sgx_enclave.c 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753
  1. #include "ocall_types.h"
  2. #include "ecall_types.h"
  3. #include "sgx_internal.h"
  4. #include "sgx_enclave.h"
  5. #include "pal_security.h"
  6. #include "pal_linux_error.h"
  7. #include <asm/mman.h>
  8. #include <asm/ioctls.h>
  9. #include <asm/socket.h>
  10. #include <linux/fs.h>
  11. #include <linux/in.h>
  12. #include <linux/in6.h>
  13. #include <math.h>
  14. #include <asm/errno.h>
  15. #ifndef SOL_IPV6
  16. # define SOL_IPV6 41
  17. #endif
  18. #define ODEBUG(code, ms) do {} while (0)
  19. static int sgx_ocall_exit(void* pms)
  20. {
  21. ms_ocall_exit_t * ms = (ms_ocall_exit_t *) pms;
  22. ODEBUG(OCALL_EXIT, NULL);
  23. if (ms->ms_exitcode != (int) ((uint8_t) ms->ms_exitcode)) {
  24. SGX_DBG(DBG_E, "Saturation error in exit code %d, getting rounded down to %u\n",
  25. ms->ms_exitcode, (uint8_t) ms->ms_exitcode);
  26. ms->ms_exitcode = 255;
  27. }
  28. /* exit the whole process if exit_group() */
  29. if (ms->ms_is_exitgroup)
  30. INLINE_SYSCALL(exit_group, 1, (int)ms->ms_exitcode);
  31. /* otherwise call SGX-related thread reset and exit this thread */
  32. block_async_signals(true);
  33. ecall_thread_reset();
  34. /* threads created with pthread_create() must exit with pthread_exit() */
  35. if (!unmap_tcs())
  36. INLINE_SYSCALL(exit, 1, (int)ms->ms_exitcode);
  37. else
  38. thread_exit(&ms->ms_exitcode);
  39. return 0;
  40. }
  41. static int sgx_ocall_print_string(void * pms)
  42. {
  43. ms_ocall_print_string_t * ms = (ms_ocall_print_string_t *) pms;
  44. INLINE_SYSCALL(write, 3, 2, ms->ms_str, ms->ms_length);
  45. return 0;
  46. }
  47. static int sgx_ocall_alloc_untrusted(void * pms)
  48. {
  49. ms_ocall_alloc_untrusted_t * ms = (ms_ocall_alloc_untrusted_t *) pms;
  50. void * addr;
  51. ODEBUG(OCALL_ALLOC_UNTRUSTED, ms);
  52. addr = (void *) INLINE_SYSCALL(mmap, 6, NULL, ms->ms_size,
  53. PROT_READ|PROT_WRITE,
  54. MAP_ANONYMOUS|MAP_PRIVATE, -1, 0);
  55. if (IS_ERR_P(addr))
  56. return -ERRNO_P(addr);
  57. ms->ms_mem = addr;
  58. return 0;
  59. }
  60. static int sgx_ocall_map_untrusted(void * pms)
  61. {
  62. ms_ocall_map_untrusted_t * ms = (ms_ocall_map_untrusted_t *) pms;
  63. void * addr;
  64. ODEBUG(OCALL_MAP_UNTRUSTED, ms);
  65. addr = (void *) INLINE_SYSCALL(mmap, 6, NULL, ms->ms_size,
  66. ms->ms_prot,
  67. MAP_FILE|MAP_SHARED,
  68. ms->ms_fd, ms->ms_offset);
  69. if (IS_ERR_P(addr))
  70. return -ERRNO_P(addr);
  71. ms->ms_mem = addr;
  72. return 0;
  73. }
  74. static int sgx_ocall_unmap_untrusted(void * pms)
  75. {
  76. ms_ocall_unmap_untrusted_t * ms = (ms_ocall_unmap_untrusted_t *) pms;
  77. ODEBUG(OCALL_UNMAP_UNTRUSTED, ms);
  78. INLINE_SYSCALL(munmap, 2, ALLOC_ALIGNDOWN(ms->ms_mem),
  79. ALLOC_ALIGNUP(ms->ms_mem + ms->ms_size) -
  80. ALLOC_ALIGNDOWN(ms->ms_mem));
  81. return 0;
  82. }
  83. static int sgx_ocall_cpuid(void * pms)
  84. {
  85. ms_ocall_cpuid_t * ms = (ms_ocall_cpuid_t *) pms;
  86. ODEBUG(OCALL_CPUID, ms);
  87. __asm__ volatile ("cpuid"
  88. : "=a"(ms->ms_values[0]),
  89. "=b"(ms->ms_values[1]),
  90. "=c"(ms->ms_values[2]),
  91. "=d"(ms->ms_values[3])
  92. : "a"(ms->ms_leaf), "c"(ms->ms_subleaf) : "memory");
  93. return 0;
  94. }
  95. static int sgx_ocall_open(void * pms)
  96. {
  97. ms_ocall_open_t * ms = (ms_ocall_open_t *) pms;
  98. int ret;
  99. ODEBUG(OCALL_OPEN, ms);
  100. ret = INLINE_SYSCALL(open, 3, ms->ms_pathname, ms->ms_flags|O_CLOEXEC,
  101. ms->ms_mode);
  102. return ret;
  103. }
  104. static int sgx_ocall_close(void * pms)
  105. {
  106. ms_ocall_close_t * ms = (ms_ocall_close_t *) pms;
  107. ODEBUG(OCALL_CLOSE, ms);
  108. INLINE_SYSCALL(close, 1, ms->ms_fd);
  109. return 0;
  110. }
  111. static int sgx_ocall_read(void * pms)
  112. {
  113. ms_ocall_read_t * ms = (ms_ocall_read_t *) pms;
  114. int ret;
  115. ODEBUG(OCALL_READ, ms);
  116. ret = INLINE_SYSCALL(read, 3, ms->ms_fd, ms->ms_buf, ms->ms_count);
  117. return ret;
  118. }
  119. static int sgx_ocall_write(void * pms)
  120. {
  121. ms_ocall_write_t * ms = (ms_ocall_write_t *) pms;
  122. int ret;
  123. ODEBUG(OCALL_WRITE, ms);
  124. ret = INLINE_SYSCALL(write, 3, ms->ms_fd, ms->ms_buf, ms->ms_count);
  125. return ret;
  126. }
  127. static int sgx_ocall_fstat(void * pms)
  128. {
  129. ms_ocall_fstat_t * ms = (ms_ocall_fstat_t *) pms;
  130. int ret;
  131. ODEBUG(OCALL_FSTAT, ms);
  132. ret = INLINE_SYSCALL(fstat, 2, ms->ms_fd, &ms->ms_stat);
  133. return ret;
  134. }
  135. static int sgx_ocall_fionread(void * pms)
  136. {
  137. ms_ocall_fionread_t * ms = (ms_ocall_fionread_t *) pms;
  138. int ret, val;
  139. ODEBUG(OCALL_FIONREAD, ms);
  140. ret = INLINE_SYSCALL(ioctl, 3, ms->ms_fd, FIONREAD, &val);
  141. return IS_ERR(ret) ? ret : val;
  142. }
  143. static int sgx_ocall_fsetnonblock(void * pms)
  144. {
  145. ms_ocall_fsetnonblock_t * ms = (ms_ocall_fsetnonblock_t *) pms;
  146. int ret, flags;
  147. ODEBUG(OCALL_FSETNONBLOCK, ms);
  148. ret = INLINE_SYSCALL(fcntl, 2, ms->ms_fd, F_GETFL);
  149. if (IS_ERR(ret))
  150. return ret;
  151. flags = ret;
  152. if (ms->ms_nonblocking) {
  153. if (!(flags & O_NONBLOCK))
  154. ret = INLINE_SYSCALL(fcntl, 3, ms->ms_fd, F_SETFL,
  155. flags | O_NONBLOCK);
  156. } else {
  157. if (flags & O_NONBLOCK)
  158. ret = INLINE_SYSCALL(fcntl, 3, ms->ms_fd, F_SETFL,
  159. flags & ~O_NONBLOCK);
  160. }
  161. return ret;
  162. }
  163. static int sgx_ocall_fchmod(void * pms)
  164. {
  165. ms_ocall_fchmod_t * ms = (ms_ocall_fchmod_t *) pms;
  166. int ret;
  167. ODEBUG(OCALL_FCHMOD, ms);
  168. ret = INLINE_SYSCALL(fchmod, 2, ms->ms_fd, ms->ms_mode);
  169. return ret;
  170. }
  171. static int sgx_ocall_fsync(void * pms)
  172. {
  173. ms_ocall_fsync_t * ms = (ms_ocall_fsync_t *) pms;
  174. ODEBUG(OCALL_FSYNC, ms);
  175. INLINE_SYSCALL(fsync, 1, ms->ms_fd);
  176. return 0;
  177. }
  178. static int sgx_ocall_ftruncate(void * pms)
  179. {
  180. ms_ocall_ftruncate_t * ms = (ms_ocall_ftruncate_t *) pms;
  181. int ret;
  182. ODEBUG(OCALL_FTRUNCATE, ms);
  183. ret = INLINE_SYSCALL(ftruncate, 2, ms->ms_fd, ms->ms_length);
  184. return ret;
  185. }
  186. static int sgx_ocall_mkdir(void * pms)
  187. {
  188. ms_ocall_mkdir_t * ms = (ms_ocall_mkdir_t *) pms;
  189. int ret;
  190. ODEBUG(OCALL_MKDIR, ms);
  191. ret = INLINE_SYSCALL(mkdir, 2, ms->ms_pathname, ms->ms_mode);
  192. return ret;
  193. }
  194. static int sgx_ocall_getdents(void * pms)
  195. {
  196. ms_ocall_getdents_t * ms = (ms_ocall_getdents_t *) pms;
  197. int ret;
  198. ODEBUG(OCALL_GETDENTS, ms);
  199. ret = INLINE_SYSCALL(getdents64, 3, ms->ms_fd, ms->ms_dirp, ms->ms_size);
  200. return ret;
  201. }
  202. static int sgx_ocall_wake_thread(void * pms)
  203. {
  204. ODEBUG(OCALL_WAKE_THREAD, pms);
  205. return pms ? interrupt_thread(pms) : clone_thread();
  206. }
  207. static int sgx_ocall_create_process(void * pms)
  208. {
  209. ms_ocall_create_process_t * ms = (ms_ocall_create_process_t *) pms;
  210. ODEBUG(OCALL_CREATE_PROCESS, ms);
  211. int ret = sgx_create_process(ms->ms_uri, ms->ms_nargs, ms->ms_args, ms->ms_proc_fds);
  212. if (ret < 0)
  213. return ret;
  214. ms->ms_pid = ret;
  215. return 0;
  216. }
  217. static int sgx_ocall_futex(void * pms)
  218. {
  219. ms_ocall_futex_t * ms = (ms_ocall_futex_t *) pms;
  220. int ret;
  221. ODEBUG(OCALL_FUTEX, ms);
  222. struct timespec* ts = NULL;
  223. if (ms->ms_timeout_us >= 0) {
  224. ts = __alloca(sizeof(struct timespec));
  225. ts->tv_sec = ms->ms_timeout_us / 1000000;
  226. ts->tv_nsec = (ms->ms_timeout_us - ts->tv_sec * 1000000) * 1000;
  227. }
  228. ret = INLINE_SYSCALL(futex, 6, ms->ms_futex, ms->ms_op, ms->ms_val,
  229. ts, NULL, 0);
  230. return ret;
  231. }
  232. static int sgx_ocall_socketpair(void * pms)
  233. {
  234. ms_ocall_socketpair_t * ms = (ms_ocall_socketpair_t *) pms;
  235. int ret;
  236. ODEBUG(OCALL_SOCKETPAIR, ms);
  237. ret = INLINE_SYSCALL(socketpair, 4, ms->ms_domain,
  238. ms->ms_type|SOCK_CLOEXEC,
  239. ms->ms_protocol, &ms->ms_sockfds);
  240. return ret;
  241. }
  242. static int sock_getopt(int fd, struct sockopt * opt)
  243. {
  244. SGX_DBG(DBG_M, "sock_getopt (fd = %d, sockopt addr = %p) is not implemented \
  245. always returns 0\n", fd, opt);
  246. /* initialize *opt with constant */
  247. *opt = (struct sockopt){0};
  248. opt->reuseaddr = 1;
  249. return 0;
  250. }
  251. static int sgx_ocall_sock_listen(void * pms)
  252. {
  253. ms_ocall_sock_listen_t * ms = (ms_ocall_sock_listen_t *) pms;
  254. int ret, fd;
  255. ODEBUG(OCALL_SOCK_LISTEN, ms);
  256. ret = INLINE_SYSCALL(socket, 3, ms->ms_domain,
  257. ms->ms_type|SOCK_CLOEXEC,
  258. ms->ms_protocol);
  259. if (IS_ERR(ret))
  260. goto err;
  261. fd = ret;
  262. if (ms->ms_addr->sa_family == AF_INET6) {
  263. int ipv6only = 1;
  264. INLINE_SYSCALL(setsockopt, 5, fd, SOL_IPV6, IPV6_V6ONLY, &ipv6only,
  265. sizeof(int));
  266. }
  267. /* must set the socket to be reuseable */
  268. int reuseaddr = 1;
  269. INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_REUSEADDR, &reuseaddr,
  270. sizeof(int));
  271. ret = INLINE_SYSCALL(bind, 3, fd, ms->ms_addr, ms->ms_addrlen);
  272. if (IS_ERR(ret))
  273. goto err_fd;
  274. if (ms->ms_addr) {
  275. socklen_t addrlen = ms->ms_addrlen;
  276. ret = INLINE_SYSCALL(getsockname, 3, fd, ms->ms_addr, &addrlen);
  277. if (IS_ERR(ret))
  278. goto err_fd;
  279. ms->ms_addrlen = addrlen;
  280. }
  281. if (ms->ms_type & SOCK_STREAM) {
  282. ret = INLINE_SYSCALL(listen, 2, fd, DEFAULT_BACKLOG);
  283. if (IS_ERR(ret))
  284. goto err_fd;
  285. }
  286. ret = sock_getopt(fd, &ms->ms_sockopt);
  287. if (IS_ERR(ret))
  288. goto err_fd;
  289. return fd;
  290. err_fd:
  291. INLINE_SYSCALL(close, 1, fd);
  292. err:
  293. return ret;
  294. }
  295. static int sgx_ocall_sock_accept(void * pms)
  296. {
  297. ms_ocall_sock_accept_t * ms = (ms_ocall_sock_accept_t *) pms;
  298. int ret, fd;
  299. ODEBUG(OCALL_SOCK_ACCEPT, ms);
  300. socklen_t addrlen = ms->ms_addrlen;
  301. ret = INLINE_SYSCALL(accept4, 4, ms->ms_sockfd, ms->ms_addr,
  302. &addrlen, O_CLOEXEC);
  303. if (IS_ERR(ret))
  304. goto err;
  305. fd = ret;
  306. ret = sock_getopt(fd, &ms->ms_sockopt);
  307. if (IS_ERR(ret))
  308. goto err_fd;
  309. ms->ms_addrlen = addrlen;
  310. return fd;
  311. err_fd:
  312. INLINE_SYSCALL(close, 1, fd);
  313. err:
  314. return ret;
  315. }
  316. static int sgx_ocall_sock_connect(void * pms)
  317. {
  318. ms_ocall_sock_connect_t * ms = (ms_ocall_sock_connect_t *) pms;
  319. int ret, fd;
  320. ODEBUG(OCALL_SOCK_CONNECT, ms);
  321. ret = INLINE_SYSCALL(socket, 3, ms->ms_domain,
  322. ms->ms_type|SOCK_CLOEXEC,
  323. ms->ms_protocol);
  324. if (IS_ERR(ret))
  325. goto err;
  326. fd = ret;
  327. if (ms->ms_addr && ms->ms_addr->sa_family == AF_INET6) {
  328. int ipv6only = 1;
  329. INLINE_SYSCALL(setsockopt, 5, fd, SOL_IPV6, IPV6_V6ONLY, &ipv6only,
  330. sizeof(int));
  331. }
  332. if (ms->ms_bind_addr && ms->ms_bind_addr->sa_family) {
  333. ret = INLINE_SYSCALL(bind, 3, fd, ms->ms_bind_addr,
  334. ms->ms_bind_addrlen);
  335. if (IS_ERR(ret))
  336. goto err_fd;
  337. }
  338. if (ms->ms_addr) {
  339. ret = INLINE_SYSCALL(connect, 3, fd, ms->ms_addr, ms->ms_addrlen);
  340. if (IS_ERR(ret) && ERRNO(ret) == EINPROGRESS) {
  341. do {
  342. struct pollfd pfd = { .fd = fd, .events = POLLOUT, .revents = 0, };
  343. ret = INLINE_SYSCALL(ppoll, 4, &pfd, 1, NULL, NULL);
  344. } while (IS_ERR(ret) &&
  345. ERRNO(ret) == -EWOULDBLOCK);
  346. }
  347. if (IS_ERR(ret))
  348. goto err_fd;
  349. }
  350. if (ms->ms_bind_addr && !ms->ms_bind_addr->sa_family) {
  351. socklen_t addrlen = ms->ms_bind_addrlen;
  352. ret = INLINE_SYSCALL(getsockname, 3, fd, ms->ms_bind_addr,
  353. &addrlen);
  354. if (IS_ERR(ret))
  355. goto err_fd;
  356. ms->ms_bind_addrlen = addrlen;
  357. }
  358. ret = sock_getopt(fd, &ms->ms_sockopt);
  359. if (IS_ERR(ret))
  360. goto err_fd;
  361. return fd;
  362. err_fd:
  363. INLINE_SYSCALL(close, 1, fd);
  364. err:
  365. return ret;
  366. }
  367. static int sgx_ocall_sock_recv(void * pms)
  368. {
  369. ms_ocall_sock_recv_t * ms = (ms_ocall_sock_recv_t *) pms;
  370. int ret;
  371. ODEBUG(OCALL_SOCK_RECV, ms);
  372. struct sockaddr * addr = ms->ms_addr;
  373. socklen_t addrlen = ms->ms_addr ? ms->ms_addrlen : 0;
  374. if (ms->ms_sockfd == PAL_SEC()->mcast_srv)
  375. addr = NULL;
  376. ret = INLINE_SYSCALL(recvfrom, 6,
  377. ms->ms_sockfd, ms->ms_buf, ms->ms_count, 0,
  378. addr, addr ? &addrlen : NULL);
  379. if (!IS_ERR(ret) && addr)
  380. ms->ms_addrlen = addrlen;
  381. return ret;
  382. }
  383. static int sgx_ocall_sock_send(void * pms)
  384. {
  385. ms_ocall_sock_send_t * ms = (ms_ocall_sock_send_t *) pms;
  386. int ret;
  387. ODEBUG(OCALL_SOCK_SEND, ms);
  388. const struct sockaddr * addr = ms->ms_addr;
  389. socklen_t addrlen = ms->ms_addr ? ms->ms_addrlen : 0;
  390. struct sockaddr_in mcast_addr;
  391. if (ms->ms_sockfd == PAL_SEC()->mcast_srv) {
  392. mcast_addr.sin_family = AF_INET;
  393. inet_pton4(MCAST_GROUP, sizeof(MCAST_GROUP), &mcast_addr.sin_addr.s_addr);
  394. mcast_addr.sin_port = htons(PAL_SEC()->mcast_port);
  395. addr = (struct sockaddr *) &mcast_addr;
  396. addrlen = sizeof(struct sockaddr_in);
  397. }
  398. ret = INLINE_SYSCALL(sendto, 6,
  399. ms->ms_sockfd, ms->ms_buf, ms->ms_count, MSG_NOSIGNAL,
  400. addr, addrlen);
  401. return ret;
  402. }
  403. static int sgx_ocall_sock_recv_fd(void * pms)
  404. {
  405. ms_ocall_sock_recv_fd_t * ms = (ms_ocall_sock_recv_fd_t *) pms;
  406. int ret;
  407. ODEBUG(OCALL_SOCK_RECV_FD, ms);
  408. struct msghdr hdr;
  409. struct iovec iov[1];
  410. // receive PAL_HANDLE contents in the body
  411. char cbuf[sizeof(struct cmsghdr) + ms->ms_nfds * sizeof(int)];
  412. iov[0].iov_base = ms->ms_buf;
  413. iov[0].iov_len = ms->ms_count;
  414. // clear body memory
  415. memset(&hdr, 0, sizeof(struct msghdr));
  416. // set message header values
  417. hdr.msg_iov = iov;
  418. hdr.msg_iovlen = 1;
  419. hdr.msg_control = cbuf;
  420. hdr.msg_controllen = sizeof(struct cmsghdr) + sizeof(int) *
  421. ms->ms_nfds;
  422. hdr.msg_flags = 0;
  423. ret = INLINE_SYSCALL(recvmsg, 3, ms->ms_sockfd, &hdr, 0);
  424. if (!IS_ERR(ret)) {
  425. struct cmsghdr * chdr = CMSG_FIRSTHDR(&hdr);
  426. if (chdr &&
  427. chdr->cmsg_type == SCM_RIGHTS) {
  428. ms->ms_nfds = (chdr->cmsg_len - sizeof(struct cmsghdr)) /
  429. sizeof(int);
  430. memcpy(ms->ms_fds, CMSG_DATA(chdr), sizeof(int) * ms->ms_nfds);
  431. } else {
  432. ms->ms_nfds = 0;
  433. }
  434. }
  435. return ret;
  436. }
  437. static int sgx_ocall_sock_send_fd(void * pms)
  438. {
  439. ms_ocall_sock_send_fd_t * ms = (ms_ocall_sock_send_fd_t *) pms;
  440. int ret;
  441. ODEBUG(OCALL_SOCK_SEND_FD, ms);
  442. // Declare variables required for sending the message
  443. struct msghdr hdr; // message header
  444. struct cmsghdr * chdr; //control message header
  445. struct iovec iov[1]; // IO Vector
  446. /* Message Body Composition:
  447. IOVEC[0]: PAL_HANDLE
  448. IOVEC[1..n]: Additional handle member follow
  449. Control Message: file descriptors */
  450. // Control message buffer with added space for 2 fds (ie. max size
  451. // that it will have)
  452. char cbuf[sizeof(struct cmsghdr) + ms->ms_nfds * sizeof(int)];
  453. iov[0].iov_base = (void *) ms->ms_buf;
  454. iov[0].iov_len = ms->ms_count;
  455. hdr.msg_name = NULL;
  456. hdr.msg_namelen = 0;
  457. hdr.msg_iov = iov;
  458. hdr.msg_iovlen = 1;
  459. hdr.msg_flags = 0;
  460. hdr.msg_control = cbuf; // Control Message Buffer
  461. hdr.msg_controllen = sizeof(struct cmsghdr) + sizeof(int) * ms->ms_nfds;
  462. // Fill control message infomation for the file descriptors
  463. // Check hdr.msg_controllen >= sizeof(struct cmsghdr) to point to
  464. // cbuf, which is redundant based on the above code as we have
  465. // statically allocated memory.
  466. // or (struct cmsghdr*) cbuf
  467. chdr = CMSG_FIRSTHDR(&hdr); // Pointer to msg_control
  468. chdr->cmsg_level = SOL_SOCKET; // Originating Protocol
  469. chdr->cmsg_type = SCM_RIGHTS; // Protocol Specific Type
  470. // Length of control message = sizeof(struct cmsghdr) + nfds
  471. chdr->cmsg_len = CMSG_LEN(sizeof(int) * ms->ms_nfds);
  472. // Copy the fds below control header
  473. memcpy(CMSG_DATA(chdr), ms->ms_fds, sizeof(int) * ms->ms_nfds);
  474. // Also, Update main header with control message length (duplicate)
  475. hdr.msg_controllen = chdr->cmsg_len;
  476. ret = INLINE_SYSCALL(sendmsg, 3, ms->ms_sockfd, &hdr, MSG_NOSIGNAL);
  477. return ret;
  478. }
  479. static int sgx_ocall_sock_setopt(void * pms)
  480. {
  481. ms_ocall_sock_setopt_t * ms = (ms_ocall_sock_setopt_t *) pms;
  482. int ret;
  483. ODEBUG(OCALL_SOCK_SETOPT, ms);
  484. ret = INLINE_SYSCALL(setsockopt, 5,
  485. ms->ms_sockfd, ms->ms_level, ms->ms_optname,
  486. ms->ms_optval, ms->ms_optlen);
  487. return ret;
  488. }
  489. static int sgx_ocall_sock_shutdown(void * pms)
  490. {
  491. ms_ocall_sock_shutdown_t * ms = (ms_ocall_sock_shutdown_t *) pms;
  492. ODEBUG(OCALL_SOCK_SHUTDOWN, ms);
  493. INLINE_SYSCALL(shutdown, 2, ms->ms_sockfd, ms->ms_how);
  494. return 0;
  495. }
  496. static int sgx_ocall_gettime(void * pms)
  497. {
  498. ms_ocall_gettime_t * ms = (ms_ocall_gettime_t *) pms;
  499. ODEBUG(OCALL_GETTIME, ms);
  500. struct timeval tv;
  501. INLINE_SYSCALL(gettimeofday, 2, &tv, NULL);
  502. ms->ms_microsec = tv.tv_sec * 1000000UL + tv.tv_usec;
  503. return 0;
  504. }
  505. static int sgx_ocall_sleep(void * pms)
  506. {
  507. ms_ocall_sleep_t * ms = (ms_ocall_sleep_t *) pms;
  508. int ret;
  509. ODEBUG(OCALL_SLEEP, ms);
  510. if (!ms->ms_microsec) {
  511. INLINE_SYSCALL(sched_yield, 0);
  512. return 0;
  513. }
  514. struct timespec req, rem;
  515. unsigned long microsec = ms->ms_microsec;
  516. const unsigned long VERY_LONG_TIME_IN_US = 1000000L * 60 * 60 * 24 * 365 * 128;
  517. if (ms->ms_microsec > VERY_LONG_TIME_IN_US) {
  518. /* avoid overflow with time_t */
  519. req.tv_sec = VERY_LONG_TIME_IN_US / 1000000;
  520. req.tv_nsec = 0;
  521. } else {
  522. req.tv_sec = ms->ms_microsec / 1000000;
  523. req.tv_nsec = (microsec - req.tv_sec * 1000000) * 1000;
  524. }
  525. ret = INLINE_SYSCALL(nanosleep, 2, &req, &rem);
  526. if (IS_ERR(ret) && ERRNO(ret) == EINTR)
  527. ms->ms_microsec = rem.tv_sec * 1000000UL + rem.tv_nsec / 1000UL;
  528. return ret;
  529. }
  530. static int sgx_ocall_poll(void * pms)
  531. {
  532. ms_ocall_poll_t * ms = (ms_ocall_poll_t *) pms;
  533. int ret;
  534. ODEBUG(OCALL_POLL, ms);
  535. struct timespec * ts = NULL;
  536. if (ms->ms_timeout_us >= 0) {
  537. ts = __alloca(sizeof(struct timespec));
  538. ts->tv_sec = ms->ms_timeout_us / 1000000;
  539. ts->tv_nsec = (ms->ms_timeout_us - ts->tv_sec * 1000000) * 1000;
  540. }
  541. ret = INLINE_SYSCALL(ppoll, 4, ms->ms_fds, ms->ms_nfds, ts, NULL);
  542. return ret;
  543. }
  544. static int sgx_ocall_rename(void * pms)
  545. {
  546. ms_ocall_rename_t * ms = (ms_ocall_rename_t *) pms;
  547. int ret;
  548. ODEBUG(OCALL_RENAME, ms);
  549. ret = INLINE_SYSCALL(rename, 2, ms->ms_oldpath, ms->ms_newpath);
  550. return ret;
  551. }
  552. static int sgx_ocall_delete(void * pms)
  553. {
  554. ms_ocall_delete_t * ms = (ms_ocall_delete_t *) pms;
  555. int ret;
  556. ODEBUG(OCALL_DELETE, ms);
  557. ret = INLINE_SYSCALL(unlink, 1, ms->ms_pathname);
  558. if (IS_ERR(ret) && ERRNO(ret) == EISDIR)
  559. ret = INLINE_SYSCALL(rmdir, 1, ms->ms_pathname);
  560. return ret;
  561. }
  562. void load_gdb_command (const char * command);
  563. static int sgx_ocall_load_debug(void * pms)
  564. {
  565. const char * command = (const char *) pms;
  566. ODEBUG(OCALL_LOAD_DEBUG, (void *) command);
  567. load_gdb_command(command);
  568. return 0;
  569. }
  570. static int sgx_ocall_get_attestation(void* pms) {
  571. ms_ocall_get_attestation_t * ms = (ms_ocall_get_attestation_t *) pms;
  572. ODEBUG(OCALL_GET_ATTESTATION, ms);
  573. return retrieve_verified_quote(&ms->ms_spid, ms->ms_subkey, ms->ms_linkable, &ms->ms_report,
  574. &ms->ms_nonce, &ms->ms_attestation);
  575. }
  576. sgx_ocall_fn_t ocall_table[OCALL_NR] = {
  577. [OCALL_EXIT] = sgx_ocall_exit,
  578. [OCALL_PRINT_STRING] = sgx_ocall_print_string,
  579. [OCALL_ALLOC_UNTRUSTED] = sgx_ocall_alloc_untrusted,
  580. [OCALL_MAP_UNTRUSTED] = sgx_ocall_map_untrusted,
  581. [OCALL_UNMAP_UNTRUSTED] = sgx_ocall_unmap_untrusted,
  582. [OCALL_CPUID] = sgx_ocall_cpuid,
  583. [OCALL_OPEN] = sgx_ocall_open,
  584. [OCALL_CLOSE] = sgx_ocall_close,
  585. [OCALL_READ] = sgx_ocall_read,
  586. [OCALL_WRITE] = sgx_ocall_write,
  587. [OCALL_FSTAT] = sgx_ocall_fstat,
  588. [OCALL_FIONREAD] = sgx_ocall_fionread,
  589. [OCALL_FSETNONBLOCK] = sgx_ocall_fsetnonblock,
  590. [OCALL_FCHMOD] = sgx_ocall_fchmod,
  591. [OCALL_FSYNC] = sgx_ocall_fsync,
  592. [OCALL_FTRUNCATE] = sgx_ocall_ftruncate,
  593. [OCALL_MKDIR] = sgx_ocall_mkdir,
  594. [OCALL_GETDENTS] = sgx_ocall_getdents,
  595. [OCALL_WAKE_THREAD] = sgx_ocall_wake_thread,
  596. [OCALL_CREATE_PROCESS] = sgx_ocall_create_process,
  597. [OCALL_FUTEX] = sgx_ocall_futex,
  598. [OCALL_SOCKETPAIR] = sgx_ocall_socketpair,
  599. [OCALL_SOCK_LISTEN] = sgx_ocall_sock_listen,
  600. [OCALL_SOCK_ACCEPT] = sgx_ocall_sock_accept,
  601. [OCALL_SOCK_CONNECT] = sgx_ocall_sock_connect,
  602. [OCALL_SOCK_RECV] = sgx_ocall_sock_recv,
  603. [OCALL_SOCK_SEND] = sgx_ocall_sock_send,
  604. [OCALL_SOCK_RECV_FD] = sgx_ocall_sock_recv_fd,
  605. [OCALL_SOCK_SEND_FD] = sgx_ocall_sock_send_fd,
  606. [OCALL_SOCK_SETOPT] = sgx_ocall_sock_setopt,
  607. [OCALL_SOCK_SHUTDOWN] = sgx_ocall_sock_shutdown,
  608. [OCALL_GETTIME] = sgx_ocall_gettime,
  609. [OCALL_SLEEP] = sgx_ocall_sleep,
  610. [OCALL_POLL] = sgx_ocall_poll,
  611. [OCALL_RENAME] = sgx_ocall_rename,
  612. [OCALL_DELETE] = sgx_ocall_delete,
  613. [OCALL_LOAD_DEBUG] = sgx_ocall_load_debug,
  614. [OCALL_GET_ATTESTATION] = sgx_ocall_get_attestation,
  615. };
  616. #define EDEBUG(code, ms) do {} while (0)
  617. int ecall_enclave_start (char * args, size_t args_size, char * env, size_t env_size)
  618. {
  619. ms_ecall_enclave_start_t ms;
  620. ms.ms_args = args;
  621. ms.ms_args_size = args_size;
  622. ms.ms_env = env;
  623. ms.ms_env_size = env_size;
  624. ms.ms_sec_info = PAL_SEC();
  625. EDEBUG(ECALL_ENCLAVE_START, &ms);
  626. return sgx_ecall(ECALL_ENCLAVE_START, &ms);
  627. }
  628. int ecall_thread_start (void)
  629. {
  630. EDEBUG(ECALL_THREAD_START, NULL);
  631. return sgx_ecall(ECALL_THREAD_START, NULL);
  632. }
  633. int ecall_thread_reset(void) {
  634. EDEBUG(ECALL_THREAD_RESET, NULL);
  635. return sgx_ecall(ECALL_THREAD_RESET, NULL);
  636. }
  637. noreturn void __abort(void) {
  638. INLINE_SYSCALL(exit_group, 1, -1);
  639. while (true) {
  640. /* nothing */;
  641. }
  642. }