blob: c1b857ca97a8b99adfb42385ff95724a84c39964 [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"
Ethan Yonker3fdcda42016-11-30 12:29:37 -060043#include "exclude.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"
Ethan Yonkerbd7492d2016-12-07 13:55:01 -060065 #ifdef TW_INCLUDE_FBE
66 #include "crypto/ext4crypt/Decrypt.h"
67 #endif
Dees_Troy5bf43922012-09-07 16:07:55 -040068#endif
Dees_Troy51a0e822012-09-05 15:24:24 -040069
Ethan Yonker1b190162016-12-05 15:25:19 -060070#ifdef AB_OTA_UPDATER
71#include <hardware/hardware.h>
72#include <hardware/boot_control.h>
73#endif
74
Ethan Yonker6277c792014-09-15 14:54:30 -050075extern bool datamedia;
76
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050077TWPartitionManager::TWPartitionManager(void) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040078 mtp_was_enabled = false;
Ethan Yonker726a0202014-12-16 20:01:38 -060079 mtp_write_fd = -1;
bigbiff7abc5fe2015-01-17 16:53:12 -050080 stop_backup.set_value(0);
Ethan Yonker1b190162016-12-05 15:25:19 -060081#ifdef AB_OTA_UPDATER
82 char slot_suffix[PROPERTY_VALUE_MAX];
83 property_get("ro.boot.slot_suffix", slot_suffix, "_a");
84 Active_Slot_Display = "";
85 if (strcmp(slot_suffix, "_a") == 0)
86 Set_Active_Slot("A");
87 else
88 Set_Active_Slot("B");
89#endif
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050090}
91
Dees_Troy51a0e822012-09-05 15:24:24 -040092int TWPartitionManager::Process_Fstab(string Fstab_Filename, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -040093 FILE *fstabFile;
94 char fstab_line[MAX_FSTAB_LINE_LENGTH];
Dees Troy02a64532014-03-19 15:23:32 +000095 TWPartition* settings_partition = NULL;
Matt Mowerbf4efa32014-04-14 23:25:26 -050096 TWPartition* andsec_partition = NULL;
Ethan Yonker726a0202014-12-16 20:01:38 -060097 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 -040098
99 fstabFile = fopen(Fstab_Filename.c_str(), "rt");
100 if (fstabFile == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000101 LOGERR("Critical Error: Unable to open fstab at '%s'.\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400102 return false;
103 }
104
105 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
106 if (fstab_line[0] != '/')
107 continue;
108
Dees_Troy2a923582012-09-20 12:13:34 -0400109 if (fstab_line[strlen(fstab_line) - 1] != '\n')
110 fstab_line[strlen(fstab_line)] = '\n';
Dees_Troy2a923582012-09-20 12:13:34 -0400111
Matt Mower2b2dd152016-04-26 11:24:08 -0500112 TWPartition* partition = new TWPartition();
Matt Mower72c87ce2016-04-26 14:34:56 -0500113 if (partition->Process_Fstab_Line(fstab_line, Display_Error))
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500114 Partitions.push_back(partition);
Matt Mower72c87ce2016-04-26 14:34:56 -0500115 else
Dees_Troy5bf43922012-09-07 16:07:55 -0400116 delete partition;
Matt Mower2b2dd152016-04-26 11:24:08 -0500117
118 memset(fstab_line, 0, sizeof(fstab_line));
Dees_Troy5bf43922012-09-07 16:07:55 -0400119 }
120 fclose(fstabFile);
Matt Mower72c87ce2016-04-26 14:34:56 -0500121
122 std::vector<TWPartition*>::iterator iter;
123 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
124 (*iter)->Partition_Post_Processing(Display_Error);
125
126 if ((*iter)->Is_Storage) {
127 ++storageid;
128 (*iter)->MTP_Storage_ID = storageid;
129 }
130
131 if (!settings_partition && (*iter)->Is_Settings_Storage && (*iter)->Is_Present)
132 settings_partition = (*iter);
133 else
134 (*iter)->Is_Settings_Storage = false;
135
136 if (!andsec_partition && (*iter)->Has_Android_Secure && (*iter)->Is_Present)
137 andsec_partition = (*iter);
138 else
139 (*iter)->Has_Android_Secure = false;
140 }
141
Ethan Yonker6277c792014-09-15 14:54:30 -0500142 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) {
143 // Attempt to automatically identify /data/media emulated storage devices
144 TWPartition* Dat = Find_Partition_By_Path("/data");
145 if (Dat) {
146 LOGINFO("Using automatic handling for /data/media emulated storage device.\n");
147 datamedia = true;
that9e0593e2014-10-08 00:01:24 +0200148 Dat->Setup_Data_Media();
Ethan Yonker6277c792014-09-15 14:54:30 -0500149 settings_partition = Dat;
Ethan Yonker726a0202014-12-16 20:01:38 -0600150 // Since /data was not considered a storage partition earlier, we still need to assign an MTP ID
151 ++storageid;
152 Dat->MTP_Storage_ID = storageid;
Ethan Yonker6277c792014-09-15 14:54:30 -0500153 }
154 }
Dees Troy02a64532014-03-19 15:23:32 +0000155 if (!settings_partition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500156 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
157 if ((*iter)->Is_Storage) {
Dees Troy02a64532014-03-19 15:23:32 +0000158 settings_partition = (*iter);
Dees_Troya13d74f2013-03-24 08:54:55 -0500159 break;
160 }
161 }
Dees Troy02a64532014-03-19 15:23:32 +0000162 if (!settings_partition)
Dees_Troy2673cec2013-04-02 20:22:16 +0000163 LOGERR("Unable to locate storage partition for storing settings file.\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500164 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400165 if (!Write_Fstab()) {
166 if (Display_Error)
Dees_Troy2673cec2013-04-02 20:22:16 +0000167 LOGERR("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400168 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000169 LOGINFO("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400170 }
Matt Mowered71fa32014-04-16 13:21:47 -0500171
Matt Mowerbf4efa32014-04-14 23:25:26 -0500172 if (andsec_partition) {
173 Setup_Android_Secure_Location(andsec_partition);
Matt Mowered71fa32014-04-16 13:21:47 -0500174 } else if (settings_partition) {
Matt Mowerbf4efa32014-04-14 23:25:26 -0500175 Setup_Android_Secure_Location(settings_partition);
176 }
Matt Mowered71fa32014-04-16 13:21:47 -0500177 if (settings_partition) {
178 Setup_Settings_Storage_Partition(settings_partition);
179 }
Ethan Yonker253368a2014-11-25 15:00:52 -0600180#ifdef TW_INCLUDE_CRYPTO
Ethan Yonkercceebb82014-11-18 10:17:59 -0600181 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
182 if (Decrypt_Data && Decrypt_Data->Is_Encrypted && !Decrypt_Data->Is_Decrypted) {
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600183 if (Decrypt_Data->Is_FBE) {
184 if (DataManager::GetIntValue(TW_CRYPTO_PWTYPE) == 0) {
185 if (Decrypt_Device("!") == 0) {
186 gui_msg("decrypt_success=Successfully decrypted with default password.");
187 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
188 } else {
189 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
190 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600191 }
192 } else {
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600193 int password_type = cryptfs_get_password_type();
194 if (password_type == CRYPT_TYPE_DEFAULT) {
195 LOGINFO("Device is encrypted with the default password, attempting to decrypt.\n");
196 if (Decrypt_Device("default_password") == 0) {
197 gui_msg("decrypt_success=Successfully decrypted with default password.");
198 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
199 } else {
200 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
201 }
202 } else {
203 DataManager::SetValue("TW_CRYPTO_TYPE", password_type);
204 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600205 }
206 }
Ethan Yonker66a19492015-12-10 10:19:45 -0600207 if (Decrypt_Data && (!Decrypt_Data->Is_Encrypted || Decrypt_Data->Is_Decrypted) && Decrypt_Data->Mount(false)) {
208 Decrypt_Adopted();
209 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600210#endif
Dees_Troy51127312012-09-08 13:08:49 -0400211 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -0400212 UnMount_Main_Partitions();
Ethan Yonker1b190162016-12-05 15:25:19 -0600213#ifdef AB_OTA_UPDATER
214 DataManager::SetValue("tw_active_slot", Get_Active_Slot_Display());
215#endif
Dees_Troy5bf43922012-09-07 16:07:55 -0400216 return true;
217}
218
219int TWPartitionManager::Write_Fstab(void) {
220 FILE *fp;
221 std::vector<TWPartition*>::iterator iter;
222 string Line;
223
224 fp = fopen("/etc/fstab", "w");
225 if (fp == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000226 LOGINFO("Can not open /etc/fstab.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400227 return false;
228 }
Dees_Troy63c8df72012-09-10 14:02:05 -0400229 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy51127312012-09-08 13:08:49 -0400230 if ((*iter)->Can_Be_Mounted) {
dianlujitao7d304c72016-01-02 12:15:50 +0800231 Line = (*iter)->Actual_Block_Device + " " + (*iter)->Mount_Point + " " + (*iter)->Current_File_System + " rw 0 0\n";
Dees_Troy5bf43922012-09-07 16:07:55 -0400232 fputs(Line.c_str(), fp);
Dees_Troy91862e62013-04-04 23:48:21 +0000233 }
234 // Handle subpartition tracking
235 if ((*iter)->Is_SubPartition) {
236 TWPartition* ParentPartition = Find_Partition_By_Path((*iter)->SubPartition_Of);
237 if (ParentPartition)
238 ParentPartition->Has_SubPartition = true;
239 else
240 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 -0400241 }
242 }
243 fclose(fp);
244 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -0400245}
246
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500247void TWPartitionManager::Setup_Settings_Storage_Partition(TWPartition* Part) {
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500248 DataManager::SetValue("tw_settings_path", Part->Storage_Path);
249 DataManager::SetValue("tw_storage_path", Part->Storage_Path);
250 LOGINFO("Settings storage is '%s'\n", Part->Storage_Path.c_str());
251}
252
Matt Mowerbf4efa32014-04-14 23:25:26 -0500253void TWPartitionManager::Setup_Android_Secure_Location(TWPartition* Part) {
254 if (Part->Has_Android_Secure)
255 Part->Setup_AndSec();
Ethan Yonker6277c792014-09-15 14:54:30 -0500256 else if (!datamedia)
Matt Mowerbf4efa32014-04-14 23:25:26 -0500257 Part->Setup_AndSec();
Matt Mowerbf4efa32014-04-14 23:25:26 -0500258}
259
Dees_Troy8170a922012-09-18 15:40:25 -0400260void TWPartitionManager::Output_Partition_Logging(void) {
261 std::vector<TWPartition*>::iterator iter;
262
263 printf("\n\nPartition Logs:\n");
264 for (iter = Partitions.begin(); iter != Partitions.end(); iter++)
265 Output_Partition((*iter));
266}
267
268void TWPartitionManager::Output_Partition(TWPartition* Part) {
269 unsigned long long mb = 1048576;
270
Gary Peck004d48b2012-11-21 16:28:18 -0800271 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 -0400272 if (Part->Can_Be_Mounted) {
Gary Peck004d48b2012-11-21 16:28:18 -0800273 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 -0400274 }
Gary Peck004d48b2012-11-21 16:28:18 -0800275 printf("\n Flags: ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500276 if (Part->Can_Be_Mounted)
277 printf("Can_Be_Mounted ");
Gary Peck004d48b2012-11-21 16:28:18 -0800278 if (Part->Can_Be_Wiped)
279 printf("Can_Be_Wiped ");
Hashcodedabfd492013-08-29 22:45:30 -0700280 if (Part->Use_Rm_Rf)
281 printf("Use_Rm_Rf ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500282 if (Part->Can_Be_Backed_Up)
283 printf("Can_Be_Backed_Up ");
Gary Peck004d48b2012-11-21 16:28:18 -0800284 if (Part->Wipe_During_Factory_Reset)
285 printf("Wipe_During_Factory_Reset ");
286 if (Part->Wipe_Available_in_GUI)
287 printf("Wipe_Available_in_GUI ");
288 if (Part->Is_SubPartition)
289 printf("Is_SubPartition ");
290 if (Part->Has_SubPartition)
291 printf("Has_SubPartition ");
292 if (Part->Removable)
293 printf("Removable ");
294 if (Part->Is_Present)
295 printf("IsPresent ");
296 if (Part->Can_Be_Encrypted)
297 printf("Can_Be_Encrypted ");
298 if (Part->Is_Encrypted)
299 printf("Is_Encrypted ");
300 if (Part->Is_Decrypted)
301 printf("Is_Decrypted ");
302 if (Part->Has_Data_Media)
303 printf("Has_Data_Media ");
Dees_Troy83bd4832013-05-04 12:39:56 +0000304 if (Part->Can_Encrypt_Backup)
305 printf("Can_Encrypt_Backup ");
306 if (Part->Use_Userdata_Encryption)
307 printf("Use_Userdata_Encryption ");
Gary Peck004d48b2012-11-21 16:28:18 -0800308 if (Part->Has_Android_Secure)
309 printf("Has_Android_Secure ");
310 if (Part->Is_Storage)
311 printf("Is_Storage ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500312 if (Part->Is_Settings_Storage)
313 printf("Is_Settings_Storage ");
Dees_Troy68cab492012-12-12 19:29:35 +0000314 if (Part->Ignore_Blkid)
315 printf("Ignore_Blkid ");
Dees_Troy16c2b312013-01-15 16:51:18 +0000316 if (Part->Retain_Layout_Version)
317 printf("Retain_Layout_Version ");
Ethan Yonker253368a2014-11-25 15:00:52 -0600318 if (Part->Mount_To_Decrypt)
319 printf("Mount_To_Decrypt ");
Ethan Yonker96af84a2015-01-05 14:58:36 -0600320 if (Part->Can_Flash_Img)
321 printf("Can_Flash_Img ");
Ethan Yonker66a19492015-12-10 10:19:45 -0600322 if (Part->Is_Adopted_Storage)
323 printf("Is_Adopted_Storage ");
Ethan Yonker1b190162016-12-05 15:25:19 -0600324 if (Part->SlotSelect)
325 printf("SlotSelect ");
Gary Peck004d48b2012-11-21 16:28:18 -0800326 printf("\n");
327 if (!Part->SubPartition_Of.empty())
328 printf(" SubPartition_Of: %s\n", Part->SubPartition_Of.c_str());
329 if (!Part->Symlink_Path.empty())
330 printf(" Symlink_Path: %s\n", Part->Symlink_Path.c_str());
331 if (!Part->Symlink_Mount_Point.empty())
332 printf(" Symlink_Mount_Point: %s\n", Part->Symlink_Mount_Point.c_str());
333 if (!Part->Primary_Block_Device.empty())
334 printf(" Primary_Block_Device: %s\n", Part->Primary_Block_Device.c_str());
335 if (!Part->Alternate_Block_Device.empty())
336 printf(" Alternate_Block_Device: %s\n", Part->Alternate_Block_Device.c_str());
337 if (!Part->Decrypted_Block_Device.empty())
338 printf(" Decrypted_Block_Device: %s\n", Part->Decrypted_Block_Device.c_str());
Ethan Yonker253368a2014-11-25 15:00:52 -0600339 if (!Part->Crypto_Key_Location.empty() && Part->Crypto_Key_Location != "footer")
340 printf(" Crypto_Key_Location: %s\n", Part->Crypto_Key_Location.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800341 if (Part->Length != 0)
342 printf(" Length: %i\n", Part->Length);
343 if (!Part->Display_Name.empty())
344 printf(" Display_Name: %s\n", Part->Display_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500345 if (!Part->Storage_Name.empty())
346 printf(" Storage_Name: %s\n", Part->Storage_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800347 if (!Part->Backup_Path.empty())
348 printf(" Backup_Path: %s\n", Part->Backup_Path.c_str());
349 if (!Part->Backup_Name.empty())
350 printf(" Backup_Name: %s\n", Part->Backup_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500351 if (!Part->Backup_Display_Name.empty())
352 printf(" Backup_Display_Name: %s\n", Part->Backup_Display_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800353 if (!Part->Backup_FileName.empty())
354 printf(" Backup_FileName: %s\n", Part->Backup_FileName.c_str());
355 if (!Part->Storage_Path.empty())
356 printf(" Storage_Path: %s\n", Part->Storage_Path.c_str());
357 if (!Part->Current_File_System.empty())
358 printf(" Current_File_System: %s\n", Part->Current_File_System.c_str());
359 if (!Part->Fstab_File_System.empty())
360 printf(" Fstab_File_System: %s\n", Part->Fstab_File_System.c_str());
361 if (Part->Format_Block_Size != 0)
Ethan Yonkerc798c9c2015-10-09 11:15:26 -0500362 printf(" Format_Block_Size: %lu\n", Part->Format_Block_Size);
Dees_Troy094207a2012-09-26 12:00:39 -0400363 if (!Part->MTD_Name.empty())
364 printf(" MTD_Name: %s\n", Part->MTD_Name.c_str());
Matt Mower029a82d2017-01-08 13:12:38 -0600365 printf(" Backup_Method: %s\n", Part->Backup_Method_By_Name().c_str());
Hashcode62bd9e02013-11-19 21:59:42 -0800366 if (Part->Mount_Flags || !Part->Mount_Options.empty())
Matt Mower4ab42b12016-04-21 13:52:18 -0500367 printf(" Mount_Options: %s\n", Part->Mount_Options.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -0600368 if (Part->MTP_Storage_ID)
369 printf(" MTP_Storage_ID: %i\n", Part->MTP_Storage_ID);
Ethan Yonker6277c792014-09-15 14:54:30 -0500370 printf("\n");
Dees_Troy8170a922012-09-18 15:40:25 -0400371}
372
Dees_Troy51a0e822012-09-05 15:24:24 -0400373int TWPartitionManager::Mount_By_Path(string Path, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400374 std::vector<TWPartition*>::iterator iter;
375 int ret = false;
376 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400377 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400378
Dees_Troyd93bda52013-07-03 19:55:19 +0000379 if (Local_Path == "/tmp" || Local_Path == "/")
Dees_Troy43d8b002012-09-17 16:00:01 -0400380 return true;
381
Dees_Troy5bf43922012-09-07 16:07:55 -0400382 // Iterate through all partitions
383 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400384 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400385 ret = (*iter)->Mount(Display_Error);
386 found = true;
Dees_Troy51127312012-09-08 13:08:49 -0400387 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400388 (*iter)->Mount(Display_Error);
Dees_Troy51127312012-09-08 13:08:49 -0400389 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400390 }
391 if (found) {
392 return ret;
393 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500394 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 -0400395 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000396 LOGINFO("Mount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400397 }
398 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400399}
400
Dees_Troy51a0e822012-09-05 15:24:24 -0400401int TWPartitionManager::UnMount_By_Path(string Path, bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400402 std::vector<TWPartition*>::iterator iter;
403 int ret = false;
404 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400405 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy51127312012-09-08 13:08:49 -0400406
407 // Iterate through all partitions
408 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400409 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy51127312012-09-08 13:08:49 -0400410 ret = (*iter)->UnMount(Display_Error);
411 found = true;
412 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
413 (*iter)->UnMount(Display_Error);
414 }
415 }
416 if (found) {
417 return ret;
418 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500419 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 -0400420 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000421 LOGINFO("UnMount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400422 }
423 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400424}
425
Dees_Troy51a0e822012-09-05 15:24:24 -0400426int TWPartitionManager::Is_Mounted_By_Path(string Path) {
Dees_Troy51127312012-09-08 13:08:49 -0400427 TWPartition* Part = Find_Partition_By_Path(Path);
428
429 if (Part)
430 return Part->Is_Mounted();
431 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000432 LOGINFO("Is_Mounted: Unable to find partition for path '%s'\n", Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400433 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400434}
435
Dees_Troy5bf43922012-09-07 16:07:55 -0400436int TWPartitionManager::Mount_Current_Storage(bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400437 string current_storage_path = DataManager::GetCurrentStoragePath();
438
439 if (Mount_By_Path(current_storage_path, Display_Error)) {
440 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
441 if (FreeStorage)
442 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
443 return true;
444 }
445 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400446}
447
Dees_Troy5bf43922012-09-07 16:07:55 -0400448int TWPartitionManager::Mount_Settings_Storage(bool Display_Error) {
449 return Mount_By_Path(DataManager::GetSettingsStoragePath(), Display_Error);
450}
451
452TWPartition* TWPartitionManager::Find_Partition_By_Path(string Path) {
453 std::vector<TWPartition*>::iterator iter;
Dees_Troy38bd7602012-09-14 13:33:53 -0400454 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400455
456 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400457 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path))
Dees_Troy5bf43922012-09-07 16:07:55 -0400458 return (*iter);
459 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400460 return NULL;
461}
462
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400463int TWPartitionManager::Check_Backup_Name(bool Display_Error) {
464 // Check the backup name to ensure that it is the correct size and contains only valid characters
465 // and that a backup with that name doesn't already exist
466 char backup_name[MAX_BACKUP_NAME_LEN];
467 char backup_loc[255], tw_image_dir[255];
468 int copy_size;
469 int index, cur_char;
470 string Backup_Name, Backup_Loc;
471
472 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
473 copy_size = Backup_Name.size();
474 // Check size
475 if (copy_size > MAX_BACKUP_NAME_LEN) {
476 if (Display_Error)
Ethan Yonker74db1572015-10-28 12:44:49 -0500477 gui_err("backup_name_len=Backup name is too long.");
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400478 return -2;
479 }
480
481 // Check each character
482 strncpy(backup_name, Backup_Name.c_str(), copy_size);
Dees_Troya13d74f2013-03-24 08:54:55 -0500483 if (copy_size == 1 && strncmp(backup_name, "0", 1) == 0)
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400484 return 0; // A "0" (zero) means to use the current timestamp for the backup name
485 for (index=0; index<copy_size; index++) {
486 cur_char = (int)backup_name[index];
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500487 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 -0400488 // These are valid characters
489 // Numbers
490 // Upper case letters
491 // Lower case letters
492 // Space
493 // and -_.{}[]
494 } else {
495 if (Display_Error)
Ethan Yonker4adc33e2016-01-22 16:07:11 -0600496 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 -0400497 return -3;
498 }
499 }
500
501 // Check to make sure that a backup with this name doesn't already exist
502 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Loc);
503 strcpy(backup_loc, Backup_Loc.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500504 sprintf(tw_image_dir,"%s/%s", backup_loc, Backup_Name.c_str());
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500505 if (TWFunc::Path_Exists(tw_image_dir)) {
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400506 if (Display_Error)
Matt Mower3c366972015-12-25 19:28:31 -0600507 gui_err("backup_name_exists=A backup with that name already exists!");
bigbiffce8f83c2015-12-12 18:30:21 -0500508
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400509 return -4;
510 }
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400511 // No problems found, return 0
512 return 0;
513}
514
bigbiff bigbiffa481d362016-08-08 20:36:10 -0400515bool TWPartitionManager::Make_MD5(PartitionSettings *part_settings)
Dees_Troy43d8b002012-09-17 16:00:01 -0400516{
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500517 string command, result;
518
519 if (part_settings->Part == NULL)
520 return false;
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500521 string Full_File = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500522 twrpDigest md5sum;
Dees_Troy43d8b002012-09-17 16:00:01 -0400523
bigbiff bigbiffa481d362016-08-08 20:36:10 -0400524 if (!part_settings->generate_md5)
Dees_Troy43d8b002012-09-17 16:00:01 -0400525 return true;
Dees_Troy43d8b002012-09-17 16:00:01 -0400526
Ethan Yonker74db1572015-10-28 12:44:49 -0500527 TWFunc::GUI_Operation_Text(TW_GENERATE_MD5_TEXT, gui_parse_text("{@generating_md51}"));
528 gui_msg("generating_md52= * Generating md5...");
Dees_Troy43d8b002012-09-17 16:00:01 -0400529 if (TWFunc::Path_Exists(Full_File)) {
bigbiffce8f83c2015-12-12 18:30:21 -0500530 md5sum.setfn(Full_File);
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500531 if (md5sum.computeMD5() == 0)
532 if (md5sum.write_md5digest() == 0)
Ethan Yonker74db1572015-10-28 12:44:49 -0500533 gui_msg("md5_created= * MD5 Created.");
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500534 else
535 return -1;
536 else
Ethan Yonker74db1572015-10-28 12:44:49 -0500537 gui_err("md5_error= * MD5 Error!");
Dees_Troy43d8b002012-09-17 16:00:01 -0400538 } else {
539 char filename[512];
540 int index = 0;
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500541 string strfn;
Dees_Troy43d8b002012-09-17 16:00:01 -0400542 sprintf(filename, "%s%03i", Full_File.c_str(), index);
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500543 strfn = filename;
Dees_Troy83bd4832013-05-04 12:39:56 +0000544 while (index < 1000) {
bigbiff bigbiff65a4c732013-03-15 15:17:50 -0400545 md5sum.setfn(filename);
Dees_Troy83bd4832013-05-04 12:39:56 +0000546 if (TWFunc::Path_Exists(filename)) {
547 if (md5sum.computeMD5() == 0) {
548 if (md5sum.write_md5digest() != 0)
549 {
Ethan Yonker74db1572015-10-28 12:44:49 -0500550 gui_err("md5_error= * MD5 Error!");
Dees_Troy83bd4832013-05-04 12:39:56 +0000551 return false;
552 }
553 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500554 gui_err("md5_compute_error= * Error computing MD5.");
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500555 return false;
556 }
557 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400558 index++;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400559 sprintf(filename, "%s%03i", Full_File.c_str(), index);
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500560 strfn = filename;
Dees_Troy43d8b002012-09-17 16:00:01 -0400561 }
562 if (index == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000563 LOGERR("Backup file: '%s' not found!\n", filename);
Dees_Troy43d8b002012-09-17 16:00:01 -0400564 return false;
565 }
Ethan Yonker74db1572015-10-28 12:44:49 -0500566 gui_msg("md5_created= * MD5 Created.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400567 }
568 return true;
569}
570
bigbiffce8f83c2015-12-12 18:30:21 -0500571bool TWPartitionManager::Backup_Partition(PartitionSettings *part_settings) {
Dees_Troy43d8b002012-09-17 16:00:01 -0400572 time_t start, stop;
Matt Mower23d8aae2017-01-06 14:30:33 -0600573 int use_compression;
bigbiffce8f83c2015-12-12 18:30:21 -0500574 string backup_log = part_settings->Backup_Folder + "/recovery.log";
Dees_Troy43d8b002012-09-17 16:00:01 -0400575
bigbiffce8f83c2015-12-12 18:30:21 -0500576 if (part_settings->Part == NULL)
Dees_Troy43d8b002012-09-17 16:00:01 -0400577 return true;
578
Dees_Troy093b7642012-09-21 15:59:38 -0400579 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400580
Tom Hite5a926722014-09-15 01:31:03 +0000581 TWFunc::SetPerformanceMode(true);
Dees_Troy43d8b002012-09-17 16:00:01 -0400582 time(&start);
583
bigbiffce8f83c2015-12-12 18:30:21 -0500584 if (part_settings->Part->Backup(part_settings, &tar_fork_pid)) {
Tom Hite5a926722014-09-15 01:31:03 +0000585 bool md5Success = false;
Matt Mower029a82d2017-01-08 13:12:38 -0600586 if (part_settings->adbbackup)
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400587 md5Success = true;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400588 else
589 md5Success = Make_MD5(part_settings);
590
bigbiffce8f83c2015-12-12 18:30:21 -0500591 if (part_settings->Part->Has_SubPartition) {
Dees_Troy8170a922012-09-18 15:40:25 -0400592 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400593 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -0400594
595 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400596 if ((*subpart)->Can_Be_Backed_Up && (*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500597 part_settings->Part = *subpart;
bigbiffce8f83c2015-12-12 18:30:21 -0500598 if (!(*subpart)->Backup(part_settings, &tar_fork_pid)) {
bigbiffce8f83c2015-12-12 18:30:21 -0500599 Clean_Backup_Folder(part_settings->Backup_Folder);
bigbiffbf1d6722015-02-14 16:25:59 -0500600 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
601 tw_set_default_metadata(backup_log.c_str());
Matt Mower02899552016-12-30 18:13:51 -0600602 TWFunc::SetPerformanceMode(false);
Dees_Troy8170a922012-09-18 15:40:25 -0400603 return false;
Tom Hite5a926722014-09-15 01:31:03 +0000604 }
Dees_Troy2727b992013-08-14 20:09:30 +0000605 sync();
606 sync();
bigbiffce8f83c2015-12-12 18:30:21 -0500607 if (!part_settings->adbbackup) {
bigbiff bigbiffa481d362016-08-08 20:36:10 -0400608 if (!Make_MD5(part_settings)) {
bigbiffce8f83c2015-12-12 18:30:21 -0500609 TWFunc::SetPerformanceMode(false);
610 return false;
611 }
Tom Hite5a926722014-09-15 01:31:03 +0000612 }
Dees_Troy8170a922012-09-18 15:40:25 -0400613 }
614 }
615 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400616 time(&stop);
that203bcc92015-05-09 17:27:33 +0200617 int backup_time = (int) difftime(stop, start);
Dees_Troy2673cec2013-04-02 20:22:16 +0000618 LOGINFO("Partition Backup time: %d\n", backup_time);
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400619
bigbiffce8f83c2015-12-12 18:30:21 -0500620 if (part_settings->Part->Backup_Method == BM_FILES) {
621 part_settings->file_time += backup_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400622 } else {
bigbiffce8f83c2015-12-12 18:30:21 -0500623 part_settings->img_time += backup_time;
624
Dees_Troy43d8b002012-09-17 16:00:01 -0400625 }
Tom Hite5a926722014-09-15 01:31:03 +0000626
Matt Mower02899552016-12-30 18:13:51 -0600627 TWFunc::SetPerformanceMode(false);
Tom Hite5a926722014-09-15 01:31:03 +0000628 return md5Success;
Dees_Troy43d8b002012-09-17 16:00:01 -0400629 } else {
bigbiffce8f83c2015-12-12 18:30:21 -0500630 Clean_Backup_Folder(part_settings->Backup_Folder);
bigbiffbf1d6722015-02-14 16:25:59 -0500631 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
632 tw_set_default_metadata(backup_log.c_str());
Tom Hite5a926722014-09-15 01:31:03 +0000633 TWFunc::SetPerformanceMode(false);
Dees_Troy43d8b002012-09-17 16:00:01 -0400634 return false;
635 }
bigbiff7abc5fe2015-01-17 16:53:12 -0500636 return 0;
637}
638
bigbiffbf1d6722015-02-14 16:25:59 -0500639void TWPartitionManager::Clean_Backup_Folder(string Backup_Folder) {
640 DIR *d = opendir(Backup_Folder.c_str());
641 struct dirent *p;
642 int r;
643
Ethan Yonker74db1572015-10-28 12:44:49 -0500644 gui_msg("backup_clean=Backup Failed. Cleaning Backup Folder.");
bigbiffbf1d6722015-02-14 16:25:59 -0500645
646 if (d == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500647 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Backup_Folder)(strerror(errno)));
bigbiffbf1d6722015-02-14 16:25:59 -0500648 return;
649 }
650
Matt Mower2b18a532015-02-20 16:58:05 -0600651 while ((p = readdir(d))) {
bigbiffbf1d6722015-02-14 16:25:59 -0500652 if (!strcmp(p->d_name, ".") || !strcmp(p->d_name, ".."))
653 continue;
654
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500655 string path = Backup_Folder + "/" + p->d_name;
bigbiffbf1d6722015-02-14 16:25:59 -0500656
657 size_t dot = path.find_last_of(".") + 1;
658 if (path.substr(dot) == "win" || path.substr(dot) == "md5" || path.substr(dot) == "info") {
659 r = unlink(path.c_str());
660 if (r != 0) {
661 LOGINFO("Unable to unlink '%s: %s'\n", path.c_str(), strerror(errno));
662 }
663 }
664 }
665 closedir(d);
666}
667
Ethan Yonker472f5062016-02-25 13:47:30 -0600668int TWPartitionManager::Check_Backup_Cancel() {
669 return stop_backup.get_value();
670}
671
bigbiff7abc5fe2015-01-17 16:53:12 -0500672int TWPartitionManager::Cancel_Backup() {
673 string Backup_Folder, Backup_Name, Full_Backup_Path;
674
675 stop_backup.set_value(1);
676
677 if (tar_fork_pid != 0) {
678 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
679 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Folder);
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500680 Full_Backup_Path = Backup_Folder + "/" + Backup_Name;
bigbiff7abc5fe2015-01-17 16:53:12 -0500681 LOGINFO("Killing pid: %d\n", tar_fork_pid);
682 kill(tar_fork_pid, SIGUSR2);
683 while (kill(tar_fork_pid, 0) == 0) {
684 usleep(1000);
685 }
686 LOGINFO("Backup_Run stopped and returning false, backup cancelled.\n");
687 LOGINFO("Removing directory %s\n", Full_Backup_Path.c_str());
688 TWFunc::removeDir(Full_Backup_Path, false);
689 tar_fork_pid = 0;
690 }
691
692 return 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400693}
694
bigbiffce8f83c2015-12-12 18:30:21 -0500695int TWPartitionManager::Run_Backup(bool adbbackup) {
696 PartitionSettings part_settings;
Matt Mower23d8aae2017-01-06 14:30:33 -0600697 int partition_count = 0, disable_free_space_check = 0, do_md5 = 0;
bigbiffce8f83c2015-12-12 18:30:21 -0500698 int gui_adb_backup;
699 string Backup_Name, Backup_List, backup_path;
Matt Mower23d8aae2017-01-06 14:30:33 -0600700 unsigned long long total_bytes = 0, free_space = 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400701 TWPartition* storage = NULL;
Dees_Troy8170a922012-09-18 15:40:25 -0400702 std::vector<TWPartition*>::iterator subpart;
Dees_Troy43d8b002012-09-17 16:00:01 -0400703 struct tm *t;
Matt Mower23d8aae2017-01-06 14:30:33 -0600704 time_t seconds, total_start, total_stop;
Dees_Troya13d74f2013-03-24 08:54:55 -0500705 size_t start_pos = 0, end_pos = 0;
bigbiff7abc5fe2015-01-17 16:53:12 -0500706 stop_backup.set_value(0);
Dees_Troy43d8b002012-09-17 16:00:01 -0400707 seconds = time(0);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500708 t = localtime(&seconds);
Dees_Troy43d8b002012-09-17 16:00:01 -0400709
bigbiffce8f83c2015-12-12 18:30:21 -0500710 part_settings.img_bytes_remaining = 0;
711 part_settings.file_bytes_remaining = 0;
712 part_settings.img_time = 0;
713 part_settings.file_time = 0;
714 part_settings.img_bytes = 0;
715 part_settings.file_bytes = 0;
716 part_settings.PM_Method = PM_BACKUP;
717
718 DataManager::GetValue("tw_enable_adb_backup", gui_adb_backup);
719 if (gui_adb_backup == true)
720 adbbackup = true;
721
722 part_settings.adbbackup = adbbackup;
Dees_Troy43d8b002012-09-17 16:00:01 -0400723 time(&total_start);
724
725 Update_System_Details();
726
727 if (!Mount_Current_Storage(true))
728 return false;
729
730 DataManager::GetValue(TW_SKIP_MD5_GENERATE_VAR, do_md5);
Dees_Troy4a2a1262012-09-18 09:33:47 -0400731 if (do_md5 == 0)
bigbiffce8f83c2015-12-12 18:30:21 -0500732 part_settings.generate_md5 = true;
Dees_Troyc5865ab2012-09-24 15:08:04 -0400733 else
bigbiffce8f83c2015-12-12 18:30:21 -0500734 part_settings.generate_md5 = false;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400735
bigbiffce8f83c2015-12-12 18:30:21 -0500736 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, part_settings.Backup_Folder);
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500737 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
738 if (Backup_Name == gui_lookup("curr_date", "(Current Date)")) {
739 Backup_Name = TWFunc::Get_Current_Date();
740 } else if (Backup_Name == gui_lookup("auto_generate", "(Auto Generate)") || Backup_Name == "0" || Backup_Name.empty()) {
Dees Troyb21cc642013-09-10 17:36:41 +0000741 TWFunc::Auto_Generate_Backup_Name();
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500742 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
Dees_Troy43d8b002012-09-17 16:00:01 -0400743 }
bigbiffce8f83c2015-12-12 18:30:21 -0500744
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500745 LOGINFO("Backup Name is: '%s'\n", Backup_Name.c_str());
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500746 part_settings.Backup_Folder = part_settings.Backup_Folder + "/" + Backup_Name;
bigbiffce8f83c2015-12-12 18:30:21 -0500747
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500748 LOGINFO("Backup_Folder is: '%s'\n", part_settings.Backup_Folder.c_str());
Dees_Troy43d8b002012-09-17 16:00:01 -0400749
Dees_Troy2673cec2013-04-02 20:22:16 +0000750 LOGINFO("Calculating backup details...\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500751 DataManager::GetValue("tw_backup_list", Backup_List);
752 if (!Backup_List.empty()) {
753 end_pos = Backup_List.find(";", start_pos);
754 while (end_pos != string::npos && start_pos < Backup_List.size()) {
755 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -0500756 part_settings.Part = Find_Partition_By_Path(backup_path);
757 if (part_settings.Part != NULL) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500758 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500759 if (part_settings.Part->Backup_Method == BM_FILES)
760 part_settings.file_bytes += part_settings.Part->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500761 else
bigbiffce8f83c2015-12-12 18:30:21 -0500762 part_settings.img_bytes += part_settings.Part->Backup_Size;
763 if (part_settings.Part->Has_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500764 std::vector<TWPartition*>::iterator subpart;
765
766 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiffce8f83c2015-12-12 18:30:21 -0500767 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 -0500768 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500769 if ((*subpart)->Backup_Method == BM_FILES)
770 part_settings.file_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500771 else
bigbiffce8f83c2015-12-12 18:30:21 -0500772 part_settings.img_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500773 }
774 }
Dees_Troy8170a922012-09-18 15:40:25 -0400775 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500776 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500777 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 -0400778 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500779 start_pos = end_pos + 1;
780 end_pos = Backup_List.find(";", start_pos);
Dees_Troy43d8b002012-09-17 16:00:01 -0400781 }
782 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400783
784 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500785 gui_msg("no_partition_selected=No partitions selected for backup.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400786 return false;
787 }
bigbiffce8f83c2015-12-12 18:30:21 -0500788 if (adbbackup) {
789 if (twadbbu::Write_ADB_Stream_Header(partition_count) == false) {
790 return false;
791 }
792 }
793 total_bytes = part_settings.file_bytes + part_settings.img_bytes;
Ethan Yonker472f5062016-02-25 13:47:30 -0600794 ProgressTracking progress(total_bytes);
bigbiffce8f83c2015-12-12 18:30:21 -0500795 part_settings.progress = &progress;
796
Ethan Yonker74db1572015-10-28 12:44:49 -0500797 gui_msg(Msg("total_partitions_backup= * Total number of partitions to back up: {1}")(partition_count));
798 gui_msg(Msg("total_backup_size= * Total size of all data: {1}MB")(total_bytes / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400799 storage = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
800 if (storage != NULL) {
801 free_space = storage->Free;
Ethan Yonker74db1572015-10-28 12:44:49 -0500802 gui_msg(Msg("available_space= * Available space: {1}MB")(free_space / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400803 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500804 gui_err("unable_locate_storage=Unable to locate storage device.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400805 return false;
806 }
bigbiffbf1d6722015-02-14 16:25:59 -0500807
Matt Mowerbfccfb82016-04-25 23:22:31 -0500808 DataManager::GetValue(TW_DISABLE_FREE_SPACE_VAR, disable_free_space_check);
bigbiffce8f83c2015-12-12 18:30:21 -0500809
810 if (adbbackup)
811 disable_free_space_check = true;
812
bigbiffbf1d6722015-02-14 16:25:59 -0500813 if (!disable_free_space_check) {
814 if (free_space - (32 * 1024 * 1024) < total_bytes) {
815 // We require an extra 32MB just in case
Ethan Yonker74db1572015-10-28 12:44:49 -0500816 gui_err("no_space=Not enough free space on storage.");
bigbiffbf1d6722015-02-14 16:25:59 -0500817 return false;
818 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400819 }
bigbiffce8f83c2015-12-12 18:30:21 -0500820 part_settings.img_bytes_remaining = part_settings.img_bytes;
821 part_settings.file_bytes_remaining = part_settings.file_bytes;
Dees_Troy43d8b002012-09-17 16:00:01 -0400822
Ethan Yonker74db1572015-10-28 12:44:49 -0500823 gui_msg("backup_started=[BACKUP STARTED]");
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500824 gui_msg(Msg("backup_folder= * Backup Folder: {1}")(part_settings.Backup_Folder));
825 if (!TWFunc::Recursive_Mkdir(part_settings.Backup_Folder)) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500826 gui_err("fail_backup_folder=Failed to make backup folder.");
Dees_Troyd4b22b02013-01-18 17:17:58 +0000827 return false;
828 }
829
Dees_Troy2673cec2013-04-02 20:22:16 +0000830 DataManager::SetProgress(0.0);
Dees_Troy093b7642012-09-21 15:59:38 -0400831
Dees_Troya13d74f2013-03-24 08:54:55 -0500832 start_pos = 0;
833 end_pos = Backup_List.find(";", start_pos);
834 while (end_pos != string::npos && start_pos < Backup_List.size()) {
bigbiff7abc5fe2015-01-17 16:53:12 -0500835 if (stop_backup.get_value() != 0)
836 return -1;
Dees_Troya13d74f2013-03-24 08:54:55 -0500837 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -0500838 part_settings.Part = Find_Partition_By_Path(backup_path);
839 if (part_settings.Part != NULL) {
840 if (!Backup_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -0500841 return false;
842 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500843 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 -0500844 }
845 start_pos = end_pos + 1;
846 end_pos = Backup_List.find(";", start_pos);
847 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400848
849 // Average BPS
bigbiffce8f83c2015-12-12 18:30:21 -0500850 if (part_settings.img_time == 0)
851 part_settings.img_time = 1;
852 if (part_settings.file_time == 0)
853 part_settings.file_time = 1;
854 int img_bps = (int)part_settings.img_bytes / (int)part_settings.img_time;
855 unsigned long long file_bps = part_settings.file_bytes / (int)part_settings.file_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400856
bigbiffce8f83c2015-12-12 18:30:21 -0500857 if (part_settings.file_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -0600858 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 -0500859 if (part_settings.img_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -0600860 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 -0400861
862 time(&total_stop);
863 int total_time = (int) difftime(total_stop, total_start);
bigbiffce8f83c2015-12-12 18:30:21 -0500864
865 uint64_t actual_backup_size;
Ethan Yonker3fdcda42016-11-30 12:29:37 -0600866 if (!adbbackup) {
867 TWExclude twe;
868 actual_backup_size = twe.Get_Folder_Size(part_settings.Backup_Folder);
869 } else
bigbiffce8f83c2015-12-12 18:30:21 -0500870 actual_backup_size = part_settings.file_bytes + part_settings.img_bytes;
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500871 actual_backup_size /= (1024LLU * 1024LLU);
Dees_Troy43d8b002012-09-17 16:00:01 -0400872
bigbiffce8f83c2015-12-12 18:30:21 -0500873 int prev_img_bps = 0, use_compression = 0;
874 unsigned long long prev_file_bps = 0;
Matt Mower173cdb92016-12-31 02:49:19 -0600875 DataManager::GetValue(TW_BACKUP_AVG_IMG_RATE, prev_img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -0400876 img_bps += (prev_img_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500877 img_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -0400878
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500879 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400880 if (use_compression)
881 DataManager::GetValue(TW_BACKUP_AVG_FILE_COMP_RATE, prev_file_bps);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500882 else
Dees_Troy093b7642012-09-21 15:59:38 -0400883 DataManager::GetValue(TW_BACKUP_AVG_FILE_RATE, prev_file_bps);
884 file_bps += (prev_file_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500885 file_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -0400886
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500887 DataManager::SetValue(TW_BACKUP_AVG_IMG_RATE, img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -0400888 if (use_compression)
889 DataManager::SetValue(TW_BACKUP_AVG_FILE_COMP_RATE, file_bps);
890 else
891 DataManager::SetValue(TW_BACKUP_AVG_FILE_RATE, file_bps);
892
Ethan Yonker74db1572015-10-28 12:44:49 -0500893 gui_msg(Msg("total_backed_size=[{1} MB TOTAL BACKED UP]")(actual_backup_size));
Dees_Troy43d8b002012-09-17 16:00:01 -0400894 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -0400895 UnMount_Main_Partitions();
Ethan Yonker56db1c42015-12-20 22:49:00 -0600896 gui_msg(Msg(msg::kHighlight, "backup_completed=[BACKUP COMPLETED IN {1} SECONDS]")(total_time)); // the end
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500897 string backup_log = part_settings.Backup_Folder + "/recovery.log";
Dees_Troy3f5c4e82013-02-01 15:16:59 +0000898 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
Ethan Yonker4b94cfd2014-12-11 10:00:45 -0600899 tw_set_default_metadata(backup_log.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -0500900
901 if (part_settings.adbbackup) {
902 if (twadbbu::Write_ADB_Stream_Trailer() == false) {
903 return false;
904 }
905 }
906 part_settings.adbbackup = false;
907 DataManager::SetValue("tw_enable_adb_backup", 0);
908
Dees_Troy3f5c4e82013-02-01 15:16:59 +0000909 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -0400910}
911
bigbiffce8f83c2015-12-12 18:30:21 -0500912bool TWPartitionManager::Restore_Partition(PartitionSettings *part_settings) {
Dees_Troy4a2a1262012-09-18 09:33:47 -0400913 time_t Start, Stop;
bigbiffce8f83c2015-12-12 18:30:21 -0500914
Tom Hite5a926722014-09-15 01:31:03 +0000915 TWFunc::SetPerformanceMode(true);
bigbiffce8f83c2015-12-12 18:30:21 -0500916
Dees_Troy4a2a1262012-09-18 09:33:47 -0400917 time(&Start);
Ethan Yonker472f5062016-02-25 13:47:30 -0600918
bigbiffce8f83c2015-12-12 18:30:21 -0500919 if (!part_settings->Part->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +0000920 TWFunc::SetPerformanceMode(false);
Dees_Troy4a2a1262012-09-18 09:33:47 -0400921 return false;
Tom Hite5a926722014-09-15 01:31:03 +0000922 }
bigbiffce8f83c2015-12-12 18:30:21 -0500923 if (part_settings->Part->Has_SubPartition) {
Dees_Troy8170a922012-09-18 15:40:25 -0400924 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400925 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -0400926
927 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400928 part_settings->Part = *subpart;
929 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
930 part_settings->Part = (*subpart);
bigbiffce8f83c2015-12-12 18:30:21 -0500931 if (!(*subpart)->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +0000932 TWFunc::SetPerformanceMode(false);
Dees_Troy8170a922012-09-18 15:40:25 -0400933 return false;
Tom Hite5a926722014-09-15 01:31:03 +0000934 }
Dees_Troy8170a922012-09-18 15:40:25 -0400935 }
936 }
937 }
Dees_Troy4a2a1262012-09-18 09:33:47 -0400938 time(&Stop);
Tom Hite5a926722014-09-15 01:31:03 +0000939 TWFunc::SetPerformanceMode(false);
bigbiffce8f83c2015-12-12 18:30:21 -0500940 gui_msg(Msg("restore_part_done=[{1} done ({2} seconds)]")(part_settings->Part->Backup_Display_Name)((int)difftime(Stop, Start)));
941
Dees_Troy4a2a1262012-09-18 09:33:47 -0400942 return true;
943}
944
Ethan Yonker472f5062016-02-25 13:47:30 -0600945int TWPartitionManager::Run_Restore(const string& Restore_Name) {
bigbiffce8f83c2015-12-12 18:30:21 -0500946 PartitionSettings part_settings;
Matt Mower23d8aae2017-01-06 14:30:33 -0600947 int check_md5;
bigbiffce8f83c2015-12-12 18:30:21 -0500948
Dees_Troy4a2a1262012-09-18 09:33:47 -0400949 time_t rStart, rStop;
950 time(&rStart);
Dees_Troya13d74f2013-03-24 08:54:55 -0500951 string Restore_List, restore_path;
952 size_t start_pos = 0, end_pos;
bigbiffce8f83c2015-12-12 18:30:21 -0500953
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500954 part_settings.Backup_Folder = Restore_Name;
bigbiffce8f83c2015-12-12 18:30:21 -0500955 part_settings.Part = NULL;
956 part_settings.partition_count = 0;
957 part_settings.total_restore_size = 0;
958 part_settings.adbbackup = false;
959 part_settings.PM_Method = PM_RESTORE;
Dees_Troy43d8b002012-09-17 16:00:01 -0400960
Ethan Yonker74db1572015-10-28 12:44:49 -0500961 gui_msg("restore_started=[RESTORE STARTED]");
962 gui_msg(Msg("restore_folder=Restore folder: '{1}'")(Restore_Name));
Dees_Troy43d8b002012-09-17 16:00:01 -0400963
Dees_Troy4a2a1262012-09-18 09:33:47 -0400964 if (!Mount_Current_Storage(true))
965 return false;
966
967 DataManager::GetValue(TW_SKIP_MD5_CHECK_VAR, check_md5);
Dees_Troya13d74f2013-03-24 08:54:55 -0500968 if (check_md5 > 0) {
969 // Check MD5 files first before restoring to ensure that all of them match before starting a restore
Ethan Yonker74db1572015-10-28 12:44:49 -0500970 TWFunc::GUI_Operation_Text(TW_VERIFY_MD5_TEXT, gui_parse_text("{@verifying_md5}"));
971 gui_msg("verifying_md5=Verifying MD5");
Dees_Troya13d74f2013-03-24 08:54:55 -0500972 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500973 gui_msg("skip_md5=Skipping MD5 check based on user setting.");
Dees_Troya13d74f2013-03-24 08:54:55 -0500974 }
Ethan Yonker74db1572015-10-28 12:44:49 -0500975 gui_msg("calc_restore=Calculating restore details...");
Dees_Troya13d74f2013-03-24 08:54:55 -0500976 DataManager::GetValue("tw_restore_selected", Restore_List);
bigbiffce8f83c2015-12-12 18:30:21 -0500977
Dees_Troya13d74f2013-03-24 08:54:55 -0500978 if (!Restore_List.empty()) {
979 end_pos = Restore_List.find(";", start_pos);
980 while (end_pos != string::npos && start_pos < Restore_List.size()) {
981 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -0500982 part_settings.Part = Find_Partition_By_Path(restore_path);
983 if (part_settings.Part != NULL) {
bigbiffce8f83c2015-12-12 18:30:21 -0500984 if (part_settings.Part->Mount_Read_Only) {
985 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 -0500986 return false;
987 }
bigbiffce8f83c2015-12-12 18:30:21 -0500988
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400989 if (check_md5 > 0 && !part_settings.Part->Check_MD5(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -0500990 return false;
bigbiffce8f83c2015-12-12 18:30:21 -0500991 part_settings.partition_count++;
992 part_settings.total_restore_size += part_settings.Part->Get_Restore_Size(&part_settings);
993 if (part_settings.Part->Has_SubPartition) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400994 TWPartition *parentPart = part_settings.Part;
Dees_Troya13d74f2013-03-24 08:54:55 -0500995 std::vector<TWPartition*>::iterator subpart;
996
997 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400998 part_settings.Part = *subpart;
999 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
1000 if (check_md5 > 0 && !(*subpart)->Check_MD5(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001001 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001002 part_settings.total_restore_size += (*subpart)->Get_Restore_Size(&part_settings);
Dees_Troya13d74f2013-03-24 08:54:55 -05001003 }
1004 }
1005 }
1006 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001007 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001008 }
1009 start_pos = end_pos + 1;
1010 end_pos = Restore_List.find(";", start_pos);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001011 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001012 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001013
bigbiffce8f83c2015-12-12 18:30:21 -05001014 if (part_settings.partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001015 gui_err("no_part_restore=No partitions selected for restore.");
Dees_Troy4a2a1262012-09-18 09:33:47 -04001016 return false;
1017 }
1018
bigbiffce8f83c2015-12-12 18:30:21 -05001019 gui_msg(Msg("restore_part_count=Restoring {1} partitions...")(part_settings.partition_count));
1020 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 +00001021 DataManager::SetProgress(0.0);
bigbiffce8f83c2015-12-12 18:30:21 -05001022 ProgressTracking progress(part_settings.total_restore_size);
1023 part_settings.progress = &progress;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001024
Dees_Troya13d74f2013-03-24 08:54:55 -05001025 start_pos = 0;
1026 if (!Restore_List.empty()) {
1027 end_pos = Restore_List.find(";", start_pos);
1028 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1029 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001030
1031 part_settings.Part = Find_Partition_By_Path(restore_path);
1032 if (part_settings.Part != NULL) {
1033 part_settings.partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -05001034 if (!Restore_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001035 return false;
1036 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001037 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001038 }
1039 start_pos = end_pos + 1;
1040 end_pos = Restore_List.find(";", start_pos);
1041 }
1042 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001043 TWFunc::GUI_Operation_Text(TW_UPDATE_SYSTEM_DETAILS_TEXT, gui_parse_text("{@updating_system_details}"));
Dees_Troy43d8b002012-09-17 16:00:01 -04001044 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001045 UnMount_Main_Partitions();
Dees_Troy4a2a1262012-09-18 09:33:47 -04001046 time(&rStop);
Matt Mower3c366972015-12-25 19:28:31 -06001047 gui_msg(Msg(msg::kHighlight, "restore_completed=[RESTORE COMPLETED IN {1} SECONDS]")((int)difftime(rStop,rStart)));
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001048 DataManager::SetValue("tw_file_progress", "");
bigbiffce8f83c2015-12-12 18:30:21 -05001049
Dees_Troy63c8df72012-09-10 14:02:05 -04001050 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001051}
1052
1053void TWPartitionManager::Set_Restore_Files(string Restore_Name) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001054 // Start with the default values
Dees_Troya13d74f2013-03-24 08:54:55 -05001055 string Restore_List;
Dees_Troy83bd4832013-05-04 12:39:56 +00001056 bool get_date = true, check_encryption = true;
1057
1058 DataManager::SetValue("tw_restore_encrypted", 0);
Dees_Troy63c8df72012-09-10 14:02:05 -04001059
1060 DIR* d;
1061 d = opendir(Restore_Name.c_str());
1062 if (d == NULL)
1063 {
Ethan Yonker74db1572015-10-28 12:44:49 -05001064 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Restore_Name)(strerror(errno)));
Dees_Troy63c8df72012-09-10 14:02:05 -04001065 return;
1066 }
1067
1068 struct dirent* de;
1069 while ((de = readdir(d)) != NULL)
1070 {
1071 // Strip off three components
1072 char str[256];
1073 char* label;
1074 char* fstype = NULL;
1075 char* extn = NULL;
1076 char* ptr;
1077
1078 strcpy(str, de->d_name);
1079 if (strlen(str) <= 2)
1080 continue;
1081
1082 if (get_date) {
1083 char file_path[255];
1084 struct stat st;
1085
1086 strcpy(file_path, Restore_Name.c_str());
1087 strcat(file_path, "/");
1088 strcat(file_path, str);
1089 stat(file_path, &st);
1090 string backup_date = ctime((const time_t*)(&st.st_mtime));
1091 DataManager::SetValue(TW_RESTORE_FILE_DATE, backup_date);
1092 get_date = false;
1093 }
1094
1095 label = str;
1096 ptr = label;
1097 while (*ptr && *ptr != '.') ptr++;
1098 if (*ptr == '.')
1099 {
1100 *ptr = 0x00;
1101 ptr++;
1102 fstype = ptr;
1103 }
1104 while (*ptr && *ptr != '.') ptr++;
1105 if (*ptr == '.')
1106 {
1107 *ptr = 0x00;
1108 ptr++;
1109 extn = ptr;
1110 }
1111
Dees_Troy83bd4832013-05-04 12:39:56 +00001112 if (fstype == NULL || extn == NULL || strcmp(fstype, "log") == 0) continue;
Dees_Troya13d74f2013-03-24 08:54:55 -05001113 int extnlength = strlen(extn);
Dees_Troy83bd4832013-05-04 12:39:56 +00001114 if (extnlength != 3 && extnlength != 6) continue;
1115 if (extnlength >= 3 && strncmp(extn, "win", 3) != 0) continue;
1116 //if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
1117
1118 if (check_encryption) {
1119 string filename = Restore_Name + "/";
1120 filename += de->d_name;
1121 if (TWFunc::Get_File_Type(filename) == 2) {
1122 LOGINFO("'%s' is encrypted\n", filename.c_str());
1123 DataManager::SetValue("tw_restore_encrypted", 1);
1124 }
1125 }
Dees_Troya13d74f2013-03-24 08:54:55 -05001126 if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
Dees_Troy63c8df72012-09-10 14:02:05 -04001127
1128 TWPartition* Part = Find_Partition_By_Path(label);
1129 if (Part == NULL)
1130 {
Ethan Yonker74db1572015-10-28 12:44:49 -05001131 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 -04001132 continue;
1133 }
1134
1135 Part->Backup_FileName = de->d_name;
1136 if (strlen(extn) > 3) {
1137 Part->Backup_FileName.resize(Part->Backup_FileName.size() - strlen(extn) + 3);
1138 }
1139
James Christopher Adduonofddbdfa2016-02-29 15:08:11 -05001140 if (!Part->Is_SubPartition)
1141 Restore_List += Part->Backup_Path + ";";
Dees_Troy63c8df72012-09-10 14:02:05 -04001142 }
1143 closedir(d);
1144
Dees_Troya13d74f2013-03-24 08:54:55 -05001145 // Set the final value
1146 DataManager::SetValue("tw_restore_list", Restore_List);
1147 DataManager::SetValue("tw_restore_selected", Restore_List);
Dees_Troy51a0e822012-09-05 15:24:24 -04001148 return;
1149}
1150
1151int TWPartitionManager::Wipe_By_Path(string Path) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001152 std::vector<TWPartition*>::iterator iter;
1153 int ret = false;
1154 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -04001155 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy63c8df72012-09-10 14:02:05 -04001156
1157 // Iterate through all partitions
1158 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -04001159 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troye58d5262012-09-21 12:27:57 -04001160 if (Path == "/and-sec")
1161 ret = (*iter)->Wipe_AndSec();
1162 else
1163 ret = (*iter)->Wipe();
Dees_Troy63c8df72012-09-10 14:02:05 -04001164 found = true;
1165 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1166 (*iter)->Wipe();
1167 }
1168 }
1169 if (found) {
1170 return ret;
1171 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001172 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 -04001173 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -04001174}
1175
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001176int TWPartitionManager::Wipe_By_Path(string Path, string New_File_System) {
1177 std::vector<TWPartition*>::iterator iter;
1178 int ret = false;
1179 bool found = false;
1180 string Local_Path = TWFunc::Get_Root_Path(Path);
1181
1182 // Iterate through all partitions
1183 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1184 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1185 if (Path == "/and-sec")
1186 ret = (*iter)->Wipe_AndSec();
1187 else
1188 ret = (*iter)->Wipe(New_File_System);
1189 found = true;
1190 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1191 (*iter)->Wipe(New_File_System);
1192 }
1193 }
1194 if (found) {
1195 return ret;
1196 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001197 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 -05001198 return false;
1199}
1200
Dees_Troy51a0e822012-09-05 15:24:24 -04001201int TWPartitionManager::Factory_Reset(void) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001202 std::vector<TWPartition*>::iterator iter;
1203 int ret = true;
1204
1205 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001206 if ((*iter)->Wipe_During_Factory_Reset && (*iter)->Is_Present) {
Ethan Yonker89583ef2015-08-26 09:01:59 -05001207#ifdef TW_OEM_BUILD
1208 if ((*iter)->Mount_Point == "/data") {
1209 if (!(*iter)->Wipe_Encryption())
1210 ret = false;
1211 } else {
1212#endif
1213 if (!(*iter)->Wipe())
1214 ret = false;
1215#ifdef TW_OEM_BUILD
1216 }
1217#endif
Dees_Troy094207a2012-09-26 12:00:39 -04001218 } else if ((*iter)->Has_Android_Secure) {
1219 if (!(*iter)->Wipe_AndSec())
1220 ret = false;
Dees_Troy63c8df72012-09-10 14:02:05 -04001221 }
1222 }
Ethan Yonker9fa76ba2016-06-30 14:05:43 -05001223 TWFunc::check_and_run_script("/sbin/factoryreset.sh", "Factory Reset Script");
Dees_Troy63c8df72012-09-10 14:02:05 -04001224 return ret;
Dees_Troy51a0e822012-09-05 15:24:24 -04001225}
1226
Dees_Troy38bd7602012-09-14 13:33:53 -04001227int TWPartitionManager::Wipe_Dalvik_Cache(void) {
1228 struct stat st;
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001229 vector <string> dir;
Dees_Troy38bd7602012-09-14 13:33:53 -04001230
1231 if (!Mount_By_Path("/data", true))
1232 return false;
1233
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001234 dir.push_back("/data/dalvik-cache");
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001235 if (Mount_By_Path("/cache", false)) {
1236 dir.push_back("/cache/dalvik-cache");
1237 dir.push_back("/cache/dc");
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001238 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001239
Dees_Troy38bd7602012-09-14 13:33:53 -04001240 TWPartition* sdext = Find_Partition_By_Path("/sd-ext");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001241 if (sdext && sdext->Is_Present && sdext->Mount(false))
1242 {
1243 if (stat("/sd-ext/dalvik-cache", &st) == 0)
1244 {
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001245 dir.push_back("/sd-ext/dalvik-cache");
1246 }
1247 }
1248
1249 gui_msg("wiping_dalvik=Wiping Dalvik Cache Directories...");
1250 for (unsigned i = 0; i < dir.size(); ++i) {
1251 if (stat(dir.at(i).c_str(), &st) == 0) {
1252 TWFunc::removeDir(dir.at(i), false);
1253 gui_msg(Msg("cleaned=Cleaned: {1}...")(dir.at(i)));
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001254 }
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001255 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001256 gui_msg("dalvik_done=-- Dalvik Cache Directories Wipe Complete!");
Dees_Troy38bd7602012-09-14 13:33:53 -04001257 return true;
1258}
1259
1260int TWPartitionManager::Wipe_Rotate_Data(void) {
1261 if (!Mount_By_Path("/data", true))
1262 return false;
1263
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001264 unlink("/data/misc/akmd*");
1265 unlink("/data/misc/rild*");
Dees_Troy2673cec2013-04-02 20:22:16 +00001266 gui_print("Rotation data wiped.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001267 return true;
1268}
1269
1270int TWPartitionManager::Wipe_Battery_Stats(void) {
1271 struct stat st;
1272
1273 if (!Mount_By_Path("/data", true))
1274 return false;
1275
1276 if (0 != stat("/data/system/batterystats.bin", &st)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001277 gui_print("No Battery Stats Found. No Need To Wipe.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001278 } else {
1279 remove("/data/system/batterystats.bin");
Dees_Troy2673cec2013-04-02 20:22:16 +00001280 gui_print("Cleared battery stats.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001281 }
1282 return true;
1283}
1284
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001285int TWPartitionManager::Wipe_Android_Secure(void) {
1286 std::vector<TWPartition*>::iterator iter;
1287 int ret = false;
1288 bool found = false;
1289
1290 // Iterate through all partitions
1291 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1292 if ((*iter)->Has_Android_Secure) {
1293 ret = (*iter)->Wipe_AndSec();
1294 found = true;
1295 }
1296 }
1297 if (found) {
1298 return ret;
1299 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001300 gui_err("no_andsec=No android secure partitions found.");
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001301 }
1302 return false;
1303}
1304
Dees_Troy38bd7602012-09-14 13:33:53 -04001305int TWPartitionManager::Format_Data(void) {
1306 TWPartition* dat = Find_Partition_By_Path("/data");
1307
1308 if (dat != NULL) {
1309 if (!dat->UnMount(true))
1310 return false;
1311
1312 return dat->Wipe_Encryption();
1313 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001314 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001315 return false;
1316 }
1317 return false;
1318}
1319
1320int TWPartitionManager::Wipe_Media_From_Data(void) {
1321 TWPartition* dat = Find_Partition_By_Path("/data");
1322
1323 if (dat != NULL) {
1324 if (!dat->Has_Data_Media) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001325 LOGERR("This device does not have /data/media\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001326 return false;
1327 }
1328 if (!dat->Mount(true))
1329 return false;
1330
Ethan Yonker74db1572015-10-28 12:44:49 -05001331 gui_msg("wiping_datamedia=Wiping internal storage -- /data/media...");
Ethan Yonker726a0202014-12-16 20:01:38 -06001332 Remove_MTP_Storage(dat->MTP_Storage_ID);
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001333 TWFunc::removeDir("/data/media", false);
xiaolu9416f4f2015-06-04 08:22:23 +08001334 dat->Recreate_Media_Folder();
Ethan Yonker726a0202014-12-16 20:01:38 -06001335 Add_MTP_Storage(dat->MTP_Storage_ID);
Dees_Troy38bd7602012-09-14 13:33:53 -04001336 return true;
1337 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001338 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001339 return false;
1340 }
1341 return false;
1342}
1343
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001344int TWPartitionManager::Repair_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)->Repair();
1357 found = true;
1358 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1359 (*iter)->Repair();
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 Yonker87c7bac2014-05-25 21:41:08 -05001366 } else {
1367 LOGINFO("Repair: Unable to find partition for path '%s'\n", Local_Path.c_str());
1368 }
1369 return false;
1370}
1371
Ethan Yonkera2719152015-05-28 09:44:41 -05001372int TWPartitionManager::Resize_By_Path(string Path, bool Display_Error) {
1373 std::vector<TWPartition*>::iterator iter;
1374 int ret = false;
1375 bool found = false;
1376 string Local_Path = TWFunc::Get_Root_Path(Path);
1377
1378 if (Local_Path == "/tmp" || Local_Path == "/")
1379 return true;
1380
1381 // Iterate through all partitions
1382 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1383 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1384 ret = (*iter)->Resize();
1385 found = true;
1386 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1387 (*iter)->Resize();
1388 }
1389 }
1390 if (found) {
1391 return ret;
1392 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001393 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 -05001394 } else {
1395 LOGINFO("Resize: Unable to find partition for path '%s'\n", Local_Path.c_str());
1396 }
1397 return false;
1398}
1399
Dees_Troy51a0e822012-09-05 15:24:24 -04001400void TWPartitionManager::Update_System_Details(void) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001401 std::vector<TWPartition*>::iterator iter;
Dees_Troy51127312012-09-08 13:08:49 -04001402 int data_size = 0;
Dees_Troy5bf43922012-09-07 16:07:55 -04001403
Ethan Yonker74db1572015-10-28 12:44:49 -05001404 gui_msg("update_part_details=Updating partition details...");
Dees_Troy5bf43922012-09-07 16:07:55 -04001405 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker1b190162016-12-05 15:25:19 -06001406 (*iter)->Update_Size(true);
Dees_Troy51127312012-09-08 13:08:49 -04001407 if ((*iter)->Can_Be_Mounted) {
Dees_Troy51127312012-09-08 13:08:49 -04001408 if ((*iter)->Mount_Point == "/system") {
1409 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1410 DataManager::SetValue(TW_BACKUP_SYSTEM_SIZE, backup_display_size);
1411 } else if ((*iter)->Mount_Point == "/data" || (*iter)->Mount_Point == "/datadata") {
1412 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
1413 } else if ((*iter)->Mount_Point == "/cache") {
1414 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1415 DataManager::SetValue(TW_BACKUP_CACHE_SIZE, backup_display_size);
1416 } else if ((*iter)->Mount_Point == "/sd-ext") {
1417 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1418 DataManager::SetValue(TW_BACKUP_SDEXT_SIZE, backup_display_size);
1419 if ((*iter)->Backup_Size == 0) {
1420 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 0);
1421 DataManager::SetValue(TW_BACKUP_SDEXT_VAR, 0);
1422 } else
1423 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 1);
Dees_Troye58d5262012-09-21 12:27:57 -04001424 } else if ((*iter)->Has_Android_Secure) {
Dees_Troy8170a922012-09-18 15:40:25 -04001425 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troye58d5262012-09-21 12:27:57 -04001426 DataManager::SetValue(TW_BACKUP_ANDSEC_SIZE, backup_display_size);
Dees_Troy8170a922012-09-18 15:40:25 -04001427 if ((*iter)->Backup_Size == 0) {
1428 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 0);
1429 DataManager::SetValue(TW_BACKUP_ANDSEC_VAR, 0);
1430 } else
1431 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 1);
Dees_Troy2c50e182012-09-26 20:05:28 -04001432 } else if ((*iter)->Mount_Point == "/boot") {
1433 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1434 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1435 if ((*iter)->Backup_Size == 0) {
1436 DataManager::SetValue("tw_has_boot_partition", 0);
1437 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1438 } else
1439 DataManager::SetValue("tw_has_boot_partition", 1);
Dees_Troy51127312012-09-08 13:08:49 -04001440 }
Dees_Troyaa9cc402012-10-13 12:14:05 -04001441 } else {
1442 // Handle unmountable partitions in case we reset defaults
1443 if ((*iter)->Mount_Point == "/boot") {
1444 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1445 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1446 if ((*iter)->Backup_Size == 0) {
1447 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 0);
1448 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1449 } else
1450 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 1);
1451 } else if ((*iter)->Mount_Point == "/recovery") {
1452 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1453 DataManager::SetValue(TW_BACKUP_RECOVERY_SIZE, backup_display_size);
1454 if ((*iter)->Backup_Size == 0) {
1455 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 0);
1456 DataManager::SetValue(TW_BACKUP_RECOVERY_VAR, 0);
1457 } else
1458 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 1);
Gary Peck82599a82012-11-21 16:23:12 -08001459 } else if ((*iter)->Mount_Point == "/data") {
1460 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troyaa9cc402012-10-13 12:14:05 -04001461 }
Dees_Troy51127312012-09-08 13:08:49 -04001462 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001463 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001464 gui_msg("update_part_details_done=...done");
Dees_Troy51127312012-09-08 13:08:49 -04001465 DataManager::SetValue(TW_BACKUP_DATA_SIZE, data_size);
1466 string current_storage_path = DataManager::GetCurrentStoragePath();
1467 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
Dees_Troy8170a922012-09-18 15:40:25 -04001468 if (FreeStorage != NULL) {
1469 // Attempt to mount storage
1470 if (!FreeStorage->Mount(false)) {
Matt Mower2d50cad2015-12-03 22:26:55 -06001471 gui_msg(Msg(msg::kError, "unable_to_mount_storage=Unable to mount storage"));
1472 DataManager::SetValue(TW_STORAGE_FREE_SIZE, 0);
Dees_Troy8170a922012-09-18 15:40:25 -04001473 } else {
1474 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
1475 }
1476 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001477 LOGINFO("Unable to find storage partition '%s'.\n", current_storage_path.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -04001478 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001479 if (!Write_Fstab())
Dees_Troy2673cec2013-04-02 20:22:16 +00001480 LOGERR("Error creating fstab\n");
Dees_Troy51a0e822012-09-05 15:24:24 -04001481 return;
1482}
1483
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001484void TWPartitionManager::Post_Decrypt(const string& Block_Device) {
1485 TWPartition* dat = Find_Partition_By_Path("/data");
1486 if (dat != NULL) {
1487 DataManager::SetValue(TW_IS_DECRYPTED, 1);
1488 dat->Is_Decrypted = true;
1489 if (!Block_Device.empty()) {
1490 dat->Decrypted_Block_Device = Block_Device;
1491 gui_msg(Msg("decrypt_success_dev=Data successfully decrypted, new block device: '{1}'")(Block_Device));
1492 } else {
1493 gui_msg("decrypt_success_nodev=Data successfully decrypted");
1494 }
1495 dat->Setup_File_System(false);
1496 dat->Current_File_System = dat->Fstab_File_System; // Needed if we're ignoring blkid because encrypted devices start out as emmc
1497
1498 // Sleep for a bit so that the device will be ready
1499 sleep(1);
1500 if (dat->Has_Data_Media && dat->Mount(false) && TWFunc::Path_Exists("/data/media/0")) {
1501 dat->Storage_Path = "/data/media/0";
1502 dat->Symlink_Path = dat->Storage_Path;
1503 DataManager::SetValue("tw_storage_path", "/data/media/0");
1504 DataManager::SetValue("tw_settings_path", "/data/media/0");
1505 dat->UnMount(false);
1506 Output_Partition(dat);
1507 }
1508 Update_System_Details();
1509 UnMount_Main_Partitions();
1510 } else
1511 LOGERR("Unable to locate data partition.\n");
1512}
1513
Dees_Troy51a0e822012-09-05 15:24:24 -04001514int TWPartitionManager::Decrypt_Device(string Password) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001515#ifdef TW_INCLUDE_CRYPTO
Ethan Yonkerddb63e22017-01-19 14:01:57 -06001516 char crypto_state[PROPERTY_VALUE_MAX], crypto_blkdev[PROPERTY_VALUE_MAX];
Ethan Yonker253368a2014-11-25 15:00:52 -06001517 std::vector<TWPartition*>::iterator iter;
Dees_Troy5bf43922012-09-07 16:07:55 -04001518
Ethan Yonker253368a2014-11-25 15:00:52 -06001519 // Mount any partitions that need to be mounted for decrypt
1520 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1521 if ((*iter)->Mount_To_Decrypt) {
1522 (*iter)->Mount(true);
1523 }
a39552696ff55ce2013-01-08 16:14:56 +00001524 }
a39552696ff55ce2013-01-08 16:14:56 +00001525
Ethan Yonkera1de1492015-11-04 11:58:27 -06001526 property_get("ro.crypto.state", crypto_state, "error");
1527 if (strcmp(crypto_state, "error") == 0) {
1528 property_set("ro.crypto.state", "encrypted");
1529 // Sleep for a bit so that services can start if needed
1530 sleep(1);
1531 }
1532
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001533 if (DataManager::GetIntValue(TW_IS_FBE)) {
1534#ifdef TW_INCLUDE_FBE
1535 if (!Mount_By_Path("/data", true)) // /data has to be mounted for FBE
1536 return -1;
1537 int retry_count = 10;
1538 while (!TWFunc::Path_Exists("/data/system/users/gatekeeper.password.key") && --retry_count)
1539 usleep(2000); // A small sleep is needed after mounting /data to ensure reliable decrypt... maybe because of DE?
1540 int user_id = DataManager::GetIntValue("tw_decrypt_user_id");
1541 LOGINFO("Decrypting FBE for user %i\n", user_id);
1542 if (Decrypt_User(user_id, Password)) {
1543 Post_Decrypt("");
1544 return 0;
1545 }
1546#else
1547 LOGERR("FBE support is not present\n");
1548#endif
1549 return -1;
1550 }
1551
Ethan Yonkerddb63e22017-01-19 14:01:57 -06001552 int pwret = -1;
1553 pid_t pid = fork();
1554 if (pid < 0) {
1555 LOGERR("fork failed\n");
1556 return -1;
1557 } else if (pid == 0) {
1558 // Child process
1559 char cPassword[255];
1560 strcpy(cPassword, Password.c_str());
1561 int ret = cryptfs_check_passwd(cPassword);
1562 exit(ret);
1563 } else {
1564 // Parent
1565 int status;
1566 if (TWFunc::Wait_For_Child_Timeout(pid, &status, "Decrypt", 30))
1567 pwret = -1;
1568 else
1569 pwret = WEXITSTATUS(status) ? -1 : 0;
1570 }
a39552696ff55ce2013-01-08 16:14:56 +00001571
Ethan Yonker253368a2014-11-25 15:00:52 -06001572 // Unmount any partitions that were needed for decrypt
1573 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1574 if ((*iter)->Mount_To_Decrypt) {
1575 (*iter)->UnMount(false);
1576 }
1577 }
1578
a39552696ff55ce2013-01-08 16:14:56 +00001579 if (pwret != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001580 gui_err("fail_decrypt=Failed to decrypt data.");
Dees_Troy5bf43922012-09-07 16:07:55 -04001581 return -1;
1582 }
a39552696ff55ce2013-01-08 16:14:56 +00001583
Dees_Troy5bf43922012-09-07 16:07:55 -04001584 property_get("ro.crypto.fs_crypto_blkdev", crypto_blkdev, "error");
1585 if (strcmp(crypto_blkdev, "error") == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001586 LOGERR("Error retrieving decrypted data block device.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04001587 } else {
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001588 Post_Decrypt(crypto_blkdev);
Dees_Troy5bf43922012-09-07 16:07:55 -04001589 }
1590 return 0;
1591#else
Ethan Yonker74db1572015-10-28 12:44:49 -05001592 gui_err("no_crypto_support=No crypto support was compiled into this build.");
Dees_Troy5bf43922012-09-07 16:07:55 -04001593 return -1;
1594#endif
Dees_Troy51a0e822012-09-05 15:24:24 -04001595 return 1;
Dees_Troy51127312012-09-08 13:08:49 -04001596}
1597
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001598int TWPartitionManager::Fix_Contexts(void) {
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001599 std::vector<TWPartition*>::iterator iter;
1600 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1601 if ((*iter)->Has_Data_Media) {
1602 if ((*iter)->Mount(true)) {
1603 if (fixContexts::fixDataMediaContexts((*iter)->Mount_Point) != 0)
1604 return -1;
1605 }
1606 }
1607 }
Dees_Troy1a650e62012-10-19 20:54:32 -04001608 UnMount_Main_Partitions();
Ethan Yonker74db1572015-10-28 12:44:49 -05001609 gui_msg("done=Done.");
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001610 return 0;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04001611}
1612
Ethan Yonker66a19492015-12-10 10:19:45 -06001613TWPartition* TWPartitionManager::Find_Next_Storage(string Path, bool Exclude_Data_Media) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001614 std::vector<TWPartition*>::iterator iter = Partitions.begin();
1615
1616 if (!Path.empty()) {
1617 string Search_Path = TWFunc::Get_Root_Path(Path);
1618 for (; iter != Partitions.end(); iter++) {
Matt Mower9a561dd2016-02-22 21:25:51 -06001619 if ((*iter)->Mount_Point == Search_Path) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001620 iter++;
1621 break;
1622 }
1623 }
1624 }
1625
1626 for (; iter != Partitions.end(); iter++) {
Ethan Yonker66a19492015-12-10 10:19:45 -06001627 if (Exclude_Data_Media && (*iter)->Has_Data_Media) {
1628 // do nothing, do not return this type of partition
1629 } else if ((*iter)->Is_Storage && (*iter)->Is_Present) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001630 return (*iter);
1631 }
1632 }
1633
1634 return NULL;
1635}
1636
Dees_Troyd21618c2012-10-14 18:48:49 -04001637int TWPartitionManager::Open_Lun_File(string Partition_Path, string Lun_File) {
Dees_Troyd21618c2012-10-14 18:48:49 -04001638 TWPartition* Part = Find_Partition_By_Path(Partition_Path);
1639
1640 if (Part == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001641 LOGINFO("Unable to locate '%s' for USB storage mode.", Partition_Path.c_str());
1642 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 -04001643 return false;
1644 }
Ethan Yonker47360be2014-04-01 10:34:34 -05001645 LOGINFO("USB mount '%s', '%s' > '%s'\n", Partition_Path.c_str(), Part->Actual_Block_Device.c_str(), Lun_File.c_str());
1646 if (!Part->UnMount(true) || !Part->Is_Present)
Dees_Troyd21618c2012-10-14 18:48:49 -04001647 return false;
1648
Dees_Troy2673cec2013-04-02 20:22:16 +00001649 if (TWFunc::write_file(Lun_File, Part->Actual_Block_Device)) {
1650 LOGERR("Unable to write to ums lunfile '%s': (%s)\n", Lun_File.c_str(), strerror(errno));
Dees_Troyd21618c2012-10-14 18:48:49 -04001651 return false;
1652 }
Dees_Troyd21618c2012-10-14 18:48:49 -04001653 return true;
1654}
1655
Dees_Troy8170a922012-09-18 15:40:25 -04001656int TWPartitionManager::usb_storage_enable(void) {
Dees_Troy8170a922012-09-18 15:40:25 -04001657 char lun_file[255];
Dees_Troyd21618c2012-10-14 18:48:49 -04001658 bool has_multiple_lun = false;
Dees_Troy8170a922012-09-18 15:40:25 -04001659
Ethan Yonker47360be2014-04-01 10:34:34 -05001660 string Lun_File_str = CUSTOM_LUN_FILE;
1661 size_t found = Lun_File_str.find("%");
1662 if (found != string::npos) {
1663 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
1664 if (TWFunc::Path_Exists(lun_file))
1665 has_multiple_lun = true;
1666 }
Ethan Yonker726a0202014-12-16 20:01:38 -06001667 mtp_was_enabled = TWFunc::Toggle_MTP(false); // Must disable MTP for USB Storage
Ethan Yonker47360be2014-04-01 10:34:34 -05001668 if (!has_multiple_lun) {
1669 LOGINFO("Device doesn't have multiple lun files, mount current storage\n");
1670 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
1671 if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) == "/data") {
Ethan Yonker66a19492015-12-10 10:19:45 -06001672 TWPartition* Mount = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05001673 if (Mount) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001674 if (!Open_Lun_File(Mount->Mount_Point, lun_file)) {
1675 goto error_handle;
1676 }
Ethan Yonker47360be2014-04-01 10:34:34 -05001677 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001678 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001679 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05001680 }
1681 } else if (!Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file)) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001682 goto error_handle;
Dees_Troy8170a922012-09-18 15:40:25 -04001683 }
Dees_Troy8170a922012-09-18 15:40:25 -04001684 } else {
Ethan Yonker47360be2014-04-01 10:34:34 -05001685 LOGINFO("Device has multiple lun files\n");
1686 TWPartition* Mount1;
1687 TWPartition* Mount2;
Dees_Troy8170a922012-09-18 15:40:25 -04001688 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
Ethan Yonker66a19492015-12-10 10:19:45 -06001689 Mount1 = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05001690 if (Mount1) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001691 if (!Open_Lun_File(Mount1->Mount_Point, lun_file)) {
1692 goto error_handle;
1693 }
Matt Moweree717062014-04-26 01:46:46 -05001694 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
Ethan Yonker66a19492015-12-10 10:19:45 -06001695 Mount2 = Find_Next_Storage(Mount1->Mount_Point, true);
Matt Mower1fdcdb72016-01-25 20:53:47 -06001696 if (Mount2 && Mount2->Mount_Point != Mount1->Mount_Point) {
Matt Moweree717062014-04-26 01:46:46 -05001697 Open_Lun_File(Mount2->Mount_Point, lun_file);
Ethan Yonker47360be2014-04-01 10:34:34 -05001698 }
1699 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001700 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001701 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05001702 }
Dees_Troy8170a922012-09-18 15:40:25 -04001703 }
Matt Mowerb6ef4782014-11-06 02:24:36 -06001704 property_set("sys.storage.ums_enabled", "1");
HandyMenny37d42992014-10-26 22:15:59 +01001705 property_set("sys.usb.config", "mass_storage,adb");
Dees_Troy8170a922012-09-18 15:40:25 -04001706 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001707error_handle:
1708 if (mtp_was_enabled)
1709 if (!Enable_MTP())
1710 Disable_MTP();
1711 return false;
Dees_Troy8170a922012-09-18 15:40:25 -04001712}
1713
1714int TWPartitionManager::usb_storage_disable(void) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001715 int index, ret;
1716 char lun_file[255], ch[2] = {0, 0};
1717 string str = ch;
Dees_Troy8170a922012-09-18 15:40:25 -04001718
1719 for (index=0; index<2; index++) {
1720 sprintf(lun_file, CUSTOM_LUN_FILE, index);
Dees_Troy2673cec2013-04-02 20:22:16 +00001721 ret = TWFunc::write_file(lun_file, str);
Dees_Troy2673cec2013-04-02 20:22:16 +00001722 if (ret < 0) {
1723 break;
Dees_Troy8170a922012-09-18 15:40:25 -04001724 }
Dees_Troy8170a922012-09-18 15:40:25 -04001725 }
Dees_Troye58d5262012-09-21 12:27:57 -04001726 Mount_All_Storage();
1727 Update_System_Details();
Dees_Troycfd73ef2012-10-12 16:52:00 -04001728 UnMount_Main_Partitions();
Matt Mowerd9cb9062013-12-14 20:34:45 -06001729 property_set("sys.storage.ums_enabled", "0");
HandyMenny37d42992014-10-26 22:15:59 +01001730 property_set("sys.usb.config", "adb");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001731 if (mtp_was_enabled)
1732 if (!Enable_MTP())
1733 Disable_MTP();
Dees_Troy2673cec2013-04-02 20:22:16 +00001734 if (ret < 0 && index == 0) {
1735 LOGERR("Unable to write to ums lunfile '%s'.", lun_file);
1736 return false;
1737 } else {
1738 return true;
1739 }
Dees_Troy8170a922012-09-18 15:40:25 -04001740 return true;
Dees_Troy812660f2012-09-20 09:55:17 -04001741}
1742
1743void TWPartitionManager::Mount_All_Storage(void) {
1744 std::vector<TWPartition*>::iterator iter;
1745
1746 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1747 if ((*iter)->Is_Storage)
1748 (*iter)->Mount(false);
1749 }
Dees_Troy2c50e182012-09-26 20:05:28 -04001750}
Dees_Troy9350b8d2012-09-27 12:38:38 -04001751
Dees_Troyd0384ef2012-10-12 12:15:42 -04001752void TWPartitionManager::UnMount_Main_Partitions(void) {
1753 // Unmounts system and data if data is not data/media
1754 // Also unmounts boot if boot is mountable
Dees_Troy2673cec2013-04-02 20:22:16 +00001755 LOGINFO("Unmounting main partitions...\n");
Dees_Troyd0384ef2012-10-12 12:15:42 -04001756
1757 TWPartition* Boot_Partition = Find_Partition_By_Path("/boot");
1758
1759 UnMount_By_Path("/system", true);
Ethan Yonker6277c792014-09-15 14:54:30 -05001760 if (!datamedia)
1761 UnMount_By_Path("/data", true);
1762
Dees_Troyd0384ef2012-10-12 12:15:42 -04001763 if (Boot_Partition != NULL && Boot_Partition->Can_Be_Mounted)
1764 Boot_Partition->UnMount(true);
1765}
1766
Dees_Troy9350b8d2012-09-27 12:38:38 -04001767int TWPartitionManager::Partition_SDCard(void) {
Matt Mower23d8aae2017-01-06 14:30:33 -06001768 char temp[255];
Ethan Yonker483e9f42016-01-11 22:21:18 -06001769 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 -04001770 int ext, swap, total_size = 0, fat_size;
Dees_Troy9350b8d2012-09-27 12:38:38 -04001771
Ethan Yonker74db1572015-10-28 12:44:49 -05001772 gui_msg("start_partition_sd=Partitioning SD Card...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06001773
1774 // Locate and validate device to partition
1775 TWPartition* SDCard = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
1776
Ethan Yonker66a19492015-12-10 10:19:45 -06001777 if (SDCard->Is_Adopted_Storage)
1778 SDCard->Revert_Adopted();
1779
Ethan Yonker1eff6cd2014-09-15 13:30:42 -05001780 if (SDCard == NULL || !SDCard->Removable || SDCard->Has_Data_Media) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001781 gui_err("partition_sd_locate=Unable to locate device to partition.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001782 return false;
1783 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001784
1785 // Unmount everything
Dees_Troy9350b8d2012-09-27 12:38:38 -04001786 if (!SDCard->UnMount(true))
1787 return false;
1788 TWPartition* SDext = Find_Partition_By_Path("/sd-ext");
1789 if (SDext != NULL) {
1790 if (!SDext->UnMount(true))
1791 return false;
1792 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001793 char* swappath = getenv("SWAPPATH");
1794 if (swappath != NULL) {
1795 LOGINFO("Unmounting swap at '%s'\n", swappath);
1796 umount(swappath);
1797 }
Vojtech Bocek05534202013-09-11 08:11:56 +02001798
Ethan Yonker483e9f42016-01-11 22:21:18 -06001799 // Determine block device
1800 if (SDCard->Alternate_Block_Device.empty()) {
1801 SDCard->Find_Actual_Block_Device();
1802 Device = SDCard->Actual_Block_Device;
1803 // Just use the root block device
1804 Device.resize(strlen("/dev/block/mmcblkX"));
1805 } else {
1806 Device = SDCard->Alternate_Block_Device;
1807 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04001808
1809 // Find the size of the block device:
Ethan Yonker483e9f42016-01-11 22:21:18 -06001810 total_size = (int)(TWFunc::IOCTL_Get_Block_Size(Device.c_str()) / (1048576));
Dees_Troy9350b8d2012-09-27 12:38:38 -04001811
1812 DataManager::GetValue("tw_sdext_size", ext);
1813 DataManager::GetValue("tw_swap_size", swap);
1814 DataManager::GetValue("tw_sdpart_file_system", ext_format);
1815 fat_size = total_size - ext - swap;
Ethan Yonker483e9f42016-01-11 22:21:18 -06001816 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);
1817
1818 // Determine partition sizes
1819 if (swap == 0 && ext == 0) {
1820 fat_str = "-0";
1821 } else {
1822 memset(temp, 0, sizeof(temp));
1823 sprintf(temp, "%i", fat_size);
1824 fat_str = temp;
1825 fat_str += "MB";
1826 }
1827 if (swap == 0) {
1828 ext_str = "-0";
1829 } else {
1830 memset(temp, 0, sizeof(temp));
1831 sprintf(temp, "%i", ext);
1832 ext_str = "+";
1833 ext_str += temp;
1834 ext_str += "MB";
1835 }
1836
Dees_Troy9350b8d2012-09-27 12:38:38 -04001837 if (ext + swap > total_size) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001838 gui_err("ext_swap_size=EXT + Swap size is larger than sdcard size.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001839 return false;
1840 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001841
Ethan Yonker74db1572015-10-28 12:44:49 -05001842 gui_msg("remove_part_table=Removing partition table...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06001843 Command = "sgdisk --zap-all " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00001844 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001845 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001846 gui_err("unable_rm_part=Unable to remove partition table.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001847 Update_System_Details();
1848 return false;
1849 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001850 gui_msg(Msg("create_part=Creating {1} partition...")("FAT32"));
Captain Throwback9643a802016-05-27 11:44:51 -04001851 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 +00001852 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001853 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001854 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("FAT32"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04001855 return false;
1856 }
1857 if (ext > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001858 gui_msg(Msg("create_part=Creating {1} partition...")("EXT"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06001859 Command = "sgdisk --new=0:0:" + ext_str + " --change-name=0:\"Linux filesystem\" " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00001860 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001861 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001862 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("EXT"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04001863 Update_System_Details();
1864 return false;
1865 }
1866 }
1867 if (swap > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001868 gui_msg(Msg("create_part=Creating {1} partition...")("swap"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06001869 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 +00001870 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001871 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001872 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("swap"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04001873 Update_System_Details();
1874 return false;
1875 }
1876 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001877
1878 // Convert GPT to MBR
1879 Command = "sgdisk --gpttombr " + Device;
1880 if (TWFunc::Exec_Cmd(Command) != 0)
1881 LOGINFO("Failed to covert partition GPT to MBR\n");
1882
1883 // Tell the kernel to rescan the partition table
1884 int fd = open(Device.c_str(), O_RDONLY);
1885 ioctl(fd, BLKRRPART, 0);
1886 close(fd);
1887
1888 string format_device = Device;
1889 if (Device.substr(0, 17) == "/dev/block/mmcblk")
1890 format_device += "p";
1891
1892 // Format new partitions to proper file system
1893 if (fat_size > 0) {
1894 Command = "mkfs.fat " + format_device + "1";
1895 TWFunc::Exec_Cmd(Command);
1896 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04001897 if (ext > 0) {
1898 if (SDext == NULL) {
Ethan Yonker483e9f42016-01-11 22:21:18 -06001899 Command = "mke2fs -t " + ext_format + " -m 0 " + format_device + "2";
1900 gui_msg(Msg("format_sdext_as=Formatting sd-ext as {1}...")(ext_format));
1901 LOGINFO("Formatting sd-ext after partitioning, command: '%s'\n", Command.c_str());
1902 TWFunc::Exec_Cmd(Command);
1903 } else {
1904 SDext->Wipe(ext_format);
Dees_Troy9350b8d2012-09-27 12:38:38 -04001905 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001906 }
1907 if (swap > 0) {
1908 Command = "mkswap " + format_device;
1909 if (ext > 0)
1910 Command += "3";
1911 else
1912 Command += "2";
Vojtech Bocek05534202013-09-11 08:11:56 +02001913 TWFunc::Exec_Cmd(Command);
Dees_Troy9350b8d2012-09-27 12:38:38 -04001914 }
1915
Ethan Yonker483e9f42016-01-11 22:21:18 -06001916 // recreate TWRP folder and rewrite settings - these will be gone after sdcard is partitioned
1917 if (SDCard->Mount(true)) {
1918 string TWRP_Folder = SDCard->Mount_Point + "/TWRP";
1919 mkdir(TWRP_Folder.c_str(), 0777);
1920 DataManager::Flush();
1921 }
1922
Dees_Troy9350b8d2012-09-27 12:38:38 -04001923 Update_System_Details();
Ethan Yonker74db1572015-10-28 12:44:49 -05001924 gui_msg("part_complete=Partitioning complete.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001925 return true;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04001926}
Dees_Troya13d74f2013-03-24 08:54:55 -05001927
1928void TWPartitionManager::Get_Partition_List(string ListType, std::vector<PartitionList> *Partition_List) {
1929 std::vector<TWPartition*>::iterator iter;
1930 if (ListType == "mount") {
1931 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1932 if ((*iter)->Can_Be_Mounted && !(*iter)->Is_SubPartition) {
1933 struct PartitionList part;
1934 part.Display_Name = (*iter)->Display_Name;
1935 part.Mount_Point = (*iter)->Mount_Point;
1936 part.selected = (*iter)->Is_Mounted();
1937 Partition_List->push_back(part);
1938 }
1939 }
1940 } else if (ListType == "storage") {
1941 char free_space[255];
1942 string Current_Storage = DataManager::GetCurrentStoragePath();
1943 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1944 if ((*iter)->Is_Storage) {
1945 struct PartitionList part;
1946 sprintf(free_space, "%llu", (*iter)->Free / 1024 / 1024);
1947 part.Display_Name = (*iter)->Storage_Name + " (";
1948 part.Display_Name += free_space;
1949 part.Display_Name += "MB)";
1950 part.Mount_Point = (*iter)->Storage_Path;
1951 if ((*iter)->Storage_Path == Current_Storage)
1952 part.selected = 1;
1953 else
1954 part.selected = 0;
1955 Partition_List->push_back(part);
1956 }
1957 }
1958 } else if (ListType == "backup") {
1959 char backup_size[255];
Dees_Troy9e0b71c2013-04-08 13:35:37 +00001960 unsigned long long Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -05001961 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy9e0b71c2013-04-08 13:35:37 +00001962 if ((*iter)->Can_Be_Backed_Up && !(*iter)->Is_SubPartition && (*iter)->Is_Present) {
Dees_Troya13d74f2013-03-24 08:54:55 -05001963 struct PartitionList part;
Dees_Troy9e0b71c2013-04-08 13:35:37 +00001964 Backup_Size = (*iter)->Backup_Size;
1965 if ((*iter)->Has_SubPartition) {
1966 std::vector<TWPartition*>::iterator subpart;
1967
1968 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
1969 if ((*subpart)->Is_SubPartition && (*subpart)->Can_Be_Backed_Up && (*subpart)->Is_Present && (*subpart)->SubPartition_Of == (*iter)->Mount_Point)
1970 Backup_Size += (*subpart)->Backup_Size;
1971 }
1972 }
1973 sprintf(backup_size, "%llu", Backup_Size / 1024 / 1024);
Dees_Troya13d74f2013-03-24 08:54:55 -05001974 part.Display_Name = (*iter)->Backup_Display_Name + " (";
1975 part.Display_Name += backup_size;
1976 part.Display_Name += "MB)";
1977 part.Mount_Point = (*iter)->Backup_Path;
1978 part.selected = 0;
1979 Partition_List->push_back(part);
1980 }
1981 }
1982 } else if (ListType == "restore") {
1983 string Restore_List, restore_path;
1984 TWPartition* restore_part = NULL;
1985
1986 DataManager::GetValue("tw_restore_list", Restore_List);
1987 if (!Restore_List.empty()) {
1988 size_t start_pos = 0, end_pos = Restore_List.find(";", start_pos);
1989 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1990 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
Dees_Troy59df9262013-06-19 14:53:57 -05001991 if ((restore_part = Find_Partition_By_Path(restore_path)) != NULL) {
Dees Troy4159aed2014-02-28 17:24:43 +00001992 if ((restore_part->Backup_Name == "recovery" && !restore_part->Can_Be_Backed_Up) || restore_part->Is_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -05001993 // Don't allow restore of recovery (causes problems on some devices)
Dees_Troy59df9262013-06-19 14:53:57 -05001994 // Don't add subpartitions to the list of items
Dees_Troya13d74f2013-03-24 08:54:55 -05001995 } else {
1996 struct PartitionList part;
1997 part.Display_Name = restore_part->Backup_Display_Name;
1998 part.Mount_Point = restore_part->Backup_Path;
1999 part.selected = 1;
2000 Partition_List->push_back(part);
2001 }
2002 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002003 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05002004 }
2005 start_pos = end_pos + 1;
2006 end_pos = Restore_List.find(";", start_pos);
2007 }
2008 }
2009 } else if (ListType == "wipe") {
2010 struct PartitionList dalvik;
Ethan Yonker74db1572015-10-28 12:44:49 -05002011 dalvik.Display_Name = gui_parse_text("{@dalvik}");
Dees_Troya13d74f2013-03-24 08:54:55 -05002012 dalvik.Mount_Point = "DALVIK";
2013 dalvik.selected = 0;
2014 Partition_List->push_back(dalvik);
2015 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2016 if ((*iter)->Wipe_Available_in_GUI && !(*iter)->Is_SubPartition) {
2017 struct PartitionList part;
2018 part.Display_Name = (*iter)->Display_Name;
2019 part.Mount_Point = (*iter)->Mount_Point;
2020 part.selected = 0;
2021 Partition_List->push_back(part);
2022 }
2023 if ((*iter)->Has_Android_Secure) {
2024 struct PartitionList part;
2025 part.Display_Name = (*iter)->Backup_Display_Name;
2026 part.Mount_Point = (*iter)->Backup_Path;
2027 part.selected = 0;
2028 Partition_List->push_back(part);
2029 }
Dees_Troy74fb2e92013-04-15 14:35:47 +00002030 if ((*iter)->Has_Data_Media) {
2031 struct PartitionList datamedia;
2032 datamedia.Display_Name = (*iter)->Storage_Name;
2033 datamedia.Mount_Point = "INTERNAL";
2034 datamedia.selected = 0;
2035 Partition_List->push_back(datamedia);
2036 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002037 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002038 } else if (ListType == "flashimg") {
2039 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2040 if ((*iter)->Can_Flash_Img && (*iter)->Is_Present) {
2041 struct PartitionList part;
2042 part.Display_Name = (*iter)->Backup_Display_Name;
2043 part.Mount_Point = (*iter)->Backup_Path;
2044 part.selected = 0;
2045 Partition_List->push_back(part);
2046 }
2047 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002048 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00002049 LOGERR("Unknown list type '%s' requested for TWPartitionManager::Get_Partition_List\n", ListType.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -05002050 }
2051}
2052
2053int TWPartitionManager::Fstab_Processed(void) {
2054 return Partitions.size();
2055}
Dees_Troyd93bda52013-07-03 19:55:19 +00002056
2057void TWPartitionManager::Output_Storage_Fstab(void) {
2058 std::vector<TWPartition*>::iterator iter;
2059 char storage_partition[255];
2060 string Temp;
2061 FILE *fp = fopen("/cache/recovery/storage.fstab", "w");
2062
2063 if (fp == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002064 gui_msg(Msg(msg::kError, "unable_to_open=Unable to open '{1}'.")("/cache/recovery/storage.fstab"));
Dees_Troyd93bda52013-07-03 19:55:19 +00002065 return;
2066 }
2067
2068 // Iterate through all partitions
2069 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2070 if ((*iter)->Is_Storage) {
2071 Temp = (*iter)->Storage_Path + ";" + (*iter)->Storage_Name + ";\n";
2072 strcpy(storage_partition, Temp.c_str());
2073 fwrite(storage_partition, sizeof(storage_partition[0]), strlen(storage_partition) / sizeof(storage_partition[0]), fp);
2074 }
2075 }
2076 fclose(fp);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02002077}
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002078
2079TWPartition *TWPartitionManager::Get_Default_Storage_Partition()
2080{
2081 TWPartition *res = NULL;
2082 for (std::vector<TWPartition*>::iterator iter = Partitions.begin(); iter != Partitions.end(); ++iter) {
Matt Mowera8a89d12016-12-30 18:10:37 -06002083 if (!(*iter)->Is_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002084 continue;
2085
Matt Mowera8a89d12016-12-30 18:10:37 -06002086 if ((*iter)->Is_Settings_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002087 return *iter;
2088
Matt Mowera8a89d12016-12-30 18:10:37 -06002089 if (!res)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002090 res = *iter;
2091 }
2092 return res;
2093}
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002094
2095bool TWPartitionManager::Enable_MTP(void) {
2096#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002097 if (mtppid) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002098 gui_err("mtp_already_enabled=MTP already enabled");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002099 return true;
2100 }
Ethan Yonker726a0202014-12-16 20:01:38 -06002101
2102 int mtppipe[2];
2103
2104 if (pipe(mtppipe) < 0) {
2105 LOGERR("Error creating MTP pipe\n");
2106 return false;
2107 }
2108
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002109 char old_value[PROPERTY_VALUE_MAX];
Matt Mowere07f0102017-02-23 17:40:00 -06002110 property_get("sys.usb.config", old_value, "");
2111 if (strcmp(old_value, "mtp,adb") != 0) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002112 char vendor[PROPERTY_VALUE_MAX];
2113 char product[PROPERTY_VALUE_MAX];
2114 property_set("sys.usb.config", "none");
2115 property_get("usb.vendor", vendor, "18D1");
2116 property_get("usb.product.mtpadb", product, "4EE2");
2117 string vendorstr = vendor;
2118 string productstr = product;
2119 TWFunc::write_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2120 TWFunc::write_file("/sys/class/android_usb/android0/idProduct", productstr);
2121 property_set("sys.usb.config", "mtp,adb");
2122 }
Matt Mower653a1702017-02-16 10:38:52 -06002123 /* To enable MTP debug, use the twrp command line feature:
Ethan Yonker6d154c42014-09-03 14:19:43 -05002124 * twrp set tw_mtp_debug 1
2125 */
2126 twrpMtp *mtp = new twrpMtp(DataManager::GetIntValue("tw_mtp_debug"));
Ethan Yonker1b039202015-01-30 10:08:48 -06002127 mtppid = mtp->forkserver(mtppipe);
2128 if (mtppid) {
2129 close(mtppipe[0]); // Host closes read side
2130 mtp_write_fd = mtppipe[1];
2131 DataManager::SetValue("tw_mtp_enabled", 1);
2132 Add_All_MTP_Storage();
2133 return true;
Ethan Yonker726a0202014-12-16 20:01:38 -06002134 } else {
2135 close(mtppipe[0]);
2136 close(mtppipe[1]);
Ethan Yonker74db1572015-10-28 12:44:49 -05002137 gui_err("mtp_fail=Failed to enable MTP");
Ethan Yonker1b039202015-01-30 10:08:48 -06002138 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002139 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002140#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002141 gui_err("no_mtp=MTP support not included");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002142#endif
2143 DataManager::SetValue("tw_mtp_enabled", 0);
2144 return false;
2145}
2146
Ethan Yonker1b039202015-01-30 10:08:48 -06002147void TWPartitionManager::Add_All_MTP_Storage(void) {
2148#ifdef TW_HAS_MTP
2149 std::vector<TWPartition*>::iterator iter;
2150
2151 if (!mtppid)
2152 return; // MTP is not enabled
2153
2154 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2155 if ((*iter)->Is_Storage && (*iter)->Is_Present && (*iter)->Mount(false))
2156 Add_Remove_MTP_Storage((*iter), MTP_MESSAGE_ADD_STORAGE);
2157 }
2158#else
2159 return;
2160#endif
2161}
2162
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002163bool TWPartitionManager::Disable_MTP(void) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002164 char old_value[PROPERTY_VALUE_MAX];
Matt Mowere07f0102017-02-23 17:40:00 -06002165 property_get("sys.usb.config", old_value, "");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002166 if (strcmp(old_value, "adb") != 0) {
2167 char vendor[PROPERTY_VALUE_MAX];
2168 char product[PROPERTY_VALUE_MAX];
2169 property_set("sys.usb.config", "none");
2170 property_get("usb.vendor", vendor, "18D1");
Matt Mowere07f0102017-02-23 17:40:00 -06002171 property_get("usb.product.adb", product, "D001");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002172 string vendorstr = vendor;
2173 string productstr = product;
2174 TWFunc::write_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2175 TWFunc::write_file("/sys/class/android_usb/android0/idProduct", productstr);
2176 usleep(2000);
2177 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002178#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002179 if (mtppid) {
Ethan Yonker8613dc02014-09-11 09:28:20 -05002180 LOGINFO("Disabling MTP\n");
2181 int status;
2182 kill(mtppid, SIGKILL);
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002183 mtppid = 0;
Ethan Yonker8613dc02014-09-11 09:28:20 -05002184 // We don't care about the exit value, but this prevents a zombie process
2185 waitpid(mtppid, &status, 0);
Ethan Yonker726a0202014-12-16 20:01:38 -06002186 close(mtp_write_fd);
2187 mtp_write_fd = -1;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002188 }
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002189#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002190 property_set("sys.usb.config", "adb");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002191#ifdef TW_HAS_MTP
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002192 DataManager::SetValue("tw_mtp_enabled", 0);
2193 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002194#endif
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002195 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002196}
Ethan Yonker726a0202014-12-16 20:01:38 -06002197
2198TWPartition* TWPartitionManager::Find_Partition_By_MTP_Storage_ID(unsigned int Storage_ID) {
2199 std::vector<TWPartition*>::iterator iter;
2200
2201 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2202 if ((*iter)->MTP_Storage_ID == Storage_ID)
2203 return (*iter);
2204 }
2205 return NULL;
2206}
2207
2208bool TWPartitionManager::Add_Remove_MTP_Storage(TWPartition* Part, int message_type) {
2209#ifdef TW_HAS_MTP
2210 struct mtpmsg mtp_message;
2211
2212 if (!mtppid)
2213 return false; // MTP is disabled
2214
2215 if (mtp_write_fd < 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002216 LOGINFO("MTP: mtp_write_fd is not set\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002217 return false;
2218 }
2219
2220 if (Part) {
Ethan Yonker4bfabab2014-12-29 09:15:37 -06002221 if (Part->MTP_Storage_ID == 0)
2222 return false;
Ethan Yonker726a0202014-12-16 20:01:38 -06002223 if (message_type == MTP_MESSAGE_REMOVE_STORAGE) {
2224 mtp_message.message_type = MTP_MESSAGE_REMOVE_STORAGE; // Remove
2225 LOGINFO("sending message to remove %i\n", Part->MTP_Storage_ID);
2226 mtp_message.storage_id = Part->MTP_Storage_ID;
2227 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002228 LOGINFO("error sending message to remove storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002229 return false;
2230 } else {
2231 LOGINFO("Message sent, remove storage ID: %i\n", Part->MTP_Storage_ID);
2232 return true;
2233 }
2234 } else if (message_type == MTP_MESSAGE_ADD_STORAGE && Part->Is_Mounted()) {
2235 mtp_message.message_type = MTP_MESSAGE_ADD_STORAGE; // Add
2236 mtp_message.storage_id = Part->MTP_Storage_ID;
2237 mtp_message.path = Part->Storage_Path.c_str();
2238 mtp_message.display = Part->Storage_Name.c_str();
2239 mtp_message.maxFileSize = Part->Get_Max_FileSize();
Ethan Yonker1b039202015-01-30 10:08:48 -06002240 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 -06002241 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002242 LOGINFO("error sending message to add storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002243 return false;
2244 } else {
2245 LOGINFO("Message sent, add storage ID: %i\n", Part->MTP_Storage_ID);
2246 return true;
2247 }
2248 } else {
2249 LOGERR("Unknown MTP message type: %i\n", message_type);
2250 }
2251 } else {
2252 // This hopefully never happens as the error handling should
2253 // occur in the calling function.
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002254 LOGINFO("TWPartitionManager::Add_Remove_MTP_Storage NULL partition given\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002255 }
2256 return true;
2257#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002258 gui_err("no_mtp=MTP support not included");
Ethan Yonker726a0202014-12-16 20:01:38 -06002259 DataManager::SetValue("tw_mtp_enabled", 0);
2260 return false;
2261#endif
2262}
2263
2264bool TWPartitionManager::Add_MTP_Storage(string Mount_Point) {
2265#ifdef TW_HAS_MTP
2266 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2267 if (Part) {
2268 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2269 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002270 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002271 }
2272#endif
2273 return false;
2274}
2275
2276bool TWPartitionManager::Add_MTP_Storage(unsigned int Storage_ID) {
2277#ifdef TW_HAS_MTP
2278 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2279 if (Part) {
2280 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2281 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002282 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002283 }
2284#endif
2285 return false;
2286}
2287
2288bool TWPartitionManager::Remove_MTP_Storage(string Mount_Point) {
2289#ifdef TW_HAS_MTP
2290 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2291 if (Part) {
2292 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2293 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002294 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002295 }
2296#endif
2297 return false;
2298}
2299
2300bool TWPartitionManager::Remove_MTP_Storage(unsigned int Storage_ID) {
2301#ifdef TW_HAS_MTP
2302 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2303 if (Part) {
2304 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2305 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002306 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002307 }
2308#endif
2309 return false;
2310}
Ethan Yonker96af84a2015-01-05 14:58:36 -06002311
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002312bool TWPartitionManager::Flash_Image(string& path, string& filename) {
Matt Mower23d8aae2017-01-06 14:30:33 -06002313 int partition_count = 0;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002314 TWPartition* flash_part = NULL;
bigbiffce8f83c2015-12-12 18:30:21 -05002315 string Flash_List, flash_path, full_filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002316 size_t start_pos = 0, end_pos = 0;
2317
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002318 full_filename = path + "/" + filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002319
bigbiffce8f83c2015-12-12 18:30:21 -05002320 gui_msg("image_flash_start=[IMAGE FLASH STARTED]");
2321 gui_msg(Msg("img_to_flash=Image to flash: '{1}'")(full_filename));
2322
2323 if (!TWFunc::Path_Exists(full_filename)) {
2324 if (!Mount_By_Path(full_filename, true)) {
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002325 return false;
2326 }
bigbiffce8f83c2015-12-12 18:30:21 -05002327 if (!TWFunc::Path_Exists(full_filename)) {
2328 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")(full_filename));
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002329 return false;
2330 }
2331 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002332
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002333 PartitionSettings part_settings;
2334 part_settings.Backup_Folder = path;
2335 unsigned long long total_bytes = TWFunc::Get_File_Size(full_filename);
2336 ProgressTracking progress(total_bytes);
2337 part_settings.progress = &progress;
2338 part_settings.adbbackup = false;
2339 part_settings.PM_Method = PM_RESTORE;
2340
Ethan Yonker74db1572015-10-28 12:44:49 -05002341 gui_msg("calc_restore=Calculating restore details...");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002342 DataManager::GetValue("tw_flash_partition", Flash_List);
2343 if (!Flash_List.empty()) {
2344 end_pos = Flash_List.find(";", start_pos);
2345 while (end_pos != string::npos && start_pos < Flash_List.size()) {
2346 flash_path = Flash_List.substr(start_pos, end_pos - start_pos);
2347 flash_part = Find_Partition_By_Path(flash_path);
2348 if (flash_part != NULL) {
2349 partition_count++;
2350 if (partition_count > 1) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002351 gui_err("too_many_flash=Too many partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002352 return false;
2353 }
2354 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002355 gui_msg(Msg(msg::kError, "flash_unable_locate=Unable to locate '{1}' partition for flashing.")(flash_path));
Ethan Yonker96af84a2015-01-05 14:58:36 -06002356 return false;
2357 }
2358 start_pos = end_pos + 1;
2359 end_pos = Flash_List.find(";", start_pos);
2360 }
2361 }
2362
2363 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002364 gui_err("no_part_flash=No partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002365 return false;
2366 }
2367
2368 DataManager::SetProgress(0.0);
2369 if (flash_part) {
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002370 flash_part->Backup_FileName = filename;
2371 if (!flash_part->Flash_Image(&part_settings))
Ethan Yonker96af84a2015-01-05 14:58:36 -06002372 return false;
2373 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002374 gui_err("invalid_flash=Invalid flash partition specified.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002375 return false;
2376 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002377 gui_highlight("flash_done=IMAGE FLASH COMPLETED]");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002378 return true;
2379}
Ethan Yonker74db1572015-10-28 12:44:49 -05002380
2381void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value) {
2382 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2383 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002384 if (part->Is_Adopted_Storage) {
2385 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2386 part->Backup_Display_Name = part->Display_Name;
2387 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2388 } else {
2389 part->Display_Name = gui_lookup(resource_name, default_value);
2390 part->Backup_Display_Name = part->Display_Name;
2391 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002392 }
2393}
2394
2395void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value, const char* storage_resource_name, const char* storage_default_value) {
2396 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2397 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002398 if (part->Is_Adopted_Storage) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06002399 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup("data_backup", "Data (excl. storage)");
Ethan Yonker66a19492015-12-10 10:19:45 -06002400 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
Ethan Yonker66a19492015-12-10 10:19:45 -06002401 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2402 } else {
2403 part->Display_Name = gui_lookup(resource_name, default_value);
2404 part->Backup_Display_Name = part->Display_Name;
2405 if (part->Is_Storage)
2406 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2407 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002408 }
2409}
2410
Ethan Yonker01f4e032017-02-03 15:30:52 -06002411void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value, const char* storage_resource_name, const char* storage_default_value, const char* backup_name, const char* backup_default) {
2412 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2413 if (part) {
2414 if (part->Is_Adopted_Storage) {
2415 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup(backup_name, backup_default);
2416 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2417 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2418 } else {
2419 part->Display_Name = gui_lookup(resource_name, default_value);
2420 part->Backup_Display_Name = gui_lookup(backup_name, backup_default);
2421 if (part->Is_Storage)
2422 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2423 }
2424 }
2425}
2426
Ethan Yonker74db1572015-10-28 12:44:49 -05002427void TWPartitionManager::Translate_Partition_Display_Names() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002428 LOGINFO("Translating partition display names\n");
Ethan Yonker74db1572015-10-28 12:44:49 -05002429 Translate_Partition("/system", "system", "System");
2430 Translate_Partition("/system_image", "system_image", "System Image");
2431 Translate_Partition("/vendor", "vendor", "Vendor");
2432 Translate_Partition("/vendor_image", "vendor_image", "Vendor Image");
2433 Translate_Partition("/cache", "cache", "Cache");
Ethan Yonker74db1572015-10-28 12:44:49 -05002434 Translate_Partition("/boot", "boot", "Boot");
2435 Translate_Partition("/recovery", "recovery", "Recovery");
2436 if (!datamedia) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06002437 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage");
Ethan Yonker74db1572015-10-28 12:44:49 -05002438 Translate_Partition("/sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2439 Translate_Partition("/internal_sd", "sdcard", "SDCard", "sdcard", "SDCard");
2440 Translate_Partition("/internal_sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2441 Translate_Partition("/emmc", "sdcard", "SDCard", "sdcard", "SDCard");
Ethan Yonker01f4e032017-02-03 15:30:52 -06002442 } else {
2443 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05002444 }
Ethan Yonker01f4e032017-02-03 15:30:52 -06002445 Translate_Partition("/external_sd", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
2446 Translate_Partition("/external_sdcard", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05002447 Translate_Partition("/usb-otg", "usbotg", "USB OTG", "usbotg", "USB OTG");
2448 Translate_Partition("/sd-ext", "sdext", "SD-EXT");
2449
2450 // Android secure is a special case
2451 TWPartition* part = PartitionManager.Find_Partition_By_Path("/and-sec");
2452 if (part)
2453 part->Backup_Display_Name = gui_lookup("android_secure", "Android Secure");
2454
2455 // This updates the text on all of the storage selection buttons in the GUI
2456 DataManager::SetBackupFolder();
2457}
Ethan Yonker66a19492015-12-10 10:19:45 -06002458
2459void TWPartitionManager::Decrypt_Adopted() {
2460#ifdef TW_INCLUDE_CRYPTO
2461 if (!Mount_By_Path("/data", false)) {
2462 LOGERR("Cannot decrypt adopted storage because /data will not mount\n");
2463 return;
2464 }
2465 LOGINFO("Decrypt adopted storage starting\n");
2466 char* xmlFile = PageManager::LoadFileToBuffer("/data/system/storage.xml", NULL);
2467 xml_document<> *doc = NULL;
2468 xml_node<>* volumes = NULL;
Ethan Yonker66a19492015-12-10 10:19:45 -06002469 string Primary_Storage_UUID = "";
2470 if (xmlFile != NULL) {
2471 LOGINFO("successfully loaded storage.xml\n");
2472 doc = new xml_document<>();
2473 doc->parse<0>(xmlFile);
2474 volumes = doc->first_node("volumes");
2475 if (volumes) {
2476 xml_attribute<>* psuuid = volumes->first_attribute("primaryStorageUuid");
2477 if (psuuid) {
2478 Primary_Storage_UUID = psuuid->value();
2479 }
2480 }
2481 }
2482 std::vector<TWPartition*>::iterator adopt;
2483 for (adopt = Partitions.begin(); adopt != Partitions.end(); adopt++) {
2484 if ((*adopt)->Removable && (*adopt)->Is_Present) {
2485 if ((*adopt)->Decrypt_Adopted() == 0) {
2486 if (volumes) {
2487 xml_node<>* volume = volumes->first_node("volume");
2488 while (volume) {
2489 xml_attribute<>* guid = volume->first_attribute("partGuid");
2490 if (guid) {
2491 string GUID = (*adopt)->Adopted_GUID.c_str();
2492 GUID.insert(8, "-");
2493 GUID.insert(13, "-");
2494 GUID.insert(18, "-");
2495 GUID.insert(23, "-");
2496
2497 if (strcasecmp(GUID.c_str(), guid->value()) == 0) {
2498 xml_attribute<>* attr = volume->first_attribute("nickname");
2499 if (attr) {
2500 (*adopt)->Storage_Name = attr->value();
2501 (*adopt)->Display_Name = (*adopt)->Storage_Name;
2502 (*adopt)->Backup_Display_Name = (*adopt)->Storage_Name;
2503 LOGINFO("storage name from storage.xml is '%s'\n", attr->value());
2504 }
2505 attr = volume->first_attribute("fsUuid");
2506 if (attr && !Primary_Storage_UUID.empty() && strcmp(Primary_Storage_UUID.c_str(), attr->value()) == 0) {
2507 TWPartition* Dat = Find_Partition_By_Path("/data");
2508 if (Dat) {
2509 LOGINFO("Internal storage is found on adopted storage '%s'\n", (*adopt)->Display_Name.c_str());
2510 LOGINFO("Changing '%s' to point to '%s'\n", Dat->Symlink_Mount_Point.c_str(), (*adopt)->Storage_Path.c_str());
2511 (*adopt)->Symlink_Mount_Point = Dat->Symlink_Mount_Point;
2512 Dat->Symlink_Mount_Point = "";
2513 // Toggle mounts to ensure that the symlink mount point (probably /sdcard) is mounted to the right location
2514 Dat->UnMount(false);
2515 Dat->Mount(false);
2516 (*adopt)->UnMount(false);
2517 (*adopt)->Mount(false);
2518 Output_Partition((*adopt));
2519 }
2520 }
2521 break;
2522 }
2523 }
2524 volume = volume->next_sibling("volume");
2525 }
2526 }
2527 }
2528 }
2529 }
2530 if (xmlFile) {
2531 doc->clear();
2532 delete doc;
2533 free(xmlFile);
2534 }
2535#else
2536 LOGINFO("Decrypt_Adopted: no crypto support\n");
2537 return;
2538#endif
2539}
Ethan Yonkerfcf3f242016-02-16 12:30:26 -06002540
2541void TWPartitionManager::Remove_Partition_By_Path(string Path) {
2542 std::vector<TWPartition*>::iterator iter;
2543 string Local_Path = TWFunc::Get_Root_Path(Path);
2544
2545 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2546 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
2547 LOGINFO("Found and erasing '%s' from partition list\n", Local_Path.c_str());
2548 Partitions.erase(iter);
2549 return;
2550 }
2551 }
2552}
Ethan Yonker1b190162016-12-05 15:25:19 -06002553
2554void TWPartitionManager::Set_Active_Slot(const string& Slot) {
2555 if (Slot != "A" && Slot != "B") {
2556 LOGERR("Set_Active_Slot invalid slot '%s'\n", Slot.c_str());
2557 return;
2558 }
2559 if (Active_Slot_Display == Slot)
2560 return;
2561 LOGINFO("Setting active slot %s\n", Slot.c_str());
2562#ifdef AB_OTA_UPDATER
2563 if (!Active_Slot_Display.empty()) {
2564 const hw_module_t *hw_module;
2565 boot_control_module_t *module;
2566 int ret;
2567 ret = hw_get_module("bootctrl", &hw_module);
2568 if (ret != 0) {
2569 LOGERR("Error getting bootctrl module.\n");
2570 } else {
2571 module = (boot_control_module_t*) hw_module;
2572 module->init(module);
2573 int slot_number = 0;
2574 if (Slot == "B")
2575 slot_number = 1;
2576 if (module->setActiveBootSlot(module, slot_number))
2577 gui_msg(Msg(msg::kError, "unable_set_boot_slot=Error changing bootloader boot slot to {1}")(Slot));
2578 }
2579 DataManager::SetValue("tw_active_slot", Slot); // Doing this outside of this if block may result in a seg fault because the DataManager may not be ready yet
2580 }
2581#else
2582 LOGERR("Boot slot feature not present\n");
2583#endif
2584 Active_Slot_Display = Slot;
2585 if (Fstab_Processed())
2586 Update_System_Details();
2587}
2588string TWPartitionManager::Get_Active_Slot_Suffix() {
2589 if (Active_Slot_Display == "A")
2590 return "_a";
2591 return "_b";
2592}
2593string TWPartitionManager::Get_Active_Slot_Display() {
2594 return Active_Slot_Display;
2595}