1857c21b4SMisha Brukman //===-- ExecutionEngine.cpp - Common Implementation shared by EEs ---------===// 2996fe010SChris Lattner // 3482202a6SJohn Criswell // The LLVM Compiler Infrastructure 4482202a6SJohn Criswell // 5f3ebc3f3SChris Lattner // This file is distributed under the University of Illinois Open Source 6f3ebc3f3SChris Lattner // License. See LICENSE.TXT for details. 7482202a6SJohn Criswell // 8482202a6SJohn Criswell //===----------------------------------------------------------------------===// 9482202a6SJohn Criswell // 10996fe010SChris Lattner // This file defines the common interface used by the various execution engine 11996fe010SChris Lattner // subclasses. 12996fe010SChris Lattner // 13996fe010SChris Lattner //===----------------------------------------------------------------------===// 14996fe010SChris Lattner 156bf87df5SJeffrey Yasskin #include "llvm/ExecutionEngine/ExecutionEngine.h" 160a446fd5SBenjamin Kramer #include "llvm/ADT/STLExtras.h" 17868e3f09SDaniel Dunbar #include "llvm/ADT/SmallString.h" 18390d78b3SChris Lattner #include "llvm/ADT/Statistic.h" 19ed0881b2SChandler Carruth #include "llvm/ExecutionEngine/GenericValue.h" 20b5c7b1ffSLang Hames #include "llvm/ExecutionEngine/JITEventListener.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" 356d8dd189SChris Lattner #include "llvm/Support/MutexGuard.h" 36ed0881b2SChandler Carruth #include "llvm/Support/TargetRegistry.h" 37ccb29cd2STorok Edwin #include "llvm/Support/raw_ostream.h" 388418fdcdSDylan Noblesmith #include "llvm/Target/TargetMachine.h" 39579f0713SAnton Korobeynikov #include <cmath> 40579f0713SAnton Korobeynikov #include <cstring> 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, 51633fe146SLang Hames std::shared_ptr<RuntimeDyld::SymbolResolver> 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, 56633fe146SLang Hames std::shared_ptr<RuntimeDyld::SymbolResolver> 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 642a8a2795SRafael Espindola ExecutionEngine::ExecutionEngine(std::unique_ptr<Module> M) 653dac3f7fSLang Hames : LazyFunctionCreator(nullptr) { 664567db45SJeffrey Yasskin CompilingLazily = false; 67cdc0060eSEvan Cheng GVCompilationDisabled = false; 6884a9055eSEvan Cheng SymbolSearchingDisabled = false; 69bc876017SLang Hames 70bc876017SLang Hames // IR module verification is enabled by default in debug builds, and disabled 71bc876017SLang Hames // by default in release builds. 72bc876017SLang Hames #ifndef NDEBUG 73bc876017SLang Hames VerifyModules = true; 74bc876017SLang Hames #else 75bc876017SLang Hames VerifyModules = false; 76bc876017SLang Hames #endif 77bc876017SLang Hames 78091217beSJeffrey Yasskin assert(M && "Module is null?"); 792a8a2795SRafael Espindola Modules.push_back(std::move(M)); 80260b0c88SMisha Brukman } 81260b0c88SMisha Brukman 8292f8b30dSBrian Gaeke ExecutionEngine::~ExecutionEngine() { 83603682adSReid Spencer clearAllGlobalMappings(); 8492f8b30dSBrian Gaeke } 8592f8b30dSBrian Gaeke 86a4044332SJeffrey Yasskin namespace { 87868e3f09SDaniel Dunbar /// \brief Helper class which uses a value handler to automatically deletes the 88868e3f09SDaniel Dunbar /// memory block when the GlobalVariable is destroyed. 89a4044332SJeffrey Yasskin class GVMemoryBlock : public CallbackVH { 90a4044332SJeffrey Yasskin GVMemoryBlock(const GlobalVariable *GV) 91a4044332SJeffrey Yasskin : CallbackVH(const_cast<GlobalVariable*>(GV)) {} 92a4044332SJeffrey Yasskin 93a4044332SJeffrey Yasskin public: 94868e3f09SDaniel Dunbar /// \brief Returns the address the GlobalVariable should be written into. The 95868e3f09SDaniel Dunbar /// GVMemoryBlock object prefixes that. 96cdfe20b9SMicah Villmow static char *Create(const GlobalVariable *GV, const DataLayout& TD) { 97229907cdSChris Lattner Type *ElTy = GV->getType()->getElementType(); 98a4044332SJeffrey Yasskin size_t GVSize = (size_t)TD.getTypeAllocSize(ElTy); 99a4044332SJeffrey Yasskin void *RawMemory = ::operator new( 100f3cadce8SDavid Majnemer RoundUpToAlignment(sizeof(GVMemoryBlock), 101a4044332SJeffrey Yasskin TD.getPreferredAlignment(GV)) 102a4044332SJeffrey Yasskin + GVSize); 103a4044332SJeffrey Yasskin new(RawMemory) GVMemoryBlock(GV); 104a4044332SJeffrey Yasskin return static_cast<char*>(RawMemory) + sizeof(GVMemoryBlock); 105a4044332SJeffrey Yasskin } 106a4044332SJeffrey Yasskin 107b51ff603SCraig Topper void deleted() override { 108a4044332SJeffrey Yasskin // We allocated with operator new and with some extra memory hanging off the 109a4044332SJeffrey Yasskin // end, so don't just delete this. I'm not sure if this is actually 110a4044332SJeffrey Yasskin // required. 111a4044332SJeffrey Yasskin this->~GVMemoryBlock(); 112a4044332SJeffrey Yasskin ::operator delete(this); 113a4044332SJeffrey Yasskin } 114a4044332SJeffrey Yasskin }; 115a4044332SJeffrey Yasskin } // anonymous namespace 116a4044332SJeffrey Yasskin 117a4044332SJeffrey Yasskin char *ExecutionEngine::getMemoryForGV(const GlobalVariable *GV) { 118cdfe20b9SMicah Villmow return GVMemoryBlock::Create(GV, *getDataLayout()); 1195457ce9aSNicolas Geoffray } 1205457ce9aSNicolas Geoffray 12135907d8eSDavid Blaikie void ExecutionEngine::addObjectFile(std::unique_ptr<object::ObjectFile> O) { 12235907d8eSDavid Blaikie llvm_unreachable("ExecutionEngine subclass doesn't implement addObjectFile."); 12335907d8eSDavid Blaikie } 12435907d8eSDavid Blaikie 1257271c194SRafael Espindola void 1267271c194SRafael Espindola ExecutionEngine::addObjectFile(object::OwningBinary<object::ObjectFile> O) { 1277271c194SRafael Espindola llvm_unreachable("ExecutionEngine subclass doesn't implement addObjectFile."); 1287271c194SRafael Espindola } 1297271c194SRafael Espindola 13048af1c2aSRafael Espindola void ExecutionEngine::addArchive(object::OwningBinary<object::Archive> A) { 131acfd6289SRafael Espindola llvm_unreachable("ExecutionEngine subclass doesn't implement addArchive."); 132acfd6289SRafael Espindola } 133acfd6289SRafael Espindola 134091217beSJeffrey Yasskin bool ExecutionEngine::removeModule(Module *M) { 1352a8a2795SRafael Espindola for (auto I = Modules.begin(), E = Modules.end(); I != E; ++I) { 1362a8a2795SRafael Espindola Module *Found = I->get(); 137091217beSJeffrey Yasskin if (Found == M) { 1382a8a2795SRafael Espindola I->release(); 139324fe890SDevang Patel Modules.erase(I); 140091217beSJeffrey Yasskin clearGlobalMappingsFromModule(M); 141091217beSJeffrey Yasskin return true; 142324fe890SDevang Patel } 143324fe890SDevang Patel } 144091217beSJeffrey Yasskin return false; 145617001d8SNate Begeman } 146617001d8SNate Begeman 1470621caefSChris Lattner Function *ExecutionEngine::FindFunctionNamed(const char *FnName) { 1480621caefSChris Lattner for (unsigned i = 0, e = Modules.size(); i != e; ++i) { 1495f92a08fSKeno Fischer Function *F = Modules[i]->getFunction(FnName); 1505f92a08fSKeno Fischer if (F && !F->isDeclaration()) 1510621caefSChris Lattner return F; 1520621caefSChris Lattner } 1532617dcceSCraig Topper return nullptr; 1540621caefSChris Lattner } 1550621caefSChris Lattner 15673378eb1SKeno Fischer GlobalVariable *ExecutionEngine::FindGlobalVariableNamed(const char *Name, bool AllowInternal) { 15773378eb1SKeno Fischer for (unsigned i = 0, e = Modules.size(); i != e; ++i) { 15873378eb1SKeno Fischer GlobalVariable *GV = Modules[i]->getGlobalVariable(Name,AllowInternal); 15973378eb1SKeno Fischer if (GV && !GV->isDeclaration()) 16073378eb1SKeno Fischer return GV; 16173378eb1SKeno Fischer } 16273378eb1SKeno Fischer return nullptr; 16373378eb1SKeno Fischer } 1640621caefSChris Lattner 1653dac3f7fSLang Hames uint64_t ExecutionEngineState::RemoveMapping(StringRef Name) { 1663dac3f7fSLang Hames GlobalAddressMapTy::iterator I = GlobalAddressMap.find(Name); 1673dac3f7fSLang Hames uint64_t OldVal; 168868e3f09SDaniel Dunbar 169868e3f09SDaniel Dunbar // FIXME: This is silly, we shouldn't end up with a mapping -> 0 in the 170868e3f09SDaniel Dunbar // GlobalAddressMap. 171307c053fSJeffrey Yasskin if (I == GlobalAddressMap.end()) 1723dac3f7fSLang Hames OldVal = 0; 173307c053fSJeffrey Yasskin else { 1743dac3f7fSLang Hames GlobalAddressReverseMap.erase(I->second); 175307c053fSJeffrey Yasskin OldVal = I->second; 176307c053fSJeffrey Yasskin GlobalAddressMap.erase(I); 177307c053fSJeffrey Yasskin } 178307c053fSJeffrey Yasskin 179307c053fSJeffrey Yasskin return OldVal; 180307c053fSJeffrey Yasskin } 181307c053fSJeffrey Yasskin 1823dac3f7fSLang Hames std::string ExecutionEngine::getMangledName(const GlobalValue *GV) { 1833dac3f7fSLang Hames MutexGuard locked(lock); 184*c233f74eSRafael Espindola Mangler Mang; 1853dac3f7fSLang Hames SmallString<128> FullName; 186*c233f74eSRafael Espindola Mang.getNameWithPrefix(FullName, GV, false); 1873dac3f7fSLang Hames return FullName.str(); 1883dac3f7fSLang Hames } 1893dac3f7fSLang Hames 1906d8dd189SChris Lattner void ExecutionEngine::addGlobalMapping(const GlobalValue *GV, void *Addr) { 191c04b892fSZachary Turner MutexGuard locked(lock); 1923dac3f7fSLang Hames addGlobalMapping(getMangledName(GV), (uint64_t) Addr); 1933dac3f7fSLang Hames } 1946d8dd189SChris Lattner 1953dac3f7fSLang Hames void ExecutionEngine::addGlobalMapping(StringRef Name, uint64_t Addr) { 1963dac3f7fSLang Hames MutexGuard locked(lock); 1973dac3f7fSLang Hames 1983dac3f7fSLang Hames assert(!Name.empty() && "Empty GlobalMapping symbol name!"); 1993dac3f7fSLang Hames 2003dac3f7fSLang Hames DEBUG(dbgs() << "JIT: Map \'" << Name << "\' to [" << Addr << "]\n";); 2013dac3f7fSLang Hames uint64_t &CurVal = EEState.getGlobalAddressMap()[Name]; 2022617dcceSCraig Topper assert((!CurVal || !Addr) && "GlobalMapping already established!"); 2036d8dd189SChris Lattner CurVal = Addr; 2046d8dd189SChris Lattner 205868e3f09SDaniel Dunbar // If we are using the reverse mapping, add it too. 2062f825df6SZachary Turner if (!EEState.getGlobalAddressReverseMap().empty()) { 2073dac3f7fSLang Hames std::string &V = EEState.getGlobalAddressReverseMap()[CurVal]; 2083dac3f7fSLang Hames assert((!V.empty() || !Name.empty()) && 2093dac3f7fSLang Hames "GlobalMapping already established!"); 2103dac3f7fSLang Hames V = Name; 2116d8dd189SChris Lattner } 2126d8dd189SChris Lattner } 2136d8dd189SChris Lattner 2146d8dd189SChris Lattner void ExecutionEngine::clearAllGlobalMappings() { 215c04b892fSZachary Turner MutexGuard locked(lock); 2166d8dd189SChris Lattner 2172f825df6SZachary Turner EEState.getGlobalAddressMap().clear(); 2182f825df6SZachary Turner EEState.getGlobalAddressReverseMap().clear(); 2196d8dd189SChris Lattner } 2206d8dd189SChris Lattner 2218f83fc4dSNate Begeman void ExecutionEngine::clearGlobalMappingsFromModule(Module *M) { 222c04b892fSZachary Turner MutexGuard locked(lock); 2238f83fc4dSNate Begeman 224868e3f09SDaniel Dunbar for (Module::iterator FI = M->begin(), FE = M->end(); FI != FE; ++FI) 2253dac3f7fSLang Hames EEState.RemoveMapping(getMangledName(FI)); 2268f83fc4dSNate Begeman for (Module::global_iterator GI = M->global_begin(), GE = M->global_end(); 227868e3f09SDaniel Dunbar GI != GE; ++GI) 2283dac3f7fSLang Hames EEState.RemoveMapping(getMangledName(GI)); 2298f83fc4dSNate Begeman } 2308f83fc4dSNate Begeman 2313dac3f7fSLang Hames uint64_t ExecutionEngine::updateGlobalMapping(const GlobalValue *GV, 2323dac3f7fSLang Hames void *Addr) { 2333dac3f7fSLang Hames MutexGuard locked(lock); 2343dac3f7fSLang Hames return updateGlobalMapping(getMangledName(GV), (uint64_t) Addr); 2353dac3f7fSLang Hames } 2363dac3f7fSLang Hames 2373dac3f7fSLang Hames uint64_t ExecutionEngine::updateGlobalMapping(StringRef Name, uint64_t Addr) { 238c04b892fSZachary Turner MutexGuard locked(lock); 2396d8dd189SChris Lattner 240d0fc8f80SJeffrey Yasskin ExecutionEngineState::GlobalAddressMapTy &Map = 2412f825df6SZachary Turner EEState.getGlobalAddressMap(); 242ee181730SChris Lattner 2436d8dd189SChris Lattner // Deleting from the mapping? 2442617dcceSCraig Topper if (!Addr) 2453dac3f7fSLang Hames return EEState.RemoveMapping(Name); 246ee181730SChris Lattner 2473dac3f7fSLang Hames uint64_t &CurVal = Map[Name]; 2483dac3f7fSLang Hames uint64_t OldVal = CurVal; 249ee181730SChris Lattner 2502f825df6SZachary Turner if (CurVal && !EEState.getGlobalAddressReverseMap().empty()) 2512f825df6SZachary Turner EEState.getGlobalAddressReverseMap().erase(CurVal); 2526d8dd189SChris Lattner CurVal = Addr; 2536d8dd189SChris Lattner 254868e3f09SDaniel Dunbar // If we are using the reverse mapping, add it too. 2552f825df6SZachary Turner if (!EEState.getGlobalAddressReverseMap().empty()) { 2563dac3f7fSLang Hames std::string &V = EEState.getGlobalAddressReverseMap()[CurVal]; 2573dac3f7fSLang Hames assert((!V.empty() || !Name.empty()) && 2583dac3f7fSLang Hames "GlobalMapping already established!"); 2593dac3f7fSLang Hames V = Name; 2606d8dd189SChris Lattner } 261ee181730SChris Lattner return OldVal; 2626d8dd189SChris Lattner } 2636d8dd189SChris Lattner 2643dac3f7fSLang Hames uint64_t ExecutionEngine::getAddressToGlobalIfAvailable(StringRef S) { 2653dac3f7fSLang Hames MutexGuard locked(lock); 2663dac3f7fSLang Hames uint64_t Address = 0; 2673dac3f7fSLang Hames ExecutionEngineState::GlobalAddressMapTy::iterator I = 2683dac3f7fSLang Hames EEState.getGlobalAddressMap().find(S); 2693dac3f7fSLang Hames if (I != EEState.getGlobalAddressMap().end()) 2703dac3f7fSLang Hames Address = I->second; 2713dac3f7fSLang Hames return Address; 2723dac3f7fSLang Hames } 2733dac3f7fSLang Hames 2743dac3f7fSLang Hames 2753dac3f7fSLang Hames void *ExecutionEngine::getPointerToGlobalIfAvailable(StringRef S) { 2763dac3f7fSLang Hames MutexGuard locked(lock); 2773dac3f7fSLang Hames if (void* Address = (void *) getAddressToGlobalIfAvailable(S)) 2783dac3f7fSLang Hames return Address; 2793dac3f7fSLang Hames return nullptr; 2803dac3f7fSLang Hames } 2813dac3f7fSLang Hames 2826d8dd189SChris Lattner void *ExecutionEngine::getPointerToGlobalIfAvailable(const GlobalValue *GV) { 283c04b892fSZachary Turner MutexGuard locked(lock); 2843dac3f7fSLang Hames return getPointerToGlobalIfAvailable(getMangledName(GV)); 2856d8dd189SChris Lattner } 2866d8dd189SChris Lattner 287748e8579SChris Lattner const GlobalValue *ExecutionEngine::getGlobalValueAtAddress(void *Addr) { 288c04b892fSZachary Turner MutexGuard locked(lock); 28979876f52SReid Spencer 290748e8579SChris Lattner // If we haven't computed the reverse mapping yet, do so first. 2912f825df6SZachary Turner if (EEState.getGlobalAddressReverseMap().empty()) { 292d0fc8f80SJeffrey Yasskin for (ExecutionEngineState::GlobalAddressMapTy::iterator 2932f825df6SZachary Turner I = EEState.getGlobalAddressMap().begin(), 2943dac3f7fSLang Hames E = EEState.getGlobalAddressMap().end(); I != E; ++I) { 2953dac3f7fSLang Hames StringRef Name = I->first(); 2963dac3f7fSLang Hames uint64_t Addr = I->second; 2972f825df6SZachary Turner EEState.getGlobalAddressReverseMap().insert(std::make_pair( 2983dac3f7fSLang Hames Addr, Name)); 2993dac3f7fSLang Hames } 300748e8579SChris Lattner } 301748e8579SChris Lattner 3023dac3f7fSLang Hames std::map<uint64_t, std::string>::iterator I = 3033dac3f7fSLang Hames EEState.getGlobalAddressReverseMap().find((uint64_t) Addr); 3043dac3f7fSLang Hames 3053dac3f7fSLang Hames if (I != EEState.getGlobalAddressReverseMap().end()) { 3063dac3f7fSLang Hames StringRef Name = I->second; 3073dac3f7fSLang Hames for (unsigned i = 0, e = Modules.size(); i != e; ++i) 3083dac3f7fSLang Hames if (GlobalValue *GV = Modules[i]->getNamedValue(Name)) 3093dac3f7fSLang Hames return GV; 3103dac3f7fSLang Hames } 3113dac3f7fSLang Hames return nullptr; 312748e8579SChris Lattner } 3135a0d4829SChris Lattner 314bfd38abbSJeffrey Yasskin namespace { 315bfd38abbSJeffrey Yasskin class ArgvArray { 316c4a9942aSDylan Noblesmith std::unique_ptr<char[]> Array; 317c4a9942aSDylan Noblesmith std::vector<std::unique_ptr<char[]>> Values; 318bfd38abbSJeffrey Yasskin public: 319bfd38abbSJeffrey Yasskin /// Turn a vector of strings into a nice argv style array of pointers to null 320bfd38abbSJeffrey Yasskin /// terminated strings. 321bfd38abbSJeffrey Yasskin void *reset(LLVMContext &C, ExecutionEngine *EE, 322bfd38abbSJeffrey Yasskin const std::vector<std::string> &InputArgv); 323bfd38abbSJeffrey Yasskin }; 324bfd38abbSJeffrey Yasskin } // anonymous namespace 325bfd38abbSJeffrey Yasskin void *ArgvArray::reset(LLVMContext &C, ExecutionEngine *EE, 3265a0d4829SChris Lattner const std::vector<std::string> &InputArgv) { 327c4a9942aSDylan Noblesmith Values.clear(); // Free the old contents. 328c4a9942aSDylan Noblesmith Values.reserve(InputArgv.size()); 3295da3f051SChandler Carruth unsigned PtrSize = EE->getDataLayout()->getPointerSize(); 330c4a9942aSDylan Noblesmith Array = make_unique<char[]>((InputArgv.size()+1)*PtrSize); 3315a0d4829SChris Lattner 332c4a9942aSDylan Noblesmith DEBUG(dbgs() << "JIT: ARGV = " << (void*)Array.get() << "\n"); 333229907cdSChris Lattner Type *SBytePtr = Type::getInt8PtrTy(C); 3345a0d4829SChris Lattner 3355a0d4829SChris Lattner for (unsigned i = 0; i != InputArgv.size(); ++i) { 3365a0d4829SChris Lattner unsigned Size = InputArgv[i].size()+1; 3374b535d19SDylan Noblesmith auto Dest = make_unique<char[]>(Size); 3384b535d19SDylan Noblesmith DEBUG(dbgs() << "JIT: ARGV[" << i << "] = " << (void*)Dest.get() << "\n"); 3395a0d4829SChris Lattner 3404b535d19SDylan Noblesmith std::copy(InputArgv[i].begin(), InputArgv[i].end(), Dest.get()); 3415a0d4829SChris Lattner Dest[Size-1] = 0; 3425a0d4829SChris Lattner 343bfd38abbSJeffrey Yasskin // Endian safe: Array[i] = (PointerTy)Dest; 3444b535d19SDylan Noblesmith EE->StoreValueToMemory(PTOGV(Dest.get()), 3454b535d19SDylan Noblesmith (GenericValue*)(&Array[i*PtrSize]), SBytePtr); 3464b535d19SDylan Noblesmith Values.push_back(std::move(Dest)); 3475a0d4829SChris Lattner } 3485a0d4829SChris Lattner 3495a0d4829SChris Lattner // Null terminate it 3502617dcceSCraig Topper EE->StoreValueToMemory(PTOGV(nullptr), 351c4a9942aSDylan Noblesmith (GenericValue*)(&Array[InputArgv.size()*PtrSize]), 3525a0d4829SChris Lattner SBytePtr); 353c4a9942aSDylan Noblesmith return Array.get(); 3545a0d4829SChris Lattner } 3555a0d4829SChris Lattner 3562a8a2795SRafael Espindola void ExecutionEngine::runStaticConstructorsDestructors(Module &module, 35741fa2bd1SChris Lattner bool isDtors) { 358faae50b6SChris Lattner const char *Name = isDtors ? "llvm.global_dtors" : "llvm.global_ctors"; 3592a8a2795SRafael Espindola GlobalVariable *GV = module.getNamedGlobal(Name); 360fe36eaebSChris Lattner 361fe36eaebSChris Lattner // If this global has internal linkage, or if it has a use, then it must be 362fe36eaebSChris Lattner // an old-style (llvmgcc3) static ctor with __main linked in and in use. If 3630621caefSChris Lattner // this is the case, don't execute any of the global ctors, __main will do 3640621caefSChris Lattner // it. 3656de96a1bSRafael Espindola if (!GV || GV->isDeclaration() || GV->hasLocalLinkage()) return; 366faae50b6SChris Lattner 3670cbfcb2bSNick Lewycky // Should be an array of '{ i32, void ()* }' structs. The first value is 3680621caefSChris Lattner // the init priority, which we ignore. 36900245f42SChris Lattner ConstantArray *InitList = dyn_cast<ConstantArray>(GV->getInitializer()); 3702617dcceSCraig Topper if (!InitList) 3710f857898SNick Lewycky return; 372868e3f09SDaniel Dunbar for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i) { 37300245f42SChris Lattner ConstantStruct *CS = dyn_cast<ConstantStruct>(InitList->getOperand(i)); 3742617dcceSCraig Topper if (!CS) continue; 375faae50b6SChris Lattner 376faae50b6SChris Lattner Constant *FP = CS->getOperand(1); 377faae50b6SChris Lattner if (FP->isNullValue()) 3780f857898SNick Lewycky continue; // Found a sentinal value, ignore. 379faae50b6SChris Lattner 380868e3f09SDaniel Dunbar // Strip off constant expression casts. 381faae50b6SChris Lattner if (ConstantExpr *CE = dyn_cast<ConstantExpr>(FP)) 3826c38f0bbSReid Spencer if (CE->isCast()) 383faae50b6SChris Lattner FP = CE->getOperand(0); 384868e3f09SDaniel Dunbar 385faae50b6SChris Lattner // Execute the ctor/dtor function! 386868e3f09SDaniel Dunbar if (Function *F = dyn_cast<Function>(FP)) 387bd7b1c89SBenjamin Kramer runFunction(F, None); 388868e3f09SDaniel Dunbar 389868e3f09SDaniel Dunbar // FIXME: It is marginally lame that we just do nothing here if we see an 390868e3f09SDaniel Dunbar // entry we don't recognize. It might not be unreasonable for the verifier 391868e3f09SDaniel Dunbar // to not even allow this and just assert here. 392faae50b6SChris Lattner } 393faae50b6SChris Lattner } 3941a9a0b7bSEvan Cheng 3951a9a0b7bSEvan Cheng void ExecutionEngine::runStaticConstructorsDestructors(bool isDtors) { 3961a9a0b7bSEvan Cheng // Execute global ctors/dtors for each module in the program. 3972a8a2795SRafael Espindola for (std::unique_ptr<Module> &M : Modules) 3982a8a2795SRafael Espindola runStaticConstructorsDestructors(*M, isDtors); 3990621caefSChris Lattner } 400faae50b6SChris Lattner 401cf3e3017SDan Gohman #ifndef NDEBUG 4021202d1b1SDuncan Sands /// isTargetNullPtr - Return whether the target pointer stored at Loc is null. 4031202d1b1SDuncan Sands static bool isTargetNullPtr(ExecutionEngine *EE, void *Loc) { 4045da3f051SChandler Carruth unsigned PtrSize = EE->getDataLayout()->getPointerSize(); 4051202d1b1SDuncan Sands for (unsigned i = 0; i < PtrSize; ++i) 4061202d1b1SDuncan Sands if (*(i + (uint8_t*)Loc)) 4071202d1b1SDuncan Sands return false; 4081202d1b1SDuncan Sands return true; 4091202d1b1SDuncan Sands } 410cf3e3017SDan Gohman #endif 4111202d1b1SDuncan Sands 4125a0d4829SChris Lattner int ExecutionEngine::runFunctionAsMain(Function *Fn, 4135a0d4829SChris Lattner const std::vector<std::string> &argv, 4145a0d4829SChris Lattner const char * const * envp) { 4155a0d4829SChris Lattner std::vector<GenericValue> GVArgs; 4165a0d4829SChris Lattner GenericValue GVArgc; 41787aa65f4SReid Spencer GVArgc.IntVal = APInt(32, argv.size()); 4188c32c111SAnton Korobeynikov 4198c32c111SAnton Korobeynikov // Check main() type 420b1cad0b3SChris Lattner unsigned NumArgs = Fn->getFunctionType()->getNumParams(); 421229907cdSChris Lattner FunctionType *FTy = Fn->getFunctionType(); 422229907cdSChris Lattner Type* PPInt8Ty = Type::getInt8PtrTy(Fn->getContext())->getPointerTo(); 423868e3f09SDaniel Dunbar 424868e3f09SDaniel Dunbar // Check the argument types. 425868e3f09SDaniel Dunbar if (NumArgs > 3) 4262104b8d3SChris Lattner report_fatal_error("Invalid number of arguments of main() supplied"); 427868e3f09SDaniel Dunbar if (NumArgs >= 3 && FTy->getParamType(2) != PPInt8Ty) 428868e3f09SDaniel Dunbar report_fatal_error("Invalid type for third argument of main() supplied"); 429868e3f09SDaniel Dunbar if (NumArgs >= 2 && FTy->getParamType(1) != PPInt8Ty) 430868e3f09SDaniel Dunbar report_fatal_error("Invalid type for second argument of main() supplied"); 431868e3f09SDaniel Dunbar if (NumArgs >= 1 && !FTy->getParamType(0)->isIntegerTy(32)) 432868e3f09SDaniel Dunbar report_fatal_error("Invalid type for first argument of main() supplied"); 433868e3f09SDaniel Dunbar if (!FTy->getReturnType()->isIntegerTy() && 434868e3f09SDaniel Dunbar !FTy->getReturnType()->isVoidTy()) 435868e3f09SDaniel Dunbar report_fatal_error("Invalid return type of main() supplied"); 4368c32c111SAnton Korobeynikov 437bfd38abbSJeffrey Yasskin ArgvArray CArgv; 438bfd38abbSJeffrey Yasskin ArgvArray CEnv; 439b1cad0b3SChris Lattner if (NumArgs) { 4405a0d4829SChris Lattner GVArgs.push_back(GVArgc); // Arg #0 = argc. 441b1cad0b3SChris Lattner if (NumArgs > 1) { 44255f1c09eSOwen Anderson // Arg #1 = argv. 443bfd38abbSJeffrey Yasskin GVArgs.push_back(PTOGV(CArgv.reset(Fn->getContext(), this, argv))); 4441202d1b1SDuncan Sands assert(!isTargetNullPtr(this, GVTOP(GVArgs[1])) && 445b1cad0b3SChris Lattner "argv[0] was null after CreateArgv"); 446b1cad0b3SChris Lattner if (NumArgs > 2) { 4475a0d4829SChris Lattner std::vector<std::string> EnvVars; 4485a0d4829SChris Lattner for (unsigned i = 0; envp[i]; ++i) 449f5e2fc47SBenjamin Kramer EnvVars.emplace_back(envp[i]); 45055f1c09eSOwen Anderson // Arg #2 = envp. 451bfd38abbSJeffrey Yasskin GVArgs.push_back(PTOGV(CEnv.reset(Fn->getContext(), this, EnvVars))); 452b1cad0b3SChris Lattner } 453b1cad0b3SChris Lattner } 454b1cad0b3SChris Lattner } 455868e3f09SDaniel Dunbar 45687aa65f4SReid Spencer return runFunction(Fn, GVArgs).IntVal.getZExtValue(); 4575a0d4829SChris Lattner } 4585a0d4829SChris Lattner 459298a3a05SBenjamin Kramer EngineBuilder::EngineBuilder() : EngineBuilder(nullptr) {} 46093de2a12SLang Hames 4614a5697e6SLang Hames EngineBuilder::EngineBuilder(std::unique_ptr<Module> M) 462298a3a05SBenjamin Kramer : M(std::move(M)), WhichEngine(EngineKind::Either), ErrorStr(nullptr), 463633fe146SLang Hames OptLevel(CodeGenOpt::Default), MemMgr(nullptr), Resolver(nullptr), 464633fe146SLang Hames RelocModel(Reloc::Default), CMModel(CodeModel::JITDefault), 465633fe146SLang Hames UseOrcMCJITReplacement(false) { 466322db9eaSAlp Toker // IR module verification is enabled by default in debug builds, and disabled 467322db9eaSAlp Toker // by default in release builds. 468322db9eaSAlp Toker #ifndef NDEBUG 469322db9eaSAlp Toker VerifyModules = true; 470322db9eaSAlp Toker #else 471322db9eaSAlp Toker VerifyModules = false; 472322db9eaSAlp Toker #endif 473322db9eaSAlp Toker } 474322db9eaSAlp Toker 475298a3a05SBenjamin Kramer EngineBuilder::~EngineBuilder() = default; 476298a3a05SBenjamin Kramer 477298a3a05SBenjamin Kramer EngineBuilder &EngineBuilder::setMCJITMemoryManager( 478298a3a05SBenjamin Kramer std::unique_ptr<RTDyldMemoryManager> mcjmm) { 479633fe146SLang Hames auto SharedMM = std::shared_ptr<RTDyldMemoryManager>(std::move(mcjmm)); 480633fe146SLang Hames MemMgr = SharedMM; 481633fe146SLang Hames Resolver = SharedMM; 482633fe146SLang Hames return *this; 483633fe146SLang Hames } 484633fe146SLang Hames 485633fe146SLang Hames EngineBuilder& 486633fe146SLang Hames EngineBuilder::setMemoryManager(std::unique_ptr<MCJITMemoryManager> MM) { 487633fe146SLang Hames MemMgr = std::shared_ptr<MCJITMemoryManager>(std::move(MM)); 488633fe146SLang Hames return *this; 489633fe146SLang Hames } 490633fe146SLang Hames 491633fe146SLang Hames EngineBuilder& 492633fe146SLang Hames EngineBuilder::setSymbolResolver(std::unique_ptr<RuntimeDyld::SymbolResolver> SR) { 493633fe146SLang Hames Resolver = std::shared_ptr<RuntimeDyld::SymbolResolver>(std::move(SR)); 494298a3a05SBenjamin Kramer return *this; 495298a3a05SBenjamin Kramer } 496298a3a05SBenjamin Kramer 497add6f1d2SOwen Anderson ExecutionEngine *EngineBuilder::create(TargetMachine *TM) { 49856440fd8SAhmed Charles std::unique_ptr<TargetMachine> TheTM(TM); // Take ownership. 49925a3d816SBenjamin Kramer 500a53414fdSNick Lewycky // Make sure we can resolve symbols in the program as well. The zero arg 501a53414fdSNick Lewycky // to the function tells DynamicLibrary to load the program, not a library. 5022617dcceSCraig Topper if (sys::DynamicLibrary::LoadLibraryPermanently(nullptr, ErrorStr)) 5032617dcceSCraig Topper return nullptr; 504a53414fdSNick Lewycky 505fc8a2d5aSReid Kleckner // If the user specified a memory manager but didn't specify which engine to 506fc8a2d5aSReid Kleckner // create, we assume they only want the JIT, and we fail if they only want 507fc8a2d5aSReid Kleckner // the interpreter. 508633fe146SLang Hames if (MemMgr) { 50941fa2bd1SChris Lattner if (WhichEngine & EngineKind::JIT) 510fc8a2d5aSReid Kleckner WhichEngine = EngineKind::JIT; 51141fa2bd1SChris Lattner else { 5128bcc6445SChris Lattner if (ErrorStr) 513fc8a2d5aSReid Kleckner *ErrorStr = "Cannot create an interpreter with a memory manager."; 5142617dcceSCraig Topper return nullptr; 515fc8a2d5aSReid Kleckner } 5164bd3bd5bSBrian Gaeke } 5174bd3bd5bSBrian Gaeke 518fc8a2d5aSReid Kleckner // Unless the interpreter was explicitly selected or the JIT is not linked, 519fc8a2d5aSReid Kleckner // try making a JIT. 52025a3d816SBenjamin Kramer if ((WhichEngine & EngineKind::JIT) && TheTM) { 5217f26246aSDylan Noblesmith Triple TT(M->getTargetTriple()); 5227f26246aSDylan Noblesmith if (!TM->getTarget().hasJIT()) { 5237f26246aSDylan Noblesmith errs() << "WARNING: This target JIT is not designed for the host" 5247f26246aSDylan Noblesmith << " you are running. If bad things happen, please choose" 5257f26246aSDylan Noblesmith << " a different -march switch.\n"; 5267f26246aSDylan Noblesmith } 5277f26246aSDylan Noblesmith 528bc876017SLang Hames ExecutionEngine *EE = nullptr; 52993de2a12SLang Hames if (ExecutionEngine::OrcMCJITReplacementCtor && UseOrcMCJITReplacement) { 530633fe146SLang Hames EE = ExecutionEngine::OrcMCJITReplacementCtor(ErrorStr, std::move(MemMgr), 531633fe146SLang Hames std::move(Resolver), 53293de2a12SLang Hames std::move(TheTM)); 53393de2a12SLang Hames EE->addModule(std::move(M)); 53493de2a12SLang Hames } else if (ExecutionEngine::MCJITCtor) 535633fe146SLang Hames EE = ExecutionEngine::MCJITCtor(std::move(M), ErrorStr, std::move(MemMgr), 536633fe146SLang Hames std::move(Resolver), std::move(TheTM)); 53793de2a12SLang Hames 538bc876017SLang Hames if (EE) { 539bc876017SLang Hames EE->setVerifyModules(VerifyModules); 540bc876017SLang Hames return EE; 54141fa2bd1SChris Lattner } 542fc8a2d5aSReid Kleckner } 543fc8a2d5aSReid Kleckner 544fc8a2d5aSReid Kleckner // If we can't make a JIT and we didn't request one specifically, try making 545fc8a2d5aSReid Kleckner // an interpreter instead. 54641fa2bd1SChris Lattner if (WhichEngine & EngineKind::Interpreter) { 54741fa2bd1SChris Lattner if (ExecutionEngine::InterpCtor) 5482a8a2795SRafael Espindola return ExecutionEngine::InterpCtor(std::move(M), ErrorStr); 5498bcc6445SChris Lattner if (ErrorStr) 55041fa2bd1SChris Lattner *ErrorStr = "Interpreter has not been linked in."; 5512617dcceSCraig Topper return nullptr; 552fc8a2d5aSReid Kleckner } 553fc8a2d5aSReid Kleckner 55479cc1e3aSEric Christopher if ((WhichEngine & EngineKind::JIT) && !ExecutionEngine::MCJITCtor) { 5558bcc6445SChris Lattner if (ErrorStr) 5568bcc6445SChris Lattner *ErrorStr = "JIT has not been linked in."; 5578bcc6445SChris Lattner } 558868e3f09SDaniel Dunbar 5592617dcceSCraig Topper return nullptr; 560b5163bb9SChris Lattner } 561b5163bb9SChris Lattner 562996fe010SChris Lattner void *ExecutionEngine::getPointerToGlobal(const GlobalValue *GV) { 5631678e859SBrian Gaeke if (Function *F = const_cast<Function*>(dyn_cast<Function>(GV))) 564996fe010SChris Lattner return getPointerToFunction(F); 565996fe010SChris Lattner 566c04b892fSZachary Turner MutexGuard locked(lock); 5673dac3f7fSLang Hames if (void* P = getPointerToGlobalIfAvailable(GV)) 568868e3f09SDaniel Dunbar return P; 56969e84901SJeff Cohen 57069e84901SJeff Cohen // Global variable might have been added since interpreter started. 57169e84901SJeff Cohen if (GlobalVariable *GVar = 57269e84901SJeff Cohen const_cast<GlobalVariable *>(dyn_cast<GlobalVariable>(GV))) 57369e84901SJeff Cohen EmitGlobalVariable(GVar); 57469e84901SJeff Cohen else 575fbcc663cSTorok Edwin llvm_unreachable("Global hasn't had an address allocated yet!"); 576868e3f09SDaniel Dunbar 5773dac3f7fSLang Hames return getPointerToGlobalIfAvailable(GV); 578996fe010SChris Lattner } 579996fe010SChris Lattner 580868e3f09SDaniel Dunbar /// \brief Converts a Constant* into a GenericValue, including handling of 581868e3f09SDaniel Dunbar /// ConstantExpr values. 582996fe010SChris Lattner GenericValue ExecutionEngine::getConstantValue(const Constant *C) { 5836c38f0bbSReid Spencer // If its undefined, return the garbage. 584bcbdbfb3SJay Foad if (isa<UndefValue>(C)) { 585bcbdbfb3SJay Foad GenericValue Result; 586bcbdbfb3SJay Foad switch (C->getType()->getTypeID()) { 587be79a7acSNadav Rotem default: 588be79a7acSNadav Rotem break; 589bcbdbfb3SJay Foad case Type::IntegerTyID: 590bcbdbfb3SJay Foad case Type::X86_FP80TyID: 591bcbdbfb3SJay Foad case Type::FP128TyID: 592bcbdbfb3SJay Foad case Type::PPC_FP128TyID: 593bcbdbfb3SJay Foad // Although the value is undefined, we still have to construct an APInt 594bcbdbfb3SJay Foad // with the correct bit width. 595bcbdbfb3SJay Foad Result.IntVal = APInt(C->getType()->getPrimitiveSizeInBits(), 0); 596bcbdbfb3SJay Foad break; 5978e97f016SElena Demikhovsky case Type::StructTyID: { 5988e97f016SElena Demikhovsky // if the whole struct is 'undef' just reserve memory for the value. 5998e97f016SElena Demikhovsky if(StructType *STy = dyn_cast<StructType>(C->getType())) { 6008e97f016SElena Demikhovsky unsigned int elemNum = STy->getNumElements(); 6018e97f016SElena Demikhovsky Result.AggregateVal.resize(elemNum); 6028e97f016SElena Demikhovsky for (unsigned int i = 0; i < elemNum; ++i) { 6038e97f016SElena Demikhovsky Type *ElemTy = STy->getElementType(i); 6048e97f016SElena Demikhovsky if (ElemTy->isIntegerTy()) 6058e97f016SElena Demikhovsky Result.AggregateVal[i].IntVal = 6068e97f016SElena Demikhovsky APInt(ElemTy->getPrimitiveSizeInBits(), 0); 6078e97f016SElena Demikhovsky else if (ElemTy->isAggregateType()) { 6088e97f016SElena Demikhovsky const Constant *ElemUndef = UndefValue::get(ElemTy); 6098e97f016SElena Demikhovsky Result.AggregateVal[i] = getConstantValue(ElemUndef); 6108e97f016SElena Demikhovsky } 6118e97f016SElena Demikhovsky } 6128e97f016SElena Demikhovsky } 6138e97f016SElena Demikhovsky } 6148e97f016SElena Demikhovsky break; 615be79a7acSNadav Rotem case Type::VectorTyID: 616be79a7acSNadav Rotem // if the whole vector is 'undef' just reserve memory for the value. 617be79a7acSNadav Rotem const VectorType* VTy = dyn_cast<VectorType>(C->getType()); 618be79a7acSNadav Rotem const Type *ElemTy = VTy->getElementType(); 619be79a7acSNadav Rotem unsigned int elemNum = VTy->getNumElements(); 620be79a7acSNadav Rotem Result.AggregateVal.resize(elemNum); 621be79a7acSNadav Rotem if (ElemTy->isIntegerTy()) 622be79a7acSNadav Rotem for (unsigned int i = 0; i < elemNum; ++i) 623be79a7acSNadav Rotem Result.AggregateVal[i].IntVal = 624be79a7acSNadav Rotem APInt(ElemTy->getPrimitiveSizeInBits(), 0); 625bcbdbfb3SJay Foad break; 626bcbdbfb3SJay Foad } 627bcbdbfb3SJay Foad return Result; 628bcbdbfb3SJay Foad } 6299de0d14dSChris Lattner 630868e3f09SDaniel Dunbar // Otherwise, if the value is a ConstantExpr... 6316c38f0bbSReid Spencer if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) { 6324fd528f2SReid Spencer Constant *Op0 = CE->getOperand(0); 6339de0d14dSChris Lattner switch (CE->getOpcode()) { 6349de0d14dSChris Lattner case Instruction::GetElementPtr: { 6356c38f0bbSReid Spencer // Compute the index 6364fd528f2SReid Spencer GenericValue Result = getConstantValue(Op0); 6377c68bebbSRafael Espindola APInt Offset(DL->getPointerSizeInBits(), 0); 6387c68bebbSRafael Espindola cast<GEPOperator>(CE)->accumulateConstantOffset(*DL, Offset); 6399de0d14dSChris Lattner 64087aa65f4SReid Spencer char* tmp = (char*) Result.PointerVal; 641b6ad9822SNuno Lopes Result = PTOGV(tmp + Offset.getSExtValue()); 6429de0d14dSChris Lattner return Result; 6439de0d14dSChris Lattner } 6444fd528f2SReid Spencer case Instruction::Trunc: { 6454fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 6464fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); 6474fd528f2SReid Spencer GV.IntVal = GV.IntVal.trunc(BitWidth); 6484fd528f2SReid Spencer return GV; 6494fd528f2SReid Spencer } 6504fd528f2SReid Spencer case Instruction::ZExt: { 6514fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 6524fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); 6534fd528f2SReid Spencer GV.IntVal = GV.IntVal.zext(BitWidth); 6544fd528f2SReid Spencer return GV; 6554fd528f2SReid Spencer } 6564fd528f2SReid Spencer case Instruction::SExt: { 6574fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 6584fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); 6594fd528f2SReid Spencer GV.IntVal = GV.IntVal.sext(BitWidth); 6604fd528f2SReid Spencer return GV; 6614fd528f2SReid Spencer } 6624fd528f2SReid Spencer case Instruction::FPTrunc: { 663a1336cf5SDale Johannesen // FIXME long double 6644fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 6654fd528f2SReid Spencer GV.FloatVal = float(GV.DoubleVal); 6664fd528f2SReid Spencer return GV; 6674fd528f2SReid Spencer } 6684fd528f2SReid Spencer case Instruction::FPExt:{ 669a1336cf5SDale Johannesen // FIXME long double 6704fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 6714fd528f2SReid Spencer GV.DoubleVal = double(GV.FloatVal); 6724fd528f2SReid Spencer return GV; 6734fd528f2SReid Spencer } 6744fd528f2SReid Spencer case Instruction::UIToFP: { 6754fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 676fdd87907SChris Lattner if (CE->getType()->isFloatTy()) 6774fd528f2SReid Spencer GV.FloatVal = float(GV.IntVal.roundToDouble()); 678fdd87907SChris Lattner else if (CE->getType()->isDoubleTy()) 6794fd528f2SReid Spencer GV.DoubleVal = GV.IntVal.roundToDouble(); 680fdd87907SChris Lattner else if (CE->getType()->isX86_FP80Ty()) { 68131920b0aSBenjamin Kramer APFloat apf = APFloat::getZero(APFloat::x87DoubleExtended); 682ca24fd90SDan Gohman (void)apf.convertFromAPInt(GV.IntVal, 683ca24fd90SDan Gohman false, 6849150652bSDale Johannesen APFloat::rmNearestTiesToEven); 68554306fe4SDale Johannesen GV.IntVal = apf.bitcastToAPInt(); 686a1336cf5SDale Johannesen } 6874fd528f2SReid Spencer return GV; 6884fd528f2SReid Spencer } 6894fd528f2SReid Spencer case Instruction::SIToFP: { 6904fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 691fdd87907SChris Lattner if (CE->getType()->isFloatTy()) 6924fd528f2SReid Spencer GV.FloatVal = float(GV.IntVal.signedRoundToDouble()); 693fdd87907SChris Lattner else if (CE->getType()->isDoubleTy()) 6944fd528f2SReid Spencer GV.DoubleVal = GV.IntVal.signedRoundToDouble(); 695fdd87907SChris Lattner else if (CE->getType()->isX86_FP80Ty()) { 69631920b0aSBenjamin Kramer APFloat apf = APFloat::getZero(APFloat::x87DoubleExtended); 697ca24fd90SDan Gohman (void)apf.convertFromAPInt(GV.IntVal, 698ca24fd90SDan Gohman true, 6999150652bSDale Johannesen APFloat::rmNearestTiesToEven); 70054306fe4SDale Johannesen GV.IntVal = apf.bitcastToAPInt(); 701a1336cf5SDale Johannesen } 7024fd528f2SReid Spencer return GV; 7034fd528f2SReid Spencer } 7044fd528f2SReid Spencer case Instruction::FPToUI: // double->APInt conversion handles sign 7054fd528f2SReid Spencer case Instruction::FPToSI: { 7064fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 7074fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); 708fdd87907SChris Lattner if (Op0->getType()->isFloatTy()) 7094fd528f2SReid Spencer GV.IntVal = APIntOps::RoundFloatToAPInt(GV.FloatVal, BitWidth); 710fdd87907SChris Lattner else if (Op0->getType()->isDoubleTy()) 7114fd528f2SReid Spencer GV.IntVal = APIntOps::RoundDoubleToAPInt(GV.DoubleVal, BitWidth); 712fdd87907SChris Lattner else if (Op0->getType()->isX86_FP80Ty()) { 71329178a34STim Northover APFloat apf = APFloat(APFloat::x87DoubleExtended, GV.IntVal); 714a1336cf5SDale Johannesen uint64_t v; 7154f0bd68cSDale Johannesen bool ignored; 716a1336cf5SDale Johannesen (void)apf.convertToInteger(&v, BitWidth, 717a1336cf5SDale Johannesen CE->getOpcode()==Instruction::FPToSI, 7184f0bd68cSDale Johannesen APFloat::rmTowardZero, &ignored); 719a1336cf5SDale Johannesen GV.IntVal = v; // endian? 720a1336cf5SDale Johannesen } 7214fd528f2SReid Spencer return GV; 7224fd528f2SReid Spencer } 7236c38f0bbSReid Spencer case Instruction::PtrToInt: { 7244fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 7257c68bebbSRafael Espindola uint32_t PtrWidth = DL->getTypeSizeInBits(Op0->getType()); 726fc1f2cd3SEli Friedman assert(PtrWidth <= 64 && "Bad pointer width"); 7274fd528f2SReid Spencer GV.IntVal = APInt(PtrWidth, uintptr_t(GV.PointerVal)); 7287c68bebbSRafael Espindola uint32_t IntWidth = DL->getTypeSizeInBits(CE->getType()); 729fc1f2cd3SEli Friedman GV.IntVal = GV.IntVal.zextOrTrunc(IntWidth); 7304fd528f2SReid Spencer return GV; 7314fd528f2SReid Spencer } 7324fd528f2SReid Spencer case Instruction::IntToPtr: { 7334fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 7347c68bebbSRafael Espindola uint32_t PtrWidth = DL->getTypeSizeInBits(CE->getType()); 7354fd528f2SReid Spencer GV.IntVal = GV.IntVal.zextOrTrunc(PtrWidth); 7364fd528f2SReid Spencer assert(GV.IntVal.getBitWidth() <= 64 && "Bad pointer width"); 7374fd528f2SReid Spencer GV.PointerVal = PointerTy(uintptr_t(GV.IntVal.getZExtValue())); 7386c38f0bbSReid Spencer return GV; 7396c38f0bbSReid Spencer } 7406c38f0bbSReid Spencer case Instruction::BitCast: { 7414fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 742229907cdSChris Lattner Type* DestTy = CE->getType(); 7434fd528f2SReid Spencer switch (Op0->getType()->getTypeID()) { 744fbcc663cSTorok Edwin default: llvm_unreachable("Invalid bitcast operand"); 7454fd528f2SReid Spencer case Type::IntegerTyID: 7469dff9becSDuncan Sands assert(DestTy->isFloatingPointTy() && "invalid bitcast"); 747fdd87907SChris Lattner if (DestTy->isFloatTy()) 7484fd528f2SReid Spencer GV.FloatVal = GV.IntVal.bitsToFloat(); 749fdd87907SChris Lattner else if (DestTy->isDoubleTy()) 7504fd528f2SReid Spencer GV.DoubleVal = GV.IntVal.bitsToDouble(); 7516c38f0bbSReid Spencer break; 7524fd528f2SReid Spencer case Type::FloatTyID: 7539dff9becSDuncan Sands assert(DestTy->isIntegerTy(32) && "Invalid bitcast"); 7543447fb01SJay Foad GV.IntVal = APInt::floatToBits(GV.FloatVal); 7554fd528f2SReid Spencer break; 7564fd528f2SReid Spencer case Type::DoubleTyID: 7579dff9becSDuncan Sands assert(DestTy->isIntegerTy(64) && "Invalid bitcast"); 7583447fb01SJay Foad GV.IntVal = APInt::doubleToBits(GV.DoubleVal); 7594fd528f2SReid Spencer break; 7604fd528f2SReid Spencer case Type::PointerTyID: 76119d0b47bSDuncan Sands assert(DestTy->isPointerTy() && "Invalid bitcast"); 7624fd528f2SReid Spencer break; // getConstantValue(Op0) above already converted it 7636c38f0bbSReid Spencer } 7644fd528f2SReid Spencer return GV; 76568cbcc3eSChris Lattner } 76668cbcc3eSChris Lattner case Instruction::Add: 767a5b9645cSDan Gohman case Instruction::FAdd: 7684fd528f2SReid Spencer case Instruction::Sub: 769a5b9645cSDan Gohman case Instruction::FSub: 7704fd528f2SReid Spencer case Instruction::Mul: 771a5b9645cSDan Gohman case Instruction::FMul: 7724fd528f2SReid Spencer case Instruction::UDiv: 7734fd528f2SReid Spencer case Instruction::SDiv: 7744fd528f2SReid Spencer case Instruction::URem: 7754fd528f2SReid Spencer case Instruction::SRem: 7764fd528f2SReid Spencer case Instruction::And: 7774fd528f2SReid Spencer case Instruction::Or: 7784fd528f2SReid Spencer case Instruction::Xor: { 7794fd528f2SReid Spencer GenericValue LHS = getConstantValue(Op0); 7804fd528f2SReid Spencer GenericValue RHS = getConstantValue(CE->getOperand(1)); 7814fd528f2SReid Spencer GenericValue GV; 782c4e6bb5fSChris Lattner switch (CE->getOperand(0)->getType()->getTypeID()) { 783fbcc663cSTorok Edwin default: llvm_unreachable("Bad add type!"); 7847a9c62baSReid Spencer case Type::IntegerTyID: 7854fd528f2SReid Spencer switch (CE->getOpcode()) { 786fbcc663cSTorok Edwin default: llvm_unreachable("Invalid integer opcode"); 7874fd528f2SReid Spencer case Instruction::Add: GV.IntVal = LHS.IntVal + RHS.IntVal; break; 7884fd528f2SReid Spencer case Instruction::Sub: GV.IntVal = LHS.IntVal - RHS.IntVal; break; 7894fd528f2SReid Spencer case Instruction::Mul: GV.IntVal = LHS.IntVal * RHS.IntVal; break; 7904fd528f2SReid Spencer case Instruction::UDiv:GV.IntVal = LHS.IntVal.udiv(RHS.IntVal); break; 7914fd528f2SReid Spencer case Instruction::SDiv:GV.IntVal = LHS.IntVal.sdiv(RHS.IntVal); break; 7924fd528f2SReid Spencer case Instruction::URem:GV.IntVal = LHS.IntVal.urem(RHS.IntVal); break; 7934fd528f2SReid Spencer case Instruction::SRem:GV.IntVal = LHS.IntVal.srem(RHS.IntVal); break; 7944fd528f2SReid Spencer case Instruction::And: GV.IntVal = LHS.IntVal & RHS.IntVal; break; 7954fd528f2SReid Spencer case Instruction::Or: GV.IntVal = LHS.IntVal | RHS.IntVal; break; 7964fd528f2SReid Spencer case Instruction::Xor: GV.IntVal = LHS.IntVal ^ RHS.IntVal; break; 7974fd528f2SReid Spencer } 798c4e6bb5fSChris Lattner break; 799c4e6bb5fSChris Lattner case Type::FloatTyID: 8004fd528f2SReid Spencer switch (CE->getOpcode()) { 801fbcc663cSTorok Edwin default: llvm_unreachable("Invalid float opcode"); 802a5b9645cSDan Gohman case Instruction::FAdd: 8034fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal + RHS.FloatVal; break; 804a5b9645cSDan Gohman case Instruction::FSub: 8054fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal - RHS.FloatVal; break; 806a5b9645cSDan Gohman case Instruction::FMul: 8074fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal * RHS.FloatVal; break; 8084fd528f2SReid Spencer case Instruction::FDiv: 8094fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal / RHS.FloatVal; break; 8104fd528f2SReid Spencer case Instruction::FRem: 81193cd0f1cSChris Lattner GV.FloatVal = std::fmod(LHS.FloatVal,RHS.FloatVal); break; 8124fd528f2SReid Spencer } 813c4e6bb5fSChris Lattner break; 814c4e6bb5fSChris Lattner case Type::DoubleTyID: 8154fd528f2SReid Spencer switch (CE->getOpcode()) { 816fbcc663cSTorok Edwin default: llvm_unreachable("Invalid double opcode"); 817a5b9645cSDan Gohman case Instruction::FAdd: 8184fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal + RHS.DoubleVal; break; 819a5b9645cSDan Gohman case Instruction::FSub: 8204fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal - RHS.DoubleVal; break; 821a5b9645cSDan Gohman case Instruction::FMul: 8224fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal * RHS.DoubleVal; break; 8234fd528f2SReid Spencer case Instruction::FDiv: 8244fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal / RHS.DoubleVal; break; 8254fd528f2SReid Spencer case Instruction::FRem: 82693cd0f1cSChris Lattner GV.DoubleVal = std::fmod(LHS.DoubleVal,RHS.DoubleVal); break; 8274fd528f2SReid Spencer } 828c4e6bb5fSChris Lattner break; 829a1336cf5SDale Johannesen case Type::X86_FP80TyID: 830a1336cf5SDale Johannesen case Type::PPC_FP128TyID: 831a1336cf5SDale Johannesen case Type::FP128TyID: { 83229178a34STim Northover const fltSemantics &Sem = CE->getOperand(0)->getType()->getFltSemantics(); 83329178a34STim Northover APFloat apfLHS = APFloat(Sem, LHS.IntVal); 834a1336cf5SDale Johannesen switch (CE->getOpcode()) { 835e4f47434SDaniel Dunbar default: llvm_unreachable("Invalid long double opcode"); 836a5b9645cSDan Gohman case Instruction::FAdd: 83729178a34STim Northover apfLHS.add(APFloat(Sem, RHS.IntVal), APFloat::rmNearestTiesToEven); 83854306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt(); 839a1336cf5SDale Johannesen break; 840a5b9645cSDan Gohman case Instruction::FSub: 84129178a34STim Northover apfLHS.subtract(APFloat(Sem, RHS.IntVal), 84229178a34STim Northover APFloat::rmNearestTiesToEven); 84354306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt(); 844a1336cf5SDale Johannesen break; 845a5b9645cSDan Gohman case Instruction::FMul: 84629178a34STim Northover apfLHS.multiply(APFloat(Sem, RHS.IntVal), 84729178a34STim Northover APFloat::rmNearestTiesToEven); 84854306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt(); 849a1336cf5SDale Johannesen break; 850a1336cf5SDale Johannesen case Instruction::FDiv: 85129178a34STim Northover apfLHS.divide(APFloat(Sem, RHS.IntVal), 85229178a34STim Northover APFloat::rmNearestTiesToEven); 85354306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt(); 854a1336cf5SDale Johannesen break; 855a1336cf5SDale Johannesen case Instruction::FRem: 85629178a34STim Northover apfLHS.mod(APFloat(Sem, RHS.IntVal), 85729178a34STim Northover APFloat::rmNearestTiesToEven); 85854306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt(); 859a1336cf5SDale Johannesen break; 860a1336cf5SDale Johannesen } 861a1336cf5SDale Johannesen } 862a1336cf5SDale Johannesen break; 863c4e6bb5fSChris Lattner } 8644fd528f2SReid Spencer return GV; 8654fd528f2SReid Spencer } 8669de0d14dSChris Lattner default: 86768cbcc3eSChris Lattner break; 86868cbcc3eSChris Lattner } 869868e3f09SDaniel Dunbar 870868e3f09SDaniel Dunbar SmallString<256> Msg; 871868e3f09SDaniel Dunbar raw_svector_ostream OS(Msg); 872868e3f09SDaniel Dunbar OS << "ConstantExpr not handled: " << *CE; 873868e3f09SDaniel Dunbar report_fatal_error(OS.str()); 8749de0d14dSChris Lattner } 875996fe010SChris Lattner 876868e3f09SDaniel Dunbar // Otherwise, we have a simple constant. 8774fd528f2SReid Spencer GenericValue Result; 8786b727599SChris Lattner switch (C->getType()->getTypeID()) { 87987aa65f4SReid Spencer case Type::FloatTyID: 880bed9dc42SDale Johannesen Result.FloatVal = cast<ConstantFP>(C)->getValueAPF().convertToFloat(); 8817a9c62baSReid Spencer break; 88287aa65f4SReid Spencer case Type::DoubleTyID: 883bed9dc42SDale Johannesen Result.DoubleVal = cast<ConstantFP>(C)->getValueAPF().convertToDouble(); 88487aa65f4SReid Spencer break; 885a1336cf5SDale Johannesen case Type::X86_FP80TyID: 886a1336cf5SDale Johannesen case Type::FP128TyID: 887a1336cf5SDale Johannesen case Type::PPC_FP128TyID: 88854306fe4SDale Johannesen Result.IntVal = cast <ConstantFP>(C)->getValueAPF().bitcastToAPInt(); 889a1336cf5SDale Johannesen break; 89087aa65f4SReid Spencer case Type::IntegerTyID: 89187aa65f4SReid Spencer Result.IntVal = cast<ConstantInt>(C)->getValue(); 89287aa65f4SReid Spencer break; 893996fe010SChris Lattner case Type::PointerTyID: 8946a0fd73bSReid Spencer if (isa<ConstantPointerNull>(C)) 8952617dcceSCraig Topper Result.PointerVal = nullptr; 8966a0fd73bSReid Spencer else if (const Function *F = dyn_cast<Function>(C)) 8976a0fd73bSReid Spencer Result = PTOGV(getPointerToFunctionOrStub(const_cast<Function*>(F))); 8986a0fd73bSReid Spencer else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(C)) 8996a0fd73bSReid Spencer Result = PTOGV(getOrEmitGlobalVariable(const_cast<GlobalVariable*>(GV))); 900e6492f10SChris Lattner else 901fbcc663cSTorok Edwin llvm_unreachable("Unknown constant pointer type!"); 902996fe010SChris Lattner break; 903be79a7acSNadav Rotem case Type::VectorTyID: { 904be79a7acSNadav Rotem unsigned elemNum; 905be79a7acSNadav Rotem Type* ElemTy; 906be79a7acSNadav Rotem const ConstantDataVector *CDV = dyn_cast<ConstantDataVector>(C); 907be79a7acSNadav Rotem const ConstantVector *CV = dyn_cast<ConstantVector>(C); 908be79a7acSNadav Rotem const ConstantAggregateZero *CAZ = dyn_cast<ConstantAggregateZero>(C); 909be79a7acSNadav Rotem 910be79a7acSNadav Rotem if (CDV) { 911be79a7acSNadav Rotem elemNum = CDV->getNumElements(); 912be79a7acSNadav Rotem ElemTy = CDV->getElementType(); 913be79a7acSNadav Rotem } else if (CV || CAZ) { 914be79a7acSNadav Rotem VectorType* VTy = dyn_cast<VectorType>(C->getType()); 915be79a7acSNadav Rotem elemNum = VTy->getNumElements(); 916be79a7acSNadav Rotem ElemTy = VTy->getElementType(); 917be79a7acSNadav Rotem } else { 918be79a7acSNadav Rotem llvm_unreachable("Unknown constant vector type!"); 919be79a7acSNadav Rotem } 920be79a7acSNadav Rotem 921be79a7acSNadav Rotem Result.AggregateVal.resize(elemNum); 922be79a7acSNadav Rotem // Check if vector holds floats. 923be79a7acSNadav Rotem if(ElemTy->isFloatTy()) { 924be79a7acSNadav Rotem if (CAZ) { 925be79a7acSNadav Rotem GenericValue floatZero; 926be79a7acSNadav Rotem floatZero.FloatVal = 0.f; 927be79a7acSNadav Rotem std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(), 928be79a7acSNadav Rotem floatZero); 929be79a7acSNadav Rotem break; 930be79a7acSNadav Rotem } 931be79a7acSNadav Rotem if(CV) { 932be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i) 933be79a7acSNadav Rotem if (!isa<UndefValue>(CV->getOperand(i))) 934be79a7acSNadav Rotem Result.AggregateVal[i].FloatVal = cast<ConstantFP>( 935be79a7acSNadav Rotem CV->getOperand(i))->getValueAPF().convertToFloat(); 936be79a7acSNadav Rotem break; 937be79a7acSNadav Rotem } 938be79a7acSNadav Rotem if(CDV) 939be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i) 940be79a7acSNadav Rotem Result.AggregateVal[i].FloatVal = CDV->getElementAsFloat(i); 941be79a7acSNadav Rotem 942be79a7acSNadav Rotem break; 943be79a7acSNadav Rotem } 944be79a7acSNadav Rotem // Check if vector holds doubles. 945be79a7acSNadav Rotem if (ElemTy->isDoubleTy()) { 946be79a7acSNadav Rotem if (CAZ) { 947be79a7acSNadav Rotem GenericValue doubleZero; 948be79a7acSNadav Rotem doubleZero.DoubleVal = 0.0; 949be79a7acSNadav Rotem std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(), 950be79a7acSNadav Rotem doubleZero); 951be79a7acSNadav Rotem break; 952be79a7acSNadav Rotem } 953be79a7acSNadav Rotem if(CV) { 954be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i) 955be79a7acSNadav Rotem if (!isa<UndefValue>(CV->getOperand(i))) 956be79a7acSNadav Rotem Result.AggregateVal[i].DoubleVal = cast<ConstantFP>( 957be79a7acSNadav Rotem CV->getOperand(i))->getValueAPF().convertToDouble(); 958be79a7acSNadav Rotem break; 959be79a7acSNadav Rotem } 960be79a7acSNadav Rotem if(CDV) 961be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i) 962be79a7acSNadav Rotem Result.AggregateVal[i].DoubleVal = CDV->getElementAsDouble(i); 963be79a7acSNadav Rotem 964be79a7acSNadav Rotem break; 965be79a7acSNadav Rotem } 966be79a7acSNadav Rotem // Check if vector holds integers. 967be79a7acSNadav Rotem if (ElemTy->isIntegerTy()) { 968be79a7acSNadav Rotem if (CAZ) { 969be79a7acSNadav Rotem GenericValue intZero; 970be79a7acSNadav Rotem intZero.IntVal = APInt(ElemTy->getScalarSizeInBits(), 0ull); 971be79a7acSNadav Rotem std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(), 972be79a7acSNadav Rotem intZero); 973be79a7acSNadav Rotem break; 974be79a7acSNadav Rotem } 975be79a7acSNadav Rotem if(CV) { 976be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i) 977be79a7acSNadav Rotem if (!isa<UndefValue>(CV->getOperand(i))) 978be79a7acSNadav Rotem Result.AggregateVal[i].IntVal = cast<ConstantInt>( 979be79a7acSNadav Rotem CV->getOperand(i))->getValue(); 980be79a7acSNadav Rotem else { 981be79a7acSNadav Rotem Result.AggregateVal[i].IntVal = 982be79a7acSNadav Rotem APInt(CV->getOperand(i)->getType()->getPrimitiveSizeInBits(), 0); 983be79a7acSNadav Rotem } 984be79a7acSNadav Rotem break; 985be79a7acSNadav Rotem } 986be79a7acSNadav Rotem if(CDV) 987be79a7acSNadav Rotem for (unsigned i = 0; i < elemNum; ++i) 988be79a7acSNadav Rotem Result.AggregateVal[i].IntVal = APInt( 989be79a7acSNadav Rotem CDV->getElementType()->getPrimitiveSizeInBits(), 990be79a7acSNadav Rotem CDV->getElementAsInteger(i)); 991be79a7acSNadav Rotem 992be79a7acSNadav Rotem break; 993be79a7acSNadav Rotem } 994be79a7acSNadav Rotem llvm_unreachable("Unknown constant pointer type!"); 995be79a7acSNadav Rotem } 996be79a7acSNadav Rotem break; 997be79a7acSNadav Rotem 998996fe010SChris Lattner default: 999868e3f09SDaniel Dunbar SmallString<256> Msg; 1000868e3f09SDaniel Dunbar raw_svector_ostream OS(Msg); 1001868e3f09SDaniel Dunbar OS << "ERROR: Constant unimplemented for type: " << *C->getType(); 1002868e3f09SDaniel Dunbar report_fatal_error(OS.str()); 1003996fe010SChris Lattner } 1004868e3f09SDaniel Dunbar 1005996fe010SChris Lattner return Result; 1006996fe010SChris Lattner } 1007996fe010SChris Lattner 10081202d1b1SDuncan Sands /// StoreIntToMemory - Fills the StoreBytes bytes of memory starting from Dst 10091202d1b1SDuncan Sands /// with the integer held in IntVal. 10101202d1b1SDuncan Sands static void StoreIntToMemory(const APInt &IntVal, uint8_t *Dst, 10111202d1b1SDuncan Sands unsigned StoreBytes) { 10121202d1b1SDuncan Sands assert((IntVal.getBitWidth()+7)/8 >= StoreBytes && "Integer too small!"); 1013ad06cee2SRoman Divacky const uint8_t *Src = (const uint8_t *)IntVal.getRawData(); 10145c65cb46SDuncan Sands 101541cb64f4SRafael Espindola if (sys::IsLittleEndianHost) { 10161202d1b1SDuncan Sands // Little-endian host - the source is ordered from LSB to MSB. Order the 10171202d1b1SDuncan Sands // destination from LSB to MSB: Do a straight copy. 10185c65cb46SDuncan Sands memcpy(Dst, Src, StoreBytes); 1019868e3f09SDaniel Dunbar } else { 10205c65cb46SDuncan Sands // Big-endian host - the source is an array of 64 bit words ordered from 10211202d1b1SDuncan Sands // LSW to MSW. Each word is ordered from MSB to LSB. Order the destination 10221202d1b1SDuncan Sands // from MSB to LSB: Reverse the word order, but not the bytes in a word. 10235c65cb46SDuncan Sands while (StoreBytes > sizeof(uint64_t)) { 10245c65cb46SDuncan Sands StoreBytes -= sizeof(uint64_t); 10255c65cb46SDuncan Sands // May not be aligned so use memcpy. 10265c65cb46SDuncan Sands memcpy(Dst + StoreBytes, Src, sizeof(uint64_t)); 10275c65cb46SDuncan Sands Src += sizeof(uint64_t); 10285c65cb46SDuncan Sands } 10295c65cb46SDuncan Sands 10305c65cb46SDuncan Sands memcpy(Dst, Src + sizeof(uint64_t) - StoreBytes, StoreBytes); 1031815f8dd2SReid Spencer } 10327a9c62baSReid Spencer } 10331202d1b1SDuncan Sands 103409053e62SEvan Cheng void ExecutionEngine::StoreValueToMemory(const GenericValue &Val, 1035229907cdSChris Lattner GenericValue *Ptr, Type *Ty) { 1036cdfe20b9SMicah Villmow const unsigned StoreBytes = getDataLayout()->getTypeStoreSize(Ty); 10371202d1b1SDuncan Sands 10381202d1b1SDuncan Sands switch (Ty->getTypeID()) { 1039be79a7acSNadav Rotem default: 1040be79a7acSNadav Rotem dbgs() << "Cannot store value of type " << *Ty << "!\n"; 1041be79a7acSNadav Rotem break; 10421202d1b1SDuncan Sands case Type::IntegerTyID: 10431202d1b1SDuncan Sands StoreIntToMemory(Val.IntVal, (uint8_t*)Ptr, StoreBytes); 10441202d1b1SDuncan Sands break; 1045996fe010SChris Lattner case Type::FloatTyID: 104687aa65f4SReid Spencer *((float*)Ptr) = Val.FloatVal; 104787aa65f4SReid Spencer break; 104887aa65f4SReid Spencer case Type::DoubleTyID: 104987aa65f4SReid Spencer *((double*)Ptr) = Val.DoubleVal; 1050996fe010SChris Lattner break; 10514d7e4ee7SDale Johannesen case Type::X86_FP80TyID: 10524d7e4ee7SDale Johannesen memcpy(Ptr, Val.IntVal.getRawData(), 10); 1053a1336cf5SDale Johannesen break; 10547a9c62baSReid Spencer case Type::PointerTyID: 10551202d1b1SDuncan Sands // Ensure 64 bit target pointers are fully initialized on 32 bit hosts. 10561202d1b1SDuncan Sands if (StoreBytes != sizeof(PointerTy)) 105793da3c82SChandler Carruth memset(&(Ptr->PointerVal), 0, StoreBytes); 10581202d1b1SDuncan Sands 105987aa65f4SReid Spencer *((PointerTy*)Ptr) = Val.PointerVal; 1060996fe010SChris Lattner break; 1061be79a7acSNadav Rotem case Type::VectorTyID: 1062be79a7acSNadav Rotem for (unsigned i = 0; i < Val.AggregateVal.size(); ++i) { 1063be79a7acSNadav Rotem if (cast<VectorType>(Ty)->getElementType()->isDoubleTy()) 1064be79a7acSNadav Rotem *(((double*)Ptr)+i) = Val.AggregateVal[i].DoubleVal; 1065be79a7acSNadav Rotem if (cast<VectorType>(Ty)->getElementType()->isFloatTy()) 1066be79a7acSNadav Rotem *(((float*)Ptr)+i) = Val.AggregateVal[i].FloatVal; 1067be79a7acSNadav Rotem if (cast<VectorType>(Ty)->getElementType()->isIntegerTy()) { 1068be79a7acSNadav Rotem unsigned numOfBytes =(Val.AggregateVal[i].IntVal.getBitWidth()+7)/8; 1069be79a7acSNadav Rotem StoreIntToMemory(Val.AggregateVal[i].IntVal, 1070be79a7acSNadav Rotem (uint8_t*)Ptr + numOfBytes*i, numOfBytes); 1071be79a7acSNadav Rotem } 1072be79a7acSNadav Rotem } 1073be79a7acSNadav Rotem break; 1074996fe010SChris Lattner } 10751202d1b1SDuncan Sands 107641cb64f4SRafael Espindola if (sys::IsLittleEndianHost != getDataLayout()->isLittleEndian()) 10771202d1b1SDuncan Sands // Host and target are different endian - reverse the stored bytes. 10781202d1b1SDuncan Sands std::reverse((uint8_t*)Ptr, StoreBytes + (uint8_t*)Ptr); 1079996fe010SChris Lattner } 1080996fe010SChris Lattner 10811202d1b1SDuncan Sands /// LoadIntFromMemory - Loads the integer stored in the LoadBytes bytes starting 10821202d1b1SDuncan Sands /// from Src into IntVal, which is assumed to be wide enough and to hold zero. 10831202d1b1SDuncan Sands static void LoadIntFromMemory(APInt &IntVal, uint8_t *Src, unsigned LoadBytes) { 10841202d1b1SDuncan Sands assert((IntVal.getBitWidth()+7)/8 >= LoadBytes && "Integer too small!"); 108582b63578SDavid Greene uint8_t *Dst = reinterpret_cast<uint8_t *>( 108682b63578SDavid Greene const_cast<uint64_t *>(IntVal.getRawData())); 10875c65cb46SDuncan Sands 108841cb64f4SRafael Espindola if (sys::IsLittleEndianHost) 10895c65cb46SDuncan Sands // Little-endian host - the destination must be ordered from LSB to MSB. 10905c65cb46SDuncan Sands // The source is ordered from LSB to MSB: Do a straight copy. 10915c65cb46SDuncan Sands memcpy(Dst, Src, LoadBytes); 10925c65cb46SDuncan Sands else { 10935c65cb46SDuncan Sands // Big-endian - the destination is an array of 64 bit words ordered from 10945c65cb46SDuncan Sands // LSW to MSW. Each word must be ordered from MSB to LSB. The source is 10955c65cb46SDuncan Sands // ordered from MSB to LSB: Reverse the word order, but not the bytes in 10965c65cb46SDuncan Sands // a word. 10975c65cb46SDuncan Sands while (LoadBytes > sizeof(uint64_t)) { 10985c65cb46SDuncan Sands LoadBytes -= sizeof(uint64_t); 10995c65cb46SDuncan Sands // May not be aligned so use memcpy. 11005c65cb46SDuncan Sands memcpy(Dst, Src + LoadBytes, sizeof(uint64_t)); 11015c65cb46SDuncan Sands Dst += sizeof(uint64_t); 11025c65cb46SDuncan Sands } 11035c65cb46SDuncan Sands 11045c65cb46SDuncan Sands memcpy(Dst + sizeof(uint64_t) - LoadBytes, Src, LoadBytes); 11055c65cb46SDuncan Sands } 11067a9c62baSReid Spencer } 11071202d1b1SDuncan Sands 11081202d1b1SDuncan Sands /// FIXME: document 11091202d1b1SDuncan Sands /// 11101202d1b1SDuncan Sands void ExecutionEngine::LoadValueFromMemory(GenericValue &Result, 11111202d1b1SDuncan Sands GenericValue *Ptr, 1112229907cdSChris Lattner Type *Ty) { 1113cdfe20b9SMicah Villmow const unsigned LoadBytes = getDataLayout()->getTypeStoreSize(Ty); 11141202d1b1SDuncan Sands 11151202d1b1SDuncan Sands switch (Ty->getTypeID()) { 11161202d1b1SDuncan Sands case Type::IntegerTyID: 11171202d1b1SDuncan Sands // An APInt with all words initially zero. 11181202d1b1SDuncan Sands Result.IntVal = APInt(cast<IntegerType>(Ty)->getBitWidth(), 0); 11191202d1b1SDuncan Sands LoadIntFromMemory(Result.IntVal, (uint8_t*)Ptr, LoadBytes); 11201202d1b1SDuncan Sands break; 11217f389e8cSChris Lattner case Type::FloatTyID: 112287aa65f4SReid Spencer Result.FloatVal = *((float*)Ptr); 112387aa65f4SReid Spencer break; 112487aa65f4SReid Spencer case Type::DoubleTyID: 112587aa65f4SReid Spencer Result.DoubleVal = *((double*)Ptr); 11267f389e8cSChris Lattner break; 11277a9c62baSReid Spencer case Type::PointerTyID: 112887aa65f4SReid Spencer Result.PointerVal = *((PointerTy*)Ptr); 11297f389e8cSChris Lattner break; 1130a1336cf5SDale Johannesen case Type::X86_FP80TyID: { 1131a1336cf5SDale Johannesen // This is endian dependent, but it will only work on x86 anyway. 113226d6539eSDuncan Sands // FIXME: Will not trap if loading a signaling NaN. 1133ff306287SDuncan Sands uint64_t y[2]; 11344d7e4ee7SDale Johannesen memcpy(y, Ptr, 10); 11357a162881SJeffrey Yasskin Result.IntVal = APInt(80, y); 1136a1336cf5SDale Johannesen break; 1137a1336cf5SDale Johannesen } 1138be79a7acSNadav Rotem case Type::VectorTyID: { 1139be79a7acSNadav Rotem const VectorType *VT = cast<VectorType>(Ty); 1140be79a7acSNadav Rotem const Type *ElemT = VT->getElementType(); 1141be79a7acSNadav Rotem const unsigned numElems = VT->getNumElements(); 1142be79a7acSNadav Rotem if (ElemT->isFloatTy()) { 1143be79a7acSNadav Rotem Result.AggregateVal.resize(numElems); 1144be79a7acSNadav Rotem for (unsigned i = 0; i < numElems; ++i) 1145be79a7acSNadav Rotem Result.AggregateVal[i].FloatVal = *((float*)Ptr+i); 1146be79a7acSNadav Rotem } 1147be79a7acSNadav Rotem if (ElemT->isDoubleTy()) { 1148be79a7acSNadav Rotem Result.AggregateVal.resize(numElems); 1149be79a7acSNadav Rotem for (unsigned i = 0; i < numElems; ++i) 1150be79a7acSNadav Rotem Result.AggregateVal[i].DoubleVal = *((double*)Ptr+i); 1151be79a7acSNadav Rotem } 1152be79a7acSNadav Rotem if (ElemT->isIntegerTy()) { 1153be79a7acSNadav Rotem GenericValue intZero; 1154be79a7acSNadav Rotem const unsigned elemBitWidth = cast<IntegerType>(ElemT)->getBitWidth(); 1155be79a7acSNadav Rotem intZero.IntVal = APInt(elemBitWidth, 0); 1156be79a7acSNadav Rotem Result.AggregateVal.resize(numElems, intZero); 1157be79a7acSNadav Rotem for (unsigned i = 0; i < numElems; ++i) 1158be79a7acSNadav Rotem LoadIntFromMemory(Result.AggregateVal[i].IntVal, 1159be79a7acSNadav Rotem (uint8_t*)Ptr+((elemBitWidth+7)/8)*i, (elemBitWidth+7)/8); 1160be79a7acSNadav Rotem } 1161be79a7acSNadav Rotem break; 1162be79a7acSNadav Rotem } 11637f389e8cSChris Lattner default: 1164868e3f09SDaniel Dunbar SmallString<256> Msg; 1165868e3f09SDaniel Dunbar raw_svector_ostream OS(Msg); 1166868e3f09SDaniel Dunbar OS << "Cannot load value of type " << *Ty << "!"; 1167868e3f09SDaniel Dunbar report_fatal_error(OS.str()); 11687f389e8cSChris Lattner } 11697f389e8cSChris Lattner } 11707f389e8cSChris Lattner 1171996fe010SChris Lattner void ExecutionEngine::InitializeMemory(const Constant *Init, void *Addr) { 11720967d2dfSDavid Greene DEBUG(dbgs() << "JIT: Initializing " << Addr << " "); 1173b086d382SDale Johannesen DEBUG(Init->dump()); 117400245f42SChris Lattner if (isa<UndefValue>(Init)) 117561753bf8SChris Lattner return; 117600245f42SChris Lattner 117700245f42SChris Lattner if (const ConstantVector *CP = dyn_cast<ConstantVector>(Init)) { 117869d62138SRobert Bocchino unsigned ElementSize = 1179cdfe20b9SMicah Villmow getDataLayout()->getTypeAllocSize(CP->getType()->getElementType()); 118069d62138SRobert Bocchino for (unsigned i = 0, e = CP->getNumOperands(); i != e; ++i) 118169d62138SRobert Bocchino InitializeMemory(CP->getOperand(i), (char*)Addr+i*ElementSize); 118269d62138SRobert Bocchino return; 118300245f42SChris Lattner } 118400245f42SChris Lattner 118500245f42SChris Lattner if (isa<ConstantAggregateZero>(Init)) { 1186cdfe20b9SMicah Villmow memset(Addr, 0, (size_t)getDataLayout()->getTypeAllocSize(Init->getType())); 11871dd86b11SChris Lattner return; 118800245f42SChris Lattner } 118900245f42SChris Lattner 119000245f42SChris Lattner if (const ConstantArray *CPA = dyn_cast<ConstantArray>(Init)) { 119169ddfbfeSDan Gohman unsigned ElementSize = 1192cdfe20b9SMicah Villmow getDataLayout()->getTypeAllocSize(CPA->getType()->getElementType()); 119369ddfbfeSDan Gohman for (unsigned i = 0, e = CPA->getNumOperands(); i != e; ++i) 119469ddfbfeSDan Gohman InitializeMemory(CPA->getOperand(i), (char*)Addr+i*ElementSize); 119569ddfbfeSDan Gohman return; 119600245f42SChris Lattner } 119700245f42SChris Lattner 119800245f42SChris Lattner if (const ConstantStruct *CPS = dyn_cast<ConstantStruct>(Init)) { 119969ddfbfeSDan Gohman const StructLayout *SL = 1200cdfe20b9SMicah Villmow getDataLayout()->getStructLayout(cast<StructType>(CPS->getType())); 120169ddfbfeSDan Gohman for (unsigned i = 0, e = CPS->getNumOperands(); i != e; ++i) 120269ddfbfeSDan Gohman InitializeMemory(CPS->getOperand(i), (char*)Addr+SL->getElementOffset(i)); 120369ddfbfeSDan Gohman return; 120400245f42SChris Lattner } 120500245f42SChris Lattner 120600245f42SChris Lattner if (const ConstantDataSequential *CDS = 120700245f42SChris Lattner dyn_cast<ConstantDataSequential>(Init)) { 120800245f42SChris Lattner // CDS is already laid out in host memory order. 120900245f42SChris Lattner StringRef Data = CDS->getRawDataValues(); 121000245f42SChris Lattner memcpy(Addr, Data.data(), Data.size()); 121100245f42SChris Lattner return; 121200245f42SChris Lattner } 121300245f42SChris Lattner 121400245f42SChris Lattner if (Init->getType()->isFirstClassType()) { 1215996fe010SChris Lattner GenericValue Val = getConstantValue(Init); 1216996fe010SChris Lattner StoreValueToMemory(Val, (GenericValue*)Addr, Init->getType()); 1217996fe010SChris Lattner return; 1218996fe010SChris Lattner } 1219996fe010SChris Lattner 1220868e3f09SDaniel Dunbar DEBUG(dbgs() << "Bad Type: " << *Init->getType() << "\n"); 1221fbcc663cSTorok Edwin llvm_unreachable("Unknown constant type to initialize memory with!"); 1222996fe010SChris Lattner } 1223996fe010SChris Lattner 1224996fe010SChris Lattner /// EmitGlobals - Emit all of the global variables to memory, storing their 1225996fe010SChris Lattner /// addresses into GlobalAddress. This must make sure to copy the contents of 1226996fe010SChris Lattner /// their initializers into the memory. 1227996fe010SChris Lattner void ExecutionEngine::emitGlobals() { 1228996fe010SChris Lattner // Loop over all of the global variables in the program, allocating the memory 12290621caefSChris Lattner // to hold them. If there is more than one module, do a prepass over globals 12300621caefSChris Lattner // to figure out how the different modules should link together. 1231229907cdSChris Lattner std::map<std::pair<std::string, Type*>, 12320621caefSChris Lattner const GlobalValue*> LinkedGlobalsMap; 12330621caefSChris Lattner 12340621caefSChris Lattner if (Modules.size() != 1) { 12350621caefSChris Lattner for (unsigned m = 0, e = Modules.size(); m != e; ++m) { 1236091217beSJeffrey Yasskin Module &M = *Modules[m]; 123749bb65a4SRafael Espindola for (const auto &GV : M.globals()) { 123849bb65a4SRafael Espindola if (GV.hasLocalLinkage() || GV.isDeclaration() || 123949bb65a4SRafael Espindola GV.hasAppendingLinkage() || !GV.hasName()) 12400621caefSChris Lattner continue;// Ignore external globals and globals with internal linkage. 12410621caefSChris Lattner 12420621caefSChris Lattner const GlobalValue *&GVEntry = 124349bb65a4SRafael Espindola LinkedGlobalsMap[std::make_pair(GV.getName(), GV.getType())]; 12440621caefSChris Lattner 12450621caefSChris Lattner // If this is the first time we've seen this global, it is the canonical 12460621caefSChris Lattner // version. 12470621caefSChris Lattner if (!GVEntry) { 124849bb65a4SRafael Espindola GVEntry = &GV; 12490621caefSChris Lattner continue; 12500621caefSChris Lattner } 12510621caefSChris Lattner 12520621caefSChris Lattner // If the existing global is strong, never replace it. 12537157bb76SNico Rieck if (GVEntry->hasExternalLinkage()) 12540621caefSChris Lattner continue; 12550621caefSChris Lattner 12560621caefSChris Lattner // Otherwise, we know it's linkonce/weak, replace it if this is a strong 1257ce4396bcSDale Johannesen // symbol. FIXME is this right for common? 125849bb65a4SRafael Espindola if (GV.hasExternalLinkage() || GVEntry->hasExternalWeakLinkage()) 125949bb65a4SRafael Espindola GVEntry = &GV; 12600621caefSChris Lattner } 12610621caefSChris Lattner } 12620621caefSChris Lattner } 12630621caefSChris Lattner 12640621caefSChris Lattner std::vector<const GlobalValue*> NonCanonicalGlobals; 12650621caefSChris Lattner for (unsigned m = 0, e = Modules.size(); m != e; ++m) { 1266091217beSJeffrey Yasskin Module &M = *Modules[m]; 126749bb65a4SRafael Espindola for (const auto &GV : M.globals()) { 12680621caefSChris Lattner // In the multi-module case, see what this global maps to. 12690621caefSChris Lattner if (!LinkedGlobalsMap.empty()) { 12700621caefSChris Lattner if (const GlobalValue *GVEntry = 127149bb65a4SRafael Espindola LinkedGlobalsMap[std::make_pair(GV.getName(), GV.getType())]) { 12720621caefSChris Lattner // If something else is the canonical global, ignore this one. 127349bb65a4SRafael Espindola if (GVEntry != &GV) { 127449bb65a4SRafael Espindola NonCanonicalGlobals.push_back(&GV); 12750621caefSChris Lattner continue; 12760621caefSChris Lattner } 12770621caefSChris Lattner } 12780621caefSChris Lattner } 12790621caefSChris Lattner 128049bb65a4SRafael Espindola if (!GV.isDeclaration()) { 128149bb65a4SRafael Espindola addGlobalMapping(&GV, getMemoryForGV(&GV)); 1282996fe010SChris Lattner } else { 1283e8bbcfc2SBrian Gaeke // External variable reference. Try to use the dynamic loader to 1284e8bbcfc2SBrian Gaeke // get a pointer to it. 12850621caefSChris Lattner if (void *SymAddr = 128649bb65a4SRafael Espindola sys::DynamicLibrary::SearchForAddressOfSymbol(GV.getName())) 128749bb65a4SRafael Espindola addGlobalMapping(&GV, SymAddr); 12889de0d14dSChris Lattner else { 12892104b8d3SChris Lattner report_fatal_error("Could not resolve external global address: " 129049bb65a4SRafael Espindola +GV.getName()); 12919de0d14dSChris Lattner } 1292996fe010SChris Lattner } 12930621caefSChris Lattner } 12940621caefSChris Lattner 12950621caefSChris Lattner // If there are multiple modules, map the non-canonical globals to their 12960621caefSChris Lattner // canonical location. 12970621caefSChris Lattner if (!NonCanonicalGlobals.empty()) { 12980621caefSChris Lattner for (unsigned i = 0, e = NonCanonicalGlobals.size(); i != e; ++i) { 12990621caefSChris Lattner const GlobalValue *GV = NonCanonicalGlobals[i]; 13000621caefSChris Lattner const GlobalValue *CGV = 13010621caefSChris Lattner LinkedGlobalsMap[std::make_pair(GV->getName(), GV->getType())]; 13020621caefSChris Lattner void *Ptr = getPointerToGlobalIfAvailable(CGV); 13030621caefSChris Lattner assert(Ptr && "Canonical global wasn't codegen'd!"); 1304a67f06b9SNuno Lopes addGlobalMapping(GV, Ptr); 13050621caefSChris Lattner } 13060621caefSChris Lattner } 1307996fe010SChris Lattner 13087a9c62baSReid Spencer // Now that all of the globals are set up in memory, loop through them all 13097a9c62baSReid Spencer // and initialize their contents. 131049bb65a4SRafael Espindola for (const auto &GV : M.globals()) { 131149bb65a4SRafael Espindola if (!GV.isDeclaration()) { 13120621caefSChris Lattner if (!LinkedGlobalsMap.empty()) { 13130621caefSChris Lattner if (const GlobalValue *GVEntry = 131449bb65a4SRafael Espindola LinkedGlobalsMap[std::make_pair(GV.getName(), GV.getType())]) 131549bb65a4SRafael Espindola if (GVEntry != &GV) // Not the canonical variable. 13160621caefSChris Lattner continue; 13170621caefSChris Lattner } 131849bb65a4SRafael Espindola EmitGlobalVariable(&GV); 13196bbe3eceSChris Lattner } 13200621caefSChris Lattner } 13210621caefSChris Lattner } 13220621caefSChris Lattner } 13236bbe3eceSChris Lattner 13246bbe3eceSChris Lattner // EmitGlobalVariable - This method emits the specified global variable to the 13256bbe3eceSChris Lattner // address specified in GlobalAddresses, or allocates new memory if it's not 13266bbe3eceSChris Lattner // already in the map. 1327fbcc0aa1SChris Lattner void ExecutionEngine::EmitGlobalVariable(const GlobalVariable *GV) { 1328748e8579SChris Lattner void *GA = getPointerToGlobalIfAvailable(GV); 1329dc631735SChris Lattner 13302617dcceSCraig Topper if (!GA) { 13316bbe3eceSChris Lattner // If it's not already specified, allocate memory for the global. 13325457ce9aSNicolas Geoffray GA = getMemoryForGV(GV); 13333b442379SAndrew Kaylor 13343b442379SAndrew Kaylor // If we failed to allocate memory for this global, return. 13352617dcceSCraig Topper if (!GA) return; 13363b442379SAndrew Kaylor 1337748e8579SChris Lattner addGlobalMapping(GV, GA); 13386bbe3eceSChris Lattner } 1339fbcc0aa1SChris Lattner 13405457ce9aSNicolas Geoffray // Don't initialize if it's thread local, let the client do it. 13415457ce9aSNicolas Geoffray if (!GV->isThreadLocal()) 13426bbe3eceSChris Lattner InitializeMemory(GV->getInitializer(), GA); 13435457ce9aSNicolas Geoffray 1344229907cdSChris Lattner Type *ElTy = GV->getType()->getElementType(); 1345cdfe20b9SMicah Villmow size_t GVSize = (size_t)getDataLayout()->getTypeAllocSize(ElTy); 1346df1f1524SChris Lattner NumInitBytes += (unsigned)GVSize; 13476bbe3eceSChris Lattner ++NumGlobals; 1348996fe010SChris Lattner } 1349