1
0
Fork 0
mirror of https://github.com/Atmosphere-NX/Atmosphere.git synced 2024-12-23 02:42:09 +00:00

Stratosphere: Add support for custom KIPs/INI in Fusee

This commit is contained in:
Michael Scire 2018-04-11 21:56:11 -06:00
parent e5a0cb1abe
commit 0674c4b64f
4 changed files with 127 additions and 10 deletions

View file

@ -14,7 +14,7 @@ typedef struct {
uint32_t size; uint32_t size;
uint32_t num_processes; uint32_t num_processes;
uint32_t _0xC; uint32_t _0xC;
char kip_data[]; unsigned char kip_data[];
} ini1_header_t; } ini1_header_t;
typedef struct { typedef struct {

View file

@ -1,5 +1,6 @@
#include "utils.h" #include "utils.h"
#include "masterkey.h" #include "masterkey.h"
#include "stratosphere.h"
#include "package2.h" #include "package2.h"
#include "kip.h" #include "kip.h"
#include "se.h" #include "se.h"
@ -9,7 +10,6 @@
/* This *greatly* simplifies logic. */ /* This *greatly* simplifies logic. */
unsigned char g_patched_package2[PACKAGE2_SIZE_MAX]; unsigned char g_patched_package2[PACKAGE2_SIZE_MAX];
unsigned char g_package2_sections[PACKAGE2_SECTION_MAX][PACKAGE2_SIZE_MAX]; unsigned char g_package2_sections[PACKAGE2_SECTION_MAX][PACKAGE2_SIZE_MAX];
unsigned char g_package2_work_buffer[PACKAGE2_SIZE_MAX];
package2_header_t *g_patched_package2_header = (package2_header_t *)g_patched_package2; package2_header_t *g_patched_package2_header = (package2_header_t *)g_patched_package2;
@ -36,7 +36,7 @@ void package2_patch(void *package2_address) {
package2_fixup_header_and_section_hashes(); package2_fixup_header_and_section_hashes();
/* Relocate Package2. */ /* Relocate Package2. */
memcpy(NX_BOOTLOADER_PACKAGE2_LOAD_ADDRESS, g_patched_package2, PACKAGE2_SIZE_MAX); memcpy(NX_BOOTLOADER_PACKAGE2_LOAD_ADDRESS, g_patched_package2, sizeof(g_patched_package2));
} }
static void package2_crypt_ctr(unsigned int master_key_rev, void *dst, size_t dst_size, const void *src, size_t src_size, const void *ctr, size_t ctr_size) { static void package2_crypt_ctr(unsigned int master_key_rev, void *dst, size_t dst_size, const void *src, size_t src_size, const void *ctr, size_t ctr_size) {
@ -211,14 +211,15 @@ void package2_patch_kernel(void) {
/* TODO: What kind of patching do we want to try to do here? */ /* TODO: What kind of patching do we want to try to do here? */
} }
void package2_patch_ini1(void) {
ini1_header_t *ini_header = (ini1_header_t *)g_package2_sections[PACKAGE2_SECTION_INI1];
if (ini_header->magic != MAGIC_INI1) {
printk("Error: INI1 section appears to not contain an INI1!\n");
generic_panic();
}
/* TODO */ void package2_patch_ini1(void) {
/* TODO: Do we want to support loading another INI from sd:/whatever/INI1.bin? */
ini1_header_t *inis_to_merge[STRATOSPHERE_INI1_MAX] = {0};
inis_to_merge[STRATOSPHERE_INI1_EMBEDDED] = stratosphere_get_ini1();
inis_to_merge[STRATOSPHERE_INI1_PACKAGE2] = (ini1_header_t *)g_package2_sections[PACKAGE2_SECTION_INI1];
/* Merge all of the INI1s. */
stratosphere_merge_inis(g_package2_sections[PACKAGE2_SECTION_INI1], inis_to_merge, STRATOSPHERE_INI1_MAX);
} }
void package2_fixup_header_and_section_hashes(void) { void package2_fixup_header_and_section_hashes(void) {

View file

@ -0,0 +1,101 @@
#include "utils.h"
#include "package2.h"
#include "stratosphere.h"
#include "sd_utils.h"
#include "lib/printk.h"
unsigned char g_stratosphere_ini1[PACKAGE2_SIZE_MAX];
static bool g_initialized_stratosphere_ini1 = false;
unsigned char g_ini1_buffer[PACKAGE2_SIZE_MAX];
ini1_header_t *stratosphere_get_ini1(void) {
ini1_header_t *ini1_header = (ini1_header_t *)g_stratosphere_ini1;
if (g_initialized_stratosphere_ini1) {
return ini1_header;
}
ini1_header->magic = MAGIC_INI1;
ini1_header->size = sizeof(ini1_header_t);
ini1_header->num_processes = 0;
ini1_header->_0xC = 0;
/* TODO: When we have processes, copy them into ini1_header->kip_data here. */
g_initialized_stratosphere_ini1 = true;
return ini1_header;
}
/* Merges some number of INI1s into a single INI1. It's assumed that the INIs are in order of preference. */
void stratosphere_merge_inis(void *dst, ini1_header_t **inis, unsigned int num_inis) {
char sd_path[0x300] = {0};
/* Validate all ini headers. */
for (unsigned int i = 0; i < num_inis; i++) {
if (inis[i] == NULL || inis[i]->magic != MAGIC_INI1 || inis[i]->num_processes > INI1_MAX_KIPS) {
printk("Error: INI1s[%d] section appears to not contain an INI1!\n", i);
generic_panic();
}
}
uint64_t process_list[INI1_MAX_KIPS] = {0};
memset(g_ini1_buffer, 0, sizeof(g_ini1_buffer));
ini1_header_t *merged = (ini1_header_t *)g_ini1_buffer;
merged->magic = MAGIC_INI1;
merged->num_processes = 0;
merged->_0xC = 0;
size_t remaining_size = PACKAGE2_SIZE_MAX - sizeof(ini1_header_t);
unsigned char *current_dst_kip = merged->kip_data;
/* Actually merge into the inis. */
for (unsigned int i = 0; i < num_inis; i++) {
uint64_t offset = 0;
for (unsigned int p = 0; p < inis[i]->num_processes; p++) {
kip1_header_t *current_kip = (kip1_header_t *)(inis[i]->kip_data + offset);
if (current_kip->magic != MAGIC_KIP1) {
printk("Error: INI1s[%d][%d] appears not to be a KIP1!\n", i, p);
generic_panic();
}
bool already_loaded = false;
for (unsigned int j = 0; j < merged->num_processes; j++) {
if (process_list[j] == current_kip->title_id) {
already_loaded = true;
break;
}
}
if (already_loaded) {
continue;
}
/* Try to load an override KIP from SD, if possible. */
if (read_sd_file(current_dst_kip, remaining_size, sd_path)) {
kip1_header_t *sd_kip = (kip1_header_t *)(current_dst_kip);
if (sd_kip->magic != MAGIC_KIP1) {
printk("Error: %s is not a KIP1?\n", sd_path);
generic_panic();
} else if (sd_kip->title_id != current_kip->title_id) {
printk("Error: %s has wrong Title ID!\n", sd_path);
generic_panic();
}
current_dst_kip += kip1_get_size_from_header(sd_kip);
} else {
uint64_t current_kip_size = kip1_get_size_from_header(current_kip);
if (current_kip_size > remaining_size) {
printk("Error: Not enough space for all the KIP1s!\n");
generic_panic();
}
memcpy(current_dst_kip, current_kip, current_kip_size);
remaining_size -= current_kip_size;
current_dst_kip += current_kip_size;
}
process_list[merged->num_processes++] = current_kip->title_id;
}
}
merged->size = sizeof(ini1_header_t) + (uint32_t)(current_dst_kip - merged->kip_data);
/* Copy merged INI1 to destination. */
memcpy(dst, merged, merged->size);
}

View file

@ -0,0 +1,15 @@
#ifndef FUSEE_STRATOSPHERE_H
#define FUSEE_STRATOSPHERE_H
#include "utils.h"
#include "kip.h"
#define STRATOSPHERE_INI1_EMBEDDED 0x0
#define STRATOSPHERE_INI1_PACKAGE2 0x1
#define STRATOSPHERE_INI1_MAX 0x2
ini1_header_t *stratosphere_get_ini1(void);
void stratosphere_merge_inis(void *dst, ini1_header_t **inis, unsigned int num_inis);
#endif