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
68 .org .Lxsave_clear+512
70 .int 0, 0 /* XSTATE_BV */
71 .int 0, 0 /* XCOMP_BV */
72 .org .+48 /* reserved bits */
78 /* TCS local storage section */
79 .equ tcsls_tos, 0x00 /* initialized by loader to *offset* from image base to TOS */
80 .equ tcsls_flags, 0x08 /* initialized by loader */
81 .equ tcsls_flag_secondary, 0 /* initialized by loader; 0 = standard TCS, 1 = secondary TCS */
82 .equ tcsls_flag_init_once, 1 /* initialized by loader to 0 */
84 .equ tcsls_user_fcw, 0x0a
85 .equ tcsls_user_mxcsr, 0x0c
86 .equ tcsls_last_rsp, 0x10 /* initialized by loader to 0 */
87 .equ tcsls_panic_last_rsp, 0x18 /* initialized by loader to 0 */
88 .equ tcsls_debug_panic_buf_ptr, 0x20 /* initialized by loader to 0 */
89 .equ tcsls_user_rsp, 0x28
90 .equ tcsls_user_retip, 0x30
91 .equ tcsls_user_rbp, 0x38
92 .equ tcsls_user_r12, 0x40
93 .equ tcsls_user_r13, 0x48
94 .equ tcsls_user_r14, 0x50
95 .equ tcsls_user_r15, 0x58
96 .equ tcsls_tls_ptr, 0x60
97 .equ tcsls_tcs_addr, 0x68
99 .macro load_tcsls_flag_secondary_bool reg:req comments:vararg
100 .ifne tcsls_flag_secondary /* to convert to a bool, must be the first bit */
103 mov $(1<<tcsls_flag_secondary),%e\reg
104 and %gs:tcsls_flags,%\reg
109 .type sgx_entry,function
111 /* save user registers */
112 mov %rcx,%gs:tcsls_user_retip
113 mov %rsp,%gs:tcsls_user_rsp
114 mov %rbp,%gs:tcsls_user_rbp
115 mov %r12,%gs:tcsls_user_r12
116 mov %r13,%gs:tcsls_user_r13
117 mov %r14,%gs:tcsls_user_r14
118 mov %r15,%gs:tcsls_user_r15
119 mov %rbx,%gs:tcsls_tcs_addr
120 stmxcsr %gs:tcsls_user_mxcsr
121 fnstcw %gs:tcsls_user_fcw
122 /* reset user state */
123 cld /* x86-64 ABI requires DF to be unset at function entry/exit */
124 /* check for debug buffer pointer */
125 testb $0xff,DEBUG(%rip)
127 mov %r10,%gs:tcsls_debug_panic_buf_ptr
129 /* check for abort */
130 bt $0,.Laborted(%rip)
132 /* check if returning from usercall */
133 mov %gs:tcsls_last_rsp,%r11
137 mov %gs:tcsls_tos,%rsp /* initially, RSP is not set to the correct value */
138 /* here. This is fixed below under "adjust stack". */
139 /* check for thread init */
140 bts $tcsls_flag_init_once,%gs:tcsls_flags
143 lea IMAGE_BASE(%rip),%rax
145 mov %rsp,%gs:tcsls_tos
147 /* store caller-saved registers in callee-saved registers */
153 load_tcsls_flag_secondary_bool di /* RDI = tcs_init() argument: secondary: bool */
155 /* reload caller-saved registers */
162 /* call into main entry point */
163 load_tcsls_flag_secondary_bool cx /* RCX = entry() argument: secondary: bool */
164 call entry /* RDI, RSI, RDX, R8, R9 passed in from userspace */
165 mov %rax,%rsi /* RSI = return value */
166 /* NOP: mov %rdx,%rdx */ /* RDX = return value */
167 xor %rdi,%rdi /* RDI = normal exit */
169 /* clear general purpose register state */
170 /* RAX overwritten by ENCLU */
172 /* RCX overwritten by ENCLU */
173 /* RDX contains return value */
176 /* RDI contains exit mode */
177 /* RSI contains return value */
182 /* R12 ~ R15 set by sgx_exit */
184 /* clear extended register state */
185 mov %rdx, %rcx /* save RDX */
188 xrstor .Lxsave_clear(%rip)
189 mov %rcx, %rdx /* restore RDX */
193 /* restore user registers */
194 mov %gs:tcsls_user_r12,%r12
195 mov %gs:tcsls_user_r13,%r13
196 mov %gs:tcsls_user_r14,%r14
197 mov %gs:tcsls_user_r15,%r15
198 mov %gs:tcsls_user_retip,%rbx
199 mov %gs:tcsls_user_rsp,%rsp
200 mov %gs:tcsls_user_rbp,%rbp
201 fldcw %gs:tcsls_user_fcw
202 ldmxcsr %gs:tcsls_user_mxcsr
204 mov $0x4,%eax /* EEXIT */
212 /* This *MUST* be called with 6 parameters, otherwise register information */
216 test %rcx,%rcx /* check `abort` function argument */
217 jnz .Lusercall_abort /* abort is set, jump to abort code (unlikely forward conditional) */
218 jmp .Lusercall_save_state /* non-aborting usercall */
220 /* set aborted bit */
221 movb $1,.Laborted(%rip)
222 /* save registers in DEBUG mode, so that debugger can reconstruct the stack */
223 testb $0xff,DEBUG(%rip)
224 jz .Lusercall_noreturn
225 .Lusercall_save_state:
226 /* save callee-saved state */
236 movq %rsp,%gs:tcsls_last_rsp
238 /* clear general purpose register state */
239 /* RAX overwritten by ENCLU */
240 /* RBX set by sgx_exit */
241 /* RCX overwritten by ENCLU */
242 /* RDX contains parameter */
243 /* RSP set by sgx_exit */
244 /* RBP set by sgx_exit */
245 /* RDI contains parameter */
246 /* RSI contains parameter */
247 /* R8 contains parameter */
248 /* R9 contains parameter */
251 /* R12 ~ R15 set by sgx_exit */
252 /* extended registers/flags cleared by sgx_exit */
256 movq $0,%gs:tcsls_last_rsp
257 /* restore callee-saved state, cf. "save" above */
269 mov %rsi,%rax /* RAX = return value */
270 /* NOP: mov %rdx,%rdx */ /* RDX = return value */
274 The following functions need to be defined externally:
276 // Called by entry code on re-entry after exit
277 extern "C" fn abort_reentry() -> !;
279 // Called once when a TCS is first entered
280 extern "C" fn tcs_init(secondary: bool);
282 // Standard TCS entrypoint
283 extern "C" fn entry(p1: u64, p2: u64, p3: u64, secondary: bool, p4: u64, p5: u64) -> (u64, u64);
289 mov %gs:tcsls_tcs_addr,%rax
294 mov %gs:tcsls_tls_ptr,%rax
299 mov %rdi,%gs:tcsls_tls_ptr
302 .global take_debug_panic_buf_ptr
303 take_debug_panic_buf_ptr:
305 xchg %gs:tcsls_debug_panic_buf_ptr,%rax