1 //===-- X86WinCOFFStreamer.cpp - X86 Target WinCOFF Streamer ----*- 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 #include "X86MCTargetDesc.h"
10 #include "X86TargetStreamer.h"
11 #include "llvm/MC/MCAsmBackend.h"
12 #include "llvm/MC/MCAssembler.h"
13 #include "llvm/MC/MCCodeEmitter.h"
14 #include "llvm/MC/MCObjectWriter.h"
15 #include "llvm/MC/MCWin64EH.h"
16 #include "llvm/MC/MCWinCOFFStreamer.h"
17
18 using namespace llvm;
19
20 namespace {
21 class X86WinCOFFStreamer : public MCWinCOFFStreamer {
22 Win64EH::UnwindEmitter EHStreamer;
23 public:
X86WinCOFFStreamer(MCContext & C,std::unique_ptr<MCAsmBackend> AB,std::unique_ptr<MCCodeEmitter> CE,std::unique_ptr<MCObjectWriter> OW)24 X86WinCOFFStreamer(MCContext &C, std::unique_ptr<MCAsmBackend> AB,
25 std::unique_ptr<MCCodeEmitter> CE,
26 std::unique_ptr<MCObjectWriter> OW)
27 : MCWinCOFFStreamer(C, std::move(AB), std::move(CE), std::move(OW)) {}
28
29 void emitWinEHHandlerData(SMLoc Loc) override;
30 void emitWindowsUnwindTables(WinEH::FrameInfo *Frame) override;
31 void emitWindowsUnwindTables() override;
32 void emitCVFPOData(const MCSymbol *ProcSym, SMLoc Loc) override;
33 void finishImpl() override;
34 };
35
emitWinEHHandlerData(SMLoc Loc)36 void X86WinCOFFStreamer::emitWinEHHandlerData(SMLoc Loc) {
37 MCStreamer::emitWinEHHandlerData(Loc);
38
39 // We have to emit the unwind info now, because this directive
40 // actually switches to the .xdata section.
41 if (WinEH::FrameInfo *CurFrame = getCurrentWinFrameInfo())
42 EHStreamer.EmitUnwindInfo(*this, CurFrame, /* HandlerData = */ true);
43 }
44
emitWindowsUnwindTables(WinEH::FrameInfo * Frame)45 void X86WinCOFFStreamer::emitWindowsUnwindTables(WinEH::FrameInfo *Frame) {
46 EHStreamer.EmitUnwindInfo(*this, Frame, /* HandlerData = */ false);
47 }
48
emitWindowsUnwindTables()49 void X86WinCOFFStreamer::emitWindowsUnwindTables() {
50 if (!getNumWinFrameInfos())
51 return;
52 EHStreamer.Emit(*this);
53 }
54
emitCVFPOData(const MCSymbol * ProcSym,SMLoc Loc)55 void X86WinCOFFStreamer::emitCVFPOData(const MCSymbol *ProcSym, SMLoc Loc) {
56 X86TargetStreamer *XTS =
57 static_cast<X86TargetStreamer *>(getTargetStreamer());
58 XTS->emitFPOData(ProcSym, Loc);
59 }
60
finishImpl()61 void X86WinCOFFStreamer::finishImpl() {
62 emitFrames(nullptr);
63 emitWindowsUnwindTables();
64
65 MCWinCOFFStreamer::finishImpl();
66 }
67 } // namespace
68
createX86WinCOFFStreamer(MCContext & C,std::unique_ptr<MCAsmBackend> && AB,std::unique_ptr<MCObjectWriter> && OW,std::unique_ptr<MCCodeEmitter> && CE,bool RelaxAll,bool IncrementalLinkerCompatible)69 MCStreamer *llvm::createX86WinCOFFStreamer(MCContext &C,
70 std::unique_ptr<MCAsmBackend> &&AB,
71 std::unique_ptr<MCObjectWriter> &&OW,
72 std::unique_ptr<MCCodeEmitter> &&CE,
73 bool RelaxAll,
74 bool IncrementalLinkerCompatible) {
75 X86WinCOFFStreamer *S =
76 new X86WinCOFFStreamer(C, std::move(AB), std::move(CE), std::move(OW));
77 S->getAssembler().setRelaxAll(RelaxAll);
78 S->getAssembler().setIncrementalLinkerCompatible(IncrementalLinkerCompatible);
79 return S;
80 }
81
82