blob: 8ed23de419d1eef7d8f253c833fbafc5c48ec60c [file] [log] [blame]
Lang Hames4821c242016-05-07 03:36:38 +00001//===------------- OrcABISupport.cpp - ABI specific support code ----------===//
Lang Hames4df7ba72015-10-26 06:40:28 +00002//
Chandler Carruth2946cd72019-01-19 08:50:56 +00003// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
Lang Hames4df7ba72015-10-26 06:40:28 +00006//
7//===----------------------------------------------------------------------===//
8
Lang Hames4821c242016-05-07 03:36:38 +00009#include "llvm/ExecutionEngine/Orc/OrcABISupport.h"
Lang Hames98c2ac12015-10-19 17:43:51 +000010#include "llvm/Support/Process.h"
Lang Hames93de2a12015-01-23 21:25:00 +000011
Lang Hames93de2a12015-01-23 21:25:00 +000012namespace llvm {
Lang Hamese7380612015-02-21 20:44:36 +000013namespace orc {
Lang Hames93de2a12015-01-23 21:25:00 +000014
Lang Hamesf74e93b2016-04-29 21:32:00 +000015void OrcAArch64::writeResolverCode(uint8_t *ResolverMem, JITReentryFn ReentryFn,
16 void *CallbackMgr) {
17
18 const uint32_t ResolverCode[] = {
19 // resolver_entry:
Lang Hames2307f402016-05-01 00:14:45 +000020 0xa9bf47fd, // 0x000: stp x29, x17, [sp, #-16]!
21 0x910003fd, // 0x004: mov x29, sp
22 0xa9bf73fb, // 0x008: stp x27, x28, [sp, #-16]!
23 0xa9bf6bf9, // 0x00c: stp x25, x26, [sp, #-16]!
24 0xa9bf63f7, // 0x010: stp x23, x24, [sp, #-16]!
25 0xa9bf5bf5, // 0x014: stp x21, x22, [sp, #-16]!
26 0xa9bf53f3, // 0x018: stp x19, x20, [sp, #-16]!
27 0xa9bf3fee, // 0x01c: stp x14, x15, [sp, #-16]!
28 0xa9bf37ec, // 0x020: stp x12, x13, [sp, #-16]!
29 0xa9bf2fea, // 0x024: stp x10, x11, [sp, #-16]!
30 0xa9bf27e8, // 0x028: stp x8, x9, [sp, #-16]!
31 0xa9bf1fe6, // 0x02c: stp x6, x7, [sp, #-16]!
32 0xa9bf17e4, // 0x030: stp x4, x5, [sp, #-16]!
33 0xa9bf0fe2, // 0x034: stp x2, x3, [sp, #-16]!
34 0xa9bf07e0, // 0x038: stp x0, x1, [sp, #-16]!
35 0xadbf7ffe, // 0x03c: stp q30, q31, [sp, #-32]!
36 0xadbf77fc, // 0x040: stp q28, q29, [sp, #-32]!
37 0xadbf6ffa, // 0x044: stp q26, q27, [sp, #-32]!
38 0xadbf67f8, // 0x048: stp q24, q25, [sp, #-32]!
39 0xadbf5ff6, // 0x04c: stp q22, q23, [sp, #-32]!
40 0xadbf57f4, // 0x050: stp q20, q21, [sp, #-32]!
41 0xadbf4ff2, // 0x054: stp q18, q19, [sp, #-32]!
42 0xadbf47f0, // 0x058: stp q16, q17, [sp, #-32]!
43 0xadbf3fee, // 0x05c: stp q14, q15, [sp, #-32]!
44 0xadbf37ec, // 0x060: stp q12, q13, [sp, #-32]!
45 0xadbf2fea, // 0x064: stp q10, q11, [sp, #-32]!
46 0xadbf27e8, // 0x068: stp q8, q9, [sp, #-32]!
47 0xadbf1fe6, // 0x06c: stp q6, q7, [sp, #-32]!
48 0xadbf17e4, // 0x070: stp q4, q5, [sp, #-32]!
49 0xadbf0fe2, // 0x074: stp q2, q3, [sp, #-32]!
50 0xadbf07e0, // 0x078: stp q0, q1, [sp, #-32]!
51 0x580004e0, // 0x07c: ldr x0, Lcallbackmgr
52 0xaa1e03e1, // 0x080: mov x1, x30
53 0xd1003021, // 0x084: sub x1, x1, #12
54 0x58000442, // 0x088: ldr x2, Lreentry_fn_ptr
55 0xd63f0040, // 0x08c: blr x2
56 0xaa0003f1, // 0x090: mov x17, x0
57 0xacc107e0, // 0x094: ldp q0, q1, [sp], #32
58 0xacc10fe2, // 0x098: ldp q2, q3, [sp], #32
59 0xacc117e4, // 0x09c: ldp q4, q5, [sp], #32
60 0xacc11fe6, // 0x0a0: ldp q6, q7, [sp], #32
61 0xacc127e8, // 0x0a4: ldp q8, q9, [sp], #32
62 0xacc12fea, // 0x0a8: ldp q10, q11, [sp], #32
63 0xacc137ec, // 0x0ac: ldp q12, q13, [sp], #32
64 0xacc13fee, // 0x0b0: ldp q14, q15, [sp], #32
65 0xacc147f0, // 0x0b4: ldp q16, q17, [sp], #32
66 0xacc14ff2, // 0x0b8: ldp q18, q19, [sp], #32
67 0xacc157f4, // 0x0bc: ldp q20, q21, [sp], #32
68 0xacc15ff6, // 0x0c0: ldp q22, q23, [sp], #32
69 0xacc167f8, // 0x0c4: ldp q24, q25, [sp], #32
70 0xacc16ffa, // 0x0c8: ldp q26, q27, [sp], #32
71 0xacc177fc, // 0x0cc: ldp q28, q29, [sp], #32
72 0xacc17ffe, // 0x0d0: ldp q30, q31, [sp], #32
73 0xa8c107e0, // 0x0d4: ldp x0, x1, [sp], #16
74 0xa8c10fe2, // 0x0d8: ldp x2, x3, [sp], #16
75 0xa8c117e4, // 0x0dc: ldp x4, x5, [sp], #16
76 0xa8c11fe6, // 0x0e0: ldp x6, x7, [sp], #16
77 0xa8c127e8, // 0x0e4: ldp x8, x9, [sp], #16
78 0xa8c12fea, // 0x0e8: ldp x10, x11, [sp], #16
79 0xa8c137ec, // 0x0ec: ldp x12, x13, [sp], #16
80 0xa8c13fee, // 0x0f0: ldp x14, x15, [sp], #16
81 0xa8c153f3, // 0x0f4: ldp x19, x20, [sp], #16
82 0xa8c15bf5, // 0x0f8: ldp x21, x22, [sp], #16
83 0xa8c163f7, // 0x0fc: ldp x23, x24, [sp], #16
84 0xa8c16bf9, // 0x100: ldp x25, x26, [sp], #16
85 0xa8c173fb, // 0x104: ldp x27, x28, [sp], #16
86 0xa8c17bfd, // 0x108: ldp x29, x30, [sp], #16
87 0xd65f0220, // 0x10c: ret x17
88 0x01234567, // 0x110: Lreentry_fn_ptr:
89 0xdeadbeef, // 0x114: .quad 0
90 0x98765432, // 0x118: Lcallbackmgr:
91 0xcafef00d // 0x11c: .quad 0
Lang Hamesf74e93b2016-04-29 21:32:00 +000092 };
93
Lang Hames2307f402016-05-01 00:14:45 +000094 const unsigned ReentryFnAddrOffset = 0x110;
95 const unsigned CallbackMgrAddrOffset = 0x118;
Lang Hamesf74e93b2016-04-29 21:32:00 +000096
97 memcpy(ResolverMem, ResolverCode, sizeof(ResolverCode));
98 memcpy(ResolverMem + ReentryFnAddrOffset, &ReentryFn, sizeof(ReentryFn));
99 memcpy(ResolverMem + CallbackMgrAddrOffset, &CallbackMgr,
100 sizeof(CallbackMgr));
101}
102
103void OrcAArch64::writeTrampolines(uint8_t *TrampolineMem, void *ResolverAddr,
104 unsigned NumTrampolines) {
105
106 unsigned OffsetToPtr = alignTo(NumTrampolines * TrampolineSize, 8);
107
108 memcpy(TrampolineMem + OffsetToPtr, &ResolverAddr, sizeof(void *));
109
110 // OffsetToPtr is actually the offset from the PC for the 2nd instruction, so
111 // subtract 32-bits.
112 OffsetToPtr -= 4;
113
114 uint32_t *Trampolines = reinterpret_cast<uint32_t *>(TrampolineMem);
115
116 for (unsigned I = 0; I < NumTrampolines; ++I, OffsetToPtr -= TrampolineSize) {
117 Trampolines[3 * I + 0] = 0xaa1e03f1; // mov x17, x30
Lang Hamesd4d6a1a2017-09-15 06:50:19 +0000118 Trampolines[3 * I + 1] = 0x58000010 | (OffsetToPtr << 3); // adr x16, Lptr
Lang Hamesf74e93b2016-04-29 21:32:00 +0000119 Trampolines[3 * I + 2] = 0xd63f0200; // blr x16
120 }
121
122}
123
124Error OrcAArch64::emitIndirectStubsBlock(IndirectStubsInfo &StubsInfo,
125 unsigned MinStubs,
126 void *InitialPtrVal) {
127 // Stub format is:
128 //
129 // .section __orc_stubs
130 // stub1:
131 // ldr x0, ptr1 ; PC-rel load of ptr1
132 // br x0 ; Jump to resolver
133 // stub2:
134 // ldr x0, ptr2 ; PC-rel load of ptr2
135 // br x0 ; Jump to resolver
136 //
137 // ...
138 //
139 // .section __orc_ptrs
140 // ptr1:
141 // .quad 0x0
142 // ptr2:
143 // .quad 0x0
144 //
145 // ...
146
147 const unsigned StubSize = IndirectStubsInfo::StubSize;
148
149 // Emit at least MinStubs, rounded up to fill the pages allocated.
Lang Hamese4b4ab62019-05-08 02:11:07 +0000150 static const unsigned PageSize = sys::Process::getPageSizeEstimate();
Lang Hamesf74e93b2016-04-29 21:32:00 +0000151 unsigned NumPages = ((MinStubs * StubSize) + (PageSize - 1)) / PageSize;
152 unsigned NumStubs = (NumPages * PageSize) / StubSize;
153
154 // Allocate memory for stubs and pointers in one call.
155 std::error_code EC;
156 auto StubsMem = sys::OwningMemoryBlock(sys::Memory::allocateMappedMemory(
157 2 * NumPages * PageSize, nullptr,
158 sys::Memory::MF_READ | sys::Memory::MF_WRITE, EC));
159
160 if (EC)
161 return errorCodeToError(EC);
162
163 // Create separate MemoryBlocks representing the stubs and pointers.
164 sys::MemoryBlock StubsBlock(StubsMem.base(), NumPages * PageSize);
165 sys::MemoryBlock PtrsBlock(static_cast<char *>(StubsMem.base()) +
166 NumPages * PageSize,
167 NumPages * PageSize);
168
169 // Populate the stubs page stubs and mark it executable.
170 uint64_t *Stub = reinterpret_cast<uint64_t *>(StubsBlock.base());
171 uint64_t PtrOffsetField = static_cast<uint64_t>(NumPages * PageSize)
172 << 3;
173
174 for (unsigned I = 0; I < NumStubs; ++I)
175 Stub[I] = 0xd61f020058000010 | PtrOffsetField;
176
177 if (auto EC = sys::Memory::protectMappedMemory(
178 StubsBlock, sys::Memory::MF_READ | sys::Memory::MF_EXEC))
179 return errorCodeToError(EC);
180
181 // Initialize all pointers to point at FailureAddress.
182 void **Ptr = reinterpret_cast<void **>(PtrsBlock.base());
183 for (unsigned I = 0; I < NumStubs; ++I)
184 Ptr[I] = InitialPtrVal;
185
186 StubsInfo = IndirectStubsInfo(NumStubs, std::move(StubsMem));
187
188 return Error::success();
189}
190
Lang Hames4821c242016-05-07 03:36:38 +0000191void OrcX86_64_Base::writeTrampolines(uint8_t *TrampolineMem,
192 void *ResolverAddr,
193 unsigned NumTrampolines) {
Rafael Espindola2f344632015-11-03 16:25:20 +0000194
Rafael Espindolae63e0182015-11-03 16:40:37 +0000195 unsigned OffsetToPtr = NumTrampolines * TrampolineSize;
196
Lang Hames1fa0e0e2016-04-25 21:21:20 +0000197 memcpy(TrampolineMem + OffsetToPtr, &ResolverAddr, sizeof(void *));
Rafael Espindolae63e0182015-11-03 16:40:37 +0000198
Lang Hames1fa0e0e2016-04-25 21:21:20 +0000199 uint64_t *Trampolines = reinterpret_cast<uint64_t *>(TrampolineMem);
Rafael Espindolae63e0182015-11-03 16:40:37 +0000200 uint64_t CallIndirPCRel = 0xf1c40000000015ff;
201
202 for (unsigned I = 0; I < NumTrampolines; ++I, OffsetToPtr -= TrampolineSize)
203 Trampolines[I] = CallIndirPCRel | ((OffsetToPtr - 6) << 16);
Lang Hames27547142015-02-17 01:18:38 +0000204}
205
Lang Hames4821c242016-05-07 03:36:38 +0000206Error OrcX86_64_Base::emitIndirectStubsBlock(IndirectStubsInfo &StubsInfo,
207 unsigned MinStubs,
208 void *InitialPtrVal) {
Lang Hames98c2ac12015-10-19 17:43:51 +0000209 // Stub format is:
210 //
211 // .section __orc_stubs
212 // stub1:
213 // jmpq *ptr1(%rip)
214 // .byte 0xC4 ; <- Invalid opcode padding.
215 // .byte 0xF1
216 // stub2:
217 // jmpq *ptr2(%rip)
218 //
219 // ...
220 //
221 // .section __orc_ptrs
222 // ptr1:
223 // .quad 0x0
224 // ptr2:
225 // .quad 0x0
226 //
227 // ...
228
229 const unsigned StubSize = IndirectStubsInfo::StubSize;
230
231 // Emit at least MinStubs, rounded up to fill the pages allocated.
Lang Hamese4b4ab62019-05-08 02:11:07 +0000232 static const unsigned PageSize = sys::Process::getPageSizeEstimate();
Lang Hames98c2ac12015-10-19 17:43:51 +0000233 unsigned NumPages = ((MinStubs * StubSize) + (PageSize - 1)) / PageSize;
234 unsigned NumStubs = (NumPages * PageSize) / StubSize;
235
236 // Allocate memory for stubs and pointers in one call.
237 std::error_code EC;
Lang Hames1fa0e0e2016-04-25 21:21:20 +0000238 auto StubsMem = sys::OwningMemoryBlock(sys::Memory::allocateMappedMemory(
239 2 * NumPages * PageSize, nullptr,
240 sys::Memory::MF_READ | sys::Memory::MF_WRITE, EC));
Lang Hames98c2ac12015-10-19 17:43:51 +0000241
242 if (EC)
Lang Hamesef5a0ee2016-04-25 19:56:45 +0000243 return errorCodeToError(EC);
Lang Hames98c2ac12015-10-19 17:43:51 +0000244
245 // Create separate MemoryBlocks representing the stubs and pointers.
Lang Hames5796eb22015-10-31 00:55:32 +0000246 sys::MemoryBlock StubsBlock(StubsMem.base(), NumPages * PageSize);
Lang Hames1fa0e0e2016-04-25 21:21:20 +0000247 sys::MemoryBlock PtrsBlock(static_cast<char *>(StubsMem.base()) +
248 NumPages * PageSize,
Lang Hames98c2ac12015-10-19 17:43:51 +0000249 NumPages * PageSize);
250
251 // Populate the stubs page stubs and mark it executable.
Lang Hames1fa0e0e2016-04-25 21:21:20 +0000252 uint64_t *Stub = reinterpret_cast<uint64_t *>(StubsBlock.base());
253 uint64_t PtrOffsetField = static_cast<uint64_t>(NumPages * PageSize - 6)
254 << 16;
Lang Hames98c2ac12015-10-19 17:43:51 +0000255 for (unsigned I = 0; I < NumStubs; ++I)
256 Stub[I] = 0xF1C40000000025ff | PtrOffsetField;
257
Lang Hames1fa0e0e2016-04-25 21:21:20 +0000258 if (auto EC = sys::Memory::protectMappedMemory(
259 StubsBlock, sys::Memory::MF_READ | sys::Memory::MF_EXEC))
Lang Hamesef5a0ee2016-04-25 19:56:45 +0000260 return errorCodeToError(EC);
Lang Hames98c2ac12015-10-19 17:43:51 +0000261
262 // Initialize all pointers to point at FailureAddress.
Lang Hames1fa0e0e2016-04-25 21:21:20 +0000263 void **Ptr = reinterpret_cast<void **>(PtrsBlock.base());
Lang Hames98c2ac12015-10-19 17:43:51 +0000264 for (unsigned I = 0; I < NumStubs; ++I)
265 Ptr[I] = InitialPtrVal;
266
Lang Hamese28b1182016-02-02 19:31:15 +0000267 StubsInfo = IndirectStubsInfo(NumStubs, std::move(StubsMem));
Lang Hames98c2ac12015-10-19 17:43:51 +0000268
Lang Hamesef5a0ee2016-04-25 19:56:45 +0000269 return Error::success();
Lang Hames98c2ac12015-10-19 17:43:51 +0000270}
271
Lang Hames4821c242016-05-07 03:36:38 +0000272void OrcX86_64_SysV::writeResolverCode(uint8_t *ResolverMem,
273 JITReentryFn ReentryFn,
274 void *CallbackMgr) {
275
276 const uint8_t ResolverCode[] = {
277 // resolver_entry:
278 0x55, // 0x00: pushq %rbp
279 0x48, 0x89, 0xe5, // 0x01: movq %rsp, %rbp
280 0x50, // 0x04: pushq %rax
281 0x53, // 0x05: pushq %rbx
282 0x51, // 0x06: pushq %rcx
283 0x52, // 0x07: pushq %rdx
284 0x56, // 0x08: pushq %rsi
285 0x57, // 0x09: pushq %rdi
286 0x41, 0x50, // 0x0a: pushq %r8
287 0x41, 0x51, // 0x0c: pushq %r9
288 0x41, 0x52, // 0x0e: pushq %r10
289 0x41, 0x53, // 0x10: pushq %r11
290 0x41, 0x54, // 0x12: pushq %r12
291 0x41, 0x55, // 0x14: pushq %r13
292 0x41, 0x56, // 0x16: pushq %r14
293 0x41, 0x57, // 0x18: pushq %r15
294 0x48, 0x81, 0xec, 0x08, 0x02, 0x00, 0x00, // 0x1a: subq 0x208, %rsp
295 0x48, 0x0f, 0xae, 0x04, 0x24, // 0x21: fxsave64 (%rsp)
296 0x48, 0xbf, // 0x26: movabsq <CBMgr>, %rdi
297
298 // 0x28: Callback manager addr.
299 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
300
301 0x48, 0x8b, 0x75, 0x08, // 0x30: movq 8(%rbp), %rsi
302 0x48, 0x83, 0xee, 0x06, // 0x34: subq $6, %rsi
303 0x48, 0xb8, // 0x38: movabsq <REntry>, %rax
304
305 // 0x3a: JIT re-entry fn addr:
306 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
307
308 0xff, 0xd0, // 0x42: callq *%rax
309 0x48, 0x89, 0x45, 0x08, // 0x44: movq %rax, 8(%rbp)
310 0x48, 0x0f, 0xae, 0x0c, 0x24, // 0x48: fxrstor64 (%rsp)
311 0x48, 0x81, 0xc4, 0x08, 0x02, 0x00, 0x00, // 0x4d: addq 0x208, %rsp
312 0x41, 0x5f, // 0x54: popq %r15
313 0x41, 0x5e, // 0x56: popq %r14
314 0x41, 0x5d, // 0x58: popq %r13
315 0x41, 0x5c, // 0x5a: popq %r12
316 0x41, 0x5b, // 0x5c: popq %r11
317 0x41, 0x5a, // 0x5e: popq %r10
318 0x41, 0x59, // 0x60: popq %r9
319 0x41, 0x58, // 0x62: popq %r8
320 0x5f, // 0x64: popq %rdi
321 0x5e, // 0x65: popq %rsi
322 0x5a, // 0x66: popq %rdx
323 0x59, // 0x67: popq %rcx
324 0x5b, // 0x68: popq %rbx
325 0x58, // 0x69: popq %rax
326 0x5d, // 0x6a: popq %rbp
327 0xc3, // 0x6b: retq
328 };
329
330 const unsigned ReentryFnAddrOffset = 0x3a;
331 const unsigned CallbackMgrAddrOffset = 0x28;
332
333 memcpy(ResolverMem, ResolverCode, sizeof(ResolverCode));
334 memcpy(ResolverMem + ReentryFnAddrOffset, &ReentryFn, sizeof(ReentryFn));
335 memcpy(ResolverMem + CallbackMgrAddrOffset, &CallbackMgr,
336 sizeof(CallbackMgr));
337}
338
339void OrcX86_64_Win32::writeResolverCode(uint8_t *ResolverMem,
340 JITReentryFn ReentryFn,
341 void *CallbackMgr) {
342
343 // resolverCode is similar to OrcX86_64 with differences specific to windows x64 calling convention:
344 // arguments go into rcx, rdx and come in reverse order, shadow space allocation on stack
345 const uint8_t ResolverCode[] = {
346 // resolver_entry:
347 0x55, // 0x00: pushq %rbp
348 0x48, 0x89, 0xe5, // 0x01: movq %rsp, %rbp
349 0x50, // 0x04: pushq %rax
350 0x53, // 0x05: pushq %rbx
351 0x51, // 0x06: pushq %rcx
352 0x52, // 0x07: pushq %rdx
353 0x56, // 0x08: pushq %rsi
354 0x57, // 0x09: pushq %rdi
355 0x41, 0x50, // 0x0a: pushq %r8
356 0x41, 0x51, // 0x0c: pushq %r9
357 0x41, 0x52, // 0x0e: pushq %r10
358 0x41, 0x53, // 0x10: pushq %r11
359 0x41, 0x54, // 0x12: pushq %r12
360 0x41, 0x55, // 0x14: pushq %r13
361 0x41, 0x56, // 0x16: pushq %r14
362 0x41, 0x57, // 0x18: pushq %r15
363 0x48, 0x81, 0xec, 0x08, 0x02, 0x00, 0x00, // 0x1a: subq 0x208, %rsp
364 0x48, 0x0f, 0xae, 0x04, 0x24, // 0x21: fxsave64 (%rsp)
365
366 0x48, 0xb9, // 0x26: movabsq <CBMgr>, %rcx
367 // 0x28: Callback manager addr.
368 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
369
370 0x48, 0x8B, 0x55, 0x08, // 0x30: mov rdx, [rbp+0x8]
371 0x48, 0x83, 0xea, 0x06, // 0x34: sub rdx, 0x6
372
373 0x48, 0xb8, // 0x38: movabsq <REntry>, %rax
374 // 0x3a: JIT re-entry fn addr:
375 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
376
377 // 0x42: sub rsp, 0x20 (Allocate shadow space)
378 0x48, 0x83, 0xEC, 0x20,
379 0xff, 0xd0, // 0x46: callq *%rax
380
381 // 0x48: add rsp, 0x20 (Free shadow space)
382 0x48, 0x83, 0xC4, 0x20,
383
384 0x48, 0x89, 0x45, 0x08, // 0x4C: movq %rax, 8(%rbp)
385 0x48, 0x0f, 0xae, 0x0c, 0x24, // 0x50: fxrstor64 (%rsp)
386 0x48, 0x81, 0xc4, 0x08, 0x02, 0x00, 0x00, // 0x55: addq 0x208, %rsp
387 0x41, 0x5f, // 0x5C: popq %r15
388 0x41, 0x5e, // 0x5E: popq %r14
389 0x41, 0x5d, // 0x60: popq %r13
390 0x41, 0x5c, // 0x62: popq %r12
391 0x41, 0x5b, // 0x64: popq %r11
392 0x41, 0x5a, // 0x66: popq %r10
393 0x41, 0x59, // 0x68: popq %r9
394 0x41, 0x58, // 0x6a: popq %r8
395 0x5f, // 0x6c: popq %rdi
396 0x5e, // 0x6d: popq %rsi
397 0x5a, // 0x6e: popq %rdx
398 0x59, // 0x6f: popq %rcx
399 0x5b, // 0x70: popq %rbx
400 0x58, // 0x71: popq %rax
401 0x5d, // 0x72: popq %rbp
402 0xc3, // 0x73: retq
403 };
404
405
406 const unsigned ReentryFnAddrOffset = 0x3a;
407 const unsigned CallbackMgrAddrOffset = 0x28;
408
409 memcpy(ResolverMem, ResolverCode, sizeof(ResolverCode));
410 memcpy(ResolverMem + ReentryFnAddrOffset, &ReentryFn, sizeof(ReentryFn));
411 memcpy(ResolverMem + CallbackMgrAddrOffset, &CallbackMgr,
412 sizeof(CallbackMgr));
413}
414
Lang Hames4f8194e2016-02-10 01:02:33 +0000415void OrcI386::writeResolverCode(uint8_t *ResolverMem, JITReentryFn ReentryFn,
416 void *CallbackMgr) {
417
418 const uint8_t ResolverCode[] = {
Lang Hames1fa0e0e2016-04-25 21:21:20 +0000419 // resolver_entry:
420 0x55, // 0x00: pushl %ebp
421 0x89, 0xe5, // 0x01: movl %esp, %ebp
422 0x54, // 0x03: pushl %esp
423 0x83, 0xe4, 0xf0, // 0x04: andl $-0x10, %esp
424 0x50, // 0x07: pushl %eax
425 0x53, // 0x08: pushl %ebx
426 0x51, // 0x09: pushl %ecx
427 0x52, // 0x0a: pushl %edx
428 0x56, // 0x0b: pushl %esi
429 0x57, // 0x0c: pushl %edi
430 0x81, 0xec, 0x18, 0x02, 0x00, 0x00, // 0x0d: subl $0x218, %esp
431 0x0f, 0xae, 0x44, 0x24, 0x10, // 0x13: fxsave 0x10(%esp)
432 0x8b, 0x75, 0x04, // 0x18: movl 0x4(%ebp), %esi
433 0x83, 0xee, 0x05, // 0x1b: subl $0x5, %esi
434 0x89, 0x74, 0x24, 0x04, // 0x1e: movl %esi, 0x4(%esp)
435 0xc7, 0x04, 0x24, 0x00, 0x00, 0x00,
436 0x00, // 0x22: movl <cbmgr>, (%esp)
437 0xb8, 0x00, 0x00, 0x00, 0x00, // 0x29: movl <reentry>, %eax
438 0xff, 0xd0, // 0x2e: calll *%eax
439 0x89, 0x45, 0x04, // 0x30: movl %eax, 0x4(%ebp)
440 0x0f, 0xae, 0x4c, 0x24, 0x10, // 0x33: fxrstor 0x10(%esp)
441 0x81, 0xc4, 0x18, 0x02, 0x00, 0x00, // 0x38: addl $0x218, %esp
442 0x5f, // 0x3e: popl %edi
443 0x5e, // 0x3f: popl %esi
444 0x5a, // 0x40: popl %edx
445 0x59, // 0x41: popl %ecx
446 0x5b, // 0x42: popl %ebx
447 0x58, // 0x43: popl %eax
448 0x8b, 0x65, 0xfc, // 0x44: movl -0x4(%ebp), %esp
449 0x5d, // 0x48: popl %ebp
450 0xc3 // 0x49: retl
Lang Hames4f8194e2016-02-10 01:02:33 +0000451 };
452
Lang Hamese1fd99c2016-02-21 22:50:26 +0000453 const unsigned ReentryFnAddrOffset = 0x2a;
454 const unsigned CallbackMgrAddrOffset = 0x25;
Lang Hames4f8194e2016-02-10 01:02:33 +0000455
456 memcpy(ResolverMem, ResolverCode, sizeof(ResolverCode));
457 memcpy(ResolverMem + ReentryFnAddrOffset, &ReentryFn, sizeof(ReentryFn));
458 memcpy(ResolverMem + CallbackMgrAddrOffset, &CallbackMgr,
459 sizeof(CallbackMgr));
460}
461
462void OrcI386::writeTrampolines(uint8_t *TrampolineMem, void *ResolverAddr,
463 unsigned NumTrampolines) {
464
465 uint64_t CallRelImm = 0xF1C4C400000000e8;
466 uint64_t Resolver = reinterpret_cast<uint64_t>(ResolverAddr);
467 uint64_t ResolverRel =
Lang Hames1fa0e0e2016-04-25 21:21:20 +0000468 Resolver - reinterpret_cast<uint64_t>(TrampolineMem) - 5;
Lang Hames4f8194e2016-02-10 01:02:33 +0000469
Lang Hames1fa0e0e2016-04-25 21:21:20 +0000470 uint64_t *Trampolines = reinterpret_cast<uint64_t *>(TrampolineMem);
Lang Hames4f8194e2016-02-10 01:02:33 +0000471 for (unsigned I = 0; I < NumTrampolines; ++I, ResolverRel -= TrampolineSize)
472 Trampolines[I] = CallRelImm | (ResolverRel << 8);
473}
474
Lang Hamesef5a0ee2016-04-25 19:56:45 +0000475Error OrcI386::emitIndirectStubsBlock(IndirectStubsInfo &StubsInfo,
Lang Hames1fa0e0e2016-04-25 21:21:20 +0000476 unsigned MinStubs, void *InitialPtrVal) {
Lang Hames4f8194e2016-02-10 01:02:33 +0000477 // Stub format is:
478 //
479 // .section __orc_stubs
480 // stub1:
481 // jmpq *ptr1
482 // .byte 0xC4 ; <- Invalid opcode padding.
483 // .byte 0xF1
484 // stub2:
485 // jmpq *ptr2
486 //
487 // ...
488 //
489 // .section __orc_ptrs
490 // ptr1:
491 // .quad 0x0
492 // ptr2:
493 // .quad 0x0
494 //
495 // ...
496
497 const unsigned StubSize = IndirectStubsInfo::StubSize;
498
499 // Emit at least MinStubs, rounded up to fill the pages allocated.
Lang Hamese4b4ab62019-05-08 02:11:07 +0000500 static const unsigned PageSize = sys::Process::getPageSizeEstimate();
Lang Hames4f8194e2016-02-10 01:02:33 +0000501 unsigned NumPages = ((MinStubs * StubSize) + (PageSize - 1)) / PageSize;
502 unsigned NumStubs = (NumPages * PageSize) / StubSize;
503
504 // Allocate memory for stubs and pointers in one call.
505 std::error_code EC;
Lang Hames1fa0e0e2016-04-25 21:21:20 +0000506 auto StubsMem = sys::OwningMemoryBlock(sys::Memory::allocateMappedMemory(
507 2 * NumPages * PageSize, nullptr,
508 sys::Memory::MF_READ | sys::Memory::MF_WRITE, EC));
Lang Hames4f8194e2016-02-10 01:02:33 +0000509
510 if (EC)
Lang Hamesef5a0ee2016-04-25 19:56:45 +0000511 return errorCodeToError(EC);
Lang Hames4f8194e2016-02-10 01:02:33 +0000512
513 // Create separate MemoryBlocks representing the stubs and pointers.
514 sys::MemoryBlock StubsBlock(StubsMem.base(), NumPages * PageSize);
Lang Hames1fa0e0e2016-04-25 21:21:20 +0000515 sys::MemoryBlock PtrsBlock(static_cast<char *>(StubsMem.base()) +
516 NumPages * PageSize,
Lang Hames4f8194e2016-02-10 01:02:33 +0000517 NumPages * PageSize);
518
519 // Populate the stubs page stubs and mark it executable.
Lang Hames1fa0e0e2016-04-25 21:21:20 +0000520 uint64_t *Stub = reinterpret_cast<uint64_t *>(StubsBlock.base());
Lang Hames4f8194e2016-02-10 01:02:33 +0000521 uint64_t PtrAddr = reinterpret_cast<uint64_t>(PtrsBlock.base());
522 for (unsigned I = 0; I < NumStubs; ++I, PtrAddr += 4)
523 Stub[I] = 0xF1C40000000025ff | (PtrAddr << 16);
524
Lang Hames1fa0e0e2016-04-25 21:21:20 +0000525 if (auto EC = sys::Memory::protectMappedMemory(
526 StubsBlock, sys::Memory::MF_READ | sys::Memory::MF_EXEC))
Lang Hamesef5a0ee2016-04-25 19:56:45 +0000527 return errorCodeToError(EC);
Lang Hames4f8194e2016-02-10 01:02:33 +0000528
529 // Initialize all pointers to point at FailureAddress.
Lang Hames1fa0e0e2016-04-25 21:21:20 +0000530 void **Ptr = reinterpret_cast<void **>(PtrsBlock.base());
Lang Hames4f8194e2016-02-10 01:02:33 +0000531 for (unsigned I = 0; I < NumStubs; ++I)
532 Ptr[I] = InitialPtrVal;
533
534 StubsInfo = IndirectStubsInfo(NumStubs, std::move(StubsMem));
535
Lang Hamesef5a0ee2016-04-25 19:56:45 +0000536 return Error::success();
Lang Hames4f8194e2016-02-10 01:02:33 +0000537}
538
Simon Atanasyan15b68b82018-12-23 12:05:00 +0000539void OrcMips32_Base::writeResolverCode(uint8_t *ResolverMem,
540 JITReentryFn ReentryFn,
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000541 void *CallbackMgr, bool isBigEndian) {
542
543 const uint32_t ResolverCode[] = {
544 // resolver_entry:
545 0x27bdff98, // 0x00: addiu $sp,$sp,-104
546 0xafa20000, // 0x04: sw $v0,0($sp)
547 0xafa30004, // 0x08: sw $v1,4($sp)
548 0xafa40008, // 0x0c: sw $a0,8($sp)
549 0xafa5000c, // 0x10: sw $a1,12($sp)
550 0xafa60010, // 0x14: sw $a2,16($sp)
551 0xafa70014, // 0x18: sw $a3,20($sp)
552 0xafb00018, // 0x1c: sw $s0,24($sp)
553 0xafb1001c, // 0x20: sw $s1,28($sp)
554 0xafb20020, // 0x24: sw $s2,32($sp)
555 0xafb30024, // 0x28: sw $s3,36($sp)
556 0xafb40028, // 0x2c: sw $s4,40($sp)
557 0xafb5002c, // 0x30: sw $s5,44($sp)
558 0xafb60030, // 0x34: sw $s6,48($sp)
559 0xafb70034, // 0x38: sw $s7,52($sp)
560 0xafa80038, // 0x3c: sw $t0,56($sp)
561 0xafa9003c, // 0x40: sw $t1,60($sp)
562 0xafaa0040, // 0x44: sw $t2,64($sp)
563 0xafab0044, // 0x48: sw $t3,68($sp)
564 0xafac0048, // 0x4c: sw $t4,72($sp)
565 0xafad004c, // 0x50: sw $t5,76($sp)
566 0xafae0050, // 0x54: sw $t6,80($sp)
567 0xafaf0054, // 0x58: sw $t7,84($sp)
568 0xafb80058, // 0x5c: sw $t8,88($sp)
569 0xafb9005c, // 0x60: sw $t9,92($sp)
570 0xafbe0060, // 0x64: sw $fp,96($sp)
571 0xafbf0064, // 0x68: sw $ra,100($sp)
572
573 // Callback manager addr.
574 0x00000000, // 0x6c: lui $a0,callbackmgr
575 0x00000000, // 0x70: addiu $a0,$a0,callbackmgr
576
577 0x03e02825, // 0x74: move $a1, $ra
578 0x24a5ffec, // 0x78: addiu $a1,$a1,-20
579
580 // JIT re-entry fn addr:
581 0x00000000, // 0x7c: lui $t9,reentry
582 0x00000000, // 0x80: addiu $t9,$t9,reentry
583
584 0x0320f809, // 0x84: jalr $t9
585 0x00000000, // 0x88: nop
586 0x8fbf0064, // 0x8c: lw $ra,100($sp)
587 0x8fbe0060, // 0x90: lw $fp,96($sp)
588 0x8fb9005c, // 0x94: lw $t9,92($sp)
589 0x8fb80058, // 0x98: lw $t8,88($sp)
590 0x8faf0054, // 0x9c: lw $t7,84($sp)
591 0x8fae0050, // 0xa0: lw $t6,80($sp)
592 0x8fad004c, // 0xa4: lw $t5,76($sp)
593 0x8fac0048, // 0xa8: lw $t4,72($sp)
594 0x8fab0044, // 0xac: lw $t3,68($sp)
595 0x8faa0040, // 0xb0: lw $t2,64($sp)
596 0x8fa9003c, // 0xb4: lw $t1,60($sp)
597 0x8fa80038, // 0xb8: lw $t0,56($sp)
598 0x8fb70034, // 0xbc: lw $s7,52($sp)
599 0x8fb60030, // 0xc0: lw $s6,48($sp)
600 0x8fb5002c, // 0xc4: lw $s5,44($sp)
601 0x8fb40028, // 0xc8: lw $s4,40($sp)
602 0x8fb30024, // 0xcc: lw $s3,36($sp)
603 0x8fb20020, // 0xd0: lw $s2,32($sp)
604 0x8fb1001c, // 0xd4: lw $s1,28($sp)
605 0x8fb00018, // 0xd8: lw $s0,24($sp)
606 0x8fa70014, // 0xdc: lw $a3,20($sp)
Simon Atanasyanda299812018-12-23 12:04:55 +0000607 0x8fa60010, // 0xe0: lw $a2,16($sp)
608 0x8fa5000c, // 0xe4: lw $a1,12($sp)
609 0x8fa40008, // 0xe8: lw $a0,8($sp)
610 0x27bd0068, // 0xec: addiu $sp,$sp,104
611 0x0300f825, // 0xf0: move $ra, $t8
Simon Atanasyan789f4152019-01-12 11:12:08 +0000612 0x03200008, // 0xf4: jr $t9
613 0x00000000, // 0xf8: move $t9, $v0/v1
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000614 };
615
Simon Atanasyan15b68b82018-12-23 12:05:00 +0000616 const unsigned ReentryFnAddrOffset = 0x7c; // JIT re-entry fn addr lui
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000617 const unsigned CallbackMgrAddrOffset = 0x6c; // Callback manager addr lui
Simon Atanasyan789f4152019-01-12 11:12:08 +0000618 const unsigned Offsett = 0xf8;
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000619
620 memcpy(ResolverMem, ResolverCode, sizeof(ResolverCode));
621
Simon Atanasyanf903f782019-01-12 11:12:04 +0000622 // Depending on endian return value will be in v0 or v1.
623 uint32_t MoveVxT9 = isBigEndian ? 0x0060c825 : 0x0040c825;
624 memcpy(ResolverMem + Offsett, &MoveVxT9, sizeof(MoveVxT9));
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000625
626 uint64_t CallMgrAddr = reinterpret_cast<uint64_t>(CallbackMgr);
627 uint32_t CallMgrLUi = 0x3c040000 | (((CallMgrAddr + 0x8000) >> 16) & 0xFFFF);
628 uint32_t CallMgrADDiu = 0x24840000 | ((CallMgrAddr) & 0xFFFF);
Simon Atanasyan15b68b82018-12-23 12:05:00 +0000629 memcpy(ResolverMem + CallbackMgrAddrOffset, &CallMgrLUi, sizeof(CallMgrLUi));
630 memcpy(ResolverMem + CallbackMgrAddrOffset + 4, &CallMgrADDiu,
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000631 sizeof(CallMgrADDiu));
632
633 uint64_t ReentryAddr = reinterpret_cast<uint64_t>(ReentryFn);
634 uint32_t ReentryLUi = 0x3c190000 | (((ReentryAddr + 0x8000) >> 16) & 0xFFFF);
635 uint32_t ReentryADDiu = 0x27390000 | ((ReentryAddr) & 0xFFFF);
Simon Atanasyan15b68b82018-12-23 12:05:00 +0000636 memcpy(ResolverMem + ReentryFnAddrOffset, &ReentryLUi, sizeof(ReentryLUi));
637 memcpy(ResolverMem + ReentryFnAddrOffset + 4, &ReentryADDiu,
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000638 sizeof(ReentryADDiu));
639}
640
Simon Atanasyan15b68b82018-12-23 12:05:00 +0000641void OrcMips32_Base::writeTrampolines(uint8_t *TrampolineMem,
642 void *ResolverAddr,
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000643 unsigned NumTrampolines) {
644
645 uint32_t *Trampolines = reinterpret_cast<uint32_t *>(TrampolineMem);
646 uint64_t ResolveAddr = reinterpret_cast<uint64_t>(ResolverAddr);
647 uint32_t RHiAddr = ((ResolveAddr + 0x8000) >> 16);
648
649 for (unsigned I = 0; I < NumTrampolines; ++I) {
650 Trampolines[5 * I + 0] = 0x03e0c025; // move $t8,$ra
651 Trampolines[5 * I + 1] = 0x3c190000 | (RHiAddr & 0xFFFF); // lui $t9,resolveAddr
652 Trampolines[5 * I + 2] = 0x27390000 | (ResolveAddr & 0xFFFF); // addiu $t9,$t9,resolveAddr
653 Trampolines[5 * I + 3] = 0x0320f809; // jalr $t9
654 Trampolines[5 * I + 4] = 0x00000000; // nop
655 }
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000656}
657
658Error OrcMips32_Base::emitIndirectStubsBlock(IndirectStubsInfo &StubsInfo,
659 unsigned MinStubs,
660 void *InitialPtrVal) {
661 // Stub format is:
662 //
663 // .section __orc_stubs
664 // stub1:
665 // lui $t9, ptr1
666 // lw $t9, %lo(ptr1)($t9)
667 // jr $t9
668 // stub2:
669 // lui $t9, ptr2
670 // lw $t9,%lo(ptr1)($t9)
671 // jr $t9
672 //
673 // ...
674 //
675 // .section __orc_ptrs
676 // ptr1:
677 // .word 0x0
678 // ptr2:
679 // .word 0x0
680 //
681 // ...
682
683 const unsigned StubSize = IndirectStubsInfo::StubSize;
684
685 // Emit at least MinStubs, rounded up to fill the pages allocated.
Lang Hamese4b4ab62019-05-08 02:11:07 +0000686 static const unsigned PageSize = sys::Process::getPageSizeEstimate();
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000687 unsigned NumPages = ((MinStubs * StubSize) + (PageSize - 1)) / PageSize;
688 unsigned NumStubs = (NumPages * PageSize) / StubSize;
689
690 // Allocate memory for stubs and pointers in one call.
691 std::error_code EC;
692 auto StubsMem = sys::OwningMemoryBlock(sys::Memory::allocateMappedMemory(
693 2 * NumPages * PageSize, nullptr,
694 sys::Memory::MF_READ | sys::Memory::MF_WRITE, EC));
695
696 if (EC)
697 return errorCodeToError(EC);
698
699 // Create separate MemoryBlocks representing the stubs and pointers.
700 sys::MemoryBlock StubsBlock(StubsMem.base(), NumPages * PageSize);
701 sys::MemoryBlock PtrsBlock(static_cast<char *>(StubsMem.base()) +
702 NumPages * PageSize,
703 NumPages * PageSize);
704
705 // Populate the stubs page stubs and mark it executable.
706 uint32_t *Stub = reinterpret_cast<uint32_t *>(StubsBlock.base());
707 uint64_t PtrAddr = reinterpret_cast<uint64_t>(Stub) + NumPages * PageSize;
708
709 for (unsigned I = 0; I < NumStubs; ++I) {
710 uint32_t HiAddr = ((PtrAddr + 0x8000) >> 16);
711 Stub[4 * I + 0] = 0x3c190000 | (HiAddr & 0xFFFF); // lui $t9,ptr1
712 Stub[4 * I + 1] = 0x8f390000 | (PtrAddr & 0xFFFF); // lw $t9,%lo(ptr1)($t9)
713 Stub[4 * I + 2] = 0x03200008; // jr $t9
714 Stub[4 * I + 3] = 0x00000000; // nop
715 PtrAddr += 4;
716 }
717
718 if (auto EC = sys::Memory::protectMappedMemory(
719 StubsBlock, sys::Memory::MF_READ | sys::Memory::MF_EXEC))
720 return errorCodeToError(EC);
721
722 // Initialize all pointers to point at FailureAddress.
723 void **Ptr = reinterpret_cast<void **>(PtrsBlock.base());
724 for (unsigned I = 0; I < NumStubs; ++I)
725 Ptr[I] = InitialPtrVal;
726
727 StubsInfo = IndirectStubsInfo(NumStubs, std::move(StubsMem));
728
729 return Error::success();
730}
731
732void OrcMips64::writeResolverCode(uint8_t *ResolverMem, JITReentryFn ReentryFn,
733 void *CallbackMgr) {
734
735 const uint32_t ResolverCode[] = {
736 //resolver_entry:
737 0x67bdff30, // 0x00: daddiu $sp,$sp,-208
738 0xffa20000, // 0x04: sd v0,0(sp)
739 0xffa30008, // 0x08: sd v1,8(sp)
740 0xffa40010, // 0x0c: sd a0,16(sp)
741 0xffa50018, // 0x10: sd a1,24(sp)
742 0xffa60020, // 0x14: sd a2,32(sp)
743 0xffa70028, // 0x18: sd a3,40(sp)
744 0xffa80030, // 0x1c: sd a4,48(sp)
745 0xffa90038, // 0x20: sd a5,56(sp)
746 0xffaa0040, // 0x24: sd a6,64(sp)
747 0xffab0048, // 0x28: sd a7,72(sp)
748 0xffac0050, // 0x2c: sd t0,80(sp)
749 0xffad0058, // 0x30: sd t1,88(sp)
750 0xffae0060, // 0x34: sd t2,96(sp)
751 0xffaf0068, // 0x38: sd t3,104(sp)
752 0xffb00070, // 0x3c: sd s0,112(sp)
753 0xffb10078, // 0x40: sd s1,120(sp)
754 0xffb20080, // 0x44: sd s2,128(sp)
755 0xffb30088, // 0x48: sd s3,136(sp)
756 0xffb40090, // 0x4c: sd s4,144(sp)
757 0xffb50098, // 0x50: sd s5,152(sp)
758 0xffb600a0, // 0x54: sd s6,160(sp)
759 0xffb700a8, // 0x58: sd s7,168(sp)
760 0xffb800b0, // 0x5c: sd t8,176(sp)
761 0xffb900b8, // 0x60: sd t9,184(sp)
Simon Atanasyan4553cda2018-12-23 12:05:04 +0000762 0xffbe00c0, // 0x64: sd fp,192(sp)
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000763 0xffbf00c8, // 0x68: sd ra,200(sp)
764
765 // Callback manager addr.
766 0x00000000, // 0x6c: lui $a0,heighest(callbackmgr)
767 0x00000000, // 0x70: daddiu $a0,$a0,heigher(callbackmgr)
768 0x00000000, // 0x74: dsll $a0,$a0,16
769 0x00000000, // 0x78: daddiu $a0,$a0,hi(callbackmgr)
770 0x00000000, // 0x7c: dsll $a0,$a0,16
771 0x00000000, // 0x80: daddiu $a0,$a0,lo(callbackmgr)
772
773 0x03e02825, // 0x84: move $a1, $ra
774 0x64a5ffdc, // 0x88: daddiu $a1,$a1,-36
775
776 // JIT re-entry fn addr:
777 0x00000000, // 0x8c: lui $t9,reentry
778 0x00000000, // 0x90: daddiu $t9,$t9,reentry
779 0x00000000, // 0x94: dsll $t9,$t9,
780 0x00000000, // 0x98: daddiu $t9,$t9,
781 0x00000000, // 0x9c: dsll $t9,$t9,
782 0x00000000, // 0xa0: daddiu $t9,$t9,
783 0x0320f809, // 0xa4: jalr $t9
784 0x00000000, // 0xa8: nop
785 0xdfbf00c8, // 0xac: ld ra, 200(sp)
Simon Atanasyan4553cda2018-12-23 12:05:04 +0000786 0xdfbe00c0, // 0xb0: ld fp, 192(sp)
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000787 0xdfb900b8, // 0xb4: ld t9, 184(sp)
788 0xdfb800b0, // 0xb8: ld t8, 176(sp)
789 0xdfb700a8, // 0xbc: ld s7, 168(sp)
790 0xdfb600a0, // 0xc0: ld s6, 160(sp)
791 0xdfb50098, // 0xc4: ld s5, 152(sp)
792 0xdfb40090, // 0xc8: ld s4, 144(sp)
793 0xdfb30088, // 0xcc: ld s3, 136(sp)
794 0xdfb20080, // 0xd0: ld s2, 128(sp)
795 0xdfb10078, // 0xd4: ld s1, 120(sp)
796 0xdfb00070, // 0xd8: ld s0, 112(sp)
797 0xdfaf0068, // 0xdc: ld t3, 104(sp)
798 0xdfae0060, // 0xe0: ld t2, 96(sp)
799 0xdfad0058, // 0xe4: ld t1, 88(sp)
800 0xdfac0050, // 0xe8: ld t0, 80(sp)
801 0xdfab0048, // 0xec: ld a7, 72(sp)
802 0xdfaa0040, // 0xf0: ld a6, 64(sp)
803 0xdfa90038, // 0xf4: ld a5, 56(sp)
804 0xdfa80030, // 0xf8: ld a4, 48(sp)
805 0xdfa70028, // 0xfc: ld a3, 40(sp)
806 0xdfa60020, // 0x100: ld a2, 32(sp)
807 0xdfa50018, // 0x104: ld a1, 24(sp)
808 0xdfa40010, // 0x108: ld a0, 16(sp)
809 0xdfa30008, // 0x10c: ld v1, 8(sp)
810 0x67bd00d0, // 0x110: daddiu $sp,$sp,208
811 0x0300f825, // 0x114: move $ra, $t8
Simon Atanasyan789f4152019-01-12 11:12:08 +0000812 0x03200008, // 0x118: jr $t9
813 0x0040c825, // 0x11c: move $t9, $v0
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000814 };
815
816 const unsigned ReentryFnAddrOffset = 0x8c; // JIT re-entry fn addr lui
817 const unsigned CallbackMgrAddrOffset = 0x6c; // Callback manager addr lui
818
819 memcpy(ResolverMem, ResolverCode, sizeof(ResolverCode));
820
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000821 uint64_t CallMgrAddr = reinterpret_cast<uint64_t>(CallbackMgr);
822
823 uint32_t CallMgrLUi =
824 0x3c040000 | (((CallMgrAddr + 0x800080008000) >> 48) & 0xFFFF);
825 uint32_t CallMgrDADDiu =
Simon Atanasyan15b68b82018-12-23 12:05:00 +0000826 0x64840000 | (((CallMgrAddr + 0x80008000) >> 32) & 0xFFFF);
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000827 uint32_t CallMgrDSLL = 0x00042438;
828 uint32_t CallMgrDADDiu2 =
829 0x64840000 | ((((CallMgrAddr + 0x8000) >> 16) & 0xFFFF));
830 uint32_t CallMgrDSLL2 = 0x00042438;
Simon Atanasyan15b68b82018-12-23 12:05:00 +0000831 uint32_t CallMgrDADDiu3 = 0x64840000 | ((CallMgrAddr)&0xFFFF);
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000832
Simon Atanasyan15b68b82018-12-23 12:05:00 +0000833 memcpy(ResolverMem + CallbackMgrAddrOffset, &CallMgrLUi, sizeof(CallMgrLUi));
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000834 memcpy(ResolverMem + (CallbackMgrAddrOffset + 4), &CallMgrDADDiu,
835 sizeof(CallMgrDADDiu));
836 memcpy(ResolverMem + (CallbackMgrAddrOffset + 8), &CallMgrDSLL,
837 sizeof(CallMgrDSLL));
838 memcpy(ResolverMem + (CallbackMgrAddrOffset + 12), &CallMgrDADDiu2,
839 sizeof(CallMgrDADDiu2));
840 memcpy(ResolverMem + (CallbackMgrAddrOffset + 16), &CallMgrDSLL2,
841 sizeof(CallMgrDSLL2));
842 memcpy(ResolverMem + (CallbackMgrAddrOffset + 20), &CallMgrDADDiu3,
843 sizeof(CallMgrDADDiu3));
844
845 uint64_t ReentryAddr = reinterpret_cast<uint64_t>(ReentryFn);
846
847 uint32_t ReentryLUi =
848 0x3c190000 | (((ReentryAddr + 0x800080008000) >> 48) & 0xFFFF);
849
850 uint32_t ReentryDADDiu =
851 0x67390000 | (((ReentryAddr + 0x80008000) >> 32) & 0xFFFF);
852
853 uint32_t ReentryDSLL = 0x0019cc38;
854
855 uint32_t ReentryDADDiu2 =
856 0x67390000 | (((ReentryAddr + 0x8000) >> 16) & 0xFFFF);
857
858 uint32_t ReentryDSLL2 = 0x0019cc38;
859
Simon Atanasyan15b68b82018-12-23 12:05:00 +0000860 uint32_t ReentryDADDiu3 = 0x67390000 | ((ReentryAddr)&0xFFFF);
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000861
Simon Atanasyan15b68b82018-12-23 12:05:00 +0000862 memcpy(ResolverMem + ReentryFnAddrOffset, &ReentryLUi, sizeof(ReentryLUi));
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000863 memcpy(ResolverMem + (ReentryFnAddrOffset + 4), &ReentryDADDiu,
864 sizeof(ReentryDADDiu));
865 memcpy(ResolverMem + (ReentryFnAddrOffset + 8), &ReentryDSLL,
866 sizeof(ReentryDSLL));
867 memcpy(ResolverMem + (ReentryFnAddrOffset + 12), &ReentryDADDiu2,
868 sizeof(ReentryDADDiu2));
869 memcpy(ResolverMem + (ReentryFnAddrOffset + 16), &ReentryDSLL2,
870 sizeof(ReentryDSLL2));
871 memcpy(ResolverMem + (ReentryFnAddrOffset + 20), &ReentryDADDiu3,
872 sizeof(ReentryDADDiu3));
873}
874
875void OrcMips64::writeTrampolines(uint8_t *TrampolineMem, void *ResolverAddr,
876 unsigned NumTrampolines) {
877
878 uint32_t *Trampolines = reinterpret_cast<uint32_t *>(TrampolineMem);
879 uint64_t ResolveAddr = reinterpret_cast<uint64_t>(ResolverAddr);
880
881 uint64_t HeighestAddr = ((ResolveAddr + 0x800080008000) >> 48);
882 uint64_t HeigherAddr = ((ResolveAddr + 0x80008000) >> 32);
883 uint64_t HiAddr = ((ResolveAddr + 0x8000) >> 16);
884
885 for (unsigned I = 0; I < NumTrampolines; ++I) {
886 Trampolines[10 * I + 0] = 0x03e0c025; // move $t8,$ra
887 Trampolines[10 * I + 1] = 0x3c190000 | (HeighestAddr & 0xFFFF); // lui $t9,resolveAddr
888 Trampolines[10 * I + 2] = 0x67390000 | (HeigherAddr & 0xFFFF); // daddiu $t9,$t9,%higher(resolveAddr)
889 Trampolines[10 * I + 3] = 0x0019cc38; // dsll $t9,$t9,16
890 Trampolines[10 * I + 4] = 0x67390000 | (HiAddr & 0xFFFF); // daddiu $t9,$t9,%hi(ptr)
891 Trampolines[10 * I + 5] = 0x0019cc38; // dsll $t9,$t9,16
892 Trampolines[10 * I + 6] = 0x67390000 | (ResolveAddr & 0xFFFF); // daddiu $t9,$t9,%lo(ptr)
893 Trampolines[10 * I + 7] = 0x0320f809; // jalr $t9
894 Trampolines[10 * I + 8] = 0x00000000; // nop
895 Trampolines[10 * I + 9] = 0x00000000; // nop
896 }
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000897}
898
899Error OrcMips64::emitIndirectStubsBlock(IndirectStubsInfo &StubsInfo,
900 unsigned MinStubs,
901 void *InitialPtrVal) {
902 // Stub format is:
903 //
904 // .section __orc_stubs
905 // stub1:
906 // lui $t9,ptr1
907 // dsll $t9,$t9,16
908 // daddiu $t9,$t9,%hi(ptr)
909 // dsll $t9,$t9,16
910 // ld $t9,%lo(ptr)
911 // jr $t9
912 // stub2:
913 // lui $t9,ptr1
914 // dsll $t9,$t9,16
915 // daddiu $t9,$t9,%hi(ptr)
916 // dsll $t9,$t9,16
917 // ld $t9,%lo(ptr)
918 // jr $t9
919 //
920 // ...
921 //
922 // .section __orc_ptrs
923 // ptr1:
924 // .dword 0x0
925 // ptr2:
926 // .dword 0x0
927 //
928 // ...
929 const unsigned StubSize = IndirectStubsInfo::StubSize;
930
931 // Emit at least MinStubs, rounded up to fill the pages allocated.
Lang Hamese4b4ab62019-05-08 02:11:07 +0000932 static const unsigned PageSize = sys::Process::getPageSizeEstimate();
Petar Jovanovic5abf4bb2018-09-11 13:10:04 +0000933 unsigned NumPages = ((MinStubs * StubSize) + (PageSize - 1)) / PageSize;
934 unsigned NumStubs = (NumPages * PageSize) / StubSize;
935
936 // Allocate memory for stubs and pointers in one call.
937 std::error_code EC;
938 auto StubsMem = sys::OwningMemoryBlock(sys::Memory::allocateMappedMemory(
939 2 * NumPages * PageSize, nullptr,
940 sys::Memory::MF_READ | sys::Memory::MF_WRITE, EC));
941
942 if (EC)
943 return errorCodeToError(EC);
944
945 // Create separate MemoryBlocks representing the stubs and pointers.
946 sys::MemoryBlock StubsBlock(StubsMem.base(), NumPages * PageSize);
947 sys::MemoryBlock PtrsBlock(static_cast<char *>(StubsMem.base()) +
948 NumPages * PageSize,
949 NumPages * PageSize);
950
951 // Populate the stubs page stubs and mark it executable.
952 uint32_t *Stub = reinterpret_cast<uint32_t *>(StubsBlock.base());
953 uint64_t PtrAddr = reinterpret_cast<uint64_t>(PtrsBlock.base());
954
955 for (unsigned I = 0; I < NumStubs; ++I, PtrAddr += 8) {
956 uint64_t HeighestAddr = ((PtrAddr + 0x800080008000) >> 48);
957 uint64_t HeigherAddr = ((PtrAddr + 0x80008000) >> 32);
958 uint64_t HiAddr = ((PtrAddr + 0x8000) >> 16);
959 Stub[8 * I + 0] = 0x3c190000 | (HeighestAddr & 0xFFFF); // lui $t9,ptr1
960 Stub[8 * I + 1] = 0x67390000 | (HeigherAddr & 0xFFFF); // daddiu $t9,$t9,%higher(ptr)
961 Stub[8 * I + 2] = 0x0019cc38; // dsll $t9,$t9,16
962 Stub[8 * I + 3] = 0x67390000 | (HiAddr & 0xFFFF); // daddiu $t9,$t9,%hi(ptr)
963 Stub[8 * I + 4] = 0x0019cc38; // dsll $t9,$t9,16
964 Stub[8 * I + 5] = 0xdf390000 | (PtrAddr & 0xFFFF); // ld $t9,%lo(ptr)
965 Stub[8 * I + 6] = 0x03200008; // jr $t9
966 Stub[8 * I + 7] = 0x00000000; // nop
967 }
968
969 if (auto EC = sys::Memory::protectMappedMemory(
970 StubsBlock, sys::Memory::MF_READ | sys::Memory::MF_EXEC))
971 return errorCodeToError(EC);
972
973 // Initialize all pointers to point at FailureAddress.
974 void **Ptr = reinterpret_cast<void **>(PtrsBlock.base());
975 for (unsigned I = 0; I < NumStubs; ++I)
976 Ptr[I] = InitialPtrVal;
977
978 StubsInfo = IndirectStubsInfo(NumStubs, std::move(StubsMem));
979
980 return Error::success();
981}
Lang Hamese7380612015-02-21 20:44:36 +0000982} // End namespace orc.
983} // End namespace llvm.