| /* |
| * Copyright (C) 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. |
| */ |
| |
| #define LOG_TAG "dumpstate" |
| |
| #include "DumpstateUtil.h" |
| |
| #include <dirent.h> |
| #include <fcntl.h> |
| #include <sys/prctl.h> |
| #include <sys/wait.h> |
| #include <unistd.h> |
| |
| #include <vector> |
| |
| #include <android-base/file.h> |
| #include <android-base/properties.h> |
| #include <android-base/stringprintf.h> |
| #include <android-base/strings.h> |
| #include <android-base/unique_fd.h> |
| #include <log/log.h> |
| |
| #include "DumpstateInternal.h" |
| |
| namespace android { |
| namespace os { |
| namespace dumpstate { |
| |
| namespace { |
| |
| static constexpr const char* kSuPath = "/system/xbin/su"; |
| |
| static bool waitpid_with_timeout(pid_t pid, int timeout_ms, int* status) { |
| sigset_t child_mask, old_mask; |
| sigemptyset(&child_mask); |
| sigaddset(&child_mask, SIGCHLD); |
| |
| if (sigprocmask(SIG_BLOCK, &child_mask, &old_mask) == -1) { |
| printf("*** sigprocmask failed: %s\n", strerror(errno)); |
| return false; |
| } |
| |
| timespec ts; |
| ts.tv_sec = MSEC_TO_SEC(timeout_ms); |
| ts.tv_nsec = (timeout_ms % 1000) * 1000000; |
| int ret = TEMP_FAILURE_RETRY(sigtimedwait(&child_mask, NULL, &ts)); |
| int saved_errno = errno; |
| |
| // Set the signals back the way they were. |
| if (sigprocmask(SIG_SETMASK, &old_mask, NULL) == -1) { |
| printf("*** sigprocmask failed: %s\n", strerror(errno)); |
| if (ret == 0) { |
| return false; |
| } |
| } |
| if (ret == -1) { |
| errno = saved_errno; |
| if (errno == EAGAIN) { |
| errno = ETIMEDOUT; |
| } else { |
| printf("*** sigtimedwait failed: %s\n", strerror(errno)); |
| } |
| return false; |
| } |
| |
| pid_t child_pid = waitpid(pid, status, WNOHANG); |
| if (child_pid != pid) { |
| if (child_pid != -1) { |
| printf("*** Waiting for pid %d, got pid %d instead\n", pid, child_pid); |
| } else { |
| printf("*** waitpid failed: %s\n", strerror(errno)); |
| } |
| return false; |
| } |
| return true; |
| } |
| } // unnamed namespace |
| |
| CommandOptions CommandOptions::DEFAULT = CommandOptions::WithTimeout(10).Build(); |
| CommandOptions CommandOptions::AS_ROOT = CommandOptions::WithTimeout(10).AsRoot().Build(); |
| |
| CommandOptions::CommandOptionsBuilder::CommandOptionsBuilder(int64_t timeout_ms) : values(timeout_ms) { |
| } |
| |
| CommandOptions::CommandOptionsBuilder& CommandOptions::CommandOptionsBuilder::Always() { |
| values.always_ = true; |
| return *this; |
| } |
| |
| CommandOptions::CommandOptionsBuilder& CommandOptions::CommandOptionsBuilder::AsRoot() { |
| values.account_mode_ = SU_ROOT; |
| return *this; |
| } |
| |
| CommandOptions::CommandOptionsBuilder& CommandOptions::CommandOptionsBuilder::AsRootIfAvailable() { |
| if (!PropertiesHelper::IsUserBuild()) |
| values.account_mode_ = SU_ROOT; |
| return *this; |
| } |
| |
| CommandOptions::CommandOptionsBuilder& CommandOptions::CommandOptionsBuilder::DropRoot() { |
| values.account_mode_ = DROP_ROOT; |
| return *this; |
| } |
| |
| CommandOptions::CommandOptionsBuilder& CommandOptions::CommandOptionsBuilder::RedirectStderr() { |
| values.output_mode_ = REDIRECT_TO_STDERR; |
| return *this; |
| } |
| |
| CommandOptions::CommandOptionsBuilder& CommandOptions::CommandOptionsBuilder::Log( |
| const std::string& message) { |
| values.logging_message_ = message; |
| return *this; |
| } |
| |
| CommandOptions CommandOptions::CommandOptionsBuilder::Build() { |
| return CommandOptions(values); |
| } |
| |
| CommandOptions::CommandOptionsValues::CommandOptionsValues(int64_t timeout_ms) |
| : timeout_ms_(timeout_ms), |
| always_(false), |
| account_mode_(DONT_DROP_ROOT), |
| output_mode_(NORMAL_OUTPUT), |
| logging_message_("") { |
| } |
| |
| CommandOptions::CommandOptions(const CommandOptionsValues& values) : values(values) { |
| } |
| |
| int64_t CommandOptions::Timeout() const { |
| return MSEC_TO_SEC(values.timeout_ms_); |
| } |
| |
| int64_t CommandOptions::TimeoutInMs() const { |
| return values.timeout_ms_; |
| } |
| |
| bool CommandOptions::Always() const { |
| return values.always_; |
| } |
| |
| PrivilegeMode CommandOptions::PrivilegeMode() const { |
| return values.account_mode_; |
| } |
| |
| OutputMode CommandOptions::OutputMode() const { |
| return values.output_mode_; |
| } |
| |
| std::string CommandOptions::LoggingMessage() const { |
| return values.logging_message_; |
| } |
| |
| CommandOptions::CommandOptionsBuilder CommandOptions::WithTimeout(int64_t timeout_sec) { |
| return CommandOptions::CommandOptionsBuilder(SEC_TO_MSEC(timeout_sec)); |
| } |
| |
| CommandOptions::CommandOptionsBuilder CommandOptions::WithTimeoutInMs(int64_t timeout_ms) { |
| return CommandOptions::CommandOptionsBuilder(timeout_ms); |
| } |
| |
| std::string PropertiesHelper::build_type_ = ""; |
| int PropertiesHelper::dry_run_ = -1; |
| |
| bool PropertiesHelper::IsUserBuild() { |
| if (build_type_.empty()) { |
| build_type_ = android::base::GetProperty("ro.build.type", "user"); |
| } |
| return "user" == build_type_; |
| } |
| |
| bool PropertiesHelper::IsDryRun() { |
| if (dry_run_ == -1) { |
| dry_run_ = android::base::GetBoolProperty("dumpstate.dry_run", false) ? 1 : 0; |
| } |
| return dry_run_ == 1; |
| } |
| |
| int DumpFileToFd(int out_fd, const std::string& title, const std::string& path) { |
| android::base::unique_fd fd(TEMP_FAILURE_RETRY(open(path.c_str(), O_RDONLY | O_NONBLOCK | O_CLOEXEC))); |
| if (fd.get() < 0) { |
| int err = errno; |
| if (title.empty()) { |
| dprintf(out_fd, "*** Error dumping %s: %s\n", path.c_str(), strerror(err)); |
| } else { |
| dprintf(out_fd, "*** Error dumping %s (%s): %s\n", path.c_str(), title.c_str(), |
| strerror(err)); |
| } |
| fsync(out_fd); |
| return -1; |
| } |
| return DumpFileFromFdToFd(title, path, fd.get(), out_fd, PropertiesHelper::IsDryRun()); |
| } |
| |
| int RunCommandToFd(int fd, const std::string& title, const std::vector<std::string>& full_command, |
| const CommandOptions& options) { |
| if (full_command.empty()) { |
| MYLOGE("No arguments on RunCommandToFd(%s)\n", title.c_str()); |
| return -1; |
| } |
| |
| int size = full_command.size() + 1; // null terminated |
| int starting_index = 0; |
| if (options.PrivilegeMode() == SU_ROOT) { |
| starting_index = 2; // "su" "root" |
| size += starting_index; |
| } |
| |
| std::vector<const char*> args; |
| args.resize(size); |
| |
| std::string command_string; |
| if (options.PrivilegeMode() == SU_ROOT) { |
| args[0] = kSuPath; |
| command_string += kSuPath; |
| args[1] = "root"; |
| command_string += " root "; |
| } |
| for (size_t i = 0; i < full_command.size(); i++) { |
| args[i + starting_index] = full_command[i].data(); |
| command_string += args[i + starting_index]; |
| if (i != full_command.size() - 1) { |
| command_string += " "; |
| } |
| } |
| args[size - 1] = nullptr; |
| |
| const char* command = command_string.c_str(); |
| |
| if (options.PrivilegeMode() == SU_ROOT && PropertiesHelper::IsUserBuild()) { |
| dprintf(fd, "Skipping '%s' on user build.\n", command); |
| return 0; |
| } |
| |
| if (!title.empty()) { |
| dprintf(fd, "------ %s (%s) ------\n", title.c_str(), command); |
| fsync(fd); |
| } |
| |
| const std::string& logging_message = options.LoggingMessage(); |
| if (!logging_message.empty()) { |
| MYLOGI(logging_message.c_str(), command_string.c_str()); |
| } |
| |
| bool silent = (options.OutputMode() == REDIRECT_TO_STDERR); |
| bool redirecting_to_fd = STDOUT_FILENO != fd; |
| |
| if (PropertiesHelper::IsDryRun() && !options.Always()) { |
| if (!title.empty()) { |
| dprintf(fd, "\t(skipped on dry run)\n"); |
| } else if (redirecting_to_fd) { |
| // There is no title, but we should still print a dry-run message |
| dprintf(fd, "%s: skipped on dry run\n", command_string.c_str()); |
| } |
| fsync(fd); |
| return 0; |
| } |
| |
| const char* path = args[0]; |
| |
| uint64_t start = Nanotime(); |
| pid_t pid = fork(); |
| |
| /* handle error case */ |
| if (pid < 0) { |
| if (!silent) dprintf(fd, "*** fork: %s\n", strerror(errno)); |
| MYLOGE("*** fork: %s\n", strerror(errno)); |
| return pid; |
| } |
| |
| /* handle child case */ |
| if (pid == 0) { |
| if (options.PrivilegeMode() == DROP_ROOT && !DropRootUser()) { |
| if (!silent) { |
| dprintf(fd, "*** failed to drop root before running %s: %s\n", command, |
| strerror(errno)); |
| } |
| MYLOGE("*** could not drop root before running %s: %s\n", command, strerror(errno)); |
| return -1; |
| } |
| |
| if (silent) { |
| // Redirects stdout to stderr |
| TEMP_FAILURE_RETRY(dup2(STDERR_FILENO, STDOUT_FILENO)); |
| } else if (redirecting_to_fd) { |
| // Redirect stdout to fd |
| TEMP_FAILURE_RETRY(dup2(fd, STDOUT_FILENO)); |
| close(fd); |
| } |
| |
| /* make sure the child dies when dumpstate dies */ |
| prctl(PR_SET_PDEATHSIG, SIGKILL); |
| |
| /* just ignore SIGPIPE, will go down with parent's */ |
| struct sigaction sigact; |
| memset(&sigact, 0, sizeof(sigact)); |
| sigact.sa_handler = SIG_IGN; |
| sigaction(SIGPIPE, &sigact, NULL); |
| |
| execvp(path, (char**)args.data()); |
| // execvp's result will be handled after waitpid_with_timeout() below, but |
| // if it failed, it's safer to exit dumpstate. |
| MYLOGD("execvp on command '%s' failed (error: %s)\n", command, strerror(errno)); |
| // Must call _exit (instead of exit), otherwise it will corrupt the zip |
| // file. |
| _exit(EXIT_FAILURE); |
| } |
| |
| /* handle parent case */ |
| int status; |
| bool ret = waitpid_with_timeout(pid, options.TimeoutInMs(), &status); |
| fsync(fd); |
| |
| uint64_t elapsed = Nanotime() - start; |
| if (!ret) { |
| if (errno == ETIMEDOUT) { |
| if (!silent) |
| dprintf(fd, "*** command '%s' timed out after %.3fs (killing pid %d)\n", command, |
| static_cast<float>(elapsed) / NANOS_PER_SEC, pid); |
| MYLOGE("*** command '%s' timed out after %.3fs (killing pid %d)\n", command, |
| static_cast<float>(elapsed) / NANOS_PER_SEC, pid); |
| } else { |
| if (!silent) |
| dprintf(fd, "*** command '%s': Error after %.4fs (killing pid %d)\n", command, |
| static_cast<float>(elapsed) / NANOS_PER_SEC, pid); |
| MYLOGE("command '%s': Error after %.4fs (killing pid %d)\n", command, |
| static_cast<float>(elapsed) / NANOS_PER_SEC, pid); |
| } |
| kill(pid, SIGTERM); |
| if (!waitpid_with_timeout(pid, 5000, nullptr)) { |
| kill(pid, SIGKILL); |
| if (!waitpid_with_timeout(pid, 5000, nullptr)) { |
| if (!silent) |
| dprintf(fd, "could not kill command '%s' (pid %d) even with SIGKILL.\n", |
| command, pid); |
| MYLOGE("could not kill command '%s' (pid %d) even with SIGKILL.\n", command, pid); |
| } |
| } |
| return -1; |
| } |
| |
| if (WIFSIGNALED(status)) { |
| if (!silent) |
| dprintf(fd, "*** command '%s' failed: killed by signal %d\n", command, WTERMSIG(status)); |
| MYLOGE("*** command '%s' failed: killed by signal %d\n", command, WTERMSIG(status)); |
| } else if (WIFEXITED(status) && WEXITSTATUS(status) > 0) { |
| status = WEXITSTATUS(status); |
| if (!silent) dprintf(fd, "*** command '%s' failed: exit code %d\n", command, status); |
| MYLOGE("*** command '%s' failed: exit code %d\n", command, status); |
| } |
| |
| return status; |
| } |
| |
| int GetPidByName(const std::string& ps_name) { |
| DIR* proc_dir; |
| struct dirent* ps; |
| unsigned int pid; |
| std::string cmdline; |
| |
| if (!(proc_dir = opendir("/proc"))) { |
| MYLOGE("Can't open /proc\n"); |
| return -1; |
| } |
| |
| while ((ps = readdir(proc_dir))) { |
| if (!(pid = atoi(ps->d_name))) { |
| continue; |
| } |
| android::base::ReadFileToString("/proc/" + std::string(ps->d_name) + "/cmdline", &cmdline); |
| if (cmdline.find(ps_name) == std::string::npos) { |
| continue; |
| } else { |
| closedir(proc_dir); |
| return pid; |
| } |
| } |
| MYLOGE("can't find the pid\n"); |
| closedir(proc_dir); |
| return -1; |
| } |
| |
| } // namespace dumpstate |
| } // namespace os |
| } // namespace android |