toaruos/kernel/misc/args.c
2012-10-03 22:03:21 -07:00

127 lines
3.5 KiB
C

/* vim: tabstop=4 shiftwidth=4 noexpandtab
*
* Kernel Argument Parser
*
* Parses arguments passed by, ie, a Multiboot bootloader.
*
* Part of the ToAruOS Kernel.
* (C) 2011 Kevin Lange
*/
#include <system.h>
#include <logging.h>
#include <ata.h>
/* XXX: This should be moved */
void ext2_disk_mount(uint32_t offset_sector, uint32_t max_sector);
int read_partition_map(int device);
/**
* Parse the given arguments to the kernel.
*
* @param arg A string containing all arguments, separated by spaces.
*/
void
parse_args(
char * arg /* Arguments */
) {
/* Sanity check... */
if (!arg) { return; }
char * pch; /* Tokenizer pointer */
char * save; /* We use the reentrant form of strtok */
char * argv[1024]; /* Command tokens (space-separated elements) */
int tokenid = 0; /* argc, basically */
/* Tokenize the arguments, splitting at spaces */
pch = strtok_r(arg," ",&save);
if (!pch) { return; }
while (pch != NULL) {
argv[tokenid] = (char *)pch;
++tokenid;
pch = strtok_r(NULL," ",&save);
}
argv[tokenid] = NULL;
/* Tokens are now stored in argv. */
for (int i = 0; i < tokenid; ++i) {
/* Parse each provided argument */
char * pch_i;
char * save_i;
char * argp[1024];
int argc = 0;
pch_i = strtok_r(argv[i],"=",&save_i);
if (!pch_i) { continue; }
while (pch_i != NULL) {
argp[argc] = (char *)pch_i;
++argc;
pch_i = strtok_r(NULL,"=,",&save_i);
}
argp[argc] = NULL;
if (!strcmp(argp[0],"vid")) {
if (argc < 2) { kprintf("vid=?\n"); continue; }
uint16_t x, y;
if (argc < 4) {
x = 1024;
y = 768;
} else {
x = atoi(argp[2]);
y = atoi(argp[3]);
debug_print(NOTICE, "Requested display resolution is %dx%d", x, y);
}
if (!strcmp(argp[1],"qemu")) {
/* Bochs / Qemu Video Device */
graphics_install_bochs(x,y);
} else if (!strcmp(argp[1],"preset")) {
graphics_install_preset(x,y);
} else {
debug_print(WARNING, "Unrecognized video adapter: %s", argp[1]);
}
} else if (!strcmp(argp[0],"hdd")) {
if (argc > 1) {
debug_print(INFO, "Scanning disk...");
if (read_partition_map(0)) {
debug_print(ERROR, "Failed to read MBR.");
continue;
}
int partition = atoi(argp[1]);
debug_print(NOTICE, "Selected partition %d starts at sector %d", partition, mbr.partitions[partition].lba_first_sector);
ext2_disk_mount(mbr.partitions[partition].lba_first_sector,mbr.partitions[partition].sector_count);
} else {
ext2_disk_mount(0, 0);
}
} else if (!strcmp(argp[0],"single")) {
boot_arg = "--single";
} else if (!strcmp(argp[0],"lite")) {
boot_arg = "--special";
} else if (!strcmp(argp[0],"vgaterm")) {
boot_arg = "--vga";
} else if (!strcmp(argp[0],"start")) {
if (argc < 2) {
debug_print(WARNING, "Expected an argument to kernel option `start`. Ignoring.");
continue;
}
boot_arg_extra = argp[1];
} else if (!strcmp(argp[0],"logtoserial")) {
if (argc > 1) {
debug_level = atoi(argp[1]);
} else {
debug_level = NOTICE; /* INFO is a bit verbose for a default */
}
kprint_to_serial = 1;
debug_print(NOTICE, "Kernel serial logging enabled at level %d.", debug_level);
} else if (!strcmp(argp[0],"kernel-term")) {
if (argc > 1) {
debug_level = atoi(argp[1]);
} else {
debug_level = NOTICE; /* INFO is a bit verbose for a default */
}
kprint_to_screen = 1;
debug_print(NOTICE, "Kernel serial logging enabled at level %d.", debug_level);
} else if (!strcmp(argp[0],"read-mbr")) {
read_partition_map(0);
}
}
}