x86/hyperv: Add ghcb hvcall support for SNP VM

hyperv provides ghcb hvcall to handle VMBus
HVCALL_SIGNAL_EVENT and HVCALL_POST_MESSAGE
msg in SNP Isolation VM. Add such support.

Reviewed-by: Michael Kelley <mikelley@microsoft.com>
Signed-off-by: Tianyu Lan <Tianyu.Lan@microsoft.com>
Link: https://lore.kernel.org/r/20211025122116.264793-8-ltykernel@gmail.com
Signed-off-by: Wei Liu <wei.liu@kernel.org>

diff --git a/arch/x86/hyperv/ivm.c b/arch/x86/hyperv/ivm.c
index 9c48d6e2d8b2a307b0bfb45eca6d026cefae9f1b..4d012fd9d95d91ebecc30aa83c555369fc41a174 100644 (file)
--- a/arch/x86/hyperv/ivm.c
+++ b/arch/x86/hyperv/ivm.c
@@ -19,10 +19,85 @@
 #include <asm/hypervisor.h>
 
 #ifdef CONFIG_AMD_MEM_ENCRYPT
+
+#define GHCB_USAGE_HYPERV_CALL 1
+
 union hv_ghcb {
        struct ghcb ghcb;
+       struct {
+               u64 hypercalldata[509];
+               u64 outputgpa;
+               union {
+                       union {
+                               struct {
+                                       u32 callcode        : 16;
+                                       u32 isfast          : 1;
+                                       u32 reserved1       : 14;
+                                       u32 isnested        : 1;
+                                       u32 countofelements : 12;
+                                       u32 reserved2       : 4;
+                                       u32 repstartindex   : 12;
+                                       u32 reserved3       : 4;
+                               };
+                               u64 asuint64;
+                       } hypercallinput;
+                       union {
+                               struct {
+                                       u16 callstatus;
+                                       u16 reserved1;
+                                       u32 elementsprocessed : 12;
+                                       u32 reserved2         : 20;
+                               };
+                               u64 asunit64;
+                       } hypercalloutput;
+               };
+               u64 reserved2;
+       } hypercall;
 } __packed __aligned(HV_HYP_PAGE_SIZE);
 
+u64 hv_ghcb_hypercall(u64 control, void *input, void *output, u32 input_size)
+{
+       union hv_ghcb *hv_ghcb;
+       void **ghcb_base;
+       unsigned long flags;
+       u64 status;
+
+       if (!hv_ghcb_pg)
+               return -EFAULT;
+
+       WARN_ON(in_nmi());
+
+       local_irq_save(flags);
+       ghcb_base = (void **)this_cpu_ptr(hv_ghcb_pg);
+       hv_ghcb = (union hv_ghcb *)*ghcb_base;
+       if (!hv_ghcb) {
+               local_irq_restore(flags);
+               return -EFAULT;
+       }
+
+       hv_ghcb->ghcb.protocol_version = GHCB_PROTOCOL_MAX;
+       hv_ghcb->ghcb.ghcb_usage = GHCB_USAGE_HYPERV_CALL;
+
+       hv_ghcb->hypercall.outputgpa = (u64)output;
+       hv_ghcb->hypercall.hypercallinput.asuint64 = 0;
+       hv_ghcb->hypercall.hypercallinput.callcode = control;
+
+       if (input_size)
+               memcpy(hv_ghcb->hypercall.hypercalldata, input, input_size);
+
+       VMGEXIT();
+
+       hv_ghcb->ghcb.ghcb_usage = 0xffffffff;
+       memset(hv_ghcb->ghcb.save.valid_bitmap, 0,
+              sizeof(hv_ghcb->ghcb.save.valid_bitmap));
+
+       status = hv_ghcb->hypercall.hypercalloutput.callstatus;
+
+       local_irq_restore(flags);
+
+       return status;
+}
+
 void hv_ghcb_msr_write(u64 msr, u64 value)
 {
        union hv_ghcb *hv_ghcb;

diff --git a/drivers/hv/connection.c b/drivers/hv/connection.c
index 5e479d54918cf4836683c09115153f0c255f54be..8820ae68f20fb30de2a8a73da40c44bef9b37712 100644 (file)
--- a/drivers/hv/connection.c
+++ b/drivers/hv/connection.c
@@ -447,6 +447,10 @@ void vmbus_set_event(struct vmbus_channel *channel)
 
        ++channel->sig_events;
 
-       hv_do_fast_hypercall8(HVCALL_SIGNAL_EVENT, channel->sig_event);
+       if (hv_isolation_type_snp())
+               hv_ghcb_hypercall(HVCALL_SIGNAL_EVENT, &channel->sig_event,
+                               NULL, sizeof(channel->sig_event));
+       else
+               hv_do_fast_hypercall8(HVCALL_SIGNAL_EVENT, channel->sig_event);
 }
 EXPORT_SYMBOL_GPL(vmbus_set_event);

diff --git a/drivers/hv/hv.c b/drivers/hv/hv.c
index 943392db9e8a64cd1a3117347c4492c15e756a79..4d6480d57546def690e7fe553b7421e7f6e505a8 100644 (file)
--- a/drivers/hv/hv.c
+++ b/drivers/hv/hv.c
@@ -98,7 +98,13 @@ int hv_post_message(union hv_connection_id connection_id,
        aligned_msg->payload_size = payload_size;
        memcpy((void *)aligned_msg->payload, payload, payload_size);
 
-       status = hv_do_hypercall(HVCALL_POST_MESSAGE, aligned_msg, NULL);
+       if (hv_isolation_type_snp())
+               status = hv_ghcb_hypercall(HVCALL_POST_MESSAGE,
+                               (void *)aligned_msg, NULL,
+                               sizeof(*aligned_msg));
+       else
+               status = hv_do_hypercall(HVCALL_POST_MESSAGE,
+                               aligned_msg, NULL);
 
        /* Preemption must remain disabled until after the hypercall
         * so some other thread can't get scheduled onto this cpu and

diff --git a/drivers/hv/hv_common.c b/drivers/hv/hv_common.c
index 1fc82d237161b9595161eae0e929d90eab5550db..7be173a99f276e8a8ae95bb58d5ee4ba0c15c2ae 100644 (file)
--- a/drivers/hv/hv_common.c
+++ b/drivers/hv/hv_common.c
@@ -289,3 +289,9 @@ void __weak hyperv_cleanup(void)
 {
 }
 EXPORT_SYMBOL_GPL(hyperv_cleanup);
+
+u64 __weak hv_ghcb_hypercall(u64 control, void *input, void *output, u32 input_size)
+{
+       return HV_STATUS_INVALID_PARAMETER;
+}
+EXPORT_SYMBOL_GPL(hv_ghcb_hypercall);

diff --git a/include/asm-generic/mshyperv.h b/include/asm-generic/mshyperv.h
index 6d3ba902ebb0cbd749429f62be34800c5880b80b..3e2248ac328ebd3b22b21a4ef797bcd9a3c8f1f5 100644 (file)
--- a/include/asm-generic/mshyperv.h
+++ b/include/asm-generic/mshyperv.h
@@ -266,6 +266,7 @@ bool hv_is_hibernation_supported(void);
 enum hv_isolation_type hv_get_isolation_type(void);
 bool hv_is_isolation_supported(void);
 bool hv_isolation_type_snp(void);
+u64 hv_ghcb_hypercall(u64 control, void *input, void *output, u32 input_size);
 void hyperv_cleanup(void);
 bool hv_query_ext_cap(u64 cap_query);
 #else /* CONFIG_HYPERV */