diff --git a/clang/test/Tooling/clang-check-syntax-tree-dump.cpp b/clang/test/Tooling/clang-check-syntax-tree-dump.cpp new file mode 100644 --- /dev/null +++ b/clang/test/Tooling/clang-check-syntax-tree-dump.cpp @@ -0,0 +1,9 @@ +// RUN: clang-check -syntax-tree-dump "%s" -- 2>&1 | FileCheck %s +int abc; +// CHECK: TranslationUnit Detached +// CHECK-NEXT: `-SimpleDeclaration +// CHECK-NEXT: |-'int' +// CHECK-NEXT: |-DeclaratorList Declarators +// CHECK-NEXT: | `-SimpleDeclarator ListElement +// CHECK-NEXT: | `-'abc' +// CHECK-NEXT: `-';' diff --git a/clang/tools/clang-check/CMakeLists.txt b/clang/tools/clang-check/CMakeLists.txt --- a/clang/tools/clang-check/CMakeLists.txt +++ b/clang/tools/clang-check/CMakeLists.txt @@ -18,4 +18,5 @@ clangSerialization clangStaticAnalyzerFrontend clangTooling + clangToolingSyntax ) diff --git a/clang/tools/clang-check/ClangCheck.cpp b/clang/tools/clang-check/ClangCheck.cpp --- a/clang/tools/clang-check/ClangCheck.cpp +++ b/clang/tools/clang-check/ClangCheck.cpp @@ -24,6 +24,9 @@ #include "clang/Rewrite/Frontend/FrontendActions.h" #include "clang/StaticAnalyzer/Frontend/FrontendActions.h" #include "clang/Tooling/CommonOptionsParser.h" +#include "clang/Tooling/Syntax/BuildTree.h" +#include "clang/Tooling/Syntax/Tokens.h" +#include "clang/Tooling/Syntax/Tree.h" #include "clang/Tooling/Tooling.h" #include "llvm/ADT/STLExtras.h" #include "llvm/Option/OptTable.h" @@ -82,6 +85,10 @@ cl::desc(Options.getOptionHelpText(options::OPT_fix_what_you_can)), cl::cat(ClangCheckCategory)); +static cl::opt SyntaxTreeDump("syntax-tree-dump", + cl::desc("dump the syntax tree"), + cl::cat(ClangCheckCategory)); + namespace { // FIXME: Move FixItRewriteInPlace from lib/Rewrite/Frontend/FrontendActions.cpp @@ -131,6 +138,28 @@ } }; +class DumpSyntaxTree : public clang::ASTFrontendAction { +public: + std::unique_ptr + CreateASTConsumer(clang::CompilerInstance &CI, StringRef InFile) override { + class Consumer : public clang::ASTConsumer { + public: + Consumer(clang::CompilerInstance &CI) : Collector(CI.getPreprocessor()) {} + + void HandleTranslationUnit(clang::ASTContext &AST) override { + clang::syntax::TokenBuffer TB = std::move(Collector).consume(); + clang::syntax::Arena A(AST.getSourceManager(), AST.getLangOpts(), TB); + llvm::outs() << clang::syntax::buildSyntaxTree(A, AST)->dump( + AST.getSourceManager()); + } + + private: + clang::syntax::TokenCollector Collector; + }; + return std::make_unique(CI); + } +}; + class ClangCheckActionFactory { public: std::unique_ptr newASTConsumer() { @@ -188,6 +217,8 @@ FrontendFactory = newFrontendActionFactory(); else if (Fixit) FrontendFactory = newFrontendActionFactory(); + else if (SyntaxTreeDump) + FrontendFactory = newFrontendActionFactory(); else FrontendFactory = newFrontendActionFactory(&CheckFactory);