97 lines
3.5 KiB
C++
97 lines
3.5 KiB
C++
//===-- XRayInstrumentation.cpp - Adds XRay instrumentation to functions. -===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements a MachineFunctionPass that inserts the appropriate
|
|
// XRay instrumentation instructions. We look for XRay-specific attributes
|
|
// on the function to determine whether we should insert the replacement
|
|
// operations.
|
|
//
|
|
//===---------------------------------------------------------------------===//
|
|
|
|
#include "llvm/CodeGen/Analysis.h"
|
|
#include "llvm/CodeGen/MachineFunction.h"
|
|
#include "llvm/CodeGen/MachineFunctionPass.h"
|
|
#include "llvm/CodeGen/MachineInstrBuilder.h"
|
|
#include "llvm/CodeGen/Passes.h"
|
|
#include "llvm/Support/TargetRegistry.h"
|
|
#include "llvm/Target/TargetInstrInfo.h"
|
|
#include "llvm/Target/TargetSubtargetInfo.h"
|
|
|
|
using namespace llvm;
|
|
|
|
namespace {
|
|
struct XRayInstrumentation : public MachineFunctionPass {
|
|
static char ID;
|
|
|
|
XRayInstrumentation() : MachineFunctionPass(ID) {
|
|
initializeXRayInstrumentationPass(*PassRegistry::getPassRegistry());
|
|
}
|
|
|
|
bool runOnMachineFunction(MachineFunction &MF) override;
|
|
};
|
|
}
|
|
|
|
bool XRayInstrumentation::runOnMachineFunction(MachineFunction &MF) {
|
|
auto &F = *MF.getFunction();
|
|
auto InstrAttr = F.getFnAttribute("function-instrument");
|
|
bool AlwaysInstrument = !InstrAttr.hasAttribute(Attribute::None) &&
|
|
InstrAttr.isStringAttribute() &&
|
|
InstrAttr.getValueAsString() == "xray-always";
|
|
Attribute Attr = F.getFnAttribute("xray-instruction-threshold");
|
|
unsigned XRayThreshold = 0;
|
|
if (!AlwaysInstrument) {
|
|
if (Attr.hasAttribute(Attribute::None) || !Attr.isStringAttribute())
|
|
return false; // XRay threshold attribute not found.
|
|
if (Attr.getValueAsString().getAsInteger(10, XRayThreshold))
|
|
return false; // Invalid value for threshold.
|
|
if (F.size() < XRayThreshold)
|
|
return false; // Function is too small.
|
|
}
|
|
|
|
// FIXME: Do the loop triviality analysis here or in an earlier pass.
|
|
|
|
// First, insert an PATCHABLE_FUNCTION_ENTER as the first instruction of the
|
|
// MachineFunction.
|
|
auto &FirstMBB = *MF.begin();
|
|
auto &FirstMI = *FirstMBB.begin();
|
|
auto *TII = MF.getSubtarget().getInstrInfo();
|
|
BuildMI(FirstMBB, FirstMI, FirstMI.getDebugLoc(),
|
|
TII->get(TargetOpcode::PATCHABLE_FUNCTION_ENTER));
|
|
|
|
// Then we look for *all* terminators and returns, then replace those with
|
|
// PATCHABLE_RET instructions.
|
|
SmallVector<MachineInstr *, 4> Terminators;
|
|
for (auto &MBB : MF) {
|
|
for (auto &T : MBB.terminators()) {
|
|
// FIXME: Handle tail calls here too?
|
|
if (T.isReturn() && T.getOpcode() == TII->getReturnOpcode()) {
|
|
// Replace return instructions with:
|
|
// PATCHABLE_RET <Opcode>, <Operand>...
|
|
auto MIB = BuildMI(MBB, T, T.getDebugLoc(),
|
|
TII->get(TargetOpcode::PATCHABLE_RET))
|
|
.addImm(T.getOpcode());
|
|
for (auto &MO : T.operands())
|
|
MIB.addOperand(MO);
|
|
Terminators.push_back(&T);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
for (auto &I : Terminators)
|
|
I->eraseFromParent();
|
|
|
|
return true;
|
|
}
|
|
|
|
char XRayInstrumentation::ID = 0;
|
|
char &llvm::XRayInstrumentationID = XRayInstrumentation::ID;
|
|
INITIALIZE_PASS(XRayInstrumentation, "xray-instrumentation", "Insert XRay ops",
|
|
false, false)
|