| /* |
| TWRP is free software: you can redistribute it and/or modify |
| it under the terms of the GNU General Public License as published by |
| the Free Software Foundation, either version 3 of the License, or |
| (at your option) any later version. |
| |
| TWRP is distributed in the hope that it will be useful, |
| but WITHOUT ANY WARRANTY; without even the implied warranty of |
| MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| GNU General Public License for more details. |
| |
| You should have received a copy of the GNU General Public License |
| along with TWRP. If not, see <http://www.gnu.org/licenses/>. |
| */ |
| |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string> |
| #include <sys/stat.h> |
| #include <sys/types.h> |
| #include <time.h> |
| #include <unistd.h> |
| #include <signal.h> |
| #include "gui/twmsg.h" |
| |
| #include "cutils/properties.h" |
| #include "bootloader_message_twrp/include/bootloader_message_twrp/bootloader_message.h" |
| |
| #ifdef ANDROID_RB_RESTART |
| #include "cutils/android_reboot.h" |
| #else |
| #include <sys/reboot.h> |
| #endif |
| |
| extern "C" { |
| #include "gui/gui.h" |
| } |
| #include "set_metadata.h" |
| #include "gui/gui.hpp" |
| #include "gui/pages.hpp" |
| #include "gui/objects.hpp" |
| #include "twcommon.h" |
| #include "twrp-functions.hpp" |
| #include "data.hpp" |
| #include "partitions.hpp" |
| #ifdef __ANDROID_API_N__ |
| #include <android-base/strings.h> |
| #else |
| #include <base/strings.h> |
| #endif |
| #include "openrecoveryscript.hpp" |
| #include "variables.h" |
| #include "twrpAdbBuFifo.hpp" |
| #include "twrpApex.hpp" |
| #ifdef TW_USE_NEW_MINADBD |
| // #include "minadbd/minadbd.h" |
| #else |
| extern "C" { |
| #include "minadbd21/adb.h" |
| } |
| #endif |
| |
| //extern int adb_server_main(int is_daemon, int server_port, int /* reply_fd */); |
| |
| TWPartitionManager PartitionManager; |
| int Log_Offset; |
| bool datamedia; |
| |
| static void Print_Prop(const char *key, const char *name, void *cookie) { |
| printf("%s=%s\n", key, name); |
| } |
| |
| static void Decrypt_Page(bool SkipDecryption, bool datamedia) { |
| // Offer to decrypt if the device is encrypted |
| if (DataManager::GetIntValue(TW_IS_ENCRYPTED) != 0) { |
| if (SkipDecryption) { |
| LOGINFO("Skipping decryption\n"); |
| } else { |
| LOGINFO("Is encrypted, do decrypt page first\n"); |
| if (gui_startPage("decrypt", 1, 1) != 0) { |
| LOGERR("Failed to start decrypt GUI page.\n"); |
| } else { |
| // Check for and load custom theme if present |
| TWFunc::check_selinux_support(); |
| gui_loadCustomResources(); |
| } |
| } |
| } else if (datamedia) { |
| TWFunc::check_selinux_support(); |
| if (tw_get_default_metadata(DataManager::GetSettingsStoragePath().c_str()) != 0) { |
| LOGINFO("Failed to get default contexts and file mode for storage files.\n"); |
| } else { |
| LOGINFO("Got default contexts and file mode for storage files.\n"); |
| } |
| } |
| } |
| |
| int main(int argc, char **argv) { |
| // Recovery needs to install world-readable files, so clear umask |
| // set by init |
| umask(0); |
| |
| Log_Offset = 0; |
| |
| // Set up temporary log file (/tmp/recovery.log) |
| freopen(TMP_LOG_FILE, "a", stdout); |
| setbuf(stdout, NULL); |
| freopen(TMP_LOG_FILE, "a", stderr); |
| setbuf(stderr, NULL); |
| |
| signal(SIGPIPE, SIG_IGN); |
| |
| // Handle ADB sideload |
| if (argc == 3 && strcmp(argv[1], "--adbd") == 0) { |
| property_set("ctl.stop", "adbd"); |
| #ifdef TW_USE_NEW_MINADBD |
| //adb_server_main(0, DEFAULT_ADB_PORT, -1); TODO fix this for android8 |
| // minadbd_main(); |
| #else |
| adb_main(argv[2]); |
| #endif |
| return 0; |
| } |
| |
| #ifdef RECOVERY_SDCARD_ON_DATA |
| datamedia = true; |
| #endif |
| |
| char crash_prop_val[PROPERTY_VALUE_MAX]; |
| int crash_counter; |
| property_get("twrp.crash_counter", crash_prop_val, "-1"); |
| crash_counter = atoi(crash_prop_val) + 1; |
| snprintf(crash_prop_val, sizeof(crash_prop_val), "%d", crash_counter); |
| property_set("twrp.crash_counter", crash_prop_val); |
| property_set("ro.twrp.boot", "1"); |
| property_set("ro.twrp.version", TW_VERSION_STR); |
| |
| time_t StartupTime = time(NULL); |
| printf("Starting TWRP %s-%s on %s (pid %d)\n", TW_VERSION_STR, TW_GIT_REVISION, ctime(&StartupTime), getpid()); |
| |
| // Load default values to set DataManager constants and handle ifdefs |
| DataManager::SetDefaultValues(); |
| printf("Starting the UI...\n"); |
| gui_init(); |
| printf("=> Linking mtab\n"); |
| symlink("/proc/mounts", "/etc/mtab"); |
| std::string fstab_filename = "/etc/twrp.fstab"; |
| if (!TWFunc::Path_Exists(fstab_filename)) { |
| fstab_filename = "/etc/recovery.fstab"; |
| } |
| |
| // Begin SAR detection |
| { |
| TWPartitionManager SarPartitionManager; |
| printf("=> Processing %s for SAR-detection\n", fstab_filename.c_str()); |
| if (!SarPartitionManager.Process_Fstab(fstab_filename, 1, 1)) { |
| LOGERR("Failing out of recovery due to problem with fstab.\n"); |
| return -1; |
| } |
| |
| mkdir("/s", 0755); |
| |
| #if defined(AB_OTA_UPDATER) || defined(__ANDROID_API_Q__) |
| bool fallback_sar = true; |
| #else |
| bool fallback_sar = property_get_bool("ro.build.system_root_image", false); |
| #endif |
| |
| if(SarPartitionManager.Mount_By_Path("/s", false)) { |
| if (TWFunc::Path_Exists("/s/build.prop")) { |
| LOGINFO("SAR-DETECT: Non-SAR System detected\n"); |
| property_set("ro.twrp.sar", "0"); |
| rmdir("/system_root"); |
| } else if (TWFunc::Path_Exists("/s/system/build.prop")) { |
| LOGINFO("SAR-DETECT: SAR System detected\n"); |
| property_set("ro.twrp.sar", "1"); |
| } else { |
| LOGINFO("SAR-DETECT: No build.prop found, falling back to %s\n", fallback_sar ? "SAR" : "Non-SAR"); |
| property_set("ro.twrp.sar", fallback_sar ? "1" : "0"); |
| } |
| |
| SarPartitionManager.UnMount_By_Path("/s", false); |
| } else { |
| LOGINFO("SAR-DETECT: Could not mount system partition, falling back to %s\n", fallback_sar ? "SAR":"Non-SAR"); |
| property_set("ro.twrp.sar", fallback_sar ? "1" : "0"); |
| } |
| |
| rmdir("/s"); |
| |
| TWFunc::check_and_run_script("/system/bin/sarsetup.sh", "boot"); |
| } |
| // End SAR detection |
| |
| printf("=> Processing %s\n", fstab_filename.c_str()); |
| if (!PartitionManager.Process_Fstab(fstab_filename, 1, 0)) { |
| LOGERR("Failing out of recovery due to problem with fstab.\n"); |
| return -1; |
| } |
| PartitionManager.Output_Partition_Logging(); |
| #ifdef TW_INCLUDE_CRYPTO |
| DataManager::SetValue(TW_IS_ENCRYPTED, 1); |
| #endif |
| |
| if (PartitionManager.Get_Super_Status()) { |
| PartitionManager.Setup_Super_Devices(); |
| PartitionManager.Setup_Super_Partition(); |
| } else { |
| #ifdef TW_INCLUDE_CRYPTO |
| if (!PartitionManager.Get_Super_Status()) |
| PartitionManager.Decrypt_Data(); |
| #endif |
| } |
| |
| // Load up all the resources |
| gui_loadResources(); |
| |
| bool Shutdown = false; |
| bool SkipDecryption = false; |
| string Send_Intent = ""; |
| { |
| TWPartition* misc = PartitionManager.Find_Partition_By_Path("/misc"); |
| if (misc != NULL) { |
| if (misc->Current_File_System == "emmc") { |
| set_misc_device(misc->Actual_Block_Device.c_str()); |
| } else { |
| LOGERR("Only emmc /misc is supported\n"); |
| } |
| } |
| get_args(&argc, &argv); |
| |
| int index, index2, len; |
| char* argptr; |
| char* ptr; |
| printf("Startup Commands: "); |
| for (index = 1; index < argc; index++) { |
| argptr = argv[index]; |
| printf(" '%s'", argv[index]); |
| len = strlen(argv[index]); |
| if (*argptr == '-') {argptr++; len--;} |
| if (*argptr == '-') {argptr++; len--;} |
| if (*argptr == 'u') { |
| ptr = argptr; |
| index2 = 0; |
| while (*ptr != '=' && *ptr != '\n') |
| ptr++; |
| // skip the = before grabbing Zip_File |
| while (*ptr == '=') |
| ptr++; |
| if (*ptr) { |
| string ORSCommand = "install "; |
| ORSCommand.append(ptr); |
| |
| // If we have a map of blocks we don't need to mount data. |
| SkipDecryption = *ptr == '@'; |
| |
| if (!OpenRecoveryScript::Insert_ORS_Command(ORSCommand)) |
| break; |
| } else |
| LOGERR("argument error specifying zip file\n"); |
| } else if (*argptr == 'w') { |
| if (len == 9) { |
| if (!OpenRecoveryScript::Insert_ORS_Command("wipe data\n")) |
| break; |
| } else if (len == 10) { |
| if (!OpenRecoveryScript::Insert_ORS_Command("wipe cache\n")) |
| break; |
| } |
| // Other 'w' items are wipe_ab and wipe_package_size which are related to bricking the device remotely. We will not bother to suppor these as having TWRP probably makes "bricking" the device in this manner useless |
| } else if (*argptr == 'n') { |
| DataManager::SetValue(TW_BACKUP_NAME, gui_parse_text("{@auto_generate}")); |
| if (!OpenRecoveryScript::Insert_ORS_Command("backup BSDCAE\n")) |
| break; |
| } else if (*argptr == 'p') { |
| Shutdown = true; |
| } else if (*argptr == 's') { |
| if (strncmp(argptr, "send_intent", strlen("send_intent")) == 0) { |
| ptr = argptr + strlen("send_intent") + 1; |
| Send_Intent = *ptr; |
| } else if (strncmp(argptr, "security", strlen("security")) == 0) { |
| LOGINFO("Security update\n"); |
| } else if (strncmp(argptr, "sideload", strlen("sideload")) == 0) { |
| if (!OpenRecoveryScript::Insert_ORS_Command("sideload\n")) |
| break; |
| } else if (strncmp(argptr, "stages", strlen("stages")) == 0) { |
| LOGINFO("ignoring stages command\n"); |
| } |
| } else if (*argptr == 'r') { |
| if (strncmp(argptr, "reason", strlen("reason")) == 0) { |
| ptr = argptr + strlen("reason") + 1; |
| gui_print("%s\n", ptr); |
| } |
| } |
| } |
| printf("\n"); |
| } |
| |
| if (crash_counter == 0) { |
| property_list(Print_Prop, NULL); |
| printf("\n"); |
| } else { |
| printf("twrp.crash_counter=%d\n", crash_counter); |
| } |
| |
| // Check for and run startup script if script exists |
| TWFunc::check_and_run_script("/system/bin/runatboot.sh", "boot"); |
| TWFunc::check_and_run_script("/system/bin/postrecoveryboot.sh", "boot"); |
| |
| #ifdef TW_INCLUDE_INJECTTWRP |
| // Back up TWRP Ramdisk if needed: |
| TWPartition* Boot = PartitionManager.Find_Partition_By_Path("/boot"); |
| LOGINFO("Backing up TWRP ramdisk...\n"); |
| if (Boot == NULL || Boot->Current_File_System != "emmc") |
| TWFunc::Exec_Cmd("injecttwrp --backup /tmp/backup_recovery_ramdisk.img"); |
| else { |
| string injectcmd = "injecttwrp --backup /tmp/backup_recovery_ramdisk.img bd=" + Boot->Actual_Block_Device; |
| TWFunc::Exec_Cmd(injectcmd); |
| } |
| LOGINFO("Backup of TWRP ramdisk done.\n"); |
| #endif |
| |
| if (!PartitionManager.Get_Super_Status()) |
| Decrypt_Page(SkipDecryption, datamedia); |
| |
| // Fixup the RTC clock on devices which require it |
| if (crash_counter == 0) |
| TWFunc::Fixup_Time_On_Boot(); |
| |
| if (!PartitionManager.Get_Super_Status()) |
| DataManager::ReadSettingsFile(); |
| PageManager::LoadLanguage(DataManager::GetStrValue("tw_language")); |
| GUIConsole::Translate_Now(); |
| |
| // Run any outstanding OpenRecoveryScript |
| std::string cacheDir = TWFunc::get_log_dir(); |
| if (cacheDir == DATA_LOGS_DIR) |
| cacheDir = "/data/cache"; |
| std::string orsFile = cacheDir + "/recovery/openrecoveryscript"; |
| if ((DataManager::GetIntValue(TW_IS_ENCRYPTED) == 0 || SkipDecryption) && (TWFunc::Path_Exists(SCRIPT_FILE_TMP) || TWFunc::Path_Exists(orsFile))) { |
| OpenRecoveryScript::Run_OpenRecoveryScript(); |
| } |
| |
| #ifdef TW_HAS_MTP |
| char mtp_crash_check[PROPERTY_VALUE_MAX]; |
| property_get("mtp.crash_check", mtp_crash_check, "0"); |
| if (DataManager::GetIntValue("tw_mtp_enabled") |
| && !strcmp(mtp_crash_check, "0") && !crash_counter |
| && (!DataManager::GetIntValue(TW_IS_ENCRYPTED) || DataManager::GetIntValue(TW_IS_DECRYPTED))) { |
| property_set("mtp.crash_check", "1"); |
| LOGINFO("Starting MTP\n"); |
| if (!PartitionManager.Enable_MTP()) |
| PartitionManager.Disable_MTP(); |
| else |
| gui_msg("mtp_enabled=MTP Enabled"); |
| property_set("mtp.crash_check", "0"); |
| } else if (strcmp(mtp_crash_check, "0")) { |
| gui_warn("mtp_crash=MTP Crashed, not starting MTP on boot."); |
| DataManager::SetValue("tw_mtp_enabled", 0); |
| PartitionManager.Disable_MTP(); |
| } else if (crash_counter == 1) { |
| LOGINFO("TWRP crashed; disabling MTP as a precaution.\n"); |
| PartitionManager.Disable_MTP(); |
| } |
| #endif |
| |
| #ifndef TW_OEM_BUILD |
| // Check if system has never been changed |
| TWPartition* sys = PartitionManager.Find_Partition_By_Path(PartitionManager.Get_Android_Root_Path()); |
| TWPartition* ven = PartitionManager.Find_Partition_By_Path("/vendor"); |
| if (sys) { |
| if (sys->Get_Super_Status()) { |
| if (!PartitionManager.Prepare_All_Super_Volumes()) { |
| LOGERR("Unable to prepare super volumes.\n"); |
| } |
| sys->Mount(true); |
| if (ven) { |
| ven->Mount(true); |
| } |
| twrpApex apex; |
| if (!apex.loadApexImages()) { |
| LOGERR("Unable to load apex images from %s\n", APEX_DIR); |
| } |
| property_set("twrp.apex.loaded", "true"); |
| #ifdef TW_INCLUDE_CRYPTO |
| PartitionManager.Decrypt_Data(); |
| Decrypt_Page(SkipDecryption, datamedia); |
| std::string recoveryLogDir(DATA_LOGS_DIR); |
| recoveryLogDir += "/recovery"; |
| if (!TWFunc::Path_Exists(recoveryLogDir)) { |
| bool created = PartitionManager.Recreate_Logs_Dir(); |
| if (!created) |
| LOGERR("Unable to create log directory for TWRP\n"); |
| } |
| DataManager::ReadSettingsFile(); |
| #endif |
| } else { |
| if ((DataManager::GetIntValue("tw_mount_system_ro") == 0 && sys->Check_Lifetime_Writes() == 0) || DataManager::GetIntValue("tw_mount_system_ro") == 2) { |
| if (DataManager::GetIntValue("tw_never_show_system_ro_page") == 0) { |
| DataManager::SetValue("tw_back", "main"); |
| if (gui_startPage("system_readonly", 1, 1) != 0) { |
| LOGERR("Failed to start system_readonly GUI page.\n"); |
| } |
| } else if (DataManager::GetIntValue("tw_mount_system_ro") == 0) { |
| sys->Change_Mount_Read_Only(false); |
| if (ven) |
| ven->Change_Mount_Read_Only(false); |
| } |
| } else if (DataManager::GetIntValue("tw_mount_system_ro") == 1) { |
| // Do nothing, user selected to leave system read only |
| } else { |
| sys->Change_Mount_Read_Only(false); |
| if (ven) |
| ven->Change_Mount_Read_Only(false); |
| } |
| } |
| } |
| #endif |
| // We are doing this here to allow super partition to be set up prior to overriding properties |
| #if defined(TW_INCLUDE_LIBRESETPROP) && defined(TW_OVERRIDE_SYSTEM_PROPS) |
| if (!PartitionManager.Mount_By_Path(PartitionManager.Get_Android_Root_Path(), true)) { |
| LOGERR("Unable to mount %s\n", PartitionManager.Get_Android_Root_Path().c_str()); |
| } else { |
| stringstream override_props(EXPAND(TW_OVERRIDE_SYSTEM_PROPS)); |
| string current_prop; |
| while (getline(override_props, current_prop, ';')) { |
| string other_prop; |
| if (current_prop.find("=") != string::npos) { |
| other_prop = current_prop.substr(current_prop.find("=") + 1); |
| current_prop = current_prop.substr(0, current_prop.find("=")); |
| } else { |
| other_prop = current_prop; |
| } |
| other_prop = android::base::Trim(other_prop); |
| current_prop = android::base::Trim(current_prop); |
| string sys_val = TWFunc::System_Property_Get(other_prop, PartitionManager, PartitionManager.Get_Android_Root_Path().c_str()); |
| if (!sys_val.empty()) { |
| LOGINFO("Overriding %s with value: \"%s\" from system property %s\n", current_prop.c_str(), sys_val.c_str(), other_prop.c_str()); |
| int error = TWFunc::Property_Override(current_prop, sys_val); |
| if (error) { |
| LOGERR("Failed overriding property %s, error_code: %d\n", current_prop.c_str(), error); |
| } |
| } else { |
| LOGINFO("Not overriding %s with empty value from system property %s\n", current_prop.c_str(), other_prop.c_str()); |
| } |
| } |
| PartitionManager.UnMount_By_Path(PartitionManager.Get_Android_Root_Path(), false); |
| } |
| #endif |
| |
| TWFunc::Update_Log_File(); |
| |
| twrpAdbBuFifo *adb_bu_fifo = new twrpAdbBuFifo(); |
| adb_bu_fifo->threadAdbBuFifo(); |
| |
| // Launch the main GUI |
| gui_start(); |
| |
| #ifndef TW_OEM_BUILD |
| // Disable flashing of stock recovery |
| TWFunc::Disable_Stock_Recovery_Replace(); |
| #endif |
| |
| // Reboot |
| TWFunc::Update_Intent_File(Send_Intent); |
| delete adb_bu_fifo; |
| TWFunc::Update_Log_File(); |
| gui_msg(Msg("rebooting=Rebooting...")); |
| string Reboot_Arg; |
| DataManager::GetValue("tw_reboot_arg", Reboot_Arg); |
| if (Reboot_Arg == "recovery") |
| TWFunc::tw_reboot(rb_recovery); |
| else if (Reboot_Arg == "poweroff") |
| TWFunc::tw_reboot(rb_poweroff); |
| else if (Reboot_Arg == "bootloader") |
| TWFunc::tw_reboot(rb_bootloader); |
| else if (Reboot_Arg == "download") |
| TWFunc::tw_reboot(rb_download); |
| else if (Reboot_Arg == "edl") |
| TWFunc::tw_reboot(rb_edl); |
| else |
| TWFunc::tw_reboot(rb_system); |
| |
| return 0; |
| } |
| |