forked from facebook/redex
-
Notifications
You must be signed in to change notification settings - Fork 0
/
BundleResources.h
147 lines (132 loc) · 6.05 KB
/
BundleResources.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
/*
* Copyright (c) Meta Platforms, Inc. and affiliates.
*
* This source code is licensed under the MIT license found in the
* LICENSE file in the root directory of this source tree.
*/
#pragma once
// TODO (T91001948): Integrate protobuf dependency in supported platforms for
// open source
#ifdef HAS_PROTOBUF
#include "RedexResources.h"
#include <map>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <vector>
#include "GlobalConfig.h"
#include "androidfw/ResourceTypes.h"
#include "protocfg/config.pb.h"
#include "protores/Resources.pb.h"
#include <google/protobuf/repeated_field.h>
using ConfigValues = google::protobuf::RepeatedPtrField<aapt::pb::ConfigValue>;
class ResourcesPbFile : public ResourceTableFile {
public:
~ResourcesPbFile() override;
size_t package_count() override;
void collect_resid_values_and_hashes(
const std::vector<uint32_t>& ids,
std::map<size_t, std::vector<uint32_t>>* res_by_hash) override;
void remap_file_paths_and_serialize(
const std::vector<std::string>& resource_files,
const std::unordered_map<std::string, std::string>& old_to_new) override;
void remap_res_ids_and_serialize(
const std::vector<std::string>& resource_files,
const std::map<uint32_t, uint32_t>& old_to_new) override;
void nullify_res_ids_and_serialize(
const std::vector<std::string>& resource_files) override;
void remap_reorder_and_serialize(
const std::vector<std::string>& resource_files,
const std::map<uint32_t, uint32_t>& old_to_new) override;
bool resource_value_identical(uint32_t a_id, uint32_t b_id) override;
void get_type_names(std::vector<std::string>* type_names) override;
std::unordered_set<uint32_t> get_types_by_name(
const std::unordered_set<std::string>& type_names) override;
std::unordered_set<uint32_t> get_types_by_name_prefixes(
const std::unordered_set<std::string>& type_name_prefixes) override;
std::vector<std::string> get_files_by_rid(
uint32_t res_id,
ResourcePathType path_type = ResourcePathType::DevicePath) override;
void walk_references_for_resource(
uint32_t resID,
ResourcePathType path_type,
std::unordered_set<uint32_t>* nodes_visited,
std::unordered_set<std::string>* potential_file_paths) override;
uint64_t resource_value_count(uint32_t res_id) override;
void delete_resource(uint32_t res_id) override;
void collect_resource_data_for_file(const std::string& resources_pb_path);
size_t get_hash_from_values(const ConfigValues& config_values);
size_t obfuscate_resource_and_serialize(
const std::vector<std::string>& resource_files,
const std::map<std::string, std::string>& filepath_old_to_new,
const std::unordered_set<uint32_t>& allowed_types,
const std::unordered_set<std::string>& keep_resource_prefixes,
const std::unordered_set<std::string>& keep_resource_specific) override;
void get_configurations(
uint32_t package_id,
const std::string& name,
std::vector<android::ResTable_config>* configs) override;
std::set<android::ResTable_config> get_configs_with_values(
uint32_t id) override;
void resolve_string_values_for_resource_reference(
uint32_t ref, std::vector<std::string>* values) override;
std::unordered_map<uint32_t, resources::InlinableValue>
get_inlinable_resource_values() override;
const std::map<uint32_t, const ConfigValues>& get_res_id_to_configvalue()
const {
return m_res_id_to_configvalue;
}
std::string resolve_module_name_for_resource_id(uint32_t res_id);
std::string resolve_module_name_for_package_id(uint32_t package_id);
private:
std::map<uint32_t, std::string> m_type_id_to_names;
std::unordered_set<uint32_t> m_existed_res_ids;
std::map<uint32_t, const aapt::pb::Entry> m_res_id_to_entry;
std::map<uint32_t, const ConfigValues> m_res_id_to_configvalue;
std::map<uint32_t, std::string> m_package_id_to_module_name;
std::set<uint32_t> m_package_ids;
};
class BundleResources : public AndroidResources {
public:
explicit BundleResources(const std::string& directory)
: AndroidResources(directory) {}
~BundleResources() override;
boost::optional<int32_t> get_min_sdk() override;
ManifestClassInfo get_manifest_class_info() override;
boost::optional<std::string> get_manifest_package_name() override;
std::unordered_set<std::string> get_service_loader_classes() override;
std::unordered_set<uint32_t> get_xml_reference_attributes(
const std::string& filename) override;
void collect_layout_classes_and_attributes_for_file(
const std::string& file_path,
const std::unordered_set<std::string>& attributes_to_read,
resources::StringOrReferenceSet* out_classes,
std::unordered_multimap<std::string, resources::StringOrReference>*
out_attributes) override;
void collect_xml_attribute_string_values_for_file(
const std::string& file_path,
std::unordered_set<std::string>* out) override;
void fully_qualify_layout(
const std::unordered_map<std::string, std::string>& element_to_class_name,
const std::string& file_path,
size_t* changes) override;
size_t remap_xml_reference_attributes(
const std::string& filename,
const std::map<uint32_t, uint32_t>& kept_to_remapped_ids) override;
std::vector<std::string> find_resources_files() override;
std::unique_ptr<ResourceTableFile> load_res_table() override;
std::unordered_set<std::string> find_all_xml_files() override;
std::string get_base_assets_dir() override;
void obfuscate_xml_files(const std::unordered_set<std::string>& allowed_types,
const std::unordered_set<std::string>&
do_not_obfuscate_elements) override;
void finalize_bundle_config(const ResourceConfig& config) override;
protected:
std::vector<std::string> find_res_directories() override;
std::vector<std::string> find_lib_directories() override;
bool rename_classes_in_layout(
const std::string& file_path,
const std::map<std::string, std::string>& rename_map,
size_t* out_num_renamed) override;
};
#endif // HAS_PROTOBUF