2018-09-07 17:00:13 +02:00
|
|
|
/*
|
2019-04-08 04:00:49 +02:00
|
|
|
* Copyright (c) 2018-2019 Atmosphère-NX
|
2018-09-07 17:00:13 +02:00
|
|
|
*
|
|
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
|
|
* under the terms and conditions of the GNU General Public License,
|
|
|
|
* version 2, as published by the Free Software Foundation.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope 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/>.
|
|
|
|
*/
|
|
|
|
|
2018-05-05 23:55:40 +02:00
|
|
|
#include <stdio.h>
|
2018-05-06 17:22:12 +02:00
|
|
|
#include <stdlib.h>
|
2018-05-13 19:53:55 +02:00
|
|
|
#include <string.h>
|
2018-05-06 17:22:12 +02:00
|
|
|
#include <errno.h>
|
2018-05-07 23:32:45 +02:00
|
|
|
#include <malloc.h>
|
2018-09-23 23:22:25 +02:00
|
|
|
|
2018-04-07 23:43:54 +02:00
|
|
|
#include "utils.h"
|
2018-05-20 14:11:46 +02:00
|
|
|
#include "panic.h"
|
2018-05-21 03:50:53 +02:00
|
|
|
#include "exception_handlers.h"
|
2018-04-07 23:43:54 +02:00
|
|
|
#include "loader.h"
|
2018-05-07 23:32:45 +02:00
|
|
|
#include "chainloader.h"
|
2018-04-08 13:06:04 +02:00
|
|
|
#include "stage2.h"
|
2018-04-09 23:34:23 +02:00
|
|
|
#include "nxboot.h"
|
2018-05-05 23:55:40 +02:00
|
|
|
#include "console.h"
|
2018-05-13 23:49:50 +02:00
|
|
|
#include "fs_utils.h"
|
2018-07-19 22:07:53 +02:00
|
|
|
#include "nxfs.h"
|
2018-05-13 19:53:55 +02:00
|
|
|
#include "gpt.h"
|
2018-05-07 23:32:45 +02:00
|
|
|
#include "display/video_fb.h"
|
2018-07-19 22:07:53 +02:00
|
|
|
#include "sdmmc/sdmmc.h"
|
2018-09-26 02:13:20 +02:00
|
|
|
#include "lib/log.h"
|
2018-04-08 13:13:15 +02:00
|
|
|
|
2018-05-08 16:51:43 +02:00
|
|
|
extern void (*__program_exit_callback)(int rc);
|
2018-04-07 23:43:54 +02:00
|
|
|
|
2019-05-10 17:40:21 +02:00
|
|
|
static __attribute__((__aligned__(0x200))) stage2_args_t g_stage2_args_store;
|
2018-05-08 16:51:43 +02:00
|
|
|
static stage2_args_t *g_stage2_args;
|
|
|
|
static bool g_do_nxboot;
|
2018-04-08 13:06:04 +02:00
|
|
|
|
2018-05-08 16:51:43 +02:00
|
|
|
static void setup_env(void) {
|
2019-07-06 21:58:01 +02:00
|
|
|
/* Initialize the display and console. */
|
|
|
|
if (console_init() < 0) {
|
2018-04-08 13:13:15 +02:00
|
|
|
generic_panic();
|
|
|
|
}
|
2019-07-06 21:58:01 +02:00
|
|
|
|
2018-05-21 03:50:53 +02:00
|
|
|
/* Set up exception handlers. */
|
|
|
|
setup_exception_handlers();
|
|
|
|
|
2019-04-07 20:02:01 +02:00
|
|
|
/* Initialize the file system by mounting the SD card. */
|
|
|
|
if (nxfs_init() < 0) {
|
|
|
|
fatal_error("Failed to initialize the file system: %s\n", strerror(errno));
|
2018-05-08 16:51:43 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static void cleanup_env(void) {
|
2019-07-06 21:58:01 +02:00
|
|
|
/* Terminate the file system. */
|
2019-04-07 20:02:01 +02:00
|
|
|
nxfs_end();
|
2018-05-08 16:51:43 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
static void exit_callback(int rc) {
|
|
|
|
(void)rc;
|
|
|
|
if (g_do_nxboot) {
|
|
|
|
/* TODO: halt */
|
|
|
|
} else {
|
|
|
|
relocate_and_chainload();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Allow for main(int argc, void **argv) signature. */
|
|
|
|
#pragma GCC diagnostic ignored "-Wmain"
|
|
|
|
|
|
|
|
int main(int argc, void **argv) {
|
|
|
|
loader_ctx_t *loader_ctx = get_loader_ctx();
|
|
|
|
|
2019-07-06 21:58:01 +02:00
|
|
|
/* Check argc. */
|
2018-05-07 23:32:45 +02:00
|
|
|
if (argc != STAGE2_ARGC) {
|
2018-06-05 16:18:02 +02:00
|
|
|
generic_panic();
|
2018-05-07 23:32:45 +02:00
|
|
|
}
|
2018-07-19 22:07:53 +02:00
|
|
|
|
2019-05-10 17:40:21 +02:00
|
|
|
g_stage2_args = &g_stage2_args_store;
|
|
|
|
memcpy(g_stage2_args, (stage2_args_t *)argv[STAGE2_ARGV_ARGUMENT_STRUCT], sizeof(*g_stage2_args));
|
2018-05-07 23:32:45 +02:00
|
|
|
|
2019-07-06 21:58:01 +02:00
|
|
|
/* Check stage2 version field. */
|
2018-07-19 22:07:53 +02:00
|
|
|
if (g_stage2_args->version != 0) {
|
2018-06-05 16:18:02 +02:00
|
|
|
generic_panic();
|
2018-05-07 23:32:45 +02:00
|
|
|
}
|
2018-07-19 22:07:53 +02:00
|
|
|
|
2018-09-28 22:12:59 +02:00
|
|
|
/* Override the global logging level. */
|
|
|
|
log_set_log_level(g_stage2_args->log_level);
|
2018-07-19 22:07:53 +02:00
|
|
|
|
2019-07-06 21:58:01 +02:00
|
|
|
/* Initialize the boot environment. */
|
2018-06-05 16:18:02 +02:00
|
|
|
setup_env();
|
2019-07-06 21:58:01 +02:00
|
|
|
|
2019-01-26 09:50:38 +01:00
|
|
|
print(SCREEN_LOG_LEVEL_DEBUG | SCREEN_LOG_LEVEL_NO_PREFIX, u8"Welcome to Atmosphère Fusée Stage 2!\n");
|
2018-09-23 23:22:25 +02:00
|
|
|
print(SCREEN_LOG_LEVEL_DEBUG, "Stage 2 executing from: %s\n", (const char *)argv[STAGE2_ARGV_PROGRAM_PATH]);
|
2019-02-20 22:12:15 +01:00
|
|
|
|
|
|
|
/* Load BCT0 from SD if needed. */
|
|
|
|
if (strcmp(g_stage2_args->bct0, "") == 0) {
|
2019-05-10 17:40:21 +02:00
|
|
|
uint32_t bct_tmp_buf[sizeof(g_stage2_args->bct0) / sizeof(uint32_t)] = {0};
|
|
|
|
if (!read_from_file(bct_tmp_buf, sizeof(bct_tmp_buf) - 1, "atmosphere/BCT.ini")) {
|
2019-02-20 22:12:15 +01:00
|
|
|
fatal_error("Failed to read BCT0 from SD!\n");
|
|
|
|
}
|
2019-05-10 17:40:21 +02:00
|
|
|
memcpy(g_stage2_args->bct0, bct_tmp_buf, sizeof(bct_tmp_buf));
|
2019-02-20 22:12:15 +01:00
|
|
|
}
|
2018-05-04 23:56:01 +02:00
|
|
|
|
2018-04-07 23:43:54 +02:00
|
|
|
/* This will load all remaining binaries off of the SD. */
|
2018-05-07 23:32:45 +02:00
|
|
|
load_payload(g_stage2_args->bct0);
|
2018-09-23 23:22:25 +02:00
|
|
|
print(SCREEN_LOG_LEVEL_INFO, "Loaded payloads!\n");
|
2019-07-06 21:58:01 +02:00
|
|
|
|
|
|
|
g_do_nxboot = (loader_ctx->chainload_entrypoint == 0);
|
2018-05-08 16:51:43 +02:00
|
|
|
if (g_do_nxboot) {
|
2019-02-20 22:12:15 +01:00
|
|
|
print(SCREEN_LOG_LEVEL_INFO, "Now performing nxboot.\n");
|
2019-07-06 21:58:01 +02:00
|
|
|
|
|
|
|
/* Start boot. */
|
2018-09-15 22:08:58 +02:00
|
|
|
uint32_t boot_memaddr = nxboot_main();
|
2019-07-06 21:58:01 +02:00
|
|
|
|
|
|
|
/* Terminate the boot environment. */
|
2019-04-20 12:41:03 +02:00
|
|
|
cleanup_env();
|
2019-07-06 21:58:01 +02:00
|
|
|
|
2019-01-26 09:59:27 +01:00
|
|
|
/* Finish boot. */
|
2018-09-15 22:08:58 +02:00
|
|
|
nxboot_finish(boot_memaddr);
|
2018-05-08 16:51:43 +02:00
|
|
|
} else {
|
2018-05-07 23:32:45 +02:00
|
|
|
/* TODO: What else do we want to do in terms of argc/argv? */
|
2018-05-14 21:17:06 +02:00
|
|
|
const char *path = get_loader_ctx()->file_paths_to_load[get_loader_ctx()->file_id_of_entrypoint];
|
2019-07-26 21:38:15 +02:00
|
|
|
print(SCREEN_LOG_LEVEL_INFO, "Now chainloading.\n");
|
2018-05-08 16:51:43 +02:00
|
|
|
g_chainloader_argc = 1;
|
2018-05-07 23:32:45 +02:00
|
|
|
strcpy(g_chainloader_arg_data, path);
|
2019-07-06 21:58:01 +02:00
|
|
|
|
|
|
|
/* Terminate the boot environment. */
|
|
|
|
cleanup_env();
|
2018-04-09 23:34:23 +02:00
|
|
|
}
|
2018-05-08 16:51:43 +02:00
|
|
|
|
|
|
|
/* Finally, after the cleanup routines (__libc_fini_array, etc.) are called, chainload or halt ourselves. */
|
|
|
|
__program_exit_callback = exit_callback;
|
|
|
|
|
2018-04-07 23:43:54 +02:00
|
|
|
return 0;
|
|
|
|
}
|