summaryrefslogtreecommitdiffstats
path: root/build/clang-plugin/CanRunScriptChecker.h
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--build/clang-plugin/CanRunScriptChecker.h31
1 files changed, 31 insertions, 0 deletions
diff --git a/build/clang-plugin/CanRunScriptChecker.h b/build/clang-plugin/CanRunScriptChecker.h
new file mode 100644
index 0000000000..4516609999
--- /dev/null
+++ b/build/clang-plugin/CanRunScriptChecker.h
@@ -0,0 +1,31 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#ifndef CanRunScriptChecker_h__
+#define CanRunScriptChecker_h__
+
+#include "plugin.h"
+#include <unordered_set>
+
+class CanRunScriptChecker : public BaseCheck {
+public:
+ CanRunScriptChecker(StringRef CheckName, ContextType *Context = nullptr)
+ : BaseCheck(CheckName, Context) {}
+ void registerMatchers(MatchFinder *AstMatcher) override;
+ void check(const MatchFinder::MatchResult &Result) override;
+
+ // Simply initialize the can-run-script function set at the beginning of each
+ // translation unit.
+ void onStartOfTranslationUnit() override;
+
+private:
+ /// Runs the inner matcher on the AST to find all the can-run-script
+ /// functions using custom rules (not only the annotation).
+ void buildFuncSet(ASTContext *Context);
+
+ bool IsFuncSetBuilt;
+ std::unordered_set<const FunctionDecl *> CanRunScriptFuncs;
+};
+
+#endif