aboutsummaryrefslogtreecommitdiff
path: root/contrib/llvm-project/llvm/lib/Target/AArch64/MCTargetDesc/AArch64TargetStreamer.h
blob: 1af978a806d1230e7e89e51e69f0a5d880b57f41 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
//===-- AArch64TargetStreamer.h - AArch64 Target Streamer ------*- C++ -*--===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_LIB_TARGET_AARCH64_MCTARGETDESC_AARCH64TARGETSTREAMER_H
#define LLVM_LIB_TARGET_AARCH64_MCTARGETDESC_AARCH64TARGETSTREAMER_H

#include "llvm/MC/MCStreamer.h"

namespace {
class AArch64ELFStreamer;
}

namespace llvm {

class AArch64TargetStreamer : public MCTargetStreamer {
public:
  AArch64TargetStreamer(MCStreamer &S);
  ~AArch64TargetStreamer() override;

  void finish() override;

  /// Callback used to implement the ldr= pseudo.
  /// Add a new entry to the constant pool for the current section and return an
  /// MCExpr that can be used to refer to the constant pool location.
  const MCExpr *addConstantPoolEntry(const MCExpr *, unsigned Size, SMLoc Loc);

  /// Callback used to implemnt the .ltorg directive.
  /// Emit contents of constant pool for the current section.
  void emitCurrentConstantPool();

  /// Callback used to implement the .inst directive.
  virtual void emitInst(uint32_t Inst);

  /// Callback used to implement the .variant_pcs directive.
  virtual void emitDirectiveVariantPCS(MCSymbol *Symbol) {};

  virtual void EmitARM64WinCFIAllocStack(unsigned Size) {}
  virtual void EmitARM64WinCFISaveFPLR(int Offset) {}
  virtual void EmitARM64WinCFISaveFPLRX(int Offset) {}
  virtual void EmitARM64WinCFISaveReg(unsigned Reg, int Offset) {}
  virtual void EmitARM64WinCFISaveRegX(unsigned Reg, int Offset) {}
  virtual void EmitARM64WinCFISaveRegP(unsigned Reg, int Offset) {}
  virtual void EmitARM64WinCFISaveRegPX(unsigned Reg, int Offset) {}
  virtual void EmitARM64WinCFISaveFReg(unsigned Reg, int Offset) {}
  virtual void EmitARM64WinCFISaveFRegX(unsigned Reg, int Offset) {}
  virtual void EmitARM64WinCFISaveFRegP(unsigned Reg, int Offset) {}
  virtual void EmitARM64WinCFISaveFRegPX(unsigned Reg, int Offset) {}
  virtual void EmitARM64WinCFISetFP() {}
  virtual void EmitARM64WinCFIAddFP(unsigned Size) {}
  virtual void EmitARM64WinCFINop() {}
  virtual void EmitARM64WinCFIPrologEnd() {}
  virtual void EmitARM64WinCFIEpilogStart() {}
  virtual void EmitARM64WinCFIEpilogEnd() {}

private:
  std::unique_ptr<AssemblerConstantPools> ConstantPools;
};

class AArch64TargetELFStreamer : public AArch64TargetStreamer {
private:
  AArch64ELFStreamer &getStreamer();

  void emitInst(uint32_t Inst) override;
  void emitDirectiveVariantPCS(MCSymbol *Symbol) override;

public:
  AArch64TargetELFStreamer(MCStreamer &S) : AArch64TargetStreamer(S) {}
};

class AArch64TargetWinCOFFStreamer : public llvm::AArch64TargetStreamer {
private:
  // True if we are processing SEH directives in an epilogue.
  bool InEpilogCFI = false;

  // Symbol of the current epilog for which we are processing SEH directives.
  MCSymbol *CurrentEpilog = nullptr;
public:
  AArch64TargetWinCOFFStreamer(llvm::MCStreamer &S)
    : AArch64TargetStreamer(S) {}

  // The unwind codes on ARM64 Windows are documented at
  // https://docs.microsoft.com/en-us/cpp/build/arm64-exception-handling
  void EmitARM64WinCFIAllocStack(unsigned Size) override;
  void EmitARM64WinCFISaveFPLR(int Offset) override;
  void EmitARM64WinCFISaveFPLRX(int Offset) override;
  void EmitARM64WinCFISaveReg(unsigned Reg, int Offset) override;
  void EmitARM64WinCFISaveRegX(unsigned Reg, int Offset) override;
  void EmitARM64WinCFISaveRegP(unsigned Reg, int Offset) override;
  void EmitARM64WinCFISaveRegPX(unsigned Reg, int Offset) override;
  void EmitARM64WinCFISaveFReg(unsigned Reg, int Offset) override;
  void EmitARM64WinCFISaveFRegX(unsigned Reg, int Offset) override;
  void EmitARM64WinCFISaveFRegP(unsigned Reg, int Offset) override;
  void EmitARM64WinCFISaveFRegPX(unsigned Reg, int Offset) override;
  void EmitARM64WinCFISetFP() override;
  void EmitARM64WinCFIAddFP(unsigned Size) override;
  void EmitARM64WinCFINop() override;
  void EmitARM64WinCFIPrologEnd() override;
  void EmitARM64WinCFIEpilogStart() override;
  void EmitARM64WinCFIEpilogEnd() override;
private:
  void EmitARM64WinUnwindCode(unsigned UnwindCode, int Reg, int Offset);
};

MCTargetStreamer *
createAArch64ObjectTargetStreamer(MCStreamer &S, const MCSubtargetInfo &STI);

} // end namespace llvm

#endif