blob: 06532cba2ac2d8cdae086d32d4c54951e4de2fdb [file] [log] [blame]
Dees_Troya449a6f2013-04-07 17:50:11 -05001/*
Dees_Troya449a6f2013-04-07 17:50:11 -05002
bigbiff bigbiff34684ff2013-12-01 21:03:45 -05003ccdd
4 TWRP is free software: you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation, either version 3 of the License, or
7 (at your option) any later version.
Dees_Troya449a6f2013-04-07 17:50:11 -05008
bigbiff bigbiff34684ff2013-12-01 21:03:45 -05009 TWRP is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
Dees_Troya449a6f2013-04-07 17:50:11 -050013
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050014 You should have received a copy of the GNU General Public License
15 along with TWRP. If not, see <http://www.gnu.org/licenses/>.
Dees_Troya449a6f2013-04-07 17:50:11 -050016*/
17
18#include <stdio.h>
19#include <stdlib.h>
20#include <string.h>
21#include <sys/stat.h>
22#include <sys/types.h>
23#include <time.h>
24#include <unistd.h>
25
26#include "cutils/properties.h"
27extern "C" {
28#include "minadbd/adb.h"
29#include "bootloader.h"
30}
31
32#ifdef ANDROID_RB_RESTART
33#include "cutils/android_reboot.h"
34#else
35#include <sys/reboot.h>
36#endif
37
38extern "C" {
39#include "gui/gui.h"
40}
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;
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050056twrpDU du;
Dees_Troya449a6f2013-04-07 17:50:11 -050057
58static void Print_Prop(const char *key, const char *name, void *cookie) {
59 printf("%s=%s\n", key, name);
60}
61
62int main(int argc, char **argv) {
63 // Recovery needs to install world-readable files, so clear umask
64 // set by init
65 umask(0);
66
67 Log_Offset = 0;
68
69 // Set up temporary log file (/tmp/recovery.log)
70 freopen(TMP_LOG_FILE, "a", stdout);
71 setbuf(stdout, NULL);
72 freopen(TMP_LOG_FILE, "a", stderr);
73 setbuf(stderr, NULL);
74
75 // Handle ADB sideload
76 if (argc == 3 && strcmp(argv[1], "--adbd") == 0) {
77 adb_main(argv[2]);
78 return 0;
79 }
80
81 time_t StartupTime = time(NULL);
82 printf("Starting TWRP %s on %s", TW_VERSION_STR, ctime(&StartupTime));
83
84 // Load default values to set DataManager constants and handle ifdefs
85 DataManager::SetDefaultValues();
86 printf("Starting the UI...");
87 gui_init();
88 printf("=> Linking mtab\n");
89 symlink("/proc/mounts", "/etc/mtab");
Dees_Troy329383e2013-08-29 14:16:06 +000090 if (TWFunc::Path_Exists("/etc/twrp.fstab")) {
91 if (TWFunc::Path_Exists("/etc/recovery.fstab")) {
92 printf("Renaming regular /etc/recovery.fstab -> /etc/recovery.fstab.bak\n");
93 rename("/etc/recovery.fstab", "/etc/recovery.fstab.bak");
94 }
95 printf("Moving /etc/twrp.fstab -> /etc/recovery.fstab\n");
96 rename("/etc/twrp.fstab", "/etc/recovery.fstab");
97 }
Dees_Troya449a6f2013-04-07 17:50:11 -050098 printf("=> Processing recovery.fstab\n");
99 if (!PartitionManager.Process_Fstab("/etc/recovery.fstab", 1)) {
100 LOGERR("Failing out of recovery due to problem with recovery.fstab.\n");
101 return -1;
102 }
103 PartitionManager.Output_Partition_Logging();
104 // Load up all the resources
105 gui_loadResources();
106
Dees_Troya95f55c2013-08-17 13:14:43 +0000107#ifdef HAVE_SELINUX
bigbiff bigbiffc49d7062013-10-11 20:28:00 -0400108 if (TWFunc::Path_Exists("/prebuilt_file_contexts")) {
109 if (TWFunc::Path_Exists("/file_contexts")) {
110 printf("Renaming regular /file_contexts -> /file_contexts.bak\n");
111 rename("/file_contexts", "/file_contexts.bak");
112 }
113 printf("Moving /prebuilt_file_contexts -> /file_contexts\n");
114 rename("/prebuilt_file_contexts", "/file_contexts");
115 }
Dees_Troya95f55c2013-08-17 13:14:43 +0000116 struct selinux_opt selinux_options[] = {
117 { SELABEL_OPT_PATH, "/file_contexts" }
118 };
bigbiff bigbiffc49d7062013-10-11 20:28:00 -0400119 selinux_handle = selabel_open(SELABEL_CTX_FILE, selinux_options, 1);
Dees_Troya95f55c2013-08-17 13:14:43 +0000120 if (!selinux_handle)
121 printf("No file contexts for SELinux\n");
122 else
123 printf("SELinux contexts loaded from /file_contexts\n");
Dees Troy995e88c2013-11-26 21:39:14 +0000124 { // Check to ensure SELinux can be supported by the kernel
125 char *contexts = NULL;
Dees Troy8d0eb132013-12-06 16:55:41 +0000126
127 if (PartitionManager.Mount_By_Path("/cache", true) && TWFunc::Path_Exists("/cache/recovery")) {
128 lgetfilecon("/cache/recovery", &contexts);
129 if (!contexts) {
130 lsetfilecon("/cache/recovery", "test");
131 lgetfilecon("/cache/recovery", &contexts);
132 }
133 } else {
134 LOGINFO("Could not check /cache/recovery SELinux contexts, using /sbin/teamwin instead which may be inaccurate.\n");
135 lgetfilecon("/sbin/teamwin", &contexts);
136 }
Dees Troy995e88c2013-11-26 21:39:14 +0000137 if (!contexts) {
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500138 gui_print("Kernel does not have support for reading SELinux contexts.\n");
Dees Troy995e88c2013-11-26 21:39:14 +0000139 } else {
140 free(contexts);
141 gui_print("Full SELinux support is present.\n");
142 }
143 }
144#else
145 gui_print("No SELinux support (no libselinux).\n");
Dees_Troya95f55c2013-08-17 13:14:43 +0000146#endif
147
Dees_Troya449a6f2013-04-07 17:50:11 -0500148 PartitionManager.Mount_By_Path("/cache", true);
149
150 string Zip_File, Reboot_Value;
151 bool Cache_Wipe = false, Factory_Reset = false, Perform_Backup = false;
152
153 {
Dees_Troy1669f892013-09-04 18:35:08 +0000154 TWPartition* misc = PartitionManager.Find_Partition_By_Path("/misc");
155 if (misc != NULL) {
156 if (misc->Current_File_System == "emmc") {
157 set_device_type('e');
158 set_device_name(misc->Actual_Block_Device.c_str());
159 } else if (misc->Current_File_System == "mtd") {
160 set_device_type('m');
161 set_device_name(misc->MTD_Name.c_str());
162 } else {
163 LOGERR("Unknown file system for /misc\n");
164 }
165 }
Dees_Troya449a6f2013-04-07 17:50:11 -0500166 get_args(&argc, &argv);
167
168 int index, index2, len;
169 char* argptr;
170 char* ptr;
171 printf("Startup Commands: ");
172 for (index = 1; index < argc; index++) {
173 argptr = argv[index];
174 printf(" '%s'", argv[index]);
175 len = strlen(argv[index]);
176 if (*argptr == '-') {argptr++; len--;}
177 if (*argptr == '-') {argptr++; len--;}
178 if (*argptr == 'u') {
179 ptr = argptr;
180 index2 = 0;
181 while (*ptr != '=' && *ptr != '\n')
182 ptr++;
Kevin Steck7b69e9d2013-10-17 18:24:31 -0400183 // skip the = before grabbing Zip_File
184 while (*ptr == '=')
185 ptr++;
Dees_Troya449a6f2013-04-07 17:50:11 -0500186 if (*ptr) {
187 Zip_File = ptr;
188 } else
189 LOGERR("argument error specifying zip file\n");
190 } else if (*argptr == 'w') {
191 if (len == 9)
192 Factory_Reset = true;
193 else if (len == 10)
194 Cache_Wipe = true;
195 } else if (*argptr == 'n') {
196 Perform_Backup = true;
197 } else if (*argptr == 's') {
198 ptr = argptr;
199 index2 = 0;
200 while (*ptr != '=' && *ptr != '\n')
201 ptr++;
202 if (*ptr) {
203 Reboot_Value = *ptr;
204 }
205 }
206 }
207 }
208
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200209 char twrp_booted[PROPERTY_VALUE_MAX];
Dees_Troya449a6f2013-04-07 17:50:11 -0500210 property_get("ro.twrp.boot", twrp_booted, "0");
211 if (strcmp(twrp_booted, "0") == 0) {
212 property_list(Print_Prop, NULL);
213 printf("\n");
214 property_set("ro.twrp.boot", "1");
215 }
216
217 // Check for and run startup script if script exists
218 TWFunc::check_and_run_script("/sbin/runatboot.sh", "boot");
219 TWFunc::check_and_run_script("/sbin/postrecoveryboot.sh", "boot");
220
221#ifdef TW_INCLUDE_INJECTTWRP
222 // Back up TWRP Ramdisk if needed:
223 TWPartition* Boot = PartitionManager.Find_Partition_By_Path("/boot");
Dees_Troya449a6f2013-04-07 17:50:11 -0500224 LOGINFO("Backing up TWRP ramdisk...\n");
225 if (Boot == NULL || Boot->Current_File_System != "emmc")
Vojtech Bocek05534202013-09-11 08:11:56 +0200226 TWFunc::Exec_Cmd("injecttwrp --backup /tmp/backup_recovery_ramdisk.img");
Dees_Troya449a6f2013-04-07 17:50:11 -0500227 else {
228 string injectcmd = "injecttwrp --backup /tmp/backup_recovery_ramdisk.img bd=" + Boot->Actual_Block_Device;
Vojtech Bocek05534202013-09-11 08:11:56 +0200229 TWFunc::Exec_Cmd(injectcmd);
Dees_Troya449a6f2013-04-07 17:50:11 -0500230 }
231 LOGINFO("Backup of TWRP ramdisk done.\n");
232#endif
233
234 bool Keep_Going = true;
235 if (Perform_Backup) {
Dees Troyb21cc642013-09-10 17:36:41 +0000236 DataManager::SetValue(TW_BACKUP_NAME, "(Auto Generate)");
Dees_Troya449a6f2013-04-07 17:50:11 -0500237 if (!OpenRecoveryScript::Insert_ORS_Command("backup BSDCAE\n"))
238 Keep_Going = false;
239 }
240 if (Keep_Going && !Zip_File.empty()) {
241 string ORSCommand = "install " + Zip_File;
242
243 if (!OpenRecoveryScript::Insert_ORS_Command(ORSCommand))
244 Keep_Going = false;
245 }
246 if (Keep_Going) {
247 if (Factory_Reset) {
248 if (!OpenRecoveryScript::Insert_ORS_Command("wipe data\n"))
249 Keep_Going = false;
250 } else if (Cache_Wipe) {
251 if (!OpenRecoveryScript::Insert_ORS_Command("wipe cache\n"))
252 Keep_Going = false;
253 }
254 }
255
256 TWFunc::Update_Log_File();
257 // Offer to decrypt if the device is encrypted
258 if (DataManager::GetIntValue(TW_IS_ENCRYPTED) != 0) {
259 LOGINFO("Is encrypted, do decrypt page first\n");
260 if (gui_startPage("decrypt") != 0) {
261 LOGERR("Failed to start decrypt GUI page.\n");
262 }
263 }
264
265 // Read the settings file
266 DataManager::ReadSettingsFile();
267 // Run any outstanding OpenRecoveryScript
268 if (DataManager::GetIntValue(TW_IS_ENCRYPTED) == 0 && (TWFunc::Path_Exists(SCRIPT_FILE_TMP) || TWFunc::Path_Exists(SCRIPT_FILE_CACHE))) {
269 OpenRecoveryScript::Run_OpenRecoveryScript();
270 }
271 // Launch the main GUI
272 gui_start();
273
274 // Check for su to see if the device is rooted or not
275 if (PartitionManager.Mount_By_Path("/system", false)) {
276 // Disable flashing of stock recovery
277 if (TWFunc::Path_Exists("/system/recovery-from-boot.p")) {
278 rename("/system/recovery-from-boot.p", "/system/recovery-from-boot.bak");
279 gui_print("Renamed stock recovery file in /system to prevent\nthe stock ROM from replacing TWRP.\n");
280 }
281 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")) {
282 // Device doesn't have su installed
283 DataManager::SetValue("tw_busy", 1);
284 if (gui_startPage("installsu") != 0) {
Dees Troyf193f882013-09-11 14:56:20 +0000285 LOGERR("Failed to start SuperSU install page.\n");
Dees_Troya449a6f2013-04-07 17:50:11 -0500286 }
287 } else if (TWFunc::Check_su_Perms() > 0) {
288 // su perms are set incorrectly
Dees Troyf193f882013-09-11 14:56:20 +0000289 LOGINFO("Root permissions appear to be lost... fixing. (This will always happen on 4.3+ ROMs with SELinux.\n");
290 TWFunc::Fix_su_Perms();
Dees_Troya449a6f2013-04-07 17:50:11 -0500291 }
292 sync();
293 PartitionManager.UnMount_By_Path("/system", false);
294 }
295
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200296 // Reboot
Dees_Troya449a6f2013-04-07 17:50:11 -0500297 TWFunc::Update_Intent_File(Reboot_Value);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200298 TWFunc::Update_Log_File();
299 gui_print("Rebooting...\n");
Dees_Troya449a6f2013-04-07 17:50:11 -0500300 string Reboot_Arg;
301 DataManager::GetValue("tw_reboot_arg", Reboot_Arg);
302 if (Reboot_Arg == "recovery")
303 TWFunc::tw_reboot(rb_recovery);
304 else if (Reboot_Arg == "poweroff")
305 TWFunc::tw_reboot(rb_poweroff);
306 else if (Reboot_Arg == "bootloader")
307 TWFunc::tw_reboot(rb_bootloader);
308 else if (Reboot_Arg == "download")
309 TWFunc::tw_reboot(rb_download);
310 else
311 TWFunc::tw_reboot(rb_system);
312
313#ifdef ANDROID_RB_RESTART
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200314 android_reboot(ANDROID_RB_RESTART, 0, 0);
Dees_Troya449a6f2013-04-07 17:50:11 -0500315#else
316 reboot(RB_AUTOBOOT);
317#endif
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200318 return 0;
Dees_Troya449a6f2013-04-07 17:50:11 -0500319}