blob: c62734a925e8e53785566378515c86557dd4be5c [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 <fstream>
#include <sstream>
#include <android-base/file.h>
#include <android-base/logging.h>
#include <android-base/macros.h>
#include <android-base/scopeguard.h>
#include <android-base/stringprintf.h>
#include <android-base/unique_fd.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 {
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 CloseDescriptor(const char* descriptor_string) {
int fd = -1;
std::istringstream stream(descriptor_string);
stream >> fd;
if (!stream.fail()) {
CloseDescriptor(fd);
}
}
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);
}
// 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);
// 2) The status channel.
CloseDescriptor(arg[1]);
// 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);
}
// Now go on and run otapreopt.
// 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]);
}
// 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;
}
if (!exec_result) {
exit(213);
}
return 0;
}
} // namespace installd
} // namespace android
int main(const int argc, char *argv[]) {
return android::installd::otapreopt_chroot(argc, argv);
}