diff options
Diffstat (limited to 'openrecoveryscript.cpp')
-rw-r--r-- | openrecoveryscript.cpp | 548 |
1 files changed, 548 insertions, 0 deletions
diff --git a/openrecoveryscript.cpp b/openrecoveryscript.cpp new file mode 100644 index 000000000..bb472bf82 --- /dev/null +++ b/openrecoveryscript.cpp @@ -0,0 +1,548 @@ +/* OpenRecoveryScript class for TWRP + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License version 2 and + * only version 2 as published by the Free Software Foundation. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA + * 02110-1301, USA. + * + * The code was written from scratch by Dees_Troy dees_troy at + * yahoo + * + * Copyright (c) 2012 + */ + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <sys/stat.h> +#include <sys/vfs.h> +#include <unistd.h> +#include <vector> +#include <dirent.h> +#include <time.h> +#include <errno.h> +#include <iostream> +#include <fstream> + +#include "twrp-functions.hpp" +#include "partitions.hpp" +#include "twcommon.h" +#include "openrecoveryscript.hpp" +#include "variables.h" +#include "adb_install.h" +#include "data.hpp" +extern "C" { + #include "twinstall.h" + #include "gui/gui.h" + int TWinstall_zip(const char* path, int* wipe_cache); +} + +#define SCRIPT_COMMAND_SIZE 512 + +int OpenRecoveryScript::check_for_script_file(void) { + if (!PartitionManager.Mount_By_Path(SCRIPT_FILE_CACHE, false)) { + LOGERR("Unable to mount /cache for OpenRecoveryScript support.\n"); + return 0; + } + if (TWFunc::Path_Exists(SCRIPT_FILE_CACHE)) { + LOGINFO("Script file found: '%s'\n", SCRIPT_FILE_CACHE); + // Copy script file to /tmp + TWFunc::copy_file(SCRIPT_FILE_CACHE, SCRIPT_FILE_TMP, 0755); + // Delete the file from /cache + unlink(SCRIPT_FILE_CACHE); + return 1; + } + return 0; +} + +int OpenRecoveryScript::run_script_file(void) { + FILE *fp = fopen(SCRIPT_FILE_TMP, "r"); + int ret_val = 0, cindex, line_len, i, remove_nl, install_cmd = 0, sideload = 0; + char script_line[SCRIPT_COMMAND_SIZE], command[SCRIPT_COMMAND_SIZE], + value[SCRIPT_COMMAND_SIZE], mount[SCRIPT_COMMAND_SIZE], + value1[SCRIPT_COMMAND_SIZE], value2[SCRIPT_COMMAND_SIZE]; + char *val_start, *tok; + + if (fp != NULL) { + DataManager::SetValue(TW_SIMULATE_ACTIONS, 0); + DataManager::SetValue("ui_progress", 0); // Reset the progress bar + while (fgets(script_line, SCRIPT_COMMAND_SIZE, fp) != NULL && ret_val == 0) { + cindex = 0; + line_len = strlen(script_line); + if (line_len < 2) + continue; // there's a blank line or line is too short to contain a command + //gui_print("script line: '%s'\n", script_line); + for (i=0; i<line_len; i++) { + if ((int)script_line[i] == 32) { + cindex = i; + i = line_len; + } + } + memset(command, 0, sizeof(command)); + memset(value, 0, sizeof(value)); + if ((int)script_line[line_len - 1] == 10) + remove_nl = 2; + else + remove_nl = 1; + if (cindex != 0) { + strncpy(command, script_line, cindex); + LOGINFO("command is: '%s' and ", command); + val_start = script_line; + val_start += cindex + 1; + if ((int) *val_start == 51) + val_start++; //get rid of = at the beginning + strncpy(value, val_start, line_len - cindex - remove_nl); + LOGINFO("value is: '%s'\n", value); + } else { + strncpy(command, script_line, line_len - remove_nl + 1); + gui_print("command is: '%s' and there is no value\n", command); + } + if (strcmp(command, "install") == 0) { + // Install Zip + DataManager::SetValue("tw_action_text2", "Installing Zip"); + PartitionManager.Mount_All_Storage(); + ret_val = Install_Command(value); + install_cmd = -1; + } else if (strcmp(command, "wipe") == 0) { + // Wipe + if (strcmp(value, "cache") == 0 || strcmp(value, "/cache") == 0) { + gui_print("-- Wiping Cache Partition...\n"); + PartitionManager.Wipe_By_Path("/cache"); + gui_print("-- Cache Partition Wipe Complete!\n"); + } else if (strcmp(value, "dalvik") == 0 || strcmp(value, "dalvick") == 0 || strcmp(value, "dalvikcache") == 0 || strcmp(value, "dalvickcache") == 0) { + gui_print("-- Wiping Dalvik Cache...\n"); + PartitionManager.Wipe_Dalvik_Cache(); + gui_print("-- Dalvik Cache Wipe Complete!\n"); + } else if (strcmp(value, "data") == 0 || strcmp(value, "/data") == 0 || strcmp(value, "factory") == 0 || strcmp(value, "factoryreset") == 0) { + gui_print("-- Wiping Data Partition...\n"); + PartitionManager.Factory_Reset(); + gui_print("-- Data Partition Wipe Complete!\n"); + } else { + LOGERR("Error with wipe command value: '%s'\n", value); + ret_val = 1; + } + } else if (strcmp(command, "backup") == 0) { + // Backup + DataManager::SetValue("tw_action_text2", "Backing Up"); + tok = strtok(value, " "); + strcpy(value1, tok); + tok = strtok(NULL, " "); + if (tok != NULL) { + memset(value2, 0, sizeof(value2)); + strcpy(value2, tok); + line_len = strlen(tok); + if ((int)value2[line_len - 1] == 10 || (int)value2[line_len - 1] == 13) { + if ((int)value2[line_len - 1] == 10 || (int)value2[line_len - 1] == 13) + remove_nl = 2; + else + remove_nl = 1; + } else + remove_nl = 0; + strncpy(value2, tok, line_len - remove_nl); + DataManager::SetValue(TW_BACKUP_NAME, value2); + gui_print("Backup folder set to '%s'\n", value2); + if (PartitionManager.Check_Backup_Name(true) != 0) { + ret_val = 1; + continue; + } + } else { + char empt[50]; + strcpy(empt, "(Current Date)"); + DataManager::SetValue(TW_BACKUP_NAME, empt); + } + ret_val = Backup_Command(value1); + } else if (strcmp(command, "restore") == 0) { + // Restore + DataManager::SetValue("tw_action_text2", "Restoring"); + PartitionManager.Mount_All_Storage(); + DataManager::SetValue(TW_SKIP_MD5_CHECK_VAR, 0); + char folder_path[512], partitions[512]; + + string val = value, restore_folder, restore_partitions; + size_t pos = val.find_last_of(" "); + if (pos == string::npos) { + restore_folder = value; + partitions[0] = '\0'; + } else { + restore_folder = val.substr(0, pos); + restore_partitions = val.substr(pos + 1, val.size() - pos - 1); + strcpy(partitions, restore_partitions.c_str()); + } + strcpy(folder_path, restore_folder.c_str()); + LOGINFO("Restore folder is: '%s' and partitions: '%s'\n", folder_path, partitions); + gui_print("Restoring '%s'\n", folder_path); + + if (folder_path[0] != '/') { + char backup_folder[512]; + string folder_var; + DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, folder_var); + sprintf(backup_folder, "%s/%s", folder_var.c_str(), folder_path); + LOGINFO("Restoring relative path: '%s'\n", backup_folder); + if (!TWFunc::Path_Exists(backup_folder)) { + if (DataManager::GetIntValue(TW_HAS_DUAL_STORAGE)) { + if (DataManager::GetIntValue(TW_USE_EXTERNAL_STORAGE)) { + LOGINFO("Backup folder '%s' not found on external storage, trying internal...\n", folder_path); + DataManager::SetValue(TW_USE_EXTERNAL_STORAGE, 0); + } else { + LOGINFO("Backup folder '%s' not found on internal storage, trying external...\n", folder_path); + DataManager::SetValue(TW_USE_EXTERNAL_STORAGE, 1); + } + DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, folder_var); + sprintf(backup_folder, "%s/%s", folder_var.c_str(), folder_path); + LOGINFO("2Restoring relative path: '%s'\n", backup_folder); + } + } + strcpy(folder_path, backup_folder); + } else { + if (folder_path[strlen(folder_path) - 1] == '/') + strcat(folder_path, "."); + else + strcat(folder_path, "/."); + } + if (!TWFunc::Path_Exists(folder_path)) { + gui_print("Unable to locate backup '%s'\n", folder_path); + ret_val = 1; + continue; + } + DataManager::SetValue("tw_restore", folder_path); + + PartitionManager.Set_Restore_Files(folder_path); + string Partition_List; + int is_encrypted = 0; + DataManager::GetValue("tw_restore_encrypted", is_encrypted); + DataManager::GetValue("tw_restore_list", Partition_List); + if (strlen(partitions) != 0) { + string Restore_List; + + memset(value2, 0, sizeof(value2)); + strcpy(value2, partitions); + gui_print("Setting restore options: '%s':\n", value2); + line_len = strlen(value2); + for (i=0; i<line_len; i++) { + if ((value2[i] == 'S' || value2[i] == 's') && Partition_List.find("/system;") != string::npos) { + Restore_List += "/system;"; + gui_print("System\n"); + } else if ((value2[i] == 'D' || value2[i] == 'd') && Partition_List.find("/data;") != string::npos) { + Restore_List += "/data;"; + gui_print("Data\n"); + } else if ((value2[i] == 'C' || value2[i] == 'c') && Partition_List.find("/cache;") != string::npos) { + Restore_List += "/cache;"; + gui_print("Cache\n"); + } else if ((value2[i] == 'R' || value2[i] == 'r') && Partition_List.find("/recovery;") != string::npos) { + gui_print("Recovery -- Not allowed to restore recovery\n"); + } else if (value2[i] == '1' && DataManager::GetIntValue(TW_RESTORE_SP1_VAR) > 0) { + gui_print("%s\n", "Special1 -- No Longer Supported..."); + } else if (value2[i] == '2' && DataManager::GetIntValue(TW_RESTORE_SP2_VAR) > 0) { + gui_print("%s\n", "Special2 -- No Longer Supported..."); + } else if (value2[i] == '3' && DataManager::GetIntValue(TW_RESTORE_SP3_VAR) > 0) { + gui_print("%s\n", "Special3 -- No Longer Supported..."); + } else if ((value2[i] == 'B' || value2[i] == 'b') && Partition_List.find("/boot;") != string::npos) { + Restore_List += "/boot;"; + gui_print("Boot\n"); + } else if ((value2[i] == 'A' || value2[i] == 'a') && Partition_List.find("/and-sec;") != string::npos) { + Restore_List += "/and-sec;"; + gui_print("Android Secure\n"); + } else if ((value2[i] == 'E' || value2[i] == 'e') && Partition_List.find("/sd-ext;") != string::npos) { + Restore_List += "/sd-ext;"; + gui_print("SD-Ext\n"); + } else if (value2[i] == 'M' || value2[i] == 'm') { + DataManager::SetValue(TW_SKIP_MD5_CHECK_VAR, 1); + gui_print("MD5 check skip is on\n"); + } + } + + DataManager::SetValue("tw_restore_selected", Restore_List); + } else { + DataManager::SetValue("tw_restore_selected", Partition_List); + } + if (is_encrypted) { + LOGERR("Unable to use OpenRecoveryScript to restore an encrypted backup.\n"); + ret_val = 1; + } else if (!PartitionManager.Run_Restore(folder_path)) + ret_val = 1; + else + gui_print("Restore complete!\n"); + } else if (strcmp(command, "mount") == 0) { + // Mount + DataManager::SetValue("tw_action_text2", "Mounting"); + if (value[0] != '/') { + strcpy(mount, "/"); + strcat(mount, value); + } else + strcpy(mount, value); + if (PartitionManager.Mount_By_Path(mount, true)) + gui_print("Mounted '%s'\n", mount); + } else if (strcmp(command, "unmount") == 0 || strcmp(command, "umount") == 0) { + // Unmount + DataManager::SetValue("tw_action_text2", "Unmounting"); + if (value[0] != '/') { + strcpy(mount, "/"); + strcat(mount, value); + } else + strcpy(mount, value); + if (PartitionManager.UnMount_By_Path(mount, true)) + gui_print("Unmounted '%s'\n", mount); + } else if (strcmp(command, "set") == 0) { + // Set value + tok = strtok(value, " "); + strcpy(value1, tok); + tok = strtok(NULL, " "); + strcpy(value2, tok); + gui_print("Setting '%s' to '%s'\n", value1, value2); + DataManager::SetValue(value1, value2); + } else if (strcmp(command, "mkdir") == 0) { + // Make directory (recursive) + DataManager::SetValue("tw_action_text2", "Making Directory"); + gui_print("Making directory (recursive): '%s'\n", value); + if (TWFunc::Recursive_Mkdir(value)) { + LOGERR("Unable to create folder: '%s'\n", value); + ret_val = 1; + } + } else if (strcmp(command, "reboot") == 0) { + // Reboot + } else if (strcmp(command, "cmd") == 0) { + DataManager::SetValue("tw_action_text2", "Running Command"); + if (cindex != 0) { + string status; + TWFunc::Exec_Cmd(value, status); + } else { + LOGERR("No value given for cmd\n"); + } + } else if (strcmp(command, "print") == 0) { + gui_print("%s\n", value); + } else if (strcmp(command, "sideload") == 0) { + // ADB Sideload + DataManager::SetValue("tw_action_text2", "ADB Sideload"); + install_cmd = -1; + + int wipe_cache = 0; + string result, Sideload_File; + + if (!PartitionManager.Mount_Current_Storage(true)) { + ret_val = 1; // failure + } else { + Sideload_File = DataManager::GetCurrentStoragePath() + "/sideload.zip"; + if (TWFunc::Path_Exists(Sideload_File)) { + unlink(Sideload_File.c_str()); + } + gui_print("Starting ADB sideload feature...\n"); + DataManager::SetValue("tw_has_cancel", 1); + DataManager::SetValue("tw_cancel_action", "adbsideloadcancel"); + ret_val = apply_from_adb(Sideload_File.c_str()); + DataManager::SetValue("tw_has_cancel", 0); + if (ret_val != 0) + ret_val = 1; // failure + else if (TWinstall_zip(Sideload_File.c_str(), &wipe_cache) == 0) { + if (wipe_cache) + PartitionManager.Wipe_By_Path("/cache"); + } else { + ret_val = 1; // failure + } + sideload = 1; // Causes device to go to the home screen afterwards + gui_print("Sideload finished.\n"); + } + } else { + LOGERR("Unrecognized script command: '%s'\n", command); + ret_val = 1; + } + } + fclose(fp); + gui_print("Done processing script file\n"); + } else { + LOGERR("Error opening script file '%s'\n", SCRIPT_FILE_TMP); + return 1; + } + if (install_cmd && DataManager::GetIntValue(TW_HAS_INJECTTWRP) == 1 && DataManager::GetIntValue(TW_INJECT_AFTER_ZIP) == 1) { + string status; + gui_print("Injecting TWRP into boot image...\n"); + TWPartition* Boot = PartitionManager.Find_Partition_By_Path("/boot"); + if (Boot == NULL || Boot->Current_File_System != "emmc") + TWFunc::Exec_Cmd("injecttwrp --dump /tmp/backup_recovery_ramdisk.img /tmp/injected_boot.img --flash", status); + else { + string injectcmd = "injecttwrp --dump /tmp/backup_recovery_ramdisk.img /tmp/injected_boot.img --flash bd=" + Boot->Actual_Block_Device; + TWFunc::Exec_Cmd(injectcmd.c_str(), status); + } + gui_print("TWRP injection complete.\n"); + } + if (sideload) + ret_val = 1; // Forces booting to the home page after sideload + return ret_val; +} + +int OpenRecoveryScript::Insert_ORS_Command(string Command) { + ofstream ORSfile(SCRIPT_FILE_TMP); + if (ORSfile.is_open()) { + //if (Command.substr(Command.size() - 1, 1) != "\n") + // Command += "\n"; + LOGINFO("Inserting '%s'\n", Command.c_str()); + ORSfile << Command.c_str(); + ORSfile.close(); + return 1; + } + LOGERR("Unable to append '%s' to '%s'\n", Command.c_str(), SCRIPT_FILE_TMP); + return 0; +} + +int OpenRecoveryScript::Install_Command(string Zip) { + // Install zip + string ret_string; + int ret_val = 0, wipe_cache = 0; + + PartitionManager.Mount_All_Storage(); + if (Zip.substr(0, 1) != "/") { + // Relative path given + string Full_Path; + + Full_Path = DataManager::GetCurrentStoragePath(); + Full_Path += "/" + Zip; + LOGINFO("Full zip path: '%s'\n", Full_Path.c_str()); + if (!TWFunc::Path_Exists(Full_Path)) { + ret_string = Locate_Zip_File(Full_Path, DataManager::GetCurrentStoragePath()); + if (!ret_string.empty()) { + Full_Path = ret_string; + } else if (DataManager::GetIntValue(TW_HAS_DUAL_STORAGE)) { + if (DataManager::GetIntValue(TW_USE_EXTERNAL_STORAGE)) { + LOGINFO("Zip file not found on external storage, trying internal...\n"); + DataManager::SetValue(TW_USE_EXTERNAL_STORAGE, 0); + } else { + LOGINFO("Zip file not found on internal storage, trying external...\n"); + DataManager::SetValue(TW_USE_EXTERNAL_STORAGE, 1); + } + Full_Path = DataManager::GetCurrentStoragePath(); + Full_Path += "/" + Zip; + LOGINFO("Full zip path: '%s'\n", Full_Path.c_str()); + ret_string = Locate_Zip_File(Full_Path, DataManager::GetCurrentStoragePath()); + if (!ret_string.empty()) + Full_Path = ret_string; + } + } + Zip = Full_Path; + } else { + // Full path given + if (!TWFunc::Path_Exists(Zip)) { + ret_string = Locate_Zip_File(Zip, DataManager::GetCurrentStoragePath()); + if (!ret_string.empty()) + Zip = ret_string; + } + } + + if (!TWFunc::Path_Exists(Zip)) { + // zip file doesn't exist + gui_print("Unable to locate zip file '%s'.\n", Zip.c_str()); + ret_val = 1; + } else { + gui_print("Installing zip file '%s'\n", Zip.c_str()); + ret_val = TWinstall_zip(Zip.c_str(), &wipe_cache); + } + if (ret_val != 0) { + LOGERR("Error installing zip file '%s'\n", Zip.c_str()); + ret_val = 1; + } else if (wipe_cache) + PartitionManager.Wipe_By_Path("/cache"); + + return ret_val; +} + +string OpenRecoveryScript::Locate_Zip_File(string Zip, string Storage_Root) { + string Path = TWFunc::Get_Path(Zip); + string File = TWFunc::Get_Filename(Zip); + string pathCpy = Path; + string wholePath; + size_t pos = Path.find("/", 1); + + while (pos != string::npos) + { + pathCpy = Path.substr(pos, Path.size() - pos); + wholePath = pathCpy + "/" + File; + if (TWFunc::Path_Exists(wholePath)) + return wholePath; + wholePath = Storage_Root + "/" + wholePath; + if (TWFunc::Path_Exists(wholePath)) + return wholePath; + + pos = Path.find("/", pos + 1); + } + return ""; +} + +int OpenRecoveryScript::Backup_Command(string Options) { + char value1[SCRIPT_COMMAND_SIZE]; + int line_len, i; + string Backup_List; + + strcpy(value1, Options.c_str()); + + DataManager::SetValue(TW_USE_COMPRESSION_VAR, 0); + DataManager::SetValue(TW_SKIP_MD5_GENERATE_VAR, 0); + + gui_print("Setting backup options:\n"); + line_len = Options.size(); + for (i=0; i<line_len; i++) { + if (Options.substr(i, 1) == "S" || Options.substr(i, 1) == "s") { + Backup_List += "/system;"; + gui_print("System\n"); + } else if (Options.substr(i, 1) == "D" || Options.substr(i, 1) == "d") { + Backup_List += "/data;"; + gui_print("Data\n"); + } else if (Options.substr(i, 1) == "C" || Options.substr(i, 1) == "c") { + Backup_List += "/cache;"; + gui_print("Cache\n"); + } else if (Options.substr(i, 1) == "R" || Options.substr(i, 1) == "r") { + Backup_List += "/recovery;"; + gui_print("Recovery\n"); + } else if (Options.substr(i, 1) == "1") { + gui_print("%s\n", "Special1 -- No Longer Supported..."); + } else if (Options.substr(i, 1) == "2") { + gui_print("%s\n", "Special2 -- No Longer Supported..."); + } else if (Options.substr(i, 1) == "3") { + gui_print("%s\n", "Special3 -- No Longer Supported..."); + } else if (Options.substr(i, 1) == "B" || Options.substr(i, 1) == "b") { + Backup_List += "/boot;"; + gui_print("Boot\n"); + } else if (Options.substr(i, 1) == "A" || Options.substr(i, 1) == "a") { + Backup_List += "/and-sec;"; + gui_print("Android Secure\n"); + } else if (Options.substr(i, 1) == "E" || Options.substr(i, 1) == "e") { + Backup_List += "/sd-ext;"; + gui_print("SD-Ext\n"); + } else if (Options.substr(i, 1) == "O" || Options.substr(i, 1) == "o") { + DataManager::SetValue(TW_USE_COMPRESSION_VAR, 1); + gui_print("Compression is on\n"); + } else if (Options.substr(i, 1) == "M" || Options.substr(i, 1) == "m") { + DataManager::SetValue(TW_SKIP_MD5_GENERATE_VAR, 1); + gui_print("MD5 Generation is off\n"); + } + } + DataManager::SetValue("tw_backup_list", Backup_List); + if (!PartitionManager.Run_Backup()) { + LOGERR("Backup failed!\n"); + return 1; + } + gui_print("Backup complete!\n"); + return 0; +} + +void OpenRecoveryScript::Run_OpenRecoveryScript(void) { + DataManager::SetValue("tw_back", "main"); + DataManager::SetValue("tw_action", "openrecoveryscript"); + DataManager::SetValue("tw_has_action2", "0"); + DataManager::SetValue("tw_action2", ""); + DataManager::SetValue("tw_action2_param", ""); + DataManager::SetValue("tw_action_text1", "Running OpenRecoveryScript"); + DataManager::SetValue("tw_action_text2", ""); + DataManager::SetValue("tw_complete_text1", "OpenRecoveryScript Complete"); + DataManager::SetValue("tw_has_cancel", 0); + DataManager::SetValue("tw_show_reboot", 0); + if (gui_startPage("action_page") != 0) { + LOGERR("Failed to load OpenRecoveryScript GUI page.\n"); + } +} |