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
69#include <rte_string_fns.h>
70#include "cmdline_parse.h"
71#include "cmdline_parse_portlist.h"
72
73struct cmdline_token_ops cmdline_token_portlist_ops = {
74	.parse = cmdline_parse_portlist,
75	.complete_get_nb = NULL,
76	.complete_get_elt = NULL,
77	.get_help = cmdline_get_help_portlist,
78};
79
80static void
81parse_set_list(cmdline_portlist_t *pl, size_t low, size_t high)
82{
83	do {
84		pl->map |= (1 << low++);
85	} while (low <= high);
86}
87
88static int
89parse_ports(cmdline_portlist_t *pl, const char *str)
90{
91	size_t ps, pe;
92	const char *first, *last;
93	char *end;
94
95	for (first = str, last = first;
96	    first != NULL && last != NULL;
97	    first = last + 1) {
98
99		last = strchr(first, ',');
100
101		errno = 0;
102		ps = strtoul(first, &end, 10);
103		if (errno != 0 || end == first ||
104		    (end[0] != '-' && end[0] != 0 && end != last))
105			return -1;
106
107		/* Support for N-M portlist format */
108		if (end[0] == '-') {
109			errno = 0;
110			first = end + 1;
111			pe = strtoul(first, &end, 10);
112			if (errno != 0 || end == first ||
113			    (end[0] != 0 && end != last))
114				return -1;
115		} else {
116			pe = ps;
117		}
118
119		if (ps > pe || pe >= sizeof (pl->map) * 8)
120			return -1;
121
122		parse_set_list(pl, ps, pe);
123	}
124
125	return 0;
126}
127
128int
129cmdline_parse_portlist(__attribute__((unused)) cmdline_parse_token_hdr_t *tk,
130	const char *buf, void *res, unsigned ressize)
131{
132	unsigned int token_len = 0;
133	char portlist_str[PORTLIST_TOKEN_SIZE+1];
134	cmdline_portlist_t *pl;
135
136	if (!buf || ! *buf)
137		return -1;
138
139	if (res && ressize < sizeof(cmdline_portlist_t))
140		return -1;
141
142	pl = res;
143
144	while (!cmdline_isendoftoken(buf[token_len]) &&
145	    (token_len < PORTLIST_TOKEN_SIZE))
146		token_len++;
147
148	if (token_len >= PORTLIST_TOKEN_SIZE)
149		return -1;
150
151	snprintf(portlist_str, token_len+1, "%s", buf);
152
153	if (pl) {
154		pl->map = 0;
155		if (strcmp("all", portlist_str) == 0)
156			pl->map	= UINT32_MAX;
157		else if (parse_ports(pl, portlist_str) != 0)
158			return -1;
159	}
160
161	return token_len;
162}
163
164int
165cmdline_get_help_portlist(__attribute__((unused)) cmdline_parse_token_hdr_t *tk,
166		char *dstbuf, unsigned int size)
167{
168	int ret;
169	ret = snprintf(dstbuf, size, "range of ports as 3,4-6,8-19,20");
170	if (ret < 0)
171		return -1;
172	return 0;
173}
174