lib/Target/X86/MCTargetDesc/X86WinCOFFStreamer.cpp

   1 //===-- X86WinCOFFStreamer.cpp - X86 Target WinCOFF Streamer ----*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #include "X86MCTargetDesc.h"
  11 #include "llvm/MC/MCWinCOFFStreamer.h"
  12
  13 using namespace llvm;
  14
  15 namespace {
  16 class X86WinCOFFStreamer : public MCWinCOFFStreamer {
  17 public:
  18   X86WinCOFFStreamer(MCContext &C, MCAsmBackend &AB, MCCodeEmitter *CE,
  19                      raw_ostream &OS)
  20     : MCWinCOFFStreamer(C, AB, *CE, OS) { }
  21
  22   void EmitWinEHHandlerData() override;
  23   void EmitWindowsUnwindTables() override;
  24   void FinishImpl() override;
  25 };
  26
  27 void X86WinCOFFStreamer::EmitWinEHHandlerData() {
  28   MCStreamer::EmitWinEHHandlerData();
  29
  30   // We have to emit the unwind info now, because this directive
  31   // actually switches to the .xdata section!
  32   MCWin64EHUnwindEmitter::EmitUnwindInfo(*this, getCurrentWinFrameInfo());
  33 }
  34
  35 void X86WinCOFFStreamer::EmitWindowsUnwindTables() {
  36   if (!getNumWinFrameInfos())
  37     return;
  38   MCWin64EHUnwindEmitter::Emit(*this);
  39 }
  40
  41 void X86WinCOFFStreamer::FinishImpl() {
  42   EmitFrames(nullptr);
  43   EmitWindowsUnwindTables();
  44
  45   MCWinCOFFStreamer::FinishImpl();
  46 }
  47 }
  48
  49 namespace llvm {
  50 MCStreamer *createX86WinCOFFStreamer(MCContext &C, MCAsmBackend &AB,
  51                                      MCCodeEmitter *CE, raw_ostream &OS,
  52                                      bool RelaxAll) {
  53   X86WinCOFFStreamer *S = new X86WinCOFFStreamer(C, AB, CE, OS);
  54   S->getAssembler().setRelaxAll(RelaxAll);
  55   return S;
  56 }
  57 }
  58