The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2007 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 | |
Jerry Zhang | 6a64804 | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 17 | #include "recovery.h" |
Tao Bao | 2ac56af | 2018-04-25 16:47:04 -0700 | [diff] [blame] | 18 | |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 19 | #include <ctype.h> |
| 20 | #include <errno.h> |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 21 | #include <getopt.h> |
Tao Bao | 862a4c1 | 2016-06-02 11:16:50 -0700 | [diff] [blame] | 22 | #include <inttypes.h> |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 23 | #include <limits.h> |
| 24 | #include <linux/input.h> |
| 25 | #include <stdio.h> |
| 26 | #include <stdlib.h> |
| 27 | #include <string.h> |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 28 | #include <sys/types.h> |
Amuru SaiDeepika | 812c1a9 | 2018-10-09 12:59:50 +0530 | [diff] [blame] | 29 | #include <sys/mount.h> |
| 30 | #include <fs_mgr.h> |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 31 | #include <time.h> |
| 32 | #include <unistd.h> |
| 33 | |
Tao Bao | 3aec696 | 2018-04-20 09:24:58 -0700 | [diff] [blame] | 34 | #include <functional> |
Tao Bao | 7f19d10 | 2019-04-26 22:56:56 -0700 | [diff] [blame] | 35 | #include <iterator> |
Tao Bao | c4a18ef | 2017-02-10 00:13:30 -0800 | [diff] [blame] | 36 | #include <memory> |
Tao Bao | 862a4c1 | 2016-06-02 11:16:50 -0700 | [diff] [blame] | 37 | #include <string> |
| 38 | #include <vector> |
Tao Bao | 04ca426 | 2015-09-10 15:32:24 -0700 | [diff] [blame] | 39 | |
Elliott Hughes | 4b166f0 | 2015-12-04 15:30:20 -0800 | [diff] [blame] | 40 | #include <android-base/file.h> |
Tianjie Xu | 7b0ad9c | 2016-08-05 18:00:04 -0700 | [diff] [blame] | 41 | #include <android-base/logging.h> |
Tianjie Xu | 3c62b67 | 2016-02-05 18:25:58 -0800 | [diff] [blame] | 42 | #include <android-base/parseint.h> |
Elliott Hughes | cb22040 | 2016-09-23 15:30:55 -0700 | [diff] [blame] | 43 | #include <android-base/properties.h> |
Elliott Hughes | 4b166f0 | 2015-12-04 15:30:20 -0800 | [diff] [blame] | 44 | #include <android-base/stringprintf.h> |
Tao Bao | 862a4c1 | 2016-06-02 11:16:50 -0700 | [diff] [blame] | 45 | #include <android-base/strings.h> |
Elliott Hughes | cb22040 | 2016-09-23 15:30:55 -0700 | [diff] [blame] | 46 | #include <cutils/properties.h> /* for property_list */ |
Tao Bao | f90d9a1 | 2019-05-10 10:40:59 -0700 | [diff] [blame] | 47 | #include <fs_mgr/roots.h> |
Tianjie Xu | 8cf5c8f | 2016-09-08 20:10:11 -0700 | [diff] [blame] | 48 | #include <ziparchive/zip_archive.h> |
Yabin Cui | 99281df | 2016-02-17 12:21:52 -0800 | [diff] [blame] | 49 | |
Tao Bao | 7f19d10 | 2019-04-26 22:56:56 -0700 | [diff] [blame] | 50 | #include "bootloader_message/bootloader_message.h" |
xunchang | 2478885 | 2019-03-22 16:08:52 -0700 | [diff] [blame] | 51 | #include "install/adb_install.h" |
Tianjie Xu | f6158eb | 2019-06-11 16:09:07 -0700 | [diff] [blame] | 52 | #include "install/fuse_install.h" |
xunchang | 2478885 | 2019-03-22 16:08:52 -0700 | [diff] [blame] | 53 | #include "install/install.h" |
| 54 | #include "install/package.h" |
Alessio Balsini | a9665ce | 2019-12-09 09:26:05 +0000 | [diff] [blame] | 55 | #include "install/snapshot_utils.h" |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 56 | #include "install/wipe_data.h" |
Tao Bao | 7f19d10 | 2019-04-26 22:56:56 -0700 | [diff] [blame] | 57 | #include "install/wipe_device.h" |
Tianjie Xu | b63a221 | 2019-07-29 14:21:49 -0700 | [diff] [blame] | 58 | #include "otautil/boot_state.h" |
Tao Bao | 17054c0 | 2018-05-03 22:41:23 -0700 | [diff] [blame] | 59 | #include "otautil/dirutil.h" |
Tao Bao | 1fc5bf3 | 2017-10-06 07:43:41 -0700 | [diff] [blame] | 60 | #include "otautil/error_code.h" |
Tao Bao | 641fa97 | 2018-04-25 18:59:40 -0700 | [diff] [blame] | 61 | #include "otautil/paths.h" |
Tao Bao | 2c52639 | 2018-05-03 23:01:13 -0700 | [diff] [blame] | 62 | #include "otautil/sysutil.h" |
Tianjie Xu | b5108c3 | 2018-08-20 13:40:47 -0700 | [diff] [blame] | 63 | #include "recovery_ui/screen_ui.h" |
| 64 | #include "recovery_ui/ui.h" |
Tao Bao | 832c9cd | 2019-09-27 23:32:00 -0700 | [diff] [blame] | 65 | #include "recovery_utils/battery_utils.h" |
Tao Bao | e3f09a7 | 2019-10-01 11:55:36 -0700 | [diff] [blame] | 66 | #include "recovery_utils/logging.h" |
| 67 | #include "recovery_utils/roots.h" |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 68 | |
Tao Bao | aac9d9f | 2018-04-29 23:38:59 -0700 | [diff] [blame] | 69 | static constexpr const char* CACHE_LOG_DIR = "/cache/recovery"; |
| 70 | static constexpr const char* COMMAND_FILE = "/cache/recovery/command"; |
Tao Bao | aac9d9f | 2018-04-29 23:38:59 -0700 | [diff] [blame] | 71 | static constexpr const char* LAST_KMSG_FILE = "/cache/recovery/last_kmsg"; |
| 72 | static constexpr const char* LAST_LOG_FILE = "/cache/recovery/last_log"; |
| 73 | static constexpr const char* LOCALE_FILE = "/cache/recovery/last_locale"; |
Tao Bao | aac9d9f | 2018-04-29 23:38:59 -0700 | [diff] [blame] | 74 | static constexpr const char* CACHE_ROOT = "/cache"; |
Nick Kralevich | a9ad032 | 2014-10-22 18:38:48 -0700 | [diff] [blame] | 75 | |
Vinayak Soni | 2e15090 | 2020-11-09 14:25:03 +0530 | [diff] [blame] | 76 | #define MMC_0_TYPE_PATH "/sys/block/mmcblk0/device/type" |
| 77 | #define SDCARD_BLK_0_PATH "/dev/block/mmcblk0p1" |
| 78 | #define MMC_1_TYPE_PATH "/sys/block/mmcblk1/device/type" |
| 79 | #define SDCARD_BLK_1_PATH "/dev/block/mmcblk1p1" |
| 80 | |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 81 | static bool save_current_log = false; |
Tao Bao | ac9d94d | 2016-11-03 11:37:15 -0700 | [diff] [blame] | 82 | |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 83 | /* |
| 84 | * The recovery tool communicates with the main system through /cache files. |
| 85 | * /cache/recovery/command - INPUT - command line for tool, one arg per line |
| 86 | * /cache/recovery/log - OUTPUT - combined log file from recovery run(s) |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 87 | * |
| 88 | * The arguments which may be supplied in the recovery.command file: |
Doug Zongker | d4208f9 | 2010-09-20 12:16:13 -0700 | [diff] [blame] | 89 | * --update_package=path - verify install an OTA package file |
Tianjie Xu | e521861 | 2019-06-25 13:59:39 -0700 | [diff] [blame] | 90 | * --install_with_fuse - install the update package with FUSE. This allows installation of large |
| 91 | * packages on LP32 builds. Since the mmap will otherwise fail due to out of memory. |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 92 | * --wipe_data - erase user data (and cache), then reboot |
Tao Bao | f9f1734 | 2018-04-27 10:44:04 -0700 | [diff] [blame] | 93 | * --prompt_and_wipe_data - prompt the user that data is corrupt, with their consent erase user |
| 94 | * data (and cache), then reboot |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 95 | * --wipe_cache - wipe cache (but not user data), then reboot |
Tao Bao | f9f1734 | 2018-04-27 10:44:04 -0700 | [diff] [blame] | 96 | * --show_text - show the recovery text menu, used by some bootloader (e.g. http://b/36872519). |
Oscar Montemayor | 0523156 | 2009-11-30 08:40:57 -0800 | [diff] [blame] | 97 | * --set_encrypted_filesystem=on|off - enables / diasables encrypted fs |
Doug Zongker | e5d5ac7 | 2012-04-12 11:01:22 -0700 | [diff] [blame] | 98 | * --just_exit - do nothing; exit and reboot |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 99 | * |
| 100 | * After completing, we remove /cache/recovery/command and reboot. |
| 101 | * Arguments may also be supplied in the bootloader control block (BCB). |
| 102 | * These important scenarios must be safely restartable at any point: |
| 103 | * |
| 104 | * FACTORY RESET |
| 105 | * 1. user selects "factory reset" |
| 106 | * 2. main system writes "--wipe_data" to /cache/recovery/command |
| 107 | * 3. main system reboots into recovery |
| 108 | * 4. get_args() writes BCB with "boot-recovery" and "--wipe_data" |
| 109 | * -- after this, rebooting will restart the erase -- |
Doug Zongker | d4208f9 | 2010-09-20 12:16:13 -0700 | [diff] [blame] | 110 | * 5. erase_volume() reformats /data |
| 111 | * 6. erase_volume() reformats /cache |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 112 | * 7. FinishRecovery() erases BCB |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 113 | * -- after this, rebooting will restart the main system -- |
| 114 | * 8. main() calls reboot() to boot main system |
| 115 | * |
| 116 | * OTA INSTALL |
| 117 | * 1. main system downloads OTA package to /cache/some-filename.zip |
Doug Zongker | 9b125b0 | 2010-09-22 12:01:37 -0700 | [diff] [blame] | 118 | * 2. main system writes "--update_package=/cache/some-filename.zip" |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 119 | * 3. main system reboots into recovery |
| 120 | * 4. get_args() writes BCB with "boot-recovery" and "--update_package=..." |
| 121 | * -- after this, rebooting will attempt to reinstall the update -- |
Tao Bao | adc99ef | 2019-04-29 23:48:02 -0700 | [diff] [blame] | 122 | * 5. InstallPackage() attempts to install the update |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 123 | * NOTE: the package install must itself be restartable from any point |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 124 | * 6. FinishRecovery() erases BCB |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 125 | * -- after this, rebooting will (try to) restart the main system -- |
| 126 | * 7. ** if install failed ** |
Tao Bao | adc99ef | 2019-04-29 23:48:02 -0700 | [diff] [blame] | 127 | * 7a. PromptAndWait() shows an error icon and waits for the user |
Tao Bao | c033639 | 2016-12-13 22:29:49 -0800 | [diff] [blame] | 128 | * 7b. the user reboots (pulling the battery, etc) into the main system |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 129 | */ |
| 130 | |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 131 | static bool IsRoDebuggable() { |
| 132 | return android::base::GetBoolProperty("ro.debuggable", false); |
Elliott Hughes | f14af80 | 2015-02-10 14:46:14 -0800 | [diff] [blame] | 133 | } |
| 134 | |
Tao Bao | ec57903 | 2017-07-21 12:13:15 -0700 | [diff] [blame] | 135 | // Clear the recovery command and prepare to boot a (hopefully working) system, |
Tianjie Xu | c14d95d | 2016-03-24 11:50:34 -0700 | [diff] [blame] | 136 | // copy our log file to cache as well (for the system to read). This function is |
| 137 | // idempotent: call it as many times as you like. |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 138 | static void FinishRecovery(RecoveryUI* ui) { |
Tao Bao | 551d2c3 | 2018-05-09 20:53:13 -0700 | [diff] [blame] | 139 | std::string locale = ui->GetLocale(); |
Tao Bao | ec57903 | 2017-07-21 12:13:15 -0700 | [diff] [blame] | 140 | // Save the locale to cache, so if recovery is next started up without a '--locale' argument |
| 141 | // (e.g., directly from the bootloader) it will use the last-known locale. |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 142 | if (!locale.empty() && HasCache()) { |
Tao Bao | ec57903 | 2017-07-21 12:13:15 -0700 | [diff] [blame] | 143 | LOG(INFO) << "Saving locale \"" << locale << "\""; |
| 144 | if (ensure_path_mounted(LOCALE_FILE) != 0) { |
| 145 | LOG(ERROR) << "Failed to mount " << LOCALE_FILE; |
| 146 | } else if (!android::base::WriteStringToFile(locale, LOCALE_FILE)) { |
| 147 | PLOG(ERROR) << "Failed to save locale to " << LOCALE_FILE; |
Doug Zongker | 4f33e55 | 2012-08-23 13:16:12 -0700 | [diff] [blame] | 148 | } |
Tao Bao | ec57903 | 2017-07-21 12:13:15 -0700 | [diff] [blame] | 149 | } |
Doug Zongker | 4f33e55 | 2012-08-23 13:16:12 -0700 | [diff] [blame] | 150 | |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 151 | copy_logs(save_current_log); |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 152 | |
Tao Bao | ec57903 | 2017-07-21 12:13:15 -0700 | [diff] [blame] | 153 | // Reset to normal system boot so recovery won't cycle indefinitely. |
| 154 | std::string err; |
| 155 | if (!clear_bootloader_message(&err)) { |
| 156 | LOG(ERROR) << "Failed to clear BCB message: " << err; |
| 157 | } |
| 158 | |
| 159 | // Remove the command file, so recovery won't repeat indefinitely. |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 160 | if (HasCache()) { |
Tao Bao | ec57903 | 2017-07-21 12:13:15 -0700 | [diff] [blame] | 161 | if (ensure_path_mounted(COMMAND_FILE) != 0 || (unlink(COMMAND_FILE) && errno != ENOENT)) { |
| 162 | LOG(WARNING) << "Can't unlink " << COMMAND_FILE; |
Yabin Cui | 8b309f6 | 2016-06-24 18:22:02 -0700 | [diff] [blame] | 163 | } |
Tao Bao | ec57903 | 2017-07-21 12:13:15 -0700 | [diff] [blame] | 164 | ensure_path_unmounted(CACHE_ROOT); |
| 165 | } |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 166 | |
Tao Bao | ec57903 | 2017-07-21 12:13:15 -0700 | [diff] [blame] | 167 | sync(); // For good measure. |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 168 | } |
| 169 | |
Elliott Hughes | 30694c9 | 2015-03-25 15:16:51 -0700 | [diff] [blame] | 170 | static bool yes_no(Device* device, const char* question1, const char* question2) { |
Tao Bao | 1fe1afe | 2018-05-01 15:56:05 -0700 | [diff] [blame] | 171 | std::vector<std::string> headers{ question1, question2 }; |
| 172 | std::vector<std::string> items{ " No", " Yes" }; |
Doug Zongker | ddd6a28 | 2009-06-09 12:22:33 -0700 | [diff] [blame] | 173 | |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 174 | size_t chosen_item = device->GetUI()->ShowMenu( |
Tao Bao | 3aec696 | 2018-04-20 09:24:58 -0700 | [diff] [blame] | 175 | headers, items, 0, true, |
| 176 | std::bind(&Device::HandleMenuKey, device, std::placeholders::_1, std::placeholders::_2)); |
| 177 | return (chosen_item == 1); |
Elliott Hughes | 30694c9 | 2015-03-25 15:16:51 -0700 | [diff] [blame] | 178 | } |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 179 | |
Paul Crowley | 08404b4 | 2016-12-19 13:04:23 -0800 | [diff] [blame] | 180 | static bool ask_to_wipe_data(Device* device) { |
Tianjie Xu | 1a0a30a | 2018-10-25 15:22:07 -0700 | [diff] [blame] | 181 | std::vector<std::string> headers{ "Wipe all user data?", " THIS CAN NOT BE UNDONE!" }; |
| 182 | std::vector<std::string> items{ " Cancel", " Factory data reset" }; |
| 183 | |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 184 | size_t chosen_item = device->GetUI()->ShowPromptWipeDataConfirmationMenu( |
Tianjie Xu | 1a0a30a | 2018-10-25 15:22:07 -0700 | [diff] [blame] | 185 | headers, items, |
| 186 | std::bind(&Device::HandleMenuKey, device, std::placeholders::_1, std::placeholders::_2)); |
| 187 | |
| 188 | return (chosen_item == 1); |
Paul Crowley | 08404b4 | 2016-12-19 13:04:23 -0800 | [diff] [blame] | 189 | } |
Doug Zongker | 1066d2c | 2009-04-01 13:57:40 -0700 | [diff] [blame] | 190 | |
Hridya Valsaraju | eb6f13a | 2018-09-12 10:25:01 -0700 | [diff] [blame] | 191 | static InstallResult prompt_and_wipe_data(Device* device) { |
Tao Bao | 2bbc6d6 | 2017-08-13 23:48:55 -0700 | [diff] [blame] | 192 | // Use a single string and let ScreenRecoveryUI handles the wrapping. |
Tianjie Xu | b99e606 | 2018-10-16 15:13:09 -0700 | [diff] [blame] | 193 | std::vector<std::string> wipe_data_menu_headers{ |
Tao Bao | 2bbc6d6 | 2017-08-13 23:48:55 -0700 | [diff] [blame] | 194 | "Can't load Android system. Your data may be corrupt. " |
| 195 | "If you continue to get this message, you may need to " |
| 196 | "perform a factory data reset and erase all user data " |
Paul Crowley | 31ac0c6 | 2017-03-23 12:32:40 -0700 | [diff] [blame] | 197 | "stored on this device.", |
Tao Bao | fc5499f | 2017-02-23 19:06:53 -0800 | [diff] [blame] | 198 | }; |
Tao Bao | 1fe1afe | 2018-05-01 15:56:05 -0700 | [diff] [blame] | 199 | // clang-format off |
Tianjie Xu | b99e606 | 2018-10-16 15:13:09 -0700 | [diff] [blame] | 200 | std::vector<std::string> wipe_data_menu_items { |
Paul Crowley | 31ac0c6 | 2017-03-23 12:32:40 -0700 | [diff] [blame] | 201 | "Try again", |
| 202 | "Factory data reset", |
Tao Bao | fc5499f | 2017-02-23 19:06:53 -0800 | [diff] [blame] | 203 | }; |
Tao Bao | 1fe1afe | 2018-05-01 15:56:05 -0700 | [diff] [blame] | 204 | // clang-format on |
Tao Bao | fc5499f | 2017-02-23 19:06:53 -0800 | [diff] [blame] | 205 | for (;;) { |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 206 | size_t chosen_item = device->GetUI()->ShowPromptWipeDataMenu( |
Tianjie Xu | b99e606 | 2018-10-16 15:13:09 -0700 | [diff] [blame] | 207 | wipe_data_menu_headers, wipe_data_menu_items, |
Tao Bao | 3aec696 | 2018-04-20 09:24:58 -0700 | [diff] [blame] | 208 | std::bind(&Device::HandleMenuKey, device, std::placeholders::_1, std::placeholders::_2)); |
Jerry Zhang | b76af93 | 2018-05-22 12:08:35 -0700 | [diff] [blame] | 209 | // If ShowMenu() returned RecoveryUI::KeyError::INTERRUPTED, WaitKey() was interrupted. |
| 210 | if (chosen_item == static_cast<size_t>(RecoveryUI::KeyError::INTERRUPTED)) { |
Hridya Valsaraju | eb6f13a | 2018-09-12 10:25:01 -0700 | [diff] [blame] | 211 | return INSTALL_KEY_INTERRUPTED; |
Jerry Zhang | b76af93 | 2018-05-22 12:08:35 -0700 | [diff] [blame] | 212 | } |
Tao Bao | fc5499f | 2017-02-23 19:06:53 -0800 | [diff] [blame] | 213 | if (chosen_item != 1) { |
Hridya Valsaraju | eb6f13a | 2018-09-12 10:25:01 -0700 | [diff] [blame] | 214 | return INSTALL_SUCCESS; // Just reboot, no wipe; not a failure, user asked for it |
Paul Crowley | 08404b4 | 2016-12-19 13:04:23 -0800 | [diff] [blame] | 215 | } |
Tianjie Xu | b99e606 | 2018-10-16 15:13:09 -0700 | [diff] [blame] | 216 | |
Tao Bao | fc5499f | 2017-02-23 19:06:53 -0800 | [diff] [blame] | 217 | if (ask_to_wipe_data(device)) { |
Tianjie Xu | b63a221 | 2019-07-29 14:21:49 -0700 | [diff] [blame] | 218 | CHECK(device->GetReason().has_value()); |
| 219 | bool convert_fbe = device->GetReason().value() == "convert_fbe"; |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 220 | if (WipeData(device, convert_fbe)) { |
Hridya Valsaraju | eb6f13a | 2018-09-12 10:25:01 -0700 | [diff] [blame] | 221 | return INSTALL_SUCCESS; |
| 222 | } else { |
| 223 | return INSTALL_ERROR; |
| 224 | } |
Tao Bao | fc5499f | 2017-02-23 19:06:53 -0800 | [diff] [blame] | 225 | } |
| 226 | } |
Paul Crowley | 08404b4 | 2016-12-19 13:04:23 -0800 | [diff] [blame] | 227 | } |
| 228 | |
Nick Kralevich | a9ad032 | 2014-10-22 18:38:48 -0700 | [diff] [blame] | 229 | static void choose_recovery_file(Device* device) { |
Tao Bao | 08fc6be | 2017-03-07 00:56:27 -0800 | [diff] [blame] | 230 | std::vector<std::string> entries; |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 231 | if (HasCache()) { |
Tao Bao | 08fc6be | 2017-03-07 00:56:27 -0800 | [diff] [blame] | 232 | for (int i = 0; i < KEEP_LOG_COUNT; i++) { |
| 233 | auto add_to_entries = [&](const char* filename) { |
| 234 | std::string log_file(filename); |
| 235 | if (i > 0) { |
| 236 | log_file += "." + std::to_string(i); |
Tao Bao | bef3971 | 2015-05-04 18:50:27 -0700 | [diff] [blame] | 237 | } |
Tao Bao | 08fc6be | 2017-03-07 00:56:27 -0800 | [diff] [blame] | 238 | |
Yifan Hong | d81b8e3 | 2018-12-17 14:29:06 -0800 | [diff] [blame] | 239 | if (ensure_path_mounted(log_file) == 0 && access(log_file.c_str(), R_OK) == 0) { |
Tao Bao | 08fc6be | 2017-03-07 00:56:27 -0800 | [diff] [blame] | 240 | entries.push_back(std::move(log_file)); |
| 241 | } |
| 242 | }; |
| 243 | |
| 244 | // Add LAST_LOG_FILE + LAST_LOG_FILE.x |
| 245 | add_to_entries(LAST_LOG_FILE); |
| 246 | |
| 247 | // Add LAST_KMSG_FILE + LAST_KMSG_FILE.x |
| 248 | add_to_entries(LAST_KMSG_FILE); |
| 249 | } |
| 250 | } else { |
| 251 | // If cache partition is not found, view /tmp/recovery.log instead. |
Tao Bao | 641fa97 | 2018-04-25 18:59:40 -0700 | [diff] [blame] | 252 | if (access(Paths::Get().temporary_log_file().c_str(), R_OK) == -1) { |
Tao Bao | 08fc6be | 2017-03-07 00:56:27 -0800 | [diff] [blame] | 253 | return; |
Tianjie Xu | a54f75e | 2016-08-17 12:02:46 -0700 | [diff] [blame] | 254 | } else { |
Tao Bao | 641fa97 | 2018-04-25 18:59:40 -0700 | [diff] [blame] | 255 | entries.push_back(Paths::Get().temporary_log_file()); |
Nick Kralevich | a9ad032 | 2014-10-22 18:38:48 -0700 | [diff] [blame] | 256 | } |
Tao Bao | 08fc6be | 2017-03-07 00:56:27 -0800 | [diff] [blame] | 257 | } |
Nick Kralevich | a9ad032 | 2014-10-22 18:38:48 -0700 | [diff] [blame] | 258 | |
Tao Bao | 08fc6be | 2017-03-07 00:56:27 -0800 | [diff] [blame] | 259 | entries.push_back("Back"); |
Elliott Hughes | c049163 | 2015-05-06 12:40:05 -0700 | [diff] [blame] | 260 | |
Tao Bao | 1fe1afe | 2018-05-01 15:56:05 -0700 | [diff] [blame] | 261 | std::vector<std::string> headers{ "Select file to view" }; |
Nick Kralevich | a9ad032 | 2014-10-22 18:38:48 -0700 | [diff] [blame] | 262 | |
Tao Bao | 1fe1afe | 2018-05-01 15:56:05 -0700 | [diff] [blame] | 263 | size_t chosen_item = 0; |
Tao Bao | 08fc6be | 2017-03-07 00:56:27 -0800 | [diff] [blame] | 264 | while (true) { |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 265 | chosen_item = device->GetUI()->ShowMenu( |
Tao Bao | 1fe1afe | 2018-05-01 15:56:05 -0700 | [diff] [blame] | 266 | headers, entries, chosen_item, true, |
Tao Bao | 3aec696 | 2018-04-20 09:24:58 -0700 | [diff] [blame] | 267 | std::bind(&Device::HandleMenuKey, device, std::placeholders::_1, std::placeholders::_2)); |
Jerry Zhang | b76af93 | 2018-05-22 12:08:35 -0700 | [diff] [blame] | 268 | |
| 269 | // Handle WaitKey() interrupt. |
| 270 | if (chosen_item == static_cast<size_t>(RecoveryUI::KeyError::INTERRUPTED)) { |
| 271 | break; |
| 272 | } |
Tao Bao | 08fc6be | 2017-03-07 00:56:27 -0800 | [diff] [blame] | 273 | if (entries[chosen_item] == "Back") break; |
Nick Kralevich | a9ad032 | 2014-10-22 18:38:48 -0700 | [diff] [blame] | 274 | |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 275 | device->GetUI()->ShowFile(entries[chosen_item]); |
Tao Bao | 08fc6be | 2017-03-07 00:56:27 -0800 | [diff] [blame] | 276 | } |
Nick Kralevich | a9ad032 | 2014-10-22 18:38:48 -0700 | [diff] [blame] | 277 | } |
| 278 | |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 279 | static void run_graphics_test(RecoveryUI* ui) { |
Tao Bao | db7e898 | 2017-03-06 23:53:16 -0800 | [diff] [blame] | 280 | // Switch to graphics screen. |
| 281 | ui->ShowText(false); |
Elliott Hughes | 498cda6 | 2016-04-14 16:49:04 -0700 | [diff] [blame] | 282 | |
Tao Bao | db7e898 | 2017-03-06 23:53:16 -0800 | [diff] [blame] | 283 | ui->SetProgressType(RecoveryUI::INDETERMINATE); |
| 284 | ui->SetBackground(RecoveryUI::INSTALLING_UPDATE); |
| 285 | sleep(1); |
Elliott Hughes | 498cda6 | 2016-04-14 16:49:04 -0700 | [diff] [blame] | 286 | |
Tao Bao | db7e898 | 2017-03-06 23:53:16 -0800 | [diff] [blame] | 287 | ui->SetBackground(RecoveryUI::ERROR); |
| 288 | sleep(1); |
Elliott Hughes | 498cda6 | 2016-04-14 16:49:04 -0700 | [diff] [blame] | 289 | |
Tao Bao | db7e898 | 2017-03-06 23:53:16 -0800 | [diff] [blame] | 290 | ui->SetBackground(RecoveryUI::NO_COMMAND); |
| 291 | sleep(1); |
Elliott Hughes | 498cda6 | 2016-04-14 16:49:04 -0700 | [diff] [blame] | 292 | |
Tao Bao | db7e898 | 2017-03-06 23:53:16 -0800 | [diff] [blame] | 293 | ui->SetBackground(RecoveryUI::ERASING); |
| 294 | sleep(1); |
Elliott Hughes | 498cda6 | 2016-04-14 16:49:04 -0700 | [diff] [blame] | 295 | |
Tao Bao | db7e898 | 2017-03-06 23:53:16 -0800 | [diff] [blame] | 296 | // Calling SetBackground() after SetStage() to trigger a redraw. |
| 297 | ui->SetStage(1, 3); |
| 298 | ui->SetBackground(RecoveryUI::INSTALLING_UPDATE); |
| 299 | sleep(1); |
| 300 | ui->SetStage(2, 3); |
| 301 | ui->SetBackground(RecoveryUI::INSTALLING_UPDATE); |
| 302 | sleep(1); |
| 303 | ui->SetStage(3, 3); |
| 304 | ui->SetBackground(RecoveryUI::INSTALLING_UPDATE); |
| 305 | sleep(1); |
Elliott Hughes | 498cda6 | 2016-04-14 16:49:04 -0700 | [diff] [blame] | 306 | |
Tao Bao | db7e898 | 2017-03-06 23:53:16 -0800 | [diff] [blame] | 307 | ui->SetStage(-1, -1); |
| 308 | ui->SetBackground(RecoveryUI::INSTALLING_UPDATE); |
Elliott Hughes | 498cda6 | 2016-04-14 16:49:04 -0700 | [diff] [blame] | 309 | |
Tao Bao | db7e898 | 2017-03-06 23:53:16 -0800 | [diff] [blame] | 310 | ui->SetProgressType(RecoveryUI::DETERMINATE); |
| 311 | ui->ShowProgress(1.0, 10.0); |
| 312 | float fraction = 0.0; |
| 313 | for (size_t i = 0; i < 100; ++i) { |
| 314 | fraction += .01; |
| 315 | ui->SetProgress(fraction); |
| 316 | usleep(100000); |
| 317 | } |
| 318 | |
| 319 | ui->ShowText(true); |
Elliott Hughes | 498cda6 | 2016-04-14 16:49:04 -0700 | [diff] [blame] | 320 | } |
| 321 | |
Vinayak Soni | 2e15090 | 2020-11-09 14:25:03 +0530 | [diff] [blame] | 322 | // Check whether the mmc type of provided path (/sys/block/mmcblk*/device/type) |
| 323 | // is SD (sdcard) or not. |
| 324 | static int check_mmc_is_sdcard (const char* mmc_type_path) |
Amuru SaiDeepika | 812c1a9 | 2018-10-09 12:59:50 +0530 | [diff] [blame] | 325 | { |
Vinayak Soni | 2e15090 | 2020-11-09 14:25:03 +0530 | [diff] [blame] | 326 | std::string mmc_type; |
| 327 | |
| 328 | LOG(INFO) << "Checking mmc type for path : " << mmc_type_path; |
| 329 | |
| 330 | if (!android::base::ReadFileToString(mmc_type_path, &mmc_type)) { |
| 331 | LOG(ERROR) << "Failed to read mmc type : " << strerror(errno); |
| 332 | return -1; |
| 333 | } |
| 334 | LOG(INFO) << "MMC type is : " << mmc_type.c_str(); |
| 335 | if (!strncmp(mmc_type.c_str(), "SD", strlen("SD"))) |
| 336 | return 0; |
| 337 | else |
| 338 | return -1; |
Amuru SaiDeepika | 812c1a9 | 2018-10-09 12:59:50 +0530 | [diff] [blame] | 339 | } |
| 340 | |
Vinayak Soni | 2e15090 | 2020-11-09 14:25:03 +0530 | [diff] [blame] | 341 | // Gather mount point and other info from fstab, find the right block |
| 342 | // path where sdcard is mounted, and try mounting it. |
| 343 | static int do_sdcard_mount(RecoveryUI* ui) |
Amuru SaiDeepika | 812c1a9 | 2018-10-09 12:59:50 +0530 | [diff] [blame] | 344 | { |
Vinayak Soni | 2e15090 | 2020-11-09 14:25:03 +0530 | [diff] [blame] | 345 | int rc = 0; |
| 346 | ui->Print("Update via sdcard. Mounting sdcard\n"); |
| 347 | Volume *v = volume_for_mount_point("/sdcard"); |
| 348 | if (v == nullptr) { |
| 349 | ui->Print("Unknown volume for /sdcard. Check fstab\n"); |
| 350 | goto error; |
| 351 | } |
| 352 | if (strncmp(v->fs_type.c_str(), "vfat", sizeof("vfat"))) { |
| 353 | ui->Print("Unsupported format on the sdcard: %s\n", |
| 354 | v->fs_type.c_str()); |
| 355 | goto error; |
| 356 | } |
| 357 | |
| 358 | if (check_mmc_is_sdcard(MMC_0_TYPE_PATH) == 0) { |
| 359 | LOG(INFO) << "Mounting sdcard on " << SDCARD_BLK_0_PATH; |
| 360 | rc = mount(SDCARD_BLK_0_PATH, |
| 361 | v->mount_point.c_str(), |
| 362 | v->fs_type.c_str(), |
| 363 | v->flags, |
| 364 | v->fs_options.c_str()); |
| 365 | } |
| 366 | else if (check_mmc_is_sdcard(MMC_1_TYPE_PATH) == 0) { |
| 367 | LOG(INFO) << "Mounting sdcard on " << SDCARD_BLK_1_PATH; |
| 368 | rc = mount(SDCARD_BLK_1_PATH, |
| 369 | v->mount_point.c_str(), |
| 370 | v->fs_type.c_str(), |
| 371 | v->flags, |
| 372 | v->fs_options.c_str()); |
| 373 | } |
| 374 | else { |
| 375 | LOG(ERROR) << "Unable to get the block path for sdcard."; |
| 376 | goto error; |
| 377 | } |
| 378 | |
| 379 | if (rc) { |
| 380 | ui->Print("Failed to mount sdcard : %s\n", |
| 381 | strerror(errno)); |
| 382 | goto error; |
| 383 | } |
| 384 | ui->Print("Done mounting sdcard\n"); |
| 385 | return 0; |
| 386 | |
Amuru SaiDeepika | 812c1a9 | 2018-10-09 12:59:50 +0530 | [diff] [blame] | 387 | error: |
Vinayak Soni | 2e15090 | 2020-11-09 14:25:03 +0530 | [diff] [blame] | 388 | return -1; |
Amuru SaiDeepika | 812c1a9 | 2018-10-09 12:59:50 +0530 | [diff] [blame] | 389 | } |
| 390 | |
Yifan Hong | 13b0996 | 2020-01-29 16:51:33 -0800 | [diff] [blame] | 391 | static void WriteUpdateInProgress() { |
| 392 | std::string err; |
| 393 | if (!update_bootloader_message({ "--reason=update_in_progress" }, &err)) { |
| 394 | LOG(ERROR) << "Failed to WriteUpdateInProgress: " << err; |
Yifan Hong | c9163fd | 2020-01-27 17:37:39 -0800 | [diff] [blame] | 395 | } |
Yifan Hong | 13b0996 | 2020-01-29 16:51:33 -0800 | [diff] [blame] | 396 | } |
Yifan Hong | c9163fd | 2020-01-27 17:37:39 -0800 | [diff] [blame] | 397 | |
Yifan Hong | 13b0996 | 2020-01-29 16:51:33 -0800 | [diff] [blame] | 398 | static bool AskToReboot(Device* device, Device::BuiltinAction chosen_action) { |
Yifan Hong | c9163fd | 2020-01-27 17:37:39 -0800 | [diff] [blame] | 399 | bool is_non_ab = android::base::GetProperty("ro.boot.slot_suffix", "").empty(); |
| 400 | bool is_virtual_ab = android::base::GetBoolProperty("ro.virtual_ab.enabled", false); |
| 401 | if (!is_non_ab && !is_virtual_ab) { |
| 402 | // Only prompt for non-A/B or Virtual A/B devices. |
| 403 | return true; |
| 404 | } |
| 405 | |
| 406 | std::string header_text; |
| 407 | std::string item_text; |
| 408 | switch (chosen_action) { |
| 409 | case Device::REBOOT: |
| 410 | header_text = "reboot"; |
| 411 | item_text = " Reboot system now"; |
| 412 | break; |
| 413 | case Device::SHUTDOWN: |
| 414 | header_text = "power off"; |
| 415 | item_text = " Power off"; |
| 416 | break; |
| 417 | default: |
| 418 | LOG(FATAL) << "Invalid chosen action " << chosen_action; |
| 419 | break; |
| 420 | } |
| 421 | |
Yifan Hong | 13b0996 | 2020-01-29 16:51:33 -0800 | [diff] [blame] | 422 | std::vector<std::string> headers{ "WARNING: Previous installation has failed.", |
Yifan Hong | c9163fd | 2020-01-27 17:37:39 -0800 | [diff] [blame] | 423 | " Your device may fail to boot if you " + header_text + |
| 424 | " now.", |
| 425 | " Confirm reboot?" }; |
| 426 | std::vector<std::string> items{ " Cancel", item_text }; |
| 427 | |
| 428 | size_t chosen_item = device->GetUI()->ShowMenu( |
| 429 | headers, items, 0, true /* menu_only */, |
| 430 | std::bind(&Device::HandleMenuKey, device, std::placeholders::_1, std::placeholders::_2)); |
| 431 | |
| 432 | return (chosen_item == 1); |
| 433 | } |
| 434 | |
Tao Bao | adc99ef | 2019-04-29 23:48:02 -0700 | [diff] [blame] | 435 | // Shows the recovery UI and waits for user input. Returns one of the device builtin actions, such |
| 436 | // as REBOOT, SHUTDOWN, or REBOOT_BOOTLOADER. Returning NO_ACTION means to take the default, which |
| 437 | // is to reboot or shutdown depending on if the --shutdown_after flag was passed to recovery. |
| 438 | static Device::BuiltinAction PromptAndWait(Device* device, InstallResult status) { |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 439 | auto ui = device->GetUI(); |
Yifan Hong | 13b0996 | 2020-01-29 16:51:33 -0800 | [diff] [blame] | 440 | bool update_in_progress = (device->GetReason().value_or("") == "update_in_progress"); |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 441 | for (;;) { |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 442 | FinishRecovery(ui); |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 443 | switch (status) { |
| 444 | case INSTALL_SUCCESS: |
| 445 | case INSTALL_NONE: |
Tao Bao | adc99ef | 2019-04-29 23:48:02 -0700 | [diff] [blame] | 446 | case INSTALL_SKIPPED: |
| 447 | case INSTALL_RETRY: |
| 448 | case INSTALL_KEY_INTERRUPTED: |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 449 | ui->SetBackground(RecoveryUI::NO_COMMAND); |
| 450 | break; |
Doug Zongker | 6c8553d | 2012-09-24 10:40:47 -0700 | [diff] [blame] | 451 | |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 452 | case INSTALL_ERROR: |
| 453 | case INSTALL_CORRUPT: |
| 454 | ui->SetBackground(RecoveryUI::ERROR); |
| 455 | break; |
Tao Bao | adc99ef | 2019-04-29 23:48:02 -0700 | [diff] [blame] | 456 | |
| 457 | case INSTALL_REBOOT: |
| 458 | // All the reboots should have been handled prior to entering PromptAndWait() or immediately |
| 459 | // after installing a package. |
| 460 | LOG(FATAL) << "Invalid status code of INSTALL_REBOOT"; |
| 461 | break; |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 462 | } |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 463 | ui->SetProgressType(RecoveryUI::EMPTY); |
| 464 | |
Yifan Hong | 13b0996 | 2020-01-29 16:51:33 -0800 | [diff] [blame] | 465 | std::vector<std::string> headers; |
| 466 | if (update_in_progress) { |
| 467 | headers = { "WARNING: Previous installation has failed.", |
| 468 | " Your device may fail to boot if you reboot or power off now." }; |
| 469 | } |
| 470 | |
Tao Bao | 1fe1afe | 2018-05-01 15:56:05 -0700 | [diff] [blame] | 471 | size_t chosen_item = ui->ShowMenu( |
Yifan Hong | 13b0996 | 2020-01-29 16:51:33 -0800 | [diff] [blame] | 472 | headers, device->GetMenuItems(), 0, false, |
Tao Bao | 3aec696 | 2018-04-20 09:24:58 -0700 | [diff] [blame] | 473 | std::bind(&Device::HandleMenuKey, device, std::placeholders::_1, std::placeholders::_2)); |
Jerry Zhang | b76af93 | 2018-05-22 12:08:35 -0700 | [diff] [blame] | 474 | // Handle Interrupt key |
| 475 | if (chosen_item == static_cast<size_t>(RecoveryUI::KeyError::INTERRUPTED)) { |
| 476 | return Device::KEY_INTERRUPTED; |
| 477 | } |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 478 | // Device-specific code may take some action here. It may return one of the core actions |
| 479 | // handled in the switch statement below. |
Jerry Zhang | b76af93 | 2018-05-22 12:08:35 -0700 | [diff] [blame] | 480 | Device::BuiltinAction chosen_action = |
| 481 | (chosen_item == static_cast<size_t>(RecoveryUI::KeyError::TIMED_OUT)) |
| 482 | ? Device::REBOOT |
| 483 | : device->InvokeMenuItem(chosen_item); |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 484 | |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 485 | switch (chosen_action) { |
Mark Salyzyn | 488cc05 | 2019-05-20 10:36:16 -0700 | [diff] [blame] | 486 | case Device::REBOOT_FROM_FASTBOOT: // Can not happen |
| 487 | case Device::SHUTDOWN_FROM_FASTBOOT: // Can not happen |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 488 | case Device::NO_ACTION: |
| 489 | break; |
| 490 | |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 491 | case Device::ENTER_FASTBOOT: |
| 492 | case Device::ENTER_RECOVERY: |
Tao Bao | 75321ad | 2019-04-23 11:46:25 -0700 | [diff] [blame] | 493 | case Device::REBOOT_BOOTLOADER: |
| 494 | case Device::REBOOT_FASTBOOT: |
| 495 | case Device::REBOOT_RECOVERY: |
| 496 | case Device::REBOOT_RESCUE: |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 497 | return chosen_action; |
| 498 | |
Yifan Hong | c9163fd | 2020-01-27 17:37:39 -0800 | [diff] [blame] | 499 | case Device::REBOOT: |
| 500 | case Device::SHUTDOWN: |
Yifan Hong | 13b0996 | 2020-01-29 16:51:33 -0800 | [diff] [blame] | 501 | if (!ui->IsTextVisible()) { |
| 502 | return Device::REBOOT; |
| 503 | } |
| 504 | // okay to reboot; no need to ask. |
| 505 | if (!update_in_progress) { |
| 506 | return Device::REBOOT; |
| 507 | } |
| 508 | // An update might have been failed. Ask if user really wants to reboot. |
| 509 | if (AskToReboot(device, chosen_action)) { |
Yifan Hong | c9163fd | 2020-01-27 17:37:39 -0800 | [diff] [blame] | 510 | return Device::REBOOT; |
| 511 | } |
| 512 | break; |
| 513 | |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 514 | case Device::WIPE_DATA: |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 515 | save_current_log = true; |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 516 | if (ui->IsTextVisible()) { |
| 517 | if (ask_to_wipe_data(device)) { |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 518 | WipeData(device, false); |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 519 | } |
| 520 | } else { |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 521 | WipeData(device, false); |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 522 | return Device::NO_ACTION; |
| 523 | } |
| 524 | break; |
| 525 | |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 526 | case Device::WIPE_CACHE: { |
| 527 | save_current_log = true; |
| 528 | std::function<bool()> confirm_func = [&device]() { |
| 529 | return yes_no(device, "Wipe cache?", " THIS CAN NOT BE UNDONE!"); |
| 530 | }; |
| 531 | WipeCache(ui, ui->IsTextVisible() ? confirm_func : nullptr); |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 532 | if (!ui->IsTextVisible()) return Device::NO_ACTION; |
| 533 | break; |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 534 | } |
Tao Bao | 7b9b7db | 2019-04-19 15:22:15 -0700 | [diff] [blame] | 535 | |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 536 | case Device::APPLY_ADB_SIDELOAD: |
Tao Bao | 378bfbf | 2019-04-16 14:22:25 -0700 | [diff] [blame] | 537 | case Device::APPLY_SDCARD: |
| 538 | case Device::ENTER_RESCUE: { |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 539 | save_current_log = true; |
Tao Bao | 378bfbf | 2019-04-16 14:22:25 -0700 | [diff] [blame] | 540 | |
Yifan Hong | 13b0996 | 2020-01-29 16:51:33 -0800 | [diff] [blame] | 541 | update_in_progress = true; |
| 542 | WriteUpdateInProgress(); |
| 543 | |
Tao Bao | 378bfbf | 2019-04-16 14:22:25 -0700 | [diff] [blame] | 544 | bool adb = true; |
Tao Bao | 7b9b7db | 2019-04-19 15:22:15 -0700 | [diff] [blame] | 545 | Device::BuiltinAction reboot_action; |
Tao Bao | 378bfbf | 2019-04-16 14:22:25 -0700 | [diff] [blame] | 546 | if (chosen_action == Device::ENTER_RESCUE) { |
| 547 | // Switch to graphics screen. |
| 548 | ui->ShowText(false); |
xunchang | 5a1916b | 2019-04-22 12:18:14 -0700 | [diff] [blame] | 549 | status = ApplyFromAdb(device, true /* rescue_mode */, &reboot_action); |
Tao Bao | 378bfbf | 2019-04-16 14:22:25 -0700 | [diff] [blame] | 550 | } else if (chosen_action == Device::APPLY_ADB_SIDELOAD) { |
xunchang | 5a1916b | 2019-04-22 12:18:14 -0700 | [diff] [blame] | 551 | status = ApplyFromAdb(device, false /* rescue_mode */, &reboot_action); |
xunchang | 3cc23d5 | 2019-03-18 15:03:33 -0700 | [diff] [blame] | 552 | } else { |
Tao Bao | 378bfbf | 2019-04-16 14:22:25 -0700 | [diff] [blame] | 553 | adb = false; |
Tianjie Xu | f6158eb | 2019-06-11 16:09:07 -0700 | [diff] [blame] | 554 | status = ApplyFromSdcard(device); |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 555 | } |
xunchang | 3cc23d5 | 2019-03-18 15:03:33 -0700 | [diff] [blame] | 556 | |
Tao Bao | 7b9b7db | 2019-04-19 15:22:15 -0700 | [diff] [blame] | 557 | ui->Print("\nInstall from %s completed with status %d.\n", adb ? "ADB" : "SD card", status); |
| 558 | if (status == INSTALL_REBOOT) { |
| 559 | return reboot_action; |
| 560 | } |
| 561 | |
Yifan Hong | 13b0996 | 2020-01-29 16:51:33 -0800 | [diff] [blame] | 562 | if (status == INSTALL_SUCCESS) { |
| 563 | update_in_progress = false; |
| 564 | if (!ui->IsTextVisible()) { |
| 565 | return Device::NO_ACTION; // reboot if logs aren't visible |
| 566 | } |
| 567 | } else { |
xunchang | 3cc23d5 | 2019-03-18 15:03:33 -0700 | [diff] [blame] | 568 | ui->SetBackground(RecoveryUI::ERROR); |
| 569 | ui->Print("Installation aborted.\n"); |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 570 | copy_logs(save_current_log); |
xunchang | 3cc23d5 | 2019-03-18 15:03:33 -0700 | [diff] [blame] | 571 | } |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 572 | break; |
xunchang | 3cc23d5 | 2019-03-18 15:03:33 -0700 | [diff] [blame] | 573 | } |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 574 | |
| 575 | case Device::VIEW_RECOVERY_LOGS: |
| 576 | choose_recovery_file(device); |
| 577 | break; |
| 578 | |
| 579 | case Device::RUN_GRAPHICS_TEST: |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 580 | run_graphics_test(ui); |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 581 | break; |
| 582 | |
Tianjie Xu | 29d5575 | 2017-09-20 17:53:46 -0700 | [diff] [blame] | 583 | case Device::RUN_LOCALE_TEST: { |
| 584 | ScreenRecoveryUI* screen_ui = static_cast<ScreenRecoveryUI*>(ui); |
Tao Bao | 39c4918 | 2018-05-07 22:50:33 -0700 | [diff] [blame] | 585 | screen_ui->CheckBackgroundTextImages(); |
Tianjie Xu | 29d5575 | 2017-09-20 17:53:46 -0700 | [diff] [blame] | 586 | break; |
| 587 | } |
Alessio Balsini | a9665ce | 2019-12-09 09:26:05 +0000 | [diff] [blame] | 588 | |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 589 | case Device::MOUNT_SYSTEM: |
Alessio Balsini | a9665ce | 2019-12-09 09:26:05 +0000 | [diff] [blame] | 590 | // For Virtual A/B, set up the snapshot devices (if exist). |
| 591 | if (!CreateSnapshotPartitions()) { |
| 592 | ui->Print("Virtual A/B: snapshot partitions creation failed.\n"); |
| 593 | break; |
| 594 | } |
Tao Bao | f90d9a1 | 2019-05-10 10:40:59 -0700 | [diff] [blame] | 595 | if (ensure_path_mounted_at(android::fs_mgr::GetSystemRoot(), "/mnt/system") != -1) { |
Yifan Hong | 4932780 | 2018-11-26 14:59:09 -0800 | [diff] [blame] | 596 | ui->Print("Mounted /system.\n"); |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 597 | } |
| 598 | break; |
Jerry Zhang | b76af93 | 2018-05-22 12:08:35 -0700 | [diff] [blame] | 599 | |
| 600 | case Device::KEY_INTERRUPTED: |
| 601 | return Device::KEY_INTERRUPTED; |
Tao Bao | 50dd532 | 2017-03-07 14:57:04 -0800 | [diff] [blame] | 602 | } |
| 603 | } |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 604 | } |
| 605 | |
Tao Bao | 99f0d9e | 2016-10-13 12:46:38 -0700 | [diff] [blame] | 606 | static void print_property(const char* key, const char* name, void* /* cookie */) { |
| 607 | printf("%s=%s\n", key, name); |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 608 | } |
| 609 | |
Tao Bao | 832c9cd | 2019-09-27 23:32:00 -0700 | [diff] [blame] | 610 | static bool IsBatteryOk(int* required_battery_level) { |
| 611 | // GmsCore enters recovery mode to install package when having enough battery percentage. |
| 612 | // Normally, the threshold is 40% without charger and 20% with charger. So we check the battery |
| 613 | // level against a slightly lower limit. |
| 614 | constexpr int BATTERY_OK_PERCENTAGE = 20; |
| 615 | constexpr int BATTERY_WITH_CHARGER_OK_PERCENTAGE = 15; |
Yabin Cui | 99281df | 2016-02-17 12:21:52 -0800 | [diff] [blame] | 616 | |
Tao Bao | 832c9cd | 2019-09-27 23:32:00 -0700 | [diff] [blame] | 617 | auto battery_info = GetBatteryInfo(); |
| 618 | *required_battery_level = |
| 619 | battery_info.charging ? BATTERY_WITH_CHARGER_OK_PERCENTAGE : BATTERY_OK_PERCENTAGE; |
| 620 | return battery_info.capacity >= *required_battery_level; |
Yabin Cui | 99281df | 2016-02-17 12:21:52 -0800 | [diff] [blame] | 621 | } |
| 622 | |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 623 | // Set the retry count to |retry_count| in BCB. |
Tianjie Xu | 72449c9 | 2017-05-16 18:07:31 -0700 | [diff] [blame] | 624 | static void set_retry_bootloader_message(int retry_count, const std::vector<std::string>& args) { |
| 625 | std::vector<std::string> options; |
| 626 | for (const auto& arg : args) { |
| 627 | if (!android::base::StartsWith(arg, "--retry_count")) { |
| 628 | options.push_back(arg); |
Tianjie Xu | 3c62b67 | 2016-02-05 18:25:58 -0800 | [diff] [blame] | 629 | } |
Tianjie Xu | 72449c9 | 2017-05-16 18:07:31 -0700 | [diff] [blame] | 630 | } |
Tianjie Xu | 3c62b67 | 2016-02-05 18:25:58 -0800 | [diff] [blame] | 631 | |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 632 | // Update the retry counter in BCB. |
| 633 | options.push_back(android::base::StringPrintf("--retry_count=%d", retry_count)); |
Tianjie Xu | 72449c9 | 2017-05-16 18:07:31 -0700 | [diff] [blame] | 634 | std::string err; |
| 635 | if (!update_bootloader_message(options, &err)) { |
| 636 | LOG(ERROR) << err; |
| 637 | } |
Tianjie Xu | 3c62b67 | 2016-02-05 18:25:58 -0800 | [diff] [blame] | 638 | } |
| 639 | |
Tianjie Xu | 06e57ac | 2016-07-11 14:04:08 -0700 | [diff] [blame] | 640 | static bool bootreason_in_blacklist() { |
Tao Bao | efc3559 | 2017-01-08 22:45:47 -0800 | [diff] [blame] | 641 | std::string bootreason = android::base::GetProperty("ro.boot.bootreason", ""); |
| 642 | if (!bootreason.empty()) { |
Tao Bao | aac9d9f | 2018-04-29 23:38:59 -0700 | [diff] [blame] | 643 | // More bootreasons can be found in "system/core/bootstat/bootstat.cpp". |
| 644 | static const std::vector<std::string> kBootreasonBlacklist{ |
| 645 | "kernel_panic", |
| 646 | "Panic", |
| 647 | }; |
| 648 | for (const auto& str : kBootreasonBlacklist) { |
| 649 | if (android::base::EqualsIgnoreCase(str, bootreason)) return true; |
Tianjie Xu | 06e57ac | 2016-07-11 14:04:08 -0700 | [diff] [blame] | 650 | } |
Tao Bao | efc3559 | 2017-01-08 22:45:47 -0800 | [diff] [blame] | 651 | } |
| 652 | return false; |
Tianjie Xu | 06e57ac | 2016-07-11 14:04:08 -0700 | [diff] [blame] | 653 | } |
| 654 | |
Tao Bao | 641fa97 | 2018-04-25 18:59:40 -0700 | [diff] [blame] | 655 | static void log_failure_code(ErrorCode code, const std::string& update_package) { |
| 656 | std::vector<std::string> log_buffer = { |
| 657 | update_package, |
| 658 | "0", // install result |
| 659 | "error: " + std::to_string(code), |
| 660 | }; |
| 661 | std::string log_content = android::base::Join(log_buffer, "\n"); |
| 662 | const std::string& install_file = Paths::Get().temporary_install_file(); |
| 663 | if (!android::base::WriteStringToFile(log_content, install_file)) { |
| 664 | PLOG(ERROR) << "Failed to write " << install_file; |
| 665 | } |
Tianjie Xu | 06e57ac | 2016-07-11 14:04:08 -0700 | [diff] [blame] | 666 | |
Tao Bao | 641fa97 | 2018-04-25 18:59:40 -0700 | [diff] [blame] | 667 | // Also write the info into last_log. |
| 668 | LOG(INFO) << log_content; |
Tianjie Xu | 06e57ac | 2016-07-11 14:04:08 -0700 | [diff] [blame] | 669 | } |
| 670 | |
Jerry Zhang | 6a64804 | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 671 | Device::BuiltinAction start_recovery(Device* device, const std::vector<std::string>& args) { |
Tao Bao | f9f1734 | 2018-04-27 10:44:04 -0700 | [diff] [blame] | 672 | static constexpr struct option OPTIONS[] = { |
Hridya Valsaraju | 20c81b3 | 2018-07-27 22:09:12 -0700 | [diff] [blame] | 673 | { "fastboot", no_argument, nullptr, 0 }, |
Tianjie Xu | e521861 | 2019-06-25 13:59:39 -0700 | [diff] [blame] | 674 | { "install_with_fuse", no_argument, nullptr, 0 }, |
Tao Bao | f9f1734 | 2018-04-27 10:44:04 -0700 | [diff] [blame] | 675 | { "just_exit", no_argument, nullptr, 'x' }, |
| 676 | { "locale", required_argument, nullptr, 0 }, |
| 677 | { "prompt_and_wipe_data", no_argument, nullptr, 0 }, |
| 678 | { "reason", required_argument, nullptr, 0 }, |
Tao Bao | 75321ad | 2019-04-23 11:46:25 -0700 | [diff] [blame] | 679 | { "rescue", no_argument, nullptr, 0 }, |
Tao Bao | f9f1734 | 2018-04-27 10:44:04 -0700 | [diff] [blame] | 680 | { "retry_count", required_argument, nullptr, 0 }, |
| 681 | { "security", no_argument, nullptr, 0 }, |
| 682 | { "show_text", no_argument, nullptr, 't' }, |
| 683 | { "shutdown_after", no_argument, nullptr, 0 }, |
| 684 | { "sideload", no_argument, nullptr, 0 }, |
| 685 | { "sideload_auto_reboot", no_argument, nullptr, 0 }, |
| 686 | { "update_package", required_argument, nullptr, 0 }, |
| 687 | { "wipe_ab", no_argument, nullptr, 0 }, |
| 688 | { "wipe_cache", no_argument, nullptr, 0 }, |
| 689 | { "wipe_data", no_argument, nullptr, 0 }, |
| 690 | { "wipe_package_size", required_argument, nullptr, 0 }, |
| 691 | { nullptr, 0, nullptr, 0 }, |
| 692 | }; |
| 693 | |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 694 | const char* update_package = nullptr; |
Tianjie Xu | e521861 | 2019-06-25 13:59:39 -0700 | [diff] [blame] | 695 | bool install_with_fuse = false; // memory map the update package by default. |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 696 | bool should_wipe_data = false; |
| 697 | bool should_prompt_and_wipe_data = false; |
| 698 | bool should_wipe_cache = false; |
| 699 | bool should_wipe_ab = false; |
| 700 | size_t wipe_package_size = 0; |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 701 | bool sideload = false; |
| 702 | bool sideload_auto_reboot = false; |
Tao Bao | 75321ad | 2019-04-23 11:46:25 -0700 | [diff] [blame] | 703 | bool rescue = false; |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 704 | bool just_exit = false; |
| 705 | bool shutdown_after = false; |
| 706 | int retry_count = 0; |
| 707 | bool security_update = false; |
Jerry Zhang | 2dea53e | 2018-05-02 17:15:03 -0700 | [diff] [blame] | 708 | std::string locale; |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 709 | |
Tao Bao | 1700cc4 | 2018-07-16 22:09:59 -0700 | [diff] [blame] | 710 | auto args_to_parse = StringVectorToNullTerminatedArray(args); |
Steven Laver | 375a9a9 | 2019-08-07 15:49:46 -0700 | [diff] [blame] | 711 | InstallResult status = INSTALL_SUCCESS; |
| 712 | |
Tao Bao | 1700cc4 | 2018-07-16 22:09:59 -0700 | [diff] [blame] | 713 | |
Bill Peckham | c90493c | 2019-08-28 14:41:43 -0700 | [diff] [blame] | 714 | if (HasCache() && ensure_path_mounted(CACHE_ROOT) == 0) { |
Hari Veerubhotla | 82b044f | 2018-08-23 12:44:58 +0530 | [diff] [blame] | 715 | //Create /cache/recovery specifically if it is not created |
| 716 | //As in cases where device is booted into recovery directly after |
| 717 | //flashing recovery folder is not created in init |
Bill Peckham | c90493c | 2019-08-28 14:41:43 -0700 | [diff] [blame] | 718 | // TODO(b/140199946) Check what to pass for selabel_handle argument. |
| 719 | mkdir_recursively(CACHE_LOG_DIR, 0777, false, nullptr); |
Hari Veerubhotla | 82b044f | 2018-08-23 12:44:58 +0530 | [diff] [blame] | 720 | } |
| 721 | |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 722 | int arg; |
| 723 | int option_index; |
Tao Bao | 1700cc4 | 2018-07-16 22:09:59 -0700 | [diff] [blame] | 724 | // Parse everything before the last element (which must be a nullptr). getopt_long(3) expects a |
| 725 | // null-terminated char* array, but without counting null as an arg (i.e. argv[argc] should be |
| 726 | // nullptr). |
| 727 | while ((arg = getopt_long(args_to_parse.size() - 1, args_to_parse.data(), "", OPTIONS, |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 728 | &option_index)) != -1) { |
| 729 | switch (arg) { |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 730 | case 't': |
Jerry Zhang | 6a64804 | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 731 | // Handled in recovery_main.cpp |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 732 | break; |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 733 | case 'x': |
| 734 | just_exit = true; |
| 735 | break; |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 736 | case 0: { |
| 737 | std::string option = OPTIONS[option_index].name; |
David Anderson | 969787c | 2020-01-27 09:29:02 -0800 | [diff] [blame] | 738 | if (option == "install_with_fuse") { |
Tianjie Xu | e521861 | 2019-06-25 13:59:39 -0700 | [diff] [blame] | 739 | install_with_fuse = true; |
Tianjie Xu | b63a221 | 2019-07-29 14:21:49 -0700 | [diff] [blame] | 740 | } else if (option == "locale" || option == "fastboot" || option == "reason") { |
Jerry Zhang | 6a64804 | 2018-05-04 11:24:10 -0700 | [diff] [blame] | 741 | // Handled in recovery_main.cpp |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 742 | } else if (option == "prompt_and_wipe_data") { |
| 743 | should_prompt_and_wipe_data = true; |
Tao Bao | 75321ad | 2019-04-23 11:46:25 -0700 | [diff] [blame] | 744 | } else if (option == "rescue") { |
| 745 | rescue = true; |
Tao Bao | f9f1734 | 2018-04-27 10:44:04 -0700 | [diff] [blame] | 746 | } else if (option == "retry_count") { |
| 747 | android::base::ParseInt(optarg, &retry_count, 0); |
| 748 | } else if (option == "security") { |
| 749 | security_update = true; |
| 750 | } else if (option == "sideload") { |
| 751 | sideload = true; |
| 752 | } else if (option == "sideload_auto_reboot") { |
| 753 | sideload = true; |
| 754 | sideload_auto_reboot = true; |
| 755 | } else if (option == "shutdown_after") { |
| 756 | shutdown_after = true; |
| 757 | } else if (option == "update_package") { |
| 758 | update_package = optarg; |
| 759 | } else if (option == "wipe_ab") { |
| 760 | should_wipe_ab = true; |
| 761 | } else if (option == "wipe_cache") { |
| 762 | should_wipe_cache = true; |
| 763 | } else if (option == "wipe_data") { |
| 764 | should_wipe_data = true; |
| 765 | } else if (option == "wipe_package_size") { |
| 766 | android::base::ParseUint(optarg, &wipe_package_size); |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 767 | } |
| 768 | break; |
| 769 | } |
| 770 | case '?': |
| 771 | LOG(ERROR) << "Invalid command argument"; |
| 772 | continue; |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 773 | } |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 774 | } |
Jerry Zhang | 49fd5d2 | 2018-05-17 12:54:41 -0700 | [diff] [blame] | 775 | optind = 1; |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 776 | |
Steven Laver | 375a9a9 | 2019-08-07 15:49:46 -0700 | [diff] [blame] | 777 | // next_action indicates the next target to reboot into upon finishing the install. It could be |
| 778 | // overridden to a different reboot target per user request. |
| 779 | Device::BuiltinAction next_action = shutdown_after ? Device::SHUTDOWN : Device::REBOOT; |
| 780 | |
Tianjie Xu | b63a221 | 2019-07-29 14:21:49 -0700 | [diff] [blame] | 781 | printf("stage is [%s]\n", device->GetStage().value_or("").c_str()); |
| 782 | printf("reason is [%s]\n", device->GetReason().value_or("").c_str()); |
Doug Zongker | 02ec6b8 | 2012-08-22 17:26:40 -0700 | [diff] [blame] | 783 | |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 784 | auto ui = device->GetUI(); |
Doug Zongker | 02ec6b8 | 2012-08-22 17:26:40 -0700 | [diff] [blame] | 785 | |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 786 | // Set background string to "installing security update" for security update, |
| 787 | // otherwise set it to "installing system update". |
| 788 | ui->SetSystemUpdateText(security_update); |
| 789 | |
| 790 | int st_cur, st_max; |
Tianjie Xu | b63a221 | 2019-07-29 14:21:49 -0700 | [diff] [blame] | 791 | if (!device->GetStage().has_value() && |
| 792 | sscanf(device->GetStage().value().c_str(), "%d/%d", &st_cur, &st_max) == 2) { |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 793 | ui->SetStage(st_cur, st_max); |
| 794 | } |
| 795 | |
Jerry Zhang | 0e577ee | 2018-05-07 11:21:10 -0700 | [diff] [blame] | 796 | std::vector<std::string> title_lines = |
| 797 | android::base::Split(android::base::GetProperty("ro.bootimage.build.fingerprint", ""), ":"); |
| 798 | title_lines.insert(std::begin(title_lines), "Android Recovery"); |
| 799 | ui->SetTitle(title_lines); |
| 800 | |
Jerry Zhang | b76af93 | 2018-05-22 12:08:35 -0700 | [diff] [blame] | 801 | ui->ResetKeyInterruptStatus(); |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 802 | device->StartRecovery(); |
| 803 | |
| 804 | printf("Command:"); |
| 805 | for (const auto& arg : args) { |
| 806 | printf(" \"%s\"", arg.c_str()); |
| 807 | } |
| 808 | printf("\n\n"); |
| 809 | |
Anuj Singhal | 60951fe | 2018-05-25 15:31:40 +0530 | [diff] [blame] | 810 | if (update_package) { |
| 811 | if (!strncmp("/sdcard", update_package, 7)) { |
Vinayak Soni | 2e15090 | 2020-11-09 14:25:03 +0530 | [diff] [blame] | 812 | if(do_sdcard_mount(ui) != 0) { |
| 813 | status = INSTALL_ERROR; |
| 814 | goto error; |
Anuj Singhal | 60951fe | 2018-05-25 15:31:40 +0530 | [diff] [blame] | 815 | } |
| 816 | } |
| 817 | } |
| 818 | |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 819 | property_list(print_property, nullptr); |
| 820 | printf("\n"); |
| 821 | |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 822 | if (update_package != nullptr) { |
| 823 | // It's not entirely true that we will modify the flash. But we want |
| 824 | // to log the update attempt since update_package is non-NULL. |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 825 | save_current_log = true; |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 826 | |
Tao Bao | 832c9cd | 2019-09-27 23:32:00 -0700 | [diff] [blame] | 827 | if (int required_battery_level; retry_count == 0 && !IsBatteryOk(&required_battery_level)) { |
Tao Bao | 6d90a9d | 2018-04-26 10:40:36 -0700 | [diff] [blame] | 828 | ui->Print("battery capacity is not enough for installing package: %d%% needed\n", |
| 829 | required_battery_level); |
Tao Bao | 832c9cd | 2019-09-27 23:32:00 -0700 | [diff] [blame] | 830 | // Log the error code to last_install when installation skips due to low battery. |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 831 | log_failure_code(kLowBattery, update_package); |
| 832 | status = INSTALL_SKIPPED; |
Tianjie Xu | a6f49bd | 2018-03-26 14:32:11 -0700 | [diff] [blame] | 833 | } else if (retry_count == 0 && bootreason_in_blacklist()) { |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 834 | // Skip update-on-reboot when bootreason is kernel_panic or similar |
| 835 | ui->Print("bootreason is in the blacklist; skip OTA installation\n"); |
| 836 | log_failure_code(kBootreasonInBlacklist, update_package); |
| 837 | status = INSTALL_SKIPPED; |
Dmitri Plotnikov | 8706a98 | 2017-04-18 08:28:26 -0700 | [diff] [blame] | 838 | } else { |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 839 | // It's a fresh update. Initialize the retry_count in the BCB to 1; therefore we can later |
| 840 | // identify the interrupted update due to unexpected reboots. |
| 841 | if (retry_count == 0) { |
| 842 | set_retry_bootloader_message(retry_count + 1, args); |
Tao Bao | 7022f33 | 2017-07-25 09:52:36 -0700 | [diff] [blame] | 843 | } |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 844 | |
Tianjie Xu | 58a2769 | 2019-08-06 12:32:05 -0700 | [diff] [blame] | 845 | bool should_use_fuse = false; |
| 846 | if (!SetupPackageMount(update_package, &should_use_fuse)) { |
| 847 | LOG(INFO) << "Failed to set up the package access, skipping installation"; |
| 848 | status = INSTALL_ERROR; |
| 849 | } else if (install_with_fuse || should_use_fuse) { |
Tianjie Xu | e521861 | 2019-06-25 13:59:39 -0700 | [diff] [blame] | 850 | LOG(INFO) << "Installing package " << update_package << " with fuse"; |
| 851 | status = InstallWithFuseFromPath(update_package, ui); |
| 852 | } else if (auto memory_package = Package::CreateMemoryPackage( |
| 853 | update_package, |
| 854 | std::bind(&RecoveryUI::SetProgress, ui, std::placeholders::_1)); |
| 855 | memory_package != nullptr) { |
| 856 | status = InstallPackage(memory_package.get(), update_package, should_wipe_cache, |
| 857 | retry_count, ui); |
| 858 | } else { |
| 859 | // We may fail to memory map the package on 32 bit builds for packages with 2GiB+ size. |
| 860 | // In such cases, we will try to install the package with fuse. This is not the default |
| 861 | // installation method because it introduces a layer of indirection from the kernel space. |
| 862 | LOG(WARNING) << "Failed to memory map package " << update_package |
| 863 | << "; falling back to install with fuse"; |
| 864 | status = InstallWithFuseFromPath(update_package, ui); |
| 865 | } |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 866 | if (status != INSTALL_SUCCESS) { |
| 867 | ui->Print("Installation aborted.\n"); |
Tao Bao | aac9d9f | 2018-04-29 23:38:59 -0700 | [diff] [blame] | 868 | |
| 869 | // When I/O error or bspatch/imgpatch error happens, reboot and retry installation |
| 870 | // RETRY_LIMIT times before we abandon this OTA update. |
| 871 | static constexpr int RETRY_LIMIT = 4; |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 872 | if (status == INSTALL_RETRY && retry_count < RETRY_LIMIT) { |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 873 | copy_logs(save_current_log); |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 874 | retry_count += 1; |
| 875 | set_retry_bootloader_message(retry_count, args); |
| 876 | // Print retry count on screen. |
| 877 | ui->Print("Retry attempt %d\n", retry_count); |
| 878 | |
Tao Bao | 782dcc1 | 2019-04-29 11:23:16 -0700 | [diff] [blame] | 879 | // Reboot back into recovery to retry the update. |
Tianjie Xu | e8ca1b8 | 2020-03-13 14:25:02 -0700 | [diff] [blame] | 880 | Reboot("recovery"); |
Tianjie Xu | d9d1629 | 2017-04-20 18:08:21 -0700 | [diff] [blame] | 881 | } |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 882 | // If this is an eng or userdebug build, then automatically |
| 883 | // turn the text display on if the script fails so the error |
| 884 | // message is visible. |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 885 | if (IsRoDebuggable()) { |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 886 | ui->ShowText(true); |
Tao Bao | c679f93 | 2015-03-30 09:43:49 -0700 | [diff] [blame] | 887 | } |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 888 | } |
Doug Zongker | 8674a72 | 2010-09-15 11:08:23 -0700 | [diff] [blame] | 889 | } |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 890 | } else if (should_wipe_data) { |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 891 | save_current_log = true; |
Tianjie Xu | b63a221 | 2019-07-29 14:21:49 -0700 | [diff] [blame] | 892 | CHECK(device->GetReason().has_value()); |
| 893 | bool convert_fbe = device->GetReason().value() == "convert_fbe"; |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 894 | if (!WipeData(device, convert_fbe)) { |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 895 | status = INSTALL_ERROR; |
Doug Zongker | b1d1263 | 2014-03-18 10:32:12 -0700 | [diff] [blame] | 896 | } |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 897 | } else if (should_prompt_and_wipe_data) { |
Tao Bao | a5bbcb9 | 2018-09-17 14:32:47 -0700 | [diff] [blame] | 898 | // Trigger the logging to capture the cause, even if user chooses to not wipe data. |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 899 | save_current_log = true; |
Tao Bao | a5bbcb9 | 2018-09-17 14:32:47 -0700 | [diff] [blame] | 900 | |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 901 | ui->ShowText(true); |
| 902 | ui->SetBackground(RecoveryUI::ERROR); |
Hridya Valsaraju | eb6f13a | 2018-09-12 10:25:01 -0700 | [diff] [blame] | 903 | status = prompt_and_wipe_data(device); |
| 904 | if (status != INSTALL_KEY_INTERRUPTED) { |
| 905 | ui->ShowText(false); |
Tao Bao | 7523863 | 2015-05-27 14:46:17 -0700 | [diff] [blame] | 906 | } |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 907 | } else if (should_wipe_cache) { |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 908 | save_current_log = true; |
| 909 | if (!WipeCache(ui, nullptr)) { |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 910 | status = INSTALL_ERROR; |
| 911 | } |
| 912 | } else if (should_wipe_ab) { |
Tao Bao | 7f19d10 | 2019-04-26 22:56:56 -0700 | [diff] [blame] | 913 | if (!WipeAbDevice(device, wipe_package_size)) { |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 914 | status = INSTALL_ERROR; |
| 915 | } |
| 916 | } else if (sideload) { |
Tao Bao | 7b9b7db | 2019-04-19 15:22:15 -0700 | [diff] [blame] | 917 | // 'adb reboot sideload' acts the same as user presses key combinations to enter the sideload |
| 918 | // mode. When 'sideload-auto-reboot' is used, text display will NOT be turned on by default. And |
| 919 | // it will reboot after sideload finishes even if there are errors. This is to enable automated |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 920 | // testing. |
xunchang | 388d253 | 2019-04-12 16:22:15 -0700 | [diff] [blame] | 921 | save_current_log = true; |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 922 | if (!sideload_auto_reboot) { |
| 923 | ui->ShowText(true); |
| 924 | } |
xunchang | 5a1916b | 2019-04-22 12:18:14 -0700 | [diff] [blame] | 925 | status = ApplyFromAdb(device, false /* rescue_mode */, &next_action); |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 926 | ui->Print("\nInstall from ADB complete (status: %d).\n", status); |
| 927 | if (sideload_auto_reboot) { |
Tao Bao | 7b9b7db | 2019-04-19 15:22:15 -0700 | [diff] [blame] | 928 | status = INSTALL_REBOOT; |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 929 | ui->Print("Rebooting automatically.\n"); |
| 930 | } |
Tao Bao | 75321ad | 2019-04-23 11:46:25 -0700 | [diff] [blame] | 931 | } else if (rescue) { |
| 932 | save_current_log = true; |
xunchang | 5a1916b | 2019-04-22 12:18:14 -0700 | [diff] [blame] | 933 | status = ApplyFromAdb(device, true /* rescue_mode */, &next_action); |
Tao Bao | 75321ad | 2019-04-23 11:46:25 -0700 | [diff] [blame] | 934 | ui->Print("\nInstall from ADB complete (status: %d).\n", status); |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 935 | } else if (!just_exit) { |
| 936 | // If this is an eng or userdebug build, automatically turn on the text display if no command |
| 937 | // is specified. Note that this should be called before setting the background to avoid |
| 938 | // flickering the background image. |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 939 | if (IsRoDebuggable()) { |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 940 | ui->ShowText(true); |
| 941 | } |
| 942 | status = INSTALL_NONE; // No command specified |
| 943 | ui->SetBackground(RecoveryUI::NO_COMMAND); |
| 944 | } |
| 945 | |
Anuj Singhal | 60951fe | 2018-05-25 15:31:40 +0530 | [diff] [blame] | 946 | error: |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 947 | if (status == INSTALL_ERROR || status == INSTALL_CORRUPT) { |
| 948 | ui->SetBackground(RecoveryUI::ERROR); |
| 949 | if (!ui->IsTextVisible()) { |
| 950 | sleep(5); |
| 951 | } |
| 952 | } |
| 953 | |
Tao Bao | 7b9b7db | 2019-04-19 15:22:15 -0700 | [diff] [blame] | 954 | // Determine the next action. |
| 955 | // - If the state is INSTALL_REBOOT, device will reboot into the target as specified in |
| 956 | // `next_action`. |
| 957 | // - If the recovery menu is visible, prompt and wait for commands. |
| 958 | // - If the state is INSTALL_NONE, wait for commands (e.g. in user build, one manually boots |
| 959 | // into recovery to sideload a package or to wipe the device). |
| 960 | // - In all other cases, reboot the device. Therefore, normal users will observe the device |
| 961 | // rebooting a) immediately upon successful finish (INSTALL_SUCCESS); or b) an "error" screen |
| 962 | // for 5s followed by an automatic reboot. |
| 963 | if (status != INSTALL_REBOOT) { |
| 964 | if (status == INSTALL_NONE || ui->IsTextVisible()) { |
Tao Bao | adc99ef | 2019-04-29 23:48:02 -0700 | [diff] [blame] | 965 | auto temp = PromptAndWait(device, status); |
Tao Bao | 7b9b7db | 2019-04-19 15:22:15 -0700 | [diff] [blame] | 966 | if (temp != Device::NO_ACTION) { |
| 967 | next_action = temp; |
| 968 | } |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 969 | } |
| 970 | } |
| 971 | |
| 972 | // Save logs and clean up before rebooting or shutting down. |
Tianjie Xu | 164c60a | 2019-05-15 13:59:39 -0700 | [diff] [blame] | 973 | FinishRecovery(ui); |
Tianjie Xu | 99b73be | 2017-11-28 17:23:06 -0800 | [diff] [blame] | 974 | |
Tao Bao | 7b9b7db | 2019-04-19 15:22:15 -0700 | [diff] [blame] | 975 | return next_action; |
The Android Open Source Project | c24a8e6 | 2009-03-03 19:28:42 -0800 | [diff] [blame] | 976 | } |