limine/stage2/main.c

89 lines
2.2 KiB
C
Raw Normal View History

2020-08-29 21:02:16 +03:00
#include <limine.h>
2020-09-02 10:55:56 +03:00
#include <lib/term.h>
#include <lib/real.h>
2020-01-22 07:02:12 +03:00
#include <lib/blib.h>
#include <lib/libc.h>
2020-04-15 14:21:44 +03:00
#include <lib/part.h>
2020-01-22 09:13:19 +03:00
#include <lib/config.h>
2020-09-18 21:02:10 +03:00
#include <sys/e820.h>
#include <sys/a20.h>
2020-05-10 01:38:27 +03:00
#include <lib/print.h>
2020-04-14 06:20:55 +03:00
#include <fs/file.h>
2020-03-25 03:04:18 +03:00
#include <lib/elf.h>
2020-09-20 13:03:44 +03:00
#include <mm/pmm.h>
#include <mm/mtrr.h>
#include <protos/stivale.h>
#include <protos/stivale2.h>
2020-04-19 11:14:49 +03:00
#include <protos/linux.h>
2020-05-06 18:00:41 +03:00
#include <protos/chainload.h>
2020-06-05 21:27:52 +03:00
#include <menu.h>
2020-11-05 03:37:45 +03:00
#include <pxe/pxe.h>
#include <pxe/tftp.h>
2020-04-21 19:27:05 +03:00
2020-11-05 03:37:45 +03:00
void entry(uint8_t _boot_drive, int pxe_boot) {
boot_drive = _boot_drive;
mtrr_save();
2020-09-02 10:55:56 +03:00
term_textmode();
2020-01-25 04:05:19 +03:00
2020-08-29 21:02:16 +03:00
print("Limine " LIMINE_VERSION "\n\n");
2020-04-24 19:26:08 +03:00
if (!a20_enable())
panic("Could not enable A20 line");
2020-10-18 07:23:39 +03:00
part_create_index();
2020-11-05 03:37:45 +03:00
init_e820();
init_memmap();
2020-11-09 16:29:53 +03:00
2020-11-05 03:37:45 +03:00
if (pxe_boot) {
pxe_init();
if(init_config_pxe()) {
panic("failed to load config file");
2020-04-15 14:21:44 +03:00
}
2020-11-05 03:37:45 +03:00
print("config loaded");
} 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;
2020-11-05 03:37:45 +03:00
print("Checking partition %d...\n", i);
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:
print("Partition not found.\n");
continue;
}
print("Partition found.\n");
if (!init_config_disk(&part)) {
print("Config file found and loaded.\n");
break;
2020-01-22 09:13:19 +03:00
}
}
}
char *cmdline = menu();
2020-09-02 10:55:56 +03:00
char proto[32];
if (!config_get_value(proto, 0, 32, "PROTOCOL")) {
panic("PROTOCOL not specified");
}
if (!strcmp(proto, "stivale")) {
stivale_load(cmdline);
} else if (!strcmp(proto, "stivale2")) {
stivale2_load(cmdline);
} else if (!strcmp(proto, "linux")) {
linux_load(cmdline);
} else if (!strcmp(proto, "chainload")) {
2020-05-06 18:00:41 +03:00
chainload();
} else {
2020-04-24 19:26:08 +03:00
panic("Invalid protocol specified");
}
2020-04-15 14:21:44 +03:00
}