crypto: ccp - Fix checks for SNP_VLEK_LOAD input buffer length

The SNP_VLEK_LOAD IOCTL currently fails due to sev_cmd_buffer_len()
returning the default expected buffer length of 0 instead of the correct
value, which would be sizeof(struct sev_user_data_snp_vlek_load). Add
specific handling for SNP_VLEK_LOAD so the correct expected size is
returned.

Reported-by: Diego GonzalezVillalobos <Diego.GonzalezVillalobos@amd.com>
Cc: Diego GonzalezVillalobos <Diego.GonzalezVillalobos@amd.com>
Fixes: 332d2c1d713e ("crypto: ccp: Add the SNP_VLEK_LOAD command")
Signed-off-by: Michael Roth <michael.roth@amd.com>
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 334405461657d495e5257eb1b2083e1afabb7bd6..4f000dc2e6396919107b96dbe8b883aa69695959 100644 (file)
--- a/drivers/crypto/ccp/sev-dev.c
+++ b/drivers/crypto/ccp/sev-dev.c
@@ -234,6 +234,7 @@ static int sev_cmd_buffer_len(int cmd)
        case SEV_CMD_SNP_CONFIG:                return sizeof(struct sev_user_data_snp_config);
        case SEV_CMD_SNP_COMMIT:                return sizeof(struct sev_data_snp_commit);
        case SEV_CMD_SNP_FEATURE_INFO:          return sizeof(struct sev_data_snp_feature_info);
+       case SEV_CMD_SNP_VLEK_LOAD:             return sizeof(struct sev_user_data_snp_vlek_load);
        default:                                return 0;
        }