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 /// Callback used to implement the .variant_pcs directive. 40 virtual void emitDirectiveVariantPCS(MCSymbol *Symbol) {}; 41 42 virtual void EmitARM64WinCFIAllocStack(unsigned Size) {} 43 virtual void EmitARM64WinCFISaveFPLR(int Offset) {} 44 virtual void EmitARM64WinCFISaveFPLRX(int Offset) {} 45 virtual void EmitARM64WinCFISaveReg(unsigned Reg, int Offset) {} 46 virtual void EmitARM64WinCFISaveRegX(unsigned Reg, int Offset) {} 47 virtual void EmitARM64WinCFISaveRegP(unsigned Reg, int Offset) {} 48 virtual void EmitARM64WinCFISaveRegPX(unsigned Reg, int Offset) {} 49 virtual void EmitARM64WinCFISaveFReg(unsigned Reg, int Offset) {} 50 virtual void EmitARM64WinCFISaveFRegX(unsigned Reg, int Offset) {} 51 virtual void EmitARM64WinCFISaveFRegP(unsigned Reg, int Offset) {} 52 virtual void EmitARM64WinCFISaveFRegPX(unsigned Reg, int Offset) {} 53 virtual void EmitARM64WinCFISetFP() {} 54 virtual void EmitARM64WinCFIAddFP(unsigned Size) {} 55 virtual void EmitARM64WinCFINop() {} 56 virtual void EmitARM64WinCFIPrologEnd() {} 57 virtual void EmitARM64WinCFIEpilogStart() {} 58 virtual void EmitARM64WinCFIEpilogEnd() {} 59 60 private: 61 std::unique_ptr<AssemblerConstantPools> ConstantPools; 62 }; 63 64 class AArch64TargetELFStreamer : public AArch64TargetStreamer { 65 private: 66 AArch64ELFStreamer &getStreamer(); 67 68 void emitInst(uint32_t Inst) override; 69 void emitDirectiveVariantPCS(MCSymbol *Symbol) override; 70 71 public: 72 AArch64TargetELFStreamer(MCStreamer &S) : AArch64TargetStreamer(S) {} 73 }; 74 75 class AArch64TargetWinCOFFStreamer : public llvm::AArch64TargetStreamer { 76 private: 77 // True if we are processing SEH directives in an epilogue. 78 bool InEpilogCFI = false; 79 80 // Symbol of the current epilog for which we are processing SEH directives. 81 MCSymbol *CurrentEpilog = nullptr; 82 public: 83 AArch64TargetWinCOFFStreamer(llvm::MCStreamer &S) 84 : AArch64TargetStreamer(S) {} 85 86 // The unwind codes on ARM64 Windows are documented at 87 // https://docs.microsoft.com/en-us/cpp/build/arm64-exception-handling 88 void EmitARM64WinCFIAllocStack(unsigned Size) override; 89 void EmitARM64WinCFISaveFPLR(int Offset) override; 90 void EmitARM64WinCFISaveFPLRX(int Offset) override; 91 void EmitARM64WinCFISaveReg(unsigned Reg, int Offset) override; 92 void EmitARM64WinCFISaveRegX(unsigned Reg, int Offset) override; 93 void EmitARM64WinCFISaveRegP(unsigned Reg, int Offset) override; 94 void EmitARM64WinCFISaveRegPX(unsigned Reg, int Offset) override; 95 void EmitARM64WinCFISaveFReg(unsigned Reg, int Offset) override; 96 void EmitARM64WinCFISaveFRegX(unsigned Reg, int Offset) override; 97 void EmitARM64WinCFISaveFRegP(unsigned Reg, int Offset) override; 98 void EmitARM64WinCFISaveFRegPX(unsigned Reg, int Offset) override; 99 void EmitARM64WinCFISetFP() override; 100 void EmitARM64WinCFIAddFP(unsigned Size) override; 101 void EmitARM64WinCFINop() override; 102 void EmitARM64WinCFIPrologEnd() override; 103 void EmitARM64WinCFIEpilogStart() override; 104 void EmitARM64WinCFIEpilogEnd() override; 105 private: 106 void EmitARM64WinUnwindCode(unsigned UnwindCode, int Reg, int Offset); 107 }; 108 109 MCTargetStreamer * 110 createAArch64ObjectTargetStreamer(MCStreamer &S, const MCSubtargetInfo &STI); 111 112 } // end namespace llvm 113 114 #endif 115