llvm-project/llvm/lib/Target/LoongArch/MCTargetDesc/LoongArchELFObjectWriter.cpp
ZhaoQi 8f671a675f
[LoongArch] Always emit symbol-based relocations regardless of relaxation (#153943)
This commit changes all relocations to be relocated with symbols.

Without this commit, errors may occur in some cases, such as when using
`llc/lto+relax`, or combining relaxed and norelaxed object files using
`ld -r`.

Some tests updated.
2025-08-18 20:15:49 +08:00

106 lines
3.5 KiB
C++

//===-- LoongArchELFObjectWriter.cpp - LoongArch ELF 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/LoongArchFixupKinds.h"
#include "MCTargetDesc/LoongArchMCTargetDesc.h"
#include "llvm/BinaryFormat/ELF.h"
#include "llvm/MC/MCContext.h"
#include "llvm/MC/MCELFObjectWriter.h"
#include "llvm/MC/MCFixup.h"
#include "llvm/MC/MCObjectWriter.h"
#include "llvm/MC/MCValue.h"
#include "llvm/Support/ErrorHandling.h"
using namespace llvm;
namespace {
class LoongArchELFObjectWriter : public MCELFObjectTargetWriter {
public:
LoongArchELFObjectWriter(uint8_t OSABI, bool Is64Bit);
~LoongArchELFObjectWriter() override;
bool needsRelocateWithSymbol(const MCValue &, unsigned Type) const override {
return true;
}
protected:
unsigned getRelocType(const MCFixup &, const MCValue &,
bool IsPCRel) const override;
};
} // end namespace
LoongArchELFObjectWriter::LoongArchELFObjectWriter(uint8_t OSABI, bool Is64Bit)
: MCELFObjectTargetWriter(Is64Bit, OSABI, ELF::EM_LOONGARCH,
/*HasRelocationAddend=*/true) {}
LoongArchELFObjectWriter::~LoongArchELFObjectWriter() {}
unsigned LoongArchELFObjectWriter::getRelocType(const MCFixup &Fixup,
const MCValue &Target,
bool IsPCRel) const {
switch (Target.getSpecifier()) {
case ELF::R_LARCH_TLS_LE_HI20:
case ELF::R_LARCH_TLS_IE_PC_HI20:
case ELF::R_LARCH_TLS_IE_HI20:
case ELF::R_LARCH_TLS_LD_PC_HI20:
case ELF::R_LARCH_TLS_LD_HI20:
case ELF::R_LARCH_TLS_GD_PC_HI20:
case ELF::R_LARCH_TLS_GD_HI20:
case ELF::R_LARCH_TLS_DESC_PC_HI20:
case ELF::R_LARCH_TLS_DESC_HI20:
case ELF::R_LARCH_TLS_LE_HI20_R:
case ELF::R_LARCH_TLS_LD_PCREL20_S2:
case ELF::R_LARCH_TLS_GD_PCREL20_S2:
case ELF::R_LARCH_TLS_DESC_PCREL20_S2:
if (auto *SA = const_cast<MCSymbol *>(Target.getAddSym()))
static_cast<MCSymbolELF *>(SA)->setType(ELF::STT_TLS);
break;
default:
break;
}
auto Kind = Fixup.getKind();
if (mc::isRelocation(Fixup.getKind()))
return Kind;
switch (Kind) {
default:
reportError(Fixup.getLoc(), "Unsupported relocation type");
return ELF::R_LARCH_NONE;
case FK_Data_1:
reportError(Fixup.getLoc(), "1-byte data relocations not supported");
return ELF::R_LARCH_NONE;
case FK_Data_2:
reportError(Fixup.getLoc(), "2-byte data relocations not supported");
return ELF::R_LARCH_NONE;
case FK_Data_4:
return IsPCRel ? ELF::R_LARCH_32_PCREL : ELF::R_LARCH_32;
case FK_Data_8:
return IsPCRel ? ELF::R_LARCH_64_PCREL : ELF::R_LARCH_64;
case LoongArch::fixup_loongarch_b16:
return ELF::R_LARCH_B16;
case LoongArch::fixup_loongarch_b21:
return ELF::R_LARCH_B21;
case LoongArch::fixup_loongarch_b26:
return ELF::R_LARCH_B26;
case LoongArch::fixup_loongarch_abs_hi20:
return ELF::R_LARCH_ABS_HI20;
case LoongArch::fixup_loongarch_abs_lo12:
return ELF::R_LARCH_ABS_LO12;
case LoongArch::fixup_loongarch_abs64_lo20:
return ELF::R_LARCH_ABS64_LO20;
case LoongArch::fixup_loongarch_abs64_hi12:
return ELF::R_LARCH_ABS64_HI12;
}
}
std::unique_ptr<MCObjectTargetWriter>
llvm::createLoongArchELFObjectWriter(uint8_t OSABI, bool Is64Bit) {
return std::make_unique<LoongArchELFObjectWriter>(OSABI, Is64Bit);
}