Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 1 | //===-- llvm-rtdyld.cpp - MCJIT Testing Tool ------------------------------===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This is a testing tool for use with the MC-JIT LLVM components. |
| 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
Chandler Carruth | 4d88a1c | 2012-12-04 10:44:52 +0000 | [diff] [blame] | 14 | #include "llvm/ADT/StringMap.h" |
Zachary Turner | 6489d7b | 2015-04-23 17:37:47 +0000 | [diff] [blame] | 15 | #include "llvm/DebugInfo/DIContext.h" |
| 16 | #include "llvm/DebugInfo/DWARF/DWARFContext.h" |
Lang Hames | 633fe14 | 2015-03-30 03:37:06 +0000 | [diff] [blame] | 17 | #include "llvm/ExecutionEngine/RTDyldMemoryManager.h" |
Chandler Carruth | 4d88a1c | 2012-12-04 10:44:52 +0000 | [diff] [blame] | 18 | #include "llvm/ExecutionEngine/RuntimeDyld.h" |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 19 | #include "llvm/ExecutionEngine/RuntimeDyldChecker.h" |
| 20 | #include "llvm/MC/MCAsmInfo.h" |
| 21 | #include "llvm/MC/MCContext.h" |
Benjamin Kramer | f57c197 | 2016-01-26 16:44:37 +0000 | [diff] [blame] | 22 | #include "llvm/MC/MCDisassembler/MCDisassembler.h" |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 23 | #include "llvm/MC/MCInstPrinter.h" |
Chandler Carruth | d990388 | 2015-01-14 11:23:27 +0000 | [diff] [blame] | 24 | #include "llvm/MC/MCInstrInfo.h" |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 25 | #include "llvm/MC/MCRegisterInfo.h" |
Pete Cooper | 3de83e4 | 2015-05-15 21:58:42 +0000 | [diff] [blame] | 26 | #include "llvm/MC/MCSubtargetInfo.h" |
Rafael Espindola | 6e040c0 | 2013-04-26 20:07:33 +0000 | [diff] [blame] | 27 | #include "llvm/Object/MachO.h" |
Rafael Espindola | b109c03 | 2015-06-23 02:08:48 +0000 | [diff] [blame] | 28 | #include "llvm/Object/SymbolSize.h" |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 29 | #include "llvm/Support/CommandLine.h" |
Lang Hames | d311c0e | 2014-05-13 22:37:41 +0000 | [diff] [blame] | 30 | #include "llvm/Support/DynamicLibrary.h" |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 31 | #include "llvm/Support/ManagedStatic.h" |
| 32 | #include "llvm/Support/Memory.h" |
| 33 | #include "llvm/Support/MemoryBuffer.h" |
Alp Toker | 9f67932 | 2013-11-05 09:33:43 +0000 | [diff] [blame] | 34 | #include "llvm/Support/PrettyStackTrace.h" |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 35 | #include "llvm/Support/Signals.h" |
| 36 | #include "llvm/Support/TargetRegistry.h" |
| 37 | #include "llvm/Support/TargetSelect.h" |
Chandler Carruth | d990388 | 2015-01-14 11:23:27 +0000 | [diff] [blame] | 38 | #include "llvm/Support/raw_ostream.h" |
Lang Hames | 778ef5b | 2014-09-04 04:19:54 +0000 | [diff] [blame] | 39 | #include <list> |
Rafael Espindola | a6e9c3e | 2014-06-12 17:38:55 +0000 | [diff] [blame] | 40 | #include <system_error> |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 41 | |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 42 | using namespace llvm; |
| 43 | using namespace llvm::object; |
| 44 | |
Jim Grosbach | 7cb41d7 | 2011-04-13 15:49:40 +0000 | [diff] [blame] | 45 | static cl::list<std::string> |
| 46 | InputFileList(cl::Positional, cl::ZeroOrMore, |
| 47 | cl::desc("<input file>")); |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 48 | |
| 49 | enum ActionType { |
Andrew Kaylor | d55d701 | 2013-01-25 22:50:58 +0000 | [diff] [blame] | 50 | AC_Execute, |
Keno Fischer | 281b694 | 2015-05-30 19:44:53 +0000 | [diff] [blame] | 51 | AC_PrintObjectLineInfo, |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 52 | AC_PrintLineInfo, |
Keno Fischer | c780e8e | 2015-05-21 21:24:32 +0000 | [diff] [blame] | 53 | AC_PrintDebugLineInfo, |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 54 | AC_Verify |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 55 | }; |
| 56 | |
| 57 | static cl::opt<ActionType> |
| 58 | Action(cl::desc("Action to perform:"), |
| 59 | cl::init(AC_Execute), |
| 60 | cl::values(clEnumValN(AC_Execute, "execute", |
| 61 | "Load, link, and execute the inputs."), |
Andrew Kaylor | d55d701 | 2013-01-25 22:50:58 +0000 | [diff] [blame] | 62 | clEnumValN(AC_PrintLineInfo, "printline", |
| 63 | "Load, link, and print line information for each function."), |
Keno Fischer | c780e8e | 2015-05-21 21:24:32 +0000 | [diff] [blame] | 64 | clEnumValN(AC_PrintDebugLineInfo, "printdebugline", |
| 65 | "Load, link, and print line information for each function using the debug object"), |
Keno Fischer | 281b694 | 2015-05-30 19:44:53 +0000 | [diff] [blame] | 66 | clEnumValN(AC_PrintObjectLineInfo, "printobjline", |
| 67 | "Like -printlineinfo but does not load the object first"), |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 68 | clEnumValN(AC_Verify, "verify", |
| 69 | "Load, link and verify the resulting memory image."), |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 70 | clEnumValEnd)); |
| 71 | |
Jim Grosbach | d35159a | 2011-04-13 15:38:30 +0000 | [diff] [blame] | 72 | static cl::opt<std::string> |
| 73 | EntryPoint("entry", |
| 74 | cl::desc("Function to call as entry point."), |
| 75 | cl::init("_main")); |
| 76 | |
Lang Hames | d311c0e | 2014-05-13 22:37:41 +0000 | [diff] [blame] | 77 | static cl::list<std::string> |
| 78 | Dylibs("dylib", |
| 79 | cl::desc("Add library."), |
| 80 | cl::ZeroOrMore); |
| 81 | |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 82 | static cl::opt<std::string> |
| 83 | TripleName("triple", cl::desc("Target triple for disassembler")); |
| 84 | |
Petar Jovanovic | 280e562 | 2015-06-23 22:52:19 +0000 | [diff] [blame] | 85 | static cl::opt<std::string> |
| 86 | MCPU("mcpu", |
| 87 | cl::desc("Target a specific cpu type (-mcpu=help for details)"), |
| 88 | cl::value_desc("cpu-name"), |
| 89 | cl::init("")); |
| 90 | |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 91 | static cl::list<std::string> |
| 92 | CheckFiles("check", |
| 93 | cl::desc("File containing RuntimeDyld verifier checks."), |
| 94 | cl::ZeroOrMore); |
| 95 | |
Lang Hames | 9cb7353 | 2014-07-29 23:43:13 +0000 | [diff] [blame] | 96 | static cl::opt<uint64_t> |
Davide Italiano | b59ea90 | 2015-10-21 22:12:03 +0000 | [diff] [blame] | 97 | PreallocMemory("preallocate", |
| 98 | cl::desc("Allocate memory upfront rather than on-demand"), |
| 99 | cl::init(0)); |
| 100 | |
| 101 | static cl::opt<uint64_t> |
Lang Hames | 9cb7353 | 2014-07-29 23:43:13 +0000 | [diff] [blame] | 102 | TargetAddrStart("target-addr-start", |
| 103 | cl::desc("For -verify only: start of phony target address " |
| 104 | "range."), |
| 105 | cl::init(4096), // Start at "page 1" - no allocating at "null". |
| 106 | cl::Hidden); |
| 107 | |
| 108 | static cl::opt<uint64_t> |
| 109 | TargetAddrEnd("target-addr-end", |
| 110 | cl::desc("For -verify only: end of phony target address range."), |
| 111 | cl::init(~0ULL), |
| 112 | cl::Hidden); |
| 113 | |
| 114 | static cl::opt<uint64_t> |
| 115 | TargetSectionSep("target-section-sep", |
| 116 | cl::desc("For -verify only: Separation between sections in " |
| 117 | "phony target address space."), |
| 118 | cl::init(0), |
| 119 | cl::Hidden); |
| 120 | |
Lang Hames | 778ef5b | 2014-09-04 04:19:54 +0000 | [diff] [blame] | 121 | static cl::list<std::string> |
| 122 | SpecificSectionMappings("map-section", |
Lang Hames | 78937c2 | 2015-07-04 01:35:26 +0000 | [diff] [blame] | 123 | cl::desc("For -verify only: Map a section to a " |
| 124 | "specific address."), |
| 125 | cl::ZeroOrMore, |
| 126 | cl::Hidden); |
| 127 | |
| 128 | static cl::list<std::string> |
| 129 | DummySymbolMappings("dummy-extern", |
| 130 | cl::desc("For -verify only: Inject a symbol into the extern " |
| 131 | "symbol table."), |
| 132 | cl::ZeroOrMore, |
| 133 | cl::Hidden); |
Lang Hames | 778ef5b | 2014-09-04 04:19:54 +0000 | [diff] [blame] | 134 | |
Sanjoy Das | d5658b0 | 2015-11-23 21:47:51 +0000 | [diff] [blame] | 135 | static cl::opt<bool> |
| 136 | PrintAllocationRequests("print-alloc-requests", |
| 137 | cl::desc("Print allocation requests made to the memory " |
| 138 | "manager by RuntimeDyld"), |
| 139 | cl::Hidden); |
| 140 | |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 141 | /* *** */ |
| 142 | |
Jim Grosbach | 2dcef050 | 2011-04-04 23:04:39 +0000 | [diff] [blame] | 143 | // A trivial memory manager that doesn't do anything fancy, just uses the |
| 144 | // support library allocation routines directly. |
| 145 | class TrivialMemoryManager : public RTDyldMemoryManager { |
| 146 | public: |
Jim Grosbach | 3ed03f1 | 2011-04-12 00:23:32 +0000 | [diff] [blame] | 147 | SmallVector<sys::MemoryBlock, 16> FunctionMemory; |
Jim Grosbach | eff0a40 | 2012-01-16 22:26:39 +0000 | [diff] [blame] | 148 | SmallVector<sys::MemoryBlock, 16> DataMemory; |
| 149 | |
| 150 | uint8_t *allocateCodeSection(uintptr_t Size, unsigned Alignment, |
Craig Topper | e56917c | 2014-03-08 08:27:28 +0000 | [diff] [blame] | 151 | unsigned SectionID, |
| 152 | StringRef SectionName) override; |
Jim Grosbach | eff0a40 | 2012-01-16 22:26:39 +0000 | [diff] [blame] | 153 | uint8_t *allocateDataSection(uintptr_t Size, unsigned Alignment, |
Filip Pizlo | 7aa695e0 | 2013-10-02 00:59:25 +0000 | [diff] [blame] | 154 | unsigned SectionID, StringRef SectionName, |
Craig Topper | e56917c | 2014-03-08 08:27:28 +0000 | [diff] [blame] | 155 | bool IsReadOnly) override; |
Jim Grosbach | 3ed03f1 | 2011-04-12 00:23:32 +0000 | [diff] [blame] | 156 | |
Craig Topper | e56917c | 2014-03-08 08:27:28 +0000 | [diff] [blame] | 157 | void *getPointerToNamedFunction(const std::string &Name, |
| 158 | bool AbortOnFailure = true) override { |
Craig Topper | e6cb63e | 2014-04-25 04:24:47 +0000 | [diff] [blame] | 159 | return nullptr; |
Danil Malyshev | bfee542 | 2012-03-28 21:46:36 +0000 | [diff] [blame] | 160 | } |
| 161 | |
Craig Topper | e56917c | 2014-03-08 08:27:28 +0000 | [diff] [blame] | 162 | bool finalizeMemory(std::string *ErrMsg) override { return false; } |
Andrew Kaylor | a342cb9 | 2012-11-15 23:50:01 +0000 | [diff] [blame] | 163 | |
Lang Hames | 78937c2 | 2015-07-04 01:35:26 +0000 | [diff] [blame] | 164 | void addDummySymbol(const std::string &Name, uint64_t Addr) { |
| 165 | DummyExterns[Name] = Addr; |
| 166 | } |
| 167 | |
| 168 | RuntimeDyld::SymbolInfo findSymbol(const std::string &Name) override { |
| 169 | auto I = DummyExterns.find(Name); |
| 170 | |
| 171 | if (I != DummyExterns.end()) |
| 172 | return RuntimeDyld::SymbolInfo(I->second, JITSymbolFlags::Exported); |
| 173 | |
| 174 | return RTDyldMemoryManager::findSymbol(Name); |
| 175 | } |
| 176 | |
Tim Northover | 5af339a | 2015-06-03 18:26:52 +0000 | [diff] [blame] | 177 | void registerEHFrames(uint8_t *Addr, uint64_t LoadAddr, |
| 178 | size_t Size) override {} |
| 179 | void deregisterEHFrames(uint8_t *Addr, uint64_t LoadAddr, |
| 180 | size_t Size) override {} |
Davide Italiano | b59ea90 | 2015-10-21 22:12:03 +0000 | [diff] [blame] | 181 | |
| 182 | void preallocateSlab(uint64_t Size) { |
| 183 | std::string Err; |
| 184 | sys::MemoryBlock MB = sys::Memory::AllocateRWX(Size, nullptr, &Err); |
| 185 | if (!MB.base()) |
| 186 | report_fatal_error("Can't allocate enough memory: " + Err); |
| 187 | |
| 188 | PreallocSlab = MB; |
| 189 | UsePreallocation = true; |
| 190 | SlabSize = Size; |
| 191 | } |
| 192 | |
| 193 | uint8_t *allocateFromSlab(uintptr_t Size, unsigned Alignment, bool isCode) { |
Rui Ueyama | da00f2f | 2016-01-14 21:06:47 +0000 | [diff] [blame] | 194 | Size = alignTo(Size, Alignment); |
Davide Italiano | b59ea90 | 2015-10-21 22:12:03 +0000 | [diff] [blame] | 195 | if (CurrentSlabOffset + Size > SlabSize) |
| 196 | report_fatal_error("Can't allocate enough memory. Tune --preallocate"); |
| 197 | |
| 198 | uintptr_t OldSlabOffset = CurrentSlabOffset; |
| 199 | sys::MemoryBlock MB((void *)OldSlabOffset, Size); |
| 200 | if (isCode) |
| 201 | FunctionMemory.push_back(MB); |
| 202 | else |
| 203 | DataMemory.push_back(MB); |
| 204 | CurrentSlabOffset += Size; |
| 205 | return (uint8_t*)OldSlabOffset; |
| 206 | } |
| 207 | |
Lang Hames | 78937c2 | 2015-07-04 01:35:26 +0000 | [diff] [blame] | 208 | private: |
| 209 | std::map<std::string, uint64_t> DummyExterns; |
Davide Italiano | b59ea90 | 2015-10-21 22:12:03 +0000 | [diff] [blame] | 210 | sys::MemoryBlock PreallocSlab; |
| 211 | bool UsePreallocation = false; |
| 212 | uintptr_t SlabSize = 0; |
| 213 | uintptr_t CurrentSlabOffset = 0; |
Jim Grosbach | 2dcef050 | 2011-04-04 23:04:39 +0000 | [diff] [blame] | 214 | }; |
| 215 | |
Jim Grosbach | eff0a40 | 2012-01-16 22:26:39 +0000 | [diff] [blame] | 216 | uint8_t *TrivialMemoryManager::allocateCodeSection(uintptr_t Size, |
| 217 | unsigned Alignment, |
Filip Pizlo | 7aa695e0 | 2013-10-02 00:59:25 +0000 | [diff] [blame] | 218 | unsigned SectionID, |
| 219 | StringRef SectionName) { |
Sanjoy Das | d5658b0 | 2015-11-23 21:47:51 +0000 | [diff] [blame] | 220 | if (PrintAllocationRequests) |
| 221 | outs() << "allocateCodeSection(Size = " << Size << ", Alignment = " |
| 222 | << Alignment << ", SectionName = " << SectionName << ")\n"; |
| 223 | |
Davide Italiano | b59ea90 | 2015-10-21 22:12:03 +0000 | [diff] [blame] | 224 | if (UsePreallocation) |
| 225 | return allocateFromSlab(Size, Alignment, true /* isCode */); |
| 226 | |
Davide Italiano | 89151a0 | 2015-10-15 00:05:32 +0000 | [diff] [blame] | 227 | std::string Err; |
| 228 | sys::MemoryBlock MB = sys::Memory::AllocateRWX(Size, nullptr, &Err); |
| 229 | if (!MB.base()) |
| 230 | report_fatal_error("MemoryManager allocation failed: " + Err); |
Danil Malyshev | 8c17fbd | 2012-05-16 18:50:11 +0000 | [diff] [blame] | 231 | FunctionMemory.push_back(MB); |
| 232 | return (uint8_t*)MB.base(); |
Jim Grosbach | eff0a40 | 2012-01-16 22:26:39 +0000 | [diff] [blame] | 233 | } |
| 234 | |
| 235 | uint8_t *TrivialMemoryManager::allocateDataSection(uintptr_t Size, |
| 236 | unsigned Alignment, |
Andrew Kaylor | a342cb9 | 2012-11-15 23:50:01 +0000 | [diff] [blame] | 237 | unsigned SectionID, |
Filip Pizlo | 7aa695e0 | 2013-10-02 00:59:25 +0000 | [diff] [blame] | 238 | StringRef SectionName, |
Andrew Kaylor | a342cb9 | 2012-11-15 23:50:01 +0000 | [diff] [blame] | 239 | bool IsReadOnly) { |
Sanjoy Das | d5658b0 | 2015-11-23 21:47:51 +0000 | [diff] [blame] | 240 | if (PrintAllocationRequests) |
| 241 | outs() << "allocateDataSection(Size = " << Size << ", Alignment = " |
| 242 | << Alignment << ", SectionName = " << SectionName << ")\n"; |
| 243 | |
Davide Italiano | b59ea90 | 2015-10-21 22:12:03 +0000 | [diff] [blame] | 244 | if (UsePreallocation) |
| 245 | return allocateFromSlab(Size, Alignment, false /* isCode */); |
| 246 | |
Davide Italiano | 89151a0 | 2015-10-15 00:05:32 +0000 | [diff] [blame] | 247 | std::string Err; |
| 248 | sys::MemoryBlock MB = sys::Memory::AllocateRWX(Size, nullptr, &Err); |
| 249 | if (!MB.base()) |
| 250 | report_fatal_error("MemoryManager allocation failed: " + Err); |
Danil Malyshev | 8c17fbd | 2012-05-16 18:50:11 +0000 | [diff] [blame] | 251 | DataMemory.push_back(MB); |
| 252 | return (uint8_t*)MB.base(); |
| 253 | } |
| 254 | |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 255 | static const char *ProgramName; |
| 256 | |
Lang Hames | ee5417d | 2016-04-05 20:11:24 +0000 | [diff] [blame] | 257 | static void ErrorAndExit(const Twine &Msg) { |
Davide Italiano | c2e910d | 2015-11-20 23:12:15 +0000 | [diff] [blame] | 258 | errs() << ProgramName << ": error: " << Msg << "\n"; |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 259 | exit(1); |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 260 | } |
| 261 | |
Lang Hames | d311c0e | 2014-05-13 22:37:41 +0000 | [diff] [blame] | 262 | static void loadDylibs() { |
| 263 | for (const std::string &Dylib : Dylibs) { |
Davide Italiano | 5cdf936 | 2015-11-22 01:58:33 +0000 | [diff] [blame] | 264 | if (!sys::fs::is_regular_file(Dylib)) |
Davide Italiano | 41d0fa7 | 2015-11-21 05:58:19 +0000 | [diff] [blame] | 265 | report_fatal_error("Dylib not found: '" + Dylib + "'."); |
Davide Italiano | 5cdf936 | 2015-11-22 01:58:33 +0000 | [diff] [blame] | 266 | std::string ErrMsg; |
| 267 | if (sys::DynamicLibrary::LoadLibraryPermanently(Dylib.c_str(), &ErrMsg)) |
| 268 | report_fatal_error("Error loading '" + Dylib + "': " + ErrMsg); |
Lang Hames | d311c0e | 2014-05-13 22:37:41 +0000 | [diff] [blame] | 269 | } |
| 270 | } |
| 271 | |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 272 | /* *** */ |
| 273 | |
Keno Fischer | c780e8e | 2015-05-21 21:24:32 +0000 | [diff] [blame] | 274 | static int printLineInfoForInput(bool LoadObjects, bool UseDebugObj) { |
| 275 | assert(LoadObjects || !UseDebugObj); |
| 276 | |
Lang Hames | d311c0e | 2014-05-13 22:37:41 +0000 | [diff] [blame] | 277 | // Load any dylibs requested on the command line. |
| 278 | loadDylibs(); |
| 279 | |
Andrew Kaylor | d55d701 | 2013-01-25 22:50:58 +0000 | [diff] [blame] | 280 | // If we don't have any input files, read from stdin. |
| 281 | if (!InputFileList.size()) |
| 282 | InputFileList.push_back("-"); |
Davide Italiano | 5d7e8fd | 2015-10-12 00:57:29 +0000 | [diff] [blame] | 283 | for (auto &File : InputFileList) { |
Andrew Kaylor | d55d701 | 2013-01-25 22:50:58 +0000 | [diff] [blame] | 284 | // Instantiate a dynamic linker. |
Benjamin Kramer | 9ce7708 | 2013-08-03 22:16:31 +0000 | [diff] [blame] | 285 | TrivialMemoryManager MemMgr; |
Lang Hames | 633fe14 | 2015-03-30 03:37:06 +0000 | [diff] [blame] | 286 | RuntimeDyld Dyld(MemMgr, MemMgr); |
Andrew Kaylor | d55d701 | 2013-01-25 22:50:58 +0000 | [diff] [blame] | 287 | |
| 288 | // Load the input memory buffer. |
Andrew Kaylor | d55d701 | 2013-01-25 22:50:58 +0000 | [diff] [blame] | 289 | |
Rafael Espindola | adf21f2 | 2014-07-06 17:43:13 +0000 | [diff] [blame] | 290 | ErrorOr<std::unique_ptr<MemoryBuffer>> InputBuffer = |
Davide Italiano | 5d7e8fd | 2015-10-12 00:57:29 +0000 | [diff] [blame] | 291 | MemoryBuffer::getFileOrSTDIN(File); |
Rafael Espindola | adf21f2 | 2014-07-06 17:43:13 +0000 | [diff] [blame] | 292 | if (std::error_code EC = InputBuffer.getError()) |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 293 | ErrorAndExit("unable to read input: '" + EC.message() + "'"); |
Rafael Espindola | adf21f2 | 2014-07-06 17:43:13 +0000 | [diff] [blame] | 294 | |
Kevin Enderby | 3fcdf6a | 2016-04-06 22:14:09 +0000 | [diff] [blame] | 295 | Expected<std::unique_ptr<ObjectFile>> MaybeObj( |
Lang Hames | b5c7b1f | 2014-11-26 16:54:40 +0000 | [diff] [blame] | 296 | ObjectFile::createObjectFile((*InputBuffer)->getMemBufferRef())); |
| 297 | |
Kevin Enderby | 3fcdf6a | 2016-04-06 22:14:09 +0000 | [diff] [blame] | 298 | if (!MaybeObj) { |
| 299 | std::string Buf; |
| 300 | raw_string_ostream OS(Buf); |
| 301 | logAllUnhandledErrors(MaybeObj.takeError(), OS, ""); |
| 302 | OS.flush(); |
| 303 | ErrorAndExit("unable to create object file: '" + Buf + "'"); |
| 304 | } |
Lang Hames | b5c7b1f | 2014-11-26 16:54:40 +0000 | [diff] [blame] | 305 | |
| 306 | ObjectFile &Obj = **MaybeObj; |
| 307 | |
Keno Fischer | c780e8e | 2015-05-21 21:24:32 +0000 | [diff] [blame] | 308 | OwningBinary<ObjectFile> DebugObj; |
| 309 | std::unique_ptr<RuntimeDyld::LoadedObjectInfo> LoadedObjInfo = nullptr; |
| 310 | ObjectFile *SymbolObj = &Obj; |
| 311 | if (LoadObjects) { |
| 312 | // Load the object file |
| 313 | LoadedObjInfo = |
| 314 | Dyld.loadObject(Obj); |
Lang Hames | b5c7b1f | 2014-11-26 16:54:40 +0000 | [diff] [blame] | 315 | |
Keno Fischer | c780e8e | 2015-05-21 21:24:32 +0000 | [diff] [blame] | 316 | if (Dyld.hasError()) |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 317 | ErrorAndExit(Dyld.getErrorString()); |
Andrew Kaylor | d55d701 | 2013-01-25 22:50:58 +0000 | [diff] [blame] | 318 | |
Keno Fischer | c780e8e | 2015-05-21 21:24:32 +0000 | [diff] [blame] | 319 | // Resolve all the relocations we can. |
| 320 | Dyld.resolveRelocations(); |
Andrew Kaylor | d55d701 | 2013-01-25 22:50:58 +0000 | [diff] [blame] | 321 | |
Keno Fischer | c780e8e | 2015-05-21 21:24:32 +0000 | [diff] [blame] | 322 | if (UseDebugObj) { |
| 323 | DebugObj = LoadedObjInfo->getObjectForDebug(Obj); |
| 324 | SymbolObj = DebugObj.getBinary(); |
Lang Hames | 5c96933 | 2015-07-28 20:51:53 +0000 | [diff] [blame] | 325 | LoadedObjInfo.reset(); |
Keno Fischer | c780e8e | 2015-05-21 21:24:32 +0000 | [diff] [blame] | 326 | } |
| 327 | } |
Lang Hames | b5c7b1f | 2014-11-26 16:54:40 +0000 | [diff] [blame] | 328 | |
Ahmed Charles | 56440fd | 2014-03-06 05:51:42 +0000 | [diff] [blame] | 329 | std::unique_ptr<DIContext> Context( |
Keno Fischer | c780e8e | 2015-05-21 21:24:32 +0000 | [diff] [blame] | 330 | new DWARFContextInMemory(*SymbolObj,LoadedObjInfo.get())); |
Andrew Kaylor | d55d701 | 2013-01-25 22:50:58 +0000 | [diff] [blame] | 331 | |
Rafael Espindola | 6bf3221 | 2015-06-24 19:57:32 +0000 | [diff] [blame] | 332 | std::vector<std::pair<SymbolRef, uint64_t>> SymAddr = |
Rafael Espindola | b109c03 | 2015-06-23 02:08:48 +0000 | [diff] [blame] | 333 | object::computeSymbolSizes(*SymbolObj); |
Rafael Espindola | a82ce1d | 2015-05-31 23:15:35 +0000 | [diff] [blame] | 334 | |
Andrew Kaylor | d55d701 | 2013-01-25 22:50:58 +0000 | [diff] [blame] | 335 | // Use symbol info to iterate functions in the object. |
Rafael Espindola | 6bf3221 | 2015-06-24 19:57:32 +0000 | [diff] [blame] | 336 | for (const auto &P : SymAddr) { |
Rafael Espindola | b109c03 | 2015-06-23 02:08:48 +0000 | [diff] [blame] | 337 | object::SymbolRef Sym = P.first; |
Kevin Enderby | 5afbc1c | 2016-03-23 20:27:00 +0000 | [diff] [blame] | 338 | ErrorOr<SymbolRef::Type> TypeOrErr = Sym.getType(); |
| 339 | if (!TypeOrErr) |
| 340 | continue; |
| 341 | SymbolRef::Type Type = *TypeOrErr; |
| 342 | if (Type == object::SymbolRef::ST_Function) { |
Kevin Enderby | 81e8b7d | 2016-04-20 21:24:34 +0000 | [diff] [blame^] | 343 | Expected<StringRef> Name = Sym.getName(); |
| 344 | if (!Name) { |
| 345 | // TODO: Actually report errors helpfully. |
| 346 | consumeError(Name.takeError()); |
Rafael Espindola | da17627 | 2015-05-31 22:13:51 +0000 | [diff] [blame] | 347 | continue; |
Kevin Enderby | 81e8b7d | 2016-04-20 21:24:34 +0000 | [diff] [blame^] | 348 | } |
Rafael Espindola | ed067c4 | 2015-07-03 18:19:00 +0000 | [diff] [blame] | 349 | ErrorOr<uint64_t> AddrOrErr = Sym.getAddress(); |
| 350 | if (!AddrOrErr) |
Rafael Espindola | da17627 | 2015-05-31 22:13:51 +0000 | [diff] [blame] | 351 | continue; |
Rafael Espindola | ed067c4 | 2015-07-03 18:19:00 +0000 | [diff] [blame] | 352 | uint64_t Addr = *AddrOrErr; |
Rafael Espindola | a82ce1d | 2015-05-31 23:15:35 +0000 | [diff] [blame] | 353 | |
Rafael Espindola | b109c03 | 2015-06-23 02:08:48 +0000 | [diff] [blame] | 354 | uint64_t Size = P.second; |
Keno Fischer | c780e8e | 2015-05-21 21:24:32 +0000 | [diff] [blame] | 355 | // If we're not using the debug object, compute the address of the |
| 356 | // symbol in memory (rather than that in the unrelocated object file) |
| 357 | // and use that to query the DWARFContext. |
| 358 | if (!UseDebugObj && LoadObjects) { |
Rafael Espindola | 8bab889 | 2015-08-07 23:27:14 +0000 | [diff] [blame] | 359 | object::section_iterator Sec = *Sym.getSection(); |
Keno Fischer | c780e8e | 2015-05-21 21:24:32 +0000 | [diff] [blame] | 360 | StringRef SecName; |
| 361 | Sec->getName(SecName); |
| 362 | uint64_t SectionLoadAddress = |
Lang Hames | 2e88f4f | 2015-07-28 17:52:11 +0000 | [diff] [blame] | 363 | LoadedObjInfo->getSectionLoadAddress(*Sec); |
Keno Fischer | c780e8e | 2015-05-21 21:24:32 +0000 | [diff] [blame] | 364 | if (SectionLoadAddress != 0) |
| 365 | Addr += SectionLoadAddress - Sec->getAddress(); |
| 366 | } |
| 367 | |
Rafael Espindola | 5d0c2ff | 2015-07-02 20:55:21 +0000 | [diff] [blame] | 368 | outs() << "Function: " << *Name << ", Size = " << Size |
| 369 | << ", Addr = " << Addr << "\n"; |
Andrew Kaylor | d55d701 | 2013-01-25 22:50:58 +0000 | [diff] [blame] | 370 | |
Andrew Kaylor | 9a8ff81 | 2013-01-26 00:28:05 +0000 | [diff] [blame] | 371 | DILineInfoTable Lines = Context->getLineInfoForAddressRange(Addr, Size); |
Davide Italiano | 5d7e8fd | 2015-10-12 00:57:29 +0000 | [diff] [blame] | 372 | for (auto &D : Lines) { |
| 373 | outs() << " Line info @ " << D.first - Addr << ": " |
| 374 | << D.second.FileName << ", line:" << D.second.Line << "\n"; |
Andrew Kaylor | 9a8ff81 | 2013-01-26 00:28:05 +0000 | [diff] [blame] | 375 | } |
Andrew Kaylor | d55d701 | 2013-01-25 22:50:58 +0000 | [diff] [blame] | 376 | } |
| 377 | } |
| 378 | } |
| 379 | |
| 380 | return 0; |
| 381 | } |
| 382 | |
Davide Italiano | b59ea90 | 2015-10-21 22:12:03 +0000 | [diff] [blame] | 383 | static void doPreallocation(TrivialMemoryManager &MemMgr) { |
| 384 | // Allocate a slab of memory upfront, if required. This is used if |
| 385 | // we want to test small code models. |
| 386 | if (static_cast<intptr_t>(PreallocMemory) < 0) |
| 387 | report_fatal_error("Pre-allocated bytes of memory must be a positive integer."); |
| 388 | |
| 389 | // FIXME: Limit the amount of memory that can be preallocated? |
| 390 | if (PreallocMemory != 0) |
| 391 | MemMgr.preallocateSlab(PreallocMemory); |
| 392 | } |
| 393 | |
Jim Grosbach | 4d5284b | 2011-03-18 17:24:21 +0000 | [diff] [blame] | 394 | static int executeInput() { |
Lang Hames | d311c0e | 2014-05-13 22:37:41 +0000 | [diff] [blame] | 395 | // Load any dylibs requested on the command line. |
| 396 | loadDylibs(); |
| 397 | |
Jim Grosbach | f016b0a | 2011-03-21 22:15:52 +0000 | [diff] [blame] | 398 | // Instantiate a dynamic linker. |
Benjamin Kramer | 9ce7708 | 2013-08-03 22:16:31 +0000 | [diff] [blame] | 399 | TrivialMemoryManager MemMgr; |
Davide Italiano | b59ea90 | 2015-10-21 22:12:03 +0000 | [diff] [blame] | 400 | doPreallocation(MemMgr); |
Lang Hames | 633fe14 | 2015-03-30 03:37:06 +0000 | [diff] [blame] | 401 | RuntimeDyld Dyld(MemMgr, MemMgr); |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 402 | |
Jim Grosbach | 7cb41d7 | 2011-04-13 15:49:40 +0000 | [diff] [blame] | 403 | // If we don't have any input files, read from stdin. |
| 404 | if (!InputFileList.size()) |
| 405 | InputFileList.push_back("-"); |
Davide Italiano | 5d7e8fd | 2015-10-12 00:57:29 +0000 | [diff] [blame] | 406 | for (auto &File : InputFileList) { |
Jim Grosbach | 7cb41d7 | 2011-04-13 15:49:40 +0000 | [diff] [blame] | 407 | // Load the input memory buffer. |
Rafael Espindola | adf21f2 | 2014-07-06 17:43:13 +0000 | [diff] [blame] | 408 | ErrorOr<std::unique_ptr<MemoryBuffer>> InputBuffer = |
Davide Italiano | 5d7e8fd | 2015-10-12 00:57:29 +0000 | [diff] [blame] | 409 | MemoryBuffer::getFileOrSTDIN(File); |
Rafael Espindola | adf21f2 | 2014-07-06 17:43:13 +0000 | [diff] [blame] | 410 | if (std::error_code EC = InputBuffer.getError()) |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 411 | ErrorAndExit("unable to read input: '" + EC.message() + "'"); |
Kevin Enderby | 3fcdf6a | 2016-04-06 22:14:09 +0000 | [diff] [blame] | 412 | Expected<std::unique_ptr<ObjectFile>> MaybeObj( |
Lang Hames | b5c7b1f | 2014-11-26 16:54:40 +0000 | [diff] [blame] | 413 | ObjectFile::createObjectFile((*InputBuffer)->getMemBufferRef())); |
| 414 | |
Kevin Enderby | 3fcdf6a | 2016-04-06 22:14:09 +0000 | [diff] [blame] | 415 | if (!MaybeObj) { |
| 416 | std::string Buf; |
| 417 | raw_string_ostream OS(Buf); |
| 418 | logAllUnhandledErrors(MaybeObj.takeError(), OS, ""); |
| 419 | OS.flush(); |
| 420 | ErrorAndExit("unable to create object file: '" + Buf + "'"); |
| 421 | } |
Lang Hames | b5c7b1f | 2014-11-26 16:54:40 +0000 | [diff] [blame] | 422 | |
| 423 | ObjectFile &Obj = **MaybeObj; |
| 424 | |
Andrew Kaylor | adc7056 | 2012-10-02 21:18:39 +0000 | [diff] [blame] | 425 | // Load the object file |
Lang Hames | b5c7b1f | 2014-11-26 16:54:40 +0000 | [diff] [blame] | 426 | Dyld.loadObject(Obj); |
| 427 | if (Dyld.hasError()) { |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 428 | ErrorAndExit(Dyld.getErrorString()); |
Jim Grosbach | 7cb41d7 | 2011-04-13 15:49:40 +0000 | [diff] [blame] | 429 | } |
Jim Grosbach | 40411cc | 2011-03-22 18:19:42 +0000 | [diff] [blame] | 430 | } |
Jim Grosbach | 7cb41d7 | 2011-04-13 15:49:40 +0000 | [diff] [blame] | 431 | |
Davide Italiano | af08e1b | 2015-11-17 16:37:52 +0000 | [diff] [blame] | 432 | // Resove all the relocations we can. |
Jim Grosbach | 7cb41d7 | 2011-04-13 15:49:40 +0000 | [diff] [blame] | 433 | // FIXME: Error out if there are unresolved relocations. |
Davide Italiano | af08e1b | 2015-11-17 16:37:52 +0000 | [diff] [blame] | 434 | Dyld.resolveRelocations(); |
Jim Grosbach | 7cb41d7 | 2011-04-13 15:49:40 +0000 | [diff] [blame] | 435 | |
Jim Grosbach | d35159a | 2011-04-13 15:38:30 +0000 | [diff] [blame] | 436 | // Get the address of the entry point (_main by default). |
Lang Hames | b118603 | 2015-03-11 00:43:26 +0000 | [diff] [blame] | 437 | void *MainAddress = Dyld.getSymbolLocalAddress(EntryPoint); |
Craig Topper | e6cb63e | 2014-04-25 04:24:47 +0000 | [diff] [blame] | 438 | if (!MainAddress) |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 439 | ErrorAndExit("no definition for '" + EntryPoint + "'"); |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 440 | |
Jim Grosbach | 3ed03f1 | 2011-04-12 00:23:32 +0000 | [diff] [blame] | 441 | // Invalidate the instruction cache for each loaded function. |
Davide Italiano | 5d7e8fd | 2015-10-12 00:57:29 +0000 | [diff] [blame] | 442 | for (auto &FM : MemMgr.FunctionMemory) { |
Davide Italiano | af08e1b | 2015-11-17 16:37:52 +0000 | [diff] [blame] | 443 | |
Jim Grosbach | 3ed03f1 | 2011-04-12 00:23:32 +0000 | [diff] [blame] | 444 | // Make sure the memory is executable. |
Davide Italiano | af08e1b | 2015-11-17 16:37:52 +0000 | [diff] [blame] | 445 | // setExecutable will call InvalidateInstructionCache. |
Jim Grosbach | 3ed03f1 | 2011-04-12 00:23:32 +0000 | [diff] [blame] | 446 | std::string ErrorStr; |
Davide Italiano | 5d7e8fd | 2015-10-12 00:57:29 +0000 | [diff] [blame] | 447 | if (!sys::Memory::setExecutable(FM, &ErrorStr)) |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 448 | ErrorAndExit("unable to mark function executable: '" + ErrorStr + "'"); |
Jim Grosbach | 3ed03f1 | 2011-04-12 00:23:32 +0000 | [diff] [blame] | 449 | } |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 450 | |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 451 | // Dispatch to _main(). |
Jim Grosbach | 7cb41d7 | 2011-04-13 15:49:40 +0000 | [diff] [blame] | 452 | errs() << "loaded '" << EntryPoint << "' at: " << (void*)MainAddress << "\n"; |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 453 | |
| 454 | int (*Main)(int, const char**) = |
| 455 | (int(*)(int,const char**)) uintptr_t(MainAddress); |
| 456 | const char **Argv = new const char*[2]; |
Jim Grosbach | 7cb41d7 | 2011-04-13 15:49:40 +0000 | [diff] [blame] | 457 | // Use the name of the first input object module as argv[0] for the target. |
| 458 | Argv[0] = InputFileList[0].c_str(); |
Craig Topper | e6cb63e | 2014-04-25 04:24:47 +0000 | [diff] [blame] | 459 | Argv[1] = nullptr; |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 460 | return Main(1, Argv); |
| 461 | } |
| 462 | |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 463 | static int checkAllExpressions(RuntimeDyldChecker &Checker) { |
| 464 | for (const auto& CheckerFileName : CheckFiles) { |
Rafael Espindola | adf21f2 | 2014-07-06 17:43:13 +0000 | [diff] [blame] | 465 | ErrorOr<std::unique_ptr<MemoryBuffer>> CheckerFileBuf = |
| 466 | MemoryBuffer::getFileOrSTDIN(CheckerFileName); |
| 467 | if (std::error_code EC = CheckerFileBuf.getError()) |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 468 | ErrorAndExit("unable to read input '" + CheckerFileName + "': " + |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 469 | EC.message()); |
| 470 | |
Rafael Espindola | adf21f2 | 2014-07-06 17:43:13 +0000 | [diff] [blame] | 471 | if (!Checker.checkAllRulesInBuffer("# rtdyld-check:", |
| 472 | CheckerFileBuf.get().get())) |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 473 | ErrorAndExit("some checks in '" + CheckerFileName + "' failed"); |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 474 | } |
| 475 | return 0; |
| 476 | } |
| 477 | |
Benjamin Kramer | f044d3f | 2015-03-09 16:23:46 +0000 | [diff] [blame] | 478 | static std::map<void *, uint64_t> |
Lang Hames | 778ef5b | 2014-09-04 04:19:54 +0000 | [diff] [blame] | 479 | applySpecificSectionMappings(RuntimeDyldChecker &Checker) { |
| 480 | |
| 481 | std::map<void*, uint64_t> SpecificMappings; |
| 482 | |
| 483 | for (StringRef Mapping : SpecificSectionMappings) { |
| 484 | |
| 485 | size_t EqualsIdx = Mapping.find_first_of("="); |
Lang Hames | 78937c2 | 2015-07-04 01:35:26 +0000 | [diff] [blame] | 486 | std::string SectionIDStr = Mapping.substr(0, EqualsIdx); |
Lang Hames | 778ef5b | 2014-09-04 04:19:54 +0000 | [diff] [blame] | 487 | size_t ComaIdx = Mapping.find_first_of(","); |
| 488 | |
Davide Italiano | 07557fc | 2015-11-21 02:15:51 +0000 | [diff] [blame] | 489 | if (ComaIdx == StringRef::npos) |
| 490 | report_fatal_error("Invalid section specification '" + Mapping + |
| 491 | "'. Should be '<file name>,<section name>=<addr>'"); |
Lang Hames | 778ef5b | 2014-09-04 04:19:54 +0000 | [diff] [blame] | 492 | |
Lang Hames | 78937c2 | 2015-07-04 01:35:26 +0000 | [diff] [blame] | 493 | std::string FileName = SectionIDStr.substr(0, ComaIdx); |
| 494 | std::string SectionName = SectionIDStr.substr(ComaIdx + 1); |
Lang Hames | 778ef5b | 2014-09-04 04:19:54 +0000 | [diff] [blame] | 495 | |
| 496 | uint64_t OldAddrInt; |
| 497 | std::string ErrorMsg; |
| 498 | std::tie(OldAddrInt, ErrorMsg) = |
| 499 | Checker.getSectionAddr(FileName, SectionName, true); |
| 500 | |
Davide Italiano | 07557fc | 2015-11-21 02:15:51 +0000 | [diff] [blame] | 501 | if (ErrorMsg != "") |
| 502 | report_fatal_error(ErrorMsg); |
Lang Hames | 778ef5b | 2014-09-04 04:19:54 +0000 | [diff] [blame] | 503 | |
| 504 | void* OldAddr = reinterpret_cast<void*>(static_cast<uintptr_t>(OldAddrInt)); |
| 505 | |
Lang Hames | 78937c2 | 2015-07-04 01:35:26 +0000 | [diff] [blame] | 506 | std::string NewAddrStr = Mapping.substr(EqualsIdx + 1); |
Lang Hames | 778ef5b | 2014-09-04 04:19:54 +0000 | [diff] [blame] | 507 | uint64_t NewAddr; |
| 508 | |
Davide Italiano | 07557fc | 2015-11-21 02:15:51 +0000 | [diff] [blame] | 509 | if (StringRef(NewAddrStr).getAsInteger(0, NewAddr)) |
| 510 | report_fatal_error("Invalid section address in mapping '" + Mapping + |
| 511 | "'."); |
Lang Hames | 778ef5b | 2014-09-04 04:19:54 +0000 | [diff] [blame] | 512 | |
| 513 | Checker.getRTDyld().mapSectionAddress(OldAddr, NewAddr); |
| 514 | SpecificMappings[OldAddr] = NewAddr; |
| 515 | } |
| 516 | |
| 517 | return SpecificMappings; |
| 518 | } |
| 519 | |
Lang Hames | 9cb7353 | 2014-07-29 23:43:13 +0000 | [diff] [blame] | 520 | // Scatter sections in all directions! |
| 521 | // Remaps section addresses for -verify mode. The following command line options |
| 522 | // can be used to customize the layout of the memory within the phony target's |
| 523 | // address space: |
| 524 | // -target-addr-start <s> -- Specify where the phony target addres range starts. |
| 525 | // -target-addr-end <e> -- Specify where the phony target address range ends. |
| 526 | // -target-section-sep <d> -- Specify how big a gap should be left between the |
| 527 | // end of one section and the start of the next. |
| 528 | // Defaults to zero. Set to something big |
| 529 | // (e.g. 1 << 32) to stress-test stubs, GOTs, etc. |
| 530 | // |
Lang Hames | 78937c2 | 2015-07-04 01:35:26 +0000 | [diff] [blame] | 531 | static void remapSectionsAndSymbols(const llvm::Triple &TargetTriple, |
| 532 | TrivialMemoryManager &MemMgr, |
| 533 | RuntimeDyldChecker &Checker) { |
Lang Hames | 778ef5b | 2014-09-04 04:19:54 +0000 | [diff] [blame] | 534 | |
| 535 | // Set up a work list (section addr/size pairs). |
| 536 | typedef std::list<std::pair<void*, uint64_t>> WorklistT; |
| 537 | WorklistT Worklist; |
| 538 | |
| 539 | for (const auto& CodeSection : MemMgr.FunctionMemory) |
| 540 | Worklist.push_back(std::make_pair(CodeSection.base(), CodeSection.size())); |
| 541 | for (const auto& DataSection : MemMgr.DataMemory) |
| 542 | Worklist.push_back(std::make_pair(DataSection.base(), DataSection.size())); |
| 543 | |
| 544 | // Apply any section-specific mappings that were requested on the command |
| 545 | // line. |
| 546 | typedef std::map<void*, uint64_t> AppliedMappingsT; |
| 547 | AppliedMappingsT AppliedMappings = applySpecificSectionMappings(Checker); |
| 548 | |
| 549 | // Keep an "already allocated" mapping of section target addresses to sizes. |
| 550 | // Sections whose address mappings aren't specified on the command line will |
| 551 | // allocated around the explicitly mapped sections while maintaining the |
| 552 | // minimum separation. |
| 553 | std::map<uint64_t, uint64_t> AlreadyAllocated; |
| 554 | |
| 555 | // Move the previously applied mappings into the already-allocated map. |
| 556 | for (WorklistT::iterator I = Worklist.begin(), E = Worklist.end(); |
| 557 | I != E;) { |
| 558 | WorklistT::iterator Tmp = I; |
| 559 | ++I; |
| 560 | AppliedMappingsT::iterator AI = AppliedMappings.find(Tmp->first); |
| 561 | |
| 562 | if (AI != AppliedMappings.end()) { |
| 563 | AlreadyAllocated[AI->second] = Tmp->second; |
| 564 | Worklist.erase(Tmp); |
| 565 | } |
| 566 | } |
Lang Hames | 9cb7353 | 2014-07-29 23:43:13 +0000 | [diff] [blame] | 567 | |
| 568 | // If the -target-addr-end option wasn't explicitly passed, then set it to a |
| 569 | // sensible default based on the target triple. |
| 570 | if (TargetAddrEnd.getNumOccurrences() == 0) { |
| 571 | if (TargetTriple.isArch16Bit()) |
| 572 | TargetAddrEnd = (1ULL << 16) - 1; |
| 573 | else if (TargetTriple.isArch32Bit()) |
| 574 | TargetAddrEnd = (1ULL << 32) - 1; |
| 575 | // TargetAddrEnd already has a sensible default for 64-bit systems, so |
| 576 | // there's nothing to do in the 64-bit case. |
| 577 | } |
| 578 | |
Lang Hames | 778ef5b | 2014-09-04 04:19:54 +0000 | [diff] [blame] | 579 | // Process any elements remaining in the worklist. |
| 580 | while (!Worklist.empty()) { |
| 581 | std::pair<void*, uint64_t> CurEntry = Worklist.front(); |
| 582 | Worklist.pop_front(); |
Lang Hames | 9cb7353 | 2014-07-29 23:43:13 +0000 | [diff] [blame] | 583 | |
Lang Hames | 778ef5b | 2014-09-04 04:19:54 +0000 | [diff] [blame] | 584 | uint64_t NextSectionAddr = TargetAddrStart; |
| 585 | |
| 586 | for (const auto &Alloc : AlreadyAllocated) |
| 587 | if (NextSectionAddr + CurEntry.second + TargetSectionSep <= Alloc.first) |
| 588 | break; |
| 589 | else |
| 590 | NextSectionAddr = Alloc.first + Alloc.second + TargetSectionSep; |
| 591 | |
| 592 | AlreadyAllocated[NextSectionAddr] = CurEntry.second; |
| 593 | Checker.getRTDyld().mapSectionAddress(CurEntry.first, NextSectionAddr); |
Lang Hames | 9cb7353 | 2014-07-29 23:43:13 +0000 | [diff] [blame] | 594 | } |
| 595 | |
Lang Hames | 78937c2 | 2015-07-04 01:35:26 +0000 | [diff] [blame] | 596 | // Add dummy symbols to the memory manager. |
| 597 | for (const auto &Mapping : DummySymbolMappings) { |
| 598 | size_t EqualsIdx = Mapping.find_first_of("="); |
| 599 | |
Davide Italiano | 07557fc | 2015-11-21 02:15:51 +0000 | [diff] [blame] | 600 | if (EqualsIdx == StringRef::npos) |
| 601 | report_fatal_error("Invalid dummy symbol specification '" + Mapping + |
| 602 | "'. Should be '<symbol name>=<addr>'"); |
Lang Hames | 78937c2 | 2015-07-04 01:35:26 +0000 | [diff] [blame] | 603 | |
| 604 | std::string Symbol = Mapping.substr(0, EqualsIdx); |
| 605 | std::string AddrStr = Mapping.substr(EqualsIdx + 1); |
| 606 | |
| 607 | uint64_t Addr; |
Davide Italiano | 07557fc | 2015-11-21 02:15:51 +0000 | [diff] [blame] | 608 | if (StringRef(AddrStr).getAsInteger(0, Addr)) |
| 609 | report_fatal_error("Invalid symbol mapping '" + Mapping + "'."); |
Lang Hames | 78937c2 | 2015-07-04 01:35:26 +0000 | [diff] [blame] | 610 | |
| 611 | MemMgr.addDummySymbol(Symbol, Addr); |
| 612 | } |
Lang Hames | 9cb7353 | 2014-07-29 23:43:13 +0000 | [diff] [blame] | 613 | } |
| 614 | |
| 615 | // Load and link the objects specified on the command line, but do not execute |
| 616 | // anything. Instead, attach a RuntimeDyldChecker instance and call it to |
| 617 | // verify the correctness of the linked memory. |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 618 | static int linkAndVerify() { |
| 619 | |
| 620 | // Check for missing triple. |
Davide Italiano | 78da759 | 2015-11-21 05:44:41 +0000 | [diff] [blame] | 621 | if (TripleName == "") |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 622 | ErrorAndExit("-triple required when running in -verify mode."); |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 623 | |
| 624 | // Look up the target and build the disassembler. |
| 625 | Triple TheTriple(Triple::normalize(TripleName)); |
| 626 | std::string ErrorStr; |
| 627 | const Target *TheTarget = |
| 628 | TargetRegistry::lookupTarget("", TheTriple, ErrorStr); |
Davide Italiano | 78da759 | 2015-11-21 05:44:41 +0000 | [diff] [blame] | 629 | if (!TheTarget) |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 630 | ErrorAndExit("Error accessing target '" + TripleName + "': " + ErrorStr); |
Davide Italiano | 78da759 | 2015-11-21 05:44:41 +0000 | [diff] [blame] | 631 | |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 632 | TripleName = TheTriple.getTriple(); |
| 633 | |
| 634 | std::unique_ptr<MCSubtargetInfo> STI( |
Petar Jovanovic | 280e562 | 2015-06-23 22:52:19 +0000 | [diff] [blame] | 635 | TheTarget->createMCSubtargetInfo(TripleName, MCPU, "")); |
Davide Italiano | ebb27af | 2015-11-21 05:49:07 +0000 | [diff] [blame] | 636 | if (!STI) |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 637 | ErrorAndExit("Unable to create subtarget info!"); |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 638 | |
| 639 | std::unique_ptr<MCRegisterInfo> MRI(TheTarget->createMCRegInfo(TripleName)); |
Davide Italiano | ebb27af | 2015-11-21 05:49:07 +0000 | [diff] [blame] | 640 | if (!MRI) |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 641 | ErrorAndExit("Unable to create target register info!"); |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 642 | |
| 643 | std::unique_ptr<MCAsmInfo> MAI(TheTarget->createMCAsmInfo(*MRI, TripleName)); |
Davide Italiano | ebb27af | 2015-11-21 05:49:07 +0000 | [diff] [blame] | 644 | if (!MAI) |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 645 | ErrorAndExit("Unable to create target asm info!"); |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 646 | |
| 647 | MCContext Ctx(MAI.get(), MRI.get(), nullptr); |
| 648 | |
| 649 | std::unique_ptr<MCDisassembler> Disassembler( |
| 650 | TheTarget->createMCDisassembler(*STI, Ctx)); |
Davide Italiano | ebb27af | 2015-11-21 05:49:07 +0000 | [diff] [blame] | 651 | if (!Disassembler) |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 652 | ErrorAndExit("Unable to create disassembler!"); |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 653 | |
| 654 | std::unique_ptr<MCInstrInfo> MII(TheTarget->createMCInstrInfo()); |
| 655 | |
Daniel Sanders | 50f1723 | 2015-09-15 16:17:27 +0000 | [diff] [blame] | 656 | std::unique_ptr<MCInstPrinter> InstPrinter( |
| 657 | TheTarget->createMCInstPrinter(Triple(TripleName), 0, *MAI, *MII, *MRI)); |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 658 | |
| 659 | // Load any dylibs requested on the command line. |
| 660 | loadDylibs(); |
| 661 | |
| 662 | // Instantiate a dynamic linker. |
| 663 | TrivialMemoryManager MemMgr; |
Davide Italiano | b59ea90 | 2015-10-21 22:12:03 +0000 | [diff] [blame] | 664 | doPreallocation(MemMgr); |
Lang Hames | 633fe14 | 2015-03-30 03:37:06 +0000 | [diff] [blame] | 665 | RuntimeDyld Dyld(MemMgr, MemMgr); |
Lang Hames | 925e51b | 2014-09-03 05:42:52 +0000 | [diff] [blame] | 666 | Dyld.setProcessAllSections(true); |
Lang Hames | f7acddd | 2014-07-22 22:47:39 +0000 | [diff] [blame] | 667 | RuntimeDyldChecker Checker(Dyld, Disassembler.get(), InstPrinter.get(), |
| 668 | llvm::dbgs()); |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 669 | |
| 670 | // If we don't have any input files, read from stdin. |
| 671 | if (!InputFileList.size()) |
| 672 | InputFileList.push_back("-"); |
Davide Italiano | d91bf8e | 2015-10-10 00:45:24 +0000 | [diff] [blame] | 673 | for (auto &Filename : InputFileList) { |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 674 | // Load the input memory buffer. |
Rafael Espindola | adf21f2 | 2014-07-06 17:43:13 +0000 | [diff] [blame] | 675 | ErrorOr<std::unique_ptr<MemoryBuffer>> InputBuffer = |
Davide Italiano | d91bf8e | 2015-10-10 00:45:24 +0000 | [diff] [blame] | 676 | MemoryBuffer::getFileOrSTDIN(Filename); |
Lang Hames | b5c7b1f | 2014-11-26 16:54:40 +0000 | [diff] [blame] | 677 | |
Rafael Espindola | adf21f2 | 2014-07-06 17:43:13 +0000 | [diff] [blame] | 678 | if (std::error_code EC = InputBuffer.getError()) |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 679 | ErrorAndExit("unable to read input: '" + EC.message() + "'"); |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 680 | |
Kevin Enderby | 3fcdf6a | 2016-04-06 22:14:09 +0000 | [diff] [blame] | 681 | Expected<std::unique_ptr<ObjectFile>> MaybeObj( |
Lang Hames | b5c7b1f | 2014-11-26 16:54:40 +0000 | [diff] [blame] | 682 | ObjectFile::createObjectFile((*InputBuffer)->getMemBufferRef())); |
| 683 | |
Kevin Enderby | 3fcdf6a | 2016-04-06 22:14:09 +0000 | [diff] [blame] | 684 | if (!MaybeObj) { |
| 685 | std::string Buf; |
| 686 | raw_string_ostream OS(Buf); |
| 687 | logAllUnhandledErrors(MaybeObj.takeError(), OS, ""); |
| 688 | OS.flush(); |
| 689 | ErrorAndExit("unable to create object file: '" + Buf + "'"); |
| 690 | } |
Lang Hames | b5c7b1f | 2014-11-26 16:54:40 +0000 | [diff] [blame] | 691 | |
| 692 | ObjectFile &Obj = **MaybeObj; |
| 693 | |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 694 | // Load the object file |
Lang Hames | b5c7b1f | 2014-11-26 16:54:40 +0000 | [diff] [blame] | 695 | Dyld.loadObject(Obj); |
| 696 | if (Dyld.hasError()) { |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 697 | ErrorAndExit(Dyld.getErrorString()); |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 698 | } |
| 699 | } |
| 700 | |
Lang Hames | 78937c2 | 2015-07-04 01:35:26 +0000 | [diff] [blame] | 701 | // Re-map the section addresses into the phony target address space and add |
| 702 | // dummy symbols. |
| 703 | remapSectionsAndSymbols(TheTriple, MemMgr, Checker); |
Lang Hames | 375385f | 2014-07-30 03:12:41 +0000 | [diff] [blame] | 704 | |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 705 | // Resolve all the relocations we can. |
| 706 | Dyld.resolveRelocations(); |
| 707 | |
Lang Hames | 925e51b | 2014-09-03 05:42:52 +0000 | [diff] [blame] | 708 | // Register EH frames. |
| 709 | Dyld.registerEHFrames(); |
| 710 | |
Lang Hames | ae17268 | 2014-08-05 20:51:46 +0000 | [diff] [blame] | 711 | int ErrorCode = checkAllExpressions(Checker); |
Davide Italiano | 78da759 | 2015-11-21 05:44:41 +0000 | [diff] [blame] | 712 | if (Dyld.hasError()) |
Lang Hames | 9e964f3 | 2016-03-25 17:25:34 +0000 | [diff] [blame] | 713 | ErrorAndExit("RTDyld reported an error applying relocations:\n " + |
Davide Italiano | 78da759 | 2015-11-21 05:44:41 +0000 | [diff] [blame] | 714 | Dyld.getErrorString()); |
Lang Hames | ae17268 | 2014-08-05 20:51:46 +0000 | [diff] [blame] | 715 | |
| 716 | return ErrorCode; |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 717 | } |
| 718 | |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 719 | int main(int argc, char **argv) { |
Alp Toker | 9f67932 | 2013-11-05 09:33:43 +0000 | [diff] [blame] | 720 | sys::PrintStackTraceOnErrorSignal(); |
| 721 | PrettyStackTraceProgram X(argc, argv); |
| 722 | |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 723 | ProgramName = argv[0]; |
| 724 | llvm_shutdown_obj Y; // Call llvm_shutdown() on exit. |
| 725 | |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 726 | llvm::InitializeAllTargetInfos(); |
| 727 | llvm::InitializeAllTargetMCs(); |
| 728 | llvm::InitializeAllDisassemblers(); |
| 729 | |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 730 | cl::ParseCommandLineOptions(argc, argv, "llvm MC-JIT tool\n"); |
| 731 | |
| 732 | switch (Action) { |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 733 | case AC_Execute: |
Jim Grosbach | 4d5284b | 2011-03-18 17:24:21 +0000 | [diff] [blame] | 734 | return executeInput(); |
Keno Fischer | c780e8e | 2015-05-21 21:24:32 +0000 | [diff] [blame] | 735 | case AC_PrintDebugLineInfo: |
Keno Fischer | 281b694 | 2015-05-30 19:44:53 +0000 | [diff] [blame] | 736 | return printLineInfoForInput(/* LoadObjects */ true,/* UseDebugObj */ true); |
Andrew Kaylor | d55d701 | 2013-01-25 22:50:58 +0000 | [diff] [blame] | 737 | case AC_PrintLineInfo: |
Keno Fischer | 281b694 | 2015-05-30 19:44:53 +0000 | [diff] [blame] | 738 | return printLineInfoForInput(/* LoadObjects */ true,/* UseDebugObj */false); |
| 739 | case AC_PrintObjectLineInfo: |
| 740 | return printLineInfoForInput(/* LoadObjects */false,/* UseDebugObj */false); |
Lang Hames | e1c1138 | 2014-06-27 20:20:57 +0000 | [diff] [blame] | 741 | case AC_Verify: |
| 742 | return linkAndVerify(); |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 743 | } |
Jim Grosbach | 0072cdb | 2011-03-18 17:11:39 +0000 | [diff] [blame] | 744 | } |