Chloe M.
102 lines
1.9 KiB
C
102 lines
1.9 KiB
C
/*
|
|
* Copyright (c) 2026, Chloe M.
|
|
* Provided under the BSD-3 clause.
|
|
*
|
|
* Description: Command line parser
|
|
* Author: Chloe M.
|
|
*/
|
|
|
|
#include <drivers/bootvid/fbio.h>
|
|
#include <ex/cmdline.h>
|
|
#include <ke/bpal.h>
|
|
#include <string.h>
|
|
|
|
#define KEY_BOOTCONS "bootcons"
|
|
#define VALUE_TRUE "true"
|
|
#define VALUE_FALSE "false"
|
|
|
|
#include <ex/trace.h>
|
|
|
|
/*
|
|
* Enable or disable the boot console
|
|
*
|
|
* @CmdValue: Command value
|
|
*/
|
|
static VOID
|
|
CmdSetBootCons(CHAR *CmdValue)
|
|
{
|
|
if (CmdValue == NULL) {
|
|
return;
|
|
}
|
|
|
|
if (RtlMemCmp(CmdValue, VALUE_TRUE, sizeof(VALUE_TRUE)) == 0) {
|
|
BootVidInitCons(NULL);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* TODO: Handle ';' seperator
|
|
*/
|
|
VOID
|
|
CmdLineParse(const CHAR *CmdLine)
|
|
{
|
|
CHAR KeyBuf[128];
|
|
CHAR ValBuf[128];
|
|
USHORT KeyBufLen = 0;
|
|
USHORT ValBufLen = 0;
|
|
BOOLEAN ParseVal = false;
|
|
const CHAR *Head;
|
|
|
|
if (CmdLine == NULL) {
|
|
return;
|
|
}
|
|
|
|
Head = CmdLine;
|
|
while (*Head != '\0') {
|
|
if (*Head == '=') {
|
|
ParseVal = true;
|
|
++Head;
|
|
continue;
|
|
}
|
|
|
|
if (!ParseVal && KeyBufLen < sizeof(KeyBuf) - 1) {
|
|
KeyBuf[KeyBufLen++] = *Head++;
|
|
continue;
|
|
}
|
|
|
|
if (ParseVal && ValBufLen < sizeof(ValBuf) - 1) {
|
|
ValBuf[ValBufLen++] = *Head++;
|
|
continue;
|
|
}
|
|
|
|
/* Handle overflows */
|
|
if (!ParseVal && KeyBufLen >= sizeof(KeyBuf) - 1)
|
|
return;
|
|
if (ParseVal && ValBufLen >= sizeof(ValBuf) - 1)
|
|
return;
|
|
|
|
++Head;
|
|
}
|
|
|
|
KeyBuf[KeyBufLen] = '\0';
|
|
ValBuf[ValBufLen] = '\0';
|
|
|
|
switch (KeyBuf[0]) {
|
|
case 'b':
|
|
if (RtlMemCmp(KeyBuf, KEY_BOOTCONS, sizeof(KEY_BOOTCONS)) == 0) {
|
|
CmdSetBootCons(ValBuf);
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
VOID
|
|
ExCmdLineInit(VOID)
|
|
{
|
|
BPAL_HANDLE BpalHandle;
|
|
|
|
KeBpalGetHandle(&BpalHandle);
|
|
CmdLineParse(BpalHandle.CommandLine);
|
|
}
|