4e186f20e2
Fix a use-after-free issue related to annotateTableJump in the LoongArch target. Previously, `LoongArchPreRAExpandPseudo::annotateTableJump()` recorded a reference to a MachineOperand representing a jump table index. However, later optimizations such as the `BranchFolder` pass may delete the instruction containing this operand, leaving a dangling reference. This led to an assertion failure in `LoongArchAsmPrinter::emitJumpTableInfo()` when trying to access a freed MachineOperand via `getIndex()`. The fix avoids holding a reference to the MachineOperand. Instead, we extract and store the jump table index at the time of annotation. During `emitJumpTableInfo()`, we verify whether the recorded index still exists in the MachineFunction's jump table. If not, we skip emission for that entry. Fixes #140904
90 lines
3.2 KiB
C++
90 lines
3.2 KiB
C++
//=- LoongArchMachineFunctionInfo.h - LoongArch machine function info -----===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file declares LoongArch-specific per-machine-function information.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_LIB_TARGET_LOONGARCH_LOONGARCHMACHINEFUNCTIONINFO_H
|
|
#define LLVM_LIB_TARGET_LOONGARCH_LOONGARCHMACHINEFUNCTIONINFO_H
|
|
|
|
#include "LoongArchSubtarget.h"
|
|
#include "llvm/CodeGen/MachineFrameInfo.h"
|
|
#include "llvm/CodeGen/MachineFunction.h"
|
|
|
|
namespace llvm {
|
|
|
|
/// LoongArchMachineFunctionInfo - This class is derived from
|
|
/// MachineFunctionInfo and contains private LoongArch-specific information for
|
|
/// each MachineFunction.
|
|
class LoongArchMachineFunctionInfo : public MachineFunctionInfo {
|
|
private:
|
|
/// FrameIndex for start of varargs area
|
|
int VarArgsFrameIndex = 0;
|
|
/// Size of the save area used for varargs
|
|
int VarArgsSaveSize = 0;
|
|
|
|
/// Size of stack frame to save callee saved registers
|
|
unsigned CalleeSavedStackSize = 0;
|
|
|
|
/// FrameIndex of the spill slot when there is no scavenged register in
|
|
/// insertIndirectBranch.
|
|
int BranchRelaxationSpillFrameIndex = -1;
|
|
|
|
/// Registers that have been sign extended from i32.
|
|
SmallVector<Register, 8> SExt32Registers;
|
|
|
|
/// Pairs of `jr` instructions and corresponding JTI operands, used for the
|
|
/// `annotate-tablejump` option.
|
|
SmallVector<std::pair<MachineInstr *, int>, 4> JumpInfos;
|
|
|
|
public:
|
|
LoongArchMachineFunctionInfo(const Function &F,
|
|
const TargetSubtargetInfo *STI) {}
|
|
|
|
MachineFunctionInfo *
|
|
clone(BumpPtrAllocator &Allocator, MachineFunction &DestMF,
|
|
const DenseMap<MachineBasicBlock *, MachineBasicBlock *> &Src2DstMBB)
|
|
const override {
|
|
return DestMF.cloneInfo<LoongArchMachineFunctionInfo>(*this);
|
|
}
|
|
|
|
int getVarArgsFrameIndex() const { return VarArgsFrameIndex; }
|
|
void setVarArgsFrameIndex(int Index) { VarArgsFrameIndex = Index; }
|
|
|
|
unsigned getVarArgsSaveSize() const { return VarArgsSaveSize; }
|
|
void setVarArgsSaveSize(int Size) { VarArgsSaveSize = Size; }
|
|
|
|
unsigned getCalleeSavedStackSize() const { return CalleeSavedStackSize; }
|
|
void setCalleeSavedStackSize(unsigned Size) { CalleeSavedStackSize = Size; }
|
|
|
|
int getBranchRelaxationSpillFrameIndex() {
|
|
return BranchRelaxationSpillFrameIndex;
|
|
}
|
|
void setBranchRelaxationSpillFrameIndex(int Index) {
|
|
BranchRelaxationSpillFrameIndex = Index;
|
|
}
|
|
|
|
void addSExt32Register(Register Reg) { SExt32Registers.push_back(Reg); }
|
|
|
|
bool isSExt32Register(Register Reg) const {
|
|
return is_contained(SExt32Registers, Reg);
|
|
}
|
|
|
|
void setJumpInfo(MachineInstr *JrMI, int JTIIdx) {
|
|
JumpInfos.push_back(std::make_pair(JrMI, JTIIdx));
|
|
}
|
|
unsigned getJumpInfoSize() { return JumpInfos.size(); }
|
|
MachineInstr *getJumpInfoJrMI(unsigned Idx) { return JumpInfos[Idx].first; }
|
|
int getJumpInfoJTIIndex(unsigned Idx) { return JumpInfos[Idx].second; }
|
|
};
|
|
|
|
} // end namespace llvm
|
|
|
|
#endif // LLVM_LIB_TARGET_LOONGARCH_LOONGARCHMACHINEFUNCTIONINFO_H
|