crypto: ccp - Fix a case where SNP_SHUTDOWN is missed

If page reclaim fails in sev_ioctl_do_snp_platform_status() and SNP was
moved from UNINIT to INIT for the function, SNP is not moved back to
UNINIT state. Additionally, SNP is not required to be initialized in order
to execute the SNP_PLATFORM_STATUS command, so don't attempt to move to
INIT state and let SNP_PLATFORM_STATUS report the status as is.

Fixes: ceac7fb89e8d ("crypto: ccp - Ensure implicit SEV/SNP init and shutdown in ioctls")
Signed-off-by: Tom Lendacky <thomas.lendacky@amd.com>
Reviewed-by: Tycho Andersen (AMD) <tycho@kernel.org>
Reviewed-by: Alexey Kardashevskiy <aik@amd.com>
Signed-off-by: Tycho Andersen (AMD) <tycho@kernel.org>
Signed-off-by: Herbert Xu <herbert@gondor.apana.org.au>

diff --git a/drivers/crypto/ccp/sev-dev.c b/drivers/crypto/ccp/sev-dev.c
index 956ea609d0cce10b5ccfa2f7ab3091d41aad2c4d..6e6011e363e3bbd5a96380deb275f8f81f2389e0 100644 (file)
--- a/drivers/crypto/ccp/sev-dev.c
+++ b/drivers/crypto/ccp/sev-dev.c
@@ -2378,11 +2378,10 @@ e_free_pdh:
 static int sev_ioctl_do_snp_platform_status(struct sev_issue_cmd *argp)
 {
        struct sev_device *sev = psp_master->sev_data;
-       bool shutdown_required = false;
        struct sev_data_snp_addr buf;
        struct page *status_page;
-       int ret, error;
        void *data;
+       int ret;
 
        if (!argp->data)
                return -EINVAL;
@@ -2393,31 +2392,35 @@ static int sev_ioctl_do_snp_platform_status(struct sev_issue_cmd *argp)
 
        data = page_address(status_page);
 
-       if (!sev->snp_initialized) {
-               ret = snp_move_to_init_state(argp, &shutdown_required);
-               if (ret)
-                       goto cleanup;
-       }
-
        /*
-        * Firmware expects status page to be in firmware-owned state, otherwise
-        * it will report firmware error code INVALID_PAGE_STATE (0x1A).
+        * SNP_PLATFORM_STATUS can be executed in any SNP state. But if executed
+        * when SNP has been initialized, the status page must be firmware-owned.
         */
-       if (rmp_mark_pages_firmware(__pa(data), 1, true)) {
-               ret = -EFAULT;
-               goto cleanup;
+       if (sev->snp_initialized) {
+               /*
+                * Firmware expects the status page to be in Firmware state,
+                * otherwise it will report an error INVALID_PAGE_STATE.
+                */
+               if (rmp_mark_pages_firmware(__pa(data), 1, true)) {
+                       ret = -EFAULT;
+                       goto cleanup;
+               }
        }
 
        buf.address = __psp_pa(data);
        ret = __sev_do_cmd_locked(SEV_CMD_SNP_PLATFORM_STATUS, &buf, &argp->error);
 
-       /*
-        * Status page will be transitioned to Reclaim state upon success, or
-        * left in Firmware state in failure. Use snp_reclaim_pages() to
-        * transition either case back to Hypervisor-owned state.
-        */
-       if (snp_reclaim_pages(__pa(data), 1, true))
-               return -EFAULT;
+       if (sev->snp_initialized) {
+               /*
+                * The status page will be in Reclaim state on success, or left
+                * in Firmware state on failure. Use snp_reclaim_pages() to
+                * transition either case back to Hypervisor-owned state.
+                */
+               if (snp_reclaim_pages(__pa(data), 1, true)) {
+                       snp_leak_pages(__page_to_pfn(status_page), 1);
+                       return -EFAULT;
+               }
+       }
 
        if (ret)
                goto cleanup;
@@ -2427,9 +2430,6 @@ static int sev_ioctl_do_snp_platform_status(struct sev_issue_cmd *argp)
                ret = -EFAULT;
 
 cleanup:
-       if (shutdown_required)
-               __sev_snp_shutdown_locked(&error, false);
-
        __free_pages(status_page, 0);
        return ret;
 }