stos: ex: Add command line parser

Signed-off-by: Chloe M. <chloe@mensia.org>
2026-06-23 03:08:17 +00:00
parent 6c297030ad
commit 06e9d99925
2 changed files with 120 additions and 0 deletions
@@ -0,0 +1,101 @@
+/*
+ * 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);
+}