
This fixes what I consider to be an API flaw I've tripped over multiple times. The point this is constructed isn't well defined, so depending on where this is first called, you can conclude different information based on the MachineFunction. For example, the AMDGPU implementation inspected the MachineFrameInfo on construction for the stack objects and if the frame has calls. This kind of worked in SelectionDAG which visited all allocas up front, but broke in GlobalISel which hasn't visited any of the IR when arguments are lowered. I've run into similar problems before with the MIR parser and trying to make use of other MachineFunction fields, so I think it's best to just categorically disallow dependency on the MachineFunction state in the constructor and to always construct this at the same time as the MachineFunction itself. A missing feature I still could use is a way to access an custom analysis pass on the IR here.
79 lines
2.9 KiB
C++
79 lines
2.9 KiB
C++
//===-- ARMMachineFunctionInfo.cpp - ARM 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "ARMMachineFunctionInfo.h"
|
|
#include "ARMSubtarget.h"
|
|
|
|
using namespace llvm;
|
|
|
|
void ARMFunctionInfo::anchor() {}
|
|
|
|
static bool GetBranchTargetEnforcement(const Function &F,
|
|
const ARMSubtarget *Subtarget) {
|
|
if (!Subtarget->isMClass() || !Subtarget->hasV7Ops())
|
|
return false;
|
|
|
|
if (!F.hasFnAttribute("branch-target-enforcement")) {
|
|
if (const auto *BTE = mdconst::extract_or_null<ConstantInt>(
|
|
F.getParent()->getModuleFlag("branch-target-enforcement")))
|
|
return BTE->getZExtValue();
|
|
return false;
|
|
}
|
|
|
|
const StringRef BTIEnable =
|
|
F.getFnAttribute("branch-target-enforcement").getValueAsString();
|
|
assert(BTIEnable.equals_insensitive("true") ||
|
|
BTIEnable.equals_insensitive("false"));
|
|
return BTIEnable.equals_insensitive("true");
|
|
}
|
|
|
|
// The pair returns values for the ARMFunctionInfo members
|
|
// SignReturnAddress and SignReturnAddressAll respectively.
|
|
static std::pair<bool, bool> GetSignReturnAddress(const Function &F) {
|
|
if (!F.hasFnAttribute("sign-return-address")) {
|
|
const Module &M = *F.getParent();
|
|
if (const auto *Sign = mdconst::extract_or_null<ConstantInt>(
|
|
M.getModuleFlag("sign-return-address"))) {
|
|
if (Sign->getZExtValue()) {
|
|
if (const auto *All = mdconst::extract_or_null<ConstantInt>(
|
|
M.getModuleFlag("sign-return-address-all")))
|
|
return {true, All->getZExtValue()};
|
|
return {true, false};
|
|
}
|
|
}
|
|
return {false, false};
|
|
}
|
|
|
|
StringRef Scope = F.getFnAttribute("sign-return-address").getValueAsString();
|
|
if (Scope.equals("none"))
|
|
return {false, false};
|
|
|
|
if (Scope.equals("all"))
|
|
return {true, true};
|
|
|
|
assert(Scope.equals("non-leaf"));
|
|
return {true, false};
|
|
}
|
|
|
|
ARMFunctionInfo::ARMFunctionInfo(const Function &F,
|
|
const ARMSubtarget *Subtarget)
|
|
: isThumb(Subtarget->isThumb()), hasThumb2(Subtarget->hasThumb2()),
|
|
IsCmseNSEntry(F.hasFnAttribute("cmse_nonsecure_entry")),
|
|
IsCmseNSCall(F.hasFnAttribute("cmse_nonsecure_call")),
|
|
BranchTargetEnforcement(GetBranchTargetEnforcement(F, Subtarget)) {
|
|
if (Subtarget->isMClass() && Subtarget->hasV7Ops())
|
|
std::tie(SignReturnAddress, SignReturnAddressAll) = GetSignReturnAddress(F);
|
|
}
|
|
|
|
MachineFunctionInfo *
|
|
ARMFunctionInfo::clone(BumpPtrAllocator &Allocator, MachineFunction &DestMF,
|
|
const DenseMap<MachineBasicBlock *, MachineBasicBlock *>
|
|
&Src2DstMBB) const {
|
|
return DestMF.cloneInfo<ARMFunctionInfo>(*this);
|
|
}
|