Kostin Artem. Lab #1. Var 3. (#15)

* warning plugin

* fixed

* add -i option

* fix clang-tidy

* fix CMakeLists

* fix clang-tidy 2

---------

Co-authored-by: m-ly4 <[email protected]>

Author: aartyomm

clang/labs/lab1/kostin_artem/CMakeLists.txt

@@ -0,0 +1,14 @@
+add_llvm_library(deprWarnPlugin MODULE depr_plugin.cpp PLUGIN_TOOL clang)
+
+if(WIN32 OR CYGWIN)
+  set(LLVM_LINK_COMPONENTS
+    Support
+  )
+  clang_target_link_libraries(deprWarnPlugin PRIVATE
+    clangAST
+    clangBasic
+    clangFrontend
+    )
+endif()
+
+set(CLANG_TEST_DEPS "deprWarnPlugin" ${CLANG_TEST_DEPS} PARENT_SCOPE)

clang/labs/lab1/kostin_artem/depr_plugin.cpp

@@ -0,0 +1,64 @@
+#include "clang/AST/ASTConsumer.h"
+#include "clang/AST/RecursiveASTVisitor.h"
+#include "clang/Frontend/CompilerInstance.h"
+#include "clang/Frontend/FrontendPluginRegistry.h"
+
+using namespace clang;
+
+class CustomNodeVisitor : public RecursiveASTVisitor<CustomNodeVisitor> {
+  bool CaseInsensitive;
+
+public:
+  CustomNodeVisitor(bool CaseInsensitive) : CaseInsensitive(CaseInsensitive) {}
+  bool VisitFunctionDecl(FunctionDecl *Pfunction) { // NOLINT
+    std::string NameOfFunction = Pfunction->getNameInfo().getAsString();
+    if (CaseInsensitive) {
+      std::transform(NameOfFunction.begin(), NameOfFunction.end(),
+                     NameOfFunction.begin(), ::tolower);
+    }
+    if (NameOfFunction.find("deprecated") != std::string::npos) {
+      DiagnosticsEngine &Diagnostics =
+          Pfunction->getASTContext().getDiagnostics();
+      unsigned int DiagnosticsId = Diagnostics.getCustomDiagID(
+          DiagnosticsEngine::Warning,
+          "The function name contains \"deprecated\"");
+      SourceLocation PositionOfFunction = Pfunction->getLocation();
+      Diagnostics.Report(PositionOfFunction, DiagnosticsId) << NameOfFunction;
+    }
+    return true;
+  }
+};
+
+class CustomConsumer : public ASTConsumer {
+  bool CaseInsensitive;
+
+public:
+  explicit CustomConsumer(bool CaseInsensitive)
+      : CaseInsensitive(CaseInsensitive) {}
+  void HandleTranslationUnit(ASTContext &Context) override {
+    CustomNodeVisitor Cnv(CaseInsensitive);
+    Cnv.TraverseDecl(Context.getTranslationUnitDecl());
+  }
+};
+
+class PluginDeprFunc : public PluginASTAction {
+  bool CaseInsensitive = false;
+  std::unique_ptr<ASTConsumer>
+  CreateASTConsumer(CompilerInstance &Instance,
+                    llvm::StringRef InFile) override {
+    return std::make_unique<CustomConsumer>(CaseInsensitive);
+  }
+  bool ParseArgs(const CompilerInstance &Compiler,
+                 const std::vector<std::string> &Args) override {
+    for (const auto &arg : Args) {
+      if (arg == "-i") {
+        CaseInsensitive = true;
+      }
+    }
+    return true;
+  }
+};
+
+static FrontendPluginRegistry::Add<PluginDeprFunc>
+    X("plugin_for_deprecated_functions",
+      "If the function name contains \"deprecated\" plugin writes a warning");

clang/test/lab1/kostin_artem/depr_plugin_test.cpp

@@ -0,0 +1,40 @@
+// RUN: split-file %s %t
+// RUN: %clang_cc1 -load %llvmshlibdir/deprWarnPlugin%pluginext -plugin plugin_for_deprecated_functions %t/with_-i.cpp -plugin-arg-plugin_for_deprecated_functions -i 2>&1 | FileCheck %t/with_-i.cpp
+// RUN: %clang_cc1 -load %llvmshlibdir/deprWarnPlugin%pluginext -plugin plugin_for_deprecated_functions %t/without_-i.cpp 2>&1 | FileCheck %t/without_-i.cpp
+
+
+//--- with_-i.cpp
+
+// CHECK: warning: The function name contains "deprecated"
+void deprecatedFunction();
+
+// CHECK: warning: The function name contains "deprecated"
+void functionWithDeprecatedWord();
+
+// CHECK-NOT: warning: The function name contains "deprecated"
+void regularFunction();
+
+class SomeClass {
+// CHECK: warning: The function name contains "deprecated"
+  void functionWithDePrEcAtEdWord();
+// CHECK-NOT: warning: The function name contains "deprecated"
+  void regularFunctionAgain();
+};
+
+//--- without_-i.cpp
+
+// CHECK: warning: The function name contains "deprecated"
+void deprecatedFunction2();
+
+// CHECK-NOT: warning: The function name contains "deprecated"
+void functionWithDeprecatedWord2();
+
+// CHECK-NOT: warning: The function name contains "deprecated"
+void regularFunction2();
+
+class SomeClass2 {
+// CHECK-NOT: warning: The function name contains "deprecated"
+  void functionWithDePrEcAtEdWord2();
+// CHECK-NOT: warning: The function name contains "deprecated"
+  void regularFunctionAgain2();
+};