lib/librte_cmdline/cmdline_parse_portlist.c

   1 /*-
   2  *   BSD LICENSE
   3  *
   4  *   Copyright(c) 2010-2014 Intel Corporation. All rights reserved.
   5  *   All rights reserved.
   6  *
   7  *   Redistribution and use in source and binary forms, with or without
   8  *   modification, are permitted provided that the following conditions
   9  *   are met:
  10  *
  11  *     * Redistributions of source code must retain the above copyright
  12  *       notice, this list of conditions and the following disclaimer.
  13  *     * Redistributions in binary form must reproduce the above copyright
  14  *       notice, this list of conditions and the following disclaimer in
  15  *       the documentation and/or other materials provided with the
  16  *       distribution.
  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 FOR
  24  *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  25  *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  26  *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  27  *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  28  *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  29  *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  30  *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  31  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32  */
  33
  34 /*
  35  * Copyright (c) 2010, Keith Wiles <keith.wiles@windriver.com>
  36  * All rights reserved.
  37  * Redistribution and use in source and binary forms, with or without
  38  * modification, are permitted provided that the following conditions are met:
  39  *
  40  *     * Redistributions of source code must retain the above copyright
  41  *       notice, this list of conditions and the following disclaimer.
  42  *     * Redistributions in binary form must reproduce the above copyright
  43  *       notice, this list of conditions and the following disclaimer in the
  44  *       documentation and/or other materials provided with the distribution.
  45  *     * Neither the name of the University of California, Berkeley nor the
  46  *       names of its contributors may be used to endorse or promote products
  47  *       derived from this software without specific prior written permission.
  48  *
  49  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND ANY
  50  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  51  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  52  * DISCLAIMED. IN NO EVENT SHALL THE REGENTS AND CONTRIBUTORS BE LIABLE FOR ANY
  53  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  54  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  55  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  56  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  57  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  58  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  59  */
  60
  61 #include <stdio.h>
  62 #include <stdlib.h>
  63 #include <inttypes.h>
  64 #include <ctype.h>
  65 #include <string.h>
  66 #include <errno.h>
  67 #include <stdarg.h>
  68 #include <netinet/in.h>
  69 #ifndef __linux__
  70 #include <net/socket.h>
  71 #endif
  72
  73 #include <rte_string_fns.h>
  74 #include "cmdline_parse.h"
  75 #include "cmdline_parse_portlist.h"
  76
  77 struct cmdline_token_ops cmdline_token_portlist_ops = {
  78         .parse = cmdline_parse_portlist,
  79         .complete_get_nb = NULL,
  80         .complete_get_elt = NULL,
  81         .get_help = cmdline_get_help_portlist,
  82 };
  83
  84 static void
  85 parse_set_list(cmdline_portlist_t * pl, int low, int high)
  86 {
  87         do {
  88                 pl->map |= (1 << low++);
  89         } while (low <= high);
  90 }
  91
  92 static int
  93 parse_ports(cmdline_portlist_t * pl, const char * str)
  94 {
  95         size_t ps, pe;
  96         const char *first, *last;
  97         char *end;
  98
  99         for (first = str, last = first;
 100             first != NULL && last != NULL;
 101             first = last + 1) {
 102
 103                 last = strchr(first, ',');
 104
 105                 errno = 0;
 106                 ps = strtoul(first, &end, 10);
 107                 if (errno != 0 || end == first ||
 108                     (end[0] != '-' && end[0] != 0 && end != last))
 109                         return (-1);
 110
 111                 /* Support for N-M portlist format */
 112                 if (end[0] == '-') {
 113                         errno = 0;
 114                         first = end + 1;
 115                         pe = strtoul(first, &end, 10);
 116                         if (errno != 0 || end == first ||
 117                             (end[0] != 0 && end != last))
 118                                 return (-1);
 119                 } else {
 120                         pe = ps;
 121                 }
 122
 123                 if (ps > pe || pe >= sizeof (pl->map) * 8)
 124                         return (-1);
 125
 126                 parse_set_list(pl, ps, pe);
 127         }
 128
 129         return (0);
 130 }
 131
 132 int
 133 cmdline_parse_portlist(__attribute__((unused)) cmdline_parse_token_hdr_t *tk,
 134                 const char *buf, void *res)
 135 {
 136         unsigned int token_len = 0;
 137         char portlist_str[PORTLIST_TOKEN_SIZE+1];
 138         cmdline_portlist_t *pl;
 139
 140         if (!buf || ! *buf)
 141                 return (-1);
 142
 143         pl = res;
 144
 145         while (!cmdline_isendoftoken(buf[token_len]) &&
 146             (token_len < PORTLIST_TOKEN_SIZE))
 147                 token_len++;
 148
 149         if (token_len >= PORTLIST_TOKEN_SIZE)
 150                 return (-1);
 151
 152         rte_snprintf(portlist_str, token_len+1, "%s", buf);
 153
 154         if (pl) {
 155                 pl->map = 0;
 156                 if (strcmp("all", portlist_str) == 0)
 157                         pl->map = UINT32_MAX;
 158                 else if (parse_ports(pl, portlist_str) != 0)
 159                         return (-1);
 160         }
 161
 162         return token_len;
 163 }
 164
 165 int
 166 cmdline_get_help_portlist(__attribute__((unused)) cmdline_parse_token_hdr_t *tk,
 167                 char *dstbuf, unsigned int size)
 168 {
 169         int ret;
 170         ret = rte_snprintf(dstbuf, size, "range of ports as 3,4-6,8-19,20");
 171         if (ret < 0)
 172                 return -1;
 173         return 0;
 174 }