1f22ef01cSRoman Divacky //===-- ExecutionEngine.cpp - Common Implementation shared by EEs ---------===//
2f22ef01cSRoman Divacky //
3f22ef01cSRoman Divacky // The LLVM Compiler Infrastructure
4f22ef01cSRoman Divacky //
5f22ef01cSRoman Divacky // This file is distributed under the University of Illinois Open Source
6f22ef01cSRoman Divacky // License. See LICENSE.TXT for details.
7f22ef01cSRoman Divacky //
8f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
9f22ef01cSRoman Divacky //
10f22ef01cSRoman Divacky // This file defines the common interface used by the various execution engine
11f22ef01cSRoman Divacky // subclasses.
12f22ef01cSRoman Divacky //
13f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
14f22ef01cSRoman Divacky
15f22ef01cSRoman Divacky #include "llvm/ExecutionEngine/ExecutionEngine.h"
16ff0cc061SDimitry Andric #include "llvm/ADT/STLExtras.h"
172754fe60SDimitry Andric #include "llvm/ADT/SmallString.h"
18f22ef01cSRoman Divacky #include "llvm/ADT/Statistic.h"
19139f7f9bSDimitry Andric #include "llvm/ExecutionEngine/GenericValue.h"
2039d628a0SDimitry Andric #include "llvm/ExecutionEngine/JITEventListener.h"
21d88c1a5aSDimitry Andric #include "llvm/ExecutionEngine/ObjectCache.h"
22ff0cc061SDimitry Andric #include "llvm/ExecutionEngine/RTDyldMemoryManager.h"
23139f7f9bSDimitry Andric #include "llvm/IR/Constants.h"
24139f7f9bSDimitry Andric #include "llvm/IR/DataLayout.h"
25139f7f9bSDimitry Andric #include "llvm/IR/DerivedTypes.h"
26ff0cc061SDimitry Andric #include "llvm/IR/Mangler.h"
27139f7f9bSDimitry Andric #include "llvm/IR/Module.h"
28139f7f9bSDimitry Andric #include "llvm/IR/Operator.h"
2991bc56edSDimitry Andric #include "llvm/IR/ValueHandle.h"
3039d628a0SDimitry Andric #include "llvm/Object/Archive.h"
3191bc56edSDimitry Andric #include "llvm/Object/ObjectFile.h"
32f22ef01cSRoman Divacky #include "llvm/Support/Debug.h"
33139f7f9bSDimitry Andric #include "llvm/Support/DynamicLibrary.h"
34f22ef01cSRoman Divacky #include "llvm/Support/ErrorHandling.h"
35139f7f9bSDimitry Andric #include "llvm/Support/Host.h"
36f22ef01cSRoman Divacky #include "llvm/Support/MutexGuard.h"
37139f7f9bSDimitry Andric #include "llvm/Support/TargetRegistry.h"
38f22ef01cSRoman Divacky #include "llvm/Support/raw_ostream.h"
39bd5abe19SDimitry Andric #include "llvm/Target/TargetMachine.h"
40f22ef01cSRoman Divacky #include <cmath>
41f22ef01cSRoman Divacky #include <cstring>
42f22ef01cSRoman Divacky using namespace llvm;
43f22ef01cSRoman Divacky
4491bc56edSDimitry Andric #define DEBUG_TYPE "jit"
4591bc56edSDimitry Andric
46f22ef01cSRoman Divacky STATISTIC(NumInitBytes, "Number of bytes of global vars initialized");
47f22ef01cSRoman Divacky STATISTIC(NumGlobals , "Number of global vars initialized");
48f22ef01cSRoman Divacky
492754fe60SDimitry Andric ExecutionEngine *(*ExecutionEngine::MCJITCtor)(
5039d628a0SDimitry Andric std::unique_ptr<Module> M, std::string *ErrorStr,
51ff0cc061SDimitry Andric std::shared_ptr<MCJITMemoryManager> MemMgr,
52*4ba319b5SDimitry Andric std::shared_ptr<LegacyJITSymbolResolver> Resolver,
5339d628a0SDimitry Andric std::unique_ptr<TargetMachine> TM) = nullptr;
54ff0cc061SDimitry Andric
55ff0cc061SDimitry Andric ExecutionEngine *(*ExecutionEngine::OrcMCJITReplacementCtor)(
56ff0cc061SDimitry Andric std::string *ErrorStr, std::shared_ptr<MCJITMemoryManager> MemMgr,
57*4ba319b5SDimitry Andric std::shared_ptr<LegacyJITSymbolResolver> Resolver,
58ff0cc061SDimitry Andric std::unique_ptr<TargetMachine> TM) = nullptr;
59ff0cc061SDimitry Andric
6039d628a0SDimitry Andric ExecutionEngine *(*ExecutionEngine::InterpCtor)(std::unique_ptr<Module> M,
6191bc56edSDimitry Andric std::string *ErrorStr) =nullptr;
62f22ef01cSRoman Divacky
anchor()6339d628a0SDimitry Andric void JITEventListener::anchor() {}
6439d628a0SDimitry Andric
anchor()65d88c1a5aSDimitry Andric void ObjectCache::anchor() {}
66d88c1a5aSDimitry Andric
Init(std::unique_ptr<Module> M)677d523365SDimitry Andric void ExecutionEngine::Init(std::unique_ptr<Module> M) {
68f22ef01cSRoman Divacky CompilingLazily = false;
69f22ef01cSRoman Divacky GVCompilationDisabled = false;
70f22ef01cSRoman Divacky SymbolSearchingDisabled = false;
7191bc56edSDimitry Andric
7291bc56edSDimitry Andric // IR module verification is enabled by default in debug builds, and disabled
7391bc56edSDimitry Andric // by default in release builds.
7491bc56edSDimitry Andric #ifndef NDEBUG
7591bc56edSDimitry Andric VerifyModules = true;
7691bc56edSDimitry Andric #else
7791bc56edSDimitry Andric VerifyModules = false;
7891bc56edSDimitry Andric #endif
7991bc56edSDimitry Andric
80f22ef01cSRoman Divacky assert(M && "Module is null?");
8139d628a0SDimitry Andric Modules.push_back(std::move(M));
82f22ef01cSRoman Divacky }
83f22ef01cSRoman Divacky
ExecutionEngine(std::unique_ptr<Module> M)847d523365SDimitry Andric ExecutionEngine::ExecutionEngine(std::unique_ptr<Module> M)
857d523365SDimitry Andric : DL(M->getDataLayout()), LazyFunctionCreator(nullptr) {
867d523365SDimitry Andric Init(std::move(M));
877d523365SDimitry Andric }
887d523365SDimitry Andric
ExecutionEngine(DataLayout DL,std::unique_ptr<Module> M)897d523365SDimitry Andric ExecutionEngine::ExecutionEngine(DataLayout DL, std::unique_ptr<Module> M)
907d523365SDimitry Andric : DL(std::move(DL)), LazyFunctionCreator(nullptr) {
917d523365SDimitry Andric Init(std::move(M));
927d523365SDimitry Andric }
937d523365SDimitry Andric
~ExecutionEngine()94f22ef01cSRoman Divacky ExecutionEngine::~ExecutionEngine() {
95f22ef01cSRoman Divacky clearAllGlobalMappings();
96f22ef01cSRoman Divacky }
97f22ef01cSRoman Divacky
98f22ef01cSRoman Divacky namespace {
99*4ba319b5SDimitry Andric /// Helper class which uses a value handler to automatically deletes the
1002754fe60SDimitry Andric /// memory block when the GlobalVariable is destroyed.
1017d523365SDimitry Andric class GVMemoryBlock final : public CallbackVH {
GVMemoryBlock(const GlobalVariable * GV)102f22ef01cSRoman Divacky GVMemoryBlock(const GlobalVariable *GV)
103f22ef01cSRoman Divacky : CallbackVH(const_cast<GlobalVariable*>(GV)) {}
104f22ef01cSRoman Divacky
105f22ef01cSRoman Divacky public:
106*4ba319b5SDimitry Andric /// Returns the address the GlobalVariable should be written into. The
1072754fe60SDimitry Andric /// GVMemoryBlock object prefixes that.
Create(const GlobalVariable * GV,const DataLayout & TD)1083861d79fSDimitry Andric static char *Create(const GlobalVariable *GV, const DataLayout& TD) {
1093ca95b02SDimitry Andric Type *ElTy = GV->getValueType();
110f22ef01cSRoman Divacky size_t GVSize = (size_t)TD.getTypeAllocSize(ElTy);
111f22ef01cSRoman Divacky void *RawMemory = ::operator new(
1123ca95b02SDimitry Andric alignTo(sizeof(GVMemoryBlock), TD.getPreferredAlignment(GV)) + GVSize);
113f22ef01cSRoman Divacky new(RawMemory) GVMemoryBlock(GV);
114f22ef01cSRoman Divacky return static_cast<char*>(RawMemory) + sizeof(GVMemoryBlock);
115f22ef01cSRoman Divacky }
116f22ef01cSRoman Divacky
deleted()11791bc56edSDimitry Andric void deleted() override {
118f22ef01cSRoman Divacky // We allocated with operator new and with some extra memory hanging off the
119f22ef01cSRoman Divacky // end, so don't just delete this. I'm not sure if this is actually
120f22ef01cSRoman Divacky // required.
121f22ef01cSRoman Divacky this->~GVMemoryBlock();
122f22ef01cSRoman Divacky ::operator delete(this);
123f22ef01cSRoman Divacky }
124f22ef01cSRoman Divacky };
125f22ef01cSRoman Divacky } // anonymous namespace
126f22ef01cSRoman Divacky
getMemoryForGV(const GlobalVariable * GV)127f22ef01cSRoman Divacky char *ExecutionEngine::getMemoryForGV(const GlobalVariable *GV) {
1287d523365SDimitry Andric return GVMemoryBlock::Create(GV, getDataLayout());
129f22ef01cSRoman Divacky }
130f22ef01cSRoman Divacky
addObjectFile(std::unique_ptr<object::ObjectFile> O)13191bc56edSDimitry Andric void ExecutionEngine::addObjectFile(std::unique_ptr<object::ObjectFile> O) {
13291bc56edSDimitry Andric llvm_unreachable("ExecutionEngine subclass doesn't implement addObjectFile.");
13391bc56edSDimitry Andric }
13491bc56edSDimitry Andric
13539d628a0SDimitry Andric void
addObjectFile(object::OwningBinary<object::ObjectFile> O)13639d628a0SDimitry Andric ExecutionEngine::addObjectFile(object::OwningBinary<object::ObjectFile> O) {
13739d628a0SDimitry Andric llvm_unreachable("ExecutionEngine subclass doesn't implement addObjectFile.");
13839d628a0SDimitry Andric }
13939d628a0SDimitry Andric
addArchive(object::OwningBinary<object::Archive> A)14039d628a0SDimitry Andric void ExecutionEngine::addArchive(object::OwningBinary<object::Archive> A) {
14139d628a0SDimitry Andric llvm_unreachable("ExecutionEngine subclass doesn't implement addArchive.");
14239d628a0SDimitry Andric }
14339d628a0SDimitry Andric
removeModule(Module * M)144f22ef01cSRoman Divacky bool ExecutionEngine::removeModule(Module *M) {
14539d628a0SDimitry Andric for (auto I = Modules.begin(), E = Modules.end(); I != E; ++I) {
14639d628a0SDimitry Andric Module *Found = I->get();
147f22ef01cSRoman Divacky if (Found == M) {
14839d628a0SDimitry Andric I->release();
149f22ef01cSRoman Divacky Modules.erase(I);
150f22ef01cSRoman Divacky clearGlobalMappingsFromModule(M);
151f22ef01cSRoman Divacky return true;
152f22ef01cSRoman Divacky }
153f22ef01cSRoman Divacky }
154f22ef01cSRoman Divacky return false;
155f22ef01cSRoman Divacky }
156f22ef01cSRoman Divacky
FindFunctionNamed(StringRef FnName)157d88c1a5aSDimitry Andric Function *ExecutionEngine::FindFunctionNamed(StringRef FnName) {
158f22ef01cSRoman Divacky for (unsigned i = 0, e = Modules.size(); i != e; ++i) {
159ff0cc061SDimitry Andric Function *F = Modules[i]->getFunction(FnName);
160ff0cc061SDimitry Andric if (F && !F->isDeclaration())
161f22ef01cSRoman Divacky return F;
162f22ef01cSRoman Divacky }
16391bc56edSDimitry Andric return nullptr;
164f22ef01cSRoman Divacky }
165f22ef01cSRoman Divacky
FindGlobalVariableNamed(StringRef Name,bool AllowInternal)166d88c1a5aSDimitry Andric GlobalVariable *ExecutionEngine::FindGlobalVariableNamed(StringRef Name, bool AllowInternal) {
1678f0fd8f6SDimitry Andric for (unsigned i = 0, e = Modules.size(); i != e; ++i) {
1688f0fd8f6SDimitry Andric GlobalVariable *GV = Modules[i]->getGlobalVariable(Name,AllowInternal);
1698f0fd8f6SDimitry Andric if (GV && !GV->isDeclaration())
1708f0fd8f6SDimitry Andric return GV;
1718f0fd8f6SDimitry Andric }
1728f0fd8f6SDimitry Andric return nullptr;
1738f0fd8f6SDimitry Andric }
174f22ef01cSRoman Divacky
RemoveMapping(StringRef Name)175ff0cc061SDimitry Andric uint64_t ExecutionEngineState::RemoveMapping(StringRef Name) {
176ff0cc061SDimitry Andric GlobalAddressMapTy::iterator I = GlobalAddressMap.find(Name);
177ff0cc061SDimitry Andric uint64_t OldVal;
1782754fe60SDimitry Andric
1792754fe60SDimitry Andric // FIXME: This is silly, we shouldn't end up with a mapping -> 0 in the
1802754fe60SDimitry Andric // GlobalAddressMap.
181f22ef01cSRoman Divacky if (I == GlobalAddressMap.end())
182ff0cc061SDimitry Andric OldVal = 0;
183f22ef01cSRoman Divacky else {
184ff0cc061SDimitry Andric GlobalAddressReverseMap.erase(I->second);
185f22ef01cSRoman Divacky OldVal = I->second;
186f22ef01cSRoman Divacky GlobalAddressMap.erase(I);
187f22ef01cSRoman Divacky }
188f22ef01cSRoman Divacky
189f22ef01cSRoman Divacky return OldVal;
190f22ef01cSRoman Divacky }
191f22ef01cSRoman Divacky
getMangledName(const GlobalValue * GV)192ff0cc061SDimitry Andric std::string ExecutionEngine::getMangledName(const GlobalValue *GV) {
193b6c25e0eSDimitry Andric assert(GV->hasName() && "Global must have name.");
194b6c25e0eSDimitry Andric
195ff0cc061SDimitry Andric MutexGuard locked(lock);
196ff0cc061SDimitry Andric SmallString<128> FullName;
197b6c25e0eSDimitry Andric
198b6c25e0eSDimitry Andric const DataLayout &DL =
199b6c25e0eSDimitry Andric GV->getParent()->getDataLayout().isDefault()
2007d523365SDimitry Andric ? getDataLayout()
201b6c25e0eSDimitry Andric : GV->getParent()->getDataLayout();
202b6c25e0eSDimitry Andric
203b6c25e0eSDimitry Andric Mangler::getNameWithPrefix(FullName, GV->getName(), DL);
204ff0cc061SDimitry Andric return FullName.str();
205ff0cc061SDimitry Andric }
206ff0cc061SDimitry Andric
addGlobalMapping(const GlobalValue * GV,void * Addr)207f22ef01cSRoman Divacky void ExecutionEngine::addGlobalMapping(const GlobalValue *GV, void *Addr) {
208f22ef01cSRoman Divacky MutexGuard locked(lock);
209ff0cc061SDimitry Andric addGlobalMapping(getMangledName(GV), (uint64_t) Addr);
210ff0cc061SDimitry Andric }
211f22ef01cSRoman Divacky
addGlobalMapping(StringRef Name,uint64_t Addr)212ff0cc061SDimitry Andric void ExecutionEngine::addGlobalMapping(StringRef Name, uint64_t Addr) {
213ff0cc061SDimitry Andric MutexGuard locked(lock);
214ff0cc061SDimitry Andric
215ff0cc061SDimitry Andric assert(!Name.empty() && "Empty GlobalMapping symbol name!");
216ff0cc061SDimitry Andric
217*4ba319b5SDimitry Andric LLVM_DEBUG(dbgs() << "JIT: Map \'" << Name << "\' to [" << Addr << "]\n";);
218ff0cc061SDimitry Andric uint64_t &CurVal = EEState.getGlobalAddressMap()[Name];
21991bc56edSDimitry Andric assert((!CurVal || !Addr) && "GlobalMapping already established!");
220f22ef01cSRoman Divacky CurVal = Addr;
221f22ef01cSRoman Divacky
2222754fe60SDimitry Andric // If we are using the reverse mapping, add it too.
22391bc56edSDimitry Andric if (!EEState.getGlobalAddressReverseMap().empty()) {
224ff0cc061SDimitry Andric std::string &V = EEState.getGlobalAddressReverseMap()[CurVal];
225ff0cc061SDimitry Andric assert((!V.empty() || !Name.empty()) &&
226ff0cc061SDimitry Andric "GlobalMapping already established!");
227ff0cc061SDimitry Andric V = Name;
228f22ef01cSRoman Divacky }
229f22ef01cSRoman Divacky }
230f22ef01cSRoman Divacky
clearAllGlobalMappings()231f22ef01cSRoman Divacky void ExecutionEngine::clearAllGlobalMappings() {
232f22ef01cSRoman Divacky MutexGuard locked(lock);
233f22ef01cSRoman Divacky
23491bc56edSDimitry Andric EEState.getGlobalAddressMap().clear();
23591bc56edSDimitry Andric EEState.getGlobalAddressReverseMap().clear();
236f22ef01cSRoman Divacky }
237f22ef01cSRoman Divacky
clearGlobalMappingsFromModule(Module * M)238f22ef01cSRoman Divacky void ExecutionEngine::clearGlobalMappingsFromModule(Module *M) {
239f22ef01cSRoman Divacky MutexGuard locked(lock);
240f22ef01cSRoman Divacky
2413ca95b02SDimitry Andric for (GlobalObject &GO : M->global_objects())
2423ca95b02SDimitry Andric EEState.RemoveMapping(getMangledName(&GO));
243f22ef01cSRoman Divacky }
244f22ef01cSRoman Divacky
updateGlobalMapping(const GlobalValue * GV,void * Addr)245ff0cc061SDimitry Andric uint64_t ExecutionEngine::updateGlobalMapping(const GlobalValue *GV,
246ff0cc061SDimitry Andric void *Addr) {
247ff0cc061SDimitry Andric MutexGuard locked(lock);
248ff0cc061SDimitry Andric return updateGlobalMapping(getMangledName(GV), (uint64_t) Addr);
249ff0cc061SDimitry Andric }
250ff0cc061SDimitry Andric
updateGlobalMapping(StringRef Name,uint64_t Addr)251ff0cc061SDimitry Andric uint64_t ExecutionEngine::updateGlobalMapping(StringRef Name, uint64_t Addr) {
252f22ef01cSRoman Divacky MutexGuard locked(lock);
253f22ef01cSRoman Divacky
254f22ef01cSRoman Divacky ExecutionEngineState::GlobalAddressMapTy &Map =
25591bc56edSDimitry Andric EEState.getGlobalAddressMap();
256f22ef01cSRoman Divacky
257f22ef01cSRoman Divacky // Deleting from the mapping?
25891bc56edSDimitry Andric if (!Addr)
259ff0cc061SDimitry Andric return EEState.RemoveMapping(Name);
260f22ef01cSRoman Divacky
261ff0cc061SDimitry Andric uint64_t &CurVal = Map[Name];
262ff0cc061SDimitry Andric uint64_t OldVal = CurVal;
263f22ef01cSRoman Divacky
26491bc56edSDimitry Andric if (CurVal && !EEState.getGlobalAddressReverseMap().empty())
26591bc56edSDimitry Andric EEState.getGlobalAddressReverseMap().erase(CurVal);
266f22ef01cSRoman Divacky CurVal = Addr;
267f22ef01cSRoman Divacky
2682754fe60SDimitry Andric // If we are using the reverse mapping, add it too.
26991bc56edSDimitry Andric if (!EEState.getGlobalAddressReverseMap().empty()) {
270ff0cc061SDimitry Andric std::string &V = EEState.getGlobalAddressReverseMap()[CurVal];
271ff0cc061SDimitry Andric assert((!V.empty() || !Name.empty()) &&
272ff0cc061SDimitry Andric "GlobalMapping already established!");
273ff0cc061SDimitry Andric V = Name;
274f22ef01cSRoman Divacky }
275f22ef01cSRoman Divacky return OldVal;
276f22ef01cSRoman Divacky }
277f22ef01cSRoman Divacky
getAddressToGlobalIfAvailable(StringRef S)278ff0cc061SDimitry Andric uint64_t ExecutionEngine::getAddressToGlobalIfAvailable(StringRef S) {
279ff0cc061SDimitry Andric MutexGuard locked(lock);
280ff0cc061SDimitry Andric uint64_t Address = 0;
281ff0cc061SDimitry Andric ExecutionEngineState::GlobalAddressMapTy::iterator I =
282ff0cc061SDimitry Andric EEState.getGlobalAddressMap().find(S);
283ff0cc061SDimitry Andric if (I != EEState.getGlobalAddressMap().end())
284ff0cc061SDimitry Andric Address = I->second;
285ff0cc061SDimitry Andric return Address;
286ff0cc061SDimitry Andric }
287ff0cc061SDimitry Andric
288ff0cc061SDimitry Andric
getPointerToGlobalIfAvailable(StringRef S)289ff0cc061SDimitry Andric void *ExecutionEngine::getPointerToGlobalIfAvailable(StringRef S) {
290ff0cc061SDimitry Andric MutexGuard locked(lock);
291ff0cc061SDimitry Andric if (void* Address = (void *) getAddressToGlobalIfAvailable(S))
292ff0cc061SDimitry Andric return Address;
293ff0cc061SDimitry Andric return nullptr;
294ff0cc061SDimitry Andric }
295ff0cc061SDimitry Andric
getPointerToGlobalIfAvailable(const GlobalValue * GV)296f22ef01cSRoman Divacky void *ExecutionEngine::getPointerToGlobalIfAvailable(const GlobalValue *GV) {
297f22ef01cSRoman Divacky MutexGuard locked(lock);
298ff0cc061SDimitry Andric return getPointerToGlobalIfAvailable(getMangledName(GV));
299f22ef01cSRoman Divacky }
300f22ef01cSRoman Divacky
getGlobalValueAtAddress(void * Addr)301f22ef01cSRoman Divacky const GlobalValue *ExecutionEngine::getGlobalValueAtAddress(void *Addr) {
302f22ef01cSRoman Divacky MutexGuard locked(lock);
303f22ef01cSRoman Divacky
304f22ef01cSRoman Divacky // If we haven't computed the reverse mapping yet, do so first.
30591bc56edSDimitry Andric if (EEState.getGlobalAddressReverseMap().empty()) {
306f22ef01cSRoman Divacky for (ExecutionEngineState::GlobalAddressMapTy::iterator
30791bc56edSDimitry Andric I = EEState.getGlobalAddressMap().begin(),
308ff0cc061SDimitry Andric E = EEState.getGlobalAddressMap().end(); I != E; ++I) {
309ff0cc061SDimitry Andric StringRef Name = I->first();
310ff0cc061SDimitry Andric uint64_t Addr = I->second;
31191bc56edSDimitry Andric EEState.getGlobalAddressReverseMap().insert(std::make_pair(
312ff0cc061SDimitry Andric Addr, Name));
313ff0cc061SDimitry Andric }
314f22ef01cSRoman Divacky }
315f22ef01cSRoman Divacky
316ff0cc061SDimitry Andric std::map<uint64_t, std::string>::iterator I =
317ff0cc061SDimitry Andric EEState.getGlobalAddressReverseMap().find((uint64_t) Addr);
318ff0cc061SDimitry Andric
319ff0cc061SDimitry Andric if (I != EEState.getGlobalAddressReverseMap().end()) {
320ff0cc061SDimitry Andric StringRef Name = I->second;
321ff0cc061SDimitry Andric for (unsigned i = 0, e = Modules.size(); i != e; ++i)
322ff0cc061SDimitry Andric if (GlobalValue *GV = Modules[i]->getNamedValue(Name))
323ff0cc061SDimitry Andric return GV;
324ff0cc061SDimitry Andric }
325ff0cc061SDimitry Andric return nullptr;
326f22ef01cSRoman Divacky }
327f22ef01cSRoman Divacky
328f22ef01cSRoman Divacky namespace {
329f22ef01cSRoman Divacky class ArgvArray {
33039d628a0SDimitry Andric std::unique_ptr<char[]> Array;
33139d628a0SDimitry Andric std::vector<std::unique_ptr<char[]>> Values;
332f22ef01cSRoman Divacky public:
333f22ef01cSRoman Divacky /// Turn a vector of strings into a nice argv style array of pointers to null
334f22ef01cSRoman Divacky /// terminated strings.
335f22ef01cSRoman Divacky void *reset(LLVMContext &C, ExecutionEngine *EE,
336f22ef01cSRoman Divacky const std::vector<std::string> &InputArgv);
337f22ef01cSRoman Divacky };
338f22ef01cSRoman Divacky } // anonymous namespace
reset(LLVMContext & C,ExecutionEngine * EE,const std::vector<std::string> & InputArgv)339f22ef01cSRoman Divacky void *ArgvArray::reset(LLVMContext &C, ExecutionEngine *EE,
340f22ef01cSRoman Divacky const std::vector<std::string> &InputArgv) {
34139d628a0SDimitry Andric Values.clear(); // Free the old contents.
34239d628a0SDimitry Andric Values.reserve(InputArgv.size());
3437d523365SDimitry Andric unsigned PtrSize = EE->getDataLayout().getPointerSize();
34439d628a0SDimitry Andric Array = make_unique<char[]>((InputArgv.size()+1)*PtrSize);
345f22ef01cSRoman Divacky
346*4ba319b5SDimitry Andric LLVM_DEBUG(dbgs() << "JIT: ARGV = " << (void *)Array.get() << "\n");
3476122f3e6SDimitry Andric Type *SBytePtr = Type::getInt8PtrTy(C);
348f22ef01cSRoman Divacky
349f22ef01cSRoman Divacky for (unsigned i = 0; i != InputArgv.size(); ++i) {
350f22ef01cSRoman Divacky unsigned Size = InputArgv[i].size()+1;
35139d628a0SDimitry Andric auto Dest = make_unique<char[]>(Size);
352*4ba319b5SDimitry Andric LLVM_DEBUG(dbgs() << "JIT: ARGV[" << i << "] = " << (void *)Dest.get()
353*4ba319b5SDimitry Andric << "\n");
354f22ef01cSRoman Divacky
35539d628a0SDimitry Andric std::copy(InputArgv[i].begin(), InputArgv[i].end(), Dest.get());
356f22ef01cSRoman Divacky Dest[Size-1] = 0;
357f22ef01cSRoman Divacky
358f22ef01cSRoman Divacky // Endian safe: Array[i] = (PointerTy)Dest;
35939d628a0SDimitry Andric EE->StoreValueToMemory(PTOGV(Dest.get()),
36039d628a0SDimitry Andric (GenericValue*)(&Array[i*PtrSize]), SBytePtr);
36139d628a0SDimitry Andric Values.push_back(std::move(Dest));
362f22ef01cSRoman Divacky }
363f22ef01cSRoman Divacky
364f22ef01cSRoman Divacky // Null terminate it
36591bc56edSDimitry Andric EE->StoreValueToMemory(PTOGV(nullptr),
36639d628a0SDimitry Andric (GenericValue*)(&Array[InputArgv.size()*PtrSize]),
367f22ef01cSRoman Divacky SBytePtr);
36839d628a0SDimitry Andric return Array.get();
369f22ef01cSRoman Divacky }
370f22ef01cSRoman Divacky
runStaticConstructorsDestructors(Module & module,bool isDtors)37139d628a0SDimitry Andric void ExecutionEngine::runStaticConstructorsDestructors(Module &module,
372f22ef01cSRoman Divacky bool isDtors) {
373d88c1a5aSDimitry Andric StringRef Name(isDtors ? "llvm.global_dtors" : "llvm.global_ctors");
37439d628a0SDimitry Andric GlobalVariable *GV = module.getNamedGlobal(Name);
375f22ef01cSRoman Divacky
376f22ef01cSRoman Divacky // If this global has internal linkage, or if it has a use, then it must be
377f22ef01cSRoman Divacky // an old-style (llvmgcc3) static ctor with __main linked in and in use. If
378f22ef01cSRoman Divacky // this is the case, don't execute any of the global ctors, __main will do
379f22ef01cSRoman Divacky // it.
380f22ef01cSRoman Divacky if (!GV || GV->isDeclaration() || GV->hasLocalLinkage()) return;
381f22ef01cSRoman Divacky
3823b0f4066SDimitry Andric // Should be an array of '{ i32, void ()* }' structs. The first value is
383f22ef01cSRoman Divacky // the init priority, which we ignore.
384dff0c46cSDimitry Andric ConstantArray *InitList = dyn_cast<ConstantArray>(GV->getInitializer());
38591bc56edSDimitry Andric if (!InitList)
3863b0f4066SDimitry Andric return;
3872754fe60SDimitry Andric for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i) {
388dff0c46cSDimitry Andric ConstantStruct *CS = dyn_cast<ConstantStruct>(InitList->getOperand(i));
38991bc56edSDimitry Andric if (!CS) continue;
390f22ef01cSRoman Divacky
391f22ef01cSRoman Divacky Constant *FP = CS->getOperand(1);
392f22ef01cSRoman Divacky if (FP->isNullValue())
3933b0f4066SDimitry Andric continue; // Found a sentinal value, ignore.
394f22ef01cSRoman Divacky
3952754fe60SDimitry Andric // Strip off constant expression casts.
396f22ef01cSRoman Divacky if (ConstantExpr *CE = dyn_cast<ConstantExpr>(FP))
397f22ef01cSRoman Divacky if (CE->isCast())
398f22ef01cSRoman Divacky FP = CE->getOperand(0);
3992754fe60SDimitry Andric
400f22ef01cSRoman Divacky // Execute the ctor/dtor function!
4012754fe60SDimitry Andric if (Function *F = dyn_cast<Function>(FP))
4028f0fd8f6SDimitry Andric runFunction(F, None);
4032754fe60SDimitry Andric
4042754fe60SDimitry Andric // FIXME: It is marginally lame that we just do nothing here if we see an
4052754fe60SDimitry Andric // entry we don't recognize. It might not be unreasonable for the verifier
4062754fe60SDimitry Andric // to not even allow this and just assert here.
407f22ef01cSRoman Divacky }
408f22ef01cSRoman Divacky }
409f22ef01cSRoman Divacky
runStaticConstructorsDestructors(bool isDtors)410f22ef01cSRoman Divacky void ExecutionEngine::runStaticConstructorsDestructors(bool isDtors) {
411f22ef01cSRoman Divacky // Execute global ctors/dtors for each module in the program.
41239d628a0SDimitry Andric for (std::unique_ptr<Module> &M : Modules)
41339d628a0SDimitry Andric runStaticConstructorsDestructors(*M, isDtors);
414f22ef01cSRoman Divacky }
415f22ef01cSRoman Divacky
416f22ef01cSRoman Divacky #ifndef NDEBUG
417f22ef01cSRoman Divacky /// isTargetNullPtr - Return whether the target pointer stored at Loc is null.
isTargetNullPtr(ExecutionEngine * EE,void * Loc)418f22ef01cSRoman Divacky static bool isTargetNullPtr(ExecutionEngine *EE, void *Loc) {
4197d523365SDimitry Andric unsigned PtrSize = EE->getDataLayout().getPointerSize();
420f22ef01cSRoman Divacky for (unsigned i = 0; i < PtrSize; ++i)
421f22ef01cSRoman Divacky if (*(i + (uint8_t*)Loc))
422f22ef01cSRoman Divacky return false;
423f22ef01cSRoman Divacky return true;
424f22ef01cSRoman Divacky }
425f22ef01cSRoman Divacky #endif
426f22ef01cSRoman Divacky
runFunctionAsMain(Function * Fn,const std::vector<std::string> & argv,const char * const * envp)427f22ef01cSRoman Divacky int ExecutionEngine::runFunctionAsMain(Function *Fn,
428f22ef01cSRoman Divacky const std::vector<std::string> &argv,
429f22ef01cSRoman Divacky const char * const * envp) {
430f22ef01cSRoman Divacky std::vector<GenericValue> GVArgs;
431f22ef01cSRoman Divacky GenericValue GVArgc;
432f22ef01cSRoman Divacky GVArgc.IntVal = APInt(32, argv.size());
433f22ef01cSRoman Divacky
434f22ef01cSRoman Divacky // Check main() type
435f22ef01cSRoman Divacky unsigned NumArgs = Fn->getFunctionType()->getNumParams();
4366122f3e6SDimitry Andric FunctionType *FTy = Fn->getFunctionType();
4376122f3e6SDimitry Andric Type* PPInt8Ty = Type::getInt8PtrTy(Fn->getContext())->getPointerTo();
4382754fe60SDimitry Andric
4392754fe60SDimitry Andric // Check the argument types.
4402754fe60SDimitry Andric if (NumArgs > 3)
441f22ef01cSRoman Divacky report_fatal_error("Invalid number of arguments of main() supplied");
4422754fe60SDimitry Andric if (NumArgs >= 3 && FTy->getParamType(2) != PPInt8Ty)
4432754fe60SDimitry Andric report_fatal_error("Invalid type for third argument of main() supplied");
4442754fe60SDimitry Andric if (NumArgs >= 2 && FTy->getParamType(1) != PPInt8Ty)
4452754fe60SDimitry Andric report_fatal_error("Invalid type for second argument of main() supplied");
4462754fe60SDimitry Andric if (NumArgs >= 1 && !FTy->getParamType(0)->isIntegerTy(32))
4472754fe60SDimitry Andric report_fatal_error("Invalid type for first argument of main() supplied");
4482754fe60SDimitry Andric if (!FTy->getReturnType()->isIntegerTy() &&
4492754fe60SDimitry Andric !FTy->getReturnType()->isVoidTy())
4502754fe60SDimitry Andric report_fatal_error("Invalid return type of main() supplied");
451f22ef01cSRoman Divacky
452f22ef01cSRoman Divacky ArgvArray CArgv;
453f22ef01cSRoman Divacky ArgvArray CEnv;
454f22ef01cSRoman Divacky if (NumArgs) {
455f22ef01cSRoman Divacky GVArgs.push_back(GVArgc); // Arg #0 = argc.
456f22ef01cSRoman Divacky if (NumArgs > 1) {
457f22ef01cSRoman Divacky // Arg #1 = argv.
458f22ef01cSRoman Divacky GVArgs.push_back(PTOGV(CArgv.reset(Fn->getContext(), this, argv)));
459f22ef01cSRoman Divacky assert(!isTargetNullPtr(this, GVTOP(GVArgs[1])) &&
460f22ef01cSRoman Divacky "argv[0] was null after CreateArgv");
461f22ef01cSRoman Divacky if (NumArgs > 2) {
462f22ef01cSRoman Divacky std::vector<std::string> EnvVars;
463f22ef01cSRoman Divacky for (unsigned i = 0; envp[i]; ++i)
46497bc6c73SDimitry Andric EnvVars.emplace_back(envp[i]);
465f22ef01cSRoman Divacky // Arg #2 = envp.
466f22ef01cSRoman Divacky GVArgs.push_back(PTOGV(CEnv.reset(Fn->getContext(), this, EnvVars)));
467f22ef01cSRoman Divacky }
468f22ef01cSRoman Divacky }
469f22ef01cSRoman Divacky }
4702754fe60SDimitry Andric
471f22ef01cSRoman Divacky return runFunction(Fn, GVArgs).IntVal.getZExtValue();
472f22ef01cSRoman Divacky }
473f22ef01cSRoman Divacky
EngineBuilder()474ff0cc061SDimitry Andric EngineBuilder::EngineBuilder() : EngineBuilder(nullptr) {}
475ff0cc061SDimitry Andric
EngineBuilder(std::unique_ptr<Module> M)47639d628a0SDimitry Andric EngineBuilder::EngineBuilder(std::unique_ptr<Module> M)
477ff0cc061SDimitry Andric : M(std::move(M)), WhichEngine(EngineKind::Either), ErrorStr(nullptr),
478ff0cc061SDimitry Andric OptLevel(CodeGenOpt::Default), MemMgr(nullptr), Resolver(nullptr),
4792cab237bSDimitry Andric UseOrcMCJITReplacement(false) {
48091bc56edSDimitry Andric // IR module verification is enabled by default in debug builds, and disabled
48191bc56edSDimitry Andric // by default in release builds.
48291bc56edSDimitry Andric #ifndef NDEBUG
48391bc56edSDimitry Andric VerifyModules = true;
48491bc56edSDimitry Andric #else
48591bc56edSDimitry Andric VerifyModules = false;
48691bc56edSDimitry Andric #endif
48791bc56edSDimitry Andric }
48891bc56edSDimitry Andric
489ff0cc061SDimitry Andric EngineBuilder::~EngineBuilder() = default;
490ff0cc061SDimitry Andric
setMCJITMemoryManager(std::unique_ptr<RTDyldMemoryManager> mcjmm)491ff0cc061SDimitry Andric EngineBuilder &EngineBuilder::setMCJITMemoryManager(
492ff0cc061SDimitry Andric std::unique_ptr<RTDyldMemoryManager> mcjmm) {
493ff0cc061SDimitry Andric auto SharedMM = std::shared_ptr<RTDyldMemoryManager>(std::move(mcjmm));
494ff0cc061SDimitry Andric MemMgr = SharedMM;
495ff0cc061SDimitry Andric Resolver = SharedMM;
496ff0cc061SDimitry Andric return *this;
497ff0cc061SDimitry Andric }
498ff0cc061SDimitry Andric
499ff0cc061SDimitry Andric EngineBuilder&
setMemoryManager(std::unique_ptr<MCJITMemoryManager> MM)500ff0cc061SDimitry Andric EngineBuilder::setMemoryManager(std::unique_ptr<MCJITMemoryManager> MM) {
501ff0cc061SDimitry Andric MemMgr = std::shared_ptr<MCJITMemoryManager>(std::move(MM));
502ff0cc061SDimitry Andric return *this;
503ff0cc061SDimitry Andric }
504ff0cc061SDimitry Andric
505ff0cc061SDimitry Andric EngineBuilder &
setSymbolResolver(std::unique_ptr<LegacyJITSymbolResolver> SR)506*4ba319b5SDimitry Andric EngineBuilder::setSymbolResolver(std::unique_ptr<LegacyJITSymbolResolver> SR) {
507*4ba319b5SDimitry Andric Resolver = std::shared_ptr<LegacyJITSymbolResolver>(std::move(SR));
508ff0cc061SDimitry Andric return *this;
509ff0cc061SDimitry Andric }
510ff0cc061SDimitry Andric
create(TargetMachine * TM)511dff0c46cSDimitry Andric ExecutionEngine *EngineBuilder::create(TargetMachine *TM) {
51291bc56edSDimitry Andric std::unique_ptr<TargetMachine> TheTM(TM); // Take ownership.
513dff0c46cSDimitry Andric
514f22ef01cSRoman Divacky // Make sure we can resolve symbols in the program as well. The zero arg
515f22ef01cSRoman Divacky // to the function tells DynamicLibrary to load the program, not a library.
51691bc56edSDimitry Andric if (sys::DynamicLibrary::LoadLibraryPermanently(nullptr, ErrorStr))
51791bc56edSDimitry Andric return nullptr;
518f22ef01cSRoman Divacky
519f22ef01cSRoman Divacky // If the user specified a memory manager but didn't specify which engine to
520f22ef01cSRoman Divacky // create, we assume they only want the JIT, and we fail if they only want
521f22ef01cSRoman Divacky // the interpreter.
522ff0cc061SDimitry Andric if (MemMgr) {
523f22ef01cSRoman Divacky if (WhichEngine & EngineKind::JIT)
524f22ef01cSRoman Divacky WhichEngine = EngineKind::JIT;
525f22ef01cSRoman Divacky else {
526f22ef01cSRoman Divacky if (ErrorStr)
527f22ef01cSRoman Divacky *ErrorStr = "Cannot create an interpreter with a memory manager.";
52891bc56edSDimitry Andric return nullptr;
529f22ef01cSRoman Divacky }
530f22ef01cSRoman Divacky }
531f22ef01cSRoman Divacky
532f22ef01cSRoman Divacky // Unless the interpreter was explicitly selected or the JIT is not linked,
533f22ef01cSRoman Divacky // try making a JIT.
534dff0c46cSDimitry Andric if ((WhichEngine & EngineKind::JIT) && TheTM) {
535dff0c46cSDimitry Andric if (!TM->getTarget().hasJIT()) {
536dff0c46cSDimitry Andric errs() << "WARNING: This target JIT is not designed for the host"
537dff0c46cSDimitry Andric << " you are running. If bad things happen, please choose"
538dff0c46cSDimitry Andric << " a different -march switch.\n";
539dff0c46cSDimitry Andric }
540dff0c46cSDimitry Andric
54191bc56edSDimitry Andric ExecutionEngine *EE = nullptr;
542ff0cc061SDimitry Andric if (ExecutionEngine::OrcMCJITReplacementCtor && UseOrcMCJITReplacement) {
543ff0cc061SDimitry Andric EE = ExecutionEngine::OrcMCJITReplacementCtor(ErrorStr, std::move(MemMgr),
544ff0cc061SDimitry Andric std::move(Resolver),
54539d628a0SDimitry Andric std::move(TheTM));
546ff0cc061SDimitry Andric EE->addModule(std::move(M));
547ff0cc061SDimitry Andric } else if (ExecutionEngine::MCJITCtor)
548ff0cc061SDimitry Andric EE = ExecutionEngine::MCJITCtor(std::move(M), ErrorStr, std::move(MemMgr),
549ff0cc061SDimitry Andric std::move(Resolver), std::move(TheTM));
550ff0cc061SDimitry Andric
55191bc56edSDimitry Andric if (EE) {
55291bc56edSDimitry Andric EE->setVerifyModules(VerifyModules);
55391bc56edSDimitry Andric return EE;
554f22ef01cSRoman Divacky }
555f22ef01cSRoman Divacky }
556f22ef01cSRoman Divacky
557f22ef01cSRoman Divacky // If we can't make a JIT and we didn't request one specifically, try making
558f22ef01cSRoman Divacky // an interpreter instead.
559f22ef01cSRoman Divacky if (WhichEngine & EngineKind::Interpreter) {
560f22ef01cSRoman Divacky if (ExecutionEngine::InterpCtor)
56139d628a0SDimitry Andric return ExecutionEngine::InterpCtor(std::move(M), ErrorStr);
562f22ef01cSRoman Divacky if (ErrorStr)
563f22ef01cSRoman Divacky *ErrorStr = "Interpreter has not been linked in.";
56491bc56edSDimitry Andric return nullptr;
565f22ef01cSRoman Divacky }
566f22ef01cSRoman Divacky
56739d628a0SDimitry Andric if ((WhichEngine & EngineKind::JIT) && !ExecutionEngine::MCJITCtor) {
568f22ef01cSRoman Divacky if (ErrorStr)
569f22ef01cSRoman Divacky *ErrorStr = "JIT has not been linked in.";
570f22ef01cSRoman Divacky }
5712754fe60SDimitry Andric
57291bc56edSDimitry Andric return nullptr;
573f22ef01cSRoman Divacky }
574f22ef01cSRoman Divacky
getPointerToGlobal(const GlobalValue * GV)575f22ef01cSRoman Divacky void *ExecutionEngine::getPointerToGlobal(const GlobalValue *GV) {
576f22ef01cSRoman Divacky if (Function *F = const_cast<Function*>(dyn_cast<Function>(GV)))
577f22ef01cSRoman Divacky return getPointerToFunction(F);
578f22ef01cSRoman Divacky
579f22ef01cSRoman Divacky MutexGuard locked(lock);
580ff0cc061SDimitry Andric if (void* P = getPointerToGlobalIfAvailable(GV))
5812754fe60SDimitry Andric return P;
582f22ef01cSRoman Divacky
583f22ef01cSRoman Divacky // Global variable might have been added since interpreter started.
584f22ef01cSRoman Divacky if (GlobalVariable *GVar =
585f22ef01cSRoman Divacky const_cast<GlobalVariable *>(dyn_cast<GlobalVariable>(GV)))
586f22ef01cSRoman Divacky EmitGlobalVariable(GVar);
587f22ef01cSRoman Divacky else
588f22ef01cSRoman Divacky llvm_unreachable("Global hasn't had an address allocated yet!");
5892754fe60SDimitry Andric
590ff0cc061SDimitry Andric return getPointerToGlobalIfAvailable(GV);
591f22ef01cSRoman Divacky }
592f22ef01cSRoman Divacky
593*4ba319b5SDimitry Andric /// Converts a Constant* into a GenericValue, including handling of
5942754fe60SDimitry Andric /// ConstantExpr values.
getConstantValue(const Constant * C)595f22ef01cSRoman Divacky GenericValue ExecutionEngine::getConstantValue(const Constant *C) {
596f22ef01cSRoman Divacky // If its undefined, return the garbage.
597f22ef01cSRoman Divacky if (isa<UndefValue>(C)) {
598f22ef01cSRoman Divacky GenericValue Result;
599f22ef01cSRoman Divacky switch (C->getType()->getTypeID()) {
600139f7f9bSDimitry Andric default:
601139f7f9bSDimitry Andric break;
602f22ef01cSRoman Divacky case Type::IntegerTyID:
603f22ef01cSRoman Divacky case Type::X86_FP80TyID:
604f22ef01cSRoman Divacky case Type::FP128TyID:
605f22ef01cSRoman Divacky case Type::PPC_FP128TyID:
606f22ef01cSRoman Divacky // Although the value is undefined, we still have to construct an APInt
607f22ef01cSRoman Divacky // with the correct bit width.
608f22ef01cSRoman Divacky Result.IntVal = APInt(C->getType()->getPrimitiveSizeInBits(), 0);
609f22ef01cSRoman Divacky break;
610f785676fSDimitry Andric case Type::StructTyID: {
611f785676fSDimitry Andric // if the whole struct is 'undef' just reserve memory for the value.
612f785676fSDimitry Andric if(StructType *STy = dyn_cast<StructType>(C->getType())) {
613f785676fSDimitry Andric unsigned int elemNum = STy->getNumElements();
614f785676fSDimitry Andric Result.AggregateVal.resize(elemNum);
615f785676fSDimitry Andric for (unsigned int i = 0; i < elemNum; ++i) {
616f785676fSDimitry Andric Type *ElemTy = STy->getElementType(i);
617f785676fSDimitry Andric if (ElemTy->isIntegerTy())
618f785676fSDimitry Andric Result.AggregateVal[i].IntVal =
619f785676fSDimitry Andric APInt(ElemTy->getPrimitiveSizeInBits(), 0);
620f785676fSDimitry Andric else if (ElemTy->isAggregateType()) {
621f785676fSDimitry Andric const Constant *ElemUndef = UndefValue::get(ElemTy);
622f785676fSDimitry Andric Result.AggregateVal[i] = getConstantValue(ElemUndef);
623f785676fSDimitry Andric }
624f785676fSDimitry Andric }
625f785676fSDimitry Andric }
626f785676fSDimitry Andric }
627f785676fSDimitry Andric break;
628139f7f9bSDimitry Andric case Type::VectorTyID:
629139f7f9bSDimitry Andric // if the whole vector is 'undef' just reserve memory for the value.
6307d523365SDimitry Andric auto* VTy = dyn_cast<VectorType>(C->getType());
6317d523365SDimitry Andric Type *ElemTy = VTy->getElementType();
632139f7f9bSDimitry Andric unsigned int elemNum = VTy->getNumElements();
633139f7f9bSDimitry Andric Result.AggregateVal.resize(elemNum);
634139f7f9bSDimitry Andric if (ElemTy->isIntegerTy())
635139f7f9bSDimitry Andric for (unsigned int i = 0; i < elemNum; ++i)
636139f7f9bSDimitry Andric Result.AggregateVal[i].IntVal =
637139f7f9bSDimitry Andric APInt(ElemTy->getPrimitiveSizeInBits(), 0);
638f22ef01cSRoman Divacky break;
639f22ef01cSRoman Divacky }
640f22ef01cSRoman Divacky return Result;
641f22ef01cSRoman Divacky }
642f22ef01cSRoman Divacky
6432754fe60SDimitry Andric // Otherwise, if the value is a ConstantExpr...
644f22ef01cSRoman Divacky if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) {
645f22ef01cSRoman Divacky Constant *Op0 = CE->getOperand(0);
646f22ef01cSRoman Divacky switch (CE->getOpcode()) {
647f22ef01cSRoman Divacky case Instruction::GetElementPtr: {
648f22ef01cSRoman Divacky // Compute the index
649f22ef01cSRoman Divacky GenericValue Result = getConstantValue(Op0);
6507d523365SDimitry Andric APInt Offset(DL.getPointerSizeInBits(), 0);
6517d523365SDimitry Andric cast<GEPOperator>(CE)->accumulateConstantOffset(DL, Offset);
652f22ef01cSRoman Divacky
653f22ef01cSRoman Divacky char* tmp = (char*) Result.PointerVal;
654139f7f9bSDimitry Andric Result = PTOGV(tmp + Offset.getSExtValue());
655f22ef01cSRoman Divacky return Result;
656f22ef01cSRoman Divacky }
657f22ef01cSRoman Divacky case Instruction::Trunc: {
658f22ef01cSRoman Divacky GenericValue GV = getConstantValue(Op0);
659f22ef01cSRoman Divacky uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
660f22ef01cSRoman Divacky GV.IntVal = GV.IntVal.trunc(BitWidth);
661f22ef01cSRoman Divacky return GV;
662f22ef01cSRoman Divacky }
663f22ef01cSRoman Divacky case Instruction::ZExt: {
664f22ef01cSRoman Divacky GenericValue GV = getConstantValue(Op0);
665f22ef01cSRoman Divacky uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
666f22ef01cSRoman Divacky GV.IntVal = GV.IntVal.zext(BitWidth);
667f22ef01cSRoman Divacky return GV;
668f22ef01cSRoman Divacky }
669f22ef01cSRoman Divacky case Instruction::SExt: {
670f22ef01cSRoman Divacky GenericValue GV = getConstantValue(Op0);
671f22ef01cSRoman Divacky uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
672f22ef01cSRoman Divacky GV.IntVal = GV.IntVal.sext(BitWidth);
673f22ef01cSRoman Divacky return GV;
674f22ef01cSRoman Divacky }
675f22ef01cSRoman Divacky case Instruction::FPTrunc: {
676f22ef01cSRoman Divacky // FIXME long double
677f22ef01cSRoman Divacky GenericValue GV = getConstantValue(Op0);
678f22ef01cSRoman Divacky GV.FloatVal = float(GV.DoubleVal);
679f22ef01cSRoman Divacky return GV;
680f22ef01cSRoman Divacky }
681f22ef01cSRoman Divacky case Instruction::FPExt:{
682f22ef01cSRoman Divacky // FIXME long double
683f22ef01cSRoman Divacky GenericValue GV = getConstantValue(Op0);
684f22ef01cSRoman Divacky GV.DoubleVal = double(GV.FloatVal);
685f22ef01cSRoman Divacky return GV;
686f22ef01cSRoman Divacky }
687f22ef01cSRoman Divacky case Instruction::UIToFP: {
688f22ef01cSRoman Divacky GenericValue GV = getConstantValue(Op0);
689f22ef01cSRoman Divacky if (CE->getType()->isFloatTy())
690f22ef01cSRoman Divacky GV.FloatVal = float(GV.IntVal.roundToDouble());
691f22ef01cSRoman Divacky else if (CE->getType()->isDoubleTy())
692f22ef01cSRoman Divacky GV.DoubleVal = GV.IntVal.roundToDouble();
693f22ef01cSRoman Divacky else if (CE->getType()->isX86_FP80Ty()) {
694d88c1a5aSDimitry Andric APFloat apf = APFloat::getZero(APFloat::x87DoubleExtended());
695f22ef01cSRoman Divacky (void)apf.convertFromAPInt(GV.IntVal,
696f22ef01cSRoman Divacky false,
697f22ef01cSRoman Divacky APFloat::rmNearestTiesToEven);
698f22ef01cSRoman Divacky GV.IntVal = apf.bitcastToAPInt();
699f22ef01cSRoman Divacky }
700f22ef01cSRoman Divacky return GV;
701f22ef01cSRoman Divacky }
702f22ef01cSRoman Divacky case Instruction::SIToFP: {
703f22ef01cSRoman Divacky GenericValue GV = getConstantValue(Op0);
704f22ef01cSRoman Divacky if (CE->getType()->isFloatTy())
705f22ef01cSRoman Divacky GV.FloatVal = float(GV.IntVal.signedRoundToDouble());
706f22ef01cSRoman Divacky else if (CE->getType()->isDoubleTy())
707f22ef01cSRoman Divacky GV.DoubleVal = GV.IntVal.signedRoundToDouble();
708f22ef01cSRoman Divacky else if (CE->getType()->isX86_FP80Ty()) {
709d88c1a5aSDimitry Andric APFloat apf = APFloat::getZero(APFloat::x87DoubleExtended());
710f22ef01cSRoman Divacky (void)apf.convertFromAPInt(GV.IntVal,
711f22ef01cSRoman Divacky true,
712f22ef01cSRoman Divacky APFloat::rmNearestTiesToEven);
713f22ef01cSRoman Divacky GV.IntVal = apf.bitcastToAPInt();
714f22ef01cSRoman Divacky }
715f22ef01cSRoman Divacky return GV;
716f22ef01cSRoman Divacky }
717f22ef01cSRoman Divacky case Instruction::FPToUI: // double->APInt conversion handles sign
718f22ef01cSRoman Divacky case Instruction::FPToSI: {
719f22ef01cSRoman Divacky GenericValue GV = getConstantValue(Op0);
720f22ef01cSRoman Divacky uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
721f22ef01cSRoman Divacky if (Op0->getType()->isFloatTy())
722f22ef01cSRoman Divacky GV.IntVal = APIntOps::RoundFloatToAPInt(GV.FloatVal, BitWidth);
723f22ef01cSRoman Divacky else if (Op0->getType()->isDoubleTy())
724f22ef01cSRoman Divacky GV.IntVal = APIntOps::RoundDoubleToAPInt(GV.DoubleVal, BitWidth);
725f22ef01cSRoman Divacky else if (Op0->getType()->isX86_FP80Ty()) {
726d88c1a5aSDimitry Andric APFloat apf = APFloat(APFloat::x87DoubleExtended(), GV.IntVal);
727f22ef01cSRoman Divacky uint64_t v;
728f22ef01cSRoman Divacky bool ignored;
7297a7e6055SDimitry Andric (void)apf.convertToInteger(makeMutableArrayRef(v), BitWidth,
730f22ef01cSRoman Divacky CE->getOpcode()==Instruction::FPToSI,
731f22ef01cSRoman Divacky APFloat::rmTowardZero, &ignored);
732f22ef01cSRoman Divacky GV.IntVal = v; // endian?
733f22ef01cSRoman Divacky }
734f22ef01cSRoman Divacky return GV;
735f22ef01cSRoman Divacky }
736f22ef01cSRoman Divacky case Instruction::PtrToInt: {
737f22ef01cSRoman Divacky GenericValue GV = getConstantValue(Op0);
7387d523365SDimitry Andric uint32_t PtrWidth = DL.getTypeSizeInBits(Op0->getType());
7393861d79fSDimitry Andric assert(PtrWidth <= 64 && "Bad pointer width");
740f22ef01cSRoman Divacky GV.IntVal = APInt(PtrWidth, uintptr_t(GV.PointerVal));
7417d523365SDimitry Andric uint32_t IntWidth = DL.getTypeSizeInBits(CE->getType());
7423861d79fSDimitry Andric GV.IntVal = GV.IntVal.zextOrTrunc(IntWidth);
743f22ef01cSRoman Divacky return GV;
744f22ef01cSRoman Divacky }
745f22ef01cSRoman Divacky case Instruction::IntToPtr: {
746f22ef01cSRoman Divacky GenericValue GV = getConstantValue(Op0);
7477d523365SDimitry Andric uint32_t PtrWidth = DL.getTypeSizeInBits(CE->getType());
748f22ef01cSRoman Divacky GV.IntVal = GV.IntVal.zextOrTrunc(PtrWidth);
749f22ef01cSRoman Divacky assert(GV.IntVal.getBitWidth() <= 64 && "Bad pointer width");
750f22ef01cSRoman Divacky GV.PointerVal = PointerTy(uintptr_t(GV.IntVal.getZExtValue()));
751f22ef01cSRoman Divacky return GV;
752f22ef01cSRoman Divacky }
753f22ef01cSRoman Divacky case Instruction::BitCast: {
754f22ef01cSRoman Divacky GenericValue GV = getConstantValue(Op0);
7556122f3e6SDimitry Andric Type* DestTy = CE->getType();
756f22ef01cSRoman Divacky switch (Op0->getType()->getTypeID()) {
757f22ef01cSRoman Divacky default: llvm_unreachable("Invalid bitcast operand");
758f22ef01cSRoman Divacky case Type::IntegerTyID:
759f22ef01cSRoman Divacky assert(DestTy->isFloatingPointTy() && "invalid bitcast");
760f22ef01cSRoman Divacky if (DestTy->isFloatTy())
761f22ef01cSRoman Divacky GV.FloatVal = GV.IntVal.bitsToFloat();
762f22ef01cSRoman Divacky else if (DestTy->isDoubleTy())
763f22ef01cSRoman Divacky GV.DoubleVal = GV.IntVal.bitsToDouble();
764f22ef01cSRoman Divacky break;
765f22ef01cSRoman Divacky case Type::FloatTyID:
766f22ef01cSRoman Divacky assert(DestTy->isIntegerTy(32) && "Invalid bitcast");
7672754fe60SDimitry Andric GV.IntVal = APInt::floatToBits(GV.FloatVal);
768f22ef01cSRoman Divacky break;
769f22ef01cSRoman Divacky case Type::DoubleTyID:
770f22ef01cSRoman Divacky assert(DestTy->isIntegerTy(64) && "Invalid bitcast");
7712754fe60SDimitry Andric GV.IntVal = APInt::doubleToBits(GV.DoubleVal);
772f22ef01cSRoman Divacky break;
773f22ef01cSRoman Divacky case Type::PointerTyID:
774f22ef01cSRoman Divacky assert(DestTy->isPointerTy() && "Invalid bitcast");
775f22ef01cSRoman Divacky break; // getConstantValue(Op0) above already converted it
776f22ef01cSRoman Divacky }
777f22ef01cSRoman Divacky return GV;
778f22ef01cSRoman Divacky }
779f22ef01cSRoman Divacky case Instruction::Add:
780f22ef01cSRoman Divacky case Instruction::FAdd:
781f22ef01cSRoman Divacky case Instruction::Sub:
782f22ef01cSRoman Divacky case Instruction::FSub:
783f22ef01cSRoman Divacky case Instruction::Mul:
784f22ef01cSRoman Divacky case Instruction::FMul:
785f22ef01cSRoman Divacky case Instruction::UDiv:
786f22ef01cSRoman Divacky case Instruction::SDiv:
787f22ef01cSRoman Divacky case Instruction::URem:
788f22ef01cSRoman Divacky case Instruction::SRem:
789f22ef01cSRoman Divacky case Instruction::And:
790f22ef01cSRoman Divacky case Instruction::Or:
791f22ef01cSRoman Divacky case Instruction::Xor: {
792f22ef01cSRoman Divacky GenericValue LHS = getConstantValue(Op0);
793f22ef01cSRoman Divacky GenericValue RHS = getConstantValue(CE->getOperand(1));
794f22ef01cSRoman Divacky GenericValue GV;
795f22ef01cSRoman Divacky switch (CE->getOperand(0)->getType()->getTypeID()) {
796f22ef01cSRoman Divacky default: llvm_unreachable("Bad add type!");
797f22ef01cSRoman Divacky case Type::IntegerTyID:
798f22ef01cSRoman Divacky switch (CE->getOpcode()) {
799f22ef01cSRoman Divacky default: llvm_unreachable("Invalid integer opcode");
800f22ef01cSRoman Divacky case Instruction::Add: GV.IntVal = LHS.IntVal + RHS.IntVal; break;
801f22ef01cSRoman Divacky case Instruction::Sub: GV.IntVal = LHS.IntVal - RHS.IntVal; break;
802f22ef01cSRoman Divacky case Instruction::Mul: GV.IntVal = LHS.IntVal * RHS.IntVal; break;
803f22ef01cSRoman Divacky case Instruction::UDiv:GV.IntVal = LHS.IntVal.udiv(RHS.IntVal); break;
804f22ef01cSRoman Divacky case Instruction::SDiv:GV.IntVal = LHS.IntVal.sdiv(RHS.IntVal); break;
805f22ef01cSRoman Divacky case Instruction::URem:GV.IntVal = LHS.IntVal.urem(RHS.IntVal); break;
806f22ef01cSRoman Divacky case Instruction::SRem:GV.IntVal = LHS.IntVal.srem(RHS.IntVal); break;
807f22ef01cSRoman Divacky case Instruction::And: GV.IntVal = LHS.IntVal & RHS.IntVal; break;
808f22ef01cSRoman Divacky case Instruction::Or: GV.IntVal = LHS.IntVal | RHS.IntVal; break;
809f22ef01cSRoman Divacky case Instruction::Xor: GV.IntVal = LHS.IntVal ^ RHS.IntVal; break;
810f22ef01cSRoman Divacky }
811f22ef01cSRoman Divacky break;
812f22ef01cSRoman Divacky case Type::FloatTyID:
813f22ef01cSRoman Divacky switch (CE->getOpcode()) {
814f22ef01cSRoman Divacky default: llvm_unreachable("Invalid float opcode");
815f22ef01cSRoman Divacky case Instruction::FAdd:
816f22ef01cSRoman Divacky GV.FloatVal = LHS.FloatVal + RHS.FloatVal; break;
817f22ef01cSRoman Divacky case Instruction::FSub:
818f22ef01cSRoman Divacky GV.FloatVal = LHS.FloatVal - RHS.FloatVal; break;
819f22ef01cSRoman Divacky case Instruction::FMul:
820f22ef01cSRoman Divacky GV.FloatVal = LHS.FloatVal * RHS.FloatVal; break;
821f22ef01cSRoman Divacky case Instruction::FDiv:
822f22ef01cSRoman Divacky GV.FloatVal = LHS.FloatVal / RHS.FloatVal; break;
823f22ef01cSRoman Divacky case Instruction::FRem:
824f22ef01cSRoman Divacky GV.FloatVal = std::fmod(LHS.FloatVal,RHS.FloatVal); break;
825f22ef01cSRoman Divacky }
826f22ef01cSRoman Divacky break;
827f22ef01cSRoman Divacky case Type::DoubleTyID:
828f22ef01cSRoman Divacky switch (CE->getOpcode()) {
829f22ef01cSRoman Divacky default: llvm_unreachable("Invalid double opcode");
830f22ef01cSRoman Divacky case Instruction::FAdd:
831f22ef01cSRoman Divacky GV.DoubleVal = LHS.DoubleVal + RHS.DoubleVal; break;
832f22ef01cSRoman Divacky case Instruction::FSub:
833f22ef01cSRoman Divacky GV.DoubleVal = LHS.DoubleVal - RHS.DoubleVal; break;
834f22ef01cSRoman Divacky case Instruction::FMul:
835f22ef01cSRoman Divacky GV.DoubleVal = LHS.DoubleVal * RHS.DoubleVal; break;
836f22ef01cSRoman Divacky case Instruction::FDiv:
837f22ef01cSRoman Divacky GV.DoubleVal = LHS.DoubleVal / RHS.DoubleVal; break;
838f22ef01cSRoman Divacky case Instruction::FRem:
839f22ef01cSRoman Divacky GV.DoubleVal = std::fmod(LHS.DoubleVal,RHS.DoubleVal); break;
840f22ef01cSRoman Divacky }
841f22ef01cSRoman Divacky break;
842f22ef01cSRoman Divacky case Type::X86_FP80TyID:
843f22ef01cSRoman Divacky case Type::PPC_FP128TyID:
844f22ef01cSRoman Divacky case Type::FP128TyID: {
845139f7f9bSDimitry Andric const fltSemantics &Sem = CE->getOperand(0)->getType()->getFltSemantics();
846139f7f9bSDimitry Andric APFloat apfLHS = APFloat(Sem, LHS.IntVal);
847f22ef01cSRoman Divacky switch (CE->getOpcode()) {
8482754fe60SDimitry Andric default: llvm_unreachable("Invalid long double opcode");
849f22ef01cSRoman Divacky case Instruction::FAdd:
850139f7f9bSDimitry Andric apfLHS.add(APFloat(Sem, RHS.IntVal), APFloat::rmNearestTiesToEven);
851f22ef01cSRoman Divacky GV.IntVal = apfLHS.bitcastToAPInt();
852f22ef01cSRoman Divacky break;
853f22ef01cSRoman Divacky case Instruction::FSub:
854139f7f9bSDimitry Andric apfLHS.subtract(APFloat(Sem, RHS.IntVal),
855139f7f9bSDimitry Andric APFloat::rmNearestTiesToEven);
856f22ef01cSRoman Divacky GV.IntVal = apfLHS.bitcastToAPInt();
857f22ef01cSRoman Divacky break;
858f22ef01cSRoman Divacky case Instruction::FMul:
859139f7f9bSDimitry Andric apfLHS.multiply(APFloat(Sem, RHS.IntVal),
860139f7f9bSDimitry Andric APFloat::rmNearestTiesToEven);
861f22ef01cSRoman Divacky GV.IntVal = apfLHS.bitcastToAPInt();
862f22ef01cSRoman Divacky break;
863f22ef01cSRoman Divacky case Instruction::FDiv:
864139f7f9bSDimitry Andric apfLHS.divide(APFloat(Sem, RHS.IntVal),
865139f7f9bSDimitry Andric APFloat::rmNearestTiesToEven);
866f22ef01cSRoman Divacky GV.IntVal = apfLHS.bitcastToAPInt();
867f22ef01cSRoman Divacky break;
868f22ef01cSRoman Divacky case Instruction::FRem:
8697d523365SDimitry Andric apfLHS.mod(APFloat(Sem, RHS.IntVal));
870f22ef01cSRoman Divacky GV.IntVal = apfLHS.bitcastToAPInt();
871f22ef01cSRoman Divacky break;
872f22ef01cSRoman Divacky }
873f22ef01cSRoman Divacky }
874f22ef01cSRoman Divacky break;
875f22ef01cSRoman Divacky }
876f22ef01cSRoman Divacky return GV;
877f22ef01cSRoman Divacky }
878f22ef01cSRoman Divacky default:
879f22ef01cSRoman Divacky break;
880f22ef01cSRoman Divacky }
8812754fe60SDimitry Andric
8822754fe60SDimitry Andric SmallString<256> Msg;
8832754fe60SDimitry Andric raw_svector_ostream OS(Msg);
8842754fe60SDimitry Andric OS << "ConstantExpr not handled: " << *CE;
8852754fe60SDimitry Andric report_fatal_error(OS.str());
886f22ef01cSRoman Divacky }
887f22ef01cSRoman Divacky
8882754fe60SDimitry Andric // Otherwise, we have a simple constant.
889f22ef01cSRoman Divacky GenericValue Result;
890f22ef01cSRoman Divacky switch (C->getType()->getTypeID()) {
891f22ef01cSRoman Divacky case Type::FloatTyID:
892f22ef01cSRoman Divacky Result.FloatVal = cast<ConstantFP>(C)->getValueAPF().convertToFloat();
893f22ef01cSRoman Divacky break;
894f22ef01cSRoman Divacky case Type::DoubleTyID:
895f22ef01cSRoman Divacky Result.DoubleVal = cast<ConstantFP>(C)->getValueAPF().convertToDouble();
896f22ef01cSRoman Divacky break;
897f22ef01cSRoman Divacky case Type::X86_FP80TyID:
898f22ef01cSRoman Divacky case Type::FP128TyID:
899f22ef01cSRoman Divacky case Type::PPC_FP128TyID:
900f22ef01cSRoman Divacky Result.IntVal = cast <ConstantFP>(C)->getValueAPF().bitcastToAPInt();
901f22ef01cSRoman Divacky break;
902f22ef01cSRoman Divacky case Type::IntegerTyID:
903f22ef01cSRoman Divacky Result.IntVal = cast<ConstantInt>(C)->getValue();
904f22ef01cSRoman Divacky break;
905f22ef01cSRoman Divacky case Type::PointerTyID:
906*4ba319b5SDimitry Andric while (auto *A = dyn_cast<GlobalAlias>(C)) {
907*4ba319b5SDimitry Andric C = A->getAliasee();
908*4ba319b5SDimitry Andric }
909f22ef01cSRoman Divacky if (isa<ConstantPointerNull>(C))
91091bc56edSDimitry Andric Result.PointerVal = nullptr;
911f22ef01cSRoman Divacky else if (const Function *F = dyn_cast<Function>(C))
912f22ef01cSRoman Divacky Result = PTOGV(getPointerToFunctionOrStub(const_cast<Function*>(F)));
913f22ef01cSRoman Divacky else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(C))
914f22ef01cSRoman Divacky Result = PTOGV(getOrEmitGlobalVariable(const_cast<GlobalVariable*>(GV)));
915f22ef01cSRoman Divacky else
916f22ef01cSRoman Divacky llvm_unreachable("Unknown constant pointer type!");
917f22ef01cSRoman Divacky break;
918139f7f9bSDimitry Andric case Type::VectorTyID: {
919139f7f9bSDimitry Andric unsigned elemNum;
920139f7f9bSDimitry Andric Type* ElemTy;
921139f7f9bSDimitry Andric const ConstantDataVector *CDV = dyn_cast<ConstantDataVector>(C);
922139f7f9bSDimitry Andric const ConstantVector *CV = dyn_cast<ConstantVector>(C);
923139f7f9bSDimitry Andric const ConstantAggregateZero *CAZ = dyn_cast<ConstantAggregateZero>(C);
924139f7f9bSDimitry Andric
925139f7f9bSDimitry Andric if (CDV) {
926139f7f9bSDimitry Andric elemNum = CDV->getNumElements();
927139f7f9bSDimitry Andric ElemTy = CDV->getElementType();
928139f7f9bSDimitry Andric } else if (CV || CAZ) {
929139f7f9bSDimitry Andric VectorType* VTy = dyn_cast<VectorType>(C->getType());
930139f7f9bSDimitry Andric elemNum = VTy->getNumElements();
931139f7f9bSDimitry Andric ElemTy = VTy->getElementType();
932139f7f9bSDimitry Andric } else {
933139f7f9bSDimitry Andric llvm_unreachable("Unknown constant vector type!");
934139f7f9bSDimitry Andric }
935139f7f9bSDimitry Andric
936139f7f9bSDimitry Andric Result.AggregateVal.resize(elemNum);
937139f7f9bSDimitry Andric // Check if vector holds floats.
938139f7f9bSDimitry Andric if(ElemTy->isFloatTy()) {
939139f7f9bSDimitry Andric if (CAZ) {
940139f7f9bSDimitry Andric GenericValue floatZero;
941139f7f9bSDimitry Andric floatZero.FloatVal = 0.f;
942139f7f9bSDimitry Andric std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(),
943139f7f9bSDimitry Andric floatZero);
944139f7f9bSDimitry Andric break;
945139f7f9bSDimitry Andric }
946139f7f9bSDimitry Andric if(CV) {
947139f7f9bSDimitry Andric for (unsigned i = 0; i < elemNum; ++i)
948139f7f9bSDimitry Andric if (!isa<UndefValue>(CV->getOperand(i)))
949139f7f9bSDimitry Andric Result.AggregateVal[i].FloatVal = cast<ConstantFP>(
950139f7f9bSDimitry Andric CV->getOperand(i))->getValueAPF().convertToFloat();
951139f7f9bSDimitry Andric break;
952139f7f9bSDimitry Andric }
953139f7f9bSDimitry Andric if(CDV)
954139f7f9bSDimitry Andric for (unsigned i = 0; i < elemNum; ++i)
955139f7f9bSDimitry Andric Result.AggregateVal[i].FloatVal = CDV->getElementAsFloat(i);
956139f7f9bSDimitry Andric
957139f7f9bSDimitry Andric break;
958139f7f9bSDimitry Andric }
959139f7f9bSDimitry Andric // Check if vector holds doubles.
960139f7f9bSDimitry Andric if (ElemTy->isDoubleTy()) {
961139f7f9bSDimitry Andric if (CAZ) {
962139f7f9bSDimitry Andric GenericValue doubleZero;
963139f7f9bSDimitry Andric doubleZero.DoubleVal = 0.0;
964139f7f9bSDimitry Andric std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(),
965139f7f9bSDimitry Andric doubleZero);
966139f7f9bSDimitry Andric break;
967139f7f9bSDimitry Andric }
968139f7f9bSDimitry Andric if(CV) {
969139f7f9bSDimitry Andric for (unsigned i = 0; i < elemNum; ++i)
970139f7f9bSDimitry Andric if (!isa<UndefValue>(CV->getOperand(i)))
971139f7f9bSDimitry Andric Result.AggregateVal[i].DoubleVal = cast<ConstantFP>(
972139f7f9bSDimitry Andric CV->getOperand(i))->getValueAPF().convertToDouble();
973139f7f9bSDimitry Andric break;
974139f7f9bSDimitry Andric }
975139f7f9bSDimitry Andric if(CDV)
976139f7f9bSDimitry Andric for (unsigned i = 0; i < elemNum; ++i)
977139f7f9bSDimitry Andric Result.AggregateVal[i].DoubleVal = CDV->getElementAsDouble(i);
978139f7f9bSDimitry Andric
979139f7f9bSDimitry Andric break;
980139f7f9bSDimitry Andric }
981139f7f9bSDimitry Andric // Check if vector holds integers.
982139f7f9bSDimitry Andric if (ElemTy->isIntegerTy()) {
983139f7f9bSDimitry Andric if (CAZ) {
984139f7f9bSDimitry Andric GenericValue intZero;
985139f7f9bSDimitry Andric intZero.IntVal = APInt(ElemTy->getScalarSizeInBits(), 0ull);
986139f7f9bSDimitry Andric std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(),
987139f7f9bSDimitry Andric intZero);
988139f7f9bSDimitry Andric break;
989139f7f9bSDimitry Andric }
990139f7f9bSDimitry Andric if(CV) {
991139f7f9bSDimitry Andric for (unsigned i = 0; i < elemNum; ++i)
992139f7f9bSDimitry Andric if (!isa<UndefValue>(CV->getOperand(i)))
993139f7f9bSDimitry Andric Result.AggregateVal[i].IntVal = cast<ConstantInt>(
994139f7f9bSDimitry Andric CV->getOperand(i))->getValue();
995139f7f9bSDimitry Andric else {
996139f7f9bSDimitry Andric Result.AggregateVal[i].IntVal =
997139f7f9bSDimitry Andric APInt(CV->getOperand(i)->getType()->getPrimitiveSizeInBits(), 0);
998139f7f9bSDimitry Andric }
999139f7f9bSDimitry Andric break;
1000139f7f9bSDimitry Andric }
1001139f7f9bSDimitry Andric if(CDV)
1002139f7f9bSDimitry Andric for (unsigned i = 0; i < elemNum; ++i)
1003139f7f9bSDimitry Andric Result.AggregateVal[i].IntVal = APInt(
1004139f7f9bSDimitry Andric CDV->getElementType()->getPrimitiveSizeInBits(),
1005139f7f9bSDimitry Andric CDV->getElementAsInteger(i));
1006139f7f9bSDimitry Andric
1007139f7f9bSDimitry Andric break;
1008139f7f9bSDimitry Andric }
1009139f7f9bSDimitry Andric llvm_unreachable("Unknown constant pointer type!");
1010139f7f9bSDimitry Andric }
1011139f7f9bSDimitry Andric break;
1012139f7f9bSDimitry Andric
1013f22ef01cSRoman Divacky default:
10142754fe60SDimitry Andric SmallString<256> Msg;
10152754fe60SDimitry Andric raw_svector_ostream OS(Msg);
10162754fe60SDimitry Andric OS << "ERROR: Constant unimplemented for type: " << *C->getType();
10172754fe60SDimitry Andric report_fatal_error(OS.str());
1018f22ef01cSRoman Divacky }
10192754fe60SDimitry Andric
1020f22ef01cSRoman Divacky return Result;
1021f22ef01cSRoman Divacky }
1022f22ef01cSRoman Divacky
1023f22ef01cSRoman Divacky /// StoreIntToMemory - Fills the StoreBytes bytes of memory starting from Dst
1024f22ef01cSRoman Divacky /// with the integer held in IntVal.
StoreIntToMemory(const APInt & IntVal,uint8_t * Dst,unsigned StoreBytes)1025f22ef01cSRoman Divacky static void StoreIntToMemory(const APInt &IntVal, uint8_t *Dst,
1026f22ef01cSRoman Divacky unsigned StoreBytes) {
1027f22ef01cSRoman Divacky assert((IntVal.getBitWidth()+7)/8 >= StoreBytes && "Integer too small!");
10283861d79fSDimitry Andric const uint8_t *Src = (const uint8_t *)IntVal.getRawData();
1029f22ef01cSRoman Divacky
1030284c1978SDimitry Andric if (sys::IsLittleEndianHost) {
1031f22ef01cSRoman Divacky // Little-endian host - the source is ordered from LSB to MSB. Order the
1032f22ef01cSRoman Divacky // destination from LSB to MSB: Do a straight copy.
1033f22ef01cSRoman Divacky memcpy(Dst, Src, StoreBytes);
10342754fe60SDimitry Andric } else {
1035f22ef01cSRoman Divacky // Big-endian host - the source is an array of 64 bit words ordered from
1036f22ef01cSRoman Divacky // LSW to MSW. Each word is ordered from MSB to LSB. Order the destination
1037f22ef01cSRoman Divacky // from MSB to LSB: Reverse the word order, but not the bytes in a word.
1038f22ef01cSRoman Divacky while (StoreBytes > sizeof(uint64_t)) {
1039f22ef01cSRoman Divacky StoreBytes -= sizeof(uint64_t);
1040f22ef01cSRoman Divacky // May not be aligned so use memcpy.
1041f22ef01cSRoman Divacky memcpy(Dst + StoreBytes, Src, sizeof(uint64_t));
1042f22ef01cSRoman Divacky Src += sizeof(uint64_t);
1043f22ef01cSRoman Divacky }
1044f22ef01cSRoman Divacky
1045f22ef01cSRoman Divacky memcpy(Dst, Src + sizeof(uint64_t) - StoreBytes, StoreBytes);
1046f22ef01cSRoman Divacky }
1047f22ef01cSRoman Divacky }
1048f22ef01cSRoman Divacky
StoreValueToMemory(const GenericValue & Val,GenericValue * Ptr,Type * Ty)1049f22ef01cSRoman Divacky void ExecutionEngine::StoreValueToMemory(const GenericValue &Val,
10506122f3e6SDimitry Andric GenericValue *Ptr, Type *Ty) {
10517d523365SDimitry Andric const unsigned StoreBytes = getDataLayout().getTypeStoreSize(Ty);
1052f22ef01cSRoman Divacky
1053f22ef01cSRoman Divacky switch (Ty->getTypeID()) {
1054139f7f9bSDimitry Andric default:
1055139f7f9bSDimitry Andric dbgs() << "Cannot store value of type " << *Ty << "!\n";
1056139f7f9bSDimitry Andric break;
1057f22ef01cSRoman Divacky case Type::IntegerTyID:
1058f22ef01cSRoman Divacky StoreIntToMemory(Val.IntVal, (uint8_t*)Ptr, StoreBytes);
1059f22ef01cSRoman Divacky break;
1060f22ef01cSRoman Divacky case Type::FloatTyID:
1061f22ef01cSRoman Divacky *((float*)Ptr) = Val.FloatVal;
1062f22ef01cSRoman Divacky break;
1063f22ef01cSRoman Divacky case Type::DoubleTyID:
1064f22ef01cSRoman Divacky *((double*)Ptr) = Val.DoubleVal;
1065f22ef01cSRoman Divacky break;
1066f22ef01cSRoman Divacky case Type::X86_FP80TyID:
1067f22ef01cSRoman Divacky memcpy(Ptr, Val.IntVal.getRawData(), 10);
1068f22ef01cSRoman Divacky break;
1069f22ef01cSRoman Divacky case Type::PointerTyID:
1070f22ef01cSRoman Divacky // Ensure 64 bit target pointers are fully initialized on 32 bit hosts.
1071f22ef01cSRoman Divacky if (StoreBytes != sizeof(PointerTy))
10723b0f4066SDimitry Andric memset(&(Ptr->PointerVal), 0, StoreBytes);
1073f22ef01cSRoman Divacky
1074f22ef01cSRoman Divacky *((PointerTy*)Ptr) = Val.PointerVal;
1075f22ef01cSRoman Divacky break;
1076139f7f9bSDimitry Andric case Type::VectorTyID:
1077139f7f9bSDimitry Andric for (unsigned i = 0; i < Val.AggregateVal.size(); ++i) {
1078139f7f9bSDimitry Andric if (cast<VectorType>(Ty)->getElementType()->isDoubleTy())
1079139f7f9bSDimitry Andric *(((double*)Ptr)+i) = Val.AggregateVal[i].DoubleVal;
1080139f7f9bSDimitry Andric if (cast<VectorType>(Ty)->getElementType()->isFloatTy())
1081139f7f9bSDimitry Andric *(((float*)Ptr)+i) = Val.AggregateVal[i].FloatVal;
1082139f7f9bSDimitry Andric if (cast<VectorType>(Ty)->getElementType()->isIntegerTy()) {
1083139f7f9bSDimitry Andric unsigned numOfBytes =(Val.AggregateVal[i].IntVal.getBitWidth()+7)/8;
1084139f7f9bSDimitry Andric StoreIntToMemory(Val.AggregateVal[i].IntVal,
1085139f7f9bSDimitry Andric (uint8_t*)Ptr + numOfBytes*i, numOfBytes);
1086139f7f9bSDimitry Andric }
1087139f7f9bSDimitry Andric }
1088139f7f9bSDimitry Andric break;
1089f22ef01cSRoman Divacky }
1090f22ef01cSRoman Divacky
10917d523365SDimitry Andric if (sys::IsLittleEndianHost != getDataLayout().isLittleEndian())
1092f22ef01cSRoman Divacky // Host and target are different endian - reverse the stored bytes.
1093f22ef01cSRoman Divacky std::reverse((uint8_t*)Ptr, StoreBytes + (uint8_t*)Ptr);
1094f22ef01cSRoman Divacky }
1095f22ef01cSRoman Divacky
1096f22ef01cSRoman Divacky /// LoadIntFromMemory - Loads the integer stored in the LoadBytes bytes starting
1097f22ef01cSRoman Divacky /// from Src into IntVal, which is assumed to be wide enough and to hold zero.
LoadIntFromMemory(APInt & IntVal,uint8_t * Src,unsigned LoadBytes)1098f22ef01cSRoman Divacky static void LoadIntFromMemory(APInt &IntVal, uint8_t *Src, unsigned LoadBytes) {
1099f22ef01cSRoman Divacky assert((IntVal.getBitWidth()+7)/8 >= LoadBytes && "Integer too small!");
1100139f7f9bSDimitry Andric uint8_t *Dst = reinterpret_cast<uint8_t *>(
1101139f7f9bSDimitry Andric const_cast<uint64_t *>(IntVal.getRawData()));
1102f22ef01cSRoman Divacky
1103284c1978SDimitry Andric if (sys::IsLittleEndianHost)
1104f22ef01cSRoman Divacky // Little-endian host - the destination must be ordered from LSB to MSB.
1105f22ef01cSRoman Divacky // The source is ordered from LSB to MSB: Do a straight copy.
1106f22ef01cSRoman Divacky memcpy(Dst, Src, LoadBytes);
1107f22ef01cSRoman Divacky else {
1108f22ef01cSRoman Divacky // Big-endian - the destination is an array of 64 bit words ordered from
1109f22ef01cSRoman Divacky // LSW to MSW. Each word must be ordered from MSB to LSB. The source is
1110f22ef01cSRoman Divacky // ordered from MSB to LSB: Reverse the word order, but not the bytes in
1111f22ef01cSRoman Divacky // a word.
1112f22ef01cSRoman Divacky while (LoadBytes > sizeof(uint64_t)) {
1113f22ef01cSRoman Divacky LoadBytes -= sizeof(uint64_t);
1114f22ef01cSRoman Divacky // May not be aligned so use memcpy.
1115f22ef01cSRoman Divacky memcpy(Dst, Src + LoadBytes, sizeof(uint64_t));
1116f22ef01cSRoman Divacky Dst += sizeof(uint64_t);
1117f22ef01cSRoman Divacky }
1118f22ef01cSRoman Divacky
1119f22ef01cSRoman Divacky memcpy(Dst + sizeof(uint64_t) - LoadBytes, Src, LoadBytes);
1120f22ef01cSRoman Divacky }
1121f22ef01cSRoman Divacky }
1122f22ef01cSRoman Divacky
1123f22ef01cSRoman Divacky /// FIXME: document
1124f22ef01cSRoman Divacky ///
LoadValueFromMemory(GenericValue & Result,GenericValue * Ptr,Type * Ty)1125f22ef01cSRoman Divacky void ExecutionEngine::LoadValueFromMemory(GenericValue &Result,
1126f22ef01cSRoman Divacky GenericValue *Ptr,
11276122f3e6SDimitry Andric Type *Ty) {
11287d523365SDimitry Andric const unsigned LoadBytes = getDataLayout().getTypeStoreSize(Ty);
1129f22ef01cSRoman Divacky
1130f22ef01cSRoman Divacky switch (Ty->getTypeID()) {
1131f22ef01cSRoman Divacky case Type::IntegerTyID:
1132f22ef01cSRoman Divacky // An APInt with all words initially zero.
1133f22ef01cSRoman Divacky Result.IntVal = APInt(cast<IntegerType>(Ty)->getBitWidth(), 0);
1134f22ef01cSRoman Divacky LoadIntFromMemory(Result.IntVal, (uint8_t*)Ptr, LoadBytes);
1135f22ef01cSRoman Divacky break;
1136f22ef01cSRoman Divacky case Type::FloatTyID:
1137f22ef01cSRoman Divacky Result.FloatVal = *((float*)Ptr);
1138f22ef01cSRoman Divacky break;
1139f22ef01cSRoman Divacky case Type::DoubleTyID:
1140f22ef01cSRoman Divacky Result.DoubleVal = *((double*)Ptr);
1141f22ef01cSRoman Divacky break;
1142f22ef01cSRoman Divacky case Type::PointerTyID:
1143f22ef01cSRoman Divacky Result.PointerVal = *((PointerTy*)Ptr);
1144f22ef01cSRoman Divacky break;
1145f22ef01cSRoman Divacky case Type::X86_FP80TyID: {
1146f22ef01cSRoman Divacky // This is endian dependent, but it will only work on x86 anyway.
1147f22ef01cSRoman Divacky // FIXME: Will not trap if loading a signaling NaN.
1148f22ef01cSRoman Divacky uint64_t y[2];
1149f22ef01cSRoman Divacky memcpy(y, Ptr, 10);
11506122f3e6SDimitry Andric Result.IntVal = APInt(80, y);
1151f22ef01cSRoman Divacky break;
1152f22ef01cSRoman Divacky }
1153139f7f9bSDimitry Andric case Type::VectorTyID: {
11547d523365SDimitry Andric auto *VT = cast<VectorType>(Ty);
11557d523365SDimitry Andric Type *ElemT = VT->getElementType();
1156139f7f9bSDimitry Andric const unsigned numElems = VT->getNumElements();
1157139f7f9bSDimitry Andric if (ElemT->isFloatTy()) {
1158139f7f9bSDimitry Andric Result.AggregateVal.resize(numElems);
1159139f7f9bSDimitry Andric for (unsigned i = 0; i < numElems; ++i)
1160139f7f9bSDimitry Andric Result.AggregateVal[i].FloatVal = *((float*)Ptr+i);
1161139f7f9bSDimitry Andric }
1162139f7f9bSDimitry Andric if (ElemT->isDoubleTy()) {
1163139f7f9bSDimitry Andric Result.AggregateVal.resize(numElems);
1164139f7f9bSDimitry Andric for (unsigned i = 0; i < numElems; ++i)
1165139f7f9bSDimitry Andric Result.AggregateVal[i].DoubleVal = *((double*)Ptr+i);
1166139f7f9bSDimitry Andric }
1167139f7f9bSDimitry Andric if (ElemT->isIntegerTy()) {
1168139f7f9bSDimitry Andric GenericValue intZero;
1169139f7f9bSDimitry Andric const unsigned elemBitWidth = cast<IntegerType>(ElemT)->getBitWidth();
1170139f7f9bSDimitry Andric intZero.IntVal = APInt(elemBitWidth, 0);
1171139f7f9bSDimitry Andric Result.AggregateVal.resize(numElems, intZero);
1172139f7f9bSDimitry Andric for (unsigned i = 0; i < numElems; ++i)
1173139f7f9bSDimitry Andric LoadIntFromMemory(Result.AggregateVal[i].IntVal,
1174139f7f9bSDimitry Andric (uint8_t*)Ptr+((elemBitWidth+7)/8)*i, (elemBitWidth+7)/8);
1175139f7f9bSDimitry Andric }
1176139f7f9bSDimitry Andric break;
1177139f7f9bSDimitry Andric }
1178f22ef01cSRoman Divacky default:
11792754fe60SDimitry Andric SmallString<256> Msg;
11802754fe60SDimitry Andric raw_svector_ostream OS(Msg);
11812754fe60SDimitry Andric OS << "Cannot load value of type " << *Ty << "!";
11822754fe60SDimitry Andric report_fatal_error(OS.str());
1183f22ef01cSRoman Divacky }
1184f22ef01cSRoman Divacky }
1185f22ef01cSRoman Divacky
InitializeMemory(const Constant * Init,void * Addr)1186f22ef01cSRoman Divacky void ExecutionEngine::InitializeMemory(const Constant *Init, void *Addr) {
1187*4ba319b5SDimitry Andric LLVM_DEBUG(dbgs() << "JIT: Initializing " << Addr << " ");
1188*4ba319b5SDimitry Andric LLVM_DEBUG(Init->dump());
1189dff0c46cSDimitry Andric if (isa<UndefValue>(Init))
1190f22ef01cSRoman Divacky return;
1191dff0c46cSDimitry Andric
1192dff0c46cSDimitry Andric if (const ConstantVector *CP = dyn_cast<ConstantVector>(Init)) {
1193f22ef01cSRoman Divacky unsigned ElementSize =
11947d523365SDimitry Andric getDataLayout().getTypeAllocSize(CP->getType()->getElementType());
1195f22ef01cSRoman Divacky for (unsigned i = 0, e = CP->getNumOperands(); i != e; ++i)
1196f22ef01cSRoman Divacky InitializeMemory(CP->getOperand(i), (char*)Addr+i*ElementSize);
1197f22ef01cSRoman Divacky return;
1198dff0c46cSDimitry Andric }
1199dff0c46cSDimitry Andric
1200dff0c46cSDimitry Andric if (isa<ConstantAggregateZero>(Init)) {
12017d523365SDimitry Andric memset(Addr, 0, (size_t)getDataLayout().getTypeAllocSize(Init->getType()));
1202f22ef01cSRoman Divacky return;
1203dff0c46cSDimitry Andric }
1204dff0c46cSDimitry Andric
1205dff0c46cSDimitry Andric if (const ConstantArray *CPA = dyn_cast<ConstantArray>(Init)) {
1206f22ef01cSRoman Divacky unsigned ElementSize =
12077d523365SDimitry Andric getDataLayout().getTypeAllocSize(CPA->getType()->getElementType());
1208f22ef01cSRoman Divacky for (unsigned i = 0, e = CPA->getNumOperands(); i != e; ++i)
1209f22ef01cSRoman Divacky InitializeMemory(CPA->getOperand(i), (char*)Addr+i*ElementSize);
1210f22ef01cSRoman Divacky return;
1211dff0c46cSDimitry Andric }
1212dff0c46cSDimitry Andric
1213dff0c46cSDimitry Andric if (const ConstantStruct *CPS = dyn_cast<ConstantStruct>(Init)) {
1214f22ef01cSRoman Divacky const StructLayout *SL =
12157d523365SDimitry Andric getDataLayout().getStructLayout(cast<StructType>(CPS->getType()));
1216f22ef01cSRoman Divacky for (unsigned i = 0, e = CPS->getNumOperands(); i != e; ++i)
1217f22ef01cSRoman Divacky InitializeMemory(CPS->getOperand(i), (char*)Addr+SL->getElementOffset(i));
1218f22ef01cSRoman Divacky return;
1219dff0c46cSDimitry Andric }
1220dff0c46cSDimitry Andric
1221dff0c46cSDimitry Andric if (const ConstantDataSequential *CDS =
1222dff0c46cSDimitry Andric dyn_cast<ConstantDataSequential>(Init)) {
1223dff0c46cSDimitry Andric // CDS is already laid out in host memory order.
1224dff0c46cSDimitry Andric StringRef Data = CDS->getRawDataValues();
1225dff0c46cSDimitry Andric memcpy(Addr, Data.data(), Data.size());
1226dff0c46cSDimitry Andric return;
1227dff0c46cSDimitry Andric }
1228dff0c46cSDimitry Andric
1229dff0c46cSDimitry Andric if (Init->getType()->isFirstClassType()) {
1230f22ef01cSRoman Divacky GenericValue Val = getConstantValue(Init);
1231f22ef01cSRoman Divacky StoreValueToMemory(Val, (GenericValue*)Addr, Init->getType());
1232f22ef01cSRoman Divacky return;
1233f22ef01cSRoman Divacky }
1234f22ef01cSRoman Divacky
1235*4ba319b5SDimitry Andric LLVM_DEBUG(dbgs() << "Bad Type: " << *Init->getType() << "\n");
1236f22ef01cSRoman Divacky llvm_unreachable("Unknown constant type to initialize memory with!");
1237f22ef01cSRoman Divacky }
1238f22ef01cSRoman Divacky
1239f22ef01cSRoman Divacky /// EmitGlobals - Emit all of the global variables to memory, storing their
1240f22ef01cSRoman Divacky /// addresses into GlobalAddress. This must make sure to copy the contents of
1241f22ef01cSRoman Divacky /// their initializers into the memory.
emitGlobals()1242f22ef01cSRoman Divacky void ExecutionEngine::emitGlobals() {
1243f22ef01cSRoman Divacky // Loop over all of the global variables in the program, allocating the memory
1244f22ef01cSRoman Divacky // to hold them. If there is more than one module, do a prepass over globals
1245f22ef01cSRoman Divacky // to figure out how the different modules should link together.
12466122f3e6SDimitry Andric std::map<std::pair<std::string, Type*>,
1247f22ef01cSRoman Divacky const GlobalValue*> LinkedGlobalsMap;
1248f22ef01cSRoman Divacky
1249f22ef01cSRoman Divacky if (Modules.size() != 1) {
1250f22ef01cSRoman Divacky for (unsigned m = 0, e = Modules.size(); m != e; ++m) {
1251f22ef01cSRoman Divacky Module &M = *Modules[m];
125291bc56edSDimitry Andric for (const auto &GV : M.globals()) {
125391bc56edSDimitry Andric if (GV.hasLocalLinkage() || GV.isDeclaration() ||
125491bc56edSDimitry Andric GV.hasAppendingLinkage() || !GV.hasName())
1255f22ef01cSRoman Divacky continue;// Ignore external globals and globals with internal linkage.
1256f22ef01cSRoman Divacky
1257f22ef01cSRoman Divacky const GlobalValue *&GVEntry =
125891bc56edSDimitry Andric LinkedGlobalsMap[std::make_pair(GV.getName(), GV.getType())];
1259f22ef01cSRoman Divacky
1260f22ef01cSRoman Divacky // If this is the first time we've seen this global, it is the canonical
1261f22ef01cSRoman Divacky // version.
1262f22ef01cSRoman Divacky if (!GVEntry) {
126391bc56edSDimitry Andric GVEntry = &GV;
1264f22ef01cSRoman Divacky continue;
1265f22ef01cSRoman Divacky }
1266f22ef01cSRoman Divacky
1267f22ef01cSRoman Divacky // If the existing global is strong, never replace it.
126891bc56edSDimitry Andric if (GVEntry->hasExternalLinkage())
1269f22ef01cSRoman Divacky continue;
1270f22ef01cSRoman Divacky
1271f22ef01cSRoman Divacky // Otherwise, we know it's linkonce/weak, replace it if this is a strong
1272f22ef01cSRoman Divacky // symbol. FIXME is this right for common?
127391bc56edSDimitry Andric if (GV.hasExternalLinkage() || GVEntry->hasExternalWeakLinkage())
127491bc56edSDimitry Andric GVEntry = &GV;
1275f22ef01cSRoman Divacky }
1276f22ef01cSRoman Divacky }
1277f22ef01cSRoman Divacky }
1278f22ef01cSRoman Divacky
1279f22ef01cSRoman Divacky std::vector<const GlobalValue*> NonCanonicalGlobals;
1280f22ef01cSRoman Divacky for (unsigned m = 0, e = Modules.size(); m != e; ++m) {
1281f22ef01cSRoman Divacky Module &M = *Modules[m];
128291bc56edSDimitry Andric for (const auto &GV : M.globals()) {
1283f22ef01cSRoman Divacky // In the multi-module case, see what this global maps to.
1284f22ef01cSRoman Divacky if (!LinkedGlobalsMap.empty()) {
1285f22ef01cSRoman Divacky if (const GlobalValue *GVEntry =
128691bc56edSDimitry Andric LinkedGlobalsMap[std::make_pair(GV.getName(), GV.getType())]) {
1287f22ef01cSRoman Divacky // If something else is the canonical global, ignore this one.
128891bc56edSDimitry Andric if (GVEntry != &GV) {
128991bc56edSDimitry Andric NonCanonicalGlobals.push_back(&GV);
1290f22ef01cSRoman Divacky continue;
1291f22ef01cSRoman Divacky }
1292f22ef01cSRoman Divacky }
1293f22ef01cSRoman Divacky }
1294f22ef01cSRoman Divacky
129591bc56edSDimitry Andric if (!GV.isDeclaration()) {
129691bc56edSDimitry Andric addGlobalMapping(&GV, getMemoryForGV(&GV));
1297f22ef01cSRoman Divacky } else {
1298f22ef01cSRoman Divacky // External variable reference. Try to use the dynamic loader to
1299f22ef01cSRoman Divacky // get a pointer to it.
1300f22ef01cSRoman Divacky if (void *SymAddr =
130191bc56edSDimitry Andric sys::DynamicLibrary::SearchForAddressOfSymbol(GV.getName()))
130291bc56edSDimitry Andric addGlobalMapping(&GV, SymAddr);
1303f22ef01cSRoman Divacky else {
1304f22ef01cSRoman Divacky report_fatal_error("Could not resolve external global address: "
130591bc56edSDimitry Andric +GV.getName());
1306f22ef01cSRoman Divacky }
1307f22ef01cSRoman Divacky }
1308f22ef01cSRoman Divacky }
1309f22ef01cSRoman Divacky
1310f22ef01cSRoman Divacky // If there are multiple modules, map the non-canonical globals to their
1311f22ef01cSRoman Divacky // canonical location.
1312f22ef01cSRoman Divacky if (!NonCanonicalGlobals.empty()) {
1313f22ef01cSRoman Divacky for (unsigned i = 0, e = NonCanonicalGlobals.size(); i != e; ++i) {
1314f22ef01cSRoman Divacky const GlobalValue *GV = NonCanonicalGlobals[i];
1315f22ef01cSRoman Divacky const GlobalValue *CGV =
1316f22ef01cSRoman Divacky LinkedGlobalsMap[std::make_pair(GV->getName(), GV->getType())];
1317f22ef01cSRoman Divacky void *Ptr = getPointerToGlobalIfAvailable(CGV);
1318f22ef01cSRoman Divacky assert(Ptr && "Canonical global wasn't codegen'd!");
1319f22ef01cSRoman Divacky addGlobalMapping(GV, Ptr);
1320f22ef01cSRoman Divacky }
1321f22ef01cSRoman Divacky }
1322f22ef01cSRoman Divacky
1323f22ef01cSRoman Divacky // Now that all of the globals are set up in memory, loop through them all
1324f22ef01cSRoman Divacky // and initialize their contents.
132591bc56edSDimitry Andric for (const auto &GV : M.globals()) {
132691bc56edSDimitry Andric if (!GV.isDeclaration()) {
1327f22ef01cSRoman Divacky if (!LinkedGlobalsMap.empty()) {
1328f22ef01cSRoman Divacky if (const GlobalValue *GVEntry =
132991bc56edSDimitry Andric LinkedGlobalsMap[std::make_pair(GV.getName(), GV.getType())])
133091bc56edSDimitry Andric if (GVEntry != &GV) // Not the canonical variable.
1331f22ef01cSRoman Divacky continue;
1332f22ef01cSRoman Divacky }
133391bc56edSDimitry Andric EmitGlobalVariable(&GV);
1334f22ef01cSRoman Divacky }
1335f22ef01cSRoman Divacky }
1336f22ef01cSRoman Divacky }
1337f22ef01cSRoman Divacky }
1338f22ef01cSRoman Divacky
1339f22ef01cSRoman Divacky // EmitGlobalVariable - This method emits the specified global variable to the
1340f22ef01cSRoman Divacky // address specified in GlobalAddresses, or allocates new memory if it's not
1341f22ef01cSRoman Divacky // already in the map.
EmitGlobalVariable(const GlobalVariable * GV)1342f22ef01cSRoman Divacky void ExecutionEngine::EmitGlobalVariable(const GlobalVariable *GV) {
1343f22ef01cSRoman Divacky void *GA = getPointerToGlobalIfAvailable(GV);
1344f22ef01cSRoman Divacky
134591bc56edSDimitry Andric if (!GA) {
1346f22ef01cSRoman Divacky // If it's not already specified, allocate memory for the global.
1347f22ef01cSRoman Divacky GA = getMemoryForGV(GV);
1348f785676fSDimitry Andric
1349f785676fSDimitry Andric // If we failed to allocate memory for this global, return.
135091bc56edSDimitry Andric if (!GA) return;
1351f785676fSDimitry Andric
1352f22ef01cSRoman Divacky addGlobalMapping(GV, GA);
1353f22ef01cSRoman Divacky }
1354f22ef01cSRoman Divacky
1355f22ef01cSRoman Divacky // Don't initialize if it's thread local, let the client do it.
1356f22ef01cSRoman Divacky if (!GV->isThreadLocal())
1357f22ef01cSRoman Divacky InitializeMemory(GV->getInitializer(), GA);
1358f22ef01cSRoman Divacky
13593ca95b02SDimitry Andric Type *ElTy = GV->getValueType();
13607d523365SDimitry Andric size_t GVSize = (size_t)getDataLayout().getTypeAllocSize(ElTy);
1361f22ef01cSRoman Divacky NumInitBytes += (unsigned)GVSize;
1362f22ef01cSRoman Divacky ++NumGlobals;
1363f22ef01cSRoman Divacky }
1364