Home | History | Annotate | Download | only in veridex
      1 /*
      2  * Copyright (C) 2018 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_TOOLS_VERIDEX_PRECISE_HIDDEN_API_FINDER_H_
     18 #define ART_TOOLS_VERIDEX_PRECISE_HIDDEN_API_FINDER_H_
     19 
     20 #include "dex/method_reference.h"
     21 #include "flow_analysis.h"
     22 
     23 #include <iostream>
     24 #include <map>
     25 #include <set>
     26 #include <string>
     27 
     28 namespace art {
     29 
     30 class HiddenApi;
     31 struct HiddenApiStats;
     32 class VeridexResolver;
     33 
     34 /**
     35  * Reports known uses of hidden APIs from reflection.
     36  */
     37 class PreciseHiddenApiFinder {
     38  public:
     39   explicit PreciseHiddenApiFinder(const HiddenApi& hidden_api) : hidden_api_(hidden_api) {}
     40 
     41   // Iterate over the dex files associated with the passed resolvers to report
     42   // hidden API uses.
     43   void Run(const std::vector<std::unique_ptr<VeridexResolver>>& app_resolvers);
     44 
     45   void Dump(std::ostream& os, HiddenApiStats* stats);
     46 
     47  private:
     48   // Run over all methods of all dex files, and call `action` on each.
     49   void RunInternal(
     50       const std::vector<std::unique_ptr<VeridexResolver>>& resolvers,
     51       const std::function<void(VeridexResolver*, const ClassDataItemIterator&)>& action);
     52 
     53   // Add uses found in method `ref`.
     54   void AddUsesAt(const std::vector<ReflectAccessInfo>& accesses, MethodReference ref);
     55 
     56   const HiddenApi& hidden_api_;
     57 
     58   std::map<MethodReference, std::vector<ReflectAccessInfo>> concrete_uses_;
     59   std::map<MethodReference, std::vector<ReflectAccessInfo>> abstract_uses_;
     60 };
     61 
     62 }  // namespace art
     63 
     64 #endif  // ART_TOOLS_VERIDEX_PRECISE_HIDDEN_API_FINDER_H_
     65