Move TW zip install code to C++ so that it can use the ui->functions. Bring in mincrypt code to fix a crash during signature checking.
728 lines
21 KiB
C++
728 lines
21 KiB
C++
/* Partition Management classes for TWRP
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License version 2 and
|
|
* only version 2 as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
|
* 02110-1301, USA.
|
|
*
|
|
* The code was written from scratch by Dees_Troy dees_troy at
|
|
* yahoo
|
|
*
|
|
* Copyright (c) 2012
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/vfs.h>
|
|
#include <unistd.h>
|
|
#include <vector>
|
|
#include <dirent.h>
|
|
#include <time.h>
|
|
|
|
#include "variables.h"
|
|
#include "common.h"
|
|
#include "partitions.hpp"
|
|
#include "data.hpp"
|
|
|
|
#ifdef TW_INCLUDE_CRYPTO
|
|
#ifdef TW_INCLUDE_JB_CRYPTO
|
|
#include "crypto/jb/cryptfs.h"
|
|
#else
|
|
#include "crypto/ics/cryptfs.h"
|
|
#endif
|
|
#include "cutils/properties.h"
|
|
#endif
|
|
|
|
int TWPartitionManager::Process_Fstab(string Fstab_Filename, bool Display_Error) {
|
|
FILE *fstabFile;
|
|
char fstab_line[MAX_FSTAB_LINE_LENGTH];
|
|
|
|
fstabFile = fopen(Fstab_Filename.c_str(), "rt");
|
|
if (fstabFile == NULL) {
|
|
LOGE("Critical Error: Unable to open fstab at '%s'.\n", Fstab_Filename.c_str());
|
|
return false;
|
|
}
|
|
|
|
while (fgets(fstab_line, sizeof(fstab_line), fstabFile) != NULL) {
|
|
if (fstab_line[0] != '/')
|
|
continue;
|
|
|
|
TWPartition* partition = new TWPartition();
|
|
string line(fstab_line);
|
|
if (partition->Process_Fstab_Line(line, Display_Error)) {
|
|
Partitions.push_back(partition);
|
|
} else {
|
|
delete partition;
|
|
}
|
|
}
|
|
fclose(fstabFile);
|
|
if (!Write_Fstab()) {
|
|
if (Display_Error)
|
|
LOGE("Error creating fstab\n");
|
|
else
|
|
LOGI("Error creating fstab\n");
|
|
}
|
|
Update_System_Details();
|
|
return true;
|
|
}
|
|
|
|
int TWPartitionManager::Write_Fstab(void) {
|
|
FILE *fp;
|
|
std::vector<TWPartition*>::iterator iter;
|
|
string Line;
|
|
|
|
fp = fopen("/etc/fstab", "w");
|
|
if (fp == NULL) {
|
|
LOGI("Can not open /etc/fstab.\n");
|
|
return false;
|
|
}
|
|
for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
|
|
if ((*iter)->Can_Be_Mounted) {
|
|
if ((*iter)->Is_Decrypted)
|
|
Line = (*iter)->Decrypted_Block_Device + " " + (*iter)->Mount_Point + " " + (*iter)->Current_File_System + " rw\n";
|
|
else
|
|
Line = (*iter)->Block_Device + " " + (*iter)->Mount_Point + " " + (*iter)->Current_File_System + " rw\n";
|
|
fputs(Line.c_str(), fp);
|
|
// Handle subpartition tracking
|
|
if ((*iter)->Is_SubPartition) {
|
|
TWPartition* ParentPartition = Find_Partition_By_Path((*iter)->SubPartition_Of);
|
|
if (ParentPartition)
|
|
ParentPartition->Has_SubPartition = true;
|
|
else
|
|
LOGE("Unable to locate parent partition '%s' of '%s'\n", (*iter)->SubPartition_Of.c_str(), (*iter)->Mount_Point.c_str());
|
|
}
|
|
}
|
|
}
|
|
fclose(fp);
|
|
return true;
|
|
}
|
|
|
|
int TWPartitionManager::Mount_By_Path(string Path, bool Display_Error) {
|
|
std::vector<TWPartition*>::iterator iter;
|
|
int ret = false;
|
|
bool found = false;
|
|
string Local_Path = Get_Root_Path(Path);
|
|
|
|
// Iterate through all partitions
|
|
for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
|
|
if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
|
|
ret = (*iter)->Mount(Display_Error);
|
|
found = true;
|
|
} else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
|
|
(*iter)->Mount(Display_Error);
|
|
}
|
|
}
|
|
if (found) {
|
|
return ret;
|
|
} else if (Display_Error) {
|
|
LOGE("Mount: Unable to find partition for path '%s'\n", Local_Path.c_str());
|
|
} else {
|
|
LOGI("Mount: Unable to find partition for path '%s'\n", Local_Path.c_str());
|
|
}
|
|
return false;
|
|
}
|
|
|
|
int TWPartitionManager::Mount_By_Block(string Block, bool Display_Error) {
|
|
TWPartition* Part = Find_Partition_By_Block(Block);
|
|
|
|
if (Part) {
|
|
if (Part->Has_SubPartition) {
|
|
std::vector<TWPartition*>::iterator subpart;
|
|
|
|
for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
|
|
if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == Part->Mount_Point)
|
|
(*subpart)->Mount(Display_Error);
|
|
}
|
|
return Part->Mount(Display_Error);
|
|
} else
|
|
return Part->Mount(Display_Error);
|
|
}
|
|
if (Display_Error)
|
|
LOGE("Mount: Unable to find partition for block '%s'\n", Block.c_str());
|
|
else
|
|
LOGI("Mount: Unable to find partition for block '%s'\n", Block.c_str());
|
|
return false;
|
|
}
|
|
|
|
int TWPartitionManager::Mount_By_Name(string Name, bool Display_Error) {
|
|
TWPartition* Part = Find_Partition_By_Name(Name);
|
|
|
|
if (Part) {
|
|
if (Part->Has_SubPartition) {
|
|
std::vector<TWPartition*>::iterator subpart;
|
|
|
|
for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
|
|
if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == Part->Mount_Point)
|
|
(*subpart)->Mount(Display_Error);
|
|
}
|
|
return Part->Mount(Display_Error);
|
|
} else
|
|
return Part->Mount(Display_Error);
|
|
}
|
|
if (Display_Error)
|
|
LOGE("Mount: Unable to find partition for name '%s'\n", Name.c_str());
|
|
else
|
|
LOGI("Mount: Unable to find partition for name '%s'\n", Name.c_str());
|
|
return false;
|
|
}
|
|
|
|
int TWPartitionManager::UnMount_By_Path(string Path, bool Display_Error) {
|
|
std::vector<TWPartition*>::iterator iter;
|
|
int ret = false;
|
|
bool found = false;
|
|
string Local_Path = Get_Root_Path(Path);
|
|
|
|
// Iterate through all partitions
|
|
for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
|
|
if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
|
|
ret = (*iter)->UnMount(Display_Error);
|
|
found = true;
|
|
} else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
|
|
(*iter)->UnMount(Display_Error);
|
|
}
|
|
}
|
|
if (found) {
|
|
return ret;
|
|
} else if (Display_Error) {
|
|
LOGE("UnMount: Unable to find partition for path '%s'\n", Local_Path.c_str());
|
|
} else {
|
|
LOGI("UnMount: Unable to find partition for path '%s'\n", Local_Path.c_str());
|
|
}
|
|
return false;
|
|
}
|
|
|
|
int TWPartitionManager::UnMount_By_Block(string Block, bool Display_Error) {
|
|
TWPartition* Part = Find_Partition_By_Block(Block);
|
|
|
|
if (Part) {
|
|
if (Part->Has_SubPartition) {
|
|
std::vector<TWPartition*>::iterator subpart;
|
|
|
|
for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
|
|
if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == Part->Mount_Point)
|
|
(*subpart)->UnMount(Display_Error);
|
|
}
|
|
return Part->UnMount(Display_Error);
|
|
} else
|
|
return Part->UnMount(Display_Error);
|
|
}
|
|
if (Display_Error)
|
|
LOGE("UnMount: Unable to find partition for block '%s'\n", Block.c_str());
|
|
else
|
|
LOGI("UnMount: Unable to find partition for block '%s'\n", Block.c_str());
|
|
return false;
|
|
}
|
|
|
|
int TWPartitionManager::UnMount_By_Name(string Name, bool Display_Error) {
|
|
TWPartition* Part = Find_Partition_By_Name(Name);
|
|
|
|
if (Part) {
|
|
if (Part->Has_SubPartition) {
|
|
std::vector<TWPartition*>::iterator subpart;
|
|
|
|
for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
|
|
if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == Part->Mount_Point)
|
|
(*subpart)->UnMount(Display_Error);
|
|
}
|
|
return Part->UnMount(Display_Error);
|
|
} else
|
|
return Part->UnMount(Display_Error);
|
|
}
|
|
if (Display_Error)
|
|
LOGE("UnMount: Unable to find partition for name '%s'\n", Name.c_str());
|
|
else
|
|
LOGI("UnMount: Unable to find partition for name '%s'\n", Name.c_str());
|
|
return false;
|
|
}
|
|
|
|
int TWPartitionManager::Is_Mounted_By_Path(string Path) {
|
|
TWPartition* Part = Find_Partition_By_Path(Path);
|
|
|
|
if (Part)
|
|
return Part->Is_Mounted();
|
|
else
|
|
LOGI("Is_Mounted: Unable to find partition for path '%s'\n", Path.c_str());
|
|
return false;
|
|
}
|
|
|
|
int TWPartitionManager::Is_Mounted_By_Block(string Block) {
|
|
TWPartition* Part = Find_Partition_By_Block(Block);
|
|
|
|
if (Part)
|
|
return Part->Is_Mounted();
|
|
else
|
|
LOGI("Is_Mounted: Unable to find partition for block '%s'\n", Block.c_str());
|
|
return false;
|
|
}
|
|
|
|
int TWPartitionManager::Is_Mounted_By_Name(string Name) {
|
|
TWPartition* Part = Find_Partition_By_Name(Name);
|
|
|
|
if (Part)
|
|
return Part->Is_Mounted();
|
|
else
|
|
LOGI("Is_Mounted: Unable to find partition for name '%s'\n", Name.c_str());
|
|
return false;
|
|
}
|
|
|
|
int TWPartitionManager::Mount_Current_Storage(bool Display_Error) {
|
|
string current_storage_path = DataManager::GetCurrentStoragePath();
|
|
|
|
if (Mount_By_Path(current_storage_path, Display_Error)) {
|
|
TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
|
|
if (FreeStorage)
|
|
DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
int TWPartitionManager::Mount_Settings_Storage(bool Display_Error) {
|
|
return Mount_By_Path(DataManager::GetSettingsStoragePath(), Display_Error);
|
|
}
|
|
|
|
TWPartition* TWPartitionManager::Find_Partition_By_Path(string Path) {
|
|
std::vector<TWPartition*>::iterator iter;
|
|
string Local_Path = Get_Root_Path(Path);
|
|
|
|
for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
|
|
if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path))
|
|
return (*iter);
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
TWPartition* TWPartitionManager::Find_Partition_By_Block(string Block) {
|
|
std::vector<TWPartition*>::iterator iter;
|
|
|
|
for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
|
|
if ((*iter)->Block_Device == Block || (*iter)->Alternate_Block_Device == Block || ((*iter)->Is_Decrypted && (*iter)->Decrypted_Block_Device == Block))
|
|
return (*iter);
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
TWPartition* TWPartitionManager::Find_Partition_By_Name(string Name) {
|
|
std::vector<TWPartition*>::iterator iter;
|
|
|
|
for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
|
|
if ((*iter)->Display_Name == Name)
|
|
return (*iter);
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
int TWPartitionManager::Run_Backup(string Backup_Name) {
|
|
LOGI("STUB TWPartitionManager::Run_Backup, Backup_Name: '%s'\n", Backup_Name.c_str());
|
|
return 1;
|
|
}
|
|
|
|
int TWPartitionManager::Run_Restore(string Restore_Name) {
|
|
int check;
|
|
TWPartition* Part;
|
|
|
|
DataManager::GetValue(TW_RESTORE_SYSTEM_VAR, check);
|
|
if (check > 0) {
|
|
Part = Find_Partition_By_Path("/system");
|
|
if (Part) {
|
|
if (!Part->Restore(Restore_Name))
|
|
return false;
|
|
} else
|
|
LOGE("Restore: Unable to locate system partition.\n");
|
|
}
|
|
DataManager::GetValue(TW_RESTORE_DATA_VAR, check);
|
|
if (check > 0) {
|
|
Part = Find_Partition_By_Path("/data");
|
|
if (Part) {
|
|
if (!Part->Restore(Restore_Name))
|
|
return false;
|
|
} else
|
|
LOGE("Restore: Unable to locate data partition.\n");
|
|
}
|
|
DataManager::GetValue(TW_RESTORE_CACHE_VAR, check);
|
|
if (check > 0) {
|
|
Part = Find_Partition_By_Path("/cache");
|
|
if (Part) {
|
|
if (!Part->Restore(Restore_Name))
|
|
return false;
|
|
} else
|
|
LOGE("Restore: Unable to locate cache partition.\n");
|
|
}
|
|
DataManager::GetValue(TW_RESTORE_BOOT_VAR, check);
|
|
if (check > 0) {
|
|
Part = Find_Partition_By_Path("/boot");
|
|
if (Part) {
|
|
if (!Part->Restore(Restore_Name))
|
|
return false;
|
|
} else
|
|
LOGE("Restore: Unable to locate boot partition.\n");
|
|
}
|
|
DataManager::GetValue(TW_RESTORE_ANDSEC_VAR, check);
|
|
if (check > 0) {
|
|
Part = Find_Partition_By_Path("/.android_secure");
|
|
if (Part) {
|
|
if (!Part->Restore(Restore_Name))
|
|
return false;
|
|
} else
|
|
LOGE("Restore: Unable to locate android_secure partition.\n");
|
|
}
|
|
DataManager::GetValue(TW_RESTORE_SDEXT_VAR, check);
|
|
if (check > 0) {
|
|
Part = Find_Partition_By_Path("/sd-ext");
|
|
if (Part) {
|
|
if (!Part->Restore(Restore_Name))
|
|
return false;
|
|
} else
|
|
LOGE("Restore: Unable to locate sd-ext partition.\n");
|
|
}
|
|
#ifdef SP1_NAME
|
|
DataManager::GetValue(TW_RESTORE_SP1_VAR, check);
|
|
if (check > 0) {
|
|
Part = Find_Partition_By_Path(Get_Root_Path(SP1_NAME));
|
|
if (Part) {
|
|
if (!Part->Restore(Restore_Name))
|
|
return false;
|
|
} else
|
|
LOGE("Restore: Unable to locate %s partition.\n", SP1_NAME);
|
|
}
|
|
#endif
|
|
#ifdef SP2_NAME
|
|
DataManager::GetValue(TW_RESTORE_SP2_VAR, check);
|
|
if (check > 0) {
|
|
Part = Find_Partition_By_Path(Get_Root_Path(SP2_NAME));
|
|
if (Part) {
|
|
if (!Part->Restore(Restore_Name))
|
|
return false;
|
|
} else
|
|
LOGE("Restore: Unable to locate %s partition.\n", SP2_NAME);
|
|
}
|
|
#endif
|
|
#ifdef SP3_NAME
|
|
DataManager::GetValue(TW_RESTORE_SP3_VAR, check);
|
|
if (check > 0) {
|
|
Part = Find_Partition_By_Path(Get_Root_Path(SP3_NAME));
|
|
if (Part) {
|
|
if (!Part->Restore(Restore_Name))
|
|
return false;
|
|
} else
|
|
LOGE("Restore: Unable to locate %s partition.\n", SP3_NAME);
|
|
}
|
|
#endif
|
|
return true;
|
|
}
|
|
|
|
void TWPartitionManager::Set_Restore_Files(string Restore_Name) {
|
|
// Start with the default values
|
|
int tw_restore_system = -1;
|
|
int tw_restore_data = -1;
|
|
int tw_restore_cache = -1;
|
|
int tw_restore_recovery = -1;
|
|
int tw_restore_boot = -1;
|
|
int tw_restore_andsec = -1;
|
|
int tw_restore_sdext = -1;
|
|
int tw_restore_sp1 = -1;
|
|
int tw_restore_sp2 = -1;
|
|
int tw_restore_sp3 = -1;
|
|
bool get_date = true;
|
|
|
|
DIR* d;
|
|
d = opendir(Restore_Name.c_str());
|
|
if (d == NULL)
|
|
{
|
|
LOGE("Error opening %s\n", Restore_Name.c_str());
|
|
return;
|
|
}
|
|
|
|
struct dirent* de;
|
|
while ((de = readdir(d)) != NULL)
|
|
{
|
|
// Strip off three components
|
|
char str[256];
|
|
char* label;
|
|
char* fstype = NULL;
|
|
char* extn = NULL;
|
|
char* ptr;
|
|
|
|
strcpy(str, de->d_name);
|
|
if (strlen(str) <= 2)
|
|
continue;
|
|
|
|
if (get_date) {
|
|
char file_path[255];
|
|
struct stat st;
|
|
|
|
strcpy(file_path, Restore_Name.c_str());
|
|
strcat(file_path, "/");
|
|
strcat(file_path, str);
|
|
stat(file_path, &st);
|
|
string backup_date = ctime((const time_t*)(&st.st_mtime));
|
|
DataManager::SetValue(TW_RESTORE_FILE_DATE, backup_date);
|
|
get_date = false;
|
|
}
|
|
|
|
label = str;
|
|
ptr = label;
|
|
while (*ptr && *ptr != '.') ptr++;
|
|
if (*ptr == '.')
|
|
{
|
|
*ptr = 0x00;
|
|
ptr++;
|
|
fstype = ptr;
|
|
}
|
|
while (*ptr && *ptr != '.') ptr++;
|
|
if (*ptr == '.')
|
|
{
|
|
*ptr = 0x00;
|
|
ptr++;
|
|
extn = ptr;
|
|
}
|
|
|
|
if (extn == NULL || (strlen(extn) >= 3 && strncmp(extn, "win", 3) != 0)) continue;
|
|
|
|
TWPartition* Part = Find_Partition_By_Path(label);
|
|
if (Part == NULL)
|
|
{
|
|
LOGE(" Unable to locate partition by backup name: '%s'\n", label);
|
|
continue;
|
|
}
|
|
|
|
Part->Backup_FileName = de->d_name;
|
|
if (strlen(extn) > 3) {
|
|
Part->Backup_FileName.resize(Part->Backup_FileName.size() - strlen(extn) + 3);
|
|
}
|
|
|
|
// Now, we just need to find the correct label
|
|
if (Part->Mount_Point == "/system")
|
|
tw_restore_system = 1;
|
|
if (Part->Mount_Point == "/data")
|
|
tw_restore_data = 1;
|
|
if (Part->Mount_Point == "/cache")
|
|
tw_restore_cache = 1;
|
|
if (Part->Mount_Point == "/recovery")
|
|
tw_restore_recovery = 1;
|
|
if (Part->Mount_Point == "/boot")
|
|
tw_restore_boot = 1;
|
|
if (Part->Mount_Point == "/.android_secure")
|
|
tw_restore_andsec = 1;
|
|
if (Part->Mount_Point == "/sd-ext")
|
|
tw_restore_sdext = 1;
|
|
#ifdef SP1_NAME
|
|
if (Part->Mount_Point == Get_Root_Path(SP1_Name))
|
|
tw_restore_sp1 = 1;
|
|
#endif
|
|
#ifdef SP2_NAME
|
|
if (Part->Mount_Point == Get_Root_Path(SP2_Name))
|
|
tw_restore_sp2 = 1;
|
|
#endif
|
|
#ifdef SP3_NAME
|
|
if (Part->Mount_Point == Get_Root_Path(SP3_Name))
|
|
tw_restore_sp3 = 1;
|
|
#endif
|
|
}
|
|
closedir(d);
|
|
|
|
// Set the final values
|
|
DataManager::SetValue(TW_RESTORE_SYSTEM_VAR, tw_restore_system);
|
|
DataManager::SetValue(TW_RESTORE_DATA_VAR, tw_restore_data);
|
|
DataManager::SetValue(TW_RESTORE_CACHE_VAR, tw_restore_cache);
|
|
DataManager::SetValue(TW_RESTORE_RECOVERY_VAR, tw_restore_recovery);
|
|
DataManager::SetValue(TW_RESTORE_BOOT_VAR, tw_restore_boot);
|
|
DataManager::SetValue(TW_RESTORE_ANDSEC_VAR, tw_restore_andsec);
|
|
DataManager::SetValue(TW_RESTORE_SDEXT_VAR, tw_restore_sdext);
|
|
DataManager::SetValue(TW_RESTORE_SP1_VAR, tw_restore_sp1);
|
|
DataManager::SetValue(TW_RESTORE_SP2_VAR, tw_restore_sp2);
|
|
DataManager::SetValue(TW_RESTORE_SP3_VAR, tw_restore_sp3);
|
|
|
|
return;
|
|
}
|
|
|
|
int TWPartitionManager::Wipe_By_Path(string Path) {
|
|
std::vector<TWPartition*>::iterator iter;
|
|
int ret = false;
|
|
bool found = false;
|
|
string Local_Path = Get_Root_Path(Path);
|
|
|
|
// Iterate through all partitions
|
|
for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
|
|
if ((*iter)->Mount_Point == Local_Path || (!(*iter)->Symlink_Mount_Point.empty() && (*iter)->Symlink_Mount_Point == Local_Path)) {
|
|
ret = (*iter)->Wipe();
|
|
found = true;
|
|
} else if ((*iter)->Is_SubPartition && (*iter)->SubPartition_Of == Local_Path) {
|
|
(*iter)->Wipe();
|
|
}
|
|
}
|
|
if (found) {
|
|
return ret;
|
|
} else
|
|
LOGE("Wipe: Unable to find partition for path '%s'\n", Local_Path.c_str());
|
|
return false;
|
|
}
|
|
|
|
int TWPartitionManager::Wipe_By_Block(string Block) {
|
|
TWPartition* Part = Find_Partition_By_Block(Block);
|
|
|
|
if (Part) {
|
|
if (Part->Has_SubPartition) {
|
|
std::vector<TWPartition*>::iterator subpart;
|
|
|
|
for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
|
|
if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == Part->Mount_Point)
|
|
(*subpart)->Wipe();
|
|
}
|
|
return Part->Wipe();
|
|
} else
|
|
return Part->Wipe();
|
|
}
|
|
LOGE("Wipe: Unable to find partition for block '%s'\n", Block.c_str());
|
|
return false;
|
|
}
|
|
|
|
int TWPartitionManager::Wipe_By_Name(string Name) {
|
|
TWPartition* Part = Find_Partition_By_Name(Name);
|
|
|
|
if (Part) {
|
|
if (Part->Has_SubPartition) {
|
|
std::vector<TWPartition*>::iterator subpart;
|
|
|
|
for (subpart = Partitions.begin(); subpart != Partitions.end(); subpart++) {
|
|
if ((*subpart)->Is_SubPartition && (*subpart)->SubPartition_Of == Part->Mount_Point)
|
|
(*subpart)->Wipe();
|
|
}
|
|
return Part->Wipe();
|
|
} else
|
|
return Part->Wipe();
|
|
}
|
|
LOGE("Wipe: Unable to find partition for name '%s'\n", Name.c_str());
|
|
return false;
|
|
}
|
|
|
|
int TWPartitionManager::Factory_Reset(void) {
|
|
std::vector<TWPartition*>::iterator iter;
|
|
int ret = true;
|
|
|
|
for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
|
|
if ((*iter)->Wipe_During_Factory_Reset) {
|
|
if (!(*iter)->Wipe())
|
|
ret = false;
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
void TWPartitionManager::Refresh_Sizes(void) {
|
|
Update_System_Details();
|
|
return;
|
|
}
|
|
|
|
void TWPartitionManager::Update_System_Details(void) {
|
|
std::vector<TWPartition*>::iterator iter;
|
|
int data_size = 0;
|
|
|
|
ui_print("Updating partition details...\n");
|
|
for (iter = Partitions.begin(); iter != Partitions.end(); iter++) {
|
|
if ((*iter)->Can_Be_Mounted) {
|
|
(*iter)->Update_Size(true);
|
|
if ((*iter)->Mount_Point == "/system") {
|
|
int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
|
|
DataManager::SetValue(TW_BACKUP_SYSTEM_SIZE, backup_display_size);
|
|
} else if ((*iter)->Mount_Point == "/data" || (*iter)->Mount_Point == "/datadata") {
|
|
data_size += (int)((*iter)->Backup_Size / 1048576LLU);
|
|
} else if ((*iter)->Mount_Point == "/cache") {
|
|
int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
|
|
DataManager::SetValue(TW_BACKUP_CACHE_SIZE, backup_display_size);
|
|
} else if ((*iter)->Mount_Point == "/sd-ext") {
|
|
int backup_display_size = (int)((*iter)->Backup_Size / 1048576LLU);
|
|
DataManager::SetValue(TW_BACKUP_SDEXT_SIZE, backup_display_size);
|
|
if ((*iter)->Backup_Size == 0) {
|
|
DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 0);
|
|
DataManager::SetValue(TW_BACKUP_SDEXT_VAR, 0);
|
|
} else
|
|
DataManager::SetValue(TW_HAS_SDEXT_PARTITION, 1);
|
|
}
|
|
}
|
|
}
|
|
DataManager::SetValue(TW_BACKUP_DATA_SIZE, data_size);
|
|
string current_storage_path = DataManager::GetCurrentStoragePath();
|
|
TWPartition* FreeStorage = Find_Partition_By_Path(current_storage_path);
|
|
if (FreeStorage)
|
|
DataManager::SetValue(TW_STORAGE_FREE_SIZE, (int)(FreeStorage->Free / 1048576LLU));
|
|
else
|
|
LOGI("Unable to find storage partition '%s'.\n", current_storage_path.c_str());
|
|
if (!Write_Fstab())
|
|
LOGE("Error creating fstab\n");
|
|
return;
|
|
}
|
|
|
|
int TWPartitionManager::Decrypt_Device(string Password) {
|
|
#ifdef TW_INCLUDE_CRYPTO
|
|
int ret_val, password_len;
|
|
char crypto_blkdev[255], cPassword[255];
|
|
size_t result;
|
|
|
|
property_set("ro.crypto.state", "encrypted");
|
|
#ifdef TW_INCLUDE_JB_CRYPTO
|
|
// No extra flags needed
|
|
#else
|
|
property_set("ro.crypto.fs_type", CRYPTO_FS_TYPE);
|
|
property_set("ro.crypto.fs_real_blkdev", CRYPTO_REAL_BLKDEV);
|
|
property_set("ro.crypto.fs_mnt_point", CRYPTO_MNT_POINT);
|
|
property_set("ro.crypto.fs_options", CRYPTO_FS_OPTIONS);
|
|
property_set("ro.crypto.fs_flags", CRYPTO_FS_FLAGS);
|
|
property_set("ro.crypto.keyfile.userdata", CRYPTO_KEY_LOC);
|
|
#endif
|
|
strcpy(cPassword, Password.c_str());
|
|
if (cryptfs_check_passwd(cPassword) != 0) {
|
|
LOGE("Failed to decrypt data.\n");
|
|
return -1;
|
|
}
|
|
property_get("ro.crypto.fs_crypto_blkdev", crypto_blkdev, "error");
|
|
if (strcmp(crypto_blkdev, "error") == 0) {
|
|
LOGE("Error retrieving decrypted data block device.\n");
|
|
} else {
|
|
TWPartition* dat = Find_Partition_By_Path("/data");
|
|
if (dat != NULL) {
|
|
DataManager::SetValue(TW_DATA_BLK_DEVICE, dat->Block_Device);
|
|
DataManager::SetValue(TW_IS_DECRYPTED, 1);
|
|
dat->Is_Decrypted = true;
|
|
dat->Decrypted_Block_Device = crypto_blkdev;
|
|
ui_print("Data successfully decrypted, new block device: '%s'\n", crypto_blkdev);
|
|
// Sleep for a bit so that the device will be ready
|
|
sleep(1);
|
|
Update_System_Details();
|
|
} else
|
|
LOGE("Unable to locate data partition.\n");
|
|
}
|
|
return 0;
|
|
#else
|
|
LOGE("No crypto support was compiled into this build.\n");
|
|
return -1;
|
|
#endif
|
|
return 1;
|
|
}
|
|
|
|
string TWPartitionManager::Get_Root_Path(string Path) {
|
|
string Local_Path = Path;
|
|
|
|
// Make sure that we have a leading slash
|
|
if (Local_Path.substr(0, 1) != "/")
|
|
Local_Path = "/" + Local_Path;
|
|
|
|
// Trim the path to get the root path only
|
|
size_t position = Local_Path.find("/", 2);
|
|
if (position != string::npos) {
|
|
Local_Path.resize(position);
|
|
}
|
|
return Local_Path;
|
|
}
|