131 lines
4.7 KiB
C++
131 lines
4.7 KiB
C++
//===-- SystemZAsmPrinter.h - SystemZ LLVM assembly printer ----*- 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_SYSTEMZ_SYSTEMZASMPRINTER_H
|
|
#define LLVM_LIB_TARGET_SYSTEMZ_SYSTEMZASMPRINTER_H
|
|
|
|
#include "SystemZMCInstLower.h"
|
|
#include "SystemZTargetMachine.h"
|
|
#include "SystemZTargetStreamer.h"
|
|
#include "llvm/CodeGen/AsmPrinter.h"
|
|
#include "llvm/CodeGen/StackMaps.h"
|
|
#include "llvm/MC/MCInstBuilder.h"
|
|
#include "llvm/Support/Compiler.h"
|
|
|
|
namespace llvm {
|
|
class MCStreamer;
|
|
class MachineInstr;
|
|
class Module;
|
|
class raw_ostream;
|
|
|
|
class LLVM_LIBRARY_VISIBILITY SystemZAsmPrinter : public AsmPrinter {
|
|
private:
|
|
MCSymbol *CurrentFnPPA1Sym; // PPA1 Symbol.
|
|
MCSymbol *CurrentFnEPMarkerSym; // Entry Point Marker.
|
|
MCSymbol *PPA2Sym;
|
|
|
|
SystemZTargetStreamer *getTargetStreamer() {
|
|
MCTargetStreamer *TS = OutStreamer->getTargetStreamer();
|
|
assert(TS && "do not have a target streamer");
|
|
return static_cast<SystemZTargetStreamer *>(TS);
|
|
}
|
|
|
|
/// Call type information for XPLINK.
|
|
enum class CallType {
|
|
BASR76 = 0, // b'x000' == BASR r7,r6
|
|
BRAS7 = 1, // b'x001' == BRAS r7,ep
|
|
RESVD_2 = 2, // b'x010'
|
|
BRASL7 = 3, // b'x011' == BRASL r7,ep
|
|
RESVD_4 = 4, // b'x100'
|
|
RESVD_5 = 5, // b'x101'
|
|
BALR1415 = 6, // b'x110' == BALR r14,r15
|
|
BASR33 = 7, // b'x111' == BASR r3,r3
|
|
};
|
|
|
|
// The Associated Data Area (ADA) contains descriptors which help locating
|
|
// external symbols. For each symbol and type, the displacement into the ADA
|
|
// is stored.
|
|
class AssociatedDataAreaTable {
|
|
public:
|
|
using DisplacementTable =
|
|
MapVector<std::pair<const MCSymbol *, unsigned>, uint32_t>;
|
|
|
|
private:
|
|
const uint64_t PointerSize;
|
|
|
|
/// The mapping of name/slot type pairs to displacements.
|
|
DisplacementTable Displacements;
|
|
|
|
/// The next available displacement value. Incremented when new entries into
|
|
/// the ADA are created.
|
|
uint32_t NextDisplacement = 0;
|
|
|
|
public:
|
|
AssociatedDataAreaTable(uint64_t PointerSize) : PointerSize(PointerSize) {}
|
|
|
|
/// @brief Add a function descriptor to the ADA.
|
|
/// @param MI Pointer to an ADA_ENTRY instruction.
|
|
/// @return The displacement of the descriptor into the ADA.
|
|
uint32_t insert(const MachineOperand MO);
|
|
|
|
/// @brief Get the displacement into associated data area (ADA) for a name.
|
|
/// If no displacement is already associated with the name, assign one and
|
|
/// return it.
|
|
/// @param Sym The symbol for which the displacement should be returned.
|
|
/// @param SlotKind The ADA type.
|
|
/// @return The displacement of the descriptor into the ADA.
|
|
uint32_t insert(const MCSymbol *Sym, unsigned SlotKind);
|
|
|
|
/// Get the table of GOFF displacements. This is 'const' since it should
|
|
/// never be modified by anything except the APIs on this class.
|
|
const DisplacementTable &getTable() const { return Displacements; }
|
|
|
|
uint32_t getNextDisplacement() const { return NextDisplacement; }
|
|
};
|
|
|
|
AssociatedDataAreaTable ADATable;
|
|
|
|
void emitPPA1(MCSymbol *FnEndSym);
|
|
void emitPPA2(Module &M);
|
|
void emitADASection();
|
|
void emitIDRLSection(Module &M);
|
|
|
|
public:
|
|
SystemZAsmPrinter(TargetMachine &TM, std::unique_ptr<MCStreamer> Streamer)
|
|
: AsmPrinter(TM, std::move(Streamer)), CurrentFnPPA1Sym(nullptr),
|
|
CurrentFnEPMarkerSym(nullptr), PPA2Sym(nullptr),
|
|
ADATable(TM.getPointerSize(0)) {}
|
|
|
|
// Override AsmPrinter.
|
|
StringRef getPassName() const override { return "SystemZ Assembly Printer"; }
|
|
void emitInstruction(const MachineInstr *MI) override;
|
|
void emitMachineConstantPoolValue(MachineConstantPoolValue *MCPV) override;
|
|
void emitEndOfAsmFile(Module &M) override;
|
|
bool PrintAsmOperand(const MachineInstr *MI, unsigned OpNo,
|
|
const char *ExtraCode, raw_ostream &OS) override;
|
|
bool PrintAsmMemoryOperand(const MachineInstr *MI, unsigned OpNo,
|
|
const char *ExtraCode, raw_ostream &OS) override;
|
|
|
|
bool doInitialization(Module &M) override {
|
|
SM.reset();
|
|
return AsmPrinter::doInitialization(M);
|
|
}
|
|
void emitFunctionEntryLabel() override;
|
|
void emitFunctionBodyEnd() override;
|
|
void emitStartOfAsmFile(Module &M) override;
|
|
|
|
private:
|
|
void emitCallInformation(CallType CT);
|
|
void LowerFENTRY_CALL(const MachineInstr &MI, SystemZMCInstLower &MCIL);
|
|
void LowerSTACKMAP(const MachineInstr &MI);
|
|
void LowerPATCHPOINT(const MachineInstr &MI, SystemZMCInstLower &Lower);
|
|
void emitAttributes(Module &M);
|
|
};
|
|
} // end namespace llvm
|
|
|
|
#endif
|