1 /* 2 * Copyright (C) 2016 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #ifndef ART_LIBDEXFILE_DEX_DEX_CACHE_RESOLVED_CLASSES_H_ 18 #define ART_LIBDEXFILE_DEX_DEX_CACHE_RESOLVED_CLASSES_H_ 19 20 #include <string> 21 #include <unordered_set> 22 #include <vector> 23 24 #include "dex/dex_file_types.h" 25 26 namespace art { 27 28 // Data structure for passing around which classes belonging to a dex cache / dex file are resolved. 29 class DexCacheResolvedClasses { 30 public: DexCacheResolvedClasses(const std::string & dex_location,const std::string & base_location,uint32_t location_checksum,uint32_t num_method_ids)31 DexCacheResolvedClasses(const std::string& dex_location, 32 const std::string& base_location, 33 uint32_t location_checksum, 34 uint32_t num_method_ids) 35 : dex_location_(dex_location), 36 base_location_(base_location), 37 location_checksum_(location_checksum), 38 num_method_ids_(num_method_ids) {} 39 40 // Only compare the key elements, ignore the resolved classes. Compare(const DexCacheResolvedClasses & other)41 int Compare(const DexCacheResolvedClasses& other) const { 42 if (location_checksum_ != other.location_checksum_) { 43 return static_cast<int>(location_checksum_ - other.location_checksum_); 44 } 45 // Don't need to compare base_location_ since dex_location_ has more info. 46 return dex_location_.compare(other.dex_location_); 47 } 48 AddClass(dex::TypeIndex index)49 bool AddClass(dex::TypeIndex index) const { 50 return classes_.insert(index).second; 51 } 52 53 template <class InputIt> AddClasses(InputIt begin,InputIt end)54 void AddClasses(InputIt begin, InputIt end) const { 55 classes_.insert(begin, end); 56 } 57 GetDexLocation()58 const std::string& GetDexLocation() const { 59 return dex_location_; 60 } 61 GetBaseLocation()62 const std::string& GetBaseLocation() const { 63 return base_location_; 64 } 65 GetLocationChecksum()66 uint32_t GetLocationChecksum() const { 67 return location_checksum_; 68 } 69 GetClasses()70 const std::unordered_set<dex::TypeIndex>& GetClasses() const { 71 return classes_; 72 } 73 NumMethodIds()74 size_t NumMethodIds() const { 75 return num_method_ids_; 76 } 77 78 private: 79 const std::string dex_location_; 80 const std::string base_location_; 81 const uint32_t location_checksum_; 82 const uint32_t num_method_ids_; 83 // Array of resolved class def indexes. 84 mutable std::unordered_set<dex::TypeIndex> classes_; 85 }; 86 87 inline bool operator<(const DexCacheResolvedClasses& a, const DexCacheResolvedClasses& b) { 88 return a.Compare(b) < 0; 89 } 90 91 } // namespace art 92 93 #endif // ART_LIBDEXFILE_DEX_DEX_CACHE_RESOLVED_CLASSES_H_ 94