1 //===-- RuntimeDyldCOFF.cpp - Run-time dynamic linker for MC-JIT -*- C++ -*-==//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // Implementation of COFF support for the MC-JIT runtime dynamic linker.
10 //
11 //===----------------------------------------------------------------------===//
12
13 #include "RuntimeDyldCOFF.h"
14 #include "Targets/RuntimeDyldCOFFAArch64.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 std::make_unique<RuntimeDyldCOFFI386>(MemMgr, Resolver);
55 case Triple::thumb:
56 return std::make_unique<RuntimeDyldCOFFThumb>(MemMgr, Resolver);
57 case Triple::x86_64:
58 return std::make_unique<RuntimeDyldCOFFX86_64>(MemMgr, Resolver);
59 case Triple::aarch64:
60 return std::make_unique<RuntimeDyldCOFFAArch64>(MemMgr, Resolver);
61 }
62 }
63
64 std::unique_ptr<RuntimeDyld::LoadedObjectInfo>
loadObject(const object::ObjectFile & O)65 RuntimeDyldCOFF::loadObject(const object::ObjectFile &O) {
66 if (auto ObjSectionToIDOrErr = loadObjectImpl(O)) {
67 return std::make_unique<LoadedCOFFObjectInfo>(*this, *ObjSectionToIDOrErr);
68 } else {
69 HasError = true;
70 raw_string_ostream ErrStream(ErrorStr);
71 logAllUnhandledErrors(ObjSectionToIDOrErr.takeError(), ErrStream);
72 return nullptr;
73 }
74 }
75
getSymbolOffset(const SymbolRef & Sym)76 uint64_t RuntimeDyldCOFF::getSymbolOffset(const SymbolRef &Sym) {
77 // The value in a relocatable COFF object is the offset.
78 return Sym.getValue();
79 }
80
isCompatibleFile(const object::ObjectFile & Obj) const81 bool RuntimeDyldCOFF::isCompatibleFile(const object::ObjectFile &Obj) const {
82 return Obj.isCOFF();
83 }
84
85 } // namespace llvm
86