blob: 7ec428cbdaf279e4c86549b29206d16480fa8990 [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");
Mohd Faraz815e2402020-09-19 19:31:52 +053080 } else if (DataManager::GetIntValue(TW_CRYPTO_PWTYPE) != 0) {
bigbiff7ba75002020-04-11 20:47:09 -040081 LOGINFO("Is encrypted, do decrypt page first\n");
Mohd Faraze3948ec2020-08-14 01:40:59 +000082 if (DataManager::GetIntValue(TW_IS_FBE))
83 DataManager::SetValue("tw_crypto_user_id", "0");
bigbiff7ba75002020-04-11 20:47:09 -040084 if (gui_startPage("decrypt", 1, 1) != 0) {
85 LOGERR("Failed to start decrypt GUI page.\n");
86 } else {
87 // Check for and load custom theme if present
88 TWFunc::check_selinux_support();
89 gui_loadCustomResources();
90 }
91 }
92 } else if (datamedia) {
93 TWFunc::check_selinux_support();
94 if (tw_get_default_metadata(DataManager::GetSettingsStoragePath().c_str()) != 0) {
95 LOGINFO("Failed to get default contexts and file mode for storage files.\n");
96 } else {
97 LOGINFO("Got default contexts and file mode for storage files.\n");
98 }
99 }
100}
101
bigbiffdf8436b2020-08-30 16:22:34 -0400102static void process_recovery_mode(twrpAdbBuFifo* adb_bu_fifo, bool skip_decryption) {
Vojtech Bocek0fc15732014-07-04 01:09:50 +0200103 char crash_prop_val[PROPERTY_VALUE_MAX];
104 int crash_counter;
105 property_get("twrp.crash_counter", crash_prop_val, "-1");
106 crash_counter = atoi(crash_prop_val) + 1;
107 snprintf(crash_prop_val, sizeof(crash_prop_val), "%d", crash_counter);
108 property_set("twrp.crash_counter", crash_prop_val);
109
bigbiffdf8436b2020-08-30 16:22:34 -0400110 if (crash_counter == 0) {
111 property_list(Print_Prop, NULL);
112 printf("\n");
113 } else {
114 printf("twrp.crash_counter=%d\n", crash_counter);
115 }
Dees_Troya449a6f2013-04-07 17:50:11 -0500116
Dees_Troya449a6f2013-04-07 17:50:11 -0500117 printf("=> Linking mtab\n");
118 symlink("/proc/mounts", "/etc/mtab");
Ethan Yonker53273ec2018-03-09 11:22:54 -0600119 std::string fstab_filename = "/etc/twrp.fstab";
120 if (!TWFunc::Path_Exists(fstab_filename)) {
121 fstab_filename = "/etc/recovery.fstab";
Dees_Troy329383e2013-08-29 14:16:06 +0000122 }
Ethan Yonker53273ec2018-03-09 11:22:54 -0600123 printf("=> Processing %s\n", fstab_filename.c_str());
bigbiff8da46fa2020-09-08 16:42:34 -0400124 if (!PartitionManager.Process_Fstab(fstab_filename, 1)) {
Ethan Yonker53273ec2018-03-09 11:22:54 -0600125 LOGERR("Failing out of recovery due to problem with fstab.\n");
bigbiffdf8436b2020-08-30 16:22:34 -0400126 return;
Dees_Troya449a6f2013-04-07 17:50:11 -0500127 }
128 PartitionManager.Output_Partition_Logging();
bigbiffee7b7ff2020-03-23 15:08:27 -0400129
Captain Throwbackdc0823c2020-09-06 13:07:23 -0400130// We are doing this here to allow super partition to be set up prior to overriding properties
131#if defined(TW_INCLUDE_LIBRESETPROP) && defined(TW_OVERRIDE_SYSTEM_PROPS)
132 if (!PartitionManager.Mount_By_Path(PartitionManager.Get_Android_Root_Path(), true)) {
133 LOGERR("Unable to mount %s\n", PartitionManager.Get_Android_Root_Path().c_str());
134 } else {
135 stringstream override_props(EXPAND(TW_OVERRIDE_SYSTEM_PROPS));
136 string current_prop;
137 while (getline(override_props, current_prop, ';')) {
138 string other_prop;
139 if (current_prop.find("=") != string::npos) {
140 other_prop = current_prop.substr(current_prop.find("=") + 1);
141 current_prop = current_prop.substr(0, current_prop.find("="));
142 } else {
143 other_prop = current_prop;
144 }
145 other_prop = android::base::Trim(other_prop);
146 current_prop = android::base::Trim(current_prop);
147 string sys_val = TWFunc::System_Property_Get(other_prop, PartitionManager, PartitionManager.Get_Android_Root_Path().c_str());
148 if (!sys_val.empty()) {
149 LOGINFO("Overriding %s with value: \"%s\" from system property %s\n", current_prop.c_str(), sys_val.c_str(), other_prop.c_str());
150 int error = TWFunc::Property_Override(current_prop, sys_val);
151 if (error) {
152 LOGERR("Failed overriding property %s, error_code: %d\n", current_prop.c_str(), error);
153 }
154 } else {
155 LOGINFO("Not overriding %s with empty value from system property %s\n", current_prop.c_str(), other_prop.c_str());
156 }
157 }
158 PartitionManager.UnMount_By_Path(PartitionManager.Get_Android_Root_Path(), false);
159 }
160#endif
161
Dees_Troya449a6f2013-04-07 17:50:11 -0500162 // Check for and run startup script if script exists
bigbiffad58e1b2020-07-06 20:24:34 -0400163 TWFunc::check_and_run_script("/system/bin/runatboot.sh", "boot");
164 TWFunc::check_and_run_script("/system/bin/postrecoveryboot.sh", "boot");
Dees_Troya449a6f2013-04-07 17:50:11 -0500165
166#ifdef TW_INCLUDE_INJECTTWRP
167 // Back up TWRP Ramdisk if needed:
168 TWPartition* Boot = PartitionManager.Find_Partition_By_Path("/boot");
Dees_Troya449a6f2013-04-07 17:50:11 -0500169 LOGINFO("Backing up TWRP ramdisk...\n");
170 if (Boot == NULL || Boot->Current_File_System != "emmc")
Vojtech Bocek05534202013-09-11 08:11:56 +0200171 TWFunc::Exec_Cmd("injecttwrp --backup /tmp/backup_recovery_ramdisk.img");
Dees_Troya449a6f2013-04-07 17:50:11 -0500172 else {
173 string injectcmd = "injecttwrp --backup /tmp/backup_recovery_ramdisk.img bd=" + Boot->Actual_Block_Device;
Vojtech Bocek05534202013-09-11 08:11:56 +0200174 TWFunc::Exec_Cmd(injectcmd);
Dees_Troya449a6f2013-04-07 17:50:11 -0500175 }
176 LOGINFO("Backup of TWRP ramdisk done.\n");
177#endif
178
bigbiffdf8436b2020-08-30 16:22:34 -0400179 Decrypt_Page(skip_decryption, datamedia);
Dees_Troya449a6f2013-04-07 17:50:11 -0500180
Vojtech Bocek2005a842014-03-11 19:40:34 +0100181 // Fixup the RTC clock on devices which require it
bigbiffadc599e2020-05-28 19:36:30 +0000182 if (crash_counter == 0)
183 TWFunc::Fixup_Time_On_Boot();
Vojtech Bocek2005a842014-03-11 19:40:34 +0100184
Ian Macdonald1e0474a2020-09-09 02:43:13 +0200185 DataManager::ReadSettingsFile();
bigbiff bigbiff19874f12019-01-08 20:06:57 -0500186
Dees_Troya449a6f2013-04-07 17:50:11 -0500187 // Run any outstanding OpenRecoveryScript
bigbiff25d25b92020-06-19 16:07:38 -0400188 std::string cacheDir = TWFunc::get_log_dir();
189 if (cacheDir == DATA_LOGS_DIR)
190 cacheDir = "/data/cache";
bigbiff bigbiff19874f12019-01-08 20:06:57 -0500191 std::string orsFile = cacheDir + "/recovery/openrecoveryscript";
bigbiffdf8436b2020-08-30 16:22:34 -0400192 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 -0500193 OpenRecoveryScript::Run_OpenRecoveryScript();
194 }
Vojtech Bocekd0e38bc2014-02-03 23:36:57 +0100195
Ethan Yonker6f49e112014-09-03 21:42:49 -0500196#ifdef TW_HAS_MTP
Ethan Yonker6f49e112014-09-03 21:42:49 -0500197 char mtp_crash_check[PROPERTY_VALUE_MAX];
198 property_get("mtp.crash_check", mtp_crash_check, "0");
Matt Mower653a1702017-02-16 10:38:52 -0600199 if (DataManager::GetIntValue("tw_mtp_enabled")
200 && !strcmp(mtp_crash_check, "0") && !crash_counter
201 && (!DataManager::GetIntValue(TW_IS_ENCRYPTED) || DataManager::GetIntValue(TW_IS_DECRYPTED))) {
Ethan Yonker6f49e112014-09-03 21:42:49 -0500202 property_set("mtp.crash_check", "1");
Matt Mower653a1702017-02-16 10:38:52 -0600203 LOGINFO("Starting MTP\n");
204 if (!PartitionManager.Enable_MTP())
Ethan Yonkerdf7abac2014-12-29 10:48:17 -0600205 PartitionManager.Disable_MTP();
Matt Mower653a1702017-02-16 10:38:52 -0600206 else
207 gui_msg("mtp_enabled=MTP Enabled");
Ethan Yonker6f49e112014-09-03 21:42:49 -0500208 property_set("mtp.crash_check", "0");
Matt Mower653a1702017-02-16 10:38:52 -0600209 } else if (strcmp(mtp_crash_check, "0")) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500210 gui_warn("mtp_crash=MTP Crashed, not starting MTP on boot.");
Ethan Yonker12055fa2014-09-04 08:06:53 -0500211 DataManager::SetValue("tw_mtp_enabled", 0);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -0600212 PartitionManager.Disable_MTP();
Matt Mower653a1702017-02-16 10:38:52 -0600213 } else if (crash_counter == 1) {
214 LOGINFO("TWRP crashed; disabling MTP as a precaution.\n");
215 PartitionManager.Disable_MTP();
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400216 }
Ethan Yonker6f49e112014-09-03 21:42:49 -0500217#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400218
Ethan Yonker89583ef2015-08-26 09:01:59 -0500219#ifndef TW_OEM_BUILD
Ethan Yonkereb32b1f2015-05-18 10:23:03 -0500220 // Check if system has never been changed
Captain Throwback9d6feb52018-07-27 10:05:24 -0400221 TWPartition* sys = PartitionManager.Find_Partition_By_Path(PartitionManager.Get_Android_Root_Path());
Ethan Yonker1673e3d2015-10-26 21:51:58 -0500222 TWPartition* ven = PartitionManager.Find_Partition_By_Path("/vendor");
Ethan Yonker0e2c6572015-06-04 11:30:02 -0500223 if (sys) {
bigbiffee7b7ff2020-03-23 15:08:27 -0400224 if (sys->Get_Super_Status()) {
bigbiff7ba75002020-04-11 20:47:09 -0400225#ifdef TW_INCLUDE_CRYPTO
bigbiff25d25b92020-06-19 16:07:38 -0400226 std::string recoveryLogDir(DATA_LOGS_DIR);
227 recoveryLogDir += "/recovery";
228 if (!TWFunc::Path_Exists(recoveryLogDir)) {
229 bool created = PartitionManager.Recreate_Logs_Dir();
230 if (!created)
231 LOGERR("Unable to create log directory for TWRP\n");
232 }
bigbiff7ba75002020-04-11 20:47:09 -0400233 DataManager::ReadSettingsFile();
234#endif
bigbiffee7b7ff2020-03-23 15:08:27 -0400235 } else {
236 if ((DataManager::GetIntValue("tw_mount_system_ro") == 0 && sys->Check_Lifetime_Writes() == 0) || DataManager::GetIntValue("tw_mount_system_ro") == 2) {
237 if (DataManager::GetIntValue("tw_never_show_system_ro_page") == 0) {
238 DataManager::SetValue("tw_back", "main");
239 if (gui_startPage("system_readonly", 1, 1) != 0) {
240 LOGERR("Failed to start system_readonly GUI page.\n");
241 }
242 } else if (DataManager::GetIntValue("tw_mount_system_ro") == 0) {
243 sys->Change_Mount_Read_Only(false);
244 if (ven)
245 ven->Change_Mount_Read_Only(false);
Ethan Yonker0e2c6572015-06-04 11:30:02 -0500246 }
bigbiffee7b7ff2020-03-23 15:08:27 -0400247 } else if (DataManager::GetIntValue("tw_mount_system_ro") == 1) {
248 // Do nothing, user selected to leave system read only
249 } else {
Ethan Yonker0e2c6572015-06-04 11:30:02 -0500250 sys->Change_Mount_Read_Only(false);
Ethan Yonker1673e3d2015-10-26 21:51:58 -0500251 if (ven)
252 ven->Change_Mount_Read_Only(false);
Ethan Yonkereb32b1f2015-05-18 10:23:03 -0500253 }
Ethan Yonkereb32b1f2015-05-18 10:23:03 -0500254 }
255 }
Ethan Yonker89583ef2015-08-26 09:01:59 -0500256#endif
bigbiff7ba75002020-04-11 20:47:09 -0400257
bigbiffaac58612020-08-30 11:18:39 -0400258 TWFunc::Update_Log_File();
259
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -0400260 adb_bu_fifo->threadAdbBuFifo();
261
Ethan Yonker89583ef2015-08-26 09:01:59 -0500262#ifndef TW_OEM_BUILD
Ethan Yonker9132d912015-02-02 10:32:49 -0600263 // Disable flashing of stock recovery
264 TWFunc::Disable_Stock_Recovery_Replace();
Ethan Yonker89583ef2015-08-26 09:01:59 -0500265#endif
bigbiffdf8436b2020-08-30 16:22:34 -0400266}
Dees_Troya449a6f2013-04-07 17:50:11 -0500267
bigbiffdf8436b2020-08-30 16:22:34 -0400268static void reboot() {
Ethan Yonker74db1572015-10-28 12:44:49 -0500269 gui_msg(Msg("rebooting=Rebooting..."));
bigbiffdf8436b2020-08-30 16:22:34 -0400270 TWFunc::Update_Log_File();
Dees_Troya449a6f2013-04-07 17:50:11 -0500271 string Reboot_Arg;
bigbiffdf8436b2020-08-30 16:22:34 -0400272
Dees_Troya449a6f2013-04-07 17:50:11 -0500273 DataManager::GetValue("tw_reboot_arg", Reboot_Arg);
274 if (Reboot_Arg == "recovery")
275 TWFunc::tw_reboot(rb_recovery);
276 else if (Reboot_Arg == "poweroff")
277 TWFunc::tw_reboot(rb_poweroff);
278 else if (Reboot_Arg == "bootloader")
279 TWFunc::tw_reboot(rb_bootloader);
280 else if (Reboot_Arg == "download")
281 TWFunc::tw_reboot(rb_download);
mauronofrioe9a49ef2018-10-03 13:38:16 +0200282 else if (Reboot_Arg == "edl")
283 TWFunc::tw_reboot(rb_edl);
bigbiffdf8436b2020-08-30 16:22:34 -0400284 else if (Reboot_Arg == "fastboot")
285 TWFunc::tw_reboot(rb_fastboot);
Dees_Troya449a6f2013-04-07 17:50:11 -0500286 else
287 TWFunc::tw_reboot(rb_system);
bigbiffdf8436b2020-08-30 16:22:34 -0400288}
289
290int main(int argc, char **argv) {
291 // Recovery needs to install world-readable files, so clear umask
292 // set by init
293 umask(0);
294
295 Log_Offset = 0;
296
297 // Set up temporary log file (/tmp/recovery.log)
298 freopen(TMP_LOG_FILE, "a", stdout);
299 setbuf(stdout, NULL);
300 freopen(TMP_LOG_FILE, "a", stderr);
301 setbuf(stderr, NULL);
302
303 signal(SIGPIPE, SIG_IGN);
304
305 // Handle ADB sideload
306 if (argc == 3 && strcmp(argv[1], "--adbd") == 0) {
307 property_set("ctl.stop", "adbd");
308#ifdef TW_USE_NEW_MINADBD
309 //adb_server_main(0, DEFAULT_ADB_PORT, -1); TODO fix this for android8
310 // minadbd_main();
311#else
312 adb_main(argv[2]);
313#endif
314 return 0;
315 }
316
317#ifdef RECOVERY_SDCARD_ON_DATA
318 datamedia = true;
319#endif
320
321 property_set("ro.twrp.boot", "1");
322 property_set("ro.twrp.version", TW_VERSION_STR);
323
324 time_t StartupTime = time(NULL);
325 printf("Starting TWRP %s-%s on %s (pid %d)\n", TW_VERSION_STR, TW_GIT_REVISION, ctime(&StartupTime), getpid());
326
327 // Load default values to set DataManager constants and handle ifdefs
328 DataManager::SetDefaultValues();
329 printf("Starting the UI...\n");
330 gui_init();
331
332 // Load up all the resources
333 gui_loadResources();
334
335 PageManager::LoadLanguage(DataManager::GetStrValue("tw_language"));
336 GUIConsole::Translate_Now();
337
338 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
358 // Launch the main GUI
359 gui_start();
360 delete adb_bu_fifo;
361 TWFunc::Update_Intent_File(startup.Get_Intent());
362
363 reboot();
Dees_Troya449a6f2013-04-07 17:50:11 -0500364
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200365 return 0;
Dees_Troya449a6f2013-04-07 17:50:11 -0500366}