blob: 738f6ce3cc1970b1c2a8cf19bca00bd7dddd2d83 [file] [log] [blame]
Dees Troy3be70a82013-10-22 14:25:12 +00001/*
Ethan Yonker472f5062016-02-25 13:47:30 -06002 Copyright 2014 to 2016 TeamWin
Dees Troy3be70a82013-10-22 14:25:12 +00003 This file is part of TWRP/TeamWin Recovery Project.
4
5 TWRP is free software: you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation, either version 3 of the License, or
8 (at your option) any later version.
9
10 TWRP is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
14
15 You should have received a copy of the GNU General Public License
16 along with TWRP. If not, see <http://www.gnu.org/licenses/>.
17*/
Dees_Troy51a0e822012-09-05 15:24:24 -040018
19#include <stdio.h>
20#include <stdlib.h>
21#include <string.h>
22#include <sys/stat.h>
23#include <sys/vfs.h>
24#include <unistd.h>
Dees_Troy5bf43922012-09-07 16:07:55 -040025#include <vector>
Dees_Troy63c8df72012-09-10 14:02:05 -040026#include <dirent.h>
27#include <time.h>
Dees_Troy8170a922012-09-18 15:40:25 -040028#include <errno.h>
29#include <fcntl.h>
bigbiffce8f83c2015-12-12 18:30:21 -050030#include <zlib.h>
bigbiff bigbiff9c754052013-01-09 09:09:08 -050031#include <iostream>
32#include <iomanip>
Ethan Yonker8613dc02014-09-11 09:28:20 -050033#include <sys/wait.h>
Ethan Yonker483e9f42016-01-11 22:21:18 -060034#include <linux/fs.h>
35#include <sys/mount.h>
Dees_Troy51a0e822012-09-05 15:24:24 -040036#include "variables.h"
Dees_Troy2673cec2013-04-02 20:22:16 +000037#include "twcommon.h"
Dees_Troy51a0e822012-09-05 15:24:24 -040038#include "partitions.hpp"
Dees_Troy5bf43922012-09-07 16:07:55 -040039#include "data.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040040#include "twrp-functions.hpp"
Ethan Yonkerb5fab762016-01-28 14:03:33 -060041#include "fixContexts.hpp"
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -050042#include "twrpDigest.hpp"
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050043#include "twrpDU.hpp"
Ethan Yonker4b94cfd2014-12-11 10:00:45 -060044#include "set_metadata.h"
bigbiff7abc5fe2015-01-17 16:53:12 -050045#include "tw_atomic.hpp"
Ethan Yonker74db1572015-10-28 12:44:49 -050046#include "gui/gui.hpp"
Ethan Yonker472f5062016-02-25 13:47:30 -060047#include "progresstracking.hpp"
bigbiffce8f83c2015-12-12 18:30:21 -050048#include "adbbu/libtwadbbu.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040049
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040050#ifdef TW_HAS_MTP
51#include "mtp/mtp_MtpServer.hpp"
52#include "mtp/twrpMtp.hpp"
Ethan Yonker726a0202014-12-16 20:01:38 -060053#include "mtp/MtpMessage.hpp"
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040054#endif
55
Dees Troy6f6441d2014-01-23 02:07:03 +000056extern "C" {
57 #include "cutils/properties.h"
Ethan Yonker74db1572015-10-28 12:44:49 -050058 #include "gui/gui.h"
Dees Troy6f6441d2014-01-23 02:07:03 +000059}
60
Dees_Troy5bf43922012-09-07 16:07:55 -040061#ifdef TW_INCLUDE_CRYPTO
Ethan Yonker253368a2014-11-25 15:00:52 -060062 #include "crypto/lollipop/cryptfs.h"
Ethan Yonker66a19492015-12-10 10:19:45 -060063 #include "gui/rapidxml.hpp"
64 #include "gui/pages.hpp"
Dees_Troy5bf43922012-09-07 16:07:55 -040065#endif
Dees_Troy51a0e822012-09-05 15:24:24 -040066
Ethan Yonker6277c792014-09-15 14:54:30 -050067extern bool datamedia;
68
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050069TWPartitionManager::TWPartitionManager(void) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040070 mtp_was_enabled = false;
Ethan Yonker726a0202014-12-16 20:01:38 -060071 mtp_write_fd = -1;
bigbiff7abc5fe2015-01-17 16:53:12 -050072 stop_backup.set_value(0);
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050073}
74
Dees_Troy51a0e822012-09-05 15:24:24 -040075int TWPartitionManager::Process_Fstab(string Fstab_Filename, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -040076 FILE *fstabFile;
77 char fstab_line[MAX_FSTAB_LINE_LENGTH];
Dees Troy02a64532014-03-19 15:23:32 +000078 TWPartition* settings_partition = NULL;
Matt Mowerbf4efa32014-04-14 23:25:26 -050079 TWPartition* andsec_partition = NULL;
Ethan Yonker726a0202014-12-16 20:01:38 -060080 unsigned int storageid = 1 << 16; // upper 16 bits are for physical storage device, we pretend to have only one
Dees_Troy5bf43922012-09-07 16:07:55 -040081
82 fstabFile = fopen(Fstab_Filename.c_str(), "rt");
83 if (fstabFile == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +000084 LOGERR("Critical Error: Unable to open fstab at '%s'.\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -040085 return false;
86 }
87
88 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
89 if (fstab_line[0] != '/')
90 continue;
91
Dees_Troy2a923582012-09-20 12:13:34 -040092 if (fstab_line[strlen(fstab_line) - 1] != '\n')
93 fstab_line[strlen(fstab_line)] = '\n';
Dees_Troy2a923582012-09-20 12:13:34 -040094
Matt Mower2b2dd152016-04-26 11:24:08 -050095 TWPartition* partition = new TWPartition();
Matt Mower72c87ce2016-04-26 14:34:56 -050096 if (partition->Process_Fstab_Line(fstab_line, Display_Error))
Ethan Yonkerc05c5982014-03-13 09:19:56 -050097 Partitions.push_back(partition);
Matt Mower72c87ce2016-04-26 14:34:56 -050098 else
Dees_Troy5bf43922012-09-07 16:07:55 -040099 delete partition;
Matt Mower2b2dd152016-04-26 11:24:08 -0500100
101 memset(fstab_line, 0, sizeof(fstab_line));
Dees_Troy5bf43922012-09-07 16:07:55 -0400102 }
103 fclose(fstabFile);
Matt Mower72c87ce2016-04-26 14:34:56 -0500104
105 std::vector<TWPartition*>::iterator iter;
106 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
107 (*iter)->Partition_Post_Processing(Display_Error);
108
109 if ((*iter)->Is_Storage) {
110 ++storageid;
111 (*iter)->MTP_Storage_ID = storageid;
112 }
113
114 if (!settings_partition && (*iter)->Is_Settings_Storage && (*iter)->Is_Present)
115 settings_partition = (*iter);
116 else
117 (*iter)->Is_Settings_Storage = false;
118
119 if (!andsec_partition && (*iter)->Has_Android_Secure && (*iter)->Is_Present)
120 andsec_partition = (*iter);
121 else
122 (*iter)->Has_Android_Secure = false;
123 }
124
Ethan Yonker6277c792014-09-15 14:54:30 -0500125 if (!datamedia && !settings_partition && Find_Partition_By_Path("/sdcard") == NULL && Find_Partition_By_Path("/internal_sd") == NULL && Find_Partition_By_Path("/internal_sdcard") == NULL && Find_Partition_By_Path("/emmc") == NULL) {
126 // Attempt to automatically identify /data/media emulated storage devices
127 TWPartition* Dat = Find_Partition_By_Path("/data");
128 if (Dat) {
129 LOGINFO("Using automatic handling for /data/media emulated storage device.\n");
130 datamedia = true;
that9e0593e2014-10-08 00:01:24 +0200131 Dat->Setup_Data_Media();
Ethan Yonker6277c792014-09-15 14:54:30 -0500132 settings_partition = Dat;
Ethan Yonker726a0202014-12-16 20:01:38 -0600133 // Since /data was not considered a storage partition earlier, we still need to assign an MTP ID
134 ++storageid;
135 Dat->MTP_Storage_ID = storageid;
Ethan Yonker6277c792014-09-15 14:54:30 -0500136 }
137 }
Dees Troy02a64532014-03-19 15:23:32 +0000138 if (!settings_partition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500139 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
140 if ((*iter)->Is_Storage) {
Dees Troy02a64532014-03-19 15:23:32 +0000141 settings_partition = (*iter);
Dees_Troya13d74f2013-03-24 08:54:55 -0500142 break;
143 }
144 }
Dees Troy02a64532014-03-19 15:23:32 +0000145 if (!settings_partition)
Dees_Troy2673cec2013-04-02 20:22:16 +0000146 LOGERR("Unable to locate storage partition for storing settings file.\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500147 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400148 if (!Write_Fstab()) {
149 if (Display_Error)
Dees_Troy2673cec2013-04-02 20:22:16 +0000150 LOGERR("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400151 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000152 LOGINFO("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400153 }
Matt Mowered71fa32014-04-16 13:21:47 -0500154
Matt Mowerbf4efa32014-04-14 23:25:26 -0500155 if (andsec_partition) {
156 Setup_Android_Secure_Location(andsec_partition);
Matt Mowered71fa32014-04-16 13:21:47 -0500157 } else if (settings_partition) {
Matt Mowerbf4efa32014-04-14 23:25:26 -0500158 Setup_Android_Secure_Location(settings_partition);
159 }
Matt Mowered71fa32014-04-16 13:21:47 -0500160 if (settings_partition) {
161 Setup_Settings_Storage_Partition(settings_partition);
162 }
Ethan Yonker253368a2014-11-25 15:00:52 -0600163#ifdef TW_INCLUDE_CRYPTO
Ethan Yonkercceebb82014-11-18 10:17:59 -0600164 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
165 if (Decrypt_Data && Decrypt_Data->Is_Encrypted && !Decrypt_Data->Is_Decrypted) {
166 int password_type = cryptfs_get_password_type();
167 if (password_type == CRYPT_TYPE_DEFAULT) {
168 LOGINFO("Device is encrypted with the default password, attempting to decrypt.\n");
169 if (Decrypt_Device("default_password") == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500170 gui_msg("decrypt_success=Successfully decrypted with default password.");
Ethan Yonkercceebb82014-11-18 10:17:59 -0600171 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
172 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500173 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
Ethan Yonkercceebb82014-11-18 10:17:59 -0600174 }
175 } else {
176 DataManager::SetValue("TW_CRYPTO_TYPE", password_type);
177 }
178 }
Ethan Yonker66a19492015-12-10 10:19:45 -0600179 if (Decrypt_Data && (!Decrypt_Data->Is_Encrypted || Decrypt_Data->Is_Decrypted) && Decrypt_Data->Mount(false)) {
180 Decrypt_Adopted();
181 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600182#endif
Dees_Troy51127312012-09-08 13:08:49 -0400183 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -0400184 UnMount_Main_Partitions();
Dees_Troy5bf43922012-09-07 16:07:55 -0400185 return true;
186}
187
188int TWPartitionManager::Write_Fstab(void) {
189 FILE *fp;
190 std::vector<TWPartition*>::iterator iter;
191 string Line;
192
193 fp = fopen("/etc/fstab", "w");
194 if (fp == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000195 LOGINFO("Can not open /etc/fstab.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400196 return false;
197 }
Dees_Troy63c8df72012-09-10 14:02:05 -0400198 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy51127312012-09-08 13:08:49 -0400199 if ((*iter)->Can_Be_Mounted) {
dianlujitao7d304c72016-01-02 12:15:50 +0800200 Line = (*iter)->Actual_Block_Device + " " + (*iter)->Mount_Point + " " + (*iter)->Current_File_System + " rw 0 0\n";
Dees_Troy5bf43922012-09-07 16:07:55 -0400201 fputs(Line.c_str(), fp);
Dees_Troy91862e62013-04-04 23:48:21 +0000202 }
203 // Handle subpartition tracking
204 if ((*iter)->Is_SubPartition) {
205 TWPartition* ParentPartition = Find_Partition_By_Path((*iter)->SubPartition_Of);
206 if (ParentPartition)
207 ParentPartition->Has_SubPartition = true;
208 else
209 LOGERR("Unable to locate parent partition '%s' of '%s'\n", (*iter)->SubPartition_Of.c_str(), (*iter)->Mount_Point.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400210 }
211 }
212 fclose(fp);
213 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -0400214}
215
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500216void TWPartitionManager::Setup_Settings_Storage_Partition(TWPartition* Part) {
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500217 DataManager::SetValue("tw_settings_path", Part->Storage_Path);
218 DataManager::SetValue("tw_storage_path", Part->Storage_Path);
219 LOGINFO("Settings storage is '%s'\n", Part->Storage_Path.c_str());
220}
221
Matt Mowerbf4efa32014-04-14 23:25:26 -0500222void TWPartitionManager::Setup_Android_Secure_Location(TWPartition* Part) {
223 if (Part->Has_Android_Secure)
224 Part->Setup_AndSec();
Ethan Yonker6277c792014-09-15 14:54:30 -0500225 else if (!datamedia)
Matt Mowerbf4efa32014-04-14 23:25:26 -0500226 Part->Setup_AndSec();
Matt Mowerbf4efa32014-04-14 23:25:26 -0500227}
228
Dees_Troy8170a922012-09-18 15:40:25 -0400229void TWPartitionManager::Output_Partition_Logging(void) {
230 std::vector<TWPartition*>::iterator iter;
231
232 printf("\n\nPartition Logs:\n");
233 for (iter = Partitions.begin(); iter != Partitions.end(); iter++)
234 Output_Partition((*iter));
235}
236
237void TWPartitionManager::Output_Partition(TWPartition* Part) {
238 unsigned long long mb = 1048576;
239
Gary Peck004d48b2012-11-21 16:28:18 -0800240 printf("%s | %s | Size: %iMB", Part->Mount_Point.c_str(), Part->Actual_Block_Device.c_str(), (int)(Part->Size / mb));
Dees_Troy8170a922012-09-18 15:40:25 -0400241 if (Part->Can_Be_Mounted) {
Gary Peck004d48b2012-11-21 16:28:18 -0800242 printf(" Used: %iMB Free: %iMB Backup Size: %iMB", (int)(Part->Used / mb), (int)(Part->Free / mb), (int)(Part->Backup_Size / mb));
Dees_Troy8170a922012-09-18 15:40:25 -0400243 }
Gary Peck004d48b2012-11-21 16:28:18 -0800244 printf("\n Flags: ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500245 if (Part->Can_Be_Mounted)
246 printf("Can_Be_Mounted ");
Gary Peck004d48b2012-11-21 16:28:18 -0800247 if (Part->Can_Be_Wiped)
248 printf("Can_Be_Wiped ");
Hashcodedabfd492013-08-29 22:45:30 -0700249 if (Part->Use_Rm_Rf)
250 printf("Use_Rm_Rf ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500251 if (Part->Can_Be_Backed_Up)
252 printf("Can_Be_Backed_Up ");
Gary Peck004d48b2012-11-21 16:28:18 -0800253 if (Part->Wipe_During_Factory_Reset)
254 printf("Wipe_During_Factory_Reset ");
255 if (Part->Wipe_Available_in_GUI)
256 printf("Wipe_Available_in_GUI ");
257 if (Part->Is_SubPartition)
258 printf("Is_SubPartition ");
259 if (Part->Has_SubPartition)
260 printf("Has_SubPartition ");
261 if (Part->Removable)
262 printf("Removable ");
263 if (Part->Is_Present)
264 printf("IsPresent ");
265 if (Part->Can_Be_Encrypted)
266 printf("Can_Be_Encrypted ");
267 if (Part->Is_Encrypted)
268 printf("Is_Encrypted ");
269 if (Part->Is_Decrypted)
270 printf("Is_Decrypted ");
271 if (Part->Has_Data_Media)
272 printf("Has_Data_Media ");
Dees_Troy83bd4832013-05-04 12:39:56 +0000273 if (Part->Can_Encrypt_Backup)
274 printf("Can_Encrypt_Backup ");
275 if (Part->Use_Userdata_Encryption)
276 printf("Use_Userdata_Encryption ");
Gary Peck004d48b2012-11-21 16:28:18 -0800277 if (Part->Has_Android_Secure)
278 printf("Has_Android_Secure ");
279 if (Part->Is_Storage)
280 printf("Is_Storage ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500281 if (Part->Is_Settings_Storage)
282 printf("Is_Settings_Storage ");
Dees_Troy68cab492012-12-12 19:29:35 +0000283 if (Part->Ignore_Blkid)
284 printf("Ignore_Blkid ");
Dees_Troy16c2b312013-01-15 16:51:18 +0000285 if (Part->Retain_Layout_Version)
286 printf("Retain_Layout_Version ");
Ethan Yonker253368a2014-11-25 15:00:52 -0600287 if (Part->Mount_To_Decrypt)
288 printf("Mount_To_Decrypt ");
Ethan Yonker96af84a2015-01-05 14:58:36 -0600289 if (Part->Can_Flash_Img)
290 printf("Can_Flash_Img ");
Ethan Yonker66a19492015-12-10 10:19:45 -0600291 if (Part->Is_Adopted_Storage)
292 printf("Is_Adopted_Storage ");
Gary Peck004d48b2012-11-21 16:28:18 -0800293 printf("\n");
294 if (!Part->SubPartition_Of.empty())
295 printf(" SubPartition_Of: %s\n", Part->SubPartition_Of.c_str());
296 if (!Part->Symlink_Path.empty())
297 printf(" Symlink_Path: %s\n", Part->Symlink_Path.c_str());
298 if (!Part->Symlink_Mount_Point.empty())
299 printf(" Symlink_Mount_Point: %s\n", Part->Symlink_Mount_Point.c_str());
300 if (!Part->Primary_Block_Device.empty())
301 printf(" Primary_Block_Device: %s\n", Part->Primary_Block_Device.c_str());
302 if (!Part->Alternate_Block_Device.empty())
303 printf(" Alternate_Block_Device: %s\n", Part->Alternate_Block_Device.c_str());
304 if (!Part->Decrypted_Block_Device.empty())
305 printf(" Decrypted_Block_Device: %s\n", Part->Decrypted_Block_Device.c_str());
Ethan Yonker253368a2014-11-25 15:00:52 -0600306 if (!Part->Crypto_Key_Location.empty() && Part->Crypto_Key_Location != "footer")
307 printf(" Crypto_Key_Location: %s\n", Part->Crypto_Key_Location.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800308 if (Part->Length != 0)
309 printf(" Length: %i\n", Part->Length);
310 if (!Part->Display_Name.empty())
311 printf(" Display_Name: %s\n", Part->Display_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500312 if (!Part->Storage_Name.empty())
313 printf(" Storage_Name: %s\n", Part->Storage_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800314 if (!Part->Backup_Path.empty())
315 printf(" Backup_Path: %s\n", Part->Backup_Path.c_str());
316 if (!Part->Backup_Name.empty())
317 printf(" Backup_Name: %s\n", Part->Backup_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500318 if (!Part->Backup_Display_Name.empty())
319 printf(" Backup_Display_Name: %s\n", Part->Backup_Display_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800320 if (!Part->Backup_FileName.empty())
321 printf(" Backup_FileName: %s\n", Part->Backup_FileName.c_str());
322 if (!Part->Storage_Path.empty())
323 printf(" Storage_Path: %s\n", Part->Storage_Path.c_str());
324 if (!Part->Current_File_System.empty())
325 printf(" Current_File_System: %s\n", Part->Current_File_System.c_str());
326 if (!Part->Fstab_File_System.empty())
327 printf(" Fstab_File_System: %s\n", Part->Fstab_File_System.c_str());
328 if (Part->Format_Block_Size != 0)
Ethan Yonkerc798c9c2015-10-09 11:15:26 -0500329 printf(" Format_Block_Size: %lu\n", Part->Format_Block_Size);
Dees_Troy094207a2012-09-26 12:00:39 -0400330 if (!Part->MTD_Name.empty())
331 printf(" MTD_Name: %s\n", Part->MTD_Name.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -0400332 string back_meth = Part->Backup_Method_By_Name();
Ethan Yonker6277c792014-09-15 14:54:30 -0500333 printf(" Backup_Method: %s\n", back_meth.c_str());
Hashcode62bd9e02013-11-19 21:59:42 -0800334 if (Part->Mount_Flags || !Part->Mount_Options.empty())
Matt Mower4ab42b12016-04-21 13:52:18 -0500335 printf(" Mount_Options: %s\n", Part->Mount_Options.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -0600336 if (Part->MTP_Storage_ID)
337 printf(" MTP_Storage_ID: %i\n", Part->MTP_Storage_ID);
Ethan Yonker6277c792014-09-15 14:54:30 -0500338 printf("\n");
Dees_Troy8170a922012-09-18 15:40:25 -0400339}
340
Dees_Troy51a0e822012-09-05 15:24:24 -0400341int TWPartitionManager::Mount_By_Path(string Path, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400342 std::vector<TWPartition*>::iterator iter;
343 int ret = false;
344 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400345 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400346
Dees_Troyd93bda52013-07-03 19:55:19 +0000347 if (Local_Path == "/tmp" || Local_Path == "/")
Dees_Troy43d8b002012-09-17 16:00:01 -0400348 return true;
349
Dees_Troy5bf43922012-09-07 16:07:55 -0400350 // Iterate through all partitions
351 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400352 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400353 ret = (*iter)->Mount(Display_Error);
354 found = true;
Dees_Troy51127312012-09-08 13:08:49 -0400355 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400356 (*iter)->Mount(Display_Error);
Dees_Troy51127312012-09-08 13:08:49 -0400357 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400358 }
359 if (found) {
360 return ret;
361 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500362 gui_msg(Msg(msg::kError, "unable_find_part_path=Unable to find partition for path '{1}'")(Local_Path));
Dees_Troy5bf43922012-09-07 16:07:55 -0400363 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000364 LOGINFO("Mount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400365 }
366 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400367}
368
Dees_Troy51a0e822012-09-05 15:24:24 -0400369int TWPartitionManager::UnMount_By_Path(string Path, bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400370 std::vector<TWPartition*>::iterator iter;
371 int ret = false;
372 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400373 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy51127312012-09-08 13:08:49 -0400374
375 // Iterate through all partitions
376 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400377 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy51127312012-09-08 13:08:49 -0400378 ret = (*iter)->UnMount(Display_Error);
379 found = true;
380 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
381 (*iter)->UnMount(Display_Error);
382 }
383 }
384 if (found) {
385 return ret;
386 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500387 gui_msg(Msg(msg::kError, "unable_find_part_path=Unable to find partition for path '{1}'")(Local_Path));
Dees_Troy51127312012-09-08 13:08:49 -0400388 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000389 LOGINFO("UnMount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400390 }
391 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400392}
393
Dees_Troy51a0e822012-09-05 15:24:24 -0400394int TWPartitionManager::Is_Mounted_By_Path(string Path) {
Dees_Troy51127312012-09-08 13:08:49 -0400395 TWPartition* Part = Find_Partition_By_Path(Path);
396
397 if (Part)
398 return Part->Is_Mounted();
399 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000400 LOGINFO("Is_Mounted: Unable to find partition for path '%s'\n", Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400401 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400402}
403
Dees_Troy5bf43922012-09-07 16:07:55 -0400404int TWPartitionManager::Mount_Current_Storage(bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400405 string current_storage_path = DataManager::GetCurrentStoragePath();
406
407 if (Mount_By_Path(current_storage_path, Display_Error)) {
408 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
409 if (FreeStorage)
410 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
411 return true;
412 }
413 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400414}
415
Dees_Troy5bf43922012-09-07 16:07:55 -0400416int TWPartitionManager::Mount_Settings_Storage(bool Display_Error) {
417 return Mount_By_Path(DataManager::GetSettingsStoragePath(), Display_Error);
418}
419
420TWPartition* TWPartitionManager::Find_Partition_By_Path(string Path) {
421 std::vector<TWPartition*>::iterator iter;
Dees_Troy38bd7602012-09-14 13:33:53 -0400422 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400423
424 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400425 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path))
Dees_Troy5bf43922012-09-07 16:07:55 -0400426 return (*iter);
427 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400428 return NULL;
429}
430
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400431int TWPartitionManager::Check_Backup_Name(bool Display_Error) {
432 // Check the backup name to ensure that it is the correct size and contains only valid characters
433 // and that a backup with that name doesn't already exist
434 char backup_name[MAX_BACKUP_NAME_LEN];
435 char backup_loc[255], tw_image_dir[255];
436 int copy_size;
437 int index, cur_char;
438 string Backup_Name, Backup_Loc;
439
440 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
441 copy_size = Backup_Name.size();
442 // Check size
443 if (copy_size > MAX_BACKUP_NAME_LEN) {
444 if (Display_Error)
Ethan Yonker74db1572015-10-28 12:44:49 -0500445 gui_err("backup_name_len=Backup name is too long.");
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400446 return -2;
447 }
448
449 // Check each character
450 strncpy(backup_name, Backup_Name.c_str(), copy_size);
Dees_Troya13d74f2013-03-24 08:54:55 -0500451 if (copy_size == 1 && strncmp(backup_name, "0", 1) == 0)
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400452 return 0; // A "0" (zero) means to use the current timestamp for the backup name
453 for (index=0; index<copy_size; index++) {
454 cur_char = (int)backup_name[index];
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500455 if (cur_char == 32 || (cur_char >= 48 && cur_char <= 57) || (cur_char >= 65 && cur_char <= 91) || cur_char == 93 || cur_char == 95 || (cur_char >= 97 && cur_char <= 123) || cur_char == 125 || cur_char == 45 || cur_char == 46) {
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400456 // These are valid characters
457 // Numbers
458 // Upper case letters
459 // Lower case letters
460 // Space
461 // and -_.{}[]
462 } else {
463 if (Display_Error)
Ethan Yonker4adc33e2016-01-22 16:07:11 -0600464 gui_msg(Msg(msg::kError, "backup_name_invalid=Backup name '{1}' contains invalid character: '{1}'")(Backup_Name)((char)cur_char));
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400465 return -3;
466 }
467 }
468
469 // Check to make sure that a backup with this name doesn't already exist
470 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Loc);
471 strcpy(backup_loc, Backup_Loc.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500472 sprintf(tw_image_dir,"%s/%s", backup_loc, Backup_Name.c_str());
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500473 if (TWFunc::Path_Exists(tw_image_dir)) {
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400474 if (Display_Error)
Matt Mower3c366972015-12-25 19:28:31 -0600475 gui_err("backup_name_exists=A backup with that name already exists!");
bigbiffce8f83c2015-12-12 18:30:21 -0500476
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400477 return -4;
478 }
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400479 // No problems found, return 0
480 return 0;
481}
482
bigbiff bigbiffa481d362016-08-08 20:36:10 -0400483bool TWPartitionManager::Make_MD5(PartitionSettings *part_settings)
Dees_Troy43d8b002012-09-17 16:00:01 -0400484{
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500485 string command, result;
486
487 if (part_settings->Part == NULL)
488 return false;
489 string Full_File = part_settings->Backup_Folder + part_settings->Part->Backup_FileName;
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500490 twrpDigest md5sum;
Dees_Troy43d8b002012-09-17 16:00:01 -0400491
bigbiff bigbiffa481d362016-08-08 20:36:10 -0400492 if (!part_settings->generate_md5)
Dees_Troy43d8b002012-09-17 16:00:01 -0400493 return true;
Dees_Troy43d8b002012-09-17 16:00:01 -0400494
Ethan Yonker74db1572015-10-28 12:44:49 -0500495 TWFunc::GUI_Operation_Text(TW_GENERATE_MD5_TEXT, gui_parse_text("{@generating_md51}"));
496 gui_msg("generating_md52= * Generating md5...");
Dees_Troy43d8b002012-09-17 16:00:01 -0400497 if (TWFunc::Path_Exists(Full_File)) {
bigbiffce8f83c2015-12-12 18:30:21 -0500498 md5sum.setfn(Full_File);
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500499 if (md5sum.computeMD5() == 0)
500 if (md5sum.write_md5digest() == 0)
Ethan Yonker74db1572015-10-28 12:44:49 -0500501 gui_msg("md5_created= * MD5 Created.");
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500502 else
503 return -1;
504 else
Ethan Yonker74db1572015-10-28 12:44:49 -0500505 gui_err("md5_error= * MD5 Error!");
Dees_Troy43d8b002012-09-17 16:00:01 -0400506 } else {
507 char filename[512];
508 int index = 0;
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500509 string strfn;
Dees_Troy43d8b002012-09-17 16:00:01 -0400510 sprintf(filename, "%s%03i", Full_File.c_str(), index);
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500511 strfn = filename;
Dees_Troy83bd4832013-05-04 12:39:56 +0000512 while (index < 1000) {
bigbiff bigbiff65a4c732013-03-15 15:17:50 -0400513 md5sum.setfn(filename);
Dees_Troy83bd4832013-05-04 12:39:56 +0000514 if (TWFunc::Path_Exists(filename)) {
515 if (md5sum.computeMD5() == 0) {
516 if (md5sum.write_md5digest() != 0)
517 {
Ethan Yonker74db1572015-10-28 12:44:49 -0500518 gui_err("md5_error= * MD5 Error!");
Dees_Troy83bd4832013-05-04 12:39:56 +0000519 return false;
520 }
521 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500522 gui_err("md5_compute_error= * Error computing MD5.");
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500523 return false;
524 }
525 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400526 index++;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400527 sprintf(filename, "%s%03i", Full_File.c_str(), index);
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500528 strfn = filename;
Dees_Troy43d8b002012-09-17 16:00:01 -0400529 }
530 if (index == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000531 LOGERR("Backup file: '%s' not found!\n", filename);
Dees_Troy43d8b002012-09-17 16:00:01 -0400532 return false;
533 }
Ethan Yonker74db1572015-10-28 12:44:49 -0500534 gui_msg("md5_created= * MD5 Created.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400535 }
536 return true;
537}
538
bigbiffce8f83c2015-12-12 18:30:21 -0500539
540bool TWPartitionManager::Backup_Partition(PartitionSettings *part_settings) {
Dees_Troy43d8b002012-09-17 16:00:01 -0400541 time_t start, stop;
bigbiffce8f83c2015-12-12 18:30:21 -0500542 int use_compression, adb_control_bu_fd;
543 string backup_log = part_settings->Backup_Folder + "/recovery.log";
Dees_Troy43d8b002012-09-17 16:00:01 -0400544
bigbiffce8f83c2015-12-12 18:30:21 -0500545 if (part_settings->Part == NULL)
Dees_Troy43d8b002012-09-17 16:00:01 -0400546 return true;
547
Dees_Troy093b7642012-09-21 15:59:38 -0400548 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400549
Tom Hite5a926722014-09-15 01:31:03 +0000550 TWFunc::SetPerformanceMode(true);
Dees_Troy43d8b002012-09-17 16:00:01 -0400551 time(&start);
552
bigbiffce8f83c2015-12-12 18:30:21 -0500553 if (part_settings->Part->Backup(part_settings, &tar_fork_pid)) {
Tom Hite5a926722014-09-15 01:31:03 +0000554 bool md5Success = false;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400555 if (part_settings->adbbackup) {
556 md5Success = true;
557 }
558 else
559 md5Success = Make_MD5(part_settings);
560
561 TWFunc::SetPerformanceMode(false);
bigbiffce8f83c2015-12-12 18:30:21 -0500562 if (part_settings->Part->Has_SubPartition) {
Dees_Troy8170a922012-09-18 15:40:25 -0400563 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400564 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -0400565
566 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400567 if ((*subpart)->Can_Be_Backed_Up && (*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500568 part_settings->Part = *subpart;
bigbiffce8f83c2015-12-12 18:30:21 -0500569 if (!(*subpart)->Backup(part_settings, &tar_fork_pid)) {
Tom Hite5a926722014-09-15 01:31:03 +0000570 TWFunc::SetPerformanceMode(false);
bigbiffce8f83c2015-12-12 18:30:21 -0500571 Clean_Backup_Folder(part_settings->Backup_Folder);
bigbiffbf1d6722015-02-14 16:25:59 -0500572 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
573 tw_set_default_metadata(backup_log.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -0400574 return false;
Tom Hite5a926722014-09-15 01:31:03 +0000575 }
Dees_Troy2727b992013-08-14 20:09:30 +0000576 sync();
577 sync();
bigbiffce8f83c2015-12-12 18:30:21 -0500578 if (!part_settings->adbbackup) {
bigbiff bigbiffa481d362016-08-08 20:36:10 -0400579 if (!Make_MD5(part_settings)) {
bigbiffce8f83c2015-12-12 18:30:21 -0500580 TWFunc::SetPerformanceMode(false);
581 return false;
582 }
Tom Hite5a926722014-09-15 01:31:03 +0000583 }
Dees_Troy8170a922012-09-18 15:40:25 -0400584 }
585 }
586 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400587 time(&stop);
that203bcc92015-05-09 17:27:33 +0200588 int backup_time = (int) difftime(stop, start);
Dees_Troy2673cec2013-04-02 20:22:16 +0000589 LOGINFO("Partition Backup time: %d\n", backup_time);
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400590
bigbiffce8f83c2015-12-12 18:30:21 -0500591 if (part_settings->Part->Backup_Method == BM_FILES) {
592 part_settings->file_time += backup_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400593 } else {
bigbiffce8f83c2015-12-12 18:30:21 -0500594 part_settings->img_time += backup_time;
595
Dees_Troy43d8b002012-09-17 16:00:01 -0400596 }
Tom Hite5a926722014-09-15 01:31:03 +0000597
Tom Hite5a926722014-09-15 01:31:03 +0000598 return md5Success;
Dees_Troy43d8b002012-09-17 16:00:01 -0400599 } else {
bigbiffce8f83c2015-12-12 18:30:21 -0500600 Clean_Backup_Folder(part_settings->Backup_Folder);
bigbiffbf1d6722015-02-14 16:25:59 -0500601 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
602 tw_set_default_metadata(backup_log.c_str());
Tom Hite5a926722014-09-15 01:31:03 +0000603 TWFunc::SetPerformanceMode(false);
Dees_Troy43d8b002012-09-17 16:00:01 -0400604 return false;
605 }
bigbiff7abc5fe2015-01-17 16:53:12 -0500606 return 0;
607}
608
bigbiffbf1d6722015-02-14 16:25:59 -0500609void TWPartitionManager::Clean_Backup_Folder(string Backup_Folder) {
610 DIR *d = opendir(Backup_Folder.c_str());
611 struct dirent *p;
612 int r;
613
Ethan Yonker74db1572015-10-28 12:44:49 -0500614 gui_msg("backup_clean=Backup Failed. Cleaning Backup Folder.");
bigbiffbf1d6722015-02-14 16:25:59 -0500615
616 if (d == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500617 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Backup_Folder)(strerror(errno)));
bigbiffbf1d6722015-02-14 16:25:59 -0500618 return;
619 }
620
Matt Mower2b18a532015-02-20 16:58:05 -0600621 while ((p = readdir(d))) {
bigbiffbf1d6722015-02-14 16:25:59 -0500622 if (!strcmp(p->d_name, ".") || !strcmp(p->d_name, ".."))
623 continue;
624
625 string path = Backup_Folder + p->d_name;
626
627 size_t dot = path.find_last_of(".") + 1;
628 if (path.substr(dot) == "win" || path.substr(dot) == "md5" || path.substr(dot) == "info") {
629 r = unlink(path.c_str());
630 if (r != 0) {
631 LOGINFO("Unable to unlink '%s: %s'\n", path.c_str(), strerror(errno));
632 }
633 }
634 }
635 closedir(d);
636}
637
Ethan Yonker472f5062016-02-25 13:47:30 -0600638int TWPartitionManager::Check_Backup_Cancel() {
639 return stop_backup.get_value();
640}
641
bigbiff7abc5fe2015-01-17 16:53:12 -0500642int TWPartitionManager::Cancel_Backup() {
643 string Backup_Folder, Backup_Name, Full_Backup_Path;
644
645 stop_backup.set_value(1);
646
647 if (tar_fork_pid != 0) {
648 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
649 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Folder);
650 Full_Backup_Path = Backup_Folder + "/" + Backup_Name + "/";
651 LOGINFO("Killing pid: %d\n", tar_fork_pid);
652 kill(tar_fork_pid, SIGUSR2);
653 while (kill(tar_fork_pid, 0) == 0) {
654 usleep(1000);
655 }
656 LOGINFO("Backup_Run stopped and returning false, backup cancelled.\n");
657 LOGINFO("Removing directory %s\n", Full_Backup_Path.c_str());
658 TWFunc::removeDir(Full_Backup_Path, false);
659 tar_fork_pid = 0;
660 }
661
662 return 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400663}
664
bigbiffce8f83c2015-12-12 18:30:21 -0500665int TWPartitionManager::Run_Backup(bool adbbackup) {
666 PartitionSettings part_settings;
667 int check, partition_count = 0, disable_free_space_check = 0, do_md5 = 0;
668 int gui_adb_backup;
669 string Backup_Name, Backup_List, backup_path;
670 unsigned long long total_bytes = 0, free_space = 0, subpart_size;
Dees_Troy43d8b002012-09-17 16:00:01 -0400671 TWPartition* storage = NULL;
Dees_Troy8170a922012-09-18 15:40:25 -0400672 std::vector<TWPartition*>::iterator subpart;
Dees_Troy43d8b002012-09-17 16:00:01 -0400673 struct tm *t;
674 time_t start, stop, seconds, total_start, total_stop;
Dees_Troya13d74f2013-03-24 08:54:55 -0500675 size_t start_pos = 0, end_pos = 0;
bigbiff7abc5fe2015-01-17 16:53:12 -0500676 stop_backup.set_value(0);
Dees_Troy43d8b002012-09-17 16:00:01 -0400677 seconds = time(0);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500678 t = localtime(&seconds);
Dees_Troy43d8b002012-09-17 16:00:01 -0400679
bigbiffce8f83c2015-12-12 18:30:21 -0500680 part_settings.img_bytes_remaining = 0;
681 part_settings.file_bytes_remaining = 0;
682 part_settings.img_time = 0;
683 part_settings.file_time = 0;
684 part_settings.img_bytes = 0;
685 part_settings.file_bytes = 0;
686 part_settings.PM_Method = PM_BACKUP;
687
688 DataManager::GetValue("tw_enable_adb_backup", gui_adb_backup);
689 if (gui_adb_backup == true)
690 adbbackup = true;
691
692 part_settings.adbbackup = adbbackup;
Dees_Troy43d8b002012-09-17 16:00:01 -0400693 time(&total_start);
694
695 Update_System_Details();
696
697 if (!Mount_Current_Storage(true))
698 return false;
699
bigbiffce8f83c2015-12-12 18:30:21 -0500700
Dees_Troy43d8b002012-09-17 16:00:01 -0400701 DataManager::GetValue(TW_SKIP_MD5_GENERATE_VAR, do_md5);
Dees_Troy4a2a1262012-09-18 09:33:47 -0400702 if (do_md5 == 0)
bigbiffce8f83c2015-12-12 18:30:21 -0500703 part_settings.generate_md5 = true;
Dees_Troyc5865ab2012-09-24 15:08:04 -0400704 else
bigbiffce8f83c2015-12-12 18:30:21 -0500705 part_settings.generate_md5 = false;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400706
bigbiffce8f83c2015-12-12 18:30:21 -0500707 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, part_settings.Backup_Folder);
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500708 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
709 if (Backup_Name == gui_lookup("curr_date", "(Current Date)")) {
710 Backup_Name = TWFunc::Get_Current_Date();
711 } else if (Backup_Name == gui_lookup("auto_generate", "(Auto Generate)") || Backup_Name == "0" || Backup_Name.empty()) {
Dees Troyb21cc642013-09-10 17:36:41 +0000712 TWFunc::Auto_Generate_Backup_Name();
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500713 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
Dees_Troy43d8b002012-09-17 16:00:01 -0400714 }
bigbiffce8f83c2015-12-12 18:30:21 -0500715
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500716 LOGINFO("Backup Name is: '%s'\n", Backup_Name.c_str());
717 part_settings.Backup_Folder = part_settings.Backup_Folder + "/" + Backup_Name + "/";
bigbiffce8f83c2015-12-12 18:30:21 -0500718
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500719 LOGINFO("Full_Backup_Path is: '%s'\n", part_settings.Backup_Folder.c_str());
Dees_Troy43d8b002012-09-17 16:00:01 -0400720
Dees_Troy2673cec2013-04-02 20:22:16 +0000721 LOGINFO("Calculating backup details...\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500722 DataManager::GetValue("tw_backup_list", Backup_List);
723 if (!Backup_List.empty()) {
724 end_pos = Backup_List.find(";", start_pos);
725 while (end_pos != string::npos && start_pos < Backup_List.size()) {
726 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -0500727 part_settings.Part = Find_Partition_By_Path(backup_path);
728 if (part_settings.Part != NULL) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500729 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500730 if (part_settings.Part->Backup_Method == BM_FILES)
731 part_settings.file_bytes += part_settings.Part->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500732 else
bigbiffce8f83c2015-12-12 18:30:21 -0500733 part_settings.img_bytes += part_settings.Part->Backup_Size;
734 if (part_settings.Part->Has_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500735 std::vector<TWPartition*>::iterator subpart;
736
737 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiffce8f83c2015-12-12 18:30:21 -0500738 if ((*subpart)->Can_Be_Backed_Up && (*subpart)->Is_Present && (*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == part_settings.Part->Mount_Point) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500739 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500740 if ((*subpart)->Backup_Method == BM_FILES)
741 part_settings.file_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500742 else
bigbiffce8f83c2015-12-12 18:30:21 -0500743 part_settings.img_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500744 }
745 }
Dees_Troy8170a922012-09-18 15:40:25 -0400746 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500747 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500748 gui_msg(Msg(msg::kError, "unable_to_locate_partition=Unable to locate '{1}' partition for backup calculations.")(backup_path));
Dees_Troy8170a922012-09-18 15:40:25 -0400749 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500750 start_pos = end_pos + 1;
751 end_pos = Backup_List.find(";", start_pos);
Dees_Troy43d8b002012-09-17 16:00:01 -0400752 }
753 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400754
755 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500756 gui_msg("no_partition_selected=No partitions selected for backup.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400757 return false;
758 }
bigbiffce8f83c2015-12-12 18:30:21 -0500759 if (adbbackup) {
760 if (twadbbu::Write_ADB_Stream_Header(partition_count) == false) {
761 return false;
762 }
763 }
764 total_bytes = part_settings.file_bytes + part_settings.img_bytes;
Ethan Yonker472f5062016-02-25 13:47:30 -0600765 ProgressTracking progress(total_bytes);
bigbiffce8f83c2015-12-12 18:30:21 -0500766 part_settings.progress = &progress;
767
Ethan Yonker74db1572015-10-28 12:44:49 -0500768 gui_msg(Msg("total_partitions_backup= * Total number of partitions to back up: {1}")(partition_count));
769 gui_msg(Msg("total_backup_size= * Total size of all data: {1}MB")(total_bytes / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400770 storage = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
771 if (storage != NULL) {
772 free_space = storage->Free;
Ethan Yonker74db1572015-10-28 12:44:49 -0500773 gui_msg(Msg("available_space= * Available space: {1}MB")(free_space / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400774 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500775 gui_err("unable_locate_storage=Unable to locate storage device.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400776 return false;
777 }
bigbiffbf1d6722015-02-14 16:25:59 -0500778
Matt Mowerbfccfb82016-04-25 23:22:31 -0500779 DataManager::GetValue(TW_DISABLE_FREE_SPACE_VAR, disable_free_space_check);
bigbiffce8f83c2015-12-12 18:30:21 -0500780
781 if (adbbackup)
782 disable_free_space_check = true;
783
bigbiffbf1d6722015-02-14 16:25:59 -0500784 if (!disable_free_space_check) {
785 if (free_space - (32 * 1024 * 1024) < total_bytes) {
786 // We require an extra 32MB just in case
Ethan Yonker74db1572015-10-28 12:44:49 -0500787 gui_err("no_space=Not enough free space on storage.");
bigbiffbf1d6722015-02-14 16:25:59 -0500788 return false;
789 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400790 }
bigbiffce8f83c2015-12-12 18:30:21 -0500791 part_settings.img_bytes_remaining = part_settings.img_bytes;
792 part_settings.file_bytes_remaining = part_settings.file_bytes;
Dees_Troy43d8b002012-09-17 16:00:01 -0400793
Ethan Yonker74db1572015-10-28 12:44:49 -0500794 gui_msg("backup_started=[BACKUP STARTED]");
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500795 gui_msg(Msg("backup_folder= * Backup Folder: {1}")(part_settings.Backup_Folder));
796 if (!TWFunc::Recursive_Mkdir(part_settings.Backup_Folder)) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500797 gui_err("fail_backup_folder=Failed to make backup folder.");
Dees_Troyd4b22b02013-01-18 17:17:58 +0000798 return false;
799 }
800
Dees_Troy2673cec2013-04-02 20:22:16 +0000801 DataManager::SetProgress(0.0);
Dees_Troy093b7642012-09-21 15:59:38 -0400802
Dees_Troya13d74f2013-03-24 08:54:55 -0500803 start_pos = 0;
804 end_pos = Backup_List.find(";", start_pos);
805 while (end_pos != string::npos && start_pos < Backup_List.size()) {
bigbiff7abc5fe2015-01-17 16:53:12 -0500806 if (stop_backup.get_value() != 0)
807 return -1;
Dees_Troya13d74f2013-03-24 08:54:55 -0500808 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -0500809 part_settings.Part = Find_Partition_By_Path(backup_path);
810 if (part_settings.Part != NULL) {
811 if (!Backup_Partition(&part_settings))
812
Dees_Troya13d74f2013-03-24 08:54:55 -0500813 return false;
814 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500815 gui_msg(Msg(msg::kError, "unable_to_locate_partition=Unable to locate '{1}' partition for backup calculations.")(backup_path));
Dees_Troya13d74f2013-03-24 08:54:55 -0500816 }
817 start_pos = end_pos + 1;
818 end_pos = Backup_List.find(";", start_pos);
819 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400820
821 // Average BPS
bigbiffce8f83c2015-12-12 18:30:21 -0500822 if (part_settings.img_time == 0)
823 part_settings.img_time = 1;
824 if (part_settings.file_time == 0)
825 part_settings.file_time = 1;
826 int img_bps = (int)part_settings.img_bytes / (int)part_settings.img_time;
827 unsigned long long file_bps = part_settings.file_bytes / (int)part_settings.file_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400828
bigbiffce8f83c2015-12-12 18:30:21 -0500829 if (part_settings.file_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -0600830 gui_msg(Msg("avg_backup_fs=Average backup rate for file systems: {1} MB/sec")(file_bps / (1024 * 1024)));
bigbiffce8f83c2015-12-12 18:30:21 -0500831 if (part_settings.img_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -0600832 gui_msg(Msg("avg_backup_img=Average backup rate for imaged drives: {1} MB/sec")(img_bps / (1024 * 1024)));
Dees_Troy43d8b002012-09-17 16:00:01 -0400833
834 time(&total_stop);
835 int total_time = (int) difftime(total_stop, total_start);
bigbiffce8f83c2015-12-12 18:30:21 -0500836
837 uint64_t actual_backup_size;
838 if (!adbbackup)
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500839 actual_backup_size = du.Get_Folder_Size(part_settings.Backup_Folder);
bigbiffce8f83c2015-12-12 18:30:21 -0500840 else
841 actual_backup_size = part_settings.file_bytes + part_settings.img_bytes;
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500842 actual_backup_size /= (1024LLU * 1024LLU);
Dees_Troy43d8b002012-09-17 16:00:01 -0400843
bigbiffce8f83c2015-12-12 18:30:21 -0500844 int prev_img_bps = 0, use_compression = 0;
845 unsigned long long prev_file_bps = 0;
Dees_Troy093b7642012-09-21 15:59:38 -0400846 img_bps += (prev_img_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500847 img_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -0400848
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500849 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400850 if (use_compression)
851 DataManager::GetValue(TW_BACKUP_AVG_FILE_COMP_RATE, prev_file_bps);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500852 else
Dees_Troy093b7642012-09-21 15:59:38 -0400853 DataManager::GetValue(TW_BACKUP_AVG_FILE_RATE, prev_file_bps);
854 file_bps += (prev_file_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500855 file_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -0400856
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500857 DataManager::SetValue(TW_BACKUP_AVG_IMG_RATE, img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -0400858 if (use_compression)
859 DataManager::SetValue(TW_BACKUP_AVG_FILE_COMP_RATE, file_bps);
860 else
861 DataManager::SetValue(TW_BACKUP_AVG_FILE_RATE, file_bps);
862
Ethan Yonker74db1572015-10-28 12:44:49 -0500863 gui_msg(Msg("total_backed_size=[{1} MB TOTAL BACKED UP]")(actual_backup_size));
Dees_Troy43d8b002012-09-17 16:00:01 -0400864 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -0400865 UnMount_Main_Partitions();
Ethan Yonker56db1c42015-12-20 22:49:00 -0600866 gui_msg(Msg(msg::kHighlight, "backup_completed=[BACKUP COMPLETED IN {1} SECONDS]")(total_time)); // the end
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500867 string backup_log = part_settings.Backup_Folder + "recovery.log";
Dees_Troy3f5c4e82013-02-01 15:16:59 +0000868 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
Ethan Yonker4b94cfd2014-12-11 10:00:45 -0600869 tw_set_default_metadata(backup_log.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -0500870
871 if (part_settings.adbbackup) {
872 if (twadbbu::Write_ADB_Stream_Trailer() == false) {
873 return false;
874 }
875 }
876 part_settings.adbbackup = false;
877 DataManager::SetValue("tw_enable_adb_backup", 0);
878
Dees_Troy3f5c4e82013-02-01 15:16:59 +0000879 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -0400880}
881
bigbiffce8f83c2015-12-12 18:30:21 -0500882bool TWPartitionManager::Restore_Partition(PartitionSettings *part_settings) {
Dees_Troy4a2a1262012-09-18 09:33:47 -0400883 time_t Start, Stop;
bigbiffce8f83c2015-12-12 18:30:21 -0500884
Tom Hite5a926722014-09-15 01:31:03 +0000885 TWFunc::SetPerformanceMode(true);
bigbiffce8f83c2015-12-12 18:30:21 -0500886
Dees_Troy4a2a1262012-09-18 09:33:47 -0400887 time(&Start);
Ethan Yonker472f5062016-02-25 13:47:30 -0600888
bigbiffce8f83c2015-12-12 18:30:21 -0500889
890 if (!part_settings->Part->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +0000891 TWFunc::SetPerformanceMode(false);
Dees_Troy4a2a1262012-09-18 09:33:47 -0400892 return false;
Tom Hite5a926722014-09-15 01:31:03 +0000893 }
bigbiffce8f83c2015-12-12 18:30:21 -0500894 if (part_settings->Part->Has_SubPartition) {
Dees_Troy8170a922012-09-18 15:40:25 -0400895 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400896 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -0400897
898 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400899 part_settings->Part = *subpart;
900 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
901 part_settings->Part = (*subpart);
bigbiffce8f83c2015-12-12 18:30:21 -0500902 if (!(*subpart)->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +0000903 TWFunc::SetPerformanceMode(false);
Dees_Troy8170a922012-09-18 15:40:25 -0400904 return false;
Tom Hite5a926722014-09-15 01:31:03 +0000905 }
Dees_Troy8170a922012-09-18 15:40:25 -0400906 }
907 }
908 }
Dees_Troy4a2a1262012-09-18 09:33:47 -0400909 time(&Stop);
Tom Hite5a926722014-09-15 01:31:03 +0000910 TWFunc::SetPerformanceMode(false);
bigbiffce8f83c2015-12-12 18:30:21 -0500911 gui_msg(Msg("restore_part_done=[{1} done ({2} seconds)]")(part_settings->Part->Backup_Display_Name)((int)difftime(Stop, Start)));
912
Dees_Troy4a2a1262012-09-18 09:33:47 -0400913 return true;
914}
915
Ethan Yonker472f5062016-02-25 13:47:30 -0600916int TWPartitionManager::Run_Restore(const string& Restore_Name) {
bigbiffce8f83c2015-12-12 18:30:21 -0500917 PartitionSettings part_settings;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400918 int check_md5, check, partition_count = 0;
Dees_Troya13d74f2013-03-24 08:54:55 -0500919 TWPartition* restore_part = NULL;
bigbiffce8f83c2015-12-12 18:30:21 -0500920
Dees_Troy4a2a1262012-09-18 09:33:47 -0400921 time_t rStart, rStop;
922 time(&rStart);
Dees_Troya13d74f2013-03-24 08:54:55 -0500923 string Restore_List, restore_path;
924 size_t start_pos = 0, end_pos;
bigbiffce8f83c2015-12-12 18:30:21 -0500925
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500926 part_settings.Backup_Folder = Restore_Name;
bigbiffce8f83c2015-12-12 18:30:21 -0500927 part_settings.Part = NULL;
928 part_settings.partition_count = 0;
929 part_settings.total_restore_size = 0;
930 part_settings.adbbackup = false;
931 part_settings.PM_Method = PM_RESTORE;
Dees_Troy43d8b002012-09-17 16:00:01 -0400932
Ethan Yonker74db1572015-10-28 12:44:49 -0500933 gui_msg("restore_started=[RESTORE STARTED]");
934 gui_msg(Msg("restore_folder=Restore folder: '{1}'")(Restore_Name));
Dees_Troy43d8b002012-09-17 16:00:01 -0400935
Dees_Troy4a2a1262012-09-18 09:33:47 -0400936 if (!Mount_Current_Storage(true))
937 return false;
938
939 DataManager::GetValue(TW_SKIP_MD5_CHECK_VAR, check_md5);
Dees_Troya13d74f2013-03-24 08:54:55 -0500940 if (check_md5 > 0) {
941 // Check MD5 files first before restoring to ensure that all of them match before starting a restore
Ethan Yonker74db1572015-10-28 12:44:49 -0500942 TWFunc::GUI_Operation_Text(TW_VERIFY_MD5_TEXT, gui_parse_text("{@verifying_md5}"));
943 gui_msg("verifying_md5=Verifying MD5");
Dees_Troya13d74f2013-03-24 08:54:55 -0500944 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500945 gui_msg("skip_md5=Skipping MD5 check based on user setting.");
Dees_Troya13d74f2013-03-24 08:54:55 -0500946 }
Ethan Yonker74db1572015-10-28 12:44:49 -0500947 gui_msg("calc_restore=Calculating restore details...");
Dees_Troya13d74f2013-03-24 08:54:55 -0500948 DataManager::GetValue("tw_restore_selected", Restore_List);
bigbiffce8f83c2015-12-12 18:30:21 -0500949
Dees_Troya13d74f2013-03-24 08:54:55 -0500950 if (!Restore_List.empty()) {
951 end_pos = Restore_List.find(";", start_pos);
952 while (end_pos != string::npos && start_pos < Restore_List.size()) {
953 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -0500954 part_settings.Part = Find_Partition_By_Path(restore_path);
955 if (part_settings.Part != NULL) {
bigbiffce8f83c2015-12-12 18:30:21 -0500956 if (part_settings.Part->Mount_Read_Only) {
957 gui_msg(Msg(msg::kError, "restore_read_only=Cannot restore {1} -- mounted read only.")(part_settings.Part->Backup_Display_Name));
Ethan Yonkereb32b1f2015-05-18 10:23:03 -0500958 return false;
959 }
bigbiffce8f83c2015-12-12 18:30:21 -0500960
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400961 if (check_md5 > 0 && !part_settings.Part->Check_MD5(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -0500962 return false;
bigbiffce8f83c2015-12-12 18:30:21 -0500963 part_settings.partition_count++;
964 part_settings.total_restore_size += part_settings.Part->Get_Restore_Size(&part_settings);
965 if (part_settings.Part->Has_SubPartition) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400966 TWPartition *parentPart = part_settings.Part;
Dees_Troya13d74f2013-03-24 08:54:55 -0500967 std::vector<TWPartition*>::iterator subpart;
968
969 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400970 part_settings.Part = *subpart;
971 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
972 if (check_md5 > 0 && !(*subpart)->Check_MD5(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -0500973 return false;
bigbiffce8f83c2015-12-12 18:30:21 -0500974 part_settings.total_restore_size += (*subpart)->Get_Restore_Size(&part_settings);
Dees_Troya13d74f2013-03-24 08:54:55 -0500975 }
976 }
977 }
978 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500979 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -0500980 }
981 start_pos = end_pos + 1;
982 end_pos = Restore_List.find(";", start_pos);
Dees_Troy4a2a1262012-09-18 09:33:47 -0400983 }
Dees_Troy4a2a1262012-09-18 09:33:47 -0400984 }
Dees_Troy4a2a1262012-09-18 09:33:47 -0400985
bigbiffce8f83c2015-12-12 18:30:21 -0500986 if (part_settings.partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500987 gui_err("no_part_restore=No partitions selected for restore.");
Dees_Troy4a2a1262012-09-18 09:33:47 -0400988 return false;
989 }
990
bigbiffce8f83c2015-12-12 18:30:21 -0500991 gui_msg(Msg("restore_part_count=Restoring {1} partitions...")(part_settings.partition_count));
992 gui_msg(Msg("total_restore_size=Total restore size is {1}MB")(part_settings.total_restore_size / 1048576));
Dees_Troy2673cec2013-04-02 20:22:16 +0000993 DataManager::SetProgress(0.0);
bigbiffce8f83c2015-12-12 18:30:21 -0500994 ProgressTracking progress(part_settings.total_restore_size);
995 part_settings.progress = &progress;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500996
Dees_Troya13d74f2013-03-24 08:54:55 -0500997 start_pos = 0;
998 if (!Restore_List.empty()) {
999 end_pos = Restore_List.find(";", start_pos);
1000 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1001 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001002
1003 part_settings.Part = Find_Partition_By_Path(restore_path);
1004 if (part_settings.Part != NULL) {
1005 part_settings.partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -05001006 if (!Restore_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001007 return false;
1008 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001009 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001010 }
1011 start_pos = end_pos + 1;
1012 end_pos = Restore_List.find(";", start_pos);
1013 }
1014 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001015 TWFunc::GUI_Operation_Text(TW_UPDATE_SYSTEM_DETAILS_TEXT, gui_parse_text("{@updating_system_details}"));
Dees_Troy43d8b002012-09-17 16:00:01 -04001016 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001017 UnMount_Main_Partitions();
Dees_Troy4a2a1262012-09-18 09:33:47 -04001018 time(&rStop);
Matt Mower3c366972015-12-25 19:28:31 -06001019 gui_msg(Msg(msg::kHighlight, "restore_completed=[RESTORE COMPLETED IN {1} SECONDS]")((int)difftime(rStop,rStart)));
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001020 DataManager::SetValue("tw_file_progress", "");
bigbiffce8f83c2015-12-12 18:30:21 -05001021
Dees_Troy63c8df72012-09-10 14:02:05 -04001022 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001023}
1024
1025void TWPartitionManager::Set_Restore_Files(string Restore_Name) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001026 // Start with the default values
Dees_Troya13d74f2013-03-24 08:54:55 -05001027 string Restore_List;
Dees_Troy83bd4832013-05-04 12:39:56 +00001028 bool get_date = true, check_encryption = true;
1029
1030 DataManager::SetValue("tw_restore_encrypted", 0);
Dees_Troy63c8df72012-09-10 14:02:05 -04001031
1032 DIR* d;
1033 d = opendir(Restore_Name.c_str());
1034 if (d == NULL)
1035 {
Ethan Yonker74db1572015-10-28 12:44:49 -05001036 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Restore_Name)(strerror(errno)));
Dees_Troy63c8df72012-09-10 14:02:05 -04001037 return;
1038 }
1039
1040 struct dirent* de;
1041 while ((de = readdir(d)) != NULL)
1042 {
1043 // Strip off three components
1044 char str[256];
1045 char* label;
1046 char* fstype = NULL;
1047 char* extn = NULL;
1048 char* ptr;
1049
1050 strcpy(str, de->d_name);
1051 if (strlen(str) <= 2)
1052 continue;
1053
1054 if (get_date) {
1055 char file_path[255];
1056 struct stat st;
1057
1058 strcpy(file_path, Restore_Name.c_str());
1059 strcat(file_path, "/");
1060 strcat(file_path, str);
1061 stat(file_path, &st);
1062 string backup_date = ctime((const time_t*)(&st.st_mtime));
1063 DataManager::SetValue(TW_RESTORE_FILE_DATE, backup_date);
1064 get_date = false;
1065 }
1066
1067 label = str;
1068 ptr = label;
1069 while (*ptr && *ptr != '.') ptr++;
1070 if (*ptr == '.')
1071 {
1072 *ptr = 0x00;
1073 ptr++;
1074 fstype = ptr;
1075 }
1076 while (*ptr && *ptr != '.') ptr++;
1077 if (*ptr == '.')
1078 {
1079 *ptr = 0x00;
1080 ptr++;
1081 extn = ptr;
1082 }
1083
Dees_Troy83bd4832013-05-04 12:39:56 +00001084 if (fstype == NULL || extn == NULL || strcmp(fstype, "log") == 0) continue;
Dees_Troya13d74f2013-03-24 08:54:55 -05001085 int extnlength = strlen(extn);
Dees_Troy83bd4832013-05-04 12:39:56 +00001086 if (extnlength != 3 && extnlength != 6) continue;
1087 if (extnlength >= 3 && strncmp(extn, "win", 3) != 0) continue;
1088 //if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
1089
1090 if (check_encryption) {
1091 string filename = Restore_Name + "/";
1092 filename += de->d_name;
1093 if (TWFunc::Get_File_Type(filename) == 2) {
1094 LOGINFO("'%s' is encrypted\n", filename.c_str());
1095 DataManager::SetValue("tw_restore_encrypted", 1);
1096 }
1097 }
Dees_Troya13d74f2013-03-24 08:54:55 -05001098 if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
Dees_Troy63c8df72012-09-10 14:02:05 -04001099
1100 TWPartition* Part = Find_Partition_By_Path(label);
1101 if (Part == NULL)
1102 {
Ethan Yonker74db1572015-10-28 12:44:49 -05001103 gui_msg(Msg(msg::kError, "unable_locate_part_backup_name=Unable to locate partition by backup name: '{1}'")(label));
Dees_Troy63c8df72012-09-10 14:02:05 -04001104 continue;
1105 }
1106
1107 Part->Backup_FileName = de->d_name;
1108 if (strlen(extn) > 3) {
1109 Part->Backup_FileName.resize(Part->Backup_FileName.size() - strlen(extn) + 3);
1110 }
1111
James Christopher Adduonofddbdfa2016-02-29 15:08:11 -05001112 if (!Part->Is_SubPartition)
1113 Restore_List += Part->Backup_Path + ";";
Dees_Troy63c8df72012-09-10 14:02:05 -04001114 }
1115 closedir(d);
1116
Dees_Troya13d74f2013-03-24 08:54:55 -05001117 // Set the final value
1118 DataManager::SetValue("tw_restore_list", Restore_List);
1119 DataManager::SetValue("tw_restore_selected", Restore_List);
Dees_Troy51a0e822012-09-05 15:24:24 -04001120 return;
1121}
1122
1123int TWPartitionManager::Wipe_By_Path(string Path) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001124 std::vector<TWPartition*>::iterator iter;
1125 int ret = false;
1126 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -04001127 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy63c8df72012-09-10 14:02:05 -04001128
1129 // Iterate through all partitions
1130 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -04001131 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troye58d5262012-09-21 12:27:57 -04001132 if (Path == "/and-sec")
1133 ret = (*iter)->Wipe_AndSec();
1134 else
1135 ret = (*iter)->Wipe();
Dees_Troy63c8df72012-09-10 14:02:05 -04001136 found = true;
1137 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1138 (*iter)->Wipe();
1139 }
1140 }
1141 if (found) {
1142 return ret;
1143 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001144 gui_msg(Msg(msg::kError, "unable_find_part_path=Unable to find partition for path '{1}'")(Local_Path));
Dees_Troy63c8df72012-09-10 14:02:05 -04001145 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -04001146}
1147
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001148int TWPartitionManager::Wipe_By_Path(string Path, string New_File_System) {
1149 std::vector<TWPartition*>::iterator iter;
1150 int ret = false;
1151 bool found = false;
1152 string Local_Path = TWFunc::Get_Root_Path(Path);
1153
1154 // Iterate through all partitions
1155 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1156 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1157 if (Path == "/and-sec")
1158 ret = (*iter)->Wipe_AndSec();
1159 else
1160 ret = (*iter)->Wipe(New_File_System);
1161 found = true;
1162 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1163 (*iter)->Wipe(New_File_System);
1164 }
1165 }
1166 if (found) {
1167 return ret;
1168 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001169 gui_msg(Msg(msg::kError, "unable_find_part_path=Unable to find partition for path '{1}'")(Local_Path));
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001170 return false;
1171}
1172
Dees_Troy51a0e822012-09-05 15:24:24 -04001173int TWPartitionManager::Factory_Reset(void) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001174 std::vector<TWPartition*>::iterator iter;
1175 int ret = true;
1176
1177 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001178 if ((*iter)->Wipe_During_Factory_Reset && (*iter)->Is_Present) {
Ethan Yonker89583ef2015-08-26 09:01:59 -05001179#ifdef TW_OEM_BUILD
1180 if ((*iter)->Mount_Point == "/data") {
1181 if (!(*iter)->Wipe_Encryption())
1182 ret = false;
1183 } else {
1184#endif
1185 if (!(*iter)->Wipe())
1186 ret = false;
1187#ifdef TW_OEM_BUILD
1188 }
1189#endif
Dees_Troy094207a2012-09-26 12:00:39 -04001190 } else if ((*iter)->Has_Android_Secure) {
1191 if (!(*iter)->Wipe_AndSec())
1192 ret = false;
Dees_Troy63c8df72012-09-10 14:02:05 -04001193 }
1194 }
Ethan Yonker9fa76ba2016-06-30 14:05:43 -05001195 TWFunc::check_and_run_script("/sbin/factoryreset.sh", "Factory Reset Script");
Dees_Troy63c8df72012-09-10 14:02:05 -04001196 return ret;
Dees_Troy51a0e822012-09-05 15:24:24 -04001197}
1198
Dees_Troy38bd7602012-09-14 13:33:53 -04001199int TWPartitionManager::Wipe_Dalvik_Cache(void) {
1200 struct stat st;
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001201 vector <string> dir;
Dees_Troy38bd7602012-09-14 13:33:53 -04001202
1203 if (!Mount_By_Path("/data", true))
1204 return false;
1205
1206 if (!Mount_By_Path("/cache", true))
1207 return false;
1208
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001209 dir.push_back("/data/dalvik-cache");
1210 dir.push_back("/cache/dalvik-cache");
1211 dir.push_back("/cache/dc");
Ethan Yonker74db1572015-10-28 12:44:49 -05001212 gui_msg("wiping_dalvik=Wiping Dalvik Cache Directories...");
Dees_Troya13d74f2013-03-24 08:54:55 -05001213 for (unsigned i = 0; i < dir.size(); ++i) {
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001214 if (stat(dir.at(i).c_str(), &st) == 0) {
1215 TWFunc::removeDir(dir.at(i), false);
Ethan Yonker74db1572015-10-28 12:44:49 -05001216 gui_msg(Msg("cleaned=Cleaned: {1}...")(dir.at(i)));
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001217 }
1218 }
Dees_Troy38bd7602012-09-14 13:33:53 -04001219 TWPartition* sdext = Find_Partition_By_Path("/sd-ext");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001220 if (sdext && sdext->Is_Present && sdext->Mount(false))
1221 {
1222 if (stat("/sd-ext/dalvik-cache", &st) == 0)
1223 {
1224 TWFunc::removeDir("/sd-ext/dalvik-cache", false);
Ethan Yonker74db1572015-10-28 12:44:49 -05001225 gui_msg(Msg("cleaned=Cleaned: {1}...")("/sd-ext/dalvik-cache"));
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001226 }
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001227 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001228 gui_msg("dalvik_done=-- Dalvik Cache Directories Wipe Complete!");
Dees_Troy38bd7602012-09-14 13:33:53 -04001229 return true;
1230}
1231
1232int TWPartitionManager::Wipe_Rotate_Data(void) {
1233 if (!Mount_By_Path("/data", true))
1234 return false;
1235
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001236 unlink("/data/misc/akmd*");
1237 unlink("/data/misc/rild*");
Dees_Troy2673cec2013-04-02 20:22:16 +00001238 gui_print("Rotation data wiped.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001239 return true;
1240}
1241
1242int TWPartitionManager::Wipe_Battery_Stats(void) {
1243 struct stat st;
1244
1245 if (!Mount_By_Path("/data", true))
1246 return false;
1247
1248 if (0 != stat("/data/system/batterystats.bin", &st)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001249 gui_print("No Battery Stats Found. No Need To Wipe.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001250 } else {
1251 remove("/data/system/batterystats.bin");
Dees_Troy2673cec2013-04-02 20:22:16 +00001252 gui_print("Cleared battery stats.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001253 }
1254 return true;
1255}
1256
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001257int TWPartitionManager::Wipe_Android_Secure(void) {
1258 std::vector<TWPartition*>::iterator iter;
1259 int ret = false;
1260 bool found = false;
1261
1262 // Iterate through all partitions
1263 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1264 if ((*iter)->Has_Android_Secure) {
1265 ret = (*iter)->Wipe_AndSec();
1266 found = true;
1267 }
1268 }
1269 if (found) {
1270 return ret;
1271 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001272 gui_err("no_andsec=No android secure partitions found.");
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001273 }
1274 return false;
1275}
1276
Dees_Troy38bd7602012-09-14 13:33:53 -04001277int TWPartitionManager::Format_Data(void) {
1278 TWPartition* dat = Find_Partition_By_Path("/data");
1279
1280 if (dat != NULL) {
1281 if (!dat->UnMount(true))
1282 return false;
1283
1284 return dat->Wipe_Encryption();
1285 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001286 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001287 return false;
1288 }
1289 return false;
1290}
1291
1292int TWPartitionManager::Wipe_Media_From_Data(void) {
1293 TWPartition* dat = Find_Partition_By_Path("/data");
1294
1295 if (dat != NULL) {
1296 if (!dat->Has_Data_Media) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001297 LOGERR("This device does not have /data/media\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001298 return false;
1299 }
1300 if (!dat->Mount(true))
1301 return false;
1302
Ethan Yonker74db1572015-10-28 12:44:49 -05001303 gui_msg("wiping_datamedia=Wiping internal storage -- /data/media...");
Ethan Yonker726a0202014-12-16 20:01:38 -06001304 Remove_MTP_Storage(dat->MTP_Storage_ID);
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001305 TWFunc::removeDir("/data/media", false);
xiaolu9416f4f2015-06-04 08:22:23 +08001306 dat->Recreate_Media_Folder();
Ethan Yonker726a0202014-12-16 20:01:38 -06001307 Add_MTP_Storage(dat->MTP_Storage_ID);
Dees_Troy38bd7602012-09-14 13:33:53 -04001308 return true;
1309 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001310 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001311 return false;
1312 }
1313 return false;
1314}
1315
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001316int TWPartitionManager::Repair_By_Path(string Path, bool Display_Error) {
1317 std::vector<TWPartition*>::iterator iter;
1318 int ret = false;
1319 bool found = false;
1320 string Local_Path = TWFunc::Get_Root_Path(Path);
1321
1322 if (Local_Path == "/tmp" || Local_Path == "/")
1323 return true;
1324
1325 // Iterate through all partitions
1326 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1327 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1328 ret = (*iter)->Repair();
1329 found = true;
1330 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1331 (*iter)->Repair();
1332 }
1333 }
1334 if (found) {
1335 return ret;
1336 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001337 gui_msg(Msg(msg::kError, "unable_find_part_path=Unable to find partition for path '{1}'")(Local_Path));
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001338 } else {
1339 LOGINFO("Repair: Unable to find partition for path '%s'\n", Local_Path.c_str());
1340 }
1341 return false;
1342}
1343
Ethan Yonkera2719152015-05-28 09:44:41 -05001344int TWPartitionManager::Resize_By_Path(string Path, bool Display_Error) {
1345 std::vector<TWPartition*>::iterator iter;
1346 int ret = false;
1347 bool found = false;
1348 string Local_Path = TWFunc::Get_Root_Path(Path);
1349
1350 if (Local_Path == "/tmp" || Local_Path == "/")
1351 return true;
1352
1353 // Iterate through all partitions
1354 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1355 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1356 ret = (*iter)->Resize();
1357 found = true;
1358 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1359 (*iter)->Resize();
1360 }
1361 }
1362 if (found) {
1363 return ret;
1364 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001365 gui_msg(Msg(msg::kError, "unable_find_part_path=Unable to find partition for path '{1}'")(Local_Path));
Ethan Yonkera2719152015-05-28 09:44:41 -05001366 } else {
1367 LOGINFO("Resize: Unable to find partition for path '%s'\n", Local_Path.c_str());
1368 }
1369 return false;
1370}
1371
Dees_Troy51a0e822012-09-05 15:24:24 -04001372void TWPartitionManager::Update_System_Details(void) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001373 std::vector<TWPartition*>::iterator iter;
Dees_Troy51127312012-09-08 13:08:49 -04001374 int data_size = 0;
Dees_Troy5bf43922012-09-07 16:07:55 -04001375
Ethan Yonker74db1572015-10-28 12:44:49 -05001376 gui_msg("update_part_details=Updating partition details...");
Dees_Troy5bf43922012-09-07 16:07:55 -04001377 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy51127312012-09-08 13:08:49 -04001378 if ((*iter)->Can_Be_Mounted) {
1379 (*iter)->Update_Size(true);
1380 if ((*iter)->Mount_Point == "/system") {
1381 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1382 DataManager::SetValue(TW_BACKUP_SYSTEM_SIZE, backup_display_size);
1383 } else if ((*iter)->Mount_Point == "/data" || (*iter)->Mount_Point == "/datadata") {
1384 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
1385 } else if ((*iter)->Mount_Point == "/cache") {
1386 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1387 DataManager::SetValue(TW_BACKUP_CACHE_SIZE, backup_display_size);
1388 } else if ((*iter)->Mount_Point == "/sd-ext") {
1389 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1390 DataManager::SetValue(TW_BACKUP_SDEXT_SIZE, backup_display_size);
1391 if ((*iter)->Backup_Size == 0) {
1392 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 0);
1393 DataManager::SetValue(TW_BACKUP_SDEXT_VAR, 0);
1394 } else
1395 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 1);
Dees_Troye58d5262012-09-21 12:27:57 -04001396 } else if ((*iter)->Has_Android_Secure) {
Dees_Troy8170a922012-09-18 15:40:25 -04001397 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troye58d5262012-09-21 12:27:57 -04001398 DataManager::SetValue(TW_BACKUP_ANDSEC_SIZE, backup_display_size);
Dees_Troy8170a922012-09-18 15:40:25 -04001399 if ((*iter)->Backup_Size == 0) {
1400 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 0);
1401 DataManager::SetValue(TW_BACKUP_ANDSEC_VAR, 0);
1402 } else
1403 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 1);
Dees_Troy2c50e182012-09-26 20:05:28 -04001404 } else if ((*iter)->Mount_Point == "/boot") {
1405 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1406 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1407 if ((*iter)->Backup_Size == 0) {
1408 DataManager::SetValue("tw_has_boot_partition", 0);
1409 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1410 } else
1411 DataManager::SetValue("tw_has_boot_partition", 1);
Dees_Troy51127312012-09-08 13:08:49 -04001412 }
Dees_Troyaa9cc402012-10-13 12:14:05 -04001413 } else {
1414 // Handle unmountable partitions in case we reset defaults
1415 if ((*iter)->Mount_Point == "/boot") {
1416 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1417 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1418 if ((*iter)->Backup_Size == 0) {
1419 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 0);
1420 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1421 } else
1422 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 1);
1423 } else if ((*iter)->Mount_Point == "/recovery") {
1424 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1425 DataManager::SetValue(TW_BACKUP_RECOVERY_SIZE, backup_display_size);
1426 if ((*iter)->Backup_Size == 0) {
1427 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 0);
1428 DataManager::SetValue(TW_BACKUP_RECOVERY_VAR, 0);
1429 } else
1430 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 1);
Gary Peck82599a82012-11-21 16:23:12 -08001431 } else if ((*iter)->Mount_Point == "/data") {
1432 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troyaa9cc402012-10-13 12:14:05 -04001433 }
Dees_Troy51127312012-09-08 13:08:49 -04001434 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001435 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001436 gui_msg("update_part_details_done=...done");
Dees_Troy51127312012-09-08 13:08:49 -04001437 DataManager::SetValue(TW_BACKUP_DATA_SIZE, data_size);
1438 string current_storage_path = DataManager::GetCurrentStoragePath();
1439 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
Dees_Troy8170a922012-09-18 15:40:25 -04001440 if (FreeStorage != NULL) {
1441 // Attempt to mount storage
1442 if (!FreeStorage->Mount(false)) {
Matt Mower2d50cad2015-12-03 22:26:55 -06001443 gui_msg(Msg(msg::kError, "unable_to_mount_storage=Unable to mount storage"));
1444 DataManager::SetValue(TW_STORAGE_FREE_SIZE, 0);
Dees_Troy8170a922012-09-18 15:40:25 -04001445 } else {
1446 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
1447 }
1448 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001449 LOGINFO("Unable to find storage partition '%s'.\n", current_storage_path.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -04001450 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001451 if (!Write_Fstab())
Dees_Troy2673cec2013-04-02 20:22:16 +00001452 LOGERR("Error creating fstab\n");
Dees_Troy51a0e822012-09-05 15:24:24 -04001453 return;
1454}
1455
1456int TWPartitionManager::Decrypt_Device(string Password) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001457#ifdef TW_INCLUDE_CRYPTO
1458 int ret_val, password_len;
Ethan Yonkera1de1492015-11-04 11:58:27 -06001459 char crypto_state[PROPERTY_VALUE_MAX], crypto_blkdev[PROPERTY_VALUE_MAX], cPassword[255];
Dees_Troy5bf43922012-09-07 16:07:55 -04001460 size_t result;
Ethan Yonker253368a2014-11-25 15:00:52 -06001461 std::vector<TWPartition*>::iterator iter;
Dees_Troy5bf43922012-09-07 16:07:55 -04001462
Ethan Yonker253368a2014-11-25 15:00:52 -06001463 // Mount any partitions that need to be mounted for decrypt
1464 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1465 if ((*iter)->Mount_To_Decrypt) {
1466 (*iter)->Mount(true);
1467 }
a39552696ff55ce2013-01-08 16:14:56 +00001468 }
a39552696ff55ce2013-01-08 16:14:56 +00001469
Ethan Yonkera1de1492015-11-04 11:58:27 -06001470 property_get("ro.crypto.state", crypto_state, "error");
1471 if (strcmp(crypto_state, "error") == 0) {
1472 property_set("ro.crypto.state", "encrypted");
1473 // Sleep for a bit so that services can start if needed
1474 sleep(1);
1475 }
1476
Dees_Troy5bf43922012-09-07 16:07:55 -04001477 strcpy(cPassword, Password.c_str());
a39552696ff55ce2013-01-08 16:14:56 +00001478 int pwret = cryptfs_check_passwd(cPassword);
1479
Ethan Yonker253368a2014-11-25 15:00:52 -06001480 // Unmount any partitions that were needed for decrypt
1481 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1482 if ((*iter)->Mount_To_Decrypt) {
1483 (*iter)->UnMount(false);
1484 }
1485 }
1486
a39552696ff55ce2013-01-08 16:14:56 +00001487 if (pwret != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001488 gui_err("fail_decrypt=Failed to decrypt data.");
Dees_Troy5bf43922012-09-07 16:07:55 -04001489 return -1;
1490 }
a39552696ff55ce2013-01-08 16:14:56 +00001491
Dees_Troy5bf43922012-09-07 16:07:55 -04001492 property_get("ro.crypto.fs_crypto_blkdev", crypto_blkdev, "error");
1493 if (strcmp(crypto_blkdev, "error") == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001494 LOGERR("Error retrieving decrypted data block device.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04001495 } else {
1496 TWPartition* dat = Find_Partition_By_Path("/data");
1497 if (dat != NULL) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001498 DataManager::SetValue(TW_IS_DECRYPTED, 1);
1499 dat->Is_Decrypted = true;
1500 dat->Decrypted_Block_Device = crypto_blkdev;
Gary Peck82599a82012-11-21 16:23:12 -08001501 dat->Setup_File_System(false);
Dees_Troy74fb2e92013-04-15 14:35:47 +00001502 dat->Current_File_System = dat->Fstab_File_System; // Needed if we're ignoring blkid because encrypted devices start out as emmc
Matt Mower3c366972015-12-25 19:28:31 -06001503 gui_msg(Msg("decrypt_success_dev=Data successfully decrypted, new block device: '{1}'")(crypto_blkdev));
a39552696ff55ce2013-01-08 16:14:56 +00001504
Dees_Troy5bf43922012-09-07 16:07:55 -04001505 // Sleep for a bit so that the device will be ready
1506 sleep(1);
Ethan Yonker6277c792014-09-15 14:54:30 -05001507 if (dat->Has_Data_Media && dat->Mount(false) && TWFunc::Path_Exists("/data/media/0")) {
Dees_Troy16b74352012-11-14 22:27:31 +00001508 dat->Storage_Path = "/data/media/0";
1509 dat->Symlink_Path = dat->Storage_Path;
Dees Troy98fb46c2013-12-04 16:56:45 +00001510 DataManager::SetValue("tw_storage_path", "/data/media/0");
Ethan Yonkerec0fa4a2014-11-20 09:51:03 -06001511 DataManager::SetValue("tw_settings_path", "/data/media/0");
Dees_Troy16b74352012-11-14 22:27:31 +00001512 dat->UnMount(false);
Dees_Troydc8bc1b2013-01-17 01:39:28 +00001513 Output_Partition(dat);
Dees_Troy16b74352012-11-14 22:27:31 +00001514 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001515 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001516 UnMount_Main_Partitions();
Dees_Troy5bf43922012-09-07 16:07:55 -04001517 } else
Dees_Troy2673cec2013-04-02 20:22:16 +00001518 LOGERR("Unable to locate data partition.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04001519 }
1520 return 0;
1521#else
Ethan Yonker74db1572015-10-28 12:44:49 -05001522 gui_err("no_crypto_support=No crypto support was compiled into this build.");
Dees_Troy5bf43922012-09-07 16:07:55 -04001523 return -1;
1524#endif
Dees_Troy51a0e822012-09-05 15:24:24 -04001525 return 1;
Dees_Troy51127312012-09-08 13:08:49 -04001526}
1527
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001528int TWPartitionManager::Fix_Contexts(void) {
thata3d31fb2014-12-21 22:27:40 +01001529#ifdef HAVE_SELINUX
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001530 std::vector<TWPartition*>::iterator iter;
1531 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1532 if ((*iter)->Has_Data_Media) {
1533 if ((*iter)->Mount(true)) {
1534 if (fixContexts::fixDataMediaContexts((*iter)->Mount_Point) != 0)
1535 return -1;
1536 }
1537 }
1538 }
Dees_Troy1a650e62012-10-19 20:54:32 -04001539 UnMount_Main_Partitions();
Ethan Yonker74db1572015-10-28 12:44:49 -05001540 gui_msg("done=Done.");
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001541 return 0;
1542#else
1543 LOGERR("Cannot fix contexts, no selinux support present.\n");
1544 return -1;
1545#endif
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04001546}
1547
Ethan Yonker66a19492015-12-10 10:19:45 -06001548TWPartition* TWPartitionManager::Find_Next_Storage(string Path, bool Exclude_Data_Media) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001549 std::vector<TWPartition*>::iterator iter = Partitions.begin();
1550
1551 if (!Path.empty()) {
1552 string Search_Path = TWFunc::Get_Root_Path(Path);
1553 for (; iter != Partitions.end(); iter++) {
Matt Mower9a561dd2016-02-22 21:25:51 -06001554 if ((*iter)->Mount_Point == Search_Path) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001555 iter++;
1556 break;
1557 }
1558 }
1559 }
1560
1561 for (; iter != Partitions.end(); iter++) {
Ethan Yonker66a19492015-12-10 10:19:45 -06001562 if (Exclude_Data_Media && (*iter)->Has_Data_Media) {
1563 // do nothing, do not return this type of partition
1564 } else if ((*iter)->Is_Storage && (*iter)->Is_Present) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001565 return (*iter);
1566 }
1567 }
1568
1569 return NULL;
1570}
1571
Dees_Troyd21618c2012-10-14 18:48:49 -04001572int TWPartitionManager::Open_Lun_File(string Partition_Path, string Lun_File) {
Dees_Troyd21618c2012-10-14 18:48:49 -04001573 TWPartition* Part = Find_Partition_By_Path(Partition_Path);
1574
1575 if (Part == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001576 LOGINFO("Unable to locate '%s' for USB storage mode.", Partition_Path.c_str());
1577 gui_msg(Msg(msg::kError, "unable_find_part_path=Unable to find partition for path '{1}'")(Partition_Path));
Dees_Troyd21618c2012-10-14 18:48:49 -04001578 return false;
1579 }
Ethan Yonker47360be2014-04-01 10:34:34 -05001580 LOGINFO("USB mount '%s', '%s' > '%s'\n", Partition_Path.c_str(), Part->Actual_Block_Device.c_str(), Lun_File.c_str());
1581 if (!Part->UnMount(true) || !Part->Is_Present)
Dees_Troyd21618c2012-10-14 18:48:49 -04001582 return false;
1583
Dees_Troy2673cec2013-04-02 20:22:16 +00001584 if (TWFunc::write_file(Lun_File, Part->Actual_Block_Device)) {
1585 LOGERR("Unable to write to ums lunfile '%s': (%s)\n", Lun_File.c_str(), strerror(errno));
Dees_Troyd21618c2012-10-14 18:48:49 -04001586 return false;
1587 }
Dees_Troyd21618c2012-10-14 18:48:49 -04001588 return true;
1589}
1590
Dees_Troy8170a922012-09-18 15:40:25 -04001591int TWPartitionManager::usb_storage_enable(void) {
Dees_Troyd21618c2012-10-14 18:48:49 -04001592 int has_dual, has_data_media;
Dees_Troy8170a922012-09-18 15:40:25 -04001593 char lun_file[255];
Dees_Troyd21618c2012-10-14 18:48:49 -04001594 bool has_multiple_lun = false;
Dees_Troy8170a922012-09-18 15:40:25 -04001595
Dees_Troy8170a922012-09-18 15:40:25 -04001596 DataManager::GetValue(TW_HAS_DATA_MEDIA, has_data_media);
Ethan Yonker47360be2014-04-01 10:34:34 -05001597 string Lun_File_str = CUSTOM_LUN_FILE;
1598 size_t found = Lun_File_str.find("%");
1599 if (found != string::npos) {
1600 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
1601 if (TWFunc::Path_Exists(lun_file))
1602 has_multiple_lun = true;
1603 }
Ethan Yonker726a0202014-12-16 20:01:38 -06001604 mtp_was_enabled = TWFunc::Toggle_MTP(false); // Must disable MTP for USB Storage
Ethan Yonker47360be2014-04-01 10:34:34 -05001605 if (!has_multiple_lun) {
1606 LOGINFO("Device doesn't have multiple lun files, mount current storage\n");
1607 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
1608 if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) == "/data") {
Ethan Yonker66a19492015-12-10 10:19:45 -06001609 TWPartition* Mount = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05001610 if (Mount) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001611 if (!Open_Lun_File(Mount->Mount_Point, lun_file)) {
1612 goto error_handle;
1613 }
Ethan Yonker47360be2014-04-01 10:34:34 -05001614 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001615 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001616 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05001617 }
1618 } else if (!Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file)) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001619 goto error_handle;
Dees_Troy8170a922012-09-18 15:40:25 -04001620 }
Dees_Troy8170a922012-09-18 15:40:25 -04001621 } else {
Ethan Yonker47360be2014-04-01 10:34:34 -05001622 LOGINFO("Device has multiple lun files\n");
1623 TWPartition* Mount1;
1624 TWPartition* Mount2;
Dees_Troy8170a922012-09-18 15:40:25 -04001625 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
Ethan Yonker66a19492015-12-10 10:19:45 -06001626 Mount1 = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05001627 if (Mount1) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001628 if (!Open_Lun_File(Mount1->Mount_Point, lun_file)) {
1629 goto error_handle;
1630 }
Matt Moweree717062014-04-26 01:46:46 -05001631 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
Ethan Yonker66a19492015-12-10 10:19:45 -06001632 Mount2 = Find_Next_Storage(Mount1->Mount_Point, true);
Matt Mower1fdcdb72016-01-25 20:53:47 -06001633 if (Mount2 && Mount2->Mount_Point != Mount1->Mount_Point) {
Matt Moweree717062014-04-26 01:46:46 -05001634 Open_Lun_File(Mount2->Mount_Point, lun_file);
Ethan Yonker47360be2014-04-01 10:34:34 -05001635 }
1636 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001637 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001638 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05001639 }
Dees_Troy8170a922012-09-18 15:40:25 -04001640 }
Matt Mowerb6ef4782014-11-06 02:24:36 -06001641 property_set("sys.storage.ums_enabled", "1");
HandyMenny37d42992014-10-26 22:15:59 +01001642 property_set("sys.usb.config", "mass_storage,adb");
Dees_Troy8170a922012-09-18 15:40:25 -04001643 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001644error_handle:
1645 if (mtp_was_enabled)
1646 if (!Enable_MTP())
1647 Disable_MTP();
1648 return false;
Dees_Troy8170a922012-09-18 15:40:25 -04001649}
1650
1651int TWPartitionManager::usb_storage_disable(void) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001652 int index, ret;
1653 char lun_file[255], ch[2] = {0, 0};
1654 string str = ch;
Dees_Troy8170a922012-09-18 15:40:25 -04001655
1656 for (index=0; index<2; index++) {
1657 sprintf(lun_file, CUSTOM_LUN_FILE, index);
Dees_Troy2673cec2013-04-02 20:22:16 +00001658 ret = TWFunc::write_file(lun_file, str);
Dees_Troy2673cec2013-04-02 20:22:16 +00001659 if (ret < 0) {
1660 break;
Dees_Troy8170a922012-09-18 15:40:25 -04001661 }
Dees_Troy8170a922012-09-18 15:40:25 -04001662 }
Dees_Troye58d5262012-09-21 12:27:57 -04001663 Mount_All_Storage();
1664 Update_System_Details();
Dees_Troycfd73ef2012-10-12 16:52:00 -04001665 UnMount_Main_Partitions();
Matt Mowerd9cb9062013-12-14 20:34:45 -06001666 property_set("sys.storage.ums_enabled", "0");
HandyMenny37d42992014-10-26 22:15:59 +01001667 property_set("sys.usb.config", "adb");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001668 if (mtp_was_enabled)
1669 if (!Enable_MTP())
1670 Disable_MTP();
Dees_Troy2673cec2013-04-02 20:22:16 +00001671 if (ret < 0 && index == 0) {
1672 LOGERR("Unable to write to ums lunfile '%s'.", lun_file);
1673 return false;
1674 } else {
1675 return true;
1676 }
Dees_Troy8170a922012-09-18 15:40:25 -04001677 return true;
Dees_Troy812660f2012-09-20 09:55:17 -04001678}
1679
1680void TWPartitionManager::Mount_All_Storage(void) {
1681 std::vector<TWPartition*>::iterator iter;
1682
1683 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1684 if ((*iter)->Is_Storage)
1685 (*iter)->Mount(false);
1686 }
Dees_Troy2c50e182012-09-26 20:05:28 -04001687}
Dees_Troy9350b8d2012-09-27 12:38:38 -04001688
Dees_Troyd0384ef2012-10-12 12:15:42 -04001689void TWPartitionManager::UnMount_Main_Partitions(void) {
1690 // Unmounts system and data if data is not data/media
1691 // Also unmounts boot if boot is mountable
Dees_Troy2673cec2013-04-02 20:22:16 +00001692 LOGINFO("Unmounting main partitions...\n");
Dees_Troyd0384ef2012-10-12 12:15:42 -04001693
1694 TWPartition* Boot_Partition = Find_Partition_By_Path("/boot");
1695
1696 UnMount_By_Path("/system", true);
Ethan Yonker6277c792014-09-15 14:54:30 -05001697 if (!datamedia)
1698 UnMount_By_Path("/data", true);
1699
Dees_Troyd0384ef2012-10-12 12:15:42 -04001700 if (Boot_Partition != NULL && Boot_Partition->Can_Be_Mounted)
1701 Boot_Partition->UnMount(true);
1702}
1703
Dees_Troy9350b8d2012-09-27 12:38:38 -04001704int TWPartitionManager::Partition_SDCard(void) {
1705 char mkdir_path[255], temp[255], line[512];
Ethan Yonker483e9f42016-01-11 22:21:18 -06001706 string Storage_Path, Command, Device, fat_str, ext_str, start_loc, end_loc, ext_format, sd_path, tmpdevice;
Dees_Troy9350b8d2012-09-27 12:38:38 -04001707 int ext, swap, total_size = 0, fat_size;
Dees_Troy9350b8d2012-09-27 12:38:38 -04001708
Ethan Yonker74db1572015-10-28 12:44:49 -05001709 gui_msg("start_partition_sd=Partitioning SD Card...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06001710
1711 // Locate and validate device to partition
1712 TWPartition* SDCard = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
1713
Ethan Yonker66a19492015-12-10 10:19:45 -06001714 if (SDCard->Is_Adopted_Storage)
1715 SDCard->Revert_Adopted();
1716
Ethan Yonker1eff6cd2014-09-15 13:30:42 -05001717 if (SDCard == NULL || !SDCard->Removable || SDCard->Has_Data_Media) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001718 gui_err("partition_sd_locate=Unable to locate device to partition.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001719 return false;
1720 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001721
1722 // Unmount everything
Dees_Troy9350b8d2012-09-27 12:38:38 -04001723 if (!SDCard->UnMount(true))
1724 return false;
1725 TWPartition* SDext = Find_Partition_By_Path("/sd-ext");
1726 if (SDext != NULL) {
1727 if (!SDext->UnMount(true))
1728 return false;
1729 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001730 char* swappath = getenv("SWAPPATH");
1731 if (swappath != NULL) {
1732 LOGINFO("Unmounting swap at '%s'\n", swappath);
1733 umount(swappath);
1734 }
Vojtech Bocek05534202013-09-11 08:11:56 +02001735
Ethan Yonker483e9f42016-01-11 22:21:18 -06001736 // Determine block device
1737 if (SDCard->Alternate_Block_Device.empty()) {
1738 SDCard->Find_Actual_Block_Device();
1739 Device = SDCard->Actual_Block_Device;
1740 // Just use the root block device
1741 Device.resize(strlen("/dev/block/mmcblkX"));
1742 } else {
1743 Device = SDCard->Alternate_Block_Device;
1744 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04001745
1746 // Find the size of the block device:
Ethan Yonker483e9f42016-01-11 22:21:18 -06001747 total_size = (int)(TWFunc::IOCTL_Get_Block_Size(Device.c_str()) / (1048576));
Dees_Troy9350b8d2012-09-27 12:38:38 -04001748
1749 DataManager::GetValue("tw_sdext_size", ext);
1750 DataManager::GetValue("tw_swap_size", swap);
1751 DataManager::GetValue("tw_sdpart_file_system", ext_format);
1752 fat_size = total_size - ext - swap;
Ethan Yonker483e9f42016-01-11 22:21:18 -06001753 LOGINFO("sd card mount point %s block device is '%s', sdcard size is: %iMB, fat size: %iMB, ext size: %iMB, ext system: '%s', swap size: %iMB\n", DataManager::GetCurrentStoragePath().c_str(), Device.c_str(), total_size, fat_size, ext, ext_format.c_str(), swap);
1754
1755 // Determine partition sizes
1756 if (swap == 0 && ext == 0) {
1757 fat_str = "-0";
1758 } else {
1759 memset(temp, 0, sizeof(temp));
1760 sprintf(temp, "%i", fat_size);
1761 fat_str = temp;
1762 fat_str += "MB";
1763 }
1764 if (swap == 0) {
1765 ext_str = "-0";
1766 } else {
1767 memset(temp, 0, sizeof(temp));
1768 sprintf(temp, "%i", ext);
1769 ext_str = "+";
1770 ext_str += temp;
1771 ext_str += "MB";
1772 }
1773
Dees_Troy9350b8d2012-09-27 12:38:38 -04001774 if (ext + swap > total_size) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001775 gui_err("ext_swap_size=EXT + Swap size is larger than sdcard size.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001776 return false;
1777 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001778
Ethan Yonker74db1572015-10-28 12:44:49 -05001779 gui_msg("remove_part_table=Removing partition table...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06001780 Command = "sgdisk --zap-all " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00001781 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001782 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001783 gui_err("unable_rm_part=Unable to remove partition table.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001784 Update_System_Details();
1785 return false;
1786 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001787 gui_msg(Msg("create_part=Creating {1} partition...")("FAT32"));
Captain Throwback9643a802016-05-27 11:44:51 -04001788 Command = "sgdisk --new=0:0:" + fat_str + " --change-name=0:\"Microsoft basic data\" --typecode=0:EBD0A0A2-B9E5-4433-87C0-68B6B72699C7 " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00001789 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001790 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001791 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("FAT32"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04001792 return false;
1793 }
1794 if (ext > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001795 gui_msg(Msg("create_part=Creating {1} partition...")("EXT"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06001796 Command = "sgdisk --new=0:0:" + ext_str + " --change-name=0:\"Linux filesystem\" " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00001797 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001798 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001799 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("EXT"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04001800 Update_System_Details();
1801 return false;
1802 }
1803 }
1804 if (swap > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001805 gui_msg(Msg("create_part=Creating {1} partition...")("swap"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06001806 Command = "sgdisk --new=0:0:-0 --change-name=0:\"Linux swap\" --typecode=0:0657FD6D-A4AB-43C4-84E5-0933C84B4F4F " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00001807 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001808 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001809 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("swap"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04001810 Update_System_Details();
1811 return false;
1812 }
1813 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001814
1815 // Convert GPT to MBR
1816 Command = "sgdisk --gpttombr " + Device;
1817 if (TWFunc::Exec_Cmd(Command) != 0)
1818 LOGINFO("Failed to covert partition GPT to MBR\n");
1819
1820 // Tell the kernel to rescan the partition table
1821 int fd = open(Device.c_str(), O_RDONLY);
1822 ioctl(fd, BLKRRPART, 0);
1823 close(fd);
1824
1825 string format_device = Device;
1826 if (Device.substr(0, 17) == "/dev/block/mmcblk")
1827 format_device += "p";
1828
1829 // Format new partitions to proper file system
1830 if (fat_size > 0) {
1831 Command = "mkfs.fat " + format_device + "1";
1832 TWFunc::Exec_Cmd(Command);
1833 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04001834 if (ext > 0) {
1835 if (SDext == NULL) {
Ethan Yonker483e9f42016-01-11 22:21:18 -06001836 Command = "mke2fs -t " + ext_format + " -m 0 " + format_device + "2";
1837 gui_msg(Msg("format_sdext_as=Formatting sd-ext as {1}...")(ext_format));
1838 LOGINFO("Formatting sd-ext after partitioning, command: '%s'\n", Command.c_str());
1839 TWFunc::Exec_Cmd(Command);
1840 } else {
1841 SDext->Wipe(ext_format);
Dees_Troy9350b8d2012-09-27 12:38:38 -04001842 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001843 }
1844 if (swap > 0) {
1845 Command = "mkswap " + format_device;
1846 if (ext > 0)
1847 Command += "3";
1848 else
1849 Command += "2";
Vojtech Bocek05534202013-09-11 08:11:56 +02001850 TWFunc::Exec_Cmd(Command);
Dees_Troy9350b8d2012-09-27 12:38:38 -04001851 }
1852
Ethan Yonker483e9f42016-01-11 22:21:18 -06001853 // recreate TWRP folder and rewrite settings - these will be gone after sdcard is partitioned
1854 if (SDCard->Mount(true)) {
1855 string TWRP_Folder = SDCard->Mount_Point + "/TWRP";
1856 mkdir(TWRP_Folder.c_str(), 0777);
1857 DataManager::Flush();
1858 }
1859
Dees_Troy9350b8d2012-09-27 12:38:38 -04001860 Update_System_Details();
Ethan Yonker74db1572015-10-28 12:44:49 -05001861 gui_msg("part_complete=Partitioning complete.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001862 return true;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04001863}
Dees_Troya13d74f2013-03-24 08:54:55 -05001864
1865void TWPartitionManager::Get_Partition_List(string ListType, std::vector<PartitionList> *Partition_List) {
1866 std::vector<TWPartition*>::iterator iter;
1867 if (ListType == "mount") {
1868 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1869 if ((*iter)->Can_Be_Mounted && !(*iter)->Is_SubPartition) {
1870 struct PartitionList part;
1871 part.Display_Name = (*iter)->Display_Name;
1872 part.Mount_Point = (*iter)->Mount_Point;
1873 part.selected = (*iter)->Is_Mounted();
1874 Partition_List->push_back(part);
1875 }
1876 }
1877 } else if (ListType == "storage") {
1878 char free_space[255];
1879 string Current_Storage = DataManager::GetCurrentStoragePath();
1880 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1881 if ((*iter)->Is_Storage) {
1882 struct PartitionList part;
1883 sprintf(free_space, "%llu", (*iter)->Free / 1024 / 1024);
1884 part.Display_Name = (*iter)->Storage_Name + " (";
1885 part.Display_Name += free_space;
1886 part.Display_Name += "MB)";
1887 part.Mount_Point = (*iter)->Storage_Path;
1888 if ((*iter)->Storage_Path == Current_Storage)
1889 part.selected = 1;
1890 else
1891 part.selected = 0;
1892 Partition_List->push_back(part);
1893 }
1894 }
1895 } else if (ListType == "backup") {
1896 char backup_size[255];
Dees_Troy9e0b71c2013-04-08 13:35:37 +00001897 unsigned long long Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -05001898 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy9e0b71c2013-04-08 13:35:37 +00001899 if ((*iter)->Can_Be_Backed_Up && !(*iter)->Is_SubPartition && (*iter)->Is_Present) {
Dees_Troya13d74f2013-03-24 08:54:55 -05001900 struct PartitionList part;
Dees_Troy9e0b71c2013-04-08 13:35:37 +00001901 Backup_Size = (*iter)->Backup_Size;
1902 if ((*iter)->Has_SubPartition) {
1903 std::vector<TWPartition*>::iterator subpart;
1904
1905 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
1906 if ((*subpart)->Is_SubPartition && (*subpart)->Can_Be_Backed_Up && (*subpart)->Is_Present && (*subpart)->SubPartition_Of == (*iter)->Mount_Point)
1907 Backup_Size += (*subpart)->Backup_Size;
1908 }
1909 }
1910 sprintf(backup_size, "%llu", Backup_Size / 1024 / 1024);
Dees_Troya13d74f2013-03-24 08:54:55 -05001911 part.Display_Name = (*iter)->Backup_Display_Name + " (";
1912 part.Display_Name += backup_size;
1913 part.Display_Name += "MB)";
1914 part.Mount_Point = (*iter)->Backup_Path;
1915 part.selected = 0;
1916 Partition_List->push_back(part);
1917 }
1918 }
1919 } else if (ListType == "restore") {
1920 string Restore_List, restore_path;
1921 TWPartition* restore_part = NULL;
1922
1923 DataManager::GetValue("tw_restore_list", Restore_List);
1924 if (!Restore_List.empty()) {
1925 size_t start_pos = 0, end_pos = Restore_List.find(";", start_pos);
1926 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1927 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
Dees_Troy59df9262013-06-19 14:53:57 -05001928 if ((restore_part = Find_Partition_By_Path(restore_path)) != NULL) {
Dees Troy4159aed2014-02-28 17:24:43 +00001929 if ((restore_part->Backup_Name == "recovery" && !restore_part->Can_Be_Backed_Up) || restore_part->Is_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -05001930 // Don't allow restore of recovery (causes problems on some devices)
Dees_Troy59df9262013-06-19 14:53:57 -05001931 // Don't add subpartitions to the list of items
Dees_Troya13d74f2013-03-24 08:54:55 -05001932 } else {
1933 struct PartitionList part;
1934 part.Display_Name = restore_part->Backup_Display_Name;
1935 part.Mount_Point = restore_part->Backup_Path;
1936 part.selected = 1;
1937 Partition_List->push_back(part);
1938 }
1939 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001940 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001941 }
1942 start_pos = end_pos + 1;
1943 end_pos = Restore_List.find(";", start_pos);
1944 }
1945 }
1946 } else if (ListType == "wipe") {
1947 struct PartitionList dalvik;
Ethan Yonker74db1572015-10-28 12:44:49 -05001948 dalvik.Display_Name = gui_parse_text("{@dalvik}");
Dees_Troya13d74f2013-03-24 08:54:55 -05001949 dalvik.Mount_Point = "DALVIK";
1950 dalvik.selected = 0;
1951 Partition_List->push_back(dalvik);
1952 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1953 if ((*iter)->Wipe_Available_in_GUI && !(*iter)->Is_SubPartition) {
1954 struct PartitionList part;
1955 part.Display_Name = (*iter)->Display_Name;
1956 part.Mount_Point = (*iter)->Mount_Point;
1957 part.selected = 0;
1958 Partition_List->push_back(part);
1959 }
1960 if ((*iter)->Has_Android_Secure) {
1961 struct PartitionList part;
1962 part.Display_Name = (*iter)->Backup_Display_Name;
1963 part.Mount_Point = (*iter)->Backup_Path;
1964 part.selected = 0;
1965 Partition_List->push_back(part);
1966 }
Dees_Troy74fb2e92013-04-15 14:35:47 +00001967 if ((*iter)->Has_Data_Media) {
1968 struct PartitionList datamedia;
1969 datamedia.Display_Name = (*iter)->Storage_Name;
1970 datamedia.Mount_Point = "INTERNAL";
1971 datamedia.selected = 0;
1972 Partition_List->push_back(datamedia);
1973 }
Dees_Troya13d74f2013-03-24 08:54:55 -05001974 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06001975 } else if (ListType == "flashimg") {
1976 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1977 if ((*iter)->Can_Flash_Img && (*iter)->Is_Present) {
1978 struct PartitionList part;
1979 part.Display_Name = (*iter)->Backup_Display_Name;
1980 part.Mount_Point = (*iter)->Backup_Path;
1981 part.selected = 0;
1982 Partition_List->push_back(part);
1983 }
1984 }
Dees_Troya13d74f2013-03-24 08:54:55 -05001985 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001986 LOGERR("Unknown list type '%s' requested for TWPartitionManager::Get_Partition_List\n", ListType.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -05001987 }
1988}
1989
1990int TWPartitionManager::Fstab_Processed(void) {
1991 return Partitions.size();
1992}
Dees_Troyd93bda52013-07-03 19:55:19 +00001993
1994void TWPartitionManager::Output_Storage_Fstab(void) {
1995 std::vector<TWPartition*>::iterator iter;
1996 char storage_partition[255];
1997 string Temp;
1998 FILE *fp = fopen("/cache/recovery/storage.fstab", "w");
1999
2000 if (fp == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002001 gui_msg(Msg(msg::kError, "unable_to_open=Unable to open '{1}'.")("/cache/recovery/storage.fstab"));
Dees_Troyd93bda52013-07-03 19:55:19 +00002002 return;
2003 }
2004
2005 // Iterate through all partitions
2006 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2007 if ((*iter)->Is_Storage) {
2008 Temp = (*iter)->Storage_Path + ";" + (*iter)->Storage_Name + ";\n";
2009 strcpy(storage_partition, Temp.c_str());
2010 fwrite(storage_partition, sizeof(storage_partition[0]), strlen(storage_partition) / sizeof(storage_partition[0]), fp);
2011 }
2012 }
2013 fclose(fp);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02002014}
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002015
2016TWPartition *TWPartitionManager::Get_Default_Storage_Partition()
2017{
2018 TWPartition *res = NULL;
2019 for (std::vector<TWPartition*>::iterator iter = Partitions.begin(); iter != Partitions.end(); ++iter) {
2020 if(!(*iter)->Is_Storage)
2021 continue;
2022
2023 if((*iter)->Is_Settings_Storage)
2024 return *iter;
2025
2026 if(!res)
2027 res = *iter;
2028 }
2029 return res;
2030}
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002031
2032bool TWPartitionManager::Enable_MTP(void) {
2033#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002034 if (mtppid) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002035 gui_err("mtp_already_enabled=MTP already enabled");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002036 return true;
2037 }
2038 //Launch MTP Responder
2039 LOGINFO("Starting MTP\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002040
2041 int mtppipe[2];
2042
2043 if (pipe(mtppipe) < 0) {
2044 LOGERR("Error creating MTP pipe\n");
2045 return false;
2046 }
2047
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002048 char old_value[PROPERTY_VALUE_MAX];
2049 property_get("sys.usb.config", old_value, "error");
2050 if (strcmp(old_value, "error") != 0 && strcmp(old_value, "mtp,adb") != 0) {
2051 char vendor[PROPERTY_VALUE_MAX];
2052 char product[PROPERTY_VALUE_MAX];
2053 property_set("sys.usb.config", "none");
2054 property_get("usb.vendor", vendor, "18D1");
2055 property_get("usb.product.mtpadb", product, "4EE2");
2056 string vendorstr = vendor;
2057 string productstr = product;
2058 TWFunc::write_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2059 TWFunc::write_file("/sys/class/android_usb/android0/idProduct", productstr);
2060 property_set("sys.usb.config", "mtp,adb");
2061 }
Ethan Yonker6d154c42014-09-03 14:19:43 -05002062 /* To enable MTP debug, use the twrp command line feature to
2063 * twrp set tw_mtp_debug 1
2064 */
2065 twrpMtp *mtp = new twrpMtp(DataManager::GetIntValue("tw_mtp_debug"));
Ethan Yonker1b039202015-01-30 10:08:48 -06002066 mtppid = mtp->forkserver(mtppipe);
2067 if (mtppid) {
2068 close(mtppipe[0]); // Host closes read side
2069 mtp_write_fd = mtppipe[1];
2070 DataManager::SetValue("tw_mtp_enabled", 1);
2071 Add_All_MTP_Storage();
2072 return true;
Ethan Yonker726a0202014-12-16 20:01:38 -06002073 } else {
2074 close(mtppipe[0]);
2075 close(mtppipe[1]);
Ethan Yonker74db1572015-10-28 12:44:49 -05002076 gui_err("mtp_fail=Failed to enable MTP");
Ethan Yonker1b039202015-01-30 10:08:48 -06002077 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002078 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002079#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002080 gui_err("no_mtp=MTP support not included");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002081#endif
2082 DataManager::SetValue("tw_mtp_enabled", 0);
2083 return false;
2084}
2085
Ethan Yonker1b039202015-01-30 10:08:48 -06002086void TWPartitionManager::Add_All_MTP_Storage(void) {
2087#ifdef TW_HAS_MTP
2088 std::vector<TWPartition*>::iterator iter;
2089
2090 if (!mtppid)
2091 return; // MTP is not enabled
2092
2093 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2094 if ((*iter)->Is_Storage && (*iter)->Is_Present && (*iter)->Mount(false))
2095 Add_Remove_MTP_Storage((*iter), MTP_MESSAGE_ADD_STORAGE);
2096 }
2097#else
2098 return;
2099#endif
2100}
2101
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002102bool TWPartitionManager::Disable_MTP(void) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002103 char old_value[PROPERTY_VALUE_MAX];
2104 property_get("sys.usb.config", old_value, "error");
2105 if (strcmp(old_value, "adb") != 0) {
2106 char vendor[PROPERTY_VALUE_MAX];
2107 char product[PROPERTY_VALUE_MAX];
2108 property_set("sys.usb.config", "none");
2109 property_get("usb.vendor", vendor, "18D1");
2110 property_get("usb.product.adb", product, "D002");
2111 string vendorstr = vendor;
2112 string productstr = product;
2113 TWFunc::write_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2114 TWFunc::write_file("/sys/class/android_usb/android0/idProduct", productstr);
2115 usleep(2000);
2116 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002117#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002118 if (mtppid) {
Ethan Yonker8613dc02014-09-11 09:28:20 -05002119 LOGINFO("Disabling MTP\n");
2120 int status;
2121 kill(mtppid, SIGKILL);
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002122 mtppid = 0;
Ethan Yonker8613dc02014-09-11 09:28:20 -05002123 // We don't care about the exit value, but this prevents a zombie process
2124 waitpid(mtppid, &status, 0);
Ethan Yonker726a0202014-12-16 20:01:38 -06002125 close(mtp_write_fd);
2126 mtp_write_fd = -1;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002127 }
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002128#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002129 property_set("sys.usb.config", "adb");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002130#ifdef TW_HAS_MTP
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002131 DataManager::SetValue("tw_mtp_enabled", 0);
2132 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002133#endif
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002134 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002135}
Ethan Yonker726a0202014-12-16 20:01:38 -06002136
2137TWPartition* TWPartitionManager::Find_Partition_By_MTP_Storage_ID(unsigned int Storage_ID) {
2138 std::vector<TWPartition*>::iterator iter;
2139
2140 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2141 if ((*iter)->MTP_Storage_ID == Storage_ID)
2142 return (*iter);
2143 }
2144 return NULL;
2145}
2146
2147bool TWPartitionManager::Add_Remove_MTP_Storage(TWPartition* Part, int message_type) {
2148#ifdef TW_HAS_MTP
2149 struct mtpmsg mtp_message;
2150
2151 if (!mtppid)
2152 return false; // MTP is disabled
2153
2154 if (mtp_write_fd < 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002155 LOGINFO("MTP: mtp_write_fd is not set\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002156 return false;
2157 }
2158
2159 if (Part) {
Ethan Yonker4bfabab2014-12-29 09:15:37 -06002160 if (Part->MTP_Storage_ID == 0)
2161 return false;
Ethan Yonker726a0202014-12-16 20:01:38 -06002162 if (message_type == MTP_MESSAGE_REMOVE_STORAGE) {
2163 mtp_message.message_type = MTP_MESSAGE_REMOVE_STORAGE; // Remove
2164 LOGINFO("sending message to remove %i\n", Part->MTP_Storage_ID);
2165 mtp_message.storage_id = Part->MTP_Storage_ID;
2166 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002167 LOGINFO("error sending message to remove storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002168 return false;
2169 } else {
2170 LOGINFO("Message sent, remove storage ID: %i\n", Part->MTP_Storage_ID);
2171 return true;
2172 }
2173 } else if (message_type == MTP_MESSAGE_ADD_STORAGE && Part->Is_Mounted()) {
2174 mtp_message.message_type = MTP_MESSAGE_ADD_STORAGE; // Add
2175 mtp_message.storage_id = Part->MTP_Storage_ID;
2176 mtp_message.path = Part->Storage_Path.c_str();
2177 mtp_message.display = Part->Storage_Name.c_str();
2178 mtp_message.maxFileSize = Part->Get_Max_FileSize();
Ethan Yonker1b039202015-01-30 10:08:48 -06002179 LOGINFO("sending message to add %i '%s' '%s'\n", mtp_message.storage_id, mtp_message.path, mtp_message.display);
Ethan Yonker726a0202014-12-16 20:01:38 -06002180 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002181 LOGINFO("error sending message to add storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002182 return false;
2183 } else {
2184 LOGINFO("Message sent, add storage ID: %i\n", Part->MTP_Storage_ID);
2185 return true;
2186 }
2187 } else {
2188 LOGERR("Unknown MTP message type: %i\n", message_type);
2189 }
2190 } else {
2191 // This hopefully never happens as the error handling should
2192 // occur in the calling function.
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002193 LOGINFO("TWPartitionManager::Add_Remove_MTP_Storage NULL partition given\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002194 }
2195 return true;
2196#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002197 gui_err("no_mtp=MTP support not included");
Ethan Yonker726a0202014-12-16 20:01:38 -06002198 DataManager::SetValue("tw_mtp_enabled", 0);
2199 return false;
2200#endif
2201}
2202
2203bool TWPartitionManager::Add_MTP_Storage(string Mount_Point) {
2204#ifdef TW_HAS_MTP
2205 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2206 if (Part) {
2207 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2208 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002209 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002210 }
2211#endif
2212 return false;
2213}
2214
2215bool TWPartitionManager::Add_MTP_Storage(unsigned int Storage_ID) {
2216#ifdef TW_HAS_MTP
2217 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2218 if (Part) {
2219 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2220 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002221 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002222 }
2223#endif
2224 return false;
2225}
2226
2227bool TWPartitionManager::Remove_MTP_Storage(string Mount_Point) {
2228#ifdef TW_HAS_MTP
2229 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2230 if (Part) {
2231 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2232 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002233 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002234 }
2235#endif
2236 return false;
2237}
2238
2239bool TWPartitionManager::Remove_MTP_Storage(unsigned int Storage_ID) {
2240#ifdef TW_HAS_MTP
2241 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2242 if (Part) {
2243 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2244 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002245 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002246 }
2247#endif
2248 return false;
2249}
Ethan Yonker96af84a2015-01-05 14:58:36 -06002250
bigbiffce8f83c2015-12-12 18:30:21 -05002251bool TWPartitionManager::Flash_Image(PartitionSettings *part_settings) {
Ethan Yonker96af84a2015-01-05 14:58:36 -06002252 int check, partition_count = 0;
2253 TWPartition* flash_part = NULL;
bigbiffce8f83c2015-12-12 18:30:21 -05002254 string Flash_List, flash_path, full_filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002255 size_t start_pos = 0, end_pos = 0;
2256
Ethan Yonkerdcf2b672016-09-13 14:41:53 -05002257 full_filename = part_settings->Backup_Folder;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002258
bigbiffce8f83c2015-12-12 18:30:21 -05002259 gui_msg("image_flash_start=[IMAGE FLASH STARTED]");
2260 gui_msg(Msg("img_to_flash=Image to flash: '{1}'")(full_filename));
2261
2262 if (!TWFunc::Path_Exists(full_filename)) {
2263 if (!Mount_By_Path(full_filename, true)) {
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002264 return false;
2265 }
bigbiffce8f83c2015-12-12 18:30:21 -05002266 if (!TWFunc::Path_Exists(full_filename)) {
2267 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")(full_filename));
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002268 return false;
2269 }
2270 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002271
Ethan Yonker74db1572015-10-28 12:44:49 -05002272 gui_msg("calc_restore=Calculating restore details...");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002273 DataManager::GetValue("tw_flash_partition", Flash_List);
2274 if (!Flash_List.empty()) {
2275 end_pos = Flash_List.find(";", start_pos);
2276 while (end_pos != string::npos && start_pos < Flash_List.size()) {
2277 flash_path = Flash_List.substr(start_pos, end_pos - start_pos);
2278 flash_part = Find_Partition_By_Path(flash_path);
2279 if (flash_part != NULL) {
2280 partition_count++;
2281 if (partition_count > 1) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002282 gui_err("too_many_flash=Too many partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002283 return false;
2284 }
2285 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002286 gui_msg(Msg(msg::kError, "flash_unable_locate=Unable to locate '{1}' partition for flashing.")(flash_path));
Ethan Yonker96af84a2015-01-05 14:58:36 -06002287 return false;
2288 }
2289 start_pos = end_pos + 1;
2290 end_pos = Flash_List.find(";", start_pos);
2291 }
2292 }
2293
2294 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002295 gui_err("no_part_flash=No partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002296 return false;
2297 }
2298
2299 DataManager::SetProgress(0.0);
2300 if (flash_part) {
bigbiffce8f83c2015-12-12 18:30:21 -05002301 if (!flash_part->Flash_Image(part_settings))
Ethan Yonker96af84a2015-01-05 14:58:36 -06002302 return false;
2303 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002304 gui_err("invalid_flash=Invalid flash partition specified.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002305 return false;
2306 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002307 gui_highlight("flash_done=IMAGE FLASH COMPLETED]");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002308 return true;
2309}
Ethan Yonker74db1572015-10-28 12:44:49 -05002310
2311void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value) {
2312 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2313 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002314 if (part->Is_Adopted_Storage) {
2315 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2316 part->Backup_Display_Name = part->Display_Name;
2317 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2318 } else {
2319 part->Display_Name = gui_lookup(resource_name, default_value);
2320 part->Backup_Display_Name = part->Display_Name;
2321 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002322 }
2323}
2324
2325void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value, const char* storage_resource_name, const char* storage_default_value) {
2326 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2327 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002328 if (part->Is_Adopted_Storage) {
2329 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2330 part->Backup_Display_Name = part->Display_Name;
2331 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2332 } else {
2333 part->Display_Name = gui_lookup(resource_name, default_value);
2334 part->Backup_Display_Name = part->Display_Name;
2335 if (part->Is_Storage)
2336 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2337 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002338 }
2339}
2340
2341void TWPartitionManager::Translate_Partition_Display_Names() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002342 LOGINFO("Translating partition display names\n");
Ethan Yonker74db1572015-10-28 12:44:49 -05002343 Translate_Partition("/system", "system", "System");
2344 Translate_Partition("/system_image", "system_image", "System Image");
2345 Translate_Partition("/vendor", "vendor", "Vendor");
2346 Translate_Partition("/vendor_image", "vendor_image", "Vendor Image");
2347 Translate_Partition("/cache", "cache", "Cache");
2348 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage");
2349 Translate_Partition("/boot", "boot", "Boot");
2350 Translate_Partition("/recovery", "recovery", "Recovery");
2351 if (!datamedia) {
2352 Translate_Partition("/sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2353 Translate_Partition("/internal_sd", "sdcard", "SDCard", "sdcard", "SDCard");
2354 Translate_Partition("/internal_sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2355 Translate_Partition("/emmc", "sdcard", "SDCard", "sdcard", "SDCard");
2356 }
2357 Translate_Partition("/external_sd", "microsd", "Micro SDCard", "microsd", "Micro SDCard");
2358 Translate_Partition("/external_sdcard", "microsd", "Micro SDCard", "microsd", "Micro SDCard");
2359 Translate_Partition("/usb-otg", "usbotg", "USB OTG", "usbotg", "USB OTG");
2360 Translate_Partition("/sd-ext", "sdext", "SD-EXT");
2361
2362 // Android secure is a special case
2363 TWPartition* part = PartitionManager.Find_Partition_By_Path("/and-sec");
2364 if (part)
2365 part->Backup_Display_Name = gui_lookup("android_secure", "Android Secure");
2366
2367 // This updates the text on all of the storage selection buttons in the GUI
2368 DataManager::SetBackupFolder();
2369}
Ethan Yonker66a19492015-12-10 10:19:45 -06002370
2371void TWPartitionManager::Decrypt_Adopted() {
2372#ifdef TW_INCLUDE_CRYPTO
2373 if (!Mount_By_Path("/data", false)) {
2374 LOGERR("Cannot decrypt adopted storage because /data will not mount\n");
2375 return;
2376 }
2377 LOGINFO("Decrypt adopted storage starting\n");
2378 char* xmlFile = PageManager::LoadFileToBuffer("/data/system/storage.xml", NULL);
2379 xml_document<> *doc = NULL;
2380 xml_node<>* volumes = NULL;
2381 xml_node<>* volume = NULL;
2382 string Primary_Storage_UUID = "";
2383 if (xmlFile != NULL) {
2384 LOGINFO("successfully loaded storage.xml\n");
2385 doc = new xml_document<>();
2386 doc->parse<0>(xmlFile);
2387 volumes = doc->first_node("volumes");
2388 if (volumes) {
2389 xml_attribute<>* psuuid = volumes->first_attribute("primaryStorageUuid");
2390 if (psuuid) {
2391 Primary_Storage_UUID = psuuid->value();
2392 }
2393 }
2394 }
2395 std::vector<TWPartition*>::iterator adopt;
2396 for (adopt = Partitions.begin(); adopt != Partitions.end(); adopt++) {
2397 if ((*adopt)->Removable && (*adopt)->Is_Present) {
2398 if ((*adopt)->Decrypt_Adopted() == 0) {
2399 if (volumes) {
2400 xml_node<>* volume = volumes->first_node("volume");
2401 while (volume) {
2402 xml_attribute<>* guid = volume->first_attribute("partGuid");
2403 if (guid) {
2404 string GUID = (*adopt)->Adopted_GUID.c_str();
2405 GUID.insert(8, "-");
2406 GUID.insert(13, "-");
2407 GUID.insert(18, "-");
2408 GUID.insert(23, "-");
2409
2410 if (strcasecmp(GUID.c_str(), guid->value()) == 0) {
2411 xml_attribute<>* attr = volume->first_attribute("nickname");
2412 if (attr) {
2413 (*adopt)->Storage_Name = attr->value();
2414 (*adopt)->Display_Name = (*adopt)->Storage_Name;
2415 (*adopt)->Backup_Display_Name = (*adopt)->Storage_Name;
2416 LOGINFO("storage name from storage.xml is '%s'\n", attr->value());
2417 }
2418 attr = volume->first_attribute("fsUuid");
2419 if (attr && !Primary_Storage_UUID.empty() && strcmp(Primary_Storage_UUID.c_str(), attr->value()) == 0) {
2420 TWPartition* Dat = Find_Partition_By_Path("/data");
2421 if (Dat) {
2422 LOGINFO("Internal storage is found on adopted storage '%s'\n", (*adopt)->Display_Name.c_str());
2423 LOGINFO("Changing '%s' to point to '%s'\n", Dat->Symlink_Mount_Point.c_str(), (*adopt)->Storage_Path.c_str());
2424 (*adopt)->Symlink_Mount_Point = Dat->Symlink_Mount_Point;
2425 Dat->Symlink_Mount_Point = "";
2426 // Toggle mounts to ensure that the symlink mount point (probably /sdcard) is mounted to the right location
2427 Dat->UnMount(false);
2428 Dat->Mount(false);
2429 (*adopt)->UnMount(false);
2430 (*adopt)->Mount(false);
2431 Output_Partition((*adopt));
2432 }
2433 }
2434 break;
2435 }
2436 }
2437 volume = volume->next_sibling("volume");
2438 }
2439 }
2440 }
2441 }
2442 }
2443 if (xmlFile) {
2444 doc->clear();
2445 delete doc;
2446 free(xmlFile);
2447 }
2448#else
2449 LOGINFO("Decrypt_Adopted: no crypto support\n");
2450 return;
2451#endif
2452}
Ethan Yonkerfcf3f242016-02-16 12:30:26 -06002453
2454void TWPartitionManager::Remove_Partition_By_Path(string Path) {
2455 std::vector<TWPartition*>::iterator iter;
2456 string Local_Path = TWFunc::Get_Root_Path(Path);
2457
2458 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2459 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
2460 LOGINFO("Found and erasing '%s' from partition list\n", Local_Path.c_str());
2461 Partitions.erase(iter);
2462 return;
2463 }
2464 }
2465}