Files
clang-p2996/llvm/lib/Target/AArch64/MCTargetDesc/AArch64TargetStreamer.h
Arthur Eubanks ad25344620 [MC][CodeGen] Emit constant pools earlier
Previously we would emit constant pool entries for ldr inline asm at the
very end of AsmPrinter::doFinalization(). However, if we're emitting
dwarf aranges, that would end all sections with aranges. Then if we have
constant pool entries to be emitted in those same sections, we'd hit an
assert that the section has already been ended.

We want to emit constant pool entries before emitting dwarf aranges.
This patch splits out arm32/64's constant pool entry emission into its
own MCTargetStreamer virtual method.

Fixes PR51208

Reviewed By: MaskRay

Differential Revision: https://reviews.llvm.org/D107314
2021-08-03 20:55:31 -07:00

134 lines
5.2 KiB
C++

//===-- 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;
void emitConstantPools() 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 .note.gnu.property section.
void emitNoteSection(unsigned Flags);
/// 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 emitARM64WinCFISaveR19R20X(int Offset) {}
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 emitARM64WinCFISaveLRPair(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 emitARM64WinCFISaveNext() {}
virtual void emitARM64WinCFIPrologEnd() {}
virtual void emitARM64WinCFIEpilogStart() {}
virtual void emitARM64WinCFIEpilogEnd() {}
virtual void emitARM64WinCFITrapFrame() {}
virtual void emitARM64WinCFIMachineFrame() {}
virtual void emitARM64WinCFIContext() {}
virtual void emitARM64WinCFIClearUnwoundToCall() {}
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 emitARM64WinCFISaveR19R20X(int Offset) 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 emitARM64WinCFISaveLRPair(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 emitARM64WinCFISaveNext() override;
void emitARM64WinCFIPrologEnd() override;
void emitARM64WinCFIEpilogStart() override;
void emitARM64WinCFIEpilogEnd() override;
void emitARM64WinCFITrapFrame() override;
void emitARM64WinCFIMachineFrame() override;
void emitARM64WinCFIContext() override;
void emitARM64WinCFIClearUnwoundToCall() override;
private:
void emitARM64WinUnwindCode(unsigned UnwindCode, int Reg, int Offset);
};
MCTargetStreamer *
createAArch64ObjectTargetStreamer(MCStreamer &S, const MCSubtargetInfo &STI);
} // end namespace llvm
#endif