1 /* This symbol is used at runtime to figure out the virtual address that the */
2 /* enclave is loaded at. */
7 .section ".note.x86_64-fortanix-unknown-sgx", "", @note
9 .long 1f - 0f /* name length (not including padding) */
10 .long 3f - 2f /* desc length (not including padding) */
11 .long 1 /* type = NT_VERSION */
12 0: .asciz "toolchain-version" /* name */
14 2: .long 0 /* desc - toolchain version number, 32-bit LE */
18 /* The XSAVE area needs to be a large chunk of readable memory, but since we are */
19 /* going to restore everything to its initial state (XSTATE_BV=0), only certain */
20 /* parts need to have a defined value. In particular: */
22 /* * MXCSR in the legacy area. This register is always restored if RFBM[1] or */
23 /* RFBM[2] is set, regardless of the value of XSTATE_BV */
31 /* We can store a bunch of data in the gap between MXCSR and the XSAVE header */
33 /* The following symbols point at read-only data that will be filled in by the */
36 /* When using this macro, don't forget to adjust the linker version script! */
37 .macro globvar name:req size:req
45 /* The base address (relative to enclave start) of the heap area */
47 /* The heap size in bytes */
49 /* Value of the RELA entry in the dynamic table */
51 /* Value of the RELACOUNT entry in the dynamic table */
53 /* The enclave size in bytes */
54 globvar ENCLAVE_SIZE 8
55 /* The base address (relative to enclave start) of the enclave configuration area */
56 globvar CFGDATA_BASE 8
57 /* Non-zero if debugging is enabled, zero otherwise */
59 /* The base address (relative to enclave start) of the enclave text section */
61 /* The size in bytes of enclacve text section */
63 /* The base address (relative to enclave start) of the enclave EH_FRM_HDR section */
64 globvar EH_FRM_HDR_BASE 8
65 /* The size in bytes of enclacve EH_FRM_HDR section */
66 globvar EH_FRM_HDR_SIZE 8
69 .asciz "Re-entered aborted enclave!"
70 .Lreentry_panic_msg_end:
72 .org .Lxsave_clear+512
74 .int 0, 0 /* XSTATE_BV */
75 .int 0, 0 /* XCOMP_BV */
76 .org .+48 /* reserved bits */
82 /* TCS local storage section */
83 .equ tcsls_tos, 0x00 /* initialized by loader to *offset* from image base to TOS */
84 .equ tcsls_flags, 0x08 /* initialized by loader */
85 .equ tcsls_flag_secondary, 0 /* initialized by loader; 0 = standard TCS, 1 = secondary TCS */
86 .equ tcsls_flag_init_once, 1 /* initialized by loader to 0 */
88 .equ tcsls_user_fcw, 0x0a
89 .equ tcsls_user_mxcsr, 0x0c
90 .equ tcsls_last_rsp, 0x10 /* initialized by loader to 0 */
91 .equ tcsls_panic_last_rsp, 0x18 /* initialized by loader to 0 */
92 .equ tcsls_debug_panic_buf_ptr, 0x20 /* initialized by loader to 0 */
93 .equ tcsls_user_rsp, 0x28
94 .equ tcsls_user_retip, 0x30
95 .equ tcsls_user_rbp, 0x38
96 .equ tcsls_user_r12, 0x40
97 .equ tcsls_user_r13, 0x48
98 .equ tcsls_user_r14, 0x50
99 .equ tcsls_user_r15, 0x58
100 .equ tcsls_tls_ptr, 0x60
101 .equ tcsls_tcs_addr, 0x68
103 .macro load_tcsls_flag_secondary_bool reg:req comments:vararg
104 .ifne tcsls_flag_secondary /* to convert to a bool, must be the first bit */
107 mov $(1<<tcsls_flag_secondary),%e\reg
108 and %gs:tcsls_flags,%\reg
113 .type sgx_entry,function
115 /* save user registers */
116 mov %rcx,%gs:tcsls_user_retip
117 mov %rsp,%gs:tcsls_user_rsp
118 mov %rbp,%gs:tcsls_user_rbp
119 mov %r12,%gs:tcsls_user_r12
120 mov %r13,%gs:tcsls_user_r13
121 mov %r14,%gs:tcsls_user_r14
122 mov %r15,%gs:tcsls_user_r15
123 mov %rbx,%gs:tcsls_tcs_addr
124 stmxcsr %gs:tcsls_user_mxcsr
125 fnstcw %gs:tcsls_user_fcw
126 /* reset user state */
127 cld /* x86-64 ABI requires DF to be unset at function entry/exit */
128 /* check for debug buffer pointer */
129 testb $0xff,DEBUG(%rip)
131 mov %r10,%gs:tcsls_debug_panic_buf_ptr
133 /* check for abort */
134 bt $0,.Laborted(%rip)
136 /* check if returning from usercall */
137 mov %gs:tcsls_last_rsp,%r11
141 mov %gs:tcsls_tos,%rsp /* initially, RSP is not set to the correct value */
142 /* here. This is fixed below under "adjust stack". */
143 /* check for thread init */
144 bts $tcsls_flag_init_once,%gs:tcsls_flags
147 lea IMAGE_BASE(%rip),%rax
149 mov %rsp,%gs:tcsls_tos
151 /* store caller-saved registers in callee-saved registers */
157 load_tcsls_flag_secondary_bool di /* RDI = tcs_init() argument: secondary: bool */
159 /* reload caller-saved registers */
166 /* call into main entry point */
167 load_tcsls_flag_secondary_bool cx /* RCX = entry() argument: secondary: bool */
168 call entry /* RDI, RSI, RDX, R8, R9 passed in from userspace */
169 mov %rax,%rsi /* RSI = return value */
170 /* NOP: mov %rdx,%rdx */ /* RDX = return value */
171 xor %rdi,%rdi /* RDI = normal exit */
173 /* clear general purpose register state */
174 /* RAX overwritten by ENCLU */
176 /* RCX overwritten by ENCLU */
177 /* RDX contains return value */
180 /* RDI contains exit mode */
181 /* RSI contains return value */
186 /* R12 ~ R15 set by sgx_exit */
188 /* clear extended register state */
189 mov %rdx, %rcx /* save RDX */
192 xrstor .Lxsave_clear(%rip)
193 mov %rcx, %rdx /* restore RDX */
197 /* restore user registers */
198 mov %gs:tcsls_user_r12,%r12
199 mov %gs:tcsls_user_r13,%r13
200 mov %gs:tcsls_user_r14,%r14
201 mov %gs:tcsls_user_r15,%r15
202 mov %gs:tcsls_user_retip,%rbx
203 mov %gs:tcsls_user_rsp,%rsp
204 mov %gs:tcsls_user_rbp,%rbp
205 fldcw %gs:tcsls_user_fcw
206 ldmxcsr %gs:tcsls_user_mxcsr
208 mov $0x4,%eax /* EEXIT */
213 lea .Lreentry_panic_msg(%rip),%rdi
214 mov $.Lreentry_panic_msg_end-.Lreentry_panic_msg,%esi
218 /* This *MUST* be called with 6 parameters, otherwise register information */
222 test %rcx,%rcx /* check `abort` function argument */
223 jnz .Lusercall_abort /* abort is set, jump to abort code (unlikely forward conditional) */
224 jmp .Lusercall_save_state /* non-aborting usercall */
226 /* set aborted bit */
227 movb $1,.Laborted(%rip)
228 /* save registers in DEBUG mode, so that debugger can reconstruct the stack */
229 testb $0xff,DEBUG(%rip)
230 jz .Lusercall_noreturn
231 .Lusercall_save_state:
232 /* save callee-saved state */
242 movq %rsp,%gs:tcsls_last_rsp
244 /* clear general purpose register state */
245 /* RAX overwritten by ENCLU */
246 /* RBX set by sgx_exit */
247 /* RCX overwritten by ENCLU */
248 /* RDX contains parameter */
249 /* RSP set by sgx_exit */
250 /* RBP set by sgx_exit */
251 /* RDI contains parameter */
252 /* RSI contains parameter */
253 /* R8 contains parameter */
254 /* R9 contains parameter */
257 /* R12 ~ R15 set by sgx_exit */
258 /* extended registers/flags cleared by sgx_exit */
262 movq $0,%gs:tcsls_last_rsp
263 /* restore callee-saved state, cf. "save" above */
275 mov %rsi,%rax /* RAX = return value */
276 /* NOP: mov %rdx,%rdx */ /* RDX = return value */
280 The following functions need to be defined externally:
282 // Called by entry code when it needs to panic
283 extern "C" fn panic_msg(msg: &'static str) -> ! {
287 // Called once when a TCS is first entered
288 extern "C" fn tcs_init(secondary: bool);
290 // Standard TCS entrypoint
291 extern "C" fn entry(p1: u64, p2: u64, p3: u64, secondary: bool, p4: u64, p5: u64) -> (u64, u64);
297 mov %gs:tcsls_tcs_addr,%rax
302 mov %gs:tcsls_tls_ptr,%rax
307 mov %rdi,%gs:tcsls_tls_ptr
310 .global take_debug_panic_buf_ptr
311 take_debug_panic_buf_ptr:
313 xchg %gs:tcsls_debug_panic_buf_ptr,%rax