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