Move full VM prot bits calculation to sgx_encl_page_alloc() so that we don't pass duplicate data in the add page flow (@prot and @secinfo hold intersecting data). Signed-off-by: Jarkko Sakkinen <jarkko.sakkinen@xxxxxxxxxxxxxxx> --- arch/x86/kernel/cpu/sgx/ioctl.c | 44 ++++++++++++++++----------------- 1 file changed, 21 insertions(+), 23 deletions(-) diff --git a/arch/x86/kernel/cpu/sgx/ioctl.c b/arch/x86/kernel/cpu/sgx/ioctl.c index 5423d7c45d5e..ead9fb2d9b69 100644 --- a/arch/x86/kernel/cpu/sgx/ioctl.c +++ b/arch/x86/kernel/cpu/sgx/ioctl.c @@ -130,10 +130,10 @@ static int sgx_validate_secs(const struct sgx_secs *secs, static struct sgx_encl_page *sgx_encl_page_alloc(struct sgx_encl *encl, unsigned long addr, - unsigned long prot, - u64 page_type) + u64 secinfo_flags) { struct sgx_encl_page *encl_page; + unsigned long prot; encl_page = kzalloc(sizeof(*encl_page), GFP_KERNEL); if (!encl_page) @@ -142,9 +142,22 @@ static struct sgx_encl_page *sgx_encl_page_alloc(struct sgx_encl *encl, encl_page->desc = addr; encl_page->encl = encl; - if (page_type == SGX_SECINFO_TCS) + if (secinfo_flags & SGX_SECINFO_TCS) encl_page->desc |= SGX_ENCL_PAGE_TCS; + + prot = _calc_vm_trans(secinfo_flags, SGX_SECINFO_R, PROT_READ) | + _calc_vm_trans(secinfo_flags, SGX_SECINFO_W, PROT_WRITE) | + _calc_vm_trans(secinfo_flags, SGX_SECINFO_X, PROT_EXEC); + + /* + * TCS pages must always RW set for CPU access while the SECINFO + * permissions are *always* zero - the CPU ignores the user provided + * values and silently overwrites them with zero permissions. + */ + if ((secinfo_flags & SGX_SECINFO_PAGE_TYPE_MASK) == SGX_SECINFO_TCS) + prot |= PROT_READ | PROT_WRITE; + /* Calculate maximum of the VM flags for the page. */ encl_page->vm_max_prot_bits = calc_vm_prot_bits(prot, 0); @@ -318,7 +331,7 @@ static int __sgx_encl_add_page(struct sgx_encl *encl, struct sgx_encl_page *encl_page, struct sgx_epc_page *epc_page, struct sgx_secinfo *secinfo, unsigned long src, - unsigned long prot, unsigned long mrmask) + unsigned long mrmask) { struct sgx_pageinfo pginfo; struct vm_area_struct *vma; @@ -375,15 +388,14 @@ static int __sgx_encl_add_page(struct sgx_encl *encl, static int sgx_encl_add_page(struct sgx_encl *encl, struct sgx_enclave_add_page *addp, - struct sgx_secinfo *secinfo, unsigned long prot) + struct sgx_secinfo *secinfo) { - u64 page_type = secinfo->flags & SGX_SECINFO_PAGE_TYPE_MASK; struct sgx_encl_page *encl_page; struct sgx_epc_page *epc_page; struct sgx_va_page *va_page; int ret; - encl_page = sgx_encl_page_alloc(encl, addp->addr, prot, page_type); + encl_page = sgx_encl_page_alloc(encl, addp->addr, secinfo->flags); if (IS_ERR(encl_page)) return PTR_ERR(encl_page); @@ -407,7 +419,7 @@ static int sgx_encl_add_page(struct sgx_encl *encl, goto err_out_shrink; ret = __sgx_encl_add_page(encl, encl_page, epc_page, secinfo, - addp->src, prot, addp->mrmask); + addp->src, addp->mrmask); if (ret) goto err_out; @@ -450,7 +462,6 @@ static long sgx_ioc_enclave_add_page(struct file *filep, void __user *arg) struct sgx_encl *encl = filep->private_data; struct sgx_enclave_add_page addp; struct sgx_secinfo secinfo; - unsigned long prot; if (!(encl->flags & SGX_ENCL_CREATED)) return -EINVAL; @@ -472,20 +483,7 @@ static long sgx_ioc_enclave_add_page(struct file *filep, void __user *arg) if (sgx_validate_secinfo(&secinfo)) return -EINVAL; - /* Set prot bits matching to the SECINFO permissions. */ - prot = _calc_vm_trans(secinfo.flags, SGX_SECINFO_R, PROT_READ) | - _calc_vm_trans(secinfo.flags, SGX_SECINFO_W, PROT_WRITE) | - _calc_vm_trans(secinfo.flags, SGX_SECINFO_X, PROT_EXEC); - - /* - * TCS pages must always RW set for CPU access while the SECINFO - * permissions are *always* zero - the CPU ignores the user provided - * values and silently overwrites with zero permissions. - */ - if ((secinfo.flags & SGX_SECINFO_PAGE_TYPE_MASK) == SGX_SECINFO_TCS) - prot |= PROT_READ | PROT_WRITE; - - return sgx_encl_add_page(encl, &addp, &secinfo, prot); + return sgx_encl_add_page(encl, &addp, &secinfo); } static int __sgx_get_key_hash(struct crypto_shash *tfm, const void *modulus, -- 2.20.1