blob: 581e67e46773d454adebc792c9e1b5c60c5b7f12 [file] [log] [blame]
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -08001/*
2 * Copyright (C) 2007 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#include <ctype.h>
18#include <errno.h>
19#include <fcntl.h>
20#include <getopt.h>
21#include <limits.h>
22#include <linux/input.h>
23#include <stdio.h>
24#include <stdlib.h>
25#include <string.h>
Doug Zongker23ceeea2010-07-08 17:27:55 -070026#include <sys/stat.h>
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080027#include <sys/types.h>
28#include <time.h>
29#include <unistd.h>
Doug Zongker8674a722010-09-15 11:08:23 -070030#include <dirent.h>
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080031
32#include "bootloader.h"
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080033#include "common.h"
34#include "cutils/properties.h"
Dees_Troy38bd7602012-09-14 13:33:53 -040035#ifdef ANDROID_RB_RESTART
Ken Sumrall6e4472a2011-03-07 23:37:27 -080036#include "cutils/android_reboot.h"
Dees_Troy38bd7602012-09-14 13:33:53 -040037#else
38#include <sys/reboot.h>
39#endif
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080040#include "install.h"
41#include "minui/minui.h"
42#include "minzip/DirUtil.h"
43#include "roots.h"
Doug Zongker28ce47c2011-10-28 10:33:05 -070044#include "ui.h"
Doug Zongker211aebc2011-10-28 15:13:10 -070045#include "screen_ui.h"
Doug Zongkerdaefc1d2011-10-31 09:34:15 -070046#include "device.h"
Doug Zongker9270a202012-01-09 15:16:13 -080047#include "adb_install.h"
48extern "C" {
49#include "minadbd/adb.h"
50}
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080051
Dees_Troy7d15c252012-09-05 20:47:21 -040052extern "C" {
Dees_Troy7d15c252012-09-05 20:47:21 -040053#include "data.h"
54#include "gui/gui.h"
55}
56#include "partitions.hpp"
Dees_Troy51127312012-09-08 13:08:49 -040057#include "variables.h"
Dees_Troy812660f2012-09-20 09:55:17 -040058#include "openrecoveryscript.hpp"
Dees_Troya58bead2012-09-27 09:49:29 -040059#include "twrp-functions.hpp"
Dees_Troy7d15c252012-09-05 20:47:21 -040060
Dees_Troy5bf43922012-09-07 16:07:55 -040061TWPartitionManager PartitionManager;
Dees_Troy7d15c252012-09-05 20:47:21 -040062
Stephen Smalley779701d2012-02-09 14:13:23 -050063struct selabel_handle *sehandle;
64
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080065static const struct option OPTIONS[] = {
66 { "send_intent", required_argument, NULL, 's' },
67 { "update_package", required_argument, NULL, 'u' },
68 { "wipe_data", no_argument, NULL, 'w' },
69 { "wipe_cache", no_argument, NULL, 'c' },
Doug Zongker4bc98062010-09-03 11:00:13 -070070 { "show_text", no_argument, NULL, 't' },
Doug Zongkere5d5ac72012-04-12 11:01:22 -070071 { "just_exit", no_argument, NULL, 'x' },
Dees_Troy83a3b122012-10-01 14:16:43 -040072 { "nandroid", no_argument, NULL, 'n' },
Doug Zongker988500b2009-10-06 14:41:38 -070073 { NULL, 0, NULL, 0 },
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080074};
75
Doug Zongkerd4208f92010-09-20 12:16:13 -070076static const char *COMMAND_FILE = "/cache/recovery/command";
77static const char *INTENT_FILE = "/cache/recovery/intent";
78static const char *LOG_FILE = "/cache/recovery/log";
Doug Zongker2c3539e2010-09-29 13:21:30 -070079static const char *LAST_LOG_FILE = "/cache/recovery/last_log";
Doug Zongkerd0181b82011-10-19 10:51:12 -070080static const char *LAST_INSTALL_FILE = "/cache/recovery/last_install";
Michael Ward9d2629c2011-06-23 22:14:24 -070081static const char *CACHE_ROOT = "/cache";
Doug Zongkerd4208f92010-09-20 12:16:13 -070082static const char *SDCARD_ROOT = "/sdcard";
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080083static const char *TEMPORARY_LOG_FILE = "/tmp/recovery.log";
Doug Zongkerd0181b82011-10-19 10:51:12 -070084static const char *TEMPORARY_INSTALL_FILE = "/tmp/last_install";
Doug Zongkerd4208f92010-09-20 12:16:13 -070085static const char *SIDELOAD_TEMP_DIR = "/tmp/sideload";
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080086
Doug Zongker211aebc2011-10-28 15:13:10 -070087RecoveryUI* ui = NULL;
88
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080089/*
90 * The recovery tool communicates with the main system through /cache files.
91 * /cache/recovery/command - INPUT - command line for tool, one arg per line
92 * /cache/recovery/log - OUTPUT - combined log file from recovery run(s)
93 * /cache/recovery/intent - OUTPUT - intent that was passed in
94 *
95 * The arguments which may be supplied in the recovery.command file:
96 * --send_intent=anystring - write the text out to recovery.intent
Doug Zongkerd4208f92010-09-20 12:16:13 -070097 * --update_package=path - verify install an OTA package file
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080098 * --wipe_data - erase user data (and cache), then reboot
99 * --wipe_cache - wipe cache (but not user data), then reboot
Oscar Montemayor05231562009-11-30 08:40:57 -0800100 * --set_encrypted_filesystem=on|off - enables / diasables encrypted fs
Doug Zongkere5d5ac72012-04-12 11:01:22 -0700101 * --just_exit - do nothing; exit and reboot
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800102 *
103 * After completing, we remove /cache/recovery/command and reboot.
104 * Arguments may also be supplied in the bootloader control block (BCB).
105 * These important scenarios must be safely restartable at any point:
106 *
107 * FACTORY RESET
108 * 1. user selects "factory reset"
109 * 2. main system writes "--wipe_data" to /cache/recovery/command
110 * 3. main system reboots into recovery
111 * 4. get_args() writes BCB with "boot-recovery" and "--wipe_data"
112 * -- after this, rebooting will restart the erase --
Doug Zongkerd4208f92010-09-20 12:16:13 -0700113 * 5. erase_volume() reformats /data
114 * 6. erase_volume() reformats /cache
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800115 * 7. finish_recovery() erases BCB
116 * -- after this, rebooting will restart the main system --
117 * 8. main() calls reboot() to boot main system
118 *
119 * OTA INSTALL
120 * 1. main system downloads OTA package to /cache/some-filename.zip
Doug Zongker9b125b02010-09-22 12:01:37 -0700121 * 2. main system writes "--update_package=/cache/some-filename.zip"
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800122 * 3. main system reboots into recovery
123 * 4. get_args() writes BCB with "boot-recovery" and "--update_package=..."
124 * -- after this, rebooting will attempt to reinstall the update --
125 * 5. install_package() attempts to install the update
126 * NOTE: the package install must itself be restartable from any point
127 * 6. finish_recovery() erases BCB
128 * -- after this, rebooting will (try to) restart the main system --
129 * 7. ** if install failed **
130 * 7a. prompt_and_wait() shows an error icon and waits for the user
131 * 7b; the user reboots (pulling the battery, etc) into the main system
132 * 8. main() calls maybe_install_firmware_update()
133 * ** if the update contained radio/hboot firmware **:
134 * 8a. m_i_f_u() writes BCB with "boot-recovery" and "--wipe_cache"
135 * -- after this, rebooting will reformat cache & restart main system --
136 * 8b. m_i_f_u() writes firmware image into raw cache partition
137 * 8c. m_i_f_u() writes BCB with "update-radio/hboot" and "--wipe_cache"
138 * -- after this, rebooting will attempt to reinstall firmware --
139 * 8d. bootloader tries to flash firmware
140 * 8e. bootloader writes BCB with "boot-recovery" (keeping "--wipe_cache")
141 * -- after this, rebooting will reformat cache & restart main system --
Doug Zongkerd4208f92010-09-20 12:16:13 -0700142 * 8f. erase_volume() reformats /cache
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800143 * 8g. finish_recovery() erases BCB
144 * -- after this, rebooting will (try to) restart the main system --
145 * 9. main() calls reboot() to boot main system
146 */
147
148static const int MAX_ARG_LENGTH = 4096;
149static const int MAX_ARGS = 100;
150
Doug Zongkerd4208f92010-09-20 12:16:13 -0700151// open a given path, mounting partitions as necessary
Doug Zongker469243e2011-04-12 09:28:10 -0700152FILE*
Doug Zongkerd4208f92010-09-20 12:16:13 -0700153fopen_path(const char *path, const char *mode) {
154 if (ensure_path_mounted(path) != 0) {
155 LOGE("Can't mount %s\n", path);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800156 return NULL;
157 }
158
159 // When writing, try to create the containing directory, if necessary.
160 // Use generous permissions, the system (init.rc) will reset them.
Stephen Smalley779701d2012-02-09 14:13:23 -0500161 if (strchr("wa", mode[0])) dirCreateHierarchy(path, 0777, NULL, 1, sehandle);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800162
163 FILE *fp = fopen(path, mode);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800164 return fp;
165}
166
167// close a file, log an error if the error indicator is set
168static void
169check_and_fclose(FILE *fp, const char *name) {
170 fflush(fp);
171 if (ferror(fp)) LOGE("Error in %s\n(%s)\n", name, strerror(errno));
172 fclose(fp);
173}
174
175// command line args come from, in decreasing precedence:
176// - the actual command line
177// - the bootloader control block (one per line, after "recovery")
178// - the contents of COMMAND_FILE (one per line)
179static void
180get_args(int *argc, char ***argv) {
181 struct bootloader_message boot;
182 memset(&boot, 0, sizeof(boot));
183 get_bootloader_message(&boot); // this may fail, leaving a zeroed structure
184
185 if (boot.command[0] != 0 && boot.command[0] != 255) {
186 LOGI("Boot command: %.*s\n", sizeof(boot.command), boot.command);
187 }
188
189 if (boot.status[0] != 0 && boot.status[0] != 255) {
190 LOGI("Boot status: %.*s\n", sizeof(boot.status), boot.status);
191 }
192
193 // --- if arguments weren't supplied, look in the bootloader control block
194 if (*argc <= 1) {
195 boot.recovery[sizeof(boot.recovery) - 1] = '\0'; // Ensure termination
196 const char *arg = strtok(boot.recovery, "\n");
197 if (arg != NULL && !strcmp(arg, "recovery")) {
198 *argv = (char **) malloc(sizeof(char *) * MAX_ARGS);
199 (*argv)[0] = strdup(arg);
200 for (*argc = 1; *argc < MAX_ARGS; ++*argc) {
201 if ((arg = strtok(NULL, "\n")) == NULL) break;
202 (*argv)[*argc] = strdup(arg);
203 }
204 LOGI("Got arguments from boot message\n");
205 } else if (boot.recovery[0] != 0 && boot.recovery[0] != 255) {
206 LOGE("Bad boot message\n\"%.20s\"\n", boot.recovery);
207 }
208 }
209
210 // --- if that doesn't work, try the command file
211 if (*argc <= 1) {
Doug Zongkerd4208f92010-09-20 12:16:13 -0700212 FILE *fp = fopen_path(COMMAND_FILE, "r");
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800213 if (fp != NULL) {
214 char *argv0 = (*argv)[0];
215 *argv = (char **) malloc(sizeof(char *) * MAX_ARGS);
216 (*argv)[0] = argv0; // use the same program name
217
218 char buf[MAX_ARG_LENGTH];
219 for (*argc = 1; *argc < MAX_ARGS; ++*argc) {
220 if (!fgets(buf, sizeof(buf), fp)) break;
221 (*argv)[*argc] = strdup(strtok(buf, "\r\n")); // Strip newline.
222 }
223
224 check_and_fclose(fp, COMMAND_FILE);
225 LOGI("Got arguments from %s\n", COMMAND_FILE);
226 }
227 }
228
229 // --> write the arguments we have back into the bootloader control block
230 // always boot into recovery after this (until finish_recovery() is called)
231 strlcpy(boot.command, "boot-recovery", sizeof(boot.command));
232 strlcpy(boot.recovery, "recovery\n", sizeof(boot.recovery));
233 int i;
234 for (i = 1; i < *argc; ++i) {
235 strlcat(boot.recovery, (*argv)[i], sizeof(boot.recovery));
236 strlcat(boot.recovery, "\n", sizeof(boot.recovery));
237 }
238 set_bootloader_message(&boot);
239}
240
Doug Zongker34c98df2009-08-18 12:05:45 -0700241static void
Oscar Montemayor05231562009-11-30 08:40:57 -0800242set_sdcard_update_bootloader_message() {
Doug Zongker34c98df2009-08-18 12:05:45 -0700243 struct bootloader_message boot;
244 memset(&boot, 0, sizeof(boot));
245 strlcpy(boot.command, "boot-recovery", sizeof(boot.command));
246 strlcpy(boot.recovery, "recovery\n", sizeof(boot.recovery));
247 set_bootloader_message(&boot);
248}
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800249
Doug Zongker2c3539e2010-09-29 13:21:30 -0700250// How much of the temp log we have copied to the copy in cache.
Dees_Troy7d15c252012-09-05 20:47:21 -0400251//static long tmplog_offset = 0;
Doug Zongker2c3539e2010-09-29 13:21:30 -0700252
253static void
Doug Zongkerd0181b82011-10-19 10:51:12 -0700254copy_log_file(const char* source, const char* destination, int append) {
Doug Zongker2c3539e2010-09-29 13:21:30 -0700255 FILE *log = fopen_path(destination, append ? "a" : "w");
256 if (log == NULL) {
257 LOGE("Can't open %s\n", destination);
258 } else {
Doug Zongkerd0181b82011-10-19 10:51:12 -0700259 FILE *tmplog = fopen(source, "r");
260 if (tmplog != NULL) {
Doug Zongker2c3539e2010-09-29 13:21:30 -0700261 if (append) {
262 fseek(tmplog, tmplog_offset, SEEK_SET); // Since last write
263 }
264 char buf[4096];
265 while (fgets(buf, sizeof(buf), tmplog)) fputs(buf, log);
266 if (append) {
267 tmplog_offset = ftell(tmplog);
268 }
Doug Zongkerd0181b82011-10-19 10:51:12 -0700269 check_and_fclose(tmplog, source);
Doug Zongker2c3539e2010-09-29 13:21:30 -0700270 }
271 check_and_fclose(log, destination);
272 }
273}
274
275
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800276// clear the recovery command and prepare to boot a (hopefully working) system,
277// copy our log file to cache as well (for the system to read), and
278// record any intent we were asked to communicate back to the system.
279// this function is idempotent: call it as many times as you like.
280static void
Oscar Montemayor05231562009-11-30 08:40:57 -0800281finish_recovery(const char *send_intent) {
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800282 // By this point, we're ready to return to the main system...
283 if (send_intent != NULL) {
Doug Zongkerd4208f92010-09-20 12:16:13 -0700284 FILE *fp = fopen_path(INTENT_FILE, "w");
Jay Freeman (saurik)619ec2f2008-11-17 01:56:05 +0000285 if (fp == NULL) {
286 LOGE("Can't open %s\n", INTENT_FILE);
287 } else {
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800288 fputs(send_intent, fp);
289 check_and_fclose(fp, INTENT_FILE);
290 }
291 }
292
293 // Copy logs to cache so the system can find out what happened.
Doug Zongkerd0181b82011-10-19 10:51:12 -0700294 copy_log_file(TEMPORARY_LOG_FILE, LOG_FILE, true);
295 copy_log_file(TEMPORARY_LOG_FILE, LAST_LOG_FILE, false);
296 copy_log_file(TEMPORARY_INSTALL_FILE, LAST_INSTALL_FILE, false);
297 chmod(LOG_FILE, 0600);
298 chown(LOG_FILE, 1000, 1000); // system user
Doug Zongker2c3539e2010-09-29 13:21:30 -0700299 chmod(LAST_LOG_FILE, 0640);
Doug Zongkerd0181b82011-10-19 10:51:12 -0700300 chmod(LAST_INSTALL_FILE, 0644);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800301
Doug Zongkere5d5ac72012-04-12 11:01:22 -0700302 // Reset to normal system boot so recovery won't cycle indefinitely.
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800303 struct bootloader_message boot;
304 memset(&boot, 0, sizeof(boot));
305 set_bootloader_message(&boot);
306
307 // Remove the command file, so recovery won't repeat indefinitely.
Doug Zongkerd4208f92010-09-20 12:16:13 -0700308 if (ensure_path_mounted(COMMAND_FILE) != 0 ||
309 (unlink(COMMAND_FILE) && errno != ENOENT)) {
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800310 LOGW("Can't unlink %s\n", COMMAND_FILE);
311 }
312
Doug Zongkerd0181b82011-10-19 10:51:12 -0700313 ensure_path_unmounted(CACHE_ROOT);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800314 sync(); // For good measure.
315}
316
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800317static int
Doug Zongkerd4208f92010-09-20 12:16:13 -0700318erase_volume(const char *volume) {
Dees_Troy4b20e772013-01-17 18:50:23 +0000319 return !PartitionManager.Wipe_By_Path(volume);
Doug Zongker211aebc2011-10-28 15:13:10 -0700320 ui->SetBackground(RecoveryUI::INSTALLING);
321 ui->SetProgressType(RecoveryUI::INDETERMINATE);
322 ui->Print("Formatting %s...\n", volume);
Doug Zongker2c3539e2010-09-29 13:21:30 -0700323
Doug Zongkerd0181b82011-10-19 10:51:12 -0700324 ensure_path_unmounted(volume);
325
Doug Zongker2c3539e2010-09-29 13:21:30 -0700326 if (strcmp(volume, "/cache") == 0) {
327 // Any part of the log we'd copied to cache is now gone.
328 // Reset the pointer so we copy from the beginning of the temp
329 // log.
330 tmplog_offset = 0;
331 }
332
Doug Zongkerd4208f92010-09-20 12:16:13 -0700333 return format_volume(volume);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800334}
335
Doug Zongker23ceeea2010-07-08 17:27:55 -0700336static char*
Doug Zongkerd4208f92010-09-20 12:16:13 -0700337copy_sideloaded_package(const char* original_path) {
338 if (ensure_path_mounted(original_path) != 0) {
339 LOGE("Can't mount %s\n", original_path);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700340 return NULL;
341 }
342
Doug Zongkerd4208f92010-09-20 12:16:13 -0700343 if (ensure_path_mounted(SIDELOAD_TEMP_DIR) != 0) {
Doug Zongker23ceeea2010-07-08 17:27:55 -0700344 LOGE("Can't mount %s\n", SIDELOAD_TEMP_DIR);
345 return NULL;
346 }
347
Doug Zongkerd4208f92010-09-20 12:16:13 -0700348 if (mkdir(SIDELOAD_TEMP_DIR, 0700) != 0) {
Doug Zongker23ceeea2010-07-08 17:27:55 -0700349 if (errno != EEXIST) {
350 LOGE("Can't mkdir %s (%s)\n", SIDELOAD_TEMP_DIR, strerror(errno));
351 return NULL;
352 }
353 }
354
Doug Zongkerd4208f92010-09-20 12:16:13 -0700355 // verify that SIDELOAD_TEMP_DIR is exactly what we expect: a
356 // directory, owned by root, readable and writable only by root.
Doug Zongker23ceeea2010-07-08 17:27:55 -0700357 struct stat st;
Doug Zongkerd4208f92010-09-20 12:16:13 -0700358 if (stat(SIDELOAD_TEMP_DIR, &st) != 0) {
359 LOGE("failed to stat %s (%s)\n", SIDELOAD_TEMP_DIR, strerror(errno));
Doug Zongker23ceeea2010-07-08 17:27:55 -0700360 return NULL;
361 }
362 if (!S_ISDIR(st.st_mode)) {
Doug Zongkerd4208f92010-09-20 12:16:13 -0700363 LOGE("%s isn't a directory\n", SIDELOAD_TEMP_DIR);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700364 return NULL;
365 }
366 if ((st.st_mode & 0777) != 0700) {
Doug Zongkerd4208f92010-09-20 12:16:13 -0700367 LOGE("%s has perms %o\n", SIDELOAD_TEMP_DIR, st.st_mode);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700368 return NULL;
369 }
370 if (st.st_uid != 0) {
Doug Zongkerd4208f92010-09-20 12:16:13 -0700371 LOGE("%s owned by %lu; not root\n", SIDELOAD_TEMP_DIR, st.st_uid);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700372 return NULL;
373 }
374
Doug Zongkerd4208f92010-09-20 12:16:13 -0700375 char copy_path[PATH_MAX];
376 strcpy(copy_path, SIDELOAD_TEMP_DIR);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700377 strcat(copy_path, "/package.zip");
378
Doug Zongker28ce47c2011-10-28 10:33:05 -0700379 char* buffer = (char*)malloc(BUFSIZ);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700380 if (buffer == NULL) {
381 LOGE("Failed to allocate buffer\n");
382 return NULL;
383 }
384
385 size_t read;
386 FILE* fin = fopen(original_path, "rb");
387 if (fin == NULL) {
388 LOGE("Failed to open %s (%s)\n", original_path, strerror(errno));
389 return NULL;
390 }
391 FILE* fout = fopen(copy_path, "wb");
392 if (fout == NULL) {
393 LOGE("Failed to open %s (%s)\n", copy_path, strerror(errno));
394 return NULL;
395 }
396
397 while ((read = fread(buffer, 1, BUFSIZ, fin)) > 0) {
398 if (fwrite(buffer, 1, read, fout) != read) {
399 LOGE("Short write of %s (%s)\n", copy_path, strerror(errno));
400 return NULL;
401 }
402 }
403
404 free(buffer);
405
406 if (fclose(fout) != 0) {
407 LOGE("Failed to close %s (%s)\n", copy_path, strerror(errno));
408 return NULL;
409 }
410
411 if (fclose(fin) != 0) {
412 LOGE("Failed to close %s (%s)\n", original_path, strerror(errno));
413 return NULL;
414 }
415
416 // "adb push" is happy to overwrite read-only files when it's
417 // running as root, but we'll try anyway.
418 if (chmod(copy_path, 0400) != 0) {
419 LOGE("Failed to chmod %s (%s)\n", copy_path, strerror(errno));
420 return NULL;
421 }
422
Doug Zongkerd4208f92010-09-20 12:16:13 -0700423 return strdup(copy_path);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700424}
425
Doug Zongker28ce47c2011-10-28 10:33:05 -0700426static const char**
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700427prepend_title(const char* const* headers) {
Doug Zongker28ce47c2011-10-28 10:33:05 -0700428 const char* title[] = { "Android system recovery <"
429 EXPAND(RECOVERY_API_VERSION) "e>",
430 "",
431 NULL };
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800432
Doug Zongkerd6837852009-06-17 22:07:13 -0700433 // count the number of lines in our title, plus the
Doug Zongkerf93d8162009-09-22 15:16:02 -0700434 // caller-provided headers.
Doug Zongkerd6837852009-06-17 22:07:13 -0700435 int count = 0;
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700436 const char* const* p;
Doug Zongkerd6837852009-06-17 22:07:13 -0700437 for (p = title; *p; ++p, ++count);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700438 for (p = headers; *p; ++p, ++count);
Doug Zongkerd6837852009-06-17 22:07:13 -0700439
Doug Zongker28ce47c2011-10-28 10:33:05 -0700440 const char** new_headers = (const char**)malloc((count+1) * sizeof(char*));
441 const char** h = new_headers;
Doug Zongkerd6837852009-06-17 22:07:13 -0700442 for (p = title; *p; ++p, ++h) *h = *p;
Doug Zongkerf93d8162009-09-22 15:16:02 -0700443 for (p = headers; *p; ++p, ++h) *h = *p;
Doug Zongkerd6837852009-06-17 22:07:13 -0700444 *h = NULL;
445
Doug Zongkerf93d8162009-09-22 15:16:02 -0700446 return new_headers;
447}
448
449static int
Doug Zongker28ce47c2011-10-28 10:33:05 -0700450get_menu_selection(const char* const * headers, const char* const * items,
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700451 int menu_only, int initial_selection, Device* device) {
Doug Zongkerf93d8162009-09-22 15:16:02 -0700452 // throw away keys pressed previously, so user doesn't
453 // accidentally trigger menu items.
Doug Zongker211aebc2011-10-28 15:13:10 -0700454 ui->FlushKeys();
Doug Zongkerf93d8162009-09-22 15:16:02 -0700455
Doug Zongker211aebc2011-10-28 15:13:10 -0700456 ui->StartMenu(headers, items, initial_selection);
Doug Zongker8674a722010-09-15 11:08:23 -0700457 int selected = initial_selection;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800458 int chosen_item = -1;
459
Doug Zongkerf93d8162009-09-22 15:16:02 -0700460 while (chosen_item < 0) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700461 int key = ui->WaitKey();
462 int visible = ui->IsTextVisible();
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800463
Doug Zongker5cae4452011-01-25 13:15:30 -0800464 if (key == -1) { // ui_wait_key() timed out
Doug Zongker211aebc2011-10-28 15:13:10 -0700465 if (ui->WasTextEverVisible()) {
Doug Zongker5cae4452011-01-25 13:15:30 -0800466 continue;
467 } else {
468 LOGI("timed out waiting for key input; rebooting.\n");
Doug Zongker211aebc2011-10-28 15:13:10 -0700469 ui->EndMenu();
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700470 return 0; // XXX fixme
Doug Zongker5cae4452011-01-25 13:15:30 -0800471 }
472 }
473
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700474 int action = device->HandleMenuKey(key, visible);
Doug Zongkerddd6a282009-06-09 12:22:33 -0700475
476 if (action < 0) {
477 switch (action) {
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700478 case Device::kHighlightUp:
Doug Zongkerddd6a282009-06-09 12:22:33 -0700479 --selected;
Doug Zongker211aebc2011-10-28 15:13:10 -0700480 selected = ui->SelectMenu(selected);
Doug Zongkerddd6a282009-06-09 12:22:33 -0700481 break;
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700482 case Device::kHighlightDown:
Doug Zongkerddd6a282009-06-09 12:22:33 -0700483 ++selected;
Doug Zongker211aebc2011-10-28 15:13:10 -0700484 selected = ui->SelectMenu(selected);
Doug Zongkerddd6a282009-06-09 12:22:33 -0700485 break;
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700486 case Device::kInvokeItem:
Doug Zongkerddd6a282009-06-09 12:22:33 -0700487 chosen_item = selected;
488 break;
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700489 case Device::kNoAction:
Doug Zongkerddd6a282009-06-09 12:22:33 -0700490 break;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800491 }
Doug Zongkerf93d8162009-09-22 15:16:02 -0700492 } else if (!menu_only) {
Doug Zongkerddd6a282009-06-09 12:22:33 -0700493 chosen_item = action;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800494 }
Doug Zongkerf93d8162009-09-22 15:16:02 -0700495 }
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800496
Doug Zongker211aebc2011-10-28 15:13:10 -0700497 ui->EndMenu();
Doug Zongkerf93d8162009-09-22 15:16:02 -0700498 return chosen_item;
499}
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800500
Doug Zongker8674a722010-09-15 11:08:23 -0700501static int compare_string(const void* a, const void* b) {
502 return strcmp(*(const char**)a, *(const char**)b);
503}
504
505static int
Doug Zongkerd0181b82011-10-19 10:51:12 -0700506update_directory(const char* path, const char* unmount_when_done,
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700507 int* wipe_cache, Device* device) {
Michael Ward9d2629c2011-06-23 22:14:24 -0700508 ensure_path_mounted(path);
Doug Zongkerc18eeb82010-09-21 16:49:26 -0700509
Doug Zongker8674a722010-09-15 11:08:23 -0700510 const char* MENU_HEADERS[] = { "Choose a package to install:",
Doug Zongkerd4208f92010-09-20 12:16:13 -0700511 path,
Doug Zongker8674a722010-09-15 11:08:23 -0700512 "",
513 NULL };
514 DIR* d;
515 struct dirent* de;
Doug Zongkerd4208f92010-09-20 12:16:13 -0700516 d = opendir(path);
Doug Zongker8674a722010-09-15 11:08:23 -0700517 if (d == NULL) {
518 LOGE("error opening %s: %s\n", path, strerror(errno));
Michael Ward9d2629c2011-06-23 22:14:24 -0700519 if (unmount_when_done != NULL) {
520 ensure_path_unmounted(unmount_when_done);
521 }
Doug Zongker8674a722010-09-15 11:08:23 -0700522 return 0;
523 }
524
Doug Zongker28ce47c2011-10-28 10:33:05 -0700525 const char** headers = prepend_title(MENU_HEADERS);
Doug Zongker8674a722010-09-15 11:08:23 -0700526
527 int d_size = 0;
528 int d_alloc = 10;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700529 char** dirs = (char**)malloc(d_alloc * sizeof(char*));
Doug Zongker8674a722010-09-15 11:08:23 -0700530 int z_size = 1;
531 int z_alloc = 10;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700532 char** zips = (char**)malloc(z_alloc * sizeof(char*));
Doug Zongker8674a722010-09-15 11:08:23 -0700533 zips[0] = strdup("../");
534
535 while ((de = readdir(d)) != NULL) {
536 int name_len = strlen(de->d_name);
537
538 if (de->d_type == DT_DIR) {
539 // skip "." and ".." entries
540 if (name_len == 1 && de->d_name[0] == '.') continue;
541 if (name_len == 2 && de->d_name[0] == '.' &&
542 de->d_name[1] == '.') continue;
543
544 if (d_size >= d_alloc) {
545 d_alloc *= 2;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700546 dirs = (char**)realloc(dirs, d_alloc * sizeof(char*));
Doug Zongker8674a722010-09-15 11:08:23 -0700547 }
Doug Zongker28ce47c2011-10-28 10:33:05 -0700548 dirs[d_size] = (char*)malloc(name_len + 2);
Doug Zongker8674a722010-09-15 11:08:23 -0700549 strcpy(dirs[d_size], de->d_name);
550 dirs[d_size][name_len] = '/';
551 dirs[d_size][name_len+1] = '\0';
552 ++d_size;
553 } else if (de->d_type == DT_REG &&
554 name_len >= 4 &&
555 strncasecmp(de->d_name + (name_len-4), ".zip", 4) == 0) {
556 if (z_size >= z_alloc) {
557 z_alloc *= 2;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700558 zips = (char**)realloc(zips, z_alloc * sizeof(char*));
Doug Zongker8674a722010-09-15 11:08:23 -0700559 }
560 zips[z_size++] = strdup(de->d_name);
561 }
562 }
563 closedir(d);
564
565 qsort(dirs, d_size, sizeof(char*), compare_string);
566 qsort(zips, z_size, sizeof(char*), compare_string);
567
568 // append dirs to the zips list
569 if (d_size + z_size + 1 > z_alloc) {
570 z_alloc = d_size + z_size + 1;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700571 zips = (char**)realloc(zips, z_alloc * sizeof(char*));
Doug Zongker8674a722010-09-15 11:08:23 -0700572 }
573 memcpy(zips + z_size, dirs, d_size * sizeof(char*));
574 free(dirs);
575 z_size += d_size;
576 zips[z_size] = NULL;
577
578 int result;
579 int chosen_item = 0;
580 do {
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700581 chosen_item = get_menu_selection(headers, zips, 1, chosen_item, device);
Doug Zongker8674a722010-09-15 11:08:23 -0700582
583 char* item = zips[chosen_item];
584 int item_len = strlen(item);
585 if (chosen_item == 0) { // item 0 is always "../"
Michael Ward9d2629c2011-06-23 22:14:24 -0700586 // go up but continue browsing (if the caller is update_directory)
Doug Zongker8674a722010-09-15 11:08:23 -0700587 result = -1;
588 break;
589 } else if (item[item_len-1] == '/') {
590 // recurse down into a subdirectory
591 char new_path[PATH_MAX];
Doug Zongkerd4208f92010-09-20 12:16:13 -0700592 strlcpy(new_path, path, PATH_MAX);
593 strlcat(new_path, "/", PATH_MAX);
Doug Zongker8674a722010-09-15 11:08:23 -0700594 strlcat(new_path, item, PATH_MAX);
Doug Zongkerd4208f92010-09-20 12:16:13 -0700595 new_path[strlen(new_path)-1] = '\0'; // truncate the trailing '/'
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700596 result = update_directory(new_path, unmount_when_done, wipe_cache, device);
Doug Zongker8674a722010-09-15 11:08:23 -0700597 if (result >= 0) break;
598 } else {
599 // selected a zip file: attempt to install it, and return
600 // the status to the caller.
601 char new_path[PATH_MAX];
Doug Zongkerd4208f92010-09-20 12:16:13 -0700602 strlcpy(new_path, path, PATH_MAX);
Doug Zongkerc18eeb82010-09-21 16:49:26 -0700603 strlcat(new_path, "/", PATH_MAX);
Doug Zongker8674a722010-09-15 11:08:23 -0700604 strlcat(new_path, item, PATH_MAX);
605
Doug Zongker211aebc2011-10-28 15:13:10 -0700606 ui->Print("\n-- Install %s ...\n", path);
Doug Zongker8674a722010-09-15 11:08:23 -0700607 set_sdcard_update_bootloader_message();
Doug Zongkerd4208f92010-09-20 12:16:13 -0700608 char* copy = copy_sideloaded_package(new_path);
Michael Ward9d2629c2011-06-23 22:14:24 -0700609 if (unmount_when_done != NULL) {
610 ensure_path_unmounted(unmount_when_done);
611 }
Doug Zongkerd4208f92010-09-20 12:16:13 -0700612 if (copy) {
Doug Zongkerd0181b82011-10-19 10:51:12 -0700613 result = install_package(copy, wipe_cache, TEMPORARY_INSTALL_FILE);
Doug Zongkerd4208f92010-09-20 12:16:13 -0700614 free(copy);
615 } else {
616 result = INSTALL_ERROR;
617 }
Doug Zongker8674a722010-09-15 11:08:23 -0700618 break;
619 }
620 } while (true);
621
622 int i;
623 for (i = 0; i < z_size; ++i) free(zips[i]);
624 free(zips);
625 free(headers);
626
Michael Ward9d2629c2011-06-23 22:14:24 -0700627 if (unmount_when_done != NULL) {
628 ensure_path_unmounted(unmount_when_done);
629 }
Doug Zongker8674a722010-09-15 11:08:23 -0700630 return result;
631}
632
Doug Zongkerf93d8162009-09-22 15:16:02 -0700633static void
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700634wipe_data(int confirm, Device* device) {
Doug Zongkerf93d8162009-09-22 15:16:02 -0700635 if (confirm) {
Doug Zongker28ce47c2011-10-28 10:33:05 -0700636 static const char** title_headers = NULL;
Doug Zongkerddd6a282009-06-09 12:22:33 -0700637
Doug Zongkerf93d8162009-09-22 15:16:02 -0700638 if (title_headers == NULL) {
Doug Zongker28ce47c2011-10-28 10:33:05 -0700639 const char* headers[] = { "Confirm wipe of all user data?",
640 " THIS CAN NOT BE UNDONE.",
641 "",
642 NULL };
Doug Zongker8674a722010-09-15 11:08:23 -0700643 title_headers = prepend_title((const char**)headers);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700644 }
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800645
Doug Zongker28ce47c2011-10-28 10:33:05 -0700646 const char* items[] = { " No",
647 " No",
648 " No",
649 " No",
650 " No",
651 " No",
652 " No",
653 " Yes -- delete all user data", // [7]
654 " No",
655 " No",
656 " No",
657 NULL };
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800658
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700659 int chosen_item = get_menu_selection(title_headers, items, 1, 0, device);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700660 if (chosen_item != 7) {
661 return;
662 }
663 }
Doug Zongker1066d2c2009-04-01 13:57:40 -0700664
Doug Zongker211aebc2011-10-28 15:13:10 -0700665 ui->Print("\n-- Wiping data...\n");
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700666 device->WipeData();
Doug Zongkerd4208f92010-09-20 12:16:13 -0700667 erase_volume("/data");
668 erase_volume("/cache");
Doug Zongker211aebc2011-10-28 15:13:10 -0700669 ui->Print("Data wipe complete.\n");
Doug Zongkerf93d8162009-09-22 15:16:02 -0700670}
671
672static void
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700673prompt_and_wait(Device* device) {
674 const char* const* headers = prepend_title(device->GetMenuHeaders());
Doug Zongkerf93d8162009-09-22 15:16:02 -0700675
676 for (;;) {
677 finish_recovery(NULL);
Doug Zongker211aebc2011-10-28 15:13:10 -0700678 ui->SetProgressType(RecoveryUI::EMPTY);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700679
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700680 int chosen_item = get_menu_selection(headers, device->GetMenuItems(), 0, 0, device);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700681
682 // device-specific code may take some action here. It may
683 // return one of the core actions handled in the switch
684 // statement below.
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700685 chosen_item = device->InvokeMenuItem(chosen_item);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700686
Michael Ward9d2629c2011-06-23 22:14:24 -0700687 int status;
Doug Zongkerd0181b82011-10-19 10:51:12 -0700688 int wipe_cache;
Doug Zongkerf93d8162009-09-22 15:16:02 -0700689 switch (chosen_item) {
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700690 case Device::REBOOT:
Doug Zongkerf93d8162009-09-22 15:16:02 -0700691 return;
692
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700693 case Device::WIPE_DATA:
694 wipe_data(ui->IsTextVisible(), device);
Doug Zongker211aebc2011-10-28 15:13:10 -0700695 if (!ui->IsTextVisible()) return;
Doug Zongkerf93d8162009-09-22 15:16:02 -0700696 break;
697
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700698 case Device::WIPE_CACHE:
Doug Zongker211aebc2011-10-28 15:13:10 -0700699 ui->Print("\n-- Wiping cache...\n");
Doug Zongkerd4208f92010-09-20 12:16:13 -0700700 erase_volume("/cache");
Doug Zongker211aebc2011-10-28 15:13:10 -0700701 ui->Print("Cache wipe complete.\n");
702 if (!ui->IsTextVisible()) return;
Doug Zongkerf93d8162009-09-22 15:16:02 -0700703 break;
704
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700705 case Device::APPLY_EXT:
Doug Zongkerd9428e32011-12-13 16:03:28 -0800706 // Some packages expect /cache to be mounted (eg,
707 // standard incremental packages expect to use /cache
708 // as scratch space).
709 ensure_path_mounted(CACHE_ROOT);
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700710 status = update_directory(SDCARD_ROOT, SDCARD_ROOT, &wipe_cache, device);
Doug Zongkerd0181b82011-10-19 10:51:12 -0700711 if (status == INSTALL_SUCCESS && wipe_cache) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700712 ui->Print("\n-- Wiping cache (at package request)...\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700713 if (erase_volume("/cache")) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700714 ui->Print("Cache wipe failed.\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700715 } else {
Doug Zongker211aebc2011-10-28 15:13:10 -0700716 ui->Print("Cache wipe complete.\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700717 }
718 }
Doug Zongker8674a722010-09-15 11:08:23 -0700719 if (status >= 0) {
720 if (status != INSTALL_SUCCESS) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700721 ui->SetBackground(RecoveryUI::ERROR);
722 ui->Print("Installation aborted.\n");
723 } else if (!ui->IsTextVisible()) {
Doug Zongker8674a722010-09-15 11:08:23 -0700724 return; // reboot if logs aren't visible
725 } else {
Doug Zongker211aebc2011-10-28 15:13:10 -0700726 ui->Print("\nInstall from sdcard complete.\n");
Doug Zongker8674a722010-09-15 11:08:23 -0700727 }
Doug Zongkerf93d8162009-09-22 15:16:02 -0700728 }
729 break;
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700730
731 case Device::APPLY_CACHE:
Michael Ward9d2629c2011-06-23 22:14:24 -0700732 // Don't unmount cache at the end of this.
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700733 status = update_directory(CACHE_ROOT, NULL, &wipe_cache, device);
Doug Zongkerd0181b82011-10-19 10:51:12 -0700734 if (status == INSTALL_SUCCESS && wipe_cache) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700735 ui->Print("\n-- Wiping cache (at package request)...\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700736 if (erase_volume("/cache")) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700737 ui->Print("Cache wipe failed.\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700738 } else {
Doug Zongker211aebc2011-10-28 15:13:10 -0700739 ui->Print("Cache wipe complete.\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700740 }
741 }
Michael Ward9d2629c2011-06-23 22:14:24 -0700742 if (status >= 0) {
743 if (status != INSTALL_SUCCESS) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700744 ui->SetBackground(RecoveryUI::ERROR);
745 ui->Print("Installation aborted.\n");
746 } else if (!ui->IsTextVisible()) {
Michael Ward9d2629c2011-06-23 22:14:24 -0700747 return; // reboot if logs aren't visible
748 } else {
Doug Zongker211aebc2011-10-28 15:13:10 -0700749 ui->Print("\nInstall from cache complete.\n");
Michael Ward9d2629c2011-06-23 22:14:24 -0700750 }
751 }
752 break;
Doug Zongker9270a202012-01-09 15:16:13 -0800753
754 case Device::APPLY_ADB_SIDELOAD:
755 ensure_path_mounted(CACHE_ROOT);
756 status = apply_from_adb(ui, &wipe_cache, TEMPORARY_INSTALL_FILE);
757 if (status >= 0) {
758 if (status != INSTALL_SUCCESS) {
759 ui->SetBackground(RecoveryUI::ERROR);
760 ui->Print("Installation aborted.\n");
761 } else if (!ui->IsTextVisible()) {
762 return; // reboot if logs aren't visible
763 } else {
764 ui->Print("\nInstall from ADB complete.\n");
765 }
766 }
767 break;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800768 }
769 }
770}
771
772static void
Oscar Montemayor05231562009-11-30 08:40:57 -0800773print_property(const char *key, const char *name, void *cookie) {
Doug Zongker56c51052010-07-01 09:18:44 -0700774 printf("%s=%s\n", key, name);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800775}
776
777int
Oscar Montemayor05231562009-11-30 08:40:57 -0800778main(int argc, char **argv) {
Dees_Troycfb63ae2012-09-19 14:30:17 -0400779 // Recovery needs to install world-readable files, so clear umask
780 // set by init
781 umask(0);
782
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800783 time_t start = time(NULL);
784
785 // If these fail, there's not really anywhere to complain...
786 freopen(TEMPORARY_LOG_FILE, "a", stdout); setbuf(stdout, NULL);
787 freopen(TEMPORARY_LOG_FILE, "a", stderr); setbuf(stderr, NULL);
Doug Zongker9270a202012-01-09 15:16:13 -0800788
789 // If this binary is started with the single argument "--adbd",
790 // instead of being the normal recovery binary, it turns into kind
791 // of a stripped-down version of adbd that only supports the
792 // 'sideload' command. Note this must be a real argument, not
793 // anything in the command file or bootloader control block; the
794 // only way recovery should be run with this argument is when it
795 // starts a copy of itself from the apply_from_adb() function.
Dees_Troy9a4b5692012-09-19 15:09:45 -0400796 if (argc == 3 && strcmp(argv[1], "--adbd") == 0) {
797 adb_main(argv[2]);
Doug Zongker9270a202012-01-09 15:16:13 -0800798 return 0;
799 }
800
Dees_Troy51127312012-09-08 13:08:49 -0400801 printf("Starting TWRP %s on %s", TW_VERSION_STR, ctime(&start));
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800802
Dees_Troy32c8eb82012-09-11 15:28:06 -0400803 Device* device = make_device();
804 ui = device->GetUI();
Doug Zongker211aebc2011-10-28 15:13:10 -0700805
Dees_Troy7d15c252012-09-05 20:47:21 -0400806 //ui->Init();
807 //ui->SetBackground(RecoveryUI::NONE);
Dees_Troy7c2dec82012-09-26 09:49:14 -0400808 //load_volume_table();
Dees_Troy7d15c252012-09-05 20:47:21 -0400809
810 // Load default values to set DataManager constants and handle ifdefs
811 DataManager_LoadDefaults();
812 printf("Starting the UI...");
Dees_Troy51127312012-09-08 13:08:49 -0400813 gui_init();
Dees_Troy7d15c252012-09-05 20:47:21 -0400814 printf("=> Linking mtab\n");
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500815 symlink("/proc/mounts", "/etc/mtab");
Dees_Troy7d15c252012-09-05 20:47:21 -0400816 printf("=> Processing recovery.fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400817 if (!PartitionManager.Process_Fstab("/etc/recovery.fstab", 1)) {
Dees_Troy7d15c252012-09-05 20:47:21 -0400818 LOGE("Failing out of recovery due to problem with recovery.fstab.\n");
819 //return -1;
820 }
Dees_Troy8170a922012-09-18 15:40:25 -0400821 PartitionManager.Output_Partition_Logging();
Dees_Troy7d15c252012-09-05 20:47:21 -0400822 // Load up all the resources
823 gui_loadResources();
824
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800825 get_args(&argc, &argv);
826
827 int previous_runs = 0;
828 const char *send_intent = NULL;
829 const char *update_package = NULL;
830 int wipe_data = 0, wipe_cache = 0;
Doug Zongkere5d5ac72012-04-12 11:01:22 -0700831 bool just_exit = false;
Dees_Troy83a3b122012-10-01 14:16:43 -0400832 bool perform_backup = false;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800833
834 int arg;
835 while ((arg = getopt_long(argc, argv, "", OPTIONS, NULL)) != -1) {
836 switch (arg) {
837 case 'p': previous_runs = atoi(optarg); break;
838 case 's': send_intent = optarg; break;
839 case 'u': update_package = optarg; break;
840 case 'w': wipe_data = wipe_cache = 1; break;
841 case 'c': wipe_cache = 1; break;
Doug Zongker211aebc2011-10-28 15:13:10 -0700842 case 't': ui->ShowText(true); break;
Doug Zongkere5d5ac72012-04-12 11:01:22 -0700843 case 'x': just_exit = true; break;
Dees_Troy83a3b122012-10-01 14:16:43 -0400844 case 'n': perform_backup = true; LOGI("nandroid\n"); break;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800845 case '?':
846 LOGE("Invalid command argument\n");
847 continue;
848 }
849 }
850
Stephen Smalley779701d2012-02-09 14:13:23 -0500851#ifdef HAVE_SELINUX
852 struct selinux_opt seopts[] = {
853 { SELABEL_OPT_PATH, "/file_contexts" }
854 };
855
856 sehandle = selabel_open(SELABEL_CTX_FILE, seopts, 1);
857
858 if (!sehandle) {
859 fprintf(stderr, "Warning: No file_contexts\n");
Kenny Root038818c2012-04-08 11:03:01 -0700860 ui->Print("Warning: No file_contexts\n");
Stephen Smalley779701d2012-02-09 14:13:23 -0500861 }
862#endif
863
Dees_Troy7d15c252012-09-05 20:47:21 -0400864 //device->StartRecovery();
Doug Zongkerefa1bab2010-02-01 15:59:12 -0800865
Doug Zongker56c51052010-07-01 09:18:44 -0700866 printf("Command:");
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800867 for (arg = 0; arg < argc; arg++) {
Doug Zongker56c51052010-07-01 09:18:44 -0700868 printf(" \"%s\"", argv[arg]);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800869 }
Doug Zongker9b125b02010-09-22 12:01:37 -0700870 printf("\n");
871
872 if (update_package) {
873 // For backwards compatibility on the cache partition only, if
874 // we're given an old 'root' path "CACHE:foo", change it to
875 // "/cache/foo".
876 if (strncmp(update_package, "CACHE:", 6) == 0) {
877 int len = strlen(update_package) + 10;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700878 char* modified_path = (char*)malloc(len);
Doug Zongker9b125b02010-09-22 12:01:37 -0700879 strlcpy(modified_path, "/cache/", len);
880 strlcat(modified_path, update_package+6, len);
881 printf("(replacing path \"%s\" with \"%s\")\n",
882 update_package, modified_path);
883 update_package = modified_path;
884 }
885 }
886 printf("\n");
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800887
888 property_list(print_property, NULL);
Doug Zongker56c51052010-07-01 09:18:44 -0700889 printf("\n");
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800890
Dees_Troy7d15c252012-09-05 20:47:21 -0400891 // Check for and run startup script if script exists
Dees_Troya58bead2012-09-27 09:49:29 -0400892 TWFunc::check_and_run_script("/sbin/runatboot.sh", "boot");
893 TWFunc::check_and_run_script("/sbin/postrecoveryboot.sh", "boot");
Dees_Troy7d15c252012-09-05 20:47:21 -0400894
895#ifdef TW_INCLUDE_INJECTTWRP
896 // Back up TWRP Ramdisk if needed:
Dees_Troy06b4fe92012-10-16 11:43:20 -0400897 TWPartition* Boot = PartitionManager.Find_Partition_By_Path("/boot");
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500898 string result;
Dees_Troy7d15c252012-09-05 20:47:21 -0400899 LOGI("Backing up TWRP ramdisk...\n");
Dees_Troy06b4fe92012-10-16 11:43:20 -0400900 if (Boot == NULL || Boot->Current_File_System != "emmc")
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500901 TWFunc::Exec_Cmd("injecttwrp --backup /tmp/backup_recovery_ramdisk.img", result);
Dees_Troy06b4fe92012-10-16 11:43:20 -0400902 else {
903 string injectcmd = "injecttwrp --backup /tmp/backup_recovery_ramdisk.img bd=" + Boot->Actual_Block_Device;
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500904 TWFunc::Exec_Cmd(injectcmd, result);
Dees_Troy06b4fe92012-10-16 11:43:20 -0400905 }
Dees_Troy7d15c252012-09-05 20:47:21 -0400906 LOGI("Backup of TWRP ramdisk done.\n");
907#endif
908
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800909 int status = INSTALL_SUCCESS;
910
Dees_Troy83a3b122012-10-01 14:16:43 -0400911 if (perform_backup) {
912 char empt[50];
913 gui_console_only();
914 strcpy(empt, "(Current Date)");
915 DataManager_SetStrValue(TW_BACKUP_NAME, empt);
916 if (OpenRecoveryScript::Backup_Command("BSDCAE") != 0)
917 status = INSTALL_ERROR;
918 }
919 if (status == INSTALL_SUCCESS) { // Prevent other actions if backup failed
Doug Zongker540d57f2011-01-18 13:36:58 -0800920 if (update_package != NULL) {
Dees_Troy7d15c252012-09-05 20:47:21 -0400921 gui_console_only();
Dees_Troy83a3b122012-10-01 14:16:43 -0400922 if (OpenRecoveryScript::Install_Command(update_package) == 0)
923 status = INSTALL_SUCCESS;
924 else
925 status = INSTALL_ERROR;
926 /*
Doug Zongkerd0181b82011-10-19 10:51:12 -0700927 status = install_package(update_package, &wipe_cache, TEMPORARY_INSTALL_FILE);
928 if (status == INSTALL_SUCCESS && wipe_cache) {
929 if (erase_volume("/cache")) {
930 LOGE("Cache wipe (requested by package) failed.");
931 }
932 }
Doug Zongker211aebc2011-10-28 15:13:10 -0700933 if (status != INSTALL_SUCCESS) ui->Print("Installation aborted.\n");
Dees_Troy83a3b122012-10-01 14:16:43 -0400934 */
Doug Zongkerb128f542009-06-18 15:07:14 -0700935 } else if (wipe_data) {
Dees_Troy7d15c252012-09-05 20:47:21 -0400936 gui_console_only();
Dees_Troy83a3b122012-10-01 14:16:43 -0400937 if (!PartitionManager.Factory_Reset()) status = INSTALL_ERROR;
938 /*
939 if (device->WipeData()) status = INSTALL_ERROR;
940 if (erase_volume("/data")) status = INSTALL_ERROR;
941 if (wipe_cache && erase_volume("/cache")) status = INSTALL_ERROR;
942 */
Doug Zongker211aebc2011-10-28 15:13:10 -0700943 if (status != INSTALL_SUCCESS) ui->Print("Data wipe failed.\n");
Doug Zongkerb128f542009-06-18 15:07:14 -0700944 } else if (wipe_cache) {
Dees_Troy7d15c252012-09-05 20:47:21 -0400945 gui_console_only();
Doug Zongkerd4208f92010-09-20 12:16:13 -0700946 if (wipe_cache && erase_volume("/cache")) status = INSTALL_ERROR;
Doug Zongker211aebc2011-10-28 15:13:10 -0700947 if (status != INSTALL_SUCCESS) ui->Print("Cache wipe failed.\n");
Doug Zongkere5d5ac72012-04-12 11:01:22 -0700948 } else if (!just_exit) {
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800949 status = INSTALL_ERROR; // No command specified
950 }
Dees_Troy83a3b122012-10-01 14:16:43 -0400951 }
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800952
Dees_Troy7d15c252012-09-05 20:47:21 -0400953 //if (status != INSTALL_SUCCESS) ui->SetBackground(RecoveryUI::ERROR);
954 if (status != INSTALL_SUCCESS /*|| ui->IsTextVisible()*/) {
Dees_Troy812660f2012-09-20 09:55:17 -0400955 finish_recovery(NULL);
Dees_Troy7d15c252012-09-05 20:47:21 -0400956 DataManager_ReadSettingsFile();
Dees_Troyc7c43412012-10-02 23:03:01 -0400957 if (PartitionManager.Mount_By_Path("/system", false) && TWFunc::Path_Exists("/system/recovery-from-boot.p")) {
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500958 rename("/system/recovery-from-boot.p", "/system/recovery-from-boot.bak");
Dees_Troyc7c43412012-10-02 23:03:01 -0400959 ui_print("Renamed stock recovery file in /system to prevent\nthe stock ROM from replacing TWRP.\n");
960 }
Dees_Troyd0384ef2012-10-12 12:15:42 -0400961 PartitionManager.UnMount_By_Path("/system", false);
Dees_Troy812660f2012-09-20 09:55:17 -0400962 if (DataManager_GetIntValue(TW_IS_ENCRYPTED) == 0 && OpenRecoveryScript::check_for_script_file()) {
963 gui_console_only();
Dees_Troy83a3b122012-10-01 14:16:43 -0400964 if (OpenRecoveryScript::run_script_file() != 0) {
Dees_Troy812660f2012-09-20 09:55:17 -0400965 // There was an error, boot the recovery
966 gui_start();
967 } else {
968 usleep(2000000); // Sleep for 2 seconds before rebooting
969 }
970 } else
971 gui_start();
Dees_Troy7d15c252012-09-05 20:47:21 -0400972 //prompt_and_wait(device);
Doug Zongker8674a722010-09-15 11:08:23 -0700973 }
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800974
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800975 // Otherwise, get ready to boot the main system...
976 finish_recovery(send_intent);
Doug Zongker211aebc2011-10-28 15:13:10 -0700977 ui->Print("Rebooting...\n");
Dees_Troy38bd7602012-09-14 13:33:53 -0400978#ifdef ANDROID_RB_RESTART
Ken Sumrall6e4472a2011-03-07 23:37:27 -0800979 android_reboot(ANDROID_RB_RESTART, 0, 0);
Dees_Troy38bd7602012-09-14 13:33:53 -0400980#else
981 reboot(RB_AUTOBOOT);
982#endif
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800983 return EXIT_SUCCESS;
984}