blob: 572cd7c9308db06ec6473d8c3ab9849a657f5300 [file] [log] [blame]
Dees Troy3be70a82013-10-22 14:25:12 +00001/*
bigbiff7abc5fe2015-01-17 16:53:12 -05002 Copyright 2014 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>
Dees_Troy51a0e822012-09-05 15:24:24 -040033#include "variables.h"
Dees_Troy2673cec2013-04-02 20:22:16 +000034#include "twcommon.h"
Dees_Troy51a0e822012-09-05 15:24:24 -040035#include "partitions.hpp"
Dees_Troy5bf43922012-09-07 16:07:55 -040036#include "data.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040037#include "twrp-functions.hpp"
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -040038#include "fixPermissions.hpp"
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -050039#include "twrpDigest.hpp"
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050040#include "twrpDU.hpp"
Ethan Yonker4b94cfd2014-12-11 10:00:45 -060041#include "set_metadata.h"
bigbiff7abc5fe2015-01-17 16:53:12 -050042#include "tw_atomic.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040043
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040044#ifdef TW_HAS_MTP
45#include "mtp/mtp_MtpServer.hpp"
46#include "mtp/twrpMtp.hpp"
Ethan Yonker726a0202014-12-16 20:01:38 -060047#include "mtp/MtpMessage.hpp"
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040048#endif
49
Dees Troy6f6441d2014-01-23 02:07:03 +000050extern "C" {
51 #include "cutils/properties.h"
52}
53
Dees_Troy5bf43922012-09-07 16:07:55 -040054#ifdef TW_INCLUDE_CRYPTO
Ethan Yonker253368a2014-11-25 15:00:52 -060055 #include "crypto/lollipop/cryptfs.h"
Dees_Troy5bf43922012-09-07 16:07:55 -040056#endif
Dees_Troy51a0e822012-09-05 15:24:24 -040057
Ethan Yonker6277c792014-09-15 14:54:30 -050058extern bool datamedia;
59
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050060TWPartitionManager::TWPartitionManager(void) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040061 mtp_was_enabled = false;
Ethan Yonker726a0202014-12-16 20:01:38 -060062 mtp_write_fd = -1;
bigbiff7abc5fe2015-01-17 16:53:12 -050063 stop_backup.set_value(0);
64 tar_fork_pid = 0;
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050065}
66
Dees_Troy51a0e822012-09-05 15:24:24 -040067int TWPartitionManager::Process_Fstab(string Fstab_Filename, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -040068 FILE *fstabFile;
69 char fstab_line[MAX_FSTAB_LINE_LENGTH];
Dees Troy02a64532014-03-19 15:23:32 +000070 TWPartition* settings_partition = NULL;
Matt Mowerbf4efa32014-04-14 23:25:26 -050071 TWPartition* andsec_partition = NULL;
Ethan Yonker726a0202014-12-16 20:01:38 -060072 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 -040073
74 fstabFile = fopen(Fstab_Filename.c_str(), "rt");
75 if (fstabFile == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +000076 LOGERR("Critical Error: Unable to open fstab at '%s'.\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -040077 return false;
78 }
79
80 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
81 if (fstab_line[0] != '/')
82 continue;
83
Dees_Troy2a923582012-09-20 12:13:34 -040084 if (fstab_line[strlen(fstab_line) - 1] != '\n')
85 fstab_line[strlen(fstab_line)] = '\n';
that9e0593e2014-10-08 00:01:24 +020086 TWPartition* partition = new TWPartition();
Dees_Troy2a923582012-09-20 12:13:34 -040087 string line = fstab_line;
Dees_Troyab10ee22012-09-21 14:27:30 -040088 memset(fstab_line, 0, sizeof(fstab_line));
Dees_Troy2a923582012-09-20 12:13:34 -040089
Dees_Troy5bf43922012-09-07 16:07:55 -040090 if (partition->Process_Fstab_Line(line, Display_Error)) {
Ethan Yonker726a0202014-12-16 20:01:38 -060091 if (partition->Is_Storage) {
92 ++storageid;
93 partition->MTP_Storage_ID = storageid;
94 }
Matt Mowered71fa32014-04-16 13:21:47 -050095 if (!settings_partition && partition->Is_Settings_Storage && partition->Is_Present) {
Dees Troy02a64532014-03-19 15:23:32 +000096 settings_partition = partition;
Dees_Troya13d74f2013-03-24 08:54:55 -050097 } else {
98 partition->Is_Settings_Storage = false;
Dees_Troya13d74f2013-03-24 08:54:55 -050099 }
Matt Mowered71fa32014-04-16 13:21:47 -0500100 if (!andsec_partition && partition->Has_Android_Secure && partition->Is_Present) {
Matt Mowerbf4efa32014-04-14 23:25:26 -0500101 andsec_partition = partition;
102 } else {
103 partition->Has_Android_Secure = false;
104 }
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500105 Partitions.push_back(partition);
Dees_Troy5bf43922012-09-07 16:07:55 -0400106 } else {
107 delete partition;
108 }
109 }
110 fclose(fstabFile);
Ethan Yonker6277c792014-09-15 14:54:30 -0500111 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) {
112 // Attempt to automatically identify /data/media emulated storage devices
113 TWPartition* Dat = Find_Partition_By_Path("/data");
114 if (Dat) {
115 LOGINFO("Using automatic handling for /data/media emulated storage device.\n");
116 datamedia = true;
that9e0593e2014-10-08 00:01:24 +0200117 Dat->Setup_Data_Media();
Ethan Yonker6277c792014-09-15 14:54:30 -0500118 settings_partition = Dat;
Ethan Yonker726a0202014-12-16 20:01:38 -0600119 // Since /data was not considered a storage partition earlier, we still need to assign an MTP ID
120 ++storageid;
121 Dat->MTP_Storage_ID = storageid;
Ethan Yonker6277c792014-09-15 14:54:30 -0500122 }
123 }
Dees Troy02a64532014-03-19 15:23:32 +0000124 if (!settings_partition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500125 std::vector<TWPartition*>::iterator iter;
126 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
127 if ((*iter)->Is_Storage) {
Dees Troy02a64532014-03-19 15:23:32 +0000128 settings_partition = (*iter);
Dees_Troya13d74f2013-03-24 08:54:55 -0500129 break;
130 }
131 }
Dees Troy02a64532014-03-19 15:23:32 +0000132 if (!settings_partition)
Dees_Troy2673cec2013-04-02 20:22:16 +0000133 LOGERR("Unable to locate storage partition for storing settings file.\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500134 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400135 if (!Write_Fstab()) {
136 if (Display_Error)
Dees_Troy2673cec2013-04-02 20:22:16 +0000137 LOGERR("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400138 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000139 LOGINFO("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400140 }
Matt Mowered71fa32014-04-16 13:21:47 -0500141
Matt Mowerbf4efa32014-04-14 23:25:26 -0500142 if (andsec_partition) {
143 Setup_Android_Secure_Location(andsec_partition);
Matt Mowered71fa32014-04-16 13:21:47 -0500144 } else if (settings_partition) {
Matt Mowerbf4efa32014-04-14 23:25:26 -0500145 Setup_Android_Secure_Location(settings_partition);
146 }
Matt Mowered71fa32014-04-16 13:21:47 -0500147 if (settings_partition) {
148 Setup_Settings_Storage_Partition(settings_partition);
149 }
Ethan Yonker253368a2014-11-25 15:00:52 -0600150#ifdef TW_INCLUDE_CRYPTO
Ethan Yonkercceebb82014-11-18 10:17:59 -0600151 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
152 if (Decrypt_Data && Decrypt_Data->Is_Encrypted && !Decrypt_Data->Is_Decrypted) {
153 int password_type = cryptfs_get_password_type();
154 if (password_type == CRYPT_TYPE_DEFAULT) {
155 LOGINFO("Device is encrypted with the default password, attempting to decrypt.\n");
156 if (Decrypt_Device("default_password") == 0) {
157 gui_print("Successfully decrypted with default password.\n");
158 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
159 } else {
160 LOGERR("Unable to decrypt with default password.");
161 LOGERR("You may need to perform a Format Data.\n");
162 }
163 } else {
164 DataManager::SetValue("TW_CRYPTO_TYPE", password_type);
165 }
166 }
167#endif
Dees_Troy51127312012-09-08 13:08:49 -0400168 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -0400169 UnMount_Main_Partitions();
Dees_Troy5bf43922012-09-07 16:07:55 -0400170 return true;
171}
172
173int TWPartitionManager::Write_Fstab(void) {
174 FILE *fp;
175 std::vector<TWPartition*>::iterator iter;
176 string Line;
177
178 fp = fopen("/etc/fstab", "w");
179 if (fp == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000180 LOGINFO("Can not open /etc/fstab.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400181 return false;
182 }
Dees_Troy63c8df72012-09-10 14:02:05 -0400183 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy51127312012-09-08 13:08:49 -0400184 if ((*iter)->Can_Be_Mounted) {
Dees_Troy38bd7602012-09-14 13:33:53 -0400185 Line = (*iter)->Actual_Block_Device + " " + (*iter)->Mount_Point + " " + (*iter)->Current_File_System + " rw\n";
Dees_Troy5bf43922012-09-07 16:07:55 -0400186 fputs(Line.c_str(), fp);
Dees_Troy91862e62013-04-04 23:48:21 +0000187 }
188 // Handle subpartition tracking
189 if ((*iter)->Is_SubPartition) {
190 TWPartition* ParentPartition = Find_Partition_By_Path((*iter)->SubPartition_Of);
191 if (ParentPartition)
192 ParentPartition->Has_SubPartition = true;
193 else
194 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 -0400195 }
196 }
197 fclose(fp);
198 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -0400199}
200
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500201void TWPartitionManager::Setup_Settings_Storage_Partition(TWPartition* Part) {
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500202 DataManager::SetValue("tw_settings_path", Part->Storage_Path);
203 DataManager::SetValue("tw_storage_path", Part->Storage_Path);
204 LOGINFO("Settings storage is '%s'\n", Part->Storage_Path.c_str());
205}
206
Matt Mowerbf4efa32014-04-14 23:25:26 -0500207void TWPartitionManager::Setup_Android_Secure_Location(TWPartition* Part) {
208 if (Part->Has_Android_Secure)
209 Part->Setup_AndSec();
Ethan Yonker6277c792014-09-15 14:54:30 -0500210 else if (!datamedia)
Matt Mowerbf4efa32014-04-14 23:25:26 -0500211 Part->Setup_AndSec();
Matt Mowerbf4efa32014-04-14 23:25:26 -0500212}
213
Dees_Troy8170a922012-09-18 15:40:25 -0400214void TWPartitionManager::Output_Partition_Logging(void) {
215 std::vector<TWPartition*>::iterator iter;
216
217 printf("\n\nPartition Logs:\n");
218 for (iter = Partitions.begin(); iter != Partitions.end(); iter++)
219 Output_Partition((*iter));
220}
221
222void TWPartitionManager::Output_Partition(TWPartition* Part) {
223 unsigned long long mb = 1048576;
224
Gary Peck004d48b2012-11-21 16:28:18 -0800225 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 -0400226 if (Part->Can_Be_Mounted) {
Gary Peck004d48b2012-11-21 16:28:18 -0800227 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 -0400228 }
Gary Peck004d48b2012-11-21 16:28:18 -0800229 printf("\n Flags: ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500230 if (Part->Can_Be_Mounted)
231 printf("Can_Be_Mounted ");
Gary Peck004d48b2012-11-21 16:28:18 -0800232 if (Part->Can_Be_Wiped)
233 printf("Can_Be_Wiped ");
Hashcodedabfd492013-08-29 22:45:30 -0700234 if (Part->Use_Rm_Rf)
235 printf("Use_Rm_Rf ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500236 if (Part->Can_Be_Backed_Up)
237 printf("Can_Be_Backed_Up ");
Gary Peck004d48b2012-11-21 16:28:18 -0800238 if (Part->Wipe_During_Factory_Reset)
239 printf("Wipe_During_Factory_Reset ");
240 if (Part->Wipe_Available_in_GUI)
241 printf("Wipe_Available_in_GUI ");
242 if (Part->Is_SubPartition)
243 printf("Is_SubPartition ");
244 if (Part->Has_SubPartition)
245 printf("Has_SubPartition ");
246 if (Part->Removable)
247 printf("Removable ");
248 if (Part->Is_Present)
249 printf("IsPresent ");
250 if (Part->Can_Be_Encrypted)
251 printf("Can_Be_Encrypted ");
252 if (Part->Is_Encrypted)
253 printf("Is_Encrypted ");
254 if (Part->Is_Decrypted)
255 printf("Is_Decrypted ");
256 if (Part->Has_Data_Media)
257 printf("Has_Data_Media ");
Dees_Troy83bd4832013-05-04 12:39:56 +0000258 if (Part->Can_Encrypt_Backup)
259 printf("Can_Encrypt_Backup ");
260 if (Part->Use_Userdata_Encryption)
261 printf("Use_Userdata_Encryption ");
Gary Peck004d48b2012-11-21 16:28:18 -0800262 if (Part->Has_Android_Secure)
263 printf("Has_Android_Secure ");
264 if (Part->Is_Storage)
265 printf("Is_Storage ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500266 if (Part->Is_Settings_Storage)
267 printf("Is_Settings_Storage ");
Dees_Troy68cab492012-12-12 19:29:35 +0000268 if (Part->Ignore_Blkid)
269 printf("Ignore_Blkid ");
Dees_Troy16c2b312013-01-15 16:51:18 +0000270 if (Part->Retain_Layout_Version)
271 printf("Retain_Layout_Version ");
Ethan Yonker253368a2014-11-25 15:00:52 -0600272 if (Part->Mount_To_Decrypt)
273 printf("Mount_To_Decrypt ");
Ethan Yonker96af84a2015-01-05 14:58:36 -0600274 if (Part->Can_Flash_Img)
275 printf("Can_Flash_Img ");
Gary Peck004d48b2012-11-21 16:28:18 -0800276 printf("\n");
277 if (!Part->SubPartition_Of.empty())
278 printf(" SubPartition_Of: %s\n", Part->SubPartition_Of.c_str());
279 if (!Part->Symlink_Path.empty())
280 printf(" Symlink_Path: %s\n", Part->Symlink_Path.c_str());
281 if (!Part->Symlink_Mount_Point.empty())
282 printf(" Symlink_Mount_Point: %s\n", Part->Symlink_Mount_Point.c_str());
283 if (!Part->Primary_Block_Device.empty())
284 printf(" Primary_Block_Device: %s\n", Part->Primary_Block_Device.c_str());
285 if (!Part->Alternate_Block_Device.empty())
286 printf(" Alternate_Block_Device: %s\n", Part->Alternate_Block_Device.c_str());
287 if (!Part->Decrypted_Block_Device.empty())
288 printf(" Decrypted_Block_Device: %s\n", Part->Decrypted_Block_Device.c_str());
Ethan Yonker253368a2014-11-25 15:00:52 -0600289 if (!Part->Crypto_Key_Location.empty() && Part->Crypto_Key_Location != "footer")
290 printf(" Crypto_Key_Location: %s\n", Part->Crypto_Key_Location.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800291 if (Part->Length != 0)
292 printf(" Length: %i\n", Part->Length);
293 if (!Part->Display_Name.empty())
294 printf(" Display_Name: %s\n", Part->Display_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500295 if (!Part->Storage_Name.empty())
296 printf(" Storage_Name: %s\n", Part->Storage_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800297 if (!Part->Backup_Path.empty())
298 printf(" Backup_Path: %s\n", Part->Backup_Path.c_str());
299 if (!Part->Backup_Name.empty())
300 printf(" Backup_Name: %s\n", Part->Backup_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500301 if (!Part->Backup_Display_Name.empty())
302 printf(" Backup_Display_Name: %s\n", Part->Backup_Display_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800303 if (!Part->Backup_FileName.empty())
304 printf(" Backup_FileName: %s\n", Part->Backup_FileName.c_str());
305 if (!Part->Storage_Path.empty())
306 printf(" Storage_Path: %s\n", Part->Storage_Path.c_str());
307 if (!Part->Current_File_System.empty())
308 printf(" Current_File_System: %s\n", Part->Current_File_System.c_str());
309 if (!Part->Fstab_File_System.empty())
310 printf(" Fstab_File_System: %s\n", Part->Fstab_File_System.c_str());
311 if (Part->Format_Block_Size != 0)
312 printf(" Format_Block_Size: %i\n", Part->Format_Block_Size);
Dees_Troy094207a2012-09-26 12:00:39 -0400313 if (!Part->MTD_Name.empty())
314 printf(" MTD_Name: %s\n", Part->MTD_Name.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -0400315 string back_meth = Part->Backup_Method_By_Name();
Ethan Yonker6277c792014-09-15 14:54:30 -0500316 printf(" Backup_Method: %s\n", back_meth.c_str());
Hashcode62bd9e02013-11-19 21:59:42 -0800317 if (Part->Mount_Flags || !Part->Mount_Options.empty())
318 printf(" Mount_Flags=0x%8x, Mount_Options=%s\n", Part->Mount_Flags, Part->Mount_Options.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -0600319 if (Part->MTP_Storage_ID)
320 printf(" MTP_Storage_ID: %i\n", Part->MTP_Storage_ID);
Ethan Yonker6277c792014-09-15 14:54:30 -0500321 printf("\n");
Dees_Troy8170a922012-09-18 15:40:25 -0400322}
323
Dees_Troy51a0e822012-09-05 15:24:24 -0400324int TWPartitionManager::Mount_By_Path(string Path, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400325 std::vector<TWPartition*>::iterator iter;
326 int ret = false;
327 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400328 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400329
Dees_Troyd93bda52013-07-03 19:55:19 +0000330 if (Local_Path == "/tmp" || Local_Path == "/")
Dees_Troy43d8b002012-09-17 16:00:01 -0400331 return true;
332
Dees_Troy5bf43922012-09-07 16:07:55 -0400333 // Iterate through all partitions
334 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400335 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400336 ret = (*iter)->Mount(Display_Error);
337 found = true;
Dees_Troy51127312012-09-08 13:08:49 -0400338 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400339 (*iter)->Mount(Display_Error);
Dees_Troy51127312012-09-08 13:08:49 -0400340 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400341 }
342 if (found) {
343 return ret;
344 } else if (Display_Error) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000345 LOGERR("Mount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400346 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000347 LOGINFO("Mount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400348 }
349 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400350}
351
Dees_Troy51a0e822012-09-05 15:24:24 -0400352int TWPartitionManager::UnMount_By_Path(string Path, bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400353 std::vector<TWPartition*>::iterator iter;
354 int ret = false;
355 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400356 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy51127312012-09-08 13:08:49 -0400357
358 // Iterate through all partitions
359 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400360 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy51127312012-09-08 13:08:49 -0400361 ret = (*iter)->UnMount(Display_Error);
362 found = true;
363 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
364 (*iter)->UnMount(Display_Error);
365 }
366 }
367 if (found) {
368 return ret;
369 } else if (Display_Error) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000370 LOGERR("UnMount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400371 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000372 LOGINFO("UnMount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400373 }
374 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400375}
376
Dees_Troy51a0e822012-09-05 15:24:24 -0400377int TWPartitionManager::Is_Mounted_By_Path(string Path) {
Dees_Troy51127312012-09-08 13:08:49 -0400378 TWPartition* Part = Find_Partition_By_Path(Path);
379
380 if (Part)
381 return Part->Is_Mounted();
382 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000383 LOGINFO("Is_Mounted: Unable to find partition for path '%s'\n", Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400384 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400385}
386
Dees_Troy5bf43922012-09-07 16:07:55 -0400387int TWPartitionManager::Mount_Current_Storage(bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400388 string current_storage_path = DataManager::GetCurrentStoragePath();
389
390 if (Mount_By_Path(current_storage_path, Display_Error)) {
391 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
392 if (FreeStorage)
393 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
394 return true;
395 }
396 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400397}
398
Dees_Troy5bf43922012-09-07 16:07:55 -0400399int TWPartitionManager::Mount_Settings_Storage(bool Display_Error) {
400 return Mount_By_Path(DataManager::GetSettingsStoragePath(), Display_Error);
401}
402
403TWPartition* TWPartitionManager::Find_Partition_By_Path(string Path) {
404 std::vector<TWPartition*>::iterator iter;
Dees_Troy38bd7602012-09-14 13:33:53 -0400405 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400406
407 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400408 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path))
Dees_Troy5bf43922012-09-07 16:07:55 -0400409 return (*iter);
410 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400411 return NULL;
412}
413
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400414int TWPartitionManager::Check_Backup_Name(bool Display_Error) {
415 // Check the backup name to ensure that it is the correct size and contains only valid characters
416 // and that a backup with that name doesn't already exist
417 char backup_name[MAX_BACKUP_NAME_LEN];
418 char backup_loc[255], tw_image_dir[255];
419 int copy_size;
420 int index, cur_char;
421 string Backup_Name, Backup_Loc;
422
423 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
424 copy_size = Backup_Name.size();
425 // Check size
426 if (copy_size > MAX_BACKUP_NAME_LEN) {
427 if (Display_Error)
Dees_Troy2673cec2013-04-02 20:22:16 +0000428 LOGERR("Backup name is too long.\n");
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400429 return -2;
430 }
431
432 // Check each character
433 strncpy(backup_name, Backup_Name.c_str(), copy_size);
Dees_Troya13d74f2013-03-24 08:54:55 -0500434 if (copy_size == 1 && strncmp(backup_name, "0", 1) == 0)
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400435 return 0; // A "0" (zero) means to use the current timestamp for the backup name
436 for (index=0; index<copy_size; index++) {
437 cur_char = (int)backup_name[index];
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500438 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 -0400439 // These are valid characters
440 // Numbers
441 // Upper case letters
442 // Lower case letters
443 // Space
444 // and -_.{}[]
445 } else {
446 if (Display_Error)
Dees_Troy2673cec2013-04-02 20:22:16 +0000447 LOGERR("Backup name '%s' contains invalid character: '%c'\n", backup_name, (char)cur_char);
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400448 return -3;
449 }
450 }
451
452 // Check to make sure that a backup with this name doesn't already exist
453 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Loc);
454 strcpy(backup_loc, Backup_Loc.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500455 sprintf(tw_image_dir,"%s/%s", backup_loc, Backup_Name.c_str());
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500456 if (TWFunc::Path_Exists(tw_image_dir)) {
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400457 if (Display_Error)
Dees_Troy2673cec2013-04-02 20:22:16 +0000458 LOGERR("A backup with this name already exists.\n");
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400459 return -4;
460 }
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400461 // No problems found, return 0
462 return 0;
463}
464
Dees_Troy43d8b002012-09-17 16:00:01 -0400465bool TWPartitionManager::Make_MD5(bool generate_md5, string Backup_Folder, string Backup_Filename)
466{
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500467 string command;
Dees_Troy43d8b002012-09-17 16:00:01 -0400468 string Full_File = Backup_Folder + Backup_Filename;
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500469 string result;
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500470 twrpDigest md5sum;
Dees_Troy43d8b002012-09-17 16:00:01 -0400471
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500472 if (!generate_md5)
Dees_Troy43d8b002012-09-17 16:00:01 -0400473 return true;
Dees_Troy43d8b002012-09-17 16:00:01 -0400474
Dees_Troyb46a6842012-09-25 11:06:46 -0400475 TWFunc::GUI_Operation_Text(TW_GENERATE_MD5_TEXT, "Generating MD5");
Dees_Troy2673cec2013-04-02 20:22:16 +0000476 gui_print(" * Generating md5...\n");
Dees_Troy43d8b002012-09-17 16:00:01 -0400477
478 if (TWFunc::Path_Exists(Full_File)) {
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500479 md5sum.setfn(Backup_Folder + Backup_Filename);
480 if (md5sum.computeMD5() == 0)
481 if (md5sum.write_md5digest() == 0)
Dees_Troy2673cec2013-04-02 20:22:16 +0000482 gui_print(" * MD5 Created.\n");
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500483 else
484 return -1;
485 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000486 gui_print(" * MD5 Error!\n");
Dees_Troy43d8b002012-09-17 16:00:01 -0400487 } else {
488 char filename[512];
489 int index = 0;
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500490 string strfn;
Dees_Troy43d8b002012-09-17 16:00:01 -0400491 sprintf(filename, "%s%03i", Full_File.c_str(), index);
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500492 strfn = filename;
Dees_Troy83bd4832013-05-04 12:39:56 +0000493 while (index < 1000) {
bigbiff bigbiff65a4c732013-03-15 15:17:50 -0400494 md5sum.setfn(filename);
Dees_Troy83bd4832013-05-04 12:39:56 +0000495 if (TWFunc::Path_Exists(filename)) {
496 if (md5sum.computeMD5() == 0) {
497 if (md5sum.write_md5digest() != 0)
498 {
499 gui_print(" * MD5 Error.\n");
500 return false;
501 }
502 } else {
503 gui_print(" * Error computing MD5.\n");
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500504 return false;
505 }
506 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400507 index++;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400508 sprintf(filename, "%s%03i", Full_File.c_str(), index);
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500509 strfn = filename;
Dees_Troy43d8b002012-09-17 16:00:01 -0400510 }
511 if (index == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000512 LOGERR("Backup file: '%s' not found!\n", filename);
Dees_Troy43d8b002012-09-17 16:00:01 -0400513 return false;
514 }
Dees_Troy2673cec2013-04-02 20:22:16 +0000515 gui_print(" * MD5 Created.\n");
Dees_Troy43d8b002012-09-17 16:00:01 -0400516 }
517 return true;
518}
519
Dees_Troy093b7642012-09-21 15:59:38 -0400520bool TWPartitionManager::Backup_Partition(TWPartition* Part, string Backup_Folder, bool generate_md5, unsigned long long* img_bytes_remaining, unsigned long long* file_bytes_remaining, unsigned long *img_time, unsigned long *file_time, unsigned long long *img_bytes, unsigned long long *file_bytes) {
Dees_Troy43d8b002012-09-17 16:00:01 -0400521 time_t start, stop;
that203bcc92015-05-09 17:27:33 +0200522 int use_compression;
Dees_Troy093b7642012-09-21 15:59:38 -0400523 float pos;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500524 unsigned long long total_size, current_size;
Dees_Troy43d8b002012-09-17 16:00:01 -0400525
bigbiffbf1d6722015-02-14 16:25:59 -0500526 string backup_log = Backup_Folder + "recovery.log";
527
Dees_Troy43d8b002012-09-17 16:00:01 -0400528 if (Part == NULL)
529 return true;
530
Dees_Troy093b7642012-09-21 15:59:38 -0400531 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400532
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500533 total_size = *file_bytes + *img_bytes;
534 current_size = *file_bytes + *img_bytes - *file_bytes_remaining - *img_bytes_remaining;
that203bcc92015-05-09 17:27:33 +0200535 // Set the position
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500536 pos = ((float)(current_size) / (float)(total_size));
Dees_Troy2673cec2013-04-02 20:22:16 +0000537 DataManager::SetProgress(pos);
Dees_Troy093b7642012-09-21 15:59:38 -0400538
Tom Hite5a926722014-09-15 01:31:03 +0000539 TWFunc::SetPerformanceMode(true);
Dees_Troy43d8b002012-09-17 16:00:01 -0400540 time(&start);
541
bigbiff7abc5fe2015-01-17 16:53:12 -0500542 if (Part->Backup(Backup_Folder, &total_size, &current_size, tar_fork_pid)) {
Tom Hite5a926722014-09-15 01:31:03 +0000543 bool md5Success = false;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500544 current_size += Part->Backup_Size;
545 pos = (float)((float)(current_size) / (float)(total_size));
546 DataManager::SetProgress(pos);
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) {
bigbiff7abc5fe2015-01-17 16:53:12 -0500552 if (!(*subpart)->Backup(Backup_Folder, &total_size, &current_size, tar_fork_pid)) {
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_Troy093b7642012-09-21 15:59:38 -0400565 if (Part->Backup_Method == 1) {
566 *file_bytes_remaining -= (*subpart)->Backup_Size;
567 } else {
568 *img_bytes_remaining -= (*subpart)->Backup_Size;
569 }
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500570 current_size += Part->Backup_Size;
571 pos = (float)(current_size / total_size);
572 DataManager::SetProgress(pos);
Dees_Troy8170a922012-09-18 15:40:25 -0400573 }
574 }
575 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400576 time(&stop);
that203bcc92015-05-09 17:27:33 +0200577 int backup_time = (int) difftime(stop, start);
Dees_Troy2673cec2013-04-02 20:22:16 +0000578 LOGINFO("Partition Backup time: %d\n", backup_time);
Dees_Troy43d8b002012-09-17 16:00:01 -0400579 if (Part->Backup_Method == 1) {
580 *file_bytes_remaining -= Part->Backup_Size;
Dees_Troy093b7642012-09-21 15:59:38 -0400581 *file_time += backup_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400582 } else {
583 *img_bytes_remaining -= Part->Backup_Size;
Dees_Troy093b7642012-09-21 15:59:38 -0400584 *img_time += backup_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400585 }
Tom Hite5a926722014-09-15 01:31:03 +0000586
587 md5Success = Make_MD5(generate_md5, Backup_Folder, Part->Backup_FileName);
588 TWFunc::SetPerformanceMode(false);
589 return md5Success;
Dees_Troy43d8b002012-09-17 16:00:01 -0400590 } else {
bigbiffbf1d6722015-02-14 16:25:59 -0500591 Clean_Backup_Folder(Backup_Folder);
592 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
593 tw_set_default_metadata(backup_log.c_str());
Tom Hite5a926722014-09-15 01:31:03 +0000594 TWFunc::SetPerformanceMode(false);
Dees_Troy43d8b002012-09-17 16:00:01 -0400595 return false;
596 }
bigbiff7abc5fe2015-01-17 16:53:12 -0500597 return 0;
598}
599
bigbiffbf1d6722015-02-14 16:25:59 -0500600void TWPartitionManager::Clean_Backup_Folder(string Backup_Folder) {
601 DIR *d = opendir(Backup_Folder.c_str());
602 struct dirent *p;
603 int r;
604
605 gui_print("Backup Failed.\nCleaning Backup Folder\n");
606
607 if (d == NULL) {
608 LOGERR("Error opening dir: '%s'\n", Backup_Folder.c_str());
609 return;
610 }
611
Matt Mower2b18a532015-02-20 16:58:05 -0600612 while ((p = readdir(d))) {
bigbiffbf1d6722015-02-14 16:25:59 -0500613 if (!strcmp(p->d_name, ".") || !strcmp(p->d_name, ".."))
614 continue;
615
616 string path = Backup_Folder + p->d_name;
617
618 size_t dot = path.find_last_of(".") + 1;
619 if (path.substr(dot) == "win" || path.substr(dot) == "md5" || path.substr(dot) == "info") {
620 r = unlink(path.c_str());
621 if (r != 0) {
622 LOGINFO("Unable to unlink '%s: %s'\n", path.c_str(), strerror(errno));
623 }
624 }
625 }
626 closedir(d);
627}
628
bigbiff7abc5fe2015-01-17 16:53:12 -0500629int TWPartitionManager::Cancel_Backup() {
630 string Backup_Folder, Backup_Name, Full_Backup_Path;
631
632 stop_backup.set_value(1);
633
634 if (tar_fork_pid != 0) {
635 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
636 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Folder);
637 Full_Backup_Path = Backup_Folder + "/" + Backup_Name + "/";
638 LOGINFO("Killing pid: %d\n", tar_fork_pid);
639 kill(tar_fork_pid, SIGUSR2);
640 while (kill(tar_fork_pid, 0) == 0) {
641 usleep(1000);
642 }
643 LOGINFO("Backup_Run stopped and returning false, backup cancelled.\n");
644 LOGINFO("Removing directory %s\n", Full_Backup_Path.c_str());
645 TWFunc::removeDir(Full_Backup_Path, false);
646 tar_fork_pid = 0;
647 }
648
649 return 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400650}
651
652int TWPartitionManager::Run_Backup(void) {
Matt Mower2b18a532015-02-20 16:58:05 -0600653 int check, do_md5, partition_count = 0, disable_free_space_check = 0;
Dees_Troya13d74f2013-03-24 08:54:55 -0500654 string Backup_Folder, Backup_Name, Full_Backup_Path, Backup_List, backup_path;
Dees_Troy8170a922012-09-18 15:40:25 -0400655 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 -0400656 unsigned long img_time = 0, file_time = 0;
Dees_Troya13d74f2013-03-24 08:54:55 -0500657 TWPartition* backup_part = NULL;
Dees_Troy43d8b002012-09-17 16:00:01 -0400658 TWPartition* storage = NULL;
Dees_Troy8170a922012-09-18 15:40:25 -0400659 std::vector<TWPartition*>::iterator subpart;
Dees_Troy43d8b002012-09-17 16:00:01 -0400660 struct tm *t;
661 time_t start, stop, seconds, total_start, total_stop;
Dees_Troya13d74f2013-03-24 08:54:55 -0500662 size_t start_pos = 0, end_pos = 0;
bigbiff7abc5fe2015-01-17 16:53:12 -0500663 stop_backup.set_value(0);
Dees_Troy43d8b002012-09-17 16:00:01 -0400664 seconds = time(0);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500665 t = localtime(&seconds);
Dees_Troy43d8b002012-09-17 16:00:01 -0400666
667 time(&total_start);
668
669 Update_System_Details();
670
671 if (!Mount_Current_Storage(true))
672 return false;
673
674 DataManager::GetValue(TW_SKIP_MD5_GENERATE_VAR, do_md5);
Dees_Troy4a2a1262012-09-18 09:33:47 -0400675 if (do_md5 == 0)
Dees_Troy43d8b002012-09-17 16:00:01 -0400676 do_md5 = true;
Dees_Troyc5865ab2012-09-24 15:08:04 -0400677 else
678 do_md5 = false;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400679
Dees_Troy43d8b002012-09-17 16:00:01 -0400680 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Folder);
681 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
Dees Troyb21cc642013-09-10 17:36:41 +0000682 if (Backup_Name == "(Current Date)") {
683 Backup_Name = TWFunc::Get_Current_Date();
684 } else if (Backup_Name == "(Auto Generate)" || Backup_Name == "0" || Backup_Name.empty()) {
685 TWFunc::Auto_Generate_Backup_Name();
686 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
Dees_Troy43d8b002012-09-17 16:00:01 -0400687 }
Dees_Troy2673cec2013-04-02 20:22:16 +0000688 LOGINFO("Backup Name is: '%s'\n", Backup_Name.c_str());
Dees_Troy43d8b002012-09-17 16:00:01 -0400689 Full_Backup_Path = Backup_Folder + "/" + Backup_Name + "/";
Dees_Troy2673cec2013-04-02 20:22:16 +0000690 LOGINFO("Full_Backup_Path is: '%s'\n", Full_Backup_Path.c_str());
Dees_Troy43d8b002012-09-17 16:00:01 -0400691
Dees_Troy2673cec2013-04-02 20:22:16 +0000692 LOGINFO("Calculating backup details...\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500693 DataManager::GetValue("tw_backup_list", Backup_List);
694 if (!Backup_List.empty()) {
695 end_pos = Backup_List.find(";", start_pos);
696 while (end_pos != string::npos && start_pos < Backup_List.size()) {
697 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
698 backup_part = Find_Partition_By_Path(backup_path);
699 if (backup_part != NULL) {
700 partition_count++;
701 if (backup_part->Backup_Method == 1)
702 file_bytes += backup_part->Backup_Size;
703 else
704 img_bytes += backup_part->Backup_Size;
705 if (backup_part->Has_SubPartition) {
706 std::vector<TWPartition*>::iterator subpart;
707
708 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
Dees_Troy9e0b71c2013-04-08 13:35:37 +0000709 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 -0500710 partition_count++;
711 if ((*subpart)->Backup_Method == 1)
712 file_bytes += (*subpart)->Backup_Size;
713 else
714 img_bytes += (*subpart)->Backup_Size;
715 }
716 }
Dees_Troy8170a922012-09-18 15:40:25 -0400717 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500718 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000719 LOGERR("Unable to locate '%s' partition for backup calculations.\n", backup_path.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -0400720 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500721 start_pos = end_pos + 1;
722 end_pos = Backup_List.find(";", start_pos);
Dees_Troy43d8b002012-09-17 16:00:01 -0400723 }
724 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400725
726 if (partition_count == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000727 gui_print("No partitions selected for backup.\n");
Dees_Troy43d8b002012-09-17 16:00:01 -0400728 return false;
729 }
730 total_bytes = file_bytes + img_bytes;
Dees_Troy2673cec2013-04-02 20:22:16 +0000731 gui_print(" * Total number of partitions to back up: %d\n", partition_count);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500732 gui_print(" * Total size of all data: %lluMB\n", total_bytes / 1024 / 1024);
Dees_Troy43d8b002012-09-17 16:00:01 -0400733 storage = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
734 if (storage != NULL) {
735 free_space = storage->Free;
Dees_Troy2673cec2013-04-02 20:22:16 +0000736 gui_print(" * Available space: %lluMB\n", free_space / 1024 / 1024);
Dees_Troy43d8b002012-09-17 16:00:01 -0400737 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000738 LOGERR("Unable to locate storage device.\n");
Dees_Troy43d8b002012-09-17 16:00:01 -0400739 return false;
740 }
bigbiffbf1d6722015-02-14 16:25:59 -0500741
742 DataManager::GetValue("tw_disable_free_space", disable_free_space_check);
743 if (!disable_free_space_check) {
744 if (free_space - (32 * 1024 * 1024) < total_bytes) {
745 // We require an extra 32MB just in case
746 LOGERR("Not enough free space on storage.\n");
747 return false;
748 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400749 }
750 img_bytes_remaining = img_bytes;
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500751 file_bytes_remaining = file_bytes;
Dees_Troy43d8b002012-09-17 16:00:01 -0400752
Dees_Troy2673cec2013-04-02 20:22:16 +0000753 gui_print("\n[BACKUP STARTED]\n");
754 gui_print(" * Backup Folder: %s\n", Full_Backup_Path.c_str());
Dees_Troyd4b22b02013-01-18 17:17:58 +0000755 if (!TWFunc::Recursive_Mkdir(Full_Backup_Path)) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000756 LOGERR("Failed to make backup folder.\n");
Dees_Troyd4b22b02013-01-18 17:17:58 +0000757 return false;
758 }
759
Dees_Troy2673cec2013-04-02 20:22:16 +0000760 DataManager::SetProgress(0.0);
Dees_Troy093b7642012-09-21 15:59:38 -0400761
Dees_Troya13d74f2013-03-24 08:54:55 -0500762 start_pos = 0;
763 end_pos = Backup_List.find(";", start_pos);
764 while (end_pos != string::npos && start_pos < Backup_List.size()) {
bigbiff7abc5fe2015-01-17 16:53:12 -0500765 if (stop_backup.get_value() != 0)
766 return -1;
Dees_Troya13d74f2013-03-24 08:54:55 -0500767 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
768 backup_part = Find_Partition_By_Path(backup_path);
769 if (backup_part != NULL) {
770 if (!Backup_Partition(backup_part, Full_Backup_Path, do_md5, &img_bytes_remaining, &file_bytes_remaining, &img_time, &file_time, &img_bytes, &file_bytes))
771 return false;
772 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000773 LOGERR("Unable to locate '%s' partition for backup process.\n", backup_path.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500774 }
775 start_pos = end_pos + 1;
776 end_pos = Backup_List.find(";", start_pos);
777 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400778
779 // Average BPS
780 if (img_time == 0)
781 img_time = 1;
782 if (file_time == 0)
783 file_time = 1;
Dees_Troy093b7642012-09-21 15:59:38 -0400784 int img_bps = (int)img_bytes / (int)img_time;
bigbiff bigbiff2c57d782013-02-19 10:09:21 -0500785 unsigned long long file_bps = file_bytes / (int)file_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400786
Dees_Troy2673cec2013-04-02 20:22:16 +0000787 gui_print("Average backup rate for file systems: %llu MB/sec\n", (file_bps / (1024 * 1024)));
788 gui_print("Average backup rate for imaged drives: %lu MB/sec\n", (img_bps / (1024 * 1024)));
Dees_Troy43d8b002012-09-17 16:00:01 -0400789
790 time(&total_stop);
791 int total_time = (int) difftime(total_stop, total_start);
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500792 uint64_t actual_backup_size = du.Get_Folder_Size(Full_Backup_Path);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500793 actual_backup_size /= (1024LLU * 1024LLU);
Dees_Troy43d8b002012-09-17 16:00:01 -0400794
bigbiff bigbiff2c57d782013-02-19 10:09:21 -0500795 int prev_img_bps, use_compression;
796 unsigned long long prev_file_bps;
Dees_Troy093b7642012-09-21 15:59:38 -0400797 DataManager::GetValue(TW_BACKUP_AVG_IMG_RATE, prev_img_bps);
798 img_bps += (prev_img_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500799 img_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -0400800
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500801 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400802 if (use_compression)
803 DataManager::GetValue(TW_BACKUP_AVG_FILE_COMP_RATE, prev_file_bps);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500804 else
Dees_Troy093b7642012-09-21 15:59:38 -0400805 DataManager::GetValue(TW_BACKUP_AVG_FILE_RATE, prev_file_bps);
806 file_bps += (prev_file_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500807 file_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -0400808
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500809 DataManager::SetValue(TW_BACKUP_AVG_IMG_RATE, img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -0400810 if (use_compression)
811 DataManager::SetValue(TW_BACKUP_AVG_FILE_COMP_RATE, file_bps);
812 else
813 DataManager::SetValue(TW_BACKUP_AVG_FILE_RATE, file_bps);
814
Dees_Troy2673cec2013-04-02 20:22:16 +0000815 gui_print("[%llu MB TOTAL BACKED UP]\n", actual_backup_size);
Dees_Troy43d8b002012-09-17 16:00:01 -0400816 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -0400817 UnMount_Main_Partitions();
Ethan Yonkerbf2cb1c2014-07-02 10:15:54 -0500818 gui_print_color("highlight", "[BACKUP COMPLETED IN %d SECONDS]\n\n", total_time); // the end
Dees_Troy3f5c4e82013-02-01 15:16:59 +0000819 string backup_log = Full_Backup_Path + "recovery.log";
820 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
Ethan Yonker4b94cfd2014-12-11 10:00:45 -0600821 tw_set_default_metadata(backup_log.c_str());
Dees_Troy3f5c4e82013-02-01 15:16:59 +0000822 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -0400823}
824
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500825bool TWPartitionManager::Restore_Partition(TWPartition* Part, string Restore_Name, int partition_count, const unsigned long long *total_restore_size, unsigned long long *already_restored_size) {
Dees_Troy4a2a1262012-09-18 09:33:47 -0400826 time_t Start, Stop;
Tom Hite5a926722014-09-15 01:31:03 +0000827 TWFunc::SetPerformanceMode(true);
Dees_Troy4a2a1262012-09-18 09:33:47 -0400828 time(&Start);
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500829 //DataManager::ShowProgress(1.0 / (float)partition_count, 150);
Tom Hite5a926722014-09-15 01:31:03 +0000830 if (!Part->Restore(Restore_Name, total_restore_size, already_restored_size)) {
831 TWFunc::SetPerformanceMode(false);
Dees_Troy4a2a1262012-09-18 09:33:47 -0400832 return false;
Tom Hite5a926722014-09-15 01:31:03 +0000833 }
Dees_Troy8170a922012-09-18 15:40:25 -0400834 if (Part->Has_SubPartition) {
835 std::vector<TWPartition*>::iterator subpart;
836
837 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
838 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == Part->Mount_Point) {
Tom Hite5a926722014-09-15 01:31:03 +0000839 if (!(*subpart)->Restore(Restore_Name, total_restore_size, already_restored_size)) {
840 TWFunc::SetPerformanceMode(false);
Dees_Troy8170a922012-09-18 15:40:25 -0400841 return false;
Tom Hite5a926722014-09-15 01:31:03 +0000842 }
Dees_Troy8170a922012-09-18 15:40:25 -0400843 }
844 }
845 }
Dees_Troy4a2a1262012-09-18 09:33:47 -0400846 time(&Stop);
Tom Hite5a926722014-09-15 01:31:03 +0000847 TWFunc::SetPerformanceMode(false);
Dees_Troy2673cec2013-04-02 20:22:16 +0000848 gui_print("[%s done (%d seconds)]\n\n", Part->Backup_Display_Name.c_str(), (int)difftime(Stop, Start));
Dees_Troy4a2a1262012-09-18 09:33:47 -0400849 return true;
850}
851
Dees_Troy51a0e822012-09-05 15:24:24 -0400852int TWPartitionManager::Run_Restore(string Restore_Name) {
Dees_Troy4a2a1262012-09-18 09:33:47 -0400853 int check_md5, check, partition_count = 0;
Dees_Troya13d74f2013-03-24 08:54:55 -0500854 TWPartition* restore_part = NULL;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400855 time_t rStart, rStop;
856 time(&rStart);
Dees_Troya13d74f2013-03-24 08:54:55 -0500857 string Restore_List, restore_path;
858 size_t start_pos = 0, end_pos;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500859 unsigned long long total_restore_size = 0, already_restored_size = 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400860
Dees_Troy2673cec2013-04-02 20:22:16 +0000861 gui_print("\n[RESTORE STARTED]\n\n");
862 gui_print("Restore folder: '%s'\n", Restore_Name.c_str());
Dees_Troy43d8b002012-09-17 16:00:01 -0400863
Dees_Troy4a2a1262012-09-18 09:33:47 -0400864 if (!Mount_Current_Storage(true))
865 return false;
866
867 DataManager::GetValue(TW_SKIP_MD5_CHECK_VAR, check_md5);
Dees_Troya13d74f2013-03-24 08:54:55 -0500868 if (check_md5 > 0) {
869 // Check MD5 files first before restoring to ensure that all of them match before starting a restore
870 TWFunc::GUI_Operation_Text(TW_VERIFY_MD5_TEXT, "Verifying MD5");
Dees_Troy2673cec2013-04-02 20:22:16 +0000871 gui_print("Verifying MD5...\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500872 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000873 gui_print("Skipping MD5 check based on user setting.\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500874 }
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500875 gui_print("Calculating restore details...\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500876 DataManager::GetValue("tw_restore_selected", Restore_List);
877 if (!Restore_List.empty()) {
878 end_pos = Restore_List.find(";", start_pos);
879 while (end_pos != string::npos && start_pos < Restore_List.size()) {
880 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
881 restore_part = Find_Partition_By_Path(restore_path);
882 if (restore_part != NULL) {
Ethan Yonkereb32b1f2015-05-18 10:23:03 -0500883 if (restore_part->Mount_Read_Only) {
884 LOGERR("Cannot restore %s -- mounted read only.\n", restore_part->Backup_Display_Name.c_str());
885 return false;
886 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500887 if (check_md5 > 0 && !restore_part->Check_MD5(Restore_Name))
888 return false;
Ethan Yonkereb32b1f2015-05-18 10:23:03 -0500889 partition_count++;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500890 total_restore_size += restore_part->Get_Restore_Size(Restore_Name);
Dees_Troya13d74f2013-03-24 08:54:55 -0500891 if (restore_part->Has_SubPartition) {
892 std::vector<TWPartition*>::iterator subpart;
893
894 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
895 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == restore_part->Mount_Point) {
bigbiff bigbiff07338812014-03-30 14:56:41 -0400896 if (check_md5 > 0 && !(*subpart)->Check_MD5(Restore_Name))
Dees_Troya13d74f2013-03-24 08:54:55 -0500897 return false;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500898 total_restore_size += (*subpart)->Get_Restore_Size(Restore_Name);
Dees_Troya13d74f2013-03-24 08:54:55 -0500899 }
900 }
901 }
902 } else {
Dees_Troy59df9262013-06-19 14:53:57 -0500903 LOGERR("Unable to locate '%s' partition for restoring (restore list).\n", restore_path.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500904 }
905 start_pos = end_pos + 1;
906 end_pos = Restore_List.find(";", start_pos);
Dees_Troy4a2a1262012-09-18 09:33:47 -0400907 }
Dees_Troy4a2a1262012-09-18 09:33:47 -0400908 }
Dees_Troy4a2a1262012-09-18 09:33:47 -0400909
910 if (partition_count == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000911 LOGERR("No partitions selected for restore.\n");
Dees_Troy4a2a1262012-09-18 09:33:47 -0400912 return false;
913 }
914
Dees_Troy2673cec2013-04-02 20:22:16 +0000915 gui_print("Restoring %i partitions...\n", partition_count);
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500916 gui_print("Total restore size is %lluMB\n", total_restore_size / 1048576);
Dees_Troy2673cec2013-04-02 20:22:16 +0000917 DataManager::SetProgress(0.0);
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500918
Dees_Troya13d74f2013-03-24 08:54:55 -0500919 start_pos = 0;
920 if (!Restore_List.empty()) {
921 end_pos = Restore_List.find(";", start_pos);
922 while (end_pos != string::npos && start_pos < Restore_List.size()) {
923 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
924 restore_part = Find_Partition_By_Path(restore_path);
925 if (restore_part != NULL) {
926 partition_count++;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500927 if (!Restore_Partition(restore_part, Restore_Name, partition_count, &total_restore_size, &already_restored_size))
Dees_Troya13d74f2013-03-24 08:54:55 -0500928 return false;
929 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000930 LOGERR("Unable to locate '%s' partition for restoring.\n", restore_path.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500931 }
932 start_pos = end_pos + 1;
933 end_pos = Restore_List.find(";", start_pos);
934 }
935 }
Dees_Troyb46a6842012-09-25 11:06:46 -0400936 TWFunc::GUI_Operation_Text(TW_UPDATE_SYSTEM_DETAILS_TEXT, "Updating System Details");
Dees_Troy43d8b002012-09-17 16:00:01 -0400937 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -0400938 UnMount_Main_Partitions();
Dees_Troy4a2a1262012-09-18 09:33:47 -0400939 time(&rStop);
Ethan Yonkerbf2cb1c2014-07-02 10:15:54 -0500940 gui_print_color("highlight", "[RESTORE COMPLETED IN %d SECONDS]\n\n",(int)difftime(rStop,rStart));
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500941 DataManager::SetValue("tw_file_progress", "");
Dees_Troy63c8df72012-09-10 14:02:05 -0400942 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -0400943}
944
945void TWPartitionManager::Set_Restore_Files(string Restore_Name) {
Dees_Troy63c8df72012-09-10 14:02:05 -0400946 // Start with the default values
Dees_Troya13d74f2013-03-24 08:54:55 -0500947 string Restore_List;
Dees_Troy83bd4832013-05-04 12:39:56 +0000948 bool get_date = true, check_encryption = true;
949
950 DataManager::SetValue("tw_restore_encrypted", 0);
Dees_Troy63c8df72012-09-10 14:02:05 -0400951
952 DIR* d;
953 d = opendir(Restore_Name.c_str());
954 if (d == NULL)
955 {
Dees_Troy2673cec2013-04-02 20:22:16 +0000956 LOGERR("Error opening %s\n", Restore_Name.c_str());
Dees_Troy63c8df72012-09-10 14:02:05 -0400957 return;
958 }
959
960 struct dirent* de;
961 while ((de = readdir(d)) != NULL)
962 {
963 // Strip off three components
964 char str[256];
965 char* label;
966 char* fstype = NULL;
967 char* extn = NULL;
968 char* ptr;
969
970 strcpy(str, de->d_name);
971 if (strlen(str) <= 2)
972 continue;
973
974 if (get_date) {
975 char file_path[255];
976 struct stat st;
977
978 strcpy(file_path, Restore_Name.c_str());
979 strcat(file_path, "/");
980 strcat(file_path, str);
981 stat(file_path, &st);
982 string backup_date = ctime((const time_t*)(&st.st_mtime));
983 DataManager::SetValue(TW_RESTORE_FILE_DATE, backup_date);
984 get_date = false;
985 }
986
987 label = str;
988 ptr = label;
989 while (*ptr && *ptr != '.') ptr++;
990 if (*ptr == '.')
991 {
992 *ptr = 0x00;
993 ptr++;
994 fstype = ptr;
995 }
996 while (*ptr && *ptr != '.') ptr++;
997 if (*ptr == '.')
998 {
999 *ptr = 0x00;
1000 ptr++;
1001 extn = ptr;
1002 }
1003
Dees_Troy83bd4832013-05-04 12:39:56 +00001004 if (fstype == NULL || extn == NULL || strcmp(fstype, "log") == 0) continue;
Dees_Troya13d74f2013-03-24 08:54:55 -05001005 int extnlength = strlen(extn);
Dees_Troy83bd4832013-05-04 12:39:56 +00001006 if (extnlength != 3 && extnlength != 6) continue;
1007 if (extnlength >= 3 && strncmp(extn, "win", 3) != 0) continue;
1008 //if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
1009
1010 if (check_encryption) {
1011 string filename = Restore_Name + "/";
1012 filename += de->d_name;
1013 if (TWFunc::Get_File_Type(filename) == 2) {
1014 LOGINFO("'%s' is encrypted\n", filename.c_str());
1015 DataManager::SetValue("tw_restore_encrypted", 1);
1016 }
1017 }
Dees_Troya13d74f2013-03-24 08:54:55 -05001018 if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
Dees_Troy63c8df72012-09-10 14:02:05 -04001019
1020 TWPartition* Part = Find_Partition_By_Path(label);
1021 if (Part == NULL)
1022 {
Dees_Troy2673cec2013-04-02 20:22:16 +00001023 LOGERR(" Unable to locate partition by backup name: '%s'\n", label);
Dees_Troy63c8df72012-09-10 14:02:05 -04001024 continue;
1025 }
1026
1027 Part->Backup_FileName = de->d_name;
1028 if (strlen(extn) > 3) {
1029 Part->Backup_FileName.resize(Part->Backup_FileName.size() - strlen(extn) + 3);
1030 }
1031
Dees_Troya13d74f2013-03-24 08:54:55 -05001032 Restore_List += Part->Backup_Path + ";";
Dees_Troy63c8df72012-09-10 14:02:05 -04001033 }
1034 closedir(d);
1035
Dees_Troya13d74f2013-03-24 08:54:55 -05001036 // Set the final value
1037 DataManager::SetValue("tw_restore_list", Restore_List);
1038 DataManager::SetValue("tw_restore_selected", Restore_List);
Dees_Troy51a0e822012-09-05 15:24:24 -04001039 return;
1040}
1041
1042int TWPartitionManager::Wipe_By_Path(string Path) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001043 std::vector<TWPartition*>::iterator iter;
1044 int ret = false;
1045 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -04001046 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy63c8df72012-09-10 14:02:05 -04001047
1048 // Iterate through all partitions
1049 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -04001050 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troye58d5262012-09-21 12:27:57 -04001051 if (Path == "/and-sec")
1052 ret = (*iter)->Wipe_AndSec();
1053 else
1054 ret = (*iter)->Wipe();
Dees_Troy63c8df72012-09-10 14:02:05 -04001055 found = true;
1056 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1057 (*iter)->Wipe();
1058 }
1059 }
1060 if (found) {
1061 return ret;
1062 } else
Dees_Troy2673cec2013-04-02 20:22:16 +00001063 LOGERR("Wipe: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy63c8df72012-09-10 14:02:05 -04001064 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -04001065}
1066
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001067int TWPartitionManager::Wipe_By_Path(string Path, string New_File_System) {
1068 std::vector<TWPartition*>::iterator iter;
1069 int ret = false;
1070 bool found = false;
1071 string Local_Path = TWFunc::Get_Root_Path(Path);
1072
1073 // Iterate through all partitions
1074 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1075 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1076 if (Path == "/and-sec")
1077 ret = (*iter)->Wipe_AndSec();
1078 else
1079 ret = (*iter)->Wipe(New_File_System);
1080 found = true;
1081 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1082 (*iter)->Wipe(New_File_System);
1083 }
1084 }
1085 if (found) {
1086 return ret;
1087 } else
1088 LOGERR("Wipe: Unable to find partition for path '%s'\n", Local_Path.c_str());
1089 return false;
1090}
1091
Dees_Troy51a0e822012-09-05 15:24:24 -04001092int TWPartitionManager::Factory_Reset(void) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001093 std::vector<TWPartition*>::iterator iter;
1094 int ret = true;
1095
1096 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001097 if ((*iter)->Wipe_During_Factory_Reset && (*iter)->Is_Present) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001098 if (!(*iter)->Wipe())
1099 ret = false;
Dees_Troy094207a2012-09-26 12:00:39 -04001100 } else if ((*iter)->Has_Android_Secure) {
1101 if (!(*iter)->Wipe_AndSec())
1102 ret = false;
Dees_Troy63c8df72012-09-10 14:02:05 -04001103 }
1104 }
1105 return ret;
Dees_Troy51a0e822012-09-05 15:24:24 -04001106}
1107
Dees_Troy38bd7602012-09-14 13:33:53 -04001108int TWPartitionManager::Wipe_Dalvik_Cache(void) {
1109 struct stat st;
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001110 vector <string> dir;
Dees_Troy38bd7602012-09-14 13:33:53 -04001111
1112 if (!Mount_By_Path("/data", true))
1113 return false;
1114
1115 if (!Mount_By_Path("/cache", true))
1116 return false;
1117
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001118 dir.push_back("/data/dalvik-cache");
1119 dir.push_back("/cache/dalvik-cache");
1120 dir.push_back("/cache/dc");
Dees_Troy2673cec2013-04-02 20:22:16 +00001121 gui_print("\nWiping Dalvik Cache Directories...\n");
Dees_Troya13d74f2013-03-24 08:54:55 -05001122 for (unsigned i = 0; i < dir.size(); ++i) {
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001123 if (stat(dir.at(i).c_str(), &st) == 0) {
1124 TWFunc::removeDir(dir.at(i), false);
Dees_Troy2673cec2013-04-02 20:22:16 +00001125 gui_print("Cleaned: %s...\n", dir.at(i).c_str());
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001126 }
1127 }
Dees_Troy38bd7602012-09-14 13:33:53 -04001128 TWPartition* sdext = Find_Partition_By_Path("/sd-ext");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001129 if (sdext && sdext->Is_Present && sdext->Mount(false))
1130 {
1131 if (stat("/sd-ext/dalvik-cache", &st) == 0)
1132 {
1133 TWFunc::removeDir("/sd-ext/dalvik-cache", false);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001134 gui_print("Cleaned: /sd-ext/dalvik-cache...\n");
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001135 }
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001136 }
Dees_Troy2673cec2013-04-02 20:22:16 +00001137 gui_print("-- Dalvik Cache Directories Wipe Complete!\n\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001138 return true;
1139}
1140
1141int TWPartitionManager::Wipe_Rotate_Data(void) {
1142 if (!Mount_By_Path("/data", true))
1143 return false;
1144
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001145 unlink("/data/misc/akmd*");
1146 unlink("/data/misc/rild*");
Dees_Troy2673cec2013-04-02 20:22:16 +00001147 gui_print("Rotation data wiped.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001148 return true;
1149}
1150
1151int TWPartitionManager::Wipe_Battery_Stats(void) {
1152 struct stat st;
1153
1154 if (!Mount_By_Path("/data", true))
1155 return false;
1156
1157 if (0 != stat("/data/system/batterystats.bin", &st)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001158 gui_print("No Battery Stats Found. No Need To Wipe.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001159 } else {
1160 remove("/data/system/batterystats.bin");
Dees_Troy2673cec2013-04-02 20:22:16 +00001161 gui_print("Cleared battery stats.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001162 }
1163 return true;
1164}
1165
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001166int TWPartitionManager::Wipe_Android_Secure(void) {
1167 std::vector<TWPartition*>::iterator iter;
1168 int ret = false;
1169 bool found = false;
1170
1171 // Iterate through all partitions
1172 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1173 if ((*iter)->Has_Android_Secure) {
1174 ret = (*iter)->Wipe_AndSec();
1175 found = true;
1176 }
1177 }
1178 if (found) {
1179 return ret;
1180 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001181 LOGERR("No android secure partitions found.\n");
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001182 }
1183 return false;
1184}
1185
Dees_Troy38bd7602012-09-14 13:33:53 -04001186int TWPartitionManager::Format_Data(void) {
1187 TWPartition* dat = Find_Partition_By_Path("/data");
1188
1189 if (dat != NULL) {
1190 if (!dat->UnMount(true))
1191 return false;
1192
1193 return dat->Wipe_Encryption();
1194 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001195 LOGERR("Unable to locate /data.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001196 return false;
1197 }
1198 return false;
1199}
1200
1201int TWPartitionManager::Wipe_Media_From_Data(void) {
1202 TWPartition* dat = Find_Partition_By_Path("/data");
1203
1204 if (dat != NULL) {
1205 if (!dat->Has_Data_Media) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001206 LOGERR("This device does not have /data/media\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001207 return false;
1208 }
1209 if (!dat->Mount(true))
1210 return false;
1211
Dees_Troy2673cec2013-04-02 20:22:16 +00001212 gui_print("Wiping internal storage -- /data/media...\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06001213 Remove_MTP_Storage(dat->MTP_Storage_ID);
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001214 TWFunc::removeDir("/data/media", false);
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001215 if (mkdir("/data/media", S_IRWXU | S_IRWXG | S_IWGRP | S_IXGRP) != 0) {
Ethan Yonker726a0202014-12-16 20:01:38 -06001216 Add_MTP_Storage(dat->MTP_Storage_ID);
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001217 return false;
1218 }
Dees_Troy38bd7602012-09-14 13:33:53 -04001219 if (dat->Has_Data_Media) {
1220 dat->Recreate_Media_Folder();
Dees_Troy74fb2e92013-04-15 14:35:47 +00001221 // Unmount and remount - slightly hackish way to ensure that the "/sdcard" folder is still mounted properly after wiping
1222 dat->UnMount(false);
1223 dat->Mount(false);
Dees_Troy38bd7602012-09-14 13:33:53 -04001224 }
Ethan Yonker726a0202014-12-16 20:01:38 -06001225 Add_MTP_Storage(dat->MTP_Storage_ID);
Dees_Troy38bd7602012-09-14 13:33:53 -04001226 return true;
1227 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001228 LOGERR("Unable to locate /data.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001229 return false;
1230 }
1231 return false;
1232}
1233
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001234int TWPartitionManager::Repair_By_Path(string Path, bool Display_Error) {
1235 std::vector<TWPartition*>::iterator iter;
1236 int ret = false;
1237 bool found = false;
1238 string Local_Path = TWFunc::Get_Root_Path(Path);
1239
1240 if (Local_Path == "/tmp" || Local_Path == "/")
1241 return true;
1242
1243 // Iterate through all partitions
1244 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1245 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1246 ret = (*iter)->Repair();
1247 found = true;
1248 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1249 (*iter)->Repair();
1250 }
1251 }
1252 if (found) {
1253 return ret;
1254 } else if (Display_Error) {
1255 LOGERR("Repair: Unable to find partition for path '%s'\n", Local_Path.c_str());
1256 } else {
1257 LOGINFO("Repair: Unable to find partition for path '%s'\n", Local_Path.c_str());
1258 }
1259 return false;
1260}
1261
Ethan Yonkera2719152015-05-28 09:44:41 -05001262int TWPartitionManager::Resize_By_Path(string Path, bool Display_Error) {
1263 std::vector<TWPartition*>::iterator iter;
1264 int ret = false;
1265 bool found = false;
1266 string Local_Path = TWFunc::Get_Root_Path(Path);
1267
1268 if (Local_Path == "/tmp" || Local_Path == "/")
1269 return true;
1270
1271 // Iterate through all partitions
1272 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1273 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1274 ret = (*iter)->Resize();
1275 found = true;
1276 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1277 (*iter)->Resize();
1278 }
1279 }
1280 if (found) {
1281 return ret;
1282 } else if (Display_Error) {
1283 LOGERR("Resize: Unable to find partition for path '%s'\n", Local_Path.c_str());
1284 } else {
1285 LOGINFO("Resize: Unable to find partition for path '%s'\n", Local_Path.c_str());
1286 }
1287 return false;
1288}
1289
Dees_Troy51a0e822012-09-05 15:24:24 -04001290void TWPartitionManager::Update_System_Details(void) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001291 std::vector<TWPartition*>::iterator iter;
Dees_Troy51127312012-09-08 13:08:49 -04001292 int data_size = 0;
Dees_Troy5bf43922012-09-07 16:07:55 -04001293
Dees_Troy2673cec2013-04-02 20:22:16 +00001294 gui_print("Updating partition details...\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04001295 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy51127312012-09-08 13:08:49 -04001296 if ((*iter)->Can_Be_Mounted) {
1297 (*iter)->Update_Size(true);
1298 if ((*iter)->Mount_Point == "/system") {
1299 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1300 DataManager::SetValue(TW_BACKUP_SYSTEM_SIZE, backup_display_size);
1301 } else if ((*iter)->Mount_Point == "/data" || (*iter)->Mount_Point == "/datadata") {
1302 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
1303 } else if ((*iter)->Mount_Point == "/cache") {
1304 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1305 DataManager::SetValue(TW_BACKUP_CACHE_SIZE, backup_display_size);
1306 } else if ((*iter)->Mount_Point == "/sd-ext") {
1307 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1308 DataManager::SetValue(TW_BACKUP_SDEXT_SIZE, backup_display_size);
1309 if ((*iter)->Backup_Size == 0) {
1310 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 0);
1311 DataManager::SetValue(TW_BACKUP_SDEXT_VAR, 0);
1312 } else
1313 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 1);
Dees_Troye58d5262012-09-21 12:27:57 -04001314 } else if ((*iter)->Has_Android_Secure) {
Dees_Troy8170a922012-09-18 15:40:25 -04001315 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troye58d5262012-09-21 12:27:57 -04001316 DataManager::SetValue(TW_BACKUP_ANDSEC_SIZE, backup_display_size);
Dees_Troy8170a922012-09-18 15:40:25 -04001317 if ((*iter)->Backup_Size == 0) {
1318 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 0);
1319 DataManager::SetValue(TW_BACKUP_ANDSEC_VAR, 0);
1320 } else
1321 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 1);
Dees_Troy2c50e182012-09-26 20:05:28 -04001322 } else if ((*iter)->Mount_Point == "/boot") {
1323 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1324 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1325 if ((*iter)->Backup_Size == 0) {
1326 DataManager::SetValue("tw_has_boot_partition", 0);
1327 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1328 } else
1329 DataManager::SetValue("tw_has_boot_partition", 1);
Dees_Troy51127312012-09-08 13:08:49 -04001330 }
Dees_Troyab10ee22012-09-21 14:27:30 -04001331#ifdef SP1_NAME
1332 if ((*iter)->Backup_Name == EXPAND(SP1_NAME)) {
1333 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1334 DataManager::SetValue(TW_BACKUP_SP1_SIZE, backup_display_size);
1335 }
1336#endif
1337#ifdef SP2_NAME
1338 if ((*iter)->Backup_Name == EXPAND(SP2_NAME)) {
1339 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1340 DataManager::SetValue(TW_BACKUP_SP2_SIZE, backup_display_size);
1341 }
1342#endif
1343#ifdef SP3_NAME
1344 if ((*iter)->Backup_Name == EXPAND(SP3_NAME)) {
1345 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1346 DataManager::SetValue(TW_BACKUP_SP3_SIZE, backup_display_size);
1347 }
1348#endif
Dees_Troyaa9cc402012-10-13 12:14:05 -04001349 } else {
1350 // Handle unmountable partitions in case we reset defaults
1351 if ((*iter)->Mount_Point == "/boot") {
1352 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1353 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1354 if ((*iter)->Backup_Size == 0) {
1355 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 0);
1356 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1357 } else
1358 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 1);
1359 } else if ((*iter)->Mount_Point == "/recovery") {
1360 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1361 DataManager::SetValue(TW_BACKUP_RECOVERY_SIZE, backup_display_size);
1362 if ((*iter)->Backup_Size == 0) {
1363 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 0);
1364 DataManager::SetValue(TW_BACKUP_RECOVERY_VAR, 0);
1365 } else
1366 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 1);
Gary Peck82599a82012-11-21 16:23:12 -08001367 } else if ((*iter)->Mount_Point == "/data") {
1368 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troyaa9cc402012-10-13 12:14:05 -04001369 }
1370#ifdef SP1_NAME
1371 if ((*iter)->Backup_Name == EXPAND(SP1_NAME)) {
1372 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1373 DataManager::SetValue(TW_BACKUP_SP1_SIZE, backup_display_size);
1374 }
1375#endif
1376#ifdef SP2_NAME
1377 if ((*iter)->Backup_Name == EXPAND(SP2_NAME)) {
1378 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1379 DataManager::SetValue(TW_BACKUP_SP2_SIZE, backup_display_size);
1380 }
1381#endif
1382#ifdef SP3_NAME
1383 if ((*iter)->Backup_Name == EXPAND(SP3_NAME)) {
1384 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1385 DataManager::SetValue(TW_BACKUP_SP3_SIZE, backup_display_size);
1386 }
1387#endif
Dees_Troy51127312012-09-08 13:08:49 -04001388 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001389 }
that39b7c162013-11-22 22:07:33 +01001390 gui_print("...done\n");
Dees_Troy51127312012-09-08 13:08:49 -04001391 DataManager::SetValue(TW_BACKUP_DATA_SIZE, data_size);
1392 string current_storage_path = DataManager::GetCurrentStoragePath();
1393 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
Dees_Troy8170a922012-09-18 15:40:25 -04001394 if (FreeStorage != NULL) {
1395 // Attempt to mount storage
1396 if (!FreeStorage->Mount(false)) {
1397 // We couldn't mount storage... check to see if we have dual storage
1398 int has_dual_storage;
1399 DataManager::GetValue(TW_HAS_DUAL_STORAGE, has_dual_storage);
1400 if (has_dual_storage == 1) {
1401 // We have dual storage, see if we're using the internal storage that should always be present
1402 if (current_storage_path == DataManager::GetSettingsStoragePath()) {
Dees_Troyab10ee22012-09-21 14:27:30 -04001403 if (!FreeStorage->Is_Encrypted) {
1404 // Not able to use internal, so error!
Dees_Troy2673cec2013-04-02 20:22:16 +00001405 LOGERR("Unable to mount internal storage.\n");
Dees_Troyab10ee22012-09-21 14:27:30 -04001406 }
Dees_Troy8170a922012-09-18 15:40:25 -04001407 DataManager::SetValue(TW_STORAGE_FREE_SIZE, 0);
1408 } else {
1409 // We were using external, flip to internal
1410 DataManager::SetValue(TW_USE_EXTERNAL_STORAGE, 0);
1411 current_storage_path = DataManager::GetCurrentStoragePath();
1412 FreeStorage = Find_Partition_By_Path(current_storage_path);
1413 if (FreeStorage != NULL) {
1414 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
1415 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001416 LOGERR("Unable to locate internal storage partition.\n");
Dees_Troy8170a922012-09-18 15:40:25 -04001417 DataManager::SetValue(TW_STORAGE_FREE_SIZE, 0);
1418 }
1419 }
1420 } else {
1421 // No dual storage and unable to mount storage, error!
Dees_Troy2673cec2013-04-02 20:22:16 +00001422 LOGERR("Unable to mount storage.\n");
Dees_Troy8170a922012-09-18 15:40:25 -04001423 DataManager::SetValue(TW_STORAGE_FREE_SIZE, 0);
1424 }
1425 } else {
1426 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
1427 }
1428 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001429 LOGINFO("Unable to find storage partition '%s'.\n", current_storage_path.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -04001430 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001431 if (!Write_Fstab())
Dees_Troy2673cec2013-04-02 20:22:16 +00001432 LOGERR("Error creating fstab\n");
Dees_Troy51a0e822012-09-05 15:24:24 -04001433 return;
1434}
1435
1436int TWPartitionManager::Decrypt_Device(string Password) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001437#ifdef TW_INCLUDE_CRYPTO
1438 int ret_val, password_len;
1439 char crypto_blkdev[255], cPassword[255];
1440 size_t result;
Ethan Yonker253368a2014-11-25 15:00:52 -06001441 std::vector<TWPartition*>::iterator iter;
Dees_Troy5bf43922012-09-07 16:07:55 -04001442
1443 property_set("ro.crypto.state", "encrypted");
a39552696ff55ce2013-01-08 16:14:56 +00001444
Ethan Yonker253368a2014-11-25 15:00:52 -06001445 // Mount any partitions that need to be mounted for decrypt
1446 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1447 if ((*iter)->Mount_To_Decrypt) {
1448 (*iter)->Mount(true);
1449 }
a39552696ff55ce2013-01-08 16:14:56 +00001450 }
a39552696ff55ce2013-01-08 16:14:56 +00001451
Dees_Troy5bf43922012-09-07 16:07:55 -04001452 strcpy(cPassword, Password.c_str());
a39552696ff55ce2013-01-08 16:14:56 +00001453 int pwret = cryptfs_check_passwd(cPassword);
1454
Ethan Yonker253368a2014-11-25 15:00:52 -06001455 // Unmount any partitions that were needed for decrypt
1456 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1457 if ((*iter)->Mount_To_Decrypt) {
1458 (*iter)->UnMount(false);
1459 }
1460 }
1461
a39552696ff55ce2013-01-08 16:14:56 +00001462 if (pwret != 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001463 LOGERR("Failed to decrypt data.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04001464 return -1;
1465 }
a39552696ff55ce2013-01-08 16:14:56 +00001466
Dees_Troy5bf43922012-09-07 16:07:55 -04001467 property_get("ro.crypto.fs_crypto_blkdev", crypto_blkdev, "error");
1468 if (strcmp(crypto_blkdev, "error") == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001469 LOGERR("Error retrieving decrypted data block device.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04001470 } else {
1471 TWPartition* dat = Find_Partition_By_Path("/data");
1472 if (dat != NULL) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001473 DataManager::SetValue(TW_DATA_BLK_DEVICE, dat->Primary_Block_Device);
Dees_Troy5bf43922012-09-07 16:07:55 -04001474 DataManager::SetValue(TW_IS_DECRYPTED, 1);
1475 dat->Is_Decrypted = true;
1476 dat->Decrypted_Block_Device = crypto_blkdev;
Gary Peck82599a82012-11-21 16:23:12 -08001477 dat->Setup_File_System(false);
Dees_Troy74fb2e92013-04-15 14:35:47 +00001478 dat->Current_File_System = dat->Fstab_File_System; // Needed if we're ignoring blkid because encrypted devices start out as emmc
Dees_Troy2673cec2013-04-02 20:22:16 +00001479 gui_print("Data successfully decrypted, new block device: '%s'\n", crypto_blkdev);
a39552696ff55ce2013-01-08 16:14:56 +00001480
Dees_Troy5bf43922012-09-07 16:07:55 -04001481 // Sleep for a bit so that the device will be ready
1482 sleep(1);
Ethan Yonker6277c792014-09-15 14:54:30 -05001483 if (dat->Has_Data_Media && dat->Mount(false) && TWFunc::Path_Exists("/data/media/0")) {
Dees_Troy16b74352012-11-14 22:27:31 +00001484 dat->Storage_Path = "/data/media/0";
1485 dat->Symlink_Path = dat->Storage_Path;
Dees Troy98fb46c2013-12-04 16:56:45 +00001486 DataManager::SetValue("tw_storage_path", "/data/media/0");
Ethan Yonkerec0fa4a2014-11-20 09:51:03 -06001487 DataManager::SetValue("tw_settings_path", "/data/media/0");
Dees_Troy16b74352012-11-14 22:27:31 +00001488 dat->UnMount(false);
Dees_Troydc8bc1b2013-01-17 01:39:28 +00001489 Output_Partition(dat);
Dees_Troy16b74352012-11-14 22:27:31 +00001490 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001491 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001492 UnMount_Main_Partitions();
Dees_Troy5bf43922012-09-07 16:07:55 -04001493 } else
Dees_Troy2673cec2013-04-02 20:22:16 +00001494 LOGERR("Unable to locate data partition.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04001495 }
1496 return 0;
1497#else
Dees_Troy2673cec2013-04-02 20:22:16 +00001498 LOGERR("No crypto support was compiled into this build.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04001499 return -1;
1500#endif
Dees_Troy51a0e822012-09-05 15:24:24 -04001501 return 1;
Dees_Troy51127312012-09-08 13:08:49 -04001502}
1503
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04001504int TWPartitionManager::Fix_Permissions(void) {
1505 int result = 0;
1506 if (!Mount_By_Path("/data", true))
1507 return false;
1508
1509 if (!Mount_By_Path("/system", true))
1510 return false;
1511
1512 Mount_By_Path("/sd-ext", false);
1513
1514 fixPermissions perms;
1515 result = perms.fixPerms(true, false);
thata3d31fb2014-12-21 22:27:40 +01001516#ifdef HAVE_SELINUX
1517 if (result == 0 && DataManager::GetIntValue("tw_fixperms_restorecon") == 1)
1518 result = perms.fixContexts();
1519#endif
Dees_Troy1a650e62012-10-19 20:54:32 -04001520 UnMount_Main_Partitions();
Dees_Troy2673cec2013-04-02 20:22:16 +00001521 gui_print("Done.\n\n");
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04001522 return result;
1523}
1524
Ethan Yonker47360be2014-04-01 10:34:34 -05001525TWPartition* TWPartitionManager::Find_Next_Storage(string Path, string Exclude) {
1526 std::vector<TWPartition*>::iterator iter = Partitions.begin();
1527
1528 if (!Path.empty()) {
1529 string Search_Path = TWFunc::Get_Root_Path(Path);
1530 for (; iter != Partitions.end(); iter++) {
1531 if ((*iter)->Mount_Point == Search_Path) {
1532 iter++;
1533 break;
1534 }
1535 }
1536 }
1537
1538 for (; iter != Partitions.end(); iter++) {
1539 if ((*iter)->Is_Storage && (*iter)->Is_Present && (*iter)->Mount_Point != Exclude) {
1540 return (*iter);
1541 }
1542 }
1543
1544 return NULL;
1545}
1546
Dees_Troyd21618c2012-10-14 18:48:49 -04001547int TWPartitionManager::Open_Lun_File(string Partition_Path, string Lun_File) {
Dees_Troyd21618c2012-10-14 18:48:49 -04001548 TWPartition* Part = Find_Partition_By_Path(Partition_Path);
1549
1550 if (Part == NULL) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001551 LOGERR("Unable to locate '%s' for USB storage mode.", Partition_Path.c_str());
Dees_Troyd21618c2012-10-14 18:48:49 -04001552 return false;
1553 }
Ethan Yonker47360be2014-04-01 10:34:34 -05001554 LOGINFO("USB mount '%s', '%s' > '%s'\n", Partition_Path.c_str(), Part->Actual_Block_Device.c_str(), Lun_File.c_str());
1555 if (!Part->UnMount(true) || !Part->Is_Present)
Dees_Troyd21618c2012-10-14 18:48:49 -04001556 return false;
1557
Dees_Troy2673cec2013-04-02 20:22:16 +00001558 if (TWFunc::write_file(Lun_File, Part->Actual_Block_Device)) {
1559 LOGERR("Unable to write to ums lunfile '%s': (%s)\n", Lun_File.c_str(), strerror(errno));
Dees_Troyd21618c2012-10-14 18:48:49 -04001560 return false;
1561 }
Dees_Troyd21618c2012-10-14 18:48:49 -04001562 return true;
1563}
1564
Dees_Troy8170a922012-09-18 15:40:25 -04001565int TWPartitionManager::usb_storage_enable(void) {
Dees_Troyd21618c2012-10-14 18:48:49 -04001566 int has_dual, has_data_media;
Dees_Troy8170a922012-09-18 15:40:25 -04001567 char lun_file[255];
Dees_Troyd21618c2012-10-14 18:48:49 -04001568 bool has_multiple_lun = false;
Dees_Troy8170a922012-09-18 15:40:25 -04001569
Dees_Troy8170a922012-09-18 15:40:25 -04001570 DataManager::GetValue(TW_HAS_DATA_MEDIA, has_data_media);
Ethan Yonker47360be2014-04-01 10:34:34 -05001571 string Lun_File_str = CUSTOM_LUN_FILE;
1572 size_t found = Lun_File_str.find("%");
1573 if (found != string::npos) {
1574 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
1575 if (TWFunc::Path_Exists(lun_file))
1576 has_multiple_lun = true;
1577 }
Ethan Yonker726a0202014-12-16 20:01:38 -06001578 mtp_was_enabled = TWFunc::Toggle_MTP(false); // Must disable MTP for USB Storage
Ethan Yonker47360be2014-04-01 10:34:34 -05001579 if (!has_multiple_lun) {
1580 LOGINFO("Device doesn't have multiple lun files, mount current storage\n");
1581 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
1582 if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) == "/data") {
1583 TWPartition* Mount = Find_Next_Storage("", "/data");
1584 if (Mount) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001585 if (!Open_Lun_File(Mount->Mount_Point, lun_file)) {
1586 goto error_handle;
1587 }
Ethan Yonker47360be2014-04-01 10:34:34 -05001588 } else {
1589 LOGERR("Unable to find storage partition to mount to USB\n");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001590 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05001591 }
1592 } else if (!Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file)) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001593 goto error_handle;
Dees_Troy8170a922012-09-18 15:40:25 -04001594 }
Dees_Troy8170a922012-09-18 15:40:25 -04001595 } else {
Ethan Yonker47360be2014-04-01 10:34:34 -05001596 LOGINFO("Device has multiple lun files\n");
1597 TWPartition* Mount1;
1598 TWPartition* Mount2;
Dees_Troy8170a922012-09-18 15:40:25 -04001599 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
Ethan Yonker47360be2014-04-01 10:34:34 -05001600 Mount1 = Find_Next_Storage("", "/data");
1601 if (Mount1) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001602 if (!Open_Lun_File(Mount1->Mount_Point, lun_file)) {
1603 goto error_handle;
1604 }
Matt Moweree717062014-04-26 01:46:46 -05001605 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
Ethan Yonker47360be2014-04-01 10:34:34 -05001606 Mount2 = Find_Next_Storage(Mount1->Mount_Point, "/data");
1607 if (Mount2) {
Matt Moweree717062014-04-26 01:46:46 -05001608 Open_Lun_File(Mount2->Mount_Point, lun_file);
Ethan Yonker47360be2014-04-01 10:34:34 -05001609 }
1610 } else {
1611 LOGERR("Unable to find storage partition to mount to USB\n");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001612 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05001613 }
Dees_Troy8170a922012-09-18 15:40:25 -04001614 }
Matt Mowerb6ef4782014-11-06 02:24:36 -06001615 property_set("sys.storage.ums_enabled", "1");
Dees_Troy8170a922012-09-18 15:40:25 -04001616 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001617error_handle:
1618 if (mtp_was_enabled)
1619 if (!Enable_MTP())
1620 Disable_MTP();
1621 return false;
Dees_Troy8170a922012-09-18 15:40:25 -04001622}
1623
1624int TWPartitionManager::usb_storage_disable(void) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001625 int index, ret;
1626 char lun_file[255], ch[2] = {0, 0};
1627 string str = ch;
Dees_Troy8170a922012-09-18 15:40:25 -04001628
1629 for (index=0; index<2; index++) {
1630 sprintf(lun_file, CUSTOM_LUN_FILE, index);
Dees_Troy2673cec2013-04-02 20:22:16 +00001631 ret = TWFunc::write_file(lun_file, str);
Dees_Troy2673cec2013-04-02 20:22:16 +00001632 if (ret < 0) {
1633 break;
Dees_Troy8170a922012-09-18 15:40:25 -04001634 }
Dees_Troy8170a922012-09-18 15:40:25 -04001635 }
Dees_Troye58d5262012-09-21 12:27:57 -04001636 Mount_All_Storage();
1637 Update_System_Details();
Dees_Troycfd73ef2012-10-12 16:52:00 -04001638 UnMount_Main_Partitions();
Matt Mowerd9cb9062013-12-14 20:34:45 -06001639 property_set("sys.storage.ums_enabled", "0");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001640 if (mtp_was_enabled)
1641 if (!Enable_MTP())
1642 Disable_MTP();
Dees_Troy2673cec2013-04-02 20:22:16 +00001643 if (ret < 0 && index == 0) {
1644 LOGERR("Unable to write to ums lunfile '%s'.", lun_file);
1645 return false;
1646 } else {
1647 return true;
1648 }
Dees_Troy8170a922012-09-18 15:40:25 -04001649 return true;
Dees_Troy812660f2012-09-20 09:55:17 -04001650}
1651
1652void TWPartitionManager::Mount_All_Storage(void) {
1653 std::vector<TWPartition*>::iterator iter;
1654
1655 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1656 if ((*iter)->Is_Storage)
1657 (*iter)->Mount(false);
1658 }
Dees_Troy2c50e182012-09-26 20:05:28 -04001659}
Dees_Troy9350b8d2012-09-27 12:38:38 -04001660
Dees_Troyd0384ef2012-10-12 12:15:42 -04001661void TWPartitionManager::UnMount_Main_Partitions(void) {
1662 // Unmounts system and data if data is not data/media
1663 // Also unmounts boot if boot is mountable
Dees_Troy2673cec2013-04-02 20:22:16 +00001664 LOGINFO("Unmounting main partitions...\n");
Dees_Troyd0384ef2012-10-12 12:15:42 -04001665
1666 TWPartition* Boot_Partition = Find_Partition_By_Path("/boot");
1667
1668 UnMount_By_Path("/system", true);
Ethan Yonker6277c792014-09-15 14:54:30 -05001669 if (!datamedia)
1670 UnMount_By_Path("/data", true);
1671
Dees_Troyd0384ef2012-10-12 12:15:42 -04001672 if (Boot_Partition != NULL && Boot_Partition->Can_Be_Mounted)
1673 Boot_Partition->UnMount(true);
1674}
1675
Dees_Troy9350b8d2012-09-27 12:38:38 -04001676int TWPartitionManager::Partition_SDCard(void) {
1677 char mkdir_path[255], temp[255], line[512];
1678 string Command, Device, fat_str, ext_str, swap_str, start_loc, end_loc, ext_format, sd_path, tmpdevice;
1679 int ext, swap, total_size = 0, fat_size;
1680 FILE* fp;
1681
Dees_Troy2673cec2013-04-02 20:22:16 +00001682 gui_print("Partitioning SD Card...\n");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001683#ifdef TW_EXTERNAL_STORAGE_PATH
1684 TWPartition* SDCard = Find_Partition_By_Path(EXPAND(TW_EXTERNAL_STORAGE_PATH));
1685#else
1686 TWPartition* SDCard = Find_Partition_By_Path("/sdcard");
1687#endif
Ethan Yonker1eff6cd2014-09-15 13:30:42 -05001688 if (SDCard == NULL || !SDCard->Removable || SDCard->Has_Data_Media) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001689 LOGERR("Unable to locate device to partition.\n");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001690 return false;
1691 }
1692 if (!SDCard->UnMount(true))
1693 return false;
1694 TWPartition* SDext = Find_Partition_By_Path("/sd-ext");
1695 if (SDext != NULL) {
1696 if (!SDext->UnMount(true))
1697 return false;
1698 }
Vojtech Bocek05534202013-09-11 08:11:56 +02001699
1700 TWFunc::Exec_Cmd("umount \"$SWAPPATH\"");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001701 Device = SDCard->Actual_Block_Device;
1702 // Just use the root block device
1703 Device.resize(strlen("/dev/block/mmcblkX"));
1704
1705 // Find the size of the block device:
1706 fp = fopen("/proc/partitions", "rt");
1707 if (fp == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001708 LOGERR("Unable to open /proc/partitions\n");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001709 return false;
1710 }
1711
1712 while (fgets(line, sizeof(line), fp) != NULL)
1713 {
1714 unsigned long major, minor, blocks;
1715 char device[512];
1716 char tmpString[64];
1717
1718 if (strlen(line) < 7 || line[0] == 'm') continue;
1719 sscanf(line + 1, "%lu %lu %lu %s", &major, &minor, &blocks, device);
1720
1721 tmpdevice = "/dev/block/";
1722 tmpdevice += device;
1723 if (tmpdevice == Device) {
1724 // Adjust block size to byte size
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001725 total_size = (int)(blocks * 1024ULL / 1000000LLU);
Dees_Troy9350b8d2012-09-27 12:38:38 -04001726 break;
1727 }
1728 }
1729 fclose(fp);
1730
1731 DataManager::GetValue("tw_sdext_size", ext);
1732 DataManager::GetValue("tw_swap_size", swap);
1733 DataManager::GetValue("tw_sdpart_file_system", ext_format);
1734 fat_size = total_size - ext - swap;
Dees_Troy2673cec2013-04-02 20:22:16 +00001735 LOGINFO("sd card block device is '%s', sdcard size is: %iMB, fat size: %iMB, ext size: %iMB, ext system: '%s', swap size: %iMB\n", Device.c_str(), total_size, fat_size, ext, ext_format.c_str(), swap);
Dees_Troy9350b8d2012-09-27 12:38:38 -04001736 memset(temp, 0, sizeof(temp));
1737 sprintf(temp, "%i", fat_size);
1738 fat_str = temp;
1739 memset(temp, 0, sizeof(temp));
1740 sprintf(temp, "%i", fat_size + ext);
1741 ext_str = temp;
1742 memset(temp, 0, sizeof(temp));
1743 sprintf(temp, "%i", fat_size + ext + swap);
1744 swap_str = temp;
1745 if (ext + swap > total_size) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001746 LOGERR("EXT + Swap size is larger than sdcard size.\n");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001747 return false;
1748 }
Dees_Troy2673cec2013-04-02 20:22:16 +00001749 gui_print("Removing partition table...\n");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001750 Command = "parted -s " + Device + " mklabel msdos";
Dees_Troy2673cec2013-04-02 20:22:16 +00001751 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001752 if (TWFunc::Exec_Cmd(Command) != 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001753 LOGERR("Unable to remove partition table.\n");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001754 Update_System_Details();
1755 return false;
1756 }
Dees_Troy2673cec2013-04-02 20:22:16 +00001757 gui_print("Creating FAT32 partition...\n");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001758 Command = "parted " + Device + " mkpartfs primary fat32 0 " + fat_str + "MB";
Dees_Troy2673cec2013-04-02 20:22:16 +00001759 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001760 if (TWFunc::Exec_Cmd(Command) != 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001761 LOGERR("Unable to create FAT32 partition.\n");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001762 return false;
1763 }
1764 if (ext > 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001765 gui_print("Creating EXT partition...\n");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001766 Command = "parted " + Device + " mkpartfs primary ext2 " + fat_str + "MB " + ext_str + "MB";
Dees_Troy2673cec2013-04-02 20:22:16 +00001767 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001768 if (TWFunc::Exec_Cmd(Command) != 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001769 LOGERR("Unable to create EXT partition.\n");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001770 Update_System_Details();
1771 return false;
1772 }
1773 }
1774 if (swap > 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001775 gui_print("Creating swap partition...\n");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001776 Command = "parted " + Device + " mkpartfs primary linux-swap " + ext_str + "MB " + swap_str + "MB";
Dees_Troy2673cec2013-04-02 20:22:16 +00001777 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001778 if (TWFunc::Exec_Cmd(Command) != 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001779 LOGERR("Unable to create swap partition.\n");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001780 Update_System_Details();
1781 return false;
1782 }
1783 }
1784 // recreate TWRP folder and rewrite settings - these will be gone after sdcard is partitioned
1785#ifdef TW_EXTERNAL_STORAGE_PATH
1786 Mount_By_Path(EXPAND(TW_EXTERNAL_STORAGE_PATH), 1);
1787 DataManager::GetValue(TW_EXTERNAL_PATH, sd_path);
1788 memset(mkdir_path, 0, sizeof(mkdir_path));
1789 sprintf(mkdir_path, "%s/TWRP", sd_path.c_str());
1790#else
1791 Mount_By_Path("/sdcard", 1);
1792 strcpy(mkdir_path, "/sdcard/TWRP");
1793#endif
1794 mkdir(mkdir_path, 0777);
1795 DataManager::Flush();
1796#ifdef TW_EXTERNAL_STORAGE_PATH
1797 DataManager::SetValue(TW_ZIP_EXTERNAL_VAR, EXPAND(TW_EXTERNAL_STORAGE_PATH));
1798 if (DataManager::GetIntValue(TW_USE_EXTERNAL_STORAGE) == 1)
1799 DataManager::SetValue(TW_ZIP_LOCATION_VAR, EXPAND(TW_EXTERNAL_STORAGE_PATH));
1800#else
1801 DataManager::SetValue(TW_ZIP_EXTERNAL_VAR, "/sdcard");
1802 if (DataManager::GetIntValue(TW_USE_EXTERNAL_STORAGE) == 1)
1803 DataManager::SetValue(TW_ZIP_LOCATION_VAR, "/sdcard");
1804#endif
1805 if (ext > 0) {
1806 if (SDext == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001807 LOGERR("Unable to locate sd-ext partition.\n");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001808 return false;
1809 }
1810 Command = "mke2fs -t " + ext_format + " -m 0 " + SDext->Actual_Block_Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00001811 gui_print("Formatting sd-ext as %s...\n", ext_format.c_str());
1812 LOGINFO("Formatting sd-ext after partitioning, command: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02001813 TWFunc::Exec_Cmd(Command);
Dees_Troy9350b8d2012-09-27 12:38:38 -04001814 }
1815
1816 Update_System_Details();
Dees_Troy2673cec2013-04-02 20:22:16 +00001817 gui_print("Partitioning complete.\n");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001818 return true;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04001819}
Dees_Troya13d74f2013-03-24 08:54:55 -05001820
1821void TWPartitionManager::Get_Partition_List(string ListType, std::vector<PartitionList> *Partition_List) {
1822 std::vector<TWPartition*>::iterator iter;
1823 if (ListType == "mount") {
1824 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1825 if ((*iter)->Can_Be_Mounted && !(*iter)->Is_SubPartition) {
1826 struct PartitionList part;
1827 part.Display_Name = (*iter)->Display_Name;
1828 part.Mount_Point = (*iter)->Mount_Point;
1829 part.selected = (*iter)->Is_Mounted();
1830 Partition_List->push_back(part);
1831 }
1832 }
1833 } else if (ListType == "storage") {
1834 char free_space[255];
1835 string Current_Storage = DataManager::GetCurrentStoragePath();
1836 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1837 if ((*iter)->Is_Storage) {
1838 struct PartitionList part;
1839 sprintf(free_space, "%llu", (*iter)->Free / 1024 / 1024);
1840 part.Display_Name = (*iter)->Storage_Name + " (";
1841 part.Display_Name += free_space;
1842 part.Display_Name += "MB)";
1843 part.Mount_Point = (*iter)->Storage_Path;
1844 if ((*iter)->Storage_Path == Current_Storage)
1845 part.selected = 1;
1846 else
1847 part.selected = 0;
1848 Partition_List->push_back(part);
1849 }
1850 }
1851 } else if (ListType == "backup") {
1852 char backup_size[255];
Dees_Troy9e0b71c2013-04-08 13:35:37 +00001853 unsigned long long Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -05001854 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy9e0b71c2013-04-08 13:35:37 +00001855 if ((*iter)->Can_Be_Backed_Up && !(*iter)->Is_SubPartition && (*iter)->Is_Present) {
Dees_Troya13d74f2013-03-24 08:54:55 -05001856 struct PartitionList part;
Dees_Troy9e0b71c2013-04-08 13:35:37 +00001857 Backup_Size = (*iter)->Backup_Size;
1858 if ((*iter)->Has_SubPartition) {
1859 std::vector<TWPartition*>::iterator subpart;
1860
1861 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
1862 if ((*subpart)->Is_SubPartition && (*subpart)->Can_Be_Backed_Up && (*subpart)->Is_Present && (*subpart)->SubPartition_Of == (*iter)->Mount_Point)
1863 Backup_Size += (*subpart)->Backup_Size;
1864 }
1865 }
1866 sprintf(backup_size, "%llu", Backup_Size / 1024 / 1024);
Dees_Troya13d74f2013-03-24 08:54:55 -05001867 part.Display_Name = (*iter)->Backup_Display_Name + " (";
1868 part.Display_Name += backup_size;
1869 part.Display_Name += "MB)";
1870 part.Mount_Point = (*iter)->Backup_Path;
1871 part.selected = 0;
1872 Partition_List->push_back(part);
1873 }
1874 }
1875 } else if (ListType == "restore") {
1876 string Restore_List, restore_path;
1877 TWPartition* restore_part = NULL;
1878
1879 DataManager::GetValue("tw_restore_list", Restore_List);
1880 if (!Restore_List.empty()) {
1881 size_t start_pos = 0, end_pos = Restore_List.find(";", start_pos);
1882 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1883 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
Dees_Troy59df9262013-06-19 14:53:57 -05001884 if ((restore_part = Find_Partition_By_Path(restore_path)) != NULL) {
Dees Troy4159aed2014-02-28 17:24:43 +00001885 if ((restore_part->Backup_Name == "recovery" && !restore_part->Can_Be_Backed_Up) || restore_part->Is_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -05001886 // Don't allow restore of recovery (causes problems on some devices)
Dees_Troy59df9262013-06-19 14:53:57 -05001887 // Don't add subpartitions to the list of items
Dees_Troya13d74f2013-03-24 08:54:55 -05001888 } else {
1889 struct PartitionList part;
1890 part.Display_Name = restore_part->Backup_Display_Name;
1891 part.Mount_Point = restore_part->Backup_Path;
1892 part.selected = 1;
1893 Partition_List->push_back(part);
1894 }
1895 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001896 LOGERR("Unable to locate '%s' partition for restore.\n", restore_path.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -05001897 }
1898 start_pos = end_pos + 1;
1899 end_pos = Restore_List.find(";", start_pos);
1900 }
1901 }
1902 } else if (ListType == "wipe") {
1903 struct PartitionList dalvik;
1904 dalvik.Display_Name = "Dalvik Cache";
1905 dalvik.Mount_Point = "DALVIK";
1906 dalvik.selected = 0;
1907 Partition_List->push_back(dalvik);
1908 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1909 if ((*iter)->Wipe_Available_in_GUI && !(*iter)->Is_SubPartition) {
1910 struct PartitionList part;
1911 part.Display_Name = (*iter)->Display_Name;
1912 part.Mount_Point = (*iter)->Mount_Point;
1913 part.selected = 0;
1914 Partition_List->push_back(part);
1915 }
1916 if ((*iter)->Has_Android_Secure) {
1917 struct PartitionList part;
1918 part.Display_Name = (*iter)->Backup_Display_Name;
1919 part.Mount_Point = (*iter)->Backup_Path;
1920 part.selected = 0;
1921 Partition_List->push_back(part);
1922 }
Dees_Troy74fb2e92013-04-15 14:35:47 +00001923 if ((*iter)->Has_Data_Media) {
1924 struct PartitionList datamedia;
1925 datamedia.Display_Name = (*iter)->Storage_Name;
1926 datamedia.Mount_Point = "INTERNAL";
1927 datamedia.selected = 0;
1928 Partition_List->push_back(datamedia);
1929 }
Dees_Troya13d74f2013-03-24 08:54:55 -05001930 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06001931 } else if (ListType == "flashimg") {
1932 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1933 if ((*iter)->Can_Flash_Img && (*iter)->Is_Present) {
1934 struct PartitionList part;
1935 part.Display_Name = (*iter)->Backup_Display_Name;
1936 part.Mount_Point = (*iter)->Backup_Path;
1937 part.selected = 0;
1938 Partition_List->push_back(part);
1939 }
1940 }
Dees_Troya13d74f2013-03-24 08:54:55 -05001941 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001942 LOGERR("Unknown list type '%s' requested for TWPartitionManager::Get_Partition_List\n", ListType.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -05001943 }
1944}
1945
1946int TWPartitionManager::Fstab_Processed(void) {
1947 return Partitions.size();
1948}
Dees_Troyd93bda52013-07-03 19:55:19 +00001949
1950void TWPartitionManager::Output_Storage_Fstab(void) {
1951 std::vector<TWPartition*>::iterator iter;
1952 char storage_partition[255];
1953 string Temp;
1954 FILE *fp = fopen("/cache/recovery/storage.fstab", "w");
1955
1956 if (fp == NULL) {
1957 LOGERR("Unable to open '/cache/recovery/storage.fstab'.\n");
1958 return;
1959 }
1960
1961 // Iterate through all partitions
1962 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1963 if ((*iter)->Is_Storage) {
1964 Temp = (*iter)->Storage_Path + ";" + (*iter)->Storage_Name + ";\n";
1965 strcpy(storage_partition, Temp.c_str());
1966 fwrite(storage_partition, sizeof(storage_partition[0]), strlen(storage_partition) / sizeof(storage_partition[0]), fp);
1967 }
1968 }
1969 fclose(fp);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001970}
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02001971
1972TWPartition *TWPartitionManager::Get_Default_Storage_Partition()
1973{
1974 TWPartition *res = NULL;
1975 for (std::vector<TWPartition*>::iterator iter = Partitions.begin(); iter != Partitions.end(); ++iter) {
1976 if(!(*iter)->Is_Storage)
1977 continue;
1978
1979 if((*iter)->Is_Settings_Storage)
1980 return *iter;
1981
1982 if(!res)
1983 res = *iter;
1984 }
1985 return res;
1986}
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001987
1988bool TWPartitionManager::Enable_MTP(void) {
1989#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05001990 if (mtppid) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001991 LOGERR("MTP already enabled\n");
1992 return true;
1993 }
1994 //Launch MTP Responder
1995 LOGINFO("Starting MTP\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06001996
1997 int mtppipe[2];
1998
1999 if (pipe(mtppipe) < 0) {
2000 LOGERR("Error creating MTP pipe\n");
2001 return false;
2002 }
2003
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002004 char old_value[PROPERTY_VALUE_MAX];
2005 property_get("sys.usb.config", old_value, "error");
2006 if (strcmp(old_value, "error") != 0 && strcmp(old_value, "mtp,adb") != 0) {
2007 char vendor[PROPERTY_VALUE_MAX];
2008 char product[PROPERTY_VALUE_MAX];
2009 property_set("sys.usb.config", "none");
2010 property_get("usb.vendor", vendor, "18D1");
2011 property_get("usb.product.mtpadb", product, "4EE2");
2012 string vendorstr = vendor;
2013 string productstr = product;
2014 TWFunc::write_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2015 TWFunc::write_file("/sys/class/android_usb/android0/idProduct", productstr);
2016 property_set("sys.usb.config", "mtp,adb");
2017 }
Ethan Yonker6d154c42014-09-03 14:19:43 -05002018 /* To enable MTP debug, use the twrp command line feature to
2019 * twrp set tw_mtp_debug 1
2020 */
2021 twrpMtp *mtp = new twrpMtp(DataManager::GetIntValue("tw_mtp_debug"));
Ethan Yonker1b039202015-01-30 10:08:48 -06002022 mtppid = mtp->forkserver(mtppipe);
2023 if (mtppid) {
2024 close(mtppipe[0]); // Host closes read side
2025 mtp_write_fd = mtppipe[1];
2026 DataManager::SetValue("tw_mtp_enabled", 1);
2027 Add_All_MTP_Storage();
2028 return true;
Ethan Yonker726a0202014-12-16 20:01:38 -06002029 } else {
2030 close(mtppipe[0]);
2031 close(mtppipe[1]);
Ethan Yonker1b039202015-01-30 10:08:48 -06002032 LOGERR("Failed to enable MTP\n");
2033 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002034 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002035#else
2036 LOGERR("MTP support not included\n");
2037#endif
2038 DataManager::SetValue("tw_mtp_enabled", 0);
2039 return false;
2040}
2041
Ethan Yonker1b039202015-01-30 10:08:48 -06002042void TWPartitionManager::Add_All_MTP_Storage(void) {
2043#ifdef TW_HAS_MTP
2044 std::vector<TWPartition*>::iterator iter;
2045
2046 if (!mtppid)
2047 return; // MTP is not enabled
2048
2049 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2050 if ((*iter)->Is_Storage && (*iter)->Is_Present && (*iter)->Mount(false))
2051 Add_Remove_MTP_Storage((*iter), MTP_MESSAGE_ADD_STORAGE);
2052 }
2053#else
2054 return;
2055#endif
2056}
2057
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002058bool TWPartitionManager::Disable_MTP(void) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002059 char old_value[PROPERTY_VALUE_MAX];
2060 property_get("sys.usb.config", old_value, "error");
2061 if (strcmp(old_value, "adb") != 0) {
2062 char vendor[PROPERTY_VALUE_MAX];
2063 char product[PROPERTY_VALUE_MAX];
2064 property_set("sys.usb.config", "none");
2065 property_get("usb.vendor", vendor, "18D1");
2066 property_get("usb.product.adb", product, "D002");
2067 string vendorstr = vendor;
2068 string productstr = product;
2069 TWFunc::write_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2070 TWFunc::write_file("/sys/class/android_usb/android0/idProduct", productstr);
2071 usleep(2000);
2072 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002073#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002074 if (mtppid) {
Ethan Yonker8613dc02014-09-11 09:28:20 -05002075 LOGINFO("Disabling MTP\n");
2076 int status;
2077 kill(mtppid, SIGKILL);
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002078 mtppid = 0;
Ethan Yonker8613dc02014-09-11 09:28:20 -05002079 // We don't care about the exit value, but this prevents a zombie process
2080 waitpid(mtppid, &status, 0);
Ethan Yonker726a0202014-12-16 20:01:38 -06002081 close(mtp_write_fd);
2082 mtp_write_fd = -1;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002083 }
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002084#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002085 property_set("sys.usb.config", "adb");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002086#ifdef TW_HAS_MTP
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002087 DataManager::SetValue("tw_mtp_enabled", 0);
2088 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002089#endif
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002090 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002091}
Ethan Yonker726a0202014-12-16 20:01:38 -06002092
2093TWPartition* TWPartitionManager::Find_Partition_By_MTP_Storage_ID(unsigned int Storage_ID) {
2094 std::vector<TWPartition*>::iterator iter;
2095
2096 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2097 if ((*iter)->MTP_Storage_ID == Storage_ID)
2098 return (*iter);
2099 }
2100 return NULL;
2101}
2102
2103bool TWPartitionManager::Add_Remove_MTP_Storage(TWPartition* Part, int message_type) {
2104#ifdef TW_HAS_MTP
2105 struct mtpmsg mtp_message;
2106
2107 if (!mtppid)
2108 return false; // MTP is disabled
2109
2110 if (mtp_write_fd < 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002111 LOGINFO("MTP: mtp_write_fd is not set\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002112 return false;
2113 }
2114
2115 if (Part) {
Ethan Yonker4bfabab2014-12-29 09:15:37 -06002116 if (Part->MTP_Storage_ID == 0)
2117 return false;
Ethan Yonker726a0202014-12-16 20:01:38 -06002118 if (message_type == MTP_MESSAGE_REMOVE_STORAGE) {
2119 mtp_message.message_type = MTP_MESSAGE_REMOVE_STORAGE; // Remove
2120 LOGINFO("sending message to remove %i\n", Part->MTP_Storage_ID);
2121 mtp_message.storage_id = Part->MTP_Storage_ID;
2122 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002123 LOGINFO("error sending message to remove storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002124 return false;
2125 } else {
2126 LOGINFO("Message sent, remove storage ID: %i\n", Part->MTP_Storage_ID);
2127 return true;
2128 }
2129 } else if (message_type == MTP_MESSAGE_ADD_STORAGE && Part->Is_Mounted()) {
2130 mtp_message.message_type = MTP_MESSAGE_ADD_STORAGE; // Add
2131 mtp_message.storage_id = Part->MTP_Storage_ID;
2132 mtp_message.path = Part->Storage_Path.c_str();
2133 mtp_message.display = Part->Storage_Name.c_str();
2134 mtp_message.maxFileSize = Part->Get_Max_FileSize();
Ethan Yonker1b039202015-01-30 10:08:48 -06002135 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 -06002136 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002137 LOGINFO("error sending message to add storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002138 return false;
2139 } else {
2140 LOGINFO("Message sent, add storage ID: %i\n", Part->MTP_Storage_ID);
2141 return true;
2142 }
2143 } else {
2144 LOGERR("Unknown MTP message type: %i\n", message_type);
2145 }
2146 } else {
2147 // This hopefully never happens as the error handling should
2148 // occur in the calling function.
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002149 LOGINFO("TWPartitionManager::Add_Remove_MTP_Storage NULL partition given\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002150 }
2151 return true;
2152#else
2153 LOGERR("MTP support not included\n");
2154 DataManager::SetValue("tw_mtp_enabled", 0);
2155 return false;
2156#endif
2157}
2158
2159bool TWPartitionManager::Add_MTP_Storage(string Mount_Point) {
2160#ifdef TW_HAS_MTP
2161 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2162 if (Part) {
2163 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2164 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002165 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002166 }
2167#endif
2168 return false;
2169}
2170
2171bool TWPartitionManager::Add_MTP_Storage(unsigned int Storage_ID) {
2172#ifdef TW_HAS_MTP
2173 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2174 if (Part) {
2175 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2176 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002177 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002178 }
2179#endif
2180 return false;
2181}
2182
2183bool TWPartitionManager::Remove_MTP_Storage(string Mount_Point) {
2184#ifdef TW_HAS_MTP
2185 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2186 if (Part) {
2187 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2188 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002189 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002190 }
2191#endif
2192 return false;
2193}
2194
2195bool TWPartitionManager::Remove_MTP_Storage(unsigned int Storage_ID) {
2196#ifdef TW_HAS_MTP
2197 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2198 if (Part) {
2199 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2200 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002201 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002202 }
2203#endif
2204 return false;
2205}
Ethan Yonker96af84a2015-01-05 14:58:36 -06002206
2207bool TWPartitionManager::Flash_Image(string Filename) {
2208 int check, partition_count = 0;
2209 TWPartition* flash_part = NULL;
2210 string Flash_List, flash_path;
2211 size_t start_pos = 0, end_pos = 0;
2212
2213 gui_print("\n[IMAGE FLASH STARTED]\n\n");
2214 gui_print("Image to flash: '%s'\n", Filename.c_str());
2215
2216 if (!Mount_Current_Storage(true))
2217 return false;
2218
2219 gui_print("Calculating restore details...\n");
2220 DataManager::GetValue("tw_flash_partition", Flash_List);
2221 if (!Flash_List.empty()) {
2222 end_pos = Flash_List.find(";", start_pos);
2223 while (end_pos != string::npos && start_pos < Flash_List.size()) {
2224 flash_path = Flash_List.substr(start_pos, end_pos - start_pos);
2225 flash_part = Find_Partition_By_Path(flash_path);
2226 if (flash_part != NULL) {
2227 partition_count++;
2228 if (partition_count > 1) {
2229 LOGERR("Too many partitions selected for flashing.\n");
2230 return false;
2231 }
2232 } else {
2233 LOGERR("Unable to locate '%s' partition for flashing (flash list).\n", flash_path.c_str());
2234 return false;
2235 }
2236 start_pos = end_pos + 1;
2237 end_pos = Flash_List.find(";", start_pos);
2238 }
2239 }
2240
2241 if (partition_count == 0) {
2242 LOGERR("No partitions selected for flashing.\n");
2243 return false;
2244 }
2245
2246 DataManager::SetProgress(0.0);
2247 if (flash_part) {
2248 if (!flash_part->Flash_Image(Filename))
2249 return false;
2250 } else {
2251 LOGERR("Invalid flash partition specified.\n");
2252 return false;
2253 }
2254 gui_print_color("highlight", "[IMAGE FLASH COMPLETED]\n\n");
2255 return true;
2256}