Initial import from http://www.droids-corp.org/hg/libcmdline/rev/db316e4289a1
[libcmdline.git] / src / lib / cmdline_parse_ipaddr.c
1 /*
2  * Copyright (c) 2009, Olivier MATZ <zer0@droids-corp.org>
3  * All rights reserved.
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are met:
6  *
7  *     * Redistributions of source code must retain the above copyright
8  *       notice, this list of conditions and the following disclaimer.
9  *     * Redistributions in binary form must reproduce the above copyright
10  *       notice, this list of conditions and the following disclaimer in the
11  *       documentation and/or other materials provided with the distribution.
12  *     * Neither the name of the University of California, Berkeley nor the
13  *       names of its contributors may be used to endorse or promote products
14  *       derived from this software without specific prior written permission.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND ANY
17  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
19  * DISCLAIMED. IN NO EVENT SHALL THE REGENTS AND CONTRIBUTORS BE LIABLE FOR ANY
20  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
21  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
22  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
23  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
25  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26  */
27
28 #include <stdio.h>
29 #include <stdlib.h>
30 #include <inttypes.h>
31 #include <ctype.h>
32 #include <string.h>
33 #include <errno.h>
34 #include <netinet/in.h>
35 #include <arpa/inet.h>
36
37 #include "cmdline_parse.h"
38 #include "cmdline_parse_ipaddr.h"
39
40 struct cmdline_token_ops cmdline_token_ipaddr_ops = {
41         .parse = cmdline_parse_ipaddr,
42         .complete_get_nb = NULL,
43         .complete_get_elt = NULL,
44         .get_help = cmdline_get_help_ipaddr,
45 };
46
47
48 int 
49 cmdline_parse_ipaddr(cmdline_parse_token_hdr_t *tk, const char *buf, void *res)
50 {
51         struct cmdline_token_ipaddr *tk2 = (struct cmdline_token_ipaddr *)tk;
52         unsigned int token_len = 0;
53         char ip_str[INET6_ADDRSTRLEN+4]; /* '+4' is for prefixlen (if any) */
54         cmdline_ipaddr_t *ipaddr = res;
55         char *prefix, *prefix_end;
56         long prefixlen;
57
58         if (! *buf)
59                 return -1;
60
61         while (!cmdline_isendoftoken(buf[token_len]))
62                 token_len++;
63                         
64         /* if token is too big... */
65         if (token_len >= INET6_ADDRSTRLEN+4)
66                 return -1;
67
68         snprintf(ip_str, token_len+1, "%s", buf);
69
70         /* convert the network prefix */
71         if (tk2->ipaddr_data.flags & CMDLINE_IPADDR_NETWORK) {
72                 prefix = strrchr(ip_str, '/');
73                 if (prefix == NULL)
74                         return -1;
75                 *prefix = '\0';
76                 prefix ++;
77                 errno = 0;
78                 prefixlen = strtol(prefix, &prefix_end, 10);
79                 if (errno || (*prefix_end != '\0') )
80                         return -1;
81                 ipaddr->prefixlen = prefixlen;
82         }
83         else {
84                 ipaddr->prefixlen = 0;  
85         }
86
87         /* convert the IP addr */
88         if ((tk2->ipaddr_data.flags & CMDLINE_IPADDR_V4) &&
89             inet_pton(AF_INET, ip_str, &ipaddr->addr.ipv4) == 1) {
90                 ipaddr->family = AF_INET;
91                 return token_len;
92         }
93         if ((tk2->ipaddr_data.flags & CMDLINE_IPADDR_V6) &&
94             inet_pton(AF_INET6, ip_str, &ipaddr->addr.ipv6) == 1) {
95                 ipaddr->family = AF_INET6;
96                 return token_len;
97         }
98         return -1;
99                         
100 }
101
102 int cmdline_get_help_ipaddr(cmdline_parse_token_hdr_t *tk, char *dstbuf,
103                             unsigned int size)
104 {
105         struct cmdline_token_ipaddr *tk2 = (struct cmdline_token_ipaddr *)tk;
106
107         switch (tk2->ipaddr_data.flags) {
108         case CMDLINE_IPADDR_V4:
109                 snprintf(dstbuf, size, "IPv4");
110                 break;
111         case CMDLINE_IPADDR_V6:
112                 snprintf(dstbuf, size, "IPv6");
113                 break;
114         case CMDLINE_IPADDR_V4|CMDLINE_IPADDR_V6:
115                 snprintf(dstbuf, size, "IPv4/IPv6");
116                 break;
117         case CMDLINE_IPADDR_NETWORK|CMDLINE_IPADDR_V4:
118                 snprintf(dstbuf, size, "IPv4 network");
119                 break;
120         case CMDLINE_IPADDR_NETWORK|CMDLINE_IPADDR_V6:
121                 snprintf(dstbuf, size, "IPv6 network");
122                 break;
123         case CMDLINE_IPADDR_NETWORK|CMDLINE_IPADDR_V4|CMDLINE_IPADDR_V6:
124                 snprintf(dstbuf, size, "IPv4/IPv6 network");
125                 break;
126         default:
127                 snprintf(dstbuf, size, "IPaddr (bad flags)");
128                 break;
129         }
130         return 0;
131 }