1 //===-- RuntimeDyldCOFF.cpp - Run-time dynamic linker for MC-JIT -*- C++ -*-==//
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 // Implementation of COFF support for the MC-JIT runtime dynamic linker.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "RuntimeDyldCOFF.h"
15 #include "Targets/RuntimeDyldCOFFI386.h"
16 #include "Targets/RuntimeDyldCOFFThumb.h"
17 #include "Targets/RuntimeDyldCOFFX86_64.h"
18 #include "llvm/ADT/STLExtras.h"
19 #include "llvm/ADT/Triple.h"
20 #include "llvm/Object/ObjectFile.h"
21 
22 using namespace llvm;
23 using namespace llvm::object;
24 
25 #define DEBUG_TYPE "dyld"
26 
27 namespace {
28 
29 class LoadedCOFFObjectInfo final
30     : public LoadedObjectInfoHelper<LoadedCOFFObjectInfo,
31                                     RuntimeDyld::LoadedObjectInfo> {
32 public:
LoadedCOFFObjectInfo(RuntimeDyldImpl & RTDyld,RuntimeDyld::LoadedObjectInfo::ObjSectionToIDMap ObjSecToIDMap)33   LoadedCOFFObjectInfo(
34       RuntimeDyldImpl &RTDyld,
35       RuntimeDyld::LoadedObjectInfo::ObjSectionToIDMap ObjSecToIDMap)
36       : LoadedObjectInfoHelper(RTDyld, std::move(ObjSecToIDMap)) {}
37 
38   OwningBinary<ObjectFile>
getObjectForDebug(const ObjectFile & Obj) const39   getObjectForDebug(const ObjectFile &Obj) const override {
40     return OwningBinary<ObjectFile>();
41   }
42 };
43 }
44 
45 namespace llvm {
46 
47 std::unique_ptr<RuntimeDyldCOFF>
create(Triple::ArchType Arch,RuntimeDyld::MemoryManager & MemMgr,JITSymbolResolver & Resolver)48 llvm::RuntimeDyldCOFF::create(Triple::ArchType Arch,
49                               RuntimeDyld::MemoryManager &MemMgr,
50                               JITSymbolResolver &Resolver) {
51   switch (Arch) {
52   default: llvm_unreachable("Unsupported target for RuntimeDyldCOFF.");
53   case Triple::x86:
54     return make_unique<RuntimeDyldCOFFI386>(MemMgr, Resolver);
55   case Triple::thumb:
56     return make_unique<RuntimeDyldCOFFThumb>(MemMgr, Resolver);
57   case Triple::x86_64:
58     return make_unique<RuntimeDyldCOFFX86_64>(MemMgr, Resolver);
59   }
60 }
61 
62 std::unique_ptr<RuntimeDyld::LoadedObjectInfo>
loadObject(const object::ObjectFile & O)63 RuntimeDyldCOFF::loadObject(const object::ObjectFile &O) {
64   if (auto ObjSectionToIDOrErr = loadObjectImpl(O)) {
65     return llvm::make_unique<LoadedCOFFObjectInfo>(*this, *ObjSectionToIDOrErr);
66   } else {
67     HasError = true;
68     raw_string_ostream ErrStream(ErrorStr);
69     logAllUnhandledErrors(ObjSectionToIDOrErr.takeError(), ErrStream, "");
70     return nullptr;
71   }
72 }
73 
getSymbolOffset(const SymbolRef & Sym)74 uint64_t RuntimeDyldCOFF::getSymbolOffset(const SymbolRef &Sym) {
75   // The value in a relocatable COFF object is the offset.
76   return Sym.getValue();
77 }
78 
isCompatibleFile(const object::ObjectFile & Obj) const79 bool RuntimeDyldCOFF::isCompatibleFile(const object::ObjectFile &Obj) const {
80   return Obj.isCOFF();
81 }
82 
83 } // namespace llvm
84