KVM: Enable cpuid check in VM and adds NIs test cases

KVM/qemu/tests/x86_cpuid.py

@@ -0,0 +1,143 @@
+#!/usr/bin/python3
+
+# SPDX-License-Identifier: GPL-2.0-only
+# Copyright (c) 2024 Intel Corporation
+
+# Author: Xudong Hao <[email protected]>
+#
+# History: Aug. 2024 - Xudong Hao - creation
+
+import os
+import sys
+from avocado.utils import process
+from avocado.core import exceptions
+from virttest import error_context, env_process
+from virttest import data_dir, asset
+from virttest import utils_package
+
+curr_dir = os.path.dirname(os.path.abspath(__file__))
+check_dir = "%s/../../../BM/instruction-check" % curr_dir
+sys.path.append(check_dir)
+from feature_list import cpuid_info
+
+
+def get_baremetal_dir(params):
+    """
+    Get the test provider's BM absolute path.
+    :param params: Dictionary with the test parameters
+    """
+    provider = params["provider"]
+    provider_info = asset.get_test_provider_info(provider)
+    if provider_info["uri"].startswith("file://"):
+        provider_dir = provider_info["uri"][7:]
+    else:
+        provider_dir = data_dir.get_test_provider_dir(provider)
+    baremetal_dir = os.path.join(provider_dir, "BM")
+
+    return baremetal_dir
+
+
+def prepare_cpuid(test, params, src_dir, vm=None, session=None):
+    """
+    Compile cpuid test tool in host or guest.
+    Return the execuable test tool with absolute path.
+    :param test: QEMU test object
+    :param params: Dictionary with the test parameters
+    :param src_dir: Test tool srouce code directory of absolute path
+    :param vm: The vm object
+    :param session: Guest session
+    """
+    source_file = params["source_file"]
+    exec_file = params["exec_file"]
+    src_cpuid = os.path.join(src_dir, source_file)
+    if session:
+        test_dir = params["test_dir"]
+        vm.copy_files_to(src_cpuid, test_dir)
+        if session.cmd_status("rpm -q gcc"):
+            if session.cmd_status("yum -y install gcc"):
+                test.cancel("Fail to install gcc in vm, please retest "
+                            "this case again.")
+    else:
+        test_dir = src_dir
+        s, o = process.getstatusoutput("rpm -q gcc", shell=True)
+        if s != 0:
+            install_status = utils_package.package_install("gcc")
+            if not install_status:
+                test.cancel("Failed to install gcc")
+
+    compile_cmd = "cd %s && gcc %s -o %s" % (test_dir, source_file, exec_file)
+    if session:
+        status = session.cmd_status(compile_cmd)
+    else:
+        status = process.system(compile_cmd, shell=True)
+    if status:
+        raise exceptions.TestError("Test suite compile failed.")
+
+    return os.path.join(test_dir, exec_file)
+
+
+def check_cpuid(params, feature, exec_bin, session=None):
+    """
+    Run cpuid test in host or guest
+    :param params: Dictionary with the test parameters
+    :param feature: The feature to be checked
+    :param exec_bin: The execuable bianry tool with absolute path
+    :param session: Guest session
+    """
+    args = cpuid_info[feature]
+    args_str = ' '.join(args)
+
+    check_cmd = '%s %s' % (exec_bin, args_str)
+    func = process.getstatusoutput
+    if session:
+        func = session.cmd_status_output
+    s, o = func(check_cmd)
+
+    return s
+
+
+@error_context.context_aware
+def run(test, params, env):
+    """
+    Check cpuid inside guest.
+    1) Check if current cpuid are support in host, if no, cancel test
+    2) Boot up guest
+    3) Check cpuid/s in guest
+    4) Shutdown guest
+    :param test: QEMU test object
+    :param params: Dictionary with the test parameters
+    :param env: Dictionary with test environment.
+    """
+    test_dir = params["test_dir"]
+    bm_dir = get_baremetal_dir(params)
+    src_dir = "%s/tools/cpuid_check" % bm_dir
+
+    check_host_cpuid = params.get_boolean("check_host_cpuid")
+    if check_host_cpuid:
+        host_exec_bin = prepare_cpuid(test, params, src_dir)
+
+        feature_names = params["features"]
+        for feature_name in feature_names.split():
+            host_status = check_cpuid(params, feature_name, host_exec_bin)
+            if host_status:
+                process.system("rm %s -f" % host_exec_bin, shell=True, ignore_status=True)
+                test.cancel("Platform doesn't support %s" % feature_name)
+        test.log.info("Host cpuid check %s pass.", feature_names)
+
+    try:
+        params["start_vm"] = "yes"
+        vm_name = params['main_vm']
+        env_process.preprocess_vm(test, params, env, vm_name)
+        vm = env.get_vm(vm_name)
+        session = vm.wait_for_login()
+
+        vm_exec_bin = prepare_cpuid(test, params, src_dir, vm, session)
+        for feature_name in feature_names.split():
+            guest_status = check_cpuid(params, feature_name, vm_exec_bin, session)
+            if guest_status:
+                test.fail("%s cpuid check fail in guest." % feature_name)
+        test.log.info("Guest cpuid check %s pass.", feature_names)
+    finally:
+        process.system("rm %s -f" % host_exec_bin, shell=True, ignore_status=True)
+        session.cmd("rm %s/cpuid* -rf" % test_dir, ignore_all_errors=True)
+        session.close()

KVM/qemu/x86_cpuid.cfg

@@ -0,0 +1,45 @@
+- x86_cpuid:
+    type = x86_cpuid
+    kill_vm_on_error = yes
+    start_vm = no
+    check_host_cpuid= yes
+    vga = std
+    test_dir = '/home'
+    source_file = 'cpuid_check.c'
+    exec_file = 'cpuid_check'
+    auto_cpu_model = "no"
+    cpu_model = host
+    variants:
+        - avx512_fp16:
+            # The name should be same as the KEY/s in cpuid_info dictionary
+            # "BM/instruction-check/feature_list.py"
+            features = "AVX512_FP16"
+        - serialize:
+            features = "SERIALIZE"
+        - tsxldtrk:
+            features = "TSXLDTRK"
+        - avx_vnni:
+            features = "AVX_VNNI"
+        - amx_bf16:
+            features = "AMX_BF16"
+        - amx_int8:
+            features = "AMX_INT8"
+        - amx_tile:
+            features = "AMX_TILE"
+        - amx_fp16:
+            features = "AMX_FP16"
+        - prefetchi:
+            features = "PREFETCHI"
+        - avx_ifma:
+            features = "AVX_IFMA"
+        - avx_vnni_int8:
+            features = "AVX_VNNI_INT8"
+        - avx_ne_convert:
+            features = "AVX_NE_CONVERT"
+        - cmpccxadd:
+            features = "CMPCCXADD"
+    variants:
+        - vm:
+        - tdvm:
+            machine_type_extra_params = "kernel-irqchip=split"
+            vm_secure_guest_type = tdx