summaryrefslogtreecommitdiff
path: root/host/commands/emugen/getopt.c
blob: 35235386b75acefe21e5e39bce4efc893b0cd9c9 (plain)
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
#include "getopt.h"

#include <stdio.h>
#include <string.h>

#define  _getprogname() nargv[0]

int opterr = 1;
int optind = 1;
int optopt = 0;
char* optarg;

int getopt(int argc, char* const argv[], const char* ostr) {
    static const char kEmpty[] = "";
    static const char* place = kEmpty;
    if (!*place) {
        if (optind >= argc)
            return -1;

        const char* arg = argv[optind];
        if (arg[0] != '-') {
            // Not an option.
            return -1;
        }
        if (arg[1] == '-' && !arg[2]) {
            // '--' -> end of options.
            return -1;
        }
        if (!arg[1]) {
            // Single '-', If the program wants it, treat it as an option.
            // Otherwise, it's the end of options.
            if (!strchr(ostr, '-')) {
                return -1;
            }
            optopt = '-';
            place = arg + 1;
        } else {
            optopt = arg[1];
            place = arg + 2;
        }
    };

    char* oindex = strchr(ostr, optopt);
    if (!oindex) {
        // Unsupported option.
        (void)fprintf(stderr, "%s: illegal option -- %c\n", argv[0]);
        return '?';
    }
    if (oindex[1] != ':') {
        // No argument needed.
        optarg = NULL;
        if (!*place)
            optind++;
        return optopt;
    }

    // This option needs an argument. Either after the option character,
    // or the argument that follows.
    if (*place) {
        optarg = (char *)place;
    } else if (argc > ++optind) {
        optarg = (char *)argv[optind];
    } else if (oindex[2] == ':') {
        // Optional argument is missing.
        place = kEmpty;
        optarg = NULL;
        return optopt;
    } else {
        // Missing argument.
        place = kEmpty;
        (void)fprintf(stderr, "%s: option requires an argument --%c\n",
                      argv[0], optopt);
        return ':';
    }
    return optopt;
}