blob: eb7b6043151053f8cae791b12e14a1118033a520 [file] [log] [blame]
Dees Troy3be70a82013-10-22 14:25:12 +00001/*
bigbiffaed1bdf2021-08-03 18:21:39 -04002 Copyright 2014 to 2021 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>
bigbiff25d25b92020-06-19 16:07:38 -040031#include <grp.h>
32#include <pwd.h>
bigbiffce8f83c2015-12-12 18:30:21 -050033#include <zlib.h>
bigbiff bigbiff9c754052013-01-09 09:09:08 -050034#include <iostream>
35#include <iomanip>
Ethan Yonker8613dc02014-09-11 09:28:20 -050036#include <sys/wait.h>
Ethan Yonker483e9f42016-01-11 22:21:18 -060037#include <linux/fs.h>
38#include <sys/mount.h>
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060039#include <sys/poll.h>
40#include <sys/socket.h>
bigbiffcfa875c2021-06-20 16:20:27 -040041#include <sys/stat.h>
42#include <sys/types.h>
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060043#include <linux/types.h>
44#include <linux/netlink.h>
bigbiffee7b7ff2020-03-23 15:08:27 -040045#include <android-base/chrono_utils.h>
46#include <android-base/file.h>
47#include <android-base/logging.h>
bigbiff7ba75002020-04-11 20:47:09 -040048#include <android-base/properties.h>
bigbiffee7b7ff2020-03-23 15:08:27 -040049#include <android-base/strings.h>
50#include <fstab/fstab.h>
51#include <fs_avb/fs_avb.h>
52#include <fs_mgr.h>
53#include <fs_mgr_dm_linear.h>
54#include <fs_mgr_overlayfs.h>
bigbiffcfa875c2021-06-20 16:20:27 -040055#include <fs_mgr/roots.h>
bigbiffee7b7ff2020-03-23 15:08:27 -040056#include <libgsi/libgsi.h>
57#include <liblp/liblp.h>
bigbiffcfa875c2021-06-20 16:20:27 -040058#include <libgsi/libgsi.h>
59#include <liblp/builder.h>
60#include <libsnapshot/snapshot.h>
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060061
Dees_Troy51a0e822012-09-05 15:24:24 -040062#include "variables.h"
Dees_Troy2673cec2013-04-02 20:22:16 +000063#include "twcommon.h"
Dees_Troy51a0e822012-09-05 15:24:24 -040064#include "partitions.hpp"
Dees_Troy5bf43922012-09-07 16:07:55 -040065#include "data.hpp"
bigbiffaed1bdf2021-08-03 18:21:39 -040066#include "startupArgs.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040067#include "twrp-functions.hpp"
Ethan Yonkerb5fab762016-01-28 14:03:33 -060068#include "fixContexts.hpp"
Ethan Yonker3fdcda42016-11-30 12:29:37 -060069#include "exclude.hpp"
Ethan Yonker4b94cfd2014-12-11 10:00:45 -060070#include "set_metadata.h"
bigbiff7abc5fe2015-01-17 16:53:12 -050071#include "tw_atomic.hpp"
Ethan Yonker74db1572015-10-28 12:44:49 -050072#include "gui/gui.hpp"
Ethan Yonker472f5062016-02-25 13:47:30 -060073#include "progresstracking.hpp"
bigbiff bigbiff56cf5642016-08-19 17:43:45 -040074#include "twrpDigestDriver.hpp"
bigbiffad58e1b2020-07-06 20:24:34 -040075#include "twrpRepacker.hpp"
bigbiffce8f83c2015-12-12 18:30:21 -050076#include "adbbu/libtwadbbu.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040077
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040078#ifdef TW_HAS_MTP
bigbiff bigbiffaf32bb92018-12-18 18:39:53 -050079#ifdef TW_HAS_LEGACY_MTP
80#include "mtp/legacy/mtp_MtpServer.hpp"
81#include "mtp/legacy/twrpMtp.hpp"
82#include "mtp/legacy/MtpMessage.hpp"
83#else
84#include "mtp/ffs/mtp_MtpServer.hpp"
85#include "mtp/ffs/twrpMtp.hpp"
86#include "mtp/ffs/MtpMessage.hpp"
87#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040088#endif
89
Dees Troy6f6441d2014-01-23 02:07:03 +000090extern "C" {
91 #include "cutils/properties.h"
Ethan Yonker74db1572015-10-28 12:44:49 -050092 #include "gui/gui.h"
Dees Troy6f6441d2014-01-23 02:07:03 +000093}
94
Dees_Troy5bf43922012-09-07 16:07:55 -040095#ifdef TW_INCLUDE_CRYPTO
bigbiff7ba75002020-04-11 20:47:09 -040096#include "crypto/fde/cryptfs.h"
97#include "gui/rapidxml.hpp"
98#include "gui/pages.hpp"
99#ifdef TW_INCLUDE_FBE
bigbiffa957f072021-03-07 18:20:29 -0500100#include "Decrypt.h"
bigbiff7ba75002020-04-11 20:47:09 -0400101#ifdef TW_INCLUDE_FBE_METADATA_DECRYPT
102 #ifdef USE_FSCRYPT
bigbiffa957f072021-03-07 18:20:29 -0500103 #include "MetadataCrypt.h"
bigbiffadc599e2020-05-28 19:36:30 +0000104 #endif
bigbiff7ba75002020-04-11 20:47:09 -0400105#endif
106#endif
107#ifdef TW_CRYPTO_USE_SYSTEM_VOLD
108#include "crypto/vold_decrypt/vold_decrypt.h"
109#endif
Dees_Troy5bf43922012-09-07 16:07:55 -0400110#endif
Dees_Troy51a0e822012-09-05 15:24:24 -0400111
Ethan Yonker1b190162016-12-05 15:25:19 -0600112#ifdef AB_OTA_UPDATER
dianlujitaob20dfaa2021-12-16 16:56:26 +0800113#include <android/hardware/boot/1.0/IBootControl.h>
114using android::hardware::boot::V1_0::CommandResult;
115using android::hardware::boot::V1_0::IBootControl;
Ethan Yonker1b190162016-12-05 15:25:19 -0600116#endif
117
bigbiffcfa875c2021-06-20 16:20:27 -0400118using android::fs_mgr::DestroyLogicalPartition;
bigbiffee7b7ff2020-03-23 15:08:27 -0400119using android::fs_mgr::Fstab;
120using android::fs_mgr::FstabEntry;
bigbiffcfa875c2021-06-20 16:20:27 -0400121using android::fs_mgr::MetadataBuilder;
Ethan Yonker6277c792014-09-15 14:54:30 -0500122
bigbiffee7b7ff2020-03-23 15:08:27 -0400123extern bool datamedia;
Noah Jacobson81d638d2019-04-28 00:10:07 -0400124std::vector<users_struct> Users_List;
125
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500126TWPartitionManager::TWPartitionManager(void) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400127 mtp_was_enabled = false;
Ethan Yonker726a0202014-12-16 20:01:38 -0600128 mtp_write_fd = -1;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600129 uevent_pfd.fd = -1;
bigbiff7abc5fe2015-01-17 16:53:12 -0500130 stop_backup.set_value(0);
Ethan Yonker1b190162016-12-05 15:25:19 -0600131#ifdef AB_OTA_UPDATER
132 char slot_suffix[PROPERTY_VALUE_MAX];
Ethan Yonkerfefe5912017-09-30 22:22:13 -0500133 property_get("ro.boot.slot_suffix", slot_suffix, "error");
134 if (strcmp(slot_suffix, "error") == 0)
135 property_get("ro.boot.slot", slot_suffix, "error");
Ethan Yonker1b190162016-12-05 15:25:19 -0600136 Active_Slot_Display = "";
Ethan Yonkerfefe5912017-09-30 22:22:13 -0500137 if (strcmp(slot_suffix, "_a") == 0 || strcmp(slot_suffix, "a") == 0)
Ethan Yonker1b190162016-12-05 15:25:19 -0600138 Set_Active_Slot("A");
139 else
140 Set_Active_Slot("B");
141#endif
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500142}
143
Captain Throwback37c3aef2021-10-06 11:38:18 -0400144void TWPartitionManager::Set_Crypto_State() {
145 char crypto_state[PROPERTY_VALUE_MAX];
146 property_get("ro.crypto.state", crypto_state, "error");
147 if (strcmp(crypto_state, "error") == 0)
148 property_set("ro.crypto.state", "encrypted");
149}
150
151int TWPartitionManager::Set_Crypto_Type(const char* crypto_type) {
152 char type_prop[PROPERTY_VALUE_MAX];
153 property_get("ro.crypto.type", type_prop, "error");
154 if (strcmp(type_prop, "error") == 0)
155 property_set("ro.crypto.type", crypto_type);
Captain Throwback18aea272020-10-29 12:43:17 -0400156 // Sleep for a bit so that services can start if needed
157 sleep(1);
158 return 0;
159}
160
bigbiff22851b92021-09-01 16:46:57 -0400161int TWPartitionManager::Process_Fstab(string Fstab_Filename, bool Display_Error, bool recovery_mode) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400162 FILE *fstabFile;
163 char fstab_line[MAX_FSTAB_LINE_LENGTH];
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600164 std::map<string, Flags_Map> twrp_flags;
165
166 fstabFile = fopen("/etc/twrp.flags", "rt");
167 if (fstabFile != NULL) {
Ian Macdonald160e8d32020-09-29 20:47:31 +0200168 LOGINFO("Reading /etc/twrp.flags\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600169 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600170 size_t line_size = strlen(fstab_line);
171 if (fstab_line[line_size - 1] != '\n')
172 fstab_line[line_size] = '\n';
173 Flags_Map line_flags;
174 line_flags.Primary_Block_Device = "";
175 line_flags.Alternate_Block_Device = "";
176 line_flags.fstab_line = (char*)malloc(MAX_FSTAB_LINE_LENGTH);
177 if (!line_flags.fstab_line) {
178 LOGERR("malloc error on line_flags.fstab_line\n");
179 return false;
180 }
181 memcpy(line_flags.fstab_line, fstab_line, MAX_FSTAB_LINE_LENGTH);
182 bool found_separator = false;
183 char *fs_loc = NULL;
184 char *block_loc = NULL;
185 char *flags_loc = NULL;
186 size_t index, item_index = 0;
187 for (index = 0; index < line_size; index++) {
188 if (fstab_line[index] <= 32) {
189 fstab_line[index] = '\0';
190 found_separator = true;
191 } else if (found_separator) {
192 if (item_index == 0) {
193 fs_loc = fstab_line + index;
194 } else if (item_index == 1) {
195 block_loc = fstab_line + index;
196 } else if (item_index > 1) {
197 char *ptr = fstab_line + index;
198 if (*ptr == '/') {
199 line_flags.Alternate_Block_Device = ptr;
200 } else if (strlen(ptr) > strlen("flags=") && strncmp(ptr, "flags=", strlen("flags=")) == 0) {
201 flags_loc = ptr;
202 // Once we find the flags=, we're done scanning the line
203 break;
204 }
205 }
206 found_separator = false;
207 item_index++;
208 }
209 }
210 if (block_loc)
211 line_flags.Primary_Block_Device = block_loc;
212 if (fs_loc)
213 line_flags.File_System = fs_loc;
214 if (flags_loc)
215 line_flags.Flags = flags_loc;
216 string Mount_Point = fstab_line;
217 twrp_flags[Mount_Point] = line_flags;
218 memset(fstab_line, 0, sizeof(fstab_line));
219 }
220 fclose(fstabFile);
221 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400222
223 fstabFile = fopen(Fstab_Filename.c_str(), "rt");
224 if (fstabFile == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000225 LOGERR("Critical Error: Unable to open fstab at '%s'.\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400226 return false;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600227 } else
228 LOGINFO("Reading %s\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400229
230 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600231 if (strstr(fstab_line, "swap"))
232 continue; // Skip swap in recovery
233
bigbiff998f8392021-08-06 19:19:33 -0400234 if (fstab_line[0] == '#')
235 continue;
236
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600237 size_t line_size = strlen(fstab_line);
238 if (fstab_line[line_size - 1] != '\n')
239 fstab_line[line_size] = '\n';
Dees_Troy2a923582012-09-20 12:13:34 -0400240
Matt Mower2b2dd152016-04-26 11:24:08 -0500241 TWPartition* partition = new TWPartition();
bigbiff8da46fa2020-09-08 16:42:34 -0400242 if (partition->Process_Fstab_Line(fstab_line, Display_Error, &twrp_flags))
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500243 Partitions.push_back(partition);
Matt Mower72c87ce2016-04-26 14:34:56 -0500244 else
Dees_Troy5bf43922012-09-07 16:07:55 -0400245 delete partition;
Matt Mower2b2dd152016-04-26 11:24:08 -0500246
247 memset(fstab_line, 0, sizeof(fstab_line));
Dees_Troy5bf43922012-09-07 16:07:55 -0400248 }
249 fclose(fstabFile);
Matt Mower72c87ce2016-04-26 14:34:56 -0500250
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600251 if (twrp_flags.size() > 0) {
252 LOGINFO("Processing remaining twrp.flags\n");
253 // Add any items from twrp.flags that did not exist in the recovery.fstab
254 for (std::map<string, Flags_Map>::iterator mapit=twrp_flags.begin(); mapit!=twrp_flags.end(); mapit++) {
255 if (Find_Partition_By_Path(mapit->first) == NULL) {
256 TWPartition* partition = new TWPartition();
bigbiff8da46fa2020-09-08 16:42:34 -0400257 if (partition->Process_Fstab_Line(mapit->second.fstab_line, Display_Error, NULL))
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600258 Partitions.push_back(partition);
259 else
260 delete partition;
261 }
262 if (mapit->second.fstab_line)
263 free(mapit->second.fstab_line);
264 mapit->second.fstab_line = NULL;
265 }
266 }
Mohd Faraze3948ec2020-08-14 01:40:59 +0000267 if (Get_Super_Status()) {
268 Setup_Super_Devices();
269 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600270 LOGINFO("Done processing fstab files\n");
271
bigbiff22851b92021-09-01 16:46:57 -0400272 if (recovery_mode) {
273 Setup_Fstab_Partitions(Display_Error);
Mohd Faraze3948ec2020-08-14 01:40:59 +0000274 }
bigbiff7ba75002020-04-11 20:47:09 -0400275 return true;
276}
277
bigbiff22851b92021-09-01 16:46:57 -0400278void TWPartitionManager::Setup_Fstab_Partitions(bool Display_Error) {
279 TWPartition* settings_partition = NULL;
280 TWPartition* andsec_partition = NULL;
281 std::vector<TWPartition*>::iterator iter;
282 unsigned int storageid = 1 << 16; // upper 16 bits are for physical storage device, we pretend to have only one
283
284 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
285 (*iter)->Partition_Post_Processing(Display_Error);
286
287 if ((*iter)->Is_Storage) {
288 ++storageid;
289 (*iter)->MTP_Storage_ID = storageid;
290 }
291
292 if (!settings_partition && (*iter)->Is_Settings_Storage && (*iter)->Is_Present)
293 settings_partition = (*iter);
294 else
295 (*iter)->Is_Settings_Storage = false;
296
297 if (!andsec_partition && (*iter)->Has_Android_Secure && (*iter)->Is_Present)
298 andsec_partition = (*iter);
299 else
300 (*iter)->Has_Android_Secure = false;
301
302 if ((*iter)->Is_Super)
303 Prepare_Super_Volume((*iter));
304 }
305
me-cafebabef29cf9e2022-07-28 03:19:20 +0800306 Unlock_Block_Partitions();
307
bigbiff22851b92021-09-01 16:46:57 -0400308 //Setup Apex before decryption
309 TWPartition* sys = PartitionManager.Find_Partition_By_Path(PartitionManager.Get_Android_Root_Path());
310 TWPartition* ven = PartitionManager.Find_Partition_By_Path("/vendor");
311 if (sys) {
312 if (sys->Get_Super_Status()) {
313 sys->Mount(true);
314 if (ven) {
315 ven->Mount(true);
316 }
317 #ifdef TW_EXCLUDE_APEX
318 LOGINFO("Apex is disabled in this build\n");
319 #else
320 twrpApex apex;
321 if (!apex.loadApexImages()) {
322 LOGERR("Unable to load apex images from %s\n", APEX_DIR);
323 property_set("twrp.apex.loaded", "false");
324 } else {
325 property_set("twrp.apex.loaded", "true");
326 }
327 TWFunc::check_and_run_script("/sbin/resyncapex.sh", "apex");
328 #endif
329 }
330 }
331 #ifndef USE_VENDOR_LIBS
332 if (ven)
333 ven->UnMount(true);
334 if (sys)
335 sys->UnMount(true);
336 #endif
337
338 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) {
339 // Attempt to automatically identify /data/media emulated storage devices
340 TWPartition* Dat = Find_Partition_By_Path("/data");
341 if (Dat) {
342 LOGINFO("Using automatic handling for /data/media emulated storage device.\n");
343 datamedia = true;
344 Dat->Setup_Data_Media();
345 settings_partition = Dat;
346 // Since /data was not considered a storage partition earlier, we still need to assign an MTP ID
347 ++storageid;
348 Dat->MTP_Storage_ID = storageid;
349 }
350 }
351 if (!settings_partition) {
352 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
353 if ((*iter)->Is_Storage) {
354 settings_partition = (*iter);
355 break;
356 }
357 }
358 if (!settings_partition)
359 LOGERR("Unable to locate storage partition for storing settings file.\n");
360 }
361 if (!Write_Fstab()) {
362 if (Display_Error)
363 LOGERR("Error creating fstab\n");
364 else
365 LOGINFO("Error creating fstab\n");
366 }
367
368 if (andsec_partition) {
369 Setup_Android_Secure_Location(andsec_partition);
370 } else if (settings_partition) {
371 Setup_Android_Secure_Location(settings_partition);
372 }
373 if (settings_partition) {
374 Setup_Settings_Storage_Partition(settings_partition);
375 }
376
377 #ifdef TW_INCLUDE_CRYPTO
378 DataManager::SetValue(TW_IS_ENCRYPTED, 1);
379 Decrypt_Data();
380 #endif
381
382 Update_System_Details();
383 if (Get_Super_Status())
384 Setup_Super_Partition();
385 UnMount_Main_Partitions();
386 #ifdef AB_OTA_UPDATER
387 DataManager::SetValue("tw_active_slot", Get_Active_Slot_Display());
388 #endif
389 setup_uevent();
390}
391
bigbiff7ba75002020-04-11 20:47:09 -0400392int TWPartitionManager::Write_Fstab(void) {
393 FILE *fp;
394 std::vector<TWPartition*>::iterator iter;
395 string Line;
396
397 fp = fopen("/etc/fstab", "w");
398 if (fp == NULL) {
399 LOGINFO("Can not open /etc/fstab.\n");
400 return false;
401 }
402 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
403 if ((*iter)->Can_Be_Mounted) {
Mohd Faraz2a1e3752020-09-30 10:55:48 +0530404 Line = (*iter)->Actual_Block_Device + " " + (*iter)->Mount_Point + " " + (*iter)->Current_File_System +
405 ((*iter)->Mount_Read_Only ? " ro " : " rw ") + "0 0\n";
bigbiff7ba75002020-04-11 20:47:09 -0400406 fputs(Line.c_str(), fp);
407 }
408 // Handle subpartition tracking
409 if ((*iter)->Is_SubPartition) {
410 TWPartition* ParentPartition = Find_Partition_By_Path((*iter)->SubPartition_Of);
411 if (ParentPartition)
412 ParentPartition->Has_SubPartition = true;
413 else
414 LOGERR("Unable to locate parent partition '%s' of '%s'\n", (*iter)->SubPartition_Of.c_str(), (*iter)->Mount_Point.c_str());
415 }
416 }
417 fclose(fp);
418 return true;
419}
420
421void TWPartitionManager::Decrypt_Data() {
422 #ifdef TW_INCLUDE_CRYPTO
Ethan Yonkercceebb82014-11-18 10:17:59 -0600423 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
424 if (Decrypt_Data && Decrypt_Data->Is_Encrypted && !Decrypt_Data->Is_Decrypted) {
Captain Throwback37c3aef2021-10-06 11:38:18 -0400425 Set_Crypto_State();
bigbiffcda15882021-12-30 17:01:44 -0500426 TWPartition* Key_Directory_Partition = Find_Partition_By_Path(Decrypt_Data->Key_Directory);
427 if (Key_Directory_Partition != nullptr)
428 if (!Key_Directory_Partition->Is_Mounted())
429 Mount_By_Path(Decrypt_Data->Key_Directory, false);
430 if (!Decrypt_Data->Key_Directory.empty()) {
431 Set_Crypto_Type("file");
Ethan Yonker93382822018-11-01 15:25:31 -0500432#ifdef TW_INCLUDE_FBE_METADATA_DECRYPT
bigbiff7ba75002020-04-11 20:47:09 -0400433#ifdef USE_FSCRYPT
434 if (fscrypt_mount_metadata_encrypted(Decrypt_Data->Actual_Block_Device, Decrypt_Data->Mount_Point, false)) {
bigbiffa957f072021-03-07 18:20:29 -0500435 std::string crypto_blkdev = android::base::GetProperty("ro.crypto.fs_crypto_blkdev", "error");
Mohd Faraze3948ec2020-08-14 01:40:59 +0000436 Decrypt_Data->Decrypted_Block_Device = crypto_blkdev;
bigbiff7ba75002020-04-11 20:47:09 -0400437 LOGINFO("Successfully decrypted metadata encrypted data partition with new block device: '%s'\n", crypto_blkdev.c_str());
bigbiff7ba75002020-04-11 20:47:09 -0400438#endif
Ethan Yonker93382822018-11-01 15:25:31 -0500439 Decrypt_Data->Is_Decrypted = true; // Needed to make the mount function work correctly
440 int retry_count = 10;
441 while (!Decrypt_Data->Mount(false) && --retry_count)
442 usleep(500);
443 if (Decrypt_Data->Mount(false)) {
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400444 if (!Decrypt_Data->Decrypt_FBE_DE()) {
bigbiff437d86f2021-09-26 16:57:14 -0400445 LOGERR("Unable to decrypt FBE device\n");
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400446 }
447
Ethan Yonker93382822018-11-01 15:25:31 -0500448 } else {
449 LOGINFO("Failed to mount data after metadata decrypt\n");
450 }
451 } else {
452 LOGINFO("Unable to decrypt metadata encryption\n");
453 }
454#else
455 LOGERR("Metadata FBE decrypt support not present in this TWRP\n");
456#endif
457 }
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600458 if (Decrypt_Data->Is_FBE) {
459 if (DataManager::GetIntValue(TW_CRYPTO_PWTYPE) == 0) {
460 if (Decrypt_Device("!") == 0) {
461 gui_msg("decrypt_success=Successfully decrypted with default password.");
462 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
463 } else {
464 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
465 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600466 }
467 } else {
bigbiffbbbfe172021-05-30 15:52:41 -0400468 LOGINFO("FBE setup failed. Trying FDE...");
Captain Throwback37c3aef2021-10-06 11:38:18 -0400469 Set_Crypto_State();
470 Set_Crypto_Type("block");
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600471 int password_type = cryptfs_get_password_type();
472 if (password_type == CRYPT_TYPE_DEFAULT) {
473 LOGINFO("Device is encrypted with the default password, attempting to decrypt.\n");
474 if (Decrypt_Device("default_password") == 0) {
475 gui_msg("decrypt_success=Successfully decrypted with default password.");
476 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
477 } else {
478 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
479 }
480 } else {
481 DataManager::SetValue("TW_CRYPTO_TYPE", password_type);
Noah Jacobson81d638d2019-04-28 00:10:07 -0400482 DataManager::SetValue("tw_crypto_pwtype_0", password_type);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600483 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600484 }
485 }
bigbiffcda15882021-12-30 17:01:44 -0500486 if (Decrypt_Data && (!Decrypt_Data->Is_Encrypted || Decrypt_Data->Is_Decrypted)) {
Ethan Yonker66a19492015-12-10 10:19:45 -0600487 Decrypt_Adopted();
488 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600489#endif
Dees_Troy51a0e822012-09-05 15:24:24 -0400490}
491
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500492void TWPartitionManager::Setup_Settings_Storage_Partition(TWPartition* Part) {
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500493 DataManager::SetValue("tw_settings_path", Part->Storage_Path);
494 DataManager::SetValue("tw_storage_path", Part->Storage_Path);
495 LOGINFO("Settings storage is '%s'\n", Part->Storage_Path.c_str());
496}
497
Matt Mowerbf4efa32014-04-14 23:25:26 -0500498void TWPartitionManager::Setup_Android_Secure_Location(TWPartition* Part) {
499 if (Part->Has_Android_Secure)
500 Part->Setup_AndSec();
Ethan Yonker6277c792014-09-15 14:54:30 -0500501 else if (!datamedia)
Matt Mowerbf4efa32014-04-14 23:25:26 -0500502 Part->Setup_AndSec();
Matt Mowerbf4efa32014-04-14 23:25:26 -0500503}
504
Dees_Troy8170a922012-09-18 15:40:25 -0400505void TWPartitionManager::Output_Partition_Logging(void) {
506 std::vector<TWPartition*>::iterator iter;
507
508 printf("\n\nPartition Logs:\n");
509 for (iter = Partitions.begin(); iter != Partitions.end(); iter++)
510 Output_Partition((*iter));
511}
512
513void TWPartitionManager::Output_Partition(TWPartition* Part) {
514 unsigned long long mb = 1048576;
515
Gary Peck004d48b2012-11-21 16:28:18 -0800516 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 -0400517 if (Part->Can_Be_Mounted) {
Gary Peck004d48b2012-11-21 16:28:18 -0800518 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 -0400519 }
Gary Peck004d48b2012-11-21 16:28:18 -0800520 printf("\n Flags: ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500521 if (Part->Can_Be_Mounted)
522 printf("Can_Be_Mounted ");
Gary Peck004d48b2012-11-21 16:28:18 -0800523 if (Part->Can_Be_Wiped)
524 printf("Can_Be_Wiped ");
Hashcodedabfd492013-08-29 22:45:30 -0700525 if (Part->Use_Rm_Rf)
526 printf("Use_Rm_Rf ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500527 if (Part->Can_Be_Backed_Up)
528 printf("Can_Be_Backed_Up ");
Gary Peck004d48b2012-11-21 16:28:18 -0800529 if (Part->Wipe_During_Factory_Reset)
530 printf("Wipe_During_Factory_Reset ");
531 if (Part->Wipe_Available_in_GUI)
532 printf("Wipe_Available_in_GUI ");
533 if (Part->Is_SubPartition)
534 printf("Is_SubPartition ");
535 if (Part->Has_SubPartition)
536 printf("Has_SubPartition ");
537 if (Part->Removable)
538 printf("Removable ");
539 if (Part->Is_Present)
540 printf("IsPresent ");
541 if (Part->Can_Be_Encrypted)
542 printf("Can_Be_Encrypted ");
543 if (Part->Is_Encrypted)
544 printf("Is_Encrypted ");
545 if (Part->Is_Decrypted)
546 printf("Is_Decrypted ");
547 if (Part->Has_Data_Media)
548 printf("Has_Data_Media ");
Dees_Troy83bd4832013-05-04 12:39:56 +0000549 if (Part->Can_Encrypt_Backup)
550 printf("Can_Encrypt_Backup ");
551 if (Part->Use_Userdata_Encryption)
552 printf("Use_Userdata_Encryption ");
Gary Peck004d48b2012-11-21 16:28:18 -0800553 if (Part->Has_Android_Secure)
554 printf("Has_Android_Secure ");
555 if (Part->Is_Storage)
556 printf("Is_Storage ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500557 if (Part->Is_Settings_Storage)
558 printf("Is_Settings_Storage ");
Dees_Troy68cab492012-12-12 19:29:35 +0000559 if (Part->Ignore_Blkid)
560 printf("Ignore_Blkid ");
Ethan Yonker253368a2014-11-25 15:00:52 -0600561 if (Part->Mount_To_Decrypt)
562 printf("Mount_To_Decrypt ");
Ethan Yonker96af84a2015-01-05 14:58:36 -0600563 if (Part->Can_Flash_Img)
564 printf("Can_Flash_Img ");
Ethan Yonker66a19492015-12-10 10:19:45 -0600565 if (Part->Is_Adopted_Storage)
566 printf("Is_Adopted_Storage ");
Ethan Yonker1b190162016-12-05 15:25:19 -0600567 if (Part->SlotSelect)
568 printf("SlotSelect ");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600569 if (Part->Mount_Read_Only)
570 printf("Mount_Read_Only ");
bigbiffcfa875c2021-06-20 16:20:27 -0400571 if (Part->Is_Super)
572 printf("Is_Super ");
Gary Peck004d48b2012-11-21 16:28:18 -0800573 printf("\n");
574 if (!Part->SubPartition_Of.empty())
575 printf(" SubPartition_Of: %s\n", Part->SubPartition_Of.c_str());
576 if (!Part->Symlink_Path.empty())
577 printf(" Symlink_Path: %s\n", Part->Symlink_Path.c_str());
578 if (!Part->Symlink_Mount_Point.empty())
579 printf(" Symlink_Mount_Point: %s\n", Part->Symlink_Mount_Point.c_str());
580 if (!Part->Primary_Block_Device.empty())
581 printf(" Primary_Block_Device: %s\n", Part->Primary_Block_Device.c_str());
582 if (!Part->Alternate_Block_Device.empty())
583 printf(" Alternate_Block_Device: %s\n", Part->Alternate_Block_Device.c_str());
584 if (!Part->Decrypted_Block_Device.empty())
585 printf(" Decrypted_Block_Device: %s\n", Part->Decrypted_Block_Device.c_str());
dianlujitao4879b372018-12-03 18:45:47 +0800586 if (!Part->Crypto_Key_Location.empty())
Ethan Yonker253368a2014-11-25 15:00:52 -0600587 printf(" Crypto_Key_Location: %s\n", Part->Crypto_Key_Location.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800588 if (Part->Length != 0)
589 printf(" Length: %i\n", Part->Length);
590 if (!Part->Display_Name.empty())
591 printf(" Display_Name: %s\n", Part->Display_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500592 if (!Part->Storage_Name.empty())
593 printf(" Storage_Name: %s\n", Part->Storage_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800594 if (!Part->Backup_Path.empty())
595 printf(" Backup_Path: %s\n", Part->Backup_Path.c_str());
596 if (!Part->Backup_Name.empty())
597 printf(" Backup_Name: %s\n", Part->Backup_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500598 if (!Part->Backup_Display_Name.empty())
599 printf(" Backup_Display_Name: %s\n", Part->Backup_Display_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800600 if (!Part->Backup_FileName.empty())
601 printf(" Backup_FileName: %s\n", Part->Backup_FileName.c_str());
602 if (!Part->Storage_Path.empty())
603 printf(" Storage_Path: %s\n", Part->Storage_Path.c_str());
604 if (!Part->Current_File_System.empty())
605 printf(" Current_File_System: %s\n", Part->Current_File_System.c_str());
606 if (!Part->Fstab_File_System.empty())
607 printf(" Fstab_File_System: %s\n", Part->Fstab_File_System.c_str());
608 if (Part->Format_Block_Size != 0)
Ethan Yonkerc798c9c2015-10-09 11:15:26 -0500609 printf(" Format_Block_Size: %lu\n", Part->Format_Block_Size);
Dees_Troy094207a2012-09-26 12:00:39 -0400610 if (!Part->MTD_Name.empty())
611 printf(" MTD_Name: %s\n", Part->MTD_Name.c_str());
Matt Mower029a82d2017-01-08 13:12:38 -0600612 printf(" Backup_Method: %s\n", Part->Backup_Method_By_Name().c_str());
Hashcode62bd9e02013-11-19 21:59:42 -0800613 if (Part->Mount_Flags || !Part->Mount_Options.empty())
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600614 printf(" Mount_Flags: %i, Mount_Options: %s\n", Part->Mount_Flags, Part->Mount_Options.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -0600615 if (Part->MTP_Storage_ID)
616 printf(" MTP_Storage_ID: %i\n", Part->MTP_Storage_ID);
Ethan Yonker93382822018-11-01 15:25:31 -0500617 if (!Part->Key_Directory.empty())
618 printf(" Metadata Key Directory: %s\n", Part->Key_Directory.c_str());
Ethan Yonker6277c792014-09-15 14:54:30 -0500619 printf("\n");
Dees_Troy8170a922012-09-18 15:40:25 -0400620}
621
Dees_Troy51a0e822012-09-05 15:24:24 -0400622int TWPartitionManager::Mount_By_Path(string Path, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400623 std::vector<TWPartition*>::iterator iter;
624 int ret = false;
625 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400626 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400627
bigbiffd58ba182020-03-23 10:02:29 -0400628 if (Local_Path == "/tmp" || Local_Path == "/")
Dees_Troy43d8b002012-09-17 16:00:01 -0400629 return true;
630
Dees_Troy5bf43922012-09-07 16:07:55 -0400631 // Iterate through all partitions
632 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400633 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400634 ret = (*iter)->Mount(Display_Error);
635 found = true;
Dees_Troy51127312012-09-08 13:08:49 -0400636 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400637 (*iter)->Mount(Display_Error);
Dees_Troy51127312012-09-08 13:08:49 -0400638 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400639 }
640 if (found) {
641 return ret;
642 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500643 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 -0400644 }
645 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400646}
647
Dees_Troy51a0e822012-09-05 15:24:24 -0400648int TWPartitionManager::UnMount_By_Path(string Path, bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400649 std::vector<TWPartition*>::iterator iter;
650 int ret = false;
651 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400652 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy51127312012-09-08 13:08:49 -0400653
654 // Iterate through all partitions
655 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400656 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy51127312012-09-08 13:08:49 -0400657 ret = (*iter)->UnMount(Display_Error);
658 found = true;
659 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
660 (*iter)->UnMount(Display_Error);
661 }
662 }
663 if (found) {
664 return ret;
665 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500666 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 -0400667 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000668 LOGINFO("UnMount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400669 }
670 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400671}
672
Dees_Troy51a0e822012-09-05 15:24:24 -0400673int TWPartitionManager::Is_Mounted_By_Path(string Path) {
Dees_Troy51127312012-09-08 13:08:49 -0400674 TWPartition* Part = Find_Partition_By_Path(Path);
675
676 if (Part)
677 return Part->Is_Mounted();
678 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000679 LOGINFO("Is_Mounted: Unable to find partition for path '%s'\n", Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400680 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400681}
682
Dees_Troy5bf43922012-09-07 16:07:55 -0400683int TWPartitionManager::Mount_Current_Storage(bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400684 string current_storage_path = DataManager::GetCurrentStoragePath();
685
686 if (Mount_By_Path(current_storage_path, Display_Error)) {
687 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
688 if (FreeStorage)
689 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
690 return true;
691 }
692 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400693}
694
Dees_Troy5bf43922012-09-07 16:07:55 -0400695int TWPartitionManager::Mount_Settings_Storage(bool Display_Error) {
696 return Mount_By_Path(DataManager::GetSettingsStoragePath(), Display_Error);
697}
698
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600699TWPartition* TWPartitionManager::Find_Partition_By_Path(const string& Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400700 std::vector<TWPartition*>::iterator iter;
Dees_Troy38bd7602012-09-14 13:33:53 -0400701 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400702
dianlujitaob76a73a2020-04-30 20:33:20 +0800703 if (Local_Path == "/system")
704 Local_Path = Get_Android_Root_Path();
Dees_Troy5bf43922012-09-07 16:07:55 -0400705 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400706 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path))
Dees_Troy5bf43922012-09-07 16:07:55 -0400707 return (*iter);
708 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400709 return NULL;
710}
711
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600712TWPartition* TWPartitionManager::Find_Partition_By_Block_Device(const string& Block_Device) {
713 std::vector<TWPartition*>::iterator iter;
714
715 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
716 if ((*iter)->Primary_Block_Device == Block_Device || (!(*iter)->Actual_Block_Device.empty() && (*iter)->Actual_Block_Device == Block_Device))
717 return (*iter);
718 }
719 return NULL;
720}
721
Ethan Yonker53796e72019-01-11 22:49:52 -0600722int TWPartitionManager::Check_Backup_Name(const std::string& Backup_Name, bool Display_Error, bool Must_Be_Unique) {
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400723 // Check the backup name to ensure that it is the correct size and contains only valid characters
724 // and that a backup with that name doesn't already exist
725 char backup_name[MAX_BACKUP_NAME_LEN];
726 char backup_loc[255], tw_image_dir[255];
727 int copy_size;
728 int index, cur_char;
Ethan Yonker53796e72019-01-11 22:49:52 -0600729 string Backup_Loc;
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400730
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400731 copy_size = Backup_Name.size();
732 // Check size
733 if (copy_size > MAX_BACKUP_NAME_LEN) {
734 if (Display_Error)
Ethan Yonker74db1572015-10-28 12:44:49 -0500735 gui_err("backup_name_len=Backup name is too long.");
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400736 return -2;
737 }
738
739 // Check each character
740 strncpy(backup_name, Backup_Name.c_str(), copy_size);
Dees_Troya13d74f2013-03-24 08:54:55 -0500741 if (copy_size == 1 && strncmp(backup_name, "0", 1) == 0)
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400742 return 0; // A "0" (zero) means to use the current timestamp for the backup name
743 for (index=0; index<copy_size; index++) {
744 cur_char = (int)backup_name[index];
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500745 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 -0400746 // These are valid characters
747 // Numbers
748 // Upper case letters
749 // Lower case letters
750 // Space
751 // and -_.{}[]
752 } else {
753 if (Display_Error)
Ethan Yonker4adc33e2016-01-22 16:07:11 -0600754 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 -0400755 return -3;
756 }
757 }
758
Ethan Yonker53796e72019-01-11 22:49:52 -0600759 if (Must_Be_Unique) {
760 // Check to make sure that a backup with this name doesn't already exist
761 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Loc);
762 strcpy(backup_loc, Backup_Loc.c_str());
763 sprintf(tw_image_dir,"%s/%s", backup_loc, Backup_Name.c_str());
764 if (TWFunc::Path_Exists(tw_image_dir)) {
765 if (Display_Error)
766 gui_err("backup_name_exists=A backup with that name already exists!");
bigbiffce8f83c2015-12-12 18:30:21 -0500767
Ethan Yonker53796e72019-01-11 22:49:52 -0600768 return -4;
769 }
770 // Backup is unique
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400771 }
Ethan Yonker53796e72019-01-11 22:49:52 -0600772 // No problems found
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400773 return 0;
774}
775
bigbiffce8f83c2015-12-12 18:30:21 -0500776bool TWPartitionManager::Backup_Partition(PartitionSettings *part_settings) {
Dees_Troy43d8b002012-09-17 16:00:01 -0400777 time_t start, stop;
Matt Mower23d8aae2017-01-06 14:30:33 -0600778 int use_compression;
bigbiffce8f83c2015-12-12 18:30:21 -0500779 string backup_log = part_settings->Backup_Folder + "/recovery.log";
Dees_Troy43d8b002012-09-17 16:00:01 -0400780
bigbiffce8f83c2015-12-12 18:30:21 -0500781 if (part_settings->Part == NULL)
Dees_Troy43d8b002012-09-17 16:00:01 -0400782 return true;
783
Dees_Troy093b7642012-09-21 15:59:38 -0400784 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400785
Tom Hite5a926722014-09-15 01:31:03 +0000786 TWFunc::SetPerformanceMode(true);
Dees_Troy43d8b002012-09-17 16:00:01 -0400787 time(&start);
788
bigbiffce8f83c2015-12-12 18:30:21 -0500789 if (part_settings->Part->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500790 sync();
791 sync();
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400792 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500793 if (!part_settings->adbbackup && part_settings->generate_digest) {
794 if (!twrpDigestDriver::Make_Digest(Full_Filename))
795 goto backup_error;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400796 }
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400797
bigbiffce8f83c2015-12-12 18:30:21 -0500798 if (part_settings->Part->Has_SubPartition) {
Dees_Troy8170a922012-09-18 15:40:25 -0400799 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400800 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -0400801
802 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400803 if ((*subpart)->Can_Be_Backed_Up && (*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500804 part_settings->Part = *subpart;
bigbiffce8f83c2015-12-12 18:30:21 -0500805 if (!(*subpart)->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500806 goto backup_error;
Tom Hite5a926722014-09-15 01:31:03 +0000807 }
Dees_Troy2727b992013-08-14 20:09:30 +0000808 sync();
809 sync();
bigbiff bigbifff5955b12019-05-18 17:28:58 -0400810 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400811 if (!part_settings->adbbackup && part_settings->generate_digest) {
812 if (!twrpDigestDriver::Make_Digest(Full_Filename)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500813 goto backup_error;
bigbiffce8f83c2015-12-12 18:30:21 -0500814 }
Tom Hite5a926722014-09-15 01:31:03 +0000815 }
Dees_Troy8170a922012-09-18 15:40:25 -0400816 }
817 }
818 }
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400819
Dees_Troy43d8b002012-09-17 16:00:01 -0400820 time(&stop);
that203bcc92015-05-09 17:27:33 +0200821 int backup_time = (int) difftime(stop, start);
Dees_Troy2673cec2013-04-02 20:22:16 +0000822 LOGINFO("Partition Backup time: %d\n", backup_time);
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400823
bigbiffce8f83c2015-12-12 18:30:21 -0500824 if (part_settings->Part->Backup_Method == BM_FILES) {
825 part_settings->file_time += backup_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400826 } else {
bigbiffce8f83c2015-12-12 18:30:21 -0500827 part_settings->img_time += backup_time;
828
Dees_Troy43d8b002012-09-17 16:00:01 -0400829 }
Tom Hite5a926722014-09-15 01:31:03 +0000830
Matt Mower02899552016-12-30 18:13:51 -0600831 TWFunc::SetPerformanceMode(false);
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500832 return true;
Dees_Troy43d8b002012-09-17 16:00:01 -0400833 }
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500834backup_error:
835 Clean_Backup_Folder(part_settings->Backup_Folder);
836 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
837 tw_set_default_metadata(backup_log.c_str());
838 TWFunc::SetPerformanceMode(false);
839 return false;
bigbiff7abc5fe2015-01-17 16:53:12 -0500840}
841
bigbiffbf1d6722015-02-14 16:25:59 -0500842void TWPartitionManager::Clean_Backup_Folder(string Backup_Folder) {
843 DIR *d = opendir(Backup_Folder.c_str());
844 struct dirent *p;
845 int r;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400846 vector<string> ext;
847
848 //extensions we should delete when cleaning
849 ext.push_back("win");
850 ext.push_back("md5");
851 ext.push_back("sha2");
852 ext.push_back("info");
bigbiffbf1d6722015-02-14 16:25:59 -0500853
Ethan Yonker74db1572015-10-28 12:44:49 -0500854 gui_msg("backup_clean=Backup Failed. Cleaning Backup Folder.");
bigbiffbf1d6722015-02-14 16:25:59 -0500855
856 if (d == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500857 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Backup_Folder)(strerror(errno)));
bigbiffbf1d6722015-02-14 16:25:59 -0500858 return;
859 }
860
Matt Mower2b18a532015-02-20 16:58:05 -0600861 while ((p = readdir(d))) {
bigbiffbf1d6722015-02-14 16:25:59 -0500862 if (!strcmp(p->d_name, ".") || !strcmp(p->d_name, ".."))
863 continue;
864
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500865 string path = Backup_Folder + "/" + p->d_name;
bigbiffbf1d6722015-02-14 16:25:59 -0500866
867 size_t dot = path.find_last_of(".") + 1;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400868 for (vector<string>::const_iterator i = ext.begin(); i != ext.end(); ++i) {
869 if (path.substr(dot) == *i) {
870 r = unlink(path.c_str());
871 if (r != 0)
872 LOGINFO("Unable to unlink '%s: %s'\n", path.c_str(), strerror(errno));
bigbiffbf1d6722015-02-14 16:25:59 -0500873 }
874 }
875 }
876 closedir(d);
877}
878
Ethan Yonker472f5062016-02-25 13:47:30 -0600879int TWPartitionManager::Check_Backup_Cancel() {
880 return stop_backup.get_value();
881}
882
bigbiff7abc5fe2015-01-17 16:53:12 -0500883int TWPartitionManager::Cancel_Backup() {
884 string Backup_Folder, Backup_Name, Full_Backup_Path;
885
886 stop_backup.set_value(1);
887
888 if (tar_fork_pid != 0) {
889 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
890 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Folder);
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500891 Full_Backup_Path = Backup_Folder + "/" + Backup_Name;
bigbiff7abc5fe2015-01-17 16:53:12 -0500892 LOGINFO("Killing pid: %d\n", tar_fork_pid);
893 kill(tar_fork_pid, SIGUSR2);
894 while (kill(tar_fork_pid, 0) == 0) {
895 usleep(1000);
896 }
897 LOGINFO("Backup_Run stopped and returning false, backup cancelled.\n");
898 LOGINFO("Removing directory %s\n", Full_Backup_Path.c_str());
899 TWFunc::removeDir(Full_Backup_Path, false);
900 tar_fork_pid = 0;
901 }
902
903 return 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400904}
905
bigbiffce8f83c2015-12-12 18:30:21 -0500906int TWPartitionManager::Run_Backup(bool adbbackup) {
907 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400908 int partition_count = 0, disable_free_space_check = 0, skip_digest = 0;
bigbiffce8f83c2015-12-12 18:30:21 -0500909 string Backup_Name, Backup_List, backup_path;
Matt Mower23d8aae2017-01-06 14:30:33 -0600910 unsigned long long total_bytes = 0, free_space = 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400911 TWPartition* storage = NULL;
912 struct tm *t;
Matt Mower23d8aae2017-01-06 14:30:33 -0600913 time_t seconds, total_start, total_stop;
Dees_Troya13d74f2013-03-24 08:54:55 -0500914 size_t start_pos = 0, end_pos = 0;
bigbiff7abc5fe2015-01-17 16:53:12 -0500915 stop_backup.set_value(0);
Dees_Troy43d8b002012-09-17 16:00:01 -0400916 seconds = time(0);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500917 t = localtime(&seconds);
Dees_Troy43d8b002012-09-17 16:00:01 -0400918
bigbiffce8f83c2015-12-12 18:30:21 -0500919 part_settings.img_bytes_remaining = 0;
920 part_settings.file_bytes_remaining = 0;
921 part_settings.img_time = 0;
922 part_settings.file_time = 0;
923 part_settings.img_bytes = 0;
924 part_settings.file_bytes = 0;
925 part_settings.PM_Method = PM_BACKUP;
926
bigbiffce8f83c2015-12-12 18:30:21 -0500927 part_settings.adbbackup = adbbackup;
Dees_Troy43d8b002012-09-17 16:00:01 -0400928 time(&total_start);
929
930 Update_System_Details();
931
932 if (!Mount_Current_Storage(true))
933 return false;
934
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400935 DataManager::GetValue(TW_SKIP_DIGEST_GENERATE_VAR, skip_digest);
936 if (skip_digest == 0)
937 part_settings.generate_digest = true;
Dees_Troyc5865ab2012-09-24 15:08:04 -0400938 else
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400939 part_settings.generate_digest = false;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400940
bigbiffce8f83c2015-12-12 18:30:21 -0500941 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, part_settings.Backup_Folder);
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500942 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
943 if (Backup_Name == gui_lookup("curr_date", "(Current Date)")) {
944 Backup_Name = TWFunc::Get_Current_Date();
945 } else if (Backup_Name == gui_lookup("auto_generate", "(Auto Generate)") || Backup_Name == "0" || Backup_Name.empty()) {
Dees Troyb21cc642013-09-10 17:36:41 +0000946 TWFunc::Auto_Generate_Backup_Name();
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500947 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
Dees_Troy43d8b002012-09-17 16:00:01 -0400948 }
bigbiffce8f83c2015-12-12 18:30:21 -0500949
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500950 LOGINFO("Backup Name is: '%s'\n", Backup_Name.c_str());
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500951 part_settings.Backup_Folder = part_settings.Backup_Folder + "/" + Backup_Name;
bigbiffce8f83c2015-12-12 18:30:21 -0500952
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500953 LOGINFO("Backup_Folder is: '%s'\n", part_settings.Backup_Folder.c_str());
Dees_Troy43d8b002012-09-17 16:00:01 -0400954
Dees_Troy2673cec2013-04-02 20:22:16 +0000955 LOGINFO("Calculating backup details...\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500956 DataManager::GetValue("tw_backup_list", Backup_List);
bigbiff7ba75002020-04-11 20:47:09 -0400957 LOGINFO("Backup_List: %s\n", Backup_List.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500958 if (!Backup_List.empty()) {
959 end_pos = Backup_List.find(";", start_pos);
960 while (end_pos != string::npos && start_pos < Backup_List.size()) {
961 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiff7ba75002020-04-11 20:47:09 -0400962 LOGINFO("backup_path: %s\n", backup_path.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -0500963 part_settings.Part = Find_Partition_By_Path(backup_path);
964 if (part_settings.Part != NULL) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500965 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500966 if (part_settings.Part->Backup_Method == BM_FILES)
967 part_settings.file_bytes += part_settings.Part->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500968 else
bigbiffce8f83c2015-12-12 18:30:21 -0500969 part_settings.img_bytes += part_settings.Part->Backup_Size;
970 if (part_settings.Part->Has_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500971 std::vector<TWPartition*>::iterator subpart;
972
973 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiffce8f83c2015-12-12 18:30:21 -0500974 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 -0500975 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500976 if ((*subpart)->Backup_Method == BM_FILES)
977 part_settings.file_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500978 else
bigbiffce8f83c2015-12-12 18:30:21 -0500979 part_settings.img_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500980 }
981 }
Dees_Troy8170a922012-09-18 15:40:25 -0400982 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500983 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500984 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 -0400985 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500986 start_pos = end_pos + 1;
987 end_pos = Backup_List.find(";", start_pos);
Dees_Troy43d8b002012-09-17 16:00:01 -0400988 }
989 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400990
991 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500992 gui_msg("no_partition_selected=No partitions selected for backup.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400993 return false;
994 }
bigbiffce8f83c2015-12-12 18:30:21 -0500995 if (adbbackup) {
996 if (twadbbu::Write_ADB_Stream_Header(partition_count) == false) {
997 return false;
998 }
999 }
1000 total_bytes = part_settings.file_bytes + part_settings.img_bytes;
Ethan Yonker472f5062016-02-25 13:47:30 -06001001 ProgressTracking progress(total_bytes);
bigbiffce8f83c2015-12-12 18:30:21 -05001002 part_settings.progress = &progress;
1003
Ethan Yonker74db1572015-10-28 12:44:49 -05001004 gui_msg(Msg("total_partitions_backup= * Total number of partitions to back up: {1}")(partition_count));
1005 gui_msg(Msg("total_backup_size= * Total size of all data: {1}MB")(total_bytes / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -04001006 storage = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
1007 if (storage != NULL) {
1008 free_space = storage->Free;
Ethan Yonker74db1572015-10-28 12:44:49 -05001009 gui_msg(Msg("available_space= * Available space: {1}MB")(free_space / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -04001010 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001011 gui_err("unable_locate_storage=Unable to locate storage device.");
Dees_Troy43d8b002012-09-17 16:00:01 -04001012 return false;
1013 }
bigbiffbf1d6722015-02-14 16:25:59 -05001014
Matt Mowerbfccfb82016-04-25 23:22:31 -05001015 DataManager::GetValue(TW_DISABLE_FREE_SPACE_VAR, disable_free_space_check);
bigbiffce8f83c2015-12-12 18:30:21 -05001016
1017 if (adbbackup)
1018 disable_free_space_check = true;
1019
bigbiffbf1d6722015-02-14 16:25:59 -05001020 if (!disable_free_space_check) {
1021 if (free_space - (32 * 1024 * 1024) < total_bytes) {
1022 // We require an extra 32MB just in case
Ethan Yonker74db1572015-10-28 12:44:49 -05001023 gui_err("no_space=Not enough free space on storage.");
bigbiffbf1d6722015-02-14 16:25:59 -05001024 return false;
1025 }
Dees_Troy43d8b002012-09-17 16:00:01 -04001026 }
bigbiffce8f83c2015-12-12 18:30:21 -05001027 part_settings.img_bytes_remaining = part_settings.img_bytes;
1028 part_settings.file_bytes_remaining = part_settings.file_bytes;
Dees_Troy43d8b002012-09-17 16:00:01 -04001029
Ethan Yonker74db1572015-10-28 12:44:49 -05001030 gui_msg("backup_started=[BACKUP STARTED]");
bigbiff bigbiffcdd97c72019-03-29 19:12:33 -04001031
1032 int is_decrypted = 0;
1033 int is_encrypted = 0;
1034
1035 DataManager::GetValue(TW_IS_DECRYPTED, is_decrypted);
1036 DataManager::GetValue(TW_IS_ENCRYPTED, is_encrypted);
1037 if (!adbbackup || (!is_encrypted || (is_encrypted && is_decrypted))) {
1038 gui_msg(Msg("backup_folder= * Backup Folder: {1}")(part_settings.Backup_Folder));
1039 if (!TWFunc::Recursive_Mkdir(part_settings.Backup_Folder)) {
1040 gui_err("fail_backup_folder=Failed to make backup folder.");
1041 return false;
1042 }
Dees_Troyd4b22b02013-01-18 17:17:58 +00001043 }
1044
Dees_Troy2673cec2013-04-02 20:22:16 +00001045 DataManager::SetProgress(0.0);
Dees_Troy093b7642012-09-21 15:59:38 -04001046
Dees_Troya13d74f2013-03-24 08:54:55 -05001047 start_pos = 0;
1048 end_pos = Backup_List.find(";", start_pos);
1049 while (end_pos != string::npos && start_pos < Backup_List.size()) {
bigbiff7abc5fe2015-01-17 16:53:12 -05001050 if (stop_backup.get_value() != 0)
1051 return -1;
Dees_Troya13d74f2013-03-24 08:54:55 -05001052 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001053 part_settings.Part = Find_Partition_By_Path(backup_path);
1054 if (part_settings.Part != NULL) {
1055 if (!Backup_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001056 return false;
1057 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001058 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 -05001059 }
1060 start_pos = end_pos + 1;
1061 end_pos = Backup_List.find(";", start_pos);
1062 }
Dees_Troy43d8b002012-09-17 16:00:01 -04001063
1064 // Average BPS
bigbiffce8f83c2015-12-12 18:30:21 -05001065 if (part_settings.img_time == 0)
1066 part_settings.img_time = 1;
1067 if (part_settings.file_time == 0)
1068 part_settings.file_time = 1;
1069 int img_bps = (int)part_settings.img_bytes / (int)part_settings.img_time;
1070 unsigned long long file_bps = part_settings.file_bytes / (int)part_settings.file_time;
Dees_Troy43d8b002012-09-17 16:00:01 -04001071
bigbiffce8f83c2015-12-12 18:30:21 -05001072 if (part_settings.file_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -06001073 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 -05001074 if (part_settings.img_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -06001075 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 -04001076
1077 time(&total_stop);
1078 int total_time = (int) difftime(total_stop, total_start);
bigbiffce8f83c2015-12-12 18:30:21 -05001079
1080 uint64_t actual_backup_size;
Ethan Yonker3fdcda42016-11-30 12:29:37 -06001081 if (!adbbackup) {
1082 TWExclude twe;
1083 actual_backup_size = twe.Get_Folder_Size(part_settings.Backup_Folder);
1084 } else
bigbiffce8f83c2015-12-12 18:30:21 -05001085 actual_backup_size = part_settings.file_bytes + part_settings.img_bytes;
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001086 actual_backup_size /= (1024LLU * 1024LLU);
Dees_Troy43d8b002012-09-17 16:00:01 -04001087
bigbiffce8f83c2015-12-12 18:30:21 -05001088 int prev_img_bps = 0, use_compression = 0;
1089 unsigned long long prev_file_bps = 0;
Matt Mower173cdb92016-12-31 02:49:19 -06001090 DataManager::GetValue(TW_BACKUP_AVG_IMG_RATE, prev_img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -04001091 img_bps += (prev_img_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001092 img_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -04001093
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001094 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -04001095 if (use_compression)
1096 DataManager::GetValue(TW_BACKUP_AVG_FILE_COMP_RATE, prev_file_bps);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001097 else
Dees_Troy093b7642012-09-21 15:59:38 -04001098 DataManager::GetValue(TW_BACKUP_AVG_FILE_RATE, prev_file_bps);
1099 file_bps += (prev_file_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001100 file_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -04001101
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001102 DataManager::SetValue(TW_BACKUP_AVG_IMG_RATE, img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -04001103 if (use_compression)
1104 DataManager::SetValue(TW_BACKUP_AVG_FILE_COMP_RATE, file_bps);
1105 else
1106 DataManager::SetValue(TW_BACKUP_AVG_FILE_RATE, file_bps);
1107
Ethan Yonker74db1572015-10-28 12:44:49 -05001108 gui_msg(Msg("total_backed_size=[{1} MB TOTAL BACKED UP]")(actual_backup_size));
Dees_Troy43d8b002012-09-17 16:00:01 -04001109 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001110 UnMount_Main_Partitions();
Ethan Yonker56db1c42015-12-20 22:49:00 -06001111 gui_msg(Msg(msg::kHighlight, "backup_completed=[BACKUP COMPLETED IN {1} SECONDS]")(total_time)); // the end
Ethan Yonkere080c1f2016-09-19 13:50:25 -05001112 string backup_log = part_settings.Backup_Folder + "/recovery.log";
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001113 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
Ethan Yonker4b94cfd2014-12-11 10:00:45 -06001114 tw_set_default_metadata(backup_log.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -05001115
1116 if (part_settings.adbbackup) {
1117 if (twadbbu::Write_ADB_Stream_Trailer() == false) {
1118 return false;
1119 }
1120 }
1121 part_settings.adbbackup = false;
1122 DataManager::SetValue("tw_enable_adb_backup", 0);
1123
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001124 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001125}
1126
bigbiffce8f83c2015-12-12 18:30:21 -05001127bool TWPartitionManager::Restore_Partition(PartitionSettings *part_settings) {
Dees_Troy4a2a1262012-09-18 09:33:47 -04001128 time_t Start, Stop;
bigbiffce8f83c2015-12-12 18:30:21 -05001129
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001130 if (part_settings->adbbackup) {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001131 std::string partName = part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win";
1132 LOGINFO("setting backup name: %s\n", partName.c_str());
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001133 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
1134 }
1135
Tom Hite5a926722014-09-15 01:31:03 +00001136 TWFunc::SetPerformanceMode(true);
bigbiffce8f83c2015-12-12 18:30:21 -05001137
Dees_Troy4a2a1262012-09-18 09:33:47 -04001138 time(&Start);
Ethan Yonker472f5062016-02-25 13:47:30 -06001139
bigbiffce8f83c2015-12-12 18:30:21 -05001140 if (!part_settings->Part->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001141 TWFunc::SetPerformanceMode(false);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001142 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001143 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001144 if (part_settings->Part->Has_SubPartition && !part_settings->adbbackup) {
Dees_Troy8170a922012-09-18 15:40:25 -04001145 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001146 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -04001147
1148 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001149 part_settings->Part = *subpart;
1150 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
1151 part_settings->Part = (*subpart);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001152 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
bigbiffce8f83c2015-12-12 18:30:21 -05001153 if (!(*subpart)->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001154 TWFunc::SetPerformanceMode(false);
Dees_Troy8170a922012-09-18 15:40:25 -04001155 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001156 }
Dees_Troy8170a922012-09-18 15:40:25 -04001157 }
1158 }
1159 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001160 time(&Stop);
Tom Hite5a926722014-09-15 01:31:03 +00001161 TWFunc::SetPerformanceMode(false);
bigbiffce8f83c2015-12-12 18:30:21 -05001162 gui_msg(Msg("restore_part_done=[{1} done ({2} seconds)]")(part_settings->Part->Backup_Display_Name)((int)difftime(Stop, Start)));
1163
Dees_Troy4a2a1262012-09-18 09:33:47 -04001164 return true;
1165}
1166
Ethan Yonker472f5062016-02-25 13:47:30 -06001167int TWPartitionManager::Run_Restore(const string& Restore_Name) {
bigbiffce8f83c2015-12-12 18:30:21 -05001168 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001169 int check_digest;
bigbiffce8f83c2015-12-12 18:30:21 -05001170
Dees_Troy4a2a1262012-09-18 09:33:47 -04001171 time_t rStart, rStop;
1172 time(&rStart);
Dees_Troya13d74f2013-03-24 08:54:55 -05001173 string Restore_List, restore_path;
1174 size_t start_pos = 0, end_pos;
bigbiffce8f83c2015-12-12 18:30:21 -05001175
Ethan Yonkerdcf2b672016-09-13 14:41:53 -05001176 part_settings.Backup_Folder = Restore_Name;
bigbiffce8f83c2015-12-12 18:30:21 -05001177 part_settings.Part = NULL;
1178 part_settings.partition_count = 0;
1179 part_settings.total_restore_size = 0;
1180 part_settings.adbbackup = false;
1181 part_settings.PM_Method = PM_RESTORE;
Dees_Troy43d8b002012-09-17 16:00:01 -04001182
Ethan Yonker74db1572015-10-28 12:44:49 -05001183 gui_msg("restore_started=[RESTORE STARTED]");
1184 gui_msg(Msg("restore_folder=Restore folder: '{1}'")(Restore_Name));
Dees_Troy43d8b002012-09-17 16:00:01 -04001185
Dees_Troy4a2a1262012-09-18 09:33:47 -04001186 if (!Mount_Current_Storage(true))
1187 return false;
1188
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001189 DataManager::GetValue(TW_SKIP_DIGEST_CHECK_VAR, check_digest);
1190 if (check_digest > 0) {
1191 // Check Digest files first before restoring to ensure that all of them match before starting a restore
1192 TWFunc::GUI_Operation_Text(TW_VERIFY_DIGEST_TEXT, gui_parse_text("{@verifying_digest}"));
1193 gui_msg("verifying_digest=Verifying Digest");
Dees_Troya13d74f2013-03-24 08:54:55 -05001194 } else {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001195 gui_msg("skip_digest=Skipping Digest check based on user setting.");
Dees_Troya13d74f2013-03-24 08:54:55 -05001196 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001197 gui_msg("calc_restore=Calculating restore details...");
Dees_Troya13d74f2013-03-24 08:54:55 -05001198 DataManager::GetValue("tw_restore_selected", Restore_List);
bigbiffce8f83c2015-12-12 18:30:21 -05001199
Dees_Troya13d74f2013-03-24 08:54:55 -05001200 if (!Restore_List.empty()) {
1201 end_pos = Restore_List.find(";", start_pos);
1202 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1203 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001204 part_settings.Part = Find_Partition_By_Path(restore_path);
1205 if (part_settings.Part != NULL) {
bigbiffce8f83c2015-12-12 18:30:21 -05001206 if (part_settings.Part->Mount_Read_Only) {
1207 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 -05001208 return false;
1209 }
bigbiffce8f83c2015-12-12 18:30:21 -05001210
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001211 string Full_Filename = part_settings.Backup_Folder + "/" + part_settings.Part->Backup_FileName;
1212
bigbiff4a60bee2021-01-23 14:08:03 -05001213 if (tw_get_default_metadata(Get_Android_Root_Path().c_str()) != 0) {
1214 gui_msg(Msg(msg::kWarning, "restore_system_context=Unable to get default context for {1} -- Android may not boot.")(Get_Android_Root_Path()));
1215 }
1216
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001217 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001218 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001219 part_settings.partition_count++;
1220 part_settings.total_restore_size += part_settings.Part->Get_Restore_Size(&part_settings);
1221 if (part_settings.Part->Has_SubPartition) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001222 TWPartition *parentPart = part_settings.Part;
Dees_Troya13d74f2013-03-24 08:54:55 -05001223 std::vector<TWPartition*>::iterator subpart;
1224
1225 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001226 part_settings.Part = *subpart;
1227 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001228 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001229 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001230 part_settings.total_restore_size += (*subpart)->Get_Restore_Size(&part_settings);
Dees_Troya13d74f2013-03-24 08:54:55 -05001231 }
1232 }
1233 }
1234 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001235 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001236 }
1237 start_pos = end_pos + 1;
1238 end_pos = Restore_List.find(";", start_pos);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001239 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001240 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001241
bigbiffce8f83c2015-12-12 18:30:21 -05001242 if (part_settings.partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001243 gui_err("no_part_restore=No partitions selected for restore.");
Dees_Troy4a2a1262012-09-18 09:33:47 -04001244 return false;
1245 }
1246
bigbiffce8f83c2015-12-12 18:30:21 -05001247 gui_msg(Msg("restore_part_count=Restoring {1} partitions...")(part_settings.partition_count));
1248 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 +00001249 DataManager::SetProgress(0.0);
bigbiffce8f83c2015-12-12 18:30:21 -05001250 ProgressTracking progress(part_settings.total_restore_size);
1251 part_settings.progress = &progress;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001252
Dees_Troya13d74f2013-03-24 08:54:55 -05001253 start_pos = 0;
1254 if (!Restore_List.empty()) {
1255 end_pos = Restore_List.find(";", start_pos);
1256 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1257 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001258
1259 part_settings.Part = Find_Partition_By_Path(restore_path);
1260 if (part_settings.Part != NULL) {
1261 part_settings.partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -05001262 if (!Restore_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001263 return false;
1264 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001265 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001266 }
1267 start_pos = end_pos + 1;
1268 end_pos = Restore_List.find(";", start_pos);
1269 }
1270 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001271 TWFunc::GUI_Operation_Text(TW_UPDATE_SYSTEM_DETAILS_TEXT, gui_parse_text("{@updating_system_details}"));
bigbiff4a60bee2021-01-23 14:08:03 -05001272 tw_set_default_metadata(Get_Android_Root_Path().c_str());
Captain Throwback9d6feb52018-07-27 10:05:24 -04001273 UnMount_By_Path(Get_Android_Root_Path(), false);
Dees_Troy43d8b002012-09-17 16:00:01 -04001274 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001275 UnMount_Main_Partitions();
Dees_Troy4a2a1262012-09-18 09:33:47 -04001276 time(&rStop);
Matt Mower3c366972015-12-25 19:28:31 -06001277 gui_msg(Msg(msg::kHighlight, "restore_completed=[RESTORE COMPLETED IN {1} SECONDS]")((int)difftime(rStop,rStart)));
bigbiff8f5fad32021-11-03 20:39:02 -04001278 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
1279 if (Decrypt_Data && Decrypt_Data->Is_Encrypted)
1280 gui_msg(Msg(msg::kWarning, "reboot_after_restore=It is recommended to reboot Android once after first boot."));
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001281 DataManager::SetValue("tw_file_progress", "");
bigbiffce8f83c2015-12-12 18:30:21 -05001282
Dees_Troy63c8df72012-09-10 14:02:05 -04001283 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001284}
1285
1286void TWPartitionManager::Set_Restore_Files(string Restore_Name) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001287 // Start with the default values
Dees_Troya13d74f2013-03-24 08:54:55 -05001288 string Restore_List;
Dees_Troy83bd4832013-05-04 12:39:56 +00001289 bool get_date = true, check_encryption = true;
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001290 bool adbbackup = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001291
1292 DataManager::SetValue("tw_restore_encrypted", 0);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001293 if (twadbbu::Check_ADB_Backup_File(Restore_Name)) {
1294 vector<string> adb_files;
1295 adb_files = twadbbu::Get_ADB_Backup_Files(Restore_Name);
1296 for (unsigned int i = 0; i < adb_files.size(); ++i) {
1297 string adb_restore_file = adb_files.at(i);
1298 std::size_t pos = adb_restore_file.find_first_of(".");
1299 std::string path = "/" + adb_restore_file.substr(0, pos);
1300 Restore_List = path + ";";
1301 TWPartition* Part = Find_Partition_By_Path(path);
1302 Part->Backup_FileName = TWFunc::Get_Filename(adb_restore_file);
1303 adbbackup = true;
1304 }
1305 DataManager::SetValue("tw_enable_adb_backup", 1);
Dees_Troy63c8df72012-09-10 14:02:05 -04001306 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001307 else {
1308 DIR* d;
1309 d = opendir(Restore_Name.c_str());
1310 if (d == NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001311 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001312 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Restore_Name)(strerror(errno)));
1313 return;
Dees_Troy63c8df72012-09-10 14:02:05 -04001314 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001315
1316 struct dirent* de;
1317 while ((de = readdir(d)) != NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001318 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001319 // Strip off three components
1320 char str[256];
1321 char* label;
1322 char* fstype = NULL;
1323 char* extn = NULL;
1324 char* ptr;
Dees_Troy63c8df72012-09-10 14:02:05 -04001325
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001326 strcpy(str, de->d_name);
1327 if (strlen(str) <= 2)
1328 continue;
Dees_Troy83bd4832013-05-04 12:39:56 +00001329
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001330 if (get_date) {
1331 char file_path[255];
1332 struct stat st;
1333
1334 strcpy(file_path, Restore_Name.c_str());
1335 strcat(file_path, "/");
1336 strcat(file_path, str);
1337 stat(file_path, &st);
1338 string backup_date = ctime((const time_t*)(&st.st_mtime));
1339 DataManager::SetValue(TW_RESTORE_FILE_DATE, backup_date);
1340 get_date = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001341 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001342
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001343 label = str;
1344 ptr = label;
1345 while (*ptr && *ptr != '.') ptr++;
1346 if (*ptr == '.')
1347 {
1348 *ptr = 0x00;
1349 ptr++;
1350 fstype = ptr;
1351 }
1352 while (*ptr && *ptr != '.') ptr++;
1353 if (*ptr == '.')
1354 {
1355 *ptr = 0x00;
1356 ptr++;
1357 extn = ptr;
1358 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001359
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001360 if (fstype == NULL || extn == NULL || strcmp(fstype, "log") == 0) continue;
1361 int extnlength = strlen(extn);
1362 if (extnlength != 3 && extnlength != 6) continue;
1363 if (extnlength >= 3 && strncmp(extn, "win", 3) != 0) continue;
1364 //if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
Dees_Troy63c8df72012-09-10 14:02:05 -04001365
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001366 if (check_encryption) {
1367 string filename = Restore_Name + "/";
1368 filename += de->d_name;
1369 if (TWFunc::Get_File_Type(filename) == 2) {
1370 LOGINFO("'%s' is encrypted\n", filename.c_str());
1371 DataManager::SetValue("tw_restore_encrypted", 1);
1372 }
1373 }
1374 if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
1375
1376 TWPartition* Part = Find_Partition_By_Path(label);
1377 if (Part == NULL)
1378 {
1379 gui_msg(Msg(msg::kError, "unable_locate_part_backup_name=Unable to locate partition by backup name: '{1}'")(label));
1380 continue;
1381 }
1382
1383 Part->Backup_FileName = de->d_name;
1384 if (strlen(extn) > 3) {
1385 Part->Backup_FileName.resize(Part->Backup_FileName.size() - strlen(extn) + 3);
1386 }
1387
dianlujitaob76a73a2020-04-30 20:33:20 +08001388 if (!Part->Is_SubPartition) {
1389 if (Part->Backup_Path == Get_Android_Root_Path())
1390 Restore_List += "/system;";
1391 else
1392 Restore_List += Part->Backup_Path + ";";
1393 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001394 }
1395 closedir(d);
Dees_Troy63c8df72012-09-10 14:02:05 -04001396 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001397
1398 if (adbbackup) {
1399 Restore_List = "ADB_Backup;";
1400 adbbackup = false;
1401 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001402
Dees_Troya13d74f2013-03-24 08:54:55 -05001403 // Set the final value
1404 DataManager::SetValue("tw_restore_list", Restore_List);
1405 DataManager::SetValue("tw_restore_selected", Restore_List);
Dees_Troy51a0e822012-09-05 15:24:24 -04001406 return;
1407}
1408
1409int TWPartitionManager::Wipe_By_Path(string Path) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001410 std::vector<TWPartition*>::iterator iter;
Mohd Farazfdd49bb2020-04-21 18:56:44 +05301411 std::vector < TWPartition * >::iterator iter1;
Dees_Troy63c8df72012-09-10 14:02:05 -04001412 int ret = false;
1413 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -04001414 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy63c8df72012-09-10 14:02:05 -04001415
dianlujitaob76a73a2020-04-30 20:33:20 +08001416 if (Local_Path == "/system")
1417 Local_Path = Get_Android_Root_Path();
bigbiffaed1bdf2021-08-03 18:21:39 -04001418 if (Path == "/cache") {
1419 TWPartition* cache = Find_Partition_By_Path("/cache");
1420 if (cache == nullptr) {
1421 TWPartition* dat = Find_Partition_By_Path("/data");
1422 if (dat) {
1423 dat->Wipe_Data_Cache();
1424 found = true;
1425 }
1426 }
1427 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001428 // Iterate through all partitions
1429 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -04001430 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Mohd Farazfdd49bb2020-04-21 18:56:44 +05301431 // iterate through all partitions since some legacy devices uses other partitions as vendor causes issues while wiping
1432 (*iter)->Find_Actual_Block_Device();
1433 for (iter1 = Partitions.begin (); iter1 != Partitions.end (); iter1++)
1434 {
1435 (*iter1)->Find_Actual_Block_Device();
1436 if ((*iter)->Actual_Block_Device == (*iter1)->Actual_Block_Device && (*iter)->Mount_Point != (*iter1)->Mount_Point)
1437 (*iter1)->UnMount(false);
1438 }
Dees_Troye58d5262012-09-21 12:27:57 -04001439 if (Path == "/and-sec")
1440 ret = (*iter)->Wipe_AndSec();
1441 else
1442 ret = (*iter)->Wipe();
Dees_Troy63c8df72012-09-10 14:02:05 -04001443 found = true;
1444 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1445 (*iter)->Wipe();
1446 }
1447 }
1448 if (found) {
1449 return ret;
1450 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001451 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 -04001452 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -04001453}
1454
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001455int TWPartitionManager::Wipe_By_Path(string Path, string New_File_System) {
1456 std::vector<TWPartition*>::iterator iter;
1457 int ret = false;
1458 bool found = false;
1459 string Local_Path = TWFunc::Get_Root_Path(Path);
1460
1461 // Iterate through all partitions
1462 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1463 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1464 if (Path == "/and-sec")
1465 ret = (*iter)->Wipe_AndSec();
1466 else
1467 ret = (*iter)->Wipe(New_File_System);
1468 found = true;
1469 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1470 (*iter)->Wipe(New_File_System);
1471 }
1472 }
1473 if (found) {
1474 return ret;
1475 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001476 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 -05001477 return false;
1478}
1479
Dees_Troy51a0e822012-09-05 15:24:24 -04001480int TWPartitionManager::Factory_Reset(void) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001481 std::vector<TWPartition*>::iterator iter;
1482 int ret = true;
1483
1484 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001485 if ((*iter)->Wipe_During_Factory_Reset && (*iter)->Is_Present) {
Ethan Yonker89583ef2015-08-26 09:01:59 -05001486#ifdef TW_OEM_BUILD
1487 if ((*iter)->Mount_Point == "/data") {
1488 if (!(*iter)->Wipe_Encryption())
1489 ret = false;
1490 } else {
1491#endif
1492 if (!(*iter)->Wipe())
1493 ret = false;
1494#ifdef TW_OEM_BUILD
1495 }
1496#endif
Dees_Troy094207a2012-09-26 12:00:39 -04001497 } else if ((*iter)->Has_Android_Secure) {
1498 if (!(*iter)->Wipe_AndSec())
1499 ret = false;
Dees_Troy63c8df72012-09-10 14:02:05 -04001500 }
1501 }
bigbiffad58e1b2020-07-06 20:24:34 -04001502 TWFunc::check_and_run_script("/system/bin/factoryreset.sh", "Factory Reset Script");
Dees_Troy63c8df72012-09-10 14:02:05 -04001503 return ret;
Dees_Troy51a0e822012-09-05 15:24:24 -04001504}
1505
Dees_Troy38bd7602012-09-14 13:33:53 -04001506int TWPartitionManager::Wipe_Dalvik_Cache(void) {
1507 struct stat st;
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001508 vector <string> dir;
Dees_Troy38bd7602012-09-14 13:33:53 -04001509
1510 if (!Mount_By_Path("/data", true))
1511 return false;
1512
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001513 dir.push_back("/data/dalvik-cache");
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001514
bigbiff25d25b92020-06-19 16:07:38 -04001515 std::string cacheDir = TWFunc::get_log_dir();
1516 if (cacheDir == CACHE_LOGS_DIR) {
1517 if (!PartitionManager.Mount_By_Path(CACHE_LOGS_DIR, false)) {
1518 LOGINFO("Unable to mount %s for wiping cache.\n", CACHE_LOGS_DIR);
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001519 }
1520 dir.push_back(cacheDir + "dalvik-cache");
1521 dir.push_back(cacheDir + "/dc");
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001522 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001523
Dees_Troy38bd7602012-09-14 13:33:53 -04001524 TWPartition* sdext = Find_Partition_By_Path("/sd-ext");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001525 if (sdext && sdext->Is_Present && sdext->Mount(false))
1526 {
1527 if (stat("/sd-ext/dalvik-cache", &st) == 0)
1528 {
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001529 dir.push_back("/sd-ext/dalvik-cache");
1530 }
1531 }
1532
bigbiff25d25b92020-06-19 16:07:38 -04001533 if (cacheDir == CACHE_LOGS_DIR) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001534 gui_msg("wiping_cache_dalvik=Wiping Dalvik Cache Directories...");
1535 } else {
1536 gui_msg("wiping_dalvik=Wiping Dalvik Directory...");
1537 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001538 for (unsigned i = 0; i < dir.size(); ++i) {
1539 if (stat(dir.at(i).c_str(), &st) == 0) {
1540 TWFunc::removeDir(dir.at(i), false);
1541 gui_msg(Msg("cleaned=Cleaned: {1}...")(dir.at(i)));
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001542 }
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001543 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001544
bigbiff25d25b92020-06-19 16:07:38 -04001545 if (cacheDir == CACHE_LOGS_DIR) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001546 gui_msg("cache_dalvik_done=-- Dalvik Cache Directories Wipe Complete!");
1547 } else {
1548 gui_msg("dalvik_done=-- Dalvik Directory Wipe Complete!");
1549 }
1550
Dees_Troy38bd7602012-09-14 13:33:53 -04001551 return true;
1552}
1553
1554int TWPartitionManager::Wipe_Rotate_Data(void) {
1555 if (!Mount_By_Path("/data", true))
1556 return false;
1557
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001558 unlink("/data/misc/akmd*");
1559 unlink("/data/misc/rild*");
Dees_Troy2673cec2013-04-02 20:22:16 +00001560 gui_print("Rotation data wiped.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001561 return true;
1562}
1563
1564int TWPartitionManager::Wipe_Battery_Stats(void) {
1565 struct stat st;
1566
1567 if (!Mount_By_Path("/data", true))
1568 return false;
1569
1570 if (0 != stat("/data/system/batterystats.bin", &st)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001571 gui_print("No Battery Stats Found. No Need To Wipe.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001572 } else {
1573 remove("/data/system/batterystats.bin");
Dees_Troy2673cec2013-04-02 20:22:16 +00001574 gui_print("Cleared battery stats.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001575 }
1576 return true;
1577}
1578
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001579int TWPartitionManager::Wipe_Android_Secure(void) {
1580 std::vector<TWPartition*>::iterator iter;
1581 int ret = false;
1582 bool found = false;
1583
1584 // Iterate through all partitions
1585 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1586 if ((*iter)->Has_Android_Secure) {
1587 ret = (*iter)->Wipe_AndSec();
1588 found = true;
1589 }
1590 }
1591 if (found) {
1592 return ret;
1593 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001594 gui_err("no_andsec=No android secure partitions found.");
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001595 }
1596 return false;
1597}
1598
Dees_Troy38bd7602012-09-14 13:33:53 -04001599int TWPartitionManager::Format_Data(void) {
1600 TWPartition* dat = Find_Partition_By_Path("/data");
bigbiffcfa875c2021-06-20 16:20:27 -04001601 TWPartition* metadata = Find_Partition_By_Path("/metadata");
1602 if (metadata != NULL)
1603 metadata->UnMount(false);
Dees_Troy38bd7602012-09-14 13:33:53 -04001604
1605 if (dat != NULL) {
bigbiffab76bd72021-10-11 10:17:48 -04001606 if (android::base::GetBoolProperty("ro.virtual_ab.enabled", false)) {
bigbiffcfa875c2021-06-20 16:20:27 -04001607#ifndef TW_EXCLUDE_APEX
1608 twrpApex apex;
1609 apex.Unmount();
1610#endif
1611 if (metadata != NULL)
1612 metadata->Mount(true);
bigbiffd21252f2021-09-18 15:56:32 -04001613 if (!Check_Pending_Merges())
bigbiffcfa875c2021-06-20 16:20:27 -04001614 return false;
1615 }
Dees_Troy38bd7602012-09-14 13:33:53 -04001616 return dat->Wipe_Encryption();
1617 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001618 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001619 return false;
1620 }
1621 return false;
1622}
1623
1624int TWPartitionManager::Wipe_Media_From_Data(void) {
1625 TWPartition* dat = Find_Partition_By_Path("/data");
1626
1627 if (dat != NULL) {
1628 if (!dat->Has_Data_Media) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001629 LOGERR("This device does not have /data/media\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001630 return false;
1631 }
1632 if (!dat->Mount(true))
1633 return false;
1634
Ethan Yonker74db1572015-10-28 12:44:49 -05001635 gui_msg("wiping_datamedia=Wiping internal storage -- /data/media...");
Ethan Yonker726a0202014-12-16 20:01:38 -06001636 Remove_MTP_Storage(dat->MTP_Storage_ID);
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001637 TWFunc::removeDir("/data/media", false);
xiaolu9416f4f2015-06-04 08:22:23 +08001638 dat->Recreate_Media_Folder();
Ethan Yonker726a0202014-12-16 20:01:38 -06001639 Add_MTP_Storage(dat->MTP_Storage_ID);
Dees_Troy38bd7602012-09-14 13:33:53 -04001640 return true;
1641 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001642 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001643 return false;
1644 }
1645 return false;
1646}
1647
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001648int TWPartitionManager::Repair_By_Path(string Path, bool Display_Error) {
1649 std::vector<TWPartition*>::iterator iter;
1650 int ret = false;
1651 bool found = false;
1652 string Local_Path = TWFunc::Get_Root_Path(Path);
1653
1654 if (Local_Path == "/tmp" || Local_Path == "/")
1655 return true;
1656