build/clang-plugin/CanRunScriptChecker.h
author Dimi Lee <dlee@mozilla.com>
Sat, 08 Aug 2020 06:00:00 +0000
changeset 543991 fa0dbdf15f291e814b4854d515d7ef3e4548b7fb
parent 387474 88c9b8587db69f36601f45e6584ac2d1c5e459b5
permissions -rw-r--r--
Bug 1658010 - Add null pointer check before notifying content block event r=xeonchen Differential Revision: https://phabricator.services.mozilla.com/D86421

/* 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