Magic-Mirror/hactool
1
0
Fork 0
mirror of https://github.com/SciresM/hactool.git synced 2024-11-27 14:30:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
0c65ac1e70
hactool/main.c
Michael Scire 0c65ac1e70 Implement --basenca support (closes #1).
2018-01-27 03:24:01 -08:00

276 lines
10 KiB
C
Raw Blame History

#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"
" --basenca Set Base NCA 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;
ncatool_ctx_t base_ctx; /* Context for base NCA, if used. */
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(&base_ctx, 0, sizeof(base_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},
{"basenca", 1, NULL, 16},
{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_file != NULL) {
usage();
return EXIT_FAILURE;
}
if ((ctx.tool_ctx->base_file = fopen(optarg, "rb")) == NULL) {
fprintf(stderr, "unable to open %s: %s\n", optarg, strerror(errno));
return EXIT_FAILURE;
}
ctx.tool_ctx->base_file_type = BASEFILE_ROMFS;
break;
case 16:
if (ctx.tool_ctx->base_file != NULL) {
usage();
return EXIT_FAILURE;
}
if ((ctx.tool_ctx->base_file = fopen(optarg, "rb")) == NULL) {
fprintf(stderr, "unable to open %s: %s\n", optarg, strerror(errno));
return EXIT_FAILURE;
}
ctx.tool_ctx->base_file_type = BASEFILE_NCA;
ctx.tool_ctx->base_nca_ctx = malloc(sizeof(*ctx.tool_ctx->base_nca_ctx));
if (ctx.tool_ctx->base_nca_ctx == NULL) {
fprintf(stderr, "Failed to allocate base NCA context!\n");
return EXIT_FAILURE;
}
nca_init(ctx.tool_ctx->base_nca_ctx);
base_ctx.file = ctx.tool_ctx->base_file;
ctx.tool_ctx->base_nca_ctx->file = base_ctx.file;
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;
}
if (ctx.tool_ctx->base_nca_ctx != NULL) {
memcpy(&base_ctx.settings.keyset, &tool_ctx.settings.keyset, sizeof(nca_keyset_t));
ctx.tool_ctx->base_nca_ctx->tool_ctx = &base_ctx;
nca_process(ctx.tool_ctx->base_nca_ctx);
int found_romfs = 0;
for (unsigned int i = 0; i < 4; i++) {
if (ctx.tool_ctx->base_nca_ctx->section_contexts[i].is_present && ctx.tool_ctx->base_nca_ctx->section_contexts[i].type == ROMFS) {
found_romfs = 1;
break;
}
}
if (found_romfs == 0) {
fprintf(stderr, "Unable to locate RomFS in base NCA!\n");
return EXIT_FAILURE;
}
}
ctx.file = tool_ctx.file;
nca_process(&ctx);
nca_free_section_contexts(&ctx);
if (ctx.tool_ctx->base_file != NULL) {
fclose(ctx.tool_ctx->base_file);
if (ctx.tool_ctx->base_file_type == BASEFILE_NCA) {
nca_free_section_contexts(ctx.tool_ctx->base_nca_ctx);
free(ctx.tool_ctx->base_nca_ctx);
}
}
if (ctx.file != NULL) {
fclose(ctx.file);
}
printf("Done!\n");
return EXIT_SUCCESS;
}
Reference in New Issue View Git Blame Copy Permalink