151 lines
5.1 KiB
C++
151 lines
5.1 KiB
C++
//= AArch64WinCOFFObjectWriter.cpp - AArch64 Windows COFF Object Writer 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
|
|
//
|
|
//===---------------------------------------------------------------------===//
|
|
|
|
#include "MCTargetDesc/AArch64FixupKinds.h"
|
|
#include "MCTargetDesc/AArch64MCExpr.h"
|
|
#include "llvm/ADT/Twine.h"
|
|
#include "llvm/BinaryFormat/COFF.h"
|
|
#include "llvm/MC/MCAsmBackend.h"
|
|
#include "llvm/MC/MCContext.h"
|
|
#include "llvm/MC/MCExpr.h"
|
|
#include "llvm/MC/MCFixup.h"
|
|
#include "llvm/MC/MCFixupKindInfo.h"
|
|
#include "llvm/MC/MCObjectWriter.h"
|
|
#include "llvm/MC/MCValue.h"
|
|
#include "llvm/MC/MCWinCOFFObjectWriter.h"
|
|
#include "llvm/Support/ErrorHandling.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include <cassert>
|
|
|
|
using namespace llvm;
|
|
|
|
namespace {
|
|
|
|
class AArch64WinCOFFObjectWriter : public MCWinCOFFObjectTargetWriter {
|
|
public:
|
|
AArch64WinCOFFObjectWriter()
|
|
: MCWinCOFFObjectTargetWriter(COFF::IMAGE_FILE_MACHINE_ARM64) {}
|
|
|
|
~AArch64WinCOFFObjectWriter() override = default;
|
|
|
|
unsigned getRelocType(MCContext &Ctx, const MCValue &Target,
|
|
const MCFixup &Fixup, bool IsCrossSection,
|
|
const MCAsmBackend &MAB) const override;
|
|
|
|
bool recordRelocation(const MCFixup &) const override;
|
|
};
|
|
|
|
} // end anonymous namespace
|
|
|
|
unsigned AArch64WinCOFFObjectWriter::getRelocType(
|
|
MCContext &Ctx, const MCValue &Target, const MCFixup &Fixup,
|
|
bool IsCrossSection, const MCAsmBackend &MAB) const {
|
|
auto Modifier = Target.isAbsolute() ? MCSymbolRefExpr::VK_None
|
|
: Target.getSymA()->getKind();
|
|
const MCExpr *Expr = Fixup.getValue();
|
|
|
|
if (const AArch64MCExpr *A64E = dyn_cast<AArch64MCExpr>(Expr)) {
|
|
AArch64MCExpr::VariantKind RefKind = A64E->getKind();
|
|
switch (AArch64MCExpr::getSymbolLoc(RefKind)) {
|
|
case AArch64MCExpr::VK_ABS:
|
|
case AArch64MCExpr::VK_SECREL:
|
|
// Supported
|
|
break;
|
|
default:
|
|
Ctx.reportError(Fixup.getLoc(), "relocation variant " +
|
|
A64E->getVariantKindName() +
|
|
" unsupported on COFF targets");
|
|
return COFF::IMAGE_REL_ARM64_ABSOLUTE; // Dummy return value
|
|
}
|
|
}
|
|
|
|
switch (static_cast<unsigned>(Fixup.getKind())) {
|
|
default: {
|
|
if (const AArch64MCExpr *A64E = dyn_cast<AArch64MCExpr>(Expr)) {
|
|
Ctx.reportError(Fixup.getLoc(), "relocation type " +
|
|
A64E->getVariantKindName() +
|
|
" unsupported on COFF targets");
|
|
} else {
|
|
const MCFixupKindInfo &Info = MAB.getFixupKindInfo(Fixup.getKind());
|
|
Ctx.reportError(Fixup.getLoc(), Twine("relocation type ") + Info.Name +
|
|
" unsupported on COFF targets");
|
|
}
|
|
return COFF::IMAGE_REL_ARM64_ABSOLUTE; // Dummy return value
|
|
}
|
|
|
|
case FK_Data_4:
|
|
switch (Modifier) {
|
|
default:
|
|
return COFF::IMAGE_REL_ARM64_ADDR32;
|
|
case MCSymbolRefExpr::VK_COFF_IMGREL32:
|
|
return COFF::IMAGE_REL_ARM64_ADDR32NB;
|
|
case MCSymbolRefExpr::VK_SECREL:
|
|
return COFF::IMAGE_REL_ARM64_SECREL;
|
|
}
|
|
|
|
case FK_Data_8:
|
|
return COFF::IMAGE_REL_ARM64_ADDR64;
|
|
|
|
case FK_SecRel_2:
|
|
return COFF::IMAGE_REL_ARM64_SECTION;
|
|
|
|
case FK_SecRel_4:
|
|
return COFF::IMAGE_REL_ARM64_SECREL;
|
|
|
|
case AArch64::fixup_aarch64_add_imm12:
|
|
if (const AArch64MCExpr *A64E = dyn_cast<AArch64MCExpr>(Expr)) {
|
|
AArch64MCExpr::VariantKind RefKind = A64E->getKind();
|
|
if (RefKind == AArch64MCExpr::VK_SECREL_LO12)
|
|
return COFF::IMAGE_REL_ARM64_SECREL_LOW12A;
|
|
if (RefKind == AArch64MCExpr::VK_SECREL_HI12)
|
|
return COFF::IMAGE_REL_ARM64_SECREL_HIGH12A;
|
|
}
|
|
return COFF::IMAGE_REL_ARM64_PAGEOFFSET_12A;
|
|
|
|
case AArch64::fixup_aarch64_ldst_imm12_scale1:
|
|
case AArch64::fixup_aarch64_ldst_imm12_scale2:
|
|
case AArch64::fixup_aarch64_ldst_imm12_scale4:
|
|
case AArch64::fixup_aarch64_ldst_imm12_scale8:
|
|
case AArch64::fixup_aarch64_ldst_imm12_scale16:
|
|
if (const AArch64MCExpr *A64E = dyn_cast<AArch64MCExpr>(Expr)) {
|
|
AArch64MCExpr::VariantKind RefKind = A64E->getKind();
|
|
if (RefKind == AArch64MCExpr::VK_SECREL_LO12)
|
|
return COFF::IMAGE_REL_ARM64_SECREL_LOW12L;
|
|
}
|
|
return COFF::IMAGE_REL_ARM64_PAGEOFFSET_12L;
|
|
|
|
case AArch64::fixup_aarch64_pcrel_adr_imm21:
|
|
return COFF::IMAGE_REL_ARM64_REL21;
|
|
|
|
case AArch64::fixup_aarch64_pcrel_adrp_imm21:
|
|
return COFF::IMAGE_REL_ARM64_PAGEBASE_REL21;
|
|
|
|
case AArch64::fixup_aarch64_pcrel_branch14:
|
|
return COFF::IMAGE_REL_ARM64_BRANCH14;
|
|
|
|
case AArch64::fixup_aarch64_pcrel_branch19:
|
|
return COFF::IMAGE_REL_ARM64_BRANCH19;
|
|
|
|
case AArch64::fixup_aarch64_pcrel_branch26:
|
|
case AArch64::fixup_aarch64_pcrel_call26:
|
|
return COFF::IMAGE_REL_ARM64_BRANCH26;
|
|
}
|
|
}
|
|
|
|
bool AArch64WinCOFFObjectWriter::recordRelocation(const MCFixup &Fixup) const {
|
|
return true;
|
|
}
|
|
|
|
namespace llvm {
|
|
|
|
std::unique_ptr<MCObjectTargetWriter> createAArch64WinCOFFObjectWriter() {
|
|
return std::make_unique<AArch64WinCOFFObjectWriter>();
|
|
}
|
|
|
|
} // end namespace llvm
|