75b3a4b507
This builds on the support being added to LLVM to import and export
registries from DLLs. This will allow us to pick up the registry
entries added in the DLL's copy of FrontendPluginRegistry.
This will allow us to use plugins on Windows using:
$ clang-cl -Xclang -load -Xclang plugin.dll \
-Xclang -add-plugin -Xclang foo
llvm-svn: 260265
125 lines
4.4 KiB
C++
125 lines
4.4 KiB
C++
//===- PrintFunctionNames.cpp ---------------------------------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Example clang plugin which simply prints the names of all the top-level decls
|
|
// in the input file.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "clang/Frontend/FrontendPluginRegistry.h"
|
|
#include "clang/AST/AST.h"
|
|
#include "clang/AST/ASTConsumer.h"
|
|
#include "clang/AST/RecursiveASTVisitor.h"
|
|
#include "clang/Frontend/CompilerInstance.h"
|
|
#include "clang/Sema/Sema.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
using namespace clang;
|
|
|
|
namespace {
|
|
|
|
class PrintFunctionsConsumer : public ASTConsumer {
|
|
CompilerInstance &Instance;
|
|
std::set<std::string> ParsedTemplates;
|
|
|
|
public:
|
|
PrintFunctionsConsumer(CompilerInstance &Instance,
|
|
std::set<std::string> ParsedTemplates)
|
|
: Instance(Instance), ParsedTemplates(ParsedTemplates) {}
|
|
|
|
bool HandleTopLevelDecl(DeclGroupRef DG) override {
|
|
for (DeclGroupRef::iterator i = DG.begin(), e = DG.end(); i != e; ++i) {
|
|
const Decl *D = *i;
|
|
if (const NamedDecl *ND = dyn_cast<NamedDecl>(D))
|
|
llvm::errs() << "top-level-decl: \"" << ND->getNameAsString() << "\"\n";
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
void HandleTranslationUnit(ASTContext& context) override {
|
|
if (!Instance.getLangOpts().DelayedTemplateParsing)
|
|
return;
|
|
|
|
// This demonstrates how to force instantiation of some templates in
|
|
// -fdelayed-template-parsing mode. (Note: Doing this unconditionally for
|
|
// all templates is similar to not using -fdelayed-template-parsig in the
|
|
// first place.)
|
|
// The advantage of doing this in HandleTranslationUnit() is that all
|
|
// codegen (when using -add-plugin) is completely finished and this can't
|
|
// affect the compiler output.
|
|
struct Visitor : public RecursiveASTVisitor<Visitor> {
|
|
const std::set<std::string> &ParsedTemplates;
|
|
Visitor(const std::set<std::string> &ParsedTemplates)
|
|
: ParsedTemplates(ParsedTemplates) {}
|
|
bool VisitFunctionDecl(FunctionDecl *FD) {
|
|
if (FD->isLateTemplateParsed() &&
|
|
ParsedTemplates.count(FD->getNameAsString()))
|
|
LateParsedDecls.insert(FD);
|
|
return true;
|
|
}
|
|
|
|
std::set<FunctionDecl*> LateParsedDecls;
|
|
} v(ParsedTemplates);
|
|
v.TraverseDecl(context.getTranslationUnitDecl());
|
|
clang::Sema &sema = Instance.getSema();
|
|
for (const FunctionDecl *FD : v.LateParsedDecls) {
|
|
clang::LateParsedTemplate* LPT = sema.LateParsedTemplateMap.lookup(FD);
|
|
sema.LateTemplateParser(sema.OpaqueParser, *LPT);
|
|
llvm::errs() << "late-parsed-decl: \"" << FD->getNameAsString() << "\"\n";
|
|
}
|
|
}
|
|
};
|
|
|
|
class PrintFunctionNamesAction : public PluginASTAction {
|
|
std::set<std::string> ParsedTemplates;
|
|
protected:
|
|
std::unique_ptr<ASTConsumer> CreateASTConsumer(CompilerInstance &CI,
|
|
llvm::StringRef) override {
|
|
return llvm::make_unique<PrintFunctionsConsumer>(CI, ParsedTemplates);
|
|
}
|
|
|
|
bool ParseArgs(const CompilerInstance &CI,
|
|
const std::vector<std::string> &args) override {
|
|
for (unsigned i = 0, e = args.size(); i != e; ++i) {
|
|
llvm::errs() << "PrintFunctionNames arg = " << args[i] << "\n";
|
|
|
|
// Example error handling.
|
|
DiagnosticsEngine &D = CI.getDiagnostics();
|
|
if (args[i] == "-an-error") {
|
|
unsigned DiagID = D.getCustomDiagID(DiagnosticsEngine::Error,
|
|
"invalid argument '%0'");
|
|
D.Report(DiagID) << args[i];
|
|
return false;
|
|
} else if (args[i] == "-parse-template") {
|
|
if (i + 1 >= e) {
|
|
D.Report(D.getCustomDiagID(DiagnosticsEngine::Error,
|
|
"missing -parse-template argument"));
|
|
return false;
|
|
}
|
|
++i;
|
|
ParsedTemplates.insert(args[i]);
|
|
}
|
|
}
|
|
if (!args.empty() && args[0] == "help")
|
|
PrintHelp(llvm::errs());
|
|
|
|
return true;
|
|
}
|
|
void PrintHelp(llvm::raw_ostream& ros) {
|
|
ros << "Help for PrintFunctionNames plugin goes here\n";
|
|
}
|
|
|
|
};
|
|
|
|
}
|
|
|
|
static FrontendPluginRegistry::Add<PrintFunctionNamesAction>
|
|
X("print-fns", "print function names");
|
|
LLVM_EXPORT_REGISTRY(FrontendPluginRegistry)
|