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