Koushik K. Dutta | 6060e5c | 2010-02-11 22:27:06 -0800 | [diff] [blame] | 1 | #include <ctype.h> |
| 2 | #include <errno.h> |
| 3 | #include <fcntl.h> |
| 4 | #include <getopt.h> |
| 5 | #include <limits.h> |
| 6 | #include <linux/input.h> |
| 7 | #include <stdio.h> |
| 8 | #include <stdlib.h> |
| 9 | #include <string.h> |
| 10 | #include <sys/reboot.h> |
| 11 | #include <sys/types.h> |
| 12 | #include <time.h> |
| 13 | #include <unistd.h> |
| 14 | |
Koushik K. Dutta | e923487 | 2010-02-12 00:43:24 -0800 | [diff] [blame] | 15 | #include <sys/wait.h> |
| 16 | #include <sys/limits.h> |
| 17 | #include <dirent.h> |
Koushik K. Dutta | 49f5689 | 2010-02-25 14:51:05 -0800 | [diff] [blame] | 18 | #include <sys/stat.h> |
Koushik K. Dutta | e923487 | 2010-02-12 00:43:24 -0800 | [diff] [blame] | 19 | |
Koushik K. Dutta | 33370db | 2010-02-25 11:39:07 -0800 | [diff] [blame] | 20 | #include <signal.h> |
| 21 | #include <sys/wait.h> |
| 22 | |
Koushik K. Dutta | 6060e5c | 2010-02-11 22:27:06 -0800 | [diff] [blame] | 23 | #include "bootloader.h" |
| 24 | #include "common.h" |
| 25 | #include "cutils/properties.h" |
| 26 | #include "firmware.h" |
| 27 | #include "install.h" |
| 28 | #include "minui/minui.h" |
| 29 | #include "minzip/DirUtil.h" |
| 30 | #include "roots.h" |
| 31 | #include "recovery_ui.h" |
| 32 | |
Koushik K. Dutta | f68aaaf | 2010-03-07 13:39:21 -0800 | [diff] [blame] | 33 | #include "commands.h" |
| 34 | #include "amend/amend.h" |
| 35 | |
Koushik Dutta | 14239d2 | 2010-06-14 15:02:48 -0700 | [diff] [blame] | 36 | #include "mtdutils/mtdutils.h" |
Koushik K. Dutta | a85d7cc | 2010-03-12 17:00:58 -0800 | [diff] [blame] | 37 | #include "mtdutils/dump_image.h" |
| 38 | #include "../../external/yaffs2/yaffs2/utils/mkyaffs2image.h" |
Koushik K. Dutta | 54305a8 | 2010-03-12 17:43:26 -0800 | [diff] [blame] | 39 | #include "../../external/yaffs2/yaffs2/utils/unyaffs.h" |
Koushik K. Dutta | a85d7cc | 2010-03-12 17:00:58 -0800 | [diff] [blame] | 40 | |
Koushik K. Dutta | ee57bbc | 2010-03-12 23:21:12 -0800 | [diff] [blame] | 41 | #include "extendedcommands.h" |
| 42 | #include "nandroid.h" |
| 43 | |
Koushik K. Dutta | 6060e5c | 2010-02-11 22:27:06 -0800 | [diff] [blame] | 44 | int signature_check_enabled = 1; |
| 45 | int script_assert_enabled = 1; |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 46 | static const char *SDCARD_PACKAGE_FILE = "SDCARD:update.zip"; |
Koushik K. Dutta | 6060e5c | 2010-02-11 22:27:06 -0800 | [diff] [blame] | 47 | |
| 48 | void |
| 49 | toggle_signature_check() |
| 50 | { |
| 51 | signature_check_enabled = !signature_check_enabled; |
| 52 | ui_print("Signature Check: %s\n", signature_check_enabled ? "Enabled" : "Disabled"); |
| 53 | } |
| 54 | |
| 55 | void toggle_script_asserts() |
| 56 | { |
| 57 | script_assert_enabled = !script_assert_enabled; |
Koushik K. Dutta | e923487 | 2010-02-12 00:43:24 -0800 | [diff] [blame] | 58 | ui_print("Script Asserts: %s\n", script_assert_enabled ? "Enabled" : "Disabled"); |
| 59 | } |
| 60 | |
Koushik K. Dutta | ea46fe2 | 2010-03-08 02:58:04 -0800 | [diff] [blame] | 61 | int install_zip(const char* packagefilepath) |
Koushik K. Dutta | e923487 | 2010-02-12 00:43:24 -0800 | [diff] [blame] | 62 | { |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 63 | ui_print("\n-- Installing: %s\n", packagefilepath); |
| 64 | set_sdcard_update_bootloader_message(); |
| 65 | int status = install_package(packagefilepath); |
Koushik K. Dutta | f3534d0 | 2010-04-18 18:06:24 -0700 | [diff] [blame] | 66 | ui_reset_progress(); |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 67 | if (status != INSTALL_SUCCESS) { |
| 68 | ui_set_background(BACKGROUND_ICON_ERROR); |
| 69 | ui_print("Installation aborted.\n"); |
Koushik K. Dutta | ea46fe2 | 2010-03-08 02:58:04 -0800 | [diff] [blame] | 70 | return 1; |
Koushik K. Dutta | 79ce82c | 2010-02-25 12:03:17 -0800 | [diff] [blame] | 71 | } |
| 72 | if (firmware_update_pending()) { |
| 73 | ui_print("\nReboot via menu to complete\ninstallation.\n"); |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 74 | } |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 75 | ui_set_background(BACKGROUND_ICON_NONE); |
Koushik K. Dutta | 99fb6fe | 2010-03-03 00:42:58 -0800 | [diff] [blame] | 76 | ui_print("\nInstall from sdcard complete.\n"); |
Koushik K. Dutta | ea46fe2 | 2010-03-08 02:58:04 -0800 | [diff] [blame] | 77 | return 0; |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 78 | } |
| 79 | |
| 80 | char* INSTALL_MENU_ITEMS[] = { "apply sdcard:update.zip", |
| 81 | "choose zip from sdcard", |
| 82 | "toggle signature verification", |
| 83 | "toggle script asserts", |
| 84 | NULL }; |
| 85 | #define ITEM_APPLY_SDCARD 0 |
| 86 | #define ITEM_CHOOSE_ZIP 1 |
| 87 | #define ITEM_SIG_CHECK 2 |
| 88 | #define ITEM_ASSERTS 3 |
| 89 | |
| 90 | void show_install_update_menu() |
| 91 | { |
| 92 | static char* headers[] = { "Apply update from .zip file on SD card", |
| 93 | "", |
| 94 | NULL |
| 95 | }; |
| 96 | for (;;) |
| 97 | { |
| 98 | int chosen_item = get_menu_selection(headers, INSTALL_MENU_ITEMS, 0); |
| 99 | switch (chosen_item) |
| 100 | { |
| 101 | case ITEM_ASSERTS: |
| 102 | toggle_script_asserts(); |
| 103 | break; |
| 104 | case ITEM_SIG_CHECK: |
| 105 | toggle_signature_check(); |
| 106 | break; |
| 107 | case ITEM_APPLY_SDCARD: |
| 108 | install_zip(SDCARD_PACKAGE_FILE); |
| 109 | break; |
| 110 | case ITEM_CHOOSE_ZIP: |
| 111 | show_choose_zip_menu(); |
| 112 | break; |
| 113 | default: |
| 114 | return; |
| 115 | } |
| 116 | |
| 117 | } |
| 118 | } |
| 119 | |
Koushik K. Dutta | 1d53c4e | 2010-04-02 16:46:21 -0700 | [diff] [blame] | 120 | void free_string_array(char** array) |
| 121 | { |
| 122 | if (array == NULL) |
| 123 | return; |
| 124 | char* cursor = array[0]; |
| 125 | int i = 0; |
| 126 | while (cursor != NULL) |
| 127 | { |
| 128 | free(cursor); |
| 129 | cursor = array[++i]; |
| 130 | } |
| 131 | free(array); |
| 132 | } |
| 133 | |
Koushik K. Dutta | 49f5689 | 2010-02-25 14:51:05 -0800 | [diff] [blame] | 134 | char** gather_files(const char* directory, const char* fileExtensionOrDirectory, int* numFiles) |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 135 | { |
Koushik K. Dutta | e923487 | 2010-02-12 00:43:24 -0800 | [diff] [blame] | 136 | char path[PATH_MAX] = ""; |
| 137 | DIR *dir; |
| 138 | struct dirent *de; |
| 139 | int total = 0; |
| 140 | int i; |
Koushik K. Dutta | ee57bbc | 2010-03-12 23:21:12 -0800 | [diff] [blame] | 141 | char** files = NULL; |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 142 | int pass; |
| 143 | *numFiles = 0; |
| 144 | int dirLen = strlen(directory); |
Koushik K. Dutta | e923487 | 2010-02-12 00:43:24 -0800 | [diff] [blame] | 145 | |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 146 | dir = opendir(directory); |
| 147 | if (dir == NULL) { |
| 148 | ui_print("Couldn't open directory.\n"); |
| 149 | return NULL; |
| 150 | } |
| 151 | |
Koushik K. Dutta | ee57bbc | 2010-03-12 23:21:12 -0800 | [diff] [blame] | 152 | int extension_length = 0; |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 153 | if (fileExtensionOrDirectory != NULL) |
| 154 | extension_length = strlen(fileExtensionOrDirectory); |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 155 | |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 156 | int isCounting = 1; |
| 157 | i = 0; |
| 158 | for (pass = 0; pass < 2; pass++) { |
| 159 | while ((de=readdir(dir)) != NULL) { |
| 160 | // skip hidden files |
| 161 | if (de->d_name[0] == '.') |
| 162 | continue; |
| 163 | |
| 164 | // NULL means that we are gathering directories, so skip this |
| 165 | if (fileExtensionOrDirectory != NULL) |
| 166 | { |
| 167 | // make sure that we can have the desired extension (prevent seg fault) |
| 168 | if (strlen(de->d_name) < extension_length) |
| 169 | continue; |
| 170 | // compare the extension |
| 171 | if (strcmp(de->d_name + strlen(de->d_name) - extension_length, fileExtensionOrDirectory) != 0) |
| 172 | continue; |
| 173 | } |
| 174 | else |
| 175 | { |
| 176 | struct stat info; |
Koushik K. Dutta | 1d53c4e | 2010-04-02 16:46:21 -0700 | [diff] [blame] | 177 | char fullFileName[PATH_MAX]; |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 178 | strcpy(fullFileName, directory); |
| 179 | strcat(fullFileName, de->d_name); |
| 180 | stat(fullFileName, &info); |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 181 | // make sure it is a directory |
| 182 | if (!(S_ISDIR(info.st_mode))) |
| 183 | continue; |
| 184 | } |
| 185 | |
| 186 | if (pass == 0) |
| 187 | { |
| 188 | total++; |
| 189 | continue; |
| 190 | } |
| 191 | |
Koushik K. Dutta | 49f5689 | 2010-02-25 14:51:05 -0800 | [diff] [blame] | 192 | files[i] = (char*) malloc(dirLen + strlen(de->d_name) + 2); |
| 193 | strcpy(files[i], directory); |
| 194 | strcat(files[i], de->d_name); |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 195 | if (fileExtensionOrDirectory == NULL) |
| 196 | strcat(files[i], "/"); |
| 197 | i++; |
| 198 | } |
| 199 | if (pass == 1) |
| 200 | break; |
| 201 | if (total == 0) |
| 202 | break; |
| 203 | rewinddir(dir); |
| 204 | *numFiles = total; |
| 205 | files = (char**) malloc((total+1)*sizeof(char*)); |
| 206 | files[total]=NULL; |
| 207 | } |
Koushik K. Dutta | 49f5689 | 2010-02-25 14:51:05 -0800 | [diff] [blame] | 208 | |
| 209 | if(closedir(dir) < 0) { |
| 210 | LOGE("Failed to close directory."); |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 211 | } |
| 212 | |
| 213 | if (total==0) { |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 214 | return NULL; |
| 215 | } |
| 216 | |
Koushik Dutta | 789ab6b | 2010-07-01 00:23:25 -0700 | [diff] [blame] | 217 | // sort the result |
| 218 | if (files != NULL) { |
| 219 | for (i = 0; i < total; i++) { |
| 220 | int curMax = -1; |
| 221 | int j; |
| 222 | for (j = 0; j < total - i; j++) { |
| 223 | if (curMax == -1 || strcmp(files[curMax], files[j]) < 0) |
| 224 | curMax = j; |
| 225 | } |
| 226 | char* temp = files[curMax]; |
| 227 | files[curMax] = files[total - i - 1]; |
| 228 | files[total - i - 1] = temp; |
| 229 | } |
| 230 | } |
| 231 | |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 232 | return files; |
Koushik K. Dutta | 49f5689 | 2010-02-25 14:51:05 -0800 | [diff] [blame] | 233 | } |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 234 | |
Koushik K. Dutta | 49f5689 | 2010-02-25 14:51:05 -0800 | [diff] [blame] | 235 | // pass in NULL for fileExtensionOrDirectory and you will get a directory chooser |
| 236 | char* choose_file_menu(const char* directory, const char* fileExtensionOrDirectory, const char* headers[]) |
| 237 | { |
| 238 | char path[PATH_MAX] = ""; |
| 239 | DIR *dir; |
| 240 | struct dirent *de; |
| 241 | int numFiles = 0; |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 242 | int numDirs = 0; |
Koushik K. Dutta | 49f5689 | 2010-02-25 14:51:05 -0800 | [diff] [blame] | 243 | int i; |
Koushik K. Dutta | 1d53c4e | 2010-04-02 16:46:21 -0700 | [diff] [blame] | 244 | char* return_value = NULL; |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 245 | int dir_len = strlen(directory); |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 246 | |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 247 | char** files = gather_files(directory, fileExtensionOrDirectory, &numFiles); |
Koushik K. Dutta | ee57bbc | 2010-03-12 23:21:12 -0800 | [diff] [blame] | 248 | char** dirs = NULL; |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 249 | if (fileExtensionOrDirectory != NULL) |
| 250 | dirs = gather_files(directory, NULL, &numDirs); |
| 251 | int total = numDirs + numFiles; |
| 252 | if (total == 0) |
| 253 | { |
| 254 | ui_print("No files found.\n"); |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 255 | } |
Koushik K. Dutta | 1d53c4e | 2010-04-02 16:46:21 -0700 | [diff] [blame] | 256 | else |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 257 | { |
Koushik K. Dutta | 1d53c4e | 2010-04-02 16:46:21 -0700 | [diff] [blame] | 258 | char** list = (char**) malloc((total + 1) * sizeof(char*)); |
| 259 | list[total] = NULL; |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 260 | |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 261 | |
Koushik K. Dutta | 1d53c4e | 2010-04-02 16:46:21 -0700 | [diff] [blame] | 262 | for (i = 0 ; i < numDirs; i++) |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 263 | { |
Koushik K. Dutta | 1d53c4e | 2010-04-02 16:46:21 -0700 | [diff] [blame] | 264 | list[i] = strdup(dirs[i] + dir_len); |
| 265 | } |
| 266 | |
| 267 | for (i = 0 ; i < numFiles; i++) |
| 268 | { |
| 269 | list[numDirs + i] = strdup(files[i] + dir_len); |
| 270 | } |
| 271 | |
| 272 | for (;;) |
| 273 | { |
| 274 | int chosen_item = get_menu_selection(headers, list, 0); |
| 275 | if (chosen_item == GO_BACK) |
| 276 | break; |
| 277 | static char ret[PATH_MAX]; |
| 278 | if (chosen_item < numDirs) |
| 279 | { |
| 280 | char* subret = choose_file_menu(dirs[chosen_item], fileExtensionOrDirectory, headers); |
| 281 | if (subret != NULL) |
| 282 | { |
| 283 | strcpy(ret, subret); |
| 284 | return_value = ret; |
| 285 | break; |
| 286 | } |
| 287 | continue; |
| 288 | } |
| 289 | strcpy(ret, files[chosen_item - numDirs]); |
| 290 | return_value = ret; |
| 291 | break; |
| 292 | } |
| 293 | free_string_array(list); |
Koushik K. Dutta | 981b0cd | 2010-02-22 08:53:34 -0800 | [diff] [blame] | 294 | } |
Koushik K. Dutta | 1d53c4e | 2010-04-02 16:46:21 -0700 | [diff] [blame] | 295 | |
| 296 | free_string_array(files); |
| 297 | free_string_array(dirs); |
| 298 | return return_value; |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 299 | } |
| 300 | |
| 301 | void show_choose_zip_menu() |
| 302 | { |
Koushik K. Dutta | e923487 | 2010-02-12 00:43:24 -0800 | [diff] [blame] | 303 | if (ensure_root_path_mounted("SDCARD:") != 0) { |
| 304 | LOGE ("Can't mount /sdcard\n"); |
| 305 | return; |
| 306 | } |
| 307 | |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 308 | static char* headers[] = { "Choose a zip to apply", |
| 309 | "", |
| 310 | NULL |
| 311 | }; |
| 312 | |
| 313 | char* file = choose_file_menu("/sdcard/", ".zip", headers); |
| 314 | if (file == NULL) |
| 315 | return; |
| 316 | char sdcard_package_file[1024]; |
| 317 | strcpy(sdcard_package_file, "SDCARD:"); |
| 318 | strcat(sdcard_package_file, file + strlen("/sdcard/")); |
| 319 | install_zip(sdcard_package_file); |
| 320 | } |
| 321 | |
Koushik K. Dutta | 33370db | 2010-02-25 11:39:07 -0800 | [diff] [blame] | 322 | // This was pulled from bionic: The default system command always looks |
| 323 | // for shell in /system/bin/sh. This is bad. |
| 324 | #define _PATH_BSHELL "/sbin/sh" |
Koushik K. Dutta | ee57bbc | 2010-03-12 23:21:12 -0800 | [diff] [blame] | 325 | |
Koushik K. Dutta | 33370db | 2010-02-25 11:39:07 -0800 | [diff] [blame] | 326 | extern char **environ; |
| 327 | int |
Koushik K. Dutta | ee57bbc | 2010-03-12 23:21:12 -0800 | [diff] [blame] | 328 | __system(const char *command) |
Koushik K. Dutta | 33370db | 2010-02-25 11:39:07 -0800 | [diff] [blame] | 329 | { |
| 330 | pid_t pid; |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 331 | sig_t intsave, quitsave; |
| 332 | sigset_t mask, omask; |
| 333 | int pstat; |
| 334 | char *argp[] = {"sh", "-c", NULL, NULL}; |
Koushik K. Dutta | 33370db | 2010-02-25 11:39:07 -0800 | [diff] [blame] | 335 | |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 336 | if (!command) /* just checking... */ |
| 337 | return(1); |
Koushik K. Dutta | 33370db | 2010-02-25 11:39:07 -0800 | [diff] [blame] | 338 | |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 339 | argp[2] = (char *)command; |
Koushik K. Dutta | 33370db | 2010-02-25 11:39:07 -0800 | [diff] [blame] | 340 | |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 341 | sigemptyset(&mask); |
| 342 | sigaddset(&mask, SIGCHLD); |
| 343 | sigprocmask(SIG_BLOCK, &mask, &omask); |
| 344 | switch (pid = vfork()) { |
| 345 | case -1: /* error */ |
| 346 | sigprocmask(SIG_SETMASK, &omask, NULL); |
| 347 | return(-1); |
| 348 | case 0: /* child */ |
| 349 | sigprocmask(SIG_SETMASK, &omask, NULL); |
| 350 | execve(_PATH_BSHELL, argp, environ); |
Koushik K. Dutta | 33370db | 2010-02-25 11:39:07 -0800 | [diff] [blame] | 351 | _exit(127); |
| 352 | } |
| 353 | |
Koushik K. Dutta | 001c5b5 | 2010-02-25 14:53:57 -0800 | [diff] [blame] | 354 | intsave = (sig_t) bsd_signal(SIGINT, SIG_IGN); |
| 355 | quitsave = (sig_t) bsd_signal(SIGQUIT, SIG_IGN); |
| 356 | pid = waitpid(pid, (int *)&pstat, 0); |
| 357 | sigprocmask(SIG_SETMASK, &omask, NULL); |
| 358 | (void)bsd_signal(SIGINT, intsave); |
| 359 | (void)bsd_signal(SIGQUIT, quitsave); |
| 360 | return (pid == -1 ? -1 : pstat); |
Koushik K. Dutta | 33370db | 2010-02-25 11:39:07 -0800 | [diff] [blame] | 361 | } |
| 362 | |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 363 | void show_nandroid_restore_menu() |
| 364 | { |
Koushik K. Dutta | 54305a8 | 2010-03-12 17:43:26 -0800 | [diff] [blame] | 365 | if (ensure_root_path_mounted("SDCARD:") != 0) { |
| 366 | LOGE ("Can't mount /sdcard\n"); |
Koushik K. Dutta | ee57bbc | 2010-03-12 23:21:12 -0800 | [diff] [blame] | 367 | return; |
Koushik K. Dutta | 54305a8 | 2010-03-12 17:43:26 -0800 | [diff] [blame] | 368 | } |
| 369 | |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 370 | static char* headers[] = { "Choose an image to restore", |
| 371 | "", |
| 372 | NULL |
| 373 | }; |
Koushik K. Dutta | e923487 | 2010-02-12 00:43:24 -0800 | [diff] [blame] | 374 | |
Koushik K. Dutta | 49f5689 | 2010-02-25 14:51:05 -0800 | [diff] [blame] | 375 | char* file = choose_file_menu("/sdcard/clockworkmod/backup/", NULL, headers); |
Koushik K. Dutta | bcdd003 | 2010-02-21 21:10:25 -0800 | [diff] [blame] | 376 | if (file == NULL) |
| 377 | return; |
Koushik K. Dutta | 68b0190 | 2010-04-01 12:20:39 -0700 | [diff] [blame] | 378 | nandroid_restore(file, 1, 1, 1, 1, 1); |
Koushik K. Dutta | 0317378 | 2010-02-26 14:14:23 -0800 | [diff] [blame] | 379 | } |
| 380 | |
Koushik K. Dutta | b9546a8 | 2010-03-14 22:42:30 -0700 | [diff] [blame] | 381 | void show_mount_usb_storage_menu() |
Koushik K. Dutta | 0317378 | 2010-02-26 14:14:23 -0800 | [diff] [blame] | 382 | { |
Koushik Dutta | 598cfc7 | 2010-06-20 09:42:47 -0700 | [diff] [blame] | 383 | char command[PATH_MAX]; |
| 384 | sprintf(command, "echo %s > /sys/devices/platform/usb_mass_storage/lun0/file", SDCARD_DEVICE_PRIMARY); |
| 385 | __system(command); |
Koushik K. Dutta | 0317378 | 2010-02-26 14:14:23 -0800 | [diff] [blame] | 386 | static char* headers[] = { "USB Mass Storage device", |
| 387 | "Leaving this menu unmount", |
| 388 | "your SD card from your PC.", |
| 389 | "", |
| 390 | NULL |
| 391 | }; |
| 392 | |
| 393 | static char* list[] = { "Unmount", NULL }; |
| 394 | |
| 395 | for (;;) |
| 396 | { |
| 397 | int chosen_item = get_menu_selection(headers, list, 0); |
| 398 | if (chosen_item == GO_BACK || chosen_item == 0) |
| 399 | break; |
| 400 | } |
| 401 | |
Koushik K. Dutta | f721594 | 2010-03-16 13:34:51 -0700 | [diff] [blame] | 402 | __system("echo '' > /sys/devices/platform/usb_mass_storage/lun0/file"); |
| 403 | __system("echo 0 > /sys/devices/platform/usb_mass_storage/lun0/enable"); |
Koushik K. Dutta | e81cb75 | 2010-02-26 17:44:33 -0800 | [diff] [blame] | 404 | } |
Koushik K. Dutta | f68aaaf | 2010-03-07 13:39:21 -0800 | [diff] [blame] | 405 | |
Koushik K. Dutta | 16f0b49 | 2010-03-19 14:37:11 -0700 | [diff] [blame] | 406 | int confirm_format() |
| 407 | { |
| 408 | static char* title_headers[] = { "Confirm format?", |
| 409 | " THIS CAN NOT BE UNDONE.", |
| 410 | "", |
| 411 | NULL, |
| 412 | }; |
| 413 | |
| 414 | char* items[] = { " No", |
| 415 | " No", |
| 416 | " No", |
| 417 | " No", |
| 418 | " No", |
| 419 | " No", |
| 420 | " No", |
| 421 | " Yes -- wipe partition", // [7] |
| 422 | " No", |
| 423 | " No", |
| 424 | " No", |
| 425 | NULL }; |
| 426 | |
| 427 | int chosen_item = get_menu_selection(title_headers, items, 0); |
| 428 | return chosen_item == 7; |
| 429 | } |
| 430 | |
Koushik Dutta | 2f73e58 | 2010-04-18 16:00:21 -0700 | [diff] [blame] | 431 | int format_non_mtd_device(const char* root) |
Koushik K. Dutta | 3f99539 | 2010-03-30 23:29:43 -0700 | [diff] [blame] | 432 | { |
Koushik K. Dutta | f3534d0 | 2010-04-18 18:06:24 -0700 | [diff] [blame] | 433 | // if this is SDEXT:, don't worry about it. |
| 434 | if (0 == strcmp(root, "SDEXT:")) |
| 435 | { |
| 436 | struct stat st; |
Koushik Dutta | 598cfc7 | 2010-06-20 09:42:47 -0700 | [diff] [blame] | 437 | if (0 != stat(SDEXT_DEVICE, &st)) |
Koushik K. Dutta | f3534d0 | 2010-04-18 18:06:24 -0700 | [diff] [blame] | 438 | { |
| 439 | ui_print("No app2sd partition found. Skipping format of /sd-ext.\n"); |
| 440 | return 0; |
| 441 | } |
| 442 | } |
Koushik Dutta | 2f73e58 | 2010-04-18 16:00:21 -0700 | [diff] [blame] | 443 | |
Koushik K. Dutta | 3f99539 | 2010-03-30 23:29:43 -0700 | [diff] [blame] | 444 | char path[PATH_MAX]; |
| 445 | translate_root_path(root, path, PATH_MAX); |
| 446 | if (0 != ensure_root_path_mounted(root)) |
| 447 | { |
| 448 | ui_print("Error mounting %s!\n", path); |
Koushik Dutta | cd44ab9 | 2010-06-23 00:02:14 -0700 | [diff] [blame] | 449 | ui_print("Skipping format...\n"); |
| 450 | return 0; |
Koushik K. Dutta | 3f99539 | 2010-03-30 23:29:43 -0700 | [diff] [blame] | 451 | } |
| 452 | |
| 453 | static char tmp[PATH_MAX]; |
| 454 | sprintf(tmp, "rm -rf %s/*", path); |
| 455 | __system(tmp); |
| 456 | sprintf(tmp, "rm -rf %s/.*", path); |
| 457 | __system(tmp); |
| 458 | |
| 459 | ensure_root_path_unmounted(root); |
| 460 | return 0; |
| 461 | } |
| 462 | |
| 463 | #define MOUNTABLE_COUNT 5 |
| 464 | #define MTD_COUNT 4 |
| 465 | #define MMC_COUNT 2 |
Koushik K. Dutta | b9546a8 | 2010-03-14 22:42:30 -0700 | [diff] [blame] | 466 | |
Koushik K. Dutta | 5899ac9 | 2010-03-19 13:34:36 -0700 | [diff] [blame] | 467 | void show_partition_menu() |
Koushik K. Dutta | b9546a8 | 2010-03-14 22:42:30 -0700 | [diff] [blame] | 468 | { |
| 469 | static char* headers[] = { "Mount and unmount partitions", |
| 470 | "", |
| 471 | NULL |
| 472 | }; |
| 473 | |
| 474 | typedef char* string; |
Koushik K. Dutta | 16f0b49 | 2010-03-19 14:37:11 -0700 | [diff] [blame] | 475 | string mounts[MOUNTABLE_COUNT][3] = { |
| 476 | { "mount /system", "unmount /system", "SYSTEM:" }, |
| 477 | { "mount /data", "unmount /data", "DATA:" }, |
| 478 | { "mount /cache", "unmount /cache", "CACHE:" }, |
Koushik K. Dutta | 3f99539 | 2010-03-30 23:29:43 -0700 | [diff] [blame] | 479 | { "mount /sdcard", "unmount /sdcard", "SDCARD:" }, |
| 480 | { "mount /sd-ext", "unmount /sd-ext", "SDEXT:" } |
Koushik K. Dutta | b9546a8 | 2010-03-14 22:42:30 -0700 | [diff] [blame] | 481 | }; |
| 482 | |
Koushik K. Dutta | 16f0b49 | 2010-03-19 14:37:11 -0700 | [diff] [blame] | 483 | string mtds[MTD_COUNT][2] = { |
Koushik K. Dutta | 3f99539 | 2010-03-30 23:29:43 -0700 | [diff] [blame] | 484 | { "format boot", "BOOT:" }, |
| 485 | { "format system", "SYSTEM:" }, |
| 486 | { "format data", "DATA:" }, |
| 487 | { "format cache", "CACHE:" }, |
| 488 | }; |
| 489 | |
| 490 | string mmcs[MMC_COUNT][3] = { |
| 491 | { "format sdcard", "SDCARD:" }, |
Koushik K. Dutta | a5f64e6 | 2010-04-05 15:44:57 -0700 | [diff] [blame] | 492 | { "format sd-ext", "SDEXT:" } |
Koushik K. Dutta | 16f0b49 | 2010-03-19 14:37:11 -0700 | [diff] [blame] | 493 | }; |
| 494 | |
Koushik K. Dutta | b9546a8 | 2010-03-14 22:42:30 -0700 | [diff] [blame] | 495 | for (;;) |
| 496 | { |
Koushik K. Dutta | 5899ac9 | 2010-03-19 13:34:36 -0700 | [diff] [blame] | 497 | int ismounted[MOUNTABLE_COUNT]; |
Koushik K. Dutta | b9546a8 | 2010-03-14 22:42:30 -0700 | [diff] [blame] | 498 | int i; |
Koushik K. Dutta | 3f99539 | 2010-03-30 23:29:43 -0700 | [diff] [blame] | 499 | static string options[MOUNTABLE_COUNT + MTD_COUNT + MMC_COUNT + 1 + 1]; // mountables, format mtds, format mmcs, usb storage, null |
Koushik K. Dutta | 5899ac9 | 2010-03-19 13:34:36 -0700 | [diff] [blame] | 500 | for (i = 0; i < MOUNTABLE_COUNT; i++) |
Koushik K. Dutta | b9546a8 | 2010-03-14 22:42:30 -0700 | [diff] [blame] | 501 | { |
Koushik K. Dutta | 16f0b49 | 2010-03-19 14:37:11 -0700 | [diff] [blame] | 502 | ismounted[i] = is_root_path_mounted(mounts[i][2]); |
Koushik K. Dutta | b9546a8 | 2010-03-14 22:42:30 -0700 | [diff] [blame] | 503 | options[i] = ismounted[i] ? mounts[i][1] : mounts[i][0]; |
| 504 | } |
Koushik K. Dutta | 5899ac9 | 2010-03-19 13:34:36 -0700 | [diff] [blame] | 505 | |
| 506 | for (i = 0; i < MTD_COUNT; i++) |
| 507 | { |
Koushik K. Dutta | 16f0b49 | 2010-03-19 14:37:11 -0700 | [diff] [blame] | 508 | options[MOUNTABLE_COUNT + i] = mtds[i][0]; |
Koushik K. Dutta | 5899ac9 | 2010-03-19 13:34:36 -0700 | [diff] [blame] | 509 | } |
Koushik K. Dutta | 3f99539 | 2010-03-30 23:29:43 -0700 | [diff] [blame] | 510 | |
| 511 | for (i = 0; i < MMC_COUNT; i++) |
| 512 | { |
| 513 | options[MOUNTABLE_COUNT + MTD_COUNT + i] = mmcs[i][0]; |
| 514 | } |
Koushik K. Dutta | b9546a8 | 2010-03-14 22:42:30 -0700 | [diff] [blame] | 515 | |
Koushik K. Dutta | 3f99539 | 2010-03-30 23:29:43 -0700 | [diff] [blame] | 516 | options[MOUNTABLE_COUNT + MTD_COUNT + MMC_COUNT] = "mount USB storage"; |
| 517 | options[MOUNTABLE_COUNT + MTD_COUNT + MMC_COUNT + 1] = NULL; |
Koushik K. Dutta | b9546a8 | 2010-03-14 22:42:30 -0700 | [diff] [blame] | 518 | |
| 519 | int chosen_item = get_menu_selection(headers, options, 0); |
| 520 | if (chosen_item == GO_BACK) |
| 521 | break; |
Koushik K. Dutta | 3f99539 | 2010-03-30 23:29:43 -0700 | [diff] [blame] | 522 | if (chosen_item == MOUNTABLE_COUNT + MTD_COUNT + MMC_COUNT) |
Koushik K. Dutta | b9546a8 | 2010-03-14 22:42:30 -0700 | [diff] [blame] | 523 | { |
| 524 | show_mount_usb_storage_menu(); |
| 525 | } |
Koushik K. Dutta | 5899ac9 | 2010-03-19 13:34:36 -0700 | [diff] [blame] | 526 | else if (chosen_item < MOUNTABLE_COUNT) |
Koushik K. Dutta | b9546a8 | 2010-03-14 22:42:30 -0700 | [diff] [blame] | 527 | { |
| 528 | if (ismounted[chosen_item]) |
| 529 | { |
Koushik K. Dutta | 16f0b49 | 2010-03-19 14:37:11 -0700 | [diff] [blame] | 530 | if (0 != ensure_root_path_unmounted(mounts[chosen_item][2])) |
| 531 | ui_print("Error unmounting %s!\n", mounts[chosen_item][2]); |
Koushik K. Dutta | b9546a8 | 2010-03-14 22:42:30 -0700 | [diff] [blame] | 532 | } |
| 533 | else |
| 534 | { |
Koushik K. Dutta | 16f0b49 | 2010-03-19 14:37:11 -0700 | [diff] [blame] | 535 | if (0 != ensure_root_path_mounted(mounts[chosen_item][2])) |
| 536 | ui_print("Error mounting %s!\n", mounts[chosen_item][2]); |
Koushik K. Dutta | b9546a8 | 2010-03-14 22:42:30 -0700 | [diff] [blame] | 537 | } |
| 538 | } |
Koushik K. Dutta | 5899ac9 | 2010-03-19 13:34:36 -0700 | [diff] [blame] | 539 | else if (chosen_item < MOUNTABLE_COUNT + MTD_COUNT) |
| 540 | { |
| 541 | chosen_item = chosen_item - MOUNTABLE_COUNT; |
Koushik K. Dutta | 16f0b49 | 2010-03-19 14:37:11 -0700 | [diff] [blame] | 542 | if (!confirm_format()) |
| 543 | continue; |
| 544 | ui_print("Formatting %s...\n", mtds[chosen_item][1]); |
Koushik K. Dutta | 3f99539 | 2010-03-30 23:29:43 -0700 | [diff] [blame] | 545 | if (0 != format_root_device(mtds[chosen_item][1])) |
| 546 | ui_print("Error formatting %s!\n", mtds[chosen_item][1]); |
| 547 | else |
| 548 | ui_print("Done.\n"); |
| 549 | } |
| 550 | else if (chosen_item < MOUNTABLE_COUNT + MTD_COUNT + MMC_COUNT) |
| 551 | { |
| 552 | chosen_item = chosen_item - MOUNTABLE_COUNT - MTD_COUNT; |
| 553 | if (!confirm_format()) |
| 554 | continue; |
| 555 | ui_print("Formatting %s...\n", mmcs[chosen_item][1]); |
Koushik Dutta | 2f73e58 | 2010-04-18 16:00:21 -0700 | [diff] [blame] | 556 | if (0 != format_non_mtd_device(mmcs[chosen_item][1])) |
Koushik K. Dutta | 3f99539 | 2010-03-30 23:29:43 -0700 | [diff] [blame] | 557 | ui_print("Error formatting %s!\n", mmcs[chosen_item][1]); |
| 558 | else |
| 559 | ui_print("Done.\n"); |
Koushik K. Dutta | 5899ac9 | 2010-03-19 13:34:36 -0700 | [diff] [blame] | 560 | } |
Koushik K. Dutta | b9546a8 | 2010-03-14 22:42:30 -0700 | [diff] [blame] | 561 | } |
| 562 | } |
| 563 | |
Koushik K. Dutta | 72a1db6 | 2010-03-07 14:10:26 -0800 | [diff] [blame] | 564 | #define EXTENDEDCOMMAND_SCRIPT "/cache/recovery/extendedcommand" |
| 565 | |
| 566 | int extendedcommand_file_exists() |
Koushik K. Dutta | f68aaaf | 2010-03-07 13:39:21 -0800 | [diff] [blame] | 567 | { |
Koushik K. Dutta | f68aaaf | 2010-03-07 13:39:21 -0800 | [diff] [blame] | 568 | struct stat file_info; |
Koushik K. Dutta | 72a1db6 | 2010-03-07 14:10:26 -0800 | [diff] [blame] | 569 | return 0 == stat(EXTENDEDCOMMAND_SCRIPT, &file_info); |
| 570 | } |
| 571 | |
Koushik K. Dutta | 3836f72 | 2010-03-11 22:17:43 -0800 | [diff] [blame] | 572 | int run_script_from_buffer(char* script_data, int script_len, char* filename) |
Koushik K. Dutta | 72a1db6 | 2010-03-07 14:10:26 -0800 | [diff] [blame] | 573 | { |
Koushik K. Dutta | f68aaaf | 2010-03-07 13:39:21 -0800 | [diff] [blame] | 574 | /* Parse the script. Note that the script and parse tree are never freed. |
| 575 | */ |
| 576 | const AmCommandList *commands = parseAmendScript(script_data, script_len); |
| 577 | if (commands == NULL) { |
| 578 | printf("Syntax error in update script\n"); |
| 579 | return 1; |
| 580 | } else { |
Koushik K. Dutta | 72a1db6 | 2010-03-07 14:10:26 -0800 | [diff] [blame] | 581 | printf("Parsed %.*s\n", script_len, filename); |
Koushik K. Dutta | f68aaaf | 2010-03-07 13:39:21 -0800 | [diff] [blame] | 582 | } |
| 583 | |
| 584 | /* Execute the script. |
| 585 | */ |
| 586 | int ret = execCommandList((ExecContext *)1, commands); |
| 587 | if (ret != 0) { |
| 588 | int num = ret; |
Koushik K. Dutta | ee57bbc | 2010-03-12 23:21:12 -0800 | [diff] [blame] | 589 | char *line = NULL, *next = script_data; |
Koushik K. Dutta | f68aaaf | 2010-03-07 13:39:21 -0800 | [diff] [blame] | 590 | while (next != NULL && ret-- > 0) { |
| 591 | line = next; |
| 592 | next = memchr(line, '\n', script_data + script_len - line); |
| 593 | if (next != NULL) *next++ = '\0'; |
| 594 | } |
| 595 | printf("Failure at line %d:\n%s\n", num, next ? line : "(not found)"); |
| 596 | return 1; |
| 597 | } |
| 598 | |
| 599 | return 0; |
| 600 | } |
Koushik K. Dutta | 72a1db6 | 2010-03-07 14:10:26 -0800 | [diff] [blame] | 601 | |
Koushik K. Dutta | 6771aca | 2010-04-03 23:28:39 -0700 | [diff] [blame] | 602 | int run_script(char* filename) |
Koushik K. Dutta | 3836f72 | 2010-03-11 22:17:43 -0800 | [diff] [blame] | 603 | { |
| 604 | struct stat file_info; |
| 605 | if (0 != stat(filename, &file_info)) { |
| 606 | printf("Error executing stat on file: %s\n", filename); |
| 607 | return 1; |
| 608 | } |
| 609 | |
| 610 | int script_len = file_info.st_size; |
Koushik K. Dutta | 707fa6d | 2010-03-23 11:44:33 -0700 | [diff] [blame] | 611 | char* script_data = (char*)malloc(script_len + 1); |
Koushik K. Dutta | 3836f72 | 2010-03-11 22:17:43 -0800 | [diff] [blame] | 612 | FILE *file = fopen(filename, "rb"); |
| 613 | fread(script_data, script_len, 1, file); |
Koushik K. Dutta | 707fa6d | 2010-03-23 11:44:33 -0700 | [diff] [blame] | 614 | // supposedly not necessary, but let's be safe. |
| 615 | script_data[script_len] = '\0'; |
Koushik K. Dutta | 3836f72 | 2010-03-11 22:17:43 -0800 | [diff] [blame] | 616 | fclose(file); |
Koushik Dutta | e25908b | 2010-06-21 08:16:19 -0700 | [diff] [blame] | 617 | LOGI("Running script:\n"); |
| 618 | LOGI("\n%s\n", script_data); |
Koushik K. Dutta | 3836f72 | 2010-03-11 22:17:43 -0800 | [diff] [blame] | 619 | |
Koushik K. Dutta | f3534d0 | 2010-04-18 18:06:24 -0700 | [diff] [blame] | 620 | int ret = run_script_from_buffer(script_data, script_len, filename); |
Koushik K. Dutta | 1d53c4e | 2010-04-02 16:46:21 -0700 | [diff] [blame] | 621 | free(script_data); |
| 622 | return ret; |
Koushik K. Dutta | 3836f72 | 2010-03-11 22:17:43 -0800 | [diff] [blame] | 623 | } |
| 624 | |
Koushik K. Dutta | 72a1db6 | 2010-03-07 14:10:26 -0800 | [diff] [blame] | 625 | int run_and_remove_extendedcommand() |
| 626 | { |
Koushik K. Dutta | 6771aca | 2010-04-03 23:28:39 -0700 | [diff] [blame] | 627 | char tmp[PATH_MAX]; |
| 628 | sprintf(tmp, "cp %s /tmp/%s", EXTENDEDCOMMAND_SCRIPT, basename(EXTENDEDCOMMAND_SCRIPT)); |
| 629 | __system(tmp); |
| 630 | remove(EXTENDEDCOMMAND_SCRIPT); |
Koushik K. Dutta | 3a25cf5 | 2010-03-08 19:22:41 -0800 | [diff] [blame] | 631 | int i = 0; |
| 632 | for (i = 20; i > 0; i--) { |
| 633 | ui_print("Waiting for SD Card to mount (%ds)\n", i); |
| 634 | if (ensure_root_path_mounted("SDCARD:") == 0) { |
| 635 | ui_print("SD Card mounted...\n"); |
| 636 | break; |
| 637 | } |
| 638 | sleep(1); |
| 639 | } |
| 640 | if (i == 0) { |
| 641 | ui_print("Timed out waiting for SD card... continuing anyways."); |
| 642 | } |
| 643 | |
Koushik K. Dutta | 6771aca | 2010-04-03 23:28:39 -0700 | [diff] [blame] | 644 | sprintf(tmp, "/tmp/%s", basename(EXTENDEDCOMMAND_SCRIPT)); |
| 645 | return run_script(tmp); |
Koushik K. Dutta | 72a1db6 | 2010-03-07 14:10:26 -0800 | [diff] [blame] | 646 | } |
| 647 | |
| 648 | int amend_main(int argc, char** argv) |
| 649 | { |
| 650 | if (argc != 2) |
| 651 | { |
| 652 | printf("Usage: amend <script>\n"); |
| 653 | return 0; |
| 654 | } |
| 655 | |
| 656 | RecoveryCommandContext ctx = { NULL }; |
| 657 | if (register_update_commands(&ctx)) { |
| 658 | LOGE("Can't install update commands\n"); |
| 659 | } |
Koushik K. Dutta | 6771aca | 2010-04-03 23:28:39 -0700 | [diff] [blame] | 660 | return run_script(argv[1]); |
Koushik K. Dutta | 5306db5 | 2010-03-08 14:09:35 -0800 | [diff] [blame] | 661 | } |
Koushik K. Dutta | 5899ac9 | 2010-03-19 13:34:36 -0700 | [diff] [blame] | 662 | |
Koushik K. Dutta | fe84a7f | 2010-03-25 18:19:23 -0700 | [diff] [blame] | 663 | void show_nandroid_advanced_restore_menu() |
| 664 | { |
| 665 | if (ensure_root_path_mounted("SDCARD:") != 0) { |
| 666 | LOGE ("Can't mount /sdcard\n"); |
| 667 | return; |
| 668 | } |
| 669 | |
| 670 | static char* advancedheaders[] = { "Choose an image to restore", |
| 671 | "", |
| 672 | "Choose an image to restore", |
| 673 | "first. The next menu will", |
| 674 | "you more options.", |
| 675 | "", |
| 676 | NULL |
| 677 | }; |
| 678 | |
| 679 | char* file = choose_file_menu("/sdcard/clockworkmod/backup/", NULL, advancedheaders); |
| 680 | if (file == NULL) |
| 681 | return; |
| 682 | |
| 683 | static char* headers[] = { "Nandroid Advanced Restore", |
| 684 | "", |
| 685 | NULL |
| 686 | }; |
| 687 | |
| 688 | static char* list[] = { "Restore boot", |
| 689 | "Restore system", |
| 690 | "Restore data", |
| 691 | "Restore cache", |
Koushik K. Dutta | 68b0190 | 2010-04-01 12:20:39 -0700 | [diff] [blame] | 692 | "Restore sd-ext", |
Koushik K. Dutta | fe84a7f | 2010-03-25 18:19:23 -0700 | [diff] [blame] | 693 | NULL |
| 694 | }; |
| 695 | |
| 696 | int chosen_item = get_menu_selection(headers, list, 0); |
| 697 | switch (chosen_item) |
| 698 | { |
| 699 | case 0: |
Koushik K. Dutta | 68b0190 | 2010-04-01 12:20:39 -0700 | [diff] [blame] | 700 | nandroid_restore(file, 1, 0, 0, 0, 0); |
Koushik K. Dutta | fe84a7f | 2010-03-25 18:19:23 -0700 | [diff] [blame] | 701 | break; |
| 702 | case 1: |
Koushik K. Dutta | 68b0190 | 2010-04-01 12:20:39 -0700 | [diff] [blame] | 703 | nandroid_restore(file, 0, 1, 0, 0, 0); |
Koushik K. Dutta | fe84a7f | 2010-03-25 18:19:23 -0700 | [diff] [blame] | 704 | break; |
| 705 | case 2: |
Koushik K. Dutta | 68b0190 | 2010-04-01 12:20:39 -0700 | [diff] [blame] | 706 | nandroid_restore(file, 0, 0, 1, 0, 0); |
Koushik K. Dutta | fe84a7f | 2010-03-25 18:19:23 -0700 | [diff] [blame] | 707 | break; |
| 708 | case 3: |
Koushik K. Dutta | 68b0190 | 2010-04-01 12:20:39 -0700 | [diff] [blame] | 709 | nandroid_restore(file, 0, 0, 0, 1, 0); |
Koushik K. Dutta | fe84a7f | 2010-03-25 18:19:23 -0700 | [diff] [blame] | 710 | break; |
Koushik Dutta | 2f73e58 | 2010-04-18 16:00:21 -0700 | [diff] [blame] | 711 | case 4: |
| 712 | nandroid_restore(file, 0, 0, 0, 0, 1); |
| 713 | break; |
Koushik K. Dutta | fe84a7f | 2010-03-25 18:19:23 -0700 | [diff] [blame] | 714 | } |
| 715 | } |
| 716 | |
Koushik K. Dutta | 5899ac9 | 2010-03-19 13:34:36 -0700 | [diff] [blame] | 717 | void show_nandroid_menu() |
| 718 | { |
| 719 | static char* headers[] = { "Nandroid", |
| 720 | "", |
| 721 | NULL |
| 722 | }; |
| 723 | |
| 724 | static char* list[] = { "Backup", |
| 725 | "Restore", |
Koushik K. Dutta | fe84a7f | 2010-03-25 18:19:23 -0700 | [diff] [blame] | 726 | "Advanced Restore", |
Koushik K. Dutta | 5899ac9 | 2010-03-19 13:34:36 -0700 | [diff] [blame] | 727 | NULL |
| 728 | }; |
| 729 | |
| 730 | int chosen_item = get_menu_selection(headers, list, 0); |
| 731 | switch (chosen_item) |
| 732 | { |
| 733 | case 0: |
| 734 | { |
Koushik K. Dutta | 5899ac9 | 2010-03-19 13:34:36 -0700 | [diff] [blame] | 735 | char backup_path[PATH_MAX]; |
Koushik K. Dutta | 6a26e7c | 2010-03-27 15:26:11 -0700 | [diff] [blame] | 736 | time_t t = time(NULL); |
| 737 | struct tm *tmp = localtime(&t); |
| 738 | if (tmp == NULL) |
| 739 | { |
| 740 | struct timeval tp; |
| 741 | gettimeofday(&tp, NULL); |
| 742 | sprintf(backup_path, "/sdcard/clockworkmod/backup/%d", tp.tv_sec); |
| 743 | } |
| 744 | else |
| 745 | { |
| 746 | strftime(backup_path, sizeof(backup_path), "/sdcard/clockworkmod/backup/%F.%H.%M.%S", tmp); |
| 747 | } |
Koushik K. Dutta | 5899ac9 | 2010-03-19 13:34:36 -0700 | [diff] [blame] | 748 | nandroid_backup(backup_path); |
| 749 | } |
| 750 | break; |
| 751 | case 1: |
| 752 | show_nandroid_restore_menu(); |
| 753 | break; |
Koushik K. Dutta | fe84a7f | 2010-03-25 18:19:23 -0700 | [diff] [blame] | 754 | case 2: |
| 755 | show_nandroid_advanced_restore_menu(); |
| 756 | break; |
Koushik K. Dutta | 5899ac9 | 2010-03-19 13:34:36 -0700 | [diff] [blame] | 757 | } |
Koushik K. Dutta | a496b51 | 2010-03-19 14:51:45 -0700 | [diff] [blame] | 758 | } |
| 759 | |
Koushik Dutta | fd1579b | 2010-05-01 12:46:55 -0700 | [diff] [blame] | 760 | void wipe_battery_stats() |
| 761 | { |
| 762 | ensure_root_path_mounted("DATA:"); |
| 763 | remove("/data/system/batterystats.bin"); |
| 764 | ensure_root_path_unmounted("DATA:"); |
| 765 | } |
| 766 | |
Koushik K. Dutta | a496b51 | 2010-03-19 14:51:45 -0700 | [diff] [blame] | 767 | void show_advanced_menu() |
| 768 | { |
| 769 | static char* headers[] = { "Advanced and Debugging Menu", |
| 770 | "", |
| 771 | NULL |
| 772 | }; |
| 773 | |
Koushik K. Dutta | 7fe4d7b | 2010-04-06 23:04:52 -0700 | [diff] [blame] | 774 | static char* list[] = { "Reboot Recovery", |
Koushik Dutta | 49af23c | 2010-06-21 13:45:51 -0700 | [diff] [blame] | 775 | "Wipe Dalvik Cache", |
Koushik Dutta | fd1579b | 2010-05-01 12:46:55 -0700 | [diff] [blame] | 776 | "Wipe Battery Stats", |
Koushik Dutta | 598cfc7 | 2010-06-20 09:42:47 -0700 | [diff] [blame] | 777 | "Report Error", |
Koushik K. Dutta | 7fe4d7b | 2010-04-06 23:04:52 -0700 | [diff] [blame] | 778 | "Key Test", |
Koushik K. Dutta | a496b51 | 2010-03-19 14:51:45 -0700 | [diff] [blame] | 779 | NULL |
| 780 | }; |
| 781 | |
| 782 | for (;;) |
| 783 | { |
| 784 | int chosen_item = get_menu_selection(headers, list, 0); |
| 785 | if (chosen_item == GO_BACK) |
| 786 | break; |
| 787 | switch (chosen_item) |
| 788 | { |
| 789 | case 0: |
Koushik K. Dutta | 7fe4d7b | 2010-04-06 23:04:52 -0700 | [diff] [blame] | 790 | __reboot(LINUX_REBOOT_MAGIC1, LINUX_REBOOT_MAGIC2, LINUX_REBOOT_CMD_RESTART2, "recovery"); |
| 791 | break; |
| 792 | case 1: |
Koushik Dutta | 49af23c | 2010-06-21 13:45:51 -0700 | [diff] [blame] | 793 | if (0 != ensure_root_path_mounted("DATA:")) |
| 794 | break; |
| 795 | __system("rm -r /data/dalvik-cache"); |
| 796 | ensure_root_path_unmounted("DATA:"); |
Koushik Dutta | fd1579b | 2010-05-01 12:46:55 -0700 | [diff] [blame] | 797 | break; |
| 798 | case 2: |
Koushik Dutta | 49af23c | 2010-06-21 13:45:51 -0700 | [diff] [blame] | 799 | wipe_battery_stats(); |
Koushik Dutta | 598cfc7 | 2010-06-20 09:42:47 -0700 | [diff] [blame] | 800 | break; |
| 801 | case 3: |
Koushik Dutta | 49af23c | 2010-06-21 13:45:51 -0700 | [diff] [blame] | 802 | handle_failure(1); |
| 803 | break; |
| 804 | case 4: |
Koushik K. Dutta | a496b51 | 2010-03-19 14:51:45 -0700 | [diff] [blame] | 805 | { |
| 806 | ui_print("Outputting key codes.\n"); |
| 807 | ui_print("Go back to end debugging.\n"); |
| 808 | int key; |
| 809 | int action; |
| 810 | do |
| 811 | { |
| 812 | key = ui_wait_key(); |
| 813 | action = device_handle_key(key, 1); |
| 814 | ui_print("Key: %d\n", key); |
| 815 | } |
| 816 | while (action != GO_BACK); |
| 817 | } |
| 818 | } |
| 819 | } |
Koushik Dutta | 14239d2 | 2010-06-14 15:02:48 -0700 | [diff] [blame] | 820 | } |
| 821 | |
| 822 | void write_fstab_root(char *root_path, FILE *file) |
| 823 | { |
| 824 | RootInfo *info = get_root_info_for_path(root_path); |
Koushik Dutta | a6522b3 | 2010-06-20 13:16:06 -0700 | [diff] [blame] | 825 | if (info == NULL) { |
| 826 | LOGW("Unable to get root info for %s during fstab generation!", root_path); |
Koushik Dutta | 598cfc7 | 2010-06-20 09:42:47 -0700 | [diff] [blame] | 827 | return; |
Koushik Dutta | a6522b3 | 2010-06-20 13:16:06 -0700 | [diff] [blame] | 828 | } |
Koushik Dutta | 14239d2 | 2010-06-14 15:02:48 -0700 | [diff] [blame] | 829 | MtdPartition *mtd = get_root_mtd_partition(root_path); |
| 830 | if (mtd != NULL) |
| 831 | { |
| 832 | fprintf(file, "/dev/block/mtdblock%d ", mtd->device_index); |
| 833 | } |
| 834 | else |
| 835 | { |
Koushik Dutta | 8ec9418 | 2010-06-21 12:27:43 -0700 | [diff] [blame] | 836 | // only SDCARD: seems to be using device2. |
| 837 | // and mmcblkXp1 is the fallback/device2. |
| 838 | // However, generally, mmcblkXp1 is usually where the |
| 839 | // FAT partition is located... so favor that. |
| 840 | if (NULL == info->device2) |
| 841 | fprintf(file, "%s ", info->device); |
| 842 | else |
| 843 | fprintf(file, "%s ", info->device2); |
Koushik Dutta | 14239d2 | 2010-06-14 15:02:48 -0700 | [diff] [blame] | 844 | } |
| 845 | |
| 846 | fprintf(file, "%s ", info->mount_point); |
| 847 | fprintf(file, "%s rw\n", info->filesystem); |
| 848 | } |
| 849 | |
| 850 | void create_fstab() |
| 851 | { |
Koushik Dutta | cd44ab9 | 2010-06-23 00:02:14 -0700 | [diff] [blame] | 852 | __system("touch /etc/mtab"); |
Koushik Dutta | 14239d2 | 2010-06-14 15:02:48 -0700 | [diff] [blame] | 853 | FILE *file = fopen("/etc/fstab", "w"); |
Koushik Dutta | a6522b3 | 2010-06-20 13:16:06 -0700 | [diff] [blame] | 854 | if (file == NULL) { |
| 855 | LOGW("Unable to create /etc/fstab!"); |
Koushik Dutta | 598cfc7 | 2010-06-20 09:42:47 -0700 | [diff] [blame] | 856 | return; |
Koushik Dutta | a6522b3 | 2010-06-20 13:16:06 -0700 | [diff] [blame] | 857 | } |
Koushik Dutta | 14239d2 | 2010-06-14 15:02:48 -0700 | [diff] [blame] | 858 | write_fstab_root("CACHE:", file); |
| 859 | write_fstab_root("DATA:", file); |
Koushik Dutta | a6522b3 | 2010-06-20 13:16:06 -0700 | [diff] [blame] | 860 | #ifdef HAS_DATADATA |
| 861 | write_fstab_root("DATADATA:", file); |
| 862 | #endif |
Koushik Dutta | 14239d2 | 2010-06-14 15:02:48 -0700 | [diff] [blame] | 863 | write_fstab_root("SYSTEM:", file); |
| 864 | write_fstab_root("SDCARD:", file); |
| 865 | write_fstab_root("SDEXT:", file); |
| 866 | fclose(file); |
Koushik Dutta | 598cfc7 | 2010-06-20 09:42:47 -0700 | [diff] [blame] | 867 | } |
| 868 | |
| 869 | void handle_failure(int ret) |
| 870 | { |
| 871 | if (ret == 0) |
| 872 | return; |
| 873 | if (0 != ensure_root_path_mounted("SDCARD:")) |
| 874 | return; |
Koushik Dutta | a6522b3 | 2010-06-20 13:16:06 -0700 | [diff] [blame] | 875 | mkdir("/sdcard/clockworkmod", S_IRWXU); |
| 876 | __system("cp /tmp/recovery.log /sdcard/clockworkmod/recovery.log"); |
Koushik Dutta | 598cfc7 | 2010-06-20 09:42:47 -0700 | [diff] [blame] | 877 | ui_print("/tmp/recovery.log was copied to /sdcard/clockworkmod/recovery.log. Please open ROM Manager to report the issue."); |
Koushik K. Dutta | 5899ac9 | 2010-03-19 13:34:36 -0700 | [diff] [blame] | 878 | } |