blob: 43cd9dafe523683dfb09882b345f2f45a6010d25 [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>
Doug Zongker7c3ae452013-05-14 11:03:02 -070018#include <dirent.h>
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080019#include <errno.h>
20#include <fcntl.h>
21#include <getopt.h>
22#include <limits.h>
23#include <linux/input.h>
Doug Zongker7c3ae452013-05-14 11:03:02 -070024#include <stdarg.h>
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080025#include <stdio.h>
26#include <stdlib.h>
27#include <string.h>
Doug Zongker23ceeea2010-07-08 17:27:55 -070028#include <sys/stat.h>
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080029#include <sys/types.h>
30#include <time.h>
31#include <unistd.h>
32
33#include "bootloader.h"
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080034#include "common.h"
35#include "cutils/properties.h"
Ken Sumrall6e4472a2011-03-07 23:37:27 -080036#include "cutils/android_reboot.h"
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080037#include "install.h"
38#include "minui/minui.h"
39#include "minzip/DirUtil.h"
40#include "roots.h"
Doug Zongker28ce47c2011-10-28 10:33:05 -070041#include "ui.h"
Doug Zongker211aebc2011-10-28 15:13:10 -070042#include "screen_ui.h"
Doug Zongkerdaefc1d2011-10-31 09:34:15 -070043#include "device.h"
Doug Zongker9270a202012-01-09 15:16:13 -080044#include "adb_install.h"
45extern "C" {
46#include "minadbd/adb.h"
47}
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080048
Stephen Smalley779701d2012-02-09 14:13:23 -050049struct selabel_handle *sehandle;
50
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080051static const struct option OPTIONS[] = {
52 { "send_intent", required_argument, NULL, 's' },
53 { "update_package", required_argument, NULL, 'u' },
54 { "wipe_data", no_argument, NULL, 'w' },
55 { "wipe_cache", no_argument, NULL, 'c' },
Doug Zongker4bc98062010-09-03 11:00:13 -070056 { "show_text", no_argument, NULL, 't' },
Doug Zongkere5d5ac72012-04-12 11:01:22 -070057 { "just_exit", no_argument, NULL, 'x' },
Doug Zongker02ec6b82012-08-22 17:26:40 -070058 { "locale", required_argument, NULL, 'l' },
Doug Zongkerc87bab12013-11-25 13:53:25 -080059 { "stages", required_argument, NULL, 'g' },
Doug Zongker988500b2009-10-06 14:41:38 -070060 { NULL, 0, NULL, 0 },
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080061};
62
Doug Zongkerda1ebae2013-05-16 11:23:48 -070063#define LAST_LOG_FILE "/cache/recovery/last_log"
64
Doug Zongker6d0d7ac2013-07-09 13:34:55 -070065static const char *CACHE_LOG_DIR = "/cache/recovery";
Doug Zongkerd4208f92010-09-20 12:16:13 -070066static const char *COMMAND_FILE = "/cache/recovery/command";
67static const char *INTENT_FILE = "/cache/recovery/intent";
68static const char *LOG_FILE = "/cache/recovery/log";
Doug Zongkerd0181b82011-10-19 10:51:12 -070069static const char *LAST_INSTALL_FILE = "/cache/recovery/last_install";
Doug Zongker8b240cc2012-08-29 15:19:29 -070070static const char *LOCALE_FILE = "/cache/recovery/last_locale";
Michael Ward9d2629c2011-06-23 22:14:24 -070071static const char *CACHE_ROOT = "/cache";
Doug Zongkerd4208f92010-09-20 12:16:13 -070072static const char *SDCARD_ROOT = "/sdcard";
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080073static const char *TEMPORARY_LOG_FILE = "/tmp/recovery.log";
Doug Zongkerd0181b82011-10-19 10:51:12 -070074static const char *TEMPORARY_INSTALL_FILE = "/tmp/last_install";
Doug Zongkerd4208f92010-09-20 12:16:13 -070075static const char *SIDELOAD_TEMP_DIR = "/tmp/sideload";
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080076
Doug Zongker211aebc2011-10-28 15:13:10 -070077RecoveryUI* ui = NULL;
Doug Zongker02ec6b82012-08-22 17:26:40 -070078char* locale = NULL;
Doug Zongkerc0441d12013-07-31 11:28:24 -070079char recovery_version[PROPERTY_VALUE_MAX+1];
Doug Zongkerc87bab12013-11-25 13:53:25 -080080char* stage = NULL;
Doug Zongker211aebc2011-10-28 15:13:10 -070081
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080082/*
83 * The recovery tool communicates with the main system through /cache files.
84 * /cache/recovery/command - INPUT - command line for tool, one arg per line
85 * /cache/recovery/log - OUTPUT - combined log file from recovery run(s)
86 * /cache/recovery/intent - OUTPUT - intent that was passed in
87 *
88 * The arguments which may be supplied in the recovery.command file:
89 * --send_intent=anystring - write the text out to recovery.intent
Doug Zongkerd4208f92010-09-20 12:16:13 -070090 * --update_package=path - verify install an OTA package file
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080091 * --wipe_data - erase user data (and cache), then reboot
92 * --wipe_cache - wipe cache (but not user data), then reboot
Oscar Montemayor05231562009-11-30 08:40:57 -080093 * --set_encrypted_filesystem=on|off - enables / diasables encrypted fs
Doug Zongkere5d5ac72012-04-12 11:01:22 -070094 * --just_exit - do nothing; exit and reboot
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080095 *
96 * After completing, we remove /cache/recovery/command and reboot.
97 * Arguments may also be supplied in the bootloader control block (BCB).
98 * These important scenarios must be safely restartable at any point:
99 *
100 * FACTORY RESET
101 * 1. user selects "factory reset"
102 * 2. main system writes "--wipe_data" to /cache/recovery/command
103 * 3. main system reboots into recovery
104 * 4. get_args() writes BCB with "boot-recovery" and "--wipe_data"
105 * -- after this, rebooting will restart the erase --
Doug Zongkerd4208f92010-09-20 12:16:13 -0700106 * 5. erase_volume() reformats /data
107 * 6. erase_volume() reformats /cache
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800108 * 7. finish_recovery() erases BCB
109 * -- after this, rebooting will restart the main system --
110 * 8. main() calls reboot() to boot main system
111 *
112 * OTA INSTALL
113 * 1. main system downloads OTA package to /cache/some-filename.zip
Doug Zongker9b125b02010-09-22 12:01:37 -0700114 * 2. main system writes "--update_package=/cache/some-filename.zip"
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800115 * 3. main system reboots into recovery
116 * 4. get_args() writes BCB with "boot-recovery" and "--update_package=..."
117 * -- after this, rebooting will attempt to reinstall the update --
118 * 5. install_package() attempts to install the update
119 * NOTE: the package install must itself be restartable from any point
120 * 6. finish_recovery() erases BCB
121 * -- after this, rebooting will (try to) restart the main system --
122 * 7. ** if install failed **
123 * 7a. prompt_and_wait() shows an error icon and waits for the user
124 * 7b; the user reboots (pulling the battery, etc) into the main system
125 * 8. main() calls maybe_install_firmware_update()
126 * ** if the update contained radio/hboot firmware **:
127 * 8a. m_i_f_u() writes BCB with "boot-recovery" and "--wipe_cache"
128 * -- after this, rebooting will reformat cache & restart main system --
129 * 8b. m_i_f_u() writes firmware image into raw cache partition
130 * 8c. m_i_f_u() writes BCB with "update-radio/hboot" and "--wipe_cache"
131 * -- after this, rebooting will attempt to reinstall firmware --
132 * 8d. bootloader tries to flash firmware
133 * 8e. bootloader writes BCB with "boot-recovery" (keeping "--wipe_cache")
134 * -- after this, rebooting will reformat cache & restart main system --
Doug Zongkerd4208f92010-09-20 12:16:13 -0700135 * 8f. erase_volume() reformats /cache
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800136 * 8g. finish_recovery() erases BCB
137 * -- after this, rebooting will (try to) restart the main system --
138 * 9. main() calls reboot() to boot main system
139 */
140
141static const int MAX_ARG_LENGTH = 4096;
142static const int MAX_ARGS = 100;
143
Doug Zongkerd4208f92010-09-20 12:16:13 -0700144// open a given path, mounting partitions as necessary
Doug Zongker469243e2011-04-12 09:28:10 -0700145FILE*
Doug Zongkerd4208f92010-09-20 12:16:13 -0700146fopen_path(const char *path, const char *mode) {
147 if (ensure_path_mounted(path) != 0) {
148 LOGE("Can't mount %s\n", path);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800149 return NULL;
150 }
151
152 // When writing, try to create the containing directory, if necessary.
153 // Use generous permissions, the system (init.rc) will reset them.
Stephen Smalley779701d2012-02-09 14:13:23 -0500154 if (strchr("wa", mode[0])) dirCreateHierarchy(path, 0777, NULL, 1, sehandle);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800155
156 FILE *fp = fopen(path, mode);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800157 return fp;
158}
159
160// close a file, log an error if the error indicator is set
161static void
162check_and_fclose(FILE *fp, const char *name) {
163 fflush(fp);
164 if (ferror(fp)) LOGE("Error in %s\n(%s)\n", name, strerror(errno));
165 fclose(fp);
166}
167
168// command line args come from, in decreasing precedence:
169// - the actual command line
170// - the bootloader control block (one per line, after "recovery")
171// - the contents of COMMAND_FILE (one per line)
172static void
173get_args(int *argc, char ***argv) {
174 struct bootloader_message boot;
175 memset(&boot, 0, sizeof(boot));
176 get_bootloader_message(&boot); // this may fail, leaving a zeroed structure
Doug Zongkerc87bab12013-11-25 13:53:25 -0800177 stage = strndup(boot.stage, sizeof(boot.stage));
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800178
179 if (boot.command[0] != 0 && boot.command[0] != 255) {
180 LOGI("Boot command: %.*s\n", sizeof(boot.command), boot.command);
181 }
182
183 if (boot.status[0] != 0 && boot.status[0] != 255) {
184 LOGI("Boot status: %.*s\n", sizeof(boot.status), boot.status);
185 }
186
187 // --- if arguments weren't supplied, look in the bootloader control block
188 if (*argc <= 1) {
189 boot.recovery[sizeof(boot.recovery) - 1] = '\0'; // Ensure termination
190 const char *arg = strtok(boot.recovery, "\n");
191 if (arg != NULL && !strcmp(arg, "recovery")) {
192 *argv = (char **) malloc(sizeof(char *) * MAX_ARGS);
193 (*argv)[0] = strdup(arg);
194 for (*argc = 1; *argc < MAX_ARGS; ++*argc) {
195 if ((arg = strtok(NULL, "\n")) == NULL) break;
196 (*argv)[*argc] = strdup(arg);
197 }
198 LOGI("Got arguments from boot message\n");
199 } else if (boot.recovery[0] != 0 && boot.recovery[0] != 255) {
200 LOGE("Bad boot message\n\"%.20s\"\n", boot.recovery);
201 }
202 }
203
204 // --- if that doesn't work, try the command file
205 if (*argc <= 1) {
Doug Zongkerd4208f92010-09-20 12:16:13 -0700206 FILE *fp = fopen_path(COMMAND_FILE, "r");
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800207 if (fp != NULL) {
Jin Feng93ffa752013-06-04 17:46:24 +0800208 char *token;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800209 char *argv0 = (*argv)[0];
210 *argv = (char **) malloc(sizeof(char *) * MAX_ARGS);
211 (*argv)[0] = argv0; // use the same program name
212
213 char buf[MAX_ARG_LENGTH];
214 for (*argc = 1; *argc < MAX_ARGS; ++*argc) {
215 if (!fgets(buf, sizeof(buf), fp)) break;
Jin Feng93ffa752013-06-04 17:46:24 +0800216 token = strtok(buf, "\r\n");
217 if (token != NULL) {
218 (*argv)[*argc] = strdup(token); // Strip newline.
219 } else {
220 --*argc;
221 }
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800222 }
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.
251static long tmplog_offset = 0;
252
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
Doug Zongkerda1ebae2013-05-16 11:23:48 -0700275// Rename last_log -> last_log.1 -> last_log.2 -> ... -> last_log.$max
276// Overwrites any existing last_log.$max.
277static void
278rotate_last_logs(int max) {
279 char oldfn[256];
280 char newfn[256];
281
282 int i;
283 for (i = max-1; i >= 0; --i) {
284 snprintf(oldfn, sizeof(oldfn), (i==0) ? LAST_LOG_FILE : (LAST_LOG_FILE ".%d"), i);
285 snprintf(newfn, sizeof(newfn), LAST_LOG_FILE ".%d", i+1);
286 // ignore errors
287 rename(oldfn, newfn);
288 }
289}
Doug Zongker2c3539e2010-09-29 13:21:30 -0700290
Doug Zongkerf24fd7e2013-07-02 11:43:25 -0700291static void
292copy_logs() {
293 // Copy logs to cache so the system can find out what happened.
294 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
299 chmod(LAST_LOG_FILE, 0640);
300 chmod(LAST_INSTALL_FILE, 0644);
301 sync();
302}
303
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800304// clear the recovery command and prepare to boot a (hopefully working) system,
305// copy our log file to cache as well (for the system to read), and
306// record any intent we were asked to communicate back to the system.
307// this function is idempotent: call it as many times as you like.
308static void
Oscar Montemayor05231562009-11-30 08:40:57 -0800309finish_recovery(const char *send_intent) {
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800310 // By this point, we're ready to return to the main system...
311 if (send_intent != NULL) {
Doug Zongkerd4208f92010-09-20 12:16:13 -0700312 FILE *fp = fopen_path(INTENT_FILE, "w");
Jay Freeman (saurik)619ec2f2008-11-17 01:56:05 +0000313 if (fp == NULL) {
314 LOGE("Can't open %s\n", INTENT_FILE);
315 } else {
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800316 fputs(send_intent, fp);
317 check_and_fclose(fp, INTENT_FILE);
318 }
319 }
320
Doug Zongker4f33e552012-08-23 13:16:12 -0700321 // Save the locale to cache, so if recovery is next started up
322 // without a --locale argument (eg, directly from the bootloader)
323 // it will use the last-known locale.
324 if (locale != NULL) {
325 LOGI("Saving locale \"%s\"\n", locale);
326 FILE* fp = fopen_path(LOCALE_FILE, "w");
327 fwrite(locale, 1, strlen(locale), fp);
328 fflush(fp);
329 fsync(fileno(fp));
330 check_and_fclose(fp, LOCALE_FILE);
331 }
332
Doug Zongkerf24fd7e2013-07-02 11:43:25 -0700333 copy_logs();
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800334
Doug Zongkere5d5ac72012-04-12 11:01:22 -0700335 // Reset to normal system boot so recovery won't cycle indefinitely.
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800336 struct bootloader_message boot;
337 memset(&boot, 0, sizeof(boot));
338 set_bootloader_message(&boot);
339
340 // Remove the command file, so recovery won't repeat indefinitely.
Doug Zongkerd4208f92010-09-20 12:16:13 -0700341 if (ensure_path_mounted(COMMAND_FILE) != 0 ||
342 (unlink(COMMAND_FILE) && errno != ENOENT)) {
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800343 LOGW("Can't unlink %s\n", COMMAND_FILE);
344 }
345
Doug Zongkerd0181b82011-10-19 10:51:12 -0700346 ensure_path_unmounted(CACHE_ROOT);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800347 sync(); // For good measure.
348}
349
Doug Zongker6d0d7ac2013-07-09 13:34:55 -0700350typedef struct _saved_log_file {
351 char* name;
352 struct stat st;
353 unsigned char* data;
354 struct _saved_log_file* next;
355} saved_log_file;
356
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800357static int
Doug Zongkerd4208f92010-09-20 12:16:13 -0700358erase_volume(const char *volume) {
Doug Zongker6d0d7ac2013-07-09 13:34:55 -0700359 bool is_cache = (strcmp(volume, CACHE_ROOT) == 0);
360
Doug Zongker02ec6b82012-08-22 17:26:40 -0700361 ui->SetBackground(RecoveryUI::ERASING);
Doug Zongker211aebc2011-10-28 15:13:10 -0700362 ui->SetProgressType(RecoveryUI::INDETERMINATE);
Doug Zongker6d0d7ac2013-07-09 13:34:55 -0700363
364 saved_log_file* head = NULL;
365
366 if (is_cache) {
367 // If we're reformatting /cache, we load any
368 // "/cache/recovery/last*" files into memory, so we can restore
369 // them after the reformat.
370
371 ensure_path_mounted(volume);
372
373 DIR* d;
374 struct dirent* de;
375 d = opendir(CACHE_LOG_DIR);
376 if (d) {
377 char path[PATH_MAX];
378 strcpy(path, CACHE_LOG_DIR);
379 strcat(path, "/");
380 int path_len = strlen(path);
381 while ((de = readdir(d)) != NULL) {
382 if (strncmp(de->d_name, "last", 4) == 0) {
383 saved_log_file* p = (saved_log_file*) malloc(sizeof(saved_log_file));
384 strcpy(path+path_len, de->d_name);
385 p->name = strdup(path);
386 if (stat(path, &(p->st)) == 0) {
387 // truncate files to 512kb
388 if (p->st.st_size > (1 << 19)) {
389 p->st.st_size = 1 << 19;
390 }
391 p->data = (unsigned char*) malloc(p->st.st_size);
392 FILE* f = fopen(path, "rb");
393 fread(p->data, 1, p->st.st_size, f);
394 fclose(f);
395 p->next = head;
396 head = p;
397 } else {
398 free(p);
399 }
400 }
401 }
402 closedir(d);
403 } else {
404 if (errno != ENOENT) {
405 printf("opendir failed: %s\n", strerror(errno));
406 }
407 }
408 }
409
Doug Zongker211aebc2011-10-28 15:13:10 -0700410 ui->Print("Formatting %s...\n", volume);
Doug Zongker2c3539e2010-09-29 13:21:30 -0700411
Doug Zongkerd0181b82011-10-19 10:51:12 -0700412 ensure_path_unmounted(volume);
Doug Zongker6d0d7ac2013-07-09 13:34:55 -0700413 int result = format_volume(volume);
Doug Zongkerd0181b82011-10-19 10:51:12 -0700414
Doug Zongker6d0d7ac2013-07-09 13:34:55 -0700415 if (is_cache) {
416 while (head) {
417 FILE* f = fopen_path(head->name, "wb");
418 if (f) {
419 fwrite(head->data, 1, head->st.st_size, f);
420 fclose(f);
421 chmod(head->name, head->st.st_mode);
422 chown(head->name, head->st.st_uid, head->st.st_gid);
423 }
424 free(head->name);
425 free(head->data);
426 saved_log_file* temp = head->next;
427 free(head);
428 head = temp;
429 }
430
Doug Zongker2c3539e2010-09-29 13:21:30 -0700431 // Any part of the log we'd copied to cache is now gone.
432 // Reset the pointer so we copy from the beginning of the temp
433 // log.
434 tmplog_offset = 0;
Doug Zongker6d0d7ac2013-07-09 13:34:55 -0700435 copy_logs();
Doug Zongker2c3539e2010-09-29 13:21:30 -0700436 }
437
Doug Zongker6d0d7ac2013-07-09 13:34:55 -0700438 return result;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800439}
440
Doug Zongker23ceeea2010-07-08 17:27:55 -0700441static char*
Doug Zongkerd4208f92010-09-20 12:16:13 -0700442copy_sideloaded_package(const char* original_path) {
443 if (ensure_path_mounted(original_path) != 0) {
444 LOGE("Can't mount %s\n", original_path);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700445 return NULL;
446 }
447
Doug Zongkerd4208f92010-09-20 12:16:13 -0700448 if (ensure_path_mounted(SIDELOAD_TEMP_DIR) != 0) {
Doug Zongker23ceeea2010-07-08 17:27:55 -0700449 LOGE("Can't mount %s\n", SIDELOAD_TEMP_DIR);
450 return NULL;
451 }
452
Doug Zongkerd4208f92010-09-20 12:16:13 -0700453 if (mkdir(SIDELOAD_TEMP_DIR, 0700) != 0) {
Doug Zongker23ceeea2010-07-08 17:27:55 -0700454 if (errno != EEXIST) {
455 LOGE("Can't mkdir %s (%s)\n", SIDELOAD_TEMP_DIR, strerror(errno));
456 return NULL;
457 }
458 }
459
Doug Zongkerd4208f92010-09-20 12:16:13 -0700460 // verify that SIDELOAD_TEMP_DIR is exactly what we expect: a
461 // directory, owned by root, readable and writable only by root.
Doug Zongker23ceeea2010-07-08 17:27:55 -0700462 struct stat st;
Doug Zongkerd4208f92010-09-20 12:16:13 -0700463 if (stat(SIDELOAD_TEMP_DIR, &st) != 0) {
464 LOGE("failed to stat %s (%s)\n", SIDELOAD_TEMP_DIR, strerror(errno));
Doug Zongker23ceeea2010-07-08 17:27:55 -0700465 return NULL;
466 }
467 if (!S_ISDIR(st.st_mode)) {
Doug Zongkerd4208f92010-09-20 12:16:13 -0700468 LOGE("%s isn't a directory\n", SIDELOAD_TEMP_DIR);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700469 return NULL;
470 }
471 if ((st.st_mode & 0777) != 0700) {
Doug Zongkerd4208f92010-09-20 12:16:13 -0700472 LOGE("%s has perms %o\n", SIDELOAD_TEMP_DIR, st.st_mode);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700473 return NULL;
474 }
475 if (st.st_uid != 0) {
Doug Zongkerd4208f92010-09-20 12:16:13 -0700476 LOGE("%s owned by %lu; not root\n", SIDELOAD_TEMP_DIR, st.st_uid);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700477 return NULL;
478 }
479
Doug Zongkerd4208f92010-09-20 12:16:13 -0700480 char copy_path[PATH_MAX];
481 strcpy(copy_path, SIDELOAD_TEMP_DIR);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700482 strcat(copy_path, "/package.zip");
483
Doug Zongker28ce47c2011-10-28 10:33:05 -0700484 char* buffer = (char*)malloc(BUFSIZ);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700485 if (buffer == NULL) {
486 LOGE("Failed to allocate buffer\n");
487 return NULL;
488 }
489
490 size_t read;
491 FILE* fin = fopen(original_path, "rb");
492 if (fin == NULL) {
493 LOGE("Failed to open %s (%s)\n", original_path, strerror(errno));
494 return NULL;
495 }
496 FILE* fout = fopen(copy_path, "wb");
497 if (fout == NULL) {
498 LOGE("Failed to open %s (%s)\n", copy_path, strerror(errno));
499 return NULL;
500 }
501
502 while ((read = fread(buffer, 1, BUFSIZ, fin)) > 0) {
503 if (fwrite(buffer, 1, read, fout) != read) {
504 LOGE("Short write of %s (%s)\n", copy_path, strerror(errno));
505 return NULL;
506 }
507 }
508
509 free(buffer);
510
511 if (fclose(fout) != 0) {
512 LOGE("Failed to close %s (%s)\n", copy_path, strerror(errno));
513 return NULL;
514 }
515
516 if (fclose(fin) != 0) {
517 LOGE("Failed to close %s (%s)\n", original_path, strerror(errno));
518 return NULL;
519 }
520
521 // "adb push" is happy to overwrite read-only files when it's
522 // running as root, but we'll try anyway.
523 if (chmod(copy_path, 0400) != 0) {
524 LOGE("Failed to chmod %s (%s)\n", copy_path, strerror(errno));
525 return NULL;
526 }
527
Doug Zongkerd4208f92010-09-20 12:16:13 -0700528 return strdup(copy_path);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700529}
530
Doug Zongker28ce47c2011-10-28 10:33:05 -0700531static const char**
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700532prepend_title(const char* const* headers) {
Doug Zongkerd6837852009-06-17 22:07:13 -0700533 // count the number of lines in our title, plus the
Doug Zongkerf93d8162009-09-22 15:16:02 -0700534 // caller-provided headers.
Doug Zongkerc0441d12013-07-31 11:28:24 -0700535 int count = 3; // our title has 3 lines
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700536 const char* const* p;
Doug Zongkerf93d8162009-09-22 15:16:02 -0700537 for (p = headers; *p; ++p, ++count);
Doug Zongkerd6837852009-06-17 22:07:13 -0700538
Doug Zongker28ce47c2011-10-28 10:33:05 -0700539 const char** new_headers = (const char**)malloc((count+1) * sizeof(char*));
540 const char** h = new_headers;
Doug Zongkerc0441d12013-07-31 11:28:24 -0700541 *(h++) = "Android system recovery <" EXPAND(RECOVERY_API_VERSION) "e>";
542 *(h++) = recovery_version;
543 *(h++) = "";
Doug Zongkerf93d8162009-09-22 15:16:02 -0700544 for (p = headers; *p; ++p, ++h) *h = *p;
Doug Zongkerd6837852009-06-17 22:07:13 -0700545 *h = NULL;
546
Doug Zongkerf93d8162009-09-22 15:16:02 -0700547 return new_headers;
548}
549
550static int
Doug Zongker28ce47c2011-10-28 10:33:05 -0700551get_menu_selection(const char* const * headers, const char* const * items,
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700552 int menu_only, int initial_selection, Device* device) {
Doug Zongkerf93d8162009-09-22 15:16:02 -0700553 // throw away keys pressed previously, so user doesn't
554 // accidentally trigger menu items.
Doug Zongker211aebc2011-10-28 15:13:10 -0700555 ui->FlushKeys();
Doug Zongkerf93d8162009-09-22 15:16:02 -0700556
Doug Zongker211aebc2011-10-28 15:13:10 -0700557 ui->StartMenu(headers, items, initial_selection);
Doug Zongker8674a722010-09-15 11:08:23 -0700558 int selected = initial_selection;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800559 int chosen_item = -1;
560
Doug Zongkerf93d8162009-09-22 15:16:02 -0700561 while (chosen_item < 0) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700562 int key = ui->WaitKey();
563 int visible = ui->IsTextVisible();
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800564
Doug Zongker5cae4452011-01-25 13:15:30 -0800565 if (key == -1) { // ui_wait_key() timed out
Doug Zongker211aebc2011-10-28 15:13:10 -0700566 if (ui->WasTextEverVisible()) {
Doug Zongker5cae4452011-01-25 13:15:30 -0800567 continue;
568 } else {
569 LOGI("timed out waiting for key input; rebooting.\n");
Doug Zongker211aebc2011-10-28 15:13:10 -0700570 ui->EndMenu();
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700571 return 0; // XXX fixme
Doug Zongker5cae4452011-01-25 13:15:30 -0800572 }
573 }
574
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700575 int action = device->HandleMenuKey(key, visible);
Doug Zongkerddd6a282009-06-09 12:22:33 -0700576
577 if (action < 0) {
578 switch (action) {
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700579 case Device::kHighlightUp:
Doug Zongkerddd6a282009-06-09 12:22:33 -0700580 --selected;
Doug Zongker211aebc2011-10-28 15:13:10 -0700581 selected = ui->SelectMenu(selected);
Doug Zongkerddd6a282009-06-09 12:22:33 -0700582 break;
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700583 case Device::kHighlightDown:
Doug Zongkerddd6a282009-06-09 12:22:33 -0700584 ++selected;
Doug Zongker211aebc2011-10-28 15:13:10 -0700585 selected = ui->SelectMenu(selected);
Doug Zongkerddd6a282009-06-09 12:22:33 -0700586 break;
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700587 case Device::kInvokeItem:
Doug Zongkerddd6a282009-06-09 12:22:33 -0700588 chosen_item = selected;
589 break;
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700590 case Device::kNoAction:
Doug Zongkerddd6a282009-06-09 12:22:33 -0700591 break;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800592 }
Doug Zongkerf93d8162009-09-22 15:16:02 -0700593 } else if (!menu_only) {
Doug Zongkerddd6a282009-06-09 12:22:33 -0700594 chosen_item = action;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800595 }
Doug Zongkerf93d8162009-09-22 15:16:02 -0700596 }
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800597
Doug Zongker211aebc2011-10-28 15:13:10 -0700598 ui->EndMenu();
Doug Zongkerf93d8162009-09-22 15:16:02 -0700599 return chosen_item;
600}
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800601
Doug Zongker8674a722010-09-15 11:08:23 -0700602static int compare_string(const void* a, const void* b) {
603 return strcmp(*(const char**)a, *(const char**)b);
604}
605
606static int
Doug Zongkerd0181b82011-10-19 10:51:12 -0700607update_directory(const char* path, const char* unmount_when_done,
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700608 int* wipe_cache, Device* device) {
Michael Ward9d2629c2011-06-23 22:14:24 -0700609 ensure_path_mounted(path);
Doug Zongkerc18eeb82010-09-21 16:49:26 -0700610
Doug Zongker8674a722010-09-15 11:08:23 -0700611 const char* MENU_HEADERS[] = { "Choose a package to install:",
Doug Zongkerd4208f92010-09-20 12:16:13 -0700612 path,
Doug Zongker8674a722010-09-15 11:08:23 -0700613 "",
614 NULL };
615 DIR* d;
616 struct dirent* de;
Doug Zongkerd4208f92010-09-20 12:16:13 -0700617 d = opendir(path);
Doug Zongker8674a722010-09-15 11:08:23 -0700618 if (d == NULL) {
619 LOGE("error opening %s: %s\n", path, strerror(errno));
Michael Ward9d2629c2011-06-23 22:14:24 -0700620 if (unmount_when_done != NULL) {
621 ensure_path_unmounted(unmount_when_done);
622 }
Doug Zongker8674a722010-09-15 11:08:23 -0700623 return 0;
624 }
625
Doug Zongker28ce47c2011-10-28 10:33:05 -0700626 const char** headers = prepend_title(MENU_HEADERS);
Doug Zongker8674a722010-09-15 11:08:23 -0700627
628 int d_size = 0;
629 int d_alloc = 10;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700630 char** dirs = (char**)malloc(d_alloc * sizeof(char*));
Doug Zongker8674a722010-09-15 11:08:23 -0700631 int z_size = 1;
632 int z_alloc = 10;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700633 char** zips = (char**)malloc(z_alloc * sizeof(char*));
Doug Zongker8674a722010-09-15 11:08:23 -0700634 zips[0] = strdup("../");
635
636 while ((de = readdir(d)) != NULL) {
637 int name_len = strlen(de->d_name);
638
639 if (de->d_type == DT_DIR) {
640 // skip "." and ".." entries
641 if (name_len == 1 && de->d_name[0] == '.') continue;
642 if (name_len == 2 && de->d_name[0] == '.' &&
643 de->d_name[1] == '.') continue;
644
645 if (d_size >= d_alloc) {
646 d_alloc *= 2;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700647 dirs = (char**)realloc(dirs, d_alloc * sizeof(char*));
Doug Zongker8674a722010-09-15 11:08:23 -0700648 }
Doug Zongker28ce47c2011-10-28 10:33:05 -0700649 dirs[d_size] = (char*)malloc(name_len + 2);
Doug Zongker8674a722010-09-15 11:08:23 -0700650 strcpy(dirs[d_size], de->d_name);
651 dirs[d_size][name_len] = '/';
652 dirs[d_size][name_len+1] = '\0';
653 ++d_size;
654 } else if (de->d_type == DT_REG &&
655 name_len >= 4 &&
656 strncasecmp(de->d_name + (name_len-4), ".zip", 4) == 0) {
657 if (z_size >= z_alloc) {
658 z_alloc *= 2;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700659 zips = (char**)realloc(zips, z_alloc * sizeof(char*));
Doug Zongker8674a722010-09-15 11:08:23 -0700660 }
661 zips[z_size++] = strdup(de->d_name);
662 }
663 }
664 closedir(d);
665
666 qsort(dirs, d_size, sizeof(char*), compare_string);
667 qsort(zips, z_size, sizeof(char*), compare_string);
668
669 // append dirs to the zips list
670 if (d_size + z_size + 1 > z_alloc) {
671 z_alloc = d_size + z_size + 1;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700672 zips = (char**)realloc(zips, z_alloc * sizeof(char*));
Doug Zongker8674a722010-09-15 11:08:23 -0700673 }
674 memcpy(zips + z_size, dirs, d_size * sizeof(char*));
675 free(dirs);
676 z_size += d_size;
677 zips[z_size] = NULL;
678
679 int result;
680 int chosen_item = 0;
681 do {
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700682 chosen_item = get_menu_selection(headers, zips, 1, chosen_item, device);
Doug Zongker8674a722010-09-15 11:08:23 -0700683
684 char* item = zips[chosen_item];
685 int item_len = strlen(item);
686 if (chosen_item == 0) { // item 0 is always "../"
Michael Ward9d2629c2011-06-23 22:14:24 -0700687 // go up but continue browsing (if the caller is update_directory)
Doug Zongker8674a722010-09-15 11:08:23 -0700688 result = -1;
689 break;
690 } else if (item[item_len-1] == '/') {
691 // recurse down into a subdirectory
692 char new_path[PATH_MAX];
Doug Zongkerd4208f92010-09-20 12:16:13 -0700693 strlcpy(new_path, path, PATH_MAX);
694 strlcat(new_path, "/", PATH_MAX);
Doug Zongker8674a722010-09-15 11:08:23 -0700695 strlcat(new_path, item, PATH_MAX);
Doug Zongkerd4208f92010-09-20 12:16:13 -0700696 new_path[strlen(new_path)-1] = '\0'; // truncate the trailing '/'
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700697 result = update_directory(new_path, unmount_when_done, wipe_cache, device);
Doug Zongker8674a722010-09-15 11:08:23 -0700698 if (result >= 0) break;
699 } else {
700 // selected a zip file: attempt to install it, and return
701 // the status to the caller.
702 char new_path[PATH_MAX];
Doug Zongkerd4208f92010-09-20 12:16:13 -0700703 strlcpy(new_path, path, PATH_MAX);
Doug Zongkerc18eeb82010-09-21 16:49:26 -0700704 strlcat(new_path, "/", PATH_MAX);
Doug Zongker8674a722010-09-15 11:08:23 -0700705 strlcat(new_path, item, PATH_MAX);
706
Doug Zongker211aebc2011-10-28 15:13:10 -0700707 ui->Print("\n-- Install %s ...\n", path);
Doug Zongker8674a722010-09-15 11:08:23 -0700708 set_sdcard_update_bootloader_message();
Doug Zongkerd4208f92010-09-20 12:16:13 -0700709 char* copy = copy_sideloaded_package(new_path);
Michael Ward9d2629c2011-06-23 22:14:24 -0700710 if (unmount_when_done != NULL) {
711 ensure_path_unmounted(unmount_when_done);
712 }
Doug Zongkerd4208f92010-09-20 12:16:13 -0700713 if (copy) {
Doug Zongkerd0181b82011-10-19 10:51:12 -0700714 result = install_package(copy, wipe_cache, TEMPORARY_INSTALL_FILE);
Doug Zongkerd4208f92010-09-20 12:16:13 -0700715 free(copy);
716 } else {
717 result = INSTALL_ERROR;
718 }
Doug Zongker8674a722010-09-15 11:08:23 -0700719 break;
720 }
721 } while (true);
722
723 int i;
724 for (i = 0; i < z_size; ++i) free(zips[i]);
725 free(zips);
726 free(headers);
727
Michael Ward9d2629c2011-06-23 22:14:24 -0700728 if (unmount_when_done != NULL) {
729 ensure_path_unmounted(unmount_when_done);
730 }
Doug Zongker8674a722010-09-15 11:08:23 -0700731 return result;
732}
733
Doug Zongkerf93d8162009-09-22 15:16:02 -0700734static void
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700735wipe_data(int confirm, Device* device) {
Doug Zongkerf93d8162009-09-22 15:16:02 -0700736 if (confirm) {
Doug Zongker28ce47c2011-10-28 10:33:05 -0700737 static const char** title_headers = NULL;
Doug Zongkerddd6a282009-06-09 12:22:33 -0700738
Doug Zongkerf93d8162009-09-22 15:16:02 -0700739 if (title_headers == NULL) {
Doug Zongker28ce47c2011-10-28 10:33:05 -0700740 const char* headers[] = { "Confirm wipe of all user data?",
741 " THIS CAN NOT BE UNDONE.",
742 "",
743 NULL };
Doug Zongker8674a722010-09-15 11:08:23 -0700744 title_headers = prepend_title((const char**)headers);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700745 }
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800746
Doug Zongker28ce47c2011-10-28 10:33:05 -0700747 const char* items[] = { " No",
748 " No",
749 " No",
750 " No",
751 " No",
752 " No",
753 " No",
754 " Yes -- delete all user data", // [7]
755 " No",
756 " No",
757 " No",
758 NULL };
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800759
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700760 int chosen_item = get_menu_selection(title_headers, items, 1, 0, device);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700761 if (chosen_item != 7) {
762 return;
763 }
764 }
Doug Zongker1066d2c2009-04-01 13:57:40 -0700765
Doug Zongker211aebc2011-10-28 15:13:10 -0700766 ui->Print("\n-- Wiping data...\n");
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700767 device->WipeData();
Doug Zongkerd4208f92010-09-20 12:16:13 -0700768 erase_volume("/data");
769 erase_volume("/cache");
Doug Zongker211aebc2011-10-28 15:13:10 -0700770 ui->Print("Data wipe complete.\n");
Doug Zongkerf93d8162009-09-22 15:16:02 -0700771}
772
773static void
Doug Zongker6c8553d2012-09-24 10:40:47 -0700774prompt_and_wait(Device* device, int status) {
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700775 const char* const* headers = prepend_title(device->GetMenuHeaders());
Doug Zongkerf93d8162009-09-22 15:16:02 -0700776
777 for (;;) {
778 finish_recovery(NULL);
Doug Zongker6c8553d2012-09-24 10:40:47 -0700779 switch (status) {
780 case INSTALL_SUCCESS:
781 case INSTALL_NONE:
782 ui->SetBackground(RecoveryUI::NO_COMMAND);
783 break;
784
785 case INSTALL_ERROR:
786 case INSTALL_CORRUPT:
787 ui->SetBackground(RecoveryUI::ERROR);
788 break;
789 }
Doug Zongker211aebc2011-10-28 15:13:10 -0700790 ui->SetProgressType(RecoveryUI::EMPTY);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700791
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700792 int chosen_item = get_menu_selection(headers, device->GetMenuItems(), 0, 0, device);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700793
794 // device-specific code may take some action here. It may
795 // return one of the core actions handled in the switch
796 // statement below.
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700797 chosen_item = device->InvokeMenuItem(chosen_item);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700798
Doug Zongkerd0181b82011-10-19 10:51:12 -0700799 int wipe_cache;
Doug Zongkerf93d8162009-09-22 15:16:02 -0700800 switch (chosen_item) {
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700801 case Device::REBOOT:
Doug Zongkerf93d8162009-09-22 15:16:02 -0700802 return;
803
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700804 case Device::WIPE_DATA:
805 wipe_data(ui->IsTextVisible(), device);
Doug Zongker211aebc2011-10-28 15:13:10 -0700806 if (!ui->IsTextVisible()) return;
Doug Zongkerf93d8162009-09-22 15:16:02 -0700807 break;
808
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700809 case Device::WIPE_CACHE:
Doug Zongker211aebc2011-10-28 15:13:10 -0700810 ui->Print("\n-- Wiping cache...\n");
Doug Zongkerd4208f92010-09-20 12:16:13 -0700811 erase_volume("/cache");
Doug Zongker211aebc2011-10-28 15:13:10 -0700812 ui->Print("Cache wipe complete.\n");
813 if (!ui->IsTextVisible()) return;
Doug Zongkerf93d8162009-09-22 15:16:02 -0700814 break;
815
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700816 case Device::APPLY_EXT:
817 status = update_directory(SDCARD_ROOT, SDCARD_ROOT, &wipe_cache, device);
Doug Zongkerd0181b82011-10-19 10:51:12 -0700818 if (status == INSTALL_SUCCESS && wipe_cache) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700819 ui->Print("\n-- Wiping cache (at package request)...\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700820 if (erase_volume("/cache")) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700821 ui->Print("Cache wipe failed.\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700822 } else {
Doug Zongker211aebc2011-10-28 15:13:10 -0700823 ui->Print("Cache wipe complete.\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700824 }
825 }
Doug Zongker8674a722010-09-15 11:08:23 -0700826 if (status >= 0) {
827 if (status != INSTALL_SUCCESS) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700828 ui->SetBackground(RecoveryUI::ERROR);
829 ui->Print("Installation aborted.\n");
830 } else if (!ui->IsTextVisible()) {
Doug Zongker8674a722010-09-15 11:08:23 -0700831 return; // reboot if logs aren't visible
832 } else {
Doug Zongker211aebc2011-10-28 15:13:10 -0700833 ui->Print("\nInstall from sdcard complete.\n");
Doug Zongker8674a722010-09-15 11:08:23 -0700834 }
Doug Zongkerf93d8162009-09-22 15:16:02 -0700835 }
836 break;
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700837
838 case Device::APPLY_CACHE:
Michael Ward9d2629c2011-06-23 22:14:24 -0700839 // Don't unmount cache at the end of this.
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700840 status = update_directory(CACHE_ROOT, NULL, &wipe_cache, device);
Doug Zongkerd0181b82011-10-19 10:51:12 -0700841 if (status == INSTALL_SUCCESS && wipe_cache) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700842 ui->Print("\n-- Wiping cache (at package request)...\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700843 if (erase_volume("/cache")) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700844 ui->Print("Cache wipe failed.\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700845 } else {
Doug Zongker211aebc2011-10-28 15:13:10 -0700846 ui->Print("Cache wipe complete.\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700847 }
848 }
Michael Ward9d2629c2011-06-23 22:14:24 -0700849 if (status >= 0) {
850 if (status != INSTALL_SUCCESS) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700851 ui->SetBackground(RecoveryUI::ERROR);
852 ui->Print("Installation aborted.\n");
853 } else if (!ui->IsTextVisible()) {
Michael Ward9d2629c2011-06-23 22:14:24 -0700854 return; // reboot if logs aren't visible
855 } else {
Doug Zongker211aebc2011-10-28 15:13:10 -0700856 ui->Print("\nInstall from cache complete.\n");
Michael Ward9d2629c2011-06-23 22:14:24 -0700857 }
858 }
859 break;
Doug Zongker9270a202012-01-09 15:16:13 -0800860
861 case Device::APPLY_ADB_SIDELOAD:
Doug Zongker9270a202012-01-09 15:16:13 -0800862 status = apply_from_adb(ui, &wipe_cache, TEMPORARY_INSTALL_FILE);
863 if (status >= 0) {
864 if (status != INSTALL_SUCCESS) {
865 ui->SetBackground(RecoveryUI::ERROR);
866 ui->Print("Installation aborted.\n");
Doug Zongkerf24fd7e2013-07-02 11:43:25 -0700867 copy_logs();
Doug Zongker9270a202012-01-09 15:16:13 -0800868 } else if (!ui->IsTextVisible()) {
869 return; // reboot if logs aren't visible
870 } else {
871 ui->Print("\nInstall from ADB complete.\n");
872 }
873 }
874 break;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800875 }
876 }
877}
878
879static void
Oscar Montemayor05231562009-11-30 08:40:57 -0800880print_property(const char *key, const char *name, void *cookie) {
Doug Zongker56c51052010-07-01 09:18:44 -0700881 printf("%s=%s\n", key, name);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800882}
883
Doug Zongker02ec6b82012-08-22 17:26:40 -0700884static void
885load_locale_from_cache() {
886 FILE* fp = fopen_path(LOCALE_FILE, "r");
887 char buffer[80];
888 if (fp != NULL) {
889 fgets(buffer, sizeof(buffer), fp);
890 int j = 0;
Doug Zongker5fa8c232012-09-18 12:37:02 -0700891 unsigned int i;
Doug Zongker02ec6b82012-08-22 17:26:40 -0700892 for (i = 0; i < sizeof(buffer) && buffer[i]; ++i) {
893 if (!isspace(buffer[i])) {
894 buffer[j++] = buffer[i];
895 }
896 }
897 buffer[j] = 0;
898 locale = strdup(buffer);
Devin Kim6016d082012-10-08 09:47:37 -0700899 check_and_fclose(fp, LOCALE_FILE);
Doug Zongker02ec6b82012-08-22 17:26:40 -0700900 }
901}
902
Doug Zongker7c3ae452013-05-14 11:03:02 -0700903static RecoveryUI* gCurrentUI = NULL;
904
905void
906ui_print(const char* format, ...) {
907 char buffer[256];
908
909 va_list ap;
910 va_start(ap, format);
911 vsnprintf(buffer, sizeof(buffer), format, ap);
912 va_end(ap);
913
914 if (gCurrentUI != NULL) {
915 gCurrentUI->Print("%s", buffer);
916 } else {
917 fputs(buffer, stdout);
918 }
919}
920
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800921int
Oscar Montemayor05231562009-11-30 08:40:57 -0800922main(int argc, char **argv) {
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800923 time_t start = time(NULL);
924
925 // If these fail, there's not really anywhere to complain...
926 freopen(TEMPORARY_LOG_FILE, "a", stdout); setbuf(stdout, NULL);
927 freopen(TEMPORARY_LOG_FILE, "a", stderr); setbuf(stderr, NULL);
Doug Zongker9270a202012-01-09 15:16:13 -0800928
929 // If this binary is started with the single argument "--adbd",
930 // instead of being the normal recovery binary, it turns into kind
931 // of a stripped-down version of adbd that only supports the
932 // 'sideload' command. Note this must be a real argument, not
933 // anything in the command file or bootloader control block; the
934 // only way recovery should be run with this argument is when it
935 // starts a copy of itself from the apply_from_adb() function.
936 if (argc == 2 && strcmp(argv[1], "--adbd") == 0) {
937 adb_main();
938 return 0;
939 }
940
Doug Zongker56c51052010-07-01 09:18:44 -0700941 printf("Starting recovery on %s", ctime(&start));
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800942
Doug Zongkerd4208f92010-09-20 12:16:13 -0700943 load_volume_table();
Doug Zongkerda1ebae2013-05-16 11:23:48 -0700944 ensure_path_mounted(LAST_LOG_FILE);
Doug Zongkerf24fd7e2013-07-02 11:43:25 -0700945 rotate_last_logs(10);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800946 get_args(&argc, &argv);
947
948 int previous_runs = 0;
949 const char *send_intent = NULL;
950 const char *update_package = NULL;
Doug Zongker02ec6b82012-08-22 17:26:40 -0700951 int wipe_data = 0, wipe_cache = 0, show_text = 0;
Doug Zongkere5d5ac72012-04-12 11:01:22 -0700952 bool just_exit = false;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800953
954 int arg;
955 while ((arg = getopt_long(argc, argv, "", OPTIONS, NULL)) != -1) {
956 switch (arg) {
957 case 'p': previous_runs = atoi(optarg); break;
958 case 's': send_intent = optarg; break;
959 case 'u': update_package = optarg; break;
960 case 'w': wipe_data = wipe_cache = 1; break;
961 case 'c': wipe_cache = 1; break;
Doug Zongker02ec6b82012-08-22 17:26:40 -0700962 case 't': show_text = 1; break;
Doug Zongkere5d5ac72012-04-12 11:01:22 -0700963 case 'x': just_exit = true; break;
Doug Zongker02ec6b82012-08-22 17:26:40 -0700964 case 'l': locale = optarg; break;
Doug Zongkerc87bab12013-11-25 13:53:25 -0800965 case 'g': {
966 if (stage == NULL || *stage == '\0') {
967 char buffer[20] = "1/";
968 strncat(buffer, optarg, sizeof(buffer)-3);
969 stage = strdup(buffer);
970 }
971 break;
972 }
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800973 case '?':
974 LOGE("Invalid command argument\n");
975 continue;
976 }
977 }
978
Doug Zongker02ec6b82012-08-22 17:26:40 -0700979 if (locale == NULL) {
980 load_locale_from_cache();
981 }
982 printf("locale is [%s]\n", locale);
Doug Zongkerc87bab12013-11-25 13:53:25 -0800983 printf("stage is [%s]\n", stage, stage);
Doug Zongker02ec6b82012-08-22 17:26:40 -0700984
985 Device* device = make_device();
986 ui = device->GetUI();
Doug Zongker7c3ae452013-05-14 11:03:02 -0700987 gCurrentUI = ui;
Doug Zongker02ec6b82012-08-22 17:26:40 -0700988
989 ui->Init();
Doug Zongkerc87bab12013-11-25 13:53:25 -0800990
991 int st_cur, st_max;
992 if (stage != NULL && sscanf(stage, "%d/%d", &st_cur, &st_max) == 2) {
993 ui->SetStage(st_cur, st_max);
994 }
995
Doug Zongker5fa8c232012-09-18 12:37:02 -0700996 ui->SetLocale(locale);
Doug Zongker02ec6b82012-08-22 17:26:40 -0700997 ui->SetBackground(RecoveryUI::NONE);
998 if (show_text) ui->ShowText(true);
999
Stephen Smalley779701d2012-02-09 14:13:23 -05001000 struct selinux_opt seopts[] = {
1001 { SELABEL_OPT_PATH, "/file_contexts" }
1002 };
1003
1004 sehandle = selabel_open(SELABEL_CTX_FILE, seopts, 1);
1005
1006 if (!sehandle) {
Doug Zongkerfafc85b2013-07-09 12:29:45 -07001007 ui->Print("Warning: No file_contexts\n");
Stephen Smalley779701d2012-02-09 14:13:23 -05001008 }
Stephen Smalley779701d2012-02-09 14:13:23 -05001009
Doug Zongkerdaefc1d2011-10-31 09:34:15 -07001010 device->StartRecovery();
Doug Zongkerefa1bab2010-02-01 15:59:12 -08001011
Doug Zongker56c51052010-07-01 09:18:44 -07001012 printf("Command:");
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -08001013 for (arg = 0; arg < argc; arg++) {
Doug Zongker56c51052010-07-01 09:18:44 -07001014 printf(" \"%s\"", argv[arg]);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -08001015 }
Doug Zongker9b125b02010-09-22 12:01:37 -07001016 printf("\n");
1017
1018 if (update_package) {
1019 // For backwards compatibility on the cache partition only, if
1020 // we're given an old 'root' path "CACHE:foo", change it to
1021 // "/cache/foo".
1022 if (strncmp(update_package, "CACHE:", 6) == 0) {
1023 int len = strlen(update_package) + 10;
Doug Zongker28ce47c2011-10-28 10:33:05 -07001024 char* modified_path = (char*)malloc(len);
Doug Zongker9b125b02010-09-22 12:01:37 -07001025 strlcpy(modified_path, "/cache/", len);
1026 strlcat(modified_path, update_package+6, len);
1027 printf("(replacing path \"%s\" with \"%s\")\n",
1028 update_package, modified_path);
1029 update_package = modified_path;
1030 }
1031 }
1032 printf("\n");
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -08001033
1034 property_list(print_property, NULL);
Doug Zongkerc0441d12013-07-31 11:28:24 -07001035 property_get("ro.build.display.id", recovery_version, "");
Doug Zongker56c51052010-07-01 09:18:44 -07001036 printf("\n");
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -08001037
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -08001038 int status = INSTALL_SUCCESS;
1039
Doug Zongker540d57f2011-01-18 13:36:58 -08001040 if (update_package != NULL) {
Doug Zongkerd0181b82011-10-19 10:51:12 -07001041 status = install_package(update_package, &wipe_cache, TEMPORARY_INSTALL_FILE);
1042 if (status == INSTALL_SUCCESS && wipe_cache) {
1043 if (erase_volume("/cache")) {
1044 LOGE("Cache wipe (requested by package) failed.");
1045 }
1046 }
Doug Zongker7c3ae452013-05-14 11:03:02 -07001047 if (status != INSTALL_SUCCESS) {
1048 ui->Print("Installation aborted.\n");
1049
1050 // If this is an eng or userdebug build, then automatically
1051 // turn the text display on if the script fails so the error
1052 // message is visible.
1053 char buffer[PROPERTY_VALUE_MAX+1];
1054 property_get("ro.build.fingerprint", buffer, "");
1055 if (strstr(buffer, ":userdebug/") || strstr(buffer, ":eng/")) {
1056 ui->ShowText(true);
1057 }
1058 }
Doug Zongkerb128f542009-06-18 15:07:14 -07001059 } else if (wipe_data) {
Doug Zongkerdaefc1d2011-10-31 09:34:15 -07001060 if (device->WipeData()) status = INSTALL_ERROR;
Doug Zongkerd4208f92010-09-20 12:16:13 -07001061 if (erase_volume("/data")) status = INSTALL_ERROR;
1062 if (wipe_cache && erase_volume("/cache")) status = INSTALL_ERROR;
Doug Zongker211aebc2011-10-28 15:13:10 -07001063 if (status != INSTALL_SUCCESS) ui->Print("Data wipe failed.\n");
Doug Zongkerb128f542009-06-18 15:07:14 -07001064 } else if (wipe_cache) {
Doug Zongkerd4208f92010-09-20 12:16:13 -07001065 if (wipe_cache && erase_volume("/cache")) status = INSTALL_ERROR;
Doug Zongker211aebc2011-10-28 15:13:10 -07001066 if (status != INSTALL_SUCCESS) ui->Print("Cache wipe failed.\n");
Doug Zongkere5d5ac72012-04-12 11:01:22 -07001067 } else if (!just_exit) {
Doug Zongker02ec6b82012-08-22 17:26:40 -07001068 status = INSTALL_NONE; // No command specified
1069 ui->SetBackground(RecoveryUI::NO_COMMAND);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -08001070 }
1071
Doug Zongker02ec6b82012-08-22 17:26:40 -07001072 if (status == INSTALL_ERROR || status == INSTALL_CORRUPT) {
Doug Zongkerf24fd7e2013-07-02 11:43:25 -07001073 copy_logs();
Doug Zongker02ec6b82012-08-22 17:26:40 -07001074 ui->SetBackground(RecoveryUI::ERROR);
1075 }
Doug Zongker211aebc2011-10-28 15:13:10 -07001076 if (status != INSTALL_SUCCESS || ui->IsTextVisible()) {
Doug Zongker6c8553d2012-09-24 10:40:47 -07001077 prompt_and_wait(device, status);
Doug Zongker8674a722010-09-15 11:08:23 -07001078 }
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -08001079
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -08001080 // Otherwise, get ready to boot the main system...
1081 finish_recovery(send_intent);
Doug Zongker211aebc2011-10-28 15:13:10 -07001082 ui->Print("Rebooting...\n");
Doug Zongker3b5a9872013-09-03 14:29:54 -07001083 property_set(ANDROID_RB_PROPERTY, "reboot,");
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -08001084 return EXIT_SUCCESS;
1085}