blob: 7fcce2c9c15dd29155e8735fe5c3e1e12541256f [file] [log] [blame]
/*
** Copyright 2016, The Android Open Source Project
**
** Licensed under the Apache License, Version 2.0 (the "License");
** you may not use this file except in compliance with the License.
** You may obtain a copy of the License at
**
** http://www.apache.org/licenses/LICENSE-2.0
**
** Unless required by applicable law or agreed to in writing, software
** distributed under the License is distributed on an "AS IS" BASIS,
** WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
** See the License for the specific language governing permissions and
** limitations under the License.
*/
#include <fcntl.h>
#include <linux/unistd.h>
#include <sys/mount.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <array>
#include <chrono>
#include <fstream>
#include <sstream>
#include <thread>
#include <android-base/file.h>
#include <android-base/logging.h>
#include <android-base/macros.h>
#include <android-base/parseint.h>
#include <android-base/scopeguard.h>
#include <android-base/stringprintf.h>
#include <android-base/strings.h>
#include <android-base/unique_fd.h>
#include <android/content/pm/IOtaDexopt.h>
#include <binder/IServiceManager.h>
#include <libdm/dm.h>
#include <selinux/android.h>
#include "installd_constants.h"
#include "otapreopt_utils.h"
#ifndef LOG_TAG
#define LOG_TAG "otapreopt"
#endif
using android::base::StringPrintf;
namespace android {
namespace installd {
using namespace std::literals::chrono_literals;
static void CloseDescriptor(int fd) {
if (fd >= 0) {
int result = close(fd);
UNUSED(result); // Ignore result. Printing to logcat will open a new descriptor
// that we do *not* want.
}
}
static void ActivateApexPackages() {
std::vector<std::string> apexd_cmd{"/system/bin/apexd", "--otachroot-bootstrap"};
std::string apexd_error_msg;
bool exec_result = Exec(apexd_cmd, &apexd_error_msg);
if (!exec_result) {
PLOG(ERROR) << "Running otapreopt failed: " << apexd_error_msg;
exit(220);
}
}
static void DeactivateApexPackages() {
std::vector<std::string> apexd_cmd{"/system/bin/apexd", "--unmount-all"};
std::string apexd_error_msg;
bool exec_result = Exec(apexd_cmd, &apexd_error_msg);
if (!exec_result) {
PLOG(ERROR) << "Running /system/bin/apexd --unmount-all failed: " << apexd_error_msg;
}
}
static void TryExtraMount(const char* name, const char* slot, const char* target) {
std::string partition_name = StringPrintf("%s%s", name, slot);
// See whether update_engine mounted a logical partition.
{
auto& dm = dm::DeviceMapper::Instance();
if (dm.GetState(partition_name) != dm::DmDeviceState::INVALID) {
std::string path;
if (dm.GetDmDevicePathByName(partition_name, &path)) {
int mount_result = mount(path.c_str(),
target,
"ext4",
MS_RDONLY,
/* data */ nullptr);
if (mount_result == 0) {
return;
}
}
}
}
// Fall back and attempt a direct mount.
std::string block_device = StringPrintf("/dev/block/by-name/%s", partition_name.c_str());
int mount_result = mount(block_device.c_str(),
target,
"ext4",
MS_RDONLY,
/* data */ nullptr);
UNUSED(mount_result);
}
static android::sp<android::content::pm::IOtaDexopt> GetDexoptService() {
auto binder = android::defaultServiceManager()->getService(android::String16("otadexopt"));
if (binder == nullptr) {
return nullptr;
}
return android::interface_cast<android::content::pm::IOtaDexopt>(binder);
}
static bool RunDexoptCommand(int argc, char **arg, const std::string& dexopt_cmd) {
// Incoming: cmd + status-fd + target-slot + cmd... | Incoming | = argc
// Outgoing: cmd + target-slot + cmd... | Outgoing | = argc - 1
std::vector<std::string> cmd;
cmd.reserve(argc);
cmd.push_back("/system/bin/otapreopt");
// The first parameter is the status file descriptor, skip.
for (size_t i = 2; i < static_cast<size_t>(argc); ++i) {
cmd.push_back(arg[i]);
}
for (const std::string& part : android::base::Split(dexopt_cmd, " ")) {
cmd.push_back(part);
}
// Fork and execute otapreopt in its own process.
std::string error_msg;
bool exec_result = Exec(cmd, &error_msg);
if (!exec_result) {
LOG(ERROR) << "Running otapreopt failed: " << error_msg;
}
return exec_result;
}
// Entry for otapreopt_chroot. Expected parameters are:
// [cmd] [status-fd] [target-slot] "dexopt" [dexopt-params]
// The file descriptor denoted by status-fd will be closed. The rest of the parameters will
// be passed on to otapreopt in the chroot.
static int otapreopt_chroot(const int argc, char **arg) {
// Validate arguments
// We need the command, status channel and target slot, at a minimum.
if(argc < 3) {
PLOG(ERROR) << "Not enough arguments.";
exit(208);
}
// Close all file descriptors. They are coming from the caller, we do not want to pass them
// on across our fork/exec into a different domain.
// 1) Default descriptors.
CloseDescriptor(STDIN_FILENO);
CloseDescriptor(STDOUT_FILENO);
CloseDescriptor(STDERR_FILENO);
int fd;
if (!android::base::ParseInt(arg[1], &fd)) {
LOG(ERROR) << "Failed to parse " << arg[1];
exit(225);
}
// Add O_CLOEXEC to status channel, since we don't want to pass it across fork/exec, but we need
// to keep it open in otapreopt_chroot to report progress
if (fcntl(fd, F_SETFD, FD_CLOEXEC) == -1) {
PLOG(ERROR) << "Failed to set O_CLOEXEC on " << fd;
exit(226);
}
// We need to run the otapreopt tool from the postinstall partition. As such, set up a
// mount namespace and change root.
// Create our own mount namespace.
if (unshare(CLONE_NEWNS) != 0) {
PLOG(ERROR) << "Failed to unshare() for otapreopt.";
exit(200);
}
// Make postinstall private, so that our changes don't propagate.
if (mount("", "/postinstall", nullptr, MS_PRIVATE, nullptr) != 0) {
PLOG(ERROR) << "Failed to mount private.";
exit(201);
}
// Bind mount necessary directories.
constexpr const char* kBindMounts[] = {
"/data", "/dev", "/proc", "/sys"
};
for (size_t i = 0; i < arraysize(kBindMounts); ++i) {
std::string trg = StringPrintf("/postinstall%s", kBindMounts[i]);
if (mount(kBindMounts[i], trg.c_str(), nullptr, MS_BIND, nullptr) != 0) {
PLOG(ERROR) << "Failed to bind-mount " << kBindMounts[i];
exit(202);
}
}
// Try to mount the vendor partition. update_engine doesn't do this for us, but we
// want it for vendor APKs.
// Notes:
// 1) We pretty much guess a name here and hope to find the partition by name.
// It is just as complicated and brittle to scan /proc/mounts. But this requires
// validating the target-slot so as not to try to mount some totally random path.
// 2) We're in a mount namespace here, so when we die, this will be cleaned up.
// 3) Ignore errors. Printing anything at this stage will open a file descriptor
// for logging.
if (!ValidateTargetSlotSuffix(arg[2])) {
LOG(ERROR) << "Target slot suffix not legal: " << arg[2];
exit(207);
}
TryExtraMount("vendor", arg[2], "/postinstall/vendor");
// Try to mount the product partition. update_engine doesn't do this for us, but we
// want it for product APKs. Same notes as vendor above.
TryExtraMount("product", arg[2], "/postinstall/product");
// Try to mount the system_ext partition. update_engine doesn't do this for
// us, but we want it for system_ext APKs. Same notes as vendor and product
// above.
TryExtraMount("system_ext", arg[2], "/postinstall/system_ext");
constexpr const char* kPostInstallLinkerconfig = "/postinstall/linkerconfig";
// Try to mount /postinstall/linkerconfig. we will set it up after performing the chroot
if (mount("tmpfs", kPostInstallLinkerconfig, "tmpfs", 0, nullptr) != 0) {
PLOG(ERROR) << "Failed to mount a tmpfs for " << kPostInstallLinkerconfig;
exit(215);
}
// Setup APEX mount point and its security context.
static constexpr const char* kPostinstallApexDir = "/postinstall/apex";
// The following logic is similar to the one in system/core/rootdir/init.rc:
//
// mount tmpfs tmpfs /apex nodev noexec nosuid
// chmod 0755 /apex
// chown root root /apex
// restorecon /apex
//
// except we perform the `restorecon` step just after mounting the tmpfs
// filesystem in /postinstall/apex, so that this directory is correctly
// labeled (with type `postinstall_apex_mnt_dir`) and may be manipulated in
// following operations (`chmod`, `chown`, etc.) following policies
// restricted to `postinstall_apex_mnt_dir`:
//
// mount tmpfs tmpfs /postinstall/apex nodev noexec nosuid
// restorecon /postinstall/apex
// chmod 0755 /postinstall/apex
// chown root root /postinstall/apex
//
if (mount("tmpfs", kPostinstallApexDir, "tmpfs", MS_NODEV | MS_NOEXEC | MS_NOSUID, nullptr)
!= 0) {
PLOG(ERROR) << "Failed to mount tmpfs in " << kPostinstallApexDir;
exit(209);
}
if (selinux_android_restorecon(kPostinstallApexDir, 0) < 0) {
PLOG(ERROR) << "Failed to restorecon " << kPostinstallApexDir;
exit(214);
}
if (chmod(kPostinstallApexDir, 0755) != 0) {
PLOG(ERROR) << "Failed to chmod " << kPostinstallApexDir << " to 0755";
exit(210);
}
if (chown(kPostinstallApexDir, 0, 0) != 0) {
PLOG(ERROR) << "Failed to chown " << kPostinstallApexDir << " to root:root";
exit(211);
}
// Chdir into /postinstall.
if (chdir("/postinstall") != 0) {
PLOG(ERROR) << "Unable to chdir into /postinstall.";
exit(203);
}
// Make /postinstall the root in our mount namespace.
if (chroot(".") != 0) {
PLOG(ERROR) << "Failed to chroot";
exit(204);
}
if (chdir("/") != 0) {
PLOG(ERROR) << "Unable to chdir into /.";
exit(205);
}
// Call apexd --unmount-all to free up loop and dm block devices, so that we can re-use
// them during the next invocation. Since otapreopt_chroot calls exit in case something goes
// wrong we need to register our own atexit handler.
// We want to register this handler before actually activating apex packages. This is mostly
// due to the fact that if fail to unmount apexes, then on the next run of otapreopt_chroot
// we will ask for new loop devices instead of re-using existing ones, and we really don't want
// to do that. :)
if (atexit(DeactivateApexPackages) != 0) {
LOG(ERROR) << "Failed to register atexit hander";
exit(206);
}
// Try to mount APEX packages in "/apex" in the chroot dir. We need at least
// the ART APEX, as it is required by otapreopt to run dex2oat.
ActivateApexPackages();
auto cleanup = android::base::make_scope_guard([](){
std::vector<std::string> apexd_cmd{"/system/bin/apexd", "--unmount-all"};
std::string apexd_error_msg;
bool exec_result = Exec(apexd_cmd, &apexd_error_msg);
if (!exec_result) {
PLOG(ERROR) << "Running /system/bin/apexd --unmount-all failed: " << apexd_error_msg;
}
});
// Check that an ART APEX has been activated; clean up and exit
// early otherwise.
static constexpr const std::string_view kRequiredApexs[] = {
"com.android.art",
"com.android.runtime",
"com.android.sdkext", // For derive_classpath
};
std::array<bool, arraysize(kRequiredApexs)> found_apexs{ false, false };
DIR* apex_dir = opendir("/apex");
if (apex_dir == nullptr) {
PLOG(ERROR) << "unable to open /apex";
exit(220);
}
for (dirent* entry = readdir(apex_dir); entry != nullptr; entry = readdir(apex_dir)) {
for (int i = 0; i < found_apexs.size(); i++) {
if (kRequiredApexs[i] == std::string_view(entry->d_name)) {
found_apexs[i] = true;
break;
}
}
}
closedir(apex_dir);
auto it = std::find(found_apexs.cbegin(), found_apexs.cend(), false);
if (it != found_apexs.cend()) {
LOG(ERROR) << "No activated " << kRequiredApexs[std::distance(found_apexs.cbegin(), it)]
<< " package!";
exit(221);
}
// Setup /linkerconfig. Doing it after the chroot means it doesn't need its own category
if (selinux_android_restorecon("/linkerconfig", 0) < 0) {
PLOG(ERROR) << "Failed to restorecon /linkerconfig";
exit(219);
}
std::vector<std::string> linkerconfig_cmd{"/apex/com.android.runtime/bin/linkerconfig",
"--target", "/linkerconfig"};
std::string linkerconfig_error_msg;
bool linkerconfig_exec_result = Exec(linkerconfig_cmd, &linkerconfig_error_msg);
if (!linkerconfig_exec_result) {
LOG(ERROR) << "Running linkerconfig failed: " << linkerconfig_error_msg;
exit(218);
}
android::sp<android::content::pm::IOtaDexopt> dexopt = GetDexoptService();
if (dexopt == nullptr) {
LOG(ERROR) << "Failed to find otadexopt service";
exit(222);
}
android::base::borrowed_fd status_fd(fd);
// Now go on and run otapreopt.
constexpr const int kMaximumPackages = 1000;
for (int iter = 0; iter < kMaximumPackages; iter++) {
android::String16 cmd;
android::binder::Status status = dexopt->nextDexoptCommand(&cmd);
if (!status.isOk()) {
LOG(ERROR) << "Failed to retrieve next dexopt command";
// Should we fail instead?
exit(224);
}
if (!RunDexoptCommand(argc, arg, android::String8(cmd).string())) {
exit(213);
}
float progress;
status = dexopt->getProgress(&progress);
if (!status.isOk()) {
LOG(ERROR) << "Failed to retrieve dexopt progress";
continue;
}
LOG(VERBOSE) << "Progress: " << progress;
std::string progress_str = StringPrintf("global_progress %.2f\n", progress);
if (!android::base::WriteStringToFd(progress_str, status_fd)) {
PLOG(ERROR) << "Failed to write '" << progress_str << "' to " << status_fd.get();
}
bool done;
status = dexopt->isDone(&done);
if (!status.isOk()) {
LOG(WARNING) << "Failed to check if dexopt is done";
continue;
}
if (done) {
LOG(INFO) << "dexopt is done";
break;
}
std::this_thread::sleep_for(1s);
}
return 0;
}
} // namespace installd
} // namespace android
int main(const int argc, char *argv[]) {
return android::installd::otapreopt_chroot(argc, argv);
}