mirror of
https://github.com/CTCaer/hekate
synced 2024-11-16 17:09:27 +00:00
1453 lines
42 KiB
C
1453 lines
42 KiB
C
/*
|
|
* Copyright (c) 2018 naehrwert
|
|
* Copyright (c) 2018 Rajko Stojadinovic
|
|
* Copyright (c) 2018-2020 CTCaer
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms and conditions of the GNU General Public License,
|
|
* version 2, as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope 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, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
//! fix the dram stuff and the pop ups
|
|
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "gui.h"
|
|
#include "fe_emmc_tools.h"
|
|
#include "fe_emummc_tools.h"
|
|
#include "../../../common/memory_map.h"
|
|
#include "../config/config.h"
|
|
#include "../libs/fatfs/ff.h"
|
|
#include "../mem/heap.h"
|
|
#include "../sec/se.h"
|
|
#include "../sec/se_t210.h"
|
|
#include "../storage/mbr_gpt.h"
|
|
#include "../storage/nx_emmc.h"
|
|
#include "../storage/sdmmc.h"
|
|
#include "../utils/btn.h"
|
|
#include "../utils/sprintf.h"
|
|
#include "../utils/util.h"
|
|
|
|
#define NUM_SECTORS_PER_ITER 8192 // 4MB Cache.
|
|
#define OUT_FILENAME_SZ 128
|
|
#define HASH_FILENAME_SZ (OUT_FILENAME_SZ + 11) // 11 == strlen(".sha256sums")
|
|
#define SHA256_SZ 0x20
|
|
|
|
extern sdmmc_t sd_sdmmc;
|
|
extern sdmmc_storage_t sd_storage;
|
|
extern FATFS sd_fs;
|
|
extern hekate_config h_cfg;
|
|
|
|
extern bool sd_mount();
|
|
extern void sd_unmount(bool deinit);
|
|
extern void emmcsn_path_impl(char *path, char *sub_dir, char *filename, sdmmc_storage_t *storage);
|
|
|
|
static void get_valid_partition(u32 *sector_start, u32 *sector_size, u32 *part_idx, bool backup)
|
|
{
|
|
sd_mount();
|
|
mbr_t *mbr = (mbr_t *)calloc(sizeof(mbr_t), 1);
|
|
sdmmc_storage_read(&sd_storage, 0, 1, mbr);
|
|
|
|
*part_idx = 0;
|
|
int i = 0;
|
|
u32 curr_part_size = 0;
|
|
for (i = 1; i < 4; i++)
|
|
{
|
|
curr_part_size = mbr->partitions[i].size_sct;
|
|
*sector_start = mbr->partitions[i].start_sct;
|
|
u8 type = mbr->partitions[i].type;
|
|
if ((curr_part_size >= *sector_size) && *sector_start && type != 0x83 && (!backup || type == 0xE0))
|
|
break;
|
|
}
|
|
if (i < 4)
|
|
*part_idx = i;
|
|
else
|
|
{
|
|
*sector_start = 0;
|
|
*sector_size = 0;
|
|
*part_idx = 0;
|
|
goto out;
|
|
}
|
|
|
|
if (!backup)
|
|
*sector_start = *sector_start + 0x8000;
|
|
else
|
|
{
|
|
*sector_size = curr_part_size;
|
|
sdmmc_storage_read(&sd_storage, *sector_start + 0xC001, 1, mbr);
|
|
if (!memcmp(mbr, "EFI PART", 8))
|
|
{
|
|
*sector_start = *sector_start + 0x8000;
|
|
goto out;
|
|
}
|
|
sdmmc_storage_read(&sd_storage, *sector_start + 0x4001, 1, mbr);
|
|
if (!memcmp(mbr, "EFI PART", 8))
|
|
goto out;
|
|
|
|
sdmmc_storage_read(&sd_storage, *sector_start + 0x4003, 1, mbr);
|
|
if (!memcmp(mbr, "EFI PART", 8))
|
|
{
|
|
*sector_start = *sector_start + 2;
|
|
goto out;
|
|
}
|
|
}
|
|
|
|
out:
|
|
free(mbr);
|
|
}
|
|
|
|
static lv_obj_t *create_mbox_text(char *text, bool button_ok)
|
|
{
|
|
lv_obj_t *dark_bg = lv_obj_create(lv_scr_act(), NULL);
|
|
lv_obj_set_style(dark_bg, &mbox_darken);
|
|
lv_obj_set_size(dark_bg, LV_HOR_RES, LV_VER_RES);
|
|
|
|
static const char *mbox_btn_map[] = { "\211", "\222OK", "\211", "" };
|
|
lv_obj_t *mbox = lv_mbox_create(dark_bg, NULL);
|
|
lv_mbox_set_recolor_text(mbox, true);
|
|
lv_obj_set_width(mbox, LV_HOR_RES / 9 * 6);
|
|
|
|
lv_mbox_set_text(mbox, text);
|
|
if (button_ok)
|
|
lv_mbox_add_btns(mbox, mbox_btn_map, mbox_action);
|
|
|
|
lv_obj_align(mbox, NULL, LV_ALIGN_CENTER, 0, 0);
|
|
lv_obj_set_top(mbox, true);
|
|
|
|
return dark_bg;
|
|
}
|
|
|
|
static void _update_filename(char *outFilename, u32 sdPathLen, u32 currPartIdx)
|
|
{
|
|
if (currPartIdx < 10)
|
|
{
|
|
outFilename[sdPathLen] = '0';
|
|
itoa(currPartIdx, &outFilename[sdPathLen + 1], 10);
|
|
}
|
|
else
|
|
itoa(currPartIdx, &outFilename[sdPathLen], 10);
|
|
}
|
|
|
|
static int _dump_emmc_verify(emmc_tool_gui_t *gui, sdmmc_storage_t *storage, u32 lba_curr, char *outFilename, emmc_part_t *part)
|
|
{
|
|
FIL fp;
|
|
FIL hashFp;
|
|
u8 sparseShouldVerify = 4;
|
|
u32 prevPct = 200;
|
|
u32 sdFileSector = 0;
|
|
int res = 0;
|
|
const char hexa[] = "0123456789abcdef";
|
|
DWORD *clmt = NULL;
|
|
|
|
u8 hashEm[SHA256_SZ];
|
|
u8 hashSd[SHA256_SZ];
|
|
|
|
if (f_open(&fp, outFilename, FA_READ) == FR_OK)
|
|
{
|
|
if (h_cfg.verification == 3)
|
|
{
|
|
char hashFilename[HASH_FILENAME_SZ];
|
|
strncpy(hashFilename, outFilename, OUT_FILENAME_SZ - 1);
|
|
strcat(hashFilename, ".sha256sums");
|
|
|
|
res = f_open(&hashFp, hashFilename, FA_CREATE_ALWAYS | FA_WRITE);
|
|
if (res)
|
|
{
|
|
f_close(&fp);
|
|
|
|
s_printf(gui->txt_buf,
|
|
"\n#FF0000 Hash file could not be written (error %d)!#\n"
|
|
"#FF0000 Aborting..#\n", res);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 1;
|
|
}
|
|
|
|
char chunkSizeAscii[10];
|
|
itoa(NUM_SECTORS_PER_ITER * NX_EMMC_BLOCKSIZE, chunkSizeAscii, 10);
|
|
chunkSizeAscii[9] = '\0';
|
|
|
|
f_puts("# chunksize: ", &hashFp);
|
|
f_puts(chunkSizeAscii, &hashFp);
|
|
f_puts("\n", &hashFp);
|
|
}
|
|
|
|
u32 totalSectorsVer = (u32)((u64)f_size(&fp) >> (u64)9);
|
|
|
|
u8 *bufEm = (u8 *)EMMC_BUF_ALIGNED;
|
|
u8 *bufSd = (u8 *)SDXC_BUF_ALIGNED;
|
|
|
|
u32 pct = (u64)((u64)(lba_curr - part->lba_start) * 100u) / (u64)(part->lba_end - part->lba_start);
|
|
lv_bar_set_value(gui->bar, pct);
|
|
lv_bar_set_style(gui->bar, LV_BAR_STYLE_BG, gui->bar_teal_bg);
|
|
lv_bar_set_style(gui->bar, LV_BAR_STYLE_INDIC, lv_theme_get_current()->bar.indic);
|
|
s_printf(gui->txt_buf, " "SYMBOL_DOT" %d%%", pct);
|
|
lv_label_set_text(gui->label_pct, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
clmt = f_expand_cltbl(&fp, 0x400000, 0);
|
|
|
|
u32 num = 0;
|
|
while (totalSectorsVer > 0)
|
|
{
|
|
num = MIN(totalSectorsVer, NUM_SECTORS_PER_ITER);
|
|
|
|
// Check every time or every 4.
|
|
// Every 4 protects from fake sd, sector corruption and frequent I/O corruption.
|
|
// Full provides all that, plus protection from extremely rare I/O corruption.
|
|
if ((h_cfg.verification >= 2) || !(sparseShouldVerify % 4))
|
|
{
|
|
if (!sdmmc_storage_read(storage, lba_curr, num, bufEm))
|
|
{
|
|
s_printf(gui->txt_buf,
|
|
"\n#FF0000 Failed to read %d blocks (@LBA %08X),#\n"
|
|
"#FF0000 from eMMC! Verification failed..#\n",
|
|
num, lba_curr);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
free(clmt);
|
|
f_close(&fp);
|
|
if (h_cfg.verification == 3)
|
|
f_close(&hashFp);
|
|
|
|
return 1;
|
|
}
|
|
manual_system_maintenance(false);
|
|
se_calc_sha256(hashEm, NULL, bufEm, num << 9, 0, SHA_INIT_HASH, false);
|
|
|
|
f_lseek(&fp, (u64)sdFileSector << (u64)9);
|
|
if (f_read_fast(&fp, bufSd, num << 9))
|
|
{
|
|
s_printf(gui->txt_buf,
|
|
"\n#FF0000 Failed to read %d blocks (@LBA %08X),#\n"
|
|
"#FF0000 from SD card! Verification failed..#\n",
|
|
num, lba_curr);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
free(clmt);
|
|
f_close(&fp);
|
|
if (h_cfg.verification == 3)
|
|
f_close(&hashFp);
|
|
|
|
return 1;
|
|
}
|
|
manual_system_maintenance(false);
|
|
se_calc_sha256_finalize(hashEm, NULL);
|
|
se_calc_sha256(hashSd, NULL, bufSd, num << 9, 0, SHA_INIT_HASH, true);
|
|
res = memcmp(hashEm, hashSd, 0x10);
|
|
|
|
if (res)
|
|
{
|
|
s_printf(gui->txt_buf,
|
|
"\n#FF0000 SD & eMMC data (@LBA %08X) do not match!#\n"
|
|
"\n#FF0000 Verification failed..#\n",
|
|
lba_curr);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
free(clmt);
|
|
f_close(&fp);
|
|
if (h_cfg.verification == 3)
|
|
f_close(&hashFp);
|
|
|
|
return 1;
|
|
}
|
|
|
|
if (h_cfg.verification == 3)
|
|
{
|
|
// Transform computed hash to readable hexadecimal
|
|
char hashStr[SHA256_SZ * 2 + 1];
|
|
char *hashStrPtr = hashStr;
|
|
for (int i = 0; i < SHA256_SZ; i++)
|
|
{
|
|
*(hashStrPtr++) = hexa[hashSd[i] >> 4];
|
|
*(hashStrPtr++) = hexa[hashSd[i] & 0x0F];
|
|
}
|
|
hashStr[SHA256_SZ * 2] = '\0';
|
|
|
|
f_puts(hashStr, &hashFp);
|
|
f_puts("\n", &hashFp);
|
|
}
|
|
}
|
|
|
|
pct = (u64)((u64)(lba_curr - part->lba_start) * 100u) / (u64)(part->lba_end - part->lba_start);
|
|
if (pct != prevPct)
|
|
{
|
|
lv_bar_set_value(gui->bar, pct);
|
|
s_printf(gui->txt_buf, " "SYMBOL_DOT" %d%%", pct);
|
|
lv_label_set_text(gui->label_pct, gui->txt_buf);
|
|
prevPct = pct;
|
|
}
|
|
|
|
manual_system_maintenance(false);
|
|
|
|
lba_curr += num;
|
|
totalSectorsVer -= num;
|
|
sdFileSector += num;
|
|
sparseShouldVerify++;
|
|
|
|
// Check for cancellation combo.
|
|
if (btn_read_vol() == (BTN_VOL_UP | BTN_VOL_DOWN))
|
|
{
|
|
s_printf(gui->txt_buf, "#FFDD00 Verification was cancelled!#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
msleep(1000);
|
|
|
|
free(clmt);
|
|
f_close(&fp);
|
|
f_close(&hashFp);
|
|
|
|
return 0;
|
|
}
|
|
}
|
|
free(clmt);
|
|
f_close(&fp);
|
|
f_close(&hashFp);
|
|
|
|
lv_bar_set_value(gui->bar, pct);
|
|
s_printf(gui->txt_buf, " "SYMBOL_DOT" %d%%", pct);
|
|
lv_label_set_text(gui->label_pct, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
s_printf(gui->txt_buf, "#FFDD00 File not found or could not be loaded!#\n#FFDD00 Verification failed..#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
bool partial_sd_full_unmount = false;
|
|
|
|
static int _dump_emmc_part(emmc_tool_gui_t *gui, char *sd_path, sdmmc_storage_t *storage, emmc_part_t *part)
|
|
{
|
|
const u32 FAT32_FILESIZE_LIMIT = 0xFFFFFFFF;
|
|
const u32 SECTORS_TO_MIB_COEFF = 11;
|
|
|
|
partial_sd_full_unmount = false;
|
|
|
|
u32 multipartSplitSize = (1u << 31);
|
|
u32 totalSectors = part->lba_end - part->lba_start + 1;
|
|
u32 currPartIdx = 0;
|
|
u32 numSplitParts = 0;
|
|
u32 maxSplitParts = 0;
|
|
bool isSmallSdCard = false;
|
|
bool partialDumpInProgress = false;
|
|
int res = 0;
|
|
char *outFilename = sd_path;
|
|
u32 sdPathLen = strlen(sd_path);
|
|
|
|
FIL partialIdxFp;
|
|
char partialIdxFilename[12];
|
|
strcpy(partialIdxFilename, "partial.idx");
|
|
|
|
s_printf(gui->txt_buf, "#96FF00 SD Card free space:# %d MiB\n#96FF00 Total backup size:# %d MiB\n\n",
|
|
sd_fs.free_clst * sd_fs.csize >> SECTORS_TO_MIB_COEFF,
|
|
totalSectors >> SECTORS_TO_MIB_COEFF);
|
|
lv_label_ins_text(gui->label_info, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
lv_bar_set_value(gui->bar, 0);
|
|
lv_label_set_text(gui->label_pct, " "SYMBOL_DOT" 0%");
|
|
lv_bar_set_style(gui->bar, LV_BAR_STYLE_BG, lv_theme_get_current()->bar.bg);
|
|
lv_bar_set_style(gui->bar, LV_BAR_STYLE_INDIC, gui->bar_white_ind);
|
|
manual_system_maintenance(true);
|
|
|
|
// 1GB parts for sd cards 8GB and less.
|
|
if ((sd_storage.csd.capacity >> (20 - sd_storage.csd.read_blkbits)) <= 8192)
|
|
multipartSplitSize = (1u << 30);
|
|
// Maximum parts fitting the free space available.
|
|
maxSplitParts = (sd_fs.free_clst * sd_fs.csize) / (multipartSplitSize / NX_EMMC_BLOCKSIZE);
|
|
|
|
// Check if the USER partition or the RAW eMMC fits the sd card free space.
|
|
if (totalSectors > (sd_fs.free_clst * sd_fs.csize))
|
|
{
|
|
isSmallSdCard = true;
|
|
|
|
s_printf(gui->txt_buf, "\n#FFBA00 Free space is smaller than backup size.#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
if (!maxSplitParts)
|
|
{
|
|
s_printf(gui->txt_buf, "#FFDD00 Not enough free space for Partial Backup!#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 0;
|
|
}
|
|
}
|
|
// Check if we are continuing a previous raw eMMC or USER partition backup in progress.
|
|
if (f_open(&partialIdxFp, partialIdxFilename, FA_READ) == FR_OK && totalSectors > (FAT32_FILESIZE_LIMIT / NX_EMMC_BLOCKSIZE))
|
|
{
|
|
s_printf(gui->txt_buf, "\n#AEFD14 Partial Backup in progress. Continuing...#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
partialDumpInProgress = true;
|
|
// Force partial dumping, even if the card is larger.
|
|
isSmallSdCard = true;
|
|
|
|
f_read(&partialIdxFp, &currPartIdx, 4, NULL);
|
|
f_close(&partialIdxFp);
|
|
|
|
if (!maxSplitParts)
|
|
{
|
|
s_printf(gui->txt_buf, "\n#FFDD00 Not enough free space for Partial Backup!#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 0;
|
|
}
|
|
|
|
// Increase maxSplitParts to accommodate previously backed up parts.
|
|
maxSplitParts += currPartIdx;
|
|
}
|
|
else if (isSmallSdCard)
|
|
{
|
|
s_printf(gui->txt_buf, "\n#FFBA00 Partial Backup enabled (%d MiB parts)...#\n", multipartSplitSize >> 20);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
}
|
|
|
|
// Check if filesystem is FAT32 or the free space is smaller and backup in parts.
|
|
if (((sd_fs.fs_type != FS_EXFAT) && totalSectors > (FAT32_FILESIZE_LIMIT / NX_EMMC_BLOCKSIZE)) || isSmallSdCard)
|
|
{
|
|
u32 multipartSplitSectors = multipartSplitSize / NX_EMMC_BLOCKSIZE;
|
|
numSplitParts = (totalSectors + multipartSplitSectors - 1) / multipartSplitSectors;
|
|
|
|
outFilename[sdPathLen++] = '.';
|
|
|
|
// Continue from where we left, if Partial Backup in progress.
|
|
_update_filename(outFilename, sdPathLen, partialDumpInProgress ? currPartIdx : 0);
|
|
}
|
|
|
|
FIL fp;
|
|
if (!f_open(&fp, outFilename, FA_READ))
|
|
{
|
|
f_close(&fp);
|
|
|
|
lv_obj_t *warn_mbox_bg = create_mbox_text(
|
|
"#FFDD00 An existing backup has been detected!#\n\n"
|
|
"Press #FF8000 POWER# to Continue.\nPress #FF8000 VOL# to abort.", false);
|
|
manual_system_maintenance(true);
|
|
|
|
if (!(btn_wait() & BTN_POWER))
|
|
{
|
|
lv_obj_del(warn_mbox_bg);
|
|
return 0;
|
|
}
|
|
lv_obj_del(warn_mbox_bg);
|
|
}
|
|
|
|
s_printf(gui->txt_buf, "#96FF00 Filepath:#\n%s\n#96FF00 Filename:# #FF8000 %s#",
|
|
gui->base_path, outFilename + strlen(gui->base_path));
|
|
lv_label_ins_text(gui->label_info, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
res = f_open(&fp, outFilename, FA_CREATE_ALWAYS | FA_WRITE);
|
|
if (res)
|
|
{
|
|
s_printf(gui->txt_buf, "\n#FF0000 Error (%d) while creating#\n#FFDD00 %s#\n", res, outFilename);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 0;
|
|
}
|
|
|
|
u8 *buf = (u8 *)MIXD_BUF_ALIGNED;
|
|
|
|
u32 lba_curr = part->lba_start;
|
|
u32 lbaStartPart = part->lba_start;
|
|
u32 bytesWritten = 0;
|
|
u32 prevPct = 200;
|
|
int retryCount = 0;
|
|
DWORD *clmt = NULL;
|
|
|
|
// Continue from where we left, if Partial Backup in progress.
|
|
if (partialDumpInProgress)
|
|
{
|
|
lba_curr += currPartIdx * (multipartSplitSize / NX_EMMC_BLOCKSIZE);
|
|
totalSectors -= currPartIdx * (multipartSplitSize / NX_EMMC_BLOCKSIZE);
|
|
lbaStartPart = lba_curr; // Update the start LBA for verification.
|
|
}
|
|
u64 totalSize = (u64)((u64)totalSectors << 9);
|
|
if (!isSmallSdCard && (sd_fs.fs_type == FS_EXFAT || totalSize <= FAT32_FILESIZE_LIMIT))
|
|
clmt = f_expand_cltbl(&fp, 0x400000, totalSize);
|
|
else
|
|
clmt = f_expand_cltbl(&fp, 0x400000, MIN(totalSize, multipartSplitSize));
|
|
|
|
u32 num = 0;
|
|
u32 pct = 0;
|
|
|
|
lv_obj_set_opa_scale(gui->bar, LV_OPA_COVER);
|
|
lv_obj_set_opa_scale(gui->label_pct, LV_OPA_COVER);
|
|
while (totalSectors > 0)
|
|
{
|
|
if (numSplitParts != 0 && bytesWritten >= multipartSplitSize)
|
|
{
|
|
f_close(&fp);
|
|
free(clmt);
|
|
memset(&fp, 0, sizeof(fp));
|
|
currPartIdx++;
|
|
|
|
if (h_cfg.verification)
|
|
{
|
|
// Verify part.
|
|
if (_dump_emmc_verify(gui, storage, lbaStartPart, outFilename, part))
|
|
{
|
|
s_printf(gui->txt_buf, "#FFDD00 Please try again...#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 0;
|
|
}
|
|
lv_bar_set_style(gui->bar, LV_BAR_STYLE_BG, lv_theme_get_current()->bar.bg);
|
|
lv_bar_set_style(gui->bar, LV_BAR_STYLE_INDIC, gui->bar_white_ind);
|
|
}
|
|
|
|
_update_filename(outFilename, sdPathLen, currPartIdx);
|
|
|
|
// Always create partial.idx before next part, in case a fatal error occurs.
|
|
if (isSmallSdCard)
|
|
{
|
|
// Create partial backup index file.
|
|
if (f_open(&partialIdxFp, partialIdxFilename, FA_CREATE_ALWAYS | FA_WRITE) == FR_OK)
|
|
{
|
|
f_write(&partialIdxFp, &currPartIdx, 4, NULL);
|
|
f_close(&partialIdxFp);
|
|
}
|
|
else
|
|
{
|
|
s_printf(gui->txt_buf, "#FF0000 Error creating partial.idx file!#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 0;
|
|
}
|
|
|
|
// More parts to backup that do not currently fit the sd card free space or fatal error.
|
|
if (currPartIdx >= maxSplitParts)
|
|
{
|
|
create_mbox_text(
|
|
"#96FF00 Partial Backup in progress!#\n\n"
|
|
"#96FF00 1.# Press OK to unmount SD Card.\n"
|
|
"#96FF00 2.# Remove SD Card and move files to free space.\n"
|
|
"#FFDD00 Don\'t move the partial.idx file!#\n"
|
|
"#96FF00 3.# Re-insert SD Card.\n"
|
|
"#96FF00 4.# Select the SAME option again to continue.", true);
|
|
|
|
partial_sd_full_unmount = true;
|
|
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
// Create next part.
|
|
s_printf(gui->txt_buf, "%s#", outFilename + strlen(gui->base_path));
|
|
lv_label_cut_text(gui->label_info,
|
|
strlen(lv_label_get_text(gui->label_info)) - strlen(outFilename + strlen(gui->base_path)) - 1,
|
|
strlen(outFilename + strlen(gui->base_path)) + 1);
|
|
lv_label_ins_text(gui->label_info, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
lbaStartPart = lba_curr;
|
|
res = f_open(&fp, outFilename, FA_CREATE_ALWAYS | FA_WRITE);
|
|
if (res)
|
|
{
|
|
s_printf(gui->txt_buf, "\n#FF0000 Error (%d) while creating#\n#FFDD00 %s#\n", res, outFilename);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 0;
|
|
}
|
|
|
|
bytesWritten = 0;
|
|
|
|
totalSize = (u64)((u64)totalSectors << 9);
|
|
clmt = f_expand_cltbl(&fp, 0x400000, MIN(totalSize, multipartSplitSize));
|
|
}
|
|
|
|
retryCount = 0;
|
|
num = MIN(totalSectors, NUM_SECTORS_PER_ITER);
|
|
while (!sdmmc_storage_read(storage, lba_curr, num, buf))
|
|
{
|
|
s_printf(gui->txt_buf,
|
|
"\n#FFDD00 Error reading %d blocks @ LBA %08X,#\n"
|
|
"#FFDD00 from eMMC (try %d). #",
|
|
num, lba_curr, ++retryCount);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
msleep(150);
|
|
if (retryCount >= 3)
|
|
{
|
|
s_printf(gui->txt_buf, "#FF0000 Aborting...#\nPlease try again...\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
f_close(&fp);
|
|
free(clmt);
|
|
f_unlink(outFilename);
|
|
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
s_printf(gui->txt_buf, "#FFDD00 Retrying...#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
}
|
|
}
|
|
manual_system_maintenance(false);
|
|
|
|
res = f_write_fast(&fp, buf, NX_EMMC_BLOCKSIZE * num);
|
|
|
|
if (res)
|
|
{
|
|
s_printf(gui->txt_buf, "\n#FF0000 Fatal error (%d) when writing to SD Card#\nPlease try again...\n", res);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
f_close(&fp);
|
|
free(clmt);
|
|
f_unlink(outFilename);
|
|
|
|
return 0;
|
|
}
|
|
|
|
manual_system_maintenance(false);
|
|
|
|
pct = (u64)((u64)(lba_curr - part->lba_start) * 100u) / (u64)(part->lba_end - part->lba_start);
|
|
if (pct != prevPct)
|
|
{
|
|
lv_bar_set_value(gui->bar, pct);
|
|
s_printf(gui->txt_buf, " "SYMBOL_DOT" %d%%", pct);
|
|
lv_label_set_text(gui->label_pct, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
prevPct = pct;
|
|
}
|
|
|
|
lba_curr += num;
|
|
totalSectors -= num;
|
|
bytesWritten += num * NX_EMMC_BLOCKSIZE;
|
|
|
|
// Force a flush after a lot of data if not splitting.
|
|
if (numSplitParts == 0 && bytesWritten >= multipartSplitSize)
|
|
{
|
|
f_sync(&fp);
|
|
bytesWritten = 0;
|
|
}
|
|
|
|
// Check for cancellation combo.
|
|
if (btn_read_vol() == (BTN_VOL_UP | BTN_VOL_DOWN))
|
|
{
|
|
s_printf(gui->txt_buf, "\n#FFDD00 The backup was cancelled!#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
msleep(1500);
|
|
|
|
f_close(&fp);
|
|
free(clmt);
|
|
f_unlink(outFilename);
|
|
|
|
return 0;
|
|
}
|
|
}
|
|
lv_bar_set_value(gui->bar, 100);
|
|
lv_label_set_text(gui->label_pct, " "SYMBOL_DOT" 100%");
|
|
manual_system_maintenance(true);
|
|
|
|
// Backup operation ended successfully.
|
|
f_close(&fp);
|
|
free(clmt);
|
|
|
|
if (h_cfg.verification)
|
|
{
|
|
// Verify last part or single file backup.
|
|
if (_dump_emmc_verify(gui, storage, lbaStartPart, outFilename, part))
|
|
{
|
|
s_printf(gui->txt_buf, "\n#FFDD00 Please try again...#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 0;
|
|
}
|
|
lv_bar_set_value(gui->bar, 100);
|
|
lv_label_set_text(gui->label_pct, " "SYMBOL_DOT" 100%");
|
|
manual_system_maintenance(true);
|
|
}
|
|
|
|
// Remove partial backup index file if no fatal errors occurred.
|
|
if (isSmallSdCard)
|
|
{
|
|
f_unlink(partialIdxFilename);
|
|
|
|
create_mbox_text(
|
|
"#96FF00 Partial Backup done!#\n\n"
|
|
"You can now join the files if needed\nand get the complete eMMC RAW GPP backup.", true);
|
|
|
|
partial_sd_full_unmount = true;
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
void dump_emmc_selected(emmcPartType_t dumpType, emmc_tool_gui_t *gui)
|
|
{
|
|
int res = 0;
|
|
u32 timer = 0;
|
|
|
|
char *txt_buf = (char *)malloc(0x1000);
|
|
gui->txt_buf = txt_buf;
|
|
|
|
s_printf(txt_buf, "");
|
|
lv_label_set_text(gui->label_log, txt_buf);
|
|
|
|
lv_label_set_text(gui->label_info, "Checking for available free space...");
|
|
manual_system_maintenance(true);
|
|
|
|
if (!sd_mount())
|
|
{
|
|
lv_label_set_text(gui->label_info, "#FFDD00 Failed to init SD!#");
|
|
goto out;
|
|
}
|
|
|
|
// Get SD Card free space for Partial Backup.
|
|
f_getfree("", &sd_fs.free_clst, NULL);
|
|
|
|
sdmmc_storage_t storage;
|
|
sdmmc_t sdmmc;
|
|
if (!sdmmc_storage_init_mmc(&storage, &sdmmc, SDMMC_4, SDMMC_BUS_WIDTH_8, 4))
|
|
{
|
|
lv_label_set_text(gui->label_info, "#FFDD00 Failed to init eMMC!#");
|
|
goto out;
|
|
}
|
|
|
|
int i = 0;
|
|
char sdPath[OUT_FILENAME_SZ];
|
|
// Create Restore folders, if they do not exist.
|
|
emmcsn_path_impl(sdPath, "/restore", "", &storage);
|
|
emmcsn_path_impl(sdPath, "/restore/partitions", "", &storage);
|
|
emmcsn_path_impl(sdPath, "", "", &storage);
|
|
gui->base_path = (char *)malloc(strlen(sdPath) + 1);
|
|
strcpy(gui->base_path, sdPath);
|
|
|
|
timer = get_tmr_s();
|
|
if (dumpType & PART_BOOT)
|
|
{
|
|
const u32 BOOT_PART_SIZE = storage.ext_csd.boot_mult << 17;
|
|
|
|
emmc_part_t bootPart;
|
|
memset(&bootPart, 0, sizeof(bootPart));
|
|
bootPart.lba_start = 0;
|
|
bootPart.lba_end = (BOOT_PART_SIZE / NX_EMMC_BLOCKSIZE) - 1;
|
|
for (i = 0; i < 2; i++)
|
|
{
|
|
strcpy(bootPart.name, "BOOT");
|
|
bootPart.name[4] = (u8)('0' + i);
|
|
bootPart.name[5] = 0;
|
|
|
|
s_printf(txt_buf, "#00DDFF %02d: %s#\n#00DDFF Range: 0x%08X - 0x%08X#\n\n",
|
|
i, bootPart.name, bootPart.lba_start, bootPart.lba_end);
|
|
lv_label_set_text(gui->label_info, txt_buf);
|
|
s_printf(txt_buf, "%02d: %s... ", i, bootPart.name);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
sdmmc_storage_set_mmc_partition(&storage, i + 1);
|
|
|
|
emmcsn_path_impl(sdPath, "", bootPart.name, &storage);
|
|
res = _dump_emmc_part(gui, sdPath, &storage, &bootPart);
|
|
|
|
if (!res)
|
|
s_printf(txt_buf, "#FFDD00 Failed!#\n");
|
|
else
|
|
s_printf(txt_buf, "Done!\n");
|
|
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, txt_buf);
|
|
manual_system_maintenance(true);
|
|
}
|
|
}
|
|
|
|
if ((dumpType & PART_SYSTEM) || (dumpType & PART_USER) || (dumpType & PART_RAW))
|
|
{
|
|
sdmmc_storage_set_mmc_partition(&storage, 0);
|
|
|
|
if ((dumpType & PART_SYSTEM) || (dumpType & PART_USER))
|
|
{
|
|
emmcsn_path_impl(sdPath, "/partitions", "", &storage);
|
|
gui->base_path = (char *)malloc(strlen(sdPath) + 1);
|
|
strcpy(gui->base_path, sdPath);
|
|
|
|
LIST_INIT(gpt);
|
|
nx_emmc_gpt_parse(&gpt, &storage);
|
|
LIST_FOREACH_ENTRY(emmc_part_t, part, &gpt, link)
|
|
{
|
|
if ((dumpType & PART_USER) == 0 && !strcmp(part->name, "USER"))
|
|
continue;
|
|
if ((dumpType & PART_SYSTEM) == 0 && strcmp(part->name, "USER"))
|
|
continue;
|
|
|
|
s_printf(txt_buf, "#00DDFF %02d: %s#\n#00DDFF Range: 0x%08X - 0x%08X#\n\n",
|
|
i, part->name, part->lba_start, part->lba_end);
|
|
lv_label_set_text(gui->label_info, txt_buf);
|
|
s_printf(txt_buf, "%02d: %s... ", i, part->name);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, txt_buf);
|
|
manual_system_maintenance(true);
|
|
i++;
|
|
|
|
emmcsn_path_impl(sdPath, "/partitions", part->name, &storage);
|
|
res = _dump_emmc_part(gui, sdPath, &storage, part);
|
|
// If a part failed, don't continue.
|
|
if (!res)
|
|
{
|
|
s_printf(txt_buf, "#FFDD00 Failed!#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, txt_buf);
|
|
break;
|
|
}
|
|
else
|
|
s_printf(txt_buf, "Done!\n");
|
|
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, txt_buf);
|
|
manual_system_maintenance(true);
|
|
}
|
|
nx_emmc_gpt_free(&gpt);
|
|
}
|
|
|
|
if (dumpType & PART_RAW)
|
|
{
|
|
// Get GP partition size dynamically.
|
|
const u32 RAW_AREA_NUM_SECTORS = storage.sec_cnt;
|
|
|
|
emmc_part_t rawPart;
|
|
memset(&rawPart, 0, sizeof(rawPart));
|
|
rawPart.lba_start = 0;
|
|
rawPart.lba_end = RAW_AREA_NUM_SECTORS - 1;
|
|
strcpy(rawPart.name, "rawnand.bin");
|
|
{
|
|
s_printf(txt_buf, "#00DDFF %02d: %s#\n#00DDFF Range: 0x%08X - 0x%08X#\n\n",
|
|
i, rawPart.name, rawPart.lba_start, rawPart.lba_end);
|
|
lv_label_set_text(gui->label_info, txt_buf);
|
|
s_printf(txt_buf, "%02d: %s... ", i, rawPart.name);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
i++;
|
|
|
|
emmcsn_path_impl(sdPath, "", rawPart.name, &storage);
|
|
res = _dump_emmc_part(gui, sdPath, &storage, &rawPart);
|
|
|
|
if (!res)
|
|
s_printf(txt_buf, "#FFDD00 Failed!#\n");
|
|
else
|
|
s_printf(txt_buf, "Done!\n");
|
|
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, txt_buf);
|
|
manual_system_maintenance(true);
|
|
}
|
|
}
|
|
}
|
|
|
|
timer = get_tmr_s() - timer;
|
|
sdmmc_storage_end(&storage);
|
|
|
|
if (res && h_cfg.verification)
|
|
s_printf(txt_buf, "Time taken: %dm %ds.\n#96FF00 Finished and verified!#", timer / 60, timer % 60);
|
|
else if (res)
|
|
s_printf(txt_buf, "Time taken: %dm %ds.\nFinished!", timer / 60, timer % 60);
|
|
else
|
|
s_printf(txt_buf, "Time taken: %dm %ds.", timer / 60, timer % 60);
|
|
|
|
lv_label_set_text(gui->label_finish, txt_buf);
|
|
|
|
out:
|
|
free(txt_buf);
|
|
free(gui->base_path);
|
|
if (!partial_sd_full_unmount)
|
|
sd_unmount(false);
|
|
else
|
|
{
|
|
partial_sd_full_unmount = false;
|
|
sd_unmount(true);
|
|
}
|
|
}
|
|
|
|
static int _restore_emmc_part(emmc_tool_gui_t *gui, char *sd_path, int active_part, sdmmc_storage_t *storage, emmc_part_t *part, bool allow_multi_part)
|
|
{
|
|
const u32 SECTORS_TO_MIB_COEFF = 11;
|
|
|
|
u32 totalSectors = part->lba_end - part->lba_start + 1;
|
|
u32 currPartIdx = 0;
|
|
u32 numSplitParts = 0;
|
|
u32 lbaStartPart = part->lba_start;
|
|
int res = 0;
|
|
char *outFilename = sd_path;
|
|
u32 sdPathLen = strlen(sd_path);
|
|
u64 fileSize = 0;
|
|
u64 totalCheckFileSize = 0;
|
|
|
|
FIL fp;
|
|
FILINFO fno;
|
|
|
|
lv_bar_set_value(gui->bar, 0);
|
|
lv_label_set_text(gui->label_pct, " "SYMBOL_DOT" 0%");
|
|
lv_bar_set_style(gui->bar, LV_BAR_STYLE_BG, lv_theme_get_current()->bar.bg);
|
|
lv_bar_set_style(gui->bar, LV_BAR_STYLE_INDIC, gui->bar_white_ind);
|
|
manual_system_maintenance(true);
|
|
|
|
bool use_multipart = false;
|
|
bool check_4MB_aligned = true;
|
|
|
|
if (allow_multi_part)
|
|
{
|
|
// Check to see if there is a combined file and if so then use that.
|
|
if (f_stat(outFilename, &fno))
|
|
{
|
|
// If not, check if there are partial files and the total size matches.
|
|
s_printf(gui->txt_buf, "\nNo single file, checking for part files...\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
outFilename[sdPathLen++] = '.';
|
|
|
|
_update_filename(outFilename, sdPathLen, numSplitParts);
|
|
|
|
s_printf(gui->txt_buf, "#96FF00 Filepath:#\n%s\n#96FF00 Filename:# #FF8000 %s#",
|
|
gui->base_path, outFilename + strlen(gui->base_path));
|
|
lv_label_ins_text(gui->label_info, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
|
|
// Stat total size of the part files.
|
|
while ((u32)((u64)totalCheckFileSize >> (u64)9) != totalSectors)
|
|
{
|
|
_update_filename(outFilename, sdPathLen, numSplitParts);
|
|
|
|
s_printf(gui->txt_buf, "%s#", outFilename + strlen(gui->base_path));
|
|
lv_label_cut_text(gui->label_info,
|
|
strlen(lv_label_get_text(gui->label_info)) - strlen(outFilename + strlen(gui->base_path)) - 1,
|
|
strlen(outFilename + strlen(gui->base_path)) + 1);
|
|
lv_label_ins_text(gui->label_info, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
if (f_stat(outFilename, &fno) && !gui->raw_emummc)
|
|
{
|
|
s_printf(gui->txt_buf, "#FFDD00 Error (%d) file not found#\n#FFDD00 %s.#\n#FFDD00 Aborting...#", res, outFilename);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 0;
|
|
}
|
|
else if (f_stat(outFilename, &fno) && gui->raw_emummc)
|
|
{
|
|
totalSectors = (u32)((u64)totalCheckFileSize >> (u64)9);
|
|
}
|
|
else
|
|
{
|
|
totalCheckFileSize += (u64)fno.fsize;
|
|
|
|
if (check_4MB_aligned && (((u64)fno.fsize) % 0x400000))
|
|
{
|
|
s_printf(gui->txt_buf, "#FFDD00 The split file must be a#\n#FFDD00 multiple of 4 MiB.#\n#FFDD00 Aborting...#", res, outFilename);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 0;
|
|
}
|
|
|
|
check_4MB_aligned = false;
|
|
}
|
|
|
|
numSplitParts++;
|
|
}
|
|
|
|
s_printf(gui->txt_buf, "%X sectors total.\n", (u32)((u64)totalCheckFileSize >> (u64)9));
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
if ((u32)((u64)totalCheckFileSize >> (u64)9) != totalSectors)
|
|
{
|
|
s_printf(gui->txt_buf, "#FF0000 Size of SD Card split backup does not match,#\n#FF0000 eMMC's selected part size!#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
use_multipart = true;
|
|
_update_filename(outFilename, sdPathLen, 0);
|
|
}
|
|
}
|
|
}
|
|
|
|
res = f_open(&fp, outFilename, FA_READ);
|
|
if (use_multipart)
|
|
{
|
|
s_printf(gui->txt_buf, "%s#", outFilename + strlen(gui->base_path));
|
|
lv_label_cut_text(gui->label_info,
|
|
strlen(lv_label_get_text(gui->label_info)) - strlen(outFilename + strlen(gui->base_path)) - 1,
|
|
strlen(outFilename + strlen(gui->base_path)) + 1);
|
|
lv_label_ins_text(gui->label_info, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
}
|
|
else
|
|
{
|
|
s_printf(gui->txt_buf, "#96FF00 Filepath:#\n%s\n#96FF00 Filename:# #FF8000 %s#",
|
|
gui->base_path, outFilename + strlen(gui->base_path));
|
|
lv_label_ins_text(gui->label_info, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
}
|
|
manual_system_maintenance(true);
|
|
if (res)
|
|
{
|
|
if (res != FR_NO_FILE)
|
|
{
|
|
s_printf(gui->txt_buf, "\n#FF0000 Error (%d) while opening file. Continuing...#\n", res);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
}
|
|
else
|
|
{
|
|
s_printf(gui->txt_buf, "\n#FFDD00 Error (%d) file not found. Continuing...#\n", res);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
else if (!use_multipart && (((u32)((u64)f_size(&fp) >> (u64)9)) != totalSectors)) // Check total restore size vs emmc size.
|
|
{
|
|
if (!gui->raw_emummc)
|
|
{
|
|
s_printf(gui->txt_buf, "\n#FF0000 Size of the SD Card backup does not match,#\n#FF0000 eMMC's selected part size.#\n", res);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
f_close(&fp);
|
|
|
|
return 0;
|
|
}
|
|
else
|
|
totalSectors = (u32)((u64)f_size(&fp) >> (u64)9);
|
|
}
|
|
else
|
|
{
|
|
fileSize = (u64)f_size(&fp);
|
|
s_printf(gui->txt_buf, "\nTotal restore size: %d MiB.\n",
|
|
(u32)((use_multipart ? (u64)totalCheckFileSize : fileSize) >> (u64)9) >> SECTORS_TO_MIB_COEFF);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
}
|
|
|
|
u8 *buf = (u8 *)MIXD_BUF_ALIGNED;
|
|
|
|
u32 lba_curr = part->lba_start;
|
|
u32 bytesWritten = 0;
|
|
u32 prevPct = 200;
|
|
int retryCount = 0;
|
|
|
|
u32 num = 0;
|
|
u32 pct = 0;
|
|
|
|
DWORD *clmt = f_expand_cltbl(&fp, 0x400000, 0);
|
|
|
|
u32 sector_start = 0, part_idx = 0;
|
|
u32 sector_size = totalSectors;
|
|
u32 sd_sector_off = 0;
|
|
|
|
if (gui->raw_emummc)
|
|
{
|
|
get_valid_partition(§or_start, §or_size, &part_idx, false);
|
|
if (!part_idx)
|
|
{
|
|
s_printf(gui->txt_buf, "#FFDD00 Failed to find a partition...#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 0;
|
|
}
|
|
sd_sector_off = sector_start + (0x2000 * active_part);
|
|
}
|
|
|
|
lv_obj_set_opa_scale(gui->bar, LV_OPA_COVER);
|
|
lv_obj_set_opa_scale(gui->label_pct, LV_OPA_COVER);
|
|
while (totalSectors > 0)
|
|
{
|
|
// If we have more than one part, check the size for the split parts and make sure that the bytes written is not more than that.
|
|
if (numSplitParts != 0 && bytesWritten >= fileSize)
|
|
{
|
|
// If we have more bytes written then close the file pointer and increase the part index we are using
|
|
f_close(&fp);
|
|
free(clmt);
|
|
memset(&fp, 0, sizeof(fp));
|
|
currPartIdx++;
|
|
|
|
if (h_cfg.verification && !gui->raw_emummc)
|
|
{
|
|
// Verify part.
|
|
if (_dump_emmc_verify(gui, storage, lbaStartPart, outFilename, part))
|
|
{
|
|
s_printf(gui->txt_buf, "#FFDD00 Please try again...#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
_update_filename(outFilename, sdPathLen, currPartIdx);
|
|
|
|
// Read from next part.
|
|
s_printf(gui->txt_buf, "%s#", outFilename + strlen(gui->base_path));
|
|
lv_label_cut_text(gui->label_info,
|
|
strlen(lv_label_get_text(gui->label_info)) - strlen(outFilename + strlen(gui->base_path)) - 1,
|
|
strlen(outFilename + strlen(gui->base_path)) + 1);
|
|
lv_label_ins_text(gui->label_info, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
lbaStartPart = lba_curr;
|
|
|
|
// Try to open the next file part
|
|
res = f_open(&fp, outFilename, FA_READ);
|
|
if (res)
|
|
{
|
|
s_printf(gui->txt_buf, "#FF0000 Error (%d) while opening file#\n#FFDD00 %s!#\n", res, outFilename);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 0;
|
|
}
|
|
fileSize = (u64)f_size(&fp);
|
|
bytesWritten = 0;
|
|
clmt = f_expand_cltbl(&fp, 0x400000, 0);
|
|
}
|
|
|
|
retryCount = 0;
|
|
num = MIN(totalSectors, NUM_SECTORS_PER_ITER);
|
|
|
|
res = f_read_fast(&fp, buf, num << 9);
|
|
manual_system_maintenance(false);
|
|
|
|
if (res)
|
|
{
|
|
s_printf(gui->txt_buf,
|
|
"\n#FF0000 Fatal error (%d) when reading from SD!#\n"
|
|
"#FF0000 Your device may be in an inoperative state!#\n"
|
|
"#FFDD00 Please try again now!#\n", res);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
f_close(&fp);
|
|
free(clmt);
|
|
return 0;
|
|
}
|
|
if (!gui->raw_emummc)
|
|
res = !sdmmc_storage_write(storage, lba_curr, num, buf);
|
|
else
|
|
res = !sdmmc_storage_write(&sd_storage, lba_curr + sd_sector_off, num, buf);
|
|
|
|
manual_system_maintenance(false);
|
|
|
|
while (res)
|
|
{
|
|
s_printf(gui->txt_buf,
|
|
"#FFDD00 Error reading %d blocks @ LBA %08X,#\n"
|
|
"#FFDD00 from eMMC (try %d). #",
|
|
num, lba_curr, ++retryCount);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
msleep(150);
|
|
if (retryCount >= 3)
|
|
{
|
|
s_printf(gui->txt_buf, "#FF0000 Aborting...#\n"
|
|
"#FF0000 Your device may be in an inoperative state!#\n"
|
|
"#FFDD00 Please try again now!#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
f_close(&fp);
|
|
free(clmt);
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
s_printf(gui->txt_buf, "#FFDD00 Retrying...#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
}
|
|
if (!gui->raw_emummc)
|
|
res = !sdmmc_storage_write(storage, lba_curr, num, buf);
|
|
else
|
|
res = !sdmmc_storage_write(&sd_storage, lba_curr + sd_sector_off, num, buf);
|
|
manual_system_maintenance(false);
|
|
}
|
|
pct = (u64)((u64)(lba_curr - part->lba_start) * 100u) / (u64)(part->lba_end - part->lba_start);
|
|
if (pct != prevPct)
|
|
{
|
|
lv_bar_set_value(gui->bar, pct);
|
|
s_printf(gui->txt_buf, " "SYMBOL_DOT" %d%%", pct);
|
|
lv_label_set_text(gui->label_pct, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
prevPct = pct;
|
|
}
|
|
|
|
lba_curr += num;
|
|
totalSectors -= num;
|
|
bytesWritten += num * NX_EMMC_BLOCKSIZE;
|
|
}
|
|
lv_bar_set_value(gui->bar, 100);
|
|
lv_label_set_text(gui->label_pct, " "SYMBOL_DOT" 100%");
|
|
manual_system_maintenance(true);
|
|
|
|
// Restore operation ended successfully.
|
|
f_close(&fp);
|
|
free(clmt);
|
|
|
|
if (h_cfg.verification && !gui->raw_emummc)
|
|
{
|
|
// Verify restored data.
|
|
if (_dump_emmc_verify(gui, storage, lbaStartPart, outFilename, part))
|
|
{
|
|
s_printf(gui->txt_buf, "#FFDD00 Please try again...#\n");
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, gui->txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
return 0;
|
|
}
|
|
lv_bar_set_value(gui->bar, 100);
|
|
lv_label_set_text(gui->label_pct, " "SYMBOL_DOT" 100%");
|
|
manual_system_maintenance(true);
|
|
}
|
|
|
|
if (gui->raw_emummc)
|
|
{
|
|
char sdPath[OUT_FILENAME_SZ];
|
|
// Create Restore folders, if they do not exist.
|
|
f_mkdir("emuMMC");
|
|
s_printf(sdPath, "emuMMC/RAW%d", part_idx);
|
|
f_mkdir(sdPath);
|
|
strcat(sdPath, "/raw_based");
|
|
FIL fp_raw;
|
|
f_open(&fp_raw, sdPath, FA_CREATE_ALWAYS | FA_WRITE);
|
|
f_write(&fp_raw, §or_start, 4, NULL);
|
|
f_close(&fp_raw);
|
|
|
|
s_printf(sdPath, "emuMMC/RAW%d", part_idx);
|
|
save_emummc_cfg(part_idx, sector_start, sdPath);
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
void restore_emmc_selected(emmcPartType_t restoreType, emmc_tool_gui_t *gui)
|
|
{
|
|
int res = 0;
|
|
u32 timer = 0;
|
|
|
|
char *txt_buf = (char *)malloc(0x1000);
|
|
gui->txt_buf = txt_buf;
|
|
|
|
s_printf(txt_buf, "");
|
|
lv_label_set_text(gui->label_log, txt_buf);
|
|
|
|
manual_system_maintenance(true);
|
|
|
|
s_printf(txt_buf,
|
|
"#FFDD00 This may render your device inoperative!#\n\n"
|
|
"#FFDD00 Are you really sure?#");
|
|
if ((restoreType & PART_BOOT) || (restoreType & PART_GP_ALL))
|
|
{
|
|
s_printf(txt_buf + strlen(txt_buf),
|
|
"\n\nThe mode you selected will only restore\nthe partitions that it can find.\n"
|
|
"If it is not found, it will be skipped\nand continue with the next.");
|
|
}
|
|
|
|
u32 orig_msg_len = strlen(txt_buf);
|
|
|
|
lv_obj_t *warn_mbox_bg = create_mbox_text(txt_buf, false);
|
|
lv_obj_t *warn_mbox = lv_obj_get_child(warn_mbox_bg, NULL);
|
|
|
|
u8 failsafe_wait = 6;
|
|
while (failsafe_wait > 0)
|
|
{
|
|
s_printf(txt_buf + orig_msg_len, "\n\n#888888 Wait... (%ds)#", failsafe_wait);
|
|
lv_mbox_set_text(warn_mbox, txt_buf);
|
|
msleep(1000);
|
|
manual_system_maintenance(true);
|
|
failsafe_wait--;
|
|
}
|
|
|
|
s_printf(txt_buf + orig_msg_len, "\n\nPress #FF8000 POWER# to Continue.\nPress #FF8000 VOL# to abort.");
|
|
lv_mbox_set_text(warn_mbox, txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
u32 btn = btn_wait();
|
|
if (!(btn & BTN_POWER))
|
|
{
|
|
lv_label_set_text(gui->label_info, "#FFDD00 Restore operation was aborted!#");
|
|
lv_obj_del(warn_mbox_bg);
|
|
goto out;
|
|
}
|
|
lv_obj_del(warn_mbox_bg);
|
|
manual_system_maintenance(true);
|
|
|
|
if (!sd_mount())
|
|
{
|
|
lv_label_set_text(gui->label_info, "#FFDD00 Failed to init SD!#");
|
|
goto out;
|
|
}
|
|
|
|
sdmmc_storage_t storage;
|
|
sdmmc_t sdmmc;
|
|
if (!sdmmc_storage_init_mmc(&storage, &sdmmc, SDMMC_4, SDMMC_BUS_WIDTH_8, 4))
|
|
{
|
|
lv_label_set_text(gui->label_info, "#FFDD00 Failed to init eMMC!#");
|
|
goto out;
|
|
}
|
|
|
|
int i = 0;
|
|
char sdPath[OUT_FILENAME_SZ];
|
|
|
|
emmcsn_path_impl(sdPath, "/restore", "", &storage);
|
|
gui->base_path = (char *)malloc(strlen(sdPath) + 1);
|
|
strcpy(gui->base_path, sdPath);
|
|
|
|
timer = get_tmr_s();
|
|
if (restoreType & PART_BOOT)
|
|
{
|
|
const u32 BOOT_PART_SIZE = storage.ext_csd.boot_mult << 17;
|
|
|
|
emmc_part_t bootPart;
|
|
memset(&bootPart, 0, sizeof(bootPart));
|
|
bootPart.lba_start = 0;
|
|
bootPart.lba_end = (BOOT_PART_SIZE / NX_EMMC_BLOCKSIZE) - 1;
|
|
for (i = 0; i < 2; i++)
|
|
{
|
|
strcpy(bootPart.name, "BOOT");
|
|
bootPart.name[4] = (u8)('0' + i);
|
|
bootPart.name[5] = 0;
|
|
|
|
s_printf(txt_buf, "#00DDFF %02d: %s#\n#00DDFF Range: 0x%08X - 0x%08X#\n\n\n\n\n",
|
|
i, bootPart.name, bootPart.lba_start, bootPart.lba_end);
|
|
lv_label_set_text(gui->label_info, txt_buf);
|
|
s_printf(txt_buf, "%02d: %s... ", i, bootPart.name);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, txt_buf);
|
|
manual_system_maintenance(true);
|
|
|
|
sdmmc_storage_set_mmc_partition(&storage, i + 1);
|
|
|
|
emmcsn_path_impl(sdPath, "/restore", bootPart.name, &storage);
|
|
res = _restore_emmc_part(gui, sdPath, i, &storage, &bootPart, false);
|
|
|
|
if (!res)
|
|
s_printf(txt_buf, "#FFDD00 Failed!#\n");
|
|
else
|
|
s_printf(txt_buf, "Done!\n");
|
|
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, txt_buf);
|
|
manual_system_maintenance(true);
|
|
}
|
|
}
|
|
|
|
if (restoreType & PART_GP_ALL)
|
|
{
|
|
emmcsn_path_impl(sdPath, "/restore/partitions", "", &storage);
|
|
gui->base_path = (char *)malloc(strlen(sdPath) + 1);
|
|
strcpy(gui->base_path, sdPath);
|
|
|
|
sdmmc_storage_set_mmc_partition(&storage, 0);
|
|
|
|
LIST_INIT(gpt);
|
|
nx_emmc_gpt_parse(&gpt, &storage);
|
|
LIST_FOREACH_ENTRY(emmc_part_t, part, &gpt, link)
|
|
{
|
|
s_printf(txt_buf, "#00DDFF %02d: %s#\n#00DDFF Range: 0x%08X - 0x%08X#\n\n\n\n\n",
|
|
i, part->name, part->lba_start, part->lba_end);
|
|
lv_label_set_text(gui->label_info, txt_buf);
|
|
s_printf(txt_buf, "%02d: %s... ", i, part->name);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, txt_buf);
|
|
manual_system_maintenance(true);
|
|
i++;
|
|
|
|
emmcsn_path_impl(sdPath, "/restore/partitions", part->name, &storage);
|
|
res = _restore_emmc_part(gui, sdPath, 0, &storage, part, false);
|
|
|
|
if (!res)
|
|
s_printf(txt_buf, "#FFDD00 Failed!#\n");
|
|
else
|
|
s_printf(txt_buf, "Done!\n");
|
|
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, txt_buf);
|
|
manual_system_maintenance(true);
|
|
}
|
|
nx_emmc_gpt_free(&gpt);
|
|
}
|
|
|
|
if (restoreType & PART_RAW)
|
|
{
|
|
// Get GP partition size dynamically.
|
|
const u32 RAW_AREA_NUM_SECTORS = storage.sec_cnt;
|
|
|
|
emmc_part_t rawPart;
|
|
memset(&rawPart, 0, sizeof(rawPart));
|
|
rawPart.lba_start = 0;
|
|
rawPart.lba_end = RAW_AREA_NUM_SECTORS - 1;
|
|
strcpy(rawPart.name, "rawnand.bin");
|
|
{
|
|
s_printf(txt_buf, "#00DDFF %02d: %s#\n#00DDFF Range: 0x%08X - 0x%08X#\n\n\n\n\n",
|
|
i, rawPart.name, rawPart.lba_start, rawPart.lba_end);
|
|
lv_label_set_text(gui->label_info, txt_buf);
|
|
s_printf(txt_buf, "%02d: %s... ", i, rawPart.name);
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, txt_buf);
|
|
manual_system_maintenance(true);
|
|
i++;
|
|
|
|
emmcsn_path_impl(sdPath, "/restore", rawPart.name, &storage);
|
|
res = _restore_emmc_part(gui, sdPath, 2, &storage, &rawPart, true);
|
|
|
|
if (!res)
|
|
s_printf(txt_buf, "#FFDD00 Failed!#\n");
|
|
else
|
|
s_printf(txt_buf, "Done!\n");
|
|
|
|
lv_label_ins_text(gui->label_log, LV_LABEL_POS_LAST, txt_buf);
|
|
manual_system_maintenance(true);
|
|
}
|
|
}
|
|
|
|
timer = get_tmr_s() - timer;
|
|
sdmmc_storage_end(&storage);
|
|
|
|
if (res && h_cfg.verification)
|
|
s_printf(txt_buf, "Time taken: %dm %ds.\n#96FF00 Finished and verified!#", timer / 60, timer % 60);
|
|
else if (res)
|
|
s_printf(txt_buf, "Time taken: %dm %ds.\nFinished!", timer / 60, timer % 60);
|
|
else
|
|
s_printf(txt_buf, "Time taken: %dm %ds.", timer / 60, timer % 60);
|
|
|
|
lv_label_set_text(gui->label_finish, txt_buf);
|
|
|
|
out:
|
|
free(txt_buf);
|
|
free(gui->base_path);
|
|
sd_unmount(false);
|
|
}
|