blob: 555723faa11c36a3cade353ca946aad0b0414ccd [file] [log] [blame]
Dees Troy3be70a82013-10-22 14:25:12 +00001/*
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04002 Copyright 2014 to 2017 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>
bigbiffce8f83c2015-12-12 18:30:21 -050031#include <zlib.h>
bigbiff bigbiff9c754052013-01-09 09:09:08 -050032#include <iostream>
33#include <iomanip>
Ethan Yonker8613dc02014-09-11 09:28:20 -050034#include <sys/wait.h>
Ethan Yonker483e9f42016-01-11 22:21:18 -060035#include <linux/fs.h>
36#include <sys/mount.h>
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060037
38#include <sys/poll.h>
39#include <sys/socket.h>
40#include <linux/types.h>
41#include <linux/netlink.h>
42
Dees_Troy51a0e822012-09-05 15:24:24 -040043#include "variables.h"
Dees_Troy2673cec2013-04-02 20:22:16 +000044#include "twcommon.h"
Dees_Troy51a0e822012-09-05 15:24:24 -040045#include "partitions.hpp"
Dees_Troy5bf43922012-09-07 16:07:55 -040046#include "data.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040047#include "twrp-functions.hpp"
Ethan Yonkerb5fab762016-01-28 14:03:33 -060048#include "fixContexts.hpp"
Ethan Yonker3fdcda42016-11-30 12:29:37 -060049#include "exclude.hpp"
Ethan Yonker4b94cfd2014-12-11 10:00:45 -060050#include "set_metadata.h"
bigbiff7abc5fe2015-01-17 16:53:12 -050051#include "tw_atomic.hpp"
Ethan Yonker74db1572015-10-28 12:44:49 -050052#include "gui/gui.hpp"
Ethan Yonker472f5062016-02-25 13:47:30 -060053#include "progresstracking.hpp"
bigbiff bigbiff56cf5642016-08-19 17:43:45 -040054#include "twrpDigestDriver.hpp"
bigbiffce8f83c2015-12-12 18:30:21 -050055#include "adbbu/libtwadbbu.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040056
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040057#ifdef TW_HAS_MTP
bigbiff bigbiffaf32bb92018-12-18 18:39:53 -050058#ifdef TW_HAS_LEGACY_MTP
59#include "mtp/legacy/mtp_MtpServer.hpp"
60#include "mtp/legacy/twrpMtp.hpp"
61#include "mtp/legacy/MtpMessage.hpp"
62#else
63#include "mtp/ffs/mtp_MtpServer.hpp"
64#include "mtp/ffs/twrpMtp.hpp"
65#include "mtp/ffs/MtpMessage.hpp"
66#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040067#endif
68
Dees Troy6f6441d2014-01-23 02:07:03 +000069extern "C" {
70 #include "cutils/properties.h"
Ethan Yonker74db1572015-10-28 12:44:49 -050071 #include "gui/gui.h"
Dees Troy6f6441d2014-01-23 02:07:03 +000072}
73
Dees_Troy5bf43922012-09-07 16:07:55 -040074#ifdef TW_INCLUDE_CRYPTO
Noah Jacobson5a79f672019-04-28 00:10:07 -040075#include "crypto/fde/cryptfs.h"
76#include "gui/rapidxml.hpp"
77#include "gui/pages.hpp"
78#ifdef TW_INCLUDE_FBE
79#include "crypto/ext4crypt/Decrypt.h"
80#ifdef TW_INCLUDE_FBE_METADATA_DECRYPT
81#include "crypto/ext4crypt/MetadataCrypt.h"
82#endif
83#endif
84#ifdef TW_CRYPTO_USE_SYSTEM_VOLD
85#include "crypto/vold_decrypt/vold_decrypt.h"
86#endif
Dees_Troy5bf43922012-09-07 16:07:55 -040087#endif
Dees_Troy51a0e822012-09-05 15:24:24 -040088
Ethan Yonker1b190162016-12-05 15:25:19 -060089#ifdef AB_OTA_UPDATER
90#include <hardware/hardware.h>
91#include <hardware/boot_control.h>
92#endif
93
Ethan Yonker6277c792014-09-15 14:54:30 -050094extern bool datamedia;
Noah Jacobson5a79f672019-04-28 00:10:07 -040095std::vector<users_struct> Users_List;
Ethan Yonker6277c792014-09-15 14:54:30 -050096
bigbiff bigbiff34684ff2013-12-01 21:03:45 -050097TWPartitionManager::TWPartitionManager(void) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040098 mtp_was_enabled = false;
Ethan Yonker726a0202014-12-16 20:01:38 -060099 mtp_write_fd = -1;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600100 uevent_pfd.fd = -1;
bigbiff7abc5fe2015-01-17 16:53:12 -0500101 stop_backup.set_value(0);
Ethan Yonker1b190162016-12-05 15:25:19 -0600102#ifdef AB_OTA_UPDATER
103 char slot_suffix[PROPERTY_VALUE_MAX];
Ethan Yonkerfefe5912017-09-30 22:22:13 -0500104 property_get("ro.boot.slot_suffix", slot_suffix, "error");
105 if (strcmp(slot_suffix, "error") == 0)
106 property_get("ro.boot.slot", slot_suffix, "error");
Ethan Yonker1b190162016-12-05 15:25:19 -0600107 Active_Slot_Display = "";
Ethan Yonkerfefe5912017-09-30 22:22:13 -0500108 if (strcmp(slot_suffix, "_a") == 0 || strcmp(slot_suffix, "a") == 0)
Ethan Yonker1b190162016-12-05 15:25:19 -0600109 Set_Active_Slot("A");
110 else
111 Set_Active_Slot("B");
112#endif
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500113}
114
Captain Throwback07b8e052020-10-29 12:43:17 -0400115int TWPartitionManager::Set_FDE_Encrypt_Status(void) {
116 property_set("ro.crypto.state", "encrypted");
117 property_set("ro.crypto.type", "block");
118 // Sleep for a bit so that services can start if needed
119 sleep(1);
120 return 0;
121}
122
Chaosmasterf6e42ce2020-01-27 00:17:04 +0100123int TWPartitionManager::Process_Fstab(string Fstab_Filename, bool Display_Error, bool Sar_Detect) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400124 FILE *fstabFile;
125 char fstab_line[MAX_FSTAB_LINE_LENGTH];
Dees Troy02a64532014-03-19 15:23:32 +0000126 TWPartition* settings_partition = NULL;
Matt Mowerbf4efa32014-04-14 23:25:26 -0500127 TWPartition* andsec_partition = NULL;
Ethan Yonker726a0202014-12-16 20:01:38 -0600128 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 -0600129 std::map<string, Flags_Map> twrp_flags;
130
131 fstabFile = fopen("/etc/twrp.flags", "rt");
132 if (fstabFile != NULL) {
133 LOGINFO("reading /etc/twrp.flags\n");
134 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
135 if (fstab_line[0] != '/')
136 continue;
137
138 size_t line_size = strlen(fstab_line);
139 if (fstab_line[line_size - 1] != '\n')
140 fstab_line[line_size] = '\n';
141 Flags_Map line_flags;
142 line_flags.Primary_Block_Device = "";
143 line_flags.Alternate_Block_Device = "";
144 line_flags.fstab_line = (char*)malloc(MAX_FSTAB_LINE_LENGTH);
145 if (!line_flags.fstab_line) {
146 LOGERR("malloc error on line_flags.fstab_line\n");
147 return false;
148 }
149 memcpy(line_flags.fstab_line, fstab_line, MAX_FSTAB_LINE_LENGTH);
150 bool found_separator = false;
151 char *fs_loc = NULL;
152 char *block_loc = NULL;
153 char *flags_loc = NULL;
154 size_t index, item_index = 0;
155 for (index = 0; index < line_size; index++) {
156 if (fstab_line[index] <= 32) {
157 fstab_line[index] = '\0';
158 found_separator = true;
159 } else if (found_separator) {
160 if (item_index == 0) {
161 fs_loc = fstab_line + index;
162 } else if (item_index == 1) {
163 block_loc = fstab_line + index;
164 } else if (item_index > 1) {
165 char *ptr = fstab_line + index;
166 if (*ptr == '/') {
167 line_flags.Alternate_Block_Device = ptr;
168 } else if (strlen(ptr) > strlen("flags=") && strncmp(ptr, "flags=", strlen("flags=")) == 0) {
169 flags_loc = ptr;
170 // Once we find the flags=, we're done scanning the line
171 break;
172 }
173 }
174 found_separator = false;
175 item_index++;
176 }
177 }
178 if (block_loc)
179 line_flags.Primary_Block_Device = block_loc;
180 if (fs_loc)
181 line_flags.File_System = fs_loc;
182 if (flags_loc)
183 line_flags.Flags = flags_loc;
184 string Mount_Point = fstab_line;
185 twrp_flags[Mount_Point] = line_flags;
186 memset(fstab_line, 0, sizeof(fstab_line));
187 }
188 fclose(fstabFile);
189 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400190
191 fstabFile = fopen(Fstab_Filename.c_str(), "rt");
192 if (fstabFile == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000193 LOGERR("Critical Error: Unable to open fstab at '%s'.\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400194 return false;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600195 } else
196 LOGINFO("Reading %s\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400197
198 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
199 if (fstab_line[0] != '/')
200 continue;
201
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600202 if (strstr(fstab_line, "swap"))
203 continue; // Skip swap in recovery
204
205 size_t line_size = strlen(fstab_line);
206 if (fstab_line[line_size - 1] != '\n')
207 fstab_line[line_size] = '\n';
Dees_Troy2a923582012-09-20 12:13:34 -0400208
Matt Mower2b2dd152016-04-26 11:24:08 -0500209 TWPartition* partition = new TWPartition();
Chaosmasterf6e42ce2020-01-27 00:17:04 +0100210 if (partition->Process_Fstab_Line(fstab_line, Display_Error, &twrp_flags, Sar_Detect))
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500211 Partitions.push_back(partition);
Matt Mower72c87ce2016-04-26 14:34:56 -0500212 else
Dees_Troy5bf43922012-09-07 16:07:55 -0400213 delete partition;
Matt Mower2b2dd152016-04-26 11:24:08 -0500214
215 memset(fstab_line, 0, sizeof(fstab_line));
Dees_Troy5bf43922012-09-07 16:07:55 -0400216 }
217 fclose(fstabFile);
Matt Mower72c87ce2016-04-26 14:34:56 -0500218
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600219 if (twrp_flags.size() > 0) {
220 LOGINFO("Processing remaining twrp.flags\n");
221 // Add any items from twrp.flags that did not exist in the recovery.fstab
222 for (std::map<string, Flags_Map>::iterator mapit=twrp_flags.begin(); mapit!=twrp_flags.end(); mapit++) {
223 if (Find_Partition_By_Path(mapit->first) == NULL) {
224 TWPartition* partition = new TWPartition();
Chaosmasterf6e42ce2020-01-27 00:17:04 +0100225 if (partition->Process_Fstab_Line(mapit->second.fstab_line, Display_Error, NULL, Sar_Detect))
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600226 Partitions.push_back(partition);
227 else
228 delete partition;
229 }
230 if (mapit->second.fstab_line)
231 free(mapit->second.fstab_line);
232 mapit->second.fstab_line = NULL;
233 }
234 }
235 LOGINFO("Done processing fstab files\n");
236
Matt Mower72c87ce2016-04-26 14:34:56 -0500237 std::vector<TWPartition*>::iterator iter;
238 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Chaosmasterf6e42ce2020-01-27 00:17:04 +0100239 if (Sar_Detect) {
240 if ((*iter)->Mount_Point == "/s")
241 return true;
242 else
243 continue;
244 }
245
Matt Mower72c87ce2016-04-26 14:34:56 -0500246 (*iter)->Partition_Post_Processing(Display_Error);
247
248 if ((*iter)->Is_Storage) {
249 ++storageid;
250 (*iter)->MTP_Storage_ID = storageid;
251 }
252
253 if (!settings_partition && (*iter)->Is_Settings_Storage && (*iter)->Is_Present)
254 settings_partition = (*iter);
255 else
256 (*iter)->Is_Settings_Storage = false;
257
258 if (!andsec_partition && (*iter)->Has_Android_Secure && (*iter)->Is_Present)
259 andsec_partition = (*iter);
260 else
261 (*iter)->Has_Android_Secure = false;
262 }
263
Ethan Yonker6277c792014-09-15 14:54:30 -0500264 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) {
265 // Attempt to automatically identify /data/media emulated storage devices
266 TWPartition* Dat = Find_Partition_By_Path("/data");
267 if (Dat) {
268 LOGINFO("Using automatic handling for /data/media emulated storage device.\n");
269 datamedia = true;
that9e0593e2014-10-08 00:01:24 +0200270 Dat->Setup_Data_Media();
Ethan Yonker6277c792014-09-15 14:54:30 -0500271 settings_partition = Dat;
Ethan Yonker726a0202014-12-16 20:01:38 -0600272 // Since /data was not considered a storage partition earlier, we still need to assign an MTP ID
273 ++storageid;
274 Dat->MTP_Storage_ID = storageid;
Ethan Yonker6277c792014-09-15 14:54:30 -0500275 }
276 }
Dees Troy02a64532014-03-19 15:23:32 +0000277 if (!settings_partition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500278 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
279 if ((*iter)->Is_Storage) {
Dees Troy02a64532014-03-19 15:23:32 +0000280 settings_partition = (*iter);
Dees_Troya13d74f2013-03-24 08:54:55 -0500281 break;
282 }
283 }
Dees Troy02a64532014-03-19 15:23:32 +0000284 if (!settings_partition)
Dees_Troy2673cec2013-04-02 20:22:16 +0000285 LOGERR("Unable to locate storage partition for storing settings file.\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500286 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400287 if (!Write_Fstab()) {
288 if (Display_Error)
Dees_Troy2673cec2013-04-02 20:22:16 +0000289 LOGERR("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400290 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000291 LOGINFO("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400292 }
Matt Mowered71fa32014-04-16 13:21:47 -0500293
Matt Mowerbf4efa32014-04-14 23:25:26 -0500294 if (andsec_partition) {
295 Setup_Android_Secure_Location(andsec_partition);
Matt Mowered71fa32014-04-16 13:21:47 -0500296 } else if (settings_partition) {
Matt Mowerbf4efa32014-04-14 23:25:26 -0500297 Setup_Android_Secure_Location(settings_partition);
298 }
Matt Mowered71fa32014-04-16 13:21:47 -0500299 if (settings_partition) {
300 Setup_Settings_Storage_Partition(settings_partition);
301 }
Ethan Yonker253368a2014-11-25 15:00:52 -0600302#ifdef TW_INCLUDE_CRYPTO
Ethan Yonkercceebb82014-11-18 10:17:59 -0600303 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
304 if (Decrypt_Data && Decrypt_Data->Is_Encrypted && !Decrypt_Data->Is_Decrypted) {
nebrassy674bb0c2020-10-24 21:08:20 +0200305 property_set("ro.crypto.state", "encrypted");
Ethan Yonker93382822018-11-01 15:25:31 -0500306 if (!Decrypt_Data->Key_Directory.empty() && Mount_By_Path(Decrypt_Data->Key_Directory, false)) {
nebrassy674bb0c2020-10-24 21:08:20 +0200307 property_set("ro.crypto.type", "file");
Ethan Yonker93382822018-11-01 15:25:31 -0500308#ifdef TW_INCLUDE_FBE_METADATA_DECRYPT
309 if (e4crypt_mount_metadata_encrypted(Decrypt_Data->Mount_Point, false, Decrypt_Data->Key_Directory, Decrypt_Data->Actual_Block_Device, &Decrypt_Data->Decrypted_Block_Device)) {
310 LOGINFO("Successfully decrypted metadata encrypted data partition with new block device: '%s'\n", Decrypt_Data->Decrypted_Block_Device.c_str());
Ethan Yonker93382822018-11-01 15:25:31 -0500311 Decrypt_Data->Is_Decrypted = true; // Needed to make the mount function work correctly
312 int retry_count = 10;
313 while (!Decrypt_Data->Mount(false) && --retry_count)
314 usleep(500);
315 if (Decrypt_Data->Mount(false)) {
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400316 if (!Decrypt_Data->Decrypt_FBE_DE()) {
mauronofrio1db94322019-11-23 23:13:04 +0100317 char wrappedvalue[PROPERTY_VALUE_MAX];
318 property_get("fbe.data.wrappedkey", wrappedvalue, "");
319 std::string wrappedkeyvalue(wrappedvalue);
320 if (wrappedkeyvalue == "true") {
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400321 LOGERR("Unable to decrypt FBE device\n");
mauronofrio1db94322019-11-23 23:13:04 +0100322 } else {
323 LOGINFO("Trying wrapped key.\n");
324 property_set("fbe.data.wrappedkey", "true");
325 if (!Decrypt_Data->Decrypt_FBE_DE()) {
326 LOGERR("Unable to decrypt FBE device\n");
327 }
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400328 }
329 }
330
Ethan Yonker93382822018-11-01 15:25:31 -0500331 } else {
332 LOGINFO("Failed to mount data after metadata decrypt\n");
333 }
334 } else {
335 LOGINFO("Unable to decrypt metadata encryption\n");
336 }
337#else
338 LOGERR("Metadata FBE decrypt support not present in this TWRP\n");
339#endif
340 }
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600341 if (Decrypt_Data->Is_FBE) {
342 if (DataManager::GetIntValue(TW_CRYPTO_PWTYPE) == 0) {
343 if (Decrypt_Device("!") == 0) {
344 gui_msg("decrypt_success=Successfully decrypted with default password.");
345 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
346 } else {
347 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
348 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600349 }
350 } else {
Captain Throwback07b8e052020-10-29 12:43:17 -0400351 Set_FDE_Encrypt_Status();
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600352 int password_type = cryptfs_get_password_type();
353 if (password_type == CRYPT_TYPE_DEFAULT) {
354 LOGINFO("Device is encrypted with the default password, attempting to decrypt.\n");
355 if (Decrypt_Device("default_password") == 0) {
356 gui_msg("decrypt_success=Successfully decrypted with default password.");
357 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
358 } else {
359 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
360 }
361 } else {
362 DataManager::SetValue("TW_CRYPTO_TYPE", password_type);
Noah Jacobson5a79f672019-04-28 00:10:07 -0400363 DataManager::SetValue("tw_crypto_pwtype_0", password_type);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600364 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600365 }
366 }
Noah Jacobson5a79f672019-04-28 00:10:07 -0400367 if (Decrypt_Data && (!Decrypt_Data->Is_Encrypted || Decrypt_Data->Is_Decrypted) &&
368 Decrypt_Data->Mount(false)) {
Ethan Yonker66a19492015-12-10 10:19:45 -0600369 Decrypt_Adopted();
370 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600371#endif
Dees_Troy51127312012-09-08 13:08:49 -0400372 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -0400373 UnMount_Main_Partitions();
Ethan Yonker1b190162016-12-05 15:25:19 -0600374#ifdef AB_OTA_UPDATER
375 DataManager::SetValue("tw_active_slot", Get_Active_Slot_Display());
376#endif
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600377 setup_uevent();
Dees_Troy5bf43922012-09-07 16:07:55 -0400378 return true;
379}
380
381int TWPartitionManager::Write_Fstab(void) {
382 FILE *fp;
383 std::vector<TWPartition*>::iterator iter;
384 string Line;
385
386 fp = fopen("/etc/fstab", "w");
387 if (fp == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000388 LOGINFO("Can not open /etc/fstab.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400389 return false;
390 }
Dees_Troy63c8df72012-09-10 14:02:05 -0400391 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy51127312012-09-08 13:08:49 -0400392 if ((*iter)->Can_Be_Mounted) {
dianlujitao7d304c72016-01-02 12:15:50 +0800393 Line = (*iter)->Actual_Block_Device + " " + (*iter)->Mount_Point + " " + (*iter)->Current_File_System + " rw 0 0\n";
Dees_Troy5bf43922012-09-07 16:07:55 -0400394 fputs(Line.c_str(), fp);
Dees_Troy91862e62013-04-04 23:48:21 +0000395 }
396 // Handle subpartition tracking
397 if ((*iter)->Is_SubPartition) {
398 TWPartition* ParentPartition = Find_Partition_By_Path((*iter)->SubPartition_Of);
399 if (ParentPartition)
400 ParentPartition->Has_SubPartition = true;
401 else
402 LOGERR("Unable to locate parent partition '%s' of '%s'\n", (*iter)->SubPartition_Of.c_str(), (*iter)->Mount_Point.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400403 }
404 }
405 fclose(fp);
406 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -0400407}
408
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500409void TWPartitionManager::Setup_Settings_Storage_Partition(TWPartition* Part) {
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500410 DataManager::SetValue("tw_settings_path", Part->Storage_Path);
411 DataManager::SetValue("tw_storage_path", Part->Storage_Path);
412 LOGINFO("Settings storage is '%s'\n", Part->Storage_Path.c_str());
413}
414
Matt Mowerbf4efa32014-04-14 23:25:26 -0500415void TWPartitionManager::Setup_Android_Secure_Location(TWPartition* Part) {
416 if (Part->Has_Android_Secure)
417 Part->Setup_AndSec();
Ethan Yonker6277c792014-09-15 14:54:30 -0500418 else if (!datamedia)
Matt Mowerbf4efa32014-04-14 23:25:26 -0500419 Part->Setup_AndSec();
Matt Mowerbf4efa32014-04-14 23:25:26 -0500420}
421
Dees_Troy8170a922012-09-18 15:40:25 -0400422void TWPartitionManager::Output_Partition_Logging(void) {
423 std::vector<TWPartition*>::iterator iter;
424
425 printf("\n\nPartition Logs:\n");
426 for (iter = Partitions.begin(); iter != Partitions.end(); iter++)
427 Output_Partition((*iter));
428}
429
430void TWPartitionManager::Output_Partition(TWPartition* Part) {
431 unsigned long long mb = 1048576;
432
Gary Peck004d48b2012-11-21 16:28:18 -0800433 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 -0400434 if (Part->Can_Be_Mounted) {
Gary Peck004d48b2012-11-21 16:28:18 -0800435 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 -0400436 }
Gary Peck004d48b2012-11-21 16:28:18 -0800437 printf("\n Flags: ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500438 if (Part->Can_Be_Mounted)
439 printf("Can_Be_Mounted ");
Gary Peck004d48b2012-11-21 16:28:18 -0800440 if (Part->Can_Be_Wiped)
441 printf("Can_Be_Wiped ");
Hashcodedabfd492013-08-29 22:45:30 -0700442 if (Part->Use_Rm_Rf)
443 printf("Use_Rm_Rf ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500444 if (Part->Can_Be_Backed_Up)
445 printf("Can_Be_Backed_Up ");
Gary Peck004d48b2012-11-21 16:28:18 -0800446 if (Part->Wipe_During_Factory_Reset)
447 printf("Wipe_During_Factory_Reset ");
448 if (Part->Wipe_Available_in_GUI)
449 printf("Wipe_Available_in_GUI ");
450 if (Part->Is_SubPartition)
451 printf("Is_SubPartition ");
452 if (Part->Has_SubPartition)
453 printf("Has_SubPartition ");
454 if (Part->Removable)
455 printf("Removable ");
456 if (Part->Is_Present)
457 printf("IsPresent ");
458 if (Part->Can_Be_Encrypted)
459 printf("Can_Be_Encrypted ");
460 if (Part->Is_Encrypted)
461 printf("Is_Encrypted ");
462 if (Part->Is_Decrypted)
463 printf("Is_Decrypted ");
464 if (Part->Has_Data_Media)
465 printf("Has_Data_Media ");
Dees_Troy83bd4832013-05-04 12:39:56 +0000466 if (Part->Can_Encrypt_Backup)
467 printf("Can_Encrypt_Backup ");
468 if (Part->Use_Userdata_Encryption)
469 printf("Use_Userdata_Encryption ");
Gary Peck004d48b2012-11-21 16:28:18 -0800470 if (Part->Has_Android_Secure)
471 printf("Has_Android_Secure ");
472 if (Part->Is_Storage)
473 printf("Is_Storage ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500474 if (Part->Is_Settings_Storage)
475 printf("Is_Settings_Storage ");
Dees_Troy68cab492012-12-12 19:29:35 +0000476 if (Part->Ignore_Blkid)
477 printf("Ignore_Blkid ");
Dees_Troy16c2b312013-01-15 16:51:18 +0000478 if (Part->Retain_Layout_Version)
479 printf("Retain_Layout_Version ");
Ethan Yonker253368a2014-11-25 15:00:52 -0600480 if (Part->Mount_To_Decrypt)
481 printf("Mount_To_Decrypt ");
Ethan Yonker96af84a2015-01-05 14:58:36 -0600482 if (Part->Can_Flash_Img)
483 printf("Can_Flash_Img ");
Ethan Yonker66a19492015-12-10 10:19:45 -0600484 if (Part->Is_Adopted_Storage)
485 printf("Is_Adopted_Storage ");
Ethan Yonker1b190162016-12-05 15:25:19 -0600486 if (Part->SlotSelect)
487 printf("SlotSelect ");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600488 if (Part->Mount_Read_Only)
489 printf("Mount_Read_Only ");
Gary Peck004d48b2012-11-21 16:28:18 -0800490 printf("\n");
491 if (!Part->SubPartition_Of.empty())
492 printf(" SubPartition_Of: %s\n", Part->SubPartition_Of.c_str());
493 if (!Part->Symlink_Path.empty())
494 printf(" Symlink_Path: %s\n", Part->Symlink_Path.c_str());
495 if (!Part->Symlink_Mount_Point.empty())
496 printf(" Symlink_Mount_Point: %s\n", Part->Symlink_Mount_Point.c_str());
497 if (!Part->Primary_Block_Device.empty())
498 printf(" Primary_Block_Device: %s\n", Part->Primary_Block_Device.c_str());
499 if (!Part->Alternate_Block_Device.empty())
500 printf(" Alternate_Block_Device: %s\n", Part->Alternate_Block_Device.c_str());
501 if (!Part->Decrypted_Block_Device.empty())
502 printf(" Decrypted_Block_Device: %s\n", Part->Decrypted_Block_Device.c_str());
dianlujitao4879b372018-12-03 18:45:47 +0800503 if (!Part->Crypto_Key_Location.empty())
Ethan Yonker253368a2014-11-25 15:00:52 -0600504 printf(" Crypto_Key_Location: %s\n", Part->Crypto_Key_Location.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800505 if (Part->Length != 0)
506 printf(" Length: %i\n", Part->Length);
507 if (!Part->Display_Name.empty())
508 printf(" Display_Name: %s\n", Part->Display_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500509 if (!Part->Storage_Name.empty())
510 printf(" Storage_Name: %s\n", Part->Storage_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800511 if (!Part->Backup_Path.empty())
512 printf(" Backup_Path: %s\n", Part->Backup_Path.c_str());
513 if (!Part->Backup_Name.empty())
514 printf(" Backup_Name: %s\n", Part->Backup_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500515 if (!Part->Backup_Display_Name.empty())
516 printf(" Backup_Display_Name: %s\n", Part->Backup_Display_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800517 if (!Part->Backup_FileName.empty())
518 printf(" Backup_FileName: %s\n", Part->Backup_FileName.c_str());
519 if (!Part->Storage_Path.empty())
520 printf(" Storage_Path: %s\n", Part->Storage_Path.c_str());
521 if (!Part->Current_File_System.empty())
522 printf(" Current_File_System: %s\n", Part->Current_File_System.c_str());
523 if (!Part->Fstab_File_System.empty())
524 printf(" Fstab_File_System: %s\n", Part->Fstab_File_System.c_str());
525 if (Part->Format_Block_Size != 0)
Ethan Yonkerc798c9c2015-10-09 11:15:26 -0500526 printf(" Format_Block_Size: %lu\n", Part->Format_Block_Size);
Dees_Troy094207a2012-09-26 12:00:39 -0400527 if (!Part->MTD_Name.empty())
528 printf(" MTD_Name: %s\n", Part->MTD_Name.c_str());
Matt Mower029a82d2017-01-08 13:12:38 -0600529 printf(" Backup_Method: %s\n", Part->Backup_Method_By_Name().c_str());
Hashcode62bd9e02013-11-19 21:59:42 -0800530 if (Part->Mount_Flags || !Part->Mount_Options.empty())
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600531 printf(" Mount_Flags: %i, Mount_Options: %s\n", Part->Mount_Flags, Part->Mount_Options.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -0600532 if (Part->MTP_Storage_ID)
533 printf(" MTP_Storage_ID: %i\n", Part->MTP_Storage_ID);
Ethan Yonker93382822018-11-01 15:25:31 -0500534 if (!Part->Key_Directory.empty())
535 printf(" Metadata Key Directory: %s\n", Part->Key_Directory.c_str());
Ethan Yonker6277c792014-09-15 14:54:30 -0500536 printf("\n");
Dees_Troy8170a922012-09-18 15:40:25 -0400537}
538
Dees_Troy51a0e822012-09-05 15:24:24 -0400539int TWPartitionManager::Mount_By_Path(string Path, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400540 std::vector<TWPartition*>::iterator iter;
541 int ret = false;
542 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400543 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400544
Captain Throwbackd54e1da2020-03-04 16:49:51 -0500545 if (Local_Path == "/tmp" || Local_Path == "/" || Local_Path == "/etc")
Dees_Troy43d8b002012-09-17 16:00:01 -0400546 return true;
547
Dees_Troy5bf43922012-09-07 16:07:55 -0400548 // Iterate through all partitions
549 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400550 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400551 ret = (*iter)->Mount(Display_Error);
552 found = true;
Dees_Troy51127312012-09-08 13:08:49 -0400553 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400554 (*iter)->Mount(Display_Error);
Dees_Troy51127312012-09-08 13:08:49 -0400555 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400556 }
557 if (found) {
558 return ret;
559 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500560 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 -0400561 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000562 LOGINFO("Mount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400563 }
564 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400565}
566
Dees_Troy51a0e822012-09-05 15:24:24 -0400567int TWPartitionManager::UnMount_By_Path(string Path, bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400568 std::vector<TWPartition*>::iterator iter;
569 int ret = false;
570 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400571 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy51127312012-09-08 13:08:49 -0400572
573 // Iterate through all partitions
574 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400575 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy51127312012-09-08 13:08:49 -0400576 ret = (*iter)->UnMount(Display_Error);
577 found = true;
578 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
579 (*iter)->UnMount(Display_Error);
580 }
581 }
582 if (found) {
583 return ret;
584 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500585 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 -0400586 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000587 LOGINFO("UnMount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400588 }
589 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400590}
591
Dees_Troy51a0e822012-09-05 15:24:24 -0400592int TWPartitionManager::Is_Mounted_By_Path(string Path) {
Dees_Troy51127312012-09-08 13:08:49 -0400593 TWPartition* Part = Find_Partition_By_Path(Path);
594
595 if (Part)
596 return Part->Is_Mounted();
597 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000598 LOGINFO("Is_Mounted: Unable to find partition for path '%s'\n", Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400599 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400600}
601
Dees_Troy5bf43922012-09-07 16:07:55 -0400602int TWPartitionManager::Mount_Current_Storage(bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400603 string current_storage_path = DataManager::GetCurrentStoragePath();
604
605 if (Mount_By_Path(current_storage_path, Display_Error)) {
606 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
607 if (FreeStorage)
608 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
609 return true;
610 }
611 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400612}
613
Dees_Troy5bf43922012-09-07 16:07:55 -0400614int TWPartitionManager::Mount_Settings_Storage(bool Display_Error) {
615 return Mount_By_Path(DataManager::GetSettingsStoragePath(), Display_Error);
616}
617
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600618TWPartition* TWPartitionManager::Find_Partition_By_Path(const string& Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400619 std::vector<TWPartition*>::iterator iter;
Dees_Troy38bd7602012-09-14 13:33:53 -0400620 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400621
dianlujitaob76a73a2020-04-30 20:33:20 +0800622 if (Local_Path == "/system")
623 Local_Path = Get_Android_Root_Path();
Dees_Troy5bf43922012-09-07 16:07:55 -0400624 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400625 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path))
Dees_Troy5bf43922012-09-07 16:07:55 -0400626 return (*iter);
627 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400628 return NULL;
629}
630
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600631TWPartition* TWPartitionManager::Find_Partition_By_Block_Device(const string& Block_Device) {
632 std::vector<TWPartition*>::iterator iter;
633
634 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
635 if ((*iter)->Primary_Block_Device == Block_Device || (!(*iter)->Actual_Block_Device.empty() && (*iter)->Actual_Block_Device == Block_Device))
636 return (*iter);
637 }
638 return NULL;
639}
640
Ethan Yonker53796e72019-01-11 22:49:52 -0600641int TWPartitionManager::Check_Backup_Name(const std::string& Backup_Name, bool Display_Error, bool Must_Be_Unique) {
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400642 // Check the backup name to ensure that it is the correct size and contains only valid characters
643 // and that a backup with that name doesn't already exist
644 char backup_name[MAX_BACKUP_NAME_LEN];
645 char backup_loc[255], tw_image_dir[255];
646 int copy_size;
647 int index, cur_char;
Ethan Yonker53796e72019-01-11 22:49:52 -0600648 string Backup_Loc;
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400649
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400650 copy_size = Backup_Name.size();
651 // Check size
652 if (copy_size > MAX_BACKUP_NAME_LEN) {
653 if (Display_Error)
Ethan Yonker74db1572015-10-28 12:44:49 -0500654 gui_err("backup_name_len=Backup name is too long.");
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400655 return -2;
656 }
657
658 // Check each character
659 strncpy(backup_name, Backup_Name.c_str(), copy_size);
Dees_Troya13d74f2013-03-24 08:54:55 -0500660 if (copy_size == 1 && strncmp(backup_name, "0", 1) == 0)
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400661 return 0; // A "0" (zero) means to use the current timestamp for the backup name
662 for (index=0; index<copy_size; index++) {
663 cur_char = (int)backup_name[index];
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500664 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 -0400665 // These are valid characters
666 // Numbers
667 // Upper case letters
668 // Lower case letters
669 // Space
670 // and -_.{}[]
671 } else {
672 if (Display_Error)
Ethan Yonker4adc33e2016-01-22 16:07:11 -0600673 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 -0400674 return -3;
675 }
676 }
677
Ethan Yonker53796e72019-01-11 22:49:52 -0600678 if (Must_Be_Unique) {
679 // Check to make sure that a backup with this name doesn't already exist
680 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Loc);
681 strcpy(backup_loc, Backup_Loc.c_str());
682 sprintf(tw_image_dir,"%s/%s", backup_loc, Backup_Name.c_str());
683 if (TWFunc::Path_Exists(tw_image_dir)) {
684 if (Display_Error)
685 gui_err("backup_name_exists=A backup with that name already exists!");
bigbiffce8f83c2015-12-12 18:30:21 -0500686
Ethan Yonker53796e72019-01-11 22:49:52 -0600687 return -4;
688 }
689 // Backup is unique
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400690 }
Ethan Yonker53796e72019-01-11 22:49:52 -0600691 // No problems found
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400692 return 0;
693}
694
bigbiffce8f83c2015-12-12 18:30:21 -0500695bool TWPartitionManager::Backup_Partition(PartitionSettings *part_settings) {
Dees_Troy43d8b002012-09-17 16:00:01 -0400696 time_t start, stop;
Matt Mower23d8aae2017-01-06 14:30:33 -0600697 int use_compression;
bigbiffce8f83c2015-12-12 18:30:21 -0500698 string backup_log = part_settings->Backup_Folder + "/recovery.log";
Dees_Troy43d8b002012-09-17 16:00:01 -0400699
bigbiffce8f83c2015-12-12 18:30:21 -0500700 if (part_settings->Part == NULL)
Dees_Troy43d8b002012-09-17 16:00:01 -0400701 return true;
702
Dees_Troy093b7642012-09-21 15:59:38 -0400703 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400704
Tom Hite5a926722014-09-15 01:31:03 +0000705 TWFunc::SetPerformanceMode(true);
Dees_Troy43d8b002012-09-17 16:00:01 -0400706 time(&start);
707
bigbiffce8f83c2015-12-12 18:30:21 -0500708 if (part_settings->Part->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500709 sync();
710 sync();
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400711 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500712 if (!part_settings->adbbackup && part_settings->generate_digest) {
713 if (!twrpDigestDriver::Make_Digest(Full_Filename))
714 goto backup_error;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400715 }
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400716
bigbiffce8f83c2015-12-12 18:30:21 -0500717 if (part_settings->Part->Has_SubPartition) {
Dees_Troy8170a922012-09-18 15:40:25 -0400718 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400719 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -0400720
721 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400722 if ((*subpart)->Can_Be_Backed_Up && (*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500723 part_settings->Part = *subpart;
bigbiffce8f83c2015-12-12 18:30:21 -0500724 if (!(*subpart)->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500725 goto backup_error;
Tom Hite5a926722014-09-15 01:31:03 +0000726 }
Dees_Troy2727b992013-08-14 20:09:30 +0000727 sync();
728 sync();
bigbiff bigbifff5955b12019-05-18 17:28:58 -0400729 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400730 if (!part_settings->adbbackup && part_settings->generate_digest) {
731 if (!twrpDigestDriver::Make_Digest(Full_Filename)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500732 goto backup_error;
bigbiffce8f83c2015-12-12 18:30:21 -0500733 }
Tom Hite5a926722014-09-15 01:31:03 +0000734 }
Dees_Troy8170a922012-09-18 15:40:25 -0400735 }
736 }
737 }
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400738
Dees_Troy43d8b002012-09-17 16:00:01 -0400739 time(&stop);
that203bcc92015-05-09 17:27:33 +0200740 int backup_time = (int) difftime(stop, start);
Dees_Troy2673cec2013-04-02 20:22:16 +0000741 LOGINFO("Partition Backup time: %d\n", backup_time);
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400742
bigbiffce8f83c2015-12-12 18:30:21 -0500743 if (part_settings->Part->Backup_Method == BM_FILES) {
744 part_settings->file_time += backup_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400745 } else {
bigbiffce8f83c2015-12-12 18:30:21 -0500746 part_settings->img_time += backup_time;
747
Dees_Troy43d8b002012-09-17 16:00:01 -0400748 }
Tom Hite5a926722014-09-15 01:31:03 +0000749
Matt Mower02899552016-12-30 18:13:51 -0600750 TWFunc::SetPerformanceMode(false);
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500751 return true;
Dees_Troy43d8b002012-09-17 16:00:01 -0400752 }
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500753backup_error:
754 Clean_Backup_Folder(part_settings->Backup_Folder);
755 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
756 tw_set_default_metadata(backup_log.c_str());
757 TWFunc::SetPerformanceMode(false);
758 return false;
bigbiff7abc5fe2015-01-17 16:53:12 -0500759}
760
bigbiffbf1d6722015-02-14 16:25:59 -0500761void TWPartitionManager::Clean_Backup_Folder(string Backup_Folder) {
762 DIR *d = opendir(Backup_Folder.c_str());
763 struct dirent *p;
764 int r;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400765 vector<string> ext;
766
767 //extensions we should delete when cleaning
768 ext.push_back("win");
769 ext.push_back("md5");
770 ext.push_back("sha2");
771 ext.push_back("info");
bigbiffbf1d6722015-02-14 16:25:59 -0500772
Ethan Yonker74db1572015-10-28 12:44:49 -0500773 gui_msg("backup_clean=Backup Failed. Cleaning Backup Folder.");
bigbiffbf1d6722015-02-14 16:25:59 -0500774
775 if (d == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500776 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Backup_Folder)(strerror(errno)));
bigbiffbf1d6722015-02-14 16:25:59 -0500777 return;
778 }
779
Matt Mower2b18a532015-02-20 16:58:05 -0600780 while ((p = readdir(d))) {
bigbiffbf1d6722015-02-14 16:25:59 -0500781 if (!strcmp(p->d_name, ".") || !strcmp(p->d_name, ".."))
782 continue;
783
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500784 string path = Backup_Folder + "/" + p->d_name;
bigbiffbf1d6722015-02-14 16:25:59 -0500785
786 size_t dot = path.find_last_of(".") + 1;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400787 for (vector<string>::const_iterator i = ext.begin(); i != ext.end(); ++i) {
788 if (path.substr(dot) == *i) {
789 r = unlink(path.c_str());
790 if (r != 0)
791 LOGINFO("Unable to unlink '%s: %s'\n", path.c_str(), strerror(errno));
bigbiffbf1d6722015-02-14 16:25:59 -0500792 }
793 }
794 }
795 closedir(d);
796}
797
Ethan Yonker472f5062016-02-25 13:47:30 -0600798int TWPartitionManager::Check_Backup_Cancel() {
799 return stop_backup.get_value();
800}
801
bigbiff7abc5fe2015-01-17 16:53:12 -0500802int TWPartitionManager::Cancel_Backup() {
803 string Backup_Folder, Backup_Name, Full_Backup_Path;
804
805 stop_backup.set_value(1);
806
807 if (tar_fork_pid != 0) {
808 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
809 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Folder);
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500810 Full_Backup_Path = Backup_Folder + "/" + Backup_Name;
bigbiff7abc5fe2015-01-17 16:53:12 -0500811 LOGINFO("Killing pid: %d\n", tar_fork_pid);
812 kill(tar_fork_pid, SIGUSR2);
813 while (kill(tar_fork_pid, 0) == 0) {
814 usleep(1000);
815 }
816 LOGINFO("Backup_Run stopped and returning false, backup cancelled.\n");
817 LOGINFO("Removing directory %s\n", Full_Backup_Path.c_str());
818 TWFunc::removeDir(Full_Backup_Path, false);
819 tar_fork_pid = 0;
820 }
821
822 return 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400823}
824
bigbiffce8f83c2015-12-12 18:30:21 -0500825int TWPartitionManager::Run_Backup(bool adbbackup) {
826 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400827 int partition_count = 0, disable_free_space_check = 0, skip_digest = 0;
bigbiffce8f83c2015-12-12 18:30:21 -0500828 string Backup_Name, Backup_List, backup_path;
Matt Mower23d8aae2017-01-06 14:30:33 -0600829 unsigned long long total_bytes = 0, free_space = 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400830 TWPartition* storage = NULL;
Dees_Troy8170a922012-09-18 15:40:25 -0400831 std::vector<TWPartition*>::iterator subpart;
Dees_Troy43d8b002012-09-17 16:00:01 -0400832 struct tm *t;
Matt Mower23d8aae2017-01-06 14:30:33 -0600833 time_t seconds, total_start, total_stop;
Dees_Troya13d74f2013-03-24 08:54:55 -0500834 size_t start_pos = 0, end_pos = 0;
bigbiff7abc5fe2015-01-17 16:53:12 -0500835 stop_backup.set_value(0);
Dees_Troy43d8b002012-09-17 16:00:01 -0400836 seconds = time(0);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500837 t = localtime(&seconds);
Dees_Troy43d8b002012-09-17 16:00:01 -0400838
bigbiffce8f83c2015-12-12 18:30:21 -0500839 part_settings.img_bytes_remaining = 0;
840 part_settings.file_bytes_remaining = 0;
841 part_settings.img_time = 0;
842 part_settings.file_time = 0;
843 part_settings.img_bytes = 0;
844 part_settings.file_bytes = 0;
845 part_settings.PM_Method = PM_BACKUP;
846
bigbiffce8f83c2015-12-12 18:30:21 -0500847 part_settings.adbbackup = adbbackup;
Dees_Troy43d8b002012-09-17 16:00:01 -0400848 time(&total_start);
849
850 Update_System_Details();
851
852 if (!Mount_Current_Storage(true))
853 return false;
854
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400855 DataManager::GetValue(TW_SKIP_DIGEST_GENERATE_VAR, skip_digest);
856 if (skip_digest == 0)
857 part_settings.generate_digest = true;
Dees_Troyc5865ab2012-09-24 15:08:04 -0400858 else
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400859 part_settings.generate_digest = false;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400860
bigbiffce8f83c2015-12-12 18:30:21 -0500861 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, part_settings.Backup_Folder);
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500862 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
863 if (Backup_Name == gui_lookup("curr_date", "(Current Date)")) {
864 Backup_Name = TWFunc::Get_Current_Date();
865 } else if (Backup_Name == gui_lookup("auto_generate", "(Auto Generate)") || Backup_Name == "0" || Backup_Name.empty()) {
Dees Troyb21cc642013-09-10 17:36:41 +0000866 TWFunc::Auto_Generate_Backup_Name();
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500867 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
Dees_Troy43d8b002012-09-17 16:00:01 -0400868 }
bigbiffce8f83c2015-12-12 18:30:21 -0500869
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500870 LOGINFO("Backup Name is: '%s'\n", Backup_Name.c_str());
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500871 part_settings.Backup_Folder = part_settings.Backup_Folder + "/" + Backup_Name;
bigbiffce8f83c2015-12-12 18:30:21 -0500872
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500873 LOGINFO("Backup_Folder is: '%s'\n", part_settings.Backup_Folder.c_str());
Dees_Troy43d8b002012-09-17 16:00:01 -0400874
Dees_Troy2673cec2013-04-02 20:22:16 +0000875 LOGINFO("Calculating backup details...\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500876 DataManager::GetValue("tw_backup_list", Backup_List);
877 if (!Backup_List.empty()) {
878 end_pos = Backup_List.find(";", start_pos);
879 while (end_pos != string::npos && start_pos < Backup_List.size()) {
880 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -0500881 part_settings.Part = Find_Partition_By_Path(backup_path);
882 if (part_settings.Part != NULL) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500883 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500884 if (part_settings.Part->Backup_Method == BM_FILES)
885 part_settings.file_bytes += part_settings.Part->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500886 else
bigbiffce8f83c2015-12-12 18:30:21 -0500887 part_settings.img_bytes += part_settings.Part->Backup_Size;
888 if (part_settings.Part->Has_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500889 std::vector<TWPartition*>::iterator subpart;
890
891 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiffce8f83c2015-12-12 18:30:21 -0500892 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 -0500893 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500894 if ((*subpart)->Backup_Method == BM_FILES)
895 part_settings.file_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500896 else
bigbiffce8f83c2015-12-12 18:30:21 -0500897 part_settings.img_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500898 }
899 }
Dees_Troy8170a922012-09-18 15:40:25 -0400900 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500901 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500902 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 -0400903 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500904 start_pos = end_pos + 1;
905 end_pos = Backup_List.find(";", start_pos);
Dees_Troy43d8b002012-09-17 16:00:01 -0400906 }
907 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400908
909 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500910 gui_msg("no_partition_selected=No partitions selected for backup.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400911 return false;
912 }
bigbiffce8f83c2015-12-12 18:30:21 -0500913 if (adbbackup) {
914 if (twadbbu::Write_ADB_Stream_Header(partition_count) == false) {
915 return false;
916 }
917 }
918 total_bytes = part_settings.file_bytes + part_settings.img_bytes;
Ethan Yonker472f5062016-02-25 13:47:30 -0600919 ProgressTracking progress(total_bytes);
bigbiffce8f83c2015-12-12 18:30:21 -0500920 part_settings.progress = &progress;
921
Ethan Yonker74db1572015-10-28 12:44:49 -0500922 gui_msg(Msg("total_partitions_backup= * Total number of partitions to back up: {1}")(partition_count));
923 gui_msg(Msg("total_backup_size= * Total size of all data: {1}MB")(total_bytes / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400924 storage = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
925 if (storage != NULL) {
926 free_space = storage->Free;
Ethan Yonker74db1572015-10-28 12:44:49 -0500927 gui_msg(Msg("available_space= * Available space: {1}MB")(free_space / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400928 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500929 gui_err("unable_locate_storage=Unable to locate storage device.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400930 return false;
931 }
bigbiffbf1d6722015-02-14 16:25:59 -0500932
Matt Mowerbfccfb82016-04-25 23:22:31 -0500933 DataManager::GetValue(TW_DISABLE_FREE_SPACE_VAR, disable_free_space_check);
bigbiffce8f83c2015-12-12 18:30:21 -0500934
935 if (adbbackup)
936 disable_free_space_check = true;
937
bigbiffbf1d6722015-02-14 16:25:59 -0500938 if (!disable_free_space_check) {
939 if (free_space - (32 * 1024 * 1024) < total_bytes) {
940 // We require an extra 32MB just in case
Ethan Yonker74db1572015-10-28 12:44:49 -0500941 gui_err("no_space=Not enough free space on storage.");
bigbiffbf1d6722015-02-14 16:25:59 -0500942 return false;
943 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400944 }
bigbiffce8f83c2015-12-12 18:30:21 -0500945 part_settings.img_bytes_remaining = part_settings.img_bytes;
946 part_settings.file_bytes_remaining = part_settings.file_bytes;
Dees_Troy43d8b002012-09-17 16:00:01 -0400947
Ethan Yonker74db1572015-10-28 12:44:49 -0500948 gui_msg("backup_started=[BACKUP STARTED]");
bigbiff bigbiffcdd97c72019-03-29 19:12:33 -0400949
950 int is_decrypted = 0;
951 int is_encrypted = 0;
952
953 DataManager::GetValue(TW_IS_DECRYPTED, is_decrypted);
954 DataManager::GetValue(TW_IS_ENCRYPTED, is_encrypted);
955 if (!adbbackup || (!is_encrypted || (is_encrypted && is_decrypted))) {
956 gui_msg(Msg("backup_folder= * Backup Folder: {1}")(part_settings.Backup_Folder));
957 if (!TWFunc::Recursive_Mkdir(part_settings.Backup_Folder)) {
958 gui_err("fail_backup_folder=Failed to make backup folder.");
959 return false;
960 }
Dees_Troyd4b22b02013-01-18 17:17:58 +0000961 }
962
Dees_Troy2673cec2013-04-02 20:22:16 +0000963 DataManager::SetProgress(0.0);
Dees_Troy093b7642012-09-21 15:59:38 -0400964
Dees_Troya13d74f2013-03-24 08:54:55 -0500965 start_pos = 0;
966 end_pos = Backup_List.find(";", start_pos);
967 while (end_pos != string::npos && start_pos < Backup_List.size()) {
bigbiff7abc5fe2015-01-17 16:53:12 -0500968 if (stop_backup.get_value() != 0)
969 return -1;
Dees_Troya13d74f2013-03-24 08:54:55 -0500970 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -0500971 part_settings.Part = Find_Partition_By_Path(backup_path);
972 if (part_settings.Part != NULL) {
973 if (!Backup_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -0500974 return false;
975 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500976 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 -0500977 }
978 start_pos = end_pos + 1;
979 end_pos = Backup_List.find(";", start_pos);
980 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400981
982 // Average BPS
bigbiffce8f83c2015-12-12 18:30:21 -0500983 if (part_settings.img_time == 0)
984 part_settings.img_time = 1;
985 if (part_settings.file_time == 0)
986 part_settings.file_time = 1;
987 int img_bps = (int)part_settings.img_bytes / (int)part_settings.img_time;
988 unsigned long long file_bps = part_settings.file_bytes / (int)part_settings.file_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400989
bigbiffce8f83c2015-12-12 18:30:21 -0500990 if (part_settings.file_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -0600991 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 -0500992 if (part_settings.img_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -0600993 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 -0400994
995 time(&total_stop);
996 int total_time = (int) difftime(total_stop, total_start);
bigbiffce8f83c2015-12-12 18:30:21 -0500997
998 uint64_t actual_backup_size;
Ethan Yonker3fdcda42016-11-30 12:29:37 -0600999 if (!adbbackup) {
1000 TWExclude twe;
1001 actual_backup_size = twe.Get_Folder_Size(part_settings.Backup_Folder);
1002 } else
bigbiffce8f83c2015-12-12 18:30:21 -05001003 actual_backup_size = part_settings.file_bytes + part_settings.img_bytes;
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001004 actual_backup_size /= (1024LLU * 1024LLU);
Dees_Troy43d8b002012-09-17 16:00:01 -04001005
bigbiffce8f83c2015-12-12 18:30:21 -05001006 int prev_img_bps = 0, use_compression = 0;
1007 unsigned long long prev_file_bps = 0;
Matt Mower173cdb92016-12-31 02:49:19 -06001008 DataManager::GetValue(TW_BACKUP_AVG_IMG_RATE, prev_img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -04001009 img_bps += (prev_img_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001010 img_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -04001011
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001012 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -04001013 if (use_compression)
1014 DataManager::GetValue(TW_BACKUP_AVG_FILE_COMP_RATE, prev_file_bps);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001015 else
Dees_Troy093b7642012-09-21 15:59:38 -04001016 DataManager::GetValue(TW_BACKUP_AVG_FILE_RATE, prev_file_bps);
1017 file_bps += (prev_file_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001018 file_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -04001019
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001020 DataManager::SetValue(TW_BACKUP_AVG_IMG_RATE, img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -04001021 if (use_compression)
1022 DataManager::SetValue(TW_BACKUP_AVG_FILE_COMP_RATE, file_bps);
1023 else
1024 DataManager::SetValue(TW_BACKUP_AVG_FILE_RATE, file_bps);
1025
Ethan Yonker74db1572015-10-28 12:44:49 -05001026 gui_msg(Msg("total_backed_size=[{1} MB TOTAL BACKED UP]")(actual_backup_size));
Dees_Troy43d8b002012-09-17 16:00:01 -04001027 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001028 UnMount_Main_Partitions();
Ethan Yonker56db1c42015-12-20 22:49:00 -06001029 gui_msg(Msg(msg::kHighlight, "backup_completed=[BACKUP COMPLETED IN {1} SECONDS]")(total_time)); // the end
Ethan Yonkere080c1f2016-09-19 13:50:25 -05001030 string backup_log = part_settings.Backup_Folder + "/recovery.log";
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001031 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
Ethan Yonker4b94cfd2014-12-11 10:00:45 -06001032 tw_set_default_metadata(backup_log.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -05001033
1034 if (part_settings.adbbackup) {
1035 if (twadbbu::Write_ADB_Stream_Trailer() == false) {
1036 return false;
1037 }
1038 }
1039 part_settings.adbbackup = false;
1040 DataManager::SetValue("tw_enable_adb_backup", 0);
1041
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001042 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001043}
1044
bigbiffce8f83c2015-12-12 18:30:21 -05001045bool TWPartitionManager::Restore_Partition(PartitionSettings *part_settings) {
Dees_Troy4a2a1262012-09-18 09:33:47 -04001046 time_t Start, Stop;
bigbiffce8f83c2015-12-12 18:30:21 -05001047
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001048 if (part_settings->adbbackup) {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001049 std::string partName = part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win";
1050 LOGINFO("setting backup name: %s\n", partName.c_str());
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001051 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
1052 }
1053
Tom Hite5a926722014-09-15 01:31:03 +00001054 TWFunc::SetPerformanceMode(true);
bigbiffce8f83c2015-12-12 18:30:21 -05001055
Dees_Troy4a2a1262012-09-18 09:33:47 -04001056 time(&Start);
Ethan Yonker472f5062016-02-25 13:47:30 -06001057
bigbiffce8f83c2015-12-12 18:30:21 -05001058 if (!part_settings->Part->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001059 TWFunc::SetPerformanceMode(false);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001060 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001061 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001062 if (part_settings->Part->Has_SubPartition && !part_settings->adbbackup) {
Dees_Troy8170a922012-09-18 15:40:25 -04001063 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001064 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -04001065
1066 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001067 part_settings->Part = *subpart;
1068 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
1069 part_settings->Part = (*subpart);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001070 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
bigbiffce8f83c2015-12-12 18:30:21 -05001071 if (!(*subpart)->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001072 TWFunc::SetPerformanceMode(false);
Dees_Troy8170a922012-09-18 15:40:25 -04001073 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001074 }
Dees_Troy8170a922012-09-18 15:40:25 -04001075 }
1076 }
1077 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001078 time(&Stop);
Tom Hite5a926722014-09-15 01:31:03 +00001079 TWFunc::SetPerformanceMode(false);
bigbiffce8f83c2015-12-12 18:30:21 -05001080 gui_msg(Msg("restore_part_done=[{1} done ({2} seconds)]")(part_settings->Part->Backup_Display_Name)((int)difftime(Stop, Start)));
1081
Dees_Troy4a2a1262012-09-18 09:33:47 -04001082 return true;
1083}
1084
Ethan Yonker472f5062016-02-25 13:47:30 -06001085int TWPartitionManager::Run_Restore(const string& Restore_Name) {
bigbiffce8f83c2015-12-12 18:30:21 -05001086 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001087 int check_digest;
bigbiffce8f83c2015-12-12 18:30:21 -05001088
Dees_Troy4a2a1262012-09-18 09:33:47 -04001089 time_t rStart, rStop;
1090 time(&rStart);
Dees_Troya13d74f2013-03-24 08:54:55 -05001091 string Restore_List, restore_path;
1092 size_t start_pos = 0, end_pos;
bigbiffce8f83c2015-12-12 18:30:21 -05001093
Ethan Yonkerdcf2b672016-09-13 14:41:53 -05001094 part_settings.Backup_Folder = Restore_Name;
bigbiffce8f83c2015-12-12 18:30:21 -05001095 part_settings.Part = NULL;
1096 part_settings.partition_count = 0;
1097 part_settings.total_restore_size = 0;
1098 part_settings.adbbackup = false;
1099 part_settings.PM_Method = PM_RESTORE;
Dees_Troy43d8b002012-09-17 16:00:01 -04001100
Ethan Yonker74db1572015-10-28 12:44:49 -05001101 gui_msg("restore_started=[RESTORE STARTED]");
1102 gui_msg(Msg("restore_folder=Restore folder: '{1}'")(Restore_Name));
Dees_Troy43d8b002012-09-17 16:00:01 -04001103
Dees_Troy4a2a1262012-09-18 09:33:47 -04001104 if (!Mount_Current_Storage(true))
1105 return false;
1106
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001107 DataManager::GetValue(TW_SKIP_DIGEST_CHECK_VAR, check_digest);
1108 if (check_digest > 0) {
1109 // Check Digest files first before restoring to ensure that all of them match before starting a restore
1110 TWFunc::GUI_Operation_Text(TW_VERIFY_DIGEST_TEXT, gui_parse_text("{@verifying_digest}"));
1111 gui_msg("verifying_digest=Verifying Digest");
Dees_Troya13d74f2013-03-24 08:54:55 -05001112 } else {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001113 gui_msg("skip_digest=Skipping Digest check based on user setting.");
Dees_Troya13d74f2013-03-24 08:54:55 -05001114 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001115 gui_msg("calc_restore=Calculating restore details...");
Dees_Troya13d74f2013-03-24 08:54:55 -05001116 DataManager::GetValue("tw_restore_selected", Restore_List);
bigbiffce8f83c2015-12-12 18:30:21 -05001117
Dees_Troya13d74f2013-03-24 08:54:55 -05001118 if (!Restore_List.empty()) {
1119 end_pos = Restore_List.find(";", start_pos);
1120 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1121 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001122 part_settings.Part = Find_Partition_By_Path(restore_path);
1123 if (part_settings.Part != NULL) {
bigbiffce8f83c2015-12-12 18:30:21 -05001124 if (part_settings.Part->Mount_Read_Only) {
1125 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 -05001126 return false;
1127 }
bigbiffce8f83c2015-12-12 18:30:21 -05001128
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001129 string Full_Filename = part_settings.Backup_Folder + "/" + part_settings.Part->Backup_FileName;
1130
bigbiff35d2bfd2021-01-23 14:08:03 -05001131 if (tw_get_default_metadata(Get_Android_Root_Path().c_str()) != 0) {
1132 gui_msg(Msg(msg::kWarning, "restore_system_context=Unable to get default context for {1} -- Android may not boot.")(Get_Android_Root_Path()));
1133 }
1134
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001135 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001136 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001137 part_settings.partition_count++;
1138 part_settings.total_restore_size += part_settings.Part->Get_Restore_Size(&part_settings);
1139 if (part_settings.Part->Has_SubPartition) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001140 TWPartition *parentPart = part_settings.Part;
Dees_Troya13d74f2013-03-24 08:54:55 -05001141 std::vector<TWPartition*>::iterator subpart;
1142
1143 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001144 part_settings.Part = *subpart;
1145 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001146 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001147 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001148 part_settings.total_restore_size += (*subpart)->Get_Restore_Size(&part_settings);
Dees_Troya13d74f2013-03-24 08:54:55 -05001149 }
1150 }
1151 }
1152 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001153 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001154 }
1155 start_pos = end_pos + 1;
1156 end_pos = Restore_List.find(";", start_pos);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001157 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001158 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001159
bigbiffce8f83c2015-12-12 18:30:21 -05001160 if (part_settings.partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001161 gui_err("no_part_restore=No partitions selected for restore.");
Dees_Troy4a2a1262012-09-18 09:33:47 -04001162 return false;
1163 }
1164
bigbiffce8f83c2015-12-12 18:30:21 -05001165 gui_msg(Msg("restore_part_count=Restoring {1} partitions...")(part_settings.partition_count));
1166 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 +00001167 DataManager::SetProgress(0.0);
bigbiffce8f83c2015-12-12 18:30:21 -05001168 ProgressTracking progress(part_settings.total_restore_size);
1169 part_settings.progress = &progress;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001170
Dees_Troya13d74f2013-03-24 08:54:55 -05001171 start_pos = 0;
1172 if (!Restore_List.empty()) {
1173 end_pos = Restore_List.find(";", start_pos);
1174 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1175 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001176
1177 part_settings.Part = Find_Partition_By_Path(restore_path);
1178 if (part_settings.Part != NULL) {
1179 part_settings.partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -05001180 if (!Restore_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001181 return false;
1182 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001183 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001184 }
1185 start_pos = end_pos + 1;
1186 end_pos = Restore_List.find(";", start_pos);
1187 }
1188 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001189 TWFunc::GUI_Operation_Text(TW_UPDATE_SYSTEM_DETAILS_TEXT, gui_parse_text("{@updating_system_details}"));
bigbiff35d2bfd2021-01-23 14:08:03 -05001190 tw_set_default_metadata(Get_Android_Root_Path().c_str());
Captain Throwback9d6feb52018-07-27 10:05:24 -04001191 UnMount_By_Path(Get_Android_Root_Path(), false);
Dees_Troy43d8b002012-09-17 16:00:01 -04001192 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001193 UnMount_Main_Partitions();
Dees_Troy4a2a1262012-09-18 09:33:47 -04001194 time(&rStop);
Matt Mower3c366972015-12-25 19:28:31 -06001195 gui_msg(Msg(msg::kHighlight, "restore_completed=[RESTORE COMPLETED IN {1} SECONDS]")((int)difftime(rStop,rStart)));
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001196 DataManager::SetValue("tw_file_progress", "");
bigbiffce8f83c2015-12-12 18:30:21 -05001197
Dees_Troy63c8df72012-09-10 14:02:05 -04001198 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001199}
1200
1201void TWPartitionManager::Set_Restore_Files(string Restore_Name) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001202 // Start with the default values
Dees_Troya13d74f2013-03-24 08:54:55 -05001203 string Restore_List;
Dees_Troy83bd4832013-05-04 12:39:56 +00001204 bool get_date = true, check_encryption = true;
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001205 bool adbbackup = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001206
1207 DataManager::SetValue("tw_restore_encrypted", 0);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001208 if (twadbbu::Check_ADB_Backup_File(Restore_Name)) {
1209 vector<string> adb_files;
1210 adb_files = twadbbu::Get_ADB_Backup_Files(Restore_Name);
1211 for (unsigned int i = 0; i < adb_files.size(); ++i) {
1212 string adb_restore_file = adb_files.at(i);
1213 std::size_t pos = adb_restore_file.find_first_of(".");
1214 std::string path = "/" + adb_restore_file.substr(0, pos);
1215 Restore_List = path + ";";
1216 TWPartition* Part = Find_Partition_By_Path(path);
1217 Part->Backup_FileName = TWFunc::Get_Filename(adb_restore_file);
1218 adbbackup = true;
1219 }
1220 DataManager::SetValue("tw_enable_adb_backup", 1);
Dees_Troy63c8df72012-09-10 14:02:05 -04001221 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001222 else {
1223 DIR* d;
1224 d = opendir(Restore_Name.c_str());
1225 if (d == NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001226 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001227 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Restore_Name)(strerror(errno)));
1228 return;
Dees_Troy63c8df72012-09-10 14:02:05 -04001229 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001230
1231 struct dirent* de;
1232 while ((de = readdir(d)) != NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001233 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001234 // Strip off three components
1235 char str[256];
1236 char* label;
1237 char* fstype = NULL;
1238 char* extn = NULL;
1239 char* ptr;
Dees_Troy63c8df72012-09-10 14:02:05 -04001240
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001241 strcpy(str, de->d_name);
1242 if (strlen(str) <= 2)
1243 continue;
Dees_Troy83bd4832013-05-04 12:39:56 +00001244
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001245 if (get_date) {
1246 char file_path[255];
1247 struct stat st;
1248
1249 strcpy(file_path, Restore_Name.c_str());
1250 strcat(file_path, "/");
1251 strcat(file_path, str);
1252 stat(file_path, &st);
1253 string backup_date = ctime((const time_t*)(&st.st_mtime));
1254 DataManager::SetValue(TW_RESTORE_FILE_DATE, backup_date);
1255 get_date = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001256 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001257
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001258 label = str;
1259 ptr = label;
1260 while (*ptr && *ptr != '.') ptr++;
1261 if (*ptr == '.')
1262 {
1263 *ptr = 0x00;
1264 ptr++;
1265 fstype = ptr;
1266 }
1267 while (*ptr && *ptr != '.') ptr++;
1268 if (*ptr == '.')
1269 {
1270 *ptr = 0x00;
1271 ptr++;
1272 extn = ptr;
1273 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001274
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001275 if (fstype == NULL || extn == NULL || strcmp(fstype, "log") == 0) continue;
1276 int extnlength = strlen(extn);
1277 if (extnlength != 3 && extnlength != 6) continue;
1278 if (extnlength >= 3 && strncmp(extn, "win", 3) != 0) continue;
1279 //if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
Dees_Troy63c8df72012-09-10 14:02:05 -04001280
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001281 if (check_encryption) {
1282 string filename = Restore_Name + "/";
1283 filename += de->d_name;
1284 if (TWFunc::Get_File_Type(filename) == 2) {
1285 LOGINFO("'%s' is encrypted\n", filename.c_str());
1286 DataManager::SetValue("tw_restore_encrypted", 1);
1287 }
1288 }
1289 if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
1290
1291 TWPartition* Part = Find_Partition_By_Path(label);
1292 if (Part == NULL)
1293 {
1294 gui_msg(Msg(msg::kError, "unable_locate_part_backup_name=Unable to locate partition by backup name: '{1}'")(label));
1295 continue;
1296 }
1297
1298 Part->Backup_FileName = de->d_name;
1299 if (strlen(extn) > 3) {
1300 Part->Backup_FileName.resize(Part->Backup_FileName.size() - strlen(extn) + 3);
1301 }
1302
dianlujitaob76a73a2020-04-30 20:33:20 +08001303 if (!Part->Is_SubPartition) {
1304 if (Part->Backup_Path == Get_Android_Root_Path())
1305 Restore_List += "/system;";
1306 else
1307 Restore_List += Part->Backup_Path + ";";
1308 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001309 }
1310 closedir(d);
Dees_Troy63c8df72012-09-10 14:02:05 -04001311 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001312
1313 if (adbbackup) {
1314 Restore_List = "ADB_Backup;";
1315 adbbackup = false;
1316 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001317
Dees_Troya13d74f2013-03-24 08:54:55 -05001318 // Set the final value
1319 DataManager::SetValue("tw_restore_list", Restore_List);
1320 DataManager::SetValue("tw_restore_selected", Restore_List);
Dees_Troy51a0e822012-09-05 15:24:24 -04001321 return;
1322}
1323
1324int TWPartitionManager::Wipe_By_Path(string Path) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001325 std::vector<TWPartition*>::iterator iter;
Mohd Faraz77a31912020-04-21 18:56:44 +05301326 std::vector < TWPartition * >::iterator iter1;
Dees_Troy63c8df72012-09-10 14:02:05 -04001327 int ret = false;
1328 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -04001329 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy63c8df72012-09-10 14:02:05 -04001330
dianlujitaob76a73a2020-04-30 20:33:20 +08001331 if (Local_Path == "/system")
1332 Local_Path = Get_Android_Root_Path();
Dees_Troy63c8df72012-09-10 14:02:05 -04001333 // Iterate through all partitions
1334 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -04001335 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Mohd Faraz77a31912020-04-21 18:56:44 +05301336 // iterate through all partitions since some legacy devices uses other partitions as vendor causes issues while wiping
1337 (*iter)->Find_Actual_Block_Device();
1338 for (iter1 = Partitions.begin (); iter1 != Partitions.end (); iter1++)
1339 {
1340 (*iter1)->Find_Actual_Block_Device();
1341 if ((*iter)->Actual_Block_Device == (*iter1)->Actual_Block_Device && (*iter)->Mount_Point != (*iter1)->Mount_Point)
1342 (*iter1)->UnMount(false);
1343 }
Dees_Troye58d5262012-09-21 12:27:57 -04001344 if (Path == "/and-sec")
1345 ret = (*iter)->Wipe_AndSec();
1346 else
1347 ret = (*iter)->Wipe();
Dees_Troy63c8df72012-09-10 14:02:05 -04001348 found = true;
1349 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1350 (*iter)->Wipe();
1351 }
1352 }
1353 if (found) {
1354 return ret;
1355 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001356 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 -04001357 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -04001358}
1359
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001360int TWPartitionManager::Wipe_By_Path(string Path, string New_File_System) {
1361 std::vector<TWPartition*>::iterator iter;
1362 int ret = false;
1363 bool found = false;
1364 string Local_Path = TWFunc::Get_Root_Path(Path);
1365
1366 // Iterate through all partitions
1367 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1368 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1369 if (Path == "/and-sec")
1370 ret = (*iter)->Wipe_AndSec();
1371 else
1372 ret = (*iter)->Wipe(New_File_System);
1373 found = true;
1374 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1375 (*iter)->Wipe(New_File_System);
1376 }
1377 }
1378 if (found) {
1379 return ret;
1380 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001381 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 -05001382 return false;
1383}
1384
Dees_Troy51a0e822012-09-05 15:24:24 -04001385int TWPartitionManager::Factory_Reset(void) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001386 std::vector<TWPartition*>::iterator iter;
1387 int ret = true;
1388
1389 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001390 if ((*iter)->Wipe_During_Factory_Reset && (*iter)->Is_Present) {
Ethan Yonker89583ef2015-08-26 09:01:59 -05001391#ifdef TW_OEM_BUILD
1392 if ((*iter)->Mount_Point == "/data") {
1393 if (!(*iter)->Wipe_Encryption())
1394 ret = false;
1395 } else {
1396#endif
1397 if (!(*iter)->Wipe())
1398 ret = false;
1399#ifdef TW_OEM_BUILD
1400 }
1401#endif
Dees_Troy094207a2012-09-26 12:00:39 -04001402 } else if ((*iter)->Has_Android_Secure) {
1403 if (!(*iter)->Wipe_AndSec())
1404 ret = false;
Dees_Troy63c8df72012-09-10 14:02:05 -04001405 }
1406 }
Ethan Yonker9fa76ba2016-06-30 14:05:43 -05001407 TWFunc::check_and_run_script("/sbin/factoryreset.sh", "Factory Reset Script");
Dees_Troy63c8df72012-09-10 14:02:05 -04001408 return ret;
Dees_Troy51a0e822012-09-05 15:24:24 -04001409}
1410
Dees_Troy38bd7602012-09-14 13:33:53 -04001411int TWPartitionManager::Wipe_Dalvik_Cache(void) {
1412 struct stat st;
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001413 vector <string> dir;
Dees_Troy38bd7602012-09-14 13:33:53 -04001414
1415 if (!Mount_By_Path("/data", true))
1416 return false;
1417
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001418 dir.push_back("/data/dalvik-cache");
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001419
bigbiff349ea552020-06-19 16:07:38 -04001420 std::string cacheDir = TWFunc::get_log_dir();
1421 if (cacheDir == CACHE_LOGS_DIR) {
1422 if (!PartitionManager.Mount_By_Path(CACHE_LOGS_DIR, false)) {
1423 LOGINFO("Unable to mount %s for wiping cache.\n", CACHE_LOGS_DIR);
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001424 }
1425 dir.push_back(cacheDir + "dalvik-cache");
1426 dir.push_back(cacheDir + "/dc");
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001427 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001428
Dees_Troy38bd7602012-09-14 13:33:53 -04001429 TWPartition* sdext = Find_Partition_By_Path("/sd-ext");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001430 if (sdext && sdext->Is_Present && sdext->Mount(false))
1431 {
1432 if (stat("/sd-ext/dalvik-cache", &st) == 0)
1433 {
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001434 dir.push_back("/sd-ext/dalvik-cache");
1435 }
1436 }
1437
bigbiff349ea552020-06-19 16:07:38 -04001438 if (cacheDir == CACHE_LOGS_DIR) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001439 gui_msg("wiping_cache_dalvik=Wiping Dalvik Cache Directories...");
1440 } else {
1441 gui_msg("wiping_dalvik=Wiping Dalvik Directory...");
1442 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001443 for (unsigned i = 0; i < dir.size(); ++i) {
1444 if (stat(dir.at(i).c_str(), &st) == 0) {
1445 TWFunc::removeDir(dir.at(i), false);
1446 gui_msg(Msg("cleaned=Cleaned: {1}...")(dir.at(i)));
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001447 }
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001448 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001449
bigbiff349ea552020-06-19 16:07:38 -04001450 if (cacheDir == CACHE_LOGS_DIR) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001451 gui_msg("cache_dalvik_done=-- Dalvik Cache Directories Wipe Complete!");
1452 } else {
1453 gui_msg("dalvik_done=-- Dalvik Directory Wipe Complete!");
1454 }
1455
Dees_Troy38bd7602012-09-14 13:33:53 -04001456 return true;
1457}
1458
1459int TWPartitionManager::Wipe_Rotate_Data(void) {
1460 if (!Mount_By_Path("/data", true))
1461 return false;
1462
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001463 unlink("/data/misc/akmd*");
1464 unlink("/data/misc/rild*");
Dees_Troy2673cec2013-04-02 20:22:16 +00001465 gui_print("Rotation data wiped.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001466 return true;
1467}
1468
1469int TWPartitionManager::Wipe_Battery_Stats(void) {
1470 struct stat st;
1471
1472 if (!Mount_By_Path("/data", true))
1473 return false;
1474
1475 if (0 != stat("/data/system/batterystats.bin", &st)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001476 gui_print("No Battery Stats Found. No Need To Wipe.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001477 } else {
1478 remove("/data/system/batterystats.bin");
Dees_Troy2673cec2013-04-02 20:22:16 +00001479 gui_print("Cleared battery stats.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001480 }
1481 return true;
1482}
1483
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001484int TWPartitionManager::Wipe_Android_Secure(void) {
1485 std::vector<TWPartition*>::iterator iter;
1486 int ret = false;
1487 bool found = false;
1488
1489 // Iterate through all partitions
1490 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1491 if ((*iter)->Has_Android_Secure) {
1492 ret = (*iter)->Wipe_AndSec();
1493 found = true;
1494 }
1495 }
1496 if (found) {
1497 return ret;
1498 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001499 gui_err("no_andsec=No android secure partitions found.");
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001500 }
1501 return false;
1502}
1503
Dees_Troy38bd7602012-09-14 13:33:53 -04001504int TWPartitionManager::Format_Data(void) {
1505 TWPartition* dat = Find_Partition_By_Path("/data");
1506
1507 if (dat != NULL) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001508 return dat->Wipe_Encryption();
1509 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001510 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001511 return false;
1512 }
1513 return false;
1514}
1515
1516int TWPartitionManager::Wipe_Media_From_Data(void) {
1517 TWPartition* dat = Find_Partition_By_Path("/data");
1518
1519 if (dat != NULL) {
1520 if (!dat->Has_Data_Media) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001521 LOGERR("This device does not have /data/media\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001522 return false;
1523 }
1524 if (!dat->Mount(true))
1525 return false;
1526
Ethan Yonker74db1572015-10-28 12:44:49 -05001527 gui_msg("wiping_datamedia=Wiping internal storage -- /data/media...");
Ethan Yonker726a0202014-12-16 20:01:38 -06001528 Remove_MTP_Storage(dat->MTP_Storage_ID);
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001529 TWFunc::removeDir("/data/media", false);
xiaolu9416f4f2015-06-04 08:22:23 +08001530 dat->Recreate_Media_Folder();
Ethan Yonker726a0202014-12-16 20:01:38 -06001531 Add_MTP_Storage(dat->MTP_Storage_ID);
Dees_Troy38bd7602012-09-14 13:33:53 -04001532 return true;
1533 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001534 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001535 return false;
1536 }
1537 return false;
1538}
1539
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001540int TWPartitionManager::Repair_By_Path(string Path, bool Display_Error) {
1541 std::vector<TWPartition*>::iterator iter;
1542 int ret = false;
1543 bool found = false;
1544 string Local_Path = TWFunc::Get_Root_Path(Path);
1545
1546 if (Local_Path == "/tmp" || Local_Path == "/")
1547 return true;
1548
1549 // Iterate through all partitions
1550 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1551 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1552 ret = (*iter)->Repair();
1553 found = true;
1554 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1555 (*iter)->Repair();
1556 }
1557 }
1558 if (found) {
1559 return ret;
1560 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001561 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 -05001562 } else {
1563 LOGINFO("Repair: Unable to find partition for path '%s'\n", Local_Path.c_str());
1564 }
1565 return false;
1566}
1567
Ethan Yonkera2719152015-05-28 09:44:41 -05001568int TWPartitionManager::Resize_By_Path(string Path, bool Display_Error) {
1569 std::vector<TWPartition*>::iterator iter;
1570 int ret = false;
1571 bool found = false;
1572 string Local_Path = TWFunc::Get_Root_Path(Path);
1573
1574 if (Local_Path == "/tmp" || Local_Path == "/")
1575 return true;
1576
1577 // Iterate through all partitions
1578 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1579 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1580 ret = (*iter)->Resize();
1581 found = true;
1582 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1583 (*iter)->Resize();
1584 }
1585 }
1586 if (found) {
1587 return ret;
1588 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001589 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 -05001590 } else {
1591 LOGINFO("Resize: Unable to find partition for path '%s'\n", Local_Path.c_str());
1592 }
1593 return false;
1594}
1595
Dees_Troy51a0e822012-09-05 15:24:24 -04001596void TWPartitionManager::Update_System_Details(void) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001597 std::vector<TWPartition*>::iterator iter;
Dees_Troy51127312012-09-08 13:08:49 -04001598 int data_size = 0;
Dees_Troy5bf43922012-09-07 16:07:55 -04001599
Ethan Yonker74db1572015-10-28 12:44:49 -05001600 gui_msg("update_part_details=Updating partition details...");
Dees_Troy5bf43922012-09-07 16:07:55 -04001601 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker1b190162016-12-05 15:25:19 -06001602 (*iter)->Update_Size(true);
Dees_Troy51127312012-09-08 13:08:49 -04001603 if ((*iter)->Can_Be_Mounted) {
Captain Throwback9d6feb52018-07-27 10:05:24 -04001604 if ((*iter)->Mount_Point == Get_Android_Root_Path()) {
Dees_Troy51127312012-09-08 13:08:49 -04001605 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1606 DataManager::SetValue(TW_BACKUP_SYSTEM_SIZE, backup_display_size);
Ethan Yonker76bbd3a2019-05-10 10:50:04 -05001607 TWFunc::Is_TWRP_App_In_System();
Dees_Troy51127312012-09-08 13:08:49 -04001608 } else if ((*iter)->Mount_Point == "/data" || (*iter)->Mount_Point == "/datadata") {
1609 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
1610 } else if ((*iter)->Mount_Point == "/cache") {
1611 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1612 DataManager::SetValue(TW_BACKUP_CACHE_SIZE, backup_display_size);
1613 } else if ((*iter)->Mount_Point == "/sd-ext") {
1614 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1615 DataManager::SetValue(TW_BACKUP_SDEXT_SIZE, backup_display_size);
1616 if ((*iter)->Backup_Size == 0) {
1617 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 0);
1618 DataManager::SetValue(TW_BACKUP_SDEXT_VAR, 0);
1619 } else
1620 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 1);
Dees_Troye58d5262012-09-21 12:27:57 -04001621 } else if ((*iter)->Has_Android_Secure) {
Dees_Troy8170a922012-09-18 15:40:25 -04001622 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troye58d5262012-09-21 12:27:57 -04001623 DataManager::SetValue(TW_BACKUP_ANDSEC_SIZE, backup_display_size);
Dees_Troy8170a922012-09-18 15:40:25 -04001624 if ((*iter)->Backup_Size == 0) {
1625 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 0);
1626 DataManager::SetValue(TW_BACKUP_ANDSEC_VAR, 0);
1627 } else
1628 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 1);
Dees_Troy2c50e182012-09-26 20:05:28 -04001629 } else if ((*iter)->Mount_Point == "/boot") {
1630 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1631 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1632 if ((*iter)->Backup_Size == 0) {
1633 DataManager::SetValue("tw_has_boot_partition", 0);
1634 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1635 } else
1636 DataManager::SetValue("tw_has_boot_partition", 1);
Dees_Troy51127312012-09-08 13:08:49 -04001637 }
Dees_Troyaa9cc402012-10-13 12:14:05 -04001638 } else {
1639 // Handle unmountable partitions in case we reset defaults
1640 if ((*iter)->Mount_Point == "/boot") {
1641 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1642 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1643 if ((*iter)->Backup_Size == 0) {
1644 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 0);
1645 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1646 } else
1647 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 1);
1648 } else if ((*iter)->Mount_Point == "/recovery") {
1649 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1650 DataManager::SetValue(TW_BACKUP_RECOVERY_SIZE, backup_display_size);
1651 if ((*iter)->Backup_Size == 0) {
1652 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 0);
1653 DataManager::SetValue(TW_BACKUP_RECOVERY_VAR, 0);
1654 } else
1655 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 1);
Gary Peck82599a82012-11-21 16:23:12 -08001656 } else if ((*iter)->Mount_Point == "/data") {
1657 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troyaa9cc402012-10-13 12:14:05 -04001658 }
Dees_Troy51127312012-09-08 13:08:49 -04001659 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001660 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001661 gui_msg("update_part_details_done=...done");
Dees_Troy51127312012-09-08 13:08:49 -04001662 DataManager::SetValue(TW_BACKUP_DATA_SIZE, data_size);
1663 string current_storage_path = DataManager::GetCurrentStoragePath();
1664 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
Dees_Troy8170a922012-09-18 15:40:25 -04001665 if (FreeStorage != NULL) {
1666 // Attempt to mount storage
1667 if (!FreeStorage->Mount(false)) {
Matt Mower2d50cad2015-12-03 22:26:55 -06001668 gui_msg(Msg(msg::kError, "unable_to_mount_storage=Unable to mount storage"));
1669 DataManager::SetValue(TW_STORAGE_FREE_SIZE, 0);
Dees_Troy8170a922012-09-18 15:40:25 -04001670 } else {
1671 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
1672 }
1673 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001674 LOGINFO("Unable to find storage partition '%s'.\n", current_storage_path.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -04001675 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001676 if (!Write_Fstab())
Dees_Troy2673cec2013-04-02 20:22:16 +00001677 LOGERR("Error creating fstab\n");
Dees_Troy51a0e822012-09-05 15:24:24 -04001678 return;
1679}
1680
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001681void TWPartitionManager::Post_Decrypt(const string& Block_Device) {
1682 TWPartition* dat = Find_Partition_By_Path("/data");
1683 if (dat != NULL) {
1684 DataManager::SetValue(TW_IS_DECRYPTED, 1);
1685 dat->Is_Decrypted = true;
1686 if (!Block_Device.empty()) {
1687 dat->Decrypted_Block_Device = Block_Device;
1688 gui_msg(Msg("decrypt_success_dev=Data successfully decrypted, new block device: '{1}'")(Block_Device));
1689 } else {
1690 gui_msg("decrypt_success_nodev=Data successfully decrypted");
1691 }
Noah Jacobson5a79f672019-04-28 00:10:07 -04001692 property_set("twrp.decrypt.done", "true");
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001693 dat->Setup_File_System(false);
Noah Jacobson5a79f672019-04-28 00:10:07 -04001694 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 -06001695
Noah Jacobson5a79f672019-04-28 00:10:07 -04001696 sleep(1); // Sleep for a bit so that the device will be ready
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001697 if (dat->Has_Data_Media && dat->Mount(false) && TWFunc::Path_Exists("/data/media/0")) {
1698 dat->Storage_Path = "/data/media/0";
1699 dat->Symlink_Path = dat->Storage_Path;
1700 DataManager::SetValue("tw_storage_path", "/data/media/0");
1701 DataManager::SetValue("tw_settings_path", "/data/media/0");
1702 dat->UnMount(false);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001703 }
epicX98053c32021-01-04 13:01:31 +05301704 DataManager::LoadTWRPFolderInfo();
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001705 Update_System_Details();
nkk71ffb02bd2017-06-04 23:12:16 +03001706 Output_Partition(dat);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001707 UnMount_Main_Partitions();
1708 } else
1709 LOGERR("Unable to locate data partition.\n");
1710}
1711
Noah Jacobson5a79f672019-04-28 00:10:07 -04001712void TWPartitionManager::Parse_Users() {
1713#ifdef TW_INCLUDE_FBE
1714 char user_check_result[PROPERTY_VALUE_MAX];
1715 for (int userId = 0; userId <= 9999; userId++) {
1716 string prop = "twrp.user." + to_string(userId) + ".decrypt";
1717 property_get(prop.c_str(), user_check_result, "-1");
1718 if (strcmp(user_check_result, "-1") != 0) {
1719 if (userId < 0 || userId > 9999) {
1720 LOGINFO("Incorrect user id %d\n", userId);
1721 continue;
1722 }
1723 struct users_struct user;
1724 user.userId = to_string(userId);
1725
1726 // Attempt to get name of user. Fallback to user ID if this fails.
1727 char* userFile = PageManager::LoadFileToBuffer("/data/system/users/" + to_string(userId) + ".xml", NULL);
1728 if (userFile == NULL)
1729 user.userName = to_string(userId);
1730 else {
1731 xml_document<> *userXml = new xml_document<>();
1732 userXml->parse<0>(userFile);
1733 xml_node<>* userNode = userXml->first_node("user");
1734 if (userNode == nullptr) {
1735 user.userName = to_string(userId);
1736 } else {
1737 xml_node<>* nameNode = userNode->first_node("name");
1738 if (nameNode == nullptr)
1739 user.userName = to_string(userId);
1740 else {
1741 string userName = nameNode->value();
1742 user.userName = userName + " (" + to_string(userId) + ")";
1743 }
1744 }
1745 }
1746
1747 string filename;
1748 user.type = Get_Password_Type(userId, filename);
1749
1750 user.isDecrypted = false;
1751 if (strcmp(user_check_result, "1") == 0)
1752 user.isDecrypted = true;
1753 Users_List.push_back(user);
1754 }
1755 }
1756 Check_Users_Decryption_Status();
1757#endif
1758}
1759
1760std::vector<users_struct>* TWPartitionManager::Get_Users_List() {
1761 return &Users_List;
1762}
1763
1764void TWPartitionManager::Mark_User_Decrypted(int userID) {
1765#ifdef TW_INCLUDE_FBE
1766 std::vector<users_struct>::iterator iter;
1767 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1768 if (atoi((*iter).userId.c_str()) == userID) {
1769 (*iter).isDecrypted = true;
1770 string user_prop_decrypted = "twrp.user." + to_string(userID) + ".decrypt";
1771 property_set(user_prop_decrypted.c_str(), "1");
1772 break;
1773 }
1774 }
1775 Check_Users_Decryption_Status();
1776#endif
1777}
1778
1779void TWPartitionManager::Check_Users_Decryption_Status() {
1780#ifdef TW_INCLUDE_FBE
1781 int all_is_decrypted = 1;
1782 std::vector<users_struct>::iterator iter;
1783 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1784 if (!(*iter).isDecrypted) {
Noah Jacobsonbf0b0fb2020-06-05 12:47:20 -04001785 LOGINFO("User %s is not decrypted.\n", (*iter).userId.c_str());
Noah Jacobson5a79f672019-04-28 00:10:07 -04001786 all_is_decrypted = 0;
1787 break;
1788 }
1789 }
1790 if (all_is_decrypted == 1) {
Noah Jacobsonbf0b0fb2020-06-05 12:47:20 -04001791 LOGINFO("All found users are decrypted.\n");
Noah Jacobson5a79f672019-04-28 00:10:07 -04001792 DataManager::SetValue("tw_all_users_decrypted", "1");
1793 property_set("twrp.all.users.decrypted", "true");
1794 } else
1795 DataManager::SetValue("tw_all_users_decrypted", "0");
1796#endif
1797}
1798
1799int TWPartitionManager::Decrypt_Device(string Password, int user_id) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001800#ifdef TW_INCLUDE_CRYPTO
Ethan Yonkerddb63e22017-01-19 14:01:57 -06001801 char crypto_state[PROPERTY_VALUE_MAX], crypto_blkdev[PROPERTY_VALUE_MAX];
Ethan Yonker253368a2014-11-25 15:00:52 -06001802 std::vector<TWPartition*>::iterator iter;
Dees_Troy5bf43922012-09-07 16:07:55 -04001803
Ethan Yonker253368a2014-11-25 15:00:52 -06001804 // Mount any partitions that need to be mounted for decrypt
1805 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1806 if ((*iter)->Mount_To_Decrypt) {
1807 (*iter)->Mount(true);
1808 }
a39552696ff55ce2013-01-08 16:14:56 +00001809 }
oshmouna82a7542018-04-10 17:45:55 +02001810 property_set("twrp.mount_to_decrypt", "1");
a39552696ff55ce2013-01-08 16:14:56 +00001811
Ethan Yonkera1de1492015-11-04 11:58:27 -06001812 property_get("ro.crypto.state", crypto_state, "error");
1813 if (strcmp(crypto_state, "error") == 0) {
Captain Throwback07b8e052020-10-29 12:43:17 -04001814 Set_FDE_Encrypt_Status();
Ethan Yonkera1de1492015-11-04 11:58:27 -06001815 }
1816
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001817 if (DataManager::GetIntValue(TW_IS_FBE)) {
1818#ifdef TW_INCLUDE_FBE
1819 if (!Mount_By_Path("/data", true)) // /data has to be mounted for FBE
1820 return -1;
Noah Jacobson5a79f672019-04-28 00:10:07 -04001821
1822 bool user_need_decrypt = false;
1823 std::vector<users_struct>::iterator iter;
1824 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1825 if (atoi((*iter).userId.c_str()) == user_id && !(*iter).isDecrypted) {
1826 user_need_decrypt = true;
1827 }
1828 }
1829 if (!user_need_decrypt) {
1830 LOGINFO("User %d does not require decryption\n", user_id);
1831 return 0;
1832 }
1833
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001834 int retry_count = 10;
1835 while (!TWFunc::Path_Exists("/data/system/users/gatekeeper.password.key") && --retry_count)
Noah Jacobson5a79f672019-04-28 00:10:07 -04001836 usleep(2000); // A small sleep is needed after mounting /data to ensure reliable decrypt...maybe because of DE?
1837 gui_msg(Msg("decrypting_user_fbe=Attempting to decrypt FBE for user {1}...")(user_id));
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001838 if (Decrypt_User(user_id, Password)) {
Noah Jacobson5a79f672019-04-28 00:10:07 -04001839 gui_msg(Msg("decrypt_user_success_fbe=User {1} Decrypted Successfully")(user_id));
1840 Mark_User_Decrypted(user_id);
1841 if (user_id == 0) {
1842 // When decrypting user 0 also try all other users
1843 std::vector<users_struct>::iterator iter;
1844 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1845 if ((*iter).userId == "0" || (*iter).isDecrypted)
1846 continue;
1847
1848 int tmp_user_id = atoi((*iter).userId.c_str());
1849 gui_msg(Msg("decrypting_user_fbe=Attempting to decrypt FBE for user {1}...")(tmp_user_id));
1850 if (Decrypt_User(tmp_user_id, Password) ||
1851 (Password != "!" && Decrypt_User(tmp_user_id, "!"))) { // "!" means default password
1852 gui_msg(Msg("decrypt_user_success_fbe=User {1} Decrypted Successfully")(tmp_user_id));
1853 Mark_User_Decrypted(tmp_user_id);
1854 } else {
1855 gui_msg(Msg("decrypt_user_fail_fbe=Failed to decrypt user {1}")(tmp_user_id));
1856 }
1857 }
1858 Post_Decrypt("");
1859 }
1860
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001861 return 0;
Noah Jacobson5a79f672019-04-28 00:10:07 -04001862 } else {
1863 gui_msg(Msg(msg::kError, "decrypt_user_fail_fbe=Failed to decrypt user {1}")(user_id));
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001864 }
1865#else
1866 LOGERR("FBE support is not present\n");
1867#endif
1868 return -1;
1869 }
1870
Noah Jacobson5a79f672019-04-28 00:10:07 -04001871 char isdecrypteddata[PROPERTY_VALUE_MAX];
1872 property_get("twrp.decrypt.done", isdecrypteddata, "");
1873 if (strcmp(isdecrypteddata, "true") == 0) {
1874 LOGINFO("Data has no decryption required\n");
1875 return 0;
1876 }
1877
Ethan Yonkerddb63e22017-01-19 14:01:57 -06001878 int pwret = -1;
1879 pid_t pid = fork();
1880 if (pid < 0) {
1881 LOGERR("fork failed\n");
1882 return -1;
1883 } else if (pid == 0) {
1884 // Child process
1885 char cPassword[255];
1886 strcpy(cPassword, Password.c_str());
1887 int ret = cryptfs_check_passwd(cPassword);
1888 exit(ret);
1889 } else {
1890 // Parent
1891 int status;
1892 if (TWFunc::Wait_For_Child_Timeout(pid, &status, "Decrypt", 30))
1893 pwret = -1;
1894 else
1895 pwret = WEXITSTATUS(status) ? -1 : 0;
1896 }
a39552696ff55ce2013-01-08 16:14:56 +00001897
nkk7171c6c502017-01-05 23:55:05 +02001898#ifdef TW_CRYPTO_USE_SYSTEM_VOLD
1899 if (pwret != 0) {
1900 pwret = vold_decrypt(Password);
nkk717d1222a2017-11-06 17:27:21 +02001901 switch (pwret) {
1902 case VD_SUCCESS:
1903 break;
1904 case VD_ERR_MISSING_VDC:
1905 gui_msg(Msg(msg::kError, "decrypt_data_vold_os_missing=Missing files needed for vold decrypt: {1}")("/system/bin/vdc"));
1906 break;
1907 case VD_ERR_MISSING_VOLD:
1908 gui_msg(Msg(msg::kError, "decrypt_data_vold_os_missing=Missing files needed for vold decrypt: {1}")("/system/bin/vold"));
1909 break;
1910 }
nkk7171c6c502017-01-05 23:55:05 +02001911 }
1912#endif // TW_CRYPTO_USE_SYSTEM_VOLD
1913
Ethan Yonker253368a2014-11-25 15:00:52 -06001914 // Unmount any partitions that were needed for decrypt
1915 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1916 if ((*iter)->Mount_To_Decrypt) {
1917 (*iter)->UnMount(false);
1918 }
1919 }
oshmouna82a7542018-04-10 17:45:55 +02001920 property_set("twrp.mount_to_decrypt", "0");
Ethan Yonker253368a2014-11-25 15:00:52 -06001921
a39552696ff55ce2013-01-08 16:14:56 +00001922 if (pwret != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001923 gui_err("fail_decrypt=Failed to decrypt data.");
Dees_Troy5bf43922012-09-07 16:07:55 -04001924 return -1;
1925 }
a39552696ff55ce2013-01-08 16:14:56 +00001926
Dees_Troy5bf43922012-09-07 16:07:55 -04001927 property_get("ro.crypto.fs_crypto_blkdev", crypto_blkdev, "error");
1928 if (strcmp(crypto_blkdev, "error") == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001929 LOGERR("Error retrieving decrypted data block device.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04001930 } else {
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001931 Post_Decrypt(crypto_blkdev);
Dees_Troy5bf43922012-09-07 16:07:55 -04001932 }
1933 return 0;
1934#else
Ethan Yonker74db1572015-10-28 12:44:49 -05001935 gui_err("no_crypto_support=No crypto support was compiled into this build.");
Dees_Troy5bf43922012-09-07 16:07:55 -04001936 return -1;
1937#endif
Dees_Troy51a0e822012-09-05 15:24:24 -04001938 return 1;
Dees_Troy51127312012-09-08 13:08:49 -04001939}
1940
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001941int TWPartitionManager::Fix_Contexts(void) {
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001942 std::vector<TWPartition*>::iterator iter;
1943 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1944 if ((*iter)->Has_Data_Media) {
1945 if ((*iter)->Mount(true)) {
1946 if (fixContexts::fixDataMediaContexts((*iter)->Mount_Point) != 0)
1947 return -1;
1948 }
1949 }
1950 }
Dees_Troy1a650e62012-10-19 20:54:32 -04001951 UnMount_Main_Partitions();
Ethan Yonker74db1572015-10-28 12:44:49 -05001952 gui_msg("done=Done.");
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001953 return 0;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04001954}
1955
Ethan Yonker66a19492015-12-10 10:19:45 -06001956TWPartition* TWPartitionManager::Find_Next_Storage(string Path, bool Exclude_Data_Media) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001957 std::vector<TWPartition*>::iterator iter = Partitions.begin();
1958
1959 if (!Path.empty()) {
1960 string Search_Path = TWFunc::Get_Root_Path(Path);
1961 for (; iter != Partitions.end(); iter++) {
Matt Mower9a561dd2016-02-22 21:25:51 -06001962 if ((*iter)->Mount_Point == Search_Path) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001963 iter++;
1964 break;
1965 }
1966 }
1967 }
1968
1969 for (; iter != Partitions.end(); iter++) {
Ethan Yonker66a19492015-12-10 10:19:45 -06001970 if (Exclude_Data_Media && (*iter)->Has_Data_Media) {
1971 // do nothing, do not return this type of partition
1972 } else if ((*iter)->Is_Storage && (*iter)->Is_Present) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001973 return (*iter);
1974 }
1975 }
1976
1977 return NULL;
1978}
1979
Dees_Troyd21618c2012-10-14 18:48:49 -04001980int TWPartitionManager::Open_Lun_File(string Partition_Path, string Lun_File) {
Dees_Troyd21618c2012-10-14 18:48:49 -04001981 TWPartition* Part = Find_Partition_By_Path(Partition_Path);
1982
1983 if (Part == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001984 LOGINFO("Unable to locate '%s' for USB storage mode.", Partition_Path.c_str());
1985 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 -04001986 return false;
1987 }
Ethan Yonker47360be2014-04-01 10:34:34 -05001988 LOGINFO("USB mount '%s', '%s' > '%s'\n", Partition_Path.c_str(), Part->Actual_Block_Device.c_str(), Lun_File.c_str());
1989 if (!Part->UnMount(true) || !Part->Is_Present)
Dees_Troyd21618c2012-10-14 18:48:49 -04001990 return false;
1991
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06001992 if (TWFunc::write_to_file(Lun_File, Part->Actual_Block_Device)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001993 LOGERR("Unable to write to ums lunfile '%s': (%s)\n", Lun_File.c_str(), strerror(errno));
Dees_Troyd21618c2012-10-14 18:48:49 -04001994 return false;
1995 }
Dees_Troyd21618c2012-10-14 18:48:49 -04001996 return true;
1997}
1998
Dees_Troy8170a922012-09-18 15:40:25 -04001999int TWPartitionManager::usb_storage_enable(void) {
Dees_Troy8170a922012-09-18 15:40:25 -04002000 char lun_file[255];
Dees_Troyd21618c2012-10-14 18:48:49 -04002001 bool has_multiple_lun = false;
Dees_Troy8170a922012-09-18 15:40:25 -04002002
Ethan Yonker47360be2014-04-01 10:34:34 -05002003 string Lun_File_str = CUSTOM_LUN_FILE;
2004 size_t found = Lun_File_str.find("%");
2005 if (found != string::npos) {
2006 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
2007 if (TWFunc::Path_Exists(lun_file))
2008 has_multiple_lun = true;
2009 }
Ethan Yonker726a0202014-12-16 20:01:38 -06002010 mtp_was_enabled = TWFunc::Toggle_MTP(false); // Must disable MTP for USB Storage
Ethan Yonker47360be2014-04-01 10:34:34 -05002011 if (!has_multiple_lun) {
2012 LOGINFO("Device doesn't have multiple lun files, mount current storage\n");
2013 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
2014 if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) == "/data") {
Ethan Yonker66a19492015-12-10 10:19:45 -06002015 TWPartition* Mount = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05002016 if (Mount) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002017 if (!Open_Lun_File(Mount->Mount_Point, lun_file)) {
2018 goto error_handle;
2019 }
Ethan Yonker47360be2014-04-01 10:34:34 -05002020 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002021 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002022 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05002023 }
2024 } else if (!Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file)) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002025 goto error_handle;
Dees_Troy8170a922012-09-18 15:40:25 -04002026 }
Dees_Troy8170a922012-09-18 15:40:25 -04002027 } else {
Ethan Yonker47360be2014-04-01 10:34:34 -05002028 LOGINFO("Device has multiple lun files\n");
2029 TWPartition* Mount1;
2030 TWPartition* Mount2;
Dees_Troy8170a922012-09-18 15:40:25 -04002031 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
Ethan Yonker66a19492015-12-10 10:19:45 -06002032 Mount1 = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05002033 if (Mount1) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002034 if (!Open_Lun_File(Mount1->Mount_Point, lun_file)) {
2035 goto error_handle;
2036 }
Matt Moweree717062014-04-26 01:46:46 -05002037 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
Ethan Yonker66a19492015-12-10 10:19:45 -06002038 Mount2 = Find_Next_Storage(Mount1->Mount_Point, true);
Matt Mower1fdcdb72016-01-25 20:53:47 -06002039 if (Mount2 && Mount2->Mount_Point != Mount1->Mount_Point) {
Matt Moweree717062014-04-26 01:46:46 -05002040 Open_Lun_File(Mount2->Mount_Point, lun_file);
Ethan Yonker47360be2014-04-01 10:34:34 -05002041 }
2042 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002043 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002044 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05002045 }
Dees_Troy8170a922012-09-18 15:40:25 -04002046 }
Matt Mowerb6ef4782014-11-06 02:24:36 -06002047 property_set("sys.storage.ums_enabled", "1");
HandyMenny37d42992014-10-26 22:15:59 +01002048 property_set("sys.usb.config", "mass_storage,adb");
Dees_Troy8170a922012-09-18 15:40:25 -04002049 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002050error_handle:
2051 if (mtp_was_enabled)
2052 if (!Enable_MTP())
2053 Disable_MTP();
2054 return false;
Dees_Troy8170a922012-09-18 15:40:25 -04002055}
2056
2057int TWPartitionManager::usb_storage_disable(void) {
Dees_Troy2673cec2013-04-02 20:22:16 +00002058 int index, ret;
2059 char lun_file[255], ch[2] = {0, 0};
2060 string str = ch;
Dees_Troy8170a922012-09-18 15:40:25 -04002061
2062 for (index=0; index<2; index++) {
2063 sprintf(lun_file, CUSTOM_LUN_FILE, index);
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002064 ret = TWFunc::write_to_file(lun_file, str);
Dees_Troy2673cec2013-04-02 20:22:16 +00002065 if (ret < 0) {
2066 break;
Dees_Troy8170a922012-09-18 15:40:25 -04002067 }
Dees_Troy8170a922012-09-18 15:40:25 -04002068 }
Dees_Troye58d5262012-09-21 12:27:57 -04002069 Mount_All_Storage();
2070 Update_System_Details();
Dees_Troycfd73ef2012-10-12 16:52:00 -04002071 UnMount_Main_Partitions();
Matt Mowerd9cb9062013-12-14 20:34:45 -06002072 property_set("sys.storage.ums_enabled", "0");
HandyMenny37d42992014-10-26 22:15:59 +01002073 property_set("sys.usb.config", "adb");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002074 if (mtp_was_enabled)
2075 if (!Enable_MTP())
2076 Disable_MTP();
Dees_Troy2673cec2013-04-02 20:22:16 +00002077 if (ret < 0 && index == 0) {
2078 LOGERR("Unable to write to ums lunfile '%s'.", lun_file);
2079 return false;
2080 } else {
2081 return true;
2082 }
Dees_Troy8170a922012-09-18 15:40:25 -04002083 return true;
Dees_Troy812660f2012-09-20 09:55:17 -04002084}
2085
2086void TWPartitionManager::Mount_All_Storage(void) {
2087 std::vector<TWPartition*>::iterator iter;
2088
2089 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2090 if ((*iter)->Is_Storage)
2091 (*iter)->Mount(false);
2092 }
Dees_Troy2c50e182012-09-26 20:05:28 -04002093}
Dees_Troy9350b8d2012-09-27 12:38:38 -04002094
Dees_Troyd0384ef2012-10-12 12:15:42 -04002095void TWPartitionManager::UnMount_Main_Partitions(void) {
2096 // Unmounts system and data if data is not data/media
2097 // Also unmounts boot if boot is mountable
Dees_Troy2673cec2013-04-02 20:22:16 +00002098 LOGINFO("Unmounting main partitions...\n");
Dees_Troyd0384ef2012-10-12 12:15:42 -04002099
2100 TWPartition* Boot_Partition = Find_Partition_By_Path("/boot");
2101
Captain Throwback9d6feb52018-07-27 10:05:24 -04002102 UnMount_By_Path(Get_Android_Root_Path(), true);
Ethan Yonker6277c792014-09-15 14:54:30 -05002103 if (!datamedia)
2104 UnMount_By_Path("/data", true);
2105
Dees_Troyd0384ef2012-10-12 12:15:42 -04002106 if (Boot_Partition != NULL && Boot_Partition->Can_Be_Mounted)
2107 Boot_Partition->UnMount(true);
2108}
2109
Dees_Troy9350b8d2012-09-27 12:38:38 -04002110int TWPartitionManager::Partition_SDCard(void) {
Matt Mower23d8aae2017-01-06 14:30:33 -06002111 char temp[255];
Ethan Yonker483e9f42016-01-11 22:21:18 -06002112 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 -04002113 int ext, swap, total_size = 0, fat_size;
Dees_Troy9350b8d2012-09-27 12:38:38 -04002114
Ethan Yonker74db1572015-10-28 12:44:49 -05002115 gui_msg("start_partition_sd=Partitioning SD Card...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06002116
2117 // Locate and validate device to partition
2118 TWPartition* SDCard = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
2119
Ethan Yonker66a19492015-12-10 10:19:45 -06002120 if (SDCard->Is_Adopted_Storage)
2121 SDCard->Revert_Adopted();
2122
Ethan Yonker1eff6cd2014-09-15 13:30:42 -05002123 if (SDCard == NULL || !SDCard->Removable || SDCard->Has_Data_Media) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002124 gui_err("partition_sd_locate=Unable to locate device to partition.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002125 return false;
2126 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002127
2128 // Unmount everything
Dees_Troy9350b8d2012-09-27 12:38:38 -04002129 if (!SDCard->UnMount(true))
2130 return false;
2131 TWPartition* SDext = Find_Partition_By_Path("/sd-ext");
2132 if (SDext != NULL) {
2133 if (!SDext->UnMount(true))
2134 return false;
2135 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002136 char* swappath = getenv("SWAPPATH");
2137 if (swappath != NULL) {
2138 LOGINFO("Unmounting swap at '%s'\n", swappath);
2139 umount(swappath);
2140 }
Vojtech Bocek05534202013-09-11 08:11:56 +02002141
Ethan Yonker483e9f42016-01-11 22:21:18 -06002142 // Determine block device
2143 if (SDCard->Alternate_Block_Device.empty()) {
2144 SDCard->Find_Actual_Block_Device();
2145 Device = SDCard->Actual_Block_Device;
2146 // Just use the root block device
2147 Device.resize(strlen("/dev/block/mmcblkX"));
2148 } else {
2149 Device = SDCard->Alternate_Block_Device;
2150 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04002151
2152 // Find the size of the block device:
Ethan Yonker483e9f42016-01-11 22:21:18 -06002153 total_size = (int)(TWFunc::IOCTL_Get_Block_Size(Device.c_str()) / (1048576));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002154
2155 DataManager::GetValue("tw_sdext_size", ext);
2156 DataManager::GetValue("tw_swap_size", swap);
2157 DataManager::GetValue("tw_sdpart_file_system", ext_format);
2158 fat_size = total_size - ext - swap;
Ethan Yonker483e9f42016-01-11 22:21:18 -06002159 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);
2160
2161 // Determine partition sizes
2162 if (swap == 0 && ext == 0) {
2163 fat_str = "-0";
2164 } else {
2165 memset(temp, 0, sizeof(temp));
2166 sprintf(temp, "%i", fat_size);
2167 fat_str = temp;
2168 fat_str += "MB";
2169 }
2170 if (swap == 0) {
2171 ext_str = "-0";
2172 } else {
2173 memset(temp, 0, sizeof(temp));
2174 sprintf(temp, "%i", ext);
2175 ext_str = "+";
2176 ext_str += temp;
2177 ext_str += "MB";
2178 }
2179
Dees_Troy9350b8d2012-09-27 12:38:38 -04002180 if (ext + swap > total_size) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002181 gui_err("ext_swap_size=EXT + Swap size is larger than sdcard size.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002182 return false;
2183 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002184
Ethan Yonker74db1572015-10-28 12:44:49 -05002185 gui_msg("remove_part_table=Removing partition table...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06002186 Command = "sgdisk --zap-all " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002187 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002188 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002189 gui_err("unable_rm_part=Unable to remove partition table.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002190 Update_System_Details();
2191 return false;
2192 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002193 gui_msg(Msg("create_part=Creating {1} partition...")("FAT32"));
Captain Throwback9643a802016-05-27 11:44:51 -04002194 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 +00002195 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002196 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002197 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("FAT32"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002198 return false;
2199 }
2200 if (ext > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002201 gui_msg(Msg("create_part=Creating {1} partition...")("EXT"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06002202 Command = "sgdisk --new=0:0:" + ext_str + " --change-name=0:\"Linux filesystem\" " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002203 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002204 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002205 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("EXT"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002206 Update_System_Details();
2207 return false;
2208 }
2209 }
2210 if (swap > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002211 gui_msg(Msg("create_part=Creating {1} partition...")("swap"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06002212 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 +00002213 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002214 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002215 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("swap"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002216 Update_System_Details();
2217 return false;
2218 }
2219 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002220
2221 // Convert GPT to MBR
2222 Command = "sgdisk --gpttombr " + Device;
2223 if (TWFunc::Exec_Cmd(Command) != 0)
2224 LOGINFO("Failed to covert partition GPT to MBR\n");
2225
2226 // Tell the kernel to rescan the partition table
2227 int fd = open(Device.c_str(), O_RDONLY);
2228 ioctl(fd, BLKRRPART, 0);
2229 close(fd);
2230
2231 string format_device = Device;
2232 if (Device.substr(0, 17) == "/dev/block/mmcblk")
2233 format_device += "p";
2234
2235 // Format new partitions to proper file system
2236 if (fat_size > 0) {
2237 Command = "mkfs.fat " + format_device + "1";
2238 TWFunc::Exec_Cmd(Command);
2239 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04002240 if (ext > 0) {
2241 if (SDext == NULL) {
Ethan Yonker483e9f42016-01-11 22:21:18 -06002242 Command = "mke2fs -t " + ext_format + " -m 0 " + format_device + "2";
2243 gui_msg(Msg("format_sdext_as=Formatting sd-ext as {1}...")(ext_format));
2244 LOGINFO("Formatting sd-ext after partitioning, command: '%s'\n", Command.c_str());
2245 TWFunc::Exec_Cmd(Command);
2246 } else {
2247 SDext->Wipe(ext_format);
Dees_Troy9350b8d2012-09-27 12:38:38 -04002248 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002249 }
2250 if (swap > 0) {
2251 Command = "mkswap " + format_device;
2252 if (ext > 0)
2253 Command += "3";
2254 else
2255 Command += "2";
Vojtech Bocek05534202013-09-11 08:11:56 +02002256 TWFunc::Exec_Cmd(Command);
Dees_Troy9350b8d2012-09-27 12:38:38 -04002257 }
2258
Ethan Yonker483e9f42016-01-11 22:21:18 -06002259 // recreate TWRP folder and rewrite settings - these will be gone after sdcard is partitioned
2260 if (SDCard->Mount(true)) {
2261 string TWRP_Folder = SDCard->Mount_Point + "/TWRP";
2262 mkdir(TWRP_Folder.c_str(), 0777);
2263 DataManager::Flush();
2264 }
2265
Dees_Troy9350b8d2012-09-27 12:38:38 -04002266 Update_System_Details();
Ethan Yonker74db1572015-10-28 12:44:49 -05002267 gui_msg("part_complete=Partitioning complete.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002268 return true;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04002269}
Dees_Troya13d74f2013-03-24 08:54:55 -05002270
2271void TWPartitionManager::Get_Partition_List(string ListType, std::vector<PartitionList> *Partition_List) {
2272 std::vector<TWPartition*>::iterator iter;
2273 if (ListType == "mount") {
2274 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002275 if ((*iter)->Can_Be_Mounted) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002276 struct PartitionList part;
2277 part.Display_Name = (*iter)->Display_Name;
2278 part.Mount_Point = (*iter)->Mount_Point;
2279 part.selected = (*iter)->Is_Mounted();
2280 Partition_List->push_back(part);
2281 }
2282 }
2283 } else if (ListType == "storage") {
2284 char free_space[255];
2285 string Current_Storage = DataManager::GetCurrentStoragePath();
2286 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2287 if ((*iter)->Is_Storage) {
2288 struct PartitionList part;
2289 sprintf(free_space, "%llu", (*iter)->Free / 1024 / 1024);
2290 part.Display_Name = (*iter)->Storage_Name + " (";
2291 part.Display_Name += free_space;
2292 part.Display_Name += "MB)";
2293 part.Mount_Point = (*iter)->Storage_Path;
2294 if ((*iter)->Storage_Path == Current_Storage)
2295 part.selected = 1;
2296 else
2297 part.selected = 0;
2298 Partition_List->push_back(part);
2299 }
2300 }
2301 } else if (ListType == "backup") {
2302 char backup_size[255];
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002303 unsigned long long Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -05002304 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002305 if ((*iter)->Can_Be_Backed_Up && !(*iter)->Is_SubPartition && (*iter)->Is_Present) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002306 struct PartitionList part;
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002307 Backup_Size = (*iter)->Backup_Size;
2308 if ((*iter)->Has_SubPartition) {
2309 std::vector<TWPartition*>::iterator subpart;
2310
2311 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
2312 if ((*subpart)->Is_SubPartition && (*subpart)->Can_Be_Backed_Up && (*subpart)->Is_Present && (*subpart)->SubPartition_Of == (*iter)->Mount_Point)
2313 Backup_Size += (*subpart)->Backup_Size;
2314 }
2315 }
2316 sprintf(backup_size, "%llu", Backup_Size / 1024 / 1024);
Dees_Troya13d74f2013-03-24 08:54:55 -05002317 part.Display_Name = (*iter)->Backup_Display_Name + " (";
2318 part.Display_Name += backup_size;
2319 part.Display_Name += "MB)";
2320 part.Mount_Point = (*iter)->Backup_Path;
2321 part.selected = 0;
2322 Partition_List->push_back(part);
2323 }
2324 }
2325 } else if (ListType == "restore") {
2326 string Restore_List, restore_path;
2327 TWPartition* restore_part = NULL;
2328
2329 DataManager::GetValue("tw_restore_list", Restore_List);
2330 if (!Restore_List.empty()) {
2331 size_t start_pos = 0, end_pos = Restore_List.find(";", start_pos);
2332 while (end_pos != string::npos && start_pos < Restore_List.size()) {
2333 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04002334 struct PartitionList part;
2335 if (restore_path.compare("ADB_Backup") == 0) {
2336 part.Display_Name = "ADB Backup";
2337 part.Mount_Point = "ADB Backup";
2338 part.selected = 1;
2339 Partition_List->push_back(part);
2340 break;
2341 }
Dees_Troy59df9262013-06-19 14:53:57 -05002342 if ((restore_part = Find_Partition_By_Path(restore_path)) != NULL) {
Dees Troy4159aed2014-02-28 17:24:43 +00002343 if ((restore_part->Backup_Name == "recovery" && !restore_part->Can_Be_Backed_Up) || restore_part->Is_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002344 // Don't allow restore of recovery (causes problems on some devices)
Dees_Troy59df9262013-06-19 14:53:57 -05002345 // Don't add subpartitions to the list of items
Dees_Troya13d74f2013-03-24 08:54:55 -05002346 } else {
Dees_Troya13d74f2013-03-24 08:54:55 -05002347 part.Display_Name = restore_part->Backup_Display_Name;
2348 part.Mount_Point = restore_part->Backup_Path;
2349 part.selected = 1;
2350 Partition_List->push_back(part);
2351 }
2352 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002353 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05002354 }
2355 start_pos = end_pos + 1;
2356 end_pos = Restore_List.find(";", start_pos);
2357 }
2358 }
2359 } else if (ListType == "wipe") {
2360 struct PartitionList dalvik;
Ethan Yonker74db1572015-10-28 12:44:49 -05002361 dalvik.Display_Name = gui_parse_text("{@dalvik}");
Dees_Troya13d74f2013-03-24 08:54:55 -05002362 dalvik.Mount_Point = "DALVIK";
2363 dalvik.selected = 0;
2364 Partition_List->push_back(dalvik);
2365 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2366 if ((*iter)->Wipe_Available_in_GUI && !(*iter)->Is_SubPartition) {
2367 struct PartitionList part;
2368 part.Display_Name = (*iter)->Display_Name;
2369 part.Mount_Point = (*iter)->Mount_Point;
2370 part.selected = 0;
2371 Partition_List->push_back(part);
2372 }
2373 if ((*iter)->Has_Android_Secure) {
2374 struct PartitionList part;
2375 part.Display_Name = (*iter)->Backup_Display_Name;
2376 part.Mount_Point = (*iter)->Backup_Path;
2377 part.selected = 0;
2378 Partition_List->push_back(part);
2379 }
Dees_Troy74fb2e92013-04-15 14:35:47 +00002380 if ((*iter)->Has_Data_Media) {
2381 struct PartitionList datamedia;
2382 datamedia.Display_Name = (*iter)->Storage_Name;
2383 datamedia.Mount_Point = "INTERNAL";
2384 datamedia.selected = 0;
2385 Partition_List->push_back(datamedia);
2386 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002387 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002388 } else if (ListType == "flashimg") {
2389 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2390 if ((*iter)->Can_Flash_Img && (*iter)->Is_Present) {
2391 struct PartitionList part;
2392 part.Display_Name = (*iter)->Backup_Display_Name;
2393 part.Mount_Point = (*iter)->Backup_Path;
2394 part.selected = 0;
2395 Partition_List->push_back(part);
2396 }
2397 }
Ethan Yonker53796e72019-01-11 22:49:52 -06002398 if (DataManager::GetIntValue("tw_has_repack_tools") != 0 && DataManager::GetIntValue("tw_has_boot_slots") != 0) {
2399 TWPartition* boot = Find_Partition_By_Path("/boot");
2400 if (boot) {
2401 // Allow flashing kernels and ramdisks
2402 struct PartitionList repack_ramdisk;
2403 repack_ramdisk.Display_Name = gui_lookup("install_twrp_ramdisk", "Install Recovery Ramdisk");
2404 repack_ramdisk.Mount_Point = "/repack_ramdisk";
2405 repack_ramdisk.selected = 0;
2406 Partition_List->push_back(repack_ramdisk);
2407 /*struct PartitionList repack_kernel; For now let's leave repacking kernels under advanced only
2408 repack_kernel.Display_Name = gui_lookup("install_kernel", "Install Kernel");
2409 repack_kernel.Mount_Point = "/repack_kernel";
2410 repack_kernel.selected = 0;
2411 Partition_List->push_back(repack_kernel);*/
2412 }
2413 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002414 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00002415 LOGERR("Unknown list type '%s' requested for TWPartitionManager::Get_Partition_List\n", ListType.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -05002416 }
2417}
2418
2419int TWPartitionManager::Fstab_Processed(void) {
2420 return Partitions.size();
2421}
Dees_Troyd93bda52013-07-03 19:55:19 +00002422
2423void TWPartitionManager::Output_Storage_Fstab(void) {
2424 std::vector<TWPartition*>::iterator iter;
2425 char storage_partition[255];
bigbiff bigbiffe4bdb152019-03-23 18:33:17 -04002426 std::string Temp;
bigbiff349ea552020-06-19 16:07:38 -04002427 std::string cacheDir = TWFunc::get_log_dir();
bigbiff bigbiffe4bdb152019-03-23 18:33:17 -04002428
2429 if (cacheDir.empty()) {
2430 LOGINFO("Unable to find cache directory\n");
2431 return;
2432 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002433
bigbiff349ea552020-06-19 16:07:38 -04002434 std::string storageFstab = TWFunc::get_log_dir() + "recovery/storage.fstab";
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002435 FILE *fp = fopen(storageFstab.c_str(), "w");
Dees_Troyd93bda52013-07-03 19:55:19 +00002436
2437 if (fp == NULL) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002438 gui_msg(Msg(msg::kError, "unable_to_open=Unable to open '{1}'.")(storageFstab));
Dees_Troyd93bda52013-07-03 19:55:19 +00002439 return;
2440 }
2441
2442 // Iterate through all partitions
2443 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2444 if ((*iter)->Is_Storage) {
2445 Temp = (*iter)->Storage_Path + ";" + (*iter)->Storage_Name + ";\n";
2446 strcpy(storage_partition, Temp.c_str());
2447 fwrite(storage_partition, sizeof(storage_partition[0]), strlen(storage_partition) / sizeof(storage_partition[0]), fp);
2448 }
2449 }
2450 fclose(fp);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02002451}
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002452
2453TWPartition *TWPartitionManager::Get_Default_Storage_Partition()
2454{
2455 TWPartition *res = NULL;
2456 for (std::vector<TWPartition*>::iterator iter = Partitions.begin(); iter != Partitions.end(); ++iter) {
Matt Mowera8a89d12016-12-30 18:10:37 -06002457 if (!(*iter)->Is_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002458 continue;
2459
Matt Mowera8a89d12016-12-30 18:10:37 -06002460 if ((*iter)->Is_Settings_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002461 return *iter;
2462
Matt Mowera8a89d12016-12-30 18:10:37 -06002463 if (!res)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002464 res = *iter;
2465 }
2466 return res;
2467}
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002468
2469bool TWPartitionManager::Enable_MTP(void) {
2470#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002471 if (mtppid) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002472 gui_err("mtp_already_enabled=MTP already enabled");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002473 return true;
2474 }
Ethan Yonker726a0202014-12-16 20:01:38 -06002475
2476 int mtppipe[2];
2477
2478 if (pipe(mtppipe) < 0) {
2479 LOGERR("Error creating MTP pipe\n");
2480 return false;
2481 }
2482
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002483 char old_value[PROPERTY_VALUE_MAX];
Matt Mowere07f0102017-02-23 17:40:00 -06002484 property_get("sys.usb.config", old_value, "");
2485 if (strcmp(old_value, "mtp,adb") != 0) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002486 char vendor[PROPERTY_VALUE_MAX];
2487 char product[PROPERTY_VALUE_MAX];
2488 property_set("sys.usb.config", "none");
2489 property_get("usb.vendor", vendor, "18D1");
2490 property_get("usb.product.mtpadb", product, "4EE2");
2491 string vendorstr = vendor;
2492 string productstr = product;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002493 TWFunc::write_to_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2494 TWFunc::write_to_file("/sys/class/android_usb/android0/idProduct", productstr);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002495 property_set("sys.usb.config", "mtp,adb");
2496 }
Matt Mower653a1702017-02-16 10:38:52 -06002497 /* To enable MTP debug, use the twrp command line feature:
Ethan Yonker6d154c42014-09-03 14:19:43 -05002498 * twrp set tw_mtp_debug 1
2499 */
2500 twrpMtp *mtp = new twrpMtp(DataManager::GetIntValue("tw_mtp_debug"));
Ethan Yonker1b039202015-01-30 10:08:48 -06002501 mtppid = mtp->forkserver(mtppipe);
2502 if (mtppid) {
2503 close(mtppipe[0]); // Host closes read side
2504 mtp_write_fd = mtppipe[1];
2505 DataManager::SetValue("tw_mtp_enabled", 1);
2506 Add_All_MTP_Storage();
2507 return true;
Ethan Yonker726a0202014-12-16 20:01:38 -06002508 } else {
2509 close(mtppipe[0]);
2510 close(mtppipe[1]);
Ethan Yonker74db1572015-10-28 12:44:49 -05002511 gui_err("mtp_fail=Failed to enable MTP");
Ethan Yonker1b039202015-01-30 10:08:48 -06002512 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002513 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002514#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002515 gui_err("no_mtp=MTP support not included");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002516#endif
2517 DataManager::SetValue("tw_mtp_enabled", 0);
2518 return false;
2519}
2520
Ethan Yonker1b039202015-01-30 10:08:48 -06002521void TWPartitionManager::Add_All_MTP_Storage(void) {
2522#ifdef TW_HAS_MTP
2523 std::vector<TWPartition*>::iterator iter;
2524
2525 if (!mtppid)
2526 return; // MTP is not enabled
2527
2528 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2529 if ((*iter)->Is_Storage && (*iter)->Is_Present && (*iter)->Mount(false))
2530 Add_Remove_MTP_Storage((*iter), MTP_MESSAGE_ADD_STORAGE);
2531 }
2532#else
2533 return;
2534#endif
2535}
2536
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002537bool TWPartitionManager::Disable_MTP(void) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002538 char old_value[PROPERTY_VALUE_MAX];
bigbiff bigbiffaf32bb92018-12-18 18:39:53 -05002539 property_set("sys.usb.ffs.mtp.ready", "0");
Matt Mowere07f0102017-02-23 17:40:00 -06002540 property_get("sys.usb.config", old_value, "");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002541 if (strcmp(old_value, "adb") != 0) {
2542 char vendor[PROPERTY_VALUE_MAX];
2543 char product[PROPERTY_VALUE_MAX];
2544 property_set("sys.usb.config", "none");
2545 property_get("usb.vendor", vendor, "18D1");
Matt Mowere07f0102017-02-23 17:40:00 -06002546 property_get("usb.product.adb", product, "D001");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002547 string vendorstr = vendor;
2548 string productstr = product;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002549 TWFunc::write_to_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2550 TWFunc::write_to_file("/sys/class/android_usb/android0/idProduct", productstr);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002551 usleep(2000);
2552 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002553#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002554 if (mtppid) {
Ethan Yonker8613dc02014-09-11 09:28:20 -05002555 LOGINFO("Disabling MTP\n");
2556 int status;
2557 kill(mtppid, SIGKILL);
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002558 mtppid = 0;
Ethan Yonker8613dc02014-09-11 09:28:20 -05002559 // We don't care about the exit value, but this prevents a zombie process
2560 waitpid(mtppid, &status, 0);
Ethan Yonker726a0202014-12-16 20:01:38 -06002561 close(mtp_write_fd);
2562 mtp_write_fd = -1;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002563 }
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002564#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002565 property_set("sys.usb.config", "adb");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002566#ifdef TW_HAS_MTP
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002567 DataManager::SetValue("tw_mtp_enabled", 0);
2568 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002569#endif
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002570 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002571}
Ethan Yonker726a0202014-12-16 20:01:38 -06002572
2573TWPartition* TWPartitionManager::Find_Partition_By_MTP_Storage_ID(unsigned int Storage_ID) {
2574 std::vector<TWPartition*>::iterator iter;
2575
2576 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2577 if ((*iter)->MTP_Storage_ID == Storage_ID)
2578 return (*iter);
2579 }
2580 return NULL;
2581}
2582
2583bool TWPartitionManager::Add_Remove_MTP_Storage(TWPartition* Part, int message_type) {
2584#ifdef TW_HAS_MTP
2585 struct mtpmsg mtp_message;
2586
2587 if (!mtppid)
2588 return false; // MTP is disabled
2589
2590 if (mtp_write_fd < 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002591 LOGINFO("MTP: mtp_write_fd is not set\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002592 return false;
2593 }
2594
2595 if (Part) {
Ethan Yonker4bfabab2014-12-29 09:15:37 -06002596 if (Part->MTP_Storage_ID == 0)
2597 return false;
Ethan Yonker726a0202014-12-16 20:01:38 -06002598 if (message_type == MTP_MESSAGE_REMOVE_STORAGE) {
2599 mtp_message.message_type = MTP_MESSAGE_REMOVE_STORAGE; // Remove
2600 LOGINFO("sending message to remove %i\n", Part->MTP_Storage_ID);
2601 mtp_message.storage_id = Part->MTP_Storage_ID;
2602 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002603 LOGINFO("error sending message to remove storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002604 return false;
2605 } else {
2606 LOGINFO("Message sent, remove storage ID: %i\n", Part->MTP_Storage_ID);
2607 return true;
2608 }
2609 } else if (message_type == MTP_MESSAGE_ADD_STORAGE && Part->Is_Mounted()) {
2610 mtp_message.message_type = MTP_MESSAGE_ADD_STORAGE; // Add
2611 mtp_message.storage_id = Part->MTP_Storage_ID;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002612 if (Part->Storage_Path.size() >= sizeof(mtp_message.path)) {
2613 LOGERR("Storage path '%s' too large for mtpmsg\n", Part->Storage_Path.c_str());
2614 return false;
2615 }
2616 strcpy(mtp_message.path, Part->Storage_Path.c_str());
2617 if (Part->Storage_Name.size() >= sizeof(mtp_message.display)) {
2618 LOGERR("Storage name '%s' too large for mtpmsg\n", Part->Storage_Name.c_str());
2619 return false;
2620 }
2621 strcpy(mtp_message.display, Part->Storage_Name.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002622 mtp_message.maxFileSize = Part->Get_Max_FileSize();
Ethan Yonker1b039202015-01-30 10:08:48 -06002623 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 -06002624 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002625 LOGINFO("error sending message to add storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002626 return false;
2627 } else {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002628 LOGINFO("Message sent, add storage ID: %i '%s'\n", Part->MTP_Storage_ID, mtp_message.path);
Ethan Yonker726a0202014-12-16 20:01:38 -06002629 return true;
2630 }
2631 } else {
2632 LOGERR("Unknown MTP message type: %i\n", message_type);
2633 }
2634 } else {
2635 // This hopefully never happens as the error handling should
2636 // occur in the calling function.
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002637 LOGINFO("TWPartitionManager::Add_Remove_MTP_Storage NULL partition given\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002638 }
2639 return true;
2640#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002641 gui_err("no_mtp=MTP support not included");
Ethan Yonker726a0202014-12-16 20:01:38 -06002642 DataManager::SetValue("tw_mtp_enabled", 0);
2643 return false;
2644#endif
2645}
2646
2647bool TWPartitionManager::Add_MTP_Storage(string Mount_Point) {
2648#ifdef TW_HAS_MTP
2649 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2650 if (Part) {
2651 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2652 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002653 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002654 }
2655#endif
2656 return false;
2657}
2658
2659bool TWPartitionManager::Add_MTP_Storage(unsigned int Storage_ID) {
2660#ifdef TW_HAS_MTP
2661 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2662 if (Part) {
2663 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2664 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002665 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002666 }
2667#endif
2668 return false;
2669}
2670
2671bool TWPartitionManager::Remove_MTP_Storage(string Mount_Point) {
2672#ifdef TW_HAS_MTP
2673 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2674 if (Part) {
2675 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2676 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002677 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002678 }
2679#endif
2680 return false;
2681}
2682
2683bool TWPartitionManager::Remove_MTP_Storage(unsigned int Storage_ID) {
2684#ifdef TW_HAS_MTP
2685 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2686 if (Part) {
2687 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2688 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002689 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002690 }
2691#endif
2692 return false;
2693}
Ethan Yonker96af84a2015-01-05 14:58:36 -06002694
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002695bool TWPartitionManager::Flash_Image(string& path, string& filename) {
Matt Mower23d8aae2017-01-06 14:30:33 -06002696 int partition_count = 0;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002697 TWPartition* flash_part = NULL;
bigbiffce8f83c2015-12-12 18:30:21 -05002698 string Flash_List, flash_path, full_filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002699 size_t start_pos = 0, end_pos = 0;
2700
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002701 full_filename = path + "/" + filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002702
bigbiffce8f83c2015-12-12 18:30:21 -05002703 gui_msg("image_flash_start=[IMAGE FLASH STARTED]");
2704 gui_msg(Msg("img_to_flash=Image to flash: '{1}'")(full_filename));
2705
2706 if (!TWFunc::Path_Exists(full_filename)) {
2707 if (!Mount_By_Path(full_filename, true)) {
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002708 return false;
2709 }
bigbiffce8f83c2015-12-12 18:30:21 -05002710 if (!TWFunc::Path_Exists(full_filename)) {
2711 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")(full_filename));
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002712 return false;
2713 }
2714 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002715
Ethan Yonker53796e72019-01-11 22:49:52 -06002716 DataManager::GetValue("tw_flash_partition", Flash_List);
2717 Repack_Type repack = REPLACE_NONE;
2718 if (Flash_List == "/repack_ramdisk;") {
2719 repack = REPLACE_RAMDISK;
2720 } else if (Flash_List == "/repack_kernel;") {
2721 repack = REPLACE_KERNEL;
2722 }
2723 if (repack != REPLACE_NONE) {
2724 Repack_Options_struct Repack_Options;
2725 Repack_Options.Type = repack;
2726 Repack_Options.Disable_Verity = false;
2727 Repack_Options.Disable_Force_Encrypt = false;
2728 Repack_Options.Backup_First = DataManager::GetIntValue("tw_repack_backup_first") != 0;
2729 return Repack_Images(full_filename, Repack_Options);
2730 }
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002731 PartitionSettings part_settings;
2732 part_settings.Backup_Folder = path;
2733 unsigned long long total_bytes = TWFunc::Get_File_Size(full_filename);
2734 ProgressTracking progress(total_bytes);
2735 part_settings.progress = &progress;
2736 part_settings.adbbackup = false;
2737 part_settings.PM_Method = PM_RESTORE;
Ethan Yonker74db1572015-10-28 12:44:49 -05002738 gui_msg("calc_restore=Calculating restore details...");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002739 if (!Flash_List.empty()) {
2740 end_pos = Flash_List.find(";", start_pos);
2741 while (end_pos != string::npos && start_pos < Flash_List.size()) {
2742 flash_path = Flash_List.substr(start_pos, end_pos - start_pos);
2743 flash_part = Find_Partition_By_Path(flash_path);
2744 if (flash_part != NULL) {
2745 partition_count++;
2746 if (partition_count > 1) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002747 gui_err("too_many_flash=Too many partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002748 return false;
2749 }
2750 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002751 gui_msg(Msg(msg::kError, "flash_unable_locate=Unable to locate '{1}' partition for flashing.")(flash_path));
Ethan Yonker96af84a2015-01-05 14:58:36 -06002752 return false;
2753 }
2754 start_pos = end_pos + 1;
2755 end_pos = Flash_List.find(";", start_pos);
2756 }
2757 }
2758
2759 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002760 gui_err("no_part_flash=No partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002761 return false;
2762 }
2763
2764 DataManager::SetProgress(0.0);
2765 if (flash_part) {
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002766 flash_part->Backup_FileName = filename;
2767 if (!flash_part->Flash_Image(&part_settings))
Ethan Yonker96af84a2015-01-05 14:58:36 -06002768 return false;
2769 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002770 gui_err("invalid_flash=Invalid flash partition specified.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002771 return false;
2772 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002773 gui_highlight("flash_done=IMAGE FLASH COMPLETED]");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002774 return true;
2775}
Ethan Yonker74db1572015-10-28 12:44:49 -05002776
2777void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value) {
2778 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2779 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002780 if (part->Is_Adopted_Storage) {
2781 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2782 part->Backup_Display_Name = part->Display_Name;
2783 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2784 } else {
2785 part->Display_Name = gui_lookup(resource_name, default_value);
2786 part->Backup_Display_Name = part->Display_Name;
2787 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002788 }
2789}
2790
2791void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value, const char* storage_resource_name, const char* storage_default_value) {
2792 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2793 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002794 if (part->Is_Adopted_Storage) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06002795 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup("data_backup", "Data (excl. storage)");
Ethan Yonker66a19492015-12-10 10:19:45 -06002796 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
Ethan Yonker66a19492015-12-10 10:19:45 -06002797 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2798 } else {
2799 part->Display_Name = gui_lookup(resource_name, default_value);
2800 part->Backup_Display_Name = part->Display_Name;
2801 if (part->Is_Storage)
2802 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2803 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002804 }
2805}
2806
Ethan Yonker01f4e032017-02-03 15:30:52 -06002807void 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) {
2808 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2809 if (part) {
2810 if (part->Is_Adopted_Storage) {
2811 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup(backup_name, backup_default);
2812 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2813 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2814 } else {
2815 part->Display_Name = gui_lookup(resource_name, default_value);
2816 part->Backup_Display_Name = gui_lookup(backup_name, backup_default);
2817 if (part->Is_Storage)
2818 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2819 }
2820 }
2821}
2822
Ethan Yonker74db1572015-10-28 12:44:49 -05002823void TWPartitionManager::Translate_Partition_Display_Names() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002824 LOGINFO("Translating partition display names\n");
Ethan Yonker74db1572015-10-28 12:44:49 -05002825 Translate_Partition("/system", "system", "System");
2826 Translate_Partition("/system_image", "system_image", "System Image");
2827 Translate_Partition("/vendor", "vendor", "Vendor");
2828 Translate_Partition("/vendor_image", "vendor_image", "Vendor Image");
2829 Translate_Partition("/cache", "cache", "Cache");
Ethan Yonker74db1572015-10-28 12:44:49 -05002830 Translate_Partition("/boot", "boot", "Boot");
2831 Translate_Partition("/recovery", "recovery", "Recovery");
2832 if (!datamedia) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06002833 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage");
Ethan Yonker74db1572015-10-28 12:44:49 -05002834 Translate_Partition("/sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2835 Translate_Partition("/internal_sd", "sdcard", "SDCard", "sdcard", "SDCard");
2836 Translate_Partition("/internal_sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2837 Translate_Partition("/emmc", "sdcard", "SDCard", "sdcard", "SDCard");
Ethan Yonker01f4e032017-02-03 15:30:52 -06002838 } else {
2839 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05002840 }
Ethan Yonker01f4e032017-02-03 15:30:52 -06002841 Translate_Partition("/external_sd", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
2842 Translate_Partition("/external_sdcard", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05002843 Translate_Partition("/usb-otg", "usbotg", "USB OTG", "usbotg", "USB OTG");
2844 Translate_Partition("/sd-ext", "sdext", "SD-EXT");
2845
2846 // Android secure is a special case
2847 TWPartition* part = PartitionManager.Find_Partition_By_Path("/and-sec");
2848 if (part)
2849 part->Backup_Display_Name = gui_lookup("android_secure", "Android Secure");
2850
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002851 std::vector<TWPartition*>::iterator sysfs;
2852 for (sysfs = Partitions.begin(); sysfs != Partitions.end(); sysfs++) {
2853 if (!(*sysfs)->Sysfs_Entry.empty()) {
2854 Translate_Partition((*sysfs)->Mount_Point.c_str(), "autostorage", "Storage", "autostorage", "Storage");
2855 }
2856 }
2857
Ethan Yonker74db1572015-10-28 12:44:49 -05002858 // This updates the text on all of the storage selection buttons in the GUI
2859 DataManager::SetBackupFolder();
2860}
Ethan Yonker66a19492015-12-10 10:19:45 -06002861
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002862bool TWPartitionManager::Decrypt_Adopted() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002863#ifdef TW_INCLUDE_CRYPTO
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002864 bool ret = false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002865 if (!Mount_By_Path("/data", false)) {
2866 LOGERR("Cannot decrypt adopted storage because /data will not mount\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002867 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002868 }
2869 LOGINFO("Decrypt adopted storage starting\n");
2870 char* xmlFile = PageManager::LoadFileToBuffer("/data/system/storage.xml", NULL);
2871 xml_document<> *doc = NULL;
2872 xml_node<>* volumes = NULL;
Ethan Yonker66a19492015-12-10 10:19:45 -06002873 string Primary_Storage_UUID = "";
2874 if (xmlFile != NULL) {
2875 LOGINFO("successfully loaded storage.xml\n");
2876 doc = new xml_document<>();
2877 doc->parse<0>(xmlFile);
2878 volumes = doc->first_node("volumes");
2879 if (volumes) {
2880 xml_attribute<>* psuuid = volumes->first_attribute("primaryStorageUuid");
2881 if (psuuid) {
2882 Primary_Storage_UUID = psuuid->value();
2883 }
2884 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002885 } else {
2886 LOGINFO("No /data/system/storage.xml for adopted storage\n");
2887 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002888 }
2889 std::vector<TWPartition*>::iterator adopt;
2890 for (adopt = Partitions.begin(); adopt != Partitions.end(); adopt++) {
Peter Caic50044b2019-10-17 08:49:10 +08002891 if ((*adopt)->Removable && !(*adopt)->Is_Present && (*adopt)->Adopted_Mount_Delay > 0) {
2892 // On some devices, the external mmc driver takes some time
2893 // to recognize the card, in which case the "actual block device"
2894 // would not have been found yet. We wait the specified delay
2895 // and then try again.
2896 LOGINFO("Sleeping %d seconds for adopted storage.\n", (*adopt)->Adopted_Mount_Delay);
2897 sleep((*adopt)->Adopted_Mount_Delay);
2898 (*adopt)->Find_Actual_Block_Device();
2899 }
2900
Ethan Yonker66a19492015-12-10 10:19:45 -06002901 if ((*adopt)->Removable && (*adopt)->Is_Present) {
2902 if ((*adopt)->Decrypt_Adopted() == 0) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002903 ret = true;
Ethan Yonker66a19492015-12-10 10:19:45 -06002904 if (volumes) {
2905 xml_node<>* volume = volumes->first_node("volume");
2906 while (volume) {
2907 xml_attribute<>* guid = volume->first_attribute("partGuid");
2908 if (guid) {
2909 string GUID = (*adopt)->Adopted_GUID.c_str();
2910 GUID.insert(8, "-");
2911 GUID.insert(13, "-");
2912 GUID.insert(18, "-");
2913 GUID.insert(23, "-");
2914
2915 if (strcasecmp(GUID.c_str(), guid->value()) == 0) {
2916 xml_attribute<>* attr = volume->first_attribute("nickname");
nkk71ffb02bd2017-06-04 23:12:16 +03002917 if (attr && attr->value() && strlen(attr->value()) > 0) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002918 (*adopt)->Storage_Name = attr->value();
2919 (*adopt)->Display_Name = (*adopt)->Storage_Name;
2920 (*adopt)->Backup_Display_Name = (*adopt)->Storage_Name;
2921 LOGINFO("storage name from storage.xml is '%s'\n", attr->value());
2922 }
2923 attr = volume->first_attribute("fsUuid");
2924 if (attr && !Primary_Storage_UUID.empty() && strcmp(Primary_Storage_UUID.c_str(), attr->value()) == 0) {
2925 TWPartition* Dat = Find_Partition_By_Path("/data");
2926 if (Dat) {
2927 LOGINFO("Internal storage is found on adopted storage '%s'\n", (*adopt)->Display_Name.c_str());
2928 LOGINFO("Changing '%s' to point to '%s'\n", Dat->Symlink_Mount_Point.c_str(), (*adopt)->Storage_Path.c_str());
2929 (*adopt)->Symlink_Mount_Point = Dat->Symlink_Mount_Point;
2930 Dat->Symlink_Mount_Point = "";
2931 // Toggle mounts to ensure that the symlink mount point (probably /sdcard) is mounted to the right location
2932 Dat->UnMount(false);
2933 Dat->Mount(false);
2934 (*adopt)->UnMount(false);
2935 (*adopt)->Mount(false);
Ethan Yonker66a19492015-12-10 10:19:45 -06002936 }
2937 }
2938 break;
2939 }
2940 }
2941 volume = volume->next_sibling("volume");
2942 }
2943 }
nkk71ffb02bd2017-06-04 23:12:16 +03002944 Update_System_Details();
2945 Output_Partition((*adopt));
Ethan Yonker66a19492015-12-10 10:19:45 -06002946 }
2947 }
2948 }
2949 if (xmlFile) {
2950 doc->clear();
2951 delete doc;
2952 free(xmlFile);
2953 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002954 return ret;
Ethan Yonker66a19492015-12-10 10:19:45 -06002955#else
2956 LOGINFO("Decrypt_Adopted: no crypto support\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002957 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002958#endif
2959}
Ethan Yonkerfcf3f242016-02-16 12:30:26 -06002960
2961void TWPartitionManager::Remove_Partition_By_Path(string Path) {
2962 std::vector<TWPartition*>::iterator iter;
2963 string Local_Path = TWFunc::Get_Root_Path(Path);
2964
2965 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2966 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
2967 LOGINFO("Found and erasing '%s' from partition list\n", Local_Path.c_str());
2968 Partitions.erase(iter);
2969 return;
2970 }
2971 }
2972}
Ethan Yonker1b190162016-12-05 15:25:19 -06002973
2974void TWPartitionManager::Set_Active_Slot(const string& Slot) {
2975 if (Slot != "A" && Slot != "B") {
2976 LOGERR("Set_Active_Slot invalid slot '%s'\n", Slot.c_str());
2977 return;
2978 }
2979 if (Active_Slot_Display == Slot)
2980 return;
2981 LOGINFO("Setting active slot %s\n", Slot.c_str());
2982#ifdef AB_OTA_UPDATER
2983 if (!Active_Slot_Display.empty()) {
2984 const hw_module_t *hw_module;
2985 boot_control_module_t *module;
2986 int ret;
2987 ret = hw_get_module("bootctrl", &hw_module);
2988 if (ret != 0) {
2989 LOGERR("Error getting bootctrl module.\n");
2990 } else {
2991 module = (boot_control_module_t*) hw_module;
2992 module->init(module);
2993 int slot_number = 0;
2994 if (Slot == "B")
2995 slot_number = 1;
2996 if (module->setActiveBootSlot(module, slot_number))
2997 gui_msg(Msg(msg::kError, "unable_set_boot_slot=Error changing bootloader boot slot to {1}")(Slot));
2998 }
2999 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
3000 }
3001#else
3002 LOGERR("Boot slot feature not present\n");
3003#endif
3004 Active_Slot_Display = Slot;
3005 if (Fstab_Processed())
3006 Update_System_Details();
3007}
3008string TWPartitionManager::Get_Active_Slot_Suffix() {
3009 if (Active_Slot_Display == "A")
3010 return "_a";
3011 return "_b";
3012}
3013string TWPartitionManager::Get_Active_Slot_Display() {
3014 return Active_Slot_Display;
3015}
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003016
Captain Throwback9d6feb52018-07-27 10:05:24 -04003017string TWPartitionManager::Get_Android_Root_Path() {
Chaosmasterf6e42ce2020-01-27 00:17:04 +01003018 if (property_get_bool("ro.twrp.sar", false))
dianlujitao58f1a632020-01-16 23:03:56 +08003019 return "/system_root";
3020 return "/system";
Captain Throwback9d6feb52018-07-27 10:05:24 -04003021}
3022
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003023void TWPartitionManager::Remove_Uevent_Devices(const string& Mount_Point) {
3024 std::vector<TWPartition*>::iterator iter;
3025
3026 for (iter = Partitions.begin(); iter != Partitions.end(); ) {
3027 if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Mount_Point) {
3028 TWPartition *part = *iter;
3029 LOGINFO("%s was removed by uevent data\n", (*iter)->Mount_Point.c_str());
3030 (*iter)->UnMount(false);
3031 rmdir((*iter)->Mount_Point.c_str());
3032 iter = Partitions.erase(iter);
3033 delete part;
3034 } else {
3035 iter++;
3036 }
3037 }
3038}
3039
3040void TWPartitionManager::Handle_Uevent(const Uevent_Block_Data& uevent_data) {
3041 std::vector<TWPartition*>::iterator iter;
3042
3043 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3044 if (!(*iter)->Sysfs_Entry.empty()) {
3045 string device;
3046 size_t wildcard = (*iter)->Sysfs_Entry.find("*");
3047 if (wildcard != string::npos) {
3048 device = (*iter)->Sysfs_Entry.substr(0, wildcard);
3049 } else {
3050 device = (*iter)->Sysfs_Entry;
3051 }
3052 if (device == uevent_data.sysfs_path.substr(0, device.size())) {
3053 // Found a match
3054 if (uevent_data.action == "add") {
3055 (*iter)->Primary_Block_Device = "/dev/block/" + uevent_data.block_device;
3056 (*iter)->Alternate_Block_Device = (*iter)->Primary_Block_Device;
3057 (*iter)->Is_Present = true;
3058 LOGINFO("Found a match '%s' '%s'\n", uevent_data.block_device.c_str(), device.c_str());
3059 if (!Decrypt_Adopted()) {
3060 LOGINFO("No adopted storage so finding actual block device\n");
3061 (*iter)->Find_Actual_Block_Device();
3062 }
3063 return;
3064 } else if (uevent_data.action == "remove") {
3065 (*iter)->Is_Present = false;
3066 (*iter)->Primary_Block_Device = "";
3067 (*iter)->Actual_Block_Device = "";
3068 Remove_Uevent_Devices((*iter)->Mount_Point);
3069 return;
3070 }
3071 }
3072 }
3073 }
3074 LOGINFO("Found no matching fstab entry for uevent device '%s' - %s\n", uevent_data.sysfs_path.c_str(), uevent_data.action.c_str());
3075}
3076
3077void TWPartitionManager::setup_uevent() {
3078 struct sockaddr_nl nls;
3079
3080 if (uevent_pfd.fd >= 0) {
3081 LOGINFO("uevent already set up\n");
3082 return;
3083 }
3084
3085 // Open hotplug event netlink socket
3086 memset(&nls,0,sizeof(struct sockaddr_nl));
3087 nls.nl_family = AF_NETLINK;
3088 nls.nl_pid = getpid();
3089 nls.nl_groups = -1;
3090 uevent_pfd.events = POLLIN;
3091 uevent_pfd.fd = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_KOBJECT_UEVENT);
3092 if (uevent_pfd.fd==-1) {
3093 LOGERR("uevent not root\n");
3094 return;
3095 }
3096
3097 // Listen to netlink socket
3098 if (::bind(uevent_pfd.fd, (struct sockaddr *) &nls, sizeof(struct sockaddr_nl)) < 0) {
3099 LOGERR("Bind failed\n");
3100 return;
3101 }
3102 set_select_fd();
3103 Coldboot();
3104}
3105
3106Uevent_Block_Data TWPartitionManager::get_event_block_values(char *buf, int len) {
3107 Uevent_Block_Data ret;
3108 ret.subsystem = "";
3109 char *ptr = buf;
3110 const char *end = buf + len;
3111
3112 buf[len - 1] = '\0';
3113 while (ptr < end) {
3114 if (strncmp(ptr, "ACTION=", strlen("ACTION=")) == 0) {
3115 ptr += strlen("ACTION=");
3116 ret.action = ptr;
3117 } else if (strncmp(ptr, "SUBSYSTEM=", strlen("SUBSYSTEM=")) == 0) {
3118 ptr += strlen("SUBSYSTEM=");
3119 ret.subsystem = ptr;
3120 } else if (strncmp(ptr, "DEVTYPE=", strlen("DEVTYPE=")) == 0) {
3121 ptr += strlen("DEVTYPE=");
3122 ret.type = ptr;
3123 } else if (strncmp(ptr, "DEVPATH=", strlen("DEVPATH=")) == 0) {
3124 ptr += strlen("DEVPATH=");
3125 ret.sysfs_path += ptr;
3126 } else if (strncmp(ptr, "DEVNAME=", strlen("DEVNAME=")) == 0) {
3127 ptr += strlen("DEVNAME=");
3128 ret.block_device += ptr;
3129 } else if (strncmp(ptr, "MAJOR=", strlen("MAJOR=")) == 0) {
3130 ptr += strlen("MAJOR=");
3131 ret.major = atoi(ptr);
3132 } else if (strncmp(ptr, "MINOR=", strlen("MINOR=")) == 0) {
3133 ptr += strlen("MINOR=");
3134 ret.minor = atoi(ptr);
3135 }
3136 ptr += strlen(ptr) + 1;
3137 }
3138 return ret;
3139}
3140
3141void TWPartitionManager::read_uevent() {
3142 char buf[1024];
3143
3144 int len = recv(uevent_pfd.fd, buf, sizeof(buf), MSG_DONTWAIT);
3145 if (len == -1) {
Mauronofrio Matarrese9632f302019-08-04 17:02:41 +01003146 LOGINFO("recv error on uevent\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003147 return;
3148 }
3149 /*int i = 0; // Print all uevent output for test /debug
3150 while (i<len) {
3151 printf("%s\n", buf+i);
3152 i += strlen(buf+i)+1;
3153 }*/
3154 Uevent_Block_Data uevent_data = get_event_block_values(buf, len);
3155 if (uevent_data.subsystem == "block" && uevent_data.type == "disk") {
3156 PartitionManager.Handle_Uevent(uevent_data);
3157 }
3158}
3159
3160void TWPartitionManager::close_uevent() {
3161 if (uevent_pfd.fd > 0)
3162 close(uevent_pfd.fd);
3163 uevent_pfd.fd = -1;
3164}
3165
3166void TWPartitionManager::Add_Partition(TWPartition* Part) {
3167 Partitions.push_back(Part);
3168}
3169
3170void TWPartitionManager::Coldboot_Scan(std::vector<string> *sysfs_entries, const string& Path, int depth) {
3171 string Real_Path = Path;
3172 char real_path[PATH_MAX];
3173 if (realpath(Path.c_str(), &real_path[0])) {
3174 string Real_Path = real_path;
3175 std::vector<string>::iterator iter;
3176 for (iter = sysfs_entries->begin(); iter != sysfs_entries->end(); iter++) {
3177 if (Real_Path.find((*iter)) != string::npos) {
3178 string Write_Path = Real_Path + "/uevent";
3179 if (TWFunc::Path_Exists(Write_Path)) {
3180 const char* write_val = "add\n";
3181 TWFunc::write_to_file(Write_Path, write_val);
3182 break;
3183 }
3184 }
3185 }
3186 }
3187
3188 DIR* d = opendir(Path.c_str());
3189 if (d != NULL) {
3190 struct dirent* de;
3191 while ((de = readdir(d)) != NULL) {
3192 if (de->d_name[0] == '.' || (de->d_type != DT_DIR && depth > 0))
3193 continue;
3194 if (strlen(de->d_name) >= 4 && (strncmp(de->d_name, "ram", 3) == 0 || strncmp(de->d_name, "loop", 4) == 0))
3195 continue;
3196
3197 string item = Path + "/";
3198 item.append(de->d_name);
3199 Coldboot_Scan(sysfs_entries, item, depth + 1);
3200 }
3201 closedir(d);
3202 }
3203}
3204
3205void TWPartitionManager::Coldboot() {
3206 std::vector<TWPartition*>::iterator iter;
3207 std::vector<string> sysfs_entries;
3208
3209 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3210 if (!(*iter)->Sysfs_Entry.empty()) {
3211 size_t wildcard_pos = (*iter)->Sysfs_Entry.find("*");
3212 if (wildcard_pos == string::npos)
3213 wildcard_pos = (*iter)->Sysfs_Entry.size();
3214 sysfs_entries.push_back((*iter)->Sysfs_Entry.substr(0, wildcard_pos));
3215 }
3216 }
3217
3218 if (sysfs_entries.size() > 0)
3219 Coldboot_Scan(&sysfs_entries, "/sys/block", 0);
3220}
Ethan Yonker53796e72019-01-11 22:49:52 -06003221
3222bool TWPartitionManager::Prepare_Empty_Folder(const std::string& Folder) {
3223 if (TWFunc::Path_Exists(Folder))
3224 TWFunc::removeDir(Folder, false);
3225 return TWFunc::Recursive_Mkdir(Folder);
3226}
3227
3228bool TWPartitionManager::Prepare_Repack(TWPartition* Part, const std::string& Temp_Folder_Destination, const bool Create_Backup, const std::string& Backup_Name) {
3229 if (!Part) {
3230 LOGERR("Partition was null!\n");
3231 return false;
3232 }
3233 if (!Prepare_Empty_Folder(Temp_Folder_Destination))
3234 return false;
3235 std::string target_image = Temp_Folder_Destination + "boot.img";
3236 PartitionSettings part_settings;
3237 part_settings.Part = Part;
3238 if (Create_Backup) {
3239 if (Check_Backup_Name(Backup_Name, true, false) != 0)
3240 return false;
3241 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, part_settings.Backup_Folder);
3242 part_settings.Backup_Folder = part_settings.Backup_Folder + "/" + TWFunc::Get_Current_Date() + " " + Backup_Name + "/";
3243 if (!TWFunc::Recursive_Mkdir(part_settings.Backup_Folder))
3244 return false;
3245 } else
3246 part_settings.Backup_Folder = Temp_Folder_Destination;
3247 part_settings.adbbackup = false;
3248 part_settings.generate_digest = false;
3249 part_settings.generate_md5 = false;
3250 part_settings.PM_Method = PM_BACKUP;
3251 part_settings.progress = NULL;
3252 pid_t not_a_pid = 0;
3253 if (!Part->Backup(&part_settings, &not_a_pid))
3254 return false;
3255 std::string backed_up_image = part_settings.Backup_Folder;
3256 backed_up_image += Part->Backup_FileName;
3257 target_image = Temp_Folder_Destination + "boot.img";
3258 if (Create_Backup) {
3259 std::string source = part_settings.Backup_Folder + Part->Backup_FileName;
3260 if (TWFunc::copy_file(source, target_image, 0644) != 0) {
3261 LOGERR("Failed to copy backup file '%s' to temp folder target '%s'\n", source.c_str(), target_image.c_str());
3262 return false;
3263 }
3264 } else {
3265 if (rename(backed_up_image.c_str(), target_image.c_str()) != 0) {
3266 LOGERR("Failed to rename '%s' to '%s'\n", backed_up_image.c_str(), target_image.c_str());
3267 return false;
3268 }
3269 }
3270 return Prepare_Repack(target_image, Temp_Folder_Destination, false, false);
3271}
3272
3273bool TWPartitionManager::Prepare_Repack(const std::string& Source_Path, const std::string& Temp_Folder_Destination, const bool Copy_Source, const bool Create_Destination) {
3274 if (Create_Destination) {
3275 if (!Prepare_Empty_Folder(Temp_Folder_Destination))
3276 return false;
3277 }
3278 if (Copy_Source) {
3279 std::string destination = Temp_Folder_Destination + "/boot.img";
3280 if (TWFunc::copy_file(Source_Path, destination, 0644))
3281 return false;
3282 }
Mohd Faraz5738e762020-05-10 04:18:30 +05303283 std::string command = "cd " + Temp_Folder_Destination + " && /sbin/magiskboot unpack -h '" + Source_Path +"'";
Ethan Yonker53796e72019-01-11 22:49:52 -06003284 if (TWFunc::Exec_Cmd(command) != 0) {
3285 LOGINFO("Error unpacking %s!\n", Source_Path.c_str());
3286 gui_msg(Msg(msg::kError, "unpack_error=Error unpacking image."));
3287 return false;
3288 }
3289 return true;
3290}
3291
3292bool TWPartitionManager::Repack_Images(const std::string& Target_Image, const struct Repack_Options_struct& Repack_Options) {
3293 if (!TWFunc::Path_Exists("/sbin/magiskboot")) {
3294 LOGERR("Image repacking tool not present in this TWRP build!");
3295 return false;
3296 }
3297 DataManager::SetProgress(0);
3298 TWPartition* part = PartitionManager.Find_Partition_By_Path("/boot");
3299 if (part)
3300 gui_msg(Msg("unpacking_image=Unpacking {1}...")(part->Display_Name));
3301 else {
3302 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/boot"));
3303 return false;
3304 }
3305 if (!PartitionManager.Prepare_Repack(part, REPACK_ORIG_DIR, Repack_Options.Backup_First, gui_lookup("repack", "Repack")))
3306 return false;
3307 DataManager::SetProgress(.25);
3308 gui_msg(Msg("unpacking_image=Unpacking {1}...")(Target_Image));
3309 if (!PartitionManager.Prepare_Repack(Target_Image, REPACK_NEW_DIR, true))
3310 return false;
3311 DataManager::SetProgress(.5);
3312 gui_msg(Msg("repacking_image=Repacking {1}...")(part->Display_Name));
3313 std::string path = REPACK_NEW_DIR;
3314 if (Repack_Options.Type == REPLACE_KERNEL) {
3315 // When we replace the kernel, what we really do is copy the boot partition ramdisk into the new image's folder
3316 if (TWFunc::copy_file(REPACK_ORIG_DIR "ramdisk.cpio", REPACK_NEW_DIR "ramdisk.cpio", 0644)) {
3317 LOGERR("Failed to copy ramdisk\n");
3318 return false;
3319 }
3320 } else if (Repack_Options.Type == REPLACE_RAMDISK) {
3321 // Repack the ramdisk
3322 if (TWFunc::copy_file(REPACK_NEW_DIR "ramdisk.cpio", REPACK_ORIG_DIR "ramdisk.cpio", 0644)) {
3323 LOGERR("Failed to copy ramdisk\n");
3324 return false;
3325 }
3326 path = REPACK_ORIG_DIR;
3327 } else {
3328 LOGERR("Invalid repacking options specified\n");
3329 return false;
3330 }
3331 if (Repack_Options.Disable_Verity)
3332 LOGERR("Disabling verity is not implemented yet\n");
3333 if (Repack_Options.Disable_Force_Encrypt)
3334 LOGERR("Disabling force encrypt is not implemented yet\n");
Mohd Faraz5738e762020-05-10 04:18:30 +05303335 std::string command = "cd " + path + " && /sbin/magiskboot repack " + path + "boot.img";
Ethan Yonker53796e72019-01-11 22:49:52 -06003336 if (TWFunc::Exec_Cmd(command) != 0) {
3337 gui_msg(Msg(msg::kError, "repack_error=Error repacking image."));
3338 return false;
3339 }
3340 DataManager::SetProgress(.75);
3341 std::string file = "new-boot.img";
3342 DataManager::SetValue("tw_flash_partition", "/boot;");
3343 if (!PartitionManager.Flash_Image(path, file)) {
3344 LOGINFO("Error flashing new image\n");
3345 return false;
3346 }
3347 DataManager::SetProgress(1);
3348 TWFunc::removeDir(REPACK_ORIG_DIR, false);
Ethan Yonker9f5dd312019-05-15 15:17:36 -05003349 if (part->SlotSelect && Repack_Options.Type == REPLACE_RAMDISK) {
3350 LOGINFO("Switching slots to flash ramdisk to both partitions\n");
3351 string Current_Slot = Get_Active_Slot_Display();
3352 if (Current_Slot == "A")
3353 Set_Active_Slot("B");
3354 else
3355 Set_Active_Slot("A");
3356 DataManager::SetProgress(.25);
3357 if (!PartitionManager.Prepare_Repack(part, REPACK_ORIG_DIR, Repack_Options.Backup_First, gui_lookup("repack", "Repack")))
3358 return false;
3359 if (TWFunc::copy_file(REPACK_NEW_DIR "ramdisk.cpio", REPACK_ORIG_DIR "ramdisk.cpio", 0644)) {
3360 LOGERR("Failed to copy ramdisk\n");
3361 return false;
3362 }
3363 path = REPACK_ORIG_DIR;
Mohd Faraz5738e762020-05-10 04:18:30 +05303364 command = "cd " + path + " && /sbin/magiskboot repack " + path + "boot.img";
Ethan Yonker9f5dd312019-05-15 15:17:36 -05003365 if (TWFunc::Exec_Cmd(command) != 0) {
3366 gui_msg(Msg(msg::kError, "repack_error=Error repacking image."));
3367 return false;
3368 }
3369 DataManager::SetProgress(.75);
3370 std::string file = "new-boot.img";
3371 DataManager::SetValue("tw_flash_partition", "/boot;");
3372 if (!PartitionManager.Flash_Image(path, file)) {
3373 LOGINFO("Error flashing new image\n");
3374 return false;
3375 }
3376 DataManager::SetProgress(1);
3377 TWFunc::removeDir(REPACK_ORIG_DIR, false);
3378 Set_Active_Slot(Current_Slot);
3379 }
Ethan Yonker53796e72019-01-11 22:49:52 -06003380 TWFunc::removeDir(REPACK_NEW_DIR, false);
3381 return true;
3382}