GCC Code Coverage Report
Directory: ./ Exec Total Coverage
File: usr.sbin/relayctl/parser.c Lines: 0 70 0.0 %
Date: 2017-11-07 Branches: 0 62 0.0 %

Line Branch Exec Source
1
/*	$OpenBSD: parser.c,v 1.27 2015/01/22 17:42:09 reyk Exp $	*/
2
3
/*
4
 * Copyright (c) 2006 Pierre-Yves Ritschard <pyr@openbsd.org>
5
 * Copyright (c) 2004 Esben Norby <norby@openbsd.org>
6
 * Copyright (c) 2003, 2004 Henning Brauer <henning@openbsd.org>
7
 *
8
 * Permission to use, copy, modify, and distribute this software for any
9
 * purpose with or without fee is hereby granted, provided that the above
10
 * copyright notice and this permission notice appear in all copies.
11
 *
12
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
13
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
14
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
15
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
16
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
17
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
18
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
19
 */
20
21
#include <sys/types.h>
22
23
#include <limits.h>
24
#include <stdio.h>
25
#include <stdlib.h>
26
#include <string.h>
27
28
#include "relayd.h"
29
#include "parser.h"
30
31
enum token_type {
32
	NOTOKEN,
33
	ENDTOKEN,
34
	HOSTID,
35
	TABLEID,
36
	RDRID,
37
	KEYWORD,
38
	PATH
39
};
40
41
struct token {
42
	enum token_type		 type;
43
	const char		*keyword;
44
	int			 value;
45
	const struct token	*next;
46
};
47
48
static const struct token t_main[];
49
static const struct token t_show[];
50
static const struct token t_rdr[];
51
static const struct token t_table[];
52
static const struct token t_host[];
53
static const struct token t_rdr_id[];
54
static const struct token t_table_id[];
55
static const struct token t_host_id[];
56
static const struct token t_log[];
57
static const struct token t_load[];
58
59
static const struct token t_main[] = {
60
	{KEYWORD,	"monitor",	MONITOR,	NULL},
61
	{KEYWORD,	"show",		NONE,		t_show},
62
	{KEYWORD,	"load",		LOAD,		t_load},
63
	{KEYWORD,	"poll",		POLL,		NULL},
64
	{KEYWORD,	"reload",	RELOAD,		NULL},
65
	{KEYWORD,	"stop",		SHUTDOWN,	NULL},
66
	{KEYWORD,	"redirect",	NONE,		t_rdr},
67
	{KEYWORD,	"table",	NONE,		t_table},
68
	{KEYWORD,	"host",		NONE,		t_host},
69
	{KEYWORD,	"log",		NONE,		t_log},
70
	{ENDTOKEN,	"",		NONE,		NULL}
71
};
72
73
static const struct token t_show[] = {
74
	{KEYWORD,	"summary",	SHOW_SUM,	NULL},
75
	{KEYWORD,	"hosts",	SHOW_HOSTS,	NULL},
76
	{KEYWORD,	"redirects",	SHOW_RDRS,	NULL},
77
	{KEYWORD,	"relays",	SHOW_RELAYS,	NULL},
78
	{KEYWORD,	"routers",	SHOW_ROUTERS,	NULL},
79
	{KEYWORD,	"sessions",	SHOW_SESSIONS,	NULL},
80
	{ENDTOKEN,	"",		NONE,		NULL}
81
};
82
83
static const struct token t_rdr[] = {
84
	{NOTOKEN,	"",		NONE,		NULL},
85
	{KEYWORD,	"disable",	RDR_DISABLE,	t_rdr_id},
86
	{KEYWORD,	"enable",	RDR_ENABLE,	t_rdr_id},
87
	{ENDTOKEN,	"",		NONE,		NULL}
88
};
89
90
static const struct token t_table[] = {
91
	{NOTOKEN,	"",		NONE,		NULL},
92
	{KEYWORD,	"disable",	TABLE_DISABLE,	t_table_id},
93
	{KEYWORD,	"enable",	TABLE_ENABLE,	t_table_id},
94
	{ENDTOKEN,	"",		NONE,		NULL}
95
};
96
97
static const struct token t_host[] = {
98
	{NOTOKEN,	"",		NONE,		NULL},
99
	{KEYWORD,	"disable",	HOST_DISABLE,	t_host_id},
100
	{KEYWORD,	"enable",	HOST_ENABLE,	t_host_id},
101
	{ENDTOKEN,	"",		NONE,		NULL}
102
};
103
104
static const struct token t_rdr_id[] = {
105
	{RDRID,		"",		NONE,		NULL},
106
	{ENDTOKEN,	"",		NONE,		NULL}
107
};
108
109
static const struct token t_table_id[] = {
110
	{TABLEID,	"",		NONE,		NULL},
111
	{ENDTOKEN,	"",		NONE,		NULL}
112
};
113
114
static const struct token t_host_id[] = {
115
	{HOSTID,	"",		NONE,		NULL},
116
	{ENDTOKEN,	"",		NONE,		NULL}
117
};
118
119
static const struct token t_log[] = {
120
	{KEYWORD,	"verbose",	LOG_VERBOSE,	NULL},
121
	{KEYWORD,	"brief",	LOG_BRIEF,	NULL},
122
	{ENDTOKEN,	"",		NONE,		NULL}
123
};
124
125
static const struct token t_load[] = {
126
	{PATH,		"",		NONE,		NULL},
127
	{ENDTOKEN,	"",		NONE,		NULL}
128
};
129
130
static const struct token *match_token(const char *, const struct token *,
131
    struct parse_result *);
