tinyargs/parser.c at main · 0xF55/tinyargs · GitHub

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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
#include "funcs.h"
#include "types.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>

/*
    Parse() - Responsible for parsing command-line arguments
*/
BOOL Parse(ArgTable* argTable) {

    for (int i = 0; i < argTable->Argc; i++) {

        CHAR* flag = argTable->Argv[i];

        /*
            Skip invalid or short flags
        */
        if (flag[0] != '-' || strlen(flag) < 2)
            continue;

        /*
            Handle built-in help flags (-h, --help)
        */
        if (strcmp(flag, "-h") == 0 || strcmp(flag, "--help") == 0 || argTable->Argc == 1) {
            Usage(argTable);
            continue;
        }

        /*
            Find the argument node by its flag
        */
        Argument* node = FindArgument(argTable, flag);
        if (node == NULL) {
            fprintf(stderr, "[!] Unknown Flag '%s'\n", flag);
            return FALSE;
        }

        /*
            Process argument based on its type
        */
        switch (node->Type) {

            /*
                Boolean flag (no value required)
            */
        case ARG_BOOL: {
            BoolArg* boolArg = (BoolArg*)node->Arg;
            boolArg->value = TRUE;
            node->UseDefault = FALSE;
            break;
        }

                     /*
                         Integer argument (requires numeric value)
                     */
        case ARG_INT: {
            IntArg* intArg = (IntArg*)node->Arg;
            CHAR* nextarg = argTable->Argv[i + 1];

            if (nextarg != NULL && strlen(nextarg) > 0) {
                BOOL valid = TRUE;

                // Validate that argument contains only digits
                for (CHAR* c = nextarg; *c != '\0'; c++) {
                    if (!isdigit(*c)) {
                        valid = FALSE;
                        break;
                    }
                }

                if (valid) {
                    intArg->value = atoi(nextarg);
                    node->UseDefault = FALSE;
                }
                else {
                    fprintf(stderr, "[!] Invalid integer for '%s': %s\n", flag, nextarg);
                }

                i++;
            }
            break;
        }

                    /*
                        String argument (requires value)
                    */
        case ARG_STRING: {
            CHAR* nextarg = argTable->Argv[i + 1];

            if (nextarg != NULL && strcmp(nextarg, "") != 0) {
                StringArg* val = (StringArg*)node->Arg;
                val->value = nextarg;
                node->UseDefault = FALSE;
                i++;
                continue;
            }

            fprintf(stderr, "[!] Missing value for %s\n", node->Flag);
            break;
        }

                       /*
                           Unknown / unsupported argument type
                       */
        default: {
            fprintf(stderr, "[!] Unknown argument type for '%s'\n", flag);
            break;
        }
        }
    }

    return TRUE;
}