760 lines
		
	
	
		
			21 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			760 lines
		
	
	
		
			21 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| #include <ctype.h>
 | |
| #include <errno.h>
 | |
| #include <fcntl.h>
 | |
| #include <getopt.h>
 | |
| #include <limits.h>
 | |
| #include <linux/input.h>
 | |
| #include <stdio.h>
 | |
| #include <stdlib.h>
 | |
| #include <string.h>
 | |
| #include <sys/reboot.h>
 | |
| #include <sys/types.h>
 | |
| #include <time.h>
 | |
| #include <unistd.h>
 | |
| 
 | |
| #include <sys/wait.h>
 | |
| #include <sys/limits.h>
 | |
| #include <dirent.h>
 | |
| #include <sys/stat.h>
 | |
| 
 | |
| #include <signal.h>
 | |
| #include <sys/wait.h>
 | |
| 
 | |
| #include "bootloader.h"
 | |
| #include "common.h"
 | |
| #include "cutils/properties.h"
 | |
| #include "firmware.h"
 | |
| #include "install.h"
 | |
| #include "minui/minui.h"
 | |
| #include "minzip/DirUtil.h"
 | |
| #include "roots.h"
 | |
| #include "recovery_ui.h"
 | |
| 
 | |
| #include "commands.h"
 | |
| #include "amend/amend.h"
 | |
| 
 | |
| #include "mtdutils/dump_image.h"
 | |
| #include "../../external/yaffs2/yaffs2/utils/mkyaffs2image.h"
 | |
| #include "../../external/yaffs2/yaffs2/utils/unyaffs.h"
 | |
| 
 | |
| #include "extendedcommands.h"
 | |
| #include "nandroid.h"
 | |
| 
 | |
| int signature_check_enabled = 1;
 | |
| int script_assert_enabled = 1;
 | |
| static const char *SDCARD_PACKAGE_FILE = "SDCARD:update.zip";
 | |
| 
 | |
| void
 | |
| toggle_signature_check()
 | |
| {
 | |
|     signature_check_enabled = !signature_check_enabled;
 | |
|     ui_print("Signature Check: %s\n", signature_check_enabled ? "Enabled" : "Disabled");
 | |
| }
 | |
| 
 | |
| void toggle_script_asserts()
 | |
| {
 | |
|     script_assert_enabled = !script_assert_enabled;
 | |
|     ui_print("Script Asserts: %s\n", script_assert_enabled ? "Enabled" : "Disabled");
 | |
| }
 | |
| 
 | |
| int install_zip(const char* packagefilepath)
 | |
