limine/src/main.c

112 lines
2.9 KiB
C
Raw Normal View History

2019-05-15 07:08:56 +03:00
asm (
".section .entry\n\t"
2019-05-15 07:08:56 +03:00
"xor dh, dh\n\t"
"push edx\n\t"
"call main\n\t"
);
#include <drivers/vga_textmode.h>
#include <lib/real.h>
2020-01-22 07:02:12 +03:00
#include <lib/blib.h>
#include <lib/libc.h>
#include <lib/mbr.h>
2020-01-22 09:13:19 +03:00
#include <lib/config.h>
2020-04-14 06:20:55 +03:00
#include <fs/file.h>
2020-01-25 04:05:19 +03:00
#include <sys/interrupt.h>
2020-03-25 03:04:18 +03:00
#include <lib/elf.h>
#include <protos/stivale.h>
2020-01-21 13:42:17 +03:00
extern symbol bss_begin;
extern symbol bss_end;
2019-05-15 07:08:56 +03:00
2020-01-22 09:13:19 +03:00
static int config_loaded = 0;
2019-05-15 07:08:56 +03:00
void main(int boot_drive) {
2020-04-14 06:20:55 +03:00
struct file_handle f;
2020-03-15 02:12:09 +03:00
2020-01-21 13:42:17 +03:00
// Zero out .bss section
for (uint8_t *p = bss_begin; p < bss_end; p++)
*p = 0;
// Initial prompt.
init_vga_textmode();
2020-01-25 04:05:19 +03:00
init_idt();
2019-05-31 08:19:02 +03:00
print("qLoader 2\n\n");
2020-01-22 09:13:19 +03:00
print("=> Boot drive: %x\n", boot_drive);
2020-01-21 13:42:17 +03:00
// Enumerate partitions.
struct mbr_part parts[4];
for (int i = 0; i < 4; i++) {
2020-01-22 09:13:19 +03:00
print("=> Checking for partition %d...\n", i);
int ret = mbr_get_part(&parts[i], boot_drive, i);
if (ret) {
2020-01-22 09:13:19 +03:00
print(" Not found!\n");
} else {
2020-01-22 09:13:19 +03:00
print(" Found!\n");
if (!config_loaded) {
2020-03-30 22:24:36 +03:00
if (!init_config(boot_drive, i)) {
2020-01-22 09:13:19 +03:00
config_loaded = 1;
print(" Config file found and loaded!\n");
}
}
}
}
2020-03-28 04:52:59 +03:00
int drive, part, timeout;
char path[128], cmdline[128], proto[64];
2020-01-22 09:13:19 +03:00
if (config_loaded) {
char buf[32];
2020-03-30 23:27:15 +03:00
if (!config_get_value(buf, 0, 32, "KERNEL_DRIVE")) {
2020-03-27 05:45:36 +03:00
print("KERNEL_DRIVE not specified, using boot drive (%x)", boot_drive);
drive = boot_drive;
} else {
drive = (int)strtoui(buf);
}
2020-03-30 23:27:15 +03:00
if (!config_get_value(buf, 0, 64, "TIMEOUT")) {
2020-03-28 04:52:59 +03:00
timeout = 5;
} else {
timeout = (int)strtoui(buf);
}
2020-03-30 23:27:15 +03:00
config_get_value(buf, 0, 32, "KERNEL_PARTITION");
2020-01-22 09:13:19 +03:00
part = (int)strtoui(buf);
2020-03-30 23:27:15 +03:00
config_get_value(path, 0, 128, "KERNEL_PATH");
config_get_value(cmdline, 0, 128, "KERNEL_CMDLINE");
config_get_value(proto, 0, 64, "KERNEL_PROTO");
2020-01-22 09:13:19 +03:00
} else {
print(" !! NO CONFIG FILE FOUND ON BOOT DRIVE !!");
for (;;);
}
2020-01-25 05:06:56 +03:00
print("\n");
2020-03-28 04:52:59 +03:00
for (int i = timeout; i; i--) {
2020-01-25 05:06:56 +03:00
print("\rBooting in %d (press any key to edit command line)...", i);
if (pit_sleep_and_quit_on_keypress(18)) {
print("\n\n> ");
gets(cmdline, cmdline, 128);
break;
}
}
print("\n");
2020-01-25 05:06:56 +03:00
2020-04-14 06:20:55 +03:00
fopen(&f, drive, part, path);
if (!strcmp(proto, "stivale")) {
stivale_load(&f, cmdline);
} else if (!strcmp(proto, "qword")) {
2020-04-14 06:20:55 +03:00
fread(&f, (void *)0x100000, 0, f.size);
// Boot the kernel.
asm volatile (
"cli\n\t"
"jmp 0x100000\n\t"
:
: "b" (cmdline)
: "memory"
);
} else {
print("Invalid protocol specified: `%s`.\n", proto);
for (;;);
}
}