summaryrefslogtreecommitdiff
path: root/src/main.c
blob: 454a8e80178eb6e801be4a5c33ed912776558836 (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
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
#include "client/addr.h"
#include "client/client.h"

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/select.h>

#define DEFAULT_PORT 53

static void help() {
    printf("wig: a simple dns client\n");
    printf("usage: wig [@server] [options] domain [qtype]\n");
    printf("arguments: \n");
    printf("\t@server\t specify a dns server (@example.com)\n");
    printf("\t-p\t specify a port (-p 2000)\n");
    printf("\t-t\t force dns request to do tcp first\n");
    exit(EXIT_SUCCESS);
}

static void parse_server (char* argv, char** server) {
    if (*server != NULL) {
        printf("error: only one dns server argument is allowed\n");
        exit(EXIT_FAILURE);
    }
    if (strlen(argv) < 1) {
        printf("error: dns server argument cannot be empty\n");
        exit(EXIT_FAILURE);
    }
    *server = argv;
}

static int parse_arguent (int argc, char** argv, Options* options) {
    if (strlen(argv[0]) < 2) {
        printf("error: expected option after -\n");
        exit(EXIT_FAILURE);
    }

    if (strcmp(argv[0], "-p") == 0) {
        if (argc < 2) {
            printf("error: expected port after -p\n");
            exit(EXIT_FAILURE);
        }
        if ((options->port = strtoul(argv[1], NULL, 10)) == 0) {
            printf("error: %s is not a valid port\n", argv[1]);
            exit(EXIT_FAILURE);
        }
        return 1;
    } else if (strcmp(argv[0], "-t") == 0) {
        options->force_tcp = true;
        return 0;
    } else if (
        strcmp(argv[0], "-h") == 0 ||
        strcmp(argv[0], "--help") == 0
    ) {
        help();
    }

    printf("error: unkown option %s\n", argv[0]);
    exit(EXIT_FAILURE);
}

int main(int argc, char** argv) {

    if (argc < 1) return EXIT_FAILURE;

    int dargs = 1;
   
    char* server = NULL;
    Options options;
    options.port = DEFAULT_PORT;
    options.force_tcp = false;
    
    for (int i = 1; i < argc; i++) {
        char s = argv[i][0];
        if (s == '@') {
            dargs++;
            parse_server(argv[i] + 1, &server);
        } else if (s == '-') {
            int o = parse_arguent(argc - i, &argv[i], &options);
            dargs++;
            dargs += o;
            i += o;
        } else {
            break;
        }
    }

    if (argc - dargs < 1) {
        printf("usage: wig [@server] [options] domain [qtype]\n");
        return EXIT_FAILURE;
    }

    IpAddr addr;

    if (server != NULL) {
        if (!ip_addr_name(server, &addr)) {
            printf("error: %s is not a valid domain\n", server);
            exit(EXIT_FAILURE);
        } 
    } else {
        resolve_default_server(&addr);
    }

    Question* arr;
    uint8_t len;

    resolve_questions(argc - dargs, &argv[dargs], &len, &arr);
    resolve(addr, options, arr, len);

    return EXIT_SUCCESS;
}