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 #include "benchmark/benchmark.h"
18 
19 #include "android-base/stringprintf.h"
20 #include "androidfw/ApkAssets.h"
21 #include "androidfw/AssetManager.h"
22 #include "androidfw/AssetManager2.h"
23 #include "androidfw/ResourceTypes.h"
24 
25 #include "BenchmarkHelpers.h"
26 #include "data/basic/R.h"
27 #include "data/libclient/R.h"
28 #include "data/styles/R.h"
29 
30 namespace app = com::android::app;
31 namespace basic = com::android::basic;
32 namespace libclient = com::android::libclient;
33 
34 namespace android {
35 
36 constexpr const static char* kFrameworkPath = "/system/framework/framework-res.apk";
37 
BM_AssetManagerLoadAssets(benchmark::State & state)38 static void BM_AssetManagerLoadAssets(benchmark::State& state) {
39   std::string path = GetTestDataPath() + "/basic/basic.apk";
40   while (state.KeepRunning()) {
41     auto apk = ApkAssets::Load(path);
42     AssetManager2 assets;
43     assets.SetApkAssets({apk});
44   }
45 }
46 BENCHMARK(BM_AssetManagerLoadAssets);
47 
BM_AssetManagerLoadAssetsOld(benchmark::State & state)48 static void BM_AssetManagerLoadAssetsOld(benchmark::State& state) {
49   String8 path((GetTestDataPath() + "/basic/basic.apk").data());
50   while (state.KeepRunning()) {
51     AssetManager assets;
52     assets.addAssetPath(path, nullptr /* cookie */, false /* appAsLib */,
53                         false /* isSystemAsset */);
54 
55     // Force creation.
56     assets.getResources(true);
57   }
58 }
59 BENCHMARK(BM_AssetManagerLoadAssetsOld);
60 
BM_AssetManagerLoadFrameworkAssets(benchmark::State & state)61 static void BM_AssetManagerLoadFrameworkAssets(benchmark::State& state) {
62   std::string path = kFrameworkPath;
63   while (state.KeepRunning()) {
64     auto apk = ApkAssets::Load(path);
65     AssetManager2 assets;
66     assets.SetApkAssets({apk});
67   }
68 }
69 BENCHMARK(BM_AssetManagerLoadFrameworkAssets);
70 
BM_AssetManagerLoadFrameworkAssetsOld(benchmark::State & state)71 static void BM_AssetManagerLoadFrameworkAssetsOld(benchmark::State& state) {
72   String8 path(kFrameworkPath);
73   while (state.KeepRunning()) {
74     AssetManager assets;
75     assets.addAssetPath(path, nullptr /* cookie */, false /* appAsLib */,
76                         false /* isSystemAsset */);
77 
78     // Force creation.
79     assets.getResources(true);
80   }
81 }
82 BENCHMARK(BM_AssetManagerLoadFrameworkAssetsOld);
83 
BM_AssetManagerGetResource(benchmark::State & state,uint32_t resid)84 static void BM_AssetManagerGetResource(benchmark::State& state, uint32_t resid) {
85   GetResourceBenchmark({GetTestDataPath() + "/basic/basic.apk"}, nullptr /*config*/, resid, state);
86 }
87 BENCHMARK_CAPTURE(BM_AssetManagerGetResource, number1, basic::R::integer::number1);
88 BENCHMARK_CAPTURE(BM_AssetManagerGetResource, deep_ref, basic::R::integer::deep_ref);
89 
BM_AssetManagerGetResourceOld(benchmark::State & state,uint32_t resid)90 static void BM_AssetManagerGetResourceOld(benchmark::State& state, uint32_t resid) {
91   GetResourceBenchmarkOld({GetTestDataPath() + "/basic/basic.apk"}, nullptr /*config*/, resid,
92                           state);
93 }
94 BENCHMARK_CAPTURE(BM_AssetManagerGetResourceOld, number1, basic::R::integer::number1);
95 BENCHMARK_CAPTURE(BM_AssetManagerGetResourceOld, deep_ref, basic::R::integer::deep_ref);
96 
BM_AssetManagerGetLibraryResource(benchmark::State & state)97 static void BM_AssetManagerGetLibraryResource(benchmark::State& state) {
98   GetResourceBenchmark(
99       {GetTestDataPath() + "/lib_two/lib_two.apk", GetTestDataPath() + "/lib_one/lib_one.apk",
100        GetTestDataPath() + "/libclient/libclient.apk"},
101       nullptr /*config*/, libclient::R::string::foo_one, state);
102 }
103 BENCHMARK(BM_AssetManagerGetLibraryResource);
104 
BM_AssetManagerGetLibraryResourceOld(benchmark::State & state)105 static void BM_AssetManagerGetLibraryResourceOld(benchmark::State& state) {
106   GetResourceBenchmarkOld(
107       {GetTestDataPath() + "/lib_two/lib_two.apk", GetTestDataPath() + "/lib_one/lib_one.apk",
108        GetTestDataPath() + "/libclient/libclient.apk"},
109       nullptr /*config*/, libclient::R::string::foo_one, state);
110 }
111 BENCHMARK(BM_AssetManagerGetLibraryResourceOld);
112 
113 constexpr static const uint32_t kStringOkId = 0x0104000au;
114 
BM_AssetManagerGetResourceFrameworkLocale(benchmark::State & state)115 static void BM_AssetManagerGetResourceFrameworkLocale(benchmark::State& state) {
116   ResTable_config config;
117   memset(&config, 0, sizeof(config));
118   memcpy(config.language, "fr", 2);
119   GetResourceBenchmark({kFrameworkPath}, &config, kStringOkId, state);
120 }
121 BENCHMARK(BM_AssetManagerGetResourceFrameworkLocale);
122 
BM_AssetManagerGetResourceFrameworkLocaleOld(benchmark::State & state)123 static void BM_AssetManagerGetResourceFrameworkLocaleOld(benchmark::State& state) {
124   ResTable_config config;
125   memset(&config, 0, sizeof(config));
126   memcpy(config.language, "fr", 2);
127   GetResourceBenchmarkOld({kFrameworkPath}, &config, kStringOkId, state);
128 }
129 BENCHMARK(BM_AssetManagerGetResourceFrameworkLocaleOld);
130 
BM_AssetManagerGetBag(benchmark::State & state)131 static void BM_AssetManagerGetBag(benchmark::State& state) {
132   auto apk = ApkAssets::Load(GetTestDataPath() + "/styles/styles.apk");
133   if (apk == nullptr) {
134     state.SkipWithError("Failed to load assets");
135     return;
136   }
137 
138   AssetManager2 assets;
139   assets.SetApkAssets({apk});
140 
141   while (state.KeepRunning()) {
142     auto bag = assets.GetBag(app::R::style::StyleTwo);
143     if (!bag.has_value()) {
144       state.SkipWithError("Failed to load get bag");
145       return;
146     }
147     const auto bag_end = end(*bag);
148     for (auto iter = begin(*bag); iter != bag_end; ++iter) {
149       uint32_t key = iter->key;
150       Res_value value = iter->value;
151       benchmark::DoNotOptimize(key);
152       benchmark::DoNotOptimize(value);
153     }
154   }
155 }
156 BENCHMARK(BM_AssetManagerGetBag);
157 
BM_AssetManagerGetBagOld(benchmark::State & state)158 static void BM_AssetManagerGetBagOld(benchmark::State& state) {
159   AssetManager assets;
160   if (!assets.addAssetPath(String8((GetTestDataPath() + "/styles/styles.apk").data()),
161                            nullptr /*cookie*/, false /*appAsLib*/, false /*isSystemAssets*/)) {
162     state.SkipWithError("Failed to load assets");
163     return;
164   }
165 
166   const ResTable& table = assets.getResources(true);
167 
168   while (state.KeepRunning()) {
169     const ResTable::bag_entry* bag_begin;
170     const ssize_t N = table.lockBag(app::R::style::StyleTwo, &bag_begin);
171     const ResTable::bag_entry* const bag_end = bag_begin + N;
172     for (auto iter = bag_begin; iter != bag_end; ++iter) {
173       uint32_t key = iter->map.name.ident;
174       Res_value value = iter->map.value;
175       benchmark::DoNotOptimize(key);
176       benchmark::DoNotOptimize(value);
177     }
178     table.unlockBag(bag_begin);
179   }
180 }
181 BENCHMARK(BM_AssetManagerGetBagOld);
182 
BM_AssetManagerGetResourceLocales(benchmark::State & state)183 static void BM_AssetManagerGetResourceLocales(benchmark::State& state) {
184   auto apk = ApkAssets::Load(kFrameworkPath);
185   if (apk == nullptr) {
186     state.SkipWithError("Failed to load assets");
187     return;
188   }
189 
190   AssetManager2 assets;
191   assets.SetApkAssets({apk});
192 
193   while (state.KeepRunning()) {
194     std::set<std::string> locales =
195         assets.GetResourceLocales(false /*exclude_system*/, true /*merge_equivalent_languages*/);
196     benchmark::DoNotOptimize(locales);
197   }
198 }
199 BENCHMARK(BM_AssetManagerGetResourceLocales);
200 
BM_AssetManagerGetResourceLocalesOld(benchmark::State & state)201 static void BM_AssetManagerGetResourceLocalesOld(benchmark::State& state) {
202   AssetManager assets;
203   if (!assets.addAssetPath(String8(kFrameworkPath), nullptr /*cookie*/, false /*appAsLib*/,
204                            true /*isSystemAssets*/)) {
205     state.SkipWithError("Failed to load assets");
206     return;
207   }
208 
209   const ResTable& table = assets.getResources(true);
210 
211   while (state.KeepRunning()) {
212     Vector<String8> locales;
213     table.getLocales(&locales, true /*includeSystemLocales*/, true /*mergeEquivalentLangs*/);
214     benchmark::DoNotOptimize(locales);
215   }
216 }
217 BENCHMARK(BM_AssetManagerGetResourceLocalesOld);
218 
BM_AssetManagerSetConfigurationFramework(benchmark::State & state)219 static void BM_AssetManagerSetConfigurationFramework(benchmark::State& state) {
220   auto apk = ApkAssets::Load(kFrameworkPath);
221   if (apk == nullptr) {
222     state.SkipWithError("Failed to load assets");
223     return;
224   }
225 
226   AssetManager2 assets;
227   assets.SetApkAssets({apk});
228 
229   ResTable_config config;
230   memset(&config, 0, sizeof(config));
231   std::vector<ResTable_config> configs;
232   configs.push_back(config);
233 
234   while (state.KeepRunning()) {
235     configs[0].sdkVersion = ~configs[0].sdkVersion;
236     assets.SetConfigurations(configs);
237   }
238 }
239 BENCHMARK(BM_AssetManagerSetConfigurationFramework);
240 
BM_AssetManagerSetConfigurationFrameworkOld(benchmark::State & state)241 static void BM_AssetManagerSetConfigurationFrameworkOld(benchmark::State& state) {
242   AssetManager assets;
243   if (!assets.addAssetPath(String8(kFrameworkPath), nullptr /*cookie*/, false /*appAsLib*/,
244                            true /*isSystemAssets*/)) {
245     state.SkipWithError("Failed to load assets");
246     return;
247   }
248 
249   const ResTable& table = assets.getResources(true);
250 
251   ResTable_config config;
252   memset(&config, 0, sizeof(config));
253 
254   while (state.KeepRunning()) {
255     config.sdkVersion = ~config.sdkVersion;
256     assets.setConfiguration(config);
257   }
258 }
259 BENCHMARK(BM_AssetManagerSetConfigurationFrameworkOld);
260 
261 }  // namespace android
262