132
static void show_valid_args(const struct token *);
133
134
struct parse_result *
135
parse(int argc, char *argv[])
136
{
137
	static struct parse_result	res;
138
	const struct token	*table = t_main;
139
	const struct token	*match;
140
141
	bzero(&res, sizeof(res));
142
143
	while (argc >= 0) {
144
		if ((match = match_token(argv[0], table, &res)) == NULL) {
145
			fprintf(stderr, "valid commands/args:\n");
146
			show_valid_args(table);
147
			return (NULL);
148
		}
149
150
		argc--;
151
		argv++;
152
153
		if (match->type == NOTOKEN || match->next == NULL)
154
			break;
155
156
		table = match->next;
157
	}
158
159
	if (argc > 0) {
160
		fprintf(stderr, "superfluous argument: %s\n", argv[0]);
161
		return (NULL);
162
	}
163
164
	return (&res);
165
}
166
167
static const struct token *
168
match_token(const char *word, const struct token *table,
169
    struct parse_result *res)
170
{
171
	u_int			 i, match;
172
	const struct token	*t = NULL;
173
	const char		*errstr;
174
175
	match = 0;
176
177
	for (i = 0; table[i].type != ENDTOKEN; i++) {
178
		switch (table[i].type) {
179
		case NOTOKEN:
180
			if (word == NULL || strlen(word) == 0) {
181
				match++;
182
				t = &table[i];
183
			}
184
			break;
185
		case KEYWORD:
186
			if (word != NULL && strncmp(word, table[i].keyword,
187
			    strlen(word)) == 0) {
188
				match++;
189
				t = &table[i];
190
				if (t->value)
191
					res->action = t->value;
192
			}
193
			break;
194
		case HOSTID:
195
			if (word == NULL)
196
				break;
197
			res->id.id = strtonum(word, 0, UINT_MAX, &errstr);
198
			if (errstr) {
199
				strlcpy(res->id.name, word,
200
				    sizeof(res->id.name));
201
				res->id.id = EMPTY_ID;
202
			}
203
			t = &table[i];
204
			match++;
205
			break;
206
		case TABLEID:
207
			if (word == NULL)
208
				break;
209
			res->id.id = strtonum(word, 0, UINT_MAX, &errstr);
210
			if (errstr) {
211
				strlcpy(res->id.name, word,
212
				    sizeof(res->id.name));
213
				res->id.id = EMPTY_ID;
214
			}
215
			t = &table[i];
216
			match++;
217
			break;
218
		case RDRID:
219
			if (word == NULL)
220
				break;
221
			res->id.id = strtonum(word, 0, UINT_MAX, &errstr);
222
			if (errstr) {
223
				strlcpy(res->id.name, word,
224
				    sizeof(res->id.name));
225
				res->id.id = EMPTY_ID;
226
			}
227
			t = &table[i];
228
			match++;
229
			break;
230
		case PATH:
231
			if (!match && word != NULL && strlen(word) > 0) {
232
				res->path = strdup(word);
233
				match++;
234
				t = &table[i];
235
			}
236
			break;
237
		case ENDTOKEN:
238
			break;
239
		}
240
	}
241
242
	if (match != 1) {
243
		if (word == NULL)
244
			fprintf(stderr, "missing argument:\n");
245
		else if (match > 1)
246
			fprintf(stderr, "ambiguous argument: %s\n", word);
247
		else if (match < 1)
248
			fprintf(stderr, "unknown argument: %s\n", word);
249
		return (NULL);
250
	}
251
252
	return (t);
253
}
254
255
static void
256
show_valid_args(const struct token *table)
257
{
258
	int	i;
259
260
	for (i = 0; table[i].type != ENDTOKEN; i++) {
261
		switch (table[i].type) {
262
		case NOTOKEN:
263
			fprintf(stderr, "  <cr>\n");
264
			break;
265
		case KEYWORD:
266
			fprintf(stderr, "  %s\n", table[i].keyword);
267
			break;
268
		case RDRID:
269
			fprintf(stderr, "  <redirectid>\n");
270
			break;
271
		case TABLEID:
272
			fprintf(stderr, "  <tableid>\n");
273
			break;
274
		case HOSTID:
275
			fprintf(stderr, "  <hostid>\n");
276
			break;
277
		case PATH:
278
			fprintf(stderr, "  <path>\n");
279
			break;
280
		case ENDTOKEN:
281
			break;
282
		}
283
	}
284
}