boot/cmdline.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

#include "config.h"
#include "boot.h"
#include <stddef.h>
#include "lib/string.h"
#include "lib/printf.h"

const char *s_cmdline = NULL;

struct kvpair
{
    const char *key, *value;
} s_arguments[CMDLINE_MAX_ARGUMENTS] = { 0 };
size_t s_numargs = 0;

void
cmdline_init(char *cmdline)
{
    s_cmdline = cmdline;
    size_t last_arg = 0;
    const char *arg = cmdline;

    for(char *s = cmdline; *s; s++) {
        if(*s == '=') {
            *(s++) = 0;
            s_arguments[last_arg].key = arg;
            arg = s;
            for(; *s && (*s != ' '); s++) {
                if(*s == '"') {
                    s++;
                    for(; *s && (*s != '"'); s++);
                    s++;
                }
            }
            *(s++) = 0;
            s_arguments[last_arg++].value = arg;
            arg = s;
        }
    }
    s_numargs = last_arg;
}

void
cmdline_print()
{
    for(size_t i = 0; i < s_numargs; i++) {
        const char *arg_key = s_arguments[i].key;
        const char *arg_value = s_arguments[i].value;

        kprintf("cmdline: %s=%s\n", arg_key, arg_value);
    }
}

const char*
cmdline_getarg(const char *key)
{
    for(size_t i = 0; i < s_numargs; i++) {
        const char *arg_key = s_arguments[i].key;
        const char *arg_value = s_arguments[i].value;

        if(strncmp(arg_key, key, strlen(arg_key)) == 0) {
            return arg_value;
        }
    }
    return NULL;
}