Merge pull request #1986 from marcov/kvm-required

kata-check: require kvm modules for amd64

cli/kata-check_amd64.go

@@ -117,16 +117,20 @@ func setCPUtype(hypervisorType vc.HypervisorType) error {
 			archRequiredKernelModules = map[string]kernelModule{
 				kernelModkvm: {
 					desc:     msgKernelVM,
+					required: true,
 				},
 				kernelModkvmintel: {
 					desc:       "Intel KVM",
 					parameters: kvmIntelParams,
+					required:   true,
 				},
 				kernelModvhost: {
 					desc:     msgKernelVirtio,
+					required: true,
 				},
 				kernelModvhostnet: {
 					desc:     msgKernelVirtioNet,
+					required: true,
 				},
 				kernelModvhostvsock: {
 					desc:     msgKernelVirtioVhostVsock,
@@ -144,12 +148,15 @@ func setCPUtype(hypervisorType vc.HypervisorType) error {
 			archRequiredKernelModules = map[string]kernelModule{
 				kernelModvhm: {
 					desc:     "Intel ACRN",
+					required: false,
 				},
 				kernelModvhost: {
 					desc:     msgKernelVirtio,
+					required: false,
 				},
 				kernelModvhostnet: {
 					desc:     msgKernelVirtioNet,
+					required: false,
 				},
 			}
 		default:

cli/kata-check_test.go

@@ -854,3 +854,54 @@ func TestCheckKernelParamHandler(t *testing.T) {
 	// single error (due to "param1"'s value being different)
 	checkKernelParamHandler(assert, testDataToCreate, testDataToExpect, nil, false, uint32(1))
 }
+
+func TestArchRequiredKernelModules(t *testing.T) {
+	assert := assert.New(t)
+
+	tmpdir, err := ioutil.TempDir("", "")
+	assert.NoError(err)
+	defer os.RemoveAll(tmpdir)
+
+	_, config, err := makeRuntimeConfig(tmpdir)
+	assert.NoError(err)
+
+	err = setCPUtype(config.HypervisorType)
+	assert.NoError(err)
+
+	if len(archRequiredKernelModules) == 0 {
+		// No modules to check
+		return
+	}
+
+	dir, err := ioutil.TempDir("", "")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer os.RemoveAll(dir)
+
+	savedModProbeCmd := modProbeCmd
+	savedSysModuleDir := sysModuleDir
+
+	// XXX: override
+	sysModuleDir = filepath.Join(dir, "sys/module")
+	modProbeCmd = "false"
+
+	defer func() {
+		sysModuleDir = savedSysModuleDir
+		modProbeCmd = savedModProbeCmd
+	}()
+
+	// Running check with no modules
+	count, err := checkKernelModules(archRequiredKernelModules, nil)
+	assert.NoError(err)
+
+	// Test that count returned matches the # of modules with required set.
+	expectedCount := 0
+	for _, module := range archRequiredKernelModules {
+		if module.required {
+			expectedCount++
+		}
+	}
+
+	assert.EqualValues(count, expectedCount)
+}