LLVM 19.0.0git
PerGraphGOTAndPLTStubsBuilder.h
Go to the documentation of this file.
1//===--------------- PerGraphGOTAndPLTStubBuilder.h -------------*- 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// Construct GOT and PLT entries for each graph.
10//
11//===----------------------------------------------------------------------===//
12
13#ifndef LLVM_EXECUTIONENGINE_JITLINK_PERGRAPHGOTANDPLTSTUBSBUILDER_H
14#define LLVM_EXECUTIONENGINE_JITLINK_PERGRAPHGOTANDPLTSTUBSBUILDER_H
15
17#include "llvm/Support/Debug.h"
18
19#define DEBUG_TYPE "jitlink"
20
21namespace llvm {
22namespace jitlink {
23
24/// Per-object GOT and PLT Stub builder.
25///
26/// Constructs GOT entries and PLT stubs in every graph for referenced symbols.
27/// Building these blocks in every graph is likely to lead to duplicate entries
28/// in the JITLinkDylib, but allows graphs to be trivially removed independently
29/// without affecting other graphs (since those other graphs will have their own
30/// copies of any required entries).
31template <typename BuilderImplT>
33public:
35
36 static Error asPass(LinkGraph &G) { return BuilderImplT(G).run(); }
37
39 LLVM_DEBUG(dbgs() << "Running Per-Graph GOT and Stubs builder:\n");
40
41 // We're going to be adding new blocks, but we don't want to iterate over
42 // the new ones, so build a worklist.
43 std::vector<Block *> Worklist(G.blocks().begin(), G.blocks().end());
44
45 for (auto *B : Worklist)
46 for (auto &E : B->edges()) {
47 if (impl().isGOTEdgeToFix(E)) {
49 dbgs() << " Fixing " << G.getEdgeKindName(E.getKind())
50 << " edge at " << B->getFixupAddress(E) << " ("
51 << B->getAddress() << " + "
52 << formatv("{0:x}", E.getOffset()) << ")\n";
53 });
54 impl().fixGOTEdge(E, getGOTEntry(E.getTarget()));
55 } else if (impl().isExternalBranchEdge(E)) {
57 dbgs() << " Fixing " << G.getEdgeKindName(E.getKind())
58 << " edge at " << B->getFixupAddress(E) << " ("
59 << B->getAddress() << " + "
60 << formatv("{0:x}", E.getOffset()) << ")\n";
61 });
62 impl().fixPLTEdge(E, getPLTStub(E.getTarget()));
63 }
64 }
65
66 return Error::success();
67 }
68
69protected:
71 assert(Target.hasName() && "GOT edge cannot point to anonymous target");
72
73 auto GOTEntryI = GOTEntries.find(Target.getName());
74
75 // Build the entry if it doesn't exist.
76 if (GOTEntryI == GOTEntries.end()) {
77 auto &GOTEntry = impl().createGOTEntry(Target);
79 dbgs() << " Created GOT entry for " << Target.getName() << ": "
80 << GOTEntry << "\n";
81 });
82 GOTEntryI =
83 GOTEntries.insert(std::make_pair(Target.getName(), &GOTEntry)).first;
84 }
85
86 assert(GOTEntryI != GOTEntries.end() && "Could not get GOT entry symbol");
88 { dbgs() << " Using GOT entry " << *GOTEntryI->second << "\n"; });
89 return *GOTEntryI->second;
90 }
91
93 assert(Target.hasName() &&
94 "External branch edge can not point to an anonymous target");
95 auto StubI = PLTStubs.find(Target.getName());
96
97 if (StubI == PLTStubs.end()) {
98 auto &StubSymbol = impl().createPLTStub(Target);
100 dbgs() << " Created PLT stub for " << Target.getName() << ": "
101 << StubSymbol << "\n";
102 });
103 StubI =
104 PLTStubs.insert(std::make_pair(Target.getName(), &StubSymbol)).first;
105 }
106
107 assert(StubI != PLTStubs.end() && "Count not get stub symbol");
108 LLVM_DEBUG({ dbgs() << " Using PLT stub " << *StubI->second << "\n"; });
109 return *StubI->second;
110 }
111
113
114private:
115 BuilderImplT &impl() { return static_cast<BuilderImplT &>(*this); }
116
119};
120
121} // end namespace jitlink
122} // end namespace llvm
123
124#undef DEBUG_TYPE
125
126#endif // LLVM_EXECUTIONENGINE_JITLINK_PERGRAPHGOTANDPLTSTUBSBUILDER_H
static GCRegistry::Add< OcamlGC > B("ocaml", "ocaml 3.10-compatible GC")
static GCRegistry::Add< CoreCLRGC > E("coreclr", "CoreCLR-compatible GC")
#define LLVM_DEBUG(X)
Definition: Debug.h:101
assert(ImpDefSCC.getReg()==AMDGPU::SCC &&ImpDefSCC.isDef())
Lightweight error class with error context and mandatory checking.
Definition: Error.h:160
static ErrorSuccess success()
Create a success value.
Definition: Error.h:334
Target - Wrapper for Target specific information.
const char * getName() const
getName - Get the target name.
This is an optimization pass for GlobalISel generic memory operations.
Definition: AddressRanges.h:18
auto formatv(const char *Fmt, Ts &&... Vals) -> formatv_object< decltype(std::make_tuple(detail::build_format_adapter(std::forward< Ts >(Vals))...))>
raw_ostream & dbgs()
dbgs() - This returns a reference to a raw_ostream for debugging messages.
Definition: Debug.cpp:163