blob: 3c2b1c65e6107c5421064e2d688322d9ae85ead5 [file] [log] [blame]
Dees Troy3be70a82013-10-22 14:25:12 +00001/*
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04002 Copyright 2014 to 2017 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>
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060025#include <map>
Dees_Troy5bf43922012-09-07 16:07:55 -040026#include <vector>
Dees_Troy63c8df72012-09-10 14:02:05 -040027#include <dirent.h>
28#include <time.h>
Dees_Troy8170a922012-09-18 15:40:25 -040029#include <errno.h>
30#include <fcntl.h>
bigbiffce8f83c2015-12-12 18:30:21 -050031#include <zlib.h>
bigbiff bigbiff9c754052013-01-09 09:09:08 -050032#include <iostream>
33#include <iomanip>
Ethan Yonker8613dc02014-09-11 09:28:20 -050034#include <sys/wait.h>
Ethan Yonker483e9f42016-01-11 22:21:18 -060035#include <linux/fs.h>
36#include <sys/mount.h>
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060037
38#include <sys/poll.h>
39#include <sys/socket.h>
40#include <linux/types.h>
41#include <linux/netlink.h>
42
Dees_Troy51a0e822012-09-05 15:24:24 -040043#include "variables.h"
Dees_Troy2673cec2013-04-02 20:22:16 +000044#include "twcommon.h"
Dees_Troy51a0e822012-09-05 15:24:24 -040045#include "partitions.hpp"
Dees_Troy5bf43922012-09-07 16:07:55 -040046#include "data.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040047#include "twrp-functions.hpp"
Ethan Yonkerb5fab762016-01-28 14:03:33 -060048#include "fixContexts.hpp"
Ethan Yonker3fdcda42016-11-30 12:29:37 -060049#include "exclude.hpp"
Ethan Yonker4b94cfd2014-12-11 10:00:45 -060050#include "set_metadata.h"
bigbiff7abc5fe2015-01-17 16:53:12 -050051#include "tw_atomic.hpp"
Ethan Yonker74db1572015-10-28 12:44:49 -050052#include "gui/gui.hpp"
Ethan Yonker472f5062016-02-25 13:47:30 -060053#include "progresstracking.hpp"
bigbiff bigbiff56cf5642016-08-19 17:43:45 -040054#include "twrpDigestDriver.hpp"
bigbiffce8f83c2015-12-12 18:30:21 -050055#include "adbbu/libtwadbbu.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040056
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040057#ifdef TW_HAS_MTP
bigbiff bigbiffaf32bb92018-12-18 18:39:53 -050058#ifdef TW_HAS_LEGACY_MTP
59#include "mtp/legacy/mtp_MtpServer.hpp"
60#include "mtp/legacy/twrpMtp.hpp"
61#include "mtp/legacy/MtpMessage.hpp"
62#else
63#include "mtp/ffs/mtp_MtpServer.hpp"
64#include "mtp/ffs/twrpMtp.hpp"
65#include "mtp/ffs/MtpMessage.hpp"
66#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040067#endif
68
Dees Troy6f6441d2014-01-23 02:07:03 +000069extern "C" {
70 #include "cutils/properties.h"
Ethan Yonker74db1572015-10-28 12:44:49 -050071 #include "gui/gui.h"
Dees Troy6f6441d2014-01-23 02:07:03 +000072}
73
Dees_Troy5bf43922012-09-07 16:07:55 -040074#ifdef TW_INCLUDE_CRYPTO
Ethan Yonker98661c12018-10-17 08:39:28 -050075 #include "crypto/fde/cryptfs.h"
Ethan Yonker66a19492015-12-10 10:19:45 -060076 #include "gui/rapidxml.hpp"
77 #include "gui/pages.hpp"
Ethan Yonkerbd7492d2016-12-07 13:55:01 -060078 #ifdef TW_INCLUDE_FBE
79 #include "crypto/ext4crypt/Decrypt.h"
Ethan Yonker93382822018-11-01 15:25:31 -050080 #ifdef TW_INCLUDE_FBE_METADATA_DECRYPT
81 #include "crypto/ext4crypt/MetadataCrypt.h"
82 #endif
Ethan Yonkerbd7492d2016-12-07 13:55:01 -060083 #endif
nkk7171c6c502017-01-05 23:55:05 +020084 #ifdef TW_CRYPTO_USE_SYSTEM_VOLD
85 #include "crypto/vold_decrypt/vold_decrypt.h"
86 #endif
Dees_Troy5bf43922012-09-07 16:07:55 -040087#endif
Dees_Troy51a0e822012-09-05 15:24:24 -040088
Ethan Yonker1b190162016-12-05 15:25:19 -060089#ifdef AB_OTA_UPDATER
90#include <hardware/hardware.h>
91#include <hardware/boot_control.h>
92#endif
93
Ethan Yonker6277c792014-09-15 14:54:30 -050094extern bool datamedia;
95
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050096TWPartitionManager::TWPartitionManager(void) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040097 mtp_was_enabled = false;
Ethan Yonker726a0202014-12-16 20:01:38 -060098 mtp_write_fd = -1;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060099 uevent_pfd.fd = -1;
bigbiff7abc5fe2015-01-17 16:53:12 -0500100 stop_backup.set_value(0);
Ethan Yonker1b190162016-12-05 15:25:19 -0600101#ifdef AB_OTA_UPDATER
102 char slot_suffix[PROPERTY_VALUE_MAX];
Ethan Yonkerfefe5912017-09-30 22:22:13 -0500103 property_get("ro.boot.slot_suffix", slot_suffix, "error");
104 if (strcmp(slot_suffix, "error") == 0)
105 property_get("ro.boot.slot", slot_suffix, "error");
Ethan Yonker1b190162016-12-05 15:25:19 -0600106 Active_Slot_Display = "";
Ethan Yonkerfefe5912017-09-30 22:22:13 -0500107 if (strcmp(slot_suffix, "_a") == 0 || strcmp(slot_suffix, "a") == 0)
Ethan Yonker1b190162016-12-05 15:25:19 -0600108 Set_Active_Slot("A");
109 else
110 Set_Active_Slot("B");
111#endif
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500112}
113
Dees_Troy51a0e822012-09-05 15:24:24 -0400114int TWPartitionManager::Process_Fstab(string Fstab_Filename, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400115 FILE *fstabFile;
116 char fstab_line[MAX_FSTAB_LINE_LENGTH];
Dees Troy02a64532014-03-19 15:23:32 +0000117 TWPartition* settings_partition = NULL;
Matt Mowerbf4efa32014-04-14 23:25:26 -0500118 TWPartition* andsec_partition = NULL;
Ethan Yonker726a0202014-12-16 20:01:38 -0600119 unsigned int storageid = 1 << 16; // upper 16 bits are for physical storage device, we pretend to have only one
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600120 std::map<string, Flags_Map> twrp_flags;
121
122 fstabFile = fopen("/etc/twrp.flags", "rt");
123 if (fstabFile != NULL) {
124 LOGINFO("reading /etc/twrp.flags\n");
125 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
126 if (fstab_line[0] != '/')
127 continue;
128
129 size_t line_size = strlen(fstab_line);
130 if (fstab_line[line_size - 1] != '\n')
131 fstab_line[line_size] = '\n';
132 Flags_Map line_flags;
133 line_flags.Primary_Block_Device = "";
134 line_flags.Alternate_Block_Device = "";
135 line_flags.fstab_line = (char*)malloc(MAX_FSTAB_LINE_LENGTH);
136 if (!line_flags.fstab_line) {
137 LOGERR("malloc error on line_flags.fstab_line\n");
138 return false;
139 }
140 memcpy(line_flags.fstab_line, fstab_line, MAX_FSTAB_LINE_LENGTH);
141 bool found_separator = false;
142 char *fs_loc = NULL;
143 char *block_loc = NULL;
144 char *flags_loc = NULL;
145 size_t index, item_index = 0;
146 for (index = 0; index < line_size; index++) {
147 if (fstab_line[index] <= 32) {
148 fstab_line[index] = '\0';
149 found_separator = true;
150 } else if (found_separator) {
151 if (item_index == 0) {
152 fs_loc = fstab_line + index;
153 } else if (item_index == 1) {
154 block_loc = fstab_line + index;
155 } else if (item_index > 1) {
156 char *ptr = fstab_line + index;
157 if (*ptr == '/') {
158 line_flags.Alternate_Block_Device = ptr;
159 } else if (strlen(ptr) > strlen("flags=") && strncmp(ptr, "flags=", strlen("flags=")) == 0) {
160 flags_loc = ptr;
161 // Once we find the flags=, we're done scanning the line
162 break;
163 }
164 }
165 found_separator = false;
166 item_index++;
167 }
168 }
169 if (block_loc)
170 line_flags.Primary_Block_Device = block_loc;
171 if (fs_loc)
172 line_flags.File_System = fs_loc;
173 if (flags_loc)
174 line_flags.Flags = flags_loc;
175 string Mount_Point = fstab_line;
176 twrp_flags[Mount_Point] = line_flags;
177 memset(fstab_line, 0, sizeof(fstab_line));
178 }
179 fclose(fstabFile);
180 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400181
182 fstabFile = fopen(Fstab_Filename.c_str(), "rt");
183 if (fstabFile == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000184 LOGERR("Critical Error: Unable to open fstab at '%s'.\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400185 return false;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600186 } else
187 LOGINFO("Reading %s\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400188
189 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
190 if (fstab_line[0] != '/')
191 continue;
192
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600193 if (strstr(fstab_line, "swap"))
194 continue; // Skip swap in recovery
195
196 size_t line_size = strlen(fstab_line);
197 if (fstab_line[line_size - 1] != '\n')
198 fstab_line[line_size] = '\n';
Dees_Troy2a923582012-09-20 12:13:34 -0400199
Matt Mower2b2dd152016-04-26 11:24:08 -0500200 TWPartition* partition = new TWPartition();
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600201 if (partition->Process_Fstab_Line(fstab_line, Display_Error, &twrp_flags))
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500202 Partitions.push_back(partition);
Matt Mower72c87ce2016-04-26 14:34:56 -0500203 else
Dees_Troy5bf43922012-09-07 16:07:55 -0400204 delete partition;
Matt Mower2b2dd152016-04-26 11:24:08 -0500205
206 memset(fstab_line, 0, sizeof(fstab_line));
Dees_Troy5bf43922012-09-07 16:07:55 -0400207 }
208 fclose(fstabFile);
Matt Mower72c87ce2016-04-26 14:34:56 -0500209
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600210 if (twrp_flags.size() > 0) {
211 LOGINFO("Processing remaining twrp.flags\n");
212 // Add any items from twrp.flags that did not exist in the recovery.fstab
213 for (std::map<string, Flags_Map>::iterator mapit=twrp_flags.begin(); mapit!=twrp_flags.end(); mapit++) {
214 if (Find_Partition_By_Path(mapit->first) == NULL) {
215 TWPartition* partition = new TWPartition();
216 if (partition->Process_Fstab_Line(mapit->second.fstab_line, Display_Error, NULL))
217 Partitions.push_back(partition);
218 else
219 delete partition;
220 }
221 if (mapit->second.fstab_line)
222 free(mapit->second.fstab_line);
223 mapit->second.fstab_line = NULL;
224 }
225 }
226 LOGINFO("Done processing fstab files\n");
227
Matt Mower72c87ce2016-04-26 14:34:56 -0500228 std::vector<TWPartition*>::iterator iter;
229 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
230 (*iter)->Partition_Post_Processing(Display_Error);
231
232 if ((*iter)->Is_Storage) {
233 ++storageid;
234 (*iter)->MTP_Storage_ID = storageid;
235 }
236
237 if (!settings_partition && (*iter)->Is_Settings_Storage && (*iter)->Is_Present)
238 settings_partition = (*iter);
239 else
240 (*iter)->Is_Settings_Storage = false;
241
242 if (!andsec_partition && (*iter)->Has_Android_Secure && (*iter)->Is_Present)
243 andsec_partition = (*iter);
244 else
245 (*iter)->Has_Android_Secure = false;
246 }
247
Ethan Yonker6277c792014-09-15 14:54:30 -0500248 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) {
249 // Attempt to automatically identify /data/media emulated storage devices
250 TWPartition* Dat = Find_Partition_By_Path("/data");
251 if (Dat) {
252 LOGINFO("Using automatic handling for /data/media emulated storage device.\n");
253 datamedia = true;
that9e0593e2014-10-08 00:01:24 +0200254 Dat->Setup_Data_Media();
Ethan Yonker6277c792014-09-15 14:54:30 -0500255 settings_partition = Dat;
Ethan Yonker726a0202014-12-16 20:01:38 -0600256 // Since /data was not considered a storage partition earlier, we still need to assign an MTP ID
257 ++storageid;
258 Dat->MTP_Storage_ID = storageid;
Ethan Yonker6277c792014-09-15 14:54:30 -0500259 }
260 }
Dees Troy02a64532014-03-19 15:23:32 +0000261 if (!settings_partition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500262 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
263 if ((*iter)->Is_Storage) {
Dees Troy02a64532014-03-19 15:23:32 +0000264 settings_partition = (*iter);
Dees_Troya13d74f2013-03-24 08:54:55 -0500265 break;
266 }
267 }
Dees Troy02a64532014-03-19 15:23:32 +0000268 if (!settings_partition)
Dees_Troy2673cec2013-04-02 20:22:16 +0000269 LOGERR("Unable to locate storage partition for storing settings file.\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500270 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400271 if (!Write_Fstab()) {
272 if (Display_Error)
Dees_Troy2673cec2013-04-02 20:22:16 +0000273 LOGERR("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400274 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000275 LOGINFO("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400276 }
Matt Mowered71fa32014-04-16 13:21:47 -0500277
Matt Mowerbf4efa32014-04-14 23:25:26 -0500278 if (andsec_partition) {
279 Setup_Android_Secure_Location(andsec_partition);
Matt Mowered71fa32014-04-16 13:21:47 -0500280 } else if (settings_partition) {
Matt Mowerbf4efa32014-04-14 23:25:26 -0500281 Setup_Android_Secure_Location(settings_partition);
282 }
Matt Mowered71fa32014-04-16 13:21:47 -0500283 if (settings_partition) {
284 Setup_Settings_Storage_Partition(settings_partition);
285 }
Ethan Yonker253368a2014-11-25 15:00:52 -0600286#ifdef TW_INCLUDE_CRYPTO
Ethan Yonkercceebb82014-11-18 10:17:59 -0600287 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
288 if (Decrypt_Data && Decrypt_Data->Is_Encrypted && !Decrypt_Data->Is_Decrypted) {
Ethan Yonker93382822018-11-01 15:25:31 -0500289 if (!Decrypt_Data->Key_Directory.empty() && Mount_By_Path(Decrypt_Data->Key_Directory, false)) {
290#ifdef TW_INCLUDE_FBE_METADATA_DECRYPT
291 if (e4crypt_mount_metadata_encrypted(Decrypt_Data->Mount_Point, false, Decrypt_Data->Key_Directory, Decrypt_Data->Actual_Block_Device, &Decrypt_Data->Decrypted_Block_Device)) {
292 LOGINFO("Successfully decrypted metadata encrypted data partition with new block device: '%s'\n", Decrypt_Data->Decrypted_Block_Device.c_str());
293 property_set("ro.crypto.state", "encrypted");
294 Decrypt_Data->Is_Decrypted = true; // Needed to make the mount function work correctly
295 int retry_count = 10;
296 while (!Decrypt_Data->Mount(false) && --retry_count)
297 usleep(500);
298 if (Decrypt_Data->Mount(false)) {
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400299 if (!Decrypt_Data->Decrypt_FBE_DE()) {
300 LOGINFO("Trying wrapped key.\n");
301 property_set("fbe.data.wrappedkey", "true");
302 if (!Decrypt_Data->Decrypt_FBE_DE()) {
303 LOGERR("Unable to decrypt FBE device\n");
304 }
305 }
306
Ethan Yonker93382822018-11-01 15:25:31 -0500307 } else {
308 LOGINFO("Failed to mount data after metadata decrypt\n");
309 }
310 } else {
311 LOGINFO("Unable to decrypt metadata encryption\n");
312 }
313#else
314 LOGERR("Metadata FBE decrypt support not present in this TWRP\n");
315#endif
316 }
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600317 if (Decrypt_Data->Is_FBE) {
318 if (DataManager::GetIntValue(TW_CRYPTO_PWTYPE) == 0) {
319 if (Decrypt_Device("!") == 0) {
320 gui_msg("decrypt_success=Successfully decrypted with default password.");
321 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
322 } else {
323 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
324 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600325 }
326 } else {
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600327 int password_type = cryptfs_get_password_type();
328 if (password_type == CRYPT_TYPE_DEFAULT) {
329 LOGINFO("Device is encrypted with the default password, attempting to decrypt.\n");
330 if (Decrypt_Device("default_password") == 0) {
331 gui_msg("decrypt_success=Successfully decrypted with default password.");
332 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
333 } else {
334 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
335 }
336 } else {
337 DataManager::SetValue("TW_CRYPTO_TYPE", password_type);
338 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600339 }
340 }
Ethan Yonker66a19492015-12-10 10:19:45 -0600341 if (Decrypt_Data && (!Decrypt_Data->Is_Encrypted || Decrypt_Data->Is_Decrypted) && Decrypt_Data->Mount(false)) {
342 Decrypt_Adopted();
343 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600344#endif
Dees_Troy51127312012-09-08 13:08:49 -0400345 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -0400346 UnMount_Main_Partitions();
Ethan Yonker1b190162016-12-05 15:25:19 -0600347#ifdef AB_OTA_UPDATER
348 DataManager::SetValue("tw_active_slot", Get_Active_Slot_Display());
349#endif
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600350 setup_uevent();
Dees_Troy5bf43922012-09-07 16:07:55 -0400351 return true;
352}
353
354int TWPartitionManager::Write_Fstab(void) {
355 FILE *fp;
356 std::vector<TWPartition*>::iterator iter;
357 string Line;
358
359 fp = fopen("/etc/fstab", "w");
360 if (fp == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000361 LOGINFO("Can not open /etc/fstab.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400362 return false;
363 }
Dees_Troy63c8df72012-09-10 14:02:05 -0400364 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy51127312012-09-08 13:08:49 -0400365 if ((*iter)->Can_Be_Mounted) {
dianlujitao7d304c72016-01-02 12:15:50 +0800366 Line = (*iter)->Actual_Block_Device + " " + (*iter)->Mount_Point + " " + (*iter)->Current_File_System + " rw 0 0\n";
Dees_Troy5bf43922012-09-07 16:07:55 -0400367 fputs(Line.c_str(), fp);
Dees_Troy91862e62013-04-04 23:48:21 +0000368 }
369 // Handle subpartition tracking
370 if ((*iter)->Is_SubPartition) {
371 TWPartition* ParentPartition = Find_Partition_By_Path((*iter)->SubPartition_Of);
372 if (ParentPartition)
373 ParentPartition->Has_SubPartition = true;
374 else
375 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 -0400376 }
377 }
378 fclose(fp);
379 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -0400380}
381
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500382void TWPartitionManager::Setup_Settings_Storage_Partition(TWPartition* Part) {
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500383 DataManager::SetValue("tw_settings_path", Part->Storage_Path);
384 DataManager::SetValue("tw_storage_path", Part->Storage_Path);
385 LOGINFO("Settings storage is '%s'\n", Part->Storage_Path.c_str());
386}
387
Matt Mowerbf4efa32014-04-14 23:25:26 -0500388void TWPartitionManager::Setup_Android_Secure_Location(TWPartition* Part) {
389 if (Part->Has_Android_Secure)
390 Part->Setup_AndSec();
Ethan Yonker6277c792014-09-15 14:54:30 -0500391 else if (!datamedia)
Matt Mowerbf4efa32014-04-14 23:25:26 -0500392 Part->Setup_AndSec();
Matt Mowerbf4efa32014-04-14 23:25:26 -0500393}
394
Dees_Troy8170a922012-09-18 15:40:25 -0400395void TWPartitionManager::Output_Partition_Logging(void) {
396 std::vector<TWPartition*>::iterator iter;
397
398 printf("\n\nPartition Logs:\n");
399 for (iter = Partitions.begin(); iter != Partitions.end(); iter++)
400 Output_Partition((*iter));
401}
402
403void TWPartitionManager::Output_Partition(TWPartition* Part) {
404 unsigned long long mb = 1048576;
405
Gary Peck004d48b2012-11-21 16:28:18 -0800406 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 -0400407 if (Part->Can_Be_Mounted) {
Gary Peck004d48b2012-11-21 16:28:18 -0800408 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 -0400409 }
Gary Peck004d48b2012-11-21 16:28:18 -0800410 printf("\n Flags: ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500411 if (Part->Can_Be_Mounted)
412 printf("Can_Be_Mounted ");
Gary Peck004d48b2012-11-21 16:28:18 -0800413 if (Part->Can_Be_Wiped)
414 printf("Can_Be_Wiped ");
Hashcodedabfd492013-08-29 22:45:30 -0700415 if (Part->Use_Rm_Rf)
416 printf("Use_Rm_Rf ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500417 if (Part->Can_Be_Backed_Up)
418 printf("Can_Be_Backed_Up ");
Gary Peck004d48b2012-11-21 16:28:18 -0800419 if (Part->Wipe_During_Factory_Reset)
420 printf("Wipe_During_Factory_Reset ");
421 if (Part->Wipe_Available_in_GUI)
422 printf("Wipe_Available_in_GUI ");
423 if (Part->Is_SubPartition)
424 printf("Is_SubPartition ");
425 if (Part->Has_SubPartition)
426 printf("Has_SubPartition ");
427 if (Part->Removable)
428 printf("Removable ");
429 if (Part->Is_Present)
430 printf("IsPresent ");
431 if (Part->Can_Be_Encrypted)
432 printf("Can_Be_Encrypted ");
433 if (Part->Is_Encrypted)
434 printf("Is_Encrypted ");
435 if (Part->Is_Decrypted)
436 printf("Is_Decrypted ");
437 if (Part->Has_Data_Media)
438 printf("Has_Data_Media ");
Dees_Troy83bd4832013-05-04 12:39:56 +0000439 if (Part->Can_Encrypt_Backup)
440 printf("Can_Encrypt_Backup ");
441 if (Part->Use_Userdata_Encryption)
442 printf("Use_Userdata_Encryption ");
Gary Peck004d48b2012-11-21 16:28:18 -0800443 if (Part->Has_Android_Secure)
444 printf("Has_Android_Secure ");
445 if (Part->Is_Storage)
446 printf("Is_Storage ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500447 if (Part->Is_Settings_Storage)
448 printf("Is_Settings_Storage ");
Dees_Troy68cab492012-12-12 19:29:35 +0000449 if (Part->Ignore_Blkid)
450 printf("Ignore_Blkid ");
Dees_Troy16c2b312013-01-15 16:51:18 +0000451 if (Part->Retain_Layout_Version)
452 printf("Retain_Layout_Version ");
Ethan Yonker253368a2014-11-25 15:00:52 -0600453 if (Part->Mount_To_Decrypt)
454 printf("Mount_To_Decrypt ");
Ethan Yonker96af84a2015-01-05 14:58:36 -0600455 if (Part->Can_Flash_Img)
456 printf("Can_Flash_Img ");
Ethan Yonker66a19492015-12-10 10:19:45 -0600457 if (Part->Is_Adopted_Storage)
458 printf("Is_Adopted_Storage ");
Ethan Yonker1b190162016-12-05 15:25:19 -0600459 if (Part->SlotSelect)
460 printf("SlotSelect ");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600461 if (Part->Mount_Read_Only)
462 printf("Mount_Read_Only ");
Gary Peck004d48b2012-11-21 16:28:18 -0800463 printf("\n");
464 if (!Part->SubPartition_Of.empty())
465 printf(" SubPartition_Of: %s\n", Part->SubPartition_Of.c_str());
466 if (!Part->Symlink_Path.empty())
467 printf(" Symlink_Path: %s\n", Part->Symlink_Path.c_str());
468 if (!Part->Symlink_Mount_Point.empty())
469 printf(" Symlink_Mount_Point: %s\n", Part->Symlink_Mount_Point.c_str());
470 if (!Part->Primary_Block_Device.empty())
471 printf(" Primary_Block_Device: %s\n", Part->Primary_Block_Device.c_str());
472 if (!Part->Alternate_Block_Device.empty())
473 printf(" Alternate_Block_Device: %s\n", Part->Alternate_Block_Device.c_str());
474 if (!Part->Decrypted_Block_Device.empty())
475 printf(" Decrypted_Block_Device: %s\n", Part->Decrypted_Block_Device.c_str());
dianlujitao4879b372018-12-03 18:45:47 +0800476 if (!Part->Crypto_Key_Location.empty())
Ethan Yonker253368a2014-11-25 15:00:52 -0600477 printf(" Crypto_Key_Location: %s\n", Part->Crypto_Key_Location.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800478 if (Part->Length != 0)
479 printf(" Length: %i\n", Part->Length);
480 if (!Part->Display_Name.empty())
481 printf(" Display_Name: %s\n", Part->Display_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500482 if (!Part->Storage_Name.empty())
483 printf(" Storage_Name: %s\n", Part->Storage_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800484 if (!Part->Backup_Path.empty())
485 printf(" Backup_Path: %s\n", Part->Backup_Path.c_str());
486 if (!Part->Backup_Name.empty())
487 printf(" Backup_Name: %s\n", Part->Backup_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500488 if (!Part->Backup_Display_Name.empty())
489 printf(" Backup_Display_Name: %s\n", Part->Backup_Display_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800490 if (!Part->Backup_FileName.empty())
491 printf(" Backup_FileName: %s\n", Part->Backup_FileName.c_str());
492 if (!Part->Storage_Path.empty())
493 printf(" Storage_Path: %s\n", Part->Storage_Path.c_str());
494 if (!Part->Current_File_System.empty())
495 printf(" Current_File_System: %s\n", Part->Current_File_System.c_str());
496 if (!Part->Fstab_File_System.empty())
497 printf(" Fstab_File_System: %s\n", Part->Fstab_File_System.c_str());
498 if (Part->Format_Block_Size != 0)
Ethan Yonkerc798c9c2015-10-09 11:15:26 -0500499 printf(" Format_Block_Size: %lu\n", Part->Format_Block_Size);
Dees_Troy094207a2012-09-26 12:00:39 -0400500 if (!Part->MTD_Name.empty())
501 printf(" MTD_Name: %s\n", Part->MTD_Name.c_str());
Matt Mower029a82d2017-01-08 13:12:38 -0600502 printf(" Backup_Method: %s\n", Part->Backup_Method_By_Name().c_str());
Hashcode62bd9e02013-11-19 21:59:42 -0800503 if (Part->Mount_Flags || !Part->Mount_Options.empty())
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600504 printf(" Mount_Flags: %i, Mount_Options: %s\n", Part->Mount_Flags, Part->Mount_Options.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -0600505 if (Part->MTP_Storage_ID)
506 printf(" MTP_Storage_ID: %i\n", Part->MTP_Storage_ID);
Ethan Yonker93382822018-11-01 15:25:31 -0500507 if (!Part->Key_Directory.empty())
508 printf(" Metadata Key Directory: %s\n", Part->Key_Directory.c_str());
Ethan Yonker6277c792014-09-15 14:54:30 -0500509 printf("\n");
Dees_Troy8170a922012-09-18 15:40:25 -0400510}
511
Dees_Troy51a0e822012-09-05 15:24:24 -0400512int TWPartitionManager::Mount_By_Path(string Path, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400513 std::vector<TWPartition*>::iterator iter;
514 int ret = false;
515 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400516 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400517
Captain Throwbackd54e1da2020-03-04 16:49:51 -0500518 if (Local_Path == "/tmp" || Local_Path == "/" || Local_Path == "/etc")
Dees_Troy43d8b002012-09-17 16:00:01 -0400519 return true;
520
Dees_Troy5bf43922012-09-07 16:07:55 -0400521 // Iterate through all partitions
522 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400523 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400524 ret = (*iter)->Mount(Display_Error);
525 found = true;
Dees_Troy51127312012-09-08 13:08:49 -0400526 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400527 (*iter)->Mount(Display_Error);
Dees_Troy51127312012-09-08 13:08:49 -0400528 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400529 }
530 if (found) {
531 return ret;
532 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500533 gui_msg(Msg(msg::kError, "unable_find_part_path=Unable to find partition for path '{1}'")(Local_Path));
Dees_Troy5bf43922012-09-07 16:07:55 -0400534 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000535 LOGINFO("Mount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400536 }
537 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400538}
539
Dees_Troy51a0e822012-09-05 15:24:24 -0400540int TWPartitionManager::UnMount_By_Path(string Path, bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400541 std::vector<TWPartition*>::iterator iter;
542 int ret = false;
543 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400544 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy51127312012-09-08 13:08:49 -0400545
546 // Iterate through all partitions
547 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400548 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy51127312012-09-08 13:08:49 -0400549 ret = (*iter)->UnMount(Display_Error);
550 found = true;
551 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
552 (*iter)->UnMount(Display_Error);
553 }
554 }
555 if (found) {
556 return ret;
557 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500558 gui_msg(Msg(msg::kError, "unable_find_part_path=Unable to find partition for path '{1}'")(Local_Path));
Dees_Troy51127312012-09-08 13:08:49 -0400559 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000560 LOGINFO("UnMount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400561 }
562 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400563}
564
Dees_Troy51a0e822012-09-05 15:24:24 -0400565int TWPartitionManager::Is_Mounted_By_Path(string Path) {
Dees_Troy51127312012-09-08 13:08:49 -0400566 TWPartition* Part = Find_Partition_By_Path(Path);
567
568 if (Part)
569 return Part->Is_Mounted();
570 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000571 LOGINFO("Is_Mounted: Unable to find partition for path '%s'\n", Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400572 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400573}
574
Dees_Troy5bf43922012-09-07 16:07:55 -0400575int TWPartitionManager::Mount_Current_Storage(bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400576 string current_storage_path = DataManager::GetCurrentStoragePath();
577
578 if (Mount_By_Path(current_storage_path, Display_Error)) {
579 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
580 if (FreeStorage)
581 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
582 return true;
583 }
584 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400585}
586
Dees_Troy5bf43922012-09-07 16:07:55 -0400587int TWPartitionManager::Mount_Settings_Storage(bool Display_Error) {
588 return Mount_By_Path(DataManager::GetSettingsStoragePath(), Display_Error);
589}
590
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600591TWPartition* TWPartitionManager::Find_Partition_By_Path(const string& Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400592 std::vector<TWPartition*>::iterator iter;
Dees_Troy38bd7602012-09-14 13:33:53 -0400593 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400594
595 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400596 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path))
Dees_Troy5bf43922012-09-07 16:07:55 -0400597 return (*iter);
598 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400599 return NULL;
600}
601
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600602TWPartition* TWPartitionManager::Find_Partition_By_Block_Device(const string& Block_Device) {
603 std::vector<TWPartition*>::iterator iter;
604
605 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
606 if ((*iter)->Primary_Block_Device == Block_Device || (!(*iter)->Actual_Block_Device.empty() && (*iter)->Actual_Block_Device == Block_Device))
607 return (*iter);
608 }
609 return NULL;
610}
611
Ethan Yonker53796e72019-01-11 22:49:52 -0600612int TWPartitionManager::Check_Backup_Name(const std::string& Backup_Name, bool Display_Error, bool Must_Be_Unique) {
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400613 // Check the backup name to ensure that it is the correct size and contains only valid characters
614 // and that a backup with that name doesn't already exist
615 char backup_name[MAX_BACKUP_NAME_LEN];
616 char backup_loc[255], tw_image_dir[255];
617 int copy_size;
618 int index, cur_char;
Ethan Yonker53796e72019-01-11 22:49:52 -0600619 string Backup_Loc;
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400620
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400621 copy_size = Backup_Name.size();
622 // Check size
623 if (copy_size > MAX_BACKUP_NAME_LEN) {
624 if (Display_Error)
Ethan Yonker74db1572015-10-28 12:44:49 -0500625 gui_err("backup_name_len=Backup name is too long.");
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400626 return -2;
627 }
628
629 // Check each character
630 strncpy(backup_name, Backup_Name.c_str(), copy_size);
Dees_Troya13d74f2013-03-24 08:54:55 -0500631 if (copy_size == 1 && strncmp(backup_name, "0", 1) == 0)
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400632 return 0; // A "0" (zero) means to use the current timestamp for the backup name
633 for (index=0; index<copy_size; index++) {
634 cur_char = (int)backup_name[index];
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500635 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 -0400636 // These are valid characters
637 // Numbers
638 // Upper case letters
639 // Lower case letters
640 // Space
641 // and -_.{}[]
642 } else {
643 if (Display_Error)
Ethan Yonker4adc33e2016-01-22 16:07:11 -0600644 gui_msg(Msg(msg::kError, "backup_name_invalid=Backup name '{1}' contains invalid character: '{1}'")(Backup_Name)((char)cur_char));
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400645 return -3;
646 }
647 }
648
Ethan Yonker53796e72019-01-11 22:49:52 -0600649 if (Must_Be_Unique) {
650 // Check to make sure that a backup with this name doesn't already exist
651 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Loc);
652 strcpy(backup_loc, Backup_Loc.c_str());
653 sprintf(tw_image_dir,"%s/%s", backup_loc, Backup_Name.c_str());
654 if (TWFunc::Path_Exists(tw_image_dir)) {
655 if (Display_Error)
656 gui_err("backup_name_exists=A backup with that name already exists!");
bigbiffce8f83c2015-12-12 18:30:21 -0500657
Ethan Yonker53796e72019-01-11 22:49:52 -0600658 return -4;
659 }
660 // Backup is unique
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400661 }
Ethan Yonker53796e72019-01-11 22:49:52 -0600662 // No problems found
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400663 return 0;
664}
665
bigbiffce8f83c2015-12-12 18:30:21 -0500666bool TWPartitionManager::Backup_Partition(PartitionSettings *part_settings) {
Dees_Troy43d8b002012-09-17 16:00:01 -0400667 time_t start, stop;
Matt Mower23d8aae2017-01-06 14:30:33 -0600668 int use_compression;
bigbiffce8f83c2015-12-12 18:30:21 -0500669 string backup_log = part_settings->Backup_Folder + "/recovery.log";
Dees_Troy43d8b002012-09-17 16:00:01 -0400670
bigbiffce8f83c2015-12-12 18:30:21 -0500671 if (part_settings->Part == NULL)
Dees_Troy43d8b002012-09-17 16:00:01 -0400672 return true;
673
Dees_Troy093b7642012-09-21 15:59:38 -0400674 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400675
Tom Hite5a926722014-09-15 01:31:03 +0000676 TWFunc::SetPerformanceMode(true);
Dees_Troy43d8b002012-09-17 16:00:01 -0400677 time(&start);
678
bigbiffce8f83c2015-12-12 18:30:21 -0500679 if (part_settings->Part->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500680 sync();
681 sync();
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400682 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500683 if (!part_settings->adbbackup && part_settings->generate_digest) {
684 if (!twrpDigestDriver::Make_Digest(Full_Filename))
685 goto backup_error;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400686 }
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400687
bigbiffce8f83c2015-12-12 18:30:21 -0500688 if (part_settings->Part->Has_SubPartition) {
Dees_Troy8170a922012-09-18 15:40:25 -0400689 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400690 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -0400691
692 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400693 if ((*subpart)->Can_Be_Backed_Up && (*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500694 part_settings->Part = *subpart;
bigbiffce8f83c2015-12-12 18:30:21 -0500695 if (!(*subpart)->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500696 goto backup_error;
Tom Hite5a926722014-09-15 01:31:03 +0000697 }
Dees_Troy2727b992013-08-14 20:09:30 +0000698 sync();
699 sync();
bigbiff bigbifff5955b12019-05-18 17:28:58 -0400700 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400701 if (!part_settings->adbbackup && part_settings->generate_digest) {
702 if (!twrpDigestDriver::Make_Digest(Full_Filename)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500703 goto backup_error;
bigbiffce8f83c2015-12-12 18:30:21 -0500704 }
Tom Hite5a926722014-09-15 01:31:03 +0000705 }
Dees_Troy8170a922012-09-18 15:40:25 -0400706 }
707 }
708 }
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400709
Dees_Troy43d8b002012-09-17 16:00:01 -0400710 time(&stop);
that203bcc92015-05-09 17:27:33 +0200711 int backup_time = (int) difftime(stop, start);
Dees_Troy2673cec2013-04-02 20:22:16 +0000712 LOGINFO("Partition Backup time: %d\n", backup_time);
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400713
bigbiffce8f83c2015-12-12 18:30:21 -0500714 if (part_settings->Part->Backup_Method == BM_FILES) {
715 part_settings->file_time += backup_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400716 } else {
bigbiffce8f83c2015-12-12 18:30:21 -0500717 part_settings->img_time += backup_time;
718
Dees_Troy43d8b002012-09-17 16:00:01 -0400719 }
Tom Hite5a926722014-09-15 01:31:03 +0000720
Matt Mower02899552016-12-30 18:13:51 -0600721 TWFunc::SetPerformanceMode(false);
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500722 return true;
Dees_Troy43d8b002012-09-17 16:00:01 -0400723 }
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500724backup_error:
725 Clean_Backup_Folder(part_settings->Backup_Folder);
726 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
727 tw_set_default_metadata(backup_log.c_str());
728 TWFunc::SetPerformanceMode(false);
729 return false;
bigbiff7abc5fe2015-01-17 16:53:12 -0500730}
731
bigbiffbf1d6722015-02-14 16:25:59 -0500732void TWPartitionManager::Clean_Backup_Folder(string Backup_Folder) {
733 DIR *d = opendir(Backup_Folder.c_str());
734 struct dirent *p;
735 int r;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400736 vector<string> ext;
737
738 //extensions we should delete when cleaning
739 ext.push_back("win");
740 ext.push_back("md5");
741 ext.push_back("sha2");
742 ext.push_back("info");
bigbiffbf1d6722015-02-14 16:25:59 -0500743
Ethan Yonker74db1572015-10-28 12:44:49 -0500744 gui_msg("backup_clean=Backup Failed. Cleaning Backup Folder.");
bigbiffbf1d6722015-02-14 16:25:59 -0500745
746 if (d == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500747 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Backup_Folder)(strerror(errno)));
bigbiffbf1d6722015-02-14 16:25:59 -0500748 return;
749 }
750
Matt Mower2b18a532015-02-20 16:58:05 -0600751 while ((p = readdir(d))) {
bigbiffbf1d6722015-02-14 16:25:59 -0500752 if (!strcmp(p->d_name, ".") || !strcmp(p->d_name, ".."))
753 continue;
754
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500755 string path = Backup_Folder + "/" + p->d_name;
bigbiffbf1d6722015-02-14 16:25:59 -0500756
757 size_t dot = path.find_last_of(".") + 1;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400758 for (vector<string>::const_iterator i = ext.begin(); i != ext.end(); ++i) {
759 if (path.substr(dot) == *i) {
760 r = unlink(path.c_str());
761 if (r != 0)
762 LOGINFO("Unable to unlink '%s: %s'\n", path.c_str(), strerror(errno));
bigbiffbf1d6722015-02-14 16:25:59 -0500763 }
764 }
765 }
766 closedir(d);
767}
768
Ethan Yonker472f5062016-02-25 13:47:30 -0600769int TWPartitionManager::Check_Backup_Cancel() {
770 return stop_backup.get_value();
771}
772
bigbiff7abc5fe2015-01-17 16:53:12 -0500773int TWPartitionManager::Cancel_Backup() {
774 string Backup_Folder, Backup_Name, Full_Backup_Path;
775
776 stop_backup.set_value(1);
777
778 if (tar_fork_pid != 0) {
779 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
780 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Folder);
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500781 Full_Backup_Path = Backup_Folder + "/" + Backup_Name;
bigbiff7abc5fe2015-01-17 16:53:12 -0500782 LOGINFO("Killing pid: %d\n", tar_fork_pid);
783 kill(tar_fork_pid, SIGUSR2);
784 while (kill(tar_fork_pid, 0) == 0) {
785 usleep(1000);
786 }
787 LOGINFO("Backup_Run stopped and returning false, backup cancelled.\n");
788 LOGINFO("Removing directory %s\n", Full_Backup_Path.c_str());
789 TWFunc::removeDir(Full_Backup_Path, false);
790 tar_fork_pid = 0;
791 }
792
793 return 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400794}
795
bigbiffce8f83c2015-12-12 18:30:21 -0500796int TWPartitionManager::Run_Backup(bool adbbackup) {
797 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400798 int partition_count = 0, disable_free_space_check = 0, skip_digest = 0;
bigbiffce8f83c2015-12-12 18:30:21 -0500799 string Backup_Name, Backup_List, backup_path;
Matt Mower23d8aae2017-01-06 14:30:33 -0600800 unsigned long long total_bytes = 0, free_space = 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400801 TWPartition* storage = NULL;
Dees_Troy8170a922012-09-18 15:40:25 -0400802 std::vector<TWPartition*>::iterator subpart;
Dees_Troy43d8b002012-09-17 16:00:01 -0400803 struct tm *t;
Matt Mower23d8aae2017-01-06 14:30:33 -0600804 time_t seconds, total_start, total_stop;
Dees_Troya13d74f2013-03-24 08:54:55 -0500805 size_t start_pos = 0, end_pos = 0;
bigbiff7abc5fe2015-01-17 16:53:12 -0500806 stop_backup.set_value(0);
Dees_Troy43d8b002012-09-17 16:00:01 -0400807 seconds = time(0);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500808 t = localtime(&seconds);
Dees_Troy43d8b002012-09-17 16:00:01 -0400809
bigbiffce8f83c2015-12-12 18:30:21 -0500810 part_settings.img_bytes_remaining = 0;
811 part_settings.file_bytes_remaining = 0;
812 part_settings.img_time = 0;
813 part_settings.file_time = 0;
814 part_settings.img_bytes = 0;
815 part_settings.file_bytes = 0;
816 part_settings.PM_Method = PM_BACKUP;
817
bigbiffce8f83c2015-12-12 18:30:21 -0500818 part_settings.adbbackup = adbbackup;
Dees_Troy43d8b002012-09-17 16:00:01 -0400819 time(&total_start);
820
821 Update_System_Details();
822
823 if (!Mount_Current_Storage(true))
824 return false;
825
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400826 DataManager::GetValue(TW_SKIP_DIGEST_GENERATE_VAR, skip_digest);
827 if (skip_digest == 0)
828 part_settings.generate_digest = true;
Dees_Troyc5865ab2012-09-24 15:08:04 -0400829 else
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400830 part_settings.generate_digest = false;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400831
bigbiffce8f83c2015-12-12 18:30:21 -0500832 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, part_settings.Backup_Folder);
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500833 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
834 if (Backup_Name == gui_lookup("curr_date", "(Current Date)")) {
835 Backup_Name = TWFunc::Get_Current_Date();
836 } else if (Backup_Name == gui_lookup("auto_generate", "(Auto Generate)") || Backup_Name == "0" || Backup_Name.empty()) {
Dees Troyb21cc642013-09-10 17:36:41 +0000837 TWFunc::Auto_Generate_Backup_Name();
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500838 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
Dees_Troy43d8b002012-09-17 16:00:01 -0400839 }
bigbiffce8f83c2015-12-12 18:30:21 -0500840
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500841 LOGINFO("Backup Name is: '%s'\n", Backup_Name.c_str());
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500842 part_settings.Backup_Folder = part_settings.Backup_Folder + "/" + Backup_Name;
bigbiffce8f83c2015-12-12 18:30:21 -0500843
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500844 LOGINFO("Backup_Folder is: '%s'\n", part_settings.Backup_Folder.c_str());
Dees_Troy43d8b002012-09-17 16:00:01 -0400845
Dees_Troy2673cec2013-04-02 20:22:16 +0000846 LOGINFO("Calculating backup details...\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500847 DataManager::GetValue("tw_backup_list", Backup_List);
848 if (!Backup_List.empty()) {
849 end_pos = Backup_List.find(";", start_pos);
850 while (end_pos != string::npos && start_pos < Backup_List.size()) {
851 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -0500852 part_settings.Part = Find_Partition_By_Path(backup_path);
853 if (part_settings.Part != NULL) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500854 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500855 if (part_settings.Part->Backup_Method == BM_FILES)
856 part_settings.file_bytes += part_settings.Part->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500857 else
bigbiffce8f83c2015-12-12 18:30:21 -0500858 part_settings.img_bytes += part_settings.Part->Backup_Size;
859 if (part_settings.Part->Has_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500860 std::vector<TWPartition*>::iterator subpart;
861
862 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiffce8f83c2015-12-12 18:30:21 -0500863 if ((*subpart)->Can_Be_Backed_Up && (*subpart)->Is_Present && (*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == part_settings.Part->Mount_Point) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500864 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500865 if ((*subpart)->Backup_Method == BM_FILES)
866 part_settings.file_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500867 else
bigbiffce8f83c2015-12-12 18:30:21 -0500868 part_settings.img_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500869 }
870 }
Dees_Troy8170a922012-09-18 15:40:25 -0400871 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500872 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500873 gui_msg(Msg(msg::kError, "unable_to_locate_partition=Unable to locate '{1}' partition for backup calculations.")(backup_path));
Dees_Troy8170a922012-09-18 15:40:25 -0400874 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500875 start_pos = end_pos + 1;
876 end_pos = Backup_List.find(";", start_pos);
Dees_Troy43d8b002012-09-17 16:00:01 -0400877 }
878 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400879
880 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500881 gui_msg("no_partition_selected=No partitions selected for backup.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400882 return false;
883 }
bigbiffce8f83c2015-12-12 18:30:21 -0500884 if (adbbackup) {
885 if (twadbbu::Write_ADB_Stream_Header(partition_count) == false) {
886 return false;
887 }
888 }
889 total_bytes = part_settings.file_bytes + part_settings.img_bytes;
Ethan Yonker472f5062016-02-25 13:47:30 -0600890 ProgressTracking progress(total_bytes);
bigbiffce8f83c2015-12-12 18:30:21 -0500891 part_settings.progress = &progress;
892
Ethan Yonker74db1572015-10-28 12:44:49 -0500893 gui_msg(Msg("total_partitions_backup= * Total number of partitions to back up: {1}")(partition_count));
894 gui_msg(Msg("total_backup_size= * Total size of all data: {1}MB")(total_bytes / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400895 storage = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
896 if (storage != NULL) {
897 free_space = storage->Free;
Ethan Yonker74db1572015-10-28 12:44:49 -0500898 gui_msg(Msg("available_space= * Available space: {1}MB")(free_space / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400899 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500900 gui_err("unable_locate_storage=Unable to locate storage device.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400901 return false;
902 }
bigbiffbf1d6722015-02-14 16:25:59 -0500903
Matt Mowerbfccfb82016-04-25 23:22:31 -0500904 DataManager::GetValue(TW_DISABLE_FREE_SPACE_VAR, disable_free_space_check);
bigbiffce8f83c2015-12-12 18:30:21 -0500905
906 if (adbbackup)
907 disable_free_space_check = true;
908
bigbiffbf1d6722015-02-14 16:25:59 -0500909 if (!disable_free_space_check) {
910 if (free_space - (32 * 1024 * 1024) < total_bytes) {
911 // We require an extra 32MB just in case
Ethan Yonker74db1572015-10-28 12:44:49 -0500912 gui_err("no_space=Not enough free space on storage.");
bigbiffbf1d6722015-02-14 16:25:59 -0500913 return false;
914 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400915 }
bigbiffce8f83c2015-12-12 18:30:21 -0500916 part_settings.img_bytes_remaining = part_settings.img_bytes;
917 part_settings.file_bytes_remaining = part_settings.file_bytes;
Dees_Troy43d8b002012-09-17 16:00:01 -0400918
Ethan Yonker74db1572015-10-28 12:44:49 -0500919 gui_msg("backup_started=[BACKUP STARTED]");
bigbiff bigbiffcdd97c72019-03-29 19:12:33 -0400920
921 int is_decrypted = 0;
922 int is_encrypted = 0;
923
924 DataManager::GetValue(TW_IS_DECRYPTED, is_decrypted);
925 DataManager::GetValue(TW_IS_ENCRYPTED, is_encrypted);
926 if (!adbbackup || (!is_encrypted || (is_encrypted && is_decrypted))) {
927 gui_msg(Msg("backup_folder= * Backup Folder: {1}")(part_settings.Backup_Folder));
928 if (!TWFunc::Recursive_Mkdir(part_settings.Backup_Folder)) {
929 gui_err("fail_backup_folder=Failed to make backup folder.");
930 return false;
931 }
Dees_Troyd4b22b02013-01-18 17:17:58 +0000932 }
933
Dees_Troy2673cec2013-04-02 20:22:16 +0000934 DataManager::SetProgress(0.0);
Dees_Troy093b7642012-09-21 15:59:38 -0400935
Dees_Troya13d74f2013-03-24 08:54:55 -0500936 start_pos = 0;
937 end_pos = Backup_List.find(";", start_pos);
938 while (end_pos != string::npos && start_pos < Backup_List.size()) {
bigbiff7abc5fe2015-01-17 16:53:12 -0500939 if (stop_backup.get_value() != 0)
940 return -1;
Dees_Troya13d74f2013-03-24 08:54:55 -0500941 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -0500942 part_settings.Part = Find_Partition_By_Path(backup_path);
943 if (part_settings.Part != NULL) {
944 if (!Backup_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -0500945 return false;
946 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500947 gui_msg(Msg(msg::kError, "unable_to_locate_partition=Unable to locate '{1}' partition for backup calculations.")(backup_path));
Dees_Troya13d74f2013-03-24 08:54:55 -0500948 }
949 start_pos = end_pos + 1;
950 end_pos = Backup_List.find(";", start_pos);
951 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400952
953 // Average BPS
bigbiffce8f83c2015-12-12 18:30:21 -0500954 if (part_settings.img_time == 0)
955 part_settings.img_time = 1;
956 if (part_settings.file_time == 0)
957 part_settings.file_time = 1;
958 int img_bps = (int)part_settings.img_bytes / (int)part_settings.img_time;
959 unsigned long long file_bps = part_settings.file_bytes / (int)part_settings.file_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400960
bigbiffce8f83c2015-12-12 18:30:21 -0500961 if (part_settings.file_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -0600962 gui_msg(Msg("avg_backup_fs=Average backup rate for file systems: {1} MB/sec")(file_bps / (1024 * 1024)));
bigbiffce8f83c2015-12-12 18:30:21 -0500963 if (part_settings.img_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -0600964 gui_msg(Msg("avg_backup_img=Average backup rate for imaged drives: {1} MB/sec")(img_bps / (1024 * 1024)));
Dees_Troy43d8b002012-09-17 16:00:01 -0400965
966 time(&total_stop);
967 int total_time = (int) difftime(total_stop, total_start);
bigbiffce8f83c2015-12-12 18:30:21 -0500968
969 uint64_t actual_backup_size;
Ethan Yonker3fdcda42016-11-30 12:29:37 -0600970 if (!adbbackup) {
971 TWExclude twe;
972 actual_backup_size = twe.Get_Folder_Size(part_settings.Backup_Folder);
973 } else
bigbiffce8f83c2015-12-12 18:30:21 -0500974 actual_backup_size = part_settings.file_bytes + part_settings.img_bytes;
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500975 actual_backup_size /= (1024LLU * 1024LLU);
Dees_Troy43d8b002012-09-17 16:00:01 -0400976
bigbiffce8f83c2015-12-12 18:30:21 -0500977 int prev_img_bps = 0, use_compression = 0;
978 unsigned long long prev_file_bps = 0;
Matt Mower173cdb92016-12-31 02:49:19 -0600979 DataManager::GetValue(TW_BACKUP_AVG_IMG_RATE, prev_img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -0400980 img_bps += (prev_img_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500981 img_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -0400982
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500983 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400984 if (use_compression)
985 DataManager::GetValue(TW_BACKUP_AVG_FILE_COMP_RATE, prev_file_bps);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500986 else
Dees_Troy093b7642012-09-21 15:59:38 -0400987 DataManager::GetValue(TW_BACKUP_AVG_FILE_RATE, prev_file_bps);
988 file_bps += (prev_file_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500989 file_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -0400990
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500991 DataManager::SetValue(TW_BACKUP_AVG_IMG_RATE, img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -0400992 if (use_compression)
993 DataManager::SetValue(TW_BACKUP_AVG_FILE_COMP_RATE, file_bps);
994 else
995 DataManager::SetValue(TW_BACKUP_AVG_FILE_RATE, file_bps);
996
Ethan Yonker74db1572015-10-28 12:44:49 -0500997 gui_msg(Msg("total_backed_size=[{1} MB TOTAL BACKED UP]")(actual_backup_size));
Dees_Troy43d8b002012-09-17 16:00:01 -0400998 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -0400999 UnMount_Main_Partitions();
Ethan Yonker56db1c42015-12-20 22:49:00 -06001000 gui_msg(Msg(msg::kHighlight, "backup_completed=[BACKUP COMPLETED IN {1} SECONDS]")(total_time)); // the end
Ethan Yonkere080c1f2016-09-19 13:50:25 -05001001 string backup_log = part_settings.Backup_Folder + "/recovery.log";
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001002 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
Ethan Yonker4b94cfd2014-12-11 10:00:45 -06001003 tw_set_default_metadata(backup_log.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -05001004
1005 if (part_settings.adbbackup) {
1006 if (twadbbu::Write_ADB_Stream_Trailer() == false) {
1007 return false;
1008 }
1009 }
1010 part_settings.adbbackup = false;
1011 DataManager::SetValue("tw_enable_adb_backup", 0);
1012
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001013 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001014}
1015
bigbiffce8f83c2015-12-12 18:30:21 -05001016bool TWPartitionManager::Restore_Partition(PartitionSettings *part_settings) {
Dees_Troy4a2a1262012-09-18 09:33:47 -04001017 time_t Start, Stop;
bigbiffce8f83c2015-12-12 18:30:21 -05001018
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001019 if (part_settings->adbbackup) {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001020 std::string partName = part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win";
1021 LOGINFO("setting backup name: %s\n", partName.c_str());
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001022 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
1023 }
1024
Tom Hite5a926722014-09-15 01:31:03 +00001025 TWFunc::SetPerformanceMode(true);
bigbiffce8f83c2015-12-12 18:30:21 -05001026
Dees_Troy4a2a1262012-09-18 09:33:47 -04001027 time(&Start);
Ethan Yonker472f5062016-02-25 13:47:30 -06001028
bigbiffce8f83c2015-12-12 18:30:21 -05001029 if (!part_settings->Part->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001030 TWFunc::SetPerformanceMode(false);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001031 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001032 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001033 if (part_settings->Part->Has_SubPartition && !part_settings->adbbackup) {
Dees_Troy8170a922012-09-18 15:40:25 -04001034 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001035 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -04001036
1037 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001038 part_settings->Part = *subpart;
1039 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
1040 part_settings->Part = (*subpart);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001041 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
bigbiffce8f83c2015-12-12 18:30:21 -05001042 if (!(*subpart)->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001043 TWFunc::SetPerformanceMode(false);
Dees_Troy8170a922012-09-18 15:40:25 -04001044 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001045 }
Dees_Troy8170a922012-09-18 15:40:25 -04001046 }
1047 }
1048 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001049 time(&Stop);
Tom Hite5a926722014-09-15 01:31:03 +00001050 TWFunc::SetPerformanceMode(false);
bigbiffce8f83c2015-12-12 18:30:21 -05001051 gui_msg(Msg("restore_part_done=[{1} done ({2} seconds)]")(part_settings->Part->Backup_Display_Name)((int)difftime(Stop, Start)));
1052
Dees_Troy4a2a1262012-09-18 09:33:47 -04001053 return true;
1054}
1055
Ethan Yonker472f5062016-02-25 13:47:30 -06001056int TWPartitionManager::Run_Restore(const string& Restore_Name) {
bigbiffce8f83c2015-12-12 18:30:21 -05001057 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001058 int check_digest;
bigbiffce8f83c2015-12-12 18:30:21 -05001059
Dees_Troy4a2a1262012-09-18 09:33:47 -04001060 time_t rStart, rStop;
1061 time(&rStart);
Dees_Troya13d74f2013-03-24 08:54:55 -05001062 string Restore_List, restore_path;
1063 size_t start_pos = 0, end_pos;
bigbiffce8f83c2015-12-12 18:30:21 -05001064
Ethan Yonkerdcf2b672016-09-13 14:41:53 -05001065 part_settings.Backup_Folder = Restore_Name;
bigbiffce8f83c2015-12-12 18:30:21 -05001066 part_settings.Part = NULL;
1067 part_settings.partition_count = 0;
1068 part_settings.total_restore_size = 0;
1069 part_settings.adbbackup = false;
1070 part_settings.PM_Method = PM_RESTORE;
Dees_Troy43d8b002012-09-17 16:00:01 -04001071
Ethan Yonker74db1572015-10-28 12:44:49 -05001072 gui_msg("restore_started=[RESTORE STARTED]");
1073 gui_msg(Msg("restore_folder=Restore folder: '{1}'")(Restore_Name));
Dees_Troy43d8b002012-09-17 16:00:01 -04001074
Dees_Troy4a2a1262012-09-18 09:33:47 -04001075 if (!Mount_Current_Storage(true))
1076 return false;
1077
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001078 DataManager::GetValue(TW_SKIP_DIGEST_CHECK_VAR, check_digest);
1079 if (check_digest > 0) {
1080 // Check Digest files first before restoring to ensure that all of them match before starting a restore
1081 TWFunc::GUI_Operation_Text(TW_VERIFY_DIGEST_TEXT, gui_parse_text("{@verifying_digest}"));
1082 gui_msg("verifying_digest=Verifying Digest");
Dees_Troya13d74f2013-03-24 08:54:55 -05001083 } else {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001084 gui_msg("skip_digest=Skipping Digest check based on user setting.");
Dees_Troya13d74f2013-03-24 08:54:55 -05001085 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001086 gui_msg("calc_restore=Calculating restore details...");
Dees_Troya13d74f2013-03-24 08:54:55 -05001087 DataManager::GetValue("tw_restore_selected", Restore_List);
bigbiffce8f83c2015-12-12 18:30:21 -05001088
Dees_Troya13d74f2013-03-24 08:54:55 -05001089 if (!Restore_List.empty()) {
1090 end_pos = Restore_List.find(";", start_pos);
1091 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1092 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001093 part_settings.Part = Find_Partition_By_Path(restore_path);
1094 if (part_settings.Part != NULL) {
bigbiffce8f83c2015-12-12 18:30:21 -05001095 if (part_settings.Part->Mount_Read_Only) {
1096 gui_msg(Msg(msg::kError, "restore_read_only=Cannot restore {1} -- mounted read only.")(part_settings.Part->Backup_Display_Name));
Ethan Yonkereb32b1f2015-05-18 10:23:03 -05001097 return false;
1098 }
bigbiffce8f83c2015-12-12 18:30:21 -05001099
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001100 string Full_Filename = part_settings.Backup_Folder + "/" + part_settings.Part->Backup_FileName;
1101
1102 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001103 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001104 part_settings.partition_count++;
1105 part_settings.total_restore_size += part_settings.Part->Get_Restore_Size(&part_settings);
1106 if (part_settings.Part->Has_SubPartition) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001107 TWPartition *parentPart = part_settings.Part;
Dees_Troya13d74f2013-03-24 08:54:55 -05001108 std::vector<TWPartition*>::iterator subpart;
1109
1110 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001111 part_settings.Part = *subpart;
1112 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001113 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001114 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001115 part_settings.total_restore_size += (*subpart)->Get_Restore_Size(&part_settings);
Dees_Troya13d74f2013-03-24 08:54:55 -05001116 }
1117 }
1118 }
1119 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001120 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001121 }
1122 start_pos = end_pos + 1;
1123 end_pos = Restore_List.find(";", start_pos);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001124 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001125 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001126
bigbiffce8f83c2015-12-12 18:30:21 -05001127 if (part_settings.partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001128 gui_err("no_part_restore=No partitions selected for restore.");
Dees_Troy4a2a1262012-09-18 09:33:47 -04001129 return false;
1130 }
1131
bigbiffce8f83c2015-12-12 18:30:21 -05001132 gui_msg(Msg("restore_part_count=Restoring {1} partitions...")(part_settings.partition_count));
1133 gui_msg(Msg("total_restore_size=Total restore size is {1}MB")(part_settings.total_restore_size / 1048576));
Dees_Troy2673cec2013-04-02 20:22:16 +00001134 DataManager::SetProgress(0.0);
bigbiffce8f83c2015-12-12 18:30:21 -05001135 ProgressTracking progress(part_settings.total_restore_size);
1136 part_settings.progress = &progress;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001137
Dees_Troya13d74f2013-03-24 08:54:55 -05001138 start_pos = 0;
1139 if (!Restore_List.empty()) {
1140 end_pos = Restore_List.find(";", start_pos);
1141 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1142 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001143
1144 part_settings.Part = Find_Partition_By_Path(restore_path);
1145 if (part_settings.Part != NULL) {
1146 part_settings.partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -05001147 if (!Restore_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001148 return false;
1149 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001150 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001151 }
1152 start_pos = end_pos + 1;
1153 end_pos = Restore_List.find(";", start_pos);
1154 }
1155 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001156 TWFunc::GUI_Operation_Text(TW_UPDATE_SYSTEM_DETAILS_TEXT, gui_parse_text("{@updating_system_details}"));
Captain Throwback9d6feb52018-07-27 10:05:24 -04001157 UnMount_By_Path(Get_Android_Root_Path(), false);
Dees_Troy43d8b002012-09-17 16:00:01 -04001158 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001159 UnMount_Main_Partitions();
Dees_Troy4a2a1262012-09-18 09:33:47 -04001160 time(&rStop);
Matt Mower3c366972015-12-25 19:28:31 -06001161 gui_msg(Msg(msg::kHighlight, "restore_completed=[RESTORE COMPLETED IN {1} SECONDS]")((int)difftime(rStop,rStart)));
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001162 DataManager::SetValue("tw_file_progress", "");
bigbiffce8f83c2015-12-12 18:30:21 -05001163
Dees_Troy63c8df72012-09-10 14:02:05 -04001164 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001165}
1166
1167void TWPartitionManager::Set_Restore_Files(string Restore_Name) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001168 // Start with the default values
Dees_Troya13d74f2013-03-24 08:54:55 -05001169 string Restore_List;
Dees_Troy83bd4832013-05-04 12:39:56 +00001170 bool get_date = true, check_encryption = true;
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001171 bool adbbackup = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001172
1173 DataManager::SetValue("tw_restore_encrypted", 0);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001174 if (twadbbu::Check_ADB_Backup_File(Restore_Name)) {
1175 vector<string> adb_files;
1176 adb_files = twadbbu::Get_ADB_Backup_Files(Restore_Name);
1177 for (unsigned int i = 0; i < adb_files.size(); ++i) {
1178 string adb_restore_file = adb_files.at(i);
1179 std::size_t pos = adb_restore_file.find_first_of(".");
1180 std::string path = "/" + adb_restore_file.substr(0, pos);
1181 Restore_List = path + ";";
1182 TWPartition* Part = Find_Partition_By_Path(path);
1183 Part->Backup_FileName = TWFunc::Get_Filename(adb_restore_file);
1184 adbbackup = true;
1185 }
1186 DataManager::SetValue("tw_enable_adb_backup", 1);
Dees_Troy63c8df72012-09-10 14:02:05 -04001187 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001188 else {
1189 DIR* d;
1190 d = opendir(Restore_Name.c_str());
1191 if (d == NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001192 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001193 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Restore_Name)(strerror(errno)));
1194 return;
Dees_Troy63c8df72012-09-10 14:02:05 -04001195 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001196
1197 struct dirent* de;
1198 while ((de = readdir(d)) != NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001199 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001200 // Strip off three components
1201 char str[256];
1202 char* label;
1203 char* fstype = NULL;
1204 char* extn = NULL;
1205 char* ptr;
Dees_Troy63c8df72012-09-10 14:02:05 -04001206
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001207 strcpy(str, de->d_name);
1208 if (strlen(str) <= 2)
1209 continue;
Dees_Troy83bd4832013-05-04 12:39:56 +00001210
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001211 if (get_date) {
1212 char file_path[255];
1213 struct stat st;
1214
1215 strcpy(file_path, Restore_Name.c_str());
1216 strcat(file_path, "/");
1217 strcat(file_path, str);
1218 stat(file_path, &st);
1219 string backup_date = ctime((const time_t*)(&st.st_mtime));
1220 DataManager::SetValue(TW_RESTORE_FILE_DATE, backup_date);
1221 get_date = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001222 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001223
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001224 label = str;
1225 ptr = label;
1226 while (*ptr && *ptr != '.') ptr++;
1227 if (*ptr == '.')
1228 {
1229 *ptr = 0x00;
1230 ptr++;
1231 fstype = ptr;
1232 }
1233 while (*ptr && *ptr != '.') ptr++;
1234 if (*ptr == '.')
1235 {
1236 *ptr = 0x00;
1237 ptr++;
1238 extn = ptr;
1239 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001240
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001241 if (fstype == NULL || extn == NULL || strcmp(fstype, "log") == 0) continue;
1242 int extnlength = strlen(extn);
1243 if (extnlength != 3 && extnlength != 6) continue;
1244 if (extnlength >= 3 && strncmp(extn, "win", 3) != 0) continue;
1245 //if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
Dees_Troy63c8df72012-09-10 14:02:05 -04001246
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001247 if (check_encryption) {
1248 string filename = Restore_Name + "/";
1249 filename += de->d_name;
1250 if (TWFunc::Get_File_Type(filename) == 2) {
1251 LOGINFO("'%s' is encrypted\n", filename.c_str());
1252 DataManager::SetValue("tw_restore_encrypted", 1);
1253 }
1254 }
1255 if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
1256
1257 TWPartition* Part = Find_Partition_By_Path(label);
1258 if (Part == NULL)
1259 {
1260 gui_msg(Msg(msg::kError, "unable_locate_part_backup_name=Unable to locate partition by backup name: '{1}'")(label));
1261 continue;
1262 }
1263
1264 Part->Backup_FileName = de->d_name;
1265 if (strlen(extn) > 3) {
1266 Part->Backup_FileName.resize(Part->Backup_FileName.size() - strlen(extn) + 3);
1267 }
1268
1269 if (!Part->Is_SubPartition)
1270 Restore_List += Part->Backup_Path + ";";
1271 }
1272 closedir(d);
Dees_Troy63c8df72012-09-10 14:02:05 -04001273 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001274
1275 if (adbbackup) {
1276 Restore_List = "ADB_Backup;";
1277 adbbackup = false;
1278 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001279
Dees_Troya13d74f2013-03-24 08:54:55 -05001280 // Set the final value
1281 DataManager::SetValue("tw_restore_list", Restore_List);
1282 DataManager::SetValue("tw_restore_selected", Restore_List);
Dees_Troy51a0e822012-09-05 15:24:24 -04001283 return;
1284}
1285
1286int TWPartitionManager::Wipe_By_Path(string Path) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001287 std::vector<TWPartition*>::iterator iter;
Mohd Faraz77a31912020-04-21 18:56:44 +05301288 std::vector < TWPartition * >::iterator iter1;
Dees_Troy63c8df72012-09-10 14:02:05 -04001289 int ret = false;
1290 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -04001291 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy63c8df72012-09-10 14:02:05 -04001292
1293 // Iterate through all partitions
1294 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -04001295 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Mohd Faraz77a31912020-04-21 18:56:44 +05301296 // iterate through all partitions since some legacy devices uses other partitions as vendor causes issues while wiping
1297 (*iter)->Find_Actual_Block_Device();
1298 for (iter1 = Partitions.begin (); iter1 != Partitions.end (); iter1++)
1299 {
1300 (*iter1)->Find_Actual_Block_Device();
1301 if ((*iter)->Actual_Block_Device == (*iter1)->Actual_Block_Device && (*iter)->Mount_Point != (*iter1)->Mount_Point)
1302 (*iter1)->UnMount(false);
1303 }
Dees_Troye58d5262012-09-21 12:27:57 -04001304 if (Path == "/and-sec")
1305 ret = (*iter)->Wipe_AndSec();
1306 else
1307 ret = (*iter)->Wipe();
Dees_Troy63c8df72012-09-10 14:02:05 -04001308 found = true;
1309 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1310 (*iter)->Wipe();
1311 }
1312 }
1313 if (found) {
1314 return ret;
1315 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001316 gui_msg(Msg(msg::kError, "unable_find_part_path=Unable to find partition for path '{1}'")(Local_Path));
Dees_Troy63c8df72012-09-10 14:02:05 -04001317 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -04001318}
1319
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001320int TWPartitionManager::Wipe_By_Path(string Path, string New_File_System) {
1321 std::vector<TWPartition*>::iterator iter;
1322 int ret = false;
1323 bool found = false;
1324 string Local_Path = TWFunc::Get_Root_Path(Path);
1325
1326 // Iterate through all partitions
1327 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1328 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1329 if (Path == "/and-sec")
1330 ret = (*iter)->Wipe_AndSec();
1331 else
1332 ret = (*iter)->Wipe(New_File_System);
1333 found = true;
1334 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1335 (*iter)->Wipe(New_File_System);
1336 }
1337 }
1338 if (found) {
1339 return ret;
1340 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001341 gui_msg(Msg(msg::kError, "unable_find_part_path=Unable to find partition for path '{1}'")(Local_Path));
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001342 return false;
1343}
1344
Dees_Troy51a0e822012-09-05 15:24:24 -04001345int TWPartitionManager::Factory_Reset(void) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001346 std::vector<TWPartition*>::iterator iter;
1347 int ret = true;
1348
1349 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001350 if ((*iter)->Wipe_During_Factory_Reset && (*iter)->Is_Present) {
Ethan Yonker89583ef2015-08-26 09:01:59 -05001351#ifdef TW_OEM_BUILD
1352 if ((*iter)->Mount_Point == "/data") {
1353 if (!(*iter)->Wipe_Encryption())
1354 ret = false;
1355 } else {
1356#endif
1357 if (!(*iter)->Wipe())
1358 ret = false;
1359#ifdef TW_OEM_BUILD
1360 }
1361#endif
Dees_Troy094207a2012-09-26 12:00:39 -04001362 } else if ((*iter)->Has_Android_Secure) {
1363 if (!(*iter)->Wipe_AndSec())
1364 ret = false;
Dees_Troy63c8df72012-09-10 14:02:05 -04001365 }
1366 }
Ethan Yonker9fa76ba2016-06-30 14:05:43 -05001367 TWFunc::check_and_run_script("/sbin/factoryreset.sh", "Factory Reset Script");
Dees_Troy63c8df72012-09-10 14:02:05 -04001368 return ret;
Dees_Troy51a0e822012-09-05 15:24:24 -04001369}
1370
Dees_Troy38bd7602012-09-14 13:33:53 -04001371int TWPartitionManager::Wipe_Dalvik_Cache(void) {
1372 struct stat st;
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001373 vector <string> dir;
Dees_Troy38bd7602012-09-14 13:33:53 -04001374
1375 if (!Mount_By_Path("/data", true))
1376 return false;
1377
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001378 dir.push_back("/data/dalvik-cache");
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001379
1380 std::string cacheDir = TWFunc::get_cache_dir();
1381 if (cacheDir == NON_AB_CACHE_DIR) {
1382 if (!PartitionManager.Mount_By_Path(NON_AB_CACHE_DIR, false)) {
1383 LOGINFO("Unable to mount %s for wiping cache.\n", NON_AB_CACHE_DIR);
1384 }
1385 dir.push_back(cacheDir + "dalvik-cache");
1386 dir.push_back(cacheDir + "/dc");
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001387 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001388
Dees_Troy38bd7602012-09-14 13:33:53 -04001389 TWPartition* sdext = Find_Partition_By_Path("/sd-ext");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001390 if (sdext && sdext->Is_Present && sdext->Mount(false))
1391 {
1392 if (stat("/sd-ext/dalvik-cache", &st) == 0)
1393 {
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001394 dir.push_back("/sd-ext/dalvik-cache");
1395 }
1396 }
1397
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001398 if (cacheDir == NON_AB_CACHE_DIR) {
1399 gui_msg("wiping_cache_dalvik=Wiping Dalvik Cache Directories...");
1400 } else {
1401 gui_msg("wiping_dalvik=Wiping Dalvik Directory...");
1402 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001403 for (unsigned i = 0; i < dir.size(); ++i) {
1404 if (stat(dir.at(i).c_str(), &st) == 0) {
1405 TWFunc::removeDir(dir.at(i), false);
1406 gui_msg(Msg("cleaned=Cleaned: {1}...")(dir.at(i)));
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001407 }
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001408 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001409
1410 if (cacheDir == NON_AB_CACHE_DIR) {
1411 gui_msg("cache_dalvik_done=-- Dalvik Cache Directories Wipe Complete!");
1412 } else {
1413 gui_msg("dalvik_done=-- Dalvik Directory Wipe Complete!");
1414 }
1415
Dees_Troy38bd7602012-09-14 13:33:53 -04001416 return true;
1417}
1418
1419int TWPartitionManager::Wipe_Rotate_Data(void) {
1420 if (!Mount_By_Path("/data", true))
1421 return false;
1422
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001423 unlink("/data/misc/akmd*");
1424 unlink("/data/misc/rild*");
Dees_Troy2673cec2013-04-02 20:22:16 +00001425 gui_print("Rotation data wiped.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001426 return true;
1427}
1428
1429int TWPartitionManager::Wipe_Battery_Stats(void) {
1430 struct stat st;
1431
1432 if (!Mount_By_Path("/data", true))
1433 return false;
1434
1435 if (0 != stat("/data/system/batterystats.bin", &st)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001436 gui_print("No Battery Stats Found. No Need To Wipe.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001437 } else {
1438 remove("/data/system/batterystats.bin");
Dees_Troy2673cec2013-04-02 20:22:16 +00001439 gui_print("Cleared battery stats.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001440 }
1441 return true;
1442}
1443
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001444int TWPartitionManager::Wipe_Android_Secure(void) {
1445 std::vector<TWPartition*>::iterator iter;
1446 int ret = false;
1447 bool found = false;
1448
1449 // Iterate through all partitions
1450 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1451 if ((*iter)->Has_Android_Secure) {
1452 ret = (*iter)->Wipe_AndSec();
1453 found = true;
1454 }
1455 }
1456 if (found) {
1457 return ret;
1458 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001459 gui_err("no_andsec=No android secure partitions found.");
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001460 }
1461 return false;
1462}
1463
Dees_Troy38bd7602012-09-14 13:33:53 -04001464int TWPartitionManager::Format_Data(void) {
1465 TWPartition* dat = Find_Partition_By_Path("/data");
1466
1467 if (dat != NULL) {
1468 if (!dat->UnMount(true))
1469 return false;
1470
1471 return dat->Wipe_Encryption();
1472 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001473 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001474 return false;
1475 }
1476 return false;
1477}
1478
1479int TWPartitionManager::Wipe_Media_From_Data(void) {
1480 TWPartition* dat = Find_Partition_By_Path("/data");
1481
1482 if (dat != NULL) {
1483 if (!dat->Has_Data_Media) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001484 LOGERR("This device does not have /data/media\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001485 return false;
1486 }
1487 if (!dat->Mount(true))
1488 return false;
1489
Ethan Yonker74db1572015-10-28 12:44:49 -05001490 gui_msg("wiping_datamedia=Wiping internal storage -- /data/media...");
Ethan Yonker726a0202014-12-16 20:01:38 -06001491 Remove_MTP_Storage(dat->MTP_Storage_ID);
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001492 TWFunc::removeDir("/data/media", false);
xiaolu9416f4f2015-06-04 08:22:23 +08001493 dat->Recreate_Media_Folder();
Ethan Yonker726a0202014-12-16 20:01:38 -06001494 Add_MTP_Storage(dat->MTP_Storage_ID);
Dees_Troy38bd7602012-09-14 13:33:53 -04001495 return true;
1496 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001497 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001498 return false;
1499 }
1500 return false;
1501}
1502
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001503int TWPartitionManager::Repair_By_Path(string Path, bool Display_Error) {
1504 std::vector<TWPartition*>::iterator iter;
1505 int ret = false;
1506 bool found = false;
1507 string Local_Path = TWFunc::Get_Root_Path(Path);
1508
1509 if (Local_Path == "/tmp" || Local_Path == "/")
1510 return true;
1511
1512 // Iterate through all partitions
1513 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1514 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1515 ret = (*iter)->Repair();
1516 found = true;
1517 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1518 (*iter)->Repair();
1519 }
1520 }
1521 if (found) {
1522 return ret;
1523 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001524 gui_msg(Msg(msg::kError, "unable_find_part_path=Unable to find partition for path '{1}'")(Local_Path));
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001525 } else {
1526 LOGINFO("Repair: Unable to find partition for path '%s'\n", Local_Path.c_str());
1527 }
1528 return false;
1529}
1530
Ethan Yonkera2719152015-05-28 09:44:41 -05001531int TWPartitionManager::Resize_By_Path(string Path, bool Display_Error) {
1532 std::vector<TWPartition*>::iterator iter;
1533 int ret = false;
1534 bool found = false;
1535 string Local_Path = TWFunc::Get_Root_Path(Path);
1536
1537 if (Local_Path == "/tmp" || Local_Path == "/")
1538 return true;
1539
1540 // Iterate through all partitions
1541 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1542 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1543 ret = (*iter)->Resize();
1544 found = true;
1545 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1546 (*iter)->Resize();
1547 }
1548 }
1549 if (found) {
1550 return ret;
1551 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001552 gui_msg(Msg(msg::kError, "unable_find_part_path=Unable to find partition for path '{1}'")(Local_Path));
Ethan Yonkera2719152015-05-28 09:44:41 -05001553 } else {
1554 LOGINFO("Resize: Unable to find partition for path '%s'\n", Local_Path.c_str());
1555 }
1556 return false;
1557}
1558
Dees_Troy51a0e822012-09-05 15:24:24 -04001559void TWPartitionManager::Update_System_Details(void) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001560 std::vector<TWPartition*>::iterator iter;
Dees_Troy51127312012-09-08 13:08:49 -04001561 int data_size = 0;
Dees_Troy5bf43922012-09-07 16:07:55 -04001562
Ethan Yonker74db1572015-10-28 12:44:49 -05001563 gui_msg("update_part_details=Updating partition details...");
Dees_Troy5bf43922012-09-07 16:07:55 -04001564 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker1b190162016-12-05 15:25:19 -06001565 (*iter)->Update_Size(true);
Dees_Troy51127312012-09-08 13:08:49 -04001566 if ((*iter)->Can_Be_Mounted) {
Captain Throwback9d6feb52018-07-27 10:05:24 -04001567 if ((*iter)->Mount_Point == Get_Android_Root_Path()) {
Dees_Troy51127312012-09-08 13:08:49 -04001568 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1569 DataManager::SetValue(TW_BACKUP_SYSTEM_SIZE, backup_display_size);
Ethan Yonker76bbd3a2019-05-10 10:50:04 -05001570 TWFunc::Is_TWRP_App_In_System();
Dees_Troy51127312012-09-08 13:08:49 -04001571 } else if ((*iter)->Mount_Point == "/data" || (*iter)->Mount_Point == "/datadata") {
1572 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
1573 } else if ((*iter)->Mount_Point == "/cache") {
1574 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1575 DataManager::SetValue(TW_BACKUP_CACHE_SIZE, backup_display_size);
1576 } else if ((*iter)->Mount_Point == "/sd-ext") {
1577 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1578 DataManager::SetValue(TW_BACKUP_SDEXT_SIZE, backup_display_size);
1579 if ((*iter)->Backup_Size == 0) {
1580 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 0);
1581 DataManager::SetValue(TW_BACKUP_SDEXT_VAR, 0);
1582 } else
1583 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 1);
Dees_Troye58d5262012-09-21 12:27:57 -04001584 } else if ((*iter)->Has_Android_Secure) {
Dees_Troy8170a922012-09-18 15:40:25 -04001585 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troye58d5262012-09-21 12:27:57 -04001586 DataManager::SetValue(TW_BACKUP_ANDSEC_SIZE, backup_display_size);
Dees_Troy8170a922012-09-18 15:40:25 -04001587 if ((*iter)->Backup_Size == 0) {
1588 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 0);
1589 DataManager::SetValue(TW_BACKUP_ANDSEC_VAR, 0);
1590 } else
1591 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 1);
Dees_Troy2c50e182012-09-26 20:05:28 -04001592 } else if ((*iter)->Mount_Point == "/boot") {
1593 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1594 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1595 if ((*iter)->Backup_Size == 0) {
1596 DataManager::SetValue("tw_has_boot_partition", 0);
1597 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1598 } else
1599 DataManager::SetValue("tw_has_boot_partition", 1);
Dees_Troy51127312012-09-08 13:08:49 -04001600 }
Dees_Troyaa9cc402012-10-13 12:14:05 -04001601 } else {
1602 // Handle unmountable partitions in case we reset defaults
1603 if ((*iter)->Mount_Point == "/boot") {
1604 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1605 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1606 if ((*iter)->Backup_Size == 0) {
1607 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 0);
1608 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1609 } else
1610 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 1);
1611 } else if ((*iter)->Mount_Point == "/recovery") {
1612 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1613 DataManager::SetValue(TW_BACKUP_RECOVERY_SIZE, backup_display_size);
1614 if ((*iter)->Backup_Size == 0) {
1615 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 0);
1616 DataManager::SetValue(TW_BACKUP_RECOVERY_VAR, 0);
1617 } else
1618 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 1);
Gary Peck82599a82012-11-21 16:23:12 -08001619 } else if ((*iter)->Mount_Point == "/data") {
1620 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troyaa9cc402012-10-13 12:14:05 -04001621 }
Dees_Troy51127312012-09-08 13:08:49 -04001622 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001623 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001624 gui_msg("update_part_details_done=...done");
Dees_Troy51127312012-09-08 13:08:49 -04001625 DataManager::SetValue(TW_BACKUP_DATA_SIZE, data_size);
1626 string current_storage_path = DataManager::GetCurrentStoragePath();
1627 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
Dees_Troy8170a922012-09-18 15:40:25 -04001628 if (FreeStorage != NULL) {
1629 // Attempt to mount storage
1630 if (!FreeStorage->Mount(false)) {
Matt Mower2d50cad2015-12-03 22:26:55 -06001631 gui_msg(Msg(msg::kError, "unable_to_mount_storage=Unable to mount storage"));
1632 DataManager::SetValue(TW_STORAGE_FREE_SIZE, 0);
Dees_Troy8170a922012-09-18 15:40:25 -04001633 } else {
1634 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
1635 }
1636 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001637 LOGINFO("Unable to find storage partition '%s'.\n", current_storage_path.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -04001638 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001639 if (!Write_Fstab())
Dees_Troy2673cec2013-04-02 20:22:16 +00001640 LOGERR("Error creating fstab\n");
Dees_Troy51a0e822012-09-05 15:24:24 -04001641 return;
1642}
1643
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001644void TWPartitionManager::Post_Decrypt(const string& Block_Device) {
1645 TWPartition* dat = Find_Partition_By_Path("/data");
1646 if (dat != NULL) {
1647 DataManager::SetValue(TW_IS_DECRYPTED, 1);
1648 dat->Is_Decrypted = true;
1649 if (!Block_Device.empty()) {
1650 dat->Decrypted_Block_Device = Block_Device;
1651 gui_msg(Msg("decrypt_success_dev=Data successfully decrypted, new block device: '{1}'")(Block_Device));
1652 } else {
1653 gui_msg("decrypt_success_nodev=Data successfully decrypted");
1654 }
1655 dat->Setup_File_System(false);
1656 dat->Current_File_System = dat->Fstab_File_System; // Needed if we're ignoring blkid because encrypted devices start out as emmc
1657
1658 // Sleep for a bit so that the device will be ready
1659 sleep(1);
1660 if (dat->Has_Data_Media && dat->Mount(false) && TWFunc::Path_Exists("/data/media/0")) {
1661 dat->Storage_Path = "/data/media/0";
1662 dat->Symlink_Path = dat->Storage_Path;
1663 DataManager::SetValue("tw_storage_path", "/data/media/0");
1664 DataManager::SetValue("tw_settings_path", "/data/media/0");
1665 dat->UnMount(false);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001666 }
1667 Update_System_Details();
nkk71ffb02bd2017-06-04 23:12:16 +03001668 Output_Partition(dat);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001669 UnMount_Main_Partitions();
1670 } else
1671 LOGERR("Unable to locate data partition.\n");
1672}
1673
Dees_Troy51a0e822012-09-05 15:24:24 -04001674int TWPartitionManager::Decrypt_Device(string Password) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001675#ifdef TW_INCLUDE_CRYPTO
Ethan Yonkerddb63e22017-01-19 14:01:57 -06001676 char crypto_state[PROPERTY_VALUE_MAX], crypto_blkdev[PROPERTY_VALUE_MAX];
Ethan Yonker253368a2014-11-25 15:00:52 -06001677 std::vector<TWPartition*>::iterator iter;
Dees_Troy5bf43922012-09-07 16:07:55 -04001678
Ethan Yonker253368a2014-11-25 15:00:52 -06001679 // Mount any partitions that need to be mounted for decrypt
1680 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1681 if ((*iter)->Mount_To_Decrypt) {
1682 (*iter)->Mount(true);
1683 }
a39552696ff55ce2013-01-08 16:14:56 +00001684 }
oshmouna82a7542018-04-10 17:45:55 +02001685 property_set("twrp.mount_to_decrypt", "1");
a39552696ff55ce2013-01-08 16:14:56 +00001686
Ethan Yonkera1de1492015-11-04 11:58:27 -06001687 property_get("ro.crypto.state", crypto_state, "error");
1688 if (strcmp(crypto_state, "error") == 0) {
1689 property_set("ro.crypto.state", "encrypted");
1690 // Sleep for a bit so that services can start if needed
1691 sleep(1);
1692 }
1693
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001694 if (DataManager::GetIntValue(TW_IS_FBE)) {
1695#ifdef TW_INCLUDE_FBE
1696 if (!Mount_By_Path("/data", true)) // /data has to be mounted for FBE
1697 return -1;
1698 int retry_count = 10;
1699 while (!TWFunc::Path_Exists("/data/system/users/gatekeeper.password.key") && --retry_count)
1700 usleep(2000); // A small sleep is needed after mounting /data to ensure reliable decrypt... maybe because of DE?
1701 int user_id = DataManager::GetIntValue("tw_decrypt_user_id");
1702 LOGINFO("Decrypting FBE for user %i\n", user_id);
1703 if (Decrypt_User(user_id, Password)) {
1704 Post_Decrypt("");
1705 return 0;
1706 }
1707#else
1708 LOGERR("FBE support is not present\n");
1709#endif
1710 return -1;
1711 }
1712
Ethan Yonkerddb63e22017-01-19 14:01:57 -06001713 int pwret = -1;
1714 pid_t pid = fork();
1715 if (pid < 0) {
1716 LOGERR("fork failed\n");
1717 return -1;
1718 } else if (pid == 0) {
1719 // Child process
1720 char cPassword[255];
1721 strcpy(cPassword, Password.c_str());
1722 int ret = cryptfs_check_passwd(cPassword);
1723 exit(ret);
1724 } else {
1725 // Parent
1726 int status;
1727 if (TWFunc::Wait_For_Child_Timeout(pid, &status, "Decrypt", 30))
1728 pwret = -1;
1729 else
1730 pwret = WEXITSTATUS(status) ? -1 : 0;
1731 }
a39552696ff55ce2013-01-08 16:14:56 +00001732
nkk7171c6c502017-01-05 23:55:05 +02001733#ifdef TW_CRYPTO_USE_SYSTEM_VOLD
1734 if (pwret != 0) {
1735 pwret = vold_decrypt(Password);
nkk717d1222a2017-11-06 17:27:21 +02001736 switch (pwret) {
1737 case VD_SUCCESS:
1738 break;
1739 case VD_ERR_MISSING_VDC:
1740 gui_msg(Msg(msg::kError, "decrypt_data_vold_os_missing=Missing files needed for vold decrypt: {1}")("/system/bin/vdc"));
1741 break;
1742 case VD_ERR_MISSING_VOLD:
1743 gui_msg(Msg(msg::kError, "decrypt_data_vold_os_missing=Missing files needed for vold decrypt: {1}")("/system/bin/vold"));
1744 break;
1745 }
nkk7171c6c502017-01-05 23:55:05 +02001746 }
1747#endif // TW_CRYPTO_USE_SYSTEM_VOLD
1748
Ethan Yonker253368a2014-11-25 15:00:52 -06001749 // Unmount any partitions that were needed for decrypt
1750 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1751 if ((*iter)->Mount_To_Decrypt) {
1752 (*iter)->UnMount(false);
1753 }
1754 }
oshmouna82a7542018-04-10 17:45:55 +02001755 property_set("twrp.mount_to_decrypt", "0");
Ethan Yonker253368a2014-11-25 15:00:52 -06001756
a39552696ff55ce2013-01-08 16:14:56 +00001757 if (pwret != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001758 gui_err("fail_decrypt=Failed to decrypt data.");
Dees_Troy5bf43922012-09-07 16:07:55 -04001759 return -1;
1760 }
a39552696ff55ce2013-01-08 16:14:56 +00001761
Dees_Troy5bf43922012-09-07 16:07:55 -04001762 property_get("ro.crypto.fs_crypto_blkdev", crypto_blkdev, "error");
1763 if (strcmp(crypto_blkdev, "error") == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001764 LOGERR("Error retrieving decrypted data block device.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04001765 } else {
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001766 Post_Decrypt(crypto_blkdev);
Dees_Troy5bf43922012-09-07 16:07:55 -04001767 }
1768 return 0;
1769#else
Ethan Yonker74db1572015-10-28 12:44:49 -05001770 gui_err("no_crypto_support=No crypto support was compiled into this build.");
Dees_Troy5bf43922012-09-07 16:07:55 -04001771 return -1;
1772#endif
Dees_Troy51a0e822012-09-05 15:24:24 -04001773 return 1;
Dees_Troy51127312012-09-08 13:08:49 -04001774}
1775
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001776int TWPartitionManager::Fix_Contexts(void) {
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001777 std::vector<TWPartition*>::iterator iter;
1778 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1779 if ((*iter)->Has_Data_Media) {
1780 if ((*iter)->Mount(true)) {
1781 if (fixContexts::fixDataMediaContexts((*iter)->Mount_Point) != 0)
1782 return -1;
1783 }
1784 }
1785 }
Dees_Troy1a650e62012-10-19 20:54:32 -04001786 UnMount_Main_Partitions();
Ethan Yonker74db1572015-10-28 12:44:49 -05001787 gui_msg("done=Done.");
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001788 return 0;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04001789}
1790
Ethan Yonker66a19492015-12-10 10:19:45 -06001791TWPartition* TWPartitionManager::Find_Next_Storage(string Path, bool Exclude_Data_Media) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001792 std::vector<TWPartition*>::iterator iter = Partitions.begin();
1793
1794 if (!Path.empty()) {
1795 string Search_Path = TWFunc::Get_Root_Path(Path);
1796 for (; iter != Partitions.end(); iter++) {
Matt Mower9a561dd2016-02-22 21:25:51 -06001797 if ((*iter)->Mount_Point == Search_Path) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001798 iter++;
1799 break;
1800 }
1801 }
1802 }
1803
1804 for (; iter != Partitions.end(); iter++) {
Ethan Yonker66a19492015-12-10 10:19:45 -06001805 if (Exclude_Data_Media && (*iter)->Has_Data_Media) {
1806 // do nothing, do not return this type of partition
1807 } else if ((*iter)->Is_Storage && (*iter)->Is_Present) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001808 return (*iter);
1809 }
1810 }
1811
1812 return NULL;
1813}
1814
Dees_Troyd21618c2012-10-14 18:48:49 -04001815int TWPartitionManager::Open_Lun_File(string Partition_Path, string Lun_File) {
Dees_Troyd21618c2012-10-14 18:48:49 -04001816 TWPartition* Part = Find_Partition_By_Path(Partition_Path);
1817
1818 if (Part == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001819 LOGINFO("Unable to locate '%s' for USB storage mode.", Partition_Path.c_str());
1820 gui_msg(Msg(msg::kError, "unable_find_part_path=Unable to find partition for path '{1}'")(Partition_Path));
Dees_Troyd21618c2012-10-14 18:48:49 -04001821 return false;
1822 }
Ethan Yonker47360be2014-04-01 10:34:34 -05001823 LOGINFO("USB mount '%s', '%s' > '%s'\n", Partition_Path.c_str(), Part->Actual_Block_Device.c_str(), Lun_File.c_str());
1824 if (!Part->UnMount(true) || !Part->Is_Present)
Dees_Troyd21618c2012-10-14 18:48:49 -04001825 return false;
1826
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06001827 if (TWFunc::write_to_file(Lun_File, Part->Actual_Block_Device)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001828 LOGERR("Unable to write to ums lunfile '%s': (%s)\n", Lun_File.c_str(), strerror(errno));
Dees_Troyd21618c2012-10-14 18:48:49 -04001829 return false;
1830 }
Dees_Troyd21618c2012-10-14 18:48:49 -04001831 return true;
1832}
1833
Dees_Troy8170a922012-09-18 15:40:25 -04001834int TWPartitionManager::usb_storage_enable(void) {
Dees_Troy8170a922012-09-18 15:40:25 -04001835 char lun_file[255];
Dees_Troyd21618c2012-10-14 18:48:49 -04001836 bool has_multiple_lun = false;
Dees_Troy8170a922012-09-18 15:40:25 -04001837
Ethan Yonker47360be2014-04-01 10:34:34 -05001838 string Lun_File_str = CUSTOM_LUN_FILE;
1839 size_t found = Lun_File_str.find("%");
1840 if (found != string::npos) {
1841 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
1842 if (TWFunc::Path_Exists(lun_file))
1843 has_multiple_lun = true;
1844 }
Ethan Yonker726a0202014-12-16 20:01:38 -06001845 mtp_was_enabled = TWFunc::Toggle_MTP(false); // Must disable MTP for USB Storage
Ethan Yonker47360be2014-04-01 10:34:34 -05001846 if (!has_multiple_lun) {
1847 LOGINFO("Device doesn't have multiple lun files, mount current storage\n");
1848 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
1849 if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) == "/data") {
Ethan Yonker66a19492015-12-10 10:19:45 -06001850 TWPartition* Mount = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05001851 if (Mount) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001852 if (!Open_Lun_File(Mount->Mount_Point, lun_file)) {
1853 goto error_handle;
1854 }
Ethan Yonker47360be2014-04-01 10:34:34 -05001855 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001856 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001857 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05001858 }
1859 } else if (!Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file)) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001860 goto error_handle;
Dees_Troy8170a922012-09-18 15:40:25 -04001861 }
Dees_Troy8170a922012-09-18 15:40:25 -04001862 } else {
Ethan Yonker47360be2014-04-01 10:34:34 -05001863 LOGINFO("Device has multiple lun files\n");
1864 TWPartition* Mount1;
1865 TWPartition* Mount2;
Dees_Troy8170a922012-09-18 15:40:25 -04001866 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
Ethan Yonker66a19492015-12-10 10:19:45 -06001867 Mount1 = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05001868 if (Mount1) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001869 if (!Open_Lun_File(Mount1->Mount_Point, lun_file)) {
1870 goto error_handle;
1871 }
Matt Moweree717062014-04-26 01:46:46 -05001872 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
Ethan Yonker66a19492015-12-10 10:19:45 -06001873 Mount2 = Find_Next_Storage(Mount1->Mount_Point, true);
Matt Mower1fdcdb72016-01-25 20:53:47 -06001874 if (Mount2 && Mount2->Mount_Point != Mount1->Mount_Point) {
Matt Moweree717062014-04-26 01:46:46 -05001875 Open_Lun_File(Mount2->Mount_Point, lun_file);
Ethan Yonker47360be2014-04-01 10:34:34 -05001876 }
1877 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001878 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001879 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05001880 }
Dees_Troy8170a922012-09-18 15:40:25 -04001881 }
Matt Mowerb6ef4782014-11-06 02:24:36 -06001882 property_set("sys.storage.ums_enabled", "1");
HandyMenny37d42992014-10-26 22:15:59 +01001883 property_set("sys.usb.config", "mass_storage,adb");
Dees_Troy8170a922012-09-18 15:40:25 -04001884 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001885error_handle:
1886 if (mtp_was_enabled)
1887 if (!Enable_MTP())
1888 Disable_MTP();
1889 return false;
Dees_Troy8170a922012-09-18 15:40:25 -04001890}
1891
1892int TWPartitionManager::usb_storage_disable(void) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001893 int index, ret;
1894 char lun_file[255], ch[2] = {0, 0};
1895 string str = ch;
Dees_Troy8170a922012-09-18 15:40:25 -04001896
1897 for (index=0; index<2; index++) {
1898 sprintf(lun_file, CUSTOM_LUN_FILE, index);
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06001899 ret = TWFunc::write_to_file(lun_file, str);
Dees_Troy2673cec2013-04-02 20:22:16 +00001900 if (ret < 0) {
1901 break;
Dees_Troy8170a922012-09-18 15:40:25 -04001902 }
Dees_Troy8170a922012-09-18 15:40:25 -04001903 }
Dees_Troye58d5262012-09-21 12:27:57 -04001904 Mount_All_Storage();
1905 Update_System_Details();
Dees_Troycfd73ef2012-10-12 16:52:00 -04001906 UnMount_Main_Partitions();
Matt Mowerd9cb9062013-12-14 20:34:45 -06001907 property_set("sys.storage.ums_enabled", "0");
HandyMenny37d42992014-10-26 22:15:59 +01001908 property_set("sys.usb.config", "adb");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001909 if (mtp_was_enabled)
1910 if (!Enable_MTP())
1911 Disable_MTP();
Dees_Troy2673cec2013-04-02 20:22:16 +00001912 if (ret < 0 && index == 0) {
1913 LOGERR("Unable to write to ums lunfile '%s'.", lun_file);
1914 return false;
1915 } else {
1916 return true;
1917 }
Dees_Troy8170a922012-09-18 15:40:25 -04001918 return true;
Dees_Troy812660f2012-09-20 09:55:17 -04001919}
1920
1921void TWPartitionManager::Mount_All_Storage(void) {
1922 std::vector<TWPartition*>::iterator iter;
1923
1924 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1925 if ((*iter)->Is_Storage)
1926 (*iter)->Mount(false);
1927 }
Dees_Troy2c50e182012-09-26 20:05:28 -04001928}
Dees_Troy9350b8d2012-09-27 12:38:38 -04001929
Dees_Troyd0384ef2012-10-12 12:15:42 -04001930void TWPartitionManager::UnMount_Main_Partitions(void) {
1931 // Unmounts system and data if data is not data/media
1932 // Also unmounts boot if boot is mountable
Dees_Troy2673cec2013-04-02 20:22:16 +00001933 LOGINFO("Unmounting main partitions...\n");
Dees_Troyd0384ef2012-10-12 12:15:42 -04001934
1935 TWPartition* Boot_Partition = Find_Partition_By_Path("/boot");
1936
Captain Throwback9d6feb52018-07-27 10:05:24 -04001937 UnMount_By_Path(Get_Android_Root_Path(), true);
Ethan Yonker6277c792014-09-15 14:54:30 -05001938 if (!datamedia)
1939 UnMount_By_Path("/data", true);
1940
Dees_Troyd0384ef2012-10-12 12:15:42 -04001941 if (Boot_Partition != NULL && Boot_Partition->Can_Be_Mounted)
1942 Boot_Partition->UnMount(true);
1943}
1944
Dees_Troy9350b8d2012-09-27 12:38:38 -04001945int TWPartitionManager::Partition_SDCard(void) {
Matt Mower23d8aae2017-01-06 14:30:33 -06001946 char temp[255];
Ethan Yonker483e9f42016-01-11 22:21:18 -06001947 string Storage_Path, Command, Device, fat_str, ext_str, start_loc, end_loc, ext_format, sd_path, tmpdevice;
Dees_Troy9350b8d2012-09-27 12:38:38 -04001948 int ext, swap, total_size = 0, fat_size;
Dees_Troy9350b8d2012-09-27 12:38:38 -04001949
Ethan Yonker74db1572015-10-28 12:44:49 -05001950 gui_msg("start_partition_sd=Partitioning SD Card...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06001951
1952 // Locate and validate device to partition
1953 TWPartition* SDCard = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
1954
Ethan Yonker66a19492015-12-10 10:19:45 -06001955 if (SDCard->Is_Adopted_Storage)
1956 SDCard->Revert_Adopted();
1957
Ethan Yonker1eff6cd2014-09-15 13:30:42 -05001958 if (SDCard == NULL || !SDCard->Removable || SDCard->Has_Data_Media) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001959 gui_err("partition_sd_locate=Unable to locate device to partition.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04001960 return false;
1961 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001962
1963 // Unmount everything
Dees_Troy9350b8d2012-09-27 12:38:38 -04001964 if (!SDCard->UnMount(true))
1965 return false;
1966 TWPartition* SDext = Find_Partition_By_Path("/sd-ext");
1967 if (SDext != NULL) {
1968 if (!SDext->UnMount(true))
1969 return false;
1970 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06001971 char* swappath = getenv("SWAPPATH");
1972 if (swappath != NULL) {
1973 LOGINFO("Unmounting swap at '%s'\n", swappath);
1974 umount(swappath);
1975 }
Vojtech Bocek05534202013-09-11 08:11:56 +02001976
Ethan Yonker483e9f42016-01-11 22:21:18 -06001977 // Determine block device
1978 if (SDCard->Alternate_Block_Device.empty()) {
1979 SDCard->Find_Actual_Block_Device();
1980 Device = SDCard->Actual_Block_Device;
1981 // Just use the root block device
1982 Device.resize(strlen("/dev/block/mmcblkX"));
1983 } else {
1984 Device = SDCard->Alternate_Block_Device;
1985 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04001986
1987 // Find the size of the block device:
Ethan Yonker483e9f42016-01-11 22:21:18 -06001988 total_size = (int)(TWFunc::IOCTL_Get_Block_Size(Device.c_str()) / (1048576));
Dees_Troy9350b8d2012-09-27 12:38:38 -04001989
1990 DataManager::GetValue("tw_sdext_size", ext);
1991 DataManager::GetValue("tw_swap_size", swap);
1992 DataManager::GetValue("tw_sdpart_file_system", ext_format);
1993 fat_size = total_size - ext - swap;
Ethan Yonker483e9f42016-01-11 22:21:18 -06001994 LOGINFO("sd card mount point %s block device is '%s', sdcard size is: %iMB, fat size: %iMB, ext size: %iMB, ext system: '%s', swap size: %iMB\n", DataManager::GetCurrentStoragePath().c_str(), Device.c_str(), total_size, fat_size, ext, ext_format.c_str(), swap);
1995
1996 // Determine partition sizes
1997 if (swap == 0 && ext == 0) {
1998 fat_str = "-0";
1999 } else {
2000 memset(temp, 0, sizeof(temp));
2001 sprintf(temp, "%i", fat_size);
2002 fat_str = temp;
2003 fat_str += "MB";
2004 }
2005 if (swap == 0) {
2006 ext_str = "-0";
2007 } else {
2008 memset(temp, 0, sizeof(temp));
2009 sprintf(temp, "%i", ext);
2010 ext_str = "+";
2011 ext_str += temp;
2012 ext_str += "MB";
2013 }
2014
Dees_Troy9350b8d2012-09-27 12:38:38 -04002015 if (ext + swap > total_size) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002016 gui_err("ext_swap_size=EXT + Swap size is larger than sdcard size.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002017 return false;
2018 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002019
Ethan Yonker74db1572015-10-28 12:44:49 -05002020 gui_msg("remove_part_table=Removing partition table...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06002021 Command = "sgdisk --zap-all " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002022 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002023 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002024 gui_err("unable_rm_part=Unable to remove partition table.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002025 Update_System_Details();
2026 return false;
2027 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002028 gui_msg(Msg("create_part=Creating {1} partition...")("FAT32"));
Captain Throwback9643a802016-05-27 11:44:51 -04002029 Command = "sgdisk --new=0:0:" + fat_str + " --change-name=0:\"Microsoft basic data\" --typecode=0:EBD0A0A2-B9E5-4433-87C0-68B6B72699C7 " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002030 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002031 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002032 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("FAT32"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002033 return false;
2034 }
2035 if (ext > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002036 gui_msg(Msg("create_part=Creating {1} partition...")("EXT"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06002037 Command = "sgdisk --new=0:0:" + ext_str + " --change-name=0:\"Linux filesystem\" " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002038 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002039 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002040 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("EXT"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002041 Update_System_Details();
2042 return false;
2043 }
2044 }
2045 if (swap > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002046 gui_msg(Msg("create_part=Creating {1} partition...")("swap"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06002047 Command = "sgdisk --new=0:0:-0 --change-name=0:\"Linux swap\" --typecode=0:0657FD6D-A4AB-43C4-84E5-0933C84B4F4F " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002048 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002049 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002050 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("swap"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002051 Update_System_Details();
2052 return false;
2053 }
2054 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002055
2056 // Convert GPT to MBR
2057 Command = "sgdisk --gpttombr " + Device;
2058 if (TWFunc::Exec_Cmd(Command) != 0)
2059 LOGINFO("Failed to covert partition GPT to MBR\n");
2060
2061 // Tell the kernel to rescan the partition table
2062 int fd = open(Device.c_str(), O_RDONLY);
2063 ioctl(fd, BLKRRPART, 0);
2064 close(fd);
2065
2066 string format_device = Device;
2067 if (Device.substr(0, 17) == "/dev/block/mmcblk")
2068 format_device += "p";
2069
2070 // Format new partitions to proper file system
2071 if (fat_size > 0) {
2072 Command = "mkfs.fat " + format_device + "1";
2073 TWFunc::Exec_Cmd(Command);
2074 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04002075 if (ext > 0) {
2076 if (SDext == NULL) {
Ethan Yonker483e9f42016-01-11 22:21:18 -06002077 Command = "mke2fs -t " + ext_format + " -m 0 " + format_device + "2";
2078 gui_msg(Msg("format_sdext_as=Formatting sd-ext as {1}...")(ext_format));
2079 LOGINFO("Formatting sd-ext after partitioning, command: '%s'\n", Command.c_str());
2080 TWFunc::Exec_Cmd(Command);
2081 } else {
2082 SDext->Wipe(ext_format);
Dees_Troy9350b8d2012-09-27 12:38:38 -04002083 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002084 }
2085 if (swap > 0) {
2086 Command = "mkswap " + format_device;
2087 if (ext > 0)
2088 Command += "3";
2089 else
2090 Command += "2";
Vojtech Bocek05534202013-09-11 08:11:56 +02002091 TWFunc::Exec_Cmd(Command);
Dees_Troy9350b8d2012-09-27 12:38:38 -04002092 }
2093
Ethan Yonker483e9f42016-01-11 22:21:18 -06002094 // recreate TWRP folder and rewrite settings - these will be gone after sdcard is partitioned
2095 if (SDCard->Mount(true)) {
2096 string TWRP_Folder = SDCard->Mount_Point + "/TWRP";
2097 mkdir(TWRP_Folder.c_str(), 0777);
2098 DataManager::Flush();
2099 }
2100
Dees_Troy9350b8d2012-09-27 12:38:38 -04002101 Update_System_Details();
Ethan Yonker74db1572015-10-28 12:44:49 -05002102 gui_msg("part_complete=Partitioning complete.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002103 return true;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04002104}
Dees_Troya13d74f2013-03-24 08:54:55 -05002105
2106void TWPartitionManager::Get_Partition_List(string ListType, std::vector<PartitionList> *Partition_List) {
2107 std::vector<TWPartition*>::iterator iter;
2108 if (ListType == "mount") {
2109 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002110 if ((*iter)->Can_Be_Mounted) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002111 struct PartitionList part;
2112 part.Display_Name = (*iter)->Display_Name;
2113 part.Mount_Point = (*iter)->Mount_Point;
2114 part.selected = (*iter)->Is_Mounted();
2115 Partition_List->push_back(part);
2116 }
2117 }
2118 } else if (ListType == "storage") {
2119 char free_space[255];
2120 string Current_Storage = DataManager::GetCurrentStoragePath();
2121 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2122 if ((*iter)->Is_Storage) {
2123 struct PartitionList part;
2124 sprintf(free_space, "%llu", (*iter)->Free / 1024 / 1024);
2125 part.Display_Name = (*iter)->Storage_Name + " (";
2126 part.Display_Name += free_space;
2127 part.Display_Name += "MB)";
2128 part.Mount_Point = (*iter)->Storage_Path;
2129 if ((*iter)->Storage_Path == Current_Storage)
2130 part.selected = 1;
2131 else
2132 part.selected = 0;
2133 Partition_List->push_back(part);
2134 }
2135 }
2136 } else if (ListType == "backup") {
2137 char backup_size[255];
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002138 unsigned long long Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -05002139 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002140 if ((*iter)->Can_Be_Backed_Up && !(*iter)->Is_SubPartition && (*iter)->Is_Present) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002141 struct PartitionList part;
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002142 Backup_Size = (*iter)->Backup_Size;
2143 if ((*iter)->Has_SubPartition) {
2144 std::vector<TWPartition*>::iterator subpart;
2145
2146 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
2147 if ((*subpart)->Is_SubPartition && (*subpart)->Can_Be_Backed_Up && (*subpart)->Is_Present && (*subpart)->SubPartition_Of == (*iter)->Mount_Point)
2148 Backup_Size += (*subpart)->Backup_Size;
2149 }
2150 }
2151 sprintf(backup_size, "%llu", Backup_Size / 1024 / 1024);
Dees_Troya13d74f2013-03-24 08:54:55 -05002152 part.Display_Name = (*iter)->Backup_Display_Name + " (";
2153 part.Display_Name += backup_size;
2154 part.Display_Name += "MB)";
2155 part.Mount_Point = (*iter)->Backup_Path;
2156 part.selected = 0;
2157 Partition_List->push_back(part);
2158 }
2159 }
2160 } else if (ListType == "restore") {
2161 string Restore_List, restore_path;
2162 TWPartition* restore_part = NULL;
2163
2164 DataManager::GetValue("tw_restore_list", Restore_List);
2165 if (!Restore_List.empty()) {
2166 size_t start_pos = 0, end_pos = Restore_List.find(";", start_pos);
2167 while (end_pos != string::npos && start_pos < Restore_List.size()) {
2168 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04002169 struct PartitionList part;
2170 if (restore_path.compare("ADB_Backup") == 0) {
2171 part.Display_Name = "ADB Backup";
2172 part.Mount_Point = "ADB Backup";
2173 part.selected = 1;
2174 Partition_List->push_back(part);
2175 break;
2176 }
Dees_Troy59df9262013-06-19 14:53:57 -05002177 if ((restore_part = Find_Partition_By_Path(restore_path)) != NULL) {
Dees Troy4159aed2014-02-28 17:24:43 +00002178 if ((restore_part->Backup_Name == "recovery" && !restore_part->Can_Be_Backed_Up) || restore_part->Is_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002179 // Don't allow restore of recovery (causes problems on some devices)
Dees_Troy59df9262013-06-19 14:53:57 -05002180 // Don't add subpartitions to the list of items
Dees_Troya13d74f2013-03-24 08:54:55 -05002181 } else {
Dees_Troya13d74f2013-03-24 08:54:55 -05002182 part.Display_Name = restore_part->Backup_Display_Name;
2183 part.Mount_Point = restore_part->Backup_Path;
2184 part.selected = 1;
2185 Partition_List->push_back(part);
2186 }
2187 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002188 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05002189 }
2190 start_pos = end_pos + 1;
2191 end_pos = Restore_List.find(";", start_pos);
2192 }
2193 }
2194 } else if (ListType == "wipe") {
2195 struct PartitionList dalvik;
Ethan Yonker74db1572015-10-28 12:44:49 -05002196 dalvik.Display_Name = gui_parse_text("{@dalvik}");
Dees_Troya13d74f2013-03-24 08:54:55 -05002197 dalvik.Mount_Point = "DALVIK";
2198 dalvik.selected = 0;
2199 Partition_List->push_back(dalvik);
2200 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2201 if ((*iter)->Wipe_Available_in_GUI && !(*iter)->Is_SubPartition) {
2202 struct PartitionList part;
2203 part.Display_Name = (*iter)->Display_Name;
2204 part.Mount_Point = (*iter)->Mount_Point;
2205 part.selected = 0;
2206 Partition_List->push_back(part);
2207 }
2208 if ((*iter)->Has_Android_Secure) {
2209 struct PartitionList part;
2210 part.Display_Name = (*iter)->Backup_Display_Name;
2211 part.Mount_Point = (*iter)->Backup_Path;
2212 part.selected = 0;
2213 Partition_List->push_back(part);
2214 }
Dees_Troy74fb2e92013-04-15 14:35:47 +00002215 if ((*iter)->Has_Data_Media) {
2216 struct PartitionList datamedia;
2217 datamedia.Display_Name = (*iter)->Storage_Name;
2218 datamedia.Mount_Point = "INTERNAL";
2219 datamedia.selected = 0;
2220 Partition_List->push_back(datamedia);
2221 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002222 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002223 } else if (ListType == "flashimg") {
2224 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2225 if ((*iter)->Can_Flash_Img && (*iter)->Is_Present) {
2226 struct PartitionList part;
2227 part.Display_Name = (*iter)->Backup_Display_Name;
2228 part.Mount_Point = (*iter)->Backup_Path;
2229 part.selected = 0;
2230 Partition_List->push_back(part);
2231 }
2232 }
Ethan Yonker53796e72019-01-11 22:49:52 -06002233 if (DataManager::GetIntValue("tw_has_repack_tools") != 0 && DataManager::GetIntValue("tw_has_boot_slots") != 0) {
2234 TWPartition* boot = Find_Partition_By_Path("/boot");
2235 if (boot) {
2236 // Allow flashing kernels and ramdisks
2237 struct PartitionList repack_ramdisk;
2238 repack_ramdisk.Display_Name = gui_lookup("install_twrp_ramdisk", "Install Recovery Ramdisk");
2239 repack_ramdisk.Mount_Point = "/repack_ramdisk";
2240 repack_ramdisk.selected = 0;
2241 Partition_List->push_back(repack_ramdisk);
2242 /*struct PartitionList repack_kernel; For now let's leave repacking kernels under advanced only
2243 repack_kernel.Display_Name = gui_lookup("install_kernel", "Install Kernel");
2244 repack_kernel.Mount_Point = "/repack_kernel";
2245 repack_kernel.selected = 0;
2246 Partition_List->push_back(repack_kernel);*/
2247 }
2248 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002249 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00002250 LOGERR("Unknown list type '%s' requested for TWPartitionManager::Get_Partition_List\n", ListType.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -05002251 }
2252}
2253
2254int TWPartitionManager::Fstab_Processed(void) {
2255 return Partitions.size();
2256}
Dees_Troyd93bda52013-07-03 19:55:19 +00002257
2258void TWPartitionManager::Output_Storage_Fstab(void) {
2259 std::vector<TWPartition*>::iterator iter;
2260 char storage_partition[255];
bigbiff bigbiffe4bdb152019-03-23 18:33:17 -04002261 std::string Temp;
2262 std::string cacheDir = TWFunc::get_cache_dir();
2263
2264 if (cacheDir.empty()) {
2265 LOGINFO("Unable to find cache directory\n");
2266 return;
2267 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002268
2269 std::string storageFstab = TWFunc::get_cache_dir() + "recovery/storage.fstab";
2270 FILE *fp = fopen(storageFstab.c_str(), "w");
Dees_Troyd93bda52013-07-03 19:55:19 +00002271
2272 if (fp == NULL) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002273 gui_msg(Msg(msg::kError, "unable_to_open=Unable to open '{1}'.")(storageFstab));
Dees_Troyd93bda52013-07-03 19:55:19 +00002274 return;
2275 }
2276
2277 // Iterate through all partitions
2278 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2279 if ((*iter)->Is_Storage) {
2280 Temp = (*iter)->Storage_Path + ";" + (*iter)->Storage_Name + ";\n";
2281 strcpy(storage_partition, Temp.c_str());
2282 fwrite(storage_partition, sizeof(storage_partition[0]), strlen(storage_partition) / sizeof(storage_partition[0]), fp);
2283 }
2284 }
2285 fclose(fp);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02002286}
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002287
2288TWPartition *TWPartitionManager::Get_Default_Storage_Partition()
2289{
2290 TWPartition *res = NULL;
2291 for (std::vector<TWPartition*>::iterator iter = Partitions.begin(); iter != Partitions.end(); ++iter) {
Matt Mowera8a89d12016-12-30 18:10:37 -06002292 if (!(*iter)->Is_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002293 continue;
2294
Matt Mowera8a89d12016-12-30 18:10:37 -06002295 if ((*iter)->Is_Settings_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002296 return *iter;
2297
Matt Mowera8a89d12016-12-30 18:10:37 -06002298 if (!res)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002299 res = *iter;
2300 }
2301 return res;
2302}
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002303
2304bool TWPartitionManager::Enable_MTP(void) {
2305#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002306 if (mtppid) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002307 gui_err("mtp_already_enabled=MTP already enabled");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002308 return true;
2309 }
Ethan Yonker726a0202014-12-16 20:01:38 -06002310
2311 int mtppipe[2];
2312
2313 if (pipe(mtppipe) < 0) {
2314 LOGERR("Error creating MTP pipe\n");
2315 return false;
2316 }
2317
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002318 char old_value[PROPERTY_VALUE_MAX];
Matt Mowere07f0102017-02-23 17:40:00 -06002319 property_get("sys.usb.config", old_value, "");
2320 if (strcmp(old_value, "mtp,adb") != 0) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002321 char vendor[PROPERTY_VALUE_MAX];
2322 char product[PROPERTY_VALUE_MAX];
2323 property_set("sys.usb.config", "none");
2324 property_get("usb.vendor", vendor, "18D1");
2325 property_get("usb.product.mtpadb", product, "4EE2");
2326 string vendorstr = vendor;
2327 string productstr = product;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002328 TWFunc::write_to_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2329 TWFunc::write_to_file("/sys/class/android_usb/android0/idProduct", productstr);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002330 property_set("sys.usb.config", "mtp,adb");
2331 }
Matt Mower653a1702017-02-16 10:38:52 -06002332 /* To enable MTP debug, use the twrp command line feature:
Ethan Yonker6d154c42014-09-03 14:19:43 -05002333 * twrp set tw_mtp_debug 1
2334 */
2335 twrpMtp *mtp = new twrpMtp(DataManager::GetIntValue("tw_mtp_debug"));
Ethan Yonker1b039202015-01-30 10:08:48 -06002336 mtppid = mtp->forkserver(mtppipe);
2337 if (mtppid) {
2338 close(mtppipe[0]); // Host closes read side
2339 mtp_write_fd = mtppipe[1];
2340 DataManager::SetValue("tw_mtp_enabled", 1);
2341 Add_All_MTP_Storage();
2342 return true;
Ethan Yonker726a0202014-12-16 20:01:38 -06002343 } else {
2344 close(mtppipe[0]);
2345 close(mtppipe[1]);
Ethan Yonker74db1572015-10-28 12:44:49 -05002346 gui_err("mtp_fail=Failed to enable MTP");
Ethan Yonker1b039202015-01-30 10:08:48 -06002347 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002348 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002349#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002350 gui_err("no_mtp=MTP support not included");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002351#endif
2352 DataManager::SetValue("tw_mtp_enabled", 0);
2353 return false;
2354}
2355
Ethan Yonker1b039202015-01-30 10:08:48 -06002356void TWPartitionManager::Add_All_MTP_Storage(void) {
2357#ifdef TW_HAS_MTP
2358 std::vector<TWPartition*>::iterator iter;
2359
2360 if (!mtppid)
2361 return; // MTP is not enabled
2362
2363 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2364 if ((*iter)->Is_Storage && (*iter)->Is_Present && (*iter)->Mount(false))
2365 Add_Remove_MTP_Storage((*iter), MTP_MESSAGE_ADD_STORAGE);
2366 }
2367#else
2368 return;
2369#endif
2370}
2371
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002372bool TWPartitionManager::Disable_MTP(void) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002373 char old_value[PROPERTY_VALUE_MAX];
bigbiff bigbiffaf32bb92018-12-18 18:39:53 -05002374 property_set("sys.usb.ffs.mtp.ready", "0");
Matt Mowere07f0102017-02-23 17:40:00 -06002375 property_get("sys.usb.config", old_value, "");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002376 if (strcmp(old_value, "adb") != 0) {
2377 char vendor[PROPERTY_VALUE_MAX];
2378 char product[PROPERTY_VALUE_MAX];
2379 property_set("sys.usb.config", "none");
2380 property_get("usb.vendor", vendor, "18D1");
Matt Mowere07f0102017-02-23 17:40:00 -06002381 property_get("usb.product.adb", product, "D001");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002382 string vendorstr = vendor;
2383 string productstr = product;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002384 TWFunc::write_to_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2385 TWFunc::write_to_file("/sys/class/android_usb/android0/idProduct", productstr);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002386 usleep(2000);
2387 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002388#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002389 if (mtppid) {
Ethan Yonker8613dc02014-09-11 09:28:20 -05002390 LOGINFO("Disabling MTP\n");
2391 int status;
2392 kill(mtppid, SIGKILL);
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002393 mtppid = 0;
Ethan Yonker8613dc02014-09-11 09:28:20 -05002394 // We don't care about the exit value, but this prevents a zombie process
2395 waitpid(mtppid, &status, 0);
Ethan Yonker726a0202014-12-16 20:01:38 -06002396 close(mtp_write_fd);
2397 mtp_write_fd = -1;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002398 }
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002399#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002400 property_set("sys.usb.config", "adb");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002401#ifdef TW_HAS_MTP
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002402 DataManager::SetValue("tw_mtp_enabled", 0);
2403 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002404#endif
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002405 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002406}
Ethan Yonker726a0202014-12-16 20:01:38 -06002407
2408TWPartition* TWPartitionManager::Find_Partition_By_MTP_Storage_ID(unsigned int Storage_ID) {
2409 std::vector<TWPartition*>::iterator iter;
2410
2411 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2412 if ((*iter)->MTP_Storage_ID == Storage_ID)
2413 return (*iter);
2414 }
2415 return NULL;
2416}
2417
2418bool TWPartitionManager::Add_Remove_MTP_Storage(TWPartition* Part, int message_type) {
2419#ifdef TW_HAS_MTP
2420 struct mtpmsg mtp_message;
2421
2422 if (!mtppid)
2423 return false; // MTP is disabled
2424
2425 if (mtp_write_fd < 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002426 LOGINFO("MTP: mtp_write_fd is not set\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002427 return false;
2428 }
2429
2430 if (Part) {
Ethan Yonker4bfabab2014-12-29 09:15:37 -06002431 if (Part->MTP_Storage_ID == 0)
2432 return false;
Ethan Yonker726a0202014-12-16 20:01:38 -06002433 if (message_type == MTP_MESSAGE_REMOVE_STORAGE) {
2434 mtp_message.message_type = MTP_MESSAGE_REMOVE_STORAGE; // Remove
2435 LOGINFO("sending message to remove %i\n", Part->MTP_Storage_ID);
2436 mtp_message.storage_id = Part->MTP_Storage_ID;
2437 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002438 LOGINFO("error sending message to remove storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002439 return false;
2440 } else {
2441 LOGINFO("Message sent, remove storage ID: %i\n", Part->MTP_Storage_ID);
2442 return true;
2443 }
2444 } else if (message_type == MTP_MESSAGE_ADD_STORAGE && Part->Is_Mounted()) {
2445 mtp_message.message_type = MTP_MESSAGE_ADD_STORAGE; // Add
2446 mtp_message.storage_id = Part->MTP_Storage_ID;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002447 if (Part->Storage_Path.size() >= sizeof(mtp_message.path)) {
2448 LOGERR("Storage path '%s' too large for mtpmsg\n", Part->Storage_Path.c_str());
2449 return false;
2450 }
2451 strcpy(mtp_message.path, Part->Storage_Path.c_str());
2452 if (Part->Storage_Name.size() >= sizeof(mtp_message.display)) {
2453 LOGERR("Storage name '%s' too large for mtpmsg\n", Part->Storage_Name.c_str());
2454 return false;
2455 }
2456 strcpy(mtp_message.display, Part->Storage_Name.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002457 mtp_message.maxFileSize = Part->Get_Max_FileSize();
Ethan Yonker1b039202015-01-30 10:08:48 -06002458 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 -06002459 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002460 LOGINFO("error sending message to add storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002461 return false;
2462 } else {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002463 LOGINFO("Message sent, add storage ID: %i '%s'\n", Part->MTP_Storage_ID, mtp_message.path);
Ethan Yonker726a0202014-12-16 20:01:38 -06002464 return true;
2465 }
2466 } else {
2467 LOGERR("Unknown MTP message type: %i\n", message_type);
2468 }
2469 } else {
2470 // This hopefully never happens as the error handling should
2471 // occur in the calling function.
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002472 LOGINFO("TWPartitionManager::Add_Remove_MTP_Storage NULL partition given\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002473 }
2474 return true;
2475#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002476 gui_err("no_mtp=MTP support not included");
Ethan Yonker726a0202014-12-16 20:01:38 -06002477 DataManager::SetValue("tw_mtp_enabled", 0);
2478 return false;
2479#endif
2480}
2481
2482bool TWPartitionManager::Add_MTP_Storage(string Mount_Point) {
2483#ifdef TW_HAS_MTP
2484 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2485 if (Part) {
2486 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2487 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002488 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002489 }
2490#endif
2491 return false;
2492}
2493
2494bool TWPartitionManager::Add_MTP_Storage(unsigned int Storage_ID) {
2495#ifdef TW_HAS_MTP
2496 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2497 if (Part) {
2498 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2499 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002500 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002501 }
2502#endif
2503 return false;
2504}
2505
2506bool TWPartitionManager::Remove_MTP_Storage(string Mount_Point) {
2507#ifdef TW_HAS_MTP
2508 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2509 if (Part) {
2510 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2511 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002512 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002513 }
2514#endif
2515 return false;
2516}
2517
2518bool TWPartitionManager::Remove_MTP_Storage(unsigned int Storage_ID) {
2519#ifdef TW_HAS_MTP
2520 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2521 if (Part) {
2522 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2523 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002524 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002525 }
2526#endif
2527 return false;
2528}
Ethan Yonker96af84a2015-01-05 14:58:36 -06002529
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002530bool TWPartitionManager::Flash_Image(string& path, string& filename) {
Matt Mower23d8aae2017-01-06 14:30:33 -06002531 int partition_count = 0;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002532 TWPartition* flash_part = NULL;
bigbiffce8f83c2015-12-12 18:30:21 -05002533 string Flash_List, flash_path, full_filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002534 size_t start_pos = 0, end_pos = 0;
2535
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002536 full_filename = path + "/" + filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002537
bigbiffce8f83c2015-12-12 18:30:21 -05002538 gui_msg("image_flash_start=[IMAGE FLASH STARTED]");
2539 gui_msg(Msg("img_to_flash=Image to flash: '{1}'")(full_filename));
2540
2541 if (!TWFunc::Path_Exists(full_filename)) {
2542 if (!Mount_By_Path(full_filename, true)) {
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002543 return false;
2544 }
bigbiffce8f83c2015-12-12 18:30:21 -05002545 if (!TWFunc::Path_Exists(full_filename)) {
2546 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")(full_filename));
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002547 return false;
2548 }
2549 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002550
Ethan Yonker53796e72019-01-11 22:49:52 -06002551 DataManager::GetValue("tw_flash_partition", Flash_List);
2552 Repack_Type repack = REPLACE_NONE;
2553 if (Flash_List == "/repack_ramdisk;") {
2554 repack = REPLACE_RAMDISK;
2555 } else if (Flash_List == "/repack_kernel;") {
2556 repack = REPLACE_KERNEL;
2557 }
2558 if (repack != REPLACE_NONE) {
2559 Repack_Options_struct Repack_Options;
2560 Repack_Options.Type = repack;
2561 Repack_Options.Disable_Verity = false;
2562 Repack_Options.Disable_Force_Encrypt = false;
2563 Repack_Options.Backup_First = DataManager::GetIntValue("tw_repack_backup_first") != 0;
2564 return Repack_Images(full_filename, Repack_Options);
2565 }
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002566 PartitionSettings part_settings;
2567 part_settings.Backup_Folder = path;
2568 unsigned long long total_bytes = TWFunc::Get_File_Size(full_filename);
2569 ProgressTracking progress(total_bytes);
2570 part_settings.progress = &progress;
2571 part_settings.adbbackup = false;
2572 part_settings.PM_Method = PM_RESTORE;
Ethan Yonker74db1572015-10-28 12:44:49 -05002573 gui_msg("calc_restore=Calculating restore details...");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002574 if (!Flash_List.empty()) {
2575 end_pos = Flash_List.find(";", start_pos);
2576 while (end_pos != string::npos && start_pos < Flash_List.size()) {
2577 flash_path = Flash_List.substr(start_pos, end_pos - start_pos);
2578 flash_part = Find_Partition_By_Path(flash_path);
2579 if (flash_part != NULL) {
2580 partition_count++;
2581 if (partition_count > 1) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002582 gui_err("too_many_flash=Too many partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002583 return false;
2584 }
2585 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002586 gui_msg(Msg(msg::kError, "flash_unable_locate=Unable to locate '{1}' partition for flashing.")(flash_path));
Ethan Yonker96af84a2015-01-05 14:58:36 -06002587 return false;
2588 }
2589 start_pos = end_pos + 1;
2590 end_pos = Flash_List.find(";", start_pos);
2591 }
2592 }
2593
2594 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002595 gui_err("no_part_flash=No partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002596 return false;
2597 }
2598
2599 DataManager::SetProgress(0.0);
2600 if (flash_part) {
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002601 flash_part->Backup_FileName = filename;
2602 if (!flash_part->Flash_Image(&part_settings))
Ethan Yonker96af84a2015-01-05 14:58:36 -06002603 return false;
2604 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002605 gui_err("invalid_flash=Invalid flash partition specified.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002606 return false;
2607 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002608 gui_highlight("flash_done=IMAGE FLASH COMPLETED]");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002609 return true;
2610}
Ethan Yonker74db1572015-10-28 12:44:49 -05002611
2612void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value) {
2613 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2614 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002615 if (part->Is_Adopted_Storage) {
2616 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2617 part->Backup_Display_Name = part->Display_Name;
2618 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2619 } else {
2620 part->Display_Name = gui_lookup(resource_name, default_value);
2621 part->Backup_Display_Name = part->Display_Name;
2622 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002623 }
2624}
2625
2626void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value, const char* storage_resource_name, const char* storage_default_value) {
2627 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2628 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002629 if (part->Is_Adopted_Storage) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06002630 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup("data_backup", "Data (excl. storage)");
Ethan Yonker66a19492015-12-10 10:19:45 -06002631 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
Ethan Yonker66a19492015-12-10 10:19:45 -06002632 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2633 } else {
2634 part->Display_Name = gui_lookup(resource_name, default_value);
2635 part->Backup_Display_Name = part->Display_Name;
2636 if (part->Is_Storage)
2637 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2638 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002639 }
2640}
2641
Ethan Yonker01f4e032017-02-03 15:30:52 -06002642void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value, const char* storage_resource_name, const char* storage_default_value, const char* backup_name, const char* backup_default) {
2643 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2644 if (part) {
2645 if (part->Is_Adopted_Storage) {
2646 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup(backup_name, backup_default);
2647 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2648 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2649 } else {
2650 part->Display_Name = gui_lookup(resource_name, default_value);
2651 part->Backup_Display_Name = gui_lookup(backup_name, backup_default);
2652 if (part->Is_Storage)
2653 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2654 }
2655 }
2656}
2657
Ethan Yonker74db1572015-10-28 12:44:49 -05002658void TWPartitionManager::Translate_Partition_Display_Names() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002659 LOGINFO("Translating partition display names\n");
Ethan Yonker74db1572015-10-28 12:44:49 -05002660 Translate_Partition("/system", "system", "System");
2661 Translate_Partition("/system_image", "system_image", "System Image");
2662 Translate_Partition("/vendor", "vendor", "Vendor");
2663 Translate_Partition("/vendor_image", "vendor_image", "Vendor Image");
2664 Translate_Partition("/cache", "cache", "Cache");
Ethan Yonker74db1572015-10-28 12:44:49 -05002665 Translate_Partition("/boot", "boot", "Boot");
2666 Translate_Partition("/recovery", "recovery", "Recovery");
2667 if (!datamedia) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06002668 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage");
Ethan Yonker74db1572015-10-28 12:44:49 -05002669 Translate_Partition("/sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2670 Translate_Partition("/internal_sd", "sdcard", "SDCard", "sdcard", "SDCard");
2671 Translate_Partition("/internal_sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2672 Translate_Partition("/emmc", "sdcard", "SDCard", "sdcard", "SDCard");
Ethan Yonker01f4e032017-02-03 15:30:52 -06002673 } else {
2674 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05002675 }
Ethan Yonker01f4e032017-02-03 15:30:52 -06002676 Translate_Partition("/external_sd", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
2677 Translate_Partition("/external_sdcard", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05002678 Translate_Partition("/usb-otg", "usbotg", "USB OTG", "usbotg", "USB OTG");
2679 Translate_Partition("/sd-ext", "sdext", "SD-EXT");
2680
2681 // Android secure is a special case
2682 TWPartition* part = PartitionManager.Find_Partition_By_Path("/and-sec");
2683 if (part)
2684 part->Backup_Display_Name = gui_lookup("android_secure", "Android Secure");
2685
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002686 std::vector<TWPartition*>::iterator sysfs;
2687 for (sysfs = Partitions.begin(); sysfs != Partitions.end(); sysfs++) {
2688 if (!(*sysfs)->Sysfs_Entry.empty()) {
2689 Translate_Partition((*sysfs)->Mount_Point.c_str(), "autostorage", "Storage", "autostorage", "Storage");
2690 }
2691 }
2692
Ethan Yonker74db1572015-10-28 12:44:49 -05002693 // This updates the text on all of the storage selection buttons in the GUI
2694 DataManager::SetBackupFolder();
2695}
Ethan Yonker66a19492015-12-10 10:19:45 -06002696
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002697bool TWPartitionManager::Decrypt_Adopted() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002698#ifdef TW_INCLUDE_CRYPTO
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002699 bool ret = false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002700 if (!Mount_By_Path("/data", false)) {
2701 LOGERR("Cannot decrypt adopted storage because /data will not mount\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002702 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002703 }
2704 LOGINFO("Decrypt adopted storage starting\n");
2705 char* xmlFile = PageManager::LoadFileToBuffer("/data/system/storage.xml", NULL);
2706 xml_document<> *doc = NULL;
2707 xml_node<>* volumes = NULL;
Ethan Yonker66a19492015-12-10 10:19:45 -06002708 string Primary_Storage_UUID = "";
2709 if (xmlFile != NULL) {
2710 LOGINFO("successfully loaded storage.xml\n");
2711 doc = new xml_document<>();
2712 doc->parse<0>(xmlFile);
2713 volumes = doc->first_node("volumes");
2714 if (volumes) {
2715 xml_attribute<>* psuuid = volumes->first_attribute("primaryStorageUuid");
2716 if (psuuid) {
2717 Primary_Storage_UUID = psuuid->value();
2718 }
2719 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002720 } else {
2721 LOGINFO("No /data/system/storage.xml for adopted storage\n");
2722 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002723 }
2724 std::vector<TWPartition*>::iterator adopt;
2725 for (adopt = Partitions.begin(); adopt != Partitions.end(); adopt++) {
2726 if ((*adopt)->Removable && (*adopt)->Is_Present) {
2727 if ((*adopt)->Decrypt_Adopted() == 0) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002728 ret = true;
Ethan Yonker66a19492015-12-10 10:19:45 -06002729 if (volumes) {
2730 xml_node<>* volume = volumes->first_node("volume");
2731 while (volume) {
2732 xml_attribute<>* guid = volume->first_attribute("partGuid");
2733 if (guid) {
2734 string GUID = (*adopt)->Adopted_GUID.c_str();
2735 GUID.insert(8, "-");
2736 GUID.insert(13, "-");
2737 GUID.insert(18, "-");
2738 GUID.insert(23, "-");
2739
2740 if (strcasecmp(GUID.c_str(), guid->value()) == 0) {
2741 xml_attribute<>* attr = volume->first_attribute("nickname");
nkk71ffb02bd2017-06-04 23:12:16 +03002742 if (attr && attr->value() && strlen(attr->value()) > 0) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002743 (*adopt)->Storage_Name = attr->value();
2744 (*adopt)->Display_Name = (*adopt)->Storage_Name;
2745 (*adopt)->Backup_Display_Name = (*adopt)->Storage_Name;
2746 LOGINFO("storage name from storage.xml is '%s'\n", attr->value());
2747 }
2748 attr = volume->first_attribute("fsUuid");
2749 if (attr && !Primary_Storage_UUID.empty() && strcmp(Primary_Storage_UUID.c_str(), attr->value()) == 0) {
2750 TWPartition* Dat = Find_Partition_By_Path("/data");
2751 if (Dat) {
2752 LOGINFO("Internal storage is found on adopted storage '%s'\n", (*adopt)->Display_Name.c_str());
2753 LOGINFO("Changing '%s' to point to '%s'\n", Dat->Symlink_Mount_Point.c_str(), (*adopt)->Storage_Path.c_str());
2754 (*adopt)->Symlink_Mount_Point = Dat->Symlink_Mount_Point;
2755 Dat->Symlink_Mount_Point = "";
2756 // Toggle mounts to ensure that the symlink mount point (probably /sdcard) is mounted to the right location
2757 Dat->UnMount(false);
2758 Dat->Mount(false);
2759 (*adopt)->UnMount(false);
2760 (*adopt)->Mount(false);
Ethan Yonker66a19492015-12-10 10:19:45 -06002761 }
2762 }
2763 break;
2764 }
2765 }
2766 volume = volume->next_sibling("volume");
2767 }
2768 }
nkk71ffb02bd2017-06-04 23:12:16 +03002769 Update_System_Details();
2770 Output_Partition((*adopt));
Ethan Yonker66a19492015-12-10 10:19:45 -06002771 }
2772 }
2773 }
2774 if (xmlFile) {
2775 doc->clear();
2776 delete doc;
2777 free(xmlFile);
2778 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002779 return ret;
Ethan Yonker66a19492015-12-10 10:19:45 -06002780#else
2781 LOGINFO("Decrypt_Adopted: no crypto support\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002782 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002783#endif
2784}
Ethan Yonkerfcf3f242016-02-16 12:30:26 -06002785
2786void TWPartitionManager::Remove_Partition_By_Path(string Path) {
2787 std::vector<TWPartition*>::iterator iter;
2788 string Local_Path = TWFunc::Get_Root_Path(Path);
2789
2790 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2791 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
2792 LOGINFO("Found and erasing '%s' from partition list\n", Local_Path.c_str());
2793 Partitions.erase(iter);
2794 return;
2795 }
2796 }
2797}
Ethan Yonker1b190162016-12-05 15:25:19 -06002798
2799void TWPartitionManager::Set_Active_Slot(const string& Slot) {
2800 if (Slot != "A" && Slot != "B") {
2801 LOGERR("Set_Active_Slot invalid slot '%s'\n", Slot.c_str());
2802 return;
2803 }
2804 if (Active_Slot_Display == Slot)
2805 return;
2806 LOGINFO("Setting active slot %s\n", Slot.c_str());
2807#ifdef AB_OTA_UPDATER
2808 if (!Active_Slot_Display.empty()) {
2809 const hw_module_t *hw_module;
2810 boot_control_module_t *module;
2811 int ret;
2812 ret = hw_get_module("bootctrl", &hw_module);
2813 if (ret != 0) {
2814 LOGERR("Error getting bootctrl module.\n");
2815 } else {
2816 module = (boot_control_module_t*) hw_module;
2817 module->init(module);
2818 int slot_number = 0;
2819 if (Slot == "B")
2820 slot_number = 1;
2821 if (module->setActiveBootSlot(module, slot_number))
2822 gui_msg(Msg(msg::kError, "unable_set_boot_slot=Error changing bootloader boot slot to {1}")(Slot));
2823 }
2824 DataManager::SetValue("tw_active_slot", Slot); // Doing this outside of this if block may result in a seg fault because the DataManager may not be ready yet
2825 }
2826#else
2827 LOGERR("Boot slot feature not present\n");
2828#endif
2829 Active_Slot_Display = Slot;
2830 if (Fstab_Processed())
2831 Update_System_Details();
2832}
2833string TWPartitionManager::Get_Active_Slot_Suffix() {
2834 if (Active_Slot_Display == "A")
2835 return "_a";
2836 return "_b";
2837}
2838string TWPartitionManager::Get_Active_Slot_Display() {
2839 return Active_Slot_Display;
2840}
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002841
Captain Throwback9d6feb52018-07-27 10:05:24 -04002842string TWPartitionManager::Get_Android_Root_Path() {
2843 std::string Android_Root = getenv("ANDROID_ROOT");
2844 if (Android_Root == "")
2845 Android_Root = "/system";
2846 return Android_Root;
2847}
2848
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002849void TWPartitionManager::Remove_Uevent_Devices(const string& Mount_Point) {
2850 std::vector<TWPartition*>::iterator iter;
2851
2852 for (iter = Partitions.begin(); iter != Partitions.end(); ) {
2853 if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Mount_Point) {
2854 TWPartition *part = *iter;
2855 LOGINFO("%s was removed by uevent data\n", (*iter)->Mount_Point.c_str());
2856 (*iter)->UnMount(false);
2857 rmdir((*iter)->Mount_Point.c_str());
2858 iter = Partitions.erase(iter);
2859 delete part;
2860 } else {
2861 iter++;
2862 }
2863 }
2864}
2865
2866void TWPartitionManager::Handle_Uevent(const Uevent_Block_Data& uevent_data) {
2867 std::vector<TWPartition*>::iterator iter;
2868
2869 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2870 if (!(*iter)->Sysfs_Entry.empty()) {
2871 string device;
2872 size_t wildcard = (*iter)->Sysfs_Entry.find("*");
2873 if (wildcard != string::npos) {
2874 device = (*iter)->Sysfs_Entry.substr(0, wildcard);
2875 } else {
2876 device = (*iter)->Sysfs_Entry;
2877 }
2878 if (device == uevent_data.sysfs_path.substr(0, device.size())) {
2879 // Found a match
2880 if (uevent_data.action == "add") {
2881 (*iter)->Primary_Block_Device = "/dev/block/" + uevent_data.block_device;
2882 (*iter)->Alternate_Block_Device = (*iter)->Primary_Block_Device;
2883 (*iter)->Is_Present = true;
2884 LOGINFO("Found a match '%s' '%s'\n", uevent_data.block_device.c_str(), device.c_str());
2885 if (!Decrypt_Adopted()) {
2886 LOGINFO("No adopted storage so finding actual block device\n");
2887 (*iter)->Find_Actual_Block_Device();
2888 }
2889 return;
2890 } else if (uevent_data.action == "remove") {
2891 (*iter)->Is_Present = false;
2892 (*iter)->Primary_Block_Device = "";
2893 (*iter)->Actual_Block_Device = "";
2894 Remove_Uevent_Devices((*iter)->Mount_Point);
2895 return;
2896 }
2897 }
2898 }
2899 }
2900 LOGINFO("Found no matching fstab entry for uevent device '%s' - %s\n", uevent_data.sysfs_path.c_str(), uevent_data.action.c_str());
2901}
2902
2903void TWPartitionManager::setup_uevent() {
2904 struct sockaddr_nl nls;
2905
2906 if (uevent_pfd.fd >= 0) {
2907 LOGINFO("uevent already set up\n");
2908 return;
2909 }
2910
2911 // Open hotplug event netlink socket
2912 memset(&nls,0,sizeof(struct sockaddr_nl));
2913 nls.nl_family = AF_NETLINK;
2914 nls.nl_pid = getpid();
2915 nls.nl_groups = -1;
2916 uevent_pfd.events = POLLIN;
2917 uevent_pfd.fd = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_KOBJECT_UEVENT);
2918 if (uevent_pfd.fd==-1) {
2919 LOGERR("uevent not root\n");
2920 return;
2921 }
2922
2923 // Listen to netlink socket
2924 if (::bind(uevent_pfd.fd, (struct sockaddr *) &nls, sizeof(struct sockaddr_nl)) < 0) {
2925 LOGERR("Bind failed\n");
2926 return;
2927 }
2928 set_select_fd();
2929 Coldboot();
2930}
2931
2932Uevent_Block_Data TWPartitionManager::get_event_block_values(char *buf, int len) {
2933 Uevent_Block_Data ret;
2934 ret.subsystem = "";
2935 char *ptr = buf;
2936 const char *end = buf + len;
2937
2938 buf[len - 1] = '\0';
2939 while (ptr < end) {
2940 if (strncmp(ptr, "ACTION=", strlen("ACTION=")) == 0) {
2941 ptr += strlen("ACTION=");
2942 ret.action = ptr;
2943 } else if (strncmp(ptr, "SUBSYSTEM=", strlen("SUBSYSTEM=")) == 0) {
2944 ptr += strlen("SUBSYSTEM=");
2945 ret.subsystem = ptr;
2946 } else if (strncmp(ptr, "DEVTYPE=", strlen("DEVTYPE=")) == 0) {
2947 ptr += strlen("DEVTYPE=");
2948 ret.type = ptr;
2949 } else if (strncmp(ptr, "DEVPATH=", strlen("DEVPATH=")) == 0) {
2950 ptr += strlen("DEVPATH=");
2951 ret.sysfs_path += ptr;
2952 } else if (strncmp(ptr, "DEVNAME=", strlen("DEVNAME=")) == 0) {
2953 ptr += strlen("DEVNAME=");
2954 ret.block_device += ptr;
2955 } else if (strncmp(ptr, "MAJOR=", strlen("MAJOR=")) == 0) {
2956 ptr += strlen("MAJOR=");
2957 ret.major = atoi(ptr);
2958 } else if (strncmp(ptr, "MINOR=", strlen("MINOR=")) == 0) {
2959 ptr += strlen("MINOR=");
2960 ret.minor = atoi(ptr);
2961 }
2962 ptr += strlen(ptr) + 1;
2963 }
2964 return ret;
2965}
2966
2967void TWPartitionManager::read_uevent() {
2968 char buf[1024];
2969
2970 int len = recv(uevent_pfd.fd, buf, sizeof(buf), MSG_DONTWAIT);
2971 if (len == -1) {
Mauronofrio Matarrese9632f302019-08-04 17:02:41 +01002972 LOGINFO("recv error on uevent\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002973 return;
2974 }
2975 /*int i = 0; // Print all uevent output for test /debug
2976 while (i<len) {
2977 printf("%s\n", buf+i);
2978 i += strlen(buf+i)+1;
2979 }*/
2980 Uevent_Block_Data uevent_data = get_event_block_values(buf, len);
2981 if (uevent_data.subsystem == "block" && uevent_data.type == "disk") {
2982 PartitionManager.Handle_Uevent(uevent_data);
2983 }
2984}
2985
2986void TWPartitionManager::close_uevent() {
2987 if (uevent_pfd.fd > 0)
2988 close(uevent_pfd.fd);
2989 uevent_pfd.fd = -1;
2990}
2991
2992void TWPartitionManager::Add_Partition(TWPartition* Part) {
2993 Partitions.push_back(Part);
2994}
2995
2996void TWPartitionManager::Coldboot_Scan(std::vector<string> *sysfs_entries, const string& Path, int depth) {
2997 string Real_Path = Path;
2998 char real_path[PATH_MAX];
2999 if (realpath(Path.c_str(), &real_path[0])) {
3000 string Real_Path = real_path;
3001 std::vector<string>::iterator iter;
3002 for (iter = sysfs_entries->begin(); iter != sysfs_entries->end(); iter++) {
3003 if (Real_Path.find((*iter)) != string::npos) {
3004 string Write_Path = Real_Path + "/uevent";
3005 if (TWFunc::Path_Exists(Write_Path)) {
3006 const char* write_val = "add\n";
3007 TWFunc::write_to_file(Write_Path, write_val);
3008 break;
3009 }
3010 }
3011 }
3012 }
3013
3014 DIR* d = opendir(Path.c_str());
3015 if (d != NULL) {
3016 struct dirent* de;
3017 while ((de = readdir(d)) != NULL) {
3018 if (de->d_name[0] == '.' || (de->d_type != DT_DIR && depth > 0))
3019 continue;
3020 if (strlen(de->d_name) >= 4 && (strncmp(de->d_name, "ram", 3) == 0 || strncmp(de->d_name, "loop", 4) == 0))
3021 continue;
3022
3023 string item = Path + "/";
3024 item.append(de->d_name);
3025 Coldboot_Scan(sysfs_entries, item, depth + 1);
3026 }
3027 closedir(d);
3028 }
3029}
3030
3031void TWPartitionManager::Coldboot() {
3032 std::vector<TWPartition*>::iterator iter;
3033 std::vector<string> sysfs_entries;
3034
3035 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3036 if (!(*iter)->Sysfs_Entry.empty()) {
3037 size_t wildcard_pos = (*iter)->Sysfs_Entry.find("*");
3038 if (wildcard_pos == string::npos)
3039 wildcard_pos = (*iter)->Sysfs_Entry.size();
3040 sysfs_entries.push_back((*iter)->Sysfs_Entry.substr(0, wildcard_pos));
3041 }
3042 }
3043
3044 if (sysfs_entries.size() > 0)
3045 Coldboot_Scan(&sysfs_entries, "/sys/block", 0);
3046}
Ethan Yonker53796e72019-01-11 22:49:52 -06003047
3048bool TWPartitionManager::Prepare_Empty_Folder(const std::string& Folder) {
3049 if (TWFunc::Path_Exists(Folder))
3050 TWFunc::removeDir(Folder, false);
3051 return TWFunc::Recursive_Mkdir(Folder);
3052}
3053
3054bool TWPartitionManager::Prepare_Repack(TWPartition* Part, const std::string& Temp_Folder_Destination, const bool Create_Backup, const std::string& Backup_Name) {
3055 if (!Part) {
3056 LOGERR("Partition was null!\n");
3057 return false;
3058 }
3059 if (!Prepare_Empty_Folder(Temp_Folder_Destination))
3060 return false;
3061 std::string target_image = Temp_Folder_Destination + "boot.img";
3062 PartitionSettings part_settings;
3063 part_settings.Part = Part;
3064 if (Create_Backup) {
3065 if (Check_Backup_Name(Backup_Name, true, false) != 0)
3066 return false;
3067 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, part_settings.Backup_Folder);
3068 part_settings.Backup_Folder = part_settings.Backup_Folder + "/" + TWFunc::Get_Current_Date() + " " + Backup_Name + "/";
3069 if (!TWFunc::Recursive_Mkdir(part_settings.Backup_Folder))
3070 return false;
3071 } else
3072 part_settings.Backup_Folder = Temp_Folder_Destination;
3073 part_settings.adbbackup = false;
3074 part_settings.generate_digest = false;
3075 part_settings.generate_md5 = false;
3076 part_settings.PM_Method = PM_BACKUP;
3077 part_settings.progress = NULL;
3078 pid_t not_a_pid = 0;
3079 if (!Part->Backup(&part_settings, &not_a_pid))
3080 return false;
3081 std::string backed_up_image = part_settings.Backup_Folder;
3082 backed_up_image += Part->Backup_FileName;
3083 target_image = Temp_Folder_Destination + "boot.img";
3084 if (Create_Backup) {
3085 std::string source = part_settings.Backup_Folder + Part->Backup_FileName;
3086 if (TWFunc::copy_file(source, target_image, 0644) != 0) {
3087 LOGERR("Failed to copy backup file '%s' to temp folder target '%s'\n", source.c_str(), target_image.c_str());
3088 return false;
3089 }
3090 } else {
3091 if (rename(backed_up_image.c_str(), target_image.c_str()) != 0) {
3092 LOGERR("Failed to rename '%s' to '%s'\n", backed_up_image.c_str(), target_image.c_str());
3093 return false;
3094 }
3095 }
3096 return Prepare_Repack(target_image, Temp_Folder_Destination, false, false);
3097}
3098
3099bool TWPartitionManager::Prepare_Repack(const std::string& Source_Path, const std::string& Temp_Folder_Destination, const bool Copy_Source, const bool Create_Destination) {
3100 if (Create_Destination) {
3101 if (!Prepare_Empty_Folder(Temp_Folder_Destination))
3102 return false;
3103 }
3104 if (Copy_Source) {
3105 std::string destination = Temp_Folder_Destination + "/boot.img";
3106 if (TWFunc::copy_file(Source_Path, destination, 0644))
3107 return false;
3108 }
Mauronofrio Matarrese82a2f452019-04-04 22:39:12 +01003109 std::string command = "cd " + Temp_Folder_Destination + " && /sbin/magiskboot --unpack -h '" + Source_Path +"'";
Ethan Yonker53796e72019-01-11 22:49:52 -06003110 if (TWFunc::Exec_Cmd(command) != 0) {
3111 LOGINFO("Error unpacking %s!\n", Source_Path.c_str());
3112 gui_msg(Msg(msg::kError, "unpack_error=Error unpacking image."));
3113 return false;
3114 }
3115 return true;
3116}
3117
3118bool TWPartitionManager::Repack_Images(const std::string& Target_Image, const struct Repack_Options_struct& Repack_Options) {
3119 if (!TWFunc::Path_Exists("/sbin/magiskboot")) {
3120 LOGERR("Image repacking tool not present in this TWRP build!");
3121 return false;
3122 }
3123 DataManager::SetProgress(0);
3124 TWPartition* part = PartitionManager.Find_Partition_By_Path("/boot");
3125 if (part)
3126 gui_msg(Msg("unpacking_image=Unpacking {1}...")(part->Display_Name));
3127 else {
3128 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/boot"));
3129 return false;
3130 }
3131 if (!PartitionManager.Prepare_Repack(part, REPACK_ORIG_DIR, Repack_Options.Backup_First, gui_lookup("repack", "Repack")))
3132 return false;
3133 DataManager::SetProgress(.25);
3134 gui_msg(Msg("unpacking_image=Unpacking {1}...")(Target_Image));
3135 if (!PartitionManager.Prepare_Repack(Target_Image, REPACK_NEW_DIR, true))
3136 return false;
3137 DataManager::SetProgress(.5);
3138 gui_msg(Msg("repacking_image=Repacking {1}...")(part->Display_Name));
3139 std::string path = REPACK_NEW_DIR;
3140 if (Repack_Options.Type == REPLACE_KERNEL) {
3141 // When we replace the kernel, what we really do is copy the boot partition ramdisk into the new image's folder
3142 if (TWFunc::copy_file(REPACK_ORIG_DIR "ramdisk.cpio", REPACK_NEW_DIR "ramdisk.cpio", 0644)) {
3143 LOGERR("Failed to copy ramdisk\n");
3144 return false;
3145 }
3146 } else if (Repack_Options.Type == REPLACE_RAMDISK) {
3147 // Repack the ramdisk
3148 if (TWFunc::copy_file(REPACK_NEW_DIR "ramdisk.cpio", REPACK_ORIG_DIR "ramdisk.cpio", 0644)) {
3149 LOGERR("Failed to copy ramdisk\n");
3150 return false;
3151 }
3152 path = REPACK_ORIG_DIR;
3153 } else {
3154 LOGERR("Invalid repacking options specified\n");
3155 return false;
3156 }
3157 if (Repack_Options.Disable_Verity)
3158 LOGERR("Disabling verity is not implemented yet\n");
3159 if (Repack_Options.Disable_Force_Encrypt)
3160 LOGERR("Disabling force encrypt is not implemented yet\n");
3161 std::string command = "cd " + path + " && /sbin/magiskboot --repack " + path + "boot.img";
3162 if (TWFunc::Exec_Cmd(command) != 0) {
3163 gui_msg(Msg(msg::kError, "repack_error=Error repacking image."));
3164 return false;
3165 }
3166 DataManager::SetProgress(.75);
3167 std::string file = "new-boot.img";
3168 DataManager::SetValue("tw_flash_partition", "/boot;");
3169 if (!PartitionManager.Flash_Image(path, file)) {
3170 LOGINFO("Error flashing new image\n");
3171 return false;
3172 }
3173 DataManager::SetProgress(1);
3174 TWFunc::removeDir(REPACK_ORIG_DIR, false);
Ethan Yonker9f5dd312019-05-15 15:17:36 -05003175 if (part->SlotSelect && Repack_Options.Type == REPLACE_RAMDISK) {
3176 LOGINFO("Switching slots to flash ramdisk to both partitions\n");
3177 string Current_Slot = Get_Active_Slot_Display();
3178 if (Current_Slot == "A")
3179 Set_Active_Slot("B");
3180 else
3181 Set_Active_Slot("A");
3182 DataManager::SetProgress(.25);
3183 if (!PartitionManager.Prepare_Repack(part, REPACK_ORIG_DIR, Repack_Options.Backup_First, gui_lookup("repack", "Repack")))
3184 return false;
3185 if (TWFunc::copy_file(REPACK_NEW_DIR "ramdisk.cpio", REPACK_ORIG_DIR "ramdisk.cpio", 0644)) {
3186 LOGERR("Failed to copy ramdisk\n");
3187 return false;
3188 }
3189 path = REPACK_ORIG_DIR;
3190 command = "cd " + path + " && /sbin/magiskboot --repack " + path + "boot.img";
3191 if (TWFunc::Exec_Cmd(command) != 0) {
3192 gui_msg(Msg(msg::kError, "repack_error=Error repacking image."));
3193 return false;
3194 }
3195 DataManager::SetProgress(.75);
3196 std::string file = "new-boot.img";
3197 DataManager::SetValue("tw_flash_partition", "/boot;");
3198 if (!PartitionManager.Flash_Image(path, file)) {
3199 LOGINFO("Error flashing new image\n");
3200 return false;
3201 }
3202 DataManager::SetProgress(1);
3203 TWFunc::removeDir(REPACK_ORIG_DIR, false);
3204 Set_Active_Slot(Current_Slot);
3205 }
Ethan Yonker53796e72019-01-11 22:49:52 -06003206 TWFunc::removeDir(REPACK_NEW_DIR, false);
3207 return true;
3208}