blob: b24beb3e4f91f58dcd8c1cb9e4a0123e2bb0fac4 [file] [log] [blame]
Dees Troy3be70a82013-10-22 14:25:12 +00001/*
bigbiffaed1bdf2021-08-03 18:21:39 -04002 Copyright 2014 to 2021 TeamWin
Dees Troy3be70a82013-10-22 14:25:12 +00003 This file is part of TWRP/TeamWin Recovery Project.
4
5 TWRP is free software: you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation, either version 3 of the License, or
8 (at your option) any later version.
9
10 TWRP is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
14
15 You should have received a copy of the GNU General Public License
16 along with TWRP. If not, see <http://www.gnu.org/licenses/>.
17*/
Dees_Troy51a0e822012-09-05 15:24:24 -040018
19#include <stdio.h>
20#include <stdlib.h>
21#include <string.h>
22#include <sys/stat.h>
23#include <sys/vfs.h>
24#include <unistd.h>
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060025#include <map>
Dees_Troy5bf43922012-09-07 16:07:55 -040026#include <vector>
Dees_Troy63c8df72012-09-10 14:02:05 -040027#include <dirent.h>
28#include <time.h>
Dees_Troy8170a922012-09-18 15:40:25 -040029#include <errno.h>
30#include <fcntl.h>
bigbiff25d25b92020-06-19 16:07:38 -040031#include <grp.h>
32#include <pwd.h>
bigbiffce8f83c2015-12-12 18:30:21 -050033#include <zlib.h>
bigbiff bigbiff9c754052013-01-09 09:09:08 -050034#include <iostream>
35#include <iomanip>
Ethan Yonker8613dc02014-09-11 09:28:20 -050036#include <sys/wait.h>
Ethan Yonker483e9f42016-01-11 22:21:18 -060037#include <linux/fs.h>
38#include <sys/mount.h>
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060039#include <sys/poll.h>
40#include <sys/socket.h>
bigbiffcfa875c2021-06-20 16:20:27 -040041#include <sys/stat.h>
42#include <sys/types.h>
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060043#include <linux/types.h>
44#include <linux/netlink.h>
bigbiffee7b7ff2020-03-23 15:08:27 -040045#include <android-base/chrono_utils.h>
46#include <android-base/file.h>
47#include <android-base/logging.h>
bigbiff7ba75002020-04-11 20:47:09 -040048#include <android-base/properties.h>
bigbiffee7b7ff2020-03-23 15:08:27 -040049#include <android-base/strings.h>
50#include <fstab/fstab.h>
51#include <fs_avb/fs_avb.h>
52#include <fs_mgr.h>
53#include <fs_mgr_dm_linear.h>
54#include <fs_mgr_overlayfs.h>
bigbiffcfa875c2021-06-20 16:20:27 -040055#include <fs_mgr/roots.h>
bigbiffee7b7ff2020-03-23 15:08:27 -040056#include <libgsi/libgsi.h>
57#include <liblp/liblp.h>
bigbiffcfa875c2021-06-20 16:20:27 -040058#include <libgsi/libgsi.h>
59#include <liblp/builder.h>
60#include <libsnapshot/snapshot.h>
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060061
Dees_Troy51a0e822012-09-05 15:24:24 -040062#include "variables.h"
Dees_Troy2673cec2013-04-02 20:22:16 +000063#include "twcommon.h"
Dees_Troy51a0e822012-09-05 15:24:24 -040064#include "partitions.hpp"
Dees_Troy5bf43922012-09-07 16:07:55 -040065#include "data.hpp"
bigbiffaed1bdf2021-08-03 18:21:39 -040066#include "startupArgs.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040067#include "twrp-functions.hpp"
Ethan Yonkerb5fab762016-01-28 14:03:33 -060068#include "fixContexts.hpp"
Ethan Yonker3fdcda42016-11-30 12:29:37 -060069#include "exclude.hpp"
Ethan Yonker4b94cfd2014-12-11 10:00:45 -060070#include "set_metadata.h"
bigbiff7abc5fe2015-01-17 16:53:12 -050071#include "tw_atomic.hpp"
Ethan Yonker74db1572015-10-28 12:44:49 -050072#include "gui/gui.hpp"
Ethan Yonker472f5062016-02-25 13:47:30 -060073#include "progresstracking.hpp"
bigbiff bigbiff56cf5642016-08-19 17:43:45 -040074#include "twrpDigestDriver.hpp"
bigbiffad58e1b2020-07-06 20:24:34 -040075#include "twrpRepacker.hpp"
bigbiffce8f83c2015-12-12 18:30:21 -050076#include "adbbu/libtwadbbu.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040077
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040078#ifdef TW_HAS_MTP
bigbiff bigbiffaf32bb92018-12-18 18:39:53 -050079#ifdef TW_HAS_LEGACY_MTP
80#include "mtp/legacy/mtp_MtpServer.hpp"
81#include "mtp/legacy/twrpMtp.hpp"
82#include "mtp/legacy/MtpMessage.hpp"
83#else
84#include "mtp/ffs/mtp_MtpServer.hpp"
85#include "mtp/ffs/twrpMtp.hpp"
86#include "mtp/ffs/MtpMessage.hpp"
87#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040088#endif
89
Dees Troy6f6441d2014-01-23 02:07:03 +000090extern "C" {
91 #include "cutils/properties.h"
Ethan Yonker74db1572015-10-28 12:44:49 -050092 #include "gui/gui.h"
Dees Troy6f6441d2014-01-23 02:07:03 +000093}
94
Dees_Troy5bf43922012-09-07 16:07:55 -040095#ifdef TW_INCLUDE_CRYPTO
bigbiff7ba75002020-04-11 20:47:09 -040096#include "crypto/fde/cryptfs.h"
97#include "gui/rapidxml.hpp"
98#include "gui/pages.hpp"
99#ifdef TW_INCLUDE_FBE
bigbiffa957f072021-03-07 18:20:29 -0500100#include "Decrypt.h"
bigbiff7ba75002020-04-11 20:47:09 -0400101#ifdef TW_INCLUDE_FBE_METADATA_DECRYPT
102 #ifdef USE_FSCRYPT
bigbiffa957f072021-03-07 18:20:29 -0500103 #include "MetadataCrypt.h"
bigbiffadc599e2020-05-28 19:36:30 +0000104 #endif
bigbiff7ba75002020-04-11 20:47:09 -0400105#endif
106#endif
107#ifdef TW_CRYPTO_USE_SYSTEM_VOLD
108#include "crypto/vold_decrypt/vold_decrypt.h"
109#endif
Dees_Troy5bf43922012-09-07 16:07:55 -0400110#endif
Dees_Troy51a0e822012-09-05 15:24:24 -0400111
Ethan Yonker1b190162016-12-05 15:25:19 -0600112#ifdef AB_OTA_UPDATER
dianlujitaob20dfaa2021-12-16 16:56:26 +0800113#include <android/hardware/boot/1.0/IBootControl.h>
114using android::hardware::boot::V1_0::CommandResult;
115using android::hardware::boot::V1_0::IBootControl;
Ethan Yonker1b190162016-12-05 15:25:19 -0600116#endif
117
bigbiffcfa875c2021-06-20 16:20:27 -0400118using android::fs_mgr::DestroyLogicalPartition;
bigbiffee7b7ff2020-03-23 15:08:27 -0400119using android::fs_mgr::Fstab;
120using android::fs_mgr::FstabEntry;
bigbiffcfa875c2021-06-20 16:20:27 -0400121using android::fs_mgr::MetadataBuilder;
Ethan Yonker6277c792014-09-15 14:54:30 -0500122
bigbiffee7b7ff2020-03-23 15:08:27 -0400123extern bool datamedia;
Noah Jacobson81d638d2019-04-28 00:10:07 -0400124std::vector<users_struct> Users_List;
125
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500126TWPartitionManager::TWPartitionManager(void) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400127 mtp_was_enabled = false;
Ethan Yonker726a0202014-12-16 20:01:38 -0600128 mtp_write_fd = -1;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600129 uevent_pfd.fd = -1;
bigbiff7abc5fe2015-01-17 16:53:12 -0500130 stop_backup.set_value(0);
Ethan Yonker1b190162016-12-05 15:25:19 -0600131#ifdef AB_OTA_UPDATER
132 char slot_suffix[PROPERTY_VALUE_MAX];
Ethan Yonkerfefe5912017-09-30 22:22:13 -0500133 property_get("ro.boot.slot_suffix", slot_suffix, "error");
134 if (strcmp(slot_suffix, "error") == 0)
135 property_get("ro.boot.slot", slot_suffix, "error");
Ethan Yonker1b190162016-12-05 15:25:19 -0600136 Active_Slot_Display = "";
Ethan Yonkerfefe5912017-09-30 22:22:13 -0500137 if (strcmp(slot_suffix, "_a") == 0 || strcmp(slot_suffix, "a") == 0)
Ethan Yonker1b190162016-12-05 15:25:19 -0600138 Set_Active_Slot("A");
139 else
140 Set_Active_Slot("B");
141#endif
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500142}
143
Captain Throwback37c3aef2021-10-06 11:38:18 -0400144void TWPartitionManager::Set_Crypto_State() {
145 char crypto_state[PROPERTY_VALUE_MAX];
146 property_get("ro.crypto.state", crypto_state, "error");
147 if (strcmp(crypto_state, "error") == 0)
148 property_set("ro.crypto.state", "encrypted");
149}
150
151int TWPartitionManager::Set_Crypto_Type(const char* crypto_type) {
152 char type_prop[PROPERTY_VALUE_MAX];
153 property_get("ro.crypto.type", type_prop, "error");
154 if (strcmp(type_prop, "error") == 0)
155 property_set("ro.crypto.type", crypto_type);
Captain Throwback18aea272020-10-29 12:43:17 -0400156 // Sleep for a bit so that services can start if needed
157 sleep(1);
158 return 0;
159}
160
bigbiff22851b92021-09-01 16:46:57 -0400161int TWPartitionManager::Process_Fstab(string Fstab_Filename, bool Display_Error, bool recovery_mode) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400162 FILE *fstabFile;
163 char fstab_line[MAX_FSTAB_LINE_LENGTH];
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600164 std::map<string, Flags_Map> twrp_flags;
165
166 fstabFile = fopen("/etc/twrp.flags", "rt");
167 if (fstabFile != NULL) {
Ian Macdonald160e8d32020-09-29 20:47:31 +0200168 LOGINFO("Reading /etc/twrp.flags\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600169 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600170 size_t line_size = strlen(fstab_line);
171 if (fstab_line[line_size - 1] != '\n')
172 fstab_line[line_size] = '\n';
173 Flags_Map line_flags;
174 line_flags.Primary_Block_Device = "";
175 line_flags.Alternate_Block_Device = "";
176 line_flags.fstab_line = (char*)malloc(MAX_FSTAB_LINE_LENGTH);
177 if (!line_flags.fstab_line) {
178 LOGERR("malloc error on line_flags.fstab_line\n");
179 return false;
180 }
181 memcpy(line_flags.fstab_line, fstab_line, MAX_FSTAB_LINE_LENGTH);
182 bool found_separator = false;
183 char *fs_loc = NULL;
184 char *block_loc = NULL;
185 char *flags_loc = NULL;
186 size_t index, item_index = 0;
187 for (index = 0; index < line_size; index++) {
188 if (fstab_line[index] <= 32) {
189 fstab_line[index] = '\0';
190 found_separator = true;
191 } else if (found_separator) {
192 if (item_index == 0) {
193 fs_loc = fstab_line + index;
194 } else if (item_index == 1) {
195 block_loc = fstab_line + index;
196 } else if (item_index > 1) {
197 char *ptr = fstab_line + index;
198 if (*ptr == '/') {
199 line_flags.Alternate_Block_Device = ptr;
200 } else if (strlen(ptr) > strlen("flags=") && strncmp(ptr, "flags=", strlen("flags=")) == 0) {
201 flags_loc = ptr;
202 // Once we find the flags=, we're done scanning the line
203 break;
204 }
205 }
206 found_separator = false;
207 item_index++;
208 }
209 }
210 if (block_loc)
211 line_flags.Primary_Block_Device = block_loc;
212 if (fs_loc)
213 line_flags.File_System = fs_loc;
214 if (flags_loc)
215 line_flags.Flags = flags_loc;
216 string Mount_Point = fstab_line;
217 twrp_flags[Mount_Point] = line_flags;
218 memset(fstab_line, 0, sizeof(fstab_line));
219 }
220 fclose(fstabFile);
221 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400222
223 fstabFile = fopen(Fstab_Filename.c_str(), "rt");
224 if (fstabFile == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000225 LOGERR("Critical Error: Unable to open fstab at '%s'.\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400226 return false;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600227 } else
228 LOGINFO("Reading %s\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400229
230 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600231 if (strstr(fstab_line, "swap"))
232 continue; // Skip swap in recovery
233
bigbiff998f8392021-08-06 19:19:33 -0400234 if (fstab_line[0] == '#')
235 continue;
236
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600237 size_t line_size = strlen(fstab_line);
238 if (fstab_line[line_size - 1] != '\n')
239 fstab_line[line_size] = '\n';
Dees_Troy2a923582012-09-20 12:13:34 -0400240
Matt Mower2b2dd152016-04-26 11:24:08 -0500241 TWPartition* partition = new TWPartition();
bigbiff8da46fa2020-09-08 16:42:34 -0400242 if (partition->Process_Fstab_Line(fstab_line, Display_Error, &twrp_flags))
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500243 Partitions.push_back(partition);
Matt Mower72c87ce2016-04-26 14:34:56 -0500244 else
Dees_Troy5bf43922012-09-07 16:07:55 -0400245 delete partition;
Matt Mower2b2dd152016-04-26 11:24:08 -0500246
247 memset(fstab_line, 0, sizeof(fstab_line));
Dees_Troy5bf43922012-09-07 16:07:55 -0400248 }
249 fclose(fstabFile);
Matt Mower72c87ce2016-04-26 14:34:56 -0500250
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600251 if (twrp_flags.size() > 0) {
252 LOGINFO("Processing remaining twrp.flags\n");
253 // Add any items from twrp.flags that did not exist in the recovery.fstab
254 for (std::map<string, Flags_Map>::iterator mapit=twrp_flags.begin(); mapit!=twrp_flags.end(); mapit++) {
255 if (Find_Partition_By_Path(mapit->first) == NULL) {
256 TWPartition* partition = new TWPartition();
bigbiff8da46fa2020-09-08 16:42:34 -0400257 if (partition->Process_Fstab_Line(mapit->second.fstab_line, Display_Error, NULL))
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600258 Partitions.push_back(partition);
259 else
260 delete partition;
261 }
262 if (mapit->second.fstab_line)
263 free(mapit->second.fstab_line);
264 mapit->second.fstab_line = NULL;
265 }
266 }
Mohd Faraze3948ec2020-08-14 01:40:59 +0000267 if (Get_Super_Status()) {
268 Setup_Super_Devices();
269 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600270 LOGINFO("Done processing fstab files\n");
271
bigbiff22851b92021-09-01 16:46:57 -0400272 if (recovery_mode) {
273 Setup_Fstab_Partitions(Display_Error);
Mohd Faraze3948ec2020-08-14 01:40:59 +0000274 }
bigbiff7ba75002020-04-11 20:47:09 -0400275 return true;
276}
277
bigbiff22851b92021-09-01 16:46:57 -0400278void TWPartitionManager::Setup_Fstab_Partitions(bool Display_Error) {
279 TWPartition* settings_partition = NULL;
280 TWPartition* andsec_partition = NULL;
281 std::vector<TWPartition*>::iterator iter;
282 unsigned int storageid = 1 << 16; // upper 16 bits are for physical storage device, we pretend to have only one
283
284 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
285 (*iter)->Partition_Post_Processing(Display_Error);
286
287 if ((*iter)->Is_Storage) {
288 ++storageid;
289 (*iter)->MTP_Storage_ID = storageid;
290 }
291
292 if (!settings_partition && (*iter)->Is_Settings_Storage && (*iter)->Is_Present)
293 settings_partition = (*iter);
294 else
295 (*iter)->Is_Settings_Storage = false;
296
297 if (!andsec_partition && (*iter)->Has_Android_Secure && (*iter)->Is_Present)
298 andsec_partition = (*iter);
299 else
300 (*iter)->Has_Android_Secure = false;
301
302 if ((*iter)->Is_Super)
303 Prepare_Super_Volume((*iter));
304 }
305
306 //Setup Apex before decryption
307 TWPartition* sys = PartitionManager.Find_Partition_By_Path(PartitionManager.Get_Android_Root_Path());
308 TWPartition* ven = PartitionManager.Find_Partition_By_Path("/vendor");
309 if (sys) {
310 if (sys->Get_Super_Status()) {
311 sys->Mount(true);
312 if (ven) {
313 ven->Mount(true);
314 }
315 #ifdef TW_EXCLUDE_APEX
316 LOGINFO("Apex is disabled in this build\n");
317 #else
318 twrpApex apex;
319 if (!apex.loadApexImages()) {
320 LOGERR("Unable to load apex images from %s\n", APEX_DIR);
321 property_set("twrp.apex.loaded", "false");
322 } else {
323 property_set("twrp.apex.loaded", "true");
324 }
325 TWFunc::check_and_run_script("/sbin/resyncapex.sh", "apex");
326 #endif
327 }
328 }
329 #ifndef USE_VENDOR_LIBS
330 if (ven)
331 ven->UnMount(true);
332 if (sys)
333 sys->UnMount(true);
334 #endif
335
336 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) {
337 // Attempt to automatically identify /data/media emulated storage devices
338 TWPartition* Dat = Find_Partition_By_Path("/data");
339 if (Dat) {
340 LOGINFO("Using automatic handling for /data/media emulated storage device.\n");
341 datamedia = true;
342 Dat->Setup_Data_Media();
343 settings_partition = Dat;
344 // Since /data was not considered a storage partition earlier, we still need to assign an MTP ID
345 ++storageid;
346 Dat->MTP_Storage_ID = storageid;
347 }
348 }
349 if (!settings_partition) {
350 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
351 if ((*iter)->Is_Storage) {
352 settings_partition = (*iter);
353 break;
354 }
355 }
356 if (!settings_partition)
357 LOGERR("Unable to locate storage partition for storing settings file.\n");
358 }
359 if (!Write_Fstab()) {
360 if (Display_Error)
361 LOGERR("Error creating fstab\n");
362 else
363 LOGINFO("Error creating fstab\n");
364 }
365
366 if (andsec_partition) {
367 Setup_Android_Secure_Location(andsec_partition);
368 } else if (settings_partition) {
369 Setup_Android_Secure_Location(settings_partition);
370 }
371 if (settings_partition) {
372 Setup_Settings_Storage_Partition(settings_partition);
373 }
374
375 #ifdef TW_INCLUDE_CRYPTO
376 DataManager::SetValue(TW_IS_ENCRYPTED, 1);
377 Decrypt_Data();
378 #endif
379
380 Update_System_Details();
381 if (Get_Super_Status())
382 Setup_Super_Partition();
383 UnMount_Main_Partitions();
384 #ifdef AB_OTA_UPDATER
385 DataManager::SetValue("tw_active_slot", Get_Active_Slot_Display());
386 #endif
387 setup_uevent();
388}
389
bigbiff7ba75002020-04-11 20:47:09 -0400390int TWPartitionManager::Write_Fstab(void) {
391 FILE *fp;
392 std::vector<TWPartition*>::iterator iter;
393 string Line;
394
395 fp = fopen("/etc/fstab", "w");
396 if (fp == NULL) {
397 LOGINFO("Can not open /etc/fstab.\n");
398 return false;
399 }
400 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
401 if ((*iter)->Can_Be_Mounted) {
Mohd Faraz2a1e3752020-09-30 10:55:48 +0530402 Line = (*iter)->Actual_Block_Device + " " + (*iter)->Mount_Point + " " + (*iter)->Current_File_System +
403 ((*iter)->Mount_Read_Only ? " ro " : " rw ") + "0 0\n";
bigbiff7ba75002020-04-11 20:47:09 -0400404 fputs(Line.c_str(), fp);
405 }
406 // Handle subpartition tracking
407 if ((*iter)->Is_SubPartition) {
408 TWPartition* ParentPartition = Find_Partition_By_Path((*iter)->SubPartition_Of);
409 if (ParentPartition)
410 ParentPartition->Has_SubPartition = true;
411 else
412 LOGERR("Unable to locate parent partition '%s' of '%s'\n", (*iter)->SubPartition_Of.c_str(), (*iter)->Mount_Point.c_str());
413 }
414 }
415 fclose(fp);
416 return true;
417}
418
419void TWPartitionManager::Decrypt_Data() {
420 #ifdef TW_INCLUDE_CRYPTO
Ethan Yonkercceebb82014-11-18 10:17:59 -0600421 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
422 if (Decrypt_Data && Decrypt_Data->Is_Encrypted && !Decrypt_Data->Is_Decrypted) {
Captain Throwback37c3aef2021-10-06 11:38:18 -0400423 Set_Crypto_State();
bigbiffcda15882021-12-30 17:01:44 -0500424 TWPartition* Key_Directory_Partition = Find_Partition_By_Path(Decrypt_Data->Key_Directory);
425 if (Key_Directory_Partition != nullptr)
426 if (!Key_Directory_Partition->Is_Mounted())
427 Mount_By_Path(Decrypt_Data->Key_Directory, false);
428 if (!Decrypt_Data->Key_Directory.empty()) {
429 Set_Crypto_Type("file");
Ethan Yonker93382822018-11-01 15:25:31 -0500430#ifdef TW_INCLUDE_FBE_METADATA_DECRYPT
bigbiff7ba75002020-04-11 20:47:09 -0400431#ifdef USE_FSCRYPT
432 if (fscrypt_mount_metadata_encrypted(Decrypt_Data->Actual_Block_Device, Decrypt_Data->Mount_Point, false)) {
bigbiffa957f072021-03-07 18:20:29 -0500433 std::string crypto_blkdev = android::base::GetProperty("ro.crypto.fs_crypto_blkdev", "error");
Mohd Faraze3948ec2020-08-14 01:40:59 +0000434 Decrypt_Data->Decrypted_Block_Device = crypto_blkdev;
bigbiff7ba75002020-04-11 20:47:09 -0400435 LOGINFO("Successfully decrypted metadata encrypted data partition with new block device: '%s'\n", crypto_blkdev.c_str());
bigbiff7ba75002020-04-11 20:47:09 -0400436#endif
Ethan Yonker93382822018-11-01 15:25:31 -0500437 Decrypt_Data->Is_Decrypted = true; // Needed to make the mount function work correctly
438 int retry_count = 10;
439 while (!Decrypt_Data->Mount(false) && --retry_count)
440 usleep(500);
441 if (Decrypt_Data->Mount(false)) {
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400442 if (!Decrypt_Data->Decrypt_FBE_DE()) {
bigbiff437d86f2021-09-26 16:57:14 -0400443 LOGERR("Unable to decrypt FBE device\n");
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400444 }
445
Ethan Yonker93382822018-11-01 15:25:31 -0500446 } else {
447 LOGINFO("Failed to mount data after metadata decrypt\n");
448 }
449 } else {
450 LOGINFO("Unable to decrypt metadata encryption\n");
451 }
452#else
453 LOGERR("Metadata FBE decrypt support not present in this TWRP\n");
454#endif
455 }
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600456 if (Decrypt_Data->Is_FBE) {
457 if (DataManager::GetIntValue(TW_CRYPTO_PWTYPE) == 0) {
458 if (Decrypt_Device("!") == 0) {
459 gui_msg("decrypt_success=Successfully decrypted with default password.");
460 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
461 } else {
462 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
463 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600464 }
465 } else {
bigbiffbbbfe172021-05-30 15:52:41 -0400466 LOGINFO("FBE setup failed. Trying FDE...");
Captain Throwback37c3aef2021-10-06 11:38:18 -0400467 Set_Crypto_State();
468 Set_Crypto_Type("block");
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600469 int password_type = cryptfs_get_password_type();
470 if (password_type == CRYPT_TYPE_DEFAULT) {
471 LOGINFO("Device is encrypted with the default password, attempting to decrypt.\n");
472 if (Decrypt_Device("default_password") == 0) {
473 gui_msg("decrypt_success=Successfully decrypted with default password.");
474 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
475 } else {
476 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
477 }
478 } else {
479 DataManager::SetValue("TW_CRYPTO_TYPE", password_type);
Noah Jacobson81d638d2019-04-28 00:10:07 -0400480 DataManager::SetValue("tw_crypto_pwtype_0", password_type);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600481 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600482 }
483 }
bigbiffcda15882021-12-30 17:01:44 -0500484 if (Decrypt_Data && (!Decrypt_Data->Is_Encrypted || Decrypt_Data->Is_Decrypted)) {
Ethan Yonker66a19492015-12-10 10:19:45 -0600485 Decrypt_Adopted();
486 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600487#endif
Dees_Troy51a0e822012-09-05 15:24:24 -0400488}
489
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500490void TWPartitionManager::Setup_Settings_Storage_Partition(TWPartition* Part) {
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500491 DataManager::SetValue("tw_settings_path", Part->Storage_Path);
492 DataManager::SetValue("tw_storage_path", Part->Storage_Path);
493 LOGINFO("Settings storage is '%s'\n", Part->Storage_Path.c_str());
494}
495
Matt Mowerbf4efa32014-04-14 23:25:26 -0500496void TWPartitionManager::Setup_Android_Secure_Location(TWPartition* Part) {
497 if (Part->Has_Android_Secure)
498 Part->Setup_AndSec();
Ethan Yonker6277c792014-09-15 14:54:30 -0500499 else if (!datamedia)
Matt Mowerbf4efa32014-04-14 23:25:26 -0500500 Part->Setup_AndSec();
Matt Mowerbf4efa32014-04-14 23:25:26 -0500501}
502
Dees_Troy8170a922012-09-18 15:40:25 -0400503void TWPartitionManager::Output_Partition_Logging(void) {
504 std::vector<TWPartition*>::iterator iter;
505
506 printf("\n\nPartition Logs:\n");
507 for (iter = Partitions.begin(); iter != Partitions.end(); iter++)
508 Output_Partition((*iter));
509}
510
511void TWPartitionManager::Output_Partition(TWPartition* Part) {
512 unsigned long long mb = 1048576;
513
Gary Peck004d48b2012-11-21 16:28:18 -0800514 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 -0400515 if (Part->Can_Be_Mounted) {
Gary Peck004d48b2012-11-21 16:28:18 -0800516 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 -0400517 }
Gary Peck004d48b2012-11-21 16:28:18 -0800518 printf("\n Flags: ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500519 if (Part->Can_Be_Mounted)
520 printf("Can_Be_Mounted ");
Gary Peck004d48b2012-11-21 16:28:18 -0800521 if (Part->Can_Be_Wiped)
522 printf("Can_Be_Wiped ");
Hashcodedabfd492013-08-29 22:45:30 -0700523 if (Part->Use_Rm_Rf)
524 printf("Use_Rm_Rf ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500525 if (Part->Can_Be_Backed_Up)
526 printf("Can_Be_Backed_Up ");
Gary Peck004d48b2012-11-21 16:28:18 -0800527 if (Part->Wipe_During_Factory_Reset)
528 printf("Wipe_During_Factory_Reset ");
529 if (Part->Wipe_Available_in_GUI)
530 printf("Wipe_Available_in_GUI ");
531 if (Part->Is_SubPartition)
532 printf("Is_SubPartition ");
533 if (Part->Has_SubPartition)
534 printf("Has_SubPartition ");
535 if (Part->Removable)
536 printf("Removable ");
537 if (Part->Is_Present)
538 printf("IsPresent ");
539 if (Part->Can_Be_Encrypted)
540 printf("Can_Be_Encrypted ");
541 if (Part->Is_Encrypted)
542 printf("Is_Encrypted ");
543 if (Part->Is_Decrypted)
544 printf("Is_Decrypted ");
545 if (Part->Has_Data_Media)
546 printf("Has_Data_Media ");
Dees_Troy83bd4832013-05-04 12:39:56 +0000547 if (Part->Can_Encrypt_Backup)
548 printf("Can_Encrypt_Backup ");
549 if (Part->Use_Userdata_Encryption)
550 printf("Use_Userdata_Encryption ");
Gary Peck004d48b2012-11-21 16:28:18 -0800551 if (Part->Has_Android_Secure)
552 printf("Has_Android_Secure ");
553 if (Part->Is_Storage)
554 printf("Is_Storage ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500555 if (Part->Is_Settings_Storage)
556 printf("Is_Settings_Storage ");
Dees_Troy68cab492012-12-12 19:29:35 +0000557 if (Part->Ignore_Blkid)
558 printf("Ignore_Blkid ");
Ethan Yonker253368a2014-11-25 15:00:52 -0600559 if (Part->Mount_To_Decrypt)
560 printf("Mount_To_Decrypt ");
Ethan Yonker96af84a2015-01-05 14:58:36 -0600561 if (Part->Can_Flash_Img)
562 printf("Can_Flash_Img ");
Ethan Yonker66a19492015-12-10 10:19:45 -0600563 if (Part->Is_Adopted_Storage)
564 printf("Is_Adopted_Storage ");
Ethan Yonker1b190162016-12-05 15:25:19 -0600565 if (Part->SlotSelect)
566 printf("SlotSelect ");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600567 if (Part->Mount_Read_Only)
568 printf("Mount_Read_Only ");
bigbiffcfa875c2021-06-20 16:20:27 -0400569 if (Part->Is_Super)
570 printf("Is_Super ");
Gary Peck004d48b2012-11-21 16:28:18 -0800571 printf("\n");
572 if (!Part->SubPartition_Of.empty())
573 printf(" SubPartition_Of: %s\n", Part->SubPartition_Of.c_str());
574 if (!Part->Symlink_Path.empty())
575 printf(" Symlink_Path: %s\n", Part->Symlink_Path.c_str());
576 if (!Part->Symlink_Mount_Point.empty())
577 printf(" Symlink_Mount_Point: %s\n", Part->Symlink_Mount_Point.c_str());
578 if (!Part->Primary_Block_Device.empty())
579 printf(" Primary_Block_Device: %s\n", Part->Primary_Block_Device.c_str());
580 if (!Part->Alternate_Block_Device.empty())
581 printf(" Alternate_Block_Device: %s\n", Part->Alternate_Block_Device.c_str());
582 if (!Part->Decrypted_Block_Device.empty())
583 printf(" Decrypted_Block_Device: %s\n", Part->Decrypted_Block_Device.c_str());
dianlujitao4879b372018-12-03 18:45:47 +0800584 if (!Part->Crypto_Key_Location.empty())
Ethan Yonker253368a2014-11-25 15:00:52 -0600585 printf(" Crypto_Key_Location: %s\n", Part->Crypto_Key_Location.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800586 if (Part->Length != 0)
587 printf(" Length: %i\n", Part->Length);
588 if (!Part->Display_Name.empty())
589 printf(" Display_Name: %s\n", Part->Display_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500590 if (!Part->Storage_Name.empty())
591 printf(" Storage_Name: %s\n", Part->Storage_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800592 if (!Part->Backup_Path.empty())
593 printf(" Backup_Path: %s\n", Part->Backup_Path.c_str());
594 if (!Part->Backup_Name.empty())
595 printf(" Backup_Name: %s\n", Part->Backup_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500596 if (!Part->Backup_Display_Name.empty())
597 printf(" Backup_Display_Name: %s\n", Part->Backup_Display_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800598 if (!Part->Backup_FileName.empty())
599 printf(" Backup_FileName: %s\n", Part->Backup_FileName.c_str());
600 if (!Part->Storage_Path.empty())
601 printf(" Storage_Path: %s\n", Part->Storage_Path.c_str());
602 if (!Part->Current_File_System.empty())
603 printf(" Current_File_System: %s\n", Part->Current_File_System.c_str());
604 if (!Part->Fstab_File_System.empty())
605 printf(" Fstab_File_System: %s\n", Part->Fstab_File_System.c_str());
606 if (Part->Format_Block_Size != 0)
Ethan Yonkerc798c9c2015-10-09 11:15:26 -0500607 printf(" Format_Block_Size: %lu\n", Part->Format_Block_Size);
Dees_Troy094207a2012-09-26 12:00:39 -0400608 if (!Part->MTD_Name.empty())
609 printf(" MTD_Name: %s\n", Part->MTD_Name.c_str());
Matt Mower029a82d2017-01-08 13:12:38 -0600610 printf(" Backup_Method: %s\n", Part->Backup_Method_By_Name().c_str());
Hashcode62bd9e02013-11-19 21:59:42 -0800611 if (Part->Mount_Flags || !Part->Mount_Options.empty())
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600612 printf(" Mount_Flags: %i, Mount_Options: %s\n", Part->Mount_Flags, Part->Mount_Options.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -0600613 if (Part->MTP_Storage_ID)
614 printf(" MTP_Storage_ID: %i\n", Part->MTP_Storage_ID);
Ethan Yonker93382822018-11-01 15:25:31 -0500615 if (!Part->Key_Directory.empty())
616 printf(" Metadata Key Directory: %s\n", Part->Key_Directory.c_str());
Ethan Yonker6277c792014-09-15 14:54:30 -0500617 printf("\n");
Dees_Troy8170a922012-09-18 15:40:25 -0400618}
619
Dees_Troy51a0e822012-09-05 15:24:24 -0400620int TWPartitionManager::Mount_By_Path(string Path, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400621 std::vector<TWPartition*>::iterator iter;
622 int ret = false;
623 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400624 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400625
bigbiffd58ba182020-03-23 10:02:29 -0400626 if (Local_Path == "/tmp" || Local_Path == "/")
Dees_Troy43d8b002012-09-17 16:00:01 -0400627 return true;
628
Dees_Troy5bf43922012-09-07 16:07:55 -0400629 // Iterate through all partitions
630 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400631 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400632 ret = (*iter)->Mount(Display_Error);
633 found = true;
Dees_Troy51127312012-09-08 13:08:49 -0400634 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400635 (*iter)->Mount(Display_Error);
Dees_Troy51127312012-09-08 13:08:49 -0400636 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400637 }
638 if (found) {
639 return ret;
640 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500641 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 -0400642 }
643 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400644}
645
Dees_Troy51a0e822012-09-05 15:24:24 -0400646int TWPartitionManager::UnMount_By_Path(string Path, bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400647 std::vector<TWPartition*>::iterator iter;
648 int ret = false;
649 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400650 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy51127312012-09-08 13:08:49 -0400651
652 // Iterate through all partitions
653 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400654 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy51127312012-09-08 13:08:49 -0400655 ret = (*iter)->UnMount(Display_Error);
656 found = true;
657 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
658 (*iter)->UnMount(Display_Error);
659 }
660 }
661 if (found) {
662 return ret;
663 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500664 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 -0400665 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000666 LOGINFO("UnMount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400667 }
668 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400669}
670
Dees_Troy51a0e822012-09-05 15:24:24 -0400671int TWPartitionManager::Is_Mounted_By_Path(string Path) {
Dees_Troy51127312012-09-08 13:08:49 -0400672 TWPartition* Part = Find_Partition_By_Path(Path);
673
674 if (Part)
675 return Part->Is_Mounted();
676 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000677 LOGINFO("Is_Mounted: Unable to find partition for path '%s'\n", Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400678 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400679}
680
Dees_Troy5bf43922012-09-07 16:07:55 -0400681int TWPartitionManager::Mount_Current_Storage(bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400682 string current_storage_path = DataManager::GetCurrentStoragePath();
683
684 if (Mount_By_Path(current_storage_path, Display_Error)) {
685 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
686 if (FreeStorage)
687 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
688 return true;
689 }
690 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400691}
692
Dees_Troy5bf43922012-09-07 16:07:55 -0400693int TWPartitionManager::Mount_Settings_Storage(bool Display_Error) {
694 return Mount_By_Path(DataManager::GetSettingsStoragePath(), Display_Error);
695}
696
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600697TWPartition* TWPartitionManager::Find_Partition_By_Path(const string& Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400698 std::vector<TWPartition*>::iterator iter;
Dees_Troy38bd7602012-09-14 13:33:53 -0400699 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400700
dianlujitaob76a73a2020-04-30 20:33:20 +0800701 if (Local_Path == "/system")
702 Local_Path = Get_Android_Root_Path();
Dees_Troy5bf43922012-09-07 16:07:55 -0400703 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400704 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path))
Dees_Troy5bf43922012-09-07 16:07:55 -0400705 return (*iter);
706 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400707 return NULL;
708}
709
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600710TWPartition* TWPartitionManager::Find_Partition_By_Block_Device(const string& Block_Device) {
711 std::vector<TWPartition*>::iterator iter;
712
713 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
714 if ((*iter)->Primary_Block_Device == Block_Device || (!(*iter)->Actual_Block_Device.empty() && (*iter)->Actual_Block_Device == Block_Device))
715 return (*iter);
716 }
717 return NULL;
718}
719
Ethan Yonker53796e72019-01-11 22:49:52 -0600720int TWPartitionManager::Check_Backup_Name(const std::string& Backup_Name, bool Display_Error, bool Must_Be_Unique) {
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400721 // Check the backup name to ensure that it is the correct size and contains only valid characters
722 // and that a backup with that name doesn't already exist
723 char backup_name[MAX_BACKUP_NAME_LEN];
724 char backup_loc[255], tw_image_dir[255];
725 int copy_size;
726 int index, cur_char;
Ethan Yonker53796e72019-01-11 22:49:52 -0600727 string Backup_Loc;
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400728
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400729 copy_size = Backup_Name.size();
730 // Check size
731 if (copy_size > MAX_BACKUP_NAME_LEN) {
732 if (Display_Error)
Ethan Yonker74db1572015-10-28 12:44:49 -0500733 gui_err("backup_name_len=Backup name is too long.");
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400734 return -2;
735 }
736
737 // Check each character
738 strncpy(backup_name, Backup_Name.c_str(), copy_size);
Dees_Troya13d74f2013-03-24 08:54:55 -0500739 if (copy_size == 1 && strncmp(backup_name, "0", 1) == 0)
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400740 return 0; // A "0" (zero) means to use the current timestamp for the backup name
741 for (index=0; index<copy_size; index++) {
742 cur_char = (int)backup_name[index];
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500743 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 -0400744 // These are valid characters
745 // Numbers
746 // Upper case letters
747 // Lower case letters
748 // Space
749 // and -_.{}[]
750 } else {
751 if (Display_Error)
Ethan Yonker4adc33e2016-01-22 16:07:11 -0600752 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 -0400753 return -3;
754 }
755 }
756
Ethan Yonker53796e72019-01-11 22:49:52 -0600757 if (Must_Be_Unique) {
758 // Check to make sure that a backup with this name doesn't already exist
759 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Loc);
760 strcpy(backup_loc, Backup_Loc.c_str());
761 sprintf(tw_image_dir,"%s/%s", backup_loc, Backup_Name.c_str());
762 if (TWFunc::Path_Exists(tw_image_dir)) {
763 if (Display_Error)
764 gui_err("backup_name_exists=A backup with that name already exists!");
bigbiffce8f83c2015-12-12 18:30:21 -0500765
Ethan Yonker53796e72019-01-11 22:49:52 -0600766 return -4;
767 }
768 // Backup is unique
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400769 }
Ethan Yonker53796e72019-01-11 22:49:52 -0600770 // No problems found
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400771 return 0;
772}
773
bigbiffce8f83c2015-12-12 18:30:21 -0500774bool TWPartitionManager::Backup_Partition(PartitionSettings *part_settings) {
Dees_Troy43d8b002012-09-17 16:00:01 -0400775 time_t start, stop;
Matt Mower23d8aae2017-01-06 14:30:33 -0600776 int use_compression;
bigbiffce8f83c2015-12-12 18:30:21 -0500777 string backup_log = part_settings->Backup_Folder + "/recovery.log";
Dees_Troy43d8b002012-09-17 16:00:01 -0400778
bigbiffce8f83c2015-12-12 18:30:21 -0500779 if (part_settings->Part == NULL)
Dees_Troy43d8b002012-09-17 16:00:01 -0400780 return true;
781
Dees_Troy093b7642012-09-21 15:59:38 -0400782 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400783
Tom Hite5a926722014-09-15 01:31:03 +0000784 TWFunc::SetPerformanceMode(true);
Dees_Troy43d8b002012-09-17 16:00:01 -0400785 time(&start);
786
bigbiffce8f83c2015-12-12 18:30:21 -0500787 if (part_settings->Part->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500788 sync();
789 sync();
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400790 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500791 if (!part_settings->adbbackup && part_settings->generate_digest) {
792 if (!twrpDigestDriver::Make_Digest(Full_Filename))
793 goto backup_error;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400794 }
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400795
bigbiffce8f83c2015-12-12 18:30:21 -0500796 if (part_settings->Part->Has_SubPartition) {
Dees_Troy8170a922012-09-18 15:40:25 -0400797 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400798 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -0400799
800 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400801 if ((*subpart)->Can_Be_Backed_Up && (*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500802 part_settings->Part = *subpart;
bigbiffce8f83c2015-12-12 18:30:21 -0500803 if (!(*subpart)->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500804 goto backup_error;
Tom Hite5a926722014-09-15 01:31:03 +0000805 }
Dees_Troy2727b992013-08-14 20:09:30 +0000806 sync();
807 sync();
bigbiff bigbifff5955b12019-05-18 17:28:58 -0400808 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400809 if (!part_settings->adbbackup && part_settings->generate_digest) {
810 if (!twrpDigestDriver::Make_Digest(Full_Filename)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500811 goto backup_error;
bigbiffce8f83c2015-12-12 18:30:21 -0500812 }
Tom Hite5a926722014-09-15 01:31:03 +0000813 }
Dees_Troy8170a922012-09-18 15:40:25 -0400814 }
815 }
816 }
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400817
Dees_Troy43d8b002012-09-17 16:00:01 -0400818 time(&stop);
that203bcc92015-05-09 17:27:33 +0200819 int backup_time = (int) difftime(stop, start);
Dees_Troy2673cec2013-04-02 20:22:16 +0000820 LOGINFO("Partition Backup time: %d\n", backup_time);
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400821
bigbiffce8f83c2015-12-12 18:30:21 -0500822 if (part_settings->Part->Backup_Method == BM_FILES) {
823 part_settings->file_time += backup_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400824 } else {
bigbiffce8f83c2015-12-12 18:30:21 -0500825 part_settings->img_time += backup_time;
826
Dees_Troy43d8b002012-09-17 16:00:01 -0400827 }
Tom Hite5a926722014-09-15 01:31:03 +0000828
Matt Mower02899552016-12-30 18:13:51 -0600829 TWFunc::SetPerformanceMode(false);
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500830 return true;
Dees_Troy43d8b002012-09-17 16:00:01 -0400831 }
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500832backup_error:
833 Clean_Backup_Folder(part_settings->Backup_Folder);
834 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
835 tw_set_default_metadata(backup_log.c_str());
836 TWFunc::SetPerformanceMode(false);
837 return false;
bigbiff7abc5fe2015-01-17 16:53:12 -0500838}
839
bigbiffbf1d6722015-02-14 16:25:59 -0500840void TWPartitionManager::Clean_Backup_Folder(string Backup_Folder) {
841 DIR *d = opendir(Backup_Folder.c_str());
842 struct dirent *p;
843 int r;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400844 vector<string> ext;
845
846 //extensions we should delete when cleaning
847 ext.push_back("win");
848 ext.push_back("md5");
849 ext.push_back("sha2");
850 ext.push_back("info");
bigbiffbf1d6722015-02-14 16:25:59 -0500851
Ethan Yonker74db1572015-10-28 12:44:49 -0500852 gui_msg("backup_clean=Backup Failed. Cleaning Backup Folder.");
bigbiffbf1d6722015-02-14 16:25:59 -0500853
854 if (d == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500855 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Backup_Folder)(strerror(errno)));
bigbiffbf1d6722015-02-14 16:25:59 -0500856 return;
857 }
858
Matt Mower2b18a532015-02-20 16:58:05 -0600859 while ((p = readdir(d))) {
bigbiffbf1d6722015-02-14 16:25:59 -0500860 if (!strcmp(p->d_name, ".") || !strcmp(p->d_name, ".."))
861 continue;
862
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500863 string path = Backup_Folder + "/" + p->d_name;
bigbiffbf1d6722015-02-14 16:25:59 -0500864
865 size_t dot = path.find_last_of(".") + 1;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400866 for (vector<string>::const_iterator i = ext.begin(); i != ext.end(); ++i) {
867 if (path.substr(dot) == *i) {
868 r = unlink(path.c_str());
869 if (r != 0)
870 LOGINFO("Unable to unlink '%s: %s'\n", path.c_str(), strerror(errno));
bigbiffbf1d6722015-02-14 16:25:59 -0500871 }
872 }
873 }
874 closedir(d);
875}
876
Ethan Yonker472f5062016-02-25 13:47:30 -0600877int TWPartitionManager::Check_Backup_Cancel() {
878 return stop_backup.get_value();
879}
880
bigbiff7abc5fe2015-01-17 16:53:12 -0500881int TWPartitionManager::Cancel_Backup() {
882 string Backup_Folder, Backup_Name, Full_Backup_Path;
883
884 stop_backup.set_value(1);
885
886 if (tar_fork_pid != 0) {
887 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
888 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Folder);
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500889 Full_Backup_Path = Backup_Folder + "/" + Backup_Name;
bigbiff7abc5fe2015-01-17 16:53:12 -0500890 LOGINFO("Killing pid: %d\n", tar_fork_pid);
891 kill(tar_fork_pid, SIGUSR2);
892 while (kill(tar_fork_pid, 0) == 0) {
893 usleep(1000);
894 }
895 LOGINFO("Backup_Run stopped and returning false, backup cancelled.\n");
896 LOGINFO("Removing directory %s\n", Full_Backup_Path.c_str());
897 TWFunc::removeDir(Full_Backup_Path, false);
898 tar_fork_pid = 0;
899 }
900
901 return 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400902}
903
bigbiffce8f83c2015-12-12 18:30:21 -0500904int TWPartitionManager::Run_Backup(bool adbbackup) {
905 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400906 int partition_count = 0, disable_free_space_check = 0, skip_digest = 0;
bigbiffce8f83c2015-12-12 18:30:21 -0500907 string Backup_Name, Backup_List, backup_path;
Matt Mower23d8aae2017-01-06 14:30:33 -0600908 unsigned long long total_bytes = 0, free_space = 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400909 TWPartition* storage = NULL;
910 struct tm *t;
Matt Mower23d8aae2017-01-06 14:30:33 -0600911 time_t seconds, total_start, total_stop;
Dees_Troya13d74f2013-03-24 08:54:55 -0500912 size_t start_pos = 0, end_pos = 0;
bigbiff7abc5fe2015-01-17 16:53:12 -0500913 stop_backup.set_value(0);
Dees_Troy43d8b002012-09-17 16:00:01 -0400914 seconds = time(0);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500915 t = localtime(&seconds);
Dees_Troy43d8b002012-09-17 16:00:01 -0400916
bigbiffce8f83c2015-12-12 18:30:21 -0500917 part_settings.img_bytes_remaining = 0;
918 part_settings.file_bytes_remaining = 0;
919 part_settings.img_time = 0;
920 part_settings.file_time = 0;
921 part_settings.img_bytes = 0;
922 part_settings.file_bytes = 0;
923 part_settings.PM_Method = PM_BACKUP;
924
bigbiffce8f83c2015-12-12 18:30:21 -0500925 part_settings.adbbackup = adbbackup;
Dees_Troy43d8b002012-09-17 16:00:01 -0400926 time(&total_start);
927
928 Update_System_Details();
929
930 if (!Mount_Current_Storage(true))
931 return false;
932
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400933 DataManager::GetValue(TW_SKIP_DIGEST_GENERATE_VAR, skip_digest);
934 if (skip_digest == 0)
935 part_settings.generate_digest = true;
Dees_Troyc5865ab2012-09-24 15:08:04 -0400936 else
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400937 part_settings.generate_digest = false;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400938
bigbiffce8f83c2015-12-12 18:30:21 -0500939 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, part_settings.Backup_Folder);
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500940 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
941 if (Backup_Name == gui_lookup("curr_date", "(Current Date)")) {
942 Backup_Name = TWFunc::Get_Current_Date();
943 } else if (Backup_Name == gui_lookup("auto_generate", "(Auto Generate)") || Backup_Name == "0" || Backup_Name.empty()) {
Dees Troyb21cc642013-09-10 17:36:41 +0000944 TWFunc::Auto_Generate_Backup_Name();
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500945 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
Dees_Troy43d8b002012-09-17 16:00:01 -0400946 }
bigbiffce8f83c2015-12-12 18:30:21 -0500947
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500948 LOGINFO("Backup Name is: '%s'\n", Backup_Name.c_str());
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500949 part_settings.Backup_Folder = part_settings.Backup_Folder + "/" + Backup_Name;
bigbiffce8f83c2015-12-12 18:30:21 -0500950
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500951 LOGINFO("Backup_Folder is: '%s'\n", part_settings.Backup_Folder.c_str());
Dees_Troy43d8b002012-09-17 16:00:01 -0400952
Dees_Troy2673cec2013-04-02 20:22:16 +0000953 LOGINFO("Calculating backup details...\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500954 DataManager::GetValue("tw_backup_list", Backup_List);
bigbiff7ba75002020-04-11 20:47:09 -0400955 LOGINFO("Backup_List: %s\n", Backup_List.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500956 if (!Backup_List.empty()) {
957 end_pos = Backup_List.find(";", start_pos);
958 while (end_pos != string::npos && start_pos < Backup_List.size()) {
959 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiff7ba75002020-04-11 20:47:09 -0400960 LOGINFO("backup_path: %s\n", backup_path.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -0500961 part_settings.Part = Find_Partition_By_Path(backup_path);
962 if (part_settings.Part != NULL) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500963 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500964 if (part_settings.Part->Backup_Method == BM_FILES)
965 part_settings.file_bytes += part_settings.Part->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500966 else
bigbiffce8f83c2015-12-12 18:30:21 -0500967 part_settings.img_bytes += part_settings.Part->Backup_Size;
968 if (part_settings.Part->Has_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500969 std::vector<TWPartition*>::iterator subpart;
970
971 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiffce8f83c2015-12-12 18:30:21 -0500972 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 -0500973 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500974 if ((*subpart)->Backup_Method == BM_FILES)
975 part_settings.file_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500976 else
bigbiffce8f83c2015-12-12 18:30:21 -0500977 part_settings.img_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500978 }
979 }
Dees_Troy8170a922012-09-18 15:40:25 -0400980 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500981 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500982 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 -0400983 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500984 start_pos = end_pos + 1;
985 end_pos = Backup_List.find(";", start_pos);
Dees_Troy43d8b002012-09-17 16:00:01 -0400986 }
987 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400988
989 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500990 gui_msg("no_partition_selected=No partitions selected for backup.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400991 return false;
992 }
bigbiffce8f83c2015-12-12 18:30:21 -0500993 if (adbbackup) {
994 if (twadbbu::Write_ADB_Stream_Header(partition_count) == false) {
995 return false;
996 }
997 }
998 total_bytes = part_settings.file_bytes + part_settings.img_bytes;
Ethan Yonker472f5062016-02-25 13:47:30 -0600999 ProgressTracking progress(total_bytes);
bigbiffce8f83c2015-12-12 18:30:21 -05001000 part_settings.progress = &progress;
1001
Ethan Yonker74db1572015-10-28 12:44:49 -05001002 gui_msg(Msg("total_partitions_backup= * Total number of partitions to back up: {1}")(partition_count));
1003 gui_msg(Msg("total_backup_size= * Total size of all data: {1}MB")(total_bytes / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -04001004 storage = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
1005 if (storage != NULL) {
1006 free_space = storage->Free;
Ethan Yonker74db1572015-10-28 12:44:49 -05001007 gui_msg(Msg("available_space= * Available space: {1}MB")(free_space / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -04001008 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001009 gui_err("unable_locate_storage=Unable to locate storage device.");
Dees_Troy43d8b002012-09-17 16:00:01 -04001010 return false;
1011 }
bigbiffbf1d6722015-02-14 16:25:59 -05001012
Matt Mowerbfccfb82016-04-25 23:22:31 -05001013 DataManager::GetValue(TW_DISABLE_FREE_SPACE_VAR, disable_free_space_check);
bigbiffce8f83c2015-12-12 18:30:21 -05001014
1015 if (adbbackup)
1016 disable_free_space_check = true;
1017
bigbiffbf1d6722015-02-14 16:25:59 -05001018 if (!disable_free_space_check) {
1019 if (free_space - (32 * 1024 * 1024) < total_bytes) {
1020 // We require an extra 32MB just in case
Ethan Yonker74db1572015-10-28 12:44:49 -05001021 gui_err("no_space=Not enough free space on storage.");
bigbiffbf1d6722015-02-14 16:25:59 -05001022 return false;
1023 }
Dees_Troy43d8b002012-09-17 16:00:01 -04001024 }
bigbiffce8f83c2015-12-12 18:30:21 -05001025 part_settings.img_bytes_remaining = part_settings.img_bytes;
1026 part_settings.file_bytes_remaining = part_settings.file_bytes;
Dees_Troy43d8b002012-09-17 16:00:01 -04001027
Ethan Yonker74db1572015-10-28 12:44:49 -05001028 gui_msg("backup_started=[BACKUP STARTED]");
bigbiff bigbiffcdd97c72019-03-29 19:12:33 -04001029
1030 int is_decrypted = 0;
1031 int is_encrypted = 0;
1032
1033 DataManager::GetValue(TW_IS_DECRYPTED, is_decrypted);
1034 DataManager::GetValue(TW_IS_ENCRYPTED, is_encrypted);
1035 if (!adbbackup || (!is_encrypted || (is_encrypted && is_decrypted))) {
1036 gui_msg(Msg("backup_folder= * Backup Folder: {1}")(part_settings.Backup_Folder));
1037 if (!TWFunc::Recursive_Mkdir(part_settings.Backup_Folder)) {
1038 gui_err("fail_backup_folder=Failed to make backup folder.");
1039 return false;
1040 }
Dees_Troyd4b22b02013-01-18 17:17:58 +00001041 }
1042
Dees_Troy2673cec2013-04-02 20:22:16 +00001043 DataManager::SetProgress(0.0);
Dees_Troy093b7642012-09-21 15:59:38 -04001044
Dees_Troya13d74f2013-03-24 08:54:55 -05001045 start_pos = 0;
1046 end_pos = Backup_List.find(";", start_pos);
1047 while (end_pos != string::npos && start_pos < Backup_List.size()) {
bigbiff7abc5fe2015-01-17 16:53:12 -05001048 if (stop_backup.get_value() != 0)
1049 return -1;
Dees_Troya13d74f2013-03-24 08:54:55 -05001050 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001051 part_settings.Part = Find_Partition_By_Path(backup_path);
1052 if (part_settings.Part != NULL) {
1053 if (!Backup_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001054 return false;
1055 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001056 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 -05001057 }
1058 start_pos = end_pos + 1;
1059 end_pos = Backup_List.find(";", start_pos);
1060 }
Dees_Troy43d8b002012-09-17 16:00:01 -04001061
1062 // Average BPS
bigbiffce8f83c2015-12-12 18:30:21 -05001063 if (part_settings.img_time == 0)
1064 part_settings.img_time = 1;
1065 if (part_settings.file_time == 0)
1066 part_settings.file_time = 1;
1067 int img_bps = (int)part_settings.img_bytes / (int)part_settings.img_time;
1068 unsigned long long file_bps = part_settings.file_bytes / (int)part_settings.file_time;
Dees_Troy43d8b002012-09-17 16:00:01 -04001069
bigbiffce8f83c2015-12-12 18:30:21 -05001070 if (part_settings.file_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -06001071 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 -05001072 if (part_settings.img_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -06001073 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 -04001074
1075 time(&total_stop);
1076 int total_time = (int) difftime(total_stop, total_start);
bigbiffce8f83c2015-12-12 18:30:21 -05001077
1078 uint64_t actual_backup_size;
Ethan Yonker3fdcda42016-11-30 12:29:37 -06001079 if (!adbbackup) {
1080 TWExclude twe;
1081 actual_backup_size = twe.Get_Folder_Size(part_settings.Backup_Folder);
1082 } else
bigbiffce8f83c2015-12-12 18:30:21 -05001083 actual_backup_size = part_settings.file_bytes + part_settings.img_bytes;
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001084 actual_backup_size /= (1024LLU * 1024LLU);
Dees_Troy43d8b002012-09-17 16:00:01 -04001085
bigbiffce8f83c2015-12-12 18:30:21 -05001086 int prev_img_bps = 0, use_compression = 0;
1087 unsigned long long prev_file_bps = 0;
Matt Mower173cdb92016-12-31 02:49:19 -06001088 DataManager::GetValue(TW_BACKUP_AVG_IMG_RATE, prev_img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -04001089 img_bps += (prev_img_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001090 img_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -04001091
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001092 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -04001093 if (use_compression)
1094 DataManager::GetValue(TW_BACKUP_AVG_FILE_COMP_RATE, prev_file_bps);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001095 else
Dees_Troy093b7642012-09-21 15:59:38 -04001096 DataManager::GetValue(TW_BACKUP_AVG_FILE_RATE, prev_file_bps);
1097 file_bps += (prev_file_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001098 file_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -04001099
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001100 DataManager::SetValue(TW_BACKUP_AVG_IMG_RATE, img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -04001101 if (use_compression)
1102 DataManager::SetValue(TW_BACKUP_AVG_FILE_COMP_RATE, file_bps);
1103 else
1104 DataManager::SetValue(TW_BACKUP_AVG_FILE_RATE, file_bps);
1105
Ethan Yonker74db1572015-10-28 12:44:49 -05001106 gui_msg(Msg("total_backed_size=[{1} MB TOTAL BACKED UP]")(actual_backup_size));
Dees_Troy43d8b002012-09-17 16:00:01 -04001107 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001108 UnMount_Main_Partitions();
Ethan Yonker56db1c42015-12-20 22:49:00 -06001109 gui_msg(Msg(msg::kHighlight, "backup_completed=[BACKUP COMPLETED IN {1} SECONDS]")(total_time)); // the end
Ethan Yonkere080c1f2016-09-19 13:50:25 -05001110 string backup_log = part_settings.Backup_Folder + "/recovery.log";
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001111 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
Ethan Yonker4b94cfd2014-12-11 10:00:45 -06001112 tw_set_default_metadata(backup_log.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -05001113
1114 if (part_settings.adbbackup) {
1115 if (twadbbu::Write_ADB_Stream_Trailer() == false) {
1116 return false;
1117 }
1118 }
1119 part_settings.adbbackup = false;
1120 DataManager::SetValue("tw_enable_adb_backup", 0);
1121
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001122 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001123}
1124
bigbiffce8f83c2015-12-12 18:30:21 -05001125bool TWPartitionManager::Restore_Partition(PartitionSettings *part_settings) {
Dees_Troy4a2a1262012-09-18 09:33:47 -04001126 time_t Start, Stop;
bigbiffce8f83c2015-12-12 18:30:21 -05001127
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001128 if (part_settings->adbbackup) {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001129 std::string partName = part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win";
1130 LOGINFO("setting backup name: %s\n", partName.c_str());
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001131 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
1132 }
1133
Tom Hite5a926722014-09-15 01:31:03 +00001134 TWFunc::SetPerformanceMode(true);
bigbiffce8f83c2015-12-12 18:30:21 -05001135
Dees_Troy4a2a1262012-09-18 09:33:47 -04001136 time(&Start);
Ethan Yonker472f5062016-02-25 13:47:30 -06001137
bigbiffce8f83c2015-12-12 18:30:21 -05001138 if (!part_settings->Part->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001139 TWFunc::SetPerformanceMode(false);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001140 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001141 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001142 if (part_settings->Part->Has_SubPartition && !part_settings->adbbackup) {
Dees_Troy8170a922012-09-18 15:40:25 -04001143 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001144 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -04001145
1146 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001147 part_settings->Part = *subpart;
1148 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
1149 part_settings->Part = (*subpart);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001150 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
bigbiffce8f83c2015-12-12 18:30:21 -05001151 if (!(*subpart)->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001152 TWFunc::SetPerformanceMode(false);
Dees_Troy8170a922012-09-18 15:40:25 -04001153 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001154 }
Dees_Troy8170a922012-09-18 15:40:25 -04001155 }
1156 }
1157 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001158 time(&Stop);
Tom Hite5a926722014-09-15 01:31:03 +00001159 TWFunc::SetPerformanceMode(false);
bigbiffce8f83c2015-12-12 18:30:21 -05001160 gui_msg(Msg("restore_part_done=[{1} done ({2} seconds)]")(part_settings->Part->Backup_Display_Name)((int)difftime(Stop, Start)));
1161
Dees_Troy4a2a1262012-09-18 09:33:47 -04001162 return true;
1163}
1164
Ethan Yonker472f5062016-02-25 13:47:30 -06001165int TWPartitionManager::Run_Restore(const string& Restore_Name) {
bigbiffce8f83c2015-12-12 18:30:21 -05001166 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001167 int check_digest;
bigbiffce8f83c2015-12-12 18:30:21 -05001168
Dees_Troy4a2a1262012-09-18 09:33:47 -04001169 time_t rStart, rStop;
1170 time(&rStart);
Dees_Troya13d74f2013-03-24 08:54:55 -05001171 string Restore_List, restore_path;
1172 size_t start_pos = 0, end_pos;
bigbiffce8f83c2015-12-12 18:30:21 -05001173
Ethan Yonkerdcf2b672016-09-13 14:41:53 -05001174 part_settings.Backup_Folder = Restore_Name;
bigbiffce8f83c2015-12-12 18:30:21 -05001175 part_settings.Part = NULL;
1176 part_settings.partition_count = 0;
1177 part_settings.total_restore_size = 0;
1178 part_settings.adbbackup = false;
1179 part_settings.PM_Method = PM_RESTORE;
Dees_Troy43d8b002012-09-17 16:00:01 -04001180
Ethan Yonker74db1572015-10-28 12:44:49 -05001181 gui_msg("restore_started=[RESTORE STARTED]");
1182 gui_msg(Msg("restore_folder=Restore folder: '{1}'")(Restore_Name));
Dees_Troy43d8b002012-09-17 16:00:01 -04001183
Dees_Troy4a2a1262012-09-18 09:33:47 -04001184 if (!Mount_Current_Storage(true))
1185 return false;
1186
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001187 DataManager::GetValue(TW_SKIP_DIGEST_CHECK_VAR, check_digest);
1188 if (check_digest > 0) {
1189 // Check Digest files first before restoring to ensure that all of them match before starting a restore
1190 TWFunc::GUI_Operation_Text(TW_VERIFY_DIGEST_TEXT, gui_parse_text("{@verifying_digest}"));
1191 gui_msg("verifying_digest=Verifying Digest");
Dees_Troya13d74f2013-03-24 08:54:55 -05001192 } else {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001193 gui_msg("skip_digest=Skipping Digest check based on user setting.");
Dees_Troya13d74f2013-03-24 08:54:55 -05001194 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001195 gui_msg("calc_restore=Calculating restore details...");
Dees_Troya13d74f2013-03-24 08:54:55 -05001196 DataManager::GetValue("tw_restore_selected", Restore_List);
bigbiffce8f83c2015-12-12 18:30:21 -05001197
Dees_Troya13d74f2013-03-24 08:54:55 -05001198 if (!Restore_List.empty()) {
1199 end_pos = Restore_List.find(";", start_pos);
1200 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1201 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001202 part_settings.Part = Find_Partition_By_Path(restore_path);
1203 if (part_settings.Part != NULL) {
bigbiffce8f83c2015-12-12 18:30:21 -05001204 if (part_settings.Part->Mount_Read_Only) {
1205 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 -05001206 return false;
1207 }
bigbiffce8f83c2015-12-12 18:30:21 -05001208
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001209 string Full_Filename = part_settings.Backup_Folder + "/" + part_settings.Part->Backup_FileName;
1210
bigbiff4a60bee2021-01-23 14:08:03 -05001211 if (tw_get_default_metadata(Get_Android_Root_Path().c_str()) != 0) {
1212 gui_msg(Msg(msg::kWarning, "restore_system_context=Unable to get default context for {1} -- Android may not boot.")(Get_Android_Root_Path()));
1213 }
1214
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001215 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001216 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001217 part_settings.partition_count++;
1218 part_settings.total_restore_size += part_settings.Part->Get_Restore_Size(&part_settings);
1219 if (part_settings.Part->Has_SubPartition) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001220 TWPartition *parentPart = part_settings.Part;
Dees_Troya13d74f2013-03-24 08:54:55 -05001221 std::vector<TWPartition*>::iterator subpart;
1222
1223 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001224 part_settings.Part = *subpart;
1225 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001226 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001227 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001228 part_settings.total_restore_size += (*subpart)->Get_Restore_Size(&part_settings);
Dees_Troya13d74f2013-03-24 08:54:55 -05001229 }
1230 }
1231 }
1232 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001233 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001234 }
1235 start_pos = end_pos + 1;
1236 end_pos = Restore_List.find(";", start_pos);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001237 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001238 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001239
bigbiffce8f83c2015-12-12 18:30:21 -05001240 if (part_settings.partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001241 gui_err("no_part_restore=No partitions selected for restore.");
Dees_Troy4a2a1262012-09-18 09:33:47 -04001242 return false;
1243 }
1244
bigbiffce8f83c2015-12-12 18:30:21 -05001245 gui_msg(Msg("restore_part_count=Restoring {1} partitions...")(part_settings.partition_count));
1246 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 +00001247 DataManager::SetProgress(0.0);
bigbiffce8f83c2015-12-12 18:30:21 -05001248 ProgressTracking progress(part_settings.total_restore_size);
1249 part_settings.progress = &progress;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001250
Dees_Troya13d74f2013-03-24 08:54:55 -05001251 start_pos = 0;
1252 if (!Restore_List.empty()) {
1253 end_pos = Restore_List.find(";", start_pos);
1254 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1255 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001256
1257 part_settings.Part = Find_Partition_By_Path(restore_path);
1258 if (part_settings.Part != NULL) {
1259 part_settings.partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -05001260 if (!Restore_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001261 return false;
1262 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001263 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001264 }
1265 start_pos = end_pos + 1;
1266 end_pos = Restore_List.find(";", start_pos);
1267 }
1268 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001269 TWFunc::GUI_Operation_Text(TW_UPDATE_SYSTEM_DETAILS_TEXT, gui_parse_text("{@updating_system_details}"));
bigbiff4a60bee2021-01-23 14:08:03 -05001270 tw_set_default_metadata(Get_Android_Root_Path().c_str());
Captain Throwback9d6feb52018-07-27 10:05:24 -04001271 UnMount_By_Path(Get_Android_Root_Path(), false);
Dees_Troy43d8b002012-09-17 16:00:01 -04001272 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001273 UnMount_Main_Partitions();
Dees_Troy4a2a1262012-09-18 09:33:47 -04001274 time(&rStop);
Matt Mower3c366972015-12-25 19:28:31 -06001275 gui_msg(Msg(msg::kHighlight, "restore_completed=[RESTORE COMPLETED IN {1} SECONDS]")((int)difftime(rStop,rStart)));
bigbiff8f5fad32021-11-03 20:39:02 -04001276 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
1277 if (Decrypt_Data && Decrypt_Data->Is_Encrypted)
1278 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 -05001279 DataManager::SetValue("tw_file_progress", "");
bigbiffce8f83c2015-12-12 18:30:21 -05001280
Dees_Troy63c8df72012-09-10 14:02:05 -04001281 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001282}
1283
1284void TWPartitionManager::Set_Restore_Files(string Restore_Name) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001285 // Start with the default values
Dees_Troya13d74f2013-03-24 08:54:55 -05001286 string Restore_List;
Dees_Troy83bd4832013-05-04 12:39:56 +00001287 bool get_date = true, check_encryption = true;
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001288 bool adbbackup = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001289
1290 DataManager::SetValue("tw_restore_encrypted", 0);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001291 if (twadbbu::Check_ADB_Backup_File(Restore_Name)) {
1292 vector<string> adb_files;
1293 adb_files = twadbbu::Get_ADB_Backup_Files(Restore_Name);
1294 for (unsigned int i = 0; i < adb_files.size(); ++i) {
1295 string adb_restore_file = adb_files.at(i);
1296 std::size_t pos = adb_restore_file.find_first_of(".");
1297 std::string path = "/" + adb_restore_file.substr(0, pos);
1298 Restore_List = path + ";";
1299 TWPartition* Part = Find_Partition_By_Path(path);
1300 Part->Backup_FileName = TWFunc::Get_Filename(adb_restore_file);
1301 adbbackup = true;
1302 }
1303 DataManager::SetValue("tw_enable_adb_backup", 1);
Dees_Troy63c8df72012-09-10 14:02:05 -04001304 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001305 else {
1306 DIR* d;
1307 d = opendir(Restore_Name.c_str());
1308 if (d == NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001309 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001310 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Restore_Name)(strerror(errno)));
1311 return;
Dees_Troy63c8df72012-09-10 14:02:05 -04001312 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001313
1314 struct dirent* de;
1315 while ((de = readdir(d)) != NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001316 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001317 // Strip off three components
1318 char str[256];
1319 char* label;
1320 char* fstype = NULL;
1321 char* extn = NULL;
1322 char* ptr;
Dees_Troy63c8df72012-09-10 14:02:05 -04001323
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001324 strcpy(str, de->d_name);
1325 if (strlen(str) <= 2)
1326 continue;
Dees_Troy83bd4832013-05-04 12:39:56 +00001327
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001328 if (get_date) {
1329 char file_path[255];
1330 struct stat st;
1331
1332 strcpy(file_path, Restore_Name.c_str());
1333 strcat(file_path, "/");
1334 strcat(file_path, str);
1335 stat(file_path, &st);
1336 string backup_date = ctime((const time_t*)(&st.st_mtime));
1337 DataManager::SetValue(TW_RESTORE_FILE_DATE, backup_date);
1338 get_date = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001339 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001340
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001341 label = str;
1342 ptr = label;
1343 while (*ptr && *ptr != '.') ptr++;
1344 if (*ptr == '.')
1345 {
1346 *ptr = 0x00;
1347 ptr++;
1348 fstype = ptr;
1349 }
1350 while (*ptr && *ptr != '.') ptr++;
1351 if (*ptr == '.')
1352 {
1353 *ptr = 0x00;
1354 ptr++;
1355 extn = ptr;
1356 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001357
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001358 if (fstype == NULL || extn == NULL || strcmp(fstype, "log") == 0) continue;
1359 int extnlength = strlen(extn);
1360 if (extnlength != 3 && extnlength != 6) continue;
1361 if (extnlength >= 3 && strncmp(extn, "win", 3) != 0) continue;
1362 //if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
Dees_Troy63c8df72012-09-10 14:02:05 -04001363
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001364 if (check_encryption) {
1365 string filename = Restore_Name + "/";
1366 filename += de->d_name;
1367 if (TWFunc::Get_File_Type(filename) == 2) {
1368 LOGINFO("'%s' is encrypted\n", filename.c_str());
1369 DataManager::SetValue("tw_restore_encrypted", 1);
1370 }
1371 }
1372 if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
1373
1374 TWPartition* Part = Find_Partition_By_Path(label);
1375 if (Part == NULL)
1376 {
1377 gui_msg(Msg(msg::kError, "unable_locate_part_backup_name=Unable to locate partition by backup name: '{1}'")(label));
1378 continue;
1379 }
1380
1381 Part->Backup_FileName = de->d_name;
1382 if (strlen(extn) > 3) {
1383 Part->Backup_FileName.resize(Part->Backup_FileName.size() - strlen(extn) + 3);
1384 }
1385
dianlujitaob76a73a2020-04-30 20:33:20 +08001386 if (!Part->Is_SubPartition) {
1387 if (Part->Backup_Path == Get_Android_Root_Path())
1388 Restore_List += "/system;";
1389 else
1390 Restore_List += Part->Backup_Path + ";";
1391 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001392 }
1393 closedir(d);
Dees_Troy63c8df72012-09-10 14:02:05 -04001394 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001395
1396 if (adbbackup) {
1397 Restore_List = "ADB_Backup;";
1398 adbbackup = false;
1399 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001400
Dees_Troya13d74f2013-03-24 08:54:55 -05001401 // Set the final value
1402 DataManager::SetValue("tw_restore_list", Restore_List);
1403 DataManager::SetValue("tw_restore_selected", Restore_List);
Dees_Troy51a0e822012-09-05 15:24:24 -04001404 return;
1405}
1406
1407int TWPartitionManager::Wipe_By_Path(string Path) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001408 std::vector<TWPartition*>::iterator iter;
Mohd Farazfdd49bb2020-04-21 18:56:44 +05301409 std::vector < TWPartition * >::iterator iter1;
Dees_Troy63c8df72012-09-10 14:02:05 -04001410 int ret = false;
1411 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -04001412 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy63c8df72012-09-10 14:02:05 -04001413
dianlujitaob76a73a2020-04-30 20:33:20 +08001414 if (Local_Path == "/system")
1415 Local_Path = Get_Android_Root_Path();
bigbiffaed1bdf2021-08-03 18:21:39 -04001416 if (Path == "/cache") {
1417 TWPartition* cache = Find_Partition_By_Path("/cache");
1418 if (cache == nullptr) {
1419 TWPartition* dat = Find_Partition_By_Path("/data");
1420 if (dat) {
1421 dat->Wipe_Data_Cache();
1422 found = true;
1423 }
1424 }
1425 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001426 // Iterate through all partitions
1427 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -04001428 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Mohd Farazfdd49bb2020-04-21 18:56:44 +05301429 // iterate through all partitions since some legacy devices uses other partitions as vendor causes issues while wiping
1430 (*iter)->Find_Actual_Block_Device();
1431 for (iter1 = Partitions.begin (); iter1 != Partitions.end (); iter1++)
1432 {
1433 (*iter1)->Find_Actual_Block_Device();
1434 if ((*iter)->Actual_Block_Device == (*iter1)->Actual_Block_Device && (*iter)->Mount_Point != (*iter1)->Mount_Point)
1435 (*iter1)->UnMount(false);
1436 }
Dees_Troye58d5262012-09-21 12:27:57 -04001437 if (Path == "/and-sec")
1438 ret = (*iter)->Wipe_AndSec();
1439 else
1440 ret = (*iter)->Wipe();
Dees_Troy63c8df72012-09-10 14:02:05 -04001441 found = true;
1442 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1443 (*iter)->Wipe();
1444 }
1445 }
1446 if (found) {
1447 return ret;
1448 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001449 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 -04001450 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -04001451}
1452
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001453int TWPartitionManager::Wipe_By_Path(string Path, string New_File_System) {
1454 std::vector<TWPartition*>::iterator iter;
1455 int ret = false;
1456 bool found = false;
1457 string Local_Path = TWFunc::Get_Root_Path(Path);
1458
1459 // Iterate through all partitions
1460 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1461 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1462 if (Path == "/and-sec")
1463 ret = (*iter)->Wipe_AndSec();
1464 else
1465 ret = (*iter)->Wipe(New_File_System);
1466 found = true;
1467 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1468 (*iter)->Wipe(New_File_System);
1469 }
1470 }
1471 if (found) {
1472 return ret;
1473 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001474 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 -05001475 return false;
1476}
1477
Dees_Troy51a0e822012-09-05 15:24:24 -04001478int TWPartitionManager::Factory_Reset(void) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001479 std::vector<TWPartition*>::iterator iter;
1480 int ret = true;
1481
1482 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001483 if ((*iter)->Wipe_During_Factory_Reset && (*iter)->Is_Present) {
Ethan Yonker89583ef2015-08-26 09:01:59 -05001484#ifdef TW_OEM_BUILD
1485 if ((*iter)->Mount_Point == "/data") {
1486 if (!(*iter)->Wipe_Encryption())
1487 ret = false;
1488 } else {
1489#endif
1490 if (!(*iter)->Wipe())
1491 ret = false;
1492#ifdef TW_OEM_BUILD
1493 }
1494#endif
Dees_Troy094207a2012-09-26 12:00:39 -04001495 } else if ((*iter)->Has_Android_Secure) {
1496 if (!(*iter)->Wipe_AndSec())
1497 ret = false;
Dees_Troy63c8df72012-09-10 14:02:05 -04001498 }
1499 }
bigbiffad58e1b2020-07-06 20:24:34 -04001500 TWFunc::check_and_run_script("/system/bin/factoryreset.sh", "Factory Reset Script");
Dees_Troy63c8df72012-09-10 14:02:05 -04001501 return ret;
Dees_Troy51a0e822012-09-05 15:24:24 -04001502}
1503
Dees_Troy38bd7602012-09-14 13:33:53 -04001504int TWPartitionManager::Wipe_Dalvik_Cache(void) {
1505 struct stat st;
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001506 vector <string> dir;
Dees_Troy38bd7602012-09-14 13:33:53 -04001507
1508 if (!Mount_By_Path("/data", true))
1509 return false;
1510
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001511 dir.push_back("/data/dalvik-cache");
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001512
bigbiff25d25b92020-06-19 16:07:38 -04001513 std::string cacheDir = TWFunc::get_log_dir();
1514 if (cacheDir == CACHE_LOGS_DIR) {
1515 if (!PartitionManager.Mount_By_Path(CACHE_LOGS_DIR, false)) {
1516 LOGINFO("Unable to mount %s for wiping cache.\n", CACHE_LOGS_DIR);
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001517 }
1518 dir.push_back(cacheDir + "dalvik-cache");
1519 dir.push_back(cacheDir + "/dc");
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001520 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001521
Dees_Troy38bd7602012-09-14 13:33:53 -04001522 TWPartition* sdext = Find_Partition_By_Path("/sd-ext");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001523 if (sdext && sdext->Is_Present && sdext->Mount(false))
1524 {
1525 if (stat("/sd-ext/dalvik-cache", &st) == 0)
1526 {
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001527 dir.push_back("/sd-ext/dalvik-cache");
1528 }
1529 }
1530
bigbiff25d25b92020-06-19 16:07:38 -04001531 if (cacheDir == CACHE_LOGS_DIR) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001532 gui_msg("wiping_cache_dalvik=Wiping Dalvik Cache Directories...");
1533 } else {
1534 gui_msg("wiping_dalvik=Wiping Dalvik Directory...");
1535 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001536 for (unsigned i = 0; i < dir.size(); ++i) {
1537 if (stat(dir.at(i).c_str(), &st) == 0) {
1538 TWFunc::removeDir(dir.at(i), false);
1539 gui_msg(Msg("cleaned=Cleaned: {1}...")(dir.at(i)));
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001540 }
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001541 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001542
bigbiff25d25b92020-06-19 16:07:38 -04001543 if (cacheDir == CACHE_LOGS_DIR) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001544 gui_msg("cache_dalvik_done=-- Dalvik Cache Directories Wipe Complete!");
1545 } else {
1546 gui_msg("dalvik_done=-- Dalvik Directory Wipe Complete!");
1547 }
1548
Dees_Troy38bd7602012-09-14 13:33:53 -04001549 return true;
1550}
1551
1552int TWPartitionManager::Wipe_Rotate_Data(void) {
1553 if (!Mount_By_Path("/data", true))
1554 return false;
1555
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001556 unlink("/data/misc/akmd*");
1557 unlink("/data/misc/rild*");
Dees_Troy2673cec2013-04-02 20:22:16 +00001558 gui_print("Rotation data wiped.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001559 return true;
1560}
1561
1562int TWPartitionManager::Wipe_Battery_Stats(void) {
1563 struct stat st;
1564
1565 if (!Mount_By_Path("/data", true))
1566 return false;
1567
1568 if (0 != stat("/data/system/batterystats.bin", &st)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001569 gui_print("No Battery Stats Found. No Need To Wipe.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001570 } else {
1571 remove("/data/system/batterystats.bin");
Dees_Troy2673cec2013-04-02 20:22:16 +00001572 gui_print("Cleared battery stats.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001573 }
1574 return true;
1575}
1576
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001577int TWPartitionManager::Wipe_Android_Secure(void) {
1578 std::vector<TWPartition*>::iterator iter;
1579 int ret = false;
1580 bool found = false;
1581
1582 // Iterate through all partitions
1583 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1584 if ((*iter)->Has_Android_Secure) {
1585 ret = (*iter)->Wipe_AndSec();
1586 found = true;
1587 }
1588 }
1589 if (found) {
1590 return ret;
1591 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001592 gui_err("no_andsec=No android secure partitions found.");
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001593 }
1594 return false;
1595}
1596
Dees_Troy38bd7602012-09-14 13:33:53 -04001597int TWPartitionManager::Format_Data(void) {
1598 TWPartition* dat = Find_Partition_By_Path("/data");
bigbiffcfa875c2021-06-20 16:20:27 -04001599 TWPartition* metadata = Find_Partition_By_Path("/metadata");
1600 if (metadata != NULL)
1601 metadata->UnMount(false);
Dees_Troy38bd7602012-09-14 13:33:53 -04001602
1603 if (dat != NULL) {
bigbiffab76bd72021-10-11 10:17:48 -04001604 if (android::base::GetBoolProperty("ro.virtual_ab.enabled", false)) {
bigbiffcfa875c2021-06-20 16:20:27 -04001605#ifndef TW_EXCLUDE_APEX
1606 twrpApex apex;
1607 apex.Unmount();
1608#endif
1609 if (metadata != NULL)
1610 metadata->Mount(true);
bigbiffd21252f2021-09-18 15:56:32 -04001611 if (!Check_Pending_Merges())
bigbiffcfa875c2021-06-20 16:20:27 -04001612 return false;
1613 }
Dees_Troy38bd7602012-09-14 13:33:53 -04001614 return dat->Wipe_Encryption();
1615 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001616 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001617 return false;
1618 }
1619 return false;
1620}
1621
1622int TWPartitionManager::Wipe_Media_From_Data(void) {
1623 TWPartition* dat = Find_Partition_By_Path("/data");
1624
1625 if (dat != NULL) {
1626 if (!dat->Has_Data_Media) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001627 LOGERR("This device does not have /data/media\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001628 return false;
1629 }
1630 if (!dat->Mount(true))
1631 return false;
1632
Ethan Yonker74db1572015-10-28 12:44:49 -05001633 gui_msg("wiping_datamedia=Wiping internal storage -- /data/media...");
Ethan Yonker726a0202014-12-16 20:01:38 -06001634 Remove_MTP_Storage(dat->MTP_Storage_ID);
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001635 TWFunc::removeDir("/data/media", false);
xiaolu9416f4f2015-06-04 08:22:23 +08001636 dat->Recreate_Media_Folder();
Ethan Yonker726a0202014-12-16 20:01:38 -06001637 Add_MTP_Storage(dat->MTP_Storage_ID);
Dees_Troy38bd7602012-09-14 13:33:53 -04001638 return true;
1639 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001640 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001641 return false;
1642 }
1643 return false;
1644}
1645
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001646int TWPartitionManager::Repair_By_Path(string Path, bool Display_Error) {
1647 std::vector<TWPartition*>::iterator iter;
1648 int ret = false;
1649 bool found = false;
1650 string Local_Path = TWFunc::Get_Root_Path(Path);
1651
1652 if (Local_Path == "/tmp" || Local_Path == "/")
1653 return true;
1654
1655 // Iterate through all partitions
1656 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1657 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1658 ret = (*iter)->Repair();
1659 found = true;
1660 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1661 (*iter)->Repair();
1662 }
1663 }
1664 if (found) {
1665 return ret;
1666 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001667 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 -05001668 } else {
1669 LOGINFO("Repair: Unable to find partition for path '%s'\n", Local_Path.c_str());
1670 }
1671 return false;
1672}
1673
Ethan Yonkera2719152015-05-28 09:44:41 -05001674int TWPartitionManager::Resize_By_Path(string Path, bool Display_Error) {
1675 std::vector<TWPartition*>::iterator iter;
1676 int ret = false;
1677 bool found = false;
1678 string Local_Path = TWFunc::Get_Root_Path(Path);
1679
1680 if (Local_Path == "/tmp" || Local_Path == "/")
1681 return true;
1682
1683 // Iterate through all partitions
1684 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1685 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1686 ret = (*iter)->Resize();
1687 found = true;
1688 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1689 (*iter)->Resize();
1690 }
1691 }
1692 if (found) {
1693 return ret;
1694 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001695 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 -05001696 } else {
1697 LOGINFO("Resize: Unable to find partition for path '%s'\n", Local_Path.c_str());
1698 }
1699 return false;
1700}
1701
Dees_Troy51a0e822012-09-05 15:24:24 -04001702void TWPartitionManager::Update_System_Details(void) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001703 std::vector<TWPartition*>::iterator iter;
Dees_Troy51127312012-09-08 13:08:49 -04001704 int data_size = 0;
Dees_Troy5bf43922012-09-07 16:07:55 -04001705
Ethan Yonker74db1572015-10-28 12:44:49 -05001706 gui_msg("update_part_details=Updating partition details...");
Dees_Troy5bf43922012-09-07 16:07:55 -04001707 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker1b190162016-12-05 15:25:19 -06001708 (*iter)->Update_Size(true);
Dees_Troy51127312012-09-08 13:08:49 -04001709 if ((*iter)->Can_Be_Mounted) {
Captain Throwback9d6feb52018-07-27 10:05:24 -04001710 if ((*iter)->Mount_Point == Get_Android_Root_Path()) {
Dees_Troy51127312012-09-08 13:08:49 -04001711 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1712 DataManager::SetValue(TW_BACKUP_SYSTEM_SIZE, backup_display_size);
Ethan Yonker76bbd3a2019-05-10 10:50:04 -05001713 TWFunc::Is_TWRP_App_In_System();
Dees_Troy51127312012-09-08 13:08:49 -04001714 } else if ((*iter)->Mount_Point == "/data" || (*iter)->Mount_Point == "/datadata") {
1715 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
1716 } else if ((*iter)->Mount_Point == "/cache") {
1717 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1718 DataManager::SetValue(TW_BACKUP_CACHE_SIZE, backup_display_size);
1719 } else if ((*iter)->Mount_Point == "/sd-ext") {
1720 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1721 DataManager::SetValue(TW_BACKUP_SDEXT_SIZE, backup_display_size);
1722 if ((*iter)->Backup_Size == 0) {
1723 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 0);
1724 DataManager::SetValue(TW_BACKUP_SDEXT_VAR, 0);
1725 } else
1726 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 1);
Dees_Troye58d5262012-09-21 12:27:57 -04001727 } else if ((*iter)->Has_Android_Secure) {
Dees_Troy8170a922012-09-18 15:40:25 -04001728 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troye58d5262012-09-21 12:27:57 -04001729 DataManager::SetValue(TW_BACKUP_ANDSEC_SIZE, backup_display_size);
Dees_Troy8170a922012-09-18 15:40:25 -04001730 if ((*iter)->Backup_Size == 0) {
1731 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 0);
1732 DataManager::SetValue(TW_BACKUP_ANDSEC_VAR, 0);
1733 } else
1734 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 1);
Dees_Troy2c50e182012-09-26 20:05:28 -04001735 } else if ((*iter)->Mount_Point == "/boot") {
1736 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1737 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1738 if ((*iter)->Backup_Size == 0) {
1739 DataManager::SetValue("tw_has_boot_partition", 0);
1740 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1741 } else
1742 DataManager::SetValue("tw_has_boot_partition", 1);
Dees_Troy51127312012-09-08 13:08:49 -04001743 }
Dees_Troyaa9cc402012-10-13 12:14:05 -04001744 } else {
1745 // Handle unmountable partitions in case we reset defaults
1746 if ((*iter)->Mount_Point == "/boot") {
1747 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1748 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1749 if ((*iter)->Backup_Size == 0) {
1750 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 0);
1751 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1752 } else
1753 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 1);
1754 } else if ((*iter)->Mount_Point == "/recovery") {
1755 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1756 DataManager::SetValue(TW_BACKUP_RECOVERY_SIZE, backup_display_size);
1757 if ((*iter)->Backup_Size == 0) {
1758 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 0);
1759 DataManager::SetValue(TW_BACKUP_RECOVERY_VAR, 0);
1760 } else
1761 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 1);
Gary Peck82599a82012-11-21 16:23:12 -08001762 } else if ((*iter)->Mount_Point == "/data") {
1763 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troyaa9cc402012-10-13 12:14:05 -04001764 }
Dees_Troy51127312012-09-08 13:08:49 -04001765 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001766 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001767 gui_msg("update_part_details_done=...done");
Dees_Troy51127312012-09-08 13:08:49 -04001768 DataManager::SetValue(TW_BACKUP_DATA_SIZE, data_size);
1769 string current_storage_path = DataManager::GetCurrentStoragePath();
1770 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
Dees_Troy8170a922012-09-18 15:40:25 -04001771 if (FreeStorage != NULL) {
1772 // Attempt to mount storage
bigbiff25d25b92020-06-19 16:07:38 -04001773 if (!FreeStorage->Mount(false)) {
1774 gui_msg(Msg(msg::kWarning, "unable_to_mount_storage=Unable to mount storage"));
1775 DataManager::SetValue(TW_STORAGE_FREE_SIZE, 0);
1776 } else {
1777 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
Dees_Troy8170a922012-09-18 15:40:25 -04001778 }
1779 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001780 LOGINFO("Unable to find storage partition '%s'.\n", current_storage_path.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -04001781 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001782 if (!Write_Fstab())
Dees_Troy2673cec2013-04-02 20:22:16 +00001783 LOGERR("Error creating fstab\n");
Dees_Troy51a0e822012-09-05 15:24:24 -04001784 return;
1785}
1786
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001787void TWPartitionManager::Post_Decrypt(const string& Block_Device) {
1788 TWPartition* dat = Find_Partition_By_Path("/data");
bigbiffc2cb3852020-09-11 14:03:31 -04001789
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001790 if (dat != NULL) {
bigbiffaed1bdf2021-08-03 18:21:39 -04001791 // reparse for /cache/recovery/command
1792 static constexpr const char* COMMAND_FILE = "/data/cache/command";
1793 if (TWFunc::Path_Exists(COMMAND_FILE)) {
1794 startupArgs startup;
1795 std::string content;
1796 TWFunc::read_file(COMMAND_FILE, content);
1797 std::vector<std::string> args = {content};
1798 startup.processRecoveryArgs(args, 0);
1799 }
1800
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001801 DataManager::SetValue(TW_IS_DECRYPTED, 1);
1802 dat->Is_Decrypted = true;
1803 if (!Block_Device.empty()) {
1804 dat->Decrypted_Block_Device = Block_Device;
1805 gui_msg(Msg("decrypt_success_dev=Data successfully decrypted, new block device: '{1}'")(Block_Device));
1806 } else {
1807 gui_msg("decrypt_success_nodev=Data successfully decrypted");
1808 }
Noah Jacobson81d638d2019-04-28 00:10:07 -04001809 property_set("twrp.decrypt.done", "true");
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001810 dat->Setup_File_System(false);
Noah Jacobson81d638d2019-04-28 00:10:07 -04001811 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 -06001812
Noah Jacobson81d638d2019-04-28 00:10:07 -04001813 sleep(1); // Sleep for a bit so that the device will be ready
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001814 if (dat->Has_Data_Media && dat->Mount(false) && TWFunc::Path_Exists("/data/media/0")) {
1815 dat->Storage_Path = "/data/media/0";
1816 dat->Symlink_Path = dat->Storage_Path;
1817 DataManager::SetValue("tw_storage_path", "/data/media/0");
1818 DataManager::SetValue("tw_settings_path", "/data/media/0");
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001819 }
epicXa721f952021-01-04 13:01:31 +05301820 DataManager::LoadTWRPFolderInfo();
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001821 Update_System_Details();
nkk71ffb02bd2017-06-04 23:12:16 +03001822 Output_Partition(dat);
bigbiffcda15882021-12-30 17:01:44 -05001823 if (!android::base::StartsWith(dat->Actual_Block_Device, "/dev/block/mmcblk")) {
1824 if (!dat->Bind_Mount(false))
1825 LOGERR("Unable to bind mount /sdcard to %s\n", dat->Storage_Path.c_str());
1826 }
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001827 } else
1828 LOGERR("Unable to locate data partition.\n");
1829}
1830
Noah Jacobson81d638d2019-04-28 00:10:07 -04001831void TWPartitionManager::Parse_Users() {
1832#ifdef TW_INCLUDE_FBE
1833 char user_check_result[PROPERTY_VALUE_MAX];
1834 for (int userId = 0; userId <= 9999; userId++) {
1835 string prop = "twrp.user." + to_string(userId) + ".decrypt";
1836 property_get(prop.c_str(), user_check_result, "-1");
1837 if (strcmp(user_check_result, "-1") != 0) {
1838 if (userId < 0 || userId > 9999) {
1839 LOGINFO("Incorrect user id %d\n", userId);
1840 continue;
1841 }
1842 struct users_struct user;
1843 user.userId = to_string(userId);
1844
1845 // Attempt to get name of user. Fallback to user ID if this fails.
zhenyolka90850472021-11-08 19:05:30 +03001846 std::string path = "/data/system/users/" + to_string(userId) + ".xml";
1847 if ((atoi(TWFunc::System_Property_Get("ro.build.version.sdk").c_str()) > 30) && TWFunc::Path_Exists(path)) {
1848 if(!TWFunc::Check_Xml_Format(path))
1849 user.userName = to_string(userId);
bigbiffa957f072021-03-07 18:20:29 -05001850 }
Noah Jacobson81d638d2019-04-28 00:10:07 -04001851 else {
zhenyolka90850472021-11-08 19:05:30 +03001852 char* userFile = PageManager::LoadFileToBuffer(path, NULL);
1853 if (userFile == NULL) {
Noah Jacobson81d638d2019-04-28 00:10:07 -04001854 user.userName = to_string(userId);
zhenyolka90850472021-11-08 19:05:30 +03001855 }
1856 else {
1857 xml_document<> *userXml = new xml_document<>();
1858 userXml->parse<0>(userFile);
1859 xml_node<>* userNode = userXml->first_node("user");
1860 if (userNode == nullptr) {
Noah Jacobson81d638d2019-04-28 00:10:07 -04001861 user.userName = to_string(userId);
zhenyolka90850472021-11-08 19:05:30 +03001862 } else {
1863 xml_node<>* nameNode = userNode->first_node("name");
1864 if (nameNode == nullptr)
1865 user.userName = to_string(userId);
1866 else {
1867 string userName = nameNode->value();
1868 user.userName = userName + " (" + to_string(userId) + ")";
1869 }
Noah Jacobson81d638d2019-04-28 00:10:07 -04001870 }
1871 }
1872 }
1873
1874 string filename;
1875 user.type = Get_Password_Type(userId, filename);
1876
1877 user.isDecrypted = false;
1878 if (strcmp(user_check_result, "1") == 0)
1879 user.isDecrypted = true;
1880 Users_List.push_back(user);
1881 }
1882 }
1883 Check_Users_Decryption_Status();
1884#endif
1885}
1886
1887std::vector<users_struct>* TWPartitionManager::Get_Users_List() {
1888 return &Users_List;
1889}
1890
1891void TWPartitionManager::Mark_User_Decrypted(int userID) {
1892#ifdef TW_INCLUDE_FBE
1893 std::vector<users_struct>::iterator iter;
1894 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1895 if (atoi((*iter).userId.c_str()) == userID) {
1896 (*iter).isDecrypted = true;
1897 string user_prop_decrypted = "twrp.user." + to_string(userID) + ".decrypt";
1898 property_set(user_prop_decrypted.c_str(), "1");
1899 break;
1900 }
1901 }
1902 Check_Users_Decryption_Status();
1903#endif
1904}
1905
1906void TWPartitionManager::Check_Users_Decryption_Status() {
1907#ifdef TW_INCLUDE_FBE
1908 int all_is_decrypted = 1;
1909 std::vector<users_struct>::iterator iter;
1910 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1911 if (!(*iter).isDecrypted) {
1912 LOGINFO("User %s is not decrypted.\n", (*iter).userId.c_str());
1913 all_is_decrypted = 0;
1914 break;
1915 }
1916 }
1917 if (all_is_decrypted == 1) {
1918 LOGINFO("All found users are decrypted.\n");
1919 DataManager::SetValue("tw_all_users_decrypted", "1");
1920 property_set("twrp.all.users.decrypted", "true");
1921 } else
1922 DataManager::SetValue("tw_all_users_decrypted", "0");
1923#endif
1924}
1925
1926int TWPartitionManager::Decrypt_Device(string Password, int user_id) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001927#ifdef TW_INCLUDE_CRYPTO
Captain Throwback37c3aef2021-10-06 11:38:18 -04001928 char crypto_blkdev[PROPERTY_VALUE_MAX];
Ethan Yonker253368a2014-11-25 15:00:52 -06001929 std::vector<TWPartition*>::iterator iter;
Dees_Troy5bf43922012-09-07 16:07:55 -04001930
Ethan Yonker253368a2014-11-25 15:00:52 -06001931 // Mount any partitions that need to be mounted for decrypt
1932 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1933 if ((*iter)->Mount_To_Decrypt) {
1934 (*iter)->Mount(true);
1935 }
a39552696ff55ce2013-01-08 16:14:56 +00001936 }
oshmouna82a7542018-04-10 17:45:55 +02001937 property_set("twrp.mount_to_decrypt", "1");
a39552696ff55ce2013-01-08 16:14:56 +00001938
Captain Throwback37c3aef2021-10-06 11:38:18 -04001939 Set_Crypto_State();
1940 Set_Crypto_Type("block");
Ethan Yonkera1de1492015-11-04 11:58:27 -06001941
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001942 if (DataManager::GetIntValue(TW_IS_FBE)) {
1943#ifdef TW_INCLUDE_FBE
1944 if (!Mount_By_Path("/data", true)) // /data has to be mounted for FBE
1945 return -1;
Noah Jacobson81d638d2019-04-28 00:10:07 -04001946
1947 bool user_need_decrypt = false;
1948 std::vector<users_struct>::iterator iter;
1949 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1950 if (atoi((*iter).userId.c_str()) == user_id && !(*iter).isDecrypted) {
1951 user_need_decrypt = true;
1952 }
1953 }
1954 if (!user_need_decrypt) {
1955 LOGINFO("User %d does not require decryption\n", user_id);
1956 return 0;
1957 }
1958
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001959 int retry_count = 10;
1960 while (!TWFunc::Path_Exists("/data/system/users/gatekeeper.password.key") && --retry_count)
Noah Jacobson81d638d2019-04-28 00:10:07 -04001961 usleep(2000); // A small sleep is needed after mounting /data to ensure reliable decrypt...maybe because of DE?
1962 gui_msg(Msg("decrypting_user_fbe=Attempting to decrypt FBE for user {1}...")(user_id));
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001963 if (Decrypt_User(user_id, Password)) {
Noah Jacobson81d638d2019-04-28 00:10:07 -04001964 gui_msg(Msg("decrypt_user_success_fbe=User {1} Decrypted Successfully")(user_id));
1965 Mark_User_Decrypted(user_id);
1966 if (user_id == 0) {
1967 // When decrypting user 0 also try all other users
1968 std::vector<users_struct>::iterator iter;
1969 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1970 if ((*iter).userId == "0" || (*iter).isDecrypted)
1971 continue;
1972
1973 int tmp_user_id = atoi((*iter).userId.c_str());
1974 gui_msg(Msg("decrypting_user_fbe=Attempting to decrypt FBE for user {1}...")(tmp_user_id));
1975 if (Decrypt_User(tmp_user_id, Password) ||
1976 (Password != "!" && Decrypt_User(tmp_user_id, "!"))) { // "!" means default password
1977 gui_msg(Msg("decrypt_user_success_fbe=User {1} Decrypted Successfully")(tmp_user_id));
1978 Mark_User_Decrypted(tmp_user_id);
1979 } else {
1980 gui_msg(Msg("decrypt_user_fail_fbe=Failed to decrypt user {1}")(tmp_user_id));
1981 }
1982 }
1983 Post_Decrypt("");
1984 }
1985
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001986 return 0;
Noah Jacobson81d638d2019-04-28 00:10:07 -04001987 } else {
1988 gui_msg(Msg(msg::kError, "decrypt_user_fail_fbe=Failed to decrypt user {1}")(user_id));
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001989 }
1990#else
1991 LOGERR("FBE support is not present\n");
1992#endif
1993 return -1;
1994 }
1995
Noah Jacobson81d638d2019-04-28 00:10:07 -04001996 char isdecrypteddata[PROPERTY_VALUE_MAX];
1997 property_get("twrp.decrypt.done", isdecrypteddata, "");
1998 if (strcmp(isdecrypteddata, "true") == 0) {
1999 LOGINFO("Data has no decryption required\n");
2000 return 0;
2001 }
2002
Ethan Yonkerddb63e22017-01-19 14:01:57 -06002003 int pwret = -1;
2004 pid_t pid = fork();
2005 if (pid < 0) {
2006 LOGERR("fork failed\n");
2007 return -1;
2008 } else if (pid == 0) {
2009 // Child process
2010 char cPassword[255];
2011 strcpy(cPassword, Password.c_str());
2012 int ret = cryptfs_check_passwd(cPassword);
2013 exit(ret);
2014 } else {
2015 // Parent
2016 int status;
2017 if (TWFunc::Wait_For_Child_Timeout(pid, &status, "Decrypt", 30))
2018 pwret = -1;
2019 else
2020 pwret = WEXITSTATUS(status) ? -1 : 0;
2021 }
a39552696ff55ce2013-01-08 16:14:56 +00002022
nkk7171c6c502017-01-05 23:55:05 +02002023#ifdef TW_CRYPTO_USE_SYSTEM_VOLD
2024 if (pwret != 0) {
2025 pwret = vold_decrypt(Password);
nkk717d1222a2017-11-06 17:27:21 +02002026 switch (pwret) {
2027 case VD_SUCCESS:
2028 break;
2029 case VD_ERR_MISSING_VDC:
2030 gui_msg(Msg(msg::kError, "decrypt_data_vold_os_missing=Missing files needed for vold decrypt: {1}")("/system/bin/vdc"));
2031 break;
2032 case VD_ERR_MISSING_VOLD:
2033 gui_msg(Msg(msg::kError, "decrypt_data_vold_os_missing=Missing files needed for vold decrypt: {1}")("/system/bin/vold"));
2034 break;
2035 }
nkk7171c6c502017-01-05 23:55:05 +02002036 }
2037#endif // TW_CRYPTO_USE_SYSTEM_VOLD
2038
Ethan Yonker253368a2014-11-25 15:00:52 -06002039 // Unmount any partitions that were needed for decrypt
2040 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2041 if ((*iter)->Mount_To_Decrypt) {
2042 (*iter)->UnMount(false);
2043 }
2044 }
oshmouna82a7542018-04-10 17:45:55 +02002045 property_set("twrp.mount_to_decrypt", "0");
Ethan Yonker253368a2014-11-25 15:00:52 -06002046
a39552696ff55ce2013-01-08 16:14:56 +00002047 if (pwret != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002048 gui_err("fail_decrypt=Failed to decrypt data.");
Dees_Troy5bf43922012-09-07 16:07:55 -04002049 return -1;
2050 }
a39552696ff55ce2013-01-08 16:14:56 +00002051
Dees_Troy5bf43922012-09-07 16:07:55 -04002052 property_get("ro.crypto.fs_crypto_blkdev", crypto_blkdev, "error");
2053 if (strcmp(crypto_blkdev, "error") == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00002054 LOGERR("Error retrieving decrypted data block device.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04002055 } else {
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06002056 Post_Decrypt(crypto_blkdev);
Dees_Troy5bf43922012-09-07 16:07:55 -04002057 }
2058 return 0;
2059#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002060 gui_err("no_crypto_support=No crypto support was compiled into this build.");
Dees_Troy5bf43922012-09-07 16:07:55 -04002061 return -1;
2062#endif
Dees_Troy51a0e822012-09-05 15:24:24 -04002063 return 1;
Dees_Troy51127312012-09-08 13:08:49 -04002064}
2065
Ethan Yonkerb5fab762016-01-28 14:03:33 -06002066int TWPartitionManager::Fix_Contexts(void) {
Ethan Yonkerb5fab762016-01-28 14:03:33 -06002067 std::vector<TWPartition*>::iterator iter;
2068 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2069 if ((*iter)->Has_Data_Media) {
2070 if ((*iter)->Mount(true)) {
2071 if (fixContexts::fixDataMediaContexts((*iter)->Mount_Point) != 0)
2072 return -1;
2073 }
2074 }
2075 }
Dees_Troy1a650e62012-10-19 20:54:32 -04002076 UnMount_Main_Partitions();
Ethan Yonker74db1572015-10-28 12:44:49 -05002077 gui_msg("done=Done.");
Ethan Yonkerb5fab762016-01-28 14:03:33 -06002078 return 0;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04002079}
2080
Ethan Yonker66a19492015-12-10 10:19:45 -06002081TWPartition* TWPartitionManager::Find_Next_Storage(string Path, bool Exclude_Data_Media) {
Ethan Yonker47360be2014-04-01 10:34:34 -05002082 std::vector<TWPartition*>::iterator iter = Partitions.begin();
2083
2084 if (!Path.empty()) {
2085 string Search_Path = TWFunc::Get_Root_Path(Path);
2086 for (; iter != Partitions.end(); iter++) {
Matt Mower9a561dd2016-02-22 21:25:51 -06002087 if ((*iter)->Mount_Point == Search_Path) {
Ethan Yonker47360be2014-04-01 10:34:34 -05002088 iter++;
2089 break;
2090 }
2091 }
2092 }
2093
2094 for (; iter != Partitions.end(); iter++) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002095 if (Exclude_Data_Media && (*iter)->Has_Data_Media) {
2096 // do nothing, do not return this type of partition
2097 } else if ((*iter)->Is_Storage && (*iter)->Is_Present) {
Ethan Yonker47360be2014-04-01 10:34:34 -05002098 return (*iter);
2099 }
2100 }
2101
2102 return NULL;
2103}
2104
Dees_Troyd21618c2012-10-14 18:48:49 -04002105int TWPartitionManager::Open_Lun_File(string Partition_Path, string Lun_File) {
Dees_Troyd21618c2012-10-14 18:48:49 -04002106 TWPartition* Part = Find_Partition_By_Path(Partition_Path);
2107
2108 if (Part == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002109 LOGINFO("Unable to locate '%s' for USB storage mode.", Partition_Path.c_str());
2110 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 -04002111 return false;
2112 }
Ethan Yonker47360be2014-04-01 10:34:34 -05002113 LOGINFO("USB mount '%s', '%s' > '%s'\n", Partition_Path.c_str(), Part->Actual_Block_Device.c_str(), Lun_File.c_str());
2114 if (!Part->UnMount(true) || !Part->Is_Present)
Dees_Troyd21618c2012-10-14 18:48:49 -04002115 return false;
2116
bigbiff22851b92021-09-01 16:46:57 -04002117 if (!TWFunc::write_to_file(Lun_File, Part->Actual_Block_Device)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00002118 LOGERR("Unable to write to ums lunfile '%s': (%s)\n", Lun_File.c_str(), strerror(errno));
Dees_Troyd21618c2012-10-14 18:48:49 -04002119 return false;
2120 }
Dees_Troyd21618c2012-10-14 18:48:49 -04002121 return true;
2122}
2123
Dees_Troy8170a922012-09-18 15:40:25 -04002124int TWPartitionManager::usb_storage_enable(void) {
Dees_Troy8170a922012-09-18 15:40:25 -04002125 char lun_file[255];
Dees_Troyd21618c2012-10-14 18:48:49 -04002126 bool has_multiple_lun = false;
Dees_Troy8170a922012-09-18 15:40:25 -04002127
Ethan Yonker47360be2014-04-01 10:34:34 -05002128 string Lun_File_str = CUSTOM_LUN_FILE;
2129 size_t found = Lun_File_str.find("%");
2130 if (found != string::npos) {
2131 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
2132 if (TWFunc::Path_Exists(lun_file))
2133 has_multiple_lun = true;
2134 }
Ethan Yonker726a0202014-12-16 20:01:38 -06002135 mtp_was_enabled = TWFunc::Toggle_MTP(false); // Must disable MTP for USB Storage
Ethan Yonker47360be2014-04-01 10:34:34 -05002136 if (!has_multiple_lun) {
2137 LOGINFO("Device doesn't have multiple lun files, mount current storage\n");
2138 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
2139 if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) == "/data") {
Ethan Yonker66a19492015-12-10 10:19:45 -06002140 TWPartition* Mount = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05002141 if (Mount) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002142 if (!Open_Lun_File(Mount->Mount_Point, lun_file)) {
2143 goto error_handle;
2144 }
Ethan Yonker47360be2014-04-01 10:34:34 -05002145 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002146 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002147 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05002148 }
2149 } else if (!Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file)) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002150 goto error_handle;
Dees_Troy8170a922012-09-18 15:40:25 -04002151 }
Dees_Troy8170a922012-09-18 15:40:25 -04002152 } else {
Ethan Yonker47360be2014-04-01 10:34:34 -05002153 LOGINFO("Device has multiple lun files\n");
2154 TWPartition* Mount1;
2155 TWPartition* Mount2;
Dees_Troy8170a922012-09-18 15:40:25 -04002156 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
Ethan Yonker66a19492015-12-10 10:19:45 -06002157 Mount1 = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05002158 if (Mount1) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002159 if (!Open_Lun_File(Mount1->Mount_Point, lun_file)) {
2160 goto error_handle;
2161 }
Matt Moweree717062014-04-26 01:46:46 -05002162 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
Ethan Yonker66a19492015-12-10 10:19:45 -06002163 Mount2 = Find_Next_Storage(Mount1->Mount_Point, true);
Matt Mower1fdcdb72016-01-25 20:53:47 -06002164 if (Mount2 && Mount2->Mount_Point != Mount1->Mount_Point) {
Matt Moweree717062014-04-26 01:46:46 -05002165 Open_Lun_File(Mount2->Mount_Point, lun_file);
nkk713c1e35f2017-03-24 18:28:03 +02002166 // Mimic single lun code: Mount CurrentStoragePath if it's not /data
2167 } else if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) != "/data") {
2168 Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file);
Ethan Yonker47360be2014-04-01 10:34:34 -05002169 }
nkk713c1e35f2017-03-24 18:28:03 +02002170 // Mimic single lun code: Mount CurrentStoragePath if it's not /data
2171 } else if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) != "/data" && !Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file)) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002172 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002173 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05002174 }
Dees_Troy8170a922012-09-18 15:40:25 -04002175 }
Matt Mowerb6ef4782014-11-06 02:24:36 -06002176 property_set("sys.storage.ums_enabled", "1");
HandyMenny37d42992014-10-26 22:15:59 +01002177 property_set("sys.usb.config", "mass_storage,adb");
Dees_Troy8170a922012-09-18 15:40:25 -04002178 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002179error_handle:
2180 if (mtp_was_enabled)
2181 if (!Enable_MTP())
2182 Disable_MTP();
2183 return false;
Dees_Troy8170a922012-09-18 15:40:25 -04002184}
2185
2186int TWPartitionManager::usb_storage_disable(void) {
bigbiff22851b92021-09-01 16:46:57 -04002187 int index, ret = 0;
Dees_Troy2673cec2013-04-02 20:22:16 +00002188 char lun_file[255], ch[2] = {0, 0};
2189 string str = ch;
Dees_Troy8170a922012-09-18 15:40:25 -04002190
2191 for (index=0; index<2; index++) {
2192 sprintf(lun_file, CUSTOM_LUN_FILE, index);
bigbiff22851b92021-09-01 16:46:57 -04002193 if (!TWFunc::write_to_file(lun_file, str)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00002194 break;
bigbiff22851b92021-09-01 16:46:57 -04002195 ret = -1;
Dees_Troy8170a922012-09-18 15:40:25 -04002196 }
Dees_Troy8170a922012-09-18 15:40:25 -04002197 }
Dees_Troye58d5262012-09-21 12:27:57 -04002198 Mount_All_Storage();
2199 Update_System_Details();
Dees_Troycfd73ef2012-10-12 16:52:00 -04002200 UnMount_Main_Partitions();
Matt Mowerd9cb9062013-12-14 20:34:45 -06002201 property_set("sys.storage.ums_enabled", "0");
HandyMenny37d42992014-10-26 22:15:59 +01002202 property_set("sys.usb.config", "adb");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002203 if (mtp_was_enabled)
2204 if (!Enable_MTP())
2205 Disable_MTP();
Dees_Troy2673cec2013-04-02 20:22:16 +00002206 if (ret < 0 && index == 0) {
2207 LOGERR("Unable to write to ums lunfile '%s'.", lun_file);
2208 return false;
2209 } else {
2210 return true;
2211 }
Dees_Troy8170a922012-09-18 15:40:25 -04002212 return true;
Dees_Troy812660f2012-09-20 09:55:17 -04002213}
2214
2215void TWPartitionManager::Mount_All_Storage(void) {
2216 std::vector<TWPartition*>::iterator iter;
2217
2218 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2219 if ((*iter)->Is_Storage)
2220 (*iter)->Mount(false);
2221 }
Dees_Troy2c50e182012-09-26 20:05:28 -04002222}
Dees_Troy9350b8d2012-09-27 12:38:38 -04002223
Dees_Troyd0384ef2012-10-12 12:15:42 -04002224void TWPartitionManager::UnMount_Main_Partitions(void) {
2225 // Unmounts system and data if data is not data/media
2226 // Also unmounts boot if boot is mountable
Dees_Troy2673cec2013-04-02 20:22:16 +00002227 LOGINFO("Unmounting main partitions...\n");
Dees_Troyd0384ef2012-10-12 12:15:42 -04002228
Mohd Faraze3948ec2020-08-14 01:40:59 +00002229 TWPartition *Partition = Find_Partition_By_Path ("/vendor");
Dees_Troyd0384ef2012-10-12 12:15:42 -04002230
Mohd Faraze3948ec2020-08-14 01:40:59 +00002231 if (Partition != NULL) UnMount_By_Path("/vendor", false);
2232 UnMount_By_Path (Get_Android_Root_Path(), true);
2233 Partition = Find_Partition_By_Path ("/product");
2234 if (Partition != NULL) UnMount_By_Path("/product", false);
Ethan Yonker6277c792014-09-15 14:54:30 -05002235 if (!datamedia)
2236 UnMount_By_Path("/data", true);
2237
Mohd Faraze3948ec2020-08-14 01:40:59 +00002238 Partition = Find_Partition_By_Path ("/boot");
2239 if (Partition != NULL && Partition->Can_Be_Mounted)
2240 Partition->UnMount(true);
Dees_Troyd0384ef2012-10-12 12:15:42 -04002241}
2242
Dees_Troy9350b8d2012-09-27 12:38:38 -04002243int TWPartitionManager::Partition_SDCard(void) {
Matt Mower23d8aae2017-01-06 14:30:33 -06002244 char temp[255];
Ethan Yonker483e9f42016-01-11 22:21:18 -06002245 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 -04002246 int ext, swap, total_size = 0, fat_size;
Dees_Troy9350b8d2012-09-27 12:38:38 -04002247
Ethan Yonker74db1572015-10-28 12:44:49 -05002248 gui_msg("start_partition_sd=Partitioning SD Card...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06002249
2250 // Locate and validate device to partition
2251 TWPartition* SDCard = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
2252
Ethan Yonker66a19492015-12-10 10:19:45 -06002253 if (SDCard->Is_Adopted_Storage)
2254 SDCard->Revert_Adopted();
2255
Ethan Yonker1eff6cd2014-09-15 13:30:42 -05002256 if (SDCard == NULL || !SDCard->Removable || SDCard->Has_Data_Media) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002257 gui_err("partition_sd_locate=Unable to locate device to partition.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002258 return false;
2259 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002260
2261 // Unmount everything
Dees_Troy9350b8d2012-09-27 12:38:38 -04002262 if (!SDCard->UnMount(true))
2263 return false;
2264 TWPartition* SDext = Find_Partition_By_Path("/sd-ext");
2265 if (SDext != NULL) {
2266 if (!SDext->UnMount(true))
2267 return false;
2268 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002269 char* swappath = getenv("SWAPPATH");
2270 if (swappath != NULL) {
2271 LOGINFO("Unmounting swap at '%s'\n", swappath);
2272 umount(swappath);
2273 }
Vojtech Bocek05534202013-09-11 08:11:56 +02002274
Ethan Yonker483e9f42016-01-11 22:21:18 -06002275 // Determine block device
2276 if (SDCard->Alternate_Block_Device.empty()) {
2277 SDCard->Find_Actual_Block_Device();
2278 Device = SDCard->Actual_Block_Device;
2279 // Just use the root block device
2280 Device.resize(strlen("/dev/block/mmcblkX"));
2281 } else {
2282 Device = SDCard->Alternate_Block_Device;
2283 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04002284
2285 // Find the size of the block device:
Ethan Yonker483e9f42016-01-11 22:21:18 -06002286 total_size = (int)(TWFunc::IOCTL_Get_Block_Size(Device.c_str()) / (1048576));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002287
2288 DataManager::GetValue("tw_sdext_size", ext);
2289 DataManager::GetValue("tw_swap_size", swap);
2290 DataManager::GetValue("tw_sdpart_file_system", ext_format);
2291 fat_size = total_size - ext - swap;
Ethan Yonker483e9f42016-01-11 22:21:18 -06002292 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);
2293
2294 // Determine partition sizes
2295 if (swap == 0 && ext == 0) {
2296 fat_str = "-0";
2297 } else {
2298 memset(temp, 0, sizeof(temp));
2299 sprintf(temp, "%i", fat_size);
2300 fat_str = temp;
2301 fat_str += "MB";
2302 }
2303 if (swap == 0) {
2304 ext_str = "-0";
2305 } else {
2306 memset(temp, 0, sizeof(temp));
2307 sprintf(temp, "%i", ext);
2308 ext_str = "+";
2309 ext_str += temp;
2310 ext_str += "MB";
2311 }
2312
Dees_Troy9350b8d2012-09-27 12:38:38 -04002313 if (ext + swap > total_size) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002314 gui_err("ext_swap_size=EXT + Swap size is larger than sdcard size.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002315 return false;
2316 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002317
Ethan Yonker74db1572015-10-28 12:44:49 -05002318 gui_msg("remove_part_table=Removing partition table...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06002319 Command = "sgdisk --zap-all " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002320 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002321 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002322 gui_err("unable_rm_part=Unable to remove partition table.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002323 Update_System_Details();
2324 return false;
2325 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002326 gui_msg(Msg("create_part=Creating {1} partition...")("FAT32"));
Captain Throwback9643a802016-05-27 11:44:51 -04002327 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 +00002328 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002329 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002330 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("FAT32"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002331 return false;
2332 }
2333 if (ext > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002334 gui_msg(Msg("create_part=Creating {1} partition...")("EXT"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06002335 Command = "sgdisk --new=0:0:" + ext_str + " --change-name=0:\"Linux filesystem\" " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002336 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002337 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002338 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("EXT"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002339 Update_System_Details();
2340 return false;
2341 }
2342 }
2343 if (swap > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002344 gui_msg(Msg("create_part=Creating {1} partition...")("swap"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06002345 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 +00002346 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002347 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002348 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("swap"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002349 Update_System_Details();
2350 return false;
2351 }
2352 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002353
2354 // Convert GPT to MBR
2355 Command = "sgdisk --gpttombr " + Device;
2356 if (TWFunc::Exec_Cmd(Command) != 0)
2357 LOGINFO("Failed to covert partition GPT to MBR\n");
2358
2359 // Tell the kernel to rescan the partition table
2360 int fd = open(Device.c_str(), O_RDONLY);
2361 ioctl(fd, BLKRRPART, 0);
2362 close(fd);
2363
2364 string format_device = Device;
2365 if (Device.substr(0, 17) == "/dev/block/mmcblk")
2366 format_device += "p";
2367
2368 // Format new partitions to proper file system
2369 if (fat_size > 0) {
2370 Command = "mkfs.fat " + format_device + "1";
2371 TWFunc::Exec_Cmd(Command);
2372 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04002373 if (ext > 0) {
2374 if (SDext == NULL) {
Ethan Yonker483e9f42016-01-11 22:21:18 -06002375 Command = "mke2fs -t " + ext_format + " -m 0 " + format_device + "2";
2376 gui_msg(Msg("format_sdext_as=Formatting sd-ext as {1}...")(ext_format));
2377 LOGINFO("Formatting sd-ext after partitioning, command: '%s'\n", Command.c_str());
2378 TWFunc::Exec_Cmd(Command);
2379 } else {
2380 SDext->Wipe(ext_format);
Dees_Troy9350b8d2012-09-27 12:38:38 -04002381 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002382 }
2383 if (swap > 0) {
2384 Command = "mkswap " + format_device;
2385 if (ext > 0)
2386 Command += "3";
2387 else
2388 Command += "2";
Vojtech Bocek05534202013-09-11 08:11:56 +02002389 TWFunc::Exec_Cmd(Command);
Dees_Troy9350b8d2012-09-27 12:38:38 -04002390 }
2391
Ethan Yonker483e9f42016-01-11 22:21:18 -06002392 // recreate TWRP folder and rewrite settings - these will be gone after sdcard is partitioned
2393 if (SDCard->Mount(true)) {
2394 string TWRP_Folder = SDCard->Mount_Point + "/TWRP";
2395 mkdir(TWRP_Folder.c_str(), 0777);
2396 DataManager::Flush();
2397 }
2398
Dees_Troy9350b8d2012-09-27 12:38:38 -04002399 Update_System_Details();
Ethan Yonker74db1572015-10-28 12:44:49 -05002400 gui_msg("part_complete=Partitioning complete.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002401 return true;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04002402}
Dees_Troya13d74f2013-03-24 08:54:55 -05002403
2404void TWPartitionManager::Get_Partition_List(string ListType, std::vector<PartitionList> *Partition_List) {
2405 std::vector<TWPartition*>::iterator iter;
2406 if (ListType == "mount") {
2407 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002408 if ((*iter)->Can_Be_Mounted) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002409 struct PartitionList part;
2410 part.Display_Name = (*iter)->Display_Name;
2411 part.Mount_Point = (*iter)->Mount_Point;
2412 part.selected = (*iter)->Is_Mounted();
2413 Partition_List->push_back(part);
2414 }
2415 }
2416 } else if (ListType == "storage") {
2417 char free_space[255];
2418 string Current_Storage = DataManager::GetCurrentStoragePath();
2419 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2420 if ((*iter)->Is_Storage) {
2421 struct PartitionList part;
2422 sprintf(free_space, "%llu", (*iter)->Free / 1024 / 1024);
2423 part.Display_Name = (*iter)->Storage_Name + " (";
2424 part.Display_Name += free_space;
2425 part.Display_Name += "MB)";
2426 part.Mount_Point = (*iter)->Storage_Path;
2427 if ((*iter)->Storage_Path == Current_Storage)
2428 part.selected = 1;
2429 else
2430 part.selected = 0;
2431 Partition_List->push_back(part);
2432 }
2433 }
2434 } else if (ListType == "backup") {
2435 char backup_size[255];
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002436 unsigned long long Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -05002437 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002438 if ((*iter)->Can_Be_Backed_Up && !(*iter)->Is_SubPartition && (*iter)->Is_Present) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002439 struct PartitionList part;
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002440 Backup_Size = (*iter)->Backup_Size;
2441 if ((*iter)->Has_SubPartition) {
2442 std::vector<TWPartition*>::iterator subpart;
2443
2444 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
2445 if ((*subpart)->Is_SubPartition && (*subpart)->Can_Be_Backed_Up && (*subpart)->Is_Present && (*subpart)->SubPartition_Of == (*iter)->Mount_Point)
2446 Backup_Size += (*subpart)->Backup_Size;
2447 }
2448 }
2449 sprintf(backup_size, "%llu", Backup_Size / 1024 / 1024);
Dees_Troya13d74f2013-03-24 08:54:55 -05002450 part.Display_Name = (*iter)->Backup_Display_Name + " (";
2451 part.Display_Name += backup_size;
2452 part.Display_Name += "MB)";
2453 part.Mount_Point = (*iter)->Backup_Path;
2454 part.selected = 0;
2455 Partition_List->push_back(part);
2456 }
2457 }
2458 } else if (ListType == "restore") {
2459 string Restore_List, restore_path;
2460 TWPartition* restore_part = NULL;
2461
2462 DataManager::GetValue("tw_restore_list", Restore_List);
2463 if (!Restore_List.empty()) {
2464 size_t start_pos = 0, end_pos = Restore_List.find(";", start_pos);
2465 while (end_pos != string::npos && start_pos < Restore_List.size()) {
2466 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04002467 struct PartitionList part;
2468 if (restore_path.compare("ADB_Backup") == 0) {
2469 part.Display_Name = "ADB Backup";
2470 part.Mount_Point = "ADB Backup";
2471 part.selected = 1;
2472 Partition_List->push_back(part);
2473 break;
2474 }
Dees_Troy59df9262013-06-19 14:53:57 -05002475 if ((restore_part = Find_Partition_By_Path(restore_path)) != NULL) {
Dees Troy4159aed2014-02-28 17:24:43 +00002476 if ((restore_part->Backup_Name == "recovery" && !restore_part->Can_Be_Backed_Up) || restore_part->Is_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002477 // Don't allow restore of recovery (causes problems on some devices)
Dees_Troy59df9262013-06-19 14:53:57 -05002478 // Don't add subpartitions to the list of items
Dees_Troya13d74f2013-03-24 08:54:55 -05002479 } else {
Dees_Troya13d74f2013-03-24 08:54:55 -05002480 part.Display_Name = restore_part->Backup_Display_Name;
2481 part.Mount_Point = restore_part->Backup_Path;
2482 part.selected = 1;
2483 Partition_List->push_back(part);
2484 }
2485 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002486 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05002487 }
2488 start_pos = end_pos + 1;
2489 end_pos = Restore_List.find(";", start_pos);
2490 }
2491 }
2492 } else if (ListType == "wipe") {
2493 struct PartitionList dalvik;
Ethan Yonker74db1572015-10-28 12:44:49 -05002494 dalvik.Display_Name = gui_parse_text("{@dalvik}");
Dees_Troya13d74f2013-03-24 08:54:55 -05002495 dalvik.Mount_Point = "DALVIK";
2496 dalvik.selected = 0;
2497 Partition_List->push_back(dalvik);
2498 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2499 if ((*iter)->Wipe_Available_in_GUI && !(*iter)->Is_SubPartition) {
2500 struct PartitionList part;
2501 part.Display_Name = (*iter)->Display_Name;
2502 part.Mount_Point = (*iter)->Mount_Point;
2503 part.selected = 0;
2504 Partition_List->push_back(part);
2505 }
2506 if ((*iter)->Has_Android_Secure) {
2507 struct PartitionList part;
2508 part.Display_Name = (*iter)->Backup_Display_Name;
2509 part.Mount_Point = (*iter)->Backup_Path;
2510 part.selected = 0;
2511 Partition_List->push_back(part);
2512 }
Dees_Troy74fb2e92013-04-15 14:35:47 +00002513 if ((*iter)->Has_Data_Media) {
2514 struct PartitionList datamedia;
2515 datamedia.Display_Name = (*iter)->Storage_Name;
2516 datamedia.Mount_Point = "INTERNAL";
2517 datamedia.selected = 0;
2518 Partition_List->push_back(datamedia);
2519 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002520 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002521 } else if (ListType == "flashimg") {
2522 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2523 if ((*iter)->Can_Flash_Img && (*iter)->Is_Present) {
2524 struct PartitionList part;
2525 part.Display_Name = (*iter)->Backup_Display_Name;
2526 part.Mount_Point = (*iter)->Backup_Path;
2527 part.selected = 0;
2528 Partition_List->push_back(part);
2529 }
2530 }
Ethan Yonker53796e72019-01-11 22:49:52 -06002531 if (DataManager::GetIntValue("tw_has_repack_tools") != 0 && DataManager::GetIntValue("tw_has_boot_slots") != 0) {
2532 TWPartition* boot = Find_Partition_By_Path("/boot");
2533 if (boot) {
2534 // Allow flashing kernels and ramdisks
2535 struct PartitionList repack_ramdisk;
2536 repack_ramdisk.Display_Name = gui_lookup("install_twrp_ramdisk", "Install Recovery Ramdisk");
2537 repack_ramdisk.Mount_Point = "/repack_ramdisk";
2538 repack_ramdisk.selected = 0;
2539 Partition_List->push_back(repack_ramdisk);
2540 /*struct PartitionList repack_kernel; For now let's leave repacking kernels under advanced only
2541 repack_kernel.Display_Name = gui_lookup("install_kernel", "Install Kernel");
2542 repack_kernel.Mount_Point = "/repack_kernel";
2543 repack_kernel.selected = 0;
2544 Partition_List->push_back(repack_kernel);*/
2545 }
2546 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002547 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00002548 LOGERR("Unknown list type '%s' requested for TWPartitionManager::Get_Partition_List\n", ListType.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -05002549 }
2550}
2551
2552int TWPartitionManager::Fstab_Processed(void) {
2553 return Partitions.size();
2554}
Dees_Troyd93bda52013-07-03 19:55:19 +00002555
2556void TWPartitionManager::Output_Storage_Fstab(void) {
2557 std::vector<TWPartition*>::iterator iter;
2558 char storage_partition[255];
bigbiff bigbiffe4bdb152019-03-23 18:33:17 -04002559 std::string Temp;
bigbiff25d25b92020-06-19 16:07:38 -04002560 std::string cacheDir = TWFunc::get_log_dir();
bigbiff bigbiffe4bdb152019-03-23 18:33:17 -04002561
2562 if (cacheDir.empty()) {
2563 LOGINFO("Unable to find cache directory\n");
2564 return;
2565 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002566
bigbiff25d25b92020-06-19 16:07:38 -04002567 std::string storageFstab = TWFunc::get_log_dir() + "recovery/storage.fstab";
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002568 FILE *fp = fopen(storageFstab.c_str(), "w");
Dees_Troyd93bda52013-07-03 19:55:19 +00002569
2570 if (fp == NULL) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002571 gui_msg(Msg(msg::kError, "unable_to_open=Unable to open '{1}'.")(storageFstab));
Dees_Troyd93bda52013-07-03 19:55:19 +00002572 return;
2573 }
2574
2575 // Iterate through all partitions
2576 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2577 if ((*iter)->Is_Storage) {
2578 Temp = (*iter)->Storage_Path + ";" + (*iter)->Storage_Name + ";\n";
2579 strcpy(storage_partition, Temp.c_str());
2580 fwrite(storage_partition, sizeof(storage_partition[0]), strlen(storage_partition) / sizeof(storage_partition[0]), fp);
2581 }
2582 }
2583 fclose(fp);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02002584}
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002585
2586TWPartition *TWPartitionManager::Get_Default_Storage_Partition()
2587{
2588 TWPartition *res = NULL;
2589 for (std::vector<TWPartition*>::iterator iter = Partitions.begin(); iter != Partitions.end(); ++iter) {
Matt Mowera8a89d12016-12-30 18:10:37 -06002590 if (!(*iter)->Is_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002591 continue;
2592
Matt Mowera8a89d12016-12-30 18:10:37 -06002593 if ((*iter)->Is_Settings_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002594 return *iter;
2595
Matt Mowera8a89d12016-12-30 18:10:37 -06002596 if (!res)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002597 res = *iter;
2598 }
2599 return res;
2600}
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002601
2602bool TWPartitionManager::Enable_MTP(void) {
2603#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002604 if (mtppid) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002605 gui_err("mtp_already_enabled=MTP already enabled");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002606 return true;
2607 }
Ethan Yonker726a0202014-12-16 20:01:38 -06002608
2609 int mtppipe[2];
2610
2611 if (pipe(mtppipe) < 0) {
2612 LOGERR("Error creating MTP pipe\n");
2613 return false;
2614 }
2615
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002616 char old_value[PROPERTY_VALUE_MAX];
Matt Mowere07f0102017-02-23 17:40:00 -06002617 property_get("sys.usb.config", old_value, "");
2618 if (strcmp(old_value, "mtp,adb") != 0) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002619 char vendor[PROPERTY_VALUE_MAX];
2620 char product[PROPERTY_VALUE_MAX];
2621 property_set("sys.usb.config", "none");
2622 property_get("usb.vendor", vendor, "18D1");
2623 property_get("usb.product.mtpadb", product, "4EE2");
2624 string vendorstr = vendor;
2625 string productstr = product;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002626 TWFunc::write_to_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2627 TWFunc::write_to_file("/sys/class/android_usb/android0/idProduct", productstr);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002628 property_set("sys.usb.config", "mtp,adb");
2629 }
Matt Mower653a1702017-02-16 10:38:52 -06002630 /* To enable MTP debug, use the twrp command line feature:
Ethan Yonker6d154c42014-09-03 14:19:43 -05002631 * twrp set tw_mtp_debug 1
2632 */
2633 twrpMtp *mtp = new twrpMtp(DataManager::GetIntValue("tw_mtp_debug"));
Ethan Yonker1b039202015-01-30 10:08:48 -06002634 mtppid = mtp->forkserver(mtppipe);
2635 if (mtppid) {
2636 close(mtppipe[0]); // Host closes read side
2637 mtp_write_fd = mtppipe[1];
2638 DataManager::SetValue("tw_mtp_enabled", 1);
2639 Add_All_MTP_Storage();
2640 return true;
Ethan Yonker726a0202014-12-16 20:01:38 -06002641 } else {
2642 close(mtppipe[0]);
2643 close(mtppipe[1]);
Ethan Yonker74db1572015-10-28 12:44:49 -05002644 gui_err("mtp_fail=Failed to enable MTP");
Ethan Yonker1b039202015-01-30 10:08:48 -06002645 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002646 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002647#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002648 gui_err("no_mtp=MTP support not included");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002649#endif
2650 DataManager::SetValue("tw_mtp_enabled", 0);
2651 return false;
2652}
2653
Ethan Yonker1b039202015-01-30 10:08:48 -06002654void TWPartitionManager::Add_All_MTP_Storage(void) {
2655#ifdef TW_HAS_MTP
2656 std::vector<TWPartition*>::iterator iter;
2657
2658 if (!mtppid)
2659 return; // MTP is not enabled
2660
2661 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2662 if ((*iter)->Is_Storage && (*iter)->Is_Present && (*iter)->Mount(false))
2663 Add_Remove_MTP_Storage((*iter), MTP_MESSAGE_ADD_STORAGE);
2664 }
2665#else
2666 return;
2667#endif
2668}
2669
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002670bool TWPartitionManager::Disable_MTP(void) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002671 char old_value[PROPERTY_VALUE_MAX];
bigbiff bigbiffaf32bb92018-12-18 18:39:53 -05002672 property_set("sys.usb.ffs.mtp.ready", "0");
Matt Mowere07f0102017-02-23 17:40:00 -06002673 property_get("sys.usb.config", old_value, "");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002674 if (strcmp(old_value, "adb") != 0) {
2675 char vendor[PROPERTY_VALUE_MAX];
2676 char product[PROPERTY_VALUE_MAX];
2677 property_set("sys.usb.config", "none");
2678 property_get("usb.vendor", vendor, "18D1");
Matt Mowere07f0102017-02-23 17:40:00 -06002679 property_get("usb.product.adb", product, "D001");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002680 string vendorstr = vendor;
2681 string productstr = product;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002682 TWFunc::write_to_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2683 TWFunc::write_to_file("/sys/class/android_usb/android0/idProduct", productstr);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002684 usleep(2000);
2685 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002686#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002687 if (mtppid) {
Ethan Yonker8613dc02014-09-11 09:28:20 -05002688 LOGINFO("Disabling MTP\n");
2689 int status;
2690 kill(mtppid, SIGKILL);
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002691 mtppid = 0;
Ethan Yonker8613dc02014-09-11 09:28:20 -05002692 // We don't care about the exit value, but this prevents a zombie process
2693 waitpid(mtppid, &status, 0);
Ethan Yonker726a0202014-12-16 20:01:38 -06002694 close(mtp_write_fd);
2695 mtp_write_fd = -1;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002696 }
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002697#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002698 property_set("sys.usb.config", "adb");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002699#ifdef TW_HAS_MTP
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002700 DataManager::SetValue("tw_mtp_enabled", 0);
2701 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002702#endif
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002703 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002704}
Ethan Yonker726a0202014-12-16 20:01:38 -06002705
2706TWPartition* TWPartitionManager::Find_Partition_By_MTP_Storage_ID(unsigned int Storage_ID) {
2707 std::vector<TWPartition*>::iterator iter;
2708
2709 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2710 if ((*iter)->MTP_Storage_ID == Storage_ID)
2711 return (*iter);
2712 }
2713 return NULL;
2714}
2715
2716bool TWPartitionManager::Add_Remove_MTP_Storage(TWPartition* Part, int message_type) {
2717#ifdef TW_HAS_MTP
2718 struct mtpmsg mtp_message;
2719
2720 if (!mtppid)
2721 return false; // MTP is disabled
2722
2723 if (mtp_write_fd < 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002724 LOGINFO("MTP: mtp_write_fd is not set\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002725 return false;
2726 }
2727
2728 if (Part) {
Ethan Yonker4bfabab2014-12-29 09:15:37 -06002729 if (Part->MTP_Storage_ID == 0)
2730 return false;
Ethan Yonker726a0202014-12-16 20:01:38 -06002731 if (message_type == MTP_MESSAGE_REMOVE_STORAGE) {
2732 mtp_message.message_type = MTP_MESSAGE_REMOVE_STORAGE; // Remove
2733 LOGINFO("sending message to remove %i\n", Part->MTP_Storage_ID);
2734 mtp_message.storage_id = Part->MTP_Storage_ID;
2735 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002736 LOGINFO("error sending message to remove storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002737 return false;
2738 } else {
2739 LOGINFO("Message sent, remove storage ID: %i\n", Part->MTP_Storage_ID);
2740 return true;
2741 }
2742 } else if (message_type == MTP_MESSAGE_ADD_STORAGE && Part->Is_Mounted()) {
2743 mtp_message.message_type = MTP_MESSAGE_ADD_STORAGE; // Add
2744 mtp_message.storage_id = Part->MTP_Storage_ID;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002745 if (Part->Storage_Path.size() >= sizeof(mtp_message.path)) {
2746 LOGERR("Storage path '%s' too large for mtpmsg\n", Part->Storage_Path.c_str());
2747 return false;
2748 }
2749 strcpy(mtp_message.path, Part->Storage_Path.c_str());
2750 if (Part->Storage_Name.size() >= sizeof(mtp_message.display)) {
2751 LOGERR("Storage name '%s' too large for mtpmsg\n", Part->Storage_Name.c_str());
2752 return false;
2753 }
2754 strcpy(mtp_message.display, Part->Storage_Name.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002755 mtp_message.maxFileSize = Part->Get_Max_FileSize();
Ethan Yonker1b039202015-01-30 10:08:48 -06002756 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 -06002757 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002758 LOGINFO("error sending message to add storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002759 return false;
2760 } else {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002761 LOGINFO("Message sent, add storage ID: %i '%s'\n", Part->MTP_Storage_ID, mtp_message.path);
Ethan Yonker726a0202014-12-16 20:01:38 -06002762 return true;
2763 }
2764 } else {
2765 LOGERR("Unknown MTP message type: %i\n", message_type);
2766 }
2767 } else {
2768 // This hopefully never happens as the error handling should
2769 // occur in the calling function.
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002770 LOGINFO("TWPartitionManager::Add_Remove_MTP_Storage NULL partition given\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002771 }
2772 return true;
2773#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002774 gui_err("no_mtp=MTP support not included");
Ethan Yonker726a0202014-12-16 20:01:38 -06002775 DataManager::SetValue("tw_mtp_enabled", 0);
2776 return false;
2777#endif
2778}
2779
2780bool TWPartitionManager::Add_MTP_Storage(string Mount_Point) {
2781#ifdef TW_HAS_MTP
2782 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2783 if (Part) {
2784 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2785 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002786 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002787 }
2788#endif
2789 return false;
2790}
2791
2792bool TWPartitionManager::Add_MTP_Storage(unsigned int Storage_ID) {
2793#ifdef TW_HAS_MTP
2794 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2795 if (Part) {
2796 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2797 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002798 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002799 }
2800#endif
2801 return false;
2802}
2803
2804bool TWPartitionManager::Remove_MTP_Storage(string Mount_Point) {
2805#ifdef TW_HAS_MTP
2806 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2807 if (Part) {
2808 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2809 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002810 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002811 }
2812#endif
2813 return false;
2814}
2815
2816bool TWPartitionManager::Remove_MTP_Storage(unsigned int Storage_ID) {
2817#ifdef TW_HAS_MTP
2818 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2819 if (Part) {
2820 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2821 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002822 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002823 }
2824#endif
2825 return false;
2826}
Ethan Yonker96af84a2015-01-05 14:58:36 -06002827
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002828bool TWPartitionManager::Flash_Image(string& path, string& filename) {
bigbiffad58e1b2020-07-06 20:24:34 -04002829 twrpRepacker repacker;
Matt Mower23d8aae2017-01-06 14:30:33 -06002830 int partition_count = 0;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002831 TWPartition* flash_part = NULL;
bigbiffce8f83c2015-12-12 18:30:21 -05002832 string Flash_List, flash_path, full_filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002833 size_t start_pos = 0, end_pos = 0;
2834
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002835 full_filename = path + "/" + filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002836
bigbiff1f9e4842020-10-31 11:33:15 -04002837 Unlock_Block_Partitions();
2838
bigbiffce8f83c2015-12-12 18:30:21 -05002839 gui_msg("image_flash_start=[IMAGE FLASH STARTED]");
2840 gui_msg(Msg("img_to_flash=Image to flash: '{1}'")(full_filename));
2841
2842 if (!TWFunc::Path_Exists(full_filename)) {
2843 if (!Mount_By_Path(full_filename, true)) {
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002844 return false;
2845 }
bigbiffce8f83c2015-12-12 18:30:21 -05002846 if (!TWFunc::Path_Exists(full_filename)) {
2847 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")(full_filename));
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002848 return false;
2849 }
2850 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002851
Ethan Yonker53796e72019-01-11 22:49:52 -06002852 DataManager::GetValue("tw_flash_partition", Flash_List);
2853 Repack_Type repack = REPLACE_NONE;
2854 if (Flash_List == "/repack_ramdisk;") {
2855 repack = REPLACE_RAMDISK;
2856 } else if (Flash_List == "/repack_kernel;") {
2857 repack = REPLACE_KERNEL;
2858 }
2859 if (repack != REPLACE_NONE) {
2860 Repack_Options_struct Repack_Options;
2861 Repack_Options.Type = repack;
2862 Repack_Options.Disable_Verity = false;
2863 Repack_Options.Disable_Force_Encrypt = false;
2864 Repack_Options.Backup_First = DataManager::GetIntValue("tw_repack_backup_first") != 0;
bigbiffad58e1b2020-07-06 20:24:34 -04002865 return repacker.Repack_Image_And_Flash(full_filename, Repack_Options);
Ethan Yonker53796e72019-01-11 22:49:52 -06002866 }
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002867 PartitionSettings part_settings;
2868 part_settings.Backup_Folder = path;
2869 unsigned long long total_bytes = TWFunc::Get_File_Size(full_filename);
2870 ProgressTracking progress(total_bytes);
2871 part_settings.progress = &progress;
2872 part_settings.adbbackup = false;
2873 part_settings.PM_Method = PM_RESTORE;
Ethan Yonker74db1572015-10-28 12:44:49 -05002874 gui_msg("calc_restore=Calculating restore details...");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002875 if (!Flash_List.empty()) {
2876 end_pos = Flash_List.find(";", start_pos);
2877 while (end_pos != string::npos && start_pos < Flash_List.size()) {
2878 flash_path = Flash_List.substr(start_pos, end_pos - start_pos);
2879 flash_part = Find_Partition_By_Path(flash_path);
2880 if (flash_part != NULL) {
2881 partition_count++;
2882 if (partition_count > 1) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002883 gui_err("too_many_flash=Too many partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002884 return false;
2885 }
2886 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002887 gui_msg(Msg(msg::kError, "flash_unable_locate=Unable to locate '{1}' partition for flashing.")(flash_path));
Ethan Yonker96af84a2015-01-05 14:58:36 -06002888 return false;
2889 }
2890 start_pos = end_pos + 1;
2891 end_pos = Flash_List.find(";", start_pos);
2892 }
2893 }
2894
2895 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002896 gui_err("no_part_flash=No partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002897 return false;
2898 }
2899
2900 DataManager::SetProgress(0.0);
2901 if (flash_part) {
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002902 flash_part->Backup_FileName = filename;
2903 if (!flash_part->Flash_Image(&part_settings))
Ethan Yonker96af84a2015-01-05 14:58:36 -06002904 return false;
2905 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002906 gui_err("invalid_flash=Invalid flash partition specified.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002907 return false;
2908 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002909 gui_highlight("flash_done=IMAGE FLASH COMPLETED]");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002910 return true;
2911}
Ethan Yonker74db1572015-10-28 12:44:49 -05002912
2913void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value) {
2914 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2915 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002916 if (part->Is_Adopted_Storage) {
2917 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2918 part->Backup_Display_Name = part->Display_Name;
2919 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2920 } else {
2921 part->Display_Name = gui_lookup(resource_name, default_value);
2922 part->Backup_Display_Name = part->Display_Name;
2923 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002924 }
2925}
2926
2927void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value, const char* storage_resource_name, const char* storage_default_value) {
2928 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2929 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002930 if (part->Is_Adopted_Storage) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06002931 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup("data_backup", "Data (excl. storage)");
Ethan Yonker66a19492015-12-10 10:19:45 -06002932 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
Ethan Yonker66a19492015-12-10 10:19:45 -06002933 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2934 } else {
2935 part->Display_Name = gui_lookup(resource_name, default_value);
2936 part->Backup_Display_Name = part->Display_Name;
2937 if (part->Is_Storage)
2938 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2939 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002940 }
2941}
2942
Ethan Yonker01f4e032017-02-03 15:30:52 -06002943void 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) {
2944 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2945 if (part) {
2946 if (part->Is_Adopted_Storage) {
2947 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup(backup_name, backup_default);
2948 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2949 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2950 } else {
2951 part->Display_Name = gui_lookup(resource_name, default_value);
2952 part->Backup_Display_Name = gui_lookup(backup_name, backup_default);
2953 if (part->Is_Storage)
2954 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2955 }
2956 }
2957}
2958
Ethan Yonker74db1572015-10-28 12:44:49 -05002959void TWPartitionManager::Translate_Partition_Display_Names() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002960 LOGINFO("Translating partition display names\n");
Ethan Yonker74db1572015-10-28 12:44:49 -05002961 Translate_Partition("/system", "system", "System");
2962 Translate_Partition("/system_image", "system_image", "System Image");
2963 Translate_Partition("/vendor", "vendor", "Vendor");
2964 Translate_Partition("/vendor_image", "vendor_image", "Vendor Image");
2965 Translate_Partition("/cache", "cache", "Cache");
Ethan Yonker74db1572015-10-28 12:44:49 -05002966 Translate_Partition("/boot", "boot", "Boot");
2967 Translate_Partition("/recovery", "recovery", "Recovery");
2968 if (!datamedia) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06002969 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage");
Ethan Yonker74db1572015-10-28 12:44:49 -05002970 Translate_Partition("/sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2971 Translate_Partition("/internal_sd", "sdcard", "SDCard", "sdcard", "SDCard");
2972 Translate_Partition("/internal_sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2973 Translate_Partition("/emmc", "sdcard", "SDCard", "sdcard", "SDCard");
Ethan Yonker01f4e032017-02-03 15:30:52 -06002974 } else {
2975 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05002976 }
Ethan Yonker01f4e032017-02-03 15:30:52 -06002977 Translate_Partition("/external_sd", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
2978 Translate_Partition("/external_sdcard", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05002979 Translate_Partition("/usb-otg", "usbotg", "USB OTG", "usbotg", "USB OTG");
2980 Translate_Partition("/sd-ext", "sdext", "SD-EXT");
2981
2982 // Android secure is a special case
2983 TWPartition* part = PartitionManager.Find_Partition_By_Path("/and-sec");
2984 if (part)
2985 part->Backup_Display_Name = gui_lookup("android_secure", "Android Secure");
2986
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002987 std::vector<TWPartition*>::iterator sysfs;
2988 for (sysfs = Partitions.begin(); sysfs != Partitions.end(); sysfs++) {
2989 if (!(*sysfs)->Sysfs_Entry.empty()) {
2990 Translate_Partition((*sysfs)->Mount_Point.c_str(), "autostorage", "Storage", "autostorage", "Storage");
2991 }
2992 }
2993
Ethan Yonker74db1572015-10-28 12:44:49 -05002994 // This updates the text on all of the storage selection buttons in the GUI
2995 DataManager::SetBackupFolder();
2996}
Ethan Yonker66a19492015-12-10 10:19:45 -06002997
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002998bool TWPartitionManager::Decrypt_Adopted() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002999#ifdef TW_INCLUDE_CRYPTO
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003000 bool ret = false;
Ethan Yonker66a19492015-12-10 10:19:45 -06003001 if (!Mount_By_Path("/data", false)) {
3002 LOGERR("Cannot decrypt adopted storage because /data will not mount\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003003 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06003004 }
zhenyolka90850472021-11-08 19:05:30 +03003005
3006 // In Android 12 xml format changed. Previously it was human-readable format with xml tags
3007 // now it's ABX (Android Binary Xml). Sadly, rapidxml can't parse it, so check xml format firstly
3008 std::string path = "/data/system/storage.xml";
3009 if ((atoi(TWFunc::System_Property_Get("ro.build.version.sdk").c_str()) > 30) && TWFunc::Path_Exists(path))
3010 if(!TWFunc::Check_Xml_Format(path)) {
3011 LOGINFO("Android 12+: storage.xml is in ABX format. Skipping adopted storage decryption\n");
3012 return false;
3013 }
3014
Ethan Yonker66a19492015-12-10 10:19:45 -06003015 LOGINFO("Decrypt adopted storage starting\n");
3016 char* xmlFile = PageManager::LoadFileToBuffer("/data/system/storage.xml", NULL);
3017 xml_document<> *doc = NULL;
3018 xml_node<>* volumes = NULL;
Ethan Yonker66a19492015-12-10 10:19:45 -06003019 string Primary_Storage_UUID = "";
3020 if (xmlFile != NULL) {
3021 LOGINFO("successfully loaded storage.xml\n");
3022 doc = new xml_document<>();
3023 doc->parse<0>(xmlFile);
3024 volumes = doc->first_node("volumes");
3025 if (volumes) {
3026 xml_attribute<>* psuuid = volumes->first_attribute("primaryStorageUuid");
3027 if (psuuid) {
3028 Primary_Storage_UUID = psuuid->value();
3029 }
3030 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003031 } else {
3032 LOGINFO("No /data/system/storage.xml for adopted storage\n");
3033 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06003034 }
3035 std::vector<TWPartition*>::iterator adopt;
3036 for (adopt = Partitions.begin(); adopt != Partitions.end(); adopt++) {
Peter Cai439d60c2019-10-17 08:49:10 +08003037 if ((*adopt)->Removable && !(*adopt)->Is_Present && (*adopt)->Adopted_Mount_Delay > 0) {
3038 // On some devices, the external mmc driver takes some time
3039 // to recognize the card, in which case the "actual block device"
3040 // would not have been found yet. We wait the specified delay
3041 // and then try again.
3042 LOGINFO("Sleeping %d seconds for adopted storage.\n", (*adopt)->Adopted_Mount_Delay);
3043 sleep((*adopt)->Adopted_Mount_Delay);
3044 (*adopt)->Find_Actual_Block_Device();
3045 }
3046
Ethan Yonker66a19492015-12-10 10:19:45 -06003047 if ((*adopt)->Removable && (*adopt)->Is_Present) {
3048 if ((*adopt)->Decrypt_Adopted() == 0) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003049 ret = true;
Ethan Yonker66a19492015-12-10 10:19:45 -06003050 if (volumes) {
3051 xml_node<>* volume = volumes->first_node("volume");
3052 while (volume) {
3053 xml_attribute<>* guid = volume->first_attribute("partGuid");
3054 if (guid) {
3055 string GUID = (*adopt)->Adopted_GUID.c_str();
3056 GUID.insert(8, "-");
3057 GUID.insert(13, "-");
3058 GUID.insert(18, "-");
3059 GUID.insert(23, "-");
3060
3061 if (strcasecmp(GUID.c_str(), guid->value()) == 0) {
3062 xml_attribute<>* attr = volume->first_attribute("nickname");
nkk71ffb02bd2017-06-04 23:12:16 +03003063 if (attr && attr->value() && strlen(attr->value()) > 0) {
Ethan Yonker66a19492015-12-10 10:19:45 -06003064 (*adopt)->Storage_Name = attr->value();
3065 (*adopt)->Display_Name = (*adopt)->Storage_Name;
3066 (*adopt)->Backup_Display_Name = (*adopt)->Storage_Name;
3067 LOGINFO("storage name from storage.xml is '%s'\n", attr->value());
3068 }
3069 attr = volume->first_attribute("fsUuid");
3070 if (attr && !Primary_Storage_UUID.empty() && strcmp(Primary_Storage_UUID.c_str(), attr->value()) == 0) {
3071 TWPartition* Dat = Find_Partition_By_Path("/data");
3072 if (Dat) {
3073 LOGINFO("Internal storage is found on adopted storage '%s'\n", (*adopt)->Display_Name.c_str());
3074 LOGINFO("Changing '%s' to point to '%s'\n", Dat->Symlink_Mount_Point.c_str(), (*adopt)->Storage_Path.c_str());
3075 (*adopt)->Symlink_Mount_Point = Dat->Symlink_Mount_Point;
3076 Dat->Symlink_Mount_Point = "";
3077 // Toggle mounts to ensure that the symlink mount point (probably /sdcard) is mounted to the right location
3078 Dat->UnMount(false);
3079 Dat->Mount(false);
3080 (*adopt)->UnMount(false);
3081 (*adopt)->Mount(false);
Ethan Yonker66a19492015-12-10 10:19:45 -06003082 }
3083 }
3084 break;
3085 }
3086 }
3087 volume = volume->next_sibling("volume");
3088 }
3089 }
nkk71ffb02bd2017-06-04 23:12:16 +03003090 Update_System_Details();
3091 Output_Partition((*adopt));
Ethan Yonker66a19492015-12-10 10:19:45 -06003092 }
3093 }
3094 }
3095 if (xmlFile) {
3096 doc->clear();
3097 delete doc;
3098 free(xmlFile);
3099 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003100 return ret;
Ethan Yonker66a19492015-12-10 10:19:45 -06003101#else
3102 LOGINFO("Decrypt_Adopted: no crypto support\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003103 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06003104#endif
3105}
Ethan Yonkerfcf3f242016-02-16 12:30:26 -06003106
3107void TWPartitionManager::Remove_Partition_By_Path(string Path) {
3108 std::vector<TWPartition*>::iterator iter;
3109 string Local_Path = TWFunc::Get_Root_Path(Path);
3110
3111 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3112 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
3113 LOGINFO("Found and erasing '%s' from partition list\n", Local_Path.c_str());
3114 Partitions.erase(iter);
3115 return;
3116 }
3117 }
3118}
Ethan Yonker1b190162016-12-05 15:25:19 -06003119
bigbiff74a46272021-07-31 19:02:26 -04003120void TWPartitionManager::Override_Active_Slot(const string& Slot) {
3121 LOGINFO("Overriding slot to '%s'\n", Slot.c_str());
3122 Active_Slot_Display = Slot;
3123 DataManager::SetValue("tw_active_slot", Slot);
nebrassy0fc1ee22021-11-21 03:32:47 +01003124 PartitionManager.Update_System_Details();
bigbiff74a46272021-07-31 19:02:26 -04003125}
3126
Ethan Yonker1b190162016-12-05 15:25:19 -06003127void TWPartitionManager::Set_Active_Slot(const string& Slot) {
3128 if (Slot != "A" && Slot != "B") {
3129 LOGERR("Set_Active_Slot invalid slot '%s'\n", Slot.c_str());
3130 return;
3131 }
3132 if (Active_Slot_Display == Slot)
3133 return;
3134 LOGINFO("Setting active slot %s\n", Slot.c_str());
3135#ifdef AB_OTA_UPDATER
3136 if (!Active_Slot_Display.empty()) {
dianlujitaob20dfaa2021-12-16 16:56:26 +08003137 android::sp<IBootControl> module = IBootControl::getService();
3138 if (module == nullptr) {
Ethan Yonker1b190162016-12-05 15:25:19 -06003139 LOGERR("Error getting bootctrl module.\n");
3140 } else {
dianlujitaob20dfaa2021-12-16 16:56:26 +08003141 uint32_t slot_number = 0;
Ethan Yonker1b190162016-12-05 15:25:19 -06003142 if (Slot == "B")
3143 slot_number = 1;
dianlujitaob20dfaa2021-12-16 16:56:26 +08003144 CommandResult result;
3145 auto ret = module->setActiveBootSlot(slot_number, [&result]
3146 (const CommandResult &cb_result) { result = cb_result; });
3147 if (!ret.isOk() || !result.success)
Ethan Yonker1b190162016-12-05 15:25:19 -06003148 gui_msg(Msg(msg::kError, "unable_set_boot_slot=Error changing bootloader boot slot to {1}")(Slot));
3149 }
3150 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
3151 }
3152#else
3153 LOGERR("Boot slot feature not present\n");
3154#endif
3155 Active_Slot_Display = Slot;
3156 if (Fstab_Processed())
3157 Update_System_Details();
3158}
3159string TWPartitionManager::Get_Active_Slot_Suffix() {
3160 if (Active_Slot_Display == "A")
3161 return "_a";
3162 return "_b";
3163}
3164string TWPartitionManager::Get_Active_Slot_Display() {
3165 return Active_Slot_Display;
3166}
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003167
Captain Throwback9d6feb52018-07-27 10:05:24 -04003168string TWPartitionManager::Get_Android_Root_Path() {
bigbiff8da46fa2020-09-08 16:42:34 -04003169 return "/system_root";
Captain Throwback9d6feb52018-07-27 10:05:24 -04003170}
3171
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003172void TWPartitionManager::Remove_Uevent_Devices(const string& Mount_Point) {
3173 std::vector<TWPartition*>::iterator iter;
3174
3175 for (iter = Partitions.begin(); iter != Partitions.end(); ) {
3176 if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Mount_Point) {
3177 TWPartition *part = *iter;
3178 LOGINFO("%s was removed by uevent data\n", (*iter)->Mount_Point.c_str());
3179 (*iter)->UnMount(false);
3180 rmdir((*iter)->Mount_Point.c_str());
3181 iter = Partitions.erase(iter);
3182 delete part;
3183 } else {
3184 iter++;
3185 }
3186 }
3187}
3188
3189void TWPartitionManager::Handle_Uevent(const Uevent_Block_Data& uevent_data) {
3190 std::vector<TWPartition*>::iterator iter;
3191
3192 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3193 if (!(*iter)->Sysfs_Entry.empty()) {
3194 string device;
3195 size_t wildcard = (*iter)->Sysfs_Entry.find("*");
3196 if (wildcard != string::npos) {
3197 device = (*iter)->Sysfs_Entry.substr(0, wildcard);
3198 } else {
3199 device = (*iter)->Sysfs_Entry;
3200 }
3201 if (device == uevent_data.sysfs_path.substr(0, device.size())) {
3202 // Found a match
3203 if (uevent_data.action == "add") {
3204 (*iter)->Primary_Block_Device = "/dev/block/" + uevent_data.block_device;
3205 (*iter)->Alternate_Block_Device = (*iter)->Primary_Block_Device;
3206 (*iter)->Is_Present = true;
3207 LOGINFO("Found a match '%s' '%s'\n", uevent_data.block_device.c_str(), device.c_str());
3208 if (!Decrypt_Adopted()) {
3209 LOGINFO("No adopted storage so finding actual block device\n");
3210 (*iter)->Find_Actual_Block_Device();
3211 }
3212 return;
3213 } else if (uevent_data.action == "remove") {
3214 (*iter)->Is_Present = false;
3215 (*iter)->Primary_Block_Device = "";
3216 (*iter)->Actual_Block_Device = "";
3217 Remove_Uevent_Devices((*iter)->Mount_Point);
3218 return;
3219 }
3220 }
3221 }
3222 }
bigbiffee7b7ff2020-03-23 15:08:27 -04003223
3224 if (!PartitionManager.Get_Super_Status())
3225 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 -06003226}
3227
3228void TWPartitionManager::setup_uevent() {
3229 struct sockaddr_nl nls;
3230
3231 if (uevent_pfd.fd >= 0) {
3232 LOGINFO("uevent already set up\n");
3233 return;
3234 }
3235
3236 // Open hotplug event netlink socket
3237 memset(&nls,0,sizeof(struct sockaddr_nl));
3238 nls.nl_family = AF_NETLINK;
3239 nls.nl_pid = getpid();
3240 nls.nl_groups = -1;
3241 uevent_pfd.events = POLLIN;
3242 uevent_pfd.fd = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_KOBJECT_UEVENT);
3243 if (uevent_pfd.fd==-1) {
3244 LOGERR("uevent not root\n");
3245 return;
3246 }
3247
3248 // Listen to netlink socket
3249 if (::bind(uevent_pfd.fd, (struct sockaddr *) &nls, sizeof(struct sockaddr_nl)) < 0) {
3250 LOGERR("Bind failed\n");
3251 return;
3252 }
3253 set_select_fd();
3254 Coldboot();
3255}
3256
3257Uevent_Block_Data TWPartitionManager::get_event_block_values(char *buf, int len) {
3258 Uevent_Block_Data ret;
3259 ret.subsystem = "";
3260 char *ptr = buf;
3261 const char *end = buf + len;
3262
3263 buf[len - 1] = '\0';
3264 while (ptr < end) {
3265 if (strncmp(ptr, "ACTION=", strlen("ACTION=")) == 0) {
3266 ptr += strlen("ACTION=");
3267 ret.action = ptr;
3268 } else if (strncmp(ptr, "SUBSYSTEM=", strlen("SUBSYSTEM=")) == 0) {
3269 ptr += strlen("SUBSYSTEM=");
3270 ret.subsystem = ptr;
3271 } else if (strncmp(ptr, "DEVTYPE=", strlen("DEVTYPE=")) == 0) {
3272 ptr += strlen("DEVTYPE=");
3273 ret.type = ptr;
3274 } else if (strncmp(ptr, "DEVPATH=", strlen("DEVPATH=")) == 0) {
3275 ptr += strlen("DEVPATH=");
3276 ret.sysfs_path += ptr;
3277 } else if (strncmp(ptr, "DEVNAME=", strlen("DEVNAME=")) == 0) {
3278 ptr += strlen("DEVNAME=");
3279 ret.block_device += ptr;
3280 } else if (strncmp(ptr, "MAJOR=", strlen("MAJOR=")) == 0) {
3281 ptr += strlen("MAJOR=");
3282 ret.major = atoi(ptr);
3283 } else if (strncmp(ptr, "MINOR=", strlen("MINOR=")) == 0) {
3284 ptr += strlen("MINOR=");
3285 ret.minor = atoi(ptr);
3286 }
3287 ptr += strlen(ptr) + 1;
3288 }
3289 return ret;
3290}
3291
3292void TWPartitionManager::read_uevent() {
3293 char buf[1024];
3294
3295 int len = recv(uevent_pfd.fd, buf, sizeof(buf), MSG_DONTWAIT);
3296 if (len == -1) {
Mauronofrio Matarresec1bb76e2019-08-04 17:02:41 +01003297 LOGINFO("recv error on uevent\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003298 return;
3299 }
3300 /*int i = 0; // Print all uevent output for test /debug
3301 while (i<len) {
3302 printf("%s\n", buf+i);
3303 i += strlen(buf+i)+1;
3304 }*/
3305 Uevent_Block_Data uevent_data = get_event_block_values(buf, len);
3306 if (uevent_data.subsystem == "block" && uevent_data.type == "disk") {
3307 PartitionManager.Handle_Uevent(uevent_data);
3308 }
3309}
3310
3311void TWPartitionManager::close_uevent() {
3312 if (uevent_pfd.fd > 0)
3313 close(uevent_pfd.fd);
3314 uevent_pfd.fd = -1;
3315}
3316
3317void TWPartitionManager::Add_Partition(TWPartition* Part) {
3318 Partitions.push_back(Part);
3319}
3320
3321void TWPartitionManager::Coldboot_Scan(std::vector<string> *sysfs_entries, const string& Path, int depth) {
3322 string Real_Path = Path;
3323 char real_path[PATH_MAX];
3324 if (realpath(Path.c_str(), &real_path[0])) {
3325 string Real_Path = real_path;
3326 std::vector<string>::iterator iter;
3327 for (iter = sysfs_entries->begin(); iter != sysfs_entries->end(); iter++) {
3328 if (Real_Path.find((*iter)) != string::npos) {
3329 string Write_Path = Real_Path + "/uevent";
3330 if (TWFunc::Path_Exists(Write_Path)) {
3331 const char* write_val = "add\n";
3332 TWFunc::write_to_file(Write_Path, write_val);
3333 break;
3334 }
3335 }
3336 }
3337 }
3338
3339 DIR* d = opendir(Path.c_str());
3340 if (d != NULL) {
3341 struct dirent* de;
3342 while ((de = readdir(d)) != NULL) {
3343 if (de->d_name[0] == '.' || (de->d_type != DT_DIR && depth > 0))
3344 continue;
3345 if (strlen(de->d_name) >= 4 && (strncmp(de->d_name, "ram", 3) == 0 || strncmp(de->d_name, "loop", 4) == 0))
3346 continue;
3347
3348 string item = Path + "/";
3349 item.append(de->d_name);
3350 Coldboot_Scan(sysfs_entries, item, depth + 1);
3351 }
3352 closedir(d);
3353 }
3354}
3355
3356void TWPartitionManager::Coldboot() {
3357 std::vector<TWPartition*>::iterator iter;
3358 std::vector<string> sysfs_entries;
3359
3360 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3361 if (!(*iter)->Sysfs_Entry.empty()) {
3362 size_t wildcard_pos = (*iter)->Sysfs_Entry.find("*");
3363 if (wildcard_pos == string::npos)
3364 wildcard_pos = (*iter)->Sysfs_Entry.size();
3365 sysfs_entries.push_back((*iter)->Sysfs_Entry.substr(0, wildcard_pos));
3366 }
3367 }
3368
3369 if (sysfs_entries.size() > 0)
3370 Coldboot_Scan(&sysfs_entries, "/sys/block", 0);
3371}
Ethan Yonker53796e72019-01-11 22:49:52 -06003372
3373bool TWPartitionManager::Prepare_Empty_Folder(const std::string& Folder) {
3374 if (TWFunc::Path_Exists(Folder))
3375 TWFunc::removeDir(Folder, false);
3376 return TWFunc::Recursive_Mkdir(Folder);
3377}
3378
bigbiffcfa875c2021-06-20 16:20:27 -04003379std::string TWPartitionManager::Get_Bare_Partition_Name(std::string Mount_Point) {
3380 if (Mount_Point == "/system_root")
3381 return "system";
3382 else
3383 return TWFunc::Remove_Beginning_Slash(Mount_Point);
3384}
3385
bigbiffee7b7ff2020-03-23 15:08:27 -04003386bool TWPartitionManager::Prepare_Super_Volume(TWPartition* twrpPart) {
3387 Fstab fstab;
bigbiffcfa875c2021-06-20 16:20:27 -04003388 std::string bare_partition_name = Get_Bare_Partition_Name(twrpPart->Get_Mount_Point());
bigbiffee7b7ff2020-03-23 15:08:27 -04003389
bigbiff998f8392021-08-06 19:19:33 -04003390 Super_Partition_List.push_back(bare_partition_name);
bigbiffee7b7ff2020-03-23 15:08:27 -04003391 LOGINFO("Trying to prepare %s from super partition\n", bare_partition_name.c_str());
3392
3393 std::string blk_device_partition;
3394#ifdef AB_OTA_UPDATER
3395 blk_device_partition = bare_partition_name + PartitionManager.Get_Active_Slot_Suffix();
3396#else
3397 blk_device_partition = bare_partition_name;
3398#endif
3399
3400 FstabEntry fstabEntry = {
3401 .blk_device = blk_device_partition,
3402 .mount_point = twrpPart->Get_Mount_Point(),
3403 .fs_type = twrpPart->Current_File_System,
3404 .fs_mgr_flags.logical = twrpPart->Is_Super,
3405 };
3406
3407 fstab.emplace_back(fstabEntry);
3408 if (!fs_mgr_update_logical_partition(&fstabEntry)) {
Captain Throwbackc6939102021-04-06 12:57:30 -04003409 LOGINFO("unable to update logical partition: %s\n", twrpPart->Get_Mount_Point().c_str());
bigbiffee7b7ff2020-03-23 15:08:27 -04003410 return false;
3411 }
3412
bigbiff32cbabe2020-04-12 19:16:19 -04003413 while (access(fstabEntry.blk_device.c_str(), F_OK) != 0) {
3414 usleep(100);
3415 }
3416
bigbiffee7b7ff2020-03-23 15:08:27 -04003417 twrpPart->Set_Block_Device(fstabEntry.blk_device);
3418 twrpPart->Update_Size(true);
3419 twrpPart->Change_Mount_Read_Only(true);
3420 twrpPart->Set_Can_Be_Backed_Up(false);
3421 twrpPart->Set_Can_Be_Wiped(false);
Mohd Faraz7fc70502021-04-22 20:44:11 +02003422 LOGINFO("Symlinking %s => /dev/block/bootdevice/by-name/%s \n", fstabEntry.blk_device.c_str(), bare_partition_name.c_str());
3423 symlink(fstabEntry.blk_device.c_str(), ("/dev/block/bootdevice/by-name/" + bare_partition_name).c_str());
3424
bigbiffee7b7ff2020-03-23 15:08:27 -04003425 return true;
3426}
3427
3428bool TWPartitionManager::Prepare_All_Super_Volumes() {
3429 bool status = true;
3430 std::vector<TWPartition*>::iterator iter;
3431
3432 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3433 if ((*iter)->Is_Super) {
3434 if (!Prepare_Super_Volume(*iter)) {
3435 status = false;
3436 }
3437 PartitionManager.Output_Partition(*iter);
3438 }
3439 }
3440 Update_System_Details();
3441 return status;
3442}
3443
bigbiffee7b7ff2020-03-23 15:08:27 -04003444std::string TWPartitionManager::Get_Super_Partition() {
3445 int slot_number = Get_Active_Slot_Display() == "A" ? 0 : 1;
bigbiff998f8392021-08-06 19:19:33 -04003446 std::string super_device = fs_mgr_get_super_partition_name(slot_number);
bigbiffee7b7ff2020-03-23 15:08:27 -04003447 return "/dev/block/by-name/" + super_device;
3448}
3449
3450void TWPartitionManager::Setup_Super_Devices() {
3451 std::string superPart = Get_Super_Partition();
3452 android::fs_mgr::CreateLogicalPartitions(superPart);
bigbiff7ba75002020-04-11 20:47:09 -04003453}
3454
3455void TWPartitionManager::Setup_Super_Partition() {
bigbiffee7b7ff2020-03-23 15:08:27 -04003456 TWPartition* superPartition = new TWPartition();
bigbiff7ba75002020-04-11 20:47:09 -04003457 std::string superPart = Get_Super_Partition();
3458
3459 superPartition->Backup_Path = "/super";
3460 superPartition->Mount_Point = "/super";
bigbiffee7b7ff2020-03-23 15:08:27 -04003461 superPartition->Actual_Block_Device = superPart;
3462 superPartition->Alternate_Block_Device = superPart;
bigbiff998f8392021-08-06 19:19:33 -04003463 superPartition->Backup_Display_Name = "Super (";
3464 // Add first 4 items to fstab as logical that you would like to display in Backup_Display_Name
3465 // for the Super partition
3466 int list_size = Super_Partition_List.size();
3467 int orig_list_size = list_size;
3468 int max_display_size = 3; // total of 4 items since we start at 0
3469
3470 for (auto partition: Super_Partition_List) {
3471 superPartition->Backup_Display_Name = superPartition->Backup_Display_Name + partition;
3472 if ((orig_list_size - list_size) == max_display_size) {
3473 break;
3474 }
3475 if (list_size != 1)
3476 superPartition->Backup_Display_Name = superPartition->Backup_Display_Name + " ";
3477 list_size--;
3478 }
3479 superPartition->Backup_Display_Name += ")";
bigbiffee7b7ff2020-03-23 15:08:27 -04003480 superPartition->Can_Flash_Img = true;
bigbiffee7b7ff2020-03-23 15:08:27 -04003481 superPartition->Current_File_System = "emmc";
3482 superPartition->Can_Be_Backed_Up = true;
3483 superPartition->Is_Present = true;
3484 superPartition->Is_SubPartition = false;
bigbiff7ba75002020-04-11 20:47:09 -04003485 superPartition->Setup_Image();
bigbiffee7b7ff2020-03-23 15:08:27 -04003486 Add_Partition(superPartition);
3487 PartitionManager.Output_Partition(superPartition);
bigbiffee7b7ff2020-03-23 15:08:27 -04003488}
3489
3490bool TWPartitionManager::Get_Super_Status() {
bigbiffdf8436b2020-08-30 16:22:34 -04003491 std::string fastboot_mode = android::base::GetProperty("sys.usb.config", "");
3492 if (fastboot_mode == "fastboot") {
3493 return false;
3494 }
3495 else
3496 return access(Get_Super_Partition().c_str(), F_OK) == 0;
bigbiffee7b7ff2020-03-23 15:08:27 -04003497}
bigbiff25d25b92020-06-19 16:07:38 -04003498
3499bool TWPartitionManager::Recreate_Logs_Dir() {
3500#ifdef TW_INCLUDE_FBE
3501 struct passwd pd;
3502 struct passwd *pwdptr = &pd;
3503 struct passwd *tempPd;
3504 char pwdBuf[512];
3505 int uid = 0, gid = 0;
3506
3507 if ((getpwnam_r("system", pwdptr, pwdBuf, sizeof(pwdBuf), &tempPd)) != 0) {
3508 LOGERR("unable to get system user id\n");
3509 return false;
3510 } else {
3511 struct group grp;
3512 struct group *grpptr = &grp;
3513 struct group *tempGrp;
3514 char grpBuf[512];
3515
3516 if ((getgrnam_r("cache", grpptr, grpBuf, sizeof(grpBuf), &tempGrp)) != 0) {
3517 LOGERR("unable to get cache group id\n");
3518 return false;
3519 } else {
3520 uid = pd.pw_uid;
3521 gid = grp.gr_gid;
3522 std::string abLogsRecoveryDir(DATA_LOGS_DIR);
3523 abLogsRecoveryDir += "/recovery/";
3524
3525 if (!TWFunc::Create_Dir_Recursive(abLogsRecoveryDir, S_IRWXU | S_IRWXG | S_IWGRP | S_IXGRP, uid, gid)) {
3526 LOGERR("Unable to recreate %s\n", abLogsRecoveryDir.c_str());
3527 return false;
3528 }
3529 if (setfilecon(abLogsRecoveryDir.c_str(), "u:object_r:cache_file:s0") != 0) {
3530 LOGERR("Unable to set contexts for %s\n", abLogsRecoveryDir.c_str());
3531 return false;
3532 }
3533 }
3534 }
3535#endif
3536 return true;
3537}
bigbiff1f9e4842020-10-31 11:33:15 -04003538
3539void TWPartitionManager::Unlock_Block_Partitions() {
3540 int fd, OFF = 0;
3541
3542 const std::string block_path = "/dev/block/";
3543 DIR* d = opendir(block_path.c_str());
3544 if (d != NULL) {
3545 struct dirent* de;
3546 while ((de = readdir(d)) != NULL) {
3547 if (de->d_type == DT_BLK) {
3548 std::string block_device = block_path + de->d_name;
bigbiff1f9e4842020-10-31 11:33:15 -04003549 if ((fd = open(block_device.c_str(), O_RDONLY | O_CLOEXEC)) < 0) {
3550 LOGERR("unable to open block device %s: %s\n", block_device.c_str(), strerror(errno));
3551 continue;
3552 }
3553 if (ioctl(fd, BLKROSET, &OFF) == -1) {
3554 LOGERR("Unable to unlock %s for flashing: %s\n", block_device.c_str());
3555 continue;
3556 }
3557 close(fd);
3558 }
3559 }
3560 closedir(d);
3561 }
bigbiffaf253cd2020-12-06 16:50:55 -05003562}
bigbiffcfa875c2021-06-20 16:20:27 -04003563
3564bool TWPartitionManager::Unmap_Super_Devices() {
3565 bool destroyed = false;
3566#ifndef TW_EXCLUDE_APEX
3567 twrpApex apex;
3568 apex.Unmount();
3569#endif
3570 for (auto iter = Partitions.begin(); iter != Partitions.end();) {
3571 LOGINFO("Checking partition: %s\n", (*iter)->Get_Mount_Point().c_str());
3572 if ((*iter)->Is_Super) {
3573 TWPartition *part = *iter;
3574 std::string bare_partition_name = Get_Bare_Partition_Name((*iter)->Get_Mount_Point());
3575 std::string blk_device_partition = bare_partition_name + PartitionManager.Get_Active_Slot_Suffix();
3576 (*iter)->UnMount(false);
3577 LOGINFO("removing dynamic partition: %s\n", blk_device_partition.c_str());
3578 destroyed = DestroyLogicalPartition(blk_device_partition);
3579 std::string cow_partition = blk_device_partition + "-cow";
3580 std::string cow_partition_path = "/dev/block/mapper/" + cow_partition;
3581 struct stat st;
3582 if (lstat(cow_partition_path.c_str(), &st) == 0) {
3583 LOGINFO("removing cow partition: %s\n", cow_partition.c_str());
3584 destroyed = DestroyLogicalPartition(cow_partition);
3585 }
bigbiffcfa875c2021-06-20 16:20:27 -04003586 iter = Partitions.erase(iter);
3587 delete part;
3588 if (!destroyed) {
3589 return false;
3590 }
3591 } else {
3592 ++iter;
3593 }
3594 }
3595 return true;
3596}
bigbiffd21252f2021-09-18 15:56:32 -04003597
3598
3599bool TWPartitionManager::Check_Pending_Merges() {
3600 auto sm = android::snapshot::SnapshotManager::NewForFirstStageMount();
3601 if (!sm) {
3602 LOGERR("Unable to call snapshot manager\n");
3603 return false;
3604 }
3605
3606 if (!Unmap_Super_Devices()) {
3607 LOGERR("Unable to unmap dynamic partitions.\n");
3608 return false;
3609 }
3610
3611 auto callback = [&]() -> void {
3612 double progress;
3613 sm->GetUpdateState(&progress);
3614 LOGINFO("waiting for merge to complete: %.2f\n", progress);
3615 };
3616
3617 LOGINFO("checking for merges\n");
3618 if (!sm->HandleImminentDataWipe(callback)) {
3619 LOGERR("Unable to check merge status\n");
3620 return false;
3621 }
3622 return true;
zhenyolka90850472021-11-08 19:05:30 +03003623}