lib/Target/AArch64/MCTargetDesc/AArch64TargetStreamer.h

   1 //===-- AArch64TargetStreamer.h - AArch64 Target 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 #ifndef LLVM_LIB_TARGET_AARCH64_MCTARGETDESC_AARCH64TARGETSTREAMER_H
  10 #define LLVM_LIB_TARGET_AARCH64_MCTARGETDESC_AARCH64TARGETSTREAMER_H
  11
  12 #include "llvm/MC/MCStreamer.h"
  13
  14 namespace {
  15 class AArch64ELFStreamer;
  16 }
  17
  18 namespace llvm {
  19
  20 class AArch64TargetStreamer : public MCTargetStreamer {
  21 public:
  22   AArch64TargetStreamer(MCStreamer &S);
  23   ~AArch64TargetStreamer() override;
  24
  25   void finish() override;
  26
  27   /// Callback used to implement the ldr= pseudo.
  28   /// Add a new entry to the constant pool for the current section and return an
  29   /// MCExpr that can be used to refer to the constant pool location.
  30   const MCExpr *addConstantPoolEntry(const MCExpr *, unsigned Size, SMLoc Loc);
  31
  32   /// Callback used to implemnt the .ltorg directive.
  33   /// Emit contents of constant pool for the current section.
  34   void emitCurrentConstantPool();
  35
  36   /// Callback used to implement the .inst directive.
  37   virtual void emitInst(uint32_t Inst);
  38
  39   virtual void EmitARM64WinCFIAllocStack(unsigned Size) {}
  40   virtual void EmitARM64WinCFISaveFPLR(int Offset) {}
  41   virtual void EmitARM64WinCFISaveFPLRX(int Offset) {}
  42   virtual void EmitARM64WinCFISaveReg(unsigned Reg, int Offset) {}
  43   virtual void EmitARM64WinCFISaveRegX(unsigned Reg, int Offset) {}
  44   virtual void EmitARM64WinCFISaveRegP(unsigned Reg, int Offset) {}
  45   virtual void EmitARM64WinCFISaveRegPX(unsigned Reg, int Offset) {}
  46   virtual void EmitARM64WinCFISaveFReg(unsigned Reg, int Offset) {}
  47   virtual void EmitARM64WinCFISaveFRegX(unsigned Reg, int Offset) {}
  48   virtual void EmitARM64WinCFISaveFRegP(unsigned Reg, int Offset) {}
  49   virtual void EmitARM64WinCFISaveFRegPX(unsigned Reg, int Offset) {}
  50   virtual void EmitARM64WinCFISetFP() {}
  51   virtual void EmitARM64WinCFIAddFP(unsigned Size) {}
  52   virtual void EmitARM64WinCFINop() {}
  53   virtual void EmitARM64WinCFIPrologEnd() {}
  54   virtual void EmitARM64WinCFIEpilogStart() {}
  55   virtual void EmitARM64WinCFIEpilogEnd() {}
  56
  57 private:
  58   std::unique_ptr<AssemblerConstantPools> ConstantPools;
  59 };
  60
  61 class AArch64TargetELFStreamer : public AArch64TargetStreamer {
  62 private:
  63   AArch64ELFStreamer &getStreamer();
  64
  65   void emitInst(uint32_t Inst) override;
  66
  67 public:
  68   AArch64TargetELFStreamer(MCStreamer &S) : AArch64TargetStreamer(S) {}
  69 };
  70
  71 class AArch64TargetWinCOFFStreamer : public llvm::AArch64TargetStreamer {
  72 private:
  73   // True if we are processing SEH directives in an epilogue.
  74   bool InEpilogCFI = false;
  75
  76   // Symbol of the current epilog for which we are processing SEH directives.
  77   MCSymbol *CurrentEpilog = nullptr;
  78 public:
  79   AArch64TargetWinCOFFStreamer(llvm::MCStreamer &S)
  80     : AArch64TargetStreamer(S) {}
  81
  82   // The unwind codes on ARM64 Windows are documented at
  83   // https://docs.microsoft.com/en-us/cpp/build/arm64-exception-handling
  84   void EmitARM64WinCFIAllocStack(unsigned Size) override;
  85   void EmitARM64WinCFISaveFPLR(int Offset) override;
  86   void EmitARM64WinCFISaveFPLRX(int Offset) override;
  87   void EmitARM64WinCFISaveReg(unsigned Reg, int Offset) override;
  88   void EmitARM64WinCFISaveRegX(unsigned Reg, int Offset) override;
  89   void EmitARM64WinCFISaveRegP(unsigned Reg, int Offset) override;
  90   void EmitARM64WinCFISaveRegPX(unsigned Reg, int Offset) override;
  91   void EmitARM64WinCFISaveFReg(unsigned Reg, int Offset) override;
  92   void EmitARM64WinCFISaveFRegX(unsigned Reg, int Offset) override;
  93   void EmitARM64WinCFISaveFRegP(unsigned Reg, int Offset) override;
  94   void EmitARM64WinCFISaveFRegPX(unsigned Reg, int Offset) override;
  95   void EmitARM64WinCFISetFP() override;
  96   void EmitARM64WinCFIAddFP(unsigned Size) override;
  97   void EmitARM64WinCFINop() override;
  98   void EmitARM64WinCFIPrologEnd() override;
  99   void EmitARM64WinCFIEpilogStart() override;
 100   void EmitARM64WinCFIEpilogEnd() override;
 101 private:
 102   void EmitARM64WinUnwindCode(unsigned UnwindCode, int Reg, int Offset);
 103 };
 104
 105 MCTargetStreamer *
 106 createAArch64ObjectTargetStreamer(MCStreamer &S, const MCSubtargetInfo &STI);
 107
 108 } // end namespace llvm
 109
 110 #endif