blob: 2849af759eae3f3261075673ca747f0766ac5fa2 [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>
23
24#include "cutils/properties.h"
25extern "C" {
26#include "minadbd/adb.h"
27#include "bootloader.h"
28}
29
30#ifdef ANDROID_RB_RESTART
31#include "cutils/android_reboot.h"
32#else
33#include <sys/reboot.h>
34#endif
35
36extern "C" {
37#include "gui/gui.h"
38}
39#include "twcommon.h"
40#include "twrp-functions.hpp"
41#include "data.hpp"
42#include "partitions.hpp"
43#include "openrecoveryscript.hpp"
44#include "variables.h"
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050045#include "twrpDU.hpp"
Dees_Troya449a6f2013-04-07 17:50:11 -050046
Dees_Troya95f55c2013-08-17 13:14:43 +000047#ifdef HAVE_SELINUX
48#include "selinux/label.h"
49struct selabel_handle *selinux_handle;
50#endif
51
Dees_Troya449a6f2013-04-07 17:50:11 -050052TWPartitionManager PartitionManager;
53int Log_Offset;
Ethan Yonker6277c792014-09-15 14:54:30 -050054bool datamedia;
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050055twrpDU du;
Dees_Troya449a6f2013-04-07 17:50:11 -050056
57static void Print_Prop(const char *key, const char *name, void *cookie) {
58 printf("%s=%s\n", key, name);
59}
60
61int main(int argc, char **argv) {
62 // Recovery needs to install world-readable files, so clear umask
63 // set by init
64 umask(0);
65
66 Log_Offset = 0;
67
68 // Set up temporary log file (/tmp/recovery.log)
69 freopen(TMP_LOG_FILE, "a", stdout);
70 setbuf(stdout, NULL);
71 freopen(TMP_LOG_FILE, "a", stderr);
72 setbuf(stderr, NULL);
73
74 // Handle ADB sideload
75 if (argc == 3 && strcmp(argv[1], "--adbd") == 0) {
76 adb_main(argv[2]);
77 return 0;
78 }
79
Ethan Yonker6277c792014-09-15 14:54:30 -050080#ifdef RECOVERY_SDCARD_ON_DATA
81 datamedia = true;
82#endif
83
Vojtech Bocek0fc15732014-07-04 01:09:50 +020084 char crash_prop_val[PROPERTY_VALUE_MAX];
85 int crash_counter;
86 property_get("twrp.crash_counter", crash_prop_val, "-1");
87 crash_counter = atoi(crash_prop_val) + 1;
88 snprintf(crash_prop_val, sizeof(crash_prop_val), "%d", crash_counter);
89 property_set("twrp.crash_counter", crash_prop_val);
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040090 property_set("ro.twrp.boot", "1");
91 property_set("ro.twrp.version", TW_VERSION_STR);
Vojtech Bocek0fc15732014-07-04 01:09:50 +020092
Dees_Troya449a6f2013-04-07 17:50:11 -050093 time_t StartupTime = time(NULL);
Ethan Yonkera1674162014-11-06 08:35:10 -060094 printf("Starting TWRP %s on %s (pid %d)", TW_VERSION_STR, ctime(&StartupTime), getpid());
Dees_Troya449a6f2013-04-07 17:50:11 -050095
96 // Load default values to set DataManager constants and handle ifdefs
97 DataManager::SetDefaultValues();
98 printf("Starting the UI...");
99 gui_init();
100 printf("=> Linking mtab\n");
101 symlink("/proc/mounts", "/etc/mtab");
Dees_Troy329383e2013-08-29 14:16:06 +0000102 if (TWFunc::Path_Exists("/etc/twrp.fstab")) {
103 if (TWFunc::Path_Exists("/etc/recovery.fstab")) {
104 printf("Renaming regular /etc/recovery.fstab -> /etc/recovery.fstab.bak\n");
105 rename("/etc/recovery.fstab", "/etc/recovery.fstab.bak");
106 }
107 printf("Moving /etc/twrp.fstab -> /etc/recovery.fstab\n");
108 rename("/etc/twrp.fstab", "/etc/recovery.fstab");
109 }
Dees_Troya449a6f2013-04-07 17:50:11 -0500110 printf("=> Processing recovery.fstab\n");
111 if (!PartitionManager.Process_Fstab("/etc/recovery.fstab", 1)) {
112 LOGERR("Failing out of recovery due to problem with recovery.fstab.\n");
113 return -1;
114 }
115 PartitionManager.Output_Partition_Logging();
116 // Load up all the resources
117 gui_loadResources();
118
Dees_Troya95f55c2013-08-17 13:14:43 +0000119#ifdef HAVE_SELINUX
bigbiff bigbiffc49d7062013-10-11 20:28:00 -0400120 if (TWFunc::Path_Exists("/prebuilt_file_contexts")) {
121 if (TWFunc::Path_Exists("/file_contexts")) {
122 printf("Renaming regular /file_contexts -> /file_contexts.bak\n");
123 rename("/file_contexts", "/file_contexts.bak");
124 }
125 printf("Moving /prebuilt_file_contexts -> /file_contexts\n");
126 rename("/prebuilt_file_contexts", "/file_contexts");
127 }
Dees_Troya95f55c2013-08-17 13:14:43 +0000128 struct selinux_opt selinux_options[] = {
129 { SELABEL_OPT_PATH, "/file_contexts" }
130 };
bigbiff bigbiffc49d7062013-10-11 20:28:00 -0400131 selinux_handle = selabel_open(SELABEL_CTX_FILE, selinux_options, 1);
Dees_Troya95f55c2013-08-17 13:14:43 +0000132 if (!selinux_handle)
133 printf("No file contexts for SELinux\n");
134 else
135 printf("SELinux contexts loaded from /file_contexts\n");
Dees Troy995e88c2013-11-26 21:39:14 +0000136 { // Check to ensure SELinux can be supported by the kernel
137 char *contexts = NULL;
Dees Troy8d0eb132013-12-06 16:55:41 +0000138
139 if (PartitionManager.Mount_By_Path("/cache", true) && TWFunc::Path_Exists("/cache/recovery")) {
140 lgetfilecon("/cache/recovery", &contexts);
141 if (!contexts) {
142 lsetfilecon("/cache/recovery", "test");
143 lgetfilecon("/cache/recovery", &contexts);
144 }
145 } else {
146 LOGINFO("Could not check /cache/recovery SELinux contexts, using /sbin/teamwin instead which may be inaccurate.\n");
147 lgetfilecon("/sbin/teamwin", &contexts);
148 }
Dees Troy995e88c2013-11-26 21:39:14 +0000149 if (!contexts) {
Ethan Yonkerbf2cb1c2014-07-02 10:15:54 -0500150 gui_print_color("warning", "Kernel does not have support for reading SELinux contexts.\n");
Dees Troy995e88c2013-11-26 21:39:14 +0000151 } else {
152 free(contexts);
153 gui_print("Full SELinux support is present.\n");
154 }
155 }
156#else
Ethan Yonkerbf2cb1c2014-07-02 10:15:54 -0500157 gui_print_color("warning", "No SELinux support (no libselinux).\n");
Dees_Troya95f55c2013-08-17 13:14:43 +0000158#endif
159
Dees_Troya449a6f2013-04-07 17:50:11 -0500160 PartitionManager.Mount_By_Path("/cache", true);
161
162 string Zip_File, Reboot_Value;
Ethan Yonkera1674162014-11-06 08:35:10 -0600163 bool Cache_Wipe = false, Factory_Reset = false, Perform_Backup = false, Shutdown = false;
Dees_Troya449a6f2013-04-07 17:50:11 -0500164
165 {
Dees_Troy1669f892013-09-04 18:35:08 +0000166 TWPartition* misc = PartitionManager.Find_Partition_By_Path("/misc");
167 if (misc != NULL) {
168 if (misc->Current_File_System == "emmc") {
169 set_device_type('e');
170 set_device_name(misc->Actual_Block_Device.c_str());
171 } else if (misc->Current_File_System == "mtd") {
172 set_device_type('m');
173 set_device_name(misc->MTD_Name.c_str());
174 } else {
175 LOGERR("Unknown file system for /misc\n");
176 }
177 }
Dees_Troya449a6f2013-04-07 17:50:11 -0500178 get_args(&argc, &argv);
179
180 int index, index2, len;
181 char* argptr;
182 char* ptr;
183 printf("Startup Commands: ");
184 for (index = 1; index < argc; index++) {
185 argptr = argv[index];
186 printf(" '%s'", argv[index]);
187 len = strlen(argv[index]);
188 if (*argptr == '-') {argptr++; len--;}
189 if (*argptr == '-') {argptr++; len--;}
190 if (*argptr == 'u') {
191 ptr = argptr;
192 index2 = 0;
193 while (*ptr != '=' && *ptr != '\n')
194 ptr++;
Kevin Steck7b69e9d2013-10-17 18:24:31 -0400195 // skip the = before grabbing Zip_File
196 while (*ptr == '=')
197 ptr++;
Dees_Troya449a6f2013-04-07 17:50:11 -0500198 if (*ptr) {
199 Zip_File = ptr;
200 } else
201 LOGERR("argument error specifying zip file\n");
202 } else if (*argptr == 'w') {
203 if (len == 9)
204 Factory_Reset = true;
205 else if (len == 10)
206 Cache_Wipe = true;
207 } else if (*argptr == 'n') {
208 Perform_Backup = true;
Ethan Yonkera1674162014-11-06 08:35:10 -0600209 } else if (*argptr == 'p') {
210 Shutdown = true;
Dees_Troya449a6f2013-04-07 17:50:11 -0500211 } else if (*argptr == 's') {
212 ptr = argptr;
213 index2 = 0;
214 while (*ptr != '=' && *ptr != '\n')
215 ptr++;
216 if (*ptr) {
217 Reboot_Value = *ptr;
218 }
219 }
220 }
Vojtech Bocek0fc15732014-07-04 01:09:50 +0200221 printf("\n");
Dees_Troya449a6f2013-04-07 17:50:11 -0500222 }
223
Vojtech Bocek0fc15732014-07-04 01:09:50 +0200224 if(crash_counter == 0) {
Dees_Troya449a6f2013-04-07 17:50:11 -0500225 property_list(Print_Prop, NULL);
226 printf("\n");
Vojtech Bocek0fc15732014-07-04 01:09:50 +0200227 } else {
228 printf("twrp.crash_counter=%d\n", crash_counter);
Dees_Troya449a6f2013-04-07 17:50:11 -0500229 }
230
231 // Check for and run startup script if script exists
232 TWFunc::check_and_run_script("/sbin/runatboot.sh", "boot");
233 TWFunc::check_and_run_script("/sbin/postrecoveryboot.sh", "boot");
234
235#ifdef TW_INCLUDE_INJECTTWRP
236 // Back up TWRP Ramdisk if needed:
237 TWPartition* Boot = PartitionManager.Find_Partition_By_Path("/boot");
Dees_Troya449a6f2013-04-07 17:50:11 -0500238 LOGINFO("Backing up TWRP ramdisk...\n");
239 if (Boot == NULL || Boot->Current_File_System != "emmc")
Vojtech Bocek05534202013-09-11 08:11:56 +0200240 TWFunc::Exec_Cmd("injecttwrp --backup /tmp/backup_recovery_ramdisk.img");
Dees_Troya449a6f2013-04-07 17:50:11 -0500241 else {
242 string injectcmd = "injecttwrp --backup /tmp/backup_recovery_ramdisk.img bd=" + Boot->Actual_Block_Device;
Vojtech Bocek05534202013-09-11 08:11:56 +0200243 TWFunc::Exec_Cmd(injectcmd);
Dees_Troya449a6f2013-04-07 17:50:11 -0500244 }
245 LOGINFO("Backup of TWRP ramdisk done.\n");
246#endif
247
248 bool Keep_Going = true;
249 if (Perform_Backup) {
Dees Troyb21cc642013-09-10 17:36:41 +0000250 DataManager::SetValue(TW_BACKUP_NAME, "(Auto Generate)");
Dees_Troya449a6f2013-04-07 17:50:11 -0500251 if (!OpenRecoveryScript::Insert_ORS_Command("backup BSDCAE\n"))
252 Keep_Going = false;
253 }
254 if (Keep_Going && !Zip_File.empty()) {
255 string ORSCommand = "install " + Zip_File;
256
257 if (!OpenRecoveryScript::Insert_ORS_Command(ORSCommand))
258 Keep_Going = false;
259 }
260 if (Keep_Going) {
261 if (Factory_Reset) {
262 if (!OpenRecoveryScript::Insert_ORS_Command("wipe data\n"))
263 Keep_Going = false;
264 } else if (Cache_Wipe) {
265 if (!OpenRecoveryScript::Insert_ORS_Command("wipe cache\n"))
266 Keep_Going = false;
267 }
268 }
269
270 TWFunc::Update_Log_File();
271 // Offer to decrypt if the device is encrypted
272 if (DataManager::GetIntValue(TW_IS_ENCRYPTED) != 0) {
273 LOGINFO("Is encrypted, do decrypt page first\n");
274 if (gui_startPage("decrypt") != 0) {
275 LOGERR("Failed to start decrypt GUI page.\n");
276 }
277 }
278
279 // Read the settings file
Ethan Yonker73da42c2014-09-03 11:30:33 -0500280#ifdef TW_HAS_MTP
281 // We unmount partitions sometimes during early boot which may override
282 // the default of MTP being enabled by auto toggling MTP off. This
283 // will force it back to enabled then get overridden by the settings
284 // file, assuming that an entry for tw_mtp_enabled is set.
285 DataManager::SetValue("tw_mtp_enabled", 1);
286#endif
Dees_Troya449a6f2013-04-07 17:50:11 -0500287 DataManager::ReadSettingsFile();
Vojtech Bocek2005a842014-03-11 19:40:34 +0100288
289 // Fixup the RTC clock on devices which require it
Vojtech Bocek67351dc2014-07-03 15:22:41 +0200290 if(crash_counter == 0)
291 TWFunc::Fixup_Time_On_Boot();
Vojtech Bocek2005a842014-03-11 19:40:34 +0100292
Dees_Troya449a6f2013-04-07 17:50:11 -0500293 // Run any outstanding OpenRecoveryScript
294 if (DataManager::GetIntValue(TW_IS_ENCRYPTED) == 0 && (TWFunc::Path_Exists(SCRIPT_FILE_TMP) || TWFunc::Path_Exists(SCRIPT_FILE_CACHE))) {
295 OpenRecoveryScript::Run_OpenRecoveryScript();
296 }
Vojtech Bocekd0e38bc2014-02-03 23:36:57 +0100297
Ethan Yonker6f49e112014-09-03 21:42:49 -0500298#ifdef TW_HAS_MTP
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400299 // Enable MTP?
Ethan Yonker6f49e112014-09-03 21:42:49 -0500300 char mtp_crash_check[PROPERTY_VALUE_MAX];
301 property_get("mtp.crash_check", mtp_crash_check, "0");
302 if (strcmp(mtp_crash_check, "0") == 0) {
303 property_set("mtp.crash_check", "1");
304 if (DataManager::GetIntValue(TW_IS_ENCRYPTED) != 0) {
305 if (DataManager::GetIntValue(TW_IS_DECRYPTED) != 0 && DataManager::GetIntValue("tw_mtp_enabled") == 1) {
306 LOGINFO("Enabling MTP during startup\n");
307 if (!PartitionManager.Enable_MTP())
308 PartitionManager.Disable_MTP();
309 else
310 gui_print("MTP Enabled\n");
311 }
312 } else if (DataManager::GetIntValue("tw_mtp_enabled") == 1) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400313 LOGINFO("Enabling MTP during startup\n");
314 if (!PartitionManager.Enable_MTP())
315 PartitionManager.Disable_MTP();
316 else
317 gui_print("MTP Enabled\n");
318 }
Ethan Yonker6f49e112014-09-03 21:42:49 -0500319 property_set("mtp.crash_check", "0");
320 } else {
321 gui_print_color("warning", "MTP Crashed, not starting MTP on boot.\n");
Ethan Yonker12055fa2014-09-04 08:06:53 -0500322 DataManager::SetValue("tw_mtp_enabled", 0);
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400323 }
Ethan Yonker6f49e112014-09-03 21:42:49 -0500324#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400325
Dees_Troya449a6f2013-04-07 17:50:11 -0500326 // Launch the main GUI
327 gui_start();
328
329 // Check for su to see if the device is rooted or not
330 if (PartitionManager.Mount_By_Path("/system", false)) {
331 // Disable flashing of stock recovery
dhacker2930f100e2014-07-15 21:33:29 -0400332 if (TWFunc::Path_Exists("/system/recovery-from-boot.p") && TWFunc::Path_Exists("/system/etc/install-recovery.sh")) {
Dees_Troya449a6f2013-04-07 17:50:11 -0500333 rename("/system/recovery-from-boot.p", "/system/recovery-from-boot.bak");
334 gui_print("Renamed stock recovery file in /system to prevent\nthe stock ROM from replacing TWRP.\n");
335 }
336 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")) {
337 // Device doesn't have su installed
338 DataManager::SetValue("tw_busy", 1);
339 if (gui_startPage("installsu") != 0) {
Dees Troyf193f882013-09-11 14:56:20 +0000340 LOGERR("Failed to start SuperSU install page.\n");
Dees_Troya449a6f2013-04-07 17:50:11 -0500341 }
342 } else if (TWFunc::Check_su_Perms() > 0) {
343 // su perms are set incorrectly
Dees Troyf193f882013-09-11 14:56:20 +0000344 LOGINFO("Root permissions appear to be lost... fixing. (This will always happen on 4.3+ ROMs with SELinux.\n");
345 TWFunc::Fix_su_Perms();
Dees_Troya449a6f2013-04-07 17:50:11 -0500346 }
347 sync();
348 PartitionManager.UnMount_By_Path("/system", false);
349 }
350
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200351 // Reboot
Dees_Troya449a6f2013-04-07 17:50:11 -0500352 TWFunc::Update_Intent_File(Reboot_Value);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200353 TWFunc::Update_Log_File();
354 gui_print("Rebooting...\n");
Dees_Troya449a6f2013-04-07 17:50:11 -0500355 string Reboot_Arg;
356 DataManager::GetValue("tw_reboot_arg", Reboot_Arg);
357 if (Reboot_Arg == "recovery")
358 TWFunc::tw_reboot(rb_recovery);
359 else if (Reboot_Arg == "poweroff")
360 TWFunc::tw_reboot(rb_poweroff);
361 else if (Reboot_Arg == "bootloader")
362 TWFunc::tw_reboot(rb_bootloader);
363 else if (Reboot_Arg == "download")
364 TWFunc::tw_reboot(rb_download);
365 else
366 TWFunc::tw_reboot(rb_system);
367
368#ifdef ANDROID_RB_RESTART
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200369 android_reboot(ANDROID_RB_RESTART, 0, 0);
Dees_Troya449a6f2013-04-07 17:50:11 -0500370#else
371 reboot(RB_AUTOBOOT);
372#endif
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200373 return 0;
Dees_Troya449a6f2013-04-07 17:50:11 -0500374}