LLVM 19.0.0git
MCLinkerOptimizationHint.h
Go to the documentation of this file.
1//===- MCLinkerOptimizationHint.h - LOH interface ---------------*- C++ -*-===//
2//
3//
4// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
5// See https://llvm.org/LICENSE.txt for license information.
6// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
7//
8//===----------------------------------------------------------------------===//
9//
10// This file declares some helpers classes to handle Linker Optimization Hint
11// (LOH).
12//
13// FIXME: LOH interface supports only MachO format at the moment.
14//===----------------------------------------------------------------------===//
15
16#ifndef LLVM_MC_MCLINKEROPTIMIZATIONHINT_H
17#define LLVM_MC_MCLINKEROPTIMIZATIONHINT_H
18
20#include "llvm/ADT/StringRef.h"
22#include <cassert>
23#include <cstdint>
24
25namespace llvm {
26
27class MachObjectWriter;
28class MCAsmLayout;
29class MCSymbol;
30class raw_ostream;
31
32/// Linker Optimization Hint Type.
34 MCLOH_AdrpAdrp = 0x1u, ///< Adrp xY, _v1@PAGE -> Adrp xY, _v2@PAGE.
35 MCLOH_AdrpLdr = 0x2u, ///< Adrp _v@PAGE -> Ldr _v@PAGEOFF.
36 MCLOH_AdrpAddLdr = 0x3u, ///< Adrp _v@PAGE -> Add _v@PAGEOFF -> Ldr.
37 MCLOH_AdrpLdrGotLdr = 0x4u, ///< Adrp _v@GOTPAGE -> Ldr _v@GOTPAGEOFF -> Ldr.
38 MCLOH_AdrpAddStr = 0x5u, ///< Adrp _v@PAGE -> Add _v@PAGEOFF -> Str.
39 MCLOH_AdrpLdrGotStr = 0x6u, ///< Adrp _v@GOTPAGE -> Ldr _v@GOTPAGEOFF -> Str.
40 MCLOH_AdrpAdd = 0x7u, ///< Adrp _v@PAGE -> Add _v@PAGEOFF.
41 MCLOH_AdrpLdrGot = 0x8u ///< Adrp _v@GOTPAGE -> Ldr _v@GOTPAGEOFF.
42};
43
45 return StringRef(".loh");
46}
47
48static inline bool isValidMCLOHType(unsigned Kind) {
49 return Kind >= MCLOH_AdrpAdrp && Kind <= MCLOH_AdrpLdrGot;
50}
51
52static inline int MCLOHNameToId(StringRef Name) {
53#define MCLOHCaseNameToId(Name) .Case(#Name, MCLOH_ ## Name)
55 MCLOHCaseNameToId(AdrpAdrp)
56 MCLOHCaseNameToId(AdrpLdr)
57 MCLOHCaseNameToId(AdrpAddLdr)
58 MCLOHCaseNameToId(AdrpLdrGotLdr)
59 MCLOHCaseNameToId(AdrpAddStr)
60 MCLOHCaseNameToId(AdrpLdrGotStr)
61 MCLOHCaseNameToId(AdrpAdd)
62 MCLOHCaseNameToId(AdrpLdrGot)
63 .Default(-1);
64#undef MCLOHCaseNameToId
65}
66
67static inline StringRef MCLOHIdToName(MCLOHType Kind) {
68#define MCLOHCaseIdToName(Name) case MCLOH_ ## Name: return StringRef(#Name);
69 switch (Kind) {
70 MCLOHCaseIdToName(AdrpAdrp);
71 MCLOHCaseIdToName(AdrpLdr);
72 MCLOHCaseIdToName(AdrpAddLdr);
73 MCLOHCaseIdToName(AdrpLdrGotLdr);
74 MCLOHCaseIdToName(AdrpAddStr);
75 MCLOHCaseIdToName(AdrpLdrGotStr);
76 MCLOHCaseIdToName(AdrpAdd);
77 MCLOHCaseIdToName(AdrpLdrGot);
78 }
79 return StringRef();
80#undef MCLOHCaseIdToName
81}
82
83static inline int MCLOHIdToNbArgs(MCLOHType Kind) {
84 switch (Kind) {
85 // LOH with two arguments
86 case MCLOH_AdrpAdrp:
87 case MCLOH_AdrpLdr:
88 case MCLOH_AdrpAdd:
90 return 2;
91 // LOH with three arguments
96 return 3;
97 }
98 return -1;
99}
100
101/// Store Linker Optimization Hint information (LOH).
103 MCLOHType Kind;
104
105 /// Arguments of this directive. Order matters.
107
108 /// Emit this directive in \p OutStream using the information available
109 /// in the given \p ObjWriter and \p Layout to get the address of the
110 /// arguments within the object file.
111 void emit_impl(raw_ostream &OutStream, const MachObjectWriter &ObjWriter,
112 const MCAsmLayout &Layout) const;
113
114public:
116
118 : Kind(Kind), Args(Args.begin(), Args.end()) {
119 assert(isValidMCLOHType(Kind) && "Invalid LOH directive type!");
120 }
121
122 MCLOHType getKind() const { return Kind; }
123
124 const LOHArgs &getArgs() const { return Args; }
125
126 /// Emit this directive as:
127 /// <kind, numArgs, addr1, ..., addrN>
128 void emit(MachObjectWriter &ObjWriter, const MCAsmLayout &Layout) const;
129
130 /// Get the size in bytes of this directive if emitted in \p ObjWriter with
131 /// the given \p Layout.
132 uint64_t getEmitSize(const MachObjectWriter &ObjWriter,
133 const MCAsmLayout &Layout) const;
134};
135
137 /// Keep track of the emit size of all the LOHs.
138 mutable uint64_t EmitSize = 0;
139
140 /// Keep track of all LOH directives.
142
143public:
145
146 MCLOHContainer() = default;
147
148 /// Const accessor to the directives.
150 return Directives;
151 }
152
153 /// Add the directive of the given kind \p Kind with the given arguments
154 /// \p Args to the container.
156 Directives.push_back(MCLOHDirective(Kind, Args));
157 }
158
159 /// Get the size of the directives if emitted.
161 const MCAsmLayout &Layout) const {
162 if (!EmitSize) {
163 for (const MCLOHDirective &D : Directives)
164 EmitSize += D.getEmitSize(ObjWriter, Layout);
165 }
166 return EmitSize;
167 }
168
169 /// Emit all Linker Optimization Hint in one big table.
170 /// Each line of the table is emitted by LOHDirective::emit.
171 void emit(MachObjectWriter &ObjWriter, const MCAsmLayout &Layout) const {
172 for (const MCLOHDirective &D : Directives)
173 D.emit(ObjWriter, Layout);
174 }
175
176 void reset() {
177 Directives.clear();
178 EmitSize = 0;
179 }
180};
181
182// Add types for specialized template using MCSymbol.
185
186} // end namespace llvm
187
188#endif // LLVM_MC_MCLINKEROPTIMIZATIONHINT_H
static GCRegistry::Add< StatepointGC > D("statepoint-example", "an example strategy for statepoint")
dxil metadata emit
std::string Name
#define MCLOHCaseIdToName(Name)
#define MCLOHCaseNameToId(Name)
assert(ImpDefSCC.getReg()==AMDGPU::SCC &&ImpDefSCC.isDef())
This file defines the SmallVector class.
This file implements the StringSwitch template, which mimics a switch() statement whose cases are str...
Encapsulates the layout of an assembly file at a particular point in time.
Definition: MCAsmLayout.h:28
SmallVectorImpl< MCLOHDirective > LOHDirectives
uint64_t getEmitSize(const MachObjectWriter &ObjWriter, const MCAsmLayout &Layout) const
Get the size of the directives if emitted.
MCLOHContainer()=default
const LOHDirectives & getDirectives() const
Const accessor to the directives.
void addDirective(MCLOHType Kind, const MCLOHDirective::LOHArgs &Args)
Add the directive of the given kind Kind with the given arguments Args to the container.
void emit(MachObjectWriter &ObjWriter, const MCAsmLayout &Layout) const
Emit all Linker Optimization Hint in one big table.
Store Linker Optimization Hint information (LOH).
const LOHArgs & getArgs() const
MCLOHDirective(MCLOHType Kind, const LOHArgs &Args)
uint64_t getEmitSize(const MachObjectWriter &ObjWriter, const MCAsmLayout &Layout) const
Get the size in bytes of this directive if emitted in ObjWriter with the given Layout.
SmallVectorImpl< MCSymbol * > LOHArgs
This class consists of common code factored out of the SmallVector class to reduce code duplication b...
Definition: SmallVector.h:586
void push_back(const T &Elt)
Definition: SmallVector.h:426
This is a 'vector' (really, a variable-sized array), optimized for the case when the array is small.
Definition: SmallVector.h:1209
StringRef - Represent a constant reference to a string, i.e.
Definition: StringRef.h:50
A switch()-like statement whose cases are string literals.
Definition: StringSwitch.h:44
This class implements an extremely fast bulk output stream that can only output to a stream.
Definition: raw_ostream.h:52
This is an optimization pass for GlobalISel generic memory operations.
Definition: AddressRanges.h:18
static int MCLOHNameToId(StringRef Name)
static StringRef MCLOHDirectiveName()
static bool isValidMCLOHType(unsigned Kind)
static int MCLOHIdToNbArgs(MCLOHType Kind)
MCLOHType
Linker Optimization Hint Type.
@ MCLOH_AdrpAddLdr
Adrp _v@PAGE -> Add _v@PAGEOFF -> Ldr.
@ MCLOH_AdrpLdrGotStr
Adrp _v@GOTPAGE -> Ldr _v@GOTPAGEOFF -> Str.
@ MCLOH_AdrpLdrGotLdr
Adrp _v@GOTPAGE -> Ldr _v@GOTPAGEOFF -> Ldr.
@ MCLOH_AdrpLdrGot
Adrp _v@GOTPAGE -> Ldr _v@GOTPAGEOFF.
@ MCLOH_AdrpLdr
Adrp _v@PAGE -> Ldr _v@PAGEOFF.
@ MCLOH_AdrpAdd
Adrp _v@PAGE -> Add _v@PAGEOFF.
@ MCLOH_AdrpAddStr
Adrp _v@PAGE -> Add _v@PAGEOFF -> Str.
@ MCLOH_AdrpAdrp
Adrp xY, _v1@PAGE -> Adrp xY, _v2@PAGE.
static StringRef MCLOHIdToName(MCLOHType Kind)