summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
Diffstat
-rw-r--r--clang/examples/PrintFunctionNames/PrintFunctionNames.cpp55
-rw-r--r--clang/lib/Parse/Parser.cpp49
-rw-r--r--clang/test/Frontend/plugin-delayed-template.cpp17
3 files changed, 99 insertions, 22 deletions
diff --git a/clang/examples/PrintFunctionNames/PrintFunctionNames.cpp b/clang/examples/PrintFunctionNames/PrintFunctionNames.cpp
index 39443f597fa..9f8f6e3f050 100644
--- a/clang/examples/PrintFunctionNames/PrintFunctionNames.cpp
+++ b/clang/examples/PrintFunctionNames/PrintFunctionNames.cpp
@@ -15,14 +15,23 @@
#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;
@@ -32,13 +41,47 @@ public:
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>();
+ return llvm::make_unique<PrintFunctionsConsumer>(CI, ParsedTemplates);
}
bool ParseArgs(const CompilerInstance &CI,
@@ -47,12 +90,20 @@ protected:
llvm::errs() << "PrintFunctionNames arg = " << args[i] << "\n";
// Example error handling.
+ DiagnosticsEngine &D = CI.getDiagnostics();
if (args[i] == "-an-error") {
- DiagnosticsEngine &D = CI.getDiagnostics();
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")
diff --git a/clang/lib/Parse/Parser.cpp b/clang/lib/Parse/Parser.cpp
index a45eaa0e333..dea7a6998a2 100644
--- a/clang/lib/Parse/Parser.cpp
+++ b/clang/lib/Parse/Parser.cpp
@@ -38,6 +38,26 @@ public:
return false;
}
};
+
+/// \brief RAIIObject to destroy the contents of a SmallVector of
+/// TemplateIdAnnotation pointers and clear the vector.
+class DestroyTemplateIdAnnotationsRAIIObj {
+ SmallVectorImpl<TemplateIdAnnotation *> &Container;
+
+public:
+ DestroyTemplateIdAnnotationsRAIIObj(
+ SmallVectorImpl<TemplateIdAnnotation *> &Container)
+ : Container(Container) {}
+
+ ~DestroyTemplateIdAnnotationsRAIIObj() {
+ for (SmallVectorImpl<TemplateIdAnnotation *>::iterator I =
+ Container.begin(),
+ E = Container.end();
+ I != E; ++I)
+ (*I)->Destroy();
+ Container.clear();
+ }
+};
} // end anonymous namespace
IdentifierInfo *Parser::getSEHExceptKeyword() {
@@ -414,6 +434,15 @@ Parser::~Parser() {
PP.clearCodeCompletionHandler();
+ if (getLangOpts().DelayedTemplateParsing &&
+ !PP.isIncrementalProcessingEnabled() && !TemplateIds.empty()) {
+ // If an ASTConsumer parsed delay-parsed templates in their
+ // HandleTranslationUnit() method, TemplateIds created there were not
+ // guarded by a DestroyTemplateIdAnnotationsRAIIObj object in
+ // ParseTopLevelDecl(). Destroy them here.
+ DestroyTemplateIdAnnotationsRAIIObj CleanupRAII(TemplateIds);
+ }
+
assert(TemplateIds.empty() && "Still alive TemplateIdAnnotations around?");
}
@@ -490,26 +519,6 @@ void Parser::Initialize() {
ConsumeToken();
}
-namespace {
- /// \brief RAIIObject to destroy the contents of a SmallVector of
- /// TemplateIdAnnotation pointers and clear the vector.
- class DestroyTemplateIdAnnotationsRAIIObj {
- SmallVectorImpl<TemplateIdAnnotation *> &Container;
- public:
- DestroyTemplateIdAnnotationsRAIIObj(SmallVectorImpl<TemplateIdAnnotation *>
- &Container)
- : Container(Container) {}
-
- ~DestroyTemplateIdAnnotationsRAIIObj() {
- for (SmallVectorImpl<TemplateIdAnnotation *>::iterator I =
- Container.begin(), E = Container.end();
- I != E; ++I)
- (*I)->Destroy();
- Container.clear();
- }
- };
-}
-
void Parser::LateTemplateParserCleanupCallback(void *P) {
// While this RAII helper doesn't bracket any actual work, the destructor will
// clean up annotations that were created during ActOnEndOfTranslationUnit
diff --git a/clang/test/Frontend/plugin-delayed-template.cpp b/clang/test/Frontend/plugin-delayed-template.cpp
new file mode 100644
index 00000000000..c57ec307689
--- /dev/null
+++ b/clang/test/Frontend/plugin-delayed-template.cpp
@@ -0,0 +1,17 @@
+// RUN: %clang_cc1 -fdelayed-template-parsing -load %llvmshlibdir/PrintFunctionNames%pluginext -plugin print-fns -plugin-arg-print-fns -parse-template -plugin-arg-print-fns ForcedTemplate %s 2>&1 | FileCheck %s
+// REQUIRES: plugins, examples
+
+template <typename T>
+void TemplateDep();
+
+// CHECK: top-level-decl: "ForcedTemplate"
+// The plugin should force parsing of this template, even though it's
+// not used and -fdelayed-template-parsing is specified.
+// CHECK: warning: expression result unused
+// CHECK: late-parsed-decl: "ForcedTemplate"
+template <typename T>
+void ForcedTemplate() {
+ TemplateDep<T>(); // Shouldn't crash.
+
+ ""; // Triggers the warning checked for above.
+}
OpenPOWER on IntegriCloud