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