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 15ee937c80SChris Lattner #define DEBUG_TYPE "jit" 166bf87df5SJeffrey Yasskin #include "llvm/ExecutionEngine/ExecutionEngine.h" 176bf87df5SJeffrey Yasskin 18996fe010SChris Lattner #include "llvm/Constants.h" 19260b0c88SMisha Brukman #include "llvm/DerivedTypes.h" 204bb926d9SMicah Villmow #include "llvm/Instructions.h" 21996fe010SChris Lattner #include "llvm/Module.h" 22ad481312SChris Lattner #include "llvm/ExecutionEngine/GenericValue.h" 23868e3f09SDaniel Dunbar #include "llvm/ADT/SmallString.h" 24390d78b3SChris Lattner #include "llvm/ADT/Statistic.h" 257c16caa3SReid Spencer #include "llvm/Support/Debug.h" 266c2d233eSTorok Edwin #include "llvm/Support/ErrorHandling.h" 276d8dd189SChris Lattner #include "llvm/Support/MutexGuard.h" 286bf87df5SJeffrey Yasskin #include "llvm/Support/ValueHandle.h" 29ccb29cd2STorok Edwin #include "llvm/Support/raw_ostream.h" 30447762daSMichael J. Spencer #include "llvm/Support/DynamicLibrary.h" 31447762daSMichael J. Spencer #include "llvm/Support/Host.h" 327f26246aSDylan Noblesmith #include "llvm/Support/TargetRegistry.h" 33cdfe20b9SMicah Villmow #include "llvm/DataLayout.h" 348418fdcdSDylan Noblesmith #include "llvm/Target/TargetMachine.h" 35579f0713SAnton Korobeynikov #include <cmath> 36579f0713SAnton Korobeynikov #include <cstring> 3729681deeSChris Lattner using namespace llvm; 38996fe010SChris Lattner 39c346ecd7SChris Lattner STATISTIC(NumInitBytes, "Number of bytes of global vars initialized"); 40c346ecd7SChris Lattner STATISTIC(NumGlobals , "Number of global vars initialized"); 41996fe010SChris Lattner 4231faefffSJeffrey Yasskin ExecutionEngine *(*ExecutionEngine::JITCtor)( 4331faefffSJeffrey Yasskin Module *M, 44fc8a2d5aSReid Kleckner std::string *ErrorStr, 45fc8a2d5aSReid Kleckner JITMemoryManager *JMM, 46700d08e1SEric Christopher bool GVsWithCode, 478418fdcdSDylan Noblesmith TargetMachine *TM) = 0; 4870ff8b05SDaniel Dunbar ExecutionEngine *(*ExecutionEngine::MCJITCtor)( 4970ff8b05SDaniel Dunbar Module *M, 5070ff8b05SDaniel Dunbar std::string *ErrorStr, 5170ff8b05SDaniel Dunbar JITMemoryManager *JMM, 5270ff8b05SDaniel Dunbar bool GVsWithCode, 538418fdcdSDylan Noblesmith TargetMachine *TM) = 0; 54091217beSJeffrey Yasskin ExecutionEngine *(*ExecutionEngine::InterpCtor)(Module *M, 55fc8a2d5aSReid Kleckner std::string *ErrorStr) = 0; 562d52c1b8SChris Lattner 57091217beSJeffrey Yasskin ExecutionEngine::ExecutionEngine(Module *M) 58f98e981cSJeffrey Yasskin : EEState(*this), 59abc7901eSDuncan Sands LazyFunctionCreator(0), 60abc7901eSDuncan Sands ExceptionTableRegister(0), 61abc7901eSDuncan Sands ExceptionTableDeregister(0) { 624567db45SJeffrey Yasskin CompilingLazily = false; 63cdc0060eSEvan Cheng GVCompilationDisabled = false; 6484a9055eSEvan Cheng SymbolSearchingDisabled = false; 65091217beSJeffrey Yasskin Modules.push_back(M); 66091217beSJeffrey Yasskin assert(M && "Module is null?"); 67260b0c88SMisha Brukman } 68260b0c88SMisha Brukman 6992f8b30dSBrian Gaeke ExecutionEngine::~ExecutionEngine() { 70603682adSReid Spencer clearAllGlobalMappings(); 710621caefSChris Lattner for (unsigned i = 0, e = Modules.size(); i != e; ++i) 720621caefSChris Lattner delete Modules[i]; 7392f8b30dSBrian Gaeke } 7492f8b30dSBrian Gaeke 75abc7901eSDuncan Sands void ExecutionEngine::DeregisterAllTables() { 76abc7901eSDuncan Sands if (ExceptionTableDeregister) { 77f045b7abSEric Christopher DenseMap<const Function*, void*>::iterator it = AllExceptionTables.begin(); 78f045b7abSEric Christopher DenseMap<const Function*, void*>::iterator ite = AllExceptionTables.end(); 79f045b7abSEric Christopher for (; it != ite; ++it) 80f045b7abSEric Christopher ExceptionTableDeregister(it->second); 81abc7901eSDuncan Sands AllExceptionTables.clear(); 82abc7901eSDuncan Sands } 83abc7901eSDuncan Sands } 84abc7901eSDuncan Sands 85a4044332SJeffrey Yasskin namespace { 86868e3f09SDaniel Dunbar /// \brief Helper class which uses a value handler to automatically deletes the 87868e3f09SDaniel Dunbar /// memory block when the GlobalVariable is destroyed. 88a4044332SJeffrey Yasskin class GVMemoryBlock : public CallbackVH { 89a4044332SJeffrey Yasskin GVMemoryBlock(const GlobalVariable *GV) 90a4044332SJeffrey Yasskin : CallbackVH(const_cast<GlobalVariable*>(GV)) {} 91a4044332SJeffrey Yasskin 92a4044332SJeffrey Yasskin public: 93868e3f09SDaniel Dunbar /// \brief Returns the address the GlobalVariable should be written into. The 94868e3f09SDaniel Dunbar /// GVMemoryBlock object prefixes that. 95cdfe20b9SMicah Villmow static char *Create(const GlobalVariable *GV, const DataLayout& TD) { 96229907cdSChris Lattner Type *ElTy = GV->getType()->getElementType(); 97a4044332SJeffrey Yasskin size_t GVSize = (size_t)TD.getTypeAllocSize(ElTy); 98a4044332SJeffrey Yasskin void *RawMemory = ::operator new( 99cdfe20b9SMicah Villmow DataLayout::RoundUpAlignment(sizeof(GVMemoryBlock), 100a4044332SJeffrey Yasskin TD.getPreferredAlignment(GV)) 101a4044332SJeffrey Yasskin + GVSize); 102a4044332SJeffrey Yasskin new(RawMemory) GVMemoryBlock(GV); 103a4044332SJeffrey Yasskin return static_cast<char*>(RawMemory) + sizeof(GVMemoryBlock); 104a4044332SJeffrey Yasskin } 105a4044332SJeffrey Yasskin 106a4044332SJeffrey Yasskin virtual void deleted() { 107a4044332SJeffrey Yasskin // We allocated with operator new and with some extra memory hanging off the 108a4044332SJeffrey Yasskin // end, so don't just delete this. I'm not sure if this is actually 109a4044332SJeffrey Yasskin // required. 110a4044332SJeffrey Yasskin this->~GVMemoryBlock(); 111a4044332SJeffrey Yasskin ::operator delete(this); 112a4044332SJeffrey Yasskin } 113a4044332SJeffrey Yasskin }; 114a4044332SJeffrey Yasskin } // anonymous namespace 115a4044332SJeffrey Yasskin 116a4044332SJeffrey Yasskin char *ExecutionEngine::getMemoryForGV(const GlobalVariable *GV) { 117cdfe20b9SMicah Villmow return GVMemoryBlock::Create(GV, *getDataLayout()); 1185457ce9aSNicolas Geoffray } 1195457ce9aSNicolas Geoffray 120091217beSJeffrey Yasskin bool ExecutionEngine::removeModule(Module *M) { 121091217beSJeffrey Yasskin for(SmallVector<Module *, 1>::iterator I = Modules.begin(), 122324fe890SDevang Patel E = Modules.end(); I != E; ++I) { 123091217beSJeffrey Yasskin Module *Found = *I; 124091217beSJeffrey Yasskin if (Found == M) { 125324fe890SDevang Patel Modules.erase(I); 126091217beSJeffrey Yasskin clearGlobalMappingsFromModule(M); 127091217beSJeffrey Yasskin return true; 128324fe890SDevang Patel } 129324fe890SDevang Patel } 130091217beSJeffrey Yasskin return false; 131617001d8SNate Begeman } 132617001d8SNate Begeman 1330621caefSChris Lattner Function *ExecutionEngine::FindFunctionNamed(const char *FnName) { 1340621caefSChris Lattner for (unsigned i = 0, e = Modules.size(); i != e; ++i) { 135091217beSJeffrey Yasskin if (Function *F = Modules[i]->getFunction(FnName)) 1360621caefSChris Lattner return F; 1370621caefSChris Lattner } 1380621caefSChris Lattner return 0; 1390621caefSChris Lattner } 1400621caefSChris Lattner 1410621caefSChris Lattner 142868e3f09SDaniel Dunbar void *ExecutionEngineState::RemoveMapping(const MutexGuard &, 143868e3f09SDaniel Dunbar const GlobalValue *ToUnmap) { 144d0fc8f80SJeffrey Yasskin GlobalAddressMapTy::iterator I = GlobalAddressMap.find(ToUnmap); 145307c053fSJeffrey Yasskin void *OldVal; 146868e3f09SDaniel Dunbar 147868e3f09SDaniel Dunbar // FIXME: This is silly, we shouldn't end up with a mapping -> 0 in the 148868e3f09SDaniel Dunbar // GlobalAddressMap. 149307c053fSJeffrey Yasskin if (I == GlobalAddressMap.end()) 150307c053fSJeffrey Yasskin OldVal = 0; 151307c053fSJeffrey Yasskin else { 152307c053fSJeffrey Yasskin OldVal = I->second; 153307c053fSJeffrey Yasskin GlobalAddressMap.erase(I); 154307c053fSJeffrey Yasskin } 155307c053fSJeffrey Yasskin 156307c053fSJeffrey Yasskin GlobalAddressReverseMap.erase(OldVal); 157307c053fSJeffrey Yasskin return OldVal; 158307c053fSJeffrey Yasskin } 159307c053fSJeffrey Yasskin 1606d8dd189SChris Lattner void ExecutionEngine::addGlobalMapping(const GlobalValue *GV, void *Addr) { 1616d8dd189SChris Lattner MutexGuard locked(lock); 1626d8dd189SChris Lattner 1630967d2dfSDavid Greene DEBUG(dbgs() << "JIT: Map \'" << GV->getName() 1649813b0b0SDaniel Dunbar << "\' to [" << Addr << "]\n";); 165d0fc8f80SJeffrey Yasskin void *&CurVal = EEState.getGlobalAddressMap(locked)[GV]; 1666d8dd189SChris Lattner assert((CurVal == 0 || Addr == 0) && "GlobalMapping already established!"); 1676d8dd189SChris Lattner CurVal = Addr; 1686d8dd189SChris Lattner 169868e3f09SDaniel Dunbar // If we are using the reverse mapping, add it too. 170f98e981cSJeffrey Yasskin if (!EEState.getGlobalAddressReverseMap(locked).empty()) { 1716bf87df5SJeffrey Yasskin AssertingVH<const GlobalValue> &V = 172f98e981cSJeffrey Yasskin EEState.getGlobalAddressReverseMap(locked)[Addr]; 1736d8dd189SChris Lattner assert((V == 0 || GV == 0) && "GlobalMapping already established!"); 1746d8dd189SChris Lattner V = GV; 1756d8dd189SChris Lattner } 1766d8dd189SChris Lattner } 1776d8dd189SChris Lattner 1786d8dd189SChris Lattner void ExecutionEngine::clearAllGlobalMappings() { 1796d8dd189SChris Lattner MutexGuard locked(lock); 1806d8dd189SChris Lattner 181f98e981cSJeffrey Yasskin EEState.getGlobalAddressMap(locked).clear(); 182f98e981cSJeffrey Yasskin EEState.getGlobalAddressReverseMap(locked).clear(); 1836d8dd189SChris Lattner } 1846d8dd189SChris Lattner 1858f83fc4dSNate Begeman void ExecutionEngine::clearGlobalMappingsFromModule(Module *M) { 1868f83fc4dSNate Begeman MutexGuard locked(lock); 1878f83fc4dSNate Begeman 188868e3f09SDaniel Dunbar for (Module::iterator FI = M->begin(), FE = M->end(); FI != FE; ++FI) 189f98e981cSJeffrey Yasskin EEState.RemoveMapping(locked, FI); 1908f83fc4dSNate Begeman for (Module::global_iterator GI = M->global_begin(), GE = M->global_end(); 191868e3f09SDaniel Dunbar GI != GE; ++GI) 192f98e981cSJeffrey Yasskin EEState.RemoveMapping(locked, GI); 1938f83fc4dSNate Begeman } 1948f83fc4dSNate Begeman 195ee181730SChris Lattner void *ExecutionEngine::updateGlobalMapping(const GlobalValue *GV, void *Addr) { 1966d8dd189SChris Lattner MutexGuard locked(lock); 1976d8dd189SChris Lattner 198d0fc8f80SJeffrey Yasskin ExecutionEngineState::GlobalAddressMapTy &Map = 199f98e981cSJeffrey Yasskin EEState.getGlobalAddressMap(locked); 200ee181730SChris Lattner 2016d8dd189SChris Lattner // Deleting from the mapping? 202868e3f09SDaniel Dunbar if (Addr == 0) 203f98e981cSJeffrey Yasskin return EEState.RemoveMapping(locked, GV); 204ee181730SChris Lattner 205d0fc8f80SJeffrey Yasskin void *&CurVal = Map[GV]; 206ee181730SChris Lattner void *OldVal = CurVal; 207ee181730SChris Lattner 208f98e981cSJeffrey Yasskin if (CurVal && !EEState.getGlobalAddressReverseMap(locked).empty()) 209f98e981cSJeffrey Yasskin EEState.getGlobalAddressReverseMap(locked).erase(CurVal); 2106d8dd189SChris Lattner CurVal = Addr; 2116d8dd189SChris Lattner 212868e3f09SDaniel Dunbar // If we are using the reverse mapping, add it too. 213f98e981cSJeffrey Yasskin if (!EEState.getGlobalAddressReverseMap(locked).empty()) { 2146bf87df5SJeffrey Yasskin AssertingVH<const GlobalValue> &V = 215f98e981cSJeffrey Yasskin EEState.getGlobalAddressReverseMap(locked)[Addr]; 2166d8dd189SChris Lattner assert((V == 0 || GV == 0) && "GlobalMapping already established!"); 2176d8dd189SChris Lattner V = GV; 2186d8dd189SChris Lattner } 219ee181730SChris Lattner return OldVal; 2206d8dd189SChris Lattner } 2216d8dd189SChris Lattner 2226d8dd189SChris Lattner void *ExecutionEngine::getPointerToGlobalIfAvailable(const GlobalValue *GV) { 2236d8dd189SChris Lattner MutexGuard locked(lock); 2246d8dd189SChris Lattner 225d0fc8f80SJeffrey Yasskin ExecutionEngineState::GlobalAddressMapTy::iterator I = 226d0fc8f80SJeffrey Yasskin EEState.getGlobalAddressMap(locked).find(GV); 227f98e981cSJeffrey Yasskin return I != EEState.getGlobalAddressMap(locked).end() ? I->second : 0; 2286d8dd189SChris Lattner } 2296d8dd189SChris Lattner 230748e8579SChris Lattner const GlobalValue *ExecutionEngine::getGlobalValueAtAddress(void *Addr) { 23179876f52SReid Spencer MutexGuard locked(lock); 23279876f52SReid Spencer 233748e8579SChris Lattner // If we haven't computed the reverse mapping yet, do so first. 234f98e981cSJeffrey Yasskin if (EEState.getGlobalAddressReverseMap(locked).empty()) { 235d0fc8f80SJeffrey Yasskin for (ExecutionEngineState::GlobalAddressMapTy::iterator 236f98e981cSJeffrey Yasskin I = EEState.getGlobalAddressMap(locked).begin(), 237f98e981cSJeffrey Yasskin E = EEState.getGlobalAddressMap(locked).end(); I != E; ++I) 238e4f47434SDaniel Dunbar EEState.getGlobalAddressReverseMap(locked).insert(std::make_pair( 239e4f47434SDaniel Dunbar I->second, I->first)); 240748e8579SChris Lattner } 241748e8579SChris Lattner 2426bf87df5SJeffrey Yasskin std::map<void *, AssertingVH<const GlobalValue> >::iterator I = 243f98e981cSJeffrey Yasskin EEState.getGlobalAddressReverseMap(locked).find(Addr); 244f98e981cSJeffrey Yasskin return I != EEState.getGlobalAddressReverseMap(locked).end() ? I->second : 0; 245748e8579SChris Lattner } 2465a0d4829SChris Lattner 247bfd38abbSJeffrey Yasskin namespace { 248bfd38abbSJeffrey Yasskin class ArgvArray { 249bfd38abbSJeffrey Yasskin char *Array; 250bfd38abbSJeffrey Yasskin std::vector<char*> Values; 251bfd38abbSJeffrey Yasskin public: 252bfd38abbSJeffrey Yasskin ArgvArray() : Array(NULL) {} 253bfd38abbSJeffrey Yasskin ~ArgvArray() { clear(); } 254bfd38abbSJeffrey Yasskin void clear() { 255bfd38abbSJeffrey Yasskin delete[] Array; 256bfd38abbSJeffrey Yasskin Array = NULL; 257bfd38abbSJeffrey Yasskin for (size_t I = 0, E = Values.size(); I != E; ++I) { 258bfd38abbSJeffrey Yasskin delete[] Values[I]; 259bfd38abbSJeffrey Yasskin } 260bfd38abbSJeffrey Yasskin Values.clear(); 261bfd38abbSJeffrey Yasskin } 262bfd38abbSJeffrey Yasskin /// Turn a vector of strings into a nice argv style array of pointers to null 263bfd38abbSJeffrey Yasskin /// terminated strings. 264bfd38abbSJeffrey Yasskin void *reset(LLVMContext &C, ExecutionEngine *EE, 265bfd38abbSJeffrey Yasskin const std::vector<std::string> &InputArgv); 266bfd38abbSJeffrey Yasskin }; 267bfd38abbSJeffrey Yasskin } // anonymous namespace 268bfd38abbSJeffrey Yasskin void *ArgvArray::reset(LLVMContext &C, ExecutionEngine *EE, 2695a0d4829SChris Lattner const std::vector<std::string> &InputArgv) { 270bfd38abbSJeffrey Yasskin clear(); // Free the old contents. 2714bb926d9SMicah Villmow unsigned PtrSize = EE->getDataLayout()->getPointerSize(0); 272bfd38abbSJeffrey Yasskin Array = new char[(InputArgv.size()+1)*PtrSize]; 2735a0d4829SChris Lattner 274bfd38abbSJeffrey Yasskin DEBUG(dbgs() << "JIT: ARGV = " << (void*)Array << "\n"); 275229907cdSChris Lattner Type *SBytePtr = Type::getInt8PtrTy(C); 2765a0d4829SChris Lattner 2775a0d4829SChris Lattner for (unsigned i = 0; i != InputArgv.size(); ++i) { 2785a0d4829SChris Lattner unsigned Size = InputArgv[i].size()+1; 2795a0d4829SChris Lattner char *Dest = new char[Size]; 280bfd38abbSJeffrey Yasskin Values.push_back(Dest); 2810967d2dfSDavid Greene DEBUG(dbgs() << "JIT: ARGV[" << i << "] = " << (void*)Dest << "\n"); 2825a0d4829SChris Lattner 2835a0d4829SChris Lattner std::copy(InputArgv[i].begin(), InputArgv[i].end(), Dest); 2845a0d4829SChris Lattner Dest[Size-1] = 0; 2855a0d4829SChris Lattner 286bfd38abbSJeffrey Yasskin // Endian safe: Array[i] = (PointerTy)Dest; 287bfd38abbSJeffrey Yasskin EE->StoreValueToMemory(PTOGV(Dest), (GenericValue*)(Array+i*PtrSize), 2885a0d4829SChris Lattner SBytePtr); 2895a0d4829SChris Lattner } 2905a0d4829SChris Lattner 2915a0d4829SChris Lattner // Null terminate it 2925a0d4829SChris Lattner EE->StoreValueToMemory(PTOGV(0), 293bfd38abbSJeffrey Yasskin (GenericValue*)(Array+InputArgv.size()*PtrSize), 2945a0d4829SChris Lattner SBytePtr); 295bfd38abbSJeffrey Yasskin return Array; 2965a0d4829SChris Lattner } 2975a0d4829SChris Lattner 29841fa2bd1SChris Lattner void ExecutionEngine::runStaticConstructorsDestructors(Module *module, 29941fa2bd1SChris Lattner bool isDtors) { 300faae50b6SChris Lattner const char *Name = isDtors ? "llvm.global_dtors" : "llvm.global_ctors"; 3011a9a0b7bSEvan Cheng GlobalVariable *GV = module->getNamedGlobal(Name); 302fe36eaebSChris Lattner 303fe36eaebSChris Lattner // If this global has internal linkage, or if it has a use, then it must be 304fe36eaebSChris Lattner // an old-style (llvmgcc3) static ctor with __main linked in and in use. If 3050621caefSChris Lattner // this is the case, don't execute any of the global ctors, __main will do 3060621caefSChris Lattner // it. 3076de96a1bSRafael Espindola if (!GV || GV->isDeclaration() || GV->hasLocalLinkage()) return; 308faae50b6SChris Lattner 3090cbfcb2bSNick Lewycky // Should be an array of '{ i32, void ()* }' structs. The first value is 3100621caefSChris Lattner // the init priority, which we ignore. 31100245f42SChris Lattner ConstantArray *InitList = dyn_cast<ConstantArray>(GV->getInitializer()); 31200245f42SChris Lattner if (InitList == 0) 3130f857898SNick Lewycky return; 314868e3f09SDaniel Dunbar for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i) { 31500245f42SChris Lattner ConstantStruct *CS = dyn_cast<ConstantStruct>(InitList->getOperand(i)); 31600245f42SChris Lattner if (CS == 0) continue; 317faae50b6SChris Lattner 318faae50b6SChris Lattner Constant *FP = CS->getOperand(1); 319faae50b6SChris Lattner if (FP->isNullValue()) 3200f857898SNick Lewycky continue; // Found a sentinal value, ignore. 321faae50b6SChris Lattner 322868e3f09SDaniel Dunbar // Strip off constant expression casts. 323faae50b6SChris Lattner if (ConstantExpr *CE = dyn_cast<ConstantExpr>(FP)) 3246c38f0bbSReid Spencer if (CE->isCast()) 325faae50b6SChris Lattner FP = CE->getOperand(0); 326868e3f09SDaniel Dunbar 327faae50b6SChris Lattner // Execute the ctor/dtor function! 328868e3f09SDaniel Dunbar if (Function *F = dyn_cast<Function>(FP)) 329faae50b6SChris Lattner runFunction(F, std::vector<GenericValue>()); 330868e3f09SDaniel Dunbar 331868e3f09SDaniel Dunbar // FIXME: It is marginally lame that we just do nothing here if we see an 332868e3f09SDaniel Dunbar // entry we don't recognize. It might not be unreasonable for the verifier 333868e3f09SDaniel Dunbar // to not even allow this and just assert here. 334faae50b6SChris Lattner } 335faae50b6SChris Lattner } 3361a9a0b7bSEvan Cheng 3371a9a0b7bSEvan Cheng void ExecutionEngine::runStaticConstructorsDestructors(bool isDtors) { 3381a9a0b7bSEvan Cheng // Execute global ctors/dtors for each module in the program. 339868e3f09SDaniel Dunbar for (unsigned i = 0, e = Modules.size(); i != e; ++i) 340868e3f09SDaniel Dunbar runStaticConstructorsDestructors(Modules[i], isDtors); 3410621caefSChris Lattner } 342faae50b6SChris Lattner 343cf3e3017SDan Gohman #ifndef NDEBUG 3441202d1b1SDuncan Sands /// isTargetNullPtr - Return whether the target pointer stored at Loc is null. 3451202d1b1SDuncan Sands static bool isTargetNullPtr(ExecutionEngine *EE, void *Loc) { 3464bb926d9SMicah Villmow unsigned PtrSize = EE->getDataLayout()->getPointerSize(0); 3471202d1b1SDuncan Sands for (unsigned i = 0; i < PtrSize; ++i) 3481202d1b1SDuncan Sands if (*(i + (uint8_t*)Loc)) 3491202d1b1SDuncan Sands return false; 3501202d1b1SDuncan Sands return true; 3511202d1b1SDuncan Sands } 352cf3e3017SDan Gohman #endif 3531202d1b1SDuncan Sands 3545a0d4829SChris Lattner int ExecutionEngine::runFunctionAsMain(Function *Fn, 3555a0d4829SChris Lattner const std::vector<std::string> &argv, 3565a0d4829SChris Lattner const char * const * envp) { 3575a0d4829SChris Lattner std::vector<GenericValue> GVArgs; 3585a0d4829SChris Lattner GenericValue GVArgc; 35987aa65f4SReid Spencer GVArgc.IntVal = APInt(32, argv.size()); 3608c32c111SAnton Korobeynikov 3618c32c111SAnton Korobeynikov // Check main() type 362b1cad0b3SChris Lattner unsigned NumArgs = Fn->getFunctionType()->getNumParams(); 363229907cdSChris Lattner FunctionType *FTy = Fn->getFunctionType(); 364229907cdSChris Lattner Type* PPInt8Ty = Type::getInt8PtrTy(Fn->getContext())->getPointerTo(); 365868e3f09SDaniel Dunbar 366868e3f09SDaniel Dunbar // Check the argument types. 367868e3f09SDaniel Dunbar if (NumArgs > 3) 3682104b8d3SChris Lattner report_fatal_error("Invalid number of arguments of main() supplied"); 369868e3f09SDaniel Dunbar if (NumArgs >= 3 && FTy->getParamType(2) != PPInt8Ty) 370868e3f09SDaniel Dunbar report_fatal_error("Invalid type for third argument of main() supplied"); 371868e3f09SDaniel Dunbar if (NumArgs >= 2 && FTy->getParamType(1) != PPInt8Ty) 372868e3f09SDaniel Dunbar report_fatal_error("Invalid type for second argument of main() supplied"); 373868e3f09SDaniel Dunbar if (NumArgs >= 1 && !FTy->getParamType(0)->isIntegerTy(32)) 374868e3f09SDaniel Dunbar report_fatal_error("Invalid type for first argument of main() supplied"); 375868e3f09SDaniel Dunbar if (!FTy->getReturnType()->isIntegerTy() && 376868e3f09SDaniel Dunbar !FTy->getReturnType()->isVoidTy()) 377868e3f09SDaniel Dunbar report_fatal_error("Invalid return type of main() supplied"); 3788c32c111SAnton Korobeynikov 379bfd38abbSJeffrey Yasskin ArgvArray CArgv; 380bfd38abbSJeffrey Yasskin ArgvArray CEnv; 381b1cad0b3SChris Lattner if (NumArgs) { 3825a0d4829SChris Lattner GVArgs.push_back(GVArgc); // Arg #0 = argc. 383b1cad0b3SChris Lattner if (NumArgs > 1) { 38455f1c09eSOwen Anderson // Arg #1 = argv. 385bfd38abbSJeffrey Yasskin GVArgs.push_back(PTOGV(CArgv.reset(Fn->getContext(), this, argv))); 3861202d1b1SDuncan Sands assert(!isTargetNullPtr(this, GVTOP(GVArgs[1])) && 387b1cad0b3SChris Lattner "argv[0] was null after CreateArgv"); 388b1cad0b3SChris Lattner if (NumArgs > 2) { 3895a0d4829SChris Lattner std::vector<std::string> EnvVars; 3905a0d4829SChris Lattner for (unsigned i = 0; envp[i]; ++i) 3915a0d4829SChris Lattner EnvVars.push_back(envp[i]); 39255f1c09eSOwen Anderson // Arg #2 = envp. 393bfd38abbSJeffrey Yasskin GVArgs.push_back(PTOGV(CEnv.reset(Fn->getContext(), this, EnvVars))); 394b1cad0b3SChris Lattner } 395b1cad0b3SChris Lattner } 396b1cad0b3SChris Lattner } 397868e3f09SDaniel Dunbar 39887aa65f4SReid Spencer return runFunction(Fn, GVArgs).IntVal.getZExtValue(); 3995a0d4829SChris Lattner } 4005a0d4829SChris Lattner 401091217beSJeffrey Yasskin ExecutionEngine *ExecutionEngine::create(Module *M, 402603682adSReid Spencer bool ForceInterpreter, 4037ff05bf5SEvan Cheng std::string *ErrorStr, 40470415d97SJeffrey Yasskin CodeGenOpt::Level OptLevel, 40570415d97SJeffrey Yasskin bool GVsWithCode) { 406add6f1d2SOwen Anderson EngineBuilder EB = EngineBuilder(M) 407fc8a2d5aSReid Kleckner .setEngineKind(ForceInterpreter 408fc8a2d5aSReid Kleckner ? EngineKind::Interpreter 409fc8a2d5aSReid Kleckner : EngineKind::JIT) 410fc8a2d5aSReid Kleckner .setErrorStr(ErrorStr) 411fc8a2d5aSReid Kleckner .setOptLevel(OptLevel) 412add6f1d2SOwen Anderson .setAllocateGVsWithCode(GVsWithCode); 413add6f1d2SOwen Anderson 414add6f1d2SOwen Anderson return EB.create(); 415fc8a2d5aSReid Kleckner } 4164bd3bd5bSBrian Gaeke 4170bd34fbdSDylan Noblesmith /// createJIT - This is the factory method for creating a JIT for the current 4180bd34fbdSDylan Noblesmith /// machine, it does not fall back to the interpreter. This takes ownership 4190bd34fbdSDylan Noblesmith /// of the module. 4200bd34fbdSDylan Noblesmith ExecutionEngine *ExecutionEngine::createJIT(Module *M, 4210bd34fbdSDylan Noblesmith std::string *ErrorStr, 4220bd34fbdSDylan Noblesmith JITMemoryManager *JMM, 42319a58df9SDylan Noblesmith CodeGenOpt::Level OL, 4240bd34fbdSDylan Noblesmith bool GVsWithCode, 4252129f596SEvan Cheng Reloc::Model RM, 4260bd34fbdSDylan Noblesmith CodeModel::Model CMM) { 4270bd34fbdSDylan Noblesmith if (ExecutionEngine::JITCtor == 0) { 4280bd34fbdSDylan Noblesmith if (ErrorStr) 4290bd34fbdSDylan Noblesmith *ErrorStr = "JIT has not been linked in."; 4300bd34fbdSDylan Noblesmith return 0; 4310bd34fbdSDylan Noblesmith } 4320bd34fbdSDylan Noblesmith 4330bd34fbdSDylan Noblesmith // Use the defaults for extra parameters. Users can use EngineBuilder to 4340bd34fbdSDylan Noblesmith // set them. 435add6f1d2SOwen Anderson EngineBuilder EB(M); 436add6f1d2SOwen Anderson EB.setEngineKind(EngineKind::JIT); 437add6f1d2SOwen Anderson EB.setErrorStr(ErrorStr); 438add6f1d2SOwen Anderson EB.setRelocationModel(RM); 439add6f1d2SOwen Anderson EB.setCodeModel(CMM); 440add6f1d2SOwen Anderson EB.setAllocateGVsWithCode(GVsWithCode); 441add6f1d2SOwen Anderson EB.setOptLevel(OL); 442add6f1d2SOwen Anderson EB.setJITMemoryManager(JMM); 4430bd34fbdSDylan Noblesmith 444dff24786SPeter Collingbourne // TODO: permit custom TargetOptions here 445add6f1d2SOwen Anderson TargetMachine *TM = EB.selectTarget(); 4460bd34fbdSDylan Noblesmith if (!TM || (ErrorStr && ErrorStr->length() > 0)) return 0; 4470bd34fbdSDylan Noblesmith 4487f26246aSDylan Noblesmith return ExecutionEngine::JITCtor(M, ErrorStr, JMM, GVsWithCode, TM); 4490bd34fbdSDylan Noblesmith } 4500bd34fbdSDylan Noblesmith 451add6f1d2SOwen Anderson ExecutionEngine *EngineBuilder::create(TargetMachine *TM) { 45225a3d816SBenjamin Kramer OwningPtr<TargetMachine> TheTM(TM); // Take ownership. 45325a3d816SBenjamin Kramer 454a53414fdSNick Lewycky // Make sure we can resolve symbols in the program as well. The zero arg 455a53414fdSNick Lewycky // to the function tells DynamicLibrary to load the program, not a library. 456a53414fdSNick Lewycky if (sys::DynamicLibrary::LoadLibraryPermanently(0, ErrorStr)) 457a53414fdSNick Lewycky return 0; 458a53414fdSNick Lewycky 459fc8a2d5aSReid Kleckner // If the user specified a memory manager but didn't specify which engine to 460fc8a2d5aSReid Kleckner // create, we assume they only want the JIT, and we fail if they only want 461fc8a2d5aSReid Kleckner // the interpreter. 462fc8a2d5aSReid Kleckner if (JMM) { 46341fa2bd1SChris Lattner if (WhichEngine & EngineKind::JIT) 464fc8a2d5aSReid Kleckner WhichEngine = EngineKind::JIT; 46541fa2bd1SChris Lattner else { 4668bcc6445SChris Lattner if (ErrorStr) 467fc8a2d5aSReid Kleckner *ErrorStr = "Cannot create an interpreter with a memory manager."; 46841fa2bd1SChris Lattner return 0; 469fc8a2d5aSReid Kleckner } 4704bd3bd5bSBrian Gaeke } 4714bd3bd5bSBrian Gaeke 472fc8a2d5aSReid Kleckner // Unless the interpreter was explicitly selected or the JIT is not linked, 473fc8a2d5aSReid Kleckner // try making a JIT. 47425a3d816SBenjamin Kramer if ((WhichEngine & EngineKind::JIT) && TheTM) { 4757f26246aSDylan Noblesmith Triple TT(M->getTargetTriple()); 4767f26246aSDylan Noblesmith if (!TM->getTarget().hasJIT()) { 4777f26246aSDylan Noblesmith errs() << "WARNING: This target JIT is not designed for the host" 4787f26246aSDylan Noblesmith << " you are running. If bad things happen, please choose" 4797f26246aSDylan Noblesmith << " a different -march switch.\n"; 4807f26246aSDylan Noblesmith } 4817f26246aSDylan Noblesmith 48270ff8b05SDaniel Dunbar if (UseMCJIT && ExecutionEngine::MCJITCtor) { 48370ff8b05SDaniel Dunbar ExecutionEngine *EE = 4847f26246aSDylan Noblesmith ExecutionEngine::MCJITCtor(M, ErrorStr, JMM, 48525a3d816SBenjamin Kramer AllocateGVsWithCode, TheTM.take()); 48670ff8b05SDaniel Dunbar if (EE) return EE; 48770ff8b05SDaniel Dunbar } else if (ExecutionEngine::JITCtor) { 48841fa2bd1SChris Lattner ExecutionEngine *EE = 4897f26246aSDylan Noblesmith ExecutionEngine::JITCtor(M, ErrorStr, JMM, 49025a3d816SBenjamin Kramer AllocateGVsWithCode, TheTM.take()); 49141fa2bd1SChris Lattner if (EE) return EE; 49241fa2bd1SChris Lattner } 493fc8a2d5aSReid Kleckner } 494fc8a2d5aSReid Kleckner 495fc8a2d5aSReid Kleckner // If we can't make a JIT and we didn't request one specifically, try making 496fc8a2d5aSReid Kleckner // an interpreter instead. 49741fa2bd1SChris Lattner if (WhichEngine & EngineKind::Interpreter) { 49841fa2bd1SChris Lattner if (ExecutionEngine::InterpCtor) 499091217beSJeffrey Yasskin return ExecutionEngine::InterpCtor(M, ErrorStr); 5008bcc6445SChris Lattner if (ErrorStr) 50141fa2bd1SChris Lattner *ErrorStr = "Interpreter has not been linked in."; 50241fa2bd1SChris Lattner return 0; 503fc8a2d5aSReid Kleckner } 504fc8a2d5aSReid Kleckner 505bea6753fSJim Grosbach if ((WhichEngine & EngineKind::JIT) && ExecutionEngine::JITCtor == 0 && 506bea6753fSJim Grosbach ExecutionEngine::MCJITCtor == 0) { 5078bcc6445SChris Lattner if (ErrorStr) 5088bcc6445SChris Lattner *ErrorStr = "JIT has not been linked in."; 5098bcc6445SChris Lattner } 510868e3f09SDaniel Dunbar 51141fa2bd1SChris Lattner return 0; 512b5163bb9SChris Lattner } 513b5163bb9SChris Lattner 514996fe010SChris Lattner void *ExecutionEngine::getPointerToGlobal(const GlobalValue *GV) { 5151678e859SBrian Gaeke if (Function *F = const_cast<Function*>(dyn_cast<Function>(GV))) 516996fe010SChris Lattner return getPointerToFunction(F); 517996fe010SChris Lattner 51879876f52SReid Spencer MutexGuard locked(lock); 519868e3f09SDaniel Dunbar if (void *P = EEState.getGlobalAddressMap(locked)[GV]) 520868e3f09SDaniel Dunbar return P; 52169e84901SJeff Cohen 52269e84901SJeff Cohen // Global variable might have been added since interpreter started. 52369e84901SJeff Cohen if (GlobalVariable *GVar = 52469e84901SJeff Cohen const_cast<GlobalVariable *>(dyn_cast<GlobalVariable>(GV))) 52569e84901SJeff Cohen EmitGlobalVariable(GVar); 52669e84901SJeff Cohen else 527fbcc663cSTorok Edwin llvm_unreachable("Global hasn't had an address allocated yet!"); 528868e3f09SDaniel Dunbar 529d0fc8f80SJeffrey Yasskin return EEState.getGlobalAddressMap(locked)[GV]; 530996fe010SChris Lattner } 531996fe010SChris Lattner 532868e3f09SDaniel Dunbar /// \brief Converts a Constant* into a GenericValue, including handling of 533868e3f09SDaniel Dunbar /// ConstantExpr values. 534996fe010SChris Lattner GenericValue ExecutionEngine::getConstantValue(const Constant *C) { 5356c38f0bbSReid Spencer // If its undefined, return the garbage. 536bcbdbfb3SJay Foad if (isa<UndefValue>(C)) { 537bcbdbfb3SJay Foad GenericValue Result; 538bcbdbfb3SJay Foad switch (C->getType()->getTypeID()) { 539bcbdbfb3SJay Foad case Type::IntegerTyID: 540bcbdbfb3SJay Foad case Type::X86_FP80TyID: 541bcbdbfb3SJay Foad case Type::FP128TyID: 542bcbdbfb3SJay Foad case Type::PPC_FP128TyID: 543bcbdbfb3SJay Foad // Although the value is undefined, we still have to construct an APInt 544bcbdbfb3SJay Foad // with the correct bit width. 545bcbdbfb3SJay Foad Result.IntVal = APInt(C->getType()->getPrimitiveSizeInBits(), 0); 546bcbdbfb3SJay Foad break; 547bcbdbfb3SJay Foad default: 548bcbdbfb3SJay Foad break; 549bcbdbfb3SJay Foad } 550bcbdbfb3SJay Foad return Result; 551bcbdbfb3SJay Foad } 5529de0d14dSChris Lattner 553868e3f09SDaniel Dunbar // Otherwise, if the value is a ConstantExpr... 5546c38f0bbSReid Spencer if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) { 5554fd528f2SReid Spencer Constant *Op0 = CE->getOperand(0); 5569de0d14dSChris Lattner switch (CE->getOpcode()) { 5579de0d14dSChris Lattner case Instruction::GetElementPtr: { 5586c38f0bbSReid Spencer // Compute the index 5594fd528f2SReid Spencer GenericValue Result = getConstantValue(Op0); 560c44bd78aSChris Lattner SmallVector<Value*, 8> Indices(CE->op_begin()+1, CE->op_end()); 561bf904773SJay Foad uint64_t Offset = TD->getIndexedOffset(Op0->getType(), Indices); 5629de0d14dSChris Lattner 56387aa65f4SReid Spencer char* tmp = (char*) Result.PointerVal; 56487aa65f4SReid Spencer Result = PTOGV(tmp + Offset); 5659de0d14dSChris Lattner return Result; 5669de0d14dSChris Lattner } 5674fd528f2SReid Spencer case Instruction::Trunc: { 5684fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 5694fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); 5704fd528f2SReid Spencer GV.IntVal = GV.IntVal.trunc(BitWidth); 5714fd528f2SReid Spencer return GV; 5724fd528f2SReid Spencer } 5734fd528f2SReid Spencer case Instruction::ZExt: { 5744fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 5754fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); 5764fd528f2SReid Spencer GV.IntVal = GV.IntVal.zext(BitWidth); 5774fd528f2SReid Spencer return GV; 5784fd528f2SReid Spencer } 5794fd528f2SReid Spencer case Instruction::SExt: { 5804fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 5814fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); 5824fd528f2SReid Spencer GV.IntVal = GV.IntVal.sext(BitWidth); 5834fd528f2SReid Spencer return GV; 5844fd528f2SReid Spencer } 5854fd528f2SReid Spencer case Instruction::FPTrunc: { 586a1336cf5SDale Johannesen // FIXME long double 5874fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 5884fd528f2SReid Spencer GV.FloatVal = float(GV.DoubleVal); 5894fd528f2SReid Spencer return GV; 5904fd528f2SReid Spencer } 5914fd528f2SReid Spencer case Instruction::FPExt:{ 592a1336cf5SDale Johannesen // FIXME long double 5934fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 5944fd528f2SReid Spencer GV.DoubleVal = double(GV.FloatVal); 5954fd528f2SReid Spencer return GV; 5964fd528f2SReid Spencer } 5974fd528f2SReid Spencer case Instruction::UIToFP: { 5984fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 599fdd87907SChris Lattner if (CE->getType()->isFloatTy()) 6004fd528f2SReid Spencer GV.FloatVal = float(GV.IntVal.roundToDouble()); 601fdd87907SChris Lattner else if (CE->getType()->isDoubleTy()) 6024fd528f2SReid Spencer GV.DoubleVal = GV.IntVal.roundToDouble(); 603fdd87907SChris Lattner else if (CE->getType()->isX86_FP80Ty()) { 60431920b0aSBenjamin Kramer APFloat apf = APFloat::getZero(APFloat::x87DoubleExtended); 605ca24fd90SDan Gohman (void)apf.convertFromAPInt(GV.IntVal, 606ca24fd90SDan Gohman false, 6079150652bSDale Johannesen APFloat::rmNearestTiesToEven); 60854306fe4SDale Johannesen GV.IntVal = apf.bitcastToAPInt(); 609a1336cf5SDale Johannesen } 6104fd528f2SReid Spencer return GV; 6114fd528f2SReid Spencer } 6124fd528f2SReid Spencer case Instruction::SIToFP: { 6134fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 614fdd87907SChris Lattner if (CE->getType()->isFloatTy()) 6154fd528f2SReid Spencer GV.FloatVal = float(GV.IntVal.signedRoundToDouble()); 616fdd87907SChris Lattner else if (CE->getType()->isDoubleTy()) 6174fd528f2SReid Spencer GV.DoubleVal = GV.IntVal.signedRoundToDouble(); 618fdd87907SChris Lattner else if (CE->getType()->isX86_FP80Ty()) { 61931920b0aSBenjamin Kramer APFloat apf = APFloat::getZero(APFloat::x87DoubleExtended); 620ca24fd90SDan Gohman (void)apf.convertFromAPInt(GV.IntVal, 621ca24fd90SDan Gohman true, 6229150652bSDale Johannesen APFloat::rmNearestTiesToEven); 62354306fe4SDale Johannesen GV.IntVal = apf.bitcastToAPInt(); 624a1336cf5SDale Johannesen } 6254fd528f2SReid Spencer return GV; 6264fd528f2SReid Spencer } 6274fd528f2SReid Spencer case Instruction::FPToUI: // double->APInt conversion handles sign 6284fd528f2SReid Spencer case Instruction::FPToSI: { 6294fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 6304fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); 631fdd87907SChris Lattner if (Op0->getType()->isFloatTy()) 6324fd528f2SReid Spencer GV.IntVal = APIntOps::RoundFloatToAPInt(GV.FloatVal, BitWidth); 633fdd87907SChris Lattner else if (Op0->getType()->isDoubleTy()) 6344fd528f2SReid Spencer GV.IntVal = APIntOps::RoundDoubleToAPInt(GV.DoubleVal, BitWidth); 635fdd87907SChris Lattner else if (Op0->getType()->isX86_FP80Ty()) { 636a1336cf5SDale Johannesen APFloat apf = APFloat(GV.IntVal); 637a1336cf5SDale Johannesen uint64_t v; 6384f0bd68cSDale Johannesen bool ignored; 639a1336cf5SDale Johannesen (void)apf.convertToInteger(&v, BitWidth, 640a1336cf5SDale Johannesen CE->getOpcode()==Instruction::FPToSI, 6414f0bd68cSDale Johannesen APFloat::rmTowardZero, &ignored); 642a1336cf5SDale Johannesen GV.IntVal = v; // endian? 643a1336cf5SDale Johannesen } 6444fd528f2SReid Spencer return GV; 6454fd528f2SReid Spencer } 6466c38f0bbSReid Spencer case Instruction::PtrToInt: { 6474fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 648*bf3eeb2dSMicah Villmow assert(CE->getOperand(1)->getType()->isPointerTy() && 649*bf3eeb2dSMicah Villmow "Must be a pointer type!"); 650*bf3eeb2dSMicah Villmow uint32_t PtrWidth = TD->getTypeSizeInBits(CE->getOperand(1)->getType()); 6514fd528f2SReid Spencer GV.IntVal = APInt(PtrWidth, uintptr_t(GV.PointerVal)); 6524fd528f2SReid Spencer return GV; 6534fd528f2SReid Spencer } 6544fd528f2SReid Spencer case Instruction::IntToPtr: { 6554fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 656*bf3eeb2dSMicah Villmow assert(CE->getOperand(1)->getType()->isPointerTy() && 657*bf3eeb2dSMicah Villmow "Must be a pointer type!"); 658*bf3eeb2dSMicah Villmow uint32_t PtrWidth = TD->getTypeSizeInBits(CE->getType()); 6594fd528f2SReid Spencer if (PtrWidth != GV.IntVal.getBitWidth()) 6604fd528f2SReid Spencer GV.IntVal = GV.IntVal.zextOrTrunc(PtrWidth); 6614fd528f2SReid Spencer assert(GV.IntVal.getBitWidth() <= 64 && "Bad pointer width"); 6624fd528f2SReid Spencer GV.PointerVal = PointerTy(uintptr_t(GV.IntVal.getZExtValue())); 6636c38f0bbSReid Spencer return GV; 6646c38f0bbSReid Spencer } 6656c38f0bbSReid Spencer case Instruction::BitCast: { 6664fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 667229907cdSChris Lattner Type* DestTy = CE->getType(); 6684fd528f2SReid Spencer switch (Op0->getType()->getTypeID()) { 669fbcc663cSTorok Edwin default: llvm_unreachable("Invalid bitcast operand"); 6704fd528f2SReid Spencer case Type::IntegerTyID: 6719dff9becSDuncan Sands assert(DestTy->isFloatingPointTy() && "invalid bitcast"); 672fdd87907SChris Lattner if (DestTy->isFloatTy()) 6734fd528f2SReid Spencer GV.FloatVal = GV.IntVal.bitsToFloat(); 674fdd87907SChris Lattner else if (DestTy->isDoubleTy()) 6754fd528f2SReid Spencer GV.DoubleVal = GV.IntVal.bitsToDouble(); 6766c38f0bbSReid Spencer break; 6774fd528f2SReid Spencer case Type::FloatTyID: 6789dff9becSDuncan Sands assert(DestTy->isIntegerTy(32) && "Invalid bitcast"); 6793447fb01SJay Foad GV.IntVal = APInt::floatToBits(GV.FloatVal); 6804fd528f2SReid Spencer break; 6814fd528f2SReid Spencer case Type::DoubleTyID: 6829dff9becSDuncan Sands assert(DestTy->isIntegerTy(64) && "Invalid bitcast"); 6833447fb01SJay Foad GV.IntVal = APInt::doubleToBits(GV.DoubleVal); 6844fd528f2SReid Spencer break; 6854fd528f2SReid Spencer case Type::PointerTyID: 68619d0b47bSDuncan Sands assert(DestTy->isPointerTy() && "Invalid bitcast"); 6874fd528f2SReid Spencer break; // getConstantValue(Op0) above already converted it 6886c38f0bbSReid Spencer } 6894fd528f2SReid Spencer return GV; 69068cbcc3eSChris Lattner } 69168cbcc3eSChris Lattner case Instruction::Add: 692a5b9645cSDan Gohman case Instruction::FAdd: 6934fd528f2SReid Spencer case Instruction::Sub: 694a5b9645cSDan Gohman case Instruction::FSub: 6954fd528f2SReid Spencer case Instruction::Mul: 696a5b9645cSDan Gohman case Instruction::FMul: 6974fd528f2SReid Spencer case Instruction::UDiv: 6984fd528f2SReid Spencer case Instruction::SDiv: 6994fd528f2SReid Spencer case Instruction::URem: 7004fd528f2SReid Spencer case Instruction::SRem: 7014fd528f2SReid Spencer case Instruction::And: 7024fd528f2SReid Spencer case Instruction::Or: 7034fd528f2SReid Spencer case Instruction::Xor: { 7044fd528f2SReid Spencer GenericValue LHS = getConstantValue(Op0); 7054fd528f2SReid Spencer GenericValue RHS = getConstantValue(CE->getOperand(1)); 7064fd528f2SReid Spencer GenericValue GV; 707c4e6bb5fSChris Lattner switch (CE->getOperand(0)->getType()->getTypeID()) { 708fbcc663cSTorok Edwin default: llvm_unreachable("Bad add type!"); 7097a9c62baSReid Spencer case Type::IntegerTyID: 7104fd528f2SReid Spencer switch (CE->getOpcode()) { 711fbcc663cSTorok Edwin default: llvm_unreachable("Invalid integer opcode"); 7124fd528f2SReid Spencer case Instruction::Add: GV.IntVal = LHS.IntVal + RHS.IntVal; break; 7134fd528f2SReid Spencer case Instruction::Sub: GV.IntVal = LHS.IntVal - RHS.IntVal; break; 7144fd528f2SReid Spencer case Instruction::Mul: GV.IntVal = LHS.IntVal * RHS.IntVal; break; 7154fd528f2SReid Spencer case Instruction::UDiv:GV.IntVal = LHS.IntVal.udiv(RHS.IntVal); break; 7164fd528f2SReid Spencer case Instruction::SDiv:GV.IntVal = LHS.IntVal.sdiv(RHS.IntVal); break; 7174fd528f2SReid Spencer case Instruction::URem:GV.IntVal = LHS.IntVal.urem(RHS.IntVal); break; 7184fd528f2SReid Spencer case Instruction::SRem:GV.IntVal = LHS.IntVal.srem(RHS.IntVal); break; 7194fd528f2SReid Spencer case Instruction::And: GV.IntVal = LHS.IntVal & RHS.IntVal; break; 7204fd528f2SReid Spencer case Instruction::Or: GV.IntVal = LHS.IntVal | RHS.IntVal; break; 7214fd528f2SReid Spencer case Instruction::Xor: GV.IntVal = LHS.IntVal ^ RHS.IntVal; break; 7224fd528f2SReid Spencer } 723c4e6bb5fSChris Lattner break; 724c4e6bb5fSChris Lattner case Type::FloatTyID: 7254fd528f2SReid Spencer switch (CE->getOpcode()) { 726fbcc663cSTorok Edwin default: llvm_unreachable("Invalid float opcode"); 727a5b9645cSDan Gohman case Instruction::FAdd: 7284fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal + RHS.FloatVal; break; 729a5b9645cSDan Gohman case Instruction::FSub: 7304fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal - RHS.FloatVal; break; 731a5b9645cSDan Gohman case Instruction::FMul: 7324fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal * RHS.FloatVal; break; 7334fd528f2SReid Spencer case Instruction::FDiv: 7344fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal / RHS.FloatVal; break; 7354fd528f2SReid Spencer case Instruction::FRem: 73693cd0f1cSChris Lattner GV.FloatVal = std::fmod(LHS.FloatVal,RHS.FloatVal); break; 7374fd528f2SReid Spencer } 738c4e6bb5fSChris Lattner break; 739c4e6bb5fSChris Lattner case Type::DoubleTyID: 7404fd528f2SReid Spencer switch (CE->getOpcode()) { 741fbcc663cSTorok Edwin default: llvm_unreachable("Invalid double opcode"); 742a5b9645cSDan Gohman case Instruction::FAdd: 7434fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal + RHS.DoubleVal; break; 744a5b9645cSDan Gohman case Instruction::FSub: 7454fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal - RHS.DoubleVal; break; 746a5b9645cSDan Gohman case Instruction::FMul: 7474fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal * RHS.DoubleVal; break; 7484fd528f2SReid Spencer case Instruction::FDiv: 7494fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal / RHS.DoubleVal; break; 7504fd528f2SReid Spencer case Instruction::FRem: 75193cd0f1cSChris Lattner GV.DoubleVal = std::fmod(LHS.DoubleVal,RHS.DoubleVal); break; 7524fd528f2SReid Spencer } 753c4e6bb5fSChris Lattner break; 754a1336cf5SDale Johannesen case Type::X86_FP80TyID: 755a1336cf5SDale Johannesen case Type::PPC_FP128TyID: 756a1336cf5SDale Johannesen case Type::FP128TyID: { 757a1336cf5SDale Johannesen APFloat apfLHS = APFloat(LHS.IntVal); 758a1336cf5SDale Johannesen switch (CE->getOpcode()) { 759e4f47434SDaniel Dunbar default: llvm_unreachable("Invalid long double opcode"); 760a5b9645cSDan Gohman case Instruction::FAdd: 761a1336cf5SDale Johannesen apfLHS.add(APFloat(RHS.IntVal), APFloat::rmNearestTiesToEven); 76254306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt(); 763a1336cf5SDale Johannesen break; 764a5b9645cSDan Gohman case Instruction::FSub: 765a1336cf5SDale Johannesen apfLHS.subtract(APFloat(RHS.IntVal), APFloat::rmNearestTiesToEven); 76654306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt(); 767a1336cf5SDale Johannesen break; 768a5b9645cSDan Gohman case Instruction::FMul: 769a1336cf5SDale Johannesen apfLHS.multiply(APFloat(RHS.IntVal), APFloat::rmNearestTiesToEven); 77054306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt(); 771a1336cf5SDale Johannesen break; 772a1336cf5SDale Johannesen case Instruction::FDiv: 773a1336cf5SDale Johannesen apfLHS.divide(APFloat(RHS.IntVal), APFloat::rmNearestTiesToEven); 77454306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt(); 775a1336cf5SDale Johannesen break; 776a1336cf5SDale Johannesen case Instruction::FRem: 777a1336cf5SDale Johannesen apfLHS.mod(APFloat(RHS.IntVal), APFloat::rmNearestTiesToEven); 77854306fe4SDale Johannesen GV.IntVal = apfLHS.bitcastToAPInt(); 779a1336cf5SDale Johannesen break; 780a1336cf5SDale Johannesen } 781a1336cf5SDale Johannesen } 782a1336cf5SDale Johannesen break; 783c4e6bb5fSChris Lattner } 7844fd528f2SReid Spencer return GV; 7854fd528f2SReid Spencer } 7869de0d14dSChris Lattner default: 78768cbcc3eSChris Lattner break; 78868cbcc3eSChris Lattner } 789868e3f09SDaniel Dunbar 790868e3f09SDaniel Dunbar SmallString<256> Msg; 791868e3f09SDaniel Dunbar raw_svector_ostream OS(Msg); 792868e3f09SDaniel Dunbar OS << "ConstantExpr not handled: " << *CE; 793868e3f09SDaniel Dunbar report_fatal_error(OS.str()); 7949de0d14dSChris Lattner } 795996fe010SChris Lattner 796868e3f09SDaniel Dunbar // Otherwise, we have a simple constant. 7974fd528f2SReid Spencer GenericValue Result; 7986b727599SChris Lattner switch (C->getType()->getTypeID()) { 79987aa65f4SReid Spencer case Type::FloatTyID: 800bed9dc42SDale Johannesen Result.FloatVal = cast<ConstantFP>(C)->getValueAPF().convertToFloat(); 8017a9c62baSReid Spencer break; 80287aa65f4SReid Spencer case Type::DoubleTyID: 803bed9dc42SDale Johannesen Result.DoubleVal = cast<ConstantFP>(C)->getValueAPF().convertToDouble(); 80487aa65f4SReid Spencer break; 805a1336cf5SDale Johannesen case Type::X86_FP80TyID: 806a1336cf5SDale Johannesen case Type::FP128TyID: 807a1336cf5SDale Johannesen case Type::PPC_FP128TyID: 80854306fe4SDale Johannesen Result.IntVal = cast <ConstantFP>(C)->getValueAPF().bitcastToAPInt(); 809a1336cf5SDale Johannesen break; 81087aa65f4SReid Spencer case Type::IntegerTyID: 81187aa65f4SReid Spencer Result.IntVal = cast<ConstantInt>(C)->getValue(); 81287aa65f4SReid Spencer break; 813996fe010SChris Lattner case Type::PointerTyID: 8146a0fd73bSReid Spencer if (isa<ConstantPointerNull>(C)) 815996fe010SChris Lattner Result.PointerVal = 0; 8166a0fd73bSReid Spencer else if (const Function *F = dyn_cast<Function>(C)) 8176a0fd73bSReid Spencer Result = PTOGV(getPointerToFunctionOrStub(const_cast<Function*>(F))); 8186a0fd73bSReid Spencer else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(C)) 8196a0fd73bSReid Spencer Result = PTOGV(getOrEmitGlobalVariable(const_cast<GlobalVariable*>(GV))); 8200c778f70SChris Lattner else if (const BlockAddress *BA = dyn_cast<BlockAddress>(C)) 8210c778f70SChris Lattner Result = PTOGV(getPointerToBasicBlock(const_cast<BasicBlock*>( 8220c778f70SChris Lattner BA->getBasicBlock()))); 823e6492f10SChris Lattner else 824fbcc663cSTorok Edwin llvm_unreachable("Unknown constant pointer type!"); 825996fe010SChris Lattner break; 826996fe010SChris Lattner default: 827868e3f09SDaniel Dunbar SmallString<256> Msg; 828868e3f09SDaniel Dunbar raw_svector_ostream OS(Msg); 829868e3f09SDaniel Dunbar OS << "ERROR: Constant unimplemented for type: " << *C->getType(); 830868e3f09SDaniel Dunbar report_fatal_error(OS.str()); 831996fe010SChris Lattner } 832868e3f09SDaniel Dunbar 833996fe010SChris Lattner return Result; 834996fe010SChris Lattner } 835996fe010SChris Lattner 8361202d1b1SDuncan Sands /// StoreIntToMemory - Fills the StoreBytes bytes of memory starting from Dst 8371202d1b1SDuncan Sands /// with the integer held in IntVal. 8381202d1b1SDuncan Sands static void StoreIntToMemory(const APInt &IntVal, uint8_t *Dst, 8391202d1b1SDuncan Sands unsigned StoreBytes) { 8401202d1b1SDuncan Sands assert((IntVal.getBitWidth()+7)/8 >= StoreBytes && "Integer too small!"); 841ad06cee2SRoman Divacky const uint8_t *Src = (const uint8_t *)IntVal.getRawData(); 8425c65cb46SDuncan Sands 843868e3f09SDaniel Dunbar if (sys::isLittleEndianHost()) { 8441202d1b1SDuncan Sands // Little-endian host - the source is ordered from LSB to MSB. Order the 8451202d1b1SDuncan Sands // destination from LSB to MSB: Do a straight copy. 8465c65cb46SDuncan Sands memcpy(Dst, Src, StoreBytes); 847868e3f09SDaniel Dunbar } else { 8485c65cb46SDuncan Sands // Big-endian host - the source is an array of 64 bit words ordered from 8491202d1b1SDuncan Sands // LSW to MSW. Each word is ordered from MSB to LSB. Order the destination 8501202d1b1SDuncan Sands // from MSB to LSB: Reverse the word order, but not the bytes in a word. 8515c65cb46SDuncan Sands while (StoreBytes > sizeof(uint64_t)) { 8525c65cb46SDuncan Sands StoreBytes -= sizeof(uint64_t); 8535c65cb46SDuncan Sands // May not be aligned so use memcpy. 8545c65cb46SDuncan Sands memcpy(Dst + StoreBytes, Src, sizeof(uint64_t)); 8555c65cb46SDuncan Sands Src += sizeof(uint64_t); 8565c65cb46SDuncan Sands } 8575c65cb46SDuncan Sands 8585c65cb46SDuncan Sands memcpy(Dst, Src + sizeof(uint64_t) - StoreBytes, StoreBytes); 859815f8dd2SReid Spencer } 8607a9c62baSReid Spencer } 8611202d1b1SDuncan Sands 86209053e62SEvan Cheng void ExecutionEngine::StoreValueToMemory(const GenericValue &Val, 863229907cdSChris Lattner GenericValue *Ptr, Type *Ty) { 864cdfe20b9SMicah Villmow const unsigned StoreBytes = getDataLayout()->getTypeStoreSize(Ty); 8651202d1b1SDuncan Sands 8661202d1b1SDuncan Sands switch (Ty->getTypeID()) { 8671202d1b1SDuncan Sands case Type::IntegerTyID: 8681202d1b1SDuncan Sands StoreIntToMemory(Val.IntVal, (uint8_t*)Ptr, StoreBytes); 8691202d1b1SDuncan Sands break; 870996fe010SChris Lattner case Type::FloatTyID: 87187aa65f4SReid Spencer *((float*)Ptr) = Val.FloatVal; 87287aa65f4SReid Spencer break; 87387aa65f4SReid Spencer case Type::DoubleTyID: 87487aa65f4SReid Spencer *((double*)Ptr) = Val.DoubleVal; 875996fe010SChris Lattner break; 8764d7e4ee7SDale Johannesen case Type::X86_FP80TyID: 8774d7e4ee7SDale Johannesen memcpy(Ptr, Val.IntVal.getRawData(), 10); 878a1336cf5SDale Johannesen break; 8797a9c62baSReid Spencer case Type::PointerTyID: 8801202d1b1SDuncan Sands // Ensure 64 bit target pointers are fully initialized on 32 bit hosts. 8811202d1b1SDuncan Sands if (StoreBytes != sizeof(PointerTy)) 88293da3c82SChandler Carruth memset(&(Ptr->PointerVal), 0, StoreBytes); 8831202d1b1SDuncan Sands 88487aa65f4SReid Spencer *((PointerTy*)Ptr) = Val.PointerVal; 885996fe010SChris Lattner break; 886996fe010SChris Lattner default: 8870967d2dfSDavid Greene dbgs() << "Cannot store value of type " << *Ty << "!\n"; 888996fe010SChris Lattner } 8891202d1b1SDuncan Sands 890cdfe20b9SMicah Villmow if (sys::isLittleEndianHost() != getDataLayout()->isLittleEndian()) 8911202d1b1SDuncan Sands // Host and target are different endian - reverse the stored bytes. 8921202d1b1SDuncan Sands std::reverse((uint8_t*)Ptr, StoreBytes + (uint8_t*)Ptr); 893996fe010SChris Lattner } 894996fe010SChris Lattner 8951202d1b1SDuncan Sands /// LoadIntFromMemory - Loads the integer stored in the LoadBytes bytes starting 8961202d1b1SDuncan Sands /// from Src into IntVal, which is assumed to be wide enough and to hold zero. 8971202d1b1SDuncan Sands static void LoadIntFromMemory(APInt &IntVal, uint8_t *Src, unsigned LoadBytes) { 8981202d1b1SDuncan Sands assert((IntVal.getBitWidth()+7)/8 >= LoadBytes && "Integer too small!"); 8991202d1b1SDuncan Sands uint8_t *Dst = (uint8_t *)IntVal.getRawData(); 9005c65cb46SDuncan Sands 901aa121227SChris Lattner if (sys::isLittleEndianHost()) 9025c65cb46SDuncan Sands // Little-endian host - the destination must be ordered from LSB to MSB. 9035c65cb46SDuncan Sands // The source is ordered from LSB to MSB: Do a straight copy. 9045c65cb46SDuncan Sands memcpy(Dst, Src, LoadBytes); 9055c65cb46SDuncan Sands else { 9065c65cb46SDuncan Sands // Big-endian - the destination is an array of 64 bit words ordered from 9075c65cb46SDuncan Sands // LSW to MSW. Each word must be ordered from MSB to LSB. The source is 9085c65cb46SDuncan Sands // ordered from MSB to LSB: Reverse the word order, but not the bytes in 9095c65cb46SDuncan Sands // a word. 9105c65cb46SDuncan Sands while (LoadBytes > sizeof(uint64_t)) { 9115c65cb46SDuncan Sands LoadBytes -= sizeof(uint64_t); 9125c65cb46SDuncan Sands // May not be aligned so use memcpy. 9135c65cb46SDuncan Sands memcpy(Dst, Src + LoadBytes, sizeof(uint64_t)); 9145c65cb46SDuncan Sands Dst += sizeof(uint64_t); 9155c65cb46SDuncan Sands } 9165c65cb46SDuncan Sands 9175c65cb46SDuncan Sands memcpy(Dst + sizeof(uint64_t) - LoadBytes, Src, LoadBytes); 9185c65cb46SDuncan Sands } 9197a9c62baSReid Spencer } 9201202d1b1SDuncan Sands 9211202d1b1SDuncan Sands /// FIXME: document 9221202d1b1SDuncan Sands /// 9231202d1b1SDuncan Sands void ExecutionEngine::LoadValueFromMemory(GenericValue &Result, 9241202d1b1SDuncan Sands GenericValue *Ptr, 925229907cdSChris Lattner Type *Ty) { 926cdfe20b9SMicah Villmow const unsigned LoadBytes = getDataLayout()->getTypeStoreSize(Ty); 9271202d1b1SDuncan Sands 9281202d1b1SDuncan Sands switch (Ty->getTypeID()) { 9291202d1b1SDuncan Sands case Type::IntegerTyID: 9301202d1b1SDuncan Sands // An APInt with all words initially zero. 9311202d1b1SDuncan Sands Result.IntVal = APInt(cast<IntegerType>(Ty)->getBitWidth(), 0); 9321202d1b1SDuncan Sands LoadIntFromMemory(Result.IntVal, (uint8_t*)Ptr, LoadBytes); 9331202d1b1SDuncan Sands break; 9347f389e8cSChris Lattner case Type::FloatTyID: 93587aa65f4SReid Spencer Result.FloatVal = *((float*)Ptr); 93687aa65f4SReid Spencer break; 93787aa65f4SReid Spencer case Type::DoubleTyID: 93887aa65f4SReid Spencer Result.DoubleVal = *((double*)Ptr); 9397f389e8cSChris Lattner break; 9407a9c62baSReid Spencer case Type::PointerTyID: 94187aa65f4SReid Spencer Result.PointerVal = *((PointerTy*)Ptr); 9427f389e8cSChris Lattner break; 943a1336cf5SDale Johannesen case Type::X86_FP80TyID: { 944a1336cf5SDale Johannesen // This is endian dependent, but it will only work on x86 anyway. 94526d6539eSDuncan Sands // FIXME: Will not trap if loading a signaling NaN. 946ff306287SDuncan Sands uint64_t y[2]; 9474d7e4ee7SDale Johannesen memcpy(y, Ptr, 10); 9487a162881SJeffrey Yasskin Result.IntVal = APInt(80, y); 949a1336cf5SDale Johannesen break; 950a1336cf5SDale Johannesen } 9517f389e8cSChris Lattner default: 952868e3f09SDaniel Dunbar SmallString<256> Msg; 953868e3f09SDaniel Dunbar raw_svector_ostream OS(Msg); 954868e3f09SDaniel Dunbar OS << "Cannot load value of type " << *Ty << "!"; 955868e3f09SDaniel Dunbar report_fatal_error(OS.str()); 9567f389e8cSChris Lattner } 9577f389e8cSChris Lattner } 9587f389e8cSChris Lattner 959996fe010SChris Lattner void ExecutionEngine::InitializeMemory(const Constant *Init, void *Addr) { 9600967d2dfSDavid Greene DEBUG(dbgs() << "JIT: Initializing " << Addr << " "); 961b086d382SDale Johannesen DEBUG(Init->dump()); 96200245f42SChris Lattner if (isa<UndefValue>(Init)) 96361753bf8SChris Lattner return; 96400245f42SChris Lattner 96500245f42SChris Lattner if (const ConstantVector *CP = dyn_cast<ConstantVector>(Init)) { 96669d62138SRobert Bocchino unsigned ElementSize = 967cdfe20b9SMicah Villmow getDataLayout()->getTypeAllocSize(CP->getType()->getElementType()); 96869d62138SRobert Bocchino for (unsigned i = 0, e = CP->getNumOperands(); i != e; ++i) 96969d62138SRobert Bocchino InitializeMemory(CP->getOperand(i), (char*)Addr+i*ElementSize); 97069d62138SRobert Bocchino return; 97100245f42SChris Lattner } 97200245f42SChris Lattner 97300245f42SChris Lattner if (isa<ConstantAggregateZero>(Init)) { 974cdfe20b9SMicah Villmow memset(Addr, 0, (size_t)getDataLayout()->getTypeAllocSize(Init->getType())); 9751dd86b11SChris Lattner return; 97600245f42SChris Lattner } 97700245f42SChris Lattner 97800245f42SChris Lattner if (const ConstantArray *CPA = dyn_cast<ConstantArray>(Init)) { 97969ddfbfeSDan Gohman unsigned ElementSize = 980cdfe20b9SMicah Villmow getDataLayout()->getTypeAllocSize(CPA->getType()->getElementType()); 98169ddfbfeSDan Gohman for (unsigned i = 0, e = CPA->getNumOperands(); i != e; ++i) 98269ddfbfeSDan Gohman InitializeMemory(CPA->getOperand(i), (char*)Addr+i*ElementSize); 98369ddfbfeSDan Gohman return; 98400245f42SChris Lattner } 98500245f42SChris Lattner 98600245f42SChris Lattner if (const ConstantStruct *CPS = dyn_cast<ConstantStruct>(Init)) { 98769ddfbfeSDan Gohman const StructLayout *SL = 988cdfe20b9SMicah Villmow getDataLayout()->getStructLayout(cast<StructType>(CPS->getType())); 98969ddfbfeSDan Gohman for (unsigned i = 0, e = CPS->getNumOperands(); i != e; ++i) 99069ddfbfeSDan Gohman InitializeMemory(CPS->getOperand(i), (char*)Addr+SL->getElementOffset(i)); 99169ddfbfeSDan Gohman return; 99200245f42SChris Lattner } 99300245f42SChris Lattner 99400245f42SChris Lattner if (const ConstantDataSequential *CDS = 99500245f42SChris Lattner dyn_cast<ConstantDataSequential>(Init)) { 99600245f42SChris Lattner // CDS is already laid out in host memory order. 99700245f42SChris Lattner StringRef Data = CDS->getRawDataValues(); 99800245f42SChris Lattner memcpy(Addr, Data.data(), Data.size()); 99900245f42SChris Lattner return; 100000245f42SChris Lattner } 100100245f42SChris Lattner 100200245f42SChris Lattner if (Init->getType()->isFirstClassType()) { 1003996fe010SChris Lattner GenericValue Val = getConstantValue(Init); 1004996fe010SChris Lattner StoreValueToMemory(Val, (GenericValue*)Addr, Init->getType()); 1005996fe010SChris Lattner return; 1006996fe010SChris Lattner } 1007996fe010SChris Lattner 1008868e3f09SDaniel Dunbar DEBUG(dbgs() << "Bad Type: " << *Init->getType() << "\n"); 1009fbcc663cSTorok Edwin llvm_unreachable("Unknown constant type to initialize memory with!"); 1010996fe010SChris Lattner } 1011996fe010SChris Lattner 1012996fe010SChris Lattner /// EmitGlobals - Emit all of the global variables to memory, storing their 1013996fe010SChris Lattner /// addresses into GlobalAddress. This must make sure to copy the contents of 1014996fe010SChris Lattner /// their initializers into the memory. 1015996fe010SChris Lattner void ExecutionEngine::emitGlobals() { 1016996fe010SChris Lattner // Loop over all of the global variables in the program, allocating the memory 10170621caefSChris Lattner // to hold them. If there is more than one module, do a prepass over globals 10180621caefSChris Lattner // to figure out how the different modules should link together. 1019229907cdSChris Lattner std::map<std::pair<std::string, Type*>, 10200621caefSChris Lattner const GlobalValue*> LinkedGlobalsMap; 10210621caefSChris Lattner 10220621caefSChris Lattner if (Modules.size() != 1) { 10230621caefSChris Lattner for (unsigned m = 0, e = Modules.size(); m != e; ++m) { 1024091217beSJeffrey Yasskin Module &M = *Modules[m]; 10250621caefSChris Lattner for (Module::const_global_iterator I = M.global_begin(), 10260621caefSChris Lattner E = M.global_end(); I != E; ++I) { 10270621caefSChris Lattner const GlobalValue *GV = I; 10286de96a1bSRafael Espindola if (GV->hasLocalLinkage() || GV->isDeclaration() || 10290621caefSChris Lattner GV->hasAppendingLinkage() || !GV->hasName()) 10300621caefSChris Lattner continue;// Ignore external globals and globals with internal linkage. 10310621caefSChris Lattner 10320621caefSChris Lattner const GlobalValue *&GVEntry = 10330621caefSChris Lattner LinkedGlobalsMap[std::make_pair(GV->getName(), GV->getType())]; 10340621caefSChris Lattner 10350621caefSChris Lattner // If this is the first time we've seen this global, it is the canonical 10360621caefSChris Lattner // version. 10370621caefSChris Lattner if (!GVEntry) { 10380621caefSChris Lattner GVEntry = GV; 10390621caefSChris Lattner continue; 10400621caefSChris Lattner } 10410621caefSChris Lattner 10420621caefSChris Lattner // If the existing global is strong, never replace it. 1043d61d39ecSAnton Korobeynikov if (GVEntry->hasExternalLinkage() || 1044d61d39ecSAnton Korobeynikov GVEntry->hasDLLImportLinkage() || 1045d61d39ecSAnton Korobeynikov GVEntry->hasDLLExportLinkage()) 10460621caefSChris Lattner continue; 10470621caefSChris Lattner 10480621caefSChris Lattner // Otherwise, we know it's linkonce/weak, replace it if this is a strong 1049ce4396bcSDale Johannesen // symbol. FIXME is this right for common? 105012c94949SAnton Korobeynikov if (GV->hasExternalLinkage() || GVEntry->hasExternalWeakLinkage()) 10510621caefSChris Lattner GVEntry = GV; 10520621caefSChris Lattner } 10530621caefSChris Lattner } 10540621caefSChris Lattner } 10550621caefSChris Lattner 10560621caefSChris Lattner std::vector<const GlobalValue*> NonCanonicalGlobals; 10570621caefSChris Lattner for (unsigned m = 0, e = Modules.size(); m != e; ++m) { 1058091217beSJeffrey Yasskin Module &M = *Modules[m]; 10598ffb6611SChris Lattner for (Module::const_global_iterator I = M.global_begin(), E = M.global_end(); 10600621caefSChris Lattner I != E; ++I) { 10610621caefSChris Lattner // In the multi-module case, see what this global maps to. 10620621caefSChris Lattner if (!LinkedGlobalsMap.empty()) { 10630621caefSChris Lattner if (const GlobalValue *GVEntry = 10640621caefSChris Lattner LinkedGlobalsMap[std::make_pair(I->getName(), I->getType())]) { 10650621caefSChris Lattner // If something else is the canonical global, ignore this one. 10660621caefSChris Lattner if (GVEntry != &*I) { 10670621caefSChris Lattner NonCanonicalGlobals.push_back(I); 10680621caefSChris Lattner continue; 10690621caefSChris Lattner } 10700621caefSChris Lattner } 10710621caefSChris Lattner } 10720621caefSChris Lattner 10735301e7c6SReid Spencer if (!I->isDeclaration()) { 10745457ce9aSNicolas Geoffray addGlobalMapping(I, getMemoryForGV(I)); 1075996fe010SChris Lattner } else { 1076e8bbcfc2SBrian Gaeke // External variable reference. Try to use the dynamic loader to 1077e8bbcfc2SBrian Gaeke // get a pointer to it. 10780621caefSChris Lattner if (void *SymAddr = 10795899e340SDaniel Dunbar sys::DynamicLibrary::SearchForAddressOfSymbol(I->getName())) 1080748e8579SChris Lattner addGlobalMapping(I, SymAddr); 10819de0d14dSChris Lattner else { 10822104b8d3SChris Lattner report_fatal_error("Could not resolve external global address: " 10836c2d233eSTorok Edwin +I->getName()); 10849de0d14dSChris Lattner } 1085996fe010SChris Lattner } 10860621caefSChris Lattner } 10870621caefSChris Lattner 10880621caefSChris Lattner // If there are multiple modules, map the non-canonical globals to their 10890621caefSChris Lattner // canonical location. 10900621caefSChris Lattner if (!NonCanonicalGlobals.empty()) { 10910621caefSChris Lattner for (unsigned i = 0, e = NonCanonicalGlobals.size(); i != e; ++i) { 10920621caefSChris Lattner const GlobalValue *GV = NonCanonicalGlobals[i]; 10930621caefSChris Lattner const GlobalValue *CGV = 10940621caefSChris Lattner LinkedGlobalsMap[std::make_pair(GV->getName(), GV->getType())]; 10950621caefSChris Lattner void *Ptr = getPointerToGlobalIfAvailable(CGV); 10960621caefSChris Lattner assert(Ptr && "Canonical global wasn't codegen'd!"); 1097a67f06b9SNuno Lopes addGlobalMapping(GV, Ptr); 10980621caefSChris Lattner } 10990621caefSChris Lattner } 1100996fe010SChris Lattner 11017a9c62baSReid Spencer // Now that all of the globals are set up in memory, loop through them all 11027a9c62baSReid Spencer // and initialize their contents. 11038ffb6611SChris Lattner for (Module::const_global_iterator I = M.global_begin(), E = M.global_end(); 11040621caefSChris Lattner I != E; ++I) { 11055301e7c6SReid Spencer if (!I->isDeclaration()) { 11060621caefSChris Lattner if (!LinkedGlobalsMap.empty()) { 11070621caefSChris Lattner if (const GlobalValue *GVEntry = 11080621caefSChris Lattner LinkedGlobalsMap[std::make_pair(I->getName(), I->getType())]) 11090621caefSChris Lattner if (GVEntry != &*I) // Not the canonical variable. 11100621caefSChris Lattner continue; 11110621caefSChris Lattner } 11126bbe3eceSChris Lattner EmitGlobalVariable(I); 11136bbe3eceSChris Lattner } 11140621caefSChris Lattner } 11150621caefSChris Lattner } 11160621caefSChris Lattner } 11176bbe3eceSChris Lattner 11186bbe3eceSChris Lattner // EmitGlobalVariable - This method emits the specified global variable to the 11196bbe3eceSChris Lattner // address specified in GlobalAddresses, or allocates new memory if it's not 11206bbe3eceSChris Lattner // already in the map. 1121fbcc0aa1SChris Lattner void ExecutionEngine::EmitGlobalVariable(const GlobalVariable *GV) { 1122748e8579SChris Lattner void *GA = getPointerToGlobalIfAvailable(GV); 1123dc631735SChris Lattner 11246bbe3eceSChris Lattner if (GA == 0) { 11256bbe3eceSChris Lattner // If it's not already specified, allocate memory for the global. 11265457ce9aSNicolas Geoffray GA = getMemoryForGV(GV); 1127748e8579SChris Lattner addGlobalMapping(GV, GA); 11286bbe3eceSChris Lattner } 1129fbcc0aa1SChris Lattner 11305457ce9aSNicolas Geoffray // Don't initialize if it's thread local, let the client do it. 11315457ce9aSNicolas Geoffray if (!GV->isThreadLocal()) 11326bbe3eceSChris Lattner InitializeMemory(GV->getInitializer(), GA); 11335457ce9aSNicolas Geoffray 1134229907cdSChris Lattner Type *ElTy = GV->getType()->getElementType(); 1135cdfe20b9SMicah Villmow size_t GVSize = (size_t)getDataLayout()->getTypeAllocSize(ElTy); 1136df1f1524SChris Lattner NumInitBytes += (unsigned)GVSize; 11376bbe3eceSChris Lattner ++NumGlobals; 1138996fe010SChris Lattner } 1139f98e981cSJeffrey Yasskin 1140d0fc8f80SJeffrey Yasskin ExecutionEngineState::ExecutionEngineState(ExecutionEngine &EE) 1141d0fc8f80SJeffrey Yasskin : EE(EE), GlobalAddressMap(this) { 1142f98e981cSJeffrey Yasskin } 1143f98e981cSJeffrey Yasskin 1144868e3f09SDaniel Dunbar sys::Mutex * 1145868e3f09SDaniel Dunbar ExecutionEngineState::AddressMapConfig::getMutex(ExecutionEngineState *EES) { 1146d0fc8f80SJeffrey Yasskin return &EES->EE.lock; 1147d0fc8f80SJeffrey Yasskin } 1148868e3f09SDaniel Dunbar 1149868e3f09SDaniel Dunbar void ExecutionEngineState::AddressMapConfig::onDelete(ExecutionEngineState *EES, 1150868e3f09SDaniel Dunbar const GlobalValue *Old) { 1151d0fc8f80SJeffrey Yasskin void *OldVal = EES->GlobalAddressMap.lookup(Old); 1152d0fc8f80SJeffrey Yasskin EES->GlobalAddressReverseMap.erase(OldVal); 1153d0fc8f80SJeffrey Yasskin } 1154d0fc8f80SJeffrey Yasskin 1155868e3f09SDaniel Dunbar void ExecutionEngineState::AddressMapConfig::onRAUW(ExecutionEngineState *, 1156868e3f09SDaniel Dunbar const GlobalValue *, 1157868e3f09SDaniel Dunbar const GlobalValue *) { 1158a2886c21SCraig Topper llvm_unreachable("The ExecutionEngine doesn't know how to handle a" 1159f98e981cSJeffrey Yasskin " RAUW on a value it has a global mapping for."); 1160f98e981cSJeffrey Yasskin } 1161