0001
0002
0003
0004 #include <stddef.h>
0005 #include "defines.h"
0006
0007
0008
0009
0010
0011
0012 static uint8_t encl_buffer[8192] = { 1 };
0013
0014 enum sgx_enclu_function {
0015 EACCEPT = 0x5,
0016 EMODPE = 0x6,
0017 };
0018
0019 static void do_encl_emodpe(void *_op)
0020 {
0021 struct sgx_secinfo secinfo __aligned(sizeof(struct sgx_secinfo)) = {0};
0022 struct encl_op_emodpe *op = _op;
0023
0024 secinfo.flags = op->flags;
0025
0026 asm volatile(".byte 0x0f, 0x01, 0xd7"
0027 :
0028 : "a" (EMODPE),
0029 "b" (&secinfo),
0030 "c" (op->epc_addr));
0031 }
0032
0033 static void do_encl_eaccept(void *_op)
0034 {
0035 struct sgx_secinfo secinfo __aligned(sizeof(struct sgx_secinfo)) = {0};
0036 struct encl_op_eaccept *op = _op;
0037 int rax;
0038
0039 secinfo.flags = op->flags;
0040
0041 asm volatile(".byte 0x0f, 0x01, 0xd7"
0042 : "=a" (rax)
0043 : "a" (EACCEPT),
0044 "b" (&secinfo),
0045 "c" (op->epc_addr));
0046
0047 op->ret = rax;
0048 }
0049
0050 static void *memcpy(void *dest, const void *src, size_t n)
0051 {
0052 size_t i;
0053
0054 for (i = 0; i < n; i++)
0055 ((char *)dest)[i] = ((char *)src)[i];
0056
0057 return dest;
0058 }
0059
0060 static void *memset(void *dest, int c, size_t n)
0061 {
0062 size_t i;
0063
0064 for (i = 0; i < n; i++)
0065 ((char *)dest)[i] = c;
0066
0067 return dest;
0068 }
0069
0070 static void do_encl_init_tcs_page(void *_op)
0071 {
0072 struct encl_op_init_tcs_page *op = _op;
0073 void *tcs = (void *)op->tcs_page;
0074 uint32_t val_32;
0075
0076 memset(tcs, 0, 16);
0077 memcpy(tcs + 16, &op->ssa, 8);
0078 memset(tcs + 24, 0, 4);
0079 val_32 = 1;
0080 memcpy(tcs + 28, &val_32, 4);
0081 memcpy(tcs + 32, &op->entry, 8);
0082 memset(tcs + 40, 0, 24);
0083 val_32 = 0xFFFFFFFF;
0084 memcpy(tcs + 64, &val_32, 4);
0085 memcpy(tcs + 68, &val_32, 4);
0086 memset(tcs + 72, 0, 4024);
0087 }
0088
0089 static void do_encl_op_put_to_buf(void *op)
0090 {
0091 struct encl_op_put_to_buf *op2 = op;
0092
0093 memcpy(&encl_buffer[0], &op2->value, 8);
0094 }
0095
0096 static void do_encl_op_get_from_buf(void *op)
0097 {
0098 struct encl_op_get_from_buf *op2 = op;
0099
0100 memcpy(&op2->value, &encl_buffer[0], 8);
0101 }
0102
0103 static void do_encl_op_put_to_addr(void *_op)
0104 {
0105 struct encl_op_put_to_addr *op = _op;
0106
0107 memcpy((void *)op->addr, &op->value, 8);
0108 }
0109
0110 static void do_encl_op_get_from_addr(void *_op)
0111 {
0112 struct encl_op_get_from_addr *op = _op;
0113
0114 memcpy(&op->value, (void *)op->addr, 8);
0115 }
0116
0117 static void do_encl_op_nop(void *_op)
0118 {
0119
0120 }
0121
0122 void encl_body(void *rdi, void *rsi)
0123 {
0124 const void (*encl_op_array[ENCL_OP_MAX])(void *) = {
0125 do_encl_op_put_to_buf,
0126 do_encl_op_get_from_buf,
0127 do_encl_op_put_to_addr,
0128 do_encl_op_get_from_addr,
0129 do_encl_op_nop,
0130 do_encl_eaccept,
0131 do_encl_emodpe,
0132 do_encl_init_tcs_page,
0133 };
0134
0135 struct encl_op_header *op = (struct encl_op_header *)rdi;
0136
0137 if (op->type < ENCL_OP_MAX)
0138 (*encl_op_array[op->type])(op);
0139 }