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 }