Atmosphere/fusee/fusee-secondary/src/main.c

103 lines
2.7 KiB
C
Raw Normal View History

#include <stdio.h>
2018-05-06 15:22:12 +00:00
#include <stdlib.h>
#include <string.h>
2018-05-06 15:22:12 +00:00
#include <errno.h>
#include <malloc.h>
2018-04-07 21:43:54 +00:00
#include "utils.h"
2018-05-20 12:11:46 +00:00
#include "panic.h"
2018-04-07 21:43:54 +00:00
#include "loader.h"
#include "chainloader.h"
#include "stage2.h"
#include "nxboot.h"
#include "console.h"
2018-05-13 21:49:50 +00:00
#include "fs_utils.h"
#include "switch_fs.h"
#include "gpt.h"
#include "display/video_fb.h"
2018-04-08 11:13:15 +00:00
extern void (*__program_exit_callback)(int rc);
2018-04-07 21:43:54 +00:00
static stage2_args_t *g_stage2_args;
static bool g_do_nxboot;
static void setup_env(void) {
2018-05-20 12:11:46 +00:00
/* Check for panics. */
check_and_display_panic();
/* Set the console up. */
if (console_init() == -1) {
2018-04-08 11:13:15 +00:00
generic_panic();
}
2018-05-04 21:56:01 +00:00
if(switchfs_mount_all() == -1) {
perror("Failed to mount at least one parition");
generic_panic();
}
/* TODO: What other hardware init should we do here? */
}
static void cleanup_env(void) {
/* Unmount everything (this causes all open files to be flushed and closed) */
switchfs_unmount_all();
//console_end();
}
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();
/* Initialize the display, console, FS, etc. */
setup_env();
if (argc != STAGE2_ARGC) {
printf("Error: Invalid argc (expected %d, got %d)!\n", STAGE2_ARGC, argc);
generic_panic();
}
g_stage2_args = (stage2_args_t *)argv[STAGE2_ARGV_ARGUMENT_STRUCT];
if(g_stage2_args->version != 0) {
printf("Error: Incorrect Stage2 args version (expected %lu, got %lu)!\n", 0ul, g_stage2_args->version);
generic_panic();
}
2018-05-06 11:03:45 +00:00
printf(u8"Welcome to Atmosphère Fusée Stage 2!\n");
printf("Stage 2 executing from: %s\n", (const char *)argv[STAGE2_ARGV_PROGRAM_PATH]);
2018-05-04 21:56:01 +00:00
2018-04-07 21:43:54 +00:00
/* This will load all remaining binaries off of the SD. */
load_payload(g_stage2_args->bct0);
2018-05-04 21:56:01 +00:00
printf("Loaded payloads!\n");
2018-05-04 21:56:01 +00:00
g_do_nxboot = loader_ctx->chainload_entrypoint == 0;
if (g_do_nxboot) {
2018-05-13 22:06:48 +00:00
printf("Now performing nxboot.\n");
nxboot_main();
} else {
/* TODO: What else do we want to do in terms of argc/argv? */
const char *path = get_loader_ctx()->file_paths_to_load[get_loader_ctx()->file_id_of_entrypoint];
2018-05-13 22:06:48 +00:00
printf("Now chainloading.\n");
g_chainloader_argc = 1;
strcpy(g_chainloader_arg_data, path);
}
/* Deinitialize the display, console, FS, etc. */
cleanup_env();
/* Finally, after the cleanup routines (__libc_fini_array, etc.) are called, chainload or halt ourselves. */
__program_exit_callback = exit_callback;
2018-04-07 21:43:54 +00:00
return 0;
}