annotate llvm/lib/ExecutionEngine/ExecutionEngine.cpp @ 150:1d019706d866

LLVM10
author anatofuz
date Thu, 13 Feb 2020 15:10:13 +0900
parents
children 0572611fdcc8
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
150
anatofuz
parents:
diff changeset
1 //===-- ExecutionEngine.cpp - Common Implementation shared by EEs ---------===//
anatofuz
parents:
diff changeset
2 //
anatofuz
parents:
diff changeset
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
anatofuz
parents:
diff changeset
4 // See https://llvm.org/LICENSE.txt for license information.
anatofuz
parents:
diff changeset
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
anatofuz
parents:
diff changeset
6 //
anatofuz
parents:
diff changeset
7 //===----------------------------------------------------------------------===//
anatofuz
parents:
diff changeset
8 //
anatofuz
parents:
diff changeset
9 // This file defines the common interface used by the various execution engine
anatofuz
parents:
diff changeset
10 // subclasses.
anatofuz
parents:
diff changeset
11 //
anatofuz
parents:
diff changeset
12 //===----------------------------------------------------------------------===//
anatofuz
parents:
diff changeset
13
anatofuz
parents:
diff changeset
14 #include "llvm/ExecutionEngine/ExecutionEngine.h"
anatofuz
parents:
diff changeset
15 #include "llvm/ADT/STLExtras.h"
anatofuz
parents:
diff changeset
16 #include "llvm/ADT/SmallString.h"
anatofuz
parents:
diff changeset
17 #include "llvm/ADT/Statistic.h"
anatofuz
parents:
diff changeset
18 #include "llvm/ExecutionEngine/GenericValue.h"
anatofuz
parents:
diff changeset
19 #include "llvm/ExecutionEngine/JITEventListener.h"
anatofuz
parents:
diff changeset
20 #include "llvm/ExecutionEngine/ObjectCache.h"
anatofuz
parents:
diff changeset
21 #include "llvm/ExecutionEngine/RTDyldMemoryManager.h"
anatofuz
parents:
diff changeset
22 #include "llvm/IR/Constants.h"
anatofuz
parents:
diff changeset
23 #include "llvm/IR/DataLayout.h"
anatofuz
parents:
diff changeset
24 #include "llvm/IR/DerivedTypes.h"
anatofuz
parents:
diff changeset
25 #include "llvm/IR/Mangler.h"
anatofuz
parents:
diff changeset
26 #include "llvm/IR/Module.h"
anatofuz
parents:
diff changeset
27 #include "llvm/IR/Operator.h"
anatofuz
parents:
diff changeset
28 #include "llvm/IR/ValueHandle.h"
anatofuz
parents:
diff changeset
29 #include "llvm/Object/Archive.h"
anatofuz
parents:
diff changeset
30 #include "llvm/Object/ObjectFile.h"
anatofuz
parents:
diff changeset
31 #include "llvm/Support/Debug.h"
anatofuz
parents:
diff changeset
32 #include "llvm/Support/DynamicLibrary.h"
anatofuz
parents:
diff changeset
33 #include "llvm/Support/ErrorHandling.h"
anatofuz
parents:
diff changeset
34 #include "llvm/Support/Host.h"
anatofuz
parents:
diff changeset
35 #include "llvm/Support/TargetRegistry.h"
anatofuz
parents:
diff changeset
36 #include "llvm/Support/raw_ostream.h"
anatofuz
parents:
diff changeset
37 #include "llvm/Target/TargetMachine.h"
anatofuz
parents:
diff changeset
38 #include <cmath>
anatofuz
parents:
diff changeset
39 #include <cstring>
anatofuz
parents:
diff changeset
40 #include <mutex>
anatofuz
parents:
diff changeset
41 using namespace llvm;
anatofuz
parents:
diff changeset
42
anatofuz
parents:
diff changeset
43 #define DEBUG_TYPE "jit"
anatofuz
parents:
diff changeset
44
anatofuz
parents:
diff changeset
45 STATISTIC(NumInitBytes, "Number of bytes of global vars initialized");
anatofuz
parents:
diff changeset
46 STATISTIC(NumGlobals , "Number of global vars initialized");
anatofuz
parents:
diff changeset
47
anatofuz
parents:
diff changeset
48 ExecutionEngine *(*ExecutionEngine::MCJITCtor)(
anatofuz
parents:
diff changeset
49 std::unique_ptr<Module> M, std::string *ErrorStr,
anatofuz
parents:
diff changeset
50 std::shared_ptr<MCJITMemoryManager> MemMgr,
anatofuz
parents:
diff changeset
51 std::shared_ptr<LegacyJITSymbolResolver> Resolver,
anatofuz
parents:
diff changeset
52 std::unique_ptr<TargetMachine> TM) = nullptr;
anatofuz
parents:
diff changeset
53
anatofuz
parents:
diff changeset
54 ExecutionEngine *(*ExecutionEngine::OrcMCJITReplacementCtor)(
anatofuz
parents:
diff changeset
55 std::string *ErrorStr, std::shared_ptr<MCJITMemoryManager> MemMgr,
anatofuz
parents:
diff changeset
56 std::shared_ptr<LegacyJITSymbolResolver> Resolver,
anatofuz
parents:
diff changeset
57 std::unique_ptr<TargetMachine> TM) = nullptr;
anatofuz
parents:
diff changeset
58
anatofuz
parents:
diff changeset
59 ExecutionEngine *(*ExecutionEngine::InterpCtor)(std::unique_ptr<Module> M,
anatofuz
parents:
diff changeset
60 std::string *ErrorStr) =nullptr;
anatofuz
parents:
diff changeset
61
anatofuz
parents:
diff changeset
62 void JITEventListener::anchor() {}
anatofuz
parents:
diff changeset
63
anatofuz
parents:
diff changeset
64 void ObjectCache::anchor() {}
anatofuz
parents:
diff changeset
65
anatofuz
parents:
diff changeset
66 void ExecutionEngine::Init(std::unique_ptr<Module> M) {
anatofuz
parents:
diff changeset
67 CompilingLazily = false;
anatofuz
parents:
diff changeset
68 GVCompilationDisabled = false;
anatofuz
parents:
diff changeset
69 SymbolSearchingDisabled = false;
anatofuz
parents:
diff changeset
70
anatofuz
parents:
diff changeset
71 // IR module verification is enabled by default in debug builds, and disabled
anatofuz
parents:
diff changeset
72 // by default in release builds.
anatofuz
parents:
diff changeset
73 #ifndef NDEBUG
anatofuz
parents:
diff changeset
74 VerifyModules = true;
anatofuz
parents:
diff changeset
75 #else
anatofuz
parents:
diff changeset
76 VerifyModules = false;
anatofuz
parents:
diff changeset
77 #endif
anatofuz
parents:
diff changeset
78
anatofuz
parents:
diff changeset
79 assert(M && "Module is null?");
anatofuz
parents:
diff changeset
80 Modules.push_back(std::move(M));
anatofuz
parents:
diff changeset
81 }
anatofuz
parents:
diff changeset
82
anatofuz
parents:
diff changeset
83 ExecutionEngine::ExecutionEngine(std::unique_ptr<Module> M)
anatofuz
parents:
diff changeset
84 : DL(M->getDataLayout()), LazyFunctionCreator(nullptr) {
anatofuz
parents:
diff changeset
85 Init(std::move(M));
anatofuz
parents:
diff changeset
86 }
anatofuz
parents:
diff changeset
87
anatofuz
parents:
diff changeset
88 ExecutionEngine::ExecutionEngine(DataLayout DL, std::unique_ptr<Module> M)
anatofuz
parents:
diff changeset
89 : DL(std::move(DL)), LazyFunctionCreator(nullptr) {
anatofuz
parents:
diff changeset
90 Init(std::move(M));
anatofuz
parents:
diff changeset
91 }
anatofuz
parents:
diff changeset
92
anatofuz
parents:
diff changeset
93 ExecutionEngine::~ExecutionEngine() {
anatofuz
parents:
diff changeset
94 clearAllGlobalMappings();
anatofuz
parents:
diff changeset
95 }
anatofuz
parents:
diff changeset
96
anatofuz
parents:
diff changeset
97 namespace {
anatofuz
parents:
diff changeset
98 /// Helper class which uses a value handler to automatically deletes the
anatofuz
parents:
diff changeset
99 /// memory block when the GlobalVariable is destroyed.
anatofuz
parents:
diff changeset
100 class GVMemoryBlock final : public CallbackVH {
anatofuz
parents:
diff changeset
101 GVMemoryBlock(const GlobalVariable *GV)
anatofuz
parents:
diff changeset
102 : CallbackVH(const_cast<GlobalVariable*>(GV)) {}
anatofuz
parents:
diff changeset
103
anatofuz
parents:
diff changeset
104 public:
anatofuz
parents:
diff changeset
105 /// Returns the address the GlobalVariable should be written into. The
anatofuz
parents:
diff changeset
106 /// GVMemoryBlock object prefixes that.
anatofuz
parents:
diff changeset
107 static char *Create(const GlobalVariable *GV, const DataLayout& TD) {
anatofuz
parents:
diff changeset
108 Type *ElTy = GV->getValueType();
anatofuz
parents:
diff changeset
109 size_t GVSize = (size_t)TD.getTypeAllocSize(ElTy);
anatofuz
parents:
diff changeset
110 void *RawMemory = ::operator new(
anatofuz
parents:
diff changeset
111 alignTo(sizeof(GVMemoryBlock), TD.getPreferredAlignment(GV)) + GVSize);
anatofuz
parents:
diff changeset
112 new(RawMemory) GVMemoryBlock(GV);
anatofuz
parents:
diff changeset
113 return static_cast<char*>(RawMemory) + sizeof(GVMemoryBlock);
anatofuz
parents:
diff changeset
114 }
anatofuz
parents:
diff changeset
115
anatofuz
parents:
diff changeset
116 void deleted() override {
anatofuz
parents:
diff changeset
117 // We allocated with operator new and with some extra memory hanging off the
anatofuz
parents:
diff changeset
118 // end, so don't just delete this. I'm not sure if this is actually
anatofuz
parents:
diff changeset
119 // required.
anatofuz
parents:
diff changeset
120 this->~GVMemoryBlock();
anatofuz
parents:
diff changeset
121 ::operator delete(this);
anatofuz
parents:
diff changeset
122 }
anatofuz
parents:
diff changeset
123 };
anatofuz
parents:
diff changeset
124 } // anonymous namespace
anatofuz
parents:
diff changeset
125
anatofuz
parents:
diff changeset
126 char *ExecutionEngine::getMemoryForGV(const GlobalVariable *GV) {
anatofuz
parents:
diff changeset
127 return GVMemoryBlock::Create(GV, getDataLayout());
anatofuz
parents:
diff changeset
128 }
anatofuz
parents:
diff changeset
129
anatofuz
parents:
diff changeset
130 void ExecutionEngine::addObjectFile(std::unique_ptr<object::ObjectFile> O) {
anatofuz
parents:
diff changeset
131 llvm_unreachable("ExecutionEngine subclass doesn't implement addObjectFile.");
anatofuz
parents:
diff changeset
132 }
anatofuz
parents:
diff changeset
133
anatofuz
parents:
diff changeset
134 void
anatofuz
parents:
diff changeset
135 ExecutionEngine::addObjectFile(object::OwningBinary<object::ObjectFile> O) {
anatofuz
parents:
diff changeset
136 llvm_unreachable("ExecutionEngine subclass doesn't implement addObjectFile.");
anatofuz
parents:
diff changeset
137 }
anatofuz
parents:
diff changeset
138
anatofuz
parents:
diff changeset
139 void ExecutionEngine::addArchive(object::OwningBinary<object::Archive> A) {
anatofuz
parents:
diff changeset
140 llvm_unreachable("ExecutionEngine subclass doesn't implement addArchive.");
anatofuz
parents:
diff changeset
141 }
anatofuz
parents:
diff changeset
142
anatofuz
parents:
diff changeset
143 bool ExecutionEngine::removeModule(Module *M) {
anatofuz
parents:
diff changeset
144 for (auto I = Modules.begin(), E = Modules.end(); I != E; ++I) {
anatofuz
parents:
diff changeset
145 Module *Found = I->get();
anatofuz
parents:
diff changeset
146 if (Found == M) {
anatofuz
parents:
diff changeset
147 I->release();
anatofuz
parents:
diff changeset
148 Modules.erase(I);
anatofuz
parents:
diff changeset
149 clearGlobalMappingsFromModule(M);
anatofuz
parents:
diff changeset
150 return true;
anatofuz
parents:
diff changeset
151 }
anatofuz
parents:
diff changeset
152 }
anatofuz
parents:
diff changeset
153 return false;
anatofuz
parents:
diff changeset
154 }
anatofuz
parents:
diff changeset
155
anatofuz
parents:
diff changeset
156 Function *ExecutionEngine::FindFunctionNamed(StringRef FnName) {
anatofuz
parents:
diff changeset
157 for (unsigned i = 0, e = Modules.size(); i != e; ++i) {
anatofuz
parents:
diff changeset
158 Function *F = Modules[i]->getFunction(FnName);
anatofuz
parents:
diff changeset
159 if (F && !F->isDeclaration())
anatofuz
parents:
diff changeset
160 return F;
anatofuz
parents:
diff changeset
161 }
anatofuz
parents:
diff changeset
162 return nullptr;
anatofuz
parents:
diff changeset
163 }
anatofuz
parents:
diff changeset
164
anatofuz
parents:
diff changeset
165 GlobalVariable *ExecutionEngine::FindGlobalVariableNamed(StringRef Name, bool AllowInternal) {
anatofuz
parents:
diff changeset
166 for (unsigned i = 0, e = Modules.size(); i != e; ++i) {
anatofuz
parents:
diff changeset
167 GlobalVariable *GV = Modules[i]->getGlobalVariable(Name,AllowInternal);
anatofuz
parents:
diff changeset
168 if (GV && !GV->isDeclaration())
anatofuz
parents:
diff changeset
169 return GV;
anatofuz
parents:
diff changeset
170 }
anatofuz
parents:
diff changeset
171 return nullptr;
anatofuz
parents:
diff changeset
172 }
anatofuz
parents:
diff changeset
173
anatofuz
parents:
diff changeset
174 uint64_t ExecutionEngineState::RemoveMapping(StringRef Name) {
anatofuz
parents:
diff changeset
175 GlobalAddressMapTy::iterator I = GlobalAddressMap.find(Name);
anatofuz
parents:
diff changeset
176 uint64_t OldVal;
anatofuz
parents:
diff changeset
177
anatofuz
parents:
diff changeset
178 // FIXME: This is silly, we shouldn't end up with a mapping -> 0 in the
anatofuz
parents:
diff changeset
179 // GlobalAddressMap.
anatofuz
parents:
diff changeset
180 if (I == GlobalAddressMap.end())
anatofuz
parents:
diff changeset
181 OldVal = 0;
anatofuz
parents:
diff changeset
182 else {
anatofuz
parents:
diff changeset
183 GlobalAddressReverseMap.erase(I->second);
anatofuz
parents:
diff changeset
184 OldVal = I->second;
anatofuz
parents:
diff changeset
185 GlobalAddressMap.erase(I);
anatofuz
parents:
diff changeset
186 }
anatofuz
parents:
diff changeset
187
anatofuz
parents:
diff changeset
188 return OldVal;
anatofuz
parents:
diff changeset
189 }
anatofuz
parents:
diff changeset
190
anatofuz
parents:
diff changeset
191 std::string ExecutionEngine::getMangledName(const GlobalValue *GV) {
anatofuz
parents:
diff changeset
192 assert(GV->hasName() && "Global must have name.");
anatofuz
parents:
diff changeset
193
anatofuz
parents:
diff changeset
194 std::lock_guard<sys::Mutex> locked(lock);
anatofuz
parents:
diff changeset
195 SmallString<128> FullName;
anatofuz
parents:
diff changeset
196
anatofuz
parents:
diff changeset
197 const DataLayout &DL =
anatofuz
parents:
diff changeset
198 GV->getParent()->getDataLayout().isDefault()
anatofuz
parents:
diff changeset
199 ? getDataLayout()
anatofuz
parents:
diff changeset
200 : GV->getParent()->getDataLayout();
anatofuz
parents:
diff changeset
201
anatofuz
parents:
diff changeset
202 Mangler::getNameWithPrefix(FullName, GV->getName(), DL);
anatofuz
parents:
diff changeset
203 return std::string(FullName.str());
anatofuz
parents:
diff changeset
204 }
anatofuz
parents:
diff changeset
205
anatofuz
parents:
diff changeset
206 void ExecutionEngine::addGlobalMapping(const GlobalValue *GV, void *Addr) {
anatofuz
parents:
diff changeset
207 std::lock_guard<sys::Mutex> locked(lock);
anatofuz
parents:
diff changeset
208 addGlobalMapping(getMangledName(GV), (uint64_t) Addr);
anatofuz
parents:
diff changeset
209 }
anatofuz
parents:
diff changeset
210
anatofuz
parents:
diff changeset
211 void ExecutionEngine::addGlobalMapping(StringRef Name, uint64_t Addr) {
anatofuz
parents:
diff changeset
212 std::lock_guard<sys::Mutex> locked(lock);
anatofuz
parents:
diff changeset
213
anatofuz
parents:
diff changeset
214 assert(!Name.empty() && "Empty GlobalMapping symbol name!");
anatofuz
parents:
diff changeset
215
anatofuz
parents:
diff changeset
216 LLVM_DEBUG(dbgs() << "JIT: Map \'" << Name << "\' to [" << Addr << "]\n";);
anatofuz
parents:
diff changeset
217 uint64_t &CurVal = EEState.getGlobalAddressMap()[Name];
anatofuz
parents:
diff changeset
218 assert((!CurVal || !Addr) && "GlobalMapping already established!");
anatofuz
parents:
diff changeset
219 CurVal = Addr;
anatofuz
parents:
diff changeset
220
anatofuz
parents:
diff changeset
221 // If we are using the reverse mapping, add it too.
anatofuz
parents:
diff changeset
222 if (!EEState.getGlobalAddressReverseMap().empty()) {
anatofuz
parents:
diff changeset
223 std::string &V = EEState.getGlobalAddressReverseMap()[CurVal];
anatofuz
parents:
diff changeset
224 assert((!V.empty() || !Name.empty()) &&
anatofuz
parents:
diff changeset
225 "GlobalMapping already established!");
anatofuz
parents:
diff changeset
226 V = std::string(Name);
anatofuz
parents:
diff changeset
227 }
anatofuz
parents:
diff changeset
228 }
anatofuz
parents:
diff changeset
229
anatofuz
parents:
diff changeset
230 void ExecutionEngine::clearAllGlobalMappings() {
anatofuz
parents:
diff changeset
231 std::lock_guard<sys::Mutex> locked(lock);
anatofuz
parents:
diff changeset
232
anatofuz
parents:
diff changeset
233 EEState.getGlobalAddressMap().clear();
anatofuz
parents:
diff changeset
234 EEState.getGlobalAddressReverseMap().clear();
anatofuz
parents:
diff changeset
235 }
anatofuz
parents:
diff changeset
236
anatofuz
parents:
diff changeset
237 void ExecutionEngine::clearGlobalMappingsFromModule(Module *M) {
anatofuz
parents:
diff changeset
238 std::lock_guard<sys::Mutex> locked(lock);
anatofuz
parents:
diff changeset
239
anatofuz
parents:
diff changeset
240 for (GlobalObject &GO : M->global_objects())
anatofuz
parents:
diff changeset
241 EEState.RemoveMapping(getMangledName(&GO));
anatofuz
parents:
diff changeset
242 }
anatofuz
parents:
diff changeset
243
anatofuz
parents:
diff changeset
244 uint64_t ExecutionEngine::updateGlobalMapping(const GlobalValue *GV,
anatofuz
parents:
diff changeset
245 void *Addr) {
anatofuz
parents:
diff changeset
246 std::lock_guard<sys::Mutex> locked(lock);
anatofuz
parents:
diff changeset
247 return updateGlobalMapping(getMangledName(GV), (uint64_t) Addr);
anatofuz
parents:
diff changeset
248 }
anatofuz
parents:
diff changeset
249
anatofuz
parents:
diff changeset
250 uint64_t ExecutionEngine::updateGlobalMapping(StringRef Name, uint64_t Addr) {
anatofuz
parents:
diff changeset
251 std::lock_guard<sys::Mutex> locked(lock);
anatofuz
parents:
diff changeset
252
anatofuz
parents:
diff changeset
253 ExecutionEngineState::GlobalAddressMapTy &Map =
anatofuz
parents:
diff changeset
254 EEState.getGlobalAddressMap();
anatofuz
parents:
diff changeset
255
anatofuz
parents:
diff changeset
256 // Deleting from the mapping?
anatofuz
parents:
diff changeset
257 if (!Addr)
anatofuz
parents:
diff changeset
258 return EEState.RemoveMapping(Name);
anatofuz
parents:
diff changeset
259
anatofuz
parents:
diff changeset
260 uint64_t &CurVal = Map[Name];
anatofuz
parents:
diff changeset
261 uint64_t OldVal = CurVal;
anatofuz
parents:
diff changeset
262
anatofuz
parents:
diff changeset
263 if (CurVal && !EEState.getGlobalAddressReverseMap().empty())
anatofuz
parents:
diff changeset
264 EEState.getGlobalAddressReverseMap().erase(CurVal);
anatofuz
parents:
diff changeset
265 CurVal = Addr;
anatofuz
parents:
diff changeset
266
anatofuz
parents:
diff changeset
267 // If we are using the reverse mapping, add it too.
anatofuz
parents:
diff changeset
268 if (!EEState.getGlobalAddressReverseMap().empty()) {
anatofuz
parents:
diff changeset
269 std::string &V = EEState.getGlobalAddressReverseMap()[CurVal];
anatofuz
parents:
diff changeset
270 assert((!V.empty() || !Name.empty()) &&
anatofuz
parents:
diff changeset
271 "GlobalMapping already established!");
anatofuz
parents:
diff changeset
272 V = std::string(Name);
anatofuz
parents:
diff changeset
273 }
anatofuz
parents:
diff changeset
274 return OldVal;
anatofuz
parents:
diff changeset
275 }
anatofuz
parents:
diff changeset
276
anatofuz
parents:
diff changeset
277 uint64_t ExecutionEngine::getAddressToGlobalIfAvailable(StringRef S) {
anatofuz
parents:
diff changeset
278 std::lock_guard<sys::Mutex> locked(lock);
anatofuz
parents:
diff changeset
279 uint64_t Address = 0;
anatofuz
parents:
diff changeset
280 ExecutionEngineState::GlobalAddressMapTy::iterator I =
anatofuz
parents:
diff changeset
281 EEState.getGlobalAddressMap().find(S);
anatofuz
parents:
diff changeset
282 if (I != EEState.getGlobalAddressMap().end())
anatofuz
parents:
diff changeset
283 Address = I->second;
anatofuz
parents:
diff changeset
284 return Address;
anatofuz
parents:
diff changeset
285 }
anatofuz
parents:
diff changeset
286
anatofuz
parents:
diff changeset
287
anatofuz
parents:
diff changeset
288 void *ExecutionEngine::getPointerToGlobalIfAvailable(StringRef S) {
anatofuz
parents:
diff changeset
289 std::lock_guard<sys::Mutex> locked(lock);
anatofuz
parents:
diff changeset
290 if (void* Address = (void *) getAddressToGlobalIfAvailable(S))
anatofuz
parents:
diff changeset
291 return Address;
anatofuz
parents:
diff changeset
292 return nullptr;
anatofuz
parents:
diff changeset
293 }
anatofuz
parents:
diff changeset
294
anatofuz
parents:
diff changeset
295 void *ExecutionEngine::getPointerToGlobalIfAvailable(const GlobalValue *GV) {
anatofuz
parents:
diff changeset
296 std::lock_guard<sys::Mutex> locked(lock);
anatofuz
parents:
diff changeset
297 return getPointerToGlobalIfAvailable(getMangledName(GV));
anatofuz
parents:
diff changeset
298 }
anatofuz
parents:
diff changeset
299
anatofuz
parents:
diff changeset
300 const GlobalValue *ExecutionEngine::getGlobalValueAtAddress(void *Addr) {
anatofuz
parents:
diff changeset
301 std::lock_guard<sys::Mutex> locked(lock);
anatofuz
parents:
diff changeset
302
anatofuz
parents:
diff changeset
303 // If we haven't computed the reverse mapping yet, do so first.
anatofuz
parents:
diff changeset
304 if (EEState.getGlobalAddressReverseMap().empty()) {
anatofuz
parents:
diff changeset
305 for (ExecutionEngineState::GlobalAddressMapTy::iterator
anatofuz
parents:
diff changeset
306 I = EEState.getGlobalAddressMap().begin(),
anatofuz
parents:
diff changeset
307 E = EEState.getGlobalAddressMap().end(); I != E; ++I) {
anatofuz
parents:
diff changeset
308 StringRef Name = I->first();
anatofuz
parents:
diff changeset
309 uint64_t Addr = I->second;
anatofuz
parents:
diff changeset
310 EEState.getGlobalAddressReverseMap().insert(
anatofuz
parents:
diff changeset
311 std::make_pair(Addr, std::string(Name)));
anatofuz
parents:
diff changeset
312 }
anatofuz
parents:
diff changeset
313 }
anatofuz
parents:
diff changeset
314
anatofuz
parents:
diff changeset
315 std::map<uint64_t, std::string>::iterator I =
anatofuz
parents:
diff changeset
316 EEState.getGlobalAddressReverseMap().find((uint64_t) Addr);
anatofuz
parents:
diff changeset
317
anatofuz
parents:
diff changeset
318 if (I != EEState.getGlobalAddressReverseMap().end()) {
anatofuz
parents:
diff changeset
319 StringRef Name = I->second;
anatofuz
parents:
diff changeset
320 for (unsigned i = 0, e = Modules.size(); i != e; ++i)
anatofuz
parents:
diff changeset
321 if (GlobalValue *GV = Modules[i]->getNamedValue(Name))
anatofuz
parents:
diff changeset
322 return GV;
anatofuz
parents:
diff changeset
323 }
anatofuz
parents:
diff changeset
324 return nullptr;
anatofuz
parents:
diff changeset
325 }
anatofuz
parents:
diff changeset
326
anatofuz
parents:
diff changeset
327 namespace {
anatofuz
parents:
diff changeset
328 class ArgvArray {
anatofuz
parents:
diff changeset
329 std::unique_ptr<char[]> Array;
anatofuz
parents:
diff changeset
330 std::vector<std::unique_ptr<char[]>> Values;
anatofuz
parents:
diff changeset
331 public:
anatofuz
parents:
diff changeset
332 /// Turn a vector of strings into a nice argv style array of pointers to null
anatofuz
parents:
diff changeset
333 /// terminated strings.
anatofuz
parents:
diff changeset
334 void *reset(LLVMContext &C, ExecutionEngine *EE,
anatofuz
parents:
diff changeset
335 const std::vector<std::string> &InputArgv);
anatofuz
parents:
diff changeset
336 };
anatofuz
parents:
diff changeset
337 } // anonymous namespace
anatofuz
parents:
diff changeset
338 void *ArgvArray::reset(LLVMContext &C, ExecutionEngine *EE,
anatofuz
parents:
diff changeset
339 const std::vector<std::string> &InputArgv) {
anatofuz
parents:
diff changeset
340 Values.clear(); // Free the old contents.
anatofuz
parents:
diff changeset
341 Values.reserve(InputArgv.size());
anatofuz
parents:
diff changeset
342 unsigned PtrSize = EE->getDataLayout().getPointerSize();
anatofuz
parents:
diff changeset
343 Array = std::make_unique<char[]>((InputArgv.size()+1)*PtrSize);
anatofuz
parents:
diff changeset
344
anatofuz
parents:
diff changeset
345 LLVM_DEBUG(dbgs() << "JIT: ARGV = " << (void *)Array.get() << "\n");
anatofuz
parents:
diff changeset
346 Type *SBytePtr = Type::getInt8PtrTy(C);
anatofuz
parents:
diff changeset
347
anatofuz
parents:
diff changeset
348 for (unsigned i = 0; i != InputArgv.size(); ++i) {
anatofuz
parents:
diff changeset
349 unsigned Size = InputArgv[i].size()+1;
anatofuz
parents:
diff changeset
350 auto Dest = std::make_unique<char[]>(Size);
anatofuz
parents:
diff changeset
351 LLVM_DEBUG(dbgs() << "JIT: ARGV[" << i << "] = " << (void *)Dest.get()
anatofuz
parents:
diff changeset
352 << "\n");
anatofuz
parents:
diff changeset
353
anatofuz
parents:
diff changeset
354 std::copy(InputArgv[i].begin(), InputArgv[i].end(), Dest.get());
anatofuz
parents:
diff changeset
355 Dest[Size-1] = 0;
anatofuz
parents:
diff changeset
356
anatofuz
parents:
diff changeset
357 // Endian safe: Array[i] = (PointerTy)Dest;
anatofuz
parents:
diff changeset
358 EE->StoreValueToMemory(PTOGV(Dest.get()),
anatofuz
parents:
diff changeset
359 (GenericValue*)(&Array[i*PtrSize]), SBytePtr);
anatofuz
parents:
diff changeset
360 Values.push_back(std::move(Dest));
anatofuz
parents:
diff changeset
361 }
anatofuz
parents:
diff changeset
362
anatofuz
parents:
diff changeset
363 // Null terminate it
anatofuz
parents:
diff changeset
364 EE->StoreValueToMemory(PTOGV(nullptr),
anatofuz
parents:
diff changeset
365 (GenericValue*)(&Array[InputArgv.size()*PtrSize]),
anatofuz
parents:
diff changeset
366 SBytePtr);
anatofuz
parents:
diff changeset
367 return Array.get();
anatofuz
parents:
diff changeset
368 }
anatofuz
parents:
diff changeset
369
anatofuz
parents:
diff changeset
370 void ExecutionEngine::runStaticConstructorsDestructors(Module &module,
anatofuz
parents:
diff changeset
371 bool isDtors) {
anatofuz
parents:
diff changeset
372 StringRef Name(isDtors ? "llvm.global_dtors" : "llvm.global_ctors");
anatofuz
parents:
diff changeset
373 GlobalVariable *GV = module.getNamedGlobal(Name);
anatofuz
parents:
diff changeset
374
anatofuz
parents:
diff changeset
375 // If this global has internal linkage, or if it has a use, then it must be
anatofuz
parents:
diff changeset
376 // an old-style (llvmgcc3) static ctor with __main linked in and in use. If
anatofuz
parents:
diff changeset
377 // this is the case, don't execute any of the global ctors, __main will do
anatofuz
parents:
diff changeset
378 // it.
anatofuz
parents:
diff changeset
379 if (!GV || GV->isDeclaration() || GV->hasLocalLinkage()) return;
anatofuz
parents:
diff changeset
380
anatofuz
parents:
diff changeset
381 // Should be an array of '{ i32, void ()* }' structs. The first value is
anatofuz
parents:
diff changeset
382 // the init priority, which we ignore.
anatofuz
parents:
diff changeset
383 ConstantArray *InitList = dyn_cast<ConstantArray>(GV->getInitializer());
anatofuz
parents:
diff changeset
384 if (!InitList)
anatofuz
parents:
diff changeset
385 return;
anatofuz
parents:
diff changeset
386 for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i) {
anatofuz
parents:
diff changeset
387 ConstantStruct *CS = dyn_cast<ConstantStruct>(InitList->getOperand(i));
anatofuz
parents:
diff changeset
388 if (!CS) continue;
anatofuz
parents:
diff changeset
389
anatofuz
parents:
diff changeset
390 Constant *FP = CS->getOperand(1);
anatofuz
parents:
diff changeset
391 if (FP->isNullValue())
anatofuz
parents:
diff changeset
392 continue; // Found a sentinal value, ignore.
anatofuz
parents:
diff changeset
393
anatofuz
parents:
diff changeset
394 // Strip off constant expression casts.
anatofuz
parents:
diff changeset
395 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(FP))
anatofuz
parents:
diff changeset
396 if (CE->isCast())
anatofuz
parents:
diff changeset
397 FP = CE->getOperand(0);
anatofuz
parents:
diff changeset
398
anatofuz
parents:
diff changeset
399 // Execute the ctor/dtor function!
anatofuz
parents:
diff changeset
400 if (Function *F = dyn_cast<Function>(FP))
anatofuz
parents:
diff changeset
401 runFunction(F, None);
anatofuz
parents:
diff changeset
402
anatofuz
parents:
diff changeset
403 // FIXME: It is marginally lame that we just do nothing here if we see an
anatofuz
parents:
diff changeset
404 // entry we don't recognize. It might not be unreasonable for the verifier
anatofuz
parents:
diff changeset
405 // to not even allow this and just assert here.
anatofuz
parents:
diff changeset
406 }
anatofuz
parents:
diff changeset
407 }
anatofuz
parents:
diff changeset
408
anatofuz
parents:
diff changeset
409 void ExecutionEngine::runStaticConstructorsDestructors(bool isDtors) {
anatofuz
parents:
diff changeset
410 // Execute global ctors/dtors for each module in the program.
anatofuz
parents:
diff changeset
411 for (std::unique_ptr<Module> &M : Modules)
anatofuz
parents:
diff changeset
412 runStaticConstructorsDestructors(*M, isDtors);
anatofuz
parents:
diff changeset
413 }
anatofuz
parents:
diff changeset
414
anatofuz
parents:
diff changeset
415 #ifndef NDEBUG
anatofuz
parents:
diff changeset
416 /// isTargetNullPtr - Return whether the target pointer stored at Loc is null.
anatofuz
parents:
diff changeset
417 static bool isTargetNullPtr(ExecutionEngine *EE, void *Loc) {
anatofuz
parents:
diff changeset
418 unsigned PtrSize = EE->getDataLayout().getPointerSize();
anatofuz
parents:
diff changeset
419 for (unsigned i = 0; i < PtrSize; ++i)
anatofuz
parents:
diff changeset
420 if (*(i + (uint8_t*)Loc))
anatofuz
parents:
diff changeset
421 return false;
anatofuz
parents:
diff changeset
422 return true;
anatofuz
parents:
diff changeset
423 }
anatofuz
parents:
diff changeset
424 #endif
anatofuz
parents:
diff changeset
425
anatofuz
parents:
diff changeset
426 int ExecutionEngine::runFunctionAsMain(Function *Fn,
anatofuz
parents:
diff changeset
427 const std::vector<std::string> &argv,
anatofuz
parents:
diff changeset
428 const char * const * envp) {
anatofuz
parents:
diff changeset
429 std::vector<GenericValue> GVArgs;
anatofuz
parents:
diff changeset
430 GenericValue GVArgc;
anatofuz
parents:
diff changeset
431 GVArgc.IntVal = APInt(32, argv.size());
anatofuz
parents:
diff changeset
432
anatofuz
parents:
diff changeset
433 // Check main() type
anatofuz
parents:
diff changeset
434 unsigned NumArgs = Fn->getFunctionType()->getNumParams();
anatofuz
parents:
diff changeset
435 FunctionType *FTy = Fn->getFunctionType();
anatofuz
parents:
diff changeset
436 Type* PPInt8Ty = Type::getInt8PtrTy(Fn->getContext())->getPointerTo();
anatofuz
parents:
diff changeset
437
anatofuz
parents:
diff changeset
438 // Check the argument types.
anatofuz
parents:
diff changeset
439 if (NumArgs > 3)
anatofuz
parents:
diff changeset
440 report_fatal_error("Invalid number of arguments of main() supplied");
anatofuz
parents:
diff changeset
441 if (NumArgs >= 3 && FTy->getParamType(2) != PPInt8Ty)
anatofuz
parents:
diff changeset
442 report_fatal_error("Invalid type for third argument of main() supplied");
anatofuz
parents:
diff changeset
443 if (NumArgs >= 2 && FTy->getParamType(1) != PPInt8Ty)
anatofuz
parents:
diff changeset
444 report_fatal_error("Invalid type for second argument of main() supplied");
anatofuz
parents:
diff changeset
445 if (NumArgs >= 1 && !FTy->getParamType(0)->isIntegerTy(32))
anatofuz
parents:
diff changeset
446 report_fatal_error("Invalid type for first argument of main() supplied");
anatofuz
parents:
diff changeset
447 if (!FTy->getReturnType()->isIntegerTy() &&
anatofuz
parents:
diff changeset
448 !FTy->getReturnType()->isVoidTy())
anatofuz
parents:
diff changeset
449 report_fatal_error("Invalid return type of main() supplied");
anatofuz
parents:
diff changeset
450
anatofuz
parents:
diff changeset
451 ArgvArray CArgv;
anatofuz
parents:
diff changeset
452 ArgvArray CEnv;
anatofuz
parents:
diff changeset
453 if (NumArgs) {
anatofuz
parents:
diff changeset
454 GVArgs.push_back(GVArgc); // Arg #0 = argc.
anatofuz
parents:
diff changeset
455 if (NumArgs > 1) {
anatofuz
parents:
diff changeset
456 // Arg #1 = argv.
anatofuz
parents:
diff changeset
457 GVArgs.push_back(PTOGV(CArgv.reset(Fn->getContext(), this, argv)));
anatofuz
parents:
diff changeset
458 assert(!isTargetNullPtr(this, GVTOP(GVArgs[1])) &&
anatofuz
parents:
diff changeset
459 "argv[0] was null after CreateArgv");
anatofuz
parents:
diff changeset
460 if (NumArgs > 2) {
anatofuz
parents:
diff changeset
461 std::vector<std::string> EnvVars;
anatofuz
parents:
diff changeset
462 for (unsigned i = 0; envp[i]; ++i)
anatofuz
parents:
diff changeset
463 EnvVars.emplace_back(envp[i]);
anatofuz
parents:
diff changeset
464 // Arg #2 = envp.
anatofuz
parents:
diff changeset
465 GVArgs.push_back(PTOGV(CEnv.reset(Fn->getContext(), this, EnvVars)));
anatofuz
parents:
diff changeset
466 }
anatofuz
parents:
diff changeset
467 }
anatofuz
parents:
diff changeset
468 }
anatofuz
parents:
diff changeset
469
anatofuz
parents:
diff changeset
470 return runFunction(Fn, GVArgs).IntVal.getZExtValue();
anatofuz
parents:
diff changeset
471 }
anatofuz
parents:
diff changeset
472
anatofuz
parents:
diff changeset
473 EngineBuilder::EngineBuilder() : EngineBuilder(nullptr) {}
anatofuz
parents:
diff changeset
474
anatofuz
parents:
diff changeset
475 EngineBuilder::EngineBuilder(std::unique_ptr<Module> M)
anatofuz
parents:
diff changeset
476 : M(std::move(M)), WhichEngine(EngineKind::Either), ErrorStr(nullptr),
anatofuz
parents:
diff changeset
477 OptLevel(CodeGenOpt::Default), MemMgr(nullptr), Resolver(nullptr),
anatofuz
parents:
diff changeset
478 UseOrcMCJITReplacement(false) {
anatofuz
parents:
diff changeset
479 // IR module verification is enabled by default in debug builds, and disabled
anatofuz
parents:
diff changeset
480 // by default in release builds.
anatofuz
parents:
diff changeset
481 #ifndef NDEBUG
anatofuz
parents:
diff changeset
482 VerifyModules = true;
anatofuz
parents:
diff changeset
483 #else
anatofuz
parents:
diff changeset
484 VerifyModules = false;
anatofuz
parents:
diff changeset
485 #endif
anatofuz
parents:
diff changeset
486 }
anatofuz
parents:
diff changeset
487
anatofuz
parents:
diff changeset
488 EngineBuilder::~EngineBuilder() = default;
anatofuz
parents:
diff changeset
489
anatofuz
parents:
diff changeset
490 EngineBuilder &EngineBuilder::setMCJITMemoryManager(
anatofuz
parents:
diff changeset
491 std::unique_ptr<RTDyldMemoryManager> mcjmm) {
anatofuz
parents:
diff changeset
492 auto SharedMM = std::shared_ptr<RTDyldMemoryManager>(std::move(mcjmm));
anatofuz
parents:
diff changeset
493 MemMgr = SharedMM;
anatofuz
parents:
diff changeset
494 Resolver = SharedMM;
anatofuz
parents:
diff changeset
495 return *this;
anatofuz
parents:
diff changeset
496 }
anatofuz
parents:
diff changeset
497
anatofuz
parents:
diff changeset
498 EngineBuilder&
anatofuz
parents:
diff changeset
499 EngineBuilder::setMemoryManager(std::unique_ptr<MCJITMemoryManager> MM) {
anatofuz
parents:
diff changeset
500 MemMgr = std::shared_ptr<MCJITMemoryManager>(std::move(MM));
anatofuz
parents:
diff changeset
501 return *this;
anatofuz
parents:
diff changeset
502 }
anatofuz
parents:
diff changeset
503
anatofuz
parents:
diff changeset
504 EngineBuilder &
anatofuz
parents:
diff changeset
505 EngineBuilder::setSymbolResolver(std::unique_ptr<LegacyJITSymbolResolver> SR) {
anatofuz
parents:
diff changeset
506 Resolver = std::shared_ptr<LegacyJITSymbolResolver>(std::move(SR));
anatofuz
parents:
diff changeset
507 return *this;
anatofuz
parents:
diff changeset
508 }
anatofuz
parents:
diff changeset
509
anatofuz
parents:
diff changeset
510 ExecutionEngine *EngineBuilder::create(TargetMachine *TM) {
anatofuz
parents:
diff changeset
511 std::unique_ptr<TargetMachine> TheTM(TM); // Take ownership.
anatofuz
parents:
diff changeset
512
anatofuz
parents:
diff changeset
513 // Make sure we can resolve symbols in the program as well. The zero arg
anatofuz
parents:
diff changeset
514 // to the function tells DynamicLibrary to load the program, not a library.
anatofuz
parents:
diff changeset
515 if (sys::DynamicLibrary::LoadLibraryPermanently(nullptr, ErrorStr))
anatofuz
parents:
diff changeset
516 return nullptr;
anatofuz
parents:
diff changeset
517
anatofuz
parents:
diff changeset
518 // If the user specified a memory manager but didn't specify which engine to
anatofuz
parents:
diff changeset
519 // create, we assume they only want the JIT, and we fail if they only want
anatofuz
parents:
diff changeset
520 // the interpreter.
anatofuz
parents:
diff changeset
521 if (MemMgr) {
anatofuz
parents:
diff changeset
522 if (WhichEngine & EngineKind::JIT)
anatofuz
parents:
diff changeset
523 WhichEngine = EngineKind::JIT;
anatofuz
parents:
diff changeset
524 else {
anatofuz
parents:
diff changeset
525 if (ErrorStr)
anatofuz
parents:
diff changeset
526 *ErrorStr = "Cannot create an interpreter with a memory manager.";
anatofuz
parents:
diff changeset
527 return nullptr;
anatofuz
parents:
diff changeset
528 }
anatofuz
parents:
diff changeset
529 }
anatofuz
parents:
diff changeset
530
anatofuz
parents:
diff changeset
531 // Unless the interpreter was explicitly selected or the JIT is not linked,
anatofuz
parents:
diff changeset
532 // try making a JIT.
anatofuz
parents:
diff changeset
533 if ((WhichEngine & EngineKind::JIT) && TheTM) {
anatofuz
parents:
diff changeset
534 if (!TM->getTarget().hasJIT()) {
anatofuz
parents:
diff changeset
535 errs() << "WARNING: This target JIT is not designed for the host"
anatofuz
parents:
diff changeset
536 << " you are running. If bad things happen, please choose"
anatofuz
parents:
diff changeset
537 << " a different -march switch.\n";
anatofuz
parents:
diff changeset
538 }
anatofuz
parents:
diff changeset
539
anatofuz
parents:
diff changeset
540 ExecutionEngine *EE = nullptr;
anatofuz
parents:
diff changeset
541 if (ExecutionEngine::OrcMCJITReplacementCtor && UseOrcMCJITReplacement) {
anatofuz
parents:
diff changeset
542 EE = ExecutionEngine::OrcMCJITReplacementCtor(ErrorStr, std::move(MemMgr),
anatofuz
parents:
diff changeset
543 std::move(Resolver),
anatofuz
parents:
diff changeset
544 std::move(TheTM));
anatofuz
parents:
diff changeset
545 EE->addModule(std::move(M));
anatofuz
parents:
diff changeset
546 } else if (ExecutionEngine::MCJITCtor)
anatofuz
parents:
diff changeset
547 EE = ExecutionEngine::MCJITCtor(std::move(M), ErrorStr, std::move(MemMgr),
anatofuz
parents:
diff changeset
548 std::move(Resolver), std::move(TheTM));
anatofuz
parents:
diff changeset
549
anatofuz
parents:
diff changeset
550 if (EE) {
anatofuz
parents:
diff changeset
551 EE->setVerifyModules(VerifyModules);
anatofuz
parents:
diff changeset
552 return EE;
anatofuz
parents:
diff changeset
553 }
anatofuz
parents:
diff changeset
554 }
anatofuz
parents:
diff changeset
555
anatofuz
parents:
diff changeset
556 // If we can't make a JIT and we didn't request one specifically, try making
anatofuz
parents:
diff changeset
557 // an interpreter instead.
anatofuz
parents:
diff changeset
558 if (WhichEngine & EngineKind::Interpreter) {
anatofuz
parents:
diff changeset
559 if (ExecutionEngine::InterpCtor)
anatofuz
parents:
diff changeset
560 return ExecutionEngine::InterpCtor(std::move(M), ErrorStr);
anatofuz
parents:
diff changeset
561 if (ErrorStr)
anatofuz
parents:
diff changeset
562 *ErrorStr = "Interpreter has not been linked in.";
anatofuz
parents:
diff changeset
563 return nullptr;
anatofuz
parents:
diff changeset
564 }
anatofuz
parents:
diff changeset
565
anatofuz
parents:
diff changeset
566 if ((WhichEngine & EngineKind::JIT) && !ExecutionEngine::MCJITCtor) {
anatofuz
parents:
diff changeset
567 if (ErrorStr)
anatofuz
parents:
diff changeset
568 *ErrorStr = "JIT has not been linked in.";
anatofuz
parents:
diff changeset
569 }
anatofuz
parents:
diff changeset
570
anatofuz
parents:
diff changeset
571 return nullptr;
anatofuz
parents:
diff changeset
572 }
anatofuz
parents:
diff changeset
573
anatofuz
parents:
diff changeset
574 void *ExecutionEngine::getPointerToGlobal(const GlobalValue *GV) {
anatofuz
parents:
diff changeset
575 if (Function *F = const_cast<Function*>(dyn_cast<Function>(GV)))
anatofuz
parents:
diff changeset
576 return getPointerToFunction(F);
anatofuz
parents:
diff changeset
577
anatofuz
parents:
diff changeset
578 std::lock_guard<sys::Mutex> locked(lock);
anatofuz
parents:
diff changeset
579 if (void* P = getPointerToGlobalIfAvailable(GV))
anatofuz
parents:
diff changeset
580 return P;
anatofuz
parents:
diff changeset
581
anatofuz
parents:
diff changeset
582 // Global variable might have been added since interpreter started.
anatofuz
parents:
diff changeset
583 if (GlobalVariable *GVar =
anatofuz
parents:
diff changeset
584 const_cast<GlobalVariable *>(dyn_cast<GlobalVariable>(GV)))
anatofuz
parents:
diff changeset
585 EmitGlobalVariable(GVar);
anatofuz
parents:
diff changeset
586 else
anatofuz
parents:
diff changeset
587 llvm_unreachable("Global hasn't had an address allocated yet!");
anatofuz
parents:
diff changeset
588
anatofuz
parents:
diff changeset
589 return getPointerToGlobalIfAvailable(GV);
anatofuz
parents:
diff changeset
590 }
anatofuz
parents:
diff changeset
591
anatofuz
parents:
diff changeset
592 /// Converts a Constant* into a GenericValue, including handling of
anatofuz
parents:
diff changeset
593 /// ConstantExpr values.
anatofuz
parents:
diff changeset
594 GenericValue ExecutionEngine::getConstantValue(const Constant *C) {
anatofuz
parents:
diff changeset
595 // If its undefined, return the garbage.
anatofuz
parents:
diff changeset
596 if (isa<UndefValue>(C)) {
anatofuz
parents:
diff changeset
597 GenericValue Result;
anatofuz
parents:
diff changeset
598 switch (C->getType()->getTypeID()) {
anatofuz
parents:
diff changeset
599 default:
anatofuz
parents:
diff changeset
600 break;
anatofuz
parents:
diff changeset
601 case Type::IntegerTyID:
anatofuz
parents:
diff changeset
602 case Type::X86_FP80TyID:
anatofuz
parents:
diff changeset
603 case Type::FP128TyID:
anatofuz
parents:
diff changeset
604 case Type::PPC_FP128TyID:
anatofuz
parents:
diff changeset
605 // Although the value is undefined, we still have to construct an APInt
anatofuz
parents:
diff changeset
606 // with the correct bit width.
anatofuz
parents:
diff changeset
607 Result.IntVal = APInt(C->getType()->getPrimitiveSizeInBits(), 0);
anatofuz
parents:
diff changeset
608 break;
anatofuz
parents:
diff changeset
609 case Type::StructTyID: {
anatofuz
parents:
diff changeset
610 // if the whole struct is 'undef' just reserve memory for the value.
anatofuz
parents:
diff changeset
611 if(StructType *STy = dyn_cast<StructType>(C->getType())) {
anatofuz
parents:
diff changeset
612 unsigned int elemNum = STy->getNumElements();
anatofuz
parents:
diff changeset
613 Result.AggregateVal.resize(elemNum);
anatofuz
parents:
diff changeset
614 for (unsigned int i = 0; i < elemNum; ++i) {
anatofuz
parents:
diff changeset
615 Type *ElemTy = STy->getElementType(i);
anatofuz
parents:
diff changeset
616 if (ElemTy->isIntegerTy())
anatofuz
parents:
diff changeset
617 Result.AggregateVal[i].IntVal =
anatofuz
parents:
diff changeset
618 APInt(ElemTy->getPrimitiveSizeInBits(), 0);
anatofuz
parents:
diff changeset
619 else if (ElemTy->isAggregateType()) {
anatofuz
parents:
diff changeset
620 const Constant *ElemUndef = UndefValue::get(ElemTy);
anatofuz
parents:
diff changeset
621 Result.AggregateVal[i] = getConstantValue(ElemUndef);
anatofuz
parents:
diff changeset
622 }
anatofuz
parents:
diff changeset
623 }
anatofuz
parents:
diff changeset
624 }
anatofuz
parents:
diff changeset
625 }
anatofuz
parents:
diff changeset
626 break;
anatofuz
parents:
diff changeset
627 case Type::VectorTyID:
anatofuz
parents:
diff changeset
628 // if the whole vector is 'undef' just reserve memory for the value.
anatofuz
parents:
diff changeset
629 auto* VTy = cast<VectorType>(C->getType());
anatofuz
parents:
diff changeset
630 Type *ElemTy = VTy->getElementType();
anatofuz
parents:
diff changeset
631 unsigned int elemNum = VTy->getNumElements();
anatofuz
parents:
diff changeset
632 Result.AggregateVal.resize(elemNum);
anatofuz
parents:
diff changeset
633 if (ElemTy->isIntegerTy())
anatofuz
parents:
diff changeset
634 for (unsigned int i = 0; i < elemNum; ++i)
anatofuz
parents:
diff changeset
635 Result.AggregateVal[i].IntVal =
anatofuz
parents:
diff changeset
636 APInt(ElemTy->getPrimitiveSizeInBits(), 0);
anatofuz
parents:
diff changeset
637 break;
anatofuz
parents:
diff changeset
638 }
anatofuz
parents:
diff changeset
639 return Result;
anatofuz
parents:
diff changeset
640 }
anatofuz
parents:
diff changeset
641
anatofuz
parents:
diff changeset
642 // Otherwise, if the value is a ConstantExpr...
anatofuz
parents:
diff changeset
643 if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) {
anatofuz
parents:
diff changeset
644 Constant *Op0 = CE->getOperand(0);
anatofuz
parents:
diff changeset
645 switch (CE->getOpcode()) {
anatofuz
parents:
diff changeset
646 case Instruction::GetElementPtr: {
anatofuz
parents:
diff changeset
647 // Compute the index
anatofuz
parents:
diff changeset
648 GenericValue Result = getConstantValue(Op0);
anatofuz
parents:
diff changeset
649 APInt Offset(DL.getPointerSizeInBits(), 0);
anatofuz
parents:
diff changeset
650 cast<GEPOperator>(CE)->accumulateConstantOffset(DL, Offset);
anatofuz
parents:
diff changeset
651
anatofuz
parents:
diff changeset
652 char* tmp = (char*) Result.PointerVal;
anatofuz
parents:
diff changeset
653 Result = PTOGV(tmp + Offset.getSExtValue());
anatofuz
parents:
diff changeset
654 return Result;
anatofuz
parents:
diff changeset
655 }
anatofuz
parents:
diff changeset
656 case Instruction::Trunc: {
anatofuz
parents:
diff changeset
657 GenericValue GV = getConstantValue(Op0);
anatofuz
parents:
diff changeset
658 uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
anatofuz
parents:
diff changeset
659 GV.IntVal = GV.IntVal.trunc(BitWidth);
anatofuz
parents:
diff changeset
660 return GV;
anatofuz
parents:
diff changeset
661 }
anatofuz
parents:
diff changeset
662 case Instruction::ZExt: {
anatofuz
parents:
diff changeset
663 GenericValue GV = getConstantValue(Op0);
anatofuz
parents:
diff changeset
664 uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
anatofuz
parents:
diff changeset
665 GV.IntVal = GV.IntVal.zext(BitWidth);
anatofuz
parents:
diff changeset
666 return GV;
anatofuz
parents:
diff changeset
667 }
anatofuz
parents:
diff changeset
668 case Instruction::SExt: {
anatofuz
parents:
diff changeset
669 GenericValue GV = getConstantValue(Op0);
anatofuz
parents:
diff changeset
670 uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
anatofuz
parents:
diff changeset
671 GV.IntVal = GV.IntVal.sext(BitWidth);
anatofuz
parents:
diff changeset
672 return GV;
anatofuz
parents:
diff changeset
673 }
anatofuz
parents:
diff changeset
674 case Instruction::FPTrunc: {
anatofuz
parents:
diff changeset
675 // FIXME long double
anatofuz
parents:
diff changeset
676 GenericValue GV = getConstantValue(Op0);
anatofuz
parents:
diff changeset
677 GV.FloatVal = float(GV.DoubleVal);
anatofuz
parents:
diff changeset
678 return GV;
anatofuz
parents:
diff changeset
679 }
anatofuz
parents:
diff changeset
680 case Instruction::FPExt:{
anatofuz
parents:
diff changeset
681 // FIXME long double
anatofuz
parents:
diff changeset
682 GenericValue GV = getConstantValue(Op0);
anatofuz
parents:
diff changeset
683 GV.DoubleVal = double(GV.FloatVal);
anatofuz
parents:
diff changeset
684 return GV;
anatofuz
parents:
diff changeset
685 }
anatofuz
parents:
diff changeset
686 case Instruction::UIToFP: {
anatofuz
parents:
diff changeset
687 GenericValue GV = getConstantValue(Op0);
anatofuz
parents:
diff changeset
688 if (CE->getType()->isFloatTy())
anatofuz
parents:
diff changeset
689 GV.FloatVal = float(GV.IntVal.roundToDouble());
anatofuz
parents:
diff changeset
690 else if (CE->getType()->isDoubleTy())
anatofuz
parents:
diff changeset
691 GV.DoubleVal = GV.IntVal.roundToDouble();
anatofuz
parents:
diff changeset
692 else if (CE->getType()->isX86_FP80Ty()) {
anatofuz
parents:
diff changeset
693 APFloat apf = APFloat::getZero(APFloat::x87DoubleExtended());
anatofuz
parents:
diff changeset
694 (void)apf.convertFromAPInt(GV.IntVal,
anatofuz
parents:
diff changeset
695 false,
anatofuz
parents:
diff changeset
696 APFloat::rmNearestTiesToEven);
anatofuz
parents:
diff changeset
697 GV.IntVal = apf.bitcastToAPInt();
anatofuz
parents:
diff changeset
698 }
anatofuz
parents:
diff changeset
699 return GV;
anatofuz
parents:
diff changeset
700 }
anatofuz
parents:
diff changeset
701 case Instruction::SIToFP: {
anatofuz
parents:
diff changeset
702 GenericValue GV = getConstantValue(Op0);
anatofuz
parents:
diff changeset
703 if (CE->getType()->isFloatTy())
anatofuz
parents:
diff changeset
704 GV.FloatVal = float(GV.IntVal.signedRoundToDouble());
anatofuz
parents:
diff changeset
705 else if (CE->getType()->isDoubleTy())
anatofuz
parents:
diff changeset
706 GV.DoubleVal = GV.IntVal.signedRoundToDouble();
anatofuz
parents:
diff changeset
707 else if (CE->getType()->isX86_FP80Ty()) {
anatofuz
parents:
diff changeset
708 APFloat apf = APFloat::getZero(APFloat::x87DoubleExtended());
anatofuz
parents:
diff changeset
709 (void)apf.convertFromAPInt(GV.IntVal,
anatofuz
parents:
diff changeset
710 true,
anatofuz
parents:
diff changeset
711 APFloat::rmNearestTiesToEven);
anatofuz
parents:
diff changeset
712 GV.IntVal = apf.bitcastToAPInt();
anatofuz
parents:
diff changeset
713 }
anatofuz
parents:
diff changeset
714 return GV;
anatofuz
parents:
diff changeset
715 }
anatofuz
parents:
diff changeset
716 case Instruction::FPToUI: // double->APInt conversion handles sign
anatofuz
parents:
diff changeset
717 case Instruction::FPToSI: {
anatofuz
parents:
diff changeset
718 GenericValue GV = getConstantValue(Op0);
anatofuz
parents:
diff changeset
719 uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
anatofuz
parents:
diff changeset
720 if (Op0->getType()->isFloatTy())
anatofuz
parents:
diff changeset
721 GV.IntVal = APIntOps::RoundFloatToAPInt(GV.FloatVal, BitWidth);
anatofuz
parents:
diff changeset
722 else if (Op0->getType()->isDoubleTy())
anatofuz
parents:
diff changeset
723 GV.IntVal = APIntOps::RoundDoubleToAPInt(GV.DoubleVal, BitWidth);
anatofuz
parents:
diff changeset
724 else if (Op0->getType()->isX86_FP80Ty()) {
anatofuz
parents:
diff changeset
725 APFloat apf = APFloat(APFloat::x87DoubleExtended(), GV.IntVal);
anatofuz
parents:
diff changeset
726 uint64_t v;
anatofuz
parents:
diff changeset
727 bool ignored;
anatofuz
parents:
diff changeset
728 (void)apf.convertToInteger(makeMutableArrayRef(v), BitWidth,
anatofuz
parents:
diff changeset
729 CE->getOpcode()==Instruction::FPToSI,
anatofuz
parents:
diff changeset
730 APFloat::rmTowardZero, &ignored);
anatofuz
parents:
diff changeset
731 GV.IntVal = v; // endian?
anatofuz
parents:
diff changeset
732 }
anatofuz
parents:
diff changeset
733 return GV;
anatofuz
parents:
diff changeset
734 }
anatofuz
parents:
diff changeset
735 case Instruction::PtrToInt: {
anatofuz
parents:
diff changeset
736 GenericValue GV = getConstantValue(Op0);
anatofuz
parents:
diff changeset
737 uint32_t PtrWidth = DL.getTypeSizeInBits(Op0->getType());
anatofuz
parents:
diff changeset
738 assert(PtrWidth <= 64 && "Bad pointer width");
anatofuz
parents:
diff changeset
739 GV.IntVal = APInt(PtrWidth, uintptr_t(GV.PointerVal));
anatofuz
parents:
diff changeset
740 uint32_t IntWidth = DL.getTypeSizeInBits(CE->getType());
anatofuz
parents:
diff changeset
741 GV.IntVal = GV.IntVal.zextOrTrunc(IntWidth);
anatofuz
parents:
diff changeset
742 return GV;
anatofuz
parents:
diff changeset
743 }
anatofuz
parents:
diff changeset
744 case Instruction::IntToPtr: {
anatofuz
parents:
diff changeset
745 GenericValue GV = getConstantValue(Op0);
anatofuz
parents:
diff changeset
746 uint32_t PtrWidth = DL.getTypeSizeInBits(CE->getType());
anatofuz
parents:
diff changeset
747 GV.IntVal = GV.IntVal.zextOrTrunc(PtrWidth);
anatofuz
parents:
diff changeset
748 assert(GV.IntVal.getBitWidth() <= 64 && "Bad pointer width");
anatofuz
parents:
diff changeset
749 GV.PointerVal = PointerTy(uintptr_t(GV.IntVal.getZExtValue()));
anatofuz
parents:
diff changeset
750 return GV;
anatofuz
parents:
diff changeset
751 }
anatofuz
parents:
diff changeset
752 case Instruction::BitCast: {
anatofuz
parents:
diff changeset
753 GenericValue GV = getConstantValue(Op0);
anatofuz
parents:
diff changeset
754 Type* DestTy = CE->getType();
anatofuz
parents:
diff changeset
755 switch (Op0->getType()->getTypeID()) {
anatofuz
parents:
diff changeset
756 default: llvm_unreachable("Invalid bitcast operand");
anatofuz
parents:
diff changeset
757 case Type::IntegerTyID:
anatofuz
parents:
diff changeset
758 assert(DestTy->isFloatingPointTy() && "invalid bitcast");
anatofuz
parents:
diff changeset
759 if (DestTy->isFloatTy())
anatofuz
parents:
diff changeset
760 GV.FloatVal = GV.IntVal.bitsToFloat();
anatofuz
parents:
diff changeset
761 else if (DestTy->isDoubleTy())
anatofuz
parents:
diff changeset
762 GV.DoubleVal = GV.IntVal.bitsToDouble();
anatofuz
parents:
diff changeset
763 break;
anatofuz
parents:
diff changeset
764 case Type::FloatTyID:
anatofuz
parents:
diff changeset
765 assert(DestTy->isIntegerTy(32) && "Invalid bitcast");
anatofuz
parents:
diff changeset
766 GV.IntVal = APInt::floatToBits(GV.FloatVal);
anatofuz
parents:
diff changeset
767 break;
anatofuz
parents:
diff changeset
768 case Type::DoubleTyID:
anatofuz
parents:
diff changeset
769 assert(DestTy->isIntegerTy(64) && "Invalid bitcast");
anatofuz
parents:
diff changeset
770 GV.IntVal = APInt::doubleToBits(GV.DoubleVal);
anatofuz
parents:
diff changeset
771 break;
anatofuz
parents:
diff changeset
772 case Type::PointerTyID:
anatofuz
parents:
diff changeset
773 assert(DestTy->isPointerTy() && "Invalid bitcast");
anatofuz
parents:
diff changeset
774 break; // getConstantValue(Op0) above already converted it
anatofuz
parents:
diff changeset
775 }
anatofuz
parents:
diff changeset
776 return GV;
anatofuz
parents:
diff changeset
777 }
anatofuz
parents:
diff changeset
778 case Instruction::Add:
anatofuz
parents:
diff changeset
779 case Instruction::FAdd:
anatofuz
parents:
diff changeset
780 case Instruction::Sub:
anatofuz
parents:
diff changeset
781 case Instruction::FSub:
anatofuz
parents:
diff changeset
782 case Instruction::Mul:
anatofuz
parents:
diff changeset
783 case Instruction::FMul:
anatofuz
parents:
diff changeset
784 case Instruction::UDiv:
anatofuz
parents:
diff changeset
785 case Instruction::SDiv:
anatofuz
parents:
diff changeset
786 case Instruction::URem:
anatofuz
parents:
diff changeset
787 case Instruction::SRem:
anatofuz
parents:
diff changeset
788 case Instruction::And:
anatofuz
parents:
diff changeset
789 case Instruction::Or:
anatofuz
parents:
diff changeset
790 case Instruction::Xor: {
anatofuz
parents:
diff changeset
791 GenericValue LHS = getConstantValue(Op0);
anatofuz
parents:
diff changeset
792 GenericValue RHS = getConstantValue(CE->getOperand(1));
anatofuz
parents:
diff changeset
793 GenericValue GV;
anatofuz
parents:
diff changeset
794 switch (CE->getOperand(0)->getType()->getTypeID()) {
anatofuz
parents:
diff changeset
795 default: llvm_unreachable("Bad add type!");
anatofuz
parents:
diff changeset
796 case Type::IntegerTyID:
anatofuz
parents:
diff changeset
797 switch (CE->getOpcode()) {
anatofuz
parents:
diff changeset
798 default: llvm_unreachable("Invalid integer opcode");
anatofuz
parents:
diff changeset
799 case Instruction::Add: GV.IntVal = LHS.IntVal + RHS.IntVal; break;
anatofuz
parents:
diff changeset
800 case Instruction::Sub: GV.IntVal = LHS.IntVal - RHS.IntVal; break;
anatofuz
parents:
diff changeset
801 case Instruction::Mul: GV.IntVal = LHS.IntVal * RHS.IntVal; break;
anatofuz
parents:
diff changeset
802 case Instruction::UDiv:GV.IntVal = LHS.IntVal.udiv(RHS.IntVal); break;
anatofuz
parents:
diff changeset
803 case Instruction::SDiv:GV.IntVal = LHS.IntVal.sdiv(RHS.IntVal); break;
anatofuz
parents:
diff changeset
804 case Instruction::URem:GV.IntVal = LHS.IntVal.urem(RHS.IntVal); break;
anatofuz
parents:
diff changeset
805 case Instruction::SRem:GV.IntVal = LHS.IntVal.srem(RHS.IntVal); break;
anatofuz
parents:
diff changeset
806 case Instruction::And: GV.IntVal = LHS.IntVal & RHS.IntVal; break;
anatofuz
parents:
diff changeset
807 case Instruction::Or: GV.IntVal = LHS.IntVal | RHS.IntVal; break;
anatofuz
parents:
diff changeset
808 case Instruction::Xor: GV.IntVal = LHS.IntVal ^ RHS.IntVal; break;
anatofuz
parents:
diff changeset
809 }
anatofuz
parents:
diff changeset
810 break;
anatofuz
parents:
diff changeset
811 case Type::FloatTyID:
anatofuz
parents:
diff changeset
812 switch (CE->getOpcode()) {
anatofuz
parents:
diff changeset
813 default: llvm_unreachable("Invalid float opcode");
anatofuz
parents:
diff changeset
814 case Instruction::FAdd:
anatofuz
parents:
diff changeset
815 GV.FloatVal = LHS.FloatVal + RHS.FloatVal; break;
anatofuz
parents:
diff changeset
816 case Instruction::FSub:
anatofuz
parents:
diff changeset
817 GV.FloatVal = LHS.FloatVal - RHS.FloatVal; break;
anatofuz
parents:
diff changeset
818 case Instruction::FMul:
anatofuz
parents:
diff changeset
819 GV.FloatVal = LHS.FloatVal * RHS.FloatVal; break;
anatofuz
parents:
diff changeset
820 case Instruction::FDiv:
anatofuz
parents:
diff changeset
821 GV.FloatVal = LHS.FloatVal / RHS.FloatVal; break;
anatofuz
parents:
diff changeset
822 case Instruction::FRem:
anatofuz
parents:
diff changeset
823 GV.FloatVal = std::fmod(LHS.FloatVal,RHS.FloatVal); break;
anatofuz
parents:
diff changeset
824 }
anatofuz
parents:
diff changeset
825 break;
anatofuz
parents:
diff changeset
826 case Type::DoubleTyID:
anatofuz
parents:
diff changeset
827 switch (CE->getOpcode()) {
anatofuz
parents:
diff changeset
828 default: llvm_unreachable("Invalid double opcode");
anatofuz
parents:
diff changeset
829 case Instruction::FAdd:
anatofuz
parents:
diff changeset
830 GV.DoubleVal = LHS.DoubleVal + RHS.DoubleVal; break;
anatofuz
parents:
diff changeset
831 case Instruction::FSub:
anatofuz
parents:
diff changeset
832 GV.DoubleVal = LHS.DoubleVal - RHS.DoubleVal; break;
anatofuz
parents:
diff changeset
833 case Instruction::FMul:
anatofuz
parents:
diff changeset
834 GV.DoubleVal = LHS.DoubleVal * RHS.DoubleVal; break;
anatofuz
parents:
diff changeset
835 case Instruction::FDiv:
anatofuz
parents:
diff changeset
836 GV.DoubleVal = LHS.DoubleVal / RHS.DoubleVal; break;
anatofuz
parents:
diff changeset
837 case Instruction::FRem:
anatofuz
parents:
diff changeset
838 GV.DoubleVal = std::fmod(LHS.DoubleVal,RHS.DoubleVal); break;
anatofuz
parents:
diff changeset
839 }
anatofuz
parents:
diff changeset
840 break;
anatofuz
parents:
diff changeset
841 case Type::X86_FP80TyID:
anatofuz
parents:
diff changeset
842 case Type::PPC_FP128TyID:
anatofuz
parents:
diff changeset
843 case Type::FP128TyID: {
anatofuz
parents:
diff changeset
844 const fltSemantics &Sem = CE->getOperand(0)->getType()->getFltSemantics();
anatofuz
parents:
diff changeset
845 APFloat apfLHS = APFloat(Sem, LHS.IntVal);
anatofuz
parents:
diff changeset
846 switch (CE->getOpcode()) {
anatofuz
parents:
diff changeset
847 default: llvm_unreachable("Invalid long double opcode");
anatofuz
parents:
diff changeset
848 case Instruction::FAdd:
anatofuz
parents:
diff changeset
849 apfLHS.add(APFloat(Sem, RHS.IntVal), APFloat::rmNearestTiesToEven);
anatofuz
parents:
diff changeset
850 GV.IntVal = apfLHS.bitcastToAPInt();
anatofuz
parents:
diff changeset
851 break;
anatofuz
parents:
diff changeset
852 case Instruction::FSub:
anatofuz
parents:
diff changeset
853 apfLHS.subtract(APFloat(Sem, RHS.IntVal),
anatofuz
parents:
diff changeset
854 APFloat::rmNearestTiesToEven);
anatofuz
parents:
diff changeset
855 GV.IntVal = apfLHS.bitcastToAPInt();
anatofuz
parents:
diff changeset
856 break;
anatofuz
parents:
diff changeset
857 case Instruction::FMul:
anatofuz
parents:
diff changeset
858 apfLHS.multiply(APFloat(Sem, RHS.IntVal),
anatofuz
parents:
diff changeset
859 APFloat::rmNearestTiesToEven);
anatofuz
parents:
diff changeset
860 GV.IntVal = apfLHS.bitcastToAPInt();
anatofuz
parents:
diff changeset
861 break;
anatofuz
parents:
diff changeset
862 case Instruction::FDiv:
anatofuz
parents:
diff changeset
863 apfLHS.divide(APFloat(Sem, RHS.IntVal),
anatofuz
parents:
diff changeset
864 APFloat::rmNearestTiesToEven);
anatofuz
parents:
diff changeset
865 GV.IntVal = apfLHS.bitcastToAPInt();
anatofuz
parents:
diff changeset
866 break;
anatofuz
parents:
diff changeset
867 case Instruction::FRem:
anatofuz
parents:
diff changeset
868 apfLHS.mod(APFloat(Sem, RHS.IntVal));
anatofuz
parents:
diff changeset
869 GV.IntVal = apfLHS.bitcastToAPInt();
anatofuz
parents:
diff changeset
870 break;
anatofuz
parents:
diff changeset
871 }
anatofuz
parents:
diff changeset
872 }
anatofuz
parents:
diff changeset
873 break;
anatofuz
parents:
diff changeset
874 }
anatofuz
parents:
diff changeset
875 return GV;
anatofuz
parents:
diff changeset
876 }
anatofuz
parents:
diff changeset
877 default:
anatofuz
parents:
diff changeset
878 break;
anatofuz
parents:
diff changeset
879 }
anatofuz
parents:
diff changeset
880
anatofuz
parents:
diff changeset
881 SmallString<256> Msg;
anatofuz
parents:
diff changeset
882 raw_svector_ostream OS(Msg);
anatofuz
parents:
diff changeset
883 OS << "ConstantExpr not handled: " << *CE;
anatofuz
parents:
diff changeset
884 report_fatal_error(OS.str());
anatofuz
parents:
diff changeset
885 }
anatofuz
parents:
diff changeset
886
anatofuz
parents:
diff changeset
887 // Otherwise, we have a simple constant.
anatofuz
parents:
diff changeset
888 GenericValue Result;
anatofuz
parents:
diff changeset
889 switch (C->getType()->getTypeID()) {
anatofuz
parents:
diff changeset
890 case Type::FloatTyID:
anatofuz
parents:
diff changeset
891 Result.FloatVal = cast<ConstantFP>(C)->getValueAPF().convertToFloat();
anatofuz
parents:
diff changeset
892 break;
anatofuz
parents:
diff changeset
893 case Type::DoubleTyID:
anatofuz
parents:
diff changeset
894 Result.DoubleVal = cast<ConstantFP>(C)->getValueAPF().convertToDouble();
anatofuz
parents:
diff changeset
895 break;
anatofuz
parents:
diff changeset
896 case Type::X86_FP80TyID:
anatofuz
parents:
diff changeset
897 case Type::FP128TyID:
anatofuz
parents:
diff changeset
898 case Type::PPC_FP128TyID:
anatofuz
parents:
diff changeset
899 Result.IntVal = cast <ConstantFP>(C)->getValueAPF().bitcastToAPInt();
anatofuz
parents:
diff changeset
900 break;
anatofuz
parents:
diff changeset
901 case Type::IntegerTyID:
anatofuz
parents:
diff changeset
902 Result.IntVal = cast<ConstantInt>(C)->getValue();
anatofuz
parents:
diff changeset
903 break;
anatofuz
parents:
diff changeset
904 case Type::PointerTyID:
anatofuz
parents:
diff changeset
905 while (auto *A = dyn_cast<GlobalAlias>(C)) {
anatofuz
parents:
diff changeset
906 C = A->getAliasee();
anatofuz
parents:
diff changeset
907 }
anatofuz
parents:
diff changeset
908 if (isa<ConstantPointerNull>(C))
anatofuz
parents:
diff changeset
909 Result.PointerVal = nullptr;
anatofuz
parents:
diff changeset
910 else if (const Function *F = dyn_cast<Function>(C))
anatofuz
parents:
diff changeset
911 Result = PTOGV(getPointerToFunctionOrStub(const_cast<Function*>(F)));
anatofuz
parents:
diff changeset
912 else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(C))
anatofuz
parents:
diff changeset
913 Result = PTOGV(getOrEmitGlobalVariable(const_cast<GlobalVariable*>(GV)));
anatofuz
parents:
diff changeset
914 else
anatofuz
parents:
diff changeset
915 llvm_unreachable("Unknown constant pointer type!");
anatofuz
parents:
diff changeset
916 break;
anatofuz
parents:
diff changeset
917 case Type::VectorTyID: {
anatofuz
parents:
diff changeset
918 unsigned elemNum;
anatofuz
parents:
diff changeset
919 Type* ElemTy;
anatofuz
parents:
diff changeset
920 const ConstantDataVector *CDV = dyn_cast<ConstantDataVector>(C);
anatofuz
parents:
diff changeset
921 const ConstantVector *CV = dyn_cast<ConstantVector>(C);
anatofuz
parents:
diff changeset
922 const ConstantAggregateZero *CAZ = dyn_cast<ConstantAggregateZero>(C);
anatofuz
parents:
diff changeset
923
anatofuz
parents:
diff changeset
924 if (CDV) {
anatofuz
parents:
diff changeset
925 elemNum = CDV->getNumElements();
anatofuz
parents:
diff changeset
926 ElemTy = CDV->getElementType();
anatofuz
parents:
diff changeset
927 } else if (CV || CAZ) {
anatofuz
parents:
diff changeset
928 auto* VTy = cast<VectorType>(C->getType());
anatofuz
parents:
diff changeset
929 elemNum = VTy->getNumElements();
anatofuz
parents:
diff changeset
930 ElemTy = VTy->getElementType();
anatofuz
parents:
diff changeset
931 } else {
anatofuz
parents:
diff changeset
932 llvm_unreachable("Unknown constant vector type!");
anatofuz
parents:
diff changeset
933 }
anatofuz
parents:
diff changeset
934
anatofuz
parents:
diff changeset
935 Result.AggregateVal.resize(elemNum);
anatofuz
parents:
diff changeset
936 // Check if vector holds floats.
anatofuz
parents:
diff changeset
937 if(ElemTy->isFloatTy()) {
anatofuz
parents:
diff changeset
938 if (CAZ) {
anatofuz
parents:
diff changeset
939 GenericValue floatZero;
anatofuz
parents:
diff changeset
940 floatZero.FloatVal = 0.f;
anatofuz
parents:
diff changeset
941 std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(),
anatofuz
parents:
diff changeset
942 floatZero);
anatofuz
parents:
diff changeset
943 break;
anatofuz
parents:
diff changeset
944 }
anatofuz
parents:
diff changeset
945 if(CV) {
anatofuz
parents:
diff changeset
946 for (unsigned i = 0; i < elemNum; ++i)
anatofuz
parents:
diff changeset
947 if (!isa<UndefValue>(CV->getOperand(i)))
anatofuz
parents:
diff changeset
948 Result.AggregateVal[i].FloatVal = cast<ConstantFP>(
anatofuz
parents:
diff changeset
949 CV->getOperand(i))->getValueAPF().convertToFloat();
anatofuz
parents:
diff changeset
950 break;
anatofuz
parents:
diff changeset
951 }
anatofuz
parents:
diff changeset
952 if(CDV)
anatofuz
parents:
diff changeset
953 for (unsigned i = 0; i < elemNum; ++i)
anatofuz
parents:
diff changeset
954 Result.AggregateVal[i].FloatVal = CDV->getElementAsFloat(i);
anatofuz
parents:
diff changeset
955
anatofuz
parents:
diff changeset
956 break;
anatofuz
parents:
diff changeset
957 }
anatofuz
parents:
diff changeset
958 // Check if vector holds doubles.
anatofuz
parents:
diff changeset
959 if (ElemTy->isDoubleTy()) {
anatofuz
parents:
diff changeset
960 if (CAZ) {
anatofuz
parents:
diff changeset
961 GenericValue doubleZero;
anatofuz
parents:
diff changeset
962 doubleZero.DoubleVal = 0.0;
anatofuz
parents:
diff changeset
963 std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(),
anatofuz
parents:
diff changeset
964 doubleZero);
anatofuz
parents:
diff changeset
965 break;
anatofuz
parents:
diff changeset
966 }
anatofuz
parents:
diff changeset
967 if(CV) {
anatofuz
parents:
diff changeset
968 for (unsigned i = 0; i < elemNum; ++i)
anatofuz
parents:
diff changeset
969 if (!isa<UndefValue>(CV->getOperand(i)))
anatofuz
parents:
diff changeset
970 Result.AggregateVal[i].DoubleVal = cast<ConstantFP>(
anatofuz
parents:
diff changeset
971 CV->getOperand(i))->getValueAPF().convertToDouble();
anatofuz
parents:
diff changeset
972 break;
anatofuz
parents:
diff changeset
973 }
anatofuz
parents:
diff changeset
974 if(CDV)
anatofuz
parents:
diff changeset
975 for (unsigned i = 0; i < elemNum; ++i)
anatofuz
parents:
diff changeset
976 Result.AggregateVal[i].DoubleVal = CDV->getElementAsDouble(i);
anatofuz
parents:
diff changeset
977
anatofuz
parents:
diff changeset
978 break;
anatofuz
parents:
diff changeset
979 }
anatofuz
parents:
diff changeset
980 // Check if vector holds integers.
anatofuz
parents:
diff changeset
981 if (ElemTy->isIntegerTy()) {
anatofuz
parents:
diff changeset
982 if (CAZ) {
anatofuz
parents:
diff changeset
983 GenericValue intZero;
anatofuz
parents:
diff changeset
984 intZero.IntVal = APInt(ElemTy->getScalarSizeInBits(), 0ull);
anatofuz
parents:
diff changeset
985 std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(),
anatofuz
parents:
diff changeset
986 intZero);
anatofuz
parents:
diff changeset
987 break;
anatofuz
parents:
diff changeset
988 }
anatofuz
parents:
diff changeset
989 if(CV) {
anatofuz
parents:
diff changeset
990 for (unsigned i = 0; i < elemNum; ++i)
anatofuz
parents:
diff changeset
991 if (!isa<UndefValue>(CV->getOperand(i)))
anatofuz
parents:
diff changeset
992 Result.AggregateVal[i].IntVal = cast<ConstantInt>(
anatofuz
parents:
diff changeset
993 CV->getOperand(i))->getValue();
anatofuz
parents:
diff changeset
994 else {
anatofuz
parents:
diff changeset
995 Result.AggregateVal[i].IntVal =
anatofuz
parents:
diff changeset
996 APInt(CV->getOperand(i)->getType()->getPrimitiveSizeInBits(), 0);
anatofuz
parents:
diff changeset
997 }
anatofuz
parents:
diff changeset
998 break;
anatofuz
parents:
diff changeset
999 }
anatofuz
parents:
diff changeset
1000 if(CDV)
anatofuz
parents:
diff changeset
1001 for (unsigned i = 0; i < elemNum; ++i)
anatofuz
parents:
diff changeset
1002 Result.AggregateVal[i].IntVal = APInt(
anatofuz
parents:
diff changeset
1003 CDV->getElementType()->getPrimitiveSizeInBits(),
anatofuz
parents:
diff changeset
1004 CDV->getElementAsInteger(i));
anatofuz
parents:
diff changeset
1005
anatofuz
parents:
diff changeset
1006 break;
anatofuz
parents:
diff changeset
1007 }
anatofuz
parents:
diff changeset
1008 llvm_unreachable("Unknown constant pointer type!");
anatofuz
parents:
diff changeset
1009 }
anatofuz
parents:
diff changeset
1010 break;
anatofuz
parents:
diff changeset
1011
anatofuz
parents:
diff changeset
1012 default:
anatofuz
parents:
diff changeset
1013 SmallString<256> Msg;
anatofuz
parents:
diff changeset
1014 raw_svector_ostream OS(Msg);
anatofuz
parents:
diff changeset
1015 OS << "ERROR: Constant unimplemented for type: " << *C->getType();
anatofuz
parents:
diff changeset
1016 report_fatal_error(OS.str());
anatofuz
parents:
diff changeset
1017 }
anatofuz
parents:
diff changeset
1018
anatofuz
parents:
diff changeset
1019 return Result;
anatofuz
parents:
diff changeset
1020 }
anatofuz
parents:
diff changeset
1021
anatofuz
parents:
diff changeset
1022 void ExecutionEngine::StoreValueToMemory(const GenericValue &Val,
anatofuz
parents:
diff changeset
1023 GenericValue *Ptr, Type *Ty) {
anatofuz
parents:
diff changeset
1024 const unsigned StoreBytes = getDataLayout().getTypeStoreSize(Ty);
anatofuz
parents:
diff changeset
1025
anatofuz
parents:
diff changeset
1026 switch (Ty->getTypeID()) {
anatofuz
parents:
diff changeset
1027 default:
anatofuz
parents:
diff changeset
1028 dbgs() << "Cannot store value of type " << *Ty << "!\n";
anatofuz
parents:
diff changeset
1029 break;
anatofuz
parents:
diff changeset
1030 case Type::IntegerTyID:
anatofuz
parents:
diff changeset
1031 StoreIntToMemory(Val.IntVal, (uint8_t*)Ptr, StoreBytes);
anatofuz
parents:
diff changeset
1032 break;
anatofuz
parents:
diff changeset
1033 case Type::FloatTyID:
anatofuz
parents:
diff changeset
1034 *((float*)Ptr) = Val.FloatVal;
anatofuz
parents:
diff changeset
1035 break;
anatofuz
parents:
diff changeset
1036 case Type::DoubleTyID:
anatofuz
parents:
diff changeset
1037 *((double*)Ptr) = Val.DoubleVal;
anatofuz
parents:
diff changeset
1038 break;
anatofuz
parents:
diff changeset
1039 case Type::X86_FP80TyID:
anatofuz
parents:
diff changeset
1040 memcpy(Ptr, Val.IntVal.getRawData(), 10);
anatofuz
parents:
diff changeset
1041 break;
anatofuz
parents:
diff changeset
1042 case Type::PointerTyID:
anatofuz
parents:
diff changeset
1043 // Ensure 64 bit target pointers are fully initialized on 32 bit hosts.
anatofuz
parents:
diff changeset
1044 if (StoreBytes != sizeof(PointerTy))
anatofuz
parents:
diff changeset
1045 memset(&(Ptr->PointerVal), 0, StoreBytes);
anatofuz
parents:
diff changeset
1046
anatofuz
parents:
diff changeset
1047 *((PointerTy*)Ptr) = Val.PointerVal;
anatofuz
parents:
diff changeset
1048 break;
anatofuz
parents:
diff changeset
1049 case Type::VectorTyID:
anatofuz
parents:
diff changeset
1050 for (unsigned i = 0; i < Val.AggregateVal.size(); ++i) {
anatofuz
parents:
diff changeset
1051 if (cast<VectorType>(Ty)->getElementType()->isDoubleTy())
anatofuz
parents:
diff changeset
1052 *(((double*)Ptr)+i) = Val.AggregateVal[i].DoubleVal;
anatofuz
parents:
diff changeset
1053 if (cast<VectorType>(Ty)->getElementType()->isFloatTy())
anatofuz
parents:
diff changeset
1054 *(((float*)Ptr)+i) = Val.AggregateVal[i].FloatVal;
anatofuz
parents:
diff changeset
1055 if (cast<VectorType>(Ty)->getElementType()->isIntegerTy()) {
anatofuz
parents:
diff changeset
1056 unsigned numOfBytes =(Val.AggregateVal[i].IntVal.getBitWidth()+7)/8;
anatofuz
parents:
diff changeset
1057 StoreIntToMemory(Val.AggregateVal[i].IntVal,
anatofuz
parents:
diff changeset
1058 (uint8_t*)Ptr + numOfBytes*i, numOfBytes);
anatofuz
parents:
diff changeset
1059 }
anatofuz
parents:
diff changeset
1060 }
anatofuz
parents:
diff changeset
1061 break;
anatofuz
parents:
diff changeset
1062 }
anatofuz
parents:
diff changeset
1063
anatofuz
parents:
diff changeset
1064 if (sys::IsLittleEndianHost != getDataLayout().isLittleEndian())
anatofuz
parents:
diff changeset
1065 // Host and target are different endian - reverse the stored bytes.
anatofuz
parents:
diff changeset
1066 std::reverse((uint8_t*)Ptr, StoreBytes + (uint8_t*)Ptr);
anatofuz
parents:
diff changeset
1067 }
anatofuz
parents:
diff changeset
1068
anatofuz
parents:
diff changeset
1069 /// FIXME: document
anatofuz
parents:
diff changeset
1070 ///
anatofuz
parents:
diff changeset
1071 void ExecutionEngine::LoadValueFromMemory(GenericValue &Result,
anatofuz
parents:
diff changeset
1072 GenericValue *Ptr,
anatofuz
parents:
diff changeset
1073 Type *Ty) {
anatofuz
parents:
diff changeset
1074 const unsigned LoadBytes = getDataLayout().getTypeStoreSize(Ty);
anatofuz
parents:
diff changeset
1075
anatofuz
parents:
diff changeset
1076 switch (Ty->getTypeID()) {
anatofuz
parents:
diff changeset
1077 case Type::IntegerTyID:
anatofuz
parents:
diff changeset
1078 // An APInt with all words initially zero.
anatofuz
parents:
diff changeset
1079 Result.IntVal = APInt(cast<IntegerType>(Ty)->getBitWidth(), 0);
anatofuz
parents:
diff changeset
1080 LoadIntFromMemory(Result.IntVal, (uint8_t*)Ptr, LoadBytes);
anatofuz
parents:
diff changeset
1081 break;
anatofuz
parents:
diff changeset
1082 case Type::FloatTyID:
anatofuz
parents:
diff changeset
1083 Result.FloatVal = *((float*)Ptr);
anatofuz
parents:
diff changeset
1084 break;
anatofuz
parents:
diff changeset
1085 case Type::DoubleTyID:
anatofuz
parents:
diff changeset
1086 Result.DoubleVal = *((double*)Ptr);
anatofuz
parents:
diff changeset
1087 break;
anatofuz
parents:
diff changeset
1088 case Type::PointerTyID:
anatofuz
parents:
diff changeset
1089 Result.PointerVal = *((PointerTy*)Ptr);
anatofuz
parents:
diff changeset
1090 break;
anatofuz
parents:
diff changeset
1091 case Type::X86_FP80TyID: {
anatofuz
parents:
diff changeset
1092 // This is endian dependent, but it will only work on x86 anyway.
anatofuz
parents:
diff changeset
1093 // FIXME: Will not trap if loading a signaling NaN.
anatofuz
parents:
diff changeset
1094 uint64_t y[2];
anatofuz
parents:
diff changeset
1095 memcpy(y, Ptr, 10);
anatofuz
parents:
diff changeset
1096 Result.IntVal = APInt(80, y);
anatofuz
parents:
diff changeset
1097 break;
anatofuz
parents:
diff changeset
1098 }
anatofuz
parents:
diff changeset
1099 case Type::VectorTyID: {
anatofuz
parents:
diff changeset
1100 auto *VT = cast<VectorType>(Ty);
anatofuz
parents:
diff changeset
1101 Type *ElemT = VT->getElementType();
anatofuz
parents:
diff changeset
1102 const unsigned numElems = VT->getNumElements();
anatofuz
parents:
diff changeset
1103 if (ElemT->isFloatTy()) {
anatofuz
parents:
diff changeset
1104 Result.AggregateVal.resize(numElems);
anatofuz
parents:
diff changeset
1105 for (unsigned i = 0; i < numElems; ++i)
anatofuz
parents:
diff changeset
1106 Result.AggregateVal[i].FloatVal = *((float*)Ptr+i);
anatofuz
parents:
diff changeset
1107 }
anatofuz
parents:
diff changeset
1108 if (ElemT->isDoubleTy()) {
anatofuz
parents:
diff changeset
1109 Result.AggregateVal.resize(numElems);
anatofuz
parents:
diff changeset
1110 for (unsigned i = 0; i < numElems; ++i)
anatofuz
parents:
diff changeset
1111 Result.AggregateVal[i].DoubleVal = *((double*)Ptr+i);
anatofuz
parents:
diff changeset
1112 }
anatofuz
parents:
diff changeset
1113 if (ElemT->isIntegerTy()) {
anatofuz
parents:
diff changeset
1114 GenericValue intZero;
anatofuz
parents:
diff changeset
1115 const unsigned elemBitWidth = cast<IntegerType>(ElemT)->getBitWidth();
anatofuz
parents:
diff changeset
1116 intZero.IntVal = APInt(elemBitWidth, 0);
anatofuz
parents:
diff changeset
1117 Result.AggregateVal.resize(numElems, intZero);
anatofuz
parents:
diff changeset
1118 for (unsigned i = 0; i < numElems; ++i)
anatofuz
parents:
diff changeset
1119 LoadIntFromMemory(Result.AggregateVal[i].IntVal,
anatofuz
parents:
diff changeset
1120 (uint8_t*)Ptr+((elemBitWidth+7)/8)*i, (elemBitWidth+7)/8);
anatofuz
parents:
diff changeset
1121 }
anatofuz
parents:
diff changeset
1122 break;
anatofuz
parents:
diff changeset
1123 }
anatofuz
parents:
diff changeset
1124 default:
anatofuz
parents:
diff changeset
1125 SmallString<256> Msg;
anatofuz
parents:
diff changeset
1126 raw_svector_ostream OS(Msg);
anatofuz
parents:
diff changeset
1127 OS << "Cannot load value of type " << *Ty << "!";
anatofuz
parents:
diff changeset
1128 report_fatal_error(OS.str());
anatofuz
parents:
diff changeset
1129 }
anatofuz
parents:
diff changeset
1130 }
anatofuz
parents:
diff changeset
1131
anatofuz
parents:
diff changeset
1132 void ExecutionEngine::InitializeMemory(const Constant *Init, void *Addr) {
anatofuz
parents:
diff changeset
1133 LLVM_DEBUG(dbgs() << "JIT: Initializing " << Addr << " ");
anatofuz
parents:
diff changeset
1134 LLVM_DEBUG(Init->dump());
anatofuz
parents:
diff changeset
1135 if (isa<UndefValue>(Init))
anatofuz
parents:
diff changeset
1136 return;
anatofuz
parents:
diff changeset
1137
anatofuz
parents:
diff changeset
1138 if (const ConstantVector *CP = dyn_cast<ConstantVector>(Init)) {
anatofuz
parents:
diff changeset
1139 unsigned ElementSize =
anatofuz
parents:
diff changeset
1140 getDataLayout().getTypeAllocSize(CP->getType()->getElementType());
anatofuz
parents:
diff changeset
1141 for (unsigned i = 0, e = CP->getNumOperands(); i != e; ++i)
anatofuz
parents:
diff changeset
1142 InitializeMemory(CP->getOperand(i), (char*)Addr+i*ElementSize);
anatofuz
parents:
diff changeset
1143 return;
anatofuz
parents:
diff changeset
1144 }
anatofuz
parents:
diff changeset
1145
anatofuz
parents:
diff changeset
1146 if (isa<ConstantAggregateZero>(Init)) {
anatofuz
parents:
diff changeset
1147 memset(Addr, 0, (size_t)getDataLayout().getTypeAllocSize(Init->getType()));
anatofuz
parents:
diff changeset
1148 return;
anatofuz
parents:
diff changeset
1149 }
anatofuz
parents:
diff changeset
1150
anatofuz
parents:
diff changeset
1151 if (const ConstantArray *CPA = dyn_cast<ConstantArray>(Init)) {
anatofuz
parents:
diff changeset
1152 unsigned ElementSize =
anatofuz
parents:
diff changeset
1153 getDataLayout().getTypeAllocSize(CPA->getType()->getElementType());
anatofuz
parents:
diff changeset
1154 for (unsigned i = 0, e = CPA->getNumOperands(); i != e; ++i)
anatofuz
parents:
diff changeset
1155 InitializeMemory(CPA->getOperand(i), (char*)Addr+i*ElementSize);
anatofuz
parents:
diff changeset
1156 return;
anatofuz
parents:
diff changeset
1157 }
anatofuz
parents:
diff changeset
1158
anatofuz
parents:
diff changeset
1159 if (const ConstantStruct *CPS = dyn_cast<ConstantStruct>(Init)) {
anatofuz
parents:
diff changeset
1160 const StructLayout *SL =
anatofuz
parents:
diff changeset
1161 getDataLayout().getStructLayout(cast<StructType>(CPS->getType()));
anatofuz
parents:
diff changeset
1162 for (unsigned i = 0, e = CPS->getNumOperands(); i != e; ++i)
anatofuz
parents:
diff changeset
1163 InitializeMemory(CPS->getOperand(i), (char*)Addr+SL->getElementOffset(i));
anatofuz
parents:
diff changeset
1164 return;
anatofuz
parents:
diff changeset
1165 }
anatofuz
parents:
diff changeset
1166
anatofuz
parents:
diff changeset
1167 if (const ConstantDataSequential *CDS =
anatofuz
parents:
diff changeset
1168 dyn_cast<ConstantDataSequential>(Init)) {
anatofuz
parents:
diff changeset
1169 // CDS is already laid out in host memory order.
anatofuz
parents:
diff changeset
1170 StringRef Data = CDS->getRawDataValues();
anatofuz
parents:
diff changeset
1171 memcpy(Addr, Data.data(), Data.size());
anatofuz
parents:
diff changeset
1172 return;
anatofuz
parents:
diff changeset
1173 }
anatofuz
parents:
diff changeset
1174
anatofuz
parents:
diff changeset
1175 if (Init->getType()->isFirstClassType()) {
anatofuz
parents:
diff changeset
1176 GenericValue Val = getConstantValue(Init);
anatofuz
parents:
diff changeset
1177 StoreValueToMemory(Val, (GenericValue*)Addr, Init->getType());
anatofuz
parents:
diff changeset
1178 return;
anatofuz
parents:
diff changeset
1179 }
anatofuz
parents:
diff changeset
1180
anatofuz
parents:
diff changeset
1181 LLVM_DEBUG(dbgs() << "Bad Type: " << *Init->getType() << "\n");
anatofuz
parents:
diff changeset
1182 llvm_unreachable("Unknown constant type to initialize memory with!");
anatofuz
parents:
diff changeset
1183 }
anatofuz
parents:
diff changeset
1184
anatofuz
parents:
diff changeset
1185 /// EmitGlobals - Emit all of the global variables to memory, storing their
anatofuz
parents:
diff changeset
1186 /// addresses into GlobalAddress. This must make sure to copy the contents of
anatofuz
parents:
diff changeset
1187 /// their initializers into the memory.
anatofuz
parents:
diff changeset
1188 void ExecutionEngine::emitGlobals() {
anatofuz
parents:
diff changeset
1189 // Loop over all of the global variables in the program, allocating the memory
anatofuz
parents:
diff changeset
1190 // to hold them. If there is more than one module, do a prepass over globals
anatofuz
parents:
diff changeset
1191 // to figure out how the different modules should link together.
anatofuz
parents:
diff changeset
1192 std::map<std::pair<std::string, Type*>,
anatofuz
parents:
diff changeset
1193 const GlobalValue*> LinkedGlobalsMap;
anatofuz
parents:
diff changeset
1194
anatofuz
parents:
diff changeset
1195 if (Modules.size() != 1) {
anatofuz
parents:
diff changeset
1196 for (unsigned m = 0, e = Modules.size(); m != e; ++m) {
anatofuz
parents:
diff changeset
1197 Module &M = *Modules[m];
anatofuz
parents:
diff changeset
1198 for (const auto &GV : M.globals()) {
anatofuz
parents:
diff changeset
1199 if (GV.hasLocalLinkage() || GV.isDeclaration() ||
anatofuz
parents:
diff changeset
1200 GV.hasAppendingLinkage() || !GV.hasName())
anatofuz
parents:
diff changeset
1201 continue;// Ignore external globals and globals with internal linkage.
anatofuz
parents:
diff changeset
1202
anatofuz
parents:
diff changeset
1203 const GlobalValue *&GVEntry = LinkedGlobalsMap[std::make_pair(
anatofuz
parents:
diff changeset
1204 std::string(GV.getName()), GV.getType())];
anatofuz
parents:
diff changeset
1205
anatofuz
parents:
diff changeset
1206 // If this is the first time we've seen this global, it is the canonical
anatofuz
parents:
diff changeset
1207 // version.
anatofuz
parents:
diff changeset
1208 if (!GVEntry) {
anatofuz
parents:
diff changeset
1209 GVEntry = &GV;
anatofuz
parents:
diff changeset
1210 continue;
anatofuz
parents:
diff changeset
1211 }
anatofuz
parents:
diff changeset
1212
anatofuz
parents:
diff changeset
1213 // If the existing global is strong, never replace it.
anatofuz
parents:
diff changeset
1214 if (GVEntry->hasExternalLinkage())
anatofuz
parents:
diff changeset
1215 continue;
anatofuz
parents:
diff changeset
1216
anatofuz
parents:
diff changeset
1217 // Otherwise, we know it's linkonce/weak, replace it if this is a strong
anatofuz
parents:
diff changeset
1218 // symbol. FIXME is this right for common?
anatofuz
parents:
diff changeset
1219 if (GV.hasExternalLinkage() || GVEntry->hasExternalWeakLinkage())
anatofuz
parents:
diff changeset
1220 GVEntry = &GV;
anatofuz
parents:
diff changeset
1221 }
anatofuz
parents:
diff changeset
1222 }
anatofuz
parents:
diff changeset
1223 }
anatofuz
parents:
diff changeset
1224
anatofuz
parents:
diff changeset
1225 std::vector<const GlobalValue*> NonCanonicalGlobals;
anatofuz
parents:
diff changeset
1226 for (unsigned m = 0, e = Modules.size(); m != e; ++m) {
anatofuz
parents:
diff changeset
1227 Module &M = *Modules[m];
anatofuz
parents:
diff changeset
1228 for (const auto &GV : M.globals()) {
anatofuz
parents:
diff changeset
1229 // In the multi-module case, see what this global maps to.
anatofuz
parents:
diff changeset
1230 if (!LinkedGlobalsMap.empty()) {
anatofuz
parents:
diff changeset
1231 if (const GlobalValue *GVEntry = LinkedGlobalsMap[std::make_pair(
anatofuz
parents:
diff changeset
1232 std::string(GV.getName()), GV.getType())]) {
anatofuz
parents:
diff changeset
1233 // If something else is the canonical global, ignore this one.
anatofuz
parents:
diff changeset
1234 if (GVEntry != &GV) {
anatofuz
parents:
diff changeset
1235 NonCanonicalGlobals.push_back(&GV);
anatofuz
parents:
diff changeset
1236 continue;
anatofuz
parents:
diff changeset
1237 }
anatofuz
parents:
diff changeset
1238 }
anatofuz
parents:
diff changeset
1239 }
anatofuz
parents:
diff changeset
1240
anatofuz
parents:
diff changeset
1241 if (!GV.isDeclaration()) {
anatofuz
parents:
diff changeset
1242 addGlobalMapping(&GV, getMemoryForGV(&GV));
anatofuz
parents:
diff changeset
1243 } else {
anatofuz
parents:
diff changeset
1244 // External variable reference. Try to use the dynamic loader to
anatofuz
parents:
diff changeset
1245 // get a pointer to it.
anatofuz
parents:
diff changeset
1246 if (void *SymAddr = sys::DynamicLibrary::SearchForAddressOfSymbol(
anatofuz
parents:
diff changeset
1247 std::string(GV.getName())))
anatofuz
parents:
diff changeset
1248 addGlobalMapping(&GV, SymAddr);
anatofuz
parents:
diff changeset
1249 else {
anatofuz
parents:
diff changeset
1250 report_fatal_error("Could not resolve external global address: "
anatofuz
parents:
diff changeset
1251 +GV.getName());
anatofuz
parents:
diff changeset
1252 }
anatofuz
parents:
diff changeset
1253 }
anatofuz
parents:
diff changeset
1254 }
anatofuz
parents:
diff changeset
1255
anatofuz
parents:
diff changeset
1256 // If there are multiple modules, map the non-canonical globals to their
anatofuz
parents:
diff changeset
1257 // canonical location.
anatofuz
parents:
diff changeset
1258 if (!NonCanonicalGlobals.empty()) {
anatofuz
parents:
diff changeset
1259 for (unsigned i = 0, e = NonCanonicalGlobals.size(); i != e; ++i) {
anatofuz
parents:
diff changeset
1260 const GlobalValue *GV = NonCanonicalGlobals[i];
anatofuz
parents:
diff changeset
1261 const GlobalValue *CGV = LinkedGlobalsMap[std::make_pair(
anatofuz
parents:
diff changeset
1262 std::string(GV->getName()), GV->getType())];
anatofuz
parents:
diff changeset
1263 void *Ptr = getPointerToGlobalIfAvailable(CGV);
anatofuz
parents:
diff changeset
1264 assert(Ptr && "Canonical global wasn't codegen'd!");
anatofuz
parents:
diff changeset
1265 addGlobalMapping(GV, Ptr);
anatofuz
parents:
diff changeset
1266 }
anatofuz
parents:
diff changeset
1267 }
anatofuz
parents:
diff changeset
1268
anatofuz
parents:
diff changeset
1269 // Now that all of the globals are set up in memory, loop through them all
anatofuz
parents:
diff changeset
1270 // and initialize their contents.
anatofuz
parents:
diff changeset
1271 for (const auto &GV : M.globals()) {
anatofuz
parents:
diff changeset
1272 if (!GV.isDeclaration()) {
anatofuz
parents:
diff changeset
1273 if (!LinkedGlobalsMap.empty()) {
anatofuz
parents:
diff changeset
1274 if (const GlobalValue *GVEntry = LinkedGlobalsMap[std::make_pair(
anatofuz
parents:
diff changeset
1275 std::string(GV.getName()), GV.getType())])
anatofuz
parents:
diff changeset
1276 if (GVEntry != &GV) // Not the canonical variable.
anatofuz
parents:
diff changeset
1277 continue;
anatofuz
parents:
diff changeset
1278 }
anatofuz
parents:
diff changeset
1279 EmitGlobalVariable(&GV);
anatofuz
parents:
diff changeset
1280 }
anatofuz
parents:
diff changeset
1281 }
anatofuz
parents:
diff changeset
1282 }
anatofuz
parents:
diff changeset
1283 }
anatofuz
parents:
diff changeset
1284
anatofuz
parents:
diff changeset
1285 // EmitGlobalVariable - This method emits the specified global variable to the
anatofuz
parents:
diff changeset
1286 // address specified in GlobalAddresses, or allocates new memory if it's not
anatofuz
parents:
diff changeset
1287 // already in the map.
anatofuz
parents:
diff changeset
1288 void ExecutionEngine::EmitGlobalVariable(const GlobalVariable *GV) {
anatofuz
parents:
diff changeset
1289 void *GA = getPointerToGlobalIfAvailable(GV);
anatofuz
parents:
diff changeset
1290
anatofuz
parents:
diff changeset
1291 if (!GA) {
anatofuz
parents:
diff changeset
1292 // If it's not already specified, allocate memory for the global.
anatofuz
parents:
diff changeset
1293 GA = getMemoryForGV(GV);
anatofuz
parents:
diff changeset
1294
anatofuz
parents:
diff changeset
1295 // If we failed to allocate memory for this global, return.
anatofuz
parents:
diff changeset
1296 if (!GA) return;
anatofuz
parents:
diff changeset
1297
anatofuz
parents:
diff changeset
1298 addGlobalMapping(GV, GA);
anatofuz
parents:
diff changeset
1299 }
anatofuz
parents:
diff changeset
1300
anatofuz
parents:
diff changeset
1301 // Don't initialize if it's thread local, let the client do it.
anatofuz
parents:
diff changeset
1302 if (!GV->isThreadLocal())
anatofuz
parents:
diff changeset
1303 InitializeMemory(GV->getInitializer(), GA);
anatofuz
parents:
diff changeset
1304
anatofuz
parents:
diff changeset
1305 Type *ElTy = GV->getValueType();
anatofuz
parents:
diff changeset
1306 size_t GVSize = (size_t)getDataLayout().getTypeAllocSize(ElTy);
anatofuz
parents:
diff changeset
1307 NumInitBytes += (unsigned)GVSize;
anatofuz
parents:
diff changeset
1308 ++NumGlobals;
anatofuz
parents:
diff changeset
1309 }