| Lang Hames | 4821c24 | 2016-05-07 03:36:38 +0000 | [diff] [blame] | 1 | //===------------- OrcABISupport.cpp - ABI specific support code ----------===// |
| Lang Hames | 4df7ba7 | 2015-10-26 06:40:28 +0000 | [diff] [blame] | 2 | // |
| Chandler Carruth | 2946cd7 | 2019-01-19 08:50:56 +0000 | [diff] [blame] | 3 | // 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 Hames | 4df7ba7 | 2015-10-26 06:40:28 +0000 | [diff] [blame] | 6 | // |
| 7 | //===----------------------------------------------------------------------===// |
| 8 | |
| Lang Hames | 4821c24 | 2016-05-07 03:36:38 +0000 | [diff] [blame] | 9 | #include "llvm/ExecutionEngine/Orc/OrcABISupport.h" |
| Lang Hames | 98c2ac1 | 2015-10-19 17:43:51 +0000 | [diff] [blame] | 10 | #include "llvm/Support/Process.h" |
| Lang Hames | 93de2a1 | 2015-01-23 21:25:00 +0000 | [diff] [blame] | 11 | |
| Lang Hames | 93de2a1 | 2015-01-23 21:25:00 +0000 | [diff] [blame] | 12 | namespace llvm { |
| Lang Hames | e738061 | 2015-02-21 20:44:36 +0000 | [diff] [blame] | 13 | namespace orc { |
| Lang Hames | 93de2a1 | 2015-01-23 21:25:00 +0000 | [diff] [blame] | 14 | |
| Lang Hames | f74e93b | 2016-04-29 21:32:00 +0000 | [diff] [blame] | 15 | void OrcAArch64::writeResolverCode(uint8_t *ResolverMem, JITReentryFn ReentryFn, |
| 16 | void *CallbackMgr) { |
| 17 | |
| 18 | const uint32_t ResolverCode[] = { |
| 19 | // resolver_entry: |
| Lang Hames | 2307f40 | 2016-05-01 00:14:45 +0000 | [diff] [blame] | 20 | 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 Hames | f74e93b | 2016-04-29 21:32:00 +0000 | [diff] [blame] | 92 | }; |
| 93 | |
| Lang Hames | 2307f40 | 2016-05-01 00:14:45 +0000 | [diff] [blame] | 94 | const unsigned ReentryFnAddrOffset = 0x110; |
| 95 | const unsigned CallbackMgrAddrOffset = 0x118; |
| Lang Hames | f74e93b | 2016-04-29 21:32:00 +0000 | [diff] [blame] | 96 | |
| 97 | memcpy(ResolverMem, ResolverCode, sizeof(ResolverCode)); |
| 98 | memcpy(ResolverMem + ReentryFnAddrOffset, &ReentryFn, sizeof(ReentryFn)); |
| 99 | memcpy(ResolverMem + CallbackMgrAddrOffset, &CallbackMgr, |
| 100 | sizeof(CallbackMgr)); |
| 101 | } |
| 102 | |
| 103 | void 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 Hames | d4d6a1a | 2017-09-15 06:50:19 +0000 | [diff] [blame] | 118 | Trampolines[3 * I + 1] = 0x58000010 | (OffsetToPtr << 3); // adr x16, Lptr |
| Lang Hames | f74e93b | 2016-04-29 21:32:00 +0000 | [diff] [blame] | 119 | Trampolines[3 * I + 2] = 0xd63f0200; // blr x16 |
| 120 | } |
| 121 | |
| 122 | } |
| 123 | |
| 124 | Error 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 Hames | e4b4ab6 | 2019-05-08 02:11:07 +0000 | [diff] [blame] | 150 | static const unsigned PageSize = sys::Process::getPageSizeEstimate(); |
| Lang Hames | f74e93b | 2016-04-29 21:32:00 +0000 | [diff] [blame] | 151 | 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 Hames | 4821c24 | 2016-05-07 03:36:38 +0000 | [diff] [blame] | 191 | void OrcX86_64_Base::writeTrampolines(uint8_t *TrampolineMem, |
| 192 | void *ResolverAddr, |
| 193 | unsigned NumTrampolines) { |
| Rafael Espindola | 2f34463 | 2015-11-03 16:25:20 +0000 | [diff] [blame] | 194 | |
| Rafael Espindola | e63e018 | 2015-11-03 16:40:37 +0000 | [diff] [blame] | 195 | unsigned OffsetToPtr = NumTrampolines * TrampolineSize; |
| 196 | |
| Lang Hames | 1fa0e0e | 2016-04-25 21:21:20 +0000 | [diff] [blame] | 197 | memcpy(TrampolineMem + OffsetToPtr, &ResolverAddr, sizeof(void *)); |
| Rafael Espindola | e63e018 | 2015-11-03 16:40:37 +0000 | [diff] [blame] | 198 | |
| Lang Hames | 1fa0e0e | 2016-04-25 21:21:20 +0000 | [diff] [blame] | 199 | uint64_t *Trampolines = reinterpret_cast<uint64_t *>(TrampolineMem); |
| Rafael Espindola | e63e018 | 2015-11-03 16:40:37 +0000 | [diff] [blame] | 200 | uint64_t CallIndirPCRel = 0xf1c40000000015ff; |
| 201 | |
| 202 | for (unsigned I = 0; I < NumTrampolines; ++I, OffsetToPtr -= TrampolineSize) |
| 203 | Trampolines[I] = CallIndirPCRel | ((OffsetToPtr - 6) << 16); |
| Lang Hames | 2754714 | 2015-02-17 01:18:38 +0000 | [diff] [blame] | 204 | } |
| 205 | |
| Lang Hames | 4821c24 | 2016-05-07 03:36:38 +0000 | [diff] [blame] | 206 | Error OrcX86_64_Base::emitIndirectStubsBlock(IndirectStubsInfo &StubsInfo, |
| 207 | unsigned MinStubs, |
| 208 | void *InitialPtrVal) { |
| Lang Hames | 98c2ac1 | 2015-10-19 17:43:51 +0000 | [diff] [blame] | 209 | // 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 Hames | e4b4ab6 | 2019-05-08 02:11:07 +0000 | [diff] [blame] | 232 | static const unsigned PageSize = sys::Process::getPageSizeEstimate(); |
| Lang Hames | 98c2ac1 | 2015-10-19 17:43:51 +0000 | [diff] [blame] | 233 | 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 Hames | 1fa0e0e | 2016-04-25 21:21:20 +0000 | [diff] [blame] | 238 | auto StubsMem = sys::OwningMemoryBlock(sys::Memory::allocateMappedMemory( |
| 239 | 2 * NumPages * PageSize, nullptr, |
| 240 | sys::Memory::MF_READ | sys::Memory::MF_WRITE, EC)); |
| Lang Hames | 98c2ac1 | 2015-10-19 17:43:51 +0000 | [diff] [blame] | 241 | |
| 242 | if (EC) |
| Lang Hames | ef5a0ee | 2016-04-25 19:56:45 +0000 | [diff] [blame] | 243 | return errorCodeToError(EC); |
| Lang Hames | 98c2ac1 | 2015-10-19 17:43:51 +0000 | [diff] [blame] | 244 | |
| 245 | // Create separate MemoryBlocks representing the stubs and pointers. |
| Lang Hames | 5796eb2 | 2015-10-31 00:55:32 +0000 | [diff] [blame] | 246 | sys::MemoryBlock StubsBlock(StubsMem.base(), NumPages * PageSize); |
| Lang Hames | 1fa0e0e | 2016-04-25 21:21:20 +0000 | [diff] [blame] | 247 | sys::MemoryBlock PtrsBlock(static_cast<char *>(StubsMem.base()) + |
| 248 | NumPages * PageSize, |
| Lang Hames | 98c2ac1 | 2015-10-19 17:43:51 +0000 | [diff] [blame] | 249 | NumPages * PageSize); |
| 250 | |
| 251 | // Populate the stubs page stubs and mark it executable. |
| Lang Hames | 1fa0e0e | 2016-04-25 21:21:20 +0000 | [diff] [blame] | 252 | uint64_t *Stub = reinterpret_cast<uint64_t *>(StubsBlock.base()); |
| 253 | uint64_t PtrOffsetField = static_cast<uint64_t>(NumPages * PageSize - 6) |
| 254 | << 16; |
| Lang Hames | 98c2ac1 | 2015-10-19 17:43:51 +0000 | [diff] [blame] | 255 | for (unsigned I = 0; I < NumStubs; ++I) |
| 256 | Stub[I] = 0xF1C40000000025ff | PtrOffsetField; |
| 257 | |
| Lang Hames | 1fa0e0e | 2016-04-25 21:21:20 +0000 | [diff] [blame] | 258 | if (auto EC = sys::Memory::protectMappedMemory( |
| 259 | StubsBlock, sys::Memory::MF_READ | sys::Memory::MF_EXEC)) |
| Lang Hames | ef5a0ee | 2016-04-25 19:56:45 +0000 | [diff] [blame] | 260 | return errorCodeToError(EC); |
| Lang Hames | 98c2ac1 | 2015-10-19 17:43:51 +0000 | [diff] [blame] | 261 | |
| 262 | // Initialize all pointers to point at FailureAddress. |
| Lang Hames | 1fa0e0e | 2016-04-25 21:21:20 +0000 | [diff] [blame] | 263 | void **Ptr = reinterpret_cast<void **>(PtrsBlock.base()); |
| Lang Hames | 98c2ac1 | 2015-10-19 17:43:51 +0000 | [diff] [blame] | 264 | for (unsigned I = 0; I < NumStubs; ++I) |
| 265 | Ptr[I] = InitialPtrVal; |
| 266 | |
| Lang Hames | e28b118 | 2016-02-02 19:31:15 +0000 | [diff] [blame] | 267 | StubsInfo = IndirectStubsInfo(NumStubs, std::move(StubsMem)); |
| Lang Hames | 98c2ac1 | 2015-10-19 17:43:51 +0000 | [diff] [blame] | 268 | |
| Lang Hames | ef5a0ee | 2016-04-25 19:56:45 +0000 | [diff] [blame] | 269 | return Error::success(); |
| Lang Hames | 98c2ac1 | 2015-10-19 17:43:51 +0000 | [diff] [blame] | 270 | } |
| 271 | |
| Lang Hames | 4821c24 | 2016-05-07 03:36:38 +0000 | [diff] [blame] | 272 | void 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 | |
| 339 | void 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 Hames | 4f8194e | 2016-02-10 01:02:33 +0000 | [diff] [blame] | 415 | void OrcI386::writeResolverCode(uint8_t *ResolverMem, JITReentryFn ReentryFn, |
| 416 | void *CallbackMgr) { |
| 417 | |
| 418 | const uint8_t ResolverCode[] = { |
| Lang Hames | 1fa0e0e | 2016-04-25 21:21:20 +0000 | [diff] [blame] | 419 | // 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 Hames | 4f8194e | 2016-02-10 01:02:33 +0000 | [diff] [blame] | 451 | }; |
| 452 | |
| Lang Hames | e1fd99c | 2016-02-21 22:50:26 +0000 | [diff] [blame] | 453 | const unsigned ReentryFnAddrOffset = 0x2a; |
| 454 | const unsigned CallbackMgrAddrOffset = 0x25; |
| Lang Hames | 4f8194e | 2016-02-10 01:02:33 +0000 | [diff] [blame] | 455 | |
| 456 | memcpy(ResolverMem, ResolverCode, sizeof(ResolverCode)); |
| 457 | memcpy(ResolverMem + ReentryFnAddrOffset, &ReentryFn, sizeof(ReentryFn)); |
| 458 | memcpy(ResolverMem + CallbackMgrAddrOffset, &CallbackMgr, |
| 459 | sizeof(CallbackMgr)); |
| 460 | } |
| 461 | |
| 462 | void 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 Hames | 1fa0e0e | 2016-04-25 21:21:20 +0000 | [diff] [blame] | 468 | Resolver - reinterpret_cast<uint64_t>(TrampolineMem) - 5; |
| Lang Hames | 4f8194e | 2016-02-10 01:02:33 +0000 | [diff] [blame] | 469 | |
| Lang Hames | 1fa0e0e | 2016-04-25 21:21:20 +0000 | [diff] [blame] | 470 | uint64_t *Trampolines = reinterpret_cast<uint64_t *>(TrampolineMem); |
| Lang Hames | 4f8194e | 2016-02-10 01:02:33 +0000 | [diff] [blame] | 471 | for (unsigned I = 0; I < NumTrampolines; ++I, ResolverRel -= TrampolineSize) |
| 472 | Trampolines[I] = CallRelImm | (ResolverRel << 8); |
| 473 | } |
| 474 | |
| Lang Hames | ef5a0ee | 2016-04-25 19:56:45 +0000 | [diff] [blame] | 475 | Error OrcI386::emitIndirectStubsBlock(IndirectStubsInfo &StubsInfo, |
| Lang Hames | 1fa0e0e | 2016-04-25 21:21:20 +0000 | [diff] [blame] | 476 | unsigned MinStubs, void *InitialPtrVal) { |
| Lang Hames | 4f8194e | 2016-02-10 01:02:33 +0000 | [diff] [blame] | 477 | // 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 Hames | e4b4ab6 | 2019-05-08 02:11:07 +0000 | [diff] [blame] | 500 | static const unsigned PageSize = sys::Process::getPageSizeEstimate(); |
| Lang Hames | 4f8194e | 2016-02-10 01:02:33 +0000 | [diff] [blame] | 501 | 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 Hames | 1fa0e0e | 2016-04-25 21:21:20 +0000 | [diff] [blame] | 506 | auto StubsMem = sys::OwningMemoryBlock(sys::Memory::allocateMappedMemory( |
| 507 | 2 * NumPages * PageSize, nullptr, |
| 508 | sys::Memory::MF_READ | sys::Memory::MF_WRITE, EC)); |
| Lang Hames | 4f8194e | 2016-02-10 01:02:33 +0000 | [diff] [blame] | 509 | |
| 510 | if (EC) |
| Lang Hames | ef5a0ee | 2016-04-25 19:56:45 +0000 | [diff] [blame] | 511 | return errorCodeToError(EC); |
| Lang Hames | 4f8194e | 2016-02-10 01:02:33 +0000 | [diff] [blame] | 512 | |
| 513 | // Create separate MemoryBlocks representing the stubs and pointers. |
| 514 | sys::MemoryBlock StubsBlock(StubsMem.base(), NumPages * PageSize); |
| Lang Hames | 1fa0e0e | 2016-04-25 21:21:20 +0000 | [diff] [blame] | 515 | sys::MemoryBlock PtrsBlock(static_cast<char *>(StubsMem.base()) + |
| 516 | NumPages * PageSize, |
| Lang Hames | 4f8194e | 2016-02-10 01:02:33 +0000 | [diff] [blame] | 517 | NumPages * PageSize); |
| 518 | |
| 519 | // Populate the stubs page stubs and mark it executable. |
| Lang Hames | 1fa0e0e | 2016-04-25 21:21:20 +0000 | [diff] [blame] | 520 | uint64_t *Stub = reinterpret_cast<uint64_t *>(StubsBlock.base()); |
| Lang Hames | 4f8194e | 2016-02-10 01:02:33 +0000 | [diff] [blame] | 521 | 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 Hames | 1fa0e0e | 2016-04-25 21:21:20 +0000 | [diff] [blame] | 525 | if (auto EC = sys::Memory::protectMappedMemory( |
| 526 | StubsBlock, sys::Memory::MF_READ | sys::Memory::MF_EXEC)) |
| Lang Hames | ef5a0ee | 2016-04-25 19:56:45 +0000 | [diff] [blame] | 527 | return errorCodeToError(EC); |
| Lang Hames | 4f8194e | 2016-02-10 01:02:33 +0000 | [diff] [blame] | 528 | |
| 529 | // Initialize all pointers to point at FailureAddress. |
| Lang Hames | 1fa0e0e | 2016-04-25 21:21:20 +0000 | [diff] [blame] | 530 | void **Ptr = reinterpret_cast<void **>(PtrsBlock.base()); |
| Lang Hames | 4f8194e | 2016-02-10 01:02:33 +0000 | [diff] [blame] | 531 | for (unsigned I = 0; I < NumStubs; ++I) |
| 532 | Ptr[I] = InitialPtrVal; |
| 533 | |
| 534 | StubsInfo = IndirectStubsInfo(NumStubs, std::move(StubsMem)); |
| 535 | |
| Lang Hames | ef5a0ee | 2016-04-25 19:56:45 +0000 | [diff] [blame] | 536 | return Error::success(); |
| Lang Hames | 4f8194e | 2016-02-10 01:02:33 +0000 | [diff] [blame] | 537 | } |
| 538 | |
| Simon Atanasyan | 15b68b8 | 2018-12-23 12:05:00 +0000 | [diff] [blame] | 539 | void OrcMips32_Base::writeResolverCode(uint8_t *ResolverMem, |
| 540 | JITReentryFn ReentryFn, |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 541 | 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 Atanasyan | da29981 | 2018-12-23 12:04:55 +0000 | [diff] [blame] | 607 | 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 Atanasyan | 789f415 | 2019-01-12 11:12:08 +0000 | [diff] [blame] | 612 | 0x03200008, // 0xf4: jr $t9 |
| 613 | 0x00000000, // 0xf8: move $t9, $v0/v1 |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 614 | }; |
| 615 | |
| Simon Atanasyan | 15b68b8 | 2018-12-23 12:05:00 +0000 | [diff] [blame] | 616 | const unsigned ReentryFnAddrOffset = 0x7c; // JIT re-entry fn addr lui |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 617 | const unsigned CallbackMgrAddrOffset = 0x6c; // Callback manager addr lui |
| Simon Atanasyan | 789f415 | 2019-01-12 11:12:08 +0000 | [diff] [blame] | 618 | const unsigned Offsett = 0xf8; |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 619 | |
| 620 | memcpy(ResolverMem, ResolverCode, sizeof(ResolverCode)); |
| 621 | |
| Simon Atanasyan | f903f78 | 2019-01-12 11:12:04 +0000 | [diff] [blame] | 622 | // 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 Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 625 | |
| 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 Atanasyan | 15b68b8 | 2018-12-23 12:05:00 +0000 | [diff] [blame] | 629 | memcpy(ResolverMem + CallbackMgrAddrOffset, &CallMgrLUi, sizeof(CallMgrLUi)); |
| 630 | memcpy(ResolverMem + CallbackMgrAddrOffset + 4, &CallMgrADDiu, |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 631 | 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 Atanasyan | 15b68b8 | 2018-12-23 12:05:00 +0000 | [diff] [blame] | 636 | memcpy(ResolverMem + ReentryFnAddrOffset, &ReentryLUi, sizeof(ReentryLUi)); |
| 637 | memcpy(ResolverMem + ReentryFnAddrOffset + 4, &ReentryADDiu, |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 638 | sizeof(ReentryADDiu)); |
| 639 | } |
| 640 | |
| Simon Atanasyan | 15b68b8 | 2018-12-23 12:05:00 +0000 | [diff] [blame] | 641 | void OrcMips32_Base::writeTrampolines(uint8_t *TrampolineMem, |
| 642 | void *ResolverAddr, |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 643 | 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 Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 656 | } |
| 657 | |
| 658 | Error 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 Hames | e4b4ab6 | 2019-05-08 02:11:07 +0000 | [diff] [blame] | 686 | static const unsigned PageSize = sys::Process::getPageSizeEstimate(); |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 687 | 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 | |
| 732 | void 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 Atanasyan | 4553cda | 2018-12-23 12:05:04 +0000 | [diff] [blame] | 762 | 0xffbe00c0, // 0x64: sd fp,192(sp) |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 763 | 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 Atanasyan | 4553cda | 2018-12-23 12:05:04 +0000 | [diff] [blame] | 786 | 0xdfbe00c0, // 0xb0: ld fp, 192(sp) |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 787 | 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 Atanasyan | 789f415 | 2019-01-12 11:12:08 +0000 | [diff] [blame] | 812 | 0x03200008, // 0x118: jr $t9 |
| 813 | 0x0040c825, // 0x11c: move $t9, $v0 |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 814 | }; |
| 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 Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 821 | uint64_t CallMgrAddr = reinterpret_cast<uint64_t>(CallbackMgr); |
| 822 | |
| 823 | uint32_t CallMgrLUi = |
| 824 | 0x3c040000 | (((CallMgrAddr + 0x800080008000) >> 48) & 0xFFFF); |
| 825 | uint32_t CallMgrDADDiu = |
| Simon Atanasyan | 15b68b8 | 2018-12-23 12:05:00 +0000 | [diff] [blame] | 826 | 0x64840000 | (((CallMgrAddr + 0x80008000) >> 32) & 0xFFFF); |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 827 | uint32_t CallMgrDSLL = 0x00042438; |
| 828 | uint32_t CallMgrDADDiu2 = |
| 829 | 0x64840000 | ((((CallMgrAddr + 0x8000) >> 16) & 0xFFFF)); |
| 830 | uint32_t CallMgrDSLL2 = 0x00042438; |
| Simon Atanasyan | 15b68b8 | 2018-12-23 12:05:00 +0000 | [diff] [blame] | 831 | uint32_t CallMgrDADDiu3 = 0x64840000 | ((CallMgrAddr)&0xFFFF); |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 832 | |
| Simon Atanasyan | 15b68b8 | 2018-12-23 12:05:00 +0000 | [diff] [blame] | 833 | memcpy(ResolverMem + CallbackMgrAddrOffset, &CallMgrLUi, sizeof(CallMgrLUi)); |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 834 | 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 Atanasyan | 15b68b8 | 2018-12-23 12:05:00 +0000 | [diff] [blame] | 860 | uint32_t ReentryDADDiu3 = 0x67390000 | ((ReentryAddr)&0xFFFF); |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 861 | |
| Simon Atanasyan | 15b68b8 | 2018-12-23 12:05:00 +0000 | [diff] [blame] | 862 | memcpy(ResolverMem + ReentryFnAddrOffset, &ReentryLUi, sizeof(ReentryLUi)); |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 863 | 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 | |
| 875 | void 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 Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 897 | } |
| 898 | |
| 899 | Error 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 Hames | e4b4ab6 | 2019-05-08 02:11:07 +0000 | [diff] [blame] | 932 | static const unsigned PageSize = sys::Process::getPageSizeEstimate(); |
| Petar Jovanovic | 5abf4bb | 2018-09-11 13:10:04 +0000 | [diff] [blame] | 933 | 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 Hames | e738061 | 2015-02-21 20:44:36 +0000 | [diff] [blame] | 982 | } // End namespace orc. |
| 983 | } // End namespace llvm. |