update

1 files changed, 234 insertions, 0 deletions

diff --git a/cgrep.cpp b/cgrep.cpp
new file mode 100644
index 0000000..4f66876
--- /dev/null
+++ b/cgrep.cpp
@@ -0,0 +1,234 @@
+
+/*first line intentionally left blank.*/
+/*************************************************************************************************/
+//-*-c++-*-
+/*Copyright (C) 2018 Farzad Sadeghi
+ * Licensed under GPL-3.0
+ * */
+/*************************************************************************************************/
+/*included modules*/
+/*project headers*/
+/*standard headers*/
+#include <cassert>
+#include <cstdlib>
+#include <dirent.h>
+#include <fstream>
+#include <iostream>
+#include <regex>
+#include <string>
+#include <vector>
+/*LLVM headers*/
+#include "clang/AST/AST.h"
+#include "clang/AST/ASTConsumer.h"
+#include "clang/ASTMatchers/ASTMatchFinder.h"
+#include "clang/ASTMatchers/ASTMatchers.h"
+#include "clang/Basic/LLVM.h"
+#include "clang/Frontend/CompilerInstance.h"
+#include "clang/Frontend/FrontendActions.h"
+#include "clang/Lex/Lexer.h"
+#include "clang/Rewrite/Core/Rewriter.h"
+#include "clang/Tooling/CommonOptionsParser.h"
+#include "clang/Tooling/Tooling.h"
+#include "llvm/Support/raw_ostream.h"
+/*************************************************************************************************/
+/*used namespaces*/
+using namespace llvm;
+using namespace clang;
+using namespace clang::ast_matchers;
+using namespace clang::driver;
+using namespace clang::tooling;
+/*************************************************************************************************/
+namespace {
+static llvm::cl::OptionCategory CGrepCat("cgrep options");
+cl::opt<std::string> CO_DIRECTORY("dir", cl::desc(""), cl::init(""), cl::cat(CGrepCat), cl::Optional);
+cl::opt<std::string> CO_REGEX("regex", cl::desc(""), cl::init(""), cl::cat(CGrepCat), cl::Optional);
+cl::opt<bool> CO_FUNCTION("func", cl::desc(""), cl::init(false), cl::cat(CGrepCat), cl::Optional);
+cl::opt<bool> CO_MEM_FUNCTION("memfunc", cl::desc(""), cl::init(false), cl::cat(CGrepCat), cl::Optional);
+cl::opt<bool> CO_VAR("var", cl::desc(""), cl::init(false), cl::cat(CGrepCat), cl::Optional);
+cl::opt<bool> CO_MEMVAR("memvar", cl::desc(""), cl::init(false), cl::cat(CGrepCat), cl::Optional);
+cl::opt<bool> CO_CLASS("class", cl::desc(""), cl::init(false), cl::cat(CGrepCat), cl::Optional);
+}
+/*************************************************************************************************/
+std::string regex_preprocessor(std::string rx_str) {
+  std::string ret_rx_str;
+  return ret_rx_str;
+}
+
+void regex_handlaer(std::string rx_str) {
+  std::regex regex(regex_preprocessor(rx_str));
+  return void();
+}
+/*************************************************************************************************/
+class CalledFunc : public MatchFinder::MatchCallback {
+public:
+  CalledFunc(Rewriter &Rewrite) : Rewrite(Rewrite) {}
+
+  virtual void run(const MatchFinder::MatchResult &MR) {
+    const FunctionDecl *FD = MR.Nodes.getNodeAs<clang::FunctionDecl>("funcdecl");
+    if (FD) {
+      SourceRange SR = FD->getSourceRange();
+      std::string name = FD->getNameAsString();
+    }
+  }
+
+private:
+  Rewriter &Rewrite [[maybe_unused]];
+};
+/*************************************************************************************************/
+class CalledVar : public MatchFinder::MatchCallback {
+public:
+  CalledVar(Rewriter &Rewrite) : Rewrite(Rewrite) {}
+
+  virtual void run(const MatchFinder::MatchResult &MR) {
+    const VarDecl *VD = MR.Nodes.getNodeAs<clang::VarDecl>("vardecl");
+    if (VD) {
+      SourceRange SR = VD->getSourceRange();
+      std::string name = VD->getNameAsString();
+    }
+  }
+
+private:
+  Rewriter &Rewrite [[maybe_unused]];
+};
+/*************************************************************************************************/
+class FuncDecl : public MatchFinder::MatchCallback {
+public:
+  FuncDecl(Rewriter &Rewrite) : Rewrite(Rewrite) {}
+
+  virtual void run(const MatchFinder::MatchResult &MR) {}
+
+private:
+  Rewriter &Rewrite [[maybe_unused]];
+};
+/*************************************************************************************************/
+class VDecl : public MatchFinder::MatchCallback {
+public:
+  VDecl(Rewriter &Rewrite) : Rewrite(Rewrite) {}
+
+  virtual void run(const MatchFinder::MatchResult &MR) {}
+
+private:
+  Rewriter &Rewrite [[maybe_unused]];
+};
+/*************************************************************************************************/
+class ClassDecl : public MatchFinder::MatchCallback {
+public:
+  ClassDecl(Rewriter &Rewrite) : Rewrite(Rewrite) {}
+
+  virtual void run(const MatchFinder::MatchResult &MR) {}
+
+private:
+  Rewriter &Rewrite [[maybe_unused]];
+};
+/*************************************************************************************************/
+class PPInclusion : public PPCallbacks {
+public:
+  explicit PPInclusion(SourceManager *SM, Rewriter *Rewrite)
+      : SM(*SM), Rewrite(*Rewrite) {}
+
+  virtual void MacroDefined(const Token &MacroNameTok,
+                            const MacroDirective *MD) {}
+
+  virtual void MacroExpands(const Token &MacroNameTok,
+                            const MacroDefinition &MD, SourceRange Range,
+                            const MacroArgs *Args) {}
+
+#if __clang_major__ <= 6
+  virtual void InclusionDirective(SourceLocation HashLoc,
+                                  const Token &IncludeTok, StringRef FileName,
+                                  bool IsAngled, CharSourceRange FilenameRange,
+                                  const FileEntry *File, StringRef SearchPath,
+                                  StringRef RelativePath,
+                                  const clang::Module *Imported) {
+#elif __clang_major__ >= 8
+  virtual void InclusionDirective(SourceLocation HashLoc,
+                                  const Token &IncludeTok, StringRef FileName,
+                                  bool IsAngled, CharSourceRange FilenameRange,
+                                  const FileEntry *File, StringRef SearchPath,
+                                  StringRef RelativePath,
+                                  const clang::Module *Imported,
+                                  SrcMgr::CharacteristicKind FileType) {
+#endif
+  }
+
+private:
+  const SourceManager &SM [[maybe_unused]];
+  Rewriter &Rewrite [[maybe_unused]];
+};
+/*************************************************************************************************/
+// brief A Clang Diagnostic Consumer that does nothing.
+class BlankDiagConsumer : public clang::DiagnosticConsumer {
+public:
+  BlankDiagConsumer() = default;
+  virtual ~BlankDiagConsumer() {}
+  virtual void HandleDiagnostic(DiagnosticsEngine::Level DiagLevel,
+                                const Diagnostic &Info) override {}
+};
+/*************************************************************************************************/
+class MyASTConsumer : public ASTConsumer {
+public:
+  MyASTConsumer(Rewriter &R)
+      : funcDeclHandler(R), HandlerForVar(R), HandlerForClass(R),
+        HandlerForCalledFunc(R), HandlerForCalledVar(R) {
+#if 1
+    Matcher.addMatcher(functionDecl().bind("funcdecl"), &funcDeclHandler);
+    Matcher.addMatcher(
+        varDecl(anyOf(unless(hasDescendant(expr(anything()))),
+                      hasDescendant(expr(anything()).bind("expr"))))
+            .bind("vardecl"),
+        &HandlerForVar);
+    Matcher.addMatcher(recordDecl(isClass()).bind("classdecl"),
+                       &HandlerForClass);
+    Matcher.addMatcher(declRefExpr().bind("calledvar"), &HandlerForCalledVar);
+#endif
+  }
+
+  void HandleTranslationUnit(ASTContext &Context) override {
+    Matcher.matchAST(Context);
+  }
+
+private:
+  FuncDecl funcDeclHandler;
+  VDecl HandlerForVar;
+  ClassDecl HandlerForClass;
+  CalledFunc HandlerForCalledFunc;
+  CalledVar HandlerForCalledVar;
+  MatchFinder Matcher;
+};
+/*************************************************************************************************/
+class AppFrontendAction : public ASTFrontendAction {
+public:
+  AppFrontendAction() {}
+  ~AppFrontendAction() { delete BDCProto; }
+
+  void EndSourceFileAction() override {
+    std::error_code EC;
+    TheRewriter.getEditBuffer(TheRewriter.getSourceMgr().getMainFileID())
+        .write(llvm::outs());
+  }
+
+  std::unique_ptr<ASTConsumer> CreateASTConsumer(CompilerInstance &CI,
+                                                 StringRef file) override {
+    CI.getPreprocessor().addPPCallbacks(
+        llvm::make_unique<PPInclusion>(&CI.getSourceManager(), &TheRewriter));
+    DiagnosticsEngine &DE = CI.getPreprocessor().getDiagnostics();
+    DE.setClient(BDCProto, false);
+    TheRewriter.setSourceMgr(CI.getSourceManager(), CI.getLangOpts());
+    return llvm::make_unique<MyASTConsumer>(TheRewriter);
+  }
+
+private:
+  BlankDiagConsumer *BDCProto = new BlankDiagConsumer;
+  Rewriter TheRewriter;
+};
+/*************************************************************************************************/
+/*************************************************************************************************/
+/*Main*/
+int main(int argc, const char **argv) {
+  CommonOptionsParser op(argc, argv, CGrepCat);
+  const std::vector<std::string> &SourcePathList [[maybe_unused]] = op.getSourcePathList();
+  ClangTool Tool(op.getCompilations(), op.getSourcePathList());
+  int ret = Tool.run(newFrontendActionFactory<AppFrontendAction>().get());
+  return ret;
+}
+/*************************************************************************************************/