| {
 | |
|     ui_print("\n-- Installing: %s\n", packagefilepath);
 | |
|     set_sdcard_update_bootloader_message();
 | |
|     int status = install_package(packagefilepath);
 | |
| 	ui_reset_progress();
 | |
|     if (status != INSTALL_SUCCESS) {
 | |
|         ui_set_background(BACKGROUND_ICON_ERROR);
 | |
|         ui_print("Installation aborted.\n");
 | |
|         return 1;
 | |
|     } 
 | |
|     if (firmware_update_pending()) {
 | |
|         ui_print("\nReboot via menu to complete\ninstallation.\n");
 | |
|     }
 | |
|     ui_set_background(BACKGROUND_ICON_NONE);
 | |
|     ui_print("\nInstall from sdcard complete.\n");
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| char* INSTALL_MENU_ITEMS[] = {  "apply sdcard:update.zip",
 | |
|                                 "choose zip from sdcard",
 | |
|                                 "toggle signature verification",
 | |
|                                 "toggle script asserts",
 | |
|                                 NULL };
 | |
| #define ITEM_APPLY_SDCARD     0
 | |
| #define ITEM_CHOOSE_ZIP       1
 | |
| #define ITEM_SIG_CHECK        2
 | |
| #define ITEM_ASSERTS          3
 | |
| 
 | |
| void show_install_update_menu()
 | |
| {
 | |
|     static char* headers[] = {  "Apply update from .zip file on SD card",
 | |
|                                 "",
 | |
|                                 NULL 
 | |
|     };
 | |
|     for (;;)
 | |
|     {
 | |
|         int chosen_item = get_menu_selection(headers, INSTALL_MENU_ITEMS, 0);
 | |
|         switch (chosen_item)
 | |
|         {
 | |
|             case ITEM_ASSERTS:
 | |
|                 toggle_script_asserts();
 | |
|                 break;
 | |
|             case ITEM_SIG_CHECK:
 | |
|                 toggle_signature_check();
 | |
|                 break;
 | |
|             case ITEM_APPLY_SDCARD:
 | |
|                 install_zip(SDCARD_PACKAGE_FILE);
 | |
|                 break;
 | |
|             case ITEM_CHOOSE_ZIP:
 | |
|                 show_choose_zip_menu();
 | |
|                 break;
 | |
|             default:
 | |
|                 return;
 | |
|         }
 | |
|         
 | |
|     }
 | |
| }
 | |
| 
 | |
| void free_string_array(char** array)
 | |
| {
 | |
|     if (array == NULL)
 | |
|         return;
 | |
|     char* cursor = array[0];
 | |
|     int i = 0;
 | |
|     while (cursor != NULL)
 | |
|     {
 | |
|         free(cursor);
 | |
|         cursor = array[++i];
 | |
|     }
 | |
|     free(array);
 | |
| }
 | |
| 
 | |
| char** gather_files(const char* directory, const char* fileExtensionOrDirectory, int* numFiles)
 | |
| {
 | |
|     char path[PATH_MAX] = "";
 | |
|     DIR *dir;
 | |
|     struct dirent *de;
 | |
|     int total = 0;
 | |
|     int i;
 | |
|     char** files = NULL;
 | |
|     int pass;
 | |
|     *numFiles = 0;
 | |
|     int dirLen = strlen(directory);
 | |
| 
 | |
|     dir = opendir(directory);
 | |
|     if (dir == NULL) {
 | |
|         ui_print("Couldn't open directory.\n");
 | |
|         return NULL;
 | |
|     }
 | |
|   
 | |
|     int extension_length = 0;
 | |
|     if (fileExtensionOrDirectory != NULL)
 | |
|         extension_length = strlen(fileExtensionOrDirectory);
 | |
|   
 | |
|     int isCounting = 1;
 | |
|     i = 0;
 | |
|     for (pass = 0; pass < 2; pass++) {
 | |
|         while ((de=readdir(dir)) != NULL) {
 | |
|             // skip hidden files
 | |
|             if (de->d_name[0] == '.')
 | |
|                 continue;
 | |
|             
 | |
|             // NULL means that we are gathering directories, so skip this
 | |
|             if (fileExtensionOrDirectory != NULL)
 | |
|             {
 | |
|                 // make sure that we can have the desired extension (prevent seg fault)
 | |
|                 if (strlen(de->d_name) < extension_length)
 | |
|                     continue;
 | |
|                 // compare the extension
 | |
|                 if (strcmp(de->d_name + strlen(de->d_name) - extension_length, fileExtensionOrDirectory) != 0)
 | |
|                     continue;
 | |
|             }
 | |
|             else
 | |
|             {
 | |
|                 struct stat info;
 | |
|                 char fullFileName[PATH_MAX];
 | |
|                 strcpy(fullFileName, directory);
 | |
|                 strcat(fullFileName, de->d_name);
 | |
|                 stat(fullFileName, &info);
 | |
|                 // make sure it is a directory
 | |
|                 if (!(S_ISDIR(info.st_mode)))
 | |
|                     continue;
 | |
|             }
 | |
|             
 | |
|             if (pass == 0)
 | |
|             {
 | |
|                 total++;
 | |
|                 continue;
 | |
|             }
 | |
|             
 | |
|             files[i] = (char*) malloc(dirLen + strlen(de->d_name) + 2);
 | |
|             strcpy(files[i], directory);
 | |
|             strcat(files[i], de->d_name);
 | |
|             if (fileExtensionOrDirectory == NULL)
 | |
|                 strcat(files[i], "/");
 | |
|             i++;
 | |
|         }
 | |
|         if (pass == 1)
 | |
|             break;
 | |
|         if (total == 0)
 | |
|             break;
 | |
|         rewinddir(dir);
 | |
|         *numFiles = total;
 | |
|         files = (char**) malloc((total+1)*sizeof(char*));
 | |
|         files[total]=NULL;
 | |
|     }
 | |
| 
 | |
|     if(closedir(dir) < 0) {
 | |
|         LOGE("Failed to close directory.");
 | |
|     }
 | |
| 
 | |
|     if (total==0) {
 | |
|         return NULL;
 | |
|     }
 | |
| 
 | |
|     return files;
 | |
| }
 | |
| 
 | |
| // pass in NULL for fileExtensionOrDirectory and you will get a directory chooser
 | |
| char* choose_file_menu(const char* directory, const char* fileExtensionOrDirectory, const char* headers[])
 | |
| {
 | |
|     char path[PATH_MAX] = "";
 | |
|     DIR *dir;
 | |
|     struct dirent *de;
 | |
|     int numFiles = 0;
 | |
|     int numDirs = 0;
 | |
|     int i;
 | |
|     char* return_value = NULL;
 | |
|     int dir_len = strlen(directory);
 | |
| 
 | |
|     char** files = gather_files(directory, fileExtensionOrDirectory, &numFiles);
 | |
|     char** dirs = NULL;
 | |
|     if (fileExtensionOrDirectory != NULL)
 | |
|         dirs = gather_files(directory, NULL, &numDirs);
 | |
|     int total = numDirs + numFiles;
 | |
|     if (total == 0)
 | |
|     {
 | |
|         ui_print("No files found.\n");
 | |
|     }
 | |
|     else
 | |
|     {
 | |
|         char** list = (char**) malloc((total + 1) * sizeof(char*));
 | |
|         list[total] = NULL;
 | |
| 
 | |
| 
 | |
|         for (i = 0 ; i < numDirs; i++)
 | |
|         {
 | |
|             list[i] = strdup(dirs[i] + dir_len);
 | |
|         }
 | |
| 
 | |
|         for (i = 0 ; i < numFiles; i++)
 | |
|         {
 | |
|             list[numDirs + i] = strdup(files[i] + dir_len);
 | |
|         }
 | |
| 
 | |
|         for (;;)
 | |
|         {
 | |
|             int chosen_item = get_menu_selection(headers, list, 0);
 | |
|             if (chosen_item == GO_BACK)
 | |
|                 break;
 | |
|             static char ret[PATH_MAX];
 | |
|             if (chosen_item < numDirs)
 | |
|             {
 | |
|                 char* subret = choose_file_menu(dirs[chosen_item], fileExtensionOrDirectory, headers);
 | |
|                 if (subret != NULL)
 | |
|                 {
 | |
|                     strcpy(ret, subret);
 | |
|                     return_value = ret;
 | |
|                     break;
 | |
|                 }
 | |
|                 continue;
 | |
|             } 
 | |
|             strcpy(ret, files[chosen_item - numDirs]);
 | |
|             return_value = ret;
 | |
|             break;
 | |
|         }
 | |
|         free_string_array(list);
 | |
|     }
 | |
| 
 | |
|     free_string_array(files);
 | |
|     free_string_array(dirs);
 | |
|     return return_value;
 | |
| }
 | |
| 
 | |
| void show_choose_zip_menu()
 | |
| {
 | |
|     if (ensure_root_path_mounted("SDCARD:") != 0) {
 | |
|         LOGE ("Can't mount /sdcard\n");
 | |
|         return;
 | |
|     }
 | |
| 
 | |
|     static char* headers[] = {  "Choose a zip to apply",
 | |
|                                 "",
 | |
|                                 NULL 
 | |
|     };
 | |
|     
 | |
|     char* file = choose_file_menu("/sdcard/", ".zip", headers);
 | |
|     if (file == NULL)
 | |
|         return;
 | |
|     char sdcard_package_file[1024];
 | |
|     strcpy(sdcard_package_file, "SDCARD:");
 | |
|     strcat(sdcard_package_file,  file + strlen("/sdcard/"));
 | |
|     install_zip(sdcard_package_file);
 | |
| }
 | |
| 
 | |
| // This was pulled from bionic: The default system command always looks
 | |
| // for shell in /system/bin/sh. This is bad.
 | |
| #define _PATH_BSHELL "/sbin/sh"
 | |
| 
 | |
| extern char **environ;
 | |
| int
 | |
| __system(const char *command)
 | |
| {
 | |
|   pid_t pid;
 | |
|     sig_t intsave, quitsave;
 | |
|     sigset_t mask, omask;
 | |
|     int pstat;
 | |
|     char *argp[] = {"sh", "-c", NULL, NULL};
 | |
| 
 | |
|     if (!command)        /* just checking... */
 | |
|         return(1);
 | |
| 
 | |
|     argp[2] = (char *)command;
 | |
| 
 | |
|     sigemptyset(&mask);
 | |
|     sigaddset(&mask, SIGCHLD);
 | |
|     sigprocmask(SIG_BLOCK, &mask, &omask);
 | |
|     switch (pid = vfork()) {
 | |
|     case -1:            /* error */
 | |
|         sigprocmask(SIG_SETMASK, &omask, NULL);
 | |
|         return(-1);
 | |
|     case 0:                /* child */
 | |
|         sigprocmask(SIG_SETMASK, &omask, NULL);
 | |
|         execve(_PATH_BSHELL, argp, environ);
 | |
|     _exit(127);
 | |
|   }
 | |
| 
 | |
|     intsave = (sig_t)  bsd_signal(SIGINT, SIG_IGN);
 | |
|     quitsave = (sig_t) bsd_signal(SIGQUIT, SIG_IGN);
 | |
|     pid = waitpid(pid, (int *)&pstat, 0);
 | |
|     sigprocmask(SIG_SETMASK, &omask, NULL);
 | |
|     (void)bsd_signal(SIGINT, intsave);
 | |
|     (void)bsd_signal(SIGQUIT, quitsave);
 | |
|     return (pid == -1 ? -1 : pstat);
 | |
| }
 | |
| 
 | |
| void show_nandroid_restore_menu()
 | |
| {
 | |
|     if (ensure_root_path_mounted("SDCARD:") != 0) {
 | |
|         LOGE ("Can't mount /sdcard\n");
 | |
|         return;
 | |
|     }
 | |
|     
 | |
|     static char* headers[] = {  "Choose an image to restore",
 | |
|                                 "",
 | |
|                                 NULL 
 | |
|     };
 | |
| 
 | |
|     char* file = choose_file_menu("/sdcard/clockworkmod/backup/", NULL, headers);
 | |
|     if (file == NULL)
 | |
|         return;
 | |
|     nandroid_restore(file, 1, 1, 1, 1, 1);
 | |
| }
 | |
| 
 | |
| void show_mount_usb_storage_menu()
 | |
| {
 | |
|     __system("echo /dev/block/mmcblk0 > /sys/devices/platform/usb_mass_storage/lun0/file");
 | |
|     static char* headers[] = {  "USB Mass Storage device",
 | |
|                                 "Leaving this menu unmount",
 | |
|                                 "your SD card from your PC.",
 | |
|                                 "",
 | |
|                                 NULL 
 | |
|     };
 | |
|     
 | |
|     static char* list[] = { "Unmount", NULL };
 | |
|     
 | |
|     for (;;)
 | |
|     {
 | |
|         int chosen_item = get_menu_selection(headers, list, 0);
 | |
|         if (chosen_item == GO_BACK || chosen_item == 0)
 | |
|             break;
 | |
|     }
 | |
|     
 | |
|     __system("echo '' > /sys/devices/platform/usb_mass_storage/lun0/file");
 | |
|     __system("echo 0 > /sys/devices/platform/usb_mass_storage/lun0/enable");
 | |
| }
 | |
| 
 | |
| int confirm_format()
 | |
| {
 | |
|     static char* title_headers[] = { "Confirm format?",
 | |
|                             "  THIS CAN NOT BE UNDONE.",
 | |
|                             "",
 | |
|                             NULL,
 | |
|                         };
 | |
| 
 | |
|     char* items[] = { " No",
 | |
|                       " No",
 | |
|                       " No",
 | |
|                       " No",
 | |
|                       " No",
 | |
|                       " No",
 | |
|                       " No",
 | |
|                       " Yes -- wipe partition",   // [7]
 | |
|                       " No",
 | |
|                       " No",
 | |
|                       " No",
 | |
|                       NULL };
 | |
| 
 | |
|     int chosen_item = get_menu_selection(title_headers, items, 0);
 | |
|     return chosen_item == 7;
 | |
| }
 | |
| 
 | |
| int format_mmc_device(char* root)
 | |
| {
 | |
|     char path[PATH_MAX];
 | |
|     translate_root_path(root, path, PATH_MAX);
 | |
|     if (0 != ensure_root_path_mounted(root))
 | |
|     {
 | |
|         ui_print("Error mounting %s!\n", path);
 | |
|         return 1;
 | |
|     }
 | |
| 
 | |
|     static char tmp[PATH_MAX];
 | |
|     sprintf(tmp, "rm -rf %s/*", path);
 | |
|     __system(tmp);
 | |
|     sprintf(tmp, "rm -rf %s/.*", path);
 | |
|     __system(tmp);
 | |
|     
 | |
|     ensure_root_path_unmounted(root);
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| #define MOUNTABLE_COUNT 5
 | |
| #define MTD_COUNT 4
 | |
| #define MMC_COUNT 2
 | |
| 
 | |
| void show_partition_menu()
 | |
| {
 | |
|     static char* headers[] = {  "Mount and unmount partitions",
 | |
|                                 "",
 | |
|                                 NULL 
 | |
|     };
 | |
| 
 | |
|     typedef char* string;
 | |
|     string mounts[MOUNTABLE_COUNT][3] = { 
 | |
|         { "mount /system", "unmount /system", "SYSTEM:" },
 | |
|         { "mount /data", "unmount /data", "DATA:" },
 | |
|         { "mount /cache", "unmount /cache", "CACHE:" },
 | |
|         { "mount /sdcard", "unmount /sdcard", "SDCARD:" },
 | |
|         { "mount /sd-ext", "unmount /sd-ext", "SDEXT:" }
 | |
|         };
 | |
|         
 | |
|     string mtds[MTD_COUNT][2] = {
 | |
|         { "format boot", "BOOT:" },
 | |
|         { "format system", "SYSTEM:" },
 | |
|         { "format data", "DATA:" },
 | |
|         { "format cache", "CACHE:" },
 | |
|     };
 | |
|     
 | |
|     string mmcs[MMC_COUNT][3] = {
 | |
|       { "format sdcard", "SDCARD:" },
 | |
|       { "format sd-ext", "SDEXT:" }  
 | |
|     };
 | |
|         
 | |
|     for (;;)
 | |
|     {
 | |
|         int ismounted[MOUNTABLE_COUNT];
 | |
|         int i;
 | |
|         static string options[MOUNTABLE_COUNT + MTD_COUNT + MMC_COUNT + 1 + 1]; // mountables, format mtds, format mmcs, usb storage, null
 | |
|         for (i = 0; i < MOUNTABLE_COUNT; i++)
 | |
|         {
 | |
|             ismounted[i] = is_root_path_mounted(mounts[i][2]);
 | |
|             options[i] = ismounted[i] ? mounts[i][1] : mounts[i][0];
 | |
|         }
 | |
|         
 | |
|         for (i = 0; i < MTD_COUNT; i++)
 | |
|         {
 | |
|             options[MOUNTABLE_COUNT + i] = mtds[i][0];
 | |
|         }
 | |
|             
 | |
|         for (i = 0; i < MMC_COUNT; i++)
 | |
|         {
 | |
|             options[MOUNTABLE_COUNT + MTD_COUNT + i] = mmcs[i][0];
 | |
|         }
 | |
|     
 | |
|         options[MOUNTABLE_COUNT + MTD_COUNT + MMC_COUNT] = "mount USB storage";
 | |
|         options[MOUNTABLE_COUNT + MTD_COUNT + MMC_COUNT + 1] = NULL;
 | |
|         
 | |
|         int chosen_item = get_menu_selection(headers, options, 0);
 | |
|         if (chosen_item == GO_BACK)
 | |
|             break;
 | |
|         if (chosen_item == MOUNTABLE_COUNT + MTD_COUNT + MMC_COUNT)
 | |
|         {
 | |
|             show_mount_usb_storage_menu();
 | |
|         }
 | |
|         else if (chosen_item < MOUNTABLE_COUNT)
 | |
|         {
 | |
|             if (ismounted[chosen_item])
 | |
|             {
 | |
|                 if (0 != ensure_root_path_unmounted(mounts[chosen_item][2]))
 | |
|                     ui_print("Error unmounting %s!\n", mounts[chosen_item][2]);
 | |
|             }
 | |
|             else
 | |
|             {
 | |
|                 if (0 != ensure_root_path_mounted(mounts[chosen_item][2]))
 | |
|                     ui_print("Error mounting %s!\n", mounts[chosen_item][2]);
 | |
|             }
 | |
|         }
 | |
|         else if (chosen_item < MOUNTABLE_COUNT + MTD_COUNT)
 | |
|         {
 | |
|             chosen_item = chosen_item - MOUNTABLE_COUNT;
 | |
|             if (!confirm_format())
 | |
|                 continue;
 | |
|             ui_print("Formatting %s...\n", mtds[chosen_item][1]);
 | |
|             if (0 != format_root_device(mtds[chosen_item][1]))
 | |
|                 ui_print("Error formatting %s!\n", mtds[chosen_item][1]);
 | |
|             else
 | |
|                 ui_print("Done.\n");
 | |
|         }
 | |
|         else if (chosen_item < MOUNTABLE_COUNT + MTD_COUNT + MMC_COUNT)
 | |
|         {
 | |
|             chosen_item = chosen_item - MOUNTABLE_COUNT - MTD_COUNT;
 | |
|             if (!confirm_format())
 | |
|                 continue;
 | |
|             ui_print("Formatting %s...\n", mmcs[chosen_item][1]);
 | |
|             if (0 != format_mmc_device(mmcs[chosen_item][1]))
 | |
|                 ui_print("Error formatting %s!\n", mmcs[chosen_item][1]);
 | |
|             else
 | |
|                 ui_print("Done.\n");
 | |
|         }
 | |
|     }
 | |
| }
 | |
| 
 | |
| #define EXTENDEDCOMMAND_SCRIPT "/cache/recovery/extendedcommand"
 | |
| 
 | |
| int extendedcommand_file_exists()
 | |
| {
 | |
|     struct stat file_info;
 | |
|     return 0 == stat(EXTENDEDCOMMAND_SCRIPT, &file_info);
 | |
| }
 | |
| 
 | |
| int run_script_from_buffer(char* script_data, int script_len, char* filename)
 | |
| {
 | |
|     /* Parse the script.  Note that the script and parse tree are never freed.
 | |
|      */
 | |
|     const AmCommandList *commands = parseAmendScript(script_data, script_len);
 | |
|     if (commands == NULL) {
 | |
|         printf("Syntax error in update script\n");
 | |
|         return 1;
 | |
|     } else {
 | |
|         printf("Parsed %.*s\n", script_len, filename);
 | |
|     }
 | |
| 
 | |
|     /* Execute the script.
 | |
|      */
 | |
|     int ret = execCommandList((ExecContext *)1, commands);
 | |
|     if (ret != 0) {
 | |
|         int num = ret;
 | |
|         char *line = NULL, *next = script_data;
 | |
|         while (next != NULL && ret-- > 0) {
 | |
|             line = next;
 | |
|             next = memchr(line, '\n', script_data + script_len - line);
 | |
|             if (next != NULL) *next++ = '\0';
 | |
|         }
 | |
|         printf("Failure at line %d:\n%s\n", num, next ? line : "(not found)");
 | |
|         return 1;
 | |
|     }    
 | |
|     
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| int run_script(char* filename)
 | |
| {
 | |
|     struct stat file_info;
 | |
|     if (0 != stat(filename, &file_info)) {
 | |
|         printf("Error executing stat on file: %s\n", filename);
 | |
|         return 1;
 | |
|     }
 | |
|     
 | |
|     int script_len = file_info.st_size;
 | |
|     char* script_data = (char*)malloc(script_len + 1);
 | |
|     FILE *file = fopen(filename, "rb");
 | |
|     fread(script_data, script_len, 1, file);
 | |
|     // supposedly not necessary, but let's be safe.
 | |
|     script_data[script_len] = '\0';
 | |
|     fclose(file);
 | |
| 
 | |
| 	int ret = run_script_from_buffer(script_data, script_len, filename);
 | |
|     free(script_data);
 | |
|     return ret;
 | |
| }
 | |
| 
 | |
| int run_and_remove_extendedcommand()
 | |
| {
 | |
|     __system("touch /tmp/.installscript");
 | |
|     char tmp[PATH_MAX];
 | |
|     sprintf(tmp, "cp %s /tmp/%s", EXTENDEDCOMMAND_SCRIPT, basename(EXTENDEDCOMMAND_SCRIPT));
 | |
|     __system(tmp);
 | |
|     remove(EXTENDEDCOMMAND_SCRIPT);
 | |
|     int i = 0;
 | |
|     for (i = 20; i > 0; i--) {
 | |
|         ui_print("Waiting for SD Card to mount (%ds)\n", i);
 | |
|         if (ensure_root_path_mounted("SDCARD:") == 0) {
 | |
|             ui_print("SD Card mounted...\n");
 | |
|             break;
 | |
|         }
 | |
|         sleep(1);
 | |
|     }
 | |
|     if (i == 0) {
 | |
|         ui_print("Timed out waiting for SD card... continuing anyways.");
 | |
|     }
 | |
|     
 | |
|     sprintf(tmp, "/tmp/%s", basename(EXTENDEDCOMMAND_SCRIPT));
 | |
|     return run_script(tmp);
 | |
| }
 | |
| 
 | |
| int amend_main(int argc, char** argv)
 | |
| {
 | |
|     if (argc != 2) 
 | |
|     {
 | |
|         printf("Usage: amend <script>\n");
 | |
|         return 0;
 | |
|     }
 | |
| 
 | |
|     RecoveryCommandContext ctx = { NULL };
 | |
|     if (register_update_commands(&ctx)) {
 | |
|         LOGE("Can't install update commands\n");
 | |
|     }
 | |
|     return run_script(argv[1]);
 | |
| }
 | |
| 
 | |
| void show_nandroid_advanced_restore_menu()
 | |
| {
 | |
|     if (ensure_root_path_mounted("SDCARD:") != 0) {
 | |
|         LOGE ("Can't mount /sdcard\n");
 | |
|         return;
 | |
|     }
 | |
| 
 | |
|     static char* advancedheaders[] = {  "Choose an image to restore",
 | |
|                                 "",
 | |
|                                 "Choose an image to restore",
 | |
|                                 "first. The next menu will",
 | |
|                                 "you more options.",
 | |
|                                 "",
 | |
|                                 NULL
 | |
|     };
 | |
| 
 | |
|     char* file = choose_file_menu("/sdcard/clockworkmod/backup/", NULL, advancedheaders);
 | |
|     if (file == NULL)
 | |
|         return;
 | |
| 
 | |
|     static char* headers[] = {  "Nandroid Advanced Restore",
 | |
|                                 "",
 | |
|                                 NULL
 | |
|     };
 | |
| 
 | |
|     static char* list[] = { "Restore boot",
 | |
|                             "Restore system",
 | |
|                             "Restore data",
 | |
|                             "Restore cache",
 | |
|                             "Restore sd-ext",
 | |
|                             NULL
 | |
|     };
 | |
| 
 | |
|     int chosen_item = get_menu_selection(headers, list, 0);
 | |
|     switch (chosen_item)
 | |
|     {
 | |
|         case 0:
 | |
|             nandroid_restore(file, 1, 0, 0, 0, 0);
 | |
|             break;
 | |
|         case 1:
 | |
|             nandroid_restore(file, 0, 1, 0, 0, 0);
 | |
|             break;
 | |
|         case 2:
 | |
|             nandroid_restore(file, 0, 0, 1, 0, 0);
 | |
|             break;
 | |
|         case 3:
 | |
|             nandroid_restore(file, 0, 0, 0, 1, 0);
 | |
|             break;
 | |
|     }
 | |
| }
 | |
| 
 | |
| void show_nandroid_menu()
 | |
| {
 | |
|     static char* headers[] = {  "Nandroid",
 | |
|                                 "",
 | |
|                                 NULL 
 | |
|     };
 | |
| 
 | |
|     static char* list[] = { "Backup", 
 | |
|                             "Restore",
 | |
|                             "Advanced Restore",
 | |
|                             NULL
 | |
|     };
 | |
| 
 | |
|     int chosen_item = get_menu_selection(headers, list, 0);
 | |
|     switch (chosen_item)
 | |
|     {
 | |
|         case 0:
 | |
|             {
 | |
|                 char backup_path[PATH_MAX];
 | |
|                 time_t t = time(NULL);
 | |
|                 struct tm *tmp = localtime(&t);
 | |
|                 if (tmp == NULL)
 | |
|                 {
 | |
|                     struct timeval tp;
 | |
|                     gettimeofday(&tp, NULL);
 | |
|                     sprintf(backup_path, "/sdcard/clockworkmod/backup/%d", tp.tv_sec);
 | |
|                 }
 | |
|                 else
 | |
|                 {
 | |
|                     strftime(backup_path, sizeof(backup_path), "/sdcard/clockworkmod/backup/%F.%H.%M.%S", tmp);
 | |
|                 }
 | |
|                 nandroid_backup(backup_path);
 | |
|             }
 | |
|             break;
 | |
|         case 1:
 | |
|             show_nandroid_restore_menu();
 | |
|             break;
 | |
|         case 2:
 | |
|             show_nandroid_advanced_restore_menu();
 | |
|             break;
 | |
|     }
 | |
| }
 | |
| 
 | |
| void show_advanced_menu()
 | |
| {
 | |
|     static char* headers[] = {  "Advanced and Debugging Menu",
 | |
|                                 "",
 | |
|                                 NULL
 | |
|     };
 | |
| 
 | |
|     static char* list[] = { "Key Test",
 | |
|                             NULL
 | |
|     };
 | |
| 
 | |
|     for (;;)
 | |
|     {
 | |
|         int chosen_item = get_menu_selection(headers, list, 0);
 | |
|         if (chosen_item == GO_BACK)
 | |
|             break;
 | |
|         switch (chosen_item)
 | |
|         {
 | |
|             case 0:
 | |
|             {
 | |
|                 ui_print("Outputting key codes.\n");
 | |
|                 ui_print("Go back to end debugging.\n");
 | |
|                 int key;
 | |
|                 int action;
 | |
|                 do
 | |
|                 {
 | |
|                     key = ui_wait_key();
 | |
|                     action = device_handle_key(key, 1);
 | |
|                     ui_print("Key: %d\n", key);
 | |
|                 }
 | |
|                 while (action != GO_BACK);
 | |
|             }
 | |
|         }
 | |
|     }
 | |
| } |