INIT_VCPU ioctl

Implement the KVM_TDX_INIT_VCPU ioctl.

Signed-off-by: Jake Correnti <[email protected]>

Cargo.toml

@@ -14,4 +14,6 @@ rust-version = "1.71"
 
 [dependencies]
 bitflags = "2.4.2"
+kvm-bindings = "0.7.0"
 kvm-ioctls = "0.16.0"
+vmm-sys-util = "0.12.1"

src/vcpu/linux/mod.rs

@@ -1 +1,3 @@
 // SPDX-License-Identifier: Apache-2.0
+
+pub mod ioctl;

src/vcpu/mod.rs

@@ -1 +1,27 @@
 // SPDX-License-Identifier: Apache-2.0
+
+mod linux;
+
+use crate::linux::{Cmd, TdxError};
+use kvm_bindings::*;
+use vmm_sys_util::*;
+
+vmm_sys_util::ioctl_iowr_nr!(KVM_MEMORY_ENCRYPT_OP, KVMIO, 0xba, std::os::raw::c_ulong);
+
+impl crate::vm::TdxVm {
+    /// TDX specific VCPU initialization using a TDVF HOB address
+    pub fn init_vcpu(vcpu: &kvm_ioctls::VcpuFd, hob_addr: u64) -> Result<(), TdxError> {
+        let mut cmd = Cmd {
+            id: 2,
+            flags: 0,
+            data: hob_addr as *const u64 as _,
+            error: 0,
+            _unused: 0,
+        };
+        let ret = unsafe { ioctl::ioctl_with_mut_ptr(vcpu, KVM_MEMORY_ENCRYPT_OP(), &mut cmd) };
+        if ret < 0 {
+            return Err(TdxError::from(ret));
+        }
+        Ok(())
+    }
+}