summaryrefslogtreecommitdiff
path: root/src/iptest.c
blob: 50877365a90f5df5e4fa6f433d58129abacc36d4 (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
113
114
115
116
117
/*
 * iptest.c: iptest IPv4/IPv6 validator
 *
 * Maintainer: Daniil Baturin <daniil at baturin dot org>
 *
 * Copyright (C) 2013 SO3Group
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#include "config.h"
#include "iptest.h"

static const struct option options[] =
{
    { "is-valid", required_argument, NULL, 'a' },
    { "is-ipv4", required_argument, NULL, 'b' },
    { "is-ipv4-host", required_argument, NULL, 'c'},
    { "is-ipv4-net", required_argument, NULL, 'd'},
    { "is-ipv6", required_argument, NULL, 'g' },
    { "is-ipv6-net", required_argument, NULL, 'G' },
    { NULL, no_argument, NULL, 0 }
};

void help(void)
{
    static char *message = \
"--is-valid STR    Check if STR is a valid IPv4 or IPv6 address\n\
                   or subnet\n\
--is-ipv4 STR Check if STR is a valid IPv4 address with mask\n";

    printf("%s", message);
}

int main(int argc, char* argv[])
{
    char *address_str = "";
    int action = 0;

    int option_index = 0;
    char c;
    int option_count = 0;

    while( (c = getopt_long(argc, argv, "abcd?", options, &option_index)) != -1 )
    {
         switch(c)
         {
             case 'a':
                 action = IS_VALID;
                 address_str = optarg;
                 break;
             case 'b':
                 action = IS_IPV4;
                 address_str = optarg;
                 break;
             case 'c':
                 action = IS_IPV4_HOST;
                 address_str = optarg;
                 break;
             case 'd':
                 action = IS_IPV4_NET;
                 address_str = optarg;
                 break;
             case 'I':
                 action = IS_IPV6;
                 address_str = optarg;
                 break;
             case '?':
             default:
                fprintf(stderr, "Invalid option\n");
                break;
         }
         option_count++;
    }

    if( option_count != 1 )
    {
        fprintf(stderr, "Wrong options number, exactly one option expected!\n");
        help();
        return(1);
    }

    CIDR *address;
    address = cidr_from_str(address_str);
    int result = EXIT_FAILURE;

    switch(action)
    {
        case IS_VALID:
            result = is_valid_address(address);
            break;
        case IS_IPV4:
            result = is_ipv4(address);
            break;
        case IS_IPV4_HOST:
            result = is_ipv4_host(address);
            break;
        case IS_IPV4_NET:
            result = is_ipv4_net(address);
            break;
    }

    return(result);
}