blob: cefcf6404c8ca2770acf842650909dc309c33af1 [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.
Alex Light9c3d87d2021-03-03 16:49:08 -0800198 // init.environ.rc and etc/classpath have the same format for
199 // environment variable exports and can be matched by the same regex.
200 // TODO Just like with the system-properties above we really should have
201 // common code between init and otapreopt to deal with reading these
202 // things. See b/181182967
203 static constexpr const char* kEnvironmentVariableSources[] = {
204 "/init.environ.rc", "/etc/classpath"
205 };
206
Andreas Gampe1842af32016-03-16 14:28:50 -0700207 std::regex export_regex("\\s*export\\s+(\\S+)\\s+(\\S+)");
Alex Light9c3d87d2021-03-03 16:49:08 -0800208 for (const char* env_vars_file : kEnvironmentVariableSources) {
209 bool parse_result = ParseFile(env_vars_file, [&](const std::string& line) {
210 std::smatch export_match;
211 if (!std::regex_match(line, export_match, export_regex)) {
212 return true;
213 }
214
215 if (export_match.size() != 3) {
216 return true;
217 }
218
219 std::string name = export_match[1].str();
220 std::string value = export_match[2].str();
221
222 system_properties_.SetProperty(name, value);
223
Andreas Gampe1842af32016-03-16 14:28:50 -0700224 return true;
Alex Light9c3d87d2021-03-03 16:49:08 -0800225 });
226 if (!parse_result) {
227 return false;
Andreas Gampe1842af32016-03-16 14:28:50 -0700228 }
Andreas Gampe73dae112015-11-19 14:12:14 -0800229 }
Andreas Gamped089ca12016-06-27 14:25:30 -0700230 if (system_properties_.GetProperty(kAndroidDataPathPropertyName) == nullptr) {
231 return false;
232 }
233 android_data_ = *system_properties_.GetProperty(kAndroidDataPathPropertyName);
234
235 if (system_properties_.GetProperty(kAndroidRootPathPropertyName) == nullptr) {
236 return false;
237 }
238 android_root_ = *system_properties_.GetProperty(kAndroidRootPathPropertyName);
239
240 if (system_properties_.GetProperty(kBootClassPathPropertyName) == nullptr) {
241 return false;
242 }
243 boot_classpath_ = *system_properties_.GetProperty(kBootClassPathPropertyName);
244
245 if (system_properties_.GetProperty(ASEC_MOUNTPOINT_ENV_NAME) == nullptr) {
246 return false;
247 }
248 asec_mountpoint_ = *system_properties_.GetProperty(ASEC_MOUNTPOINT_ENV_NAME);
249
250 return true;
251 }
252
253 const std::string& GetAndroidData() const {
254 return android_data_;
255 }
256
257 const std::string& GetAndroidRoot() const {
258 return android_root_;
259 }
260
261 const std::string GetOtaDirectoryPrefix() const {
262 return GetAndroidData() + "/ota";
263 }
264
265 bool CheckAndInitializeInstalldGlobals() {
266 // init_globals_from_data_and_root requires "ASEC_MOUNTPOINT" in the environment. We
267 // do not use any datapath that includes this, but we'll still have to set it.
268 CHECK(system_properties_.GetProperty(ASEC_MOUNTPOINT_ENV_NAME) != nullptr);
269 int result = setenv(ASEC_MOUNTPOINT_ENV_NAME, asec_mountpoint_.c_str(), 0);
270 if (result != 0) {
271 LOG(ERROR) << "Could not set ASEC_MOUNTPOINT environment variable";
272 return false;
273 }
274
275 if (!init_globals_from_data_and_root(GetAndroidData().c_str(), GetAndroidRoot().c_str())) {
276 LOG(ERROR) << "Could not initialize globals; exiting.";
277 return false;
278 }
279
280 // This is different from the normal installd. We only do the base
281 // directory, the rest will be created on demand when each app is compiled.
282 if (access(GetOtaDirectoryPrefix().c_str(), R_OK) < 0) {
283 LOG(ERROR) << "Could not access " << GetOtaDirectoryPrefix();
284 return false;
Andreas Gampe1842af32016-03-16 14:28:50 -0700285 }
Andreas Gampe73dae112015-11-19 14:12:14 -0800286
287 return true;
288 }
289
Shubham Ajmera45c87432017-06-22 11:10:27 -0700290 bool ParseBool(const char* in) {
291 if (strcmp(in, "true") == 0) {
292 return true;
293 }
294 return false;
295 }
296
Andreas Gampec4ced4f2017-04-14 20:39:56 -0700297 bool ParseUInt(const char* in, uint32_t* out) {
298 char* end;
299 long long int result = strtoll(in, &end, 0);
300 if (in == end || *end != '\0') {
301 return false;
302 }
303 if (result < std::numeric_limits<uint32_t>::min() ||
304 std::numeric_limits<uint32_t>::max() < result) {
305 return false;
306 }
307 *out = static_cast<uint32_t>(result);
308 return true;
309 }
Andreas Gamped089ca12016-06-27 14:25:30 -0700310
Andreas Gampec4ced4f2017-04-14 20:39:56 -0700311 bool ReadArguments(int argc, char** argv) {
Calin Juravlec9e76792018-02-01 14:44:56 +0000312 return parameters_.ReadArguments(argc, const_cast<const char**>(argv));
Andreas Gampe73dae112015-11-19 14:12:14 -0800313 }
314
Orion Hodson640cb762020-03-26 09:24:41 +0000315 void PrepareEnvironmentVariables() {
Andreas Gamped089ca12016-06-27 14:25:30 -0700316 environ_.push_back(StringPrintf("BOOTCLASSPATH=%s", boot_classpath_.c_str()));
317 environ_.push_back(StringPrintf("ANDROID_DATA=%s", GetOTADataDirectory().c_str()));
318 environ_.push_back(StringPrintf("ANDROID_ROOT=%s", android_root_.c_str()));
Andreas Gampe73dae112015-11-19 14:12:14 -0800319
320 for (const std::string& e : environ_) {
321 putenv(const_cast<char*>(e.c_str()));
322 }
323 }
324
Orion Hodson640cb762020-03-26 09:24:41 +0000325 // Ensure that we have the right boot image and cache file structures.
326 bool EnsureBootImageAndDalvikCache() const {
Calin Juravlec9e76792018-02-01 14:44:56 +0000327 if (parameters_.instruction_set == nullptr) {
Andreas Gampe73dae112015-11-19 14:12:14 -0800328 LOG(ERROR) << "Instruction set missing.";
329 return false;
330 }
Calin Juravlec9e76792018-02-01 14:44:56 +0000331 const char* isa = parameters_.instruction_set;
Andreas Gamped089ca12016-06-27 14:25:30 -0700332 std::string dalvik_cache = GetOTADataDirectory() + "/" + DALVIK_CACHE;
Andreas Gampe73dae112015-11-19 14:12:14 -0800333 std::string isa_path = dalvik_cache + "/" + isa;
Andreas Gampe73dae112015-11-19 14:12:14 -0800334
Andreas Gamped089ca12016-06-27 14:25:30 -0700335 // Reset umask in otapreopt, so that we control the the access for the files we create.
336 umask(0);
337
Andreas Gampe73dae112015-11-19 14:12:14 -0800338 // Create the directories, if necessary.
339 if (access(dalvik_cache.c_str(), F_OK) != 0) {
Andreas Gamped089ca12016-06-27 14:25:30 -0700340 if (!CreatePath(dalvik_cache)) {
341 PLOG(ERROR) << "Could not create dalvik-cache dir " << dalvik_cache;
Andreas Gampe73dae112015-11-19 14:12:14 -0800342 return false;
343 }
344 }
345 if (access(isa_path.c_str(), F_OK) != 0) {
Andreas Gamped089ca12016-06-27 14:25:30 -0700346 if (!CreatePath(isa_path)) {
Andreas Gampe73dae112015-11-19 14:12:14 -0800347 PLOG(ERROR) << "Could not create dalvik-cache isa dir";
348 return false;
349 }
350 }
351
Orion Hodson640cb762020-03-26 09:24:41 +0000352 // Clear cached artifacts.
353 ClearDirectory(isa_path);
354
355 // Check whether we have a boot image.
Andreas Gampebd7aef12018-10-23 13:58:44 -0700356 // TODO: check that the files are correct wrt/ jars.
Orion Hodson640cb762020-03-26 09:24:41 +0000357 std::string preopted_boot_art_path =
358 StringPrintf("/apex/com.android.art/javalib/%s/boot.art", isa);
359 if (access(preopted_boot_art_path.c_str(), F_OK) != 0) {
360 PLOG(ERROR) << "Bad access() to " << preopted_boot_art_path;
361 return false;
Andreas Gampebd7aef12018-10-23 13:58:44 -0700362 }
363
Orion Hodson640cb762020-03-26 09:24:41 +0000364 return true;
Andreas Gampe5709b572016-02-12 17:42:59 -0800365 }
366
Andreas Gamped089ca12016-06-27 14:25:30 -0700367 static bool CreatePath(const std::string& path) {
368 // Create the given path. Use string processing instead of dirname, as dirname's need for
369 // a writable char buffer is painful.
370
371 // First, try to use the full path.
372 if (mkdir(path.c_str(), 0711) == 0) {
373 return true;
374 }
375 if (errno != ENOENT) {
376 PLOG(ERROR) << "Could not create path " << path;
377 return false;
378 }
379
380 // Now find the parent and try that first.
381 size_t last_slash = path.find_last_of('/');
382 if (last_slash == std::string::npos || last_slash == 0) {
383 PLOG(ERROR) << "Could not create " << path;
384 return false;
385 }
386
387 if (!CreatePath(path.substr(0, last_slash))) {
388 return false;
389 }
390
391 if (mkdir(path.c_str(), 0711) == 0) {
392 return true;
393 }
394 PLOG(ERROR) << "Could not create " << path;
395 return false;
396 }
397
398 static void ClearDirectory(const std::string& dir) {
399 DIR* c_dir = opendir(dir.c_str());
400 if (c_dir == nullptr) {
401 PLOG(WARNING) << "Unable to open " << dir << " to delete it's contents";
402 return;
403 }
404
405 for (struct dirent* de = readdir(c_dir); de != nullptr; de = readdir(c_dir)) {
406 const char* name = de->d_name;
407 if (strcmp(name, ".") == 0 || strcmp(name, "..") == 0) {
408 continue;
409 }
410 // We only want to delete regular files and symbolic links.
411 std::string file = StringPrintf("%s/%s", dir.c_str(), name);
412 if (de->d_type != DT_REG && de->d_type != DT_LNK) {
413 LOG(WARNING) << "Unexpected file "
414 << file
415 << " of type "
416 << std::hex
417 << de->d_type
418 << " encountered.";
419 } else {
420 // Try to unlink the file.
421 if (unlink(file.c_str()) != 0) {
422 PLOG(ERROR) << "Unable to unlink " << file;
423 }
424 }
425 }
426 CHECK_EQ(0, closedir(c_dir)) << "Unable to close directory.";
427 }
428
Andreas Gampe73dae112015-11-19 14:12:14 -0800429 static const char* ParseNull(const char* arg) {
430 return (strcmp(arg, "!") == 0) ? nullptr : arg;
431 }
432
Andreas Gamped089ca12016-06-27 14:25:30 -0700433 bool ShouldSkipPreopt() const {
Andreas Gampe56f79f92016-06-08 15:11:37 -0700434 // There's one thing we have to be careful about: we may/will be asked to compile an app
435 // living in the system image. This may be a valid request - if the app wasn't compiled,
436 // e.g., if the system image wasn't large enough to include preopted files. However, the
437 // data we have is from the old system, so the driver (the OTA service) can't actually
438 // know. Thus, we will get requests for apps that have preopted components. To avoid
439 // duplication (we'd generate files that are not used and are *not* cleaned up), do two
440 // simple checks:
441 //
442 // 1) Does the apk_path start with the value of ANDROID_ROOT? (~in the system image)
443 // (For simplicity, assume the value of ANDROID_ROOT does not contain a symlink.)
444 //
445 // 2) If you replace the name in the apk_path with "oat," does the path exist?
446 // (=have a subdirectory for preopted files)
447 //
448 // If the answer to both is yes, skip the dexopt.
449 //
450 // Note: while one may think it's OK to call dexopt and it will fail (because APKs should
451 // be stripped), that's not true for APKs signed outside the build system (so the
452 // jar content must be exactly the same).
453
454 // (This is ugly as it's the only thing where we need to understand the contents
Calin Juravlec9e76792018-02-01 14:44:56 +0000455 // of parameters_, but it beats postponing the decision or using the call-
Andreas Gampe56f79f92016-06-08 15:11:37 -0700456 // backs to do weird things.)
Calin Juravlec9e76792018-02-01 14:44:56 +0000457 const char* apk_path = parameters_.apk_path;
Andreas Gampec4ced4f2017-04-14 20:39:56 -0700458 CHECK(apk_path != nullptr);
Elliott Hughes969e4f82017-12-20 12:34:09 -0800459 if (StartsWith(apk_path, android_root_)) {
Andreas Gampec4ced4f2017-04-14 20:39:56 -0700460 const char* last_slash = strrchr(apk_path, '/');
Andreas Gampe56f79f92016-06-08 15:11:37 -0700461 if (last_slash != nullptr) {
Andreas Gampec4ced4f2017-04-14 20:39:56 -0700462 std::string path(apk_path, last_slash - apk_path + 1);
Andreas Gampe56f79f92016-06-08 15:11:37 -0700463 CHECK(EndsWith(path, "/"));
464 path = path + "oat";
465 if (access(path.c_str(), F_OK) == 0) {
Calin Juravleb3591f62017-11-17 16:38:17 -0800466 LOG(INFO) << "Skipping A/B OTA preopt of already preopted package " << apk_path;
Andreas Gamped089ca12016-06-27 14:25:30 -0700467 return true;
Andreas Gampe56f79f92016-06-08 15:11:37 -0700468 }
469 }
470 }
471
Andreas Gamped089ca12016-06-27 14:25:30 -0700472 // Another issue is unavailability of files in the new system. If the partition
473 // layout changes, otapreopt_chroot may not know about this. Then files from that
474 // partition will not be available and fail to build. This is problematic, as
475 // this tool will wipe the OTA artifact cache and try again (for robustness after
476 // a failed OTA with remaining cache artifacts).
Andreas Gampec4ced4f2017-04-14 20:39:56 -0700477 if (access(apk_path, F_OK) != 0) {
Calin Juravleb3591f62017-11-17 16:38:17 -0800478 LOG(WARNING) << "Skipping A/B OTA preopt of non-existing package " << apk_path;
Andreas Gamped089ca12016-06-27 14:25:30 -0700479 return true;
480 }
481
482 return false;
483 }
484
Calin Juravlec9e76792018-02-01 14:44:56 +0000485 // Run dexopt with the parameters of parameters_.
Calin Juravlecfcd6aa2018-01-18 20:23:17 -0800486 // TODO(calin): embed the profile name in the parameters.
Andreas Gampeb39d2f02017-04-17 20:04:02 -0700487 int Dexopt() {
Alex Light6c857b72021-02-25 13:37:13 -0800488 std::string error;
489 int res = dexopt(parameters_.apk_path,
490 parameters_.uid,
491 parameters_.pkgName,
492 parameters_.instruction_set,
493 parameters_.dexopt_needed,
494 parameters_.oat_dir,
495 parameters_.dexopt_flags,
496 parameters_.compiler_filter,
497 parameters_.volume_uuid,
498 parameters_.shared_libraries,
499 parameters_.se_info,
500 parameters_.downgrade,
501 parameters_.target_sdk_version,
502 parameters_.profile_name,
503 parameters_.dex_metadata_path,
504 parameters_.compilation_reason,
505 &error);
506 if (res != 0) {
507 LOG(ERROR) << "During preopt of " << parameters_.apk_path << " got result " << res
508 << " error: " << error;
509 }
510 return res;
Andreas Gampe73dae112015-11-19 14:12:14 -0800511 }
512
Andreas Gampeb39d2f02017-04-17 20:04:02 -0700513 int RunPreopt() {
514 if (ShouldSkipPreopt()) {
515 return 0;
516 }
517
518 int dexopt_result = Dexopt();
519 if (dexopt_result == 0) {
520 return 0;
521 }
522
Andreas Gampeb39d2f02017-04-17 20:04:02 -0700523 // If this was a profile-guided run, we may have profile version issues. Try to downgrade,
524 // if possible.
Calin Juravlec9e76792018-02-01 14:44:56 +0000525 if ((parameters_.dexopt_flags & DEXOPT_PROFILE_GUIDED) == 0) {
Andreas Gampeb39d2f02017-04-17 20:04:02 -0700526 return dexopt_result;
527 }
528
529 LOG(WARNING) << "Downgrading compiler filter in an attempt to progress compilation";
Calin Juravlec9e76792018-02-01 14:44:56 +0000530 parameters_.dexopt_flags &= ~DEXOPT_PROFILE_GUIDED;
Andreas Gampeb39d2f02017-04-17 20:04:02 -0700531 return Dexopt();
532 }
533
Andreas Gampe73dae112015-11-19 14:12:14 -0800534 ////////////////////////////////////
535 // Helpers, mostly taken from ART //
536 ////////////////////////////////////
537
Andreas Gampe73dae112015-11-19 14:12:14 -0800538 // Choose a random relocation offset. Taken from art/runtime/gc/image_space.cc.
539 static int32_t ChooseRelocationOffsetDelta(int32_t min_delta, int32_t max_delta) {
540 constexpr size_t kPageSize = PAGE_SIZE;
Elliott Hughes0beed272020-07-29 14:28:25 -0700541 static_assert(IsPowerOfTwo(kPageSize), "page size must be power of two");
Andreas Gampe73dae112015-11-19 14:12:14 -0800542 CHECK_EQ(min_delta % kPageSize, 0u);
543 CHECK_EQ(max_delta % kPageSize, 0u);
544 CHECK_LT(min_delta, max_delta);
545
546 std::default_random_engine generator;
547 generator.seed(GetSeed());
548 std::uniform_int_distribution<int32_t> distribution(min_delta, max_delta);
549 int32_t r = distribution(generator);
550 if (r % 2 == 0) {
551 r = RoundUp(r, kPageSize);
552 } else {
553 r = RoundDown(r, kPageSize);
554 }
555 CHECK_LE(min_delta, r);
556 CHECK_GE(max_delta, r);
557 CHECK_EQ(r % kPageSize, 0u);
558 return r;
559 }
560
561 static uint64_t GetSeed() {
562#ifdef __BIONIC__
563 // Bionic exposes arc4random, use it.
564 uint64_t random_data;
565 arc4random_buf(&random_data, sizeof(random_data));
566 return random_data;
567#else
568#error "This is only supposed to run with bionic. Otherwise, implement..."
569#endif
570 }
571
572 void AddCompilerOptionFromSystemProperty(const char* system_property,
573 const char* prefix,
574 bool runtime,
Andreas Gamped089ca12016-06-27 14:25:30 -0700575 std::vector<std::string>& out) const {
576 const std::string* value = system_properties_.GetProperty(system_property);
Andreas Gampe73dae112015-11-19 14:12:14 -0800577 if (value != nullptr) {
578 if (runtime) {
579 out.push_back("--runtime-arg");
580 }
581 if (prefix != nullptr) {
582 out.push_back(StringPrintf("%s%s", prefix, value->c_str()));
583 } else {
584 out.push_back(*value);
585 }
586 }
587 }
588
Andreas Gamped089ca12016-06-27 14:25:30 -0700589 static constexpr const char* kBootClassPathPropertyName = "BOOTCLASSPATH";
590 static constexpr const char* kAndroidRootPathPropertyName = "ANDROID_ROOT";
591 static constexpr const char* kAndroidDataPathPropertyName = "ANDROID_DATA";
592 // The index of the instruction-set string inside the package parameters. Needed for
593 // some special-casing that requires knowledge of the instruction-set.
594 static constexpr size_t kISAIndex = 3;
595
Andreas Gampe73dae112015-11-19 14:12:14 -0800596 // Stores the system properties read out of the B partition. We need to use these properties
597 // to compile, instead of the A properties we could get from init/get_property.
598 SystemProperties system_properties_;
599
Andreas Gamped089ca12016-06-27 14:25:30 -0700600 // Some select properties that are always needed.
Andreas Gamped089ca12016-06-27 14:25:30 -0700601 std::string android_root_;
602 std::string android_data_;
603 std::string boot_classpath_;
604 std::string asec_mountpoint_;
605
Calin Juravlec9e76792018-02-01 14:44:56 +0000606 OTAPreoptParameters parameters_;
Andreas Gampe73dae112015-11-19 14:12:14 -0800607
608 // Store environment values we need to set.
609 std::vector<std::string> environ_;
610};
611
612OTAPreoptService gOps;
613
614////////////////////////
615// Plug-in functions. //
616////////////////////////
617
618int get_property(const char *key, char *value, const char *default_value) {
Andreas Gampe73dae112015-11-19 14:12:14 -0800619 return gOps.GetProperty(key, value, default_value);
620}
621
622// Compute the output path of
623bool calculate_oat_file_path(char path[PKG_PATH_MAX], const char *oat_dir,
624 const char *apk_path,
625 const char *instruction_set) {
Dan Austin9c8f93a2016-06-03 16:15:54 -0700626 const char *file_name_start;
627 const char *file_name_end;
Andreas Gampe73dae112015-11-19 14:12:14 -0800628
629 file_name_start = strrchr(apk_path, '/');
630 if (file_name_start == nullptr) {
631 ALOGE("apk_path '%s' has no '/'s in it\n", apk_path);
632 return false;
633 }
634 file_name_end = strrchr(file_name_start, '.');
635 if (file_name_end == nullptr) {
636 ALOGE("apk_path '%s' has no extension\n", apk_path);
637 return false;
638 }
639
640 // Calculate file_name
641 file_name_start++; // Move past '/', is valid as file_name_end is valid.
642 size_t file_name_len = file_name_end - file_name_start;
643 std::string file_name(file_name_start, file_name_len);
644
645 // <apk_parent_dir>/oat/<isa>/<file_name>.odex.b
Andreas Gamped089ca12016-06-27 14:25:30 -0700646 snprintf(path,
647 PKG_PATH_MAX,
648 "%s/%s/%s.odex.%s",
649 oat_dir,
650 instruction_set,
651 file_name.c_str(),
652 gOps.GetTargetSlot().c_str());
Andreas Gampe73dae112015-11-19 14:12:14 -0800653 return true;
654}
655
656/*
657 * Computes the odex file for the given apk_path and instruction_set.
658 * /system/framework/whatever.jar -> /system/framework/oat/<isa>/whatever.odex
659 *
660 * Returns false if it failed to determine the odex file path.
661 */
662bool calculate_odex_file_path(char path[PKG_PATH_MAX], const char *apk_path,
663 const char *instruction_set) {
Andreas Gampe73dae112015-11-19 14:12:14 -0800664 const char *path_end = strrchr(apk_path, '/');
665 if (path_end == nullptr) {
666 ALOGE("apk_path '%s' has no '/'s in it?!\n", apk_path);
667 return false;
668 }
669 std::string path_component(apk_path, path_end - apk_path);
670
671 const char *name_begin = path_end + 1;
672 const char *extension_start = strrchr(name_begin, '.');
673 if (extension_start == nullptr) {
674 ALOGE("apk_path '%s' has no extension.\n", apk_path);
675 return false;
676 }
677 std::string name_component(name_begin, extension_start - name_begin);
678
Andreas Gamped089ca12016-06-27 14:25:30 -0700679 std::string new_path = StringPrintf("%s/oat/%s/%s.odex.%s",
Andreas Gampe73dae112015-11-19 14:12:14 -0800680 path_component.c_str(),
681 instruction_set,
Andreas Gamped089ca12016-06-27 14:25:30 -0700682 name_component.c_str(),
683 gOps.GetTargetSlot().c_str());
684 if (new_path.length() >= PKG_PATH_MAX) {
685 LOG(ERROR) << "apk_path of " << apk_path << " is too long: " << new_path;
686 return false;
687 }
Andreas Gampe73dae112015-11-19 14:12:14 -0800688 strcpy(path, new_path.c_str());
689 return true;
690}
691
692bool create_cache_path(char path[PKG_PATH_MAX],
693 const char *src,
694 const char *instruction_set) {
695 size_t srclen = strlen(src);
696
697 /* demand that we are an absolute path */
698 if ((src == 0) || (src[0] != '/') || strstr(src,"..")) {
699 return false;
700 }
701
702 if (srclen > PKG_PATH_MAX) { // XXX: PKG_NAME_MAX?
703 return false;
704 }
705
706 std::string from_src = std::string(src + 1);
707 std::replace(from_src.begin(), from_src.end(), '/', '@');
708
709 std::string assembled_path = StringPrintf("%s/%s/%s/%s%s",
Andreas Gamped089ca12016-06-27 14:25:30 -0700710 gOps.GetOTADataDirectory().c_str(),
Andreas Gampe73dae112015-11-19 14:12:14 -0800711 DALVIK_CACHE,
712 instruction_set,
713 from_src.c_str(),
David Brazdil249c1792016-09-06 15:35:28 +0100714 DALVIK_CACHE_POSTFIX);
Andreas Gampe73dae112015-11-19 14:12:14 -0800715
716 if (assembled_path.length() + 1 > PKG_PATH_MAX) {
717 return false;
718 }
719 strcpy(path, assembled_path.c_str());
720
721 return true;
722}
723
Andreas Gampe73dae112015-11-19 14:12:14 -0800724static int log_callback(int type, const char *fmt, ...) {
725 va_list ap;
726 int priority;
727
728 switch (type) {
729 case SELINUX_WARNING:
730 priority = ANDROID_LOG_WARN;
731 break;
732 case SELINUX_INFO:
733 priority = ANDROID_LOG_INFO;
734 break;
735 default:
736 priority = ANDROID_LOG_ERROR;
737 break;
738 }
739 va_start(ap, fmt);
740 LOG_PRI_VA(priority, "SELinux", fmt, ap);
741 va_end(ap);
742 return 0;
743}
744
745static int otapreopt_main(const int argc, char *argv[]) {
746 int selinux_enabled = (is_selinux_enabled() > 0);
747
748 setenv("ANDROID_LOG_TAGS", "*:v", 1);
749 android::base::InitLogging(argv);
750
Andreas Gampe73dae112015-11-19 14:12:14 -0800751 if (argc < 2) {
752 ALOGE("Expecting parameters");
753 exit(1);
754 }
755
756 union selinux_callback cb;
757 cb.func_log = log_callback;
758 selinux_set_callback(SELINUX_CB_LOG, cb);
759
Andreas Gampe73dae112015-11-19 14:12:14 -0800760 if (selinux_enabled && selinux_status_open(true) < 0) {
761 ALOGE("Could not open selinux status; exiting.\n");
762 exit(1);
763 }
764
765 int ret = android::installd::gOps.Main(argc, argv);
766
767 return ret;
768}
769
770} // namespace installd
771} // namespace android
772
773int main(const int argc, char *argv[]) {
774 return android::installd::otapreopt_main(argc, argv);
775}