1857c21b4SMisha Brukman //===-- ExecutionEngine.cpp - Common Implementation shared by EEs ---------===// 2996fe010SChris Lattner // 3482202a6SJohn Criswell // The LLVM Compiler Infrastructure 4482202a6SJohn Criswell // 5482202a6SJohn Criswell // This file was developed by the LLVM research group and is distributed under 6482202a6SJohn Criswell // the University of Illinois Open Source 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" 16996fe010SChris Lattner #include "llvm/Constants.h" 17260b0c88SMisha Brukman #include "llvm/DerivedTypes.h" 18996fe010SChris Lattner #include "llvm/Module.h" 19260b0c88SMisha Brukman #include "llvm/ModuleProvider.h" 2070e37278SReid Spencer #include "llvm/ADT/Statistic.h" 21260b0c88SMisha Brukman #include "llvm/ExecutionEngine/ExecutionEngine.h" 22ad481312SChris Lattner #include "llvm/ExecutionEngine/GenericValue.h" 237c16caa3SReid Spencer #include "llvm/Support/Debug.h" 246d8dd189SChris Lattner #include "llvm/Support/MutexGuard.h" 2570e37278SReid Spencer #include "llvm/System/DynamicLibrary.h" 2670e37278SReid Spencer #include "llvm/Target/TargetData.h" 2729681deeSChris Lattner using namespace llvm; 28996fe010SChris Lattner 29c346ecd7SChris Lattner STATISTIC(NumInitBytes, "Number of bytes of global vars initialized"); 30c346ecd7SChris Lattner STATISTIC(NumGlobals , "Number of global vars initialized"); 31996fe010SChris Lattner 322d52c1b8SChris Lattner ExecutionEngine::EECtorFn ExecutionEngine::JITCtor = 0; 332d52c1b8SChris Lattner ExecutionEngine::EECtorFn ExecutionEngine::InterpCtor = 0; 342d52c1b8SChris Lattner 350621caefSChris Lattner ExecutionEngine::ExecutionEngine(ModuleProvider *P) { 3687aee74cSChris Lattner LazyCompilationDisabled = false; 370621caefSChris Lattner Modules.push_back(P); 38260b0c88SMisha Brukman assert(P && "ModuleProvider is null?"); 39260b0c88SMisha Brukman } 40260b0c88SMisha Brukman 410621caefSChris Lattner ExecutionEngine::ExecutionEngine(Module *M) { 4287aee74cSChris Lattner LazyCompilationDisabled = false; 43260b0c88SMisha Brukman assert(M && "Module is null?"); 440621caefSChris Lattner Modules.push_back(new ExistingModuleProvider(M)); 45260b0c88SMisha Brukman } 46260b0c88SMisha Brukman 4792f8b30dSBrian Gaeke ExecutionEngine::~ExecutionEngine() { 48603682adSReid Spencer clearAllGlobalMappings(); 490621caefSChris Lattner for (unsigned i = 0, e = Modules.size(); i != e; ++i) 500621caefSChris Lattner delete Modules[i]; 5192f8b30dSBrian Gaeke } 5292f8b30dSBrian Gaeke 530621caefSChris Lattner /// FindFunctionNamed - Search all of the active modules to find the one that 540621caefSChris Lattner /// defines FnName. This is very slow operation and shouldn't be used for 550621caefSChris Lattner /// general code. 560621caefSChris Lattner Function *ExecutionEngine::FindFunctionNamed(const char *FnName) { 570621caefSChris Lattner for (unsigned i = 0, e = Modules.size(); i != e; ++i) { 581241d6d5SReid Spencer if (Function *F = Modules[i]->getModule()->getFunction(FnName)) 590621caefSChris Lattner return F; 600621caefSChris Lattner } 610621caefSChris Lattner return 0; 620621caefSChris Lattner } 630621caefSChris Lattner 640621caefSChris Lattner 656d8dd189SChris Lattner /// addGlobalMapping - Tell the execution engine that the specified global is 666d8dd189SChris Lattner /// at the specified location. This is used internally as functions are JIT'd 676d8dd189SChris Lattner /// and as global variables are laid out in memory. It can and should also be 686d8dd189SChris Lattner /// used by clients of the EE that want to have an LLVM global overlay 696d8dd189SChris Lattner /// existing data in memory. 706d8dd189SChris Lattner void ExecutionEngine::addGlobalMapping(const GlobalValue *GV, void *Addr) { 716d8dd189SChris Lattner MutexGuard locked(lock); 726d8dd189SChris Lattner 736d8dd189SChris Lattner void *&CurVal = state.getGlobalAddressMap(locked)[GV]; 746d8dd189SChris Lattner assert((CurVal == 0 || Addr == 0) && "GlobalMapping already established!"); 756d8dd189SChris Lattner CurVal = Addr; 766d8dd189SChris Lattner 776d8dd189SChris Lattner // If we are using the reverse mapping, add it too 786d8dd189SChris Lattner if (!state.getGlobalAddressReverseMap(locked).empty()) { 796d8dd189SChris Lattner const GlobalValue *&V = state.getGlobalAddressReverseMap(locked)[Addr]; 806d8dd189SChris Lattner assert((V == 0 || GV == 0) && "GlobalMapping already established!"); 816d8dd189SChris Lattner V = GV; 826d8dd189SChris Lattner } 836d8dd189SChris Lattner } 846d8dd189SChris Lattner 856d8dd189SChris Lattner /// clearAllGlobalMappings - Clear all global mappings and start over again 866d8dd189SChris Lattner /// use in dynamic compilation scenarios when you want to move globals 876d8dd189SChris Lattner void ExecutionEngine::clearAllGlobalMappings() { 886d8dd189SChris Lattner MutexGuard locked(lock); 896d8dd189SChris Lattner 906d8dd189SChris Lattner state.getGlobalAddressMap(locked).clear(); 916d8dd189SChris Lattner state.getGlobalAddressReverseMap(locked).clear(); 926d8dd189SChris Lattner } 936d8dd189SChris Lattner 946d8dd189SChris Lattner /// updateGlobalMapping - Replace an existing mapping for GV with a new 956d8dd189SChris Lattner /// address. This updates both maps as required. If "Addr" is null, the 966d8dd189SChris Lattner /// entry for the global is removed from the mappings. 976d8dd189SChris Lattner void ExecutionEngine::updateGlobalMapping(const GlobalValue *GV, void *Addr) { 986d8dd189SChris Lattner MutexGuard locked(lock); 996d8dd189SChris Lattner 1006d8dd189SChris Lattner // Deleting from the mapping? 1016d8dd189SChris Lattner if (Addr == 0) { 1026d8dd189SChris Lattner state.getGlobalAddressMap(locked).erase(GV); 1036d8dd189SChris Lattner if (!state.getGlobalAddressReverseMap(locked).empty()) 1046d8dd189SChris Lattner state.getGlobalAddressReverseMap(locked).erase(Addr); 1056d8dd189SChris Lattner return; 1066d8dd189SChris Lattner } 1076d8dd189SChris Lattner 1086d8dd189SChris Lattner void *&CurVal = state.getGlobalAddressMap(locked)[GV]; 1096d8dd189SChris Lattner if (CurVal && !state.getGlobalAddressReverseMap(locked).empty()) 1106d8dd189SChris Lattner state.getGlobalAddressReverseMap(locked).erase(CurVal); 1116d8dd189SChris Lattner CurVal = Addr; 1126d8dd189SChris Lattner 1136d8dd189SChris Lattner // If we are using the reverse mapping, add it too 1146d8dd189SChris Lattner if (!state.getGlobalAddressReverseMap(locked).empty()) { 1156d8dd189SChris Lattner const GlobalValue *&V = state.getGlobalAddressReverseMap(locked)[Addr]; 1166d8dd189SChris Lattner assert((V == 0 || GV == 0) && "GlobalMapping already established!"); 1176d8dd189SChris Lattner V = GV; 1186d8dd189SChris Lattner } 1196d8dd189SChris Lattner } 1206d8dd189SChris Lattner 1216d8dd189SChris Lattner /// getPointerToGlobalIfAvailable - This returns the address of the specified 1226d8dd189SChris Lattner /// global value if it is has already been codegen'd, otherwise it returns null. 1236d8dd189SChris Lattner /// 1246d8dd189SChris Lattner void *ExecutionEngine::getPointerToGlobalIfAvailable(const GlobalValue *GV) { 1256d8dd189SChris Lattner MutexGuard locked(lock); 1266d8dd189SChris Lattner 1276d8dd189SChris Lattner std::map<const GlobalValue*, void*>::iterator I = 1286d8dd189SChris Lattner state.getGlobalAddressMap(locked).find(GV); 1296d8dd189SChris Lattner return I != state.getGlobalAddressMap(locked).end() ? I->second : 0; 1306d8dd189SChris Lattner } 1316d8dd189SChris Lattner 132748e8579SChris Lattner /// getGlobalValueAtAddress - Return the LLVM global value object that starts 133748e8579SChris Lattner /// at the specified address. 134748e8579SChris Lattner /// 135748e8579SChris Lattner const GlobalValue *ExecutionEngine::getGlobalValueAtAddress(void *Addr) { 13679876f52SReid Spencer MutexGuard locked(lock); 13779876f52SReid Spencer 138748e8579SChris Lattner // If we haven't computed the reverse mapping yet, do so first. 13979876f52SReid Spencer if (state.getGlobalAddressReverseMap(locked).empty()) { 1406d8dd189SChris Lattner for (std::map<const GlobalValue*, void *>::iterator 1416d8dd189SChris Lattner I = state.getGlobalAddressMap(locked).begin(), 1426d8dd189SChris Lattner E = state.getGlobalAddressMap(locked).end(); I != E; ++I) 1436d8dd189SChris Lattner state.getGlobalAddressReverseMap(locked).insert(std::make_pair(I->second, 1446d8dd189SChris Lattner I->first)); 145748e8579SChris Lattner } 146748e8579SChris Lattner 147748e8579SChris Lattner std::map<void *, const GlobalValue*>::iterator I = 14879876f52SReid Spencer state.getGlobalAddressReverseMap(locked).find(Addr); 14979876f52SReid Spencer return I != state.getGlobalAddressReverseMap(locked).end() ? I->second : 0; 150748e8579SChris Lattner } 1515a0d4829SChris Lattner 1525a0d4829SChris Lattner // CreateArgv - Turn a vector of strings into a nice argv style array of 1535a0d4829SChris Lattner // pointers to null terminated strings. 1545a0d4829SChris Lattner // 1555a0d4829SChris Lattner static void *CreateArgv(ExecutionEngine *EE, 1565a0d4829SChris Lattner const std::vector<std::string> &InputArgv) { 15720a631fdSOwen Anderson unsigned PtrSize = EE->getTargetData()->getPointerSize(); 1585a0d4829SChris Lattner char *Result = new char[(InputArgv.size()+1)*PtrSize]; 1595a0d4829SChris Lattner 1605834fdb3SBill Wendling DOUT << "ARGV = " << (void*)Result << "\n"; 1610d54e78aSReid Spencer const Type *SBytePtr = PointerType::get(Type::Int8Ty); 1625a0d4829SChris Lattner 1635a0d4829SChris Lattner for (unsigned i = 0; i != InputArgv.size(); ++i) { 1645a0d4829SChris Lattner unsigned Size = InputArgv[i].size()+1; 1655a0d4829SChris Lattner char *Dest = new char[Size]; 1665834fdb3SBill Wendling DOUT << "ARGV[" << i << "] = " << (void*)Dest << "\n"; 1675a0d4829SChris Lattner 1685a0d4829SChris Lattner std::copy(InputArgv[i].begin(), InputArgv[i].end(), Dest); 1695a0d4829SChris Lattner Dest[Size-1] = 0; 1705a0d4829SChris Lattner 1715a0d4829SChris Lattner // Endian safe: Result[i] = (PointerTy)Dest; 1725a0d4829SChris Lattner EE->StoreValueToMemory(PTOGV(Dest), (GenericValue*)(Result+i*PtrSize), 1735a0d4829SChris Lattner SBytePtr); 1745a0d4829SChris Lattner } 1755a0d4829SChris Lattner 1765a0d4829SChris Lattner // Null terminate it 1775a0d4829SChris Lattner EE->StoreValueToMemory(PTOGV(0), 1785a0d4829SChris Lattner (GenericValue*)(Result+InputArgv.size()*PtrSize), 1795a0d4829SChris Lattner SBytePtr); 1805a0d4829SChris Lattner return Result; 1815a0d4829SChris Lattner } 1825a0d4829SChris Lattner 183faae50b6SChris Lattner 184faae50b6SChris Lattner /// runStaticConstructorsDestructors - This method is used to execute all of 1850621caefSChris Lattner /// the static constructors or destructors for a program, depending on the 186faae50b6SChris Lattner /// value of isDtors. 187faae50b6SChris Lattner void ExecutionEngine::runStaticConstructorsDestructors(bool isDtors) { 188faae50b6SChris Lattner const char *Name = isDtors ? "llvm.global_dtors" : "llvm.global_ctors"; 1890621caefSChris Lattner 1900621caefSChris Lattner // Execute global ctors/dtors for each module in the program. 1910621caefSChris Lattner for (unsigned m = 0, e = Modules.size(); m != e; ++m) { 1920621caefSChris Lattner GlobalVariable *GV = Modules[m]->getModule()->getNamedGlobal(Name); 193fe36eaebSChris Lattner 194fe36eaebSChris Lattner // If this global has internal linkage, or if it has a use, then it must be 195fe36eaebSChris Lattner // an old-style (llvmgcc3) static ctor with __main linked in and in use. If 1960621caefSChris Lattner // this is the case, don't execute any of the global ctors, __main will do 1970621caefSChris Lattner // it. 1985301e7c6SReid Spencer if (!GV || GV->isDeclaration() || GV->hasInternalLinkage()) continue; 199faae50b6SChris Lattner 2000621caefSChris Lattner // Should be an array of '{ int, void ()* }' structs. The first value is 2010621caefSChris Lattner // the init priority, which we ignore. 202faae50b6SChris Lattner ConstantArray *InitList = dyn_cast<ConstantArray>(GV->getInitializer()); 2030621caefSChris Lattner if (!InitList) continue; 204faae50b6SChris Lattner for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i) 2050621caefSChris Lattner if (ConstantStruct *CS = 2060621caefSChris Lattner dyn_cast<ConstantStruct>(InitList->getOperand(i))) { 2070621caefSChris Lattner if (CS->getNumOperands() != 2) break; // Not array of 2-element structs. 208faae50b6SChris Lattner 209faae50b6SChris Lattner Constant *FP = CS->getOperand(1); 210faae50b6SChris Lattner if (FP->isNullValue()) 2110621caefSChris Lattner break; // Found a null terminator, exit. 212faae50b6SChris Lattner 213faae50b6SChris Lattner if (ConstantExpr *CE = dyn_cast<ConstantExpr>(FP)) 2146c38f0bbSReid Spencer if (CE->isCast()) 215faae50b6SChris Lattner FP = CE->getOperand(0); 216faae50b6SChris Lattner if (Function *F = dyn_cast<Function>(FP)) { 217faae50b6SChris Lattner // Execute the ctor/dtor function! 218faae50b6SChris Lattner runFunction(F, std::vector<GenericValue>()); 219faae50b6SChris Lattner } 220faae50b6SChris Lattner } 221faae50b6SChris Lattner } 2220621caefSChris Lattner } 223faae50b6SChris Lattner 2245a0d4829SChris Lattner /// runFunctionAsMain - This is a helper function which wraps runFunction to 2255a0d4829SChris Lattner /// handle the common task of starting up main with the specified argc, argv, 2265a0d4829SChris Lattner /// and envp parameters. 2275a0d4829SChris Lattner int ExecutionEngine::runFunctionAsMain(Function *Fn, 2285a0d4829SChris Lattner const std::vector<std::string> &argv, 2295a0d4829SChris Lattner const char * const * envp) { 2305a0d4829SChris Lattner std::vector<GenericValue> GVArgs; 2315a0d4829SChris Lattner GenericValue GVArgc; 23287aa65f4SReid Spencer GVArgc.IntVal = APInt(32, argv.size()); 233b1cad0b3SChris Lattner unsigned NumArgs = Fn->getFunctionType()->getNumParams(); 234b1cad0b3SChris Lattner if (NumArgs) { 2355a0d4829SChris Lattner GVArgs.push_back(GVArgc); // Arg #0 = argc. 236b1cad0b3SChris Lattner if (NumArgs > 1) { 2375a0d4829SChris Lattner GVArgs.push_back(PTOGV(CreateArgv(this, argv))); // Arg #1 = argv. 238b1cad0b3SChris Lattner assert(((char **)GVTOP(GVArgs[1]))[0] && 239b1cad0b3SChris Lattner "argv[0] was null after CreateArgv"); 240b1cad0b3SChris Lattner if (NumArgs > 2) { 2415a0d4829SChris Lattner std::vector<std::string> EnvVars; 2425a0d4829SChris Lattner for (unsigned i = 0; envp[i]; ++i) 2435a0d4829SChris Lattner EnvVars.push_back(envp[i]); 2445a0d4829SChris Lattner GVArgs.push_back(PTOGV(CreateArgv(this, EnvVars))); // Arg #2 = envp. 245b1cad0b3SChris Lattner } 246b1cad0b3SChris Lattner } 247b1cad0b3SChris Lattner } 24887aa65f4SReid Spencer return runFunction(Fn, GVArgs).IntVal.getZExtValue(); 2495a0d4829SChris Lattner } 2505a0d4829SChris Lattner 251260b0c88SMisha Brukman /// If possible, create a JIT, unless the caller specifically requests an 252260b0c88SMisha Brukman /// Interpreter or there's an error. If even an Interpreter cannot be created, 253260b0c88SMisha Brukman /// NULL is returned. 254857c21b4SMisha Brukman /// 2552f1e2002SMisha Brukman ExecutionEngine *ExecutionEngine::create(ModuleProvider *MP, 256603682adSReid Spencer bool ForceInterpreter, 257603682adSReid Spencer std::string *ErrorStr) { 2584bd3bd5bSBrian Gaeke ExecutionEngine *EE = 0; 2594bd3bd5bSBrian Gaeke 260c8c6c03dSChris Lattner // Unless the interpreter was explicitly selected, try making a JIT. 2612d52c1b8SChris Lattner if (!ForceInterpreter && JITCtor) 262603682adSReid Spencer EE = JITCtor(MP, ErrorStr); 2634bd3bd5bSBrian Gaeke 2644bd3bd5bSBrian Gaeke // If we can't make a JIT, make an interpreter instead. 2652d52c1b8SChris Lattner if (EE == 0 && InterpCtor) 266603682adSReid Spencer EE = InterpCtor(MP, ErrorStr); 267c8c6c03dSChris Lattner 2680b2de9f2SChris Lattner if (EE) { 26970e37278SReid Spencer // Make sure we can resolve symbols in the program as well. The zero arg 27070e37278SReid Spencer // to the function tells DynamicLibrary to load the program, not a library. 27163539389SChris Lattner try { 27270e37278SReid Spencer sys::DynamicLibrary::LoadLibraryPermanently(0); 27363539389SChris Lattner } catch (...) { 27463539389SChris Lattner } 2750b2de9f2SChris Lattner } 27670e37278SReid Spencer 2774bd3bd5bSBrian Gaeke return EE; 2784bd3bd5bSBrian Gaeke } 2794bd3bd5bSBrian Gaeke 280857c21b4SMisha Brukman /// getPointerToGlobal - This returns the address of the specified global 281857c21b4SMisha Brukman /// value. This may involve code generation if it's a function. 282857c21b4SMisha Brukman /// 283996fe010SChris Lattner void *ExecutionEngine::getPointerToGlobal(const GlobalValue *GV) { 2841678e859SBrian Gaeke if (Function *F = const_cast<Function*>(dyn_cast<Function>(GV))) 285996fe010SChris Lattner return getPointerToFunction(F); 286996fe010SChris Lattner 28779876f52SReid Spencer MutexGuard locked(lock); 28869e84901SJeff Cohen void *p = state.getGlobalAddressMap(locked)[GV]; 28969e84901SJeff Cohen if (p) 29069e84901SJeff Cohen return p; 29169e84901SJeff Cohen 29269e84901SJeff Cohen // Global variable might have been added since interpreter started. 29369e84901SJeff Cohen if (GlobalVariable *GVar = 29469e84901SJeff Cohen const_cast<GlobalVariable *>(dyn_cast<GlobalVariable>(GV))) 29569e84901SJeff Cohen EmitGlobalVariable(GVar); 29669e84901SJeff Cohen else 2974da5e17cSChris Lattner assert(0 && "Global hasn't had an address allocated yet!"); 29879876f52SReid Spencer return state.getGlobalAddressMap(locked)[GV]; 299996fe010SChris Lattner } 300996fe010SChris Lattner 3016c38f0bbSReid Spencer /// This function converts a Constant* into a GenericValue. The interesting 3026c38f0bbSReid Spencer /// part is if C is a ConstantExpr. 3036c38f0bbSReid Spencer /// @brief Get a GenericValue for a Constnat* 304996fe010SChris Lattner GenericValue ExecutionEngine::getConstantValue(const Constant *C) { 3056c38f0bbSReid Spencer // If its undefined, return the garbage. 306*4fd528f2SReid Spencer if (isa<UndefValue>(C)) 307*4fd528f2SReid Spencer return GenericValue(); 3089de0d14dSChris Lattner 3096c38f0bbSReid Spencer // If the value is a ConstantExpr 3106c38f0bbSReid Spencer if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) { 311*4fd528f2SReid Spencer Constant *Op0 = CE->getOperand(0); 3129de0d14dSChris Lattner switch (CE->getOpcode()) { 3139de0d14dSChris Lattner case Instruction::GetElementPtr: { 3146c38f0bbSReid Spencer // Compute the index 315*4fd528f2SReid Spencer GenericValue Result = getConstantValue(Op0); 316c44bd78aSChris Lattner SmallVector<Value*, 8> Indices(CE->op_begin()+1, CE->op_end()); 3179de0d14dSChris Lattner uint64_t Offset = 318*4fd528f2SReid Spencer TD->getIndexedOffset(Op0->getType(), &Indices[0], Indices.size()); 3199de0d14dSChris Lattner 32087aa65f4SReid Spencer char* tmp = (char*) Result.PointerVal; 32187aa65f4SReid Spencer Result = PTOGV(tmp + Offset); 3229de0d14dSChris Lattner return Result; 3239de0d14dSChris Lattner } 324*4fd528f2SReid Spencer case Instruction::Trunc: { 325*4fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 326*4fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); 327*4fd528f2SReid Spencer GV.IntVal = GV.IntVal.trunc(BitWidth); 328*4fd528f2SReid Spencer return GV; 329*4fd528f2SReid Spencer } 330*4fd528f2SReid Spencer case Instruction::ZExt: { 331*4fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 332*4fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); 333*4fd528f2SReid Spencer GV.IntVal = GV.IntVal.zext(BitWidth); 334*4fd528f2SReid Spencer return GV; 335*4fd528f2SReid Spencer } 336*4fd528f2SReid Spencer case Instruction::SExt: { 337*4fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 338*4fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); 339*4fd528f2SReid Spencer GV.IntVal = GV.IntVal.sext(BitWidth); 340*4fd528f2SReid Spencer return GV; 341*4fd528f2SReid Spencer } 342*4fd528f2SReid Spencer case Instruction::FPTrunc: { 343*4fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 344*4fd528f2SReid Spencer GV.FloatVal = float(GV.DoubleVal); 345*4fd528f2SReid Spencer return GV; 346*4fd528f2SReid Spencer } 347*4fd528f2SReid Spencer case Instruction::FPExt:{ 348*4fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 349*4fd528f2SReid Spencer GV.DoubleVal = double(GV.FloatVal); 350*4fd528f2SReid Spencer return GV; 351*4fd528f2SReid Spencer } 352*4fd528f2SReid Spencer case Instruction::UIToFP: { 353*4fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 354*4fd528f2SReid Spencer if (CE->getType() == Type::FloatTy) 355*4fd528f2SReid Spencer GV.FloatVal = float(GV.IntVal.roundToDouble()); 356*4fd528f2SReid Spencer else 357*4fd528f2SReid Spencer GV.DoubleVal = GV.IntVal.roundToDouble(); 358*4fd528f2SReid Spencer return GV; 359*4fd528f2SReid Spencer } 360*4fd528f2SReid Spencer case Instruction::SIToFP: { 361*4fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 362*4fd528f2SReid Spencer if (CE->getType() == Type::FloatTy) 363*4fd528f2SReid Spencer GV.FloatVal = float(GV.IntVal.signedRoundToDouble()); 364*4fd528f2SReid Spencer else 365*4fd528f2SReid Spencer GV.DoubleVal = GV.IntVal.signedRoundToDouble(); 366*4fd528f2SReid Spencer return GV; 367*4fd528f2SReid Spencer } 368*4fd528f2SReid Spencer case Instruction::FPToUI: // double->APInt conversion handles sign 369*4fd528f2SReid Spencer case Instruction::FPToSI: { 370*4fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 371*4fd528f2SReid Spencer uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); 372*4fd528f2SReid Spencer if (Op0->getType() == Type::FloatTy) 373*4fd528f2SReid Spencer GV.IntVal = APIntOps::RoundFloatToAPInt(GV.FloatVal, BitWidth); 374*4fd528f2SReid Spencer else 375*4fd528f2SReid Spencer GV.IntVal = APIntOps::RoundDoubleToAPInt(GV.DoubleVal, BitWidth); 376*4fd528f2SReid Spencer return GV; 377*4fd528f2SReid Spencer } 3786c38f0bbSReid Spencer case Instruction::PtrToInt: { 379*4fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 380*4fd528f2SReid Spencer uint32_t PtrWidth = TD->getPointerSizeInBits(); 381*4fd528f2SReid Spencer GV.IntVal = APInt(PtrWidth, uintptr_t(GV.PointerVal)); 382*4fd528f2SReid Spencer return GV; 383*4fd528f2SReid Spencer } 384*4fd528f2SReid Spencer case Instruction::IntToPtr: { 385*4fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 386*4fd528f2SReid Spencer uint32_t PtrWidth = TD->getPointerSizeInBits(); 387*4fd528f2SReid Spencer if (PtrWidth != GV.IntVal.getBitWidth()) 388*4fd528f2SReid Spencer GV.IntVal = GV.IntVal.zextOrTrunc(PtrWidth); 389*4fd528f2SReid Spencer assert(GV.IntVal.getBitWidth() <= 64 && "Bad pointer width"); 390*4fd528f2SReid Spencer GV.PointerVal = PointerTy(uintptr_t(GV.IntVal.getZExtValue())); 3916c38f0bbSReid Spencer return GV; 3926c38f0bbSReid Spencer } 3936c38f0bbSReid Spencer case Instruction::BitCast: { 394*4fd528f2SReid Spencer GenericValue GV = getConstantValue(Op0); 395*4fd528f2SReid Spencer const Type* DestTy = CE->getType(); 396*4fd528f2SReid Spencer switch (Op0->getType()->getTypeID()) { 397*4fd528f2SReid Spencer default: assert(0 && "Invalid bitcast operand"); 398*4fd528f2SReid Spencer case Type::IntegerTyID: 399*4fd528f2SReid Spencer assert(DestTy->isFloatingPoint() && "invalid bitcast"); 400*4fd528f2SReid Spencer if (DestTy == Type::FloatTy) 401*4fd528f2SReid Spencer GV.FloatVal = GV.IntVal.bitsToFloat(); 402*4fd528f2SReid Spencer else if (DestTy == Type::DoubleTy) 403*4fd528f2SReid Spencer GV.DoubleVal = GV.IntVal.bitsToDouble(); 4046c38f0bbSReid Spencer break; 405*4fd528f2SReid Spencer case Type::FloatTyID: 406*4fd528f2SReid Spencer assert(DestTy == Type::Int32Ty && "Invalid bitcast"); 407*4fd528f2SReid Spencer GV.IntVal.floatToBits(GV.FloatVal); 408*4fd528f2SReid Spencer break; 409*4fd528f2SReid Spencer case Type::DoubleTyID: 410*4fd528f2SReid Spencer assert(DestTy == Type::Int64Ty && "Invalid bitcast"); 411*4fd528f2SReid Spencer GV.IntVal.doubleToBits(GV.DoubleVal); 412*4fd528f2SReid Spencer break; 413*4fd528f2SReid Spencer case Type::PointerTyID: 414*4fd528f2SReid Spencer assert(isa<PointerType>(DestTy) && "Invalid bitcast"); 415*4fd528f2SReid Spencer break; // getConstantValue(Op0) above already converted it 4166c38f0bbSReid Spencer } 417*4fd528f2SReid Spencer return GV; 41868cbcc3eSChris Lattner } 41968cbcc3eSChris Lattner case Instruction::Add: 420*4fd528f2SReid Spencer case Instruction::Sub: 421*4fd528f2SReid Spencer case Instruction::Mul: 422*4fd528f2SReid Spencer case Instruction::UDiv: 423*4fd528f2SReid Spencer case Instruction::SDiv: 424*4fd528f2SReid Spencer case Instruction::URem: 425*4fd528f2SReid Spencer case Instruction::SRem: 426*4fd528f2SReid Spencer case Instruction::And: 427*4fd528f2SReid Spencer case Instruction::Or: 428*4fd528f2SReid Spencer case Instruction::Xor: { 429*4fd528f2SReid Spencer GenericValue LHS = getConstantValue(Op0); 430*4fd528f2SReid Spencer GenericValue RHS = getConstantValue(CE->getOperand(1)); 431*4fd528f2SReid Spencer GenericValue GV; 432c4e6bb5fSChris Lattner switch (CE->getOperand(0)->getType()->getTypeID()) { 433c4e6bb5fSChris Lattner default: assert(0 && "Bad add type!"); abort(); 4347a9c62baSReid Spencer case Type::IntegerTyID: 435*4fd528f2SReid Spencer switch (CE->getOpcode()) { 436*4fd528f2SReid Spencer default: assert(0 && "Invalid integer opcode"); 437*4fd528f2SReid Spencer case Instruction::Add: GV.IntVal = LHS.IntVal + RHS.IntVal; break; 438*4fd528f2SReid Spencer case Instruction::Sub: GV.IntVal = LHS.IntVal - RHS.IntVal; break; 439*4fd528f2SReid Spencer case Instruction::Mul: GV.IntVal = LHS.IntVal * RHS.IntVal; break; 440*4fd528f2SReid Spencer case Instruction::UDiv:GV.IntVal = LHS.IntVal.udiv(RHS.IntVal); break; 441*4fd528f2SReid Spencer case Instruction::SDiv:GV.IntVal = LHS.IntVal.sdiv(RHS.IntVal); break; 442*4fd528f2SReid Spencer case Instruction::URem:GV.IntVal = LHS.IntVal.urem(RHS.IntVal); break; 443*4fd528f2SReid Spencer case Instruction::SRem:GV.IntVal = LHS.IntVal.srem(RHS.IntVal); break; 444*4fd528f2SReid Spencer case Instruction::And: GV.IntVal = LHS.IntVal & RHS.IntVal; break; 445*4fd528f2SReid Spencer case Instruction::Or: GV.IntVal = LHS.IntVal | RHS.IntVal; break; 446*4fd528f2SReid Spencer case Instruction::Xor: GV.IntVal = LHS.IntVal ^ RHS.IntVal; break; 447*4fd528f2SReid Spencer } 448c4e6bb5fSChris Lattner break; 449c4e6bb5fSChris Lattner case Type::FloatTyID: 450*4fd528f2SReid Spencer switch (CE->getOpcode()) { 451*4fd528f2SReid Spencer default: assert(0 && "Invalid float opcode"); abort(); 452*4fd528f2SReid Spencer case Instruction::Add: 453*4fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal + RHS.FloatVal; break; 454*4fd528f2SReid Spencer case Instruction::Sub: 455*4fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal - RHS.FloatVal; break; 456*4fd528f2SReid Spencer case Instruction::Mul: 457*4fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal * RHS.FloatVal; break; 458*4fd528f2SReid Spencer case Instruction::FDiv: 459*4fd528f2SReid Spencer GV.FloatVal = LHS.FloatVal / RHS.FloatVal; break; 460*4fd528f2SReid Spencer case Instruction::FRem: 461*4fd528f2SReid Spencer GV.FloatVal = ::fmodf(LHS.FloatVal,RHS.FloatVal); break; 462*4fd528f2SReid Spencer } 463c4e6bb5fSChris Lattner break; 464c4e6bb5fSChris Lattner case Type::DoubleTyID: 465*4fd528f2SReid Spencer switch (CE->getOpcode()) { 466*4fd528f2SReid Spencer default: assert(0 && "Invalid double opcode"); abort(); 467*4fd528f2SReid Spencer case Instruction::Add: 468*4fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal + RHS.DoubleVal; break; 469*4fd528f2SReid Spencer case Instruction::Sub: 470*4fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal - RHS.DoubleVal; break; 471*4fd528f2SReid Spencer case Instruction::Mul: 472*4fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal * RHS.DoubleVal; break; 473*4fd528f2SReid Spencer case Instruction::FDiv: 474*4fd528f2SReid Spencer GV.DoubleVal = LHS.DoubleVal / RHS.DoubleVal; break; 475*4fd528f2SReid Spencer case Instruction::FRem: 476*4fd528f2SReid Spencer GV.DoubleVal = ::fmod(LHS.DoubleVal,RHS.DoubleVal); break; 477*4fd528f2SReid Spencer } 478c4e6bb5fSChris Lattner break; 479c4e6bb5fSChris Lattner } 480*4fd528f2SReid Spencer return GV; 481*4fd528f2SReid Spencer } 4829de0d14dSChris Lattner default: 48368cbcc3eSChris Lattner break; 48468cbcc3eSChris Lattner } 485*4fd528f2SReid Spencer cerr << "ConstantExpr not handled: " << *CE << "\n"; 4869de0d14dSChris Lattner abort(); 4879de0d14dSChris Lattner } 488996fe010SChris Lattner 489*4fd528f2SReid Spencer GenericValue Result; 4906b727599SChris Lattner switch (C->getType()->getTypeID()) { 49187aa65f4SReid Spencer case Type::FloatTyID: 49287aa65f4SReid Spencer Result.FloatVal = (float)cast<ConstantFP>(C)->getValue(); 4937a9c62baSReid Spencer break; 49487aa65f4SReid Spencer case Type::DoubleTyID: 49587aa65f4SReid Spencer Result.DoubleVal = (double)cast<ConstantFP>(C)->getValue(); 49687aa65f4SReid Spencer break; 49787aa65f4SReid Spencer case Type::IntegerTyID: 49887aa65f4SReid Spencer Result.IntVal = cast<ConstantInt>(C)->getValue(); 49987aa65f4SReid Spencer break; 500996fe010SChris Lattner case Type::PointerTyID: 5016a0fd73bSReid Spencer if (isa<ConstantPointerNull>(C)) 502996fe010SChris Lattner Result.PointerVal = 0; 5036a0fd73bSReid Spencer else if (const Function *F = dyn_cast<Function>(C)) 5046a0fd73bSReid Spencer Result = PTOGV(getPointerToFunctionOrStub(const_cast<Function*>(F))); 5056a0fd73bSReid Spencer else if (const GlobalVariable* GV = dyn_cast<GlobalVariable>(C)) 5066a0fd73bSReid Spencer Result = PTOGV(getOrEmitGlobalVariable(const_cast<GlobalVariable*>(GV))); 507e6492f10SChris Lattner else 508996fe010SChris Lattner assert(0 && "Unknown constant pointer type!"); 509996fe010SChris Lattner break; 510996fe010SChris Lattner default: 511*4fd528f2SReid Spencer cerr << "ERROR: Constant unimplemented for type: " << *C->getType() << "\n"; 5129de0d14dSChris Lattner abort(); 513996fe010SChris Lattner } 514996fe010SChris Lattner return Result; 515996fe010SChris Lattner } 516996fe010SChris Lattner 5174ca2ea5bSNate Begeman /// StoreValueToMemory - Stores the data in Val of type Ty at address Ptr. Ptr 5184ca2ea5bSNate Begeman /// is the address of the memory at which to store Val, cast to GenericValue *. 5194ca2ea5bSNate Begeman /// It is not a pointer to a GenericValue containing the address at which to 5204ca2ea5bSNate Begeman /// store Val. 521857c21b4SMisha Brukman /// 5224e42790cSReid Spencer void ExecutionEngine::StoreValueToMemory(const GenericValue &Val, GenericValue *Ptr, 523996fe010SChris Lattner const Type *Ty) { 5246b727599SChris Lattner switch (Ty->getTypeID()) { 5257a9c62baSReid Spencer case Type::IntegerTyID: { 5267a9c62baSReid Spencer unsigned BitWidth = cast<IntegerType>(Ty)->getBitWidth(); 52701f7e06dSReid Spencer GenericValue TmpVal = Val; 5287a9c62baSReid Spencer if (BitWidth <= 8) 52987aa65f4SReid Spencer *((uint8_t*)Ptr) = uint8_t(Val.IntVal.getZExtValue()); 5307a9c62baSReid Spencer else if (BitWidth <= 16) { 53187aa65f4SReid Spencer *((uint16_t*)Ptr) = uint16_t(Val.IntVal.getZExtValue()); 5327a9c62baSReid Spencer } else if (BitWidth <= 32) { 53387aa65f4SReid Spencer *((uint32_t*)Ptr) = uint32_t(Val.IntVal.getZExtValue()); 5347a9c62baSReid Spencer } else if (BitWidth <= 64) { 5354e42790cSReid Spencer *((uint64_t*)Ptr) = uint64_t(Val.IntVal.getZExtValue()); 536815f8dd2SReid Spencer } else { 537815f8dd2SReid Spencer uint64_t *Dest = (uint64_t*)Ptr; 53887aa65f4SReid Spencer const uint64_t *Src = Val.IntVal.getRawData(); 53987aa65f4SReid Spencer for (uint32_t i = 0; i < Val.IntVal.getNumWords(); ++i) 540815f8dd2SReid Spencer Dest[i] = Src[i]; 541815f8dd2SReid Spencer } 542996fe010SChris Lattner break; 5437a9c62baSReid Spencer } 544996fe010SChris Lattner case Type::FloatTyID: 54587aa65f4SReid Spencer *((float*)Ptr) = Val.FloatVal; 54687aa65f4SReid Spencer break; 54787aa65f4SReid Spencer case Type::DoubleTyID: 54887aa65f4SReid Spencer *((double*)Ptr) = Val.DoubleVal; 549996fe010SChris Lattner break; 5507a9c62baSReid Spencer case Type::PointerTyID: 55187aa65f4SReid Spencer *((PointerTy*)Ptr) = Val.PointerVal; 552996fe010SChris Lattner break; 553996fe010SChris Lattner default: 554f3baad3eSBill Wendling cerr << "Cannot store value of type " << *Ty << "!\n"; 555996fe010SChris Lattner } 556996fe010SChris Lattner } 557996fe010SChris Lattner 558857c21b4SMisha Brukman /// FIXME: document 559857c21b4SMisha Brukman /// 56000919f57SReid Spencer void ExecutionEngine::LoadValueFromMemory(GenericValue &Result, 56100919f57SReid Spencer GenericValue *Ptr, 5627f389e8cSChris Lattner const Type *Ty) { 5636b727599SChris Lattner switch (Ty->getTypeID()) { 5647a9c62baSReid Spencer case Type::IntegerTyID: { 5657a9c62baSReid Spencer unsigned BitWidth = cast<IntegerType>(Ty)->getBitWidth(); 5667a9c62baSReid Spencer if (BitWidth <= 8) 56787aa65f4SReid Spencer Result.IntVal = APInt(BitWidth, *((uint8_t*)Ptr)); 5687a9c62baSReid Spencer else if (BitWidth <= 16) { 56987aa65f4SReid Spencer Result.IntVal = APInt(BitWidth, *((uint16_t*)Ptr)); 5707a9c62baSReid Spencer } else if (BitWidth <= 32) { 57187aa65f4SReid Spencer Result.IntVal = APInt(BitWidth, *((uint32_t*)Ptr)); 5727a9c62baSReid Spencer } else if (BitWidth <= 64) { 57387aa65f4SReid Spencer Result.IntVal = APInt(BitWidth, *((uint64_t*)Ptr)); 5747a9c62baSReid Spencer } else 57587aa65f4SReid Spencer Result.IntVal = APInt(BitWidth, BitWidth/64, (uint64_t*)Ptr); 5767f389e8cSChris Lattner break; 5777a9c62baSReid Spencer } 5787f389e8cSChris Lattner case Type::FloatTyID: 57987aa65f4SReid Spencer Result.FloatVal = *((float*)Ptr); 58087aa65f4SReid Spencer break; 58187aa65f4SReid Spencer case Type::DoubleTyID: 58287aa65f4SReid Spencer Result.DoubleVal = *((double*)Ptr); 5837f389e8cSChris Lattner break; 5847a9c62baSReid Spencer case Type::PointerTyID: 58587aa65f4SReid Spencer Result.PointerVal = *((PointerTy*)Ptr); 5867f389e8cSChris Lattner break; 5877f389e8cSChris Lattner default: 588f3baad3eSBill Wendling cerr << "Cannot load value of type " << *Ty << "!\n"; 5897f389e8cSChris Lattner abort(); 5907f389e8cSChris Lattner } 5917f389e8cSChris Lattner } 5927f389e8cSChris Lattner 593996fe010SChris Lattner // InitializeMemory - Recursive function to apply a Constant value into the 594996fe010SChris Lattner // specified memory location... 595996fe010SChris Lattner // 596996fe010SChris Lattner void ExecutionEngine::InitializeMemory(const Constant *Init, void *Addr) { 59761753bf8SChris Lattner if (isa<UndefValue>(Init)) { 59861753bf8SChris Lattner return; 599d84d35baSReid Spencer } else if (const ConstantVector *CP = dyn_cast<ConstantVector>(Init)) { 60069d62138SRobert Bocchino unsigned ElementSize = 60120a631fdSOwen Anderson getTargetData()->getTypeSize(CP->getType()->getElementType()); 60269d62138SRobert Bocchino for (unsigned i = 0, e = CP->getNumOperands(); i != e; ++i) 60369d62138SRobert Bocchino InitializeMemory(CP->getOperand(i), (char*)Addr+i*ElementSize); 60469d62138SRobert Bocchino return; 60561753bf8SChris Lattner } else if (Init->getType()->isFirstClassType()) { 606996fe010SChris Lattner GenericValue Val = getConstantValue(Init); 607996fe010SChris Lattner StoreValueToMemory(Val, (GenericValue*)Addr, Init->getType()); 608996fe010SChris Lattner return; 609834b1272SChris Lattner } else if (isa<ConstantAggregateZero>(Init)) { 61020a631fdSOwen Anderson memset(Addr, 0, (size_t)getTargetData()->getTypeSize(Init->getType())); 611834b1272SChris Lattner return; 612996fe010SChris Lattner } 613996fe010SChris Lattner 6146b727599SChris Lattner switch (Init->getType()->getTypeID()) { 615996fe010SChris Lattner case Type::ArrayTyID: { 616996fe010SChris Lattner const ConstantArray *CPA = cast<ConstantArray>(Init); 617996fe010SChris Lattner unsigned ElementSize = 61820a631fdSOwen Anderson getTargetData()->getTypeSize(CPA->getType()->getElementType()); 61983243725SAlkis Evlogimenos for (unsigned i = 0, e = CPA->getNumOperands(); i != e; ++i) 62083243725SAlkis Evlogimenos InitializeMemory(CPA->getOperand(i), (char*)Addr+i*ElementSize); 621996fe010SChris Lattner return; 622996fe010SChris Lattner } 623996fe010SChris Lattner 624996fe010SChris Lattner case Type::StructTyID: { 625996fe010SChris Lattner const ConstantStruct *CPS = cast<ConstantStruct>(Init); 626996fe010SChris Lattner const StructLayout *SL = 62720a631fdSOwen Anderson getTargetData()->getStructLayout(cast<StructType>(CPS->getType())); 62883243725SAlkis Evlogimenos for (unsigned i = 0, e = CPS->getNumOperands(); i != e; ++i) 629c473d8e4SChris Lattner InitializeMemory(CPS->getOperand(i), (char*)Addr+SL->getElementOffset(i)); 630996fe010SChris Lattner return; 631996fe010SChris Lattner } 632996fe010SChris Lattner 633996fe010SChris Lattner default: 634f3baad3eSBill Wendling cerr << "Bad Type: " << *Init->getType() << "\n"; 635996fe010SChris Lattner assert(0 && "Unknown constant type to initialize memory with!"); 636996fe010SChris Lattner } 637996fe010SChris Lattner } 638996fe010SChris Lattner 639996fe010SChris Lattner /// EmitGlobals - Emit all of the global variables to memory, storing their 640996fe010SChris Lattner /// addresses into GlobalAddress. This must make sure to copy the contents of 641996fe010SChris Lattner /// their initializers into the memory. 642996fe010SChris Lattner /// 643996fe010SChris Lattner void ExecutionEngine::emitGlobals() { 64420a631fdSOwen Anderson const TargetData *TD = getTargetData(); 645996fe010SChris Lattner 646996fe010SChris Lattner // Loop over all of the global variables in the program, allocating the memory 6470621caefSChris Lattner // to hold them. If there is more than one module, do a prepass over globals 6480621caefSChris Lattner // to figure out how the different modules should link together. 6490621caefSChris Lattner // 6500621caefSChris Lattner std::map<std::pair<std::string, const Type*>, 6510621caefSChris Lattner const GlobalValue*> LinkedGlobalsMap; 6520621caefSChris Lattner 6530621caefSChris Lattner if (Modules.size() != 1) { 6540621caefSChris Lattner for (unsigned m = 0, e = Modules.size(); m != e; ++m) { 6550621caefSChris Lattner Module &M = *Modules[m]->getModule(); 6560621caefSChris Lattner for (Module::const_global_iterator I = M.global_begin(), 6570621caefSChris Lattner E = M.global_end(); I != E; ++I) { 6580621caefSChris Lattner const GlobalValue *GV = I; 6595301e7c6SReid Spencer if (GV->hasInternalLinkage() || GV->isDeclaration() || 6600621caefSChris Lattner GV->hasAppendingLinkage() || !GV->hasName()) 6610621caefSChris Lattner continue;// Ignore external globals and globals with internal linkage. 6620621caefSChris Lattner 6630621caefSChris Lattner const GlobalValue *&GVEntry = 6640621caefSChris Lattner LinkedGlobalsMap[std::make_pair(GV->getName(), GV->getType())]; 6650621caefSChris Lattner 6660621caefSChris Lattner // If this is the first time we've seen this global, it is the canonical 6670621caefSChris Lattner // version. 6680621caefSChris Lattner if (!GVEntry) { 6690621caefSChris Lattner GVEntry = GV; 6700621caefSChris Lattner continue; 6710621caefSChris Lattner } 6720621caefSChris Lattner 6730621caefSChris Lattner // If the existing global is strong, never replace it. 674d61d39ecSAnton Korobeynikov if (GVEntry->hasExternalLinkage() || 675d61d39ecSAnton Korobeynikov GVEntry->hasDLLImportLinkage() || 676d61d39ecSAnton Korobeynikov GVEntry->hasDLLExportLinkage()) 6770621caefSChris Lattner continue; 6780621caefSChris Lattner 6790621caefSChris Lattner // Otherwise, we know it's linkonce/weak, replace it if this is a strong 6800621caefSChris Lattner // symbol. 68112c94949SAnton Korobeynikov if (GV->hasExternalLinkage() || GVEntry->hasExternalWeakLinkage()) 6820621caefSChris Lattner GVEntry = GV; 6830621caefSChris Lattner } 6840621caefSChris Lattner } 6850621caefSChris Lattner } 6860621caefSChris Lattner 6870621caefSChris Lattner std::vector<const GlobalValue*> NonCanonicalGlobals; 6880621caefSChris Lattner for (unsigned m = 0, e = Modules.size(); m != e; ++m) { 6890621caefSChris Lattner Module &M = *Modules[m]->getModule(); 6908ffb6611SChris Lattner for (Module::const_global_iterator I = M.global_begin(), E = M.global_end(); 6910621caefSChris Lattner I != E; ++I) { 6920621caefSChris Lattner // In the multi-module case, see what this global maps to. 6930621caefSChris Lattner if (!LinkedGlobalsMap.empty()) { 6940621caefSChris Lattner if (const GlobalValue *GVEntry = 6950621caefSChris Lattner LinkedGlobalsMap[std::make_pair(I->getName(), I->getType())]) { 6960621caefSChris Lattner // If something else is the canonical global, ignore this one. 6970621caefSChris Lattner if (GVEntry != &*I) { 6980621caefSChris Lattner NonCanonicalGlobals.push_back(I); 6990621caefSChris Lattner continue; 7000621caefSChris Lattner } 7010621caefSChris Lattner } 7020621caefSChris Lattner } 7030621caefSChris Lattner 7045301e7c6SReid Spencer if (!I->isDeclaration()) { 7050621caefSChris Lattner // Get the type of the global. 706996fe010SChris Lattner const Type *Ty = I->getType()->getElementType(); 707996fe010SChris Lattner 708996fe010SChris Lattner // Allocate some memory for it! 70920a631fdSOwen Anderson unsigned Size = TD->getTypeSize(Ty); 7106bbe3eceSChris Lattner addGlobalMapping(I, new char[Size]); 711996fe010SChris Lattner } else { 712e8bbcfc2SBrian Gaeke // External variable reference. Try to use the dynamic loader to 713e8bbcfc2SBrian Gaeke // get a pointer to it. 7140621caefSChris Lattner if (void *SymAddr = 7150621caefSChris Lattner sys::DynamicLibrary::SearchForAddressOfSymbol(I->getName().c_str())) 716748e8579SChris Lattner addGlobalMapping(I, SymAddr); 7179de0d14dSChris Lattner else { 718f3baad3eSBill Wendling cerr << "Could not resolve external global address: " 7199de0d14dSChris Lattner << I->getName() << "\n"; 7209de0d14dSChris Lattner abort(); 7219de0d14dSChris Lattner } 722996fe010SChris Lattner } 7230621caefSChris Lattner } 7240621caefSChris Lattner 7250621caefSChris Lattner // If there are multiple modules, map the non-canonical globals to their 7260621caefSChris Lattner // canonical location. 7270621caefSChris Lattner if (!NonCanonicalGlobals.empty()) { 7280621caefSChris Lattner for (unsigned i = 0, e = NonCanonicalGlobals.size(); i != e; ++i) { 7290621caefSChris Lattner const GlobalValue *GV = NonCanonicalGlobals[i]; 7300621caefSChris Lattner const GlobalValue *CGV = 7310621caefSChris Lattner LinkedGlobalsMap[std::make_pair(GV->getName(), GV->getType())]; 7320621caefSChris Lattner void *Ptr = getPointerToGlobalIfAvailable(CGV); 7330621caefSChris Lattner assert(Ptr && "Canonical global wasn't codegen'd!"); 7340621caefSChris Lattner addGlobalMapping(GV, getPointerToGlobalIfAvailable(CGV)); 7350621caefSChris Lattner } 7360621caefSChris Lattner } 737996fe010SChris Lattner 7387a9c62baSReid Spencer // Now that all of the globals are set up in memory, loop through them all 7397a9c62baSReid Spencer // and initialize their contents. 7408ffb6611SChris Lattner for (Module::const_global_iterator I = M.global_begin(), E = M.global_end(); 7410621caefSChris Lattner I != E; ++I) { 7425301e7c6SReid Spencer if (!I->isDeclaration()) { 7430621caefSChris Lattner if (!LinkedGlobalsMap.empty()) { 7440621caefSChris Lattner if (const GlobalValue *GVEntry = 7450621caefSChris Lattner LinkedGlobalsMap[std::make_pair(I->getName(), I->getType())]) 7460621caefSChris Lattner if (GVEntry != &*I) // Not the canonical variable. 7470621caefSChris Lattner continue; 7480621caefSChris Lattner } 7496bbe3eceSChris Lattner EmitGlobalVariable(I); 7506bbe3eceSChris Lattner } 7510621caefSChris Lattner } 7520621caefSChris Lattner } 7530621caefSChris Lattner } 7546bbe3eceSChris Lattner 7556bbe3eceSChris Lattner // EmitGlobalVariable - This method emits the specified global variable to the 7566bbe3eceSChris Lattner // address specified in GlobalAddresses, or allocates new memory if it's not 7576bbe3eceSChris Lattner // already in the map. 758fbcc0aa1SChris Lattner void ExecutionEngine::EmitGlobalVariable(const GlobalVariable *GV) { 759748e8579SChris Lattner void *GA = getPointerToGlobalIfAvailable(GV); 7605834fdb3SBill Wendling DOUT << "Global '" << GV->getName() << "' -> " << GA << "\n"; 761dc631735SChris Lattner 762fbcc0aa1SChris Lattner const Type *ElTy = GV->getType()->getElementType(); 76320a631fdSOwen Anderson size_t GVSize = (size_t)getTargetData()->getTypeSize(ElTy); 7646bbe3eceSChris Lattner if (GA == 0) { 7656bbe3eceSChris Lattner // If it's not already specified, allocate memory for the global. 766d215992bSChris Lattner GA = new char[GVSize]; 767748e8579SChris Lattner addGlobalMapping(GV, GA); 7686bbe3eceSChris Lattner } 769fbcc0aa1SChris Lattner 7706bbe3eceSChris Lattner InitializeMemory(GV->getInitializer(), GA); 771df1f1524SChris Lattner NumInitBytes += (unsigned)GVSize; 7726bbe3eceSChris Lattner ++NumGlobals; 773996fe010SChris Lattner } 774