Misha Brukman | 857c21b | 2003-10-10 17:45:12 +0000 | [diff] [blame] | 1 | //===-- ExecutionEngine.cpp - Common Implementation shared by EEs ---------===// |
Misha Brukman | 835702a | 2005-04-21 22:36:52 +0000 | [diff] [blame] | 2 | // |
John Criswell | 482202a | 2003-10-20 19:43:21 +0000 | [diff] [blame] | 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | f3ebc3f | 2007-12-29 20:36:04 +0000 | [diff] [blame] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Misha Brukman | 835702a | 2005-04-21 22:36:52 +0000 | [diff] [blame] | 7 | // |
John Criswell | 482202a | 2003-10-20 19:43:21 +0000 | [diff] [blame] | 8 | //===----------------------------------------------------------------------===// |
Misha Brukman | 835702a | 2005-04-21 22:36:52 +0000 | [diff] [blame] | 9 | // |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 10 | // This file defines the common interface used by the various execution engine |
| 11 | // subclasses. |
| 12 | // |
| 13 | //===----------------------------------------------------------------------===// |
| 14 | |
Jeffrey Yasskin | 6bf87df | 2009-08-07 19:54:29 +0000 | [diff] [blame] | 15 | #include "llvm/ExecutionEngine/ExecutionEngine.h" |
Benjamin Kramer | 0a446fd | 2015-03-01 21:28:53 +0000 | [diff] [blame] | 16 | #include "llvm/ADT/STLExtras.h" |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 17 | #include "llvm/ADT/SmallString.h" |
Chris Lattner | 390d78b | 2009-08-23 22:49:13 +0000 | [diff] [blame] | 18 | #include "llvm/ADT/Statistic.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 19 | #include "llvm/ExecutionEngine/GenericValue.h" |
Lang Hames | b5c7b1f | 2014-11-26 16:54:40 +0000 | [diff] [blame] | 20 | #include "llvm/ExecutionEngine/JITEventListener.h" |
Lang Hames | 633fe14 | 2015-03-30 03:37:06 +0000 | [diff] [blame] | 21 | #include "llvm/ExecutionEngine/RTDyldMemoryManager.h" |
Chandler Carruth | 9fb823b | 2013-01-02 11:36:10 +0000 | [diff] [blame] | 22 | #include "llvm/IR/Constants.h" |
| 23 | #include "llvm/IR/DataLayout.h" |
| 24 | #include "llvm/IR/DerivedTypes.h" |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 25 | #include "llvm/IR/Mangler.h" |
Chandler Carruth | 9fb823b | 2013-01-02 11:36:10 +0000 | [diff] [blame] | 26 | #include "llvm/IR/Module.h" |
| 27 | #include "llvm/IR/Operator.h" |
Chandler Carruth | 4220e9c | 2014-03-04 11:17:44 +0000 | [diff] [blame] | 28 | #include "llvm/IR/ValueHandle.h" |
Rafael Espindola | dd39657 | 2014-08-01 19:28:15 +0000 | [diff] [blame] | 29 | #include "llvm/Object/Archive.h" |
David Blaikie | 35907d8 | 2014-04-29 22:04:55 +0000 | [diff] [blame] | 30 | #include "llvm/Object/ObjectFile.h" |
Reid Spencer | 7c16caa | 2004-09-01 22:55:40 +0000 | [diff] [blame] | 31 | #include "llvm/Support/Debug.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 32 | #include "llvm/Support/DynamicLibrary.h" |
Torok Edwin | 6c2d233 | 2009-07-07 17:32:34 +0000 | [diff] [blame] | 33 | #include "llvm/Support/ErrorHandling.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 34 | #include "llvm/Support/Host.h" |
Chris Lattner | 6d8dd18 | 2006-05-08 22:00:52 +0000 | [diff] [blame] | 35 | #include "llvm/Support/MutexGuard.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 36 | #include "llvm/Support/TargetRegistry.h" |
Torok Edwin | ccb29cd | 2009-07-11 13:10:19 +0000 | [diff] [blame] | 37 | #include "llvm/Support/raw_ostream.h" |
Dylan Noblesmith | 8418fdc | 2011-05-13 21:51:29 +0000 | [diff] [blame] | 38 | #include "llvm/Target/TargetMachine.h" |
Anton Korobeynikov | 579f071 | 2008-02-20 11:08:44 +0000 | [diff] [blame] | 39 | #include <cmath> |
| 40 | #include <cstring> |
Chris Lattner | 29681de | 2003-11-19 21:08:57 +0000 | [diff] [blame] | 41 | using namespace llvm; |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 42 | |
Chandler Carruth | f58e376 | 2014-04-22 03:04:17 +0000 | [diff] [blame] | 43 | #define DEBUG_TYPE "jit" |
| 44 | |
Chris Lattner | c346ecd | 2006-12-19 22:43:32 +0000 | [diff] [blame] | 45 | STATISTIC(NumInitBytes, "Number of bytes of global vars initialized"); |
| 46 | STATISTIC(NumGlobals , "Number of global vars initialized"); |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 47 | |
Daniel Dunbar | 70ff8b0 | 2010-11-17 16:06:37 +0000 | [diff] [blame] | 48 | ExecutionEngine *(*ExecutionEngine::MCJITCtor)( |
David Blaikie | 196e323 | 2014-09-02 22:41:07 +0000 | [diff] [blame] | 49 | std::unique_ptr<Module> M, std::string *ErrorStr, |
Lang Hames | 633fe14 | 2015-03-30 03:37:06 +0000 | [diff] [blame] | 50 | std::shared_ptr<MCJITMemoryManager> MemMgr, |
| 51 | std::shared_ptr<RuntimeDyld::SymbolResolver> Resolver, |
Lang Hames | 4a5697e | 2014-12-03 00:51:19 +0000 | [diff] [blame] | 52 | std::unique_ptr<TargetMachine> TM) = nullptr; |
Lang Hames | 93de2a1 | 2015-01-23 21:25:00 +0000 | [diff] [blame] | 53 | |
| 54 | ExecutionEngine *(*ExecutionEngine::OrcMCJITReplacementCtor)( |
Lang Hames | 633fe14 | 2015-03-30 03:37:06 +0000 | [diff] [blame] | 55 | std::string *ErrorStr, std::shared_ptr<MCJITMemoryManager> MemMgr, |
| 56 | std::shared_ptr<RuntimeDyld::SymbolResolver> Resolver, |
Lang Hames | 93de2a1 | 2015-01-23 21:25:00 +0000 | [diff] [blame] | 57 | std::unique_ptr<TargetMachine> TM) = nullptr; |
| 58 | |
Rafael Espindola | 2a8a279 | 2014-08-19 04:04:25 +0000 | [diff] [blame] | 59 | ExecutionEngine *(*ExecutionEngine::InterpCtor)(std::unique_ptr<Module> M, |
Craig Topper | 2617dcc | 2014-04-15 06:32:26 +0000 | [diff] [blame] | 60 | std::string *ErrorStr) =nullptr; |
Chris Lattner | 2d52c1b | 2006-03-22 06:07:50 +0000 | [diff] [blame] | 61 | |
Lang Hames | 7ea98e1 | 2014-11-27 01:41:16 +0000 | [diff] [blame] | 62 | void JITEventListener::anchor() {} |
| 63 | |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 64 | void ExecutionEngine::Init(std::unique_ptr<Module> M) { |
Jeffrey Yasskin | 4567db4 | 2009-10-27 20:30:28 +0000 | [diff] [blame] | 65 | CompilingLazily = false; |
Evan Cheng | cdc0060 | 2008-09-24 16:25:55 +0000 | [diff] [blame] | 66 | GVCompilationDisabled = false; |
Evan Cheng | 84a9055 | 2008-06-17 16:49:02 +0000 | [diff] [blame] | 67 | SymbolSearchingDisabled = false; |
Lang Hames | bc87601 | 2014-04-18 06:48:23 +0000 | [diff] [blame] | 68 | |
| 69 | // IR module verification is enabled by default in debug builds, and disabled |
| 70 | // by default in release builds. |
| 71 | #ifndef NDEBUG |
| 72 | VerifyModules = true; |
| 73 | #else |
| 74 | VerifyModules = false; |
| 75 | #endif |
| 76 | |
Jeffrey Yasskin | 091217b | 2010-01-27 20:34:15 +0000 | [diff] [blame] | 77 | assert(M && "Module is null?"); |
Rafael Espindola | 2a8a279 | 2014-08-19 04:04:25 +0000 | [diff] [blame] | 78 | Modules.push_back(std::move(M)); |
Misha Brukman | 260b0c8 | 2003-10-16 21:18:05 +0000 | [diff] [blame] | 79 | } |
| 80 | |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 81 | ExecutionEngine::ExecutionEngine(std::unique_ptr<Module> M) |
| 82 | : DL(M->getDataLayout()), LazyFunctionCreator(nullptr) { |
| 83 | Init(std::move(M)); |
| 84 | } |
| 85 | |
| 86 | ExecutionEngine::ExecutionEngine(DataLayout DL, std::unique_ptr<Module> M) |
| 87 | : DL(std::move(DL)), LazyFunctionCreator(nullptr) { |
| 88 | Init(std::move(M)); |
| 89 | } |
| 90 | |
Brian Gaeke | 92f8b30 | 2003-09-04 22:57:27 +0000 | [diff] [blame] | 91 | ExecutionEngine::~ExecutionEngine() { |
Reid Spencer | 603682a | 2007-03-03 18:19:18 +0000 | [diff] [blame] | 92 | clearAllGlobalMappings(); |
Brian Gaeke | 92f8b30 | 2003-09-04 22:57:27 +0000 | [diff] [blame] | 93 | } |
| 94 | |
Jeffrey Yasskin | a404433 | 2010-03-27 04:53:56 +0000 | [diff] [blame] | 95 | namespace { |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 96 | /// \brief Helper class which uses a value handler to automatically deletes the |
| 97 | /// memory block when the GlobalVariable is destroyed. |
Jeffrey Yasskin | a404433 | 2010-03-27 04:53:56 +0000 | [diff] [blame] | 98 | class GVMemoryBlock : public CallbackVH { |
| 99 | GVMemoryBlock(const GlobalVariable *GV) |
| 100 | : CallbackVH(const_cast<GlobalVariable*>(GV)) {} |
| 101 | |
| 102 | public: |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 103 | /// \brief Returns the address the GlobalVariable should be written into. The |
| 104 | /// GVMemoryBlock object prefixes that. |
Micah Villmow | cdfe20b | 2012-10-08 16:38:25 +0000 | [diff] [blame] | 105 | static char *Create(const GlobalVariable *GV, const DataLayout& TD) { |
Chris Lattner | 229907c | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 106 | Type *ElTy = GV->getType()->getElementType(); |
Jeffrey Yasskin | a404433 | 2010-03-27 04:53:56 +0000 | [diff] [blame] | 107 | size_t GVSize = (size_t)TD.getTypeAllocSize(ElTy); |
| 108 | void *RawMemory = ::operator new( |
David Majnemer | f3cadce | 2014-10-20 06:13:33 +0000 | [diff] [blame] | 109 | RoundUpToAlignment(sizeof(GVMemoryBlock), |
| 110 | TD.getPreferredAlignment(GV)) |
Jeffrey Yasskin | a404433 | 2010-03-27 04:53:56 +0000 | [diff] [blame] | 111 | + GVSize); |
| 112 | new(RawMemory) GVMemoryBlock(GV); |
| 113 | return static_cast<char*>(RawMemory) + sizeof(GVMemoryBlock); |
| 114 | } |
| 115 | |
Craig Topper | b51ff60 | 2014-03-08 07:51:20 +0000 | [diff] [blame] | 116 | void deleted() override { |
Jeffrey Yasskin | a404433 | 2010-03-27 04:53:56 +0000 | [diff] [blame] | 117 | // We allocated with operator new and with some extra memory hanging off the |
| 118 | // end, so don't just delete this. I'm not sure if this is actually |
| 119 | // required. |
| 120 | this->~GVMemoryBlock(); |
| 121 | ::operator delete(this); |
| 122 | } |
| 123 | }; |
| 124 | } // anonymous namespace |
| 125 | |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 126 | char *ExecutionEngine::getMemoryForGV(const GlobalVariable *GV) { |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 127 | return GVMemoryBlock::Create(GV, getDataLayout()); |
Nicolas Geoffray | 5457ce9 | 2008-10-25 15:41:43 +0000 | [diff] [blame] | 128 | } |
| 129 | |
David Blaikie | 35907d8 | 2014-04-29 22:04:55 +0000 | [diff] [blame] | 130 | void ExecutionEngine::addObjectFile(std::unique_ptr<object::ObjectFile> O) { |
| 131 | llvm_unreachable("ExecutionEngine subclass doesn't implement addObjectFile."); |
| 132 | } |
| 133 | |
Rafael Espindola | 7271c19 | 2014-08-26 21:04:04 +0000 | [diff] [blame] | 134 | void |
| 135 | ExecutionEngine::addObjectFile(object::OwningBinary<object::ObjectFile> O) { |
| 136 | llvm_unreachable("ExecutionEngine subclass doesn't implement addObjectFile."); |
| 137 | } |
| 138 | |
Rafael Espindola | 48af1c2 | 2014-08-19 18:44:46 +0000 | [diff] [blame] | 139 | void ExecutionEngine::addArchive(object::OwningBinary<object::Archive> A) { |
Rafael Espindola | acfd628 | 2014-08-01 18:49:24 +0000 | [diff] [blame] | 140 | llvm_unreachable("ExecutionEngine subclass doesn't implement addArchive."); |
| 141 | } |
| 142 | |
Jeffrey Yasskin | 091217b | 2010-01-27 20:34:15 +0000 | [diff] [blame] | 143 | bool ExecutionEngine::removeModule(Module *M) { |
Rafael Espindola | 2a8a279 | 2014-08-19 04:04:25 +0000 | [diff] [blame] | 144 | for (auto I = Modules.begin(), E = Modules.end(); I != E; ++I) { |
| 145 | Module *Found = I->get(); |
Jeffrey Yasskin | 091217b | 2010-01-27 20:34:15 +0000 | [diff] [blame] | 146 | if (Found == M) { |
Rafael Espindola | 2a8a279 | 2014-08-19 04:04:25 +0000 | [diff] [blame] | 147 | I->release(); |
Devang Patel | 324fe89 | 2007-10-15 19:56:32 +0000 | [diff] [blame] | 148 | Modules.erase(I); |
Jeffrey Yasskin | 091217b | 2010-01-27 20:34:15 +0000 | [diff] [blame] | 149 | clearGlobalMappingsFromModule(M); |
| 150 | return true; |
Devang Patel | 324fe89 | 2007-10-15 19:56:32 +0000 | [diff] [blame] | 151 | } |
| 152 | } |
Jeffrey Yasskin | 091217b | 2010-01-27 20:34:15 +0000 | [diff] [blame] | 153 | return false; |
Nate Begeman | 617001d | 2009-01-23 19:27:28 +0000 | [diff] [blame] | 154 | } |
| 155 | |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 156 | Function *ExecutionEngine::FindFunctionNamed(const char *FnName) { |
| 157 | for (unsigned i = 0, e = Modules.size(); i != e; ++i) { |
Keno Fischer | 5f92a08 | 2015-01-27 19:29:00 +0000 | [diff] [blame] | 158 | Function *F = Modules[i]->getFunction(FnName); |
| 159 | if (F && !F->isDeclaration()) |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 160 | return F; |
| 161 | } |
Craig Topper | 2617dcc | 2014-04-15 06:32:26 +0000 | [diff] [blame] | 162 | return nullptr; |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 163 | } |
| 164 | |
Keno Fischer | 73378eb | 2015-06-20 00:55:58 +0000 | [diff] [blame] | 165 | GlobalVariable *ExecutionEngine::FindGlobalVariableNamed(const char *Name, bool AllowInternal) { |
| 166 | for (unsigned i = 0, e = Modules.size(); i != e; ++i) { |
| 167 | GlobalVariable *GV = Modules[i]->getGlobalVariable(Name,AllowInternal); |
| 168 | if (GV && !GV->isDeclaration()) |
| 169 | return GV; |
| 170 | } |
| 171 | return nullptr; |
| 172 | } |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 173 | |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 174 | uint64_t ExecutionEngineState::RemoveMapping(StringRef Name) { |
| 175 | GlobalAddressMapTy::iterator I = GlobalAddressMap.find(Name); |
| 176 | uint64_t OldVal; |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 177 | |
| 178 | // FIXME: This is silly, we shouldn't end up with a mapping -> 0 in the |
| 179 | // GlobalAddressMap. |
Jeffrey Yasskin | 307c053 | 2009-10-09 22:10:27 +0000 | [diff] [blame] | 180 | if (I == GlobalAddressMap.end()) |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 181 | OldVal = 0; |
Jeffrey Yasskin | 307c053 | 2009-10-09 22:10:27 +0000 | [diff] [blame] | 182 | else { |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 183 | GlobalAddressReverseMap.erase(I->second); |
Jeffrey Yasskin | 307c053 | 2009-10-09 22:10:27 +0000 | [diff] [blame] | 184 | OldVal = I->second; |
| 185 | GlobalAddressMap.erase(I); |
| 186 | } |
| 187 | |
Jeffrey Yasskin | 307c053 | 2009-10-09 22:10:27 +0000 | [diff] [blame] | 188 | return OldVal; |
| 189 | } |
| 190 | |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 191 | std::string ExecutionEngine::getMangledName(const GlobalValue *GV) { |
| 192 | MutexGuard locked(lock); |
Rafael Espindola | c233f74 | 2015-06-23 13:59:29 +0000 | [diff] [blame] | 193 | Mangler Mang; |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 194 | SmallString<128> FullName; |
Rafael Espindola | c233f74 | 2015-06-23 13:59:29 +0000 | [diff] [blame] | 195 | Mang.getNameWithPrefix(FullName, GV, false); |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 196 | return FullName.str(); |
| 197 | } |
| 198 | |
Chris Lattner | 6d8dd18 | 2006-05-08 22:00:52 +0000 | [diff] [blame] | 199 | void ExecutionEngine::addGlobalMapping(const GlobalValue *GV, void *Addr) { |
Zachary Turner | c04b892 | 2014-06-20 21:07:14 +0000 | [diff] [blame] | 200 | MutexGuard locked(lock); |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 201 | addGlobalMapping(getMangledName(GV), (uint64_t) Addr); |
| 202 | } |
Evan Cheng | 5cc53c3 | 2008-09-18 07:54:21 +0000 | [diff] [blame] | 203 | |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 204 | void ExecutionEngine::addGlobalMapping(StringRef Name, uint64_t Addr) { |
| 205 | MutexGuard locked(lock); |
| 206 | |
| 207 | assert(!Name.empty() && "Empty GlobalMapping symbol name!"); |
| 208 | |
| 209 | DEBUG(dbgs() << "JIT: Map \'" << Name << "\' to [" << Addr << "]\n";); |
| 210 | uint64_t &CurVal = EEState.getGlobalAddressMap()[Name]; |
Craig Topper | 2617dcc | 2014-04-15 06:32:26 +0000 | [diff] [blame] | 211 | assert((!CurVal || !Addr) && "GlobalMapping already established!"); |
Chris Lattner | 6d8dd18 | 2006-05-08 22:00:52 +0000 | [diff] [blame] | 212 | CurVal = Addr; |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 213 | |
| 214 | // If we are using the reverse mapping, add it too. |
Zachary Turner | 2f825df | 2014-06-16 20:54:28 +0000 | [diff] [blame] | 215 | if (!EEState.getGlobalAddressReverseMap().empty()) { |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 216 | std::string &V = EEState.getGlobalAddressReverseMap()[CurVal]; |
| 217 | assert((!V.empty() || !Name.empty()) && |
| 218 | "GlobalMapping already established!"); |
| 219 | V = Name; |
Chris Lattner | 6d8dd18 | 2006-05-08 22:00:52 +0000 | [diff] [blame] | 220 | } |
| 221 | } |
| 222 | |
Chris Lattner | 6d8dd18 | 2006-05-08 22:00:52 +0000 | [diff] [blame] | 223 | void ExecutionEngine::clearAllGlobalMappings() { |
Zachary Turner | c04b892 | 2014-06-20 21:07:14 +0000 | [diff] [blame] | 224 | MutexGuard locked(lock); |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 225 | |
Zachary Turner | 2f825df | 2014-06-16 20:54:28 +0000 | [diff] [blame] | 226 | EEState.getGlobalAddressMap().clear(); |
| 227 | EEState.getGlobalAddressReverseMap().clear(); |
Chris Lattner | 6d8dd18 | 2006-05-08 22:00:52 +0000 | [diff] [blame] | 228 | } |
| 229 | |
Nate Begeman | 8f83fc4 | 2008-05-21 16:34:48 +0000 | [diff] [blame] | 230 | void ExecutionEngine::clearGlobalMappingsFromModule(Module *M) { |
Zachary Turner | c04b892 | 2014-06-20 21:07:14 +0000 | [diff] [blame] | 231 | MutexGuard locked(lock); |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 232 | |
| 233 | for (Module::iterator FI = M->begin(), FE = M->end(); FI != FE; ++FI) |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 234 | EEState.RemoveMapping(getMangledName(FI)); |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 235 | for (Module::global_iterator GI = M->global_begin(), GE = M->global_end(); |
| 236 | GI != GE; ++GI) |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 237 | EEState.RemoveMapping(getMangledName(GI)); |
Nate Begeman | 8f83fc4 | 2008-05-21 16:34:48 +0000 | [diff] [blame] | 238 | } |
| 239 | |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 240 | uint64_t ExecutionEngine::updateGlobalMapping(const GlobalValue *GV, |
| 241 | void *Addr) { |
| 242 | MutexGuard locked(lock); |
| 243 | return updateGlobalMapping(getMangledName(GV), (uint64_t) Addr); |
| 244 | } |
| 245 | |
| 246 | uint64_t ExecutionEngine::updateGlobalMapping(StringRef Name, uint64_t Addr) { |
Zachary Turner | c04b892 | 2014-06-20 21:07:14 +0000 | [diff] [blame] | 247 | MutexGuard locked(lock); |
Chris Lattner | ee18173 | 2008-04-04 04:47:41 +0000 | [diff] [blame] | 248 | |
Jeffrey Yasskin | d0fc8f8 | 2009-10-23 22:37:43 +0000 | [diff] [blame] | 249 | ExecutionEngineState::GlobalAddressMapTy &Map = |
Zachary Turner | 2f825df | 2014-06-16 20:54:28 +0000 | [diff] [blame] | 250 | EEState.getGlobalAddressMap(); |
Chris Lattner | ee18173 | 2008-04-04 04:47:41 +0000 | [diff] [blame] | 251 | |
Chris Lattner | 6d8dd18 | 2006-05-08 22:00:52 +0000 | [diff] [blame] | 252 | // Deleting from the mapping? |
Craig Topper | 2617dcc | 2014-04-15 06:32:26 +0000 | [diff] [blame] | 253 | if (!Addr) |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 254 | return EEState.RemoveMapping(Name); |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 255 | |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 256 | uint64_t &CurVal = Map[Name]; |
| 257 | uint64_t OldVal = CurVal; |
Chris Lattner | ee18173 | 2008-04-04 04:47:41 +0000 | [diff] [blame] | 258 | |
Zachary Turner | 2f825df | 2014-06-16 20:54:28 +0000 | [diff] [blame] | 259 | if (CurVal && !EEState.getGlobalAddressReverseMap().empty()) |
| 260 | EEState.getGlobalAddressReverseMap().erase(CurVal); |
Chris Lattner | 6d8dd18 | 2006-05-08 22:00:52 +0000 | [diff] [blame] | 261 | CurVal = Addr; |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 262 | |
| 263 | // If we are using the reverse mapping, add it too. |
Zachary Turner | 2f825df | 2014-06-16 20:54:28 +0000 | [diff] [blame] | 264 | if (!EEState.getGlobalAddressReverseMap().empty()) { |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 265 | std::string &V = EEState.getGlobalAddressReverseMap()[CurVal]; |
| 266 | assert((!V.empty() || !Name.empty()) && |
| 267 | "GlobalMapping already established!"); |
| 268 | V = Name; |
Chris Lattner | 6d8dd18 | 2006-05-08 22:00:52 +0000 | [diff] [blame] | 269 | } |
Chris Lattner | ee18173 | 2008-04-04 04:47:41 +0000 | [diff] [blame] | 270 | return OldVal; |
Chris Lattner | 6d8dd18 | 2006-05-08 22:00:52 +0000 | [diff] [blame] | 271 | } |
| 272 | |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 273 | uint64_t ExecutionEngine::getAddressToGlobalIfAvailable(StringRef S) { |
| 274 | MutexGuard locked(lock); |
| 275 | uint64_t Address = 0; |
| 276 | ExecutionEngineState::GlobalAddressMapTy::iterator I = |
| 277 | EEState.getGlobalAddressMap().find(S); |
| 278 | if (I != EEState.getGlobalAddressMap().end()) |
| 279 | Address = I->second; |
| 280 | return Address; |
| 281 | } |
| 282 | |
| 283 | |
| 284 | void *ExecutionEngine::getPointerToGlobalIfAvailable(StringRef S) { |
| 285 | MutexGuard locked(lock); |
| 286 | if (void* Address = (void *) getAddressToGlobalIfAvailable(S)) |
| 287 | return Address; |
| 288 | return nullptr; |
| 289 | } |
| 290 | |
Chris Lattner | 6d8dd18 | 2006-05-08 22:00:52 +0000 | [diff] [blame] | 291 | void *ExecutionEngine::getPointerToGlobalIfAvailable(const GlobalValue *GV) { |
Zachary Turner | c04b892 | 2014-06-20 21:07:14 +0000 | [diff] [blame] | 292 | MutexGuard locked(lock); |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 293 | return getPointerToGlobalIfAvailable(getMangledName(GV)); |
Chris Lattner | 6d8dd18 | 2006-05-08 22:00:52 +0000 | [diff] [blame] | 294 | } |
| 295 | |
Chris Lattner | 748e857 | 2003-12-31 20:21:04 +0000 | [diff] [blame] | 296 | const GlobalValue *ExecutionEngine::getGlobalValueAtAddress(void *Addr) { |
Zachary Turner | c04b892 | 2014-06-20 21:07:14 +0000 | [diff] [blame] | 297 | MutexGuard locked(lock); |
Reid Spencer | 79876f5 | 2005-07-12 15:51:55 +0000 | [diff] [blame] | 298 | |
Chris Lattner | 748e857 | 2003-12-31 20:21:04 +0000 | [diff] [blame] | 299 | // If we haven't computed the reverse mapping yet, do so first. |
Zachary Turner | 2f825df | 2014-06-16 20:54:28 +0000 | [diff] [blame] | 300 | if (EEState.getGlobalAddressReverseMap().empty()) { |
Jeffrey Yasskin | d0fc8f8 | 2009-10-23 22:37:43 +0000 | [diff] [blame] | 301 | for (ExecutionEngineState::GlobalAddressMapTy::iterator |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 302 | I = EEState.getGlobalAddressMap().begin(), |
| 303 | E = EEState.getGlobalAddressMap().end(); I != E; ++I) { |
| 304 | StringRef Name = I->first(); |
| 305 | uint64_t Addr = I->second; |
Zachary Turner | 2f825df | 2014-06-16 20:54:28 +0000 | [diff] [blame] | 306 | EEState.getGlobalAddressReverseMap().insert(std::make_pair( |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 307 | Addr, Name)); |
| 308 | } |
Chris Lattner | 748e857 | 2003-12-31 20:21:04 +0000 | [diff] [blame] | 309 | } |
| 310 | |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 311 | std::map<uint64_t, std::string>::iterator I = |
| 312 | EEState.getGlobalAddressReverseMap().find((uint64_t) Addr); |
| 313 | |
| 314 | if (I != EEState.getGlobalAddressReverseMap().end()) { |
| 315 | StringRef Name = I->second; |
| 316 | for (unsigned i = 0, e = Modules.size(); i != e; ++i) |
| 317 | if (GlobalValue *GV = Modules[i]->getNamedValue(Name)) |
| 318 | return GV; |
| 319 | } |
| 320 | return nullptr; |
Chris Lattner | 748e857 | 2003-12-31 20:21:04 +0000 | [diff] [blame] | 321 | } |
Chris Lattner | 5a0d482 | 2003-12-26 06:50:30 +0000 | [diff] [blame] | 322 | |
Jeffrey Yasskin | bfd38ab | 2010-03-26 00:59:12 +0000 | [diff] [blame] | 323 | namespace { |
| 324 | class ArgvArray { |
Dylan Noblesmith | c4a9942 | 2014-08-25 00:58:18 +0000 | [diff] [blame] | 325 | std::unique_ptr<char[]> Array; |
| 326 | std::vector<std::unique_ptr<char[]>> Values; |
Jeffrey Yasskin | bfd38ab | 2010-03-26 00:59:12 +0000 | [diff] [blame] | 327 | public: |
Jeffrey Yasskin | bfd38ab | 2010-03-26 00:59:12 +0000 | [diff] [blame] | 328 | /// Turn a vector of strings into a nice argv style array of pointers to null |
| 329 | /// terminated strings. |
| 330 | void *reset(LLVMContext &C, ExecutionEngine *EE, |
| 331 | const std::vector<std::string> &InputArgv); |
| 332 | }; |
| 333 | } // anonymous namespace |
| 334 | void *ArgvArray::reset(LLVMContext &C, ExecutionEngine *EE, |
| 335 | const std::vector<std::string> &InputArgv) { |
Dylan Noblesmith | c4a9942 | 2014-08-25 00:58:18 +0000 | [diff] [blame] | 336 | Values.clear(); // Free the old contents. |
| 337 | Values.reserve(InputArgv.size()); |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 338 | unsigned PtrSize = EE->getDataLayout().getPointerSize(); |
Dylan Noblesmith | c4a9942 | 2014-08-25 00:58:18 +0000 | [diff] [blame] | 339 | Array = make_unique<char[]>((InputArgv.size()+1)*PtrSize); |
Chris Lattner | 5a0d482 | 2003-12-26 06:50:30 +0000 | [diff] [blame] | 340 | |
Dylan Noblesmith | c4a9942 | 2014-08-25 00:58:18 +0000 | [diff] [blame] | 341 | DEBUG(dbgs() << "JIT: ARGV = " << (void*)Array.get() << "\n"); |
Chris Lattner | 229907c | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 342 | Type *SBytePtr = Type::getInt8PtrTy(C); |
Chris Lattner | 5a0d482 | 2003-12-26 06:50:30 +0000 | [diff] [blame] | 343 | |
| 344 | for (unsigned i = 0; i != InputArgv.size(); ++i) { |
| 345 | unsigned Size = InputArgv[i].size()+1; |
Dylan Noblesmith | 4b535d1 | 2014-08-26 02:03:28 +0000 | [diff] [blame] | 346 | auto Dest = make_unique<char[]>(Size); |
| 347 | DEBUG(dbgs() << "JIT: ARGV[" << i << "] = " << (void*)Dest.get() << "\n"); |
Misha Brukman | 835702a | 2005-04-21 22:36:52 +0000 | [diff] [blame] | 348 | |
Dylan Noblesmith | 4b535d1 | 2014-08-26 02:03:28 +0000 | [diff] [blame] | 349 | std::copy(InputArgv[i].begin(), InputArgv[i].end(), Dest.get()); |
Chris Lattner | 5a0d482 | 2003-12-26 06:50:30 +0000 | [diff] [blame] | 350 | Dest[Size-1] = 0; |
Misha Brukman | 835702a | 2005-04-21 22:36:52 +0000 | [diff] [blame] | 351 | |
Jeffrey Yasskin | bfd38ab | 2010-03-26 00:59:12 +0000 | [diff] [blame] | 352 | // Endian safe: Array[i] = (PointerTy)Dest; |
Dylan Noblesmith | 4b535d1 | 2014-08-26 02:03:28 +0000 | [diff] [blame] | 353 | EE->StoreValueToMemory(PTOGV(Dest.get()), |
| 354 | (GenericValue*)(&Array[i*PtrSize]), SBytePtr); |
| 355 | Values.push_back(std::move(Dest)); |
Chris Lattner | 5a0d482 | 2003-12-26 06:50:30 +0000 | [diff] [blame] | 356 | } |
| 357 | |
| 358 | // Null terminate it |
Craig Topper | 2617dcc | 2014-04-15 06:32:26 +0000 | [diff] [blame] | 359 | EE->StoreValueToMemory(PTOGV(nullptr), |
Dylan Noblesmith | c4a9942 | 2014-08-25 00:58:18 +0000 | [diff] [blame] | 360 | (GenericValue*)(&Array[InputArgv.size()*PtrSize]), |
Chris Lattner | 5a0d482 | 2003-12-26 06:50:30 +0000 | [diff] [blame] | 361 | SBytePtr); |
Dylan Noblesmith | c4a9942 | 2014-08-25 00:58:18 +0000 | [diff] [blame] | 362 | return Array.get(); |
Chris Lattner | 5a0d482 | 2003-12-26 06:50:30 +0000 | [diff] [blame] | 363 | } |
| 364 | |
Rafael Espindola | 2a8a279 | 2014-08-19 04:04:25 +0000 | [diff] [blame] | 365 | void ExecutionEngine::runStaticConstructorsDestructors(Module &module, |
Chris Lattner | 41fa2bd | 2009-09-23 01:46:04 +0000 | [diff] [blame] | 366 | bool isDtors) { |
Chris Lattner | faae50b | 2006-03-08 18:42:46 +0000 | [diff] [blame] | 367 | const char *Name = isDtors ? "llvm.global_dtors" : "llvm.global_ctors"; |
Rafael Espindola | 2a8a279 | 2014-08-19 04:04:25 +0000 | [diff] [blame] | 368 | GlobalVariable *GV = module.getNamedGlobal(Name); |
Evan Cheng | 1a9a0b7 | 2008-09-30 15:51:21 +0000 | [diff] [blame] | 369 | |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 370 | // If this global has internal linkage, or if it has a use, then it must be |
| 371 | // an old-style (llvmgcc3) static ctor with __main linked in and in use. If |
| 372 | // this is the case, don't execute any of the global ctors, __main will do |
| 373 | // it. |
| 374 | if (!GV || GV->isDeclaration() || GV->hasLocalLinkage()) return; |
| 375 | |
Nick Lewycky | 0cbfcb2 | 2011-04-06 20:38:44 +0000 | [diff] [blame] | 376 | // Should be an array of '{ i32, void ()* }' structs. The first value is |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 377 | // the init priority, which we ignore. |
Chris Lattner | 00245f4 | 2012-01-24 13:41:11 +0000 | [diff] [blame] | 378 | ConstantArray *InitList = dyn_cast<ConstantArray>(GV->getInitializer()); |
Craig Topper | 2617dcc | 2014-04-15 06:32:26 +0000 | [diff] [blame] | 379 | if (!InitList) |
Nick Lewycky | 0f85789 | 2011-04-11 22:11:20 +0000 | [diff] [blame] | 380 | return; |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 381 | for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i) { |
Chris Lattner | 00245f4 | 2012-01-24 13:41:11 +0000 | [diff] [blame] | 382 | ConstantStruct *CS = dyn_cast<ConstantStruct>(InitList->getOperand(i)); |
Craig Topper | 2617dcc | 2014-04-15 06:32:26 +0000 | [diff] [blame] | 383 | if (!CS) continue; |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 384 | |
| 385 | Constant *FP = CS->getOperand(1); |
| 386 | if (FP->isNullValue()) |
Nick Lewycky | 0f85789 | 2011-04-11 22:11:20 +0000 | [diff] [blame] | 387 | continue; // Found a sentinal value, ignore. |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 388 | |
| 389 | // Strip off constant expression casts. |
| 390 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(FP)) |
| 391 | if (CE->isCast()) |
| 392 | FP = CE->getOperand(0); |
| 393 | |
| 394 | // Execute the ctor/dtor function! |
| 395 | if (Function *F = dyn_cast<Function>(FP)) |
Benjamin Kramer | bd7b1c8 | 2015-06-13 19:50:29 +0000 | [diff] [blame] | 396 | runFunction(F, None); |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 397 | |
| 398 | // FIXME: It is marginally lame that we just do nothing here if we see an |
| 399 | // entry we don't recognize. It might not be unreasonable for the verifier |
| 400 | // to not even allow this and just assert here. |
| 401 | } |
Evan Cheng | 1a9a0b7 | 2008-09-30 15:51:21 +0000 | [diff] [blame] | 402 | } |
| 403 | |
Evan Cheng | 1a9a0b7 | 2008-09-30 15:51:21 +0000 | [diff] [blame] | 404 | void ExecutionEngine::runStaticConstructorsDestructors(bool isDtors) { |
| 405 | // Execute global ctors/dtors for each module in the program. |
Rafael Espindola | 2a8a279 | 2014-08-19 04:04:25 +0000 | [diff] [blame] | 406 | for (std::unique_ptr<Module> &M : Modules) |
| 407 | runStaticConstructorsDestructors(*M, isDtors); |
Chris Lattner | faae50b | 2006-03-08 18:42:46 +0000 | [diff] [blame] | 408 | } |
| 409 | |
Dan Gohman | cf3e301 | 2008-08-26 01:38:29 +0000 | [diff] [blame] | 410 | #ifndef NDEBUG |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 411 | /// isTargetNullPtr - Return whether the target pointer stored at Loc is null. |
| 412 | static bool isTargetNullPtr(ExecutionEngine *EE, void *Loc) { |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 413 | unsigned PtrSize = EE->getDataLayout().getPointerSize(); |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 414 | for (unsigned i = 0; i < PtrSize; ++i) |
| 415 | if (*(i + (uint8_t*)Loc)) |
| 416 | return false; |
| 417 | return true; |
| 418 | } |
Dan Gohman | cf3e301 | 2008-08-26 01:38:29 +0000 | [diff] [blame] | 419 | #endif |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 420 | |
Chris Lattner | 5a0d482 | 2003-12-26 06:50:30 +0000 | [diff] [blame] | 421 | int ExecutionEngine::runFunctionAsMain(Function *Fn, |
| 422 | const std::vector<std::string> &argv, |
| 423 | const char * const * envp) { |
| 424 | std::vector<GenericValue> GVArgs; |
| 425 | GenericValue GVArgc; |
Reid Spencer | 87aa65f | 2007-03-06 03:04:04 +0000 | [diff] [blame] | 426 | GVArgc.IntVal = APInt(32, argv.size()); |
Anton Korobeynikov | 8c32c11 | 2007-06-03 19:17:35 +0000 | [diff] [blame] | 427 | |
| 428 | // Check main() type |
Chris Lattner | b1cad0b | 2004-08-16 01:05:35 +0000 | [diff] [blame] | 429 | unsigned NumArgs = Fn->getFunctionType()->getNumParams(); |
Chris Lattner | 229907c | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 430 | FunctionType *FTy = Fn->getFunctionType(); |
| 431 | Type* PPInt8Ty = Type::getInt8PtrTy(Fn->getContext())->getPointerTo(); |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 432 | |
| 433 | // Check the argument types. |
| 434 | if (NumArgs > 3) |
| 435 | report_fatal_error("Invalid number of arguments of main() supplied"); |
| 436 | if (NumArgs >= 3 && FTy->getParamType(2) != PPInt8Ty) |
| 437 | report_fatal_error("Invalid type for third argument of main() supplied"); |
| 438 | if (NumArgs >= 2 && FTy->getParamType(1) != PPInt8Ty) |
| 439 | report_fatal_error("Invalid type for second argument of main() supplied"); |
| 440 | if (NumArgs >= 1 && !FTy->getParamType(0)->isIntegerTy(32)) |
| 441 | report_fatal_error("Invalid type for first argument of main() supplied"); |
| 442 | if (!FTy->getReturnType()->isIntegerTy() && |
| 443 | !FTy->getReturnType()->isVoidTy()) |
| 444 | report_fatal_error("Invalid return type of main() supplied"); |
| 445 | |
Jeffrey Yasskin | bfd38ab | 2010-03-26 00:59:12 +0000 | [diff] [blame] | 446 | ArgvArray CArgv; |
| 447 | ArgvArray CEnv; |
Chris Lattner | b1cad0b | 2004-08-16 01:05:35 +0000 | [diff] [blame] | 448 | if (NumArgs) { |
| 449 | GVArgs.push_back(GVArgc); // Arg #0 = argc. |
| 450 | if (NumArgs > 1) { |
Owen Anderson | 55f1c09 | 2009-08-13 21:58:54 +0000 | [diff] [blame] | 451 | // Arg #1 = argv. |
Jeffrey Yasskin | bfd38ab | 2010-03-26 00:59:12 +0000 | [diff] [blame] | 452 | GVArgs.push_back(PTOGV(CArgv.reset(Fn->getContext(), this, argv))); |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 453 | assert(!isTargetNullPtr(this, GVTOP(GVArgs[1])) && |
Chris Lattner | b1cad0b | 2004-08-16 01:05:35 +0000 | [diff] [blame] | 454 | "argv[0] was null after CreateArgv"); |
| 455 | if (NumArgs > 2) { |
| 456 | std::vector<std::string> EnvVars; |
| 457 | for (unsigned i = 0; envp[i]; ++i) |
Benjamin Kramer | f5e2fc4 | 2015-05-29 19:43:39 +0000 | [diff] [blame] | 458 | EnvVars.emplace_back(envp[i]); |
Owen Anderson | 55f1c09 | 2009-08-13 21:58:54 +0000 | [diff] [blame] | 459 | // Arg #2 = envp. |
Jeffrey Yasskin | bfd38ab | 2010-03-26 00:59:12 +0000 | [diff] [blame] | 460 | GVArgs.push_back(PTOGV(CEnv.reset(Fn->getContext(), this, EnvVars))); |
Chris Lattner | b1cad0b | 2004-08-16 01:05:35 +0000 | [diff] [blame] | 461 | } |
| 462 | } |
| 463 | } |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 464 | |
Reid Spencer | 87aa65f | 2007-03-06 03:04:04 +0000 | [diff] [blame] | 465 | return runFunction(Fn, GVArgs).IntVal.getZExtValue(); |
Chris Lattner | 5a0d482 | 2003-12-26 06:50:30 +0000 | [diff] [blame] | 466 | } |
| 467 | |
Benjamin Kramer | 298a3a0 | 2015-03-06 16:21:15 +0000 | [diff] [blame] | 468 | EngineBuilder::EngineBuilder() : EngineBuilder(nullptr) {} |
Lang Hames | 93de2a1 | 2015-01-23 21:25:00 +0000 | [diff] [blame] | 469 | |
Lang Hames | 4a5697e | 2014-12-03 00:51:19 +0000 | [diff] [blame] | 470 | EngineBuilder::EngineBuilder(std::unique_ptr<Module> M) |
Benjamin Kramer | 298a3a0 | 2015-03-06 16:21:15 +0000 | [diff] [blame] | 471 | : M(std::move(M)), WhichEngine(EngineKind::Either), ErrorStr(nullptr), |
Lang Hames | 633fe14 | 2015-03-30 03:37:06 +0000 | [diff] [blame] | 472 | OptLevel(CodeGenOpt::Default), MemMgr(nullptr), Resolver(nullptr), |
| 473 | RelocModel(Reloc::Default), CMModel(CodeModel::JITDefault), |
| 474 | UseOrcMCJITReplacement(false) { |
Alp Toker | 322db9e | 2014-05-31 21:26:17 +0000 | [diff] [blame] | 475 | // IR module verification is enabled by default in debug builds, and disabled |
| 476 | // by default in release builds. |
| 477 | #ifndef NDEBUG |
| 478 | VerifyModules = true; |
| 479 | #else |
| 480 | VerifyModules = false; |
| 481 | #endif |
| 482 | } |
| 483 | |
Benjamin Kramer | 298a3a0 | 2015-03-06 16:21:15 +0000 | [diff] [blame] | 484 | EngineBuilder::~EngineBuilder() = default; |
| 485 | |
| 486 | EngineBuilder &EngineBuilder::setMCJITMemoryManager( |
| 487 | std::unique_ptr<RTDyldMemoryManager> mcjmm) { |
Lang Hames | 633fe14 | 2015-03-30 03:37:06 +0000 | [diff] [blame] | 488 | auto SharedMM = std::shared_ptr<RTDyldMemoryManager>(std::move(mcjmm)); |
| 489 | MemMgr = SharedMM; |
| 490 | Resolver = SharedMM; |
| 491 | return *this; |
| 492 | } |
| 493 | |
| 494 | EngineBuilder& |
| 495 | EngineBuilder::setMemoryManager(std::unique_ptr<MCJITMemoryManager> MM) { |
| 496 | MemMgr = std::shared_ptr<MCJITMemoryManager>(std::move(MM)); |
| 497 | return *this; |
| 498 | } |
| 499 | |
| 500 | EngineBuilder& |
| 501 | EngineBuilder::setSymbolResolver(std::unique_ptr<RuntimeDyld::SymbolResolver> SR) { |
| 502 | Resolver = std::shared_ptr<RuntimeDyld::SymbolResolver>(std::move(SR)); |
Benjamin Kramer | 298a3a0 | 2015-03-06 16:21:15 +0000 | [diff] [blame] | 503 | return *this; |
| 504 | } |
| 505 | |
Owen Anderson | add6f1d | 2012-03-23 17:40:56 +0000 | [diff] [blame] | 506 | ExecutionEngine *EngineBuilder::create(TargetMachine *TM) { |
Ahmed Charles | 56440fd | 2014-03-06 05:51:42 +0000 | [diff] [blame] | 507 | std::unique_ptr<TargetMachine> TheTM(TM); // Take ownership. |
Benjamin Kramer | 25a3d81 | 2012-04-08 14:53:14 +0000 | [diff] [blame] | 508 | |
Nick Lewycky | a53414f | 2008-03-08 02:49:45 +0000 | [diff] [blame] | 509 | // Make sure we can resolve symbols in the program as well. The zero arg |
| 510 | // to the function tells DynamicLibrary to load the program, not a library. |
Craig Topper | 2617dcc | 2014-04-15 06:32:26 +0000 | [diff] [blame] | 511 | if (sys::DynamicLibrary::LoadLibraryPermanently(nullptr, ErrorStr)) |
| 512 | return nullptr; |
Filip Pizlo | 9bc53e8 | 2013-05-14 19:29:00 +0000 | [diff] [blame] | 513 | |
Reid Kleckner | fc8a2d5 | 2009-07-18 00:42:18 +0000 | [diff] [blame] | 514 | // If the user specified a memory manager but didn't specify which engine to |
| 515 | // create, we assume they only want the JIT, and we fail if they only want |
| 516 | // the interpreter. |
Lang Hames | 633fe14 | 2015-03-30 03:37:06 +0000 | [diff] [blame] | 517 | if (MemMgr) { |
Chris Lattner | 41fa2bd | 2009-09-23 01:46:04 +0000 | [diff] [blame] | 518 | if (WhichEngine & EngineKind::JIT) |
Reid Kleckner | fc8a2d5 | 2009-07-18 00:42:18 +0000 | [diff] [blame] | 519 | WhichEngine = EngineKind::JIT; |
Chris Lattner | 41fa2bd | 2009-09-23 01:46:04 +0000 | [diff] [blame] | 520 | else { |
Chris Lattner | 8bcc644 | 2009-09-23 02:03:49 +0000 | [diff] [blame] | 521 | if (ErrorStr) |
| 522 | *ErrorStr = "Cannot create an interpreter with a memory manager."; |
Craig Topper | 2617dcc | 2014-04-15 06:32:26 +0000 | [diff] [blame] | 523 | return nullptr; |
Reid Kleckner | fc8a2d5 | 2009-07-18 00:42:18 +0000 | [diff] [blame] | 524 | } |
| 525 | } |
Brian Gaeke | 4bd3bd5 | 2003-09-03 20:34:19 +0000 | [diff] [blame] | 526 | |
Reid Kleckner | fc8a2d5 | 2009-07-18 00:42:18 +0000 | [diff] [blame] | 527 | // Unless the interpreter was explicitly selected or the JIT is not linked, |
| 528 | // try making a JIT. |
Benjamin Kramer | 25a3d81 | 2012-04-08 14:53:14 +0000 | [diff] [blame] | 529 | if ((WhichEngine & EngineKind::JIT) && TheTM) { |
Dylan Noblesmith | 7f26246 | 2011-12-12 04:20:36 +0000 | [diff] [blame] | 530 | Triple TT(M->getTargetTriple()); |
Owen Anderson | add6f1d | 2012-03-23 17:40:56 +0000 | [diff] [blame] | 531 | if (!TM->getTarget().hasJIT()) { |
| 532 | errs() << "WARNING: This target JIT is not designed for the host" |
| 533 | << " you are running. If bad things happen, please choose" |
| 534 | << " a different -march switch.\n"; |
| 535 | } |
Dylan Noblesmith | 7f26246 | 2011-12-12 04:20:36 +0000 | [diff] [blame] | 536 | |
Lang Hames | bc87601 | 2014-04-18 06:48:23 +0000 | [diff] [blame] | 537 | ExecutionEngine *EE = nullptr; |
Lang Hames | 93de2a1 | 2015-01-23 21:25:00 +0000 | [diff] [blame] | 538 | if (ExecutionEngine::OrcMCJITReplacementCtor && UseOrcMCJITReplacement) { |
Lang Hames | 633fe14 | 2015-03-30 03:37:06 +0000 | [diff] [blame] | 539 | EE = ExecutionEngine::OrcMCJITReplacementCtor(ErrorStr, std::move(MemMgr), |
| 540 | std::move(Resolver), |
Lang Hames | 93de2a1 | 2015-01-23 21:25:00 +0000 | [diff] [blame] | 541 | std::move(TheTM)); |
| 542 | EE->addModule(std::move(M)); |
| 543 | } else if (ExecutionEngine::MCJITCtor) |
Lang Hames | 633fe14 | 2015-03-30 03:37:06 +0000 | [diff] [blame] | 544 | EE = ExecutionEngine::MCJITCtor(std::move(M), ErrorStr, std::move(MemMgr), |
| 545 | std::move(Resolver), std::move(TheTM)); |
Lang Hames | 93de2a1 | 2015-01-23 21:25:00 +0000 | [diff] [blame] | 546 | |
Lang Hames | bc87601 | 2014-04-18 06:48:23 +0000 | [diff] [blame] | 547 | if (EE) { |
| 548 | EE->setVerifyModules(VerifyModules); |
| 549 | return EE; |
Chris Lattner | 41fa2bd | 2009-09-23 01:46:04 +0000 | [diff] [blame] | 550 | } |
Reid Kleckner | fc8a2d5 | 2009-07-18 00:42:18 +0000 | [diff] [blame] | 551 | } |
| 552 | |
| 553 | // If we can't make a JIT and we didn't request one specifically, try making |
| 554 | // an interpreter instead. |
Chris Lattner | 41fa2bd | 2009-09-23 01:46:04 +0000 | [diff] [blame] | 555 | if (WhichEngine & EngineKind::Interpreter) { |
| 556 | if (ExecutionEngine::InterpCtor) |
Rafael Espindola | 2a8a279 | 2014-08-19 04:04:25 +0000 | [diff] [blame] | 557 | return ExecutionEngine::InterpCtor(std::move(M), ErrorStr); |
Chris Lattner | 8bcc644 | 2009-09-23 02:03:49 +0000 | [diff] [blame] | 558 | if (ErrorStr) |
| 559 | *ErrorStr = "Interpreter has not been linked in."; |
Craig Topper | 2617dcc | 2014-04-15 06:32:26 +0000 | [diff] [blame] | 560 | return nullptr; |
Reid Kleckner | fc8a2d5 | 2009-07-18 00:42:18 +0000 | [diff] [blame] | 561 | } |
Chris Lattner | 8bcc644 | 2009-09-23 02:03:49 +0000 | [diff] [blame] | 562 | |
Eric Christopher | 79cc1e3 | 2014-09-02 22:28:02 +0000 | [diff] [blame] | 563 | if ((WhichEngine & EngineKind::JIT) && !ExecutionEngine::MCJITCtor) { |
Chris Lattner | 8bcc644 | 2009-09-23 02:03:49 +0000 | [diff] [blame] | 564 | if (ErrorStr) |
| 565 | *ErrorStr = "JIT has not been linked in."; |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 566 | } |
| 567 | |
Craig Topper | 2617dcc | 2014-04-15 06:32:26 +0000 | [diff] [blame] | 568 | return nullptr; |
Brian Gaeke | 4bd3bd5 | 2003-09-03 20:34:19 +0000 | [diff] [blame] | 569 | } |
| 570 | |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 571 | void *ExecutionEngine::getPointerToGlobal(const GlobalValue *GV) { |
Brian Gaeke | 1678e85 | 2003-08-13 18:16:14 +0000 | [diff] [blame] | 572 | if (Function *F = const_cast<Function*>(dyn_cast<Function>(GV))) |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 573 | return getPointerToFunction(F); |
| 574 | |
Zachary Turner | c04b892 | 2014-06-20 21:07:14 +0000 | [diff] [blame] | 575 | MutexGuard locked(lock); |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 576 | if (void* P = getPointerToGlobalIfAvailable(GV)) |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 577 | return P; |
Jeff Cohen | 69e84901 | 2006-02-07 05:11:57 +0000 | [diff] [blame] | 578 | |
| 579 | // Global variable might have been added since interpreter started. |
| 580 | if (GlobalVariable *GVar = |
| 581 | const_cast<GlobalVariable *>(dyn_cast<GlobalVariable>(GV))) |
| 582 | EmitGlobalVariable(GVar); |
| 583 | else |
Torok Edwin | fbcc663 | 2009-07-14 16:55:14 +0000 | [diff] [blame] | 584 | llvm_unreachable("Global hasn't had an address allocated yet!"); |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 585 | |
Lang Hames | 3dac3f7 | 2015-03-31 20:31:14 +0000 | [diff] [blame] | 586 | return getPointerToGlobalIfAvailable(GV); |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 587 | } |
| 588 | |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 589 | /// \brief Converts a Constant* into a GenericValue, including handling of |
| 590 | /// ConstantExpr values. |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 591 | GenericValue ExecutionEngine::getConstantValue(const Constant *C) { |
Reid Spencer | 6c38f0b | 2006-11-27 01:05:10 +0000 | [diff] [blame] | 592 | // If its undefined, return the garbage. |
Jay Foad | bcbdbfb | 2010-01-15 08:32:58 +0000 | [diff] [blame] | 593 | if (isa<UndefValue>(C)) { |
| 594 | GenericValue Result; |
| 595 | switch (C->getType()->getTypeID()) { |
Nadav Rotem | be79a7a | 2013-04-01 15:53:30 +0000 | [diff] [blame] | 596 | default: |
| 597 | break; |
Jay Foad | bcbdbfb | 2010-01-15 08:32:58 +0000 | [diff] [blame] | 598 | case Type::IntegerTyID: |
| 599 | case Type::X86_FP80TyID: |
| 600 | case Type::FP128TyID: |
| 601 | case Type::PPC_FP128TyID: |
| 602 | // Although the value is undefined, we still have to construct an APInt |
| 603 | // with the correct bit width. |
| 604 | Result.IntVal = APInt(C->getType()->getPrimitiveSizeInBits(), 0); |
| 605 | break; |
Elena Demikhovsky | 8e97f01 | 2013-09-12 10:48:23 +0000 | [diff] [blame] | 606 | case Type::StructTyID: { |
| 607 | // if the whole struct is 'undef' just reserve memory for the value. |
| 608 | if(StructType *STy = dyn_cast<StructType>(C->getType())) { |
| 609 | unsigned int elemNum = STy->getNumElements(); |
| 610 | Result.AggregateVal.resize(elemNum); |
| 611 | for (unsigned int i = 0; i < elemNum; ++i) { |
| 612 | Type *ElemTy = STy->getElementType(i); |
| 613 | if (ElemTy->isIntegerTy()) |
| 614 | Result.AggregateVal[i].IntVal = |
| 615 | APInt(ElemTy->getPrimitiveSizeInBits(), 0); |
| 616 | else if (ElemTy->isAggregateType()) { |
| 617 | const Constant *ElemUndef = UndefValue::get(ElemTy); |
| 618 | Result.AggregateVal[i] = getConstantValue(ElemUndef); |
| 619 | } |
| 620 | } |
| 621 | } |
| 622 | } |
| 623 | break; |
Nadav Rotem | be79a7a | 2013-04-01 15:53:30 +0000 | [diff] [blame] | 624 | case Type::VectorTyID: |
| 625 | // if the whole vector is 'undef' just reserve memory for the value. |
| 626 | const VectorType* VTy = dyn_cast<VectorType>(C->getType()); |
| 627 | const Type *ElemTy = VTy->getElementType(); |
| 628 | unsigned int elemNum = VTy->getNumElements(); |
| 629 | Result.AggregateVal.resize(elemNum); |
| 630 | if (ElemTy->isIntegerTy()) |
| 631 | for (unsigned int i = 0; i < elemNum; ++i) |
Elena Demikhovsky | 8e97f01 | 2013-09-12 10:48:23 +0000 | [diff] [blame] | 632 | Result.AggregateVal[i].IntVal = |
Nadav Rotem | be79a7a | 2013-04-01 15:53:30 +0000 | [diff] [blame] | 633 | APInt(ElemTy->getPrimitiveSizeInBits(), 0); |
Jay Foad | bcbdbfb | 2010-01-15 08:32:58 +0000 | [diff] [blame] | 634 | break; |
| 635 | } |
| 636 | return Result; |
| 637 | } |
Chris Lattner | 9de0d14 | 2003-04-23 19:01:49 +0000 | [diff] [blame] | 638 | |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 639 | // Otherwise, if the value is a ConstantExpr... |
Reid Spencer | 6c38f0b | 2006-11-27 01:05:10 +0000 | [diff] [blame] | 640 | if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) { |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 641 | Constant *Op0 = CE->getOperand(0); |
Chris Lattner | 9de0d14 | 2003-04-23 19:01:49 +0000 | [diff] [blame] | 642 | switch (CE->getOpcode()) { |
| 643 | case Instruction::GetElementPtr: { |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 644 | // Compute the index |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 645 | GenericValue Result = getConstantValue(Op0); |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 646 | APInt Offset(DL.getPointerSizeInBits(), 0); |
| 647 | cast<GEPOperator>(CE)->accumulateConstantOffset(DL, Offset); |
Misha Brukman | 835702a | 2005-04-21 22:36:52 +0000 | [diff] [blame] | 648 | |
Reid Spencer | 87aa65f | 2007-03-06 03:04:04 +0000 | [diff] [blame] | 649 | char* tmp = (char*) Result.PointerVal; |
Nuno Lopes | b6ad982 | 2012-12-30 16:25:48 +0000 | [diff] [blame] | 650 | Result = PTOGV(tmp + Offset.getSExtValue()); |
Chris Lattner | 9de0d14 | 2003-04-23 19:01:49 +0000 | [diff] [blame] | 651 | return Result; |
| 652 | } |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 653 | case Instruction::Trunc: { |
| 654 | GenericValue GV = getConstantValue(Op0); |
| 655 | uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); |
| 656 | GV.IntVal = GV.IntVal.trunc(BitWidth); |
| 657 | return GV; |
| 658 | } |
| 659 | case Instruction::ZExt: { |
| 660 | GenericValue GV = getConstantValue(Op0); |
| 661 | uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); |
| 662 | GV.IntVal = GV.IntVal.zext(BitWidth); |
| 663 | return GV; |
| 664 | } |
| 665 | case Instruction::SExt: { |
| 666 | GenericValue GV = getConstantValue(Op0); |
| 667 | uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); |
| 668 | GV.IntVal = GV.IntVal.sext(BitWidth); |
| 669 | return GV; |
| 670 | } |
| 671 | case Instruction::FPTrunc: { |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 672 | // FIXME long double |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 673 | GenericValue GV = getConstantValue(Op0); |
| 674 | GV.FloatVal = float(GV.DoubleVal); |
| 675 | return GV; |
| 676 | } |
| 677 | case Instruction::FPExt:{ |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 678 | // FIXME long double |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 679 | GenericValue GV = getConstantValue(Op0); |
| 680 | GV.DoubleVal = double(GV.FloatVal); |
| 681 | return GV; |
| 682 | } |
| 683 | case Instruction::UIToFP: { |
| 684 | GenericValue GV = getConstantValue(Op0); |
Chris Lattner | fdd8790 | 2009-10-05 05:54:46 +0000 | [diff] [blame] | 685 | if (CE->getType()->isFloatTy()) |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 686 | GV.FloatVal = float(GV.IntVal.roundToDouble()); |
Chris Lattner | fdd8790 | 2009-10-05 05:54:46 +0000 | [diff] [blame] | 687 | else if (CE->getType()->isDoubleTy()) |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 688 | GV.DoubleVal = GV.IntVal.roundToDouble(); |
Chris Lattner | fdd8790 | 2009-10-05 05:54:46 +0000 | [diff] [blame] | 689 | else if (CE->getType()->isX86_FP80Ty()) { |
Benjamin Kramer | 31920b0 | 2010-12-04 15:28:22 +0000 | [diff] [blame] | 690 | APFloat apf = APFloat::getZero(APFloat::x87DoubleExtended); |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 691 | (void)apf.convertFromAPInt(GV.IntVal, |
Dan Gohman | ca24fd9 | 2008-02-29 01:27:13 +0000 | [diff] [blame] | 692 | false, |
| 693 | APFloat::rmNearestTiesToEven); |
Dale Johannesen | 54306fe | 2008-10-09 18:53:47 +0000 | [diff] [blame] | 694 | GV.IntVal = apf.bitcastToAPInt(); |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 695 | } |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 696 | return GV; |
| 697 | } |
| 698 | case Instruction::SIToFP: { |
| 699 | GenericValue GV = getConstantValue(Op0); |
Chris Lattner | fdd8790 | 2009-10-05 05:54:46 +0000 | [diff] [blame] | 700 | if (CE->getType()->isFloatTy()) |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 701 | GV.FloatVal = float(GV.IntVal.signedRoundToDouble()); |
Chris Lattner | fdd8790 | 2009-10-05 05:54:46 +0000 | [diff] [blame] | 702 | else if (CE->getType()->isDoubleTy()) |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 703 | GV.DoubleVal = GV.IntVal.signedRoundToDouble(); |
Chris Lattner | fdd8790 | 2009-10-05 05:54:46 +0000 | [diff] [blame] | 704 | else if (CE->getType()->isX86_FP80Ty()) { |
Benjamin Kramer | 31920b0 | 2010-12-04 15:28:22 +0000 | [diff] [blame] | 705 | APFloat apf = APFloat::getZero(APFloat::x87DoubleExtended); |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 706 | (void)apf.convertFromAPInt(GV.IntVal, |
Dan Gohman | ca24fd9 | 2008-02-29 01:27:13 +0000 | [diff] [blame] | 707 | true, |
| 708 | APFloat::rmNearestTiesToEven); |
Dale Johannesen | 54306fe | 2008-10-09 18:53:47 +0000 | [diff] [blame] | 709 | GV.IntVal = apf.bitcastToAPInt(); |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 710 | } |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 711 | return GV; |
| 712 | } |
| 713 | case Instruction::FPToUI: // double->APInt conversion handles sign |
| 714 | case Instruction::FPToSI: { |
| 715 | GenericValue GV = getConstantValue(Op0); |
| 716 | uint32_t BitWidth = cast<IntegerType>(CE->getType())->getBitWidth(); |
Chris Lattner | fdd8790 | 2009-10-05 05:54:46 +0000 | [diff] [blame] | 717 | if (Op0->getType()->isFloatTy()) |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 718 | GV.IntVal = APIntOps::RoundFloatToAPInt(GV.FloatVal, BitWidth); |
Chris Lattner | fdd8790 | 2009-10-05 05:54:46 +0000 | [diff] [blame] | 719 | else if (Op0->getType()->isDoubleTy()) |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 720 | GV.IntVal = APIntOps::RoundDoubleToAPInt(GV.DoubleVal, BitWidth); |
Chris Lattner | fdd8790 | 2009-10-05 05:54:46 +0000 | [diff] [blame] | 721 | else if (Op0->getType()->isX86_FP80Ty()) { |
Tim Northover | 29178a3 | 2013-01-22 09:46:31 +0000 | [diff] [blame] | 722 | APFloat apf = APFloat(APFloat::x87DoubleExtended, GV.IntVal); |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 723 | uint64_t v; |
Dale Johannesen | 4f0bd68 | 2008-10-09 23:00:39 +0000 | [diff] [blame] | 724 | bool ignored; |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 725 | (void)apf.convertToInteger(&v, BitWidth, |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 726 | CE->getOpcode()==Instruction::FPToSI, |
Dale Johannesen | 4f0bd68 | 2008-10-09 23:00:39 +0000 | [diff] [blame] | 727 | APFloat::rmTowardZero, &ignored); |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 728 | GV.IntVal = v; // endian? |
| 729 | } |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 730 | return GV; |
| 731 | } |
Reid Spencer | 6c38f0b | 2006-11-27 01:05:10 +0000 | [diff] [blame] | 732 | case Instruction::PtrToInt: { |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 733 | GenericValue GV = getConstantValue(Op0); |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 734 | uint32_t PtrWidth = DL.getTypeSizeInBits(Op0->getType()); |
Eli Friedman | fc1f2cd | 2012-10-30 22:21:55 +0000 | [diff] [blame] | 735 | assert(PtrWidth <= 64 && "Bad pointer width"); |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 736 | GV.IntVal = APInt(PtrWidth, uintptr_t(GV.PointerVal)); |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 737 | uint32_t IntWidth = DL.getTypeSizeInBits(CE->getType()); |
Eli Friedman | fc1f2cd | 2012-10-30 22:21:55 +0000 | [diff] [blame] | 738 | GV.IntVal = GV.IntVal.zextOrTrunc(IntWidth); |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 739 | return GV; |
| 740 | } |
| 741 | case Instruction::IntToPtr: { |
| 742 | GenericValue GV = getConstantValue(Op0); |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 743 | uint32_t PtrWidth = DL.getTypeSizeInBits(CE->getType()); |
Eli Friedman | fc1f2cd | 2012-10-30 22:21:55 +0000 | [diff] [blame] | 744 | GV.IntVal = GV.IntVal.zextOrTrunc(PtrWidth); |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 745 | assert(GV.IntVal.getBitWidth() <= 64 && "Bad pointer width"); |
| 746 | GV.PointerVal = PointerTy(uintptr_t(GV.IntVal.getZExtValue())); |
Reid Spencer | 6c38f0b | 2006-11-27 01:05:10 +0000 | [diff] [blame] | 747 | return GV; |
| 748 | } |
| 749 | case Instruction::BitCast: { |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 750 | GenericValue GV = getConstantValue(Op0); |
Chris Lattner | 229907c | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 751 | Type* DestTy = CE->getType(); |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 752 | switch (Op0->getType()->getTypeID()) { |
Torok Edwin | fbcc663 | 2009-07-14 16:55:14 +0000 | [diff] [blame] | 753 | default: llvm_unreachable("Invalid bitcast operand"); |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 754 | case Type::IntegerTyID: |
Duncan Sands | 9dff9be | 2010-02-15 16:12:20 +0000 | [diff] [blame] | 755 | assert(DestTy->isFloatingPointTy() && "invalid bitcast"); |
Chris Lattner | fdd8790 | 2009-10-05 05:54:46 +0000 | [diff] [blame] | 756 | if (DestTy->isFloatTy()) |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 757 | GV.FloatVal = GV.IntVal.bitsToFloat(); |
Chris Lattner | fdd8790 | 2009-10-05 05:54:46 +0000 | [diff] [blame] | 758 | else if (DestTy->isDoubleTy()) |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 759 | GV.DoubleVal = GV.IntVal.bitsToDouble(); |
| 760 | break; |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 761 | case Type::FloatTyID: |
Duncan Sands | 9dff9be | 2010-02-15 16:12:20 +0000 | [diff] [blame] | 762 | assert(DestTy->isIntegerTy(32) && "Invalid bitcast"); |
Jay Foad | 3447fb0 | 2010-11-28 21:04:48 +0000 | [diff] [blame] | 763 | GV.IntVal = APInt::floatToBits(GV.FloatVal); |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 764 | break; |
| 765 | case Type::DoubleTyID: |
Duncan Sands | 9dff9be | 2010-02-15 16:12:20 +0000 | [diff] [blame] | 766 | assert(DestTy->isIntegerTy(64) && "Invalid bitcast"); |
Jay Foad | 3447fb0 | 2010-11-28 21:04:48 +0000 | [diff] [blame] | 767 | GV.IntVal = APInt::doubleToBits(GV.DoubleVal); |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 768 | break; |
| 769 | case Type::PointerTyID: |
Duncan Sands | 19d0b47 | 2010-02-16 11:11:14 +0000 | [diff] [blame] | 770 | assert(DestTy->isPointerTy() && "Invalid bitcast"); |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 771 | break; // getConstantValue(Op0) above already converted it |
| 772 | } |
| 773 | return GV; |
Chris Lattner | 9de0d14 | 2003-04-23 19:01:49 +0000 | [diff] [blame] | 774 | } |
Chris Lattner | 68cbcc3 | 2003-05-14 17:51:49 +0000 | [diff] [blame] | 775 | case Instruction::Add: |
Dan Gohman | a5b9645 | 2009-06-04 22:49:04 +0000 | [diff] [blame] | 776 | case Instruction::FAdd: |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 777 | case Instruction::Sub: |
Dan Gohman | a5b9645 | 2009-06-04 22:49:04 +0000 | [diff] [blame] | 778 | case Instruction::FSub: |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 779 | case Instruction::Mul: |
Dan Gohman | a5b9645 | 2009-06-04 22:49:04 +0000 | [diff] [blame] | 780 | case Instruction::FMul: |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 781 | case Instruction::UDiv: |
| 782 | case Instruction::SDiv: |
| 783 | case Instruction::URem: |
| 784 | case Instruction::SRem: |
| 785 | case Instruction::And: |
| 786 | case Instruction::Or: |
| 787 | case Instruction::Xor: { |
| 788 | GenericValue LHS = getConstantValue(Op0); |
| 789 | GenericValue RHS = getConstantValue(CE->getOperand(1)); |
| 790 | GenericValue GV; |
Chris Lattner | c4e6bb5 | 2004-07-11 08:01:11 +0000 | [diff] [blame] | 791 | switch (CE->getOperand(0)->getType()->getTypeID()) { |
Torok Edwin | fbcc663 | 2009-07-14 16:55:14 +0000 | [diff] [blame] | 792 | default: llvm_unreachable("Bad add type!"); |
Reid Spencer | 7a9c62b | 2007-01-12 07:05:14 +0000 | [diff] [blame] | 793 | case Type::IntegerTyID: |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 794 | switch (CE->getOpcode()) { |
Torok Edwin | fbcc663 | 2009-07-14 16:55:14 +0000 | [diff] [blame] | 795 | default: llvm_unreachable("Invalid integer opcode"); |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 796 | case Instruction::Add: GV.IntVal = LHS.IntVal + RHS.IntVal; break; |
| 797 | case Instruction::Sub: GV.IntVal = LHS.IntVal - RHS.IntVal; break; |
| 798 | case Instruction::Mul: GV.IntVal = LHS.IntVal * RHS.IntVal; break; |
| 799 | case Instruction::UDiv:GV.IntVal = LHS.IntVal.udiv(RHS.IntVal); break; |
| 800 | case Instruction::SDiv:GV.IntVal = LHS.IntVal.sdiv(RHS.IntVal); break; |
| 801 | case Instruction::URem:GV.IntVal = LHS.IntVal.urem(RHS.IntVal); break; |
| 802 | case Instruction::SRem:GV.IntVal = LHS.IntVal.srem(RHS.IntVal); break; |
| 803 | case Instruction::And: GV.IntVal = LHS.IntVal & RHS.IntVal; break; |
| 804 | case Instruction::Or: GV.IntVal = LHS.IntVal | RHS.IntVal; break; |
| 805 | case Instruction::Xor: GV.IntVal = LHS.IntVal ^ RHS.IntVal; break; |
| 806 | } |
Chris Lattner | c4e6bb5 | 2004-07-11 08:01:11 +0000 | [diff] [blame] | 807 | break; |
| 808 | case Type::FloatTyID: |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 809 | switch (CE->getOpcode()) { |
Torok Edwin | fbcc663 | 2009-07-14 16:55:14 +0000 | [diff] [blame] | 810 | default: llvm_unreachable("Invalid float opcode"); |
Dan Gohman | a5b9645 | 2009-06-04 22:49:04 +0000 | [diff] [blame] | 811 | case Instruction::FAdd: |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 812 | GV.FloatVal = LHS.FloatVal + RHS.FloatVal; break; |
Dan Gohman | a5b9645 | 2009-06-04 22:49:04 +0000 | [diff] [blame] | 813 | case Instruction::FSub: |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 814 | GV.FloatVal = LHS.FloatVal - RHS.FloatVal; break; |
Dan Gohman | a5b9645 | 2009-06-04 22:49:04 +0000 | [diff] [blame] | 815 | case Instruction::FMul: |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 816 | GV.FloatVal = LHS.FloatVal * RHS.FloatVal; break; |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 817 | case Instruction::FDiv: |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 818 | GV.FloatVal = LHS.FloatVal / RHS.FloatVal; break; |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 819 | case Instruction::FRem: |
Chris Lattner | 93cd0f1c | 2010-05-15 17:10:24 +0000 | [diff] [blame] | 820 | GV.FloatVal = std::fmod(LHS.FloatVal,RHS.FloatVal); break; |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 821 | } |
Chris Lattner | c4e6bb5 | 2004-07-11 08:01:11 +0000 | [diff] [blame] | 822 | break; |
| 823 | case Type::DoubleTyID: |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 824 | switch (CE->getOpcode()) { |
Torok Edwin | fbcc663 | 2009-07-14 16:55:14 +0000 | [diff] [blame] | 825 | default: llvm_unreachable("Invalid double opcode"); |
Dan Gohman | a5b9645 | 2009-06-04 22:49:04 +0000 | [diff] [blame] | 826 | case Instruction::FAdd: |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 827 | GV.DoubleVal = LHS.DoubleVal + RHS.DoubleVal; break; |
Dan Gohman | a5b9645 | 2009-06-04 22:49:04 +0000 | [diff] [blame] | 828 | case Instruction::FSub: |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 829 | GV.DoubleVal = LHS.DoubleVal - RHS.DoubleVal; break; |
Dan Gohman | a5b9645 | 2009-06-04 22:49:04 +0000 | [diff] [blame] | 830 | case Instruction::FMul: |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 831 | GV.DoubleVal = LHS.DoubleVal * RHS.DoubleVal; break; |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 832 | case Instruction::FDiv: |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 833 | GV.DoubleVal = LHS.DoubleVal / RHS.DoubleVal; break; |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 834 | case Instruction::FRem: |
Chris Lattner | 93cd0f1c | 2010-05-15 17:10:24 +0000 | [diff] [blame] | 835 | GV.DoubleVal = std::fmod(LHS.DoubleVal,RHS.DoubleVal); break; |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 836 | } |
Chris Lattner | c4e6bb5 | 2004-07-11 08:01:11 +0000 | [diff] [blame] | 837 | break; |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 838 | case Type::X86_FP80TyID: |
| 839 | case Type::PPC_FP128TyID: |
| 840 | case Type::FP128TyID: { |
Tim Northover | 29178a3 | 2013-01-22 09:46:31 +0000 | [diff] [blame] | 841 | const fltSemantics &Sem = CE->getOperand(0)->getType()->getFltSemantics(); |
| 842 | APFloat apfLHS = APFloat(Sem, LHS.IntVal); |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 843 | switch (CE->getOpcode()) { |
Daniel Dunbar | e4f4743 | 2010-11-13 00:55:42 +0000 | [diff] [blame] | 844 | default: llvm_unreachable("Invalid long double opcode"); |
Dan Gohman | a5b9645 | 2009-06-04 22:49:04 +0000 | [diff] [blame] | 845 | case Instruction::FAdd: |
Tim Northover | 29178a3 | 2013-01-22 09:46:31 +0000 | [diff] [blame] | 846 | apfLHS.add(APFloat(Sem, RHS.IntVal), APFloat::rmNearestTiesToEven); |
Dale Johannesen | 54306fe | 2008-10-09 18:53:47 +0000 | [diff] [blame] | 847 | GV.IntVal = apfLHS.bitcastToAPInt(); |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 848 | break; |
Dan Gohman | a5b9645 | 2009-06-04 22:49:04 +0000 | [diff] [blame] | 849 | case Instruction::FSub: |
Tim Northover | 29178a3 | 2013-01-22 09:46:31 +0000 | [diff] [blame] | 850 | apfLHS.subtract(APFloat(Sem, RHS.IntVal), |
| 851 | APFloat::rmNearestTiesToEven); |
Dale Johannesen | 54306fe | 2008-10-09 18:53:47 +0000 | [diff] [blame] | 852 | GV.IntVal = apfLHS.bitcastToAPInt(); |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 853 | break; |
Dan Gohman | a5b9645 | 2009-06-04 22:49:04 +0000 | [diff] [blame] | 854 | case Instruction::FMul: |
Tim Northover | 29178a3 | 2013-01-22 09:46:31 +0000 | [diff] [blame] | 855 | apfLHS.multiply(APFloat(Sem, RHS.IntVal), |
| 856 | APFloat::rmNearestTiesToEven); |
Dale Johannesen | 54306fe | 2008-10-09 18:53:47 +0000 | [diff] [blame] | 857 | GV.IntVal = apfLHS.bitcastToAPInt(); |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 858 | break; |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 859 | case Instruction::FDiv: |
Tim Northover | 29178a3 | 2013-01-22 09:46:31 +0000 | [diff] [blame] | 860 | apfLHS.divide(APFloat(Sem, RHS.IntVal), |
| 861 | APFloat::rmNearestTiesToEven); |
Dale Johannesen | 54306fe | 2008-10-09 18:53:47 +0000 | [diff] [blame] | 862 | GV.IntVal = apfLHS.bitcastToAPInt(); |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 863 | break; |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 864 | case Instruction::FRem: |
Tim Northover | 29178a3 | 2013-01-22 09:46:31 +0000 | [diff] [blame] | 865 | apfLHS.mod(APFloat(Sem, RHS.IntVal), |
| 866 | APFloat::rmNearestTiesToEven); |
Dale Johannesen | 54306fe | 2008-10-09 18:53:47 +0000 | [diff] [blame] | 867 | GV.IntVal = apfLHS.bitcastToAPInt(); |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 868 | break; |
| 869 | } |
| 870 | } |
| 871 | break; |
Chris Lattner | c4e6bb5 | 2004-07-11 08:01:11 +0000 | [diff] [blame] | 872 | } |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 873 | return GV; |
| 874 | } |
Chris Lattner | 68cbcc3 | 2003-05-14 17:51:49 +0000 | [diff] [blame] | 875 | default: |
| 876 | break; |
| 877 | } |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 878 | |
| 879 | SmallString<256> Msg; |
| 880 | raw_svector_ostream OS(Msg); |
| 881 | OS << "ConstantExpr not handled: " << *CE; |
| 882 | report_fatal_error(OS.str()); |
Chris Lattner | 68cbcc3 | 2003-05-14 17:51:49 +0000 | [diff] [blame] | 883 | } |
Misha Brukman | 835702a | 2005-04-21 22:36:52 +0000 | [diff] [blame] | 884 | |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 885 | // Otherwise, we have a simple constant. |
Reid Spencer | 4fd528f | 2007-03-06 22:23:15 +0000 | [diff] [blame] | 886 | GenericValue Result; |
Chris Lattner | 6b72759 | 2004-06-17 18:19:28 +0000 | [diff] [blame] | 887 | switch (C->getType()->getTypeID()) { |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 888 | case Type::FloatTyID: |
| 889 | Result.FloatVal = cast<ConstantFP>(C)->getValueAPF().convertToFloat(); |
Reid Spencer | 7a9c62b | 2007-01-12 07:05:14 +0000 | [diff] [blame] | 890 | break; |
Reid Spencer | 87aa65f | 2007-03-06 03:04:04 +0000 | [diff] [blame] | 891 | case Type::DoubleTyID: |
Dale Johannesen | bed9dc4 | 2007-09-06 18:13:44 +0000 | [diff] [blame] | 892 | Result.DoubleVal = cast<ConstantFP>(C)->getValueAPF().convertToDouble(); |
Reid Spencer | 87aa65f | 2007-03-06 03:04:04 +0000 | [diff] [blame] | 893 | break; |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 894 | case Type::X86_FP80TyID: |
| 895 | case Type::FP128TyID: |
| 896 | case Type::PPC_FP128TyID: |
Dale Johannesen | 54306fe | 2008-10-09 18:53:47 +0000 | [diff] [blame] | 897 | Result.IntVal = cast <ConstantFP>(C)->getValueAPF().bitcastToAPInt(); |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 898 | break; |
Reid Spencer | 87aa65f | 2007-03-06 03:04:04 +0000 | [diff] [blame] | 899 | case Type::IntegerTyID: |
| 900 | Result.IntVal = cast<ConstantInt>(C)->getValue(); |
| 901 | break; |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 902 | case Type::PointerTyID: |
Reid Spencer | 6a0fd73 | 2004-07-18 00:41:27 +0000 | [diff] [blame] | 903 | if (isa<ConstantPointerNull>(C)) |
Craig Topper | 2617dcc | 2014-04-15 06:32:26 +0000 | [diff] [blame] | 904 | Result.PointerVal = nullptr; |
Reid Spencer | 6a0fd73 | 2004-07-18 00:41:27 +0000 | [diff] [blame] | 905 | else if (const Function *F = dyn_cast<Function>(C)) |
| 906 | Result = PTOGV(getPointerToFunctionOrStub(const_cast<Function*>(F))); |
Chris Lattner | 0c778f7 | 2009-10-29 05:26:09 +0000 | [diff] [blame] | 907 | else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(C)) |
Reid Spencer | 6a0fd73 | 2004-07-18 00:41:27 +0000 | [diff] [blame] | 908 | Result = PTOGV(getOrEmitGlobalVariable(const_cast<GlobalVariable*>(GV))); |
| 909 | else |
Torok Edwin | fbcc663 | 2009-07-14 16:55:14 +0000 | [diff] [blame] | 910 | llvm_unreachable("Unknown constant pointer type!"); |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 911 | break; |
Nadav Rotem | be79a7a | 2013-04-01 15:53:30 +0000 | [diff] [blame] | 912 | case Type::VectorTyID: { |
| 913 | unsigned elemNum; |
| 914 | Type* ElemTy; |
| 915 | const ConstantDataVector *CDV = dyn_cast<ConstantDataVector>(C); |
| 916 | const ConstantVector *CV = dyn_cast<ConstantVector>(C); |
| 917 | const ConstantAggregateZero *CAZ = dyn_cast<ConstantAggregateZero>(C); |
| 918 | |
| 919 | if (CDV) { |
| 920 | elemNum = CDV->getNumElements(); |
| 921 | ElemTy = CDV->getElementType(); |
| 922 | } else if (CV || CAZ) { |
| 923 | VectorType* VTy = dyn_cast<VectorType>(C->getType()); |
| 924 | elemNum = VTy->getNumElements(); |
| 925 | ElemTy = VTy->getElementType(); |
| 926 | } else { |
| 927 | llvm_unreachable("Unknown constant vector type!"); |
| 928 | } |
| 929 | |
| 930 | Result.AggregateVal.resize(elemNum); |
| 931 | // Check if vector holds floats. |
| 932 | if(ElemTy->isFloatTy()) { |
| 933 | if (CAZ) { |
| 934 | GenericValue floatZero; |
| 935 | floatZero.FloatVal = 0.f; |
| 936 | std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(), |
| 937 | floatZero); |
| 938 | break; |
| 939 | } |
| 940 | if(CV) { |
| 941 | for (unsigned i = 0; i < elemNum; ++i) |
| 942 | if (!isa<UndefValue>(CV->getOperand(i))) |
| 943 | Result.AggregateVal[i].FloatVal = cast<ConstantFP>( |
| 944 | CV->getOperand(i))->getValueAPF().convertToFloat(); |
| 945 | break; |
| 946 | } |
| 947 | if(CDV) |
| 948 | for (unsigned i = 0; i < elemNum; ++i) |
| 949 | Result.AggregateVal[i].FloatVal = CDV->getElementAsFloat(i); |
| 950 | |
| 951 | break; |
| 952 | } |
| 953 | // Check if vector holds doubles. |
| 954 | if (ElemTy->isDoubleTy()) { |
| 955 | if (CAZ) { |
| 956 | GenericValue doubleZero; |
| 957 | doubleZero.DoubleVal = 0.0; |
| 958 | std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(), |
| 959 | doubleZero); |
| 960 | break; |
| 961 | } |
| 962 | if(CV) { |
| 963 | for (unsigned i = 0; i < elemNum; ++i) |
| 964 | if (!isa<UndefValue>(CV->getOperand(i))) |
| 965 | Result.AggregateVal[i].DoubleVal = cast<ConstantFP>( |
| 966 | CV->getOperand(i))->getValueAPF().convertToDouble(); |
| 967 | break; |
| 968 | } |
| 969 | if(CDV) |
| 970 | for (unsigned i = 0; i < elemNum; ++i) |
| 971 | Result.AggregateVal[i].DoubleVal = CDV->getElementAsDouble(i); |
| 972 | |
| 973 | break; |
| 974 | } |
| 975 | // Check if vector holds integers. |
| 976 | if (ElemTy->isIntegerTy()) { |
| 977 | if (CAZ) { |
| 978 | GenericValue intZero; |
| 979 | intZero.IntVal = APInt(ElemTy->getScalarSizeInBits(), 0ull); |
| 980 | std::fill(Result.AggregateVal.begin(), Result.AggregateVal.end(), |
| 981 | intZero); |
| 982 | break; |
| 983 | } |
| 984 | if(CV) { |
| 985 | for (unsigned i = 0; i < elemNum; ++i) |
| 986 | if (!isa<UndefValue>(CV->getOperand(i))) |
| 987 | Result.AggregateVal[i].IntVal = cast<ConstantInt>( |
| 988 | CV->getOperand(i))->getValue(); |
| 989 | else { |
| 990 | Result.AggregateVal[i].IntVal = |
| 991 | APInt(CV->getOperand(i)->getType()->getPrimitiveSizeInBits(), 0); |
| 992 | } |
| 993 | break; |
| 994 | } |
| 995 | if(CDV) |
| 996 | for (unsigned i = 0; i < elemNum; ++i) |
| 997 | Result.AggregateVal[i].IntVal = APInt( |
| 998 | CDV->getElementType()->getPrimitiveSizeInBits(), |
| 999 | CDV->getElementAsInteger(i)); |
| 1000 | |
| 1001 | break; |
| 1002 | } |
| 1003 | llvm_unreachable("Unknown constant pointer type!"); |
| 1004 | } |
| 1005 | break; |
| 1006 | |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 1007 | default: |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 1008 | SmallString<256> Msg; |
| 1009 | raw_svector_ostream OS(Msg); |
| 1010 | OS << "ERROR: Constant unimplemented for type: " << *C->getType(); |
| 1011 | report_fatal_error(OS.str()); |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 1012 | } |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 1013 | |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 1014 | return Result; |
| 1015 | } |
| 1016 | |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 1017 | /// StoreIntToMemory - Fills the StoreBytes bytes of memory starting from Dst |
| 1018 | /// with the integer held in IntVal. |
| 1019 | static void StoreIntToMemory(const APInt &IntVal, uint8_t *Dst, |
| 1020 | unsigned StoreBytes) { |
| 1021 | assert((IntVal.getBitWidth()+7)/8 >= StoreBytes && "Integer too small!"); |
Roman Divacky | ad06cee | 2012-09-05 22:26:57 +0000 | [diff] [blame] | 1022 | const uint8_t *Src = (const uint8_t *)IntVal.getRawData(); |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 1023 | |
Rafael Espindola | 41cb64f | 2013-04-15 14:44:24 +0000 | [diff] [blame] | 1024 | if (sys::IsLittleEndianHost) { |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 1025 | // Little-endian host - the source is ordered from LSB to MSB. Order the |
| 1026 | // destination from LSB to MSB: Do a straight copy. |
| 1027 | memcpy(Dst, Src, StoreBytes); |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 1028 | } else { |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 1029 | // Big-endian host - the source is an array of 64 bit words ordered from |
| 1030 | // LSW to MSW. Each word is ordered from MSB to LSB. Order the destination |
| 1031 | // from MSB to LSB: Reverse the word order, but not the bytes in a word. |
| 1032 | while (StoreBytes > sizeof(uint64_t)) { |
| 1033 | StoreBytes -= sizeof(uint64_t); |
| 1034 | // May not be aligned so use memcpy. |
| 1035 | memcpy(Dst + StoreBytes, Src, sizeof(uint64_t)); |
| 1036 | Src += sizeof(uint64_t); |
| 1037 | } |
| 1038 | |
| 1039 | memcpy(Dst, Src + sizeof(uint64_t) - StoreBytes, StoreBytes); |
| 1040 | } |
| 1041 | } |
| 1042 | |
Evan Cheng | 09053e6 | 2008-11-04 06:10:31 +0000 | [diff] [blame] | 1043 | void ExecutionEngine::StoreValueToMemory(const GenericValue &Val, |
Chris Lattner | 229907c | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 1044 | GenericValue *Ptr, Type *Ty) { |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 1045 | const unsigned StoreBytes = getDataLayout().getTypeStoreSize(Ty); |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 1046 | |
Reid Spencer | 87aa65f | 2007-03-06 03:04:04 +0000 | [diff] [blame] | 1047 | switch (Ty->getTypeID()) { |
Nadav Rotem | be79a7a | 2013-04-01 15:53:30 +0000 | [diff] [blame] | 1048 | default: |
| 1049 | dbgs() << "Cannot store value of type " << *Ty << "!\n"; |
| 1050 | break; |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 1051 | case Type::IntegerTyID: |
| 1052 | StoreIntToMemory(Val.IntVal, (uint8_t*)Ptr, StoreBytes); |
Reid Spencer | 87aa65f | 2007-03-06 03:04:04 +0000 | [diff] [blame] | 1053 | break; |
Reid Spencer | 87aa65f | 2007-03-06 03:04:04 +0000 | [diff] [blame] | 1054 | case Type::FloatTyID: |
| 1055 | *((float*)Ptr) = Val.FloatVal; |
| 1056 | break; |
| 1057 | case Type::DoubleTyID: |
| 1058 | *((double*)Ptr) = Val.DoubleVal; |
| 1059 | break; |
Dale Johannesen | 4d7e4ee | 2009-03-24 18:16:17 +0000 | [diff] [blame] | 1060 | case Type::X86_FP80TyID: |
| 1061 | memcpy(Ptr, Val.IntVal.getRawData(), 10); |
| 1062 | break; |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 1063 | case Type::PointerTyID: |
| 1064 | // Ensure 64 bit target pointers are fully initialized on 32 bit hosts. |
| 1065 | if (StoreBytes != sizeof(PointerTy)) |
Chandler Carruth | 93da3c8 | 2011-04-28 08:37:18 +0000 | [diff] [blame] | 1066 | memset(&(Ptr->PointerVal), 0, StoreBytes); |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 1067 | |
Reid Spencer | 87aa65f | 2007-03-06 03:04:04 +0000 | [diff] [blame] | 1068 | *((PointerTy*)Ptr) = Val.PointerVal; |
| 1069 | break; |
Nadav Rotem | be79a7a | 2013-04-01 15:53:30 +0000 | [diff] [blame] | 1070 | case Type::VectorTyID: |
| 1071 | for (unsigned i = 0; i < Val.AggregateVal.size(); ++i) { |
| 1072 | if (cast<VectorType>(Ty)->getElementType()->isDoubleTy()) |
| 1073 | *(((double*)Ptr)+i) = Val.AggregateVal[i].DoubleVal; |
| 1074 | if (cast<VectorType>(Ty)->getElementType()->isFloatTy()) |
| 1075 | *(((float*)Ptr)+i) = Val.AggregateVal[i].FloatVal; |
| 1076 | if (cast<VectorType>(Ty)->getElementType()->isIntegerTy()) { |
| 1077 | unsigned numOfBytes =(Val.AggregateVal[i].IntVal.getBitWidth()+7)/8; |
| 1078 | StoreIntToMemory(Val.AggregateVal[i].IntVal, |
| 1079 | (uint8_t*)Ptr + numOfBytes*i, numOfBytes); |
| 1080 | } |
| 1081 | } |
| 1082 | break; |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 1083 | } |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 1084 | |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 1085 | if (sys::IsLittleEndianHost != getDataLayout().isLittleEndian()) |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 1086 | // Host and target are different endian - reverse the stored bytes. |
| 1087 | std::reverse((uint8_t*)Ptr, StoreBytes + (uint8_t*)Ptr); |
| 1088 | } |
| 1089 | |
| 1090 | /// LoadIntFromMemory - Loads the integer stored in the LoadBytes bytes starting |
| 1091 | /// from Src into IntVal, which is assumed to be wide enough and to hold zero. |
| 1092 | static void LoadIntFromMemory(APInt &IntVal, uint8_t *Src, unsigned LoadBytes) { |
| 1093 | assert((IntVal.getBitWidth()+7)/8 >= LoadBytes && "Integer too small!"); |
David Greene | 82b6357 | 2013-01-14 21:04:45 +0000 | [diff] [blame] | 1094 | uint8_t *Dst = reinterpret_cast<uint8_t *>( |
| 1095 | const_cast<uint64_t *>(IntVal.getRawData())); |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 1096 | |
Rafael Espindola | 41cb64f | 2013-04-15 14:44:24 +0000 | [diff] [blame] | 1097 | if (sys::IsLittleEndianHost) |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 1098 | // Little-endian host - the destination must be ordered from LSB to MSB. |
| 1099 | // The source is ordered from LSB to MSB: Do a straight copy. |
| 1100 | memcpy(Dst, Src, LoadBytes); |
| 1101 | else { |
| 1102 | // Big-endian - the destination is an array of 64 bit words ordered from |
| 1103 | // LSW to MSW. Each word must be ordered from MSB to LSB. The source is |
| 1104 | // ordered from MSB to LSB: Reverse the word order, but not the bytes in |
| 1105 | // a word. |
| 1106 | while (LoadBytes > sizeof(uint64_t)) { |
| 1107 | LoadBytes -= sizeof(uint64_t); |
| 1108 | // May not be aligned so use memcpy. |
| 1109 | memcpy(Dst, Src + LoadBytes, sizeof(uint64_t)); |
| 1110 | Dst += sizeof(uint64_t); |
| 1111 | } |
| 1112 | |
| 1113 | memcpy(Dst + sizeof(uint64_t) - LoadBytes, Src, LoadBytes); |
| 1114 | } |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 1115 | } |
| 1116 | |
Misha Brukman | 857c21b | 2003-10-10 17:45:12 +0000 | [diff] [blame] | 1117 | /// FIXME: document |
| 1118 | /// |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 1119 | void ExecutionEngine::LoadValueFromMemory(GenericValue &Result, |
Duncan Sands | cd4a6be | 2008-03-10 16:38:37 +0000 | [diff] [blame] | 1120 | GenericValue *Ptr, |
Chris Lattner | 229907c | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 1121 | Type *Ty) { |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 1122 | const unsigned LoadBytes = getDataLayout().getTypeStoreSize(Ty); |
Duncan Sands | 5c65cb4 | 2007-12-10 17:43:13 +0000 | [diff] [blame] | 1123 | |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 1124 | switch (Ty->getTypeID()) { |
| 1125 | case Type::IntegerTyID: |
| 1126 | // An APInt with all words initially zero. |
| 1127 | Result.IntVal = APInt(cast<IntegerType>(Ty)->getBitWidth(), 0); |
| 1128 | LoadIntFromMemory(Result.IntVal, (uint8_t*)Ptr, LoadBytes); |
| 1129 | break; |
Reid Spencer | 87aa65f | 2007-03-06 03:04:04 +0000 | [diff] [blame] | 1130 | case Type::FloatTyID: |
| 1131 | Result.FloatVal = *((float*)Ptr); |
| 1132 | break; |
| 1133 | case Type::DoubleTyID: |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 1134 | Result.DoubleVal = *((double*)Ptr); |
Reid Spencer | 87aa65f | 2007-03-06 03:04:04 +0000 | [diff] [blame] | 1135 | break; |
Duncan Sands | 1202d1b | 2007-12-14 19:38:31 +0000 | [diff] [blame] | 1136 | case Type::PointerTyID: |
Reid Spencer | 87aa65f | 2007-03-06 03:04:04 +0000 | [diff] [blame] | 1137 | Result.PointerVal = *((PointerTy*)Ptr); |
| 1138 | break; |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 1139 | case Type::X86_FP80TyID: { |
| 1140 | // This is endian dependent, but it will only work on x86 anyway. |
Duncan Sands | 26d6539 | 2007-12-15 17:37:40 +0000 | [diff] [blame] | 1141 | // FIXME: Will not trap if loading a signaling NaN. |
Dale Johannesen | 4d7e4ee | 2009-03-24 18:16:17 +0000 | [diff] [blame] | 1142 | uint64_t y[2]; |
| 1143 | memcpy(y, Ptr, 10); |
Jeffrey Yasskin | 7a16288 | 2011-07-18 21:45:40 +0000 | [diff] [blame] | 1144 | Result.IntVal = APInt(80, y); |
Dale Johannesen | a1336cf | 2007-09-17 18:44:13 +0000 | [diff] [blame] | 1145 | break; |
| 1146 | } |
Nadav Rotem | be79a7a | 2013-04-01 15:53:30 +0000 | [diff] [blame] | 1147 | case Type::VectorTyID: { |
| 1148 | const VectorType *VT = cast<VectorType>(Ty); |
| 1149 | const Type *ElemT = VT->getElementType(); |
| 1150 | const unsigned numElems = VT->getNumElements(); |
| 1151 | if (ElemT->isFloatTy()) { |
| 1152 | Result.AggregateVal.resize(numElems); |
| 1153 | for (unsigned i = 0; i < numElems; ++i) |
| 1154 | Result.AggregateVal[i].FloatVal = *((float*)Ptr+i); |
| 1155 | } |
| 1156 | if (ElemT->isDoubleTy()) { |
| 1157 | Result.AggregateVal.resize(numElems); |
| 1158 | for (unsigned i = 0; i < numElems; ++i) |
| 1159 | Result.AggregateVal[i].DoubleVal = *((double*)Ptr+i); |
| 1160 | } |
| 1161 | if (ElemT->isIntegerTy()) { |
| 1162 | GenericValue intZero; |
| 1163 | const unsigned elemBitWidth = cast<IntegerType>(ElemT)->getBitWidth(); |
| 1164 | intZero.IntVal = APInt(elemBitWidth, 0); |
| 1165 | Result.AggregateVal.resize(numElems, intZero); |
| 1166 | for (unsigned i = 0; i < numElems; ++i) |
| 1167 | LoadIntFromMemory(Result.AggregateVal[i].IntVal, |
| 1168 | (uint8_t*)Ptr+((elemBitWidth+7)/8)*i, (elemBitWidth+7)/8); |
| 1169 | } |
| 1170 | break; |
| 1171 | } |
Reid Spencer | 87aa65f | 2007-03-06 03:04:04 +0000 | [diff] [blame] | 1172 | default: |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 1173 | SmallString<256> Msg; |
| 1174 | raw_svector_ostream OS(Msg); |
| 1175 | OS << "Cannot load value of type " << *Ty << "!"; |
| 1176 | report_fatal_error(OS.str()); |
Chris Lattner | 7f389e8 | 2003-05-08 16:52:16 +0000 | [diff] [blame] | 1177 | } |
Chris Lattner | 7f389e8 | 2003-05-08 16:52:16 +0000 | [diff] [blame] | 1178 | } |
| 1179 | |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 1180 | void ExecutionEngine::InitializeMemory(const Constant *Init, void *Addr) { |
David Greene | 0967d2d | 2010-01-05 01:27:39 +0000 | [diff] [blame] | 1181 | DEBUG(dbgs() << "JIT: Initializing " << Addr << " "); |
Dale Johannesen | b086d38 | 2008-08-07 01:30:15 +0000 | [diff] [blame] | 1182 | DEBUG(Init->dump()); |
Chris Lattner | 00245f4 | 2012-01-24 13:41:11 +0000 | [diff] [blame] | 1183 | if (isa<UndefValue>(Init)) |
Chris Lattner | 61753bf | 2004-10-16 18:19:26 +0000 | [diff] [blame] | 1184 | return; |
Chris Lattner | 00245f4 | 2012-01-24 13:41:11 +0000 | [diff] [blame] | 1185 | |
| 1186 | if (const ConstantVector *CP = dyn_cast<ConstantVector>(Init)) { |
Robert Bocchino | 69d6213 | 2006-01-20 18:18:40 +0000 | [diff] [blame] | 1187 | unsigned ElementSize = |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 1188 | getDataLayout().getTypeAllocSize(CP->getType()->getElementType()); |
Robert Bocchino | 69d6213 | 2006-01-20 18:18:40 +0000 | [diff] [blame] | 1189 | for (unsigned i = 0, e = CP->getNumOperands(); i != e; ++i) |
| 1190 | InitializeMemory(CP->getOperand(i), (char*)Addr+i*ElementSize); |
| 1191 | return; |
Chris Lattner | 00245f4 | 2012-01-24 13:41:11 +0000 | [diff] [blame] | 1192 | } |
| 1193 | |
| 1194 | if (isa<ConstantAggregateZero>(Init)) { |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 1195 | memset(Addr, 0, (size_t)getDataLayout().getTypeAllocSize(Init->getType())); |
Chris Lattner | 1dd86b1 | 2008-02-15 00:57:28 +0000 | [diff] [blame] | 1196 | return; |
Chris Lattner | 00245f4 | 2012-01-24 13:41:11 +0000 | [diff] [blame] | 1197 | } |
| 1198 | |
| 1199 | if (const ConstantArray *CPA = dyn_cast<ConstantArray>(Init)) { |
Dan Gohman | 69ddfbf | 2008-05-20 03:20:09 +0000 | [diff] [blame] | 1200 | unsigned ElementSize = |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 1201 | getDataLayout().getTypeAllocSize(CPA->getType()->getElementType()); |
Dan Gohman | 69ddfbf | 2008-05-20 03:20:09 +0000 | [diff] [blame] | 1202 | for (unsigned i = 0, e = CPA->getNumOperands(); i != e; ++i) |
| 1203 | InitializeMemory(CPA->getOperand(i), (char*)Addr+i*ElementSize); |
| 1204 | return; |
Chris Lattner | 00245f4 | 2012-01-24 13:41:11 +0000 | [diff] [blame] | 1205 | } |
| 1206 | |
| 1207 | if (const ConstantStruct *CPS = dyn_cast<ConstantStruct>(Init)) { |
Dan Gohman | 69ddfbf | 2008-05-20 03:20:09 +0000 | [diff] [blame] | 1208 | const StructLayout *SL = |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 1209 | getDataLayout().getStructLayout(cast<StructType>(CPS->getType())); |
Dan Gohman | 69ddfbf | 2008-05-20 03:20:09 +0000 | [diff] [blame] | 1210 | for (unsigned i = 0, e = CPS->getNumOperands(); i != e; ++i) |
| 1211 | InitializeMemory(CPS->getOperand(i), (char*)Addr+SL->getElementOffset(i)); |
| 1212 | return; |
Chris Lattner | 00245f4 | 2012-01-24 13:41:11 +0000 | [diff] [blame] | 1213 | } |
| 1214 | |
| 1215 | if (const ConstantDataSequential *CDS = |
| 1216 | dyn_cast<ConstantDataSequential>(Init)) { |
| 1217 | // CDS is already laid out in host memory order. |
| 1218 | StringRef Data = CDS->getRawDataValues(); |
| 1219 | memcpy(Addr, Data.data(), Data.size()); |
| 1220 | return; |
| 1221 | } |
| 1222 | |
| 1223 | if (Init->getType()->isFirstClassType()) { |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 1224 | GenericValue Val = getConstantValue(Init); |
| 1225 | StoreValueToMemory(Val, (GenericValue*)Addr, Init->getType()); |
| 1226 | return; |
| 1227 | } |
| 1228 | |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 1229 | DEBUG(dbgs() << "Bad Type: " << *Init->getType() << "\n"); |
Torok Edwin | fbcc663 | 2009-07-14 16:55:14 +0000 | [diff] [blame] | 1230 | llvm_unreachable("Unknown constant type to initialize memory with!"); |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 1231 | } |
| 1232 | |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 1233 | /// EmitGlobals - Emit all of the global variables to memory, storing their |
| 1234 | /// addresses into GlobalAddress. This must make sure to copy the contents of |
| 1235 | /// their initializers into the memory. |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 1236 | void ExecutionEngine::emitGlobals() { |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 1237 | // Loop over all of the global variables in the program, allocating the memory |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1238 | // to hold them. If there is more than one module, do a prepass over globals |
| 1239 | // to figure out how the different modules should link together. |
Chris Lattner | 229907c | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 1240 | std::map<std::pair<std::string, Type*>, |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1241 | const GlobalValue*> LinkedGlobalsMap; |
Misha Brukman | 835702a | 2005-04-21 22:36:52 +0000 | [diff] [blame] | 1242 | |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1243 | if (Modules.size() != 1) { |
| 1244 | for (unsigned m = 0, e = Modules.size(); m != e; ++m) { |
Jeffrey Yasskin | 091217b | 2010-01-27 20:34:15 +0000 | [diff] [blame] | 1245 | Module &M = *Modules[m]; |
Rafael Espindola | 49bb65a | 2014-05-08 18:17:44 +0000 | [diff] [blame] | 1246 | for (const auto &GV : M.globals()) { |
| 1247 | if (GV.hasLocalLinkage() || GV.isDeclaration() || |
| 1248 | GV.hasAppendingLinkage() || !GV.hasName()) |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1249 | continue;// Ignore external globals and globals with internal linkage. |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 1250 | |
| 1251 | const GlobalValue *&GVEntry = |
Rafael Espindola | 49bb65a | 2014-05-08 18:17:44 +0000 | [diff] [blame] | 1252 | LinkedGlobalsMap[std::make_pair(GV.getName(), GV.getType())]; |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1253 | |
| 1254 | // If this is the first time we've seen this global, it is the canonical |
| 1255 | // version. |
| 1256 | if (!GVEntry) { |
Rafael Espindola | 49bb65a | 2014-05-08 18:17:44 +0000 | [diff] [blame] | 1257 | GVEntry = &GV; |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1258 | continue; |
| 1259 | } |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 1260 | |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1261 | // If the existing global is strong, never replace it. |
Nico Rieck | 7157bb7 | 2014-01-14 15:22:47 +0000 | [diff] [blame] | 1262 | if (GVEntry->hasExternalLinkage()) |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1263 | continue; |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 1264 | |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1265 | // Otherwise, we know it's linkonce/weak, replace it if this is a strong |
Dale Johannesen | ce4396b | 2008-05-14 20:12:51 +0000 | [diff] [blame] | 1266 | // symbol. FIXME is this right for common? |
Rafael Espindola | 49bb65a | 2014-05-08 18:17:44 +0000 | [diff] [blame] | 1267 | if (GV.hasExternalLinkage() || GVEntry->hasExternalWeakLinkage()) |
| 1268 | GVEntry = &GV; |
Chris Lattner | 9de0d14 | 2003-04-23 19:01:49 +0000 | [diff] [blame] | 1269 | } |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 1270 | } |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1271 | } |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 1272 | |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1273 | std::vector<const GlobalValue*> NonCanonicalGlobals; |
| 1274 | for (unsigned m = 0, e = Modules.size(); m != e; ++m) { |
Jeffrey Yasskin | 091217b | 2010-01-27 20:34:15 +0000 | [diff] [blame] | 1275 | Module &M = *Modules[m]; |
Rafael Espindola | 49bb65a | 2014-05-08 18:17:44 +0000 | [diff] [blame] | 1276 | for (const auto &GV : M.globals()) { |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1277 | // In the multi-module case, see what this global maps to. |
| 1278 | if (!LinkedGlobalsMap.empty()) { |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 1279 | if (const GlobalValue *GVEntry = |
Rafael Espindola | 49bb65a | 2014-05-08 18:17:44 +0000 | [diff] [blame] | 1280 | LinkedGlobalsMap[std::make_pair(GV.getName(), GV.getType())]) { |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1281 | // If something else is the canonical global, ignore this one. |
Rafael Espindola | 49bb65a | 2014-05-08 18:17:44 +0000 | [diff] [blame] | 1282 | if (GVEntry != &GV) { |
| 1283 | NonCanonicalGlobals.push_back(&GV); |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1284 | continue; |
| 1285 | } |
| 1286 | } |
| 1287 | } |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 1288 | |
Rafael Espindola | 49bb65a | 2014-05-08 18:17:44 +0000 | [diff] [blame] | 1289 | if (!GV.isDeclaration()) { |
| 1290 | addGlobalMapping(&GV, getMemoryForGV(&GV)); |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1291 | } else { |
| 1292 | // External variable reference. Try to use the dynamic loader to |
| 1293 | // get a pointer to it. |
| 1294 | if (void *SymAddr = |
Rafael Espindola | 49bb65a | 2014-05-08 18:17:44 +0000 | [diff] [blame] | 1295 | sys::DynamicLibrary::SearchForAddressOfSymbol(GV.getName())) |
| 1296 | addGlobalMapping(&GV, SymAddr); |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1297 | else { |
Chris Lattner | 2104b8d | 2010-04-07 22:58:41 +0000 | [diff] [blame] | 1298 | report_fatal_error("Could not resolve external global address: " |
Rafael Espindola | 49bb65a | 2014-05-08 18:17:44 +0000 | [diff] [blame] | 1299 | +GV.getName()); |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1300 | } |
| 1301 | } |
| 1302 | } |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 1303 | |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1304 | // If there are multiple modules, map the non-canonical globals to their |
| 1305 | // canonical location. |
| 1306 | if (!NonCanonicalGlobals.empty()) { |
| 1307 | for (unsigned i = 0, e = NonCanonicalGlobals.size(); i != e; ++i) { |
| 1308 | const GlobalValue *GV = NonCanonicalGlobals[i]; |
| 1309 | const GlobalValue *CGV = |
| 1310 | LinkedGlobalsMap[std::make_pair(GV->getName(), GV->getType())]; |
| 1311 | void *Ptr = getPointerToGlobalIfAvailable(CGV); |
| 1312 | assert(Ptr && "Canonical global wasn't codegen'd!"); |
Nuno Lopes | a67f06b | 2008-10-14 10:04:52 +0000 | [diff] [blame] | 1313 | addGlobalMapping(GV, Ptr); |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1314 | } |
| 1315 | } |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 1316 | |
| 1317 | // Now that all of the globals are set up in memory, loop through them all |
Reid Spencer | 7a9c62b | 2007-01-12 07:05:14 +0000 | [diff] [blame] | 1318 | // and initialize their contents. |
Rafael Espindola | 49bb65a | 2014-05-08 18:17:44 +0000 | [diff] [blame] | 1319 | for (const auto &GV : M.globals()) { |
| 1320 | if (!GV.isDeclaration()) { |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1321 | if (!LinkedGlobalsMap.empty()) { |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 1322 | if (const GlobalValue *GVEntry = |
Rafael Espindola | 49bb65a | 2014-05-08 18:17:44 +0000 | [diff] [blame] | 1323 | LinkedGlobalsMap[std::make_pair(GV.getName(), GV.getType())]) |
| 1324 | if (GVEntry != &GV) // Not the canonical variable. |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1325 | continue; |
| 1326 | } |
Rafael Espindola | 49bb65a | 2014-05-08 18:17:44 +0000 | [diff] [blame] | 1327 | EmitGlobalVariable(&GV); |
Chris Lattner | 0621cae | 2006-08-16 01:24:12 +0000 | [diff] [blame] | 1328 | } |
| 1329 | } |
| 1330 | } |
Chris Lattner | 6bbe3ec | 2003-12-20 02:45:37 +0000 | [diff] [blame] | 1331 | } |
| 1332 | |
| 1333 | // EmitGlobalVariable - This method emits the specified global variable to the |
| 1334 | // address specified in GlobalAddresses, or allocates new memory if it's not |
| 1335 | // already in the map. |
Chris Lattner | fbcc0aa | 2003-12-20 03:36:47 +0000 | [diff] [blame] | 1336 | void ExecutionEngine::EmitGlobalVariable(const GlobalVariable *GV) { |
Chris Lattner | 748e857 | 2003-12-31 20:21:04 +0000 | [diff] [blame] | 1337 | void *GA = getPointerToGlobalIfAvailable(GV); |
Chris Lattner | dc63173 | 2004-02-08 19:33:23 +0000 | [diff] [blame] | 1338 | |
Craig Topper | 2617dcc | 2014-04-15 06:32:26 +0000 | [diff] [blame] | 1339 | if (!GA) { |
Chris Lattner | 6bbe3ec | 2003-12-20 02:45:37 +0000 | [diff] [blame] | 1340 | // If it's not already specified, allocate memory for the global. |
Nicolas Geoffray | 5457ce9 | 2008-10-25 15:41:43 +0000 | [diff] [blame] | 1341 | GA = getMemoryForGV(GV); |
Andrew Kaylor | 3b44237 | 2013-11-15 17:52:54 +0000 | [diff] [blame] | 1342 | |
| 1343 | // If we failed to allocate memory for this global, return. |
Craig Topper | 2617dcc | 2014-04-15 06:32:26 +0000 | [diff] [blame] | 1344 | if (!GA) return; |
Andrew Kaylor | 3b44237 | 2013-11-15 17:52:54 +0000 | [diff] [blame] | 1345 | |
Chris Lattner | 748e857 | 2003-12-31 20:21:04 +0000 | [diff] [blame] | 1346 | addGlobalMapping(GV, GA); |
Chris Lattner | 6bbe3ec | 2003-12-20 02:45:37 +0000 | [diff] [blame] | 1347 | } |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 1348 | |
Nicolas Geoffray | 5457ce9 | 2008-10-25 15:41:43 +0000 | [diff] [blame] | 1349 | // Don't initialize if it's thread local, let the client do it. |
| 1350 | if (!GV->isThreadLocal()) |
| 1351 | InitializeMemory(GV->getInitializer(), GA); |
Daniel Dunbar | 868e3f0 | 2010-11-13 02:48:57 +0000 | [diff] [blame] | 1352 | |
Chris Lattner | 229907c | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 1353 | Type *ElTy = GV->getType()->getElementType(); |
Mehdi Amini | a3fcefb | 2015-07-16 16:34:23 +0000 | [diff] [blame^] | 1354 | size_t GVSize = (size_t)getDataLayout().getTypeAllocSize(ElTy); |
Chris Lattner | df1f152 | 2005-01-08 20:13:19 +0000 | [diff] [blame] | 1355 | NumInitBytes += (unsigned)GVSize; |
Chris Lattner | 6bbe3ec | 2003-12-20 02:45:37 +0000 | [diff] [blame] | 1356 | ++NumGlobals; |
Chris Lattner | 996fe01 | 2002-12-24 00:01:05 +0000 | [diff] [blame] | 1357 | } |