blob: 5f743d40c1dcc83be4681285e7db38ab5b311b17 [file] [log] [blame]
Dees_Troya449a6f2013-04-07 17:50:11 -05001/*
bigbiffdf8436b2020-08-30 16:22:34 -04002 Copyright 2012-2020 TeamWin
3 This file is part of TWRP/TeamWin Recovery Project.
Dees_Troya449a6f2013-04-07 17:50:11 -05004
bigbiffdf8436b2020-08-30 16:22:34 -04005 TWRP is free software: you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation, either version 3 of the License, or
8 (at your option) any later version.
Dees_Troya449a6f2013-04-07 17:50:11 -05009
bigbiffdf8436b2020-08-30 16:22:34 -040010 TWRP is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
14
15 You should have received a copy of the GNU General Public License
16 along with TWRP. If not, see <http://www.gnu.org/licenses/>.
Dees_Troya449a6f2013-04-07 17:50:11 -050017*/
18
19#include <stdio.h>
20#include <stdlib.h>
Ethan Yonker74db1572015-10-28 12:44:49 -050021#include <string>
Dees_Troya449a6f2013-04-07 17:50:11 -050022#include <sys/stat.h>
23#include <sys/types.h>
24#include <time.h>
25#include <unistd.h>
Ethan Yonkerf9f99bc2014-12-29 09:10:34 -060026#include <signal.h>
Ethan Yonker74db1572015-10-28 12:44:49 -050027#include "gui/twmsg.h"
Dees_Troya449a6f2013-04-07 17:50:11 -050028
29#include "cutils/properties.h"
Dees_Troya449a6f2013-04-07 17:50:11 -050030
31#ifdef ANDROID_RB_RESTART
32#include "cutils/android_reboot.h"
33#else
34#include <sys/reboot.h>
35#endif
36
37extern "C" {
38#include "gui/gui.h"
39}
Ethan Yonkerf1179622016-08-25 15:32:21 -050040#include "set_metadata.h"
Ethan Yonker74db1572015-10-28 12:44:49 -050041#include "gui/gui.hpp"
42#include "gui/pages.hpp"
43#include "gui/objects.hpp"
Dees_Troya449a6f2013-04-07 17:50:11 -050044#include "twcommon.h"
45#include "twrp-functions.hpp"
46#include "data.hpp"
47#include "partitions.hpp"
Mohd Faraz0d9a62e2020-02-20 00:10:26 +053048#ifdef __ANDROID_API_N__
49#include <android-base/strings.h>
50#else
51#include <base/strings.h>
52#endif
Dees_Troya449a6f2013-04-07 17:50:11 -050053#include "openrecoveryscript.hpp"
54#include "variables.h"
bigbiffdf8436b2020-08-30 16:22:34 -040055#include "startupArgs.hpp"
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -040056#include "twrpAdbBuFifo.hpp"
Ethan Yonkerc798c9c2015-10-09 11:15:26 -050057#ifdef TW_USE_NEW_MINADBD
bigbiffd58ba182020-03-23 10:02:29 -040058// #include "minadbd/minadbd.h"
Ethan Yonkerc798c9c2015-10-09 11:15:26 -050059#else
60extern "C" {
Ethan Yonker84d61ce2017-05-10 16:11:35 -050061#include "minadbd21/adb.h"
Ethan Yonkerc798c9c2015-10-09 11:15:26 -050062}
63#endif
Dees_Troya449a6f2013-04-07 17:50:11 -050064
Ethan Yonker8373cfe2017-09-08 06:50:54 -050065//extern int adb_server_main(int is_daemon, int server_port, int /* reply_fd */);
Ethan Yonker534d4e02016-08-26 10:05:03 -050066
Dees_Troya449a6f2013-04-07 17:50:11 -050067TWPartitionManager PartitionManager;
68int Log_Offset;
Ethan Yonker6277c792014-09-15 14:54:30 -050069bool datamedia;
Dees_Troya449a6f2013-04-07 17:50:11 -050070
71static void Print_Prop(const char *key, const char *name, void *cookie) {
72 printf("%s=%s\n", key, name);
73}
74
bigbiff7ba75002020-04-11 20:47:09 -040075static void Decrypt_Page(bool SkipDecryption, bool datamedia) {
76 // Offer to decrypt if the device is encrypted
77 if (DataManager::GetIntValue(TW_IS_ENCRYPTED) != 0) {
78 if (SkipDecryption) {
79 LOGINFO("Skipping decryption\n");
Captain Throwback89b56a12021-02-10 18:25:07 -050080 PartitionManager.Update_System_Details();
Mohd Faraz815e2402020-09-19 19:31:52 +053081 } else if (DataManager::GetIntValue(TW_CRYPTO_PWTYPE) != 0) {
bigbiff7ba75002020-04-11 20:47:09 -040082 LOGINFO("Is encrypted, do decrypt page first\n");
Mohd Faraze3948ec2020-08-14 01:40:59 +000083 if (DataManager::GetIntValue(TW_IS_FBE))
84 DataManager::SetValue("tw_crypto_user_id", "0");
bigbiff7ba75002020-04-11 20:47:09 -040085 if (gui_startPage("decrypt", 1, 1) != 0) {
86 LOGERR("Failed to start decrypt GUI page.\n");
87 } else {
88 // Check for and load custom theme if present
89 TWFunc::check_selinux_support();
90 gui_loadCustomResources();
91 }
92 }
93 } else if (datamedia) {
Captain Throwback89b56a12021-02-10 18:25:07 -050094 PartitionManager.Update_System_Details();
bigbiff7ba75002020-04-11 20:47:09 -040095 TWFunc::check_selinux_support();
96 if (tw_get_default_metadata(DataManager::GetSettingsStoragePath().c_str()) != 0) {
97 LOGINFO("Failed to get default contexts and file mode for storage files.\n");
98 } else {
99 LOGINFO("Got default contexts and file mode for storage files.\n");
100 }
101 }
102}
103
bigbiffdf8436b2020-08-30 16:22:34 -0400104static void process_recovery_mode(twrpAdbBuFifo* adb_bu_fifo, bool skip_decryption) {
Vojtech Bocek0fc15732014-07-04 01:09:50 +0200105 char crash_prop_val[PROPERTY_VALUE_MAX];
106 int crash_counter;
107 property_get("twrp.crash_counter", crash_prop_val, "-1");
108 crash_counter = atoi(crash_prop_val) + 1;
109 snprintf(crash_prop_val, sizeof(crash_prop_val), "%d", crash_counter);
110 property_set("twrp.crash_counter", crash_prop_val);
111
bigbiffdf8436b2020-08-30 16:22:34 -0400112 if (crash_counter == 0) {
113 property_list(Print_Prop, NULL);
114 printf("\n");
115 } else {
116 printf("twrp.crash_counter=%d\n", crash_counter);
117 }
Dees_Troya449a6f2013-04-07 17:50:11 -0500118
Dees_Troya449a6f2013-04-07 17:50:11 -0500119 printf("=> Linking mtab\n");
120 symlink("/proc/mounts", "/etc/mtab");
Ethan Yonker53273ec2018-03-09 11:22:54 -0600121 std::string fstab_filename = "/etc/twrp.fstab";
122 if (!TWFunc::Path_Exists(fstab_filename)) {
123 fstab_filename = "/etc/recovery.fstab";
Dees_Troy329383e2013-08-29 14:16:06 +0000124 }
Ethan Yonker53273ec2018-03-09 11:22:54 -0600125 printf("=> Processing %s\n", fstab_filename.c_str());
bigbiff8da46fa2020-09-08 16:42:34 -0400126 if (!PartitionManager.Process_Fstab(fstab_filename, 1)) {
Ethan Yonker53273ec2018-03-09 11:22:54 -0600127 LOGERR("Failing out of recovery due to problem with fstab.\n");
bigbiffdf8436b2020-08-30 16:22:34 -0400128 return;
Dees_Troya449a6f2013-04-07 17:50:11 -0500129 }
bigbiffee7b7ff2020-03-23 15:08:27 -0400130
Captain Throwbackdc0823c2020-09-06 13:07:23 -0400131// We are doing this here to allow super partition to be set up prior to overriding properties
132#if defined(TW_INCLUDE_LIBRESETPROP) && defined(TW_OVERRIDE_SYSTEM_PROPS)
133 if (!PartitionManager.Mount_By_Path(PartitionManager.Get_Android_Root_Path(), true)) {
134 LOGERR("Unable to mount %s\n", PartitionManager.Get_Android_Root_Path().c_str());
135 } else {
136 stringstream override_props(EXPAND(TW_OVERRIDE_SYSTEM_PROPS));
137 string current_prop;
138 while (getline(override_props, current_prop, ';')) {
139 string other_prop;
140 if (current_prop.find("=") != string::npos) {
141 other_prop = current_prop.substr(current_prop.find("=") + 1);
142 current_prop = current_prop.substr(0, current_prop.find("="));
143 } else {
144 other_prop = current_prop;
145 }
146 other_prop = android::base::Trim(other_prop);
147 current_prop = android::base::Trim(current_prop);
148 string sys_val = TWFunc::System_Property_Get(other_prop, PartitionManager, PartitionManager.Get_Android_Root_Path().c_str());
149 if (!sys_val.empty()) {
150 LOGINFO("Overriding %s with value: \"%s\" from system property %s\n", current_prop.c_str(), sys_val.c_str(), other_prop.c_str());
151 int error = TWFunc::Property_Override(current_prop, sys_val);
152 if (error) {
153 LOGERR("Failed overriding property %s, error_code: %d\n", current_prop.c_str(), error);
154 }
155 } else {
156 LOGINFO("Not overriding %s with empty value from system property %s\n", current_prop.c_str(), other_prop.c_str());
157 }
158 }
159 PartitionManager.UnMount_By_Path(PartitionManager.Get_Android_Root_Path(), false);
160 }
161#endif
162
Dees_Troya449a6f2013-04-07 17:50:11 -0500163 // Check for and run startup script if script exists
bigbiffad58e1b2020-07-06 20:24:34 -0400164 TWFunc::check_and_run_script("/system/bin/runatboot.sh", "boot");
165 TWFunc::check_and_run_script("/system/bin/postrecoveryboot.sh", "boot");
Dees_Troya449a6f2013-04-07 17:50:11 -0500166
167#ifdef TW_INCLUDE_INJECTTWRP
168 // Back up TWRP Ramdisk if needed:
169 TWPartition* Boot = PartitionManager.Find_Partition_By_Path("/boot");
Dees_Troya449a6f2013-04-07 17:50:11 -0500170 LOGINFO("Backing up TWRP ramdisk...\n");
171 if (Boot == NULL || Boot->Current_File_System != "emmc")
Vojtech Bocek05534202013-09-11 08:11:56 +0200172 TWFunc::Exec_Cmd("injecttwrp --backup /tmp/backup_recovery_ramdisk.img");
Dees_Troya449a6f2013-04-07 17:50:11 -0500173 else {
174 string injectcmd = "injecttwrp --backup /tmp/backup_recovery_ramdisk.img bd=" + Boot->Actual_Block_Device;
Vojtech Bocek05534202013-09-11 08:11:56 +0200175 TWFunc::Exec_Cmd(injectcmd);
Dees_Troya449a6f2013-04-07 17:50:11 -0500176 }
177 LOGINFO("Backup of TWRP ramdisk done.\n");
178#endif
179
bigbiffdf8436b2020-08-30 16:22:34 -0400180 Decrypt_Page(skip_decryption, datamedia);
Captain Throwback89b56a12021-02-10 18:25:07 -0500181 PartitionManager.Output_Partition_Logging();
Dees_Troya449a6f2013-04-07 17:50:11 -0500182
Vojtech Bocek2005a842014-03-11 19:40:34 +0100183 // Fixup the RTC clock on devices which require it
bigbiffadc599e2020-05-28 19:36:30 +0000184 if (crash_counter == 0)
185 TWFunc::Fixup_Time_On_Boot();
Vojtech Bocek2005a842014-03-11 19:40:34 +0100186
epicXa721f952021-01-04 13:01:31 +0530187 DataManager::LoadTWRPFolderInfo();
Ian Macdonald1e0474a2020-09-09 02:43:13 +0200188 DataManager::ReadSettingsFile();
bigbiff bigbiff19874f12019-01-08 20:06:57 -0500189
Dees_Troya449a6f2013-04-07 17:50:11 -0500190 // Run any outstanding OpenRecoveryScript
bigbiff25d25b92020-06-19 16:07:38 -0400191 std::string cacheDir = TWFunc::get_log_dir();
192 if (cacheDir == DATA_LOGS_DIR)
193 cacheDir = "/data/cache";
bigbiff bigbiff19874f12019-01-08 20:06:57 -0500194 std::string orsFile = cacheDir + "/recovery/openrecoveryscript";
bigbiffdf8436b2020-08-30 16:22:34 -0400195 if ((DataManager::GetIntValue(TW_IS_ENCRYPTED) == 0 || skip_decryption) && (TWFunc::Path_Exists(SCRIPT_FILE_TMP) || TWFunc::Path_Exists(orsFile))) {
Dees_Troya449a6f2013-04-07 17:50:11 -0500196 OpenRecoveryScript::Run_OpenRecoveryScript();
197 }
Vojtech Bocekd0e38bc2014-02-03 23:36:57 +0100198
Ethan Yonker6f49e112014-09-03 21:42:49 -0500199#ifdef TW_HAS_MTP
Ethan Yonker6f49e112014-09-03 21:42:49 -0500200 char mtp_crash_check[PROPERTY_VALUE_MAX];
201 property_get("mtp.crash_check", mtp_crash_check, "0");
Matt Mower653a1702017-02-16 10:38:52 -0600202 if (DataManager::GetIntValue("tw_mtp_enabled")
203 && !strcmp(mtp_crash_check, "0") && !crash_counter
204 && (!DataManager::GetIntValue(TW_IS_ENCRYPTED) || DataManager::GetIntValue(TW_IS_DECRYPTED))) {
Ethan Yonker6f49e112014-09-03 21:42:49 -0500205 property_set("mtp.crash_check", "1");
Matt Mower653a1702017-02-16 10:38:52 -0600206 LOGINFO("Starting MTP\n");
207 if (!PartitionManager.Enable_MTP())
Ethan Yonkerdf7abac2014-12-29 10:48:17 -0600208 PartitionManager.Disable_MTP();
Matt Mower653a1702017-02-16 10:38:52 -0600209 else
210 gui_msg("mtp_enabled=MTP Enabled");
Ethan Yonker6f49e112014-09-03 21:42:49 -0500211 property_set("mtp.crash_check", "0");
Matt Mower653a1702017-02-16 10:38:52 -0600212 } else if (strcmp(mtp_crash_check, "0")) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500213 gui_warn("mtp_crash=MTP Crashed, not starting MTP on boot.");
Ethan Yonker12055fa2014-09-04 08:06:53 -0500214 DataManager::SetValue("tw_mtp_enabled", 0);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -0600215 PartitionManager.Disable_MTP();
Matt Mower653a1702017-02-16 10:38:52 -0600216 } else if (crash_counter == 1) {
217 LOGINFO("TWRP crashed; disabling MTP as a precaution.\n");
218 PartitionManager.Disable_MTP();
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400219 }
Ethan Yonker6f49e112014-09-03 21:42:49 -0500220#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400221
Ethan Yonker89583ef2015-08-26 09:01:59 -0500222#ifndef TW_OEM_BUILD
Ethan Yonkereb32b1f2015-05-18 10:23:03 -0500223 // Check if system has never been changed
Captain Throwback9d6feb52018-07-27 10:05:24 -0400224 TWPartition* sys = PartitionManager.Find_Partition_By_Path(PartitionManager.Get_Android_Root_Path());
Ethan Yonker1673e3d2015-10-26 21:51:58 -0500225 TWPartition* ven = PartitionManager.Find_Partition_By_Path("/vendor");
Ethan Yonker0e2c6572015-06-04 11:30:02 -0500226 if (sys) {
bigbiffee7b7ff2020-03-23 15:08:27 -0400227 if (sys->Get_Super_Status()) {
bigbiff7ba75002020-04-11 20:47:09 -0400228#ifdef TW_INCLUDE_CRYPTO
bigbiff25d25b92020-06-19 16:07:38 -0400229 std::string recoveryLogDir(DATA_LOGS_DIR);
230 recoveryLogDir += "/recovery";
231 if (!TWFunc::Path_Exists(recoveryLogDir)) {
232 bool created = PartitionManager.Recreate_Logs_Dir();
233 if (!created)
234 LOGERR("Unable to create log directory for TWRP\n");
235 }
bigbiff7ba75002020-04-11 20:47:09 -0400236 DataManager::ReadSettingsFile();
237#endif
bigbiffee7b7ff2020-03-23 15:08:27 -0400238 } else {
239 if ((DataManager::GetIntValue("tw_mount_system_ro") == 0 && sys->Check_Lifetime_Writes() == 0) || DataManager::GetIntValue("tw_mount_system_ro") == 2) {
240 if (DataManager::GetIntValue("tw_never_show_system_ro_page") == 0) {
241 DataManager::SetValue("tw_back", "main");
242 if (gui_startPage("system_readonly", 1, 1) != 0) {
243 LOGERR("Failed to start system_readonly GUI page.\n");
244 }
245 } else if (DataManager::GetIntValue("tw_mount_system_ro") == 0) {
246 sys->Change_Mount_Read_Only(false);
247 if (ven)
248 ven->Change_Mount_Read_Only(false);
Ethan Yonker0e2c6572015-06-04 11:30:02 -0500249 }
bigbiffee7b7ff2020-03-23 15:08:27 -0400250 } else if (DataManager::GetIntValue("tw_mount_system_ro") == 1) {
251 // Do nothing, user selected to leave system read only
252 } else {
Ethan Yonker0e2c6572015-06-04 11:30:02 -0500253 sys->Change_Mount_Read_Only(false);
Ethan Yonker1673e3d2015-10-26 21:51:58 -0500254 if (ven)
255 ven->Change_Mount_Read_Only(false);
Ethan Yonkereb32b1f2015-05-18 10:23:03 -0500256 }
Ethan Yonkereb32b1f2015-05-18 10:23:03 -0500257 }
258 }
Ethan Yonker89583ef2015-08-26 09:01:59 -0500259#endif
bigbiff7ba75002020-04-11 20:47:09 -0400260
bigbiffaac58612020-08-30 11:18:39 -0400261 TWFunc::Update_Log_File();
262
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -0400263 adb_bu_fifo->threadAdbBuFifo();
264
Ethan Yonker89583ef2015-08-26 09:01:59 -0500265#ifndef TW_OEM_BUILD
Ethan Yonker9132d912015-02-02 10:32:49 -0600266 // Disable flashing of stock recovery
267 TWFunc::Disable_Stock_Recovery_Replace();
Ethan Yonker89583ef2015-08-26 09:01:59 -0500268#endif
bigbiffdf8436b2020-08-30 16:22:34 -0400269}
Dees_Troya449a6f2013-04-07 17:50:11 -0500270
bigbiffdf8436b2020-08-30 16:22:34 -0400271static void reboot() {
Ethan Yonker74db1572015-10-28 12:44:49 -0500272 gui_msg(Msg("rebooting=Rebooting..."));
bigbiffdf8436b2020-08-30 16:22:34 -0400273 TWFunc::Update_Log_File();
Dees_Troya449a6f2013-04-07 17:50:11 -0500274 string Reboot_Arg;
bigbiffdf8436b2020-08-30 16:22:34 -0400275
Dees_Troya449a6f2013-04-07 17:50:11 -0500276 DataManager::GetValue("tw_reboot_arg", Reboot_Arg);
277 if (Reboot_Arg == "recovery")
278 TWFunc::tw_reboot(rb_recovery);
279 else if (Reboot_Arg == "poweroff")
280 TWFunc::tw_reboot(rb_poweroff);
281 else if (Reboot_Arg == "bootloader")
282 TWFunc::tw_reboot(rb_bootloader);
283 else if (Reboot_Arg == "download")
284 TWFunc::tw_reboot(rb_download);
mauronofrioe9a49ef2018-10-03 13:38:16 +0200285 else if (Reboot_Arg == "edl")
286 TWFunc::tw_reboot(rb_edl);
bigbiffdf8436b2020-08-30 16:22:34 -0400287 else if (Reboot_Arg == "fastboot")
288 TWFunc::tw_reboot(rb_fastboot);
Dees_Troya449a6f2013-04-07 17:50:11 -0500289 else
290 TWFunc::tw_reboot(rb_system);
bigbiffdf8436b2020-08-30 16:22:34 -0400291}
292
293int main(int argc, char **argv) {
294 // Recovery needs to install world-readable files, so clear umask
295 // set by init
296 umask(0);
297
298 Log_Offset = 0;
299
300 // Set up temporary log file (/tmp/recovery.log)
301 freopen(TMP_LOG_FILE, "a", stdout);
302 setbuf(stdout, NULL);
303 freopen(TMP_LOG_FILE, "a", stderr);
304 setbuf(stderr, NULL);
305
306 signal(SIGPIPE, SIG_IGN);
307
308 // Handle ADB sideload
309 if (argc == 3 && strcmp(argv[1], "--adbd") == 0) {
310 property_set("ctl.stop", "adbd");
311#ifdef TW_USE_NEW_MINADBD
312 //adb_server_main(0, DEFAULT_ADB_PORT, -1); TODO fix this for android8
313 // minadbd_main();
314#else
315 adb_main(argv[2]);
316#endif
317 return 0;
318 }
319
320#ifdef RECOVERY_SDCARD_ON_DATA
321 datamedia = true;
322#endif
323
324 property_set("ro.twrp.boot", "1");
325 property_set("ro.twrp.version", TW_VERSION_STR);
326
327 time_t StartupTime = time(NULL);
328 printf("Starting TWRP %s-%s on %s (pid %d)\n", TW_VERSION_STR, TW_GIT_REVISION, ctime(&StartupTime), getpid());
329
330 // Load default values to set DataManager constants and handle ifdefs
331 DataManager::SetDefaultValues();
332 printf("Starting the UI...\n");
333 gui_init();
334
335 // Load up all the resources
336 gui_loadResources();
337
bigbiffdf8436b2020-08-30 16:22:34 -0400338 startupArgs startup;
339 startup.parse(&argc, &argv);
340 twrpAdbBuFifo *adb_bu_fifo = new twrpAdbBuFifo();
341 TWFunc::Clear_Bootloader_Message();
342
343 if (startup.Get_Fastboot_Mode()) {
344 LOGINFO("starting fastboot\n");
345 gui_msg(Msg("fastboot_console_msg=Entered Fastboot mode..."));
346 if (gui_startPage("fastboot", 1, 1) != 0) {
347 LOGERR("Failed to start fastbootd page.\n");
348 }
349 delete adb_bu_fifo;
350 TWFunc::Update_Intent_File(startup.Get_Intent());
351
352 reboot();
353 return 0;
354 } else {
355 process_recovery_mode(adb_bu_fifo, startup.Should_Skip_Decryption());
356 }
357
Ian Macdonalda9910222020-10-27 18:02:21 +0100358 PageManager::LoadLanguage(DataManager::GetStrValue("tw_language"));
359 GUIConsole::Translate_Now();
360
epicX271bb3a2020-12-30 01:03:18 +0530361 TWFunc::checkforapp(); //Checking compatibility for TWRP app
362
bigbiffdf8436b2020-08-30 16:22:34 -0400363 // Launch the main GUI
364 gui_start();
365 delete adb_bu_fifo;
366 TWFunc::Update_Intent_File(startup.Get_Intent());
367
368 reboot();
Dees_Troya449a6f2013-04-07 17:50:11 -0500369
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200370 return 0;
Dees_Troya449a6f2013-04-07 17:50:11 -0500371}