src/lib/cmdline_parse_ipaddr.c

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