summaryrefslogtreecommitdiffstats
path: root/build/clang-plugin/BaseCheck.h
blob: 867b82d2ad353e07b6beef8842e57cf12d81207f (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
32
33
34
/* 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 BaseCheck_h__
#define BaseCheck_h__

class MozContext {};
typedef MozContext ContextType;

class BaseCheck : public MatchFinder::MatchCallback {
public:
  BaseCheck(StringRef CheckName, ContextType *Context) {}
  virtual void registerMatchers(MatchFinder *Finder) {}
  virtual void registerPPCallbacks(CompilerInstance &CI) {}
  virtual void check(const MatchFinder::MatchResult &Result) {}
  DiagnosticBuilder diag(SourceLocation Loc, StringRef Description,
                         DiagnosticIDs::Level Level = DiagnosticIDs::Warning) {
    DiagnosticsEngine &Diag = Context->getDiagnostics();
    unsigned ID = Diag.getDiagnosticIDs()->getCustomDiagID(Level, Description);
    return Diag.Report(Loc, ID);
  }

private:
  void run(const MatchFinder::MatchResult &Result) override {
    Context = Result.Context;
    check(Result);
  }

private:
  ASTContext *Context;
};

#endif