blob: 33f29b4e4c3c0db046024d7d0150ccb929c85e4e [file] [log] [blame]
Misha Brukman4afac182003-10-10 17:45:12 +00001//===-- ExecutionEngine.cpp - Common Implementation shared by EEs ---------===//
Misha Brukmanedf128a2005-04-21 22:36:52 +00002//
John Criswellb576c942003-10-20 19:43:21 +00003// The LLVM Compiler Infrastructure
4//
Chris Lattner4ee451d2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Misha Brukmanedf128a2005-04-21 22:36:52 +00007//
John Criswellb576c942003-10-20 19:43:21 +00008//===----------------------------------------------------------------------===//
Misha Brukmanedf128a2005-04-21 22:36:52 +00009//
Chris Lattnerbd199fb2002-12-24 00:01:05 +000010// This file defines the common interface used by the various execution engine
11// subclasses.
12//
13//===----------------------------------------------------------------------===//
14
Chris Lattner3785fad2003-08-05 17:00:32 +000015#define DEBUG_TYPE "jit"
Chris Lattnerbd199fb2002-12-24 00:01:05 +000016#include "llvm/Constants.h"
Misha Brukman19684162003-10-16 21:18:05 +000017#include "llvm/DerivedTypes.h"
Chris Lattnerbd199fb2002-12-24 00:01:05 +000018#include "llvm/Module.h"
Misha Brukman19684162003-10-16 21:18:05 +000019#include "llvm/ModuleProvider.h"
Reid Spencerdf5a37e2004-11-29 14:11:29 +000020#include "llvm/ADT/Statistic.h"
Duncan Sands8a43e9e2007-12-14 19:38:31 +000021#include "llvm/Config/alloca.h"
Misha Brukman19684162003-10-16 21:18:05 +000022#include "llvm/ExecutionEngine/ExecutionEngine.h"
Chris Lattnerfd131292003-09-05 20:08:15 +000023#include "llvm/ExecutionEngine/GenericValue.h"
Reid Spencer551ccae2004-09-01 22:55:40 +000024#include "llvm/Support/Debug.h"
Chris Lattnere7fd5532006-05-08 22:00:52 +000025#include "llvm/Support/MutexGuard.h"
Reid Spencerdf5a37e2004-11-29 14:11:29 +000026#include "llvm/System/DynamicLibrary.h"
Duncan Sands67f1c492007-12-12 23:03:45 +000027#include "llvm/System/Host.h"
Reid Spencerdf5a37e2004-11-29 14:11:29 +000028#include "llvm/Target/TargetData.h"
Anton Korobeynikovae9f3a32008-02-20 11:08:44 +000029#include <cmath>
30#include <cstring>
Chris Lattnerc2ee9b92003-11-19 21:08:57 +000031using namespace llvm;
Chris Lattnerbd199fb2002-12-24 00:01:05 +000032
Chris Lattner36343732006-12-19 22:43:32 +000033STATISTIC(NumInitBytes, "Number of bytes of global vars initialized");
34STATISTIC(NumGlobals , "Number of global vars initialized");
Chris Lattnerbd199fb2002-12-24 00:01:05 +000035
Chris Lattner2fe4bb02006-03-22 06:07:50 +000036ExecutionEngine::EECtorFn ExecutionEngine::JITCtor = 0;
37ExecutionEngine::EECtorFn ExecutionEngine::InterpCtor = 0;
Nicolas Geoffrayafe6c2b2008-02-13 18:39:37 +000038ExecutionEngine::EERegisterFn ExecutionEngine::ExceptionTableRegister = 0;
39
Chris Lattner2fe4bb02006-03-22 06:07:50 +000040
Chris Lattnerd958a5a2007-10-22 02:50:12 +000041ExecutionEngine::ExecutionEngine(ModuleProvider *P) : LazyFunctionCreator(0) {
Chris Lattner3d6e33d2006-11-09 19:31:15 +000042 LazyCompilationDisabled = false;
Evan Cheng446531e2008-09-24 16:25:55 +000043 GVCompilationDisabled = false;
Evan Cheng1b088f32008-06-17 16:49:02 +000044 SymbolSearchingDisabled = false;
Nate Begemand6b7a242009-02-18 08:31:02 +000045 DlsymStubsEnabled = false;
Chris Lattnerfe854032006-08-16 01:24:12 +000046 Modules.push_back(P);
Misha Brukman19684162003-10-16 21:18:05 +000047 assert(P && "ModuleProvider is null?");
48}
49
Brian Gaeke8e539482003-09-04 22:57:27 +000050ExecutionEngine::~ExecutionEngine() {
Reid Spencerd4c0e622007-03-03 18:19:18 +000051 clearAllGlobalMappings();
Chris Lattnerfe854032006-08-16 01:24:12 +000052 for (unsigned i = 0, e = Modules.size(); i != e; ++i)
53 delete Modules[i];
Brian Gaeke8e539482003-09-04 22:57:27 +000054}
55
Nicolas Geoffray46fa1392008-10-25 15:41:43 +000056char* ExecutionEngine::getMemoryForGV(const GlobalVariable* GV) {
57 const Type *ElTy = GV->getType()->getElementType();
Duncan Sandsceb4d1a2009-01-12 20:38:59 +000058 size_t GVSize = (size_t)getTargetData()->getTypePaddedSize(ElTy);
Nicolas Geoffray46fa1392008-10-25 15:41:43 +000059 return new char[GVSize];
60}
61
Devang Patel73d0e212007-10-15 19:56:32 +000062/// removeModuleProvider - Remove a ModuleProvider from the list of modules.
Nate Begeman60789e42009-01-23 19:27:28 +000063/// Relases the Module from the ModuleProvider, materializing it in the
64/// process, and returns the materialized Module.
Devang Patel73d0e212007-10-15 19:56:32 +000065Module* ExecutionEngine::removeModuleProvider(ModuleProvider *P,
66 std::string *ErrInfo) {
67 for(SmallVector<ModuleProvider *, 1>::iterator I = Modules.begin(),
68 E = Modules.end(); I != E; ++I) {
69 ModuleProvider *MP = *I;
70 if (MP == P) {
71 Modules.erase(I);
Nate Begemanf049e072008-05-21 16:34:48 +000072 clearGlobalMappingsFromModule(MP->getModule());
Devang Patel73d0e212007-10-15 19:56:32 +000073 return MP->releaseModule(ErrInfo);
74 }
75 }
76 return NULL;
77}
78
Nate Begeman60789e42009-01-23 19:27:28 +000079/// deleteModuleProvider - Remove a ModuleProvider from the list of modules,
80/// and deletes the ModuleProvider and owned Module. Avoids materializing
81/// the underlying module.
82void ExecutionEngine::deleteModuleProvider(ModuleProvider *P,
83 std::string *ErrInfo) {
84 for(SmallVector<ModuleProvider *, 1>::iterator I = Modules.begin(),
85 E = Modules.end(); I != E; ++I) {
86 ModuleProvider *MP = *I;
87 if (MP == P) {
88 Modules.erase(I);
89 clearGlobalMappingsFromModule(MP->getModule());
90 delete MP;
91 return;
92 }
93 }
94}
95
Chris Lattnerfe854032006-08-16 01:24:12 +000096/// FindFunctionNamed - Search all of the active modules to find the one that
97/// defines FnName. This is very slow operation and shouldn't be used for
98/// general code.
99Function *ExecutionEngine::FindFunctionNamed(const char *FnName) {
100 for (unsigned i = 0, e = Modules.size(); i != e; ++i) {
Reid Spencer688b0492007-02-05 21:19:13 +0000101 if (Function *F = Modules[i]->getModule()->getFunction(FnName))
Chris Lattnerfe854032006-08-16 01:24:12 +0000102 return F;
103 }
104 return 0;
105}
106
107
Chris Lattnere7fd5532006-05-08 22:00:52 +0000108/// addGlobalMapping - Tell the execution engine that the specified global is
109/// at the specified location. This is used internally as functions are JIT'd
110/// and as global variables are laid out in memory. It can and should also be
111/// used by clients of the EE that want to have an LLVM global overlay
112/// existing data in memory.
113void ExecutionEngine::addGlobalMapping(const GlobalValue *GV, void *Addr) {
114 MutexGuard locked(lock);
Evan Chengbc4707a2008-09-18 07:54:21 +0000115
Evan Cheng366cf292008-11-07 22:30:29 +0000116 DOUT << "JIT: Map \'" << GV->getNameStart() << "\' to [" << Addr << "]\n";
Chris Lattnere7fd5532006-05-08 22:00:52 +0000117 void *&CurVal = state.getGlobalAddressMap(locked)[GV];
118 assert((CurVal == 0 || Addr == 0) && "GlobalMapping already established!");
119 CurVal = Addr;
120
121 // If we are using the reverse mapping, add it too
122 if (!state.getGlobalAddressReverseMap(locked).empty()) {
123 const GlobalValue *&V = state.getGlobalAddressReverseMap(locked)[Addr];
124 assert((V == 0 || GV == 0) && "GlobalMapping already established!");
125 V = GV;
126 }
127}
128
129/// clearAllGlobalMappings - Clear all global mappings and start over again
130/// use in dynamic compilation scenarios when you want to move globals
131void ExecutionEngine::clearAllGlobalMappings() {
132 MutexGuard locked(lock);
133
134 state.getGlobalAddressMap(locked).clear();
135 state.getGlobalAddressReverseMap(locked).clear();
136}
137
Nate Begemanf049e072008-05-21 16:34:48 +0000138/// clearGlobalMappingsFromModule - Clear all global mappings that came from a
139/// particular module, because it has been removed from the JIT.
140void ExecutionEngine::clearGlobalMappingsFromModule(Module *M) {
141 MutexGuard locked(lock);
142
143 for (Module::iterator FI = M->begin(), FE = M->end(); FI != FE; ++FI) {
144 state.getGlobalAddressMap(locked).erase(FI);
145 state.getGlobalAddressReverseMap(locked).erase(FI);
146 }
147 for (Module::global_iterator GI = M->global_begin(), GE = M->global_end();
148 GI != GE; ++GI) {
149 state.getGlobalAddressMap(locked).erase(GI);
150 state.getGlobalAddressReverseMap(locked).erase(GI);
151 }
152}
153
Chris Lattnere7fd5532006-05-08 22:00:52 +0000154/// updateGlobalMapping - Replace an existing mapping for GV with a new
155/// address. This updates both maps as required. If "Addr" is null, the
156/// entry for the global is removed from the mappings.
Chris Lattnerf4cc3092008-04-04 04:47:41 +0000157void *ExecutionEngine::updateGlobalMapping(const GlobalValue *GV, void *Addr) {
Chris Lattnere7fd5532006-05-08 22:00:52 +0000158 MutexGuard locked(lock);
Chris Lattnerf4cc3092008-04-04 04:47:41 +0000159
160 std::map<const GlobalValue*, void *> &Map = state.getGlobalAddressMap(locked);
161
Chris Lattnere7fd5532006-05-08 22:00:52 +0000162 // Deleting from the mapping?
163 if (Addr == 0) {
Chris Lattnerf4cc3092008-04-04 04:47:41 +0000164 std::map<const GlobalValue*, void *>::iterator I = Map.find(GV);
165 void *OldVal;
166 if (I == Map.end())
167 OldVal = 0;
168 else {
169 OldVal = I->second;
170 Map.erase(I);
171 }
172
Chris Lattnere7fd5532006-05-08 22:00:52 +0000173 if (!state.getGlobalAddressReverseMap(locked).empty())
174 state.getGlobalAddressReverseMap(locked).erase(Addr);
Chris Lattnerf4cc3092008-04-04 04:47:41 +0000175 return OldVal;
Chris Lattnere7fd5532006-05-08 22:00:52 +0000176 }
177
Chris Lattnerf4cc3092008-04-04 04:47:41 +0000178 void *&CurVal = Map[GV];
179 void *OldVal = CurVal;
180
Chris Lattnere7fd5532006-05-08 22:00:52 +0000181 if (CurVal && !state.getGlobalAddressReverseMap(locked).empty())
182 state.getGlobalAddressReverseMap(locked).erase(CurVal);
183 CurVal = Addr;
184
185 // If we are using the reverse mapping, add it too
186 if (!state.getGlobalAddressReverseMap(locked).empty()) {
187 const GlobalValue *&V = state.getGlobalAddressReverseMap(locked)[Addr];
188 assert((V == 0 || GV == 0) && "GlobalMapping already established!");
189 V = GV;
190 }
Chris Lattnerf4cc3092008-04-04 04:47:41 +0000191 return OldVal;
Chris Lattnere7fd5532006-05-08 22:00:52 +0000192}
193
194/// getPointerToGlobalIfAvailable - This returns the address of the specified
195/// global value if it is has already been codegen'd, otherwise it returns null.
196///
197void *ExecutionEngine::getPointerToGlobalIfAvailable(const GlobalValue *GV) {
198 MutexGuard locked(lock);
199
200 std::map<const GlobalValue*, void*>::iterator I =
201 state.getGlobalAddressMap(locked).find(GV);
202 return I != state.getGlobalAddressMap(locked).end() ? I->second : 0;
203}
204
Chris Lattner55d86482003-12-31 20:21:04 +0000205/// getGlobalValueAtAddress - Return the LLVM global value object that starts
206/// at the specified address.
207///
208const GlobalValue *ExecutionEngine::getGlobalValueAtAddress(void *Addr) {
Reid Spenceree448632005-07-12 15:51:55 +0000209 MutexGuard locked(lock);
210
Chris Lattner55d86482003-12-31 20:21:04 +0000211 // If we haven't computed the reverse mapping yet, do so first.
Reid Spenceree448632005-07-12 15:51:55 +0000212 if (state.getGlobalAddressReverseMap(locked).empty()) {
Chris Lattnere7fd5532006-05-08 22:00:52 +0000213 for (std::map<const GlobalValue*, void *>::iterator
214 I = state.getGlobalAddressMap(locked).begin(),
215 E = state.getGlobalAddressMap(locked).end(); I != E; ++I)
216 state.getGlobalAddressReverseMap(locked).insert(std::make_pair(I->second,
217 I->first));
Chris Lattner55d86482003-12-31 20:21:04 +0000218 }
219
220 std::map<void *, const GlobalValue*>::iterator I =
Reid Spenceree448632005-07-12 15:51:55 +0000221 state.getGlobalAddressReverseMap(locked).find(Addr);
222 return I != state.getGlobalAddressReverseMap(locked).end() ? I->second : 0;
Chris Lattner55d86482003-12-31 20:21:04 +0000223}
Chris Lattner87f03102003-12-26 06:50:30 +0000224
225// CreateArgv - Turn a vector of strings into a nice argv style array of
226// pointers to null terminated strings.
227//
228static void *CreateArgv(ExecutionEngine *EE,
229 const std::vector<std::string> &InputArgv) {
Owen Andersona69571c2006-05-03 01:29:57 +0000230 unsigned PtrSize = EE->getTargetData()->getPointerSize();
Chris Lattner87f03102003-12-26 06:50:30 +0000231 char *Result = new char[(InputArgv.size()+1)*PtrSize];
232
Evan Chengeb5d95a2008-11-06 17:46:04 +0000233 DOUT << "JIT: ARGV = " << (void*)Result << "\n";
Christopher Lamb43ad6b32007-12-17 01:12:55 +0000234 const Type *SBytePtr = PointerType::getUnqual(Type::Int8Ty);
Chris Lattner87f03102003-12-26 06:50:30 +0000235
236 for (unsigned i = 0; i != InputArgv.size(); ++i) {
237 unsigned Size = InputArgv[i].size()+1;
238 char *Dest = new char[Size];
Evan Chengeb5d95a2008-11-06 17:46:04 +0000239 DOUT << "JIT: ARGV[" << i << "] = " << (void*)Dest << "\n";
Misha Brukmanedf128a2005-04-21 22:36:52 +0000240
Chris Lattner87f03102003-12-26 06:50:30 +0000241 std::copy(InputArgv[i].begin(), InputArgv[i].end(), Dest);
242 Dest[Size-1] = 0;
Misha Brukmanedf128a2005-04-21 22:36:52 +0000243
Chris Lattner87f03102003-12-26 06:50:30 +0000244 // Endian safe: Result[i] = (PointerTy)Dest;
245 EE->StoreValueToMemory(PTOGV(Dest), (GenericValue*)(Result+i*PtrSize),
246 SBytePtr);
247 }
248
249 // Null terminate it
250 EE->StoreValueToMemory(PTOGV(0),
251 (GenericValue*)(Result+InputArgv.size()*PtrSize),
252 SBytePtr);
253 return Result;
254}
255
Chris Lattner9ca6cda2006-03-08 18:42:46 +0000256
257/// runStaticConstructorsDestructors - This method is used to execute all of
Evan Cheng18314dc2008-09-30 15:51:21 +0000258/// the static constructors or destructors for a module, depending on the
Chris Lattner9ca6cda2006-03-08 18:42:46 +0000259/// value of isDtors.
Evan Cheng18314dc2008-09-30 15:51:21 +0000260void ExecutionEngine::runStaticConstructorsDestructors(Module *module, bool isDtors) {
Chris Lattner9ca6cda2006-03-08 18:42:46 +0000261 const char *Name = isDtors ? "llvm.global_dtors" : "llvm.global_ctors";
Chris Lattner9ca6cda2006-03-08 18:42:46 +0000262
Chris Lattnerfe854032006-08-16 01:24:12 +0000263 // Execute global ctors/dtors for each module in the program.
Chris Lattnerfe854032006-08-16 01:24:12 +0000264
Evan Cheng18314dc2008-09-30 15:51:21 +0000265 GlobalVariable *GV = module->getNamedGlobal(Name);
266
267 // If this global has internal linkage, or if it has a use, then it must be
268 // an old-style (llvmgcc3) static ctor with __main linked in and in use. If
269 // this is the case, don't execute any of the global ctors, __main will do
270 // it.
Rafael Espindolabb46f522009-01-15 20:18:42 +0000271 if (!GV || GV->isDeclaration() || GV->hasLocalLinkage()) return;
Evan Cheng18314dc2008-09-30 15:51:21 +0000272
273 // Should be an array of '{ int, void ()* }' structs. The first value is
274 // the init priority, which we ignore.
275 ConstantArray *InitList = dyn_cast<ConstantArray>(GV->getInitializer());
276 if (!InitList) return;
277 for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i)
278 if (ConstantStruct *CS =
279 dyn_cast<ConstantStruct>(InitList->getOperand(i))) {
280 if (CS->getNumOperands() != 2) return; // Not array of 2-element structs.
281
282 Constant *FP = CS->getOperand(1);
283 if (FP->isNullValue())
284 break; // Found a null terminator, exit.
285
286 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(FP))
287 if (CE->isCast())
288 FP = CE->getOperand(0);
289 if (Function *F = dyn_cast<Function>(FP)) {
290 // Execute the ctor/dtor function!
291 runFunction(F, std::vector<GenericValue>());
292 }
293 }
294}
295
296/// runStaticConstructorsDestructors - This method is used to execute all of
297/// the static constructors or destructors for a program, depending on the
298/// value of isDtors.
299void ExecutionEngine::runStaticConstructorsDestructors(bool isDtors) {
300 // Execute global ctors/dtors for each module in the program.
301 for (unsigned m = 0, e = Modules.size(); m != e; ++m)
302 runStaticConstructorsDestructors(Modules[m]->getModule(), isDtors);
Chris Lattner9ca6cda2006-03-08 18:42:46 +0000303}
304
Dan Gohmanb6e3d6c2008-08-26 01:38:29 +0000305#ifndef NDEBUG
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000306/// isTargetNullPtr - Return whether the target pointer stored at Loc is null.
307static bool isTargetNullPtr(ExecutionEngine *EE, void *Loc) {
308 unsigned PtrSize = EE->getTargetData()->getPointerSize();
309 for (unsigned i = 0; i < PtrSize; ++i)
310 if (*(i + (uint8_t*)Loc))
311 return false;
312 return true;
313}
Dan Gohmanb6e3d6c2008-08-26 01:38:29 +0000314#endif
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000315
Chris Lattner87f03102003-12-26 06:50:30 +0000316/// runFunctionAsMain - This is a helper function which wraps runFunction to
317/// handle the common task of starting up main with the specified argc, argv,
318/// and envp parameters.
319int ExecutionEngine::runFunctionAsMain(Function *Fn,
320 const std::vector<std::string> &argv,
321 const char * const * envp) {
322 std::vector<GenericValue> GVArgs;
323 GenericValue GVArgc;
Reid Spencer8fb0f192007-03-06 03:04:04 +0000324 GVArgc.IntVal = APInt(32, argv.size());
Anton Korobeynikov499d8f02007-06-03 19:17:35 +0000325
326 // Check main() type
Chris Lattnerf24d0992004-08-16 01:05:35 +0000327 unsigned NumArgs = Fn->getFunctionType()->getNumParams();
Anton Korobeynikov499d8f02007-06-03 19:17:35 +0000328 const FunctionType *FTy = Fn->getFunctionType();
Christopher Lamb43ad6b32007-12-17 01:12:55 +0000329 const Type* PPInt8Ty =
330 PointerType::getUnqual(PointerType::getUnqual(Type::Int8Ty));
Anton Korobeynikov499d8f02007-06-03 19:17:35 +0000331 switch (NumArgs) {
332 case 3:
333 if (FTy->getParamType(2) != PPInt8Ty) {
334 cerr << "Invalid type for third argument of main() supplied\n";
335 abort();
336 }
Anton Korobeynikovfb450862007-06-03 19:20:49 +0000337 // FALLS THROUGH
Anton Korobeynikov499d8f02007-06-03 19:17:35 +0000338 case 2:
339 if (FTy->getParamType(1) != PPInt8Ty) {
340 cerr << "Invalid type for second argument of main() supplied\n";
341 abort();
342 }
Anton Korobeynikovfb450862007-06-03 19:20:49 +0000343 // FALLS THROUGH
Anton Korobeynikov499d8f02007-06-03 19:17:35 +0000344 case 1:
345 if (FTy->getParamType(0) != Type::Int32Ty) {
346 cerr << "Invalid type for first argument of main() supplied\n";
347 abort();
348 }
Anton Korobeynikovfb450862007-06-03 19:20:49 +0000349 // FALLS THROUGH
Anton Korobeynikov499d8f02007-06-03 19:17:35 +0000350 case 0:
Chris Lattner45f36832009-02-04 17:48:18 +0000351 if (!isa<IntegerType>(FTy->getReturnType()) &&
Anton Korobeynikov499d8f02007-06-03 19:17:35 +0000352 FTy->getReturnType() != Type::VoidTy) {
353 cerr << "Invalid return type of main() supplied\n";
354 abort();
355 }
356 break;
357 default:
358 cerr << "Invalid number of arguments of main() supplied\n";
359 abort();
360 }
361
Chris Lattnerf24d0992004-08-16 01:05:35 +0000362 if (NumArgs) {
363 GVArgs.push_back(GVArgc); // Arg #0 = argc.
364 if (NumArgs > 1) {
365 GVArgs.push_back(PTOGV(CreateArgv(this, argv))); // Arg #1 = argv.
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000366 assert(!isTargetNullPtr(this, GVTOP(GVArgs[1])) &&
Chris Lattnerf24d0992004-08-16 01:05:35 +0000367 "argv[0] was null after CreateArgv");
368 if (NumArgs > 2) {
369 std::vector<std::string> EnvVars;
370 for (unsigned i = 0; envp[i]; ++i)
371 EnvVars.push_back(envp[i]);
372 GVArgs.push_back(PTOGV(CreateArgv(this, EnvVars))); // Arg #2 = envp.
373 }
374 }
375 }
Reid Spencer8fb0f192007-03-06 03:04:04 +0000376 return runFunction(Fn, GVArgs).IntVal.getZExtValue();
Chris Lattner87f03102003-12-26 06:50:30 +0000377}
378
Misha Brukman19684162003-10-16 21:18:05 +0000379/// If possible, create a JIT, unless the caller specifically requests an
380/// Interpreter or there's an error. If even an Interpreter cannot be created,
Misha Brukmanedf128a2005-04-21 22:36:52 +0000381/// NULL is returned.
Misha Brukman4afac182003-10-10 17:45:12 +0000382///
Misha Brukmanedf128a2005-04-21 22:36:52 +0000383ExecutionEngine *ExecutionEngine::create(ModuleProvider *MP,
Reid Spencerd4c0e622007-03-03 18:19:18 +0000384 bool ForceInterpreter,
Evan Cheng502f20b2008-08-08 08:11:34 +0000385 std::string *ErrorStr,
386 bool Fast) {
Brian Gaeke82d82772003-09-03 20:34:19 +0000387 ExecutionEngine *EE = 0;
388
Nick Lewycky6456d862008-03-08 02:49:45 +0000389 // Make sure we can resolve symbols in the program as well. The zero arg
390 // to the function tells DynamicLibrary to load the program, not a library.
391 if (sys::DynamicLibrary::LoadLibraryPermanently(0, ErrorStr))
392 return 0;
393
Chris Lattner73011782003-12-28 09:44:37 +0000394 // Unless the interpreter was explicitly selected, try making a JIT.
Chris Lattner2fe4bb02006-03-22 06:07:50 +0000395 if (!ForceInterpreter && JITCtor)
Evan Cheng502f20b2008-08-08 08:11:34 +0000396 EE = JITCtor(MP, ErrorStr, Fast);
Brian Gaeke82d82772003-09-03 20:34:19 +0000397
398 // If we can't make a JIT, make an interpreter instead.
Chris Lattner2fe4bb02006-03-22 06:07:50 +0000399 if (EE == 0 && InterpCtor)
Evan Cheng502f20b2008-08-08 08:11:34 +0000400 EE = InterpCtor(MP, ErrorStr, Fast);
Chris Lattner73011782003-12-28 09:44:37 +0000401
Brian Gaeke82d82772003-09-03 20:34:19 +0000402 return EE;
403}
404
Chris Lattner8b5295b2007-10-21 22:57:11 +0000405ExecutionEngine *ExecutionEngine::create(Module *M) {
406 return create(new ExistingModuleProvider(M));
407}
408
Misha Brukman4afac182003-10-10 17:45:12 +0000409/// getPointerToGlobal - This returns the address of the specified global
410/// value. This may involve code generation if it's a function.
411///
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000412void *ExecutionEngine::getPointerToGlobal(const GlobalValue *GV) {
Brian Gaeke37df4602003-08-13 18:16:14 +0000413 if (Function *F = const_cast<Function*>(dyn_cast<Function>(GV)))
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000414 return getPointerToFunction(F);
415
Reid Spenceree448632005-07-12 15:51:55 +0000416 MutexGuard locked(lock);
Jeff Cohen68835dd2006-02-07 05:11:57 +0000417 void *p = state.getGlobalAddressMap(locked)[GV];
418 if (p)
419 return p;
420
421 // Global variable might have been added since interpreter started.
422 if (GlobalVariable *GVar =
423 const_cast<GlobalVariable *>(dyn_cast<GlobalVariable>(GV)))
424 EmitGlobalVariable(GVar);
425 else
Chris Lattner64f150f2007-02-14 06:20:04 +0000426 assert(0 && "Global hasn't had an address allocated yet!");
Reid Spenceree448632005-07-12 15:51:55 +0000427 return state.getGlobalAddressMap(locked)[GV];
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000428}
429
Reid Spencer3da59db2006-11-27 01:05:10 +0000430/// This function converts a Constant* into a GenericValue. The interesting
431/// part is if C is a ConstantExpr.
Reid Spencerba28cb92007-08-11 15:57:56 +0000432/// @brief Get a GenericValue for a Constant*
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000433GenericValue ExecutionEngine::getConstantValue(const Constant *C) {
Reid Spencer3da59db2006-11-27 01:05:10 +0000434 // If its undefined, return the garbage.
Reid Spencerbce30f12007-03-06 22:23:15 +0000435 if (isa<UndefValue>(C))
436 return GenericValue();
Chris Lattnerd8c03bf2003-04-23 19:01:49 +0000437
Reid Spencer3da59db2006-11-27 01:05:10 +0000438 // If the value is a ConstantExpr
439 if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) {
Reid Spencerbce30f12007-03-06 22:23:15 +0000440 Constant *Op0 = CE->getOperand(0);
Chris Lattnerd8c03bf2003-04-23 19:01:49 +0000441 switch (CE->getOpcode()) {
442 case Instruction::GetElementPtr: {
Reid Spencer3da59db2006-11-27 01:05:10 +0000443 // Compute the index
Reid Spencerbce30f12007-03-06 22:23:15 +0000444 GenericValue Result = getConstantValue(Op0);
Chris Lattner829621c2007-02-10 20:35:22 +0000445 SmallVector<Value*, 8> Indices(CE->op_begin()+1, CE->op_end());
Chris Lattnerd8c03bf2003-04-23 19:01:49 +0000446 uint64_t Offset =
Reid Spencerbce30f12007-03-06 22:23:15 +0000447 TD->getIndexedOffset(Op0->getType(), &Indices[0], Indices.size());
Misha Brukmanedf128a2005-04-21 22:36:52 +0000448
Reid Spencer8fb0f192007-03-06 03:04:04 +0000449 char* tmp = (char*) Result.PointerVal;
450 Result = PTOGV(tmp + Offset);
Chris Lattnerd8c03bf2003-04-23 19:01:49 +0000451 return Result;
452 }
Reid Spencerbce30f12007-03-06 22:23:15 +0000453 case Instruction::Trunc: {
454 GenericValue GV = getConstantValue(Op0);
455 uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
456 GV.IntVal = GV.IntVal.trunc(BitWidth);
457 return GV;
458 }
459 case Instruction::ZExt: {
460 GenericValue GV = getConstantValue(Op0);
461 uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
462 GV.IntVal = GV.IntVal.zext(BitWidth);
463 return GV;
464 }
465 case Instruction::SExt: {
466 GenericValue GV = getConstantValue(Op0);
467 uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
468 GV.IntVal = GV.IntVal.sext(BitWidth);
469 return GV;
470 }
471 case Instruction::FPTrunc: {
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000472 // FIXME long double
Reid Spencerbce30f12007-03-06 22:23:15 +0000473 GenericValue GV = getConstantValue(Op0);
474 GV.FloatVal = float(GV.DoubleVal);
475 return GV;
476 }
477 case Instruction::FPExt:{
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000478 // FIXME long double
Reid Spencerbce30f12007-03-06 22:23:15 +0000479 GenericValue GV = getConstantValue(Op0);
480 GV.DoubleVal = double(GV.FloatVal);
481 return GV;
482 }
483 case Instruction::UIToFP: {
484 GenericValue GV = getConstantValue(Op0);
485 if (CE->getType() == Type::FloatTy)
486 GV.FloatVal = float(GV.IntVal.roundToDouble());
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000487 else if (CE->getType() == Type::DoubleTy)
Reid Spencerbce30f12007-03-06 22:23:15 +0000488 GV.DoubleVal = GV.IntVal.roundToDouble();
Dale Johannesen910993e2007-09-21 22:09:37 +0000489 else if (CE->getType() == Type::X86_FP80Ty) {
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000490 const uint64_t zero[] = {0, 0};
491 APFloat apf = APFloat(APInt(80, 2, zero));
Dan Gohman62824062008-02-29 01:27:13 +0000492 (void)apf.convertFromAPInt(GV.IntVal,
493 false,
494 APFloat::rmNearestTiesToEven);
Dale Johannesen7111b022008-10-09 18:53:47 +0000495 GV.IntVal = apf.bitcastToAPInt();
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000496 }
Reid Spencerbce30f12007-03-06 22:23:15 +0000497 return GV;
498 }
499 case Instruction::SIToFP: {
500 GenericValue GV = getConstantValue(Op0);
501 if (CE->getType() == Type::FloatTy)
502 GV.FloatVal = float(GV.IntVal.signedRoundToDouble());
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000503 else if (CE->getType() == Type::DoubleTy)
Reid Spencerbce30f12007-03-06 22:23:15 +0000504 GV.DoubleVal = GV.IntVal.signedRoundToDouble();
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000505 else if (CE->getType() == Type::X86_FP80Ty) {
506 const uint64_t zero[] = { 0, 0};
507 APFloat apf = APFloat(APInt(80, 2, zero));
Dan Gohman62824062008-02-29 01:27:13 +0000508 (void)apf.convertFromAPInt(GV.IntVal,
509 true,
510 APFloat::rmNearestTiesToEven);
Dale Johannesen7111b022008-10-09 18:53:47 +0000511 GV.IntVal = apf.bitcastToAPInt();
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000512 }
Reid Spencerbce30f12007-03-06 22:23:15 +0000513 return GV;
514 }
515 case Instruction::FPToUI: // double->APInt conversion handles sign
516 case Instruction::FPToSI: {
517 GenericValue GV = getConstantValue(Op0);
518 uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
519 if (Op0->getType() == Type::FloatTy)
520 GV.IntVal = APIntOps::RoundFloatToAPInt(GV.FloatVal, BitWidth);
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000521 else if (Op0->getType() == Type::DoubleTy)
Reid Spencerbce30f12007-03-06 22:23:15 +0000522 GV.IntVal = APIntOps::RoundDoubleToAPInt(GV.DoubleVal, BitWidth);
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000523 else if (Op0->getType() == Type::X86_FP80Ty) {
524 APFloat apf = APFloat(GV.IntVal);
525 uint64_t v;
Dale Johannesen23a98552008-10-09 23:00:39 +0000526 bool ignored;
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000527 (void)apf.convertToInteger(&v, BitWidth,
528 CE->getOpcode()==Instruction::FPToSI,
Dale Johannesen23a98552008-10-09 23:00:39 +0000529 APFloat::rmTowardZero, &ignored);
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000530 GV.IntVal = v; // endian?
531 }
Reid Spencerbce30f12007-03-06 22:23:15 +0000532 return GV;
533 }
Reid Spencer3da59db2006-11-27 01:05:10 +0000534 case Instruction::PtrToInt: {
Reid Spencerbce30f12007-03-06 22:23:15 +0000535 GenericValue GV = getConstantValue(Op0);
536 uint32_t PtrWidth = TD->getPointerSizeInBits();
537 GV.IntVal = APInt(PtrWidth, uintptr_t(GV.PointerVal));
538 return GV;
539 }
540 case Instruction::IntToPtr: {
541 GenericValue GV = getConstantValue(Op0);
542 uint32_t PtrWidth = TD->getPointerSizeInBits();
543 if (PtrWidth != GV.IntVal.getBitWidth())
544 GV.IntVal = GV.IntVal.zextOrTrunc(PtrWidth);
545 assert(GV.IntVal.getBitWidth() <= 64 && "Bad pointer width");
546 GV.PointerVal = PointerTy(uintptr_t(GV.IntVal.getZExtValue()));
Reid Spencer3da59db2006-11-27 01:05:10 +0000547 return GV;
548 }
549 case Instruction::BitCast: {
Reid Spencerbce30f12007-03-06 22:23:15 +0000550 GenericValue GV = getConstantValue(Op0);
551 const Type* DestTy = CE->getType();
552 switch (Op0->getType()->getTypeID()) {
553 default: assert(0 && "Invalid bitcast operand");
554 case Type::IntegerTyID:
555 assert(DestTy->isFloatingPoint() && "invalid bitcast");
556 if (DestTy == Type::FloatTy)
557 GV.FloatVal = GV.IntVal.bitsToFloat();
558 else if (DestTy == Type::DoubleTy)
559 GV.DoubleVal = GV.IntVal.bitsToDouble();
560 break;
561 case Type::FloatTyID:
562 assert(DestTy == Type::Int32Ty && "Invalid bitcast");
563 GV.IntVal.floatToBits(GV.FloatVal);
564 break;
565 case Type::DoubleTyID:
566 assert(DestTy == Type::Int64Ty && "Invalid bitcast");
567 GV.IntVal.doubleToBits(GV.DoubleVal);
568 break;
569 case Type::PointerTyID:
570 assert(isa<PointerType>(DestTy) && "Invalid bitcast");
571 break; // getConstantValue(Op0) above already converted it
572 }
573 return GV;
Chris Lattnerd8c03bf2003-04-23 19:01:49 +0000574 }
Chris Lattner9a231222003-05-14 17:51:49 +0000575 case Instruction::Add:
Reid Spencerbce30f12007-03-06 22:23:15 +0000576 case Instruction::Sub:
577 case Instruction::Mul:
578 case Instruction::UDiv:
579 case Instruction::SDiv:
580 case Instruction::URem:
581 case Instruction::SRem:
582 case Instruction::And:
583 case Instruction::Or:
584 case Instruction::Xor: {
585 GenericValue LHS = getConstantValue(Op0);
586 GenericValue RHS = getConstantValue(CE->getOperand(1));
587 GenericValue GV;
Chris Lattner5f90cb82004-07-11 08:01:11 +0000588 switch (CE->getOperand(0)->getType()->getTypeID()) {
589 default: assert(0 && "Bad add type!"); abort();
Reid Spencera54b7cb2007-01-12 07:05:14 +0000590 case Type::IntegerTyID:
Reid Spencerbce30f12007-03-06 22:23:15 +0000591 switch (CE->getOpcode()) {
592 default: assert(0 && "Invalid integer opcode");
593 case Instruction::Add: GV.IntVal = LHS.IntVal + RHS.IntVal; break;
594 case Instruction::Sub: GV.IntVal = LHS.IntVal - RHS.IntVal; break;
595 case Instruction::Mul: GV.IntVal = LHS.IntVal * RHS.IntVal; break;
596 case Instruction::UDiv:GV.IntVal = LHS.IntVal.udiv(RHS.IntVal); break;
597 case Instruction::SDiv:GV.IntVal = LHS.IntVal.sdiv(RHS.IntVal); break;
598 case Instruction::URem:GV.IntVal = LHS.IntVal.urem(RHS.IntVal); break;
599 case Instruction::SRem:GV.IntVal = LHS.IntVal.srem(RHS.IntVal); break;
600 case Instruction::And: GV.IntVal = LHS.IntVal & RHS.IntVal; break;
601 case Instruction::Or: GV.IntVal = LHS.IntVal | RHS.IntVal; break;
602 case Instruction::Xor: GV.IntVal = LHS.IntVal ^ RHS.IntVal; break;
603 }
Chris Lattner5f90cb82004-07-11 08:01:11 +0000604 break;
605 case Type::FloatTyID:
Reid Spencerbce30f12007-03-06 22:23:15 +0000606 switch (CE->getOpcode()) {
607 default: assert(0 && "Invalid float opcode"); abort();
608 case Instruction::Add:
609 GV.FloatVal = LHS.FloatVal + RHS.FloatVal; break;
610 case Instruction::Sub:
611 GV.FloatVal = LHS.FloatVal - RHS.FloatVal; break;
612 case Instruction::Mul:
613 GV.FloatVal = LHS.FloatVal * RHS.FloatVal; break;
614 case Instruction::FDiv:
615 GV.FloatVal = LHS.FloatVal / RHS.FloatVal; break;
616 case Instruction::FRem:
617 GV.FloatVal = ::fmodf(LHS.FloatVal,RHS.FloatVal); break;
618 }
Chris Lattner5f90cb82004-07-11 08:01:11 +0000619 break;
620 case Type::DoubleTyID:
Reid Spencerbce30f12007-03-06 22:23:15 +0000621 switch (CE->getOpcode()) {
622 default: assert(0 && "Invalid double opcode"); abort();
623 case Instruction::Add:
624 GV.DoubleVal = LHS.DoubleVal + RHS.DoubleVal; break;
625 case Instruction::Sub:
626 GV.DoubleVal = LHS.DoubleVal - RHS.DoubleVal; break;
627 case Instruction::Mul:
628 GV.DoubleVal = LHS.DoubleVal * RHS.DoubleVal; break;
629 case Instruction::FDiv:
630 GV.DoubleVal = LHS.DoubleVal / RHS.DoubleVal; break;
631 case Instruction::FRem:
632 GV.DoubleVal = ::fmod(LHS.DoubleVal,RHS.DoubleVal); break;
633 }
Chris Lattner5f90cb82004-07-11 08:01:11 +0000634 break;
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000635 case Type::X86_FP80TyID:
636 case Type::PPC_FP128TyID:
637 case Type::FP128TyID: {
638 APFloat apfLHS = APFloat(LHS.IntVal);
639 switch (CE->getOpcode()) {
640 default: assert(0 && "Invalid long double opcode"); abort();
641 case Instruction::Add:
642 apfLHS.add(APFloat(RHS.IntVal), APFloat::rmNearestTiesToEven);
Dale Johannesen7111b022008-10-09 18:53:47 +0000643 GV.IntVal = apfLHS.bitcastToAPInt();
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000644 break;
645 case Instruction::Sub:
646 apfLHS.subtract(APFloat(RHS.IntVal), APFloat::rmNearestTiesToEven);
Dale Johannesen7111b022008-10-09 18:53:47 +0000647 GV.IntVal = apfLHS.bitcastToAPInt();
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000648 break;
649 case Instruction::Mul:
650 apfLHS.multiply(APFloat(RHS.IntVal), APFloat::rmNearestTiesToEven);
Dale Johannesen7111b022008-10-09 18:53:47 +0000651 GV.IntVal = apfLHS.bitcastToAPInt();
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000652 break;
653 case Instruction::FDiv:
654 apfLHS.divide(APFloat(RHS.IntVal), APFloat::rmNearestTiesToEven);
Dale Johannesen7111b022008-10-09 18:53:47 +0000655 GV.IntVal = apfLHS.bitcastToAPInt();
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000656 break;
657 case Instruction::FRem:
658 apfLHS.mod(APFloat(RHS.IntVal), APFloat::rmNearestTiesToEven);
Dale Johannesen7111b022008-10-09 18:53:47 +0000659 GV.IntVal = apfLHS.bitcastToAPInt();
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000660 break;
661 }
662 }
663 break;
Chris Lattner5f90cb82004-07-11 08:01:11 +0000664 }
Reid Spencerbce30f12007-03-06 22:23:15 +0000665 return GV;
666 }
Chris Lattner9a231222003-05-14 17:51:49 +0000667 default:
668 break;
669 }
Reid Spencerbce30f12007-03-06 22:23:15 +0000670 cerr << "ConstantExpr not handled: " << *CE << "\n";
Chris Lattner9a231222003-05-14 17:51:49 +0000671 abort();
672 }
Misha Brukmanedf128a2005-04-21 22:36:52 +0000673
Reid Spencerbce30f12007-03-06 22:23:15 +0000674 GenericValue Result;
Chris Lattnerf70c22b2004-06-17 18:19:28 +0000675 switch (C->getType()->getTypeID()) {
Reid Spencer8fb0f192007-03-06 03:04:04 +0000676 case Type::FloatTyID:
Dale Johannesen43421b32007-09-06 18:13:44 +0000677 Result.FloatVal = cast<ConstantFP>(C)->getValueAPF().convertToFloat();
Reid Spencera54b7cb2007-01-12 07:05:14 +0000678 break;
Reid Spencer8fb0f192007-03-06 03:04:04 +0000679 case Type::DoubleTyID:
Dale Johannesen43421b32007-09-06 18:13:44 +0000680 Result.DoubleVal = cast<ConstantFP>(C)->getValueAPF().convertToDouble();
Reid Spencer8fb0f192007-03-06 03:04:04 +0000681 break;
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000682 case Type::X86_FP80TyID:
683 case Type::FP128TyID:
684 case Type::PPC_FP128TyID:
Dale Johannesen7111b022008-10-09 18:53:47 +0000685 Result.IntVal = cast <ConstantFP>(C)->getValueAPF().bitcastToAPInt();
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000686 break;
Reid Spencer8fb0f192007-03-06 03:04:04 +0000687 case Type::IntegerTyID:
688 Result.IntVal = cast<ConstantInt>(C)->getValue();
689 break;
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000690 case Type::PointerTyID:
Reid Spencer40cf2f92004-07-18 00:41:27 +0000691 if (isa<ConstantPointerNull>(C))
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000692 Result.PointerVal = 0;
Reid Spencer40cf2f92004-07-18 00:41:27 +0000693 else if (const Function *F = dyn_cast<Function>(C))
694 Result = PTOGV(getPointerToFunctionOrStub(const_cast<Function*>(F)));
695 else if (const GlobalVariable* GV = dyn_cast<GlobalVariable>(C))
696 Result = PTOGV(getOrEmitGlobalVariable(const_cast<GlobalVariable*>(GV)));
697 else
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000698 assert(0 && "Unknown constant pointer type!");
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000699 break;
700 default:
Reid Spencerbce30f12007-03-06 22:23:15 +0000701 cerr << "ERROR: Constant unimplemented for type: " << *C->getType() << "\n";
Chris Lattnerd8c03bf2003-04-23 19:01:49 +0000702 abort();
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000703 }
704 return Result;
705}
706
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000707/// StoreIntToMemory - Fills the StoreBytes bytes of memory starting from Dst
708/// with the integer held in IntVal.
709static void StoreIntToMemory(const APInt &IntVal, uint8_t *Dst,
710 unsigned StoreBytes) {
711 assert((IntVal.getBitWidth()+7)/8 >= StoreBytes && "Integer too small!");
712 uint8_t *Src = (uint8_t *)IntVal.getRawData();
713
Chris Lattnerb67c9582009-01-22 19:53:00 +0000714 if (sys::isLittleEndianHost())
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000715 // Little-endian host - the source is ordered from LSB to MSB. Order the
716 // destination from LSB to MSB: Do a straight copy.
717 memcpy(Dst, Src, StoreBytes);
718 else {
719 // Big-endian host - the source is an array of 64 bit words ordered from
720 // LSW to MSW. Each word is ordered from MSB to LSB. Order the destination
721 // from MSB to LSB: Reverse the word order, but not the bytes in a word.
722 while (StoreBytes > sizeof(uint64_t)) {
723 StoreBytes -= sizeof(uint64_t);
724 // May not be aligned so use memcpy.
725 memcpy(Dst + StoreBytes, Src, sizeof(uint64_t));
726 Src += sizeof(uint64_t);
727 }
728
729 memcpy(Dst, Src + sizeof(uint64_t) - StoreBytes, StoreBytes);
730 }
731}
732
Nate Begeman37efe672006-04-22 18:53:45 +0000733/// StoreValueToMemory - Stores the data in Val of type Ty at address Ptr. Ptr
734/// is the address of the memory at which to store Val, cast to GenericValue *.
735/// It is not a pointer to a GenericValue containing the address at which to
736/// store Val.
Evan Cheng89687e32008-11-04 06:10:31 +0000737void ExecutionEngine::StoreValueToMemory(const GenericValue &Val,
738 GenericValue *Ptr, const Type *Ty) {
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000739 const unsigned StoreBytes = getTargetData()->getTypeStoreSize(Ty);
740
Reid Spencer8fb0f192007-03-06 03:04:04 +0000741 switch (Ty->getTypeID()) {
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000742 case Type::IntegerTyID:
743 StoreIntToMemory(Val.IntVal, (uint8_t*)Ptr, StoreBytes);
Reid Spencer8fb0f192007-03-06 03:04:04 +0000744 break;
Reid Spencer8fb0f192007-03-06 03:04:04 +0000745 case Type::FloatTyID:
746 *((float*)Ptr) = Val.FloatVal;
747 break;
748 case Type::DoubleTyID:
749 *((double*)Ptr) = Val.DoubleVal;
750 break;
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000751 case Type::X86_FP80TyID: {
752 uint16_t *Dest = (uint16_t*)Ptr;
753 const uint16_t *Src = (uint16_t*)Val.IntVal.getRawData();
754 // This is endian dependent, but it will only work on x86 anyway.
755 Dest[0] = Src[4];
756 Dest[1] = Src[0];
757 Dest[2] = Src[1];
758 Dest[3] = Src[2];
759 Dest[4] = Src[3];
760 break;
761 }
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000762 case Type::PointerTyID:
763 // Ensure 64 bit target pointers are fully initialized on 32 bit hosts.
764 if (StoreBytes != sizeof(PointerTy))
765 memset(Ptr, 0, StoreBytes);
766
Reid Spencer8fb0f192007-03-06 03:04:04 +0000767 *((PointerTy*)Ptr) = Val.PointerVal;
768 break;
769 default:
770 cerr << "Cannot store value of type " << *Ty << "!\n";
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000771 }
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000772
Chris Lattnerb67c9582009-01-22 19:53:00 +0000773 if (sys::isLittleEndianHost() != getTargetData()->isLittleEndian())
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000774 // Host and target are different endian - reverse the stored bytes.
775 std::reverse((uint8_t*)Ptr, StoreBytes + (uint8_t*)Ptr);
776}
777
778/// LoadIntFromMemory - Loads the integer stored in the LoadBytes bytes starting
779/// from Src into IntVal, which is assumed to be wide enough and to hold zero.
780static void LoadIntFromMemory(APInt &IntVal, uint8_t *Src, unsigned LoadBytes) {
781 assert((IntVal.getBitWidth()+7)/8 >= LoadBytes && "Integer too small!");
782 uint8_t *Dst = (uint8_t *)IntVal.getRawData();
783
Chris Lattnerb67c9582009-01-22 19:53:00 +0000784 if (sys::isLittleEndianHost())
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000785 // Little-endian host - the destination must be ordered from LSB to MSB.
786 // The source is ordered from LSB to MSB: Do a straight copy.
787 memcpy(Dst, Src, LoadBytes);
788 else {
789 // Big-endian - the destination is an array of 64 bit words ordered from
790 // LSW to MSW. Each word must be ordered from MSB to LSB. The source is
791 // ordered from MSB to LSB: Reverse the word order, but not the bytes in
792 // a word.
793 while (LoadBytes > sizeof(uint64_t)) {
794 LoadBytes -= sizeof(uint64_t);
795 // May not be aligned so use memcpy.
796 memcpy(Dst, Src + LoadBytes, sizeof(uint64_t));
797 Dst += sizeof(uint64_t);
798 }
799
800 memcpy(Dst + sizeof(uint64_t) - LoadBytes, Src, LoadBytes);
801 }
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000802}
803
Misha Brukman4afac182003-10-10 17:45:12 +0000804/// FIXME: document
805///
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000806void ExecutionEngine::LoadValueFromMemory(GenericValue &Result,
Duncan Sands08bfe262008-03-10 16:38:37 +0000807 GenericValue *Ptr,
808 const Type *Ty) {
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000809 const unsigned LoadBytes = getTargetData()->getTypeStoreSize(Ty);
Duncan Sands1eff7042007-12-10 17:43:13 +0000810
Chris Lattnerb67c9582009-01-22 19:53:00 +0000811 if (sys::isLittleEndianHost() != getTargetData()->isLittleEndian()) {
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000812 // Host and target are different endian - reverse copy the stored
813 // bytes into a buffer, and load from that.
814 uint8_t *Src = (uint8_t*)Ptr;
815 uint8_t *Buf = (uint8_t*)alloca(LoadBytes);
816 std::reverse_copy(Src, Src + LoadBytes, Buf);
817 Ptr = (GenericValue*)Buf;
Reid Spencer8fb0f192007-03-06 03:04:04 +0000818 }
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000819
820 switch (Ty->getTypeID()) {
821 case Type::IntegerTyID:
822 // An APInt with all words initially zero.
823 Result.IntVal = APInt(cast<IntegerType>(Ty)->getBitWidth(), 0);
824 LoadIntFromMemory(Result.IntVal, (uint8_t*)Ptr, LoadBytes);
825 break;
Reid Spencer8fb0f192007-03-06 03:04:04 +0000826 case Type::FloatTyID:
827 Result.FloatVal = *((float*)Ptr);
828 break;
829 case Type::DoubleTyID:
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000830 Result.DoubleVal = *((double*)Ptr);
Reid Spencer8fb0f192007-03-06 03:04:04 +0000831 break;
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000832 case Type::PointerTyID:
Reid Spencer8fb0f192007-03-06 03:04:04 +0000833 Result.PointerVal = *((PointerTy*)Ptr);
834 break;
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000835 case Type::X86_FP80TyID: {
836 // This is endian dependent, but it will only work on x86 anyway.
Duncan Sands9e4635a2007-12-15 17:37:40 +0000837 // FIXME: Will not trap if loading a signaling NaN.
Duncan Sandsdd65a732007-11-28 10:36:19 +0000838 uint16_t *p = (uint16_t*)Ptr;
839 union {
840 uint16_t x[8];
841 uint64_t y[2];
842 };
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000843 x[0] = p[1];
844 x[1] = p[2];
845 x[2] = p[3];
846 x[3] = p[4];
847 x[4] = p[0];
Duncan Sandsdd65a732007-11-28 10:36:19 +0000848 Result.IntVal = APInt(80, 2, y);
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000849 break;
850 }
Reid Spencer8fb0f192007-03-06 03:04:04 +0000851 default:
852 cerr << "Cannot load value of type " << *Ty << "!\n";
853 abort();
Chris Lattnerf88b9a62003-05-08 16:52:16 +0000854 }
Chris Lattnerf88b9a62003-05-08 16:52:16 +0000855}
856
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000857// InitializeMemory - Recursive function to apply a Constant value into the
858// specified memory location...
859//
860void ExecutionEngine::InitializeMemory(const Constant *Init, void *Addr) {
Evan Chengeb5d95a2008-11-06 17:46:04 +0000861 DOUT << "JIT: Initializing " << Addr << " ";
Dale Johannesendd947ea2008-08-07 01:30:15 +0000862 DEBUG(Init->dump());
Chris Lattnerbd1d3822004-10-16 18:19:26 +0000863 if (isa<UndefValue>(Init)) {
864 return;
Reid Spencer9d6565a2007-02-15 02:26:10 +0000865 } else if (const ConstantVector *CP = dyn_cast<ConstantVector>(Init)) {
Robert Bocchino7c2b7c72006-01-20 18:18:40 +0000866 unsigned ElementSize =
Duncan Sandsceb4d1a2009-01-12 20:38:59 +0000867 getTargetData()->getTypePaddedSize(CP->getType()->getElementType());
Robert Bocchino7c2b7c72006-01-20 18:18:40 +0000868 for (unsigned i = 0, e = CP->getNumOperands(); i != e; ++i)
869 InitializeMemory(CP->getOperand(i), (char*)Addr+i*ElementSize);
870 return;
Chris Lattnerb6e1dd72008-02-15 00:57:28 +0000871 } else if (isa<ConstantAggregateZero>(Init)) {
Duncan Sandsceb4d1a2009-01-12 20:38:59 +0000872 memset(Addr, 0, (size_t)getTargetData()->getTypePaddedSize(Init->getType()));
Chris Lattnerb6e1dd72008-02-15 00:57:28 +0000873 return;
Dan Gohman638e3782008-05-20 03:20:09 +0000874 } else if (const ConstantArray *CPA = dyn_cast<ConstantArray>(Init)) {
875 unsigned ElementSize =
Duncan Sandsceb4d1a2009-01-12 20:38:59 +0000876 getTargetData()->getTypePaddedSize(CPA->getType()->getElementType());
Dan Gohman638e3782008-05-20 03:20:09 +0000877 for (unsigned i = 0, e = CPA->getNumOperands(); i != e; ++i)
878 InitializeMemory(CPA->getOperand(i), (char*)Addr+i*ElementSize);
879 return;
880 } else if (const ConstantStruct *CPS = dyn_cast<ConstantStruct>(Init)) {
881 const StructLayout *SL =
882 getTargetData()->getStructLayout(cast<StructType>(CPS->getType()));
883 for (unsigned i = 0, e = CPS->getNumOperands(); i != e; ++i)
884 InitializeMemory(CPS->getOperand(i), (char*)Addr+SL->getElementOffset(i));
885 return;
Chris Lattnerbd1d3822004-10-16 18:19:26 +0000886 } else if (Init->getType()->isFirstClassType()) {
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000887 GenericValue Val = getConstantValue(Init);
888 StoreValueToMemory(Val, (GenericValue*)Addr, Init->getType());
889 return;
890 }
891
Dan Gohman638e3782008-05-20 03:20:09 +0000892 cerr << "Bad Type: " << *Init->getType() << "\n";
893 assert(0 && "Unknown constant type to initialize memory with!");
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000894}
895
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000896/// EmitGlobals - Emit all of the global variables to memory, storing their
897/// addresses into GlobalAddress. This must make sure to copy the contents of
898/// their initializers into the memory.
899///
900void ExecutionEngine::emitGlobals() {
Misha Brukmanedf128a2005-04-21 22:36:52 +0000901
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000902 // Loop over all of the global variables in the program, allocating the memory
Chris Lattnerfe854032006-08-16 01:24:12 +0000903 // to hold them. If there is more than one module, do a prepass over globals
904 // to figure out how the different modules should link together.
905 //
906 std::map<std::pair<std::string, const Type*>,
907 const GlobalValue*> LinkedGlobalsMap;
Misha Brukmanedf128a2005-04-21 22:36:52 +0000908
Chris Lattnerfe854032006-08-16 01:24:12 +0000909 if (Modules.size() != 1) {
910 for (unsigned m = 0, e = Modules.size(); m != e; ++m) {
911 Module &M = *Modules[m]->getModule();
912 for (Module::const_global_iterator I = M.global_begin(),
913 E = M.global_end(); I != E; ++I) {
914 const GlobalValue *GV = I;
Rafael Espindolabb46f522009-01-15 20:18:42 +0000915 if (GV->hasLocalLinkage() || GV->isDeclaration() ||
Chris Lattnerfe854032006-08-16 01:24:12 +0000916 GV->hasAppendingLinkage() || !GV->hasName())
917 continue;// Ignore external globals and globals with internal linkage.
918
919 const GlobalValue *&GVEntry =
920 LinkedGlobalsMap[std::make_pair(GV->getName(), GV->getType())];
921
922 // If this is the first time we've seen this global, it is the canonical
923 // version.
924 if (!GVEntry) {
925 GVEntry = GV;
926 continue;
927 }
928
929 // If the existing global is strong, never replace it.
Anton Korobeynikovb74ed072006-09-14 18:23:27 +0000930 if (GVEntry->hasExternalLinkage() ||
931 GVEntry->hasDLLImportLinkage() ||
932 GVEntry->hasDLLExportLinkage())
Chris Lattnerfe854032006-08-16 01:24:12 +0000933 continue;
934
935 // Otherwise, we know it's linkonce/weak, replace it if this is a strong
Dale Johannesenaafce772008-05-14 20:12:51 +0000936 // symbol. FIXME is this right for common?
Anton Korobeynikov78ee7b72006-12-01 00:25:12 +0000937 if (GV->hasExternalLinkage() || GVEntry->hasExternalWeakLinkage())
Chris Lattnerfe854032006-08-16 01:24:12 +0000938 GVEntry = GV;
Chris Lattnerd8c03bf2003-04-23 19:01:49 +0000939 }
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000940 }
Chris Lattnerfe854032006-08-16 01:24:12 +0000941 }
942
943 std::vector<const GlobalValue*> NonCanonicalGlobals;
944 for (unsigned m = 0, e = Modules.size(); m != e; ++m) {
945 Module &M = *Modules[m]->getModule();
946 for (Module::const_global_iterator I = M.global_begin(), E = M.global_end();
947 I != E; ++I) {
948 // In the multi-module case, see what this global maps to.
949 if (!LinkedGlobalsMap.empty()) {
950 if (const GlobalValue *GVEntry =
951 LinkedGlobalsMap[std::make_pair(I->getName(), I->getType())]) {
952 // If something else is the canonical global, ignore this one.
953 if (GVEntry != &*I) {
954 NonCanonicalGlobals.push_back(I);
955 continue;
956 }
957 }
958 }
959
Reid Spencer5cbf9852007-01-30 20:08:39 +0000960 if (!I->isDeclaration()) {
Nicolas Geoffray46fa1392008-10-25 15:41:43 +0000961 addGlobalMapping(I, getMemoryForGV(I));
Chris Lattnerfe854032006-08-16 01:24:12 +0000962 } else {
963 // External variable reference. Try to use the dynamic loader to
964 // get a pointer to it.
965 if (void *SymAddr =
966 sys::DynamicLibrary::SearchForAddressOfSymbol(I->getName().c_str()))
967 addGlobalMapping(I, SymAddr);
968 else {
Bill Wendlinge8156192006-12-07 01:30:32 +0000969 cerr << "Could not resolve external global address: "
970 << I->getName() << "\n";
Chris Lattnerfe854032006-08-16 01:24:12 +0000971 abort();
972 }
973 }
974 }
975
976 // If there are multiple modules, map the non-canonical globals to their
977 // canonical location.
978 if (!NonCanonicalGlobals.empty()) {
979 for (unsigned i = 0, e = NonCanonicalGlobals.size(); i != e; ++i) {
980 const GlobalValue *GV = NonCanonicalGlobals[i];
981 const GlobalValue *CGV =
982 LinkedGlobalsMap[std::make_pair(GV->getName(), GV->getType())];
983 void *Ptr = getPointerToGlobalIfAvailable(CGV);
984 assert(Ptr && "Canonical global wasn't codegen'd!");
Nuno Lopesc8ed9022008-10-14 10:04:52 +0000985 addGlobalMapping(GV, Ptr);
Chris Lattnerfe854032006-08-16 01:24:12 +0000986 }
987 }
988
Reid Spencera54b7cb2007-01-12 07:05:14 +0000989 // Now that all of the globals are set up in memory, loop through them all
990 // and initialize their contents.
Chris Lattnerfe854032006-08-16 01:24:12 +0000991 for (Module::const_global_iterator I = M.global_begin(), E = M.global_end();
992 I != E; ++I) {
Reid Spencer5cbf9852007-01-30 20:08:39 +0000993 if (!I->isDeclaration()) {
Chris Lattnerfe854032006-08-16 01:24:12 +0000994 if (!LinkedGlobalsMap.empty()) {
995 if (const GlobalValue *GVEntry =
996 LinkedGlobalsMap[std::make_pair(I->getName(), I->getType())])
997 if (GVEntry != &*I) // Not the canonical variable.
998 continue;
999 }
1000 EmitGlobalVariable(I);
1001 }
1002 }
1003 }
Chris Lattner24b0a182003-12-20 02:45:37 +00001004}
1005
1006// EmitGlobalVariable - This method emits the specified global variable to the
1007// address specified in GlobalAddresses, or allocates new memory if it's not
1008// already in the map.
Chris Lattnerc07ed132003-12-20 03:36:47 +00001009void ExecutionEngine::EmitGlobalVariable(const GlobalVariable *GV) {
Chris Lattner55d86482003-12-31 20:21:04 +00001010 void *GA = getPointerToGlobalIfAvailable(GV);
Chris Lattner23c47242004-02-08 19:33:23 +00001011
Chris Lattner24b0a182003-12-20 02:45:37 +00001012 if (GA == 0) {
1013 // If it's not already specified, allocate memory for the global.
Nicolas Geoffray46fa1392008-10-25 15:41:43 +00001014 GA = getMemoryForGV(GV);
Chris Lattner55d86482003-12-31 20:21:04 +00001015 addGlobalMapping(GV, GA);
Chris Lattner24b0a182003-12-20 02:45:37 +00001016 }
Nicolas Geoffray46fa1392008-10-25 15:41:43 +00001017
1018 // Don't initialize if it's thread local, let the client do it.
1019 if (!GV->isThreadLocal())
1020 InitializeMemory(GV->getInitializer(), GA);
1021
1022 const Type *ElTy = GV->getType()->getElementType();
Duncan Sandsceb4d1a2009-01-12 20:38:59 +00001023 size_t GVSize = (size_t)getTargetData()->getTypePaddedSize(ElTy);
Chris Lattner813c8152005-01-08 20:13:19 +00001024 NumInitBytes += (unsigned)GVSize;
Chris Lattner24b0a182003-12-20 02:45:37 +00001025 ++NumGlobals;
Chris Lattnerbd199fb2002-12-24 00:01:05 +00001026}