lib/cmd/ucg_cmd_parse_ipaddr.c

   1 /*
   2  * Copyright (c) 2009-2015, Olivier MATZ <zer0@droids-corp.org>
   3  *
   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 /*-
  29  * Copyright (c) <2010>, Intel Corporation
  30  * All rights reserved.
  31  *
  32  * Redistribution and use in source and binary forms, with or without
  33  * modification, are permitted provided that the following conditions
  34  * are met:
  35  *
  36  * - Redistributions of source code must retain the above copyright
  37  *   notice, this list of conditions and the following disclaimer.
  38  *
  39  * - Redistributions in binary form must reproduce the above copyright
  40  *   notice, this list of conditions and the following disclaimer in
  41  *   the documentation and/or other materials provided with the
  42  *   distribution.
  43  *
  44  * - Neither the name of Intel Corporation nor the names of its
  45  *   contributors may be used to endorse or promote products derived
  46  *   from this software without specific prior written permission.
  47  *
  48  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  49  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  50  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  51  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  52  * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  53  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  54  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  55  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  56  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  57  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  58  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
  59  * OF THE POSSIBILITY OF SUCH DAMAGE.
  60  */
  61
  62 /*
  63  * For inet_ntop() functions:
  64  *
  65  * Copyright (c) 1996 by Internet Software Consortium.
  66  *
  67  * Permission to use, copy, modify, and distribute this software for any
  68  * purpose with or without fee is hereby granted, provided that the above
  69  * copyright notice and this permission notice appear in all copies.
  70  *
  71  * THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SOFTWARE CONSORTIUM DISCLAIMS
  72  * ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES
  73  * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL INTERNET SOFTWARE
  74  * CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
  75  * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
  76  * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
  77  * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
  78  * SOFTWARE.
  79  */
  80
  81
  82 #include <stdio.h>
  83 #include <stdlib.h>
  84 #include <inttypes.h>
  85 #include <ctype.h>
  86 #include <string.h>
  87 #include <errno.h>
  88 #include <netinet/in.h>
  89 #ifndef __linux__
  90 #include <net/socket.h>
  91 #endif
  92
  93 #include <ucg_cmd_parse.h>
  94 #include <ucg_cmd_parse_ipaddr.h>
  95
  96 #define INADDRSZ 4
  97 #define IN6ADDRSZ 16
  98
  99 /*
 100  * WARNING: Don't even consider trying to compile this on a system where
 101  * sizeof(int) < 4.  sizeof(int) > 4 is fine; all the world's not a VAX.
 102  */
 103
 104 static int inet_pton4(const char *src, unsigned char *dst);
 105 static int inet_pton6(const char *src, unsigned char *dst);
 106
 107 /* int
 108  * inet_pton(af, src, dst)
 109  *      convert from presentation format (which usually means ASCII printable)
 110  *      to network format (which is usually some kind of binary format).
 111  * return:
 112  *      1 if the address was valid for the specified address family
 113  *      0 if the address wasn't valid (`dst' is untouched in this case)
 114  *      -1 if some other error occurred (`dst' is untouched in this case, too)
 115  * author:
 116  *      Paul Vixie, 1996.
 117  */
 118 static int
 119 my_inet_pton(int af, const char *src, void *dst)
 120 {
 121         switch (af) {
 122                 case AF_INET:
 123                         return (inet_pton4(src, dst));
 124                 case AF_INET6:
 125                         return (inet_pton6(src, dst));
 126                 default:
 127                         errno = EAFNOSUPPORT;
 128                         return (-1);
 129         }
 130         /* NOTREACHED */
 131 }
 132
 133 /* int
 134  * inet_pton4(src, dst)
 135  *      like inet_aton() but without all the hexadecimal and shorthand.
 136  * return:
 137  *      1 if `src' is a valid dotted quad, else 0.
 138  * notice:
 139  *      does not touch `dst' unless it's returning 1.
 140  * author:
 141  *      Paul Vixie, 1996.
 142  */
 143 static int
 144 inet_pton4(const char *src, unsigned char *dst)
 145 {
 146         static const char digits[] = "0123456789";
 147         int saw_digit, octets, ch;
 148         unsigned char tmp[INADDRSZ], *tp;
 149
 150         saw_digit = 0;
 151         octets = 0;
 152         *(tp = tmp) = 0;
 153         while ((ch = *src++) != '\0') {
 154                 const char *pch;
 155
 156                 if ((pch = strchr(digits, ch)) != NULL) {
 157                         unsigned int new = *tp * 10 + (pch - digits);
 158
 159                         if (new > 255)
 160                                 return (0);
 161                         if (! saw_digit) {
 162                                 if (++octets > 4)
 163                                         return (0);
 164                                 saw_digit = 1;
 165                         }
 166                         *tp = (unsigned char)new;
 167                 } else if (ch == '.' && saw_digit) {
 168                         if (octets == 4)
 169                                 return (0);
 170                         *++tp = 0;
 171                         saw_digit = 0;
 172                 } else
 173                         return (0);
 174         }
 175         if (octets < 4)
 176                 return (0);
 177
 178         memcpy(dst, tmp, INADDRSZ);
 179         return (1);
 180 }
 181
 182 /* int
 183  * inet_pton6(src, dst)
 184  *      convert presentation level address to network order binary form.
 185  * return:
 186  *      1 if `src' is a valid [RFC1884 2.2] address, else 0.
 187  * notice:
 188  *      (1) does not touch `dst' unless it's returning 1.
 189  *      (2) :: in a full address is silently ignored.
 190  * credit:
 191  *      inspired by Mark Andrews.
 192  * author:
 193  *      Paul Vixie, 1996.
 194  */
 195 static int
 196 inet_pton6(const char *src, unsigned char *dst)
 197 {
 198         static const char xdigits_l[] = "0123456789abcdef",
 199                 xdigits_u[] = "0123456789ABCDEF";
 200         unsigned char tmp[IN6ADDRSZ], *tp, *endp, *colonp;
 201         const char *xdigits, *curtok;
 202         int ch, saw_xdigit, count_xdigit;
 203         unsigned int val;
 204
 205         memset((tp = tmp), '\0', IN6ADDRSZ);
 206         endp = tp + IN6ADDRSZ;
 207         colonp = NULL;
 208         /* Leading :: requires some special handling. */
 209         if (*src == ':')
 210                 if (*++src != ':')
 211                         return (0);
 212         curtok = src;
 213         saw_xdigit = count_xdigit = 0;
 214         val = 0;
 215         while ((ch = *src++) != '\0') {
 216                 const char *pch;
 217
 218                 if ((pch = strchr((xdigits = xdigits_l), ch)) == NULL)
 219                         pch = strchr((xdigits = xdigits_u), ch);
 220                 if (pch != NULL) {
 221                         if (count_xdigit >= 4)
 222                                 return (0);
 223                         val <<= 4;
 224                         val |= (pch - xdigits);
 225                         if (val > 0xffff)
 226                                 return (0);
 227                         saw_xdigit = 1;
 228                         count_xdigit++;
 229                         continue;
 230                 }
 231                 if (ch == ':') {
 232                         curtok = src;
 233                         if (!saw_xdigit) {
 234                                 if (colonp)
 235                                         return (0);
 236                                 colonp = tp;
 237                                 continue;
 238                         } else if (*src == '\0') {
 239                                 return (0);
 240                         }
 241                         if (tp + sizeof(int16_t) > endp)
 242                                 return (0);
 243                         *tp++ = (unsigned char) ((val >> 8) & 0xff);
 244                         *tp++ = (unsigned char) (val & 0xff);
 245                         saw_xdigit = 0;
 246                         count_xdigit = 0;
 247                         val = 0;
 248                         continue;
 249                 }
 250                 if (ch == '.' && ((tp + INADDRSZ) <= endp) &&
 251                     inet_pton4(curtok, tp) > 0) {
 252                         tp += INADDRSZ;
 253                         saw_xdigit = 0;
 254                         count_xdigit = 0;
 255                         break;  /* '\0' was seen by inet_pton4(). */
 256                 }
 257                 return (0);
 258         }
 259         if (saw_xdigit) {
 260                 if (tp + sizeof(int16_t) > endp)
 261                         return (0);
 262                 *tp++ = (unsigned char) ((val >> 8) & 0xff);
 263                 *tp++ = (unsigned char) (val & 0xff);
 264         }
 265         if (colonp != NULL) {
 266                 /*
 267                  * Since some memmove()'s erroneously fail to handle
 268                  * overlapping regions, we'll do the shift by hand.
 269                  */
 270                 const int n = tp - colonp;
 271                 int i;
 272
 273                 for (i = 1; i <= n; i++) {
 274                         endp[- i] = colonp[n - i];
 275                         colonp[n - i] = 0;
 276                 }
 277                 tp = endp;
 278         }
 279         if (tp != endp)
 280                 return (0);
 281         memcpy(dst, tmp, IN6ADDRSZ);
 282         return (1);
 283 }
 284
 285 static int
 286 cmd_parse_ipaddr(ucg_cmd_tk_hdr_t *tk, const char *buf,
 287         void *res, unsigned ressize)
 288 {
 289         struct ucg_cmd_tk_ipaddr *tk2 =
 290                 (struct ucg_cmd_tk_ipaddr *)tk;
 291         unsigned int token_len = 0;
 292         char ip_str[INET6_ADDRSTRLEN+4]; /* '+4' is for prefixlen (if any) */
 293         ucg_cmd_ipaddr_t ipaddr;
 294         char *prefix, *prefix_end;
 295         long prefixlen;
 296
 297         if (res && ressize < sizeof(ucg_cmd_ipaddr_t))
 298                 return -1;
 299
 300         memset(&ipaddr, 0, sizeof(ipaddr));
 301
 302         /* if token is too big... */
 303         token_len = snprintf(ip_str, sizeof(ip_str), "%s", buf);
 304         if (token_len >= sizeof(ip_str))
 305                 return -1;
 306
 307         /* convert the network prefix */
 308         if (tk2->ipaddr_data.flags & UCG_CMD_IPADDR_NETWORK) {
 309                 prefix = strrchr(ip_str, '/');
 310                 if (prefix == NULL)
 311                         return -1;
 312                 *prefix = '\0';
 313                 prefix ++;
 314                 errno = 0;
 315                 prefixlen = strtol(prefix, &prefix_end, 10);
 316                 if (errno || (*prefix_end != '\0') )
 317                         return -1;
 318                 ipaddr.prefixlen = prefixlen;
 319         }
 320         else {
 321                 ipaddr.prefixlen = 0;
 322         }
 323
 324         /* convert the IP addr */
 325         if ((tk2->ipaddr_data.flags & UCG_CMD_IPADDR_V4) &&
 326             my_inet_pton(AF_INET, ip_str, &ipaddr.addr.ipv4) == 1) {
 327                 ipaddr.family = AF_INET;
 328                 if (res != NULL)
 329                         memcpy(res, &ipaddr, sizeof(ipaddr));
 330                 return 0;
 331         }
 332         if ((tk2->ipaddr_data.flags & UCG_CMD_IPADDR_V6) &&
 333             my_inet_pton(AF_INET6, ip_str, &ipaddr.addr.ipv6) == 1) {
 334                 ipaddr.family = AF_INET6;
 335                 if (res != NULL)
 336                         memcpy(res, &ipaddr, sizeof(ipaddr));
 337                 return 0;
 338         }
 339         return -1;
 340
 341 }
 342
 343 static int
 344 cmd_help_ipaddr(ucg_cmd_tk_hdr_t *tk, char *dstbuf,
 345                     unsigned int size)
 346 {
 347         struct ucg_cmd_tk_ipaddr *tk2 =
 348                 (struct ucg_cmd_tk_ipaddr *)tk;
 349
 350         switch (tk2->ipaddr_data.flags) {
 351         case UCG_CMD_IPADDR_V4:
 352                 snprintf(dstbuf, size, "<IPv4>");
 353                 break;
 354         case UCG_CMD_IPADDR_V6:
 355                 snprintf(dstbuf, size, "<IPv6>");
 356                 break;
 357         case UCG_CMD_IPADDR_V4 | UCG_CMD_IPADDR_V6:
 358                 snprintf(dstbuf, size, "<IPv4/IPv6>");
 359                 break;
 360         case UCG_CMD_IPADDR_NETWORK | UCG_CMD_IPADDR_V4:
 361                 snprintf(dstbuf, size, "<IPv4 network>");
 362                 break;
 363         case UCG_CMD_IPADDR_NETWORK | UCG_CMD_IPADDR_V6:
 364                 snprintf(dstbuf, size, "<IPv6 network>");
 365                 break;
 366         case UCG_CMD_IPADDR_NETWORK | UCG_CMD_IPADDR_V4 |
 367                 UCG_CMD_IPADDR_V6:
 368                 snprintf(dstbuf, size, "<IPv4/IPv6 network>");
 369                 break;
 370         default:
 371                 snprintf(dstbuf, size, "<IPaddr (bad flags)>");
 372                 break;
 373         }
 374         return 0;
 375 }
 376
 377 struct ucg_cmd_tk_ops cmd_token_ipaddr_ops = {
 378         .parse = cmd_parse_ipaddr,
 379         .complete_start = NULL,
 380         .complete_iterate = NULL,
 381         .complete_end = NULL,
 382         .help = cmd_help_ipaddr,
 383 };
 384