mirror of
https://github.com/DarkMatterCore/nxdumptool.git
synced 2024-11-23 02:36:41 +00:00
Extracted RomFS dumping via USB.
This commit is contained in:
parent
b39ab27a52
commit
a02c806b60
5 changed files with 181 additions and 168 deletions
|
@ -144,16 +144,16 @@ NX_INLINE RomFileSystemFileEntry *bktrGetFileEntryByPath(BktrContext *ctx, const
|
||||||
return romfsGetFileEntryByPath(&(ctx->patch_romfs_ctx), path);
|
return romfsGetFileEntryByPath(&(ctx->patch_romfs_ctx), path);
|
||||||
}
|
}
|
||||||
|
|
||||||
NX_INLINE bool bktrGeneratePathFromDirectoryEntry(BktrContext *ctx, RomFileSystemDirectoryEntry *dir_entry, char *out_path, size_t out_path_size)
|
NX_INLINE bool bktrGeneratePathFromDirectoryEntry(BktrContext *ctx, RomFileSystemDirectoryEntry *dir_entry, char *out_path, size_t out_path_size, u8 illegal_char_replace_type)
|
||||||
{
|
{
|
||||||
if (!ctx) return false;
|
if (!ctx) return false;
|
||||||
return romfsGeneratePathFromDirectoryEntry(&(ctx->patch_romfs_ctx), dir_entry, out_path, out_path_size);
|
return romfsGeneratePathFromDirectoryEntry(&(ctx->patch_romfs_ctx), dir_entry, out_path, out_path_size, illegal_char_replace_type);
|
||||||
}
|
}
|
||||||
|
|
||||||
NX_INLINE bool bktrGeneratePathFromFileEntry(BktrContext *ctx, RomFileSystemFileEntry *file_entry, char *out_path, size_t out_path_size)
|
NX_INLINE bool bktrGeneratePathFromFileEntry(BktrContext *ctx, RomFileSystemFileEntry *file_entry, char *out_path, size_t out_path_size, u8 illegal_char_replace_type)
|
||||||
{
|
{
|
||||||
if (!ctx) return false;
|
if (!ctx) return false;
|
||||||
return romfsGeneratePathFromFileEntry(&(ctx->patch_romfs_ctx), file_entry, out_path, out_path_size);
|
return romfsGeneratePathFromFileEntry(&(ctx->patch_romfs_ctx), file_entry, out_path, out_path_size, illegal_char_replace_type);
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif /* __BKTR_H__ */
|
#endif /* __BKTR_H__ */
|
||||||
|
|
309
source/main.c
309
source/main.c
|
@ -23,8 +23,6 @@
|
||||||
#include <threads.h>
|
#include <threads.h>
|
||||||
#include <stdarg.h>
|
#include <stdarg.h>
|
||||||
|
|
||||||
//#include "lvgl_helper.h"
|
|
||||||
|
|
||||||
#include "utils.h"
|
#include "utils.h"
|
||||||
#include "bktr.h"
|
#include "bktr.h"
|
||||||
#include "gamecard.h"
|
#include "gamecard.h"
|
||||||
|
@ -122,13 +120,15 @@ static CondVar g_readCondvar = 0, g_writeCondvar = 0;
|
||||||
|
|
||||||
typedef struct
|
typedef struct
|
||||||
{
|
{
|
||||||
FILE *fileobj;
|
//FILE *fileobj;
|
||||||
|
RomFileSystemContext *romfs_ctx;
|
||||||
void *data;
|
void *data;
|
||||||
size_t data_size;
|
size_t data_size;
|
||||||
size_t data_written;
|
size_t data_written;
|
||||||
size_t total_size;
|
size_t total_size;
|
||||||
bool read_error;
|
bool read_error;
|
||||||
bool write_error;
|
bool write_error;
|
||||||
|
bool transfer_cancelled;
|
||||||
} ThreadSharedData;
|
} ThreadSharedData;
|
||||||
|
|
||||||
static void consolePrint(const char *text, ...)
|
static void consolePrint(const char *text, ...)
|
||||||
|
@ -156,37 +156,89 @@ static int read_thread_func(void *arg)
|
||||||
return -2;
|
return -2;
|
||||||
}
|
}
|
||||||
|
|
||||||
for(u64 offset = 0, blksize = TEST_BUF_SIZE; offset < shared_data->total_size; offset += blksize)
|
u64 file_table_offset = 0;
|
||||||
{
|
RomFileSystemFileEntry *file_entry = NULL;
|
||||||
if (blksize > (shared_data->total_size - offset)) blksize = (shared_data->total_size - offset);
|
char path[FS_MAX_PATH] = {0};
|
||||||
|
|
||||||
shared_data->read_error = !gamecardReadStorage(buf, blksize, offset);
|
while(file_table_offset < shared_data->romfs_ctx->file_table_size)
|
||||||
|
{
|
||||||
|
/* Check if the transfer has been cancelled by the user */
|
||||||
|
if (shared_data->transfer_cancelled)
|
||||||
|
{
|
||||||
|
condvarWakeAll(&g_writeCondvar);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Retrieve RomFS file entry information */
|
||||||
|
shared_data->read_error = (!(file_entry = romfsGetFileEntryByOffset(shared_data->romfs_ctx, file_table_offset)) || \
|
||||||
|
!romfsGeneratePathFromFileEntry(shared_data->romfs_ctx, file_entry, path, FS_MAX_PATH, RomFileSystemPathIllegalCharReplaceType_IllegalFsChars));
|
||||||
if (shared_data->read_error)
|
if (shared_data->read_error)
|
||||||
{
|
{
|
||||||
condvarWakeAll(&g_writeCondvar);
|
condvarWakeAll(&g_writeCondvar);
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/* Wait until the previous file data chunk has been written */
|
||||||
mutexLock(&g_fileMutex);
|
mutexLock(&g_fileMutex);
|
||||||
|
|
||||||
if (shared_data->data_size && !shared_data->write_error) condvarWait(&g_readCondvar, &g_fileMutex);
|
if (shared_data->data_size && !shared_data->write_error) condvarWait(&g_readCondvar, &g_fileMutex);
|
||||||
|
mutexUnlock(&g_fileMutex);
|
||||||
|
if (shared_data->write_error) break;
|
||||||
|
|
||||||
if (shared_data->write_error)
|
/* Send current file properties */
|
||||||
|
shared_data->read_error = !usbSendFileProperties(file_entry->size, path);
|
||||||
|
if (shared_data->read_error)
|
||||||
{
|
{
|
||||||
mutexUnlock(&g_fileMutex);
|
condvarWakeAll(&g_writeCondvar);
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
memcpy(shared_data->data, buf, blksize);
|
for(u64 offset = 0, blksize = TEST_BUF_SIZE; offset < file_entry->size; offset += blksize)
|
||||||
shared_data->data_size = blksize;
|
{
|
||||||
|
if (blksize > (file_entry->size - offset)) blksize = (file_entry->size - offset);
|
||||||
|
|
||||||
mutexUnlock(&g_fileMutex);
|
/* Check if the transfer has been cancelled by the user */
|
||||||
condvarWakeAll(&g_writeCondvar);
|
if (shared_data->transfer_cancelled)
|
||||||
|
{
|
||||||
|
condvarWakeAll(&g_writeCondvar);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Read current file data chunk */
|
||||||
|
shared_data->read_error = !romfsReadFileEntryData(shared_data->romfs_ctx, file_entry, buf, blksize, offset);
|
||||||
|
if (shared_data->read_error)
|
||||||
|
{
|
||||||
|
condvarWakeAll(&g_writeCondvar);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Wait until the previous file data chunk has been written */
|
||||||
|
mutexLock(&g_fileMutex);
|
||||||
|
|
||||||
|
if (shared_data->data_size && !shared_data->write_error) condvarWait(&g_readCondvar, &g_fileMutex);
|
||||||
|
|
||||||
|
if (shared_data->write_error)
|
||||||
|
{
|
||||||
|
mutexUnlock(&g_fileMutex);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Copy current file data chunk to the shared buffer */
|
||||||
|
memcpy(shared_data->data, buf, blksize);
|
||||||
|
shared_data->data_size = blksize;
|
||||||
|
|
||||||
|
/* Wake up the write thread to continue writing data */
|
||||||
|
mutexUnlock(&g_fileMutex);
|
||||||
|
condvarWakeAll(&g_writeCondvar);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (shared_data->read_error || shared_data->write_error || shared_data->transfer_cancelled) break;
|
||||||
|
|
||||||
|
file_table_offset += ALIGN_UP(sizeof(RomFileSystemFileEntry) + file_entry->name_length, 4);
|
||||||
}
|
}
|
||||||
|
|
||||||
free(buf);
|
free(buf);
|
||||||
|
|
||||||
return 0;
|
return (shared_data->read_error ? -3 : 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
static int write_thread_func(void *arg)
|
static int write_thread_func(void *arg)
|
||||||
|
@ -200,11 +252,12 @@ static int write_thread_func(void *arg)
|
||||||
|
|
||||||
while(shared_data->data_written < shared_data->total_size)
|
while(shared_data->data_written < shared_data->total_size)
|
||||||
{
|
{
|
||||||
|
/* Wait until the current file data chunk has been read */
|
||||||
mutexLock(&g_fileMutex);
|
mutexLock(&g_fileMutex);
|
||||||
|
|
||||||
if (!shared_data->data_size && !shared_data->read_error) condvarWait(&g_writeCondvar, &g_fileMutex);
|
if (!shared_data->data_size && !shared_data->read_error) condvarWait(&g_writeCondvar, &g_fileMutex);
|
||||||
|
|
||||||
if (shared_data->read_error)
|
if (shared_data->read_error || shared_data->transfer_cancelled)
|
||||||
{
|
{
|
||||||
mutexUnlock(&g_fileMutex);
|
mutexUnlock(&g_fileMutex);
|
||||||
break;
|
break;
|
||||||
|
@ -212,6 +265,7 @@ static int write_thread_func(void *arg)
|
||||||
|
|
||||||
//shared_data->write_error = (fwrite(shared_data->data, 1, shared_data->data_size, shared_data->fileobj) != shared_data->data_size);
|
//shared_data->write_error = (fwrite(shared_data->data, 1, shared_data->data_size, shared_data->fileobj) != shared_data->data_size);
|
||||||
|
|
||||||
|
/* Write current file data chunk */
|
||||||
shared_data->write_error = !usbSendFileData(shared_data->data, shared_data->data_size);
|
shared_data->write_error = !usbSendFileData(shared_data->data, shared_data->data_size);
|
||||||
if (!shared_data->write_error)
|
if (!shared_data->write_error)
|
||||||
{
|
{
|
||||||
|
@ -219,6 +273,7 @@ static int write_thread_func(void *arg)
|
||||||
shared_data->data_size = 0;
|
shared_data->data_size = 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/* Wake up the read thread to continue reading data */
|
||||||
mutexUnlock(&g_fileMutex);
|
mutexUnlock(&g_fileMutex);
|
||||||
condvarWakeAll(&g_readCondvar);
|
condvarWakeAll(&g_readCondvar);
|
||||||
|
|
||||||
|
@ -247,55 +302,25 @@ int main(int argc, char *argv[])
|
||||||
goto out;
|
goto out;
|
||||||
}
|
}
|
||||||
|
|
||||||
/*lv_test();
|
|
||||||
|
|
||||||
while(appletMainLoop())
|
|
||||||
{
|
|
||||||
lv_task_handler();
|
|
||||||
if (lvglHelperGetExitFlag()) break;
|
|
||||||
}*/
|
|
||||||
|
|
||||||
u8 *buf = NULL;
|
u8 *buf = NULL;
|
||||||
FILE *tmp_file = NULL;
|
|
||||||
|
|
||||||
Ticket base_tik = {0}, update_tik = {0};
|
Ticket tik = {0};
|
||||||
NcaContext *base_nca_ctx = NULL, *update_nca_ctx = NULL;
|
NcaContext *nca_ctx = NULL;
|
||||||
NcmContentStorage ncm_storage = {0};
|
NcmContentStorage ncm_storage = {0};
|
||||||
|
|
||||||
BktrContext bktr_ctx = {0};
|
|
||||||
RomFileSystemFileEntry *bktr_file_entry = NULL;
|
|
||||||
|
|
||||||
Result rc = 0;
|
Result rc = 0;
|
||||||
|
|
||||||
ThreadSharedData shared_data = {0};
|
ThreadSharedData shared_data = {0};
|
||||||
thrd_t read_thread, write_thread;
|
thrd_t read_thread, write_thread;
|
||||||
|
|
||||||
|
char path[FS_MAX_PATH] = {0};
|
||||||
|
LrLocationResolver resolver = {0};
|
||||||
|
NcmContentInfo content_info = {0};
|
||||||
|
|
||||||
|
RomFileSystemContext romfs_ctx = {0};
|
||||||
|
|
||||||
//mkdir("sdmc:/nxdt_test", 0744);
|
//mkdir("sdmc:/nxdt_test", 0744);
|
||||||
|
|
||||||
// SSBU's Base Program NCA
|
|
||||||
NcmContentInfo base_program_content_info = {
|
|
||||||
.content_id = {
|
|
||||||
.c = { 0x48, 0xBB, 0xEA, 0xB6, 0x3E, 0x73, 0x88, 0x69, 0x8D, 0xE4, 0x74, 0x43, 0x49, 0x00, 0xE1, 0x04 }
|
|
||||||
},
|
|
||||||
.size = {
|
|
||||||
0x00, 0x40, 0x24, 0x62, 0x03, 0x00
|
|
||||||
},
|
|
||||||
.content_type = NcmContentType_Program,
|
|
||||||
.id_offset = 0
|
|
||||||
};
|
|
||||||
|
|
||||||
// SSBU's Update Program NCA
|
|
||||||
NcmContentInfo update_program_content_info = {
|
|
||||||
.content_id = {
|
|
||||||
.c = { 0x83, 0x7A, 0xD9, 0x78, 0x3E, 0xB2, 0x3A, 0xFA, 0xB0, 0x4B, 0xF5, 0x71, 0x55, 0x43, 0x6E, 0x5D }
|
|
||||||
},
|
|
||||||
.size = {
|
|
||||||
0x00, 0x6E, 0xE9, 0x84, 0x00, 0x00
|
|
||||||
},
|
|
||||||
.content_type = NcmContentType_Program,
|
|
||||||
.id_offset = 0
|
|
||||||
};
|
|
||||||
|
|
||||||
buf = usbAllocatePageAlignedBuffer(TEST_BUF_SIZE);
|
buf = usbAllocatePageAlignedBuffer(TEST_BUF_SIZE);
|
||||||
if (!buf)
|
if (!buf)
|
||||||
{
|
{
|
||||||
|
@ -305,23 +330,14 @@ int main(int argc, char *argv[])
|
||||||
|
|
||||||
consolePrint("buf succeeded\n");
|
consolePrint("buf succeeded\n");
|
||||||
|
|
||||||
base_nca_ctx = calloc(1, sizeof(NcaContext));
|
nca_ctx = calloc(1, sizeof(NcaContext));
|
||||||
if (!base_nca_ctx)
|
if (!nca_ctx)
|
||||||
{
|
{
|
||||||
consolePrint("base nca ctx buf failed\n");
|
consolePrint("nca ctx buf failed\n");
|
||||||
goto out2;
|
goto out2;
|
||||||
}
|
}
|
||||||
|
|
||||||
consolePrint("base nca ctx buf succeeded\n");
|
consolePrint("nca ctx buf succeeded\n");
|
||||||
|
|
||||||
update_nca_ctx = calloc(1, sizeof(NcaContext));
|
|
||||||
if (!update_nca_ctx)
|
|
||||||
{
|
|
||||||
consolePrint("update nca ctx buf failed\n");
|
|
||||||
goto out2;
|
|
||||||
}
|
|
||||||
|
|
||||||
consolePrint("update nca ctx buf succeeded\n");
|
|
||||||
|
|
||||||
rc = ncmOpenContentStorage(&ncm_storage, NcmStorageId_SdCard);
|
rc = ncmOpenContentStorage(&ncm_storage, NcmStorageId_SdCard);
|
||||||
if (R_FAILED(rc))
|
if (R_FAILED(rc))
|
||||||
|
@ -332,108 +348,80 @@ int main(int argc, char *argv[])
|
||||||
|
|
||||||
consolePrint("ncm open storage succeeded\n");
|
consolePrint("ncm open storage succeeded\n");
|
||||||
|
|
||||||
/*if (!ncaInitializeContext(base_nca_ctx, NcmStorageId_SdCard, &ncm_storage, 0, &base_program_content_info, &base_tik))
|
rc = lrInitialize();
|
||||||
|
if (R_FAILED(rc))
|
||||||
{
|
{
|
||||||
consolePrint("base nca initialize ctx failed\n");
|
consolePrint("lrInitialize failed\n");
|
||||||
goto out2;
|
goto out2;
|
||||||
}
|
}
|
||||||
|
|
||||||
tmp_file = fopen("sdmc:/nxdt_test/base_nca_ctx.bin", "wb");
|
consolePrint("lrInitialize succeeded\n");
|
||||||
if (tmp_file)
|
|
||||||
{
|
|
||||||
fwrite(base_nca_ctx, 1, sizeof(NcaContext), tmp_file);
|
|
||||||
fclose(tmp_file);
|
|
||||||
tmp_file = NULL;
|
|
||||||
consolePrint("base nca ctx saved\n");
|
|
||||||
} else {
|
|
||||||
consolePrint("base nca ctx not saved\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!ncaInitializeContext(update_nca_ctx, NcmStorageId_SdCard, &ncm_storage, 0, &update_program_content_info, &update_tik))
|
rc = lrOpenLocationResolver(NcmStorageId_SdCard, &resolver);
|
||||||
|
if (R_FAILED(rc))
|
||||||
{
|
{
|
||||||
consolePrint("update nca initialize ctx failed\n");
|
consolePrint("lrOpenLocationResolver failed\n");
|
||||||
goto out2;
|
goto out2;
|
||||||
}
|
}
|
||||||
|
|
||||||
tmp_file = fopen("sdmc:/nxdt_test/update_nca_ctx.bin", "wb");
|
consolePrint("lrOpenLocationResolver succeeded\n");
|
||||||
if (tmp_file)
|
|
||||||
{
|
|
||||||
fwrite(update_nca_ctx, 1, sizeof(NcaContext), tmp_file);
|
|
||||||
fclose(tmp_file);
|
|
||||||
tmp_file = NULL;
|
|
||||||
consolePrint("update nca ctx saved\n");
|
|
||||||
} else {
|
|
||||||
consolePrint("update nca ctx not saved\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!bktrInitializeContext(&bktr_ctx, &(base_nca_ctx->fs_contexts[1]), &(update_nca_ctx->fs_contexts[1])))
|
rc = lrLrResolveProgramPath(&resolver, (u64)0x01006F8002326000, path); // ACNH 0x01006F8002326000 | Smash 0x01006A800016E000 | Dark Souls 0x01004AB00A260000
|
||||||
|
if (R_FAILED(rc))
|
||||||
{
|
{
|
||||||
consolePrint("bktr initialize ctx failed\n");
|
consolePrint("lrLrResolveProgramPath failed\n");
|
||||||
goto out2;
|
goto out2;
|
||||||
}
|
}
|
||||||
|
|
||||||
consolePrint("bktr initialize ctx succeeded\n");
|
consolePrint("lrLrResolveProgramPath succeeded\n");
|
||||||
|
|
||||||
tmp_file = fopen("sdmc:/nxdt_test/bktr_ctx.bin", "wb");
|
memmove(path, strrchr(path, '/') + 1, SHA256_HASH_SIZE + 4);
|
||||||
if (tmp_file)
|
path[SHA256_HASH_SIZE + 4] = '\0';
|
||||||
|
|
||||||
|
consolePrint("Program NCA: %s\n", path);
|
||||||
|
|
||||||
|
for(u32 i = 0; i < SHA256_HASH_SIZE; i++)
|
||||||
{
|
{
|
||||||
fwrite(&bktr_ctx, 1, sizeof(BktrContext), tmp_file);
|
char val = (('a' <= path[i] && path[i] <= 'f') ? (path[i] - 'a' + 0xA) : (path[i] - '0'));
|
||||||
fclose(tmp_file);
|
if ((i & 1) == 0) val <<= 4;
|
||||||
tmp_file = NULL;
|
content_info.content_id.c[i >> 1] |= val;
|
||||||
consolePrint("bktr ctx saved\n");
|
|
||||||
} else {
|
|
||||||
consolePrint("bktr ctx not saved\n");
|
|
||||||
}
|
}
|
||||||
|
|
||||||
bktr_file_entry = bktrGetFileEntryByPath(&bktr_ctx, "/data.arc");
|
content_info.content_type = NcmContentType_Program;
|
||||||
if (!bktr_file_entry)
|
|
||||||
|
u64 content_size = 0;
|
||||||
|
rc = ncmContentStorageGetSizeFromContentId(&ncm_storage, (s64*)&content_size, &(content_info.content_id));
|
||||||
|
if (R_FAILED(rc))
|
||||||
{
|
{
|
||||||
consolePrint("bktr get file entry by path failed\n");
|
consolePrint("ncmContentStorageGetSizeFromContentId failed\n");
|
||||||
goto out2;
|
goto out2;
|
||||||
}
|
}
|
||||||
|
|
||||||
consolePrint("bktr get file entry by path success: %.*s | 0x%lX\n", bktr_file_entry->name_length, bktr_file_entry->name, bktr_file_entry->size);*/
|
consolePrint("ncmContentStorageGetSizeFromContentId succeeded\n");
|
||||||
|
|
||||||
|
memcpy(&(content_info.size), &content_size, 6);
|
||||||
|
|
||||||
|
if (!ncaInitializeContext(nca_ctx, NcmStorageId_SdCard, &ncm_storage, 0, &content_info, &tik))
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/*if (!utilsCreateConcatenationFile("sdmc:/nxdt_test/gamecard.xci"))
|
|
||||||
{
|
{
|
||||||
consolePrint("create concatenationfile failed\n");
|
consolePrint("nca initialize ctx failed\n");
|
||||||
goto out2;
|
goto out2;
|
||||||
}
|
}
|
||||||
|
|
||||||
consolePrint("create concatenationfile success\n");
|
consolePrint("nca initialize ctx succeeded\n");
|
||||||
|
|
||||||
tmp_file = fopen("sdmc:/nxdt_test/gamecard.xci", "wb");
|
if (!romfsInitializeContext(&romfs_ctx, &(nca_ctx->fs_contexts[1])))
|
||||||
if (!tmp_file)
|
|
||||||
{
|
{
|
||||||
consolePrint("open concatenationfile failed\n");
|
consolePrint("romfs initialize ctx failed\n");
|
||||||
goto out2;
|
goto out2;
|
||||||
}
|
}
|
||||||
|
|
||||||
consolePrint("open concatenationfile success\n");*/
|
consolePrint("romfs initialize ctx succeeded\n");
|
||||||
|
|
||||||
|
shared_data.romfs_ctx = &romfs_ctx;
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
consolePrint("waiting for gamecard to be ready...\n");
|
|
||||||
|
|
||||||
while(appletMainLoop())
|
|
||||||
{
|
|
||||||
if (gamecardIsReady()) break;
|
|
||||||
}
|
|
||||||
|
|
||||||
//shared_data.fileobj = tmp_file;
|
|
||||||
shared_data.fileobj = NULL;
|
|
||||||
shared_data.data = buf;
|
shared_data.data = buf;
|
||||||
shared_data.data_size = 0;
|
shared_data.data_size = 0;
|
||||||
shared_data.data_written = 0;
|
shared_data.data_written = 0;
|
||||||
gamecardGetTotalSize(&(shared_data.total_size));
|
romfsGetTotalDataSize(&romfs_ctx, &(shared_data.total_size));
|
||||||
|
|
||||||
consolePrint("waiting for usb connection... ");
|
consolePrint("waiting for usb connection... ");
|
||||||
|
|
||||||
|
@ -445,17 +433,7 @@ int main(int argc, char *argv[])
|
||||||
|
|
||||||
consolePrint("success\n");
|
consolePrint("success\n");
|
||||||
|
|
||||||
consolePrint("sending file properties... ");
|
consolePrint("creating threads\n");
|
||||||
|
|
||||||
if (!usbSendFileProperties(shared_data.total_size, "gamecard.xci"))
|
|
||||||
{
|
|
||||||
consolePrint("failed\n");
|
|
||||||
goto out2;
|
|
||||||
}
|
|
||||||
|
|
||||||
consolePrint("success\n");
|
|
||||||
|
|
||||||
consolePrint("creating threads\n\n");
|
|
||||||
thrd_create(&read_thread, read_thread_func, &shared_data);
|
thrd_create(&read_thread, read_thread_func, &shared_data);
|
||||||
thrd_create(&write_thread, write_thread_func, &shared_data);
|
thrd_create(&write_thread, write_thread_func, &shared_data);
|
||||||
|
|
||||||
|
@ -465,6 +443,11 @@ int main(int argc, char *argv[])
|
||||||
|
|
||||||
time_t start = time(NULL);
|
time_t start = time(NULL);
|
||||||
|
|
||||||
|
time_t btn_cancel_start_tmr = 0, btn_cancel_end_tmr = 0;
|
||||||
|
bool btn_cancel_cur_state = false, btn_cancel_prev_state = false;
|
||||||
|
|
||||||
|
consolePrint("hold b to cancel\n\n");
|
||||||
|
|
||||||
while(shared_data.data_written < shared_data.total_size)
|
while(shared_data.data_written < shared_data.total_size)
|
||||||
{
|
{
|
||||||
if (shared_data.read_error || shared_data.write_error) break;
|
if (shared_data.read_error || shared_data.write_error) break;
|
||||||
|
@ -473,6 +456,29 @@ int main(int argc, char *argv[])
|
||||||
struct tm *ts = localtime(&now);
|
struct tm *ts = localtime(&now);
|
||||||
size_t size = shared_data.data_written;
|
size_t size = shared_data.data_written;
|
||||||
|
|
||||||
|
hidScanInput();
|
||||||
|
btn_cancel_cur_state = (utilsHidKeysAllHeld() & KEY_B);
|
||||||
|
|
||||||
|
if (btn_cancel_cur_state && btn_cancel_cur_state != btn_cancel_prev_state)
|
||||||
|
{
|
||||||
|
btn_cancel_start_tmr = now;
|
||||||
|
} else
|
||||||
|
if (btn_cancel_cur_state && btn_cancel_cur_state == btn_cancel_prev_state)
|
||||||
|
{
|
||||||
|
btn_cancel_end_tmr = now;
|
||||||
|
if ((btn_cancel_end_tmr - btn_cancel_start_tmr) >= 3)
|
||||||
|
{
|
||||||
|
mutexLock(&g_fileMutex);
|
||||||
|
shared_data.transfer_cancelled = true;
|
||||||
|
mutexUnlock(&g_fileMutex);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
btn_cancel_start_tmr = btn_cancel_end_tmr = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
btn_cancel_prev_state = btn_cancel_cur_state;
|
||||||
|
|
||||||
if (prev_time == ts->tm_sec || prev_size == size) continue;
|
if (prev_time == ts->tm_sec || prev_size == size) continue;
|
||||||
|
|
||||||
percent = (u8)((size * 100) / shared_data.total_size);
|
percent = (u8)((size * 100) / shared_data.total_size);
|
||||||
|
@ -484,9 +490,6 @@ int main(int argc, char *argv[])
|
||||||
consoleUpdate(NULL);
|
consoleUpdate(NULL);
|
||||||
}
|
}
|
||||||
|
|
||||||
//fclose(tmp_file);
|
|
||||||
//tmp_file = NULL;
|
|
||||||
|
|
||||||
start = (time(NULL) - start);
|
start = (time(NULL) - start);
|
||||||
|
|
||||||
consolePrint("\nwaiting for threads to join\n");
|
consolePrint("\nwaiting for threads to join\n");
|
||||||
|
@ -501,6 +504,12 @@ int main(int argc, char *argv[])
|
||||||
goto out2;
|
goto out2;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (shared_data.transfer_cancelled)
|
||||||
|
{
|
||||||
|
consolePrint("process cancelled\n");
|
||||||
|
goto out2;
|
||||||
|
}
|
||||||
|
|
||||||
consolePrint("process completed in %lu seconds\n", start);
|
consolePrint("process completed in %lu seconds\n", start);
|
||||||
|
|
||||||
|
|
||||||
|
@ -519,15 +528,11 @@ out2:
|
||||||
consolePrint("press any button to exit\n");
|
consolePrint("press any button to exit\n");
|
||||||
utilsWaitForButtonPress();
|
utilsWaitForButtonPress();
|
||||||
|
|
||||||
if (tmp_file) fclose(tmp_file);
|
lrExit();
|
||||||
|
|
||||||
bktrFreeContext(&bktr_ctx);
|
|
||||||
|
|
||||||
if (serviceIsActive(&(ncm_storage.s))) ncmContentStorageClose(&ncm_storage);
|
if (serviceIsActive(&(ncm_storage.s))) ncmContentStorageClose(&ncm_storage);
|
||||||
|
|
||||||
if (update_nca_ctx) free(update_nca_ctx);
|
if (nca_ctx) free(nca_ctx);
|
||||||
|
|
||||||
if (base_nca_ctx) free(base_nca_ctx);
|
|
||||||
|
|
||||||
if (buf) free(buf);
|
if (buf) free(buf);
|
||||||
|
|
||||||
|
|
|
@ -359,14 +359,15 @@ out:
|
||||||
return file_entry;
|
return file_entry;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool romfsGeneratePathFromDirectoryEntry(RomFileSystemContext *ctx, RomFileSystemDirectoryEntry *dir_entry, char *out_path, size_t out_path_size)
|
bool romfsGeneratePathFromDirectoryEntry(RomFileSystemContext *ctx, RomFileSystemDirectoryEntry *dir_entry, char *out_path, size_t out_path_size, u8 illegal_char_replace_type)
|
||||||
{
|
{
|
||||||
size_t path_len = 0;
|
size_t path_len = 0;
|
||||||
u32 dir_offset = ROMFS_VOID_ENTRY, dir_entries_count = 0;
|
u32 dir_offset = ROMFS_VOID_ENTRY, dir_entries_count = 0;
|
||||||
RomFileSystemDirectoryEntry **dir_entries = NULL, **tmp_dir_entries = NULL;
|
RomFileSystemDirectoryEntry **dir_entries = NULL, **tmp_dir_entries = NULL;
|
||||||
bool success = false;
|
bool success = false;
|
||||||
|
|
||||||
if (!ctx || !ctx->dir_table || !ctx->dir_table_size || !dir_entry || (!dir_entry->name_length && dir_entry->parent_offset) || !out_path || out_path_size < 2)
|
if (!ctx || !ctx->dir_table || !ctx->dir_table_size || !dir_entry || (!dir_entry->name_length && dir_entry->parent_offset) || !out_path || out_path_size < 2 || \
|
||||||
|
illegal_char_replace_type > RomFileSystemPathIllegalCharReplaceType_KeepAsciiCharsOnly)
|
||||||
{
|
{
|
||||||
LOGFILE("Invalid parameters!");
|
LOGFILE("Invalid parameters!");
|
||||||
return false;
|
return false;
|
||||||
|
@ -428,6 +429,8 @@ bool romfsGeneratePathFromDirectoryEntry(RomFileSystemContext *ctx, RomFileSyste
|
||||||
{
|
{
|
||||||
strcat(out_path, "/");
|
strcat(out_path, "/");
|
||||||
strncat(out_path, dir_entries[i - 1]->name, dir_entries[i - 1]->name_length);
|
strncat(out_path, dir_entries[i - 1]->name, dir_entries[i - 1]->name_length);
|
||||||
|
if (illegal_char_replace_type) utilsReplaceIllegalCharacters(out_path + (strlen(out_path) - dir_entries[i - 1]->name_length), \
|
||||||
|
illegal_char_replace_type == RomFileSystemPathIllegalCharReplaceType_KeepAsciiCharsOnly);
|
||||||
}
|
}
|
||||||
|
|
||||||
success = true;
|
success = true;
|
||||||
|
@ -438,20 +441,20 @@ out:
|
||||||
return success;
|
return success;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool romfsGeneratePathFromFileEntry(RomFileSystemContext *ctx, RomFileSystemFileEntry *file_entry, char *out_path, size_t out_path_size)
|
bool romfsGeneratePathFromFileEntry(RomFileSystemContext *ctx, RomFileSystemFileEntry *file_entry, char *out_path, size_t out_path_size, u8 illegal_char_replace_type)
|
||||||
{
|
{
|
||||||
size_t path_len = 0;
|
size_t path_len = 0;
|
||||||
RomFileSystemDirectoryEntry *dir_entry = NULL;
|
RomFileSystemDirectoryEntry *dir_entry = NULL;
|
||||||
|
|
||||||
if (!ctx || !ctx->file_table || !ctx->file_table_size || !file_entry || !file_entry->name_length || !out_path || out_path_size < 2 || \
|
if (!ctx || !ctx->file_table || !ctx->file_table_size || !file_entry || !file_entry->name_length || !out_path || out_path_size < 2 || \
|
||||||
!(dir_entry = romfsGetDirectoryEntryByOffset(ctx, file_entry->parent_offset)))
|
!(dir_entry = romfsGetDirectoryEntryByOffset(ctx, file_entry->parent_offset)) || illegal_char_replace_type > RomFileSystemPathIllegalCharReplaceType_KeepAsciiCharsOnly)
|
||||||
{
|
{
|
||||||
LOGFILE("Invalid parameters!");
|
LOGFILE("Invalid parameters!");
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Retrieve full RomFS path up to the file entry name */
|
/* Retrieve full RomFS path up to the file entry name */
|
||||||
if (!romfsGeneratePathFromDirectoryEntry(ctx, dir_entry, out_path, out_path_size))
|
if (!romfsGeneratePathFromDirectoryEntry(ctx, dir_entry, out_path, out_path_size, illegal_char_replace_type))
|
||||||
{
|
{
|
||||||
LOGFILE("Failed to retrieve RomFS directory path!");
|
LOGFILE("Failed to retrieve RomFS directory path!");
|
||||||
return false;
|
return false;
|
||||||
|
@ -466,8 +469,10 @@ bool romfsGeneratePathFromFileEntry(RomFileSystemContext *ctx, RomFileSystemFile
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Concatenate file entry name */
|
/* Concatenate file entry name */
|
||||||
strcat(out_path, "/");
|
if (file_entry->parent_offset) strcat(out_path, "/");
|
||||||
strncat(out_path, file_entry->name, file_entry->name_length);
|
strncat(out_path, file_entry->name, file_entry->name_length);
|
||||||
|
if (illegal_char_replace_type) utilsReplaceIllegalCharacters(out_path + (strlen(out_path) - file_entry->name_length), \
|
||||||
|
illegal_char_replace_type == RomFileSystemPathIllegalCharReplaceType_KeepAsciiCharsOnly);
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
|
@ -106,6 +106,12 @@ typedef struct {
|
||||||
NcaHierarchicalIntegrityPatch cur_format_patch; ///< Used with NCA2/NCA3 RomFS sections.
|
NcaHierarchicalIntegrityPatch cur_format_patch; ///< Used with NCA2/NCA3 RomFS sections.
|
||||||
} RomFileSystemFileEntryPatch;
|
} RomFileSystemFileEntryPatch;
|
||||||
|
|
||||||
|
typedef enum {
|
||||||
|
RomFileSystemPathIllegalCharReplaceType_None = 0,
|
||||||
|
RomFileSystemPathIllegalCharReplaceType_IllegalFsChars = 1,
|
||||||
|
RomFileSystemPathIllegalCharReplaceType_KeepAsciiCharsOnly = 2
|
||||||
|
} RomFileSystemPathIllegalCharReplaceType;
|
||||||
|
|
||||||
/// Initializes a RomFS context.
|
/// Initializes a RomFS context.
|
||||||
bool romfsInitializeContext(RomFileSystemContext *out, NcaFsSectionContext *nca_fs_ctx);
|
bool romfsInitializeContext(RomFileSystemContext *out, NcaFsSectionContext *nca_fs_ctx);
|
||||||
|
|
||||||
|
@ -141,10 +147,10 @@ RomFileSystemDirectoryEntry *romfsGetDirectoryEntryByPath(RomFileSystemContext *
|
||||||
RomFileSystemFileEntry *romfsGetFileEntryByPath(RomFileSystemContext *ctx, const char *path);
|
RomFileSystemFileEntry *romfsGetFileEntryByPath(RomFileSystemContext *ctx, const char *path);
|
||||||
|
|
||||||
/// Generates a path string from a RomFS directory entry.
|
/// Generates a path string from a RomFS directory entry.
|
||||||
bool romfsGeneratePathFromDirectoryEntry(RomFileSystemContext *ctx, RomFileSystemDirectoryEntry *dir_entry, char *out_path, size_t out_path_size);
|
bool romfsGeneratePathFromDirectoryEntry(RomFileSystemContext *ctx, RomFileSystemDirectoryEntry *dir_entry, char *out_path, size_t out_path_size, u8 illegal_char_replace_type);
|
||||||
|
|
||||||
/// Generates a path string from a RomFS file entry.
|
/// Generates a path string from a RomFS file entry.
|
||||||
bool romfsGeneratePathFromFileEntry(RomFileSystemContext *ctx, RomFileSystemFileEntry *file_entry, char *out_path, size_t out_path_size);
|
bool romfsGeneratePathFromFileEntry(RomFileSystemContext *ctx, RomFileSystemFileEntry *file_entry, char *out_path, size_t out_path_size, u8 illegal_char_replace_type);
|
||||||
|
|
||||||
/// Generates HierarchicalSha256 (NCA0) / HierarchicalIntegrity (NCA2/NCA3) FS section patch data using a RomFS context + file entry, which can be used to replace NCA data in content dumping operations.
|
/// Generates HierarchicalSha256 (NCA0) / HierarchicalIntegrity (NCA2/NCA3) FS section patch data using a RomFS context + file entry, which can be used to replace NCA data in content dumping operations.
|
||||||
/// Input offset must be relative to the start of the RomFS file entry data.
|
/// Input offset must be relative to the start of the RomFS file entry data.
|
||||||
|
|
|
@ -78,8 +78,8 @@ bool servicesInitialize(void)
|
||||||
|
|
||||||
for(u32 i = 0; i < g_serviceInfoCount; i++)
|
for(u32 i = 0; i < g_serviceInfoCount; i++)
|
||||||
{
|
{
|
||||||
/* Check if this service has been already initialized */
|
/* Check if this service has been already initialized or if it actually has a valid initialize function */
|
||||||
if (g_serviceInfo[i].initialized) continue;
|
if (g_serviceInfo[i].initialized || g_serviceInfo[i].init_func == NULL) continue;
|
||||||
|
|
||||||
/* Check if this service depends on a condition function */
|
/* Check if this service depends on a condition function */
|
||||||
if (g_serviceInfo[i].cond_func != NULL)
|
if (g_serviceInfo[i].cond_func != NULL)
|
||||||
|
@ -89,9 +89,6 @@ bool servicesInitialize(void)
|
||||||
if (!g_serviceInfo[i].cond_func(&(g_serviceInfo[i]))) continue;
|
if (!g_serviceInfo[i].cond_func(&(g_serviceInfo[i]))) continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Check if this service has a valid initialize function */
|
|
||||||
if (g_serviceInfo[i].init_func == NULL) continue;
|
|
||||||
|
|
||||||
/* Initialize service */
|
/* Initialize service */
|
||||||
rc = g_serviceInfo[i].init_func();
|
rc = g_serviceInfo[i].init_func();
|
||||||
if (R_FAILED(rc))
|
if (R_FAILED(rc))
|
||||||
|
|
Loading…
Reference in a new issue