rulimine/stage2/main.c

95 lines
2.4 KiB
C

#include <lib/term.h>
#include <lib/real.h>
#include <lib/blib.h>
#include <lib/libc.h>
#include <lib/part.h>
#include <lib/config.h>
#include <lib/trace.h>
#include <sys/e820.h>
#include <sys/a20.h>
#include <lib/print.h>
#include <fs/file.h>
#include <lib/elf.h>
#include <mm/pmm.h>
#include <mm/mtrr.h>
#include <protos/stivale.h>
#include <protos/stivale2.h>
#include <protos/linux.h>
#include <protos/chainload.h>
#include <menu.h>
#include <pxe/pxe.h>
#include <pxe/tftp.h>
#include <lib/tinf.h>
void entry(uint8_t _boot_drive, int pxe_boot, void *_tinf_gzip_uncompress) {
boot_drive = _boot_drive;
tinf_gzip_uncompress = _tinf_gzip_uncompress;
booted_from_pxe = pxe_boot;
mtrr_save();
term_textmode();
print("Limine " LIMINE_VERSION "\n\n");
if (!a20_enable())
panic("Could not enable A20 line");
init_e820();
init_memmap();
part_create_index();
if (pxe_boot) {
pxe_init();
if (init_config_pxe()) {
panic("Failed to load config file");
}
print("Config loaded via PXE\n");
} else {
print("Boot drive: %x\n", boot_drive);
// Look for config file.
print("Searching for config file...\n");
for (int i = 0; ; i++) {
struct part part;
int ret = part_get(&part, boot_drive, i);
switch (ret) {
case INVALID_TABLE:
panic("Partition table of boot drive is invalid.");
case END_OF_TABLE:
panic("Config file not found.");
case NO_PARTITION:
continue;
}
if (!init_config_disk(&part)) {
print("Config file found and loaded.\n");
boot_partition = i;
break;
}
}
}
trace_init();
char *cmdline;
char *config = menu(&cmdline);
char *proto = config_get_value(config, 0, "PROTOCOL");
if (proto == NULL) {
panic("PROTOCOL not specified");
}
if (!strcmp(proto, "stivale")) {
stivale_load(config, cmdline);
} else if (!strcmp(proto, "stivale2")) {
stivale2_load(config, cmdline, pxe_boot);
} else if (!strcmp(proto, "linux")) {
linux_load(config, cmdline);
} else if (!strcmp(proto, "chainload")) {
chainload(config);
} else {
panic("Invalid protocol specified");
}
}