blob: da39a9dafd1327a2b8f89ebaa3635b7695ee0882 [file] [log] [blame]
Andreas Gampe73dae112015-11-19 14:12:14 -08001/*
2 ** Copyright 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 <algorithm>
18#include <inttypes.h>
Andreas Gampec4ced4f2017-04-14 20:39:56 -070019#include <limits>
Andreas Gampe73dae112015-11-19 14:12:14 -080020#include <random>
Andreas Gampe1842af32016-03-16 14:28:50 -070021#include <regex>
Andreas Gampe73dae112015-11-19 14:12:14 -080022#include <selinux/android.h>
23#include <selinux/avc.h>
24#include <stdlib.h>
25#include <string.h>
26#include <sys/capability.h>
27#include <sys/prctl.h>
28#include <sys/stat.h>
Andreas Gampe73dae112015-11-19 14:12:14 -080029
30#include <android-base/logging.h>
31#include <android-base/macros.h>
32#include <android-base/stringprintf.h>
Andreas Gampe6db8db92016-06-03 10:22:19 -070033#include <android-base/strings.h>
Andreas Gampe73dae112015-11-19 14:12:14 -080034#include <cutils/fs.h>
Andreas Gampe73dae112015-11-19 14:12:14 -080035#include <cutils/properties.h>
Mark Salyzyn7823e122016-09-29 08:08:05 -070036#include <log/log.h>
Andreas Gampe73dae112015-11-19 14:12:14 -080037#include <private/android_filesystem_config.h>
38
Jeff Sharkey6c2c0562016-12-07 12:12:00 -070039#include "dexopt.h"
Jeff Sharkeyf3e30b92016-12-09 17:06:57 -070040#include "file_parsing.h"
41#include "globals.h"
Andreas Gampec4ced4f2017-04-14 20:39:56 -070042#include "installd_constants.h"
Jeff Sharkeyf3e30b92016-12-09 17:06:57 -070043#include "installd_deps.h" // Need to fill in requirements of commands.
Calin Juravlec9e76792018-02-01 14:44:56 +000044#include "otapreopt_parameters.h"
Jeff Sharkeyf3e30b92016-12-09 17:06:57 -070045#include "otapreopt_utils.h"
46#include "system_properties.h"
47#include "utils.h"
Jeff Sharkey6c2c0562016-12-07 12:12:00 -070048
Andreas Gampe73dae112015-11-19 14:12:14 -080049#ifndef LOG_TAG
50#define LOG_TAG "otapreopt"
51#endif
52
53#define BUFFER_MAX 1024 /* input buffer for commands */
54#define TOKEN_MAX 16 /* max number of arguments in buffer */
55#define REPLY_MAX 256 /* largest reply allowed */
56
Andreas Gampe56f79f92016-06-08 15:11:37 -070057using android::base::EndsWith;
Andreas Gampe6db8db92016-06-03 10:22:19 -070058using android::base::Split;
Andreas Gampe56f79f92016-06-08 15:11:37 -070059using android::base::StartsWith;
Andreas Gampe73dae112015-11-19 14:12:14 -080060using android::base::StringPrintf;
61
62namespace android {
63namespace installd {
64
Andreas Gampeef21fd22017-05-22 13:36:06 -070065// Check expected values for dexopt flags. If you need to change this:
66//
67// RUN AN A/B OTA TO MAKE SURE THINGS STILL WORK!
68//
69// You most likely need to increase the protocol version and all that entails!
70
71static_assert(DEXOPT_PUBLIC == 1 << 1, "DEXOPT_PUBLIC unexpected.");
72static_assert(DEXOPT_DEBUGGABLE == 1 << 2, "DEXOPT_DEBUGGABLE unexpected.");
73static_assert(DEXOPT_BOOTCOMPLETE == 1 << 3, "DEXOPT_BOOTCOMPLETE unexpected.");
74static_assert(DEXOPT_PROFILE_GUIDED == 1 << 4, "DEXOPT_PROFILE_GUIDED unexpected.");
75static_assert(DEXOPT_SECONDARY_DEX == 1 << 5, "DEXOPT_SECONDARY_DEX unexpected.");
76static_assert(DEXOPT_FORCE == 1 << 6, "DEXOPT_FORCE unexpected.");
77static_assert(DEXOPT_STORAGE_CE == 1 << 7, "DEXOPT_STORAGE_CE unexpected.");
78static_assert(DEXOPT_STORAGE_DE == 1 << 8, "DEXOPT_STORAGE_DE unexpected.");
David Brazdil22cce5a2018-02-12 18:04:59 -080079static_assert(DEXOPT_ENABLE_HIDDEN_API_CHECKS == 1 << 10,
80 "DEXOPT_ENABLE_HIDDEN_API_CHECKS unexpected");
Mathieu Chartier351bc942018-03-06 13:55:58 -080081static_assert(DEXOPT_GENERATE_COMPACT_DEX == 1 << 11, "DEXOPT_GENERATE_COMPACT_DEX unexpected");
Mathieu Chartierad45a1b2018-03-12 17:55:06 -070082static_assert(DEXOPT_GENERATE_APP_IMAGE == 1 << 12, "DEXOPT_GENERATE_APP_IMAGE unexpected");
Andreas Gampeef21fd22017-05-22 13:36:06 -070083
Patrick Baumann2271b3e2020-04-14 17:03:00 -070084static_assert(DEXOPT_MASK == (0x3dfe | DEXOPT_IDLE_BACKGROUND_JOB),
Andreas Gamped32eec22018-02-28 16:02:51 -080085 "DEXOPT_MASK unexpected.");
Andreas Gampeef21fd22017-05-22 13:36:06 -070086
87
Andreas Gampea64a6272018-07-10 10:43:47 -070088template<typename T>
89static constexpr bool IsPowerOfTwo(T x) {
90 static_assert(std::is_integral<T>::value, "T must be integral");
91 // TODO: assert unsigned. There is currently many uses with signed values.
92 return (x & (x - 1)) == 0;
93}
Andreas Gampeef21fd22017-05-22 13:36:06 -070094
Andreas Gampe73dae112015-11-19 14:12:14 -080095template<typename T>
96static constexpr T RoundDown(T x, typename std::decay<T>::type n) {
Elliott Hughes0beed272020-07-29 14:28:25 -070097 return (x & -n);
Andreas Gampe73dae112015-11-19 14:12:14 -080098}
99
100template<typename T>
101static constexpr T RoundUp(T x, typename std::remove_reference<T>::type n) {
102 return RoundDown(x + n - 1, n);
103}
104
105class OTAPreoptService {
106 public:
Andreas Gampe73dae112015-11-19 14:12:14 -0800107 // Main driver. Performs the following steps.
108 //
109 // 1) Parse options (read system properties etc from B partition).
110 //
111 // 2) Read in package data.
112 //
113 // 3) Prepare environment variables.
114 //
115 // 4) Prepare(compile) boot image, if necessary.
116 //
117 // 5) Run update.
118 int Main(int argc, char** argv) {
Andreas Gamped089ca12016-06-27 14:25:30 -0700119 if (!ReadArguments(argc, argv)) {
120 LOG(ERROR) << "Failed reading command line.";
121 return 1;
122 }
123
Andreas Gampe73dae112015-11-19 14:12:14 -0800124 if (!ReadSystemProperties()) {
125 LOG(ERROR)<< "Failed reading system properties.";
Andreas Gamped089ca12016-06-27 14:25:30 -0700126 return 2;
Andreas Gampe73dae112015-11-19 14:12:14 -0800127 }
128
129 if (!ReadEnvironment()) {
130 LOG(ERROR) << "Failed reading environment properties.";
Andreas Gamped089ca12016-06-27 14:25:30 -0700131 return 3;
Andreas Gampe73dae112015-11-19 14:12:14 -0800132 }
133
Andreas Gamped089ca12016-06-27 14:25:30 -0700134 if (!CheckAndInitializeInstalldGlobals()) {
135 LOG(ERROR) << "Failed initializing globals.";
136 return 4;
Andreas Gampe73dae112015-11-19 14:12:14 -0800137 }
138
Orion Hodson640cb762020-03-26 09:24:41 +0000139 PrepareEnvironmentVariables();
Andreas Gampe73dae112015-11-19 14:12:14 -0800140
Orion Hodson640cb762020-03-26 09:24:41 +0000141 if (!EnsureBootImageAndDalvikCache()) {
142 LOG(ERROR) << "Bad boot image.";
Andreas Gamped089ca12016-06-27 14:25:30 -0700143 return 5;
Andreas Gampe73dae112015-11-19 14:12:14 -0800144 }
145
146 int dexopt_retcode = RunPreopt();
147
148 return dexopt_retcode;
149 }
150
Andreas Gamped089ca12016-06-27 14:25:30 -0700151 int GetProperty(const char* key, char* value, const char* default_value) const {
Andreas Gampe73dae112015-11-19 14:12:14 -0800152 const std::string* prop_value = system_properties_.GetProperty(key);
153 if (prop_value == nullptr) {
154 if (default_value == nullptr) {
155 return 0;
156 }
157 // Copy in the default value.
Jeff Sharkey1b9d9a62017-09-21 14:51:09 -0600158 strlcpy(value, default_value, kPropertyValueMax - 1);
Andreas Gampe73dae112015-11-19 14:12:14 -0800159 value[kPropertyValueMax - 1] = 0;
160 return strlen(default_value);// TODO: Need to truncate?
161 }
Andreas Gampe5696e632017-09-26 20:41:48 -0700162 size_t size = std::min(kPropertyValueMax - 1, prop_value->length()) + 1;
Jeff Sharkey1b9d9a62017-09-21 14:51:09 -0600163 strlcpy(value, prop_value->data(), size);
Andreas Gampe5696e632017-09-26 20:41:48 -0700164 return static_cast<int>(size - 1);
Andreas Gampe73dae112015-11-19 14:12:14 -0800165 }
166
Andreas Gamped089ca12016-06-27 14:25:30 -0700167 std::string GetOTADataDirectory() const {
Calin Juravlec9e76792018-02-01 14:44:56 +0000168 return StringPrintf("%s/%s", GetOtaDirectoryPrefix().c_str(), GetTargetSlot().c_str());
Andreas Gamped089ca12016-06-27 14:25:30 -0700169 }
170
171 const std::string& GetTargetSlot() const {
Calin Juravlec9e76792018-02-01 14:44:56 +0000172 return parameters_.target_slot;
Andreas Gamped089ca12016-06-27 14:25:30 -0700173 }
174
Andreas Gampe73dae112015-11-19 14:12:14 -0800175private:
Andreas Gamped089ca12016-06-27 14:25:30 -0700176
Andreas Gampe73dae112015-11-19 14:12:14 -0800177 bool ReadSystemProperties() {
Alex Lightc36d0012021-03-03 16:10:12 -0800178 // TODO This file does not have a stable format. It should be read by
179 // code shared by init and otapreopt. See b/181182967#comment80
Andreas Gampe1842af32016-03-16 14:28:50 -0700180 static constexpr const char* kPropertyFiles[] = {
Alex Lightc36d0012021-03-03 16:10:12 -0800181 "/system/build.prop"
Andreas Gampe1842af32016-03-16 14:28:50 -0700182 };
Andreas Gampe73dae112015-11-19 14:12:14 -0800183
Andreas Gampe1842af32016-03-16 14:28:50 -0700184 for (size_t i = 0; i < arraysize(kPropertyFiles); ++i) {
185 if (!system_properties_.Load(kPropertyFiles[i])) {
186 return false;
187 }
188 }
189
190 return true;
Andreas Gampe73dae112015-11-19 14:12:14 -0800191 }
192
193 bool ReadEnvironment() {
Andreas Gampe1842af32016-03-16 14:28:50 -0700194 // Parse the environment variables from init.environ.rc, which have the form
195 // export NAME VALUE
196 // For simplicity, don't respect string quotation. The values we are interested in can be
197 // encoded without them.
198 std::regex export_regex("\\s*export\\s+(\\S+)\\s+(\\S+)");
199 bool parse_result = ParseFile("/init.environ.rc", [&](const std::string& line) {
200 std::smatch export_match;
201 if (!std::regex_match(line, export_match, export_regex)) {
202 return true;
203 }
Andreas Gampe73dae112015-11-19 14:12:14 -0800204
Andreas Gampe1842af32016-03-16 14:28:50 -0700205 if (export_match.size() != 3) {
206 return true;
207 }
208
209 std::string name = export_match[1].str();
210 std::string value = export_match[2].str();
211
212 system_properties_.SetProperty(name, value);
213
214 return true;
215 });
216 if (!parse_result) {
Andreas Gampe73dae112015-11-19 14:12:14 -0800217 return false;
218 }
Andreas Gampe1842af32016-03-16 14:28:50 -0700219
Andreas Gamped089ca12016-06-27 14:25:30 -0700220 if (system_properties_.GetProperty(kAndroidDataPathPropertyName) == nullptr) {
221 return false;
222 }
223 android_data_ = *system_properties_.GetProperty(kAndroidDataPathPropertyName);
224
225 if (system_properties_.GetProperty(kAndroidRootPathPropertyName) == nullptr) {
226 return false;
227 }
228 android_root_ = *system_properties_.GetProperty(kAndroidRootPathPropertyName);
229
230 if (system_properties_.GetProperty(kBootClassPathPropertyName) == nullptr) {
231 return false;
232 }
233 boot_classpath_ = *system_properties_.GetProperty(kBootClassPathPropertyName);
234
235 if (system_properties_.GetProperty(ASEC_MOUNTPOINT_ENV_NAME) == nullptr) {
236 return false;
237 }
238 asec_mountpoint_ = *system_properties_.GetProperty(ASEC_MOUNTPOINT_ENV_NAME);
239
240 return true;
241 }
242
243 const std::string& GetAndroidData() const {
244 return android_data_;
245 }
246
247 const std::string& GetAndroidRoot() const {
248 return android_root_;
249 }
250
251 const std::string GetOtaDirectoryPrefix() const {
252 return GetAndroidData() + "/ota";
253 }
254
255 bool CheckAndInitializeInstalldGlobals() {
256 // init_globals_from_data_and_root requires "ASEC_MOUNTPOINT" in the environment. We
257 // do not use any datapath that includes this, but we'll still have to set it.
258 CHECK(system_properties_.GetProperty(ASEC_MOUNTPOINT_ENV_NAME) != nullptr);
259 int result = setenv(ASEC_MOUNTPOINT_ENV_NAME, asec_mountpoint_.c_str(), 0);
260 if (result != 0) {
261 LOG(ERROR) << "Could not set ASEC_MOUNTPOINT environment variable";
262 return false;
263 }
264
265 if (!init_globals_from_data_and_root(GetAndroidData().c_str(), GetAndroidRoot().c_str())) {
266 LOG(ERROR) << "Could not initialize globals; exiting.";
267 return false;
268 }
269
270 // This is different from the normal installd. We only do the base
271 // directory, the rest will be created on demand when each app is compiled.
272 if (access(GetOtaDirectoryPrefix().c_str(), R_OK) < 0) {
273 LOG(ERROR) << "Could not access " << GetOtaDirectoryPrefix();
274 return false;
Andreas Gampe1842af32016-03-16 14:28:50 -0700275 }
Andreas Gampe73dae112015-11-19 14:12:14 -0800276
277 return true;
278 }
279
Shubham Ajmera45c87432017-06-22 11:10:27 -0700280 bool ParseBool(const char* in) {
281 if (strcmp(in, "true") == 0) {
282 return true;
283 }
284 return false;
285 }
286
Andreas Gampec4ced4f2017-04-14 20:39:56 -0700287 bool ParseUInt(const char* in, uint32_t* out) {
288 char* end;
289 long long int result = strtoll(in, &end, 0);
290 if (in == end || *end != '\0') {
291 return false;
292 }
293 if (result < std::numeric_limits<uint32_t>::min() ||
294 std::numeric_limits<uint32_t>::max() < result) {
295 return false;
296 }
297 *out = static_cast<uint32_t>(result);
298 return true;
299 }
Andreas Gamped089ca12016-06-27 14:25:30 -0700300
Andreas Gampec4ced4f2017-04-14 20:39:56 -0700301 bool ReadArguments(int argc, char** argv) {
Calin Juravlec9e76792018-02-01 14:44:56 +0000302 return parameters_.ReadArguments(argc, const_cast<const char**>(argv));
Andreas Gampe73dae112015-11-19 14:12:14 -0800303 }
304
Orion Hodson640cb762020-03-26 09:24:41 +0000305 void PrepareEnvironmentVariables() {
Andreas Gamped089ca12016-06-27 14:25:30 -0700306 environ_.push_back(StringPrintf("BOOTCLASSPATH=%s", boot_classpath_.c_str()));
307 environ_.push_back(StringPrintf("ANDROID_DATA=%s", GetOTADataDirectory().c_str()));
308 environ_.push_back(StringPrintf("ANDROID_ROOT=%s", android_root_.c_str()));
Andreas Gampe73dae112015-11-19 14:12:14 -0800309
310 for (const std::string& e : environ_) {
311 putenv(const_cast<char*>(e.c_str()));
312 }
313 }
314
Orion Hodson640cb762020-03-26 09:24:41 +0000315 // Ensure that we have the right boot image and cache file structures.
316 bool EnsureBootImageAndDalvikCache() const {
Calin Juravlec9e76792018-02-01 14:44:56 +0000317 if (parameters_.instruction_set == nullptr) {
Andreas Gampe73dae112015-11-19 14:12:14 -0800318 LOG(ERROR) << "Instruction set missing.";
319 return false;
320 }
Calin Juravlec9e76792018-02-01 14:44:56 +0000321 const char* isa = parameters_.instruction_set;
Andreas Gamped089ca12016-06-27 14:25:30 -0700322 std::string dalvik_cache = GetOTADataDirectory() + "/" + DALVIK_CACHE;
Andreas Gampe73dae112015-11-19 14:12:14 -0800323 std::string isa_path = dalvik_cache + "/" + isa;
Andreas Gampe73dae112015-11-19 14:12:14 -0800324
Andreas Gamped089ca12016-06-27 14:25:30 -0700325 // Reset umask in otapreopt, so that we control the the access for the files we create.
326 umask(0);
327
Andreas Gampe73dae112015-11-19 14:12:14 -0800328 // Create the directories, if necessary.
329 if (access(dalvik_cache.c_str(), F_OK) != 0) {
Andreas Gamped089ca12016-06-27 14:25:30 -0700330 if (!CreatePath(dalvik_cache)) {
331 PLOG(ERROR) << "Could not create dalvik-cache dir " << dalvik_cache;
Andreas Gampe73dae112015-11-19 14:12:14 -0800332 return false;
333 }
334 }
335 if (access(isa_path.c_str(), F_OK) != 0) {
Andreas Gamped089ca12016-06-27 14:25:30 -0700336 if (!CreatePath(isa_path)) {
Andreas Gampe73dae112015-11-19 14:12:14 -0800337 PLOG(ERROR) << "Could not create dalvik-cache isa dir";
338 return false;
339 }
340 }
341
Orion Hodson640cb762020-03-26 09:24:41 +0000342 // Clear cached artifacts.
343 ClearDirectory(isa_path);
344
345 // Check whether we have a boot image.
Andreas Gampebd7aef12018-10-23 13:58:44 -0700346 // TODO: check that the files are correct wrt/ jars.
Orion Hodson640cb762020-03-26 09:24:41 +0000347 std::string preopted_boot_art_path =
348 StringPrintf("/apex/com.android.art/javalib/%s/boot.art", isa);
349 if (access(preopted_boot_art_path.c_str(), F_OK) != 0) {
350 PLOG(ERROR) << "Bad access() to " << preopted_boot_art_path;
351 return false;
Andreas Gampebd7aef12018-10-23 13:58:44 -0700352 }
353
Orion Hodson640cb762020-03-26 09:24:41 +0000354 return true;
Andreas Gampe5709b572016-02-12 17:42:59 -0800355 }
356
Andreas Gamped089ca12016-06-27 14:25:30 -0700357 static bool CreatePath(const std::string& path) {
358 // Create the given path. Use string processing instead of dirname, as dirname's need for
359 // a writable char buffer is painful.
360
361 // First, try to use the full path.
362 if (mkdir(path.c_str(), 0711) == 0) {
363 return true;
364 }
365 if (errno != ENOENT) {
366 PLOG(ERROR) << "Could not create path " << path;
367 return false;
368 }
369
370 // Now find the parent and try that first.
371 size_t last_slash = path.find_last_of('/');
372 if (last_slash == std::string::npos || last_slash == 0) {
373 PLOG(ERROR) << "Could not create " << path;
374 return false;
375 }
376
377 if (!CreatePath(path.substr(0, last_slash))) {
378 return false;
379 }
380
381 if (mkdir(path.c_str(), 0711) == 0) {
382 return true;
383 }
384 PLOG(ERROR) << "Could not create " << path;
385 return false;
386 }
387
388 static void ClearDirectory(const std::string& dir) {
389 DIR* c_dir = opendir(dir.c_str());
390 if (c_dir == nullptr) {
391 PLOG(WARNING) << "Unable to open " << dir << " to delete it's contents";
392 return;
393 }
394
395 for (struct dirent* de = readdir(c_dir); de != nullptr; de = readdir(c_dir)) {
396 const char* name = de->d_name;
397 if (strcmp(name, ".") == 0 || strcmp(name, "..") == 0) {
398 continue;
399 }
400 // We only want to delete regular files and symbolic links.
401 std::string file = StringPrintf("%s/%s", dir.c_str(), name);
402 if (de->d_type != DT_REG && de->d_type != DT_LNK) {
403 LOG(WARNING) << "Unexpected file "
404 << file
405 << " of type "
406 << std::hex
407 << de->d_type
408 << " encountered.";
409 } else {
410 // Try to unlink the file.
411 if (unlink(file.c_str()) != 0) {
412 PLOG(ERROR) << "Unable to unlink " << file;
413 }
414 }
415 }
416 CHECK_EQ(0, closedir(c_dir)) << "Unable to close directory.";
417 }
418
Andreas Gampe73dae112015-11-19 14:12:14 -0800419 static const char* ParseNull(const char* arg) {
420 return (strcmp(arg, "!") == 0) ? nullptr : arg;
421 }
422
Andreas Gamped089ca12016-06-27 14:25:30 -0700423 bool ShouldSkipPreopt() const {
Andreas Gampe56f79f92016-06-08 15:11:37 -0700424 // There's one thing we have to be careful about: we may/will be asked to compile an app
425 // living in the system image. This may be a valid request - if the app wasn't compiled,
426 // e.g., if the system image wasn't large enough to include preopted files. However, the
427 // data we have is from the old system, so the driver (the OTA service) can't actually
428 // know. Thus, we will get requests for apps that have preopted components. To avoid
429 // duplication (we'd generate files that are not used and are *not* cleaned up), do two
430 // simple checks:
431 //
432 // 1) Does the apk_path start with the value of ANDROID_ROOT? (~in the system image)
433 // (For simplicity, assume the value of ANDROID_ROOT does not contain a symlink.)
434 //
435 // 2) If you replace the name in the apk_path with "oat," does the path exist?
436 // (=have a subdirectory for preopted files)
437 //
438 // If the answer to both is yes, skip the dexopt.
439 //
440 // Note: while one may think it's OK to call dexopt and it will fail (because APKs should
441 // be stripped), that's not true for APKs signed outside the build system (so the
442 // jar content must be exactly the same).
443
444 // (This is ugly as it's the only thing where we need to understand the contents
Calin Juravlec9e76792018-02-01 14:44:56 +0000445 // of parameters_, but it beats postponing the decision or using the call-
Andreas Gampe56f79f92016-06-08 15:11:37 -0700446 // backs to do weird things.)
Calin Juravlec9e76792018-02-01 14:44:56 +0000447 const char* apk_path = parameters_.apk_path;
Andreas Gampec4ced4f2017-04-14 20:39:56 -0700448 CHECK(apk_path != nullptr);
Elliott Hughes969e4f82017-12-20 12:34:09 -0800449 if (StartsWith(apk_path, android_root_)) {
Andreas Gampec4ced4f2017-04-14 20:39:56 -0700450 const char* last_slash = strrchr(apk_path, '/');
Andreas Gampe56f79f92016-06-08 15:11:37 -0700451 if (last_slash != nullptr) {
Andreas Gampec4ced4f2017-04-14 20:39:56 -0700452 std::string path(apk_path, last_slash - apk_path + 1);
Andreas Gampe56f79f92016-06-08 15:11:37 -0700453 CHECK(EndsWith(path, "/"));
454 path = path + "oat";
455 if (access(path.c_str(), F_OK) == 0) {
Calin Juravleb3591f62017-11-17 16:38:17 -0800456 LOG(INFO) << "Skipping A/B OTA preopt of already preopted package " << apk_path;
Andreas Gamped089ca12016-06-27 14:25:30 -0700457 return true;
Andreas Gampe56f79f92016-06-08 15:11:37 -0700458 }
459 }
460 }
461
Andreas Gamped089ca12016-06-27 14:25:30 -0700462 // Another issue is unavailability of files in the new system. If the partition
463 // layout changes, otapreopt_chroot may not know about this. Then files from that
464 // partition will not be available and fail to build. This is problematic, as
465 // this tool will wipe the OTA artifact cache and try again (for robustness after
466 // a failed OTA with remaining cache artifacts).
Andreas Gampec4ced4f2017-04-14 20:39:56 -0700467 if (access(apk_path, F_OK) != 0) {
Calin Juravleb3591f62017-11-17 16:38:17 -0800468 LOG(WARNING) << "Skipping A/B OTA preopt of non-existing package " << apk_path;
Andreas Gamped089ca12016-06-27 14:25:30 -0700469 return true;
470 }
471
472 return false;
473 }
474
Calin Juravlec9e76792018-02-01 14:44:56 +0000475 // Run dexopt with the parameters of parameters_.
Calin Juravlecfcd6aa2018-01-18 20:23:17 -0800476 // TODO(calin): embed the profile name in the parameters.
Andreas Gampeb39d2f02017-04-17 20:04:02 -0700477 int Dexopt() {
Alex Light6c857b72021-02-25 13:37:13 -0800478 std::string error;
479 int res = dexopt(parameters_.apk_path,
480 parameters_.uid,
481 parameters_.pkgName,
482 parameters_.instruction_set,
483 parameters_.dexopt_needed,
484 parameters_.oat_dir,
485 parameters_.dexopt_flags,
486 parameters_.compiler_filter,
487 parameters_.volume_uuid,
488 parameters_.shared_libraries,
489 parameters_.se_info,
490 parameters_.downgrade,
491 parameters_.target_sdk_version,
492 parameters_.profile_name,
493 parameters_.dex_metadata_path,
494 parameters_.compilation_reason,
495 &error);
496 if (res != 0) {
497 LOG(ERROR) << "During preopt of " << parameters_.apk_path << " got result " << res
498 << " error: " << error;
499 }
500 return res;
Andreas Gampe73dae112015-11-19 14:12:14 -0800501 }
502
Andreas Gampeb39d2f02017-04-17 20:04:02 -0700503 int RunPreopt() {
504 if (ShouldSkipPreopt()) {
505 return 0;
506 }
507
508 int dexopt_result = Dexopt();
509 if (dexopt_result == 0) {
510 return 0;
511 }
512
Andreas Gampeb39d2f02017-04-17 20:04:02 -0700513 // If this was a profile-guided run, we may have profile version issues. Try to downgrade,
514 // if possible.
Calin Juravlec9e76792018-02-01 14:44:56 +0000515 if ((parameters_.dexopt_flags & DEXOPT_PROFILE_GUIDED) == 0) {
Andreas Gampeb39d2f02017-04-17 20:04:02 -0700516 return dexopt_result;
517 }
518
519 LOG(WARNING) << "Downgrading compiler filter in an attempt to progress compilation";
Calin Juravlec9e76792018-02-01 14:44:56 +0000520 parameters_.dexopt_flags &= ~DEXOPT_PROFILE_GUIDED;
Andreas Gampeb39d2f02017-04-17 20:04:02 -0700521 return Dexopt();
522 }
523
Andreas Gampe73dae112015-11-19 14:12:14 -0800524 ////////////////////////////////////
525 // Helpers, mostly taken from ART //
526 ////////////////////////////////////
527
Andreas Gampe73dae112015-11-19 14:12:14 -0800528 // Choose a random relocation offset. Taken from art/runtime/gc/image_space.cc.
529 static int32_t ChooseRelocationOffsetDelta(int32_t min_delta, int32_t max_delta) {
530 constexpr size_t kPageSize = PAGE_SIZE;
Elliott Hughes0beed272020-07-29 14:28:25 -0700531 static_assert(IsPowerOfTwo(kPageSize), "page size must be power of two");
Andreas Gampe73dae112015-11-19 14:12:14 -0800532 CHECK_EQ(min_delta % kPageSize, 0u);
533 CHECK_EQ(max_delta % kPageSize, 0u);
534 CHECK_LT(min_delta, max_delta);
535
536 std::default_random_engine generator;
537 generator.seed(GetSeed());
538 std::uniform_int_distribution<int32_t> distribution(min_delta, max_delta);
539 int32_t r = distribution(generator);
540 if (r % 2 == 0) {
541 r = RoundUp(r, kPageSize);
542 } else {
543 r = RoundDown(r, kPageSize);
544 }
545 CHECK_LE(min_delta, r);
546 CHECK_GE(max_delta, r);
547 CHECK_EQ(r % kPageSize, 0u);
548 return r;
549 }
550
551 static uint64_t GetSeed() {
552#ifdef __BIONIC__
553 // Bionic exposes arc4random, use it.
554 uint64_t random_data;
555 arc4random_buf(&random_data, sizeof(random_data));
556 return random_data;
557#else
558#error "This is only supposed to run with bionic. Otherwise, implement..."
559#endif
560 }
561
562 void AddCompilerOptionFromSystemProperty(const char* system_property,
563 const char* prefix,
564 bool runtime,
Andreas Gamped089ca12016-06-27 14:25:30 -0700565 std::vector<std::string>& out) const {
566 const std::string* value = system_properties_.GetProperty(system_property);
Andreas Gampe73dae112015-11-19 14:12:14 -0800567 if (value != nullptr) {
568 if (runtime) {
569 out.push_back("--runtime-arg");
570 }
571 if (prefix != nullptr) {
572 out.push_back(StringPrintf("%s%s", prefix, value->c_str()));
573 } else {
574 out.push_back(*value);
575 }
576 }
577 }
578
Andreas Gamped089ca12016-06-27 14:25:30 -0700579 static constexpr const char* kBootClassPathPropertyName = "BOOTCLASSPATH";
580 static constexpr const char* kAndroidRootPathPropertyName = "ANDROID_ROOT";
581 static constexpr const char* kAndroidDataPathPropertyName = "ANDROID_DATA";
582 // The index of the instruction-set string inside the package parameters. Needed for
583 // some special-casing that requires knowledge of the instruction-set.
584 static constexpr size_t kISAIndex = 3;
585
Andreas Gampe73dae112015-11-19 14:12:14 -0800586 // Stores the system properties read out of the B partition. We need to use these properties
587 // to compile, instead of the A properties we could get from init/get_property.
588 SystemProperties system_properties_;
589
Andreas Gamped089ca12016-06-27 14:25:30 -0700590 // Some select properties that are always needed.
Andreas Gamped089ca12016-06-27 14:25:30 -0700591 std::string android_root_;
592 std::string android_data_;
593 std::string boot_classpath_;
594 std::string asec_mountpoint_;
595
Calin Juravlec9e76792018-02-01 14:44:56 +0000596 OTAPreoptParameters parameters_;
Andreas Gampe73dae112015-11-19 14:12:14 -0800597
598 // Store environment values we need to set.
599 std::vector<std::string> environ_;
600};
601
602OTAPreoptService gOps;
603
604////////////////////////
605// Plug-in functions. //
606////////////////////////
607
608int get_property(const char *key, char *value, const char *default_value) {
Andreas Gampe73dae112015-11-19 14:12:14 -0800609 return gOps.GetProperty(key, value, default_value);
610}
611
612// Compute the output path of
613bool calculate_oat_file_path(char path[PKG_PATH_MAX], const char *oat_dir,
614 const char *apk_path,
615 const char *instruction_set) {
Dan Austin9c8f93a2016-06-03 16:15:54 -0700616 const char *file_name_start;
617 const char *file_name_end;
Andreas Gampe73dae112015-11-19 14:12:14 -0800618
619 file_name_start = strrchr(apk_path, '/');
620 if (file_name_start == nullptr) {
621 ALOGE("apk_path '%s' has no '/'s in it\n", apk_path);
622 return false;
623 }
624 file_name_end = strrchr(file_name_start, '.');
625 if (file_name_end == nullptr) {
626 ALOGE("apk_path '%s' has no extension\n", apk_path);
627 return false;
628 }
629
630 // Calculate file_name
631 file_name_start++; // Move past '/', is valid as file_name_end is valid.
632 size_t file_name_len = file_name_end - file_name_start;
633 std::string file_name(file_name_start, file_name_len);
634
635 // <apk_parent_dir>/oat/<isa>/<file_name>.odex.b
Andreas Gamped089ca12016-06-27 14:25:30 -0700636 snprintf(path,
637 PKG_PATH_MAX,
638 "%s/%s/%s.odex.%s",
639 oat_dir,
640 instruction_set,
641 file_name.c_str(),
642 gOps.GetTargetSlot().c_str());
Andreas Gampe73dae112015-11-19 14:12:14 -0800643 return true;
644}
645
646/*
647 * Computes the odex file for the given apk_path and instruction_set.
648 * /system/framework/whatever.jar -> /system/framework/oat/<isa>/whatever.odex
649 *
650 * Returns false if it failed to determine the odex file path.
651 */
652bool calculate_odex_file_path(char path[PKG_PATH_MAX], const char *apk_path,
653 const char *instruction_set) {
Andreas Gampe73dae112015-11-19 14:12:14 -0800654 const char *path_end = strrchr(apk_path, '/');
655 if (path_end == nullptr) {
656 ALOGE("apk_path '%s' has no '/'s in it?!\n", apk_path);
657 return false;
658 }
659 std::string path_component(apk_path, path_end - apk_path);
660
661 const char *name_begin = path_end + 1;
662 const char *extension_start = strrchr(name_begin, '.');
663 if (extension_start == nullptr) {
664 ALOGE("apk_path '%s' has no extension.\n", apk_path);
665 return false;
666 }
667 std::string name_component(name_begin, extension_start - name_begin);
668
Andreas Gamped089ca12016-06-27 14:25:30 -0700669 std::string new_path = StringPrintf("%s/oat/%s/%s.odex.%s",
Andreas Gampe73dae112015-11-19 14:12:14 -0800670 path_component.c_str(),
671 instruction_set,
Andreas Gamped089ca12016-06-27 14:25:30 -0700672 name_component.c_str(),
673 gOps.GetTargetSlot().c_str());
674 if (new_path.length() >= PKG_PATH_MAX) {
675 LOG(ERROR) << "apk_path of " << apk_path << " is too long: " << new_path;
676 return false;
677 }
Andreas Gampe73dae112015-11-19 14:12:14 -0800678 strcpy(path, new_path.c_str());
679 return true;
680}
681
682bool create_cache_path(char path[PKG_PATH_MAX],
683 const char *src,
684 const char *instruction_set) {
685 size_t srclen = strlen(src);
686
687 /* demand that we are an absolute path */
688 if ((src == 0) || (src[0] != '/') || strstr(src,"..")) {
689 return false;
690 }
691
692 if (srclen > PKG_PATH_MAX) { // XXX: PKG_NAME_MAX?
693 return false;
694 }
695
696 std::string from_src = std::string(src + 1);
697 std::replace(from_src.begin(), from_src.end(), '/', '@');
698
699 std::string assembled_path = StringPrintf("%s/%s/%s/%s%s",
Andreas Gamped089ca12016-06-27 14:25:30 -0700700 gOps.GetOTADataDirectory().c_str(),
Andreas Gampe73dae112015-11-19 14:12:14 -0800701 DALVIK_CACHE,
702 instruction_set,
703 from_src.c_str(),
David Brazdil249c1792016-09-06 15:35:28 +0100704 DALVIK_CACHE_POSTFIX);
Andreas Gampe73dae112015-11-19 14:12:14 -0800705
706 if (assembled_path.length() + 1 > PKG_PATH_MAX) {
707 return false;
708 }
709 strcpy(path, assembled_path.c_str());
710
711 return true;
712}
713
Andreas Gampe73dae112015-11-19 14:12:14 -0800714static int log_callback(int type, const char *fmt, ...) {
715 va_list ap;
716 int priority;
717
718 switch (type) {
719 case SELINUX_WARNING:
720 priority = ANDROID_LOG_WARN;
721 break;
722 case SELINUX_INFO:
723 priority = ANDROID_LOG_INFO;
724 break;
725 default:
726 priority = ANDROID_LOG_ERROR;
727 break;
728 }
729 va_start(ap, fmt);
730 LOG_PRI_VA(priority, "SELinux", fmt, ap);
731 va_end(ap);
732 return 0;
733}
734
735static int otapreopt_main(const int argc, char *argv[]) {
736 int selinux_enabled = (is_selinux_enabled() > 0);
737
738 setenv("ANDROID_LOG_TAGS", "*:v", 1);
739 android::base::InitLogging(argv);
740
Andreas Gampe73dae112015-11-19 14:12:14 -0800741 if (argc < 2) {
742 ALOGE("Expecting parameters");
743 exit(1);
744 }
745
746 union selinux_callback cb;
747 cb.func_log = log_callback;
748 selinux_set_callback(SELINUX_CB_LOG, cb);
749
Andreas Gampe73dae112015-11-19 14:12:14 -0800750 if (selinux_enabled && selinux_status_open(true) < 0) {
751 ALOGE("Could not open selinux status; exiting.\n");
752 exit(1);
753 }
754
755 int ret = android::installd::gOps.Main(argc, argv);
756
757 return ret;
758}
759
760} // namespace installd
761} // namespace android
762
763int main(const int argc, char *argv[]) {
764 return android::installd::otapreopt_main(argc, argv);
765}