LLVM 19.0.0git
MCWinCOFFStreamer.cpp
Go to the documentation of this file.
1//===- llvm/MC/MCWinCOFFStreamer.cpp --------------------------------------===//
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// This file contains an implementation of a Windows COFF object file streamer.
10//
11//===----------------------------------------------------------------------===//
12
16#include "llvm/ADT/Twine.h"
19#include "llvm/MC/MCAssembler.h"
21#include "llvm/MC/MCContext.h"
22#include "llvm/MC/MCExpr.h"
23#include "llvm/MC/MCFixup.h"
24#include "llvm/MC/MCFragment.h"
33#include "llvm/Support/SMLoc.h"
36#include <algorithm>
37#include <cstdint>
38
39using namespace llvm;
40
41#define DEBUG_TYPE "WinCOFFStreamer"
42
44 std::unique_ptr<MCAsmBackend> MAB,
45 std::unique_ptr<MCCodeEmitter> CE,
46 std::unique_ptr<MCObjectWriter> OW)
47 : MCObjectStreamer(Context, std::move(MAB), std::move(OW), std::move(CE)),
48 CurSymbol(nullptr) {}
49
51 const MCSubtargetInfo &STI) {
53
56 getAssembler().getEmitter().encodeInstruction(Inst, Code, Fixups, STI);
57
58 // Add the fixups and data.
59 for (unsigned i = 0, e = Fixups.size(); i != e; ++i) {
60 Fixups[i].setOffset(Fixups[i].getOffset() + DF->getContents().size());
61 DF->getFixups().push_back(Fixups[i]);
62 }
63 DF->setHasInstructions(STI);
64 DF->getContents().append(Code.begin(), Code.end());
65}
66
67void MCWinCOFFStreamer::initSections(bool NoExecStack,
68 const MCSubtargetInfo &STI) {
69 // FIXME: this is identical to the ELF one.
70 // This emulates the same behavior of GNU as. This makes it easier
71 // to compare the output as the major sections are in the same order.
72 switchSection(getContext().getObjectFileInfo()->getTextSection());
73 emitCodeAlignment(Align(4), &STI);
74
75 switchSection(getContext().getObjectFileInfo()->getDataSection());
76 emitCodeAlignment(Align(4), &STI);
77
78 switchSection(getContext().getObjectFileInfo()->getBSSSection());
79 emitCodeAlignment(Align(4), &STI);
80
81 switchSection(getContext().getObjectFileInfo()->getTextSection());
82}
83
85 changeSectionImpl(Section, Subsection);
86 // Ensure that the first and the second symbols relative to the section are
87 // the section symbol and the COMDAT symbol.
88 getAssembler().registerSymbol(*Section->getBeginSymbol());
89 if (auto *Sym = cast<MCSectionCOFF>(Section)->getCOMDATSymbol())
91}
92
94 auto *Symbol = cast<MCSymbolCOFF>(S);
95 MCObjectStreamer::emitLabel(Symbol, Loc);
96}
97
99 // Let the target do whatever target specific stuff it needs to do.
101
102 switch (Flag) {
103 // None of these require COFF specific handling.
105 case MCAF_Code16:
106 case MCAF_Code32:
107 case MCAF_Code64:
108 break;
110 llvm_unreachable("COFF doesn't support .subsections_via_symbols");
111 }
112}
113
115 llvm_unreachable("not implemented");
116}
117
120 auto *Symbol = cast<MCSymbolCOFF>(S);
121 getAssembler().registerSymbol(*Symbol);
122
123 switch (Attribute) {
124 default: return false;
126 case MCSA_Weak:
127 Symbol->setWeakExternalCharacteristics(COFF::IMAGE_WEAK_EXTERN_SEARCH_ALIAS);
128 Symbol->setExternal(true);
129 break;
130 case MCSA_WeakAntiDep:
131 Symbol->setWeakExternalCharacteristics(COFF::IMAGE_WEAK_EXTERN_ANTI_DEPENDENCY);
132 Symbol->setExternal(true);
133 Symbol->setIsWeakExternal(true);
134 break;
135 case MCSA_Global:
136 Symbol->setExternal(true);
137 break;
138 case MCSA_AltEntry:
139 llvm_unreachable("COFF doesn't support the .alt_entry attribute");
140 }
141
142 return true;
143}
144
145void MCWinCOFFStreamer::emitSymbolDesc(MCSymbol *Symbol, unsigned DescValue) {
146 llvm_unreachable("not implemented");
147}
148
150 auto *Symbol = cast<MCSymbolCOFF>(S);
151 if (CurSymbol)
152 Error("starting a new symbol definition without completing the "
153 "previous one");
154 CurSymbol = Symbol;
155}
156
158 if (!CurSymbol) {
159 Error("storage class specified outside of symbol definition");
160 return;
161 }
162
164 Error("storage class value '" + Twine(StorageClass) +
165 "' out of range");
166 return;
167 }
168
170 cast<MCSymbolCOFF>(CurSymbol)->setClass((uint16_t)StorageClass);
171}
172
174 if (!CurSymbol) {
175 Error("symbol type specified outside of a symbol definition");
176 return;
177 }
178
179 if (Type & ~0xffff) {
180 Error("type value '" + Twine(Type) + "' out of range");
181 return;
182 }
183
185 cast<MCSymbolCOFF>(CurSymbol)->setType((uint16_t)Type);
186}
187
189 if (!CurSymbol)
190 Error("ending symbol definition without starting one");
191 CurSymbol = nullptr;
192}
193
195 // SafeSEH is a feature specific to 32-bit x86. It does not exist (and is
196 // unnecessary) on all platforms which use table-based exception dispatch.
197 if (getContext().getTargetTriple().getArch() != Triple::x86)
198 return;
199
200 const MCSymbolCOFF *CSymbol = cast<MCSymbolCOFF>(Symbol);
201 if (CSymbol->isSafeSEH())
202 return;
203
205 changeSection(SXData);
206 SXData->ensureMinAlignment(Align(4));
207
208 insert(getContext().allocFragment<MCSymbolIdFragment>(Symbol));
209 getAssembler().registerSymbol(*Symbol);
210 CSymbol->setIsSafeSEH();
211
212 // The Microsoft linker requires that the symbol type of a handler be
213 // function. Go ahead and oblige it here.
216}
217
220 Sec->ensureMinAlignment(Align(4));
221
222 insert(getContext().allocFragment<MCSymbolIdFragment>(Symbol));
223 getAssembler().registerSymbol(*Symbol);
224}
225
227 visitUsedSymbol(*Symbol);
229 const MCSymbolRefExpr *SRE = MCSymbolRefExpr::create(Symbol, getContext());
230 MCFixup Fixup = MCFixup::create(DF->getContents().size(), SRE, FK_SecRel_2);
231 DF->getFixups().push_back(Fixup);
232 DF->getContents().resize(DF->getContents().size() + 2, 0);
233}
234
237 visitUsedSymbol(*Symbol);
239 // Create Symbol A for the relocation relative reference.
240 const MCExpr *MCE = MCSymbolRefExpr::create(Symbol, getContext());
241 // Add the constant offset, if given.
242 if (Offset)
245 // Build the secrel32 relocation.
246 MCFixup Fixup = MCFixup::create(DF->getContents().size(), MCE, FK_SecRel_4);
247 // Record the relocation.
248 DF->getFixups().push_back(Fixup);
249 // Emit 4 bytes (zeros) to the object file.
250 DF->getContents().resize(DF->getContents().size() + 4, 0);
251}
252
254 int64_t Offset) {
255 visitUsedSymbol(*Symbol);
257 // Create Symbol A for the relocation relative reference.
258 const MCExpr *MCE = MCSymbolRefExpr::create(
260 // Add the constant offset, if given.
261 if (Offset)
264 // Build the imgrel relocation.
265 MCFixup Fixup = MCFixup::create(DF->getContents().size(), MCE, FK_Data_4);
266 // Record the relocation.
267 DF->getFixups().push_back(Fixup);
268 // Emit 4 bytes (zeros) to the object file.
269 DF->getContents().resize(DF->getContents().size() + 4, 0);
270}
271
273 Align ByteAlignment) {
274 auto *Symbol = cast<MCSymbolCOFF>(S);
275
276 const Triple &T = getContext().getTargetTriple();
277 if (T.isWindowsMSVCEnvironment()) {
278 if (ByteAlignment > 32)
279 report_fatal_error("alignment is limited to 32-bytes");
280
281 // Round size up to alignment so that we will honor the alignment request.
282 Size = std::max(Size, ByteAlignment.value());
283 }
284
285 getAssembler().registerSymbol(*Symbol);
286 Symbol->setExternal(true);
287 Symbol->setCommon(Size, ByteAlignment);
288
289 if (!T.isWindowsMSVCEnvironment() && ByteAlignment > 1) {
293
294 OS << " -aligncomm:\"" << Symbol->getName() << "\","
295 << Log2_32_Ceil(ByteAlignment.value());
296
297 pushSection();
300 popSection();
301 }
302}
303
305 Align ByteAlignment) {
306 auto *Symbol = cast<MCSymbolCOFF>(S);
307
309 pushSection();
310 switchSection(Section);
311 emitValueToAlignment(ByteAlignment, 0, 1, 0);
312 emitLabel(Symbol);
313 Symbol->setExternal(false);
315 popSection();
316}
317
319 const MCSymbol *Symbol) {
320 auto *Alias = cast<MCSymbolCOFF>(AliasS);
322
323 getAssembler().registerSymbol(*Symbol);
324 Alias->setVariableValue(MCSymbolRefExpr::create(
326}
327
329 uint64_t Size, Align ByteAlignment,
330 SMLoc Loc) {
331 llvm_unreachable("not implemented");
332}
333
335 uint64_t Size, Align ByteAlignment) {
336 llvm_unreachable("not implemented");
337}
338
339// TODO: Implement this if you want to emit .comment section in COFF obj files.
341 llvm_unreachable("not implemented");
342}
343
345 llvm_unreachable("not implemented");
346}
347
349 const MCSymbolRefExpr *To,
350 uint64_t Count) {
351 // Ignore temporary symbols for now.
352 if (!From->getSymbol().isTemporary() && !To->getSymbol().isTemporary())
353 getAssembler().CGProfile.push_back({From, To, Count});
354}
355
357 const MCSymbol *S = &SRE->getSymbol();
358 if (getAssembler().registerSymbol(*S))
359 cast<MCSymbolCOFF>(S)->setExternal(true);
360}
361
363 MCAssembler &Asm = getAssembler();
364 if (Asm.getWriter().getEmitAddrsigSection()) {
365 // Register the section.
366 switchSection(Asm.getContext().getCOFFSection(".llvm_addrsig",
368 }
369 if (!Asm.CGProfile.empty()) {
370 for (MCAssembler::CGProfileEntry &E : Asm.CGProfile) {
373 }
374 switchSection(Asm.getContext().getCOFFSection(".llvm.call-graph-profile",
376 }
377
379}
380
381void MCWinCOFFStreamer::Error(const Twine &Msg) const {
382 getContext().reportError(SMLoc(), Msg);
383}
BlockVerifier::State From
static RegisterPass< DebugifyFunctionPass > DF("debugify-function", "Attach debug info to a function")
uint64_t Size
Symbol * Sym
Definition: ELF_riscv.cpp:479
PowerPC TLS Dynamic Call Fixup
raw_pwrite_stream & OS
This file defines the SmallString class.
This file defines the SmallVector class.
Lightweight error class with error context and mandatory checking.
Definition: Error.h:160
virtual void handleAssemblerFlag(MCAssemblerFlag Flag)
Handle any target-specific assembler flags. By default, do nothing.
Definition: MCAsmBackend.h:224
MCCodeEmitter & getEmitter() const
Definition: MCAssembler.h:287
MCAsmBackend & getBackend() const
Definition: MCAssembler.h:285
std::vector< CGProfileEntry > CGProfile
Definition: MCAssembler.h:357
bool registerSymbol(const MCSymbol &Symbol)
static const MCBinaryExpr * createAdd(const MCExpr *LHS, const MCExpr *RHS, MCContext &Ctx)
Definition: MCExpr.h:532
virtual void encodeInstruction(const MCInst &Inst, SmallVectorImpl< char > &CB, SmallVectorImpl< MCFixup > &Fixups, const MCSubtargetInfo &STI) const =0
Encode the given Inst to bytes and append to CB.
static const MCConstantExpr * create(int64_t Value, MCContext &Ctx, bool PrintInHex=false, unsigned SizeInBytes=0)
Definition: MCExpr.cpp:193
Context object for machine code objects.
Definition: MCContext.h:83
const MCObjectFileInfo * getObjectFileInfo() const
Definition: MCContext.h:416
void reportError(SMLoc L, const Twine &Msg)
Definition: MCContext.cpp:1069
const Triple & getTargetTriple() const
Definition: MCContext.h:400
Fragment for data and encoded instructions.
Definition: MCFragment.h:231
Base class for the full range of assembler expressions which are needed for parsing.
Definition: MCExpr.h:34
Encode information on a single operation to perform on a byte sequence (e.g., an encoded instruction)...
Definition: MCFixup.h:71
static MCFixup create(uint32_t Offset, const MCExpr *Value, MCFixupKind Kind, SMLoc Loc=SMLoc())
Definition: MCFixup.h:87
Instances of this class represent a single low-level machine instruction.
Definition: MCInst.h:184
MCSection * getSXDataSection() const
MCSection * getBSSSection() const
MCSection * getDrectveSection() const
Streaming object file generation interface.
void emitValueToAlignment(Align Alignment, int64_t Value=0, unsigned ValueSize=1, unsigned MaxBytesToEmit=0) override
Emit some number of copies of Value until the byte alignment ByteAlignment is reached.
MCDataFragment * getOrCreateDataFragment(const MCSubtargetInfo *STI=nullptr)
Get a data fragment to write into, creating a new one if the current fragment is not a data fragment.
MCAssembler & getAssembler()
void emitBytes(StringRef Data) override
Emit the bytes in Data into the output.
bool changeSectionImpl(MCSection *Section, uint32_t Subsection)
void visitUsedSymbol(const MCSymbol &Sym) override
void insert(MCFragment *F)
void emitLabel(MCSymbol *Symbol, SMLoc Loc=SMLoc()) override
Emit a label for Symbol into the current section.
void finishImpl() override
Streamer specific finalization.
void emitCodeAlignment(Align ByteAlignment, const MCSubtargetInfo *STI, unsigned MaxBytesToEmit=0) override
Emit nops until the byte alignment ByteAlignment is reached.
Instances of this class represent a uniqued identifier for a section in the current translation unit.
Definition: MCSection.h:36
void ensureMinAlignment(Align MinAlignment)
Makes sure that Alignment is at least MinAlignment.
Definition: MCSection.h:150
bool popSection()
Restore the current and previous section from the section stack.
MCContext & getContext() const
Definition: MCStreamer.h:300
void pushSection()
Save the current and previous section on the section stack.
Definition: MCStreamer.h:416
virtual void switchSection(MCSection *Section, uint32_t Subsec=0)
Set the current section where code is being emitted to Section.
MCSection * getCurrentSectionOnly() const
Definition: MCStreamer.h:398
void emitZeros(uint64_t NumBytes)
Emit NumBytes worth of zeros.
Definition: MCStreamer.cpp:229
Generic base class for all target subtargets.
void setIsSafeSEH() const
Definition: MCSymbolCOFF.h:65
void setType(uint16_t Ty) const
Definition: MCSymbolCOFF.h:39
bool isSafeSEH() const
Definition: MCSymbolCOFF.h:62
Represent a reference to a symbol from inside an expression.
Definition: MCExpr.h:188
const MCSymbol & getSymbol() const
Definition: MCExpr.h:406
static const MCSymbolRefExpr * create(const MCSymbol *Symbol, MCContext &Ctx)
Definition: MCExpr.h:393
MCSymbol - Instances of this class represent a symbol name in the MC file, and MCSymbols are created ...
Definition: MCSymbol.h:41
void setExternal(bool Value) const
Definition: MCSymbol.h:407
bool isTemporary() const
isTemporary - Check if this is an assembler temporary symbol.
Definition: MCSymbol.h:222
void emitCOFFSectionIndex(MCSymbol const *Symbol) override
Emits a COFF section index.
void emitAssemblerFlag(MCAssemblerFlag Flag) override
Note in the output the specified Flag.
void finalizeCGProfileEntry(const MCSymbolRefExpr *&S)
void emitThumbFunc(MCSymbol *Func) override
Note in the output that the specified Func is a Thumb mode function (ARM target only).
void emitCOFFImgRel32(MCSymbol const *Symbol, int64_t Offset) override
Emits a COFF image relative relocation.
void emitLabel(MCSymbol *Symbol, SMLoc Loc=SMLoc()) override
Emit a label for Symbol into the current section.
void emitZerofill(MCSection *Section, MCSymbol *Symbol, uint64_t Size, Align ByteAlignment, SMLoc Loc=SMLoc()) override
Emit the zerofill section and an optional symbol.
void emitWeakReference(MCSymbol *Alias, const MCSymbol *Symbol) override
Emit an weak reference from Alias to Symbol.
void initSections(bool NoExecStack, const MCSubtargetInfo &STI) override
Create the default sections and set the initial one.
void emitCOFFSafeSEH(MCSymbol const *Symbol) override
void emitCommonSymbol(MCSymbol *Symbol, uint64_t Size, Align ByteAlignment) override
Emit a common symbol.
void endCOFFSymbolDef() override
Marks the end of the symbol definition.
MCWinCOFFStreamer(MCContext &Context, std::unique_ptr< MCAsmBackend > MAB, std::unique_ptr< MCCodeEmitter > CE, std::unique_ptr< MCObjectWriter > OW)
void emitTBSSSymbol(MCSection *Section, MCSymbol *Symbol, uint64_t Size, Align ByteAlignment) override
Emit a thread local bss (.tbss) symbol.
void changeSection(MCSection *Section, uint32_t Subsection=0) override
This is called by popSection and switchSection, if the current section changes.
void emitCOFFSecRel32(MCSymbol const *Symbol, uint64_t Offset) override
Emits a COFF section relative relocation.
void finishImpl() override
Streamer specific finalization.
void emitIdent(StringRef IdentString) override
Emit the "identifiers" directive.
void emitCGProfileEntry(const MCSymbolRefExpr *From, const MCSymbolRefExpr *To, uint64_t Count) override
void emitLocalCommonSymbol(MCSymbol *Symbol, uint64_t Size, Align ByteAlignment) override
Emit a local common (.lcomm) symbol.
void emitCOFFSymbolType(int Type) override
Emit the type of the symbol.
void emitWinEHHandlerData(SMLoc Loc) override
void emitSymbolDesc(MCSymbol *Symbol, unsigned DescValue) override
Set the DescValue for the Symbol.
const MCSymbol * CurSymbol
bool emitSymbolAttribute(MCSymbol *Symbol, MCSymbolAttr Attribute) override
Add the given Attribute to Symbol.
void emitCOFFSymbolIndex(MCSymbol const *Symbol) override
Emits the symbol table index of a Symbol into the current section.
void beginCOFFSymbolDef(MCSymbol const *Symbol) override
Start emitting COFF symbol definition.
void emitInstToData(const MCInst &Inst, const MCSubtargetInfo &STI) override
void emitCOFFSymbolStorageClass(int StorageClass) override
Emit the storage class of the symbol.
Represents a location in source code.
Definition: SMLoc.h:23
SmallString - A SmallString is just a SmallVector with methods and accessors that make it work better...
Definition: SmallString.h:26
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
Triple - Helper class for working with autoconf configuration names.
Definition: Triple.h:44
Twine - A lightweight data structure for efficiently representing the concatenation of temporary valu...
Definition: Twine.h:81
The instances of the Type class are immutable: once they are created, they are never changed.
Definition: Type.h:45
A raw_ostream that writes to an SmallVector or SmallString.
Definition: raw_ostream.h:691
#define llvm_unreachable(msg)
Marks that the current location is not supposed to be reachable.
@ IMAGE_SCN_LNK_REMOVE
Definition: COFF.h:307
@ SSC_Invalid
Definition: COFF.h:218
@ IMAGE_WEAK_EXTERN_SEARCH_ALIAS
Definition: COFF.h:456
@ IMAGE_WEAK_EXTERN_ANTI_DEPENDENCY
Definition: COFF.h:457
@ IMAGE_SYM_DTYPE_FUNCTION
A function that returns a base type.
Definition: COFF.h:275
@ SCT_COMPLEX_TYPE_SHIFT
Type is formed as (base + (derived << SCT_COMPLEX_TYPE_SHIFT))
Definition: COFF.h:279
StorageClass
Definition: XCOFF.h:170
This is an optimization pass for GlobalISel generic memory operations.
Definition: AddressRanges.h:18
unsigned Log2_32_Ceil(uint32_t Value)
Return the ceil log base 2 of the specified value, 32 if the value is zero.
Definition: MathExtras.h:353
@ Offset
Definition: DWP.cpp:480
static Error getOffset(const SymbolRef &Sym, SectionRef Sec, uint64_t &Result)
void report_fatal_error(Error Err, bool gen_crash_diag=true)
Report a serious error, calling any installed error handler.
Definition: Error.cpp:167
@ FK_SecRel_2
A two-byte section relative fixup.
Definition: MCFixup.h:41
@ FK_Data_4
A four-byte fixup.
Definition: MCFixup.h:25
@ FK_SecRel_4
A four-byte section relative fixup.
Definition: MCFixup.h:42
MCAssemblerFlag
Definition: MCDirectives.h:53
@ MCAF_SyntaxUnified
.syntax (ARM/ELF)
Definition: MCDirectives.h:54
@ MCAF_Code64
.code64 (X86)
Definition: MCDirectives.h:58
@ MCAF_Code16
.code16 (X86) / .code 16 (ARM)
Definition: MCDirectives.h:56
@ MCAF_Code32
.code32 (X86) / .code 32 (ARM)
Definition: MCDirectives.h:57
@ MCAF_SubsectionsViaSymbols
.subsections_via_symbols (MachO)
Definition: MCDirectives.h:55
OutputIt move(R &&Range, OutputIt Out)
Provide wrappers to std::move which take ranges instead of having to pass begin/end explicitly.
Definition: STLExtras.h:1849
MCSymbolAttr
Definition: MCDirectives.h:18
@ MCSA_WeakReference
.weak_reference (MachO)
Definition: MCDirectives.h:47
@ MCSA_AltEntry
.alt_entry (MachO)
Definition: MCDirectives.h:41
@ MCSA_Weak
.weak
Definition: MCDirectives.h:45
@ MCSA_Global
.type _foo, @gnu_unique_object
Definition: MCDirectives.h:30
@ MCSA_WeakAntiDep
.weak_anti_dep (COFF)
Definition: MCDirectives.h:49
Implement std::hash so that hash_code can be used in STL containers.
Definition: BitVector.h:858
This struct is a compact representation of a valid (non-zero power of two) alignment.
Definition: Alignment.h:39
const MCSymbolRefExpr * From
Definition: MCAssembler.h:353
const MCSymbolRefExpr * To
Definition: MCAssembler.h:354