Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===-- RTDyldMemoryManager.cpp - Memory manager for MC-JIT -----*- C++ -*-===// |
2 | // |
||
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 |
||
6 | // |
||
7 | //===----------------------------------------------------------------------===// |
||
8 | // |
||
9 | // Interface of the runtime dynamic memory manager base class. |
||
10 | // |
||
11 | //===----------------------------------------------------------------------===// |
||
12 | |||
13 | #ifndef LLVM_EXECUTIONENGINE_RTDYLDMEMORYMANAGER_H |
||
14 | #define LLVM_EXECUTIONENGINE_RTDYLDMEMORYMANAGER_H |
||
15 | |||
16 | #include "llvm-c/ExecutionEngine.h" |
||
17 | #include "llvm/ExecutionEngine/JITSymbol.h" |
||
18 | #include "llvm/ExecutionEngine/RuntimeDyld.h" |
||
19 | #include "llvm/Support/CBindingWrapping.h" |
||
20 | #include <cstddef> |
||
21 | #include <cstdint> |
||
22 | #include <string> |
||
23 | |||
24 | namespace llvm { |
||
25 | |||
26 | class ExecutionEngine; |
||
27 | |||
28 | namespace object { |
||
29 | class ObjectFile; |
||
30 | } // end namespace object |
||
31 | |||
32 | class MCJITMemoryManager : public RuntimeDyld::MemoryManager { |
||
33 | public: |
||
34 | // Don't hide the notifyObjectLoaded method from RuntimeDyld::MemoryManager. |
||
35 | using RuntimeDyld::MemoryManager::notifyObjectLoaded; |
||
36 | |||
37 | /// This method is called after an object has been loaded into memory but |
||
38 | /// before relocations are applied to the loaded sections. The object load |
||
39 | /// may have been initiated by MCJIT to resolve an external symbol for another |
||
40 | /// object that is being finalized. In that case, the object about which |
||
41 | /// the memory manager is being notified will be finalized immediately after |
||
42 | /// the memory manager returns from this call. |
||
43 | /// |
||
44 | /// Memory managers which are preparing code for execution in an external |
||
45 | /// address space can use this call to remap the section addresses for the |
||
46 | /// newly loaded object. |
||
47 | virtual void notifyObjectLoaded(ExecutionEngine *EE, |
||
48 | const object::ObjectFile &) {} |
||
49 | |||
50 | private: |
||
51 | void anchor() override; |
||
52 | }; |
||
53 | |||
54 | // RuntimeDyld clients often want to handle the memory management of |
||
55 | // what gets placed where. For JIT clients, this is the subset of |
||
56 | // JITMemoryManager required for dynamic loading of binaries. |
||
57 | // |
||
58 | // FIXME: As the RuntimeDyld fills out, additional routines will be needed |
||
59 | // for the varying types of objects to be allocated. |
||
60 | class RTDyldMemoryManager : public MCJITMemoryManager, |
||
61 | public LegacyJITSymbolResolver { |
||
62 | public: |
||
63 | RTDyldMemoryManager() = default; |
||
64 | RTDyldMemoryManager(const RTDyldMemoryManager&) = delete; |
||
65 | void operator=(const RTDyldMemoryManager&) = delete; |
||
66 | ~RTDyldMemoryManager() override; |
||
67 | |||
68 | /// Register EH frames in the current process. |
||
69 | static void registerEHFramesInProcess(uint8_t *Addr, size_t Size); |
||
70 | |||
71 | /// Deregister EH frames in the current proces. |
||
72 | static void deregisterEHFramesInProcess(uint8_t *Addr, size_t Size); |
||
73 | |||
74 | void registerEHFrames(uint8_t *Addr, uint64_t LoadAddr, size_t Size) override; |
||
75 | void deregisterEHFrames() override; |
||
76 | |||
77 | /// This method returns the address of the specified function or variable in |
||
78 | /// the current process. |
||
79 | static uint64_t getSymbolAddressInProcess(const std::string &Name); |
||
80 | |||
81 | /// Legacy symbol lookup - DEPRECATED! Please override findSymbol instead. |
||
82 | /// |
||
83 | /// This method returns the address of the specified function or variable. |
||
84 | /// It is used to resolve symbols during module linking. |
||
85 | virtual uint64_t getSymbolAddress(const std::string &Name) { |
||
86 | return getSymbolAddressInProcess(Name); |
||
87 | } |
||
88 | |||
89 | /// This method returns a RuntimeDyld::SymbolInfo for the specified function |
||
90 | /// or variable. It is used to resolve symbols during module linking. |
||
91 | /// |
||
92 | /// By default this falls back on the legacy lookup method: |
||
93 | /// 'getSymbolAddress'. The address returned by getSymbolAddress is treated as |
||
94 | /// a strong, exported symbol, consistent with historical treatment by |
||
95 | /// RuntimeDyld. |
||
96 | /// |
||
97 | /// Clients writing custom RTDyldMemoryManagers are encouraged to override |
||
98 | /// this method and return a SymbolInfo with the flags set correctly. This is |
||
99 | /// necessary for RuntimeDyld to correctly handle weak and non-exported symbols. |
||
100 | JITSymbol findSymbol(const std::string &Name) override { |
||
101 | return JITSymbol(getSymbolAddress(Name), JITSymbolFlags::Exported); |
||
102 | } |
||
103 | |||
104 | /// Legacy symbol lookup -- DEPRECATED! Please override |
||
105 | /// findSymbolInLogicalDylib instead. |
||
106 | /// |
||
107 | /// Default to treating all modules as separate. |
||
108 | virtual uint64_t getSymbolAddressInLogicalDylib(const std::string &Name) { |
||
109 | return 0; |
||
110 | } |
||
111 | |||
112 | /// Default to treating all modules as separate. |
||
113 | /// |
||
114 | /// By default this falls back on the legacy lookup method: |
||
115 | /// 'getSymbolAddressInLogicalDylib'. The address returned by |
||
116 | /// getSymbolAddressInLogicalDylib is treated as a strong, exported symbol, |
||
117 | /// consistent with historical treatment by RuntimeDyld. |
||
118 | /// |
||
119 | /// Clients writing custom RTDyldMemoryManagers are encouraged to override |
||
120 | /// this method and return a SymbolInfo with the flags set correctly. This is |
||
121 | /// necessary for RuntimeDyld to correctly handle weak and non-exported symbols. |
||
122 | JITSymbol |
||
123 | findSymbolInLogicalDylib(const std::string &Name) override { |
||
124 | return JITSymbol(getSymbolAddressInLogicalDylib(Name), |
||
125 | JITSymbolFlags::Exported); |
||
126 | } |
||
127 | |||
128 | /// This method returns the address of the specified function. As such it is |
||
129 | /// only useful for resolving library symbols, not code generated symbols. |
||
130 | /// |
||
131 | /// If \p AbortOnFailure is false and no function with the given name is |
||
132 | /// found, this function returns a null pointer. Otherwise, it prints a |
||
133 | /// message to stderr and aborts. |
||
134 | /// |
||
135 | /// This function is deprecated for memory managers to be used with |
||
136 | /// MCJIT or RuntimeDyld. Use getSymbolAddress instead. |
||
137 | virtual void *getPointerToNamedFunction(const std::string &Name, |
||
138 | bool AbortOnFailure = true); |
||
139 | |||
140 | protected: |
||
141 | struct EHFrame { |
||
142 | uint8_t *Addr; |
||
143 | size_t Size; |
||
144 | }; |
||
145 | typedef std::vector<EHFrame> EHFrameInfos; |
||
146 | EHFrameInfos EHFrames; |
||
147 | |||
148 | private: |
||
149 | void anchor() override; |
||
150 | }; |
||
151 | |||
152 | // Create wrappers for C Binding types (see CBindingWrapping.h). |
||
153 | DEFINE_SIMPLE_CONVERSION_FUNCTIONS( |
||
154 | RTDyldMemoryManager, LLVMMCJITMemoryManagerRef) |
||
155 | |||
156 | } // end namespace llvm |
||
157 | |||
158 | #endif // LLVM_EXECUTIONENGINE_RTDYLDMEMORYMANAGER_H |