KVM: pVMX: Implement setup_mce operation

Enable the pKVM VMX host to setup guest MCE via the setup_mce PV
interface.

Signed-off-by: Chuanxiao Dong <[email protected]>

Author: cxdong

arch/x86/kvm/vmx/pkvm_host.c

@@ -1016,6 +1016,11 @@ static struct kvm_x86_nested_ops pkvm_nested_ops = {
 	.write_log_dirty = pkvm_nested_write_pml_buffer,
 };
 
+static void pkvm_setup_mce(struct kvm_vcpu *vcpu)
+{
+	KVM_BUG_ON(pkvm_hypercall(setup_mce), vcpu->kvm);
+}
+
 static bool pkvm_apic_init_signal_blocked(struct kvm_vcpu *vcpu)
 {
 	/*
@@ -1119,6 +1124,8 @@ struct kvm_x86_ops pkvm_host_vt_x86_ops __initdata = {
 	.pi_update_irte = vmx_pi_update_irte,
 	.pi_start_bypass = vmx_pi_start_bypass,
 
+	.setup_mce = pkvm_setup_mce,
+
 	.apic_init_signal_blocked = pkvm_apic_init_signal_blocked,
 
 	.vcpu_deliver_sipi_vector = kvm_vcpu_deliver_sipi_vector,