checkenv/checkenv.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
66

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

static void
help(void)
{
    printf("usage: ./checkenv <ENV>:<...>:<...>\n");
    printf("[-h]    Display this help menu\n");
}

static int
check_envs(const char *envlist)
{
    char *p, *tok;
    char *env;

    p = strdup(envlist);
    if (p == NULL) {
        printf("fatal: out of memory\n");
        return -1;
    }

    tok = strtok(p, ":");
    while (tok != NULL) {
        if ((env = getenv(tok)) == NULL) {
            printf("env '%s' not set\n", tok);
            return -1;
        }
        tok = strtok(NULL, ":");
    }

    return 0;
}

int
main(int argc, char **argv)
{
    int opt;
    char *envlist;
    int error;

    if (argc < 2) {
        printf("fatal: too few arguments\n");
        help();
        return -1;
    }

    while ((opt = getopt(argc, argv, "h")) != -1) {
        switch (opt) {
        case 'h':
            help();
            return -1;
        }
    }

    while (optind < argc) {
        envlist = argv[optind++];
        if ((error = check_envs(envlist)) < 0) {
            return error;
        }
    }

    return 0;
}