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 // 12996fe010SChris Lattner //===----------------------------------------------------------------------===// 13996fe010SChris Lattner 146bf87df5SJeffrey Yasskin #include "llvm/ExecutionEngine/ExecutionEngine.h" 150a446fd5SBenjamin Kramer #include "llvm/ADT/STLExtras.h" 16868e3f09SDaniel Dunbar #include "llvm/ADT/SmallString.h" 17390d78b3SChris Lattner #include "llvm/ADT/Statistic.h" 18ed0881b2SChandler Carruth #include "llvm/ExecutionEngine/GenericValue.h" 19b5c7b1ffSLang Hames #include "llvm/ExecutionEngine/JITEventListener.h" 203301c7eeSLang Hames #include "llvm/ExecutionEngine/ObjectCache.h" 21633fe146SLang Hames #include "llvm/ExecutionEngine/RTDyldMemoryManager.h" 229fb823bbSChandler Carruth #include "llvm/IR/Constants.h" 239fb823bbSChandler Carruth #include "llvm/IR/DataLayout.h" 249fb823bbSChandler Carruth #include "llvm/IR/DerivedTypes.h" 253dac3f7fSLang Hames #include "llvm/IR/Mangler.h" 269fb823bbSChandler Carruth #include "llvm/IR/Module.h" 279fb823bbSChandler Carruth #include "llvm/IR/Operator.h" 284220e9c1SChandler Carruth #include "llvm/IR/ValueHandle.h" 29dd39657aSRafael Espindola #include "llvm/Object/Archive.h" 3035907d8eSDavid Blaikie #include "llvm/Object/ObjectFile.h" 317c16caa3SReid Spencer #include "llvm/Support/Debug.h" 32ed0881b2SChandler Carruth #include "llvm/Support/DynamicLibrary.h" 336c2d233eSTorok Edwin #include "llvm/Support/ErrorHandling.h" 34ed0881b2SChandler Carruth #include "llvm/Support/Host.h" 35ed0881b2SChandler Carruth #include "llvm/Support/TargetRegistry.h" 36ccb29cd2STorok Edwin #include "llvm/Support/raw_ostream.h" 378418fdcdSDylan Noblesmith #include "llvm/Target/TargetMachine.h" 38579f0713SAnton Korobeynikov #include <cmath> 39579f0713SAnton Korobeynikov #include <cstring> 403d5360a4SBenjamin Kramer #include <mutex> 4129681deeSChris Lattner using namespace llvm; 42996fe010SChris Lattner 43f58e376dSChandler Carruth #define DEBUG_TYPE "jit" 44f58e376dSChandler Carruth 45c346ecd7SChris Lattner STATISTIC(NumInitBytes, "Number of bytes of global vars initialized"); 46c346ecd7SChris Lattner STATISTIC(NumGlobals , "Number of global vars initialized"); 47996fe010SChris Lattner 4870ff8b05SDaniel Dunbar ExecutionEngine *(*ExecutionEngine::MCJITCtor)( 49196e323cSDavid Blaikie std::unique_ptr<Module> M, std::string *ErrorStr, 50633fe146SLang Hames std::shared_ptr<MCJITMemoryManager> MemMgr, 51b72f4845SLang Hames std::shared_ptr<LegacyJITSymbolResolver> Resolver, 524a5697e6SLang Hames std::unique_ptr<TargetMachine> TM) = nullptr; 5393de2a12SLang Hames 5493de2a12SLang Hames ExecutionEngine *(*ExecutionEngine::OrcMCJITReplacementCtor)( 55633fe146SLang Hames std::string *ErrorStr, std::shared_ptr<MCJITMemoryManager> MemMgr, 56b72f4845SLang Hames std::shared_ptr<LegacyJITSymbolResolver> Resolver, 5793de2a12SLang Hames std::unique_ptr<TargetMachine> TM) = nullptr; 5893de2a12SLang Hames 592a8a2795SRafael Espindola ExecutionEngine *(*ExecutionEngine::InterpCtor)(std::unique_ptr<Module> M, 602617dcceSCraig Topper std::string *ErrorStr) =nullptr; 612d52c1b8SChris Lattner 627ea98e14SLang Hames void JITEventListener::anchor() {} 637ea98e14SLang Hames 643301c7eeSLang Hames void ObjectCache::anchor() {} 653301c7eeSLang Hames 66a3fcefb6SMehdi Amini void ExecutionEngine::Init(std::unique_ptr<Module> M) { 674567db45SJeffrey Yasskin CompilingLazily = false; 68cdc0060eSEvan Cheng GVCompilationDisabled = false; 6984a9055eSEvan Cheng SymbolSearchingDisabled = false; 70bc876017SLang Hames 71bc876017SLang Hames // IR module verification is enabled by default in debug builds, and disabled 72bc876017SLang Hames // by default in release builds. 73bc876017SLang Hames #ifndef NDEBUG 74bc876017SLang Hames VerifyModules = true; 75bc876017SLang Hames #else 76bc876017SLang Hames VerifyModules = false; 77bc876017SLang Hames #endif 78bc876017SLang Hames 79091217beSJeffrey Yasskin assert(M && "Module is null?"); 802a8a2795SRafael Espindola Modules.push_back(std::move(M)); 81260b0c88SMisha Brukman } 82260b0c88SMisha Brukman 83a3fcefb6SMehdi Amini ExecutionEngine::ExecutionEngine(std::unique_ptr<Module> M) 84a3fcefb6SMehdi Amini : DL(M->getDataLayout()), LazyFunctionCreator(nullptr) { 85a3fcefb6SMehdi Amini Init(std::move(M)); 86a3fcefb6SMehdi Amini } 87a3fcefb6SMehdi Amini 88a3fcefb6SMehdi Amini ExecutionEngine::ExecutionEngine(DataLayout DL, std::unique_ptr<Module> M) 89a3fcefb6SMehdi Amini : DL(std::move(DL)), LazyFunctionCreator(nullptr) { 90a3fcefb6SMehdi Amini Init(std::move(M)); 91a3fcefb6SMehdi Amini } 92a3fcefb6SMehdi Amini 9392f8b30dSBrian Gaeke ExecutionEngine::~ExecutionEngine() { 94603682adSReid Spencer clearAllGlobalMappings(); 9592f8b30dSBrian Gaeke } 9692f8b30dSBrian Gaeke 97a4044332SJeffrey Yasskin namespace { 985f8f34e4SAdrian Prantl /// Helper class which uses a value handler to automatically deletes the 99868e3f09SDaniel Dunbar /// memory block when the GlobalVariable is destroyed. 100774b584fSDavid Blaikie class GVMemoryBlock final : public CallbackVH { 101a4044332SJeffrey Yasskin GVMemoryBlock(const GlobalVariable *GV) 102a4044332SJeffrey Yasskin : CallbackVH(const_cast<GlobalVariable*>(GV)) {} 103a4044332SJeffrey Yasskin 104a4044332SJeffrey Yasskin public: 1055f8f34e4SAdrian Prantl /// Returns the address the GlobalVariable should be written into. The 106868e3f09SDaniel Dunbar /// GVMemoryBlock object prefixes that. 107cdfe20b9SMicah Villmow static char *Create(const GlobalVariable *GV, const DataLayout& TD) { 1085f6eaac6SManuel Jacob Type *ElTy = GV->getValueType(); 109a4044332SJeffrey Yasskin size_t GVSize = (size_t)TD.getTypeAllocSize(ElTy); 110a4044332SJeffrey Yasskin void *RawMemory = ::operator new( 111da00f2fdSRui Ueyama alignTo(sizeof(GVMemoryBlock), TD.getPreferredAlignment(GV)) + GVSize); 112a4044332SJeffrey Yasskin new(RawMemory) GVMemoryBlock(GV); 113a4044332SJeffrey Yasskin return static_cast<char*>(RawMemory) + sizeof(GVMemoryBlock); 114a4044332SJeffrey Yasskin } 115a4044332SJeffrey Yasskin 116b51ff603SCraig Topper void deleted() override { 117a4044332SJeffrey Yasskin // We allocated with operator new and with some extra memory hanging off the 118a4044332SJeffrey Yasskin // end, so don't just delete this. I'm not sure if this is actually 119a4044332SJeffrey Yasskin // required. 120a4044332SJeffrey Yasskin this->~GVMemoryBlock(); 121a4044332SJeffrey Yasskin ::operator delete(this); 122a4044332SJeffrey Yasskin } 123a4044332SJeffrey Yasskin }; 124a4044332SJeffrey Yasskin } // anonymous namespace 125a4044332SJeffrey Yasskin 126a4044332SJeffrey Yasskin char *ExecutionEngine::getMemoryForGV(const GlobalVariable *GV) { 127a3fcefb6SMehdi Amini return GVMemoryBlock::Create(GV, getDataLayout()); 1285457ce9aSNicolas Geoffray } 1295457ce9aSNicolas Geoffray 13035907d8eSDavid Blaikie void ExecutionEngine::addObjectFile(std::unique_ptr<object::ObjectFile> O) { 13135907d8eSDavid Blaikie llvm_unreachable("ExecutionEngine subclass doesn't implement addObjectFile."); 13235907d8eSDavid Blaikie } 13335907d8eSDavid Blaikie 1347271c194SRafael Espindola void 1357271c194SRafael Espindola ExecutionEngine::addObjectFile(object::OwningBinary<object::ObjectFile> O) { 1367271c194SRafael Espindola llvm_unreachable("ExecutionEngine subclass doesn't implement addObjectFile."); 1377271c194SRafael Espindola } 1387271c194SRafael Espindola 13948af1c2aSRafael Espindola void ExecutionEngine::addArchive(object::OwningBinary<object::Archive> A) { 140acfd6289SRafael Espindola llvm_unreachable("ExecutionEngine subclass doesn't implement addArchive."); 141acfd6289SRafael Espindola } 142acfd6289SRafael Espindola 143091217beSJeffrey Yasskin bool ExecutionEngine::removeModule(Module *M) { 1442a8a2795SRafael Espindola for (auto I = Modules.begin(), E = Modules.end(); I != E; ++I) { 1452a8a2795SRafael Espindola Module *Found = I->get(); 146091217beSJeffrey Yasskin if (Found == M) { 1472a8a2795SRafael Espindola I->release(); 148324fe890SDevang Patel Modules.erase(I); 149091217beSJeffrey Yasskin clearGlobalMappingsFromModule(M); 150091217beSJeffrey Yasskin return true; 151324fe890SDevang Patel } 152324fe890SDevang Patel } 153091217beSJeffrey Yasskin return false; 154617001d8SNate Begeman } 155617001d8SNate Begeman 1567419e940SMehdi Amini Function *ExecutionEngine::FindFunctionNamed(StringRef FnName) { 1570621caefSChris Lattner for (unsigned i = 0, e = Modules.size(); i != e; ++i) { 1585f92a08fSKeno Fischer Function *F = Modules[i]->getFunction(FnName); 1595f92a08fSKeno Fischer if (F && !F->isDeclaration()) 1600621caefSChris Lattner return F; 1610621caefSChris Lattner } 1622617dcceSCraig Topper return nullptr; 1630621caefSChris Lattner } 1640621caefSChris Lattner 1657419e940SMehdi Amini GlobalVariable *ExecutionEngine::FindGlobalVariableNamed(StringRef Name, bool AllowInternal) { 16673378eb1SKeno Fischer for (unsigned i = 0, e = Modules.size(); i != e; ++i) { 16773378eb1SKeno Fischer GlobalVariable *GV = Modules[i]->getGlobalVariable(Name,AllowInternal); 16873378eb1SKeno Fischer if (GV && !GV->isDeclaration()) 16973378eb1SKeno Fischer return GV; 17073378eb1SKeno Fischer } 17173378eb1SKeno Fischer return nullptr; 17273378eb1SKeno Fischer } 1730621caefSChris Lattner 1743dac3f7fSLang Hames uint64_t ExecutionEngineState::RemoveMapping(StringRef Name) { 1753dac3f7fSLang Hames GlobalAddressMapTy::iterator I = GlobalAddressMap.find(Name); 1763dac3f7fSLang Hames uint64_t OldVal; 177868e3f09SDaniel Dunbar 178868e3f09SDaniel Dunbar // FIXME: This is silly, we shouldn't end up with a mapping -> 0 in the 179868e3f09SDaniel Dunbar // GlobalAddressMap. 180307c053fSJeffrey Yasskin if (I == GlobalAddressMap.end()) 1813dac3f7fSLang Hames OldVal = 0; 182307c053fSJeffrey Yasskin else { 1833dac3f7fSLang Hames GlobalAddressReverseMap.erase(I->second); 184307c053fSJeffrey Yasskin OldVal = I->second; 185307c053fSJeffrey Yasskin GlobalAddressMap.erase(I); 186307c053fSJeffrey Yasskin } 187307c053fSJeffrey Yasskin 188307c053fSJeffrey Yasskin return OldVal; 189307c053fSJeffrey Yasskin } 190307c053fSJeffrey Yasskin 1913dac3f7fSLang Hames std::string ExecutionEngine::getMangledName(const GlobalValue *GV) { 1923393cfdeSLang Hames assert(GV->hasName() && "Global must have name."); 1933393cfdeSLang Hames 1943d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock); 1953dac3f7fSLang Hames SmallString<128> FullName; 1963393cfdeSLang Hames 1973393cfdeSLang Hames const DataLayout &DL = 1983393cfdeSLang Hames GV->getParent()->getDataLayout().isDefault() 1993393cfdeSLang Hames ? getDataLayout() 2003393cfdeSLang Hames : GV->getParent()->getDataLayout(); 2013393cfdeSLang Hames 2023393cfdeSLang Hames Mangler::getNameWithPrefix(FullName, GV->getName(), DL); 2033dac3f7fSLang Hames return FullName.str(); 2043dac3f7fSLang Hames } 2053dac3f7fSLang Hames 2066d8dd189SChris Lattner void ExecutionEngine::addGlobalMapping(const GlobalValue *GV, void *Addr) { 2073d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock); 2083dac3f7fSLang Hames addGlobalMapping(getMangledName(GV), (uint64_t) Addr); 2093dac3f7fSLang Hames } 2106d8dd189SChris Lattner 2113dac3f7fSLang Hames void ExecutionEngine::addGlobalMapping(StringRef Name, uint64_t Addr) { 2123d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock); 2133dac3f7fSLang Hames 2143dac3f7fSLang Hames assert(!Name.empty() && "Empty GlobalMapping symbol name!"); 2153dac3f7fSLang Hames 216d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "JIT: Map \'" << Name << "\' to [" << Addr << "]\n";); 2173dac3f7fSLang Hames uint64_t &CurVal = EEState.getGlobalAddressMap()[Name]; 2182617dcceSCraig Topper assert((!CurVal || !Addr) && "GlobalMapping already established!"); 2196d8dd189SChris Lattner CurVal = Addr; 2206d8dd189SChris Lattner 221868e3f09SDaniel Dunbar // If we are using the reverse mapping, add it too. 2222f825df6SZachary Turner if (!EEState.getGlobalAddressReverseMap().empty()) { 2233dac3f7fSLang Hames std::string &V = EEState.getGlobalAddressReverseMap()[CurVal]; 2243dac3f7fSLang Hames assert((!V.empty() || !Name.empty()) && 2253dac3f7fSLang Hames "GlobalMapping already established!"); 2263dac3f7fSLang Hames V = Name; 2276d8dd189SChris Lattner } 2286d8dd189SChris Lattner } 2296d8dd189SChris Lattner 2306d8dd189SChris Lattner void ExecutionEngine::clearAllGlobalMappings() { 2313d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock); 2326d8dd189SChris Lattner 2332f825df6SZachary Turner EEState.getGlobalAddressMap().clear(); 2342f825df6SZachary Turner EEState.getGlobalAddressReverseMap().clear(); 2356d8dd189SChris Lattner } 2366d8dd189SChris Lattner 2378f83fc4dSNate Begeman void ExecutionEngine::clearGlobalMappingsFromModule(Module *M) { 2383d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock); 2398f83fc4dSNate Begeman 2406d88fde3SPeter Collingbourne for (GlobalObject &GO : M->global_objects()) 2416d88fde3SPeter Collingbourne EEState.RemoveMapping(getMangledName(&GO)); 2428f83fc4dSNate Begeman } 2438f83fc4dSNate Begeman 2443dac3f7fSLang Hames uint64_t ExecutionEngine::updateGlobalMapping(const GlobalValue *GV, 2453dac3f7fSLang Hames void *Addr) { 2463d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock); 2473dac3f7fSLang Hames return updateGlobalMapping(getMangledName(GV), (uint64_t) Addr); 2483dac3f7fSLang Hames } 2493dac3f7fSLang Hames 2503dac3f7fSLang Hames uint64_t ExecutionEngine::updateGlobalMapping(StringRef Name, uint64_t Addr) { 2513d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock); 2526d8dd189SChris Lattner 253d0fc8f80SJeffrey Yasskin ExecutionEngineState::GlobalAddressMapTy &Map = 2542f825df6SZachary Turner EEState.getGlobalAddressMap(); 255ee181730SChris Lattner 2566d8dd189SChris Lattner // Deleting from the mapping? 2572617dcceSCraig Topper if (!Addr) 2583dac3f7fSLang Hames return EEState.RemoveMapping(Name); 259ee181730SChris Lattner 2603dac3f7fSLang Hames uint64_t &CurVal = Map[Name]; 2613dac3f7fSLang Hames uint64_t OldVal = CurVal; 262ee181730SChris Lattner 2632f825df6SZachary Turner if (CurVal && !EEState.getGlobalAddressReverseMap().empty()) 2642f825df6SZachary Turner EEState.getGlobalAddressReverseMap().erase(CurVal); 2656d8dd189SChris Lattner CurVal = Addr; 2666d8dd189SChris Lattner 267868e3f09SDaniel Dunbar // If we are using the reverse mapping, add it too. 2682f825df6SZachary Turner if (!EEState.getGlobalAddressReverseMap().empty()) { 2693dac3f7fSLang Hames std::string &V = EEState.getGlobalAddressReverseMap()[CurVal]; 2703dac3f7fSLang Hames assert((!V.empty() || !Name.empty()) && 2713dac3f7fSLang Hames "GlobalMapping already established!"); 2723dac3f7fSLang Hames V = Name; 2736d8dd189SChris Lattner } 274ee181730SChris Lattner return OldVal; 2756d8dd189SChris Lattner } 2766d8dd189SChris Lattner 2773dac3f7fSLang Hames uint64_t ExecutionEngine::getAddressToGlobalIfAvailable(StringRef S) { 2783d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock); 2793dac3f7fSLang Hames uint64_t Address = 0; 2803dac3f7fSLang Hames ExecutionEngineState::GlobalAddressMapTy::iterator I = 2813dac3f7fSLang Hames EEState.getGlobalAddressMap().find(S); 2823dac3f7fSLang Hames if (I != EEState.getGlobalAddressMap().end()) 2833dac3f7fSLang Hames Address = I->second; 2843dac3f7fSLang Hames return Address; 2853dac3f7fSLang Hames } 2863dac3f7fSLang Hames 2873dac3f7fSLang Hames 2883dac3f7fSLang Hames void *ExecutionEngine::getPointerToGlobalIfAvailable(StringRef S) { 2893d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock); 2903dac3f7fSLang Hames if (void* Address = (void *) getAddressToGlobalIfAvailable(S)) 2913dac3f7fSLang Hames return Address; 2923dac3f7fSLang Hames return nullptr; 2933dac3f7fSLang Hames } 2943dac3f7fSLang Hames 2956d8dd189SChris Lattner void *ExecutionEngine::getPointerToGlobalIfAvailable(const GlobalValue *GV) { 2963d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock); 2973dac3f7fSLang Hames return getPointerToGlobalIfAvailable(getMangledName(GV)); 2986d8dd189SChris Lattner } 2996d8dd189SChris Lattner 300748e8579SChris Lattner const GlobalValue *ExecutionEngine::getGlobalValueAtAddress(void *Addr) { 3013d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock); 30279876f52SReid Spencer 303748e8579SChris Lattner // If we haven't computed the reverse mapping yet, do so first. 3042f825df6SZachary Turner if (EEState.getGlobalAddressReverseMap().empty()) { 305d0fc8f80SJeffrey Yasskin for (ExecutionEngineState::GlobalAddressMapTy::iterator 3062f825df6SZachary Turner I = EEState.getGlobalAddressMap().begin(), 3073dac3f7fSLang Hames E = EEState.getGlobalAddressMap().end(); I != E; ++I) { 3083dac3f7fSLang Hames StringRef Name = I->first(); 3093dac3f7fSLang Hames uint64_t Addr = I->second; 3102f825df6SZachary Turner EEState.getGlobalAddressReverseMap().insert(std::make_pair( 3113dac3f7fSLang Hames Addr, Name)); 3123dac3f7fSLang Hames } 313748e8579SChris Lattner } 314748e8579SChris Lattner 3153dac3f7fSLang Hames std::map<uint64_t, std::string>::iterator I = 3163dac3f7fSLang Hames EEState.getGlobalAddressReverseMap().find((uint64_t) Addr); 3173dac3f7fSLang Hames 3183dac3f7fSLang Hames if (I != EEState.getGlobalAddressReverseMap().end()) { 3193dac3f7fSLang Hames StringRef Name = I->second; 3203dac3f7fSLang Hames for (unsigned i = 0, e = Modules.size(); i != e; ++i) 3213dac3f7fSLang Hames if (GlobalValue *GV = Modules[i]->getNamedValue(Name)) 3223dac3f7fSLang Hames return GV; 3233dac3f7fSLang Hames } 3243dac3f7fSLang Hames return nullptr; 325748e8579SChris Lattner } 3265a0d4829SChris Lattner 327bfd38abbSJeffrey Yasskin namespace { 328bfd38abbSJeffrey Yasskin class ArgvArray { 329c4a9942aSDylan Noblesmith std::unique_ptr<char[]> Array; 330c4a9942aSDylan Noblesmith std::vector<std::unique_ptr<char[]>> Values; 331bfd38abbSJeffrey Yasskin public: 332bfd38abbSJeffrey Yasskin /// Turn a vector of strings into a nice argv style array of pointers to null 333bfd38abbSJeffrey Yasskin /// terminated strings. 334bfd38abbSJeffrey Yasskin void *reset(LLVMContext &C, ExecutionEngine *EE, 335bfd38abbSJeffrey Yasskin const std::vector<std::string> &InputArgv); 336bfd38abbSJeffrey Yasskin }; 337bfd38abbSJeffrey Yasskin } // anonymous namespace 338bfd38abbSJeffrey Yasskin void *ArgvArray::reset(LLVMContext &C, ExecutionEngine *EE, 3395a0d4829SChris Lattner const std::vector<std::string> &InputArgv) { 340c4a9942aSDylan Noblesmith Values.clear(); // Free the old contents. 341c4a9942aSDylan Noblesmith Values.reserve(InputArgv.size()); 342a3fcefb6SMehdi Amini unsigned PtrSize = EE->getDataLayout().getPointerSize(); 343*0eaee545SJonas Devlieghere Array = std::make_unique<char[]>((InputArgv.size()+1)*PtrSize); 3445a0d4829SChris Lattner 345d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "JIT: ARGV = " << (void *)Array.get() << "\n"); 346229907cdSChris Lattner Type *SBytePtr = Type::getInt8PtrTy(C); 3475a0d4829SChris Lattner 3485a0d4829SChris Lattner for (unsigned i = 0; i != InputArgv.size(); ++i) { 3495a0d4829SChris Lattner unsigned Size = InputArgv[i].size()+1; 350*0eaee545SJonas Devlieghere auto Dest = std::make_unique<char[]>(Size); 351d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "JIT: ARGV[" << i << "] = " << (void *)Dest.get() 352d34e60caSNicola Zaghen << "\n"); 3535a0d4829SChris Lattner 3544b535d19SDylan Noblesmith std::copy(InputArgv[i].begin(), InputArgv[i].end(), Dest.get()); 3555a0d4829SChris Lattner Dest[Size-1] = 0; 3565a0d4829SChris Lattner 357bfd38abbSJeffrey Yasskin // Endian safe: Array[i] = (PointerTy)Dest; 3584b535d19SDylan Noblesmith EE->StoreValueToMemory(PTOGV(Dest.get()), 3594b535d19SDylan Noblesmith (GenericValue*)(&Array[i*PtrSize]), SBytePtr); 3604b535d19SDylan Noblesmith Values.push_back(std::move(Dest)); 3615a0d4829SChris Lattner } 3625a0d4829SChris Lattner 3635a0d4829SChris Lattner // Null terminate it 3642617dcceSCraig Topper EE->StoreValueToMemory(PTOGV(nullptr), 365c4a9942aSDylan Noblesmith (GenericValue*)(&Array[InputArgv.size()*PtrSize]), 3665a0d4829SChris Lattner SBytePtr); 367c4a9942aSDylan Noblesmith return Array.get(); 3685a0d4829SChris Lattner } 3695a0d4829SChris Lattner 3702a8a2795SRafael Espindola void ExecutionEngine::runStaticConstructorsDestructors(Module &module, 37141fa2bd1SChris Lattner bool isDtors) { 3727419e940SMehdi Amini StringRef Name(isDtors ? "llvm.global_dtors" : "llvm.global_ctors"); 3732a8a2795SRafael Espindola GlobalVariable *GV = module.getNamedGlobal(Name); 374fe36eaebSChris Lattner 375fe36eaebSChris Lattner // If this global has internal linkage, or if it has a use, then it must be 376fe36eaebSChris Lattner // an old-style (llvmgcc3) static ctor with __main linked in and in use. If 3770621caefSChris Lattner // this is the case, don't execute any of the global ctors, __main will do 3780621caefSChris Lattner // it. 3796de96a1bSRafael Espindola if (!GV || GV->isDeclaration() || GV->hasLocalLinkage()) return; 380faae50b6SChris Lattner 3810cbfcb2bSNick Lewycky // Should be an array of '{ i32, void ()* }' structs. The first value is 3820621caefSChris Lattner // the init priority, which we ignore. 38300245f42SChris Lattner ConstantArray *InitList = dyn_cast<ConstantArray>(GV->getInitializer()); 3842617dcceSCraig Topper if (!InitList) 3850f857898SNick Lewycky return; 386868e3f09SDaniel Dunbar for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i) { 38700245f42SChris Lattner ConstantStruct *CS = dyn_cast<ConstantStruct>(InitList->getOperand(i)); 3882617dcceSCraig Topper if (!CS) continue; 389faae50b6SChris Lattner 390faae50b6SChris Lattner Constant *FP = CS->getOperand(1); 391faae50b6SChris Lattner if (FP->isNullValue()) 3920f857898SNick Lewycky continue; // Found a sentinal value, ignore. 393faae50b6SChris Lattner 394868e3f09SDaniel Dunbar // Strip off constant expression casts. 395faae50b6SChris Lattner if (ConstantExpr *CE = dyn_cast<ConstantExpr>(FP)) 3966c38f0bbSReid Spencer if (CE->isCast()) 397faae50b6SChris Lattner FP = CE->getOperand(0); 398868e3f09SDaniel Dunbar 399faae50b6SChris Lattner // Execute the ctor/dtor function! 400868e3f09SDaniel Dunbar if (Function *F = dyn_cast<Function>(FP)) 401bd7b1c89SBenjamin Kramer runFunction(F, None); 402868e3f09SDaniel Dunbar 403868e3f09SDaniel Dunbar // FIXME: It is marginally lame that we just do nothing here if we see an 404868e3f09SDaniel Dunbar // entry we don't recognize. It might not be unreasonable for the verifier 405868e3f09SDaniel Dunbar // to not even allow this and just assert here. 406faae50b6SChris Lattner } 407faae50b6SChris Lattner } 4081a9a0b7bSEvan Cheng 4091a9a0b7bSEvan Cheng void ExecutionEngine::runStaticConstructorsDestructors(bool isDtors) { 4101a9a0b7bSEvan Cheng // Execute global ctors/dtors for each module in the program. 4112a8a2795SRafael Espindola for (std::unique_ptr<Module> &M : Modules) 4122a8a2795SRafael Espindola runStaticConstructorsDestructors(*M, isDtors); 4130621caefSChris Lattner } 414faae50b6SChris Lattner 415cf3e3017SDan Gohman #ifndef NDEBUG 4161202d1b1SDuncan Sands /// isTargetNullPtr - Return whether the target pointer stored at Loc is null. 4171202d1b1SDuncan Sands static bool isTargetNullPtr(ExecutionEngine *EE, void *Loc) { 418a3fcefb6SMehdi Amini unsigned PtrSize = EE->getDataLayout().getPointerSize(); 4191202d1b1SDuncan Sands for (unsigned i = 0; i < PtrSize; ++i) 4201202d1b1SDuncan Sands if (*(i + (uint8_t*)Loc)) 4211202d1b1SDuncan Sands return false; 4221202d1b1SDuncan Sands return true; 4231202d1b1SDuncan Sands } 424cf3e3017SDan Gohman #endif 4251202d1b1SDuncan Sands 4265a0d4829SChris Lattner int ExecutionEngine::runFunctionAsMain(Function *Fn, 4275a0d4829SChris Lattner const std::vector<std::string> &argv, 4285a0d4829SChris Lattner const char * const * envp) { 4295a0d4829SChris Lattner std::vector<GenericValue> GVArgs; 4305a0d4829SChris Lattner GenericValue GVArgc; 43187aa65f4SReid Spencer GVArgc.IntVal = APInt(32, argv.size()); 4328c32c111SAnton Korobeynikov 4338c32c111SAnton Korobeynikov // Check main() type 434b1cad0b3SChris Lattner unsigned NumArgs = Fn->getFunctionType()->getNumParams(); 435229907cdSChris Lattner FunctionType *FTy = Fn->getFunctionType(); 436229907cdSChris Lattner Type* PPInt8Ty = Type::getInt8PtrTy(Fn->getContext())->getPointerTo(); 437868e3f09SDaniel Dunbar 438868e3f09SDaniel Dunbar // Check the argument types. 439868e3f09SDaniel Dunbar if (NumArgs > 3) 4402104b8d3SChris Lattner report_fatal_error("Invalid number of arguments of main() supplied"); 441868e3f09SDaniel Dunbar if (NumArgs >= 3 && FTy->getParamType(2) != PPInt8Ty) 442868e3f09SDaniel Dunbar report_fatal_error("Invalid type for third argument of main() supplied"); 443868e3f09SDaniel Dunbar if (NumArgs >= 2 && FTy->getParamType(1) != PPInt8Ty) 444868e3f09SDaniel Dunbar report_fatal_error("Invalid type for second argument of main() supplied"); 445868e3f09SDaniel Dunbar if (NumArgs >= 1 && !FTy->getParamType(0)->isIntegerTy(32)) 446868e3f09SDaniel Dunbar report_fatal_error("Invalid type for first argument of main() supplied"); 447868e3f09SDaniel Dunbar if (!FTy->getReturnType()->isIntegerTy() && 448868e3f09SDaniel Dunbar !FTy->getReturnType()->isVoidTy()) 449868e3f09SDaniel Dunbar report_fatal_error("Invalid return type of main() supplied"); 4508c32c111SAnton Korobeynikov 451bfd38abbSJeffrey Yasskin ArgvArray CArgv; 452bfd38abbSJeffrey Yasskin ArgvArray CEnv; 453b1cad0b3SChris Lattner if (NumArgs) { 4545a0d4829SChris Lattner GVArgs.push_back(GVArgc); // Arg #0 = argc. 455b1cad0b3SChris Lattner if (NumArgs > 1) { 45655f1c09eSOwen Anderson // Arg #1 = argv. 457bfd38abbSJeffrey Yasskin GVArgs.push_back(PTOGV(CArgv.reset(Fn->getContext(), this, argv))); 4581202d1b1SDuncan Sands assert(!isTargetNullPtr(this, GVTOP(GVArgs[1])) && 459b1cad0b3SChris Lattner "argv[0] was null after CreateArgv"); 460b1cad0b3SChris Lattner if (NumArgs > 2) { 4615a0d4829SChris Lattner std::vector<std::string> EnvVars; 4625a0d4829SChris Lattner for (unsigned i = 0; envp[i]; ++i) 463f5e2fc47SBenjamin Kramer EnvVars.emplace_back(envp[i]); 46455f1c09eSOwen Anderson // Arg #2 = envp. 465bfd38abbSJeffrey Yasskin GVArgs.push_back(PTOGV(CEnv.reset(Fn->getContext(), this, EnvVars))); 466b1cad0b3SChris Lattner } 467b1cad0b3SChris Lattner } 468b1cad0b3SChris Lattner } 469868e3f09SDaniel Dunbar 47087aa65f4SReid Spencer return runFunction(Fn, GVArgs).IntVal.getZExtValue(); 4715a0d4829SChris Lattner } 4725a0d4829SChris Lattner 473298a3a05SBenjamin Kramer EngineBuilder::EngineBuilder() : EngineBuilder(nullptr) {} 47493de2a12SLang Hames 4754a5697e6SLang Hames EngineBuilder::EngineBuilder(std::unique_ptr<Module> M) 476298a3a05SBenjamin Kramer : M(std::move(M)), WhichEngine(EngineKind::Either), ErrorStr(nullptr), 477633fe146SLang Hames OptLevel(CodeGenOpt::Default), MemMgr(nullptr), Resolver(nullptr), 47879e238afSRafael Espindola UseOrcMCJITReplacement(false) { 479322db9eaSAlp Toker // IR module verification is enabled by default in debug builds, and disabled 480322db9eaSAlp Toker // by default in release builds. 481322db9eaSAlp Toker #ifndef NDEBUG 482322db9eaSAlp Toker VerifyModules = true; 483322db9eaSAlp Toker #else 484322db9eaSAlp Toker VerifyModules = false; 485322db9eaSAlp Toker #endif 486322db9eaSAlp Toker } 487322db9eaSAlp Toker 488298a3a05SBenjamin Kramer EngineBuilder::~EngineBuilder() = default; 489298a3a05SBenjamin Kramer 490298a3a05SBenjamin Kramer EngineBuilder &EngineBuilder::setMCJITMemoryManager( 491298a3a05SBenjamin Kramer std::unique_ptr<RTDyldMemoryManager> mcjmm) { 492633fe146SLang Hames auto SharedMM = std::shared_ptr<RTDyldMemoryManager>(std::move(mcjmm)); 493633fe146SLang Hames MemMgr = SharedMM; 494633fe146SLang Hames Resolver = SharedMM; 495633fe146SLang Hames return *this; 496633fe146SLang Hames } 497633fe146SLang Hames 498633fe146SLang Hames EngineBuilder& 499633fe146SLang Hames EngineBuilder::setMemoryManager(std::unique_ptr<MCJITMemoryManager> MM) { 500633fe146SLang Hames MemMgr = std::shared_ptr<MCJITMemoryManager>(std::move(MM)); 501633fe146SLang Hames return *this; 502633fe146SLang Hames } 503633fe146SLang Hames 504633fe146SLang Hames EngineBuilder & 505b72f4845SLang Hames EngineBuilder::setSymbolResolver(std::unique_ptr<LegacyJITSymbolResolver> SR) { 506b72f4845SLang Hames Resolver = std::shared_ptr<LegacyJITSymbolResolver>(std::move(SR)); 507298a3a05SBenjamin Kramer return *this; 508298a3a05SBenjamin Kramer } 509298a3a05SBenjamin Kramer 510add6f1d2SOwen Anderson ExecutionEngine *EngineBuilder::create(TargetMachine *TM) { 51156440fd8SAhmed Charles std::unique_ptr<TargetMachine> TheTM(TM); // Take ownership. 51225a3d816SBenjamin Kramer 513a53414fdSNick Lewycky // Make sure we can resolve symbols in the program as well. The zero arg 514a53414fdSNick Lewycky // to the function tells DynamicLibrary to load the program, not a library. 5152617dcceSCraig Topper if (sys::DynamicLibrary::LoadLibraryPermanently(nullptr, ErrorStr)) 5162617dcceSCraig Topper return nullptr; 517a53414fdSNick Lewycky 518fc8a2d5aSReid Kleckner // If the user specified a memory manager but didn't specify which engine to 519fc8a2d5aSReid Kleckner // create, we assume they only want the JIT, and we fail if they only want 520fc8a2d5aSReid Kleckner // the interpreter. 521633fe146SLang Hames if (MemMgr) { 52241fa2bd1SChris Lattner if (WhichEngine & EngineKind::JIT) 523fc8a2d5aSReid Kleckner WhichEngine = EngineKind::JIT; 52441fa2bd1SChris Lattner else { 5258bcc6445SChris Lattner if (ErrorStr) 526fc8a2d5aSReid Kleckner *ErrorStr = "Cannot create an interpreter with a memory manager."; 5272617dcceSCraig Topper return nullptr; 528fc8a2d5aSReid Kleckner } 5294bd3bd5bSBrian Gaeke } 5304bd3bd5bSBrian Gaeke 531fc8a2d5aSReid Kleckner // Unless the interpreter was explicitly selected or the JIT is not linked, 532fc8a2d5aSReid Kleckner // try making a JIT. 53325a3d816SBenjamin Kramer if ((WhichEngine & EngineKind::JIT) && TheTM) { 5347f26246aSDylan Noblesmith if (!TM->getTarget().hasJIT()) { 5357f26246aSDylan Noblesmith errs() << "WARNING: This target JIT is not designed for the host" 5367f26246aSDylan Noblesmith << " you are running. If bad things happen, please choose" 5377f26246aSDylan Noblesmith << " a different -march switch.\n"; 5387f26246aSDylan Noblesmith } 5397f26246aSDylan Noblesmith 540bc876017SLang Hames ExecutionEngine *EE = nullptr; 54193de2a12SLang Hames if (ExecutionEngine::OrcMCJITReplacementCtor && UseOrcMCJITReplacement) { 542633fe146SLang Hames EE = ExecutionEngine::OrcMCJITReplacementCtor(ErrorStr, std::move(MemMgr), 543633fe146SLang Hames std::move(Resolver), 54493de2a12SLang Hames std::move(TheTM)); 54593de2a12SLang Hames EE->addModule(std::move(M)); 54693de2a12SLang Hames } else if (ExecutionEngine::MCJITCtor) 547633fe146SLang Hames EE = ExecutionEngine::MCJITCtor(std::move(M), ErrorStr, std::move(MemMgr), 548633fe146SLang Hames std::move(Resolver), std::move(TheTM)); 54993de2a12SLang Hames 550bc876017SLang Hames if (EE) { 551bc876017SLang Hames EE->setVerifyModules(VerifyModules); 552bc876017SLang Hames return EE; 55341fa2bd1SChris Lattner } 554fc8a2d5aSReid Kleckner } 555fc8a2d5aSReid Kleckner 556fc8a2d5aSReid Kleckner // If we can't make a JIT and we didn't request one specifically, try making 557fc8a2d5aSReid Kleckner // an interpreter instead. 55841fa2bd1SChris Lattner if (WhichEngine & EngineKind::Interpreter) { 55941fa2bd1SChris Lattner if (ExecutionEngine::InterpCtor) 5602a8a2795SRafael Espindola return ExecutionEngine::InterpCtor(std::move(M), ErrorStr); 5618bcc6445SChris Lattner if (ErrorStr) 56241fa2bd1SChris Lattner *ErrorStr = "Interpreter has not been linked in."; 5632617dcceSCraig Topper return nullptr; 564fc8a2d5aSReid Kleckner } 565fc8a2d5aSReid Kleckner 56679cc1e3aSEric Christopher if ((WhichEngine & EngineKind::JIT) && !ExecutionEngine::MCJITCtor) { 5678bcc6445SChris Lattner if (ErrorStr) 5688bcc6445SChris Lattner *ErrorStr = "JIT has not been linked in."; 5698bcc6445SChris Lattner } 570868e3f09SDaniel Dunbar 5712617dcceSCraig Topper return nullptr; 572b5163bb9SChris Lattner } 573b5163bb9SChris Lattner 574996fe010SChris Lattner void *ExecutionEngine::getPointerToGlobal(const GlobalValue *GV) { 5751678e859SBrian Gaeke if (Function *F = const_cast<Function*>(dyn_cast<Function>(GV))) 576996fe010SChris Lattner return getPointerToFunction(F); 577996fe010SChris Lattner 5783d5360a4SBenjamin Kramer std::lock_guard<sys::Mutex> locked(lock); 5793dac3f7fSLang Hames if (void* P = getPointerToGlobalIfAvailable(GV)) 580868e3f09SDaniel Dunbar return P; 58169e84901SJeff Cohen 58269e84901SJeff Cohen // Global variable might have been added since interpreter started. 58369e84901SJeff Cohen if (GlobalVariable *GVar = 58469e84901SJeff Cohen const_cast<GlobalVariable *>(dyn_cast<GlobalVariable>(GV))) 58569e84901SJeff Cohen EmitGlobalVariable(GVar); 58669e84901SJeff Cohen else 587fbcc663cSTorok Edwin llvm_unreachable("Global hasn't had an address allocated yet!"); 588868e3f09SDaniel Dunbar 5893dac3f7fSLang Hames return getPointerToGlobalIfAvailable(GV); 590996fe010SChris Lattner } 591996fe010SChris Lattner 5925f8f34e4SAdrian Prantl /// Converts a Constant* into a GenericValue, including handling of 593868e3f09SDaniel Dunbar /// ConstantExpr values. 594996fe010SChris Lattner GenericValue ExecutionEngine::getConstantValue(const Constant *C) { 5956c38f0bbSReid Spencer // If its undefined, return the garbage. 596bcbdbfb3SJay Foad if (isa<UndefValue>(C)) { 597bcbdbfb3SJay Foad GenericValue Result; 598bcbdbfb3SJay Foad switch (C->getType()->getTypeID()) { 599be79a7acSNadav Rotem default: 600be79a7acSNadav Rotem break; 601bcbdbfb3SJay Foad case Type::IntegerTyID: 602bcbdbfb3SJay Foad case Type::X86_FP80TyID: 603bcbdbfb3SJay Foad case Type::FP128TyID: 604bcbdbfb3SJay Foad case Type::PPC_FP128TyID: 605bcbdbfb3SJay Foad // Although the value is undefined, we still have to construct an APInt 606bcbdbfb3SJay Foad // with the correct bit width. 607bcbdbfb3SJay Foad Result.IntVal = APInt(C->getType()->getPrimitiveSizeInBits(), 0); 608bcbdbfb3SJay Foad break; 6098e97f016SElena Demikhovsky case Type::StructTyID: { 6108e97f016SElena Demikhovsky // if the whole struct is 'undef' just reserve memory for the value. 6118e97f016SElena Demikhovsky if(StructType *STy = dyn_cast<StructType>(C->getType())) { 6128e97f016SElena Demikhovsky unsigned int elemNum = STy->getNumElements(); 6138e97f016SElena Demikhovsky Result.AggregateVal.resize(elemNum); 6148e97f016SElena Demikhovsky for (unsigned int i = 0; i < elemNum; ++i) { 6158e97f016SElena Demikhovsky Type *ElemTy = STy->getElementType(i); 6168e97f016SElena Demikhovsky if (ElemTy->isIntegerTy()) 6178e97f016SElena Demikhovsky Result.AggregateVal[i].IntVal = 6188e97f016SElena Demikhovsky APInt(ElemTy->getPrimitiveSizeInBits(), 0); 6198e97f016SElena Demikhovsky else if (ElemTy->isAggregateType()) { 6208e97f016SElena Demikhovsky const Constant *ElemUndef = UndefValue::get(ElemTy); 6218e97f016SElena Demikhovsky Result.AggregateVal[i] = getConstantValue(ElemUndef); 6228e97f016SElena Demikhovsky } 6238e97f016SElena Demikhovsky } 6248e97f016SElena Demikhovsky } 6258e97f016SElena Demikhovsky } 6268e97f016SElena Demikhovsky break; 627be79a7acSNadav Rotem case Type::VectorTyID: 628be79a7acSNadav Rotem // if the whole vector is 'undef' just reserve memory for the value. 629e3dcce97SCraig Topper auto* VTy = dyn_cast<VectorType>(C->getType()); 630e3dcce97SCraig Topper Type *ElemTy = VTy->getElementType(); 631be79a7acSNadav Rotem unsigned int elemNum = VTy->getNumElements(); 632be79a7acSNadav Rotem Result.AggregateVal.resize(elemNum); 633be79a7acSNadav Rotem if (ElemTy->isIntegerTy()) 634be79a7acSNadav Rotem for (unsigned int i = 0; i < elemNum; ++i) 635be79a7acSNadav Rotem Result.AggregateVal[i].IntVal = 636be79a7acSNadav Rotem APInt(ElemTy->getPrimitiveSizeInBits(), 0); 637bcbdbfb3SJay Foad break; 638bcbdbfb3SJay Foad } 639bcbdbfb3SJay Foad return Result; 640bcbdbfb3SJay Foad } 6419de0d14dSChris Lattner 642868e3f09SDaniel Dunbar // Otherwise, if the value is a ConstantExpr... 6436c38f0bbSReid Spencer if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) { 6444fd528f2SReid Spencer Constant *Op0 = CE->getOperand(0); 6459de0d14dSChris Lattner switch (CE->getOpcode()) { 6469de0d14dSChris Lattner case Instruction::GetElementPtr: { 6476c38f0bbSReid Spencer // Compute the index 6484fd528f2SReid Spencer GenericValue Result = getConstantValue(Op0); 649a3fcefb6SMehdi Amini APInt Offset(DL.getPointerSizeInBits(), 0); 650a3fcefb6SMehdi Amini cast<GEPOperator>(CE)->accumulateConstantOffset(DL, Offset); 6519de0d14dSChris Lattner 65287aa65f4SReid Spencer char* tmp = (char*) Result.PointerVal; 653b6ad9822SNuno Lopes Result = PTOGV(tmp + Offset.getSExtValue()); 6549de0d14dSChris Lattner return Result; 6559de0d14dSChris Lattner } 6564fd528f2SReid Spencer case Instruction::Trunc: { 6574fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 6584fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); 6594fd528f2SReid Spencer GV.IntVal = GV.IntVal.trunc(BitWidth); 6604fd528f2SReid Spencer return GV; 6614fd528f2SReid Spencer } 6624fd528f2SReid Spencer case Instruction::ZExt: { 6634fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 6644fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); 6654fd528f2SReid Spencer GV.IntVal = GV.IntVal.zext(BitWidth); 6664fd528f2SReid Spencer return GV; 6674fd528f2SReid Spencer } 6684fd528f2SReid Spencer case Instruction::SExt: { 6694fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 6704fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); 6714fd528f2SReid Spencer GV.IntVal = GV.IntVal.sext(BitWidth); 6724fd528f2SReid Spencer return GV; 6734fd528f2SReid Spencer } 6744fd528f2SReid Spencer case Instruction::FPTrunc: { 675a1336cf5SDale Johannesen // FIXME long double 6764fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 6774fd528f2SReid Spencer GV.FloatVal = float(GV.DoubleVal); 6784fd528f2SReid Spencer return GV; 6794fd528f2SReid Spencer } 6804fd528f2SReid Spencer case Instruction::FPExt:{ 681a1336cf5SDale Johannesen // FIXME long double 6824fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 6834fd528f2SReid Spencer GV.DoubleVal = double(GV.FloatVal); 6844fd528f2SReid Spencer return GV; 6854fd528f2SReid Spencer } 6864fd528f2SReid Spencer case Instruction::UIToFP: { 6874fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 688fdd87907SChris Lattner if (CE->getType()->isFloatTy()) 6894fd528f2SReid Spencer GV.FloatVal = float(GV.IntVal.roundToDouble()); 690fdd87907SChris Lattner else if (CE->getType()->isDoubleTy()) 6914fd528f2SReid Spencer GV.DoubleVal = GV.IntVal.roundToDouble(); 692fdd87907SChris Lattner else if (CE->getType()->isX86_FP80Ty()) { 69317c7f703SStephan Bergmann APFloat apf = APFloat::getZero(APFloat::x87DoubleExtended()); 694ca24fd90SDan Gohman (void)apf.convertFromAPInt(GV.IntVal, 695ca24fd90SDan Gohman false, 6969150652bSDale Johannesen APFloat::rmNearestTiesToEven); 69754306fe4SDale Johannesen GV.IntVal = apf.bitcastToAPInt(); 698a1336cf5SDale Johannesen } 6994fd528f2SReid Spencer return GV; 7004fd528f2SReid Spencer } 7014fd528f2SReid Spencer case Instruction::SIToFP: { 7024fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 703fdd87907SChris Lattner if (CE->getType()->isFloatTy()) 7044fd528f2SReid Spencer GV.FloatVal = float(GV.IntVal.signedRoundToDouble()); 705fdd87907SChris Lattner else if (CE->getType()->isDoubleTy()) 7064fd528f2SReid Spencer GV.DoubleVal = GV.IntVal.signedRoundToDouble(); 707fdd87907SChris Lattner else if (CE->getType()->isX86_FP80Ty()) { 70817c7f703SStephan Bergmann APFloat apf = APFloat::getZero(APFloat::x87DoubleExtended()); 709ca24fd90SDan Gohman (void)apf.convertFromAPInt(GV.IntVal, 710ca24fd90SDan Gohman true, 7119150652bSDale Johannesen APFloat::rmNearestTiesToEven); 71254306fe4SDale Johannesen GV.IntVal = apf.bitcastToAPInt(); 713a1336cf5SDale Johannesen } 7144fd528f2SReid Spencer return GV; 7154fd528f2SReid Spencer } 7164fd528f2SReid Spencer case Instruction::FPToUI: // double->APInt conversion handles sign 7174fd528f2SReid Spencer case Instruction::FPToSI: { 7184fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 7194fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); 720fdd87907SChris Lattner if (Op0->getType()->isFloatTy()) 7214fd528f2SReid Spencer GV.IntVal = APIntOps::RoundFloatToAPInt(GV.FloatVal, BitWidth); 722fdd87907SChris Lattner else if (Op0->getType()->isDoubleTy()) 7234fd528f2SReid Spencer GV.IntVal = APIntOps::RoundDoubleToAPInt(GV.DoubleVal, BitWidth); 724fdd87907SChris Lattner else if (Op0->getType()->isX86_FP80Ty()) { 72517c7f703SStephan Bergmann APFloat apf = APFloat(APFloat::x87DoubleExtended(), GV.IntVal); 726a1336cf5SDale Johannesen uint64_t v; 7274f0bd68cSDale Johannesen bool ignored; 72800b34996SSimon Pilgrim (void)apf.convertToInteger(makeMutableArrayRef(v), BitWidth, 729a1336cf5SDale Johannesen CE->getOpcode()==Instruction::FPToSI, 7304f0bd68cSDale Johannesen APFloat::rmTowardZero, &ignored); 731a1336cf5SDale Johannesen GV.IntVal = v; // endian? 732a1336cf5SDale Johannesen } 7334fd528f2SReid Spencer return GV; 7344fd528f2SReid Spencer } 7356c38f0bbSReid Spencer case Instruction::PtrToInt: { 7364fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 737a3fcefb6SMehdi Amini uint32_t PtrWidth = DL.getTypeSizeInBits(Op0->getType()); 738fc1f2cd3SEli Friedman assert(PtrWidth <= 64 && "Bad pointer width"); 7394fd528f2SReid Spencer GV.IntVal = APInt(PtrWidth, uintptr_t(GV.PointerVal)); 740a3fcefb6SMehdi Amini uint32_t IntWidth = DL.getTypeSizeInBits(CE->getType()); 741fc1f2cd3SEli Friedman GV.IntVal = GV.IntVal.zextOrTrunc(IntWidth); 7424fd528f2SReid Spencer return GV; 7434fd528f2SReid Spencer } 7444fd528f2SReid Spencer case Instruction::IntToPtr: { 7454fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 746a3fcefb6SMehdi Amini uint32_t PtrWidth = DL.getTypeSizeInBits(CE->getType()); 7474fd528f2SReid Spencer GV.IntVal = GV.IntVal.zextOrTrunc(PtrWidth); 7484fd528f2SReid Spencer assert(GV.IntVal.getBitWidth() <= 64 && "Bad pointer width"); 7494fd528f2SReid Spencer GV.PointerVal = PointerTy(uintptr_t(GV.IntVal.getZExtValue())); 7506c38f0bbSReid Spencer return GV; 7516c38f0bbSReid Spencer } 7526c38f0bbSReid Spencer case Instruction::BitCast: { 7534fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 754229907cdSChris Lattner Type* DestTy = CE->getType(); 7554fd528f2SReid Spencer switch (Op0->getType()->getTypeID()) { 756fbcc663cSTorok Edwin default: llvm_unreachable("Invalid bitcast operand"); 7574fd528f2SReid Spencer case Type::IntegerTyID: 7589dff9becSDuncan Sands assert(DestTy->isFloatingPointTy() && "invalid bitcast"); 759fdd87907SChris Lattner if (DestTy->isFloatTy()) 7604fd528f2SReid Spencer GV.FloatVal = GV.IntVal.bitsToFloat(); 761fdd87907SChris Lattner else if (DestTy->isDoubleTy()) 7624fd528f2SReid Spencer GV.DoubleVal = GV.IntVal.bitsToDouble(); 7636c38f0bbSReid Spencer break; 7644fd528f2SReid Spencer case Type::FloatTyID: 7659dff9becSDuncan Sands assert(DestTy->isIntegerTy(32) && "Invalid bitcast"); 7663447fb01SJay Foad GV.IntVal = APInt::floatToBits(GV.FloatVal); 7674fd528f2SReid Spencer break; 7684fd528f2SReid Spencer case Type::DoubleTyID: 7699dff9becSDuncan Sands assert(DestTy->isIntegerTy(64) && "Invalid bitcast"); 7703447fb01SJay Foad GV.IntVal = APInt::doubleToBits(GV.DoubleVal); 7714fd528f2SReid Spencer break; 7724fd528f2SReid Spencer case Type::PointerTyID: 77319d0b47bSDuncan Sands assert(DestTy->isPointerTy() && "Invalid bitcast"); 7744fd528f2SReid Spencer break; // getConstantValue(Op0) above already converted it 7756c38f0bbSReid Spencer } 7764fd528f2SReid Spencer return GV; 77768cbcc3eSChris Lattner } 77868cbcc3eSChris Lattner case Instruction::Add: 779a5b9645cSDan Gohman case Instruction::FAdd: 7804fd528f2SReid Spencer case Instruction::Sub: 781a5b9645cSDan Gohman case Instruction::FSub: 7824fd528f2SReid Spencer case Instruction::Mul: 783a5b9645cSDan Gohman case Instruction::FMul: 7844fd528f2SReid Spencer case Instruction::UDiv: 7854fd528f2SReid Spencer case Instruction::SDiv: 7864fd528f2SReid Spencer case Instruction::URem: 7874fd528f2SReid Spencer case Instruction::SRem: 7884fd528f2SReid Spencer case Instruction::And: 7894fd528f2SReid Spencer case Instruction::Or: 7904fd528f2SReid Spencer case Instruction::Xor: { 7914fd528f2SReid Spencer GenericValue LHS = getConstantValue(Op0); 7924fd528f2SReid Spencer GenericValue RHS = getConstantValue(CE->getOperand(1)); 7934fd528f2SReid Spencer GenericValue GV; 794c4e6bb5fSChris Lattner switch (CE->getOperand(0)->getType()->getTypeID()) { 795fbcc663cSTorok Edwin default: llvm_unreachable("Bad add type!"); 7967a9c62baSReid Spencer case Type::IntegerTyID: 7974fd528f2SReid Spencer switch (CE->getOpcode()) { 798fbcc663cSTorok Edwin default: llvm_unreachable("Invalid integer opcode"); 7994fd528f2SReid Spencer case Instruction::Add: GV.IntVal = LHS.IntVal + RHS.IntVal; break; 8004fd528f2SReid Spencer case Instruction::Sub: GV.IntVal = LHS.IntVal - RHS.IntVal; break; 8014fd528f2SReid Spencer case Instruction::Mul: GV.IntVal = LHS.IntVal * RHS.IntVal; break; 8024fd528f2SReid Spencer case Instruction::UDiv:GV.IntVal = LHS.IntVal.udiv(RHS.IntVal); break; 8034fd528f2SReid Spencer case Instruction::SDiv:GV.IntVal = LHS.IntVal.sdiv(RHS.IntVal); break; 8044fd528f2SReid Spencer case Instruction::URem:GV.IntVal = LHS.IntVal.urem(RHS.IntVal); break; 8054fd528f2SReid Spencer case Instruction::SRem:GV.IntVal = LHS.IntVal.srem(RHS.IntVal); break; 8064fd528f2SReid Spencer case Instruction::And: GV.IntVal = LHS.IntVal & RHS.IntVal; break; 8074fd528f2SReid Spencer case Instruction::Or: GV.IntVal = LHS.IntVal | RHS.IntVal; break; 8084fd528f2SReid Spencer case Instruction::Xor: GV.IntVal = LHS.IntVal ^ RHS.IntVal; break; 8094fd528f2SReid Spencer } 810c4e6bb5fSChris Lattner break; 811c4e6bb5fSChris Lattner case Type::FloatTyID: 8124fd528f2SReid Spencer switch (CE->getOpcode()) { 813fbcc663cSTorok Edwin default: llvm_unreachable("Invalid float opcode"); 814a5b9645cSDan Gohman case Instruction::FAdd: 8154fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal + RHS.FloatVal; break; 816a5b9645cSDan Gohman case Instruction::FSub: 8174fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal - RHS.FloatVal; break; 818a5b9645cSDan Gohman case Instruction::FMul: 8194fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal * RHS.FloatVal; break; 8204fd528f2SReid Spencer case Instruction::FDiv: 8214fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal / RHS.FloatVal; break; 8224fd528f2SReid Spencer case Instruction::FRem: 82393cd0f1cSChris Lattner GV.FloatVal = std::fmod(LHS.FloatVal,RHS.FloatVal); break; 8244fd528f2SReid Spencer } 825c4e6bb5fSChris Lattner break; 826c4e6bb5fSChris Lattner case Type::DoubleTyID: 8274fd528f2SReid Spencer switch (CE->getOpcode()) { 828fbcc663cSTorok Edwin default: llvm_unreachable("Invalid double opcode"); 829a5b9645cSDan Gohman case Instruction::FAdd: 8304fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal + RHS.DoubleVal; break; 831a5b9645cSDan Gohman case Instruction::FSub: 8324fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal - RHS.DoubleVal; break; 833a5b9645cSDan Gohman case Instruction::FMul: 8344fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal * RHS.DoubleVal; break; 8354fd528f2SReid Spencer case Instruction::FDiv: 8364fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal / RHS.DoubleVal; break; 8374fd528f2SReid Spencer case Instruction::FRem: 83893cd0f1cSChris Lattner GV.DoubleVal = std::fmod(LHS.DoubleVal,RHS.DoubleVal); break; 8394fd528f2SReid Spencer } 840c4e6bb5fSChris Lattner break; 841a1336cf5SDale Johannesen case Type::X86_FP80TyID: 842a1336cf5SDale Johannesen case Type::PPC_FP128TyID: 843a1336cf5SDale Johannesen case Type::FP128TyID: { 84429178a34STim Northover const fltSemantics &Sem = CE->getOperand(0)->getType()->getFltSemantics(); 84529178a34STim Northover APFloat apfLHS = APFloat(Sem, LHS.IntVal); 846a1336cf5SDale Johannesen switch (CE->getOpcode()) { 847e4f47434SDaniel Dunbar default: llvm_unreachable("Invalid long double opcode"); 848a5b9645cSDan Gohman case Instruction::FAdd: 84929178a34STim Northover apfLHS.add(APFloat(Sem, RHS.IntVal), APFloat::rmNearestTiesToEven); 85054306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt(); 851a1336cf5SDale Johannesen break; 852a5b9645cSDan Gohman case Instruction::FSub: 85329178a34STim Northover apfLHS.subtract(APFloat(Sem, RHS.IntVal), 85429178a34STim Northover APFloat::rmNearestTiesToEven); 85554306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt(); 856a1336cf5SDale Johannesen break; 857a5b9645cSDan Gohman case Instruction::FMul: 85829178a34STim Northover apfLHS.multiply(APFloat(Sem, RHS.IntVal), 85929178a34STim Northover APFloat::rmNearestTiesToEven); 86054306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt(); 861a1336cf5SDale Johannesen break; 862a1336cf5SDale Johannesen case Instruction::FDiv: 86329178a34STim Northover apfLHS.divide(APFloat(Sem, RHS.IntVal), 86429178a34STim Northover APFloat::rmNearestTiesToEven); 86554306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt(); 866a1336cf5SDale Johannesen break; 867a1336cf5SDale Johannesen case Instruction::FRem: 868b12db0e4SStephen Canon apfLHS.mod(APFloat(Sem, RHS.IntVal)); 86954306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt(); 870a1336cf5SDale Johannesen break; 871a1336cf5SDale Johannesen } 872a1336cf5SDale Johannesen } 873a1336cf5SDale Johannesen break; 874c4e6bb5fSChris Lattner } 8754fd528f2SReid Spencer return GV; 8764fd528f2SReid Spencer } 8779de0d14dSChris Lattner default: 87868cbcc3eSChris Lattner break; 87968cbcc3eSChris Lattner } 880868e3f09SDaniel Dunbar 881868e3f09SDaniel Dunbar SmallString<256> Msg; 882868e3f09SDaniel Dunbar raw_svector_ostream OS(Msg); 883868e3f09SDaniel Dunbar OS << "ConstantExpr not handled: " << *CE; 884868e3f09SDaniel Dunbar report_fatal_error(OS.str()); 8859de0d14dSChris Lattner } 886996fe010SChris Lattner 887868e3f09SDaniel Dunbar // Otherwise, we have a simple constant. 8884fd528f2SReid Spencer GenericValue Result; 8896b727599SChris Lattner switch (C->getType()->getTypeID()) { 89087aa65f4SReid Spencer case Type::FloatTyID: 891bed9dc42SDale Johannesen Result.FloatVal = cast<ConstantFP>(C)->getValueAPF().convertToFloat(); 8927a9c62baSReid Spencer break; 89387aa65f4SReid Spencer case Type::DoubleTyID: 894bed9dc42SDale Johannesen Result.DoubleVal = cast<ConstantFP>(C)->getValueAPF().convertToDouble(); 89587aa65f4SReid Spencer break; 896a1336cf5SDale Johannesen case Type::X86_FP80TyID: 897a1336cf5SDale Johannesen case Type::FP128TyID: 898a1336cf5SDale Johannesen case Type::PPC_FP128TyID: 89954306fe4SDale Johannesen Result.IntVal = cast <ConstantFP>(C)->getValueAPF().bitcastToAPInt(); 900a1336cf5SDale Johannesen break; 90187aa65f4SReid Spencer case Type::IntegerTyID: 90287aa65f4SReid Spencer Result.IntVal = cast<ConstantInt>(C)->getValue(); 90387aa65f4SReid Spencer break; 904996fe010SChris Lattner case Type::PointerTyID: 9057d60b905SLang Hames while (auto *A = dyn_cast<GlobalAlias>(C)) { 9067d60b905SLang Hames C = A->getAliasee(); 9077d60b905SLang Hames } 9086a0fd73bSReid Spencer if (isa<ConstantPointerNull>(C)) 9092617dcceSCraig Topper Result.PointerVal = nullptr; 9106a0fd73bSReid Spencer else if (const Function *F = dyn_cast<Function>(C)) 9116a0fd73bSReid Spencer Result = PTOGV(getPointerToFunctionOrStub(const_cast<Function*>(F))); 9126a0fd73bSReid Spencer else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(C)) 9136a0fd73bSReid Spencer Result = PTOGV(getOrEmitGlobalVariable(const_cast<GlobalVariable*>(GV))); 914e6492f10SChris Lattner else 915fbcc663cSTorok Edwin llvm_unreachable("Unknown constant pointer type!"); 916996fe010SChris Lattner break; 917be79a7acSNadav Rotem case Type::VectorTyID: { 918be79a7acSNadav Rotem unsigned elemNum; 919be79a7acSNadav Rotem Type* ElemTy; 920be79a7acSNadav Rotem const ConstantDataVector *CDV = dyn_cast<ConstantDataVector>(C); 921be79a7acSNadav Rotem const ConstantVector *CV = dyn_cast<ConstantVector>(C); 922be79a7acSNadav Rotem const ConstantAggregateZero *CAZ = dyn_cast<ConstantAggregateZero>(C); 923be79a7acSNadav Rotem 924be79a7acSNadav Rotem if (CDV) { 925be79a7acSNadav Rotem elemNum = CDV->getNumElements(); 926be79a7acSNadav Rotem ElemTy = CDV->getElementType(); 927be79a7acSNadav Rotem } else if (CV || CAZ) { 928be79a7acSNadav Rotem VectorType* VTy = dyn_cast<VectorType>(C->getType()); 929be79a7acSNadav Rotem elemNum = VTy->getNumElements(); 930be79a7acSNadav Rotem ElemTy = VTy->getElementType(); 931be79a7acSNadav Rotem } else { 932be79a7acSNadav Rotem llvm_unreachable("Unknown constant vector type!"); 933be79a7acSNadav Rotem } 934be79a7acSNadav Rotem 935be79a7acSNadav Rotem Result.AggregateVal.resize(elemNum); 936be79a7acSNadav Rotem // Check if vector holds floats. 937be79a7acSNadav Rotem if(ElemTy->isFloatTy()) { 938be79a7acSNadav Rotem if (CAZ) { 939be79a7acSNadav Rotem GenericValue floatZero; 940be79a7acSNadav Rotem floatZero.FloatVal = 0.f; 941be79a7acSNadav Rotem std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(), 942be79a7acSNadav Rotem floatZero); 943be79a7acSNadav Rotem break; 944be79a7acSNadav Rotem } 945be79a7acSNadav Rotem if(CV) { 946be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i) 947be79a7acSNadav Rotem if (!isa<UndefValue>(CV->getOperand(i))) 948be79a7acSNadav Rotem Result.AggregateVal[i].FloatVal = cast<ConstantFP>( 949be79a7acSNadav Rotem CV->getOperand(i))->getValueAPF().convertToFloat(); 950be79a7acSNadav Rotem break; 951be79a7acSNadav Rotem } 952be79a7acSNadav Rotem if(CDV) 953be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i) 954be79a7acSNadav Rotem Result.AggregateVal[i].FloatVal = CDV->getElementAsFloat(i); 955be79a7acSNadav Rotem 956be79a7acSNadav Rotem break; 957be79a7acSNadav Rotem } 958be79a7acSNadav Rotem // Check if vector holds doubles. 959be79a7acSNadav Rotem if (ElemTy->isDoubleTy()) { 960be79a7acSNadav Rotem if (CAZ) { 961be79a7acSNadav Rotem GenericValue doubleZero; 962be79a7acSNadav Rotem doubleZero.DoubleVal = 0.0; 963be79a7acSNadav Rotem std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(), 964be79a7acSNadav Rotem doubleZero); 965be79a7acSNadav Rotem break; 966be79a7acSNadav Rotem } 967be79a7acSNadav Rotem if(CV) { 968be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i) 969be79a7acSNadav Rotem if (!isa<UndefValue>(CV->getOperand(i))) 970be79a7acSNadav Rotem Result.AggregateVal[i].DoubleVal = cast<ConstantFP>( 971be79a7acSNadav Rotem CV->getOperand(i))->getValueAPF().convertToDouble(); 972be79a7acSNadav Rotem break; 973be79a7acSNadav Rotem } 974be79a7acSNadav Rotem if(CDV) 975be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i) 976be79a7acSNadav Rotem Result.AggregateVal[i].DoubleVal = CDV->getElementAsDouble(i); 977be79a7acSNadav Rotem 978be79a7acSNadav Rotem break; 979be79a7acSNadav Rotem } 980be79a7acSNadav Rotem // Check if vector holds integers. 981be79a7acSNadav Rotem if (ElemTy->isIntegerTy()) { 982be79a7acSNadav Rotem if (CAZ) { 983be79a7acSNadav Rotem GenericValue intZero; 984be79a7acSNadav Rotem intZero.IntVal = APInt(ElemTy->getScalarSizeInBits(), 0ull); 985be79a7acSNadav Rotem std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(), 986be79a7acSNadav Rotem intZero); 987be79a7acSNadav Rotem break; 988be79a7acSNadav Rotem } 989be79a7acSNadav Rotem if(CV) { 990be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i) 991be79a7acSNadav Rotem if (!isa<UndefValue>(CV->getOperand(i))) 992be79a7acSNadav Rotem Result.AggregateVal[i].IntVal = cast<ConstantInt>( 993be79a7acSNadav Rotem CV->getOperand(i))->getValue(); 994be79a7acSNadav Rotem else { 995be79a7acSNadav Rotem Result.AggregateVal[i].IntVal = 996be79a7acSNadav Rotem APInt(CV->getOperand(i)->getType()->getPrimitiveSizeInBits(), 0); 997be79a7acSNadav Rotem } 998be79a7acSNadav Rotem break; 999be79a7acSNadav Rotem } 1000be79a7acSNadav Rotem if(CDV) 1001be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i) 1002be79a7acSNadav Rotem Result.AggregateVal[i].IntVal = APInt( 1003be79a7acSNadav Rotem CDV->getElementType()->getPrimitiveSizeInBits(), 1004be79a7acSNadav Rotem CDV->getElementAsInteger(i)); 1005be79a7acSNadav Rotem 1006be79a7acSNadav Rotem break; 1007be79a7acSNadav Rotem } 1008be79a7acSNadav Rotem llvm_unreachable("Unknown constant pointer type!"); 1009be79a7acSNadav Rotem } 1010be79a7acSNadav Rotem break; 1011be79a7acSNadav Rotem 1012996fe010SChris Lattner default: 1013868e3f09SDaniel Dunbar SmallString<256> Msg; 1014868e3f09SDaniel Dunbar raw_svector_ostream OS(Msg); 1015868e3f09SDaniel Dunbar OS << "ERROR: Constant unimplemented for type: " << *C->getType(); 1016868e3f09SDaniel Dunbar report_fatal_error(OS.str()); 1017996fe010SChris Lattner } 1018868e3f09SDaniel Dunbar 1019996fe010SChris Lattner return Result; 1020996fe010SChris Lattner } 1021996fe010SChris Lattner 102209053e62SEvan Cheng void ExecutionEngine::StoreValueToMemory(const GenericValue &Val, 1023229907cdSChris Lattner GenericValue *Ptr, Type *Ty) { 1024a3fcefb6SMehdi Amini const unsigned StoreBytes = getDataLayout().getTypeStoreSize(Ty); 10251202d1b1SDuncan Sands 10261202d1b1SDuncan Sands switch (Ty->getTypeID()) { 1027be79a7acSNadav Rotem default: 1028be79a7acSNadav Rotem dbgs() << "Cannot store value of type " << *Ty << "!\n"; 1029be79a7acSNadav Rotem break; 10301202d1b1SDuncan Sands case Type::IntegerTyID: 10311202d1b1SDuncan Sands StoreIntToMemory(Val.IntVal, (uint8_t*)Ptr, StoreBytes); 10321202d1b1SDuncan Sands break; 1033996fe010SChris Lattner case Type::FloatTyID: 103487aa65f4SReid Spencer *((float*)Ptr) = Val.FloatVal; 103587aa65f4SReid Spencer break; 103687aa65f4SReid Spencer case Type::DoubleTyID: 103787aa65f4SReid Spencer *((double*)Ptr) = Val.DoubleVal; 1038996fe010SChris Lattner break; 10394d7e4ee7SDale Johannesen case Type::X86_FP80TyID: 10404d7e4ee7SDale Johannesen memcpy(Ptr, Val.IntVal.getRawData(), 10); 1041a1336cf5SDale Johannesen break; 10427a9c62baSReid Spencer case Type::PointerTyID: 10431202d1b1SDuncan Sands // Ensure 64 bit target pointers are fully initialized on 32 bit hosts. 10441202d1b1SDuncan Sands if (StoreBytes != sizeof(PointerTy)) 104593da3c82SChandler Carruth memset(&(Ptr->PointerVal), 0, StoreBytes); 10461202d1b1SDuncan Sands 104787aa65f4SReid Spencer *((PointerTy*)Ptr) = Val.PointerVal; 1048996fe010SChris Lattner break; 1049be79a7acSNadav Rotem case Type::VectorTyID: 1050be79a7acSNadav Rotem for (unsigned i = 0; i < Val.AggregateVal.size(); ++i) { 1051be79a7acSNadav Rotem if (cast<VectorType>(Ty)->getElementType()->isDoubleTy()) 1052be79a7acSNadav Rotem *(((double*)Ptr)+i) = Val.AggregateVal[i].DoubleVal; 1053be79a7acSNadav Rotem if (cast<VectorType>(Ty)->getElementType()->isFloatTy()) 1054be79a7acSNadav Rotem *(((float*)Ptr)+i) = Val.AggregateVal[i].FloatVal; 1055be79a7acSNadav Rotem if (cast<VectorType>(Ty)->getElementType()->isIntegerTy()) { 1056be79a7acSNadav Rotem unsigned numOfBytes =(Val.AggregateVal[i].IntVal.getBitWidth()+7)/8; 1057be79a7acSNadav Rotem StoreIntToMemory(Val.AggregateVal[i].IntVal, 1058be79a7acSNadav Rotem (uint8_t*)Ptr + numOfBytes*i, numOfBytes); 1059be79a7acSNadav Rotem } 1060be79a7acSNadav Rotem } 1061be79a7acSNadav Rotem break; 1062996fe010SChris Lattner } 10631202d1b1SDuncan Sands 1064a3fcefb6SMehdi Amini if (sys::IsLittleEndianHost != getDataLayout().isLittleEndian()) 10651202d1b1SDuncan Sands // Host and target are different endian - reverse the stored bytes. 10661202d1b1SDuncan Sands std::reverse((uint8_t*)Ptr, StoreBytes + (uint8_t*)Ptr); 1067996fe010SChris Lattner } 1068996fe010SChris Lattner 10691202d1b1SDuncan Sands /// FIXME: document 10701202d1b1SDuncan Sands /// 10711202d1b1SDuncan Sands void ExecutionEngine::LoadValueFromMemory(GenericValue &Result, 10721202d1b1SDuncan Sands GenericValue *Ptr, 1073229907cdSChris Lattner Type *Ty) { 1074a3fcefb6SMehdi Amini const unsigned LoadBytes = getDataLayout().getTypeStoreSize(Ty); 10751202d1b1SDuncan Sands 10761202d1b1SDuncan Sands switch (Ty->getTypeID()) { 10771202d1b1SDuncan Sands case Type::IntegerTyID: 10781202d1b1SDuncan Sands // An APInt with all words initially zero. 10791202d1b1SDuncan Sands Result.IntVal = APInt(cast<IntegerType>(Ty)->getBitWidth(), 0); 10801202d1b1SDuncan Sands LoadIntFromMemory(Result.IntVal, (uint8_t*)Ptr, LoadBytes); 10811202d1b1SDuncan Sands break; 10827f389e8cSChris Lattner case Type::FloatTyID: 108387aa65f4SReid Spencer Result.FloatVal = *((float*)Ptr); 108487aa65f4SReid Spencer break; 108587aa65f4SReid Spencer case Type::DoubleTyID: 108687aa65f4SReid Spencer Result.DoubleVal = *((double*)Ptr); 10877f389e8cSChris Lattner break; 10887a9c62baSReid Spencer case Type::PointerTyID: 108987aa65f4SReid Spencer Result.PointerVal = *((PointerTy*)Ptr); 10907f389e8cSChris Lattner break; 1091a1336cf5SDale Johannesen case Type::X86_FP80TyID: { 1092a1336cf5SDale Johannesen // This is endian dependent, but it will only work on x86 anyway. 109326d6539eSDuncan Sands // FIXME: Will not trap if loading a signaling NaN. 1094ff306287SDuncan Sands uint64_t y[2]; 10954d7e4ee7SDale Johannesen memcpy(y, Ptr, 10); 10967a162881SJeffrey Yasskin Result.IntVal = APInt(80, y); 1097a1336cf5SDale Johannesen break; 1098a1336cf5SDale Johannesen } 1099be79a7acSNadav Rotem case Type::VectorTyID: { 1100e3dcce97SCraig Topper auto *VT = cast<VectorType>(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 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. 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 12030621caefSChris Lattner const GlobalValue *&GVEntry = 120449bb65a4SRafael Espindola LinkedGlobalsMap[std::make_pair(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()) { 12310621caefSChris Lattner if (const GlobalValue *GVEntry = 123249bb65a4SRafael Espindola LinkedGlobalsMap[std::make_pair(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. 12460621caefSChris Lattner if (void *SymAddr = 124749bb65a4SRafael Espindola sys::DynamicLibrary::SearchForAddressOfSymbol(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()) { 12590621caefSChris Lattner for (unsigned i = 0, e = NonCanonicalGlobals.size(); i != e; ++i) { 12600621caefSChris Lattner const GlobalValue *GV = NonCanonicalGlobals[i]; 12610621caefSChris Lattner const GlobalValue *CGV = 12620621caefSChris Lattner LinkedGlobalsMap[std::make_pair(GV->getName(), GV->getType())]; 12630621caefSChris Lattner void *Ptr = getPointerToGlobalIfAvailable(CGV); 12640621caefSChris Lattner assert(Ptr && "Canonical global wasn't codegen'd!"); 1265a67f06b9SNuno Lopes addGlobalMapping(GV, Ptr); 12660621caefSChris Lattner } 12670621caefSChris Lattner } 1268996fe010SChris Lattner 12697a9c62baSReid Spencer // Now that all of the globals are set up in memory, loop through them all 12707a9c62baSReid Spencer // and initialize their contents. 127149bb65a4SRafael Espindola for (const auto &GV : M.globals()) { 127249bb65a4SRafael Espindola if (!GV.isDeclaration()) { 12730621caefSChris Lattner if (!LinkedGlobalsMap.empty()) { 12740621caefSChris Lattner if (const GlobalValue *GVEntry = 127549bb65a4SRafael Espindola LinkedGlobalsMap[std::make_pair(GV.getName(), GV.getType())]) 127649bb65a4SRafael Espindola if (GVEntry != &GV) // Not the canonical variable. 12770621caefSChris Lattner continue; 12780621caefSChris Lattner } 127949bb65a4SRafael Espindola EmitGlobalVariable(&GV); 12806bbe3eceSChris Lattner } 12810621caefSChris Lattner } 12820621caefSChris Lattner } 12830621caefSChris Lattner } 12846bbe3eceSChris Lattner 12856bbe3eceSChris Lattner // EmitGlobalVariable - This method emits the specified global variable to the 12866bbe3eceSChris Lattner // address specified in GlobalAddresses, or allocates new memory if it's not 12876bbe3eceSChris Lattner // already in the map. 1288fbcc0aa1SChris Lattner void ExecutionEngine::EmitGlobalVariable(const GlobalVariable *GV) { 1289748e8579SChris Lattner void *GA = getPointerToGlobalIfAvailable(GV); 1290dc631735SChris Lattner 12912617dcceSCraig Topper if (!GA) { 12926bbe3eceSChris Lattner // If it's not already specified, allocate memory for the global. 12935457ce9aSNicolas Geoffray GA = getMemoryForGV(GV); 12943b442379SAndrew Kaylor 12953b442379SAndrew Kaylor // If we failed to allocate memory for this global, return. 12962617dcceSCraig Topper if (!GA) return; 12973b442379SAndrew Kaylor 1298748e8579SChris Lattner addGlobalMapping(GV, GA); 12996bbe3eceSChris Lattner } 1300fbcc0aa1SChris Lattner 13015457ce9aSNicolas Geoffray // Don't initialize if it's thread local, let the client do it. 13025457ce9aSNicolas Geoffray if (!GV->isThreadLocal()) 13036bbe3eceSChris Lattner InitializeMemory(GV->getInitializer(), GA); 13045457ce9aSNicolas Geoffray 13055f6eaac6SManuel Jacob Type *ElTy = GV->getValueType(); 1306a3fcefb6SMehdi Amini size_t GVSize = (size_t)getDataLayout().getTypeAllocSize(ElTy); 1307df1f1524SChris Lattner NumInitBytes += (unsigned)GVSize; 13086bbe3eceSChris Lattner ++NumGlobals; 1309996fe010SChris Lattner } 1310