summaryrefslogtreecommitdiffstats
path: root/build/clang-plugin/CanRunScriptChecker.h
blob: 45166099997e2b65e78f9de304c9c25336f61b5c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
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