blob: efcaed889f64efb778a80060ddcfc9cdbb9dbe21 [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"
bigbiffce8f83c2015-12-12 18:30:21 -050070#include "adbbu/libtwadbbu.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040071
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040072#ifdef TW_HAS_MTP
bigbiff bigbiffaf32bb92018-12-18 18:39:53 -050073#ifdef TW_HAS_LEGACY_MTP
74#include "mtp/legacy/mtp_MtpServer.hpp"
75#include "mtp/legacy/twrpMtp.hpp"
76#include "mtp/legacy/MtpMessage.hpp"
77#else
78#include "mtp/ffs/mtp_MtpServer.hpp"
79#include "mtp/ffs/twrpMtp.hpp"
80#include "mtp/ffs/MtpMessage.hpp"
81#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040082#endif
83
Dees Troy6f6441d2014-01-23 02:07:03 +000084extern "C" {
85 #include "cutils/properties.h"
Ethan Yonker74db1572015-10-28 12:44:49 -050086 #include "gui/gui.h"
Dees Troy6f6441d2014-01-23 02:07:03 +000087}
88
Dees_Troy5bf43922012-09-07 16:07:55 -040089#ifdef TW_INCLUDE_CRYPTO
bigbiff7ba75002020-04-11 20:47:09 -040090#include "crypto/fde/cryptfs.h"
91#include "gui/rapidxml.hpp"
92#include "gui/pages.hpp"
93#ifdef TW_INCLUDE_FBE
94#include "crypto/ext4crypt/Decrypt.h"
95#ifdef TW_INCLUDE_FBE_METADATA_DECRYPT
96 #ifdef USE_FSCRYPT
97 #include "crypto/fscrypt/MetadataCrypt.h"
98 #else
99 #include "crypto/ext4crypt/MetadataCrypt.h"
bigbiffadc599e2020-05-28 19:36:30 +0000100 #endif
bigbiff7ba75002020-04-11 20:47:09 -0400101#endif
102#endif
103#ifdef TW_CRYPTO_USE_SYSTEM_VOLD
104#include "crypto/vold_decrypt/vold_decrypt.h"
105#endif
Dees_Troy5bf43922012-09-07 16:07:55 -0400106#endif
Dees_Troy51a0e822012-09-05 15:24:24 -0400107
Ethan Yonker1b190162016-12-05 15:25:19 -0600108#ifdef AB_OTA_UPDATER
109#include <hardware/hardware.h>
110#include <hardware/boot_control.h>
111#endif
112
bigbiffee7b7ff2020-03-23 15:08:27 -0400113using android::fs_mgr::Fstab;
114using android::fs_mgr::FstabEntry;
Ethan Yonker6277c792014-09-15 14:54:30 -0500115
bigbiffee7b7ff2020-03-23 15:08:27 -0400116extern bool datamedia;
117
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500118TWPartitionManager::TWPartitionManager(void) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400119 mtp_was_enabled = false;
Ethan Yonker726a0202014-12-16 20:01:38 -0600120 mtp_write_fd = -1;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600121 uevent_pfd.fd = -1;
bigbiff7abc5fe2015-01-17 16:53:12 -0500122 stop_backup.set_value(0);
Ethan Yonker1b190162016-12-05 15:25:19 -0600123#ifdef AB_OTA_UPDATER
124 char slot_suffix[PROPERTY_VALUE_MAX];
Ethan Yonkerfefe5912017-09-30 22:22:13 -0500125 property_get("ro.boot.slot_suffix", slot_suffix, "error");
126 if (strcmp(slot_suffix, "error") == 0)
127 property_get("ro.boot.slot", slot_suffix, "error");
Ethan Yonker1b190162016-12-05 15:25:19 -0600128 Active_Slot_Display = "";
Ethan Yonkerfefe5912017-09-30 22:22:13 -0500129 if (strcmp(slot_suffix, "_a") == 0 || strcmp(slot_suffix, "a") == 0)
Ethan Yonker1b190162016-12-05 15:25:19 -0600130 Set_Active_Slot("A");
131 else
132 Set_Active_Slot("B");
133#endif
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500134}
135
Chaosmasterf6e42ce2020-01-27 00:17:04 +0100136int TWPartitionManager::Process_Fstab(string Fstab_Filename, bool Display_Error, bool Sar_Detect) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400137 FILE *fstabFile;
138 char fstab_line[MAX_FSTAB_LINE_LENGTH];
Dees Troy02a64532014-03-19 15:23:32 +0000139 TWPartition* settings_partition = NULL;
Matt Mowerbf4efa32014-04-14 23:25:26 -0500140 TWPartition* andsec_partition = NULL;
Ethan Yonker726a0202014-12-16 20:01:38 -0600141 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 -0600142 std::map<string, Flags_Map> twrp_flags;
143
144 fstabFile = fopen("/etc/twrp.flags", "rt");
145 if (fstabFile != NULL) {
146 LOGINFO("reading /etc/twrp.flags\n");
147 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
148 if (fstab_line[0] != '/')
149 continue;
150
151 size_t line_size = strlen(fstab_line);
152 if (fstab_line[line_size - 1] != '\n')
153 fstab_line[line_size] = '\n';
154 Flags_Map line_flags;
155 line_flags.Primary_Block_Device = "";
156 line_flags.Alternate_Block_Device = "";
157 line_flags.fstab_line = (char*)malloc(MAX_FSTAB_LINE_LENGTH);
158 if (!line_flags.fstab_line) {
159 LOGERR("malloc error on line_flags.fstab_line\n");
160 return false;
161 }
162 memcpy(line_flags.fstab_line, fstab_line, MAX_FSTAB_LINE_LENGTH);
163 bool found_separator = false;
164 char *fs_loc = NULL;
165 char *block_loc = NULL;
166 char *flags_loc = NULL;
167 size_t index, item_index = 0;
168 for (index = 0; index < line_size; index++) {
169 if (fstab_line[index] <= 32) {
170 fstab_line[index] = '\0';
171 found_separator = true;
172 } else if (found_separator) {
173 if (item_index == 0) {
174 fs_loc = fstab_line + index;
175 } else if (item_index == 1) {
176 block_loc = fstab_line + index;
177 } else if (item_index > 1) {
178 char *ptr = fstab_line + index;
179 if (*ptr == '/') {
180 line_flags.Alternate_Block_Device = ptr;
181 } else if (strlen(ptr) > strlen("flags=") && strncmp(ptr, "flags=", strlen("flags=")) == 0) {
182 flags_loc = ptr;
183 // Once we find the flags=, we're done scanning the line
184 break;
185 }
186 }
187 found_separator = false;
188 item_index++;
189 }
190 }
191 if (block_loc)
192 line_flags.Primary_Block_Device = block_loc;
193 if (fs_loc)
194 line_flags.File_System = fs_loc;
195 if (flags_loc)
196 line_flags.Flags = flags_loc;
197 string Mount_Point = fstab_line;
198 twrp_flags[Mount_Point] = line_flags;
199 memset(fstab_line, 0, sizeof(fstab_line));
200 }
201 fclose(fstabFile);
202 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400203
204 fstabFile = fopen(Fstab_Filename.c_str(), "rt");
205 if (fstabFile == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000206 LOGERR("Critical Error: Unable to open fstab at '%s'.\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400207 return false;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600208 } else
209 LOGINFO("Reading %s\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400210
211 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
bigbiffee7b7ff2020-03-23 15:08:27 -0400212 bool isSuper = Is_Super_Partition(fstab_line);
213
214 if (!isSuper && fstab_line[0] != '/')
Dees_Troy5bf43922012-09-07 16:07:55 -0400215 continue;
216
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600217 if (strstr(fstab_line, "swap"))
218 continue; // Skip swap in recovery
219
220 size_t line_size = strlen(fstab_line);
221 if (fstab_line[line_size - 1] != '\n')
222 fstab_line[line_size] = '\n';
Dees_Troy2a923582012-09-20 12:13:34 -0400223
Matt Mower2b2dd152016-04-26 11:24:08 -0500224 TWPartition* partition = new TWPartition();
Chaosmasterf6e42ce2020-01-27 00:17:04 +0100225 if (partition->Process_Fstab_Line(fstab_line, Display_Error, &twrp_flags, Sar_Detect))
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500226 Partitions.push_back(partition);
Matt Mower72c87ce2016-04-26 14:34:56 -0500227 else
Dees_Troy5bf43922012-09-07 16:07:55 -0400228 delete partition;
Matt Mower2b2dd152016-04-26 11:24:08 -0500229
230 memset(fstab_line, 0, sizeof(fstab_line));
Dees_Troy5bf43922012-09-07 16:07:55 -0400231 }
232 fclose(fstabFile);
Matt Mower72c87ce2016-04-26 14:34:56 -0500233
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600234 if (twrp_flags.size() > 0) {
235 LOGINFO("Processing remaining twrp.flags\n");
236 // Add any items from twrp.flags that did not exist in the recovery.fstab
237 for (std::map<string, Flags_Map>::iterator mapit=twrp_flags.begin(); mapit!=twrp_flags.end(); mapit++) {
238 if (Find_Partition_By_Path(mapit->first) == NULL) {
239 TWPartition* partition = new TWPartition();
Chaosmasterf6e42ce2020-01-27 00:17:04 +0100240 if (partition->Process_Fstab_Line(mapit->second.fstab_line, Display_Error, NULL, Sar_Detect))
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600241 Partitions.push_back(partition);
242 else
243 delete partition;
244 }
245 if (mapit->second.fstab_line)
246 free(mapit->second.fstab_line);
247 mapit->second.fstab_line = NULL;
248 }
249 }
250 LOGINFO("Done processing fstab files\n");
251
Matt Mower72c87ce2016-04-26 14:34:56 -0500252 std::vector<TWPartition*>::iterator iter;
253 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Chaosmasterf6e42ce2020-01-27 00:17:04 +0100254 if (Sar_Detect) {
255 if ((*iter)->Mount_Point == "/s")
256 return true;
257 else
258 continue;
259 }
260
Matt Mower72c87ce2016-04-26 14:34:56 -0500261 (*iter)->Partition_Post_Processing(Display_Error);
262
263 if ((*iter)->Is_Storage) {
264 ++storageid;
265 (*iter)->MTP_Storage_ID = storageid;
266 }
267
268 if (!settings_partition && (*iter)->Is_Settings_Storage && (*iter)->Is_Present)
269 settings_partition = (*iter);
270 else
271 (*iter)->Is_Settings_Storage = false;
272
273 if (!andsec_partition && (*iter)->Has_Android_Secure && (*iter)->Is_Present)
274 andsec_partition = (*iter);
275 else
276 (*iter)->Has_Android_Secure = false;
277 }
278
Ethan Yonker6277c792014-09-15 14:54:30 -0500279 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) {
280 // Attempt to automatically identify /data/media emulated storage devices
281 TWPartition* Dat = Find_Partition_By_Path("/data");
282 if (Dat) {
283 LOGINFO("Using automatic handling for /data/media emulated storage device.\n");
284 datamedia = true;
that9e0593e2014-10-08 00:01:24 +0200285 Dat->Setup_Data_Media();
Ethan Yonker6277c792014-09-15 14:54:30 -0500286 settings_partition = Dat;
Ethan Yonker726a0202014-12-16 20:01:38 -0600287 // Since /data was not considered a storage partition earlier, we still need to assign an MTP ID
288 ++storageid;
289 Dat->MTP_Storage_ID = storageid;
Ethan Yonker6277c792014-09-15 14:54:30 -0500290 }
291 }
Dees Troy02a64532014-03-19 15:23:32 +0000292 if (!settings_partition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500293 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
294 if ((*iter)->Is_Storage) {
Dees Troy02a64532014-03-19 15:23:32 +0000295 settings_partition = (*iter);
Dees_Troya13d74f2013-03-24 08:54:55 -0500296 break;
297 }
298 }
Dees Troy02a64532014-03-19 15:23:32 +0000299 if (!settings_partition)
Dees_Troy2673cec2013-04-02 20:22:16 +0000300 LOGERR("Unable to locate storage partition for storing settings file.\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500301 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400302 if (!Write_Fstab()) {
303 if (Display_Error)
Dees_Troy2673cec2013-04-02 20:22:16 +0000304 LOGERR("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400305 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000306 LOGINFO("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400307 }
Matt Mowered71fa32014-04-16 13:21:47 -0500308
Matt Mowerbf4efa32014-04-14 23:25:26 -0500309 if (andsec_partition) {
310 Setup_Android_Secure_Location(andsec_partition);
Matt Mowered71fa32014-04-16 13:21:47 -0500311 } else if (settings_partition) {
Matt Mowerbf4efa32014-04-14 23:25:26 -0500312 Setup_Android_Secure_Location(settings_partition);
313 }
Matt Mowered71fa32014-04-16 13:21:47 -0500314 if (settings_partition) {
315 Setup_Settings_Storage_Partition(settings_partition);
316 }
bigbiff7ba75002020-04-11 20:47:09 -0400317
318 Update_System_Details();
319 UnMount_Main_Partitions();
320#ifdef AB_OTA_UPDATER
321 DataManager::SetValue("tw_active_slot", Get_Active_Slot_Display());
322#endif
323 setup_uevent();
324 return true;
325}
326
327int TWPartitionManager::Write_Fstab(void) {
328 FILE *fp;
329 std::vector<TWPartition*>::iterator iter;
330 string Line;
331
332 fp = fopen("/etc/fstab", "w");
333 if (fp == NULL) {
334 LOGINFO("Can not open /etc/fstab.\n");
335 return false;
336 }
337 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
338 if ((*iter)->Can_Be_Mounted) {
339 Line = (*iter)->Actual_Block_Device + " " + (*iter)->Mount_Point + " " + (*iter)->Current_File_System + " rw 0 0\n";
340 fputs(Line.c_str(), fp);
341 }
342 // Handle subpartition tracking
343 if ((*iter)->Is_SubPartition) {
344 TWPartition* ParentPartition = Find_Partition_By_Path((*iter)->SubPartition_Of);
345 if (ParentPartition)
346 ParentPartition->Has_SubPartition = true;
347 else
348 LOGERR("Unable to locate parent partition '%s' of '%s'\n", (*iter)->SubPartition_Of.c_str(), (*iter)->Mount_Point.c_str());
349 }
350 }
351 fclose(fp);
352 return true;
353}
354
355void TWPartitionManager::Decrypt_Data() {
356 #ifdef TW_INCLUDE_CRYPTO
Ethan Yonkercceebb82014-11-18 10:17:59 -0600357 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
358 if (Decrypt_Data && Decrypt_Data->Is_Encrypted && !Decrypt_Data->Is_Decrypted) {
Ethan Yonker93382822018-11-01 15:25:31 -0500359 if (!Decrypt_Data->Key_Directory.empty() && Mount_By_Path(Decrypt_Data->Key_Directory, false)) {
360#ifdef TW_INCLUDE_FBE_METADATA_DECRYPT
bigbiff7ba75002020-04-11 20:47:09 -0400361#ifdef USE_FSCRYPT
362 if (fscrypt_mount_metadata_encrypted(Decrypt_Data->Actual_Block_Device, Decrypt_Data->Mount_Point, false)) {
363 std::string crypto_blkdev = android::base::GetProperty("ro.crypto.fs_crypto_blkdev", "error");
364 LOGINFO("Successfully decrypted metadata encrypted data partition with new block device: '%s'\n", crypto_blkdev.c_str());
365#else
Ethan Yonker93382822018-11-01 15:25:31 -0500366 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 -0400367 LOGINFO("Successfully decrypted metadata encrypted data partition with new block device: '%s'\n",
368 Decrypt_Data->Decrypted_Block_Device.c_str());
369#endif
Ethan Yonker93382822018-11-01 15:25:31 -0500370 property_set("ro.crypto.state", "encrypted");
371 Decrypt_Data->Is_Decrypted = true; // Needed to make the mount function work correctly
372 int retry_count = 10;
373 while (!Decrypt_Data->Mount(false) && --retry_count)
374 usleep(500);
375 if (Decrypt_Data->Mount(false)) {
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400376 if (!Decrypt_Data->Decrypt_FBE_DE()) {
mauronofriofc79aa62019-11-23 23:13:04 +0100377 char wrappedvalue[PROPERTY_VALUE_MAX];
378 property_get("fbe.data.wrappedkey", wrappedvalue, "");
379 std::string wrappedkeyvalue(wrappedvalue);
380 if (wrappedkeyvalue == "true") {
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400381 LOGERR("Unable to decrypt FBE device\n");
mauronofriofc79aa62019-11-23 23:13:04 +0100382 } else {
383 LOGINFO("Trying wrapped key.\n");
384 property_set("fbe.data.wrappedkey", "true");
385 if (!Decrypt_Data->Decrypt_FBE_DE()) {
386 LOGERR("Unable to decrypt FBE device\n");
387 }
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400388 }
389 }
390
Ethan Yonker93382822018-11-01 15:25:31 -0500391 } else {
392 LOGINFO("Failed to mount data after metadata decrypt\n");
393 }
394 } else {
395 LOGINFO("Unable to decrypt metadata encryption\n");
396 }
397#else
398 LOGERR("Metadata FBE decrypt support not present in this TWRP\n");
399#endif
400 }
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600401 if (Decrypt_Data->Is_FBE) {
402 if (DataManager::GetIntValue(TW_CRYPTO_PWTYPE) == 0) {
403 if (Decrypt_Device("!") == 0) {
404 gui_msg("decrypt_success=Successfully decrypted with default password.");
405 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
406 } else {
407 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
408 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600409 }
410 } else {
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600411 int password_type = cryptfs_get_password_type();
412 if (password_type == CRYPT_TYPE_DEFAULT) {
413 LOGINFO("Device is encrypted with the default password, attempting to decrypt.\n");
414 if (Decrypt_Device("default_password") == 0) {
415 gui_msg("decrypt_success=Successfully decrypted with default password.");
416 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
417 } else {
418 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
419 }
420 } else {
421 DataManager::SetValue("TW_CRYPTO_TYPE", password_type);
422 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600423 }
424 }
bigbiffadc599e2020-05-28 19:36:30 +0000425 if (Decrypt_Data && (!Decrypt_Data->Is_Encrypted || Decrypt_Data->Is_Decrypted) && Decrypt_Data->Mount(false)) {
Ethan Yonker66a19492015-12-10 10:19:45 -0600426 Decrypt_Adopted();
427 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600428#endif
Dees_Troy51a0e822012-09-05 15:24:24 -0400429}
430
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500431void TWPartitionManager::Setup_Settings_Storage_Partition(TWPartition* Part) {
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500432 DataManager::SetValue("tw_settings_path", Part->Storage_Path);
433 DataManager::SetValue("tw_storage_path", Part->Storage_Path);
434 LOGINFO("Settings storage is '%s'\n", Part->Storage_Path.c_str());
435}
436
Matt Mowerbf4efa32014-04-14 23:25:26 -0500437void TWPartitionManager::Setup_Android_Secure_Location(TWPartition* Part) {
438 if (Part->Has_Android_Secure)
439 Part->Setup_AndSec();
Ethan Yonker6277c792014-09-15 14:54:30 -0500440 else if (!datamedia)
Matt Mowerbf4efa32014-04-14 23:25:26 -0500441 Part->Setup_AndSec();
Matt Mowerbf4efa32014-04-14 23:25:26 -0500442}
443
Dees_Troy8170a922012-09-18 15:40:25 -0400444void TWPartitionManager::Output_Partition_Logging(void) {
445 std::vector<TWPartition*>::iterator iter;
446
447 printf("\n\nPartition Logs:\n");
448 for (iter = Partitions.begin(); iter != Partitions.end(); iter++)
449 Output_Partition((*iter));
450}
451
452void TWPartitionManager::Output_Partition(TWPartition* Part) {
453 unsigned long long mb = 1048576;
454
Gary Peck004d48b2012-11-21 16:28:18 -0800455 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 -0400456 if (Part->Can_Be_Mounted) {
Gary Peck004d48b2012-11-21 16:28:18 -0800457 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 -0400458 }
Gary Peck004d48b2012-11-21 16:28:18 -0800459 printf("\n Flags: ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500460 if (Part->Can_Be_Mounted)
461 printf("Can_Be_Mounted ");
Gary Peck004d48b2012-11-21 16:28:18 -0800462 if (Part->Can_Be_Wiped)
463 printf("Can_Be_Wiped ");
Hashcodedabfd492013-08-29 22:45:30 -0700464 if (Part->Use_Rm_Rf)
465 printf("Use_Rm_Rf ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500466 if (Part->Can_Be_Backed_Up)
467 printf("Can_Be_Backed_Up ");
Gary Peck004d48b2012-11-21 16:28:18 -0800468 if (Part->Wipe_During_Factory_Reset)
469 printf("Wipe_During_Factory_Reset ");
470 if (Part->Wipe_Available_in_GUI)
471 printf("Wipe_Available_in_GUI ");
472 if (Part->Is_SubPartition)
473 printf("Is_SubPartition ");
474 if (Part->Has_SubPartition)
475 printf("Has_SubPartition ");
476 if (Part->Removable)
477 printf("Removable ");
478 if (Part->Is_Present)
479 printf("IsPresent ");
480 if (Part->Can_Be_Encrypted)
481 printf("Can_Be_Encrypted ");
482 if (Part->Is_Encrypted)
483 printf("Is_Encrypted ");
484 if (Part->Is_Decrypted)
485 printf("Is_Decrypted ");
486 if (Part->Has_Data_Media)
487 printf("Has_Data_Media ");
Dees_Troy83bd4832013-05-04 12:39:56 +0000488 if (Part->Can_Encrypt_Backup)
489 printf("Can_Encrypt_Backup ");
490 if (Part->Use_Userdata_Encryption)
491 printf("Use_Userdata_Encryption ");
Gary Peck004d48b2012-11-21 16:28:18 -0800492 if (Part->Has_Android_Secure)
493 printf("Has_Android_Secure ");
494 if (Part->Is_Storage)
495 printf("Is_Storage ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500496 if (Part->Is_Settings_Storage)
497 printf("Is_Settings_Storage ");
Dees_Troy68cab492012-12-12 19:29:35 +0000498 if (Part->Ignore_Blkid)
499 printf("Ignore_Blkid ");
Ethan Yonker253368a2014-11-25 15:00:52 -0600500 if (Part->Mount_To_Decrypt)
501 printf("Mount_To_Decrypt ");
Ethan Yonker96af84a2015-01-05 14:58:36 -0600502 if (Part->Can_Flash_Img)
503 printf("Can_Flash_Img ");
Ethan Yonker66a19492015-12-10 10:19:45 -0600504 if (Part->Is_Adopted_Storage)
505 printf("Is_Adopted_Storage ");
Ethan Yonker1b190162016-12-05 15:25:19 -0600506 if (Part->SlotSelect)
507 printf("SlotSelect ");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600508 if (Part->Mount_Read_Only)
509 printf("Mount_Read_Only ");
Gary Peck004d48b2012-11-21 16:28:18 -0800510 printf("\n");
511 if (!Part->SubPartition_Of.empty())
512 printf(" SubPartition_Of: %s\n", Part->SubPartition_Of.c_str());
513 if (!Part->Symlink_Path.empty())
514 printf(" Symlink_Path: %s\n", Part->Symlink_Path.c_str());
515 if (!Part->Symlink_Mount_Point.empty())
516 printf(" Symlink_Mount_Point: %s\n", Part->Symlink_Mount_Point.c_str());
517 if (!Part->Primary_Block_Device.empty())
518 printf(" Primary_Block_Device: %s\n", Part->Primary_Block_Device.c_str());
519 if (!Part->Alternate_Block_Device.empty())
520 printf(" Alternate_Block_Device: %s\n", Part->Alternate_Block_Device.c_str());
521 if (!Part->Decrypted_Block_Device.empty())
522 printf(" Decrypted_Block_Device: %s\n", Part->Decrypted_Block_Device.c_str());
dianlujitao4879b372018-12-03 18:45:47 +0800523 if (!Part->Crypto_Key_Location.empty())
Ethan Yonker253368a2014-11-25 15:00:52 -0600524 printf(" Crypto_Key_Location: %s\n", Part->Crypto_Key_Location.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800525 if (Part->Length != 0)
526 printf(" Length: %i\n", Part->Length);
527 if (!Part->Display_Name.empty())
528 printf(" Display_Name: %s\n", Part->Display_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500529 if (!Part->Storage_Name.empty())
530 printf(" Storage_Name: %s\n", Part->Storage_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800531 if (!Part->Backup_Path.empty())
532 printf(" Backup_Path: %s\n", Part->Backup_Path.c_str());
533 if (!Part->Backup_Name.empty())
534 printf(" Backup_Name: %s\n", Part->Backup_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500535 if (!Part->Backup_Display_Name.empty())
536 printf(" Backup_Display_Name: %s\n", Part->Backup_Display_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800537 if (!Part->Backup_FileName.empty())
538 printf(" Backup_FileName: %s\n", Part->Backup_FileName.c_str());
539 if (!Part->Storage_Path.empty())
540 printf(" Storage_Path: %s\n", Part->Storage_Path.c_str());
541 if (!Part->Current_File_System.empty())
542 printf(" Current_File_System: %s\n", Part->Current_File_System.c_str());
543 if (!Part->Fstab_File_System.empty())
544 printf(" Fstab_File_System: %s\n", Part->Fstab_File_System.c_str());
545 if (Part->Format_Block_Size != 0)
Ethan Yonkerc798c9c2015-10-09 11:15:26 -0500546 printf(" Format_Block_Size: %lu\n", Part->Format_Block_Size);
Dees_Troy094207a2012-09-26 12:00:39 -0400547 if (!Part->MTD_Name.empty())
548 printf(" MTD_Name: %s\n", Part->MTD_Name.c_str());
Matt Mower029a82d2017-01-08 13:12:38 -0600549 printf(" Backup_Method: %s\n", Part->Backup_Method_By_Name().c_str());
Hashcode62bd9e02013-11-19 21:59:42 -0800550 if (Part->Mount_Flags || !Part->Mount_Options.empty())
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600551 printf(" Mount_Flags: %i, Mount_Options: %s\n", Part->Mount_Flags, Part->Mount_Options.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -0600552 if (Part->MTP_Storage_ID)
553 printf(" MTP_Storage_ID: %i\n", Part->MTP_Storage_ID);
Ethan Yonker93382822018-11-01 15:25:31 -0500554 if (!Part->Key_Directory.empty())
555 printf(" Metadata Key Directory: %s\n", Part->Key_Directory.c_str());
Ethan Yonker6277c792014-09-15 14:54:30 -0500556 printf("\n");
Dees_Troy8170a922012-09-18 15:40:25 -0400557}
558
Dees_Troy51a0e822012-09-05 15:24:24 -0400559int TWPartitionManager::Mount_By_Path(string Path, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400560 std::vector<TWPartition*>::iterator iter;
561 int ret = false;
562 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400563 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400564
bigbiffd58ba182020-03-23 10:02:29 -0400565 if (Local_Path == "/tmp" || Local_Path == "/")
Dees_Troy43d8b002012-09-17 16:00:01 -0400566 return true;
567
Dees_Troy5bf43922012-09-07 16:07:55 -0400568 // Iterate through all partitions
569 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400570 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400571 ret = (*iter)->Mount(Display_Error);
572 found = true;
Dees_Troy51127312012-09-08 13:08:49 -0400573 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400574 (*iter)->Mount(Display_Error);
Dees_Troy51127312012-09-08 13:08:49 -0400575 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400576 }
577 if (found) {
578 return ret;
579 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500580 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 -0400581 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000582 LOGINFO("Mount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400583 }
584 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400585}
586
Dees_Troy51a0e822012-09-05 15:24:24 -0400587int TWPartitionManager::UnMount_By_Path(string Path, bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400588 std::vector<TWPartition*>::iterator iter;
589 int ret = false;
590 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400591 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy51127312012-09-08 13:08:49 -0400592
593 // Iterate through all partitions
594 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400595 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy51127312012-09-08 13:08:49 -0400596 ret = (*iter)->UnMount(Display_Error);
597 found = true;
598 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
599 (*iter)->UnMount(Display_Error);
600 }
601 }
602 if (found) {
603 return ret;
604 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500605 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 -0400606 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000607 LOGINFO("UnMount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400608 }
609 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400610}
611
Dees_Troy51a0e822012-09-05 15:24:24 -0400612int TWPartitionManager::Is_Mounted_By_Path(string Path) {
Dees_Troy51127312012-09-08 13:08:49 -0400613 TWPartition* Part = Find_Partition_By_Path(Path);
614
615 if (Part)
616 return Part->Is_Mounted();
617 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000618 LOGINFO("Is_Mounted: Unable to find partition for path '%s'\n", Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400619 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400620}
621
Dees_Troy5bf43922012-09-07 16:07:55 -0400622int TWPartitionManager::Mount_Current_Storage(bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400623 string current_storage_path = DataManager::GetCurrentStoragePath();
624
625 if (Mount_By_Path(current_storage_path, Display_Error)) {
626 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
627 if (FreeStorage)
628 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
629 return true;
630 }
631 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400632}
633
Dees_Troy5bf43922012-09-07 16:07:55 -0400634int TWPartitionManager::Mount_Settings_Storage(bool Display_Error) {
635 return Mount_By_Path(DataManager::GetSettingsStoragePath(), Display_Error);
636}
637
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600638TWPartition* TWPartitionManager::Find_Partition_By_Path(const string& Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400639 std::vector<TWPartition*>::iterator iter;
Dees_Troy38bd7602012-09-14 13:33:53 -0400640 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400641
dianlujitaob76a73a2020-04-30 20:33:20 +0800642 if (Local_Path == "/system")
643 Local_Path = Get_Android_Root_Path();
Dees_Troy5bf43922012-09-07 16:07:55 -0400644 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400645 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path))
Dees_Troy5bf43922012-09-07 16:07:55 -0400646 return (*iter);
647 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400648 return NULL;
649}
650
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600651TWPartition* TWPartitionManager::Find_Partition_By_Block_Device(const string& Block_Device) {
652 std::vector<TWPartition*>::iterator iter;
653
654 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
655 if ((*iter)->Primary_Block_Device == Block_Device || (!(*iter)->Actual_Block_Device.empty() && (*iter)->Actual_Block_Device == Block_Device))
656 return (*iter);
657 }
658 return NULL;
659}
660
Ethan Yonker53796e72019-01-11 22:49:52 -0600661int TWPartitionManager::Check_Backup_Name(const std::string& Backup_Name, bool Display_Error, bool Must_Be_Unique) {
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400662 // Check the backup name to ensure that it is the correct size and contains only valid characters
663 // and that a backup with that name doesn't already exist
664 char backup_name[MAX_BACKUP_NAME_LEN];
665 char backup_loc[255], tw_image_dir[255];
666 int copy_size;
667 int index, cur_char;
Ethan Yonker53796e72019-01-11 22:49:52 -0600668 string Backup_Loc;
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400669
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400670 copy_size = Backup_Name.size();
671 // Check size
672 if (copy_size > MAX_BACKUP_NAME_LEN) {
673 if (Display_Error)
Ethan Yonker74db1572015-10-28 12:44:49 -0500674 gui_err("backup_name_len=Backup name is too long.");
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400675 return -2;
676 }
677
678 // Check each character
679 strncpy(backup_name, Backup_Name.c_str(), copy_size);
Dees_Troya13d74f2013-03-24 08:54:55 -0500680 if (copy_size == 1 && strncmp(backup_name, "0", 1) == 0)
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400681 return 0; // A "0" (zero) means to use the current timestamp for the backup name
682 for (index=0; index<copy_size; index++) {
683 cur_char = (int)backup_name[index];
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500684 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 -0400685 // These are valid characters
686 // Numbers
687 // Upper case letters
688 // Lower case letters
689 // Space
690 // and -_.{}[]
691 } else {
692 if (Display_Error)
Ethan Yonker4adc33e2016-01-22 16:07:11 -0600693 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 -0400694 return -3;
695 }
696 }
697
Ethan Yonker53796e72019-01-11 22:49:52 -0600698 if (Must_Be_Unique) {
699 // Check to make sure that a backup with this name doesn't already exist
700 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Loc);
701 strcpy(backup_loc, Backup_Loc.c_str());
702 sprintf(tw_image_dir,"%s/%s", backup_loc, Backup_Name.c_str());
703 if (TWFunc::Path_Exists(tw_image_dir)) {
704 if (Display_Error)
705 gui_err("backup_name_exists=A backup with that name already exists!");
bigbiffce8f83c2015-12-12 18:30:21 -0500706
Ethan Yonker53796e72019-01-11 22:49:52 -0600707 return -4;
708 }
709 // Backup is unique
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400710 }
Ethan Yonker53796e72019-01-11 22:49:52 -0600711 // No problems found
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400712 return 0;
713}
714
bigbiffce8f83c2015-12-12 18:30:21 -0500715bool TWPartitionManager::Backup_Partition(PartitionSettings *part_settings) {
Dees_Troy43d8b002012-09-17 16:00:01 -0400716 time_t start, stop;
Matt Mower23d8aae2017-01-06 14:30:33 -0600717 int use_compression;
bigbiffce8f83c2015-12-12 18:30:21 -0500718 string backup_log = part_settings->Backup_Folder + "/recovery.log";
Dees_Troy43d8b002012-09-17 16:00:01 -0400719
bigbiffce8f83c2015-12-12 18:30:21 -0500720 if (part_settings->Part == NULL)
Dees_Troy43d8b002012-09-17 16:00:01 -0400721 return true;
722
Dees_Troy093b7642012-09-21 15:59:38 -0400723 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400724
Tom Hite5a926722014-09-15 01:31:03 +0000725 TWFunc::SetPerformanceMode(true);
Dees_Troy43d8b002012-09-17 16:00:01 -0400726 time(&start);
727
bigbiffce8f83c2015-12-12 18:30:21 -0500728 if (part_settings->Part->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500729 sync();
730 sync();
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400731 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500732 if (!part_settings->adbbackup && part_settings->generate_digest) {
733 if (!twrpDigestDriver::Make_Digest(Full_Filename))
734 goto backup_error;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400735 }
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400736
bigbiffce8f83c2015-12-12 18:30:21 -0500737 if (part_settings->Part->Has_SubPartition) {
Dees_Troy8170a922012-09-18 15:40:25 -0400738 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400739 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -0400740
741 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400742 if ((*subpart)->Can_Be_Backed_Up && (*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500743 part_settings->Part = *subpart;
bigbiffce8f83c2015-12-12 18:30:21 -0500744 if (!(*subpart)->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500745 goto backup_error;
Tom Hite5a926722014-09-15 01:31:03 +0000746 }
Dees_Troy2727b992013-08-14 20:09:30 +0000747 sync();
748 sync();
bigbiff bigbifff5955b12019-05-18 17:28:58 -0400749 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400750 if (!part_settings->adbbackup && part_settings->generate_digest) {
751 if (!twrpDigestDriver::Make_Digest(Full_Filename)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500752 goto backup_error;
bigbiffce8f83c2015-12-12 18:30:21 -0500753 }
Tom Hite5a926722014-09-15 01:31:03 +0000754 }
Dees_Troy8170a922012-09-18 15:40:25 -0400755 }
756 }
757 }
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400758
Dees_Troy43d8b002012-09-17 16:00:01 -0400759 time(&stop);
that203bcc92015-05-09 17:27:33 +0200760 int backup_time = (int) difftime(stop, start);
Dees_Troy2673cec2013-04-02 20:22:16 +0000761 LOGINFO("Partition Backup time: %d\n", backup_time);
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400762
bigbiffce8f83c2015-12-12 18:30:21 -0500763 if (part_settings->Part->Backup_Method == BM_FILES) {
764 part_settings->file_time += backup_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400765 } else {
bigbiffce8f83c2015-12-12 18:30:21 -0500766 part_settings->img_time += backup_time;
767
Dees_Troy43d8b002012-09-17 16:00:01 -0400768 }
Tom Hite5a926722014-09-15 01:31:03 +0000769
Matt Mower02899552016-12-30 18:13:51 -0600770 TWFunc::SetPerformanceMode(false);
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500771 return true;
Dees_Troy43d8b002012-09-17 16:00:01 -0400772 }
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500773backup_error:
774 Clean_Backup_Folder(part_settings->Backup_Folder);
775 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
776 tw_set_default_metadata(backup_log.c_str());
777 TWFunc::SetPerformanceMode(false);
778 return false;
bigbiff7abc5fe2015-01-17 16:53:12 -0500779}
780
bigbiffbf1d6722015-02-14 16:25:59 -0500781void TWPartitionManager::Clean_Backup_Folder(string Backup_Folder) {
782 DIR *d = opendir(Backup_Folder.c_str());
783 struct dirent *p;
784 int r;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400785 vector<string> ext;
786
787 //extensions we should delete when cleaning
788 ext.push_back("win");
789 ext.push_back("md5");
790 ext.push_back("sha2");
791 ext.push_back("info");
bigbiffbf1d6722015-02-14 16:25:59 -0500792
Ethan Yonker74db1572015-10-28 12:44:49 -0500793 gui_msg("backup_clean=Backup Failed. Cleaning Backup Folder.");
bigbiffbf1d6722015-02-14 16:25:59 -0500794
795 if (d == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500796 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Backup_Folder)(strerror(errno)));
bigbiffbf1d6722015-02-14 16:25:59 -0500797 return;
798 }
799
Matt Mower2b18a532015-02-20 16:58:05 -0600800 while ((p = readdir(d))) {
bigbiffbf1d6722015-02-14 16:25:59 -0500801 if (!strcmp(p->d_name, ".") || !strcmp(p->d_name, ".."))
802 continue;
803
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500804 string path = Backup_Folder + "/" + p->d_name;
bigbiffbf1d6722015-02-14 16:25:59 -0500805
806 size_t dot = path.find_last_of(".") + 1;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400807 for (vector<string>::const_iterator i = ext.begin(); i != ext.end(); ++i) {
808 if (path.substr(dot) == *i) {
809 r = unlink(path.c_str());
810 if (r != 0)
811 LOGINFO("Unable to unlink '%s: %s'\n", path.c_str(), strerror(errno));
bigbiffbf1d6722015-02-14 16:25:59 -0500812 }
813 }
814 }
815 closedir(d);
816}
817
Ethan Yonker472f5062016-02-25 13:47:30 -0600818int TWPartitionManager::Check_Backup_Cancel() {
819 return stop_backup.get_value();
820}
821
bigbiff7abc5fe2015-01-17 16:53:12 -0500822int TWPartitionManager::Cancel_Backup() {
823 string Backup_Folder, Backup_Name, Full_Backup_Path;
824
825 stop_backup.set_value(1);
826
827 if (tar_fork_pid != 0) {
828 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
829 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Folder);
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500830 Full_Backup_Path = Backup_Folder + "/" + Backup_Name;
bigbiff7abc5fe2015-01-17 16:53:12 -0500831 LOGINFO("Killing pid: %d\n", tar_fork_pid);
832 kill(tar_fork_pid, SIGUSR2);
833 while (kill(tar_fork_pid, 0) == 0) {
834 usleep(1000);
835 }
836 LOGINFO("Backup_Run stopped and returning false, backup cancelled.\n");
837 LOGINFO("Removing directory %s\n", Full_Backup_Path.c_str());
838 TWFunc::removeDir(Full_Backup_Path, false);
839 tar_fork_pid = 0;
840 }
841
842 return 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400843}
844
bigbiffce8f83c2015-12-12 18:30:21 -0500845int TWPartitionManager::Run_Backup(bool adbbackup) {
846 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400847 int partition_count = 0, disable_free_space_check = 0, skip_digest = 0;
bigbiffce8f83c2015-12-12 18:30:21 -0500848 string Backup_Name, Backup_List, backup_path;
Matt Mower23d8aae2017-01-06 14:30:33 -0600849 unsigned long long total_bytes = 0, free_space = 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400850 TWPartition* storage = NULL;
851 struct tm *t;
Matt Mower23d8aae2017-01-06 14:30:33 -0600852 time_t seconds, total_start, total_stop;
Dees_Troya13d74f2013-03-24 08:54:55 -0500853 size_t start_pos = 0, end_pos = 0;
bigbiff7abc5fe2015-01-17 16:53:12 -0500854 stop_backup.set_value(0);
Dees_Troy43d8b002012-09-17 16:00:01 -0400855 seconds = time(0);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500856 t = localtime(&seconds);
Dees_Troy43d8b002012-09-17 16:00:01 -0400857
bigbiffce8f83c2015-12-12 18:30:21 -0500858 part_settings.img_bytes_remaining = 0;
859 part_settings.file_bytes_remaining = 0;
860 part_settings.img_time = 0;
861 part_settings.file_time = 0;
862 part_settings.img_bytes = 0;
863 part_settings.file_bytes = 0;
864 part_settings.PM_Method = PM_BACKUP;
865
bigbiffce8f83c2015-12-12 18:30:21 -0500866 part_settings.adbbackup = adbbackup;
Dees_Troy43d8b002012-09-17 16:00:01 -0400867 time(&total_start);
868
869 Update_System_Details();
870
871 if (!Mount_Current_Storage(true))
872 return false;
873
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400874 DataManager::GetValue(TW_SKIP_DIGEST_GENERATE_VAR, skip_digest);
875 if (skip_digest == 0)
876 part_settings.generate_digest = true;
Dees_Troyc5865ab2012-09-24 15:08:04 -0400877 else
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400878 part_settings.generate_digest = false;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400879
bigbiffce8f83c2015-12-12 18:30:21 -0500880 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, part_settings.Backup_Folder);
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500881 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
882 if (Backup_Name == gui_lookup("curr_date", "(Current Date)")) {
883 Backup_Name = TWFunc::Get_Current_Date();
884 } else if (Backup_Name == gui_lookup("auto_generate", "(Auto Generate)") || Backup_Name == "0" || Backup_Name.empty()) {
Dees Troyb21cc642013-09-10 17:36:41 +0000885 TWFunc::Auto_Generate_Backup_Name();
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500886 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
Dees_Troy43d8b002012-09-17 16:00:01 -0400887 }
bigbiffce8f83c2015-12-12 18:30:21 -0500888
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500889 LOGINFO("Backup Name is: '%s'\n", Backup_Name.c_str());
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500890 part_settings.Backup_Folder = part_settings.Backup_Folder + "/" + Backup_Name;
bigbiffce8f83c2015-12-12 18:30:21 -0500891
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500892 LOGINFO("Backup_Folder is: '%s'\n", part_settings.Backup_Folder.c_str());
Dees_Troy43d8b002012-09-17 16:00:01 -0400893
Dees_Troy2673cec2013-04-02 20:22:16 +0000894 LOGINFO("Calculating backup details...\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500895 DataManager::GetValue("tw_backup_list", Backup_List);
bigbiff7ba75002020-04-11 20:47:09 -0400896 LOGINFO("Backup_List: %s\n", Backup_List.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500897 if (!Backup_List.empty()) {
898 end_pos = Backup_List.find(";", start_pos);
899 while (end_pos != string::npos && start_pos < Backup_List.size()) {
900 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiff7ba75002020-04-11 20:47:09 -0400901 LOGINFO("backup_path: %s\n", backup_path.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -0500902 part_settings.Part = Find_Partition_By_Path(backup_path);
903 if (part_settings.Part != NULL) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500904 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500905 if (part_settings.Part->Backup_Method == BM_FILES)
906 part_settings.file_bytes += part_settings.Part->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500907 else
bigbiffce8f83c2015-12-12 18:30:21 -0500908 part_settings.img_bytes += part_settings.Part->Backup_Size;
909 if (part_settings.Part->Has_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500910 std::vector<TWPartition*>::iterator subpart;
911
912 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiffce8f83c2015-12-12 18:30:21 -0500913 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 -0500914 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500915 if ((*subpart)->Backup_Method == BM_FILES)
916 part_settings.file_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500917 else
bigbiffce8f83c2015-12-12 18:30:21 -0500918 part_settings.img_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500919 }
920 }
Dees_Troy8170a922012-09-18 15:40:25 -0400921 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500922 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500923 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 -0400924 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500925 start_pos = end_pos + 1;
926 end_pos = Backup_List.find(";", start_pos);
Dees_Troy43d8b002012-09-17 16:00:01 -0400927 }
928 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400929
930 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500931 gui_msg("no_partition_selected=No partitions selected for backup.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400932 return false;
933 }
bigbiffce8f83c2015-12-12 18:30:21 -0500934 if (adbbackup) {
935 if (twadbbu::Write_ADB_Stream_Header(partition_count) == false) {
936 return false;
937 }
938 }
939 total_bytes = part_settings.file_bytes + part_settings.img_bytes;
Ethan Yonker472f5062016-02-25 13:47:30 -0600940 ProgressTracking progress(total_bytes);
bigbiffce8f83c2015-12-12 18:30:21 -0500941 part_settings.progress = &progress;
942
Ethan Yonker74db1572015-10-28 12:44:49 -0500943 gui_msg(Msg("total_partitions_backup= * Total number of partitions to back up: {1}")(partition_count));
944 gui_msg(Msg("total_backup_size= * Total size of all data: {1}MB")(total_bytes / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400945 storage = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
946 if (storage != NULL) {
947 free_space = storage->Free;
Ethan Yonker74db1572015-10-28 12:44:49 -0500948 gui_msg(Msg("available_space= * Available space: {1}MB")(free_space / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400949 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500950 gui_err("unable_locate_storage=Unable to locate storage device.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400951 return false;
952 }
bigbiffbf1d6722015-02-14 16:25:59 -0500953
Matt Mowerbfccfb82016-04-25 23:22:31 -0500954 DataManager::GetValue(TW_DISABLE_FREE_SPACE_VAR, disable_free_space_check);
bigbiffce8f83c2015-12-12 18:30:21 -0500955
956 if (adbbackup)
957 disable_free_space_check = true;
958
bigbiffbf1d6722015-02-14 16:25:59 -0500959 if (!disable_free_space_check) {
960 if (free_space - (32 * 1024 * 1024) < total_bytes) {
961 // We require an extra 32MB just in case
Ethan Yonker74db1572015-10-28 12:44:49 -0500962 gui_err("no_space=Not enough free space on storage.");
bigbiffbf1d6722015-02-14 16:25:59 -0500963 return false;
964 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400965 }
bigbiffce8f83c2015-12-12 18:30:21 -0500966 part_settings.img_bytes_remaining = part_settings.img_bytes;
967 part_settings.file_bytes_remaining = part_settings.file_bytes;
Dees_Troy43d8b002012-09-17 16:00:01 -0400968
Ethan Yonker74db1572015-10-28 12:44:49 -0500969 gui_msg("backup_started=[BACKUP STARTED]");
bigbiff bigbiffcdd97c72019-03-29 19:12:33 -0400970
971 int is_decrypted = 0;
972 int is_encrypted = 0;
973
974 DataManager::GetValue(TW_IS_DECRYPTED, is_decrypted);
975 DataManager::GetValue(TW_IS_ENCRYPTED, is_encrypted);
976 if (!adbbackup || (!is_encrypted || (is_encrypted && is_decrypted))) {
977 gui_msg(Msg("backup_folder= * Backup Folder: {1}")(part_settings.Backup_Folder));
978 if (!TWFunc::Recursive_Mkdir(part_settings.Backup_Folder)) {
979 gui_err("fail_backup_folder=Failed to make backup folder.");
980 return false;
981 }
Dees_Troyd4b22b02013-01-18 17:17:58 +0000982 }
983
Dees_Troy2673cec2013-04-02 20:22:16 +0000984 DataManager::SetProgress(0.0);
Dees_Troy093b7642012-09-21 15:59:38 -0400985
Dees_Troya13d74f2013-03-24 08:54:55 -0500986 start_pos = 0;
987 end_pos = Backup_List.find(";", start_pos);
988 while (end_pos != string::npos && start_pos < Backup_List.size()) {
bigbiff7abc5fe2015-01-17 16:53:12 -0500989 if (stop_backup.get_value() != 0)
990 return -1;
Dees_Troya13d74f2013-03-24 08:54:55 -0500991 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -0500992 part_settings.Part = Find_Partition_By_Path(backup_path);
993 if (part_settings.Part != NULL) {
994 if (!Backup_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -0500995 return false;
996 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500997 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 -0500998 }
999 start_pos = end_pos + 1;
1000 end_pos = Backup_List.find(";", start_pos);
1001 }
Dees_Troy43d8b002012-09-17 16:00:01 -04001002
1003 // Average BPS
bigbiffce8f83c2015-12-12 18:30:21 -05001004 if (part_settings.img_time == 0)
1005 part_settings.img_time = 1;
1006 if (part_settings.file_time == 0)
1007 part_settings.file_time = 1;
1008 int img_bps = (int)part_settings.img_bytes / (int)part_settings.img_time;
1009 unsigned long long file_bps = part_settings.file_bytes / (int)part_settings.file_time;
Dees_Troy43d8b002012-09-17 16:00:01 -04001010
bigbiffce8f83c2015-12-12 18:30:21 -05001011 if (part_settings.file_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -06001012 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 -05001013 if (part_settings.img_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -06001014 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 -04001015
1016 time(&total_stop);
1017 int total_time = (int) difftime(total_stop, total_start);
bigbiffce8f83c2015-12-12 18:30:21 -05001018
1019 uint64_t actual_backup_size;
Ethan Yonker3fdcda42016-11-30 12:29:37 -06001020 if (!adbbackup) {
1021 TWExclude twe;
1022 actual_backup_size = twe.Get_Folder_Size(part_settings.Backup_Folder);
1023 } else
bigbiffce8f83c2015-12-12 18:30:21 -05001024 actual_backup_size = part_settings.file_bytes + part_settings.img_bytes;
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001025 actual_backup_size /= (1024LLU * 1024LLU);
Dees_Troy43d8b002012-09-17 16:00:01 -04001026
bigbiffce8f83c2015-12-12 18:30:21 -05001027 int prev_img_bps = 0, use_compression = 0;
1028 unsigned long long prev_file_bps = 0;
Matt Mower173cdb92016-12-31 02:49:19 -06001029 DataManager::GetValue(TW_BACKUP_AVG_IMG_RATE, prev_img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -04001030 img_bps += (prev_img_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001031 img_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -04001032
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001033 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -04001034 if (use_compression)
1035 DataManager::GetValue(TW_BACKUP_AVG_FILE_COMP_RATE, prev_file_bps);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001036 else
Dees_Troy093b7642012-09-21 15:59:38 -04001037 DataManager::GetValue(TW_BACKUP_AVG_FILE_RATE, prev_file_bps);
1038 file_bps += (prev_file_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001039 file_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -04001040
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001041 DataManager::SetValue(TW_BACKUP_AVG_IMG_RATE, img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -04001042 if (use_compression)
1043 DataManager::SetValue(TW_BACKUP_AVG_FILE_COMP_RATE, file_bps);
1044 else
1045 DataManager::SetValue(TW_BACKUP_AVG_FILE_RATE, file_bps);
1046
Ethan Yonker74db1572015-10-28 12:44:49 -05001047 gui_msg(Msg("total_backed_size=[{1} MB TOTAL BACKED UP]")(actual_backup_size));
Dees_Troy43d8b002012-09-17 16:00:01 -04001048 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001049 UnMount_Main_Partitions();
Ethan Yonker56db1c42015-12-20 22:49:00 -06001050 gui_msg(Msg(msg::kHighlight, "backup_completed=[BACKUP COMPLETED IN {1} SECONDS]")(total_time)); // the end
Ethan Yonkere080c1f2016-09-19 13:50:25 -05001051 string backup_log = part_settings.Backup_Folder + "/recovery.log";
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001052 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
Ethan Yonker4b94cfd2014-12-11 10:00:45 -06001053 tw_set_default_metadata(backup_log.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -05001054
1055 if (part_settings.adbbackup) {
1056 if (twadbbu::Write_ADB_Stream_Trailer() == false) {
1057 return false;
1058 }
1059 }
1060 part_settings.adbbackup = false;
1061 DataManager::SetValue("tw_enable_adb_backup", 0);
1062
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001063 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001064}
1065
bigbiffce8f83c2015-12-12 18:30:21 -05001066bool TWPartitionManager::Restore_Partition(PartitionSettings *part_settings) {
Dees_Troy4a2a1262012-09-18 09:33:47 -04001067 time_t Start, Stop;
bigbiffce8f83c2015-12-12 18:30:21 -05001068
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001069 if (part_settings->adbbackup) {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001070 std::string partName = part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win";
1071 LOGINFO("setting backup name: %s\n", partName.c_str());
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001072 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
1073 }
1074
Tom Hite5a926722014-09-15 01:31:03 +00001075 TWFunc::SetPerformanceMode(true);
bigbiffce8f83c2015-12-12 18:30:21 -05001076
Dees_Troy4a2a1262012-09-18 09:33:47 -04001077 time(&Start);
Ethan Yonker472f5062016-02-25 13:47:30 -06001078
bigbiffce8f83c2015-12-12 18:30:21 -05001079 if (!part_settings->Part->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001080 TWFunc::SetPerformanceMode(false);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001081 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001082 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001083 if (part_settings->Part->Has_SubPartition && !part_settings->adbbackup) {
Dees_Troy8170a922012-09-18 15:40:25 -04001084 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001085 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -04001086
1087 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001088 part_settings->Part = *subpart;
1089 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
1090 part_settings->Part = (*subpart);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001091 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
bigbiffce8f83c2015-12-12 18:30:21 -05001092 if (!(*subpart)->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001093 TWFunc::SetPerformanceMode(false);
Dees_Troy8170a922012-09-18 15:40:25 -04001094 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001095 }
Dees_Troy8170a922012-09-18 15:40:25 -04001096 }
1097 }
1098 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001099 time(&Stop);
Tom Hite5a926722014-09-15 01:31:03 +00001100 TWFunc::SetPerformanceMode(false);
bigbiffce8f83c2015-12-12 18:30:21 -05001101 gui_msg(Msg("restore_part_done=[{1} done ({2} seconds)]")(part_settings->Part->Backup_Display_Name)((int)difftime(Stop, Start)));
1102
Dees_Troy4a2a1262012-09-18 09:33:47 -04001103 return true;
1104}
1105
Ethan Yonker472f5062016-02-25 13:47:30 -06001106int TWPartitionManager::Run_Restore(const string& Restore_Name) {
bigbiffce8f83c2015-12-12 18:30:21 -05001107 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001108 int check_digest;
bigbiffce8f83c2015-12-12 18:30:21 -05001109
Dees_Troy4a2a1262012-09-18 09:33:47 -04001110 time_t rStart, rStop;
1111 time(&rStart);
Dees_Troya13d74f2013-03-24 08:54:55 -05001112 string Restore_List, restore_path;
1113 size_t start_pos = 0, end_pos;
bigbiffce8f83c2015-12-12 18:30:21 -05001114
Ethan Yonkerdcf2b672016-09-13 14:41:53 -05001115 part_settings.Backup_Folder = Restore_Name;
bigbiffce8f83c2015-12-12 18:30:21 -05001116 part_settings.Part = NULL;
1117 part_settings.partition_count = 0;
1118 part_settings.total_restore_size = 0;
1119 part_settings.adbbackup = false;
1120 part_settings.PM_Method = PM_RESTORE;
Dees_Troy43d8b002012-09-17 16:00:01 -04001121
Ethan Yonker74db1572015-10-28 12:44:49 -05001122 gui_msg("restore_started=[RESTORE STARTED]");
1123 gui_msg(Msg("restore_folder=Restore folder: '{1}'")(Restore_Name));
Dees_Troy43d8b002012-09-17 16:00:01 -04001124
Dees_Troy4a2a1262012-09-18 09:33:47 -04001125 if (!Mount_Current_Storage(true))
1126 return false;
1127
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001128 DataManager::GetValue(TW_SKIP_DIGEST_CHECK_VAR, check_digest);
1129 if (check_digest > 0) {
1130 // Check Digest files first before restoring to ensure that all of them match before starting a restore
1131 TWFunc::GUI_Operation_Text(TW_VERIFY_DIGEST_TEXT, gui_parse_text("{@verifying_digest}"));
1132 gui_msg("verifying_digest=Verifying Digest");
Dees_Troya13d74f2013-03-24 08:54:55 -05001133 } else {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001134 gui_msg("skip_digest=Skipping Digest check based on user setting.");
Dees_Troya13d74f2013-03-24 08:54:55 -05001135 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001136 gui_msg("calc_restore=Calculating restore details...");
Dees_Troya13d74f2013-03-24 08:54:55 -05001137 DataManager::GetValue("tw_restore_selected", Restore_List);
bigbiffce8f83c2015-12-12 18:30:21 -05001138
Dees_Troya13d74f2013-03-24 08:54:55 -05001139 if (!Restore_List.empty()) {
1140 end_pos = Restore_List.find(";", start_pos);
1141 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1142 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001143 part_settings.Part = Find_Partition_By_Path(restore_path);
1144 if (part_settings.Part != NULL) {
bigbiffce8f83c2015-12-12 18:30:21 -05001145 if (part_settings.Part->Mount_Read_Only) {
1146 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 -05001147 return false;
1148 }
bigbiffce8f83c2015-12-12 18:30:21 -05001149
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001150 string Full_Filename = part_settings.Backup_Folder + "/" + part_settings.Part->Backup_FileName;
1151
1152 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001153 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001154 part_settings.partition_count++;
1155 part_settings.total_restore_size += part_settings.Part->Get_Restore_Size(&part_settings);
1156 if (part_settings.Part->Has_SubPartition) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001157 TWPartition *parentPart = part_settings.Part;
Dees_Troya13d74f2013-03-24 08:54:55 -05001158 std::vector<TWPartition*>::iterator subpart;
1159
1160 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001161 part_settings.Part = *subpart;
1162 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001163 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001164 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001165 part_settings.total_restore_size += (*subpart)->Get_Restore_Size(&part_settings);
Dees_Troya13d74f2013-03-24 08:54:55 -05001166 }
1167 }
1168 }
1169 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001170 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001171 }
1172 start_pos = end_pos + 1;
1173 end_pos = Restore_List.find(";", start_pos);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001174 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001175 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001176
bigbiffce8f83c2015-12-12 18:30:21 -05001177 if (part_settings.partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001178 gui_err("no_part_restore=No partitions selected for restore.");
Dees_Troy4a2a1262012-09-18 09:33:47 -04001179 return false;
1180 }
1181
bigbiffce8f83c2015-12-12 18:30:21 -05001182 gui_msg(Msg("restore_part_count=Restoring {1} partitions...")(part_settings.partition_count));
1183 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 +00001184 DataManager::SetProgress(0.0);
bigbiffce8f83c2015-12-12 18:30:21 -05001185 ProgressTracking progress(part_settings.total_restore_size);
1186 part_settings.progress = &progress;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001187
Dees_Troya13d74f2013-03-24 08:54:55 -05001188 start_pos = 0;
1189 if (!Restore_List.empty()) {
1190 end_pos = Restore_List.find(";", start_pos);
1191 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1192 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001193
1194 part_settings.Part = Find_Partition_By_Path(restore_path);
1195 if (part_settings.Part != NULL) {
1196 part_settings.partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -05001197 if (!Restore_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001198 return false;
1199 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001200 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001201 }
1202 start_pos = end_pos + 1;
1203 end_pos = Restore_List.find(";", start_pos);
1204 }
1205 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001206 TWFunc::GUI_Operation_Text(TW_UPDATE_SYSTEM_DETAILS_TEXT, gui_parse_text("{@updating_system_details}"));
Captain Throwback9d6feb52018-07-27 10:05:24 -04001207 UnMount_By_Path(Get_Android_Root_Path(), false);
Dees_Troy43d8b002012-09-17 16:00:01 -04001208 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001209 UnMount_Main_Partitions();
Dees_Troy4a2a1262012-09-18 09:33:47 -04001210 time(&rStop);
Matt Mower3c366972015-12-25 19:28:31 -06001211 gui_msg(Msg(msg::kHighlight, "restore_completed=[RESTORE COMPLETED IN {1} SECONDS]")((int)difftime(rStop,rStart)));
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001212 DataManager::SetValue("tw_file_progress", "");
bigbiffce8f83c2015-12-12 18:30:21 -05001213
Dees_Troy63c8df72012-09-10 14:02:05 -04001214 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001215}
1216
1217void TWPartitionManager::Set_Restore_Files(string Restore_Name) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001218 // Start with the default values
Dees_Troya13d74f2013-03-24 08:54:55 -05001219 string Restore_List;
Dees_Troy83bd4832013-05-04 12:39:56 +00001220 bool get_date = true, check_encryption = true;
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001221 bool adbbackup = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001222
1223 DataManager::SetValue("tw_restore_encrypted", 0);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001224 if (twadbbu::Check_ADB_Backup_File(Restore_Name)) {
1225 vector<string> adb_files;
1226 adb_files = twadbbu::Get_ADB_Backup_Files(Restore_Name);
1227 for (unsigned int i = 0; i < adb_files.size(); ++i) {
1228 string adb_restore_file = adb_files.at(i);
1229 std::size_t pos = adb_restore_file.find_first_of(".");
1230 std::string path = "/" + adb_restore_file.substr(0, pos);
1231 Restore_List = path + ";";
1232 TWPartition* Part = Find_Partition_By_Path(path);
1233 Part->Backup_FileName = TWFunc::Get_Filename(adb_restore_file);
1234 adbbackup = true;
1235 }
1236 DataManager::SetValue("tw_enable_adb_backup", 1);
Dees_Troy63c8df72012-09-10 14:02:05 -04001237 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001238 else {
1239 DIR* d;
1240 d = opendir(Restore_Name.c_str());
1241 if (d == NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001242 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001243 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Restore_Name)(strerror(errno)));
1244 return;
Dees_Troy63c8df72012-09-10 14:02:05 -04001245 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001246
1247 struct dirent* de;
1248 while ((de = readdir(d)) != NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001249 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001250 // Strip off three components
1251 char str[256];
1252 char* label;
1253 char* fstype = NULL;
1254 char* extn = NULL;
1255 char* ptr;
Dees_Troy63c8df72012-09-10 14:02:05 -04001256
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001257 strcpy(str, de->d_name);
1258 if (strlen(str) <= 2)
1259 continue;
Dees_Troy83bd4832013-05-04 12:39:56 +00001260
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001261 if (get_date) {
1262 char file_path[255];
1263 struct stat st;
1264
1265 strcpy(file_path, Restore_Name.c_str());
1266 strcat(file_path, "/");
1267 strcat(file_path, str);
1268 stat(file_path, &st);
1269 string backup_date = ctime((const time_t*)(&st.st_mtime));
1270 DataManager::SetValue(TW_RESTORE_FILE_DATE, backup_date);
1271 get_date = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001272 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001273
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001274 label = str;
1275 ptr = label;
1276 while (*ptr && *ptr != '.') ptr++;
1277 if (*ptr == '.')
1278 {
1279 *ptr = 0x00;
1280 ptr++;
1281 fstype = ptr;
1282 }
1283 while (*ptr && *ptr != '.') ptr++;
1284 if (*ptr == '.')
1285 {
1286 *ptr = 0x00;
1287 ptr++;
1288 extn = ptr;
1289 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001290
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001291 if (fstype == NULL || extn == NULL || strcmp(fstype, "log") == 0) continue;
1292 int extnlength = strlen(extn);
1293 if (extnlength != 3 && extnlength != 6) continue;
1294 if (extnlength >= 3 && strncmp(extn, "win", 3) != 0) continue;
1295 //if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
Dees_Troy63c8df72012-09-10 14:02:05 -04001296
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001297 if (check_encryption) {
1298 string filename = Restore_Name + "/";
1299 filename += de->d_name;
1300 if (TWFunc::Get_File_Type(filename) == 2) {
1301 LOGINFO("'%s' is encrypted\n", filename.c_str());
1302 DataManager::SetValue("tw_restore_encrypted", 1);
1303 }
1304 }
1305 if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
1306
1307 TWPartition* Part = Find_Partition_By_Path(label);
1308 if (Part == NULL)
1309 {
1310 gui_msg(Msg(msg::kError, "unable_locate_part_backup_name=Unable to locate partition by backup name: '{1}'")(label));
1311 continue;
1312 }
1313
1314 Part->Backup_FileName = de->d_name;
1315 if (strlen(extn) > 3) {
1316 Part->Backup_FileName.resize(Part->Backup_FileName.size() - strlen(extn) + 3);
1317 }
1318
dianlujitaob76a73a2020-04-30 20:33:20 +08001319 if (!Part->Is_SubPartition) {
1320 if (Part->Backup_Path == Get_Android_Root_Path())
1321 Restore_List += "/system;";
1322 else
1323 Restore_List += Part->Backup_Path + ";";
1324 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001325 }
1326 closedir(d);
Dees_Troy63c8df72012-09-10 14:02:05 -04001327 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001328
1329 if (adbbackup) {
1330 Restore_List = "ADB_Backup;";
1331 adbbackup = false;
1332 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001333
Dees_Troya13d74f2013-03-24 08:54:55 -05001334 // Set the final value
1335 DataManager::SetValue("tw_restore_list", Restore_List);
1336 DataManager::SetValue("tw_restore_selected", Restore_List);
Dees_Troy51a0e822012-09-05 15:24:24 -04001337 return;
1338}
1339
1340int TWPartitionManager::Wipe_By_Path(string Path) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001341 std::vector<TWPartition*>::iterator iter;
Mohd Farazfdd49bb2020-04-21 18:56:44 +05301342 std::vector < TWPartition * >::iterator iter1;
Dees_Troy63c8df72012-09-10 14:02:05 -04001343 int ret = false;
1344 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -04001345 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy63c8df72012-09-10 14:02:05 -04001346
dianlujitaob76a73a2020-04-30 20:33:20 +08001347 if (Local_Path == "/system")
1348 Local_Path = Get_Android_Root_Path();
Dees_Troy63c8df72012-09-10 14:02:05 -04001349 // Iterate through all partitions
1350 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -04001351 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Mohd Farazfdd49bb2020-04-21 18:56:44 +05301352 // iterate through all partitions since some legacy devices uses other partitions as vendor causes issues while wiping
1353 (*iter)->Find_Actual_Block_Device();
1354 for (iter1 = Partitions.begin (); iter1 != Partitions.end (); iter1++)
1355 {
1356 (*iter1)->Find_Actual_Block_Device();
1357 if ((*iter)->Actual_Block_Device == (*iter1)->Actual_Block_Device && (*iter)->Mount_Point != (*iter1)->Mount_Point)
1358 (*iter1)->UnMount(false);
1359 }
Dees_Troye58d5262012-09-21 12:27:57 -04001360 if (Path == "/and-sec")
1361 ret = (*iter)->Wipe_AndSec();
1362 else
1363 ret = (*iter)->Wipe();
Dees_Troy63c8df72012-09-10 14:02:05 -04001364 found = true;
1365 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1366 (*iter)->Wipe();
1367 }
1368 }
1369 if (found) {
1370 return ret;
1371 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001372 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 -04001373 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -04001374}
1375
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001376int TWPartitionManager::Wipe_By_Path(string Path, string New_File_System) {
1377 std::vector<TWPartition*>::iterator iter;
1378 int ret = false;
1379 bool found = false;
1380 string Local_Path = TWFunc::Get_Root_Path(Path);
1381
1382 // Iterate through all partitions
1383 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1384 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1385 if (Path == "/and-sec")
1386 ret = (*iter)->Wipe_AndSec();
1387 else
1388 ret = (*iter)->Wipe(New_File_System);
1389 found = true;
1390 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1391 (*iter)->Wipe(New_File_System);
1392 }
1393 }
1394 if (found) {
1395 return ret;
1396 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001397 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 -05001398 return false;
1399}
1400
Dees_Troy51a0e822012-09-05 15:24:24 -04001401int TWPartitionManager::Factory_Reset(void) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001402 std::vector<TWPartition*>::iterator iter;
1403 int ret = true;
1404
1405 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001406 if ((*iter)->Wipe_During_Factory_Reset && (*iter)->Is_Present) {
Ethan Yonker89583ef2015-08-26 09:01:59 -05001407#ifdef TW_OEM_BUILD
1408 if ((*iter)->Mount_Point == "/data") {
1409 if (!(*iter)->Wipe_Encryption())
1410 ret = false;
1411 } else {
1412#endif
1413 if (!(*iter)->Wipe())
1414 ret = false;
1415#ifdef TW_OEM_BUILD
1416 }
1417#endif
Dees_Troy094207a2012-09-26 12:00:39 -04001418 } else if ((*iter)->Has_Android_Secure) {
1419 if (!(*iter)->Wipe_AndSec())
1420 ret = false;
Dees_Troy63c8df72012-09-10 14:02:05 -04001421 }
1422 }
Ethan Yonker9fa76ba2016-06-30 14:05:43 -05001423 TWFunc::check_and_run_script("/sbin/factoryreset.sh", "Factory Reset Script");
Dees_Troy63c8df72012-09-10 14:02:05 -04001424 return ret;
Dees_Troy51a0e822012-09-05 15:24:24 -04001425}
1426
Dees_Troy38bd7602012-09-14 13:33:53 -04001427int TWPartitionManager::Wipe_Dalvik_Cache(void) {
1428 struct stat st;
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001429 vector <string> dir;
Dees_Troy38bd7602012-09-14 13:33:53 -04001430
1431 if (!Mount_By_Path("/data", true))
1432 return false;
1433
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001434 dir.push_back("/data/dalvik-cache");
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001435
bigbiff25d25b92020-06-19 16:07:38 -04001436 std::string cacheDir = TWFunc::get_log_dir();
1437 if (cacheDir == CACHE_LOGS_DIR) {
1438 if (!PartitionManager.Mount_By_Path(CACHE_LOGS_DIR, false)) {
1439 LOGINFO("Unable to mount %s for wiping cache.\n", CACHE_LOGS_DIR);
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001440 }
1441 dir.push_back(cacheDir + "dalvik-cache");
1442 dir.push_back(cacheDir + "/dc");
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001443 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001444
Dees_Troy38bd7602012-09-14 13:33:53 -04001445 TWPartition* sdext = Find_Partition_By_Path("/sd-ext");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001446 if (sdext && sdext->Is_Present && sdext->Mount(false))
1447 {
1448 if (stat("/sd-ext/dalvik-cache", &st) == 0)
1449 {
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001450 dir.push_back("/sd-ext/dalvik-cache");
1451 }
1452 }
1453
bigbiff25d25b92020-06-19 16:07:38 -04001454 if (cacheDir == CACHE_LOGS_DIR) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001455 gui_msg("wiping_cache_dalvik=Wiping Dalvik Cache Directories...");
1456 } else {
1457 gui_msg("wiping_dalvik=Wiping Dalvik Directory...");
1458 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001459 for (unsigned i = 0; i < dir.size(); ++i) {
1460 if (stat(dir.at(i).c_str(), &st) == 0) {
1461 TWFunc::removeDir(dir.at(i), false);
1462 gui_msg(Msg("cleaned=Cleaned: {1}...")(dir.at(i)));
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001463 }
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001464 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001465
bigbiff25d25b92020-06-19 16:07:38 -04001466 if (cacheDir == CACHE_LOGS_DIR) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001467 gui_msg("cache_dalvik_done=-- Dalvik Cache Directories Wipe Complete!");
1468 } else {
1469 gui_msg("dalvik_done=-- Dalvik Directory Wipe Complete!");
1470 }
1471
Dees_Troy38bd7602012-09-14 13:33:53 -04001472 return true;
1473}
1474
1475int TWPartitionManager::Wipe_Rotate_Data(void) {
1476 if (!Mount_By_Path("/data", true))
1477 return false;
1478
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001479 unlink("/data/misc/akmd*");
1480 unlink("/data/misc/rild*");
Dees_Troy2673cec2013-04-02 20:22:16 +00001481 gui_print("Rotation data wiped.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001482 return true;
1483}
1484
1485int TWPartitionManager::Wipe_Battery_Stats(void) {
1486 struct stat st;
1487
1488 if (!Mount_By_Path("/data", true))
1489 return false;
1490
1491 if (0 != stat("/data/system/batterystats.bin", &st)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001492 gui_print("No Battery Stats Found. No Need To Wipe.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001493 } else {
1494 remove("/data/system/batterystats.bin");
Dees_Troy2673cec2013-04-02 20:22:16 +00001495 gui_print("Cleared battery stats.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001496 }
1497 return true;
1498}
1499
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001500int TWPartitionManager::Wipe_Android_Secure(void) {
1501 std::vector<TWPartition*>::iterator iter;
1502 int ret = false;
1503 bool found = false;
1504
1505 // Iterate through all partitions
1506 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1507 if ((*iter)->Has_Android_Secure) {
1508 ret = (*iter)->Wipe_AndSec();
1509 found = true;
1510 }
1511 }
1512 if (found) {
1513 return ret;
1514 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001515 gui_err("no_andsec=No android secure partitions found.");
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001516 }
1517 return false;
1518}
1519
Dees_Troy38bd7602012-09-14 13:33:53 -04001520int TWPartitionManager::Format_Data(void) {
1521 TWPartition* dat = Find_Partition_By_Path("/data");
1522
1523 if (dat != NULL) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001524 return dat->Wipe_Encryption();
1525 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001526 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001527 return false;
1528 }
1529 return false;
1530}
1531
1532int TWPartitionManager::Wipe_Media_From_Data(void) {
1533 TWPartition* dat = Find_Partition_By_Path("/data");
1534
1535 if (dat != NULL) {
1536 if (!dat->Has_Data_Media) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001537 LOGERR("This device does not have /data/media\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001538 return false;
1539 }
1540 if (!dat->Mount(true))
1541 return false;
1542
Ethan Yonker74db1572015-10-28 12:44:49 -05001543 gui_msg("wiping_datamedia=Wiping internal storage -- /data/media...");
Ethan Yonker726a0202014-12-16 20:01:38 -06001544 Remove_MTP_Storage(dat->MTP_Storage_ID);
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001545 TWFunc::removeDir("/data/media", false);
xiaolu9416f4f2015-06-04 08:22:23 +08001546 dat->Recreate_Media_Folder();
Ethan Yonker726a0202014-12-16 20:01:38 -06001547 Add_MTP_Storage(dat->MTP_Storage_ID);
Dees_Troy38bd7602012-09-14 13:33:53 -04001548 return true;
1549 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001550 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001551 return false;
1552 }
1553 return false;
1554}
1555
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001556int TWPartitionManager::Repair_By_Path(string Path, bool Display_Error) {
1557 std::vector<TWPartition*>::iterator iter;
1558 int ret = false;
1559 bool found = false;
1560 string Local_Path = TWFunc::Get_Root_Path(Path);
1561
1562 if (Local_Path == "/tmp" || Local_Path == "/")
1563 return true;
1564
1565 // Iterate through all partitions
1566 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1567 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1568 ret = (*iter)->Repair();
1569 found = true;
1570 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1571 (*iter)->Repair();
1572 }
1573 }
1574 if (found) {
1575 return ret;
1576 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001577 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 -05001578 } else {
1579 LOGINFO("Repair: Unable to find partition for path '%s'\n", Local_Path.c_str());
1580 }
1581 return false;
1582}
1583
Ethan Yonkera2719152015-05-28 09:44:41 -05001584int TWPartitionManager::Resize_By_Path(string Path, bool Display_Error) {
1585 std::vector<TWPartition*>::iterator iter;
1586 int ret = false;
1587 bool found = false;
1588 string Local_Path = TWFunc::Get_Root_Path(Path);
1589
1590 if (Local_Path == "/tmp" || Local_Path == "/")
1591 return true;
1592
1593 // Iterate through all partitions
1594 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1595 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1596 ret = (*iter)->Resize();
1597 found = true;
1598 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1599 (*iter)->Resize();
1600 }
1601 }
1602 if (found) {
1603 return ret;
1604 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001605 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 -05001606 } else {
1607 LOGINFO("Resize: Unable to find partition for path '%s'\n", Local_Path.c_str());
1608 }
1609 return false;
1610}
1611
Dees_Troy51a0e822012-09-05 15:24:24 -04001612void TWPartitionManager::Update_System_Details(void) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001613 std::vector<TWPartition*>::iterator iter;
Dees_Troy51127312012-09-08 13:08:49 -04001614 int data_size = 0;
Dees_Troy5bf43922012-09-07 16:07:55 -04001615
Ethan Yonker74db1572015-10-28 12:44:49 -05001616 gui_msg("update_part_details=Updating partition details...");
Dees_Troy5bf43922012-09-07 16:07:55 -04001617 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker1b190162016-12-05 15:25:19 -06001618 (*iter)->Update_Size(true);
Dees_Troy51127312012-09-08 13:08:49 -04001619 if ((*iter)->Can_Be_Mounted) {
Captain Throwback9d6feb52018-07-27 10:05:24 -04001620 if ((*iter)->Mount_Point == Get_Android_Root_Path()) {
Dees_Troy51127312012-09-08 13:08:49 -04001621 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1622 DataManager::SetValue(TW_BACKUP_SYSTEM_SIZE, backup_display_size);
Ethan Yonker76bbd3a2019-05-10 10:50:04 -05001623 TWFunc::Is_TWRP_App_In_System();
Dees_Troy51127312012-09-08 13:08:49 -04001624 } else if ((*iter)->Mount_Point == "/data" || (*iter)->Mount_Point == "/datadata") {
1625 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
1626 } else if ((*iter)->Mount_Point == "/cache") {
1627 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1628 DataManager::SetValue(TW_BACKUP_CACHE_SIZE, backup_display_size);
1629 } else if ((*iter)->Mount_Point == "/sd-ext") {
1630 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1631 DataManager::SetValue(TW_BACKUP_SDEXT_SIZE, backup_display_size);
1632 if ((*iter)->Backup_Size == 0) {
1633 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 0);
1634 DataManager::SetValue(TW_BACKUP_SDEXT_VAR, 0);
1635 } else
1636 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 1);
Dees_Troye58d5262012-09-21 12:27:57 -04001637 } else if ((*iter)->Has_Android_Secure) {
Dees_Troy8170a922012-09-18 15:40:25 -04001638 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troye58d5262012-09-21 12:27:57 -04001639 DataManager::SetValue(TW_BACKUP_ANDSEC_SIZE, backup_display_size);
Dees_Troy8170a922012-09-18 15:40:25 -04001640 if ((*iter)->Backup_Size == 0) {
1641 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 0);
1642 DataManager::SetValue(TW_BACKUP_ANDSEC_VAR, 0);
1643 } else
1644 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 1);
Dees_Troy2c50e182012-09-26 20:05:28 -04001645 } else if ((*iter)->Mount_Point == "/boot") {
1646 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1647 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1648 if ((*iter)->Backup_Size == 0) {
1649 DataManager::SetValue("tw_has_boot_partition", 0);
1650 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1651 } else
1652 DataManager::SetValue("tw_has_boot_partition", 1);
Dees_Troy51127312012-09-08 13:08:49 -04001653 }
Dees_Troyaa9cc402012-10-13 12:14:05 -04001654 } else {
1655 // Handle unmountable partitions in case we reset defaults
1656 if ((*iter)->Mount_Point == "/boot") {
1657 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1658 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1659 if ((*iter)->Backup_Size == 0) {
1660 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 0);
1661 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1662 } else
1663 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 1);
1664 } else if ((*iter)->Mount_Point == "/recovery") {
1665 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1666 DataManager::SetValue(TW_BACKUP_RECOVERY_SIZE, backup_display_size);
1667 if ((*iter)->Backup_Size == 0) {
1668 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 0);
1669 DataManager::SetValue(TW_BACKUP_RECOVERY_VAR, 0);
1670 } else
1671 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 1);
Gary Peck82599a82012-11-21 16:23:12 -08001672 } else if ((*iter)->Mount_Point == "/data") {
1673 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troyaa9cc402012-10-13 12:14:05 -04001674 }
Dees_Troy51127312012-09-08 13:08:49 -04001675 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001676 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001677 gui_msg("update_part_details_done=...done");
Dees_Troy51127312012-09-08 13:08:49 -04001678 DataManager::SetValue(TW_BACKUP_DATA_SIZE, data_size);
1679 string current_storage_path = DataManager::GetCurrentStoragePath();
1680 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
Dees_Troy8170a922012-09-18 15:40:25 -04001681 if (FreeStorage != NULL) {
1682 // Attempt to mount storage
bigbiff25d25b92020-06-19 16:07:38 -04001683 if (!FreeStorage->Mount(false)) {
1684 gui_msg(Msg(msg::kWarning, "unable_to_mount_storage=Unable to mount storage"));
1685 DataManager::SetValue(TW_STORAGE_FREE_SIZE, 0);
1686 } else {
1687 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
Dees_Troy8170a922012-09-18 15:40:25 -04001688 }
1689 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001690 LOGINFO("Unable to find storage partition '%s'.\n", current_storage_path.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -04001691 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001692 if (!Write_Fstab())
Dees_Troy2673cec2013-04-02 20:22:16 +00001693 LOGERR("Error creating fstab\n");
Dees_Troy51a0e822012-09-05 15:24:24 -04001694 return;
1695}
1696
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001697void TWPartitionManager::Post_Decrypt(const string& Block_Device) {
1698 TWPartition* dat = Find_Partition_By_Path("/data");
bigbiff7ba75002020-04-11 20:47:09 -04001699#ifdef USE_FSCRYPT
1700 dat->Set_Block_Device("/dev/block/mapper/userdata");
1701#endif
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001702 if (dat != NULL) {
1703 DataManager::SetValue(TW_IS_DECRYPTED, 1);
1704 dat->Is_Decrypted = true;
1705 if (!Block_Device.empty()) {
1706 dat->Decrypted_Block_Device = Block_Device;
1707 gui_msg(Msg("decrypt_success_dev=Data successfully decrypted, new block device: '{1}'")(Block_Device));
1708 } else {
1709 gui_msg("decrypt_success_nodev=Data successfully decrypted");
1710 }
1711 dat->Setup_File_System(false);
bigbiffadc599e2020-05-28 19:36:30 +00001712 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 -06001713
bigbiffadc599e2020-05-28 19:36:30 +00001714 // Sleep for a bit so that the device will be ready
1715 sleep(1);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001716 if (dat->Has_Data_Media && dat->Mount(false) && TWFunc::Path_Exists("/data/media/0")) {
1717 dat->Storage_Path = "/data/media/0";
1718 dat->Symlink_Path = dat->Storage_Path;
1719 DataManager::SetValue("tw_storage_path", "/data/media/0");
1720 DataManager::SetValue("tw_settings_path", "/data/media/0");
1721 dat->UnMount(false);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001722 }
1723 Update_System_Details();
nkk71ffb02bd2017-06-04 23:12:16 +03001724 Output_Partition(dat);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001725 UnMount_Main_Partitions();
1726 } else
1727 LOGERR("Unable to locate data partition.\n");
1728}
1729
bigbiffadc599e2020-05-28 19:36:30 +00001730int TWPartitionManager::Decrypt_Device(string Password) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001731#ifdef TW_INCLUDE_CRYPTO
Ethan Yonkerddb63e22017-01-19 14:01:57 -06001732 char crypto_state[PROPERTY_VALUE_MAX], crypto_blkdev[PROPERTY_VALUE_MAX];
Ethan Yonker253368a2014-11-25 15:00:52 -06001733 std::vector<TWPartition*>::iterator iter;
Dees_Troy5bf43922012-09-07 16:07:55 -04001734
Ethan Yonker253368a2014-11-25 15:00:52 -06001735 // Mount any partitions that need to be mounted for decrypt
1736 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1737 if ((*iter)->Mount_To_Decrypt) {
1738 (*iter)->Mount(true);
1739 }
a39552696ff55ce2013-01-08 16:14:56 +00001740 }
oshmouna82a7542018-04-10 17:45:55 +02001741 property_set("twrp.mount_to_decrypt", "1");
a39552696ff55ce2013-01-08 16:14:56 +00001742
Ethan Yonkera1de1492015-11-04 11:58:27 -06001743 property_get("ro.crypto.state", crypto_state, "error");
1744 if (strcmp(crypto_state, "error") == 0) {
1745 property_set("ro.crypto.state", "encrypted");
1746 // Sleep for a bit so that services can start if needed
1747 sleep(1);
1748 }
1749
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001750 if (DataManager::GetIntValue(TW_IS_FBE)) {
1751#ifdef TW_INCLUDE_FBE
1752 if (!Mount_By_Path("/data", true)) // /data has to be mounted for FBE
1753 return -1;
1754 int retry_count = 10;
1755 while (!TWFunc::Path_Exists("/data/system/users/gatekeeper.password.key") && --retry_count)
bigbiffadc599e2020-05-28 19:36:30 +00001756 usleep(2000); // A small sleep is needed after mounting /data to ensure reliable decrypt... maybe because of DE?
1757 int user_id = DataManager::GetIntValue("tw_decrypt_user_id");
1758 LOGINFO("Decrypting FBE for user %i\n", user_id);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001759 if (Decrypt_User(user_id, Password)) {
bigbiffadc599e2020-05-28 19:36:30 +00001760 Post_Decrypt("");
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001761 return 0;
1762 }
1763#else
1764 LOGERR("FBE support is not present\n");
1765#endif
1766 return -1;
1767 }
1768
Ethan Yonkerddb63e22017-01-19 14:01:57 -06001769 int pwret = -1;
1770 pid_t pid = fork();
1771 if (pid < 0) {
1772 LOGERR("fork failed\n");
1773 return -1;
1774 } else if (pid == 0) {
1775 // Child process
1776 char cPassword[255];
1777 strcpy(cPassword, Password.c_str());
1778 int ret = cryptfs_check_passwd(cPassword);
1779 exit(ret);
1780 } else {
1781 // Parent
1782 int status;
1783 if (TWFunc::Wait_For_Child_Timeout(pid, &status, "Decrypt", 30))
1784 pwret = -1;
1785 else
1786 pwret = WEXITSTATUS(status) ? -1 : 0;
1787 }
a39552696ff55ce2013-01-08 16:14:56 +00001788
nkk7171c6c502017-01-05 23:55:05 +02001789#ifdef TW_CRYPTO_USE_SYSTEM_VOLD
1790 if (pwret != 0) {
1791 pwret = vold_decrypt(Password);
nkk717d1222a2017-11-06 17:27:21 +02001792 switch (pwret) {
1793 case VD_SUCCESS:
1794 break;
1795 case VD_ERR_MISSING_VDC:
1796 gui_msg(Msg(msg::kError, "decrypt_data_vold_os_missing=Missing files needed for vold decrypt: {1}")("/system/bin/vdc"));
1797 break;
1798 case VD_ERR_MISSING_VOLD:
1799 gui_msg(Msg(msg::kError, "decrypt_data_vold_os_missing=Missing files needed for vold decrypt: {1}")("/system/bin/vold"));
1800 break;
1801 }
nkk7171c6c502017-01-05 23:55:05 +02001802 }
1803#endif // TW_CRYPTO_USE_SYSTEM_VOLD
1804
Ethan Yonker253368a2014-11-25 15:00:52 -06001805 // Unmount any partitions that were needed for decrypt
1806 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1807 if ((*iter)->Mount_To_Decrypt) {
1808 (*iter)->UnMount(false);
1809 }
1810 }
oshmouna82a7542018-04-10 17:45:55 +02001811 property_set("twrp.mount_to_decrypt", "0");
Ethan Yonker253368a2014-11-25 15:00:52 -06001812
a39552696ff55ce2013-01-08 16:14:56 +00001813 if (pwret != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001814 gui_err("fail_decrypt=Failed to decrypt data.");
Dees_Troy5bf43922012-09-07 16:07:55 -04001815 return -1;
1816 }
a39552696ff55ce2013-01-08 16:14:56 +00001817
Dees_Troy5bf43922012-09-07 16:07:55 -04001818 property_get("ro.crypto.fs_crypto_blkdev", crypto_blkdev, "error");
1819 if (strcmp(crypto_blkdev, "error") == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001820 LOGERR("Error retrieving decrypted data block device.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04001821 } else {
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001822 Post_Decrypt(crypto_blkdev);
Dees_Troy5bf43922012-09-07 16:07:55 -04001823 }
1824 return 0;
1825#else
Ethan Yonker74db1572015-10-28 12:44:49 -05001826 gui_err("no_crypto_support=No crypto support was compiled into this build.");
Dees_Troy5bf43922012-09-07 16:07:55 -04001827 return -1;
1828#endif
Dees_Troy51a0e822012-09-05 15:24:24 -04001829 return 1;
Dees_Troy51127312012-09-08 13:08:49 -04001830}
1831
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001832int TWPartitionManager::Fix_Contexts(void) {
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001833 std::vector<TWPartition*>::iterator iter;
1834 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1835 if ((*iter)->Has_Data_Media) {
1836 if ((*iter)->Mount(true)) {
1837 if (fixContexts::fixDataMediaContexts((*iter)->Mount_Point) != 0)
1838 return -1;
1839 }
1840 }
1841 }
Dees_Troy1a650e62012-10-19 20:54:32 -04001842 UnMount_Main_Partitions();
Ethan Yonker74db1572015-10-28 12:44:49 -05001843 gui_msg("done=Done.");
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001844 return 0;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04001845}
1846
Ethan Yonker66a19492015-12-10 10:19:45 -06001847TWPartition* TWPartitionManager::Find_Next_Storage(string Path, bool Exclude_Data_Media) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001848 std::vector<TWPartition*>::iterator iter = Partitions.begin();
1849
1850 if (!Path.empty()) {
1851 string Search_Path = TWFunc::Get_Root_Path(Path);
1852 for (; iter != Partitions.end(); iter++) {
Matt Mower9a561dd2016-02-22 21:25:51 -06001853 if ((*iter)->Mount_Point == Search_Path) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001854 iter++;
1855 break;
1856 }
1857 }
1858 }
1859
1860 for (; iter != Partitions.end(); iter++) {
Ethan Yonker66a19492015-12-10 10:19:45 -06001861 if (Exclude_Data_Media && (*iter)->Has_Data_Media) {
1862 // do nothing, do not return this type of partition
1863 } else if ((*iter)->Is_Storage && (*iter)->Is_Present) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001864 return (*iter);
1865 }
1866 }
1867
1868 return NULL;
1869}
1870
Dees_Troyd21618c2012-10-14 18:48:49 -04001871int TWPartitionManager::Open_Lun_File(string Partition_Path, string Lun_File) {
Dees_Troyd21618c2012-10-14 18:48:49 -04001872 TWPartition* Part = Find_Partition_By_Path(Partition_Path);
1873
1874 if (Part == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001875 LOGINFO("Unable to locate '%s' for USB storage mode.", Partition_Path.c_str());
1876 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 -04001877 return false;
1878 }
Ethan Yonker47360be2014-04-01 10:34:34 -05001879 LOGINFO("USB mount '%s', '%s' > '%s'\n", Partition_Path.c_str(), Part->Actual_Block_Device.c_str(), Lun_File.c_str());
1880 if (!Part->UnMount(true) || !Part->Is_Present)
Dees_Troyd21618c2012-10-14 18:48:49 -04001881 return false;
1882
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06001883 if (TWFunc::write_to_file(Lun_File, Part->Actual_Block_Device)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001884 LOGERR("Unable to write to ums lunfile '%s': (%s)\n", Lun_File.c_str(), strerror(errno));
Dees_Troyd21618c2012-10-14 18:48:49 -04001885 return false;
1886 }
Dees_Troyd21618c2012-10-14 18:48:49 -04001887 return true;
1888}
1889
Dees_Troy8170a922012-09-18 15:40:25 -04001890int TWPartitionManager::usb_storage_enable(void) {
Dees_Troy8170a922012-09-18 15:40:25 -04001891 char lun_file[255];
Dees_Troyd21618c2012-10-14 18:48:49 -04001892 bool has_multiple_lun = false;
Dees_Troy8170a922012-09-18 15:40:25 -04001893
Ethan Yonker47360be2014-04-01 10:34:34 -05001894 string Lun_File_str = CUSTOM_LUN_FILE;
1895 size_t found = Lun_File_str.find("%");
1896 if (found != string::npos) {
1897 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
1898 if (TWFunc::Path_Exists(lun_file))
1899 has_multiple_lun = true;
1900 }
Ethan Yonker726a0202014-12-16 20:01:38 -06001901 mtp_was_enabled = TWFunc::Toggle_MTP(false); // Must disable MTP for USB Storage
Ethan Yonker47360be2014-04-01 10:34:34 -05001902 if (!has_multiple_lun) {
1903 LOGINFO("Device doesn't have multiple lun files, mount current storage\n");
1904 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
1905 if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) == "/data") {
Ethan Yonker66a19492015-12-10 10:19:45 -06001906 TWPartition* Mount = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05001907 if (Mount) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001908 if (!Open_Lun_File(Mount->Mount_Point, lun_file)) {
1909 goto error_handle;
1910 }
Ethan Yonker47360be2014-04-01 10:34:34 -05001911 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001912 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001913 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05001914 }
1915 } else if (!Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file)) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001916 goto error_handle;
Dees_Troy8170a922012-09-18 15:40:25 -04001917 }
Dees_Troy8170a922012-09-18 15:40:25 -04001918 } else {
Ethan Yonker47360be2014-04-01 10:34:34 -05001919 LOGINFO("Device has multiple lun files\n");
1920 TWPartition* Mount1;
1921 TWPartition* Mount2;
Dees_Troy8170a922012-09-18 15:40:25 -04001922 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
Ethan Yonker66a19492015-12-10 10:19:45 -06001923 Mount1 = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05001924 if (Mount1) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001925 if (!Open_Lun_File(Mount1->Mount_Point, lun_file)) {
1926 goto error_handle;
1927 }
Matt Moweree717062014-04-26 01:46:46 -05001928 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
Ethan Yonker66a19492015-12-10 10:19:45 -06001929 Mount2 = Find_Next_Storage(Mount1->Mount_Point, true);
Matt Mower1fdcdb72016-01-25 20:53:47 -06001930 if (Mount2 && Mount2->Mount_Point != Mount1->Mount_Point) {
Matt Moweree717062014-04-26 01:46:46 -05001931 Open_Lun_File(Mount2->Mount_Point, lun_file);
Ethan Yonker47360be2014-04-01 10:34:34 -05001932 }
1933 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001934 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001935 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05001936 }
Dees_Troy8170a922012-09-18 15:40:25 -04001937 }
Matt Mowerb6ef4782014-11-06 02:24:36 -06001938 property_set("sys.storage.ums_enabled", "1");
HandyMenny37d42992014-10-26 22:15:59 +01001939 property_set("sys.usb.config", "mass_storage,adb");
Dees_Troy8170a922012-09-18 15:40:25 -04001940 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001941error_handle:
1942 if (mtp_was_enabled)
1943 if (!Enable_MTP())
1944 Disable_MTP();
1945 return false;
Dees_Troy8170a922012-09-18 15:40:25 -04001946}
1947
1948int TWPartitionManager::usb_storage_disable(void) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001949 int index, ret;
1950 char lun_file[255], ch[2] = {0, 0};
1951 string str = ch;
Dees_Troy8170a922012-09-18 15:40:25 -04001952
1953 for (index=0; index<2; index++) {
1954 sprintf(lun_file, CUSTOM_LUN_FILE, index);
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06001955 ret = TWFunc::write_to_file(lun_file, str);
Dees_Troy2673cec2013-04-02 20:22:16 +00001956 if (ret < 0) {
1957 break;
Dees_Troy8170a922012-09-18 15:40:25 -04001958 }
Dees_Troy8170a922012-09-18 15:40:25 -04001959 }
Dees_Troye58d5262012-09-21 12:27:57 -04001960 Mount_All_Storage();
1961 Update_System_Details();
Dees_Troycfd73ef2012-10-12 16:52:00 -04001962 UnMount_Main_Partitions();
Matt Mowerd9cb9062013-12-14 20:34:45 -06001963 property_set("sys.storage.ums_enabled", "0");
HandyMenny37d42992014-10-26 22:15:59 +01001964 property_set("sys.usb.config", "adb");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001965 if (mtp_was_enabled)
1966 if (!Enable_MTP())
1967 Disable_MTP();
Dees_Troy2673cec2013-04-02 20:22:16 +00001968 if (ret < 0 && index == 0) {
1969 LOGERR("Unable to write to ums lunfile '%s'.", lun_file);
1970 return false;
1971 } else {
1972 return true;
1973 }
Dees_Troy8170a922012-09-18 15:40:25 -04001974 return true;
Dees_Troy812660f2012-09-20 09:55:17 -04001975}
1976
1977void TWPartitionManager::Mount_All_Storage(void) {
1978 std::vector<TWPartition*>::iterator iter;
1979
1980 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1981 if ((*iter)->Is_Storage)
1982 (*iter)->Mount(false);
1983 }
Dees_Troy2c50e182012-09-26 20:05:28 -04001984}
Dees_Troy9350b8d2012-09-27 12:38:38 -04001985
Dees_Troyd0384ef2012-10-12 12:15:42 -04001986void TWPartitionManager::UnMount_Main_Partitions(void) {
1987 // Unmounts system and data if data is not data/media
1988 // Also unmounts boot if boot is mountable
Dees_Troy2673cec2013-04-02 20:22:16 +00001989 LOGINFO("Unmounting main partitions...\n");
Dees_Troyd0384ef2012-10-12 12:15:42 -04001990
1991 TWPartition* Boot_Partition = Find_Partition_By_Path("/boot");
1992
Captain Throwback9d6feb52018-07-27 10:05:24 -04001993 UnMount_By_Path(Get_Android_Root_Path(), true);
Ethan Yonker6277c792014-09-15 14:54:30 -05001994 if (!datamedia)
1995 UnMount_By_Path("/data", true);
1996
Dees_Troyd0384ef2012-10-12 12:15:42 -04001997 if (Boot_Partition != NULL && Boot_Partition->Can_Be_Mounted)
1998 Boot_Partition->UnMount(true);
1999}
2000
Dees_Troy9350b8d2012-09-27 12:38:38 -04002001int TWPartitionManager::Partition_SDCard(void) {
Matt Mower23d8aae2017-01-06 14:30:33 -06002002 char temp[255];
Ethan Yonker483e9f42016-01-11 22:21:18 -06002003 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 -04002004 int ext, swap, total_size = 0, fat_size;
Dees_Troy9350b8d2012-09-27 12:38:38 -04002005
Ethan Yonker74db1572015-10-28 12:44:49 -05002006 gui_msg("start_partition_sd=Partitioning SD Card...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06002007
2008 // Locate and validate device to partition
2009 TWPartition* SDCard = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
2010
Ethan Yonker66a19492015-12-10 10:19:45 -06002011 if (SDCard->Is_Adopted_Storage)
2012 SDCard->Revert_Adopted();
2013
Ethan Yonker1eff6cd2014-09-15 13:30:42 -05002014 if (SDCard == NULL || !SDCard->Removable || SDCard->Has_Data_Media) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002015 gui_err("partition_sd_locate=Unable to locate device to partition.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002016 return false;
2017 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002018
2019 // Unmount everything
Dees_Troy9350b8d2012-09-27 12:38:38 -04002020 if (!SDCard->UnMount(true))
2021 return false;
2022 TWPartition* SDext = Find_Partition_By_Path("/sd-ext");
2023 if (SDext != NULL) {
2024 if (!SDext->UnMount(true))
2025 return false;
2026 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002027 char* swappath = getenv("SWAPPATH");
2028 if (swappath != NULL) {
2029 LOGINFO("Unmounting swap at '%s'\n", swappath);
2030 umount(swappath);
2031 }
Vojtech Bocek05534202013-09-11 08:11:56 +02002032
Ethan Yonker483e9f42016-01-11 22:21:18 -06002033 // Determine block device
2034 if (SDCard->Alternate_Block_Device.empty()) {
2035 SDCard->Find_Actual_Block_Device();
2036 Device = SDCard->Actual_Block_Device;
2037 // Just use the root block device
2038 Device.resize(strlen("/dev/block/mmcblkX"));
2039 } else {
2040 Device = SDCard->Alternate_Block_Device;
2041 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04002042
2043 // Find the size of the block device:
Ethan Yonker483e9f42016-01-11 22:21:18 -06002044 total_size = (int)(TWFunc::IOCTL_Get_Block_Size(Device.c_str()) / (1048576));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002045
2046 DataManager::GetValue("tw_sdext_size", ext);
2047 DataManager::GetValue("tw_swap_size", swap);
2048 DataManager::GetValue("tw_sdpart_file_system", ext_format);
2049 fat_size = total_size - ext - swap;
Ethan Yonker483e9f42016-01-11 22:21:18 -06002050 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);
2051
2052 // Determine partition sizes
2053 if (swap == 0 && ext == 0) {
2054 fat_str = "-0";
2055 } else {
2056 memset(temp, 0, sizeof(temp));
2057 sprintf(temp, "%i", fat_size);
2058 fat_str = temp;
2059 fat_str += "MB";
2060 }
2061 if (swap == 0) {
2062 ext_str = "-0";
2063 } else {
2064 memset(temp, 0, sizeof(temp));
2065 sprintf(temp, "%i", ext);
2066 ext_str = "+";
2067 ext_str += temp;
2068 ext_str += "MB";
2069 }
2070
Dees_Troy9350b8d2012-09-27 12:38:38 -04002071 if (ext + swap > total_size) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002072 gui_err("ext_swap_size=EXT + Swap size is larger than sdcard size.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002073 return false;
2074 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002075
Ethan Yonker74db1572015-10-28 12:44:49 -05002076 gui_msg("remove_part_table=Removing partition table...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06002077 Command = "sgdisk --zap-all " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002078 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002079 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002080 gui_err("unable_rm_part=Unable to remove partition table.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002081 Update_System_Details();
2082 return false;
2083 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002084 gui_msg(Msg("create_part=Creating {1} partition...")("FAT32"));
Captain Throwback9643a802016-05-27 11:44:51 -04002085 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 +00002086 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002087 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002088 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("FAT32"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002089 return false;
2090 }
2091 if (ext > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002092 gui_msg(Msg("create_part=Creating {1} partition...")("EXT"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06002093 Command = "sgdisk --new=0:0:" + ext_str + " --change-name=0:\"Linux filesystem\" " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002094 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002095 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002096 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("EXT"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002097 Update_System_Details();
2098 return false;
2099 }
2100 }
2101 if (swap > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002102 gui_msg(Msg("create_part=Creating {1} partition...")("swap"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06002103 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 +00002104 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002105 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002106 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("swap"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002107 Update_System_Details();
2108 return false;
2109 }
2110 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002111
2112 // Convert GPT to MBR
2113 Command = "sgdisk --gpttombr " + Device;
2114 if (TWFunc::Exec_Cmd(Command) != 0)
2115 LOGINFO("Failed to covert partition GPT to MBR\n");
2116
2117 // Tell the kernel to rescan the partition table
2118 int fd = open(Device.c_str(), O_RDONLY);
2119 ioctl(fd, BLKRRPART, 0);
2120 close(fd);
2121
2122 string format_device = Device;
2123 if (Device.substr(0, 17) == "/dev/block/mmcblk")
2124 format_device += "p";
2125
2126 // Format new partitions to proper file system
2127 if (fat_size > 0) {
2128 Command = "mkfs.fat " + format_device + "1";
2129 TWFunc::Exec_Cmd(Command);
2130 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04002131 if (ext > 0) {
2132 if (SDext == NULL) {
Ethan Yonker483e9f42016-01-11 22:21:18 -06002133 Command = "mke2fs -t " + ext_format + " -m 0 " + format_device + "2";
2134 gui_msg(Msg("format_sdext_as=Formatting sd-ext as {1}...")(ext_format));
2135 LOGINFO("Formatting sd-ext after partitioning, command: '%s'\n", Command.c_str());
2136 TWFunc::Exec_Cmd(Command);
2137 } else {
2138 SDext->Wipe(ext_format);
Dees_Troy9350b8d2012-09-27 12:38:38 -04002139 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002140 }
2141 if (swap > 0) {
2142 Command = "mkswap " + format_device;
2143 if (ext > 0)
2144 Command += "3";
2145 else
2146 Command += "2";
Vojtech Bocek05534202013-09-11 08:11:56 +02002147 TWFunc::Exec_Cmd(Command);
Dees_Troy9350b8d2012-09-27 12:38:38 -04002148 }
2149
Ethan Yonker483e9f42016-01-11 22:21:18 -06002150 // recreate TWRP folder and rewrite settings - these will be gone after sdcard is partitioned
2151 if (SDCard->Mount(true)) {
2152 string TWRP_Folder = SDCard->Mount_Point + "/TWRP";
2153 mkdir(TWRP_Folder.c_str(), 0777);
2154 DataManager::Flush();
2155 }
2156
Dees_Troy9350b8d2012-09-27 12:38:38 -04002157 Update_System_Details();
Ethan Yonker74db1572015-10-28 12:44:49 -05002158 gui_msg("part_complete=Partitioning complete.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002159 return true;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04002160}
Dees_Troya13d74f2013-03-24 08:54:55 -05002161
2162void TWPartitionManager::Get_Partition_List(string ListType, std::vector<PartitionList> *Partition_List) {
2163 std::vector<TWPartition*>::iterator iter;
2164 if (ListType == "mount") {
2165 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002166 if ((*iter)->Can_Be_Mounted) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002167 struct PartitionList part;
2168 part.Display_Name = (*iter)->Display_Name;
2169 part.Mount_Point = (*iter)->Mount_Point;
2170 part.selected = (*iter)->Is_Mounted();
2171 Partition_List->push_back(part);
2172 }
2173 }
2174 } else if (ListType == "storage") {
2175 char free_space[255];
2176 string Current_Storage = DataManager::GetCurrentStoragePath();
2177 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2178 if ((*iter)->Is_Storage) {
2179 struct PartitionList part;
2180 sprintf(free_space, "%llu", (*iter)->Free / 1024 / 1024);
2181 part.Display_Name = (*iter)->Storage_Name + " (";
2182 part.Display_Name += free_space;
2183 part.Display_Name += "MB)";
2184 part.Mount_Point = (*iter)->Storage_Path;
2185 if ((*iter)->Storage_Path == Current_Storage)
2186 part.selected = 1;
2187 else
2188 part.selected = 0;
2189 Partition_List->push_back(part);
2190 }
2191 }
2192 } else if (ListType == "backup") {
2193 char backup_size[255];
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002194 unsigned long long Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -05002195 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002196 if ((*iter)->Can_Be_Backed_Up && !(*iter)->Is_SubPartition && (*iter)->Is_Present) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002197 struct PartitionList part;
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002198 Backup_Size = (*iter)->Backup_Size;
2199 if ((*iter)->Has_SubPartition) {
2200 std::vector<TWPartition*>::iterator subpart;
2201
2202 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
2203 if ((*subpart)->Is_SubPartition && (*subpart)->Can_Be_Backed_Up && (*subpart)->Is_Present && (*subpart)->SubPartition_Of == (*iter)->Mount_Point)
2204 Backup_Size += (*subpart)->Backup_Size;
2205 }
2206 }
2207 sprintf(backup_size, "%llu", Backup_Size / 1024 / 1024);
Dees_Troya13d74f2013-03-24 08:54:55 -05002208 part.Display_Name = (*iter)->Backup_Display_Name + " (";
2209 part.Display_Name += backup_size;
2210 part.Display_Name += "MB)";
2211 part.Mount_Point = (*iter)->Backup_Path;
2212 part.selected = 0;
2213 Partition_List->push_back(part);
2214 }
2215 }
2216 } else if (ListType == "restore") {
2217 string Restore_List, restore_path;
2218 TWPartition* restore_part = NULL;
2219
2220 DataManager::GetValue("tw_restore_list", Restore_List);
2221 if (!Restore_List.empty()) {
2222 size_t start_pos = 0, end_pos = Restore_List.find(";", start_pos);
2223 while (end_pos != string::npos && start_pos < Restore_List.size()) {
2224 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04002225 struct PartitionList part;
2226 if (restore_path.compare("ADB_Backup") == 0) {
2227 part.Display_Name = "ADB Backup";
2228 part.Mount_Point = "ADB Backup";
2229 part.selected = 1;
2230 Partition_List->push_back(part);
2231 break;
2232 }
Dees_Troy59df9262013-06-19 14:53:57 -05002233 if ((restore_part = Find_Partition_By_Path(restore_path)) != NULL) {
Dees Troy4159aed2014-02-28 17:24:43 +00002234 if ((restore_part->Backup_Name == "recovery" && !restore_part->Can_Be_Backed_Up) || restore_part->Is_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002235 // Don't allow restore of recovery (causes problems on some devices)
Dees_Troy59df9262013-06-19 14:53:57 -05002236 // Don't add subpartitions to the list of items
Dees_Troya13d74f2013-03-24 08:54:55 -05002237 } else {
Dees_Troya13d74f2013-03-24 08:54:55 -05002238 part.Display_Name = restore_part->Backup_Display_Name;
2239 part.Mount_Point = restore_part->Backup_Path;
2240 part.selected = 1;
2241 Partition_List->push_back(part);
2242 }
2243 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002244 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05002245 }
2246 start_pos = end_pos + 1;
2247 end_pos = Restore_List.find(";", start_pos);
2248 }
2249 }
2250 } else if (ListType == "wipe") {
2251 struct PartitionList dalvik;
Ethan Yonker74db1572015-10-28 12:44:49 -05002252 dalvik.Display_Name = gui_parse_text("{@dalvik}");
Dees_Troya13d74f2013-03-24 08:54:55 -05002253 dalvik.Mount_Point = "DALVIK";
2254 dalvik.selected = 0;
2255 Partition_List->push_back(dalvik);
2256 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2257 if ((*iter)->Wipe_Available_in_GUI && !(*iter)->Is_SubPartition) {
2258 struct PartitionList part;
2259 part.Display_Name = (*iter)->Display_Name;
2260 part.Mount_Point = (*iter)->Mount_Point;
2261 part.selected = 0;
2262 Partition_List->push_back(part);
2263 }
2264 if ((*iter)->Has_Android_Secure) {
2265 struct PartitionList part;
2266 part.Display_Name = (*iter)->Backup_Display_Name;
2267 part.Mount_Point = (*iter)->Backup_Path;
2268 part.selected = 0;
2269 Partition_List->push_back(part);
2270 }
Dees_Troy74fb2e92013-04-15 14:35:47 +00002271 if ((*iter)->Has_Data_Media) {
2272 struct PartitionList datamedia;
2273 datamedia.Display_Name = (*iter)->Storage_Name;
2274 datamedia.Mount_Point = "INTERNAL";
2275 datamedia.selected = 0;
2276 Partition_List->push_back(datamedia);
2277 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002278 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002279 } else if (ListType == "flashimg") {
2280 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2281 if ((*iter)->Can_Flash_Img && (*iter)->Is_Present) {
2282 struct PartitionList part;
2283 part.Display_Name = (*iter)->Backup_Display_Name;
2284 part.Mount_Point = (*iter)->Backup_Path;
2285 part.selected = 0;
2286 Partition_List->push_back(part);
2287 }
2288 }
Ethan Yonker53796e72019-01-11 22:49:52 -06002289 if (DataManager::GetIntValue("tw_has_repack_tools") != 0 && DataManager::GetIntValue("tw_has_boot_slots") != 0) {
2290 TWPartition* boot = Find_Partition_By_Path("/boot");
2291 if (boot) {
2292 // Allow flashing kernels and ramdisks
2293 struct PartitionList repack_ramdisk;
2294 repack_ramdisk.Display_Name = gui_lookup("install_twrp_ramdisk", "Install Recovery Ramdisk");
2295 repack_ramdisk.Mount_Point = "/repack_ramdisk";
2296 repack_ramdisk.selected = 0;
2297 Partition_List->push_back(repack_ramdisk);
2298 /*struct PartitionList repack_kernel; For now let's leave repacking kernels under advanced only
2299 repack_kernel.Display_Name = gui_lookup("install_kernel", "Install Kernel");
2300 repack_kernel.Mount_Point = "/repack_kernel";
2301 repack_kernel.selected = 0;
2302 Partition_List->push_back(repack_kernel);*/
2303 }
2304 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002305 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00002306 LOGERR("Unknown list type '%s' requested for TWPartitionManager::Get_Partition_List\n", ListType.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -05002307 }
2308}
2309
2310int TWPartitionManager::Fstab_Processed(void) {
2311 return Partitions.size();
2312}
Dees_Troyd93bda52013-07-03 19:55:19 +00002313
2314void TWPartitionManager::Output_Storage_Fstab(void) {
2315 std::vector<TWPartition*>::iterator iter;
2316 char storage_partition[255];
bigbiff bigbiffe4bdb152019-03-23 18:33:17 -04002317 std::string Temp;
bigbiff25d25b92020-06-19 16:07:38 -04002318 std::string cacheDir = TWFunc::get_log_dir();
bigbiff bigbiffe4bdb152019-03-23 18:33:17 -04002319
2320 if (cacheDir.empty()) {
2321 LOGINFO("Unable to find cache directory\n");
2322 return;
2323 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002324
bigbiff25d25b92020-06-19 16:07:38 -04002325 std::string storageFstab = TWFunc::get_log_dir() + "recovery/storage.fstab";
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002326 FILE *fp = fopen(storageFstab.c_str(), "w");
Dees_Troyd93bda52013-07-03 19:55:19 +00002327
2328 if (fp == NULL) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002329 gui_msg(Msg(msg::kError, "unable_to_open=Unable to open '{1}'.")(storageFstab));
Dees_Troyd93bda52013-07-03 19:55:19 +00002330 return;
2331 }
2332
2333 // Iterate through all partitions
2334 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2335 if ((*iter)->Is_Storage) {
2336 Temp = (*iter)->Storage_Path + ";" + (*iter)->Storage_Name + ";\n";
2337 strcpy(storage_partition, Temp.c_str());
2338 fwrite(storage_partition, sizeof(storage_partition[0]), strlen(storage_partition) / sizeof(storage_partition[0]), fp);
2339 }
2340 }
2341 fclose(fp);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02002342}
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002343
2344TWPartition *TWPartitionManager::Get_Default_Storage_Partition()
2345{
2346 TWPartition *res = NULL;
2347 for (std::vector<TWPartition*>::iterator iter = Partitions.begin(); iter != Partitions.end(); ++iter) {
Matt Mowera8a89d12016-12-30 18:10:37 -06002348 if (!(*iter)->Is_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002349 continue;
2350
Matt Mowera8a89d12016-12-30 18:10:37 -06002351 if ((*iter)->Is_Settings_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002352 return *iter;
2353
Matt Mowera8a89d12016-12-30 18:10:37 -06002354 if (!res)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002355 res = *iter;
2356 }
2357 return res;
2358}
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002359
2360bool TWPartitionManager::Enable_MTP(void) {
2361#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002362 if (mtppid) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002363 gui_err("mtp_already_enabled=MTP already enabled");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002364 return true;
2365 }
Ethan Yonker726a0202014-12-16 20:01:38 -06002366
2367 int mtppipe[2];
2368
2369 if (pipe(mtppipe) < 0) {
2370 LOGERR("Error creating MTP pipe\n");
2371 return false;
2372 }
2373
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002374 char old_value[PROPERTY_VALUE_MAX];
Matt Mowere07f0102017-02-23 17:40:00 -06002375 property_get("sys.usb.config", old_value, "");
2376 if (strcmp(old_value, "mtp,adb") != 0) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002377 char vendor[PROPERTY_VALUE_MAX];
2378 char product[PROPERTY_VALUE_MAX];
2379 property_set("sys.usb.config", "none");
2380 property_get("usb.vendor", vendor, "18D1");
2381 property_get("usb.product.mtpadb", product, "4EE2");
2382 string vendorstr = vendor;
2383 string productstr = product;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002384 TWFunc::write_to_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2385 TWFunc::write_to_file("/sys/class/android_usb/android0/idProduct", productstr);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002386 property_set("sys.usb.config", "mtp,adb");
2387 }
Matt Mower653a1702017-02-16 10:38:52 -06002388 /* To enable MTP debug, use the twrp command line feature:
Ethan Yonker6d154c42014-09-03 14:19:43 -05002389 * twrp set tw_mtp_debug 1
2390 */
2391 twrpMtp *mtp = new twrpMtp(DataManager::GetIntValue("tw_mtp_debug"));
Ethan Yonker1b039202015-01-30 10:08:48 -06002392 mtppid = mtp->forkserver(mtppipe);
2393 if (mtppid) {
2394 close(mtppipe[0]); // Host closes read side
2395 mtp_write_fd = mtppipe[1];
2396 DataManager::SetValue("tw_mtp_enabled", 1);
2397 Add_All_MTP_Storage();
2398 return true;
Ethan Yonker726a0202014-12-16 20:01:38 -06002399 } else {
2400 close(mtppipe[0]);
2401 close(mtppipe[1]);
Ethan Yonker74db1572015-10-28 12:44:49 -05002402 gui_err("mtp_fail=Failed to enable MTP");
Ethan Yonker1b039202015-01-30 10:08:48 -06002403 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002404 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002405#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002406 gui_err("no_mtp=MTP support not included");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002407#endif
2408 DataManager::SetValue("tw_mtp_enabled", 0);
2409 return false;
2410}
2411
Ethan Yonker1b039202015-01-30 10:08:48 -06002412void TWPartitionManager::Add_All_MTP_Storage(void) {
2413#ifdef TW_HAS_MTP
2414 std::vector<TWPartition*>::iterator iter;
2415
2416 if (!mtppid)
2417 return; // MTP is not enabled
2418
2419 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2420 if ((*iter)->Is_Storage && (*iter)->Is_Present && (*iter)->Mount(false))
2421 Add_Remove_MTP_Storage((*iter), MTP_MESSAGE_ADD_STORAGE);
2422 }
2423#else
2424 return;
2425#endif
2426}
2427
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002428bool TWPartitionManager::Disable_MTP(void) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002429 char old_value[PROPERTY_VALUE_MAX];
bigbiff bigbiffaf32bb92018-12-18 18:39:53 -05002430 property_set("sys.usb.ffs.mtp.ready", "0");
Matt Mowere07f0102017-02-23 17:40:00 -06002431 property_get("sys.usb.config", old_value, "");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002432 if (strcmp(old_value, "adb") != 0) {
2433 char vendor[PROPERTY_VALUE_MAX];
2434 char product[PROPERTY_VALUE_MAX];
2435 property_set("sys.usb.config", "none");
2436 property_get("usb.vendor", vendor, "18D1");
Matt Mowere07f0102017-02-23 17:40:00 -06002437 property_get("usb.product.adb", product, "D001");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002438 string vendorstr = vendor;
2439 string productstr = product;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002440 TWFunc::write_to_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2441 TWFunc::write_to_file("/sys/class/android_usb/android0/idProduct", productstr);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002442 usleep(2000);
2443 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002444#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002445 if (mtppid) {
Ethan Yonker8613dc02014-09-11 09:28:20 -05002446 LOGINFO("Disabling MTP\n");
2447 int status;
2448 kill(mtppid, SIGKILL);
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002449 mtppid = 0;
Ethan Yonker8613dc02014-09-11 09:28:20 -05002450 // We don't care about the exit value, but this prevents a zombie process
2451 waitpid(mtppid, &status, 0);
Ethan Yonker726a0202014-12-16 20:01:38 -06002452 close(mtp_write_fd);
2453 mtp_write_fd = -1;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002454 }
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002455#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002456 property_set("sys.usb.config", "adb");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002457#ifdef TW_HAS_MTP
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002458 DataManager::SetValue("tw_mtp_enabled", 0);
2459 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002460#endif
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002461 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002462}
Ethan Yonker726a0202014-12-16 20:01:38 -06002463
2464TWPartition* TWPartitionManager::Find_Partition_By_MTP_Storage_ID(unsigned int Storage_ID) {
2465 std::vector<TWPartition*>::iterator iter;
2466
2467 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2468 if ((*iter)->MTP_Storage_ID == Storage_ID)
2469 return (*iter);
2470 }
2471 return NULL;
2472}
2473
2474bool TWPartitionManager::Add_Remove_MTP_Storage(TWPartition* Part, int message_type) {
2475#ifdef TW_HAS_MTP
2476 struct mtpmsg mtp_message;
2477
2478 if (!mtppid)
2479 return false; // MTP is disabled
2480
2481 if (mtp_write_fd < 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002482 LOGINFO("MTP: mtp_write_fd is not set\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002483 return false;
2484 }
2485
2486 if (Part) {
Ethan Yonker4bfabab2014-12-29 09:15:37 -06002487 if (Part->MTP_Storage_ID == 0)
2488 return false;
Ethan Yonker726a0202014-12-16 20:01:38 -06002489 if (message_type == MTP_MESSAGE_REMOVE_STORAGE) {
2490 mtp_message.message_type = MTP_MESSAGE_REMOVE_STORAGE; // Remove
2491 LOGINFO("sending message to remove %i\n", Part->MTP_Storage_ID);
2492 mtp_message.storage_id = Part->MTP_Storage_ID;
2493 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002494 LOGINFO("error sending message to remove storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002495 return false;
2496 } else {
2497 LOGINFO("Message sent, remove storage ID: %i\n", Part->MTP_Storage_ID);
2498 return true;
2499 }
2500 } else if (message_type == MTP_MESSAGE_ADD_STORAGE && Part->Is_Mounted()) {
2501 mtp_message.message_type = MTP_MESSAGE_ADD_STORAGE; // Add
2502 mtp_message.storage_id = Part->MTP_Storage_ID;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002503 if (Part->Storage_Path.size() >= sizeof(mtp_message.path)) {
2504 LOGERR("Storage path '%s' too large for mtpmsg\n", Part->Storage_Path.c_str());
2505 return false;
2506 }
2507 strcpy(mtp_message.path, Part->Storage_Path.c_str());
2508 if (Part->Storage_Name.size() >= sizeof(mtp_message.display)) {
2509 LOGERR("Storage name '%s' too large for mtpmsg\n", Part->Storage_Name.c_str());
2510 return false;
2511 }
2512 strcpy(mtp_message.display, Part->Storage_Name.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002513 mtp_message.maxFileSize = Part->Get_Max_FileSize();
Ethan Yonker1b039202015-01-30 10:08:48 -06002514 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 -06002515 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002516 LOGINFO("error sending message to add storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002517 return false;
2518 } else {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002519 LOGINFO("Message sent, add storage ID: %i '%s'\n", Part->MTP_Storage_ID, mtp_message.path);
Ethan Yonker726a0202014-12-16 20:01:38 -06002520 return true;
2521 }
2522 } else {
2523 LOGERR("Unknown MTP message type: %i\n", message_type);
2524 }
2525 } else {
2526 // This hopefully never happens as the error handling should
2527 // occur in the calling function.
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002528 LOGINFO("TWPartitionManager::Add_Remove_MTP_Storage NULL partition given\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002529 }
2530 return true;
2531#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002532 gui_err("no_mtp=MTP support not included");
Ethan Yonker726a0202014-12-16 20:01:38 -06002533 DataManager::SetValue("tw_mtp_enabled", 0);
2534 return false;
2535#endif
2536}
2537
2538bool TWPartitionManager::Add_MTP_Storage(string Mount_Point) {
2539#ifdef TW_HAS_MTP
2540 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2541 if (Part) {
2542 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2543 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002544 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002545 }
2546#endif
2547 return false;
2548}
2549
2550bool TWPartitionManager::Add_MTP_Storage(unsigned int Storage_ID) {
2551#ifdef TW_HAS_MTP
2552 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2553 if (Part) {
2554 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2555 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002556 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002557 }
2558#endif
2559 return false;
2560}
2561
2562bool TWPartitionManager::Remove_MTP_Storage(string Mount_Point) {
2563#ifdef TW_HAS_MTP
2564 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2565 if (Part) {
2566 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2567 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002568 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002569 }
2570#endif
2571 return false;
2572}
2573
2574bool TWPartitionManager::Remove_MTP_Storage(unsigned int Storage_ID) {
2575#ifdef TW_HAS_MTP
2576 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2577 if (Part) {
2578 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2579 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002580 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002581 }
2582#endif
2583 return false;
2584}
Ethan Yonker96af84a2015-01-05 14:58:36 -06002585
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002586bool TWPartitionManager::Flash_Image(string& path, string& filename) {
Matt Mower23d8aae2017-01-06 14:30:33 -06002587 int partition_count = 0;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002588 TWPartition* flash_part = NULL;
bigbiffce8f83c2015-12-12 18:30:21 -05002589 string Flash_List, flash_path, full_filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002590 size_t start_pos = 0, end_pos = 0;
2591
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002592 full_filename = path + "/" + filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002593
bigbiffce8f83c2015-12-12 18:30:21 -05002594 gui_msg("image_flash_start=[IMAGE FLASH STARTED]");
2595 gui_msg(Msg("img_to_flash=Image to flash: '{1}'")(full_filename));
2596
2597 if (!TWFunc::Path_Exists(full_filename)) {
2598 if (!Mount_By_Path(full_filename, true)) {
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002599 return false;
2600 }
bigbiffce8f83c2015-12-12 18:30:21 -05002601 if (!TWFunc::Path_Exists(full_filename)) {
2602 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")(full_filename));
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002603 return false;
2604 }
2605 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002606
Ethan Yonker53796e72019-01-11 22:49:52 -06002607 DataManager::GetValue("tw_flash_partition", Flash_List);
2608 Repack_Type repack = REPLACE_NONE;
2609 if (Flash_List == "/repack_ramdisk;") {
2610 repack = REPLACE_RAMDISK;
2611 } else if (Flash_List == "/repack_kernel;") {
2612 repack = REPLACE_KERNEL;
2613 }
2614 if (repack != REPLACE_NONE) {
2615 Repack_Options_struct Repack_Options;
2616 Repack_Options.Type = repack;
2617 Repack_Options.Disable_Verity = false;
2618 Repack_Options.Disable_Force_Encrypt = false;
2619 Repack_Options.Backup_First = DataManager::GetIntValue("tw_repack_backup_first") != 0;
2620 return Repack_Images(full_filename, Repack_Options);
2621 }
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002622 PartitionSettings part_settings;
2623 part_settings.Backup_Folder = path;
2624 unsigned long long total_bytes = TWFunc::Get_File_Size(full_filename);
2625 ProgressTracking progress(total_bytes);
2626 part_settings.progress = &progress;
2627 part_settings.adbbackup = false;
2628 part_settings.PM_Method = PM_RESTORE;
Ethan Yonker74db1572015-10-28 12:44:49 -05002629 gui_msg("calc_restore=Calculating restore details...");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002630 if (!Flash_List.empty()) {
2631 end_pos = Flash_List.find(";", start_pos);
2632 while (end_pos != string::npos && start_pos < Flash_List.size()) {
2633 flash_path = Flash_List.substr(start_pos, end_pos - start_pos);
2634 flash_part = Find_Partition_By_Path(flash_path);
2635 if (flash_part != NULL) {
2636 partition_count++;
2637 if (partition_count > 1) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002638 gui_err("too_many_flash=Too many partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002639 return false;
2640 }
2641 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002642 gui_msg(Msg(msg::kError, "flash_unable_locate=Unable to locate '{1}' partition for flashing.")(flash_path));
Ethan Yonker96af84a2015-01-05 14:58:36 -06002643 return false;
2644 }
2645 start_pos = end_pos + 1;
2646 end_pos = Flash_List.find(";", start_pos);
2647 }
2648 }
2649
2650 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002651 gui_err("no_part_flash=No partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002652 return false;
2653 }
2654
2655 DataManager::SetProgress(0.0);
2656 if (flash_part) {
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002657 flash_part->Backup_FileName = filename;
2658 if (!flash_part->Flash_Image(&part_settings))
Ethan Yonker96af84a2015-01-05 14:58:36 -06002659 return false;
2660 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002661 gui_err("invalid_flash=Invalid flash partition specified.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002662 return false;
2663 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002664 gui_highlight("flash_done=IMAGE FLASH COMPLETED]");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002665 return true;
2666}
Ethan Yonker74db1572015-10-28 12:44:49 -05002667
2668void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value) {
2669 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2670 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002671 if (part->Is_Adopted_Storage) {
2672 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2673 part->Backup_Display_Name = part->Display_Name;
2674 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2675 } else {
2676 part->Display_Name = gui_lookup(resource_name, default_value);
2677 part->Backup_Display_Name = part->Display_Name;
2678 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002679 }
2680}
2681
2682void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value, const char* storage_resource_name, const char* storage_default_value) {
2683 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2684 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002685 if (part->Is_Adopted_Storage) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06002686 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup("data_backup", "Data (excl. storage)");
Ethan Yonker66a19492015-12-10 10:19:45 -06002687 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
Ethan Yonker66a19492015-12-10 10:19:45 -06002688 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2689 } else {
2690 part->Display_Name = gui_lookup(resource_name, default_value);
2691 part->Backup_Display_Name = part->Display_Name;
2692 if (part->Is_Storage)
2693 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2694 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002695 }
2696}
2697
Ethan Yonker01f4e032017-02-03 15:30:52 -06002698void 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) {
2699 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2700 if (part) {
2701 if (part->Is_Adopted_Storage) {
2702 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup(backup_name, backup_default);
2703 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2704 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2705 } else {
2706 part->Display_Name = gui_lookup(resource_name, default_value);
2707 part->Backup_Display_Name = gui_lookup(backup_name, backup_default);
2708 if (part->Is_Storage)
2709 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2710 }
2711 }
2712}
2713
Ethan Yonker74db1572015-10-28 12:44:49 -05002714void TWPartitionManager::Translate_Partition_Display_Names() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002715 LOGINFO("Translating partition display names\n");
Ethan Yonker74db1572015-10-28 12:44:49 -05002716 Translate_Partition("/system", "system", "System");
2717 Translate_Partition("/system_image", "system_image", "System Image");
2718 Translate_Partition("/vendor", "vendor", "Vendor");
2719 Translate_Partition("/vendor_image", "vendor_image", "Vendor Image");
2720 Translate_Partition("/cache", "cache", "Cache");
Ethan Yonker74db1572015-10-28 12:44:49 -05002721 Translate_Partition("/boot", "boot", "Boot");
2722 Translate_Partition("/recovery", "recovery", "Recovery");
2723 if (!datamedia) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06002724 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage");
Ethan Yonker74db1572015-10-28 12:44:49 -05002725 Translate_Partition("/sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2726 Translate_Partition("/internal_sd", "sdcard", "SDCard", "sdcard", "SDCard");
2727 Translate_Partition("/internal_sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2728 Translate_Partition("/emmc", "sdcard", "SDCard", "sdcard", "SDCard");
Ethan Yonker01f4e032017-02-03 15:30:52 -06002729 } else {
2730 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05002731 }
Ethan Yonker01f4e032017-02-03 15:30:52 -06002732 Translate_Partition("/external_sd", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
2733 Translate_Partition("/external_sdcard", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05002734 Translate_Partition("/usb-otg", "usbotg", "USB OTG", "usbotg", "USB OTG");
2735 Translate_Partition("/sd-ext", "sdext", "SD-EXT");
2736
2737 // Android secure is a special case
2738 TWPartition* part = PartitionManager.Find_Partition_By_Path("/and-sec");
2739 if (part)
2740 part->Backup_Display_Name = gui_lookup("android_secure", "Android Secure");
2741
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002742 std::vector<TWPartition*>::iterator sysfs;
2743 for (sysfs = Partitions.begin(); sysfs != Partitions.end(); sysfs++) {
2744 if (!(*sysfs)->Sysfs_Entry.empty()) {
2745 Translate_Partition((*sysfs)->Mount_Point.c_str(), "autostorage", "Storage", "autostorage", "Storage");
2746 }
2747 }
2748
Ethan Yonker74db1572015-10-28 12:44:49 -05002749 // This updates the text on all of the storage selection buttons in the GUI
2750 DataManager::SetBackupFolder();
2751}
Ethan Yonker66a19492015-12-10 10:19:45 -06002752
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002753bool TWPartitionManager::Decrypt_Adopted() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002754#ifdef TW_INCLUDE_CRYPTO
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002755 bool ret = false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002756 if (!Mount_By_Path("/data", false)) {
2757 LOGERR("Cannot decrypt adopted storage because /data will not mount\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002758 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002759 }
2760 LOGINFO("Decrypt adopted storage starting\n");
2761 char* xmlFile = PageManager::LoadFileToBuffer("/data/system/storage.xml", NULL);
2762 xml_document<> *doc = NULL;
2763 xml_node<>* volumes = NULL;
Ethan Yonker66a19492015-12-10 10:19:45 -06002764 string Primary_Storage_UUID = "";
2765 if (xmlFile != NULL) {
2766 LOGINFO("successfully loaded storage.xml\n");
2767 doc = new xml_document<>();
2768 doc->parse<0>(xmlFile);
2769 volumes = doc->first_node("volumes");
2770 if (volumes) {
2771 xml_attribute<>* psuuid = volumes->first_attribute("primaryStorageUuid");
2772 if (psuuid) {
2773 Primary_Storage_UUID = psuuid->value();
2774 }
2775 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002776 } else {
2777 LOGINFO("No /data/system/storage.xml for adopted storage\n");
2778 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002779 }
2780 std::vector<TWPartition*>::iterator adopt;
2781 for (adopt = Partitions.begin(); adopt != Partitions.end(); adopt++) {
2782 if ((*adopt)->Removable && (*adopt)->Is_Present) {
2783 if ((*adopt)->Decrypt_Adopted() == 0) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002784 ret = true;
Ethan Yonker66a19492015-12-10 10:19:45 -06002785 if (volumes) {
2786 xml_node<>* volume = volumes->first_node("volume");
2787 while (volume) {
2788 xml_attribute<>* guid = volume->first_attribute("partGuid");
2789 if (guid) {
2790 string GUID = (*adopt)->Adopted_GUID.c_str();
2791 GUID.insert(8, "-");
2792 GUID.insert(13, "-");
2793 GUID.insert(18, "-");
2794 GUID.insert(23, "-");
2795
2796 if (strcasecmp(GUID.c_str(), guid->value()) == 0) {
2797 xml_attribute<>* attr = volume->first_attribute("nickname");
nkk71ffb02bd2017-06-04 23:12:16 +03002798 if (attr && attr->value() && strlen(attr->value()) > 0) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002799 (*adopt)->Storage_Name = attr->value();
2800 (*adopt)->Display_Name = (*adopt)->Storage_Name;
2801 (*adopt)->Backup_Display_Name = (*adopt)->Storage_Name;
2802 LOGINFO("storage name from storage.xml is '%s'\n", attr->value());
2803 }
2804 attr = volume->first_attribute("fsUuid");
2805 if (attr && !Primary_Storage_UUID.empty() && strcmp(Primary_Storage_UUID.c_str(), attr->value()) == 0) {
2806 TWPartition* Dat = Find_Partition_By_Path("/data");
2807 if (Dat) {
2808 LOGINFO("Internal storage is found on adopted storage '%s'\n", (*adopt)->Display_Name.c_str());
2809 LOGINFO("Changing '%s' to point to '%s'\n", Dat->Symlink_Mount_Point.c_str(), (*adopt)->Storage_Path.c_str());
2810 (*adopt)->Symlink_Mount_Point = Dat->Symlink_Mount_Point;
2811 Dat->Symlink_Mount_Point = "";
2812 // Toggle mounts to ensure that the symlink mount point (probably /sdcard) is mounted to the right location
2813 Dat->UnMount(false);
2814 Dat->Mount(false);
2815 (*adopt)->UnMount(false);
2816 (*adopt)->Mount(false);
Ethan Yonker66a19492015-12-10 10:19:45 -06002817 }
2818 }
2819 break;
2820 }
2821 }
2822 volume = volume->next_sibling("volume");
2823 }
2824 }
nkk71ffb02bd2017-06-04 23:12:16 +03002825 Update_System_Details();
2826 Output_Partition((*adopt));
Ethan Yonker66a19492015-12-10 10:19:45 -06002827 }
2828 }
2829 }
2830 if (xmlFile) {
2831 doc->clear();
2832 delete doc;
2833 free(xmlFile);
2834 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002835 return ret;
Ethan Yonker66a19492015-12-10 10:19:45 -06002836#else
2837 LOGINFO("Decrypt_Adopted: no crypto support\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002838 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002839#endif
2840}
Ethan Yonkerfcf3f242016-02-16 12:30:26 -06002841
2842void TWPartitionManager::Remove_Partition_By_Path(string Path) {
2843 std::vector<TWPartition*>::iterator iter;
2844 string Local_Path = TWFunc::Get_Root_Path(Path);
2845
2846 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2847 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
2848 LOGINFO("Found and erasing '%s' from partition list\n", Local_Path.c_str());
2849 Partitions.erase(iter);
2850 return;
2851 }
2852 }
2853}
Ethan Yonker1b190162016-12-05 15:25:19 -06002854
2855void TWPartitionManager::Set_Active_Slot(const string& Slot) {
2856 if (Slot != "A" && Slot != "B") {
2857 LOGERR("Set_Active_Slot invalid slot '%s'\n", Slot.c_str());
2858 return;
2859 }
2860 if (Active_Slot_Display == Slot)
2861 return;
2862 LOGINFO("Setting active slot %s\n", Slot.c_str());
2863#ifdef AB_OTA_UPDATER
2864 if (!Active_Slot_Display.empty()) {
2865 const hw_module_t *hw_module;
2866 boot_control_module_t *module;
2867 int ret;
2868 ret = hw_get_module("bootctrl", &hw_module);
2869 if (ret != 0) {
2870 LOGERR("Error getting bootctrl module.\n");
2871 } else {
2872 module = (boot_control_module_t*) hw_module;
2873 module->init(module);
2874 int slot_number = 0;
2875 if (Slot == "B")
2876 slot_number = 1;
2877 if (module->setActiveBootSlot(module, slot_number))
2878 gui_msg(Msg(msg::kError, "unable_set_boot_slot=Error changing bootloader boot slot to {1}")(Slot));
2879 }
2880 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
2881 }
2882#else
2883 LOGERR("Boot slot feature not present\n");
2884#endif
2885 Active_Slot_Display = Slot;
2886 if (Fstab_Processed())
2887 Update_System_Details();
2888}
2889string TWPartitionManager::Get_Active_Slot_Suffix() {
2890 if (Active_Slot_Display == "A")
2891 return "_a";
2892 return "_b";
2893}
2894string TWPartitionManager::Get_Active_Slot_Display() {
2895 return Active_Slot_Display;
2896}
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002897
Captain Throwback9d6feb52018-07-27 10:05:24 -04002898string TWPartitionManager::Get_Android_Root_Path() {
Chaosmasterf6e42ce2020-01-27 00:17:04 +01002899 if (property_get_bool("ro.twrp.sar", false))
dianlujitao58f1a632020-01-16 23:03:56 +08002900 return "/system_root";
2901 return "/system";
Captain Throwback9d6feb52018-07-27 10:05:24 -04002902}
2903
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002904void TWPartitionManager::Remove_Uevent_Devices(const string& Mount_Point) {
2905 std::vector<TWPartition*>::iterator iter;
2906
2907 for (iter = Partitions.begin(); iter != Partitions.end(); ) {
2908 if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Mount_Point) {
2909 TWPartition *part = *iter;
2910 LOGINFO("%s was removed by uevent data\n", (*iter)->Mount_Point.c_str());
2911 (*iter)->UnMount(false);
2912 rmdir((*iter)->Mount_Point.c_str());
2913 iter = Partitions.erase(iter);
2914 delete part;
2915 } else {
2916 iter++;
2917 }
2918 }
2919}
2920
2921void TWPartitionManager::Handle_Uevent(const Uevent_Block_Data& uevent_data) {
2922 std::vector<TWPartition*>::iterator iter;
2923
2924 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2925 if (!(*iter)->Sysfs_Entry.empty()) {
2926 string device;
2927 size_t wildcard = (*iter)->Sysfs_Entry.find("*");
2928 if (wildcard != string::npos) {
2929 device = (*iter)->Sysfs_Entry.substr(0, wildcard);
2930 } else {
2931 device = (*iter)->Sysfs_Entry;
2932 }
2933 if (device == uevent_data.sysfs_path.substr(0, device.size())) {
2934 // Found a match
2935 if (uevent_data.action == "add") {
2936 (*iter)->Primary_Block_Device = "/dev/block/" + uevent_data.block_device;
2937 (*iter)->Alternate_Block_Device = (*iter)->Primary_Block_Device;
2938 (*iter)->Is_Present = true;
2939 LOGINFO("Found a match '%s' '%s'\n", uevent_data.block_device.c_str(), device.c_str());
2940 if (!Decrypt_Adopted()) {
2941 LOGINFO("No adopted storage so finding actual block device\n");
2942 (*iter)->Find_Actual_Block_Device();
2943 }
2944 return;
2945 } else if (uevent_data.action == "remove") {
2946 (*iter)->Is_Present = false;
2947 (*iter)->Primary_Block_Device = "";
2948 (*iter)->Actual_Block_Device = "";
2949 Remove_Uevent_Devices((*iter)->Mount_Point);
2950 return;
2951 }
2952 }
2953 }
2954 }
bigbiffee7b7ff2020-03-23 15:08:27 -04002955
2956 if (!PartitionManager.Get_Super_Status())
2957 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 -06002958}
2959
2960void TWPartitionManager::setup_uevent() {
2961 struct sockaddr_nl nls;
2962
2963 if (uevent_pfd.fd >= 0) {
2964 LOGINFO("uevent already set up\n");
2965 return;
2966 }
2967
2968 // Open hotplug event netlink socket
2969 memset(&nls,0,sizeof(struct sockaddr_nl));
2970 nls.nl_family = AF_NETLINK;
2971 nls.nl_pid = getpid();
2972 nls.nl_groups = -1;
2973 uevent_pfd.events = POLLIN;
2974 uevent_pfd.fd = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_KOBJECT_UEVENT);
2975 if (uevent_pfd.fd==-1) {
2976 LOGERR("uevent not root\n");
2977 return;
2978 }
2979
2980 // Listen to netlink socket
2981 if (::bind(uevent_pfd.fd, (struct sockaddr *) &nls, sizeof(struct sockaddr_nl)) < 0) {
2982 LOGERR("Bind failed\n");
2983 return;
2984 }
2985 set_select_fd();
2986 Coldboot();
2987}
2988
2989Uevent_Block_Data TWPartitionManager::get_event_block_values(char *buf, int len) {
2990 Uevent_Block_Data ret;
2991 ret.subsystem = "";
2992 char *ptr = buf;
2993 const char *end = buf + len;
2994
2995 buf[len - 1] = '\0';
2996 while (ptr < end) {
2997 if (strncmp(ptr, "ACTION=", strlen("ACTION=")) == 0) {
2998 ptr += strlen("ACTION=");
2999 ret.action = ptr;
3000 } else if (strncmp(ptr, "SUBSYSTEM=", strlen("SUBSYSTEM=")) == 0) {
3001 ptr += strlen("SUBSYSTEM=");
3002 ret.subsystem = ptr;
3003 } else if (strncmp(ptr, "DEVTYPE=", strlen("DEVTYPE=")) == 0) {
3004 ptr += strlen("DEVTYPE=");
3005 ret.type = ptr;
3006 } else if (strncmp(ptr, "DEVPATH=", strlen("DEVPATH=")) == 0) {
3007 ptr += strlen("DEVPATH=");
3008 ret.sysfs_path += ptr;
3009 } else if (strncmp(ptr, "DEVNAME=", strlen("DEVNAME=")) == 0) {
3010 ptr += strlen("DEVNAME=");
3011 ret.block_device += ptr;
3012 } else if (strncmp(ptr, "MAJOR=", strlen("MAJOR=")) == 0) {
3013 ptr += strlen("MAJOR=");
3014 ret.major = atoi(ptr);
3015 } else if (strncmp(ptr, "MINOR=", strlen("MINOR=")) == 0) {
3016 ptr += strlen("MINOR=");
3017 ret.minor = atoi(ptr);
3018 }
3019 ptr += strlen(ptr) + 1;
3020 }
3021 return ret;
3022}
3023
3024void TWPartitionManager::read_uevent() {
3025 char buf[1024];
3026
3027 int len = recv(uevent_pfd.fd, buf, sizeof(buf), MSG_DONTWAIT);
3028 if (len == -1) {
bigbiffd58ba182020-03-23 10:02:29 -04003029 LOGERR("recv error on uevent\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003030 return;
3031 }
3032 /*int i = 0; // Print all uevent output for test /debug
3033 while (i<len) {
3034 printf("%s\n", buf+i);
3035 i += strlen(buf+i)+1;
3036 }*/
3037 Uevent_Block_Data uevent_data = get_event_block_values(buf, len);
3038 if (uevent_data.subsystem == "block" && uevent_data.type == "disk") {
3039 PartitionManager.Handle_Uevent(uevent_data);
3040 }
3041}
3042
3043void TWPartitionManager::close_uevent() {
3044 if (uevent_pfd.fd > 0)
3045 close(uevent_pfd.fd);
3046 uevent_pfd.fd = -1;
3047}
3048
3049void TWPartitionManager::Add_Partition(TWPartition* Part) {
3050 Partitions.push_back(Part);
3051}
3052
3053void TWPartitionManager::Coldboot_Scan(std::vector<string> *sysfs_entries, const string& Path, int depth) {
3054 string Real_Path = Path;
3055 char real_path[PATH_MAX];
3056 if (realpath(Path.c_str(), &real_path[0])) {
3057 string Real_Path = real_path;
3058 std::vector<string>::iterator iter;
3059 for (iter = sysfs_entries->begin(); iter != sysfs_entries->end(); iter++) {
3060 if (Real_Path.find((*iter)) != string::npos) {
3061 string Write_Path = Real_Path + "/uevent";
3062 if (TWFunc::Path_Exists(Write_Path)) {
3063 const char* write_val = "add\n";
3064 TWFunc::write_to_file(Write_Path, write_val);
3065 break;
3066 }
3067 }
3068 }
3069 }
3070
3071 DIR* d = opendir(Path.c_str());
3072 if (d != NULL) {
3073 struct dirent* de;
3074 while ((de = readdir(d)) != NULL) {
3075 if (de->d_name[0] == '.' || (de->d_type != DT_DIR && depth > 0))
3076 continue;
3077 if (strlen(de->d_name) >= 4 && (strncmp(de->d_name, "ram", 3) == 0 || strncmp(de->d_name, "loop", 4) == 0))
3078 continue;
3079
3080 string item = Path + "/";
3081 item.append(de->d_name);
3082 Coldboot_Scan(sysfs_entries, item, depth + 1);
3083 }
3084 closedir(d);
3085 }
3086}
3087
3088void TWPartitionManager::Coldboot() {
3089 std::vector<TWPartition*>::iterator iter;
3090 std::vector<string> sysfs_entries;
3091
3092 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3093 if (!(*iter)->Sysfs_Entry.empty()) {
3094 size_t wildcard_pos = (*iter)->Sysfs_Entry.find("*");
3095 if (wildcard_pos == string::npos)
3096 wildcard_pos = (*iter)->Sysfs_Entry.size();
3097 sysfs_entries.push_back((*iter)->Sysfs_Entry.substr(0, wildcard_pos));
3098 }
3099 }
3100
3101 if (sysfs_entries.size() > 0)
3102 Coldboot_Scan(&sysfs_entries, "/sys/block", 0);
3103}
Ethan Yonker53796e72019-01-11 22:49:52 -06003104
3105bool TWPartitionManager::Prepare_Empty_Folder(const std::string& Folder) {
3106 if (TWFunc::Path_Exists(Folder))
3107 TWFunc::removeDir(Folder, false);
3108 return TWFunc::Recursive_Mkdir(Folder);
3109}
3110
3111bool TWPartitionManager::Prepare_Repack(TWPartition* Part, const std::string& Temp_Folder_Destination, const bool Create_Backup, const std::string& Backup_Name) {
3112 if (!Part) {
3113 LOGERR("Partition was null!\n");
3114 return false;
3115 }
3116 if (!Prepare_Empty_Folder(Temp_Folder_Destination))
3117 return false;
3118 std::string target_image = Temp_Folder_Destination + "boot.img";
3119 PartitionSettings part_settings;
3120 part_settings.Part = Part;
3121 if (Create_Backup) {
3122 if (Check_Backup_Name(Backup_Name, true, false) != 0)
3123 return false;
3124 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, part_settings.Backup_Folder);
3125 part_settings.Backup_Folder = part_settings.Backup_Folder + "/" + TWFunc::Get_Current_Date() + " " + Backup_Name + "/";
3126 if (!TWFunc::Recursive_Mkdir(part_settings.Backup_Folder))
3127 return false;
3128 } else
3129 part_settings.Backup_Folder = Temp_Folder_Destination;
3130 part_settings.adbbackup = false;
3131 part_settings.generate_digest = false;
3132 part_settings.generate_md5 = false;
3133 part_settings.PM_Method = PM_BACKUP;
3134 part_settings.progress = NULL;
3135 pid_t not_a_pid = 0;
3136 if (!Part->Backup(&part_settings, &not_a_pid))
3137 return false;
3138 std::string backed_up_image = part_settings.Backup_Folder;
3139 backed_up_image += Part->Backup_FileName;
3140 target_image = Temp_Folder_Destination + "boot.img";
3141 if (Create_Backup) {
3142 std::string source = part_settings.Backup_Folder + Part->Backup_FileName;
3143 if (TWFunc::copy_file(source, target_image, 0644) != 0) {
3144 LOGERR("Failed to copy backup file '%s' to temp folder target '%s'\n", source.c_str(), target_image.c_str());
3145 return false;
3146 }
3147 } else {
3148 if (rename(backed_up_image.c_str(), target_image.c_str()) != 0) {
3149 LOGERR("Failed to rename '%s' to '%s'\n", backed_up_image.c_str(), target_image.c_str());
3150 return false;
3151 }
3152 }
3153 return Prepare_Repack(target_image, Temp_Folder_Destination, false, false);
3154}
3155
3156bool TWPartitionManager::Prepare_Repack(const std::string& Source_Path, const std::string& Temp_Folder_Destination, const bool Copy_Source, const bool Create_Destination) {
3157 if (Create_Destination) {
3158 if (!Prepare_Empty_Folder(Temp_Folder_Destination))
3159 return false;
3160 }
3161 if (Copy_Source) {
3162 std::string destination = Temp_Folder_Destination + "/boot.img";
3163 if (TWFunc::copy_file(Source_Path, destination, 0644))
3164 return false;
3165 }
Mohd Farazb98b4f72020-05-10 04:18:30 +05303166 std::string command = "cd " + Temp_Folder_Destination + " && /sbin/magiskboot unpack -h '" + Source_Path +"'";
Ethan Yonker53796e72019-01-11 22:49:52 -06003167 if (TWFunc::Exec_Cmd(command) != 0) {
3168 LOGINFO("Error unpacking %s!\n", Source_Path.c_str());
3169 gui_msg(Msg(msg::kError, "unpack_error=Error unpacking image."));
3170 return false;
3171 }
3172 return true;
3173}
3174
3175bool TWPartitionManager::Repack_Images(const std::string& Target_Image, const struct Repack_Options_struct& Repack_Options) {
3176 if (!TWFunc::Path_Exists("/sbin/magiskboot")) {
3177 LOGERR("Image repacking tool not present in this TWRP build!");
3178 return false;
3179 }
3180 DataManager::SetProgress(0);
3181 TWPartition* part = PartitionManager.Find_Partition_By_Path("/boot");
3182 if (part)
3183 gui_msg(Msg("unpacking_image=Unpacking {1}...")(part->Display_Name));
3184 else {
3185 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/boot"));
3186 return false;
3187 }
3188 if (!PartitionManager.Prepare_Repack(part, REPACK_ORIG_DIR, Repack_Options.Backup_First, gui_lookup("repack", "Repack")))
3189 return false;
3190 DataManager::SetProgress(.25);
3191 gui_msg(Msg("unpacking_image=Unpacking {1}...")(Target_Image));
3192 if (!PartitionManager.Prepare_Repack(Target_Image, REPACK_NEW_DIR, true))
3193 return false;
3194 DataManager::SetProgress(.5);
3195 gui_msg(Msg("repacking_image=Repacking {1}...")(part->Display_Name));
3196 std::string path = REPACK_NEW_DIR;
3197 if (Repack_Options.Type == REPLACE_KERNEL) {
3198 // When we replace the kernel, what we really do is copy the boot partition ramdisk into the new image's folder
3199 if (TWFunc::copy_file(REPACK_ORIG_DIR "ramdisk.cpio", REPACK_NEW_DIR "ramdisk.cpio", 0644)) {
3200 LOGERR("Failed to copy ramdisk\n");
3201 return false;
3202 }
3203 } else if (Repack_Options.Type == REPLACE_RAMDISK) {
3204 // Repack the ramdisk
3205 if (TWFunc::copy_file(REPACK_NEW_DIR "ramdisk.cpio", REPACK_ORIG_DIR "ramdisk.cpio", 0644)) {
3206 LOGERR("Failed to copy ramdisk\n");
3207 return false;
3208 }
3209 path = REPACK_ORIG_DIR;
3210 } else {
3211 LOGERR("Invalid repacking options specified\n");
3212 return false;
3213 }
3214 if (Repack_Options.Disable_Verity)
3215 LOGERR("Disabling verity is not implemented yet\n");
3216 if (Repack_Options.Disable_Force_Encrypt)
3217 LOGERR("Disabling force encrypt is not implemented yet\n");
Mohd Farazb98b4f72020-05-10 04:18:30 +05303218 std::string command = "cd " + path + " && /sbin/magiskboot repack " + path + "boot.img";
Ethan Yonker53796e72019-01-11 22:49:52 -06003219 if (TWFunc::Exec_Cmd(command) != 0) {
3220 gui_msg(Msg(msg::kError, "repack_error=Error repacking image."));
3221 return false;
3222 }
3223 DataManager::SetProgress(.75);
3224 std::string file = "new-boot.img";
3225 DataManager::SetValue("tw_flash_partition", "/boot;");
3226 if (!PartitionManager.Flash_Image(path, file)) {
3227 LOGINFO("Error flashing new image\n");
3228 return false;
3229 }
3230 DataManager::SetProgress(1);
3231 TWFunc::removeDir(REPACK_ORIG_DIR, false);
Ethan Yonker9f5dd312019-05-15 15:17:36 -05003232 if (part->SlotSelect && Repack_Options.Type == REPLACE_RAMDISK) {
3233 LOGINFO("Switching slots to flash ramdisk to both partitions\n");
3234 string Current_Slot = Get_Active_Slot_Display();
3235 if (Current_Slot == "A")
3236 Set_Active_Slot("B");
3237 else
3238 Set_Active_Slot("A");
3239 DataManager::SetProgress(.25);
3240 if (!PartitionManager.Prepare_Repack(part, REPACK_ORIG_DIR, Repack_Options.Backup_First, gui_lookup("repack", "Repack")))
3241 return false;
3242 if (TWFunc::copy_file(REPACK_NEW_DIR "ramdisk.cpio", REPACK_ORIG_DIR "ramdisk.cpio", 0644)) {
3243 LOGERR("Failed to copy ramdisk\n");
3244 return false;
3245 }
3246 path = REPACK_ORIG_DIR;
Mohd Farazb98b4f72020-05-10 04:18:30 +05303247 command = "cd " + path + " && /sbin/magiskboot repack " + path + "boot.img";
Ethan Yonker9f5dd312019-05-15 15:17:36 -05003248 if (TWFunc::Exec_Cmd(command) != 0) {
3249 gui_msg(Msg(msg::kError, "repack_error=Error repacking image."));
3250 return false;
3251 }
3252 DataManager::SetProgress(.75);
3253 std::string file = "new-boot.img";
3254 DataManager::SetValue("tw_flash_partition", "/boot;");
3255 if (!PartitionManager.Flash_Image(path, file)) {
3256 LOGINFO("Error flashing new image\n");
3257 return false;
3258 }
3259 DataManager::SetProgress(1);
3260 TWFunc::removeDir(REPACK_ORIG_DIR, false);
3261 Set_Active_Slot(Current_Slot);
3262 }
Ethan Yonker53796e72019-01-11 22:49:52 -06003263 TWFunc::removeDir(REPACK_NEW_DIR, false);
3264 return true;
3265}
bigbiffee7b7ff2020-03-23 15:08:27 -04003266
3267bool TWPartitionManager::Prepare_Super_Volume(TWPartition* twrpPart) {
3268 Fstab fstab;
3269 std::string bare_partition_name;
3270
3271 if (twrpPart->Get_Mount_Point() == "/system_root")
3272 bare_partition_name = "system";
3273 else
3274 bare_partition_name = TWFunc::Remove_Beginning_Slash(twrpPart->Get_Mount_Point());
3275
3276 LOGINFO("Trying to prepare %s from super partition\n", bare_partition_name.c_str());
3277
3278 std::string blk_device_partition;
3279#ifdef AB_OTA_UPDATER
3280 blk_device_partition = bare_partition_name + PartitionManager.Get_Active_Slot_Suffix();
3281#else
3282 blk_device_partition = bare_partition_name;
3283#endif
3284
3285 FstabEntry fstabEntry = {
3286 .blk_device = blk_device_partition,
3287 .mount_point = twrpPart->Get_Mount_Point(),
3288 .fs_type = twrpPart->Current_File_System,
3289 .fs_mgr_flags.logical = twrpPart->Is_Super,
3290 };
3291
3292 fstab.emplace_back(fstabEntry);
3293 if (!fs_mgr_update_logical_partition(&fstabEntry)) {
3294 LOGERR("unable to update logical partition: %s\n", twrpPart->Get_Mount_Point().c_str());
3295 return false;
3296 }
3297
bigbiff32cbabe2020-04-12 19:16:19 -04003298 while (access(fstabEntry.blk_device.c_str(), F_OK) != 0) {
3299 usleep(100);
3300 }
3301
bigbiffee7b7ff2020-03-23 15:08:27 -04003302 twrpPart->Set_Block_Device(fstabEntry.blk_device);
3303 twrpPart->Update_Size(true);
3304 twrpPart->Change_Mount_Read_Only(true);
3305 twrpPart->Set_Can_Be_Backed_Up(false);
3306 twrpPart->Set_Can_Be_Wiped(false);
3307 return true;
3308}
3309
3310bool TWPartitionManager::Prepare_All_Super_Volumes() {
3311 bool status = true;
3312 std::vector<TWPartition*>::iterator iter;
3313
3314 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3315 if ((*iter)->Is_Super) {
3316 if (!Prepare_Super_Volume(*iter)) {
3317 status = false;
3318 }
3319 PartitionManager.Output_Partition(*iter);
3320 }
3321 }
3322 Update_System_Details();
3323 return status;
3324}
3325
3326bool TWPartitionManager::Is_Super_Partition(const char* fstab_line) {
3327 std::vector<std::string> super_partition_list = {"system", "vendor", "odm", "product", "system_ext"};
3328
3329 for (auto&& fstab_partition_check: super_partition_list) {
3330 if (strncmp(fstab_line, fstab_partition_check.c_str(), fstab_partition_check.size()) == 0) {
3331 DataManager::SetValue(TW_IS_SUPER, "1");
3332 return true;
3333 }
3334 }
3335 return false;
3336}
3337
3338std::string TWPartitionManager::Get_Super_Partition() {
3339 int slot_number = Get_Active_Slot_Display() == "A" ? 0 : 1;
3340 std::string super_device = fs_mgr_get_super_partition_name(slot_number);
3341 return "/dev/block/by-name/" + super_device;
3342}
3343
3344void TWPartitionManager::Setup_Super_Devices() {
3345 std::string superPart = Get_Super_Partition();
3346 android::fs_mgr::CreateLogicalPartitions(superPart);
bigbiff7ba75002020-04-11 20:47:09 -04003347}
3348
3349void TWPartitionManager::Setup_Super_Partition() {
bigbiffee7b7ff2020-03-23 15:08:27 -04003350 TWPartition* superPartition = new TWPartition();
bigbiff7ba75002020-04-11 20:47:09 -04003351 std::string superPart = Get_Super_Partition();
3352
3353 superPartition->Backup_Path = "/super";
3354 superPartition->Mount_Point = "/super";
bigbiffee7b7ff2020-03-23 15:08:27 -04003355 superPartition->Actual_Block_Device = superPart;
3356 superPartition->Alternate_Block_Device = superPart;
3357 superPartition->Backup_Display_Name = "super";
3358 superPartition->Can_Flash_Img = true;
bigbiffee7b7ff2020-03-23 15:08:27 -04003359 superPartition->Current_File_System = "emmc";
3360 superPartition->Can_Be_Backed_Up = true;
3361 superPartition->Is_Present = true;
3362 superPartition->Is_SubPartition = false;
bigbiff7ba75002020-04-11 20:47:09 -04003363 superPartition->Setup_Image();
bigbiffee7b7ff2020-03-23 15:08:27 -04003364 Add_Partition(superPartition);
3365 PartitionManager.Output_Partition(superPartition);
3366 Update_System_Details();
3367}
3368
3369bool TWPartitionManager::Get_Super_Status() {
3370 return access(Get_Super_Partition().c_str(), F_OK) == 0;
3371}
bigbiff25d25b92020-06-19 16:07:38 -04003372
3373bool TWPartitionManager::Recreate_Logs_Dir() {
3374#ifdef TW_INCLUDE_FBE
3375 struct passwd pd;
3376 struct passwd *pwdptr = &pd;
3377 struct passwd *tempPd;
3378 char pwdBuf[512];
3379 int uid = 0, gid = 0;
3380
3381 if ((getpwnam_r("system", pwdptr, pwdBuf, sizeof(pwdBuf), &tempPd)) != 0) {
3382 LOGERR("unable to get system user id\n");
3383 return false;
3384 } else {
3385 struct group grp;
3386 struct group *grpptr = &grp;
3387 struct group *tempGrp;
3388 char grpBuf[512];
3389
3390 if ((getgrnam_r("cache", grpptr, grpBuf, sizeof(grpBuf), &tempGrp)) != 0) {
3391 LOGERR("unable to get cache group id\n");
3392 return false;
3393 } else {
3394 uid = pd.pw_uid;
3395 gid = grp.gr_gid;
3396 std::string abLogsRecoveryDir(DATA_LOGS_DIR);
3397 abLogsRecoveryDir += "/recovery/";
3398
3399 if (!TWFunc::Create_Dir_Recursive(abLogsRecoveryDir, S_IRWXU | S_IRWXG | S_IWGRP | S_IXGRP, uid, gid)) {
3400 LOGERR("Unable to recreate %s\n", abLogsRecoveryDir.c_str());
3401 return false;
3402 }
3403 if (setfilecon(abLogsRecoveryDir.c_str(), "u:object_r:cache_file:s0") != 0) {
3404 LOGERR("Unable to set contexts for %s\n", abLogsRecoveryDir.c_str());
3405 return false;
3406 }
3407 }
3408 }
3409#endif
3410 return true;
3411}