[qemu] Add custom kvm cpu args support

When improve `bpf_get_branch_snapshot()` helper, it requires passing LBR
into the VM like `-cpu,host,pmu=on,lbr-fmt=0x5`.

Then, in the VM, confirm the LBR is supported:

```bash
root@(none):/# dmesg | grep -i lbr
[    0.394406] Performance Events: Skylake events, 32-deep LBR, full-width counters, Intel PMU driver.
```

With this commit, the VM can be created by
`vmtest -k $(make -s image_name) --kvm-cpu-args 'host,pmu=on,lbr-fmt=0x5' -`.

Signed-off-by: Leon Hwang <[email protected]>

Author: Asphaltt

src/config.rs

@@ -100,6 +100,10 @@ pub struct Target {
     ///
     /// Arguments are only valid for kernel targets.
     pub kernel_args: Option<String>,
+    /// KVM -cpu arguments are only valid for KVM enabled targets.
+    ///
+    /// Default: host
+    pub kvm_cpu_args: Option<String>,
     /// Path to rootfs to test against.
     ///
     /// * The path is relative to `vmtest.toml`.
@@ -140,6 +144,7 @@ impl Default for Target {
             uefi: false,
             kernel: None,
             kernel_args: None,
+            kvm_cpu_args: None,
             rootfs: Self::default_rootfs(),
             arch: Self::default_arch(),
             qemu_command: None,

src/main.rs

@@ -37,6 +37,9 @@ struct Args {
     /// Additional kernel command line arguments
     #[clap(long, conflicts_with = "config")]
     kargs: Option<String>,
+    /// KVM -cpu arguments
+    #[clap(long, conflicts_with = "config", default_value = "host")]
+    kvm_cpu_args: Option<String>,
     /// Location of rootfs, default to host's /
     #[clap(short, long, conflicts_with = "config", default_value = Target::default_rootfs().into_os_string())]
     rootfs: PathBuf,
@@ -122,6 +125,7 @@ fn config(args: &Args) -> Result<Vmtest> {
                     rootfs: args.rootfs.clone(),
                     arch: args.arch.clone(),
                     kernel_args: args.kargs.clone(),
+                    kvm_cpu_args: args.kvm_cpu_args.clone(),
                     qemu_command: args.qemu_command.clone(),
                     command: args.command.join(" "),
                     vm: VMConfig::default(),

src/qemu.rs

@@ -251,13 +251,13 @@ fn guest_agent_args(sock: &Path) -> Vec<OsString> {
 }
 
 /// Generate arguments for full KVM virtualization if host supports it
-fn kvm_args(arch: &str) -> Vec<&'static str> {
+fn kvm_args<'a>(arch: &str, kvm_cpu_args: &'a str) -> Vec<&'a str> {
     let mut args = Vec::new();
 
     if host_supports_kvm(arch) {
         args.push("-enable-kvm");
         args.push("-cpu");
-        args.push("host");
+        args.push(kvm_cpu_args);
     } else {
         args.push("-cpu");
         match arch {
@@ -365,7 +365,7 @@ fn kernel_args(
     kernel: &Path,
     arch: &str,
     init: &Path,
-    additional_kargs: Option<&String>,
+    additional_kargs: Option<&str>,
 ) -> Vec<OsString> {
     let mut args = Vec::new();
 
@@ -669,7 +669,10 @@ impl Qemu {
             .stderr(Stdio::piped())
             .arg("-serial")
             .arg("mon:stdio")
-            .args(kvm_args(&target.arch))
+            .args(kvm_args(
+                &target.arch,
+                target.kvm_cpu_args.as_deref().unwrap_or("host"),
+            ))
             .args(machine_args(&target.arch))
             .args(machine_protocol_args(&qmp_sock))
             .args(guest_agent_args(&qga_sock))
@@ -691,7 +694,7 @@ impl Qemu {
                 kernel,
                 &target.arch,
                 guest_init.as_path(),
-                target.kernel_args.as_ref(),
+                target.kernel_args.as_deref(),
             ));
         } else {
             panic!("Config validation should've enforced XOR");

src/vmtest.rs

@@ -56,6 +56,13 @@ fn validate_config(config: &Config) -> Result<()> {
             );
         }
 
+        if target.kvm_cpu_args.is_none() || target.kvm_cpu_args.as_ref().unwrap().is_empty() {
+            bail!(
+                "Target '{}' must specify non-empty 'kvm_cpu_args'",
+                target.name
+            );
+        }
+
         if let Some(image) = &target.image {
             if image.as_os_str().is_empty() {
                 bail!("Target '{}' has empty image path", target.name);