1 /*
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>
19 #include <limits>
20 #include <random>
21 #include <regex>
22 #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>
29 
30 #include <android-base/logging.h>
31 #include <android-base/macros.h>
32 #include <android-base/stringprintf.h>
33 #include <android-base/strings.h>
34 #include <art_image_values.h>
35 #include <cutils/fs.h>
36 #include <cutils/properties.h>
37 #include <dex2oat_return_codes.h>
38 #include <log/log.h>
39 #include <private/android_filesystem_config.h>
40 
41 #include "dexopt.h"
42 #include "file_parsing.h"
43 #include "globals.h"
44 #include "installd_constants.h"
45 #include "installd_deps.h"  // Need to fill in requirements of commands.
46 #include "otapreopt_parameters.h"
47 #include "otapreopt_utils.h"
48 #include "system_properties.h"
49 #include "utils.h"
50 
51 #ifndef LOG_TAG
52 #define LOG_TAG "otapreopt"
53 #endif
54 
55 #define BUFFER_MAX    1024  /* input buffer for commands */
56 #define TOKEN_MAX     16    /* max number of arguments in buffer */
57 #define REPLY_MAX     256   /* largest reply allowed */
58 
59 using android::base::EndsWith;
60 using android::base::Split;
61 using android::base::StartsWith;
62 using android::base::StringPrintf;
63 
64 namespace android {
65 namespace installd {
66 
67 // Check expected values for dexopt flags. If you need to change this:
68 //
69 //   RUN AN A/B OTA TO MAKE SURE THINGS STILL WORK!
70 //
71 // You most likely need to increase the protocol version and all that entails!
72 
73 static_assert(DEXOPT_PUBLIC         == 1 << 1, "DEXOPT_PUBLIC unexpected.");
74 static_assert(DEXOPT_DEBUGGABLE     == 1 << 2, "DEXOPT_DEBUGGABLE unexpected.");
75 static_assert(DEXOPT_BOOTCOMPLETE   == 1 << 3, "DEXOPT_BOOTCOMPLETE unexpected.");
76 static_assert(DEXOPT_PROFILE_GUIDED == 1 << 4, "DEXOPT_PROFILE_GUIDED unexpected.");
77 static_assert(DEXOPT_SECONDARY_DEX  == 1 << 5, "DEXOPT_SECONDARY_DEX unexpected.");
78 static_assert(DEXOPT_FORCE          == 1 << 6, "DEXOPT_FORCE unexpected.");
79 static_assert(DEXOPT_STORAGE_CE     == 1 << 7, "DEXOPT_STORAGE_CE unexpected.");
80 static_assert(DEXOPT_STORAGE_DE     == 1 << 8, "DEXOPT_STORAGE_DE unexpected.");
81 static_assert(DEXOPT_ENABLE_HIDDEN_API_CHECKS == 1 << 10,
82         "DEXOPT_ENABLE_HIDDEN_API_CHECKS unexpected");
83 static_assert(DEXOPT_GENERATE_COMPACT_DEX == 1 << 11, "DEXOPT_GENERATE_COMPACT_DEX unexpected");
84 static_assert(DEXOPT_GENERATE_APP_IMAGE == 1 << 12, "DEXOPT_GENERATE_APP_IMAGE unexpected");
85 
86 static_assert(DEXOPT_MASK           == (0x3dfe | DEXOPT_IDLE_BACKGROUND_JOB),
87               "DEXOPT_MASK unexpected.");
88 
89 
90 template<typename T>
IsPowerOfTwo(T x)91 static constexpr bool IsPowerOfTwo(T x) {
92   static_assert(std::is_integral<T>::value, "T must be integral");
93   // TODO: assert unsigned. There is currently many uses with signed values.
94   return (x & (x - 1)) == 0;
95 }
96 
97 template<typename T>
RoundDown(T x,typename std::decay<T>::type n)98 static constexpr T RoundDown(T x, typename std::decay<T>::type n) {
99     return (x & -n);
100 }
101 
102 template<typename T>
RoundUp(T x,typename std::remove_reference<T>::type n)103 static constexpr T RoundUp(T x, typename std::remove_reference<T>::type n) {
104     return RoundDown(x + n - 1, n);
105 }
106 
107 class OTAPreoptService {
108  public:
109     // Main driver. Performs the following steps.
110     //
111     // 1) Parse options (read system properties etc from B partition).
112     //
113     // 2) Read in package data.
114     //
115     // 3) Prepare environment variables.
116     //
117     // 4) Prepare(compile) boot image, if necessary.
118     //
119     // 5) Run update.
Main(int argc,char ** argv)120     int Main(int argc, char** argv) {
121         if (!ReadArguments(argc, argv)) {
122             LOG(ERROR) << "Failed reading command line.";
123             return 1;
124         }
125 
126         if (!ReadSystemProperties()) {
127             LOG(ERROR)<< "Failed reading system properties.";
128             return 2;
129         }
130 
131         if (!ReadEnvironment()) {
132             LOG(ERROR) << "Failed reading environment properties.";
133             return 3;
134         }
135 
136         if (!CheckAndInitializeInstalldGlobals()) {
137             LOG(ERROR) << "Failed initializing globals.";
138             return 4;
139         }
140 
141         PrepareEnvironmentVariables();
142 
143         if (!EnsureBootImageAndDalvikCache()) {
144             LOG(ERROR) << "Bad boot image.";
145             return 5;
146         }
147 
148         int dexopt_retcode = RunPreopt();
149 
150         return dexopt_retcode;
151     }
152 
GetProperty(const char * key,char * value,const char * default_value) const153     int GetProperty(const char* key, char* value, const char* default_value) const {
154         const std::string* prop_value = system_properties_.GetProperty(key);
155         if (prop_value == nullptr) {
156             if (default_value == nullptr) {
157                 return 0;
158             }
159             // Copy in the default value.
160             strlcpy(value, default_value, kPropertyValueMax - 1);
161             value[kPropertyValueMax - 1] = 0;
162             return strlen(default_value);// TODO: Need to truncate?
163         }
164         size_t size = std::min(kPropertyValueMax - 1, prop_value->length()) + 1;
165         strlcpy(value, prop_value->data(), size);
166         return static_cast<int>(size - 1);
167     }
168 
GetOTADataDirectory() const169     std::string GetOTADataDirectory() const {
170         return StringPrintf("%s/%s", GetOtaDirectoryPrefix().c_str(), GetTargetSlot().c_str());
171     }
172 
GetTargetSlot() const173     const std::string& GetTargetSlot() const {
174         return parameters_.target_slot;
175     }
176 
177 private:
178 
ReadSystemProperties()179     bool ReadSystemProperties() {
180         static constexpr const char* kPropertyFiles[] = {
181                 "/default.prop", "/system/build.prop"
182         };
183 
184         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;
191     }
192 
ReadEnvironment()193     bool ReadEnvironment() {
194         // 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             }
204 
205             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) {
217             return false;
218         }
219 
220         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 
GetAndroidData() const243     const std::string& GetAndroidData() const {
244         return android_data_;
245     }
246 
GetAndroidRoot() const247     const std::string& GetAndroidRoot() const {
248         return android_root_;
249     }
250 
GetOtaDirectoryPrefix() const251     const std::string GetOtaDirectoryPrefix() const {
252         return GetAndroidData() + "/ota";
253     }
254 
CheckAndInitializeInstalldGlobals()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;
275         }
276 
277         return true;
278     }
279 
ParseBool(const char * in)280     bool ParseBool(const char* in) {
281         if (strcmp(in, "true") == 0) {
282             return true;
283         }
284         return false;
285     }
286 
ParseUInt(const char * in,uint32_t * out)287     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     }
300 
ReadArguments(int argc,char ** argv)301     bool ReadArguments(int argc, char** argv) {
302         return parameters_.ReadArguments(argc, const_cast<const char**>(argv));
303     }
304 
PrepareEnvironmentVariables()305     void PrepareEnvironmentVariables() {
306         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()));
309 
310         for (const std::string& e : environ_) {
311             putenv(const_cast<char*>(e.c_str()));
312         }
313     }
314 
315     // Ensure that we have the right boot image and cache file structures.
EnsureBootImageAndDalvikCache() const316     bool EnsureBootImageAndDalvikCache() const {
317         if (parameters_.instruction_set == nullptr) {
318             LOG(ERROR) << "Instruction set missing.";
319             return false;
320         }
321         const char* isa = parameters_.instruction_set;
322         std::string dalvik_cache = GetOTADataDirectory() + "/" + DALVIK_CACHE;
323         std::string isa_path = dalvik_cache + "/" + isa;
324 
325         // Reset umask in otapreopt, so that we control the the access for the files we create.
326         umask(0);
327 
328         // Create the directories, if necessary.
329         if (access(dalvik_cache.c_str(), F_OK) != 0) {
330             if (!CreatePath(dalvik_cache)) {
331                 PLOG(ERROR) << "Could not create dalvik-cache dir " << dalvik_cache;
332                 return false;
333             }
334         }
335         if (access(isa_path.c_str(), F_OK) != 0) {
336             if (!CreatePath(isa_path)) {
337                 PLOG(ERROR) << "Could not create dalvik-cache isa dir";
338                 return false;
339             }
340         }
341 
342         // Clear cached artifacts.
343         ClearDirectory(isa_path);
344 
345         // Check whether we have a boot image.
346         // TODO: check that the files are correct wrt/ jars.
347         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;
352         }
353 
354         return true;
355     }
356 
CreatePath(const std::string & path)357     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 
ClearDirectory(const std::string & dir)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 
ParseNull(const char * arg)419     static const char* ParseNull(const char* arg) {
420         return (strcmp(arg, "!") == 0) ? nullptr : arg;
421     }
422 
ShouldSkipPreopt() const423     bool ShouldSkipPreopt() const {
424         // 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
445         //        of parameters_, but it beats postponing the decision or using the call-
446         //        backs to do weird things.)
447         const char* apk_path = parameters_.apk_path;
448         CHECK(apk_path != nullptr);
449         if (StartsWith(apk_path, android_root_)) {
450             const char* last_slash = strrchr(apk_path, '/');
451             if (last_slash != nullptr) {
452                 std::string path(apk_path, last_slash - apk_path + 1);
453                 CHECK(EndsWith(path, "/"));
454                 path = path + "oat";
455                 if (access(path.c_str(), F_OK) == 0) {
456                     LOG(INFO) << "Skipping A/B OTA preopt of already preopted package " << apk_path;
457                     return true;
458                 }
459             }
460         }
461 
462         // 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).
467         if (access(apk_path, F_OK) != 0) {
468             LOG(WARNING) << "Skipping A/B OTA preopt of non-existing package " << apk_path;
469             return true;
470         }
471 
472         return false;
473     }
474 
475     // Run dexopt with the parameters of parameters_.
476     // TODO(calin): embed the profile name in the parameters.
Dexopt()477     int Dexopt() {
478         std::string dummy;
479         return 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                       &dummy);
496     }
497 
RunPreopt()498     int RunPreopt() {
499         if (ShouldSkipPreopt()) {
500             return 0;
501         }
502 
503         int dexopt_result = Dexopt();
504         if (dexopt_result == 0) {
505             return 0;
506         }
507 
508         // If this was a profile-guided run, we may have profile version issues. Try to downgrade,
509         // if possible.
510         if ((parameters_.dexopt_flags & DEXOPT_PROFILE_GUIDED) == 0) {
511             return dexopt_result;
512         }
513 
514         LOG(WARNING) << "Downgrading compiler filter in an attempt to progress compilation";
515         parameters_.dexopt_flags &= ~DEXOPT_PROFILE_GUIDED;
516         return Dexopt();
517     }
518 
519     ////////////////////////////////////
520     // Helpers, mostly taken from ART //
521     ////////////////////////////////////
522 
523     // Choose a random relocation offset. Taken from art/runtime/gc/image_space.cc.
ChooseRelocationOffsetDelta(int32_t min_delta,int32_t max_delta)524     static int32_t ChooseRelocationOffsetDelta(int32_t min_delta, int32_t max_delta) {
525         constexpr size_t kPageSize = PAGE_SIZE;
526         static_assert(IsPowerOfTwo(kPageSize), "page size must be power of two");
527         CHECK_EQ(min_delta % kPageSize, 0u);
528         CHECK_EQ(max_delta % kPageSize, 0u);
529         CHECK_LT(min_delta, max_delta);
530 
531         std::default_random_engine generator;
532         generator.seed(GetSeed());
533         std::uniform_int_distribution<int32_t> distribution(min_delta, max_delta);
534         int32_t r = distribution(generator);
535         if (r % 2 == 0) {
536             r = RoundUp(r, kPageSize);
537         } else {
538             r = RoundDown(r, kPageSize);
539         }
540         CHECK_LE(min_delta, r);
541         CHECK_GE(max_delta, r);
542         CHECK_EQ(r % kPageSize, 0u);
543         return r;
544     }
545 
GetSeed()546     static uint64_t GetSeed() {
547 #ifdef __BIONIC__
548         // Bionic exposes arc4random, use it.
549         uint64_t random_data;
550         arc4random_buf(&random_data, sizeof(random_data));
551         return random_data;
552 #else
553 #error "This is only supposed to run with bionic. Otherwise, implement..."
554 #endif
555     }
556 
AddCompilerOptionFromSystemProperty(const char * system_property,const char * prefix,bool runtime,std::vector<std::string> & out) const557     void AddCompilerOptionFromSystemProperty(const char* system_property,
558             const char* prefix,
559             bool runtime,
560             std::vector<std::string>& out) const {
561         const std::string* value = system_properties_.GetProperty(system_property);
562         if (value != nullptr) {
563             if (runtime) {
564                 out.push_back("--runtime-arg");
565             }
566             if (prefix != nullptr) {
567                 out.push_back(StringPrintf("%s%s", prefix, value->c_str()));
568             } else {
569                 out.push_back(*value);
570             }
571         }
572     }
573 
574     static constexpr const char* kBootClassPathPropertyName = "BOOTCLASSPATH";
575     static constexpr const char* kAndroidRootPathPropertyName = "ANDROID_ROOT";
576     static constexpr const char* kAndroidDataPathPropertyName = "ANDROID_DATA";
577     // The index of the instruction-set string inside the package parameters. Needed for
578     // some special-casing that requires knowledge of the instruction-set.
579     static constexpr size_t kISAIndex = 3;
580 
581     // Stores the system properties read out of the B partition. We need to use these properties
582     // to compile, instead of the A properties we could get from init/get_property.
583     SystemProperties system_properties_;
584 
585     // Some select properties that are always needed.
586     std::string android_root_;
587     std::string android_data_;
588     std::string boot_classpath_;
589     std::string asec_mountpoint_;
590 
591     OTAPreoptParameters parameters_;
592 
593     // Store environment values we need to set.
594     std::vector<std::string> environ_;
595 };
596 
597 OTAPreoptService gOps;
598 
599 ////////////////////////
600 // Plug-in functions. //
601 ////////////////////////
602 
get_property(const char * key,char * value,const char * default_value)603 int get_property(const char *key, char *value, const char *default_value) {
604     return gOps.GetProperty(key, value, default_value);
605 }
606 
607 // Compute the output path of
calculate_oat_file_path(char path[PKG_PATH_MAX],const char * oat_dir,const char * apk_path,const char * instruction_set)608 bool calculate_oat_file_path(char path[PKG_PATH_MAX], const char *oat_dir,
609                              const char *apk_path,
610                              const char *instruction_set) {
611     const char *file_name_start;
612     const char *file_name_end;
613 
614     file_name_start = strrchr(apk_path, '/');
615     if (file_name_start == nullptr) {
616         ALOGE("apk_path '%s' has no '/'s in it\n", apk_path);
617         return false;
618     }
619     file_name_end = strrchr(file_name_start, '.');
620     if (file_name_end == nullptr) {
621         ALOGE("apk_path '%s' has no extension\n", apk_path);
622         return false;
623     }
624 
625     // Calculate file_name
626     file_name_start++;  // Move past '/', is valid as file_name_end is valid.
627     size_t file_name_len = file_name_end - file_name_start;
628     std::string file_name(file_name_start, file_name_len);
629 
630     // <apk_parent_dir>/oat/<isa>/<file_name>.odex.b
631     snprintf(path,
632              PKG_PATH_MAX,
633              "%s/%s/%s.odex.%s",
634              oat_dir,
635              instruction_set,
636              file_name.c_str(),
637              gOps.GetTargetSlot().c_str());
638     return true;
639 }
640 
641 /*
642  * Computes the odex file for the given apk_path and instruction_set.
643  * /system/framework/whatever.jar -> /system/framework/oat/<isa>/whatever.odex
644  *
645  * Returns false if it failed to determine the odex file path.
646  */
calculate_odex_file_path(char path[PKG_PATH_MAX],const char * apk_path,const char * instruction_set)647 bool calculate_odex_file_path(char path[PKG_PATH_MAX], const char *apk_path,
648                               const char *instruction_set) {
649     const char *path_end = strrchr(apk_path, '/');
650     if (path_end == nullptr) {
651         ALOGE("apk_path '%s' has no '/'s in it?!\n", apk_path);
652         return false;
653     }
654     std::string path_component(apk_path, path_end - apk_path);
655 
656     const char *name_begin = path_end + 1;
657     const char *extension_start = strrchr(name_begin, '.');
658     if (extension_start == nullptr) {
659         ALOGE("apk_path '%s' has no extension.\n", apk_path);
660         return false;
661     }
662     std::string name_component(name_begin, extension_start - name_begin);
663 
664     std::string new_path = StringPrintf("%s/oat/%s/%s.odex.%s",
665                                         path_component.c_str(),
666                                         instruction_set,
667                                         name_component.c_str(),
668                                         gOps.GetTargetSlot().c_str());
669     if (new_path.length() >= PKG_PATH_MAX) {
670         LOG(ERROR) << "apk_path of " << apk_path << " is too long: " << new_path;
671         return false;
672     }
673     strcpy(path, new_path.c_str());
674     return true;
675 }
676 
create_cache_path(char path[PKG_PATH_MAX],const char * src,const char * instruction_set)677 bool create_cache_path(char path[PKG_PATH_MAX],
678                        const char *src,
679                        const char *instruction_set) {
680     size_t srclen = strlen(src);
681 
682         /* demand that we are an absolute path */
683     if ((src == 0) || (src[0] != '/') || strstr(src,"..")) {
684         return false;
685     }
686 
687     if (srclen > PKG_PATH_MAX) {        // XXX: PKG_NAME_MAX?
688         return false;
689     }
690 
691     std::string from_src = std::string(src + 1);
692     std::replace(from_src.begin(), from_src.end(), '/', '@');
693 
694     std::string assembled_path = StringPrintf("%s/%s/%s/%s%s",
695                                               gOps.GetOTADataDirectory().c_str(),
696                                               DALVIK_CACHE,
697                                               instruction_set,
698                                               from_src.c_str(),
699                                               DALVIK_CACHE_POSTFIX);
700 
701     if (assembled_path.length() + 1 > PKG_PATH_MAX) {
702         return false;
703     }
704     strcpy(path, assembled_path.c_str());
705 
706     return true;
707 }
708 
log_callback(int type,const char * fmt,...)709 static int log_callback(int type, const char *fmt, ...) {
710     va_list ap;
711     int priority;
712 
713     switch (type) {
714         case SELINUX_WARNING:
715             priority = ANDROID_LOG_WARN;
716             break;
717         case SELINUX_INFO:
718             priority = ANDROID_LOG_INFO;
719             break;
720         default:
721             priority = ANDROID_LOG_ERROR;
722             break;
723     }
724     va_start(ap, fmt);
725     LOG_PRI_VA(priority, "SELinux", fmt, ap);
726     va_end(ap);
727     return 0;
728 }
729 
otapreopt_main(const int argc,char * argv[])730 static int otapreopt_main(const int argc, char *argv[]) {
731     int selinux_enabled = (is_selinux_enabled() > 0);
732 
733     setenv("ANDROID_LOG_TAGS", "*:v", 1);
734     android::base::InitLogging(argv);
735 
736     if (argc < 2) {
737         ALOGE("Expecting parameters");
738         exit(1);
739     }
740 
741     union selinux_callback cb;
742     cb.func_log = log_callback;
743     selinux_set_callback(SELINUX_CB_LOG, cb);
744 
745     if (selinux_enabled && selinux_status_open(true) < 0) {
746         ALOGE("Could not open selinux status; exiting.\n");
747         exit(1);
748     }
749 
750     int ret = android::installd::gOps.Main(argc, argv);
751 
752     return ret;
753 }
754 
755 }  // namespace installd
756 }  // namespace android
757 
main(const int argc,char * argv[])758 int main(const int argc, char *argv[]) {
759     return android::installd::otapreopt_main(argc, argv);
760 }
761