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 |
|
|
} |