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_RUNTIME_DEX_CACHE_RESOLVED_CLASSES_H_
18 #define ART_RUNTIME_DEX_CACHE_RESOLVED_CLASSES_H_
19 
20 #include <string>
21 #include <unordered_set>
22 #include <vector>
23 
24 #include "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)31   DexCacheResolvedClasses(const std::string& dex_location,
32                           const std::string& base_location,
33                           uint32_t location_checksum)
34       : dex_location_(dex_location),
35         base_location_(base_location),
36         location_checksum_(location_checksum) {}
37 
38   // Only compare the key elements, ignore the resolved classes.
Compare(const DexCacheResolvedClasses & other)39   int Compare(const DexCacheResolvedClasses& other) const {
40     if (location_checksum_ != other.location_checksum_) {
41       return static_cast<int>(location_checksum_ - other.location_checksum_);
42     }
43     // Don't need to compare base_location_ since dex_location_ has more info.
44     return dex_location_.compare(other.dex_location_);
45   }
46 
AddClass(dex::TypeIndex index)47   bool AddClass(dex::TypeIndex index) const {
48     return classes_.insert(index).second;
49   }
50 
51   template <class InputIt>
AddClasses(InputIt begin,InputIt end)52   void AddClasses(InputIt begin, InputIt end) const {
53     classes_.insert(begin, end);
54   }
55 
GetDexLocation()56   const std::string& GetDexLocation() const {
57     return dex_location_;
58   }
59 
GetBaseLocation()60   const std::string& GetBaseLocation() const {
61     return base_location_;
62   }
63 
GetLocationChecksum()64   uint32_t GetLocationChecksum() const {
65     return location_checksum_;
66   }
67 
GetClasses()68   const std::unordered_set<dex::TypeIndex>& GetClasses() const {
69     return classes_;
70   }
71 
72  private:
73   const std::string dex_location_;
74   const std::string base_location_;
75   const uint32_t location_checksum_;
76   // Array of resolved class def indexes.
77   mutable std::unordered_set<dex::TypeIndex> classes_;
78 };
79 
80 inline bool operator<(const DexCacheResolvedClasses& a, const DexCacheResolvedClasses& b) {
81   return a.Compare(b) < 0;
82 }
83 
84 }  // namespace art
85 
86 #endif  // ART_RUNTIME_DEX_CACHE_RESOLVED_CLASSES_H_
87