vmm/linux/tee: Handle KVM_EXIT_MEMORY_FAULT exits

Like KVM_EXIT_HYPERCALL, KVM_EXIT_MEMORY_FAULT is used by SEV-SNP guests
to set memory regions to private or shared. Allow vCPUs encountering a
KVM_EXIT_MEMORY_FAULT to re-use the shared/private memory handler used
by KVM_EXIT_HYPERCALL to set regions to private or shared.

Co-developed-by: Matias Ezequiel Vara Larsen <[email protected]>
Signed-off-by: Tyler Fanelli <[email protected]>

Author: tylerfanelli

src/vmm/src/linux/vstate.rs

@@ -45,7 +45,7 @@ use kvm_bindings::{
     KVM_MEM_GUEST_MEMFD,
 };
 #[cfg(feature = "tee")]
-use kvm_bindings::{kvm_enable_cap, KVM_CAP_EXIT_HYPERCALL};
+use kvm_bindings::{kvm_enable_cap, KVM_CAP_EXIT_HYPERCALL, KVM_MEMORY_EXIT_FLAG_PRIVATE};
 use kvm_ioctls::{Cap::*, *};
 use utils::eventfd::EventFd;
 use utils::signal::{register_signal_handler, sigrtmin, Killable};
@@ -1238,6 +1238,17 @@ impl Vcpu {
                     self.io_bus.write(0, u64::from(addr), data);
                     Ok(VcpuEmulation::Handled)
                 }
+                #[cfg(feature = "tee")]
+                VcpuExit::MemoryFault { gpa, size, flags } => {
+                    let private = (flags & (KVM_MEMORY_EXIT_FLAG_PRIVATE as u64)) != 0;
+
+                    let mem_properties = MemoryProperties { gpa, size, private };
+
+                    self.pm_sender.0.send(mem_properties).unwrap();
+                    let _ = self.pm_sender.1.read().unwrap();
+
+                    Ok(VcpuEmulation::Handled)
+                }
                 VcpuExit::MmioRead(addr, data) => {
                     if let Some(ref mmio_bus) = self.mmio_bus {
                         mmio_bus.read(0, addr, data);