Home | History | Annotate | Download | only in dex 
      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:
     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.
     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 
     49   bool AddClass(dex::TypeIndex index) const {
     50     return classes_.insert(index).second;
     51   }
     52 
     53   template <class InputIt>
     54   void AddClasses(InputIt begin, InputIt end) const {
     55     classes_.insert(begin, end);
     56   }
     57 
     58   const std::string& GetDexLocation() const {
     59     return dex_location_;
     60   }
     61 
     62   const std::string& GetBaseLocation() const {
     63     return base_location_;
     64   }
     65 
     66   uint32_t GetLocationChecksum() const {
     67     return location_checksum_;
     68   }
     69 
     70   const std::unordered_set<dex::TypeIndex>& GetClasses() const {
     71     return classes_;
     72   }
     73 
     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