blob: 17b2c5e7e750a3cf4a8afddc42b0130b66d3073c [file] [log] [blame]
Dees Troy3be70a82013-10-22 14:25:12 +00001/*
bigbiffaed1bdf2021-08-03 18:21:39 -04002 Copyright 2014 to 2021 TeamWin
Dees Troy3be70a82013-10-22 14:25:12 +00003 This file is part of TWRP/TeamWin Recovery Project.
4
5 TWRP is free software: you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation, either version 3 of the License, or
8 (at your option) any later version.
9
10 TWRP is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
14
15 You should have received a copy of the GNU General Public License
16 along with TWRP. If not, see <http://www.gnu.org/licenses/>.
17*/
Dees_Troy51a0e822012-09-05 15:24:24 -040018
19#include <stdio.h>
20#include <stdlib.h>
21#include <string.h>
22#include <sys/stat.h>
23#include <sys/vfs.h>
24#include <unistd.h>
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060025#include <map>
Dees_Troy5bf43922012-09-07 16:07:55 -040026#include <vector>
Dees_Troy63c8df72012-09-10 14:02:05 -040027#include <dirent.h>
28#include <time.h>
Dees_Troy8170a922012-09-18 15:40:25 -040029#include <errno.h>
30#include <fcntl.h>
bigbiff25d25b92020-06-19 16:07:38 -040031#include <grp.h>
32#include <pwd.h>
bigbiffce8f83c2015-12-12 18:30:21 -050033#include <zlib.h>
bigbiff bigbiff9c754052013-01-09 09:09:08 -050034#include <iostream>
35#include <iomanip>
Ethan Yonker8613dc02014-09-11 09:28:20 -050036#include <sys/wait.h>
Ethan Yonker483e9f42016-01-11 22:21:18 -060037#include <linux/fs.h>
38#include <sys/mount.h>
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060039#include <sys/poll.h>
40#include <sys/socket.h>
bigbiffcfa875c2021-06-20 16:20:27 -040041#include <sys/stat.h>
42#include <sys/types.h>
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060043#include <linux/types.h>
44#include <linux/netlink.h>
bigbiffee7b7ff2020-03-23 15:08:27 -040045#include <android-base/chrono_utils.h>
46#include <android-base/file.h>
47#include <android-base/logging.h>
bigbiff7ba75002020-04-11 20:47:09 -040048#include <android-base/properties.h>
bigbiffee7b7ff2020-03-23 15:08:27 -040049#include <android-base/strings.h>
50#include <fstab/fstab.h>
51#include <fs_avb/fs_avb.h>
52#include <fs_mgr.h>
53#include <fs_mgr_dm_linear.h>
54#include <fs_mgr_overlayfs.h>
bigbiffcfa875c2021-06-20 16:20:27 -040055#include <fs_mgr/roots.h>
bigbiffee7b7ff2020-03-23 15:08:27 -040056#include <libgsi/libgsi.h>
57#include <liblp/liblp.h>
bigbiffcfa875c2021-06-20 16:20:27 -040058#include <libgsi/libgsi.h>
59#include <liblp/builder.h>
60#include <libsnapshot/snapshot.h>
Ethan Yonker6e8c27a2016-12-22 17:55:57 -060061
Dees_Troy51a0e822012-09-05 15:24:24 -040062#include "variables.h"
Dees_Troy2673cec2013-04-02 20:22:16 +000063#include "twcommon.h"
Dees_Troy51a0e822012-09-05 15:24:24 -040064#include "partitions.hpp"
Dees_Troy5bf43922012-09-07 16:07:55 -040065#include "data.hpp"
bigbiffaed1bdf2021-08-03 18:21:39 -040066#include "startupArgs.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040067#include "twrp-functions.hpp"
Ethan Yonkerb5fab762016-01-28 14:03:33 -060068#include "fixContexts.hpp"
Ethan Yonker3fdcda42016-11-30 12:29:37 -060069#include "exclude.hpp"
Ethan Yonker4b94cfd2014-12-11 10:00:45 -060070#include "set_metadata.h"
bigbiff7abc5fe2015-01-17 16:53:12 -050071#include "tw_atomic.hpp"
Ethan Yonker74db1572015-10-28 12:44:49 -050072#include "gui/gui.hpp"
Ethan Yonker472f5062016-02-25 13:47:30 -060073#include "progresstracking.hpp"
bigbiff bigbiff56cf5642016-08-19 17:43:45 -040074#include "twrpDigestDriver.hpp"
bigbiffad58e1b2020-07-06 20:24:34 -040075#include "twrpRepacker.hpp"
bigbiffce8f83c2015-12-12 18:30:21 -050076#include "adbbu/libtwadbbu.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040077
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040078#ifdef TW_HAS_MTP
bigbiff bigbiffaf32bb92018-12-18 18:39:53 -050079#ifdef TW_HAS_LEGACY_MTP
80#include "mtp/legacy/mtp_MtpServer.hpp"
81#include "mtp/legacy/twrpMtp.hpp"
82#include "mtp/legacy/MtpMessage.hpp"
83#else
84#include "mtp/ffs/mtp_MtpServer.hpp"
85#include "mtp/ffs/twrpMtp.hpp"
86#include "mtp/ffs/MtpMessage.hpp"
87#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040088#endif
89
Dees Troy6f6441d2014-01-23 02:07:03 +000090extern "C" {
91 #include "cutils/properties.h"
Ethan Yonker74db1572015-10-28 12:44:49 -050092 #include "gui/gui.h"
Dees Troy6f6441d2014-01-23 02:07:03 +000093}
94
Dees_Troy5bf43922012-09-07 16:07:55 -040095#ifdef TW_INCLUDE_CRYPTO
bigbiffdf53b6c2022-01-17 19:31:52 -050096// #include "crypto/fde/cryptfs.h"
bigbiff7ba75002020-04-11 20:47:09 -040097#include "gui/rapidxml.hpp"
98#include "gui/pages.hpp"
99#ifdef TW_INCLUDE_FBE
bigbiffa957f072021-03-07 18:20:29 -0500100#include "Decrypt.h"
bigbiff7ba75002020-04-11 20:47:09 -0400101#ifdef TW_INCLUDE_FBE_METADATA_DECRYPT
102 #ifdef USE_FSCRYPT
bigbiffdf53b6c2022-01-17 19:31:52 -0500103 #include "cryptfs.h"
104 #include "fscrypt-common.h"
bigbiffa957f072021-03-07 18:20:29 -0500105 #include "MetadataCrypt.h"
bigbiffadc599e2020-05-28 19:36:30 +0000106 #endif
bigbiff7ba75002020-04-11 20:47:09 -0400107#endif
108#endif
Dees_Troy5bf43922012-09-07 16:07:55 -0400109#endif
Dees_Troy51a0e822012-09-05 15:24:24 -0400110
Ethan Yonker1b190162016-12-05 15:25:19 -0600111#ifdef AB_OTA_UPDATER
dianlujitaoa90dc462021-12-16 16:56:26 +0800112#include <android/hardware/boot/1.0/IBootControl.h>
113using android::hardware::boot::V1_0::CommandResult;
114using android::hardware::boot::V1_0::IBootControl;
Ethan Yonker1b190162016-12-05 15:25:19 -0600115#endif
116
bigbiffcfa875c2021-06-20 16:20:27 -0400117using android::fs_mgr::DestroyLogicalPartition;
bigbiffee7b7ff2020-03-23 15:08:27 -0400118using android::fs_mgr::Fstab;
119using android::fs_mgr::FstabEntry;
bigbiffcfa875c2021-06-20 16:20:27 -0400120using android::fs_mgr::MetadataBuilder;
Ethan Yonker6277c792014-09-15 14:54:30 -0500121
bigbiffee7b7ff2020-03-23 15:08:27 -0400122extern bool datamedia;
Noah Jacobson81d638d2019-04-28 00:10:07 -0400123std::vector<users_struct> Users_List;
124
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100125std::string additional_fstab = "/etc/additional.fstab";
126
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500127TWPartitionManager::TWPartitionManager(void) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400128 mtp_was_enabled = false;
Ethan Yonker726a0202014-12-16 20:01:38 -0600129 mtp_write_fd = -1;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600130 uevent_pfd.fd = -1;
bigbiff7abc5fe2015-01-17 16:53:12 -0500131 stop_backup.set_value(0);
Ethan Yonker1b190162016-12-05 15:25:19 -0600132#ifdef AB_OTA_UPDATER
133 char slot_suffix[PROPERTY_VALUE_MAX];
Ethan Yonkerfefe5912017-09-30 22:22:13 -0500134 property_get("ro.boot.slot_suffix", slot_suffix, "error");
135 if (strcmp(slot_suffix, "error") == 0)
136 property_get("ro.boot.slot", slot_suffix, "error");
Ethan Yonker1b190162016-12-05 15:25:19 -0600137 Active_Slot_Display = "";
Ethan Yonkerfefe5912017-09-30 22:22:13 -0500138 if (strcmp(slot_suffix, "_a") == 0 || strcmp(slot_suffix, "a") == 0)
Ethan Yonker1b190162016-12-05 15:25:19 -0600139 Set_Active_Slot("A");
140 else
141 Set_Active_Slot("B");
142#endif
bigbiff bigbiff34684ff2013-12-01 21:03:45 -0500143}
144
Captain Throwback37c3aef2021-10-06 11:38:18 -0400145void TWPartitionManager::Set_Crypto_State() {
146 char crypto_state[PROPERTY_VALUE_MAX];
147 property_get("ro.crypto.state", crypto_state, "error");
148 if (strcmp(crypto_state, "error") == 0)
149 property_set("ro.crypto.state", "encrypted");
150}
151
152int TWPartitionManager::Set_Crypto_Type(const char* crypto_type) {
153 char type_prop[PROPERTY_VALUE_MAX];
154 property_get("ro.crypto.type", type_prop, "error");
155 if (strcmp(type_prop, "error") == 0)
156 property_set("ro.crypto.type", crypto_type);
Captain Throwback18aea272020-10-29 12:43:17 -0400157 // Sleep for a bit so that services can start if needed
158 sleep(1);
159 return 0;
160}
161
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100162void inline Reset_Prop_From_Partition(std::string prop, std::string def, TWPartition *ven, TWPartition *odm) {
163 bool prop_on_odm = false, prop_on_vendor = false;
164 string prop_value;
165 if (odm) {
166 string odm_prop = TWFunc::Partition_Property_Get(prop, PartitionManager, "/odm", "etc/build.prop");
167 if (!odm_prop.empty()) {
168 prop_on_odm = true;
169 if (TWFunc::Property_Override(prop, odm_prop) == NOT_AVAILABLE) {
170 LOGERR("Unable to override '%s' due to missing libresetprop\n", prop.c_str());
171 } else {
172 prop_value = android::base::GetProperty(prop, "");
173 LOGINFO("Setting '%s' to '%s' from /odm/etc/build.prop\n", prop.c_str(), prop_value.c_str());
174 }
175 }
176 }
177 if (ven) {
178 string vendor_prop = TWFunc::Partition_Property_Get(prop, PartitionManager, "/vendor", "build.prop");
179 if (!vendor_prop.empty()) {
180 prop_on_vendor = true;
181 if (TWFunc::Property_Override(prop, vendor_prop) == NOT_AVAILABLE) {
182 LOGERR("Unable to override '%s' due to missing libresetprop\n", prop.c_str());
183 } else {
184 prop_value = android::base::GetProperty(prop, "");
185 LOGINFO("Setting '%s' to '%s' from /vendor/build.prop\n", prop.c_str(), prop_value.c_str());
186 }
187 }
188 }
189 if (!prop_on_odm && !prop_on_vendor && !def.empty()) {
190 if (TWFunc::Property_Override(prop, def) == NOT_AVAILABLE) {
191 LOGERR("Unable to override '%s' due to missing libresetprop\n", prop.c_str());
192 } else {
193 prop_value = android::base::GetProperty(prop, "");
194 LOGINFO("Setting '%s' to default value (%s)\n", prop.c_str(), prop_value.c_str());
195 }
196 }
197 prop_value = android::base::GetProperty(prop, "");
198 if (!prop_on_odm && !prop_on_vendor && !prop_value.empty() && def.empty()) {
199 if(TWFunc::Delete_Property(prop) == NOT_AVAILABLE) {
200 LOGERR("Unable to delete '%s' due to missing libresetprop\n", prop.c_str());
201 } else {
202 LOGINFO("Deleting property '%s'\n", prop.c_str());
203 }
204 }
205}
206
bigbiff22851b92021-09-01 16:46:57 -0400207int TWPartitionManager::Process_Fstab(string Fstab_Filename, bool Display_Error, bool recovery_mode) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400208 FILE *fstabFile;
209 char fstab_line[MAX_FSTAB_LINE_LENGTH];
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100210 bool parse_userdata = false;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600211 std::map<string, Flags_Map> twrp_flags;
212
213 fstabFile = fopen("/etc/twrp.flags", "rt");
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100214 if (Get_Super_Status()) {
215 Setup_Super_Devices();
216 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600217 if (fstabFile != NULL) {
Ian Macdonald160e8d32020-09-29 20:47:31 +0200218 LOGINFO("Reading /etc/twrp.flags\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600219 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600220 size_t line_size = strlen(fstab_line);
221 if (fstab_line[line_size - 1] != '\n')
222 fstab_line[line_size] = '\n';
223 Flags_Map line_flags;
224 line_flags.Primary_Block_Device = "";
225 line_flags.Alternate_Block_Device = "";
226 line_flags.fstab_line = (char*)malloc(MAX_FSTAB_LINE_LENGTH);
227 if (!line_flags.fstab_line) {
228 LOGERR("malloc error on line_flags.fstab_line\n");
229 return false;
230 }
231 memcpy(line_flags.fstab_line, fstab_line, MAX_FSTAB_LINE_LENGTH);
232 bool found_separator = false;
233 char *fs_loc = NULL;
234 char *block_loc = NULL;
235 char *flags_loc = NULL;
236 size_t index, item_index = 0;
237 for (index = 0; index < line_size; index++) {
238 if (fstab_line[index] <= 32) {
239 fstab_line[index] = '\0';
240 found_separator = true;
241 } else if (found_separator) {
242 if (item_index == 0) {
243 fs_loc = fstab_line + index;
244 } else if (item_index == 1) {
245 block_loc = fstab_line + index;
246 } else if (item_index > 1) {
247 char *ptr = fstab_line + index;
248 if (*ptr == '/') {
249 line_flags.Alternate_Block_Device = ptr;
250 } else if (strlen(ptr) > strlen("flags=") && strncmp(ptr, "flags=", strlen("flags=")) == 0) {
251 flags_loc = ptr;
252 // Once we find the flags=, we're done scanning the line
253 break;
254 }
255 }
256 found_separator = false;
257 item_index++;
258 }
259 }
260 if (block_loc)
261 line_flags.Primary_Block_Device = block_loc;
262 if (fs_loc)
263 line_flags.File_System = fs_loc;
264 if (flags_loc)
265 line_flags.Flags = flags_loc;
266 string Mount_Point = fstab_line;
267 twrp_flags[Mount_Point] = line_flags;
268 memset(fstab_line, 0, sizeof(fstab_line));
269 }
270 fclose(fstabFile);
271 }
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100272 TWPartition *data = NULL;
273 TWPartition *meta = NULL;
274parse:
Dees_Troy5bf43922012-09-07 16:07:55 -0400275 fstabFile = fopen(Fstab_Filename.c_str(), "rt");
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100276 if (!parse_userdata && fstabFile == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000277 LOGERR("Critical Error: Unable to open fstab at '%s'.\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400278 return false;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600279 } else
280 LOGINFO("Reading %s\n", Fstab_Filename.c_str());
Dees_Troy5bf43922012-09-07 16:07:55 -0400281
282 while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600283 if (strstr(fstab_line, "swap"))
284 continue; // Skip swap in recovery
285
bigbiff998f8392021-08-06 19:19:33 -0400286 if (fstab_line[0] == '#')
287 continue;
288
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100289 if (parse_userdata) {
290 if (strstr(fstab_line, "/metadata") && !strstr(fstab_line, "/data")) {
sekaiacg01ac7722023-08-01 09:32:00 +0800291 if (meta) {
292 Partitions.erase(std::find(Partitions.begin(), Partitions.end(), meta));
293 delete meta;
294 meta = NULL;
295 }
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100296 } else if (strstr(fstab_line, "/data")) {
sekaiacg01ac7722023-08-01 09:32:00 +0800297 if (data) {
298 Partitions.erase(std::find(Partitions.begin(), Partitions.end(), data));
299 delete data;
300 data = NULL;
301 }
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100302 } else {
303 continue;
304 }
305 }
306
307
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600308 size_t line_size = strlen(fstab_line);
309 if (fstab_line[line_size - 1] != '\n')
310 fstab_line[line_size] = '\n';
Dees_Troy2a923582012-09-20 12:13:34 -0400311
Matt Mower2b2dd152016-04-26 11:24:08 -0500312 TWPartition* partition = new TWPartition();
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100313 if (partition->Process_Fstab_Line(fstab_line, Display_Error, parse_userdata ? NULL : &twrp_flags)) {
314 if (partition->Mount_Point == "/data") data = partition;
315 if (partition->Mount_Point == "/metadata") meta = partition;
316 if (partition->Is_Super && !Prepare_Super_Volume(partition)) {
317 goto clear;
318 }
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500319 Partitions.push_back(partition);
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100320 }
Matt Mower72c87ce2016-04-26 14:34:56 -0500321 else
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100322clear:
Dees_Troy5bf43922012-09-07 16:07:55 -0400323 delete partition;
Matt Mower2b2dd152016-04-26 11:24:08 -0500324
325 memset(fstab_line, 0, sizeof(fstab_line));
Dees_Troy5bf43922012-09-07 16:07:55 -0400326 }
327 fclose(fstabFile);
Matt Mower72c87ce2016-04-26 14:34:56 -0500328
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100329 if (!parse_userdata && twrp_flags.size() > 0) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600330 LOGINFO("Processing remaining twrp.flags\n");
331 // Add any items from twrp.flags that did not exist in the recovery.fstab
332 for (std::map<string, Flags_Map>::iterator mapit=twrp_flags.begin(); mapit!=twrp_flags.end(); mapit++) {
333 if (Find_Partition_By_Path(mapit->first) == NULL) {
334 TWPartition* partition = new TWPartition();
bigbiff8da46fa2020-09-08 16:42:34 -0400335 if (partition->Process_Fstab_Line(mapit->second.fstab_line, Display_Error, NULL))
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600336 Partitions.push_back(partition);
337 else
338 delete partition;
339 }
340 if (mapit->second.fstab_line)
341 free(mapit->second.fstab_line);
342 mapit->second.fstab_line = NULL;
343 }
344 }
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100345 TWPartition* ven = PartitionManager.Find_Partition_By_Path("/vendor");
346 TWPartition* odm = PartitionManager.Find_Partition_By_Path("/odm");
347 if (!parse_userdata) {
348
349 if (ven) ven->Mount(true);
350 if (odm) odm->Mount(true);
351 if (TWFunc::Find_Fstab(Fstab_Filename)) {
352 string service;
353 LOGINFO("Fstab: %s\n", Fstab_Filename.c_str());
354 TWFunc::copy_file(Fstab_Filename, additional_fstab, 0600, false);
355 Fstab_Filename = additional_fstab;
356 property_set("fstab.additional", "1");
357 TWFunc::Get_Service_From(ven, "keymaster", service);
Captain Throwback500c3d22023-08-28 17:20:35 -0400358 LOGINFO("Keymaster version: '%s'\n", TWFunc::Get_Version_From_Service(service).c_str());
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100359 property_set("keymaster_ver", TWFunc::Get_Version_From_Service(service).c_str());
360 parse_userdata = true;
361 Reset_Prop_From_Partition("ro.crypto.dm_default_key.options_format.version", "", ven, odm);
362 Reset_Prop_From_Partition("ro.crypto.volume.metadata.method", "", ven, odm);
363 Reset_Prop_From_Partition("ro.crypto.volume.options", "", ven, odm);
364 Reset_Prop_From_Partition("external_storage.projid.enabled", "", ven, odm);
365 Reset_Prop_From_Partition("external_storage.casefold.enabled", "", ven, odm);
366 Reset_Prop_From_Partition("external_storage.sdcardfs.enabled", "", ven, odm);
367 goto parse;
368 } else {
369 LOGINFO("Unable to parse vendor fstab\n");
370 }
Mohd Faraze3948ec2020-08-14 01:40:59 +0000371 }
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100372 if (ven) ven->UnMount(true);
373 if (odm) odm->UnMount(true);
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600374 LOGINFO("Done processing fstab files\n");
375
bigbiff22851b92021-09-01 16:46:57 -0400376 if (recovery_mode) {
377 Setup_Fstab_Partitions(Display_Error);
Mohd Faraze3948ec2020-08-14 01:40:59 +0000378 }
bigbiff7ba75002020-04-11 20:47:09 -0400379 return true;
380}
381
bigbiff22851b92021-09-01 16:46:57 -0400382void TWPartitionManager::Setup_Fstab_Partitions(bool Display_Error) {
383 TWPartition* settings_partition = NULL;
384 TWPartition* andsec_partition = NULL;
385 std::vector<TWPartition*>::iterator iter;
386 unsigned int storageid = 1 << 16; // upper 16 bits are for physical storage device, we pretend to have only one
387
388 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
389 (*iter)->Partition_Post_Processing(Display_Error);
390
391 if ((*iter)->Is_Storage) {
392 ++storageid;
393 (*iter)->MTP_Storage_ID = storageid;
394 }
395
396 if (!settings_partition && (*iter)->Is_Settings_Storage && (*iter)->Is_Present)
397 settings_partition = (*iter);
398 else
399 (*iter)->Is_Settings_Storage = false;
400
401 if (!andsec_partition && (*iter)->Has_Android_Secure && (*iter)->Is_Present)
402 andsec_partition = (*iter);
403 else
404 (*iter)->Has_Android_Secure = false;
bigbiff22851b92021-09-01 16:46:57 -0400405 }
406
me-cafebabed63fcaa2022-07-28 03:19:20 +0800407 Unlock_Block_Partitions();
408
bigbiff22851b92021-09-01 16:46:57 -0400409 //Setup Apex before decryption
410 TWPartition* sys = PartitionManager.Find_Partition_By_Path(PartitionManager.Get_Android_Root_Path());
411 TWPartition* ven = PartitionManager.Find_Partition_By_Path("/vendor");
412 if (sys) {
413 if (sys->Get_Super_Status()) {
414 sys->Mount(true);
415 if (ven) {
416 ven->Mount(true);
417 }
418 #ifdef TW_EXCLUDE_APEX
419 LOGINFO("Apex is disabled in this build\n");
420 #else
421 twrpApex apex;
422 if (!apex.loadApexImages()) {
423 LOGERR("Unable to load apex images from %s\n", APEX_DIR);
424 property_set("twrp.apex.loaded", "false");
425 } else {
426 property_set("twrp.apex.loaded", "true");
427 }
428 TWFunc::check_and_run_script("/sbin/resyncapex.sh", "apex");
429 #endif
430 }
431 }
432 #ifndef USE_VENDOR_LIBS
433 if (ven)
434 ven->UnMount(true);
435 if (sys)
436 sys->UnMount(true);
437 #endif
438
439 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) {
440 // Attempt to automatically identify /data/media emulated storage devices
441 TWPartition* Dat = Find_Partition_By_Path("/data");
442 if (Dat) {
443 LOGINFO("Using automatic handling for /data/media emulated storage device.\n");
444 datamedia = true;
445 Dat->Setup_Data_Media();
446 settings_partition = Dat;
447 // Since /data was not considered a storage partition earlier, we still need to assign an MTP ID
448 ++storageid;
449 Dat->MTP_Storage_ID = storageid;
450 }
451 }
452 if (!settings_partition) {
453 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
454 if ((*iter)->Is_Storage) {
455 settings_partition = (*iter);
456 break;
457 }
458 }
459 if (!settings_partition)
460 LOGERR("Unable to locate storage partition for storing settings file.\n");
461 }
462 if (!Write_Fstab()) {
463 if (Display_Error)
464 LOGERR("Error creating fstab\n");
465 else
466 LOGINFO("Error creating fstab\n");
467 }
468
469 if (andsec_partition) {
470 Setup_Android_Secure_Location(andsec_partition);
471 } else if (settings_partition) {
472 Setup_Android_Secure_Location(settings_partition);
473 }
474 if (settings_partition) {
475 Setup_Settings_Storage_Partition(settings_partition);
476 }
477
478 #ifdef TW_INCLUDE_CRYPTO
479 DataManager::SetValue(TW_IS_ENCRYPTED, 1);
480 Decrypt_Data();
481 #endif
482
483 Update_System_Details();
484 if (Get_Super_Status())
485 Setup_Super_Partition();
486 UnMount_Main_Partitions();
487 #ifdef AB_OTA_UPDATER
488 DataManager::SetValue("tw_active_slot", Get_Active_Slot_Display());
489 #endif
490 setup_uevent();
491}
492
bigbiff7ba75002020-04-11 20:47:09 -0400493int TWPartitionManager::Write_Fstab(void) {
494 FILE *fp;
495 std::vector<TWPartition*>::iterator iter;
496 string Line;
497
498 fp = fopen("/etc/fstab", "w");
499 if (fp == NULL) {
500 LOGINFO("Can not open /etc/fstab.\n");
501 return false;
502 }
503 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
504 if ((*iter)->Can_Be_Mounted) {
Mohd Faraz2a1e3752020-09-30 10:55:48 +0530505 Line = (*iter)->Actual_Block_Device + " " + (*iter)->Mount_Point + " " + (*iter)->Current_File_System +
506 ((*iter)->Mount_Read_Only ? " ro " : " rw ") + "0 0\n";
bigbiff7ba75002020-04-11 20:47:09 -0400507 fputs(Line.c_str(), fp);
508 }
509 // Handle subpartition tracking
510 if ((*iter)->Is_SubPartition) {
511 TWPartition* ParentPartition = Find_Partition_By_Path((*iter)->SubPartition_Of);
512 if (ParentPartition)
513 ParentPartition->Has_SubPartition = true;
514 else
515 LOGERR("Unable to locate parent partition '%s' of '%s'\n", (*iter)->SubPartition_Of.c_str(), (*iter)->Mount_Point.c_str());
516 }
517 }
518 fclose(fp);
519 return true;
520}
521
522void TWPartitionManager::Decrypt_Data() {
523 #ifdef TW_INCLUDE_CRYPTO
Ethan Yonkercceebb82014-11-18 10:17:59 -0600524 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
525 if (Decrypt_Data && Decrypt_Data->Is_Encrypted && !Decrypt_Data->Is_Decrypted) {
Captain Throwback37c3aef2021-10-06 11:38:18 -0400526 Set_Crypto_State();
bigbiff4fa02b52021-12-30 17:01:44 -0500527 TWPartition* Key_Directory_Partition = Find_Partition_By_Path(Decrypt_Data->Key_Directory);
528 if (Key_Directory_Partition != nullptr)
529 if (!Key_Directory_Partition->Is_Mounted())
530 Mount_By_Path(Decrypt_Data->Key_Directory, false);
531 if (!Decrypt_Data->Key_Directory.empty()) {
532 Set_Crypto_Type("file");
Ethan Yonker93382822018-11-01 15:25:31 -0500533#ifdef TW_INCLUDE_FBE_METADATA_DECRYPT
bigbiff7ba75002020-04-11 20:47:09 -0400534#ifdef USE_FSCRYPT
Mohd Faraz446ccbf2023-01-24 19:58:24 +0100535 if (android::vold::fscrypt_mount_metadata_encrypted(Decrypt_Data->Actual_Block_Device, Decrypt_Data->Mount_Point, false, false, Decrypt_Data->Current_File_System, TWFunc::Path_Exists(additional_fstab) ? additional_fstab : "")) {
bigbiffa957f072021-03-07 18:20:29 -0500536 std::string crypto_blkdev = android::base::GetProperty("ro.crypto.fs_crypto_blkdev", "error");
Mohd Faraze3948ec2020-08-14 01:40:59 +0000537 Decrypt_Data->Decrypted_Block_Device = crypto_blkdev;
bigbiff7ba75002020-04-11 20:47:09 -0400538 LOGINFO("Successfully decrypted metadata encrypted data partition with new block device: '%s'\n", crypto_blkdev.c_str());
bigbiff7ba75002020-04-11 20:47:09 -0400539#endif
Ethan Yonker93382822018-11-01 15:25:31 -0500540 Decrypt_Data->Is_Decrypted = true; // Needed to make the mount function work correctly
541 int retry_count = 10;
542 while (!Decrypt_Data->Mount(false) && --retry_count)
543 usleep(500);
544 if (Decrypt_Data->Mount(false)) {
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400545 if (!Decrypt_Data->Decrypt_FBE_DE()) {
bigbiff437d86f2021-09-26 16:57:14 -0400546 LOGERR("Unable to decrypt FBE device\n");
bigbiff bigbiff0be03b32019-08-27 20:50:31 -0400547 }
548
Ethan Yonker93382822018-11-01 15:25:31 -0500549 } else {
550 LOGINFO("Failed to mount data after metadata decrypt\n");
551 }
552 } else {
553 LOGINFO("Unable to decrypt metadata encryption\n");
554 }
555#else
556 LOGERR("Metadata FBE decrypt support not present in this TWRP\n");
557#endif
558 }
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600559 if (Decrypt_Data->Is_FBE) {
560 if (DataManager::GetIntValue(TW_CRYPTO_PWTYPE) == 0) {
561 if (Decrypt_Device("!") == 0) {
562 gui_msg("decrypt_success=Successfully decrypted with default password.");
563 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
564 } else {
565 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
566 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600567 }
568 } else {
Captain Throwback500c3d22023-08-28 17:20:35 -0400569 LOGINFO("FBE setup failed. Trying FDE...\n");
Captain Throwback37c3aef2021-10-06 11:38:18 -0400570 Set_Crypto_State();
571 Set_Crypto_Type("block");
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600572 int password_type = cryptfs_get_password_type();
573 if (password_type == CRYPT_TYPE_DEFAULT) {
574 LOGINFO("Device is encrypted with the default password, attempting to decrypt.\n");
575 if (Decrypt_Device("default_password") == 0) {
576 gui_msg("decrypt_success=Successfully decrypted with default password.");
577 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
578 } else {
579 gui_err("unable_to_decrypt=Unable to decrypt with default password.");
580 }
581 } else {
582 DataManager::SetValue("TW_CRYPTO_TYPE", password_type);
Noah Jacobson81d638d2019-04-28 00:10:07 -0400583 DataManager::SetValue("tw_crypto_pwtype_0", password_type);
Ethan Yonkerbd7492d2016-12-07 13:55:01 -0600584 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600585 }
586 }
bigbiff4fa02b52021-12-30 17:01:44 -0500587 if (Decrypt_Data && (!Decrypt_Data->Is_Encrypted || Decrypt_Data->Is_Decrypted)) {
Ethan Yonker66a19492015-12-10 10:19:45 -0600588 Decrypt_Adopted();
589 }
Ethan Yonkercceebb82014-11-18 10:17:59 -0600590#endif
Dees_Troy51a0e822012-09-05 15:24:24 -0400591}
592
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500593void TWPartitionManager::Setup_Settings_Storage_Partition(TWPartition* Part) {
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500594 DataManager::SetValue("tw_storage_path", Part->Storage_Path);
Ethan Yonkerc05c5982014-03-13 09:19:56 -0500595}
596
Matt Mowerbf4efa32014-04-14 23:25:26 -0500597void TWPartitionManager::Setup_Android_Secure_Location(TWPartition* Part) {
598 if (Part->Has_Android_Secure)
599 Part->Setup_AndSec();
Ethan Yonker6277c792014-09-15 14:54:30 -0500600 else if (!datamedia)
Matt Mowerbf4efa32014-04-14 23:25:26 -0500601 Part->Setup_AndSec();
Matt Mowerbf4efa32014-04-14 23:25:26 -0500602}
603
Dees_Troy8170a922012-09-18 15:40:25 -0400604void TWPartitionManager::Output_Partition_Logging(void) {
605 std::vector<TWPartition*>::iterator iter;
606
607 printf("\n\nPartition Logs:\n");
608 for (iter = Partitions.begin(); iter != Partitions.end(); iter++)
609 Output_Partition((*iter));
610}
611
612void TWPartitionManager::Output_Partition(TWPartition* Part) {
613 unsigned long long mb = 1048576;
614
Gary Peck004d48b2012-11-21 16:28:18 -0800615 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 -0400616 if (Part->Can_Be_Mounted) {
Gary Peck004d48b2012-11-21 16:28:18 -0800617 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 -0400618 }
Gary Peck004d48b2012-11-21 16:28:18 -0800619 printf("\n Flags: ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500620 if (Part->Can_Be_Mounted)
621 printf("Can_Be_Mounted ");
Gary Peck004d48b2012-11-21 16:28:18 -0800622 if (Part->Can_Be_Wiped)
623 printf("Can_Be_Wiped ");
Hashcodedabfd492013-08-29 22:45:30 -0700624 if (Part->Use_Rm_Rf)
625 printf("Use_Rm_Rf ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500626 if (Part->Can_Be_Backed_Up)
627 printf("Can_Be_Backed_Up ");
Gary Peck004d48b2012-11-21 16:28:18 -0800628 if (Part->Wipe_During_Factory_Reset)
629 printf("Wipe_During_Factory_Reset ");
630 if (Part->Wipe_Available_in_GUI)
631 printf("Wipe_Available_in_GUI ");
632 if (Part->Is_SubPartition)
633 printf("Is_SubPartition ");
634 if (Part->Has_SubPartition)
635 printf("Has_SubPartition ");
636 if (Part->Removable)
637 printf("Removable ");
638 if (Part->Is_Present)
639 printf("IsPresent ");
640 if (Part->Can_Be_Encrypted)
641 printf("Can_Be_Encrypted ");
642 if (Part->Is_Encrypted)
643 printf("Is_Encrypted ");
644 if (Part->Is_Decrypted)
645 printf("Is_Decrypted ");
646 if (Part->Has_Data_Media)
647 printf("Has_Data_Media ");
Dees_Troy83bd4832013-05-04 12:39:56 +0000648 if (Part->Can_Encrypt_Backup)
649 printf("Can_Encrypt_Backup ");
650 if (Part->Use_Userdata_Encryption)
651 printf("Use_Userdata_Encryption ");
Gary Peck004d48b2012-11-21 16:28:18 -0800652 if (Part->Has_Android_Secure)
653 printf("Has_Android_Secure ");
654 if (Part->Is_Storage)
655 printf("Is_Storage ");
Dees_Troya13d74f2013-03-24 08:54:55 -0500656 if (Part->Is_Settings_Storage)
657 printf("Is_Settings_Storage ");
Dees_Troy68cab492012-12-12 19:29:35 +0000658 if (Part->Ignore_Blkid)
659 printf("Ignore_Blkid ");
Ethan Yonker253368a2014-11-25 15:00:52 -0600660 if (Part->Mount_To_Decrypt)
661 printf("Mount_To_Decrypt ");
Ethan Yonker96af84a2015-01-05 14:58:36 -0600662 if (Part->Can_Flash_Img)
663 printf("Can_Flash_Img ");
Ethan Yonker66a19492015-12-10 10:19:45 -0600664 if (Part->Is_Adopted_Storage)
665 printf("Is_Adopted_Storage ");
Ethan Yonker1b190162016-12-05 15:25:19 -0600666 if (Part->SlotSelect)
667 printf("SlotSelect ");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600668 if (Part->Mount_Read_Only)
669 printf("Mount_Read_Only ");
bigbiffcfa875c2021-06-20 16:20:27 -0400670 if (Part->Is_Super)
671 printf("Is_Super ");
Gary Peck004d48b2012-11-21 16:28:18 -0800672 printf("\n");
673 if (!Part->SubPartition_Of.empty())
674 printf(" SubPartition_Of: %s\n", Part->SubPartition_Of.c_str());
675 if (!Part->Symlink_Path.empty())
676 printf(" Symlink_Path: %s\n", Part->Symlink_Path.c_str());
677 if (!Part->Symlink_Mount_Point.empty())
678 printf(" Symlink_Mount_Point: %s\n", Part->Symlink_Mount_Point.c_str());
679 if (!Part->Primary_Block_Device.empty())
680 printf(" Primary_Block_Device: %s\n", Part->Primary_Block_Device.c_str());
681 if (!Part->Alternate_Block_Device.empty())
682 printf(" Alternate_Block_Device: %s\n", Part->Alternate_Block_Device.c_str());
683 if (!Part->Decrypted_Block_Device.empty())
684 printf(" Decrypted_Block_Device: %s\n", Part->Decrypted_Block_Device.c_str());
dianlujitao4879b372018-12-03 18:45:47 +0800685 if (!Part->Crypto_Key_Location.empty())
Ethan Yonker253368a2014-11-25 15:00:52 -0600686 printf(" Crypto_Key_Location: %s\n", Part->Crypto_Key_Location.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800687 if (Part->Length != 0)
688 printf(" Length: %i\n", Part->Length);
689 if (!Part->Display_Name.empty())
690 printf(" Display_Name: %s\n", Part->Display_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500691 if (!Part->Storage_Name.empty())
692 printf(" Storage_Name: %s\n", Part->Storage_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800693 if (!Part->Backup_Path.empty())
694 printf(" Backup_Path: %s\n", Part->Backup_Path.c_str());
695 if (!Part->Backup_Name.empty())
696 printf(" Backup_Name: %s\n", Part->Backup_Name.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -0500697 if (!Part->Backup_Display_Name.empty())
698 printf(" Backup_Display_Name: %s\n", Part->Backup_Display_Name.c_str());
Gary Peck004d48b2012-11-21 16:28:18 -0800699 if (!Part->Backup_FileName.empty())
700 printf(" Backup_FileName: %s\n", Part->Backup_FileName.c_str());
701 if (!Part->Storage_Path.empty())
702 printf(" Storage_Path: %s\n", Part->Storage_Path.c_str());
703 if (!Part->Current_File_System.empty())
704 printf(" Current_File_System: %s\n", Part->Current_File_System.c_str());
705 if (!Part->Fstab_File_System.empty())
706 printf(" Fstab_File_System: %s\n", Part->Fstab_File_System.c_str());
707 if (Part->Format_Block_Size != 0)
Ethan Yonkerc798c9c2015-10-09 11:15:26 -0500708 printf(" Format_Block_Size: %lu\n", Part->Format_Block_Size);
Dees_Troy094207a2012-09-26 12:00:39 -0400709 if (!Part->MTD_Name.empty())
710 printf(" MTD_Name: %s\n", Part->MTD_Name.c_str());
Matt Mower029a82d2017-01-08 13:12:38 -0600711 printf(" Backup_Method: %s\n", Part->Backup_Method_By_Name().c_str());
Hashcode62bd9e02013-11-19 21:59:42 -0800712 if (Part->Mount_Flags || !Part->Mount_Options.empty())
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600713 printf(" Mount_Flags: %i, Mount_Options: %s\n", Part->Mount_Flags, Part->Mount_Options.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -0600714 if (Part->MTP_Storage_ID)
715 printf(" MTP_Storage_ID: %i\n", Part->MTP_Storage_ID);
Ethan Yonker93382822018-11-01 15:25:31 -0500716 if (!Part->Key_Directory.empty())
717 printf(" Metadata Key Directory: %s\n", Part->Key_Directory.c_str());
Ethan Yonker6277c792014-09-15 14:54:30 -0500718 printf("\n");
Dees_Troy8170a922012-09-18 15:40:25 -0400719}
720
Dees_Troy51a0e822012-09-05 15:24:24 -0400721int TWPartitionManager::Mount_By_Path(string Path, bool Display_Error) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400722 std::vector<TWPartition*>::iterator iter;
723 int ret = false;
724 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400725 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400726
bigbiffd58ba182020-03-23 10:02:29 -0400727 if (Local_Path == "/tmp" || Local_Path == "/")
Dees_Troy43d8b002012-09-17 16:00:01 -0400728 return true;
729
Dees_Troy5bf43922012-09-07 16:07:55 -0400730 // Iterate through all partitions
731 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400732 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400733 ret = (*iter)->Mount(Display_Error);
734 found = true;
Dees_Troy51127312012-09-08 13:08:49 -0400735 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400736 (*iter)->Mount(Display_Error);
Dees_Troy51127312012-09-08 13:08:49 -0400737 }
Dees_Troy5bf43922012-09-07 16:07:55 -0400738 }
739 if (found) {
740 return ret;
741 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500742 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 -0400743 }
744 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400745}
746
Dees_Troy51a0e822012-09-05 15:24:24 -0400747int TWPartitionManager::UnMount_By_Path(string Path, bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400748 std::vector<TWPartition*>::iterator iter;
749 int ret = false;
750 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -0400751 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy51127312012-09-08 13:08:49 -0400752
753 // Iterate through all partitions
754 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400755 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Dees_Troy51127312012-09-08 13:08:49 -0400756 ret = (*iter)->UnMount(Display_Error);
757 found = true;
758 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
759 (*iter)->UnMount(Display_Error);
760 }
761 }
762 if (found) {
763 return ret;
764 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500765 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 -0400766 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000767 LOGINFO("UnMount: Unable to find partition for path '%s'\n", Local_Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400768 }
769 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400770}
771
Dees_Troy51a0e822012-09-05 15:24:24 -0400772int TWPartitionManager::Is_Mounted_By_Path(string Path) {
Dees_Troy51127312012-09-08 13:08:49 -0400773 TWPartition* Part = Find_Partition_By_Path(Path);
774
775 if (Part)
776 return Part->Is_Mounted();
777 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000778 LOGINFO("Is_Mounted: Unable to find partition for path '%s'\n", Path.c_str());
Dees_Troy51127312012-09-08 13:08:49 -0400779 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400780}
781
Dees_Troy5bf43922012-09-07 16:07:55 -0400782int TWPartitionManager::Mount_Current_Storage(bool Display_Error) {
Dees_Troy51127312012-09-08 13:08:49 -0400783 string current_storage_path = DataManager::GetCurrentStoragePath();
784
785 if (Mount_By_Path(current_storage_path, Display_Error)) {
786 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
787 if (FreeStorage)
788 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
789 return true;
790 }
791 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -0400792}
793
Dees_Troy5bf43922012-09-07 16:07:55 -0400794int TWPartitionManager::Mount_Settings_Storage(bool Display_Error) {
795 return Mount_By_Path(DataManager::GetSettingsStoragePath(), Display_Error);
796}
797
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600798TWPartition* TWPartitionManager::Find_Partition_By_Path(const string& Path) {
Dees_Troy5bf43922012-09-07 16:07:55 -0400799 std::vector<TWPartition*>::iterator iter;
Dees_Troy38bd7602012-09-14 13:33:53 -0400800 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy5bf43922012-09-07 16:07:55 -0400801
dianlujitaob76a73a2020-04-30 20:33:20 +0800802 if (Local_Path == "/system")
803 Local_Path = Get_Android_Root_Path();
Dees_Troy5bf43922012-09-07 16:07:55 -0400804 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -0400805 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path))
Dees_Troy5bf43922012-09-07 16:07:55 -0400806 return (*iter);
807 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400808 return NULL;
809}
810
Ethan Yonker6e8c27a2016-12-22 17:55:57 -0600811TWPartition* TWPartitionManager::Find_Partition_By_Block_Device(const string& Block_Device) {
812 std::vector<TWPartition*>::iterator iter;
813
814 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
815 if ((*iter)->Primary_Block_Device == Block_Device || (!(*iter)->Actual_Block_Device.empty() && (*iter)->Actual_Block_Device == Block_Device))
816 return (*iter);
817 }
818 return NULL;
819}
820
Ethan Yonker53796e72019-01-11 22:49:52 -0600821int TWPartitionManager::Check_Backup_Name(const std::string& Backup_Name, bool Display_Error, bool Must_Be_Unique) {
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400822 // Check the backup name to ensure that it is the correct size and contains only valid characters
823 // and that a backup with that name doesn't already exist
824 char backup_name[MAX_BACKUP_NAME_LEN];
825 char backup_loc[255], tw_image_dir[255];
826 int copy_size;
827 int index, cur_char;
Ethan Yonker53796e72019-01-11 22:49:52 -0600828 string Backup_Loc;
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400829
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400830 copy_size = Backup_Name.size();
831 // Check size
832 if (copy_size > MAX_BACKUP_NAME_LEN) {
833 if (Display_Error)
Ethan Yonker74db1572015-10-28 12:44:49 -0500834 gui_err("backup_name_len=Backup name is too long.");
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400835 return -2;
836 }
837
838 // Check each character
839 strncpy(backup_name, Backup_Name.c_str(), copy_size);
Dees_Troya13d74f2013-03-24 08:54:55 -0500840 if (copy_size == 1 && strncmp(backup_name, "0", 1) == 0)
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400841 return 0; // A "0" (zero) means to use the current timestamp for the backup name
842 for (index=0; index<copy_size; index++) {
843 cur_char = (int)backup_name[index];
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500844 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 -0400845 // These are valid characters
846 // Numbers
847 // Upper case letters
848 // Lower case letters
849 // Space
850 // and -_.{}[]
851 } else {
852 if (Display_Error)
Ethan Yonker4adc33e2016-01-22 16:07:11 -0600853 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 -0400854 return -3;
855 }
856 }
857
Ethan Yonker53796e72019-01-11 22:49:52 -0600858 if (Must_Be_Unique) {
859 // Check to make sure that a backup with this name doesn't already exist
860 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Loc);
861 strcpy(backup_loc, Backup_Loc.c_str());
862 sprintf(tw_image_dir,"%s/%s", backup_loc, Backup_Name.c_str());
863 if (TWFunc::Path_Exists(tw_image_dir)) {
864 if (Display_Error)
865 gui_err("backup_name_exists=A backup with that name already exists!");
bigbiffce8f83c2015-12-12 18:30:21 -0500866
Ethan Yonker53796e72019-01-11 22:49:52 -0600867 return -4;
868 }
869 // Backup is unique
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400870 }
Ethan Yonker53796e72019-01-11 22:49:52 -0600871 // No problems found
Dees_Troyc9ff7a32012-09-27 10:09:41 -0400872 return 0;
873}
874
bigbiffce8f83c2015-12-12 18:30:21 -0500875bool TWPartitionManager::Backup_Partition(PartitionSettings *part_settings) {
Dees_Troy43d8b002012-09-17 16:00:01 -0400876 time_t start, stop;
Matt Mower23d8aae2017-01-06 14:30:33 -0600877 int use_compression;
bigbiffce8f83c2015-12-12 18:30:21 -0500878 string backup_log = part_settings->Backup_Folder + "/recovery.log";
Dees_Troy43d8b002012-09-17 16:00:01 -0400879
bigbiffce8f83c2015-12-12 18:30:21 -0500880 if (part_settings->Part == NULL)
Dees_Troy43d8b002012-09-17 16:00:01 -0400881 return true;
882
Dees_Troy093b7642012-09-21 15:59:38 -0400883 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -0400884
Tom Hite5a926722014-09-15 01:31:03 +0000885 TWFunc::SetPerformanceMode(true);
Dees_Troy43d8b002012-09-17 16:00:01 -0400886 time(&start);
887
bigbiffce8f83c2015-12-12 18:30:21 -0500888 if (part_settings->Part->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500889 sync();
890 sync();
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400891 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500892 if (!part_settings->adbbackup && part_settings->generate_digest) {
893 if (!twrpDigestDriver::Make_Digest(Full_Filename))
894 goto backup_error;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400895 }
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400896
bigbiffce8f83c2015-12-12 18:30:21 -0500897 if (part_settings->Part->Has_SubPartition) {
Dees_Troy8170a922012-09-18 15:40:25 -0400898 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400899 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -0400900
901 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400902 if ((*subpart)->Can_Be_Backed_Up && (*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
Ethan Yonkerdcf2b672016-09-13 14:41:53 -0500903 part_settings->Part = *subpart;
bigbiffce8f83c2015-12-12 18:30:21 -0500904 if (!(*subpart)->Backup(part_settings, &tar_fork_pid)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500905 goto backup_error;
Tom Hite5a926722014-09-15 01:31:03 +0000906 }
Dees_Troy2727b992013-08-14 20:09:30 +0000907 sync();
908 sync();
bigbiff bigbifff5955b12019-05-18 17:28:58 -0400909 string Full_Filename = part_settings->Backup_Folder + "/" + part_settings->Part->Backup_FileName;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400910 if (!part_settings->adbbackup && part_settings->generate_digest) {
911 if (!twrpDigestDriver::Make_Digest(Full_Filename)) {
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500912 goto backup_error;
bigbiffce8f83c2015-12-12 18:30:21 -0500913 }
Tom Hite5a926722014-09-15 01:31:03 +0000914 }
Dees_Troy8170a922012-09-18 15:40:25 -0400915 }
916 }
917 }
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400918
Dees_Troy43d8b002012-09-17 16:00:01 -0400919 time(&stop);
that203bcc92015-05-09 17:27:33 +0200920 int backup_time = (int) difftime(stop, start);
Dees_Troy2673cec2013-04-02 20:22:16 +0000921 LOGINFO("Partition Backup time: %d\n", backup_time);
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -0400922
bigbiffce8f83c2015-12-12 18:30:21 -0500923 if (part_settings->Part->Backup_Method == BM_FILES) {
924 part_settings->file_time += backup_time;
Dees_Troy43d8b002012-09-17 16:00:01 -0400925 } else {
bigbiffce8f83c2015-12-12 18:30:21 -0500926 part_settings->img_time += backup_time;
927
Dees_Troy43d8b002012-09-17 16:00:01 -0400928 }
Tom Hite5a926722014-09-15 01:31:03 +0000929
Matt Mower02899552016-12-30 18:13:51 -0600930 TWFunc::SetPerformanceMode(false);
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500931 return true;
Dees_Troy43d8b002012-09-17 16:00:01 -0400932 }
Ethan Yonkercd00a8b2017-07-06 10:23:30 -0500933backup_error:
934 Clean_Backup_Folder(part_settings->Backup_Folder);
935 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
936 tw_set_default_metadata(backup_log.c_str());
937 TWFunc::SetPerformanceMode(false);
938 return false;
bigbiff7abc5fe2015-01-17 16:53:12 -0500939}
940
bigbiffbf1d6722015-02-14 16:25:59 -0500941void TWPartitionManager::Clean_Backup_Folder(string Backup_Folder) {
942 DIR *d = opendir(Backup_Folder.c_str());
943 struct dirent *p;
944 int r;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400945 vector<string> ext;
946
947 //extensions we should delete when cleaning
948 ext.push_back("win");
949 ext.push_back("md5");
950 ext.push_back("sha2");
951 ext.push_back("info");
bigbiffbf1d6722015-02-14 16:25:59 -0500952
Ethan Yonker74db1572015-10-28 12:44:49 -0500953 gui_msg("backup_clean=Backup Failed. Cleaning Backup Folder.");
bigbiffbf1d6722015-02-14 16:25:59 -0500954
955 if (d == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500956 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Backup_Folder)(strerror(errno)));
bigbiffbf1d6722015-02-14 16:25:59 -0500957 return;
958 }
959
Matt Mower2b18a532015-02-20 16:58:05 -0600960 while ((p = readdir(d))) {
bigbiffbf1d6722015-02-14 16:25:59 -0500961 if (!strcmp(p->d_name, ".") || !strcmp(p->d_name, ".."))
962 continue;
963
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500964 string path = Backup_Folder + "/" + p->d_name;
bigbiffbf1d6722015-02-14 16:25:59 -0500965
966 size_t dot = path.find_last_of(".") + 1;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400967 for (vector<string>::const_iterator i = ext.begin(); i != ext.end(); ++i) {
968 if (path.substr(dot) == *i) {
969 r = unlink(path.c_str());
970 if (r != 0)
971 LOGINFO("Unable to unlink '%s: %s'\n", path.c_str(), strerror(errno));
bigbiffbf1d6722015-02-14 16:25:59 -0500972 }
973 }
974 }
975 closedir(d);
976}
977
Ethan Yonker472f5062016-02-25 13:47:30 -0600978int TWPartitionManager::Check_Backup_Cancel() {
979 return stop_backup.get_value();
980}
981
bigbiff7abc5fe2015-01-17 16:53:12 -0500982int TWPartitionManager::Cancel_Backup() {
983 string Backup_Folder, Backup_Name, Full_Backup_Path;
984
985 stop_backup.set_value(1);
986
987 if (tar_fork_pid != 0) {
988 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
989 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, Backup_Folder);
Ethan Yonkere080c1f2016-09-19 13:50:25 -0500990 Full_Backup_Path = Backup_Folder + "/" + Backup_Name;
bigbiff7abc5fe2015-01-17 16:53:12 -0500991 LOGINFO("Killing pid: %d\n", tar_fork_pid);
992 kill(tar_fork_pid, SIGUSR2);
993 while (kill(tar_fork_pid, 0) == 0) {
994 usleep(1000);
995 }
996 LOGINFO("Backup_Run stopped and returning false, backup cancelled.\n");
997 LOGINFO("Removing directory %s\n", Full_Backup_Path.c_str());
998 TWFunc::removeDir(Full_Backup_Path, false);
999 tar_fork_pid = 0;
1000 }
1001
1002 return 0;
Dees_Troy43d8b002012-09-17 16:00:01 -04001003}
1004
bigbiffce8f83c2015-12-12 18:30:21 -05001005int TWPartitionManager::Run_Backup(bool adbbackup) {
1006 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001007 int partition_count = 0, disable_free_space_check = 0, skip_digest = 0;
bigbiffce8f83c2015-12-12 18:30:21 -05001008 string Backup_Name, Backup_List, backup_path;
Matt Mower23d8aae2017-01-06 14:30:33 -06001009 unsigned long long total_bytes = 0, free_space = 0;
Dees_Troy43d8b002012-09-17 16:00:01 -04001010 TWPartition* storage = NULL;
1011 struct tm *t;
Matt Mower23d8aae2017-01-06 14:30:33 -06001012 time_t seconds, total_start, total_stop;
Dees_Troya13d74f2013-03-24 08:54:55 -05001013 size_t start_pos = 0, end_pos = 0;
bigbiff7abc5fe2015-01-17 16:53:12 -05001014 stop_backup.set_value(0);
Dees_Troy43d8b002012-09-17 16:00:01 -04001015 seconds = time(0);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001016 t = localtime(&seconds);
Dees_Troy43d8b002012-09-17 16:00:01 -04001017
bigbiffce8f83c2015-12-12 18:30:21 -05001018 part_settings.img_bytes_remaining = 0;
1019 part_settings.file_bytes_remaining = 0;
1020 part_settings.img_time = 0;
1021 part_settings.file_time = 0;
1022 part_settings.img_bytes = 0;
1023 part_settings.file_bytes = 0;
1024 part_settings.PM_Method = PM_BACKUP;
1025
bigbiffce8f83c2015-12-12 18:30:21 -05001026 part_settings.adbbackup = adbbackup;
Dees_Troy43d8b002012-09-17 16:00:01 -04001027 time(&total_start);
1028
1029 Update_System_Details();
1030
1031 if (!Mount_Current_Storage(true))
1032 return false;
1033
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001034 DataManager::GetValue(TW_SKIP_DIGEST_GENERATE_VAR, skip_digest);
1035 if (skip_digest == 0)
1036 part_settings.generate_digest = true;
Dees_Troyc5865ab2012-09-24 15:08:04 -04001037 else
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001038 part_settings.generate_digest = false;
Dees_Troy4a2a1262012-09-18 09:33:47 -04001039
bigbiffce8f83c2015-12-12 18:30:21 -05001040 DataManager::GetValue(TW_BACKUPS_FOLDER_VAR, part_settings.Backup_Folder);
Ethan Yonkerdcf2b672016-09-13 14:41:53 -05001041 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
1042 if (Backup_Name == gui_lookup("curr_date", "(Current Date)")) {
1043 Backup_Name = TWFunc::Get_Current_Date();
1044 } else if (Backup_Name == gui_lookup("auto_generate", "(Auto Generate)") || Backup_Name == "0" || Backup_Name.empty()) {
Dees Troyb21cc642013-09-10 17:36:41 +00001045 TWFunc::Auto_Generate_Backup_Name();
Ethan Yonkerdcf2b672016-09-13 14:41:53 -05001046 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
Dees_Troy43d8b002012-09-17 16:00:01 -04001047 }
bigbiffce8f83c2015-12-12 18:30:21 -05001048
Ethan Yonkerdcf2b672016-09-13 14:41:53 -05001049 LOGINFO("Backup Name is: '%s'\n", Backup_Name.c_str());
Ethan Yonkere080c1f2016-09-19 13:50:25 -05001050 part_settings.Backup_Folder = part_settings.Backup_Folder + "/" + Backup_Name;
bigbiffce8f83c2015-12-12 18:30:21 -05001051
Ethan Yonkere080c1f2016-09-19 13:50:25 -05001052 LOGINFO("Backup_Folder is: '%s'\n", part_settings.Backup_Folder.c_str());
Dees_Troy43d8b002012-09-17 16:00:01 -04001053
Dees_Troy2673cec2013-04-02 20:22:16 +00001054 LOGINFO("Calculating backup details...\n");
Dees_Troya13d74f2013-03-24 08:54:55 -05001055 DataManager::GetValue("tw_backup_list", Backup_List);
bigbiff7ba75002020-04-11 20:47:09 -04001056 LOGINFO("Backup_List: %s\n", Backup_List.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -05001057 if (!Backup_List.empty()) {
1058 end_pos = Backup_List.find(";", start_pos);
1059 while (end_pos != string::npos && start_pos < Backup_List.size()) {
1060 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiff7ba75002020-04-11 20:47:09 -04001061 LOGINFO("backup_path: %s\n", backup_path.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -05001062 part_settings.Part = Find_Partition_By_Path(backup_path);
1063 if (part_settings.Part != NULL) {
Dees_Troya13d74f2013-03-24 08:54:55 -05001064 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -05001065 if (part_settings.Part->Backup_Method == BM_FILES)
1066 part_settings.file_bytes += part_settings.Part->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -05001067 else
bigbiffce8f83c2015-12-12 18:30:21 -05001068 part_settings.img_bytes += part_settings.Part->Backup_Size;
1069 if (part_settings.Part->Has_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -05001070 std::vector<TWPartition*>::iterator subpart;
1071
1072 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiffce8f83c2015-12-12 18:30:21 -05001073 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 -05001074 partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -05001075 if ((*subpart)->Backup_Method == BM_FILES)
1076 part_settings.file_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -05001077 else
bigbiffce8f83c2015-12-12 18:30:21 -05001078 part_settings.img_bytes += (*subpart)->Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -05001079 }
1080 }
Dees_Troy8170a922012-09-18 15:40:25 -04001081 }
Dees_Troya13d74f2013-03-24 08:54:55 -05001082 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001083 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 -04001084 }
Dees_Troya13d74f2013-03-24 08:54:55 -05001085 start_pos = end_pos + 1;
1086 end_pos = Backup_List.find(";", start_pos);
Dees_Troy43d8b002012-09-17 16:00:01 -04001087 }
1088 }
Dees_Troy43d8b002012-09-17 16:00:01 -04001089
1090 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001091 gui_msg("no_partition_selected=No partitions selected for backup.");
Dees_Troy43d8b002012-09-17 16:00:01 -04001092 return false;
1093 }
bigbiffce8f83c2015-12-12 18:30:21 -05001094 if (adbbackup) {
1095 if (twadbbu::Write_ADB_Stream_Header(partition_count) == false) {
1096 return false;
1097 }
1098 }
1099 total_bytes = part_settings.file_bytes + part_settings.img_bytes;
Ethan Yonker472f5062016-02-25 13:47:30 -06001100 ProgressTracking progress(total_bytes);
bigbiffce8f83c2015-12-12 18:30:21 -05001101 part_settings.progress = &progress;
1102
Ethan Yonker74db1572015-10-28 12:44:49 -05001103 gui_msg(Msg("total_partitions_backup= * Total number of partitions to back up: {1}")(partition_count));
1104 gui_msg(Msg("total_backup_size= * Total size of all data: {1}MB")(total_bytes / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -04001105 storage = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
1106 if (storage != NULL) {
1107 free_space = storage->Free;
Ethan Yonker74db1572015-10-28 12:44:49 -05001108 gui_msg(Msg("available_space= * Available space: {1}MB")(free_space / 1024 / 1024));
Dees_Troy43d8b002012-09-17 16:00:01 -04001109 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001110 gui_err("unable_locate_storage=Unable to locate storage device.");
Dees_Troy43d8b002012-09-17 16:00:01 -04001111 return false;
1112 }
bigbiffbf1d6722015-02-14 16:25:59 -05001113
Matt Mowerbfccfb82016-04-25 23:22:31 -05001114 DataManager::GetValue(TW_DISABLE_FREE_SPACE_VAR, disable_free_space_check);
bigbiffce8f83c2015-12-12 18:30:21 -05001115
1116 if (adbbackup)
1117 disable_free_space_check = true;
1118
bigbiffbf1d6722015-02-14 16:25:59 -05001119 if (!disable_free_space_check) {
1120 if (free_space - (32 * 1024 * 1024) < total_bytes) {
1121 // We require an extra 32MB just in case
Ethan Yonker74db1572015-10-28 12:44:49 -05001122 gui_err("no_space=Not enough free space on storage.");
bigbiffbf1d6722015-02-14 16:25:59 -05001123 return false;
1124 }
Dees_Troy43d8b002012-09-17 16:00:01 -04001125 }
bigbiffce8f83c2015-12-12 18:30:21 -05001126 part_settings.img_bytes_remaining = part_settings.img_bytes;
1127 part_settings.file_bytes_remaining = part_settings.file_bytes;
Dees_Troy43d8b002012-09-17 16:00:01 -04001128
Ethan Yonker74db1572015-10-28 12:44:49 -05001129 gui_msg("backup_started=[BACKUP STARTED]");
bigbiff bigbiffcdd97c72019-03-29 19:12:33 -04001130
1131 int is_decrypted = 0;
1132 int is_encrypted = 0;
1133
1134 DataManager::GetValue(TW_IS_DECRYPTED, is_decrypted);
1135 DataManager::GetValue(TW_IS_ENCRYPTED, is_encrypted);
1136 if (!adbbackup || (!is_encrypted || (is_encrypted && is_decrypted))) {
1137 gui_msg(Msg("backup_folder= * Backup Folder: {1}")(part_settings.Backup_Folder));
1138 if (!TWFunc::Recursive_Mkdir(part_settings.Backup_Folder)) {
1139 gui_err("fail_backup_folder=Failed to make backup folder.");
1140 return false;
1141 }
Dees_Troyd4b22b02013-01-18 17:17:58 +00001142 }
1143
Dees_Troy2673cec2013-04-02 20:22:16 +00001144 DataManager::SetProgress(0.0);
Dees_Troy093b7642012-09-21 15:59:38 -04001145
Dees_Troya13d74f2013-03-24 08:54:55 -05001146 start_pos = 0;
1147 end_pos = Backup_List.find(";", start_pos);
1148 while (end_pos != string::npos && start_pos < Backup_List.size()) {
bigbiff7abc5fe2015-01-17 16:53:12 -05001149 if (stop_backup.get_value() != 0)
1150 return -1;
Dees_Troya13d74f2013-03-24 08:54:55 -05001151 backup_path = Backup_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001152 part_settings.Part = Find_Partition_By_Path(backup_path);
1153 if (part_settings.Part != NULL) {
1154 if (!Backup_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001155 return false;
1156 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001157 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 -05001158 }
1159 start_pos = end_pos + 1;
1160 end_pos = Backup_List.find(";", start_pos);
1161 }
Dees_Troy43d8b002012-09-17 16:00:01 -04001162
1163 // Average BPS
bigbiffce8f83c2015-12-12 18:30:21 -05001164 if (part_settings.img_time == 0)
1165 part_settings.img_time = 1;
1166 if (part_settings.file_time == 0)
1167 part_settings.file_time = 1;
1168 int img_bps = (int)part_settings.img_bytes / (int)part_settings.img_time;
1169 unsigned long long file_bps = part_settings.file_bytes / (int)part_settings.file_time;
Dees_Troy43d8b002012-09-17 16:00:01 -04001170
bigbiffce8f83c2015-12-12 18:30:21 -05001171 if (part_settings.file_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -06001172 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 -05001173 if (part_settings.img_bytes != 0)
Ethan Yonker472f5062016-02-25 13:47:30 -06001174 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 -04001175
1176 time(&total_stop);
1177 int total_time = (int) difftime(total_stop, total_start);
bigbiffce8f83c2015-12-12 18:30:21 -05001178
1179 uint64_t actual_backup_size;
Ethan Yonker3fdcda42016-11-30 12:29:37 -06001180 if (!adbbackup) {
1181 TWExclude twe;
1182 actual_backup_size = twe.Get_Folder_Size(part_settings.Backup_Folder);
1183 } else
bigbiffce8f83c2015-12-12 18:30:21 -05001184 actual_backup_size = part_settings.file_bytes + part_settings.img_bytes;
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001185 actual_backup_size /= (1024LLU * 1024LLU);
Dees_Troy43d8b002012-09-17 16:00:01 -04001186
bigbiffce8f83c2015-12-12 18:30:21 -05001187 int prev_img_bps = 0, use_compression = 0;
1188 unsigned long long prev_file_bps = 0;
Matt Mower173cdb92016-12-31 02:49:19 -06001189 DataManager::GetValue(TW_BACKUP_AVG_IMG_RATE, prev_img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -04001190 img_bps += (prev_img_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001191 img_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -04001192
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001193 DataManager::GetValue(TW_USE_COMPRESSION_VAR, use_compression);
Dees_Troy093b7642012-09-21 15:59:38 -04001194 if (use_compression)
1195 DataManager::GetValue(TW_BACKUP_AVG_FILE_COMP_RATE, prev_file_bps);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001196 else
Dees_Troy093b7642012-09-21 15:59:38 -04001197 DataManager::GetValue(TW_BACKUP_AVG_FILE_RATE, prev_file_bps);
1198 file_bps += (prev_file_bps * 4);
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001199 file_bps /= 5;
Dees_Troy093b7642012-09-21 15:59:38 -04001200
Matt Mowerfb1c4ff2014-04-16 13:43:36 -05001201 DataManager::SetValue(TW_BACKUP_AVG_IMG_RATE, img_bps);
Dees_Troy093b7642012-09-21 15:59:38 -04001202 if (use_compression)
1203 DataManager::SetValue(TW_BACKUP_AVG_FILE_COMP_RATE, file_bps);
1204 else
1205 DataManager::SetValue(TW_BACKUP_AVG_FILE_RATE, file_bps);
1206
Ethan Yonker74db1572015-10-28 12:44:49 -05001207 gui_msg(Msg("total_backed_size=[{1} MB TOTAL BACKED UP]")(actual_backup_size));
Dees_Troy43d8b002012-09-17 16:00:01 -04001208 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001209 UnMount_Main_Partitions();
Ethan Yonker56db1c42015-12-20 22:49:00 -06001210 gui_msg(Msg(msg::kHighlight, "backup_completed=[BACKUP COMPLETED IN {1} SECONDS]")(total_time)); // the end
Ethan Yonkere080c1f2016-09-19 13:50:25 -05001211 string backup_log = part_settings.Backup_Folder + "/recovery.log";
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001212 TWFunc::copy_file("/tmp/recovery.log", backup_log, 0644);
Ethan Yonker4b94cfd2014-12-11 10:00:45 -06001213 tw_set_default_metadata(backup_log.c_str());
bigbiffce8f83c2015-12-12 18:30:21 -05001214
1215 if (part_settings.adbbackup) {
1216 if (twadbbu::Write_ADB_Stream_Trailer() == false) {
1217 return false;
1218 }
1219 }
1220 part_settings.adbbackup = false;
1221 DataManager::SetValue("tw_enable_adb_backup", 0);
1222
Dees_Troy3f5c4e82013-02-01 15:16:59 +00001223 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001224}
1225
bigbiffce8f83c2015-12-12 18:30:21 -05001226bool TWPartitionManager::Restore_Partition(PartitionSettings *part_settings) {
Dees_Troy4a2a1262012-09-18 09:33:47 -04001227 time_t Start, Stop;
bigbiffce8f83c2015-12-12 18:30:21 -05001228
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001229 if (part_settings->adbbackup) {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001230 std::string partName = part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win";
1231 LOGINFO("setting backup name: %s\n", partName.c_str());
bigbiff bigbiffb5ecaad2017-03-20 18:53:53 -04001232 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
1233 }
1234
Tom Hite5a926722014-09-15 01:31:03 +00001235 TWFunc::SetPerformanceMode(true);
bigbiffce8f83c2015-12-12 18:30:21 -05001236
Dees_Troy4a2a1262012-09-18 09:33:47 -04001237 time(&Start);
Ethan Yonker472f5062016-02-25 13:47:30 -06001238
bigbiffce8f83c2015-12-12 18:30:21 -05001239 if (!part_settings->Part->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001240 TWFunc::SetPerformanceMode(false);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001241 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001242 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001243 if (part_settings->Part->Has_SubPartition && !part_settings->adbbackup) {
Dees_Troy8170a922012-09-18 15:40:25 -04001244 std::vector<TWPartition*>::iterator subpart;
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001245 TWPartition *parentPart = part_settings->Part;
Dees_Troy8170a922012-09-18 15:40:25 -04001246
1247 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001248 part_settings->Part = *subpart;
1249 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
1250 part_settings->Part = (*subpart);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001251 part_settings->Part->Set_Backup_FileName(part_settings->Part->Backup_Name + "." + part_settings->Part->Current_File_System + ".win");
bigbiffce8f83c2015-12-12 18:30:21 -05001252 if (!(*subpart)->Restore(part_settings)) {
Tom Hite5a926722014-09-15 01:31:03 +00001253 TWFunc::SetPerformanceMode(false);
Dees_Troy8170a922012-09-18 15:40:25 -04001254 return false;
Tom Hite5a926722014-09-15 01:31:03 +00001255 }
Dees_Troy8170a922012-09-18 15:40:25 -04001256 }
1257 }
1258 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001259 time(&Stop);
Tom Hite5a926722014-09-15 01:31:03 +00001260 TWFunc::SetPerformanceMode(false);
bigbiffce8f83c2015-12-12 18:30:21 -05001261 gui_msg(Msg("restore_part_done=[{1} done ({2} seconds)]")(part_settings->Part->Backup_Display_Name)((int)difftime(Stop, Start)));
1262
Dees_Troy4a2a1262012-09-18 09:33:47 -04001263 return true;
1264}
1265
Ethan Yonker472f5062016-02-25 13:47:30 -06001266int TWPartitionManager::Run_Restore(const string& Restore_Name) {
bigbiffce8f83c2015-12-12 18:30:21 -05001267 PartitionSettings part_settings;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001268 int check_digest;
bigbiffce8f83c2015-12-12 18:30:21 -05001269
Dees_Troy4a2a1262012-09-18 09:33:47 -04001270 time_t rStart, rStop;
1271 time(&rStart);
Dees_Troya13d74f2013-03-24 08:54:55 -05001272 string Restore_List, restore_path;
1273 size_t start_pos = 0, end_pos;
bigbiffce8f83c2015-12-12 18:30:21 -05001274
Ethan Yonkerdcf2b672016-09-13 14:41:53 -05001275 part_settings.Backup_Folder = Restore_Name;
bigbiffce8f83c2015-12-12 18:30:21 -05001276 part_settings.Part = NULL;
1277 part_settings.partition_count = 0;
1278 part_settings.total_restore_size = 0;
1279 part_settings.adbbackup = false;
1280 part_settings.PM_Method = PM_RESTORE;
Dees_Troy43d8b002012-09-17 16:00:01 -04001281
Ethan Yonker74db1572015-10-28 12:44:49 -05001282 gui_msg("restore_started=[RESTORE STARTED]");
1283 gui_msg(Msg("restore_folder=Restore folder: '{1}'")(Restore_Name));
Dees_Troy43d8b002012-09-17 16:00:01 -04001284
Dees_Troy4a2a1262012-09-18 09:33:47 -04001285 if (!Mount_Current_Storage(true))
1286 return false;
1287
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001288 DataManager::GetValue(TW_SKIP_DIGEST_CHECK_VAR, check_digest);
1289 if (check_digest > 0) {
1290 // Check Digest files first before restoring to ensure that all of them match before starting a restore
1291 TWFunc::GUI_Operation_Text(TW_VERIFY_DIGEST_TEXT, gui_parse_text("{@verifying_digest}"));
1292 gui_msg("verifying_digest=Verifying Digest");
Dees_Troya13d74f2013-03-24 08:54:55 -05001293 } else {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001294 gui_msg("skip_digest=Skipping Digest check based on user setting.");
Dees_Troya13d74f2013-03-24 08:54:55 -05001295 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001296 gui_msg("calc_restore=Calculating restore details...");
Dees_Troya13d74f2013-03-24 08:54:55 -05001297 DataManager::GetValue("tw_restore_selected", Restore_List);
bigbiffce8f83c2015-12-12 18:30:21 -05001298
Dees_Troya13d74f2013-03-24 08:54:55 -05001299 if (!Restore_List.empty()) {
1300 end_pos = Restore_List.find(";", start_pos);
1301 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1302 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001303 part_settings.Part = Find_Partition_By_Path(restore_path);
1304 if (part_settings.Part != NULL) {
bigbiffce8f83c2015-12-12 18:30:21 -05001305 if (part_settings.Part->Mount_Read_Only) {
1306 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 -05001307 return false;
1308 }
bigbiffce8f83c2015-12-12 18:30:21 -05001309
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001310 string Full_Filename = part_settings.Backup_Folder + "/" + part_settings.Part->Backup_FileName;
1311
bigbiff4a60bee2021-01-23 14:08:03 -05001312 if (tw_get_default_metadata(Get_Android_Root_Path().c_str()) != 0) {
1313 gui_msg(Msg(msg::kWarning, "restore_system_context=Unable to get default context for {1} -- Android may not boot.")(Get_Android_Root_Path()));
1314 }
1315
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001316 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001317 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001318 part_settings.partition_count++;
1319 part_settings.total_restore_size += part_settings.Part->Get_Restore_Size(&part_settings);
1320 if (part_settings.Part->Has_SubPartition) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001321 TWPartition *parentPart = part_settings.Part;
Dees_Troya13d74f2013-03-24 08:54:55 -05001322 std::vector<TWPartition*>::iterator subpart;
1323
1324 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
bigbiff bigbiff8fd4b092016-08-30 20:48:53 -04001325 part_settings.Part = *subpart;
1326 if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == parentPart->Mount_Point) {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -04001327 if (check_digest > 0 && !twrpDigestDriver::Check_Digest(Full_Filename))
Dees_Troya13d74f2013-03-24 08:54:55 -05001328 return false;
bigbiffce8f83c2015-12-12 18:30:21 -05001329 part_settings.total_restore_size += (*subpart)->Get_Restore_Size(&part_settings);
Dees_Troya13d74f2013-03-24 08:54:55 -05001330 }
1331 }
1332 }
1333 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001334 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001335 }
1336 start_pos = end_pos + 1;
1337 end_pos = Restore_List.find(";", start_pos);
Dees_Troy4a2a1262012-09-18 09:33:47 -04001338 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001339 }
Dees_Troy4a2a1262012-09-18 09:33:47 -04001340
bigbiffce8f83c2015-12-12 18:30:21 -05001341 if (part_settings.partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001342 gui_err("no_part_restore=No partitions selected for restore.");
Dees_Troy4a2a1262012-09-18 09:33:47 -04001343 return false;
1344 }
1345
bigbiffce8f83c2015-12-12 18:30:21 -05001346 gui_msg(Msg("restore_part_count=Restoring {1} partitions...")(part_settings.partition_count));
1347 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 +00001348 DataManager::SetProgress(0.0);
bigbiffce8f83c2015-12-12 18:30:21 -05001349 ProgressTracking progress(part_settings.total_restore_size);
1350 part_settings.progress = &progress;
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001351
Dees_Troya13d74f2013-03-24 08:54:55 -05001352 start_pos = 0;
1353 if (!Restore_List.empty()) {
1354 end_pos = Restore_List.find(";", start_pos);
1355 while (end_pos != string::npos && start_pos < Restore_List.size()) {
1356 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiffce8f83c2015-12-12 18:30:21 -05001357
1358 part_settings.Part = Find_Partition_By_Path(restore_path);
1359 if (part_settings.Part != NULL) {
1360 part_settings.partition_count++;
bigbiffce8f83c2015-12-12 18:30:21 -05001361 if (!Restore_Partition(&part_settings))
Dees_Troya13d74f2013-03-24 08:54:55 -05001362 return false;
1363 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001364 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05001365 }
1366 start_pos = end_pos + 1;
1367 end_pos = Restore_List.find(";", start_pos);
1368 }
1369 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001370 TWFunc::GUI_Operation_Text(TW_UPDATE_SYSTEM_DETAILS_TEXT, gui_parse_text("{@updating_system_details}"));
bigbiff4a60bee2021-01-23 14:08:03 -05001371 tw_set_default_metadata(Get_Android_Root_Path().c_str());
Captain Throwback9d6feb52018-07-27 10:05:24 -04001372 UnMount_By_Path(Get_Android_Root_Path(), false);
Dees_Troy43d8b002012-09-17 16:00:01 -04001373 Update_System_Details();
Dees_Troyd0384ef2012-10-12 12:15:42 -04001374 UnMount_Main_Partitions();
Dees_Troy4a2a1262012-09-18 09:33:47 -04001375 time(&rStop);
Matt Mower3c366972015-12-25 19:28:31 -06001376 gui_msg(Msg(msg::kHighlight, "restore_completed=[RESTORE COMPLETED IN {1} SECONDS]")((int)difftime(rStop,rStart)));
bigbiff8f5fad32021-11-03 20:39:02 -04001377 TWPartition* Decrypt_Data = Find_Partition_By_Path("/data");
1378 if (Decrypt_Data && Decrypt_Data->Is_Encrypted)
1379 gui_msg(Msg(msg::kWarning, "reboot_after_restore=It is recommended to reboot Android once after first boot."));
Ethan Yonker1b7a31b2014-07-03 15:09:22 -05001380 DataManager::SetValue("tw_file_progress", "");
bigbiffce8f83c2015-12-12 18:30:21 -05001381
Dees_Troy63c8df72012-09-10 14:02:05 -04001382 return true;
Dees_Troy51a0e822012-09-05 15:24:24 -04001383}
1384
1385void TWPartitionManager::Set_Restore_Files(string Restore_Name) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001386 // Start with the default values
Dees_Troya13d74f2013-03-24 08:54:55 -05001387 string Restore_List;
Dees_Troy83bd4832013-05-04 12:39:56 +00001388 bool get_date = true, check_encryption = true;
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001389 bool adbbackup = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001390
1391 DataManager::SetValue("tw_restore_encrypted", 0);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001392 if (twadbbu::Check_ADB_Backup_File(Restore_Name)) {
1393 vector<string> adb_files;
1394 adb_files = twadbbu::Get_ADB_Backup_Files(Restore_Name);
1395 for (unsigned int i = 0; i < adb_files.size(); ++i) {
1396 string adb_restore_file = adb_files.at(i);
1397 std::size_t pos = adb_restore_file.find_first_of(".");
1398 std::string path = "/" + adb_restore_file.substr(0, pos);
1399 Restore_List = path + ";";
1400 TWPartition* Part = Find_Partition_By_Path(path);
1401 Part->Backup_FileName = TWFunc::Get_Filename(adb_restore_file);
1402 adbbackup = true;
1403 }
1404 DataManager::SetValue("tw_enable_adb_backup", 1);
Dees_Troy63c8df72012-09-10 14:02:05 -04001405 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001406 else {
1407 DIR* d;
1408 d = opendir(Restore_Name.c_str());
1409 if (d == NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001410 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001411 gui_msg(Msg(msg::kError, "error_opening_strerr=Error opening: '{1}' ({2})")(Restore_Name)(strerror(errno)));
1412 return;
Dees_Troy63c8df72012-09-10 14:02:05 -04001413 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001414
1415 struct dirent* de;
1416 while ((de = readdir(d)) != NULL)
Dees_Troy63c8df72012-09-10 14:02:05 -04001417 {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001418 // Strip off three components
1419 char str[256];
1420 char* label;
1421 char* fstype = NULL;
1422 char* extn = NULL;
1423 char* ptr;
Dees_Troy63c8df72012-09-10 14:02:05 -04001424
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001425 strcpy(str, de->d_name);
1426 if (strlen(str) <= 2)
1427 continue;
Dees_Troy83bd4832013-05-04 12:39:56 +00001428
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001429 if (get_date) {
1430 char file_path[255];
1431 struct stat st;
1432
1433 strcpy(file_path, Restore_Name.c_str());
1434 strcat(file_path, "/");
1435 strcat(file_path, str);
1436 stat(file_path, &st);
1437 string backup_date = ctime((const time_t*)(&st.st_mtime));
1438 DataManager::SetValue(TW_RESTORE_FILE_DATE, backup_date);
1439 get_date = false;
Dees_Troy83bd4832013-05-04 12:39:56 +00001440 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001441
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001442 label = str;
1443 ptr = label;
1444 while (*ptr && *ptr != '.') ptr++;
1445 if (*ptr == '.')
1446 {
1447 *ptr = 0x00;
1448 ptr++;
1449 fstype = ptr;
1450 }
1451 while (*ptr && *ptr != '.') ptr++;
1452 if (*ptr == '.')
1453 {
1454 *ptr = 0x00;
1455 ptr++;
1456 extn = ptr;
1457 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001458
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001459 if (fstype == NULL || extn == NULL || strcmp(fstype, "log") == 0) continue;
1460 int extnlength = strlen(extn);
1461 if (extnlength != 3 && extnlength != 6) continue;
1462 if (extnlength >= 3 && strncmp(extn, "win", 3) != 0) continue;
1463 //if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
Dees_Troy63c8df72012-09-10 14:02:05 -04001464
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001465 if (check_encryption) {
1466 string filename = Restore_Name + "/";
1467 filename += de->d_name;
1468 if (TWFunc::Get_File_Type(filename) == 2) {
1469 LOGINFO("'%s' is encrypted\n", filename.c_str());
1470 DataManager::SetValue("tw_restore_encrypted", 1);
1471 }
1472 }
1473 if (extnlength == 6 && strncmp(extn, "win000", 6) != 0) continue;
1474
1475 TWPartition* Part = Find_Partition_By_Path(label);
1476 if (Part == NULL)
1477 {
1478 gui_msg(Msg(msg::kError, "unable_locate_part_backup_name=Unable to locate partition by backup name: '{1}'")(label));
1479 continue;
1480 }
1481
1482 Part->Backup_FileName = de->d_name;
1483 if (strlen(extn) > 3) {
1484 Part->Backup_FileName.resize(Part->Backup_FileName.size() - strlen(extn) + 3);
1485 }
1486
dianlujitaob76a73a2020-04-30 20:33:20 +08001487 if (!Part->Is_SubPartition) {
1488 if (Part->Backup_Path == Get_Android_Root_Path())
1489 Restore_List += "/system;";
1490 else
1491 Restore_List += Part->Backup_Path + ";";
1492 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001493 }
1494 closedir(d);
Dees_Troy63c8df72012-09-10 14:02:05 -04001495 }
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001496
1497 if (adbbackup) {
1498 Restore_List = "ADB_Backup;";
1499 adbbackup = false;
1500 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001501
Dees_Troya13d74f2013-03-24 08:54:55 -05001502 // Set the final value
1503 DataManager::SetValue("tw_restore_list", Restore_List);
1504 DataManager::SetValue("tw_restore_selected", Restore_List);
Dees_Troy51a0e822012-09-05 15:24:24 -04001505 return;
1506}
1507
1508int TWPartitionManager::Wipe_By_Path(string Path) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001509 std::vector<TWPartition*>::iterator iter;
Mohd Farazfdd49bb2020-04-21 18:56:44 +05301510 std::vector < TWPartition * >::iterator iter1;
Dees_Troy63c8df72012-09-10 14:02:05 -04001511 int ret = false;
1512 bool found = false;
Dees_Troy38bd7602012-09-14 13:33:53 -04001513 string Local_Path = TWFunc::Get_Root_Path(Path);
Dees_Troy63c8df72012-09-10 14:02:05 -04001514
dianlujitaob76a73a2020-04-30 20:33:20 +08001515 if (Local_Path == "/system")
1516 Local_Path = Get_Android_Root_Path();
bigbiffaed1bdf2021-08-03 18:21:39 -04001517 if (Path == "/cache") {
1518 TWPartition* cache = Find_Partition_By_Path("/cache");
1519 if (cache == nullptr) {
1520 TWPartition* dat = Find_Partition_By_Path("/data");
1521 if (dat) {
1522 dat->Wipe_Data_Cache();
1523 found = true;
1524 }
1525 }
1526 }
Dees_Troy63c8df72012-09-10 14:02:05 -04001527 // Iterate through all partitions
1528 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy657c3092012-09-10 20:32:10 -04001529 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
Mohd Farazfdd49bb2020-04-21 18:56:44 +05301530 // iterate through all partitions since some legacy devices uses other partitions as vendor causes issues while wiping
1531 (*iter)->Find_Actual_Block_Device();
1532 for (iter1 = Partitions.begin (); iter1 != Partitions.end (); iter1++)
1533 {
1534 (*iter1)->Find_Actual_Block_Device();
1535 if ((*iter)->Actual_Block_Device == (*iter1)->Actual_Block_Device && (*iter)->Mount_Point != (*iter1)->Mount_Point)
1536 (*iter1)->UnMount(false);
1537 }
Dees_Troye58d5262012-09-21 12:27:57 -04001538 if (Path == "/and-sec")
1539 ret = (*iter)->Wipe_AndSec();
1540 else
1541 ret = (*iter)->Wipe();
Dees_Troy63c8df72012-09-10 14:02:05 -04001542 found = true;
1543 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1544 (*iter)->Wipe();
1545 }
1546 }
1547 if (found) {
1548 return ret;
1549 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001550 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 -04001551 return false;
Dees_Troy51a0e822012-09-05 15:24:24 -04001552}
1553
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001554int TWPartitionManager::Wipe_By_Path(string Path, string New_File_System) {
1555 std::vector<TWPartition*>::iterator iter;
1556 int ret = false;
1557 bool found = false;
1558 string Local_Path = TWFunc::Get_Root_Path(Path);
1559
1560 // Iterate through all partitions
1561 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1562 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1563 if (Path == "/and-sec")
1564 ret = (*iter)->Wipe_AndSec();
1565 else
1566 ret = (*iter)->Wipe(New_File_System);
1567 found = true;
1568 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1569 (*iter)->Wipe(New_File_System);
1570 }
1571 }
1572 if (found) {
1573 return ret;
1574 } else
Ethan Yonker74db1572015-10-28 12:44:49 -05001575 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 -05001576 return false;
1577}
1578
Dees_Troy51a0e822012-09-05 15:24:24 -04001579int TWPartitionManager::Factory_Reset(void) {
Dees_Troy63c8df72012-09-10 14:02:05 -04001580 std::vector<TWPartition*>::iterator iter;
1581 int ret = true;
1582
1583 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy38bd7602012-09-14 13:33:53 -04001584 if ((*iter)->Wipe_During_Factory_Reset && (*iter)->Is_Present) {
Ethan Yonker89583ef2015-08-26 09:01:59 -05001585#ifdef TW_OEM_BUILD
1586 if ((*iter)->Mount_Point == "/data") {
1587 if (!(*iter)->Wipe_Encryption())
1588 ret = false;
1589 } else {
1590#endif
1591 if (!(*iter)->Wipe())
1592 ret = false;
1593#ifdef TW_OEM_BUILD
1594 }
1595#endif
Dees_Troy094207a2012-09-26 12:00:39 -04001596 } else if ((*iter)->Has_Android_Secure) {
1597 if (!(*iter)->Wipe_AndSec())
1598 ret = false;
Dees_Troy63c8df72012-09-10 14:02:05 -04001599 }
1600 }
bigbiffad58e1b2020-07-06 20:24:34 -04001601 TWFunc::check_and_run_script("/system/bin/factoryreset.sh", "Factory Reset Script");
Dees_Troy63c8df72012-09-10 14:02:05 -04001602 return ret;
Dees_Troy51a0e822012-09-05 15:24:24 -04001603}
1604
Dees_Troy38bd7602012-09-14 13:33:53 -04001605int TWPartitionManager::Wipe_Dalvik_Cache(void) {
1606 struct stat st;
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001607 vector <string> dir;
Dees_Troy38bd7602012-09-14 13:33:53 -04001608
1609 if (!Mount_By_Path("/data", true))
1610 return false;
1611
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001612 dir.push_back("/data/dalvik-cache");
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001613
bigbiff25d25b92020-06-19 16:07:38 -04001614 std::string cacheDir = TWFunc::get_log_dir();
1615 if (cacheDir == CACHE_LOGS_DIR) {
1616 if (!PartitionManager.Mount_By_Path(CACHE_LOGS_DIR, false)) {
1617 LOGINFO("Unable to mount %s for wiping cache.\n", CACHE_LOGS_DIR);
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001618 }
1619 dir.push_back(cacheDir + "dalvik-cache");
1620 dir.push_back(cacheDir + "/dc");
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001621 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001622
Dees_Troy38bd7602012-09-14 13:33:53 -04001623 TWPartition* sdext = Find_Partition_By_Path("/sd-ext");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001624 if (sdext && sdext->Is_Present && sdext->Mount(false))
1625 {
1626 if (stat("/sd-ext/dalvik-cache", &st) == 0)
1627 {
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001628 dir.push_back("/sd-ext/dalvik-cache");
1629 }
1630 }
1631
bigbiff25d25b92020-06-19 16:07:38 -04001632 if (cacheDir == CACHE_LOGS_DIR) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001633 gui_msg("wiping_cache_dalvik=Wiping Dalvik Cache Directories...");
1634 } else {
1635 gui_msg("wiping_dalvik=Wiping Dalvik Directory...");
1636 }
Ethan Yonkerff2b7882016-12-10 09:04:41 -06001637 for (unsigned i = 0; i < dir.size(); ++i) {
1638 if (stat(dir.at(i).c_str(), &st) == 0) {
1639 TWFunc::removeDir(dir.at(i), false);
1640 gui_msg(Msg("cleaned=Cleaned: {1}...")(dir.at(i)));
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001641 }
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001642 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001643
bigbiff25d25b92020-06-19 16:07:38 -04001644 if (cacheDir == CACHE_LOGS_DIR) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05001645 gui_msg("cache_dalvik_done=-- Dalvik Cache Directories Wipe Complete!");
1646 } else {
1647 gui_msg("dalvik_done=-- Dalvik Directory Wipe Complete!");
1648 }
1649
Dees_Troy38bd7602012-09-14 13:33:53 -04001650 return true;
1651}
1652
1653int TWPartitionManager::Wipe_Rotate_Data(void) {
1654 if (!Mount_By_Path("/data", true))
1655 return false;
1656
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001657 unlink("/data/misc/akmd*");
1658 unlink("/data/misc/rild*");
Dees_Troy2673cec2013-04-02 20:22:16 +00001659 gui_print("Rotation data wiped.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001660 return true;
1661}
1662
1663int TWPartitionManager::Wipe_Battery_Stats(void) {
1664 struct stat st;
1665
1666 if (!Mount_By_Path("/data", true))
1667 return false;
1668
1669 if (0 != stat("/data/system/batterystats.bin", &st)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001670 gui_print("No Battery Stats Found. No Need To Wipe.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001671 } else {
1672 remove("/data/system/batterystats.bin");
Dees_Troy2673cec2013-04-02 20:22:16 +00001673 gui_print("Cleared battery stats.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001674 }
1675 return true;
1676}
1677
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001678int TWPartitionManager::Wipe_Android_Secure(void) {
1679 std::vector<TWPartition*>::iterator iter;
1680 int ret = false;
1681 bool found = false;
1682
1683 // Iterate through all partitions
1684 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1685 if ((*iter)->Has_Android_Secure) {
1686 ret = (*iter)->Wipe_AndSec();
1687 found = true;
1688 }
1689 }
1690 if (found) {
1691 return ret;
1692 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001693 gui_err("no_andsec=No android secure partitions found.");
Dees_Troy2ff5a8d2012-09-26 14:53:02 -04001694 }
1695 return false;
1696}
1697
Dees_Troy38bd7602012-09-14 13:33:53 -04001698int TWPartitionManager::Format_Data(void) {
1699 TWPartition* dat = Find_Partition_By_Path("/data");
bigbiffcfa875c2021-06-20 16:20:27 -04001700 TWPartition* metadata = Find_Partition_By_Path("/metadata");
1701 if (metadata != NULL)
1702 metadata->UnMount(false);
Dees_Troy38bd7602012-09-14 13:33:53 -04001703
1704 if (dat != NULL) {
bigbiffab76bd72021-10-11 10:17:48 -04001705 if (android::base::GetBoolProperty("ro.virtual_ab.enabled", false)) {
bigbiffcfa875c2021-06-20 16:20:27 -04001706#ifndef TW_EXCLUDE_APEX
1707 twrpApex apex;
1708 apex.Unmount();
1709#endif
1710 if (metadata != NULL)
1711 metadata->Mount(true);
bigbiffd21252f2021-09-18 15:56:32 -04001712 if (!Check_Pending_Merges())
bigbiffcfa875c2021-06-20 16:20:27 -04001713 return false;
1714 }
Dees_Troy38bd7602012-09-14 13:33:53 -04001715 return dat->Wipe_Encryption();
1716 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001717 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001718 return false;
1719 }
1720 return false;
1721}
1722
1723int TWPartitionManager::Wipe_Media_From_Data(void) {
1724 TWPartition* dat = Find_Partition_By_Path("/data");
1725
1726 if (dat != NULL) {
1727 if (!dat->Has_Data_Media) {
Dees_Troy2673cec2013-04-02 20:22:16 +00001728 LOGERR("This device does not have /data/media\n");
Dees_Troy38bd7602012-09-14 13:33:53 -04001729 return false;
1730 }
1731 if (!dat->Mount(true))
1732 return false;
1733
Ethan Yonker74db1572015-10-28 12:44:49 -05001734 gui_msg("wiping_datamedia=Wiping internal storage -- /data/media...");
Ethan Yonker726a0202014-12-16 20:01:38 -06001735 Remove_MTP_Storage(dat->MTP_Storage_ID);
bigbiff bigbiff9c754052013-01-09 09:09:08 -05001736 TWFunc::removeDir("/data/media", false);
xiaolu9416f4f2015-06-04 08:22:23 +08001737 dat->Recreate_Media_Folder();
Ethan Yonker726a0202014-12-16 20:01:38 -06001738 Add_MTP_Storage(dat->MTP_Storage_ID);
Dees_Troy38bd7602012-09-14 13:33:53 -04001739 return true;
1740 } else {
Matt Mower3c366972015-12-25 19:28:31 -06001741 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/data"));
Dees_Troy38bd7602012-09-14 13:33:53 -04001742 return false;
1743 }
1744 return false;
1745}
1746
Ethan Yonker87c7bac2014-05-25 21:41:08 -05001747int TWPartitionManager::Repair_By_Path(string Path, bool Display_Error) {
1748 std::vector<TWPartition*>::iterator iter;
1749 int ret = false;
1750 bool found = false;
1751 string Local_Path = TWFunc::Get_Root_Path(Path);
1752
1753 if (Local_Path == "/tmp" || Local_Path == "/")
1754 return true;
1755
1756 // Iterate through all partitions
1757 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1758 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1759 ret = (*iter)->Repair();
1760 found = true;
1761 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1762 (*iter)->Repair();
1763 }
1764 }
1765 if (found) {
1766 return ret;
1767 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001768 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 -05001769 } else {
1770 LOGINFO("Repair: Unable to find partition for path '%s'\n", Local_Path.c_str());
1771 }
1772 return false;
1773}
1774
Ethan Yonkera2719152015-05-28 09:44:41 -05001775int TWPartitionManager::Resize_By_Path(string Path, bool Display_Error) {
1776 std::vector<TWPartition*>::iterator iter;
1777 int ret = false;
1778 bool found = false;
1779 string Local_Path = TWFunc::Get_Root_Path(Path);
1780
1781 if (Local_Path == "/tmp" || Local_Path == "/")
1782 return true;
1783
1784 // Iterate through all partitions
1785 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
1786 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
1787 ret = (*iter)->Resize();
1788 found = true;
1789 } else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
1790 (*iter)->Resize();
1791 }
1792 }
1793 if (found) {
1794 return ret;
1795 } else if (Display_Error) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001796 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 -05001797 } else {
1798 LOGINFO("Resize: Unable to find partition for path '%s'\n", Local_Path.c_str());
1799 }
1800 return false;
1801}
1802
Dees_Troy51a0e822012-09-05 15:24:24 -04001803void TWPartitionManager::Update_System_Details(void) {
Dees_Troy5bf43922012-09-07 16:07:55 -04001804 std::vector<TWPartition*>::iterator iter;
Dees_Troy51127312012-09-08 13:08:49 -04001805 int data_size = 0;
Dees_Troy5bf43922012-09-07 16:07:55 -04001806
Ethan Yonker74db1572015-10-28 12:44:49 -05001807 gui_msg("update_part_details=Updating partition details...");
Dees_Troy5bf43922012-09-07 16:07:55 -04001808 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker1b190162016-12-05 15:25:19 -06001809 (*iter)->Update_Size(true);
Dees_Troy51127312012-09-08 13:08:49 -04001810 if ((*iter)->Can_Be_Mounted) {
Captain Throwback9d6feb52018-07-27 10:05:24 -04001811 if ((*iter)->Mount_Point == Get_Android_Root_Path()) {
Dees_Troy51127312012-09-08 13:08:49 -04001812 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1813 DataManager::SetValue(TW_BACKUP_SYSTEM_SIZE, backup_display_size);
Ethan Yonker76bbd3a2019-05-10 10:50:04 -05001814 TWFunc::Is_TWRP_App_In_System();
Dees_Troy51127312012-09-08 13:08:49 -04001815 } else if ((*iter)->Mount_Point == "/data" || (*iter)->Mount_Point == "/datadata") {
1816 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
1817 } else if ((*iter)->Mount_Point == "/cache") {
1818 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1819 DataManager::SetValue(TW_BACKUP_CACHE_SIZE, backup_display_size);
1820 } else if ((*iter)->Mount_Point == "/sd-ext") {
1821 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1822 DataManager::SetValue(TW_BACKUP_SDEXT_SIZE, backup_display_size);
1823 if ((*iter)->Backup_Size == 0) {
1824 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 0);
1825 DataManager::SetValue(TW_BACKUP_SDEXT_VAR, 0);
1826 } else
1827 DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 1);
Dees_Troye58d5262012-09-21 12:27:57 -04001828 } else if ((*iter)->Has_Android_Secure) {
Dees_Troy8170a922012-09-18 15:40:25 -04001829 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troye58d5262012-09-21 12:27:57 -04001830 DataManager::SetValue(TW_BACKUP_ANDSEC_SIZE, backup_display_size);
Dees_Troy8170a922012-09-18 15:40:25 -04001831 if ((*iter)->Backup_Size == 0) {
1832 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 0);
1833 DataManager::SetValue(TW_BACKUP_ANDSEC_VAR, 0);
1834 } else
1835 DataManager::SetValue(TW_HAS_ANDROID_SECURE, 1);
Dees_Troy2c50e182012-09-26 20:05:28 -04001836 } else if ((*iter)->Mount_Point == "/boot") {
1837 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1838 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1839 if ((*iter)->Backup_Size == 0) {
1840 DataManager::SetValue("tw_has_boot_partition", 0);
1841 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1842 } else
1843 DataManager::SetValue("tw_has_boot_partition", 1);
Dees_Troy51127312012-09-08 13:08:49 -04001844 }
Dees_Troyaa9cc402012-10-13 12:14:05 -04001845 } else {
1846 // Handle unmountable partitions in case we reset defaults
1847 if ((*iter)->Mount_Point == "/boot") {
1848 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1849 DataManager::SetValue(TW_BACKUP_BOOT_SIZE, backup_display_size);
1850 if ((*iter)->Backup_Size == 0) {
1851 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 0);
1852 DataManager::SetValue(TW_BACKUP_BOOT_VAR, 0);
1853 } else
1854 DataManager::SetValue(TW_HAS_BOOT_PARTITION, 1);
1855 } else if ((*iter)->Mount_Point == "/recovery") {
1856 int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
1857 DataManager::SetValue(TW_BACKUP_RECOVERY_SIZE, backup_display_size);
1858 if ((*iter)->Backup_Size == 0) {
1859 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 0);
1860 DataManager::SetValue(TW_BACKUP_RECOVERY_VAR, 0);
1861 } else
1862 DataManager::SetValue(TW_HAS_RECOVERY_PARTITION, 1);
Gary Peck82599a82012-11-21 16:23:12 -08001863 } else if ((*iter)->Mount_Point == "/data") {
1864 data_size += (int)((*iter)->Backup_Size / 1048576LLU);
Dees_Troyaa9cc402012-10-13 12:14:05 -04001865 }
Dees_Troy51127312012-09-08 13:08:49 -04001866 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001867 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001868 gui_msg("update_part_details_done=...done");
Dees_Troy51127312012-09-08 13:08:49 -04001869 DataManager::SetValue(TW_BACKUP_DATA_SIZE, data_size);
1870 string current_storage_path = DataManager::GetCurrentStoragePath();
1871 TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
Dees_Troy8170a922012-09-18 15:40:25 -04001872 if (FreeStorage != NULL) {
1873 // Attempt to mount storage
bigbiff25d25b92020-06-19 16:07:38 -04001874 if (!FreeStorage->Mount(false)) {
1875 gui_msg(Msg(msg::kWarning, "unable_to_mount_storage=Unable to mount storage"));
1876 DataManager::SetValue(TW_STORAGE_FREE_SIZE, 0);
1877 } else {
1878 DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
Dees_Troy8170a922012-09-18 15:40:25 -04001879 }
1880 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00001881 LOGINFO("Unable to find storage partition '%s'.\n", current_storage_path.c_str());
Dees_Troy8170a922012-09-18 15:40:25 -04001882 }
Dees_Troy5bf43922012-09-07 16:07:55 -04001883 if (!Write_Fstab())
Dees_Troy2673cec2013-04-02 20:22:16 +00001884 LOGERR("Error creating fstab\n");
Dees_Troy51a0e822012-09-05 15:24:24 -04001885 return;
1886}
1887
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001888void TWPartitionManager::Post_Decrypt(const string& Block_Device) {
1889 TWPartition* dat = Find_Partition_By_Path("/data");
bigbiffc2cb3852020-09-11 14:03:31 -04001890
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001891 if (dat != NULL) {
bigbiffaed1bdf2021-08-03 18:21:39 -04001892 // reparse for /cache/recovery/command
1893 static constexpr const char* COMMAND_FILE = "/data/cache/command";
1894 if (TWFunc::Path_Exists(COMMAND_FILE)) {
1895 startupArgs startup;
1896 std::string content;
1897 TWFunc::read_file(COMMAND_FILE, content);
1898 std::vector<std::string> args = {content};
1899 startup.processRecoveryArgs(args, 0);
1900 }
1901
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001902 DataManager::SetValue(TW_IS_DECRYPTED, 1);
1903 dat->Is_Decrypted = true;
1904 if (!Block_Device.empty()) {
1905 dat->Decrypted_Block_Device = Block_Device;
1906 gui_msg(Msg("decrypt_success_dev=Data successfully decrypted, new block device: '{1}'")(Block_Device));
1907 } else {
1908 gui_msg("decrypt_success_nodev=Data successfully decrypted");
1909 }
Noah Jacobson81d638d2019-04-28 00:10:07 -04001910 property_set("twrp.decrypt.done", "true");
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001911 dat->Setup_File_System(false);
Noah Jacobson81d638d2019-04-28 00:10:07 -04001912 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 -06001913
Noah Jacobson81d638d2019-04-28 00:10:07 -04001914 sleep(1); // Sleep for a bit so that the device will be ready
me-cafebabe09649612022-06-10 21:53:31 +08001915
1916 // Mount only /data
1917 dat->Symlink_Path = ""; // Not to let it to bind mount /data/media again
1918 if (!dat->Mount(false)) {
1919 LOGERR("Unable to mount /data after decryption");
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001920 }
me-cafebabe09649612022-06-10 21:53:31 +08001921
1922 if (dat->Has_Data_Media && TWFunc::Path_Exists("/data/media/0")) {
1923 dat->Storage_Path = "/data/media/0";
1924 } else {
1925 dat->Storage_Path = "/data/media";
1926 }
1927 dat->Symlink_Path = dat->Storage_Path;
1928 DataManager::SetValue("tw_storage_path", dat->Symlink_Path);
Mohd Farazf9231102022-06-18 22:51:55 +02001929 DataManager::SetValue("tw_settings_path", TW_STORAGE_PATH);
me-cafebabe09649612022-06-10 21:53:31 +08001930 LOGINFO("New storage path after decryption: %s\n", dat->Storage_Path.c_str());
1931
epicXa721f952021-01-04 13:01:31 +05301932 DataManager::LoadTWRPFolderInfo();
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001933 Update_System_Details();
nkk71ffb02bd2017-06-04 23:12:16 +03001934 Output_Partition(dat);
bigbiff4fa02b52021-12-30 17:01:44 -05001935 if (!android::base::StartsWith(dat->Actual_Block_Device, "/dev/block/mmcblk")) {
1936 if (!dat->Bind_Mount(false))
1937 LOGERR("Unable to bind mount /sdcard to %s\n", dat->Storage_Path.c_str());
1938 }
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06001939 } else
1940 LOGERR("Unable to locate data partition.\n");
1941}
1942
Noah Jacobson81d638d2019-04-28 00:10:07 -04001943void TWPartitionManager::Parse_Users() {
1944#ifdef TW_INCLUDE_FBE
1945 char user_check_result[PROPERTY_VALUE_MAX];
1946 for (int userId = 0; userId <= 9999; userId++) {
1947 string prop = "twrp.user." + to_string(userId) + ".decrypt";
1948 property_get(prop.c_str(), user_check_result, "-1");
1949 if (strcmp(user_check_result, "-1") != 0) {
1950 if (userId < 0 || userId > 9999) {
1951 LOGINFO("Incorrect user id %d\n", userId);
1952 continue;
1953 }
1954 struct users_struct user;
1955 user.userId = to_string(userId);
1956
1957 // Attempt to get name of user. Fallback to user ID if this fails.
Darth98f775972022-08-05 09:39:03 +01001958 std::string path = "/data/system/users/" + to_string(userId) + ".xml";
1959 if (!TWFunc::Check_Xml_Format(path)) {
1960 string oldpath = path;
1961 if (TWFunc::abx_to_xml(oldpath, path)) {
1962 LOGINFO("Android 12+: '%s' has been converted into plain text xml (for user %s).\n", oldpath.c_str(), user.userId.c_str());
1963 }
1964 }
1965 char* userFile = PageManager::LoadFileToBuffer(path, NULL);
me-cafebabec46981d2022-07-29 01:54:43 +08001966 if (userFile == NULL) {
1967 user.userName = to_string(userId);
bigbiffa957f072021-03-07 18:20:29 -05001968 }
Noah Jacobson81d638d2019-04-28 00:10:07 -04001969 else {
me-cafebabec46981d2022-07-29 01:54:43 +08001970 xml_document<> *userXml = new xml_document<>();
1971 userXml->parse<0>(userFile);
1972 xml_node<>* userNode = userXml->first_node("user");
1973 if (userNode == nullptr) {
Noah Jacobson81d638d2019-04-28 00:10:07 -04001974 user.userName = to_string(userId);
me-cafebabec46981d2022-07-29 01:54:43 +08001975 } else {
1976 xml_node<>* nameNode = userNode->first_node("name");
1977 if (nameNode == nullptr)
Noah Jacobson81d638d2019-04-28 00:10:07 -04001978 user.userName = to_string(userId);
me-cafebabec46981d2022-07-29 01:54:43 +08001979 else {
1980 string userName = nameNode->value();
1981 user.userName = userName + " (" + to_string(userId) + ")";
Noah Jacobson81d638d2019-04-28 00:10:07 -04001982 }
1983 }
1984 }
1985
1986 string filename;
bigbiffdf53b6c2022-01-17 19:31:52 -05001987 user.type = android::keystore::Get_Password_Type(userId, filename);
Noah Jacobson81d638d2019-04-28 00:10:07 -04001988
1989 user.isDecrypted = false;
1990 if (strcmp(user_check_result, "1") == 0)
1991 user.isDecrypted = true;
1992 Users_List.push_back(user);
1993 }
1994 }
1995 Check_Users_Decryption_Status();
1996#endif
1997}
1998
1999std::vector<users_struct>* TWPartitionManager::Get_Users_List() {
2000 return &Users_List;
2001}
2002
2003void TWPartitionManager::Mark_User_Decrypted(int userID) {
2004#ifdef TW_INCLUDE_FBE
2005 std::vector<users_struct>::iterator iter;
2006 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
2007 if (atoi((*iter).userId.c_str()) == userID) {
2008 (*iter).isDecrypted = true;
2009 string user_prop_decrypted = "twrp.user." + to_string(userID) + ".decrypt";
2010 property_set(user_prop_decrypted.c_str(), "1");
2011 break;
2012 }
2013 }
2014 Check_Users_Decryption_Status();
2015#endif
2016}
2017
2018void TWPartitionManager::Check_Users_Decryption_Status() {
2019#ifdef TW_INCLUDE_FBE
2020 int all_is_decrypted = 1;
2021 std::vector<users_struct>::iterator iter;
2022 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
2023 if (!(*iter).isDecrypted) {
2024 LOGINFO("User %s is not decrypted.\n", (*iter).userId.c_str());
2025 all_is_decrypted = 0;
2026 break;
2027 }
2028 }
2029 if (all_is_decrypted == 1) {
2030 LOGINFO("All found users are decrypted.\n");
2031 DataManager::SetValue("tw_all_users_decrypted", "1");
2032 property_set("twrp.all.users.decrypted", "true");
2033 } else
2034 DataManager::SetValue("tw_all_users_decrypted", "0");
2035#endif
2036}
2037
2038int TWPartitionManager::Decrypt_Device(string Password, int user_id) {
Dees_Troy5bf43922012-09-07 16:07:55 -04002039#ifdef TW_INCLUDE_CRYPTO
Captain Throwback37c3aef2021-10-06 11:38:18 -04002040 char crypto_blkdev[PROPERTY_VALUE_MAX];
Ethan Yonker253368a2014-11-25 15:00:52 -06002041 std::vector<TWPartition*>::iterator iter;
Dees_Troy5bf43922012-09-07 16:07:55 -04002042
Ethan Yonker253368a2014-11-25 15:00:52 -06002043 // Mount any partitions that need to be mounted for decrypt
2044 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2045 if ((*iter)->Mount_To_Decrypt) {
2046 (*iter)->Mount(true);
2047 }
a39552696ff55ce2013-01-08 16:14:56 +00002048 }
oshmouna82a7542018-04-10 17:45:55 +02002049 property_set("twrp.mount_to_decrypt", "1");
a39552696ff55ce2013-01-08 16:14:56 +00002050
Captain Throwback37c3aef2021-10-06 11:38:18 -04002051 Set_Crypto_State();
2052 Set_Crypto_Type("block");
Ethan Yonkera1de1492015-11-04 11:58:27 -06002053
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06002054 if (DataManager::GetIntValue(TW_IS_FBE)) {
2055#ifdef TW_INCLUDE_FBE
2056 if (!Mount_By_Path("/data", true)) // /data has to be mounted for FBE
2057 return -1;
Noah Jacobson81d638d2019-04-28 00:10:07 -04002058
2059 bool user_need_decrypt = false;
2060 std::vector<users_struct>::iterator iter;
2061 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
2062 if (atoi((*iter).userId.c_str()) == user_id && !(*iter).isDecrypted) {
2063 user_need_decrypt = true;
2064 }
2065 }
2066 if (!user_need_decrypt) {
2067 LOGINFO("User %d does not require decryption\n", user_id);
2068 return 0;
2069 }
2070
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06002071 int retry_count = 10;
2072 while (!TWFunc::Path_Exists("/data/system/users/gatekeeper.password.key") && --retry_count)
Noah Jacobson81d638d2019-04-28 00:10:07 -04002073 usleep(2000); // A small sleep is needed after mounting /data to ensure reliable decrypt...maybe because of DE?
2074 gui_msg(Msg("decrypting_user_fbe=Attempting to decrypt FBE for user {1}...")(user_id));
bigbiffdf53b6c2022-01-17 19:31:52 -05002075 if (android::keystore::Decrypt_User(user_id, Password)) {
Noah Jacobson81d638d2019-04-28 00:10:07 -04002076 gui_msg(Msg("decrypt_user_success_fbe=User {1} Decrypted Successfully")(user_id));
2077 Mark_User_Decrypted(user_id);
2078 if (user_id == 0) {
2079 // When decrypting user 0 also try all other users
2080 std::vector<users_struct>::iterator iter;
2081 for (iter = Users_List.begin(); iter != Users_List.end(); iter++) {
2082 if ((*iter).userId == "0" || (*iter).isDecrypted)
2083 continue;
2084
2085 int tmp_user_id = atoi((*iter).userId.c_str());
2086 gui_msg(Msg("decrypting_user_fbe=Attempting to decrypt FBE for user {1}...")(tmp_user_id));
bigbiffdf53b6c2022-01-17 19:31:52 -05002087 if (android::keystore::Decrypt_User(tmp_user_id, Password) ||
2088 (Password != "!" && android::keystore::Decrypt_User(tmp_user_id, "!"))) { // "!" means default password
Noah Jacobson81d638d2019-04-28 00:10:07 -04002089 gui_msg(Msg("decrypt_user_success_fbe=User {1} Decrypted Successfully")(tmp_user_id));
2090 Mark_User_Decrypted(tmp_user_id);
2091 } else {
2092 gui_msg(Msg("decrypt_user_fail_fbe=Failed to decrypt user {1}")(tmp_user_id));
2093 }
2094 }
2095 Post_Decrypt("");
2096 }
2097
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06002098 return 0;
Noah Jacobson81d638d2019-04-28 00:10:07 -04002099 } else {
2100 gui_msg(Msg(msg::kError, "decrypt_user_fail_fbe=Failed to decrypt user {1}")(user_id));
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06002101 }
2102#else
2103 LOGERR("FBE support is not present\n");
2104#endif
2105 return -1;
2106 }
2107
Noah Jacobson81d638d2019-04-28 00:10:07 -04002108 char isdecrypteddata[PROPERTY_VALUE_MAX];
2109 property_get("twrp.decrypt.done", isdecrypteddata, "");
2110 if (strcmp(isdecrypteddata, "true") == 0) {
2111 LOGINFO("Data has no decryption required\n");
2112 return 0;
2113 }
2114
Ethan Yonkerddb63e22017-01-19 14:01:57 -06002115 int pwret = -1;
2116 pid_t pid = fork();
2117 if (pid < 0) {
2118 LOGERR("fork failed\n");
2119 return -1;
2120 } else if (pid == 0) {
2121 // Child process
2122 char cPassword[255];
2123 strcpy(cPassword, Password.c_str());
2124 int ret = cryptfs_check_passwd(cPassword);
2125 exit(ret);
2126 } else {
2127 // Parent
2128 int status;
2129 if (TWFunc::Wait_For_Child_Timeout(pid, &status, "Decrypt", 30))
2130 pwret = -1;
2131 else
2132 pwret = WEXITSTATUS(status) ? -1 : 0;
2133 }
a39552696ff55ce2013-01-08 16:14:56 +00002134
Ethan Yonker253368a2014-11-25 15:00:52 -06002135 // Unmount any partitions that were needed for decrypt
2136 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2137 if ((*iter)->Mount_To_Decrypt) {
2138 (*iter)->UnMount(false);
2139 }
2140 }
oshmouna82a7542018-04-10 17:45:55 +02002141 property_set("twrp.mount_to_decrypt", "0");
Ethan Yonker253368a2014-11-25 15:00:52 -06002142
a39552696ff55ce2013-01-08 16:14:56 +00002143 if (pwret != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002144 gui_err("fail_decrypt=Failed to decrypt data.");
Dees_Troy5bf43922012-09-07 16:07:55 -04002145 return -1;
2146 }
a39552696ff55ce2013-01-08 16:14:56 +00002147
Dees_Troy5bf43922012-09-07 16:07:55 -04002148 property_get("ro.crypto.fs_crypto_blkdev", crypto_blkdev, "error");
2149 if (strcmp(crypto_blkdev, "error") == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +00002150 LOGERR("Error retrieving decrypted data block device.\n");
Dees_Troy5bf43922012-09-07 16:07:55 -04002151 } else {
Ethan Yonkerbd7492d2016-12-07 13:55:01 -06002152 Post_Decrypt(crypto_blkdev);
Dees_Troy5bf43922012-09-07 16:07:55 -04002153 }
2154 return 0;
2155#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002156 gui_err("no_crypto_support=No crypto support was compiled into this build.");
Dees_Troy5bf43922012-09-07 16:07:55 -04002157 return -1;
2158#endif
Dees_Troy51a0e822012-09-05 15:24:24 -04002159 return 1;
Dees_Troy51127312012-09-08 13:08:49 -04002160}
2161
Ethan Yonkerb5fab762016-01-28 14:03:33 -06002162int TWPartitionManager::Fix_Contexts(void) {
Ethan Yonkerb5fab762016-01-28 14:03:33 -06002163 std::vector<TWPartition*>::iterator iter;
2164 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2165 if ((*iter)->Has_Data_Media) {
2166 if ((*iter)->Mount(true)) {
2167 if (fixContexts::fixDataMediaContexts((*iter)->Mount_Point) != 0)
2168 return -1;
2169 }
2170 }
2171 }
Dees_Troy1a650e62012-10-19 20:54:32 -04002172 UnMount_Main_Partitions();
Ethan Yonker74db1572015-10-28 12:44:49 -05002173 gui_msg("done=Done.");
Ethan Yonkerb5fab762016-01-28 14:03:33 -06002174 return 0;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04002175}
2176
Ethan Yonker66a19492015-12-10 10:19:45 -06002177TWPartition* TWPartitionManager::Find_Next_Storage(string Path, bool Exclude_Data_Media) {
Ethan Yonker47360be2014-04-01 10:34:34 -05002178 std::vector<TWPartition*>::iterator iter = Partitions.begin();
2179
2180 if (!Path.empty()) {
2181 string Search_Path = TWFunc::Get_Root_Path(Path);
2182 for (; iter != Partitions.end(); iter++) {
Matt Mower9a561dd2016-02-22 21:25:51 -06002183 if ((*iter)->Mount_Point == Search_Path) {
Ethan Yonker47360be2014-04-01 10:34:34 -05002184 iter++;
2185 break;
2186 }
2187 }
2188 }
2189
2190 for (; iter != Partitions.end(); iter++) {
Ethan Yonker66a19492015-12-10 10:19:45 -06002191 if (Exclude_Data_Media && (*iter)->Has_Data_Media) {
2192 // do nothing, do not return this type of partition
2193 } else if ((*iter)->Is_Storage && (*iter)->Is_Present) {
Ethan Yonker47360be2014-04-01 10:34:34 -05002194 return (*iter);
2195 }
2196 }
2197
2198 return NULL;
2199}
2200
Dees_Troyd21618c2012-10-14 18:48:49 -04002201int TWPartitionManager::Open_Lun_File(string Partition_Path, string Lun_File) {
Dees_Troyd21618c2012-10-14 18:48:49 -04002202 TWPartition* Part = Find_Partition_By_Path(Partition_Path);
2203
2204 if (Part == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002205 LOGINFO("Unable to locate '%s' for USB storage mode.", Partition_Path.c_str());
2206 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 -04002207 return false;
2208 }
Ethan Yonker47360be2014-04-01 10:34:34 -05002209 LOGINFO("USB mount '%s', '%s' > '%s'\n", Partition_Path.c_str(), Part->Actual_Block_Device.c_str(), Lun_File.c_str());
2210 if (!Part->UnMount(true) || !Part->Is_Present)
Dees_Troyd21618c2012-10-14 18:48:49 -04002211 return false;
2212
bigbiff22851b92021-09-01 16:46:57 -04002213 if (!TWFunc::write_to_file(Lun_File, Part->Actual_Block_Device)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00002214 LOGERR("Unable to write to ums lunfile '%s': (%s)\n", Lun_File.c_str(), strerror(errno));
Dees_Troyd21618c2012-10-14 18:48:49 -04002215 return false;
2216 }
Dees_Troyd21618c2012-10-14 18:48:49 -04002217 return true;
2218}
2219
Dees_Troy8170a922012-09-18 15:40:25 -04002220int TWPartitionManager::usb_storage_enable(void) {
Dees_Troy8170a922012-09-18 15:40:25 -04002221 char lun_file[255];
Dees_Troyd21618c2012-10-14 18:48:49 -04002222 bool has_multiple_lun = false;
Dees_Troy8170a922012-09-18 15:40:25 -04002223
Ethan Yonker47360be2014-04-01 10:34:34 -05002224 string Lun_File_str = CUSTOM_LUN_FILE;
2225 size_t found = Lun_File_str.find("%");
2226 if (found != string::npos) {
2227 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
2228 if (TWFunc::Path_Exists(lun_file))
2229 has_multiple_lun = true;
2230 }
Ethan Yonker726a0202014-12-16 20:01:38 -06002231 mtp_was_enabled = TWFunc::Toggle_MTP(false); // Must disable MTP for USB Storage
Ethan Yonker47360be2014-04-01 10:34:34 -05002232 if (!has_multiple_lun) {
2233 LOGINFO("Device doesn't have multiple lun files, mount current storage\n");
2234 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
2235 if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) == "/data") {
Ethan Yonker66a19492015-12-10 10:19:45 -06002236 TWPartition* Mount = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05002237 if (Mount) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002238 if (!Open_Lun_File(Mount->Mount_Point, lun_file)) {
2239 goto error_handle;
2240 }
Ethan Yonker47360be2014-04-01 10:34:34 -05002241 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002242 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002243 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05002244 }
2245 } else if (!Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file)) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002246 goto error_handle;
Dees_Troy8170a922012-09-18 15:40:25 -04002247 }
Dees_Troy8170a922012-09-18 15:40:25 -04002248 } else {
Ethan Yonker47360be2014-04-01 10:34:34 -05002249 LOGINFO("Device has multiple lun files\n");
2250 TWPartition* Mount1;
2251 TWPartition* Mount2;
Dees_Troy8170a922012-09-18 15:40:25 -04002252 sprintf(lun_file, CUSTOM_LUN_FILE, 0);
Ethan Yonker66a19492015-12-10 10:19:45 -06002253 Mount1 = Find_Next_Storage("", true);
Ethan Yonker47360be2014-04-01 10:34:34 -05002254 if (Mount1) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002255 if (!Open_Lun_File(Mount1->Mount_Point, lun_file)) {
2256 goto error_handle;
2257 }
Matt Moweree717062014-04-26 01:46:46 -05002258 sprintf(lun_file, CUSTOM_LUN_FILE, 1);
Ethan Yonker66a19492015-12-10 10:19:45 -06002259 Mount2 = Find_Next_Storage(Mount1->Mount_Point, true);
Matt Mower1fdcdb72016-01-25 20:53:47 -06002260 if (Mount2 && Mount2->Mount_Point != Mount1->Mount_Point) {
Matt Moweree717062014-04-26 01:46:46 -05002261 Open_Lun_File(Mount2->Mount_Point, lun_file);
nkk717aa6fc62017-03-24 18:28:03 +02002262 // Mimic single lun code: Mount CurrentStoragePath if it's not /data
2263 } else if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) != "/data") {
2264 Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file);
Ethan Yonker47360be2014-04-01 10:34:34 -05002265 }
nkk717aa6fc62017-03-24 18:28:03 +02002266 // Mimic single lun code: Mount CurrentStoragePath if it's not /data
2267 } else if (TWFunc::Get_Root_Path(DataManager::GetCurrentStoragePath()) != "/data" && !Open_Lun_File(DataManager::GetCurrentStoragePath(), lun_file)) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002268 gui_err("unable_locate_storage=Unable to locate storage device.");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002269 goto error_handle;
Ethan Yonker47360be2014-04-01 10:34:34 -05002270 }
Dees_Troy8170a922012-09-18 15:40:25 -04002271 }
Matt Mowerb6ef4782014-11-06 02:24:36 -06002272 property_set("sys.storage.ums_enabled", "1");
HandyMenny37d42992014-10-26 22:15:59 +01002273 property_set("sys.usb.config", "mass_storage,adb");
Dees_Troy8170a922012-09-18 15:40:25 -04002274 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002275error_handle:
2276 if (mtp_was_enabled)
2277 if (!Enable_MTP())
2278 Disable_MTP();
2279 return false;
Dees_Troy8170a922012-09-18 15:40:25 -04002280}
2281
2282int TWPartitionManager::usb_storage_disable(void) {
bigbiff22851b92021-09-01 16:46:57 -04002283 int index, ret = 0;
Dees_Troy2673cec2013-04-02 20:22:16 +00002284 char lun_file[255], ch[2] = {0, 0};
2285 string str = ch;
Dees_Troy8170a922012-09-18 15:40:25 -04002286
2287 for (index=0; index<2; index++) {
2288 sprintf(lun_file, CUSTOM_LUN_FILE, index);
bigbiff22851b92021-09-01 16:46:57 -04002289 if (!TWFunc::write_to_file(lun_file, str)) {
Dees_Troy2673cec2013-04-02 20:22:16 +00002290 break;
bigbiff22851b92021-09-01 16:46:57 -04002291 ret = -1;
Dees_Troy8170a922012-09-18 15:40:25 -04002292 }
Dees_Troy8170a922012-09-18 15:40:25 -04002293 }
Dees_Troye58d5262012-09-21 12:27:57 -04002294 Mount_All_Storage();
2295 Update_System_Details();
Dees_Troycfd73ef2012-10-12 16:52:00 -04002296 UnMount_Main_Partitions();
Matt Mowerd9cb9062013-12-14 20:34:45 -06002297 property_set("sys.storage.ums_enabled", "0");
HandyMenny37d42992014-10-26 22:15:59 +01002298 property_set("sys.usb.config", "adb");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002299 if (mtp_was_enabled)
2300 if (!Enable_MTP())
2301 Disable_MTP();
Dees_Troy2673cec2013-04-02 20:22:16 +00002302 if (ret < 0 && index == 0) {
2303 LOGERR("Unable to write to ums lunfile '%s'.", lun_file);
2304 return false;
2305 } else {
2306 return true;
2307 }
Dees_Troy8170a922012-09-18 15:40:25 -04002308 return true;
Dees_Troy812660f2012-09-20 09:55:17 -04002309}
2310
2311void TWPartitionManager::Mount_All_Storage(void) {
2312 std::vector<TWPartition*>::iterator iter;
2313
2314 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2315 if ((*iter)->Is_Storage)
2316 (*iter)->Mount(false);
2317 }
Dees_Troy2c50e182012-09-26 20:05:28 -04002318}
Dees_Troy9350b8d2012-09-27 12:38:38 -04002319
Dees_Troyd0384ef2012-10-12 12:15:42 -04002320void TWPartitionManager::UnMount_Main_Partitions(void) {
2321 // Unmounts system and data if data is not data/media
2322 // Also unmounts boot if boot is mountable
Dees_Troy2673cec2013-04-02 20:22:16 +00002323 LOGINFO("Unmounting main partitions...\n");
Dees_Troyd0384ef2012-10-12 12:15:42 -04002324
Mohd Faraze3948ec2020-08-14 01:40:59 +00002325 TWPartition *Partition = Find_Partition_By_Path ("/vendor");
Dees_Troyd0384ef2012-10-12 12:15:42 -04002326
Mohd Faraze3948ec2020-08-14 01:40:59 +00002327 if (Partition != NULL) UnMount_By_Path("/vendor", false);
2328 UnMount_By_Path (Get_Android_Root_Path(), true);
2329 Partition = Find_Partition_By_Path ("/product");
2330 if (Partition != NULL) UnMount_By_Path("/product", false);
Ethan Yonker6277c792014-09-15 14:54:30 -05002331 if (!datamedia)
2332 UnMount_By_Path("/data", true);
2333
Mohd Faraze3948ec2020-08-14 01:40:59 +00002334 Partition = Find_Partition_By_Path ("/boot");
2335 if (Partition != NULL && Partition->Can_Be_Mounted)
2336 Partition->UnMount(true);
Dees_Troyd0384ef2012-10-12 12:15:42 -04002337}
2338
Dees_Troy9350b8d2012-09-27 12:38:38 -04002339int TWPartitionManager::Partition_SDCard(void) {
Matt Mower23d8aae2017-01-06 14:30:33 -06002340 char temp[255];
Ethan Yonker483e9f42016-01-11 22:21:18 -06002341 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 -04002342 int ext, swap, total_size = 0, fat_size;
Dees_Troy9350b8d2012-09-27 12:38:38 -04002343
Ethan Yonker74db1572015-10-28 12:44:49 -05002344 gui_msg("start_partition_sd=Partitioning SD Card...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06002345
2346 // Locate and validate device to partition
2347 TWPartition* SDCard = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
2348
Ethan Yonker66a19492015-12-10 10:19:45 -06002349 if (SDCard->Is_Adopted_Storage)
2350 SDCard->Revert_Adopted();
2351
Ethan Yonker1eff6cd2014-09-15 13:30:42 -05002352 if (SDCard == NULL || !SDCard->Removable || SDCard->Has_Data_Media) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002353 gui_err("partition_sd_locate=Unable to locate device to partition.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002354 return false;
2355 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002356
2357 // Unmount everything
Dees_Troy9350b8d2012-09-27 12:38:38 -04002358 if (!SDCard->UnMount(true))
2359 return false;
2360 TWPartition* SDext = Find_Partition_By_Path("/sd-ext");
2361 if (SDext != NULL) {
2362 if (!SDext->UnMount(true))
2363 return false;
2364 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002365 char* swappath = getenv("SWAPPATH");
2366 if (swappath != NULL) {
2367 LOGINFO("Unmounting swap at '%s'\n", swappath);
2368 umount(swappath);
2369 }
Vojtech Bocek05534202013-09-11 08:11:56 +02002370
Ethan Yonker483e9f42016-01-11 22:21:18 -06002371 // Determine block device
2372 if (SDCard->Alternate_Block_Device.empty()) {
2373 SDCard->Find_Actual_Block_Device();
2374 Device = SDCard->Actual_Block_Device;
2375 // Just use the root block device
2376 Device.resize(strlen("/dev/block/mmcblkX"));
2377 } else {
2378 Device = SDCard->Alternate_Block_Device;
2379 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04002380
2381 // Find the size of the block device:
Ethan Yonker483e9f42016-01-11 22:21:18 -06002382 total_size = (int)(TWFunc::IOCTL_Get_Block_Size(Device.c_str()) / (1048576));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002383
2384 DataManager::GetValue("tw_sdext_size", ext);
2385 DataManager::GetValue("tw_swap_size", swap);
2386 DataManager::GetValue("tw_sdpart_file_system", ext_format);
2387 fat_size = total_size - ext - swap;
Ethan Yonker483e9f42016-01-11 22:21:18 -06002388 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);
2389
2390 // Determine partition sizes
2391 if (swap == 0 && ext == 0) {
2392 fat_str = "-0";
2393 } else {
2394 memset(temp, 0, sizeof(temp));
2395 sprintf(temp, "%i", fat_size);
2396 fat_str = temp;
2397 fat_str += "MB";
2398 }
2399 if (swap == 0) {
2400 ext_str = "-0";
2401 } else {
2402 memset(temp, 0, sizeof(temp));
2403 sprintf(temp, "%i", ext);
2404 ext_str = "+";
2405 ext_str += temp;
2406 ext_str += "MB";
2407 }
2408
Dees_Troy9350b8d2012-09-27 12:38:38 -04002409 if (ext + swap > total_size) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002410 gui_err("ext_swap_size=EXT + Swap size is larger than sdcard size.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002411 return false;
2412 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002413
Ethan Yonker74db1572015-10-28 12:44:49 -05002414 gui_msg("remove_part_table=Removing partition table...");
Ethan Yonker483e9f42016-01-11 22:21:18 -06002415 Command = "sgdisk --zap-all " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002416 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002417 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002418 gui_err("unable_rm_part=Unable to remove partition table.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002419 Update_System_Details();
2420 return false;
2421 }
Ethan Yonker74db1572015-10-28 12:44:49 -05002422 gui_msg(Msg("create_part=Creating {1} partition...")("FAT32"));
Captain Throwback9643a802016-05-27 11:44:51 -04002423 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 +00002424 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002425 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002426 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("FAT32"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002427 return false;
2428 }
2429 if (ext > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002430 gui_msg(Msg("create_part=Creating {1} partition...")("EXT"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06002431 Command = "sgdisk --new=0:0:" + ext_str + " --change-name=0:\"Linux filesystem\" " + Device;
Dees_Troy2673cec2013-04-02 20:22:16 +00002432 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002433 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002434 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("EXT"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002435 Update_System_Details();
2436 return false;
2437 }
2438 }
2439 if (swap > 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002440 gui_msg(Msg("create_part=Creating {1} partition...")("swap"));
Ethan Yonker483e9f42016-01-11 22:21:18 -06002441 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 +00002442 LOGINFO("Command is: '%s'\n", Command.c_str());
Vojtech Bocek05534202013-09-11 08:11:56 +02002443 if (TWFunc::Exec_Cmd(Command) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002444 gui_msg(Msg(msg::kError, "unable_to_create_part=Unable to create {1} partition.")("swap"));
Dees_Troy9350b8d2012-09-27 12:38:38 -04002445 Update_System_Details();
2446 return false;
2447 }
2448 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002449
2450 // Convert GPT to MBR
2451 Command = "sgdisk --gpttombr " + Device;
2452 if (TWFunc::Exec_Cmd(Command) != 0)
2453 LOGINFO("Failed to covert partition GPT to MBR\n");
2454
2455 // Tell the kernel to rescan the partition table
2456 int fd = open(Device.c_str(), O_RDONLY);
2457 ioctl(fd, BLKRRPART, 0);
2458 close(fd);
2459
2460 string format_device = Device;
2461 if (Device.substr(0, 17) == "/dev/block/mmcblk")
2462 format_device += "p";
2463
2464 // Format new partitions to proper file system
2465 if (fat_size > 0) {
2466 Command = "mkfs.fat " + format_device + "1";
2467 TWFunc::Exec_Cmd(Command);
2468 }
Dees_Troy9350b8d2012-09-27 12:38:38 -04002469 if (ext > 0) {
2470 if (SDext == NULL) {
Ethan Yonker483e9f42016-01-11 22:21:18 -06002471 Command = "mke2fs -t " + ext_format + " -m 0 " + format_device + "2";
2472 gui_msg(Msg("format_sdext_as=Formatting sd-ext as {1}...")(ext_format));
2473 LOGINFO("Formatting sd-ext after partitioning, command: '%s'\n", Command.c_str());
2474 TWFunc::Exec_Cmd(Command);
2475 } else {
2476 SDext->Wipe(ext_format);
Dees_Troy9350b8d2012-09-27 12:38:38 -04002477 }
Ethan Yonker483e9f42016-01-11 22:21:18 -06002478 }
2479 if (swap > 0) {
2480 Command = "mkswap " + format_device;
2481 if (ext > 0)
2482 Command += "3";
2483 else
2484 Command += "2";
Vojtech Bocek05534202013-09-11 08:11:56 +02002485 TWFunc::Exec_Cmd(Command);
Dees_Troy9350b8d2012-09-27 12:38:38 -04002486 }
2487
Ethan Yonker483e9f42016-01-11 22:21:18 -06002488 // recreate TWRP folder and rewrite settings - these will be gone after sdcard is partitioned
2489 if (SDCard->Mount(true)) {
2490 string TWRP_Folder = SDCard->Mount_Point + "/TWRP";
2491 mkdir(TWRP_Folder.c_str(), 0777);
2492 DataManager::Flush();
2493 }
2494
Dees_Troy9350b8d2012-09-27 12:38:38 -04002495 Update_System_Details();
Ethan Yonker74db1572015-10-28 12:44:49 -05002496 gui_msg("part_complete=Partitioning complete.");
Dees_Troy9350b8d2012-09-27 12:38:38 -04002497 return true;
bigbiff bigbiffa0f8a592012-10-09 21:01:03 -04002498}
Dees_Troya13d74f2013-03-24 08:54:55 -05002499
2500void TWPartitionManager::Get_Partition_List(string ListType, std::vector<PartitionList> *Partition_List) {
2501 std::vector<TWPartition*>::iterator iter;
2502 if (ListType == "mount") {
2503 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002504 if ((*iter)->Can_Be_Mounted) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002505 struct PartitionList part;
2506 part.Display_Name = (*iter)->Display_Name;
2507 part.Mount_Point = (*iter)->Mount_Point;
2508 part.selected = (*iter)->Is_Mounted();
2509 Partition_List->push_back(part);
2510 }
2511 }
2512 } else if (ListType == "storage") {
2513 char free_space[255];
2514 string Current_Storage = DataManager::GetCurrentStoragePath();
2515 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2516 if ((*iter)->Is_Storage) {
2517 struct PartitionList part;
2518 sprintf(free_space, "%llu", (*iter)->Free / 1024 / 1024);
2519 part.Display_Name = (*iter)->Storage_Name + " (";
2520 part.Display_Name += free_space;
2521 part.Display_Name += "MB)";
2522 part.Mount_Point = (*iter)->Storage_Path;
2523 if ((*iter)->Storage_Path == Current_Storage)
2524 part.selected = 1;
2525 else
2526 part.selected = 0;
2527 Partition_List->push_back(part);
2528 }
2529 }
2530 } else if (ListType == "backup") {
2531 char backup_size[255];
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002532 unsigned long long Backup_Size;
Dees_Troya13d74f2013-03-24 08:54:55 -05002533 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002534 if ((*iter)->Can_Be_Backed_Up && !(*iter)->Is_SubPartition && (*iter)->Is_Present) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002535 struct PartitionList part;
Dees_Troy9e0b71c2013-04-08 13:35:37 +00002536 Backup_Size = (*iter)->Backup_Size;
2537 if ((*iter)->Has_SubPartition) {
2538 std::vector<TWPartition*>::iterator subpart;
2539
2540 for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
2541 if ((*subpart)->Is_SubPartition && (*subpart)->Can_Be_Backed_Up && (*subpart)->Is_Present && (*subpart)->SubPartition_Of == (*iter)->Mount_Point)
2542 Backup_Size += (*subpart)->Backup_Size;
2543 }
2544 }
2545 sprintf(backup_size, "%llu", Backup_Size / 1024 / 1024);
Dees_Troya13d74f2013-03-24 08:54:55 -05002546 part.Display_Name = (*iter)->Backup_Display_Name + " (";
2547 part.Display_Name += backup_size;
2548 part.Display_Name += "MB)";
2549 part.Mount_Point = (*iter)->Backup_Path;
2550 part.selected = 0;
2551 Partition_List->push_back(part);
2552 }
2553 }
2554 } else if (ListType == "restore") {
2555 string Restore_List, restore_path;
2556 TWPartition* restore_part = NULL;
2557
2558 DataManager::GetValue("tw_restore_list", Restore_List);
2559 if (!Restore_List.empty()) {
2560 size_t start_pos = 0, end_pos = Restore_List.find(";", start_pos);
2561 while (end_pos != string::npos && start_pos < Restore_List.size()) {
2562 restore_path = Restore_List.substr(start_pos, end_pos - start_pos);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04002563 struct PartitionList part;
2564 if (restore_path.compare("ADB_Backup") == 0) {
2565 part.Display_Name = "ADB Backup";
2566 part.Mount_Point = "ADB Backup";
2567 part.selected = 1;
2568 Partition_List->push_back(part);
2569 break;
2570 }
Dees_Troy59df9262013-06-19 14:53:57 -05002571 if ((restore_part = Find_Partition_By_Path(restore_path)) != NULL) {
Dees Troy4159aed2014-02-28 17:24:43 +00002572 if ((restore_part->Backup_Name == "recovery" && !restore_part->Can_Be_Backed_Up) || restore_part->Is_SubPartition) {
Dees_Troya13d74f2013-03-24 08:54:55 -05002573 // Don't allow restore of recovery (causes problems on some devices)
Dees_Troy59df9262013-06-19 14:53:57 -05002574 // Don't add subpartitions to the list of items
Dees_Troya13d74f2013-03-24 08:54:55 -05002575 } else {
Dees_Troya13d74f2013-03-24 08:54:55 -05002576 part.Display_Name = restore_part->Backup_Display_Name;
2577 part.Mount_Point = restore_part->Backup_Path;
2578 part.selected = 1;
2579 Partition_List->push_back(part);
2580 }
2581 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002582 gui_msg(Msg(msg::kError, "restore_unable_locate=Unable to locate '{1}' partition for restoring.")(restore_path));
Dees_Troya13d74f2013-03-24 08:54:55 -05002583 }
2584 start_pos = end_pos + 1;
2585 end_pos = Restore_List.find(";", start_pos);
2586 }
2587 }
2588 } else if (ListType == "wipe") {
2589 struct PartitionList dalvik;
Ethan Yonker74db1572015-10-28 12:44:49 -05002590 dalvik.Display_Name = gui_parse_text("{@dalvik}");
Dees_Troya13d74f2013-03-24 08:54:55 -05002591 dalvik.Mount_Point = "DALVIK";
2592 dalvik.selected = 0;
2593 Partition_List->push_back(dalvik);
2594 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2595 if ((*iter)->Wipe_Available_in_GUI && !(*iter)->Is_SubPartition) {
2596 struct PartitionList part;
2597 part.Display_Name = (*iter)->Display_Name;
2598 part.Mount_Point = (*iter)->Mount_Point;
2599 part.selected = 0;
2600 Partition_List->push_back(part);
2601 }
2602 if ((*iter)->Has_Android_Secure) {
2603 struct PartitionList part;
2604 part.Display_Name = (*iter)->Backup_Display_Name;
2605 part.Mount_Point = (*iter)->Backup_Path;
2606 part.selected = 0;
2607 Partition_List->push_back(part);
2608 }
Dees_Troy74fb2e92013-04-15 14:35:47 +00002609 if ((*iter)->Has_Data_Media) {
2610 struct PartitionList datamedia;
2611 datamedia.Display_Name = (*iter)->Storage_Name;
2612 datamedia.Mount_Point = "INTERNAL";
2613 datamedia.selected = 0;
2614 Partition_List->push_back(datamedia);
2615 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002616 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002617 } else if (ListType == "flashimg") {
2618 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2619 if ((*iter)->Can_Flash_Img && (*iter)->Is_Present) {
2620 struct PartitionList part;
2621 part.Display_Name = (*iter)->Backup_Display_Name;
2622 part.Mount_Point = (*iter)->Backup_Path;
2623 part.selected = 0;
2624 Partition_List->push_back(part);
2625 }
2626 }
Ethan Yonker53796e72019-01-11 22:49:52 -06002627 if (DataManager::GetIntValue("tw_has_repack_tools") != 0 && DataManager::GetIntValue("tw_has_boot_slots") != 0) {
2628 TWPartition* boot = Find_Partition_By_Path("/boot");
2629 if (boot) {
2630 // Allow flashing kernels and ramdisks
2631 struct PartitionList repack_ramdisk;
2632 repack_ramdisk.Display_Name = gui_lookup("install_twrp_ramdisk", "Install Recovery Ramdisk");
2633 repack_ramdisk.Mount_Point = "/repack_ramdisk";
2634 repack_ramdisk.selected = 0;
2635 Partition_List->push_back(repack_ramdisk);
2636 /*struct PartitionList repack_kernel; For now let's leave repacking kernels under advanced only
2637 repack_kernel.Display_Name = gui_lookup("install_kernel", "Install Kernel");
2638 repack_kernel.Mount_Point = "/repack_kernel";
2639 repack_kernel.selected = 0;
2640 Partition_List->push_back(repack_kernel);*/
2641 }
2642 }
Dees_Troya13d74f2013-03-24 08:54:55 -05002643 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +00002644 LOGERR("Unknown list type '%s' requested for TWPartitionManager::Get_Partition_List\n", ListType.c_str());
Dees_Troya13d74f2013-03-24 08:54:55 -05002645 }
2646}
2647
2648int TWPartitionManager::Fstab_Processed(void) {
2649 return Partitions.size();
2650}
Dees_Troyd93bda52013-07-03 19:55:19 +00002651
2652void TWPartitionManager::Output_Storage_Fstab(void) {
2653 std::vector<TWPartition*>::iterator iter;
2654 char storage_partition[255];
bigbiff bigbiffe4bdb152019-03-23 18:33:17 -04002655 std::string Temp;
bigbiff25d25b92020-06-19 16:07:38 -04002656 std::string cacheDir = TWFunc::get_log_dir();
bigbiff bigbiffe4bdb152019-03-23 18:33:17 -04002657
2658 if (cacheDir.empty()) {
2659 LOGINFO("Unable to find cache directory\n");
2660 return;
2661 }
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002662
bigbiff25d25b92020-06-19 16:07:38 -04002663 std::string storageFstab = TWFunc::get_log_dir() + "recovery/storage.fstab";
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002664 FILE *fp = fopen(storageFstab.c_str(), "w");
Dees_Troyd93bda52013-07-03 19:55:19 +00002665
2666 if (fp == NULL) {
bigbiff bigbiff19874f12019-01-08 20:06:57 -05002667 gui_msg(Msg(msg::kError, "unable_to_open=Unable to open '{1}'.")(storageFstab));
Dees_Troyd93bda52013-07-03 19:55:19 +00002668 return;
2669 }
2670
2671 // Iterate through all partitions
2672 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2673 if ((*iter)->Is_Storage) {
2674 Temp = (*iter)->Storage_Path + ";" + (*iter)->Storage_Name + ";\n";
2675 strcpy(storage_partition, Temp.c_str());
2676 fwrite(storage_partition, sizeof(storage_partition[0]), strlen(storage_partition) / sizeof(storage_partition[0]), fp);
2677 }
2678 }
2679 fclose(fp);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02002680}
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002681
2682TWPartition *TWPartitionManager::Get_Default_Storage_Partition()
2683{
2684 TWPartition *res = NULL;
2685 for (std::vector<TWPartition*>::iterator iter = Partitions.begin(); iter != Partitions.end(); ++iter) {
Matt Mowera8a89d12016-12-30 18:10:37 -06002686 if (!(*iter)->Is_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002687 continue;
2688
Matt Mowera8a89d12016-12-30 18:10:37 -06002689 if ((*iter)->Is_Settings_Storage)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002690 return *iter;
2691
Matt Mowera8a89d12016-12-30 18:10:37 -06002692 if (!res)
Vojtech Bocek93cb1ef2014-05-12 15:41:52 +02002693 res = *iter;
2694 }
2695 return res;
2696}
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002697
2698bool TWPartitionManager::Enable_MTP(void) {
2699#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002700 if (mtppid) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002701 gui_err("mtp_already_enabled=MTP already enabled");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002702 return true;
2703 }
Ethan Yonker726a0202014-12-16 20:01:38 -06002704
2705 int mtppipe[2];
2706
2707 if (pipe(mtppipe) < 0) {
2708 LOGERR("Error creating MTP pipe\n");
2709 return false;
2710 }
2711
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002712 char old_value[PROPERTY_VALUE_MAX];
Matt Mowere07f0102017-02-23 17:40:00 -06002713 property_get("sys.usb.config", old_value, "");
2714 if (strcmp(old_value, "mtp,adb") != 0) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002715 char vendor[PROPERTY_VALUE_MAX];
2716 char product[PROPERTY_VALUE_MAX];
2717 property_set("sys.usb.config", "none");
2718 property_get("usb.vendor", vendor, "18D1");
2719 property_get("usb.product.mtpadb", product, "4EE2");
2720 string vendorstr = vendor;
2721 string productstr = product;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002722 TWFunc::write_to_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2723 TWFunc::write_to_file("/sys/class/android_usb/android0/idProduct", productstr);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002724 property_set("sys.usb.config", "mtp,adb");
2725 }
Matt Mower653a1702017-02-16 10:38:52 -06002726 /* To enable MTP debug, use the twrp command line feature:
Ethan Yonker6d154c42014-09-03 14:19:43 -05002727 * twrp set tw_mtp_debug 1
2728 */
2729 twrpMtp *mtp = new twrpMtp(DataManager::GetIntValue("tw_mtp_debug"));
Ethan Yonker1b039202015-01-30 10:08:48 -06002730 mtppid = mtp->forkserver(mtppipe);
2731 if (mtppid) {
2732 close(mtppipe[0]); // Host closes read side
2733 mtp_write_fd = mtppipe[1];
2734 DataManager::SetValue("tw_mtp_enabled", 1);
2735 Add_All_MTP_Storage();
2736 return true;
Ethan Yonker726a0202014-12-16 20:01:38 -06002737 } else {
2738 close(mtppipe[0]);
2739 close(mtppipe[1]);
Ethan Yonker74db1572015-10-28 12:44:49 -05002740 gui_err("mtp_fail=Failed to enable MTP");
Ethan Yonker1b039202015-01-30 10:08:48 -06002741 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002742 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002743#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002744 gui_err("no_mtp=MTP support not included");
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002745#endif
2746 DataManager::SetValue("tw_mtp_enabled", 0);
2747 return false;
2748}
2749
Ethan Yonker1b039202015-01-30 10:08:48 -06002750void TWPartitionManager::Add_All_MTP_Storage(void) {
2751#ifdef TW_HAS_MTP
2752 std::vector<TWPartition*>::iterator iter;
2753
2754 if (!mtppid)
2755 return; // MTP is not enabled
2756
2757 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2758 if ((*iter)->Is_Storage && (*iter)->Is_Present && (*iter)->Mount(false))
2759 Add_Remove_MTP_Storage((*iter), MTP_MESSAGE_ADD_STORAGE);
2760 }
2761#else
2762 return;
2763#endif
2764}
2765
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002766bool TWPartitionManager::Disable_MTP(void) {
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002767 char old_value[PROPERTY_VALUE_MAX];
bigbiff bigbiffaf32bb92018-12-18 18:39:53 -05002768 property_set("sys.usb.ffs.mtp.ready", "0");
Matt Mowere07f0102017-02-23 17:40:00 -06002769 property_get("sys.usb.config", old_value, "");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002770 if (strcmp(old_value, "adb") != 0) {
2771 char vendor[PROPERTY_VALUE_MAX];
2772 char product[PROPERTY_VALUE_MAX];
2773 property_set("sys.usb.config", "none");
2774 property_get("usb.vendor", vendor, "18D1");
Matt Mowere07f0102017-02-23 17:40:00 -06002775 property_get("usb.product.adb", product, "D001");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002776 string vendorstr = vendor;
2777 string productstr = product;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002778 TWFunc::write_to_file("/sys/class/android_usb/android0/idVendor", vendorstr);
2779 TWFunc::write_to_file("/sys/class/android_usb/android0/idProduct", productstr);
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002780 usleep(2000);
2781 }
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002782#ifdef TW_HAS_MTP
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002783 if (mtppid) {
Ethan Yonker8613dc02014-09-11 09:28:20 -05002784 LOGINFO("Disabling MTP\n");
2785 int status;
2786 kill(mtppid, SIGKILL);
Ethan Yonker8dfa7772014-09-04 21:48:41 -05002787 mtppid = 0;
Ethan Yonker8613dc02014-09-11 09:28:20 -05002788 // We don't care about the exit value, but this prevents a zombie process
2789 waitpid(mtppid, &status, 0);
Ethan Yonker726a0202014-12-16 20:01:38 -06002790 close(mtp_write_fd);
2791 mtp_write_fd = -1;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002792 }
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002793#endif
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002794 property_set("sys.usb.config", "adb");
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002795#ifdef TW_HAS_MTP
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002796 DataManager::SetValue("tw_mtp_enabled", 0);
2797 return true;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002798#endif
Ethan Yonkerdf7abac2014-12-29 10:48:17 -06002799 return false;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04002800}
Ethan Yonker726a0202014-12-16 20:01:38 -06002801
2802TWPartition* TWPartitionManager::Find_Partition_By_MTP_Storage_ID(unsigned int Storage_ID) {
2803 std::vector<TWPartition*>::iterator iter;
2804
2805 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
2806 if ((*iter)->MTP_Storage_ID == Storage_ID)
2807 return (*iter);
2808 }
2809 return NULL;
2810}
2811
2812bool TWPartitionManager::Add_Remove_MTP_Storage(TWPartition* Part, int message_type) {
2813#ifdef TW_HAS_MTP
2814 struct mtpmsg mtp_message;
2815
2816 if (!mtppid)
2817 return false; // MTP is disabled
2818
2819 if (mtp_write_fd < 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002820 LOGINFO("MTP: mtp_write_fd is not set\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002821 return false;
2822 }
2823
2824 if (Part) {
Ethan Yonker4bfabab2014-12-29 09:15:37 -06002825 if (Part->MTP_Storage_ID == 0)
2826 return false;
Ethan Yonker726a0202014-12-16 20:01:38 -06002827 if (message_type == MTP_MESSAGE_REMOVE_STORAGE) {
2828 mtp_message.message_type = MTP_MESSAGE_REMOVE_STORAGE; // Remove
2829 LOGINFO("sending message to remove %i\n", Part->MTP_Storage_ID);
2830 mtp_message.storage_id = Part->MTP_Storage_ID;
2831 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002832 LOGINFO("error sending message to remove storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002833 return false;
2834 } else {
2835 LOGINFO("Message sent, remove storage ID: %i\n", Part->MTP_Storage_ID);
2836 return true;
2837 }
2838 } else if (message_type == MTP_MESSAGE_ADD_STORAGE && Part->Is_Mounted()) {
2839 mtp_message.message_type = MTP_MESSAGE_ADD_STORAGE; // Add
2840 mtp_message.storage_id = Part->MTP_Storage_ID;
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002841 if (Part->Storage_Path.size() >= sizeof(mtp_message.path)) {
2842 LOGERR("Storage path '%s' too large for mtpmsg\n", Part->Storage_Path.c_str());
2843 return false;
2844 }
2845 strcpy(mtp_message.path, Part->Storage_Path.c_str());
2846 if (Part->Storage_Name.size() >= sizeof(mtp_message.display)) {
2847 LOGERR("Storage name '%s' too large for mtpmsg\n", Part->Storage_Name.c_str());
2848 return false;
2849 }
2850 strcpy(mtp_message.display, Part->Storage_Name.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002851 mtp_message.maxFileSize = Part->Get_Max_FileSize();
Ethan Yonker1b039202015-01-30 10:08:48 -06002852 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 -06002853 if (write(mtp_write_fd, &mtp_message, sizeof(mtp_message)) <= 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002854 LOGINFO("error sending message to add storage %i\n", Part->MTP_Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002855 return false;
2856 } else {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06002857 LOGINFO("Message sent, add storage ID: %i '%s'\n", Part->MTP_Storage_ID, mtp_message.path);
Ethan Yonker726a0202014-12-16 20:01:38 -06002858 return true;
2859 }
2860 } else {
2861 LOGERR("Unknown MTP message type: %i\n", message_type);
2862 }
2863 } else {
2864 // This hopefully never happens as the error handling should
2865 // occur in the calling function.
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002866 LOGINFO("TWPartitionManager::Add_Remove_MTP_Storage NULL partition given\n");
Ethan Yonker726a0202014-12-16 20:01:38 -06002867 }
2868 return true;
2869#else
Ethan Yonker74db1572015-10-28 12:44:49 -05002870 gui_err("no_mtp=MTP support not included");
Ethan Yonker726a0202014-12-16 20:01:38 -06002871 DataManager::SetValue("tw_mtp_enabled", 0);
2872 return false;
2873#endif
2874}
2875
2876bool TWPartitionManager::Add_MTP_Storage(string Mount_Point) {
2877#ifdef TW_HAS_MTP
2878 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2879 if (Part) {
2880 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2881 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002882 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002883 }
2884#endif
2885 return false;
2886}
2887
2888bool TWPartitionManager::Add_MTP_Storage(unsigned int Storage_ID) {
2889#ifdef TW_HAS_MTP
2890 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2891 if (Part) {
2892 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_ADD_STORAGE);
2893 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002894 LOGINFO("TWFunc::Add_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002895 }
2896#endif
2897 return false;
2898}
2899
2900bool TWPartitionManager::Remove_MTP_Storage(string Mount_Point) {
2901#ifdef TW_HAS_MTP
2902 TWPartition* Part = PartitionManager.Find_Partition_By_Path(Mount_Point);
2903 if (Part) {
2904 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2905 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002906 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for '%s'\n", Mount_Point.c_str());
Ethan Yonker726a0202014-12-16 20:01:38 -06002907 }
2908#endif
2909 return false;
2910}
2911
2912bool TWPartitionManager::Remove_MTP_Storage(unsigned int Storage_ID) {
2913#ifdef TW_HAS_MTP
2914 TWPartition* Part = PartitionManager.Find_Partition_By_MTP_Storage_ID(Storage_ID);
2915 if (Part) {
2916 return PartitionManager.Add_Remove_MTP_Storage(Part, MTP_MESSAGE_REMOVE_STORAGE);
2917 } else {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06002918 LOGINFO("TWFunc::Remove_MTP_Storage unable to locate partition for %i\n", Storage_ID);
Ethan Yonker726a0202014-12-16 20:01:38 -06002919 }
2920#endif
2921 return false;
2922}
Ethan Yonker96af84a2015-01-05 14:58:36 -06002923
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002924bool TWPartitionManager::Flash_Image(string& path, string& filename) {
bigbiffad58e1b2020-07-06 20:24:34 -04002925 twrpRepacker repacker;
Matt Mower23d8aae2017-01-06 14:30:33 -06002926 int partition_count = 0;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002927 TWPartition* flash_part = NULL;
bigbiffce8f83c2015-12-12 18:30:21 -05002928 string Flash_List, flash_path, full_filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002929 size_t start_pos = 0, end_pos = 0;
2930
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002931 full_filename = path + "/" + filename;
Ethan Yonker96af84a2015-01-05 14:58:36 -06002932
bigbiffce8f83c2015-12-12 18:30:21 -05002933 gui_msg("image_flash_start=[IMAGE FLASH STARTED]");
2934 gui_msg(Msg("img_to_flash=Image to flash: '{1}'")(full_filename));
2935
2936 if (!TWFunc::Path_Exists(full_filename)) {
2937 if (!Mount_By_Path(full_filename, true)) {
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002938 return false;
2939 }
bigbiffce8f83c2015-12-12 18:30:21 -05002940 if (!TWFunc::Path_Exists(full_filename)) {
2941 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")(full_filename));
Ethan Yonkerb78fbdf2016-01-15 16:46:35 -06002942 return false;
2943 }
2944 }
Ethan Yonker96af84a2015-01-05 14:58:36 -06002945
Ethan Yonker53796e72019-01-11 22:49:52 -06002946 DataManager::GetValue("tw_flash_partition", Flash_List);
2947 Repack_Type repack = REPLACE_NONE;
2948 if (Flash_List == "/repack_ramdisk;") {
2949 repack = REPLACE_RAMDISK;
2950 } else if (Flash_List == "/repack_kernel;") {
2951 repack = REPLACE_KERNEL;
2952 }
2953 if (repack != REPLACE_NONE) {
2954 Repack_Options_struct Repack_Options;
2955 Repack_Options.Type = repack;
2956 Repack_Options.Disable_Verity = false;
2957 Repack_Options.Disable_Force_Encrypt = false;
2958 Repack_Options.Backup_First = DataManager::GetIntValue("tw_repack_backup_first") != 0;
bigbiffad58e1b2020-07-06 20:24:34 -04002959 return repacker.Repack_Image_And_Flash(full_filename, Repack_Options);
Ethan Yonker53796e72019-01-11 22:49:52 -06002960 }
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002961 PartitionSettings part_settings;
2962 part_settings.Backup_Folder = path;
2963 unsigned long long total_bytes = TWFunc::Get_File_Size(full_filename);
2964 ProgressTracking progress(total_bytes);
2965 part_settings.progress = &progress;
2966 part_settings.adbbackup = false;
2967 part_settings.PM_Method = PM_RESTORE;
Ethan Yonker74db1572015-10-28 12:44:49 -05002968 gui_msg("calc_restore=Calculating restore details...");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002969 if (!Flash_List.empty()) {
2970 end_pos = Flash_List.find(";", start_pos);
2971 while (end_pos != string::npos && start_pos < Flash_List.size()) {
2972 flash_path = Flash_List.substr(start_pos, end_pos - start_pos);
2973 flash_part = Find_Partition_By_Path(flash_path);
2974 if (flash_part != NULL) {
2975 partition_count++;
2976 if (partition_count > 1) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002977 gui_err("too_many_flash=Too many partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002978 return false;
2979 }
2980 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05002981 gui_msg(Msg(msg::kError, "flash_unable_locate=Unable to locate '{1}' partition for flashing.")(flash_path));
Ethan Yonker96af84a2015-01-05 14:58:36 -06002982 return false;
2983 }
2984 start_pos = end_pos + 1;
2985 end_pos = Flash_List.find(";", start_pos);
2986 }
2987 }
2988
2989 if (partition_count == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05002990 gui_err("no_part_flash=No partitions selected for flashing.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06002991 return false;
2992 }
2993
2994 DataManager::SetProgress(0.0);
2995 if (flash_part) {
Ethan Yonkere080c1f2016-09-19 13:50:25 -05002996 flash_part->Backup_FileName = filename;
2997 if (!flash_part->Flash_Image(&part_settings))
Ethan Yonker96af84a2015-01-05 14:58:36 -06002998 return false;
2999 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05003000 gui_err("invalid_flash=Invalid flash partition specified.");
Ethan Yonker96af84a2015-01-05 14:58:36 -06003001 return false;
3002 }
Ethan Yonker74db1572015-10-28 12:44:49 -05003003 gui_highlight("flash_done=IMAGE FLASH COMPLETED]");
Ethan Yonker96af84a2015-01-05 14:58:36 -06003004 return true;
3005}
Ethan Yonker74db1572015-10-28 12:44:49 -05003006
3007void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value) {
3008 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
3009 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06003010 if (part->Is_Adopted_Storage) {
3011 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
3012 part->Backup_Display_Name = part->Display_Name;
3013 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
3014 } else {
3015 part->Display_Name = gui_lookup(resource_name, default_value);
3016 part->Backup_Display_Name = part->Display_Name;
3017 }
Ethan Yonker74db1572015-10-28 12:44:49 -05003018 }
3019}
3020
3021void TWPartitionManager::Translate_Partition(const char* path, const char* resource_name, const char* default_value, const char* storage_resource_name, const char* storage_default_value) {
3022 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
3023 if (part) {
Ethan Yonker66a19492015-12-10 10:19:45 -06003024 if (part->Is_Adopted_Storage) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06003025 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup("data_backup", "Data (excl. storage)");
Ethan Yonker66a19492015-12-10 10:19:45 -06003026 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
Ethan Yonker66a19492015-12-10 10:19:45 -06003027 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
3028 } else {
3029 part->Display_Name = gui_lookup(resource_name, default_value);
3030 part->Backup_Display_Name = part->Display_Name;
3031 if (part->Is_Storage)
3032 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
3033 }
Ethan Yonker74db1572015-10-28 12:44:49 -05003034 }
3035}
3036
Ethan Yonker01f4e032017-02-03 15:30:52 -06003037void 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) {
3038 TWPartition* part = PartitionManager.Find_Partition_By_Path(path);
3039 if (part) {
3040 if (part->Is_Adopted_Storage) {
3041 part->Backup_Display_Name = part->Display_Name + " - " + gui_lookup(backup_name, backup_default);
3042 part->Display_Name = part->Display_Name + " - " + gui_lookup("data", "Data");
3043 part->Storage_Name = part->Storage_Name + " - " + gui_lookup("adopted_storage", "Adopted Storage");
3044 } else {
3045 part->Display_Name = gui_lookup(resource_name, default_value);
3046 part->Backup_Display_Name = gui_lookup(backup_name, backup_default);
3047 if (part->Is_Storage)
3048 part->Storage_Name = gui_lookup(storage_resource_name, storage_default_value);
3049 }
3050 }
3051}
3052
Ethan Yonker74db1572015-10-28 12:44:49 -05003053void TWPartitionManager::Translate_Partition_Display_Names() {
Ethan Yonker66a19492015-12-10 10:19:45 -06003054 LOGINFO("Translating partition display names\n");
Ethan Yonker74db1572015-10-28 12:44:49 -05003055 Translate_Partition("/system", "system", "System");
3056 Translate_Partition("/system_image", "system_image", "System Image");
3057 Translate_Partition("/vendor", "vendor", "Vendor");
3058 Translate_Partition("/vendor_image", "vendor_image", "Vendor Image");
3059 Translate_Partition("/cache", "cache", "Cache");
Ethan Yonker74db1572015-10-28 12:44:49 -05003060 Translate_Partition("/boot", "boot", "Boot");
3061 Translate_Partition("/recovery", "recovery", "Recovery");
3062 if (!datamedia) {
Ethan Yonker01f4e032017-02-03 15:30:52 -06003063 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage");
Ethan Yonker74db1572015-10-28 12:44:49 -05003064 Translate_Partition("/sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
3065 Translate_Partition("/internal_sd", "sdcard", "SDCard", "sdcard", "SDCard");
3066 Translate_Partition("/internal_sdcard", "sdcard", "SDCard", "sdcard", "SDCard");
3067 Translate_Partition("/emmc", "sdcard", "SDCard", "sdcard", "SDCard");
Ethan Yonker01f4e032017-02-03 15:30:52 -06003068 } else {
3069 Translate_Partition("/data", "data", "Data", "internal", "Internal Storage", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05003070 }
Ethan Yonker01f4e032017-02-03 15:30:52 -06003071 Translate_Partition("/external_sd", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
3072 Translate_Partition("/external_sdcard", "microsd", "Micro SDCard", "microsd", "Micro SDCard", "data_backup", "Data (excl. storage)");
Ethan Yonker74db1572015-10-28 12:44:49 -05003073 Translate_Partition("/usb-otg", "usbotg", "USB OTG", "usbotg", "USB OTG");
3074 Translate_Partition("/sd-ext", "sdext", "SD-EXT");
3075
3076 // Android secure is a special case
3077 TWPartition* part = PartitionManager.Find_Partition_By_Path("/and-sec");
3078 if (part)
3079 part->Backup_Display_Name = gui_lookup("android_secure", "Android Secure");
3080
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003081 std::vector<TWPartition*>::iterator sysfs;
3082 for (sysfs = Partitions.begin(); sysfs != Partitions.end(); sysfs++) {
3083 if (!(*sysfs)->Sysfs_Entry.empty()) {
3084 Translate_Partition((*sysfs)->Mount_Point.c_str(), "autostorage", "Storage", "autostorage", "Storage");
3085 }
3086 }
3087
Ethan Yonker74db1572015-10-28 12:44:49 -05003088 // This updates the text on all of the storage selection buttons in the GUI
3089 DataManager::SetBackupFolder();
3090}
Ethan Yonker66a19492015-12-10 10:19:45 -06003091
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003092bool TWPartitionManager::Decrypt_Adopted() {
Ethan Yonker66a19492015-12-10 10:19:45 -06003093#ifdef TW_INCLUDE_CRYPTO
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003094 bool ret = false;
Ethan Yonker66a19492015-12-10 10:19:45 -06003095 if (!Mount_By_Path("/data", false)) {
3096 LOGERR("Cannot decrypt adopted storage because /data will not mount\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003097 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06003098 }
Darth98f775972022-08-05 09:39:03 +01003099
3100 std::string path = "/data/system/storage.xml";
3101 if (!TWFunc::Check_Xml_Format(path)) {
3102 std::string oldpath = path;
3103 if (TWFunc::abx_to_xml(oldpath, path)) {
3104 LOGINFO("Android 12+: '%s' has been converted into plain text xml (%s).\n", oldpath.c_str(), path.c_str());
3105 }
3106 }
3107
Fernando Oliveira25062d72023-02-01 12:52:32 -03003108 //Devices without encryption do not run the Post_Decrypt function so the "data/recovery" folder was not being created on these devices
3109 DataManager::SetValue("tw_settings_path", TW_STORAGE_PATH);
Ethan Yonker66a19492015-12-10 10:19:45 -06003110 LOGINFO("Decrypt adopted storage starting\n");
Darth98f775972022-08-05 09:39:03 +01003111 char* xmlFile = PageManager::LoadFileToBuffer(path, NULL);
Ethan Yonker66a19492015-12-10 10:19:45 -06003112 xml_document<> *doc = NULL;
3113 xml_node<>* volumes = NULL;
Ethan Yonker66a19492015-12-10 10:19:45 -06003114 string Primary_Storage_UUID = "";
3115 if (xmlFile != NULL) {
3116 LOGINFO("successfully loaded storage.xml\n");
3117 doc = new xml_document<>();
3118 doc->parse<0>(xmlFile);
3119 volumes = doc->first_node("volumes");
3120 if (volumes) {
3121 xml_attribute<>* psuuid = volumes->first_attribute("primaryStorageUuid");
3122 if (psuuid) {
3123 Primary_Storage_UUID = psuuid->value();
3124 }
3125 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003126 } else {
3127 LOGINFO("No /data/system/storage.xml for adopted storage\n");
3128 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06003129 }
3130 std::vector<TWPartition*>::iterator adopt;
3131 for (adopt = Partitions.begin(); adopt != Partitions.end(); adopt++) {
Peter Cai439d60c2019-10-17 08:49:10 +08003132 if ((*adopt)->Removable && !(*adopt)->Is_Present && (*adopt)->Adopted_Mount_Delay > 0) {
3133 // On some devices, the external mmc driver takes some time
3134 // to recognize the card, in which case the "actual block device"
3135 // would not have been found yet. We wait the specified delay
3136 // and then try again.
3137 LOGINFO("Sleeping %d seconds for adopted storage.\n", (*adopt)->Adopted_Mount_Delay);
3138 sleep((*adopt)->Adopted_Mount_Delay);
3139 (*adopt)->Find_Actual_Block_Device();
3140 }
3141
Ethan Yonker66a19492015-12-10 10:19:45 -06003142 if ((*adopt)->Removable && (*adopt)->Is_Present) {
3143 if ((*adopt)->Decrypt_Adopted() == 0) {
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003144 ret = true;
Ethan Yonker66a19492015-12-10 10:19:45 -06003145 if (volumes) {
3146 xml_node<>* volume = volumes->first_node("volume");
3147 while (volume) {
3148 xml_attribute<>* guid = volume->first_attribute("partGuid");
3149 if (guid) {
3150 string GUID = (*adopt)->Adopted_GUID.c_str();
3151 GUID.insert(8, "-");
3152 GUID.insert(13, "-");
3153 GUID.insert(18, "-");
3154 GUID.insert(23, "-");
3155
3156 if (strcasecmp(GUID.c_str(), guid->value()) == 0) {
3157 xml_attribute<>* attr = volume->first_attribute("nickname");
nkk71ffb02bd2017-06-04 23:12:16 +03003158 if (attr && attr->value() && strlen(attr->value()) > 0) {
Ethan Yonker66a19492015-12-10 10:19:45 -06003159 (*adopt)->Storage_Name = attr->value();
3160 (*adopt)->Display_Name = (*adopt)->Storage_Name;
3161 (*adopt)->Backup_Display_Name = (*adopt)->Storage_Name;
3162 LOGINFO("storage name from storage.xml is '%s'\n", attr->value());
3163 }
3164 attr = volume->first_attribute("fsUuid");
3165 if (attr && !Primary_Storage_UUID.empty() && strcmp(Primary_Storage_UUID.c_str(), attr->value()) == 0) {
3166 TWPartition* Dat = Find_Partition_By_Path("/data");
3167 if (Dat) {
3168 LOGINFO("Internal storage is found on adopted storage '%s'\n", (*adopt)->Display_Name.c_str());
3169 LOGINFO("Changing '%s' to point to '%s'\n", Dat->Symlink_Mount_Point.c_str(), (*adopt)->Storage_Path.c_str());
3170 (*adopt)->Symlink_Mount_Point = Dat->Symlink_Mount_Point;
3171 Dat->Symlink_Mount_Point = "";
3172 // Toggle mounts to ensure that the symlink mount point (probably /sdcard) is mounted to the right location
3173 Dat->UnMount(false);
3174 Dat->Mount(false);
3175 (*adopt)->UnMount(false);
3176 (*adopt)->Mount(false);
Ethan Yonker66a19492015-12-10 10:19:45 -06003177 }
3178 }
3179 break;
3180 }
3181 }
3182 volume = volume->next_sibling("volume");
3183 }
3184 }
nkk71ffb02bd2017-06-04 23:12:16 +03003185 Update_System_Details();
3186 Output_Partition((*adopt));
Ethan Yonker66a19492015-12-10 10:19:45 -06003187 }
3188 }
3189 }
3190 if (xmlFile) {
3191 doc->clear();
3192 delete doc;
3193 free(xmlFile);
3194 }
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003195 return ret;
Ethan Yonker66a19492015-12-10 10:19:45 -06003196#else
3197 LOGINFO("Decrypt_Adopted: no crypto support\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003198 return false;
Ethan Yonker66a19492015-12-10 10:19:45 -06003199#endif
3200}
Ethan Yonkerfcf3f242016-02-16 12:30:26 -06003201
3202void TWPartitionManager::Remove_Partition_By_Path(string Path) {
3203 std::vector<TWPartition*>::iterator iter;
3204 string Local_Path = TWFunc::Get_Root_Path(Path);
3205
3206 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3207 if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
3208 LOGINFO("Found and erasing '%s' from partition list\n", Local_Path.c_str());
3209 Partitions.erase(iter);
3210 return;
3211 }
3212 }
3213}
Ethan Yonker1b190162016-12-05 15:25:19 -06003214
bigbiff74a46272021-07-31 19:02:26 -04003215void TWPartitionManager::Override_Active_Slot(const string& Slot) {
3216 LOGINFO("Overriding slot to '%s'\n", Slot.c_str());
3217 Active_Slot_Display = Slot;
3218 DataManager::SetValue("tw_active_slot", Slot);
nebrassy07c4b902021-11-21 03:32:47 +01003219 PartitionManager.Update_System_Details();
bigbiff74a46272021-07-31 19:02:26 -04003220}
3221
Ethan Yonker1b190162016-12-05 15:25:19 -06003222void TWPartitionManager::Set_Active_Slot(const string& Slot) {
3223 if (Slot != "A" && Slot != "B") {
3224 LOGERR("Set_Active_Slot invalid slot '%s'\n", Slot.c_str());
3225 return;
3226 }
3227 if (Active_Slot_Display == Slot)
3228 return;
3229 LOGINFO("Setting active slot %s\n", Slot.c_str());
3230#ifdef AB_OTA_UPDATER
3231 if (!Active_Slot_Display.empty()) {
dianlujitaoa90dc462021-12-16 16:56:26 +08003232 android::sp<IBootControl> module = IBootControl::getService();
3233 if (module == nullptr) {
Ethan Yonker1b190162016-12-05 15:25:19 -06003234 LOGERR("Error getting bootctrl module.\n");
3235 } else {
dianlujitaoa90dc462021-12-16 16:56:26 +08003236 uint32_t slot_number = 0;
Ethan Yonker1b190162016-12-05 15:25:19 -06003237 if (Slot == "B")
3238 slot_number = 1;
dianlujitaoa90dc462021-12-16 16:56:26 +08003239 CommandResult result;
3240 auto ret = module->setActiveBootSlot(slot_number, [&result]
3241 (const CommandResult &cb_result) { result = cb_result; });
3242 if (!ret.isOk() || !result.success)
Ethan Yonker1b190162016-12-05 15:25:19 -06003243 gui_msg(Msg(msg::kError, "unable_set_boot_slot=Error changing bootloader boot slot to {1}")(Slot));
3244 }
3245 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
3246 }
3247#else
3248 LOGERR("Boot slot feature not present\n");
3249#endif
3250 Active_Slot_Display = Slot;
3251 if (Fstab_Processed())
3252 Update_System_Details();
3253}
3254string TWPartitionManager::Get_Active_Slot_Suffix() {
3255 if (Active_Slot_Display == "A")
3256 return "_a";
3257 return "_b";
3258}
3259string TWPartitionManager::Get_Active_Slot_Display() {
3260 return Active_Slot_Display;
3261}
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003262
Captain Throwback9d6feb52018-07-27 10:05:24 -04003263string TWPartitionManager::Get_Android_Root_Path() {
bigbiff8da46fa2020-09-08 16:42:34 -04003264 return "/system_root";
Captain Throwback9d6feb52018-07-27 10:05:24 -04003265}
3266
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003267void TWPartitionManager::Remove_Uevent_Devices(const string& Mount_Point) {
3268 std::vector<TWPartition*>::iterator iter;
3269
3270 for (iter = Partitions.begin(); iter != Partitions.end(); ) {
3271 if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Mount_Point) {
3272 TWPartition *part = *iter;
3273 LOGINFO("%s was removed by uevent data\n", (*iter)->Mount_Point.c_str());
3274 (*iter)->UnMount(false);
3275 rmdir((*iter)->Mount_Point.c_str());
3276 iter = Partitions.erase(iter);
3277 delete part;
3278 } else {
3279 iter++;
3280 }
3281 }
3282}
3283
3284void TWPartitionManager::Handle_Uevent(const Uevent_Block_Data& uevent_data) {
3285 std::vector<TWPartition*>::iterator iter;
3286
3287 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3288 if (!(*iter)->Sysfs_Entry.empty()) {
3289 string device;
3290 size_t wildcard = (*iter)->Sysfs_Entry.find("*");
3291 if (wildcard != string::npos) {
3292 device = (*iter)->Sysfs_Entry.substr(0, wildcard);
3293 } else {
3294 device = (*iter)->Sysfs_Entry;
3295 }
3296 if (device == uevent_data.sysfs_path.substr(0, device.size())) {
3297 // Found a match
3298 if (uevent_data.action == "add") {
3299 (*iter)->Primary_Block_Device = "/dev/block/" + uevent_data.block_device;
3300 (*iter)->Alternate_Block_Device = (*iter)->Primary_Block_Device;
3301 (*iter)->Is_Present = true;
3302 LOGINFO("Found a match '%s' '%s'\n", uevent_data.block_device.c_str(), device.c_str());
3303 if (!Decrypt_Adopted()) {
3304 LOGINFO("No adopted storage so finding actual block device\n");
3305 (*iter)->Find_Actual_Block_Device();
3306 }
3307 return;
3308 } else if (uevent_data.action == "remove") {
3309 (*iter)->Is_Present = false;
3310 (*iter)->Primary_Block_Device = "";
3311 (*iter)->Actual_Block_Device = "";
3312 Remove_Uevent_Devices((*iter)->Mount_Point);
3313 return;
3314 }
3315 }
3316 }
3317 }
bigbiffee7b7ff2020-03-23 15:08:27 -04003318
3319 if (!PartitionManager.Get_Super_Status())
3320 LOGINFO("Found no matching fstab entry for uevent device '%s' - %s\n", uevent_data.sysfs_path.c_str(), uevent_data.action.c_str());
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003321}
3322
3323void TWPartitionManager::setup_uevent() {
3324 struct sockaddr_nl nls;
3325
3326 if (uevent_pfd.fd >= 0) {
3327 LOGINFO("uevent already set up\n");
3328 return;
3329 }
3330
3331 // Open hotplug event netlink socket
3332 memset(&nls,0,sizeof(struct sockaddr_nl));
3333 nls.nl_family = AF_NETLINK;
3334 nls.nl_pid = getpid();
3335 nls.nl_groups = -1;
3336 uevent_pfd.events = POLLIN;
3337 uevent_pfd.fd = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_KOBJECT_UEVENT);
3338 if (uevent_pfd.fd==-1) {
3339 LOGERR("uevent not root\n");
3340 return;
3341 }
3342
3343 // Listen to netlink socket
3344 if (::bind(uevent_pfd.fd, (struct sockaddr *) &nls, sizeof(struct sockaddr_nl)) < 0) {
3345 LOGERR("Bind failed\n");
3346 return;
3347 }
3348 set_select_fd();
3349 Coldboot();
3350}
3351
3352Uevent_Block_Data TWPartitionManager::get_event_block_values(char *buf, int len) {
3353 Uevent_Block_Data ret;
3354 ret.subsystem = "";
3355 char *ptr = buf;
3356 const char *end = buf + len;
3357
3358 buf[len - 1] = '\0';
3359 while (ptr < end) {
3360 if (strncmp(ptr, "ACTION=", strlen("ACTION=")) == 0) {
3361 ptr += strlen("ACTION=");
3362 ret.action = ptr;
3363 } else if (strncmp(ptr, "SUBSYSTEM=", strlen("SUBSYSTEM=")) == 0) {
3364 ptr += strlen("SUBSYSTEM=");
3365 ret.subsystem = ptr;
3366 } else if (strncmp(ptr, "DEVTYPE=", strlen("DEVTYPE=")) == 0) {
3367 ptr += strlen("DEVTYPE=");
3368 ret.type = ptr;
3369 } else if (strncmp(ptr, "DEVPATH=", strlen("DEVPATH=")) == 0) {
3370 ptr += strlen("DEVPATH=");
3371 ret.sysfs_path += ptr;
3372 } else if (strncmp(ptr, "DEVNAME=", strlen("DEVNAME=")) == 0) {
3373 ptr += strlen("DEVNAME=");
3374 ret.block_device += ptr;
3375 } else if (strncmp(ptr, "MAJOR=", strlen("MAJOR=")) == 0) {
3376 ptr += strlen("MAJOR=");
3377 ret.major = atoi(ptr);
3378 } else if (strncmp(ptr, "MINOR=", strlen("MINOR=")) == 0) {
3379 ptr += strlen("MINOR=");
3380 ret.minor = atoi(ptr);
3381 }
3382 ptr += strlen(ptr) + 1;
3383 }
3384 return ret;
3385}
3386
3387void TWPartitionManager::read_uevent() {
3388 char buf[1024];
3389
3390 int len = recv(uevent_pfd.fd, buf, sizeof(buf), MSG_DONTWAIT);
3391 if (len == -1) {
Mauronofrio Matarresec1bb76e2019-08-04 17:02:41 +01003392 LOGINFO("recv error on uevent\n");
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06003393 return;
3394 }
3395 /*int i = 0; // Print all uevent output for test /debug
3396 while (i<len) {
3397 printf("%s\n", buf+i);
3398 i += strlen(buf+i)+1;
3399 }*/
3400 Uevent_Block_Data uevent_data = get_event_block_values(buf, len);
3401 if (uevent_data.subsystem == "block" && uevent_data.type == "disk") {
3402 PartitionManager.Handle_Uevent(uevent_data);
3403 }
3404}
3405
3406void TWPartitionManager::close_uevent() {
3407 if (uevent_pfd.fd > 0)
3408 close(uevent_pfd.fd);
3409 uevent_pfd.fd = -1;
3410}
3411
3412void TWPartitionManager::Add_Partition(TWPartition* Part) {
3413 Partitions.push_back(Part);
3414}
3415
3416void TWPartitionManager::Coldboot_Scan(std::vector<string> *sysfs_entries, const string& Path, int depth) {
3417 string Real_Path = Path;
3418 char real_path[PATH_MAX];
3419 if (realpath(Path.c_str(), &real_path[0])) {
3420 string Real_Path = real_path;
3421 std::vector<string>::iterator iter;
3422 for (iter = sysfs_entries->begin(); iter != sysfs_entries->end(); iter++) {
3423 if (Real_Path.find((*iter)) != string::npos) {
3424 string Write_Path = Real_Path + "/uevent";
3425 if (TWFunc::Path_Exists(Write_Path)) {
3426 const char* write_val = "add\n";
3427 TWFunc::write_to_file(Write_Path, write_val);
3428 break;
3429 }
3430 }
3431 }
3432 }
3433
3434 DIR* d = opendir(Path.c_str());
3435 if (d != NULL) {
3436 struct dirent* de;
3437 while ((de = readdir(d)) != NULL) {
3438 if (de->d_name[0] == '.' || (de->d_type != DT_DIR && depth > 0))
3439 continue;
3440 if (strlen(de->d_name) >= 4 && (strncmp(de->d_name, "ram", 3) == 0 || strncmp(de->d_name, "loop", 4) == 0))
3441 continue;
3442
3443 string item = Path + "/";
3444 item.append(de->d_name);
3445 Coldboot_Scan(sysfs_entries, item, depth + 1);
3446 }
3447 closedir(d);
3448 }
3449}
3450
3451void TWPartitionManager::Coldboot() {
3452 std::vector<TWPartition*>::iterator iter;
3453 std::vector<string> sysfs_entries;
3454
3455 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3456 if (!(*iter)->Sysfs_Entry.empty()) {
3457 size_t wildcard_pos = (*iter)->Sysfs_Entry.find("*");
3458 if (wildcard_pos == string::npos)
3459 wildcard_pos = (*iter)->Sysfs_Entry.size();
3460 sysfs_entries.push_back((*iter)->Sysfs_Entry.substr(0, wildcard_pos));
3461 }
3462 }
3463
3464 if (sysfs_entries.size() > 0)
3465 Coldboot_Scan(&sysfs_entries, "/sys/block", 0);
3466}
Ethan Yonker53796e72019-01-11 22:49:52 -06003467
3468bool TWPartitionManager::Prepare_Empty_Folder(const std::string& Folder) {
3469 if (TWFunc::Path_Exists(Folder))
3470 TWFunc::removeDir(Folder, false);
3471 return TWFunc::Recursive_Mkdir(Folder);
3472}
3473
bigbiffcfa875c2021-06-20 16:20:27 -04003474std::string TWPartitionManager::Get_Bare_Partition_Name(std::string Mount_Point) {
3475 if (Mount_Point == "/system_root")
3476 return "system";
3477 else
3478 return TWFunc::Remove_Beginning_Slash(Mount_Point);
3479}
3480
bigbiffee7b7ff2020-03-23 15:08:27 -04003481bool TWPartitionManager::Prepare_Super_Volume(TWPartition* twrpPart) {
3482 Fstab fstab;
bigbiffcfa875c2021-06-20 16:20:27 -04003483 std::string bare_partition_name = Get_Bare_Partition_Name(twrpPart->Get_Mount_Point());
bigbiffee7b7ff2020-03-23 15:08:27 -04003484
bigbiff998f8392021-08-06 19:19:33 -04003485 Super_Partition_List.push_back(bare_partition_name);
bigbiffee7b7ff2020-03-23 15:08:27 -04003486 LOGINFO("Trying to prepare %s from super partition\n", bare_partition_name.c_str());
3487
3488 std::string blk_device_partition;
3489#ifdef AB_OTA_UPDATER
3490 blk_device_partition = bare_partition_name + PartitionManager.Get_Active_Slot_Suffix();
3491#else
3492 blk_device_partition = bare_partition_name;
3493#endif
3494
3495 FstabEntry fstabEntry = {
3496 .blk_device = blk_device_partition,
3497 .mount_point = twrpPart->Get_Mount_Point(),
3498 .fs_type = twrpPart->Current_File_System,
3499 .fs_mgr_flags.logical = twrpPart->Is_Super,
3500 };
3501
3502 fstab.emplace_back(fstabEntry);
3503 if (!fs_mgr_update_logical_partition(&fstabEntry)) {
Captain Throwbackc6939102021-04-06 12:57:30 -04003504 LOGINFO("unable to update logical partition: %s\n", twrpPart->Get_Mount_Point().c_str());
bigbiffee7b7ff2020-03-23 15:08:27 -04003505 return false;
3506 }
3507
bigbiff32cbabe2020-04-12 19:16:19 -04003508 while (access(fstabEntry.blk_device.c_str(), F_OK) != 0) {
3509 usleep(100);
3510 }
3511
bigbiffee7b7ff2020-03-23 15:08:27 -04003512 twrpPart->Set_Block_Device(fstabEntry.blk_device);
3513 twrpPart->Update_Size(true);
bigbiffee7b7ff2020-03-23 15:08:27 -04003514 twrpPart->Set_Can_Be_Backed_Up(false);
3515 twrpPart->Set_Can_Be_Wiped(false);
me-cafebabe928234b2022-07-28 04:07:04 +08003516 if (access(("/dev/block/bootdevice/by-name/" + bare_partition_name).c_str(), F_OK) == -1) {
3517 LOGINFO("Symlinking %s => /dev/block/bootdevice/by-name/%s \n", fstabEntry.blk_device.c_str(), bare_partition_name.c_str());
3518 symlink(fstabEntry.blk_device.c_str(), ("/dev/block/bootdevice/by-name/" + bare_partition_name).c_str());
Captain Throwback88c6e652023-01-16 15:32:26 -05003519 property_set("twrp.super.symlinks_created", "true");
me-cafebabe928234b2022-07-28 04:07:04 +08003520 }
Mohd Faraz7fc70502021-04-22 20:44:11 +02003521
bigbiffee7b7ff2020-03-23 15:08:27 -04003522 return true;
3523}
3524
3525bool TWPartitionManager::Prepare_All_Super_Volumes() {
3526 bool status = true;
3527 std::vector<TWPartition*>::iterator iter;
3528
3529 for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
3530 if ((*iter)->Is_Super) {
3531 if (!Prepare_Super_Volume(*iter)) {
3532 status = false;
Mohd Faraz787cc612023-01-02 22:47:51 +01003533 Partitions.erase(iter--);
bigbiffee7b7ff2020-03-23 15:08:27 -04003534 }
3535 PartitionManager.Output_Partition(*iter);
3536 }
3537 }
3538 Update_System_Details();
3539 return status;
3540}
3541
bigbiffee7b7ff2020-03-23 15:08:27 -04003542std::string TWPartitionManager::Get_Super_Partition() {
3543 int slot_number = Get_Active_Slot_Display() == "A" ? 0 : 1;
bigbiff998f8392021-08-06 19:19:33 -04003544 std::string super_device = fs_mgr_get_super_partition_name(slot_number);
bigbiffee7b7ff2020-03-23 15:08:27 -04003545 return "/dev/block/by-name/" + super_device;
3546}
3547
3548void TWPartitionManager::Setup_Super_Devices() {
3549 std::string superPart = Get_Super_Partition();
3550 android::fs_mgr::CreateLogicalPartitions(superPart);
bigbiff7ba75002020-04-11 20:47:09 -04003551}
3552
3553void TWPartitionManager::Setup_Super_Partition() {
bigbiffee7b7ff2020-03-23 15:08:27 -04003554 TWPartition* superPartition = new TWPartition();
bigbiff7ba75002020-04-11 20:47:09 -04003555 std::string superPart = Get_Super_Partition();
3556
3557 superPartition->Backup_Path = "/super";
3558 superPartition->Mount_Point = "/super";
bigbiffee7b7ff2020-03-23 15:08:27 -04003559 superPartition->Actual_Block_Device = superPart;
3560 superPartition->Alternate_Block_Device = superPart;
bigbiff998f8392021-08-06 19:19:33 -04003561 superPartition->Backup_Display_Name = "Super (";
3562 // Add first 4 items to fstab as logical that you would like to display in Backup_Display_Name
3563 // for the Super partition
3564 int list_size = Super_Partition_List.size();
3565 int orig_list_size = list_size;
3566 int max_display_size = 3; // total of 4 items since we start at 0
3567
3568 for (auto partition: Super_Partition_List) {
3569 superPartition->Backup_Display_Name = superPartition->Backup_Display_Name + partition;
3570 if ((orig_list_size - list_size) == max_display_size) {
3571 break;
3572 }
3573 if (list_size != 1)
3574 superPartition->Backup_Display_Name = superPartition->Backup_Display_Name + " ";
3575 list_size--;
3576 }
3577 superPartition->Backup_Display_Name += ")";
bigbiffee7b7ff2020-03-23 15:08:27 -04003578 superPartition->Can_Flash_Img = true;
bigbiffee7b7ff2020-03-23 15:08:27 -04003579 superPartition->Current_File_System = "emmc";
3580 superPartition->Can_Be_Backed_Up = true;
3581 superPartition->Is_Present = true;
3582 superPartition->Is_SubPartition = false;
bigbiff7ba75002020-04-11 20:47:09 -04003583 superPartition->Setup_Image();
bigbiffee7b7ff2020-03-23 15:08:27 -04003584 Add_Partition(superPartition);
3585 PartitionManager.Output_Partition(superPartition);
bigbiffee7b7ff2020-03-23 15:08:27 -04003586}
3587
3588bool TWPartitionManager::Get_Super_Status() {
Mohd Faraz88b4bab2023-01-01 02:43:01 +05303589 return access(Get_Super_Partition().c_str(), F_OK) == 0;
bigbiffee7b7ff2020-03-23 15:08:27 -04003590}
bigbiff25d25b92020-06-19 16:07:38 -04003591
3592bool TWPartitionManager::Recreate_Logs_Dir() {
3593#ifdef TW_INCLUDE_FBE
3594 struct passwd pd;
3595 struct passwd *pwdptr = &pd;
3596 struct passwd *tempPd;
3597 char pwdBuf[512];
3598 int uid = 0, gid = 0;
3599
3600 if ((getpwnam_r("system", pwdptr, pwdBuf, sizeof(pwdBuf), &tempPd)) != 0) {
3601 LOGERR("unable to get system user id\n");
3602 return false;
3603 } else {
3604 struct group grp;
3605 struct group *grpptr = &grp;
3606 struct group *tempGrp;
3607 char grpBuf[512];
3608
3609 if ((getgrnam_r("cache", grpptr, grpBuf, sizeof(grpBuf), &tempGrp)) != 0) {
3610 LOGERR("unable to get cache group id\n");
3611 return false;
3612 } else {
3613 uid = pd.pw_uid;
3614 gid = grp.gr_gid;
3615 std::string abLogsRecoveryDir(DATA_LOGS_DIR);
3616 abLogsRecoveryDir += "/recovery/";
3617
3618 if (!TWFunc::Create_Dir_Recursive(abLogsRecoveryDir, S_IRWXU | S_IRWXG | S_IWGRP | S_IXGRP, uid, gid)) {
3619 LOGERR("Unable to recreate %s\n", abLogsRecoveryDir.c_str());
3620 return false;
3621 }
3622 if (setfilecon(abLogsRecoveryDir.c_str(), "u:object_r:cache_file:s0") != 0) {
3623 LOGERR("Unable to set contexts for %s\n", abLogsRecoveryDir.c_str());
3624 return false;
3625 }
3626 }
3627 }
3628#endif
3629 return true;
3630}
bigbiff1f9e4842020-10-31 11:33:15 -04003631
3632void TWPartitionManager::Unlock_Block_Partitions() {
3633 int fd, OFF = 0;
3634
3635 const std::string block_path = "/dev/block/";
3636 DIR* d = opendir(block_path.c_str());
3637 if (d != NULL) {
3638 struct dirent* de;
3639 while ((de = readdir(d)) != NULL) {
3640 if (de->d_type == DT_BLK) {
3641 std::string block_device = block_path + de->d_name;
bigbiff1f9e4842020-10-31 11:33:15 -04003642 if ((fd = open(block_device.c_str(), O_RDONLY | O_CLOEXEC)) < 0) {
3643 LOGERR("unable to open block device %s: %s\n", block_device.c_str(), strerror(errno));
3644 continue;
3645 }
3646 if (ioctl(fd, BLKROSET, &OFF) == -1) {
me-cafebabed63fcaa2022-07-28 03:19:20 +08003647 LOGERR("Unable to unlock %s: %s\n", block_device.c_str());
bigbiff1f9e4842020-10-31 11:33:15 -04003648 continue;
3649 }
3650 close(fd);
3651 }
3652 }
3653 closedir(d);
3654 }
bigbiffaf253cd2020-12-06 16:50:55 -05003655}
bigbiffcfa875c2021-06-20 16:20:27 -04003656
3657bool TWPartitionManager::Unmap_Super_Devices() {
3658 bool destroyed = false;
3659#ifndef TW_EXCLUDE_APEX
3660 twrpApex apex;
3661 apex.Unmount();
3662#endif
3663 for (auto iter = Partitions.begin(); iter != Partitions.end();) {
3664 LOGINFO("Checking partition: %s\n", (*iter)->Get_Mount_Point().c_str());
3665 if ((*iter)->Is_Super) {
3666 TWPartition *part = *iter;
3667 std::string bare_partition_name = Get_Bare_Partition_Name((*iter)->Get_Mount_Point());
me-cafebabe2136fc52022-07-20 06:42:18 +08003668 std::string blk_device_partition = bare_partition_name;
3669 if (DataManager::GetStrValue(TW_VIRTUAL_AB_ENABLED) == "1")
3670 blk_device_partition.append(PartitionManager.Get_Active_Slot_Suffix());
bigbiffcfa875c2021-06-20 16:20:27 -04003671 (*iter)->UnMount(false);
3672 LOGINFO("removing dynamic partition: %s\n", blk_device_partition.c_str());
3673 destroyed = DestroyLogicalPartition(blk_device_partition);
3674 std::string cow_partition = blk_device_partition + "-cow";
3675 std::string cow_partition_path = "/dev/block/mapper/" + cow_partition;
3676 struct stat st;
3677 if (lstat(cow_partition_path.c_str(), &st) == 0) {
3678 LOGINFO("removing cow partition: %s\n", cow_partition.c_str());
3679 destroyed = DestroyLogicalPartition(cow_partition);
3680 }
bigbiffcfa875c2021-06-20 16:20:27 -04003681 iter = Partitions.erase(iter);
3682 delete part;
3683 if (!destroyed) {
3684 return false;
3685 }
3686 } else {
3687 ++iter;
3688 }
3689 }
3690 return true;
3691}
bigbiffd21252f2021-09-18 15:56:32 -04003692
3693
3694bool TWPartitionManager::Check_Pending_Merges() {
3695 auto sm = android::snapshot::SnapshotManager::NewForFirstStageMount();
3696 if (!sm) {
3697 LOGERR("Unable to call snapshot manager\n");
3698 return false;
3699 }
3700
3701 if (!Unmap_Super_Devices()) {
3702 LOGERR("Unable to unmap dynamic partitions.\n");
3703 return false;
3704 }
3705
3706 auto callback = [&]() -> void {
3707 double progress;
3708 sm->GetUpdateState(&progress);
3709 LOGINFO("waiting for merge to complete: %.2f\n", progress);
3710 };
3711
3712 LOGINFO("checking for merges\n");
3713 if (!sm->HandleImminentDataWipe(callback)) {
3714 LOGERR("Unable to check merge status\n");
3715 return false;
3716 }
3717 return true;
Darth98f775972022-08-05 09:39:03 +01003718}