blob: f44544df5eae539496cf7fe045fcfbc1840ba6bf [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
bigbiffdf53b6c2022-01-17 19:31:52 -050096// #include "crypto/fde/cryptfs.h"
bigbiff7ba75002020-04-11 20:47:09 -040097#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
bigbiffdf53b6c2022-01-17 19:31:52 -0500103 #include "cryptfs.h"
104 #include "fscrypt-common.h"
bigbiffa957f072021-03-07 18:20:29 -0500105 #include "MetadataCrypt.h"
bigbiffadc599e2020-05-28 19:36:30 +0000106 #endif
bigbiff7ba75002020-04-11 20:47:09 -0400107#endif
108#endif
109#ifdef TW_CRYPTO_USE_SYSTEM_VOLD
110#include "crypto/vold_decrypt/vold_decrypt.h"
111#endif
Dees_Troy5bf43922012-09-07 16:07:55 -0400112#endif
Dees_Troy51a0e822012-09-05 15:24:24 -0400113
Ethan Yonker1b190162016-12-05 15:25:19 -0600114#ifdef AB_OTA_UPDATER
dianlujitaoa90dc462021-12-16 16:56:26 +0800115#include <android/hardware/boot/1.0/IBootControl.h>
116using android::hardware::boot::V1_0::CommandResult;
117using android::hardware::boot::V1_0::IBootControl;
Ethan Yonker1b190162016-12-05 15:25:19 -0600118#endif
119
bigbiffcfa875c2021-06-20 16:20:27 -0400120using android::fs_mgr::DestroyLogicalPartition;
bigbiffee7b7ff2020-03-23 15:08:27 -0400121using android::fs_mgr::Fstab;
122using android::fs_mgr::FstabEntry;
bigbiffcfa875c2021-06-20 16:20:27 -0400123using android::fs_mgr::MetadataBuilder;
Ethan Yonker6277c792014-09-15 14:54:30 -0500124
bigbiffee7b7ff2020-03-23 15:08:27 -0400125extern bool datamedia;
Noah Jacobson81d638d2019-04-28 00:10:07 -0400126std::vector<users_struct> Users_List;
127
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500128TWPartitionManager::TWPartitionManager(void) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400129 mtp_was_enabled = false;
Ethan Yonker726a0202014-12-16 20:01:38 -0600130 mtp_write_fd = -1;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600131 uevent_pfd.fd = -1;
bigbiff7abc5fe2015-01-17 16:53:12 -0500132 stop_backup.set_value(0);
Ethan Yonker1b190162016-12-05 15:25:19 -0600133#ifdef AB_OTA_UPDATER
134 char slot_suffix[PROPERTY_VALUE_MAX];
Ethan Yonkerfefe5912017-09-30 22:22:13 -0500135 property_get("ro.boot.slot_suffix", slot_suffix, "error");
136 if (strcmp(slot_suffix, "error") == 0)
137 property_get("ro.boot.slot", slot_suffix, "error");
Ethan Yonker1b190162016-12-05 15:25:19 -0600138 Active_Slot_Display = "";
Ethan Yonkerfefe5912017-09-30 22:22:13 -0500139 if (strcmp(slot_suffix, "_a") == 0 || strcmp(slot_suffix, "a") == 0)
Ethan Yonker1b190162016-12-05 15:25:19 -0600140 Set_Active_Slot("A");
141 else
142 Set_Active_Slot("B");
143#endif
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500144}
145
Captain Throwback37c3aef2021-10-06 11:38:18 -0400146void TWPartitionManager::Set_Crypto_State() {
147 char crypto_state[PROPERTY_VALUE_MAX];
148 property_get("ro.crypto.state", crypto_state, "error");
149 if (strcmp(crypto_state, "error") == 0)
150 property_set("ro.crypto.state", "encrypted");
151}
152
153int TWPartitionManager::Set_Crypto_Type(const char* crypto_type) {
154 char type_prop[PROPERTY_VALUE_MAX];
155 property_get("ro.crypto.type", type_prop, "error");
156 if (strcmp(type_prop, "error") == 0)
157 property_set("ro.crypto.type", crypto_type);
Captain Throwback18aea272020-10-29 12:43:17 -0400158 // Sleep for a bit so that services can start if needed
159 sleep(1);
160 return 0;
161}
162
bigbiff22851b92021-09-01 16:46:57 -0400163int TWPartitionManager::Process_Fstab(string Fstab_Filename, bool Display_Error, bool recovery_mode) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400164 FILE *fstabFile;
165 char fstab_line[MAX_FSTAB_LINE_LENGTH];
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600166 std::map<string, Flags_Map> twrp_flags;
167
168 fstabFile = fopen("/etc/twrp.flags", "rt");
169 if (fstabFile != NULL) {
Ian Macdonald160e8d32020-09-29 20:47:31 +0200170 LOGINFO("Reading /etc/twrp.flags\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600171 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600172 size_t line_size = strlen(fstab_line);
173 if (fstab_line[line_size - 1] != '\n')
174 fstab_line[line_size] = '\n';
175 Flags_Map line_flags;
176 line_flags.Primary_Block_Device = "";
177 line_flags.Alternate_Block_Device = "";
178 line_flags.fstab_line = (char*)malloc(MAX_FSTAB_LINE_LENGTH);
179 if (!line_flags.fstab_line) {
180 LOGERR("malloc error on line_flags.fstab_line\n");
181 return false;
182 }
183 memcpy(line_flags.fstab_line, fstab_line, MAX_FSTAB_LINE_LENGTH);
184 bool found_separator = false;
185 char *fs_loc = NULL;
186 char *block_loc = NULL;
187 char *flags_loc = NULL;
188 size_t index, item_index = 0;
189 for (index = 0; index < line_size; index++) {
190 if (fstab_line[index] <= 32) {
191 fstab_line[index] = '\0';
192 found_separator = true;
193 } else if (found_separator) {
194 if (item_index == 0) {
195 fs_loc = fstab_line + index;
196 } else if (item_index == 1) {
197 block_loc = fstab_line + index;
198 } else if (item_index > 1) {
199 char *ptr = fstab_line + index;
200 if (*ptr == '/') {
201 line_flags.Alternate_Block_Device = ptr;
202 } else if (strlen(ptr) > strlen("flags=") && strncmp(ptr, "flags=", strlen("flags=")) == 0) {
203 flags_loc = ptr;
204 // Once we find the flags=, we're done scanning the line
205 break;
206 }
207 }
208 found_separator = false;
209 item_index++;
210 }
211 }
212 if (block_loc)
213 line_flags.Primary_Block_Device = block_loc;
214 if (fs_loc)
215 line_flags.File_System = fs_loc;
216 if (flags_loc)
217 line_flags.Flags = flags_loc;
218 string Mount_Point = fstab_line;
219 twrp_flags[Mount_Point] = line_flags;
220 memset(fstab_line, 0, sizeof(fstab_line));
221 }
222 fclose(fstabFile);
223 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400224
225 fstabFile = fopen(Fstab_Filename.c_str(), "rt");
226 if (fstabFile == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000227 LOGERR("Critical Error: Unable to open fstab at '%s'.\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400228 return false;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600229 } else
230 LOGINFO("Reading %s\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400231
232 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600233 if (strstr(fstab_line, "swap"))
234 continue; // Skip swap in recovery
235
bigbiff998f8392021-08-06 19:19:33 -0400236 if (fstab_line[0] == '#')
237 continue;
238
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600239 size_t line_size = strlen(fstab_line);
240 if (fstab_line[line_size - 1] != '\n')
241 fstab_line[line_size] = '\n';
Dees_Troy2a923582012-09-20 12:13:34 -0400242
Matt Mower2b2dd152016-04-26 11:24:08 -0500243 TWPartition* partition = new TWPartition();
bigbiff8da46fa2020-09-08 16:42:34 -0400244 if (partition->Process_Fstab_Line(fstab_line, Display_Error, &twrp_flags))
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500245 Partitions.push_back(partition);
Matt Mower72c87ce2016-04-26 14:34:56 -0500246 else
Dees_Troy5bf43922012-09-07 16:07:55 -0400247 delete partition;
Matt Mower2b2dd152016-04-26 11:24:08 -0500248
249 memset(fstab_line, 0, sizeof(fstab_line));
Dees_Troy5bf43922012-09-07 16:07:55 -0400250 }
251 fclose(fstabFile);
Matt Mower72c87ce2016-04-26 14:34:56 -0500252
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600253 if (twrp_flags.size() > 0) {
254 LOGINFO("Processing remaining twrp.flags\n");
255 // Add any items from twrp.flags that did not exist in the recovery.fstab
256 for (std::map<string, Flags_Map>::iterator mapit=twrp_flags.begin(); mapit!=twrp_flags.end(); mapit++) {
257 if (Find_Partition_By_Path(mapit->first) == NULL) {
258 TWPartition* partition = new TWPartition();
bigbiff8da46fa2020-09-08 16:42:34 -0400259 if (partition->Process_Fstab_Line(mapit->second.fstab_line, Display_Error, NULL))
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600260 Partitions.push_back(partition);
261 else
262 delete partition;
263 }
264 if (mapit->second.fstab_line)
265 free(mapit->second.fstab_line);
266 mapit->second.fstab_line = NULL;
267 }
268 }
Mohd Faraze3948ec2020-08-14 01:40:59 +0000269 if (Get_Super_Status()) {
270 Setup_Super_Devices();
271 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600272 LOGINFO("Done processing fstab files\n");
273
bigbiff22851b92021-09-01 16:46:57 -0400274 if (recovery_mode) {
275 Setup_Fstab_Partitions(Display_Error);
Mohd Faraze3948ec2020-08-14 01:40:59 +0000276 }
bigbiff7ba75002020-04-11 20:47:09 -0400277 return true;
278}
279
bigbiff22851b92021-09-01 16:46:57 -0400280void TWPartitionManager::Setup_Fstab_Partitions(bool Display_Error) {
281 TWPartition* settings_partition = NULL;
282 TWPartition* andsec_partition = NULL;
283 std::vector<TWPartition*>::iterator iter;
284 unsigned int storageid = 1 << 16; // upper 16 bits are for physical storage device, we pretend to have only one
285
286 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
287 (*iter)->Partition_Post_Processing(Display_Error);
288
289 if ((*iter)->Is_Storage) {
290 ++storageid;
291 (*iter)->MTP_Storage_ID = storageid;
292 }
293
294 if (!settings_partition && (*iter)->Is_Settings_Storage && (*iter)->Is_Present)
295 settings_partition = (*iter);
296 else
297 (*iter)->Is_Settings_Storage = false;
298
299 if (!andsec_partition && (*iter)->Has_Android_Secure && (*iter)->Is_Present)
300 andsec_partition = (*iter);
301 else
302 (*iter)->Has_Android_Secure = false;
303
Mohd Faraz5ec04ef2023-01-02 22:47:51 +0100304 if ((*iter)->Is_Super && !Prepare_Super_Volume(*iter))
305 Partitions.erase(iter--);
bigbiff22851b92021-09-01 16:46:57 -0400306 }
307
me-cafebabed63fcaa2022-07-28 03:19:20 +0800308 Unlock_Block_Partitions();
309
bigbiff22851b92021-09-01 16:46:57 -0400310 //Setup Apex before decryption
311 TWPartition* sys = PartitionManager.Find_Partition_By_Path(PartitionManager.Get_Android_Root_Path());
312 TWPartition* ven = PartitionManager.Find_Partition_By_Path("/vendor");
313 if (sys) {
314 if (sys->Get_Super_Status()) {
315 sys->Mount(true);
316 if (ven) {
317 ven->Mount(true);
318 }
319 #ifdef TW_EXCLUDE_APEX
320 LOGINFO("Apex is disabled in this build\n");
321 #else
322 twrpApex apex;
323 if (!apex.loadApexImages()) {
324 LOGERR("Unable to load apex images from %s\n", APEX_DIR);
325 property_set("twrp.apex.loaded", "false");
326 } else {
327 property_set("twrp.apex.loaded", "true");
328 }
329 TWFunc::check_and_run_script("/sbin/resyncapex.sh", "apex");
330 #endif
331 }
332 }
333 #ifndef USE_VENDOR_LIBS
334 if (ven)
335 ven->UnMount(true);
336 if (sys)
337 sys->UnMount(true);
338 #endif
339
340 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) {
341 // Attempt to automatically identify /data/media emulated storage devices
342 TWPartition* Dat = Find_Partition_By_Path("/data");
343 if (Dat) {
344 LOGINFO("Using automatic handling for /data/media emulated storage device.\n");
345 datamedia = true;
346 Dat->Setup_Data_Media();
347 settings_partition = Dat;
348 // Since /data was not considered a storage partition earlier, we still need to assign an MTP ID
349 ++storageid;
350 Dat->MTP_Storage_ID = storageid;
351 }
352 }
353 if (!settings_partition) {
354 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
355 if ((*iter)->Is_Storage) {
356 settings_partition = (*iter);
357 break;
358 }
359 }
360 if (!settings_partition)
361 LOGERR("Unable to locate storage partition for storing settings file.\n");
362 }
363 if (!Write_Fstab()) {
364 if (Display_Error)
365 LOGERR("Error creating fstab\n");
366 else
367 LOGINFO("Error creating fstab\n");
368 }
369
370 if (andsec_partition) {
371 Setup_Android_Secure_Location(andsec_partition);
372 } else if (settings_partition) {
373 Setup_Android_Secure_Location(settings_partition);
374 }
375 if (settings_partition) {
376 Setup_Settings_Storage_Partition(settings_partition);
377 }
378
379 #ifdef TW_INCLUDE_CRYPTO
380 DataManager::SetValue(TW_IS_ENCRYPTED, 1);
381 Decrypt_Data();
382 #endif
383
384 Update_System_Details();
385 if (Get_Super_Status())
386 Setup_Super_Partition();
387 UnMount_Main_Partitions();
388 #ifdef AB_OTA_UPDATER
389 DataManager::SetValue("tw_active_slot", Get_Active_Slot_Display());
390 #endif
391 setup_uevent();
392}
393
bigbiff7ba75002020-04-11 20:47:09 -0400394int TWPartitionManager::Write_Fstab(void) {
395 FILE *fp;
396 std::vector<TWPartition*>::iterator iter;
397 string Line;
398
399 fp = fopen("/etc/fstab", "w");
400 if (fp == NULL) {
401 LOGINFO("Can not open /etc/fstab.\n");
402 return false;
403 }
404 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
405 if ((*iter)->Can_Be_Mounted) {
Mohd Faraz2a1e3752020-09-30 10:55:48 +0530406 Line = (*iter)->Actual_Block_Device + " " + (*iter)->Mount_Point + " " + (*iter)->Current_File_System +
407 ((*iter)->Mount_Read_Only ? " ro " : " rw ") + "0 0\n";
bigbiff7ba75002020-04-11 20:47:09 -0400408 fputs(Line.c_str(), fp);
409 }
410 // Handle subpartition tracking
411 if ((*iter)->Is_SubPartition) {
412 TWPartition* ParentPartition = Find_Partition_By_Path((*iter)->SubPartition_Of);
413 if (ParentPartition)
414 ParentPartition->Has_SubPartition = true;
415 else
416 LOGERR("Unable to locate parent partition '%s' of '%s'\n", (*iter)->SubPartition_Of.c_str(), (*iter)->Mount_Point.c_str());
417 }
418 }
419 fclose(fp);
420 return true;
421}
422
423void TWPartitionManager::Decrypt_Data() {
424 #ifdef TW_INCLUDE_CRYPTO
Ethan Yonkercceebb82014-11-18 10:17:59 -0600425 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
426 if (Decrypt_Data && Decrypt_Data->Is_Encrypted && !Decrypt_Data->Is_Decrypted) {
Captain Throwback37c3aef2021-10-06 11:38:18 -0400427 Set_Crypto_State();
bigbiff4fa02b52021-12-30 17:01:44 -0500428 TWPartition* Key_Directory_Partition = Find_Partition_By_Path(Decrypt_Data->Key_Directory);
429 if (Key_Directory_Partition != nullptr)
430 if (!Key_Directory_Partition->Is_Mounted())
431 Mount_By_Path(Decrypt_Data->Key_Directory, false);
432 if (!Decrypt_Data->Key_Directory.empty()) {
433 Set_Crypto_Type("file");
Ethan Yonker93382822018-11-01 15:25:31 -0500434#ifdef TW_INCLUDE_FBE_METADATA_DECRYPT
bigbiff7ba75002020-04-11 20:47:09 -0400435#ifdef USE_FSCRYPT
bigbiffdf53b6c2022-01-17 19:31:52 -0500436 if (android::vold::fscrypt_mount_metadata_encrypted(Decrypt_Data->Actual_Block_Device, Decrypt_Data->Mount_Point, false, false, Decrypt_Data->Current_File_System)) {
bigbiffa957f072021-03-07 18:20:29 -0500437 std::string crypto_blkdev = android::base::GetProperty("ro.crypto.fs_crypto_blkdev", "error");
Mohd Faraze3948ec2020-08-14 01:40:59 +0000438 Decrypt_Data->Decrypted_Block_Device = crypto_blkdev;
bigbiff7ba75002020-04-11 20:47:09 -0400439 LOGINFO("Successfully decrypted metadata encrypted data partition with new block device: '%s'\n", crypto_blkdev.c_str());
bigbiff7ba75002020-04-11 20:47:09 -0400440#endif
Ethan Yonker93382822018-11-01 15:25:31 -0500441 Decrypt_Data->Is_Decrypted = true; // Needed to make the mount function work correctly
442 int retry_count = 10;
443 while (!Decrypt_Data->Mount(false) && --retry_count)
444 usleep(500);
445 if (Decrypt_Data->Mount(false)) {
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400446 if (!Decrypt_Data->Decrypt_FBE_DE()) {
bigbiff437d86f2021-09-26 16:57:14 -0400447 LOGERR("Unable to decrypt FBE device\n");
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400448 }
449
Ethan Yonker93382822018-11-01 15:25:31 -0500450 } else {
451 LOGINFO("Failed to mount data after metadata decrypt\n");
452 }
453 } else {
454 LOGINFO("Unable to decrypt metadata encryption\n");
455 }
456#else
457 LOGERR("Metadata FBE decrypt support not present in this TWRP\n");
458#endif
459 }
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600460 if (Decrypt_Data->Is_FBE) {
461 if (DataManager::GetIntValue(TW_CRYPTO_PWTYPE) == 0) {
462 if (Decrypt_Device("!") == 0) {
463 gui_msg("decrypt_success=Successfully decrypted with default password.");
464 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
465 } else {
466 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
467 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600468 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600469 }
470 }
bigbiff4fa02b52021-12-30 17:01:44 -0500471 if (Decrypt_Data && (!Decrypt_Data->Is_Encrypted || Decrypt_Data->Is_Decrypted)) {
Ethan Yonker66a19492015-12-10 10:19:45 -0600472 Decrypt_Adopted();
473 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600474#endif
Dees_Troy51a0e822012-09-05 15:24:24 -0400475}
476
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500477void TWPartitionManager::Setup_Settings_Storage_Partition(TWPartition* Part) {
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500478 DataManager::SetValue("tw_storage_path", Part->Storage_Path);
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500479}
480
Matt Mowerbf4efa32014-04-14 23:25:26 -0500481void TWPartitionManager::Setup_Android_Secure_Location(TWPartition* Part) {
482 if (Part->Has_Android_Secure)
483 Part->Setup_AndSec();
Ethan Yonker6277c792014-09-15 14:54:30 -0500484 else if (!datamedia)
Matt Mowerbf4efa32014-04-14 23:25:26 -0500485 Part->Setup_AndSec();
Matt Mowerbf4efa32014-04-14 23:25:26 -0500486}
487
Dees_Troy8170a922012-09-18 15:40:25 -0400488void TWPartitionManager::Output_Partition_Logging(void) {
489 std::vector<TWPartition*>::iterator iter;
490
491 printf("\n\nPartition Logs:\n");
492 for (iter = Partitions.begin(); iter != Partitions.end(); iter++)
493 Output_Partition((*iter));
494}
495
496void TWPartitionManager::Output_Partition(TWPartition* Part) {
497 unsigned long long mb = 1048576;
498
Gary Peck004d48b2012-11-21 16:28:18 -0800499 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 -0400500 if (Part->Can_Be_Mounted) {
Gary Peck004d48b2012-11-21 16:28:18 -0800501 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 -0400502 }
Gary Peck004d48b2012-11-21 16:28:18 -0800503 printf("\n Flags: ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500504 if (Part->Can_Be_Mounted)
505 printf("Can_Be_Mounted ");
Gary Peck004d48b2012-11-21 16:28:18 -0800506 if (Part->Can_Be_Wiped)
507 printf("Can_Be_Wiped ");
Hashcodedabfd492013-08-29 22:45:30 -0700508 if (Part->Use_Rm_Rf)
509 printf("Use_Rm_Rf ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500510 if (Part->Can_Be_Backed_Up)
511 printf("Can_Be_Backed_Up ");
Gary Peck004d48b2012-11-21 16:28:18 -0800512 if (Part->Wipe_During_Factory_Reset)
513 printf("Wipe_During_Factory_Reset ");
514 if (Part->Wipe_Available_in_GUI)
515 printf("Wipe_Available_in_GUI ");
516 if (Part->Is_SubPartition)
517 printf("Is_SubPartition ");
518 if (Part->Has_SubPartition)
519 printf("Has_SubPartition ");
520 if (Part->Removable)
521 printf("Removable ");
522 if (Part->Is_Present)
523 printf("IsPresent ");
524 if (Part->Can_Be_Encrypted)
525 printf("Can_Be_Encrypted ");
526 if (Part->Is_Encrypted)
527 printf("Is_Encrypted ");
528 if (Part->Is_Decrypted)
529 printf("Is_Decrypted ");
530 if (Part->Has_Data_Media)
531 printf("Has_Data_Media ");
Dees_Troy83bd4832013-05-04 12:39:56 +0000532 if (Part->Can_Encrypt_Backup)
533 printf("Can_Encrypt_Backup ");
534 if (Part->Use_Userdata_Encryption)
535 printf("Use_Userdata_Encryption ");
Gary Peck004d48b2012-11-21 16:28:18 -0800536 if (Part->Has_Android_Secure)
537 printf("Has_Android_Secure ");
538 if (Part->Is_Storage)
539 printf("Is_Storage ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500540 if (Part->Is_Settings_Storage)
541 printf("Is_Settings_Storage ");
Dees_Troy68cab492012-12-12 19:29:35 +0000542 if (Part->Ignore_Blkid)
543 printf("Ignore_Blkid ");
Ethan Yonker253368a2014-11-25 15:00:52 -0600544 if (Part->Mount_To_Decrypt)
545 printf("Mount_To_Decrypt ");
Ethan Yonker96af84a2015-01-05 14:58:36 -0600546 if (Part->Can_Flash_Img)
547 printf("Can_Flash_Img ");
Ethan Yonker66a19492015-12-10 10:19:45 -0600548 if (Part->Is_Adopted_Storage)
549 printf("Is_Adopted_Storage ");
Ethan Yonker1b190162016-12-05 15:25:19 -0600550 if (Part->SlotSelect)
551 printf("SlotSelect ");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600552 if (Part->Mount_Read_Only)
553 printf("Mount_Read_Only ");
bigbiffcfa875c2021-06-20 16:20:27 -0400554 if (Part->Is_Super)
555 printf("Is_Super ");
Gary Peck004d48b2012-11-21 16:28:18 -0800556 printf("\n");
557 if (!Part->SubPartition_Of.empty())
558 printf(" SubPartition_Of: %s\n", Part->SubPartition_Of.c_str());
559 if (!Part->Symlink_Path.empty())
560 printf(" Symlink_Path: %s\n", Part->Symlink_Path.c_str());
561 if (!Part->Symlink_Mount_Point.empty())
562 printf(" Symlink_Mount_Point: %s\n", Part->Symlink_Mount_Point.c_str());
563 if (!Part->Primary_Block_Device.empty())
564 printf(" Primary_Block_Device: %s\n", Part->Primary_Block_Device.c_str());
565 if (!Part->Alternate_Block_Device.empty())
566 printf(" Alternate_Block_Device: %s\n", Part->Alternate_Block_Device.c_str());
567 if (!Part->Decrypted_Block_Device.empty())
568 printf(" Decrypted_Block_Device: %s\n", Part->Decrypted_Block_Device.c_str());
dianlujitao4879b372018-12-03 18:45:47 +0800569 if (!Part->Crypto_Key_Location.empty())
Ethan Yonker253368a2014-11-25 15:00:52 -0600570 printf(" Crypto_Key_Location: %s\n", Part->Crypto_Key_Location.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800571 if (Part->Length != 0)
572 printf(" Length: %i\n", Part->Length);
573 if (!Part->Display_Name.empty())
574 printf(" Display_Name: %s\n", Part->Display_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500575 if (!Part->Storage_Name.empty())
576 printf(" Storage_Name: %s\n", Part->Storage_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800577 if (!Part->Backup_Path.empty())
578 printf(" Backup_Path: %s\n", Part->Backup_Path.c_str());
579 if (!Part->Backup_Name.empty())
580 printf(" Backup_Name: %s\n", Part->Backup_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500581 if (!Part->Backup_Display_Name.empty())
582 printf(" Backup_Display_Name: %s\n", Part->Backup_Display_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800583 if (!Part->Backup_FileName.empty())
584 printf(" Backup_FileName: %s\n", Part->Backup_FileName.c_str());
585 if (!Part->Storage_Path.empty())
586 printf(" Storage_Path: %s\n", Part->Storage_Path.c_str());
587 if (!Part->Current_File_System.empty())
588 printf(" Current_File_System: %s\n", Part->Current_File_System.c_str());
589 if (!Part->Fstab_File_System.empty())
590 printf(" Fstab_File_System: %s\n", Part->Fstab_File_System.c_str());
591 if (Part->Format_Block_Size != 0)
Ethan Yonkerc798c9c2015-10-09 11:15:26 -0500592 printf(" Format_Block_Size: %lu\n", Part->Format_Block_Size);
Dees_Troy094207a2012-09-26 12:00:39 -0400593 if (!Part->MTD_Name.empty())
594 printf(" MTD_Name: %s\n", Part->MTD_Name.c_str());
Matt Mower029a82d2017-01-08 13:12:38 -0600595 printf(" Backup_Method: %s\n", Part->Backup_Method_By_Name().c_str());
Hashcode62bd9e02013-11-19 21:59:42 -0800596 if (Part->Mount_Flags || !Part->Mount_Options.empty())
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600597 printf(" Mount_Flags: %i, Mount_Options: %s\n", Part->Mount_Flags, Part->Mount_Options.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -0600598 if (Part->MTP_Storage_ID)
599 printf(" MTP_Storage_ID: %i\n", Part->MTP_Storage_ID);
Ethan Yonker93382822018-11-01 15:25:31 -0500600 if (!Part->Key_Directory.empty())
601 printf(" Metadata Key Directory: %s\n", Part->Key_Directory.c_str());
Ethan Yonker6277c792014-09-15 14:54:30 -0500602 printf("\n");
Dees_Troy8170a922012-09-18 15:40:25 -0400603}
604
Dees_Troy51a0e822012-09-05 15:24:24 -0400605int TWPartitionManager::Mount_By_Path(string Path, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400606 std::vector<TWPartition*>::iterator iter;
607 int ret = false;
608 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400609 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400610
bigbiffd58ba182020-03-23 10:02:29 -0400611 if (Local_Path == "/tmp" || Local_Path == "/")
Dees_Troy43d8b002012-09-17 16:00:01 -0400612 return true;
613
Dees_Troy5bf43922012-09-07 16:07:55 -0400614 // Iterate through all partitions
615 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400616 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400617 ret = (*iter)->Mount(Display_Error);
618 found = true;
Dees_Troy51127312012-09-08 13:08:49 -0400619 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400620 (*iter)->Mount(Display_Error);
Dees_Troy51127312012-09-08 13:08:49 -0400621 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400622 }
623 if (found) {
624 return ret;
625 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500626 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 -0400627 }
628 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400629}
630
Dees_Troy51a0e822012-09-05 15:24:24 -0400631int TWPartitionManager::UnMount_By_Path(string Path, bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400632 std::vector<TWPartition*>::iterator iter;
633 int ret = false;
634 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400635 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy51127312012-09-08 13:08:49 -0400636
637 // Iterate through all partitions
638 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400639 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy51127312012-09-08 13:08:49 -0400640 ret = (*iter)->UnMount(Display_Error);
641 found = true;
642 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
643 (*iter)->UnMount(Display_Error);
644 }
645 }
646 if (found) {
647 return ret;
648 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500649 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 -0400650 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000651 LOGINFO("UnMount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400652 }
653 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400654}
655
Dees_Troy51a0e822012-09-05 15:24:24 -0400656int TWPartitionManager::Is_Mounted_By_Path(string Path) {
Dees_Troy51127312012-09-08 13:08:49 -0400657 TWPartition* Part = Find_Partition_By_Path(Path);
658
659 if (Part)
660 return Part->Is_Mounted();
661 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000662 LOGINFO("Is_Mounted: Unable to find partition for path '%s'\n", Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400663 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400664}
665
Dees_Troy5bf43922012-09-07 16:07:55 -0400666int TWPartitionManager::Mount_Current_Storage(bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400667 string current_storage_path = DataManager::GetCurrentStoragePath();
668
669 if (Mount_By_Path(current_storage_path, Display_Error)) {
670 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
671 if (FreeStorage)
672 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
673 return true;
674 }
675 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400676}
677
Dees_Troy5bf43922012-09-07 16:07:55 -0400678int TWPartitionManager::Mount_Settings_Storage(bool Display_Error) {
679 return Mount_By_Path(DataManager::GetSettingsStoragePath(), Display_Error);
680}
681
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600682TWPartition* TWPartitionManager::Find_Partition_By_Path(const string& Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400683 std::vector<TWPartition*>::iterator iter;
Dees_Troy38bd7602012-09-14 13:33:53 -0400684 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400685
dianlujitaob76a73a2020-04-30 20:33:20 +0800686 if (Local_Path == "/system")
687 Local_Path = Get_Android_Root_Path();
Dees_Troy5bf43922012-09-07 16:07:55 -0400688 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400689 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path))
Dees_Troy5bf43922012-09-07 16:07:55 -0400690 return (*iter);
691 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400692 return NULL;
693}
694
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600695TWPartition* TWPartitionManager::Find_Partition_By_Block_Device(const string& Block_Device) {
696 std::vector<TWPartition*>::iterator iter;
697
698 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
699 if ((*iter)->Primary_Block_Device == Block_Device || (!(*iter)->Actual_Block_Device.empty() && (*iter)->Actual_Block_Device == Block_Device))
700 return (*iter);
701 }
702 return NULL;
703}
704
Ethan Yonker53796e72019-01-11 22:49:52 -0600705int TWPartitionManager::Check_Backup_Name(const std::string& Backup_Name, bool Display_Error, bool Must_Be_Unique) {
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400706 // Check the backup name to ensure that it is the correct size and contains only valid characters
707 // and that a backup with that name doesn't already exist
708 char backup_name[MAX_BACKUP_NAME_LEN];
709 char backup_loc[255], tw_image_dir[255];
710 int copy_size;
711 int index, cur_char;
Ethan Yonker53796e72019-01-11 22:49:52 -0600712 string Backup_Loc;
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400713
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400714 copy_size = Backup_Name.size();
715 // Check size
716 if (copy_size > MAX_BACKUP_NAME_LEN) {
717 if (Display_Error)
Ethan Yonker74db1572015-10-28 12:44:49 -0500718 gui_err("backup_name_len=Backup name is too long.");
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400719 return -2;
720 }
721
722 // Check each character
723 strncpy(backup_name, Backup_Name.c_str(), copy_size);
Dees_Troya13d74f2013-03-24 08:54:55 -0500724 if (copy_size == 1 && strncmp(backup_name, "0", 1) == 0)
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400725 return 0; // A "0" (zero) means to use the current timestamp for the backup name
726 for (index=0; index<copy_size; index++) {
727 cur_char = (int)backup_name[index];
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500728 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 -0400729 // These are valid characters
730 // Numbers
731 // Upper case letters
732 // Lower case letters
733 // Space
734 // and -_.{}[]
735 } else {
736 if (Display_Error)
Ethan Yonker4adc33e2016-01-22 16:07:11 -0600737 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 -0400738 return -3;
739 }
740 }
741
Ethan Yonker53796e72019-01-11 22:49:52 -0600742 if (Must_Be_Unique) {
743 // Check to make sure that a backup with this name doesn't already exist
744 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Loc);
745 strcpy(backup_loc, Backup_Loc.c_str());
746 sprintf(tw_image_dir,"%s/%s", backup_loc, Backup_Name.c_str());
747 if (TWFunc::Path_Exists(tw_image_dir)) {
748 if (Display_Error)
749 gui_err("backup_name_exists=A backup with that name already exists!");
bigbiffce8f83c2015-12-12 18:30:21 -0500750
Ethan Yonker53796e72019-01-11 22:49:52 -0600751 return -4;
752 }
753 // Backup is unique
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400754 }
Ethan Yonker53796e72019-01-11 22:49:52 -0600755 // No problems found
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400756 return 0;
757}
758
bigbiffce8f83c2015-12-12 18:30:21 -0500759bool TWPartitionManager::Backup_Partition(PartitionSettings *part_settings) {
Dees_Troy43d8b002012-09-17 16:00:01 -0400760 time_t start, stop;
Matt Mower23d8aae2017-01-06 14:30:33 -0600761 int use_compression;
bigbiffce8f83c2015-12-12 18:30:21 -0500762 string backup_log = part_settings->Backup_Folder + "/recovery.log";
Dees_Troy43d8b002012-09-17 16:00:01 -0400763
bigbiffce8f83c2015-12-12 18:30:21 -0500764 if (part_settings->Part == NULL)
Dees_Troy43d8b002012-09-17 16:00:01 -0400765 return true;
766
Dees_Troy093b7642012-09-21 15:59:38 -0400767 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400768
Tom Hite5a926722014-09-15 01:31:03 +0000769 TWFunc::SetPerformanceMode(true);
Dees_Troy43d8b002012-09-17 16:00:01 -0400770 time(&start);
771
bigbiffce8f83c2015-12-12 18:30:21 -0500772 if (part_settings->Part->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500773 sync();
774 sync();
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400775 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500776 if (!part_settings->adbbackup && part_settings->generate_digest) {
777 if (!twrpDigestDriver::Make_Digest(Full_Filename))
778 goto backup_error;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400779 }
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400780
bigbiffce8f83c2015-12-12 18:30:21 -0500781 if (part_settings->Part->Has_SubPartition) {
Dees_Troy8170a922012-09-18 15:40:25 -0400782 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400783 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -0400784
785 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400786 if ((*subpart)->Can_Be_Backed_Up && (*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500787 part_settings->Part = *subpart;
bigbiffce8f83c2015-12-12 18:30:21 -0500788 if (!(*subpart)->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500789 goto backup_error;
Tom Hite5a926722014-09-15 01:31:03 +0000790 }
Dees_Troy2727b992013-08-14 20:09:30 +0000791 sync();
792 sync();
bigbiff bigbifff5955b12019-05-18 17:28:58 -0400793 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400794 if (!part_settings->adbbackup && part_settings->generate_digest) {
795 if (!twrpDigestDriver::Make_Digest(Full_Filename)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500796 goto backup_error;
bigbiffce8f83c2015-12-12 18:30:21 -0500797 }
Tom Hite5a926722014-09-15 01:31:03 +0000798 }
Dees_Troy8170a922012-09-18 15:40:25 -0400799 }
800 }
801 }
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400802
Dees_Troy43d8b002012-09-17 16:00:01 -0400803 time(&stop);
that203bcc92015-05-09 17:27:33 +0200804 int backup_time = (int) difftime(stop, start);
Dees_Troy2673cec2013-04-02 20:22:16 +0000805 LOGINFO("Partition Backup time: %d\n", backup_time);
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400806
bigbiffce8f83c2015-12-12 18:30:21 -0500807 if (part_settings->Part->Backup_Method == BM_FILES) {
808 part_settings->file_time += backup_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400809 } else {
bigbiffce8f83c2015-12-12 18:30:21 -0500810 part_settings->img_time += backup_time;
811
Dees_Troy43d8b002012-09-17 16:00:01 -0400812 }
Tom Hite5a926722014-09-15 01:31:03 +0000813
Matt Mower02899552016-12-30 18:13:51 -0600814 TWFunc::SetPerformanceMode(false);
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500815 return true;
Dees_Troy43d8b002012-09-17 16:00:01 -0400816 }
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500817backup_error:
818 Clean_Backup_Folder(part_settings->Backup_Folder);
819 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
820 tw_set_default_metadata(backup_log.c_str());
821 TWFunc::SetPerformanceMode(false);
822 return false;
bigbiff7abc5fe2015-01-17 16:53:12 -0500823}
824
bigbiffbf1d6722015-02-14 16:25:59 -0500825void TWPartitionManager::Clean_Backup_Folder(string Backup_Folder) {
826 DIR *d = opendir(Backup_Folder.c_str());
827 struct dirent *p;
828 int r;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400829 vector<string> ext;
830
831 //extensions we should delete when cleaning
832 ext.push_back("win");
833 ext.push_back("md5");
834 ext.push_back("sha2");
835 ext.push_back("info");
bigbiffbf1d6722015-02-14 16:25:59 -0500836
Ethan Yonker74db1572015-10-28 12:44:49 -0500837 gui_msg("backup_clean=Backup Failed. Cleaning Backup Folder.");
bigbiffbf1d6722015-02-14 16:25:59 -0500838
839 if (d == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500840 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Backup_Folder)(strerror(errno)));
bigbiffbf1d6722015-02-14 16:25:59 -0500841 return;
842 }
843
Matt Mower2b18a532015-02-20 16:58:05 -0600844 while ((p = readdir(d))) {
bigbiffbf1d6722015-02-14 16:25:59 -0500845 if (!strcmp(p->d_name, ".") || !strcmp(p->d_name, ".."))
846 continue;
847
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500848 string path = Backup_Folder + "/" + p->d_name;
bigbiffbf1d6722015-02-14 16:25:59 -0500849
850 size_t dot = path.find_last_of(".") + 1;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400851 for (vector<string>::const_iterator i = ext.begin(); i != ext.end(); ++i) {
852 if (path.substr(dot) == *i) {
853 r = unlink(path.c_str());
854 if (r != 0)
855 LOGINFO("Unable to unlink '%s: %s'\n", path.c_str(), strerror(errno));
bigbiffbf1d6722015-02-14 16:25:59 -0500856 }
857 }
858 }
859 closedir(d);
860}
861
Ethan Yonker472f5062016-02-25 13:47:30 -0600862int TWPartitionManager::Check_Backup_Cancel() {
863 return stop_backup.get_value();
864}
865
bigbiff7abc5fe2015-01-17 16:53:12 -0500866int TWPartitionManager::Cancel_Backup() {
867 string Backup_Folder, Backup_Name, Full_Backup_Path;
868
869 stop_backup.set_value(1);
870
871 if (tar_fork_pid != 0) {
872 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
873 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Folder);
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500874 Full_Backup_Path = Backup_Folder + "/" + Backup_Name;
bigbiff7abc5fe2015-01-17 16:53:12 -0500875 LOGINFO("Killing pid: %d\n", tar_fork_pid);
876 kill(tar_fork_pid, SIGUSR2);
877 while (kill(tar_fork_pid, 0) == 0) {
878 usleep(1000);
879 }
880 LOGINFO("Backup_Run stopped and returning false, backup cancelled.\n");
881 LOGINFO("Removing directory %s\n", Full_Backup_Path.c_str());
882 TWFunc::removeDir(Full_Backup_Path, false);
883 tar_fork_pid = 0;
884 }
885
886 return 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400887}
888
bigbiffce8f83c2015-12-12 18:30:21 -0500889int TWPartitionManager::Run_Backup(bool adbbackup) {
890 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400891 int partition_count = 0, disable_free_space_check = 0, skip_digest = 0;
bigbiffce8f83c2015-12-12 18:30:21 -0500892 string Backup_Name, Backup_List, backup_path;
Matt Mower23d8aae2017-01-06 14:30:33 -0600893 unsigned long long total_bytes = 0, free_space = 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400894 TWPartition* storage = NULL;
bigbiff7df87472022-09-24 16:45:03 -0400895 time_t total_start, total_stop;
896 //time_t seconds = time(0);
897 //struct tm *t = localtime(&seconds);
Dees_Troya13d74f2013-03-24 08:54:55 -0500898 size_t start_pos = 0, end_pos = 0;
bigbiff7abc5fe2015-01-17 16:53:12 -0500899 stop_backup.set_value(0);
Dees_Troy43d8b002012-09-17 16:00:01 -0400900
bigbiffce8f83c2015-12-12 18:30:21 -0500901 part_settings.img_bytes_remaining = 0;
902 part_settings.file_bytes_remaining = 0;
903 part_settings.img_time = 0;
904 part_settings.file_time = 0;
905 part_settings.img_bytes = 0;
906 part_settings.file_bytes = 0;
907 part_settings.PM_Method = PM_BACKUP;
908
bigbiffce8f83c2015-12-12 18:30:21 -0500909 part_settings.adbbackup = adbbackup;
Dees_Troy43d8b002012-09-17 16:00:01 -0400910 time(&total_start);
911
912 Update_System_Details();
913
914 if (!Mount_Current_Storage(true))
915 return false;
916
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400917 DataManager::GetValue(TW_SKIP_DIGEST_GENERATE_VAR, skip_digest);
918 if (skip_digest == 0)
919 part_settings.generate_digest = true;
Dees_Troyc5865ab2012-09-24 15:08:04 -0400920 else
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400921 part_settings.generate_digest = false;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400922
bigbiffce8f83c2015-12-12 18:30:21 -0500923 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, part_settings.Backup_Folder);
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500924 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
925 if (Backup_Name == gui_lookup("curr_date", "(Current Date)")) {
926 Backup_Name = TWFunc::Get_Current_Date();
927 } else if (Backup_Name == gui_lookup("auto_generate", "(Auto Generate)") || Backup_Name == "0" || Backup_Name.empty()) {
Dees Troyb21cc642013-09-10 17:36:41 +0000928 TWFunc::Auto_Generate_Backup_Name();
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500929 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
Dees_Troy43d8b002012-09-17 16:00:01 -0400930 }
bigbiffce8f83c2015-12-12 18:30:21 -0500931
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500932 LOGINFO("Backup Name is: '%s'\n", Backup_Name.c_str());
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500933 part_settings.Backup_Folder = part_settings.Backup_Folder + "/" + Backup_Name;
bigbiffce8f83c2015-12-12 18:30:21 -0500934
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500935 LOGINFO("Backup_Folder is: '%s'\n", part_settings.Backup_Folder.c_str());
Dees_Troy43d8b002012-09-17 16:00:01 -0400936
Dees_Troy2673cec2013-04-02 20:22:16 +0000937 LOGINFO("Calculating backup details...\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500938 DataManager::GetValue("tw_backup_list", Backup_List);
bigbiff7ba75002020-04-11 20:47:09 -0400939 LOGINFO("Backup_List: %s\n", Backup_List.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500940 if (!Backup_List.empty()) {
941 end_pos = Backup_List.find(";", start_pos);
942 while (end_pos != string::npos && start_pos < Backup_List.size()) {
943 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiff7ba75002020-04-11 20:47:09 -0400944 LOGINFO("backup_path: %s\n", backup_path.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -0500945 part_settings.Part = Find_Partition_By_Path(backup_path);
946 if (part_settings.Part != NULL) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500947 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500948 if (part_settings.Part->Backup_Method == BM_FILES)
949 part_settings.file_bytes += part_settings.Part->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500950 else
bigbiffce8f83c2015-12-12 18:30:21 -0500951 part_settings.img_bytes += part_settings.Part->Backup_Size;
952 if (part_settings.Part->Has_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500953 std::vector<TWPartition*>::iterator subpart;
954
955 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiffce8f83c2015-12-12 18:30:21 -0500956 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 -0500957 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500958 if ((*subpart)->Backup_Method == BM_FILES)
959 part_settings.file_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500960 else
bigbiffce8f83c2015-12-12 18:30:21 -0500961 part_settings.img_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500962 }
963 }
Dees_Troy8170a922012-09-18 15:40:25 -0400964 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500965 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500966 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 -0400967 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500968 start_pos = end_pos + 1;
969 end_pos = Backup_List.find(";", start_pos);
Dees_Troy43d8b002012-09-17 16:00:01 -0400970 }
971 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400972
973 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500974 gui_msg("no_partition_selected=No partitions selected for backup.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400975 return false;
976 }
bigbiffce8f83c2015-12-12 18:30:21 -0500977 if (adbbackup) {
978 if (twadbbu::Write_ADB_Stream_Header(partition_count) == false) {
979 return false;
980 }
981 }
982 total_bytes = part_settings.file_bytes + part_settings.img_bytes;
Ethan Yonker472f5062016-02-25 13:47:30 -0600983 ProgressTracking progress(total_bytes);
bigbiffce8f83c2015-12-12 18:30:21 -0500984 part_settings.progress = &progress;
985
Ethan Yonker74db1572015-10-28 12:44:49 -0500986 gui_msg(Msg("total_partitions_backup= * Total number of partitions to back up: {1}")(partition_count));
987 gui_msg(Msg("total_backup_size= * Total size of all data: {1}MB")(total_bytes / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400988 storage = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
989 if (storage != NULL) {
990 free_space = storage->Free;
Ethan Yonker74db1572015-10-28 12:44:49 -0500991 gui_msg(Msg("available_space= * Available space: {1}MB")(free_space / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400992 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500993 gui_err("unable_locate_storage=Unable to locate storage device.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400994 return false;
995 }
bigbiffbf1d6722015-02-14 16:25:59 -0500996
Matt Mowerbfccfb82016-04-25 23:22:31 -0500997 DataManager::GetValue(TW_DISABLE_FREE_SPACE_VAR, disable_free_space_check);
bigbiffce8f83c2015-12-12 18:30:21 -0500998
999 if (adbbackup)
1000 disable_free_space_check = true;
1001
bigbiffbf1d6722015-02-14 16:25:59 -05001002 if (!disable_free_space_check) {
1003 if (free_space - (32 * 1024 * 1024) < total_bytes) {
1004 // We require an extra 32MB just in case
Ethan Yonker74db1572015-10-28 12:44:49 -05001005 gui_err("no_space=Not enough free space on storage.");
bigbiffbf1d6722015-02-14 16:25:59 -05001006 return false;
1007 }
Dees_Troy43d8b002012-09-17 16:00:01 -04001008 }
bigbiffce8f83c2015-12-12 18:30:21 -05001009 part_settings.img_bytes_remaining = part_settings.img_bytes;
1010 part_settings.file_bytes_remaining = part_settings.file_bytes;
Dees_Troy43d8b002012-09-17 16:00:01 -04001011
Ethan Yonker74db1572015-10-28 12:44:49 -05001012 gui_msg("backup_started=[BACKUP STARTED]");
bigbiff bigbiffcdd97c72019-03-29 19:12:33 -04001013
1014 int is_decrypted = 0;
1015 int is_encrypted = 0;
1016
1017 DataManager::GetValue(TW_IS_DECRYPTED, is_decrypted);
1018 DataManager::GetValue(TW_IS_ENCRYPTED, is_encrypted);
1019 if (!adbbackup || (!is_encrypted || (is_encrypted && is_decrypted))) {
1020 gui_msg(Msg("backup_folder= * Backup Folder: {1}")(part_settings.Backup_Folder));
1021 if (!TWFunc::Recursive_Mkdir(part_settings.Backup_Folder)) {
1022 gui_err("fail_backup_folder=Failed to make backup folder.");
1023 return false;
1024 }
Dees_Troyd4b22b02013-01-18 17:17:58 +00001025 }
1026
Dees_Troy2673cec2013-04-02 20:22:16 +00001027 DataManager::SetProgress(0.0);
Dees_Troy093b7642012-09-21 15:59:38 -04001028
Dees_Troya13d74f2013-03-24 08:54:55 -05001029 start_pos = 0;
1030 end_pos = Backup_List.find(";", start_pos);
1031 while (end_pos != string::npos && start_pos < Backup_List.size()) {
bigbiff7abc5fe2015-01-17 16:53:12 -05001032 if (stop_backup.get_value() != 0)
1033 return -1;
Dees_Troya13d74f2013-03-24 08:54:55 -05001034 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001035 part_settings.Part = Find_Partition_By_Path(backup_path);
1036 if (part_settings.Part != NULL) {
1037 if (!Backup_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001038 return false;
1039 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001040 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 -05001041 }
1042 start_pos = end_pos + 1;
1043 end_pos = Backup_List.find(";", start_pos);
1044 }
Dees_Troy43d8b002012-09-17 16:00:01 -04001045
1046 // Average BPS
bigbiffce8f83c2015-12-12 18:30:21 -05001047 if (part_settings.img_time == 0)
1048 part_settings.img_time = 1;
1049 if (part_settings.file_time == 0)
1050 part_settings.file_time = 1;
1051 int img_bps = (int)part_settings.img_bytes / (int)part_settings.img_time;
1052 unsigned long long file_bps = part_settings.file_bytes / (int)part_settings.file_time;
Dees_Troy43d8b002012-09-17 16:00:01 -04001053
bigbiffce8f83c2015-12-12 18:30:21 -05001054 if (part_settings.file_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -06001055 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 -05001056 if (part_settings.img_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -06001057 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 -04001058
1059 time(&total_stop);
1060 int total_time = (int) difftime(total_stop, total_start);
bigbiffce8f83c2015-12-12 18:30:21 -05001061
1062 uint64_t actual_backup_size;
Ethan Yonker3fdcda42016-11-30 12:29:37 -06001063 if (!adbbackup) {
1064 TWExclude twe;
1065 actual_backup_size = twe.Get_Folder_Size(part_settings.Backup_Folder);
1066 } else
bigbiffce8f83c2015-12-12 18:30:21 -05001067 actual_backup_size = part_settings.file_bytes + part_settings.img_bytes;
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001068 actual_backup_size /= (1024LLU * 1024LLU);
Dees_Troy43d8b002012-09-17 16:00:01 -04001069
bigbiffce8f83c2015-12-12 18:30:21 -05001070 int prev_img_bps = 0, use_compression = 0;
1071 unsigned long long prev_file_bps = 0;
Matt Mower173cdb92016-12-31 02:49:19 -06001072 DataManager::GetValue(TW_BACKUP_AVG_IMG_RATE, prev_img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -04001073 img_bps += (prev_img_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001074 img_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -04001075
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001076 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -04001077 if (use_compression)
1078 DataManager::GetValue(TW_BACKUP_AVG_FILE_COMP_RATE, prev_file_bps);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001079 else
Dees_Troy093b7642012-09-21 15:59:38 -04001080 DataManager::GetValue(TW_BACKUP_AVG_FILE_RATE, prev_file_bps);
1081 file_bps += (prev_file_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001082 file_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -04001083
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001084 DataManager::SetValue(TW_BACKUP_AVG_IMG_RATE, img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -04001085 if (use_compression)
1086 DataManager::SetValue(TW_BACKUP_AVG_FILE_COMP_RATE, file_bps);
1087 else
1088 DataManager::SetValue(TW_BACKUP_AVG_FILE_RATE, file_bps);
1089
Ethan Yonker74db1572015-10-28 12:44:49 -05001090 gui_msg(Msg("total_backed_size=[{1} MB TOTAL BACKED UP]")(actual_backup_size));
Dees_Troy43d8b002012-09-17 16:00:01 -04001091 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001092 UnMount_Main_Partitions();
Ethan Yonker56db1c42015-12-20 22:49:00 -06001093 gui_msg(Msg(msg::kHighlight, "backup_completed=[BACKUP COMPLETED IN {1} SECONDS]")(total_time)); // the end
Ethan Yonkere080c1f2016-09-19 13:50:25 -05001094 string backup_log = part_settings.Backup_Folder + "/recovery.log";
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001095 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
Ethan Yonker4b94cfd2014-12-11 10:00:45 -06001096 tw_set_default_metadata(backup_log.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -05001097
1098 if (part_settings.adbbackup) {
1099 if (twadbbu::Write_ADB_Stream_Trailer() == false) {
1100 return false;
1101 }
1102 }
1103 part_settings.adbbackup = false;
1104 DataManager::SetValue("tw_enable_adb_backup", 0);
1105
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001106 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001107}
1108
bigbiffce8f83c2015-12-12 18:30:21 -05001109bool TWPartitionManager::Restore_Partition(PartitionSettings *part_settings) {
Dees_Troy4a2a1262012-09-18 09:33:47 -04001110 time_t Start, Stop;
bigbiffce8f83c2015-12-12 18:30:21 -05001111
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001112 if (part_settings->adbbackup) {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001113 std::string partName = part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win";
1114 LOGINFO("setting backup name: %s\n", partName.c_str());
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001115 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
1116 }
1117
Tom Hite5a926722014-09-15 01:31:03 +00001118 TWFunc::SetPerformanceMode(true);
bigbiffce8f83c2015-12-12 18:30:21 -05001119
Dees_Troy4a2a1262012-09-18 09:33:47 -04001120 time(&Start);
Ethan Yonker472f5062016-02-25 13:47:30 -06001121
bigbiffce8f83c2015-12-12 18:30:21 -05001122 if (!part_settings->Part->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001123 TWFunc::SetPerformanceMode(false);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001124 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001125 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001126 if (part_settings->Part->Has_SubPartition && !part_settings->adbbackup) {
Dees_Troy8170a922012-09-18 15:40:25 -04001127 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001128 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -04001129
1130 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001131 part_settings->Part = *subpart;
1132 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
1133 part_settings->Part = (*subpart);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001134 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
bigbiffce8f83c2015-12-12 18:30:21 -05001135 if (!(*subpart)->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001136 TWFunc::SetPerformanceMode(false);
Dees_Troy8170a922012-09-18 15:40:25 -04001137 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001138 }
Dees_Troy8170a922012-09-18 15:40:25 -04001139 }
1140 }
1141 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001142 time(&Stop);
Tom Hite5a926722014-09-15 01:31:03 +00001143 TWFunc::SetPerformanceMode(false);
bigbiffce8f83c2015-12-12 18:30:21 -05001144 gui_msg(Msg("restore_part_done=[{1} done ({2} seconds)]")(part_settings->Part->Backup_Display_Name)((int)difftime(Stop, Start)));
1145
Dees_Troy4a2a1262012-09-18 09:33:47 -04001146 return true;
1147}
1148
Ethan Yonker472f5062016-02-25 13:47:30 -06001149int TWPartitionManager::Run_Restore(const string& Restore_Name) {
bigbiffce8f83c2015-12-12 18:30:21 -05001150 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001151 int check_digest;
bigbiffce8f83c2015-12-12 18:30:21 -05001152
Dees_Troy4a2a1262012-09-18 09:33:47 -04001153 time_t rStart, rStop;
1154 time(&rStart);
Dees_Troya13d74f2013-03-24 08:54:55 -05001155 string Restore_List, restore_path;
1156 size_t start_pos = 0, end_pos;
bigbiffce8f83c2015-12-12 18:30:21 -05001157
Ethan Yonkerdcf2b672016-09-13 14:41:53 -05001158 part_settings.Backup_Folder = Restore_Name;
bigbiffce8f83c2015-12-12 18:30:21 -05001159 part_settings.Part = NULL;
1160 part_settings.partition_count = 0;
1161 part_settings.total_restore_size = 0;
1162 part_settings.adbbackup = false;
1163 part_settings.PM_Method = PM_RESTORE;
Dees_Troy43d8b002012-09-17 16:00:01 -04001164
Ethan Yonker74db1572015-10-28 12:44:49 -05001165 gui_msg("restore_started=[RESTORE STARTED]");
1166 gui_msg(Msg("restore_folder=Restore folder: '{1}'")(Restore_Name));
Dees_Troy43d8b002012-09-17 16:00:01 -04001167
Dees_Troy4a2a1262012-09-18 09:33:47 -04001168 if (!Mount_Current_Storage(true))
1169 return false;
1170
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001171 DataManager::GetValue(TW_SKIP_DIGEST_CHECK_VAR, check_digest);
1172 if (check_digest > 0) {
1173 // Check Digest files first before restoring to ensure that all of them match before starting a restore
1174 TWFunc::GUI_Operation_Text(TW_VERIFY_DIGEST_TEXT, gui_parse_text("{@verifying_digest}"));
1175 gui_msg("verifying_digest=Verifying Digest");
Dees_Troya13d74f2013-03-24 08:54:55 -05001176 } else {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001177 gui_msg("skip_digest=Skipping Digest check based on user setting.");
Dees_Troya13d74f2013-03-24 08:54:55 -05001178 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001179 gui_msg("calc_restore=Calculating restore details...");
Dees_Troya13d74f2013-03-24 08:54:55 -05001180 DataManager::GetValue("tw_restore_selected", Restore_List);
bigbiffce8f83c2015-12-12 18:30:21 -05001181
Dees_Troya13d74f2013-03-24 08:54:55 -05001182 if (!Restore_List.empty()) {
1183 end_pos = Restore_List.find(";", start_pos);
1184 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1185 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001186 part_settings.Part = Find_Partition_By_Path(restore_path);
1187 if (part_settings.Part != NULL) {
bigbiffce8f83c2015-12-12 18:30:21 -05001188 if (part_settings.Part->Mount_Read_Only) {
1189 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 -05001190 return false;
1191 }
bigbiffce8f83c2015-12-12 18:30:21 -05001192
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001193 string Full_Filename = part_settings.Backup_Folder + "/" + part_settings.Part->Backup_FileName;
1194
bigbiff4a60bee2021-01-23 14:08:03 -05001195 if (tw_get_default_metadata(Get_Android_Root_Path().c_str()) != 0) {
1196 gui_msg(Msg(msg::kWarning, "restore_system_context=Unable to get default context for {1} -- Android may not boot.")(Get_Android_Root_Path()));
1197 }
1198
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001199 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001200 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001201 part_settings.partition_count++;
1202 part_settings.total_restore_size += part_settings.Part->Get_Restore_Size(&part_settings);
1203 if (part_settings.Part->Has_SubPartition) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001204 TWPartition *parentPart = part_settings.Part;
Dees_Troya13d74f2013-03-24 08:54:55 -05001205 std::vector<TWPartition*>::iterator subpart;
1206
1207 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001208 part_settings.Part = *subpart;
1209 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001210 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001211 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001212 part_settings.total_restore_size += (*subpart)->Get_Restore_Size(&part_settings);
Dees_Troya13d74f2013-03-24 08:54:55 -05001213 }
1214 }
1215 }
1216 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001217 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001218 }
1219 start_pos = end_pos + 1;
1220 end_pos = Restore_List.find(";", start_pos);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001221 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001222 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001223
bigbiffce8f83c2015-12-12 18:30:21 -05001224 if (part_settings.partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001225 gui_err("no_part_restore=No partitions selected for restore.");
Dees_Troy4a2a1262012-09-18 09:33:47 -04001226 return false;
1227 }
1228
bigbiffce8f83c2015-12-12 18:30:21 -05001229 gui_msg(Msg("restore_part_count=Restoring {1} partitions...")(part_settings.partition_count));
1230 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 +00001231 DataManager::SetProgress(0.0);
bigbiffce8f83c2015-12-12 18:30:21 -05001232 ProgressTracking progress(part_settings.total_restore_size);
1233 part_settings.progress = &progress;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001234
Dees_Troya13d74f2013-03-24 08:54:55 -05001235 start_pos = 0;
1236 if (!Restore_List.empty()) {
1237 end_pos = Restore_List.find(";", start_pos);
1238 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1239 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001240
1241 part_settings.Part = Find_Partition_By_Path(restore_path);
1242 if (part_settings.Part != NULL) {
1243 part_settings.partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -05001244 if (!Restore_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001245 return false;
1246 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001247 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001248 }
1249 start_pos = end_pos + 1;
1250 end_pos = Restore_List.find(";", start_pos);
1251 }
1252 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001253 TWFunc::GUI_Operation_Text(TW_UPDATE_SYSTEM_DETAILS_TEXT, gui_parse_text("{@updating_system_details}"));
bigbiff4a60bee2021-01-23 14:08:03 -05001254 tw_set_default_metadata(Get_Android_Root_Path().c_str());
Captain Throwback9d6feb52018-07-27 10:05:24 -04001255 UnMount_By_Path(Get_Android_Root_Path(), false);
Dees_Troy43d8b002012-09-17 16:00:01 -04001256 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001257 UnMount_Main_Partitions();
Dees_Troy4a2a1262012-09-18 09:33:47 -04001258 time(&rStop);
Matt Mower3c366972015-12-25 19:28:31 -06001259 gui_msg(Msg(msg::kHighlight, "restore_completed=[RESTORE COMPLETED IN {1} SECONDS]")((int)difftime(rStop,rStart)));
bigbiff8f5fad32021-11-03 20:39:02 -04001260 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
1261 if (Decrypt_Data && Decrypt_Data->Is_Encrypted)
1262 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 -05001263 DataManager::SetValue("tw_file_progress", "");
bigbiffce8f83c2015-12-12 18:30:21 -05001264
Dees_Troy63c8df72012-09-10 14:02:05 -04001265 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001266}
1267
1268void TWPartitionManager::Set_Restore_Files(string Restore_Name) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001269 // Start with the default values
Dees_Troya13d74f2013-03-24 08:54:55 -05001270 string Restore_List;
Dees_Troy83bd4832013-05-04 12:39:56 +00001271 bool get_date = true, check_encryption = true;
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001272 bool adbbackup = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001273
1274 DataManager::SetValue("tw_restore_encrypted", 0);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001275 if (twadbbu::Check_ADB_Backup_File(Restore_Name)) {
1276 vector<string> adb_files;
1277 adb_files = twadbbu::Get_ADB_Backup_Files(Restore_Name);
1278 for (unsigned int i = 0; i < adb_files.size(); ++i) {
1279 string adb_restore_file = adb_files.at(i);
1280 std::size_t pos = adb_restore_file.find_first_of(".");
1281 std::string path = "/" + adb_restore_file.substr(0, pos);
1282 Restore_List = path + ";";
1283 TWPartition* Part = Find_Partition_By_Path(path);
1284 Part->Backup_FileName = TWFunc::Get_Filename(adb_restore_file);
1285 adbbackup = true;
1286 }
1287 DataManager::SetValue("tw_enable_adb_backup", 1);
Dees_Troy63c8df72012-09-10 14:02:05 -04001288 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001289 else {
1290 DIR* d;
1291 d = opendir(Restore_Name.c_str());
1292 if (d == NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001293 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001294 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Restore_Name)(strerror(errno)));
1295 return;
Dees_Troy63c8df72012-09-10 14:02:05 -04001296 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001297
1298 struct dirent* de;
1299 while ((de = readdir(d)) != NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001300 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001301 // Strip off three components
1302 char str[256];
1303 char* label;
1304 char* fstype = NULL;
1305 char* extn = NULL;
1306 char* ptr;
Dees_Troy63c8df72012-09-10 14:02:05 -04001307
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001308 strcpy(str, de->d_name);
1309 if (strlen(str) <= 2)
1310 continue;
Dees_Troy83bd4832013-05-04 12:39:56 +00001311
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001312 if (get_date) {
1313 char file_path[255];
1314 struct stat st;
1315
1316 strcpy(file_path, Restore_Name.c_str());
1317 strcat(file_path, "/");
1318 strcat(file_path, str);
1319 stat(file_path, &st);
1320 string backup_date = ctime((const time_t*)(&st.st_mtime));
1321 DataManager::SetValue(TW_RESTORE_FILE_DATE, backup_date);
1322 get_date = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001323 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001324
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001325 label = str;
1326 ptr = label;
1327 while (*ptr && *ptr != '.') ptr++;
1328 if (*ptr == '.')
1329 {
1330 *ptr = 0x00;
1331 ptr++;
1332 fstype = ptr;
1333 }
1334 while (*ptr && *ptr != '.') ptr++;
1335 if (*ptr == '.')
1336 {
1337 *ptr = 0x00;
1338 ptr++;
1339 extn = ptr;
1340 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001341
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001342 if (fstype == NULL || extn == NULL || strcmp(fstype, "log") == 0) continue;
1343 int extnlength = strlen(extn);
1344 if (extnlength != 3 && extnlength != 6) continue;
1345 if (extnlength >= 3 && strncmp(extn, "win", 3) != 0) continue;
1346 //if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
Dees_Troy63c8df72012-09-10 14:02:05 -04001347
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001348 if (check_encryption) {
1349 string filename = Restore_Name + "/";
1350 filename += de->d_name;
1351 if (TWFunc::Get_File_Type(filename) == 2) {
1352 LOGINFO("'%s' is encrypted\n", filename.c_str());
1353 DataManager::SetValue("tw_restore_encrypted", 1);
1354 }
1355 }
1356 if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
1357
1358 TWPartition* Part = Find_Partition_By_Path(label);
1359 if (Part == NULL)
1360 {
1361 gui_msg(Msg(msg::kError, "unable_locate_part_backup_name=Unable to locate partition by backup name: '{1}'")(label));
1362 continue;
1363 }
1364
1365 Part->Backup_FileName = de->d_name;
1366 if (strlen(extn) > 3) {
1367 Part->Backup_FileName.resize(Part->Backup_FileName.size() - strlen(extn) + 3);
1368 }
1369
dianlujitaob76a73a2020-04-30 20:33:20 +08001370 if (!Part->Is_SubPartition) {
1371 if (Part->Backup_Path == Get_Android_Root_Path())
1372 Restore_List += "/system;";
1373 else
1374 Restore_List += Part->Backup_Path + ";";
1375 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001376 }
1377 closedir(d);
Dees_Troy63c8df72012-09-10 14:02:05 -04001378 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001379
1380 if (adbbackup) {
1381 Restore_List = "ADB_Backup;";
1382 adbbackup = false;
1383 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001384
Dees_Troya13d74f2013-03-24 08:54:55 -05001385 // Set the final value
1386 DataManager::SetValue("tw_restore_list", Restore_List);
1387 DataManager::SetValue("tw_restore_selected", Restore_List);
Dees_Troy51a0e822012-09-05 15:24:24 -04001388 return;
1389}
1390
1391int TWPartitionManager::Wipe_By_Path(string Path) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001392 std::vector<TWPartition*>::iterator iter;
Mohd Farazfdd49bb2020-04-21 18:56:44 +05301393 std::vector < TWPartition * >::iterator iter1;
Dees_Troy63c8df72012-09-10 14:02:05 -04001394 int ret = false;
1395 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -04001396 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy63c8df72012-09-10 14:02:05 -04001397
dianlujitaob76a73a2020-04-30 20:33:20 +08001398 if (Local_Path == "/system")
1399 Local_Path = Get_Android_Root_Path();
bigbiffaed1bdf2021-08-03 18:21:39 -04001400 if (Path == "/cache") {
1401 TWPartition* cache = Find_Partition_By_Path("/cache");
1402 if (cache == nullptr) {
1403 TWPartition* dat = Find_Partition_By_Path("/data");
1404 if (dat) {
1405 dat->Wipe_Data_Cache();
1406 found = true;
1407 }
1408 }
1409 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001410 // Iterate through all partitions
1411 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -04001412 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Mohd Farazfdd49bb2020-04-21 18:56:44 +05301413 // iterate through all partitions since some legacy devices uses other partitions as vendor causes issues while wiping
1414 (*iter)->Find_Actual_Block_Device();
1415 for (iter1 = Partitions.begin (); iter1 != Partitions.end (); iter1++)
1416 {
1417 (*iter1)->Find_Actual_Block_Device();
1418 if ((*iter)->Actual_Block_Device == (*iter1)->Actual_Block_Device && (*iter)->Mount_Point != (*iter1)->Mount_Point)
1419 (*iter1)->UnMount(false);
1420 }
Dees_Troye58d5262012-09-21 12:27:57 -04001421 if (Path == "/and-sec")
1422 ret = (*iter)->Wipe_AndSec();
1423 else
1424 ret = (*iter)->Wipe();
Dees_Troy63c8df72012-09-10 14:02:05 -04001425 found = true;
1426 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1427 (*iter)->Wipe();
1428 }
1429 }
1430 if (found) {
1431 return ret;
1432 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001433 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 -04001434 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -04001435}
1436
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001437int TWPartitionManager::Wipe_By_Path(string Path, string New_File_System) {
1438 std::vector<TWPartition*>::iterator iter;
1439 int ret = false;
1440 bool found = false;
1441 string Local_Path = TWFunc::Get_Root_Path(Path);
1442
1443 // Iterate through all partitions
1444 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1445 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1446 if (Path == "/and-sec")
1447 ret = (*iter)->Wipe_AndSec();
1448 else
1449 ret = (*iter)->Wipe(New_File_System);
1450 found = true;
1451 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1452 (*iter)->Wipe(New_File_System);
1453 }
1454 }
1455 if (found) {
1456 return ret;
1457 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001458 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 -05001459 return false;
1460}
1461
Dees_Troy51a0e822012-09-05 15:24:24 -04001462int TWPartitionManager::Factory_Reset(void) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001463 std::vector<TWPartition*>::iterator iter;
1464 int ret = true;
1465
1466 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001467 if ((*iter)->Wipe_During_Factory_Reset && (*iter)->Is_Present) {
Ethan Yonker89583ef2015-08-26 09:01:59 -05001468#ifdef TW_OEM_BUILD
1469 if ((*iter)->Mount_Point == "/data") {
1470 if (!(*iter)->Wipe_Encryption())
1471 ret = false;
1472 } else {
1473#endif
1474 if (!(*iter)->Wipe())
1475 ret = false;
1476#ifdef TW_OEM_BUILD
1477 }
1478#endif
Dees_Troy094207a2012-09-26 12:00:39 -04001479 } else if ((*iter)->Has_Android_Secure) {
1480 if (!(*iter)->Wipe_AndSec())
1481 ret = false;
Dees_Troy63c8df72012-09-10 14:02:05 -04001482 }
1483 }
bigbiffad58e1b2020-07-06 20:24:34 -04001484 TWFunc::check_and_run_script("/system/bin/factoryreset.sh", "Factory Reset Script");
Dees_Troy63c8df72012-09-10 14:02:05 -04001485 return ret;
Dees_Troy51a0e822012-09-05 15:24:24 -04001486}
1487
Dees_Troy38bd7602012-09-14 13:33:53 -04001488int TWPartitionManager::Wipe_Dalvik_Cache(void) {
1489 struct stat st;
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001490 vector <string> dir;
Dees_Troy38bd7602012-09-14 13:33:53 -04001491
1492 if (!Mount_By_Path("/data", true))
1493 return false;
1494
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001495 dir.push_back("/data/dalvik-cache");
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001496
bigbiff25d25b92020-06-19 16:07:38 -04001497 std::string cacheDir = TWFunc::get_log_dir();
1498 if (cacheDir == CACHE_LOGS_DIR) {
1499 if (!PartitionManager.Mount_By_Path(CACHE_LOGS_DIR, false)) {
1500 LOGINFO("Unable to mount %s for wiping cache.\n", CACHE_LOGS_DIR);
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001501 }
1502 dir.push_back(cacheDir + "dalvik-cache");
1503 dir.push_back(cacheDir + "/dc");
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001504 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001505
Dees_Troy38bd7602012-09-14 13:33:53 -04001506 TWPartition* sdext = Find_Partition_By_Path("/sd-ext");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001507 if (sdext && sdext->Is_Present && sdext->Mount(false))
1508 {
1509 if (stat("/sd-ext/dalvik-cache", &st) == 0)
1510 {
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001511 dir.push_back("/sd-ext/dalvik-cache");
1512 }
1513 }
1514
bigbiff25d25b92020-06-19 16:07:38 -04001515 if (cacheDir == CACHE_LOGS_DIR) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001516 gui_msg("wiping_cache_dalvik=Wiping Dalvik Cache Directories...");
1517 } else {
1518 gui_msg("wiping_dalvik=Wiping Dalvik Directory...");
1519 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001520 for (unsigned i = 0; i < dir.size(); ++i) {
1521 if (stat(dir.at(i).c_str(), &st) == 0) {
1522 TWFunc::removeDir(dir.at(i), false);
1523 gui_msg(Msg("cleaned=Cleaned: {1}...")(dir.at(i)));
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001524 }
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001525 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001526
bigbiff25d25b92020-06-19 16:07:38 -04001527 if (cacheDir == CACHE_LOGS_DIR) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001528 gui_msg("cache_dalvik_done=-- Dalvik Cache Directories Wipe Complete!");
1529 } else {
1530 gui_msg("dalvik_done=-- Dalvik Directory Wipe Complete!");
1531 }
1532
Dees_Troy38bd7602012-09-14 13:33:53 -04001533 return true;
1534}
1535
1536int TWPartitionManager::Wipe_Rotate_Data(void) {
1537 if (!Mount_By_Path("/data", true))
1538 return false;
1539
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001540 unlink("/data/misc/akmd*");
1541 unlink("/data/misc/rild*");
Dees_Troy2673cec2013-04-02 20:22:16 +00001542 gui_print("Rotation data wiped.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001543 return true;
1544}
1545
1546int TWPartitionManager::Wipe_Battery_Stats(void) {
1547 struct stat st;
1548
1549 if (!Mount_By_Path("/data", true))
1550 return false;
1551
1552 if (0 != stat("/data/system/batterystats.bin", &st)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001553 gui_print("No Battery Stats Found. No Need To Wipe.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001554 } else {
1555 remove("/data/system/batterystats.bin");
Dees_Troy2673cec2013-04-02 20:22:16 +00001556 gui_print("Cleared battery stats.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001557 }
1558 return true;
1559}
1560
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001561int TWPartitionManager::Wipe_Android_Secure(void) {
1562 std::vector<TWPartition*>::iterator iter;
1563 int ret = false;
1564 bool found = false;
1565
1566 // Iterate through all partitions
1567 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1568 if ((*iter)->Has_Android_Secure) {
1569 ret = (*iter)->Wipe_AndSec();
1570 found = true;
1571 }
1572 }
1573 if (found) {
1574 return ret;
1575 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001576 gui_err("no_andsec=No android secure partitions found.");
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001577 }
1578 return false;
1579}
1580
Dees_Troy38bd7602012-09-14 13:33:53 -04001581int TWPartitionManager::Format_Data(void) {
1582 TWPartition* dat = Find_Partition_By_Path("/data");
bigbiffcfa875c2021-06-20 16:20:27 -04001583 TWPartition* metadata = Find_Partition_By_Path("/metadata");
1584 if (metadata != NULL)
1585 metadata->UnMount(false);
Dees_Troy38bd7602012-09-14 13:33:53 -04001586
1587 if (dat != NULL) {
bigbiffab76bd72021-10-11 10:17:48 -04001588 if (android::base::GetBoolProperty("ro.virtual_ab.enabled", false)) {
bigbiffcfa875c2021-06-20 16:20:27 -04001589#ifndef TW_EXCLUDE_APEX
1590 twrpApex apex;
1591 apex.Unmount();
1592#endif
1593 if (metadata != NULL)
1594 metadata->Mount(true);
bigbiffd21252f2021-09-18 15:56:32 -04001595 if (!Check_Pending_Merges())
bigbiffcfa875c2021-06-20 16:20:27 -04001596 return false;
1597 }
Dees_Troy38bd7602012-09-14 13:33:53 -04001598 return dat->Wipe_Encryption();
1599 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001600 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001601 return false;
1602 }
1603 return false;
1604}
1605
1606int TWPartitionManager::Wipe_Media_From_Data(void) {
1607 TWPartition* dat = Find_Partition_By_Path("/data");
1608
1609 if (dat != NULL) {
1610 if (!dat->Has_Data_Media) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001611 LOGERR("This device does not have /data/media\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001612 return false;
1613 }
1614 if (!dat->Mount(true))
1615 return false;
1616
Ethan Yonker74db1572015-10-28 12:44:49 -05001617 gui_msg("wiping_datamedia=Wiping internal storage -- /data/media...");
Ethan Yonker726a0202014-12-16 20:01:38 -06001618 Remove_MTP_Storage(dat->MTP_Storage_ID);
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001619 TWFunc::removeDir("/data/media", false);
xiaolu9416f4f2015-06-04 08:22:23 +08001620 dat->Recreate_Media_Folder();
Ethan Yonker726a0202014-12-16 20:01:38 -06001621 Add_MTP_Storage(dat->MTP_Storage_ID);
Dees_Troy38bd7602012-09-14 13:33:53 -04001622 return true;
1623 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001624 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001625 return false;
1626 }
1627 return false;
1628}
1629
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001630int TWPartitionManager::Repair_By_Path(string Path, bool Display_Error) {
1631 std::vector<TWPartition*>::iterator iter;
1632 int ret = false;
1633 bool found = false;
1634 string Local_Path = TWFunc::Get_Root_Path(Path);
1635
1636 if (Local_Path == "/tmp" || Local_Path == "/")
1637 return true;
1638
1639 // Iterate through all partitions
1640 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1641 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1642 ret = (*iter)->Repair();
1643 found = true;
1644 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1645 (*iter)->Repair();
1646 }
1647 }
1648 if (found) {
1649 return ret;
1650 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001651 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 -05001652 } else {
1653 LOGINFO("Repair: Unable to find partition for path '%s'\n", Local_Path.c_str());
1654 }
1655 return false;
1656}
1657
Ethan Yonkera2719152015-05-28 09:44:41 -05001658int TWPartitionManager::Resize_By_Path(string Path, bool Display_Error) {
1659 std::vector<TWPartition*>::iterator iter;
1660 int ret = false;
1661 bool found = false;
1662 string Local_Path = TWFunc::Get_Root_Path(Path);
1663
1664 if (Local_Path == "/tmp" || Local_Path == "/")
1665 return true;
1666
1667 // Iterate through all partitions
1668 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1669 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1670 ret = (*iter)->Resize();
1671 found = true;
1672 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1673 (*iter)->Resize();
1674 }
1675 }
1676 if (found) {
1677 return ret;
1678 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001679 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 -05001680 } else {
1681 LOGINFO("Resize: Unable to find partition for path '%s'\n", Local_Path.c_str());
1682 }
1683 return false;
1684}
1685
Dees_Troy51a0e822012-09-05 15:24:24 -04001686void TWPartitionManager::Update_System_Details(void) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001687 std::vector<TWPartition*>::iterator iter;
Dees_Troy51127312012-09-08 13:08:49 -04001688 int data_size = 0;
Dees_Troy5bf43922012-09-07 16:07:55 -04001689
Ethan Yonker74db1572015-10-28 12:44:49 -05001690 gui_msg("update_part_details=Updating partition details...");
Dees_Troy5bf43922012-09-07 16:07:55 -04001691 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker1b190162016-12-05 15:25:19 -06001692 (*iter)->Update_Size(true);
Dees_Troy51127312012-09-08 13:08:49 -04001693 if ((*iter)->Can_Be_Mounted) {
Captain Throwback9d6feb52018-07-27 10:05:24 -04001694 if ((*iter)->Mount_Point == Get_Android_Root_Path()) {
Dees_Troy51127312012-09-08 13:08:49 -04001695 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1696 DataManager::SetValue(TW_BACKUP_SYSTEM_SIZE, backup_display_size);
Ethan Yonker76bbd3a2019-05-10 10:50:04 -05001697 TWFunc::Is_TWRP_App_In_System();
Dees_Troy51127312012-09-08 13:08:49 -04001698 } else if ((*iter)->Mount_Point == "/data" || (*iter)->Mount_Point == "/datadata") {
1699 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
1700 } else if ((*iter)->Mount_Point == "/cache") {
1701 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1702 DataManager::SetValue(TW_BACKUP_CACHE_SIZE, backup_display_size);
1703 } else if ((*iter)->Mount_Point == "/sd-ext") {
1704 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1705 DataManager::SetValue(TW_BACKUP_SDEXT_SIZE, backup_display_size);
1706 if ((*iter)->Backup_Size == 0) {
1707 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 0);
1708 DataManager::SetValue(TW_BACKUP_SDEXT_VAR, 0);
1709 } else
1710 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 1);
Dees_Troye58d5262012-09-21 12:27:57 -04001711 } else if ((*iter)->Has_Android_Secure) {
Dees_Troy8170a922012-09-18 15:40:25 -04001712 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troye58d5262012-09-21 12:27:57 -04001713 DataManager::SetValue(TW_BACKUP_ANDSEC_SIZE, backup_display_size);
Dees_Troy8170a922012-09-18 15:40:25 -04001714 if ((*iter)->Backup_Size == 0) {
1715 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 0);
1716 DataManager::SetValue(TW_BACKUP_ANDSEC_VAR, 0);
1717 } else
1718 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 1);
Dees_Troy2c50e182012-09-26 20:05:28 -04001719 } else if ((*iter)->Mount_Point == "/boot") {
1720 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1721 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1722 if ((*iter)->Backup_Size == 0) {
1723 DataManager::SetValue("tw_has_boot_partition", 0);
1724 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1725 } else
1726 DataManager::SetValue("tw_has_boot_partition", 1);
Dees_Troy51127312012-09-08 13:08:49 -04001727 }
Dees_Troyaa9cc402012-10-13 12:14:05 -04001728 } else {
1729 // Handle unmountable partitions in case we reset defaults
1730 if ((*iter)->Mount_Point == "/boot") {
1731 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1732 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1733 if ((*iter)->Backup_Size == 0) {
1734 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 0);
1735 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1736 } else
1737 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 1);
1738 } else if ((*iter)->Mount_Point == "/recovery") {
1739 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1740 DataManager::SetValue(TW_BACKUP_RECOVERY_SIZE, backup_display_size);
1741 if ((*iter)->Backup_Size == 0) {
1742 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 0);
1743 DataManager::SetValue(TW_BACKUP_RECOVERY_VAR, 0);
1744 } else
1745 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 1);
Gary Peck82599a82012-11-21 16:23:12 -08001746 } else if ((*iter)->Mount_Point == "/data") {
1747 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troyaa9cc402012-10-13 12:14:05 -04001748 }
Dees_Troy51127312012-09-08 13:08:49 -04001749 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001750 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001751 gui_msg("update_part_details_done=...done");
Dees_Troy51127312012-09-08 13:08:49 -04001752 DataManager::SetValue(TW_BACKUP_DATA_SIZE, data_size);
1753 string current_storage_path = DataManager::GetCurrentStoragePath();
1754 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
Dees_Troy8170a922012-09-18 15:40:25 -04001755 if (FreeStorage != NULL) {
1756 // Attempt to mount storage
bigbiff25d25b92020-06-19 16:07:38 -04001757 if (!FreeStorage->Mount(false)) {
1758 gui_msg(Msg(msg::kWarning, "unable_to_mount_storage=Unable to mount storage"));
1759 DataManager::SetValue(TW_STORAGE_FREE_SIZE, 0);
1760 } else {
1761 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
Dees_Troy8170a922012-09-18 15:40:25 -04001762 }
1763 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001764 LOGINFO("Unable to find storage partition '%s'.\n", current_storage_path.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -04001765 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001766 if (!Write_Fstab())
Dees_Troy2673cec2013-04-02 20:22:16 +00001767 LOGERR("Error creating fstab\n");
Dees_Troy51a0e822012-09-05 15:24:24 -04001768 return;
1769}
1770
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001771void TWPartitionManager::Post_Decrypt(const string& Block_Device) {
1772 TWPartition* dat = Find_Partition_By_Path("/data");
bigbiffc2cb3852020-09-11 14:03:31 -04001773
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001774 if (dat != NULL) {
bigbiffaed1bdf2021-08-03 18:21:39 -04001775 // reparse for /cache/recovery/command
1776 static constexpr const char* COMMAND_FILE = "/data/cache/command";
1777 if (TWFunc::Path_Exists(COMMAND_FILE)) {
1778 startupArgs startup;
1779 std::string content;
1780 TWFunc::read_file(COMMAND_FILE, content);
1781 std::vector<std::string> args = {content};
1782 startup.processRecoveryArgs(args, 0);
1783 }
1784
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001785 DataManager::SetValue(TW_IS_DECRYPTED, 1);
1786 dat->Is_Decrypted = true;
1787 if (!Block_Device.empty()) {
1788 dat->Decrypted_Block_Device = Block_Device;
1789 gui_msg(Msg("decrypt_success_dev=Data successfully decrypted, new block device: '{1}'")(Block_Device));
1790 } else {
1791 gui_msg("decrypt_success_nodev=Data successfully decrypted");
1792 }
Noah Jacobson81d638d2019-04-28 00:10:07 -04001793 property_set("twrp.decrypt.done", "true");
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001794 dat->Setup_File_System(false);
Noah Jacobson81d638d2019-04-28 00:10:07 -04001795 dat->Current_File_System = dat->Fstab_File_System; // Needed if we're ignoring blkid because encrypted devices start out as emmc
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001796
Noah Jacobson81d638d2019-04-28 00:10:07 -04001797 sleep(1); // Sleep for a bit so that the device will be ready
me-cafebabe09649612022-06-10 21:53:31 +08001798
1799 // Mount only /data
1800 dat->Symlink_Path = ""; // Not to let it to bind mount /data/media again
1801 if (!dat->Mount(false)) {
1802 LOGERR("Unable to mount /data after decryption");
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001803 }
me-cafebabe09649612022-06-10 21:53:31 +08001804
1805 if (dat->Has_Data_Media && TWFunc::Path_Exists("/data/media/0")) {
1806 dat->Storage_Path = "/data/media/0";
1807 } else {
1808 dat->Storage_Path = "/data/media";
1809 }
1810 dat->Symlink_Path = dat->Storage_Path;
1811 DataManager::SetValue("tw_storage_path", dat->Symlink_Path);
Mohd Faraz36072652022-06-18 22:51:55 +02001812 DataManager::SetValue("tw_settings_path", TW_STORAGE_PATH);
me-cafebabe09649612022-06-10 21:53:31 +08001813 LOGINFO("New storage path after decryption: %s\n", dat->Storage_Path.c_str());
1814
epicXa721f952021-01-04 13:01:31 +05301815 DataManager::LoadTWRPFolderInfo();
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001816 Update_System_Details();
nkk71ffb02bd2017-06-04 23:12:16 +03001817 Output_Partition(dat);
bigbiff4fa02b52021-12-30 17:01:44 -05001818 if (!android::base::StartsWith(dat->Actual_Block_Device, "/dev/block/mmcblk")) {
1819 if (!dat->Bind_Mount(false))
1820 LOGERR("Unable to bind mount /sdcard to %s\n", dat->Storage_Path.c_str());
1821 }
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001822 } else
1823 LOGERR("Unable to locate data partition.\n");
1824}
1825
Noah Jacobson81d638d2019-04-28 00:10:07 -04001826void TWPartitionManager::Parse_Users() {
1827#ifdef TW_INCLUDE_FBE
1828 char user_check_result[PROPERTY_VALUE_MAX];
1829 for (int userId = 0; userId <= 9999; userId++) {
1830 string prop = "twrp.user." + to_string(userId) + ".decrypt";
1831 property_get(prop.c_str(), user_check_result, "-1");
1832 if (strcmp(user_check_result, "-1") != 0) {
1833 if (userId < 0 || userId > 9999) {
1834 LOGINFO("Incorrect user id %d\n", userId);
1835 continue;
1836 }
1837 struct users_struct user;
1838 user.userId = to_string(userId);
1839
1840 // Attempt to get name of user. Fallback to user ID if this fails.
Darth98f775972022-08-05 09:39:03 +01001841 std::string path = "/data/system/users/" + to_string(userId) + ".xml";
1842 if (!TWFunc::Check_Xml_Format(path)) {
1843 string oldpath = path;
1844 if (TWFunc::abx_to_xml(oldpath, path)) {
1845 LOGINFO("Android 12+: '%s' has been converted into plain text xml (for user %s).\n", oldpath.c_str(), user.userId.c_str());
1846 }
1847 }
1848 char* userFile = PageManager::LoadFileToBuffer(path, NULL);
me-cafebabec46981d2022-07-29 01:54:43 +08001849 if (userFile == NULL) {
1850 user.userName = to_string(userId);
bigbiffa957f072021-03-07 18:20:29 -05001851 }
Noah Jacobson81d638d2019-04-28 00:10:07 -04001852 else {
me-cafebabec46981d2022-07-29 01:54:43 +08001853 xml_document<> *userXml = new xml_document<>();
1854 userXml->parse<0>(userFile);
1855 xml_node<>* userNode = userXml->first_node("user");
1856 if (userNode == nullptr) {
Noah Jacobson81d638d2019-04-28 00:10:07 -04001857 user.userName = to_string(userId);
me-cafebabec46981d2022-07-29 01:54:43 +08001858 } else {
1859 xml_node<>* nameNode = userNode->first_node("name");
1860 if (nameNode == nullptr)
Noah Jacobson81d638d2019-04-28 00:10:07 -04001861 user.userName = to_string(userId);
me-cafebabec46981d2022-07-29 01:54:43 +08001862 else {
1863 string userName = nameNode->value();
1864 user.userName = userName + " (" + to_string(userId) + ")";
Noah Jacobson81d638d2019-04-28 00:10:07 -04001865 }
1866 }
1867 }
1868
1869 string filename;
bigbiffdf53b6c2022-01-17 19:31:52 -05001870 user.type = android::keystore::Get_Password_Type(userId, filename);
Noah Jacobson81d638d2019-04-28 00:10:07 -04001871
1872 user.isDecrypted = false;
1873 if (strcmp(user_check_result, "1") == 0)
1874 user.isDecrypted = true;
1875 Users_List.push_back(user);
1876 }
1877 }
1878 Check_Users_Decryption_Status();
1879#endif
1880}
1881
1882std::vector<users_struct>* TWPartitionManager::Get_Users_List() {
1883 return &Users_List;
1884}
1885
1886void TWPartitionManager::Mark_User_Decrypted(int userID) {
1887#ifdef TW_INCLUDE_FBE
1888 std::vector<users_struct>::iterator iter;
1889 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1890 if (atoi((*iter).userId.c_str()) == userID) {
1891 (*iter).isDecrypted = true;
1892 string user_prop_decrypted = "twrp.user." + to_string(userID) + ".decrypt";
1893 property_set(user_prop_decrypted.c_str(), "1");
1894 break;
1895 }
1896 }
1897 Check_Users_Decryption_Status();
1898#endif
1899}
1900
1901void TWPartitionManager::Check_Users_Decryption_Status() {
1902#ifdef TW_INCLUDE_FBE
1903 int all_is_decrypted = 1;
1904 std::vector<users_struct>::iterator iter;
1905 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1906 if (!(*iter).isDecrypted) {
1907 LOGINFO("User %s is not decrypted.\n", (*iter).userId.c_str());
1908 all_is_decrypted = 0;
1909 break;
1910 }
1911 }
1912 if (all_is_decrypted == 1) {
1913 LOGINFO("All found users are decrypted.\n");
1914 DataManager::SetValue("tw_all_users_decrypted", "1");
1915 property_set("twrp.all.users.decrypted", "true");
1916 } else
1917 DataManager::SetValue("tw_all_users_decrypted", "0");
1918#endif
1919}
1920
1921int TWPartitionManager::Decrypt_Device(string Password, int user_id) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001922#ifdef TW_INCLUDE_CRYPTO
Captain Throwback37c3aef2021-10-06 11:38:18 -04001923 char crypto_blkdev[PROPERTY_VALUE_MAX];
Ethan Yonker253368a2014-11-25 15:00:52 -06001924 std::vector<TWPartition*>::iterator iter;
Dees_Troy5bf43922012-09-07 16:07:55 -04001925
Ethan Yonker253368a2014-11-25 15:00:52 -06001926 // Mount any partitions that need to be mounted for decrypt
1927 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1928 if ((*iter)->Mount_To_Decrypt) {
1929 (*iter)->Mount(true);
1930 }
a39552696ff55ce2013-01-08 16:14:56 +00001931 }
oshmouna82a7542018-04-10 17:45:55 +02001932 property_set("twrp.mount_to_decrypt", "1");
a39552696ff55ce2013-01-08 16:14:56 +00001933
Captain Throwback37c3aef2021-10-06 11:38:18 -04001934 Set_Crypto_State();
1935 Set_Crypto_Type("block");
Ethan Yonkera1de1492015-11-04 11:58:27 -06001936
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001937 if (DataManager::GetIntValue(TW_IS_FBE)) {
1938#ifdef TW_INCLUDE_FBE
1939 if (!Mount_By_Path("/data", true)) // /data has to be mounted for FBE
1940 return -1;
Noah Jacobson81d638d2019-04-28 00:10:07 -04001941
1942 bool user_need_decrypt = false;
1943 std::vector<users_struct>::iterator iter;
1944 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1945 if (atoi((*iter).userId.c_str()) == user_id && !(*iter).isDecrypted) {
1946 user_need_decrypt = true;
1947 }
1948 }
1949 if (!user_need_decrypt) {
1950 LOGINFO("User %d does not require decryption\n", user_id);
1951 return 0;
1952 }
1953
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001954 int retry_count = 10;
1955 while (!TWFunc::Path_Exists("/data/system/users/gatekeeper.password.key") && --retry_count)
Noah Jacobson81d638d2019-04-28 00:10:07 -04001956 usleep(2000); // A small sleep is needed after mounting /data to ensure reliable decrypt...maybe because of DE?
1957 gui_msg(Msg("decrypting_user_fbe=Attempting to decrypt FBE for user {1}...")(user_id));
bigbiffdf53b6c2022-01-17 19:31:52 -05001958 if (android::keystore::Decrypt_User(user_id, Password)) {
Noah Jacobson81d638d2019-04-28 00:10:07 -04001959 gui_msg(Msg("decrypt_user_success_fbe=User {1} Decrypted Successfully")(user_id));
1960 Mark_User_Decrypted(user_id);
1961 if (user_id == 0) {
1962 // When decrypting user 0 also try all other users
1963 std::vector<users_struct>::iterator iter;
1964 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1965 if ((*iter).userId == "0" || (*iter).isDecrypted)
1966 continue;
1967
1968 int tmp_user_id = atoi((*iter).userId.c_str());
1969 gui_msg(Msg("decrypting_user_fbe=Attempting to decrypt FBE for user {1}...")(tmp_user_id));
bigbiffdf53b6c2022-01-17 19:31:52 -05001970 if (android::keystore::Decrypt_User(tmp_user_id, Password) ||
1971 (Password != "!" && android::keystore::Decrypt_User(tmp_user_id, "!"))) { // "!" means default password
Noah Jacobson81d638d2019-04-28 00:10:07 -04001972 gui_msg(Msg("decrypt_user_success_fbe=User {1} Decrypted Successfully")(tmp_user_id));
1973 Mark_User_Decrypted(tmp_user_id);
1974 } else {
1975 gui_msg(Msg("decrypt_user_fail_fbe=Failed to decrypt user {1}")(tmp_user_id));
1976 }
1977 }
1978 Post_Decrypt("");
1979 }
1980
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001981 return 0;
Noah Jacobson81d638d2019-04-28 00:10:07 -04001982 } else {
1983 gui_msg(Msg(msg::kError, "decrypt_user_fail_fbe=Failed to decrypt user {1}")(user_id));
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001984 }
1985#else
1986 LOGERR("FBE support is not present\n");
1987#endif
1988 return -1;
1989 }
1990
Noah Jacobson81d638d2019-04-28 00:10:07 -04001991 char isdecrypteddata[PROPERTY_VALUE_MAX];
1992 property_get("twrp.decrypt.done", isdecrypteddata, "");
1993 if (strcmp(isdecrypteddata, "true") == 0) {
1994 LOGINFO("Data has no decryption required\n");
1995 return 0;
1996 }
1997
Ethan Yonkerddb63e22017-01-19 14:01:57 -06001998 int pwret = -1;
1999 pid_t pid = fork();
2000 if (pid < 0) {
2001 LOGERR("fork failed\n");
2002 return -1;
2003 } else if (pid == 0) {
2004 // Child process
2005 char cPassword[255];
2006 strcpy(cPassword, Password.c_str());
bigbiff7df87472022-09-24 16:45:03 -04002007 //int ret = cryptfs_check_passwd(cPassword);
2008 exit(0);
Ethan Yonkerddb63e22017-01-19 14:01:57 -06002009 } else {
2010 // Parent
2011 int status;
2012 if (TWFunc::Wait_For_Child_Timeout(pid, &status, "Decrypt", 30))
2013 pwret = -1;
2014 else
2015 pwret = WEXITSTATUS(status) ? -1 : 0;
2016 }
a39552696ff55ce2013-01-08 16:14:56 +00002017
nkk7171c6c502017-01-05 23:55:05 +02002018#ifdef TW_CRYPTO_USE_SYSTEM_VOLD
2019 if (pwret != 0) {
2020 pwret = vold_decrypt(Password);
nkk717d1222a2017-11-06 17:27:21 +02002021 switch (pwret) {
2022 case VD_SUCCESS:
2023 break;
2024 case VD_ERR_MISSING_VDC:
2025 gui_msg(Msg(msg::kError, "decrypt_data_vold_os_missing=Missing files needed for vold decrypt: {1}")("/system/bin/vdc"));
2026 break;
2027 case VD_ERR_MISSING_VOLD:
2028 gui_msg(Msg(msg::kError, "decrypt_data_vold_os_missing=Missing files needed for vold decrypt: {1}")("/system/bin/vold"));
2029 break;
2030 }
nkk7171c6c502017-01-05 23:55:05 +02002031 }
2032#endif // TW_CRYPTO_USE_SYSTEM_VOLD
2033
Ethan Yonker253368a2014-11-25 15:00:52 -06002034 // Unmount any partitions that were needed for decrypt
2035 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2036 if ((*iter)->Mount_To_Decrypt) {
2037 (*iter)->UnMount(false);
2038 }
2039 }
oshmouna82a7542018-04-10 17:45:55 +02002040 property_set("twrp.mount_to_decrypt", "0");
Ethan Yonker253368a2014-11-25 15:00:52 -06002041
a39552696ff55ce2013-01-08 16:14:56 +00002042 if (pwret != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002043 gui_err("fail_decrypt=Failed to decrypt data.");
Dees_Troy5bf43922012-09-07 16:07:55 -04002044 return -1;
2045 }
a39552696ff55ce2013-01-08 16:14:56 +00002046
Dees_Troy5bf43922012-09-07 16:07:55 -04002047 property_get("ro.crypto.fs_crypto_blkdev", crypto_blkdev, "error");
2048 if (strcmp(crypto_blkdev, "error") == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00002049 LOGERR("Error retrieving decrypted data block device.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04002050 } else {
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06002051 Post_Decrypt(crypto_blkdev);
Dees_Troy5bf43922012-09-07 16:07:55 -04002052 }
2053 return 0;
2054#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002055 gui_err("no_crypto_support=No crypto support was compiled into this build.");
Dees_Troy5bf43922012-09-07 16:07:55 -04002056 return -1;
2057#endif
Dees_Troy51a0e822012-09-05 15:24:24 -04002058 return 1;
Dees_Troy51127312012-09-08 13:08:49 -04002059}
2060
Ethan Yonkerb5fab762016-01-28 14:03:33 -06002061int TWPartitionManager::Fix_Contexts(void) {
Ethan Yonkerb5fab762016-01-28 14:03:33 -06002062 std::vector<TWPartition*>::iterator iter;
2063 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2064 if ((*iter)->Has_Data_Media) {
2065 if ((*iter)->Mount(true)) {
2066 if (fixContexts::fixDataMediaContexts((*iter)->Mount_Point) != 0)
2067 return -1;
2068 }
2069 }
2070 }
Dees_Troy1a650e62012-10-19 20:54:32 -04002071 UnMount_Main_Partitions();
Ethan Yonker74db1572015-10-28 12:44:49 -05002072 gui_msg("done=Done.");
Ethan Yonkerb5fab762016-01-28 14:03:33 -06002073 return 0;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04002074}
2075
Ethan Yonker66a19492015-12-10 10:19:45 -06002076TWPartition* TWPartitionManager::Find_Next_Storage(string Path, bool Exclude_Data_Media) {
Ethan Yonker47360be2014-04-01 10:34:34 -05002077 std::vector<TWPartition*>::iterator iter = Partitions.begin();
2078
2079 if (!Path.empty()) {
2080 string Search_Path = TWFunc::Get_Root_Path(Path);
2081 for (; iter != Partitions.end(); iter++) {
Matt Mower9a561dd2016-02-22 21:25:51 -06002082 if ((*iter)->Mount_Point == Search_Path) {
Ethan Yonker47360be2014-04-01 10:34:34 -05002083 iter++;
2084 break;
2085 }
2086 }
2087 }
2088
2089 for (; iter != Partitions.end(); iter++) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002090 if (Exclude_Data_Media && (*iter)->Has_Data_Media) {
2091 // do nothing, do not return this type of partition
2092 } else if ((*iter)->Is_Storage && (*iter)->Is_Present) {
Ethan Yonker47360be2014-04-01 10:34:34 -05002093 return (*iter);
2094 }
2095 }
2096
2097 return NULL;
2098}
2099
Dees_Troyd21618c2012-10-14 18:48:49 -04002100int TWPartitionManager::Open_Lun_File(string Partition_Path, string Lun_File) {
Dees_Troyd21618c2012-10-14 18:48:49 -04002101 TWPartition* Part = Find_Partition_By_Path(Partition_Path);
2102
2103 if (Part == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002104 LOGINFO("Unable to locate '%s' for USB storage mode.", Partition_Path.c_str());
2105 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 -04002106 return false;
2107 }
Ethan Yonker47360be2014-04-01 10:34:34 -05002108 LOGINFO("USB mount '%s', '%s' > '%s'\n", Partition_Path.c_str(), Part->Actual_Block_Device.c_str(), Lun_File.c_str());
2109 if (!Part->UnMount(true) || !Part->Is_Present)
Dees_Troyd21618c2012-10-14 18:48:49 -04002110 return false;
2111
bigbiff22851b92021-09-01 16:46:57 -04002112 if (!TWFunc::write_to_file(Lun_File, Part->Actual_Block_Device)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00002113 LOGERR("Unable to write to ums lunfile '%s': (%s)\n", Lun_File.c_str(), strerror(errno));
Dees_Troyd21618c2012-10-14 18:48:49 -04002114 return false;
2115 }
Dees_Troyd21618c2012-10-14 18:48:49 -04002116 return true;
2117}
2118
Dees_Troy8170a922012-09-18 15:40:25 -04002119int TWPartitionManager::usb_storage_enable(void) {
Dees_Troy8170a922012-09-18 15:40:25 -04002120 char lun_file[255];
Dees_Troyd21618c2012-10-14 18:48:49 -04002121 bool has_multiple_lun = false;
Dees_Troy8170a922012-09-18 15:40:25 -04002122
Ethan Yonker47360be2014-04-01 10:34:34 -05002123 string Lun_File_str = CUSTOM_LUN_FILE;
2124 size_t found = Lun_File_str.find("%");
2125 if (found != string::npos) {
2126 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
2127 if (TWFunc::Path_Exists(lun_file))
2128 has_multiple_lun = true;
2129 }
Ethan Yonker726a0202014-12-16 20:01:38 -06002130 mtp_was_enabled = TWFunc::Toggle_MTP(false); // Must disable MTP for USB Storage
Ethan Yonker47360be2014-04-01 10:34:34 -05002131 if (!has_multiple_lun) {
2132 LOGINFO("Device doesn't have multiple lun files, mount current storage\n");
2133 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
2134 if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) == "/data") {
Ethan Yonker66a19492015-12-10 10:19:45 -06002135 TWPartition* Mount = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05002136 if (Mount) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002137 if (!Open_Lun_File(Mount->Mount_Point, lun_file)) {
2138 goto error_handle;
2139 }
Ethan Yonker47360be2014-04-01 10:34:34 -05002140 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002141 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002142 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05002143 }
2144 } else if (!Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file)) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002145 goto error_handle;
Dees_Troy8170a922012-09-18 15:40:25 -04002146 }
Dees_Troy8170a922012-09-18 15:40:25 -04002147 } else {
Ethan Yonker47360be2014-04-01 10:34:34 -05002148 LOGINFO("Device has multiple lun files\n");
2149 TWPartition* Mount1;
2150 TWPartition* Mount2;
Dees_Troy8170a922012-09-18 15:40:25 -04002151 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
Ethan Yonker66a19492015-12-10 10:19:45 -06002152 Mount1 = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05002153 if (Mount1) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002154 if (!Open_Lun_File(Mount1->Mount_Point, lun_file)) {
2155 goto error_handle;
2156 }
Matt Moweree717062014-04-26 01:46:46 -05002157 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
Ethan Yonker66a19492015-12-10 10:19:45 -06002158 Mount2 = Find_Next_Storage(Mount1->Mount_Point, true);
Matt Mower1fdcdb72016-01-25 20:53:47 -06002159 if (Mount2 && Mount2->Mount_Point != Mount1->Mount_Point) {
Matt Moweree717062014-04-26 01:46:46 -05002160 Open_Lun_File(Mount2->Mount_Point, lun_file);
nkk717aa6fc62017-03-24 18:28:03 +02002161 // Mimic single lun code: Mount CurrentStoragePath if it's not /data
2162 } else if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) != "/data") {
2163 Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file);
Ethan Yonker47360be2014-04-01 10:34:34 -05002164 }
nkk717aa6fc62017-03-24 18:28:03 +02002165 // Mimic single lun code: Mount CurrentStoragePath if it's not /data
2166 } else if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) != "/data" && !Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file)) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002167 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002168 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05002169 }
Dees_Troy8170a922012-09-18 15:40:25 -04002170 }
Matt Mowerb6ef4782014-11-06 02:24:36 -06002171 property_set("sys.storage.ums_enabled", "1");
HandyMenny37d42992014-10-26 22:15:59 +01002172 property_set("sys.usb.config", "mass_storage,adb");
Dees_Troy8170a922012-09-18 15:40:25 -04002173 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002174error_handle:
2175 if (mtp_was_enabled)
2176 if (!Enable_MTP())
2177 Disable_MTP();
2178 return false;
Dees_Troy8170a922012-09-18 15:40:25 -04002179}
2180
2181int TWPartitionManager::usb_storage_disable(void) {
bigbiff22851b92021-09-01 16:46:57 -04002182 int index, ret = 0;
Dees_Troy2673cec2013-04-02 20:22:16 +00002183 char lun_file[255], ch[2] = {0, 0};
2184 string str = ch;
Dees_Troy8170a922012-09-18 15:40:25 -04002185
2186 for (index=0; index<2; index++) {
2187 sprintf(lun_file, CUSTOM_LUN_FILE, index);
bigbiff22851b92021-09-01 16:46:57 -04002188 if (!TWFunc::write_to_file(lun_file, str)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00002189 break;
bigbiff22851b92021-09-01 16:46:57 -04002190 ret = -1;
Dees_Troy8170a922012-09-18 15:40:25 -04002191 }
Dees_Troy8170a922012-09-18 15:40:25 -04002192 }
Dees_Troye58d5262012-09-21 12:27:57 -04002193 Mount_All_Storage();
2194 Update_System_Details();
Dees_Troycfd73ef2012-10-12 16:52:00 -04002195 UnMount_Main_Partitions();
Matt Mowerd9cb9062013-12-14 20:34:45 -06002196 property_set("sys.storage.ums_enabled", "0");
HandyMenny37d42992014-10-26 22:15:59 +01002197 property_set("sys.usb.config", "adb");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002198 if (mtp_was_enabled)
2199 if (!Enable_MTP())
2200 Disable_MTP();
Dees_Troy2673cec2013-04-02 20:22:16 +00002201 if (ret < 0 && index == 0) {
2202 LOGERR("Unable to write to ums lunfile '%s'.", lun_file);
2203 return false;
2204 } else {
2205 return true;
2206 }
Dees_Troy8170a922012-09-18 15:40:25 -04002207 return true;
Dees_Troy812660f2012-09-20 09:55:17 -04002208}
2209
2210void TWPartitionManager::Mount_All_Storage(void) {
2211 std::vector<TWPartition*>::iterator iter;
2212
2213 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2214 if ((*iter)->Is_Storage)
2215 (*iter)->Mount(false);
2216 }
Dees_Troy2c50e182012-09-26 20:05:28 -04002217}
Dees_Troy9350b8d2012-09-27 12:38:38 -04002218
Dees_Troyd0384ef2012-10-12 12:15:42 -04002219void TWPartitionManager::UnMount_Main_Partitions(void) {
2220 // Unmounts system and data if data is not data/media
2221 // Also unmounts boot if boot is mountable
Dees_Troy2673cec2013-04-02 20:22:16 +00002222 LOGINFO("Unmounting main partitions...\n");
Dees_Troyd0384ef2012-10-12 12:15:42 -04002223
Mohd Faraze3948ec2020-08-14 01:40:59 +00002224 TWPartition *Partition = Find_Partition_By_Path ("/vendor");
Dees_Troyd0384ef2012-10-12 12:15:42 -04002225
Mohd Faraze3948ec2020-08-14 01:40:59 +00002226 if (Partition != NULL) UnMount_By_Path("/vendor", false);
2227 UnMount_By_Path (Get_Android_Root_Path(), true);
2228 Partition = Find_Partition_By_Path ("/product");
2229 if (Partition != NULL) UnMount_By_Path("/product", false);
Ethan Yonker6277c792014-09-15 14:54:30 -05002230 if (!datamedia)
2231 UnMount_By_Path("/data", true);
2232
Mohd Faraze3948ec2020-08-14 01:40:59 +00002233 Partition = Find_Partition_By_Path ("/boot");
2234 if (Partition != NULL && Partition->Can_Be_Mounted)
2235 Partition->UnMount(true);
Dees_Troyd0384ef2012-10-12 12:15:42 -04002236}
2237
Dees_Troy9350b8d2012-09-27 12:38:38 -04002238int TWPartitionManager::Partition_SDCard(void) {
Matt Mower23d8aae2017-01-06 14:30:33 -06002239 char temp[255];
Ethan Yonker483e9f42016-01-11 22:21:18 -06002240 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 -04002241 int ext, swap, total_size = 0, fat_size;
Dees_Troy9350b8d2012-09-27 12:38:38 -04002242
Ethan Yonker74db1572015-10-28 12:44:49 -05002243 gui_msg("start_partition_sd=Partitioning SD Card...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06002244
2245 // Locate and validate device to partition
2246 TWPartition* SDCard = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
2247
Ethan Yonker66a19492015-12-10 10:19:45 -06002248 if (SDCard->Is_Adopted_Storage)
2249 SDCard->Revert_Adopted();
2250
Ethan Yonker1eff6cd2014-09-15 13:30:42 -05002251 if (SDCard == NULL || !SDCard->Removable || SDCard->Has_Data_Media) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002252 gui_err("partition_sd_locate=Unable to locate device to partition.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002253 return false;
2254 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002255
2256 // Unmount everything
Dees_Troy9350b8d2012-09-27 12:38:38 -04002257 if (!SDCard->UnMount(true))
2258 return false;
2259 TWPartition* SDext = Find_Partition_By_Path("/sd-ext");
2260 if (SDext != NULL) {
2261 if (!SDext->UnMount(true))
2262 return false;
2263 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002264 char* swappath = getenv("SWAPPATH");
2265 if (swappath != NULL) {
2266 LOGINFO("Unmounting swap at '%s'\n", swappath);
2267 umount(swappath);
2268 }
Vojtech Bocek05534202013-09-11 08:11:56 +02002269
Ethan Yonker483e9f42016-01-11 22:21:18 -06002270 // Determine block device
2271 if (SDCard->Alternate_Block_Device.empty()) {
2272 SDCard->Find_Actual_Block_Device();
2273 Device = SDCard->Actual_Block_Device;
2274 // Just use the root block device
2275 Device.resize(strlen("/dev/block/mmcblkX"));
2276 } else {
2277 Device = SDCard->Alternate_Block_Device;
2278 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04002279
2280 // Find the size of the block device:
Ethan Yonker483e9f42016-01-11 22:21:18 -06002281 total_size = (int)(TWFunc::IOCTL_Get_Block_Size(Device.c_str()) / (1048576));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002282
2283 DataManager::GetValue("tw_sdext_size", ext);
2284 DataManager::GetValue("tw_swap_size", swap);
2285 DataManager::GetValue("tw_sdpart_file_system", ext_format);
2286 fat_size = total_size - ext - swap;
Ethan Yonker483e9f42016-01-11 22:21:18 -06002287 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);
2288
2289 // Determine partition sizes
2290 if (swap == 0 && ext == 0) {
2291 fat_str = "-0";
2292 } else {
2293 memset(temp, 0, sizeof(temp));
2294 sprintf(temp, "%i", fat_size);
2295 fat_str = temp;
2296 fat_str += "MB";
2297 }
2298 if (swap == 0) {
2299 ext_str = "-0";
2300 } else {
2301 memset(temp, 0, sizeof(temp));
2302 sprintf(temp, "%i", ext);
2303 ext_str = "+";
2304 ext_str += temp;
2305 ext_str += "MB";
2306 }
2307
Dees_Troy9350b8d2012-09-27 12:38:38 -04002308 if (ext + swap > total_size) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002309 gui_err("ext_swap_size=EXT + Swap size is larger than sdcard size.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002310 return false;
2311 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002312
Ethan Yonker74db1572015-10-28 12:44:49 -05002313 gui_msg("remove_part_table=Removing partition table...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06002314 Command = "sgdisk --zap-all " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002315 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002316 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002317 gui_err("unable_rm_part=Unable to remove partition table.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002318 Update_System_Details();
2319 return false;
2320 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002321 gui_msg(Msg("create_part=Creating {1} partition...")("FAT32"));
Captain Throwback9643a802016-05-27 11:44:51 -04002322 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 +00002323 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002324 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002325 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("FAT32"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002326 return false;
2327 }
2328 if (ext > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002329 gui_msg(Msg("create_part=Creating {1} partition...")("EXT"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06002330 Command = "sgdisk --new=0:0:" + ext_str + " --change-name=0:\"Linux filesystem\" " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002331 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002332 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002333 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("EXT"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002334 Update_System_Details();
2335 return false;
2336 }
2337 }
2338 if (swap > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002339 gui_msg(Msg("create_part=Creating {1} partition...")("swap"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06002340 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 +00002341 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002342 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002343 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("swap"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002344 Update_System_Details();
2345 return false;
2346 }
2347 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002348
2349 // Convert GPT to MBR
2350 Command = "sgdisk --gpttombr " + Device;
2351 if (TWFunc::Exec_Cmd(Command) != 0)
2352 LOGINFO("Failed to covert partition GPT to MBR\n");
2353
2354 // Tell the kernel to rescan the partition table
2355 int fd = open(Device.c_str(), O_RDONLY);
2356 ioctl(fd, BLKRRPART, 0);
2357 close(fd);
2358
2359 string format_device = Device;
2360 if (Device.substr(0, 17) == "/dev/block/mmcblk")
2361 format_device += "p";
2362
2363 // Format new partitions to proper file system
2364 if (fat_size > 0) {
2365 Command = "mkfs.fat " + format_device + "1";
2366 TWFunc::Exec_Cmd(Command);
2367 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04002368 if (ext > 0) {
2369 if (SDext == NULL) {
Ethan Yonker483e9f42016-01-11 22:21:18 -06002370 Command = "mke2fs -t " + ext_format + " -m 0 " + format_device + "2";
2371 gui_msg(Msg("format_sdext_as=Formatting sd-ext as {1}...")(ext_format));
2372 LOGINFO("Formatting sd-ext after partitioning, command: '%s'\n", Command.c_str());
2373 TWFunc::Exec_Cmd(Command);
2374 } else {
2375 SDext->Wipe(ext_format);
Dees_Troy9350b8d2012-09-27 12:38:38 -04002376 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002377 }
2378 if (swap > 0) {
2379 Command = "mkswap " + format_device;
2380 if (ext > 0)
2381 Command += "3";
2382 else
2383 Command += "2";
Vojtech Bocek05534202013-09-11 08:11:56 +02002384 TWFunc::Exec_Cmd(Command);
Dees_Troy9350b8d2012-09-27 12:38:38 -04002385 }
2386
Ethan Yonker483e9f42016-01-11 22:21:18 -06002387 // recreate TWRP folder and rewrite settings - these will be gone after sdcard is partitioned
2388 if (SDCard->Mount(true)) {
2389 string TWRP_Folder = SDCard->Mount_Point + "/TWRP";
2390 mkdir(TWRP_Folder.c_str(), 0777);
2391 DataManager::Flush();
2392 }
2393
Dees_Troy9350b8d2012-09-27 12:38:38 -04002394 Update_System_Details();
Ethan Yonker74db1572015-10-28 12:44:49 -05002395 gui_msg("part_complete=Partitioning complete.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002396 return true;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04002397}
Dees_Troya13d74f2013-03-24 08:54:55 -05002398
2399void TWPartitionManager::Get_Partition_List(string ListType, std::vector<PartitionList> *Partition_List) {
2400 std::vector<TWPartition*>::iterator iter;
2401 if (ListType == "mount") {
2402 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002403 if ((*iter)->Can_Be_Mounted) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002404 struct PartitionList part;
2405 part.Display_Name = (*iter)->Display_Name;
2406 part.Mount_Point = (*iter)->Mount_Point;
2407 part.selected = (*iter)->Is_Mounted();
2408 Partition_List->push_back(part);
2409 }
2410 }
2411 } else if (ListType == "storage") {
2412 char free_space[255];
2413 string Current_Storage = DataManager::GetCurrentStoragePath();
2414 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2415 if ((*iter)->Is_Storage) {
2416 struct PartitionList part;
2417 sprintf(free_space, "%llu", (*iter)->Free / 1024 / 1024);
2418 part.Display_Name = (*iter)->Storage_Name + " (";
2419 part.Display_Name += free_space;
2420 part.Display_Name += "MB)";
2421 part.Mount_Point = (*iter)->Storage_Path;
2422 if ((*iter)->Storage_Path == Current_Storage)
2423 part.selected = 1;
2424 else
2425 part.selected = 0;
2426 Partition_List->push_back(part);
2427 }
2428 }
2429 } else if (ListType == "backup") {
2430 char backup_size[255];
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002431 unsigned long long Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -05002432 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002433 if ((*iter)->Can_Be_Backed_Up && !(*iter)->Is_SubPartition && (*iter)->Is_Present) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002434 struct PartitionList part;
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002435 Backup_Size = (*iter)->Backup_Size;
2436 if ((*iter)->Has_SubPartition) {
2437 std::vector<TWPartition*>::iterator subpart;
2438
2439 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
2440 if ((*subpart)->Is_SubPartition && (*subpart)->Can_Be_Backed_Up && (*subpart)->Is_Present && (*subpart)->SubPartition_Of == (*iter)->Mount_Point)
2441 Backup_Size += (*subpart)->Backup_Size;
2442 }
2443 }
2444 sprintf(backup_size, "%llu", Backup_Size / 1024 / 1024);
Dees_Troya13d74f2013-03-24 08:54:55 -05002445 part.Display_Name = (*iter)->Backup_Display_Name + " (";
2446 part.Display_Name += backup_size;
2447 part.Display_Name += "MB)";
2448 part.Mount_Point = (*iter)->Backup_Path;
2449 part.selected = 0;
2450 Partition_List->push_back(part);
2451 }
2452 }
2453 } else if (ListType == "restore") {
2454 string Restore_List, restore_path;
2455 TWPartition* restore_part = NULL;
2456
2457 DataManager::GetValue("tw_restore_list", Restore_List);
2458 if (!Restore_List.empty()) {
2459 size_t start_pos = 0, end_pos = Restore_List.find(";", start_pos);
2460 while (end_pos != string::npos && start_pos < Restore_List.size()) {
2461 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04002462 struct PartitionList part;
2463 if (restore_path.compare("ADB_Backup") == 0) {
2464 part.Display_Name = "ADB Backup";
2465 part.Mount_Point = "ADB Backup";
2466 part.selected = 1;
2467 Partition_List->push_back(part);
2468 break;
2469 }
Dees_Troy59df9262013-06-19 14:53:57 -05002470 if ((restore_part = Find_Partition_By_Path(restore_path)) != NULL) {
Dees Troy4159aed2014-02-28 17:24:43 +00002471 if ((restore_part->Backup_Name == "recovery" && !restore_part->Can_Be_Backed_Up) || restore_part->Is_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002472 // Don't allow restore of recovery (causes problems on some devices)
Dees_Troy59df9262013-06-19 14:53:57 -05002473 // Don't add subpartitions to the list of items
Dees_Troya13d74f2013-03-24 08:54:55 -05002474 } else {
Dees_Troya13d74f2013-03-24 08:54:55 -05002475 part.Display_Name = restore_part->Backup_Display_Name;
2476 part.Mount_Point = restore_part->Backup_Path;
2477 part.selected = 1;
2478 Partition_List->push_back(part);
2479 }
2480 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002481 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05002482 }
2483 start_pos = end_pos + 1;
2484 end_pos = Restore_List.find(";", start_pos);
2485 }
2486 }
2487 } else if (ListType == "wipe") {
2488 struct PartitionList dalvik;
Ethan Yonker74db1572015-10-28 12:44:49 -05002489 dalvik.Display_Name = gui_parse_text("{@dalvik}");
Dees_Troya13d74f2013-03-24 08:54:55 -05002490 dalvik.Mount_Point = "DALVIK";
2491 dalvik.selected = 0;
2492 Partition_List->push_back(dalvik);
2493 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2494 if ((*iter)->Wipe_Available_in_GUI && !(*iter)->Is_SubPartition) {
2495 struct PartitionList part;
2496 part.Display_Name = (*iter)->Display_Name;
2497 part.Mount_Point = (*iter)->Mount_Point;
2498 part.selected = 0;
2499 Partition_List->push_back(part);
2500 }
2501 if ((*iter)->Has_Android_Secure) {
2502 struct PartitionList part;
2503 part.Display_Name = (*iter)->Backup_Display_Name;
2504 part.Mount_Point = (*iter)->Backup_Path;
2505 part.selected = 0;
2506 Partition_List->push_back(part);
2507 }
Dees_Troy74fb2e92013-04-15 14:35:47 +00002508 if ((*iter)->Has_Data_Media) {
2509 struct PartitionList datamedia;
2510 datamedia.Display_Name = (*iter)->Storage_Name;
2511 datamedia.Mount_Point = "INTERNAL";
2512 datamedia.selected = 0;
2513 Partition_List->push_back(datamedia);
2514 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002515 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002516 } else if (ListType == "flashimg") {
2517 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2518 if ((*iter)->Can_Flash_Img && (*iter)->Is_Present) {
2519 struct PartitionList part;
2520 part.Display_Name = (*iter)->Backup_Display_Name;
2521 part.Mount_Point = (*iter)->Backup_Path;
2522 part.selected = 0;
2523 Partition_List->push_back(part);
2524 }
2525 }
Ethan Yonker53796e72019-01-11 22:49:52 -06002526 if (DataManager::GetIntValue("tw_has_repack_tools") != 0 && DataManager::GetIntValue("tw_has_boot_slots") != 0) {
2527 TWPartition* boot = Find_Partition_By_Path("/boot");
2528 if (boot) {
2529 // Allow flashing kernels and ramdisks
2530 struct PartitionList repack_ramdisk;
2531 repack_ramdisk.Display_Name = gui_lookup("install_twrp_ramdisk", "Install Recovery Ramdisk");
2532 repack_ramdisk.Mount_Point = "/repack_ramdisk";
2533 repack_ramdisk.selected = 0;
2534 Partition_List->push_back(repack_ramdisk);
2535 /*struct PartitionList repack_kernel; For now let's leave repacking kernels under advanced only
2536 repack_kernel.Display_Name = gui_lookup("install_kernel", "Install Kernel");
2537 repack_kernel.Mount_Point = "/repack_kernel";
2538 repack_kernel.selected = 0;
2539 Partition_List->push_back(repack_kernel);*/
2540 }
2541 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002542 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00002543 LOGERR("Unknown list type '%s' requested for TWPartitionManager::Get_Partition_List\n", ListType.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -05002544 }
2545}
2546
2547int TWPartitionManager::Fstab_Processed(void) {
2548 return Partitions.size();
2549}
Dees_Troyd93bda52013-07-03 19:55:19 +00002550
2551void TWPartitionManager::Output_Storage_Fstab(void) {
2552 std::vector<TWPartition*>::iterator iter;
2553 char storage_partition[255];
bigbiff bigbiffe4bdb152019-03-23 18:33:17 -04002554 std::string Temp;
bigbiff25d25b92020-06-19 16:07:38 -04002555 std::string cacheDir = TWFunc::get_log_dir();
bigbiff bigbiffe4bdb152019-03-23 18:33:17 -04002556
2557 if (cacheDir.empty()) {
2558 LOGINFO("Unable to find cache directory\n");
2559 return;
2560 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002561
bigbiff25d25b92020-06-19 16:07:38 -04002562 std::string storageFstab = TWFunc::get_log_dir() + "recovery/storage.fstab";
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002563 FILE *fp = fopen(storageFstab.c_str(), "w");
Dees_Troyd93bda52013-07-03 19:55:19 +00002564
2565 if (fp == NULL) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002566 gui_msg(Msg(msg::kError, "unable_to_open=Unable to open '{1}'.")(storageFstab));
Dees_Troyd93bda52013-07-03 19:55:19 +00002567 return;
2568 }
2569
2570 // Iterate through all partitions
2571 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2572 if ((*iter)->Is_Storage) {
2573 Temp = (*iter)->Storage_Path + ";" + (*iter)->Storage_Name + ";\n";
2574 strcpy(storage_partition, Temp.c_str());
2575 fwrite(storage_partition, sizeof(storage_partition[0]), strlen(storage_partition) / sizeof(storage_partition[0]), fp);
2576 }
2577 }
2578 fclose(fp);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02002579}
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002580
2581TWPartition *TWPartitionManager::Get_Default_Storage_Partition()
2582{
2583 TWPartition *res = NULL;
2584 for (std::vector<TWPartition*>::iterator iter = Partitions.begin(); iter != Partitions.end(); ++iter) {
Matt Mowera8a89d12016-12-30 18:10:37 -06002585 if (!(*iter)->Is_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002586 continue;
2587
Matt Mowera8a89d12016-12-30 18:10:37 -06002588 if ((*iter)->Is_Settings_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002589 return *iter;
2590
Matt Mowera8a89d12016-12-30 18:10:37 -06002591 if (!res)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002592 res = *iter;
2593 }
2594 return res;
2595}
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002596
2597bool TWPartitionManager::Enable_MTP(void) {
2598#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002599 if (mtppid) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002600 gui_err("mtp_already_enabled=MTP already enabled");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002601 return true;
2602 }
Ethan Yonker726a0202014-12-16 20:01:38 -06002603
2604 int mtppipe[2];
2605
2606 if (pipe(mtppipe) < 0) {
2607 LOGERR("Error creating MTP pipe\n");
2608 return false;
2609 }
2610
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002611 char old_value[PROPERTY_VALUE_MAX];
Matt Mowere07f0102017-02-23 17:40:00 -06002612 property_get("sys.usb.config", old_value, "");
2613 if (strcmp(old_value, "mtp,adb") != 0) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002614 char vendor[PROPERTY_VALUE_MAX];
2615 char product[PROPERTY_VALUE_MAX];
2616 property_set("sys.usb.config", "none");
2617 property_get("usb.vendor", vendor, "18D1");
2618 property_get("usb.product.mtpadb", product, "4EE2");
2619 string vendorstr = vendor;
2620 string productstr = product;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002621 TWFunc::write_to_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2622 TWFunc::write_to_file("/sys/class/android_usb/android0/idProduct", productstr);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002623 property_set("sys.usb.config", "mtp,adb");
2624 }
Matt Mower653a1702017-02-16 10:38:52 -06002625 /* To enable MTP debug, use the twrp command line feature:
Ethan Yonker6d154c42014-09-03 14:19:43 -05002626 * twrp set tw_mtp_debug 1
2627 */
2628 twrpMtp *mtp = new twrpMtp(DataManager::GetIntValue("tw_mtp_debug"));
Ethan Yonker1b039202015-01-30 10:08:48 -06002629 mtppid = mtp->forkserver(mtppipe);
2630 if (mtppid) {
2631 close(mtppipe[0]); // Host closes read side
2632 mtp_write_fd = mtppipe[1];
2633 DataManager::SetValue("tw_mtp_enabled", 1);
2634 Add_All_MTP_Storage();
2635 return true;
Ethan Yonker726a0202014-12-16 20:01:38 -06002636 } else {
2637 close(mtppipe[0]);
2638 close(mtppipe[1]);
Ethan Yonker74db1572015-10-28 12:44:49 -05002639 gui_err("mtp_fail=Failed to enable MTP");
Ethan Yonker1b039202015-01-30 10:08:48 -06002640 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002641 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002642#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002643 gui_err("no_mtp=MTP support not included");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002644#endif
2645 DataManager::SetValue("tw_mtp_enabled", 0);
2646 return false;
2647}
2648
Ethan Yonker1b039202015-01-30 10:08:48 -06002649void TWPartitionManager::Add_All_MTP_Storage(void) {
2650#ifdef TW_HAS_MTP
2651 std::vector<TWPartition*>::iterator iter;
2652
2653 if (!mtppid)
2654 return; // MTP is not enabled
2655
2656 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2657 if ((*iter)->Is_Storage && (*iter)->Is_Present && (*iter)->Mount(false))
2658 Add_Remove_MTP_Storage((*iter), MTP_MESSAGE_ADD_STORAGE);
2659 }
2660#else
2661 return;
2662#endif
2663}
2664
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002665bool TWPartitionManager::Disable_MTP(void) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002666 char old_value[PROPERTY_VALUE_MAX];
bigbiff bigbiffaf32bb92018-12-18 18:39:53 -05002667 property_set("sys.usb.ffs.mtp.ready", "0");
Matt Mowere07f0102017-02-23 17:40:00 -06002668 property_get("sys.usb.config", old_value, "");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002669 if (strcmp(old_value, "adb") != 0) {
2670 char vendor[PROPERTY_VALUE_MAX];
2671 char product[PROPERTY_VALUE_MAX];
2672 property_set("sys.usb.config", "none");
2673 property_get("usb.vendor", vendor, "18D1");
Matt Mowere07f0102017-02-23 17:40:00 -06002674 property_get("usb.product.adb", product, "D001");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002675 string vendorstr = vendor;
2676 string productstr = product;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002677 TWFunc::write_to_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2678 TWFunc::write_to_file("/sys/class/android_usb/android0/idProduct", productstr);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002679 usleep(2000);
2680 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002681#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002682 if (mtppid) {
Ethan Yonker8613dc02014-09-11 09:28:20 -05002683 LOGINFO("Disabling MTP\n");
2684 int status;
2685 kill(mtppid, SIGKILL);
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002686 mtppid = 0;
Ethan Yonker8613dc02014-09-11 09:28:20 -05002687 // We don't care about the exit value, but this prevents a zombie process
2688 waitpid(mtppid, &status, 0);
Ethan Yonker726a0202014-12-16 20:01:38 -06002689 close(mtp_write_fd);
2690 mtp_write_fd = -1;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002691 }
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002692#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002693 property_set("sys.usb.config", "adb");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002694#ifdef TW_HAS_MTP
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002695 DataManager::SetValue("tw_mtp_enabled", 0);
2696 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002697#endif
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002698 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002699}
Ethan Yonker726a0202014-12-16 20:01:38 -06002700
2701TWPartition* TWPartitionManager::Find_Partition_By_MTP_Storage_ID(unsigned int Storage_ID) {
2702 std::vector<TWPartition*>::iterator iter;
2703
2704 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2705 if ((*iter)->MTP_Storage_ID == Storage_ID)
2706 return (*iter);
2707 }
2708 return NULL;
2709}
2710
2711bool TWPartitionManager::Add_Remove_MTP_Storage(TWPartition* Part, int message_type) {
2712#ifdef TW_HAS_MTP
2713 struct mtpmsg mtp_message;
2714
2715 if (!mtppid)
2716 return false; // MTP is disabled
2717
2718 if (mtp_write_fd < 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002719 LOGINFO("MTP: mtp_write_fd is not set\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002720 return false;
2721 }
2722
2723 if (Part) {
Ethan Yonker4bfabab2014-12-29 09:15:37 -06002724 if (Part->MTP_Storage_ID == 0)
2725 return false;
Ethan Yonker726a0202014-12-16 20:01:38 -06002726 if (message_type == MTP_MESSAGE_REMOVE_STORAGE) {
2727 mtp_message.message_type = MTP_MESSAGE_REMOVE_STORAGE; // Remove
2728 LOGINFO("sending message to remove %i\n", Part->MTP_Storage_ID);
2729 mtp_message.storage_id = Part->MTP_Storage_ID;
2730 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002731 LOGINFO("error sending message to remove storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002732 return false;
2733 } else {
2734 LOGINFO("Message sent, remove storage ID: %i\n", Part->MTP_Storage_ID);
2735 return true;
2736 }
2737 } else if (message_type == MTP_MESSAGE_ADD_STORAGE && Part->Is_Mounted()) {
2738 mtp_message.message_type = MTP_MESSAGE_ADD_STORAGE; // Add
2739 mtp_message.storage_id = Part->MTP_Storage_ID;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002740 if (Part->Storage_Path.size() >= sizeof(mtp_message.path)) {
2741 LOGERR("Storage path '%s' too large for mtpmsg\n", Part->Storage_Path.c_str());
2742 return false;
2743 }
2744 strcpy(mtp_message.path, Part->Storage_Path.c_str());
2745 if (Part->Storage_Name.size() >= sizeof(mtp_message.display)) {
2746 LOGERR("Storage name '%s' too large for mtpmsg\n", Part->Storage_Name.c_str());
2747 return false;
2748 }
2749 strcpy(mtp_message.display, Part->Storage_Name.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002750 mtp_message.maxFileSize = Part->Get_Max_FileSize();
Ethan Yonker1b039202015-01-30 10:08:48 -06002751 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 -06002752 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002753 LOGINFO("error sending message to add storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002754 return false;
2755 } else {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002756 LOGINFO("Message sent, add storage ID: %i '%s'\n", Part->MTP_Storage_ID, mtp_message.path);
Ethan Yonker726a0202014-12-16 20:01:38 -06002757 return true;
2758 }
2759 } else {
2760 LOGERR("Unknown MTP message type: %i\n", message_type);
2761 }
2762 } else {
2763 // This hopefully never happens as the error handling should
2764 // occur in the calling function.
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002765 LOGINFO("TWPartitionManager::Add_Remove_MTP_Storage NULL partition given\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002766 }
2767 return true;
2768#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002769 gui_err("no_mtp=MTP support not included");
Ethan Yonker726a0202014-12-16 20:01:38 -06002770 DataManager::SetValue("tw_mtp_enabled", 0);
2771 return false;
2772#endif
2773}
2774
2775bool TWPartitionManager::Add_MTP_Storage(string Mount_Point) {
2776#ifdef TW_HAS_MTP
2777 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2778 if (Part) {
2779 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2780 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002781 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002782 }
2783#endif
2784 return false;
2785}
2786
2787bool TWPartitionManager::Add_MTP_Storage(unsigned int Storage_ID) {
2788#ifdef TW_HAS_MTP
2789 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2790 if (Part) {
2791 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2792 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002793 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002794 }
2795#endif
2796 return false;
2797}
2798
2799bool TWPartitionManager::Remove_MTP_Storage(string Mount_Point) {
2800#ifdef TW_HAS_MTP
2801 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2802 if (Part) {
2803 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2804 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002805 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002806 }
2807#endif
2808 return false;
2809}
2810
2811bool TWPartitionManager::Remove_MTP_Storage(unsigned int Storage_ID) {
2812#ifdef TW_HAS_MTP
2813 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2814 if (Part) {
2815 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2816 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002817 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002818 }
2819#endif
2820 return false;
2821}
Ethan Yonker96af84a2015-01-05 14:58:36 -06002822
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002823bool TWPartitionManager::Flash_Image(string& path, string& filename) {
bigbiffad58e1b2020-07-06 20:24:34 -04002824 twrpRepacker repacker;
Matt Mower23d8aae2017-01-06 14:30:33 -06002825 int partition_count = 0;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002826 TWPartition* flash_part = NULL;
bigbiffce8f83c2015-12-12 18:30:21 -05002827 string Flash_List, flash_path, full_filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002828 size_t start_pos = 0, end_pos = 0;
2829
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002830 full_filename = path + "/" + filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002831
bigbiffce8f83c2015-12-12 18:30:21 -05002832 gui_msg("image_flash_start=[IMAGE FLASH STARTED]");
2833 gui_msg(Msg("img_to_flash=Image to flash: '{1}'")(full_filename));
2834
2835 if (!TWFunc::Path_Exists(full_filename)) {
2836 if (!Mount_By_Path(full_filename, true)) {
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002837 return false;
2838 }
bigbiffce8f83c2015-12-12 18:30:21 -05002839 if (!TWFunc::Path_Exists(full_filename)) {
2840 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")(full_filename));
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002841 return false;
2842 }
2843 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002844
Ethan Yonker53796e72019-01-11 22:49:52 -06002845 DataManager::GetValue("tw_flash_partition", Flash_List);
2846 Repack_Type repack = REPLACE_NONE;
2847 if (Flash_List == "/repack_ramdisk;") {
2848 repack = REPLACE_RAMDISK;
2849 } else if (Flash_List == "/repack_kernel;") {
2850 repack = REPLACE_KERNEL;
2851 }
2852 if (repack != REPLACE_NONE) {
2853 Repack_Options_struct Repack_Options;
2854 Repack_Options.Type = repack;
2855 Repack_Options.Disable_Verity = false;
2856 Repack_Options.Disable_Force_Encrypt = false;
2857 Repack_Options.Backup_First = DataManager::GetIntValue("tw_repack_backup_first") != 0;
bigbiffad58e1b2020-07-06 20:24:34 -04002858 return repacker.Repack_Image_And_Flash(full_filename, Repack_Options);
Ethan Yonker53796e72019-01-11 22:49:52 -06002859 }
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002860 PartitionSettings part_settings;
2861 part_settings.Backup_Folder = path;
2862 unsigned long long total_bytes = TWFunc::Get_File_Size(full_filename);
2863 ProgressTracking progress(total_bytes);
2864 part_settings.progress = &progress;
2865 part_settings.adbbackup = false;
2866 part_settings.PM_Method = PM_RESTORE;
Ethan Yonker74db1572015-10-28 12:44:49 -05002867 gui_msg("calc_restore=Calculating restore details...");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002868 if (!Flash_List.empty()) {
2869 end_pos = Flash_List.find(";", start_pos);
2870 while (end_pos != string::npos && start_pos < Flash_List.size()) {
2871 flash_path = Flash_List.substr(start_pos, end_pos - start_pos);
2872 flash_part = Find_Partition_By_Path(flash_path);
2873 if (flash_part != NULL) {
2874 partition_count++;
2875 if (partition_count > 1) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002876 gui_err("too_many_flash=Too many partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002877 return false;
2878 }
2879 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002880 gui_msg(Msg(msg::kError, "flash_unable_locate=Unable to locate '{1}' partition for flashing.")(flash_path));
Ethan Yonker96af84a2015-01-05 14:58:36 -06002881 return false;
2882 }
2883 start_pos = end_pos + 1;
2884 end_pos = Flash_List.find(";", start_pos);
2885 }
2886 }
2887
2888 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002889 gui_err("no_part_flash=No partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002890 return false;
2891 }
2892
2893 DataManager::SetProgress(0.0);
2894 if (flash_part) {
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002895 flash_part->Backup_FileName = filename;
2896 if (!flash_part->Flash_Image(&part_settings))
Ethan Yonker96af84a2015-01-05 14:58:36 -06002897 return false;
2898 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002899 gui_err("invalid_flash=Invalid flash partition specified.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002900 return false;
2901 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002902 gui_highlight("flash_done=IMAGE FLASH COMPLETED]");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002903 return true;
2904}
Ethan Yonker74db1572015-10-28 12:44:49 -05002905
2906void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value) {
2907 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2908 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002909 if (part->Is_Adopted_Storage) {
2910 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2911 part->Backup_Display_Name = part->Display_Name;
2912 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2913 } else {
2914 part->Display_Name = gui_lookup(resource_name, default_value);
2915 part->Backup_Display_Name = part->Display_Name;
2916 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002917 }
2918}
2919
2920void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value, const char* storage_resource_name, const char* storage_default_value) {
2921 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2922 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002923 if (part->Is_Adopted_Storage) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06002924 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup("data_backup", "Data (excl. storage)");
Ethan Yonker66a19492015-12-10 10:19:45 -06002925 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
Ethan Yonker66a19492015-12-10 10:19:45 -06002926 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2927 } else {
2928 part->Display_Name = gui_lookup(resource_name, default_value);
2929 part->Backup_Display_Name = part->Display_Name;
2930 if (part->Is_Storage)
2931 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2932 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002933 }
2934}
2935
Ethan Yonker01f4e032017-02-03 15:30:52 -06002936void 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) {
2937 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2938 if (part) {
2939 if (part->Is_Adopted_Storage) {
2940 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup(backup_name, backup_default);
2941 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2942 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2943 } else {
2944 part->Display_Name = gui_lookup(resource_name, default_value);
2945 part->Backup_Display_Name = gui_lookup(backup_name, backup_default);
2946 if (part->Is_Storage)
2947 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2948 }
2949 }
2950}
2951
Ethan Yonker74db1572015-10-28 12:44:49 -05002952void TWPartitionManager::Translate_Partition_Display_Names() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002953 LOGINFO("Translating partition display names\n");
Ethan Yonker74db1572015-10-28 12:44:49 -05002954 Translate_Partition("/system", "system", "System");
2955 Translate_Partition("/system_image", "system_image", "System Image");
2956 Translate_Partition("/vendor", "vendor", "Vendor");
2957 Translate_Partition("/vendor_image", "vendor_image", "Vendor Image");
2958 Translate_Partition("/cache", "cache", "Cache");
Ethan Yonker74db1572015-10-28 12:44:49 -05002959 Translate_Partition("/boot", "boot", "Boot");
2960 Translate_Partition("/recovery", "recovery", "Recovery");
2961 if (!datamedia) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06002962 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage");
Ethan Yonker74db1572015-10-28 12:44:49 -05002963 Translate_Partition("/sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2964 Translate_Partition("/internal_sd", "sdcard", "SDCard", "sdcard", "SDCard");
2965 Translate_Partition("/internal_sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2966 Translate_Partition("/emmc", "sdcard", "SDCard", "sdcard", "SDCard");
Ethan Yonker01f4e032017-02-03 15:30:52 -06002967 } else {
2968 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05002969 }
Ethan Yonker01f4e032017-02-03 15:30:52 -06002970 Translate_Partition("/external_sd", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
2971 Translate_Partition("/external_sdcard", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05002972 Translate_Partition("/usb-otg", "usbotg", "USB OTG", "usbotg", "USB OTG");
2973 Translate_Partition("/sd-ext", "sdext", "SD-EXT");
2974
2975 // Android secure is a special case
2976 TWPartition* part = PartitionManager.Find_Partition_By_Path("/and-sec");
2977 if (part)
2978 part->Backup_Display_Name = gui_lookup("android_secure", "Android Secure");
2979
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002980 std::vector<TWPartition*>::iterator sysfs;
2981 for (sysfs = Partitions.begin(); sysfs != Partitions.end(); sysfs++) {
2982 if (!(*sysfs)->Sysfs_Entry.empty()) {
2983 Translate_Partition((*sysfs)->Mount_Point.c_str(), "autostorage", "Storage", "autostorage", "Storage");
2984 }
2985 }
2986
Ethan Yonker74db1572015-10-28 12:44:49 -05002987 // This updates the text on all of the storage selection buttons in the GUI
2988 DataManager::SetBackupFolder();
2989}
Ethan Yonker66a19492015-12-10 10:19:45 -06002990
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002991bool TWPartitionManager::Decrypt_Adopted() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002992#ifdef TW_INCLUDE_CRYPTO
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002993 bool ret = false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002994 if (!Mount_By_Path("/data", false)) {
2995 LOGERR("Cannot decrypt adopted storage because /data will not mount\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002996 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002997 }
Darth98f775972022-08-05 09:39:03 +01002998
2999 std::string path = "/data/system/storage.xml";
3000 if (!TWFunc::Check_Xml_Format(path)) {
3001 std::string oldpath = path;
3002 if (TWFunc::abx_to_xml(oldpath, path)) {
3003 LOGINFO("Android 12+: '%s' has been converted into plain text xml (%s).\n", oldpath.c_str(), path.c_str());
3004 }
3005 }
3006
Ethan Yonker66a19492015-12-10 10:19:45 -06003007 LOGINFO("Decrypt adopted storage starting\n");
Darth98f775972022-08-05 09:39:03 +01003008 char* xmlFile = PageManager::LoadFileToBuffer(path, NULL);
Ethan Yonker66a19492015-12-10 10:19:45 -06003009 xml_document<> *doc = NULL;
3010 xml_node<>* volumes = NULL;
Ethan Yonker66a19492015-12-10 10:19:45 -06003011 string Primary_Storage_UUID = "";
3012 if (xmlFile != NULL) {
3013 LOGINFO("successfully loaded storage.xml\n");
3014 doc = new xml_document<>();
3015 doc->parse<0>(xmlFile);
3016 volumes = doc->first_node("volumes");
3017 if (volumes) {
3018 xml_attribute<>* psuuid = volumes->first_attribute("primaryStorageUuid");
3019 if (psuuid) {
3020 Primary_Storage_UUID = psuuid->value();
3021 }
3022 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003023 } else {
3024 LOGINFO("No /data/system/storage.xml for adopted storage\n");
3025 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06003026 }
3027 std::vector<TWPartition*>::iterator adopt;
3028 for (adopt = Partitions.begin(); adopt != Partitions.end(); adopt++) {
Peter Cai439d60c2019-10-17 08:49:10 +08003029 if ((*adopt)->Removable && !(*adopt)->Is_Present && (*adopt)->Adopted_Mount_Delay > 0) {
3030 // On some devices, the external mmc driver takes some time
3031 // to recognize the card, in which case the "actual block device"
3032 // would not have been found yet. We wait the specified delay
3033 // and then try again.
3034 LOGINFO("Sleeping %d seconds for adopted storage.\n", (*adopt)->Adopted_Mount_Delay);
3035 sleep((*adopt)->Adopted_Mount_Delay);
3036 (*adopt)->Find_Actual_Block_Device();
3037 }
3038
Ethan Yonker66a19492015-12-10 10:19:45 -06003039 if ((*adopt)->Removable && (*adopt)->Is_Present) {
3040 if ((*adopt)->Decrypt_Adopted() == 0) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003041 ret = true;
Ethan Yonker66a19492015-12-10 10:19:45 -06003042 if (volumes) {
3043 xml_node<>* volume = volumes->first_node("volume");
3044 while (volume) {
3045 xml_attribute<>* guid = volume->first_attribute("partGuid");
3046 if (guid) {
3047 string GUID = (*adopt)->Adopted_GUID.c_str();
3048 GUID.insert(8, "-");
3049 GUID.insert(13, "-");
3050 GUID.insert(18, "-");
3051 GUID.insert(23, "-");
3052
3053 if (strcasecmp(GUID.c_str(), guid->value()) == 0) {
3054 xml_attribute<>* attr = volume->first_attribute("nickname");
nkk71ffb02bd2017-06-04 23:12:16 +03003055 if (attr && attr->value() && strlen(attr->value()) > 0) {
Ethan Yonker66a19492015-12-10 10:19:45 -06003056 (*adopt)->Storage_Name = attr->value();
3057 (*adopt)->Display_Name = (*adopt)->Storage_Name;
3058 (*adopt)->Backup_Display_Name = (*adopt)->Storage_Name;
3059 LOGINFO("storage name from storage.xml is '%s'\n", attr->value());
3060 }
3061 attr = volume->first_attribute("fsUuid");
3062 if (attr && !Primary_Storage_UUID.empty() && strcmp(Primary_Storage_UUID.c_str(), attr->value()) == 0) {
3063 TWPartition* Dat = Find_Partition_By_Path("/data");
3064 if (Dat) {
3065 LOGINFO("Internal storage is found on adopted storage '%s'\n", (*adopt)->Display_Name.c_str());
3066 LOGINFO("Changing '%s' to point to '%s'\n", Dat->Symlink_Mount_Point.c_str(), (*adopt)->Storage_Path.c_str());
3067 (*adopt)->Symlink_Mount_Point = Dat->Symlink_Mount_Point;
3068 Dat->Symlink_Mount_Point = "";
3069 // Toggle mounts to ensure that the symlink mount point (probably /sdcard) is mounted to the right location
3070 Dat->UnMount(false);
3071 Dat->Mount(false);
3072 (*adopt)->UnMount(false);
3073 (*adopt)->Mount(false);
Ethan Yonker66a19492015-12-10 10:19:45 -06003074 }
3075 }
3076 break;
3077 }
3078 }
3079 volume = volume->next_sibling("volume");
3080 }
3081 }
nkk71ffb02bd2017-06-04 23:12:16 +03003082 Update_System_Details();
3083 Output_Partition((*adopt));
Ethan Yonker66a19492015-12-10 10:19:45 -06003084 }
3085 }
3086 }
3087 if (xmlFile) {
3088 doc->clear();
3089 delete doc;
3090 free(xmlFile);
3091 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003092 return ret;
Ethan Yonker66a19492015-12-10 10:19:45 -06003093#else
3094 LOGINFO("Decrypt_Adopted: no crypto support\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003095 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06003096#endif
3097}
Ethan Yonkerfcf3f242016-02-16 12:30:26 -06003098
3099void TWPartitionManager::Remove_Partition_By_Path(string Path) {
3100 std::vector<TWPartition*>::iterator iter;
3101 string Local_Path = TWFunc::Get_Root_Path(Path);
3102
3103 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3104 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
3105 LOGINFO("Found and erasing '%s' from partition list\n", Local_Path.c_str());
3106 Partitions.erase(iter);
3107 return;
3108 }
3109 }
3110}
Ethan Yonker1b190162016-12-05 15:25:19 -06003111
bigbiff74a46272021-07-31 19:02:26 -04003112void TWPartitionManager::Override_Active_Slot(const string& Slot) {
3113 LOGINFO("Overriding slot to '%s'\n", Slot.c_str());
3114 Active_Slot_Display = Slot;
3115 DataManager::SetValue("tw_active_slot", Slot);
nebrassy07c4b902021-11-21 03:32:47 +01003116 PartitionManager.Update_System_Details();
bigbiff74a46272021-07-31 19:02:26 -04003117}
3118
Ethan Yonker1b190162016-12-05 15:25:19 -06003119void TWPartitionManager::Set_Active_Slot(const string& Slot) {
3120 if (Slot != "A" && Slot != "B") {
3121 LOGERR("Set_Active_Slot invalid slot '%s'\n", Slot.c_str());
3122 return;
3123 }
3124 if (Active_Slot_Display == Slot)
3125 return;
3126 LOGINFO("Setting active slot %s\n", Slot.c_str());
3127#ifdef AB_OTA_UPDATER
3128 if (!Active_Slot_Display.empty()) {
dianlujitaoa90dc462021-12-16 16:56:26 +08003129 android::sp<IBootControl> module = IBootControl::getService();
3130 if (module == nullptr) {
Ethan Yonker1b190162016-12-05 15:25:19 -06003131 LOGERR("Error getting bootctrl module.\n");
3132 } else {
dianlujitaoa90dc462021-12-16 16:56:26 +08003133 uint32_t slot_number = 0;
Ethan Yonker1b190162016-12-05 15:25:19 -06003134 if (Slot == "B")
3135 slot_number = 1;
dianlujitaoa90dc462021-12-16 16:56:26 +08003136 CommandResult result;
3137 auto ret = module->setActiveBootSlot(slot_number, [&result]
3138 (const CommandResult &cb_result) { result = cb_result; });
3139 if (!ret.isOk() || !result.success)
Ethan Yonker1b190162016-12-05 15:25:19 -06003140 gui_msg(Msg(msg::kError, "unable_set_boot_slot=Error changing bootloader boot slot to {1}")(Slot));
3141 }
3142 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
3143 }
3144#else
3145 LOGERR("Boot slot feature not present\n");
3146#endif
3147 Active_Slot_Display = Slot;
3148 if (Fstab_Processed())
3149 Update_System_Details();
3150}
3151string TWPartitionManager::Get_Active_Slot_Suffix() {
3152 if (Active_Slot_Display == "A")
3153 return "_a";
3154 return "_b";
3155}
3156string TWPartitionManager::Get_Active_Slot_Display() {
3157 return Active_Slot_Display;
3158}
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003159
Captain Throwback9d6feb52018-07-27 10:05:24 -04003160string TWPartitionManager::Get_Android_Root_Path() {
bigbiff8da46fa2020-09-08 16:42:34 -04003161 return "/system_root";
Captain Throwback9d6feb52018-07-27 10:05:24 -04003162}
3163
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003164void TWPartitionManager::Remove_Uevent_Devices(const string& Mount_Point) {
3165 std::vector<TWPartition*>::iterator iter;
3166
3167 for (iter = Partitions.begin(); iter != Partitions.end(); ) {
3168 if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Mount_Point) {
3169 TWPartition *part = *iter;
3170 LOGINFO("%s was removed by uevent data\n", (*iter)->Mount_Point.c_str());
3171 (*iter)->UnMount(false);
3172 rmdir((*iter)->Mount_Point.c_str());
3173 iter = Partitions.erase(iter);
3174 delete part;
3175 } else {
3176 iter++;
3177 }
3178 }
3179}
3180
3181void TWPartitionManager::Handle_Uevent(const Uevent_Block_Data& uevent_data) {
3182 std::vector<TWPartition*>::iterator iter;
3183
3184 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3185 if (!(*iter)->Sysfs_Entry.empty()) {
3186 string device;
3187 size_t wildcard = (*iter)->Sysfs_Entry.find("*");
3188 if (wildcard != string::npos) {
3189 device = (*iter)->Sysfs_Entry.substr(0, wildcard);
3190 } else {
3191 device = (*iter)->Sysfs_Entry;
3192 }
3193 if (device == uevent_data.sysfs_path.substr(0, device.size())) {
3194 // Found a match
3195 if (uevent_data.action == "add") {
3196 (*iter)->Primary_Block_Device = "/dev/block/" + uevent_data.block_device;
3197 (*iter)->Alternate_Block_Device = (*iter)->Primary_Block_Device;
3198 (*iter)->Is_Present = true;
3199 LOGINFO("Found a match '%s' '%s'\n", uevent_data.block_device.c_str(), device.c_str());
3200 if (!Decrypt_Adopted()) {
3201 LOGINFO("No adopted storage so finding actual block device\n");
3202 (*iter)->Find_Actual_Block_Device();
3203 }
3204 return;
3205 } else if (uevent_data.action == "remove") {
3206 (*iter)->Is_Present = false;
3207 (*iter)->Primary_Block_Device = "";
3208 (*iter)->Actual_Block_Device = "";
3209 Remove_Uevent_Devices((*iter)->Mount_Point);
3210 return;
3211 }
3212 }
3213 }
3214 }
bigbiffee7b7ff2020-03-23 15:08:27 -04003215
3216 if (!PartitionManager.Get_Super_Status())
3217 LOGINFO("Found no matching fstab entry for uevent device '%s' - %s\n", uevent_data.sysfs_path.c_str(), uevent_data.action.c_str());
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003218}
3219
3220void TWPartitionManager::setup_uevent() {
3221 struct sockaddr_nl nls;
3222
3223 if (uevent_pfd.fd >= 0) {
3224 LOGINFO("uevent already set up\n");
3225 return;
3226 }
3227
3228 // Open hotplug event netlink socket
3229 memset(&nls,0,sizeof(struct sockaddr_nl));
3230 nls.nl_family = AF_NETLINK;
3231 nls.nl_pid = getpid();
3232 nls.nl_groups = -1;
3233 uevent_pfd.events = POLLIN;
3234 uevent_pfd.fd = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_KOBJECT_UEVENT);
3235 if (uevent_pfd.fd==-1) {
3236 LOGERR("uevent not root\n");
3237 return;
3238 }
3239
3240 // Listen to netlink socket
3241 if (::bind(uevent_pfd.fd, (struct sockaddr *) &nls, sizeof(struct sockaddr_nl)) < 0) {
3242 LOGERR("Bind failed\n");
3243 return;
3244 }
3245 set_select_fd();
3246 Coldboot();
3247}
3248
3249Uevent_Block_Data TWPartitionManager::get_event_block_values(char *buf, int len) {
3250 Uevent_Block_Data ret;
3251 ret.subsystem = "";
3252 char *ptr = buf;
3253 const char *end = buf + len;
3254
3255 buf[len - 1] = '\0';
3256 while (ptr < end) {
3257 if (strncmp(ptr, "ACTION=", strlen("ACTION=")) == 0) {
3258 ptr += strlen("ACTION=");
3259 ret.action = ptr;
3260 } else if (strncmp(ptr, "SUBSYSTEM=", strlen("SUBSYSTEM=")) == 0) {
3261 ptr += strlen("SUBSYSTEM=");
3262 ret.subsystem = ptr;
3263 } else if (strncmp(ptr, "DEVTYPE=", strlen("DEVTYPE=")) == 0) {
3264 ptr += strlen("DEVTYPE=");
3265 ret.type = ptr;
3266 } else if (strncmp(ptr, "DEVPATH=", strlen("DEVPATH=")) == 0) {
3267 ptr += strlen("DEVPATH=");
3268 ret.sysfs_path += ptr;
3269 } else if (strncmp(ptr, "DEVNAME=", strlen("DEVNAME=")) == 0) {
3270 ptr += strlen("DEVNAME=");
3271 ret.block_device += ptr;
3272 } else if (strncmp(ptr, "MAJOR=", strlen("MAJOR=")) == 0) {
3273 ptr += strlen("MAJOR=");
3274 ret.major = atoi(ptr);
3275 } else if (strncmp(ptr, "MINOR=", strlen("MINOR=")) == 0) {
3276 ptr += strlen("MINOR=");
3277 ret.minor = atoi(ptr);
3278 }
3279 ptr += strlen(ptr) + 1;
3280 }
3281 return ret;
3282}
3283
3284void TWPartitionManager::read_uevent() {
3285 char buf[1024];
3286
3287 int len = recv(uevent_pfd.fd, buf, sizeof(buf), MSG_DONTWAIT);
3288 if (len == -1) {
Mauronofrio Matarresec1bb76e2019-08-04 17:02:41 +01003289 LOGINFO("recv error on uevent\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003290 return;
3291 }
3292 /*int i = 0; // Print all uevent output for test /debug
3293 while (i<len) {
3294 printf("%s\n", buf+i);
3295 i += strlen(buf+i)+1;
3296 }*/
3297 Uevent_Block_Data uevent_data = get_event_block_values(buf, len);
3298 if (uevent_data.subsystem == "block" && uevent_data.type == "disk") {
3299 PartitionManager.Handle_Uevent(uevent_data);
3300 }
3301}
3302
3303void TWPartitionManager::close_uevent() {
3304 if (uevent_pfd.fd > 0)
3305 close(uevent_pfd.fd);
3306 uevent_pfd.fd = -1;
3307}
3308
3309void TWPartitionManager::Add_Partition(TWPartition* Part) {
3310 Partitions.push_back(Part);
3311}
3312
3313void TWPartitionManager::Coldboot_Scan(std::vector<string> *sysfs_entries, const string& Path, int depth) {
3314 string Real_Path = Path;
3315 char real_path[PATH_MAX];
3316 if (realpath(Path.c_str(), &real_path[0])) {
3317 string Real_Path = real_path;
3318 std::vector<string>::iterator iter;
3319 for (iter = sysfs_entries->begin(); iter != sysfs_entries->end(); iter++) {
3320 if (Real_Path.find((*iter)) != string::npos) {
3321 string Write_Path = Real_Path + "/uevent";
3322 if (TWFunc::Path_Exists(Write_Path)) {
3323 const char* write_val = "add\n";
3324 TWFunc::write_to_file(Write_Path, write_val);
3325 break;
3326 }
3327 }
3328 }
3329 }
3330
3331 DIR* d = opendir(Path.c_str());
3332 if (d != NULL) {
3333 struct dirent* de;
3334 while ((de = readdir(d)) != NULL) {
3335 if (de->d_name[0] == '.' || (de->d_type != DT_DIR && depth > 0))
3336 continue;
3337 if (strlen(de->d_name) >= 4 && (strncmp(de->d_name, "ram", 3) == 0 || strncmp(de->d_name, "loop", 4) == 0))
3338 continue;
3339
3340 string item = Path + "/";
3341 item.append(de->d_name);
3342 Coldboot_Scan(sysfs_entries, item, depth + 1);
3343 }
3344 closedir(d);
3345 }
3346}
3347
3348void TWPartitionManager::Coldboot() {
3349 std::vector<TWPartition*>::iterator iter;
3350 std::vector<string> sysfs_entries;
3351
3352 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3353 if (!(*iter)->Sysfs_Entry.empty()) {
3354 size_t wildcard_pos = (*iter)->Sysfs_Entry.find("*");
3355 if (wildcard_pos == string::npos)
3356 wildcard_pos = (*iter)->Sysfs_Entry.size();
3357 sysfs_entries.push_back((*iter)->Sysfs_Entry.substr(0, wildcard_pos));
3358 }
3359 }
3360
3361 if (sysfs_entries.size() > 0)
3362 Coldboot_Scan(&sysfs_entries, "/sys/block", 0);
3363}
Ethan Yonker53796e72019-01-11 22:49:52 -06003364
3365bool TWPartitionManager::Prepare_Empty_Folder(const std::string& Folder) {
3366 if (TWFunc::Path_Exists(Folder))
3367 TWFunc::removeDir(Folder, false);
3368 return TWFunc::Recursive_Mkdir(Folder);
3369}
3370
bigbiffcfa875c2021-06-20 16:20:27 -04003371std::string TWPartitionManager::Get_Bare_Partition_Name(std::string Mount_Point) {
3372 if (Mount_Point == "/system_root")
3373 return "system";
3374 else
3375 return TWFunc::Remove_Beginning_Slash(Mount_Point);
3376}
3377
bigbiffee7b7ff2020-03-23 15:08:27 -04003378bool TWPartitionManager::Prepare_Super_Volume(TWPartition* twrpPart) {
3379 Fstab fstab;
bigbiffcfa875c2021-06-20 16:20:27 -04003380 std::string bare_partition_name = Get_Bare_Partition_Name(twrpPart->Get_Mount_Point());
bigbiffee7b7ff2020-03-23 15:08:27 -04003381
bigbiff998f8392021-08-06 19:19:33 -04003382 Super_Partition_List.push_back(bare_partition_name);
bigbiffee7b7ff2020-03-23 15:08:27 -04003383 LOGINFO("Trying to prepare %s from super partition\n", bare_partition_name.c_str());
3384
3385 std::string blk_device_partition;
3386#ifdef AB_OTA_UPDATER
3387 blk_device_partition = bare_partition_name + PartitionManager.Get_Active_Slot_Suffix();
3388#else
3389 blk_device_partition = bare_partition_name;
3390#endif
3391
3392 FstabEntry fstabEntry = {
3393 .blk_device = blk_device_partition,
3394 .mount_point = twrpPart->Get_Mount_Point(),
3395 .fs_type = twrpPart->Current_File_System,
3396 .fs_mgr_flags.logical = twrpPart->Is_Super,
3397 };
3398
3399 fstab.emplace_back(fstabEntry);
3400 if (!fs_mgr_update_logical_partition(&fstabEntry)) {
Captain Throwbackc6939102021-04-06 12:57:30 -04003401 LOGINFO("unable to update logical partition: %s\n", twrpPart->Get_Mount_Point().c_str());
bigbiffee7b7ff2020-03-23 15:08:27 -04003402 return false;
3403 }
3404
bigbiff32cbabe2020-04-12 19:16:19 -04003405 while (access(fstabEntry.blk_device.c_str(), F_OK) != 0) {
3406 usleep(100);
3407 }
3408
bigbiffee7b7ff2020-03-23 15:08:27 -04003409 twrpPart->Set_Block_Device(fstabEntry.blk_device);
3410 twrpPart->Update_Size(true);
bigbiffee7b7ff2020-03-23 15:08:27 -04003411 twrpPart->Set_Can_Be_Backed_Up(false);
3412 twrpPart->Set_Can_Be_Wiped(false);
me-cafebabe928234b2022-07-28 04:07:04 +08003413 if (access(("/dev/block/bootdevice/by-name/" + bare_partition_name).c_str(), F_OK) == -1) {
3414 LOGINFO("Symlinking %s => /dev/block/bootdevice/by-name/%s \n", fstabEntry.blk_device.c_str(), bare_partition_name.c_str());
3415 symlink(fstabEntry.blk_device.c_str(), ("/dev/block/bootdevice/by-name/" + bare_partition_name).c_str());
3416 }
Mohd Faraz7fc70502021-04-22 20:44:11 +02003417
bigbiffee7b7ff2020-03-23 15:08:27 -04003418 return true;
3419}
3420
3421bool TWPartitionManager::Prepare_All_Super_Volumes() {
3422 bool status = true;
3423 std::vector<TWPartition*>::iterator iter;
3424
3425 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3426 if ((*iter)->Is_Super) {
3427 if (!Prepare_Super_Volume(*iter)) {
3428 status = false;
Mohd Faraz5ec04ef2023-01-02 22:47:51 +01003429 Partitions.erase(iter--);
bigbiffee7b7ff2020-03-23 15:08:27 -04003430 }
3431 PartitionManager.Output_Partition(*iter);
3432 }
3433 }
3434 Update_System_Details();
3435 return status;
3436}
3437
bigbiffee7b7ff2020-03-23 15:08:27 -04003438std::string TWPartitionManager::Get_Super_Partition() {
3439 int slot_number = Get_Active_Slot_Display() == "A" ? 0 : 1;
bigbiff998f8392021-08-06 19:19:33 -04003440 std::string super_device = fs_mgr_get_super_partition_name(slot_number);
bigbiffee7b7ff2020-03-23 15:08:27 -04003441 return "/dev/block/by-name/" + super_device;
3442}
3443
3444void TWPartitionManager::Setup_Super_Devices() {
3445 std::string superPart = Get_Super_Partition();
3446 android::fs_mgr::CreateLogicalPartitions(superPart);
bigbiff7ba75002020-04-11 20:47:09 -04003447}
3448
3449void TWPartitionManager::Setup_Super_Partition() {
bigbiffee7b7ff2020-03-23 15:08:27 -04003450 TWPartition* superPartition = new TWPartition();
bigbiff7ba75002020-04-11 20:47:09 -04003451 std::string superPart = Get_Super_Partition();
3452
3453 superPartition->Backup_Path = "/super";
3454 superPartition->Mount_Point = "/super";
bigbiffee7b7ff2020-03-23 15:08:27 -04003455 superPartition->Actual_Block_Device = superPart;
3456 superPartition->Alternate_Block_Device = superPart;
bigbiff998f8392021-08-06 19:19:33 -04003457 superPartition->Backup_Display_Name = "Super (";
3458 // Add first 4 items to fstab as logical that you would like to display in Backup_Display_Name
3459 // for the Super partition
3460 int list_size = Super_Partition_List.size();
3461 int orig_list_size = list_size;
3462 int max_display_size = 3; // total of 4 items since we start at 0
3463
3464 for (auto partition: Super_Partition_List) {
3465 superPartition->Backup_Display_Name = superPartition->Backup_Display_Name + partition;
3466 if ((orig_list_size - list_size) == max_display_size) {
3467 break;
3468 }
3469 if (list_size != 1)
3470 superPartition->Backup_Display_Name = superPartition->Backup_Display_Name + " ";
3471 list_size--;
3472 }
3473 superPartition->Backup_Display_Name += ")";
bigbiffee7b7ff2020-03-23 15:08:27 -04003474 superPartition->Can_Flash_Img = true;
bigbiffee7b7ff2020-03-23 15:08:27 -04003475 superPartition->Current_File_System = "emmc";
3476 superPartition->Can_Be_Backed_Up = true;
3477 superPartition->Is_Present = true;
3478 superPartition->Is_SubPartition = false;
bigbiff7ba75002020-04-11 20:47:09 -04003479 superPartition->Setup_Image();
bigbiffee7b7ff2020-03-23 15:08:27 -04003480 Add_Partition(superPartition);
3481 PartitionManager.Output_Partition(superPartition);
bigbiffee7b7ff2020-03-23 15:08:27 -04003482}
3483
3484bool TWPartitionManager::Get_Super_Status() {
bigbiffdf8436b2020-08-30 16:22:34 -04003485 std::string fastboot_mode = android::base::GetProperty("sys.usb.config", "");
3486 if (fastboot_mode == "fastboot") {
3487 return false;
3488 }
3489 else
3490 return access(Get_Super_Partition().c_str(), F_OK) == 0;
bigbiffee7b7ff2020-03-23 15:08:27 -04003491}
bigbiff25d25b92020-06-19 16:07:38 -04003492
3493bool TWPartitionManager::Recreate_Logs_Dir() {
3494#ifdef TW_INCLUDE_FBE
3495 struct passwd pd;
3496 struct passwd *pwdptr = &pd;
3497 struct passwd *tempPd;
3498 char pwdBuf[512];
3499 int uid = 0, gid = 0;
3500
3501 if ((getpwnam_r("system", pwdptr, pwdBuf, sizeof(pwdBuf), &tempPd)) != 0) {
3502 LOGERR("unable to get system user id\n");
3503 return false;
3504 } else {
3505 struct group grp;
3506 struct group *grpptr = &grp;
3507 struct group *tempGrp;
3508 char grpBuf[512];
3509
3510 if ((getgrnam_r("cache", grpptr, grpBuf, sizeof(grpBuf), &tempGrp)) != 0) {
3511 LOGERR("unable to get cache group id\n");
3512 return false;
3513 } else {
3514 uid = pd.pw_uid;
3515 gid = grp.gr_gid;
3516 std::string abLogsRecoveryDir(DATA_LOGS_DIR);
3517 abLogsRecoveryDir += "/recovery/";
3518
3519 if (!TWFunc::Create_Dir_Recursive(abLogsRecoveryDir, S_IRWXU | S_IRWXG | S_IWGRP | S_IXGRP, uid, gid)) {
3520 LOGERR("Unable to recreate %s\n", abLogsRecoveryDir.c_str());
3521 return false;
3522 }
3523 if (setfilecon(abLogsRecoveryDir.c_str(), "u:object_r:cache_file:s0") != 0) {
3524 LOGERR("Unable to set contexts for %s\n", abLogsRecoveryDir.c_str());
3525 return false;
3526 }
3527 }
3528 }
3529#endif
3530 return true;
3531}
bigbiff1f9e4842020-10-31 11:33:15 -04003532
3533void TWPartitionManager::Unlock_Block_Partitions() {
3534 int fd, OFF = 0;
3535
3536 const std::string block_path = "/dev/block/";
3537 DIR* d = opendir(block_path.c_str());
3538 if (d != NULL) {
3539 struct dirent* de;
3540 while ((de = readdir(d)) != NULL) {
3541 if (de->d_type == DT_BLK) {
3542 std::string block_device = block_path + de->d_name;
bigbiff1f9e4842020-10-31 11:33:15 -04003543 if ((fd = open(block_device.c_str(), O_RDONLY | O_CLOEXEC)) < 0) {
3544 LOGERR("unable to open block device %s: %s\n", block_device.c_str(), strerror(errno));
3545 continue;
3546 }
3547 if (ioctl(fd, BLKROSET, &OFF) == -1) {
me-cafebabed63fcaa2022-07-28 03:19:20 +08003548 LOGERR("Unable to unlock %s: %s\n", block_device.c_str());
bigbiff1f9e4842020-10-31 11:33:15 -04003549 continue;
3550 }
3551 close(fd);
3552 }
3553 }
3554 closedir(d);
3555 }
bigbiffaf253cd2020-12-06 16:50:55 -05003556}
bigbiffcfa875c2021-06-20 16:20:27 -04003557
3558bool TWPartitionManager::Unmap_Super_Devices() {
3559 bool destroyed = false;
3560#ifndef TW_EXCLUDE_APEX
3561 twrpApex apex;
3562 apex.Unmount();
3563#endif
3564 for (auto iter = Partitions.begin(); iter != Partitions.end();) {
3565 LOGINFO("Checking partition: %s\n", (*iter)->Get_Mount_Point().c_str());
3566 if ((*iter)->Is_Super) {
3567 TWPartition *part = *iter;
3568 std::string bare_partition_name = Get_Bare_Partition_Name((*iter)->Get_Mount_Point());
3569 std::string blk_device_partition = bare_partition_name + PartitionManager.Get_Active_Slot_Suffix();
3570 (*iter)->UnMount(false);
3571 LOGINFO("removing dynamic partition: %s\n", blk_device_partition.c_str());
3572 destroyed = DestroyLogicalPartition(blk_device_partition);
3573 std::string cow_partition = blk_device_partition + "-cow";
3574 std::string cow_partition_path = "/dev/block/mapper/" + cow_partition;
3575 struct stat st;
3576 if (lstat(cow_partition_path.c_str(), &st) == 0) {
3577 LOGINFO("removing cow partition: %s\n", cow_partition.c_str());
3578 destroyed = DestroyLogicalPartition(cow_partition);
3579 }
bigbiffcfa875c2021-06-20 16:20:27 -04003580 iter = Partitions.erase(iter);
3581 delete part;
3582 if (!destroyed) {
3583 return false;
3584 }
3585 } else {
3586 ++iter;
3587 }
3588 }
3589 return true;
3590}
bigbiffd21252f2021-09-18 15:56:32 -04003591
3592
3593bool TWPartitionManager::Check_Pending_Merges() {
3594 auto sm = android::snapshot::SnapshotManager::NewForFirstStageMount();
3595 if (!sm) {
3596 LOGERR("Unable to call snapshot manager\n");
3597 return false;
3598 }
3599
3600 if (!Unmap_Super_Devices()) {
3601 LOGERR("Unable to unmap dynamic partitions.\n");
3602 return false;
3603 }
3604
3605 auto callback = [&]() -> void {
3606 double progress;
3607 sm->GetUpdateState(&progress);
3608 LOGINFO("waiting for merge to complete: %.2f\n", progress);
3609 };
3610
3611 LOGINFO("checking for merges\n");
3612 if (!sm->HandleImminentDataWipe(callback)) {
3613 LOGERR("Unable to check merge status\n");
3614 return false;
3615 }
3616 return true;
Darth98f775972022-08-05 09:39:03 +01003617}