6502/llvm-6502
1
0
Fork 0
mirror of https://github.com/c64scene-ar/llvm-6502.git synced 2025-06-25 00:24:26 +00:00
Code Issues Projects Releases Wiki Activity

JIT: More nitty style tweakage, aka territory marking.

Browse Source 
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@118973 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Daniel Dunbar
2010-11-13 02:48:57 +00:00
parent afd693cff3
commit 48dd875be1
2 changed files with 206 additions and 247 deletions
Download Patch File Download Diff File Expand all files Collapse all files

123
include/llvm/ExecutionEngine/ExecutionEngine.h
View File

@ -41,6 +41,8 @@ class MutexGuard;
class TargetData; class TargetData;
class Type; class Type;
/// \brief Helper class for helping synchronize access to the global address map
/// table.
class ExecutionEngineState { class ExecutionEngineState {
public: public:
struct AddressMapConfig : public ValueMapConfig<const GlobalValue*> { struct AddressMapConfig : public ValueMapConfig<const GlobalValue*> {
@ -70,8 +72,7 @@ private:
public: public:
ExecutionEngineState(ExecutionEngine &EE); ExecutionEngineState(ExecutionEngine &EE);
GlobalAddressMapTy & GlobalAddressMapTy &getGlobalAddressMap(const MutexGuard &) {
getGlobalAddressMap(const MutexGuard &) {
return GlobalAddressMap; return GlobalAddressMap;
} }
@ -80,23 +81,41 @@ public:
return GlobalAddressReverseMap; return GlobalAddressReverseMap;
} }
// Returns the address ToUnmap was mapped to. /// \brief Erase an entry from the mapping table.
///
/// \returns The address that \arg ToUnmap was happed to.
void *RemoveMapping(const MutexGuard &, const GlobalValue *ToUnmap); void *RemoveMapping(const MutexGuard &, const GlobalValue *ToUnmap);
}; };
/// \brief Abstract interface for implementation execution of LLVM modules,
/// designed to support both interpreter and just-in-time (JIT) compiler
/// implementations.
class ExecutionEngine { class ExecutionEngine {
const TargetData *TD; /// The state object holding the global address mapping, which must be
/// accessed synchronously.
//
// FIXME: There is no particular need the entire map needs to be
// synchronized. Wouldn't a reader-writer design be better here?
ExecutionEngineState EEState; ExecutionEngineState EEState;
/// The target data for the platform for which execution is being performed.
const TargetData *TD;
/// Whether lazy JIT compilation is enabled.
bool CompilingLazily; bool CompilingLazily;
/// Whether JIT compilation of external global variables is allowed.
bool GVCompilationDisabled; bool GVCompilationDisabled;
/// Whether the JIT should perform lookups of external symbols (e.g.,
/// using dlsym).
bool SymbolSearchingDisabled; bool SymbolSearchingDisabled;
friend class EngineBuilder; // To allow access to JITCtor and InterpCtor. friend class EngineBuilder; // To allow access to JITCtor and InterpCtor.
protected: protected:
/// Modules - This is a list of Modules that we are JIT'ing from. We use a /// The list of Modules that we are JIT'ing from. We use a SmallVector to
/// smallvector to optimize for the case where there is only one module. /// optimize for the case where there is only one module.
SmallVector<Module*, 1> Modules; SmallVector<Module*, 1> Modules;
void setTargetData(const TargetData *td) { void setTargetData(const TargetData *td) {
@ -104,11 +123,11 @@ protected:
} }
/// getMemoryforGV - Allocate memory for a global variable. /// getMemoryforGV - Allocate memory for a global variable.
virtual char* getMemoryForGV(const GlobalVariable* GV); virtual char *getMemoryForGV(const GlobalVariable *GV);
// To avoid having libexecutionengine depend on the JIT and interpreter // To avoid having libexecutionengine depend on the JIT and interpreter
// libraries, the JIT and Interpreter set these functions to ctor pointers // libraries, the JIT and Interpreter set these functions to ctor pointers at
// at startup time if they are linked in. // startup time if they are linked in.
static ExecutionEngine *(*JITCtor)( static ExecutionEngine *(*JITCtor)(
Module *M, Module *M,
std::string *ErrorStr, std::string *ErrorStr,
@ -123,8 +142,9 @@ protected:
std::string *ErrorStr); std::string *ErrorStr);
/// LazyFunctionCreator - If an unknown function is needed, this function /// LazyFunctionCreator - If an unknown function is needed, this function
/// pointer is invoked to create it. If this returns null, the JIT will abort. /// pointer is invoked to create it. If this returns null, the JIT will
void* (*LazyFunctionCreator)(const std::string &); /// abort.
void *(*LazyFunctionCreator)(const std::string &);
/// ExceptionTableRegister - If Exception Handling is set, the JIT will /// ExceptionTableRegister - If Exception Handling is set, the JIT will
/// register dwarf tables with this function. /// register dwarf tables with this function.
@ -134,10 +154,10 @@ protected:
std::vector<void*> AllExceptionTables; std::vector<void*> AllExceptionTables;
public: public:
/// lock - This lock is protects the ExecutionEngine, JIT, JITResolver and /// lock - This lock protects the ExecutionEngine, JIT, JITResolver and
/// JITEmitter classes. It must be held while changing the internal state of /// JITEmitter classes. It must be held while changing the internal state of
/// any of those classes. /// any of those classes.
sys::Mutex lock; // Used to make this class and subclasses thread-safe sys::Mutex lock;
//===--------------------------------------------------------------------===// //===--------------------------------------------------------------------===//
// ExecutionEngine Startup // ExecutionEngine Startup
@ -148,20 +168,18 @@ public:
/// create - This is the factory method for creating an execution engine which /// create - This is the factory method for creating an execution engine which
/// is appropriate for the current machine. This takes ownership of the /// is appropriate for the current machine. This takes ownership of the
/// module. /// module.
///
/// \param GVsWithCode - Allocating globals with code breaks
/// freeMachineCodeForFunction and is probably unsafe and bad for performance.
/// However, we have clients who depend on this behavior, so we must support
/// it. Eventually, when we're willing to break some backwards compatability,
/// this flag should be flipped to false, so that by default
/// freeMachineCodeForFunction works.
static ExecutionEngine *create(Module *M, static ExecutionEngine *create(Module *M,
bool ForceInterpreter = false, bool ForceInterpreter = false,
std::string *ErrorStr = 0, std::string *ErrorStr = 0,
CodeGenOpt::Level OptLevel = CodeGenOpt::Level OptLevel =
CodeGenOpt::Default, CodeGenOpt::Default,
// Allocating globals with code breaks
// freeMachineCodeForFunction and is probably
// unsafe and bad for performance. However,
// we have clients who depend on this
// behavior, so we must support it.
// Eventually, when we're willing to break
// some backwards compatability, this flag
// should be flipped to false, so that by
// default freeMachineCodeForFunction works.
bool GVsWithCode = true); bool GVsWithCode = true);
/// createJIT - This is the factory method for creating a JIT for the current /// createJIT - This is the factory method for creating a JIT for the current
@ -186,11 +204,10 @@ public:
Modules.push_back(M); Modules.push_back(M);
} }
//===----------------------------------------------------------------------===// //===--------------------------------------------------------------------===//
const TargetData *getTargetData() const { return TD; } const TargetData *getTargetData() const { return TD; }
/// removeModule - Remove a Module from the list of modules. Returns true if /// removeModule - Remove a Module from the list of modules. Returns true if
/// M is found. /// M is found.
virtual bool removeModule(Module *M); virtual bool removeModule(Module *M);
@ -202,17 +219,19 @@ public:
/// runFunction - Execute the specified function with the specified arguments, /// runFunction - Execute the specified function with the specified arguments,
/// and return the result. /// and return the result.
///
virtual GenericValue runFunction(Function *F, virtual GenericValue runFunction(Function *F,
const std::vector<GenericValue> &ArgValues) = 0; const std::vector<GenericValue> &ArgValues) = 0;
/// runStaticConstructorsDestructors - This method is used to execute all of /// runStaticConstructorsDestructors - This method is used to execute all of
/// the static constructors or destructors for a program, depending on the /// the static constructors or destructors for a program.
/// value of isDtors. ///
/// \param isDtors - Run the destructors instead of constructors.
void runStaticConstructorsDestructors(bool isDtors); void runStaticConstructorsDestructors(bool isDtors);
/// runStaticConstructorsDestructors - This method is used to execute all of /// runStaticConstructorsDestructors - This method is used to execute all of
/// the static constructors or destructors for a module, depending on the /// the static constructors or destructors for a particular module.
/// value of isDtors. ///
/// \param isDtors - Run the destructors instead of constructors.
void runStaticConstructorsDestructors(Module *module, bool isDtors); void runStaticConstructorsDestructors(Module *module, bool isDtors);
@ -231,8 +250,8 @@ public:
/// GlobalValue is destroyed. /// GlobalValue is destroyed.
void addGlobalMapping(const GlobalValue *GV, void *Addr); void addGlobalMapping(const GlobalValue *GV, void *Addr);
/// clearAllGlobalMappings - Clear all global mappings and start over again /// clearAllGlobalMappings - Clear all global mappings and start over again,
/// use in dynamic compilation scenarios when you want to move globals /// for use in dynamic compilation scenarios to move globals.
void clearAllGlobalMappings(); void clearAllGlobalMappings();
/// clearGlobalMappingsFromModule - Clear all global mappings that came from a /// clearGlobalMappingsFromModule - Clear all global mappings that came from a
@ -248,12 +267,10 @@ public:
/// getPointerToGlobalIfAvailable - This returns the address of the specified /// getPointerToGlobalIfAvailable - This returns the address of the specified
/// global value if it is has already been codegen'd, otherwise it returns /// global value if it is has already been codegen'd, otherwise it returns
/// null. /// null.
///
void *getPointerToGlobalIfAvailable(const GlobalValue *GV); void *getPointerToGlobalIfAvailable(const GlobalValue *GV);
/// getPointerToGlobal - This returns the address of the specified global /// getPointerToGlobal - This returns the address of the specified global
/// value. This may involve code generation if it's a function. /// value. This may involve code generation if it's a function.
///
void *getPointerToGlobal(const GlobalValue *GV); void *getPointerToGlobal(const GlobalValue *GV);
/// getPointerToFunction - The different EE's represent function bodies in /// getPointerToFunction - The different EE's represent function bodies in
@ -261,20 +278,17 @@ public:
/// pointer should look like. When F is destroyed, the ExecutionEngine will /// pointer should look like. When F is destroyed, the ExecutionEngine will
/// remove its global mapping and free any machine code. Be sure no threads /// remove its global mapping and free any machine code. Be sure no threads
/// are running inside F when that happens. /// are running inside F when that happens.
///
virtual void *getPointerToFunction(Function *F) = 0; virtual void *getPointerToFunction(Function *F) = 0;
/// getPointerToBasicBlock - The different EE's represent basic blocks in /// getPointerToBasicBlock - The different EE's represent basic blocks in
/// different ways. Return the representation for a blockaddress of the /// different ways. Return the representation for a blockaddress of the
/// specified block. /// specified block.
///
virtual void *getPointerToBasicBlock(BasicBlock *BB) = 0; virtual void *getPointerToBasicBlock(BasicBlock *BB) = 0;
/// getPointerToFunctionOrStub - If the specified function has been /// getPointerToFunctionOrStub - If the specified function has been
/// code-gen'd, return a pointer to the function. If not, compile it, or use /// code-gen'd, return a pointer to the function. If not, compile it, or use
/// a stub to implement lazy compilation if available. See /// a stub to implement lazy compilation if available. See
/// getPointerToFunction for the requirements on destroying F. /// getPointerToFunction for the requirements on destroying F.
///
virtual void *getPointerToFunctionOrStub(Function *F) { virtual void *getPointerToFunctionOrStub(Function *F) {
// Default implementation, just codegen the function. // Default implementation, just codegen the function.
return getPointerToFunction(F); return getPointerToFunction(F);
@ -288,23 +302,25 @@ public:
/// ///
const GlobalValue *getGlobalValueAtAddress(void *Addr); const GlobalValue *getGlobalValueAtAddress(void *Addr);
/// StoreValueToMemory - Stores the data in Val of type Ty at address Ptr.
/// Ptr is the address of the memory at which to store Val, cast to
/// GenericValue *. It is not a pointer to a GenericValue containing the
/// address at which to store Val.
void StoreValueToMemory(const GenericValue &Val, GenericValue *Ptr, void StoreValueToMemory(const GenericValue &Val, GenericValue *Ptr,
const Type *Ty); const Type *Ty);
void InitializeMemory(const Constant *Init, void *Addr); void InitializeMemory(const Constant *Init, void *Addr);
/// recompileAndRelinkFunction - This method is used to force a function /// recompileAndRelinkFunction - This method is used to force a function which
/// which has already been compiled to be compiled again, possibly /// has already been compiled to be compiled again, possibly after it has been
/// after it has been modified. Then the entry to the old copy is overwritten /// modified. Then the entry to the old copy is overwritten with a branch to
/// with a branch to the new copy. If there was no old copy, this acts /// the new copy. If there was no old copy, this acts just like
/// just like VM::getPointerToFunction(). /// VM::getPointerToFunction().
///
virtual void *recompileAndRelinkFunction(Function *F) = 0; virtual void *recompileAndRelinkFunction(Function *F) = 0;
/// freeMachineCodeForFunction - Release memory in the ExecutionEngine /// freeMachineCodeForFunction - Release memory in the ExecutionEngine
/// corresponding to the machine code emitted to execute this function, useful /// corresponding to the machine code emitted to execute this function, useful
/// for garbage-collecting generated code. /// for garbage-collecting generated code.
///
virtual void freeMachineCodeForFunction(Function *F) = 0; virtual void freeMachineCodeForFunction(Function *F) = 0;
/// getOrEmitGlobalVariable - Return the address of the specified global /// getOrEmitGlobalVariable - Return the address of the specified global
@ -382,8 +398,8 @@ public:
ExceptionTableDeregister = F; ExceptionTableDeregister = F;
} }
/// RegisterTable - Registers the given pointer as an exception table. It uses /// RegisterTable - Registers the given pointer as an exception table. It
/// the ExceptionTableRegister function. /// uses the ExceptionTableRegister function.
void RegisterTable(void* res) { void RegisterTable(void* res) {
if (ExceptionTableRegister) { if (ExceptionTableRegister) {
ExceptionTableRegister(res); ExceptionTableRegister(res);
@ -392,7 +408,7 @@ public:
} }
/// DeregisterAllTables - Deregisters all previously registered pointers to an /// DeregisterAllTables - Deregisters all previously registered pointers to an
/// exception tables. It uses the ExceptionTableoDeregister function. /// exception tables. It uses the ExceptionTableoDeregister function.
void DeregisterAllTables(); void DeregisterAllTables();
protected: protected:
@ -400,9 +416,6 @@ protected:
void emitGlobals(); void emitGlobals();
// EmitGlobalVariable - This method emits the specified global variable to the
// address specified in GlobalAddresses, or allocates new memory if it's not
// already in the map.
void EmitGlobalVariable(const GlobalVariable *GV); void EmitGlobalVariable(const GlobalVariable *GV);
GenericValue getConstantValue(const Constant *C); GenericValue getConstantValue(const Constant *C);
@ -423,8 +436,7 @@ namespace EngineKind {
/// stack-allocating a builder, chaining the various set* methods, and /// stack-allocating a builder, chaining the various set* methods, and
/// terminating it with a .create() call. /// terminating it with a .create() call.
class EngineBuilder { class EngineBuilder {
private:
private:
Module *M; Module *M;
EngineKind::Kind WhichEngine; EngineKind::Kind WhichEngine;
std::string *ErrorStr; std::string *ErrorStr;
@ -437,7 +449,6 @@ class EngineBuilder {
SmallVector<std::string, 4> MAttrs; SmallVector<std::string, 4> MAttrs;
/// InitEngine - Does the common initialization of default options. /// InitEngine - Does the common initialization of default options.
///
void InitEngine() { void InitEngine() {
WhichEngine = EngineKind::Either; WhichEngine = EngineKind::Either;
ErrorStr = NULL; ErrorStr = NULL;
@ -447,7 +458,7 @@ class EngineBuilder {
CMModel = CodeModel::Default; CMModel = CodeModel::Default;
} }
public: public:
/// EngineBuilder - Constructor for EngineBuilder. If create() is called and /// EngineBuilder - Constructor for EngineBuilder. If create() is called and
/// is successful, the created engine takes ownership of the module. /// is successful, the created engine takes ownership of the module.
EngineBuilder(Module *m) : M(m) { EngineBuilder(Module *m) : M(m) {

244
lib/ExecutionEngine/ExecutionEngine.cpp
View File

@ -19,6 +19,7 @@
#include "llvm/DerivedTypes.h" #include "llvm/DerivedTypes.h"
#include "llvm/Module.h" #include "llvm/Module.h"
#include "llvm/ExecutionEngine/GenericValue.h" #include "llvm/ExecutionEngine/GenericValue.h"
#include "llvm/ADT/SmallString.h"
#include "llvm/ADT/Statistic.h" #include "llvm/ADT/Statistic.h"
#include "llvm/Support/Debug.h" #include "llvm/Support/Debug.h"
#include "llvm/Support/ErrorHandling.h" #include "llvm/Support/ErrorHandling.h"
@ -68,24 +69,23 @@ ExecutionEngine::~ExecutionEngine() {
void ExecutionEngine::DeregisterAllTables() { void ExecutionEngine::DeregisterAllTables() {
if (ExceptionTableDeregister) { if (ExceptionTableDeregister) {
std::vector<void*>::iterator it = AllExceptionTables.begin(); for (std::vector<void*>::iterator it = AllExceptionTables.begin(),
std::vector<void*>::iterator ite = AllExceptionTables.end(); ie = AllExceptionTables.end(); it != ie; ++it)
for (; it != ite; ++it)
ExceptionTableDeregister(*it); ExceptionTableDeregister(*it);
AllExceptionTables.clear(); AllExceptionTables.clear();
} }
} }
namespace { namespace {
// This class automatically deletes the memory block when the GlobalVariable is /// \brief Helper class which uses a value handler to automatically deletes the
// destroyed. /// memory block when the GlobalVariable is destroyed.
class GVMemoryBlock : public CallbackVH { class GVMemoryBlock : public CallbackVH {
GVMemoryBlock(const GlobalVariable *GV) GVMemoryBlock(const GlobalVariable *GV)
: CallbackVH(const_cast<GlobalVariable*>(GV)) {} : CallbackVH(const_cast<GlobalVariable*>(GV)) {}
public: public:
// Returns the address the GlobalVariable should be written into. The /// \brief Returns the address the GlobalVariable should be written into. The
// GVMemoryBlock object prefixes that. /// GVMemoryBlock object prefixes that.
static char *Create(const GlobalVariable *GV, const TargetData& TD) { static char *Create(const GlobalVariable *GV, const TargetData& TD) {
const Type *ElTy = GV->getType()->getElementType(); const Type *ElTy = GV->getType()->getElementType();
size_t GVSize = (size_t)TD.getTypeAllocSize(ElTy); size_t GVSize = (size_t)TD.getTypeAllocSize(ElTy);
@ -107,11 +107,10 @@ public:
}; };
} // anonymous namespace } // anonymous namespace
char* ExecutionEngine::getMemoryForGV(const GlobalVariable* GV) { char *ExecutionEngine::getMemoryForGV(const GlobalVariable *GV) {
return GVMemoryBlock::Create(GV, *getTargetData()); return GVMemoryBlock::Create(GV, *getTargetData());
} }
/// removeModule - Remove a Module from the list of modules.
bool ExecutionEngine::removeModule(Module *M) { bool ExecutionEngine::removeModule(Module *M) {
for(SmallVector<Module *, 1>::iterator I = Modules.begin(), for(SmallVector<Module *, 1>::iterator I = Modules.begin(),
E = Modules.end(); I != E; ++I) { E = Modules.end(); I != E; ++I) {
@ -125,9 +124,6 @@ bool ExecutionEngine::removeModule(Module *M) {
return false; return false;
} }
/// FindFunctionNamed - Search all of the active modules to find the one that
/// defines FnName. This is very slow operation and shouldn't be used for
/// general code.
Function *ExecutionEngine::FindFunctionNamed(const char *FnName) { Function *ExecutionEngine::FindFunctionNamed(const char *FnName) {
for (unsigned i = 0, e = Modules.size(); i != e; ++i) { for (unsigned i = 0, e = Modules.size(); i != e; ++i) {
if (Function *F = Modules[i]->getFunction(FnName)) if (Function *F = Modules[i]->getFunction(FnName))
@ -137,10 +133,13 @@ Function *ExecutionEngine::FindFunctionNamed(const char *FnName) {
} }
void *ExecutionEngineState::RemoveMapping( void *ExecutionEngineState::RemoveMapping(const MutexGuard &,
const MutexGuard &, const GlobalValue *ToUnmap) { const GlobalValue *ToUnmap) {
GlobalAddressMapTy::iterator I = GlobalAddressMap.find(ToUnmap); GlobalAddressMapTy::iterator I = GlobalAddressMap.find(ToUnmap);
void *OldVal; void *OldVal;
// FIXME: This is silly, we shouldn't end up with a mapping -> 0 in the
// GlobalAddressMap.
if (I == GlobalAddressMap.end()) if (I == GlobalAddressMap.end())
OldVal = 0; OldVal = 0;
else { else {
@ -152,11 +151,6 @@ void *ExecutionEngineState::RemoveMapping(
return OldVal; return OldVal;
} }
/// addGlobalMapping - Tell the execution engine that the specified global is
/// at the specified location. This is used internally as functions are JIT'd
/// and as global variables are laid out in memory. It can and should also be
/// used by clients of the EE that want to have an LLVM global overlay
/// existing data in memory.
void ExecutionEngine::addGlobalMapping(const GlobalValue *GV, void *Addr) { void ExecutionEngine::addGlobalMapping(const GlobalValue *GV, void *Addr) {
MutexGuard locked(lock); MutexGuard locked(lock);
@ -166,7 +160,7 @@ void ExecutionEngine::addGlobalMapping(const GlobalValue *GV, void *Addr) {
assert((CurVal == 0 || Addr == 0) && "GlobalMapping already established!"); assert((CurVal == 0 || Addr == 0) && "GlobalMapping already established!");
CurVal = Addr; CurVal = Addr;
// If we are using the reverse mapping, add it too // If we are using the reverse mapping, add it too.
if (!EEState.getGlobalAddressReverseMap(locked).empty()) { if (!EEState.getGlobalAddressReverseMap(locked).empty()) {
AssertingVH<const GlobalValue> &V = AssertingVH<const GlobalValue> &V =
EEState.getGlobalAddressReverseMap(locked)[Addr]; EEState.getGlobalAddressReverseMap(locked)[Addr];
@ -175,8 +169,6 @@ void ExecutionEngine::addGlobalMapping(const GlobalValue *GV, void *Addr) {
} }
} }
/// clearAllGlobalMappings - Clear all global mappings and start over again
/// use in dynamic compilation scenarios when you want to move globals
void ExecutionEngine::clearAllGlobalMappings() { void ExecutionEngine::clearAllGlobalMappings() {
MutexGuard locked(lock); MutexGuard locked(lock);
@ -184,23 +176,16 @@ void ExecutionEngine::clearAllGlobalMappings() {
EEState.getGlobalAddressReverseMap(locked).clear(); EEState.getGlobalAddressReverseMap(locked).clear();
} }
/// clearGlobalMappingsFromModule - Clear all global mappings that came from a
/// particular module, because it has been removed from the JIT.
void ExecutionEngine::clearGlobalMappingsFromModule(Module *M) { void ExecutionEngine::clearGlobalMappingsFromModule(Module *M) {
MutexGuard locked(lock); MutexGuard locked(lock);
for (Module::iterator FI = M->begin(), FE = M->end(); FI != FE; ++FI) { for (Module::iterator FI = M->begin(), FE = M->end(); FI != FE; ++FI)
EEState.RemoveMapping(locked, FI); EEState.RemoveMapping(locked, FI);
}
for (Module::global_iterator GI = M->global_begin(), GE = M->global_end(); for (Module::global_iterator GI = M->global_begin(), GE = M->global_end();
GI != GE; ++GI) { GI != GE; ++GI)
EEState.RemoveMapping(locked, GI); EEState.RemoveMapping(locked, GI);
}
} }
/// updateGlobalMapping - Replace an existing mapping for GV with a new
/// address. This updates both maps as required. If "Addr" is null, the
/// entry for the global is removed from the mappings.
void *ExecutionEngine::updateGlobalMapping(const GlobalValue *GV, void *Addr) { void *ExecutionEngine::updateGlobalMapping(const GlobalValue *GV, void *Addr) {
MutexGuard locked(lock); MutexGuard locked(lock);
@ -208,9 +193,8 @@ void *ExecutionEngine::updateGlobalMapping(const GlobalValue *GV, void *Addr) {
EEState.getGlobalAddressMap(locked); EEState.getGlobalAddressMap(locked);
// Deleting from the mapping? // Deleting from the mapping?
if (Addr == 0) { if (Addr == 0)
return EEState.RemoveMapping(locked, GV); return EEState.RemoveMapping(locked, GV);
}
void *&CurVal = Map[GV]; void *&CurVal = Map[GV];
void *OldVal = CurVal; void *OldVal = CurVal;
@ -219,7 +203,7 @@ void *ExecutionEngine::updateGlobalMapping(const GlobalValue *GV, void *Addr) {
EEState.getGlobalAddressReverseMap(locked).erase(CurVal); EEState.getGlobalAddressReverseMap(locked).erase(CurVal);
CurVal = Addr; CurVal = Addr;
// If we are using the reverse mapping, add it too // If we are using the reverse mapping, add it too.
if (!EEState.getGlobalAddressReverseMap(locked).empty()) { if (!EEState.getGlobalAddressReverseMap(locked).empty()) {
AssertingVH<const GlobalValue> &V = AssertingVH<const GlobalValue> &V =
EEState.getGlobalAddressReverseMap(locked)[Addr]; EEState.getGlobalAddressReverseMap(locked)[Addr];
@ -229,9 +213,6 @@ void *ExecutionEngine::updateGlobalMapping(const GlobalValue *GV, void *Addr) {
return OldVal; return OldVal;
} }
/// getPointerToGlobalIfAvailable - This returns the address of the specified
/// global value if it is has already been codegen'd, otherwise it returns null.
///
void *ExecutionEngine::getPointerToGlobalIfAvailable(const GlobalValue *GV) { void *ExecutionEngine::getPointerToGlobalIfAvailable(const GlobalValue *GV) {
MutexGuard locked(lock); MutexGuard locked(lock);
@ -240,9 +221,6 @@ void *ExecutionEngine::getPointerToGlobalIfAvailable(const GlobalValue *GV) {
return I != EEState.getGlobalAddressMap(locked).end() ? I->second : 0; return I != EEState.getGlobalAddressMap(locked).end() ? I->second : 0;
} }
/// getGlobalValueAtAddress - Return the LLVM global value object that starts
/// at the specified address.
///
const GlobalValue *ExecutionEngine::getGlobalValueAtAddress(void *Addr) { const GlobalValue *ExecutionEngine::getGlobalValueAtAddress(void *Addr) {
MutexGuard locked(lock); MutexGuard locked(lock);
@ -311,54 +289,50 @@ void *ArgvArray::reset(LLVMContext &C, ExecutionEngine *EE,
return Array; return Array;
} }
/// runStaticConstructorsDestructors - This method is used to execute all of
/// the static constructors or destructors for a module, depending on the
/// value of isDtors.
void ExecutionEngine::runStaticConstructorsDestructors(Module *module, void ExecutionEngine::runStaticConstructorsDestructors(Module *module,
bool isDtors) { bool isDtors) {
const char *Name = isDtors ? "llvm.global_dtors" : "llvm.global_ctors"; const char *Name = isDtors ? "llvm.global_dtors" : "llvm.global_ctors";
GlobalVariable *GV = module->getNamedGlobal(Name);
// Execute global ctors/dtors for each module in the program. // If this global has internal linkage, or if it has a use, then it must be
// an old-style (llvmgcc3) static ctor with __main linked in and in use. If
// this is the case, don't execute any of the global ctors, __main will do
// it.
if (!GV || GV->isDeclaration() || GV->hasLocalLinkage()) return;
GlobalVariable *GV = module->getNamedGlobal(Name); // Should be an array of '{ int, void ()* }' structs. The first value is
// the init priority, which we ignore.
ConstantArray *InitList = dyn_cast<ConstantArray>(GV->getInitializer());
if (!InitList) return;
for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i) {
ConstantStruct *CS =
dyn_cast<ConstantStruct>(InitList->getOperand(i));
if (!CS) continue;
if (CS->getNumOperands() != 2) return; // Not array of 2-element structs.
// If this global has internal linkage, or if it has a use, then it must be Constant *FP = CS->getOperand(1);
// an old-style (llvmgcc3) static ctor with __main linked in and in use. If if (FP->isNullValue())
// this is the case, don't execute any of the global ctors, __main will do break; // Found a null terminator, exit.
// it.
if (!GV || GV->isDeclaration() || GV->hasLocalLinkage()) return;
// Should be an array of '{ int, void ()* }' structs. The first value is // Strip off constant expression casts.
// the init priority, which we ignore. if (ConstantExpr *CE = dyn_cast<ConstantExpr>(FP))
ConstantArray *InitList = dyn_cast<ConstantArray>(GV->getInitializer()); if (CE->isCast())
if (!InitList) return; FP = CE->getOperand(0);
for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i)
if (ConstantStruct *CS =
dyn_cast<ConstantStruct>(InitList->getOperand(i))) {
if (CS->getNumOperands() != 2) return; // Not array of 2-element structs.
Constant *FP = CS->getOperand(1); // Execute the ctor/dtor function!
if (FP->isNullValue()) if (Function *F = dyn_cast<Function>(FP))
break; // Found a null terminator, exit. runFunction(F, std::vector<GenericValue>());
if (ConstantExpr *CE = dyn_cast<ConstantExpr>(FP)) // FIXME: It is marginally lame that we just do nothing here if we see an
if (CE->isCast()) // entry we don't recognize. It might not be unreasonable for the verifier
FP = CE->getOperand(0); // to not even allow this and just assert here.
if (Function *F = dyn_cast<Function>(FP)) { }
// Execute the ctor/dtor function!
runFunction(F, std::vector<GenericValue>());
}
}
} }
/// runStaticConstructorsDestructors - This method is used to execute all of
/// the static constructors or destructors for a program, depending on the
/// value of isDtors.
void ExecutionEngine::runStaticConstructorsDestructors(bool isDtors) { void ExecutionEngine::runStaticConstructorsDestructors(bool isDtors) {
// Execute global ctors/dtors for each module in the program. // Execute global ctors/dtors for each module in the program.
for (unsigned m = 0, e = Modules.size(); m != e; ++m) for (unsigned i = 0, e = Modules.size(); i != e; ++i)
runStaticConstructorsDestructors(Modules[m], isDtors); runStaticConstructorsDestructors(Modules[i], isDtors);
} }
#ifndef NDEBUG #ifndef NDEBUG
@ -372,9 +346,6 @@ static bool isTargetNullPtr(ExecutionEngine *EE, void *Loc) {
} }
#endif #endif
/// runFunctionAsMain - This is a helper function which wraps runFunction to
/// handle the common task of starting up main with the specified argc, argv,
/// and envp parameters.
int ExecutionEngine::runFunctionAsMain(Function *Fn, int ExecutionEngine::runFunctionAsMain(Function *Fn,
const std::vector<std::string> &argv, const std::vector<std::string> &argv,
const char * const * envp) { const char * const * envp) {
@ -386,31 +357,19 @@ int ExecutionEngine::runFunctionAsMain(Function *Fn,
unsigned NumArgs = Fn->getFunctionType()->getNumParams(); unsigned NumArgs = Fn->getFunctionType()->getNumParams();
const FunctionType *FTy = Fn->getFunctionType(); const FunctionType *FTy = Fn->getFunctionType();
const Type* PPInt8Ty = Type::getInt8PtrTy(Fn->getContext())->getPointerTo(); const Type* PPInt8Ty = Type::getInt8PtrTy(Fn->getContext())->getPointerTo();
switch (NumArgs) {
case 3: // Check the argument types.
if (FTy->getParamType(2) != PPInt8Ty) { if (NumArgs > 3)
report_fatal_error("Invalid type for third argument of main() supplied"); report_fatal_error("Invalid number of arguments of main() supplied");
} if (NumArgs >= 3 && FTy->getParamType(2) != PPInt8Ty)
// FALLS THROUGH report_fatal_error("Invalid type for third argument of main() supplied");
case 2: if (NumArgs >= 2 && FTy->getParamType(1) != PPInt8Ty)
if (FTy->getParamType(1) != PPInt8Ty) { report_fatal_error("Invalid type for second argument of main() supplied");
report_fatal_error("Invalid type for second argument of main() supplied"); if (NumArgs >= 1 && !FTy->getParamType(0)->isIntegerTy(32))
} report_fatal_error("Invalid type for first argument of main() supplied");
// FALLS THROUGH if (!FTy->getReturnType()->isIntegerTy() &&
case 1: !FTy->getReturnType()->isVoidTy())
if (!FTy->getParamType(0)->isIntegerTy(32)) { report_fatal_error("Invalid return type of main() supplied");
report_fatal_error("Invalid type for first argument of main() supplied");
}
// FALLS THROUGH
case 0:
if (!FTy->getReturnType()->isIntegerTy() &&
!FTy->getReturnType()->isVoidTy()) {
report_fatal_error("Invalid return type of main() supplied");
}
break;
default:
report_fatal_error("Invalid number of arguments of main() supplied");
}
ArgvArray CArgv; ArgvArray CArgv;
ArgvArray CEnv; ArgvArray CEnv;
@ -430,13 +389,10 @@ int ExecutionEngine::runFunctionAsMain(Function *Fn,
} }
} }
} }
return runFunction(Fn, GVArgs).IntVal.getZExtValue(); return runFunction(Fn, GVArgs).IntVal.getZExtValue();
} }
/// If possible, create a JIT, unless the caller specifically requests an
/// Interpreter or there's an error. If even an Interpreter cannot be created,
/// NULL is returned.
///
ExecutionEngine *ExecutionEngine::create(Module *M, ExecutionEngine *ExecutionEngine::create(Module *M,
bool ForceInterpreter, bool ForceInterpreter,
std::string *ErrorStr, std::string *ErrorStr,
@ -497,20 +453,17 @@ ExecutionEngine *EngineBuilder::create() {
if (ErrorStr) if (ErrorStr)
*ErrorStr = "JIT has not been linked in."; *ErrorStr = "JIT has not been linked in.";
} }
return 0; return 0;
} }
/// getPointerToGlobal - This returns the address of the specified global
/// value. This may involve code generation if it's a function.
///
void *ExecutionEngine::getPointerToGlobal(const GlobalValue *GV) { void *ExecutionEngine::getPointerToGlobal(const GlobalValue *GV) {
if (Function *F = const_cast<Function*>(dyn_cast<Function>(GV))) if (Function *F = const_cast<Function*>(dyn_cast<Function>(GV)))
return getPointerToFunction(F); return getPointerToFunction(F);
MutexGuard locked(lock); MutexGuard locked(lock);
void *p = EEState.getGlobalAddressMap(locked)[GV]; if (void *P = EEState.getGlobalAddressMap(locked)[GV])
if (p) return P;
return p;
// Global variable might have been added since interpreter started. // Global variable might have been added since interpreter started.
if (GlobalVariable *GVar = if (GlobalVariable *GVar =
@ -518,12 +471,12 @@ void *ExecutionEngine::getPointerToGlobal(const GlobalValue *GV) {
EmitGlobalVariable(GVar); EmitGlobalVariable(GVar);
else else
llvm_unreachable("Global hasn't had an address allocated yet!"); llvm_unreachable("Global hasn't had an address allocated yet!");
return EEState.getGlobalAddressMap(locked)[GV]; return EEState.getGlobalAddressMap(locked)[GV];
} }
/// This function converts a Constant* into a GenericValue. The interesting /// \brief Converts a Constant* into a GenericValue, including handling of
/// part is if C is a ConstantExpr. /// ConstantExpr values.
/// @brief Get a GenericValue for a Constant*
GenericValue ExecutionEngine::getConstantValue(const Constant *C) { GenericValue ExecutionEngine::getConstantValue(const Constant *C) {
// If its undefined, return the garbage. // If its undefined, return the garbage.
if (isa<UndefValue>(C)) { if (isa<UndefValue>(C)) {
@ -543,7 +496,7 @@ GenericValue ExecutionEngine::getConstantValue(const Constant *C) {
return Result; return Result;
} }
// If the value is a ConstantExpr // Otherwise, if the value is a ConstantExpr...
if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) { if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) {
Constant *Op0 = CE->getOperand(0); Constant *Op0 = CE->getOperand(0);
switch (CE->getOpcode()) { switch (CE->getOpcode()) {
@ -778,12 +731,14 @@ GenericValue ExecutionEngine::getConstantValue(const Constant *C) {
default: default:
break; break;
} }
std::string msg;
raw_string_ostream Msg(msg); SmallString<256> Msg;
Msg << "ConstantExpr not handled: " << *CE; raw_svector_ostream OS(Msg);
report_fatal_error(Msg.str()); OS << "ConstantExpr not handled: " << *CE;
report_fatal_error(OS.str());
} }
// Otherwise, we have a simple constant.
GenericValue Result; GenericValue Result;
switch (C->getType()->getTypeID()) { switch (C->getType()->getTypeID()) {
case Type::FloatTyID: case Type::FloatTyID:
@ -814,11 +769,12 @@ GenericValue ExecutionEngine::getConstantValue(const Constant *C) {
llvm_unreachable("Unknown constant pointer type!"); llvm_unreachable("Unknown constant pointer type!");
break; break;
default: default:
std::string msg; SmallString<256> Msg;
raw_string_ostream Msg(msg); raw_svector_ostream OS(Msg);
Msg << "ERROR: Constant unimplemented for type: " << *C->getType(); OS << "ERROR: Constant unimplemented for type: " << *C->getType();
report_fatal_error(Msg.str()); report_fatal_error(OS.str());
} }
return Result; return Result;
} }
@ -829,11 +785,11 @@ static void StoreIntToMemory(const APInt &IntVal, uint8_t *Dst,
assert((IntVal.getBitWidth()+7)/8 >= StoreBytes && "Integer too small!"); assert((IntVal.getBitWidth()+7)/8 >= StoreBytes && "Integer too small!");
uint8_t *Src = (uint8_t *)IntVal.getRawData(); uint8_t *Src = (uint8_t *)IntVal.getRawData();
if (sys::isLittleEndianHost()) if (sys::isLittleEndianHost()) {
// Little-endian host - the source is ordered from LSB to MSB. Order the // Little-endian host - the source is ordered from LSB to MSB. Order the
// destination from LSB to MSB: Do a straight copy. // destination from LSB to MSB: Do a straight copy.
memcpy(Dst, Src, StoreBytes); memcpy(Dst, Src, StoreBytes);
else { } else {
// Big-endian host - the source is an array of 64 bit words ordered from // Big-endian host - the source is an array of 64 bit words ordered from
// LSW to MSW. Each word is ordered from MSB to LSB. Order the destination // LSW to MSW. Each word is ordered from MSB to LSB. Order the destination
// from MSB to LSB: Reverse the word order, but not the bytes in a word. // from MSB to LSB: Reverse the word order, but not the bytes in a word.
@ -848,10 +804,6 @@ static void StoreIntToMemory(const APInt &IntVal, uint8_t *Dst,
} }
} }
/// StoreValueToMemory - Stores the data in Val of type Ty at address Ptr. Ptr
/// is the address of the memory at which to store Val, cast to GenericValue *.
/// It is not a pointer to a GenericValue containing the address at which to
/// store Val.
void ExecutionEngine::StoreValueToMemory(const GenericValue &Val, void ExecutionEngine::StoreValueToMemory(const GenericValue &Val,
GenericValue *Ptr, const Type *Ty) { GenericValue *Ptr, const Type *Ty) {
const unsigned StoreBytes = getTargetData()->getTypeStoreSize(Ty); const unsigned StoreBytes = getTargetData()->getTypeStoreSize(Ty);
@ -942,16 +894,13 @@ void ExecutionEngine::LoadValueFromMemory(GenericValue &Result,
break; break;
} }
default: default:
std::string msg; SmallString<256> Msg;
raw_string_ostream Msg(msg); raw_svector_ostream OS(Msg);
Msg << "Cannot load value of type " << *Ty << "!"; OS << "Cannot load value of type " << *Ty << "!";
report_fatal_error(Msg.str()); report_fatal_error(OS.str());
} }
} }
// InitializeMemory - Recursive function to apply a Constant value into the
// specified memory location...
//
void ExecutionEngine::InitializeMemory(const Constant *Init, void *Addr) { void ExecutionEngine::InitializeMemory(const Constant *Init, void *Addr) {
DEBUG(dbgs() << "JIT: Initializing " << Addr << " "); DEBUG(dbgs() << "JIT: Initializing " << Addr << " ");
DEBUG(Init->dump()); DEBUG(Init->dump());
@ -984,20 +933,17 @@ void ExecutionEngine::InitializeMemory(const Constant *Init, void *Addr) {
return; return;
} }
dbgs() << "Bad Type: " << *Init->getType() << "\n"; DEBUG(dbgs() << "Bad Type: " << *Init->getType() << "\n");
llvm_unreachable("Unknown constant type to initialize memory with!"); llvm_unreachable("Unknown constant type to initialize memory with!");
} }
/// EmitGlobals - Emit all of the global variables to memory, storing their /// EmitGlobals - Emit all of the global variables to memory, storing their
/// addresses into GlobalAddress. This must make sure to copy the contents of /// addresses into GlobalAddress. This must make sure to copy the contents of
/// their initializers into the memory. /// their initializers into the memory.
///
void ExecutionEngine::emitGlobals() { void ExecutionEngine::emitGlobals() {
// Loop over all of the global variables in the program, allocating the memory // Loop over all of the global variables in the program, allocating the memory
// to hold them. If there is more than one module, do a prepass over globals // to hold them. If there is more than one module, do a prepass over globals
// to figure out how the different modules should link together. // to figure out how the different modules should link together.
//
std::map<std::pair<std::string, const Type*>, std::map<std::pair<std::string, const Type*>,
const GlobalValue*> LinkedGlobalsMap; const GlobalValue*> LinkedGlobalsMap;
@ -1123,18 +1069,20 @@ ExecutionEngineState::ExecutionEngineState(ExecutionEngine &EE)
: EE(EE), GlobalAddressMap(this) { : EE(EE), GlobalAddressMap(this) {
} }
sys::Mutex *ExecutionEngineState::AddressMapConfig::getMutex( sys::Mutex *
ExecutionEngineState *EES) { ExecutionEngineState::AddressMapConfig::getMutex(ExecutionEngineState *EES) {
return &EES->EE.lock; return &EES->EE.lock;
} }
void ExecutionEngineState::AddressMapConfig::onDelete(
ExecutionEngineState *EES, const GlobalValue *Old) { void ExecutionEngineState::AddressMapConfig::onDelete(ExecutionEngineState *EES,
const GlobalValue *Old) {
void *OldVal = EES->GlobalAddressMap.lookup(Old); void *OldVal = EES->GlobalAddressMap.lookup(Old);
EES->GlobalAddressReverseMap.erase(OldVal); EES->GlobalAddressReverseMap.erase(OldVal);
} }
void ExecutionEngineState::AddressMapConfig::onRAUW( void ExecutionEngineState::AddressMapConfig::onRAUW(ExecutionEngineState *,
ExecutionEngineState *, const GlobalValue *, const GlobalValue *) { const GlobalValue *,
const GlobalValue *) {
assert(false && "The ExecutionEngine doesn't know how to handle a" assert(false && "The ExecutionEngine doesn't know how to handle a"
" RAUW on a value it has a global mapping for."); " RAUW on a value it has a global mapping for.");
} }