2017-03-30 08:29:36 +08:00
|
|
|
//===--- XRayFunctionFilter.cpp - XRay automatic-attribution --------------===//
|
|
|
|
//
|
2019-01-19 16:50:56 +08:00
|
|
|
// 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
|
2017-03-30 08:29:36 +08:00
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
//
|
|
|
|
// User-provided filters for always/never XRay instrumenting certain functions.
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "clang/Basic/XRayLists.h"
|
|
|
|
|
|
|
|
using namespace clang;
|
|
|
|
|
|
|
|
XRayFunctionFilter::XRayFunctionFilter(
|
|
|
|
ArrayRef<std::string> AlwaysInstrumentPaths,
|
2018-04-09 12:02:09 +08:00
|
|
|
ArrayRef<std::string> NeverInstrumentPaths,
|
|
|
|
ArrayRef<std::string> AttrListPaths, SourceManager &SM)
|
2017-03-30 08:29:36 +08:00
|
|
|
: AlwaysInstrument(
|
|
|
|
llvm::SpecialCaseList::createOrDie(AlwaysInstrumentPaths)),
|
|
|
|
NeverInstrument(llvm::SpecialCaseList::createOrDie(NeverInstrumentPaths)),
|
2018-04-09 12:02:09 +08:00
|
|
|
AttrList(llvm::SpecialCaseList::createOrDie(AttrListPaths)), SM(SM) {}
|
2017-03-30 08:29:36 +08:00
|
|
|
|
|
|
|
XRayFunctionFilter::ImbueAttribute
|
|
|
|
XRayFunctionFilter::shouldImbueFunction(StringRef FunctionName) const {
|
|
|
|
// First apply the always instrument list, than if it isn't an "always" see
|
|
|
|
// whether it's treated as a "never" instrument function.
|
2018-04-09 12:02:09 +08:00
|
|
|
// TODO: Remove these as they're deprecated; use the AttrList exclusively.
|
2017-09-26 06:11:12 +08:00
|
|
|
if (AlwaysInstrument->inSection("xray_always_instrument", "fun", FunctionName,
|
2018-04-09 12:02:09 +08:00
|
|
|
"arg1") ||
|
|
|
|
AttrList->inSection("always", "fun", FunctionName, "arg1"))
|
2017-05-24 13:46:36 +08:00
|
|
|
return ImbueAttribute::ALWAYS_ARG1;
|
2017-09-26 06:11:12 +08:00
|
|
|
if (AlwaysInstrument->inSection("xray_always_instrument", "fun",
|
2018-04-09 12:02:09 +08:00
|
|
|
FunctionName) ||
|
|
|
|
AttrList->inSection("always", "fun", FunctionName))
|
2017-03-30 08:29:36 +08:00
|
|
|
return ImbueAttribute::ALWAYS;
|
2018-04-09 12:02:09 +08:00
|
|
|
|
|
|
|
if (NeverInstrument->inSection("xray_never_instrument", "fun",
|
|
|
|
FunctionName) ||
|
|
|
|
AttrList->inSection("never", "fun", FunctionName))
|
2017-03-30 08:29:36 +08:00
|
|
|
return ImbueAttribute::NEVER;
|
2018-04-09 12:02:09 +08:00
|
|
|
|
2017-03-30 08:29:36 +08:00
|
|
|
return ImbueAttribute::NONE;
|
|
|
|
}
|
|
|
|
|
|
|
|
XRayFunctionFilter::ImbueAttribute
|
|
|
|
XRayFunctionFilter::shouldImbueFunctionsInFile(StringRef Filename,
|
|
|
|
StringRef Category) const {
|
2017-09-26 06:11:12 +08:00
|
|
|
if (AlwaysInstrument->inSection("xray_always_instrument", "src", Filename,
|
2018-04-09 12:02:09 +08:00
|
|
|
Category) ||
|
|
|
|
AttrList->inSection("always", "src", Filename, Category))
|
2017-03-30 08:29:36 +08:00
|
|
|
return ImbueAttribute::ALWAYS;
|
2017-09-26 06:11:12 +08:00
|
|
|
if (NeverInstrument->inSection("xray_never_instrument", "src", Filename,
|
2018-04-09 12:02:09 +08:00
|
|
|
Category) ||
|
|
|
|
AttrList->inSection("never", "src", Filename, Category))
|
2017-03-30 08:29:36 +08:00
|
|
|
return ImbueAttribute::NEVER;
|
|
|
|
return ImbueAttribute::NONE;
|
|
|
|
}
|
|
|
|
|
|
|
|
XRayFunctionFilter::ImbueAttribute
|
|
|
|
XRayFunctionFilter::shouldImbueLocation(SourceLocation Loc,
|
|
|
|
StringRef Category) const {
|
|
|
|
if (!Loc.isValid())
|
|
|
|
return ImbueAttribute::NONE;
|
|
|
|
return this->shouldImbueFunctionsInFile(SM.getFilename(SM.getFileLoc(Loc)),
|
|
|
|
Category);
|
|
|
|
}
|