blob: 3b0e8a240fc98d11ab84450ad012df84e62a5a29 [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 Jacobson0b25b1d2019-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 Jacobson0b25b1d2019-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
Chaosmasterf6e42ce2020-01-27 00:17:04 +0100115int TWPartitionManager::Process_Fstab(string Fstab_Filename, bool Display_Error, bool Sar_Detect) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400116 FILE *fstabFile;
117 char fstab_line[MAX_FSTAB_LINE_LENGTH];
Dees Troy02a64532014-03-19 15:23:32 +0000118 TWPartition* settings_partition = NULL;
Matt Mowerbf4efa32014-04-14 23:25:26 -0500119 TWPartition* andsec_partition = NULL;
Ethan Yonker726a0202014-12-16 20:01:38 -0600120 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 -0600121 std::map<string, Flags_Map> twrp_flags;
122
123 fstabFile = fopen("/etc/twrp.flags", "rt");
124 if (fstabFile != NULL) {
125 LOGINFO("reading /etc/twrp.flags\n");
126 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
127 if (fstab_line[0] != '/')
128 continue;
129
130 size_t line_size = strlen(fstab_line);
131 if (fstab_line[line_size - 1] != '\n')
132 fstab_line[line_size] = '\n';
133 Flags_Map line_flags;
134 line_flags.Primary_Block_Device = "";
135 line_flags.Alternate_Block_Device = "";
136 line_flags.fstab_line = (char*)malloc(MAX_FSTAB_LINE_LENGTH);
137 if (!line_flags.fstab_line) {
138 LOGERR("malloc error on line_flags.fstab_line\n");
139 return false;
140 }
141 memcpy(line_flags.fstab_line, fstab_line, MAX_FSTAB_LINE_LENGTH);
142 bool found_separator = false;
143 char *fs_loc = NULL;
144 char *block_loc = NULL;
145 char *flags_loc = NULL;
146 size_t index, item_index = 0;
147 for (index = 0; index < line_size; index++) {
148 if (fstab_line[index] <= 32) {
149 fstab_line[index] = '\0';
150 found_separator = true;
151 } else if (found_separator) {
152 if (item_index == 0) {
153 fs_loc = fstab_line + index;
154 } else if (item_index == 1) {
155 block_loc = fstab_line + index;
156 } else if (item_index > 1) {
157 char *ptr = fstab_line + index;
158 if (*ptr == '/') {
159 line_flags.Alternate_Block_Device = ptr;
160 } else if (strlen(ptr) > strlen("flags=") && strncmp(ptr, "flags=", strlen("flags=")) == 0) {
161 flags_loc = ptr;
162 // Once we find the flags=, we're done scanning the line
163 break;
164 }
165 }
166 found_separator = false;
167 item_index++;
168 }
169 }
170 if (block_loc)
171 line_flags.Primary_Block_Device = block_loc;
172 if (fs_loc)
173 line_flags.File_System = fs_loc;
174 if (flags_loc)
175 line_flags.Flags = flags_loc;
176 string Mount_Point = fstab_line;
177 twrp_flags[Mount_Point] = line_flags;
178 memset(fstab_line, 0, sizeof(fstab_line));
179 }
180 fclose(fstabFile);
181 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400182
183 fstabFile = fopen(Fstab_Filename.c_str(), "rt");
184 if (fstabFile == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000185 LOGERR("Critical Error: Unable to open fstab at '%s'.\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400186 return false;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600187 } else
188 LOGINFO("Reading %s\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400189
190 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
191 if (fstab_line[0] != '/')
192 continue;
193
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600194 if (strstr(fstab_line, "swap"))
195 continue; // Skip swap in recovery
196
197 size_t line_size = strlen(fstab_line);
198 if (fstab_line[line_size - 1] != '\n')
199 fstab_line[line_size] = '\n';
Dees_Troy2a923582012-09-20 12:13:34 -0400200
Matt Mower2b2dd152016-04-26 11:24:08 -0500201 TWPartition* partition = new TWPartition();
Chaosmasterf6e42ce2020-01-27 00:17:04 +0100202 if (partition->Process_Fstab_Line(fstab_line, Display_Error, &twrp_flags, Sar_Detect))
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500203 Partitions.push_back(partition);
Matt Mower72c87ce2016-04-26 14:34:56 -0500204 else
Dees_Troy5bf43922012-09-07 16:07:55 -0400205 delete partition;
Matt Mower2b2dd152016-04-26 11:24:08 -0500206
207 memset(fstab_line, 0, sizeof(fstab_line));
Dees_Troy5bf43922012-09-07 16:07:55 -0400208 }
209 fclose(fstabFile);
Matt Mower72c87ce2016-04-26 14:34:56 -0500210
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600211 if (twrp_flags.size() > 0) {
212 LOGINFO("Processing remaining twrp.flags\n");
213 // Add any items from twrp.flags that did not exist in the recovery.fstab
214 for (std::map<string, Flags_Map>::iterator mapit=twrp_flags.begin(); mapit!=twrp_flags.end(); mapit++) {
215 if (Find_Partition_By_Path(mapit->first) == NULL) {
216 TWPartition* partition = new TWPartition();
Chaosmasterf6e42ce2020-01-27 00:17:04 +0100217 if (partition->Process_Fstab_Line(mapit->second.fstab_line, Display_Error, NULL, Sar_Detect))
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600218 Partitions.push_back(partition);
219 else
220 delete partition;
221 }
222 if (mapit->second.fstab_line)
223 free(mapit->second.fstab_line);
224 mapit->second.fstab_line = NULL;
225 }
226 }
227 LOGINFO("Done processing fstab files\n");
228
Matt Mower72c87ce2016-04-26 14:34:56 -0500229 std::vector<TWPartition*>::iterator iter;
230 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Chaosmasterf6e42ce2020-01-27 00:17:04 +0100231 if (Sar_Detect) {
232 if ((*iter)->Mount_Point == "/s")
233 return true;
234 else
235 continue;
236 }
237
Matt Mower72c87ce2016-04-26 14:34:56 -0500238 (*iter)->Partition_Post_Processing(Display_Error);
239
240 if ((*iter)->Is_Storage) {
241 ++storageid;
242 (*iter)->MTP_Storage_ID = storageid;
243 }
244
245 if (!settings_partition && (*iter)->Is_Settings_Storage && (*iter)->Is_Present)
246 settings_partition = (*iter);
247 else
248 (*iter)->Is_Settings_Storage = false;
249
250 if (!andsec_partition && (*iter)->Has_Android_Secure && (*iter)->Is_Present)
251 andsec_partition = (*iter);
252 else
253 (*iter)->Has_Android_Secure = false;
254 }
255
Ethan Yonker6277c792014-09-15 14:54:30 -0500256 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) {
257 // Attempt to automatically identify /data/media emulated storage devices
258 TWPartition* Dat = Find_Partition_By_Path("/data");
259 if (Dat) {
260 LOGINFO("Using automatic handling for /data/media emulated storage device.\n");
261 datamedia = true;
that9e0593e2014-10-08 00:01:24 +0200262 Dat->Setup_Data_Media();
Ethan Yonker6277c792014-09-15 14:54:30 -0500263 settings_partition = Dat;
Ethan Yonker726a0202014-12-16 20:01:38 -0600264 // Since /data was not considered a storage partition earlier, we still need to assign an MTP ID
265 ++storageid;
266 Dat->MTP_Storage_ID = storageid;
Ethan Yonker6277c792014-09-15 14:54:30 -0500267 }
268 }
Dees Troy02a64532014-03-19 15:23:32 +0000269 if (!settings_partition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500270 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
271 if ((*iter)->Is_Storage) {
Dees Troy02a64532014-03-19 15:23:32 +0000272 settings_partition = (*iter);
Dees_Troya13d74f2013-03-24 08:54:55 -0500273 break;
274 }
275 }
Dees Troy02a64532014-03-19 15:23:32 +0000276 if (!settings_partition)
Dees_Troy2673cec2013-04-02 20:22:16 +0000277 LOGERR("Unable to locate storage partition for storing settings file.\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500278 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400279 if (!Write_Fstab()) {
280 if (Display_Error)
Dees_Troy2673cec2013-04-02 20:22:16 +0000281 LOGERR("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400282 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000283 LOGINFO("Error creating fstab\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400284 }
Matt Mowered71fa32014-04-16 13:21:47 -0500285
Matt Mowerbf4efa32014-04-14 23:25:26 -0500286 if (andsec_partition) {
287 Setup_Android_Secure_Location(andsec_partition);
Matt Mowered71fa32014-04-16 13:21:47 -0500288 } else if (settings_partition) {
Matt Mowerbf4efa32014-04-14 23:25:26 -0500289 Setup_Android_Secure_Location(settings_partition);
290 }
Matt Mowered71fa32014-04-16 13:21:47 -0500291 if (settings_partition) {
292 Setup_Settings_Storage_Partition(settings_partition);
293 }
Ethan Yonker253368a2014-11-25 15:00:52 -0600294#ifdef TW_INCLUDE_CRYPTO
Ethan Yonkercceebb82014-11-18 10:17:59 -0600295 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
296 if (Decrypt_Data && Decrypt_Data->Is_Encrypted && !Decrypt_Data->Is_Decrypted) {
Ethan Yonker93382822018-11-01 15:25:31 -0500297 if (!Decrypt_Data->Key_Directory.empty() && Mount_By_Path(Decrypt_Data->Key_Directory, false)) {
298#ifdef TW_INCLUDE_FBE_METADATA_DECRYPT
299 if (e4crypt_mount_metadata_encrypted(Decrypt_Data->Mount_Point, false, Decrypt_Data->Key_Directory, Decrypt_Data->Actual_Block_Device, &Decrypt_Data->Decrypted_Block_Device)) {
300 LOGINFO("Successfully decrypted metadata encrypted data partition with new block device: '%s'\n", Decrypt_Data->Decrypted_Block_Device.c_str());
301 property_set("ro.crypto.state", "encrypted");
302 Decrypt_Data->Is_Decrypted = true; // Needed to make the mount function work correctly
303 int retry_count = 10;
304 while (!Decrypt_Data->Mount(false) && --retry_count)
305 usleep(500);
306 if (Decrypt_Data->Mount(false)) {
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400307 if (!Decrypt_Data->Decrypt_FBE_DE()) {
mauronofriofc79aa62019-11-23 23:13:04 +0100308 char wrappedvalue[PROPERTY_VALUE_MAX];
309 property_get("fbe.data.wrappedkey", wrappedvalue, "");
310 std::string wrappedkeyvalue(wrappedvalue);
311 if (wrappedkeyvalue == "true") {
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400312 LOGERR("Unable to decrypt FBE device\n");
mauronofriofc79aa62019-11-23 23:13:04 +0100313 } else {
314 LOGINFO("Trying wrapped key.\n");
315 property_set("fbe.data.wrappedkey", "true");
316 if (!Decrypt_Data->Decrypt_FBE_DE()) {
317 LOGERR("Unable to decrypt FBE device\n");
318 }
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400319 }
320 }
321
Ethan Yonker93382822018-11-01 15:25:31 -0500322 } else {
323 LOGINFO("Failed to mount data after metadata decrypt\n");
324 }
325 } else {
326 LOGINFO("Unable to decrypt metadata encryption\n");
327 }
328#else
329 LOGERR("Metadata FBE decrypt support not present in this TWRP\n");
330#endif
331 }
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600332 if (Decrypt_Data->Is_FBE) {
333 if (DataManager::GetIntValue(TW_CRYPTO_PWTYPE) == 0) {
334 if (Decrypt_Device("!") == 0) {
335 gui_msg("decrypt_success=Successfully decrypted with default password.");
336 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
337 } else {
338 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
339 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600340 }
341 } else {
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600342 int password_type = cryptfs_get_password_type();
343 if (password_type == CRYPT_TYPE_DEFAULT) {
344 LOGINFO("Device is encrypted with the default password, attempting to decrypt.\n");
345 if (Decrypt_Device("default_password") == 0) {
346 gui_msg("decrypt_success=Successfully decrypted with default password.");
347 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
348 } else {
349 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
350 }
351 } else {
352 DataManager::SetValue("TW_CRYPTO_TYPE", password_type);
Noah Jacobson0b25b1d2019-04-28 00:10:07 -0400353 DataManager::SetValue("tw_crypto_pwtype_0", password_type);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600354 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600355 }
356 }
Noah Jacobson0b25b1d2019-04-28 00:10:07 -0400357 if (Decrypt_Data && (!Decrypt_Data->Is_Encrypted || Decrypt_Data->Is_Decrypted) &&
358 Decrypt_Data->Mount(false)) {
Ethan Yonker66a19492015-12-10 10:19:45 -0600359 Decrypt_Adopted();
360 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600361#endif
Dees_Troy51127312012-09-08 13:08:49 -0400362 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -0400363 UnMount_Main_Partitions();
Ethan Yonker1b190162016-12-05 15:25:19 -0600364#ifdef AB_OTA_UPDATER
365 DataManager::SetValue("tw_active_slot", Get_Active_Slot_Display());
366#endif
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600367 setup_uevent();
Dees_Troy5bf43922012-09-07 16:07:55 -0400368 return true;
369}
370
371int TWPartitionManager::Write_Fstab(void) {
372 FILE *fp;
373 std::vector<TWPartition*>::iterator iter;
374 string Line;
375
376 fp = fopen("/etc/fstab", "w");
377 if (fp == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000378 LOGINFO("Can not open /etc/fstab.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -0400379 return false;
380 }
Dees_Troy63c8df72012-09-10 14:02:05 -0400381 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy51127312012-09-08 13:08:49 -0400382 if ((*iter)->Can_Be_Mounted) {
dianlujitao7d304c72016-01-02 12:15:50 +0800383 Line = (*iter)->Actual_Block_Device + " " + (*iter)->Mount_Point + " " + (*iter)->Current_File_System + " rw 0 0\n";
Dees_Troy5bf43922012-09-07 16:07:55 -0400384 fputs(Line.c_str(), fp);
Dees_Troy91862e62013-04-04 23:48:21 +0000385 }
386 // Handle subpartition tracking
387 if ((*iter)->Is_SubPartition) {
388 TWPartition* ParentPartition = Find_Partition_By_Path((*iter)->SubPartition_Of);
389 if (ParentPartition)
390 ParentPartition->Has_SubPartition = true;
391 else
392 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 -0400393 }
394 }
395 fclose(fp);
396 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -0400397}
398
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500399void TWPartitionManager::Setup_Settings_Storage_Partition(TWPartition* Part) {
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500400 DataManager::SetValue("tw_settings_path", Part->Storage_Path);
401 DataManager::SetValue("tw_storage_path", Part->Storage_Path);
402 LOGINFO("Settings storage is '%s'\n", Part->Storage_Path.c_str());
403}
404
Matt Mowerbf4efa32014-04-14 23:25:26 -0500405void TWPartitionManager::Setup_Android_Secure_Location(TWPartition* Part) {
406 if (Part->Has_Android_Secure)
407 Part->Setup_AndSec();
Ethan Yonker6277c792014-09-15 14:54:30 -0500408 else if (!datamedia)
Matt Mowerbf4efa32014-04-14 23:25:26 -0500409 Part->Setup_AndSec();
Matt Mowerbf4efa32014-04-14 23:25:26 -0500410}
411
Dees_Troy8170a922012-09-18 15:40:25 -0400412void TWPartitionManager::Output_Partition_Logging(void) {
413 std::vector<TWPartition*>::iterator iter;
414
415 printf("\n\nPartition Logs:\n");
416 for (iter = Partitions.begin(); iter != Partitions.end(); iter++)
417 Output_Partition((*iter));
418}
419
420void TWPartitionManager::Output_Partition(TWPartition* Part) {
421 unsigned long long mb = 1048576;
422
Gary Peck004d48b2012-11-21 16:28:18 -0800423 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 -0400424 if (Part->Can_Be_Mounted) {
Gary Peck004d48b2012-11-21 16:28:18 -0800425 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 -0400426 }
Gary Peck004d48b2012-11-21 16:28:18 -0800427 printf("\n Flags: ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500428 if (Part->Can_Be_Mounted)
429 printf("Can_Be_Mounted ");
Gary Peck004d48b2012-11-21 16:28:18 -0800430 if (Part->Can_Be_Wiped)
431 printf("Can_Be_Wiped ");
Hashcodedabfd492013-08-29 22:45:30 -0700432 if (Part->Use_Rm_Rf)
433 printf("Use_Rm_Rf ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500434 if (Part->Can_Be_Backed_Up)
435 printf("Can_Be_Backed_Up ");
Gary Peck004d48b2012-11-21 16:28:18 -0800436 if (Part->Wipe_During_Factory_Reset)
437 printf("Wipe_During_Factory_Reset ");
438 if (Part->Wipe_Available_in_GUI)
439 printf("Wipe_Available_in_GUI ");
440 if (Part->Is_SubPartition)
441 printf("Is_SubPartition ");
442 if (Part->Has_SubPartition)
443 printf("Has_SubPartition ");
444 if (Part->Removable)
445 printf("Removable ");
446 if (Part->Is_Present)
447 printf("IsPresent ");
448 if (Part->Can_Be_Encrypted)
449 printf("Can_Be_Encrypted ");
450 if (Part->Is_Encrypted)
451 printf("Is_Encrypted ");
452 if (Part->Is_Decrypted)
453 printf("Is_Decrypted ");
454 if (Part->Has_Data_Media)
455 printf("Has_Data_Media ");
Dees_Troy83bd4832013-05-04 12:39:56 +0000456 if (Part->Can_Encrypt_Backup)
457 printf("Can_Encrypt_Backup ");
458 if (Part->Use_Userdata_Encryption)
459 printf("Use_Userdata_Encryption ");
Gary Peck004d48b2012-11-21 16:28:18 -0800460 if (Part->Has_Android_Secure)
461 printf("Has_Android_Secure ");
462 if (Part->Is_Storage)
463 printf("Is_Storage ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500464 if (Part->Is_Settings_Storage)
465 printf("Is_Settings_Storage ");
Dees_Troy68cab492012-12-12 19:29:35 +0000466 if (Part->Ignore_Blkid)
467 printf("Ignore_Blkid ");
Dees_Troy16c2b312013-01-15 16:51:18 +0000468 if (Part->Retain_Layout_Version)
469 printf("Retain_Layout_Version ");
Ethan Yonker253368a2014-11-25 15:00:52 -0600470 if (Part->Mount_To_Decrypt)
471 printf("Mount_To_Decrypt ");
Ethan Yonker96af84a2015-01-05 14:58:36 -0600472 if (Part->Can_Flash_Img)
473 printf("Can_Flash_Img ");
Ethan Yonker66a19492015-12-10 10:19:45 -0600474 if (Part->Is_Adopted_Storage)
475 printf("Is_Adopted_Storage ");
Ethan Yonker1b190162016-12-05 15:25:19 -0600476 if (Part->SlotSelect)
477 printf("SlotSelect ");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600478 if (Part->Mount_Read_Only)
479 printf("Mount_Read_Only ");
Gary Peck004d48b2012-11-21 16:28:18 -0800480 printf("\n");
481 if (!Part->SubPartition_Of.empty())
482 printf(" SubPartition_Of: %s\n", Part->SubPartition_Of.c_str());
483 if (!Part->Symlink_Path.empty())
484 printf(" Symlink_Path: %s\n", Part->Symlink_Path.c_str());
485 if (!Part->Symlink_Mount_Point.empty())
486 printf(" Symlink_Mount_Point: %s\n", Part->Symlink_Mount_Point.c_str());
487 if (!Part->Primary_Block_Device.empty())
488 printf(" Primary_Block_Device: %s\n", Part->Primary_Block_Device.c_str());
489 if (!Part->Alternate_Block_Device.empty())
490 printf(" Alternate_Block_Device: %s\n", Part->Alternate_Block_Device.c_str());
491 if (!Part->Decrypted_Block_Device.empty())
492 printf(" Decrypted_Block_Device: %s\n", Part->Decrypted_Block_Device.c_str());
dianlujitao4879b372018-12-03 18:45:47 +0800493 if (!Part->Crypto_Key_Location.empty())
Ethan Yonker253368a2014-11-25 15:00:52 -0600494 printf(" Crypto_Key_Location: %s\n", Part->Crypto_Key_Location.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800495 if (Part->Length != 0)
496 printf(" Length: %i\n", Part->Length);
497 if (!Part->Display_Name.empty())
498 printf(" Display_Name: %s\n", Part->Display_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500499 if (!Part->Storage_Name.empty())
500 printf(" Storage_Name: %s\n", Part->Storage_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800501 if (!Part->Backup_Path.empty())
502 printf(" Backup_Path: %s\n", Part->Backup_Path.c_str());
503 if (!Part->Backup_Name.empty())
504 printf(" Backup_Name: %s\n", Part->Backup_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500505 if (!Part->Backup_Display_Name.empty())
506 printf(" Backup_Display_Name: %s\n", Part->Backup_Display_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800507 if (!Part->Backup_FileName.empty())
508 printf(" Backup_FileName: %s\n", Part->Backup_FileName.c_str());
509 if (!Part->Storage_Path.empty())
510 printf(" Storage_Path: %s\n", Part->Storage_Path.c_str());
511 if (!Part->Current_File_System.empty())
512 printf(" Current_File_System: %s\n", Part->Current_File_System.c_str());
513 if (!Part->Fstab_File_System.empty())
514 printf(" Fstab_File_System: %s\n", Part->Fstab_File_System.c_str());
515 if (Part->Format_Block_Size != 0)
Ethan Yonkerc798c9c2015-10-09 11:15:26 -0500516 printf(" Format_Block_Size: %lu\n", Part->Format_Block_Size);
Dees_Troy094207a2012-09-26 12:00:39 -0400517 if (!Part->MTD_Name.empty())
518 printf(" MTD_Name: %s\n", Part->MTD_Name.c_str());
Matt Mower029a82d2017-01-08 13:12:38 -0600519 printf(" Backup_Method: %s\n", Part->Backup_Method_By_Name().c_str());
Hashcode62bd9e02013-11-19 21:59:42 -0800520 if (Part->Mount_Flags || !Part->Mount_Options.empty())
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600521 printf(" Mount_Flags: %i, Mount_Options: %s\n", Part->Mount_Flags, Part->Mount_Options.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -0600522 if (Part->MTP_Storage_ID)
523 printf(" MTP_Storage_ID: %i\n", Part->MTP_Storage_ID);
Ethan Yonker93382822018-11-01 15:25:31 -0500524 if (!Part->Key_Directory.empty())
525 printf(" Metadata Key Directory: %s\n", Part->Key_Directory.c_str());
Ethan Yonker6277c792014-09-15 14:54:30 -0500526 printf("\n");
Dees_Troy8170a922012-09-18 15:40:25 -0400527}
528
Dees_Troy51a0e822012-09-05 15:24:24 -0400529int TWPartitionManager::Mount_By_Path(string Path, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400530 std::vector<TWPartition*>::iterator iter;
531 int ret = false;
532 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400533 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400534
bigbiffd58ba182020-03-23 10:02:29 -0400535 if (Local_Path == "/tmp" || Local_Path == "/")
Dees_Troy43d8b002012-09-17 16:00:01 -0400536 return true;
537
Dees_Troy5bf43922012-09-07 16:07:55 -0400538 // Iterate through all partitions
539 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400540 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400541 ret = (*iter)->Mount(Display_Error);
542 found = true;
Dees_Troy51127312012-09-08 13:08:49 -0400543 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400544 (*iter)->Mount(Display_Error);
Dees_Troy51127312012-09-08 13:08:49 -0400545 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400546 }
547 if (found) {
548 return ret;
549 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500550 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 -0400551 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000552 LOGINFO("Mount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400553 }
554 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400555}
556
Dees_Troy51a0e822012-09-05 15:24:24 -0400557int TWPartitionManager::UnMount_By_Path(string Path, bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400558 std::vector<TWPartition*>::iterator iter;
559 int ret = false;
560 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400561 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy51127312012-09-08 13:08:49 -0400562
563 // Iterate through all partitions
564 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400565 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy51127312012-09-08 13:08:49 -0400566 ret = (*iter)->UnMount(Display_Error);
567 found = true;
568 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
569 (*iter)->UnMount(Display_Error);
570 }
571 }
572 if (found) {
573 return ret;
574 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500575 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 -0400576 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000577 LOGINFO("UnMount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400578 }
579 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400580}
581
Dees_Troy51a0e822012-09-05 15:24:24 -0400582int TWPartitionManager::Is_Mounted_By_Path(string Path) {
Dees_Troy51127312012-09-08 13:08:49 -0400583 TWPartition* Part = Find_Partition_By_Path(Path);
584
585 if (Part)
586 return Part->Is_Mounted();
587 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000588 LOGINFO("Is_Mounted: Unable to find partition for path '%s'\n", Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400589 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400590}
591
Dees_Troy5bf43922012-09-07 16:07:55 -0400592int TWPartitionManager::Mount_Current_Storage(bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400593 string current_storage_path = DataManager::GetCurrentStoragePath();
594
595 if (Mount_By_Path(current_storage_path, Display_Error)) {
596 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
597 if (FreeStorage)
598 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
599 return true;
600 }
601 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400602}
603
Dees_Troy5bf43922012-09-07 16:07:55 -0400604int TWPartitionManager::Mount_Settings_Storage(bool Display_Error) {
605 return Mount_By_Path(DataManager::GetSettingsStoragePath(), Display_Error);
606}
607
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600608TWPartition* TWPartitionManager::Find_Partition_By_Path(const string& Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400609 std::vector<TWPartition*>::iterator iter;
Dees_Troy38bd7602012-09-14 13:33:53 -0400610 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400611
dianlujitaob76a73a2020-04-30 20:33:20 +0800612 if (Local_Path == "/system")
613 Local_Path = Get_Android_Root_Path();
Dees_Troy5bf43922012-09-07 16:07:55 -0400614 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400615 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path))
Dees_Troy5bf43922012-09-07 16:07:55 -0400616 return (*iter);
617 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400618 return NULL;
619}
620
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600621TWPartition* TWPartitionManager::Find_Partition_By_Block_Device(const string& Block_Device) {
622 std::vector<TWPartition*>::iterator iter;
623
624 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
625 if ((*iter)->Primary_Block_Device == Block_Device || (!(*iter)->Actual_Block_Device.empty() && (*iter)->Actual_Block_Device == Block_Device))
626 return (*iter);
627 }
628 return NULL;
629}
630
Ethan Yonker53796e72019-01-11 22:49:52 -0600631int TWPartitionManager::Check_Backup_Name(const std::string& Backup_Name, bool Display_Error, bool Must_Be_Unique) {
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400632 // Check the backup name to ensure that it is the correct size and contains only valid characters
633 // and that a backup with that name doesn't already exist
634 char backup_name[MAX_BACKUP_NAME_LEN];
635 char backup_loc[255], tw_image_dir[255];
636 int copy_size;
637 int index, cur_char;
Ethan Yonker53796e72019-01-11 22:49:52 -0600638 string Backup_Loc;
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400639
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400640 copy_size = Backup_Name.size();
641 // Check size
642 if (copy_size > MAX_BACKUP_NAME_LEN) {
643 if (Display_Error)
Ethan Yonker74db1572015-10-28 12:44:49 -0500644 gui_err("backup_name_len=Backup name is too long.");
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400645 return -2;
646 }
647
648 // Check each character
649 strncpy(backup_name, Backup_Name.c_str(), copy_size);
Dees_Troya13d74f2013-03-24 08:54:55 -0500650 if (copy_size == 1 && strncmp(backup_name, "0", 1) == 0)
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400651 return 0; // A "0" (zero) means to use the current timestamp for the backup name
652 for (index=0; index<copy_size; index++) {
653 cur_char = (int)backup_name[index];
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500654 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 -0400655 // These are valid characters
656 // Numbers
657 // Upper case letters
658 // Lower case letters
659 // Space
660 // and -_.{}[]
661 } else {
662 if (Display_Error)
Ethan Yonker4adc33e2016-01-22 16:07:11 -0600663 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 -0400664 return -3;
665 }
666 }
667
Ethan Yonker53796e72019-01-11 22:49:52 -0600668 if (Must_Be_Unique) {
669 // Check to make sure that a backup with this name doesn't already exist
670 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Loc);
671 strcpy(backup_loc, Backup_Loc.c_str());
672 sprintf(tw_image_dir,"%s/%s", backup_loc, Backup_Name.c_str());
673 if (TWFunc::Path_Exists(tw_image_dir)) {
674 if (Display_Error)
675 gui_err("backup_name_exists=A backup with that name already exists!");
bigbiffce8f83c2015-12-12 18:30:21 -0500676
Ethan Yonker53796e72019-01-11 22:49:52 -0600677 return -4;
678 }
679 // Backup is unique
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400680 }
Ethan Yonker53796e72019-01-11 22:49:52 -0600681 // No problems found
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400682 return 0;
683}
684
bigbiffce8f83c2015-12-12 18:30:21 -0500685bool TWPartitionManager::Backup_Partition(PartitionSettings *part_settings) {
Dees_Troy43d8b002012-09-17 16:00:01 -0400686 time_t start, stop;
Matt Mower23d8aae2017-01-06 14:30:33 -0600687 int use_compression;
bigbiffce8f83c2015-12-12 18:30:21 -0500688 string backup_log = part_settings->Backup_Folder + "/recovery.log";
Dees_Troy43d8b002012-09-17 16:00:01 -0400689
bigbiffce8f83c2015-12-12 18:30:21 -0500690 if (part_settings->Part == NULL)
Dees_Troy43d8b002012-09-17 16:00:01 -0400691 return true;
692
Dees_Troy093b7642012-09-21 15:59:38 -0400693 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400694
Tom Hite5a926722014-09-15 01:31:03 +0000695 TWFunc::SetPerformanceMode(true);
Dees_Troy43d8b002012-09-17 16:00:01 -0400696 time(&start);
697
bigbiffce8f83c2015-12-12 18:30:21 -0500698 if (part_settings->Part->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500699 sync();
700 sync();
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400701 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500702 if (!part_settings->adbbackup && part_settings->generate_digest) {
703 if (!twrpDigestDriver::Make_Digest(Full_Filename))
704 goto backup_error;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400705 }
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400706
bigbiffce8f83c2015-12-12 18:30:21 -0500707 if (part_settings->Part->Has_SubPartition) {
Dees_Troy8170a922012-09-18 15:40:25 -0400708 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400709 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -0400710
711 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400712 if ((*subpart)->Can_Be_Backed_Up && (*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500713 part_settings->Part = *subpart;
bigbiffce8f83c2015-12-12 18:30:21 -0500714 if (!(*subpart)->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500715 goto backup_error;
Tom Hite5a926722014-09-15 01:31:03 +0000716 }
Dees_Troy2727b992013-08-14 20:09:30 +0000717 sync();
718 sync();
bigbiff bigbifff5955b12019-05-18 17:28:58 -0400719 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400720 if (!part_settings->adbbackup && part_settings->generate_digest) {
721 if (!twrpDigestDriver::Make_Digest(Full_Filename)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500722 goto backup_error;
bigbiffce8f83c2015-12-12 18:30:21 -0500723 }
Tom Hite5a926722014-09-15 01:31:03 +0000724 }
Dees_Troy8170a922012-09-18 15:40:25 -0400725 }
726 }
727 }
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400728
Dees_Troy43d8b002012-09-17 16:00:01 -0400729 time(&stop);
that203bcc92015-05-09 17:27:33 +0200730 int backup_time = (int) difftime(stop, start);
Dees_Troy2673cec2013-04-02 20:22:16 +0000731 LOGINFO("Partition Backup time: %d\n", backup_time);
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400732
bigbiffce8f83c2015-12-12 18:30:21 -0500733 if (part_settings->Part->Backup_Method == BM_FILES) {
734 part_settings->file_time += backup_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400735 } else {
bigbiffce8f83c2015-12-12 18:30:21 -0500736 part_settings->img_time += backup_time;
737
Dees_Troy43d8b002012-09-17 16:00:01 -0400738 }
Tom Hite5a926722014-09-15 01:31:03 +0000739
Matt Mower02899552016-12-30 18:13:51 -0600740 TWFunc::SetPerformanceMode(false);
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500741 return true;
Dees_Troy43d8b002012-09-17 16:00:01 -0400742 }
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500743backup_error:
744 Clean_Backup_Folder(part_settings->Backup_Folder);
745 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
746 tw_set_default_metadata(backup_log.c_str());
747 TWFunc::SetPerformanceMode(false);
748 return false;
bigbiff7abc5fe2015-01-17 16:53:12 -0500749}
750
bigbiffbf1d6722015-02-14 16:25:59 -0500751void TWPartitionManager::Clean_Backup_Folder(string Backup_Folder) {
752 DIR *d = opendir(Backup_Folder.c_str());
753 struct dirent *p;
754 int r;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400755 vector<string> ext;
756
757 //extensions we should delete when cleaning
758 ext.push_back("win");
759 ext.push_back("md5");
760 ext.push_back("sha2");
761 ext.push_back("info");
bigbiffbf1d6722015-02-14 16:25:59 -0500762
Ethan Yonker74db1572015-10-28 12:44:49 -0500763 gui_msg("backup_clean=Backup Failed. Cleaning Backup Folder.");
bigbiffbf1d6722015-02-14 16:25:59 -0500764
765 if (d == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500766 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Backup_Folder)(strerror(errno)));
bigbiffbf1d6722015-02-14 16:25:59 -0500767 return;
768 }
769
Matt Mower2b18a532015-02-20 16:58:05 -0600770 while ((p = readdir(d))) {
bigbiffbf1d6722015-02-14 16:25:59 -0500771 if (!strcmp(p->d_name, ".") || !strcmp(p->d_name, ".."))
772 continue;
773
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500774 string path = Backup_Folder + "/" + p->d_name;
bigbiffbf1d6722015-02-14 16:25:59 -0500775
776 size_t dot = path.find_last_of(".") + 1;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400777 for (vector<string>::const_iterator i = ext.begin(); i != ext.end(); ++i) {
778 if (path.substr(dot) == *i) {
779 r = unlink(path.c_str());
780 if (r != 0)
781 LOGINFO("Unable to unlink '%s: %s'\n", path.c_str(), strerror(errno));
bigbiffbf1d6722015-02-14 16:25:59 -0500782 }
783 }
784 }
785 closedir(d);
786}
787
Ethan Yonker472f5062016-02-25 13:47:30 -0600788int TWPartitionManager::Check_Backup_Cancel() {
789 return stop_backup.get_value();
790}
791
bigbiff7abc5fe2015-01-17 16:53:12 -0500792int TWPartitionManager::Cancel_Backup() {
793 string Backup_Folder, Backup_Name, Full_Backup_Path;
794
795 stop_backup.set_value(1);
796
797 if (tar_fork_pid != 0) {
798 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
799 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Folder);
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500800 Full_Backup_Path = Backup_Folder + "/" + Backup_Name;
bigbiff7abc5fe2015-01-17 16:53:12 -0500801 LOGINFO("Killing pid: %d\n", tar_fork_pid);
802 kill(tar_fork_pid, SIGUSR2);
803 while (kill(tar_fork_pid, 0) == 0) {
804 usleep(1000);
805 }
806 LOGINFO("Backup_Run stopped and returning false, backup cancelled.\n");
807 LOGINFO("Removing directory %s\n", Full_Backup_Path.c_str());
808 TWFunc::removeDir(Full_Backup_Path, false);
809 tar_fork_pid = 0;
810 }
811
812 return 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400813}
814
bigbiffce8f83c2015-12-12 18:30:21 -0500815int TWPartitionManager::Run_Backup(bool adbbackup) {
816 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400817 int partition_count = 0, disable_free_space_check = 0, skip_digest = 0;
bigbiffce8f83c2015-12-12 18:30:21 -0500818 string Backup_Name, Backup_List, backup_path;
Matt Mower23d8aae2017-01-06 14:30:33 -0600819 unsigned long long total_bytes = 0, free_space = 0;
Dees_Troy43d8b002012-09-17 16:00:01 -0400820 TWPartition* storage = NULL;
821 struct tm *t;
Matt Mower23d8aae2017-01-06 14:30:33 -0600822 time_t seconds, total_start, total_stop;
Dees_Troya13d74f2013-03-24 08:54:55 -0500823 size_t start_pos = 0, end_pos = 0;
bigbiff7abc5fe2015-01-17 16:53:12 -0500824 stop_backup.set_value(0);
Dees_Troy43d8b002012-09-17 16:00:01 -0400825 seconds = time(0);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500826 t = localtime(&seconds);
Dees_Troy43d8b002012-09-17 16:00:01 -0400827
bigbiffce8f83c2015-12-12 18:30:21 -0500828 part_settings.img_bytes_remaining = 0;
829 part_settings.file_bytes_remaining = 0;
830 part_settings.img_time = 0;
831 part_settings.file_time = 0;
832 part_settings.img_bytes = 0;
833 part_settings.file_bytes = 0;
834 part_settings.PM_Method = PM_BACKUP;
835
bigbiffce8f83c2015-12-12 18:30:21 -0500836 part_settings.adbbackup = adbbackup;
Dees_Troy43d8b002012-09-17 16:00:01 -0400837 time(&total_start);
838
839 Update_System_Details();
840
841 if (!Mount_Current_Storage(true))
842 return false;
843
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400844 DataManager::GetValue(TW_SKIP_DIGEST_GENERATE_VAR, skip_digest);
845 if (skip_digest == 0)
846 part_settings.generate_digest = true;
Dees_Troyc5865ab2012-09-24 15:08:04 -0400847 else
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400848 part_settings.generate_digest = false;
Dees_Troy4a2a1262012-09-18 09:33:47 -0400849
bigbiffce8f83c2015-12-12 18:30:21 -0500850 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, part_settings.Backup_Folder);
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500851 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
852 if (Backup_Name == gui_lookup("curr_date", "(Current Date)")) {
853 Backup_Name = TWFunc::Get_Current_Date();
854 } else if (Backup_Name == gui_lookup("auto_generate", "(Auto Generate)") || Backup_Name == "0" || Backup_Name.empty()) {
Dees Troyb21cc642013-09-10 17:36:41 +0000855 TWFunc::Auto_Generate_Backup_Name();
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500856 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
Dees_Troy43d8b002012-09-17 16:00:01 -0400857 }
bigbiffce8f83c2015-12-12 18:30:21 -0500858
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500859 LOGINFO("Backup Name is: '%s'\n", Backup_Name.c_str());
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500860 part_settings.Backup_Folder = part_settings.Backup_Folder + "/" + Backup_Name;
bigbiffce8f83c2015-12-12 18:30:21 -0500861
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500862 LOGINFO("Backup_Folder is: '%s'\n", part_settings.Backup_Folder.c_str());
Dees_Troy43d8b002012-09-17 16:00:01 -0400863
Dees_Troy2673cec2013-04-02 20:22:16 +0000864 LOGINFO("Calculating backup details...\n");
Dees_Troya13d74f2013-03-24 08:54:55 -0500865 DataManager::GetValue("tw_backup_list", Backup_List);
866 if (!Backup_List.empty()) {
867 end_pos = Backup_List.find(";", start_pos);
868 while (end_pos != string::npos && start_pos < Backup_List.size()) {
869 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -0500870 part_settings.Part = Find_Partition_By_Path(backup_path);
871 if (part_settings.Part != NULL) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500872 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500873 if (part_settings.Part->Backup_Method == BM_FILES)
874 part_settings.file_bytes += part_settings.Part->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500875 else
bigbiffce8f83c2015-12-12 18:30:21 -0500876 part_settings.img_bytes += part_settings.Part->Backup_Size;
877 if (part_settings.Part->Has_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500878 std::vector<TWPartition*>::iterator subpart;
879
880 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiffce8f83c2015-12-12 18:30:21 -0500881 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 -0500882 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -0500883 if ((*subpart)->Backup_Method == BM_FILES)
884 part_settings.file_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500885 else
bigbiffce8f83c2015-12-12 18:30:21 -0500886 part_settings.img_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -0500887 }
888 }
Dees_Troy8170a922012-09-18 15:40:25 -0400889 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500890 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500891 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 -0400892 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500893 start_pos = end_pos + 1;
894 end_pos = Backup_List.find(";", start_pos);
Dees_Troy43d8b002012-09-17 16:00:01 -0400895 }
896 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400897
898 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500899 gui_msg("no_partition_selected=No partitions selected for backup.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400900 return false;
901 }
bigbiffce8f83c2015-12-12 18:30:21 -0500902 if (adbbackup) {
903 if (twadbbu::Write_ADB_Stream_Header(partition_count) == false) {
904 return false;
905 }
906 }
907 total_bytes = part_settings.file_bytes + part_settings.img_bytes;
Ethan Yonker472f5062016-02-25 13:47:30 -0600908 ProgressTracking progress(total_bytes);
bigbiffce8f83c2015-12-12 18:30:21 -0500909 part_settings.progress = &progress;
910
Ethan Yonker74db1572015-10-28 12:44:49 -0500911 gui_msg(Msg("total_partitions_backup= * Total number of partitions to back up: {1}")(partition_count));
912 gui_msg(Msg("total_backup_size= * Total size of all data: {1}MB")(total_bytes / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400913 storage = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
914 if (storage != NULL) {
915 free_space = storage->Free;
Ethan Yonker74db1572015-10-28 12:44:49 -0500916 gui_msg(Msg("available_space= * Available space: {1}MB")(free_space / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -0400917 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500918 gui_err("unable_locate_storage=Unable to locate storage device.");
Dees_Troy43d8b002012-09-17 16:00:01 -0400919 return false;
920 }
bigbiffbf1d6722015-02-14 16:25:59 -0500921
Matt Mowerbfccfb82016-04-25 23:22:31 -0500922 DataManager::GetValue(TW_DISABLE_FREE_SPACE_VAR, disable_free_space_check);
bigbiffce8f83c2015-12-12 18:30:21 -0500923
924 if (adbbackup)
925 disable_free_space_check = true;
926
bigbiffbf1d6722015-02-14 16:25:59 -0500927 if (!disable_free_space_check) {
928 if (free_space - (32 * 1024 * 1024) < total_bytes) {
929 // We require an extra 32MB just in case
Ethan Yonker74db1572015-10-28 12:44:49 -0500930 gui_err("no_space=Not enough free space on storage.");
bigbiffbf1d6722015-02-14 16:25:59 -0500931 return false;
932 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400933 }
bigbiffce8f83c2015-12-12 18:30:21 -0500934 part_settings.img_bytes_remaining = part_settings.img_bytes;
935 part_settings.file_bytes_remaining = part_settings.file_bytes;
Dees_Troy43d8b002012-09-17 16:00:01 -0400936
Ethan Yonker74db1572015-10-28 12:44:49 -0500937 gui_msg("backup_started=[BACKUP STARTED]");
bigbiff bigbiffcdd97c72019-03-29 19:12:33 -0400938
939 int is_decrypted = 0;
940 int is_encrypted = 0;
941
942 DataManager::GetValue(TW_IS_DECRYPTED, is_decrypted);
943 DataManager::GetValue(TW_IS_ENCRYPTED, is_encrypted);
944 if (!adbbackup || (!is_encrypted || (is_encrypted && is_decrypted))) {
945 gui_msg(Msg("backup_folder= * Backup Folder: {1}")(part_settings.Backup_Folder));
946 if (!TWFunc::Recursive_Mkdir(part_settings.Backup_Folder)) {
947 gui_err("fail_backup_folder=Failed to make backup folder.");
948 return false;
949 }
Dees_Troyd4b22b02013-01-18 17:17:58 +0000950 }
951
Dees_Troy2673cec2013-04-02 20:22:16 +0000952 DataManager::SetProgress(0.0);
Dees_Troy093b7642012-09-21 15:59:38 -0400953
Dees_Troya13d74f2013-03-24 08:54:55 -0500954 start_pos = 0;
955 end_pos = Backup_List.find(";", start_pos);
956 while (end_pos != string::npos && start_pos < Backup_List.size()) {
bigbiff7abc5fe2015-01-17 16:53:12 -0500957 if (stop_backup.get_value() != 0)
958 return -1;
Dees_Troya13d74f2013-03-24 08:54:55 -0500959 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -0500960 part_settings.Part = Find_Partition_By_Path(backup_path);
961 if (part_settings.Part != NULL) {
962 if (!Backup_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -0500963 return false;
964 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500965 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 -0500966 }
967 start_pos = end_pos + 1;
968 end_pos = Backup_List.find(";", start_pos);
969 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400970
971 // Average BPS
bigbiffce8f83c2015-12-12 18:30:21 -0500972 if (part_settings.img_time == 0)
973 part_settings.img_time = 1;
974 if (part_settings.file_time == 0)
975 part_settings.file_time = 1;
976 int img_bps = (int)part_settings.img_bytes / (int)part_settings.img_time;
977 unsigned long long file_bps = part_settings.file_bytes / (int)part_settings.file_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400978
bigbiffce8f83c2015-12-12 18:30:21 -0500979 if (part_settings.file_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -0600980 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 -0500981 if (part_settings.img_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -0600982 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 -0400983
984 time(&total_stop);
985 int total_time = (int) difftime(total_stop, total_start);
bigbiffce8f83c2015-12-12 18:30:21 -0500986
987 uint64_t actual_backup_size;
Ethan Yonker3fdcda42016-11-30 12:29:37 -0600988 if (!adbbackup) {
989 TWExclude twe;
990 actual_backup_size = twe.Get_Folder_Size(part_settings.Backup_Folder);
991 } else
bigbiffce8f83c2015-12-12 18:30:21 -0500992 actual_backup_size = part_settings.file_bytes + part_settings.img_bytes;
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500993 actual_backup_size /= (1024LLU * 1024LLU);
Dees_Troy43d8b002012-09-17 16:00:01 -0400994
bigbiffce8f83c2015-12-12 18:30:21 -0500995 int prev_img_bps = 0, use_compression = 0;
996 unsigned long long prev_file_bps = 0;
Matt Mower173cdb92016-12-31 02:49:19 -0600997 DataManager::GetValue(TW_BACKUP_AVG_IMG_RATE, prev_img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -0400998 img_bps += (prev_img_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500999 img_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -04001000
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001001 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -04001002 if (use_compression)
1003 DataManager::GetValue(TW_BACKUP_AVG_FILE_COMP_RATE, prev_file_bps);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001004 else
Dees_Troy093b7642012-09-21 15:59:38 -04001005 DataManager::GetValue(TW_BACKUP_AVG_FILE_RATE, prev_file_bps);
1006 file_bps += (prev_file_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001007 file_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -04001008
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001009 DataManager::SetValue(TW_BACKUP_AVG_IMG_RATE, img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -04001010 if (use_compression)
1011 DataManager::SetValue(TW_BACKUP_AVG_FILE_COMP_RATE, file_bps);
1012 else
1013 DataManager::SetValue(TW_BACKUP_AVG_FILE_RATE, file_bps);
1014
Ethan Yonker74db1572015-10-28 12:44:49 -05001015 gui_msg(Msg("total_backed_size=[{1} MB TOTAL BACKED UP]")(actual_backup_size));
Dees_Troy43d8b002012-09-17 16:00:01 -04001016 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001017 UnMount_Main_Partitions();
Ethan Yonker56db1c42015-12-20 22:49:00 -06001018 gui_msg(Msg(msg::kHighlight, "backup_completed=[BACKUP COMPLETED IN {1} SECONDS]")(total_time)); // the end
Ethan Yonkere080c1f2016-09-19 13:50:25 -05001019 string backup_log = part_settings.Backup_Folder + "/recovery.log";
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001020 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
Ethan Yonker4b94cfd2014-12-11 10:00:45 -06001021 tw_set_default_metadata(backup_log.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -05001022
1023 if (part_settings.adbbackup) {
1024 if (twadbbu::Write_ADB_Stream_Trailer() == false) {
1025 return false;
1026 }
1027 }
1028 part_settings.adbbackup = false;
1029 DataManager::SetValue("tw_enable_adb_backup", 0);
1030
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001031 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001032}
1033
bigbiffce8f83c2015-12-12 18:30:21 -05001034bool TWPartitionManager::Restore_Partition(PartitionSettings *part_settings) {
Dees_Troy4a2a1262012-09-18 09:33:47 -04001035 time_t Start, Stop;
bigbiffce8f83c2015-12-12 18:30:21 -05001036
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001037 if (part_settings->adbbackup) {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001038 std::string partName = part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win";
1039 LOGINFO("setting backup name: %s\n", partName.c_str());
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001040 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
1041 }
1042
Tom Hite5a926722014-09-15 01:31:03 +00001043 TWFunc::SetPerformanceMode(true);
bigbiffce8f83c2015-12-12 18:30:21 -05001044
Dees_Troy4a2a1262012-09-18 09:33:47 -04001045 time(&Start);
Ethan Yonker472f5062016-02-25 13:47:30 -06001046
bigbiffce8f83c2015-12-12 18:30:21 -05001047 if (!part_settings->Part->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001048 TWFunc::SetPerformanceMode(false);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001049 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001050 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001051 if (part_settings->Part->Has_SubPartition && !part_settings->adbbackup) {
Dees_Troy8170a922012-09-18 15:40:25 -04001052 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001053 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -04001054
1055 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001056 part_settings->Part = *subpart;
1057 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
1058 part_settings->Part = (*subpart);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001059 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
bigbiffce8f83c2015-12-12 18:30:21 -05001060 if (!(*subpart)->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001061 TWFunc::SetPerformanceMode(false);
Dees_Troy8170a922012-09-18 15:40:25 -04001062 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001063 }
Dees_Troy8170a922012-09-18 15:40:25 -04001064 }
1065 }
1066 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001067 time(&Stop);
Tom Hite5a926722014-09-15 01:31:03 +00001068 TWFunc::SetPerformanceMode(false);
bigbiffce8f83c2015-12-12 18:30:21 -05001069 gui_msg(Msg("restore_part_done=[{1} done ({2} seconds)]")(part_settings->Part->Backup_Display_Name)((int)difftime(Stop, Start)));
1070
Dees_Troy4a2a1262012-09-18 09:33:47 -04001071 return true;
1072}
1073
Ethan Yonker472f5062016-02-25 13:47:30 -06001074int TWPartitionManager::Run_Restore(const string& Restore_Name) {
bigbiffce8f83c2015-12-12 18:30:21 -05001075 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001076 int check_digest;
bigbiffce8f83c2015-12-12 18:30:21 -05001077
Dees_Troy4a2a1262012-09-18 09:33:47 -04001078 time_t rStart, rStop;
1079 time(&rStart);
Dees_Troya13d74f2013-03-24 08:54:55 -05001080 string Restore_List, restore_path;
1081 size_t start_pos = 0, end_pos;
bigbiffce8f83c2015-12-12 18:30:21 -05001082
Ethan Yonkerdcf2b672016-09-13 14:41:53 -05001083 part_settings.Backup_Folder = Restore_Name;
bigbiffce8f83c2015-12-12 18:30:21 -05001084 part_settings.Part = NULL;
1085 part_settings.partition_count = 0;
1086 part_settings.total_restore_size = 0;
1087 part_settings.adbbackup = false;
1088 part_settings.PM_Method = PM_RESTORE;
Dees_Troy43d8b002012-09-17 16:00:01 -04001089
Ethan Yonker74db1572015-10-28 12:44:49 -05001090 gui_msg("restore_started=[RESTORE STARTED]");
1091 gui_msg(Msg("restore_folder=Restore folder: '{1}'")(Restore_Name));
Dees_Troy43d8b002012-09-17 16:00:01 -04001092
Dees_Troy4a2a1262012-09-18 09:33:47 -04001093 if (!Mount_Current_Storage(true))
1094 return false;
1095
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001096 DataManager::GetValue(TW_SKIP_DIGEST_CHECK_VAR, check_digest);
1097 if (check_digest > 0) {
1098 // Check Digest files first before restoring to ensure that all of them match before starting a restore
1099 TWFunc::GUI_Operation_Text(TW_VERIFY_DIGEST_TEXT, gui_parse_text("{@verifying_digest}"));
1100 gui_msg("verifying_digest=Verifying Digest");
Dees_Troya13d74f2013-03-24 08:54:55 -05001101 } else {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001102 gui_msg("skip_digest=Skipping Digest check based on user setting.");
Dees_Troya13d74f2013-03-24 08:54:55 -05001103 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001104 gui_msg("calc_restore=Calculating restore details...");
Dees_Troya13d74f2013-03-24 08:54:55 -05001105 DataManager::GetValue("tw_restore_selected", Restore_List);
bigbiffce8f83c2015-12-12 18:30:21 -05001106
Dees_Troya13d74f2013-03-24 08:54:55 -05001107 if (!Restore_List.empty()) {
1108 end_pos = Restore_List.find(";", start_pos);
1109 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1110 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001111 part_settings.Part = Find_Partition_By_Path(restore_path);
1112 if (part_settings.Part != NULL) {
bigbiffce8f83c2015-12-12 18:30:21 -05001113 if (part_settings.Part->Mount_Read_Only) {
1114 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 -05001115 return false;
1116 }
bigbiffce8f83c2015-12-12 18:30:21 -05001117
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001118 string Full_Filename = part_settings.Backup_Folder + "/" + part_settings.Part->Backup_FileName;
1119
1120 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001121 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001122 part_settings.partition_count++;
1123 part_settings.total_restore_size += part_settings.Part->Get_Restore_Size(&part_settings);
1124 if (part_settings.Part->Has_SubPartition) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001125 TWPartition *parentPart = part_settings.Part;
Dees_Troya13d74f2013-03-24 08:54:55 -05001126 std::vector<TWPartition*>::iterator subpart;
1127
1128 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001129 part_settings.Part = *subpart;
1130 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001131 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001132 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001133 part_settings.total_restore_size += (*subpart)->Get_Restore_Size(&part_settings);
Dees_Troya13d74f2013-03-24 08:54:55 -05001134 }
1135 }
1136 }
1137 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001138 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001139 }
1140 start_pos = end_pos + 1;
1141 end_pos = Restore_List.find(";", start_pos);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001142 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001143 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001144
bigbiffce8f83c2015-12-12 18:30:21 -05001145 if (part_settings.partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001146 gui_err("no_part_restore=No partitions selected for restore.");
Dees_Troy4a2a1262012-09-18 09:33:47 -04001147 return false;
1148 }
1149
bigbiffce8f83c2015-12-12 18:30:21 -05001150 gui_msg(Msg("restore_part_count=Restoring {1} partitions...")(part_settings.partition_count));
1151 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 +00001152 DataManager::SetProgress(0.0);
bigbiffce8f83c2015-12-12 18:30:21 -05001153 ProgressTracking progress(part_settings.total_restore_size);
1154 part_settings.progress = &progress;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001155
Dees_Troya13d74f2013-03-24 08:54:55 -05001156 start_pos = 0;
1157 if (!Restore_List.empty()) {
1158 end_pos = Restore_List.find(";", start_pos);
1159 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1160 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001161
1162 part_settings.Part = Find_Partition_By_Path(restore_path);
1163 if (part_settings.Part != NULL) {
1164 part_settings.partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -05001165 if (!Restore_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001166 return false;
1167 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001168 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001169 }
1170 start_pos = end_pos + 1;
1171 end_pos = Restore_List.find(";", start_pos);
1172 }
1173 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001174 TWFunc::GUI_Operation_Text(TW_UPDATE_SYSTEM_DETAILS_TEXT, gui_parse_text("{@updating_system_details}"));
Captain Throwback9d6feb52018-07-27 10:05:24 -04001175 UnMount_By_Path(Get_Android_Root_Path(), false);
Dees_Troy43d8b002012-09-17 16:00:01 -04001176 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001177 UnMount_Main_Partitions();
Dees_Troy4a2a1262012-09-18 09:33:47 -04001178 time(&rStop);
Matt Mower3c366972015-12-25 19:28:31 -06001179 gui_msg(Msg(msg::kHighlight, "restore_completed=[RESTORE COMPLETED IN {1} SECONDS]")((int)difftime(rStop,rStart)));
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001180 DataManager::SetValue("tw_file_progress", "");
bigbiffce8f83c2015-12-12 18:30:21 -05001181
Dees_Troy63c8df72012-09-10 14:02:05 -04001182 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001183}
1184
1185void TWPartitionManager::Set_Restore_Files(string Restore_Name) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001186 // Start with the default values
Dees_Troya13d74f2013-03-24 08:54:55 -05001187 string Restore_List;
Dees_Troy83bd4832013-05-04 12:39:56 +00001188 bool get_date = true, check_encryption = true;
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001189 bool adbbackup = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001190
1191 DataManager::SetValue("tw_restore_encrypted", 0);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001192 if (twadbbu::Check_ADB_Backup_File(Restore_Name)) {
1193 vector<string> adb_files;
1194 adb_files = twadbbu::Get_ADB_Backup_Files(Restore_Name);
1195 for (unsigned int i = 0; i < adb_files.size(); ++i) {
1196 string adb_restore_file = adb_files.at(i);
1197 std::size_t pos = adb_restore_file.find_first_of(".");
1198 std::string path = "/" + adb_restore_file.substr(0, pos);
1199 Restore_List = path + ";";
1200 TWPartition* Part = Find_Partition_By_Path(path);
1201 Part->Backup_FileName = TWFunc::Get_Filename(adb_restore_file);
1202 adbbackup = true;
1203 }
1204 DataManager::SetValue("tw_enable_adb_backup", 1);
Dees_Troy63c8df72012-09-10 14:02:05 -04001205 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001206 else {
1207 DIR* d;
1208 d = opendir(Restore_Name.c_str());
1209 if (d == NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001210 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001211 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Restore_Name)(strerror(errno)));
1212 return;
Dees_Troy63c8df72012-09-10 14:02:05 -04001213 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001214
1215 struct dirent* de;
1216 while ((de = readdir(d)) != NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001217 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001218 // Strip off three components
1219 char str[256];
1220 char* label;
1221 char* fstype = NULL;
1222 char* extn = NULL;
1223 char* ptr;
Dees_Troy63c8df72012-09-10 14:02:05 -04001224
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001225 strcpy(str, de->d_name);
1226 if (strlen(str) <= 2)
1227 continue;
Dees_Troy83bd4832013-05-04 12:39:56 +00001228
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001229 if (get_date) {
1230 char file_path[255];
1231 struct stat st;
1232
1233 strcpy(file_path, Restore_Name.c_str());
1234 strcat(file_path, "/");
1235 strcat(file_path, str);
1236 stat(file_path, &st);
1237 string backup_date = ctime((const time_t*)(&st.st_mtime));
1238 DataManager::SetValue(TW_RESTORE_FILE_DATE, backup_date);
1239 get_date = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001240 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001241
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001242 label = str;
1243 ptr = label;
1244 while (*ptr && *ptr != '.') ptr++;
1245 if (*ptr == '.')
1246 {
1247 *ptr = 0x00;
1248 ptr++;
1249 fstype = ptr;
1250 }
1251 while (*ptr && *ptr != '.') ptr++;
1252 if (*ptr == '.')
1253 {
1254 *ptr = 0x00;
1255 ptr++;
1256 extn = ptr;
1257 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001258
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001259 if (fstype == NULL || extn == NULL || strcmp(fstype, "log") == 0) continue;
1260 int extnlength = strlen(extn);
1261 if (extnlength != 3 && extnlength != 6) continue;
1262 if (extnlength >= 3 && strncmp(extn, "win", 3) != 0) continue;
1263 //if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
Dees_Troy63c8df72012-09-10 14:02:05 -04001264
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001265 if (check_encryption) {
1266 string filename = Restore_Name + "/";
1267 filename += de->d_name;
1268 if (TWFunc::Get_File_Type(filename) == 2) {
1269 LOGINFO("'%s' is encrypted\n", filename.c_str());
1270 DataManager::SetValue("tw_restore_encrypted", 1);
1271 }
1272 }
1273 if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
1274
1275 TWPartition* Part = Find_Partition_By_Path(label);
1276 if (Part == NULL)
1277 {
1278 gui_msg(Msg(msg::kError, "unable_locate_part_backup_name=Unable to locate partition by backup name: '{1}'")(label));
1279 continue;
1280 }
1281
1282 Part->Backup_FileName = de->d_name;
1283 if (strlen(extn) > 3) {
1284 Part->Backup_FileName.resize(Part->Backup_FileName.size() - strlen(extn) + 3);
1285 }
1286
dianlujitaob76a73a2020-04-30 20:33:20 +08001287 if (!Part->Is_SubPartition) {
1288 if (Part->Backup_Path == Get_Android_Root_Path())
1289 Restore_List += "/system;";
1290 else
1291 Restore_List += Part->Backup_Path + ";";
1292 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001293 }
1294 closedir(d);
Dees_Troy63c8df72012-09-10 14:02:05 -04001295 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001296
1297 if (adbbackup) {
1298 Restore_List = "ADB_Backup;";
1299 adbbackup = false;
1300 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001301
Dees_Troya13d74f2013-03-24 08:54:55 -05001302 // Set the final value
1303 DataManager::SetValue("tw_restore_list", Restore_List);
1304 DataManager::SetValue("tw_restore_selected", Restore_List);
Dees_Troy51a0e822012-09-05 15:24:24 -04001305 return;
1306}
1307
1308int TWPartitionManager::Wipe_By_Path(string Path) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001309 std::vector<TWPartition*>::iterator iter;
Mohd Farazfdd49bb2020-04-21 18:56:44 +05301310 std::vector < TWPartition * >::iterator iter1;
Dees_Troy63c8df72012-09-10 14:02:05 -04001311 int ret = false;
1312 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -04001313 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy63c8df72012-09-10 14:02:05 -04001314
dianlujitaob76a73a2020-04-30 20:33:20 +08001315 if (Local_Path == "/system")
1316 Local_Path = Get_Android_Root_Path();
Dees_Troy63c8df72012-09-10 14:02:05 -04001317 // Iterate through all partitions
1318 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -04001319 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Mohd Farazfdd49bb2020-04-21 18:56:44 +05301320 // iterate through all partitions since some legacy devices uses other partitions as vendor causes issues while wiping
1321 (*iter)->Find_Actual_Block_Device();
1322 for (iter1 = Partitions.begin (); iter1 != Partitions.end (); iter1++)
1323 {
1324 (*iter1)->Find_Actual_Block_Device();
1325 if ((*iter)->Actual_Block_Device == (*iter1)->Actual_Block_Device && (*iter)->Mount_Point != (*iter1)->Mount_Point)
1326 (*iter1)->UnMount(false);
1327 }
Dees_Troye58d5262012-09-21 12:27:57 -04001328 if (Path == "/and-sec")
1329 ret = (*iter)->Wipe_AndSec();
1330 else
1331 ret = (*iter)->Wipe();
Dees_Troy63c8df72012-09-10 14:02:05 -04001332 found = true;
1333 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1334 (*iter)->Wipe();
1335 }
1336 }
1337 if (found) {
1338 return ret;
1339 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001340 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 -04001341 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -04001342}
1343
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001344int TWPartitionManager::Wipe_By_Path(string Path, string New_File_System) {
1345 std::vector<TWPartition*>::iterator iter;
1346 int ret = false;
1347 bool found = false;
1348 string Local_Path = TWFunc::Get_Root_Path(Path);
1349
1350 // Iterate through all partitions
1351 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1352 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1353 if (Path == "/and-sec")
1354 ret = (*iter)->Wipe_AndSec();
1355 else
1356 ret = (*iter)->Wipe(New_File_System);
1357 found = true;
1358 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1359 (*iter)->Wipe(New_File_System);
1360 }
1361 }
1362 if (found) {
1363 return ret;
1364 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001365 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 -05001366 return false;
1367}
1368
Dees_Troy51a0e822012-09-05 15:24:24 -04001369int TWPartitionManager::Factory_Reset(void) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001370 std::vector<TWPartition*>::iterator iter;
1371 int ret = true;
1372
1373 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001374 if ((*iter)->Wipe_During_Factory_Reset && (*iter)->Is_Present) {
Ethan Yonker89583ef2015-08-26 09:01:59 -05001375#ifdef TW_OEM_BUILD
1376 if ((*iter)->Mount_Point == "/data") {
1377 if (!(*iter)->Wipe_Encryption())
1378 ret = false;
1379 } else {
1380#endif
1381 if (!(*iter)->Wipe())
1382 ret = false;
1383#ifdef TW_OEM_BUILD
1384 }
1385#endif
Dees_Troy094207a2012-09-26 12:00:39 -04001386 } else if ((*iter)->Has_Android_Secure) {
1387 if (!(*iter)->Wipe_AndSec())
1388 ret = false;
Dees_Troy63c8df72012-09-10 14:02:05 -04001389 }
1390 }
Ethan Yonker9fa76ba2016-06-30 14:05:43 -05001391 TWFunc::check_and_run_script("/sbin/factoryreset.sh", "Factory Reset Script");
Dees_Troy63c8df72012-09-10 14:02:05 -04001392 return ret;
Dees_Troy51a0e822012-09-05 15:24:24 -04001393}
1394
Dees_Troy38bd7602012-09-14 13:33:53 -04001395int TWPartitionManager::Wipe_Dalvik_Cache(void) {
1396 struct stat st;
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001397 vector <string> dir;
Dees_Troy38bd7602012-09-14 13:33:53 -04001398
1399 if (!Mount_By_Path("/data", true))
1400 return false;
1401
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001402 dir.push_back("/data/dalvik-cache");
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001403
1404 std::string cacheDir = TWFunc::get_cache_dir();
1405 if (cacheDir == NON_AB_CACHE_DIR) {
1406 if (!PartitionManager.Mount_By_Path(NON_AB_CACHE_DIR, false)) {
1407 LOGINFO("Unable to mount %s for wiping cache.\n", NON_AB_CACHE_DIR);
1408 }
1409 dir.push_back(cacheDir + "dalvik-cache");
1410 dir.push_back(cacheDir + "/dc");
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001411 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001412
Dees_Troy38bd7602012-09-14 13:33:53 -04001413 TWPartition* sdext = Find_Partition_By_Path("/sd-ext");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001414 if (sdext && sdext->Is_Present && sdext->Mount(false))
1415 {
1416 if (stat("/sd-ext/dalvik-cache", &st) == 0)
1417 {
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001418 dir.push_back("/sd-ext/dalvik-cache");
1419 }
1420 }
1421
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001422 if (cacheDir == NON_AB_CACHE_DIR) {
1423 gui_msg("wiping_cache_dalvik=Wiping Dalvik Cache Directories...");
1424 } else {
1425 gui_msg("wiping_dalvik=Wiping Dalvik Directory...");
1426 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001427 for (unsigned i = 0; i < dir.size(); ++i) {
1428 if (stat(dir.at(i).c_str(), &st) == 0) {
1429 TWFunc::removeDir(dir.at(i), false);
1430 gui_msg(Msg("cleaned=Cleaned: {1}...")(dir.at(i)));
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001431 }
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001432 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001433
1434 if (cacheDir == NON_AB_CACHE_DIR) {
1435 gui_msg("cache_dalvik_done=-- Dalvik Cache Directories Wipe Complete!");
1436 } else {
1437 gui_msg("dalvik_done=-- Dalvik Directory Wipe Complete!");
1438 }
1439
Dees_Troy38bd7602012-09-14 13:33:53 -04001440 return true;
1441}
1442
1443int TWPartitionManager::Wipe_Rotate_Data(void) {
1444 if (!Mount_By_Path("/data", true))
1445 return false;
1446
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001447 unlink("/data/misc/akmd*");
1448 unlink("/data/misc/rild*");
Dees_Troy2673cec2013-04-02 20:22:16 +00001449 gui_print("Rotation data wiped.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001450 return true;
1451}
1452
1453int TWPartitionManager::Wipe_Battery_Stats(void) {
1454 struct stat st;
1455
1456 if (!Mount_By_Path("/data", true))
1457 return false;
1458
1459 if (0 != stat("/data/system/batterystats.bin", &st)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001460 gui_print("No Battery Stats Found. No Need To Wipe.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001461 } else {
1462 remove("/data/system/batterystats.bin");
Dees_Troy2673cec2013-04-02 20:22:16 +00001463 gui_print("Cleared battery stats.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001464 }
1465 return true;
1466}
1467
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001468int TWPartitionManager::Wipe_Android_Secure(void) {
1469 std::vector<TWPartition*>::iterator iter;
1470 int ret = false;
1471 bool found = false;
1472
1473 // Iterate through all partitions
1474 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1475 if ((*iter)->Has_Android_Secure) {
1476 ret = (*iter)->Wipe_AndSec();
1477 found = true;
1478 }
1479 }
1480 if (found) {
1481 return ret;
1482 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001483 gui_err("no_andsec=No android secure partitions found.");
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001484 }
1485 return false;
1486}
1487
Dees_Troy38bd7602012-09-14 13:33:53 -04001488int TWPartitionManager::Format_Data(void) {
1489 TWPartition* dat = Find_Partition_By_Path("/data");
1490
1491 if (dat != NULL) {
1492 if (!dat->UnMount(true))
1493 return false;
1494
1495 return dat->Wipe_Encryption();
1496 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001497 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001498 return false;
1499 }
1500 return false;
1501}
1502
1503int TWPartitionManager::Wipe_Media_From_Data(void) {
1504 TWPartition* dat = Find_Partition_By_Path("/data");
1505
1506 if (dat != NULL) {
1507 if (!dat->Has_Data_Media) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001508 LOGERR("This device does not have /data/media\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001509 return false;
1510 }
1511 if (!dat->Mount(true))
1512 return false;
1513
Ethan Yonker74db1572015-10-28 12:44:49 -05001514 gui_msg("wiping_datamedia=Wiping internal storage -- /data/media...");
Ethan Yonker726a0202014-12-16 20:01:38 -06001515 Remove_MTP_Storage(dat->MTP_Storage_ID);
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001516 TWFunc::removeDir("/data/media", false);
xiaolu9416f4f2015-06-04 08:22:23 +08001517 dat->Recreate_Media_Folder();
Ethan Yonker726a0202014-12-16 20:01:38 -06001518 Add_MTP_Storage(dat->MTP_Storage_ID);
Dees_Troy38bd7602012-09-14 13:33:53 -04001519 return true;
1520 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001521 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001522 return false;
1523 }
1524 return false;
1525}
1526
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001527int TWPartitionManager::Repair_By_Path(string Path, bool Display_Error) {
1528 std::vector<TWPartition*>::iterator iter;
1529 int ret = false;
1530 bool found = false;
1531 string Local_Path = TWFunc::Get_Root_Path(Path);
1532
1533 if (Local_Path == "/tmp" || Local_Path == "/")
1534 return true;
1535
1536 // Iterate through all partitions
1537 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1538 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1539 ret = (*iter)->Repair();
1540 found = true;
1541 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1542 (*iter)->Repair();
1543 }
1544 }
1545 if (found) {
1546 return ret;
1547 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001548 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 -05001549 } else {
1550 LOGINFO("Repair: Unable to find partition for path '%s'\n", Local_Path.c_str());
1551 }
1552 return false;
1553}
1554
Ethan Yonkera2719152015-05-28 09:44:41 -05001555int TWPartitionManager::Resize_By_Path(string Path, bool Display_Error) {
1556 std::vector<TWPartition*>::iterator iter;
1557 int ret = false;
1558 bool found = false;
1559 string Local_Path = TWFunc::Get_Root_Path(Path);
1560
1561 if (Local_Path == "/tmp" || Local_Path == "/")
1562 return true;
1563
1564 // Iterate through all partitions
1565 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1566 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1567 ret = (*iter)->Resize();
1568 found = true;
1569 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1570 (*iter)->Resize();
1571 }
1572 }
1573 if (found) {
1574 return ret;
1575 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001576 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 -05001577 } else {
1578 LOGINFO("Resize: Unable to find partition for path '%s'\n", Local_Path.c_str());
1579 }
1580 return false;
1581}
1582
Dees_Troy51a0e822012-09-05 15:24:24 -04001583void TWPartitionManager::Update_System_Details(void) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001584 std::vector<TWPartition*>::iterator iter;
Dees_Troy51127312012-09-08 13:08:49 -04001585 int data_size = 0;
Dees_Troy5bf43922012-09-07 16:07:55 -04001586
Ethan Yonker74db1572015-10-28 12:44:49 -05001587 gui_msg("update_part_details=Updating partition details...");
Dees_Troy5bf43922012-09-07 16:07:55 -04001588 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker1b190162016-12-05 15:25:19 -06001589 (*iter)->Update_Size(true);
Dees_Troy51127312012-09-08 13:08:49 -04001590 if ((*iter)->Can_Be_Mounted) {
Captain Throwback9d6feb52018-07-27 10:05:24 -04001591 if ((*iter)->Mount_Point == Get_Android_Root_Path()) {
Dees_Troy51127312012-09-08 13:08:49 -04001592 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1593 DataManager::SetValue(TW_BACKUP_SYSTEM_SIZE, backup_display_size);
Ethan Yonker76bbd3a2019-05-10 10:50:04 -05001594 TWFunc::Is_TWRP_App_In_System();
Dees_Troy51127312012-09-08 13:08:49 -04001595 } else if ((*iter)->Mount_Point == "/data" || (*iter)->Mount_Point == "/datadata") {
1596 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
1597 } else if ((*iter)->Mount_Point == "/cache") {
1598 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1599 DataManager::SetValue(TW_BACKUP_CACHE_SIZE, backup_display_size);
1600 } else if ((*iter)->Mount_Point == "/sd-ext") {
1601 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1602 DataManager::SetValue(TW_BACKUP_SDEXT_SIZE, backup_display_size);
1603 if ((*iter)->Backup_Size == 0) {
1604 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 0);
1605 DataManager::SetValue(TW_BACKUP_SDEXT_VAR, 0);
1606 } else
1607 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 1);
Dees_Troye58d5262012-09-21 12:27:57 -04001608 } else if ((*iter)->Has_Android_Secure) {
Dees_Troy8170a922012-09-18 15:40:25 -04001609 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troye58d5262012-09-21 12:27:57 -04001610 DataManager::SetValue(TW_BACKUP_ANDSEC_SIZE, backup_display_size);
Dees_Troy8170a922012-09-18 15:40:25 -04001611 if ((*iter)->Backup_Size == 0) {
1612 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 0);
1613 DataManager::SetValue(TW_BACKUP_ANDSEC_VAR, 0);
1614 } else
1615 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 1);
Dees_Troy2c50e182012-09-26 20:05:28 -04001616 } else if ((*iter)->Mount_Point == "/boot") {
1617 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1618 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1619 if ((*iter)->Backup_Size == 0) {
1620 DataManager::SetValue("tw_has_boot_partition", 0);
1621 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1622 } else
1623 DataManager::SetValue("tw_has_boot_partition", 1);
Dees_Troy51127312012-09-08 13:08:49 -04001624 }
Dees_Troyaa9cc402012-10-13 12:14:05 -04001625 } else {
1626 // Handle unmountable partitions in case we reset defaults
1627 if ((*iter)->Mount_Point == "/boot") {
1628 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1629 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1630 if ((*iter)->Backup_Size == 0) {
1631 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 0);
1632 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1633 } else
1634 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 1);
1635 } else if ((*iter)->Mount_Point == "/recovery") {
1636 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1637 DataManager::SetValue(TW_BACKUP_RECOVERY_SIZE, backup_display_size);
1638 if ((*iter)->Backup_Size == 0) {
1639 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 0);
1640 DataManager::SetValue(TW_BACKUP_RECOVERY_VAR, 0);
1641 } else
1642 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 1);
Gary Peck82599a82012-11-21 16:23:12 -08001643 } else if ((*iter)->Mount_Point == "/data") {
1644 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troyaa9cc402012-10-13 12:14:05 -04001645 }
Dees_Troy51127312012-09-08 13:08:49 -04001646 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001647 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001648 gui_msg("update_part_details_done=...done");
Dees_Troy51127312012-09-08 13:08:49 -04001649 DataManager::SetValue(TW_BACKUP_DATA_SIZE, data_size);
1650 string current_storage_path = DataManager::GetCurrentStoragePath();
1651 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
Dees_Troy8170a922012-09-18 15:40:25 -04001652 if (FreeStorage != NULL) {
1653 // Attempt to mount storage
1654 if (!FreeStorage->Mount(false)) {
Matt Mower2d50cad2015-12-03 22:26:55 -06001655 gui_msg(Msg(msg::kError, "unable_to_mount_storage=Unable to mount storage"));
1656 DataManager::SetValue(TW_STORAGE_FREE_SIZE, 0);
Dees_Troy8170a922012-09-18 15:40:25 -04001657 } else {
1658 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
1659 }
1660 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001661 LOGINFO("Unable to find storage partition '%s'.\n", current_storage_path.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -04001662 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001663 if (!Write_Fstab())
Dees_Troy2673cec2013-04-02 20:22:16 +00001664 LOGERR("Error creating fstab\n");
Dees_Troy51a0e822012-09-05 15:24:24 -04001665 return;
1666}
1667
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001668void TWPartitionManager::Post_Decrypt(const string& Block_Device) {
1669 TWPartition* dat = Find_Partition_By_Path("/data");
1670 if (dat != NULL) {
1671 DataManager::SetValue(TW_IS_DECRYPTED, 1);
1672 dat->Is_Decrypted = true;
1673 if (!Block_Device.empty()) {
1674 dat->Decrypted_Block_Device = Block_Device;
1675 gui_msg(Msg("decrypt_success_dev=Data successfully decrypted, new block device: '{1}'")(Block_Device));
1676 } else {
1677 gui_msg("decrypt_success_nodev=Data successfully decrypted");
1678 }
Noah Jacobson0b25b1d2019-04-28 00:10:07 -04001679 property_set("twrp.decrypt.done", "true");
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001680 dat->Setup_File_System(false);
Noah Jacobson0b25b1d2019-04-28 00:10:07 -04001681 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 -06001682
Noah Jacobson0b25b1d2019-04-28 00:10:07 -04001683 sleep(1); // Sleep for a bit so that the device will be ready
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001684 if (dat->Has_Data_Media && dat->Mount(false) && TWFunc::Path_Exists("/data/media/0")) {
1685 dat->Storage_Path = "/data/media/0";
1686 dat->Symlink_Path = dat->Storage_Path;
1687 DataManager::SetValue("tw_storage_path", "/data/media/0");
1688 DataManager::SetValue("tw_settings_path", "/data/media/0");
1689 dat->UnMount(false);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001690 }
1691 Update_System_Details();
nkk71ffb02bd2017-06-04 23:12:16 +03001692 Output_Partition(dat);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001693 UnMount_Main_Partitions();
1694 } else
1695 LOGERR("Unable to locate data partition.\n");
1696}
1697
Noah Jacobson0b25b1d2019-04-28 00:10:07 -04001698void TWPartitionManager::Parse_Users() {
1699#ifdef TW_INCLUDE_FBE
1700 char user_check_result[PROPERTY_VALUE_MAX];
1701 for (int userId = 0; userId <= 9999; userId++) {
1702 string prop = "twrp.user." + to_string(userId) + ".decrypt";
1703 property_get(prop.c_str(), user_check_result, "-1");
1704 if (strcmp(user_check_result, "-1") != 0) {
1705 if (userId < 0 || userId > 9999) {
1706 LOGINFO("Incorrect user id %d\n", userId);
1707 continue;
1708 }
1709 struct users_struct user;
1710 user.userId = to_string(userId);
1711
1712 // Attempt to get name of user. Fallback to user ID if this fails.
1713 char* userFile = PageManager::LoadFileToBuffer("/data/system/users/" + to_string(userId) + ".xml", NULL);
1714 if (userFile == NULL)
1715 user.userName = to_string(userId);
1716 else {
1717 xml_document<> *userXml = new xml_document<>();
1718 userXml->parse<0>(userFile);
1719 xml_node<>* userNode = userXml->first_node("user");
1720 if (userNode == nullptr) {
1721 user.userName = to_string(userId);
1722 } else {
1723 xml_node<>* nameNode = userNode->first_node("name");
1724 if (nameNode == nullptr)
1725 user.userName = to_string(userId);
1726 else {
1727 string userName = nameNode->value();
1728 user.userName = userName + " (" + to_string(userId) + ")";
1729 }
1730 }
1731 }
1732
1733 string filename;
1734 user.type = Get_Password_Type(userId, filename);
1735
1736 user.isDecrypted = false;
1737 if (strcmp(user_check_result, "1") == 0)
1738 user.isDecrypted = true;
1739 Users_List.push_back(user);
1740 }
1741 }
1742 Check_Users_Decryption_Status();
1743#endif
1744}
1745
1746std::vector<users_struct>* TWPartitionManager::Get_Users_List() {
1747 return &Users_List;
1748}
1749
1750void TWPartitionManager::Mark_User_Decrypted(int userID) {
1751#ifdef TW_INCLUDE_FBE
1752 std::vector<users_struct>::iterator iter;
1753 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1754 if (atoi((*iter).userId.c_str()) == userID) {
1755 (*iter).isDecrypted = true;
1756 string user_prop_decrypted = "twrp.user." + to_string(userID) + ".decrypt";
1757 property_set(user_prop_decrypted.c_str(), "1");
1758 break;
1759 }
1760 }
1761 Check_Users_Decryption_Status();
1762#endif
1763}
1764
1765void TWPartitionManager::Check_Users_Decryption_Status() {
1766#ifdef TW_INCLUDE_FBE
1767 int all_is_decrypted = 1;
1768 std::vector<users_struct>::iterator iter;
1769 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1770 if (!(*iter).isDecrypted) {
1771 all_is_decrypted = 0;
1772 break;
1773 }
1774 }
1775 if (all_is_decrypted == 1) {
1776 DataManager::SetValue("tw_all_users_decrypted", "1");
1777 property_set("twrp.all.users.decrypted", "true");
1778 } else
1779 DataManager::SetValue("tw_all_users_decrypted", "0");
1780#endif
1781}
1782
1783int TWPartitionManager::Decrypt_Device(string Password, int user_id) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001784#ifdef TW_INCLUDE_CRYPTO
Ethan Yonkerddb63e22017-01-19 14:01:57 -06001785 char crypto_state[PROPERTY_VALUE_MAX], crypto_blkdev[PROPERTY_VALUE_MAX];
Ethan Yonker253368a2014-11-25 15:00:52 -06001786 std::vector<TWPartition*>::iterator iter;
Dees_Troy5bf43922012-09-07 16:07:55 -04001787
Ethan Yonker253368a2014-11-25 15:00:52 -06001788 // Mount any partitions that need to be mounted for decrypt
1789 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1790 if ((*iter)->Mount_To_Decrypt) {
1791 (*iter)->Mount(true);
1792 }
a39552696ff55ce2013-01-08 16:14:56 +00001793 }
oshmouna82a7542018-04-10 17:45:55 +02001794 property_set("twrp.mount_to_decrypt", "1");
a39552696ff55ce2013-01-08 16:14:56 +00001795
Ethan Yonkera1de1492015-11-04 11:58:27 -06001796 property_get("ro.crypto.state", crypto_state, "error");
1797 if (strcmp(crypto_state, "error") == 0) {
1798 property_set("ro.crypto.state", "encrypted");
1799 // Sleep for a bit so that services can start if needed
1800 sleep(1);
1801 }
1802
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001803 if (DataManager::GetIntValue(TW_IS_FBE)) {
1804#ifdef TW_INCLUDE_FBE
1805 if (!Mount_By_Path("/data", true)) // /data has to be mounted for FBE
1806 return -1;
Noah Jacobson0b25b1d2019-04-28 00:10:07 -04001807
1808 bool user_need_decrypt = false;
1809 std::vector<users_struct>::iterator iter;
1810 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1811 if (atoi((*iter).userId.c_str()) == user_id && !(*iter).isDecrypted) {
1812 user_need_decrypt = true;
1813 }
1814 }
1815 if (!user_need_decrypt) {
1816 LOGINFO("User %d does not require decryption\n", user_id);
1817 return 0;
1818 }
1819
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001820 int retry_count = 10;
1821 while (!TWFunc::Path_Exists("/data/system/users/gatekeeper.password.key") && --retry_count)
Noah Jacobson0b25b1d2019-04-28 00:10:07 -04001822 usleep(2000); // A small sleep is needed after mounting /data to ensure reliable decrypt...maybe because of DE?
1823 gui_msg(Msg("decrypting_user_fbe=Attempting to decrypt FBE for user {1}...")(user_id));
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001824 if (Decrypt_User(user_id, Password)) {
Noah Jacobson0b25b1d2019-04-28 00:10:07 -04001825 gui_msg(Msg("decrypt_user_success_fbe=User {1} Decrypted Successfully")(user_id));
1826 Mark_User_Decrypted(user_id);
1827 if (user_id == 0) {
1828 // When decrypting user 0 also try all other users
1829 std::vector<users_struct>::iterator iter;
1830 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
1831 if ((*iter).userId == "0" || (*iter).isDecrypted)
1832 continue;
1833
1834 int tmp_user_id = atoi((*iter).userId.c_str());
1835 gui_msg(Msg("decrypting_user_fbe=Attempting to decrypt FBE for user {1}...")(tmp_user_id));
1836 if (Decrypt_User(tmp_user_id, Password) ||
1837 (Password != "!" && Decrypt_User(tmp_user_id, "!"))) { // "!" means default password
1838 gui_msg(Msg("decrypt_user_success_fbe=User {1} Decrypted Successfully")(tmp_user_id));
1839 Mark_User_Decrypted(tmp_user_id);
1840 } else {
1841 gui_msg(Msg("decrypt_user_fail_fbe=Failed to decrypt user {1}")(tmp_user_id));
1842 }
1843 }
1844 Post_Decrypt("");
1845 }
1846
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001847 return 0;
Noah Jacobson0b25b1d2019-04-28 00:10:07 -04001848 } else {
1849 gui_msg(Msg(msg::kError, "decrypt_user_fail_fbe=Failed to decrypt user {1}")(user_id));
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001850 }
1851#else
1852 LOGERR("FBE support is not present\n");
1853#endif
1854 return -1;
1855 }
1856
Noah Jacobson0b25b1d2019-04-28 00:10:07 -04001857 char isdecrypteddata[PROPERTY_VALUE_MAX];
1858 property_get("twrp.decrypt.done", isdecrypteddata, "");
1859 if (strcmp(isdecrypteddata, "true") == 0) {
1860 LOGINFO("Data has no decryption required\n");
1861 return 0;
1862 }
1863
Ethan Yonkerddb63e22017-01-19 14:01:57 -06001864 int pwret = -1;
1865 pid_t pid = fork();
1866 if (pid < 0) {
1867 LOGERR("fork failed\n");
1868 return -1;
1869 } else if (pid == 0) {
1870 // Child process
1871 char cPassword[255];
1872 strcpy(cPassword, Password.c_str());
1873 int ret = cryptfs_check_passwd(cPassword);
1874 exit(ret);
1875 } else {
1876 // Parent
1877 int status;
1878 if (TWFunc::Wait_For_Child_Timeout(pid, &status, "Decrypt", 30))
1879 pwret = -1;
1880 else
1881 pwret = WEXITSTATUS(status) ? -1 : 0;
1882 }
a39552696ff55ce2013-01-08 16:14:56 +00001883
nkk7171c6c502017-01-05 23:55:05 +02001884#ifdef TW_CRYPTO_USE_SYSTEM_VOLD
1885 if (pwret != 0) {
1886 pwret = vold_decrypt(Password);
nkk717d1222a2017-11-06 17:27:21 +02001887 switch (pwret) {
1888 case VD_SUCCESS:
1889 break;
1890 case VD_ERR_MISSING_VDC:
1891 gui_msg(Msg(msg::kError, "decrypt_data_vold_os_missing=Missing files needed for vold decrypt: {1}")("/system/bin/vdc"));
1892 break;
1893 case VD_ERR_MISSING_VOLD:
1894 gui_msg(Msg(msg::kError, "decrypt_data_vold_os_missing=Missing files needed for vold decrypt: {1}")("/system/bin/vold"));
1895 break;
1896 }
nkk7171c6c502017-01-05 23:55:05 +02001897 }
1898#endif // TW_CRYPTO_USE_SYSTEM_VOLD
1899
Ethan Yonker253368a2014-11-25 15:00:52 -06001900 // Unmount any partitions that were needed for decrypt
1901 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1902 if ((*iter)->Mount_To_Decrypt) {
1903 (*iter)->UnMount(false);
1904 }
1905 }
oshmouna82a7542018-04-10 17:45:55 +02001906 property_set("twrp.mount_to_decrypt", "0");
Ethan Yonker253368a2014-11-25 15:00:52 -06001907
a39552696ff55ce2013-01-08 16:14:56 +00001908 if (pwret != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001909 gui_err("fail_decrypt=Failed to decrypt data.");
Dees_Troy5bf43922012-09-07 16:07:55 -04001910 return -1;
1911 }
a39552696ff55ce2013-01-08 16:14:56 +00001912
Dees_Troy5bf43922012-09-07 16:07:55 -04001913 property_get("ro.crypto.fs_crypto_blkdev", crypto_blkdev, "error");
1914 if (strcmp(crypto_blkdev, "error") == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001915 LOGERR("Error retrieving decrypted data block device.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04001916 } else {
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001917 Post_Decrypt(crypto_blkdev);
Dees_Troy5bf43922012-09-07 16:07:55 -04001918 }
1919 return 0;
1920#else
Ethan Yonker74db1572015-10-28 12:44:49 -05001921 gui_err("no_crypto_support=No crypto support was compiled into this build.");
Dees_Troy5bf43922012-09-07 16:07:55 -04001922 return -1;
1923#endif
Dees_Troy51a0e822012-09-05 15:24:24 -04001924 return 1;
Dees_Troy51127312012-09-08 13:08:49 -04001925}
1926
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001927int TWPartitionManager::Fix_Contexts(void) {
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001928 std::vector<TWPartition*>::iterator iter;
1929 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1930 if ((*iter)->Has_Data_Media) {
1931 if ((*iter)->Mount(true)) {
1932 if (fixContexts::fixDataMediaContexts((*iter)->Mount_Point) != 0)
1933 return -1;
1934 }
1935 }
1936 }
Dees_Troy1a650e62012-10-19 20:54:32 -04001937 UnMount_Main_Partitions();
Ethan Yonker74db1572015-10-28 12:44:49 -05001938 gui_msg("done=Done.");
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001939 return 0;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04001940}
1941
Ethan Yonker66a19492015-12-10 10:19:45 -06001942TWPartition* TWPartitionManager::Find_Next_Storage(string Path, bool Exclude_Data_Media) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001943 std::vector<TWPartition*>::iterator iter = Partitions.begin();
1944
1945 if (!Path.empty()) {
1946 string Search_Path = TWFunc::Get_Root_Path(Path);
1947 for (; iter != Partitions.end(); iter++) {
Matt Mower9a561dd2016-02-22 21:25:51 -06001948 if ((*iter)->Mount_Point == Search_Path) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001949 iter++;
1950 break;
1951 }
1952 }
1953 }
1954
1955 for (; iter != Partitions.end(); iter++) {
Ethan Yonker66a19492015-12-10 10:19:45 -06001956 if (Exclude_Data_Media && (*iter)->Has_Data_Media) {
1957 // do nothing, do not return this type of partition
1958 } else if ((*iter)->Is_Storage && (*iter)->Is_Present) {
Ethan Yonker47360be2014-04-01 10:34:34 -05001959 return (*iter);
1960 }
1961 }
1962
1963 return NULL;
1964}
1965
Dees_Troyd21618c2012-10-14 18:48:49 -04001966int TWPartitionManager::Open_Lun_File(string Partition_Path, string Lun_File) {
Dees_Troyd21618c2012-10-14 18:48:49 -04001967 TWPartition* Part = Find_Partition_By_Path(Partition_Path);
1968
1969 if (Part == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001970 LOGINFO("Unable to locate '%s' for USB storage mode.", Partition_Path.c_str());
1971 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 -04001972 return false;
1973 }
Ethan Yonker47360be2014-04-01 10:34:34 -05001974 LOGINFO("USB mount '%s', '%s' > '%s'\n", Partition_Path.c_str(), Part->Actual_Block_Device.c_str(), Lun_File.c_str());
1975 if (!Part->UnMount(true) || !Part->Is_Present)
Dees_Troyd21618c2012-10-14 18:48:49 -04001976 return false;
1977
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06001978 if (TWFunc::write_to_file(Lun_File, Part->Actual_Block_Device)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001979 LOGERR("Unable to write to ums lunfile '%s': (%s)\n", Lun_File.c_str(), strerror(errno));
Dees_Troyd21618c2012-10-14 18:48:49 -04001980 return false;
1981 }
Dees_Troyd21618c2012-10-14 18:48:49 -04001982 return true;
1983}
1984
Dees_Troy8170a922012-09-18 15:40:25 -04001985int TWPartitionManager::usb_storage_enable(void) {
Dees_Troy8170a922012-09-18 15:40:25 -04001986 char lun_file[255];
Dees_Troyd21618c2012-10-14 18:48:49 -04001987 bool has_multiple_lun = false;
Dees_Troy8170a922012-09-18 15:40:25 -04001988
Ethan Yonker47360be2014-04-01 10:34:34 -05001989 string Lun_File_str = CUSTOM_LUN_FILE;
1990 size_t found = Lun_File_str.find("%");
1991 if (found != string::npos) {
1992 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
1993 if (TWFunc::Path_Exists(lun_file))
1994 has_multiple_lun = true;
1995 }
Ethan Yonker726a0202014-12-16 20:01:38 -06001996 mtp_was_enabled = TWFunc::Toggle_MTP(false); // Must disable MTP for USB Storage
Ethan Yonker47360be2014-04-01 10:34:34 -05001997 if (!has_multiple_lun) {
1998 LOGINFO("Device doesn't have multiple lun files, mount current storage\n");
1999 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
2000 if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) == "/data") {
Ethan Yonker66a19492015-12-10 10:19:45 -06002001 TWPartition* Mount = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05002002 if (Mount) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002003 if (!Open_Lun_File(Mount->Mount_Point, lun_file)) {
2004 goto error_handle;
2005 }
Ethan Yonker47360be2014-04-01 10:34:34 -05002006 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002007 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002008 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05002009 }
2010 } else if (!Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file)) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002011 goto error_handle;
Dees_Troy8170a922012-09-18 15:40:25 -04002012 }
Dees_Troy8170a922012-09-18 15:40:25 -04002013 } else {
Ethan Yonker47360be2014-04-01 10:34:34 -05002014 LOGINFO("Device has multiple lun files\n");
2015 TWPartition* Mount1;
2016 TWPartition* Mount2;
Dees_Troy8170a922012-09-18 15:40:25 -04002017 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
Ethan Yonker66a19492015-12-10 10:19:45 -06002018 Mount1 = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05002019 if (Mount1) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002020 if (!Open_Lun_File(Mount1->Mount_Point, lun_file)) {
2021 goto error_handle;
2022 }
Matt Moweree717062014-04-26 01:46:46 -05002023 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
Ethan Yonker66a19492015-12-10 10:19:45 -06002024 Mount2 = Find_Next_Storage(Mount1->Mount_Point, true);
Matt Mower1fdcdb72016-01-25 20:53:47 -06002025 if (Mount2 && Mount2->Mount_Point != Mount1->Mount_Point) {
Matt Moweree717062014-04-26 01:46:46 -05002026 Open_Lun_File(Mount2->Mount_Point, lun_file);
Ethan Yonker47360be2014-04-01 10:34:34 -05002027 }
2028 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002029 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002030 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05002031 }
Dees_Troy8170a922012-09-18 15:40:25 -04002032 }
Matt Mowerb6ef4782014-11-06 02:24:36 -06002033 property_set("sys.storage.ums_enabled", "1");
HandyMenny37d42992014-10-26 22:15:59 +01002034 property_set("sys.usb.config", "mass_storage,adb");
Dees_Troy8170a922012-09-18 15:40:25 -04002035 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002036error_handle:
2037 if (mtp_was_enabled)
2038 if (!Enable_MTP())
2039 Disable_MTP();
2040 return false;
Dees_Troy8170a922012-09-18 15:40:25 -04002041}
2042
2043int TWPartitionManager::usb_storage_disable(void) {
Dees_Troy2673cec2013-04-02 20:22:16 +00002044 int index, ret;
2045 char lun_file[255], ch[2] = {0, 0};
2046 string str = ch;
Dees_Troy8170a922012-09-18 15:40:25 -04002047
2048 for (index=0; index<2; index++) {
2049 sprintf(lun_file, CUSTOM_LUN_FILE, index);
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002050 ret = TWFunc::write_to_file(lun_file, str);
Dees_Troy2673cec2013-04-02 20:22:16 +00002051 if (ret < 0) {
2052 break;
Dees_Troy8170a922012-09-18 15:40:25 -04002053 }
Dees_Troy8170a922012-09-18 15:40:25 -04002054 }
Dees_Troye58d5262012-09-21 12:27:57 -04002055 Mount_All_Storage();
2056 Update_System_Details();
Dees_Troycfd73ef2012-10-12 16:52:00 -04002057 UnMount_Main_Partitions();
Matt Mowerd9cb9062013-12-14 20:34:45 -06002058 property_set("sys.storage.ums_enabled", "0");
HandyMenny37d42992014-10-26 22:15:59 +01002059 property_set("sys.usb.config", "adb");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002060 if (mtp_was_enabled)
2061 if (!Enable_MTP())
2062 Disable_MTP();
Dees_Troy2673cec2013-04-02 20:22:16 +00002063 if (ret < 0 && index == 0) {
2064 LOGERR("Unable to write to ums lunfile '%s'.", lun_file);
2065 return false;
2066 } else {
2067 return true;
2068 }
Dees_Troy8170a922012-09-18 15:40:25 -04002069 return true;
Dees_Troy812660f2012-09-20 09:55:17 -04002070}
2071
2072void TWPartitionManager::Mount_All_Storage(void) {
2073 std::vector<TWPartition*>::iterator iter;
2074
2075 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2076 if ((*iter)->Is_Storage)
2077 (*iter)->Mount(false);
2078 }
Dees_Troy2c50e182012-09-26 20:05:28 -04002079}
Dees_Troy9350b8d2012-09-27 12:38:38 -04002080
Dees_Troyd0384ef2012-10-12 12:15:42 -04002081void TWPartitionManager::UnMount_Main_Partitions(void) {
2082 // Unmounts system and data if data is not data/media
2083 // Also unmounts boot if boot is mountable
Dees_Troy2673cec2013-04-02 20:22:16 +00002084 LOGINFO("Unmounting main partitions...\n");
Dees_Troyd0384ef2012-10-12 12:15:42 -04002085
2086 TWPartition* Boot_Partition = Find_Partition_By_Path("/boot");
2087
Captain Throwback9d6feb52018-07-27 10:05:24 -04002088 UnMount_By_Path(Get_Android_Root_Path(), true);
Ethan Yonker6277c792014-09-15 14:54:30 -05002089 if (!datamedia)
2090 UnMount_By_Path("/data", true);
2091
Dees_Troyd0384ef2012-10-12 12:15:42 -04002092 if (Boot_Partition != NULL && Boot_Partition->Can_Be_Mounted)
2093 Boot_Partition->UnMount(true);
2094}
2095
Dees_Troy9350b8d2012-09-27 12:38:38 -04002096int TWPartitionManager::Partition_SDCard(void) {
Matt Mower23d8aae2017-01-06 14:30:33 -06002097 char temp[255];
Ethan Yonker483e9f42016-01-11 22:21:18 -06002098 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 -04002099 int ext, swap, total_size = 0, fat_size;
Dees_Troy9350b8d2012-09-27 12:38:38 -04002100
Ethan Yonker74db1572015-10-28 12:44:49 -05002101 gui_msg("start_partition_sd=Partitioning SD Card...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06002102
2103 // Locate and validate device to partition
2104 TWPartition* SDCard = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
2105
Ethan Yonker66a19492015-12-10 10:19:45 -06002106 if (SDCard->Is_Adopted_Storage)
2107 SDCard->Revert_Adopted();
2108
Ethan Yonker1eff6cd2014-09-15 13:30:42 -05002109 if (SDCard == NULL || !SDCard->Removable || SDCard->Has_Data_Media) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002110 gui_err("partition_sd_locate=Unable to locate device to partition.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002111 return false;
2112 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002113
2114 // Unmount everything
Dees_Troy9350b8d2012-09-27 12:38:38 -04002115 if (!SDCard->UnMount(true))
2116 return false;
2117 TWPartition* SDext = Find_Partition_By_Path("/sd-ext");
2118 if (SDext != NULL) {
2119 if (!SDext->UnMount(true))
2120 return false;
2121 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002122 char* swappath = getenv("SWAPPATH");
2123 if (swappath != NULL) {
2124 LOGINFO("Unmounting swap at '%s'\n", swappath);
2125 umount(swappath);
2126 }
Vojtech Bocek05534202013-09-11 08:11:56 +02002127
Ethan Yonker483e9f42016-01-11 22:21:18 -06002128 // Determine block device
2129 if (SDCard->Alternate_Block_Device.empty()) {
2130 SDCard->Find_Actual_Block_Device();
2131 Device = SDCard->Actual_Block_Device;
2132 // Just use the root block device
2133 Device.resize(strlen("/dev/block/mmcblkX"));
2134 } else {
2135 Device = SDCard->Alternate_Block_Device;
2136 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04002137
2138 // Find the size of the block device:
Ethan Yonker483e9f42016-01-11 22:21:18 -06002139 total_size = (int)(TWFunc::IOCTL_Get_Block_Size(Device.c_str()) / (1048576));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002140
2141 DataManager::GetValue("tw_sdext_size", ext);
2142 DataManager::GetValue("tw_swap_size", swap);
2143 DataManager::GetValue("tw_sdpart_file_system", ext_format);
2144 fat_size = total_size - ext - swap;
Ethan Yonker483e9f42016-01-11 22:21:18 -06002145 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);
2146
2147 // Determine partition sizes
2148 if (swap == 0 && ext == 0) {
2149 fat_str = "-0";
2150 } else {
2151 memset(temp, 0, sizeof(temp));
2152 sprintf(temp, "%i", fat_size);
2153 fat_str = temp;
2154 fat_str += "MB";
2155 }
2156 if (swap == 0) {
2157 ext_str = "-0";
2158 } else {
2159 memset(temp, 0, sizeof(temp));
2160 sprintf(temp, "%i", ext);
2161 ext_str = "+";
2162 ext_str += temp;
2163 ext_str += "MB";
2164 }
2165
Dees_Troy9350b8d2012-09-27 12:38:38 -04002166 if (ext + swap > total_size) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002167 gui_err("ext_swap_size=EXT + Swap size is larger than sdcard size.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002168 return false;
2169 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002170
Ethan Yonker74db1572015-10-28 12:44:49 -05002171 gui_msg("remove_part_table=Removing partition table...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06002172 Command = "sgdisk --zap-all " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002173 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002174 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002175 gui_err("unable_rm_part=Unable to remove partition table.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002176 Update_System_Details();
2177 return false;
2178 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002179 gui_msg(Msg("create_part=Creating {1} partition...")("FAT32"));
Captain Throwback9643a802016-05-27 11:44:51 -04002180 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 +00002181 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002182 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002183 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("FAT32"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002184 return false;
2185 }
2186 if (ext > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002187 gui_msg(Msg("create_part=Creating {1} partition...")("EXT"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06002188 Command = "sgdisk --new=0:0:" + ext_str + " --change-name=0:\"Linux filesystem\" " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002189 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002190 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002191 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("EXT"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002192 Update_System_Details();
2193 return false;
2194 }
2195 }
2196 if (swap > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002197 gui_msg(Msg("create_part=Creating {1} partition...")("swap"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06002198 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 +00002199 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002200 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002201 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("swap"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002202 Update_System_Details();
2203 return false;
2204 }
2205 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002206
2207 // Convert GPT to MBR
2208 Command = "sgdisk --gpttombr " + Device;
2209 if (TWFunc::Exec_Cmd(Command) != 0)
2210 LOGINFO("Failed to covert partition GPT to MBR\n");
2211
2212 // Tell the kernel to rescan the partition table
2213 int fd = open(Device.c_str(), O_RDONLY);
2214 ioctl(fd, BLKRRPART, 0);
2215 close(fd);
2216
2217 string format_device = Device;
2218 if (Device.substr(0, 17) == "/dev/block/mmcblk")
2219 format_device += "p";
2220
2221 // Format new partitions to proper file system
2222 if (fat_size > 0) {
2223 Command = "mkfs.fat " + format_device + "1";
2224 TWFunc::Exec_Cmd(Command);
2225 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04002226 if (ext > 0) {
2227 if (SDext == NULL) {
Ethan Yonker483e9f42016-01-11 22:21:18 -06002228 Command = "mke2fs -t " + ext_format + " -m 0 " + format_device + "2";
2229 gui_msg(Msg("format_sdext_as=Formatting sd-ext as {1}...")(ext_format));
2230 LOGINFO("Formatting sd-ext after partitioning, command: '%s'\n", Command.c_str());
2231 TWFunc::Exec_Cmd(Command);
2232 } else {
2233 SDext->Wipe(ext_format);
Dees_Troy9350b8d2012-09-27 12:38:38 -04002234 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002235 }
2236 if (swap > 0) {
2237 Command = "mkswap " + format_device;
2238 if (ext > 0)
2239 Command += "3";
2240 else
2241 Command += "2";
Vojtech Bocek05534202013-09-11 08:11:56 +02002242 TWFunc::Exec_Cmd(Command);
Dees_Troy9350b8d2012-09-27 12:38:38 -04002243 }
2244
Ethan Yonker483e9f42016-01-11 22:21:18 -06002245 // recreate TWRP folder and rewrite settings - these will be gone after sdcard is partitioned
2246 if (SDCard->Mount(true)) {
2247 string TWRP_Folder = SDCard->Mount_Point + "/TWRP";
2248 mkdir(TWRP_Folder.c_str(), 0777);
2249 DataManager::Flush();
2250 }
2251
Dees_Troy9350b8d2012-09-27 12:38:38 -04002252 Update_System_Details();
Ethan Yonker74db1572015-10-28 12:44:49 -05002253 gui_msg("part_complete=Partitioning complete.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002254 return true;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04002255}
Dees_Troya13d74f2013-03-24 08:54:55 -05002256
2257void TWPartitionManager::Get_Partition_List(string ListType, std::vector<PartitionList> *Partition_List) {
2258 std::vector<TWPartition*>::iterator iter;
2259 if (ListType == "mount") {
2260 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002261 if ((*iter)->Can_Be_Mounted) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002262 struct PartitionList part;
2263 part.Display_Name = (*iter)->Display_Name;
2264 part.Mount_Point = (*iter)->Mount_Point;
2265 part.selected = (*iter)->Is_Mounted();
2266 Partition_List->push_back(part);
2267 }
2268 }
2269 } else if (ListType == "storage") {
2270 char free_space[255];
2271 string Current_Storage = DataManager::GetCurrentStoragePath();
2272 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2273 if ((*iter)->Is_Storage) {
2274 struct PartitionList part;
2275 sprintf(free_space, "%llu", (*iter)->Free / 1024 / 1024);
2276 part.Display_Name = (*iter)->Storage_Name + " (";
2277 part.Display_Name += free_space;
2278 part.Display_Name += "MB)";
2279 part.Mount_Point = (*iter)->Storage_Path;
2280 if ((*iter)->Storage_Path == Current_Storage)
2281 part.selected = 1;
2282 else
2283 part.selected = 0;
2284 Partition_List->push_back(part);
2285 }
2286 }
2287 } else if (ListType == "backup") {
2288 char backup_size[255];
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002289 unsigned long long Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -05002290 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002291 if ((*iter)->Can_Be_Backed_Up && !(*iter)->Is_SubPartition && (*iter)->Is_Present) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002292 struct PartitionList part;
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002293 Backup_Size = (*iter)->Backup_Size;
2294 if ((*iter)->Has_SubPartition) {
2295 std::vector<TWPartition*>::iterator subpart;
2296
2297 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
2298 if ((*subpart)->Is_SubPartition && (*subpart)->Can_Be_Backed_Up && (*subpart)->Is_Present && (*subpart)->SubPartition_Of == (*iter)->Mount_Point)
2299 Backup_Size += (*subpart)->Backup_Size;
2300 }
2301 }
2302 sprintf(backup_size, "%llu", Backup_Size / 1024 / 1024);
Dees_Troya13d74f2013-03-24 08:54:55 -05002303 part.Display_Name = (*iter)->Backup_Display_Name + " (";
2304 part.Display_Name += backup_size;
2305 part.Display_Name += "MB)";
2306 part.Mount_Point = (*iter)->Backup_Path;
2307 part.selected = 0;
2308 Partition_List->push_back(part);
2309 }
2310 }
2311 } else if (ListType == "restore") {
2312 string Restore_List, restore_path;
2313 TWPartition* restore_part = NULL;
2314
2315 DataManager::GetValue("tw_restore_list", Restore_List);
2316 if (!Restore_List.empty()) {
2317 size_t start_pos = 0, end_pos = Restore_List.find(";", start_pos);
2318 while (end_pos != string::npos && start_pos < Restore_List.size()) {
2319 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04002320 struct PartitionList part;
2321 if (restore_path.compare("ADB_Backup") == 0) {
2322 part.Display_Name = "ADB Backup";
2323 part.Mount_Point = "ADB Backup";
2324 part.selected = 1;
2325 Partition_List->push_back(part);
2326 break;
2327 }
Dees_Troy59df9262013-06-19 14:53:57 -05002328 if ((restore_part = Find_Partition_By_Path(restore_path)) != NULL) {
Dees Troy4159aed2014-02-28 17:24:43 +00002329 if ((restore_part->Backup_Name == "recovery" && !restore_part->Can_Be_Backed_Up) || restore_part->Is_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002330 // Don't allow restore of recovery (causes problems on some devices)
Dees_Troy59df9262013-06-19 14:53:57 -05002331 // Don't add subpartitions to the list of items
Dees_Troya13d74f2013-03-24 08:54:55 -05002332 } else {
Dees_Troya13d74f2013-03-24 08:54:55 -05002333 part.Display_Name = restore_part->Backup_Display_Name;
2334 part.Mount_Point = restore_part->Backup_Path;
2335 part.selected = 1;
2336 Partition_List->push_back(part);
2337 }
2338 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002339 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05002340 }
2341 start_pos = end_pos + 1;
2342 end_pos = Restore_List.find(";", start_pos);
2343 }
2344 }
2345 } else if (ListType == "wipe") {
2346 struct PartitionList dalvik;
Ethan Yonker74db1572015-10-28 12:44:49 -05002347 dalvik.Display_Name = gui_parse_text("{@dalvik}");
Dees_Troya13d74f2013-03-24 08:54:55 -05002348 dalvik.Mount_Point = "DALVIK";
2349 dalvik.selected = 0;
2350 Partition_List->push_back(dalvik);
2351 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2352 if ((*iter)->Wipe_Available_in_GUI && !(*iter)->Is_SubPartition) {
2353 struct PartitionList part;
2354 part.Display_Name = (*iter)->Display_Name;
2355 part.Mount_Point = (*iter)->Mount_Point;
2356 part.selected = 0;
2357 Partition_List->push_back(part);
2358 }
2359 if ((*iter)->Has_Android_Secure) {
2360 struct PartitionList part;
2361 part.Display_Name = (*iter)->Backup_Display_Name;
2362 part.Mount_Point = (*iter)->Backup_Path;
2363 part.selected = 0;
2364 Partition_List->push_back(part);
2365 }
Dees_Troy74fb2e92013-04-15 14:35:47 +00002366 if ((*iter)->Has_Data_Media) {
2367 struct PartitionList datamedia;
2368 datamedia.Display_Name = (*iter)->Storage_Name;
2369 datamedia.Mount_Point = "INTERNAL";
2370 datamedia.selected = 0;
2371 Partition_List->push_back(datamedia);
2372 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002373 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002374 } else if (ListType == "flashimg") {
2375 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2376 if ((*iter)->Can_Flash_Img && (*iter)->Is_Present) {
2377 struct PartitionList part;
2378 part.Display_Name = (*iter)->Backup_Display_Name;
2379 part.Mount_Point = (*iter)->Backup_Path;
2380 part.selected = 0;
2381 Partition_List->push_back(part);
2382 }
2383 }
Ethan Yonker53796e72019-01-11 22:49:52 -06002384 if (DataManager::GetIntValue("tw_has_repack_tools") != 0 && DataManager::GetIntValue("tw_has_boot_slots") != 0) {
2385 TWPartition* boot = Find_Partition_By_Path("/boot");
2386 if (boot) {
2387 // Allow flashing kernels and ramdisks
2388 struct PartitionList repack_ramdisk;
2389 repack_ramdisk.Display_Name = gui_lookup("install_twrp_ramdisk", "Install Recovery Ramdisk");
2390 repack_ramdisk.Mount_Point = "/repack_ramdisk";
2391 repack_ramdisk.selected = 0;
2392 Partition_List->push_back(repack_ramdisk);
2393 /*struct PartitionList repack_kernel; For now let's leave repacking kernels under advanced only
2394 repack_kernel.Display_Name = gui_lookup("install_kernel", "Install Kernel");
2395 repack_kernel.Mount_Point = "/repack_kernel";
2396 repack_kernel.selected = 0;
2397 Partition_List->push_back(repack_kernel);*/
2398 }
2399 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002400 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00002401 LOGERR("Unknown list type '%s' requested for TWPartitionManager::Get_Partition_List\n", ListType.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -05002402 }
2403}
2404
2405int TWPartitionManager::Fstab_Processed(void) {
2406 return Partitions.size();
2407}
Dees_Troyd93bda52013-07-03 19:55:19 +00002408
2409void TWPartitionManager::Output_Storage_Fstab(void) {
2410 std::vector<TWPartition*>::iterator iter;
2411 char storage_partition[255];
bigbiff bigbiffe4bdb152019-03-23 18:33:17 -04002412 std::string Temp;
2413 std::string cacheDir = TWFunc::get_cache_dir();
2414
2415 if (cacheDir.empty()) {
2416 LOGINFO("Unable to find cache directory\n");
2417 return;
2418 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002419
2420 std::string storageFstab = TWFunc::get_cache_dir() + "recovery/storage.fstab";
2421 FILE *fp = fopen(storageFstab.c_str(), "w");
Dees_Troyd93bda52013-07-03 19:55:19 +00002422
2423 if (fp == NULL) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002424 gui_msg(Msg(msg::kError, "unable_to_open=Unable to open '{1}'.")(storageFstab));
Dees_Troyd93bda52013-07-03 19:55:19 +00002425 return;
2426 }
2427
2428 // Iterate through all partitions
2429 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2430 if ((*iter)->Is_Storage) {
2431 Temp = (*iter)->Storage_Path + ";" + (*iter)->Storage_Name + ";\n";
2432 strcpy(storage_partition, Temp.c_str());
2433 fwrite(storage_partition, sizeof(storage_partition[0]), strlen(storage_partition) / sizeof(storage_partition[0]), fp);
2434 }
2435 }
2436 fclose(fp);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02002437}
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002438
2439TWPartition *TWPartitionManager::Get_Default_Storage_Partition()
2440{
2441 TWPartition *res = NULL;
2442 for (std::vector<TWPartition*>::iterator iter = Partitions.begin(); iter != Partitions.end(); ++iter) {
Matt Mowera8a89d12016-12-30 18:10:37 -06002443 if (!(*iter)->Is_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002444 continue;
2445
Matt Mowera8a89d12016-12-30 18:10:37 -06002446 if ((*iter)->Is_Settings_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002447 return *iter;
2448
Matt Mowera8a89d12016-12-30 18:10:37 -06002449 if (!res)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002450 res = *iter;
2451 }
2452 return res;
2453}
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002454
2455bool TWPartitionManager::Enable_MTP(void) {
2456#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002457 if (mtppid) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002458 gui_err("mtp_already_enabled=MTP already enabled");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002459 return true;
2460 }
Ethan Yonker726a0202014-12-16 20:01:38 -06002461
2462 int mtppipe[2];
2463
2464 if (pipe(mtppipe) < 0) {
2465 LOGERR("Error creating MTP pipe\n");
2466 return false;
2467 }
2468
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002469 char old_value[PROPERTY_VALUE_MAX];
Matt Mowere07f0102017-02-23 17:40:00 -06002470 property_get("sys.usb.config", old_value, "");
2471 if (strcmp(old_value, "mtp,adb") != 0) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002472 char vendor[PROPERTY_VALUE_MAX];
2473 char product[PROPERTY_VALUE_MAX];
2474 property_set("sys.usb.config", "none");
2475 property_get("usb.vendor", vendor, "18D1");
2476 property_get("usb.product.mtpadb", product, "4EE2");
2477 string vendorstr = vendor;
2478 string productstr = product;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002479 TWFunc::write_to_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2480 TWFunc::write_to_file("/sys/class/android_usb/android0/idProduct", productstr);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002481 property_set("sys.usb.config", "mtp,adb");
2482 }
Matt Mower653a1702017-02-16 10:38:52 -06002483 /* To enable MTP debug, use the twrp command line feature:
Ethan Yonker6d154c42014-09-03 14:19:43 -05002484 * twrp set tw_mtp_debug 1
2485 */
2486 twrpMtp *mtp = new twrpMtp(DataManager::GetIntValue("tw_mtp_debug"));
Ethan Yonker1b039202015-01-30 10:08:48 -06002487 mtppid = mtp->forkserver(mtppipe);
2488 if (mtppid) {
2489 close(mtppipe[0]); // Host closes read side
2490 mtp_write_fd = mtppipe[1];
2491 DataManager::SetValue("tw_mtp_enabled", 1);
2492 Add_All_MTP_Storage();
2493 return true;
Ethan Yonker726a0202014-12-16 20:01:38 -06002494 } else {
2495 close(mtppipe[0]);
2496 close(mtppipe[1]);
Ethan Yonker74db1572015-10-28 12:44:49 -05002497 gui_err("mtp_fail=Failed to enable MTP");
Ethan Yonker1b039202015-01-30 10:08:48 -06002498 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002499 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002500#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002501 gui_err("no_mtp=MTP support not included");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002502#endif
2503 DataManager::SetValue("tw_mtp_enabled", 0);
2504 return false;
2505}
2506
Ethan Yonker1b039202015-01-30 10:08:48 -06002507void TWPartitionManager::Add_All_MTP_Storage(void) {
2508#ifdef TW_HAS_MTP
2509 std::vector<TWPartition*>::iterator iter;
2510
2511 if (!mtppid)
2512 return; // MTP is not enabled
2513
2514 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2515 if ((*iter)->Is_Storage && (*iter)->Is_Present && (*iter)->Mount(false))
2516 Add_Remove_MTP_Storage((*iter), MTP_MESSAGE_ADD_STORAGE);
2517 }
2518#else
2519 return;
2520#endif
2521}
2522
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002523bool TWPartitionManager::Disable_MTP(void) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002524 char old_value[PROPERTY_VALUE_MAX];
bigbiff bigbiffaf32bb92018-12-18 18:39:53 -05002525 property_set("sys.usb.ffs.mtp.ready", "0");
Matt Mowere07f0102017-02-23 17:40:00 -06002526 property_get("sys.usb.config", old_value, "");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002527 if (strcmp(old_value, "adb") != 0) {
2528 char vendor[PROPERTY_VALUE_MAX];
2529 char product[PROPERTY_VALUE_MAX];
2530 property_set("sys.usb.config", "none");
2531 property_get("usb.vendor", vendor, "18D1");
Matt Mowere07f0102017-02-23 17:40:00 -06002532 property_get("usb.product.adb", product, "D001");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002533 string vendorstr = vendor;
2534 string productstr = product;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002535 TWFunc::write_to_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2536 TWFunc::write_to_file("/sys/class/android_usb/android0/idProduct", productstr);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002537 usleep(2000);
2538 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002539#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002540 if (mtppid) {
Ethan Yonker8613dc02014-09-11 09:28:20 -05002541 LOGINFO("Disabling MTP\n");
2542 int status;
2543 kill(mtppid, SIGKILL);
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002544 mtppid = 0;
Ethan Yonker8613dc02014-09-11 09:28:20 -05002545 // We don't care about the exit value, but this prevents a zombie process
2546 waitpid(mtppid, &status, 0);
Ethan Yonker726a0202014-12-16 20:01:38 -06002547 close(mtp_write_fd);
2548 mtp_write_fd = -1;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002549 }
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002550#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002551 property_set("sys.usb.config", "adb");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002552#ifdef TW_HAS_MTP
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002553 DataManager::SetValue("tw_mtp_enabled", 0);
2554 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002555#endif
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002556 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002557}
Ethan Yonker726a0202014-12-16 20:01:38 -06002558
2559TWPartition* TWPartitionManager::Find_Partition_By_MTP_Storage_ID(unsigned int Storage_ID) {
2560 std::vector<TWPartition*>::iterator iter;
2561
2562 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2563 if ((*iter)->MTP_Storage_ID == Storage_ID)
2564 return (*iter);
2565 }
2566 return NULL;
2567}
2568
2569bool TWPartitionManager::Add_Remove_MTP_Storage(TWPartition* Part, int message_type) {
2570#ifdef TW_HAS_MTP
2571 struct mtpmsg mtp_message;
2572
2573 if (!mtppid)
2574 return false; // MTP is disabled
2575
2576 if (mtp_write_fd < 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002577 LOGINFO("MTP: mtp_write_fd is not set\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002578 return false;
2579 }
2580
2581 if (Part) {
Ethan Yonker4bfabab2014-12-29 09:15:37 -06002582 if (Part->MTP_Storage_ID == 0)
2583 return false;
Ethan Yonker726a0202014-12-16 20:01:38 -06002584 if (message_type == MTP_MESSAGE_REMOVE_STORAGE) {
2585 mtp_message.message_type = MTP_MESSAGE_REMOVE_STORAGE; // Remove
2586 LOGINFO("sending message to remove %i\n", Part->MTP_Storage_ID);
2587 mtp_message.storage_id = Part->MTP_Storage_ID;
2588 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002589 LOGINFO("error sending message to remove storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002590 return false;
2591 } else {
2592 LOGINFO("Message sent, remove storage ID: %i\n", Part->MTP_Storage_ID);
2593 return true;
2594 }
2595 } else if (message_type == MTP_MESSAGE_ADD_STORAGE && Part->Is_Mounted()) {
2596 mtp_message.message_type = MTP_MESSAGE_ADD_STORAGE; // Add
2597 mtp_message.storage_id = Part->MTP_Storage_ID;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002598 if (Part->Storage_Path.size() >= sizeof(mtp_message.path)) {
2599 LOGERR("Storage path '%s' too large for mtpmsg\n", Part->Storage_Path.c_str());
2600 return false;
2601 }
2602 strcpy(mtp_message.path, Part->Storage_Path.c_str());
2603 if (Part->Storage_Name.size() >= sizeof(mtp_message.display)) {
2604 LOGERR("Storage name '%s' too large for mtpmsg\n", Part->Storage_Name.c_str());
2605 return false;
2606 }
2607 strcpy(mtp_message.display, Part->Storage_Name.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002608 mtp_message.maxFileSize = Part->Get_Max_FileSize();
Ethan Yonker1b039202015-01-30 10:08:48 -06002609 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 -06002610 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002611 LOGINFO("error sending message to add storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002612 return false;
2613 } else {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002614 LOGINFO("Message sent, add storage ID: %i '%s'\n", Part->MTP_Storage_ID, mtp_message.path);
Ethan Yonker726a0202014-12-16 20:01:38 -06002615 return true;
2616 }
2617 } else {
2618 LOGERR("Unknown MTP message type: %i\n", message_type);
2619 }
2620 } else {
2621 // This hopefully never happens as the error handling should
2622 // occur in the calling function.
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002623 LOGINFO("TWPartitionManager::Add_Remove_MTP_Storage NULL partition given\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002624 }
2625 return true;
2626#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002627 gui_err("no_mtp=MTP support not included");
Ethan Yonker726a0202014-12-16 20:01:38 -06002628 DataManager::SetValue("tw_mtp_enabled", 0);
2629 return false;
2630#endif
2631}
2632
2633bool TWPartitionManager::Add_MTP_Storage(string Mount_Point) {
2634#ifdef TW_HAS_MTP
2635 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2636 if (Part) {
2637 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2638 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002639 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002640 }
2641#endif
2642 return false;
2643}
2644
2645bool TWPartitionManager::Add_MTP_Storage(unsigned int Storage_ID) {
2646#ifdef TW_HAS_MTP
2647 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2648 if (Part) {
2649 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2650 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002651 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002652 }
2653#endif
2654 return false;
2655}
2656
2657bool TWPartitionManager::Remove_MTP_Storage(string Mount_Point) {
2658#ifdef TW_HAS_MTP
2659 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2660 if (Part) {
2661 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2662 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002663 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002664 }
2665#endif
2666 return false;
2667}
2668
2669bool TWPartitionManager::Remove_MTP_Storage(unsigned int Storage_ID) {
2670#ifdef TW_HAS_MTP
2671 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2672 if (Part) {
2673 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2674 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002675 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002676 }
2677#endif
2678 return false;
2679}
Ethan Yonker96af84a2015-01-05 14:58:36 -06002680
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002681bool TWPartitionManager::Flash_Image(string& path, string& filename) {
Matt Mower23d8aae2017-01-06 14:30:33 -06002682 int partition_count = 0;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002683 TWPartition* flash_part = NULL;
bigbiffce8f83c2015-12-12 18:30:21 -05002684 string Flash_List, flash_path, full_filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002685 size_t start_pos = 0, end_pos = 0;
2686
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002687 full_filename = path + "/" + filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002688
bigbiffce8f83c2015-12-12 18:30:21 -05002689 gui_msg("image_flash_start=[IMAGE FLASH STARTED]");
2690 gui_msg(Msg("img_to_flash=Image to flash: '{1}'")(full_filename));
2691
2692 if (!TWFunc::Path_Exists(full_filename)) {
2693 if (!Mount_By_Path(full_filename, true)) {
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002694 return false;
2695 }
bigbiffce8f83c2015-12-12 18:30:21 -05002696 if (!TWFunc::Path_Exists(full_filename)) {
2697 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")(full_filename));
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002698 return false;
2699 }
2700 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002701
Ethan Yonker53796e72019-01-11 22:49:52 -06002702 DataManager::GetValue("tw_flash_partition", Flash_List);
2703 Repack_Type repack = REPLACE_NONE;
2704 if (Flash_List == "/repack_ramdisk;") {
2705 repack = REPLACE_RAMDISK;
2706 } else if (Flash_List == "/repack_kernel;") {
2707 repack = REPLACE_KERNEL;
2708 }
2709 if (repack != REPLACE_NONE) {
2710 Repack_Options_struct Repack_Options;
2711 Repack_Options.Type = repack;
2712 Repack_Options.Disable_Verity = false;
2713 Repack_Options.Disable_Force_Encrypt = false;
2714 Repack_Options.Backup_First = DataManager::GetIntValue("tw_repack_backup_first") != 0;
2715 return Repack_Images(full_filename, Repack_Options);
2716 }
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002717 PartitionSettings part_settings;
2718 part_settings.Backup_Folder = path;
2719 unsigned long long total_bytes = TWFunc::Get_File_Size(full_filename);
2720 ProgressTracking progress(total_bytes);
2721 part_settings.progress = &progress;
2722 part_settings.adbbackup = false;
2723 part_settings.PM_Method = PM_RESTORE;
Ethan Yonker74db1572015-10-28 12:44:49 -05002724 gui_msg("calc_restore=Calculating restore details...");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002725 if (!Flash_List.empty()) {
2726 end_pos = Flash_List.find(";", start_pos);
2727 while (end_pos != string::npos && start_pos < Flash_List.size()) {
2728 flash_path = Flash_List.substr(start_pos, end_pos - start_pos);
2729 flash_part = Find_Partition_By_Path(flash_path);
2730 if (flash_part != NULL) {
2731 partition_count++;
2732 if (partition_count > 1) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002733 gui_err("too_many_flash=Too many partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002734 return false;
2735 }
2736 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002737 gui_msg(Msg(msg::kError, "flash_unable_locate=Unable to locate '{1}' partition for flashing.")(flash_path));
Ethan Yonker96af84a2015-01-05 14:58:36 -06002738 return false;
2739 }
2740 start_pos = end_pos + 1;
2741 end_pos = Flash_List.find(";", start_pos);
2742 }
2743 }
2744
2745 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002746 gui_err("no_part_flash=No partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002747 return false;
2748 }
2749
2750 DataManager::SetProgress(0.0);
2751 if (flash_part) {
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002752 flash_part->Backup_FileName = filename;
2753 if (!flash_part->Flash_Image(&part_settings))
Ethan Yonker96af84a2015-01-05 14:58:36 -06002754 return false;
2755 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002756 gui_err("invalid_flash=Invalid flash partition specified.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002757 return false;
2758 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002759 gui_highlight("flash_done=IMAGE FLASH COMPLETED]");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002760 return true;
2761}
Ethan Yonker74db1572015-10-28 12:44:49 -05002762
2763void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value) {
2764 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2765 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002766 if (part->Is_Adopted_Storage) {
2767 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2768 part->Backup_Display_Name = part->Display_Name;
2769 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2770 } else {
2771 part->Display_Name = gui_lookup(resource_name, default_value);
2772 part->Backup_Display_Name = part->Display_Name;
2773 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002774 }
2775}
2776
2777void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value, const char* storage_resource_name, const char* storage_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) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06002781 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup("data_backup", "Data (excl. storage)");
Ethan Yonker66a19492015-12-10 10:19:45 -06002782 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
Ethan Yonker66a19492015-12-10 10:19:45 -06002783 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 if (part->Is_Storage)
2788 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2789 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002790 }
2791}
2792
Ethan Yonker01f4e032017-02-03 15:30:52 -06002793void 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) {
2794 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
2795 if (part) {
2796 if (part->Is_Adopted_Storage) {
2797 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup(backup_name, backup_default);
2798 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
2799 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
2800 } else {
2801 part->Display_Name = gui_lookup(resource_name, default_value);
2802 part->Backup_Display_Name = gui_lookup(backup_name, backup_default);
2803 if (part->Is_Storage)
2804 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
2805 }
2806 }
2807}
2808
Ethan Yonker74db1572015-10-28 12:44:49 -05002809void TWPartitionManager::Translate_Partition_Display_Names() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002810 LOGINFO("Translating partition display names\n");
Ethan Yonker74db1572015-10-28 12:44:49 -05002811 Translate_Partition("/system", "system", "System");
2812 Translate_Partition("/system_image", "system_image", "System Image");
2813 Translate_Partition("/vendor", "vendor", "Vendor");
2814 Translate_Partition("/vendor_image", "vendor_image", "Vendor Image");
2815 Translate_Partition("/cache", "cache", "Cache");
Ethan Yonker74db1572015-10-28 12:44:49 -05002816 Translate_Partition("/boot", "boot", "Boot");
2817 Translate_Partition("/recovery", "recovery", "Recovery");
2818 if (!datamedia) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06002819 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage");
Ethan Yonker74db1572015-10-28 12:44:49 -05002820 Translate_Partition("/sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2821 Translate_Partition("/internal_sd", "sdcard", "SDCard", "sdcard", "SDCard");
2822 Translate_Partition("/internal_sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
2823 Translate_Partition("/emmc", "sdcard", "SDCard", "sdcard", "SDCard");
Ethan Yonker01f4e032017-02-03 15:30:52 -06002824 } else {
2825 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05002826 }
Ethan Yonker01f4e032017-02-03 15:30:52 -06002827 Translate_Partition("/external_sd", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
2828 Translate_Partition("/external_sdcard", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05002829 Translate_Partition("/usb-otg", "usbotg", "USB OTG", "usbotg", "USB OTG");
2830 Translate_Partition("/sd-ext", "sdext", "SD-EXT");
2831
2832 // Android secure is a special case
2833 TWPartition* part = PartitionManager.Find_Partition_By_Path("/and-sec");
2834 if (part)
2835 part->Backup_Display_Name = gui_lookup("android_secure", "Android Secure");
2836
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002837 std::vector<TWPartition*>::iterator sysfs;
2838 for (sysfs = Partitions.begin(); sysfs != Partitions.end(); sysfs++) {
2839 if (!(*sysfs)->Sysfs_Entry.empty()) {
2840 Translate_Partition((*sysfs)->Mount_Point.c_str(), "autostorage", "Storage", "autostorage", "Storage");
2841 }
2842 }
2843
Ethan Yonker74db1572015-10-28 12:44:49 -05002844 // This updates the text on all of the storage selection buttons in the GUI
2845 DataManager::SetBackupFolder();
2846}
Ethan Yonker66a19492015-12-10 10:19:45 -06002847
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002848bool TWPartitionManager::Decrypt_Adopted() {
Ethan Yonker66a19492015-12-10 10:19:45 -06002849#ifdef TW_INCLUDE_CRYPTO
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002850 bool ret = false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002851 if (!Mount_By_Path("/data", false)) {
2852 LOGERR("Cannot decrypt adopted storage because /data will not mount\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002853 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002854 }
2855 LOGINFO("Decrypt adopted storage starting\n");
2856 char* xmlFile = PageManager::LoadFileToBuffer("/data/system/storage.xml", NULL);
2857 xml_document<> *doc = NULL;
2858 xml_node<>* volumes = NULL;
Ethan Yonker66a19492015-12-10 10:19:45 -06002859 string Primary_Storage_UUID = "";
2860 if (xmlFile != NULL) {
2861 LOGINFO("successfully loaded storage.xml\n");
2862 doc = new xml_document<>();
2863 doc->parse<0>(xmlFile);
2864 volumes = doc->first_node("volumes");
2865 if (volumes) {
2866 xml_attribute<>* psuuid = volumes->first_attribute("primaryStorageUuid");
2867 if (psuuid) {
2868 Primary_Storage_UUID = psuuid->value();
2869 }
2870 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002871 } else {
2872 LOGINFO("No /data/system/storage.xml for adopted storage\n");
2873 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002874 }
2875 std::vector<TWPartition*>::iterator adopt;
2876 for (adopt = Partitions.begin(); adopt != Partitions.end(); adopt++) {
2877 if ((*adopt)->Removable && (*adopt)->Is_Present) {
2878 if ((*adopt)->Decrypt_Adopted() == 0) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002879 ret = true;
Ethan Yonker66a19492015-12-10 10:19:45 -06002880 if (volumes) {
2881 xml_node<>* volume = volumes->first_node("volume");
2882 while (volume) {
2883 xml_attribute<>* guid = volume->first_attribute("partGuid");
2884 if (guid) {
2885 string GUID = (*adopt)->Adopted_GUID.c_str();
2886 GUID.insert(8, "-");
2887 GUID.insert(13, "-");
2888 GUID.insert(18, "-");
2889 GUID.insert(23, "-");
2890
2891 if (strcasecmp(GUID.c_str(), guid->value()) == 0) {
2892 xml_attribute<>* attr = volume->first_attribute("nickname");
nkk71ffb02bd2017-06-04 23:12:16 +03002893 if (attr && attr->value() && strlen(attr->value()) > 0) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002894 (*adopt)->Storage_Name = attr->value();
2895 (*adopt)->Display_Name = (*adopt)->Storage_Name;
2896 (*adopt)->Backup_Display_Name = (*adopt)->Storage_Name;
2897 LOGINFO("storage name from storage.xml is '%s'\n", attr->value());
2898 }
2899 attr = volume->first_attribute("fsUuid");
2900 if (attr && !Primary_Storage_UUID.empty() && strcmp(Primary_Storage_UUID.c_str(), attr->value()) == 0) {
2901 TWPartition* Dat = Find_Partition_By_Path("/data");
2902 if (Dat) {
2903 LOGINFO("Internal storage is found on adopted storage '%s'\n", (*adopt)->Display_Name.c_str());
2904 LOGINFO("Changing '%s' to point to '%s'\n", Dat->Symlink_Mount_Point.c_str(), (*adopt)->Storage_Path.c_str());
2905 (*adopt)->Symlink_Mount_Point = Dat->Symlink_Mount_Point;
2906 Dat->Symlink_Mount_Point = "";
2907 // Toggle mounts to ensure that the symlink mount point (probably /sdcard) is mounted to the right location
2908 Dat->UnMount(false);
2909 Dat->Mount(false);
2910 (*adopt)->UnMount(false);
2911 (*adopt)->Mount(false);
Ethan Yonker66a19492015-12-10 10:19:45 -06002912 }
2913 }
2914 break;
2915 }
2916 }
2917 volume = volume->next_sibling("volume");
2918 }
2919 }
nkk71ffb02bd2017-06-04 23:12:16 +03002920 Update_System_Details();
2921 Output_Partition((*adopt));
Ethan Yonker66a19492015-12-10 10:19:45 -06002922 }
2923 }
2924 }
2925 if (xmlFile) {
2926 doc->clear();
2927 delete doc;
2928 free(xmlFile);
2929 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002930 return ret;
Ethan Yonker66a19492015-12-10 10:19:45 -06002931#else
2932 LOGINFO("Decrypt_Adopted: no crypto support\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002933 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06002934#endif
2935}
Ethan Yonkerfcf3f242016-02-16 12:30:26 -06002936
2937void TWPartitionManager::Remove_Partition_By_Path(string Path) {
2938 std::vector<TWPartition*>::iterator iter;
2939 string Local_Path = TWFunc::Get_Root_Path(Path);
2940
2941 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2942 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
2943 LOGINFO("Found and erasing '%s' from partition list\n", Local_Path.c_str());
2944 Partitions.erase(iter);
2945 return;
2946 }
2947 }
2948}
Ethan Yonker1b190162016-12-05 15:25:19 -06002949
2950void TWPartitionManager::Set_Active_Slot(const string& Slot) {
2951 if (Slot != "A" && Slot != "B") {
2952 LOGERR("Set_Active_Slot invalid slot '%s'\n", Slot.c_str());
2953 return;
2954 }
2955 if (Active_Slot_Display == Slot)
2956 return;
2957 LOGINFO("Setting active slot %s\n", Slot.c_str());
2958#ifdef AB_OTA_UPDATER
2959 if (!Active_Slot_Display.empty()) {
2960 const hw_module_t *hw_module;
2961 boot_control_module_t *module;
2962 int ret;
2963 ret = hw_get_module("bootctrl", &hw_module);
2964 if (ret != 0) {
2965 LOGERR("Error getting bootctrl module.\n");
2966 } else {
2967 module = (boot_control_module_t*) hw_module;
2968 module->init(module);
2969 int slot_number = 0;
2970 if (Slot == "B")
2971 slot_number = 1;
2972 if (module->setActiveBootSlot(module, slot_number))
2973 gui_msg(Msg(msg::kError, "unable_set_boot_slot=Error changing bootloader boot slot to {1}")(Slot));
2974 }
2975 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
2976 }
2977#else
2978 LOGERR("Boot slot feature not present\n");
2979#endif
2980 Active_Slot_Display = Slot;
2981 if (Fstab_Processed())
2982 Update_System_Details();
2983}
2984string TWPartitionManager::Get_Active_Slot_Suffix() {
2985 if (Active_Slot_Display == "A")
2986 return "_a";
2987 return "_b";
2988}
2989string TWPartitionManager::Get_Active_Slot_Display() {
2990 return Active_Slot_Display;
2991}
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002992
Captain Throwback9d6feb52018-07-27 10:05:24 -04002993string TWPartitionManager::Get_Android_Root_Path() {
Chaosmasterf6e42ce2020-01-27 00:17:04 +01002994 if (property_get_bool("ro.twrp.sar", false))
dianlujitao58f1a632020-01-16 23:03:56 +08002995 return "/system_root";
2996 return "/system";
Captain Throwback9d6feb52018-07-27 10:05:24 -04002997}
2998
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002999void TWPartitionManager::Remove_Uevent_Devices(const string& Mount_Point) {
3000 std::vector<TWPartition*>::iterator iter;
3001
3002 for (iter = Partitions.begin(); iter != Partitions.end(); ) {
3003 if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Mount_Point) {
3004 TWPartition *part = *iter;
3005 LOGINFO("%s was removed by uevent data\n", (*iter)->Mount_Point.c_str());
3006 (*iter)->UnMount(false);
3007 rmdir((*iter)->Mount_Point.c_str());
3008 iter = Partitions.erase(iter);
3009 delete part;
3010 } else {
3011 iter++;
3012 }
3013 }
3014}
3015
3016void TWPartitionManager::Handle_Uevent(const Uevent_Block_Data& uevent_data) {
3017 std::vector<TWPartition*>::iterator iter;
3018
3019 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3020 if (!(*iter)->Sysfs_Entry.empty()) {
3021 string device;
3022 size_t wildcard = (*iter)->Sysfs_Entry.find("*");
3023 if (wildcard != string::npos) {
3024 device = (*iter)->Sysfs_Entry.substr(0, wildcard);
3025 } else {
3026 device = (*iter)->Sysfs_Entry;
3027 }
3028 if (device == uevent_data.sysfs_path.substr(0, device.size())) {
3029 // Found a match
3030 if (uevent_data.action == "add") {
3031 (*iter)->Primary_Block_Device = "/dev/block/" + uevent_data.block_device;
3032 (*iter)->Alternate_Block_Device = (*iter)->Primary_Block_Device;
3033 (*iter)->Is_Present = true;
3034 LOGINFO("Found a match '%s' '%s'\n", uevent_data.block_device.c_str(), device.c_str());
3035 if (!Decrypt_Adopted()) {
3036 LOGINFO("No adopted storage so finding actual block device\n");
3037 (*iter)->Find_Actual_Block_Device();
3038 }
3039 return;
3040 } else if (uevent_data.action == "remove") {
3041 (*iter)->Is_Present = false;
3042 (*iter)->Primary_Block_Device = "";
3043 (*iter)->Actual_Block_Device = "";
3044 Remove_Uevent_Devices((*iter)->Mount_Point);
3045 return;
3046 }
3047 }
3048 }
3049 }
3050 LOGINFO("Found no matching fstab entry for uevent device '%s' - %s\n", uevent_data.sysfs_path.c_str(), uevent_data.action.c_str());
3051}
3052
3053void TWPartitionManager::setup_uevent() {
3054 struct sockaddr_nl nls;
3055
3056 if (uevent_pfd.fd >= 0) {
3057 LOGINFO("uevent already set up\n");
3058 return;
3059 }
3060
3061 // Open hotplug event netlink socket
3062 memset(&nls,0,sizeof(struct sockaddr_nl));
3063 nls.nl_family = AF_NETLINK;
3064 nls.nl_pid = getpid();
3065 nls.nl_groups = -1;
3066 uevent_pfd.events = POLLIN;
3067 uevent_pfd.fd = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_KOBJECT_UEVENT);
3068 if (uevent_pfd.fd==-1) {
3069 LOGERR("uevent not root\n");
3070 return;
3071 }
3072
3073 // Listen to netlink socket
3074 if (::bind(uevent_pfd.fd, (struct sockaddr *) &nls, sizeof(struct sockaddr_nl)) < 0) {
3075 LOGERR("Bind failed\n");
3076 return;
3077 }
3078 set_select_fd();
3079 Coldboot();
3080}
3081
3082Uevent_Block_Data TWPartitionManager::get_event_block_values(char *buf, int len) {
3083 Uevent_Block_Data ret;
3084 ret.subsystem = "";
3085 char *ptr = buf;
3086 const char *end = buf + len;
3087
3088 buf[len - 1] = '\0';
3089 while (ptr < end) {
3090 if (strncmp(ptr, "ACTION=", strlen("ACTION=")) == 0) {
3091 ptr += strlen("ACTION=");
3092 ret.action = ptr;
3093 } else if (strncmp(ptr, "SUBSYSTEM=", strlen("SUBSYSTEM=")) == 0) {
3094 ptr += strlen("SUBSYSTEM=");
3095 ret.subsystem = ptr;
3096 } else if (strncmp(ptr, "DEVTYPE=", strlen("DEVTYPE=")) == 0) {
3097 ptr += strlen("DEVTYPE=");
3098 ret.type = ptr;
3099 } else if (strncmp(ptr, "DEVPATH=", strlen("DEVPATH=")) == 0) {
3100 ptr += strlen("DEVPATH=");
3101 ret.sysfs_path += ptr;
3102 } else if (strncmp(ptr, "DEVNAME=", strlen("DEVNAME=")) == 0) {
3103 ptr += strlen("DEVNAME=");
3104 ret.block_device += ptr;
3105 } else if (strncmp(ptr, "MAJOR=", strlen("MAJOR=")) == 0) {
3106 ptr += strlen("MAJOR=");
3107 ret.major = atoi(ptr);
3108 } else if (strncmp(ptr, "MINOR=", strlen("MINOR=")) == 0) {
3109 ptr += strlen("MINOR=");
3110 ret.minor = atoi(ptr);
3111 }
3112 ptr += strlen(ptr) + 1;
3113 }
3114 return ret;
3115}
3116
3117void TWPartitionManager::read_uevent() {
3118 char buf[1024];
3119
3120 int len = recv(uevent_pfd.fd, buf, sizeof(buf), MSG_DONTWAIT);
3121 if (len == -1) {
bigbiffd58ba182020-03-23 10:02:29 -04003122 LOGERR("recv error on uevent\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003123 return;
3124 }
3125 /*int i = 0; // Print all uevent output for test /debug
3126 while (i<len) {
3127 printf("%s\n", buf+i);
3128 i += strlen(buf+i)+1;
3129 }*/
3130 Uevent_Block_Data uevent_data = get_event_block_values(buf, len);
3131 if (uevent_data.subsystem == "block" && uevent_data.type == "disk") {
3132 PartitionManager.Handle_Uevent(uevent_data);
3133 }
3134}
3135
3136void TWPartitionManager::close_uevent() {
3137 if (uevent_pfd.fd > 0)
3138 close(uevent_pfd.fd);
3139 uevent_pfd.fd = -1;
3140}
3141
3142void TWPartitionManager::Add_Partition(TWPartition* Part) {
3143 Partitions.push_back(Part);
3144}
3145
3146void TWPartitionManager::Coldboot_Scan(std::vector<string> *sysfs_entries, const string& Path, int depth) {
3147 string Real_Path = Path;
3148 char real_path[PATH_MAX];
3149 if (realpath(Path.c_str(), &real_path[0])) {
3150 string Real_Path = real_path;
3151 std::vector<string>::iterator iter;
3152 for (iter = sysfs_entries->begin(); iter != sysfs_entries->end(); iter++) {
3153 if (Real_Path.find((*iter)) != string::npos) {
3154 string Write_Path = Real_Path + "/uevent";
3155 if (TWFunc::Path_Exists(Write_Path)) {
3156 const char* write_val = "add\n";
3157 TWFunc::write_to_file(Write_Path, write_val);
3158 break;
3159 }
3160 }
3161 }
3162 }
3163
3164 DIR* d = opendir(Path.c_str());
3165 if (d != NULL) {
3166 struct dirent* de;
3167 while ((de = readdir(d)) != NULL) {
3168 if (de->d_name[0] == '.' || (de->d_type != DT_DIR && depth > 0))
3169 continue;
3170 if (strlen(de->d_name) >= 4 && (strncmp(de->d_name, "ram", 3) == 0 || strncmp(de->d_name, "loop", 4) == 0))
3171 continue;
3172
3173 string item = Path + "/";
3174 item.append(de->d_name);
3175 Coldboot_Scan(sysfs_entries, item, depth + 1);
3176 }
3177 closedir(d);
3178 }
3179}
3180
3181void TWPartitionManager::Coldboot() {
3182 std::vector<TWPartition*>::iterator iter;
3183 std::vector<string> sysfs_entries;
3184
3185 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3186 if (!(*iter)->Sysfs_Entry.empty()) {
3187 size_t wildcard_pos = (*iter)->Sysfs_Entry.find("*");
3188 if (wildcard_pos == string::npos)
3189 wildcard_pos = (*iter)->Sysfs_Entry.size();
3190 sysfs_entries.push_back((*iter)->Sysfs_Entry.substr(0, wildcard_pos));
3191 }
3192 }
3193
3194 if (sysfs_entries.size() > 0)
3195 Coldboot_Scan(&sysfs_entries, "/sys/block", 0);
3196}
Ethan Yonker53796e72019-01-11 22:49:52 -06003197
3198bool TWPartitionManager::Prepare_Empty_Folder(const std::string& Folder) {
3199 if (TWFunc::Path_Exists(Folder))
3200 TWFunc::removeDir(Folder, false);
3201 return TWFunc::Recursive_Mkdir(Folder);
3202}
3203
3204bool TWPartitionManager::Prepare_Repack(TWPartition* Part, const std::string& Temp_Folder_Destination, const bool Create_Backup, const std::string& Backup_Name) {
3205 if (!Part) {
3206 LOGERR("Partition was null!\n");
3207 return false;
3208 }
3209 if (!Prepare_Empty_Folder(Temp_Folder_Destination))
3210 return false;
3211 std::string target_image = Temp_Folder_Destination + "boot.img";
3212 PartitionSettings part_settings;
3213 part_settings.Part = Part;
3214 if (Create_Backup) {
3215 if (Check_Backup_Name(Backup_Name, true, false) != 0)
3216 return false;
3217 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, part_settings.Backup_Folder);
3218 part_settings.Backup_Folder = part_settings.Backup_Folder + "/" + TWFunc::Get_Current_Date() + " " + Backup_Name + "/";
3219 if (!TWFunc::Recursive_Mkdir(part_settings.Backup_Folder))
3220 return false;
3221 } else
3222 part_settings.Backup_Folder = Temp_Folder_Destination;
3223 part_settings.adbbackup = false;
3224 part_settings.generate_digest = false;
3225 part_settings.generate_md5 = false;
3226 part_settings.PM_Method = PM_BACKUP;
3227 part_settings.progress = NULL;
3228 pid_t not_a_pid = 0;
3229 if (!Part->Backup(&part_settings, &not_a_pid))
3230 return false;
3231 std::string backed_up_image = part_settings.Backup_Folder;
3232 backed_up_image += Part->Backup_FileName;
3233 target_image = Temp_Folder_Destination + "boot.img";
3234 if (Create_Backup) {
3235 std::string source = part_settings.Backup_Folder + Part->Backup_FileName;
3236 if (TWFunc::copy_file(source, target_image, 0644) != 0) {
3237 LOGERR("Failed to copy backup file '%s' to temp folder target '%s'\n", source.c_str(), target_image.c_str());
3238 return false;
3239 }
3240 } else {
3241 if (rename(backed_up_image.c_str(), target_image.c_str()) != 0) {
3242 LOGERR("Failed to rename '%s' to '%s'\n", backed_up_image.c_str(), target_image.c_str());
3243 return false;
3244 }
3245 }
3246 return Prepare_Repack(target_image, Temp_Folder_Destination, false, false);
3247}
3248
3249bool TWPartitionManager::Prepare_Repack(const std::string& Source_Path, const std::string& Temp_Folder_Destination, const bool Copy_Source, const bool Create_Destination) {
3250 if (Create_Destination) {
3251 if (!Prepare_Empty_Folder(Temp_Folder_Destination))
3252 return false;
3253 }
3254 if (Copy_Source) {
3255 std::string destination = Temp_Folder_Destination + "/boot.img";
3256 if (TWFunc::copy_file(Source_Path, destination, 0644))
3257 return false;
3258 }
Mauronofrio Matarrese82a2f452019-04-04 22:39:12 +01003259 std::string command = "cd " + Temp_Folder_Destination + " && /sbin/magiskboot --unpack -h '" + Source_Path +"'";
Ethan Yonker53796e72019-01-11 22:49:52 -06003260 if (TWFunc::Exec_Cmd(command) != 0) {
3261 LOGINFO("Error unpacking %s!\n", Source_Path.c_str());
3262 gui_msg(Msg(msg::kError, "unpack_error=Error unpacking image."));
3263 return false;
3264 }
3265 return true;
3266}
3267
3268bool TWPartitionManager::Repack_Images(const std::string& Target_Image, const struct Repack_Options_struct& Repack_Options) {
3269 if (!TWFunc::Path_Exists("/sbin/magiskboot")) {
3270 LOGERR("Image repacking tool not present in this TWRP build!");
3271 return false;
3272 }
3273 DataManager::SetProgress(0);
3274 TWPartition* part = PartitionManager.Find_Partition_By_Path("/boot");
3275 if (part)
3276 gui_msg(Msg("unpacking_image=Unpacking {1}...")(part->Display_Name));
3277 else {
3278 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/boot"));
3279 return false;
3280 }
3281 if (!PartitionManager.Prepare_Repack(part, REPACK_ORIG_DIR, Repack_Options.Backup_First, gui_lookup("repack", "Repack")))
3282 return false;
3283 DataManager::SetProgress(.25);
3284 gui_msg(Msg("unpacking_image=Unpacking {1}...")(Target_Image));
3285 if (!PartitionManager.Prepare_Repack(Target_Image, REPACK_NEW_DIR, true))
3286 return false;
3287 DataManager::SetProgress(.5);
3288 gui_msg(Msg("repacking_image=Repacking {1}...")(part->Display_Name));
3289 std::string path = REPACK_NEW_DIR;
3290 if (Repack_Options.Type == REPLACE_KERNEL) {
3291 // When we replace the kernel, what we really do is copy the boot partition ramdisk into the new image's folder
3292 if (TWFunc::copy_file(REPACK_ORIG_DIR "ramdisk.cpio", REPACK_NEW_DIR "ramdisk.cpio", 0644)) {
3293 LOGERR("Failed to copy ramdisk\n");
3294 return false;
3295 }
3296 } else if (Repack_Options.Type == REPLACE_RAMDISK) {
3297 // Repack the ramdisk
3298 if (TWFunc::copy_file(REPACK_NEW_DIR "ramdisk.cpio", REPACK_ORIG_DIR "ramdisk.cpio", 0644)) {
3299 LOGERR("Failed to copy ramdisk\n");
3300 return false;
3301 }
3302 path = REPACK_ORIG_DIR;
3303 } else {
3304 LOGERR("Invalid repacking options specified\n");
3305 return false;
3306 }
3307 if (Repack_Options.Disable_Verity)
3308 LOGERR("Disabling verity is not implemented yet\n");
3309 if (Repack_Options.Disable_Force_Encrypt)
3310 LOGERR("Disabling force encrypt is not implemented yet\n");
3311 std::string command = "cd " + path + " && /sbin/magiskboot --repack " + path + "boot.img";
3312 if (TWFunc::Exec_Cmd(command) != 0) {
3313 gui_msg(Msg(msg::kError, "repack_error=Error repacking image."));
3314 return false;
3315 }
3316 DataManager::SetProgress(.75);
3317 std::string file = "new-boot.img";
3318 DataManager::SetValue("tw_flash_partition", "/boot;");
3319 if (!PartitionManager.Flash_Image(path, file)) {
3320 LOGINFO("Error flashing new image\n");
3321 return false;
3322 }
3323 DataManager::SetProgress(1);
3324 TWFunc::removeDir(REPACK_ORIG_DIR, false);
Ethan Yonker9f5dd312019-05-15 15:17:36 -05003325 if (part->SlotSelect && Repack_Options.Type == REPLACE_RAMDISK) {
3326 LOGINFO("Switching slots to flash ramdisk to both partitions\n");
3327 string Current_Slot = Get_Active_Slot_Display();
3328 if (Current_Slot == "A")
3329 Set_Active_Slot("B");
3330 else
3331 Set_Active_Slot("A");
3332 DataManager::SetProgress(.25);
3333 if (!PartitionManager.Prepare_Repack(part, REPACK_ORIG_DIR, Repack_Options.Backup_First, gui_lookup("repack", "Repack")))
3334 return false;
3335 if (TWFunc::copy_file(REPACK_NEW_DIR "ramdisk.cpio", REPACK_ORIG_DIR "ramdisk.cpio", 0644)) {
3336 LOGERR("Failed to copy ramdisk\n");
3337 return false;
3338 }
3339 path = REPACK_ORIG_DIR;
3340 command = "cd " + path + " && /sbin/magiskboot --repack " + path + "boot.img";
3341 if (TWFunc::Exec_Cmd(command) != 0) {
3342 gui_msg(Msg(msg::kError, "repack_error=Error repacking image."));
3343 return false;
3344 }
3345 DataManager::SetProgress(.75);
3346 std::string file = "new-boot.img";
3347 DataManager::SetValue("tw_flash_partition", "/boot;");
3348 if (!PartitionManager.Flash_Image(path, file)) {
3349 LOGINFO("Error flashing new image\n");
3350 return false;
3351 }
3352 DataManager::SetProgress(1);
3353 TWFunc::removeDir(REPACK_ORIG_DIR, false);
3354 Set_Active_Slot(Current_Slot);
3355 }
Ethan Yonker53796e72019-01-11 22:49:52 -06003356 TWFunc::removeDir(REPACK_NEW_DIR, false);
3357 return true;
3358}