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