blob: 3489f8cfc05b7991eb39a27523b5956ccfc85082 [file] [log] [blame]
Dees_Troya449a6f2013-04-07 17:50:11 -05001/*
bigbiff bigbiff34684ff2013-12-01 21:03:45 -05002 TWRP is free software: you can redistribute it and/or modify
3 it under the terms of the GNU General Public License as published by
4 the Free Software Foundation, either version 3 of the License, or
5 (at your option) any later version.
Dees_Troya449a6f2013-04-07 17:50:11 -05006
bigbiff bigbiff34684ff2013-12-01 21:03:45 -05007 TWRP is distributed in the hope that it will be useful,
8 but WITHOUT ANY WARRANTY; without even the implied warranty of
9 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
10 GNU General Public License for more details.
Dees_Troya449a6f2013-04-07 17:50:11 -050011
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050012 You should have received a copy of the GNU General Public License
13 along with TWRP. If not, see <http://www.gnu.org/licenses/>.
Dees_Troya449a6f2013-04-07 17:50:11 -050014*/
15
16#include <stdio.h>
17#include <stdlib.h>
18#include <string.h>
19#include <sys/stat.h>
20#include <sys/types.h>
21#include <time.h>
22#include <unistd.h>
Ethan Yonkerf9f99bc2014-12-29 09:10:34 -060023#include <signal.h>
Dees_Troya449a6f2013-04-07 17:50:11 -050024
25#include "cutils/properties.h"
26extern "C" {
27#include "minadbd/adb.h"
28#include "bootloader.h"
29}
30
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"
Ethan Yonker4b94cfd2014-12-11 10:00:45 -060039#include "set_metadata.h"
Dees_Troya449a6f2013-04-07 17:50:11 -050040}
41#include "twcommon.h"
42#include "twrp-functions.hpp"
43#include "data.hpp"
44#include "partitions.hpp"
45#include "openrecoveryscript.hpp"
46#include "variables.h"
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050047#include "twrpDU.hpp"
Dees_Troya449a6f2013-04-07 17:50:11 -050048
Dees_Troya95f55c2013-08-17 13:14:43 +000049#ifdef HAVE_SELINUX
50#include "selinux/label.h"
51struct selabel_handle *selinux_handle;
52#endif
53
Dees_Troya449a6f2013-04-07 17:50:11 -050054TWPartitionManager PartitionManager;
55int Log_Offset;
Ethan Yonker6277c792014-09-15 14:54:30 -050056bool datamedia;
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050057twrpDU du;
Dees_Troya449a6f2013-04-07 17:50:11 -050058
59static void Print_Prop(const char *key, const char *name, void *cookie) {
60 printf("%s=%s\n", key, name);
61}
62
63int main(int argc, char **argv) {
64 // Recovery needs to install world-readable files, so clear umask
65 // set by init
66 umask(0);
67
68 Log_Offset = 0;
69
70 // Set up temporary log file (/tmp/recovery.log)
71 freopen(TMP_LOG_FILE, "a", stdout);
72 setbuf(stdout, NULL);
73 freopen(TMP_LOG_FILE, "a", stderr);
74 setbuf(stderr, NULL);
75
Ethan Yonkerf9f99bc2014-12-29 09:10:34 -060076 signal(SIGPIPE, SIG_IGN);
77
Dees_Troya449a6f2013-04-07 17:50:11 -050078 // Handle ADB sideload
79 if (argc == 3 && strcmp(argv[1], "--adbd") == 0) {
thatcc8ddca2015-01-03 01:59:36 +010080 property_set("ctl.stop", "adbd");
Dees_Troya449a6f2013-04-07 17:50:11 -050081 adb_main(argv[2]);
82 return 0;
83 }
84
Ethan Yonker6277c792014-09-15 14:54:30 -050085#ifdef RECOVERY_SDCARD_ON_DATA
86 datamedia = true;
87#endif
88
Vojtech Bocek0fc15732014-07-04 01:09:50 +020089 char crash_prop_val[PROPERTY_VALUE_MAX];
90 int crash_counter;
91 property_get("twrp.crash_counter", crash_prop_val, "-1");
92 crash_counter = atoi(crash_prop_val) + 1;
93 snprintf(crash_prop_val, sizeof(crash_prop_val), "%d", crash_counter);
94 property_set("twrp.crash_counter", crash_prop_val);
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040095 property_set("ro.twrp.boot", "1");
96 property_set("ro.twrp.version", TW_VERSION_STR);
Vojtech Bocek0fc15732014-07-04 01:09:50 +020097
Dees_Troya449a6f2013-04-07 17:50:11 -050098 time_t StartupTime = time(NULL);
thatcc8ddca2015-01-03 01:59:36 +010099 printf("Starting TWRP %s on %s (pid %d)\n", TW_VERSION_STR, ctime(&StartupTime), getpid());
Dees_Troya449a6f2013-04-07 17:50:11 -0500100
101 // Load default values to set DataManager constants and handle ifdefs
102 DataManager::SetDefaultValues();
103 printf("Starting the UI...");
104 gui_init();
105 printf("=> Linking mtab\n");
106 symlink("/proc/mounts", "/etc/mtab");
Dees_Troy329383e2013-08-29 14:16:06 +0000107 if (TWFunc::Path_Exists("/etc/twrp.fstab")) {
108 if (TWFunc::Path_Exists("/etc/recovery.fstab")) {
109 printf("Renaming regular /etc/recovery.fstab -> /etc/recovery.fstab.bak\n");
110 rename("/etc/recovery.fstab", "/etc/recovery.fstab.bak");
111 }
112 printf("Moving /etc/twrp.fstab -> /etc/recovery.fstab\n");
113 rename("/etc/twrp.fstab", "/etc/recovery.fstab");
114 }
Dees_Troya449a6f2013-04-07 17:50:11 -0500115 printf("=> Processing recovery.fstab\n");
116 if (!PartitionManager.Process_Fstab("/etc/recovery.fstab", 1)) {
117 LOGERR("Failing out of recovery due to problem with recovery.fstab.\n");
118 return -1;
119 }
120 PartitionManager.Output_Partition_Logging();
121 // Load up all the resources
122 gui_loadResources();
123
Dees_Troya95f55c2013-08-17 13:14:43 +0000124#ifdef HAVE_SELINUX
bigbiff bigbiffc49d7062013-10-11 20:28:00 -0400125 if (TWFunc::Path_Exists("/prebuilt_file_contexts")) {
126 if (TWFunc::Path_Exists("/file_contexts")) {
127 printf("Renaming regular /file_contexts -> /file_contexts.bak\n");
128 rename("/file_contexts", "/file_contexts.bak");
129 }
130 printf("Moving /prebuilt_file_contexts -> /file_contexts\n");
131 rename("/prebuilt_file_contexts", "/file_contexts");
132 }
Dees_Troya95f55c2013-08-17 13:14:43 +0000133 struct selinux_opt selinux_options[] = {
134 { SELABEL_OPT_PATH, "/file_contexts" }
135 };
bigbiff bigbiffc49d7062013-10-11 20:28:00 -0400136 selinux_handle = selabel_open(SELABEL_CTX_FILE, selinux_options, 1);
Dees_Troya95f55c2013-08-17 13:14:43 +0000137 if (!selinux_handle)
138 printf("No file contexts for SELinux\n");
139 else
140 printf("SELinux contexts loaded from /file_contexts\n");
Dees Troy995e88c2013-11-26 21:39:14 +0000141 { // Check to ensure SELinux can be supported by the kernel
142 char *contexts = NULL;
Dees Troy8d0eb132013-12-06 16:55:41 +0000143
144 if (PartitionManager.Mount_By_Path("/cache", true) && TWFunc::Path_Exists("/cache/recovery")) {
145 lgetfilecon("/cache/recovery", &contexts);
146 if (!contexts) {
147 lsetfilecon("/cache/recovery", "test");
148 lgetfilecon("/cache/recovery", &contexts);
149 }
150 } else {
151 LOGINFO("Could not check /cache/recovery SELinux contexts, using /sbin/teamwin instead which may be inaccurate.\n");
152 lgetfilecon("/sbin/teamwin", &contexts);
153 }
Dees Troy995e88c2013-11-26 21:39:14 +0000154 if (!contexts) {
Ethan Yonkerbf2cb1c2014-07-02 10:15:54 -0500155 gui_print_color("warning", "Kernel does not have support for reading SELinux contexts.\n");
Dees Troy995e88c2013-11-26 21:39:14 +0000156 } else {
157 free(contexts);
158 gui_print("Full SELinux support is present.\n");
159 }
160 }
161#else
Ethan Yonkerbf2cb1c2014-07-02 10:15:54 -0500162 gui_print_color("warning", "No SELinux support (no libselinux).\n");
Dees_Troya95f55c2013-08-17 13:14:43 +0000163#endif
164
Dees_Troya449a6f2013-04-07 17:50:11 -0500165 PartitionManager.Mount_By_Path("/cache", true);
166
167 string Zip_File, Reboot_Value;
Ethan Yonkera1674162014-11-06 08:35:10 -0600168 bool Cache_Wipe = false, Factory_Reset = false, Perform_Backup = false, Shutdown = false;
Dees_Troya449a6f2013-04-07 17:50:11 -0500169
170 {
Dees_Troy1669f892013-09-04 18:35:08 +0000171 TWPartition* misc = PartitionManager.Find_Partition_By_Path("/misc");
172 if (misc != NULL) {
173 if (misc->Current_File_System == "emmc") {
174 set_device_type('e');
175 set_device_name(misc->Actual_Block_Device.c_str());
176 } else if (misc->Current_File_System == "mtd") {
177 set_device_type('m');
178 set_device_name(misc->MTD_Name.c_str());
179 } else {
180 LOGERR("Unknown file system for /misc\n");
181 }
182 }
Dees_Troya449a6f2013-04-07 17:50:11 -0500183 get_args(&argc, &argv);
184
185 int index, index2, len;
186 char* argptr;
187 char* ptr;
188 printf("Startup Commands: ");
189 for (index = 1; index < argc; index++) {
190 argptr = argv[index];
191 printf(" '%s'", argv[index]);
192 len = strlen(argv[index]);
193 if (*argptr == '-') {argptr++; len--;}
194 if (*argptr == '-') {argptr++; len--;}
195 if (*argptr == 'u') {
196 ptr = argptr;
197 index2 = 0;
198 while (*ptr != '=' && *ptr != '\n')
199 ptr++;
Kevin Steck7b69e9d2013-10-17 18:24:31 -0400200 // skip the = before grabbing Zip_File
201 while (*ptr == '=')
202 ptr++;
Dees_Troya449a6f2013-04-07 17:50:11 -0500203 if (*ptr) {
204 Zip_File = ptr;
205 } else
206 LOGERR("argument error specifying zip file\n");
207 } else if (*argptr == 'w') {
208 if (len == 9)
209 Factory_Reset = true;
210 else if (len == 10)
211 Cache_Wipe = true;
212 } else if (*argptr == 'n') {
213 Perform_Backup = true;
Ethan Yonkera1674162014-11-06 08:35:10 -0600214 } else if (*argptr == 'p') {
215 Shutdown = true;
Dees_Troya449a6f2013-04-07 17:50:11 -0500216 } else if (*argptr == 's') {
217 ptr = argptr;
218 index2 = 0;
219 while (*ptr != '=' && *ptr != '\n')
220 ptr++;
221 if (*ptr) {
222 Reboot_Value = *ptr;
223 }
224 }
225 }
Vojtech Bocek0fc15732014-07-04 01:09:50 +0200226 printf("\n");
Dees_Troya449a6f2013-04-07 17:50:11 -0500227 }
228
Vojtech Bocek0fc15732014-07-04 01:09:50 +0200229 if(crash_counter == 0) {
Dees_Troya449a6f2013-04-07 17:50:11 -0500230 property_list(Print_Prop, NULL);
231 printf("\n");
Vojtech Bocek0fc15732014-07-04 01:09:50 +0200232 } else {
233 printf("twrp.crash_counter=%d\n", crash_counter);
Dees_Troya449a6f2013-04-07 17:50:11 -0500234 }
235
236 // Check for and run startup script if script exists
237 TWFunc::check_and_run_script("/sbin/runatboot.sh", "boot");
238 TWFunc::check_and_run_script("/sbin/postrecoveryboot.sh", "boot");
239
240#ifdef TW_INCLUDE_INJECTTWRP
241 // Back up TWRP Ramdisk if needed:
242 TWPartition* Boot = PartitionManager.Find_Partition_By_Path("/boot");
Dees_Troya449a6f2013-04-07 17:50:11 -0500243 LOGINFO("Backing up TWRP ramdisk...\n");
244 if (Boot == NULL || Boot->Current_File_System != "emmc")
Vojtech Bocek05534202013-09-11 08:11:56 +0200245 TWFunc::Exec_Cmd("injecttwrp --backup /tmp/backup_recovery_ramdisk.img");
Dees_Troya449a6f2013-04-07 17:50:11 -0500246 else {
247 string injectcmd = "injecttwrp --backup /tmp/backup_recovery_ramdisk.img bd=" + Boot->Actual_Block_Device;
Vojtech Bocek05534202013-09-11 08:11:56 +0200248 TWFunc::Exec_Cmd(injectcmd);
Dees_Troya449a6f2013-04-07 17:50:11 -0500249 }
250 LOGINFO("Backup of TWRP ramdisk done.\n");
251#endif
252
253 bool Keep_Going = true;
254 if (Perform_Backup) {
Dees Troyb21cc642013-09-10 17:36:41 +0000255 DataManager::SetValue(TW_BACKUP_NAME, "(Auto Generate)");
Dees_Troya449a6f2013-04-07 17:50:11 -0500256 if (!OpenRecoveryScript::Insert_ORS_Command("backup BSDCAE\n"))
257 Keep_Going = false;
258 }
259 if (Keep_Going && !Zip_File.empty()) {
260 string ORSCommand = "install " + Zip_File;
261
262 if (!OpenRecoveryScript::Insert_ORS_Command(ORSCommand))
263 Keep_Going = false;
264 }
265 if (Keep_Going) {
266 if (Factory_Reset) {
267 if (!OpenRecoveryScript::Insert_ORS_Command("wipe data\n"))
268 Keep_Going = false;
269 } else if (Cache_Wipe) {
270 if (!OpenRecoveryScript::Insert_ORS_Command("wipe cache\n"))
271 Keep_Going = false;
272 }
273 }
274
275 TWFunc::Update_Log_File();
276 // Offer to decrypt if the device is encrypted
277 if (DataManager::GetIntValue(TW_IS_ENCRYPTED) != 0) {
278 LOGINFO("Is encrypted, do decrypt page first\n");
279 if (gui_startPage("decrypt") != 0) {
280 LOGERR("Failed to start decrypt GUI page.\n");
Ethan Yonkercf50da52015-01-12 21:59:07 -0600281 } else {
282 // Check for and load custom theme if present
283 gui_loadCustomResources();
Dees_Troya449a6f2013-04-07 17:50:11 -0500284 }
Ethan Yonker4b94cfd2014-12-11 10:00:45 -0600285 } else if (datamedia) {
286 if (tw_get_default_metadata(DataManager::GetSettingsStoragePath().c_str()) != 0) {
287 LOGERR("Failed to get default contexts and file mode for storage files.\n");
288 } else {
289 LOGINFO("Got default contexts and file mode for storage files.\n");
290 }
Dees_Troya449a6f2013-04-07 17:50:11 -0500291 }
292
293 // Read the settings file
Ethan Yonker73da42c2014-09-03 11:30:33 -0500294#ifdef TW_HAS_MTP
295 // We unmount partitions sometimes during early boot which may override
296 // the default of MTP being enabled by auto toggling MTP off. This
297 // will force it back to enabled then get overridden by the settings
298 // file, assuming that an entry for tw_mtp_enabled is set.
299 DataManager::SetValue("tw_mtp_enabled", 1);
300#endif
Dees_Troya449a6f2013-04-07 17:50:11 -0500301 DataManager::ReadSettingsFile();
Vojtech Bocek2005a842014-03-11 19:40:34 +0100302
303 // Fixup the RTC clock on devices which require it
Vojtech Bocek67351dc2014-07-03 15:22:41 +0200304 if(crash_counter == 0)
305 TWFunc::Fixup_Time_On_Boot();
Vojtech Bocek2005a842014-03-11 19:40:34 +0100306
Dees_Troya449a6f2013-04-07 17:50:11 -0500307 // Run any outstanding OpenRecoveryScript
308 if (DataManager::GetIntValue(TW_IS_ENCRYPTED) == 0 && (TWFunc::Path_Exists(SCRIPT_FILE_TMP) || TWFunc::Path_Exists(SCRIPT_FILE_CACHE))) {
309 OpenRecoveryScript::Run_OpenRecoveryScript();
310 }
Vojtech Bocekd0e38bc2014-02-03 23:36:57 +0100311
Ethan Yonker6f49e112014-09-03 21:42:49 -0500312#ifdef TW_HAS_MTP
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400313 // Enable MTP?
Ethan Yonker6f49e112014-09-03 21:42:49 -0500314 char mtp_crash_check[PROPERTY_VALUE_MAX];
315 property_get("mtp.crash_check", mtp_crash_check, "0");
316 if (strcmp(mtp_crash_check, "0") == 0) {
317 property_set("mtp.crash_check", "1");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -0600318 if (DataManager::GetIntValue("tw_mtp_enabled") == 1 && ((DataManager::GetIntValue(TW_IS_ENCRYPTED) != 0 && DataManager::GetIntValue(TW_IS_DECRYPTED) != 0) || DataManager::GetIntValue(TW_IS_ENCRYPTED) == 0)) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400319 LOGINFO("Enabling MTP during startup\n");
320 if (!PartitionManager.Enable_MTP())
321 PartitionManager.Disable_MTP();
322 else
323 gui_print("MTP Enabled\n");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -0600324 } else {
325 PartitionManager.Disable_MTP();
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400326 }
Ethan Yonker6f49e112014-09-03 21:42:49 -0500327 property_set("mtp.crash_check", "0");
328 } else {
329 gui_print_color("warning", "MTP Crashed, not starting MTP on boot.\n");
Ethan Yonker12055fa2014-09-04 08:06:53 -0500330 DataManager::SetValue("tw_mtp_enabled", 0);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -0600331 PartitionManager.Disable_MTP();
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400332 }
Ethan Yonkerdf7abac2014-12-29 10:48:17 -0600333#else
334 PartitionManager.Disable_MTP();
Ethan Yonker6f49e112014-09-03 21:42:49 -0500335#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400336
Dees_Troya449a6f2013-04-07 17:50:11 -0500337 // Launch the main GUI
338 gui_start();
339
340 // Check for su to see if the device is rooted or not
341 if (PartitionManager.Mount_By_Path("/system", false)) {
342 // Disable flashing of stock recovery
Dees Troyc8f58992014-11-20 15:29:03 +0000343 if (TWFunc::Path_Exists("/system/recovery-from-boot.p")) {
Dees_Troya449a6f2013-04-07 17:50:11 -0500344 rename("/system/recovery-from-boot.p", "/system/recovery-from-boot.bak");
345 gui_print("Renamed stock recovery file in /system to prevent\nthe stock ROM from replacing TWRP.\n");
346 }
347 if (TWFunc::Path_Exists("/supersu/su") && !TWFunc::Path_Exists("/system/bin/su") && !TWFunc::Path_Exists("/system/xbin/su") && !TWFunc::Path_Exists("/system/bin/.ext/.su")) {
348 // Device doesn't have su installed
349 DataManager::SetValue("tw_busy", 1);
350 if (gui_startPage("installsu") != 0) {
Dees Troyf193f882013-09-11 14:56:20 +0000351 LOGERR("Failed to start SuperSU install page.\n");
Dees_Troya449a6f2013-04-07 17:50:11 -0500352 }
Dees_Troya449a6f2013-04-07 17:50:11 -0500353 }
354 sync();
355 PartitionManager.UnMount_By_Path("/system", false);
356 }
357
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200358 // Reboot
Dees_Troya449a6f2013-04-07 17:50:11 -0500359 TWFunc::Update_Intent_File(Reboot_Value);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200360 TWFunc::Update_Log_File();
361 gui_print("Rebooting...\n");
Dees_Troya449a6f2013-04-07 17:50:11 -0500362 string Reboot_Arg;
363 DataManager::GetValue("tw_reboot_arg", Reboot_Arg);
364 if (Reboot_Arg == "recovery")
365 TWFunc::tw_reboot(rb_recovery);
366 else if (Reboot_Arg == "poweroff")
367 TWFunc::tw_reboot(rb_poweroff);
368 else if (Reboot_Arg == "bootloader")
369 TWFunc::tw_reboot(rb_bootloader);
370 else if (Reboot_Arg == "download")
371 TWFunc::tw_reboot(rb_download);
372 else
373 TWFunc::tw_reboot(rb_system);
374
375#ifdef ANDROID_RB_RESTART
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200376 android_reboot(ANDROID_RB_RESTART, 0, 0);
Dees_Troya449a6f2013-04-07 17:50:11 -0500377#else
378 reboot(RB_AUTOBOOT);
379#endif
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200380 return 0;
Dees_Troya449a6f2013-04-07 17:50:11 -0500381}