blob: d41eac331e881f8cc014c075477dbe9e1074bb77 [file] [log] [blame]
Dees Troy3be70a82013-10-22 14:25:12 +00001/*
bigbiffee7b7ff2020-03-23 15:08:27 -04002 Copyright 2014 to 2020 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
bigbiffee7b7ff2020-03-23 15:08:27 -040040
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060041#include <sys/poll.h>
42#include <sys/socket.h>
43#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>
55#include <libgsi/libgsi.h>
56#include <liblp/liblp.h>
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060057
Dees_Troy51a0e822012-09-05 15:24:24 -040058#include "variables.h"
Dees_Troy2673cec2013-04-02 20:22:16 +000059#include "twcommon.h"
Dees_Troy51a0e822012-09-05 15:24:24 -040060#include "partitions.hpp"
Dees_Troy5bf43922012-09-07 16:07:55 -040061#include "data.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040062#include "twrp-functions.hpp"
Ethan Yonkerb5fab762016-01-28 14:03:33 -060063#include "fixContexts.hpp"
Ethan Yonker3fdcda42016-11-30 12:29:37 -060064#include "exclude.hpp"
Ethan Yonker4b94cfd2014-12-11 10:00:45 -060065#include "set_metadata.h"
bigbiff7abc5fe2015-01-17 16:53:12 -050066#include "tw_atomic.hpp"
Ethan Yonker74db1572015-10-28 12:44:49 -050067#include "gui/gui.hpp"
Ethan Yonker472f5062016-02-25 13:47:30 -060068#include "progresstracking.hpp"
bigbiff bigbiff56cf5642016-08-19 17:43:45 -040069#include "twrpDigestDriver.hpp"
bigbiffad58e1b2020-07-06 20:24:34 -040070#include "twrpRepacker.hpp"
bigbiffce8f83c2015-12-12 18:30:21 -050071#include "adbbu/libtwadbbu.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040072
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040073#ifdef TW_HAS_MTP
bigbiff bigbiffaf32bb92018-12-18 18:39:53 -050074#ifdef TW_HAS_LEGACY_MTP
75#include "mtp/legacy/mtp_MtpServer.hpp"
76#include "mtp/legacy/twrpMtp.hpp"
77#include "mtp/legacy/MtpMessage.hpp"
78#else
79#include "mtp/ffs/mtp_MtpServer.hpp"
80#include "mtp/ffs/twrpMtp.hpp"
81#include "mtp/ffs/MtpMessage.hpp"
82#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040083#endif
84
Dees Troy6f6441d2014-01-23 02:07:03 +000085extern "C" {
86 #include "cutils/properties.h"
Ethan Yonker74db1572015-10-28 12:44:49 -050087 #include "gui/gui.h"
Dees Troy6f6441d2014-01-23 02:07:03 +000088}
89
Dees_Troy5bf43922012-09-07 16:07:55 -040090#ifdef TW_INCLUDE_CRYPTO
bigbiff7ba75002020-04-11 20:47:09 -040091#include "crypto/fde/cryptfs.h"
92#include "gui/rapidxml.hpp"
93#include "gui/pages.hpp"
94#ifdef TW_INCLUDE_FBE
95#include "crypto/ext4crypt/Decrypt.h"
96#ifdef TW_INCLUDE_FBE_METADATA_DECRYPT
97 #ifdef USE_FSCRYPT
98 #include "crypto/fscrypt/MetadataCrypt.h"
99 #else
100 #include "crypto/ext4crypt/MetadataCrypt.h"
bigbiffadc599e2020-05-28 19:36:30 +0000101 #endif
bigbiff7ba75002020-04-11 20:47:09 -0400102#endif
103#endif
104#ifdef TW_CRYPTO_USE_SYSTEM_VOLD
105#include "crypto/vold_decrypt/vold_decrypt.h"
106#endif
Dees_Troy5bf43922012-09-07 16:07:55 -0400107#endif
Dees_Troy51a0e822012-09-05 15:24:24 -0400108
Ethan Yonker1b190162016-12-05 15:25:19 -0600109#ifdef AB_OTA_UPDATER
110#include <hardware/hardware.h>
111#include <hardware/boot_control.h>
112#endif
113
bigbiffee7b7ff2020-03-23 15:08:27 -0400114using android::fs_mgr::Fstab;
115using android::fs_mgr::FstabEntry;
Ethan Yonker6277c792014-09-15 14:54:30 -0500116
bigbiffee7b7ff2020-03-23 15:08:27 -0400117extern bool datamedia;
Noah Jacobson81d638d2019-04-28 00:10:07 -0400118std::vector<users_struct> Users_List;
119
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500120TWPartitionManager::TWPartitionManager(void) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400121 mtp_was_enabled = false;
Ethan Yonker726a0202014-12-16 20:01:38 -0600122 mtp_write_fd = -1;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600123 uevent_pfd.fd = -1;
bigbiff7abc5fe2015-01-17 16:53:12 -0500124 stop_backup.set_value(0);
Ethan Yonker1b190162016-12-05 15:25:19 -0600125#ifdef AB_OTA_UPDATER
126 char slot_suffix[PROPERTY_VALUE_MAX];
Ethan Yonkerfefe5912017-09-30 22:22:13 -0500127 property_get("ro.boot.slot_suffix", slot_suffix, "error");
128 if (strcmp(slot_suffix, "error") == 0)
129 property_get("ro.boot.slot", slot_suffix, "error");
Ethan Yonker1b190162016-12-05 15:25:19 -0600130 Active_Slot_Display = "";
Ethan Yonkerfefe5912017-09-30 22:22:13 -0500131 if (strcmp(slot_suffix, "_a") == 0 || strcmp(slot_suffix, "a") == 0)
Ethan Yonker1b190162016-12-05 15:25:19 -0600132 Set_Active_Slot("A");
133 else
134 Set_Active_Slot("B");
135#endif
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500136}
137
Captain Throwback18aea272020-10-29 12:43:17 -0400138int TWPartitionManager::Set_FDE_Encrypt_Status(void) {
139 property_set("ro.crypto.state", "encrypted");
140 property_set("ro.crypto.type", "block");
141 // Sleep for a bit so that services can start if needed
142 sleep(1);
143 return 0;
144}
145
bigbiff8da46fa2020-09-08 16:42:34 -0400146int TWPartitionManager::Process_Fstab(string Fstab_Filename, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400147 FILE *fstabFile;
148 char fstab_line[MAX_FSTAB_LINE_LENGTH];
Dees Troy02a64532014-03-19 15:23:32 +0000149 TWPartition* settings_partition = NULL;
Matt Mowerbf4efa32014-04-14 23:25:26 -0500150 TWPartition* andsec_partition = NULL;
Ethan Yonker726a0202014-12-16 20:01:38 -0600151 unsigned int storageid = 1 << 16; // upper 16 bits are for physical storage device, we pretend to have only one
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600152 std::map<string, Flags_Map> twrp_flags;
153
154 fstabFile = fopen("/etc/twrp.flags", "rt");
155 if (fstabFile != NULL) {
Ian Macdonald160e8d32020-09-29 20:47:31 +0200156 LOGINFO("Reading /etc/twrp.flags\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600157 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
158 if (fstab_line[0] != '/')
159 continue;
160
161 size_t line_size = strlen(fstab_line);
162 if (fstab_line[line_size - 1] != '\n')
163 fstab_line[line_size] = '\n';
164 Flags_Map line_flags;
165 line_flags.Primary_Block_Device = "";
166 line_flags.Alternate_Block_Device = "";
167 line_flags.fstab_line = (char*)malloc(MAX_FSTAB_LINE_LENGTH);
168 if (!line_flags.fstab_line) {
169 LOGERR("malloc error on line_flags.fstab_line\n");
170 return false;
171 }
172 memcpy(line_flags.fstab_line, fstab_line, MAX_FSTAB_LINE_LENGTH);
173 bool found_separator = false;
174 char *fs_loc = NULL;
175 char *block_loc = NULL;
176 char *flags_loc = NULL;
177 size_t index, item_index = 0;
178 for (index = 0; index < line_size; index++) {
179 if (fstab_line[index] <= 32) {
180 fstab_line[index] = '\0';
181 found_separator = true;
182 } else if (found_separator) {
183 if (item_index == 0) {
184 fs_loc = fstab_line + index;
185 } else if (item_index == 1) {
186 block_loc = fstab_line + index;
187 } else if (item_index > 1) {
188 char *ptr = fstab_line + index;
189 if (*ptr == '/') {
190 line_flags.Alternate_Block_Device = ptr;
191 } else if (strlen(ptr) > strlen("flags=") && strncmp(ptr, "flags=", strlen("flags=")) == 0) {
192 flags_loc = ptr;
193 // Once we find the flags=, we're done scanning the line
194 break;
195 }
196 }
197 found_separator = false;
198 item_index++;
199 }
200 }
201 if (block_loc)
202 line_flags.Primary_Block_Device = block_loc;
203 if (fs_loc)
204 line_flags.File_System = fs_loc;
205 if (flags_loc)
206 line_flags.Flags = flags_loc;
207 string Mount_Point = fstab_line;
208 twrp_flags[Mount_Point] = line_flags;
209 memset(fstab_line, 0, sizeof(fstab_line));
210 }
211 fclose(fstabFile);
212 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400213
214 fstabFile = fopen(Fstab_Filename.c_str(), "rt");
215 if (fstabFile == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000216 LOGERR("Critical Error: Unable to open fstab at '%s'.\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400217 return false;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600218 } else
219 LOGINFO("Reading %s\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400220
221 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
bigbiffee7b7ff2020-03-23 15:08:27 -0400222 bool isSuper = Is_Super_Partition(fstab_line);
223
224 if (!isSuper && fstab_line[0] != '/')
Dees_Troy5bf43922012-09-07 16:07:55 -0400225 continue;
226
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600227 if (strstr(fstab_line, "swap"))
228 continue; // Skip swap in recovery
229
230 size_t line_size = strlen(fstab_line);
231 if (fstab_line[line_size - 1] != '\n')
232 fstab_line[line_size] = '\n';
Dees_Troy2a923582012-09-20 12:13:34 -0400233
Matt Mower2b2dd152016-04-26 11:24:08 -0500234 TWPartition* partition = new TWPartition();
bigbiff8da46fa2020-09-08 16:42:34 -0400235 if (partition->Process_Fstab_Line(fstab_line, Display_Error, &twrp_flags))
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500236 Partitions.push_back(partition);
Matt Mower72c87ce2016-04-26 14:34:56 -0500237 else
Dees_Troy5bf43922012-09-07 16:07:55 -0400238 delete partition;
Matt Mower2b2dd152016-04-26 11:24:08 -0500239
240 memset(fstab_line, 0, sizeof(fstab_line));
Dees_Troy5bf43922012-09-07 16:07:55 -0400241 }
242 fclose(fstabFile);
Matt Mower72c87ce2016-04-26 14:34:56 -0500243
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600244 if (twrp_flags.size() > 0) {
245 LOGINFO("Processing remaining twrp.flags\n");
246 // Add any items from twrp.flags that did not exist in the recovery.fstab
247 for (std::map<string, Flags_Map>::iterator mapit=twrp_flags.begin(); mapit!=twrp_flags.end(); mapit++) {
248 if (Find_Partition_By_Path(mapit->first) == NULL) {
249 TWPartition* partition = new TWPartition();
bigbiff8da46fa2020-09-08 16:42:34 -0400250 if (partition->Process_Fstab_Line(mapit->second.fstab_line, Display_Error, NULL))
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600251 Partitions.push_back(partition);
252 else
253 delete partition;
254 }
255 if (mapit->second.fstab_line)
256 free(mapit->second.fstab_line);
257 mapit->second.fstab_line = NULL;
258 }
259 }
Mohd Faraze3948ec2020-08-14 01:40:59 +0000260 if (Get_Super_Status()) {
261 Setup_Super_Devices();
262 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600263 LOGINFO("Done processing fstab files\n");
264
Matt Mower72c87ce2016-04-26 14:34:56 -0500265 std::vector<TWPartition*>::iterator iter;
266 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
267 (*iter)->Partition_Post_Processing(Display_Error);
268
269 if ((*iter)->Is_Storage) {
270 ++storageid;
271 (*iter)->MTP_Storage_ID = storageid;
272 }
273
274 if (!settings_partition && (*iter)->Is_Settings_Storage && (*iter)->Is_Present)
275 settings_partition = (*iter);
276 else
277 (*iter)->Is_Settings_Storage = false;
278
279 if (!andsec_partition && (*iter)->Has_Android_Secure && (*iter)->Is_Present)
280 andsec_partition = (*iter);
281 else
282 (*iter)->Has_Android_Secure = false;
bigbiffdf8436b2020-08-30 16:22:34 -0400283
Mohd Faraze3948ec2020-08-14 01:40:59 +0000284 if (Is_Super_Partition(TWFunc::Remove_Beginning_Slash((*iter)->Get_Mount_Point()).c_str()))
285 Prepare_Super_Volume((*iter));
286 }
287
288 //Setup Apex before decryption
289 TWPartition* sys = PartitionManager.Find_Partition_By_Path(PartitionManager.Get_Android_Root_Path());
290 TWPartition* ven = PartitionManager.Find_Partition_By_Path("/vendor");
291 if (sys) {
292 if (sys->Get_Super_Status()) {
293 sys->Mount(true);
294 if (ven) {
295 ven->Mount(true);
296 }
297 twrpApex apex;
298 if (!apex.loadApexImages()) {
299 LOGERR("Unable to load apex images from %s\n", APEX_DIR);
300 property_set("twrp.apex.loaded", "false");
301 } else {
302 property_set("twrp.apex.loaded", "true");
303 }
Mohd Farazd9e66712020-10-28 20:42:57 +0530304 TWFunc::check_and_run_script("/sbin/resyncapex.sh", "apex");
Mohd Faraze3948ec2020-08-14 01:40:59 +0000305 }
Matt Mower72c87ce2016-04-26 14:34:56 -0500306 }
Mohd Faraz9a4ef262020-10-18 23:58:21 +0530307#ifndef USE_VENDOR_LIBS
308 if (ven)
309 ven->UnMount(true);
310 if (sys)
311 sys->UnMount(true);
312#endif
Matt Mower72c87ce2016-04-26 14:34:56 -0500313
Ethan Yonker6277c792014-09-15 14:54:30 -0500314 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) {
315 // Attempt to automatically identify /data/media emulated storage devices
316 TWPartition* Dat = Find_Partition_By_Path("/data");
317 if (Dat) {
318 LOGINFO("Using automatic handling for /data/media emulated storage device.\n");
319 datamedia = true;
that9e0593e2014-10-08 00:01:24 +0200320 Dat->Setup_Data_Media();
Ethan Yonker6277c792014-09-15 14:54:30 -0500321 settings_partition = Dat;
Ethan Yonker726a0202014-12-16 20:01:38 -0600322 // Since /data was not considered a storage partition earlier, we still need to assign an MTP ID
323 ++storageid;
324 Dat->MTP_Storage_ID = storageid;
Ethan Yonker6277c792014-09-15 14:54:30 -0500325 }
326 }
Dees Troy02a64532014-03-19 15:23:32 +0000327 if (!settings_partition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500328 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
329 if ((*iter)->Is_Storage) {
Dees Troy02a64532014-03-19 15:23:32 +0000330 settings_partition = (*iter);
Dees_Troya13d74f2013-03-24 08:54:55 -0500331 break;
332 }
333 }
Dees Troy02a64532014-03-19 15:23:32 +0000334 if (!settings_partition)
Dees_Troy2673cec2013-04-02 20:22:16 +0000335 LOGERR("Unable to locate storage partition for storing settings file.\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500336 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400337 if (!Write_Fstab()) {
338 if (Display_Error)
Dees_Troy2673cec2013-04-02 20:22:16 +0000339 LOGERR("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400340 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000341 LOGINFO("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400342 }
Matt Mowered71fa32014-04-16 13:21:47 -0500343
Matt Mowerbf4efa32014-04-14 23:25:26 -0500344 if (andsec_partition) {
345 Setup_Android_Secure_Location(andsec_partition);
Matt Mowered71fa32014-04-16 13:21:47 -0500346 } else if (settings_partition) {
Matt Mowerbf4efa32014-04-14 23:25:26 -0500347 Setup_Android_Secure_Location(settings_partition);
348 }
Matt Mowered71fa32014-04-16 13:21:47 -0500349 if (settings_partition) {
350 Setup_Settings_Storage_Partition(settings_partition);
351 }
bigbiff7ba75002020-04-11 20:47:09 -0400352
Mohd Faraze3948ec2020-08-14 01:40:59 +0000353#ifdef TW_INCLUDE_CRYPTO
354 DataManager::SetValue(TW_IS_ENCRYPTED, 1);
355 Decrypt_Data();
356#endif
357
Mohd Faraze3948ec2020-08-14 01:40:59 +0000358 if (Get_Super_Status())
359 Setup_Super_Partition();
bigbiff7ba75002020-04-11 20:47:09 -0400360 UnMount_Main_Partitions();
361#ifdef AB_OTA_UPDATER
362 DataManager::SetValue("tw_active_slot", Get_Active_Slot_Display());
363#endif
364 setup_uevent();
365 return true;
366}
367
368int TWPartitionManager::Write_Fstab(void) {
369 FILE *fp;
370 std::vector<TWPartition*>::iterator iter;
371 string Line;
372
373 fp = fopen("/etc/fstab", "w");
374 if (fp == NULL) {
375 LOGINFO("Can not open /etc/fstab.\n");
376 return false;
377 }
378 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
379 if ((*iter)->Can_Be_Mounted) {
Mohd Faraz2a1e3752020-09-30 10:55:48 +0530380 Line = (*iter)->Actual_Block_Device + " " + (*iter)->Mount_Point + " " + (*iter)->Current_File_System +
381 ((*iter)->Mount_Read_Only ? " ro " : " rw ") + "0 0\n";
bigbiff7ba75002020-04-11 20:47:09 -0400382 fputs(Line.c_str(), fp);
383 }
384 // Handle subpartition tracking
385 if ((*iter)->Is_SubPartition) {
386 TWPartition* ParentPartition = Find_Partition_By_Path((*iter)->SubPartition_Of);
387 if (ParentPartition)
388 ParentPartition->Has_SubPartition = true;
389 else
390 LOGERR("Unable to locate parent partition '%s' of '%s'\n", (*iter)->SubPartition_Of.c_str(), (*iter)->Mount_Point.c_str());
391 }
392 }
393 fclose(fp);
394 return true;
395}
396
397void TWPartitionManager::Decrypt_Data() {
398 #ifdef TW_INCLUDE_CRYPTO
Ethan Yonkercceebb82014-11-18 10:17:59 -0600399 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
400 if (Decrypt_Data && Decrypt_Data->Is_Encrypted && !Decrypt_Data->Is_Decrypted) {
nebrassyf74255e2020-10-24 21:08:20 +0200401 property_set("ro.crypto.state", "encrypted");
Ethan Yonker93382822018-11-01 15:25:31 -0500402 if (!Decrypt_Data->Key_Directory.empty() && Mount_By_Path(Decrypt_Data->Key_Directory, false)) {
nebrassyf74255e2020-10-24 21:08:20 +0200403 property_set("ro.crypto.type", "file");
Ethan Yonker93382822018-11-01 15:25:31 -0500404#ifdef TW_INCLUDE_FBE_METADATA_DECRYPT
bigbiff7ba75002020-04-11 20:47:09 -0400405#ifdef USE_FSCRYPT
406 if (fscrypt_mount_metadata_encrypted(Decrypt_Data->Actual_Block_Device, Decrypt_Data->Mount_Point, false)) {
Mohd Faraze3948ec2020-08-14 01:40:59 +0000407 std::string crypto_blkdev =android::base::GetProperty("ro.crypto.fs_crypto_blkdev", "error");
408 Decrypt_Data->Decrypted_Block_Device = crypto_blkdev;
bigbiff7ba75002020-04-11 20:47:09 -0400409 LOGINFO("Successfully decrypted metadata encrypted data partition with new block device: '%s'\n", crypto_blkdev.c_str());
410#else
Ethan Yonker93382822018-11-01 15:25:31 -0500411 if (e4crypt_mount_metadata_encrypted(Decrypt_Data->Mount_Point, false, Decrypt_Data->Key_Directory, Decrypt_Data->Actual_Block_Device, &Decrypt_Data->Decrypted_Block_Device)) {
bigbiff7ba75002020-04-11 20:47:09 -0400412 LOGINFO("Successfully decrypted metadata encrypted data partition with new block device: '%s'\n",
413 Decrypt_Data->Decrypted_Block_Device.c_str());
414#endif
Ethan Yonker93382822018-11-01 15:25:31 -0500415 Decrypt_Data->Is_Decrypted = true; // Needed to make the mount function work correctly
416 int retry_count = 10;
417 while (!Decrypt_Data->Mount(false) && --retry_count)
418 usleep(500);
419 if (Decrypt_Data->Mount(false)) {
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400420 if (!Decrypt_Data->Decrypt_FBE_DE()) {
mauronofriofc79aa62019-11-23 23:13:04 +0100421 char wrappedvalue[PROPERTY_VALUE_MAX];
422 property_get("fbe.data.wrappedkey", wrappedvalue, "");
423 std::string wrappedkeyvalue(wrappedvalue);
424 if (wrappedkeyvalue == "true") {
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400425 LOGERR("Unable to decrypt FBE device\n");
mauronofriofc79aa62019-11-23 23:13:04 +0100426 } else {
427 LOGINFO("Trying wrapped key.\n");
428 property_set("fbe.data.wrappedkey", "true");
429 if (!Decrypt_Data->Decrypt_FBE_DE()) {
430 LOGERR("Unable to decrypt FBE device\n");
431 }
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400432 }
433 }
434
Ethan Yonker93382822018-11-01 15:25:31 -0500435 } else {
436 LOGINFO("Failed to mount data after metadata decrypt\n");
437 }
438 } else {
439 LOGINFO("Unable to decrypt metadata encryption\n");
440 }
441#else
442 LOGERR("Metadata FBE decrypt support not present in this TWRP\n");
443#endif
444 }
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600445 if (Decrypt_Data->Is_FBE) {
446 if (DataManager::GetIntValue(TW_CRYPTO_PWTYPE) == 0) {
447 if (Decrypt_Device("!") == 0) {
448 gui_msg("decrypt_success=Successfully decrypted with default password.");
449 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
450 } else {
451 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
452 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600453 }
454 } else {
Captain Throwback18aea272020-10-29 12:43:17 -0400455 Set_FDE_Encrypt_Status();
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600456 int password_type = cryptfs_get_password_type();
457 if (password_type == CRYPT_TYPE_DEFAULT) {
458 LOGINFO("Device is encrypted with the default password, attempting to decrypt.\n");
459 if (Decrypt_Device("default_password") == 0) {
460 gui_msg("decrypt_success=Successfully decrypted with default password.");
461 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
462 } else {
463 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
464 }
465 } else {
466 DataManager::SetValue("TW_CRYPTO_TYPE", password_type);
Noah Jacobson81d638d2019-04-28 00:10:07 -0400467 DataManager::SetValue("tw_crypto_pwtype_0", password_type);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600468 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600469 }
470 }
Noah Jacobson81d638d2019-04-28 00:10:07 -0400471 if (Decrypt_Data && (!Decrypt_Data->Is_Encrypted || Decrypt_Data->Is_Decrypted) &&
472 Decrypt_Data->Mount(false)) {
Ethan Yonker66a19492015-12-10 10:19:45 -0600473 Decrypt_Adopted();
474 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600475#endif
Dees_Troy51a0e822012-09-05 15:24:24 -0400476}
477
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500478void TWPartitionManager::Setup_Settings_Storage_Partition(TWPartition* Part) {
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500479 DataManager::SetValue("tw_settings_path", Part->Storage_Path);
480 DataManager::SetValue("tw_storage_path", Part->Storage_Path);
481 LOGINFO("Settings storage is '%s'\n", Part->Storage_Path.c_str());
482}
483
Matt Mowerbf4efa32014-04-14 23:25:26 -0500484void TWPartitionManager::Setup_Android_Secure_Location(TWPartition* Part) {
485 if (Part->Has_Android_Secure)
486 Part->Setup_AndSec();
Ethan Yonker6277c792014-09-15 14:54:30 -0500487 else if (!datamedia)
Matt Mowerbf4efa32014-04-14 23:25:26 -0500488 Part->Setup_AndSec();
Matt Mowerbf4efa32014-04-14 23:25:26 -0500489}
490
Dees_Troy8170a922012-09-18 15:40:25 -0400491void TWPartitionManager::Output_Partition_Logging(void) {
492 std::vector<TWPartition*>::iterator iter;
493
494 printf("\n\nPartition Logs:\n");
495 for (iter = Partitions.begin(); iter != Partitions.end(); iter++)
496 Output_Partition((*iter));
497}
498
499void TWPartitionManager::Output_Partition(TWPartition* Part) {
500 unsigned long long mb = 1048576;
501
Gary Peck004d48b2012-11-21 16:28:18 -0800502 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 -0400503 if (Part->Can_Be_Mounted) {
Gary Peck004d48b2012-11-21 16:28:18 -0800504 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 -0400505 }
Gary Peck004d48b2012-11-21 16:28:18 -0800506 printf("\n Flags: ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500507 if (Part->Can_Be_Mounted)
508 printf("Can_Be_Mounted ");
Gary Peck004d48b2012-11-21 16:28:18 -0800509 if (Part->Can_Be_Wiped)
510 printf("Can_Be_Wiped ");
Hashcodedabfd492013-08-29 22:45:30 -0700511 if (Part->Use_Rm_Rf)
512 printf("Use_Rm_Rf ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500513 if (Part->Can_Be_Backed_Up)
514 printf("Can_Be_Backed_Up ");
Gary Peck004d48b2012-11-21 16:28:18 -0800515 if (Part->Wipe_During_Factory_Reset)
516 printf("Wipe_During_Factory_Reset ");
517 if (Part->Wipe_Available_in_GUI)
518 printf("Wipe_Available_in_GUI ");
519 if (Part->Is_SubPartition)
520 printf("Is_SubPartition ");
521 if (Part->Has_SubPartition)
522 printf("Has_SubPartition ");
523 if (Part->Removable)
524 printf("Removable ");
525 if (Part->Is_Present)
526 printf("IsPresent ");
527 if (Part->Can_Be_Encrypted)
528 printf("Can_Be_Encrypted ");
529 if (Part->Is_Encrypted)
530 printf("Is_Encrypted ");
531 if (Part->Is_Decrypted)
532 printf("Is_Decrypted ");
533 if (Part->Has_Data_Media)
534 printf("Has_Data_Media ");
Dees_Troy83bd4832013-05-04 12:39:56 +0000535 if (Part->Can_Encrypt_Backup)
536 printf("Can_Encrypt_Backup ");
537 if (Part->Use_Userdata_Encryption)
538 printf("Use_Userdata_Encryption ");
Gary Peck004d48b2012-11-21 16:28:18 -0800539 if (Part->Has_Android_Secure)
540 printf("Has_Android_Secure ");
541 if (Part->Is_Storage)
542 printf("Is_Storage ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500543 if (Part->Is_Settings_Storage)
544 printf("Is_Settings_Storage ");
Dees_Troy68cab492012-12-12 19:29:35 +0000545 if (Part->Ignore_Blkid)
546 printf("Ignore_Blkid ");
Ethan Yonker253368a2014-11-25 15:00:52 -0600547 if (Part->Mount_To_Decrypt)
548 printf("Mount_To_Decrypt ");
Ethan Yonker96af84a2015-01-05 14:58:36 -0600549 if (Part->Can_Flash_Img)
550 printf("Can_Flash_Img ");
Ethan Yonker66a19492015-12-10 10:19:45 -0600551 if (Part->Is_Adopted_Storage)
552 printf("Is_Adopted_Storage ");
Ethan Yonker1b190162016-12-05 15:25:19 -0600553 if (Part->SlotSelect)
554 printf("SlotSelect ");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600555 if (Part->Mount_Read_Only)
556 printf("Mount_Read_Only ");
Gary Peck004d48b2012-11-21 16:28:18 -0800557 printf("\n");
558 if (!Part->SubPartition_Of.empty())
559 printf(" SubPartition_Of: %s\n", Part->SubPartition_Of.c_str());
560 if (!Part->Symlink_Path.empty())
561 printf(" Symlink_Path: %s\n", Part->Symlink_Path.c_str());
562 if (!Part->Symlink_Mount_Point.empty())
563 printf(" Symlink_Mount_Point: %s\n", Part->Symlink_Mount_Point.c_str());
564 if (!Part->Primary_Block_Device.empty())
565 printf(" Primary_Block_Device: %s\n", Part->Primary_Block_Device.c_str());
566 if (!Part->Alternate_Block_Device.empty())
567 printf(" Alternate_Block_Device: %s\n", Part->Alternate_Block_Device.c_str());
568 if (!Part->Decrypted_Block_Device.empty())
569 printf(" Decrypted_Block_Device: %s\n", Part->Decrypted_Block_Device.c_str());
dianlujitao4879b372018-12-03 18:45:47 +0800570 if (!Part->Crypto_Key_Location.empty())
Ethan Yonker253368a2014-11-25 15:00:52 -0600571 printf(" Crypto_Key_Location: %s\n", Part->Crypto_Key_Location.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800572 if (Part->Length != 0)
573 printf(" Length: %i\n", Part->Length);
574 if (!Part->Display_Name.empty())
575 printf(" Display_Name: %s\n", Part->Display_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500576 if (!Part->Storage_Name.empty())
577 printf(" Storage_Name: %s\n", Part->Storage_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800578 if (!Part->Backup_Path.empty())
579 printf(" Backup_Path: %s\n", Part->Backup_Path.c_str());
580 if (!Part->Backup_Name.empty())
581 printf(" Backup_Name: %s\n", Part->Backup_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500582 if (!Part->Backup_Display_Name.empty())
583 printf(" Backup_Display_Name: %s\n", Part->Backup_Display_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800584 if (!Part->Backup_FileName.empty())
585 printf(" Backup_FileName: %s\n", Part->Backup_FileName.c_str());
586 if (!Part->Storage_Path.empty())
587 printf(" Storage_Path: %s\n", Part->Storage_Path.c_str());
588 if (!Part->Current_File_System.empty())
589 printf(" Current_File_System: %s\n", Part->Current_File_System.c_str());
590 if (!Part->Fstab_File_System.empty())
591 printf(" Fstab_File_System: %s\n", Part->Fstab_File_System.c_str());
592 if (Part->Format_Block_Size != 0)
Ethan Yonkerc798c9c2015-10-09 11:15:26 -0500593 printf(" Format_Block_Size: %lu\n", Part->Format_Block_Size);
Dees_Troy094207a2012-09-26 12:00:39 -0400594 if (!Part->MTD_Name.empty())
595 printf(" MTD_Name: %s\n", Part->MTD_Name.c_str());
Matt Mower029a82d2017-01-08 13:12:38 -0600596 printf(" Backup_Method: %s\n", Part->Backup_Method_By_Name().c_str());
Hashcode62bd9e02013-11-19 21:59:42 -0800597 if (Part->Mount_Flags || !Part->Mount_Options.empty())
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600598 printf(" Mount_Flags: %i, Mount_Options: %s\n", Part->Mount_Flags, Part->Mount_Options.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -0600599 if (Part->MTP_Storage_ID)
600 printf(" MTP_Storage_ID: %i\n", Part->MTP_Storage_ID);
Ethan Yonker93382822018-11-01 15:25:31 -0500601 if (!Part->Key_Directory.empty())
602 printf(" Metadata Key Directory: %s\n", Part->Key_Directory.c_str());
Ethan Yonker6277c792014-09-15 14:54:30 -0500603 printf("\n");
Dees_Troy8170a922012-09-18 15:40:25 -0400604}
605
Dees_Troy51a0e822012-09-05 15:24:24 -0400606int TWPartitionManager::Mount_By_Path(string Path, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400607 std::vector<TWPartition*>::iterator iter;
608 int ret = false;
609 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400610 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400611
bigbiffd58ba182020-03-23 10:02:29 -0400612 if (Local_Path == "/tmp" || Local_Path == "/")
Dees_Troy43d8b002012-09-17 16:00:01 -0400613 return true;
614
Dees_Troy5bf43922012-09-07 16:07:55 -0400615 // Iterate through all partitions
616 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400617 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400618 ret = (*iter)->Mount(Display_Error);
619 found = true;
Dees_Troy51127312012-09-08 13:08:49 -0400620 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400621 (*iter)->Mount(Display_Error);
Dees_Troy51127312012-09-08 13:08:49 -0400622 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400623 }
624 if (found) {
625 return ret;
626 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500627 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 -0400628 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000629 LOGINFO("Mount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400630 }
631 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400632}
633
Dees_Troy51a0e822012-09-05 15:24:24 -0400634int TWPartitionManager::UnMount_By_Path(string Path, bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400635 std::vector<TWPartition*>::iterator iter;
636 int ret = false;
637 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400638 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy51127312012-09-08 13:08:49 -0400639
640 // Iterate through all partitions
641 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400642 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy51127312012-09-08 13:08:49 -0400643 ret = (*iter)->UnMount(Display_Error);
644 found = true;
645 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
646 (*iter)->UnMount(Display_Error);
647 }
648 }
649 if (found) {
650 return ret;
651 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500652 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 -0400653 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000654 LOGINFO("UnMount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400655 }
656 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400657}
658
Dees_Troy51a0e822012-09-05 15:24:24 -0400659int TWPartitionManager::Is_Mounted_By_Path(string Path) {
Dees_Troy51127312012-09-08 13:08:49 -0400660 TWPartition* Part = Find_Partition_By_Path(Path);
661
662 if (Part)
663 return Part->Is_Mounted();
664 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000665 LOGINFO("Is_Mounted: Unable to find partition for path '%s'\n", Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400666 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400667}
668
Dees_Troy5bf43922012-09-07 16:07:55 -0400669int TWPartitionManager::Mount_Current_Storage(bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400670 string current_storage_path = DataManager::GetCurrentStoragePath();
671
672 if (Mount_By_Path(current_storage_path, Display_Error)) {
673 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
674 if (FreeStorage)
675 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
676 return true;
677 }
678 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400679}
680
Dees_Troy5bf43922012-09-07 16:07:55 -0400681int TWPartitionManager::Mount_Settings_Storage(bool Display_Error) {
682 return Mount_By_Path(DataManager::GetSettingsStoragePath(), Display_Error);
683}
684
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600685TWPartition* TWPartitionManager::Find_Partition_By_Path(const string& Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400686 std::vector<TWPartition*>::iterator iter;
Dees_Troy38bd7602012-09-14 13:33:53 -0400687 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400688
dianlujitaob76a73a2020-04-30 20:33:20 +0800689 if (Local_Path == "/system")
690 Local_Path = Get_Android_Root_Path();
Dees_Troy5bf43922012-09-07 16:07:55 -0400691 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400692 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path))
Dees_Troy5bf43922012-09-07 16:07:55 -0400693 return (*iter);
694 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400695 return NULL;
696}
697
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600698TWPartition* TWPartitionManager::Find_Partition_By_Block_Device(const string& Block_Device) {
699 std::vector<TWPartition*>::iterator iter;
700
701 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
702 if ((*iter)->Primary_Block_Device == Block_Device || (!(*iter)->Actual_Block_Device.empty() && (*iter)->Actual_Block_Device == Block_Device))
703 return (*iter);
704 }
705 return NULL;
706}
707
Ethan Yonker53796e72019-01-11 22:49:52 -0600708int TWPartitionManager::Check_Backup_Name(const std::string& Backup_Name, bool Display_Error, bool Must_Be_Unique) {
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400709 // Check the backup name to ensure that it is the correct size and contains only valid characters
710 // and that a backup with that name doesn't already exist
711 char backup_name[MAX_BACKUP_NAME_LEN];
712 char backup_loc[255], tw_image_dir[255];
713 int copy_size;
714 int index, cur_char;
Ethan Yonker53796e72019-01-11 22:49:52 -0600715 string Backup_Loc;
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400716
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400717 copy_size = Backup_Name.size();
718 // Check size
719 if (copy_size > MAX_BACKUP_NAME_LEN) {
720 if (Display_Error)
Ethan Yonker74db1572015-10-28 12:44:49 -0500721 gui_err("backup_name_len=Backup name is too long.");
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400722 return -2;
723 }
724
725 // Check each character
726 strncpy(backup_name, Backup_Name.c_str(), copy_size);
Dees_Troya13d74f2013-03-24 08:54:55 -0500727 if (copy_size == 1 && strncmp(backup_name, "0", 1) == 0)
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400728 return 0; // A "0" (zero) means to use the current timestamp for the backup name
729 for (index=0; index<copy_size; index++) {
730 cur_char = (int)backup_name[index];
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500731 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 -0400732 // These are valid characters
733 // Numbers
734 // Upper case letters
735 // Lower case letters
736 // Space
737 // and -_.{}[]
738 } else {
739 if (Display_Error)
Ethan Yonker4adc33e2016-01-22 16:07:11 -0600740 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 -0400741 return -3;
742 }
743 }
744
Ethan Yonker53796e72019-01-11 22:49:52 -0600745 if (Must_Be_Unique) {
746 // Check to make sure that a backup with this name doesn't already exist
747 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Loc);
748 strcpy(backup_loc, Backup_Loc.c_str());
749 sprintf(tw_image_dir,"%s/%s", backup_loc, Backup_Name.c_str());
750 if (TWFunc::Path_Exists(tw_image_dir)) {
751 if (Display_Error)
752 gui_err("backup_name_exists=A backup with that name already exists!");
bigbiffce8f83c2015-12-12 18:30:21 -0500753
Ethan Yonker53796e72019-01-11 22:49:52 -0600754 return -4;
755 }
756 // Backup is unique
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400757 }
Ethan Yonker53796e72019-01-11 22:49:52 -0600758 // No problems found
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400759 return 0;
760}
761
bigbiffce8f83c2015-12-12 18:30:21 -0500762bool TWPartitionManager::Backup_Partition(PartitionSettings *part_settings) {
Dees_Troy43d8b002012-09-17 16:00:01 -0400763 time_t start, stop;
Matt Mower23d8aae2017-01-06 14:30:33 -0600764 int use_compression;
bigbiffce8f83c2015-12-12 18:30:21 -0500765 string backup_log = part_settings->Backup_Folder + "/recovery.log";
Dees_Troy43d8b002012-09-17 16:00:01 -0400766
bigbiffce8f83c2015-12-12 18:30:21 -0500767 if (part_settings->Part == NULL)
Dees_Troy43d8b002012-09-17 16:00:01 -0400768 return true;
769
Dees_Troy093b7642012-09-21 15:59:38 -0400770 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400771
Tom Hite5a926722014-09-15 01:31:03 +0000772 TWFunc::SetPerformanceMode(true);
Dees_Troy43d8b002012-09-17 16:00:01 -0400773 time(&start);
774
bigbiffce8f83c2015-12-12 18:30:21 -0500775 if (part_settings->Part->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500776 sync();
777 sync();
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400778 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500779 if (!part_settings->adbbackup && part_settings->generate_digest) {
780 if (!twrpDigestDriver::Make_Digest(Full_Filename))
781 goto backup_error;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400782 }
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400783
bigbiffce8f83c2015-12-12 18:30:21 -0500784 if (part_settings->Part->Has_SubPartition) {
Dees_Troy8170a922012-09-18 15:40:25 -0400785 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400786 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -0400787
788 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400789 if ((*subpart)->Can_Be_Backed_Up && (*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500790 part_settings->Part = *subpart;
bigbiffce8f83c2015-12-12 18:30:21 -0500791 if (!(*subpart)->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500792 goto backup_error;
Tom Hite5a926722014-09-15 01:31:03 +0000793 }
Dees_Troy2727b992013-08-14 20:09:30 +0000794 sync();
795 sync();
bigbiff bigbifff5955b12019-05-18 17:28:58 -0400796 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400797 if (!part_settings->adbbackup && part_settings->generate_digest) {
798 if (!twrpDigestDriver::Make_Digest(Full_Filename)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500799 goto backup_error;
bigbiffce8f83c2015-12-12 18:30:21 -0500800 }
Tom Hite5a926722014-09-15 01:31:03 +0000801 }
Dees_Troy8170a922012-09-18 15:40:25 -0400802 }
803 }
804 }
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400805
Dees_Troy43d8b002012-09-17 16:00:01 -0400806 time(&stop);
that203bcc92015-05-09 17:27:33 +0200807 int backup_time = (int) difftime(stop, start);
Dees_Troy2673cec2013-04-02 20:22:16 +0000808 LOGINFO("Partition Backup time: %d\n", backup_time);
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400809
bigbiffce8f83c2015-12-12 18:30:21 -0500810 if (part_settings->Part->Backup_Method == BM_FILES) {
811 part_settings->file_time += backup_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400812 } else {
bigbiffce8f83c2015-12-12 18:30:21 -0500813 part_settings->img_time += backup_time;
814
Dees_Troy43d8b002012-09-17 16:00:01 -0400815 }
Tom Hite5a926722014-09-15 01:31:03 +0000816
Matt Mower02899552016-12-30 18:13:51 -0600817 TWFunc::SetPerformanceMode(false);
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500818 return true;
Dees_Troy43d8b002012-09-17 16:00:01 -0400819 }
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500820backup_error:
821 Clean_Backup_Folder(part_settings->Backup_Folder);
822 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
823 tw_set_default_metadata(backup_log.c_str());
824 TWFunc::SetPerformanceMode(false);
825 return false;
bigbiff7abc5fe2015-01-17 16:53:12 -0500826}
827
bigbiffbf1d6722015-02-14 16:25:59 -0500828void TWPartitionManager::Clean_Backup_Folder(string Backup_Folder) {
829 DIR *d = opendir(Backup_Folder.c_str());
830 struct dirent *p;
831 int r;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400832 vector<string> ext;
833
834 //extensions we should delete when cleaning
835 ext.push_back("win");
836 ext.push_back("md5");
837 ext.push_back("sha2");
838 ext.push_back("info");
bigbiffbf1d6722015-02-14 16:25:59 -0500839
Ethan Yonker74db1572015-10-28 12:44:49 -0500840 gui_msg("backup_clean=Backup Failed. Cleaning Backup Folder.");
bigbiffbf1d6722015-02-14 16:25:59 -0500841
842 if (d == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500843 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Backup_Folder)(strerror(errno)));
bigbiffbf1d6722015-02-14 16:25:59 -0500844 return;
845 }
846
Matt Mower2b18a532015-02-20 16:58:05 -0600847 while ((p = readdir(d))) {
bigbiffbf1d6722015-02-14 16:25:59 -0500848 if (!strcmp(p->d_name, ".") || !strcmp(p->d_name, ".."))
849 continue;
850
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500851 string path = Backup_Folder + "/" + p->d_name;
bigbiffbf1d6722015-02-14 16:25:59 -0500852
853 size_t dot = path.find_last_of(".") + 1;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400854 for (vector<string>::const_iterator i = ext.begin(); i != ext.end(); ++i) {
855 if (path.substr(dot) == *i) {
856 r = unlink(path.c_str());
857 if (r != 0)
858 LOGINFO("Unable to unlink '%s: %s'\n", path.c_str(), strerror(errno));
bigbiffbf1d6722015-02-14 16:25:59 -0500859 }
860 }
861 }
862 closedir(d);
863}
864
Ethan Yonker472f5062016-02-25 13:47:30 -0600865int TWPartitionManager::Check_Backup_Cancel() {
866 return stop_backup.get_value();
867}
868
bigbiff7abc5fe2015-01-17 16:53:12 -0500869int TWPartitionManager::Cancel_Backup() {
870 string Backup_Folder, Backup_Name, Full_Backup_Path;
871
872 stop_backup.set_value(1);
873
874 if (tar_fork_pid != 0) {
875 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
876 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Folder);
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500877 Full_Backup_Path = Backup_Folder + "/" + Backup_Name;
bigbiff7abc5fe2015-01-17 16:53:12 -0500878 LOGINFO("Killing pid: %d\n", tar_fork_pid);
879 kill(tar_fork_pid, SIGUSR2);
880 while (kill(tar_fork_pid, 0) == 0) {
881 usleep(1000);
882 }
883 LOGINFO("Backup_Run stopped and returning false, backup cancelled.\n");
884 LOGINFO("Removing directory %s\n", Full_Backup_Path.c_str());
885 TWFunc::removeDir(Full_Backup_Path, false);
886 tar_fork_pid = 0;
887 }
888
889 return 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400890}
891
bigbiffce8f83c2015-12-12 18:30:21 -0500892int TWPartitionManager::Run_Backup(bool adbbackup) {
893 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400894 int partition_count = 0, disable_free_space_check = 0, skip_digest = 0;
bigbiffce8f83c2015-12-12 18:30:21 -0500895 string Backup_Name, Backup_List, backup_path;
Matt Mower23d8aae2017-01-06 14:30:33 -0600896 unsigned long long total_bytes = 0, free_space = 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400897 TWPartition* storage = NULL;
898 struct tm *t;
Matt Mower23d8aae2017-01-06 14:30:33 -0600899 time_t seconds, total_start, total_stop;
Dees_Troya13d74f2013-03-24 08:54:55 -0500900 size_t start_pos = 0, end_pos = 0;
bigbiff7abc5fe2015-01-17 16:53:12 -0500901 stop_backup.set_value(0);
Dees_Troy43d8b002012-09-17 16:00:01 -0400902 seconds = time(0);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500903 t = localtime(&seconds);
Dees_Troy43d8b002012-09-17 16:00:01 -0400904
bigbiffce8f83c2015-12-12 18:30:21 -0500905 part_settings.img_bytes_remaining = 0;
906 part_settings.file_bytes_remaining = 0;
907 part_settings.img_time = 0;
908 part_settings.file_time = 0;
909 part_settings.img_bytes = 0;
910 part_settings.file_bytes = 0;
911 part_settings.PM_Method = PM_BACKUP;
912
bigbiffce8f83c2015-12-12 18:30:21 -0500913 part_settings.adbbackup = adbbackup;
Dees_Troy43d8b002012-09-17 16:00:01 -0400914 time(&total_start);
915
916 Update_System_Details();
917
918 if (!Mount_Current_Storage(true))
919 return false;
920
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400921 DataManager::GetValue(TW_SKIP_DIGEST_GENERATE_VAR, skip_digest);
922 if (skip_digest == 0)
923 part_settings.generate_digest = true;
Dees_Troyc5865ab2012-09-24 15:08:04 -0400924 else
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400925 part_settings.generate_digest = false;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400926
bigbiffce8f83c2015-12-12 18:30:21 -0500927 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, part_settings.Backup_Folder);
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500928 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
929 if (Backup_Name == gui_lookup("curr_date", "(Current Date)")) {
930 Backup_Name = TWFunc::Get_Current_Date();
931 } else if (Backup_Name == gui_lookup("auto_generate", "(Auto Generate)") || Backup_Name == "0" || Backup_Name.empty()) {
Dees Troyb21cc642013-09-10 17:36:41 +0000932 TWFunc::Auto_Generate_Backup_Name();
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500933 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
Dees_Troy43d8b002012-09-17 16:00:01 -0400934 }
bigbiffce8f83c2015-12-12 18:30:21 -0500935
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500936 LOGINFO("Backup Name is: '%s'\n", Backup_Name.c_str());
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500937 part_settings.Backup_Folder = part_settings.Backup_Folder + "/" + Backup_Name;
bigbiffce8f83c2015-12-12 18:30:21 -0500938
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500939 LOGINFO("Backup_Folder is: '%s'\n", part_settings.Backup_Folder.c_str());
Dees_Troy43d8b002012-09-17 16:00:01 -0400940
Dees_Troy2673cec2013-04-02 20:22:16 +0000941 LOGINFO("Calculating backup details...\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500942 DataManager::GetValue("tw_backup_list", Backup_List);
bigbiff7ba75002020-04-11 20:47:09 -0400943 LOGINFO("Backup_List: %s\n", Backup_List.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500944 if (!Backup_List.empty()) {
945 end_pos = Backup_List.find(";", start_pos);
946 while (end_pos != string::npos && start_pos < Backup_List.size()) {
947 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiff7ba75002020-04-11 20:47:09 -0400948 LOGINFO("backup_path: %s\n", backup_path.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -0500949 part_settings.Part = Find_Partition_By_Path(backup_path);
950 if (part_settings.Part != NULL) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500951 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500952 if (part_settings.Part->Backup_Method == BM_FILES)
953 part_settings.file_bytes += part_settings.Part->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500954 else
bigbiffce8f83c2015-12-12 18:30:21 -0500955 part_settings.img_bytes += part_settings.Part->Backup_Size;
956 if (part_settings.Part->Has_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500957 std::vector<TWPartition*>::iterator subpart;
958
959 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiffce8f83c2015-12-12 18:30:21 -0500960 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 -0500961 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500962 if ((*subpart)->Backup_Method == BM_FILES)
963 part_settings.file_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500964 else
bigbiffce8f83c2015-12-12 18:30:21 -0500965 part_settings.img_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500966 }
967 }
Dees_Troy8170a922012-09-18 15:40:25 -0400968 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500969 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500970 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 -0400971 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500972 start_pos = end_pos + 1;
973 end_pos = Backup_List.find(";", start_pos);
Dees_Troy43d8b002012-09-17 16:00:01 -0400974 }
975 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400976
977 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500978 gui_msg("no_partition_selected=No partitions selected for backup.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400979 return false;
980 }
bigbiffce8f83c2015-12-12 18:30:21 -0500981 if (adbbackup) {
982 if (twadbbu::Write_ADB_Stream_Header(partition_count) == false) {
983 return false;
984 }
985 }
986 total_bytes = part_settings.file_bytes + part_settings.img_bytes;
Ethan Yonker472f5062016-02-25 13:47:30 -0600987 ProgressTracking progress(total_bytes);
bigbiffce8f83c2015-12-12 18:30:21 -0500988 part_settings.progress = &progress;
989
Ethan Yonker74db1572015-10-28 12:44:49 -0500990 gui_msg(Msg("total_partitions_backup= * Total number of partitions to back up: {1}")(partition_count));
991 gui_msg(Msg("total_backup_size= * Total size of all data: {1}MB")(total_bytes / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400992 storage = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
993 if (storage != NULL) {
994 free_space = storage->Free;
Ethan Yonker74db1572015-10-28 12:44:49 -0500995 gui_msg(Msg("available_space= * Available space: {1}MB")(free_space / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400996 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500997 gui_err("unable_locate_storage=Unable to locate storage device.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400998 return false;
999 }
bigbiffbf1d6722015-02-14 16:25:59 -05001000
Matt Mowerbfccfb82016-04-25 23:22:31 -05001001 DataManager::GetValue(TW_DISABLE_FREE_SPACE_VAR, disable_free_space_check);
bigbiffce8f83c2015-12-12 18:30:21 -05001002
1003 if (adbbackup)
1004 disable_free_space_check = true;
1005
bigbiffbf1d6722015-02-14 16:25:59 -05001006 if (!disable_free_space_check) {
1007 if (free_space - (32 * 1024 * 1024) < total_bytes) {
1008 // We require an extra 32MB just in case
Ethan Yonker74db1572015-10-28 12:44:49 -05001009 gui_err("no_space=Not enough free space on storage.");
bigbiffbf1d6722015-02-14 16:25:59 -05001010 return false;
1011 }
Dees_Troy43d8b002012-09-17 16:00:01 -04001012 }
bigbiffce8f83c2015-12-12 18:30:21 -05001013 part_settings.img_bytes_remaining = part_settings.img_bytes;
1014 part_settings.file_bytes_remaining = part_settings.file_bytes;
Dees_Troy43d8b002012-09-17 16:00:01 -04001015
Ethan Yonker74db1572015-10-28 12:44:49 -05001016 gui_msg("backup_started=[BACKUP STARTED]");
bigbiff bigbiffcdd97c72019-03-29 19:12:33 -04001017
1018 int is_decrypted = 0;
1019 int is_encrypted = 0;
1020
1021 DataManager::GetValue(TW_IS_DECRYPTED, is_decrypted);
1022 DataManager::GetValue(TW_IS_ENCRYPTED, is_encrypted);
1023 if (!adbbackup || (!is_encrypted || (is_encrypted && is_decrypted))) {
1024 gui_msg(Msg("backup_folder= * Backup Folder: {1}")(part_settings.Backup_Folder));
1025 if (!TWFunc::Recursive_Mkdir(part_settings.Backup_Folder)) {
1026 gui_err("fail_backup_folder=Failed to make backup folder.");
1027 return false;
1028 }
Dees_Troyd4b22b02013-01-18 17:17:58 +00001029 }
1030
Dees_Troy2673cec2013-04-02 20:22:16 +00001031 DataManager::SetProgress(0.0);
Dees_Troy093b7642012-09-21 15:59:38 -04001032
Dees_Troya13d74f2013-03-24 08:54:55 -05001033 start_pos = 0;
1034 end_pos = Backup_List.find(";", start_pos);
1035 while (end_pos != string::npos && start_pos < Backup_List.size()) {
bigbiff7abc5fe2015-01-17 16:53:12 -05001036 if (stop_backup.get_value() != 0)
1037 return -1;
Dees_Troya13d74f2013-03-24 08:54:55 -05001038 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001039 part_settings.Part = Find_Partition_By_Path(backup_path);
1040 if (part_settings.Part != NULL) {
1041 if (!Backup_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001042 return false;
1043 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001044 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 -05001045 }
1046 start_pos = end_pos + 1;
1047 end_pos = Backup_List.find(";", start_pos);
1048 }
Dees_Troy43d8b002012-09-17 16:00:01 -04001049
1050 // Average BPS
bigbiffce8f83c2015-12-12 18:30:21 -05001051 if (part_settings.img_time == 0)
1052 part_settings.img_time = 1;
1053 if (part_settings.file_time == 0)
1054 part_settings.file_time = 1;
1055 int img_bps = (int)part_settings.img_bytes / (int)part_settings.img_time;
1056 unsigned long long file_bps = part_settings.file_bytes / (int)part_settings.file_time;
Dees_Troy43d8b002012-09-17 16:00:01 -04001057
bigbiffce8f83c2015-12-12 18:30:21 -05001058 if (part_settings.file_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -06001059 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 -05001060 if (part_settings.img_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -06001061 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 -04001062
1063 time(&total_stop);
1064 int total_time = (int) difftime(total_stop, total_start);
bigbiffce8f83c2015-12-12 18:30:21 -05001065
1066 uint64_t actual_backup_size;
Ethan Yonker3fdcda42016-11-30 12:29:37 -06001067 if (!adbbackup) {
1068 TWExclude twe;
1069 actual_backup_size = twe.Get_Folder_Size(part_settings.Backup_Folder);
1070 } else
bigbiffce8f83c2015-12-12 18:30:21 -05001071 actual_backup_size = part_settings.file_bytes + part_settings.img_bytes;
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001072 actual_backup_size /= (1024LLU * 1024LLU);
Dees_Troy43d8b002012-09-17 16:00:01 -04001073
bigbiffce8f83c2015-12-12 18:30:21 -05001074 int prev_img_bps = 0, use_compression = 0;
1075 unsigned long long prev_file_bps = 0;
Matt Mower173cdb92016-12-31 02:49:19 -06001076 DataManager::GetValue(TW_BACKUP_AVG_IMG_RATE, prev_img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -04001077 img_bps += (prev_img_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001078 img_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -04001079
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001080 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -04001081 if (use_compression)
1082 DataManager::GetValue(TW_BACKUP_AVG_FILE_COMP_RATE, prev_file_bps);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001083 else
Dees_Troy093b7642012-09-21 15:59:38 -04001084 DataManager::GetValue(TW_BACKUP_AVG_FILE_RATE, prev_file_bps);
1085 file_bps += (prev_file_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001086 file_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -04001087
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001088 DataManager::SetValue(TW_BACKUP_AVG_IMG_RATE, img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -04001089 if (use_compression)
1090 DataManager::SetValue(TW_BACKUP_AVG_FILE_COMP_RATE, file_bps);
1091 else
1092 DataManager::SetValue(TW_BACKUP_AVG_FILE_RATE, file_bps);
1093
Ethan Yonker74db1572015-10-28 12:44:49 -05001094 gui_msg(Msg("total_backed_size=[{1} MB TOTAL BACKED UP]")(actual_backup_size));
Dees_Troy43d8b002012-09-17 16:00:01 -04001095 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001096 UnMount_Main_Partitions();
Ethan Yonker56db1c42015-12-20 22:49:00 -06001097 gui_msg(Msg(msg::kHighlight, "backup_completed=[BACKUP COMPLETED IN {1} SECONDS]")(total_time)); // the end
Ethan Yonkere080c1f2016-09-19 13:50:25 -05001098 string backup_log = part_settings.Backup_Folder + "/recovery.log";
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001099 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
Ethan Yonker4b94cfd2014-12-11 10:00:45 -06001100 tw_set_default_metadata(backup_log.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -05001101
1102 if (part_settings.adbbackup) {
1103 if (twadbbu::Write_ADB_Stream_Trailer() == false) {
1104 return false;
1105 }
1106 }
1107 part_settings.adbbackup = false;
1108 DataManager::SetValue("tw_enable_adb_backup", 0);
1109
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001110 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001111}
1112
bigbiffce8f83c2015-12-12 18:30:21 -05001113bool TWPartitionManager::Restore_Partition(PartitionSettings *part_settings) {
Dees_Troy4a2a1262012-09-18 09:33:47 -04001114 time_t Start, Stop;
bigbiffce8f83c2015-12-12 18:30:21 -05001115
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001116 if (part_settings->adbbackup) {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001117 std::string partName = part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win";
1118 LOGINFO("setting backup name: %s\n", partName.c_str());
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001119 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
1120 }
1121
Tom Hite5a926722014-09-15 01:31:03 +00001122 TWFunc::SetPerformanceMode(true);
bigbiffce8f83c2015-12-12 18:30:21 -05001123
Dees_Troy4a2a1262012-09-18 09:33:47 -04001124 time(&Start);
Ethan Yonker472f5062016-02-25 13:47:30 -06001125
bigbiffce8f83c2015-12-12 18:30:21 -05001126 if (!part_settings->Part->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001127 TWFunc::SetPerformanceMode(false);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001128 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001129 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001130 if (part_settings->Part->Has_SubPartition && !part_settings->adbbackup) {
Dees_Troy8170a922012-09-18 15:40:25 -04001131 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001132 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -04001133
1134 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001135 part_settings->Part = *subpart;
1136 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
1137 part_settings->Part = (*subpart);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001138 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
bigbiffce8f83c2015-12-12 18:30:21 -05001139 if (!(*subpart)->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001140 TWFunc::SetPerformanceMode(false);
Dees_Troy8170a922012-09-18 15:40:25 -04001141 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001142 }
Dees_Troy8170a922012-09-18 15:40:25 -04001143 }
1144 }
1145 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001146 time(&Stop);
Tom Hite5a926722014-09-15 01:31:03 +00001147 TWFunc::SetPerformanceMode(false);
bigbiffce8f83c2015-12-12 18:30:21 -05001148 gui_msg(Msg("restore_part_done=[{1} done ({2} seconds)]")(part_settings->Part->Backup_Display_Name)((int)difftime(Stop, Start)));
1149
Dees_Troy4a2a1262012-09-18 09:33:47 -04001150 return true;
1151}
1152
Ethan Yonker472f5062016-02-25 13:47:30 -06001153int TWPartitionManager::Run_Restore(const string& Restore_Name) {
bigbiffce8f83c2015-12-12 18:30:21 -05001154 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001155 int check_digest;
bigbiffce8f83c2015-12-12 18:30:21 -05001156
Dees_Troy4a2a1262012-09-18 09:33:47 -04001157 time_t rStart, rStop;
1158 time(&rStart);
Dees_Troya13d74f2013-03-24 08:54:55 -05001159 string Restore_List, restore_path;
1160 size_t start_pos = 0, end_pos;
bigbiffce8f83c2015-12-12 18:30:21 -05001161
Ethan Yonkerdcf2b672016-09-13 14:41:53 -05001162 part_settings.Backup_Folder = Restore_Name;
bigbiffce8f83c2015-12-12 18:30:21 -05001163 part_settings.Part = NULL;
1164 part_settings.partition_count = 0;
1165 part_settings.total_restore_size = 0;
1166 part_settings.adbbackup = false;
1167 part_settings.PM_Method = PM_RESTORE;
Dees_Troy43d8b002012-09-17 16:00:01 -04001168
Ethan Yonker74db1572015-10-28 12:44:49 -05001169 gui_msg("restore_started=[RESTORE STARTED]");
1170 gui_msg(Msg("restore_folder=Restore folder: '{1}'")(Restore_Name));
Dees_Troy43d8b002012-09-17 16:00:01 -04001171
Dees_Troy4a2a1262012-09-18 09:33:47 -04001172 if (!Mount_Current_Storage(true))
1173 return false;
1174
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001175 DataManager::GetValue(TW_SKIP_DIGEST_CHECK_VAR, check_digest);
1176 if (check_digest > 0) {
1177 // Check Digest files first before restoring to ensure that all of them match before starting a restore
1178 TWFunc::GUI_Operation_Text(TW_VERIFY_DIGEST_TEXT, gui_parse_text("{@verifying_digest}"));
1179 gui_msg("verifying_digest=Verifying Digest");
Dees_Troya13d74f2013-03-24 08:54:55 -05001180 } else {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001181 gui_msg("skip_digest=Skipping Digest check based on user setting.");
Dees_Troya13d74f2013-03-24 08:54:55 -05001182 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001183 gui_msg("calc_restore=Calculating restore details...");
Dees_Troya13d74f2013-03-24 08:54:55 -05001184 DataManager::GetValue("tw_restore_selected", Restore_List);
bigbiffce8f83c2015-12-12 18:30:21 -05001185
Dees_Troya13d74f2013-03-24 08:54:55 -05001186 if (!Restore_List.empty()) {
1187 end_pos = Restore_List.find(";", start_pos);
1188 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1189 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001190 part_settings.Part = Find_Partition_By_Path(restore_path);
1191 if (part_settings.Part != NULL) {
bigbiffce8f83c2015-12-12 18:30:21 -05001192 if (part_settings.Part->Mount_Read_Only) {
1193 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 -05001194 return false;
1195 }
bigbiffce8f83c2015-12-12 18:30:21 -05001196
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001197 string Full_Filename = part_settings.Backup_Folder + "/" + part_settings.Part->Backup_FileName;
1198
bigbiff71840d42021-01-23 14:08:03 -05001199 if (tw_get_default_metadata(Get_Android_Root_Path().c_str()) != 0) {
1200 gui_msg(Msg(msg::kWarning, "restore_system_context=Unable to get default context for {1} -- Android may not boot.")(Get_Android_Root_Path()));
1201 }
1202
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001203 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001204 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001205 part_settings.partition_count++;
1206 part_settings.total_restore_size += part_settings.Part->Get_Restore_Size(&part_settings);
1207 if (part_settings.Part->Has_SubPartition) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001208 TWPartition *parentPart = part_settings.Part;
Dees_Troya13d74f2013-03-24 08:54:55 -05001209 std::vector<TWPartition*>::iterator subpart;
1210
1211 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001212 part_settings.Part = *subpart;
1213 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001214 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001215 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001216 part_settings.total_restore_size += (*subpart)->Get_Restore_Size(&part_settings);
Dees_Troya13d74f2013-03-24 08:54:55 -05001217 }
1218 }
1219 }
1220 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001221 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001222 }
1223 start_pos = end_pos + 1;
1224 end_pos = Restore_List.find(";", start_pos);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001225 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001226 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001227
bigbiffce8f83c2015-12-12 18:30:21 -05001228 if (part_settings.partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001229 gui_err("no_part_restore=No partitions selected for restore.");
Dees_Troy4a2a1262012-09-18 09:33:47 -04001230 return false;
1231 }
1232
bigbiffce8f83c2015-12-12 18:30:21 -05001233 gui_msg(Msg("restore_part_count=Restoring {1} partitions...")(part_settings.partition_count));
1234 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 +00001235 DataManager::SetProgress(0.0);
bigbiffce8f83c2015-12-12 18:30:21 -05001236 ProgressTracking progress(part_settings.total_restore_size);
1237 part_settings.progress = &progress;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001238
Dees_Troya13d74f2013-03-24 08:54:55 -05001239 start_pos = 0;
1240 if (!Restore_List.empty()) {
1241 end_pos = Restore_List.find(";", start_pos);
1242 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1243 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001244
1245 part_settings.Part = Find_Partition_By_Path(restore_path);
1246 if (part_settings.Part != NULL) {
1247 part_settings.partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -05001248 if (!Restore_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001249 return false;
1250 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001251 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001252 }
1253 start_pos = end_pos + 1;
1254 end_pos = Restore_List.find(";", start_pos);
1255 }
1256 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001257 TWFunc::GUI_Operation_Text(TW_UPDATE_SYSTEM_DETAILS_TEXT, gui_parse_text("{@updating_system_details}"));
bigbiff71840d42021-01-23 14:08:03 -05001258 tw_set_default_metadata(Get_Android_Root_Path().c_str());
Captain Throwback9d6feb52018-07-27 10:05:24 -04001259 UnMount_By_Path(Get_Android_Root_Path(), false);
Dees_Troy43d8b002012-09-17 16:00:01 -04001260 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001261 UnMount_Main_Partitions();
Dees_Troy4a2a1262012-09-18 09:33:47 -04001262 time(&rStop);
Matt Mower3c366972015-12-25 19:28:31 -06001263 gui_msg(Msg(msg::kHighlight, "restore_completed=[RESTORE COMPLETED IN {1} SECONDS]")((int)difftime(rStop,rStart)));
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001264 DataManager::SetValue("tw_file_progress", "");
bigbiffce8f83c2015-12-12 18:30:21 -05001265
Dees_Troy63c8df72012-09-10 14:02:05 -04001266 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001267}
1268
1269void TWPartitionManager::Set_Restore_Files(string Restore_Name) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001270 // Start with the default values
Dees_Troya13d74f2013-03-24 08:54:55 -05001271 string Restore_List;
Dees_Troy83bd4832013-05-04 12:39:56 +00001272 bool get_date = true, check_encryption = true;
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001273 bool adbbackup = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001274
1275 DataManager::SetValue("tw_restore_encrypted", 0);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001276 if (twadbbu::Check_ADB_Backup_File(Restore_Name)) {
1277 vector<string> adb_files;
1278 adb_files = twadbbu::Get_ADB_Backup_Files(Restore_Name);
1279 for (unsigned int i = 0; i < adb_files.size(); ++i) {
1280 string adb_restore_file = adb_files.at(i);
1281 std::size_t pos = adb_restore_file.find_first_of(".");
1282 std::string path = "/" + adb_restore_file.substr(0, pos);
1283 Restore_List = path + ";";
1284 TWPartition* Part = Find_Partition_By_Path(path);
1285 Part->Backup_FileName = TWFunc::Get_Filename(adb_restore_file);
1286 adbbackup = true;
1287 }
1288 DataManager::SetValue("tw_enable_adb_backup", 1);
Dees_Troy63c8df72012-09-10 14:02:05 -04001289 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001290 else {
1291 DIR* d;
1292 d = opendir(Restore_Name.c_str());
1293 if (d == NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001294 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001295 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Restore_Name)(strerror(errno)));
1296 return;
Dees_Troy63c8df72012-09-10 14:02:05 -04001297 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001298
1299 struct dirent* de;
1300 while ((de = readdir(d)) != NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001301 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001302 // Strip off three components
1303 char str[256];
1304 char* label;
1305 char* fstype = NULL;
1306 char* extn = NULL;
1307 char* ptr;
Dees_Troy63c8df72012-09-10 14:02:05 -04001308
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001309 strcpy(str, de->d_name);
1310 if (strlen(str) <= 2)
1311 continue;
Dees_Troy83bd4832013-05-04 12:39:56 +00001312
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001313 if (get_date) {
1314 char file_path[255];
1315 struct stat st;
1316
1317 strcpy(file_path, Restore_Name.c_str());
1318 strcat(file_path, "/");
1319 strcat(file_path, str);
1320 stat(file_path, &st);
1321 string backup_date = ctime((const time_t*)(&st.st_mtime));
1322 DataManager::SetValue(TW_RESTORE_FILE_DATE, backup_date);
1323 get_date = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001324 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001325
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001326 label = str;
1327 ptr = label;
1328 while (*ptr && *ptr != '.') ptr++;
1329 if (*ptr == '.')
1330 {
1331 *ptr = 0x00;
1332 ptr++;
1333 fstype = ptr;
1334 }
1335 while (*ptr && *ptr != '.') ptr++;
1336 if (*ptr == '.')
1337 {
1338 *ptr = 0x00;
1339 ptr++;
1340 extn = ptr;
1341 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001342
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001343 if (fstype == NULL || extn == NULL || strcmp(fstype, "log") == 0) continue;
1344 int extnlength = strlen(extn);
1345 if (extnlength != 3 && extnlength != 6) continue;
1346 if (extnlength >= 3 && strncmp(extn, "win", 3) != 0) continue;
1347 //if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
Dees_Troy63c8df72012-09-10 14:02:05 -04001348
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001349 if (check_encryption) {
1350 string filename = Restore_Name + "/";
1351 filename += de->d_name;
1352 if (TWFunc::Get_File_Type(filename) == 2) {
1353 LOGINFO("'%s' is encrypted\n", filename.c_str());
1354 DataManager::SetValue("tw_restore_encrypted", 1);
1355 }
1356 }
1357 if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
1358
1359 TWPartition* Part = Find_Partition_By_Path(label);
1360 if (Part == NULL)
1361 {
1362 gui_msg(Msg(msg::kError, "unable_locate_part_backup_name=Unable to locate partition by backup name: '{1}'")(label));
1363 continue;
1364 }
1365
1366 Part->Backup_FileName = de->d_name;
1367 if (strlen(extn) > 3) {
1368 Part->Backup_FileName.resize(Part->Backup_FileName.size() - strlen(extn) + 3);
1369 }
1370
dianlujitaob76a73a2020-04-30 20:33:20 +08001371 if (!Part->Is_SubPartition) {
1372 if (Part->Backup_Path == Get_Android_Root_Path())
1373 Restore_List += "/system;";
1374 else
1375 Restore_List += Part->Backup_Path + ";";
1376 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001377 }
1378 closedir(d);
Dees_Troy63c8df72012-09-10 14:02:05 -04001379 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001380
1381 if (adbbackup) {
1382 Restore_List = "ADB_Backup;";
1383 adbbackup = false;
1384 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001385
Dees_Troya13d74f2013-03-24 08:54:55 -05001386 // Set the final value
1387 DataManager::SetValue("tw_restore_list", Restore_List);
1388 DataManager::SetValue("tw_restore_selected", Restore_List);
Dees_Troy51a0e822012-09-05 15:24:24 -04001389 return;
1390}
1391
1392int TWPartitionManager::Wipe_By_Path(string Path) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001393 std::vector<TWPartition*>::iterator iter;
Mohd Farazfdd49bb2020-04-21 18:56:44 +05301394 std::vector < TWPartition * >::iterator iter1;
Dees_Troy63c8df72012-09-10 14:02:05 -04001395 int ret = false;
1396 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -04001397 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy63c8df72012-09-10 14:02:05 -04001398
dianlujitaob76a73a2020-04-30 20:33:20 +08001399 if (Local_Path == "/system")
1400 Local_Path = Get_Android_Root_Path();
Dees_Troy63c8df72012-09-10 14:02:05 -04001401 // Iterate through all partitions
1402 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -04001403 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Mohd Farazfdd49bb2020-04-21 18:56:44 +05301404 // iterate through all partitions since some legacy devices uses other partitions as vendor causes issues while wiping
1405 (*iter)->Find_Actual_Block_Device();
1406 for (iter1 = Partitions.begin (); iter1 != Partitions.end (); iter1++)
1407 {
1408 (*iter1)->Find_Actual_Block_Device();
1409 if ((*iter)->Actual_Block_Device == (*iter1)->Actual_Block_Device && (*iter)->Mount_Point != (*iter1)->Mount_Point)
1410 (*iter1)->UnMount(false);
1411 }
Dees_Troye58d5262012-09-21 12:27:57 -04001412 if (Path == "/and-sec")
1413 ret = (*iter)->Wipe_AndSec();
1414 else
1415 ret = (*iter)->Wipe();
Dees_Troy63c8df72012-09-10 14:02:05 -04001416 found = true;
1417 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1418 (*iter)->Wipe();
1419 }
1420 }
1421 if (found) {
1422 return ret;
1423 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001424 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 -04001425 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -04001426}
1427
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001428int TWPartitionManager::Wipe_By_Path(string Path, string New_File_System) {
1429 std::vector<TWPartition*>::iterator iter;
1430 int ret = false;
1431 bool found = false;
1432 string Local_Path = TWFunc::Get_Root_Path(Path);
1433
1434 // Iterate through all partitions
1435 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1436 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1437 if (Path == "/and-sec")
1438 ret = (*iter)->Wipe_AndSec();
1439 else
1440 ret = (*iter)->Wipe(New_File_System);
1441 found = true;
1442 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1443 (*iter)->Wipe(New_File_System);
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));
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001450 return false;
1451}
1452
Dees_Troy51a0e822012-09-05 15:24:24 -04001453int TWPartitionManager::Factory_Reset(void) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001454 std::vector<TWPartition*>::iterator iter;
1455 int ret = true;
1456
1457 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001458 if ((*iter)->Wipe_During_Factory_Reset && (*iter)->Is_Present) {
Ethan Yonker89583ef2015-08-26 09:01:59 -05001459#ifdef TW_OEM_BUILD
1460 if ((*iter)->Mount_Point == "/data") {
1461 if (!(*iter)->Wipe_Encryption())
1462 ret = false;
1463 } else {
1464#endif
1465 if (!(*iter)->Wipe())
1466 ret = false;
1467#ifdef TW_OEM_BUILD
1468 }
1469#endif
Dees_Troy094207a2012-09-26 12:00:39 -04001470 } else if ((*iter)->Has_Android_Secure) {
1471 if (!(*iter)->Wipe_AndSec())
1472 ret = false;
Dees_Troy63c8df72012-09-10 14:02:05 -04001473 }
1474 }
bigbiffad58e1b2020-07-06 20:24:34 -04001475 TWFunc::check_and_run_script("/system/bin/factoryreset.sh", "Factory Reset Script");
Dees_Troy63c8df72012-09-10 14:02:05 -04001476 return ret;
Dees_Troy51a0e822012-09-05 15:24:24 -04001477}
1478
Dees_Troy38bd7602012-09-14 13:33:53 -04001479int TWPartitionManager::Wipe_Dalvik_Cache(void) {
1480 struct stat st;
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001481 vector <string> dir;
Dees_Troy38bd7602012-09-14 13:33:53 -04001482
1483 if (!Mount_By_Path("/data", true))
1484 return false;
1485
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001486 dir.push_back("/data/dalvik-cache");
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001487
bigbiff25d25b92020-06-19 16:07:38 -04001488 std::string cacheDir = TWFunc::get_log_dir();
1489 if (cacheDir == CACHE_LOGS_DIR) {
1490 if (!PartitionManager.Mount_By_Path(CACHE_LOGS_DIR, false)) {
1491 LOGINFO("Unable to mount %s for wiping cache.\n", CACHE_LOGS_DIR);
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001492 }
1493 dir.push_back(cacheDir + "dalvik-cache");
1494 dir.push_back(cacheDir + "/dc");
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001495 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001496
Dees_Troy38bd7602012-09-14 13:33:53 -04001497 TWPartition* sdext = Find_Partition_By_Path("/sd-ext");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001498 if (sdext && sdext->Is_Present && sdext->Mount(false))
1499 {
1500 if (stat("/sd-ext/dalvik-cache", &st) == 0)
1501 {
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001502 dir.push_back("/sd-ext/dalvik-cache");
1503 }
1504 }
1505
bigbiff25d25b92020-06-19 16:07:38 -04001506 if (cacheDir == CACHE_LOGS_DIR) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001507 gui_msg("wiping_cache_dalvik=Wiping Dalvik Cache Directories...");
1508 } else {
1509 gui_msg("wiping_dalvik=Wiping Dalvik Directory...");
1510 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001511 for (unsigned i = 0; i < dir.size(); ++i) {
1512 if (stat(dir.at(i).c_str(), &st) == 0) {
1513 TWFunc::removeDir(dir.at(i), false);
1514 gui_msg(Msg("cleaned=Cleaned: {1}...")(dir.at(i)));
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001515 }
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001516 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001517
bigbiff25d25b92020-06-19 16:07:38 -04001518 if (cacheDir == CACHE_LOGS_DIR) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001519 gui_msg("cache_dalvik_done=-- Dalvik Cache Directories Wipe Complete!");
1520 } else {
1521 gui_msg("dalvik_done=-- Dalvik Directory Wipe Complete!");
1522 }
1523
Dees_Troy38bd7602012-09-14 13:33:53 -04001524 return true;
1525}
1526
1527int TWPartitionManager::Wipe_Rotate_Data(void) {
1528 if (!Mount_By_Path("/data", true))
1529 return false;
1530
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001531 unlink("/data/misc/akmd*");
1532 unlink("/data/misc/rild*");
Dees_Troy2673cec2013-04-02 20:22:16 +00001533 gui_print("Rotation data wiped.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001534 return true;
1535}
1536
1537int TWPartitionManager::Wipe_Battery_Stats(void) {
1538 struct stat st;
1539
1540 if (!Mount_By_Path("/data", true))
1541 return false;
1542
1543 if (0 != stat("/data/system/batterystats.bin", &st)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001544 gui_print("No Battery Stats Found. No Need To Wipe.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001545 } else {
1546 remove("/data/system/batterystats.bin");
Dees_Troy2673cec2013-04-02 20:22:16 +00001547 gui_print("Cleared battery stats.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001548 }
1549 return true;
1550}
1551
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001552int TWPartitionManager::Wipe_Android_Secure(void) {
1553 std::vector<TWPartition*>::iterator iter;
1554 int ret = false;
1555 bool found = false;
1556
1557 // Iterate through all partitions
1558 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1559 if ((*iter)->Has_Android_Secure) {
1560 ret = (*iter)->Wipe_AndSec();
1561 found = true;
1562 }
1563 }
1564 if (found) {
1565 return ret;
1566 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001567 gui_err("no_andsec=No android secure partitions found.");
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001568 }
1569 return false;
1570}
1571
Dees_Troy38bd7602012-09-14 13:33:53 -04001572int TWPartitionManager::Format_Data(void) {
1573 TWPartition* dat = Find_Partition_By_Path("/data");
1574
1575 if (dat != NULL) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001576 return dat->Wipe_Encryption();
1577 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001578 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001579 return false;
1580 }
1581 return false;
1582}
1583
1584int TWPartitionManager::Wipe_Media_From_Data(void) {
1585 TWPartition* dat = Find_Partition_By_Path("/data");
1586
1587 if (dat != NULL) {
1588 if (!dat->Has_Data_Media) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001589 LOGERR("This device does not have /data/media\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001590 return false;
1591 }
1592 if (!dat->Mount(true))
1593 return false;
1594
Ethan Yonker74db1572015-10-28 12:44:49 -05001595 gui_msg("wiping_datamedia=Wiping internal storage -- /data/media...");
Ethan Yonker726a0202014-12-16 20:01:38 -06001596 Remove_MTP_Storage(dat->MTP_Storage_ID);
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001597 TWFunc::removeDir("/data/media", false);
xiaolu9416f4f2015-06-04 08:22:23 +08001598 dat->Recreate_Media_Folder();
Ethan Yonker726a0202014-12-16 20:01:38 -06001599 Add_MTP_Storage(dat->MTP_Storage_ID);
Dees_Troy38bd7602012-09-14 13:33:53 -04001600 return true;
1601 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001602 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001603 return false;
1604 }
1605 return false;
1606}
1607
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001608int TWPartitionManager::Repair_By_Path(string Path, bool Display_Error) {
1609 std::vector<TWPartition*>::iterator iter;
1610 int ret = false;
1611 bool found = false;
1612 string Local_Path = TWFunc::Get_Root_Path(Path);
1613
1614 if (Local_Path == "/tmp" || Local_Path == "/")
1615 return true;
1616
1617 // Iterate through all partitions
1618 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1619 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1620 ret = (*iter)->Repair();
1621 found = true;
1622 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1623 (*iter)->Repair();
1624 }
1625 }
1626 if (found) {
1627 return ret;
1628 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001629 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 -05001630 } else {
1631 LOGINFO("Repair: Unable to find partition for path '%s'\n", Local_Path.c_str());
1632 }
1633 return false;
1634}
1635
Ethan Yonkera2719152015-05-28 09:44:41 -05001636int TWPartitionManager::Resize_By_Path(string Path, bool Display_Error) {
1637 std::vector<TWPartition*>::iterator iter;
1638 int ret = false;
1639 bool found = false;
1640 string Local_Path = TWFunc::Get_Root_Path(Path);
1641
1642 if (Local_Path == "/tmp" || Local_Path == "/")
1643 return true;
1644
1645 // Iterate through all partitions
1646 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1647 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1648 ret = (*iter)->Resize();
1649 found = true;
1650 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1651 (*iter)->Resize();
1652 }
1653 }
1654 if (found) {
1655 return ret;
1656 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001657 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 -05001658 } else {
1659 LOGINFO("Resize: Unable to find partition for path '%s'\n", Local_Path.c_str());
1660 }
1661 return false;
1662}
1663
Dees_Troy51a0e822012-09-05 15:24:24 -04001664void TWPartitionManager::Update_System_Details(void) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001665 std::vector<TWPartition*>::iterator iter;
Dees_Troy51127312012-09-08 13:08:49 -04001666 int data_size = 0;
Dees_Troy5bf43922012-09-07 16:07:55 -04001667
Ethan Yonker74db1572015-10-28 12:44:49 -05001668 gui_msg("update_part_details=Updating partition details...");
Dees_Troy5bf43922012-09-07 16:07:55 -04001669 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker1b190162016-12-05 15:25:19 -06001670 (*iter)->Update_Size(true);
Dees_Troy51127312012-09-08 13:08:49 -04001671 if ((*iter)->Can_Be_Mounted) {
Captain Throwback9d6feb52018-07-27 10:05:24 -04001672 if ((*iter)->Mount_Point == Get_Android_Root_Path()) {
Dees_Troy51127312012-09-08 13:08:49 -04001673 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1674 DataManager::SetValue(TW_BACKUP_SYSTEM_SIZE, backup_display_size);
Ethan Yonker76bbd3a2019-05-10 10:50:04 -05001675 TWFunc::Is_TWRP_App_In_System();
Dees_Troy51127312012-09-08 13:08:49 -04001676 } else if ((*iter)->Mount_Point == "/data" || (*iter)->Mount_Point == "/datadata") {
1677 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
1678 } else if ((*iter)->Mount_Point == "/cache") {
1679 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1680 DataManager::SetValue(TW_BACKUP_CACHE_SIZE, backup_display_size);
1681 } else if ((*iter)->Mount_Point == "/sd-ext") {
1682 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1683 DataManager::SetValue(TW_BACKUP_SDEXT_SIZE, backup_display_size);
1684 if ((*iter)->Backup_Size == 0) {
1685 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 0);
1686 DataManager::SetValue(TW_BACKUP_SDEXT_VAR, 0);
1687 } else
1688 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 1);
Dees_Troye58d5262012-09-21 12:27:57 -04001689 } else if ((*iter)->Has_Android_Secure) {
Dees_Troy8170a922012-09-18 15:40:25 -04001690 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troye58d5262012-09-21 12:27:57 -04001691 DataManager::SetValue(TW_BACKUP_ANDSEC_SIZE, backup_display_size);
Dees_Troy8170a922012-09-18 15:40:25 -04001692 if ((*iter)->Backup_Size == 0) {
1693 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 0);
1694 DataManager::SetValue(TW_BACKUP_ANDSEC_VAR, 0);
1695 } else
1696 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 1);
Dees_Troy2c50e182012-09-26 20:05:28 -04001697 } else if ((*iter)->Mount_Point == "/boot") {
1698 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1699 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1700 if ((*iter)->Backup_Size == 0) {
1701 DataManager::SetValue("tw_has_boot_partition", 0);
1702 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1703 } else
1704 DataManager::SetValue("tw_has_boot_partition", 1);
Dees_Troy51127312012-09-08 13:08:49 -04001705 }
Dees_Troyaa9cc402012-10-13 12:14:05 -04001706 } else {
1707 // Handle unmountable partitions in case we reset defaults
1708 if ((*iter)->Mount_Point == "/boot") {
1709 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1710 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1711 if ((*iter)->Backup_Size == 0) {
1712 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 0);
1713 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1714 } else
1715 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 1);
1716 } else if ((*iter)->Mount_Point == "/recovery") {
1717 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1718 DataManager::SetValue(TW_BACKUP_RECOVERY_SIZE, backup_display_size);
1719 if ((*iter)->Backup_Size == 0) {
1720 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 0);
1721 DataManager::SetValue(TW_BACKUP_RECOVERY_VAR, 0);
1722 } else
1723 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 1);
Gary Peck82599a82012-11-21 16:23:12 -08001724 } else if ((*iter)->Mount_Point == "/data") {
1725 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troyaa9cc402012-10-13 12:14:05 -04001726 }
Dees_Troy51127312012-09-08 13:08:49 -04001727 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001728 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001729 gui_msg("update_part_details_done=...done");
Dees_Troy51127312012-09-08 13:08:49 -04001730 DataManager::SetValue(TW_BACKUP_DATA_SIZE, data_size);
1731 string current_storage_path = DataManager::GetCurrentStoragePath();
1732 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
Dees_Troy8170a922012-09-18 15:40:25 -04001733 if (FreeStorage != NULL) {
1734 // Attempt to mount storage
bigbiff25d25b92020-06-19 16:07:38 -04001735 if (!FreeStorage->Mount(false)) {
1736 gui_msg(Msg(msg::kWarning, "unable_to_mount_storage=Unable to mount storage"));
1737 DataManager::SetValue(TW_STORAGE_FREE_SIZE, 0);
1738 } else {
1739 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
Dees_Troy8170a922012-09-18 15:40:25 -04001740 }
1741 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001742 LOGINFO("Unable to find storage partition '%s'.\n", current_storage_path.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -04001743 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001744 if (!Write_Fstab())
Dees_Troy2673cec2013-04-02 20:22:16 +00001745 LOGERR("Error creating fstab\n");
Dees_Troy51a0e822012-09-05 15:24:24 -04001746 return;
1747}
1748
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001749void TWPartitionManager::Post_Decrypt(const string& Block_Device) {
1750 TWPartition* dat = Find_Partition_By_Path("/data");
bigbiffc2cb3852020-09-11 14:03:31 -04001751
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001752 if (dat != NULL) {
1753 DataManager::SetValue(TW_IS_DECRYPTED, 1);
1754 dat->Is_Decrypted = true;
1755 if (!Block_Device.empty()) {
1756 dat->Decrypted_Block_Device = Block_Device;
1757 gui_msg(Msg("decrypt_success_dev=Data successfully decrypted, new block device: '{1}'")(Block_Device));
1758 } else {
1759 gui_msg("decrypt_success_nodev=Data successfully decrypted");
1760 }
Noah Jacobson81d638d2019-04-28 00:10:07 -04001761 property_set("twrp.decrypt.done", "true");
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001762 dat->Setup_File_System(false);
Noah Jacobson81d638d2019-04-28 00:10:07 -04001763 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 -06001764
Noah Jacobson81d638d2019-04-28 00:10:07 -04001765 sleep(1); // Sleep for a bit so that the device will be ready
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001766 if (dat->Has_Data_Media && dat->Mount(false) && TWFunc::Path_Exists("/data/media/0")) {
1767 dat->Storage_Path = "/data/media/0";
1768 dat->Symlink_Path = dat->Storage_Path;
1769 DataManager::SetValue("tw_storage_path", "/data/media/0");
1770 DataManager::SetValue("tw_settings_path", "/data/media/0");
1771 dat->UnMount(false);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001772 }
epicXaaddbe02021-01-04 13:01:31 +05301773 DataManager::LoadTWRPFolderInfo();
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001774 Update_System_Details();
nkk71ffb02bd2017-06-04 23:12:16 +03001775 Output_Partition(dat);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001776 UnMount_Main_Partitions();
1777 } else
1778 LOGERR("Unable to locate data partition.\n");
1779}
1780
Noah Jacobson81d638d2019-04-28 00:10:07 -04001781void TWPartitionManager::Parse_Users() {
1782#ifdef TW_INCLUDE_FBE
1783 char user_check_result[PROPERTY_VALUE_MAX];
1784 for (int userId = 0; userId <= 9999; userId++) {
1785 string prop = "twrp.user." + to_string(userId) + ".decrypt";
1786 property_get(prop.c_str(), user_check_result, "-1");
1787 if (strcmp(user_check_result, "-1") != 0) {
1788 if (userId < 0 || userId > 9999) {
1789 LOGINFO("Incorrect user id %d\n", userId);
1790 continue;
1791 }
1792 struct users_struct user;
1793 user.userId = to_string(userId);
1794
1795 // Attempt to get name of user. Fallback to user ID if this fails.
1796 char* userFile = PageManager::LoadFileToBuffer("/data/system/users/" + to_string(userId) + ".xml", NULL);
1797 if (userFile == NULL)
1798 user.userName = to_string(userId);
1799 else {
1800 xml_document<> *userXml = new xml_document<>();
1801 userXml->parse<0>(userFile);
1802 xml_node<>* userNode = userXml->first_node("user");
1803 if (userNode == nullptr) {
1804 user.userName = to_string(userId);
1805 } else {
1806 xml_node<>* nameNode = userNode->first_node("name");
1807 if (nameNode == nullptr)
1808 user.userName = to_string(userId);
1809 else {
1810 string userName = nameNode->value();
1811 user.userName = userName + " (" + to_string(userId) + ")";
1812 }
1813 }
1814 }
1815
1816 string filename;
1817 user.type = Get_Password_Type(userId, filename);
1818
1819 user.isDecrypted = false;
1820 if (strcmp(user_check_result, "1") == 0)
1821 user.isDecrypted = true;
1822 Users_List.push_back(user);
1823 }
1824 }
1825 Check_Users_Decryption_Status();
1826#endif
1827}
1828
1829std::vector<users_struct>* TWPartitionManager::Get_Users_List() {
1830 return &Users_List;
1831}
1832
1833void TWPartitionManager::Mark_User_Decrypted(int userID) {
1834#ifdef TW_INCLUDE_FBE
1835 std::vector<users_struct>::iterator iter;
1836 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1837 if (atoi((*iter).userId.c_str()) == userID) {
1838 (*iter).isDecrypted = true;
1839 string user_prop_decrypted = "twrp.user." + to_string(userID) + ".decrypt";
1840 property_set(user_prop_decrypted.c_str(), "1");
1841 break;
1842 }
1843 }
1844 Check_Users_Decryption_Status();
1845#endif
1846}
1847
1848void TWPartitionManager::Check_Users_Decryption_Status() {
1849#ifdef TW_INCLUDE_FBE
1850 int all_is_decrypted = 1;
1851 std::vector<users_struct>::iterator iter;
1852 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1853 if (!(*iter).isDecrypted) {
1854 LOGINFO("User %s is not decrypted.\n", (*iter).userId.c_str());
1855 all_is_decrypted = 0;
1856 break;
1857 }
1858 }
1859 if (all_is_decrypted == 1) {
1860 LOGINFO("All found users are decrypted.\n");
1861 DataManager::SetValue("tw_all_users_decrypted", "1");
1862 property_set("twrp.all.users.decrypted", "true");
1863 } else
1864 DataManager::SetValue("tw_all_users_decrypted", "0");
1865#endif
1866}
1867
1868int TWPartitionManager::Decrypt_Device(string Password, int user_id) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001869#ifdef TW_INCLUDE_CRYPTO
Ethan Yonkerddb63e22017-01-19 14:01:57 -06001870 char crypto_state[PROPERTY_VALUE_MAX], crypto_blkdev[PROPERTY_VALUE_MAX];
Ethan Yonker253368a2014-11-25 15:00:52 -06001871 std::vector<TWPartition*>::iterator iter;
Dees_Troy5bf43922012-09-07 16:07:55 -04001872
Ethan Yonker253368a2014-11-25 15:00:52 -06001873 // Mount any partitions that need to be mounted for decrypt
1874 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1875 if ((*iter)->Mount_To_Decrypt) {
1876 (*iter)->Mount(true);
1877 }
a39552696ff55ce2013-01-08 16:14:56 +00001878 }
oshmouna82a7542018-04-10 17:45:55 +02001879 property_set("twrp.mount_to_decrypt", "1");
a39552696ff55ce2013-01-08 16:14:56 +00001880
Ethan Yonkera1de1492015-11-04 11:58:27 -06001881 property_get("ro.crypto.state", crypto_state, "error");
1882 if (strcmp(crypto_state, "error") == 0) {
Captain Throwback18aea272020-10-29 12:43:17 -04001883 Set_FDE_Encrypt_Status();
Ethan Yonkera1de1492015-11-04 11:58:27 -06001884 }
1885
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001886 if (DataManager::GetIntValue(TW_IS_FBE)) {
1887#ifdef TW_INCLUDE_FBE
1888 if (!Mount_By_Path("/data", true)) // /data has to be mounted for FBE
1889 return -1;
Noah Jacobson81d638d2019-04-28 00:10:07 -04001890
1891 bool user_need_decrypt = false;
1892 std::vector<users_struct>::iterator iter;
1893 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1894 if (atoi((*iter).userId.c_str()) == user_id && !(*iter).isDecrypted) {
1895 user_need_decrypt = true;
1896 }
1897 }
1898 if (!user_need_decrypt) {
1899 LOGINFO("User %d does not require decryption\n", user_id);
1900 return 0;
1901 }
1902
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001903 int retry_count = 10;
1904 while (!TWFunc::Path_Exists("/data/system/users/gatekeeper.password.key") && --retry_count)
Noah Jacobson81d638d2019-04-28 00:10:07 -04001905 usleep(2000); // A small sleep is needed after mounting /data to ensure reliable decrypt...maybe because of DE?
1906 gui_msg(Msg("decrypting_user_fbe=Attempting to decrypt FBE for user {1}...")(user_id));
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001907 if (Decrypt_User(user_id, Password)) {
Noah Jacobson81d638d2019-04-28 00:10:07 -04001908 gui_msg(Msg("decrypt_user_success_fbe=User {1} Decrypted Successfully")(user_id));
1909 Mark_User_Decrypted(user_id);
1910 if (user_id == 0) {
1911 // When decrypting user 0 also try all other users
1912 std::vector<users_struct>::iterator iter;
1913 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1914 if ((*iter).userId == "0" || (*iter).isDecrypted)
1915 continue;
1916
1917 int tmp_user_id = atoi((*iter).userId.c_str());
1918 gui_msg(Msg("decrypting_user_fbe=Attempting to decrypt FBE for user {1}...")(tmp_user_id));
1919 if (Decrypt_User(tmp_user_id, Password) ||
1920 (Password != "!" && Decrypt_User(tmp_user_id, "!"))) { // "!" means default password
1921 gui_msg(Msg("decrypt_user_success_fbe=User {1} Decrypted Successfully")(tmp_user_id));
1922 Mark_User_Decrypted(tmp_user_id);
1923 } else {
1924 gui_msg(Msg("decrypt_user_fail_fbe=Failed to decrypt user {1}")(tmp_user_id));
1925 }
1926 }
1927 Post_Decrypt("");
1928 }
1929
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001930 return 0;
Noah Jacobson81d638d2019-04-28 00:10:07 -04001931 } else {
1932 gui_msg(Msg(msg::kError, "decrypt_user_fail_fbe=Failed to decrypt user {1}")(user_id));
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001933 }
1934#else
1935 LOGERR("FBE support is not present\n");
1936#endif
1937 return -1;
1938 }
1939
Noah Jacobson81d638d2019-04-28 00:10:07 -04001940 char isdecrypteddata[PROPERTY_VALUE_MAX];
1941 property_get("twrp.decrypt.done", isdecrypteddata, "");
1942 if (strcmp(isdecrypteddata, "true") == 0) {
1943 LOGINFO("Data has no decryption required\n");
1944 return 0;
1945 }
1946
Ethan Yonkerddb63e22017-01-19 14:01:57 -06001947 int pwret = -1;
1948 pid_t pid = fork();
1949 if (pid < 0) {
1950 LOGERR("fork failed\n");
1951 return -1;
1952 } else if (pid == 0) {
1953 // Child process
1954 char cPassword[255];
1955 strcpy(cPassword, Password.c_str());
1956 int ret = cryptfs_check_passwd(cPassword);
1957 exit(ret);
1958 } else {
1959 // Parent
1960 int status;
1961 if (TWFunc::Wait_For_Child_Timeout(pid, &status, "Decrypt", 30))
1962 pwret = -1;
1963 else
1964 pwret = WEXITSTATUS(status) ? -1 : 0;
1965 }
a39552696ff55ce2013-01-08 16:14:56 +00001966
nkk7171c6c502017-01-05 23:55:05 +02001967#ifdef TW_CRYPTO_USE_SYSTEM_VOLD
1968 if (pwret != 0) {
1969 pwret = vold_decrypt(Password);
nkk717d1222a2017-11-06 17:27:21 +02001970 switch (pwret) {
1971 case VD_SUCCESS:
1972 break;
1973 case VD_ERR_MISSING_VDC:
1974 gui_msg(Msg(msg::kError, "decrypt_data_vold_os_missing=Missing files needed for vold decrypt: {1}")("/system/bin/vdc"));
1975 break;
1976 case VD_ERR_MISSING_VOLD:
1977 gui_msg(Msg(msg::kError, "decrypt_data_vold_os_missing=Missing files needed for vold decrypt: {1}")("/system/bin/vold"));
1978 break;
1979 }
nkk7171c6c502017-01-05 23:55:05 +02001980 }
1981#endif // TW_CRYPTO_USE_SYSTEM_VOLD
1982
Ethan Yonker253368a2014-11-25 15:00:52 -06001983 // Unmount any partitions that were needed for decrypt
1984 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1985 if ((*iter)->Mount_To_Decrypt) {
1986 (*iter)->UnMount(false);
1987 }
1988 }
oshmouna82a7542018-04-10 17:45:55 +02001989 property_set("twrp.mount_to_decrypt", "0");
Ethan Yonker253368a2014-11-25 15:00:52 -06001990
a39552696ff55ce2013-01-08 16:14:56 +00001991 if (pwret != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001992 gui_err("fail_decrypt=Failed to decrypt data.");
Dees_Troy5bf43922012-09-07 16:07:55 -04001993 return -1;
1994 }
a39552696ff55ce2013-01-08 16:14:56 +00001995
Dees_Troy5bf43922012-09-07 16:07:55 -04001996 property_get("ro.crypto.fs_crypto_blkdev", crypto_blkdev, "error");
1997 if (strcmp(crypto_blkdev, "error") == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001998 LOGERR("Error retrieving decrypted data block device.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04001999 } else {
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06002000 Post_Decrypt(crypto_blkdev);
Dees_Troy5bf43922012-09-07 16:07:55 -04002001 }
2002 return 0;
2003#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002004 gui_err("no_crypto_support=No crypto support was compiled into this build.");
Dees_Troy5bf43922012-09-07 16:07:55 -04002005 return -1;
2006#endif
Dees_Troy51a0e822012-09-05 15:24:24 -04002007 return 1;
Dees_Troy51127312012-09-08 13:08:49 -04002008}
2009
Ethan Yonkerb5fab762016-01-28 14:03:33 -06002010int TWPartitionManager::Fix_Contexts(void) {
Ethan Yonkerb5fab762016-01-28 14:03:33 -06002011 std::vector<TWPartition*>::iterator iter;
2012 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2013 if ((*iter)->Has_Data_Media) {
2014 if ((*iter)->Mount(true)) {
2015 if (fixContexts::fixDataMediaContexts((*iter)->Mount_Point) != 0)
2016 return -1;
2017 }
2018 }
2019 }
Dees_Troy1a650e62012-10-19 20:54:32 -04002020 UnMount_Main_Partitions();
Ethan Yonker74db1572015-10-28 12:44:49 -05002021 gui_msg("done=Done.");
Ethan Yonkerb5fab762016-01-28 14:03:33 -06002022 return 0;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04002023}
2024
Ethan Yonker66a19492015-12-10 10:19:45 -06002025TWPartition* TWPartitionManager::Find_Next_Storage(string Path, bool Exclude_Data_Media) {
Ethan Yonker47360be2014-04-01 10:34:34 -05002026 std::vector<TWPartition*>::iterator iter = Partitions.begin();
2027
2028 if (!Path.empty()) {
2029 string Search_Path = TWFunc::Get_Root_Path(Path);
2030 for (; iter != Partitions.end(); iter++) {
Matt Mower9a561dd2016-02-22 21:25:51 -06002031 if ((*iter)->Mount_Point == Search_Path) {
Ethan Yonker47360be2014-04-01 10:34:34 -05002032 iter++;
2033 break;
2034 }
2035 }
2036 }
2037
2038 for (; iter != Partitions.end(); iter++) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002039 if (Exclude_Data_Media && (*iter)->Has_Data_Media) {
2040 // do nothing, do not return this type of partition
2041 } else if ((*iter)->Is_Storage && (*iter)->Is_Present) {
Ethan Yonker47360be2014-04-01 10:34:34 -05002042 return (*iter);
2043 }
2044 }
2045
2046 return NULL;
2047}
2048
Dees_Troyd21618c2012-10-14 18:48:49 -04002049int TWPartitionManager::Open_Lun_File(string Partition_Path, string Lun_File) {
Dees_Troyd21618c2012-10-14 18:48:49 -04002050 TWPartition* Part = Find_Partition_By_Path(Partition_Path);
2051
2052 if (Part == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002053 LOGINFO("Unable to locate '%s' for USB storage mode.", Partition_Path.c_str());
2054 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 -04002055 return false;
2056 }
Ethan Yonker47360be2014-04-01 10:34:34 -05002057 LOGINFO("USB mount '%s', '%s' > '%s'\n", Partition_Path.c_str(), Part->Actual_Block_Device.c_str(), Lun_File.c_str());
2058 if (!Part->UnMount(true) || !Part->Is_Present)
Dees_Troyd21618c2012-10-14 18:48:49 -04002059 return false;
2060
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002061 if (TWFunc::write_to_file(Lun_File, Part->Actual_Block_Device)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00002062 LOGERR("Unable to write to ums lunfile '%s': (%s)\n", Lun_File.c_str(), strerror(errno));
Dees_Troyd21618c2012-10-14 18:48:49 -04002063 return false;
2064 }
Dees_Troyd21618c2012-10-14 18:48:49 -04002065 return true;
2066}
2067
Dees_Troy8170a922012-09-18 15:40:25 -04002068int TWPartitionManager::usb_storage_enable(void) {
Dees_Troy8170a922012-09-18 15:40:25 -04002069 char lun_file[255];
Dees_Troyd21618c2012-10-14 18:48:49 -04002070 bool has_multiple_lun = false;
Dees_Troy8170a922012-09-18 15:40:25 -04002071
Ethan Yonker47360be2014-04-01 10:34:34 -05002072 string Lun_File_str = CUSTOM_LUN_FILE;
2073 size_t found = Lun_File_str.find("%");
2074 if (found != string::npos) {
2075 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
2076 if (TWFunc::Path_Exists(lun_file))
2077 has_multiple_lun = true;
2078 }
Ethan Yonker726a0202014-12-16 20:01:38 -06002079 mtp_was_enabled = TWFunc::Toggle_MTP(false); // Must disable MTP for USB Storage
Ethan Yonker47360be2014-04-01 10:34:34 -05002080 if (!has_multiple_lun) {
2081 LOGINFO("Device doesn't have multiple lun files, mount current storage\n");
2082 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
2083 if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) == "/data") {
Ethan Yonker66a19492015-12-10 10:19:45 -06002084 TWPartition* Mount = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05002085 if (Mount) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002086 if (!Open_Lun_File(Mount->Mount_Point, lun_file)) {
2087 goto error_handle;
2088 }
Ethan Yonker47360be2014-04-01 10:34:34 -05002089 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002090 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002091 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05002092 }
2093 } else if (!Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file)) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002094 goto error_handle;
Dees_Troy8170a922012-09-18 15:40:25 -04002095 }
Dees_Troy8170a922012-09-18 15:40:25 -04002096 } else {
Ethan Yonker47360be2014-04-01 10:34:34 -05002097 LOGINFO("Device has multiple lun files\n");
2098 TWPartition* Mount1;
2099 TWPartition* Mount2;
Dees_Troy8170a922012-09-18 15:40:25 -04002100 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
Ethan Yonker66a19492015-12-10 10:19:45 -06002101 Mount1 = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05002102 if (Mount1) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002103 if (!Open_Lun_File(Mount1->Mount_Point, lun_file)) {
2104 goto error_handle;
2105 }
Matt Moweree717062014-04-26 01:46:46 -05002106 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
Ethan Yonker66a19492015-12-10 10:19:45 -06002107 Mount2 = Find_Next_Storage(Mount1->Mount_Point, true);
Matt Mower1fdcdb72016-01-25 20:53:47 -06002108 if (Mount2 && Mount2->Mount_Point != Mount1->Mount_Point) {
Matt Moweree717062014-04-26 01:46:46 -05002109 Open_Lun_File(Mount2->Mount_Point, lun_file);
Ethan Yonker47360be2014-04-01 10:34:34 -05002110 }
2111 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002112 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002113 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05002114 }
Dees_Troy8170a922012-09-18 15:40:25 -04002115 }
Matt Mowerb6ef4782014-11-06 02:24:36 -06002116 property_set("sys.storage.ums_enabled", "1");
HandyMenny37d42992014-10-26 22:15:59 +01002117 property_set("sys.usb.config", "mass_storage,adb");
Dees_Troy8170a922012-09-18 15:40:25 -04002118 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002119error_handle:
2120 if (mtp_was_enabled)
2121 if (!Enable_MTP())
2122 Disable_MTP();
2123 return false;
Dees_Troy8170a922012-09-18 15:40:25 -04002124}
2125
2126int TWPartitionManager::usb_storage_disable(void) {
Dees_Troy2673cec2013-04-02 20:22:16 +00002127 int index, ret;
2128 char lun_file[255], ch[2] = {0, 0};
2129 string str = ch;
Dees_Troy8170a922012-09-18 15:40:25 -04002130
2131 for (index=0; index<2; index++) {
2132 sprintf(lun_file, CUSTOM_LUN_FILE, index);
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002133 ret = TWFunc::write_to_file(lun_file, str);
Dees_Troy2673cec2013-04-02 20:22:16 +00002134 if (ret < 0) {
2135 break;
Dees_Troy8170a922012-09-18 15:40:25 -04002136 }
Dees_Troy8170a922012-09-18 15:40:25 -04002137 }
Dees_Troye58d5262012-09-21 12:27:57 -04002138 Mount_All_Storage();
2139 Update_System_Details();
Dees_Troycfd73ef2012-10-12 16:52:00 -04002140 UnMount_Main_Partitions();
Matt Mowerd9cb9062013-12-14 20:34:45 -06002141 property_set("sys.storage.ums_enabled", "0");
HandyMenny37d42992014-10-26 22:15:59 +01002142 property_set("sys.usb.config", "adb");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002143 if (mtp_was_enabled)
2144 if (!Enable_MTP())
2145 Disable_MTP();
Dees_Troy2673cec2013-04-02 20:22:16 +00002146 if (ret < 0 && index == 0) {
2147 LOGERR("Unable to write to ums lunfile '%s'.", lun_file);
2148 return false;
2149 } else {
2150 return true;
2151 }
Dees_Troy8170a922012-09-18 15:40:25 -04002152 return true;
Dees_Troy812660f2012-09-20 09:55:17 -04002153}
2154
2155void TWPartitionManager::Mount_All_Storage(void) {
2156 std::vector<TWPartition*>::iterator iter;
2157
2158 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2159 if ((*iter)->Is_Storage)
2160 (*iter)->Mount(false);
2161 }
Dees_Troy2c50e182012-09-26 20:05:28 -04002162}
Dees_Troy9350b8d2012-09-27 12:38:38 -04002163
Dees_Troyd0384ef2012-10-12 12:15:42 -04002164void TWPartitionManager::UnMount_Main_Partitions(void) {
2165 // Unmounts system and data if data is not data/media
2166 // Also unmounts boot if boot is mountable
Dees_Troy2673cec2013-04-02 20:22:16 +00002167 LOGINFO("Unmounting main partitions...\n");
Dees_Troyd0384ef2012-10-12 12:15:42 -04002168
Mohd Faraze3948ec2020-08-14 01:40:59 +00002169 TWPartition *Partition = Find_Partition_By_Path ("/vendor");
Dees_Troyd0384ef2012-10-12 12:15:42 -04002170
Mohd Faraze3948ec2020-08-14 01:40:59 +00002171 if (Partition != NULL) UnMount_By_Path("/vendor", false);
2172 UnMount_By_Path (Get_Android_Root_Path(), true);
2173 Partition = Find_Partition_By_Path ("/product");
2174 if (Partition != NULL) UnMount_By_Path("/product", false);
Ethan Yonker6277c792014-09-15 14:54:30 -05002175 if (!datamedia)
2176 UnMount_By_Path("/data", true);
2177
Mohd Faraze3948ec2020-08-14 01:40:59 +00002178 Partition = Find_Partition_By_Path ("/boot");
2179 if (Partition != NULL && Partition->Can_Be_Mounted)
2180 Partition->UnMount(true);
Dees_Troyd0384ef2012-10-12 12:15:42 -04002181}
2182
Dees_Troy9350b8d2012-09-27 12:38:38 -04002183int TWPartitionManager::Partition_SDCard(void) {
Matt Mower23d8aae2017-01-06 14:30:33 -06002184 char temp[255];
Ethan Yonker483e9f42016-01-11 22:21:18 -06002185 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 -04002186 int ext, swap, total_size = 0, fat_size;
Dees_Troy9350b8d2012-09-27 12:38:38 -04002187
Ethan Yonker74db1572015-10-28 12:44:49 -05002188 gui_msg("start_partition_sd=Partitioning SD Card...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06002189
2190 // Locate and validate device to partition
2191 TWPartition* SDCard = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
2192
Ethan Yonker66a19492015-12-10 10:19:45 -06002193 if (SDCard->Is_Adopted_Storage)
2194 SDCard->Revert_Adopted();
2195
Ethan Yonker1eff6cd2014-09-15 13:30:42 -05002196 if (SDCard == NULL || !SDCard->Removable || SDCard->Has_Data_Media) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002197 gui_err("partition_sd_locate=Unable to locate device to partition.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002198 return false;
2199 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002200
2201 // Unmount everything
Dees_Troy9350b8d2012-09-27 12:38:38 -04002202 if (!SDCard->UnMount(true))
2203 return false;
2204 TWPartition* SDext = Find_Partition_By_Path("/sd-ext");
2205 if (SDext != NULL) {
2206 if (!SDext->UnMount(true))
2207 return false;
2208 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002209 char* swappath = getenv("SWAPPATH");
2210 if (swappath != NULL) {
2211 LOGINFO("Unmounting swap at '%s'\n", swappath);
2212 umount(swappath);
2213 }
Vojtech Bocek05534202013-09-11 08:11:56 +02002214
Ethan Yonker483e9f42016-01-11 22:21:18 -06002215 // Determine block device
2216 if (SDCard->Alternate_Block_Device.empty()) {
2217 SDCard->Find_Actual_Block_Device();
2218 Device = SDCard->Actual_Block_Device;
2219 // Just use the root block device
2220 Device.resize(strlen("/dev/block/mmcblkX"));
2221 } else {
2222 Device = SDCard->Alternate_Block_Device;
2223 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04002224
2225 // Find the size of the block device:
Ethan Yonker483e9f42016-01-11 22:21:18 -06002226 total_size = (int)(TWFunc::IOCTL_Get_Block_Size(Device.c_str()) / (1048576));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002227
2228 DataManager::GetValue("tw_sdext_size", ext);
2229 DataManager::GetValue("tw_swap_size", swap);
2230 DataManager::GetValue("tw_sdpart_file_system", ext_format);
2231 fat_size = total_size - ext - swap;
Ethan Yonker483e9f42016-01-11 22:21:18 -06002232 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);
2233
2234 // Determine partition sizes
2235 if (swap == 0 && ext == 0) {
2236 fat_str = "-0";
2237 } else {
2238 memset(temp, 0, sizeof(temp));
2239 sprintf(temp, "%i", fat_size);
2240 fat_str = temp;
2241 fat_str += "MB";
2242 }
2243 if (swap == 0) {
2244 ext_str = "-0";
2245 } else {
2246 memset(temp, 0, sizeof(temp));
2247 sprintf(temp, "%i", ext);
2248 ext_str = "+";
2249 ext_str += temp;
2250 ext_str += "MB";
2251 }
2252
Dees_Troy9350b8d2012-09-27 12:38:38 -04002253 if (ext + swap > total_size) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002254 gui_err("ext_swap_size=EXT + Swap size is larger than sdcard size.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002255 return false;
2256 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002257
Ethan Yonker74db1572015-10-28 12:44:49 -05002258 gui_msg("remove_part_table=Removing partition table...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06002259 Command = "sgdisk --zap-all " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002260 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002261 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002262 gui_err("unable_rm_part=Unable to remove partition table.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002263 Update_System_Details();
2264 return false;
2265 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002266 gui_msg(Msg("create_part=Creating {1} partition...")("FAT32"));
Captain Throwback9643a802016-05-27 11:44:51 -04002267 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 +00002268 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002269 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002270 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("FAT32"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002271 return false;
2272 }
2273 if (ext > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002274 gui_msg(Msg("create_part=Creating {1} partition...")("EXT"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06002275 Command = "sgdisk --new=0:0:" + ext_str + " --change-name=0:\"Linux filesystem\" " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002276 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002277 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002278 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("EXT"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002279 Update_System_Details();
2280 return false;
2281 }
2282 }
2283 if (swap > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002284 gui_msg(Msg("create_part=Creating {1} partition...")("swap"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06002285 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 +00002286 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002287 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002288 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("swap"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002289 Update_System_Details();
2290 return false;
2291 }
2292 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002293
2294 // Convert GPT to MBR
2295 Command = "sgdisk --gpttombr " + Device;
2296 if (TWFunc::Exec_Cmd(Command) != 0)
2297 LOGINFO("Failed to covert partition GPT to MBR\n");
2298
2299 // Tell the kernel to rescan the partition table
2300 int fd = open(Device.c_str(), O_RDONLY);
2301 ioctl(fd, BLKRRPART, 0);
2302 close(fd);
2303
2304 string format_device = Device;
2305 if (Device.substr(0, 17) == "/dev/block/mmcblk")
2306 format_device += "p";
2307
2308 // Format new partitions to proper file system
2309 if (fat_size > 0) {
2310 Command = "mkfs.fat " + format_device + "1";
2311 TWFunc::Exec_Cmd(Command);
2312 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04002313 if (ext > 0) {
2314 if (SDext == NULL) {
Ethan Yonker483e9f42016-01-11 22:21:18 -06002315 Command = "mke2fs -t " + ext_format + " -m 0 " + format_device + "2";
2316 gui_msg(Msg("format_sdext_as=Formatting sd-ext as {1}...")(ext_format));
2317 LOGINFO("Formatting sd-ext after partitioning, command: '%s'\n", Command.c_str());
2318 TWFunc::Exec_Cmd(Command);
2319 } else {
2320 SDext->Wipe(ext_format);
Dees_Troy9350b8d2012-09-27 12:38:38 -04002321 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002322 }
2323 if (swap > 0) {
2324 Command = "mkswap " + format_device;
2325 if (ext > 0)
2326 Command += "3";
2327 else
2328 Command += "2";
Vojtech Bocek05534202013-09-11 08:11:56 +02002329 TWFunc::Exec_Cmd(Command);
Dees_Troy9350b8d2012-09-27 12:38:38 -04002330 }
2331
Ethan Yonker483e9f42016-01-11 22:21:18 -06002332 // recreate TWRP folder and rewrite settings - these will be gone after sdcard is partitioned
2333 if (SDCard->Mount(true)) {
2334 string TWRP_Folder = SDCard->Mount_Point + "/TWRP";
2335 mkdir(TWRP_Folder.c_str(), 0777);
2336 DataManager::Flush();
2337 }
2338
Dees_Troy9350b8d2012-09-27 12:38:38 -04002339 Update_System_Details();
Ethan Yonker74db1572015-10-28 12:44:49 -05002340 gui_msg("part_complete=Partitioning complete.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002341 return true;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04002342}
Dees_Troya13d74f2013-03-24 08:54:55 -05002343
2344void TWPartitionManager::Get_Partition_List(string ListType, std::vector<PartitionList> *Partition_List) {
2345 std::vector<TWPartition*>::iterator iter;
2346 if (ListType == "mount") {
2347 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002348 if ((*iter)->Can_Be_Mounted) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002349 struct PartitionList part;
2350 part.Display_Name = (*iter)->Display_Name;
2351 part.Mount_Point = (*iter)->Mount_Point;
2352 part.selected = (*iter)->Is_Mounted();
2353 Partition_List->push_back(part);
2354 }
2355 }
2356 } else if (ListType == "storage") {
2357 char free_space[255];
2358 string Current_Storage = DataManager::GetCurrentStoragePath();
2359 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2360 if ((*iter)->Is_Storage) {
2361 struct PartitionList part;
2362 sprintf(free_space, "%llu", (*iter)->Free / 1024 / 1024);
2363 part.Display_Name = (*iter)->Storage_Name + " (";
2364 part.Display_Name += free_space;
2365 part.Display_Name += "MB)";
2366 part.Mount_Point = (*iter)->Storage_Path;
2367 if ((*iter)->Storage_Path == Current_Storage)
2368 part.selected = 1;
2369 else
2370 part.selected = 0;
2371 Partition_List->push_back(part);
2372 }
2373 }
2374 } else if (ListType == "backup") {
2375 char backup_size[255];
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002376 unsigned long long Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -05002377 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002378 if ((*iter)->Can_Be_Backed_Up && !(*iter)->Is_SubPartition && (*iter)->Is_Present) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002379 struct PartitionList part;
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002380 Backup_Size = (*iter)->Backup_Size;
2381 if ((*iter)->Has_SubPartition) {
2382 std::vector<TWPartition*>::iterator subpart;
2383
2384 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
2385 if ((*subpart)->Is_SubPartition && (*subpart)->Can_Be_Backed_Up && (*subpart)->Is_Present && (*subpart)->SubPartition_Of == (*iter)->Mount_Point)
2386 Backup_Size += (*subpart)->Backup_Size;
2387 }
2388 }
2389 sprintf(backup_size, "%llu", Backup_Size / 1024 / 1024);
Dees_Troya13d74f2013-03-24 08:54:55 -05002390 part.Display_Name = (*iter)->Backup_Display_Name + " (";
2391 part.Display_Name += backup_size;
2392 part.Display_Name += "MB)";
2393 part.Mount_Point = (*iter)->Backup_Path;
2394 part.selected = 0;
2395 Partition_List->push_back(part);
2396 }
2397 }
2398 } else if (ListType == "restore") {
2399 string Restore_List, restore_path;
2400 TWPartition* restore_part = NULL;
2401
2402 DataManager::GetValue("tw_restore_list", Restore_List);
2403 if (!Restore_List.empty()) {
2404 size_t start_pos = 0, end_pos = Restore_List.find(";", start_pos);
2405 while (end_pos != string::npos && start_pos < Restore_List.size()) {
2406 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04002407 struct PartitionList part;
2408 if (restore_path.compare("ADB_Backup") == 0) {
2409 part.Display_Name = "ADB Backup";
2410 part.Mount_Point = "ADB Backup";
2411 part.selected = 1;
2412 Partition_List->push_back(part);
2413 break;
2414 }
Dees_Troy59df9262013-06-19 14:53:57 -05002415 if ((restore_part = Find_Partition_By_Path(restore_path)) != NULL) {
Dees Troy4159aed2014-02-28 17:24:43 +00002416 if ((restore_part->Backup_Name == "recovery" && !restore_part->Can_Be_Backed_Up) || restore_part->Is_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002417 // Don't allow restore of recovery (causes problems on some devices)
Dees_Troy59df9262013-06-19 14:53:57 -05002418 // Don't add subpartitions to the list of items
Dees_Troya13d74f2013-03-24 08:54:55 -05002419 } else {
Dees_Troya13d74f2013-03-24 08:54:55 -05002420 part.Display_Name = restore_part->Backup_Display_Name;
2421 part.Mount_Point = restore_part->Backup_Path;
2422 part.selected = 1;
2423 Partition_List->push_back(part);
2424 }
2425 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002426 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05002427 }
2428 start_pos = end_pos + 1;
2429 end_pos = Restore_List.find(";", start_pos);
2430 }
2431 }
2432 } else if (ListType == "wipe") {
2433 struct PartitionList dalvik;
Ethan Yonker74db1572015-10-28 12:44:49 -05002434 dalvik.Display_Name = gui_parse_text("{@dalvik}");
Dees_Troya13d74f2013-03-24 08:54:55 -05002435 dalvik.Mount_Point = "DALVIK";
2436 dalvik.selected = 0;
2437 Partition_List->push_back(dalvik);
2438 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2439 if ((*iter)->Wipe_Available_in_GUI && !(*iter)->Is_SubPartition) {
2440 struct PartitionList part;
2441 part.Display_Name = (*iter)->Display_Name;
2442 part.Mount_Point = (*iter)->Mount_Point;
2443 part.selected = 0;
2444 Partition_List->push_back(part);
2445 }
2446 if ((*iter)->Has_Android_Secure) {
2447 struct PartitionList part;
2448 part.Display_Name = (*iter)->Backup_Display_Name;
2449 part.Mount_Point = (*iter)->Backup_Path;
2450 part.selected = 0;
2451 Partition_List->push_back(part);
2452 }
Dees_Troy74fb2e92013-04-15 14:35:47 +00002453 if ((*iter)->Has_Data_Media) {
2454 struct PartitionList datamedia;
2455 datamedia.Display_Name = (*iter)->Storage_Name;
2456 datamedia.Mount_Point = "INTERNAL";
2457 datamedia.selected = 0;
2458 Partition_List->push_back(datamedia);
2459 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002460 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002461 } else if (ListType == "flashimg") {
2462 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2463 if ((*iter)->Can_Flash_Img && (*iter)->Is_Present) {
2464 struct PartitionList part;
2465 part.Display_Name = (*iter)->Backup_Display_Name;
2466 part.Mount_Point = (*iter)->Backup_Path;
2467 part.selected = 0;
2468 Partition_List->push_back(part);
2469 }
2470 }
Ethan Yonker53796e72019-01-11 22:49:52 -06002471 if (DataManager::GetIntValue("tw_has_repack_tools") != 0 && DataManager::GetIntValue("tw_has_boot_slots") != 0) {
2472 TWPartition* boot = Find_Partition_By_Path("/boot");
2473 if (boot) {
2474 // Allow flashing kernels and ramdisks
2475 struct PartitionList repack_ramdisk;
2476 repack_ramdisk.Display_Name = gui_lookup("install_twrp_ramdisk", "Install Recovery Ramdisk");
2477 repack_ramdisk.Mount_Point = "/repack_ramdisk";
2478 repack_ramdisk.selected = 0;
2479 Partition_List->push_back(repack_ramdisk);
2480 /*struct PartitionList repack_kernel; For now let's leave repacking kernels under advanced only
2481 repack_kernel.Display_Name = gui_lookup("install_kernel", "Install Kernel");
2482 repack_kernel.Mount_Point = "/repack_kernel";
2483 repack_kernel.selected = 0;
2484 Partition_List->push_back(repack_kernel);*/
2485 }
2486 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002487 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00002488 LOGERR("Unknown list type '%s' requested for TWPartitionManager::Get_Partition_List\n", ListType.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -05002489 }
2490}
2491
2492int TWPartitionManager::Fstab_Processed(void) {
2493 return Partitions.size();
2494}
Dees_Troyd93bda52013-07-03 19:55:19 +00002495
2496void TWPartitionManager::Output_Storage_Fstab(void) {
2497 std::vector<TWPartition*>::iterator iter;
2498 char storage_partition[255];
bigbiff bigbiffe4bdb152019-03-23 18:33:17 -04002499 std::string Temp;
bigbiff25d25b92020-06-19 16:07:38 -04002500 std::string cacheDir = TWFunc::get_log_dir();
bigbiff bigbiffe4bdb152019-03-23 18:33:17 -04002501
2502 if (cacheDir.empty()) {
2503 LOGINFO("Unable to find cache directory\n");
2504 return;
2505 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002506
bigbiff25d25b92020-06-19 16:07:38 -04002507 std::string storageFstab = TWFunc::get_log_dir() + "recovery/storage.fstab";
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002508 FILE *fp = fopen(storageFstab.c_str(), "w");
Dees_Troyd93bda52013-07-03 19:55:19 +00002509
2510 if (fp == NULL) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002511 gui_msg(Msg(msg::kError, "unable_to_open=Unable to open '{1}'.")(storageFstab));
Dees_Troyd93bda52013-07-03 19:55:19 +00002512 return;
2513 }
2514
2515 // Iterate through all partitions
2516 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2517 if ((*iter)->Is_Storage) {
2518 Temp = (*iter)->Storage_Path + ";" + (*iter)->Storage_Name + ";\n";
2519 strcpy(storage_partition, Temp.c_str());
2520 fwrite(storage_partition, sizeof(storage_partition[0]), strlen(storage_partition) / sizeof(storage_partition[0]), fp);
2521 }
2522 }
2523 fclose(fp);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02002524}
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002525
2526TWPartition *TWPartitionManager::Get_Default_Storage_Partition()
2527{
2528 TWPartition *res = NULL;
2529 for (std::vector<TWPartition*>::iterator iter = Partitions.begin(); iter != Partitions.end(); ++iter) {
Matt Mowera8a89d12016-12-30 18:10:37 -06002530 if (!(*iter)->Is_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002531 continue;
2532
Matt Mowera8a89d12016-12-30 18:10:37 -06002533 if ((*iter)->Is_Settings_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002534 return *iter;
2535
Matt Mowera8a89d12016-12-30 18:10:37 -06002536 if (!res)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002537 res = *iter;
2538 }
2539 return res;
2540}
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002541
2542bool TWPartitionManager::Enable_MTP(void) {
2543#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002544 if (mtppid) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002545 gui_err("mtp_already_enabled=MTP already enabled");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002546 return true;
2547 }
Ethan Yonker726a0202014-12-16 20:01:38 -06002548
2549 int mtppipe[2];
2550
2551 if (pipe(mtppipe) < 0) {
2552 LOGERR("Error creating MTP pipe\n");
2553 return false;
2554 }
2555
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002556 char old_value[PROPERTY_VALUE_MAX];
Matt Mowere07f0102017-02-23 17:40:00 -06002557 property_get("sys.usb.config", old_value, "");
2558 if (strcmp(old_value, "mtp,adb") != 0) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002559 char vendor[PROPERTY_VALUE_MAX];
2560 char product[PROPERTY_VALUE_MAX];
2561 property_set("sys.usb.config", "none");
2562 property_get("usb.vendor", vendor, "18D1");
2563 property_get("usb.product.mtpadb", product, "4EE2");
2564 string vendorstr = vendor;
2565 string productstr = product;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002566 TWFunc::write_to_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2567 TWFunc::write_to_file("/sys/class/android_usb/android0/idProduct", productstr);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002568 property_set("sys.usb.config", "mtp,adb");
2569 }
Matt Mower653a1702017-02-16 10:38:52 -06002570 /* To enable MTP debug, use the twrp command line feature:
Ethan Yonker6d154c42014-09-03 14:19:43 -05002571 * twrp set tw_mtp_debug 1
2572 */
2573 twrpMtp *mtp = new twrpMtp(DataManager::GetIntValue("tw_mtp_debug"));
Ethan Yonker1b039202015-01-30 10:08:48 -06002574 mtppid = mtp->forkserver(mtppipe);
2575 if (mtppid) {
2576 close(mtppipe[0]); // Host closes read side
2577 mtp_write_fd = mtppipe[1];
2578 DataManager::SetValue("tw_mtp_enabled", 1);
2579 Add_All_MTP_Storage();
2580 return true;
Ethan Yonker726a0202014-12-16 20:01:38 -06002581 } else {
2582 close(mtppipe[0]);
2583 close(mtppipe[1]);
Ethan Yonker74db1572015-10-28 12:44:49 -05002584 gui_err("mtp_fail=Failed to enable MTP");
Ethan Yonker1b039202015-01-30 10:08:48 -06002585 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002586 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002587#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002588 gui_err("no_mtp=MTP support not included");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002589#endif
2590 DataManager::SetValue("tw_mtp_enabled", 0);
2591 return false;
2592}
2593
Ethan Yonker1b039202015-01-30 10:08:48 -06002594void TWPartitionManager::Add_All_MTP_Storage(void) {
2595#ifdef TW_HAS_MTP
2596 std::vector<TWPartition*>::iterator iter;
2597
2598 if (!mtppid)
2599 return; // MTP is not enabled
2600
2601 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2602 if ((*iter)->Is_Storage && (*iter)->Is_Present && (*iter)->Mount(false))
2603 Add_Remove_MTP_Storage((*iter), MTP_MESSAGE_ADD_STORAGE);
2604 }
2605#else
2606 return;
2607#endif
2608}
2609
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002610bool TWPartitionManager::Disable_MTP(void) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002611 char old_value[PROPERTY_VALUE_MAX];
bigbiff bigbiffaf32bb92018-12-18 18:39:53 -05002612 property_set("sys.usb.ffs.mtp.ready", "0");
Matt Mowere07f0102017-02-23 17:40:00 -06002613 property_get("sys.usb.config", old_value, "");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002614 if (strcmp(old_value, "adb") != 0) {
2615 char vendor[PROPERTY_VALUE_MAX];
2616 char product[PROPERTY_VALUE_MAX];
2617 property_set("sys.usb.config", "none");
2618 property_get("usb.vendor", vendor, "18D1");
Matt Mowere07f0102017-02-23 17:40:00 -06002619 property_get("usb.product.adb", product, "D001");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002620 string vendorstr = vendor;
2621 string productstr = product;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002622 TWFunc::write_to_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2623 TWFunc::write_to_file("/sys/class/android_usb/android0/idProduct", productstr);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002624 usleep(2000);
2625 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002626#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002627 if (mtppid) {
Ethan Yonker8613dc02014-09-11 09:28:20 -05002628 LOGINFO("Disabling MTP\n");
2629 int status;
2630 kill(mtppid, SIGKILL);
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002631 mtppid = 0;
Ethan Yonker8613dc02014-09-11 09:28:20 -05002632 // We don't care about the exit value, but this prevents a zombie process
2633 waitpid(mtppid, &status, 0);
Ethan Yonker726a0202014-12-16 20:01:38 -06002634 close(mtp_write_fd);
2635 mtp_write_fd = -1;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002636 }
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002637#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002638 property_set("sys.usb.config", "adb");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002639#ifdef TW_HAS_MTP
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002640 DataManager::SetValue("tw_mtp_enabled", 0);
2641 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002642#endif
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002643 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002644}
Ethan Yonker726a0202014-12-16 20:01:38 -06002645
2646TWPartition* TWPartitionManager::Find_Partition_By_MTP_Storage_ID(unsigned int Storage_ID) {
2647 std::vector<TWPartition*>::iterator iter;
2648
2649 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2650 if ((*iter)->MTP_Storage_ID == Storage_ID)
2651 return (*iter);
2652 }
2653 return NULL;
2654}
2655
2656bool TWPartitionManager::Add_Remove_MTP_Storage(TWPartition* Part, int message_type) {
2657#ifdef TW_HAS_MTP
2658 struct mtpmsg mtp_message;
2659
2660 if (!mtppid)
2661 return false; // MTP is disabled
2662
2663 if (mtp_write_fd < 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002664 LOGINFO("MTP: mtp_write_fd is not set\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002665 return false;
2666 }
2667
2668 if (Part) {
Ethan Yonker4bfabab2014-12-29 09:15:37 -06002669 if (Part->MTP_Storage_ID == 0)
2670 return false;
Ethan Yonker726a0202014-12-16 20:01:38 -06002671 if (message_type == MTP_MESSAGE_REMOVE_STORAGE) {
2672 mtp_message.message_type = MTP_MESSAGE_REMOVE_STORAGE; // Remove
2673 LOGINFO("sending message to remove %i\n", Part->MTP_Storage_ID);
2674 mtp_message.storage_id = Part->MTP_Storage_ID;
2675 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002676 LOGINFO("error sending message to remove storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002677 return false;
2678 } else {
2679 LOGINFO("Message sent, remove storage ID: %i\n", Part->MTP_Storage_ID);
2680 return true;
2681 }
2682 } else if (message_type == MTP_MESSAGE_ADD_STORAGE && Part->Is_Mounted()) {
2683 mtp_message.message_type = MTP_MESSAGE_ADD_STORAGE; // Add
2684 mtp_message.storage_id = Part->MTP_Storage_ID;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002685 if (Part->Storage_Path.size() >= sizeof(mtp_message.path)) {
2686 LOGERR("Storage path '%s' too large for mtpmsg\n", Part->Storage_Path.c_str());
2687 return false;
2688 }
2689 strcpy(mtp_message.path, Part->Storage_Path.c_str());
2690 if (Part->Storage_Name.size() >= sizeof(mtp_message.display)) {
2691 LOGERR("Storage name '%s' too large for mtpmsg\n", Part->Storage_Name.c_str());
2692 return false;
2693 }
2694 strcpy(mtp_message.display, Part->Storage_Name.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002695 mtp_message.maxFileSize = Part->Get_Max_FileSize();
Ethan Yonker1b039202015-01-30 10:08:48 -06002696 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 -06002697 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002698 LOGINFO("error sending message to add storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002699 return false;
2700 } else {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002701 LOGINFO("Message sent, add storage ID: %i '%s'\n", Part->MTP_Storage_ID, mtp_message.path);
Ethan Yonker726a0202014-12-16 20:01:38 -06002702 return true;
2703 }
2704 } else {
2705 LOGERR("Unknown MTP message type: %i\n", message_type);
2706 }
2707 } else {
2708 // This hopefully never happens as the error handling should
2709 // occur in the calling function.
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002710 LOGINFO("TWPartitionManager::Add_Remove_MTP_Storage NULL partition given\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002711 }
2712 return true;
2713#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002714 gui_err("no_mtp=MTP support not included");
Ethan Yonker726a0202014-12-16 20:01:38 -06002715 DataManager::SetValue("tw_mtp_enabled", 0);
2716 return false;
2717#endif
2718}
2719
2720bool TWPartitionManager::Add_MTP_Storage(string Mount_Point) {
2721#ifdef TW_HAS_MTP
2722 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2723 if (Part) {
2724 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2725 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002726 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002727 }
2728#endif
2729 return false;
2730}
2731
2732bool TWPartitionManager::Add_MTP_Storage(unsigned int Storage_ID) {
2733#ifdef TW_HAS_MTP
2734 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2735 if (Part) {
2736 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2737 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002738 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002739 }
2740#endif
2741 return false;
2742}
2743
2744bool TWPartitionManager::Remove_MTP_Storage(string Mount_Point) {
2745#ifdef TW_HAS_MTP
2746 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2747 if (Part) {
2748 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2749 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002750 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002751 }
2752#endif
2753 return false;
2754}
2755
2756bool TWPartitionManager::Remove_MTP_Storage(unsigned int Storage_ID) {
2757#ifdef TW_HAS_MTP
2758 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2759 if (Part) {
2760 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2761 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002762 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002763 }
2764#endif
2765 return false;
2766}
Ethan Yonker96af84a2015-01-05 14:58:36 -06002767
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002768bool TWPartitionManager::Flash_Image(string& path, string& filename) {
bigbiffad58e1b2020-07-06 20:24:34 -04002769 twrpRepacker repacker;
Matt Mower23d8aae2017-01-06 14:30:33 -06002770 int partition_count = 0;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002771 TWPartition* flash_part = NULL;
bigbiffce8f83c2015-12-12 18:30:21 -05002772 string Flash_List, flash_path, full_filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002773 size_t start_pos = 0, end_pos = 0;
2774
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002775 full_filename = path + "/" + filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002776
bigbiff1f9e4842020-10-31 11:33:15 -04002777 Unlock_Block_Partitions();
2778
bigbiffce8f83c2015-12-12 18:30:21 -05002779 gui_msg("image_flash_start=[IMAGE FLASH STARTED]");
2780 gui_msg(Msg("img_to_flash=Image to flash: '{1}'")(full_filename));
2781
2782 if (!TWFunc::Path_Exists(full_filename)) {
2783 if (!Mount_By_Path(full_filename, true)) {
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002784 return false;
2785 }
bigbiffce8f83c2015-12-12 18:30:21 -05002786 if (!TWFunc::Path_Exists(full_filename)) {
2787 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")(full_filename));
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002788 return false;
2789 }
2790 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002791
Ethan Yonker53796e72019-01-11 22:49:52 -06002792 DataManager::GetValue("tw_flash_partition", Flash_List);
2793 Repack_Type repack = REPLACE_NONE;
2794 if (Flash_List == "/repack_ramdisk;") {
2795 repack = REPLACE_RAMDISK;
2796 } else if (Flash_List == "/repack_kernel;") {
2797 repack = REPLACE_KERNEL;
2798 }
2799 if (repack != REPLACE_NONE) {
2800 Repack_Options_struct Repack_Options;
2801 Repack_Options.Type = repack;
2802 Repack_Options.Disable_Verity = false;
2803 Repack_Options.Disable_Force_Encrypt = false;
2804 Repack_Options.Backup_First = DataManager::GetIntValue("tw_repack_backup_first") != 0;
bigbiffad58e1b2020-07-06 20:24:34 -04002805 return repacker.Repack_Image_And_Flash(full_filename, Repack_Options);
Ethan Yonker53796e72019-01-11 22:49:52 -06002806 }
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002807 PartitionSettings part_settings;
2808 part_settings.Backup_Folder = path;
2809 unsigned long long total_bytes = TWFunc::Get_File_Size(full_filename);
2810 ProgressTracking progress(total_bytes);
2811 part_settings.progress = &progress;
2812 part_settings.adbbackup = false;
2813 part_settings.PM_Method = PM_RESTORE;
Ethan Yonker74db1572015-10-28 12:44:49 -05002814 gui_msg("calc_restore=Calculating restore details...");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002815 if (!Flash_List.empty()) {
2816 end_pos = Flash_List.find(";", start_pos);
2817 while (end_pos != string::npos && start_pos < Flash_List.size()) {
2818 flash_path = Flash_List.substr(start_pos, end_pos - start_pos);
2819 flash_part = Find_Partition_By_Path(flash_path);
2820 if (flash_part != NULL) {
2821 partition_count++;
2822 if (partition_count > 1) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002823 gui_err("too_many_flash=Too many partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002824 return false;
2825 }
2826 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002827 gui_msg(Msg(msg::kError, "flash_unable_locate=Unable to locate '{1}' partition for flashing.")(flash_path));
Ethan Yonker96af84a2015-01-05 14:58:36 -06002828 return false;
2829 }
2830 start_pos = end_pos + 1;
2831 end_pos = Flash_List.find(";", start_pos);
2832 }
2833 }
2834
2835 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002836 gui_err("no_part_flash=No partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002837 return false;
2838 }
2839
2840 DataManager::SetProgress(0.0);
2841 if (flash_part) {
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002842 flash_part->Backup_FileName = filename;
2843 if (!flash_part->Flash_Image(&part_settings))
Ethan Yonker96af84a2015-01-05 14:58:36 -06002844 return false;
2845 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002846 gui_err("invalid_flash=Invalid flash partition specified.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002847 return false;
2848 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002849 gui_highlight("flash_done=IMAGE FLASH COMPLETED]");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002850 return true;
2851}
Ethan Yonker74db1572015-10-28 12:44:49 -05002852
2853void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value) {
2854 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2855 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002856 if (part->Is_Adopted_Storage) {
2857 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2858 part->Backup_Display_Name = part->Display_Name;
2859 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2860 } else {
2861 part->Display_Name = gui_lookup(resource_name, default_value);
2862 part->Backup_Display_Name = part->Display_Name;
2863 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002864 }
2865}
2866
2867void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value, const char* storage_resource_name, const char* storage_default_value) {
2868 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2869 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002870 if (part->Is_Adopted_Storage) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06002871 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup("data_backup", "Data (excl. storage)");
Ethan Yonker66a19492015-12-10 10:19:45 -06002872 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
Ethan Yonker66a19492015-12-10 10:19:45 -06002873 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2874 } else {
2875 part->Display_Name = gui_lookup(resource_name, default_value);
2876 part->Backup_Display_Name = part->Display_Name;
2877 if (part->Is_Storage)
2878 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2879 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002880 }
2881}
2882
Ethan Yonker01f4e032017-02-03 15:30:52 -06002883void 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) {
2884 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2885 if (part) {
2886 if (part->Is_Adopted_Storage) {
2887 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup(backup_name, backup_default);
2888 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2889 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2890 } else {
2891 part->Display_Name = gui_lookup(resource_name, default_value);
2892 part->Backup_Display_Name = gui_lookup(backup_name, backup_default);
2893 if (part->Is_Storage)
2894 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2895 }
2896 }
2897}
2898
Ethan Yonker74db1572015-10-28 12:44:49 -05002899void TWPartitionManager::Translate_Partition_Display_Names() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002900 LOGINFO("Translating partition display names\n");
Ethan Yonker74db1572015-10-28 12:44:49 -05002901 Translate_Partition("/system", "system", "System");
2902 Translate_Partition("/system_image", "system_image", "System Image");
2903 Translate_Partition("/vendor", "vendor", "Vendor");
2904 Translate_Partition("/vendor_image", "vendor_image", "Vendor Image");
2905 Translate_Partition("/cache", "cache", "Cache");
Ethan Yonker74db1572015-10-28 12:44:49 -05002906 Translate_Partition("/boot", "boot", "Boot");
2907 Translate_Partition("/recovery", "recovery", "Recovery");
2908 if (!datamedia) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06002909 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage");
Ethan Yonker74db1572015-10-28 12:44:49 -05002910 Translate_Partition("/sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2911 Translate_Partition("/internal_sd", "sdcard", "SDCard", "sdcard", "SDCard");
2912 Translate_Partition("/internal_sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2913 Translate_Partition("/emmc", "sdcard", "SDCard", "sdcard", "SDCard");
Ethan Yonker01f4e032017-02-03 15:30:52 -06002914 } else {
2915 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05002916 }
Ethan Yonker01f4e032017-02-03 15:30:52 -06002917 Translate_Partition("/external_sd", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
2918 Translate_Partition("/external_sdcard", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05002919 Translate_Partition("/usb-otg", "usbotg", "USB OTG", "usbotg", "USB OTG");
2920 Translate_Partition("/sd-ext", "sdext", "SD-EXT");
2921
2922 // Android secure is a special case
2923 TWPartition* part = PartitionManager.Find_Partition_By_Path("/and-sec");
2924 if (part)
2925 part->Backup_Display_Name = gui_lookup("android_secure", "Android Secure");
2926
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002927 std::vector<TWPartition*>::iterator sysfs;
2928 for (sysfs = Partitions.begin(); sysfs != Partitions.end(); sysfs++) {
2929 if (!(*sysfs)->Sysfs_Entry.empty()) {
2930 Translate_Partition((*sysfs)->Mount_Point.c_str(), "autostorage", "Storage", "autostorage", "Storage");
2931 }
2932 }
2933
Ethan Yonker74db1572015-10-28 12:44:49 -05002934 // This updates the text on all of the storage selection buttons in the GUI
2935 DataManager::SetBackupFolder();
2936}
Ethan Yonker66a19492015-12-10 10:19:45 -06002937
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002938bool TWPartitionManager::Decrypt_Adopted() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002939#ifdef TW_INCLUDE_CRYPTO
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002940 bool ret = false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002941 if (!Mount_By_Path("/data", false)) {
2942 LOGERR("Cannot decrypt adopted storage because /data will not mount\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002943 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002944 }
2945 LOGINFO("Decrypt adopted storage starting\n");
2946 char* xmlFile = PageManager::LoadFileToBuffer("/data/system/storage.xml", NULL);
2947 xml_document<> *doc = NULL;
2948 xml_node<>* volumes = NULL;
Ethan Yonker66a19492015-12-10 10:19:45 -06002949 string Primary_Storage_UUID = "";
2950 if (xmlFile != NULL) {
2951 LOGINFO("successfully loaded storage.xml\n");
2952 doc = new xml_document<>();
2953 doc->parse<0>(xmlFile);
2954 volumes = doc->first_node("volumes");
2955 if (volumes) {
2956 xml_attribute<>* psuuid = volumes->first_attribute("primaryStorageUuid");
2957 if (psuuid) {
2958 Primary_Storage_UUID = psuuid->value();
2959 }
2960 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002961 } else {
2962 LOGINFO("No /data/system/storage.xml for adopted storage\n");
2963 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002964 }
2965 std::vector<TWPartition*>::iterator adopt;
2966 for (adopt = Partitions.begin(); adopt != Partitions.end(); adopt++) {
Peter Caife14cc12019-10-17 08:49:10 +08002967 if ((*adopt)->Removable && !(*adopt)->Is_Present && (*adopt)->Adopted_Mount_Delay > 0) {
2968 // On some devices, the external mmc driver takes some time
2969 // to recognize the card, in which case the "actual block device"
2970 // would not have been found yet. We wait the specified delay
2971 // and then try again.
2972 LOGINFO("Sleeping %d seconds for adopted storage.\n", (*adopt)->Adopted_Mount_Delay);
2973 sleep((*adopt)->Adopted_Mount_Delay);
2974 (*adopt)->Find_Actual_Block_Device();
2975 }
2976
Ethan Yonker66a19492015-12-10 10:19:45 -06002977 if ((*adopt)->Removable && (*adopt)->Is_Present) {
2978 if ((*adopt)->Decrypt_Adopted() == 0) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002979 ret = true;
Ethan Yonker66a19492015-12-10 10:19:45 -06002980 if (volumes) {
2981 xml_node<>* volume = volumes->first_node("volume");
2982 while (volume) {
2983 xml_attribute<>* guid = volume->first_attribute("partGuid");
2984 if (guid) {
2985 string GUID = (*adopt)->Adopted_GUID.c_str();
2986 GUID.insert(8, "-");
2987 GUID.insert(13, "-");
2988 GUID.insert(18, "-");
2989 GUID.insert(23, "-");
2990
2991 if (strcasecmp(GUID.c_str(), guid->value()) == 0) {
2992 xml_attribute<>* attr = volume->first_attribute("nickname");
nkk71ffb02bd2017-06-04 23:12:16 +03002993 if (attr && attr->value() && strlen(attr->value()) > 0) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002994 (*adopt)->Storage_Name = attr->value();
2995 (*adopt)->Display_Name = (*adopt)->Storage_Name;
2996 (*adopt)->Backup_Display_Name = (*adopt)->Storage_Name;
2997 LOGINFO("storage name from storage.xml is '%s'\n", attr->value());
2998 }
2999 attr = volume->first_attribute("fsUuid");
3000 if (attr && !Primary_Storage_UUID.empty() && strcmp(Primary_Storage_UUID.c_str(), attr->value()) == 0) {
3001 TWPartition* Dat = Find_Partition_By_Path("/data");
3002 if (Dat) {
3003 LOGINFO("Internal storage is found on adopted storage '%s'\n", (*adopt)->Display_Name.c_str());
3004 LOGINFO("Changing '%s' to point to '%s'\n", Dat->Symlink_Mount_Point.c_str(), (*adopt)->Storage_Path.c_str());
3005 (*adopt)->Symlink_Mount_Point = Dat->Symlink_Mount_Point;
3006 Dat->Symlink_Mount_Point = "";
3007 // Toggle mounts to ensure that the symlink mount point (probably /sdcard) is mounted to the right location
3008 Dat->UnMount(false);
3009 Dat->Mount(false);
3010 (*adopt)->UnMount(false);
3011 (*adopt)->Mount(false);
Ethan Yonker66a19492015-12-10 10:19:45 -06003012 }
3013 }
3014 break;
3015 }
3016 }
3017 volume = volume->next_sibling("volume");
3018 }
3019 }
nkk71ffb02bd2017-06-04 23:12:16 +03003020 Update_System_Details();
3021 Output_Partition((*adopt));
Ethan Yonker66a19492015-12-10 10:19:45 -06003022 }
3023 }
3024 }
3025 if (xmlFile) {
3026 doc->clear();
3027 delete doc;
3028 free(xmlFile);
3029 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003030 return ret;
Ethan Yonker66a19492015-12-10 10:19:45 -06003031#else
3032 LOGINFO("Decrypt_Adopted: no crypto support\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003033 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06003034#endif
3035}
Ethan Yonkerfcf3f242016-02-16 12:30:26 -06003036
3037void TWPartitionManager::Remove_Partition_By_Path(string Path) {
3038 std::vector<TWPartition*>::iterator iter;
3039 string Local_Path = TWFunc::Get_Root_Path(Path);
3040
3041 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3042 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
3043 LOGINFO("Found and erasing '%s' from partition list\n", Local_Path.c_str());
3044 Partitions.erase(iter);
3045 return;
3046 }
3047 }
3048}
Ethan Yonker1b190162016-12-05 15:25:19 -06003049
3050void TWPartitionManager::Set_Active_Slot(const string& Slot) {
3051 if (Slot != "A" && Slot != "B") {
3052 LOGERR("Set_Active_Slot invalid slot '%s'\n", Slot.c_str());
3053 return;
3054 }
3055 if (Active_Slot_Display == Slot)
3056 return;
3057 LOGINFO("Setting active slot %s\n", Slot.c_str());
3058#ifdef AB_OTA_UPDATER
3059 if (!Active_Slot_Display.empty()) {
3060 const hw_module_t *hw_module;
3061 boot_control_module_t *module;
3062 int ret;
3063 ret = hw_get_module("bootctrl", &hw_module);
3064 if (ret != 0) {
3065 LOGERR("Error getting bootctrl module.\n");
3066 } else {
3067 module = (boot_control_module_t*) hw_module;
3068 module->init(module);
3069 int slot_number = 0;
3070 if (Slot == "B")
3071 slot_number = 1;
3072 if (module->setActiveBootSlot(module, slot_number))
3073 gui_msg(Msg(msg::kError, "unable_set_boot_slot=Error changing bootloader boot slot to {1}")(Slot));
3074 }
3075 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
3076 }
3077#else
3078 LOGERR("Boot slot feature not present\n");
3079#endif
3080 Active_Slot_Display = Slot;
3081 if (Fstab_Processed())
3082 Update_System_Details();
3083}
3084string TWPartitionManager::Get_Active_Slot_Suffix() {
3085 if (Active_Slot_Display == "A")
3086 return "_a";
3087 return "_b";
3088}
3089string TWPartitionManager::Get_Active_Slot_Display() {
3090 return Active_Slot_Display;
3091}
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003092
Captain Throwback9d6feb52018-07-27 10:05:24 -04003093string TWPartitionManager::Get_Android_Root_Path() {
bigbiff8da46fa2020-09-08 16:42:34 -04003094 return "/system_root";
Captain Throwback9d6feb52018-07-27 10:05:24 -04003095}
3096
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003097void TWPartitionManager::Remove_Uevent_Devices(const string& Mount_Point) {
3098 std::vector<TWPartition*>::iterator iter;
3099
3100 for (iter = Partitions.begin(); iter != Partitions.end(); ) {
3101 if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Mount_Point) {
3102 TWPartition *part = *iter;
3103 LOGINFO("%s was removed by uevent data\n", (*iter)->Mount_Point.c_str());
3104 (*iter)->UnMount(false);
3105 rmdir((*iter)->Mount_Point.c_str());
3106 iter = Partitions.erase(iter);
3107 delete part;
3108 } else {
3109 iter++;
3110 }
3111 }
3112}
3113
3114void TWPartitionManager::Handle_Uevent(const Uevent_Block_Data& uevent_data) {
3115 std::vector<TWPartition*>::iterator iter;
3116
3117 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3118 if (!(*iter)->Sysfs_Entry.empty()) {
3119 string device;
3120 size_t wildcard = (*iter)->Sysfs_Entry.find("*");
3121 if (wildcard != string::npos) {
3122 device = (*iter)->Sysfs_Entry.substr(0, wildcard);
3123 } else {
3124 device = (*iter)->Sysfs_Entry;
3125 }
3126 if (device == uevent_data.sysfs_path.substr(0, device.size())) {
3127 // Found a match
3128 if (uevent_data.action == "add") {
3129 (*iter)->Primary_Block_Device = "/dev/block/" + uevent_data.block_device;
3130 (*iter)->Alternate_Block_Device = (*iter)->Primary_Block_Device;
3131 (*iter)->Is_Present = true;
3132 LOGINFO("Found a match '%s' '%s'\n", uevent_data.block_device.c_str(), device.c_str());
3133 if (!Decrypt_Adopted()) {
3134 LOGINFO("No adopted storage so finding actual block device\n");
3135 (*iter)->Find_Actual_Block_Device();
3136 }
3137 return;
3138 } else if (uevent_data.action == "remove") {
3139 (*iter)->Is_Present = false;
3140 (*iter)->Primary_Block_Device = "";
3141 (*iter)->Actual_Block_Device = "";
3142 Remove_Uevent_Devices((*iter)->Mount_Point);
3143 return;
3144 }
3145 }
3146 }
3147 }
bigbiffee7b7ff2020-03-23 15:08:27 -04003148
3149 if (!PartitionManager.Get_Super_Status())
3150 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 -06003151}
3152
3153void TWPartitionManager::setup_uevent() {
3154 struct sockaddr_nl nls;
3155
3156 if (uevent_pfd.fd >= 0) {
3157 LOGINFO("uevent already set up\n");
3158 return;
3159 }
3160
3161 // Open hotplug event netlink socket
3162 memset(&nls,0,sizeof(struct sockaddr_nl));
3163 nls.nl_family = AF_NETLINK;
3164 nls.nl_pid = getpid();
3165 nls.nl_groups = -1;
3166 uevent_pfd.events = POLLIN;
3167 uevent_pfd.fd = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_KOBJECT_UEVENT);
3168 if (uevent_pfd.fd==-1) {
3169 LOGERR("uevent not root\n");
3170 return;
3171 }
3172
3173 // Listen to netlink socket
3174 if (::bind(uevent_pfd.fd, (struct sockaddr *) &nls, sizeof(struct sockaddr_nl)) < 0) {
3175 LOGERR("Bind failed\n");
3176 return;
3177 }
3178 set_select_fd();
3179 Coldboot();
3180}
3181
3182Uevent_Block_Data TWPartitionManager::get_event_block_values(char *buf, int len) {
3183 Uevent_Block_Data ret;
3184 ret.subsystem = "";
3185 char *ptr = buf;
3186 const char *end = buf + len;
3187
3188 buf[len - 1] = '\0';
3189 while (ptr < end) {
3190 if (strncmp(ptr, "ACTION=", strlen("ACTION=")) == 0) {
3191 ptr += strlen("ACTION=");
3192 ret.action = ptr;
3193 } else if (strncmp(ptr, "SUBSYSTEM=", strlen("SUBSYSTEM=")) == 0) {
3194 ptr += strlen("SUBSYSTEM=");
3195 ret.subsystem = ptr;
3196 } else if (strncmp(ptr, "DEVTYPE=", strlen("DEVTYPE=")) == 0) {
3197 ptr += strlen("DEVTYPE=");
3198 ret.type = ptr;
3199 } else if (strncmp(ptr, "DEVPATH=", strlen("DEVPATH=")) == 0) {
3200 ptr += strlen("DEVPATH=");
3201 ret.sysfs_path += ptr;
3202 } else if (strncmp(ptr, "DEVNAME=", strlen("DEVNAME=")) == 0) {
3203 ptr += strlen("DEVNAME=");
3204 ret.block_device += ptr;
3205 } else if (strncmp(ptr, "MAJOR=", strlen("MAJOR=")) == 0) {
3206 ptr += strlen("MAJOR=");
3207 ret.major = atoi(ptr);
3208 } else if (strncmp(ptr, "MINOR=", strlen("MINOR=")) == 0) {
3209 ptr += strlen("MINOR=");
3210 ret.minor = atoi(ptr);
3211 }
3212 ptr += strlen(ptr) + 1;
3213 }
3214 return ret;
3215}
3216
3217void TWPartitionManager::read_uevent() {
3218 char buf[1024];
3219
3220 int len = recv(uevent_pfd.fd, buf, sizeof(buf), MSG_DONTWAIT);
3221 if (len == -1) {
Mauronofrio Matarresec1bb76e2019-08-04 17:02:41 +01003222 LOGINFO("recv error on uevent\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003223 return;
3224 }
3225 /*int i = 0; // Print all uevent output for test /debug
3226 while (i<len) {
3227 printf("%s\n", buf+i);
3228 i += strlen(buf+i)+1;
3229 }*/
3230 Uevent_Block_Data uevent_data = get_event_block_values(buf, len);
3231 if (uevent_data.subsystem == "block" && uevent_data.type == "disk") {
3232 PartitionManager.Handle_Uevent(uevent_data);
3233 }
3234}
3235
3236void TWPartitionManager::close_uevent() {
3237 if (uevent_pfd.fd > 0)
3238 close(uevent_pfd.fd);
3239 uevent_pfd.fd = -1;
3240}
3241
3242void TWPartitionManager::Add_Partition(TWPartition* Part) {
3243 Partitions.push_back(Part);
3244}
3245
3246void TWPartitionManager::Coldboot_Scan(std::vector<string> *sysfs_entries, const string& Path, int depth) {
3247 string Real_Path = Path;
3248 char real_path[PATH_MAX];
3249 if (realpath(Path.c_str(), &real_path[0])) {
3250 string Real_Path = real_path;
3251 std::vector<string>::iterator iter;
3252 for (iter = sysfs_entries->begin(); iter != sysfs_entries->end(); iter++) {
3253 if (Real_Path.find((*iter)) != string::npos) {
3254 string Write_Path = Real_Path + "/uevent";
3255 if (TWFunc::Path_Exists(Write_Path)) {
3256 const char* write_val = "add\n";
3257 TWFunc::write_to_file(Write_Path, write_val);
3258 break;
3259 }
3260 }
3261 }
3262 }
3263
3264 DIR* d = opendir(Path.c_str());
3265 if (d != NULL) {
3266 struct dirent* de;
3267 while ((de = readdir(d)) != NULL) {
3268 if (de->d_name[0] == '.' || (de->d_type != DT_DIR && depth > 0))
3269 continue;
3270 if (strlen(de->d_name) >= 4 && (strncmp(de->d_name, "ram", 3) == 0 || strncmp(de->d_name, "loop", 4) == 0))
3271 continue;
3272
3273 string item = Path + "/";
3274 item.append(de->d_name);
3275 Coldboot_Scan(sysfs_entries, item, depth + 1);
3276 }
3277 closedir(d);
3278 }
3279}
3280
3281void TWPartitionManager::Coldboot() {
3282 std::vector<TWPartition*>::iterator iter;
3283 std::vector<string> sysfs_entries;
3284
3285 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3286 if (!(*iter)->Sysfs_Entry.empty()) {
3287 size_t wildcard_pos = (*iter)->Sysfs_Entry.find("*");
3288 if (wildcard_pos == string::npos)
3289 wildcard_pos = (*iter)->Sysfs_Entry.size();
3290 sysfs_entries.push_back((*iter)->Sysfs_Entry.substr(0, wildcard_pos));
3291 }
3292 }
3293
3294 if (sysfs_entries.size() > 0)
3295 Coldboot_Scan(&sysfs_entries, "/sys/block", 0);
3296}
Ethan Yonker53796e72019-01-11 22:49:52 -06003297
3298bool TWPartitionManager::Prepare_Empty_Folder(const std::string& Folder) {
3299 if (TWFunc::Path_Exists(Folder))
3300 TWFunc::removeDir(Folder, false);
3301 return TWFunc::Recursive_Mkdir(Folder);
3302}
3303
bigbiffee7b7ff2020-03-23 15:08:27 -04003304bool TWPartitionManager::Prepare_Super_Volume(TWPartition* twrpPart) {
3305 Fstab fstab;
3306 std::string bare_partition_name;
3307
3308 if (twrpPart->Get_Mount_Point() == "/system_root")
3309 bare_partition_name = "system";
3310 else
3311 bare_partition_name = TWFunc::Remove_Beginning_Slash(twrpPart->Get_Mount_Point());
3312
3313 LOGINFO("Trying to prepare %s from super partition\n", bare_partition_name.c_str());
3314
3315 std::string blk_device_partition;
3316#ifdef AB_OTA_UPDATER
3317 blk_device_partition = bare_partition_name + PartitionManager.Get_Active_Slot_Suffix();
3318#else
3319 blk_device_partition = bare_partition_name;
3320#endif
3321
3322 FstabEntry fstabEntry = {
3323 .blk_device = blk_device_partition,
3324 .mount_point = twrpPart->Get_Mount_Point(),
3325 .fs_type = twrpPart->Current_File_System,
3326 .fs_mgr_flags.logical = twrpPart->Is_Super,
3327 };
3328
3329 fstab.emplace_back(fstabEntry);
3330 if (!fs_mgr_update_logical_partition(&fstabEntry)) {
3331 LOGERR("unable to update logical partition: %s\n", twrpPart->Get_Mount_Point().c_str());
3332 return false;
3333 }
3334
bigbiff32cbabe2020-04-12 19:16:19 -04003335 while (access(fstabEntry.blk_device.c_str(), F_OK) != 0) {
3336 usleep(100);
3337 }
3338
bigbiffee7b7ff2020-03-23 15:08:27 -04003339 twrpPart->Set_Block_Device(fstabEntry.blk_device);
3340 twrpPart->Update_Size(true);
3341 twrpPart->Change_Mount_Read_Only(true);
3342 twrpPart->Set_Can_Be_Backed_Up(false);
3343 twrpPart->Set_Can_Be_Wiped(false);
3344 return true;
3345}
3346
3347bool TWPartitionManager::Prepare_All_Super_Volumes() {
3348 bool status = true;
3349 std::vector<TWPartition*>::iterator iter;
3350
3351 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3352 if ((*iter)->Is_Super) {
3353 if (!Prepare_Super_Volume(*iter)) {
3354 status = false;
3355 }
3356 PartitionManager.Output_Partition(*iter);
3357 }
3358 }
3359 Update_System_Details();
3360 return status;
3361}
3362
3363bool TWPartitionManager::Is_Super_Partition(const char* fstab_line) {
Mohd Faraze3948ec2020-08-14 01:40:59 +00003364 if (!Get_Super_Status())
3365 return false;
bigbiffee7b7ff2020-03-23 15:08:27 -04003366 std::vector<std::string> super_partition_list = {"system", "vendor", "odm", "product", "system_ext"};
3367
3368 for (auto&& fstab_partition_check: super_partition_list) {
3369 if (strncmp(fstab_line, fstab_partition_check.c_str(), fstab_partition_check.size()) == 0) {
3370 DataManager::SetValue(TW_IS_SUPER, "1");
3371 return true;
3372 }
3373 }
3374 return false;
3375}
3376
3377std::string TWPartitionManager::Get_Super_Partition() {
3378 int slot_number = Get_Active_Slot_Display() == "A" ? 0 : 1;
3379 std::string super_device = fs_mgr_get_super_partition_name(slot_number);
3380 return "/dev/block/by-name/" + super_device;
3381}
3382
3383void TWPartitionManager::Setup_Super_Devices() {
3384 std::string superPart = Get_Super_Partition();
3385 android::fs_mgr::CreateLogicalPartitions(superPart);
bigbiff7ba75002020-04-11 20:47:09 -04003386}
3387
3388void TWPartitionManager::Setup_Super_Partition() {
bigbiffee7b7ff2020-03-23 15:08:27 -04003389 TWPartition* superPartition = new TWPartition();
bigbiff7ba75002020-04-11 20:47:09 -04003390 std::string superPart = Get_Super_Partition();
3391
3392 superPartition->Backup_Path = "/super";
3393 superPartition->Mount_Point = "/super";
bigbiffee7b7ff2020-03-23 15:08:27 -04003394 superPartition->Actual_Block_Device = superPart;
3395 superPartition->Alternate_Block_Device = superPart;
Ian Macdonald160e8d32020-09-29 20:47:31 +02003396#ifdef BOARD_DYNAMIC_PARTITIONS_PARTITION_LIST
3397 superPartition->Backup_Display_Name = "Super (" BOARD_DYNAMIC_PARTITIONS_PARTITION_LIST ")";
3398#else
3399 superPartition->Backup_Display_Name = "Super";
3400#endif
bigbiffee7b7ff2020-03-23 15:08:27 -04003401 superPartition->Can_Flash_Img = true;
bigbiffee7b7ff2020-03-23 15:08:27 -04003402 superPartition->Current_File_System = "emmc";
3403 superPartition->Can_Be_Backed_Up = true;
3404 superPartition->Is_Present = true;
3405 superPartition->Is_SubPartition = false;
bigbiff7ba75002020-04-11 20:47:09 -04003406 superPartition->Setup_Image();
bigbiffee7b7ff2020-03-23 15:08:27 -04003407 Add_Partition(superPartition);
3408 PartitionManager.Output_Partition(superPartition);
bigbiffee7b7ff2020-03-23 15:08:27 -04003409}
3410
3411bool TWPartitionManager::Get_Super_Status() {
bigbiffdf8436b2020-08-30 16:22:34 -04003412 std::string fastboot_mode = android::base::GetProperty("sys.usb.config", "");
3413 if (fastboot_mode == "fastboot") {
3414 return false;
3415 }
3416 else
3417 return access(Get_Super_Partition().c_str(), F_OK) == 0;
bigbiffee7b7ff2020-03-23 15:08:27 -04003418}
bigbiff25d25b92020-06-19 16:07:38 -04003419
3420bool TWPartitionManager::Recreate_Logs_Dir() {
3421#ifdef TW_INCLUDE_FBE
3422 struct passwd pd;
3423 struct passwd *pwdptr = &pd;
3424 struct passwd *tempPd;
3425 char pwdBuf[512];
3426 int uid = 0, gid = 0;
3427
3428 if ((getpwnam_r("system", pwdptr, pwdBuf, sizeof(pwdBuf), &tempPd)) != 0) {
3429 LOGERR("unable to get system user id\n");
3430 return false;
3431 } else {
3432 struct group grp;
3433 struct group *grpptr = &grp;
3434 struct group *tempGrp;
3435 char grpBuf[512];
3436
3437 if ((getgrnam_r("cache", grpptr, grpBuf, sizeof(grpBuf), &tempGrp)) != 0) {
3438 LOGERR("unable to get cache group id\n");
3439 return false;
3440 } else {
3441 uid = pd.pw_uid;
3442 gid = grp.gr_gid;
3443 std::string abLogsRecoveryDir(DATA_LOGS_DIR);
3444 abLogsRecoveryDir += "/recovery/";
3445
3446 if (!TWFunc::Create_Dir_Recursive(abLogsRecoveryDir, S_IRWXU | S_IRWXG | S_IWGRP | S_IXGRP, uid, gid)) {
3447 LOGERR("Unable to recreate %s\n", abLogsRecoveryDir.c_str());
3448 return false;
3449 }
3450 if (setfilecon(abLogsRecoveryDir.c_str(), "u:object_r:cache_file:s0") != 0) {
3451 LOGERR("Unable to set contexts for %s\n", abLogsRecoveryDir.c_str());
3452 return false;
3453 }
3454 }
3455 }
3456#endif
3457 return true;
3458}
bigbiff1f9e4842020-10-31 11:33:15 -04003459
3460void TWPartitionManager::Unlock_Block_Partitions() {
3461 int fd, OFF = 0;
3462
3463 const std::string block_path = "/dev/block/";
3464 DIR* d = opendir(block_path.c_str());
3465 if (d != NULL) {
3466 struct dirent* de;
3467 while ((de = readdir(d)) != NULL) {
3468 if (de->d_type == DT_BLK) {
3469 std::string block_device = block_path + de->d_name;
bigbiff1f9e4842020-10-31 11:33:15 -04003470 if ((fd = open(block_device.c_str(), O_RDONLY | O_CLOEXEC)) < 0) {
3471 LOGERR("unable to open block device %s: %s\n", block_device.c_str(), strerror(errno));
3472 continue;
3473 }
3474 if (ioctl(fd, BLKROSET, &OFF) == -1) {
3475 LOGERR("Unable to unlock %s for flashing: %s\n", block_device.c_str());
3476 continue;
3477 }
3478 close(fd);
3479 }
3480 }
3481 closedir(d);
3482 }
bigbiffd655ffe2020-12-06 16:50:55 -05003483}