36ba8d1b93e7e0816c743f1c091303f5b8bdf3b4
[oota-llvm.git] / lib / ExecutionEngine / RuntimeDyld / RuntimeDyldMachO.h
1 //===-- RuntimeDyldMachO.h - Run-time dynamic linker for MC-JIT ---*- C++ -*-=//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // MachO support for MC-JIT runtime dynamic linker.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #ifndef LLVM_LIB_EXECUTIONENGINE_RUNTIMEDYLD_RUNTIMEDYLDMACHO_H
15 #define LLVM_LIB_EXECUTIONENGINE_RUNTIMEDYLD_RUNTIMEDYLDMACHO_H
16
17 #include "RuntimeDyldImpl.h"
18 #include "llvm/Object/MachO.h"
19 #include "llvm/Support/Format.h"
20
21 #define DEBUG_TYPE "dyld"
22
23 using namespace llvm;
24 using namespace llvm::object;
25
26 namespace llvm {
27 class RuntimeDyldMachO : public RuntimeDyldImpl {
28 protected:
29   struct SectionOffsetPair {
30     unsigned SectionID;
31     uint64_t Offset;
32   };
33
34   struct EHFrameRelatedSections {
35     EHFrameRelatedSections()
36         : EHFrameSID(RTDYLD_INVALID_SECTION_ID),
37           TextSID(RTDYLD_INVALID_SECTION_ID),
38           ExceptTabSID(RTDYLD_INVALID_SECTION_ID) {}
39
40     EHFrameRelatedSections(SID EH, SID T, SID Ex)
41         : EHFrameSID(EH), TextSID(T), ExceptTabSID(Ex) {}
42     SID EHFrameSID;
43     SID TextSID;
44     SID ExceptTabSID;
45   };
46
47   // When a module is loaded we save the SectionID of the EH frame section
48   // in a table until we receive a request to register all unregistered
49   // EH frame sections with the memory manager.
50   SmallVector<EHFrameRelatedSections, 2> UnregisteredEHFrameSections;
51
52   RuntimeDyldMachO(RuntimeDyld::MemoryManager &MemMgr,
53                    RuntimeDyld::SymbolResolver &Resolver)
54       : RuntimeDyldImpl(MemMgr, Resolver) {}
55
56   /// This convenience method uses memcpy to extract a contiguous addend (the
57   /// addend size and offset are taken from the corresponding fields of the RE).
58   int64_t memcpyAddend(const RelocationEntry &RE) const;
59
60   /// Given a relocation_iterator for a non-scattered relocation, construct a
61   /// RelocationEntry and fill in the common fields. The 'Addend' field is *not*
62   /// filled in, since immediate encodings are highly target/opcode specific.
63   /// For targets/opcodes with simple, contiguous immediates (e.g. X86) the
64   /// memcpyAddend method can be used to read the immediate.
65   RelocationEntry getRelocationEntry(unsigned SectionID,
66                                      const ObjectFile &BaseTObj,
67                                      const relocation_iterator &RI) const {
68     const MachOObjectFile &Obj =
69       static_cast<const MachOObjectFile &>(BaseTObj);
70     MachO::any_relocation_info RelInfo =
71       Obj.getRelocation(RI->getRawDataRefImpl());
72
73     bool IsPCRel = Obj.getAnyRelocationPCRel(RelInfo);
74     unsigned Size = Obj.getAnyRelocationLength(RelInfo);
75     uint64_t Offset = RI->getOffset();
76     MachO::RelocationInfoType RelType =
77       static_cast<MachO::RelocationInfoType>(Obj.getAnyRelocationType(RelInfo));
78
79     return RelocationEntry(SectionID, Offset, RelType, 0, IsPCRel, Size);
80   }
81
82   /// Construct a RelocationValueRef representing the relocation target.
83   /// For Symbols in known sections, this will return a RelocationValueRef
84   /// representing a (SectionID, Offset) pair.
85   /// For Symbols whose section is not known, this will return a
86   /// (SymbolName, Offset) pair, where the Offset is taken from the instruction
87   /// immediate (held in RE.Addend).
88   /// In both cases the Addend field is *NOT* fixed up to be PC-relative. That
89   /// should be done by the caller where appropriate by calling makePCRel on
90   /// the RelocationValueRef.
91   RelocationValueRef getRelocationValueRef(const ObjectFile &BaseTObj,
92                                            const relocation_iterator &RI,
93                                            const RelocationEntry &RE,
94                                            ObjSectionToIDMap &ObjSectionToID);
95
96   /// Make the RelocationValueRef addend PC-relative.
97   void makeValueAddendPCRel(RelocationValueRef &Value,
98                             const ObjectFile &BaseTObj,
99                             const relocation_iterator &RI,
100                             unsigned OffsetToNextPC);
101
102   /// Dump information about the relocation entry (RE) and resolved value.
103   void dumpRelocationToResolve(const RelocationEntry &RE, uint64_t Value) const;
104
105   // Return a section iterator for the section containing the given address.
106   static section_iterator getSectionByAddress(const MachOObjectFile &Obj,
107                                               uint64_t Addr);
108
109
110   // Populate __pointers section.
111   void populateIndirectSymbolPointersSection(const MachOObjectFile &Obj,
112                                              const SectionRef &PTSection,
113                                              unsigned PTSectionID);
114
115 public:
116
117   /// Create a RuntimeDyldMachO instance for the given target architecture.
118   static std::unique_ptr<RuntimeDyldMachO>
119   create(Triple::ArchType Arch,
120          RuntimeDyld::MemoryManager &MemMgr,
121          RuntimeDyld::SymbolResolver &Resolver);
122
123   std::unique_ptr<RuntimeDyld::LoadedObjectInfo>
124   loadObject(const object::ObjectFile &O) override;
125
126   SectionEntry &getSection(unsigned SectionID) { return Sections[SectionID]; }
127
128   bool isCompatibleFile(const object::ObjectFile &Obj) const override;
129 };
130
131 /// RuntimeDyldMachOTarget - Templated base class for generic MachO linker
132 /// algorithms and data structures.
133 ///
134 /// Concrete, target specific sub-classes can be accessed via the impl()
135 /// methods. (i.e. the RuntimeDyldMachO hierarchy uses the Curiously
136 /// Recurring Template Idiom). Concrete subclasses for each target
137 /// can be found in ./Targets.
138 template <typename Impl>
139 class RuntimeDyldMachOCRTPBase : public RuntimeDyldMachO {
140 private:
141   Impl &impl() { return static_cast<Impl &>(*this); }
142   const Impl &impl() const { return static_cast<const Impl &>(*this); }
143
144   unsigned char *processFDE(unsigned char *P, int64_t DeltaForText,
145                             int64_t DeltaForEH);
146
147 public:
148   RuntimeDyldMachOCRTPBase(RuntimeDyld::MemoryManager &MemMgr,
149                            RuntimeDyld::SymbolResolver &Resolver)
150     : RuntimeDyldMachO(MemMgr, Resolver) {}
151
152   void finalizeLoad(const ObjectFile &Obj,
153                     ObjSectionToIDMap &SectionMap) override;
154   void registerEHFrames() override;
155 };
156
157 } // end namespace llvm
158
159 #undef DEBUG_TYPE
160
161 #endif