From: Wei Liu <[email protected]>

Add a helper for HVCALL_ISSUE_SNP_PSP_GUEST_REQUEST so MSHV can
issue asynchronous PSP guest requests on behalf of encrypted
partitions.

Signed-off-by: Wei Liu <[email protected]>
---
 drivers/hv/mshv_root.h         |  8 +++++++
 drivers/hv/mshv_root_hv_call.c | 39 ++++++++++++++++++++++++++++++++++
 2 files changed, 47 insertions(+)

diff --git a/drivers/hv/mshv_root.h b/drivers/hv/mshv_root.h
index 1f086dcb7aa1..8abb080938fd 100644
--- a/drivers/hv/mshv_root.h
+++ b/drivers/hv/mshv_root.h
@@ -378,4 +378,12 @@ bool mshv_region_handle_gfn_fault(struct mshv_mem_region 
*region, u64 gfn);
 void mshv_region_movable_fini(struct mshv_mem_region *region);
 bool mshv_region_movable_init(struct mshv_mem_region *region);
 
+#ifdef HV_SUPPORTS_SEV_SNP_GUESTS
+int hv_call_issue_psp_guest_request(u64 partition_id, u64 req_pfn,
+                                   u64 rsp_pfn,
+                                   void (*completion_handler)(void *data,
+                                                              u64 *status),
+                                   void *completion_data);
+#endif
+
 #endif /* _MSHV_ROOT_H_ */
diff --git a/drivers/hv/mshv_root_hv_call.c b/drivers/hv/mshv_root_hv_call.c
index cb55d4d4be2e..48abe338c06f 100644
--- a/drivers/hv/mshv_root_hv_call.c
+++ b/drivers/hv/mshv_root_hv_call.c
@@ -1009,6 +1009,45 @@ int hv_unmap_stats_page(enum hv_stats_object_type type,
        return ret;
 }
 
+#ifdef HV_SUPPORTS_SEV_SNP_GUESTS
+int hv_call_issue_psp_guest_request(u64 partition_id, u64 req_pfn,
+                                   u64 rsp_pfn,
+                                   void (*completion_handler)(void *data,
+                                                              u64 *status),
+                                   void *completion_data)
+{
+       struct hv_input_issue_psp_guest_request *input;
+       unsigned long flags;
+       u64 status;
+
+       if (!completion_handler) {
+               pr_err("%s: missing completion handler\n", __func__);
+               return -EINVAL;
+       }
+
+       local_irq_save(flags);
+       input = *this_cpu_ptr(hyperv_pcpu_input_arg);
+       memset(input, 0, sizeof(*input));
+       input->partition_id = partition_id;
+       input->request_page = req_pfn;
+       input->response_page = rsp_pfn;
+       status = hv_do_hypercall(HVCALL_ISSUE_SNP_PSP_GUEST_REQUEST, input,
+                                NULL);
+       local_irq_restore(flags);
+
+       if (hv_result(status) == HV_STATUS_CALL_PENDING)
+               completion_handler(completion_data, &status);
+
+       if (!hv_result_success(status)) {
+               pr_err("%s: status=%s partition_id=%llu\n", __func__,
+                      hv_result_to_string(status), partition_id);
+               return hv_result_to_errno(status);
+       }
+
+       return 0;
+}
+#endif
+
 int hv_call_modify_spa_host_access(u64 partition_id, struct page **pages,
                                   u64 page_struct_count, u32 host_access,
                                   u32 flags, u8 acquire)
-- 
2.43.0


Reply via email to