blob: 7002cb8a4db3ab665e957ca65bb151f42239a4ca [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"
Ken Sumrall6e4472a2011-03-07 23:37:27 -080035#include "cutils/android_reboot.h"
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080036#include "install.h"
37#include "minui/minui.h"
38#include "minzip/DirUtil.h"
39#include "roots.h"
Doug Zongker28ce47c2011-10-28 10:33:05 -070040#include "ui.h"
Doug Zongker211aebc2011-10-28 15:13:10 -070041#include "screen_ui.h"
Doug Zongkerdaefc1d2011-10-31 09:34:15 -070042#include "device.h"
Doug Zongker9270a202012-01-09 15:16:13 -080043#include "adb_install.h"
44extern "C" {
45#include "minadbd/adb.h"
46}
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080047
Stephen Smalley779701d2012-02-09 14:13:23 -050048struct selabel_handle *sehandle;
49
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080050static const struct option OPTIONS[] = {
51 { "send_intent", required_argument, NULL, 's' },
52 { "update_package", required_argument, NULL, 'u' },
53 { "wipe_data", no_argument, NULL, 'w' },
54 { "wipe_cache", no_argument, NULL, 'c' },
Doug Zongker4bc98062010-09-03 11:00:13 -070055 { "show_text", no_argument, NULL, 't' },
Doug Zongkere5d5ac72012-04-12 11:01:22 -070056 { "just_exit", no_argument, NULL, 'x' },
Doug Zongker02ec6b82012-08-22 17:26:40 -070057 { "locale", required_argument, NULL, 'l' },
Doug Zongker988500b2009-10-06 14:41:38 -070058 { NULL, 0, NULL, 0 },
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080059};
60
Doug Zongkerd4208f92010-09-20 12:16:13 -070061static const char *COMMAND_FILE = "/cache/recovery/command";
62static const char *INTENT_FILE = "/cache/recovery/intent";
63static const char *LOG_FILE = "/cache/recovery/log";
Doug Zongker2c3539e2010-09-29 13:21:30 -070064static const char *LAST_LOG_FILE = "/cache/recovery/last_log";
Doug Zongkerd0181b82011-10-19 10:51:12 -070065static const char *LAST_INSTALL_FILE = "/cache/recovery/last_install";
Doug Zongker8b240cc2012-08-29 15:19:29 -070066static const char *LOCALE_FILE = "/cache/recovery/last_locale";
Michael Ward9d2629c2011-06-23 22:14:24 -070067static const char *CACHE_ROOT = "/cache";
Doug Zongkerd4208f92010-09-20 12:16:13 -070068static const char *SDCARD_ROOT = "/sdcard";
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080069static const char *TEMPORARY_LOG_FILE = "/tmp/recovery.log";
Doug Zongkerd0181b82011-10-19 10:51:12 -070070static const char *TEMPORARY_INSTALL_FILE = "/tmp/last_install";
Doug Zongkerd4208f92010-09-20 12:16:13 -070071static const char *SIDELOAD_TEMP_DIR = "/tmp/sideload";
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080072
Doug Zongker211aebc2011-10-28 15:13:10 -070073RecoveryUI* ui = NULL;
Doug Zongker02ec6b82012-08-22 17:26:40 -070074char* locale = NULL;
Doug Zongker211aebc2011-10-28 15:13:10 -070075
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080076/*
77 * The recovery tool communicates with the main system through /cache files.
78 * /cache/recovery/command - INPUT - command line for tool, one arg per line
79 * /cache/recovery/log - OUTPUT - combined log file from recovery run(s)
80 * /cache/recovery/intent - OUTPUT - intent that was passed in
81 *
82 * The arguments which may be supplied in the recovery.command file:
83 * --send_intent=anystring - write the text out to recovery.intent
Doug Zongkerd4208f92010-09-20 12:16:13 -070084 * --update_package=path - verify install an OTA package file
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080085 * --wipe_data - erase user data (and cache), then reboot
86 * --wipe_cache - wipe cache (but not user data), then reboot
Oscar Montemayor05231562009-11-30 08:40:57 -080087 * --set_encrypted_filesystem=on|off - enables / diasables encrypted fs
Doug Zongkere5d5ac72012-04-12 11:01:22 -070088 * --just_exit - do nothing; exit and reboot
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -080089 *
90 * After completing, we remove /cache/recovery/command and reboot.
91 * Arguments may also be supplied in the bootloader control block (BCB).
92 * These important scenarios must be safely restartable at any point:
93 *
94 * FACTORY RESET
95 * 1. user selects "factory reset"
96 * 2. main system writes "--wipe_data" to /cache/recovery/command
97 * 3. main system reboots into recovery
98 * 4. get_args() writes BCB with "boot-recovery" and "--wipe_data"
99 * -- after this, rebooting will restart the erase --
Doug Zongkerd4208f92010-09-20 12:16:13 -0700100 * 5. erase_volume() reformats /data
101 * 6. erase_volume() reformats /cache
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800102 * 7. finish_recovery() erases BCB
103 * -- after this, rebooting will restart the main system --
104 * 8. main() calls reboot() to boot main system
105 *
106 * OTA INSTALL
107 * 1. main system downloads OTA package to /cache/some-filename.zip
Doug Zongker9b125b02010-09-22 12:01:37 -0700108 * 2. main system writes "--update_package=/cache/some-filename.zip"
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800109 * 3. main system reboots into recovery
110 * 4. get_args() writes BCB with "boot-recovery" and "--update_package=..."
111 * -- after this, rebooting will attempt to reinstall the update --
112 * 5. install_package() attempts to install the update
113 * NOTE: the package install must itself be restartable from any point
114 * 6. finish_recovery() erases BCB
115 * -- after this, rebooting will (try to) restart the main system --
116 * 7. ** if install failed **
117 * 7a. prompt_and_wait() shows an error icon and waits for the user
118 * 7b; the user reboots (pulling the battery, etc) into the main system
119 * 8. main() calls maybe_install_firmware_update()
120 * ** if the update contained radio/hboot firmware **:
121 * 8a. m_i_f_u() writes BCB with "boot-recovery" and "--wipe_cache"
122 * -- after this, rebooting will reformat cache & restart main system --
123 * 8b. m_i_f_u() writes firmware image into raw cache partition
124 * 8c. m_i_f_u() writes BCB with "update-radio/hboot" and "--wipe_cache"
125 * -- after this, rebooting will attempt to reinstall firmware --
126 * 8d. bootloader tries to flash firmware
127 * 8e. bootloader writes BCB with "boot-recovery" (keeping "--wipe_cache")
128 * -- after this, rebooting will reformat cache & restart main system --
Doug Zongkerd4208f92010-09-20 12:16:13 -0700129 * 8f. erase_volume() reformats /cache
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800130 * 8g. finish_recovery() erases BCB
131 * -- after this, rebooting will (try to) restart the main system --
132 * 9. main() calls reboot() to boot main system
133 */
134
135static const int MAX_ARG_LENGTH = 4096;
136static const int MAX_ARGS = 100;
137
Doug Zongkerd4208f92010-09-20 12:16:13 -0700138// open a given path, mounting partitions as necessary
Doug Zongker469243e2011-04-12 09:28:10 -0700139FILE*
Doug Zongkerd4208f92010-09-20 12:16:13 -0700140fopen_path(const char *path, const char *mode) {
141 if (ensure_path_mounted(path) != 0) {
142 LOGE("Can't mount %s\n", path);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800143 return NULL;
144 }
145
146 // When writing, try to create the containing directory, if necessary.
147 // Use generous permissions, the system (init.rc) will reset them.
Stephen Smalley779701d2012-02-09 14:13:23 -0500148 if (strchr("wa", mode[0])) dirCreateHierarchy(path, 0777, NULL, 1, sehandle);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800149
150 FILE *fp = fopen(path, mode);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800151 return fp;
152}
153
154// close a file, log an error if the error indicator is set
155static void
156check_and_fclose(FILE *fp, const char *name) {
157 fflush(fp);
158 if (ferror(fp)) LOGE("Error in %s\n(%s)\n", name, strerror(errno));
159 fclose(fp);
160}
161
162// command line args come from, in decreasing precedence:
163// - the actual command line
164// - the bootloader control block (one per line, after "recovery")
165// - the contents of COMMAND_FILE (one per line)
166static void
167get_args(int *argc, char ***argv) {
168 struct bootloader_message boot;
169 memset(&boot, 0, sizeof(boot));
170 get_bootloader_message(&boot); // this may fail, leaving a zeroed structure
171
172 if (boot.command[0] != 0 && boot.command[0] != 255) {
173 LOGI("Boot command: %.*s\n", sizeof(boot.command), boot.command);
174 }
175
176 if (boot.status[0] != 0 && boot.status[0] != 255) {
177 LOGI("Boot status: %.*s\n", sizeof(boot.status), boot.status);
178 }
179
180 // --- if arguments weren't supplied, look in the bootloader control block
181 if (*argc <= 1) {
182 boot.recovery[sizeof(boot.recovery) - 1] = '\0'; // Ensure termination
183 const char *arg = strtok(boot.recovery, "\n");
184 if (arg != NULL && !strcmp(arg, "recovery")) {
185 *argv = (char **) malloc(sizeof(char *) * MAX_ARGS);
186 (*argv)[0] = strdup(arg);
187 for (*argc = 1; *argc < MAX_ARGS; ++*argc) {
188 if ((arg = strtok(NULL, "\n")) == NULL) break;
189 (*argv)[*argc] = strdup(arg);
190 }
191 LOGI("Got arguments from boot message\n");
192 } else if (boot.recovery[0] != 0 && boot.recovery[0] != 255) {
193 LOGE("Bad boot message\n\"%.20s\"\n", boot.recovery);
194 }
195 }
196
197 // --- if that doesn't work, try the command file
198 if (*argc <= 1) {
Doug Zongkerd4208f92010-09-20 12:16:13 -0700199 FILE *fp = fopen_path(COMMAND_FILE, "r");
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800200 if (fp != NULL) {
201 char *argv0 = (*argv)[0];
202 *argv = (char **) malloc(sizeof(char *) * MAX_ARGS);
203 (*argv)[0] = argv0; // use the same program name
204
205 char buf[MAX_ARG_LENGTH];
206 for (*argc = 1; *argc < MAX_ARGS; ++*argc) {
207 if (!fgets(buf, sizeof(buf), fp)) break;
208 (*argv)[*argc] = strdup(strtok(buf, "\r\n")); // Strip newline.
209 }
210
211 check_and_fclose(fp, COMMAND_FILE);
212 LOGI("Got arguments from %s\n", COMMAND_FILE);
213 }
214 }
215
216 // --> write the arguments we have back into the bootloader control block
217 // always boot into recovery after this (until finish_recovery() is called)
218 strlcpy(boot.command, "boot-recovery", sizeof(boot.command));
219 strlcpy(boot.recovery, "recovery\n", sizeof(boot.recovery));
220 int i;
221 for (i = 1; i < *argc; ++i) {
222 strlcat(boot.recovery, (*argv)[i], sizeof(boot.recovery));
223 strlcat(boot.recovery, "\n", sizeof(boot.recovery));
224 }
225 set_bootloader_message(&boot);
226}
227
Doug Zongker34c98df2009-08-18 12:05:45 -0700228static void
Oscar Montemayor05231562009-11-30 08:40:57 -0800229set_sdcard_update_bootloader_message() {
Doug Zongker34c98df2009-08-18 12:05:45 -0700230 struct bootloader_message boot;
231 memset(&boot, 0, sizeof(boot));
232 strlcpy(boot.command, "boot-recovery", sizeof(boot.command));
233 strlcpy(boot.recovery, "recovery\n", sizeof(boot.recovery));
234 set_bootloader_message(&boot);
235}
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800236
Doug Zongker2c3539e2010-09-29 13:21:30 -0700237// How much of the temp log we have copied to the copy in cache.
238static long tmplog_offset = 0;
239
240static void
Doug Zongkerd0181b82011-10-19 10:51:12 -0700241copy_log_file(const char* source, const char* destination, int append) {
Doug Zongker2c3539e2010-09-29 13:21:30 -0700242 FILE *log = fopen_path(destination, append ? "a" : "w");
243 if (log == NULL) {
244 LOGE("Can't open %s\n", destination);
245 } else {
Doug Zongkerd0181b82011-10-19 10:51:12 -0700246 FILE *tmplog = fopen(source, "r");
247 if (tmplog != NULL) {
Doug Zongker2c3539e2010-09-29 13:21:30 -0700248 if (append) {
249 fseek(tmplog, tmplog_offset, SEEK_SET); // Since last write
250 }
251 char buf[4096];
252 while (fgets(buf, sizeof(buf), tmplog)) fputs(buf, log);
253 if (append) {
254 tmplog_offset = ftell(tmplog);
255 }
Doug Zongkerd0181b82011-10-19 10:51:12 -0700256 check_and_fclose(tmplog, source);
Doug Zongker2c3539e2010-09-29 13:21:30 -0700257 }
258 check_and_fclose(log, destination);
259 }
260}
261
262
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800263// clear the recovery command and prepare to boot a (hopefully working) system,
264// copy our log file to cache as well (for the system to read), and
265// record any intent we were asked to communicate back to the system.
266// this function is idempotent: call it as many times as you like.
267static void
Oscar Montemayor05231562009-11-30 08:40:57 -0800268finish_recovery(const char *send_intent) {
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800269 // By this point, we're ready to return to the main system...
270 if (send_intent != NULL) {
Doug Zongkerd4208f92010-09-20 12:16:13 -0700271 FILE *fp = fopen_path(INTENT_FILE, "w");
Jay Freeman (saurik)619ec2f2008-11-17 01:56:05 +0000272 if (fp == NULL) {
273 LOGE("Can't open %s\n", INTENT_FILE);
274 } else {
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800275 fputs(send_intent, fp);
276 check_and_fclose(fp, INTENT_FILE);
277 }
278 }
279
Doug Zongker4f33e552012-08-23 13:16:12 -0700280 // Save the locale to cache, so if recovery is next started up
281 // without a --locale argument (eg, directly from the bootloader)
282 // it will use the last-known locale.
283 if (locale != NULL) {
284 LOGI("Saving locale \"%s\"\n", locale);
285 FILE* fp = fopen_path(LOCALE_FILE, "w");
286 fwrite(locale, 1, strlen(locale), fp);
287 fflush(fp);
288 fsync(fileno(fp));
289 check_and_fclose(fp, LOCALE_FILE);
290 }
291
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800292 // Copy logs to cache so the system can find out what happened.
Doug Zongkerd0181b82011-10-19 10:51:12 -0700293 copy_log_file(TEMPORARY_LOG_FILE, LOG_FILE, true);
294 copy_log_file(TEMPORARY_LOG_FILE, LAST_LOG_FILE, false);
295 copy_log_file(TEMPORARY_INSTALL_FILE, LAST_INSTALL_FILE, false);
296 chmod(LOG_FILE, 0600);
297 chown(LOG_FILE, 1000, 1000); // system user
Doug Zongker2c3539e2010-09-29 13:21:30 -0700298 chmod(LAST_LOG_FILE, 0640);
Doug Zongkerd0181b82011-10-19 10:51:12 -0700299 chmod(LAST_INSTALL_FILE, 0644);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800300
Doug Zongkere5d5ac72012-04-12 11:01:22 -0700301 // Reset to normal system boot so recovery won't cycle indefinitely.
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800302 struct bootloader_message boot;
303 memset(&boot, 0, sizeof(boot));
304 set_bootloader_message(&boot);
305
306 // Remove the command file, so recovery won't repeat indefinitely.
Doug Zongkerd4208f92010-09-20 12:16:13 -0700307 if (ensure_path_mounted(COMMAND_FILE) != 0 ||
308 (unlink(COMMAND_FILE) && errno != ENOENT)) {
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800309 LOGW("Can't unlink %s\n", COMMAND_FILE);
310 }
311
Doug Zongkerd0181b82011-10-19 10:51:12 -0700312 ensure_path_unmounted(CACHE_ROOT);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800313 sync(); // For good measure.
314}
315
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800316static int
Doug Zongkerd4208f92010-09-20 12:16:13 -0700317erase_volume(const char *volume) {
Doug Zongker02ec6b82012-08-22 17:26:40 -0700318 ui->SetBackground(RecoveryUI::ERASING);
Doug Zongker211aebc2011-10-28 15:13:10 -0700319 ui->SetProgressType(RecoveryUI::INDETERMINATE);
320 ui->Print("Formatting %s...\n", volume);
Doug Zongker2c3539e2010-09-29 13:21:30 -0700321
Doug Zongkerd0181b82011-10-19 10:51:12 -0700322 ensure_path_unmounted(volume);
323
Doug Zongker2c3539e2010-09-29 13:21:30 -0700324 if (strcmp(volume, "/cache") == 0) {
325 // Any part of the log we'd copied to cache is now gone.
326 // Reset the pointer so we copy from the beginning of the temp
327 // log.
328 tmplog_offset = 0;
329 }
330
Doug Zongkerd4208f92010-09-20 12:16:13 -0700331 return format_volume(volume);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800332}
333
Doug Zongker23ceeea2010-07-08 17:27:55 -0700334static char*
Doug Zongkerd4208f92010-09-20 12:16:13 -0700335copy_sideloaded_package(const char* original_path) {
336 if (ensure_path_mounted(original_path) != 0) {
337 LOGE("Can't mount %s\n", original_path);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700338 return NULL;
339 }
340
Doug Zongkerd4208f92010-09-20 12:16:13 -0700341 if (ensure_path_mounted(SIDELOAD_TEMP_DIR) != 0) {
Doug Zongker23ceeea2010-07-08 17:27:55 -0700342 LOGE("Can't mount %s\n", SIDELOAD_TEMP_DIR);
343 return NULL;
344 }
345
Doug Zongkerd4208f92010-09-20 12:16:13 -0700346 if (mkdir(SIDELOAD_TEMP_DIR, 0700) != 0) {
Doug Zongker23ceeea2010-07-08 17:27:55 -0700347 if (errno != EEXIST) {
348 LOGE("Can't mkdir %s (%s)\n", SIDELOAD_TEMP_DIR, strerror(errno));
349 return NULL;
350 }
351 }
352
Doug Zongkerd4208f92010-09-20 12:16:13 -0700353 // verify that SIDELOAD_TEMP_DIR is exactly what we expect: a
354 // directory, owned by root, readable and writable only by root.
Doug Zongker23ceeea2010-07-08 17:27:55 -0700355 struct stat st;
Doug Zongkerd4208f92010-09-20 12:16:13 -0700356 if (stat(SIDELOAD_TEMP_DIR, &st) != 0) {
357 LOGE("failed to stat %s (%s)\n", SIDELOAD_TEMP_DIR, strerror(errno));
Doug Zongker23ceeea2010-07-08 17:27:55 -0700358 return NULL;
359 }
360 if (!S_ISDIR(st.st_mode)) {
Doug Zongkerd4208f92010-09-20 12:16:13 -0700361 LOGE("%s isn't a directory\n", SIDELOAD_TEMP_DIR);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700362 return NULL;
363 }
364 if ((st.st_mode & 0777) != 0700) {
Doug Zongkerd4208f92010-09-20 12:16:13 -0700365 LOGE("%s has perms %o\n", SIDELOAD_TEMP_DIR, st.st_mode);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700366 return NULL;
367 }
368 if (st.st_uid != 0) {
Doug Zongkerd4208f92010-09-20 12:16:13 -0700369 LOGE("%s owned by %lu; not root\n", SIDELOAD_TEMP_DIR, st.st_uid);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700370 return NULL;
371 }
372
Doug Zongkerd4208f92010-09-20 12:16:13 -0700373 char copy_path[PATH_MAX];
374 strcpy(copy_path, SIDELOAD_TEMP_DIR);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700375 strcat(copy_path, "/package.zip");
376
Doug Zongker28ce47c2011-10-28 10:33:05 -0700377 char* buffer = (char*)malloc(BUFSIZ);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700378 if (buffer == NULL) {
379 LOGE("Failed to allocate buffer\n");
380 return NULL;
381 }
382
383 size_t read;
384 FILE* fin = fopen(original_path, "rb");
385 if (fin == NULL) {
386 LOGE("Failed to open %s (%s)\n", original_path, strerror(errno));
387 return NULL;
388 }
389 FILE* fout = fopen(copy_path, "wb");
390 if (fout == NULL) {
391 LOGE("Failed to open %s (%s)\n", copy_path, strerror(errno));
392 return NULL;
393 }
394
395 while ((read = fread(buffer, 1, BUFSIZ, fin)) > 0) {
396 if (fwrite(buffer, 1, read, fout) != read) {
397 LOGE("Short write of %s (%s)\n", copy_path, strerror(errno));
398 return NULL;
399 }
400 }
401
402 free(buffer);
403
404 if (fclose(fout) != 0) {
405 LOGE("Failed to close %s (%s)\n", copy_path, strerror(errno));
406 return NULL;
407 }
408
409 if (fclose(fin) != 0) {
410 LOGE("Failed to close %s (%s)\n", original_path, strerror(errno));
411 return NULL;
412 }
413
414 // "adb push" is happy to overwrite read-only files when it's
415 // running as root, but we'll try anyway.
416 if (chmod(copy_path, 0400) != 0) {
417 LOGE("Failed to chmod %s (%s)\n", copy_path, strerror(errno));
418 return NULL;
419 }
420
Doug Zongkerd4208f92010-09-20 12:16:13 -0700421 return strdup(copy_path);
Doug Zongker23ceeea2010-07-08 17:27:55 -0700422}
423
Doug Zongker28ce47c2011-10-28 10:33:05 -0700424static const char**
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700425prepend_title(const char* const* headers) {
Doug Zongker28ce47c2011-10-28 10:33:05 -0700426 const char* title[] = { "Android system recovery <"
427 EXPAND(RECOVERY_API_VERSION) "e>",
428 "",
429 NULL };
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800430
Doug Zongkerd6837852009-06-17 22:07:13 -0700431 // count the number of lines in our title, plus the
Doug Zongkerf93d8162009-09-22 15:16:02 -0700432 // caller-provided headers.
Doug Zongkerd6837852009-06-17 22:07:13 -0700433 int count = 0;
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700434 const char* const* p;
Doug Zongkerd6837852009-06-17 22:07:13 -0700435 for (p = title; *p; ++p, ++count);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700436 for (p = headers; *p; ++p, ++count);
Doug Zongkerd6837852009-06-17 22:07:13 -0700437
Doug Zongker28ce47c2011-10-28 10:33:05 -0700438 const char** new_headers = (const char**)malloc((count+1) * sizeof(char*));
439 const char** h = new_headers;
Doug Zongkerd6837852009-06-17 22:07:13 -0700440 for (p = title; *p; ++p, ++h) *h = *p;
Doug Zongkerf93d8162009-09-22 15:16:02 -0700441 for (p = headers; *p; ++p, ++h) *h = *p;
Doug Zongkerd6837852009-06-17 22:07:13 -0700442 *h = NULL;
443
Doug Zongkerf93d8162009-09-22 15:16:02 -0700444 return new_headers;
445}
446
447static int
Doug Zongker28ce47c2011-10-28 10:33:05 -0700448get_menu_selection(const char* const * headers, const char* const * items,
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700449 int menu_only, int initial_selection, Device* device) {
Doug Zongkerf93d8162009-09-22 15:16:02 -0700450 // throw away keys pressed previously, so user doesn't
451 // accidentally trigger menu items.
Doug Zongker211aebc2011-10-28 15:13:10 -0700452 ui->FlushKeys();
Doug Zongkerf93d8162009-09-22 15:16:02 -0700453
Doug Zongker211aebc2011-10-28 15:13:10 -0700454 ui->StartMenu(headers, items, initial_selection);
Doug Zongker8674a722010-09-15 11:08:23 -0700455 int selected = initial_selection;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800456 int chosen_item = -1;
457
Doug Zongkerf93d8162009-09-22 15:16:02 -0700458 while (chosen_item < 0) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700459 int key = ui->WaitKey();
460 int visible = ui->IsTextVisible();
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800461
Doug Zongker5cae4452011-01-25 13:15:30 -0800462 if (key == -1) { // ui_wait_key() timed out
Doug Zongker211aebc2011-10-28 15:13:10 -0700463 if (ui->WasTextEverVisible()) {
Doug Zongker5cae4452011-01-25 13:15:30 -0800464 continue;
465 } else {
466 LOGI("timed out waiting for key input; rebooting.\n");
Doug Zongker211aebc2011-10-28 15:13:10 -0700467 ui->EndMenu();
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700468 return 0; // XXX fixme
Doug Zongker5cae4452011-01-25 13:15:30 -0800469 }
470 }
471
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700472 int action = device->HandleMenuKey(key, visible);
Doug Zongkerddd6a282009-06-09 12:22:33 -0700473
474 if (action < 0) {
475 switch (action) {
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700476 case Device::kHighlightUp:
Doug Zongkerddd6a282009-06-09 12:22:33 -0700477 --selected;
Doug Zongker211aebc2011-10-28 15:13:10 -0700478 selected = ui->SelectMenu(selected);
Doug Zongkerddd6a282009-06-09 12:22:33 -0700479 break;
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700480 case Device::kHighlightDown:
Doug Zongkerddd6a282009-06-09 12:22:33 -0700481 ++selected;
Doug Zongker211aebc2011-10-28 15:13:10 -0700482 selected = ui->SelectMenu(selected);
Doug Zongkerddd6a282009-06-09 12:22:33 -0700483 break;
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700484 case Device::kInvokeItem:
Doug Zongkerddd6a282009-06-09 12:22:33 -0700485 chosen_item = selected;
486 break;
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700487 case Device::kNoAction:
Doug Zongkerddd6a282009-06-09 12:22:33 -0700488 break;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800489 }
Doug Zongkerf93d8162009-09-22 15:16:02 -0700490 } else if (!menu_only) {
Doug Zongkerddd6a282009-06-09 12:22:33 -0700491 chosen_item = action;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800492 }
Doug Zongkerf93d8162009-09-22 15:16:02 -0700493 }
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800494
Doug Zongker211aebc2011-10-28 15:13:10 -0700495 ui->EndMenu();
Doug Zongkerf93d8162009-09-22 15:16:02 -0700496 return chosen_item;
497}
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800498
Doug Zongker8674a722010-09-15 11:08:23 -0700499static int compare_string(const void* a, const void* b) {
500 return strcmp(*(const char**)a, *(const char**)b);
501}
502
503static int
Doug Zongkerd0181b82011-10-19 10:51:12 -0700504update_directory(const char* path, const char* unmount_when_done,
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700505 int* wipe_cache, Device* device) {
Michael Ward9d2629c2011-06-23 22:14:24 -0700506 ensure_path_mounted(path);
Doug Zongkerc18eeb82010-09-21 16:49:26 -0700507
Doug Zongker8674a722010-09-15 11:08:23 -0700508 const char* MENU_HEADERS[] = { "Choose a package to install:",
Doug Zongkerd4208f92010-09-20 12:16:13 -0700509 path,
Doug Zongker8674a722010-09-15 11:08:23 -0700510 "",
511 NULL };
512 DIR* d;
513 struct dirent* de;
Doug Zongkerd4208f92010-09-20 12:16:13 -0700514 d = opendir(path);
Doug Zongker8674a722010-09-15 11:08:23 -0700515 if (d == NULL) {
516 LOGE("error opening %s: %s\n", path, strerror(errno));
Michael Ward9d2629c2011-06-23 22:14:24 -0700517 if (unmount_when_done != NULL) {
518 ensure_path_unmounted(unmount_when_done);
519 }
Doug Zongker8674a722010-09-15 11:08:23 -0700520 return 0;
521 }
522
Doug Zongker28ce47c2011-10-28 10:33:05 -0700523 const char** headers = prepend_title(MENU_HEADERS);
Doug Zongker8674a722010-09-15 11:08:23 -0700524
525 int d_size = 0;
526 int d_alloc = 10;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700527 char** dirs = (char**)malloc(d_alloc * sizeof(char*));
Doug Zongker8674a722010-09-15 11:08:23 -0700528 int z_size = 1;
529 int z_alloc = 10;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700530 char** zips = (char**)malloc(z_alloc * sizeof(char*));
Doug Zongker8674a722010-09-15 11:08:23 -0700531 zips[0] = strdup("../");
532
533 while ((de = readdir(d)) != NULL) {
534 int name_len = strlen(de->d_name);
535
536 if (de->d_type == DT_DIR) {
537 // skip "." and ".." entries
538 if (name_len == 1 && de->d_name[0] == '.') continue;
539 if (name_len == 2 && de->d_name[0] == '.' &&
540 de->d_name[1] == '.') continue;
541
542 if (d_size >= d_alloc) {
543 d_alloc *= 2;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700544 dirs = (char**)realloc(dirs, d_alloc * sizeof(char*));
Doug Zongker8674a722010-09-15 11:08:23 -0700545 }
Doug Zongker28ce47c2011-10-28 10:33:05 -0700546 dirs[d_size] = (char*)malloc(name_len + 2);
Doug Zongker8674a722010-09-15 11:08:23 -0700547 strcpy(dirs[d_size], de->d_name);
548 dirs[d_size][name_len] = '/';
549 dirs[d_size][name_len+1] = '\0';
550 ++d_size;
551 } else if (de->d_type == DT_REG &&
552 name_len >= 4 &&
553 strncasecmp(de->d_name + (name_len-4), ".zip", 4) == 0) {
554 if (z_size >= z_alloc) {
555 z_alloc *= 2;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700556 zips = (char**)realloc(zips, z_alloc * sizeof(char*));
Doug Zongker8674a722010-09-15 11:08:23 -0700557 }
558 zips[z_size++] = strdup(de->d_name);
559 }
560 }
561 closedir(d);
562
563 qsort(dirs, d_size, sizeof(char*), compare_string);
564 qsort(zips, z_size, sizeof(char*), compare_string);
565
566 // append dirs to the zips list
567 if (d_size + z_size + 1 > z_alloc) {
568 z_alloc = d_size + z_size + 1;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700569 zips = (char**)realloc(zips, z_alloc * sizeof(char*));
Doug Zongker8674a722010-09-15 11:08:23 -0700570 }
571 memcpy(zips + z_size, dirs, d_size * sizeof(char*));
572 free(dirs);
573 z_size += d_size;
574 zips[z_size] = NULL;
575
576 int result;
577 int chosen_item = 0;
578 do {
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700579 chosen_item = get_menu_selection(headers, zips, 1, chosen_item, device);
Doug Zongker8674a722010-09-15 11:08:23 -0700580
581 char* item = zips[chosen_item];
582 int item_len = strlen(item);
583 if (chosen_item == 0) { // item 0 is always "../"
Michael Ward9d2629c2011-06-23 22:14:24 -0700584 // go up but continue browsing (if the caller is update_directory)
Doug Zongker8674a722010-09-15 11:08:23 -0700585 result = -1;
586 break;
587 } else if (item[item_len-1] == '/') {
588 // recurse down into a subdirectory
589 char new_path[PATH_MAX];
Doug Zongkerd4208f92010-09-20 12:16:13 -0700590 strlcpy(new_path, path, PATH_MAX);
591 strlcat(new_path, "/", PATH_MAX);
Doug Zongker8674a722010-09-15 11:08:23 -0700592 strlcat(new_path, item, PATH_MAX);
Doug Zongkerd4208f92010-09-20 12:16:13 -0700593 new_path[strlen(new_path)-1] = '\0'; // truncate the trailing '/'
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700594 result = update_directory(new_path, unmount_when_done, wipe_cache, device);
Doug Zongker8674a722010-09-15 11:08:23 -0700595 if (result >= 0) break;
596 } else {
597 // selected a zip file: attempt to install it, and return
598 // the status to the caller.
599 char new_path[PATH_MAX];
Doug Zongkerd4208f92010-09-20 12:16:13 -0700600 strlcpy(new_path, path, PATH_MAX);
Doug Zongkerc18eeb82010-09-21 16:49:26 -0700601 strlcat(new_path, "/", PATH_MAX);
Doug Zongker8674a722010-09-15 11:08:23 -0700602 strlcat(new_path, item, PATH_MAX);
603
Doug Zongker211aebc2011-10-28 15:13:10 -0700604 ui->Print("\n-- Install %s ...\n", path);
Doug Zongker8674a722010-09-15 11:08:23 -0700605 set_sdcard_update_bootloader_message();
Doug Zongkerd4208f92010-09-20 12:16:13 -0700606 char* copy = copy_sideloaded_package(new_path);
Michael Ward9d2629c2011-06-23 22:14:24 -0700607 if (unmount_when_done != NULL) {
608 ensure_path_unmounted(unmount_when_done);
609 }
Doug Zongkerd4208f92010-09-20 12:16:13 -0700610 if (copy) {
Doug Zongkerd0181b82011-10-19 10:51:12 -0700611 result = install_package(copy, wipe_cache, TEMPORARY_INSTALL_FILE);
Doug Zongkerd4208f92010-09-20 12:16:13 -0700612 free(copy);
613 } else {
614 result = INSTALL_ERROR;
615 }
Doug Zongker8674a722010-09-15 11:08:23 -0700616 break;
617 }
618 } while (true);
619
620 int i;
621 for (i = 0; i < z_size; ++i) free(zips[i]);
622 free(zips);
623 free(headers);
624
Michael Ward9d2629c2011-06-23 22:14:24 -0700625 if (unmount_when_done != NULL) {
626 ensure_path_unmounted(unmount_when_done);
627 }
Doug Zongker8674a722010-09-15 11:08:23 -0700628 return result;
629}
630
Doug Zongkerf93d8162009-09-22 15:16:02 -0700631static void
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700632wipe_data(int confirm, Device* device) {
Doug Zongkerf93d8162009-09-22 15:16:02 -0700633 if (confirm) {
Doug Zongker28ce47c2011-10-28 10:33:05 -0700634 static const char** title_headers = NULL;
Doug Zongkerddd6a282009-06-09 12:22:33 -0700635
Doug Zongkerf93d8162009-09-22 15:16:02 -0700636 if (title_headers == NULL) {
Doug Zongker28ce47c2011-10-28 10:33:05 -0700637 const char* headers[] = { "Confirm wipe of all user data?",
638 " THIS CAN NOT BE UNDONE.",
639 "",
640 NULL };
Doug Zongker8674a722010-09-15 11:08:23 -0700641 title_headers = prepend_title((const char**)headers);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700642 }
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800643
Doug Zongker28ce47c2011-10-28 10:33:05 -0700644 const char* items[] = { " No",
645 " No",
646 " No",
647 " No",
648 " No",
649 " No",
650 " No",
651 " Yes -- delete all user data", // [7]
652 " No",
653 " No",
654 " No",
655 NULL };
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800656
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700657 int chosen_item = get_menu_selection(title_headers, items, 1, 0, device);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700658 if (chosen_item != 7) {
659 return;
660 }
661 }
Doug Zongker1066d2c2009-04-01 13:57:40 -0700662
Doug Zongker211aebc2011-10-28 15:13:10 -0700663 ui->Print("\n-- Wiping data...\n");
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700664 device->WipeData();
Doug Zongkerd4208f92010-09-20 12:16:13 -0700665 erase_volume("/data");
666 erase_volume("/cache");
Doug Zongker211aebc2011-10-28 15:13:10 -0700667 ui->Print("Data wipe complete.\n");
Doug Zongkerf93d8162009-09-22 15:16:02 -0700668}
669
670static void
Doug Zongker6c8553d2012-09-24 10:40:47 -0700671prompt_and_wait(Device* device, int status) {
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700672 const char* const* headers = prepend_title(device->GetMenuHeaders());
Doug Zongkerf93d8162009-09-22 15:16:02 -0700673
674 for (;;) {
675 finish_recovery(NULL);
Doug Zongker6c8553d2012-09-24 10:40:47 -0700676 switch (status) {
677 case INSTALL_SUCCESS:
678 case INSTALL_NONE:
679 ui->SetBackground(RecoveryUI::NO_COMMAND);
680 break;
681
682 case INSTALL_ERROR:
683 case INSTALL_CORRUPT:
684 ui->SetBackground(RecoveryUI::ERROR);
685 break;
686 }
Doug Zongker211aebc2011-10-28 15:13:10 -0700687 ui->SetProgressType(RecoveryUI::EMPTY);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700688
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700689 int chosen_item = get_menu_selection(headers, device->GetMenuItems(), 0, 0, device);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700690
691 // device-specific code may take some action here. It may
692 // return one of the core actions handled in the switch
693 // statement below.
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700694 chosen_item = device->InvokeMenuItem(chosen_item);
Doug Zongkerf93d8162009-09-22 15:16:02 -0700695
Doug Zongkerd0181b82011-10-19 10:51:12 -0700696 int wipe_cache;
Doug Zongkerf93d8162009-09-22 15:16:02 -0700697 switch (chosen_item) {
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700698 case Device::REBOOT:
Doug Zongkerf93d8162009-09-22 15:16:02 -0700699 return;
700
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700701 case Device::WIPE_DATA:
702 wipe_data(ui->IsTextVisible(), device);
Doug Zongker211aebc2011-10-28 15:13:10 -0700703 if (!ui->IsTextVisible()) return;
Doug Zongkerf93d8162009-09-22 15:16:02 -0700704 break;
705
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700706 case Device::WIPE_CACHE:
Doug Zongker211aebc2011-10-28 15:13:10 -0700707 ui->Print("\n-- Wiping cache...\n");
Doug Zongkerd4208f92010-09-20 12:16:13 -0700708 erase_volume("/cache");
Doug Zongker211aebc2011-10-28 15:13:10 -0700709 ui->Print("Cache wipe complete.\n");
710 if (!ui->IsTextVisible()) return;
Doug Zongkerf93d8162009-09-22 15:16:02 -0700711 break;
712
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700713 case Device::APPLY_EXT:
Doug Zongkerd9428e32011-12-13 16:03:28 -0800714 // Some packages expect /cache to be mounted (eg,
715 // standard incremental packages expect to use /cache
716 // as scratch space).
717 ensure_path_mounted(CACHE_ROOT);
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700718 status = update_directory(SDCARD_ROOT, SDCARD_ROOT, &wipe_cache, device);
Doug Zongkerd0181b82011-10-19 10:51:12 -0700719 if (status == INSTALL_SUCCESS && wipe_cache) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700720 ui->Print("\n-- Wiping cache (at package request)...\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700721 if (erase_volume("/cache")) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700722 ui->Print("Cache wipe failed.\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700723 } else {
Doug Zongker211aebc2011-10-28 15:13:10 -0700724 ui->Print("Cache wipe complete.\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700725 }
726 }
Doug Zongker8674a722010-09-15 11:08:23 -0700727 if (status >= 0) {
728 if (status != INSTALL_SUCCESS) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700729 ui->SetBackground(RecoveryUI::ERROR);
730 ui->Print("Installation aborted.\n");
731 } else if (!ui->IsTextVisible()) {
Doug Zongker8674a722010-09-15 11:08:23 -0700732 return; // reboot if logs aren't visible
733 } else {
Doug Zongker211aebc2011-10-28 15:13:10 -0700734 ui->Print("\nInstall from sdcard complete.\n");
Doug Zongker8674a722010-09-15 11:08:23 -0700735 }
Doug Zongkerf93d8162009-09-22 15:16:02 -0700736 }
737 break;
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700738
739 case Device::APPLY_CACHE:
Michael Ward9d2629c2011-06-23 22:14:24 -0700740 // Don't unmount cache at the end of this.
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700741 status = update_directory(CACHE_ROOT, NULL, &wipe_cache, device);
Doug Zongkerd0181b82011-10-19 10:51:12 -0700742 if (status == INSTALL_SUCCESS && wipe_cache) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700743 ui->Print("\n-- Wiping cache (at package request)...\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700744 if (erase_volume("/cache")) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700745 ui->Print("Cache wipe failed.\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700746 } else {
Doug Zongker211aebc2011-10-28 15:13:10 -0700747 ui->Print("Cache wipe complete.\n");
Doug Zongkerd0181b82011-10-19 10:51:12 -0700748 }
749 }
Michael Ward9d2629c2011-06-23 22:14:24 -0700750 if (status >= 0) {
751 if (status != INSTALL_SUCCESS) {
Doug Zongker211aebc2011-10-28 15:13:10 -0700752 ui->SetBackground(RecoveryUI::ERROR);
753 ui->Print("Installation aborted.\n");
754 } else if (!ui->IsTextVisible()) {
Michael Ward9d2629c2011-06-23 22:14:24 -0700755 return; // reboot if logs aren't visible
756 } else {
Doug Zongker211aebc2011-10-28 15:13:10 -0700757 ui->Print("\nInstall from cache complete.\n");
Michael Ward9d2629c2011-06-23 22:14:24 -0700758 }
759 }
760 break;
Doug Zongker9270a202012-01-09 15:16:13 -0800761
762 case Device::APPLY_ADB_SIDELOAD:
763 ensure_path_mounted(CACHE_ROOT);
764 status = apply_from_adb(ui, &wipe_cache, TEMPORARY_INSTALL_FILE);
765 if (status >= 0) {
766 if (status != INSTALL_SUCCESS) {
767 ui->SetBackground(RecoveryUI::ERROR);
768 ui->Print("Installation aborted.\n");
769 } else if (!ui->IsTextVisible()) {
770 return; // reboot if logs aren't visible
771 } else {
772 ui->Print("\nInstall from ADB complete.\n");
773 }
774 }
775 break;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800776 }
777 }
778}
779
780static void
Oscar Montemayor05231562009-11-30 08:40:57 -0800781print_property(const char *key, const char *name, void *cookie) {
Doug Zongker56c51052010-07-01 09:18:44 -0700782 printf("%s=%s\n", key, name);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800783}
784
Doug Zongker02ec6b82012-08-22 17:26:40 -0700785static void
786load_locale_from_cache() {
787 FILE* fp = fopen_path(LOCALE_FILE, "r");
788 char buffer[80];
789 if (fp != NULL) {
790 fgets(buffer, sizeof(buffer), fp);
791 int j = 0;
Doug Zongker5fa8c232012-09-18 12:37:02 -0700792 unsigned int i;
Doug Zongker02ec6b82012-08-22 17:26:40 -0700793 for (i = 0; i < sizeof(buffer) && buffer[i]; ++i) {
794 if (!isspace(buffer[i])) {
795 buffer[j++] = buffer[i];
796 }
797 }
798 buffer[j] = 0;
799 locale = strdup(buffer);
Devin Kim6016d082012-10-08 09:47:37 -0700800 check_and_fclose(fp, LOCALE_FILE);
Doug Zongker02ec6b82012-08-22 17:26:40 -0700801 }
802}
803
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800804int
Oscar Montemayor05231562009-11-30 08:40:57 -0800805main(int argc, char **argv) {
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800806 time_t start = time(NULL);
807
808 // If these fail, there's not really anywhere to complain...
809 freopen(TEMPORARY_LOG_FILE, "a", stdout); setbuf(stdout, NULL);
810 freopen(TEMPORARY_LOG_FILE, "a", stderr); setbuf(stderr, NULL);
Doug Zongker9270a202012-01-09 15:16:13 -0800811
812 // If this binary is started with the single argument "--adbd",
813 // instead of being the normal recovery binary, it turns into kind
814 // of a stripped-down version of adbd that only supports the
815 // 'sideload' command. Note this must be a real argument, not
816 // anything in the command file or bootloader control block; the
817 // only way recovery should be run with this argument is when it
818 // starts a copy of itself from the apply_from_adb() function.
819 if (argc == 2 && strcmp(argv[1], "--adbd") == 0) {
820 adb_main();
821 return 0;
822 }
823
Doug Zongker56c51052010-07-01 09:18:44 -0700824 printf("Starting recovery on %s", ctime(&start));
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800825
Doug Zongkerd4208f92010-09-20 12:16:13 -0700826 load_volume_table();
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800827 get_args(&argc, &argv);
828
829 int previous_runs = 0;
830 const char *send_intent = NULL;
831 const char *update_package = NULL;
Doug Zongker02ec6b82012-08-22 17:26:40 -0700832 int wipe_data = 0, wipe_cache = 0, show_text = 0;
Doug Zongkere5d5ac72012-04-12 11:01:22 -0700833 bool just_exit = false;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800834
835 int arg;
836 while ((arg = getopt_long(argc, argv, "", OPTIONS, NULL)) != -1) {
837 switch (arg) {
838 case 'p': previous_runs = atoi(optarg); break;
839 case 's': send_intent = optarg; break;
840 case 'u': update_package = optarg; break;
841 case 'w': wipe_data = wipe_cache = 1; break;
842 case 'c': wipe_cache = 1; break;
Doug Zongker02ec6b82012-08-22 17:26:40 -0700843 case 't': show_text = 1; break;
Doug Zongkere5d5ac72012-04-12 11:01:22 -0700844 case 'x': just_exit = true; break;
Doug Zongker02ec6b82012-08-22 17:26:40 -0700845 case 'l': locale = optarg; break;
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800846 case '?':
847 LOGE("Invalid command argument\n");
848 continue;
849 }
850 }
851
Doug Zongker02ec6b82012-08-22 17:26:40 -0700852 if (locale == NULL) {
853 load_locale_from_cache();
854 }
855 printf("locale is [%s]\n", locale);
856
857 Device* device = make_device();
858 ui = device->GetUI();
859
860 ui->Init();
Doug Zongker5fa8c232012-09-18 12:37:02 -0700861 ui->SetLocale(locale);
Doug Zongker02ec6b82012-08-22 17:26:40 -0700862 ui->SetBackground(RecoveryUI::NONE);
863 if (show_text) ui->ShowText(true);
864
Stephen Smalley779701d2012-02-09 14:13:23 -0500865 struct selinux_opt seopts[] = {
866 { SELABEL_OPT_PATH, "/file_contexts" }
867 };
868
869 sehandle = selabel_open(SELABEL_CTX_FILE, seopts, 1);
870
871 if (!sehandle) {
872 fprintf(stderr, "Warning: No file_contexts\n");
Kenny Root038818c2012-04-08 11:03:01 -0700873 ui->Print("Warning: No file_contexts\n");
Stephen Smalley779701d2012-02-09 14:13:23 -0500874 }
Stephen Smalley779701d2012-02-09 14:13:23 -0500875
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700876 device->StartRecovery();
Doug Zongkerefa1bab2010-02-01 15:59:12 -0800877
Doug Zongker56c51052010-07-01 09:18:44 -0700878 printf("Command:");
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800879 for (arg = 0; arg < argc; arg++) {
Doug Zongker56c51052010-07-01 09:18:44 -0700880 printf(" \"%s\"", argv[arg]);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800881 }
Doug Zongker9b125b02010-09-22 12:01:37 -0700882 printf("\n");
883
884 if (update_package) {
885 // For backwards compatibility on the cache partition only, if
886 // we're given an old 'root' path "CACHE:foo", change it to
887 // "/cache/foo".
888 if (strncmp(update_package, "CACHE:", 6) == 0) {
889 int len = strlen(update_package) + 10;
Doug Zongker28ce47c2011-10-28 10:33:05 -0700890 char* modified_path = (char*)malloc(len);
Doug Zongker9b125b02010-09-22 12:01:37 -0700891 strlcpy(modified_path, "/cache/", len);
892 strlcat(modified_path, update_package+6, len);
893 printf("(replacing path \"%s\" with \"%s\")\n",
894 update_package, modified_path);
895 update_package = modified_path;
896 }
897 }
898 printf("\n");
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800899
900 property_list(print_property, NULL);
Doug Zongker56c51052010-07-01 09:18:44 -0700901 printf("\n");
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800902
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800903 int status = INSTALL_SUCCESS;
904
Doug Zongker540d57f2011-01-18 13:36:58 -0800905 if (update_package != NULL) {
Doug Zongkerd0181b82011-10-19 10:51:12 -0700906 status = install_package(update_package, &wipe_cache, TEMPORARY_INSTALL_FILE);
907 if (status == INSTALL_SUCCESS && wipe_cache) {
908 if (erase_volume("/cache")) {
909 LOGE("Cache wipe (requested by package) failed.");
910 }
911 }
Doug Zongker211aebc2011-10-28 15:13:10 -0700912 if (status != INSTALL_SUCCESS) ui->Print("Installation aborted.\n");
Doug Zongkerb128f542009-06-18 15:07:14 -0700913 } else if (wipe_data) {
Doug Zongkerdaefc1d2011-10-31 09:34:15 -0700914 if (device->WipeData()) status = INSTALL_ERROR;
Doug Zongkerd4208f92010-09-20 12:16:13 -0700915 if (erase_volume("/data")) status = INSTALL_ERROR;
916 if (wipe_cache && erase_volume("/cache")) status = INSTALL_ERROR;
Doug Zongker211aebc2011-10-28 15:13:10 -0700917 if (status != INSTALL_SUCCESS) ui->Print("Data wipe failed.\n");
Doug Zongkerb128f542009-06-18 15:07:14 -0700918 } else if (wipe_cache) {
Doug Zongkerd4208f92010-09-20 12:16:13 -0700919 if (wipe_cache && erase_volume("/cache")) status = INSTALL_ERROR;
Doug Zongker211aebc2011-10-28 15:13:10 -0700920 if (status != INSTALL_SUCCESS) ui->Print("Cache wipe failed.\n");
Doug Zongkere5d5ac72012-04-12 11:01:22 -0700921 } else if (!just_exit) {
Doug Zongker02ec6b82012-08-22 17:26:40 -0700922 status = INSTALL_NONE; // No command specified
923 ui->SetBackground(RecoveryUI::NO_COMMAND);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800924 }
925
Doug Zongker02ec6b82012-08-22 17:26:40 -0700926 if (status == INSTALL_ERROR || status == INSTALL_CORRUPT) {
927 ui->SetBackground(RecoveryUI::ERROR);
928 }
Doug Zongker211aebc2011-10-28 15:13:10 -0700929 if (status != INSTALL_SUCCESS || ui->IsTextVisible()) {
Doug Zongker6c8553d2012-09-24 10:40:47 -0700930 prompt_and_wait(device, status);
Doug Zongker8674a722010-09-15 11:08:23 -0700931 }
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800932
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800933 // Otherwise, get ready to boot the main system...
934 finish_recovery(send_intent);
Doug Zongker211aebc2011-10-28 15:13:10 -0700935 ui->Print("Rebooting...\n");
Ken Sumrall6e4472a2011-03-07 23:37:27 -0800936 android_reboot(ANDROID_RB_RESTART, 0, 0);
The Android Open Source Projectc24a8e62009-03-03 19:28:42 -0800937 return EXIT_SUCCESS;
938}