mirror of
https://github.com/SciresM/hactool
synced 2024-11-10 06:34:14 +00:00
fba6933668
Currently, --baseromfs allows to specify a base romfs image. The base romfs must be dumped using -r for this to work right. TODO: I would like to add support for --basenca.
226 lines
8 KiB
C
226 lines
8 KiB
C
#include <getopt.h>
|
|
#include <errno.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include "types.h"
|
|
#include "utils.h"
|
|
#include "settings.h"
|
|
#include "pki.h"
|
|
#include "nca.h"
|
|
|
|
static char *prog_name = "ncatool";
|
|
|
|
/* Print usage. Taken largely from ctrtool. */
|
|
static void usage(void) {
|
|
fprintf(stderr,
|
|
"ncatool (c) SciresM.\n"
|
|
"Built: %s %s\n"
|
|
"\n"
|
|
"Usage: %s [options...] <file>\n"
|
|
"Options:\n"
|
|
"-i, --info Show file info.\n"
|
|
" This is the default action.\n"
|
|
"-x, --extract Extract data from file.\n"
|
|
" This is also the default action.\n"
|
|
" -r, --raw Keep raw data, don't unpack.\n"
|
|
" -y, --verify Verify hashes and signatures.\n"
|
|
" -d, --dev Decrypt with development keys instead of retail.\n"
|
|
" --titlekey=key Set title key for Rights ID crypto titles.\n"
|
|
" --contentkey=key Set raw key for NCA body decryption.\n"
|
|
"NCA options:\n"
|
|
" --section0=file Specify Section 0 file path.\n"
|
|
" --section1=file Specify Section 1 file path.\n"
|
|
" --section2=file Specify Section 2 file path.\n"
|
|
" --section3=file Specify Section 3 file path.\n"
|
|
" --section0dir=dir Specify Section 0 directory path.\n"
|
|
" --section1dir=dir Specify Section 1 directory path.\n"
|
|
" --section2dir=dir Specify Section 2 directory path.\n"
|
|
" --section3dir=dir Specify Section 3 directory path.\n"
|
|
" --exefs=file Specify ExeFS file path. Overrides appropriate section file path.\n"
|
|
" --exefsdir=dir Specify ExeFS directory path. Overrides appropriate section directory path.\n"
|
|
" --romfs=file Specify RomFS file path. Overrides appropriate section file path.\n"
|
|
" --romfsdir=dir Specify RomFS directory path. Overrides appropriate section directory path.\n"
|
|
" --listromfs List files in RomFS.\n"
|
|
" --baseromfs Set Base RomFS to use with update partitions.\n"
|
|
"\n", __TIME__, __DATE__, prog_name);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
static int ishex(char c) {
|
|
if ('a' <= c && c <= 'f') return 1;
|
|
if ('A' <= c && c <= 'F') return 1;
|
|
if ('0' <= c && c <= '9') return 1;
|
|
return 0;
|
|
}
|
|
|
|
static char hextoi(char c) {
|
|
if ('a' <= c && c <= 'f') return c - 'a' + 0xA;
|
|
if ('A' <= c && c <= 'F') return c - 'A' + 0xA;
|
|
if ('0' <= c && c <= '9') return c - '0';
|
|
return 0;
|
|
}
|
|
|
|
void parse_hex_key(unsigned char *key, const char *hex) {
|
|
if (strlen(hex) != 32) {
|
|
fprintf(stderr, "Key must be 32 hex digits!\n");
|
|
usage();
|
|
}
|
|
|
|
for (unsigned int i = 0; i < 32; i++) {
|
|
if (!ishex(hex[i])) {
|
|
fprintf(stderr, "Key must be 32 hex digits!\n");
|
|
usage();
|
|
}
|
|
}
|
|
|
|
memset(key, 0, 16);
|
|
|
|
for (unsigned int i = 0; i < 32; i++) {
|
|
char val = hextoi(hex[i]);
|
|
if ((i & 1) == 0) {
|
|
val <<= 4;
|
|
}
|
|
key[i >> 1] |= val;
|
|
}
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
ncatool_ctx_t tool_ctx;
|
|
nca_ctx_t ctx;
|
|
char input_name[0x200];
|
|
|
|
prog_name = (argc < 1) ? "ncatool" : argv[0];
|
|
|
|
nca_init(&ctx);
|
|
memset(&tool_ctx, 0, sizeof(tool_ctx));
|
|
memset(input_name, 0, sizeof(input_name));
|
|
ctx.tool_ctx = &tool_ctx;
|
|
|
|
ctx.tool_ctx->action = ACTION_INFO | ACTION_EXTRACT;
|
|
pki_initialize_keyset(&tool_ctx.settings.keyset, KEYSET_RETAIL);
|
|
|
|
while (1) {
|
|
int option_index;
|
|
char c;
|
|
static struct option long_options[] =
|
|
{
|
|
{"extract", 0, NULL, 'x'},
|
|
{"info", 0, NULL, 'i'},
|
|
{"dev", 0, NULL, 'd'},
|
|
{"verify", 0, NULL, 'y'},
|
|
{"raw", 0, NULL, 'r'},
|
|
{"section0", 1, NULL, 0},
|
|
{"section1", 1, NULL, 1},
|
|
{"section2", 1, NULL, 2},
|
|
{"section3", 1, NULL, 3},
|
|
{"section0dir", 1, NULL, 4},
|
|
{"section1dir", 1, NULL, 5},
|
|
{"section2dir", 1, NULL, 6},
|
|
{"section3dir", 1, NULL, 7},
|
|
{"exefs", 1, NULL, 8},
|
|
{"romfs", 1, NULL, 9},
|
|
{"exefsdir", 1, NULL, 10},
|
|
{"romfsdir", 1, NULL, 11},
|
|
{"titlekey", 1, NULL, 12},
|
|
{"contentkey", 1, NULL, 13},
|
|
{"listromfs", 0, NULL, 14},
|
|
{"baseromfs", 1, NULL, 15},
|
|
{NULL, 0, NULL, 0},
|
|
};
|
|
|
|
c = getopt_long(argc, argv, "dryxi", long_options, &option_index);
|
|
if (c == -1)
|
|
break;
|
|
|
|
switch (c)
|
|
{
|
|
case 'i':
|
|
ctx.tool_ctx->action |= ACTION_INFO;
|
|
break;
|
|
case 'x':
|
|
ctx.tool_ctx->action |= ACTION_EXTRACT;
|
|
break;
|
|
case 'y':
|
|
ctx.tool_ctx->action |= ACTION_VERIFY;
|
|
break;
|
|
case 'r':
|
|
ctx.tool_ctx->action |= ACTION_RAW;
|
|
break;
|
|
case 'd':
|
|
pki_initialize_keyset(&tool_ctx.settings.keyset, KEYSET_DEV);
|
|
break;
|
|
|
|
case 0: filepath_set(&ctx.tool_ctx->settings.section_paths[0], optarg); break;
|
|
case 1: filepath_set(&ctx.tool_ctx->settings.section_paths[1], optarg); break;
|
|
case 2: filepath_set(&ctx.tool_ctx->settings.section_paths[2], optarg); break;
|
|
case 3: filepath_set(&ctx.tool_ctx->settings.section_paths[3], optarg); break;
|
|
case 4: filepath_set(&ctx.tool_ctx->settings.section_dir_paths[0], optarg); break;
|
|
case 5: filepath_set(&ctx.tool_ctx->settings.section_dir_paths[1], optarg); break;
|
|
case 6: filepath_set(&ctx.tool_ctx->settings.section_dir_paths[2], optarg); break;
|
|
case 7: filepath_set(&ctx.tool_ctx->settings.section_dir_paths[3], optarg); break;
|
|
case 8:
|
|
ctx.tool_ctx->settings.exefs_path.enabled = 1;
|
|
filepath_set(&ctx.tool_ctx->settings.exefs_path.path, optarg);
|
|
break;
|
|
case 9:
|
|
ctx.tool_ctx->settings.romfs_path.enabled = 1;
|
|
filepath_set(&ctx.tool_ctx->settings.romfs_path.path, optarg);
|
|
break;
|
|
case 10:
|
|
ctx.tool_ctx->settings.exefs_dir_path.enabled = 1;
|
|
filepath_set(&ctx.tool_ctx->settings.exefs_dir_path.path, optarg);
|
|
break;
|
|
case 11:
|
|
ctx.tool_ctx->settings.romfs_dir_path.enabled = 1;
|
|
filepath_set(&ctx.tool_ctx->settings.romfs_dir_path.path, optarg);
|
|
break;
|
|
case 12:
|
|
parse_hex_key(ctx.tool_ctx->settings.titlekey, optarg);
|
|
ctx.tool_ctx->settings.has_titlekey = 1;
|
|
break;
|
|
case 13:
|
|
parse_hex_key(ctx.tool_ctx->settings.contentkey, optarg);
|
|
ctx.tool_ctx->settings.has_contentkey = 1;
|
|
break;
|
|
case 14:
|
|
ctx.tool_ctx->action |= ACTION_LISTROMFS;
|
|
break;
|
|
case 15:
|
|
if ((ctx.tool_ctx->base_romfs_file = fopen(optarg, "rb")) == NULL) {
|
|
fprintf(stderr, "unable to open %s: %s\n", optarg, strerror(errno));
|
|
return EXIT_FAILURE;
|
|
}
|
|
break;
|
|
default:
|
|
usage();
|
|
return EXIT_FAILURE;
|
|
}
|
|
}
|
|
|
|
if (optind == argc - 1) {
|
|
/* Copy input file. */
|
|
strncpy(input_name, argv[optind], sizeof(input_name));
|
|
} else if ((optind < argc) || (argc == 1)) {
|
|
usage();
|
|
}
|
|
|
|
if ((tool_ctx.file = fopen(input_name, "rb")) == NULL) {
|
|
fprintf(stderr, "unable to open %s: %s\n", input_name, strerror(errno));
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
ctx.file = tool_ctx.file;
|
|
nca_process(&ctx);
|
|
nca_free_section_contexts(&ctx);
|
|
|
|
if (ctx.tool_ctx->base_romfs_file != NULL) {
|
|
fclose(ctx.tool_ctx->base_romfs_file);
|
|
}
|
|
if (ctx.file != NULL) {
|
|
fclose(ctx.file);
|
|
}
|
|
printf("Done!\n");
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|