a7b840e531
Upgrade process always copied sdmc:/boot.firm to nand:/boot.firm even if Luma3DS was chainloaded from another path. Fix this by copying from argv[0], if Luma3DS was launched from the SD card. Also use "sdmc:" and "nand:" instead of "0:" and "1:" when using FatFS. Closes #1776.
475 lines
14 KiB
C
475 lines
14 KiB
C
/*
|
|
* This file is part of Luma3DS
|
|
* Copyright (C) 2016-2021 Aurora Wright, TuxSH
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* 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, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
* Additional Terms 7.b and 7.c of GPLv3 apply to this file:
|
|
* * Requiring preservation of specified reasonable legal notices or
|
|
* author attributions in that material or in the Appropriate Legal
|
|
* Notices displayed by works containing it.
|
|
* * Prohibiting misrepresentation of the origin of that material,
|
|
* or requiring that modified versions of such material be marked in
|
|
* reasonable ways as different from the original version.
|
|
*/
|
|
|
|
#include "fs.h"
|
|
#include "memory.h"
|
|
#include "fmt.h"
|
|
#include "crypto.h"
|
|
#include "cache.h"
|
|
#include "screen.h"
|
|
#include "draw.h"
|
|
#include "utils.h"
|
|
#include "fatfs/ff.h"
|
|
#include "buttons.h"
|
|
#include "firm.h"
|
|
#include "crypto.h"
|
|
#include "strings.h"
|
|
#include "alignedseqmemcpy.h"
|
|
#include "i2c.h"
|
|
|
|
static FATFS sdFs,
|
|
nandFs;
|
|
|
|
static bool switchToMainDir(bool isSd)
|
|
{
|
|
const char *mainDir = isSd ? "/luma" : "/rw/luma";
|
|
|
|
switch(f_chdir(mainDir))
|
|
{
|
|
case FR_OK:
|
|
return true;
|
|
case FR_NO_PATH:
|
|
return f_mkdir(mainDir) == FR_OK && switchToMainDir(isSd);
|
|
default:
|
|
return false;
|
|
}
|
|
}
|
|
|
|
bool mountFs(bool isSd, bool switchToCtrNand)
|
|
{
|
|
static bool sdInitialized = false, nandInitialized = false;
|
|
if (isSd)
|
|
{
|
|
if (!sdInitialized)
|
|
sdInitialized = f_mount(&sdFs, "sdmc:", 1) == FR_OK;
|
|
return sdInitialized && switchToMainDir(true);
|
|
}
|
|
else
|
|
{
|
|
if (!nandInitialized)
|
|
nandInitialized = f_mount(&nandFs, "nand:", 1) == FR_OK;
|
|
return nandInitialized && (!switchToCtrNand || (f_chdrive("nand:") == FR_OK && switchToMainDir(false)));
|
|
}
|
|
}
|
|
|
|
u32 fileRead(void *dest, const char *path, u32 maxSize)
|
|
{
|
|
FIL file;
|
|
FRESULT result = FR_OK;
|
|
u32 ret = 0;
|
|
|
|
if(f_open(&file, path, FA_READ) != FR_OK) return ret;
|
|
|
|
u32 size = f_size(&file);
|
|
if(dest == NULL) ret = size;
|
|
else if(size <= maxSize)
|
|
result = f_read(&file, dest, size, (unsigned int *)&ret);
|
|
result |= f_close(&file);
|
|
|
|
return result == FR_OK ? ret : 0;
|
|
}
|
|
|
|
u32 getFileSize(const char *path)
|
|
{
|
|
return fileRead(NULL, path, 0);
|
|
}
|
|
|
|
bool fileWrite(const void *buffer, const char *path, u32 size)
|
|
{
|
|
FIL file;
|
|
FRESULT result = FR_OK;
|
|
|
|
switch(f_open(&file, path, FA_WRITE | FA_OPEN_ALWAYS))
|
|
{
|
|
case FR_OK:
|
|
{
|
|
unsigned int written;
|
|
result = f_write(&file, buffer, size, &written);
|
|
if(result == FR_OK) result = f_truncate(&file);
|
|
result |= f_close(&file);
|
|
|
|
return result == FR_OK && (u32)written == size;
|
|
}
|
|
case FR_NO_PATH:
|
|
for(u32 i = 1; path[i] != 0; i++)
|
|
if(path[i] == '/')
|
|
{
|
|
char folder[i + 1];
|
|
memcpy(folder, path, i);
|
|
folder[i] = 0;
|
|
result = f_mkdir(folder);
|
|
}
|
|
|
|
return result == FR_OK && fileWrite(buffer, path, size);
|
|
default:
|
|
return false;
|
|
}
|
|
}
|
|
|
|
bool fileDelete(const char *path)
|
|
{
|
|
return f_unlink(path) == FR_OK;
|
|
}
|
|
|
|
bool fileCopy(const char *pathSrc, const char *pathDst, bool replace, void *tmpBuffer, size_t bufferSize)
|
|
{
|
|
FIL fileSrc, fileDst;
|
|
FRESULT res;
|
|
|
|
res = f_open(&fileSrc, pathSrc, FA_READ);
|
|
if (res != FR_OK)
|
|
return true; // Succeed if the source file doesn't exist
|
|
|
|
size_t szSrc = f_size(&fileSrc), rem = szSrc;
|
|
|
|
res = f_open(&fileDst, pathDst, FA_WRITE | (replace ? FA_CREATE_ALWAYS : FA_CREATE_NEW));
|
|
|
|
if (res == FR_EXIST)
|
|
{
|
|
// We did not fail
|
|
f_close(&fileSrc);
|
|
return true;
|
|
}
|
|
else if (res == FR_NO_PATH)
|
|
{
|
|
const char *c;
|
|
for (c = pathDst + strlen(pathDst); *c != '/' && c >= pathDst; --c);
|
|
if (c >= pathDst && c - pathDst <= FF_MAX_LFN && *c != '\0')
|
|
{
|
|
char path[FF_MAX_LFN + 1];
|
|
strncpy(path, pathDst, c - pathDst);
|
|
path[c - pathDst] = '\0';
|
|
res = f_mkdir(path);
|
|
}
|
|
|
|
if (res == FR_OK)
|
|
res = f_open(&fileDst, pathDst, FA_WRITE | (replace ? FA_CREATE_ALWAYS : FA_CREATE_NEW));
|
|
}
|
|
|
|
if (res != FR_OK)
|
|
{
|
|
f_close(&fileSrc);
|
|
return false;
|
|
}
|
|
|
|
while (rem > 0)
|
|
{
|
|
size_t sz = rem >= bufferSize ? bufferSize : rem;
|
|
UINT n = 0;
|
|
|
|
res = f_read(&fileSrc, tmpBuffer, sz, &n);
|
|
if (n != sz)
|
|
res = FR_INT_ERR; // should not happen
|
|
|
|
if (res == FR_OK)
|
|
{
|
|
res = f_write(&fileDst, tmpBuffer, sz, &n);
|
|
if (n != sz)
|
|
res = FR_DENIED; // disk full
|
|
}
|
|
|
|
if (res != FR_OK)
|
|
{
|
|
f_close(&fileSrc);
|
|
f_close(&fileDst);
|
|
f_unlink(pathDst); // oops, failed
|
|
return false;
|
|
}
|
|
rem -= sz;
|
|
}
|
|
|
|
f_close(&fileSrc);
|
|
f_close(&fileDst);
|
|
|
|
return true;
|
|
}
|
|
|
|
bool findPayload(char *path, u32 pressed)
|
|
{
|
|
const char *pattern;
|
|
|
|
if(pressed & BUTTON_LEFT) pattern = PATTERN("left");
|
|
else if(pressed & BUTTON_RIGHT) pattern = PATTERN("right");
|
|
else if(pressed & BUTTON_UP) pattern = PATTERN("up");
|
|
else if(pressed & BUTTON_DOWN) pattern = PATTERN("down");
|
|
else if(pressed & BUTTON_START) pattern = PATTERN("start");
|
|
else if(pressed & BUTTON_B) pattern = PATTERN("b");
|
|
else if(pressed & BUTTON_X) pattern = PATTERN("x");
|
|
else if(pressed & BUTTON_Y) pattern = PATTERN("y");
|
|
else if(pressed & BUTTON_R1) pattern = PATTERN("r");
|
|
else if(pressed & BUTTON_A) pattern = PATTERN("a");
|
|
else pattern = PATTERN("select");
|
|
|
|
DIR dir;
|
|
FILINFO info;
|
|
FRESULT result;
|
|
|
|
result = f_findfirst(&dir, &info, "payloads", pattern);
|
|
|
|
if(result != FR_OK) return false;
|
|
|
|
f_closedir(&dir);
|
|
|
|
if(!info.fname[0]) return false;
|
|
|
|
sprintf(path, "payloads/%s", info.fname);
|
|
|
|
return true;
|
|
}
|
|
|
|
bool payloadMenu(char *path, bool *hasDisplayedMenu)
|
|
{
|
|
DIR dir;
|
|
|
|
*hasDisplayedMenu = false;
|
|
if(f_opendir(&dir, "payloads") != FR_OK) return false;
|
|
|
|
FILINFO info;
|
|
u32 payloadNum = 0;
|
|
char payloadList[20][49];
|
|
|
|
while(f_readdir(&dir, &info) == FR_OK && info.fname[0] != 0 && payloadNum < 20)
|
|
{
|
|
if(info.fname[0] == '.') continue;
|
|
|
|
u32 nameLength = strlen(info.fname);
|
|
|
|
if(nameLength < 6 || nameLength > 52) continue;
|
|
|
|
nameLength -= 5;
|
|
|
|
if(memcmp(info.fname + nameLength, ".firm", 5) != 0) continue;
|
|
|
|
memcpy(payloadList[payloadNum], info.fname, nameLength);
|
|
payloadList[payloadNum][nameLength] = 0;
|
|
payloadNum++;
|
|
}
|
|
|
|
if(f_closedir(&dir) != FR_OK || !payloadNum) return false;
|
|
|
|
u32 pressed = 0,
|
|
selectedPayload = 0;
|
|
|
|
if(payloadNum != 1)
|
|
{
|
|
initScreens();
|
|
*hasDisplayedMenu = true;
|
|
|
|
drawString(true, 10, 10, COLOR_TITLE, "Luma3DS chainloader");
|
|
drawString(true, 10, 10 + SPACING_Y, COLOR_TITLE, "Press A to select, START to quit");
|
|
|
|
for(u32 i = 0, posY = 10 + 3 * SPACING_Y, color = COLOR_RED; i < payloadNum; i++, posY += SPACING_Y)
|
|
{
|
|
drawString(true, 10, posY, color, payloadList[i]);
|
|
if(color == COLOR_RED) color = COLOR_WHITE;
|
|
}
|
|
|
|
while(pressed != BUTTON_A && pressed != BUTTON_START)
|
|
{
|
|
do
|
|
{
|
|
pressed = waitInput(true) & MENU_BUTTONS;
|
|
}
|
|
while(!pressed);
|
|
|
|
u32 oldSelectedPayload = selectedPayload;
|
|
|
|
switch(pressed)
|
|
{
|
|
case BUTTON_UP:
|
|
selectedPayload = !selectedPayload ? payloadNum - 1 : selectedPayload - 1;
|
|
break;
|
|
case BUTTON_DOWN:
|
|
selectedPayload = selectedPayload == payloadNum - 1 ? 0 : selectedPayload + 1;
|
|
break;
|
|
case BUTTON_LEFT:
|
|
selectedPayload = 0;
|
|
break;
|
|
case BUTTON_RIGHT:
|
|
selectedPayload = payloadNum - 1;
|
|
break;
|
|
default:
|
|
continue;
|
|
}
|
|
|
|
if(oldSelectedPayload == selectedPayload) continue;
|
|
|
|
drawString(true, 10, 10 + (3 + oldSelectedPayload) * SPACING_Y, COLOR_WHITE, payloadList[oldSelectedPayload]);
|
|
drawString(true, 10, 10 + (3 + selectedPayload) * SPACING_Y, COLOR_RED, payloadList[selectedPayload]);
|
|
}
|
|
}
|
|
|
|
if(pressed != BUTTON_START)
|
|
{
|
|
sprintf(path, "payloads/%s.firm", payloadList[selectedPayload]);
|
|
|
|
return true;
|
|
}
|
|
|
|
while(HID_PAD & MENU_BUTTONS);
|
|
wait(2000ULL);
|
|
|
|
return false;
|
|
}
|
|
|
|
u32 firmRead(void *dest, u32 firmType)
|
|
{
|
|
static const char *firmFolders[][2] = {{"00000002", "20000002"},
|
|
{"00000102", "20000102"},
|
|
{"00000202", "20000202"},
|
|
{"00000003", "20000003"},
|
|
{"00000001", "20000001"}};
|
|
|
|
char folderPath[64],
|
|
path[128];
|
|
|
|
sprintf(folderPath, "nand:/title/00040138/%s/content", firmFolders[firmType][ISN3DS ? 1 : 0]);
|
|
|
|
DIR dir;
|
|
u32 firmVersion = 0xFFFFFFFF;
|
|
|
|
if(f_opendir(&dir, folderPath) != FR_OK) goto exit;
|
|
|
|
FILINFO info;
|
|
|
|
//Parse the target directory
|
|
while(f_readdir(&dir, &info) == FR_OK && info.fname[0] != 0)
|
|
{
|
|
//Not a cxi
|
|
if(info.fname[9] != 'a' || strlen(info.fname) != 12) continue;
|
|
|
|
u32 tempVersion = hexAtoi(info.altname, 8);
|
|
|
|
//Found an older cxi
|
|
if(tempVersion < firmVersion) firmVersion = tempVersion;
|
|
}
|
|
|
|
if(f_closedir(&dir) != FR_OK || firmVersion == 0xFFFFFFFF) goto exit;
|
|
|
|
//Complete the string with the .app name
|
|
sprintf(path, "%s/%08lx.app", folderPath, firmVersion);
|
|
|
|
if(fileRead(dest, path, 0x400000 + sizeof(Cxi) + 0x200) <= sizeof(Cxi) + 0x400) firmVersion = 0xFFFFFFFF;
|
|
|
|
exit:
|
|
return firmVersion;
|
|
}
|
|
|
|
void findDumpFile(const char *folderPath, char *fileName)
|
|
{
|
|
DIR dir;
|
|
FRESULT result;
|
|
|
|
for(u32 n = 0; n <= 99999999; n++)
|
|
{
|
|
FILINFO info;
|
|
|
|
sprintf(fileName, "crash_dump_%08lu.dmp", n);
|
|
result = f_findfirst(&dir, &info, folderPath, fileName);
|
|
|
|
if(result != FR_OK || !info.fname[0]) break;
|
|
}
|
|
|
|
if(result == FR_OK) f_closedir(&dir);
|
|
}
|
|
|
|
static u8 fileCopyBuffer[0x10000];
|
|
|
|
static const u8 boot9Sha256[32] = {
|
|
0x2F, 0x88, 0x74, 0x4F, 0xEE, 0xD7, 0x17, 0x85, 0x63, 0x86, 0x40, 0x0A, 0x44, 0xBB, 0xA4, 0xB9,
|
|
0xCA, 0x62, 0xE7, 0x6A, 0x32, 0xC7, 0x15, 0xD4, 0xF3, 0x09, 0xC3, 0x99, 0xBF, 0x28, 0x16, 0x6F
|
|
};
|
|
|
|
static const u8 boot11Sha256[32] = {
|
|
0x74, 0xDA, 0xAC, 0xE1, 0xF8, 0x06, 0x7B, 0x66, 0xCC, 0x81, 0xFC, 0x30, 0x7A, 0x3F, 0xDB, 0x50,
|
|
0x9C, 0xBE, 0xDC, 0x32, 0xF9, 0x03, 0xAE, 0xBE, 0x90, 0x61, 0x44, 0xDE, 0xA7, 0xA0, 0x75, 0x12
|
|
};
|
|
|
|
static bool backupEssentialFiles(void)
|
|
{
|
|
size_t sz = sizeof(fileCopyBuffer);
|
|
|
|
bool ok = !(isSdMode && !mountFs(false, false));
|
|
|
|
ok = ok && fileCopy("nand:/ro/sys/HWCAL0.dat", "backups/HWCAL0.dat", false, fileCopyBuffer, sz);
|
|
ok = ok && fileCopy("nand:/ro/sys/HWCAL1.dat", "backups/HWCAL1.dat", false, fileCopyBuffer, sz);
|
|
|
|
ok = ok && fileCopy("nand:/rw/sys/LocalFriendCodeSeed_A", "backups/LocalFriendCodeSeed_A", false, fileCopyBuffer, sz); // often doesn't exist
|
|
ok = ok && fileCopy("nand:/rw/sys/LocalFriendCodeSeed_B", "backups/LocalFriendCodeSeed_B", false, fileCopyBuffer, sz);
|
|
|
|
ok = ok && fileCopy("nand:/rw/sys/SecureInfo_A", "backups/SecureInfo_A", false, fileCopyBuffer, sz);
|
|
ok = ok && fileCopy("nand:/rw/sys/SecureInfo_B", "backups/SecureInfo_B", false, fileCopyBuffer, sz); // often doesn't exist
|
|
|
|
if (!ok) return false;
|
|
|
|
alignedseqmemcpy(fileCopyBuffer, (const void *)0x10012000, 0x100);
|
|
if (getFileSize("backups/otp.bin") != 0x100)
|
|
ok = ok && fileWrite(fileCopyBuffer, "backups/otp.bin", 0x100);
|
|
|
|
if (!ok) return false;
|
|
|
|
// On dev boards, but not O3DS IS_DEBUGGER, hwcal is on an EEPROM chip accessed via I2C
|
|
u8 c = mcuConsoleInfo[0];
|
|
if (c == 2 || c == 4 || (ISN3DS && c == 5) || c == 6)
|
|
{
|
|
I2C_readRegBuf(I2C_DEV_EEPROM, 0, fileCopyBuffer, 0x1000); // Up to two instances of hwcal, with the second one @0x800
|
|
if (getFileSize("backups/HWCAL_01_EEPROM.dat") != 0x1000)
|
|
ok = ok && fileWrite(fileCopyBuffer, "backups/HWCAL_01_EEPROM.dat", 0x1000);
|
|
}
|
|
|
|
// B9S bootrom backups
|
|
u32 hash[32/4];
|
|
sha(hash, (const void *)0x08080000, 0x10000, SHA_256_MODE);
|
|
if (memcmp(hash, boot9Sha256, 32) == 0 && getFileSize("backups/boot9.bin") != 0x10000)
|
|
ok = ok && fileWrite((const void *)0x08080000, "backups/boot9.bin", 0x10000);
|
|
sha(hash, (const void *)0x08090000, 0x10000, SHA_256_MODE);
|
|
if (memcmp(hash, boot11Sha256, 32) == 0 && getFileSize("backups/boot11.bin") != 0x10000)
|
|
ok = ok && fileWrite((const void *)0x08090000, "backups/boot11.bin", 0x10000);
|
|
|
|
return ok;
|
|
}
|
|
|
|
bool doLumaUpgradeProcess(void)
|
|
{
|
|
// Ensure CTRNAND is mounted
|
|
bool ok = mountFs(false, false), ok2 = true;
|
|
if (!ok)
|
|
return false;
|
|
|
|
// Try to boot.firm to CTRNAND, when applicable
|
|
if (isSdMode && memcmp(launchedPathForFatfs, "sdmc:", 5) == 0)
|
|
ok = fileCopy(launchedPathForFatfs, "nand:/boot.firm", true, fileCopyBuffer, sizeof(fileCopyBuffer));
|
|
|
|
// Try to backup essential files
|
|
ok2 = backupEssentialFiles();
|
|
|
|
// Clean up some of the old files
|
|
fileDelete("sdmc:/luma/config.bin");
|
|
fileDelete("nand:/rw/luma/config.bin");
|
|
|
|
return ok && ok2;
|
|
}
|