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
14namespace {
15class AArch64ELFStreamer;
16}
17
18namespace llvm {
19
20class AArch64TargetStreamer : public MCTargetStreamer {
21public:
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
57private:
58  std::unique_ptr<AssemblerConstantPools> ConstantPools;
59};
60
61class AArch64TargetELFStreamer : public AArch64TargetStreamer {
62private:
63  AArch64ELFStreamer &getStreamer();
64
65  void emitInst(uint32_t Inst) override;
66
67public:
68  AArch64TargetELFStreamer(MCStreamer &S) : AArch64TargetStreamer(S) {}
69};
70
71class AArch64TargetWinCOFFStreamer : public llvm::AArch64TargetStreamer {
72private:
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;
78public:
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;
101private:
102  void EmitARM64WinUnwindCode(unsigned UnwindCode, int Reg, int Offset);
103};
104
105MCTargetStreamer *
106createAArch64ObjectTargetStreamer(MCStreamer &S, const MCSubtargetInfo &STI);
107
108} // end namespace llvm
109
110#endif
111