blob: c2ef5f5631de3d19279c75e9e72022fc42a1c385 [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>
bigbiff bigbiff9c754052013-01-09 09:09:08 -050030#include <iostream>
31#include <iomanip>
Ethan Yonker8613dc02014-09-11 09:28:20 -050032#include <sys/wait.h>
Ethan Yonker483e9f42016-01-11 22:21:18 -060033#include <linux/fs.h>
34#include <sys/mount.h>
Dees_Troy51a0e822012-09-05 15:24:24 -040035#include "variables.h"
Dees_Troy2673cec2013-04-02 20:22:16 +000036#include "twcommon.h"
Dees_Troy51a0e822012-09-05 15:24:24 -040037#include "partitions.hpp"
Dees_Troy5bf43922012-09-07 16:07:55 -040038#include "data.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040039#include "twrp-functions.hpp"
Ethan Yonkerb5fab762016-01-28 14:03:33 -060040#include "fixContexts.hpp"
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -050041#include "twrpDigest.hpp"
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050042#include "twrpDU.hpp"
Ethan Yonker4b94cfd2014-12-11 10:00:45 -060043#include "set_metadata.h"
bigbiff7abc5fe2015-01-17 16:53:12 -050044#include "tw_atomic.hpp"
Ethan Yonker74db1572015-10-28 12:44:49 -050045#include "gui/gui.hpp"
Ethan Yonker472f5062016-02-25 13:47:30 -060046#include "progresstracking.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040047
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040048#ifdef TW_HAS_MTP
49#include "mtp/mtp_MtpServer.hpp"
50#include "mtp/twrpMtp.hpp"
Ethan Yonker726a0202014-12-16 20:01:38 -060051#include "mtp/MtpMessage.hpp"
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040052#endif
53
Dees Troy6f6441d2014-01-23 02:07:03 +000054extern "C" {
55 #include "cutils/properties.h"
Ethan Yonker74db1572015-10-28 12:44:49 -050056 #include "gui/gui.h"
Dees Troy6f6441d2014-01-23 02:07:03 +000057}
58
Dees_Troy5bf43922012-09-07 16:07:55 -040059#ifdef TW_INCLUDE_CRYPTO
Ethan Yonker253368a2014-11-25 15:00:52 -060060 #include "crypto/lollipop/cryptfs.h"
Ethan Yonker66a19492015-12-10 10:19:45 -060061 #include "gui/rapidxml.hpp"
62 #include "gui/pages.hpp"
Dees_Troy5bf43922012-09-07 16:07:55 -040063#endif
Dees_Troy51a0e822012-09-05 15:24:24 -040064
Ethan Yonker6277c792014-09-15 14:54:30 -050065extern bool datamedia;
66
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050067TWPartitionManager::TWPartitionManager(void) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040068 mtp_was_enabled = false;
Ethan Yonker726a0202014-12-16 20:01:38 -060069 mtp_write_fd = -1;
bigbiff7abc5fe2015-01-17 16:53:12 -050070 stop_backup.set_value(0);
71 tar_fork_pid = 0;
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050072}
73
Dees_Troy51a0e822012-09-05 15:24:24 -040074int TWPartitionManager::Process_Fstab(string Fstab_Filename, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -040075 FILE *fstabFile;
76 char fstab_line[MAX_FSTAB_LINE_LENGTH];
Dees Troy02a64532014-03-19 15:23:32 +000077 TWPartition* settings_partition = NULL;
Matt Mowerbf4efa32014-04-14 23:25:26 -050078 TWPartition* andsec_partition = NULL;
Ethan Yonker726a0202014-12-16 20:01:38 -060079 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 -040080
81 fstabFile = fopen(Fstab_Filename.c_str(), "rt");
82 if (fstabFile == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +000083 LOGERR("Critical Error: Unable to open fstab at '%s'.\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -040084 return false;
85 }
86
87 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
88 if (fstab_line[0] != '/')
89 continue;
90
Dees_Troy2a923582012-09-20 12:13:34 -040091 if (fstab_line[strlen(fstab_line) - 1] != '\n')
92 fstab_line[strlen(fstab_line)] = '\n';
that9e0593e2014-10-08 00:01:24 +020093 TWPartition* partition = new TWPartition();
Dees_Troy2a923582012-09-20 12:13:34 -040094 string line = fstab_line;
Dees_Troyab10ee22012-09-21 14:27:30 -040095 memset(fstab_line, 0, sizeof(fstab_line));
Dees_Troy2a923582012-09-20 12:13:34 -040096
Dees_Troy5bf43922012-09-07 16:07:55 -040097 if (partition->Process_Fstab_Line(line, Display_Error)) {
Ethan Yonker726a0202014-12-16 20:01:38 -060098 if (partition->Is_Storage) {
99 ++storageid;
100 partition->MTP_Storage_ID = storageid;
101 }
Matt Mowered71fa32014-04-16 13:21:47 -0500102 if (!settings_partition && partition->Is_Settings_Storage && partition->Is_Present) {
Dees Troy02a64532014-03-19 15:23:32 +0000103 settings_partition = partition;
Dees_Troya13d74f2013-03-24 08:54:55 -0500104 } else {
105 partition->Is_Settings_Storage = false;
Dees_Troya13d74f2013-03-24 08:54:55 -0500106 }
Matt Mowered71fa32014-04-16 13:21:47 -0500107 if (!andsec_partition && partition->Has_Android_Secure && partition->Is_Present) {
Matt Mowerbf4efa32014-04-14 23:25:26 -0500108 andsec_partition = partition;
109 } else {
110 partition->Has_Android_Secure = false;
111 }
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500112 Partitions.push_back(partition);
Dees_Troy5bf43922012-09-07 16:07:55 -0400113 } else {
114 delete partition;
115 }
116 }
117 fclose(fstabFile);
Ethan Yonker6277c792014-09-15 14:54:30 -0500118 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) {
119 // Attempt to automatically identify /data/media emulated storage devices
120 TWPartition* Dat = Find_Partition_By_Path("/data");
121 if (Dat) {
122 LOGINFO("Using automatic handling for /data/media emulated storage device.\n");
123 datamedia = true;
that9e0593e2014-10-08 00:01:24 +0200124 Dat->Setup_Data_Media();
Ethan Yonker6277c792014-09-15 14:54:30 -0500125 settings_partition = Dat;
Ethan Yonker726a0202014-12-16 20:01:38 -0600126 // Since /data was not considered a storage partition earlier, we still need to assign an MTP ID
127 ++storageid;
128 Dat->MTP_Storage_ID = storageid;
Ethan Yonker6277c792014-09-15 14:54:30 -0500129 }
130 }
Dees Troy02a64532014-03-19 15:23:32 +0000131 if (!settings_partition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500132 std::vector<TWPartition*>::iterator iter;
133 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
134 if ((*iter)->Is_Storage) {
Dees Troy02a64532014-03-19 15:23:32 +0000135 settings_partition = (*iter);
Dees_Troya13d74f2013-03-24 08:54:55 -0500136 break;
137 }
138 }
Dees Troy02a64532014-03-19 15:23:32 +0000139 if (!settings_partition)
Dees_Troy2673cec2013-04-02 20:22:16 +0000140 LOGERR("Unable to locate storage partition for storing settings file.\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500141 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400142 if (!Write_Fstab()) {
143 if (Display_Error)
Dees_Troy2673cec2013-04-02 20:22:16 +0000144 LOGERR("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400145 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000146 LOGINFO("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400147 }
Matt Mowered71fa32014-04-16 13:21:47 -0500148
Matt Mowerbf4efa32014-04-14 23:25:26 -0500149 if (andsec_partition) {
150 Setup_Android_Secure_Location(andsec_partition);
Matt Mowered71fa32014-04-16 13:21:47 -0500151 } else if (settings_partition) {
Matt Mowerbf4efa32014-04-14 23:25:26 -0500152 Setup_Android_Secure_Location(settings_partition);
153 }
Matt Mowered71fa32014-04-16 13:21:47 -0500154 if (settings_partition) {
155 Setup_Settings_Storage_Partition(settings_partition);
156 }
Ethan Yonker253368a2014-11-25 15:00:52 -0600157#ifdef TW_INCLUDE_CRYPTO
Ethan Yonkercceebb82014-11-18 10:17:59 -0600158 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
159 if (Decrypt_Data && Decrypt_Data->Is_Encrypted && !Decrypt_Data->Is_Decrypted) {
160 int password_type = cryptfs_get_password_type();
161 if (password_type == CRYPT_TYPE_DEFAULT) {
162 LOGINFO("Device is encrypted with the default password, attempting to decrypt.\n");
163 if (Decrypt_Device("default_password") == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500164 gui_msg("decrypt_success=Successfully decrypted with default password.");
Ethan Yonkercceebb82014-11-18 10:17:59 -0600165 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
166 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500167 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
Ethan Yonkercceebb82014-11-18 10:17:59 -0600168 }
169 } else {
170 DataManager::SetValue("TW_CRYPTO_TYPE", password_type);
171 }
172 }
Ethan Yonker66a19492015-12-10 10:19:45 -0600173 if (Decrypt_Data && (!Decrypt_Data->Is_Encrypted || Decrypt_Data->Is_Decrypted) && Decrypt_Data->Mount(false)) {
174 Decrypt_Adopted();
175 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600176#endif
Dees_Troy51127312012-09-08 13:08:49 -0400177 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -0400178 UnMount_Main_Partitions();
Dees_Troy5bf43922012-09-07 16:07:55 -0400179 return true;
180}
181
182int TWPartitionManager::Write_Fstab(void) {
183 FILE *fp;
184 std::vector<TWPartition*>::iterator iter;
185 string Line;
186
187 fp = fopen("/etc/fstab", "w");
188 if (fp == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000189 LOGINFO("Can not open /etc/fstab.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400190 return false;
191 }
Dees_Troy63c8df72012-09-10 14:02:05 -0400192 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy51127312012-09-08 13:08:49 -0400193 if ((*iter)->Can_Be_Mounted) {
dianlujitao7d304c72016-01-02 12:15:50 +0800194 Line = (*iter)->Actual_Block_Device + " " + (*iter)->Mount_Point + " " + (*iter)->Current_File_System + " rw 0 0\n";
Dees_Troy5bf43922012-09-07 16:07:55 -0400195 fputs(Line.c_str(), fp);
Dees_Troy91862e62013-04-04 23:48:21 +0000196 }
197 // Handle subpartition tracking
198 if ((*iter)->Is_SubPartition) {
199 TWPartition* ParentPartition = Find_Partition_By_Path((*iter)->SubPartition_Of);
200 if (ParentPartition)
201 ParentPartition->Has_SubPartition = true;
202 else
203 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 -0400204 }
205 }
206 fclose(fp);
207 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -0400208}
209
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500210void TWPartitionManager::Setup_Settings_Storage_Partition(TWPartition* Part) {
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500211 DataManager::SetValue("tw_settings_path", Part->Storage_Path);
212 DataManager::SetValue("tw_storage_path", Part->Storage_Path);
213 LOGINFO("Settings storage is '%s'\n", Part->Storage_Path.c_str());
214}
215
Matt Mowerbf4efa32014-04-14 23:25:26 -0500216void TWPartitionManager::Setup_Android_Secure_Location(TWPartition* Part) {
217 if (Part->Has_Android_Secure)
218 Part->Setup_AndSec();
Ethan Yonker6277c792014-09-15 14:54:30 -0500219 else if (!datamedia)
Matt Mowerbf4efa32014-04-14 23:25:26 -0500220 Part->Setup_AndSec();
Matt Mowerbf4efa32014-04-14 23:25:26 -0500221}
222
Dees_Troy8170a922012-09-18 15:40:25 -0400223void TWPartitionManager::Output_Partition_Logging(void) {
224 std::vector<TWPartition*>::iterator iter;
225
226 printf("\n\nPartition Logs:\n");
227 for (iter = Partitions.begin(); iter != Partitions.end(); iter++)
228 Output_Partition((*iter));
229}
230
231void TWPartitionManager::Output_Partition(TWPartition* Part) {
232 unsigned long long mb = 1048576;
233
Gary Peck004d48b2012-11-21 16:28:18 -0800234 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 -0400235 if (Part->Can_Be_Mounted) {
Gary Peck004d48b2012-11-21 16:28:18 -0800236 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 -0400237 }
Gary Peck004d48b2012-11-21 16:28:18 -0800238 printf("\n Flags: ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500239 if (Part->Can_Be_Mounted)
240 printf("Can_Be_Mounted ");
Gary Peck004d48b2012-11-21 16:28:18 -0800241 if (Part->Can_Be_Wiped)
242 printf("Can_Be_Wiped ");
Hashcodedabfd492013-08-29 22:45:30 -0700243 if (Part->Use_Rm_Rf)
244 printf("Use_Rm_Rf ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500245 if (Part->Can_Be_Backed_Up)
246 printf("Can_Be_Backed_Up ");
Gary Peck004d48b2012-11-21 16:28:18 -0800247 if (Part->Wipe_During_Factory_Reset)
248 printf("Wipe_During_Factory_Reset ");
249 if (Part->Wipe_Available_in_GUI)
250 printf("Wipe_Available_in_GUI ");
251 if (Part->Is_SubPartition)
252 printf("Is_SubPartition ");
253 if (Part->Has_SubPartition)
254 printf("Has_SubPartition ");
255 if (Part->Removable)
256 printf("Removable ");
257 if (Part->Is_Present)
258 printf("IsPresent ");
259 if (Part->Can_Be_Encrypted)
260 printf("Can_Be_Encrypted ");
261 if (Part->Is_Encrypted)
262 printf("Is_Encrypted ");
263 if (Part->Is_Decrypted)
264 printf("Is_Decrypted ");
265 if (Part->Has_Data_Media)
266 printf("Has_Data_Media ");
Dees_Troy83bd4832013-05-04 12:39:56 +0000267 if (Part->Can_Encrypt_Backup)
268 printf("Can_Encrypt_Backup ");
269 if (Part->Use_Userdata_Encryption)
270 printf("Use_Userdata_Encryption ");
Gary Peck004d48b2012-11-21 16:28:18 -0800271 if (Part->Has_Android_Secure)
272 printf("Has_Android_Secure ");
273 if (Part->Is_Storage)
274 printf("Is_Storage ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500275 if (Part->Is_Settings_Storage)
276 printf("Is_Settings_Storage ");
Dees_Troy68cab492012-12-12 19:29:35 +0000277 if (Part->Ignore_Blkid)
278 printf("Ignore_Blkid ");
Dees_Troy16c2b312013-01-15 16:51:18 +0000279 if (Part->Retain_Layout_Version)
280 printf("Retain_Layout_Version ");
Ethan Yonker253368a2014-11-25 15:00:52 -0600281 if (Part->Mount_To_Decrypt)
282 printf("Mount_To_Decrypt ");
Ethan Yonker96af84a2015-01-05 14:58:36 -0600283 if (Part->Can_Flash_Img)
284 printf("Can_Flash_Img ");
Ethan Yonker66a19492015-12-10 10:19:45 -0600285 if (Part->Is_Adopted_Storage)
286 printf("Is_Adopted_Storage ");
Gary Peck004d48b2012-11-21 16:28:18 -0800287 printf("\n");
288 if (!Part->SubPartition_Of.empty())
289 printf(" SubPartition_Of: %s\n", Part->SubPartition_Of.c_str());
290 if (!Part->Symlink_Path.empty())
291 printf(" Symlink_Path: %s\n", Part->Symlink_Path.c_str());
292 if (!Part->Symlink_Mount_Point.empty())
293 printf(" Symlink_Mount_Point: %s\n", Part->Symlink_Mount_Point.c_str());
294 if (!Part->Primary_Block_Device.empty())
295 printf(" Primary_Block_Device: %s\n", Part->Primary_Block_Device.c_str());
296 if (!Part->Alternate_Block_Device.empty())
297 printf(" Alternate_Block_Device: %s\n", Part->Alternate_Block_Device.c_str());
298 if (!Part->Decrypted_Block_Device.empty())
299 printf(" Decrypted_Block_Device: %s\n", Part->Decrypted_Block_Device.c_str());
Ethan Yonker253368a2014-11-25 15:00:52 -0600300 if (!Part->Crypto_Key_Location.empty() && Part->Crypto_Key_Location != "footer")
301 printf(" Crypto_Key_Location: %s\n", Part->Crypto_Key_Location.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800302 if (Part->Length != 0)
303 printf(" Length: %i\n", Part->Length);
304 if (!Part->Display_Name.empty())
305 printf(" Display_Name: %s\n", Part->Display_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500306 if (!Part->Storage_Name.empty())
307 printf(" Storage_Name: %s\n", Part->Storage_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800308 if (!Part->Backup_Path.empty())
309 printf(" Backup_Path: %s\n", Part->Backup_Path.c_str());
310 if (!Part->Backup_Name.empty())
311 printf(" Backup_Name: %s\n", Part->Backup_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500312 if (!Part->Backup_Display_Name.empty())
313 printf(" Backup_Display_Name: %s\n", Part->Backup_Display_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800314 if (!Part->Backup_FileName.empty())
315 printf(" Backup_FileName: %s\n", Part->Backup_FileName.c_str());
316 if (!Part->Storage_Path.empty())
317 printf(" Storage_Path: %s\n", Part->Storage_Path.c_str());
318 if (!Part->Current_File_System.empty())
319 printf(" Current_File_System: %s\n", Part->Current_File_System.c_str());
320 if (!Part->Fstab_File_System.empty())
321 printf(" Fstab_File_System: %s\n", Part->Fstab_File_System.c_str());
322 if (Part->Format_Block_Size != 0)
Ethan Yonkerc798c9c2015-10-09 11:15:26 -0500323 printf(" Format_Block_Size: %lu\n", Part->Format_Block_Size);
Dees_Troy094207a2012-09-26 12:00:39 -0400324 if (!Part->MTD_Name.empty())
325 printf(" MTD_Name: %s\n", Part->MTD_Name.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -0400326 string back_meth = Part->Backup_Method_By_Name();
Ethan Yonker6277c792014-09-15 14:54:30 -0500327 printf(" Backup_Method: %s\n", back_meth.c_str());
Hashcode62bd9e02013-11-19 21:59:42 -0800328 if (Part->Mount_Flags || !Part->Mount_Options.empty())
329 printf(" Mount_Flags=0x%8x, Mount_Options=%s\n", Part->Mount_Flags, Part->Mount_Options.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -0600330 if (Part->MTP_Storage_ID)
331 printf(" MTP_Storage_ID: %i\n", Part->MTP_Storage_ID);
Ethan Yonker6277c792014-09-15 14:54:30 -0500332 printf("\n");
Dees_Troy8170a922012-09-18 15:40:25 -0400333}
334
Dees_Troy51a0e822012-09-05 15:24:24 -0400335int TWPartitionManager::Mount_By_Path(string Path, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400336 std::vector<TWPartition*>::iterator iter;
337 int ret = false;
338 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400339 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400340
Dees_Troyd93bda52013-07-03 19:55:19 +0000341 if (Local_Path == "/tmp" || Local_Path == "/")
Dees_Troy43d8b002012-09-17 16:00:01 -0400342 return true;
343
Dees_Troy5bf43922012-09-07 16:07:55 -0400344 // Iterate through all partitions
345 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400346 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400347 ret = (*iter)->Mount(Display_Error);
348 found = true;
Dees_Troy51127312012-09-08 13:08:49 -0400349 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400350 (*iter)->Mount(Display_Error);
Dees_Troy51127312012-09-08 13:08:49 -0400351 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400352 }
353 if (found) {
354 return ret;
355 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500356 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 -0400357 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000358 LOGINFO("Mount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400359 }
360 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400361}
362
Dees_Troy51a0e822012-09-05 15:24:24 -0400363int TWPartitionManager::UnMount_By_Path(string Path, bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400364 std::vector<TWPartition*>::iterator iter;
365 int ret = false;
366 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400367 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy51127312012-09-08 13:08:49 -0400368
369 // Iterate through all partitions
370 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400371 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy51127312012-09-08 13:08:49 -0400372 ret = (*iter)->UnMount(Display_Error);
373 found = true;
374 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
375 (*iter)->UnMount(Display_Error);
376 }
377 }
378 if (found) {
379 return ret;
380 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500381 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 -0400382 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000383 LOGINFO("UnMount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400384 }
385 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400386}
387
Dees_Troy51a0e822012-09-05 15:24:24 -0400388int TWPartitionManager::Is_Mounted_By_Path(string Path) {
Dees_Troy51127312012-09-08 13:08:49 -0400389 TWPartition* Part = Find_Partition_By_Path(Path);
390
391 if (Part)
392 return Part->Is_Mounted();
393 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000394 LOGINFO("Is_Mounted: Unable to find partition for path '%s'\n", Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400395 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400396}
397
Dees_Troy5bf43922012-09-07 16:07:55 -0400398int TWPartitionManager::Mount_Current_Storage(bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400399 string current_storage_path = DataManager::GetCurrentStoragePath();
400
401 if (Mount_By_Path(current_storage_path, Display_Error)) {
402 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
403 if (FreeStorage)
404 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
405 return true;
406 }
407 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400408}
409
Dees_Troy5bf43922012-09-07 16:07:55 -0400410int TWPartitionManager::Mount_Settings_Storage(bool Display_Error) {
411 return Mount_By_Path(DataManager::GetSettingsStoragePath(), Display_Error);
412}
413
414TWPartition* TWPartitionManager::Find_Partition_By_Path(string Path) {
415 std::vector<TWPartition*>::iterator iter;
Dees_Troy38bd7602012-09-14 13:33:53 -0400416 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400417
418 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400419 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path))
Dees_Troy5bf43922012-09-07 16:07:55 -0400420 return (*iter);
421 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400422 return NULL;
423}
424
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400425int TWPartitionManager::Check_Backup_Name(bool Display_Error) {
426 // Check the backup name to ensure that it is the correct size and contains only valid characters
427 // and that a backup with that name doesn't already exist
428 char backup_name[MAX_BACKUP_NAME_LEN];
429 char backup_loc[255], tw_image_dir[255];
430 int copy_size;
431 int index, cur_char;
432 string Backup_Name, Backup_Loc;
433
434 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
435 copy_size = Backup_Name.size();
436 // Check size
437 if (copy_size > MAX_BACKUP_NAME_LEN) {
438 if (Display_Error)
Ethan Yonker74db1572015-10-28 12:44:49 -0500439 gui_err("backup_name_len=Backup name is too long.");
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400440 return -2;
441 }
442
443 // Check each character
444 strncpy(backup_name, Backup_Name.c_str(), copy_size);
Dees_Troya13d74f2013-03-24 08:54:55 -0500445 if (copy_size == 1 && strncmp(backup_name, "0", 1) == 0)
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400446 return 0; // A "0" (zero) means to use the current timestamp for the backup name
447 for (index=0; index<copy_size; index++) {
448 cur_char = (int)backup_name[index];
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500449 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 -0400450 // These are valid characters
451 // Numbers
452 // Upper case letters
453 // Lower case letters
454 // Space
455 // and -_.{}[]
456 } else {
457 if (Display_Error)
Ethan Yonker4adc33e2016-01-22 16:07:11 -0600458 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 -0400459 return -3;
460 }
461 }
462
463 // Check to make sure that a backup with this name doesn't already exist
464 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Loc);
465 strcpy(backup_loc, Backup_Loc.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500466 sprintf(tw_image_dir,"%s/%s", backup_loc, Backup_Name.c_str());
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500467 if (TWFunc::Path_Exists(tw_image_dir)) {
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400468 if (Display_Error)
Matt Mower3c366972015-12-25 19:28:31 -0600469 gui_err("backup_name_exists=A backup with that name already exists!");
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400470 return -4;
471 }
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400472 // No problems found, return 0
473 return 0;
474}
475
Dees_Troy43d8b002012-09-17 16:00:01 -0400476bool TWPartitionManager::Make_MD5(bool generate_md5, string Backup_Folder, string Backup_Filename)
477{
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500478 string command;
Dees_Troy43d8b002012-09-17 16:00:01 -0400479 string Full_File = Backup_Folder + Backup_Filename;
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500480 string result;
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500481 twrpDigest md5sum;
Dees_Troy43d8b002012-09-17 16:00:01 -0400482
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500483 if (!generate_md5)
Dees_Troy43d8b002012-09-17 16:00:01 -0400484 return true;
Dees_Troy43d8b002012-09-17 16:00:01 -0400485
Ethan Yonker74db1572015-10-28 12:44:49 -0500486 TWFunc::GUI_Operation_Text(TW_GENERATE_MD5_TEXT, gui_parse_text("{@generating_md51}"));
487 gui_msg("generating_md52= * Generating md5...");
Dees_Troy43d8b002012-09-17 16:00:01 -0400488
489 if (TWFunc::Path_Exists(Full_File)) {
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500490 md5sum.setfn(Backup_Folder + Backup_Filename);
491 if (md5sum.computeMD5() == 0)
492 if (md5sum.write_md5digest() == 0)
Ethan Yonker74db1572015-10-28 12:44:49 -0500493 gui_msg("md5_created= * MD5 Created.");
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500494 else
495 return -1;
496 else
Ethan Yonker74db1572015-10-28 12:44:49 -0500497 gui_err("md5_error= * MD5 Error!");
Dees_Troy43d8b002012-09-17 16:00:01 -0400498 } else {
499 char filename[512];
500 int index = 0;
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500501 string strfn;
Dees_Troy43d8b002012-09-17 16:00:01 -0400502 sprintf(filename, "%s%03i", Full_File.c_str(), index);
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500503 strfn = filename;
Dees_Troy83bd4832013-05-04 12:39:56 +0000504 while (index < 1000) {
bigbiff bigbiff65a4c732013-03-15 15:17:50 -0400505 md5sum.setfn(filename);
Dees_Troy83bd4832013-05-04 12:39:56 +0000506 if (TWFunc::Path_Exists(filename)) {
507 if (md5sum.computeMD5() == 0) {
508 if (md5sum.write_md5digest() != 0)
509 {
Ethan Yonker74db1572015-10-28 12:44:49 -0500510 gui_err("md5_error= * MD5 Error!");
Dees_Troy83bd4832013-05-04 12:39:56 +0000511 return false;
512 }
513 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500514 gui_err("md5_compute_error= * Error computing MD5.");
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500515 return false;
516 }
517 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400518 index++;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400519 sprintf(filename, "%s%03i", Full_File.c_str(), index);
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500520 strfn = filename;
Dees_Troy43d8b002012-09-17 16:00:01 -0400521 }
522 if (index == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000523 LOGERR("Backup file: '%s' not found!\n", filename);
Dees_Troy43d8b002012-09-17 16:00:01 -0400524 return false;
525 }
Ethan Yonker74db1572015-10-28 12:44:49 -0500526 gui_msg("md5_created= * MD5 Created.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400527 }
528 return true;
529}
530
Ethan Yonker472f5062016-02-25 13:47:30 -0600531bool TWPartitionManager::Backup_Partition(TWPartition* Part, const string& Backup_Folder, bool generate_md5, unsigned long *img_time, unsigned long *file_time, ProgressTracking *progress) {
Dees_Troy43d8b002012-09-17 16:00:01 -0400532 time_t start, stop;
that203bcc92015-05-09 17:27:33 +0200533 int use_compression;
Dees_Troy43d8b002012-09-17 16:00:01 -0400534
bigbiffbf1d6722015-02-14 16:25:59 -0500535 string backup_log = Backup_Folder + "recovery.log";
536
Dees_Troy43d8b002012-09-17 16:00:01 -0400537 if (Part == NULL)
538 return true;
539
Dees_Troy093b7642012-09-21 15:59:38 -0400540 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400541
Tom Hite5a926722014-09-15 01:31:03 +0000542 TWFunc::SetPerformanceMode(true);
Dees_Troy43d8b002012-09-17 16:00:01 -0400543 time(&start);
544
Ethan Yonker472f5062016-02-25 13:47:30 -0600545 if (Part->Backup(Backup_Folder, tar_fork_pid, progress)) {
Tom Hite5a926722014-09-15 01:31:03 +0000546 bool md5Success = false;
Dees_Troy8170a922012-09-18 15:40:25 -0400547 if (Part->Has_SubPartition) {
548 std::vector<TWPartition*>::iterator subpart;
549
550 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500551 if ((*subpart)->Can_Be_Backed_Up && (*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == Part->Mount_Point) {
Ethan Yonker472f5062016-02-25 13:47:30 -0600552 if (!(*subpart)->Backup(Backup_Folder, tar_fork_pid, progress)) {
Tom Hite5a926722014-09-15 01:31:03 +0000553 TWFunc::SetPerformanceMode(false);
bigbiffbf1d6722015-02-14 16:25:59 -0500554 Clean_Backup_Folder(Backup_Folder);
555 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
556 tw_set_default_metadata(backup_log.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -0400557 return false;
Tom Hite5a926722014-09-15 01:31:03 +0000558 }
Dees_Troy2727b992013-08-14 20:09:30 +0000559 sync();
560 sync();
Tom Hite5a926722014-09-15 01:31:03 +0000561 if (!Make_MD5(generate_md5, Backup_Folder, (*subpart)->Backup_FileName)) {
562 TWFunc::SetPerformanceMode(false);
Dees_Troy8170a922012-09-18 15:40:25 -0400563 return false;
Tom Hite5a926722014-09-15 01:31:03 +0000564 }
Dees_Troy8170a922012-09-18 15:40:25 -0400565 }
566 }
567 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400568 time(&stop);
that203bcc92015-05-09 17:27:33 +0200569 int backup_time = (int) difftime(stop, start);
Dees_Troy2673cec2013-04-02 20:22:16 +0000570 LOGINFO("Partition Backup time: %d\n", backup_time);
Dees_Troy43d8b002012-09-17 16:00:01 -0400571 if (Part->Backup_Method == 1) {
Dees_Troy093b7642012-09-21 15:59:38 -0400572 *file_time += backup_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400573 } else {
Dees_Troy093b7642012-09-21 15:59:38 -0400574 *img_time += backup_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400575 }
Tom Hite5a926722014-09-15 01:31:03 +0000576
577 md5Success = Make_MD5(generate_md5, Backup_Folder, Part->Backup_FileName);
578 TWFunc::SetPerformanceMode(false);
579 return md5Success;
Dees_Troy43d8b002012-09-17 16:00:01 -0400580 } else {
bigbiffbf1d6722015-02-14 16:25:59 -0500581 Clean_Backup_Folder(Backup_Folder);
582 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
583 tw_set_default_metadata(backup_log.c_str());
Tom Hite5a926722014-09-15 01:31:03 +0000584 TWFunc::SetPerformanceMode(false);
Dees_Troy43d8b002012-09-17 16:00:01 -0400585 return false;
586 }
bigbiff7abc5fe2015-01-17 16:53:12 -0500587 return 0;
588}
589
bigbiffbf1d6722015-02-14 16:25:59 -0500590void TWPartitionManager::Clean_Backup_Folder(string Backup_Folder) {
591 DIR *d = opendir(Backup_Folder.c_str());
592 struct dirent *p;
593 int r;
594
Ethan Yonker74db1572015-10-28 12:44:49 -0500595 gui_msg("backup_clean=Backup Failed. Cleaning Backup Folder.");
bigbiffbf1d6722015-02-14 16:25:59 -0500596
597 if (d == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500598 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Backup_Folder)(strerror(errno)));
bigbiffbf1d6722015-02-14 16:25:59 -0500599 return;
600 }
601
Matt Mower2b18a532015-02-20 16:58:05 -0600602 while ((p = readdir(d))) {
bigbiffbf1d6722015-02-14 16:25:59 -0500603 if (!strcmp(p->d_name, ".") || !strcmp(p->d_name, ".."))
604 continue;
605
606 string path = Backup_Folder + p->d_name;
607
608 size_t dot = path.find_last_of(".") + 1;
609 if (path.substr(dot) == "win" || path.substr(dot) == "md5" || path.substr(dot) == "info") {
610 r = unlink(path.c_str());
611 if (r != 0) {
612 LOGINFO("Unable to unlink '%s: %s'\n", path.c_str(), strerror(errno));
613 }
614 }
615 }
616 closedir(d);
617}
618
Ethan Yonker472f5062016-02-25 13:47:30 -0600619int TWPartitionManager::Check_Backup_Cancel() {
620 return stop_backup.get_value();
621}
622
bigbiff7abc5fe2015-01-17 16:53:12 -0500623int TWPartitionManager::Cancel_Backup() {
624 string Backup_Folder, Backup_Name, Full_Backup_Path;
625
626 stop_backup.set_value(1);
627
628 if (tar_fork_pid != 0) {
629 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
630 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Folder);
631 Full_Backup_Path = Backup_Folder + "/" + Backup_Name + "/";
632 LOGINFO("Killing pid: %d\n", tar_fork_pid);
633 kill(tar_fork_pid, SIGUSR2);
634 while (kill(tar_fork_pid, 0) == 0) {
635 usleep(1000);
636 }
637 LOGINFO("Backup_Run stopped and returning false, backup cancelled.\n");
638 LOGINFO("Removing directory %s\n", Full_Backup_Path.c_str());
639 TWFunc::removeDir(Full_Backup_Path, false);
640 tar_fork_pid = 0;
641 }
642
643 return 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400644}
645
646int TWPartitionManager::Run_Backup(void) {
Matt Mower2b18a532015-02-20 16:58:05 -0600647 int check, do_md5, partition_count = 0, disable_free_space_check = 0;
Dees_Troya13d74f2013-03-24 08:54:55 -0500648 string Backup_Folder, Backup_Name, Full_Backup_Path, Backup_List, backup_path;
Dees_Troy8170a922012-09-18 15:40:25 -0400649 unsigned long long total_bytes = 0, file_bytes = 0, img_bytes = 0, free_space = 0, img_bytes_remaining, file_bytes_remaining, subpart_size;
Dees_Troy43d8b002012-09-17 16:00:01 -0400650 unsigned long img_time = 0, file_time = 0;
Dees_Troya13d74f2013-03-24 08:54:55 -0500651 TWPartition* backup_part = NULL;
Dees_Troy43d8b002012-09-17 16:00:01 -0400652 TWPartition* storage = NULL;
Dees_Troy8170a922012-09-18 15:40:25 -0400653 std::vector<TWPartition*>::iterator subpart;
Dees_Troy43d8b002012-09-17 16:00:01 -0400654 struct tm *t;
655 time_t start, stop, seconds, total_start, total_stop;
Dees_Troya13d74f2013-03-24 08:54:55 -0500656 size_t start_pos = 0, end_pos = 0;
bigbiff7abc5fe2015-01-17 16:53:12 -0500657 stop_backup.set_value(0);
Dees_Troy43d8b002012-09-17 16:00:01 -0400658 seconds = time(0);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500659 t = localtime(&seconds);
Dees_Troy43d8b002012-09-17 16:00:01 -0400660
661 time(&total_start);
662
663 Update_System_Details();
664
665 if (!Mount_Current_Storage(true))
666 return false;
667
668 DataManager::GetValue(TW_SKIP_MD5_GENERATE_VAR, do_md5);
Dees_Troy4a2a1262012-09-18 09:33:47 -0400669 if (do_md5 == 0)
Dees_Troy43d8b002012-09-17 16:00:01 -0400670 do_md5 = true;
Dees_Troyc5865ab2012-09-24 15:08:04 -0400671 else
672 do_md5 = false;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400673
Dees_Troy43d8b002012-09-17 16:00:01 -0400674 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Folder);
675 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
Ethan Yonker4adc33e2016-01-22 16:07:11 -0600676 if (Backup_Name == gui_lookup("curr_date", "(Current Date)")) {
Dees Troyb21cc642013-09-10 17:36:41 +0000677 Backup_Name = TWFunc::Get_Current_Date();
Ethan Yonker4adc33e2016-01-22 16:07:11 -0600678 } else if (Backup_Name == gui_lookup("auto_generate", "(Auto Generate)") || Backup_Name == "0" || Backup_Name.empty()) {
Dees Troyb21cc642013-09-10 17:36:41 +0000679 TWFunc::Auto_Generate_Backup_Name();
680 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
Dees_Troy43d8b002012-09-17 16:00:01 -0400681 }
Dees_Troy2673cec2013-04-02 20:22:16 +0000682 LOGINFO("Backup Name is: '%s'\n", Backup_Name.c_str());
Dees_Troy43d8b002012-09-17 16:00:01 -0400683 Full_Backup_Path = Backup_Folder + "/" + Backup_Name + "/";
Dees_Troy2673cec2013-04-02 20:22:16 +0000684 LOGINFO("Full_Backup_Path is: '%s'\n", Full_Backup_Path.c_str());
Dees_Troy43d8b002012-09-17 16:00:01 -0400685
Dees_Troy2673cec2013-04-02 20:22:16 +0000686 LOGINFO("Calculating backup details...\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500687 DataManager::GetValue("tw_backup_list", Backup_List);
688 if (!Backup_List.empty()) {
689 end_pos = Backup_List.find(";", start_pos);
690 while (end_pos != string::npos && start_pos < Backup_List.size()) {
691 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
692 backup_part = Find_Partition_By_Path(backup_path);
693 if (backup_part != NULL) {
694 partition_count++;
695 if (backup_part->Backup_Method == 1)
696 file_bytes += backup_part->Backup_Size;
697 else
698 img_bytes += backup_part->Backup_Size;
699 if (backup_part->Has_SubPartition) {
700 std::vector<TWPartition*>::iterator subpart;
701
702 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
Dees_Troy9e0b71c2013-04-08 13:35:37 +0000703 if ((*subpart)->Can_Be_Backed_Up && (*subpart)->Is_Present && (*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == backup_part->Mount_Point) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500704 partition_count++;
705 if ((*subpart)->Backup_Method == 1)
706 file_bytes += (*subpart)->Backup_Size;
707 else
708 img_bytes += (*subpart)->Backup_Size;
709 }
710 }
Dees_Troy8170a922012-09-18 15:40:25 -0400711 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500712 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500713 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 -0400714 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500715 start_pos = end_pos + 1;
716 end_pos = Backup_List.find(";", start_pos);
Dees_Troy43d8b002012-09-17 16:00:01 -0400717 }
718 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400719
720 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500721 gui_msg("no_partition_selected=No partitions selected for backup.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400722 return false;
723 }
724 total_bytes = file_bytes + img_bytes;
Ethan Yonker472f5062016-02-25 13:47:30 -0600725 ProgressTracking progress(total_bytes);
Ethan Yonker74db1572015-10-28 12:44:49 -0500726 gui_msg(Msg("total_partitions_backup= * Total number of partitions to back up: {1}")(partition_count));
727 gui_msg(Msg("total_backup_size= * Total size of all data: {1}MB")(total_bytes / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400728 storage = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
729 if (storage != NULL) {
730 free_space = storage->Free;
Ethan Yonker74db1572015-10-28 12:44:49 -0500731 gui_msg(Msg("available_space= * Available space: {1}MB")(free_space / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400732 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500733 gui_err("unable_locate_storage=Unable to locate storage device.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400734 return false;
735 }
bigbiffbf1d6722015-02-14 16:25:59 -0500736
737 DataManager::GetValue("tw_disable_free_space", disable_free_space_check);
738 if (!disable_free_space_check) {
739 if (free_space - (32 * 1024 * 1024) < total_bytes) {
740 // We require an extra 32MB just in case
Ethan Yonker74db1572015-10-28 12:44:49 -0500741 gui_err("no_space=Not enough free space on storage.");
bigbiffbf1d6722015-02-14 16:25:59 -0500742 return false;
743 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400744 }
745 img_bytes_remaining = img_bytes;
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500746 file_bytes_remaining = file_bytes;
Dees_Troy43d8b002012-09-17 16:00:01 -0400747
Ethan Yonker74db1572015-10-28 12:44:49 -0500748 gui_msg("backup_started=[BACKUP STARTED]");
749 gui_msg(Msg("backup_folder= * Backup Folder: {1}")(Full_Backup_Path));
Dees_Troyd4b22b02013-01-18 17:17:58 +0000750 if (!TWFunc::Recursive_Mkdir(Full_Backup_Path)) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500751 gui_err("fail_backup_folder=Failed to make backup folder.");
Dees_Troyd4b22b02013-01-18 17:17:58 +0000752 return false;
753 }
754
Dees_Troy2673cec2013-04-02 20:22:16 +0000755 DataManager::SetProgress(0.0);
Dees_Troy093b7642012-09-21 15:59:38 -0400756
Dees_Troya13d74f2013-03-24 08:54:55 -0500757 start_pos = 0;
758 end_pos = Backup_List.find(";", start_pos);
759 while (end_pos != string::npos && start_pos < Backup_List.size()) {
bigbiff7abc5fe2015-01-17 16:53:12 -0500760 if (stop_backup.get_value() != 0)
761 return -1;
Dees_Troya13d74f2013-03-24 08:54:55 -0500762 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
763 backup_part = Find_Partition_By_Path(backup_path);
764 if (backup_part != NULL) {
Ethan Yonker472f5062016-02-25 13:47:30 -0600765 if (!Backup_Partition(backup_part, Full_Backup_Path, do_md5, &img_time, &file_time, &progress))
Dees_Troya13d74f2013-03-24 08:54:55 -0500766 return false;
767 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500768 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 -0500769 }
770 start_pos = end_pos + 1;
771 end_pos = Backup_List.find(";", start_pos);
772 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400773
774 // Average BPS
775 if (img_time == 0)
776 img_time = 1;
777 if (file_time == 0)
778 file_time = 1;
Dees_Troy093b7642012-09-21 15:59:38 -0400779 int img_bps = (int)img_bytes / (int)img_time;
bigbiff bigbiff2c57d782013-02-19 10:09:21 -0500780 unsigned long long file_bps = file_bytes / (int)file_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400781
Ethan Yonker472f5062016-02-25 13:47:30 -0600782 if (file_bytes != 0)
783 gui_msg(Msg("avg_backup_fs=Average backup rate for file systems: {1} MB/sec")(file_bps / (1024 * 1024)));
784 if (img_bytes != 0)
785 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 -0400786
787 time(&total_stop);
788 int total_time = (int) difftime(total_stop, total_start);
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500789 uint64_t actual_backup_size = du.Get_Folder_Size(Full_Backup_Path);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500790 actual_backup_size /= (1024LLU * 1024LLU);
Dees_Troy43d8b002012-09-17 16:00:01 -0400791
bigbiff bigbiff2c57d782013-02-19 10:09:21 -0500792 int prev_img_bps, use_compression;
793 unsigned long long prev_file_bps;
Dees_Troy093b7642012-09-21 15:59:38 -0400794 DataManager::GetValue(TW_BACKUP_AVG_IMG_RATE, prev_img_bps);
795 img_bps += (prev_img_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500796 img_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -0400797
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500798 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400799 if (use_compression)
800 DataManager::GetValue(TW_BACKUP_AVG_FILE_COMP_RATE, prev_file_bps);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500801 else
Dees_Troy093b7642012-09-21 15:59:38 -0400802 DataManager::GetValue(TW_BACKUP_AVG_FILE_RATE, prev_file_bps);
803 file_bps += (prev_file_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500804 file_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -0400805
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500806 DataManager::SetValue(TW_BACKUP_AVG_IMG_RATE, img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -0400807 if (use_compression)
808 DataManager::SetValue(TW_BACKUP_AVG_FILE_COMP_RATE, file_bps);
809 else
810 DataManager::SetValue(TW_BACKUP_AVG_FILE_RATE, file_bps);
811
Ethan Yonker74db1572015-10-28 12:44:49 -0500812 gui_msg(Msg("total_backed_size=[{1} MB TOTAL BACKED UP]")(actual_backup_size));
Dees_Troy43d8b002012-09-17 16:00:01 -0400813 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -0400814 UnMount_Main_Partitions();
Ethan Yonker56db1c42015-12-20 22:49:00 -0600815 gui_msg(Msg(msg::kHighlight, "backup_completed=[BACKUP COMPLETED IN {1} SECONDS]")(total_time)); // the end
Dees_Troy3f5c4e82013-02-01 15:16:59 +0000816 string backup_log = Full_Backup_Path + "recovery.log";
817 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
Ethan Yonker4b94cfd2014-12-11 10:00:45 -0600818 tw_set_default_metadata(backup_log.c_str());
Dees_Troy3f5c4e82013-02-01 15:16:59 +0000819 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -0400820}
821
Ethan Yonker472f5062016-02-25 13:47:30 -0600822bool TWPartitionManager::Restore_Partition(TWPartition* Part, const string& Restore_Name, ProgressTracking *progress) {
Dees_Troy4a2a1262012-09-18 09:33:47 -0400823 time_t Start, Stop;
Tom Hite5a926722014-09-15 01:31:03 +0000824 TWFunc::SetPerformanceMode(true);
Dees_Troy4a2a1262012-09-18 09:33:47 -0400825 time(&Start);
Ethan Yonker472f5062016-02-25 13:47:30 -0600826
827 if (!Part->Restore(Restore_Name, progress)) {
Tom Hite5a926722014-09-15 01:31:03 +0000828 TWFunc::SetPerformanceMode(false);
Dees_Troy4a2a1262012-09-18 09:33:47 -0400829 return false;
Tom Hite5a926722014-09-15 01:31:03 +0000830 }
Dees_Troy8170a922012-09-18 15:40:25 -0400831 if (Part->Has_SubPartition) {
832 std::vector<TWPartition*>::iterator subpart;
833
834 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
835 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == Part->Mount_Point) {
Ethan Yonker472f5062016-02-25 13:47:30 -0600836 if (!(*subpart)->Restore(Restore_Name, progress)) {
Tom Hite5a926722014-09-15 01:31:03 +0000837 TWFunc::SetPerformanceMode(false);
Dees_Troy8170a922012-09-18 15:40:25 -0400838 return false;
Tom Hite5a926722014-09-15 01:31:03 +0000839 }
Dees_Troy8170a922012-09-18 15:40:25 -0400840 }
841 }
842 }
Dees_Troy4a2a1262012-09-18 09:33:47 -0400843 time(&Stop);
Tom Hite5a926722014-09-15 01:31:03 +0000844 TWFunc::SetPerformanceMode(false);
z31s1g2053fe02016-02-04 06:45:18 +0100845 gui_msg(Msg("restore_part_done=[{1} done ({2} seconds)]")(Part->Backup_Display_Name)((int)difftime(Stop, Start)));
Dees_Troy4a2a1262012-09-18 09:33:47 -0400846 return true;
847}
848
Ethan Yonker472f5062016-02-25 13:47:30 -0600849int TWPartitionManager::Run_Restore(const string& Restore_Name) {
Dees_Troy4a2a1262012-09-18 09:33:47 -0400850 int check_md5, check, partition_count = 0;
Dees_Troya13d74f2013-03-24 08:54:55 -0500851 TWPartition* restore_part = NULL;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400852 time_t rStart, rStop;
853 time(&rStart);
Dees_Troya13d74f2013-03-24 08:54:55 -0500854 string Restore_List, restore_path;
855 size_t start_pos = 0, end_pos;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500856 unsigned long long total_restore_size = 0, already_restored_size = 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400857
Ethan Yonker74db1572015-10-28 12:44:49 -0500858 gui_msg("restore_started=[RESTORE STARTED]");
859 gui_msg(Msg("restore_folder=Restore folder: '{1}'")(Restore_Name));
Dees_Troy43d8b002012-09-17 16:00:01 -0400860
Dees_Troy4a2a1262012-09-18 09:33:47 -0400861 if (!Mount_Current_Storage(true))
862 return false;
863
864 DataManager::GetValue(TW_SKIP_MD5_CHECK_VAR, check_md5);
Dees_Troya13d74f2013-03-24 08:54:55 -0500865 if (check_md5 > 0) {
866 // Check MD5 files first before restoring to ensure that all of them match before starting a restore
Ethan Yonker74db1572015-10-28 12:44:49 -0500867 TWFunc::GUI_Operation_Text(TW_VERIFY_MD5_TEXT, gui_parse_text("{@verifying_md5}"));
868 gui_msg("verifying_md5=Verifying MD5");
Dees_Troya13d74f2013-03-24 08:54:55 -0500869 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500870 gui_msg("skip_md5=Skipping MD5 check based on user setting.");
Dees_Troya13d74f2013-03-24 08:54:55 -0500871 }
Ethan Yonker74db1572015-10-28 12:44:49 -0500872 gui_msg("calc_restore=Calculating restore details...");
Dees_Troya13d74f2013-03-24 08:54:55 -0500873 DataManager::GetValue("tw_restore_selected", Restore_List);
874 if (!Restore_List.empty()) {
875 end_pos = Restore_List.find(";", start_pos);
876 while (end_pos != string::npos && start_pos < Restore_List.size()) {
877 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
878 restore_part = Find_Partition_By_Path(restore_path);
879 if (restore_part != NULL) {
Ethan Yonkereb32b1f2015-05-18 10:23:03 -0500880 if (restore_part->Mount_Read_Only) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500881 gui_msg(Msg(msg::kError, "restore_read_only=Cannot restore {1} -- mounted read only.")(restore_part->Backup_Display_Name));
Ethan Yonkereb32b1f2015-05-18 10:23:03 -0500882 return false;
883 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500884 if (check_md5 > 0 && !restore_part->Check_MD5(Restore_Name))
885 return false;
Ethan Yonkereb32b1f2015-05-18 10:23:03 -0500886 partition_count++;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500887 total_restore_size += restore_part->Get_Restore_Size(Restore_Name);
Dees_Troya13d74f2013-03-24 08:54:55 -0500888 if (restore_part->Has_SubPartition) {
889 std::vector<TWPartition*>::iterator subpart;
890
891 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
892 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == restore_part->Mount_Point) {
bigbiff bigbiff07338812014-03-30 14:56:41 -0400893 if (check_md5 > 0 && !(*subpart)->Check_MD5(Restore_Name))
Dees_Troya13d74f2013-03-24 08:54:55 -0500894 return false;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500895 total_restore_size += (*subpart)->Get_Restore_Size(Restore_Name);
Dees_Troya13d74f2013-03-24 08:54:55 -0500896 }
897 }
898 }
899 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500900 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -0500901 }
902 start_pos = end_pos + 1;
903 end_pos = Restore_List.find(";", start_pos);
Dees_Troy4a2a1262012-09-18 09:33:47 -0400904 }
Dees_Troy4a2a1262012-09-18 09:33:47 -0400905 }
Dees_Troy4a2a1262012-09-18 09:33:47 -0400906
907 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500908 gui_err("no_part_restore=No partitions selected for restore.");
Dees_Troy4a2a1262012-09-18 09:33:47 -0400909 return false;
910 }
911
Ethan Yonker74db1572015-10-28 12:44:49 -0500912 gui_msg(Msg("restore_part_count=Restoring {1} partitions...")(partition_count));
913 gui_msg(Msg("total_restore_size=Total restore size is {1}MB")(total_restore_size / 1048576));
Dees_Troy2673cec2013-04-02 20:22:16 +0000914 DataManager::SetProgress(0.0);
Ethan Yonker472f5062016-02-25 13:47:30 -0600915 ProgressTracking progress(total_restore_size);
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500916
Dees_Troya13d74f2013-03-24 08:54:55 -0500917 start_pos = 0;
918 if (!Restore_List.empty()) {
919 end_pos = Restore_List.find(";", start_pos);
920 while (end_pos != string::npos && start_pos < Restore_List.size()) {
921 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
922 restore_part = Find_Partition_By_Path(restore_path);
923 if (restore_part != NULL) {
924 partition_count++;
Ethan Yonker472f5062016-02-25 13:47:30 -0600925 if (!Restore_Partition(restore_part, Restore_Name, &progress))
Dees_Troya13d74f2013-03-24 08:54:55 -0500926 return false;
927 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500928 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -0500929 }
930 start_pos = end_pos + 1;
931 end_pos = Restore_List.find(";", start_pos);
932 }
933 }
Ethan Yonker74db1572015-10-28 12:44:49 -0500934 TWFunc::GUI_Operation_Text(TW_UPDATE_SYSTEM_DETAILS_TEXT, gui_parse_text("{@updating_system_details}"));
Dees_Troy43d8b002012-09-17 16:00:01 -0400935 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -0400936 UnMount_Main_Partitions();
Dees_Troy4a2a1262012-09-18 09:33:47 -0400937 time(&rStop);
Matt Mower3c366972015-12-25 19:28:31 -0600938 gui_msg(Msg(msg::kHighlight, "restore_completed=[RESTORE COMPLETED IN {1} SECONDS]")((int)difftime(rStop,rStart)));
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500939 DataManager::SetValue("tw_file_progress", "");
Dees_Troy63c8df72012-09-10 14:02:05 -0400940 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -0400941}
942
943void TWPartitionManager::Set_Restore_Files(string Restore_Name) {
Dees_Troy63c8df72012-09-10 14:02:05 -0400944 // Start with the default values
Dees_Troya13d74f2013-03-24 08:54:55 -0500945 string Restore_List;
Dees_Troy83bd4832013-05-04 12:39:56 +0000946 bool get_date = true, check_encryption = true;
947
948 DataManager::SetValue("tw_restore_encrypted", 0);
Dees_Troy63c8df72012-09-10 14:02:05 -0400949
950 DIR* d;
951 d = opendir(Restore_Name.c_str());
952 if (d == NULL)
953 {
Ethan Yonker74db1572015-10-28 12:44:49 -0500954 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Restore_Name)(strerror(errno)));
Dees_Troy63c8df72012-09-10 14:02:05 -0400955 return;
956 }
957
958 struct dirent* de;
959 while ((de = readdir(d)) != NULL)
960 {
961 // Strip off three components
962 char str[256];
963 char* label;
964 char* fstype = NULL;
965 char* extn = NULL;
966 char* ptr;
967
968 strcpy(str, de->d_name);
969 if (strlen(str) <= 2)
970 continue;
971
972 if (get_date) {
973 char file_path[255];
974 struct stat st;
975
976 strcpy(file_path, Restore_Name.c_str());
977 strcat(file_path, "/");
978 strcat(file_path, str);
979 stat(file_path, &st);
980 string backup_date = ctime((const time_t*)(&st.st_mtime));
981 DataManager::SetValue(TW_RESTORE_FILE_DATE, backup_date);
982 get_date = false;
983 }
984
985 label = str;
986 ptr = label;
987 while (*ptr && *ptr != '.') ptr++;
988 if (*ptr == '.')
989 {
990 *ptr = 0x00;
991 ptr++;
992 fstype = ptr;
993 }
994 while (*ptr && *ptr != '.') ptr++;
995 if (*ptr == '.')
996 {
997 *ptr = 0x00;
998 ptr++;
999 extn = ptr;
1000 }
1001
Dees_Troy83bd4832013-05-04 12:39:56 +00001002 if (fstype == NULL || extn == NULL || strcmp(fstype, "log") == 0) continue;
Dees_Troya13d74f2013-03-24 08:54:55 -05001003 int extnlength = strlen(extn);
Dees_Troy83bd4832013-05-04 12:39:56 +00001004 if (extnlength != 3 && extnlength != 6) continue;
1005 if (extnlength >= 3 && strncmp(extn, "win", 3) != 0) continue;
1006 //if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
1007
1008 if (check_encryption) {
1009 string filename = Restore_Name + "/";
1010 filename += de->d_name;
1011 if (TWFunc::Get_File_Type(filename) == 2) {
1012 LOGINFO("'%s' is encrypted\n", filename.c_str());
1013 DataManager::SetValue("tw_restore_encrypted", 1);
1014 }
1015 }
Dees_Troya13d74f2013-03-24 08:54:55 -05001016 if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
Dees_Troy63c8df72012-09-10 14:02:05 -04001017
1018 TWPartition* Part = Find_Partition_By_Path(label);
1019 if (Part == NULL)
1020 {
Ethan Yonker74db1572015-10-28 12:44:49 -05001021 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 -04001022 continue;
1023 }
1024
1025 Part->Backup_FileName = de->d_name;
1026 if (strlen(extn) > 3) {
1027 Part->Backup_FileName.resize(Part->Backup_FileName.size() - strlen(extn) + 3);
1028 }
1029
James Christopher Adduonofddbdfa2016-02-29 15:08:11 -05001030 if (!Part->Is_SubPartition)
1031 Restore_List += Part->Backup_Path + ";";
Dees_Troy63c8df72012-09-10 14:02:05 -04001032 }
1033 closedir(d);
1034
Dees_Troya13d74f2013-03-24 08:54:55 -05001035 // Set the final value
1036 DataManager::SetValue("tw_restore_list", Restore_List);
1037 DataManager::SetValue("tw_restore_selected", Restore_List);
Dees_Troy51a0e822012-09-05 15:24:24 -04001038 return;
1039}
1040
1041int TWPartitionManager::Wipe_By_Path(string Path) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001042 std::vector<TWPartition*>::iterator iter;
1043 int ret = false;
1044 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -04001045 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy63c8df72012-09-10 14:02:05 -04001046
1047 // Iterate through all partitions
1048 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -04001049 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troye58d5262012-09-21 12:27:57 -04001050 if (Path == "/and-sec")
1051 ret = (*iter)->Wipe_AndSec();
1052 else
1053 ret = (*iter)->Wipe();
Dees_Troy63c8df72012-09-10 14:02:05 -04001054 found = true;
1055 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1056 (*iter)->Wipe();
1057 }
1058 }
1059 if (found) {
1060 return ret;
1061 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001062 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 -04001063 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -04001064}
1065
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001066int TWPartitionManager::Wipe_By_Path(string Path, string New_File_System) {
1067 std::vector<TWPartition*>::iterator iter;
1068 int ret = false;
1069 bool found = false;
1070 string Local_Path = TWFunc::Get_Root_Path(Path);
1071
1072 // Iterate through all partitions
1073 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1074 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1075 if (Path == "/and-sec")
1076 ret = (*iter)->Wipe_AndSec();
1077 else
1078 ret = (*iter)->Wipe(New_File_System);
1079 found = true;
1080 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1081 (*iter)->Wipe(New_File_System);
1082 }
1083 }
1084 if (found) {
1085 return ret;
1086 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001087 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 -05001088 return false;
1089}
1090
Dees_Troy51a0e822012-09-05 15:24:24 -04001091int TWPartitionManager::Factory_Reset(void) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001092 std::vector<TWPartition*>::iterator iter;
1093 int ret = true;
1094
1095 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001096 if ((*iter)->Wipe_During_Factory_Reset && (*iter)->Is_Present) {
Ethan Yonker89583ef2015-08-26 09:01:59 -05001097#ifdef TW_OEM_BUILD
1098 if ((*iter)->Mount_Point == "/data") {
1099 if (!(*iter)->Wipe_Encryption())
1100 ret = false;
1101 } else {
1102#endif
1103 if (!(*iter)->Wipe())
1104 ret = false;
1105#ifdef TW_OEM_BUILD
1106 }
1107#endif
Dees_Troy094207a2012-09-26 12:00:39 -04001108 } else if ((*iter)->Has_Android_Secure) {
1109 if (!(*iter)->Wipe_AndSec())
1110 ret = false;
Dees_Troy63c8df72012-09-10 14:02:05 -04001111 }
1112 }
1113 return ret;
Dees_Troy51a0e822012-09-05 15:24:24 -04001114}
1115
Dees_Troy38bd7602012-09-14 13:33:53 -04001116int TWPartitionManager::Wipe_Dalvik_Cache(void) {
1117 struct stat st;
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001118 vector <string> dir;
Dees_Troy38bd7602012-09-14 13:33:53 -04001119
1120 if (!Mount_By_Path("/data", true))
1121 return false;
1122
1123 if (!Mount_By_Path("/cache", true))
1124 return false;
1125
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001126 dir.push_back("/data/dalvik-cache");
1127 dir.push_back("/cache/dalvik-cache");
1128 dir.push_back("/cache/dc");
Ethan Yonker74db1572015-10-28 12:44:49 -05001129 gui_msg("wiping_dalvik=Wiping Dalvik Cache Directories...");
Dees_Troya13d74f2013-03-24 08:54:55 -05001130 for (unsigned i = 0; i < dir.size(); ++i) {
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001131 if (stat(dir.at(i).c_str(), &st) == 0) {
1132 TWFunc::removeDir(dir.at(i), false);
Ethan Yonker74db1572015-10-28 12:44:49 -05001133 gui_msg(Msg("cleaned=Cleaned: {1}...")(dir.at(i)));
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001134 }
1135 }
Dees_Troy38bd7602012-09-14 13:33:53 -04001136 TWPartition* sdext = Find_Partition_By_Path("/sd-ext");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001137 if (sdext && sdext->Is_Present && sdext->Mount(false))
1138 {
1139 if (stat("/sd-ext/dalvik-cache", &st) == 0)
1140 {
1141 TWFunc::removeDir("/sd-ext/dalvik-cache", false);
Ethan Yonker74db1572015-10-28 12:44:49 -05001142 gui_msg(Msg("cleaned=Cleaned: {1}...")("/sd-ext/dalvik-cache"));
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001143 }
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001144 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001145 gui_msg("dalvik_done=-- Dalvik Cache Directories Wipe Complete!");
Dees_Troy38bd7602012-09-14 13:33:53 -04001146 return true;
1147}
1148
1149int TWPartitionManager::Wipe_Rotate_Data(void) {
1150 if (!Mount_By_Path("/data", true))
1151 return false;
1152
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001153 unlink("/data/misc/akmd*");
1154 unlink("/data/misc/rild*");
Dees_Troy2673cec2013-04-02 20:22:16 +00001155 gui_print("Rotation data wiped.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001156 return true;
1157}
1158
1159int TWPartitionManager::Wipe_Battery_Stats(void) {
1160 struct stat st;
1161
1162 if (!Mount_By_Path("/data", true))
1163 return false;
1164
1165 if (0 != stat("/data/system/batterystats.bin", &st)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001166 gui_print("No Battery Stats Found. No Need To Wipe.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001167 } else {
1168 remove("/data/system/batterystats.bin");
Dees_Troy2673cec2013-04-02 20:22:16 +00001169 gui_print("Cleared battery stats.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001170 }
1171 return true;
1172}
1173
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001174int TWPartitionManager::Wipe_Android_Secure(void) {
1175 std::vector<TWPartition*>::iterator iter;
1176 int ret = false;
1177 bool found = false;
1178
1179 // Iterate through all partitions
1180 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1181 if ((*iter)->Has_Android_Secure) {
1182 ret = (*iter)->Wipe_AndSec();
1183 found = true;
1184 }
1185 }
1186 if (found) {
1187 return ret;
1188 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001189 gui_err("no_andsec=No android secure partitions found.");
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001190 }
1191 return false;
1192}
1193
Dees_Troy38bd7602012-09-14 13:33:53 -04001194int TWPartitionManager::Format_Data(void) {
1195 TWPartition* dat = Find_Partition_By_Path("/data");
1196
1197 if (dat != NULL) {
1198 if (!dat->UnMount(true))
1199 return false;
1200
1201 return dat->Wipe_Encryption();
1202 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001203 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001204 return false;
1205 }
1206 return false;
1207}
1208
1209int TWPartitionManager::Wipe_Media_From_Data(void) {
1210 TWPartition* dat = Find_Partition_By_Path("/data");
1211
1212 if (dat != NULL) {
1213 if (!dat->Has_Data_Media) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001214 LOGERR("This device does not have /data/media\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001215 return false;
1216 }
1217 if (!dat->Mount(true))
1218 return false;
1219
Ethan Yonker74db1572015-10-28 12:44:49 -05001220 gui_msg("wiping_datamedia=Wiping internal storage -- /data/media...");
Ethan Yonker726a0202014-12-16 20:01:38 -06001221 Remove_MTP_Storage(dat->MTP_Storage_ID);
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001222 TWFunc::removeDir("/data/media", false);
xiaolu9416f4f2015-06-04 08:22:23 +08001223 dat->Recreate_Media_Folder();
Ethan Yonker726a0202014-12-16 20:01:38 -06001224 Add_MTP_Storage(dat->MTP_Storage_ID);
Dees_Troy38bd7602012-09-14 13:33:53 -04001225 return true;
1226 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001227 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001228 return false;
1229 }
1230 return false;
1231}
1232
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001233int TWPartitionManager::Repair_By_Path(string Path, bool Display_Error) {
1234 std::vector<TWPartition*>::iterator iter;
1235 int ret = false;
1236 bool found = false;
1237 string Local_Path = TWFunc::Get_Root_Path(Path);
1238
1239 if (Local_Path == "/tmp" || Local_Path == "/")
1240 return true;
1241
1242 // Iterate through all partitions
1243 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1244 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1245 ret = (*iter)->Repair();
1246 found = true;
1247 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1248 (*iter)->Repair();
1249 }
1250 }
1251 if (found) {
1252 return ret;
1253 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001254 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 -05001255 } else {
1256 LOGINFO("Repair: Unable to find partition for path '%s'\n", Local_Path.c_str());
1257 }
1258 return false;
1259}
1260
Ethan Yonkera2719152015-05-28 09:44:41 -05001261int TWPartitionManager::Resize_By_Path(string Path, bool Display_Error) {
1262 std::vector<TWPartition*>::iterator iter;
1263 int ret = false;
1264 bool found = false;
1265 string Local_Path = TWFunc::Get_Root_Path(Path);
1266
1267 if (Local_Path == "/tmp" || Local_Path == "/")
1268 return true;
1269
1270 // Iterate through all partitions
1271 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1272 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1273 ret = (*iter)->Resize();
1274 found = true;
1275 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1276 (*iter)->Resize();
1277 }
1278 }
1279 if (found) {
1280 return ret;
1281 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001282 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 -05001283 } else {
1284 LOGINFO("Resize: Unable to find partition for path '%s'\n", Local_Path.c_str());
1285 }
1286 return false;
1287}
1288
Dees_Troy51a0e822012-09-05 15:24:24 -04001289void TWPartitionManager::Update_System_Details(void) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001290 std::vector<TWPartition*>::iterator iter;
Dees_Troy51127312012-09-08 13:08:49 -04001291 int data_size = 0;
Dees_Troy5bf43922012-09-07 16:07:55 -04001292
Ethan Yonker74db1572015-10-28 12:44:49 -05001293 gui_msg("update_part_details=Updating partition details...");
Dees_Troy5bf43922012-09-07 16:07:55 -04001294 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy51127312012-09-08 13:08:49 -04001295 if ((*iter)->Can_Be_Mounted) {
1296 (*iter)->Update_Size(true);
1297 if ((*iter)->Mount_Point == "/system") {
1298 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1299 DataManager::SetValue(TW_BACKUP_SYSTEM_SIZE, backup_display_size);
1300 } else if ((*iter)->Mount_Point == "/data" || (*iter)->Mount_Point == "/datadata") {
1301 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
1302 } else if ((*iter)->Mount_Point == "/cache") {
1303 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1304 DataManager::SetValue(TW_BACKUP_CACHE_SIZE, backup_display_size);
1305 } else if ((*iter)->Mount_Point == "/sd-ext") {
1306 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1307 DataManager::SetValue(TW_BACKUP_SDEXT_SIZE, backup_display_size);
1308 if ((*iter)->Backup_Size == 0) {
1309 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 0);
1310 DataManager::SetValue(TW_BACKUP_SDEXT_VAR, 0);
1311 } else
1312 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 1);
Dees_Troye58d5262012-09-21 12:27:57 -04001313 } else if ((*iter)->Has_Android_Secure) {
Dees_Troy8170a922012-09-18 15:40:25 -04001314 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troye58d5262012-09-21 12:27:57 -04001315 DataManager::SetValue(TW_BACKUP_ANDSEC_SIZE, backup_display_size);
Dees_Troy8170a922012-09-18 15:40:25 -04001316 if ((*iter)->Backup_Size == 0) {
1317 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 0);
1318 DataManager::SetValue(TW_BACKUP_ANDSEC_VAR, 0);
1319 } else
1320 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 1);
Dees_Troy2c50e182012-09-26 20:05:28 -04001321 } else if ((*iter)->Mount_Point == "/boot") {
1322 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1323 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1324 if ((*iter)->Backup_Size == 0) {
1325 DataManager::SetValue("tw_has_boot_partition", 0);
1326 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1327 } else
1328 DataManager::SetValue("tw_has_boot_partition", 1);
Dees_Troy51127312012-09-08 13:08:49 -04001329 }
Dees_Troyaa9cc402012-10-13 12:14:05 -04001330 } else {
1331 // Handle unmountable partitions in case we reset defaults
1332 if ((*iter)->Mount_Point == "/boot") {
1333 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1334 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1335 if ((*iter)->Backup_Size == 0) {
1336 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 0);
1337 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1338 } else
1339 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 1);
1340 } else if ((*iter)->Mount_Point == "/recovery") {
1341 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1342 DataManager::SetValue(TW_BACKUP_RECOVERY_SIZE, backup_display_size);
1343 if ((*iter)->Backup_Size == 0) {
1344 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 0);
1345 DataManager::SetValue(TW_BACKUP_RECOVERY_VAR, 0);
1346 } else
1347 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 1);
Gary Peck82599a82012-11-21 16:23:12 -08001348 } else if ((*iter)->Mount_Point == "/data") {
1349 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troyaa9cc402012-10-13 12:14:05 -04001350 }
Dees_Troy51127312012-09-08 13:08:49 -04001351 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001352 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001353 gui_msg("update_part_details_done=...done");
Dees_Troy51127312012-09-08 13:08:49 -04001354 DataManager::SetValue(TW_BACKUP_DATA_SIZE, data_size);
1355 string current_storage_path = DataManager::GetCurrentStoragePath();
1356 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
Dees_Troy8170a922012-09-18 15:40:25 -04001357 if (FreeStorage != NULL) {
1358 // Attempt to mount storage
1359 if (!FreeStorage->Mount(false)) {
Matt Mower2d50cad2015-12-03 22:26:55 -06001360 gui_msg(Msg(msg::kError, "unable_to_mount_storage=Unable to mount storage"));
1361 DataManager::SetValue(TW_STORAGE_FREE_SIZE, 0);
Dees_Troy8170a922012-09-18 15:40:25 -04001362 } else {
1363 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
1364 }
1365 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001366 LOGINFO("Unable to find storage partition '%s'.\n", current_storage_path.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -04001367 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001368 if (!Write_Fstab())
Dees_Troy2673cec2013-04-02 20:22:16 +00001369 LOGERR("Error creating fstab\n");
Dees_Troy51a0e822012-09-05 15:24:24 -04001370 return;
1371}
1372
1373int TWPartitionManager::Decrypt_Device(string Password) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001374#ifdef TW_INCLUDE_CRYPTO
1375 int ret_val, password_len;
Ethan Yonkera1de1492015-11-04 11:58:27 -06001376 char crypto_state[PROPERTY_VALUE_MAX], crypto_blkdev[PROPERTY_VALUE_MAX], cPassword[255];
Dees_Troy5bf43922012-09-07 16:07:55 -04001377 size_t result;
Ethan Yonker253368a2014-11-25 15:00:52 -06001378 std::vector<TWPartition*>::iterator iter;
Dees_Troy5bf43922012-09-07 16:07:55 -04001379
Ethan Yonker253368a2014-11-25 15:00:52 -06001380 // Mount any partitions that need to be mounted for decrypt
1381 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1382 if ((*iter)->Mount_To_Decrypt) {
1383 (*iter)->Mount(true);
1384 }
a39552696ff55ce2013-01-08 16:14:56 +00001385 }
a39552696ff55ce2013-01-08 16:14:56 +00001386
Ethan Yonkera1de1492015-11-04 11:58:27 -06001387 property_get("ro.crypto.state", crypto_state, "error");
1388 if (strcmp(crypto_state, "error") == 0) {
1389 property_set("ro.crypto.state", "encrypted");
1390 // Sleep for a bit so that services can start if needed
1391 sleep(1);
1392 }
1393
Dees_Troy5bf43922012-09-07 16:07:55 -04001394 strcpy(cPassword, Password.c_str());
a39552696ff55ce2013-01-08 16:14:56 +00001395 int pwret = cryptfs_check_passwd(cPassword);
1396
Ethan Yonker253368a2014-11-25 15:00:52 -06001397 // Unmount any partitions that were needed for decrypt
1398 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1399 if ((*iter)->Mount_To_Decrypt) {
1400 (*iter)->UnMount(false);
1401 }
1402 }
1403
a39552696ff55ce2013-01-08 16:14:56 +00001404 if (pwret != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001405 gui_err("fail_decrypt=Failed to decrypt data.");
Dees_Troy5bf43922012-09-07 16:07:55 -04001406 return -1;
1407 }
a39552696ff55ce2013-01-08 16:14:56 +00001408
Dees_Troy5bf43922012-09-07 16:07:55 -04001409 property_get("ro.crypto.fs_crypto_blkdev", crypto_blkdev, "error");
1410 if (strcmp(crypto_blkdev, "error") == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001411 LOGERR("Error retrieving decrypted data block device.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04001412 } else {
1413 TWPartition* dat = Find_Partition_By_Path("/data");
1414 if (dat != NULL) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001415 DataManager::SetValue(TW_IS_DECRYPTED, 1);
1416 dat->Is_Decrypted = true;
1417 dat->Decrypted_Block_Device = crypto_blkdev;
Gary Peck82599a82012-11-21 16:23:12 -08001418 dat->Setup_File_System(false);
Dees_Troy74fb2e92013-04-15 14:35:47 +00001419 dat->Current_File_System = dat->Fstab_File_System; // Needed if we're ignoring blkid because encrypted devices start out as emmc
Matt Mower3c366972015-12-25 19:28:31 -06001420 gui_msg(Msg("decrypt_success_dev=Data successfully decrypted, new block device: '{1}'")(crypto_blkdev));
a39552696ff55ce2013-01-08 16:14:56 +00001421
Dees_Troy5bf43922012-09-07 16:07:55 -04001422 // Sleep for a bit so that the device will be ready
1423 sleep(1);
Ethan Yonker6277c792014-09-15 14:54:30 -05001424 if (dat->Has_Data_Media && dat->Mount(false) && TWFunc::Path_Exists("/data/media/0")) {
Dees_Troy16b74352012-11-14 22:27:31 +00001425 dat->Storage_Path = "/data/media/0";
1426 dat->Symlink_Path = dat->Storage_Path;
Dees Troy98fb46c2013-12-04 16:56:45 +00001427 DataManager::SetValue("tw_storage_path", "/data/media/0");
Ethan Yonkerec0fa4a2014-11-20 09:51:03 -06001428 DataManager::SetValue("tw_settings_path", "/data/media/0");
Dees_Troy16b74352012-11-14 22:27:31 +00001429 dat->UnMount(false);
Dees_Troydc8bc1b2013-01-17 01:39:28 +00001430 Output_Partition(dat);
Dees_Troy16b74352012-11-14 22:27:31 +00001431 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001432 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001433 UnMount_Main_Partitions();
Dees_Troy5bf43922012-09-07 16:07:55 -04001434 } else
Dees_Troy2673cec2013-04-02 20:22:16 +00001435 LOGERR("Unable to locate data partition.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04001436 }
1437 return 0;
1438#else
Ethan Yonker74db1572015-10-28 12:44:49 -05001439 gui_err("no_crypto_support=No crypto support was compiled into this build.");
Dees_Troy5bf43922012-09-07 16:07:55 -04001440 return -1;
1441#endif
Dees_Troy51a0e822012-09-05 15:24:24 -04001442 return 1;
Dees_Troy51127312012-09-08 13:08:49 -04001443}
1444
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001445int TWPartitionManager::Fix_Contexts(void) {
thata3d31fb2014-12-21 22:27:40 +01001446#ifdef HAVE_SELINUX
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001447 std::vector<TWPartition*>::iterator iter;
1448 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1449 if ((*iter)->Has_Data_Media) {
1450 if ((*iter)->Mount(true)) {
1451 if (fixContexts::fixDataMediaContexts((*iter)->Mount_Point) != 0)
1452 return -1;
1453 }
1454 }
1455 }
Dees_Troy1a650e62012-10-19 20:54:32 -04001456 UnMount_Main_Partitions();
Ethan Yonker74db1572015-10-28 12:44:49 -05001457 gui_msg("done=Done.");
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001458 return 0;
1459#else
1460 LOGERR("Cannot fix contexts, no selinux support present.\n");
1461 return -1;
1462#endif
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04001463}
1464
Ethan Yonker66a19492015-12-10 10:19:45 -06001465TWPartition* TWPartitionManager::Find_Next_Storage(string Path, bool Exclude_Data_Media) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001466 std::vector<TWPartition*>::iterator iter = Partitions.begin();
1467
1468 if (!Path.empty()) {
1469 string Search_Path = TWFunc::Get_Root_Path(Path);
1470 for (; iter != Partitions.end(); iter++) {
Matt Mower9a561dd2016-02-22 21:25:51 -06001471 if ((*iter)->Mount_Point == Search_Path) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001472 iter++;
1473 break;
1474 }
1475 }
1476 }
1477
1478 for (; iter != Partitions.end(); iter++) {
Ethan Yonker66a19492015-12-10 10:19:45 -06001479 if (Exclude_Data_Media && (*iter)->Has_Data_Media) {
1480 // do nothing, do not return this type of partition
1481 } else if ((*iter)->Is_Storage && (*iter)->Is_Present) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001482 return (*iter);
1483 }
1484 }
1485
1486 return NULL;
1487}
1488
Dees_Troyd21618c2012-10-14 18:48:49 -04001489int TWPartitionManager::Open_Lun_File(string Partition_Path, string Lun_File) {
Dees_Troyd21618c2012-10-14 18:48:49 -04001490 TWPartition* Part = Find_Partition_By_Path(Partition_Path);
1491
1492 if (Part == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001493 LOGINFO("Unable to locate '%s' for USB storage mode.", Partition_Path.c_str());
1494 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 -04001495 return false;
1496 }
Ethan Yonker47360be2014-04-01 10:34:34 -05001497 LOGINFO("USB mount '%s', '%s' > '%s'\n", Partition_Path.c_str(), Part->Actual_Block_Device.c_str(), Lun_File.c_str());
1498 if (!Part->UnMount(true) || !Part->Is_Present)
Dees_Troyd21618c2012-10-14 18:48:49 -04001499 return false;
1500
Dees_Troy2673cec2013-04-02 20:22:16 +00001501 if (TWFunc::write_file(Lun_File, Part->Actual_Block_Device)) {
1502 LOGERR("Unable to write to ums lunfile '%s': (%s)\n", Lun_File.c_str(), strerror(errno));
Dees_Troyd21618c2012-10-14 18:48:49 -04001503 return false;
1504 }
Dees_Troyd21618c2012-10-14 18:48:49 -04001505 return true;
1506}
1507
Dees_Troy8170a922012-09-18 15:40:25 -04001508int TWPartitionManager::usb_storage_enable(void) {
Dees_Troyd21618c2012-10-14 18:48:49 -04001509 int has_dual, has_data_media;
Dees_Troy8170a922012-09-18 15:40:25 -04001510 char lun_file[255];
Dees_Troyd21618c2012-10-14 18:48:49 -04001511 bool has_multiple_lun = false;
Dees_Troy8170a922012-09-18 15:40:25 -04001512
Dees_Troy8170a922012-09-18 15:40:25 -04001513 DataManager::GetValue(TW_HAS_DATA_MEDIA, has_data_media);
Ethan Yonker47360be2014-04-01 10:34:34 -05001514 string Lun_File_str = CUSTOM_LUN_FILE;
1515 size_t found = Lun_File_str.find("%");
1516 if (found != string::npos) {
1517 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
1518 if (TWFunc::Path_Exists(lun_file))
1519 has_multiple_lun = true;
1520 }
Ethan Yonker726a0202014-12-16 20:01:38 -06001521 mtp_was_enabled = TWFunc::Toggle_MTP(false); // Must disable MTP for USB Storage
Ethan Yonker47360be2014-04-01 10:34:34 -05001522 if (!has_multiple_lun) {
1523 LOGINFO("Device doesn't have multiple lun files, mount current storage\n");
1524 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
1525 if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) == "/data") {
Ethan Yonker66a19492015-12-10 10:19:45 -06001526 TWPartition* Mount = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05001527 if (Mount) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001528 if (!Open_Lun_File(Mount->Mount_Point, lun_file)) {
1529 goto error_handle;
1530 }
Ethan Yonker47360be2014-04-01 10:34:34 -05001531 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001532 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001533 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05001534 }
1535 } else if (!Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file)) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001536 goto error_handle;
Dees_Troy8170a922012-09-18 15:40:25 -04001537 }
Dees_Troy8170a922012-09-18 15:40:25 -04001538 } else {
Ethan Yonker47360be2014-04-01 10:34:34 -05001539 LOGINFO("Device has multiple lun files\n");
1540 TWPartition* Mount1;
1541 TWPartition* Mount2;
Dees_Troy8170a922012-09-18 15:40:25 -04001542 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
Ethan Yonker66a19492015-12-10 10:19:45 -06001543 Mount1 = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05001544 if (Mount1) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001545 if (!Open_Lun_File(Mount1->Mount_Point, lun_file)) {
1546 goto error_handle;
1547 }
Matt Moweree717062014-04-26 01:46:46 -05001548 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
Ethan Yonker66a19492015-12-10 10:19:45 -06001549 Mount2 = Find_Next_Storage(Mount1->Mount_Point, true);
Matt Mower1fdcdb72016-01-25 20:53:47 -06001550 if (Mount2 && Mount2->Mount_Point != Mount1->Mount_Point) {
Matt Moweree717062014-04-26 01:46:46 -05001551 Open_Lun_File(Mount2->Mount_Point, lun_file);
Ethan Yonker47360be2014-04-01 10:34:34 -05001552 }
1553 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001554 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001555 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05001556 }
Dees_Troy8170a922012-09-18 15:40:25 -04001557 }
Matt Mowerb6ef4782014-11-06 02:24:36 -06001558 property_set("sys.storage.ums_enabled", "1");
HandyMenny37d42992014-10-26 22:15:59 +01001559 property_set("sys.usb.config", "mass_storage,adb");
Dees_Troy8170a922012-09-18 15:40:25 -04001560 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001561error_handle:
1562 if (mtp_was_enabled)
1563 if (!Enable_MTP())
1564 Disable_MTP();
1565 return false;
Dees_Troy8170a922012-09-18 15:40:25 -04001566}
1567
1568int TWPartitionManager::usb_storage_disable(void) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001569 int index, ret;
1570 char lun_file[255], ch[2] = {0, 0};
1571 string str = ch;
Dees_Troy8170a922012-09-18 15:40:25 -04001572
1573 for (index=0; index<2; index++) {
1574 sprintf(lun_file, CUSTOM_LUN_FILE, index);
Dees_Troy2673cec2013-04-02 20:22:16 +00001575 ret = TWFunc::write_file(lun_file, str);
Dees_Troy2673cec2013-04-02 20:22:16 +00001576 if (ret < 0) {
1577 break;
Dees_Troy8170a922012-09-18 15:40:25 -04001578 }
Dees_Troy8170a922012-09-18 15:40:25 -04001579 }
Dees_Troye58d5262012-09-21 12:27:57 -04001580 Mount_All_Storage();
1581 Update_System_Details();
Dees_Troycfd73ef2012-10-12 16:52:00 -04001582 UnMount_Main_Partitions();
Matt Mowerd9cb9062013-12-14 20:34:45 -06001583 property_set("sys.storage.ums_enabled", "0");
HandyMenny37d42992014-10-26 22:15:59 +01001584 property_set("sys.usb.config", "adb");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001585 if (mtp_was_enabled)
1586 if (!Enable_MTP())
1587 Disable_MTP();
Dees_Troy2673cec2013-04-02 20:22:16 +00001588 if (ret < 0 && index == 0) {
1589 LOGERR("Unable to write to ums lunfile '%s'.", lun_file);
1590 return false;
1591 } else {
1592 return true;
1593 }
Dees_Troy8170a922012-09-18 15:40:25 -04001594 return true;
Dees_Troy812660f2012-09-20 09:55:17 -04001595}
1596
1597void TWPartitionManager::Mount_All_Storage(void) {
1598 std::vector<TWPartition*>::iterator iter;
1599
1600 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1601 if ((*iter)->Is_Storage)
1602 (*iter)->Mount(false);
1603 }
Dees_Troy2c50e182012-09-26 20:05:28 -04001604}
Dees_Troy9350b8d2012-09-27 12:38:38 -04001605
Dees_Troyd0384ef2012-10-12 12:15:42 -04001606void TWPartitionManager::UnMount_Main_Partitions(void) {
1607 // Unmounts system and data if data is not data/media
1608 // Also unmounts boot if boot is mountable
Dees_Troy2673cec2013-04-02 20:22:16 +00001609 LOGINFO("Unmounting main partitions...\n");
Dees_Troyd0384ef2012-10-12 12:15:42 -04001610
1611 TWPartition* Boot_Partition = Find_Partition_By_Path("/boot");
1612
1613 UnMount_By_Path("/system", true);
Ethan Yonker6277c792014-09-15 14:54:30 -05001614 if (!datamedia)
1615 UnMount_By_Path("/data", true);
1616
Dees_Troyd0384ef2012-10-12 12:15:42 -04001617 if (Boot_Partition != NULL && Boot_Partition->Can_Be_Mounted)
1618 Boot_Partition->UnMount(true);
1619}
1620
Dees_Troy9350b8d2012-09-27 12:38:38 -04001621int TWPartitionManager::Partition_SDCard(void) {
1622 char mkdir_path[255], temp[255], line[512];
Ethan Yonker483e9f42016-01-11 22:21:18 -06001623 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 -04001624 int ext, swap, total_size = 0, fat_size;
Dees_Troy9350b8d2012-09-27 12:38:38 -04001625
Ethan Yonker74db1572015-10-28 12:44:49 -05001626 gui_msg("start_partition_sd=Partitioning SD Card...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06001627
1628 // Locate and validate device to partition
1629 TWPartition* SDCard = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
1630
Ethan Yonker66a19492015-12-10 10:19:45 -06001631 if (SDCard->Is_Adopted_Storage)
1632 SDCard->Revert_Adopted();
1633
Ethan Yonker1eff6cd2014-09-15 13:30:42 -05001634 if (SDCard == NULL || !SDCard->Removable || SDCard->Has_Data_Media) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001635 gui_err("partition_sd_locate=Unable to locate device to partition.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001636 return false;
1637 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001638
1639 // Unmount everything
Dees_Troy9350b8d2012-09-27 12:38:38 -04001640 if (!SDCard->UnMount(true))
1641 return false;
1642 TWPartition* SDext = Find_Partition_By_Path("/sd-ext");
1643 if (SDext != NULL) {
1644 if (!SDext->UnMount(true))
1645 return false;
1646 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001647 char* swappath = getenv("SWAPPATH");
1648 if (swappath != NULL) {
1649 LOGINFO("Unmounting swap at '%s'\n", swappath);
1650 umount(swappath);
1651 }
Vojtech Bocek05534202013-09-11 08:11:56 +02001652
Ethan Yonker483e9f42016-01-11 22:21:18 -06001653 // Determine block device
1654 if (SDCard->Alternate_Block_Device.empty()) {
1655 SDCard->Find_Actual_Block_Device();
1656 Device = SDCard->Actual_Block_Device;
1657 // Just use the root block device
1658 Device.resize(strlen("/dev/block/mmcblkX"));
1659 } else {
1660 Device = SDCard->Alternate_Block_Device;
1661 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04001662
1663 // Find the size of the block device:
Ethan Yonker483e9f42016-01-11 22:21:18 -06001664 total_size = (int)(TWFunc::IOCTL_Get_Block_Size(Device.c_str()) / (1048576));
Dees_Troy9350b8d2012-09-27 12:38:38 -04001665
1666 DataManager::GetValue("tw_sdext_size", ext);
1667 DataManager::GetValue("tw_swap_size", swap);
1668 DataManager::GetValue("tw_sdpart_file_system", ext_format);
1669 fat_size = total_size - ext - swap;
Ethan Yonker483e9f42016-01-11 22:21:18 -06001670 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);
1671
1672 // Determine partition sizes
1673 if (swap == 0 && ext == 0) {
1674 fat_str = "-0";
1675 } else {
1676 memset(temp, 0, sizeof(temp));
1677 sprintf(temp, "%i", fat_size);
1678 fat_str = temp;
1679 fat_str += "MB";
1680 }
1681 if (swap == 0) {
1682 ext_str = "-0";
1683 } else {
1684 memset(temp, 0, sizeof(temp));
1685 sprintf(temp, "%i", ext);
1686 ext_str = "+";
1687 ext_str += temp;
1688 ext_str += "MB";
1689 }
1690
Dees_Troy9350b8d2012-09-27 12:38:38 -04001691 if (ext + swap > total_size) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001692 gui_err("ext_swap_size=EXT + Swap size is larger than sdcard size.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001693 return false;
1694 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001695
Ethan Yonker74db1572015-10-28 12:44:49 -05001696 gui_msg("remove_part_table=Removing partition table...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06001697 Command = "sgdisk --zap-all " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00001698 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001699 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001700 gui_err("unable_rm_part=Unable to remove partition table.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001701 Update_System_Details();
1702 return false;
1703 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001704 gui_msg(Msg("create_part=Creating {1} partition...")("FAT32"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06001705 Command = "sgdisk --new=0:0:" + fat_str + " --change-name=0:\"Microsoft basic data\" " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00001706 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001707 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001708 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("FAT32"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04001709 return false;
1710 }
1711 if (ext > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001712 gui_msg(Msg("create_part=Creating {1} partition...")("EXT"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06001713 Command = "sgdisk --new=0:0:" + ext_str + " --change-name=0:\"Linux filesystem\" " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00001714 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001715 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001716 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("EXT"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04001717 Update_System_Details();
1718 return false;
1719 }
1720 }
1721 if (swap > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001722 gui_msg(Msg("create_part=Creating {1} partition...")("swap"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06001723 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 +00001724 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001725 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001726 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("swap"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04001727 Update_System_Details();
1728 return false;
1729 }
1730 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001731
1732 // Convert GPT to MBR
1733 Command = "sgdisk --gpttombr " + Device;
1734 if (TWFunc::Exec_Cmd(Command) != 0)
1735 LOGINFO("Failed to covert partition GPT to MBR\n");
1736
1737 // Tell the kernel to rescan the partition table
1738 int fd = open(Device.c_str(), O_RDONLY);
1739 ioctl(fd, BLKRRPART, 0);
1740 close(fd);
1741
1742 string format_device = Device;
1743 if (Device.substr(0, 17) == "/dev/block/mmcblk")
1744 format_device += "p";
1745
1746 // Format new partitions to proper file system
1747 if (fat_size > 0) {
1748 Command = "mkfs.fat " + format_device + "1";
1749 TWFunc::Exec_Cmd(Command);
1750 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04001751 if (ext > 0) {
1752 if (SDext == NULL) {
Ethan Yonker483e9f42016-01-11 22:21:18 -06001753 Command = "mke2fs -t " + ext_format + " -m 0 " + format_device + "2";
1754 gui_msg(Msg("format_sdext_as=Formatting sd-ext as {1}...")(ext_format));
1755 LOGINFO("Formatting sd-ext after partitioning, command: '%s'\n", Command.c_str());
1756 TWFunc::Exec_Cmd(Command);
1757 } else {
1758 SDext->Wipe(ext_format);
Dees_Troy9350b8d2012-09-27 12:38:38 -04001759 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001760 }
1761 if (swap > 0) {
1762 Command = "mkswap " + format_device;
1763 if (ext > 0)
1764 Command += "3";
1765 else
1766 Command += "2";
Vojtech Bocek05534202013-09-11 08:11:56 +02001767 TWFunc::Exec_Cmd(Command);
Dees_Troy9350b8d2012-09-27 12:38:38 -04001768 }
1769
Ethan Yonker483e9f42016-01-11 22:21:18 -06001770 // recreate TWRP folder and rewrite settings - these will be gone after sdcard is partitioned
1771 if (SDCard->Mount(true)) {
1772 string TWRP_Folder = SDCard->Mount_Point + "/TWRP";
1773 mkdir(TWRP_Folder.c_str(), 0777);
1774 DataManager::Flush();
1775 }
1776
Dees_Troy9350b8d2012-09-27 12:38:38 -04001777 Update_System_Details();
Ethan Yonker74db1572015-10-28 12:44:49 -05001778 gui_msg("part_complete=Partitioning complete.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001779 return true;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04001780}
Dees_Troya13d74f2013-03-24 08:54:55 -05001781
1782void TWPartitionManager::Get_Partition_List(string ListType, std::vector<PartitionList> *Partition_List) {
1783 std::vector<TWPartition*>::iterator iter;
1784 if (ListType == "mount") {
1785 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1786 if ((*iter)->Can_Be_Mounted && !(*iter)->Is_SubPartition) {
1787 struct PartitionList part;
1788 part.Display_Name = (*iter)->Display_Name;
1789 part.Mount_Point = (*iter)->Mount_Point;
1790 part.selected = (*iter)->Is_Mounted();
1791 Partition_List->push_back(part);
1792 }
1793 }
1794 } else if (ListType == "storage") {
1795 char free_space[255];
1796 string Current_Storage = DataManager::GetCurrentStoragePath();
1797 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1798 if ((*iter)->Is_Storage) {
1799 struct PartitionList part;
1800 sprintf(free_space, "%llu", (*iter)->Free / 1024 / 1024);
1801 part.Display_Name = (*iter)->Storage_Name + " (";
1802 part.Display_Name += free_space;
1803 part.Display_Name += "MB)";
1804 part.Mount_Point = (*iter)->Storage_Path;
1805 if ((*iter)->Storage_Path == Current_Storage)
1806 part.selected = 1;
1807 else
1808 part.selected = 0;
1809 Partition_List->push_back(part);
1810 }
1811 }
1812 } else if (ListType == "backup") {
1813 char backup_size[255];
Dees_Troy9e0b71c2013-04-08 13:35:37 +00001814 unsigned long long Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -05001815 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy9e0b71c2013-04-08 13:35:37 +00001816 if ((*iter)->Can_Be_Backed_Up && !(*iter)->Is_SubPartition && (*iter)->Is_Present) {
Dees_Troya13d74f2013-03-24 08:54:55 -05001817 struct PartitionList part;
Dees_Troy9e0b71c2013-04-08 13:35:37 +00001818 Backup_Size = (*iter)->Backup_Size;
1819 if ((*iter)->Has_SubPartition) {
1820 std::vector<TWPartition*>::iterator subpart;
1821
1822 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
1823 if ((*subpart)->Is_SubPartition && (*subpart)->Can_Be_Backed_Up && (*subpart)->Is_Present && (*subpart)->SubPartition_Of == (*iter)->Mount_Point)
1824 Backup_Size += (*subpart)->Backup_Size;
1825 }
1826 }
1827 sprintf(backup_size, "%llu", Backup_Size / 1024 / 1024);
Dees_Troya13d74f2013-03-24 08:54:55 -05001828 part.Display_Name = (*iter)->Backup_Display_Name + " (";
1829 part.Display_Name += backup_size;
1830 part.Display_Name += "MB)";
1831 part.Mount_Point = (*iter)->Backup_Path;
1832 part.selected = 0;
1833 Partition_List->push_back(part);
1834 }
1835 }
1836 } else if (ListType == "restore") {
1837 string Restore_List, restore_path;
1838 TWPartition* restore_part = NULL;
1839
1840 DataManager::GetValue("tw_restore_list", Restore_List);
1841 if (!Restore_List.empty()) {
1842 size_t start_pos = 0, end_pos = Restore_List.find(";", start_pos);
1843 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1844 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
Dees_Troy59df9262013-06-19 14:53:57 -05001845 if ((restore_part = Find_Partition_By_Path(restore_path)) != NULL) {
Dees Troy4159aed2014-02-28 17:24:43 +00001846 if ((restore_part->Backup_Name == "recovery" && !restore_part->Can_Be_Backed_Up) || restore_part->Is_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -05001847 // Don't allow restore of recovery (causes problems on some devices)
Dees_Troy59df9262013-06-19 14:53:57 -05001848 // Don't add subpartitions to the list of items
Dees_Troya13d74f2013-03-24 08:54:55 -05001849 } else {
1850 struct PartitionList part;
1851 part.Display_Name = restore_part->Backup_Display_Name;
1852 part.Mount_Point = restore_part->Backup_Path;
1853 part.selected = 1;
1854 Partition_List->push_back(part);
1855 }
1856 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001857 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001858 }
1859 start_pos = end_pos + 1;
1860 end_pos = Restore_List.find(";", start_pos);
1861 }
1862 }
1863 } else if (ListType == "wipe") {
1864 struct PartitionList dalvik;
Ethan Yonker74db1572015-10-28 12:44:49 -05001865 dalvik.Display_Name = gui_parse_text("{@dalvik}");
Dees_Troya13d74f2013-03-24 08:54:55 -05001866 dalvik.Mount_Point = "DALVIK";
1867 dalvik.selected = 0;
1868 Partition_List->push_back(dalvik);
1869 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1870 if ((*iter)->Wipe_Available_in_GUI && !(*iter)->Is_SubPartition) {
1871 struct PartitionList part;
1872 part.Display_Name = (*iter)->Display_Name;
1873 part.Mount_Point = (*iter)->Mount_Point;
1874 part.selected = 0;
1875 Partition_List->push_back(part);
1876 }
1877 if ((*iter)->Has_Android_Secure) {
1878 struct PartitionList part;
1879 part.Display_Name = (*iter)->Backup_Display_Name;
1880 part.Mount_Point = (*iter)->Backup_Path;
1881 part.selected = 0;
1882 Partition_List->push_back(part);
1883 }
Dees_Troy74fb2e92013-04-15 14:35:47 +00001884 if ((*iter)->Has_Data_Media) {
1885 struct PartitionList datamedia;
1886 datamedia.Display_Name = (*iter)->Storage_Name;
1887 datamedia.Mount_Point = "INTERNAL";
1888 datamedia.selected = 0;
1889 Partition_List->push_back(datamedia);
1890 }
Dees_Troya13d74f2013-03-24 08:54:55 -05001891 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06001892 } else if (ListType == "flashimg") {
1893 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1894 if ((*iter)->Can_Flash_Img && (*iter)->Is_Present) {
1895 struct PartitionList part;
1896 part.Display_Name = (*iter)->Backup_Display_Name;
1897 part.Mount_Point = (*iter)->Backup_Path;
1898 part.selected = 0;
1899 Partition_List->push_back(part);
1900 }
1901 }
Dees_Troya13d74f2013-03-24 08:54:55 -05001902 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001903 LOGERR("Unknown list type '%s' requested for TWPartitionManager::Get_Partition_List\n", ListType.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -05001904 }
1905}
1906
1907int TWPartitionManager::Fstab_Processed(void) {
1908 return Partitions.size();
1909}
Dees_Troyd93bda52013-07-03 19:55:19 +00001910
1911void TWPartitionManager::Output_Storage_Fstab(void) {
1912 std::vector<TWPartition*>::iterator iter;
1913 char storage_partition[255];
1914 string Temp;
1915 FILE *fp = fopen("/cache/recovery/storage.fstab", "w");
1916
1917 if (fp == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001918 gui_msg(Msg(msg::kError, "unable_to_open=Unable to open '{1}'.")("/cache/recovery/storage.fstab"));
Dees_Troyd93bda52013-07-03 19:55:19 +00001919 return;
1920 }
1921
1922 // Iterate through all partitions
1923 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1924 if ((*iter)->Is_Storage) {
1925 Temp = (*iter)->Storage_Path + ";" + (*iter)->Storage_Name + ";\n";
1926 strcpy(storage_partition, Temp.c_str());
1927 fwrite(storage_partition, sizeof(storage_partition[0]), strlen(storage_partition) / sizeof(storage_partition[0]), fp);
1928 }
1929 }
1930 fclose(fp);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001931}
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02001932
1933TWPartition *TWPartitionManager::Get_Default_Storage_Partition()
1934{
1935 TWPartition *res = NULL;
1936 for (std::vector<TWPartition*>::iterator iter = Partitions.begin(); iter != Partitions.end(); ++iter) {
1937 if(!(*iter)->Is_Storage)
1938 continue;
1939
1940 if((*iter)->Is_Settings_Storage)
1941 return *iter;
1942
1943 if(!res)
1944 res = *iter;
1945 }
1946 return res;
1947}
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001948
1949bool TWPartitionManager::Enable_MTP(void) {
1950#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05001951 if (mtppid) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001952 gui_err("mtp_already_enabled=MTP already enabled");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001953 return true;
1954 }
1955 //Launch MTP Responder
1956 LOGINFO("Starting MTP\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06001957
1958 int mtppipe[2];
1959
1960 if (pipe(mtppipe) < 0) {
1961 LOGERR("Error creating MTP pipe\n");
1962 return false;
1963 }
1964
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06001965 char old_value[PROPERTY_VALUE_MAX];
1966 property_get("sys.usb.config", old_value, "error");
1967 if (strcmp(old_value, "error") != 0 && strcmp(old_value, "mtp,adb") != 0) {
1968 char vendor[PROPERTY_VALUE_MAX];
1969 char product[PROPERTY_VALUE_MAX];
1970 property_set("sys.usb.config", "none");
1971 property_get("usb.vendor", vendor, "18D1");
1972 property_get("usb.product.mtpadb", product, "4EE2");
1973 string vendorstr = vendor;
1974 string productstr = product;
1975 TWFunc::write_file("/sys/class/android_usb/android0/idVendor", vendorstr);
1976 TWFunc::write_file("/sys/class/android_usb/android0/idProduct", productstr);
1977 property_set("sys.usb.config", "mtp,adb");
1978 }
Ethan Yonker6d154c42014-09-03 14:19:43 -05001979 /* To enable MTP debug, use the twrp command line feature to
1980 * twrp set tw_mtp_debug 1
1981 */
1982 twrpMtp *mtp = new twrpMtp(DataManager::GetIntValue("tw_mtp_debug"));
Ethan Yonker1b039202015-01-30 10:08:48 -06001983 mtppid = mtp->forkserver(mtppipe);
1984 if (mtppid) {
1985 close(mtppipe[0]); // Host closes read side
1986 mtp_write_fd = mtppipe[1];
1987 DataManager::SetValue("tw_mtp_enabled", 1);
1988 Add_All_MTP_Storage();
1989 return true;
Ethan Yonker726a0202014-12-16 20:01:38 -06001990 } else {
1991 close(mtppipe[0]);
1992 close(mtppipe[1]);
Ethan Yonker74db1572015-10-28 12:44:49 -05001993 gui_err("mtp_fail=Failed to enable MTP");
Ethan Yonker1b039202015-01-30 10:08:48 -06001994 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001995 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001996#else
Ethan Yonker74db1572015-10-28 12:44:49 -05001997 gui_err("no_mtp=MTP support not included");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001998#endif
1999 DataManager::SetValue("tw_mtp_enabled", 0);
2000 return false;
2001}
2002
Ethan Yonker1b039202015-01-30 10:08:48 -06002003void TWPartitionManager::Add_All_MTP_Storage(void) {
2004#ifdef TW_HAS_MTP
2005 std::vector<TWPartition*>::iterator iter;
2006
2007 if (!mtppid)
2008 return; // MTP is not enabled
2009
2010 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2011 if ((*iter)->Is_Storage && (*iter)->Is_Present && (*iter)->Mount(false))
2012 Add_Remove_MTP_Storage((*iter), MTP_MESSAGE_ADD_STORAGE);
2013 }
2014#else
2015 return;
2016#endif
2017}
2018
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002019bool TWPartitionManager::Disable_MTP(void) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002020 char old_value[PROPERTY_VALUE_MAX];
2021 property_get("sys.usb.config", old_value, "error");
2022 if (strcmp(old_value, "adb") != 0) {
2023 char vendor[PROPERTY_VALUE_MAX];
2024 char product[PROPERTY_VALUE_MAX];
2025 property_set("sys.usb.config", "none");
2026 property_get("usb.vendor", vendor, "18D1");
2027 property_get("usb.product.adb", product, "D002");
2028 string vendorstr = vendor;
2029 string productstr = product;
2030 TWFunc::write_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2031 TWFunc::write_file("/sys/class/android_usb/android0/idProduct", productstr);
2032 usleep(2000);
2033 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002034#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002035 if (mtppid) {
Ethan Yonker8613dc02014-09-11 09:28:20 -05002036 LOGINFO("Disabling MTP\n");
2037 int status;
2038 kill(mtppid, SIGKILL);
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002039 mtppid = 0;
Ethan Yonker8613dc02014-09-11 09:28:20 -05002040 // We don't care about the exit value, but this prevents a zombie process
2041 waitpid(mtppid, &status, 0);
Ethan Yonker726a0202014-12-16 20:01:38 -06002042 close(mtp_write_fd);
2043 mtp_write_fd = -1;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002044 }
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002045#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002046 property_set("sys.usb.config", "adb");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002047#ifdef TW_HAS_MTP
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002048 DataManager::SetValue("tw_mtp_enabled", 0);
2049 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002050#endif
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002051 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002052}
Ethan Yonker726a0202014-12-16 20:01:38 -06002053
2054TWPartition* TWPartitionManager::Find_Partition_By_MTP_Storage_ID(unsigned int Storage_ID) {
2055 std::vector<TWPartition*>::iterator iter;
2056
2057 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2058 if ((*iter)->MTP_Storage_ID == Storage_ID)
2059 return (*iter);
2060 }
2061 return NULL;
2062}
2063
2064bool TWPartitionManager::Add_Remove_MTP_Storage(TWPartition* Part, int message_type) {
2065#ifdef TW_HAS_MTP
2066 struct mtpmsg mtp_message;
2067
2068 if (!mtppid)
2069 return false; // MTP is disabled
2070
2071 if (mtp_write_fd < 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002072 LOGINFO("MTP: mtp_write_fd is not set\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002073 return false;
2074 }
2075
2076 if (Part) {
Ethan Yonker4bfabab2014-12-29 09:15:37 -06002077 if (Part->MTP_Storage_ID == 0)
2078 return false;
Ethan Yonker726a0202014-12-16 20:01:38 -06002079 if (message_type == MTP_MESSAGE_REMOVE_STORAGE) {
2080 mtp_message.message_type = MTP_MESSAGE_REMOVE_STORAGE; // Remove
2081 LOGINFO("sending message to remove %i\n", Part->MTP_Storage_ID);
2082 mtp_message.storage_id = Part->MTP_Storage_ID;
2083 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002084 LOGINFO("error sending message to remove storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002085 return false;
2086 } else {
2087 LOGINFO("Message sent, remove storage ID: %i\n", Part->MTP_Storage_ID);
2088 return true;
2089 }
2090 } else if (message_type == MTP_MESSAGE_ADD_STORAGE && Part->Is_Mounted()) {
2091 mtp_message.message_type = MTP_MESSAGE_ADD_STORAGE; // Add
2092 mtp_message.storage_id = Part->MTP_Storage_ID;
2093 mtp_message.path = Part->Storage_Path.c_str();
2094 mtp_message.display = Part->Storage_Name.c_str();
2095 mtp_message.maxFileSize = Part->Get_Max_FileSize();
Ethan Yonker1b039202015-01-30 10:08:48 -06002096 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 -06002097 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002098 LOGINFO("error sending message to add storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002099 return false;
2100 } else {
2101 LOGINFO("Message sent, add storage ID: %i\n", Part->MTP_Storage_ID);
2102 return true;
2103 }
2104 } else {
2105 LOGERR("Unknown MTP message type: %i\n", message_type);
2106 }
2107 } else {
2108 // This hopefully never happens as the error handling should
2109 // occur in the calling function.
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002110 LOGINFO("TWPartitionManager::Add_Remove_MTP_Storage NULL partition given\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002111 }
2112 return true;
2113#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002114 gui_err("no_mtp=MTP support not included");
Ethan Yonker726a0202014-12-16 20:01:38 -06002115 DataManager::SetValue("tw_mtp_enabled", 0);
2116 return false;
2117#endif
2118}
2119
2120bool TWPartitionManager::Add_MTP_Storage(string Mount_Point) {
2121#ifdef TW_HAS_MTP
2122 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2123 if (Part) {
2124 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2125 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002126 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002127 }
2128#endif
2129 return false;
2130}
2131
2132bool TWPartitionManager::Add_MTP_Storage(unsigned int Storage_ID) {
2133#ifdef TW_HAS_MTP
2134 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2135 if (Part) {
2136 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2137 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002138 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002139 }
2140#endif
2141 return false;
2142}
2143
2144bool TWPartitionManager::Remove_MTP_Storage(string Mount_Point) {
2145#ifdef TW_HAS_MTP
2146 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2147 if (Part) {
2148 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2149 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002150 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002151 }
2152#endif
2153 return false;
2154}
2155
2156bool TWPartitionManager::Remove_MTP_Storage(unsigned int Storage_ID) {
2157#ifdef TW_HAS_MTP
2158 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2159 if (Part) {
2160 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2161 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002162 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002163 }
2164#endif
2165 return false;
2166}
Ethan Yonker96af84a2015-01-05 14:58:36 -06002167
2168bool TWPartitionManager::Flash_Image(string Filename) {
2169 int check, partition_count = 0;
2170 TWPartition* flash_part = NULL;
2171 string Flash_List, flash_path;
2172 size_t start_pos = 0, end_pos = 0;
2173
Ethan Yonker74db1572015-10-28 12:44:49 -05002174 gui_msg("image_flash_start=[IMAGE FLASH STARTED]");
2175 gui_msg(Msg("img_to_flash=Image to flash: '{1}'")(Filename));
Ethan Yonker96af84a2015-01-05 14:58:36 -06002176
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002177 if (!TWFunc::Path_Exists(Filename)) {
2178 if (!Mount_By_Path(Filename, true)) {
2179 return false;
2180 }
2181 if (!TWFunc::Path_Exists(Filename)) {
2182 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")(Filename));
2183 return false;
2184 }
2185 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002186
Ethan Yonker74db1572015-10-28 12:44:49 -05002187 gui_msg("calc_restore=Calculating restore details...");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002188 DataManager::GetValue("tw_flash_partition", Flash_List);
2189 if (!Flash_List.empty()) {
2190 end_pos = Flash_List.find(";", start_pos);
2191 while (end_pos != string::npos && start_pos < Flash_List.size()) {
2192 flash_path = Flash_List.substr(start_pos, end_pos - start_pos);
2193 flash_part = Find_Partition_By_Path(flash_path);
2194 if (flash_part != NULL) {
2195 partition_count++;
2196 if (partition_count > 1) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002197 gui_err("too_many_flash=Too many partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002198 return false;
2199 }
2200 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002201 gui_msg(Msg(msg::kError, "flash_unable_locate=Unable to locate '{1}' partition for flashing.")(flash_path));
Ethan Yonker96af84a2015-01-05 14:58:36 -06002202 return false;
2203 }
2204 start_pos = end_pos + 1;
2205 end_pos = Flash_List.find(";", start_pos);
2206 }
2207 }
2208
2209 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002210 gui_err("no_part_flash=No partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002211 return false;
2212 }
2213
2214 DataManager::SetProgress(0.0);
2215 if (flash_part) {
2216 if (!flash_part->Flash_Image(Filename))
2217 return false;
2218 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002219 gui_err("invalid_flash=Invalid flash partition specified.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002220 return false;
2221 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002222 gui_highlight("flash_done=IMAGE FLASH COMPLETED]");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002223 return true;
2224}
Ethan Yonker74db1572015-10-28 12:44:49 -05002225
2226void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value) {
2227 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2228 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002229 if (part->Is_Adopted_Storage) {
2230 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2231 part->Backup_Display_Name = part->Display_Name;
2232 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2233 } else {
2234 part->Display_Name = gui_lookup(resource_name, default_value);
2235 part->Backup_Display_Name = part->Display_Name;
2236 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002237 }
2238}
2239
2240void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value, const char* storage_resource_name, const char* storage_default_value) {
2241 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2242 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002243 if (part->Is_Adopted_Storage) {
2244 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2245 part->Backup_Display_Name = part->Display_Name;
2246 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2247 } else {
2248 part->Display_Name = gui_lookup(resource_name, default_value);
2249 part->Backup_Display_Name = part->Display_Name;
2250 if (part->Is_Storage)
2251 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2252 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002253 }
2254}
2255
2256void TWPartitionManager::Translate_Partition_Display_Names() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002257 LOGINFO("Translating partition display names\n");
Ethan Yonker74db1572015-10-28 12:44:49 -05002258 Translate_Partition("/system", "system", "System");
2259 Translate_Partition("/system_image", "system_image", "System Image");
2260 Translate_Partition("/vendor", "vendor", "Vendor");
2261 Translate_Partition("/vendor_image", "vendor_image", "Vendor Image");
2262 Translate_Partition("/cache", "cache", "Cache");
2263 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage");
2264 Translate_Partition("/boot", "boot", "Boot");
2265 Translate_Partition("/recovery", "recovery", "Recovery");
2266 if (!datamedia) {
2267 Translate_Partition("/sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2268 Translate_Partition("/internal_sd", "sdcard", "SDCard", "sdcard", "SDCard");
2269 Translate_Partition("/internal_sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2270 Translate_Partition("/emmc", "sdcard", "SDCard", "sdcard", "SDCard");
2271 }
2272 Translate_Partition("/external_sd", "microsd", "Micro SDCard", "microsd", "Micro SDCard");
2273 Translate_Partition("/external_sdcard", "microsd", "Micro SDCard", "microsd", "Micro SDCard");
2274 Translate_Partition("/usb-otg", "usbotg", "USB OTG", "usbotg", "USB OTG");
2275 Translate_Partition("/sd-ext", "sdext", "SD-EXT");
2276
2277 // Android secure is a special case
2278 TWPartition* part = PartitionManager.Find_Partition_By_Path("/and-sec");
2279 if (part)
2280 part->Backup_Display_Name = gui_lookup("android_secure", "Android Secure");
2281
2282 // This updates the text on all of the storage selection buttons in the GUI
2283 DataManager::SetBackupFolder();
2284}
Ethan Yonker66a19492015-12-10 10:19:45 -06002285
2286void TWPartitionManager::Decrypt_Adopted() {
2287#ifdef TW_INCLUDE_CRYPTO
2288 if (!Mount_By_Path("/data", false)) {
2289 LOGERR("Cannot decrypt adopted storage because /data will not mount\n");
2290 return;
2291 }
2292 LOGINFO("Decrypt adopted storage starting\n");
2293 char* xmlFile = PageManager::LoadFileToBuffer("/data/system/storage.xml", NULL);
2294 xml_document<> *doc = NULL;
2295 xml_node<>* volumes = NULL;
2296 xml_node<>* volume = NULL;
2297 string Primary_Storage_UUID = "";
2298 if (xmlFile != NULL) {
2299 LOGINFO("successfully loaded storage.xml\n");
2300 doc = new xml_document<>();
2301 doc->parse<0>(xmlFile);
2302 volumes = doc->first_node("volumes");
2303 if (volumes) {
2304 xml_attribute<>* psuuid = volumes->first_attribute("primaryStorageUuid");
2305 if (psuuid) {
2306 Primary_Storage_UUID = psuuid->value();
2307 }
2308 }
2309 }
2310 std::vector<TWPartition*>::iterator adopt;
2311 for (adopt = Partitions.begin(); adopt != Partitions.end(); adopt++) {
2312 if ((*adopt)->Removable && (*adopt)->Is_Present) {
2313 if ((*adopt)->Decrypt_Adopted() == 0) {
2314 if (volumes) {
2315 xml_node<>* volume = volumes->first_node("volume");
2316 while (volume) {
2317 xml_attribute<>* guid = volume->first_attribute("partGuid");
2318 if (guid) {
2319 string GUID = (*adopt)->Adopted_GUID.c_str();
2320 GUID.insert(8, "-");
2321 GUID.insert(13, "-");
2322 GUID.insert(18, "-");
2323 GUID.insert(23, "-");
2324
2325 if (strcasecmp(GUID.c_str(), guid->value()) == 0) {
2326 xml_attribute<>* attr = volume->first_attribute("nickname");
2327 if (attr) {
2328 (*adopt)->Storage_Name = attr->value();
2329 (*adopt)->Display_Name = (*adopt)->Storage_Name;
2330 (*adopt)->Backup_Display_Name = (*adopt)->Storage_Name;
2331 LOGINFO("storage name from storage.xml is '%s'\n", attr->value());
2332 }
2333 attr = volume->first_attribute("fsUuid");
2334 if (attr && !Primary_Storage_UUID.empty() && strcmp(Primary_Storage_UUID.c_str(), attr->value()) == 0) {
2335 TWPartition* Dat = Find_Partition_By_Path("/data");
2336 if (Dat) {
2337 LOGINFO("Internal storage is found on adopted storage '%s'\n", (*adopt)->Display_Name.c_str());
2338 LOGINFO("Changing '%s' to point to '%s'\n", Dat->Symlink_Mount_Point.c_str(), (*adopt)->Storage_Path.c_str());
2339 (*adopt)->Symlink_Mount_Point = Dat->Symlink_Mount_Point;
2340 Dat->Symlink_Mount_Point = "";
2341 // Toggle mounts to ensure that the symlink mount point (probably /sdcard) is mounted to the right location
2342 Dat->UnMount(false);
2343 Dat->Mount(false);
2344 (*adopt)->UnMount(false);
2345 (*adopt)->Mount(false);
2346 Output_Partition((*adopt));
2347 }
2348 }
2349 break;
2350 }
2351 }
2352 volume = volume->next_sibling("volume");
2353 }
2354 }
2355 }
2356 }
2357 }
2358 if (xmlFile) {
2359 doc->clear();
2360 delete doc;
2361 free(xmlFile);
2362 }
2363#else
2364 LOGINFO("Decrypt_Adopted: no crypto support\n");
2365 return;
2366#endif
2367}
Ethan Yonkerfcf3f242016-02-16 12:30:26 -06002368
2369void TWPartitionManager::Remove_Partition_By_Path(string Path) {
2370 std::vector<TWPartition*>::iterator iter;
2371 string Local_Path = TWFunc::Get_Root_Path(Path);
2372
2373 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2374 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
2375 LOGINFO("Found and erasing '%s' from partition list\n", Local_Path.c_str());
2376 Partitions.erase(iter);
2377 return;
2378 }
2379 }
2380}