forked from OSchip/llvm-project
125 lines
4.0 KiB
C++
125 lines
4.0 KiB
C++
//===- ClangExtDefMapGen.cpp -----------------------------------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===--------------------------------------------------------------------===//
|
|
//
|
|
// Clang tool which creates a list of defined functions and the files in which
|
|
// they are defined.
|
|
//
|
|
//===--------------------------------------------------------------------===//
|
|
|
|
#include "clang/AST/ASTConsumer.h"
|
|
#include "clang/AST/ASTContext.h"
|
|
#include "clang/Basic/SourceManager.h"
|
|
#include "clang/CrossTU/CrossTranslationUnit.h"
|
|
#include "clang/Frontend/CompilerInstance.h"
|
|
#include "clang/Frontend/FrontendActions.h"
|
|
#include "clang/Tooling/CommonOptionsParser.h"
|
|
#include "clang/Tooling/Tooling.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/Signals.h"
|
|
#include <sstream>
|
|
#include <string>
|
|
|
|
using namespace llvm;
|
|
using namespace clang;
|
|
using namespace clang::cross_tu;
|
|
using namespace clang::tooling;
|
|
|
|
static cl::OptionCategory ClangExtDefMapGenCategory("clang-extdefmapgen options");
|
|
|
|
class MapExtDefNamesConsumer : public ASTConsumer {
|
|
public:
|
|
MapExtDefNamesConsumer(ASTContext &Context)
|
|
: Ctx(Context), SM(Context.getSourceManager()) {}
|
|
|
|
~MapExtDefNamesConsumer() {
|
|
// Flush results to standard output.
|
|
llvm::outs() << createCrossTUIndexString(Index);
|
|
}
|
|
|
|
void HandleTranslationUnit(ASTContext &Context) override {
|
|
handleDecl(Context.getTranslationUnitDecl());
|
|
}
|
|
|
|
private:
|
|
void handleDecl(const Decl *D);
|
|
void addIfInMain(const DeclaratorDecl *DD, SourceLocation defStart);
|
|
|
|
ASTContext &Ctx;
|
|
SourceManager &SM;
|
|
llvm::StringMap<std::string> Index;
|
|
std::string CurrentFileName;
|
|
};
|
|
|
|
void MapExtDefNamesConsumer::handleDecl(const Decl *D) {
|
|
if (!D)
|
|
return;
|
|
|
|
if (const auto *FD = dyn_cast<FunctionDecl>(D)) {
|
|
if (FD->isThisDeclarationADefinition())
|
|
if (const Stmt *Body = FD->getBody())
|
|
addIfInMain(FD, Body->getBeginLoc());
|
|
} else if (const auto *VD = dyn_cast<VarDecl>(D)) {
|
|
if (cross_tu::containsConst(VD, Ctx) && VD->hasInit())
|
|
if (const Expr *Init = VD->getInit())
|
|
addIfInMain(VD, Init->getBeginLoc());
|
|
}
|
|
|
|
if (const auto *DC = dyn_cast<DeclContext>(D))
|
|
for (const Decl *D : DC->decls())
|
|
handleDecl(D);
|
|
}
|
|
|
|
void MapExtDefNamesConsumer::addIfInMain(const DeclaratorDecl *DD,
|
|
SourceLocation defStart) {
|
|
std::string LookupName = CrossTranslationUnitContext::getLookupName(DD);
|
|
if (CurrentFileName.empty()) {
|
|
CurrentFileName =
|
|
SM.getFileEntryForID(SM.getMainFileID())->tryGetRealPathName();
|
|
if (CurrentFileName.empty())
|
|
CurrentFileName = "invalid_file";
|
|
}
|
|
|
|
switch (DD->getLinkageInternal()) {
|
|
case ExternalLinkage:
|
|
case VisibleNoLinkage:
|
|
case UniqueExternalLinkage:
|
|
if (SM.isInMainFile(defStart))
|
|
Index[LookupName] = CurrentFileName;
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
class MapExtDefNamesAction : public ASTFrontendAction {
|
|
protected:
|
|
std::unique_ptr<ASTConsumer> CreateASTConsumer(CompilerInstance &CI,
|
|
llvm::StringRef) {
|
|
return llvm::make_unique<MapExtDefNamesConsumer>(CI.getASTContext());
|
|
}
|
|
};
|
|
|
|
static cl::extrahelp CommonHelp(CommonOptionsParser::HelpMessage);
|
|
|
|
int main(int argc, const char **argv) {
|
|
// Print a stack trace if we signal out.
|
|
sys::PrintStackTraceOnErrorSignal(argv[0], false);
|
|
PrettyStackTraceProgram X(argc, argv);
|
|
|
|
const char *Overview = "\nThis tool collects the USR name and location "
|
|
"of external definitions in the source files "
|
|
"(excluding headers).\n";
|
|
CommonOptionsParser OptionsParser(argc, argv, ClangExtDefMapGenCategory,
|
|
cl::ZeroOrMore, Overview);
|
|
|
|
ClangTool Tool(OptionsParser.getCompilations(),
|
|
OptionsParser.getSourcePathList());
|
|
|
|
return Tool.run(newFrontendActionFactory<MapExtDefNamesAction>().get());
|
|
}
|