blob: 2a7b914ac3447594938bec59afd81ce8ebdfb762 [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 Cheng1b088f32008-06-17 16:49:02 +000043 SymbolSearchingDisabled = false;
Chris Lattnerfe854032006-08-16 01:24:12 +000044 Modules.push_back(P);
Misha Brukman19684162003-10-16 21:18:05 +000045 assert(P && "ModuleProvider is null?");
46}
47
Brian Gaeke8e539482003-09-04 22:57:27 +000048ExecutionEngine::~ExecutionEngine() {
Reid Spencerd4c0e622007-03-03 18:19:18 +000049 clearAllGlobalMappings();
Chris Lattnerfe854032006-08-16 01:24:12 +000050 for (unsigned i = 0, e = Modules.size(); i != e; ++i)
51 delete Modules[i];
Brian Gaeke8e539482003-09-04 22:57:27 +000052}
53
Devang Patel73d0e212007-10-15 19:56:32 +000054/// removeModuleProvider - Remove a ModuleProvider from the list of modules.
55/// Release module from ModuleProvider.
56Module* ExecutionEngine::removeModuleProvider(ModuleProvider *P,
57 std::string *ErrInfo) {
58 for(SmallVector<ModuleProvider *, 1>::iterator I = Modules.begin(),
59 E = Modules.end(); I != E; ++I) {
60 ModuleProvider *MP = *I;
61 if (MP == P) {
62 Modules.erase(I);
Nate Begemanf049e072008-05-21 16:34:48 +000063 clearGlobalMappingsFromModule(MP->getModule());
Devang Patel73d0e212007-10-15 19:56:32 +000064 return MP->releaseModule(ErrInfo);
65 }
66 }
67 return NULL;
68}
69
Chris Lattnerfe854032006-08-16 01:24:12 +000070/// FindFunctionNamed - Search all of the active modules to find the one that
71/// defines FnName. This is very slow operation and shouldn't be used for
72/// general code.
73Function *ExecutionEngine::FindFunctionNamed(const char *FnName) {
74 for (unsigned i = 0, e = Modules.size(); i != e; ++i) {
Reid Spencer688b0492007-02-05 21:19:13 +000075 if (Function *F = Modules[i]->getModule()->getFunction(FnName))
Chris Lattnerfe854032006-08-16 01:24:12 +000076 return F;
77 }
78 return 0;
79}
80
81
Chris Lattnere7fd5532006-05-08 22:00:52 +000082/// addGlobalMapping - Tell the execution engine that the specified global is
83/// at the specified location. This is used internally as functions are JIT'd
84/// and as global variables are laid out in memory. It can and should also be
85/// used by clients of the EE that want to have an LLVM global overlay
86/// existing data in memory.
87void ExecutionEngine::addGlobalMapping(const GlobalValue *GV, void *Addr) {
88 MutexGuard locked(lock);
89
90 void *&CurVal = state.getGlobalAddressMap(locked)[GV];
91 assert((CurVal == 0 || Addr == 0) && "GlobalMapping already established!");
92 CurVal = Addr;
93
94 // If we are using the reverse mapping, add it too
95 if (!state.getGlobalAddressReverseMap(locked).empty()) {
96 const GlobalValue *&V = state.getGlobalAddressReverseMap(locked)[Addr];
97 assert((V == 0 || GV == 0) && "GlobalMapping already established!");
98 V = GV;
99 }
100}
101
102/// clearAllGlobalMappings - Clear all global mappings and start over again
103/// use in dynamic compilation scenarios when you want to move globals
104void ExecutionEngine::clearAllGlobalMappings() {
105 MutexGuard locked(lock);
106
107 state.getGlobalAddressMap(locked).clear();
108 state.getGlobalAddressReverseMap(locked).clear();
109}
110
Nate Begemanf049e072008-05-21 16:34:48 +0000111/// clearGlobalMappingsFromModule - Clear all global mappings that came from a
112/// particular module, because it has been removed from the JIT.
113void ExecutionEngine::clearGlobalMappingsFromModule(Module *M) {
114 MutexGuard locked(lock);
115
116 for (Module::iterator FI = M->begin(), FE = M->end(); FI != FE; ++FI) {
117 state.getGlobalAddressMap(locked).erase(FI);
118 state.getGlobalAddressReverseMap(locked).erase(FI);
119 }
120 for (Module::global_iterator GI = M->global_begin(), GE = M->global_end();
121 GI != GE; ++GI) {
122 state.getGlobalAddressMap(locked).erase(GI);
123 state.getGlobalAddressReverseMap(locked).erase(GI);
124 }
125}
126
Chris Lattnere7fd5532006-05-08 22:00:52 +0000127/// updateGlobalMapping - Replace an existing mapping for GV with a new
128/// address. This updates both maps as required. If "Addr" is null, the
129/// entry for the global is removed from the mappings.
Chris Lattnerf4cc3092008-04-04 04:47:41 +0000130void *ExecutionEngine::updateGlobalMapping(const GlobalValue *GV, void *Addr) {
Chris Lattnere7fd5532006-05-08 22:00:52 +0000131 MutexGuard locked(lock);
Chris Lattnerf4cc3092008-04-04 04:47:41 +0000132
133 std::map<const GlobalValue*, void *> &Map = state.getGlobalAddressMap(locked);
134
Chris Lattnere7fd5532006-05-08 22:00:52 +0000135 // Deleting from the mapping?
136 if (Addr == 0) {
Chris Lattnerf4cc3092008-04-04 04:47:41 +0000137 std::map<const GlobalValue*, void *>::iterator I = Map.find(GV);
138 void *OldVal;
139 if (I == Map.end())
140 OldVal = 0;
141 else {
142 OldVal = I->second;
143 Map.erase(I);
144 }
145
Chris Lattnere7fd5532006-05-08 22:00:52 +0000146 if (!state.getGlobalAddressReverseMap(locked).empty())
147 state.getGlobalAddressReverseMap(locked).erase(Addr);
Chris Lattnerf4cc3092008-04-04 04:47:41 +0000148 return OldVal;
Chris Lattnere7fd5532006-05-08 22:00:52 +0000149 }
150
Chris Lattnerf4cc3092008-04-04 04:47:41 +0000151 void *&CurVal = Map[GV];
152 void *OldVal = CurVal;
153
Chris Lattnere7fd5532006-05-08 22:00:52 +0000154 if (CurVal && !state.getGlobalAddressReverseMap(locked).empty())
155 state.getGlobalAddressReverseMap(locked).erase(CurVal);
156 CurVal = Addr;
157
158 // If we are using the reverse mapping, add it too
159 if (!state.getGlobalAddressReverseMap(locked).empty()) {
160 const GlobalValue *&V = state.getGlobalAddressReverseMap(locked)[Addr];
161 assert((V == 0 || GV == 0) && "GlobalMapping already established!");
162 V = GV;
163 }
Chris Lattnerf4cc3092008-04-04 04:47:41 +0000164 return OldVal;
Chris Lattnere7fd5532006-05-08 22:00:52 +0000165}
166
167/// getPointerToGlobalIfAvailable - This returns the address of the specified
168/// global value if it is has already been codegen'd, otherwise it returns null.
169///
170void *ExecutionEngine::getPointerToGlobalIfAvailable(const GlobalValue *GV) {
171 MutexGuard locked(lock);
172
173 std::map<const GlobalValue*, void*>::iterator I =
174 state.getGlobalAddressMap(locked).find(GV);
175 return I != state.getGlobalAddressMap(locked).end() ? I->second : 0;
176}
177
Chris Lattner55d86482003-12-31 20:21:04 +0000178/// getGlobalValueAtAddress - Return the LLVM global value object that starts
179/// at the specified address.
180///
181const GlobalValue *ExecutionEngine::getGlobalValueAtAddress(void *Addr) {
Reid Spenceree448632005-07-12 15:51:55 +0000182 MutexGuard locked(lock);
183
Chris Lattner55d86482003-12-31 20:21:04 +0000184 // If we haven't computed the reverse mapping yet, do so first.
Reid Spenceree448632005-07-12 15:51:55 +0000185 if (state.getGlobalAddressReverseMap(locked).empty()) {
Chris Lattnere7fd5532006-05-08 22:00:52 +0000186 for (std::map<const GlobalValue*, void *>::iterator
187 I = state.getGlobalAddressMap(locked).begin(),
188 E = state.getGlobalAddressMap(locked).end(); I != E; ++I)
189 state.getGlobalAddressReverseMap(locked).insert(std::make_pair(I->second,
190 I->first));
Chris Lattner55d86482003-12-31 20:21:04 +0000191 }
192
193 std::map<void *, const GlobalValue*>::iterator I =
Reid Spenceree448632005-07-12 15:51:55 +0000194 state.getGlobalAddressReverseMap(locked).find(Addr);
195 return I != state.getGlobalAddressReverseMap(locked).end() ? I->second : 0;
Chris Lattner55d86482003-12-31 20:21:04 +0000196}
Chris Lattner87f03102003-12-26 06:50:30 +0000197
198// CreateArgv - Turn a vector of strings into a nice argv style array of
199// pointers to null terminated strings.
200//
201static void *CreateArgv(ExecutionEngine *EE,
202 const std::vector<std::string> &InputArgv) {
Owen Andersona69571c2006-05-03 01:29:57 +0000203 unsigned PtrSize = EE->getTargetData()->getPointerSize();
Chris Lattner87f03102003-12-26 06:50:30 +0000204 char *Result = new char[(InputArgv.size()+1)*PtrSize];
205
Bill Wendling480f0932006-11-27 23:54:50 +0000206 DOUT << "ARGV = " << (void*)Result << "\n";
Christopher Lamb43ad6b32007-12-17 01:12:55 +0000207 const Type *SBytePtr = PointerType::getUnqual(Type::Int8Ty);
Chris Lattner87f03102003-12-26 06:50:30 +0000208
209 for (unsigned i = 0; i != InputArgv.size(); ++i) {
210 unsigned Size = InputArgv[i].size()+1;
211 char *Dest = new char[Size];
Bill Wendling480f0932006-11-27 23:54:50 +0000212 DOUT << "ARGV[" << i << "] = " << (void*)Dest << "\n";
Misha Brukmanedf128a2005-04-21 22:36:52 +0000213
Chris Lattner87f03102003-12-26 06:50:30 +0000214 std::copy(InputArgv[i].begin(), InputArgv[i].end(), Dest);
215 Dest[Size-1] = 0;
Misha Brukmanedf128a2005-04-21 22:36:52 +0000216
Chris Lattner87f03102003-12-26 06:50:30 +0000217 // Endian safe: Result[i] = (PointerTy)Dest;
218 EE->StoreValueToMemory(PTOGV(Dest), (GenericValue*)(Result+i*PtrSize),
219 SBytePtr);
220 }
221
222 // Null terminate it
223 EE->StoreValueToMemory(PTOGV(0),
224 (GenericValue*)(Result+InputArgv.size()*PtrSize),
225 SBytePtr);
226 return Result;
227}
228
Chris Lattner9ca6cda2006-03-08 18:42:46 +0000229
230/// runStaticConstructorsDestructors - This method is used to execute all of
Chris Lattnerfe854032006-08-16 01:24:12 +0000231/// the static constructors or destructors for a program, depending on the
Chris Lattner9ca6cda2006-03-08 18:42:46 +0000232/// value of isDtors.
233void ExecutionEngine::runStaticConstructorsDestructors(bool isDtors) {
234 const char *Name = isDtors ? "llvm.global_dtors" : "llvm.global_ctors";
Chris Lattner9ca6cda2006-03-08 18:42:46 +0000235
Chris Lattnerfe854032006-08-16 01:24:12 +0000236 // Execute global ctors/dtors for each module in the program.
237 for (unsigned m = 0, e = Modules.size(); m != e; ++m) {
238 GlobalVariable *GV = Modules[m]->getModule()->getNamedGlobal(Name);
239
240 // If this global has internal linkage, or if it has a use, then it must be
241 // an old-style (llvmgcc3) static ctor with __main linked in and in use. If
242 // this is the case, don't execute any of the global ctors, __main will do
243 // it.
Reid Spencer5cbf9852007-01-30 20:08:39 +0000244 if (!GV || GV->isDeclaration() || GV->hasInternalLinkage()) continue;
Chris Lattnerfe854032006-08-16 01:24:12 +0000245
246 // Should be an array of '{ int, void ()* }' structs. The first value is
247 // the init priority, which we ignore.
248 ConstantArray *InitList = dyn_cast<ConstantArray>(GV->getInitializer());
249 if (!InitList) continue;
250 for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i)
251 if (ConstantStruct *CS =
252 dyn_cast<ConstantStruct>(InitList->getOperand(i))) {
253 if (CS->getNumOperands() != 2) break; // Not array of 2-element structs.
Chris Lattner9ca6cda2006-03-08 18:42:46 +0000254
Chris Lattnerfe854032006-08-16 01:24:12 +0000255 Constant *FP = CS->getOperand(1);
256 if (FP->isNullValue())
257 break; // Found a null terminator, exit.
Chris Lattner9ca6cda2006-03-08 18:42:46 +0000258
Chris Lattnerfe854032006-08-16 01:24:12 +0000259 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(FP))
Reid Spencer3da59db2006-11-27 01:05:10 +0000260 if (CE->isCast())
Chris Lattnerfe854032006-08-16 01:24:12 +0000261 FP = CE->getOperand(0);
262 if (Function *F = dyn_cast<Function>(FP)) {
263 // Execute the ctor/dtor function!
264 runFunction(F, std::vector<GenericValue>());
265 }
Chris Lattner9ca6cda2006-03-08 18:42:46 +0000266 }
Chris Lattnerfe854032006-08-16 01:24:12 +0000267 }
Chris Lattner9ca6cda2006-03-08 18:42:46 +0000268}
269
Dan Gohmanb6e3d6c2008-08-26 01:38:29 +0000270#ifndef NDEBUG
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000271/// isTargetNullPtr - Return whether the target pointer stored at Loc is null.
272static bool isTargetNullPtr(ExecutionEngine *EE, void *Loc) {
273 unsigned PtrSize = EE->getTargetData()->getPointerSize();
274 for (unsigned i = 0; i < PtrSize; ++i)
275 if (*(i + (uint8_t*)Loc))
276 return false;
277 return true;
278}
Dan Gohmanb6e3d6c2008-08-26 01:38:29 +0000279#endif
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000280
Chris Lattner87f03102003-12-26 06:50:30 +0000281/// runFunctionAsMain - This is a helper function which wraps runFunction to
282/// handle the common task of starting up main with the specified argc, argv,
283/// and envp parameters.
284int ExecutionEngine::runFunctionAsMain(Function *Fn,
285 const std::vector<std::string> &argv,
286 const char * const * envp) {
287 std::vector<GenericValue> GVArgs;
288 GenericValue GVArgc;
Reid Spencer8fb0f192007-03-06 03:04:04 +0000289 GVArgc.IntVal = APInt(32, argv.size());
Anton Korobeynikov499d8f02007-06-03 19:17:35 +0000290
291 // Check main() type
Chris Lattnerf24d0992004-08-16 01:05:35 +0000292 unsigned NumArgs = Fn->getFunctionType()->getNumParams();
Anton Korobeynikov499d8f02007-06-03 19:17:35 +0000293 const FunctionType *FTy = Fn->getFunctionType();
Christopher Lamb43ad6b32007-12-17 01:12:55 +0000294 const Type* PPInt8Ty =
295 PointerType::getUnqual(PointerType::getUnqual(Type::Int8Ty));
Anton Korobeynikov499d8f02007-06-03 19:17:35 +0000296 switch (NumArgs) {
297 case 3:
298 if (FTy->getParamType(2) != PPInt8Ty) {
299 cerr << "Invalid type for third argument of main() supplied\n";
300 abort();
301 }
Anton Korobeynikovfb450862007-06-03 19:20:49 +0000302 // FALLS THROUGH
Anton Korobeynikov499d8f02007-06-03 19:17:35 +0000303 case 2:
304 if (FTy->getParamType(1) != PPInt8Ty) {
305 cerr << "Invalid type for second argument of main() supplied\n";
306 abort();
307 }
Anton Korobeynikovfb450862007-06-03 19:20:49 +0000308 // FALLS THROUGH
Anton Korobeynikov499d8f02007-06-03 19:17:35 +0000309 case 1:
310 if (FTy->getParamType(0) != Type::Int32Ty) {
311 cerr << "Invalid type for first argument of main() supplied\n";
312 abort();
313 }
Anton Korobeynikovfb450862007-06-03 19:20:49 +0000314 // FALLS THROUGH
Anton Korobeynikov499d8f02007-06-03 19:17:35 +0000315 case 0:
316 if (FTy->getReturnType() != Type::Int32Ty &&
317 FTy->getReturnType() != Type::VoidTy) {
318 cerr << "Invalid return type of main() supplied\n";
319 abort();
320 }
321 break;
322 default:
323 cerr << "Invalid number of arguments of main() supplied\n";
324 abort();
325 }
326
Chris Lattnerf24d0992004-08-16 01:05:35 +0000327 if (NumArgs) {
328 GVArgs.push_back(GVArgc); // Arg #0 = argc.
329 if (NumArgs > 1) {
330 GVArgs.push_back(PTOGV(CreateArgv(this, argv))); // Arg #1 = argv.
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000331 assert(!isTargetNullPtr(this, GVTOP(GVArgs[1])) &&
Chris Lattnerf24d0992004-08-16 01:05:35 +0000332 "argv[0] was null after CreateArgv");
333 if (NumArgs > 2) {
334 std::vector<std::string> EnvVars;
335 for (unsigned i = 0; envp[i]; ++i)
336 EnvVars.push_back(envp[i]);
337 GVArgs.push_back(PTOGV(CreateArgv(this, EnvVars))); // Arg #2 = envp.
338 }
339 }
340 }
Reid Spencer8fb0f192007-03-06 03:04:04 +0000341 return runFunction(Fn, GVArgs).IntVal.getZExtValue();
Chris Lattner87f03102003-12-26 06:50:30 +0000342}
343
Misha Brukman19684162003-10-16 21:18:05 +0000344/// If possible, create a JIT, unless the caller specifically requests an
345/// Interpreter or there's an error. If even an Interpreter cannot be created,
Misha Brukmanedf128a2005-04-21 22:36:52 +0000346/// NULL is returned.
Misha Brukman4afac182003-10-10 17:45:12 +0000347///
Misha Brukmanedf128a2005-04-21 22:36:52 +0000348ExecutionEngine *ExecutionEngine::create(ModuleProvider *MP,
Reid Spencerd4c0e622007-03-03 18:19:18 +0000349 bool ForceInterpreter,
Evan Cheng502f20b2008-08-08 08:11:34 +0000350 std::string *ErrorStr,
351 bool Fast) {
Brian Gaeke82d82772003-09-03 20:34:19 +0000352 ExecutionEngine *EE = 0;
353
Nick Lewycky6456d862008-03-08 02:49:45 +0000354 // Make sure we can resolve symbols in the program as well. The zero arg
355 // to the function tells DynamicLibrary to load the program, not a library.
356 if (sys::DynamicLibrary::LoadLibraryPermanently(0, ErrorStr))
357 return 0;
358
Chris Lattner73011782003-12-28 09:44:37 +0000359 // Unless the interpreter was explicitly selected, try making a JIT.
Chris Lattner2fe4bb02006-03-22 06:07:50 +0000360 if (!ForceInterpreter && JITCtor)
Evan Cheng502f20b2008-08-08 08:11:34 +0000361 EE = JITCtor(MP, ErrorStr, Fast);
Brian Gaeke82d82772003-09-03 20:34:19 +0000362
363 // If we can't make a JIT, make an interpreter instead.
Chris Lattner2fe4bb02006-03-22 06:07:50 +0000364 if (EE == 0 && InterpCtor)
Evan Cheng502f20b2008-08-08 08:11:34 +0000365 EE = InterpCtor(MP, ErrorStr, Fast);
Chris Lattner73011782003-12-28 09:44:37 +0000366
Brian Gaeke82d82772003-09-03 20:34:19 +0000367 return EE;
368}
369
Chris Lattner8b5295b2007-10-21 22:57:11 +0000370ExecutionEngine *ExecutionEngine::create(Module *M) {
371 return create(new ExistingModuleProvider(M));
372}
373
Misha Brukman4afac182003-10-10 17:45:12 +0000374/// getPointerToGlobal - This returns the address of the specified global
375/// value. This may involve code generation if it's a function.
376///
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000377void *ExecutionEngine::getPointerToGlobal(const GlobalValue *GV) {
Brian Gaeke37df4602003-08-13 18:16:14 +0000378 if (Function *F = const_cast<Function*>(dyn_cast<Function>(GV)))
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000379 return getPointerToFunction(F);
380
Reid Spenceree448632005-07-12 15:51:55 +0000381 MutexGuard locked(lock);
Jeff Cohen68835dd2006-02-07 05:11:57 +0000382 void *p = state.getGlobalAddressMap(locked)[GV];
383 if (p)
384 return p;
385
386 // Global variable might have been added since interpreter started.
387 if (GlobalVariable *GVar =
388 const_cast<GlobalVariable *>(dyn_cast<GlobalVariable>(GV)))
389 EmitGlobalVariable(GVar);
390 else
Chris Lattner64f150f2007-02-14 06:20:04 +0000391 assert(0 && "Global hasn't had an address allocated yet!");
Reid Spenceree448632005-07-12 15:51:55 +0000392 return state.getGlobalAddressMap(locked)[GV];
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000393}
394
Reid Spencer3da59db2006-11-27 01:05:10 +0000395/// This function converts a Constant* into a GenericValue. The interesting
396/// part is if C is a ConstantExpr.
Reid Spencerba28cb92007-08-11 15:57:56 +0000397/// @brief Get a GenericValue for a Constant*
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000398GenericValue ExecutionEngine::getConstantValue(const Constant *C) {
Reid Spencer3da59db2006-11-27 01:05:10 +0000399 // If its undefined, return the garbage.
Reid Spencerbce30f12007-03-06 22:23:15 +0000400 if (isa<UndefValue>(C))
401 return GenericValue();
Chris Lattnerd8c03bf2003-04-23 19:01:49 +0000402
Reid Spencer3da59db2006-11-27 01:05:10 +0000403 // If the value is a ConstantExpr
404 if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) {
Reid Spencerbce30f12007-03-06 22:23:15 +0000405 Constant *Op0 = CE->getOperand(0);
Chris Lattnerd8c03bf2003-04-23 19:01:49 +0000406 switch (CE->getOpcode()) {
407 case Instruction::GetElementPtr: {
Reid Spencer3da59db2006-11-27 01:05:10 +0000408 // Compute the index
Reid Spencerbce30f12007-03-06 22:23:15 +0000409 GenericValue Result = getConstantValue(Op0);
Chris Lattner829621c2007-02-10 20:35:22 +0000410 SmallVector<Value*, 8> Indices(CE->op_begin()+1, CE->op_end());
Chris Lattnerd8c03bf2003-04-23 19:01:49 +0000411 uint64_t Offset =
Reid Spencerbce30f12007-03-06 22:23:15 +0000412 TD->getIndexedOffset(Op0->getType(), &Indices[0], Indices.size());
Misha Brukmanedf128a2005-04-21 22:36:52 +0000413
Reid Spencer8fb0f192007-03-06 03:04:04 +0000414 char* tmp = (char*) Result.PointerVal;
415 Result = PTOGV(tmp + Offset);
Chris Lattnerd8c03bf2003-04-23 19:01:49 +0000416 return Result;
417 }
Reid Spencerbce30f12007-03-06 22:23:15 +0000418 case Instruction::Trunc: {
419 GenericValue GV = getConstantValue(Op0);
420 uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
421 GV.IntVal = GV.IntVal.trunc(BitWidth);
422 return GV;
423 }
424 case Instruction::ZExt: {
425 GenericValue GV = getConstantValue(Op0);
426 uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
427 GV.IntVal = GV.IntVal.zext(BitWidth);
428 return GV;
429 }
430 case Instruction::SExt: {
431 GenericValue GV = getConstantValue(Op0);
432 uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
433 GV.IntVal = GV.IntVal.sext(BitWidth);
434 return GV;
435 }
436 case Instruction::FPTrunc: {
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000437 // FIXME long double
Reid Spencerbce30f12007-03-06 22:23:15 +0000438 GenericValue GV = getConstantValue(Op0);
439 GV.FloatVal = float(GV.DoubleVal);
440 return GV;
441 }
442 case Instruction::FPExt:{
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000443 // FIXME long double
Reid Spencerbce30f12007-03-06 22:23:15 +0000444 GenericValue GV = getConstantValue(Op0);
445 GV.DoubleVal = double(GV.FloatVal);
446 return GV;
447 }
448 case Instruction::UIToFP: {
449 GenericValue GV = getConstantValue(Op0);
450 if (CE->getType() == Type::FloatTy)
451 GV.FloatVal = float(GV.IntVal.roundToDouble());
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000452 else if (CE->getType() == Type::DoubleTy)
Reid Spencerbce30f12007-03-06 22:23:15 +0000453 GV.DoubleVal = GV.IntVal.roundToDouble();
Dale Johannesen910993e2007-09-21 22:09:37 +0000454 else if (CE->getType() == Type::X86_FP80Ty) {
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000455 const uint64_t zero[] = {0, 0};
456 APFloat apf = APFloat(APInt(80, 2, zero));
Dan Gohman62824062008-02-29 01:27:13 +0000457 (void)apf.convertFromAPInt(GV.IntVal,
458 false,
459 APFloat::rmNearestTiesToEven);
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000460 GV.IntVal = apf.convertToAPInt();
461 }
Reid Spencerbce30f12007-03-06 22:23:15 +0000462 return GV;
463 }
464 case Instruction::SIToFP: {
465 GenericValue GV = getConstantValue(Op0);
466 if (CE->getType() == Type::FloatTy)
467 GV.FloatVal = float(GV.IntVal.signedRoundToDouble());
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000468 else if (CE->getType() == Type::DoubleTy)
Reid Spencerbce30f12007-03-06 22:23:15 +0000469 GV.DoubleVal = GV.IntVal.signedRoundToDouble();
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000470 else if (CE->getType() == Type::X86_FP80Ty) {
471 const uint64_t zero[] = { 0, 0};
472 APFloat apf = APFloat(APInt(80, 2, zero));
Dan Gohman62824062008-02-29 01:27:13 +0000473 (void)apf.convertFromAPInt(GV.IntVal,
474 true,
475 APFloat::rmNearestTiesToEven);
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000476 GV.IntVal = apf.convertToAPInt();
477 }
Reid Spencerbce30f12007-03-06 22:23:15 +0000478 return GV;
479 }
480 case Instruction::FPToUI: // double->APInt conversion handles sign
481 case Instruction::FPToSI: {
482 GenericValue GV = getConstantValue(Op0);
483 uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth();
484 if (Op0->getType() == Type::FloatTy)
485 GV.IntVal = APIntOps::RoundFloatToAPInt(GV.FloatVal, BitWidth);
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000486 else if (Op0->getType() == Type::DoubleTy)
Reid Spencerbce30f12007-03-06 22:23:15 +0000487 GV.IntVal = APIntOps::RoundDoubleToAPInt(GV.DoubleVal, BitWidth);
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000488 else if (Op0->getType() == Type::X86_FP80Ty) {
489 APFloat apf = APFloat(GV.IntVal);
490 uint64_t v;
491 (void)apf.convertToInteger(&v, BitWidth,
492 CE->getOpcode()==Instruction::FPToSI,
493 APFloat::rmTowardZero);
494 GV.IntVal = v; // endian?
495 }
Reid Spencerbce30f12007-03-06 22:23:15 +0000496 return GV;
497 }
Reid Spencer3da59db2006-11-27 01:05:10 +0000498 case Instruction::PtrToInt: {
Reid Spencerbce30f12007-03-06 22:23:15 +0000499 GenericValue GV = getConstantValue(Op0);
500 uint32_t PtrWidth = TD->getPointerSizeInBits();
501 GV.IntVal = APInt(PtrWidth, uintptr_t(GV.PointerVal));
502 return GV;
503 }
504 case Instruction::IntToPtr: {
505 GenericValue GV = getConstantValue(Op0);
506 uint32_t PtrWidth = TD->getPointerSizeInBits();
507 if (PtrWidth != GV.IntVal.getBitWidth())
508 GV.IntVal = GV.IntVal.zextOrTrunc(PtrWidth);
509 assert(GV.IntVal.getBitWidth() <= 64 && "Bad pointer width");
510 GV.PointerVal = PointerTy(uintptr_t(GV.IntVal.getZExtValue()));
Reid Spencer3da59db2006-11-27 01:05:10 +0000511 return GV;
512 }
513 case Instruction::BitCast: {
Reid Spencerbce30f12007-03-06 22:23:15 +0000514 GenericValue GV = getConstantValue(Op0);
515 const Type* DestTy = CE->getType();
516 switch (Op0->getType()->getTypeID()) {
517 default: assert(0 && "Invalid bitcast operand");
518 case Type::IntegerTyID:
519 assert(DestTy->isFloatingPoint() && "invalid bitcast");
520 if (DestTy == Type::FloatTy)
521 GV.FloatVal = GV.IntVal.bitsToFloat();
522 else if (DestTy == Type::DoubleTy)
523 GV.DoubleVal = GV.IntVal.bitsToDouble();
524 break;
525 case Type::FloatTyID:
526 assert(DestTy == Type::Int32Ty && "Invalid bitcast");
527 GV.IntVal.floatToBits(GV.FloatVal);
528 break;
529 case Type::DoubleTyID:
530 assert(DestTy == Type::Int64Ty && "Invalid bitcast");
531 GV.IntVal.doubleToBits(GV.DoubleVal);
532 break;
533 case Type::PointerTyID:
534 assert(isa<PointerType>(DestTy) && "Invalid bitcast");
535 break; // getConstantValue(Op0) above already converted it
536 }
537 return GV;
Chris Lattnerd8c03bf2003-04-23 19:01:49 +0000538 }
Chris Lattner9a231222003-05-14 17:51:49 +0000539 case Instruction::Add:
Reid Spencerbce30f12007-03-06 22:23:15 +0000540 case Instruction::Sub:
541 case Instruction::Mul:
542 case Instruction::UDiv:
543 case Instruction::SDiv:
544 case Instruction::URem:
545 case Instruction::SRem:
546 case Instruction::And:
547 case Instruction::Or:
548 case Instruction::Xor: {
549 GenericValue LHS = getConstantValue(Op0);
550 GenericValue RHS = getConstantValue(CE->getOperand(1));
551 GenericValue GV;
Chris Lattner5f90cb82004-07-11 08:01:11 +0000552 switch (CE->getOperand(0)->getType()->getTypeID()) {
553 default: assert(0 && "Bad add type!"); abort();
Reid Spencera54b7cb2007-01-12 07:05:14 +0000554 case Type::IntegerTyID:
Reid Spencerbce30f12007-03-06 22:23:15 +0000555 switch (CE->getOpcode()) {
556 default: assert(0 && "Invalid integer opcode");
557 case Instruction::Add: GV.IntVal = LHS.IntVal + RHS.IntVal; break;
558 case Instruction::Sub: GV.IntVal = LHS.IntVal - RHS.IntVal; break;
559 case Instruction::Mul: GV.IntVal = LHS.IntVal * RHS.IntVal; break;
560 case Instruction::UDiv:GV.IntVal = LHS.IntVal.udiv(RHS.IntVal); break;
561 case Instruction::SDiv:GV.IntVal = LHS.IntVal.sdiv(RHS.IntVal); break;
562 case Instruction::URem:GV.IntVal = LHS.IntVal.urem(RHS.IntVal); break;
563 case Instruction::SRem:GV.IntVal = LHS.IntVal.srem(RHS.IntVal); break;
564 case Instruction::And: GV.IntVal = LHS.IntVal & RHS.IntVal; break;
565 case Instruction::Or: GV.IntVal = LHS.IntVal | RHS.IntVal; break;
566 case Instruction::Xor: GV.IntVal = LHS.IntVal ^ RHS.IntVal; break;
567 }
Chris Lattner5f90cb82004-07-11 08:01:11 +0000568 break;
569 case Type::FloatTyID:
Reid Spencerbce30f12007-03-06 22:23:15 +0000570 switch (CE->getOpcode()) {
571 default: assert(0 && "Invalid float opcode"); abort();
572 case Instruction::Add:
573 GV.FloatVal = LHS.FloatVal + RHS.FloatVal; break;
574 case Instruction::Sub:
575 GV.FloatVal = LHS.FloatVal - RHS.FloatVal; break;
576 case Instruction::Mul:
577 GV.FloatVal = LHS.FloatVal * RHS.FloatVal; break;
578 case Instruction::FDiv:
579 GV.FloatVal = LHS.FloatVal / RHS.FloatVal; break;
580 case Instruction::FRem:
581 GV.FloatVal = ::fmodf(LHS.FloatVal,RHS.FloatVal); break;
582 }
Chris Lattner5f90cb82004-07-11 08:01:11 +0000583 break;
584 case Type::DoubleTyID:
Reid Spencerbce30f12007-03-06 22:23:15 +0000585 switch (CE->getOpcode()) {
586 default: assert(0 && "Invalid double opcode"); abort();
587 case Instruction::Add:
588 GV.DoubleVal = LHS.DoubleVal + RHS.DoubleVal; break;
589 case Instruction::Sub:
590 GV.DoubleVal = LHS.DoubleVal - RHS.DoubleVal; break;
591 case Instruction::Mul:
592 GV.DoubleVal = LHS.DoubleVal * RHS.DoubleVal; break;
593 case Instruction::FDiv:
594 GV.DoubleVal = LHS.DoubleVal / RHS.DoubleVal; break;
595 case Instruction::FRem:
596 GV.DoubleVal = ::fmod(LHS.DoubleVal,RHS.DoubleVal); break;
597 }
Chris Lattner5f90cb82004-07-11 08:01:11 +0000598 break;
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000599 case Type::X86_FP80TyID:
600 case Type::PPC_FP128TyID:
601 case Type::FP128TyID: {
602 APFloat apfLHS = APFloat(LHS.IntVal);
603 switch (CE->getOpcode()) {
604 default: assert(0 && "Invalid long double opcode"); abort();
605 case Instruction::Add:
606 apfLHS.add(APFloat(RHS.IntVal), APFloat::rmNearestTiesToEven);
607 GV.IntVal = apfLHS.convertToAPInt();
608 break;
609 case Instruction::Sub:
610 apfLHS.subtract(APFloat(RHS.IntVal), APFloat::rmNearestTiesToEven);
611 GV.IntVal = apfLHS.convertToAPInt();
612 break;
613 case Instruction::Mul:
614 apfLHS.multiply(APFloat(RHS.IntVal), APFloat::rmNearestTiesToEven);
615 GV.IntVal = apfLHS.convertToAPInt();
616 break;
617 case Instruction::FDiv:
618 apfLHS.divide(APFloat(RHS.IntVal), APFloat::rmNearestTiesToEven);
619 GV.IntVal = apfLHS.convertToAPInt();
620 break;
621 case Instruction::FRem:
622 apfLHS.mod(APFloat(RHS.IntVal), APFloat::rmNearestTiesToEven);
623 GV.IntVal = apfLHS.convertToAPInt();
624 break;
625 }
626 }
627 break;
Chris Lattner5f90cb82004-07-11 08:01:11 +0000628 }
Reid Spencerbce30f12007-03-06 22:23:15 +0000629 return GV;
630 }
Chris Lattner9a231222003-05-14 17:51:49 +0000631 default:
632 break;
633 }
Reid Spencerbce30f12007-03-06 22:23:15 +0000634 cerr << "ConstantExpr not handled: " << *CE << "\n";
Chris Lattner9a231222003-05-14 17:51:49 +0000635 abort();
636 }
Misha Brukmanedf128a2005-04-21 22:36:52 +0000637
Reid Spencerbce30f12007-03-06 22:23:15 +0000638 GenericValue Result;
Chris Lattnerf70c22b2004-06-17 18:19:28 +0000639 switch (C->getType()->getTypeID()) {
Reid Spencer8fb0f192007-03-06 03:04:04 +0000640 case Type::FloatTyID:
Dale Johannesen43421b32007-09-06 18:13:44 +0000641 Result.FloatVal = cast<ConstantFP>(C)->getValueAPF().convertToFloat();
Reid Spencera54b7cb2007-01-12 07:05:14 +0000642 break;
Reid Spencer8fb0f192007-03-06 03:04:04 +0000643 case Type::DoubleTyID:
Dale Johannesen43421b32007-09-06 18:13:44 +0000644 Result.DoubleVal = cast<ConstantFP>(C)->getValueAPF().convertToDouble();
Reid Spencer8fb0f192007-03-06 03:04:04 +0000645 break;
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000646 case Type::X86_FP80TyID:
647 case Type::FP128TyID:
648 case Type::PPC_FP128TyID:
649 Result.IntVal = cast <ConstantFP>(C)->getValueAPF().convertToAPInt();
650 break;
Reid Spencer8fb0f192007-03-06 03:04:04 +0000651 case Type::IntegerTyID:
652 Result.IntVal = cast<ConstantInt>(C)->getValue();
653 break;
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000654 case Type::PointerTyID:
Reid Spencer40cf2f92004-07-18 00:41:27 +0000655 if (isa<ConstantPointerNull>(C))
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000656 Result.PointerVal = 0;
Reid Spencer40cf2f92004-07-18 00:41:27 +0000657 else if (const Function *F = dyn_cast<Function>(C))
658 Result = PTOGV(getPointerToFunctionOrStub(const_cast<Function*>(F)));
659 else if (const GlobalVariable* GV = dyn_cast<GlobalVariable>(C))
660 Result = PTOGV(getOrEmitGlobalVariable(const_cast<GlobalVariable*>(GV)));
661 else
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000662 assert(0 && "Unknown constant pointer type!");
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000663 break;
664 default:
Reid Spencerbce30f12007-03-06 22:23:15 +0000665 cerr << "ERROR: Constant unimplemented for type: " << *C->getType() << "\n";
Chris Lattnerd8c03bf2003-04-23 19:01:49 +0000666 abort();
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000667 }
668 return Result;
669}
670
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000671/// StoreIntToMemory - Fills the StoreBytes bytes of memory starting from Dst
672/// with the integer held in IntVal.
673static void StoreIntToMemory(const APInt &IntVal, uint8_t *Dst,
674 unsigned StoreBytes) {
675 assert((IntVal.getBitWidth()+7)/8 >= StoreBytes && "Integer too small!");
676 uint8_t *Src = (uint8_t *)IntVal.getRawData();
677
678 if (sys::littleEndianHost())
679 // Little-endian host - the source is ordered from LSB to MSB. Order the
680 // destination from LSB to MSB: Do a straight copy.
681 memcpy(Dst, Src, StoreBytes);
682 else {
683 // Big-endian host - the source is an array of 64 bit words ordered from
684 // LSW to MSW. Each word is ordered from MSB to LSB. Order the destination
685 // from MSB to LSB: Reverse the word order, but not the bytes in a word.
686 while (StoreBytes > sizeof(uint64_t)) {
687 StoreBytes -= sizeof(uint64_t);
688 // May not be aligned so use memcpy.
689 memcpy(Dst + StoreBytes, Src, sizeof(uint64_t));
690 Src += sizeof(uint64_t);
691 }
692
693 memcpy(Dst, Src + sizeof(uint64_t) - StoreBytes, StoreBytes);
694 }
695}
696
Nate Begeman37efe672006-04-22 18:53:45 +0000697/// StoreValueToMemory - Stores the data in Val of type Ty at address Ptr. Ptr
698/// is the address of the memory at which to store Val, cast to GenericValue *.
699/// It is not a pointer to a GenericValue containing the address at which to
700/// store Val.
Reid Spencer415c1f72007-03-06 05:03:16 +0000701void ExecutionEngine::StoreValueToMemory(const GenericValue &Val, GenericValue *Ptr,
Misha Brukman4afac182003-10-10 17:45:12 +0000702 const Type *Ty) {
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000703 const unsigned StoreBytes = getTargetData()->getTypeStoreSize(Ty);
704
Reid Spencer8fb0f192007-03-06 03:04:04 +0000705 switch (Ty->getTypeID()) {
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000706 case Type::IntegerTyID:
707 StoreIntToMemory(Val.IntVal, (uint8_t*)Ptr, StoreBytes);
Reid Spencer8fb0f192007-03-06 03:04:04 +0000708 break;
Reid Spencer8fb0f192007-03-06 03:04:04 +0000709 case Type::FloatTyID:
710 *((float*)Ptr) = Val.FloatVal;
711 break;
712 case Type::DoubleTyID:
713 *((double*)Ptr) = Val.DoubleVal;
714 break;
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000715 case Type::X86_FP80TyID: {
716 uint16_t *Dest = (uint16_t*)Ptr;
717 const uint16_t *Src = (uint16_t*)Val.IntVal.getRawData();
718 // This is endian dependent, but it will only work on x86 anyway.
719 Dest[0] = Src[4];
720 Dest[1] = Src[0];
721 Dest[2] = Src[1];
722 Dest[3] = Src[2];
723 Dest[4] = Src[3];
724 break;
725 }
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000726 case Type::PointerTyID:
727 // Ensure 64 bit target pointers are fully initialized on 32 bit hosts.
728 if (StoreBytes != sizeof(PointerTy))
729 memset(Ptr, 0, StoreBytes);
730
Reid Spencer8fb0f192007-03-06 03:04:04 +0000731 *((PointerTy*)Ptr) = Val.PointerVal;
732 break;
733 default:
734 cerr << "Cannot store value of type " << *Ty << "!\n";
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000735 }
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000736
737 if (sys::littleEndianHost() != getTargetData()->isLittleEndian())
738 // Host and target are different endian - reverse the stored bytes.
739 std::reverse((uint8_t*)Ptr, StoreBytes + (uint8_t*)Ptr);
740}
741
742/// LoadIntFromMemory - Loads the integer stored in the LoadBytes bytes starting
743/// from Src into IntVal, which is assumed to be wide enough and to hold zero.
744static void LoadIntFromMemory(APInt &IntVal, uint8_t *Src, unsigned LoadBytes) {
745 assert((IntVal.getBitWidth()+7)/8 >= LoadBytes && "Integer too small!");
746 uint8_t *Dst = (uint8_t *)IntVal.getRawData();
747
748 if (sys::littleEndianHost())
749 // Little-endian host - the destination must be ordered from LSB to MSB.
750 // The source is ordered from LSB to MSB: Do a straight copy.
751 memcpy(Dst, Src, LoadBytes);
752 else {
753 // Big-endian - the destination is an array of 64 bit words ordered from
754 // LSW to MSW. Each word must be ordered from MSB to LSB. The source is
755 // ordered from MSB to LSB: Reverse the word order, but not the bytes in
756 // a word.
757 while (LoadBytes > sizeof(uint64_t)) {
758 LoadBytes -= sizeof(uint64_t);
759 // May not be aligned so use memcpy.
760 memcpy(Dst, Src + LoadBytes, sizeof(uint64_t));
761 Dst += sizeof(uint64_t);
762 }
763
764 memcpy(Dst + sizeof(uint64_t) - LoadBytes, Src, LoadBytes);
765 }
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000766}
767
Misha Brukman4afac182003-10-10 17:45:12 +0000768/// FIXME: document
769///
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000770void ExecutionEngine::LoadValueFromMemory(GenericValue &Result,
Duncan Sands08bfe262008-03-10 16:38:37 +0000771 GenericValue *Ptr,
772 const Type *Ty) {
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000773 const unsigned LoadBytes = getTargetData()->getTypeStoreSize(Ty);
Duncan Sands1eff7042007-12-10 17:43:13 +0000774
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000775 if (sys::littleEndianHost() != getTargetData()->isLittleEndian()) {
776 // Host and target are different endian - reverse copy the stored
777 // bytes into a buffer, and load from that.
778 uint8_t *Src = (uint8_t*)Ptr;
779 uint8_t *Buf = (uint8_t*)alloca(LoadBytes);
780 std::reverse_copy(Src, Src + LoadBytes, Buf);
781 Ptr = (GenericValue*)Buf;
Reid Spencer8fb0f192007-03-06 03:04:04 +0000782 }
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000783
784 switch (Ty->getTypeID()) {
785 case Type::IntegerTyID:
786 // An APInt with all words initially zero.
787 Result.IntVal = APInt(cast<IntegerType>(Ty)->getBitWidth(), 0);
788 LoadIntFromMemory(Result.IntVal, (uint8_t*)Ptr, LoadBytes);
789 break;
Reid Spencer8fb0f192007-03-06 03:04:04 +0000790 case Type::FloatTyID:
791 Result.FloatVal = *((float*)Ptr);
792 break;
793 case Type::DoubleTyID:
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000794 Result.DoubleVal = *((double*)Ptr);
Reid Spencer8fb0f192007-03-06 03:04:04 +0000795 break;
Duncan Sands8a43e9e2007-12-14 19:38:31 +0000796 case Type::PointerTyID:
Reid Spencer8fb0f192007-03-06 03:04:04 +0000797 Result.PointerVal = *((PointerTy*)Ptr);
798 break;
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000799 case Type::X86_FP80TyID: {
800 // This is endian dependent, but it will only work on x86 anyway.
Duncan Sands9e4635a2007-12-15 17:37:40 +0000801 // FIXME: Will not trap if loading a signaling NaN.
Duncan Sandsdd65a732007-11-28 10:36:19 +0000802 uint16_t *p = (uint16_t*)Ptr;
803 union {
804 uint16_t x[8];
805 uint64_t y[2];
806 };
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000807 x[0] = p[1];
808 x[1] = p[2];
809 x[2] = p[3];
810 x[3] = p[4];
811 x[4] = p[0];
Duncan Sandsdd65a732007-11-28 10:36:19 +0000812 Result.IntVal = APInt(80, 2, y);
Dale Johannesen1abac0d2007-09-17 18:44:13 +0000813 break;
814 }
Reid Spencer8fb0f192007-03-06 03:04:04 +0000815 default:
816 cerr << "Cannot load value of type " << *Ty << "!\n";
817 abort();
Chris Lattnerf88b9a62003-05-08 16:52:16 +0000818 }
Chris Lattnerf88b9a62003-05-08 16:52:16 +0000819}
820
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000821// InitializeMemory - Recursive function to apply a Constant value into the
822// specified memory location...
823//
824void ExecutionEngine::InitializeMemory(const Constant *Init, void *Addr) {
Dale Johannesendd947ea2008-08-07 01:30:15 +0000825 DOUT << "Initializing " << Addr;
826 DEBUG(Init->dump());
Chris Lattnerbd1d3822004-10-16 18:19:26 +0000827 if (isa<UndefValue>(Init)) {
828 return;
Reid Spencer9d6565a2007-02-15 02:26:10 +0000829 } else if (const ConstantVector *CP = dyn_cast<ConstantVector>(Init)) {
Robert Bocchino7c2b7c72006-01-20 18:18:40 +0000830 unsigned ElementSize =
Duncan Sands514ab342007-11-01 20:53:16 +0000831 getTargetData()->getABITypeSize(CP->getType()->getElementType());
Robert Bocchino7c2b7c72006-01-20 18:18:40 +0000832 for (unsigned i = 0, e = CP->getNumOperands(); i != e; ++i)
833 InitializeMemory(CP->getOperand(i), (char*)Addr+i*ElementSize);
834 return;
Chris Lattnerb6e1dd72008-02-15 00:57:28 +0000835 } else if (isa<ConstantAggregateZero>(Init)) {
836 memset(Addr, 0, (size_t)getTargetData()->getABITypeSize(Init->getType()));
837 return;
Dan Gohman638e3782008-05-20 03:20:09 +0000838 } else if (const ConstantArray *CPA = dyn_cast<ConstantArray>(Init)) {
839 unsigned ElementSize =
840 getTargetData()->getABITypeSize(CPA->getType()->getElementType());
841 for (unsigned i = 0, e = CPA->getNumOperands(); i != e; ++i)
842 InitializeMemory(CPA->getOperand(i), (char*)Addr+i*ElementSize);
843 return;
844 } else if (const ConstantStruct *CPS = dyn_cast<ConstantStruct>(Init)) {
845 const StructLayout *SL =
846 getTargetData()->getStructLayout(cast<StructType>(CPS->getType()));
847 for (unsigned i = 0, e = CPS->getNumOperands(); i != e; ++i)
848 InitializeMemory(CPS->getOperand(i), (char*)Addr+SL->getElementOffset(i));
849 return;
Chris Lattnerbd1d3822004-10-16 18:19:26 +0000850 } else if (Init->getType()->isFirstClassType()) {
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000851 GenericValue Val = getConstantValue(Init);
852 StoreValueToMemory(Val, (GenericValue*)Addr, Init->getType());
853 return;
854 }
855
Dan Gohman638e3782008-05-20 03:20:09 +0000856 cerr << "Bad Type: " << *Init->getType() << "\n";
857 assert(0 && "Unknown constant type to initialize memory with!");
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000858}
859
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000860/// EmitGlobals - Emit all of the global variables to memory, storing their
861/// addresses into GlobalAddress. This must make sure to copy the contents of
862/// their initializers into the memory.
863///
864void ExecutionEngine::emitGlobals() {
Owen Andersona69571c2006-05-03 01:29:57 +0000865 const TargetData *TD = getTargetData();
Misha Brukmanedf128a2005-04-21 22:36:52 +0000866
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000867 // Loop over all of the global variables in the program, allocating the memory
Chris Lattnerfe854032006-08-16 01:24:12 +0000868 // to hold them. If there is more than one module, do a prepass over globals
869 // to figure out how the different modules should link together.
870 //
871 std::map<std::pair<std::string, const Type*>,
872 const GlobalValue*> LinkedGlobalsMap;
Misha Brukmanedf128a2005-04-21 22:36:52 +0000873
Chris Lattnerfe854032006-08-16 01:24:12 +0000874 if (Modules.size() != 1) {
875 for (unsigned m = 0, e = Modules.size(); m != e; ++m) {
876 Module &M = *Modules[m]->getModule();
877 for (Module::const_global_iterator I = M.global_begin(),
878 E = M.global_end(); I != E; ++I) {
879 const GlobalValue *GV = I;
Reid Spencer5cbf9852007-01-30 20:08:39 +0000880 if (GV->hasInternalLinkage() || GV->isDeclaration() ||
Chris Lattnerfe854032006-08-16 01:24:12 +0000881 GV->hasAppendingLinkage() || !GV->hasName())
882 continue;// Ignore external globals and globals with internal linkage.
883
884 const GlobalValue *&GVEntry =
885 LinkedGlobalsMap[std::make_pair(GV->getName(), GV->getType())];
886
887 // If this is the first time we've seen this global, it is the canonical
888 // version.
889 if (!GVEntry) {
890 GVEntry = GV;
891 continue;
892 }
893
894 // If the existing global is strong, never replace it.
Anton Korobeynikovb74ed072006-09-14 18:23:27 +0000895 if (GVEntry->hasExternalLinkage() ||
896 GVEntry->hasDLLImportLinkage() ||
897 GVEntry->hasDLLExportLinkage())
Chris Lattnerfe854032006-08-16 01:24:12 +0000898 continue;
899
900 // Otherwise, we know it's linkonce/weak, replace it if this is a strong
Dale Johannesenaafce772008-05-14 20:12:51 +0000901 // symbol. FIXME is this right for common?
Anton Korobeynikov78ee7b72006-12-01 00:25:12 +0000902 if (GV->hasExternalLinkage() || GVEntry->hasExternalWeakLinkage())
Chris Lattnerfe854032006-08-16 01:24:12 +0000903 GVEntry = GV;
Chris Lattnerd8c03bf2003-04-23 19:01:49 +0000904 }
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000905 }
Chris Lattnerfe854032006-08-16 01:24:12 +0000906 }
907
908 std::vector<const GlobalValue*> NonCanonicalGlobals;
909 for (unsigned m = 0, e = Modules.size(); m != e; ++m) {
910 Module &M = *Modules[m]->getModule();
911 for (Module::const_global_iterator I = M.global_begin(), E = M.global_end();
912 I != E; ++I) {
913 // In the multi-module case, see what this global maps to.
914 if (!LinkedGlobalsMap.empty()) {
915 if (const GlobalValue *GVEntry =
916 LinkedGlobalsMap[std::make_pair(I->getName(), I->getType())]) {
917 // If something else is the canonical global, ignore this one.
918 if (GVEntry != &*I) {
919 NonCanonicalGlobals.push_back(I);
920 continue;
921 }
922 }
923 }
924
Reid Spencer5cbf9852007-01-30 20:08:39 +0000925 if (!I->isDeclaration()) {
Chris Lattnerfe854032006-08-16 01:24:12 +0000926 // Get the type of the global.
927 const Type *Ty = I->getType()->getElementType();
Misha Brukmanedf128a2005-04-21 22:36:52 +0000928
Chris Lattnerfe854032006-08-16 01:24:12 +0000929 // Allocate some memory for it!
Duncan Sands514ab342007-11-01 20:53:16 +0000930 unsigned Size = TD->getABITypeSize(Ty);
Chris Lattnerfe854032006-08-16 01:24:12 +0000931 addGlobalMapping(I, new char[Size]);
932 } else {
933 // External variable reference. Try to use the dynamic loader to
934 // get a pointer to it.
935 if (void *SymAddr =
936 sys::DynamicLibrary::SearchForAddressOfSymbol(I->getName().c_str()))
937 addGlobalMapping(I, SymAddr);
938 else {
Bill Wendlinge8156192006-12-07 01:30:32 +0000939 cerr << "Could not resolve external global address: "
940 << I->getName() << "\n";
Chris Lattnerfe854032006-08-16 01:24:12 +0000941 abort();
942 }
943 }
944 }
945
946 // If there are multiple modules, map the non-canonical globals to their
947 // canonical location.
948 if (!NonCanonicalGlobals.empty()) {
949 for (unsigned i = 0, e = NonCanonicalGlobals.size(); i != e; ++i) {
950 const GlobalValue *GV = NonCanonicalGlobals[i];
951 const GlobalValue *CGV =
952 LinkedGlobalsMap[std::make_pair(GV->getName(), GV->getType())];
953 void *Ptr = getPointerToGlobalIfAvailable(CGV);
954 assert(Ptr && "Canonical global wasn't codegen'd!");
955 addGlobalMapping(GV, getPointerToGlobalIfAvailable(CGV));
956 }
957 }
958
Reid Spencera54b7cb2007-01-12 07:05:14 +0000959 // Now that all of the globals are set up in memory, loop through them all
960 // and initialize their contents.
Chris Lattnerfe854032006-08-16 01:24:12 +0000961 for (Module::const_global_iterator I = M.global_begin(), E = M.global_end();
962 I != E; ++I) {
Reid Spencer5cbf9852007-01-30 20:08:39 +0000963 if (!I->isDeclaration()) {
Chris Lattnerfe854032006-08-16 01:24:12 +0000964 if (!LinkedGlobalsMap.empty()) {
965 if (const GlobalValue *GVEntry =
966 LinkedGlobalsMap[std::make_pair(I->getName(), I->getType())])
967 if (GVEntry != &*I) // Not the canonical variable.
968 continue;
969 }
970 EmitGlobalVariable(I);
971 }
972 }
973 }
Chris Lattner24b0a182003-12-20 02:45:37 +0000974}
975
976// EmitGlobalVariable - This method emits the specified global variable to the
977// address specified in GlobalAddresses, or allocates new memory if it's not
978// already in the map.
Chris Lattnerc07ed132003-12-20 03:36:47 +0000979void ExecutionEngine::EmitGlobalVariable(const GlobalVariable *GV) {
Chris Lattner55d86482003-12-31 20:21:04 +0000980 void *GA = getPointerToGlobalIfAvailable(GV);
Bill Wendling480f0932006-11-27 23:54:50 +0000981 DOUT << "Global '" << GV->getName() << "' -> " << GA << "\n";
Chris Lattner23c47242004-02-08 19:33:23 +0000982
Chris Lattnerc07ed132003-12-20 03:36:47 +0000983 const Type *ElTy = GV->getType()->getElementType();
Duncan Sands514ab342007-11-01 20:53:16 +0000984 size_t GVSize = (size_t)getTargetData()->getABITypeSize(ElTy);
Chris Lattner24b0a182003-12-20 02:45:37 +0000985 if (GA == 0) {
986 // If it's not already specified, allocate memory for the global.
Chris Lattnera98c5452004-11-19 08:44:07 +0000987 GA = new char[GVSize];
Chris Lattner55d86482003-12-31 20:21:04 +0000988 addGlobalMapping(GV, GA);
Chris Lattner24b0a182003-12-20 02:45:37 +0000989 }
Chris Lattnerc07ed132003-12-20 03:36:47 +0000990
Chris Lattner24b0a182003-12-20 02:45:37 +0000991 InitializeMemory(GV->getInitializer(), GA);
Chris Lattner813c8152005-01-08 20:13:19 +0000992 NumInitBytes += (unsigned)GVSize;
Chris Lattner24b0a182003-12-20 02:45:37 +0000993 ++NumGlobals;
Chris Lattnerbd199fb2002-12-24 00:01:05 +0000994}