95 lines
2.5 KiB
C
95 lines
2.5 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>
|
|
|
|
#if defined (uefi)
|
|
EFI_STATUS efi_main(EFI_HANDLE ImageHandle, EFI_SYSTEM_TABLE *SystemTable) {
|
|
(void)ImageHandle;
|
|
|
|
gST = SystemTable;
|
|
gBS = SystemTable->BootServices;
|
|
gRT = SystemTable->RuntimeServices;
|
|
|
|
for (;;);
|
|
}
|
|
#endif
|
|
|
|
__attribute__((section(".stage3_build_id")))
|
|
uint64_t stage3_build_id = BUILD_ID;
|
|
|
|
__attribute__((noreturn))
|
|
__attribute__((section(".stage3_entry")))
|
|
void stage3_entry(int boot_from) {
|
|
mtrr_save();
|
|
|
|
switch (boot_from) {
|
|
case BOOT_FROM_HDD:
|
|
case BOOT_FROM_CD: {
|
|
struct volume boot_volume;
|
|
volume_get_by_coord(&boot_volume, boot_drive, -1);
|
|
struct volume part = boot_volume;
|
|
for (int i = 0; ; i++) {
|
|
if (!init_config_disk(&part)) {
|
|
print("Config file found and loaded.\n");
|
|
boot_partition = i - 1;
|
|
break;
|
|
}
|
|
int ret = part_get(&part, &boot_volume, i);
|
|
switch (ret) {
|
|
case INVALID_TABLE:
|
|
case END_OF_TABLE:
|
|
panic("Config file not found.");
|
|
}
|
|
}
|
|
break;
|
|
case BOOT_FROM_PXE:
|
|
pxe_init();
|
|
if (init_config_pxe()) {
|
|
panic("Failed to load config file");
|
|
}
|
|
print("Config loaded via PXE\n");
|
|
break;
|
|
}
|
|
}
|
|
|
|
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, booted_from_pxe);
|
|
} else if (!strcmp(proto, "linux")) {
|
|
linux_load(config, cmdline);
|
|
} else if (!strcmp(proto, "chainload")) {
|
|
chainload(config);
|
|
} else {
|
|
panic("Invalid protocol specified");
|
|
}
|
|
|
|
for (;;);
|
|
}
|