On Fri, Oct 06, 2017 at 08:06:07PM -0500, Brijesh Singh wrote: > The SEV_PDH_CERT_EXPORT command can be used to export the PDH and its > certificate chain. The command is defined in SEV spec section 5.10. > > Cc: Paolo Bonzini <pbonzini@xxxxxxxxxx> > Cc: "Radim Krčmář" <rkrcmar@xxxxxxxxxx> > Cc: Borislav Petkov <bp@xxxxxxx> > Cc: Herbert Xu <herbert@xxxxxxxxxxxxxxxxxxx> > Cc: Gary Hook <gary.hook@xxxxxxx> > Cc: Tom Lendacky <thomas.lendacky@xxxxxxx> > Cc: linux-crypto@xxxxxxxxxxxxxxx > Cc: kvm@xxxxxxxxxxxxxxx > Cc: linux-kernel@xxxxxxxxxxxxxxx > Signed-off-by: Brijesh Singh <brijesh.singh@xxxxxxx> > --- > drivers/crypto/ccp/psp-dev.c | 110 +++++++++++++++++++++++++++++++++++++++++++ > 1 file changed, 110 insertions(+) > > diff --git a/drivers/crypto/ccp/psp-dev.c b/drivers/crypto/ccp/psp-dev.c > index 861c44bf2910..0a069e3c7b8c 100644 > --- a/drivers/crypto/ccp/psp-dev.c > +++ b/drivers/crypto/ccp/psp-dev.c > @@ -473,6 +473,112 @@ static int sev_ioctl_pek_cert_import(struct sev_issue_cmd *argp) > return ret; > } > > +static int sev_ioctl_pdh_cert_export(struct sev_issue_cmd *argp) > +{ > + struct sev_user_data_pdh_cert_export input; > + struct sev_data_pdh_cert_export *data; > + int ret, state, need_shutdown = 0; > + void *pdh_blob, *cert_blob; > + > + if (copy_from_user(&input, (void __user *)(uintptr_t)argp->data, > + sizeof(struct sev_user_data_pdh_cert_export))) > + return -EFAULT; > + > + data = kzalloc(sizeof(*data), GFP_KERNEL); > + if (!data) > + return -ENOMEM; > + > + pdh_blob = NULL; > + if (input.pdh_cert_address) { Here... > + if (!access_ok(VERIFY_WRITE, input.pdh_cert_address, input.pdh_cert_len) || > + (input.pdh_cert_len > SEV_FW_BLOB_MAX_SIZE)) { > + ret = -EFAULT; > + goto e_free; > + } > + > + pdh_blob = kmalloc(input.pdh_cert_len, GFP_KERNEL); > + if (!pdh_blob) { > + ret = -ENOMEM; > + goto e_free; > + } > + > + data->pdh_cert_address = __psp_pa(pdh_blob); > + data->pdh_cert_len = input.pdh_cert_len; > + } > + > + cert_blob = NULL; > + if (input.cert_chain_address) { ... and here please check the full condition where userspace queries the cert length, like before. Otherwise, the usual cleanups: --- diff --git a/drivers/crypto/ccp/psp-dev.c b/drivers/crypto/ccp/psp-dev.c index 5e69add4fea9..331d028f9445 100644 --- a/drivers/crypto/ccp/psp-dev.c +++ b/drivers/crypto/ccp/psp-dev.c @@ -476,8 +476,7 @@ static int sev_ioctl_pdh_cert_export(struct sev_issue_cmd *argp) int ret, state, need_shutdown = 0; void *pdh_blob, *cert_blob; - if (copy_from_user(&input, (void __user *)(uintptr_t)argp->data, - sizeof(struct sev_user_data_pdh_cert_export))) + if (copy_from_user(&input, (void __user *)argp->data, sizeof(input))) return -EFAULT; data = kzalloc(sizeof(*data), GFP_KERNEL); @@ -536,36 +535,37 @@ static int sev_ioctl_pdh_cert_export(struct sev_issue_cmd *argp) need_shutdown = 1; } - ret = sev_handle_cmd(SEV_CMD_PDH_CERT_EXPORT, data, &argp->error); + ret = sev_do_cmd(SEV_CMD_PDH_CERT_EXPORT, data, &argp->error); input.cert_chain_len = data->cert_chain_len; input.pdh_cert_len = data->pdh_cert_len; /* copy certificate length to userspace */ - if (copy_to_user((void __user *)(uintptr_t)argp->data, &input, - sizeof(struct sev_user_data_pdh_cert_export))) + if (copy_to_user((void __user *)argp->data, &input, sizeof(input))) ret = -EFAULT; if (ret) goto e_shutdown; /* copy PDH certificate to userspace */ - if (pdh_blob && - copy_to_user((void __user *)(uintptr_t)input.pdh_cert_address, - pdh_blob, input.pdh_cert_len)) { - ret = -EFAULT; - goto e_shutdown; + if (pdh_blob) { + if (copy_to_user((void __user *)input.pdh_cert_address, + pdh_blob, input.pdh_cert_len)) { + ret = -EFAULT; + goto e_shutdown; + } } /* copy certificate chain to userspace */ - if (cert_blob && - copy_to_user((void __user *)(uintptr_t)input.cert_chain_address, - cert_blob, input.cert_chain_len)) - ret = -EFAULT; + if (cert_blob) { + if (copy_to_user((void __user *)input.cert_chain_address, + cert_blob, input.cert_chain_len)) + ret = -EFAULT; + } e_shutdown: if (need_shutdown) - sev_handle_cmd(SEV_CMD_SHUTDOWN, 0, NULL); + sev_do_cmd(SEV_CMD_SHUTDOWN, 0, NULL); e_free_cert: kfree(cert_blob); e_free_pdh: @@ -616,10 +616,9 @@ static long sev_ioctl(struct file *file, unsigned int ioctl, unsigned long arg) ret = sev_ioctl_pek_cert_import(&input); break; - case SEV_PDH_CERT_EXPORT: { + case SEV_PDH_CERT_EXPORT: ret = sev_ioctl_pdh_cert_export(&input); break; - } default: ret = -EINVAL; -- Regards/Gruss, Boris. SUSE Linux GmbH, GF: Felix Imendörffer, Jane Smithard, Graham Norton, HRB 21284 (AG Nürnberg) --