1857c21b4SMisha Brukman //===-- ExecutionEngine.cpp - Common Implementation shared by EEs ---------===//
2996fe010SChris Lattner //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6482202a6SJohn Criswell //
7482202a6SJohn Criswell //===----------------------------------------------------------------------===//
8482202a6SJohn Criswell //
9996fe010SChris Lattner // This file defines the common interface used by the various execution engine
10996fe010SChris Lattner // subclasses.
11996fe010SChris Lattner //
129b500e56SChristopher Tetreault // FIXME: This file needs to be updated to support scalable vectors
139b500e56SChristopher Tetreault //
14996fe010SChris Lattner //===----------------------------------------------------------------------===//
15996fe010SChris Lattner
166bf87df5SJeffrey Yasskin #include "llvm/ExecutionEngine/ExecutionEngine.h"
170a446fd5SBenjamin Kramer #include "llvm/ADT/STLExtras.h"
18868e3f09SDaniel Dunbar #include "llvm/ADT/SmallString.h"
19390d78b3SChris Lattner #include "llvm/ADT/Statistic.h"
20ed0881b2SChandler Carruth #include "llvm/ExecutionEngine/GenericValue.h"
21b5c7b1ffSLang Hames #include "llvm/ExecutionEngine/JITEventListener.h"
223301c7eeSLang Hames #include "llvm/ExecutionEngine/ObjectCache.h"
23633fe146SLang Hames #include "llvm/ExecutionEngine/RTDyldMemoryManager.h"
249fb823bbSChandler Carruth #include "llvm/IR/Constants.h"
259fb823bbSChandler Carruth #include "llvm/IR/DataLayout.h"
269fb823bbSChandler Carruth #include "llvm/IR/DerivedTypes.h"
273dac3f7fSLang Hames #include "llvm/IR/Mangler.h"
289fb823bbSChandler Carruth #include "llvm/IR/Module.h"
299fb823bbSChandler Carruth #include "llvm/IR/Operator.h"
304220e9c1SChandler Carruth #include "llvm/IR/ValueHandle.h"
3189b57061SReid Kleckner #include "llvm/MC/TargetRegistry.h"
32dd39657aSRafael Espindola #include "llvm/Object/Archive.h"
3335907d8eSDavid Blaikie #include "llvm/Object/ObjectFile.h"
347c16caa3SReid Spencer #include "llvm/Support/Debug.h"
35ed0881b2SChandler Carruth #include "llvm/Support/DynamicLibrary.h"
366c2d233eSTorok Edwin #include "llvm/Support/ErrorHandling.h"
37ed0881b2SChandler Carruth #include "llvm/Support/Host.h"
38ccb29cd2STorok Edwin #include "llvm/Support/raw_ostream.h"
398418fdcdSDylan Noblesmith #include "llvm/Target/TargetMachine.h"
40579f0713SAnton Korobeynikov #include <cmath>
41579f0713SAnton Korobeynikov #include <cstring>
423d5360a4SBenjamin Kramer #include <mutex>
4329681deeSChris Lattner using namespace llvm;
44996fe010SChris Lattner
45f58e376dSChandler Carruth #define DEBUG_TYPE "jit"
46f58e376dSChandler Carruth
47c346ecd7SChris Lattner STATISTIC(NumInitBytes, "Number of bytes of global vars initialized");
48c346ecd7SChris Lattner STATISTIC(NumGlobals , "Number of global vars initialized");
49996fe010SChris Lattner
5070ff8b05SDaniel Dunbar ExecutionEngine *(*ExecutionEngine::MCJITCtor)(
51196e323cSDavid Blaikie std::unique_ptr<Module> M, std::string *ErrorStr,
52633fe146SLang Hames std::shared_ptr<MCJITMemoryManager> MemMgr,
53b72f4845SLang Hames std::shared_ptr<LegacyJITSymbolResolver> Resolver,
544a5697e6SLang Hames std::unique_ptr<TargetMachine> TM) = nullptr;
5593de2a12SLang Hames
562a8a2795SRafael Espindola ExecutionEngine *(*ExecutionEngine::InterpCtor)(std::unique_ptr<Module> M,
572617dcceSCraig Topper std::string *ErrorStr) =nullptr;
582d52c1b8SChris Lattner
anchor()597ea98e14SLang Hames void JITEventListener::anchor() {}
607ea98e14SLang Hames
anchor()613301c7eeSLang Hames void ObjectCache::anchor() {}
623301c7eeSLang Hames
Init(std::unique_ptr<Module> M)63a3fcefb6SMehdi Amini void ExecutionEngine::Init(std::unique_ptr<Module> M) {
644567db45SJeffrey Yasskin CompilingLazily = false;
65cdc0060eSEvan Cheng GVCompilationDisabled = false;
6684a9055eSEvan Cheng SymbolSearchingDisabled = false;
67bc876017SLang Hames
68bc876017SLang Hames // IR module verification is enabled by default in debug builds, and disabled
69bc876017SLang Hames // by default in release builds.
70bc876017SLang Hames #ifndef NDEBUG
71bc876017SLang Hames VerifyModules = true;
72bc876017SLang Hames #else
73bc876017SLang Hames VerifyModules = false;
74bc876017SLang Hames #endif
75bc876017SLang Hames
76091217beSJeffrey Yasskin assert(M && "Module is null?");
772a8a2795SRafael Espindola Modules.push_back(std::move(M));
78260b0c88SMisha Brukman }
79260b0c88SMisha Brukman
ExecutionEngine(std::unique_ptr<Module> M)80a3fcefb6SMehdi Amini ExecutionEngine::ExecutionEngine(std::unique_ptr<Module> M)
81a3fcefb6SMehdi Amini : DL(M->getDataLayout()), LazyFunctionCreator(nullptr) {
82a3fcefb6SMehdi Amini Init(std::move(M));
83a3fcefb6SMehdi Amini }
84a3fcefb6SMehdi Amini
ExecutionEngine(DataLayout DL,std::unique_ptr<Module> M)85a3fcefb6SMehdi Amini ExecutionEngine::ExecutionEngine(DataLayout DL, std::unique_ptr<Module> M)
86a3fcefb6SMehdi Amini : DL(std::move(DL)), LazyFunctionCreator(nullptr) {
87a3fcefb6SMehdi Amini Init(std::move(M));
88a3fcefb6SMehdi Amini }
89a3fcefb6SMehdi Amini
~ExecutionEngine()9092f8b30dSBrian Gaeke ExecutionEngine::~ExecutionEngine() {
91603682adSReid Spencer clearAllGlobalMappings();
9292f8b30dSBrian Gaeke }
9392f8b30dSBrian Gaeke
94a4044332SJeffrey Yasskin namespace {
955f8f34e4SAdrian Prantl /// Helper class which uses a value handler to automatically deletes the
96868e3f09SDaniel Dunbar /// memory block when the GlobalVariable is destroyed.
97774b584fSDavid Blaikie class GVMemoryBlock final : public CallbackVH {
GVMemoryBlock(const GlobalVariable * GV)98a4044332SJeffrey Yasskin GVMemoryBlock(const GlobalVariable *GV)
99a4044332SJeffrey Yasskin : CallbackVH(const_cast<GlobalVariable*>(GV)) {}
100a4044332SJeffrey Yasskin
101a4044332SJeffrey Yasskin public:
1025f8f34e4SAdrian Prantl /// Returns the address the GlobalVariable should be written into. The
103868e3f09SDaniel Dunbar /// GVMemoryBlock object prefixes that.
Create(const GlobalVariable * GV,const DataLayout & TD)104cdfe20b9SMicah Villmow static char *Create(const GlobalVariable *GV, const DataLayout& TD) {
1055f6eaac6SManuel Jacob Type *ElTy = GV->getValueType();
106a4044332SJeffrey Yasskin size_t GVSize = (size_t)TD.getTypeAllocSize(ElTy);
107a4044332SJeffrey Yasskin void *RawMemory = ::operator new(
108368a5e3aSGuillaume Chatelet alignTo(sizeof(GVMemoryBlock), TD.getPreferredAlign(GV)) + GVSize);
109a4044332SJeffrey Yasskin new(RawMemory) GVMemoryBlock(GV);
110a4044332SJeffrey Yasskin return static_cast<char*>(RawMemory) + sizeof(GVMemoryBlock);
111a4044332SJeffrey Yasskin }
112a4044332SJeffrey Yasskin
deleted()113b51ff603SCraig Topper void deleted() override {
114a4044332SJeffrey Yasskin // We allocated with operator new and with some extra memory hanging off the
115a4044332SJeffrey Yasskin // end, so don't just delete this. I'm not sure if this is actually
116a4044332SJeffrey Yasskin // required.
117a4044332SJeffrey Yasskin this->~GVMemoryBlock();
118a4044332SJeffrey Yasskin ::operator delete(this);
119a4044332SJeffrey Yasskin }
120a4044332SJeffrey Yasskin };
121a4044332SJeffrey Yasskin } // anonymous namespace
122a4044332SJeffrey Yasskin
getMemoryForGV(const GlobalVariable * GV)123a4044332SJeffrey Yasskin char *ExecutionEngine::getMemoryForGV(const GlobalVariable *GV) {
124a3fcefb6SMehdi Amini return GVMemoryBlock::Create(GV, getDataLayout());
1255457ce9aSNicolas Geoffray }
1265457ce9aSNicolas Geoffray
addObjectFile(std::unique_ptr<object::ObjectFile> O)12735907d8eSDavid Blaikie void ExecutionEngine::addObjectFile(std::unique_ptr<object::ObjectFile> O) {
12835907d8eSDavid Blaikie llvm_unreachable("ExecutionEngine subclass doesn't implement addObjectFile.");
12935907d8eSDavid Blaikie }
13035907d8eSDavid Blaikie
1317271c194SRafael Espindola void
addObjectFile(object::OwningBinary<object::ObjectFile> O)1327271c194SRafael Espindola ExecutionEngine::addObjectFile(object::OwningBinary<object::ObjectFile> O) {
1337271c194SRafael Espindola llvm_unreachable("ExecutionEngine subclass doesn't implement addObjectFile.");
1347271c194SRafael Espindola }
1357271c194SRafael Espindola
addArchive(object::OwningBinary<object::Archive> A)13648af1c2aSRafael Espindola void ExecutionEngine::addArchive(object::OwningBinary<object::Archive> A) {
137acfd6289SRafael Espindola llvm_unreachable("ExecutionEngine subclass doesn't implement addArchive.");
138acfd6289SRafael Espindola }
139acfd6289SRafael Espindola
removeModule(Module * M)140091217beSJeffrey Yasskin bool ExecutionEngine::removeModule(Module *M) {
1412a8a2795SRafael Espindola for (auto I = Modules.begin(), E = Modules.end(); I != E; ++I) {
1422a8a2795SRafael Espindola Module *Found = I->get();
143091217beSJeffrey Yasskin if (Found == M) {
1442a8a2795SRafael Espindola I->release();
145324fe890SDevang Patel Modules.erase(I);
146091217beSJeffrey Yasskin clearGlobalMappingsFromModule(M);
147091217beSJeffrey Yasskin return true;
148324fe890SDevang Patel }
149324fe890SDevang Patel }
150091217beSJeffrey Yasskin return false;
151617001d8SNate Begeman }
152617001d8SNate Begeman
FindFunctionNamed(StringRef FnName)1537419e940SMehdi Amini Function *ExecutionEngine::FindFunctionNamed(StringRef FnName) {
1540621caefSChris Lattner for (unsigned i = 0, e = Modules.size(); i != e; ++i) {
1555f92a08fSKeno Fischer Function *F = Modules[i]->getFunction(FnName);
1565f92a08fSKeno Fischer if (F && !F->isDeclaration())
1570621caefSChris Lattner return F;
1580621caefSChris Lattner }
1592617dcceSCraig Topper return nullptr;
1600621caefSChris Lattner }
1610621caefSChris Lattner
FindGlobalVariableNamed(StringRef Name,bool AllowInternal)1627419e940SMehdi Amini GlobalVariable *ExecutionEngine::FindGlobalVariableNamed(StringRef Name, bool AllowInternal) {
16373378eb1SKeno Fischer for (unsigned i = 0, e = Modules.size(); i != e; ++i) {
16473378eb1SKeno Fischer GlobalVariable *GV = Modules[i]->getGlobalVariable(Name,AllowInternal);
16573378eb1SKeno Fischer if (GV && !GV->isDeclaration())
16673378eb1SKeno Fischer return GV;
16773378eb1SKeno Fischer }
16873378eb1SKeno Fischer return nullptr;
16973378eb1SKeno Fischer }
1700621caefSChris Lattner
RemoveMapping(StringRef Name)1713dac3f7fSLang Hames uint64_t ExecutionEngineState::RemoveMapping(StringRef Name) {
1723dac3f7fSLang Hames GlobalAddressMapTy::iterator I = GlobalAddressMap.find(Name);
1733dac3f7fSLang Hames uint64_t OldVal;
174868e3f09SDaniel Dunbar
175868e3f09SDaniel Dunbar // FIXME: This is silly, we shouldn't end up with a mapping -> 0 in the
176868e3f09SDaniel Dunbar // GlobalAddressMap.
177307c053fSJeffrey Yasskin if (I == GlobalAddressMap.end())
1783dac3f7fSLang Hames OldVal = 0;
179307c053fSJeffrey Yasskin else {
1803dac3f7fSLang Hames GlobalAddressReverseMap.erase(I->second);
181307c053fSJeffrey Yasskin OldVal = I->second;
182307c053fSJeffrey Yasskin GlobalAddressMap.erase(I);
183307c053fSJeffrey Yasskin }
184307c053fSJeffrey Yasskin
185307c053fSJeffrey Yasskin return OldVal;
186307c053fSJeffrey Yasskin }
187307c053fSJeffrey Yasskin
getMangledName(const GlobalValue * GV)1883dac3f7fSLang Hames std::string ExecutionEngine::getMangledName(const GlobalValue *GV) {
1893393cfdeSLang Hames assert(GV->hasName() && "Global must have name.");
1903393cfdeSLang Hames
1913d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock);
1923dac3f7fSLang Hames SmallString<128> FullName;
1933393cfdeSLang Hames
1943393cfdeSLang Hames const DataLayout &DL =
1953393cfdeSLang Hames GV->getParent()->getDataLayout().isDefault()
1963393cfdeSLang Hames ? getDataLayout()
1973393cfdeSLang Hames : GV->getParent()->getDataLayout();
1983393cfdeSLang Hames
1993393cfdeSLang Hames Mangler::getNameWithPrefix(FullName, GV->getName(), DL);
200adcd0268SBenjamin Kramer return std::string(FullName.str());
2013dac3f7fSLang Hames }
2023dac3f7fSLang Hames
addGlobalMapping(const GlobalValue * GV,void * Addr)2036d8dd189SChris Lattner void ExecutionEngine::addGlobalMapping(const GlobalValue *GV, void *Addr) {
2043d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock);
2053dac3f7fSLang Hames addGlobalMapping(getMangledName(GV), (uint64_t) Addr);
2063dac3f7fSLang Hames }
2076d8dd189SChris Lattner
addGlobalMapping(StringRef Name,uint64_t Addr)2083dac3f7fSLang Hames void ExecutionEngine::addGlobalMapping(StringRef Name, uint64_t Addr) {
2093d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock);
2103dac3f7fSLang Hames
2113dac3f7fSLang Hames assert(!Name.empty() && "Empty GlobalMapping symbol name!");
2123dac3f7fSLang Hames
213d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "JIT: Map \'" << Name << "\' to [" << Addr << "]\n";);
2143dac3f7fSLang Hames uint64_t &CurVal = EEState.getGlobalAddressMap()[Name];
2152617dcceSCraig Topper assert((!CurVal || !Addr) && "GlobalMapping already established!");
2166d8dd189SChris Lattner CurVal = Addr;
2176d8dd189SChris Lattner
218868e3f09SDaniel Dunbar // If we are using the reverse mapping, add it too.
2192f825df6SZachary Turner if (!EEState.getGlobalAddressReverseMap().empty()) {
2203dac3f7fSLang Hames std::string &V = EEState.getGlobalAddressReverseMap()[CurVal];
2213dac3f7fSLang Hames assert((!V.empty() || !Name.empty()) &&
2223dac3f7fSLang Hames "GlobalMapping already established!");
223adcd0268SBenjamin Kramer V = std::string(Name);
2246d8dd189SChris Lattner }
2256d8dd189SChris Lattner }
2266d8dd189SChris Lattner
clearAllGlobalMappings()2276d8dd189SChris Lattner void ExecutionEngine::clearAllGlobalMappings() {
2283d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock);
2296d8dd189SChris Lattner
2302f825df6SZachary Turner EEState.getGlobalAddressMap().clear();
2312f825df6SZachary Turner EEState.getGlobalAddressReverseMap().clear();
2326d8dd189SChris Lattner }
2336d8dd189SChris Lattner
clearGlobalMappingsFromModule(Module * M)2348f83fc4dSNate Begeman void ExecutionEngine::clearGlobalMappingsFromModule(Module *M) {
2353d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock);
2368f83fc4dSNate Begeman
2376d88fde3SPeter Collingbourne for (GlobalObject &GO : M->global_objects())
2386d88fde3SPeter Collingbourne EEState.RemoveMapping(getMangledName(&GO));
2398f83fc4dSNate Begeman }
2408f83fc4dSNate Begeman
updateGlobalMapping(const GlobalValue * GV,void * Addr)2413dac3f7fSLang Hames uint64_t ExecutionEngine::updateGlobalMapping(const GlobalValue *GV,
2423dac3f7fSLang Hames void *Addr) {
2433d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock);
2443dac3f7fSLang Hames return updateGlobalMapping(getMangledName(GV), (uint64_t) Addr);
2453dac3f7fSLang Hames }
2463dac3f7fSLang Hames
updateGlobalMapping(StringRef Name,uint64_t Addr)2473dac3f7fSLang Hames uint64_t ExecutionEngine::updateGlobalMapping(StringRef Name, uint64_t Addr) {
2483d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock);
2496d8dd189SChris Lattner
250d0fc8f80SJeffrey Yasskin ExecutionEngineState::GlobalAddressMapTy &Map =
2512f825df6SZachary Turner EEState.getGlobalAddressMap();
252ee181730SChris Lattner
2536d8dd189SChris Lattner // Deleting from the mapping?
2542617dcceSCraig Topper if (!Addr)
2553dac3f7fSLang Hames return EEState.RemoveMapping(Name);
256ee181730SChris Lattner
2573dac3f7fSLang Hames uint64_t &CurVal = Map[Name];
2583dac3f7fSLang Hames uint64_t OldVal = CurVal;
259ee181730SChris Lattner
2602f825df6SZachary Turner if (CurVal && !EEState.getGlobalAddressReverseMap().empty())
2612f825df6SZachary Turner EEState.getGlobalAddressReverseMap().erase(CurVal);
2626d8dd189SChris Lattner CurVal = Addr;
2636d8dd189SChris Lattner
264868e3f09SDaniel Dunbar // If we are using the reverse mapping, add it too.
2652f825df6SZachary Turner if (!EEState.getGlobalAddressReverseMap().empty()) {
2663dac3f7fSLang Hames std::string &V = EEState.getGlobalAddressReverseMap()[CurVal];
2673dac3f7fSLang Hames assert((!V.empty() || !Name.empty()) &&
2683dac3f7fSLang Hames "GlobalMapping already established!");
269adcd0268SBenjamin Kramer V = std::string(Name);
2706d8dd189SChris Lattner }
271ee181730SChris Lattner return OldVal;
2726d8dd189SChris Lattner }
2736d8dd189SChris Lattner
getAddressToGlobalIfAvailable(StringRef S)2743dac3f7fSLang Hames uint64_t ExecutionEngine::getAddressToGlobalIfAvailable(StringRef S) {
2753d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock);
2763dac3f7fSLang Hames uint64_t Address = 0;
2773dac3f7fSLang Hames ExecutionEngineState::GlobalAddressMapTy::iterator I =
2783dac3f7fSLang Hames EEState.getGlobalAddressMap().find(S);
2793dac3f7fSLang Hames if (I != EEState.getGlobalAddressMap().end())
2803dac3f7fSLang Hames Address = I->second;
2813dac3f7fSLang Hames return Address;
2823dac3f7fSLang Hames }
2833dac3f7fSLang Hames
2843dac3f7fSLang Hames
getPointerToGlobalIfAvailable(StringRef S)2853dac3f7fSLang Hames void *ExecutionEngine::getPointerToGlobalIfAvailable(StringRef S) {
2863d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock);
2873dac3f7fSLang Hames if (void* Address = (void *) getAddressToGlobalIfAvailable(S))
2883dac3f7fSLang Hames return Address;
2893dac3f7fSLang Hames return nullptr;
2903dac3f7fSLang Hames }
2913dac3f7fSLang Hames
getPointerToGlobalIfAvailable(const GlobalValue * GV)2926d8dd189SChris Lattner void *ExecutionEngine::getPointerToGlobalIfAvailable(const GlobalValue *GV) {
2933d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock);
2943dac3f7fSLang Hames return getPointerToGlobalIfAvailable(getMangledName(GV));
2956d8dd189SChris Lattner }
2966d8dd189SChris Lattner
getGlobalValueAtAddress(void * Addr)297748e8579SChris Lattner const GlobalValue *ExecutionEngine::getGlobalValueAtAddress(void *Addr) {
2983d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock);
29979876f52SReid Spencer
300748e8579SChris Lattner // If we haven't computed the reverse mapping yet, do so first.
3012f825df6SZachary Turner if (EEState.getGlobalAddressReverseMap().empty()) {
302d0fc8f80SJeffrey Yasskin for (ExecutionEngineState::GlobalAddressMapTy::iterator
3032f825df6SZachary Turner I = EEState.getGlobalAddressMap().begin(),
3043dac3f7fSLang Hames E = EEState.getGlobalAddressMap().end(); I != E; ++I) {
3053dac3f7fSLang Hames StringRef Name = I->first();
3063dac3f7fSLang Hames uint64_t Addr = I->second;
307a9bc7b83SBenjamin Kramer EEState.getGlobalAddressReverseMap().insert(
308a9bc7b83SBenjamin Kramer std::make_pair(Addr, std::string(Name)));
3093dac3f7fSLang Hames }
310748e8579SChris Lattner }
311748e8579SChris Lattner
3123dac3f7fSLang Hames std::map<uint64_t, std::string>::iterator I =
3133dac3f7fSLang Hames EEState.getGlobalAddressReverseMap().find((uint64_t) Addr);
3143dac3f7fSLang Hames
3153dac3f7fSLang Hames if (I != EEState.getGlobalAddressReverseMap().end()) {
3163dac3f7fSLang Hames StringRef Name = I->second;
3173dac3f7fSLang Hames for (unsigned i = 0, e = Modules.size(); i != e; ++i)
3183dac3f7fSLang Hames if (GlobalValue *GV = Modules[i]->getNamedValue(Name))
3193dac3f7fSLang Hames return GV;
3203dac3f7fSLang Hames }
3213dac3f7fSLang Hames return nullptr;
322748e8579SChris Lattner }
3235a0d4829SChris Lattner
324bfd38abbSJeffrey Yasskin namespace {
325bfd38abbSJeffrey Yasskin class ArgvArray {
326c4a9942aSDylan Noblesmith std::unique_ptr<char[]> Array;
327c4a9942aSDylan Noblesmith std::vector<std::unique_ptr<char[]>> Values;
328bfd38abbSJeffrey Yasskin public:
329bfd38abbSJeffrey Yasskin /// Turn a vector of strings into a nice argv style array of pointers to null
330bfd38abbSJeffrey Yasskin /// terminated strings.
331bfd38abbSJeffrey Yasskin void *reset(LLVMContext &C, ExecutionEngine *EE,
332bfd38abbSJeffrey Yasskin const std::vector<std::string> &InputArgv);
333bfd38abbSJeffrey Yasskin };
334bfd38abbSJeffrey Yasskin } // anonymous namespace
reset(LLVMContext & C,ExecutionEngine * EE,const std::vector<std::string> & InputArgv)335bfd38abbSJeffrey Yasskin void *ArgvArray::reset(LLVMContext &C, ExecutionEngine *EE,
3365a0d4829SChris Lattner const std::vector<std::string> &InputArgv) {
337c4a9942aSDylan Noblesmith Values.clear(); // Free the old contents.
338c4a9942aSDylan Noblesmith Values.reserve(InputArgv.size());
339a3fcefb6SMehdi Amini unsigned PtrSize = EE->getDataLayout().getPointerSize();
3400eaee545SJonas Devlieghere Array = std::make_unique<char[]>((InputArgv.size()+1)*PtrSize);
3415a0d4829SChris Lattner
342d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "JIT: ARGV = " << (void *)Array.get() << "\n");
343229907cdSChris Lattner Type *SBytePtr = Type::getInt8PtrTy(C);
3445a0d4829SChris Lattner
3455a0d4829SChris Lattner for (unsigned i = 0; i != InputArgv.size(); ++i) {
3465a0d4829SChris Lattner unsigned Size = InputArgv[i].size()+1;
3470eaee545SJonas Devlieghere auto Dest = std::make_unique<char[]>(Size);
348d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "JIT: ARGV[" << i << "] = " << (void *)Dest.get()
349d34e60caSNicola Zaghen << "\n");
3505a0d4829SChris Lattner
3514b535d19SDylan Noblesmith std::copy(InputArgv[i].begin(), InputArgv[i].end(), Dest.get());
3525a0d4829SChris Lattner Dest[Size-1] = 0;
3535a0d4829SChris Lattner
354bfd38abbSJeffrey Yasskin // Endian safe: Array[i] = (PointerTy)Dest;
3554b535d19SDylan Noblesmith EE->StoreValueToMemory(PTOGV(Dest.get()),
3564b535d19SDylan Noblesmith (GenericValue*)(&Array[i*PtrSize]), SBytePtr);
3574b535d19SDylan Noblesmith Values.push_back(std::move(Dest));
3585a0d4829SChris Lattner }
3595a0d4829SChris Lattner
3605a0d4829SChris Lattner // Null terminate it
3612617dcceSCraig Topper EE->StoreValueToMemory(PTOGV(nullptr),
362c4a9942aSDylan Noblesmith (GenericValue*)(&Array[InputArgv.size()*PtrSize]),
3635a0d4829SChris Lattner SBytePtr);
364c4a9942aSDylan Noblesmith return Array.get();
3655a0d4829SChris Lattner }
3665a0d4829SChris Lattner
runStaticConstructorsDestructors(Module & module,bool isDtors)3672a8a2795SRafael Espindola void ExecutionEngine::runStaticConstructorsDestructors(Module &module,
36841fa2bd1SChris Lattner bool isDtors) {
3697419e940SMehdi Amini StringRef Name(isDtors ? "llvm.global_dtors" : "llvm.global_ctors");
3702a8a2795SRafael Espindola GlobalVariable *GV = module.getNamedGlobal(Name);
371fe36eaebSChris Lattner
372fe36eaebSChris Lattner // If this global has internal linkage, or if it has a use, then it must be
373fe36eaebSChris Lattner // an old-style (llvmgcc3) static ctor with __main linked in and in use. If
3740621caefSChris Lattner // this is the case, don't execute any of the global ctors, __main will do
3750621caefSChris Lattner // it.
3766de96a1bSRafael Espindola if (!GV || GV->isDeclaration() || GV->hasLocalLinkage()) return;
377faae50b6SChris Lattner
3780cbfcb2bSNick Lewycky // Should be an array of '{ i32, void ()* }' structs. The first value is
3790621caefSChris Lattner // the init priority, which we ignore.
38000245f42SChris Lattner ConstantArray *InitList = dyn_cast<ConstantArray>(GV->getInitializer());
3812617dcceSCraig Topper if (!InitList)
3820f857898SNick Lewycky return;
383868e3f09SDaniel Dunbar for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i) {
38400245f42SChris Lattner ConstantStruct *CS = dyn_cast<ConstantStruct>(InitList->getOperand(i));
3852617dcceSCraig Topper if (!CS) continue;
386faae50b6SChris Lattner
387faae50b6SChris Lattner Constant *FP = CS->getOperand(1);
388faae50b6SChris Lattner if (FP->isNullValue())
3890f857898SNick Lewycky continue; // Found a sentinal value, ignore.
390faae50b6SChris Lattner
391868e3f09SDaniel Dunbar // Strip off constant expression casts.
392faae50b6SChris Lattner if (ConstantExpr *CE = dyn_cast<ConstantExpr>(FP))
3936c38f0bbSReid Spencer if (CE->isCast())
394faae50b6SChris Lattner FP = CE->getOperand(0);
395868e3f09SDaniel Dunbar
396faae50b6SChris Lattner // Execute the ctor/dtor function!
397868e3f09SDaniel Dunbar if (Function *F = dyn_cast<Function>(FP))
398bd7b1c89SBenjamin Kramer runFunction(F, None);
399868e3f09SDaniel Dunbar
400868e3f09SDaniel Dunbar // FIXME: It is marginally lame that we just do nothing here if we see an
401868e3f09SDaniel Dunbar // entry we don't recognize. It might not be unreasonable for the verifier
402868e3f09SDaniel Dunbar // to not even allow this and just assert here.
403faae50b6SChris Lattner }
404faae50b6SChris Lattner }
4051a9a0b7bSEvan Cheng
runStaticConstructorsDestructors(bool isDtors)4061a9a0b7bSEvan Cheng void ExecutionEngine::runStaticConstructorsDestructors(bool isDtors) {
4071a9a0b7bSEvan Cheng // Execute global ctors/dtors for each module in the program.
4082a8a2795SRafael Espindola for (std::unique_ptr<Module> &M : Modules)
4092a8a2795SRafael Espindola runStaticConstructorsDestructors(*M, isDtors);
4100621caefSChris Lattner }
411faae50b6SChris Lattner
412cf3e3017SDan Gohman #ifndef NDEBUG
4131202d1b1SDuncan Sands /// isTargetNullPtr - Return whether the target pointer stored at Loc is null.
isTargetNullPtr(ExecutionEngine * EE,void * Loc)4141202d1b1SDuncan Sands static bool isTargetNullPtr(ExecutionEngine *EE, void *Loc) {
415a3fcefb6SMehdi Amini unsigned PtrSize = EE->getDataLayout().getPointerSize();
4161202d1b1SDuncan Sands for (unsigned i = 0; i < PtrSize; ++i)
4171202d1b1SDuncan Sands if (*(i + (uint8_t*)Loc))
4181202d1b1SDuncan Sands return false;
4191202d1b1SDuncan Sands return true;
4201202d1b1SDuncan Sands }
421cf3e3017SDan Gohman #endif
4221202d1b1SDuncan Sands
runFunctionAsMain(Function * Fn,const std::vector<std::string> & argv,const char * const * envp)4235a0d4829SChris Lattner int ExecutionEngine::runFunctionAsMain(Function *Fn,
4245a0d4829SChris Lattner const std::vector<std::string> &argv,
4255a0d4829SChris Lattner const char * const * envp) {
4265a0d4829SChris Lattner std::vector<GenericValue> GVArgs;
4275a0d4829SChris Lattner GenericValue GVArgc;
42887aa65f4SReid Spencer GVArgc.IntVal = APInt(32, argv.size());
4298c32c111SAnton Korobeynikov
4308c32c111SAnton Korobeynikov // Check main() type
431b1cad0b3SChris Lattner unsigned NumArgs = Fn->getFunctionType()->getNumParams();
432229907cdSChris Lattner FunctionType *FTy = Fn->getFunctionType();
433229907cdSChris Lattner Type* PPInt8Ty = Type::getInt8PtrTy(Fn->getContext())->getPointerTo();
434868e3f09SDaniel Dunbar
435868e3f09SDaniel Dunbar // Check the argument types.
436868e3f09SDaniel Dunbar if (NumArgs > 3)
4372104b8d3SChris Lattner report_fatal_error("Invalid number of arguments of main() supplied");
438868e3f09SDaniel Dunbar if (NumArgs >= 3 && FTy->getParamType(2) != PPInt8Ty)
439868e3f09SDaniel Dunbar report_fatal_error("Invalid type for third argument of main() supplied");
440868e3f09SDaniel Dunbar if (NumArgs >= 2 && FTy->getParamType(1) != PPInt8Ty)
441868e3f09SDaniel Dunbar report_fatal_error("Invalid type for second argument of main() supplied");
442868e3f09SDaniel Dunbar if (NumArgs >= 1 && !FTy->getParamType(0)->isIntegerTy(32))
443868e3f09SDaniel Dunbar report_fatal_error("Invalid type for first argument of main() supplied");
444868e3f09SDaniel Dunbar if (!FTy->getReturnType()->isIntegerTy() &&
445868e3f09SDaniel Dunbar !FTy->getReturnType()->isVoidTy())
446868e3f09SDaniel Dunbar report_fatal_error("Invalid return type of main() supplied");
4478c32c111SAnton Korobeynikov
448bfd38abbSJeffrey Yasskin ArgvArray CArgv;
449bfd38abbSJeffrey Yasskin ArgvArray CEnv;
450b1cad0b3SChris Lattner if (NumArgs) {
4515a0d4829SChris Lattner GVArgs.push_back(GVArgc); // Arg #0 = argc.
452b1cad0b3SChris Lattner if (NumArgs > 1) {
45355f1c09eSOwen Anderson // Arg #1 = argv.
454bfd38abbSJeffrey Yasskin GVArgs.push_back(PTOGV(CArgv.reset(Fn->getContext(), this, argv)));
4551202d1b1SDuncan Sands assert(!isTargetNullPtr(this, GVTOP(GVArgs[1])) &&
456b1cad0b3SChris Lattner "argv[0] was null after CreateArgv");
457b1cad0b3SChris Lattner if (NumArgs > 2) {
4585a0d4829SChris Lattner std::vector<std::string> EnvVars;
4595a0d4829SChris Lattner for (unsigned i = 0; envp[i]; ++i)
460f5e2fc47SBenjamin Kramer EnvVars.emplace_back(envp[i]);
46155f1c09eSOwen Anderson // Arg #2 = envp.
462bfd38abbSJeffrey Yasskin GVArgs.push_back(PTOGV(CEnv.reset(Fn->getContext(), this, EnvVars)));
463b1cad0b3SChris Lattner }
464b1cad0b3SChris Lattner }
465b1cad0b3SChris Lattner }
466868e3f09SDaniel Dunbar
46787aa65f4SReid Spencer return runFunction(Fn, GVArgs).IntVal.getZExtValue();
4685a0d4829SChris Lattner }
4695a0d4829SChris Lattner
EngineBuilder()470298a3a05SBenjamin Kramer EngineBuilder::EngineBuilder() : EngineBuilder(nullptr) {}
47193de2a12SLang Hames
EngineBuilder(std::unique_ptr<Module> M)4724a5697e6SLang Hames EngineBuilder::EngineBuilder(std::unique_ptr<Module> M)
473298a3a05SBenjamin Kramer : M(std::move(M)), WhichEngine(EngineKind::Either), ErrorStr(nullptr),
4746154c411SLang Hames OptLevel(CodeGenOpt::Default), MemMgr(nullptr), Resolver(nullptr) {
475322db9eaSAlp Toker // IR module verification is enabled by default in debug builds, and disabled
476322db9eaSAlp Toker // by default in release builds.
477322db9eaSAlp Toker #ifndef NDEBUG
478322db9eaSAlp Toker VerifyModules = true;
479322db9eaSAlp Toker #else
480322db9eaSAlp Toker VerifyModules = false;
481322db9eaSAlp Toker #endif
482322db9eaSAlp Toker }
483322db9eaSAlp Toker
484298a3a05SBenjamin Kramer EngineBuilder::~EngineBuilder() = default;
485298a3a05SBenjamin Kramer
setMCJITMemoryManager(std::unique_ptr<RTDyldMemoryManager> mcjmm)486298a3a05SBenjamin Kramer EngineBuilder &EngineBuilder::setMCJITMemoryManager(
487298a3a05SBenjamin Kramer std::unique_ptr<RTDyldMemoryManager> mcjmm) {
488633fe146SLang Hames auto SharedMM = std::shared_ptr<RTDyldMemoryManager>(std::move(mcjmm));
489633fe146SLang Hames MemMgr = SharedMM;
490633fe146SLang Hames Resolver = SharedMM;
491633fe146SLang Hames return *this;
492633fe146SLang Hames }
493633fe146SLang Hames
494633fe146SLang Hames EngineBuilder&
setMemoryManager(std::unique_ptr<MCJITMemoryManager> MM)495633fe146SLang Hames EngineBuilder::setMemoryManager(std::unique_ptr<MCJITMemoryManager> MM) {
496633fe146SLang Hames MemMgr = std::shared_ptr<MCJITMemoryManager>(std::move(MM));
497633fe146SLang Hames return *this;
498633fe146SLang Hames }
499633fe146SLang Hames
500633fe146SLang Hames EngineBuilder &
setSymbolResolver(std::unique_ptr<LegacyJITSymbolResolver> SR)501b72f4845SLang Hames EngineBuilder::setSymbolResolver(std::unique_ptr<LegacyJITSymbolResolver> SR) {
502b72f4845SLang Hames Resolver = std::shared_ptr<LegacyJITSymbolResolver>(std::move(SR));
503298a3a05SBenjamin Kramer return *this;
504298a3a05SBenjamin Kramer }
505298a3a05SBenjamin Kramer
create(TargetMachine * TM)506add6f1d2SOwen Anderson ExecutionEngine *EngineBuilder::create(TargetMachine *TM) {
50756440fd8SAhmed Charles std::unique_ptr<TargetMachine> TheTM(TM); // Take ownership.
50825a3d816SBenjamin Kramer
509a53414fdSNick Lewycky // Make sure we can resolve symbols in the program as well. The zero arg
510a53414fdSNick Lewycky // to the function tells DynamicLibrary to load the program, not a library.
5112617dcceSCraig Topper if (sys::DynamicLibrary::LoadLibraryPermanently(nullptr, ErrorStr))
5122617dcceSCraig Topper return nullptr;
513a53414fdSNick Lewycky
514fc8a2d5aSReid Kleckner // If the user specified a memory manager but didn't specify which engine to
515fc8a2d5aSReid Kleckner // create, we assume they only want the JIT, and we fail if they only want
516fc8a2d5aSReid Kleckner // the interpreter.
517633fe146SLang Hames if (MemMgr) {
51841fa2bd1SChris Lattner if (WhichEngine & EngineKind::JIT)
519fc8a2d5aSReid Kleckner WhichEngine = EngineKind::JIT;
52041fa2bd1SChris Lattner else {
5218bcc6445SChris Lattner if (ErrorStr)
522fc8a2d5aSReid Kleckner *ErrorStr = "Cannot create an interpreter with a memory manager.";
5232617dcceSCraig Topper return nullptr;
524fc8a2d5aSReid Kleckner }
5254bd3bd5bSBrian Gaeke }
5264bd3bd5bSBrian Gaeke
527fc8a2d5aSReid Kleckner // Unless the interpreter was explicitly selected or the JIT is not linked,
528fc8a2d5aSReid Kleckner // try making a JIT.
52925a3d816SBenjamin Kramer if ((WhichEngine & EngineKind::JIT) && TheTM) {
5307f26246aSDylan Noblesmith if (!TM->getTarget().hasJIT()) {
5317f26246aSDylan Noblesmith errs() << "WARNING: This target JIT is not designed for the host"
5327f26246aSDylan Noblesmith << " you are running. If bad things happen, please choose"
5337f26246aSDylan Noblesmith << " a different -march switch.\n";
5347f26246aSDylan Noblesmith }
5357f26246aSDylan Noblesmith
536bc876017SLang Hames ExecutionEngine *EE = nullptr;
5376154c411SLang Hames if (ExecutionEngine::MCJITCtor)
538633fe146SLang Hames EE = ExecutionEngine::MCJITCtor(std::move(M), ErrorStr, std::move(MemMgr),
539633fe146SLang Hames std::move(Resolver), std::move(TheTM));
54093de2a12SLang Hames
541bc876017SLang Hames if (EE) {
542bc876017SLang Hames EE->setVerifyModules(VerifyModules);
543bc876017SLang Hames return EE;
54441fa2bd1SChris Lattner }
545fc8a2d5aSReid Kleckner }
546fc8a2d5aSReid Kleckner
547fc8a2d5aSReid Kleckner // If we can't make a JIT and we didn't request one specifically, try making
548fc8a2d5aSReid Kleckner // an interpreter instead.
54941fa2bd1SChris Lattner if (WhichEngine & EngineKind::Interpreter) {
55041fa2bd1SChris Lattner if (ExecutionEngine::InterpCtor)
5512a8a2795SRafael Espindola return ExecutionEngine::InterpCtor(std::move(M), ErrorStr);
5528bcc6445SChris Lattner if (ErrorStr)
55341fa2bd1SChris Lattner *ErrorStr = "Interpreter has not been linked in.";
5542617dcceSCraig Topper return nullptr;
555fc8a2d5aSReid Kleckner }
556fc8a2d5aSReid Kleckner
55779cc1e3aSEric Christopher if ((WhichEngine & EngineKind::JIT) && !ExecutionEngine::MCJITCtor) {
5588bcc6445SChris Lattner if (ErrorStr)
5598bcc6445SChris Lattner *ErrorStr = "JIT has not been linked in.";
5608bcc6445SChris Lattner }
561868e3f09SDaniel Dunbar
5622617dcceSCraig Topper return nullptr;
563b5163bb9SChris Lattner }
564b5163bb9SChris Lattner
getPointerToGlobal(const GlobalValue * GV)565996fe010SChris Lattner void *ExecutionEngine::getPointerToGlobal(const GlobalValue *GV) {
5661678e859SBrian Gaeke if (Function *F = const_cast<Function*>(dyn_cast<Function>(GV)))
567996fe010SChris Lattner return getPointerToFunction(F);
568996fe010SChris Lattner
5693d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock);
5703dac3f7fSLang Hames if (void* P = getPointerToGlobalIfAvailable(GV))
571868e3f09SDaniel Dunbar return P;
57269e84901SJeff Cohen
57369e84901SJeff Cohen // Global variable might have been added since interpreter started.
57469e84901SJeff Cohen if (GlobalVariable *GVar =
57569e84901SJeff Cohen const_cast<GlobalVariable *>(dyn_cast<GlobalVariable>(GV)))
5761d49eb00SFangrui Song emitGlobalVariable(GVar);
57769e84901SJeff Cohen else
578fbcc663cSTorok Edwin llvm_unreachable("Global hasn't had an address allocated yet!");
579868e3f09SDaniel Dunbar
5803dac3f7fSLang Hames return getPointerToGlobalIfAvailable(GV);
581996fe010SChris Lattner }
582996fe010SChris Lattner
5835f8f34e4SAdrian Prantl /// Converts a Constant* into a GenericValue, including handling of
584868e3f09SDaniel Dunbar /// ConstantExpr values.
getConstantValue(const Constant * C)585996fe010SChris Lattner GenericValue ExecutionEngine::getConstantValue(const Constant *C) {
5866c38f0bbSReid Spencer // If its undefined, return the garbage.
587bcbdbfb3SJay Foad if (isa<UndefValue>(C)) {
588bcbdbfb3SJay Foad GenericValue Result;
589bcbdbfb3SJay Foad switch (C->getType()->getTypeID()) {
590be79a7acSNadav Rotem default:
591be79a7acSNadav Rotem break;
592bcbdbfb3SJay Foad case Type::IntegerTyID:
593bcbdbfb3SJay Foad case Type::X86_FP80TyID:
594bcbdbfb3SJay Foad case Type::FP128TyID:
595bcbdbfb3SJay Foad case Type::PPC_FP128TyID:
596bcbdbfb3SJay Foad // Although the value is undefined, we still have to construct an APInt
597bcbdbfb3SJay Foad // with the correct bit width.
598bcbdbfb3SJay Foad Result.IntVal = APInt(C->getType()->getPrimitiveSizeInBits(), 0);
599bcbdbfb3SJay Foad break;
6008e97f016SElena Demikhovsky case Type::StructTyID: {
6018e97f016SElena Demikhovsky // if the whole struct is 'undef' just reserve memory for the value.
6028e97f016SElena Demikhovsky if(StructType *STy = dyn_cast<StructType>(C->getType())) {
6038e97f016SElena Demikhovsky unsigned int elemNum = STy->getNumElements();
6048e97f016SElena Demikhovsky Result.AggregateVal.resize(elemNum);
6058e97f016SElena Demikhovsky for (unsigned int i = 0; i < elemNum; ++i) {
6068e97f016SElena Demikhovsky Type *ElemTy = STy->getElementType(i);
6078e97f016SElena Demikhovsky if (ElemTy->isIntegerTy())
6088e97f016SElena Demikhovsky Result.AggregateVal[i].IntVal =
6098e97f016SElena Demikhovsky APInt(ElemTy->getPrimitiveSizeInBits(), 0);
6108e97f016SElena Demikhovsky else if (ElemTy->isAggregateType()) {
6118e97f016SElena Demikhovsky const Constant *ElemUndef = UndefValue::get(ElemTy);
6128e97f016SElena Demikhovsky Result.AggregateVal[i] = getConstantValue(ElemUndef);
6138e97f016SElena Demikhovsky }
6148e97f016SElena Demikhovsky }
6158e97f016SElena Demikhovsky }
6168e97f016SElena Demikhovsky }
6178e97f016SElena Demikhovsky break;
6182dea3f12SChristopher Tetreault case Type::ScalableVectorTyID:
6199b500e56SChristopher Tetreault report_fatal_error(
6209b500e56SChristopher Tetreault "Scalable vector support not yet implemented in ExecutionEngine");
6219b500e56SChristopher Tetreault case Type::FixedVectorTyID:
622be79a7acSNadav Rotem // if the whole vector is 'undef' just reserve memory for the value.
6239b500e56SChristopher Tetreault auto *VTy = cast<FixedVectorType>(C->getType());
624e3dcce97SCraig Topper Type *ElemTy = VTy->getElementType();
625be79a7acSNadav Rotem unsigned int elemNum = VTy->getNumElements();
626be79a7acSNadav Rotem Result.AggregateVal.resize(elemNum);
627be79a7acSNadav Rotem if (ElemTy->isIntegerTy())
628be79a7acSNadav Rotem for (unsigned int i = 0; i < elemNum; ++i)
629be79a7acSNadav Rotem Result.AggregateVal[i].IntVal =
630be79a7acSNadav Rotem APInt(ElemTy->getPrimitiveSizeInBits(), 0);
631bcbdbfb3SJay Foad break;
632bcbdbfb3SJay Foad }
633bcbdbfb3SJay Foad return Result;
634bcbdbfb3SJay Foad }
6359de0d14dSChris Lattner
636868e3f09SDaniel Dunbar // Otherwise, if the value is a ConstantExpr...
6376c38f0bbSReid Spencer if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) {
6384fd528f2SReid Spencer Constant *Op0 = CE->getOperand(0);
6399de0d14dSChris Lattner switch (CE->getOpcode()) {
6409de0d14dSChris Lattner case Instruction::GetElementPtr: {
6416c38f0bbSReid Spencer // Compute the index
6424fd528f2SReid Spencer GenericValue Result = getConstantValue(Op0);
643a3fcefb6SMehdi Amini APInt Offset(DL.getPointerSizeInBits(), 0);
644a3fcefb6SMehdi Amini cast<GEPOperator>(CE)->accumulateConstantOffset(DL, Offset);
6459de0d14dSChris Lattner
64687aa65f4SReid Spencer char* tmp = (char*) Result.PointerVal;
647b6ad9822SNuno Lopes Result = PTOGV(tmp + Offset.getSExtValue());
6489de0d14dSChris Lattner return Result;
6499de0d14dSChris Lattner }
6504fd528f2SReid Spencer case Instruction::Trunc: {
6514fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0);
6524fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
6534fd528f2SReid Spencer GV.IntVal = GV.IntVal.trunc(BitWidth);
6544fd528f2SReid Spencer return GV;
6554fd528f2SReid Spencer }
6564fd528f2SReid Spencer case Instruction::ZExt: {
6574fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0);
6584fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
6594fd528f2SReid Spencer GV.IntVal = GV.IntVal.zext(BitWidth);
6604fd528f2SReid Spencer return GV;
6614fd528f2SReid Spencer }
6624fd528f2SReid Spencer case Instruction::SExt: {
6634fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0);
6644fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
6654fd528f2SReid Spencer GV.IntVal = GV.IntVal.sext(BitWidth);
6664fd528f2SReid Spencer return GV;
6674fd528f2SReid Spencer }
6684fd528f2SReid Spencer case Instruction::FPTrunc: {
669a1336cf5SDale Johannesen // FIXME long double
6704fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0);
6714fd528f2SReid Spencer GV.FloatVal = float(GV.DoubleVal);
6724fd528f2SReid Spencer return GV;
6734fd528f2SReid Spencer }
6744fd528f2SReid Spencer case Instruction::FPExt:{
675a1336cf5SDale Johannesen // FIXME long double
6764fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0);
6774fd528f2SReid Spencer GV.DoubleVal = double(GV.FloatVal);
6784fd528f2SReid Spencer return GV;
6794fd528f2SReid Spencer }
6804fd528f2SReid Spencer case Instruction::UIToFP: {
6814fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0);
682fdd87907SChris Lattner if (CE->getType()->isFloatTy())
6834fd528f2SReid Spencer GV.FloatVal = float(GV.IntVal.roundToDouble());
684fdd87907SChris Lattner else if (CE->getType()->isDoubleTy())
6854fd528f2SReid Spencer GV.DoubleVal = GV.IntVal.roundToDouble();
686fdd87907SChris Lattner else if (CE->getType()->isX86_FP80Ty()) {
68717c7f703SStephan Bergmann APFloat apf = APFloat::getZero(APFloat::x87DoubleExtended());
688ca24fd90SDan Gohman (void)apf.convertFromAPInt(GV.IntVal,
689ca24fd90SDan Gohman false,
6909150652bSDale Johannesen APFloat::rmNearestTiesToEven);
69154306fe4SDale Johannesen GV.IntVal = apf.bitcastToAPInt();
692a1336cf5SDale Johannesen }
6934fd528f2SReid Spencer return GV;
6944fd528f2SReid Spencer }
6954fd528f2SReid Spencer case Instruction::SIToFP: {
6964fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0);
697fdd87907SChris Lattner if (CE->getType()->isFloatTy())
6984fd528f2SReid Spencer GV.FloatVal = float(GV.IntVal.signedRoundToDouble());
699fdd87907SChris Lattner else if (CE->getType()->isDoubleTy())
7004fd528f2SReid Spencer GV.DoubleVal = GV.IntVal.signedRoundToDouble();
701fdd87907SChris Lattner else if (CE->getType()->isX86_FP80Ty()) {
70217c7f703SStephan Bergmann APFloat apf = APFloat::getZero(APFloat::x87DoubleExtended());
703ca24fd90SDan Gohman (void)apf.convertFromAPInt(GV.IntVal,
704ca24fd90SDan Gohman true,
7059150652bSDale Johannesen APFloat::rmNearestTiesToEven);
70654306fe4SDale Johannesen GV.IntVal = apf.bitcastToAPInt();
707a1336cf5SDale Johannesen }
7084fd528f2SReid Spencer return GV;
7094fd528f2SReid Spencer }
7104fd528f2SReid Spencer case Instruction::FPToUI: // double->APInt conversion handles sign
7114fd528f2SReid Spencer case Instruction::FPToSI: {
7124fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0);
7134fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
714fdd87907SChris Lattner if (Op0->getType()->isFloatTy())
7154fd528f2SReid Spencer GV.IntVal = APIntOps::RoundFloatToAPInt(GV.FloatVal, BitWidth);
716fdd87907SChris Lattner else if (Op0->getType()->isDoubleTy())
7174fd528f2SReid Spencer GV.IntVal = APIntOps::RoundDoubleToAPInt(GV.DoubleVal, BitWidth);
718fdd87907SChris Lattner else if (Op0->getType()->isX86_FP80Ty()) {
71917c7f703SStephan Bergmann APFloat apf = APFloat(APFloat::x87DoubleExtended(), GV.IntVal);
720a1336cf5SDale Johannesen uint64_t v;
7214f0bd68cSDale Johannesen bool ignored;
72200b34996SSimon Pilgrim (void)apf.convertToInteger(makeMutableArrayRef(v), BitWidth,
723a1336cf5SDale Johannesen CE->getOpcode()==Instruction::FPToSI,
7244f0bd68cSDale Johannesen APFloat::rmTowardZero, &ignored);
725a1336cf5SDale Johannesen GV.IntVal = v; // endian?
726a1336cf5SDale Johannesen }
7274fd528f2SReid Spencer return GV;
7284fd528f2SReid Spencer }
7296c38f0bbSReid Spencer case Instruction::PtrToInt: {
7304fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0);
731a3fcefb6SMehdi Amini uint32_t PtrWidth = DL.getTypeSizeInBits(Op0->getType());
732fc1f2cd3SEli Friedman assert(PtrWidth <= 64 && "Bad pointer width");
7334fd528f2SReid Spencer GV.IntVal = APInt(PtrWidth, uintptr_t(GV.PointerVal));
734a3fcefb6SMehdi Amini uint32_t IntWidth = DL.getTypeSizeInBits(CE->getType());
735fc1f2cd3SEli Friedman GV.IntVal = GV.IntVal.zextOrTrunc(IntWidth);
7364fd528f2SReid Spencer return GV;
7374fd528f2SReid Spencer }
7384fd528f2SReid Spencer case Instruction::IntToPtr: {
7394fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0);
740a3fcefb6SMehdi Amini uint32_t PtrWidth = DL.getTypeSizeInBits(CE->getType());
7414fd528f2SReid Spencer GV.IntVal = GV.IntVal.zextOrTrunc(PtrWidth);
7424fd528f2SReid Spencer assert(GV.IntVal.getBitWidth() <= 64 && "Bad pointer width");
7434fd528f2SReid Spencer GV.PointerVal = PointerTy(uintptr_t(GV.IntVal.getZExtValue()));
7446c38f0bbSReid Spencer return GV;
7456c38f0bbSReid Spencer }
7466c38f0bbSReid Spencer case Instruction::BitCast: {
7474fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0);
748229907cdSChris Lattner Type* DestTy = CE->getType();
7494fd528f2SReid Spencer switch (Op0->getType()->getTypeID()) {
750fbcc663cSTorok Edwin default: llvm_unreachable("Invalid bitcast operand");
7514fd528f2SReid Spencer case Type::IntegerTyID:
7529dff9becSDuncan Sands assert(DestTy->isFloatingPointTy() && "invalid bitcast");
753fdd87907SChris Lattner if (DestTy->isFloatTy())
7544fd528f2SReid Spencer GV.FloatVal = GV.IntVal.bitsToFloat();
755fdd87907SChris Lattner else if (DestTy->isDoubleTy())
7564fd528f2SReid Spencer GV.DoubleVal = GV.IntVal.bitsToDouble();
7576c38f0bbSReid Spencer break;
7584fd528f2SReid Spencer case Type::FloatTyID:
7599dff9becSDuncan Sands assert(DestTy->isIntegerTy(32) && "Invalid bitcast");
7603447fb01SJay Foad GV.IntVal = APInt::floatToBits(GV.FloatVal);
7614fd528f2SReid Spencer break;
7624fd528f2SReid Spencer case Type::DoubleTyID:
7639dff9becSDuncan Sands assert(DestTy->isIntegerTy(64) && "Invalid bitcast");
7643447fb01SJay Foad GV.IntVal = APInt::doubleToBits(GV.DoubleVal);
7654fd528f2SReid Spencer break;
7664fd528f2SReid Spencer case Type::PointerTyID:
76719d0b47bSDuncan Sands assert(DestTy->isPointerTy() && "Invalid bitcast");
7684fd528f2SReid Spencer break; // getConstantValue(Op0) above already converted it
7696c38f0bbSReid Spencer }
7704fd528f2SReid Spencer return GV;
77168cbcc3eSChris Lattner }
77268cbcc3eSChris Lattner case Instruction::Add:
773a5b9645cSDan Gohman case Instruction::FAdd:
7744fd528f2SReid Spencer case Instruction::Sub:
775a5b9645cSDan Gohman case Instruction::FSub:
7764fd528f2SReid Spencer case Instruction::Mul:
777a5b9645cSDan Gohman case Instruction::FMul:
7784fd528f2SReid Spencer case Instruction::UDiv:
7794fd528f2SReid Spencer case Instruction::SDiv:
7804fd528f2SReid Spencer case Instruction::URem:
7814fd528f2SReid Spencer case Instruction::SRem:
7824fd528f2SReid Spencer case Instruction::And:
7834fd528f2SReid Spencer case Instruction::Or:
7844fd528f2SReid Spencer case Instruction::Xor: {
7854fd528f2SReid Spencer GenericValue LHS = getConstantValue(Op0);
7864fd528f2SReid Spencer GenericValue RHS = getConstantValue(CE->getOperand(1));
7874fd528f2SReid Spencer GenericValue GV;
788c4e6bb5fSChris Lattner switch (CE->getOperand(0)->getType()->getTypeID()) {
789fbcc663cSTorok Edwin default: llvm_unreachable("Bad add type!");
7907a9c62baSReid Spencer case Type::IntegerTyID:
7914fd528f2SReid Spencer switch (CE->getOpcode()) {
792fbcc663cSTorok Edwin default: llvm_unreachable("Invalid integer opcode");
7934fd528f2SReid Spencer case Instruction::Add: GV.IntVal = LHS.IntVal + RHS.IntVal; break;
7944fd528f2SReid Spencer case Instruction::Sub: GV.IntVal = LHS.IntVal - RHS.IntVal; break;
7954fd528f2SReid Spencer case Instruction::Mul: GV.IntVal = LHS.IntVal * RHS.IntVal; break;
7964fd528f2SReid Spencer case Instruction::UDiv:GV.IntVal = LHS.IntVal.udiv(RHS.IntVal); break;
7974fd528f2SReid Spencer case Instruction::SDiv:GV.IntVal = LHS.IntVal.sdiv(RHS.IntVal); break;
7984fd528f2SReid Spencer case Instruction::URem:GV.IntVal = LHS.IntVal.urem(RHS.IntVal); break;
7994fd528f2SReid Spencer case Instruction::SRem:GV.IntVal = LHS.IntVal.srem(RHS.IntVal); break;
8004fd528f2SReid Spencer case Instruction::And: GV.IntVal = LHS.IntVal & RHS.IntVal; break;
8014fd528f2SReid Spencer case Instruction::Or: GV.IntVal = LHS.IntVal | RHS.IntVal; break;
8024fd528f2SReid Spencer case Instruction::Xor: GV.IntVal = LHS.IntVal ^ RHS.IntVal; break;
8034fd528f2SReid Spencer }
804c4e6bb5fSChris Lattner break;
805c4e6bb5fSChris Lattner case Type::FloatTyID:
8064fd528f2SReid Spencer switch (CE->getOpcode()) {
807fbcc663cSTorok Edwin default: llvm_unreachable("Invalid float opcode");
808a5b9645cSDan Gohman case Instruction::FAdd:
8094fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal + RHS.FloatVal; break;
810a5b9645cSDan Gohman case Instruction::FSub:
8114fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal - RHS.FloatVal; break;
812a5b9645cSDan Gohman case Instruction::FMul:
8134fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal * RHS.FloatVal; break;
8144fd528f2SReid Spencer case Instruction::FDiv:
8154fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal / RHS.FloatVal; break;
8164fd528f2SReid Spencer case Instruction::FRem:
81793cd0f1cSChris Lattner GV.FloatVal = std::fmod(LHS.FloatVal,RHS.FloatVal); break;
8184fd528f2SReid Spencer }
819c4e6bb5fSChris Lattner break;
820c4e6bb5fSChris Lattner case Type::DoubleTyID:
8214fd528f2SReid Spencer switch (CE->getOpcode()) {
822fbcc663cSTorok Edwin default: llvm_unreachable("Invalid double opcode");
823a5b9645cSDan Gohman case Instruction::FAdd:
8244fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal + RHS.DoubleVal; break;
825a5b9645cSDan Gohman case Instruction::FSub:
8264fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal - RHS.DoubleVal; break;
827a5b9645cSDan Gohman case Instruction::FMul:
8284fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal * RHS.DoubleVal; break;
8294fd528f2SReid Spencer case Instruction::FDiv:
8304fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal / RHS.DoubleVal; break;
8314fd528f2SReid Spencer case Instruction::FRem:
83293cd0f1cSChris Lattner GV.DoubleVal = std::fmod(LHS.DoubleVal,RHS.DoubleVal); break;
8334fd528f2SReid Spencer }
834c4e6bb5fSChris Lattner break;
835a1336cf5SDale Johannesen case Type::X86_FP80TyID:
836a1336cf5SDale Johannesen case Type::PPC_FP128TyID:
837a1336cf5SDale Johannesen case Type::FP128TyID: {
83829178a34STim Northover const fltSemantics &Sem = CE->getOperand(0)->getType()->getFltSemantics();
83929178a34STim Northover APFloat apfLHS = APFloat(Sem, LHS.IntVal);
840a1336cf5SDale Johannesen switch (CE->getOpcode()) {
841e4f47434SDaniel Dunbar default: llvm_unreachable("Invalid long double opcode");
842a5b9645cSDan Gohman case Instruction::FAdd:
84329178a34STim Northover apfLHS.add(APFloat(Sem, RHS.IntVal), APFloat::rmNearestTiesToEven);
84454306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt();
845a1336cf5SDale Johannesen break;
846a5b9645cSDan Gohman case Instruction::FSub:
84729178a34STim Northover apfLHS.subtract(APFloat(Sem, RHS.IntVal),
84829178a34STim Northover APFloat::rmNearestTiesToEven);
84954306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt();
850a1336cf5SDale Johannesen break;
851a5b9645cSDan Gohman case Instruction::FMul:
85229178a34STim Northover apfLHS.multiply(APFloat(Sem, RHS.IntVal),
85329178a34STim Northover APFloat::rmNearestTiesToEven);
85454306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt();
855a1336cf5SDale Johannesen break;
856a1336cf5SDale Johannesen case Instruction::FDiv:
85729178a34STim Northover apfLHS.divide(APFloat(Sem, RHS.IntVal),
85829178a34STim Northover APFloat::rmNearestTiesToEven);
85954306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt();
860a1336cf5SDale Johannesen break;
861a1336cf5SDale Johannesen case Instruction::FRem:
862b12db0e4SStephen Canon apfLHS.mod(APFloat(Sem, RHS.IntVal));
86354306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt();
864a1336cf5SDale Johannesen break;
865a1336cf5SDale Johannesen }
866a1336cf5SDale Johannesen }
867a1336cf5SDale Johannesen break;
868c4e6bb5fSChris Lattner }
8694fd528f2SReid Spencer return GV;
8704fd528f2SReid Spencer }
8719de0d14dSChris Lattner default:
87268cbcc3eSChris Lattner break;
87368cbcc3eSChris Lattner }
874868e3f09SDaniel Dunbar
875868e3f09SDaniel Dunbar SmallString<256> Msg;
876868e3f09SDaniel Dunbar raw_svector_ostream OS(Msg);
877868e3f09SDaniel Dunbar OS << "ConstantExpr not handled: " << *CE;
878868e3f09SDaniel Dunbar report_fatal_error(OS.str());
8799de0d14dSChris Lattner }
880996fe010SChris Lattner
881868e3f09SDaniel Dunbar // Otherwise, we have a simple constant.
8824fd528f2SReid Spencer GenericValue Result;
8836b727599SChris Lattner switch (C->getType()->getTypeID()) {
88487aa65f4SReid Spencer case Type::FloatTyID:
885bed9dc42SDale Johannesen Result.FloatVal = cast<ConstantFP>(C)->getValueAPF().convertToFloat();
8867a9c62baSReid Spencer break;
88787aa65f4SReid Spencer case Type::DoubleTyID:
888bed9dc42SDale Johannesen Result.DoubleVal = cast<ConstantFP>(C)->getValueAPF().convertToDouble();
88987aa65f4SReid Spencer break;
890a1336cf5SDale Johannesen case Type::X86_FP80TyID:
891a1336cf5SDale Johannesen case Type::FP128TyID:
892a1336cf5SDale Johannesen case Type::PPC_FP128TyID:
89354306fe4SDale Johannesen Result.IntVal = cast <ConstantFP>(C)->getValueAPF().bitcastToAPInt();
894a1336cf5SDale Johannesen break;
89587aa65f4SReid Spencer case Type::IntegerTyID:
89687aa65f4SReid Spencer Result.IntVal = cast<ConstantInt>(C)->getValue();
89787aa65f4SReid Spencer break;
898996fe010SChris Lattner case Type::PointerTyID:
8997d60b905SLang Hames while (auto *A = dyn_cast<GlobalAlias>(C)) {
9007d60b905SLang Hames C = A->getAliasee();
9017d60b905SLang Hames }
9026a0fd73bSReid Spencer if (isa<ConstantPointerNull>(C))
9032617dcceSCraig Topper Result.PointerVal = nullptr;
9046a0fd73bSReid Spencer else if (const Function *F = dyn_cast<Function>(C))
9056a0fd73bSReid Spencer Result = PTOGV(getPointerToFunctionOrStub(const_cast<Function*>(F)));
9066a0fd73bSReid Spencer else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(C))
9076a0fd73bSReid Spencer Result = PTOGV(getOrEmitGlobalVariable(const_cast<GlobalVariable*>(GV)));
908e6492f10SChris Lattner else
909fbcc663cSTorok Edwin llvm_unreachable("Unknown constant pointer type!");
910996fe010SChris Lattner break;
9119b500e56SChristopher Tetreault case Type::ScalableVectorTyID:
9129b500e56SChristopher Tetreault report_fatal_error(
9139b500e56SChristopher Tetreault "Scalable vector support not yet implemented in ExecutionEngine");
9149b500e56SChristopher Tetreault case Type::FixedVectorTyID: {
915be79a7acSNadav Rotem unsigned elemNum;
916be79a7acSNadav Rotem Type* ElemTy;
917be79a7acSNadav Rotem const ConstantDataVector *CDV = dyn_cast<ConstantDataVector>(C);
918be79a7acSNadav Rotem const ConstantVector *CV = dyn_cast<ConstantVector>(C);
919be79a7acSNadav Rotem const ConstantAggregateZero *CAZ = dyn_cast<ConstantAggregateZero>(C);
920be79a7acSNadav Rotem
921be79a7acSNadav Rotem if (CDV) {
922be79a7acSNadav Rotem elemNum = CDV->getNumElements();
923be79a7acSNadav Rotem ElemTy = CDV->getElementType();
924be79a7acSNadav Rotem } else if (CV || CAZ) {
9259b500e56SChristopher Tetreault auto *VTy = cast<FixedVectorType>(C->getType());
926be79a7acSNadav Rotem elemNum = VTy->getNumElements();
927be79a7acSNadav Rotem ElemTy = VTy->getElementType();
928be79a7acSNadav Rotem } else {
929be79a7acSNadav Rotem llvm_unreachable("Unknown constant vector type!");
930be79a7acSNadav Rotem }
931be79a7acSNadav Rotem
932be79a7acSNadav Rotem Result.AggregateVal.resize(elemNum);
933be79a7acSNadav Rotem // Check if vector holds floats.
934be79a7acSNadav Rotem if(ElemTy->isFloatTy()) {
935be79a7acSNadav Rotem if (CAZ) {
936be79a7acSNadav Rotem GenericValue floatZero;
937be79a7acSNadav Rotem floatZero.FloatVal = 0.f;
938be79a7acSNadav Rotem std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(),
939be79a7acSNadav Rotem floatZero);
940be79a7acSNadav Rotem break;
941be79a7acSNadav Rotem }
942be79a7acSNadav Rotem if(CV) {
943be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i)
944be79a7acSNadav Rotem if (!isa<UndefValue>(CV->getOperand(i)))
945be79a7acSNadav Rotem Result.AggregateVal[i].FloatVal = cast<ConstantFP>(
946be79a7acSNadav Rotem CV->getOperand(i))->getValueAPF().convertToFloat();
947be79a7acSNadav Rotem break;
948be79a7acSNadav Rotem }
949be79a7acSNadav Rotem if(CDV)
950be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i)
951be79a7acSNadav Rotem Result.AggregateVal[i].FloatVal = CDV->getElementAsFloat(i);
952be79a7acSNadav Rotem
953be79a7acSNadav Rotem break;
954be79a7acSNadav Rotem }
955be79a7acSNadav Rotem // Check if vector holds doubles.
956be79a7acSNadav Rotem if (ElemTy->isDoubleTy()) {
957be79a7acSNadav Rotem if (CAZ) {
958be79a7acSNadav Rotem GenericValue doubleZero;
959be79a7acSNadav Rotem doubleZero.DoubleVal = 0.0;
960be79a7acSNadav Rotem std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(),
961be79a7acSNadav Rotem doubleZero);
962be79a7acSNadav Rotem break;
963be79a7acSNadav Rotem }
964be79a7acSNadav Rotem if(CV) {
965be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i)
966be79a7acSNadav Rotem if (!isa<UndefValue>(CV->getOperand(i)))
967be79a7acSNadav Rotem Result.AggregateVal[i].DoubleVal = cast<ConstantFP>(
968be79a7acSNadav Rotem CV->getOperand(i))->getValueAPF().convertToDouble();
969be79a7acSNadav Rotem break;
970be79a7acSNadav Rotem }
971be79a7acSNadav Rotem if(CDV)
972be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i)
973be79a7acSNadav Rotem Result.AggregateVal[i].DoubleVal = CDV->getElementAsDouble(i);
974be79a7acSNadav Rotem
975be79a7acSNadav Rotem break;
976be79a7acSNadav Rotem }
977be79a7acSNadav Rotem // Check if vector holds integers.
978be79a7acSNadav Rotem if (ElemTy->isIntegerTy()) {
979be79a7acSNadav Rotem if (CAZ) {
980be79a7acSNadav Rotem GenericValue intZero;
981be79a7acSNadav Rotem intZero.IntVal = APInt(ElemTy->getScalarSizeInBits(), 0ull);
982be79a7acSNadav Rotem std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(),
983be79a7acSNadav Rotem intZero);
984be79a7acSNadav Rotem break;
985be79a7acSNadav Rotem }
986be79a7acSNadav Rotem if(CV) {
987be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i)
988be79a7acSNadav Rotem if (!isa<UndefValue>(CV->getOperand(i)))
989be79a7acSNadav Rotem Result.AggregateVal[i].IntVal = cast<ConstantInt>(
990be79a7acSNadav Rotem CV->getOperand(i))->getValue();
991be79a7acSNadav Rotem else {
992be79a7acSNadav Rotem Result.AggregateVal[i].IntVal =
993be79a7acSNadav Rotem APInt(CV->getOperand(i)->getType()->getPrimitiveSizeInBits(), 0);
994be79a7acSNadav Rotem }
995be79a7acSNadav Rotem break;
996be79a7acSNadav Rotem }
997be79a7acSNadav Rotem if(CDV)
998be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i)
999be79a7acSNadav Rotem Result.AggregateVal[i].IntVal = APInt(
1000be79a7acSNadav Rotem CDV->getElementType()->getPrimitiveSizeInBits(),
1001be79a7acSNadav Rotem CDV->getElementAsInteger(i));
1002be79a7acSNadav Rotem
1003be79a7acSNadav Rotem break;
1004be79a7acSNadav Rotem }
1005be79a7acSNadav Rotem llvm_unreachable("Unknown constant pointer type!");
10062dea3f12SChristopher Tetreault } break;
1007be79a7acSNadav Rotem
1008996fe010SChris Lattner default:
1009868e3f09SDaniel Dunbar SmallString<256> Msg;
1010868e3f09SDaniel Dunbar raw_svector_ostream OS(Msg);
1011868e3f09SDaniel Dunbar OS << "ERROR: Constant unimplemented for type: " << *C->getType();
1012868e3f09SDaniel Dunbar report_fatal_error(OS.str());
1013996fe010SChris Lattner }
1014868e3f09SDaniel Dunbar
1015996fe010SChris Lattner return Result;
1016996fe010SChris Lattner }
1017996fe010SChris Lattner
StoreValueToMemory(const GenericValue & Val,GenericValue * Ptr,Type * Ty)101809053e62SEvan Cheng void ExecutionEngine::StoreValueToMemory(const GenericValue &Val,
1019229907cdSChris Lattner GenericValue *Ptr, Type *Ty) {
1020a3fcefb6SMehdi Amini const unsigned StoreBytes = getDataLayout().getTypeStoreSize(Ty);
10211202d1b1SDuncan Sands
10221202d1b1SDuncan Sands switch (Ty->getTypeID()) {
1023be79a7acSNadav Rotem default:
1024be79a7acSNadav Rotem dbgs() << "Cannot store value of type " << *Ty << "!\n";
1025be79a7acSNadav Rotem break;
10261202d1b1SDuncan Sands case Type::IntegerTyID:
10271202d1b1SDuncan Sands StoreIntToMemory(Val.IntVal, (uint8_t*)Ptr, StoreBytes);
10281202d1b1SDuncan Sands break;
1029996fe010SChris Lattner case Type::FloatTyID:
103087aa65f4SReid Spencer *((float*)Ptr) = Val.FloatVal;
103187aa65f4SReid Spencer break;
103287aa65f4SReid Spencer case Type::DoubleTyID:
103387aa65f4SReid Spencer *((double*)Ptr) = Val.DoubleVal;
1034996fe010SChris Lattner break;
10354d7e4ee7SDale Johannesen case Type::X86_FP80TyID:
10364d7e4ee7SDale Johannesen memcpy(Ptr, Val.IntVal.getRawData(), 10);
1037a1336cf5SDale Johannesen break;
10387a9c62baSReid Spencer case Type::PointerTyID:
10391202d1b1SDuncan Sands // Ensure 64 bit target pointers are fully initialized on 32 bit hosts.
10401202d1b1SDuncan Sands if (StoreBytes != sizeof(PointerTy))
104193da3c82SChandler Carruth memset(&(Ptr->PointerVal), 0, StoreBytes);
10421202d1b1SDuncan Sands
104387aa65f4SReid Spencer *((PointerTy*)Ptr) = Val.PointerVal;
1044996fe010SChris Lattner break;
10452dea3f12SChristopher Tetreault case Type::FixedVectorTyID:
10462dea3f12SChristopher Tetreault case Type::ScalableVectorTyID:
1047be79a7acSNadav Rotem for (unsigned i = 0; i < Val.AggregateVal.size(); ++i) {
1048be79a7acSNadav Rotem if (cast<VectorType>(Ty)->getElementType()->isDoubleTy())
1049be79a7acSNadav Rotem *(((double*)Ptr)+i) = Val.AggregateVal[i].DoubleVal;
1050be79a7acSNadav Rotem if (cast<VectorType>(Ty)->getElementType()->isFloatTy())
1051be79a7acSNadav Rotem *(((float*)Ptr)+i) = Val.AggregateVal[i].FloatVal;
1052be79a7acSNadav Rotem if (cast<VectorType>(Ty)->getElementType()->isIntegerTy()) {
1053be79a7acSNadav Rotem unsigned numOfBytes =(Val.AggregateVal[i].IntVal.getBitWidth()+7)/8;
1054be79a7acSNadav Rotem StoreIntToMemory(Val.AggregateVal[i].IntVal,
1055be79a7acSNadav Rotem (uint8_t*)Ptr + numOfBytes*i, numOfBytes);
1056be79a7acSNadav Rotem }
1057be79a7acSNadav Rotem }
1058be79a7acSNadav Rotem break;
1059996fe010SChris Lattner }
10601202d1b1SDuncan Sands
1061a3fcefb6SMehdi Amini if (sys::IsLittleEndianHost != getDataLayout().isLittleEndian())
10621202d1b1SDuncan Sands // Host and target are different endian - reverse the stored bytes.
10631202d1b1SDuncan Sands std::reverse((uint8_t*)Ptr, StoreBytes + (uint8_t*)Ptr);
1064996fe010SChris Lattner }
1065996fe010SChris Lattner
10661202d1b1SDuncan Sands /// FIXME: document
10671202d1b1SDuncan Sands ///
LoadValueFromMemory(GenericValue & Result,GenericValue * Ptr,Type * Ty)10681202d1b1SDuncan Sands void ExecutionEngine::LoadValueFromMemory(GenericValue &Result,
10691202d1b1SDuncan Sands GenericValue *Ptr,
1070229907cdSChris Lattner Type *Ty) {
1071a3fcefb6SMehdi Amini const unsigned LoadBytes = getDataLayout().getTypeStoreSize(Ty);
10721202d1b1SDuncan Sands
10731202d1b1SDuncan Sands switch (Ty->getTypeID()) {
10741202d1b1SDuncan Sands case Type::IntegerTyID:
10751202d1b1SDuncan Sands // An APInt with all words initially zero.
10761202d1b1SDuncan Sands Result.IntVal = APInt(cast<IntegerType>(Ty)->getBitWidth(), 0);
10771202d1b1SDuncan Sands LoadIntFromMemory(Result.IntVal, (uint8_t*)Ptr, LoadBytes);
10781202d1b1SDuncan Sands break;
10797f389e8cSChris Lattner case Type::FloatTyID:
108087aa65f4SReid Spencer Result.FloatVal = *((float*)Ptr);
108187aa65f4SReid Spencer break;
108287aa65f4SReid Spencer case Type::DoubleTyID:
108387aa65f4SReid Spencer Result.DoubleVal = *((double*)Ptr);
10847f389e8cSChris Lattner break;
10857a9c62baSReid Spencer case Type::PointerTyID:
108687aa65f4SReid Spencer Result.PointerVal = *((PointerTy*)Ptr);
10877f389e8cSChris Lattner break;
1088a1336cf5SDale Johannesen case Type::X86_FP80TyID: {
1089a1336cf5SDale Johannesen // This is endian dependent, but it will only work on x86 anyway.
109026d6539eSDuncan Sands // FIXME: Will not trap if loading a signaling NaN.
1091ff306287SDuncan Sands uint64_t y[2];
10924d7e4ee7SDale Johannesen memcpy(y, Ptr, 10);
10937a162881SJeffrey Yasskin Result.IntVal = APInt(80, y);
1094a1336cf5SDale Johannesen break;
1095a1336cf5SDale Johannesen }
10969b500e56SChristopher Tetreault case Type::ScalableVectorTyID:
10979b500e56SChristopher Tetreault report_fatal_error(
10989b500e56SChristopher Tetreault "Scalable vector support not yet implemented in ExecutionEngine");
10999b500e56SChristopher Tetreault case Type::FixedVectorTyID: {
11009b500e56SChristopher Tetreault auto *VT = cast<FixedVectorType>(Ty);
1101e3dcce97SCraig Topper Type *ElemT = VT->getElementType();
1102be79a7acSNadav Rotem const unsigned numElems = VT->getNumElements();
1103be79a7acSNadav Rotem if (ElemT->isFloatTy()) {
1104be79a7acSNadav Rotem Result.AggregateVal.resize(numElems);
1105be79a7acSNadav Rotem for (unsigned i = 0; i < numElems; ++i)
1106be79a7acSNadav Rotem Result.AggregateVal[i].FloatVal = *((float*)Ptr+i);
1107be79a7acSNadav Rotem }
1108be79a7acSNadav Rotem if (ElemT->isDoubleTy()) {
1109be79a7acSNadav Rotem Result.AggregateVal.resize(numElems);
1110be79a7acSNadav Rotem for (unsigned i = 0; i < numElems; ++i)
1111be79a7acSNadav Rotem Result.AggregateVal[i].DoubleVal = *((double*)Ptr+i);
1112be79a7acSNadav Rotem }
1113be79a7acSNadav Rotem if (ElemT->isIntegerTy()) {
1114be79a7acSNadav Rotem GenericValue intZero;
1115be79a7acSNadav Rotem const unsigned elemBitWidth = cast<IntegerType>(ElemT)->getBitWidth();
1116be79a7acSNadav Rotem intZero.IntVal = APInt(elemBitWidth, 0);
1117be79a7acSNadav Rotem Result.AggregateVal.resize(numElems, intZero);
1118be79a7acSNadav Rotem for (unsigned i = 0; i < numElems; ++i)
1119be79a7acSNadav Rotem LoadIntFromMemory(Result.AggregateVal[i].IntVal,
1120be79a7acSNadav Rotem (uint8_t*)Ptr+((elemBitWidth+7)/8)*i, (elemBitWidth+7)/8);
1121be79a7acSNadav Rotem }
1122be79a7acSNadav Rotem break;
1123be79a7acSNadav Rotem }
11247f389e8cSChris Lattner default:
1125868e3f09SDaniel Dunbar SmallString<256> Msg;
1126868e3f09SDaniel Dunbar raw_svector_ostream OS(Msg);
1127868e3f09SDaniel Dunbar OS << "Cannot load value of type " << *Ty << "!";
1128868e3f09SDaniel Dunbar report_fatal_error(OS.str());
11297f389e8cSChris Lattner }
11307f389e8cSChris Lattner }
11317f389e8cSChris Lattner
InitializeMemory(const Constant * Init,void * Addr)1132996fe010SChris Lattner void ExecutionEngine::InitializeMemory(const Constant *Init, void *Addr) {
1133d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "JIT: Initializing " << Addr << " ");
1134d34e60caSNicola Zaghen LLVM_DEBUG(Init->dump());
113500245f42SChris Lattner if (isa<UndefValue>(Init))
113661753bf8SChris Lattner return;
113700245f42SChris Lattner
113800245f42SChris Lattner if (const ConstantVector *CP = dyn_cast<ConstantVector>(Init)) {
113969d62138SRobert Bocchino unsigned ElementSize =
1140a3fcefb6SMehdi Amini getDataLayout().getTypeAllocSize(CP->getType()->getElementType());
114169d62138SRobert Bocchino for (unsigned i = 0, e = CP->getNumOperands(); i != e; ++i)
114269d62138SRobert Bocchino InitializeMemory(CP->getOperand(i), (char*)Addr+i*ElementSize);
114369d62138SRobert Bocchino return;
114400245f42SChris Lattner }
114500245f42SChris Lattner
114600245f42SChris Lattner if (isa<ConstantAggregateZero>(Init)) {
1147a3fcefb6SMehdi Amini memset(Addr, 0, (size_t)getDataLayout().getTypeAllocSize(Init->getType()));
11481dd86b11SChris Lattner return;
114900245f42SChris Lattner }
115000245f42SChris Lattner
115100245f42SChris Lattner if (const ConstantArray *CPA = dyn_cast<ConstantArray>(Init)) {
115269ddfbfeSDan Gohman unsigned ElementSize =
1153a3fcefb6SMehdi Amini getDataLayout().getTypeAllocSize(CPA->getType()->getElementType());
115469ddfbfeSDan Gohman for (unsigned i = 0, e = CPA->getNumOperands(); i != e; ++i)
115569ddfbfeSDan Gohman InitializeMemory(CPA->getOperand(i), (char*)Addr+i*ElementSize);
115669ddfbfeSDan Gohman return;
115700245f42SChris Lattner }
115800245f42SChris Lattner
115900245f42SChris Lattner if (const ConstantStruct *CPS = dyn_cast<ConstantStruct>(Init)) {
116069ddfbfeSDan Gohman const StructLayout *SL =
1161a3fcefb6SMehdi Amini getDataLayout().getStructLayout(cast<StructType>(CPS->getType()));
116269ddfbfeSDan Gohman for (unsigned i = 0, e = CPS->getNumOperands(); i != e; ++i)
116369ddfbfeSDan Gohman InitializeMemory(CPS->getOperand(i), (char*)Addr+SL->getElementOffset(i));
116469ddfbfeSDan Gohman return;
116500245f42SChris Lattner }
116600245f42SChris Lattner
116700245f42SChris Lattner if (const ConstantDataSequential *CDS =
116800245f42SChris Lattner dyn_cast<ConstantDataSequential>(Init)) {
116900245f42SChris Lattner // CDS is already laid out in host memory order.
117000245f42SChris Lattner StringRef Data = CDS->getRawDataValues();
117100245f42SChris Lattner memcpy(Addr, Data.data(), Data.size());
117200245f42SChris Lattner return;
117300245f42SChris Lattner }
117400245f42SChris Lattner
117500245f42SChris Lattner if (Init->getType()->isFirstClassType()) {
1176996fe010SChris Lattner GenericValue Val = getConstantValue(Init);
1177996fe010SChris Lattner StoreValueToMemory(Val, (GenericValue*)Addr, Init->getType());
1178996fe010SChris Lattner return;
1179996fe010SChris Lattner }
1180996fe010SChris Lattner
1181d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "Bad Type: " << *Init->getType() << "\n");
1182fbcc663cSTorok Edwin llvm_unreachable("Unknown constant type to initialize memory with!");
1183996fe010SChris Lattner }
1184996fe010SChris Lattner
1185996fe010SChris Lattner /// EmitGlobals - Emit all of the global variables to memory, storing their
1186996fe010SChris Lattner /// addresses into GlobalAddress. This must make sure to copy the contents of
1187996fe010SChris Lattner /// their initializers into the memory.
emitGlobals()1188996fe010SChris Lattner void ExecutionEngine::emitGlobals() {
1189996fe010SChris Lattner // Loop over all of the global variables in the program, allocating the memory
11900621caefSChris Lattner // to hold them. If there is more than one module, do a prepass over globals
11910621caefSChris Lattner // to figure out how the different modules should link together.
1192229907cdSChris Lattner std::map<std::pair<std::string, Type*>,
11930621caefSChris Lattner const GlobalValue*> LinkedGlobalsMap;
11940621caefSChris Lattner
11950621caefSChris Lattner if (Modules.size() != 1) {
11960621caefSChris Lattner for (unsigned m = 0, e = Modules.size(); m != e; ++m) {
1197091217beSJeffrey Yasskin Module &M = *Modules[m];
119849bb65a4SRafael Espindola for (const auto &GV : M.globals()) {
119949bb65a4SRafael Espindola if (GV.hasLocalLinkage() || GV.isDeclaration() ||
120049bb65a4SRafael Espindola GV.hasAppendingLinkage() || !GV.hasName())
12010621caefSChris Lattner continue;// Ignore external globals and globals with internal linkage.
12020621caefSChris Lattner
1203adcd0268SBenjamin Kramer const GlobalValue *&GVEntry = LinkedGlobalsMap[std::make_pair(
1204adcd0268SBenjamin Kramer std::string(GV.getName()), GV.getType())];
12050621caefSChris Lattner
12060621caefSChris Lattner // If this is the first time we've seen this global, it is the canonical
12070621caefSChris Lattner // version.
12080621caefSChris Lattner if (!GVEntry) {
120949bb65a4SRafael Espindola GVEntry = &GV;
12100621caefSChris Lattner continue;
12110621caefSChris Lattner }
12120621caefSChris Lattner
12130621caefSChris Lattner // If the existing global is strong, never replace it.
12147157bb76SNico Rieck if (GVEntry->hasExternalLinkage())
12150621caefSChris Lattner continue;
12160621caefSChris Lattner
12170621caefSChris Lattner // Otherwise, we know it's linkonce/weak, replace it if this is a strong
1218ce4396bcSDale Johannesen // symbol. FIXME is this right for common?
121949bb65a4SRafael Espindola if (GV.hasExternalLinkage() || GVEntry->hasExternalWeakLinkage())
122049bb65a4SRafael Espindola GVEntry = &GV;
12210621caefSChris Lattner }
12220621caefSChris Lattner }
12230621caefSChris Lattner }
12240621caefSChris Lattner
12250621caefSChris Lattner std::vector<const GlobalValue*> NonCanonicalGlobals;
12260621caefSChris Lattner for (unsigned m = 0, e = Modules.size(); m != e; ++m) {
1227091217beSJeffrey Yasskin Module &M = *Modules[m];
122849bb65a4SRafael Espindola for (const auto &GV : M.globals()) {
12290621caefSChris Lattner // In the multi-module case, see what this global maps to.
12300621caefSChris Lattner if (!LinkedGlobalsMap.empty()) {
1231adcd0268SBenjamin Kramer if (const GlobalValue *GVEntry = LinkedGlobalsMap[std::make_pair(
1232adcd0268SBenjamin Kramer std::string(GV.getName()), GV.getType())]) {
12330621caefSChris Lattner // If something else is the canonical global, ignore this one.
123449bb65a4SRafael Espindola if (GVEntry != &GV) {
123549bb65a4SRafael Espindola NonCanonicalGlobals.push_back(&GV);
12360621caefSChris Lattner continue;
12370621caefSChris Lattner }
12380621caefSChris Lattner }
12390621caefSChris Lattner }
12400621caefSChris Lattner
124149bb65a4SRafael Espindola if (!GV.isDeclaration()) {
124249bb65a4SRafael Espindola addGlobalMapping(&GV, getMemoryForGV(&GV));
1243996fe010SChris Lattner } else {
1244e8bbcfc2SBrian Gaeke // External variable reference. Try to use the dynamic loader to
1245e8bbcfc2SBrian Gaeke // get a pointer to it.
1246adcd0268SBenjamin Kramer if (void *SymAddr = sys::DynamicLibrary::SearchForAddressOfSymbol(
1247adcd0268SBenjamin Kramer std::string(GV.getName())))
124849bb65a4SRafael Espindola addGlobalMapping(&GV, SymAddr);
12499de0d14dSChris Lattner else {
12502104b8d3SChris Lattner report_fatal_error("Could not resolve external global address: "
125149bb65a4SRafael Espindola +GV.getName());
12529de0d14dSChris Lattner }
1253996fe010SChris Lattner }
12540621caefSChris Lattner }
12550621caefSChris Lattner
12560621caefSChris Lattner // If there are multiple modules, map the non-canonical globals to their
12570621caefSChris Lattner // canonical location.
12580621caefSChris Lattner if (!NonCanonicalGlobals.empty()) {
1259*fd7d4064SKazu Hirata for (const GlobalValue *GV : NonCanonicalGlobals) {
1260adcd0268SBenjamin Kramer const GlobalValue *CGV = LinkedGlobalsMap[std::make_pair(
1261adcd0268SBenjamin Kramer std::string(GV->getName()), GV->getType())];
12620621caefSChris Lattner void *Ptr = getPointerToGlobalIfAvailable(CGV);
12630621caefSChris Lattner assert(Ptr && "Canonical global wasn't codegen'd!");
1264a67f06b9SNuno Lopes addGlobalMapping(GV, Ptr);
12650621caefSChris Lattner }
12660621caefSChris Lattner }
1267996fe010SChris Lattner
12687a9c62baSReid Spencer // Now that all of the globals are set up in memory, loop through them all
12697a9c62baSReid Spencer // and initialize their contents.
127049bb65a4SRafael Espindola for (const auto &GV : M.globals()) {
127149bb65a4SRafael Espindola if (!GV.isDeclaration()) {
12720621caefSChris Lattner if (!LinkedGlobalsMap.empty()) {
1273adcd0268SBenjamin Kramer if (const GlobalValue *GVEntry = LinkedGlobalsMap[std::make_pair(
1274adcd0268SBenjamin Kramer std::string(GV.getName()), GV.getType())])
127549bb65a4SRafael Espindola if (GVEntry != &GV) // Not the canonical variable.
12760621caefSChris Lattner continue;
12770621caefSChris Lattner }
12781d49eb00SFangrui Song emitGlobalVariable(&GV);
12796bbe3eceSChris Lattner }
12800621caefSChris Lattner }
12810621caefSChris Lattner }
12820621caefSChris Lattner }
12836bbe3eceSChris Lattner
12846bbe3eceSChris Lattner // EmitGlobalVariable - This method emits the specified global variable to the
12856bbe3eceSChris Lattner // address specified in GlobalAddresses, or allocates new memory if it's not
12866bbe3eceSChris Lattner // already in the map.
emitGlobalVariable(const GlobalVariable * GV)12871d49eb00SFangrui Song void ExecutionEngine::emitGlobalVariable(const GlobalVariable *GV) {
1288748e8579SChris Lattner void *GA = getPointerToGlobalIfAvailable(GV);
1289dc631735SChris Lattner
12902617dcceSCraig Topper if (!GA) {
12916bbe3eceSChris Lattner // If it's not already specified, allocate memory for the global.
12925457ce9aSNicolas Geoffray GA = getMemoryForGV(GV);
12933b442379SAndrew Kaylor
12943b442379SAndrew Kaylor // If we failed to allocate memory for this global, return.
12952617dcceSCraig Topper if (!GA) return;
12963b442379SAndrew Kaylor
1297748e8579SChris Lattner addGlobalMapping(GV, GA);
12986bbe3eceSChris Lattner }
1299fbcc0aa1SChris Lattner
13005457ce9aSNicolas Geoffray // Don't initialize if it's thread local, let the client do it.
13015457ce9aSNicolas Geoffray if (!GV->isThreadLocal())
13026bbe3eceSChris Lattner InitializeMemory(GV->getInitializer(), GA);
13035457ce9aSNicolas Geoffray
13045f6eaac6SManuel Jacob Type *ElTy = GV->getValueType();
1305a3fcefb6SMehdi Amini size_t GVSize = (size_t)getDataLayout().getTypeAllocSize(ElTy);
1306df1f1524SChris Lattner NumInitBytes += (unsigned)GVSize;
13076bbe3eceSChris Lattner ++NumGlobals;
1308996fe010SChris Lattner }
1309