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
|