From aee144c32207a27e2c6f54639101b5828da713d8 Mon Sep 17 00:00:00 2001 From: Ethan Yonker Date: Wed, 11 Feb 2015 16:49:19 -0600 Subject: Fix handling of mapped zip files for OTA updates AOSP started using the @ special character to pass to sysMapFile to indicate that the zip file is actually mapped to unused blocks on the data partition. Allow TWRP to do the same. Change-Id: I5bf1355a5462588086401b02de1b17ab88ed3bcd --- openrecoveryscript.cpp | 57 +++++++++++++++++++++++++++++--------------------- 1 file changed, 33 insertions(+), 24 deletions(-) (limited to 'openrecoveryscript.cpp') diff --git a/openrecoveryscript.cpp b/openrecoveryscript.cpp index d28f76202..9b71ee75c 100644 --- a/openrecoveryscript.cpp +++ b/openrecoveryscript.cpp @@ -437,34 +437,43 @@ int OpenRecoveryScript::Install_Command(string Zip) { std::vector Storage_List; string Full_Path; - PartitionManager.Mount_All_Storage(); - PartitionManager.Get_Partition_List("storage", &Storage_List); - int listSize = Storage_List.size(); - for (int i = 0; i < listSize; i++) { - if (PartitionManager.Is_Mounted_By_Path(Storage_List.at(i).Mount_Point)) { - Full_Path = Storage_List.at(i).Mount_Point + "/" + Zip; - if (TWFunc::Path_Exists(Full_Path)) { - Zip = Full_Path; - break; - } - Full_Path = Zip; - LOGINFO("Trying to find zip '%s' on '%s'...\n", Full_Path.c_str(), Storage_List.at(i).Mount_Point.c_str()); - ret_string = Locate_Zip_File(Full_Path, Storage_List.at(i).Mount_Point); - if (!ret_string.empty()) { - Zip = ret_string; - break; + if (Zip.substr(0, 1) == "@") { + // This is a special file that contains a map of blocks on the data partition + Full_Path = Zip.substr(1); + if (!PartitionManager.Mount_By_Path(Full_Path, true) || !TWFunc::Path_Exists(Full_Path)) { + gui_print("Unable to install via mapped zip '%s'\n", Full_Path.c_str()); + return 1; + } + gui_print("Installing mapped zip file '%s'\n", Full_Path.c_str()); + } else if (!TWFunc::Path_Exists(Zip)) { + PartitionManager.Mount_All_Storage(); + PartitionManager.Get_Partition_List("storage", &Storage_List); + int listSize = Storage_List.size(); + for (int i = 0; i < listSize; i++) { + if (PartitionManager.Is_Mounted_By_Path(Storage_List.at(i).Mount_Point)) { + Full_Path = Storage_List.at(i).Mount_Point + "/" + Zip; + if (TWFunc::Path_Exists(Full_Path)) { + Zip = Full_Path; + break; + } + Full_Path = Zip; + LOGINFO("Trying to find zip '%s' on '%s'...\n", Full_Path.c_str(), Storage_List.at(i).Mount_Point.c_str()); + ret_string = Locate_Zip_File(Full_Path, Storage_List.at(i).Mount_Point); + if (!ret_string.empty()) { + Zip = ret_string; + break; + } } } + 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()); } - 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); - } + 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; -- cgit v1.2.3