123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375 |
- /* -*- mode:c; c-file-style:"k&r"; c-basic-offset: 4; tab-width:4; indent-tabs-mode:nil; mode:auto-fill; fill-column:78; -*- */
- /* vim: set ts=4 sw=4 et tw=78 fo=cqt wm=0: */
- #include <pal_linux.h>
- #include <pal_rtld.h>
- #include "sgx_internal.h"
- #include "sgx_arch.h"
- #include "sgx_enclave.h"
- #include "graphene-sgx.h"
- #include <asm/errno.h>
- int gsgx_device = -1;
- int isgx_device = -1;
- #define ISGX_FILE "/dev/isgx"
- void * zero_page;
- int open_gsgx(void)
- {
- gsgx_device = INLINE_SYSCALL(open, 3, GSGX_FILE, O_RDWR, 0);
- if (IS_ERR(gsgx_device))
- return -ERRNO(gsgx_device);
- isgx_device = INLINE_SYSCALL(open, 3, ISGX_FILE, O_RDWR, 0);
- if (IS_ERR(isgx_device))
- return -ERRNO(isgx_device);
- return 0;
- }
- int read_enclave_token(int token_file, sgx_arch_token_t * token)
- {
- struct stat stat;
- int ret;
- ret = INLINE_SYSCALL(fstat, 2, token_file, &stat);
- if (IS_ERR(ret))
- return -ERRNO(ret);
- if (stat.st_size != sizeof(sgx_arch_token_t)) {
- SGX_DBG(DBG_I, "size of token size does not match\n");
- return -EINVAL;
- }
- int bytes = INLINE_SYSCALL(read, 3, token_file, token, sizeof(sgx_arch_token_t));
- if (IS_ERR(bytes))
- return -ERRNO(bytes);
- return 0;
- }
- int read_enclave_sigstruct(int sigfile, sgx_arch_sigstruct_t * sig)
- {
- struct stat stat;
- int ret;
- ret = INLINE_SYSCALL(fstat, 2, sigfile, &stat);
- if (IS_ERR(ret))
- return -ERRNO(ret);
- if (stat.st_size < sizeof(sgx_arch_sigstruct_t)) {
- SGX_DBG(DBG_I, "size of sigstruct size does not match\n");
- return -EINVAL;
- }
- int bytes = INLINE_SYSCALL(read, 3, sigfile, sig, sizeof(sgx_arch_sigstruct_t));
- if (IS_ERR(bytes))
- return -ERRNO(bytes);
- return 0;
- }
- #define SE_LEAF 0x12
- static inline void cpuid(uint32_t leaf, uint32_t subleaf, uint32_t info[4])
- {
- asm volatile("cpuid"
- : "=a"(info[0]),
- "=b"(info[1]),
- "=c"(info[2]),
- "=d"(info[3])
- : "a"(leaf),
- "c"(subleaf));
- }
- static size_t get_ssaframesize (uint64_t xfrm)
- {
- uint32_t cpuinfo[4];
- uint64_t xfrm_ex;
- int xsave_size = 0;
- cpuid(SE_LEAF, 1, cpuinfo);
- xfrm_ex = ((uint64_t) cpuinfo[3] << 32) + cpuinfo[2];
- for (int i = 2; i < 64; i++)
- if ((xfrm & (1 << i)) || (xfrm_ex & (1 << i))) {
- cpuid(0xd, i, cpuinfo);
- if (cpuinfo[0] + cpuinfo[1] > xsave_size)
- xsave_size = cpuinfo[0] + cpuinfo[1];
- }
- return ALLOC_ALIGNUP(xsave_size + sizeof(sgx_arch_gpr_t) + 1);
- }
- int check_wrfsbase_support (void)
- {
- if (gsgx_device == -1)
- return -EACCES;
- uint32_t cpuinfo[4];
- cpuid(7, 0, cpuinfo);
- if (!(cpuinfo[1] & 0x1))
- return 0;
- return 1;
- }
- int create_enclave(sgx_arch_secs_t * secs,
- unsigned long baseaddr,
- unsigned long size,
- sgx_arch_token_t * token)
- {
- int flags = MAP_SHARED;
- if (!zero_page) {
- zero_page = (void *)
- INLINE_SYSCALL(mmap, 6, NULL, pagesize,
- PROT_READ, MAP_PRIVATE|MAP_ANONYMOUS,
- -1, 0);
- if (IS_ERR_P(zero_page))
- return -ENOMEM;
- }
- memset(secs, 0, sizeof(sgx_arch_secs_t));
- secs->size = pagesize;
- while (secs->size < size)
- secs->size <<= 1;
- secs->ssaframesize = get_ssaframesize(token->attributes.xfrm) / pagesize;
- secs->miscselect = token->miscselect_mask;
- memcpy(&secs->attributes, &token->attributes,
- sizeof(sgx_arch_attributes_t));
- memcpy(&secs->mrenclave, &token->mrenclave, sizeof(sgx_arch_hash_t));
- memcpy(&secs->mrsigner, &token->mrsigner, sizeof(sgx_arch_hash_t));
- if (baseaddr) {
- secs->baseaddr = (uint64_t) baseaddr & ~(secs->size - 1);
- } else {
- secs->baseaddr = ENCLAVE_HIGH_ADDRESS;
- }
- uint64_t addr = INLINE_SYSCALL(mmap, 6, secs->baseaddr, secs->size,
- PROT_READ|PROT_WRITE|PROT_EXEC,
- flags|MAP_FIXED, isgx_device, 0);
- if (IS_ERR_P(addr)) {
- if (ERRNO_P(addr) == 1 && (flags | MAP_FIXED))
- pal_printf("Permission denied on mapping enclave. "
- "You may need to set sysctl vm.mmap_min_addr to zero\n");
- SGX_DBG(DBG_I, "enclave ECREATE failed in allocating EPC memory "
- "(errno = %d)\n", ERRNO_P(addr));
- return -ENOMEM;
- }
- secs->baseaddr = addr;
- #if SDK_DRIVER_VERSION >= KERNEL_VERSION(1, 8, 0)
- struct sgx_enclave_create param = {
- .src = (uint64_t) secs,
- };
- int ret = INLINE_SYSCALL(ioctl, 3, isgx_device, SGX_IOC_ENCLAVE_CREATE,
- ¶m);
- #else
- struct gsgx_enclave_create param = {
- .src = (uint64_t) secs,
- };
- int ret = INLINE_SYSCALL(ioctl, 3, gsgx_device, GSGX_IOCTL_ENCLAVE_CREATE,
- ¶m);
- #endif
- if (IS_ERR(ret)) {
- SGX_DBG(DBG_I, "enclave ECREATE failed in enclave creation ioctl - %d\n", ERRNO(ret));
- return -ERRNO(ret);
- }
- if (ret) {
- SGX_DBG(DBG_I, "enclave ECREATE failed - %d\n", ret);
- return -EPERM;
- }
- secs->attributes.flags |= SGX_FLAGS_INITIALIZED;
- SGX_DBG(DBG_I, "enclave created:\n");
- SGX_DBG(DBG_I, " base: 0x%016lx\n", secs->baseaddr);
- SGX_DBG(DBG_I, " size: 0x%016lx\n", secs->size);
- SGX_DBG(DBG_I, " attr: 0x%016lx\n", secs->attributes.flags);
- SGX_DBG(DBG_I, " xfrm: 0x%016lx\n", secs->attributes.xfrm);
- SGX_DBG(DBG_I, " ssaframesize: %ld\n", secs->ssaframesize);
- SGX_DBG(DBG_I, " isvprodid: 0x%08x\n", secs->isvprodid);
- SGX_DBG(DBG_I, " isvsvn: 0x%08x\n", secs->isvsvn);
- return 0;
- }
- int add_pages_to_enclave(sgx_arch_secs_t * secs,
- void * addr, void * user_addr,
- unsigned long size,
- enum sgx_page_type type, int prot,
- bool skip_eextend,
- const char * comment)
- {
- sgx_arch_secinfo_t secinfo;
- int ret;
- memset(&secinfo, 0, sizeof(sgx_arch_secinfo_t));
- switch (type) {
- case SGX_PAGE_SECS:
- return -EPERM;
- case SGX_PAGE_TCS:
- secinfo.flags |= SGX_SECINFO_FLAGS_TCS;
- break;
- case SGX_PAGE_REG:
- secinfo.flags |= SGX_SECINFO_FLAGS_REG;
- if (prot & PROT_READ)
- secinfo.flags |= SGX_SECINFO_FLAGS_R;
- if (prot & PROT_WRITE)
- secinfo.flags |= SGX_SECINFO_FLAGS_W;
- if (prot & PROT_EXEC)
- secinfo.flags |= SGX_SECINFO_FLAGS_X;
- break;
- }
- char p[4] = "---";
- const char * t = (type == SGX_PAGE_TCS) ? "TCS" : "REG";
- const char * m = skip_eextend ? "" : " measured";
- if (type == SGX_PAGE_REG) {
- if (prot & PROT_READ)
- p[0] = 'R';
- if (prot & PROT_WRITE)
- p[1] = 'W';
- if (prot & PROT_EXEC)
- p[2] = 'X';
- }
- if (size == pagesize)
- SGX_DBG(DBG_I, "adding page to enclave: %016lx [%s:%s] (%s)%s\n",
- addr, t, p, comment, m);
- else
- SGX_DBG(DBG_I, "adding pages to enclave: %016lx-%016lx [%s:%s] (%s)%s\n",
- addr, addr + size, t, p, comment, m);
- #if SDK_DRIVER_VERSION >= KERNEL_VERSION(1, 8, 0)
- struct sgx_enclave_add_page param = {
- .addr = secs->baseaddr + (uint64_t) addr,
- .src = (uint64_t) (user_addr ? : zero_page),
- .secinfo = (uint64_t) &secinfo,
- .mrmask = skip_eextend ? 0 : (uint16_t) -1,
- };
- uint64_t added_size = 0;
- while (added_size < size) {
- ret = INLINE_SYSCALL(ioctl, 3, isgx_device,
- SGX_IOC_ENCLAVE_ADD_PAGE, ¶m);
- if (IS_ERR(ret)) {
- SGX_DBG(DBG_I, "Enclave add page returned %d\n", ret);
- return -ERRNO(ret);
- }
- param.addr += pagesize;
- if (param.src != (uint64_t) zero_page) param.src += pagesize;
- added_size += pagesize;
- }
- #else
- struct gsgx_enclave_add_pages param = {
- .addr = secs->baseaddr + (uint64_t) addr,
- .user_addr = (uint64_t) user_addr,
- .size = size,
- .secinfo = (uint64_t) &secinfo,
- .flags = skip_eextend ? GSGX_ENCLAVE_ADD_PAGES_SKIP_EEXTEND : 0,
- };
- if (!user_addr) {
- param.user_addr = (unsigned long) zero_page;
- param.flags |= GSGX_ENCLAVE_ADD_PAGES_REPEAT_SRC;
- }
- ret = INLINE_SYSCALL(ioctl, 3, gsgx_device,
- GSGX_IOCTL_ENCLAVE_ADD_PAGES,
- ¶m);
- if (IS_ERR(ret)) {
- SGX_DBG(DBG_I, "Enclave add page returned %d\n", ret);
- return -ERRNO(ret);
- }
- #endif
- return 0;
- }
- int init_enclave(sgx_arch_secs_t * secs,
- sgx_arch_sigstruct_t * sigstruct,
- sgx_arch_token_t * token)
- {
- unsigned long enclave_valid_addr =
- secs->baseaddr + secs->size - pagesize;
- SGX_DBG(DBG_I, "enclave initializing:\n");
- SGX_DBG(DBG_I, " enclave id: 0x%016lx\n", enclave_valid_addr);
- SGX_DBG(DBG_I, " enclave hash:");
- for (int i = 0 ; i < sizeof(sgx_arch_hash_t) ; i++)
- SGX_DBG(DBG_I, " %02x", sigstruct->enclave_hash[i]);
- SGX_DBG(DBG_I, "\n");
- #if SDK_DRIVER_VERSION >= KERNEL_VERSION(1, 8, 0)
- struct sgx_enclave_init param = {
- .addr = enclave_valid_addr,
- .sigstruct = (uint64_t) sigstruct,
- .einittoken = (uint64_t) token,
- };
- int ret = INLINE_SYSCALL(ioctl, 3, isgx_device, SGX_IOC_ENCLAVE_INIT,
- ¶m);
- #else
- struct gsgx_enclave_init param = {
- .addr = enclave_valid_addr,
- .sigstruct = (uint64_t) sigstruct,
- .einittoken = (uint64_t) token,
- };
- int ret = INLINE_SYSCALL(ioctl, 3, gsgx_device, GSGX_IOCTL_ENCLAVE_INIT,
- ¶m);
- #endif
- if (IS_ERR(ret)) {
- return -ERRNO(ret);
- }
- if (ret) {
- const char * error;
- /* DEP 3/22/17: Try to improve error messages */
- switch(ret) {
- case SGX_INVALID_SIG_STRUCT:
- error = "Invalid SIGSTRUCT"; break;
- case SGX_INVALID_ATTRIBUTE:
- error = "Invalid enclave attribute"; break;
- case SGX_INVALID_MEASUREMENT:
- error = "Invalid measurement"; break;
- case SGX_INVALID_SIGNATURE:
- error = "Invalid signature"; break;
- case SGX_INVALID_LICENSE:
- error = "Invalid EINIT token"; break;
- case SGX_INVALID_CPUSVN:
- error = "Invalid CPU SVN"; break;
- default:
- error = "Unknown reason"; break;
- }
- SGX_DBG(DBG_I, "enclave EINIT failed - %s\n", error);
- return -EPERM;
- }
- return 0;
- }
- int destroy_enclave(void * base_addr, size_t length)
- {
- SGX_DBG(DBG_I, "destroying enclave...\n");
- int ret = INLINE_SYSCALL(munmap, 2, base_addr, length);
- if (IS_ERR(ret)) {
- SGX_DBG(DBG_I, "enclave EDESTROY failed\n");
- return -ERRNO(ret);
- }
- return 0;
- }
|