Evan Cheng | 148b6a4 | 2007-07-05 21:15:40 +0000 | [diff] [blame] | 1 | //===- ARMJITInfo.h - ARM implementation of the JIT interface --*- C++ -*-===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | 4ee451d | 2007-12-29 20:36:04 +0000 | [diff] [blame] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Evan Cheng | 148b6a4 | 2007-07-05 21:15:40 +0000 | [diff] [blame] | 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This file contains the declaration of the ARMJITInfo class. |
| 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
| 14 | #ifndef ARMJITINFO_H |
| 15 | #define ARMJITINFO_H |
| 16 | |
Evan Cheng | f1bbb95 | 2008-11-08 00:51:41 +0000 | [diff] [blame] | 17 | #include "ARMMachineFunctionInfo.h" |
Evan Cheng | 25e0478 | 2008-11-04 00:50:32 +0000 | [diff] [blame] | 18 | #include "llvm/CodeGen/MachineConstantPool.h" |
Evan Cheng | 4df60f5 | 2008-11-07 09:06:08 +0000 | [diff] [blame] | 19 | #include "llvm/CodeGen/MachineFunction.h" |
| 20 | #include "llvm/CodeGen/MachineJumpTableInfo.h" |
Evan Cheng | f1bbb95 | 2008-11-08 00:51:41 +0000 | [diff] [blame] | 21 | #include "llvm/Target/TargetJITInfo.h" |
Evan Cheng | 25e0478 | 2008-11-04 00:50:32 +0000 | [diff] [blame] | 22 | #include "llvm/ADT/DenseMap.h" |
Evan Cheng | 938b9d8 | 2008-10-31 19:55:13 +0000 | [diff] [blame] | 23 | #include "llvm/ADT/SmallVector.h" |
Evan Cheng | 148b6a4 | 2007-07-05 21:15:40 +0000 | [diff] [blame] | 24 | |
| 25 | namespace llvm { |
| 26 | class ARMTargetMachine; |
| 27 | |
| 28 | class ARMJITInfo : public TargetJITInfo { |
Evan Cheng | 0f28243 | 2008-10-29 23:55:43 +0000 | [diff] [blame] | 29 | // ConstPoolId2AddrMap - A map from constant pool ids to the corresponding |
| 30 | // CONSTPOOL_ENTRY addresses. |
Evan Cheng | 4df60f5 | 2008-11-07 09:06:08 +0000 | [diff] [blame] | 31 | SmallVector<intptr_t, 16> ConstPoolId2AddrMap; |
| 32 | |
| 33 | // JumpTableId2AddrMap - A map from inline jumptable ids to the |
| 34 | // corresponding inline jump table bases. |
| 35 | SmallVector<intptr_t, 16> JumpTableId2AddrMap; |
Evan Cheng | 0f28243 | 2008-10-29 23:55:43 +0000 | [diff] [blame] | 36 | |
Evan Cheng | 25e0478 | 2008-11-04 00:50:32 +0000 | [diff] [blame] | 37 | // PCLabelMap - A map from PC labels to addresses. |
| 38 | DenseMap<unsigned, intptr_t> PCLabelMap; |
| 39 | |
Evan Cheng | 3cc8223 | 2008-11-08 07:38:22 +0000 | [diff] [blame^] | 40 | // IsPIC - True if the relocation model is PIC. This is used to determine |
| 41 | // how to codegen function stubs. |
| 42 | bool IsPIC; |
| 43 | |
Evan Cheng | 148b6a4 | 2007-07-05 21:15:40 +0000 | [diff] [blame] | 44 | public: |
Evan Cheng | 3cc8223 | 2008-11-08 07:38:22 +0000 | [diff] [blame^] | 45 | explicit ARMJITInfo() : IsPIC(false) { useGOT = false; } |
Evan Cheng | 148b6a4 | 2007-07-05 21:15:40 +0000 | [diff] [blame] | 46 | |
| 47 | /// replaceMachineCodeForFunction - Make it so that calling the function |
| 48 | /// whose machine code is at OLD turns into a call to NEW, perhaps by |
| 49 | /// overwriting OLD with a branch to NEW. This is used for self-modifying |
| 50 | /// code. |
| 51 | /// |
| 52 | virtual void replaceMachineCodeForFunction(void *Old, void *New); |
| 53 | |
Evan Cheng | e96a490 | 2008-11-08 01:31:27 +0000 | [diff] [blame] | 54 | /// emitGlobalValueNonLazyPtr - Use the specified MachineCodeEmitter object |
| 55 | /// to emit a Mac OS X non-lazy pointer which contains the address of the |
| 56 | /// specified ptr. |
| 57 | virtual void *emitGlobalValueNonLazyPtr(const GlobalValue *GV, void *Ptr, |
| 58 | MachineCodeEmitter &MCE); |
| 59 | |
Evan Cheng | 148b6a4 | 2007-07-05 21:15:40 +0000 | [diff] [blame] | 60 | /// emitFunctionStub - Use the specified MachineCodeEmitter object to emit a |
| 61 | /// small native function that simply calls the function at the specified |
| 62 | /// address. |
Nicolas Geoffray | 51cc3c1 | 2008-04-16 20:46:05 +0000 | [diff] [blame] | 63 | virtual void *emitFunctionStub(const Function* F, void *Fn, |
| 64 | MachineCodeEmitter &MCE); |
Evan Cheng | 148b6a4 | 2007-07-05 21:15:40 +0000 | [diff] [blame] | 65 | |
| 66 | /// getLazyResolverFunction - Expose the lazy resolver to the JIT. |
| 67 | virtual LazyResolverFn getLazyResolverFunction(JITCompilerFn); |
| 68 | |
| 69 | /// relocate - Before the JIT can run a block of code that has been emitted, |
| 70 | /// it must rewrite the code to contain the actual addresses of any |
| 71 | /// referenced global symbols. |
| 72 | virtual void relocate(void *Function, MachineRelocation *MR, |
| 73 | unsigned NumRelocs, unsigned char* GOTBase); |
Evan Cheng | 25e0478 | 2008-11-04 00:50:32 +0000 | [diff] [blame] | 74 | |
Jim Grosbach | bc6d876 | 2008-10-28 18:25:49 +0000 | [diff] [blame] | 75 | /// hasCustomConstantPool - Allows a target to specify that constant |
| 76 | /// pool address resolution is handled by the target. |
| 77 | virtual bool hasCustomConstantPool() const { return true; } |
| 78 | |
Evan Cheng | 4df60f5 | 2008-11-07 09:06:08 +0000 | [diff] [blame] | 79 | /// hasCustomJumpTables - Allows a target to specify that jumptables |
| 80 | /// are emitted by the target. |
| 81 | virtual bool hasCustomJumpTables() const { return true; } |
| 82 | |
Evan Cheng | b0b5349 | 2008-11-04 09:30:48 +0000 | [diff] [blame] | 83 | /// allocateSeparateGVMemory - If true, globals should be placed in |
| 84 | /// separately allocated heap memory rather than in the same |
| 85 | /// code memory allocated by MachineCodeEmitter. |
| 86 | virtual bool allocateSeparateGVMemory() const { |
| 87 | #ifdef __APPLE__ |
| 88 | return true; |
| 89 | #else |
| 90 | return false; |
| 91 | #endif |
| 92 | } |
| 93 | |
Evan Cheng | 3cc8223 | 2008-11-08 07:38:22 +0000 | [diff] [blame^] | 94 | /// Initialize - Initialize internal stage for the function being JITted. |
| 95 | /// Resize constant pool ids to CONSTPOOL_ENTRY addresses map; resize |
| 96 | /// jump table ids to jump table bases map; remember if codegen relocation |
| 97 | /// model is PIC. |
| 98 | void Initialize(const MachineFunction &MF, bool isPIC) { |
Evan Cheng | f1bbb95 | 2008-11-08 00:51:41 +0000 | [diff] [blame] | 99 | const ARMFunctionInfo *AFI = MF.getInfo<ARMFunctionInfo>(); |
| 100 | ConstPoolId2AddrMap.resize(AFI->getNumConstPoolEntries()); |
| 101 | JumpTableId2AddrMap.resize(AFI->getNumJumpTables()); |
Evan Cheng | 3cc8223 | 2008-11-08 07:38:22 +0000 | [diff] [blame^] | 102 | IsPIC = isPIC; |
Evan Cheng | 938b9d8 | 2008-10-31 19:55:13 +0000 | [diff] [blame] | 103 | } |
| 104 | |
Evan Cheng | 0f28243 | 2008-10-29 23:55:43 +0000 | [diff] [blame] | 105 | /// getConstantPoolEntryAddr - The ARM target puts all constant |
Evan Cheng | 4df60f5 | 2008-11-07 09:06:08 +0000 | [diff] [blame] | 106 | /// pool entries into constant islands. This returns the address of the |
| 107 | /// constant pool entry of the specified index. |
Evan Cheng | 938b9d8 | 2008-10-31 19:55:13 +0000 | [diff] [blame] | 108 | intptr_t getConstantPoolEntryAddr(unsigned CPI) const { |
| 109 | assert(CPI < ConstPoolId2AddrMap.size()); |
| 110 | return ConstPoolId2AddrMap[CPI]; |
Evan Cheng | 0f28243 | 2008-10-29 23:55:43 +0000 | [diff] [blame] | 111 | } |
Jim Grosbach | bc6d876 | 2008-10-28 18:25:49 +0000 | [diff] [blame] | 112 | |
Evan Cheng | 4df60f5 | 2008-11-07 09:06:08 +0000 | [diff] [blame] | 113 | /// addConstantPoolEntryAddr - Map a Constant Pool Index to the address |
Jim Grosbach | bc6d876 | 2008-10-28 18:25:49 +0000 | [diff] [blame] | 114 | /// where its associated value is stored. When relocations are processed, |
| 115 | /// this value will be used to resolve references to the constant. |
Evan Cheng | 938b9d8 | 2008-10-31 19:55:13 +0000 | [diff] [blame] | 116 | void addConstantPoolEntryAddr(unsigned CPI, intptr_t Addr) { |
| 117 | assert(CPI < ConstPoolId2AddrMap.size()); |
| 118 | ConstPoolId2AddrMap[CPI] = Addr; |
Evan Cheng | 0f28243 | 2008-10-29 23:55:43 +0000 | [diff] [blame] | 119 | } |
Evan Cheng | 25e0478 | 2008-11-04 00:50:32 +0000 | [diff] [blame] | 120 | |
Evan Cheng | 4df60f5 | 2008-11-07 09:06:08 +0000 | [diff] [blame] | 121 | /// getJumpTableBaseAddr - The ARM target inline all jump tables within |
| 122 | /// text section of the function. This returns the address of the base of |
| 123 | /// the jump table of the specified index. |
| 124 | intptr_t getJumpTableBaseAddr(unsigned JTI) const { |
| 125 | assert(JTI < JumpTableId2AddrMap.size()); |
| 126 | return JumpTableId2AddrMap[JTI]; |
| 127 | } |
| 128 | |
| 129 | /// addJumpTableBaseAddr - Map a jump table index to the address where |
| 130 | /// the corresponding inline jump table is emitted. When relocations are |
| 131 | /// processed, this value will be used to resolve references to the |
| 132 | /// jump table. |
| 133 | void addJumpTableBaseAddr(unsigned JTI, intptr_t Addr) { |
| 134 | assert(JTI < JumpTableId2AddrMap.size()); |
| 135 | JumpTableId2AddrMap[JTI] = Addr; |
| 136 | } |
| 137 | |
Evan Cheng | 25e0478 | 2008-11-04 00:50:32 +0000 | [diff] [blame] | 138 | /// getPCLabelAddr - Retrieve the address of the PC label of the specified id. |
| 139 | intptr_t getPCLabelAddr(unsigned Id) const { |
| 140 | DenseMap<unsigned, intptr_t>::const_iterator I = PCLabelMap.find(Id); |
| 141 | assert(I != PCLabelMap.end()); |
| 142 | return I->second; |
| 143 | } |
| 144 | |
| 145 | /// addPCLabelAddr - Remember the address of the specified PC label. |
| 146 | void addPCLabelAddr(unsigned Id, intptr_t Addr) { |
| 147 | PCLabelMap.insert(std::make_pair(Id, Addr)); |
| 148 | } |
| 149 | |
| 150 | private: |
Evan Cheng | 4df60f5 | 2008-11-07 09:06:08 +0000 | [diff] [blame] | 151 | /// resolveRelocDestAddr - Resolve the resulting address of the relocation |
Evan Cheng | 25e0478 | 2008-11-04 00:50:32 +0000 | [diff] [blame] | 152 | /// if it's not already solved. Constantpool entries must be resolved by |
| 153 | /// ARM target. |
Evan Cheng | 4df60f5 | 2008-11-07 09:06:08 +0000 | [diff] [blame] | 154 | intptr_t resolveRelocDestAddr(MachineRelocation *MR) const; |
Evan Cheng | 148b6a4 | 2007-07-05 21:15:40 +0000 | [diff] [blame] | 155 | }; |
| 156 | } |
| 157 | |
| 158 | #endif |