Home | History | Annotate | Download | only in plugins
      1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #ifndef TOOLS_CLANG_PLUGINS_CHROMECLASSTESTER_H_
      6 #define TOOLS_CLANG_PLUGINS_CHROMECLASSTESTER_H_
      7 
      8 #include <set>
      9 #include <vector>
     10 
     11 #include "clang/AST/ASTConsumer.h"
     12 #include "clang/AST/TypeLoc.h"
     13 #include "clang/Frontend/CompilerInstance.h"
     14 
     15 // A class on top of ASTConsumer that forwards classes defined in Chromium
     16 // headers to subclasses which implement CheckChromeClass().
     17 class ChromeClassTester : public clang::ASTConsumer {
     18  public:
     19   explicit ChromeClassTester(clang::CompilerInstance& instance);
     20   virtual ~ChromeClassTester();
     21 
     22   // clang::ASTConsumer:
     23   virtual void HandleTagDeclDefinition(clang::TagDecl* tag);
     24   virtual bool HandleTopLevelDecl(clang::DeclGroupRef group_ref);
     25 
     26  protected:
     27   clang::CompilerInstance& instance() { return instance_; }
     28   clang::DiagnosticsEngine& diagnostic() { return diagnostic_; }
     29 
     30   // Emits a simple warning; this shouldn't be used if you require printf-style
     31   // printing.
     32   void emitWarning(clang::SourceLocation loc, const char* error);
     33 
     34   // Utility method for subclasses to check if this class is in a banned
     35   // namespace.
     36   bool InBannedNamespace(const clang::Decl* record);
     37 
     38   // Utility method for subclasses to determine the namespace of the
     39   // specified record, if any. Unnamed namespaces will be identified as
     40   // "<anonymous namespace>".
     41   std::string GetNamespace(const clang::Decl* record);
     42 
     43   // Utility method for subclasses to check if this class is within an
     44   // implementation (.cc, .cpp, .mm) file.
     45   bool InImplementationFile(clang::SourceLocation location);
     46 
     47  private:
     48   void BuildBannedLists();
     49 
     50   void CheckTag(clang::TagDecl*);
     51 
     52   // Filtered versions of tags that are only called with things defined in
     53   // chrome header files.
     54   virtual void CheckChromeClass(clang::SourceLocation record_location,
     55                                 clang::CXXRecordDecl* record) = 0;
     56 
     57   // Filtered versions of enum type that are only called with things defined
     58   // in chrome header files.
     59   virtual void CheckChromeEnum(clang::SourceLocation enum_location,
     60                                clang::EnumDecl* enum_decl) {
     61   }
     62 
     63   // Utility methods used for filtering out non-chrome classes (and ones we
     64   // deliberately ignore) in HandleTagDeclDefinition().
     65   std::string GetNamespaceImpl(const clang::DeclContext* context,
     66                                const std::string& candidate);
     67   bool InBannedDirectory(clang::SourceLocation loc);
     68   bool IsIgnoredType(const std::string& base_name);
     69 
     70   // Attempts to determine the filename for the given SourceLocation.
     71   // Returns false if the filename could not be determined.
     72   bool GetFilename(clang::SourceLocation loc, std::string* filename);
     73 
     74   clang::CompilerInstance& instance_;
     75   clang::DiagnosticsEngine& diagnostic_;
     76 
     77   // List of banned namespaces.
     78   std::vector<std::string> banned_namespaces_;
     79 
     80   // List of banned directories.
     81   std::vector<std::string> banned_directories_;
     82 
     83   // List of types that we don't check.
     84   std::set<std::string> ignored_record_names_;
     85 
     86   // List of decls to check once the current top-level decl is parsed.
     87   std::vector<clang::TagDecl*> pending_class_decls_;
     88 };
     89 
     90 #endif  // TOOLS_CLANG_PLUGINS_CHROMECLASSTESTER_H_
     91