From: Michael Roth <[email protected]>

Make the source page for populating an SNP guest_memfd instance optional
if in-place conversion/population is enabled.  If KVM can convert the page
in-place, then it's possible for guest memory to be initialized directly
from userspace by mmap()'ing the guest_memfd and writing to it while the
corresponding GPA ranges are in a 'shared' state, before converting them
to the 'private' state expected by KVM_SEV_SNP_LAUNCH_UPDATE.

Update the handling/documentation for KVM_SEV_SNP_LAUNCH_UPDATE to allow
for 'uaddr' to be set to NULL when in-place conversion is enabled, which
SNP_LAUNCH_UPDATE will then use to determine when it should/shouldn't
copy in data from a separate memory location. Continue to enforce
non-NULL when PRIVATE is tracked per-VM, not per-guest_memfd.

Signed-off-by: Michael Roth <[email protected]>
[Added src_page check in error handling path when the firmware command fails]
[Dropped ifdef CONFIG_KVM_VM_MEMORY_ATTRIBUTES]
Signed-off-by: Ackerley Tng <[email protected]>
[sean: drop explicit vm_memory_attributes references]
Signed-off-by: Sean Christopherson <[email protected]>
---
 Documentation/virt/kvm/x86/amd-memory-encryption.rst | 13 +++++++++----
 arch/x86/kvm/svm/sev.c                               | 16 +++++++++++-----
 virt/kvm/kvm_main.c                                  |  1 +
 3 files changed, 21 insertions(+), 9 deletions(-)

diff --git a/Documentation/virt/kvm/x86/amd-memory-encryption.rst 
b/Documentation/virt/kvm/x86/amd-memory-encryption.rst
index bd04a908a8dbd..29409297f1ef0 100644
--- a/Documentation/virt/kvm/x86/amd-memory-encryption.rst
+++ b/Documentation/virt/kvm/x86/amd-memory-encryption.rst
@@ -503,7 +503,8 @@ secrets.
 
 It is required that the GPA ranges initialized by this command have had the
 KVM_MEMORY_ATTRIBUTE_PRIVATE attribute set in advance. See the documentation
-for KVM_SET_MEMORY_ATTRIBUTES for more details on this aspect.
+for KVM_SET_MEMORY_ATTRIBUTES/KVM_SET_MEMORY_ATTRIBUTES2 for more details on
+this aspect.
 
 Upon success, this command is not guaranteed to have processed the entire
 range requested. Instead, the ``gfn_start``, ``uaddr``, and ``len`` fields of
@@ -511,9 +512,13 @@ range requested. Instead, the ``gfn_start``, ``uaddr``, 
and ``len`` fields of
 remaining range that has yet to be processed. The caller should continue
 calling this command until those fields indicate the entire range has been
 processed, e.g. ``len`` is 0, ``gfn_start`` is equal to the last GFN in the
-range plus 1, and ``uaddr`` is the last byte of the userspace-provided source
-buffer address plus 1. In the case where ``type`` is 
KVM_SEV_SNP_PAGE_TYPE_ZERO,
-``uaddr`` will be ignored completely.
+range plus 1, and ``uaddr`` (if specified) is the last byte of the
+userspace-provided source buffer address plus 1.
+
+In the case where ``type`` is KVM_SEV_SNP_PAGE_TYPE_ZERO, ``uaddr`` will be
+ignored completely. For all other page types, ``uaddr`` is optional if in-place
+conversion is enable, i.e. when the destination can also be the source, and is
+required if in-place conversion is disabled.
 
 Parameters (in): struct  kvm_sev_snp_launch_update
 
diff --git a/arch/x86/kvm/svm/sev.c b/arch/x86/kvm/svm/sev.c
index 74fb15551e83f..2b7569b6a8609 100644
--- a/arch/x86/kvm/svm/sev.c
+++ b/arch/x86/kvm/svm/sev.c
@@ -2330,7 +2330,13 @@ static int sev_gmem_post_populate(struct kvm *kvm, gfn_t 
gfn, kvm_pfn_t pfn,
        int level;
        int ret;
 
-       if (WARN_ON_ONCE(sev_populate_args->type != KVM_SEV_SNP_PAGE_TYPE_ZERO 
&& !src_page))
+       /*
+        * A source page is required if in-place conversion isn't enabled, as
+        * the data needs to come from a separate physical page.  Zero pages
+        * are exempt as they don't consume a source page.
+        */
+       if (!gmem_in_place_conversion &&
+           sev_populate_args->type != KVM_SEV_SNP_PAGE_TYPE_ZERO && !src_page)
                return -EINVAL;
 
        ret = snp_lookup_rmpentry((u64)pfn, &assigned, &level);
@@ -2377,7 +2383,7 @@ static int sev_gmem_post_populate(struct kvm *kvm, gfn_t 
gfn, kvm_pfn_t pfn,
         */
        if (ret && !snp_page_reclaim(kvm, pfn) &&
            sev_populate_args->type == KVM_SEV_SNP_PAGE_TYPE_CPUID &&
-           sev_populate_args->fw_error == SEV_RET_INVALID_PARAM) {
+           sev_populate_args->fw_error == SEV_RET_INVALID_PARAM && src_page) {
                void *src_vaddr = kmap_local_page(src_page);
                void *dst_vaddr = kmap_local_pfn(pfn);
 
@@ -2410,8 +2416,8 @@ static int snp_launch_update(struct kvm *kvm, struct 
kvm_sev_cmd *argp)
        if (copy_from_user(&params, u64_to_user_ptr(argp->data), 
sizeof(params)))
                return -EFAULT;
 
-       pr_debug("%s: GFN start 0x%llx length 0x%llx type %d flags %d\n", 
__func__,
-                params.gfn_start, params.len, params.type, params.flags);
+       pr_debug("%s: GFN start 0x%llx length 0x%llx type %d flags %d src 
%llx\n", __func__,
+                params.gfn_start, params.len, params.type, params.flags, 
params.uaddr);
 
        if (!params.len || !PAGE_ALIGNED(params.len) || params.flags ||
            (params.type != KVM_SEV_SNP_PAGE_TYPE_NORMAL &&
@@ -2468,7 +2474,7 @@ static int snp_launch_update(struct kvm *kvm, struct 
kvm_sev_cmd *argp)
 
        params.gfn_start += count;
        params.len -= count * PAGE_SIZE;
-       if (params.type != KVM_SEV_SNP_PAGE_TYPE_ZERO)
+       if (src && params.type != KVM_SEV_SNP_PAGE_TYPE_ZERO)
                params.uaddr += count * PAGE_SIZE;
 
        if (copy_to_user(u64_to_user_ptr(argp->data), &params, sizeof(params)))
diff --git a/virt/kvm/kvm_main.c b/virt/kvm/kvm_main.c
index 044486f128c37..dd1d18a1d2f68 100644
--- a/virt/kvm/kvm_main.c
+++ b/virt/kvm/kvm_main.c
@@ -103,6 +103,7 @@ module_param(allow_unsafe_mappings, bool, 0444);
 
 #ifdef kvm_arch_has_private_mem
 bool __ro_after_init gmem_in_place_conversion = false;
+EXPORT_SYMBOL_FOR_KVM_INTERNAL(gmem_in_place_conversion);
 #endif
 
 #define MEMORY_ATTRIBUTES_MATCH(one, two)                              \

-- 
2.55.0.rc0.738.g0c8ab3ebcc-goog



Reply via email to