| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2010 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 // Clang plugin which finds types that are affected if the types listed in the | |
| 6 // plugin parameters are changed. This is for use in determining what PPAPI | |
| 7 // C-level interfaces are affected if one or more PPAPI C structs are changed. | |
| 8 | |
| 9 #include <algorithm> | |
| 10 #include <cstdio> | |
| 11 #include <map> | |
| 12 #include <sstream> | |
| 13 #include <string> | |
| 14 #include <vector> | |
| 15 | |
| 16 #include "clang/AST/AST.h" | |
| 17 #include "clang/AST/ASTConsumer.h" | |
| 18 #include "clang/AST/CharUnits.h" | |
| 19 #include "clang/Basic/SourceManager.h" | |
| 20 #include "clang/Basic/TargetInfo.h" | |
| 21 #include "clang/Frontend/CompilerInstance.h" | |
| 22 #include "clang/Frontend/FrontendPluginRegistry.h" | |
| 23 | |
| 24 namespace { | |
| 25 | |
| 26 typedef std::vector<std::string> StringVec; | |
| 27 | |
| 28 class FindAffectedInterfacesConsumer : public clang::ASTConsumer { | |
| 29 public: | |
| 30 explicit FindAffectedInterfacesConsumer(const StringVec& changed_classes) { | |
| 31 StringVec::const_iterator iter(changed_classes.begin()); | |
| 32 StringVec::const_iterator the_end(changed_classes.end()); | |
| 33 for (; iter != the_end; ++iter) { | |
| 34 class_is_affected_map_[*iter] = true; | |
| 35 } | |
| 36 } | |
| 37 | |
| 38 private: | |
| 39 typedef std::map<std::string, bool> StringBoolMap; | |
| 40 StringBoolMap class_is_affected_map_; | |
| 41 | |
| 42 bool IsAffected(const clang::Type& type_to_check) { | |
| 43 std::string type_string( | |
| 44 type_to_check.getCanonicalTypeInternal().getAsString()); | |
| 45 std::pair<StringBoolMap::iterator, bool> iter_success_pair = | |
| 46 class_is_affected_map_.insert( | |
| 47 StringBoolMap::value_type(type_string, false)); | |
| 48 StringBoolMap::iterator iter(iter_success_pair.first); | |
| 49 bool successfully_inserted(iter_success_pair.second); | |
| 50 // If we were able to insert, then we haven't yet seen this type. Compute | |
| 51 // IsAffected and put it at the newly inserted location in the map. | |
| 52 if (successfully_inserted) { | |
| 53 if (type_to_check.isPointerType()) { | |
| 54 const clang::PointerType* pointer_type = | |
| 55 dyn_cast<clang::PointerType>(&type_to_check); | |
| 56 // Recurse to the pointee type. | |
| 57 iter->second = IsAffected(*pointer_type->getPointeeType().getTypePtr()); | |
| 58 } else if (type_to_check.isFunctionProtoType()) { | |
| 59 const clang::FunctionProtoType* func_type = | |
| 60 dyn_cast<clang::FunctionProtoType>(&type_to_check); | |
| 61 // Recurse to the return type and parameter types. | |
| 62 iter->second = IsAffected(*func_type->getResultType().getTypePtr()); | |
| 63 if (!iter->second) { | |
| 64 clang::FunctionProtoType::arg_type_iterator arg_iter = | |
| 65 func_type->arg_type_begin(); | |
| 66 clang::FunctionProtoType::arg_type_iterator arg_end = | |
| 67 func_type->arg_type_end(); | |
| 68 for (; (arg_iter != arg_end) && (!iter->second); ++arg_iter) { | |
| 69 iter->second = IsAffected(*(arg_iter->getTypePtr())); | |
| 70 } | |
| 71 } | |
| 72 } else if (type_to_check.isRecordType()) { | |
| 73 // For records (unions, structs), recurse to the fields. | |
| 74 const clang::RecordType* record = | |
| 75 dyn_cast<clang::RecordType>(&type_to_check); | |
| 76 const clang::RecordDecl* decl = record->getDecl(); | |
| 77 clang::RecordDecl::field_iterator field_iter(decl->field_begin()); | |
| 78 clang::RecordDecl::field_iterator field_end(decl->field_end()); | |
| 79 for (; (field_iter != field_end) && (!iter->second); ++field_iter) { | |
| 80 iter->second = IsAffected(*(field_iter->getType().getTypePtr())); | |
| 81 } | |
| 82 } | |
| 83 } | |
| 84 // By this point, the bool in the map at the location referenced by iter has | |
| 85 // the correct value. Either it was cached, or we computed & inserted it. | |
| 86 return iter->second; | |
| 87 } | |
| 88 | |
| 89 // Virtual function to consume top-level declarations. For each one, we check | |
| 90 // to see if it is a type definition. If it is, we print information about | |
| 91 // it. | |
| 92 virtual void HandleTopLevelDecl(clang::DeclGroupRef decl_group) { | |
| 93 clang::DeclGroupRef::iterator iter(decl_group.begin()); | |
| 94 clang::DeclGroupRef::iterator the_end(decl_group.end()); | |
| 95 for (; iter != the_end; ++iter) { | |
| 96 const clang::Decl *decl = *iter; | |
| 97 if (const clang::TypeDecl* type_decl = dyn_cast<clang::TypeDecl>(decl)) { | |
| 98 std::string name(type_decl->getNameAsString()); | |
| 99 // TagDecl covers structs, enums, unions, and classes. | |
| 100 if (const clang::TagDecl* tag = dyn_cast<clang::TagDecl>(type_decl)) { | |
| 101 // Only print out info when we find the definition; ignore forward | |
| 102 // references. | |
| 103 if (tag->isDefinition()) { | |
| 104 clang::Type* type = type_decl->getTypeForDecl(); | |
| 105 if (IsAffected(*type)) { | |
| 106 std::printf("%s\n", name.c_str()); | |
| 107 } | |
| 108 } | |
| 109 } else if (const clang::TypedefDecl* td = | |
| 110 dyn_cast<clang::TypedefDecl>(type_decl)) { | |
| 111 clang::Type* type = td->getUnderlyingType().getTypePtr(); | |
| 112 if (IsAffected(*type)) { | |
| 113 std::printf("%s\n", name.c_str()); | |
| 114 } | |
| 115 } | |
| 116 } | |
| 117 } | |
| 118 } | |
| 119 }; | |
| 120 | |
| 121 class FindAffectedInterfacesAction : public clang::PluginASTAction { | |
| 122 public: | |
| 123 FindAffectedInterfacesAction() {} | |
| 124 | |
| 125 private: | |
| 126 StringVec types_; | |
| 127 | |
| 128 virtual clang::ASTConsumer *CreateASTConsumer( | |
| 129 clang::CompilerInstance &instance, llvm::StringRef /*input_file*/) { | |
| 130 return new FindAffectedInterfacesConsumer(types_); | |
| 131 } | |
| 132 | |
| 133 virtual bool ParseArgs(const clang::CompilerInstance& /*instance*/, | |
| 134 const std::vector<std::string>& args) { | |
| 135 // Every argument is interpreted as a comma-delimited list of names of types | |
| 136 // that have been changed. | |
| 137 StringVec::const_iterator iter(args.begin()), end(args.end()); | |
| 138 for (; iter != end; ++iter) { | |
| 139 std::stringstream stream(*iter); | |
| 140 std::string type_name; | |
| 141 while (std::getline(stream, type_name, ',')) { | |
| 142 types_.push_back(type_name); | |
| 143 } | |
| 144 } | |
| 145 return true; | |
| 146 } | |
| 147 }; | |
| 148 | |
| 149 } // namespace | |
| 150 | |
| 151 static clang::FrontendPluginRegistry::Add<FindAffectedInterfacesAction> | |
| 152 X("FindAffectedInterfaces", | |
| 153 "Find interfaces affected by changes to the passes classes."); | |
| 154 | |
| OLD | NEW |