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
111