Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2018 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 | |
Tao Bao | 42c45e2 | 2018-07-31 09:37:12 -0700 | [diff] [blame] | 17 | #include <dlfcn.h> |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 18 | #include <errno.h> |
| 19 | #include <fcntl.h> |
| 20 | #include <getopt.h> |
| 21 | #include <inttypes.h> |
| 22 | #include <limits.h> |
| 23 | #include <linux/fs.h> |
| 24 | #include <stdarg.h> |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 25 | #include <stdio.h> |
| 26 | #include <stdlib.h> |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 27 | #include <string.h> |
| 28 | #include <sys/stat.h> |
| 29 | #include <sys/types.h> |
| 30 | #include <time.h> |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 31 | #include <unistd.h> |
| 32 | |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 33 | #include <atomic> |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 34 | #include <string> |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 35 | #include <thread> |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 36 | #include <vector> |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 37 | |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 38 | #include <android-base/file.h> |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 39 | #include <android-base/logging.h> |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 40 | #include <android-base/properties.h> |
| 41 | #include <android-base/strings.h> |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 42 | #include <android-base/unique_fd.h> |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 43 | #include <bootloader_message/bootloader_message.h> |
| 44 | #include <cutils/android_reboot.h> |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 45 | #include <cutils/sockets.h> |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 46 | #include <private/android_logger.h> /* private pmsg functions */ |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 47 | #include <selinux/android.h> |
| 48 | #include <selinux/label.h> |
| 49 | #include <selinux/selinux.h> |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 50 | |
| 51 | #include "common.h" |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 52 | #include "fastboot/fastboot.h" |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 53 | #include "install/wipe_data.h" |
| 54 | #include "otautil/logging.h" |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 55 | #include "otautil/paths.h" |
xunchang | 2478885 | 2019-03-22 16:08:52 -0700 | [diff] [blame] | 56 | #include "otautil/roots.h" |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 57 | #include "otautil/sysutil.h" |
| 58 | #include "recovery.h" |
Tianjie Xu | 8f39730 | 2018-08-20 13:40:47 -0700 | [diff] [blame] | 59 | #include "recovery_ui/device.h" |
| 60 | #include "recovery_ui/stub_ui.h" |
| 61 | #include "recovery_ui/ui.h" |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 62 | |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 63 | static constexpr const char* COMMAND_FILE = "/cache/recovery/command"; |
| 64 | static constexpr const char* LOCALE_FILE = "/cache/recovery/last_locale"; |
| 65 | |
| 66 | static constexpr const char* CACHE_ROOT = "/cache"; |
| 67 | |
| 68 | bool has_cache = false; |
| 69 | |
| 70 | RecoveryUI* ui = nullptr; |
| 71 | struct selabel_handle* sehandle; |
| 72 | |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 73 | static void UiLogger(android::base::LogId /* id */, android::base::LogSeverity severity, |
| 74 | const char* /* tag */, const char* /* file */, unsigned int /* line */, |
| 75 | const char* message) { |
| 76 | static constexpr char log_characters[] = "VDIWEF"; |
| 77 | if (severity >= android::base::ERROR && ui != nullptr) { |
| 78 | ui->Print("E:%s\n", message); |
| 79 | } else { |
| 80 | fprintf(stdout, "%c:%s\n", log_characters[severity], message); |
| 81 | } |
| 82 | } |
| 83 | |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 84 | // command line args come from, in decreasing precedence: |
| 85 | // - the actual command line |
| 86 | // - the bootloader control block (one per line, after "recovery") |
| 87 | // - the contents of COMMAND_FILE (one per line) |
| 88 | static std::vector<std::string> get_args(const int argc, char** const argv) { |
| 89 | CHECK_GT(argc, 0); |
| 90 | |
| 91 | bootloader_message boot = {}; |
| 92 | std::string err; |
| 93 | if (!read_bootloader_message(&boot, &err)) { |
| 94 | LOG(ERROR) << err; |
| 95 | // If fails, leave a zeroed bootloader_message. |
| 96 | boot = {}; |
| 97 | } |
| 98 | stage = std::string(boot.stage); |
| 99 | |
David Anderson | eee4e26 | 2018-08-21 13:10:45 -0700 | [diff] [blame] | 100 | std::string boot_command; |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 101 | if (boot.command[0] != 0) { |
David Anderson | eee4e26 | 2018-08-21 13:10:45 -0700 | [diff] [blame] | 102 | if (memchr(boot.command, '\0', sizeof(boot.command))) { |
| 103 | boot_command = std::string(boot.command); |
| 104 | } else { |
| 105 | boot_command = std::string(boot.command, sizeof(boot.command)); |
| 106 | } |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 107 | LOG(INFO) << "Boot command: " << boot_command; |
| 108 | } |
| 109 | |
| 110 | if (boot.status[0] != 0) { |
| 111 | std::string boot_status = std::string(boot.status, sizeof(boot.status)); |
| 112 | LOG(INFO) << "Boot status: " << boot_status; |
| 113 | } |
| 114 | |
| 115 | std::vector<std::string> args(argv, argv + argc); |
| 116 | |
| 117 | // --- if arguments weren't supplied, look in the bootloader control block |
| 118 | if (args.size() == 1) { |
| 119 | boot.recovery[sizeof(boot.recovery) - 1] = '\0'; // Ensure termination |
| 120 | std::string boot_recovery(boot.recovery); |
| 121 | std::vector<std::string> tokens = android::base::Split(boot_recovery, "\n"); |
| 122 | if (!tokens.empty() && tokens[0] == "recovery") { |
| 123 | for (auto it = tokens.begin() + 1; it != tokens.end(); it++) { |
| 124 | // Skip empty and '\0'-filled tokens. |
| 125 | if (!it->empty() && (*it)[0] != '\0') args.push_back(std::move(*it)); |
| 126 | } |
| 127 | LOG(INFO) << "Got " << args.size() << " arguments from boot message"; |
| 128 | } else if (boot.recovery[0] != 0) { |
| 129 | LOG(ERROR) << "Bad boot message: \"" << boot_recovery << "\""; |
| 130 | } |
| 131 | } |
| 132 | |
| 133 | // --- if that doesn't work, try the command file (if we have /cache). |
| 134 | if (args.size() == 1 && has_cache) { |
| 135 | std::string content; |
| 136 | if (ensure_path_mounted(COMMAND_FILE) == 0 && |
| 137 | android::base::ReadFileToString(COMMAND_FILE, &content)) { |
| 138 | std::vector<std::string> tokens = android::base::Split(content, "\n"); |
| 139 | // All the arguments in COMMAND_FILE are needed (unlike the BCB message, |
| 140 | // COMMAND_FILE doesn't use filename as the first argument). |
| 141 | for (auto it = tokens.begin(); it != tokens.end(); it++) { |
| 142 | // Skip empty and '\0'-filled tokens. |
| 143 | if (!it->empty() && (*it)[0] != '\0') args.push_back(std::move(*it)); |
| 144 | } |
| 145 | LOG(INFO) << "Got " << args.size() << " arguments from " << COMMAND_FILE; |
| 146 | } |
| 147 | } |
| 148 | |
| 149 | // Write the arguments (excluding the filename in args[0]) back into the |
| 150 | // bootloader control block. So the device will always boot into recovery to |
| 151 | // finish the pending work, until finish_recovery() is called. |
| 152 | std::vector<std::string> options(args.cbegin() + 1, args.cend()); |
| 153 | if (!update_bootloader_message(options, &err)) { |
| 154 | LOG(ERROR) << "Failed to set BCB message: " << err; |
| 155 | } |
| 156 | |
David Anderson | eee4e26 | 2018-08-21 13:10:45 -0700 | [diff] [blame] | 157 | // Finally, if no arguments were specified, check whether we should boot |
| 158 | // into fastboot. |
| 159 | if (args.size() == 1 && boot_command == "boot-fastboot") { |
| 160 | args.emplace_back("--fastboot"); |
| 161 | } |
| 162 | |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 163 | return args; |
| 164 | } |
| 165 | |
| 166 | static std::string load_locale_from_cache() { |
| 167 | if (ensure_path_mounted(LOCALE_FILE) != 0) { |
| 168 | LOG(ERROR) << "Can't mount " << LOCALE_FILE; |
| 169 | return ""; |
| 170 | } |
| 171 | |
| 172 | std::string content; |
| 173 | if (!android::base::ReadFileToString(LOCALE_FILE, &content)) { |
| 174 | PLOG(ERROR) << "Can't read " << LOCALE_FILE; |
| 175 | return ""; |
| 176 | } |
| 177 | |
| 178 | return android::base::Trim(content); |
| 179 | } |
| 180 | |
Tao Bao | e0cfab3 | 2019-03-29 15:53:23 -0700 | [diff] [blame] | 181 | // Sets the usb config to 'state'. |
| 182 | static bool SetUsbConfig(const std::string& state) { |
| 183 | android::base::SetProperty("sys.usb.config", state); |
| 184 | return android::base::WaitForProperty("sys.usb.state", state); |
| 185 | } |
| 186 | |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 187 | static void ListenRecoverySocket(RecoveryUI* ui, std::atomic<Device::BuiltinAction>& action) { |
| 188 | android::base::unique_fd sock_fd(android_get_control_socket("recovery")); |
| 189 | if (sock_fd < 0) { |
| 190 | PLOG(ERROR) << "Failed to open recovery socket"; |
| 191 | return; |
| 192 | } |
| 193 | listen(sock_fd, 4); |
| 194 | |
| 195 | while (true) { |
| 196 | android::base::unique_fd connection_fd; |
| 197 | connection_fd.reset(accept(sock_fd, nullptr, nullptr)); |
| 198 | if (connection_fd < 0) { |
| 199 | PLOG(ERROR) << "Failed to accept socket connection"; |
| 200 | continue; |
| 201 | } |
| 202 | char msg; |
| 203 | constexpr char kSwitchToFastboot = 'f'; |
| 204 | constexpr char kSwitchToRecovery = 'r'; |
| 205 | ssize_t ret = TEMP_FAILURE_RETRY(read(connection_fd, &msg, sizeof(msg))); |
| 206 | if (ret != sizeof(msg)) { |
| 207 | PLOG(ERROR) << "Couldn't read from socket"; |
| 208 | continue; |
| 209 | } |
| 210 | switch (msg) { |
| 211 | case kSwitchToRecovery: |
| 212 | action = Device::BuiltinAction::ENTER_RECOVERY; |
| 213 | break; |
| 214 | case kSwitchToFastboot: |
| 215 | action = Device::BuiltinAction::ENTER_FASTBOOT; |
| 216 | break; |
| 217 | default: |
| 218 | LOG(ERROR) << "Unrecognized char from socket " << msg; |
| 219 | continue; |
| 220 | } |
| 221 | ui->InterruptKey(); |
| 222 | } |
| 223 | } |
| 224 | |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 225 | static void redirect_stdio(const char* filename) { |
Tao Bao | 6fcd208 | 2019-01-16 09:29:17 -0800 | [diff] [blame] | 226 | android::base::unique_fd pipe_read, pipe_write; |
| 227 | // Create a pipe that allows parent process sending logs over. |
| 228 | if (!android::base::Pipe(&pipe_read, &pipe_write)) { |
| 229 | PLOG(ERROR) << "Failed to create pipe for redirecting stdio"; |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 230 | |
| 231 | // Fall back to traditional logging mode without timestamps. If these fail, there's not really |
| 232 | // anywhere to complain... |
| 233 | freopen(filename, "a", stdout); |
| 234 | setbuf(stdout, nullptr); |
| 235 | freopen(filename, "a", stderr); |
| 236 | setbuf(stderr, nullptr); |
| 237 | |
| 238 | return; |
| 239 | } |
| 240 | |
| 241 | pid_t pid = fork(); |
| 242 | if (pid == -1) { |
Tao Bao | 6fcd208 | 2019-01-16 09:29:17 -0800 | [diff] [blame] | 243 | PLOG(ERROR) << "Failed to fork for redirecting stdio"; |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 244 | |
| 245 | // Fall back to traditional logging mode without timestamps. If these fail, there's not really |
| 246 | // anywhere to complain... |
| 247 | freopen(filename, "a", stdout); |
| 248 | setbuf(stdout, nullptr); |
| 249 | freopen(filename, "a", stderr); |
| 250 | setbuf(stderr, nullptr); |
| 251 | |
| 252 | return; |
| 253 | } |
| 254 | |
| 255 | if (pid == 0) { |
Tao Bao | 6fcd208 | 2019-01-16 09:29:17 -0800 | [diff] [blame] | 256 | // Child process reads the incoming logs and doesn't write to the pipe. |
| 257 | pipe_write.reset(); |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 258 | |
| 259 | auto start = std::chrono::steady_clock::now(); |
| 260 | |
| 261 | // Child logger to actually write to the log file. |
| 262 | FILE* log_fp = fopen(filename, "ae"); |
| 263 | if (log_fp == nullptr) { |
| 264 | PLOG(ERROR) << "fopen \"" << filename << "\" failed"; |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 265 | _exit(EXIT_FAILURE); |
| 266 | } |
| 267 | |
Tao Bao | 6fcd208 | 2019-01-16 09:29:17 -0800 | [diff] [blame] | 268 | FILE* pipe_fp = android::base::Fdopen(std::move(pipe_read), "r"); |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 269 | if (pipe_fp == nullptr) { |
| 270 | PLOG(ERROR) << "fdopen failed"; |
| 271 | check_and_fclose(log_fp, filename); |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 272 | _exit(EXIT_FAILURE); |
| 273 | } |
| 274 | |
| 275 | char* line = nullptr; |
| 276 | size_t len = 0; |
| 277 | while (getline(&line, &len, pipe_fp) != -1) { |
| 278 | auto now = std::chrono::steady_clock::now(); |
| 279 | double duration = |
| 280 | std::chrono::duration_cast<std::chrono::duration<double>>(now - start).count(); |
| 281 | if (line[0] == '\n') { |
| 282 | fprintf(log_fp, "[%12.6lf]\n", duration); |
| 283 | } else { |
| 284 | fprintf(log_fp, "[%12.6lf] %s", duration, line); |
| 285 | } |
| 286 | fflush(log_fp); |
| 287 | } |
| 288 | |
| 289 | PLOG(ERROR) << "getline failed"; |
| 290 | |
Tao Bao | 6fcd208 | 2019-01-16 09:29:17 -0800 | [diff] [blame] | 291 | fclose(pipe_fp); |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 292 | free(line); |
| 293 | check_and_fclose(log_fp, filename); |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 294 | _exit(EXIT_FAILURE); |
| 295 | } else { |
| 296 | // Redirect stdout/stderr to the logger process. Close the unused read end. |
Tao Bao | 6fcd208 | 2019-01-16 09:29:17 -0800 | [diff] [blame] | 297 | pipe_read.reset(); |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 298 | |
| 299 | setbuf(stdout, nullptr); |
| 300 | setbuf(stderr, nullptr); |
| 301 | |
Tao Bao | 6fcd208 | 2019-01-16 09:29:17 -0800 | [diff] [blame] | 302 | if (dup2(pipe_write.get(), STDOUT_FILENO) == -1) { |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 303 | PLOG(ERROR) << "dup2 stdout failed"; |
| 304 | } |
Tao Bao | 6fcd208 | 2019-01-16 09:29:17 -0800 | [diff] [blame] | 305 | if (dup2(pipe_write.get(), STDERR_FILENO) == -1) { |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 306 | PLOG(ERROR) << "dup2 stderr failed"; |
| 307 | } |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 308 | } |
| 309 | } |
| 310 | |
| 311 | int main(int argc, char** argv) { |
| 312 | // We don't have logcat yet under recovery; so we'll print error on screen and log to stdout |
| 313 | // (which is redirected to recovery.log) as we used to do. |
| 314 | android::base::InitLogging(argv, &UiLogger); |
| 315 | |
| 316 | // Take last pmsg contents and rewrite it to the current pmsg session. |
| 317 | static constexpr const char filter[] = "recovery/"; |
| 318 | // Do we need to rotate? |
| 319 | bool do_rotate = false; |
| 320 | |
| 321 | __android_log_pmsg_file_read(LOG_ID_SYSTEM, ANDROID_LOG_INFO, filter, logbasename, &do_rotate); |
| 322 | // Take action to refresh pmsg contents |
| 323 | __android_log_pmsg_file_read(LOG_ID_SYSTEM, ANDROID_LOG_INFO, filter, logrotate, &do_rotate); |
| 324 | |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 325 | time_t start = time(nullptr); |
| 326 | |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 327 | // redirect_stdio should be called only in non-sideload mode. Otherwise we may have two logger |
| 328 | // instances with different timestamps. |
| 329 | redirect_stdio(Paths::Get().temporary_log_file().c_str()); |
| 330 | |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 331 | load_volume_table(); |
| 332 | has_cache = volume_for_mount_point(CACHE_ROOT) != nullptr; |
| 333 | |
| 334 | std::vector<std::string> args = get_args(argc, argv); |
Tao Bao | 1700cc4 | 2018-07-16 22:09:59 -0700 | [diff] [blame] | 335 | auto args_to_parse = StringVectorToNullTerminatedArray(args); |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 336 | |
| 337 | static constexpr struct option OPTIONS[] = { |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 338 | { "fastboot", no_argument, nullptr, 0 }, |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 339 | { "locale", required_argument, nullptr, 0 }, |
| 340 | { "show_text", no_argument, nullptr, 't' }, |
| 341 | { nullptr, 0, nullptr, 0 }, |
| 342 | }; |
| 343 | |
| 344 | bool show_text = false; |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 345 | bool fastboot = false; |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 346 | std::string locale; |
| 347 | |
| 348 | int arg; |
| 349 | int option_index; |
Tao Bao | 1700cc4 | 2018-07-16 22:09:59 -0700 | [diff] [blame] | 350 | while ((arg = getopt_long(args_to_parse.size() - 1, args_to_parse.data(), "", OPTIONS, |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 351 | &option_index)) != -1) { |
| 352 | switch (arg) { |
| 353 | case 't': |
| 354 | show_text = true; |
| 355 | break; |
| 356 | case 0: { |
| 357 | std::string option = OPTIONS[option_index].name; |
| 358 | if (option == "locale") { |
| 359 | locale = optarg; |
Hridya Valsaraju | 7f41a2c | 2018-09-19 16:29:01 -0700 | [diff] [blame] | 360 | } else if (option == "fastboot" && |
Yifan Hong | d17174c | 2018-11-16 12:49:33 -0800 | [diff] [blame] | 361 | android::base::GetBoolProperty("ro.boot.dynamic_partitions", false)) { |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 362 | fastboot = true; |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 363 | } |
| 364 | break; |
| 365 | } |
| 366 | } |
| 367 | } |
Jerry Zhang | 49fd5d2 | 2018-05-17 12:54:41 -0700 | [diff] [blame] | 368 | optind = 1; |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 369 | |
| 370 | if (locale.empty()) { |
| 371 | if (has_cache) { |
| 372 | locale = load_locale_from_cache(); |
| 373 | } |
| 374 | |
| 375 | if (locale.empty()) { |
| 376 | static constexpr const char* DEFAULT_LOCALE = "en-US"; |
| 377 | locale = DEFAULT_LOCALE; |
| 378 | } |
| 379 | } |
| 380 | |
Tao Bao | 42c45e2 | 2018-07-31 09:37:12 -0700 | [diff] [blame] | 381 | static constexpr const char* kDefaultLibRecoveryUIExt = "librecovery_ui_ext.so"; |
| 382 | // Intentionally not calling dlclose(3) to avoid potential gotchas (e.g. `make_device` may have |
| 383 | // handed out pointers to code or static [or thread-local] data and doesn't collect them all back |
| 384 | // in on dlclose). |
| 385 | void* librecovery_ui_ext = dlopen(kDefaultLibRecoveryUIExt, RTLD_NOW); |
| 386 | |
| 387 | using MakeDeviceType = decltype(&make_device); |
| 388 | MakeDeviceType make_device_func = nullptr; |
| 389 | if (librecovery_ui_ext == nullptr) { |
| 390 | printf("Failed to dlopen %s: %s\n", kDefaultLibRecoveryUIExt, dlerror()); |
| 391 | } else { |
| 392 | reinterpret_cast<void*&>(make_device_func) = dlsym(librecovery_ui_ext, "make_device"); |
| 393 | if (make_device_func == nullptr) { |
| 394 | printf("Failed to dlsym make_device: %s\n", dlerror()); |
| 395 | } |
| 396 | } |
| 397 | |
| 398 | Device* device; |
| 399 | if (make_device_func == nullptr) { |
| 400 | printf("Falling back to the default make_device() instead\n"); |
| 401 | device = make_device(); |
| 402 | } else { |
| 403 | printf("Loading make_device from %s\n", kDefaultLibRecoveryUIExt); |
| 404 | device = (*make_device_func)(); |
| 405 | } |
| 406 | |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 407 | if (android::base::GetBoolProperty("ro.boot.quiescent", false)) { |
| 408 | printf("Quiescent recovery mode.\n"); |
| 409 | device->ResetUI(new StubRecoveryUI()); |
| 410 | } else { |
| 411 | if (!device->GetUI()->Init(locale)) { |
| 412 | printf("Failed to initialize UI; using stub UI instead.\n"); |
| 413 | device->ResetUI(new StubRecoveryUI()); |
| 414 | } |
| 415 | } |
| 416 | ui = device->GetUI(); |
| 417 | |
| 418 | if (!has_cache) { |
| 419 | device->RemoveMenuItemForAction(Device::WIPE_CACHE); |
| 420 | } |
| 421 | |
Yifan Hong | d17174c | 2018-11-16 12:49:33 -0800 | [diff] [blame] | 422 | if (!android::base::GetBoolProperty("ro.boot.dynamic_partitions", false)) { |
Hridya Valsaraju | daa301e | 2018-09-18 14:48:01 -0700 | [diff] [blame] | 423 | device->RemoveMenuItemForAction(Device::ENTER_FASTBOOT); |
| 424 | } |
| 425 | |
Tao Bao | 378bfbf | 2019-04-16 14:22:25 -0700 | [diff] [blame] | 426 | if (!is_ro_debuggable()) { |
| 427 | device->RemoveMenuItemForAction(Device::ENTER_RESCUE); |
| 428 | } |
| 429 | |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 430 | ui->SetBackground(RecoveryUI::NONE); |
| 431 | if (show_text) ui->ShowText(true); |
| 432 | |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 433 | LOG(INFO) << "Starting recovery (pid " << getpid() << ") on " << ctime(&start); |
| 434 | LOG(INFO) << "locale is [" << locale << "]"; |
| 435 | |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 436 | sehandle = selinux_android_file_context_handle(); |
| 437 | selinux_android_set_sehandle(sehandle); |
| 438 | if (!sehandle) { |
| 439 | ui->Print("Warning: No file_contexts\n"); |
| 440 | } |
| 441 | |
xunchang | cd780b4 | 2019-04-15 15:24:24 -0700 | [diff] [blame] | 442 | SetLoggingSehandle(sehandle); |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 443 | |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 444 | std::atomic<Device::BuiltinAction> action; |
| 445 | std::thread listener_thread(ListenRecoverySocket, ui, std::ref(action)); |
| 446 | listener_thread.detach(); |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 447 | |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 448 | while (true) { |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 449 | std::string usb_config = fastboot ? "fastboot" : is_ro_debuggable() ? "adb" : "none"; |
| 450 | std::string usb_state = android::base::GetProperty("sys.usb.state", "none"); |
| 451 | if (usb_config != usb_state) { |
| 452 | if (!SetUsbConfig("none")) { |
| 453 | LOG(ERROR) << "Failed to clear USB config"; |
| 454 | } |
| 455 | if (!SetUsbConfig(usb_config)) { |
| 456 | LOG(ERROR) << "Failed to set USB config to " << usb_config; |
| 457 | } |
| 458 | } |
| 459 | |
David Anderson | 983e2d5 | 2019-01-02 11:35:38 -0800 | [diff] [blame] | 460 | ui->SetEnableFastbootdLogo(fastboot); |
| 461 | |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 462 | auto ret = fastboot ? StartFastboot(device, args) : start_recovery(device, args); |
| 463 | |
| 464 | if (ret == Device::KEY_INTERRUPTED) { |
| 465 | ret = action.exchange(ret); |
| 466 | if (ret == Device::NO_ACTION) { |
| 467 | continue; |
| 468 | } |
| 469 | } |
| 470 | switch (ret) { |
| 471 | case Device::SHUTDOWN: |
| 472 | ui->Print("Shutting down...\n"); |
| 473 | android::base::SetProperty(ANDROID_RB_PROPERTY, "shutdown,"); |
| 474 | break; |
| 475 | |
| 476 | case Device::REBOOT_BOOTLOADER: |
| 477 | ui->Print("Rebooting to bootloader...\n"); |
| 478 | android::base::SetProperty(ANDROID_RB_PROPERTY, "reboot,bootloader"); |
| 479 | break; |
| 480 | |
| 481 | case Device::ENTER_FASTBOOT: |
David Anderson | 2b2f423 | 2018-10-29 18:48:56 -0700 | [diff] [blame] | 482 | if (logical_partitions_mapped()) { |
| 483 | ui->Print("Partitions may be mounted - rebooting to enter fastboot."); |
| 484 | android::base::SetProperty(ANDROID_RB_PROPERTY, "reboot,fastboot"); |
| 485 | } else { |
| 486 | LOG(INFO) << "Entering fastboot"; |
| 487 | fastboot = true; |
| 488 | } |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 489 | break; |
| 490 | |
| 491 | case Device::ENTER_RECOVERY: |
| 492 | LOG(INFO) << "Entering recovery"; |
| 493 | fastboot = false; |
| 494 | break; |
| 495 | |
| 496 | default: |
| 497 | ui->Print("Rebooting...\n"); |
| 498 | reboot("reboot,"); |
| 499 | break; |
| 500 | } |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 501 | } |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 502 | |
Jerry Zhang | f5e319a | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 503 | // Should be unreachable. |
| 504 | return EXIT_SUCCESS; |
Tao Bao | 6d99d4b | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 505 | } |