xref: /freebsd/sbin/ipfw/tables.c (revision dfd822b1f0846bdaa9a14457346f7431a86d3a64)
1 /*
2  * Copyright (c) 2014 Yandex LLC
3  * Copyright (c) 2014 Alexander V. Chernikov
4  *
5  * Redistribution and use in source forms, with and without modification,
6  * are permitted provided that this entire comment appears intact.
7  *
8  * Redistribution in binary form may occur without any restrictions.
9  * Obviously, it would be nice if you gave credit where credit is due
10  * but requiring it would be too onerous.
11  *
12  * This software is provided ``AS IS'' without any warranties of any kind.
13  *
14  * in-kernel ipfw tables support.
15  */
16 
17 
18 #include <sys/types.h>
19 #include <sys/param.h>
20 #include <sys/socket.h>
21 #include <sys/sysctl.h>
22 
23 #include <ctype.h>
24 #include <err.h>
25 #include <errno.h>
26 #include <netdb.h>
27 #include <stdio.h>
28 #include <stdlib.h>
29 #include <string.h>
30 #include <sysexits.h>
31 
32 #include <net/ethernet.h>
33 #include <net/if.h>
34 #include <netinet/in.h>
35 #include <netinet/ip_fw.h>
36 #include <arpa/inet.h>
37 #include <netdb.h>
38 
39 #include "ipfw2.h"
40 
41 static void table_modify_record(ipfw_obj_header *oh, int ac, char *av[],
42     int add, int quiet, int update, int atomic);
43 static int table_flush(ipfw_obj_header *oh);
44 static int table_destroy(ipfw_obj_header *oh);
45 static int table_do_create(ipfw_obj_header *oh, ipfw_xtable_info *i);
46 static int table_do_modify(ipfw_obj_header *oh, ipfw_xtable_info *i);
47 static int table_do_swap(ipfw_obj_header *oh, char *second);
48 static void table_create(ipfw_obj_header *oh, int ac, char *av[]);
49 static void table_modify(ipfw_obj_header *oh, int ac, char *av[]);
50 static void table_lookup(ipfw_obj_header *oh, int ac, char *av[]);
51 static void table_lock(ipfw_obj_header *oh, int lock);
52 static int table_swap(ipfw_obj_header *oh, char *second);
53 static int table_get_info(ipfw_obj_header *oh, ipfw_xtable_info *i);
54 static int table_show_info(ipfw_xtable_info *i, void *arg);
55 
56 static int table_destroy_one(ipfw_xtable_info *i, void *arg);
57 static int table_flush_one(ipfw_xtable_info *i, void *arg);
58 static int table_show_one(ipfw_xtable_info *i, void *arg);
59 static int table_do_get_list(ipfw_xtable_info *i, ipfw_obj_header **poh);
60 static void table_show_list(ipfw_obj_header *oh, int need_header);
61 static void table_show_entry(ipfw_xtable_info *i, ipfw_obj_tentry *tent);
62 
63 static void tentry_fill_key(ipfw_obj_header *oh, ipfw_obj_tentry *tent,
64     char *key, int add, uint8_t *ptype, uint32_t *pvmask, ipfw_xtable_info *xi);
65 static void tentry_fill_value(ipfw_obj_header *oh, ipfw_obj_tentry *tent,
66     char *arg, uint8_t type, uint32_t vmask);
67 static void table_show_value(char *buf, size_t bufsize, ipfw_table_value *v,
68     uint32_t vmask, int print_ip);
69 
70 typedef int (table_cb_t)(ipfw_xtable_info *i, void *arg);
71 static int tables_foreach(table_cb_t *f, void *arg, int sort);
72 
73 #ifndef s6_addr32
74 #define s6_addr32 __u6_addr.__u6_addr32
75 #endif
76 
77 static struct _s_x tabletypes[] = {
78       { "addr",		IPFW_TABLE_ADDR },
79       { "mac",		IPFW_TABLE_MAC },
80       { "iface",	IPFW_TABLE_INTERFACE },
81       { "number",	IPFW_TABLE_NUMBER },
82       { "flow",		IPFW_TABLE_FLOW },
83       { NULL, 0 }
84 };
85 
86 /* Default algorithms for various table types */
87 static struct _s_x tablealgos[] = {
88       { "addr:radix",	IPFW_TABLE_ADDR },
89       { "flow:hash",	IPFW_TABLE_FLOW },
90       { "iface:array",	IPFW_TABLE_INTERFACE },
91       { "number:array",	IPFW_TABLE_NUMBER },
92       { NULL, 0 }
93 };
94 
95 static struct _s_x tablevaltypes[] = {
96       { "skipto",	IPFW_VTYPE_SKIPTO },
97       { "pipe",		IPFW_VTYPE_PIPE },
98       { "fib",		IPFW_VTYPE_FIB },
99       { "nat",		IPFW_VTYPE_NAT },
100       { "dscp",		IPFW_VTYPE_DSCP },
101       { "tag",		IPFW_VTYPE_TAG },
102       { "divert",	IPFW_VTYPE_DIVERT },
103       { "netgraph",	IPFW_VTYPE_NETGRAPH },
104       { "limit",	IPFW_VTYPE_LIMIT },
105       { "ipv4",		IPFW_VTYPE_NH4 },
106       { "ipv6",		IPFW_VTYPE_NH6 },
107       { "mark",		IPFW_VTYPE_MARK },
108       { NULL, 0 }
109 };
110 
111 static struct _s_x tablecmds[] = {
112       { "add",		TOK_ADD },
113       { "delete",	TOK_DEL },
114       { "create",	TOK_CREATE },
115       { "destroy",	TOK_DESTROY },
116       { "flush",	TOK_FLUSH },
117       { "modify",	TOK_MODIFY },
118       { "swap",		TOK_SWAP },
119       { "info",		TOK_INFO },
120       { "detail",	TOK_DETAIL },
121       { "list",		TOK_LIST },
122       { "lookup",	TOK_LOOKUP },
123       { "atomic",	TOK_ATOMIC },
124       { "lock",		TOK_LOCK },
125       { "unlock",	TOK_UNLOCK },
126       { NULL, 0 }
127 };
128 
129 static int
lookup_host(char * host,struct in_addr * ipaddr)130 lookup_host (char *host, struct in_addr *ipaddr)
131 {
132 	struct hostent *he;
133 
134 	if (!inet_aton(host, ipaddr)) {
135 		if ((he = gethostbyname(host)) == NULL)
136 			return(-1);
137 		*ipaddr = *(struct in_addr *)he->h_addr_list[0];
138 	}
139 	return(0);
140 }
141 
142 /*
143  * This one handles all table-related commands
144  * 	ipfw table NAME create ...
145  * 	ipfw table NAME modify ...
146  * 	ipfw table {NAME | all} destroy
147  * 	ipfw table NAME swap NAME
148  * 	ipfw table NAME lock
149  * 	ipfw table NAME unlock
150  * 	ipfw table NAME add addr[/masklen] [value]
151  * 	ipfw table NAME add [addr[/masklen] value] [addr[/masklen] value] ..
152  * 	ipfw table NAME delete addr[/masklen] [addr[/masklen]] ..
153  * 	ipfw table NAME lookup addr
154  * 	ipfw table {NAME | all} flush
155  * 	ipfw table {NAME | all} list
156  * 	ipfw table {NAME | all} info
157  * 	ipfw table {NAME | all} detail
158  */
159 void
ipfw_table_handler(int ac,char * av[])160 ipfw_table_handler(int ac, char *av[])
161 {
162 	int do_add, is_all;
163 	int atomic, error, tcmd;
164 	ipfw_xtable_info i;
165 	ipfw_obj_header oh;
166 	char *tablename;
167 	uint8_t set;
168 	void *arg;
169 
170 	memset(&oh, 0, sizeof(oh));
171 	is_all = 0;
172 	if (g_co.use_set != 0)
173 		set = g_co.use_set - 1;
174 	else
175 		set = 0;
176 
177 	ac--; av++;
178 	NEED1("table needs name");
179 	tablename = *av;
180 
181 	if (table_check_name(tablename) == 0) {
182 		table_fill_ntlv(&oh.ntlv, *av, set, 1);
183 		oh.idx = 1;
184 	} else {
185 		if (strcmp(tablename, "all") == 0)
186 			is_all = 1;
187 		else
188 			errx(EX_USAGE, "table name %s is invalid", tablename);
189 	}
190 	ac--; av++;
191 	NEED1("table needs command");
192 
193 	tcmd = get_token(tablecmds, *av, "table command");
194 	/* Check if atomic operation was requested */
195 	atomic = 0;
196 	if (tcmd == TOK_ATOMIC) {
197 		ac--; av++;
198 		NEED1("atomic needs command");
199 		tcmd = get_token(tablecmds, *av, "table command");
200 		switch (tcmd) {
201 		case TOK_ADD:
202 			break;
203 		default:
204 			errx(EX_USAGE, "atomic is not compatible with %s", *av);
205 		}
206 		atomic = 1;
207 	}
208 
209 	switch (tcmd) {
210 	case TOK_LIST:
211 	case TOK_INFO:
212 	case TOK_DETAIL:
213 	case TOK_FLUSH:
214 	case TOK_DESTROY:
215 		break;
216 	default:
217 		if (is_all != 0)
218 			errx(EX_USAGE, "table name required");
219 	}
220 
221 	switch (tcmd) {
222 	case TOK_ADD:
223 	case TOK_DEL:
224 		do_add = **av == 'a';
225 		ac--; av++;
226 		table_modify_record(&oh, ac, av, do_add, g_co.do_quiet,
227 		    g_co.do_quiet, atomic);
228 		break;
229 	case TOK_CREATE:
230 		ac--; av++;
231 		table_create(&oh, ac, av);
232 		break;
233 	case TOK_MODIFY:
234 		ac--; av++;
235 		table_modify(&oh, ac, av);
236 		break;
237 	case TOK_DESTROY:
238 		if (is_all == 0) {
239 			if (table_destroy(&oh) == 0)
240 				break;
241 			if (errno != ESRCH)
242 				err(EX_OSERR, "failed to destroy table %s",
243 				    tablename);
244 			/* ESRCH isn't fatal, warn if not quiet mode */
245 			if (g_co.do_quiet == 0)
246 				warn("failed to destroy table %s", tablename);
247 		} else {
248 			error = tables_foreach(table_destroy_one, &oh, 1);
249 			if (error != 0)
250 				err(EX_OSERR,
251 				    "failed to destroy tables list");
252 		}
253 		break;
254 	case TOK_FLUSH:
255 		if (is_all == 0) {
256 			if ((error = table_flush(&oh)) == 0)
257 				break;
258 			if (errno != ESRCH)
259 				err(EX_OSERR, "failed to flush table %s info",
260 				    tablename);
261 			/* ESRCH isn't fatal, warn if not quiet mode */
262 			if (g_co.do_quiet == 0)
263 				warn("failed to flush table %s info",
264 				    tablename);
265 		} else {
266 			error = tables_foreach(table_flush_one, &oh, 1);
267 			if (error != 0)
268 				err(EX_OSERR, "failed to flush tables list");
269 			/* XXX: we ignore errors here */
270 		}
271 		break;
272 	case TOK_SWAP:
273 		ac--; av++;
274 		NEED1("second table name required");
275 		table_swap(&oh, *av);
276 		break;
277 	case TOK_LOCK:
278 	case TOK_UNLOCK:
279 		table_lock(&oh, (tcmd == TOK_LOCK));
280 		break;
281 	case TOK_DETAIL:
282 	case TOK_INFO:
283 		arg = (tcmd == TOK_DETAIL) ? (void *)1 : NULL;
284 		if (is_all == 0) {
285 			if ((error = table_get_info(&oh, &i)) != 0)
286 				err(EX_OSERR, "failed to request table info");
287 			table_show_info(&i, arg);
288 		} else {
289 			error = tables_foreach(table_show_info, arg, 1);
290 			if (error != 0)
291 				err(EX_OSERR, "failed to request tables list");
292 		}
293 		break;
294 	case TOK_LIST:
295 		arg = is_all ? (void*)1 : NULL;
296 		if (is_all == 0) {
297 			if ((error = table_get_info(&oh, &i)) != 0)
298 				err(EX_OSERR, "failed to request table info");
299 			table_show_one(&i, arg);
300 		} else {
301 			error = tables_foreach(table_show_one, arg, 1);
302 			if (error != 0)
303 				err(EX_OSERR, "failed to request tables list");
304 		}
305 		break;
306 	case TOK_LOOKUP:
307 		ac--; av++;
308 		table_lookup(&oh, ac, av);
309 		break;
310 	}
311 }
312 
313 void
table_fill_ntlv(ipfw_obj_ntlv * ntlv,const char * name,uint8_t set,uint32_t uidx)314 table_fill_ntlv(ipfw_obj_ntlv *ntlv, const char *name, uint8_t set,
315     uint32_t uidx)
316 {
317 
318 	ntlv->head.type = IPFW_TLV_TBL_NAME;
319 	ntlv->head.length = sizeof(ipfw_obj_ntlv);
320 	ntlv->idx = uidx;
321 	ntlv->set = set;
322 	strlcpy(ntlv->name, name, sizeof(ntlv->name));
323 }
324 
325 static void
table_fill_objheader(ipfw_obj_header * oh,ipfw_xtable_info * i)326 table_fill_objheader(ipfw_obj_header *oh, ipfw_xtable_info *i)
327 {
328 
329 	oh->idx = 1;
330 	table_fill_ntlv(&oh->ntlv, i->tablename, i->set, 1);
331 }
332 
333 static struct _s_x tablenewcmds[] = {
334       { "type",		TOK_TYPE },
335       { "valtype",	TOK_VALTYPE },
336       { "algo",		TOK_ALGO },
337       { "limit",	TOK_LIMIT },
338       { "locked",	TOK_LOCK },
339       { "missing",	TOK_MISSING },
340       { "or-flush",	TOK_ORFLUSH },
341       { NULL, 0 }
342 };
343 
344 static struct _s_x flowtypecmds[] = {
345       { "src-ip",	IPFW_TFFLAG_SRCIP },
346       { "proto",	IPFW_TFFLAG_PROTO },
347       { "src-port",	IPFW_TFFLAG_SRCPORT },
348       { "dst-ip",	IPFW_TFFLAG_DSTIP },
349       { "dst-port",	IPFW_TFFLAG_DSTPORT },
350       { NULL, 0 }
351 };
352 
353 static int
table_parse_type(uint8_t ttype,char * p,uint8_t * tflags)354 table_parse_type(uint8_t ttype, char *p, uint8_t *tflags)
355 {
356 	uint32_t fset, fclear;
357 	char *e;
358 
359 	/* Parse type options */
360 	switch(ttype) {
361 	case IPFW_TABLE_FLOW:
362 		fset = fclear = 0;
363 		if (fill_flags(flowtypecmds, p, &e, &fset, &fclear) != 0)
364 			errx(EX_USAGE,
365 			    "unable to parse flow option %s", e);
366 		*tflags = fset;
367 		break;
368 	default:
369 		return (EX_USAGE);
370 	}
371 
372 	return (0);
373 }
374 
375 static void
table_print_type(char * tbuf,size_t size,uint8_t type,uint8_t tflags)376 table_print_type(char *tbuf, size_t size, uint8_t type, uint8_t tflags)
377 {
378 	const char *tname;
379 	int l;
380 
381 	if ((tname = match_value(tabletypes, type)) == NULL)
382 		tname = "unknown";
383 
384 	l = snprintf(tbuf, size, "%s", tname);
385 	tbuf += l;
386 	size -= l;
387 
388 	switch(type) {
389 	case IPFW_TABLE_FLOW:
390 		if (tflags != 0) {
391 			*tbuf++ = ':';
392 			l--;
393 			print_flags_buffer(tbuf, size, flowtypecmds, tflags);
394 		}
395 		break;
396 	}
397 }
398 
399 /*
400  * Creates new table
401  *
402  * ipfw table NAME create [ type { addr | iface | number | flow } ]
403  *     [ algo algoname ] [missing] [or-flush]
404  */
405 static void
table_create(ipfw_obj_header * oh,int ac,char * av[])406 table_create(ipfw_obj_header *oh, int ac, char *av[])
407 {
408 	ipfw_xtable_info xi, xie;
409 	int error, missing, orflush, tcmd, val;
410 	uint32_t fset, fclear;
411 	char *e, *p;
412 	char tbuf[128];
413 
414 	missing = orflush = 0;
415 	memset(&xi, 0, sizeof(xi));
416 	while (ac > 0) {
417 		tcmd = get_token(tablenewcmds, *av, "option");
418 		ac--; av++;
419 
420 		switch (tcmd) {
421 		case TOK_LIMIT:
422 			NEED1("limit value required");
423 			xi.limit = strtol(*av, NULL, 10);
424 			ac--; av++;
425 			break;
426 		case TOK_TYPE:
427 			NEED1("table type required");
428 			/* Type may have suboptions after ':' */
429 			if ((p = strchr(*av, ':')) != NULL)
430 				*p++ = '\0';
431 			val = match_token(tabletypes, *av);
432 			if (val == -1) {
433 				concat_tokens(tbuf, sizeof(tbuf), tabletypes,
434 				    ", ");
435 				errx(EX_USAGE,
436 				    "Unknown tabletype: %s. Supported: %s",
437 				    *av, tbuf);
438 			}
439 			xi.type = val;
440 			if (p != NULL) {
441 				error = table_parse_type(val, p, &xi.tflags);
442 				if (error != 0)
443 					errx(EX_USAGE,
444 					    "Unsupported suboptions: %s", p);
445 			}
446 			ac--; av++;
447 			break;
448 		case TOK_VALTYPE:
449 			NEED1("table value type required");
450 			fset = fclear = 0;
451 			val = fill_flags(tablevaltypes, *av, &e, &fset, &fclear);
452 			if (val != -1) {
453 				xi.vmask = fset;
454 				ac--; av++;
455 				break;
456 			}
457 			concat_tokens(tbuf, sizeof(tbuf), tablevaltypes, ", ");
458 			errx(EX_USAGE, "Unknown value type: %s. Supported: %s",
459 			    e, tbuf);
460 			break;
461 		case TOK_ALGO:
462 			NEED1("table algorithm name required");
463 			if (strlen(*av) > sizeof(xi.algoname))
464 				errx(EX_USAGE, "algorithm name too long");
465 			strlcpy(xi.algoname, *av, sizeof(xi.algoname));
466 			ac--; av++;
467 			break;
468 		case TOK_LOCK:
469 			xi.flags |= IPFW_TGFLAGS_LOCKED;
470 			break;
471 		case TOK_ORFLUSH:
472 			orflush = 1;
473 			/* FALLTHROUGH */
474 		case TOK_MISSING:
475 			missing = 1;
476 			break;
477 		}
478 	}
479 
480 	/* Set some defaults to preserve compatibility. */
481 	if (xi.algoname[0] == '\0') {
482 		const char *algo;
483 
484 		if (xi.type == 0)
485 			xi.type = IPFW_TABLE_ADDR;
486 		algo = match_value(tablealgos, xi.type);
487 		if (algo != NULL)
488 			strlcpy(xi.algoname, algo, sizeof(xi.algoname));
489 	}
490 	if (xi.vmask == 0)
491 		xi.vmask = IPFW_VTYPE_LEGACY;
492 
493 	error = table_do_create(oh, &xi);
494 
495 	if (error == 0)
496 		return;
497 
498 	if (errno != EEXIST || missing == 0)
499 		err(EX_OSERR, "Table creation failed");
500 
501 	/* Check that existing table is the same we are trying to create */
502 	if (table_get_info(oh, &xie) != 0)
503 		err(EX_OSERR, "Existing table check failed");
504 
505 	if (xi.limit != xie.limit || xi.type != xie.type ||
506 	    xi.tflags != xie.tflags || xi.vmask != xie.vmask || (
507 	    xi.algoname[0] != '\0' && strcmp(xi.algoname,
508 	    xie.algoname) != 0) || xi.flags != xie.flags)
509 		errx(EX_DATAERR, "The existing table is not compatible "
510 		    "with one you are creating.");
511 
512 	/* Flush existing table if instructed to do so */
513 	if (orflush != 0 && table_flush(oh) != 0)
514 		err(EX_OSERR, "Table flush on creation failed");
515 }
516 
517 /*
518  * Creates new table
519  *
520  * Request: [ ipfw_obj_header ipfw_xtable_info ]
521  *
522  * Returns 0 on success.
523  */
524 static int
table_do_create(ipfw_obj_header * oh,ipfw_xtable_info * i)525 table_do_create(ipfw_obj_header *oh, ipfw_xtable_info *i)
526 {
527 	char tbuf[sizeof(ipfw_obj_header) + sizeof(ipfw_xtable_info)];
528 	int error;
529 
530 	memcpy(tbuf, oh, sizeof(*oh));
531 	memcpy(tbuf + sizeof(*oh), i, sizeof(*i));
532 	oh = (ipfw_obj_header *)tbuf;
533 
534 	error = do_set3(IP_FW_TABLE_XCREATE, &oh->opheader, sizeof(tbuf));
535 
536 	return (error);
537 }
538 
539 /*
540  * Modifies existing table
541  *
542  * ipfw table NAME modify [ limit number ]
543  */
544 static void
table_modify(ipfw_obj_header * oh,int ac,char * av[])545 table_modify(ipfw_obj_header *oh, int ac, char *av[])
546 {
547 	ipfw_xtable_info xi;
548 	int tcmd;
549 
550 	memset(&xi, 0, sizeof(xi));
551 
552 	while (ac > 0) {
553 		tcmd = get_token(tablenewcmds, *av, "option");
554 		ac--; av++;
555 
556 		switch (tcmd) {
557 		case TOK_LIMIT:
558 			NEED1("limit value required");
559 			xi.limit = strtol(*av, NULL, 10);
560 			xi.mflags |= IPFW_TMFLAGS_LIMIT;
561 			ac--; av++;
562 			break;
563 		default:
564 			errx(EX_USAGE, "cmd is not supported for modification");
565 		}
566 	}
567 
568 	if (table_do_modify(oh, &xi) != 0)
569 		err(EX_OSERR, "Table modification failed");
570 }
571 
572 /*
573  * Modifies existing table.
574  *
575  * Request: [ ipfw_obj_header ipfw_xtable_info ]
576  *
577  * Returns 0 on success.
578  */
579 static int
table_do_modify(ipfw_obj_header * oh,ipfw_xtable_info * i)580 table_do_modify(ipfw_obj_header *oh, ipfw_xtable_info *i)
581 {
582 	char tbuf[sizeof(ipfw_obj_header) + sizeof(ipfw_xtable_info)];
583 	int error;
584 
585 	memcpy(tbuf, oh, sizeof(*oh));
586 	memcpy(tbuf + sizeof(*oh), i, sizeof(*i));
587 	oh = (ipfw_obj_header *)tbuf;
588 
589 	error = do_set3(IP_FW_TABLE_XMODIFY, &oh->opheader, sizeof(tbuf));
590 
591 	return (error);
592 }
593 
594 /*
595  * Locks or unlocks given table
596  */
597 static void
table_lock(ipfw_obj_header * oh,int lock)598 table_lock(ipfw_obj_header *oh, int lock)
599 {
600 	ipfw_xtable_info xi;
601 
602 	memset(&xi, 0, sizeof(xi));
603 
604 	xi.mflags |= IPFW_TMFLAGS_LOCK;
605 	xi.flags |= (lock != 0) ? IPFW_TGFLAGS_LOCKED : 0;
606 
607 	if (table_do_modify(oh, &xi) != 0)
608 		err(EX_OSERR, "Table %s failed", lock != 0 ? "lock" : "unlock");
609 }
610 
611 /*
612  * Destroys given table specified by @oh->ntlv.
613  * Returns 0 on success.
614  */
615 static int
table_destroy(ipfw_obj_header * oh)616 table_destroy(ipfw_obj_header *oh)
617 {
618 
619 	if (do_set3(IP_FW_TABLE_XDESTROY, &oh->opheader, sizeof(*oh)) != 0)
620 		return (-1);
621 
622 	return (0);
623 }
624 
625 static int
table_destroy_one(ipfw_xtable_info * i,void * arg)626 table_destroy_one(ipfw_xtable_info *i, void *arg)
627 {
628 	ipfw_obj_header *oh;
629 
630 	oh = (ipfw_obj_header *)arg;
631 	table_fill_ntlv(&oh->ntlv, i->tablename, i->set, 1);
632 	if (table_destroy(oh) != 0) {
633 		if (g_co.do_quiet == 0)
634 			warn("failed to destroy table(%s) in set %u",
635 			    i->tablename, i->set);
636 		return (-1);
637 	}
638 	return (0);
639 }
640 
641 /*
642  * Flushes given table specified by @oh->ntlv.
643  * Returns 0 on success.
644  */
645 static int
table_flush(ipfw_obj_header * oh)646 table_flush(ipfw_obj_header *oh)
647 {
648 
649 	if (do_set3(IP_FW_TABLE_XFLUSH, &oh->opheader, sizeof(*oh)) != 0)
650 		return (-1);
651 
652 	return (0);
653 }
654 
655 static int
table_do_swap(ipfw_obj_header * oh,char * second)656 table_do_swap(ipfw_obj_header *oh, char *second)
657 {
658 	char tbuf[sizeof(ipfw_obj_header) + sizeof(ipfw_obj_ntlv)];
659 	int error;
660 
661 	memset(tbuf, 0, sizeof(tbuf));
662 	memcpy(tbuf, oh, sizeof(*oh));
663 	oh = (ipfw_obj_header *)tbuf;
664 	table_fill_ntlv((ipfw_obj_ntlv *)(oh + 1), second, oh->ntlv.set, 1);
665 
666 	error = do_set3(IP_FW_TABLE_XSWAP, &oh->opheader, sizeof(tbuf));
667 
668 	return (error);
669 }
670 
671 /*
672  * Swaps given table with @second one.
673  */
674 static int
table_swap(ipfw_obj_header * oh,char * second)675 table_swap(ipfw_obj_header *oh, char *second)
676 {
677 
678 	if (table_check_name(second) != 0)
679 		errx(EX_USAGE, "table name %s is invalid", second);
680 
681 	if (table_do_swap(oh, second) == 0)
682 		return (0);
683 
684 	switch (errno) {
685 	case EINVAL:
686 		errx(EX_USAGE, "Unable to swap table: check types");
687 	case EFBIG:
688 		errx(EX_USAGE, "Unable to swap table: check limits");
689 	}
690 
691 	return (0);
692 }
693 
694 
695 /*
696  * Retrieves table in given table specified by @oh->ntlv.
697  * it inside @i.
698  * Returns 0 on success.
699  */
700 static int
table_get_info(ipfw_obj_header * oh,ipfw_xtable_info * i)701 table_get_info(ipfw_obj_header *oh, ipfw_xtable_info *i)
702 {
703 	char tbuf[sizeof(ipfw_obj_header) + sizeof(ipfw_xtable_info)];
704 	size_t sz;
705 
706 	sz = sizeof(tbuf);
707 	memset(tbuf, 0, sizeof(tbuf));
708 	memcpy(tbuf, oh, sizeof(*oh));
709 	oh = (ipfw_obj_header *)tbuf;
710 
711 	if (do_get3(IP_FW_TABLE_XINFO, &oh->opheader, &sz) != 0)
712 		return (errno);
713 
714 	if (sz < sizeof(tbuf))
715 		return (EINVAL);
716 
717 	*i = *(ipfw_xtable_info *)(oh + 1);
718 
719 	return (0);
720 }
721 
722 static struct _s_x tablealgoclass[] = {
723       { "hash",		IPFW_TACLASS_HASH },
724       { "array",	IPFW_TACLASS_ARRAY },
725       { "radix",	IPFW_TACLASS_RADIX },
726       { NULL, 0 }
727 };
728 
729 struct ta_cldata {
730 	uint8_t		taclass;
731 	uint8_t		spare4;
732 	uint16_t	itemsize;
733 	uint16_t	itemsize6;
734 	uint32_t	size;
735 	uint32_t	count;
736 };
737 
738 /*
739  * Print global/per-AF table @i algorithm info.
740  */
741 static void
table_show_tainfo(ipfw_xtable_info * i __unused,struct ta_cldata * d,const char * af,const char * taclass)742 table_show_tainfo(ipfw_xtable_info *i __unused, struct ta_cldata *d,
743     const char *af, const char *taclass)
744 {
745 
746 	switch (d->taclass) {
747 	case IPFW_TACLASS_HASH:
748 	case IPFW_TACLASS_ARRAY:
749 		printf(" %salgorithm %s info\n", af, taclass);
750 		if (d->itemsize == d->itemsize6)
751 			printf("  size: %u items: %u itemsize: %u\n",
752 			    d->size, d->count, d->itemsize);
753 		else
754 			printf("  size: %u items: %u "
755 			    "itemsize4: %u itemsize6: %u\n",
756 			    d->size, d->count,
757 			    d->itemsize, d->itemsize6);
758 		break;
759 	case IPFW_TACLASS_RADIX:
760 		printf(" %salgorithm %s info\n", af, taclass);
761 		if (d->itemsize == d->itemsize6)
762 			printf("  items: %u itemsize: %u\n",
763 			    d->count, d->itemsize);
764 		else
765 			printf("  items: %u "
766 			    "itemsize4: %u itemsize6: %u\n",
767 			    d->count, d->itemsize, d->itemsize6);
768 		break;
769 	default:
770 		printf(" algo class: %s\n", taclass);
771 	}
772 }
773 
774 static void
table_print_valheader(char * buf,size_t bufsize,uint32_t vmask)775 table_print_valheader(char *buf, size_t bufsize, uint32_t vmask)
776 {
777 
778 	if (vmask == IPFW_VTYPE_LEGACY) {
779 		snprintf(buf, bufsize, "legacy");
780 		return;
781 	}
782 
783 	memset(buf, 0, bufsize);
784 	print_flags_buffer(buf, bufsize, tablevaltypes, vmask);
785 }
786 
787 /*
788  * Prints table info struct @i in human-readable form.
789  */
790 static int
table_show_info(ipfw_xtable_info * i,void * arg)791 table_show_info(ipfw_xtable_info *i, void *arg)
792 {
793 	const char *vtype;
794 	ipfw_ta_tinfo *tainfo;
795 	int afdata, afitem;
796 	struct ta_cldata d;
797 	char ttype[64], tvtype[64];
798 
799 	table_print_type(ttype, sizeof(ttype), i->type, i->tflags);
800 	table_print_valheader(tvtype, sizeof(tvtype), i->vmask);
801 
802 	printf("--- table(%s), set(%u) ---\n", i->tablename, i->set);
803 	if ((i->flags & IPFW_TGFLAGS_LOCKED) != 0)
804 		printf(" kindex: %d, type: %s, locked\n", i->kidx, ttype);
805 	else
806 		printf(" kindex: %d, type: %s\n", i->kidx, ttype);
807 	printf(" references: %u, valtype: %s\n", i->refcnt, tvtype);
808 	printf(" algorithm: %s\n", i->algoname);
809 	printf(" items: %u, size: %u\n", i->count, i->size);
810 	if (i->limit > 0)
811 		printf(" limit: %u\n", i->limit);
812 
813 	/* Print algo-specific info if requested & set  */
814 	if (arg == NULL)
815 		return (0);
816 
817 	if ((i->ta_info.flags & IPFW_TATFLAGS_DATA) == 0)
818 		return (0);
819 	tainfo = &i->ta_info;
820 
821 	afdata = 0;
822 	afitem = 0;
823 	if (tainfo->flags & IPFW_TATFLAGS_AFDATA)
824 		afdata = 1;
825 	if (tainfo->flags & IPFW_TATFLAGS_AFITEM)
826 		afitem = 1;
827 
828 	memset(&d, 0, sizeof(d));
829 	d.taclass = tainfo->taclass4;
830 	d.size = tainfo->size4;
831 	d.count = tainfo->count4;
832 	d.itemsize = tainfo->itemsize4;
833 	if (afdata == 0 && afitem != 0)
834 		d.itemsize6 = tainfo->itemsize6;
835 	else
836 		d.itemsize6 = d.itemsize;
837 	if ((vtype = match_value(tablealgoclass, d.taclass)) == NULL)
838 		vtype = "unknown";
839 
840 	if (afdata == 0) {
841 		table_show_tainfo(i, &d, "", vtype);
842 	} else {
843 		table_show_tainfo(i, &d, "IPv4 ", vtype);
844 		memset(&d, 0, sizeof(d));
845 		d.taclass = tainfo->taclass6;
846 		if ((vtype = match_value(tablealgoclass, d.taclass)) == NULL)
847 			vtype = "unknown";
848 		d.size = tainfo->size6;
849 		d.count = tainfo->count6;
850 		d.itemsize = tainfo->itemsize6;
851 		d.itemsize6 = d.itemsize;
852 		table_show_tainfo(i, &d, "IPv6 ", vtype);
853 	}
854 
855 	return (0);
856 }
857 
858 
859 /*
860  * Function wrappers which can be used either
861  * as is or as foreach function parameter.
862  */
863 
864 static int
table_show_one(ipfw_xtable_info * i,void * arg)865 table_show_one(ipfw_xtable_info *i, void *arg)
866 {
867 	ipfw_obj_header *oh = NULL;
868 	int error;
869 	int is_all;
870 
871 	is_all = arg == NULL ? 0 : 1;
872 
873 	if ((error = table_do_get_list(i, &oh)) != 0) {
874 		err(EX_OSERR, "Error requesting table %s list", i->tablename);
875 		return (error);
876 	}
877 
878 	table_show_list(oh, is_all);
879 
880 	free(oh);
881 	return (0);
882 }
883 
884 static int
table_flush_one(ipfw_xtable_info * i,void * arg)885 table_flush_one(ipfw_xtable_info *i, void *arg)
886 {
887 	ipfw_obj_header *oh;
888 
889 	oh = (ipfw_obj_header *)arg;
890 
891 	table_fill_ntlv(&oh->ntlv, i->tablename, i->set, 1);
892 
893 	return (table_flush(oh));
894 }
895 
896 static int
table_do_modify_record(int cmd,ipfw_obj_header * oh,ipfw_obj_tentry * tent,int count,int atomic)897 table_do_modify_record(int cmd, ipfw_obj_header *oh,
898     ipfw_obj_tentry *tent, int count, int atomic)
899 {
900 	ipfw_obj_ctlv *ctlv;
901 	ipfw_obj_tentry *tent_base;
902 	caddr_t pbuf;
903 	char xbuf[sizeof(*oh) + sizeof(ipfw_obj_ctlv) + sizeof(*tent)];
904 	int error, i;
905 	size_t sz;
906 
907 	sz = sizeof(*ctlv) + sizeof(*tent) * count;
908 	if (count == 1) {
909 		memset(xbuf, 0, sizeof(xbuf));
910 		pbuf = xbuf;
911 	} else {
912 		if ((pbuf = calloc(1, sizeof(*oh) + sz)) == NULL)
913 			return (ENOMEM);
914 	}
915 
916 	memcpy(pbuf, oh, sizeof(*oh));
917 	oh = (ipfw_obj_header *)pbuf;
918 	oh->opheader.version = 1; /* Current version */
919 
920 	ctlv = (ipfw_obj_ctlv *)(oh + 1);
921 	ctlv->count = count;
922 	ctlv->head.length = sz;
923 	if (atomic != 0)
924 		ctlv->flags |= IPFW_CTF_ATOMIC;
925 
926 	tent_base = tent;
927 	memcpy(ctlv + 1, tent, sizeof(*tent) * count);
928 	tent = (ipfw_obj_tentry *)(ctlv + 1);
929 	for (i = 0; i < count; i++, tent++) {
930 		tent->head.length = sizeof(ipfw_obj_tentry);
931 		tent->idx = oh->idx;
932 	}
933 
934 	sz += sizeof(*oh);
935 	error = do_get3(cmd, &oh->opheader, &sz);
936 	if (error != 0)
937 		error = errno;
938 	tent = (ipfw_obj_tentry *)(ctlv + 1);
939 	/* Copy result back to provided buffer */
940 	memcpy(tent_base, ctlv + 1, sizeof(*tent) * count);
941 
942 	if (pbuf != xbuf)
943 		free(pbuf);
944 
945 	return (error);
946 }
947 
948 static void
table_modify_record(ipfw_obj_header * oh,int ac,char * av[],int add,int quiet,int update,int atomic)949 table_modify_record(ipfw_obj_header *oh, int ac, char *av[], int add,
950     int quiet, int update, int atomic)
951 {
952 	ipfw_obj_tentry *ptent, tent, *tent_buf;
953 	ipfw_xtable_info xi;
954 	const char *etxt, *px, *texterr;
955 	uint8_t type;
956 	uint32_t vmask;
957 	int cmd, count, error, i, ignored;
958 
959 	if (ac == 0)
960 		errx(EX_USAGE, "address required");
961 
962 	if (add != 0) {
963 		cmd = IP_FW_TABLE_XADD;
964 		texterr = "Adding record failed";
965 	} else {
966 		cmd = IP_FW_TABLE_XDEL;
967 		texterr = "Deleting record failed";
968 	}
969 
970 	/*
971 	 * Calculate number of entries:
972 	 * Assume [key val] x N for add
973 	 * and
974 	 * key x N for delete
975 	 */
976 	count = (add != 0) ? ac / 2 + 1 : ac;
977 
978 	if (count <= 1) {
979 		/* Adding single entry with/without value */
980 		memset(&tent, 0, sizeof(tent));
981 		tent_buf = &tent;
982 	} else {
983 
984 		if ((tent_buf = calloc(count, sizeof(tent))) == NULL)
985 			errx(EX_OSERR,
986 			    "Unable to allocate memory for all entries");
987 	}
988 	ptent = tent_buf;
989 
990 	memset(&xi, 0, sizeof(xi));
991 	count = 0;
992 	while (ac > 0) {
993 		tentry_fill_key(oh, ptent, *av, add, &type, &vmask, &xi);
994 
995 		/*
996 		 * Compatibility layer: auto-create table if not exists.
997 		 */
998 		if (xi.tablename[0] == '\0') {
999 			xi.type = type;
1000 			xi.vmask = vmask;
1001 			strlcpy(xi.tablename, oh->ntlv.name,
1002 			    sizeof(xi.tablename));
1003 			if (quiet == 0)
1004 				warnx("DEPRECATED: inserting data into "
1005 				    "non-existent table %s. (auto-created)",
1006 				    xi.tablename);
1007 			table_do_create(oh, &xi);
1008 		}
1009 
1010 		oh->ntlv.type = type;
1011 		ac--; av++;
1012 
1013 		if (add != 0 && ac > 0) {
1014 			tentry_fill_value(oh, ptent, *av, type, vmask);
1015 			ac--; av++;
1016 		}
1017 
1018 		if (update != 0)
1019 			ptent->head.flags |= IPFW_TF_UPDATE;
1020 
1021 		count++;
1022 		ptent++;
1023 	}
1024 
1025 	error = table_do_modify_record(cmd, oh, tent_buf, count, atomic);
1026 
1027 	/*
1028 	 * Compatibility stuff: do not yell on duplicate keys or
1029 	 * failed deletions.
1030 	 */
1031 	if (error == 0 || (error == EEXIST && add != 0) ||
1032 	    (error == ENOENT && add == 0)) {
1033 		if (quiet != 0) {
1034 			if (tent_buf != &tent)
1035 				free(tent_buf);
1036 			return;
1037 		}
1038 	}
1039 
1040 	/* Report results back */
1041 	ptent = tent_buf;
1042 	for (i = 0; i < count; ptent++, i++) {
1043 		ignored = 0;
1044 		switch (ptent->result) {
1045 		case IPFW_TR_ADDED:
1046 			px = "added";
1047 			break;
1048 		case IPFW_TR_DELETED:
1049 			px = "deleted";
1050 			break;
1051 		case IPFW_TR_UPDATED:
1052 			px = "updated";
1053 			break;
1054 		case IPFW_TR_LIMIT:
1055 			px = "limit";
1056 			ignored = 1;
1057 			break;
1058 		case IPFW_TR_ERROR:
1059 			px = "error";
1060 			ignored = 1;
1061 			break;
1062 		case IPFW_TR_NOTFOUND:
1063 			px = "notfound";
1064 			ignored = 1;
1065 			break;
1066 		case IPFW_TR_EXISTS:
1067 			px = "exists";
1068 			ignored = 1;
1069 			break;
1070 		case IPFW_TR_IGNORED:
1071 			px = "ignored";
1072 			ignored = 1;
1073 			break;
1074 		default:
1075 			px = "unknown";
1076 			ignored = 1;
1077 		}
1078 
1079 		if (error != 0 && atomic != 0 && ignored == 0)
1080 			printf("%s(reverted): ", px);
1081 		else
1082 			printf("%s: ", px);
1083 
1084 		table_show_entry(&xi, ptent);
1085 	}
1086 
1087 	if (tent_buf != &tent)
1088 		free(tent_buf);
1089 
1090 	if (error == 0)
1091 		return;
1092 
1093 	/* Try to provide more human-readable error */
1094 	switch (error) {
1095 	case EEXIST:
1096 		etxt = "record already exists";
1097 		break;
1098 	case EFBIG:
1099 		etxt = "limit hit";
1100 		break;
1101 	case ESRCH:
1102 		etxt = "table not found";
1103 		break;
1104 	case ENOENT:
1105 		etxt = "record not found";
1106 		break;
1107 	case EACCES:
1108 		etxt = "table is locked";
1109 		break;
1110 	default:
1111 		etxt = strerror(error);
1112 	}
1113 
1114 	errx(EX_OSERR, "%s: %s", texterr, etxt);
1115 }
1116 
1117 static int
table_do_lookup(ipfw_obj_header * oh,char * key,ipfw_xtable_info * xi,ipfw_obj_tentry * xtent)1118 table_do_lookup(ipfw_obj_header *oh, char *key, ipfw_xtable_info *xi,
1119     ipfw_obj_tentry *xtent)
1120 {
1121 	char xbuf[sizeof(ipfw_obj_header) + sizeof(ipfw_obj_tentry)];
1122 	ipfw_obj_tentry *tent;
1123 	uint8_t type;
1124 	uint32_t vmask;
1125 	size_t sz;
1126 
1127 	memcpy(xbuf, oh, sizeof(*oh));
1128 	oh = (ipfw_obj_header *)xbuf;
1129 	tent = (ipfw_obj_tentry *)(oh + 1);
1130 
1131 	memset(tent, 0, sizeof(*tent));
1132 	tent->head.length = sizeof(*tent);
1133 	tent->idx = 1;
1134 
1135 	tentry_fill_key(oh, tent, key, 0, &type, &vmask, xi);
1136 	oh->ntlv.type = type;
1137 
1138 	sz = sizeof(xbuf);
1139 	if (do_get3(IP_FW_TABLE_XFIND, &oh->opheader, &sz) != 0)
1140 		return (errno);
1141 
1142 	if (sz < sizeof(xbuf))
1143 		return (EINVAL);
1144 
1145 	*xtent = *tent;
1146 
1147 	return (0);
1148 }
1149 
1150 static void
table_lookup(ipfw_obj_header * oh,int ac,char * av[])1151 table_lookup(ipfw_obj_header *oh, int ac, char *av[])
1152 {
1153 	ipfw_obj_tentry xtent;
1154 	ipfw_xtable_info xi;
1155 	char key[64];
1156 	int error;
1157 
1158 	if (ac == 0)
1159 		errx(EX_USAGE, "address required");
1160 
1161 	strlcpy(key, *av, sizeof(key));
1162 
1163 	memset(&xi, 0, sizeof(xi));
1164 	error = table_do_lookup(oh, key, &xi, &xtent);
1165 
1166 	switch (error) {
1167 	case 0:
1168 		break;
1169 	case ESRCH:
1170 		errx(EX_UNAVAILABLE, "Table %s not found", oh->ntlv.name);
1171 	case ENOENT:
1172 		errx(EX_UNAVAILABLE, "Entry %s not found", *av);
1173 	case ENOTSUP:
1174 		errx(EX_UNAVAILABLE, "Table %s algo does not support "
1175 		    "\"lookup\" method", oh->ntlv.name);
1176 	default:
1177 		err(EX_OSERR, "getsockopt(IP_FW_TABLE_XFIND)");
1178 	}
1179 
1180 	table_show_entry(&xi, &xtent);
1181 }
1182 
1183 static void
tentry_fill_key_type(char * arg,ipfw_obj_tentry * tentry,uint8_t type,uint8_t tflags)1184 tentry_fill_key_type(char *arg, ipfw_obj_tentry *tentry, uint8_t type,
1185     uint8_t tflags)
1186 {
1187 	char *p, *pp;
1188 	int mask, af;
1189 	struct in6_addr *paddr, tmp;
1190 	struct ether_addr *mac;
1191 	struct tflow_entry *tfe;
1192 	uint32_t key, *pkey;
1193 	uint16_t port;
1194 	struct protoent *pent;
1195 	struct servent *sent;
1196 	int masklen;
1197 
1198 	mask = masklen = 0;
1199 	af = 0;
1200 	paddr = (struct in6_addr *)&tentry->k;
1201 
1202 	switch (type) {
1203 	case IPFW_TABLE_ADDR:
1204 		/* Remove / if exists */
1205 		if ((p = strchr(arg, '/')) != NULL) {
1206 			*p = '\0';
1207 			mask = atoi(p + 1);
1208 		}
1209 
1210 		if (inet_pton(AF_INET, arg, paddr) == 1) {
1211 			if (p != NULL && mask > 32)
1212 				errx(EX_DATAERR, "bad IPv4 mask width: %s",
1213 				    p + 1);
1214 
1215 			masklen = p ? mask : 32;
1216 			af = AF_INET;
1217 		} else if (inet_pton(AF_INET6, arg, paddr) == 1) {
1218 			if (IN6_IS_ADDR_V4COMPAT(paddr))
1219 				errx(EX_DATAERR,
1220 				    "Use IPv4 instead of v4-compatible");
1221 			if (p != NULL && mask > 128)
1222 				errx(EX_DATAERR, "bad IPv6 mask width: %s",
1223 				    p + 1);
1224 
1225 			masklen = p ? mask : 128;
1226 			af = AF_INET6;
1227 		} else {
1228 			/* Assume FQDN */
1229 			if (lookup_host(arg, (struct in_addr *)paddr) != 0)
1230 				errx(EX_NOHOST, "hostname ``%s'' unknown", arg);
1231 
1232 			masklen = 32;
1233 			type = IPFW_TABLE_ADDR;
1234 			af = AF_INET;
1235 		}
1236 		break;
1237 	case IPFW_TABLE_MAC:
1238 		/* Remove / if exists */
1239 		if ((p = strchr(arg, '/')) != NULL) {
1240 			*p = '\0';
1241 			mask = atoi(p + 1);
1242 		}
1243 
1244 		if (p != NULL && mask > 8 * ETHER_ADDR_LEN)
1245 			errx(EX_DATAERR, "bad MAC mask width: %s",
1246 			    p + 1);
1247 
1248 		if ((mac = ether_aton(arg)) == NULL)
1249 			errx(EX_DATAERR, "Incorrect MAC address");
1250 
1251 		memcpy(tentry->k.mac, mac->octet, ETHER_ADDR_LEN);
1252 		masklen = p ? mask : 8 * ETHER_ADDR_LEN;
1253 		af = AF_LINK;
1254 		break;
1255 	case IPFW_TABLE_INTERFACE:
1256 		/* Assume interface name. Copy significant data only */
1257 		mask = MIN(strlen(arg), IF_NAMESIZE - 1);
1258 		memcpy(paddr, arg, mask);
1259 		/* Set mask to exact match */
1260 		masklen = 8 * IF_NAMESIZE;
1261 		break;
1262 	case IPFW_TABLE_NUMBER:
1263 		/* Port or any other key */
1264 		key = strtol(arg, &p, 10);
1265 		if (*p != '\0')
1266 			errx(EX_DATAERR, "Invalid number: %s", arg);
1267 
1268 		pkey = (uint32_t *)paddr;
1269 		*pkey = key;
1270 		masklen = 32;
1271 		break;
1272 	case IPFW_TABLE_FLOW:
1273 		/* Assume [src-ip][,proto][,src-port][,dst-ip][,dst-port] */
1274 		tfe = &tentry->k.flow;
1275 		af = 0;
1276 
1277 		/* Handle <ipv4|ipv6> */
1278 		if ((tflags & IPFW_TFFLAG_SRCIP) != 0) {
1279 			if ((p = strchr(arg, ',')) != NULL)
1280 				*p++ = '\0';
1281 			/* Determine family using temporary storage */
1282 			if (inet_pton(AF_INET, arg, &tmp) == 1) {
1283 				if (af != 0 && af != AF_INET)
1284 					errx(EX_DATAERR,
1285 					    "Inconsistent address family\n");
1286 				af = AF_INET;
1287 				memcpy(&tfe->a.a4.sip, &tmp, 4);
1288 			} else if (inet_pton(AF_INET6, arg, &tmp) == 1) {
1289 				if (af != 0 && af != AF_INET6)
1290 					errx(EX_DATAERR,
1291 					    "Inconsistent address family\n");
1292 				af = AF_INET6;
1293 				memcpy(&tfe->a.a6.sip6, &tmp, 16);
1294 			}
1295 
1296 			arg = p;
1297 		}
1298 
1299 		/* Handle <proto-num|proto-name> */
1300 		if ((tflags & IPFW_TFFLAG_PROTO) != 0) {
1301 			if (arg == NULL)
1302 				errx(EX_DATAERR, "invalid key: proto missing");
1303 			if ((p = strchr(arg, ',')) != NULL)
1304 				*p++ = '\0';
1305 
1306 			key = strtol(arg, &pp, 10);
1307 			if (*pp != '\0') {
1308 				if ((pent = getprotobyname(arg)) == NULL)
1309 					errx(EX_DATAERR, "Unknown proto: %s",
1310 					    arg);
1311 				else
1312 					key = pent->p_proto;
1313 			}
1314 
1315 			if (key > 255)
1316 				errx(EX_DATAERR, "Bad protocol number: %u",key);
1317 
1318 			tfe->proto = key;
1319 
1320 			arg = p;
1321 		}
1322 
1323 		/* Handle <port-num|service-name> */
1324 		if ((tflags & IPFW_TFFLAG_SRCPORT) != 0) {
1325 			if (arg == NULL)
1326 				errx(EX_DATAERR, "invalid key: src port missing");
1327 			if ((p = strchr(arg, ',')) != NULL)
1328 				*p++ = '\0';
1329 
1330 			port = htons(strtol(arg, &pp, 10));
1331 			if (*pp != '\0') {
1332 				if ((sent = getservbyname(arg, NULL)) == NULL)
1333 					errx(EX_DATAERR, "Unknown service: %s",
1334 					    arg);
1335 				port = sent->s_port;
1336 			}
1337 			tfe->sport = port;
1338 			arg = p;
1339 		}
1340 
1341 		/* Handle <ipv4|ipv6>*/
1342 		if ((tflags & IPFW_TFFLAG_DSTIP) != 0) {
1343 			if (arg == NULL)
1344 				errx(EX_DATAERR, "invalid key: dst ip missing");
1345 			if ((p = strchr(arg, ',')) != NULL)
1346 				*p++ = '\0';
1347 			/* Determine family using temporary storage */
1348 			if (inet_pton(AF_INET, arg, &tmp) == 1) {
1349 				if (af != 0 && af != AF_INET)
1350 					errx(EX_DATAERR,
1351 					    "Inconsistent address family");
1352 				af = AF_INET;
1353 				memcpy(&tfe->a.a4.dip, &tmp, 4);
1354 			} else if (inet_pton(AF_INET6, arg, &tmp) == 1) {
1355 				if (af != 0 && af != AF_INET6)
1356 					errx(EX_DATAERR,
1357 					    "Inconsistent address family");
1358 				af = AF_INET6;
1359 				memcpy(&tfe->a.a6.dip6, &tmp, 16);
1360 			}
1361 
1362 			arg = p;
1363 		}
1364 
1365 		/* Handle <port-num|service-name> */
1366 		if ((tflags & IPFW_TFFLAG_DSTPORT) != 0) {
1367 			if (arg == NULL)
1368 				errx(EX_DATAERR, "invalid key: dst port missing");
1369 			if ((p = strchr(arg, ',')) != NULL)
1370 				*p++ = '\0';
1371 
1372 			port = htons(strtol(arg, &pp, 10));
1373 			if (*pp != '\0') {
1374 				if ((sent = getservbyname(arg, NULL)) == NULL)
1375 					errx(EX_DATAERR, "Unknown service: %s",
1376 					    arg);
1377 				port = sent->s_port;
1378 			}
1379 			tfe->dport = port;
1380 			arg = p;
1381 		}
1382 
1383 		tfe->af = af;
1384 
1385 		break;
1386 
1387 	default:
1388 		errx(EX_DATAERR, "Unsupported table type: %d", type);
1389 	}
1390 
1391 	tentry->subtype = af;
1392 	tentry->masklen = masklen;
1393 }
1394 
1395 /*
1396  * Tries to guess table key type.
1397  * This procedure is used in legacy table auto-create
1398  * code AND in `ipfw -n` ruleset checking.
1399  *
1400  * Imported from old table_fill_xentry() parse code.
1401  */
1402 static int
guess_key_type(char * key,uint8_t * ptype)1403 guess_key_type(char *key, uint8_t *ptype)
1404 {
1405 	char *p;
1406 	struct in6_addr addr;
1407 
1408 	if (ishexnumber(*key) != 0 || *key == ':') {
1409 		/* Remove / if exists */
1410 		if ((p = strchr(key, '/')) != NULL)
1411 			*p = '\0';
1412 
1413 		if ((inet_pton(AF_INET, key, &addr) == 1) ||
1414 		    (inet_pton(AF_INET6, key, &addr) == 1)) {
1415 			*ptype = IPFW_TABLE_CIDR;
1416 			if (p != NULL)
1417 				*p = '/';
1418 			return (0);
1419 		} else {
1420 			/* Port or any other key */
1421 			/* Skip non-base 10 entries like 'fa1' */
1422 			(void)strtol(key, &p, 10);
1423 			if (*p == '\0') {
1424 				*ptype = IPFW_TABLE_NUMBER;
1425 				return (0);
1426 			} else if ((p != key) && (*p == '.')) {
1427 				/*
1428 				 * Warn on IPv4 address strings
1429 				 * which are "valid" for inet_aton() but not
1430 				 * in inet_pton().
1431 				 *
1432 				 * Typical examples: '10.5' or '10.0.0.05'
1433 				 */
1434 				return (1);
1435 			}
1436 		}
1437 	}
1438 
1439 	if (strchr(key, '.') == NULL) {
1440 		*ptype = IPFW_TABLE_INTERFACE;
1441 		return (0);
1442 	}
1443 
1444 	if (lookup_host(key, (struct in_addr *)&addr) != 0)
1445 		return (1);
1446 
1447 	*ptype = IPFW_TABLE_CIDR;
1448 	return (0);
1449 }
1450 
1451 static void
tentry_fill_key(ipfw_obj_header * oh,ipfw_obj_tentry * tent,char * key,int add,uint8_t * ptype,uint32_t * pvmask,ipfw_xtable_info * xi)1452 tentry_fill_key(ipfw_obj_header *oh, ipfw_obj_tentry *tent, char *key,
1453     int add, uint8_t *ptype, uint32_t *pvmask, ipfw_xtable_info *xi)
1454 {
1455 	uint8_t type, tflags;
1456 	uint32_t vmask;
1457 	int error;
1458 
1459 	type = 0;
1460 	tflags = 0;
1461 	vmask = 0;
1462 
1463 	if (xi->tablename[0] == '\0')
1464 		error = table_get_info(oh, xi);
1465 	else
1466 		error = 0;
1467 
1468 	if (error == 0) {
1469 		if (g_co.test_only == 0) {
1470 			/* Table found */
1471 			type = xi->type;
1472 			tflags = xi->tflags;
1473 			vmask = xi->vmask;
1474 		} else {
1475 			/*
1476 			 * We're running `ipfw -n`
1477 			 * Compatibility layer: try to guess key type
1478 			 * before failing.
1479 			 */
1480 			if (guess_key_type(key, &type) != 0) {
1481 				/* Inknown key */
1482 				errx(EX_USAGE, "Cannot guess "
1483 				    "key '%s' type", key);
1484 			}
1485 			vmask = IPFW_VTYPE_LEGACY;
1486 		}
1487 	} else {
1488 		if (error != ESRCH)
1489 			errx(EX_OSERR, "Error requesting table %s info",
1490 			    oh->ntlv.name);
1491 		if (add == 0)
1492 			errx(EX_DATAERR, "Table %s does not exist",
1493 			    oh->ntlv.name);
1494 		/*
1495 		 * Table does not exist
1496 		 * Compatibility layer: try to guess key type before failing.
1497 		 */
1498 		if (guess_key_type(key, &type) != 0) {
1499 			/* Inknown key */
1500 			errx(EX_USAGE, "Table %s does not exist, cannot guess "
1501 			    "key '%s' type", oh->ntlv.name, key);
1502 		}
1503 
1504 		vmask = IPFW_VTYPE_LEGACY;
1505 	}
1506 
1507 	tentry_fill_key_type(key, tent, type, tflags);
1508 
1509 	*ptype = type;
1510 	*pvmask = vmask;
1511 }
1512 
1513 static void
set_legacy_value(uint32_t val,ipfw_table_value * v)1514 set_legacy_value(uint32_t val, ipfw_table_value *v)
1515 {
1516 	v->tag = val;
1517 	v->pipe = val;
1518 	v->divert = val;
1519 	v->skipto = val;
1520 	v->netgraph = val;
1521 	v->fib = val;
1522 	v->nat = val;
1523 	v->nh4 = val;
1524 	v->dscp = (uint8_t)val;
1525 	v->limit = val;
1526 }
1527 
1528 static void
tentry_fill_value(ipfw_obj_header * oh __unused,ipfw_obj_tentry * tent,char * arg,uint8_t type __unused,uint32_t vmask)1529 tentry_fill_value(ipfw_obj_header *oh __unused, ipfw_obj_tentry *tent,
1530     char *arg, uint8_t type __unused, uint32_t vmask)
1531 {
1532 	struct addrinfo hints, *res;
1533 	struct in_addr ipaddr;
1534 	const char *etype;
1535 	char *comma, *e, *n, *p;
1536 	uint32_t a4, flag, val;
1537 	ipfw_table_value *v;
1538 	uint32_t i;
1539 	int dval;
1540 
1541 	v = &tent->v.value;
1542 
1543 	/* Compat layer: keep old behavior for legacy value types */
1544 	if (vmask == IPFW_VTYPE_LEGACY) {
1545 		/* Try to interpret as number first */
1546 		val = strtoul(arg, &p, 0);
1547 		if (*p == '\0') {
1548 			set_legacy_value(val, v);
1549 			return;
1550 		}
1551 		if (inet_pton(AF_INET, arg, &val) == 1) {
1552 			set_legacy_value(ntohl(val), v);
1553 			return;
1554 		}
1555 		/* Try hostname */
1556 		if (lookup_host(arg, &ipaddr) == 0) {
1557 			set_legacy_value(ntohl(ipaddr.s_addr), v);
1558 			return;
1559 		}
1560 		errx(EX_OSERR, "Unable to parse value %s", arg);
1561 	}
1562 
1563 	/*
1564 	 * Shorthands: handle single value if vmask consists
1565 	 * of numbers only. e.g.:
1566 	 * vmask = "fib,skipto" -> treat input "1" as "1,1"
1567 	 */
1568 
1569 	n = arg;
1570 	etype = NULL;
1571 	for (i = 1; i < (1u << 31); i *= 2) {
1572 		if ((flag = (vmask & i)) == 0)
1573 			continue;
1574 		vmask &= ~flag;
1575 
1576 		if ((comma = strchr(n, ',')) != NULL)
1577 			*comma = '\0';
1578 
1579 		switch (flag) {
1580 		case IPFW_VTYPE_TAG:
1581 			v->tag = strtol(n, &e, 10);
1582 			if (*e != '\0')
1583 				etype = "tag";
1584 			break;
1585 		case IPFW_VTYPE_PIPE:
1586 			v->pipe = strtol(n, &e, 10);
1587 			if (*e != '\0')
1588 				etype = "pipe";
1589 			break;
1590 		case IPFW_VTYPE_DIVERT:
1591 			v->divert = strtol(n, &e, 10);
1592 			if (*e != '\0')
1593 				etype = "divert";
1594 			break;
1595 		case IPFW_VTYPE_SKIPTO:
1596 			v->skipto = strtol(n, &e, 10);
1597 			if (*e != '\0')
1598 				etype = "skipto";
1599 			break;
1600 		case IPFW_VTYPE_NETGRAPH:
1601 			v->netgraph = strtol(n, &e, 10);
1602 			if (*e != '\0')
1603 				etype = "netgraph";
1604 			break;
1605 		case IPFW_VTYPE_FIB:
1606 			v->fib = strtol(n, &e, 10);
1607 			if (*e != '\0')
1608 				etype = "fib";
1609 			break;
1610 		case IPFW_VTYPE_NAT:
1611 			v->nat = strtol(n, &e, 10);
1612 			if (*e != '\0')
1613 				etype = "nat";
1614 			break;
1615 		case IPFW_VTYPE_LIMIT:
1616 			v->limit = strtol(n, &e, 10);
1617 			if (*e != '\0')
1618 				etype = "limit";
1619 			break;
1620 		case IPFW_VTYPE_NH4:
1621 			if (strchr(n, '.') != NULL &&
1622 			    inet_pton(AF_INET, n, &a4) == 1) {
1623 				v->nh4 = ntohl(a4);
1624 				break;
1625 			}
1626 			if (lookup_host(n, &ipaddr) == 0) {
1627 				v->nh4 = ntohl(ipaddr.s_addr);
1628 				break;
1629 			}
1630 			etype = "ipv4";
1631 			break;
1632 		case IPFW_VTYPE_DSCP:
1633 			if (isalpha(*n)) {
1634 				if ((dval = match_token(f_ipdscp, n)) != -1) {
1635 					v->dscp = dval;
1636 					break;
1637 				} else
1638 					etype = "DSCP code";
1639 			} else {
1640 				v->dscp = strtol(n, &e, 10);
1641 				if (v->dscp > 63 || *e != '\0')
1642 					etype = "DSCP value";
1643 			}
1644 			break;
1645 		case IPFW_VTYPE_NH6:
1646 			if (strchr(n, ':') != NULL) {
1647 				memset(&hints, 0, sizeof(hints));
1648 				hints.ai_family = AF_INET6;
1649 				hints.ai_flags = AI_NUMERICHOST;
1650 				if (getaddrinfo(n, NULL, &hints, &res) == 0) {
1651 					v->nh6 = ((struct sockaddr_in6 *)
1652 					    res->ai_addr)->sin6_addr;
1653 					v->zoneid = ((struct sockaddr_in6 *)
1654 					    res->ai_addr)->sin6_scope_id;
1655 					freeaddrinfo(res);
1656 					break;
1657 				}
1658 			}
1659 			etype = "ipv6";
1660 			break;
1661 		case IPFW_VTYPE_MARK:
1662 			v->mark = strtol(n, &e, 16);
1663 			if (*e != '\0')
1664 				etype = "mark";
1665 			break;
1666 		}
1667 
1668 		if (etype != NULL)
1669 			errx(EX_USAGE, "Unable to parse %s as %s", n, etype);
1670 
1671 		if (comma != NULL)
1672 			*comma++ = ',';
1673 
1674 		if ((n = comma) != NULL)
1675 			continue;
1676 
1677 		/* End of input. */
1678 		if (vmask != 0)
1679 			errx(EX_USAGE, "Not enough fields inside value");
1680 	}
1681 }
1682 
1683 /*
1684  * Compare table names.
1685  * Honor number comparison.
1686  */
1687 static int
tablename_cmp(const void * a,const void * b)1688 tablename_cmp(const void *a, const void *b)
1689 {
1690 	const ipfw_xtable_info *ia, *ib;
1691 
1692 	ia = (const ipfw_xtable_info *)a;
1693 	ib = (const ipfw_xtable_info *)b;
1694 
1695 	return (stringnum_cmp(ia->tablename, ib->tablename));
1696 }
1697 
1698 /*
1699  * Retrieves table list from kernel,
1700  * optionally sorts it and calls requested function for each table.
1701  * Returns 0 on success.
1702  */
1703 static int
tables_foreach(table_cb_t * f,void * arg,int sort)1704 tables_foreach(table_cb_t *f, void *arg, int sort)
1705 {
1706 	ipfw_obj_lheader *olh;
1707 	ipfw_xtable_info *info;
1708 	size_t sz;
1709 	uint32_t i;
1710 
1711 	/* Start with reasonable default */
1712 	sz = sizeof(*olh) + 16 * sizeof(ipfw_xtable_info);
1713 
1714 	for (;;) {
1715 		if ((olh = calloc(1, sz)) == NULL)
1716 			return (ENOMEM);
1717 
1718 		olh->size = sz;
1719 		if (do_get3(IP_FW_TABLES_XLIST, &olh->opheader, &sz) != 0) {
1720 			sz = olh->size;
1721 			free(olh);
1722 			if (errno != ENOMEM)
1723 				return (errno);
1724 			continue;
1725 		}
1726 
1727 		if (sort != 0)
1728 			qsort(olh + 1, olh->count, olh->objsize,
1729 			    tablename_cmp);
1730 
1731 		info = (ipfw_xtable_info *)(olh + 1);
1732 		for (i = 0; i < olh->count; i++) {
1733 			if (g_co.use_set == 0 || info->set == g_co.use_set - 1)
1734 				(void)f(info, arg);
1735 			info = (ipfw_xtable_info *)((caddr_t)info +
1736 			    olh->objsize);
1737 		}
1738 		free(olh);
1739 		break;
1740 	}
1741 	return (0);
1742 }
1743 
1744 
1745 /*
1746  * Retrieves all entries for given table @i in
1747  * eXtended format. Allocate buffer large enough
1748  * to store result. Called needs to free it later.
1749  *
1750  * Returns 0 on success.
1751  */
1752 static int
table_do_get_list(ipfw_xtable_info * i,ipfw_obj_header ** poh)1753 table_do_get_list(ipfw_xtable_info *i, ipfw_obj_header **poh)
1754 {
1755 	ipfw_obj_header *oh;
1756 	size_t sz;
1757 	int c;
1758 
1759 	sz = 0;
1760 	oh = NULL;
1761 	for (c = 0; c < 8; c++) {
1762 		if (sz < i->size)
1763 			sz = i->size + 44;
1764 		if (oh != NULL)
1765 			free(oh);
1766 		if ((oh = calloc(1, sz)) == NULL)
1767 			continue;
1768 		table_fill_objheader(oh, i);
1769 		oh->opheader.version = 1; /* Current version */
1770 		if (do_get3(IP_FW_TABLE_XLIST, &oh->opheader, &sz) == 0) {
1771 			*poh = oh;
1772 			return (0);
1773 		}
1774 
1775 		if (errno != ENOMEM)
1776 			break;
1777 	}
1778 	free(oh);
1779 
1780 	return (errno);
1781 }
1782 
1783 /*
1784  * Shows all entries from @oh in human-readable format
1785  */
1786 static void
table_show_list(ipfw_obj_header * oh,int need_header)1787 table_show_list(ipfw_obj_header *oh, int need_header)
1788 {
1789 	ipfw_obj_tentry *tent;
1790 	uint32_t count;
1791 	ipfw_xtable_info *i;
1792 
1793 	i = (ipfw_xtable_info *)(oh + 1);
1794 	tent = (ipfw_obj_tentry *)(i + 1);
1795 
1796 	if (need_header)
1797 		printf("--- table(%s), set(%u) ---\n", i->tablename, i->set);
1798 
1799 	count = i->count;
1800 	while (count > 0) {
1801 		table_show_entry(i, tent);
1802 		tent = (ipfw_obj_tentry *)((caddr_t)tent + tent->head.length);
1803 		count--;
1804 	}
1805 }
1806 
1807 static void
table_show_value(char * buf,size_t bufsize,ipfw_table_value * v,uint32_t vmask,int print_ip)1808 table_show_value(char *buf, size_t bufsize, ipfw_table_value *v,
1809     uint32_t vmask, int print_ip)
1810 {
1811 	char abuf[INET6_ADDRSTRLEN + IF_NAMESIZE + 2];
1812 	struct sockaddr_in6 sa6;
1813 	uint32_t flag, i, l;
1814 	size_t sz;
1815 	struct in_addr a4;
1816 
1817 	sz = bufsize;
1818 
1819 	/*
1820 	 * Some shorthands for printing values:
1821 	 * legacy assumes all values are equal, so keep the first one.
1822 	 */
1823 	if (vmask == IPFW_VTYPE_LEGACY) {
1824 		if (print_ip != 0) {
1825 			flag = htonl(v->tag);
1826 			inet_ntop(AF_INET, &flag, buf, sz);
1827 		} else
1828 			snprintf(buf, sz, "%u", v->tag);
1829 		return;
1830 	}
1831 
1832 	for (i = 1; i < (1u << 31); i *= 2) {
1833 		if ((flag = (vmask & i)) == 0)
1834 			continue;
1835 		l = 0;
1836 
1837 		switch (flag) {
1838 		case IPFW_VTYPE_TAG:
1839 			l = snprintf(buf, sz, "%u,", v->tag);
1840 			break;
1841 		case IPFW_VTYPE_PIPE:
1842 			l = snprintf(buf, sz, "%u,", v->pipe);
1843 			break;
1844 		case IPFW_VTYPE_DIVERT:
1845 			l = snprintf(buf, sz, "%d,", v->divert);
1846 			break;
1847 		case IPFW_VTYPE_SKIPTO:
1848 			l = snprintf(buf, sz, "%d,", v->skipto);
1849 			break;
1850 		case IPFW_VTYPE_NETGRAPH:
1851 			l = snprintf(buf, sz, "%u,", v->netgraph);
1852 			break;
1853 		case IPFW_VTYPE_FIB:
1854 			l = snprintf(buf, sz, "%u,", v->fib);
1855 			break;
1856 		case IPFW_VTYPE_NAT:
1857 			l = snprintf(buf, sz, "%u,", v->nat);
1858 			break;
1859 		case IPFW_VTYPE_LIMIT:
1860 			l = snprintf(buf, sz, "%u,", v->limit);
1861 			break;
1862 		case IPFW_VTYPE_NH4:
1863 			a4.s_addr = htonl(v->nh4);
1864 			inet_ntop(AF_INET, &a4, abuf, sizeof(abuf));
1865 			l = snprintf(buf, sz, "%s,", abuf);
1866 			break;
1867 		case IPFW_VTYPE_DSCP:
1868 			l = snprintf(buf, sz, "%d,", v->dscp);
1869 			break;
1870 		case IPFW_VTYPE_NH6:
1871 			sa6.sin6_family = AF_INET6;
1872 			sa6.sin6_len = sizeof(sa6);
1873 			sa6.sin6_addr = v->nh6;
1874 			sa6.sin6_port = 0;
1875 			sa6.sin6_scope_id = v->zoneid;
1876 			if (getnameinfo((const struct sockaddr *)&sa6,
1877 			    sa6.sin6_len, abuf, sizeof(abuf), NULL, 0,
1878 			    NI_NUMERICHOST) == 0)
1879 				l = snprintf(buf, sz, "%s,", abuf);
1880 			break;
1881 		case IPFW_VTYPE_MARK:
1882 			l = snprintf(buf, sz, "%#x,", v->mark);
1883 			break;
1884 		}
1885 
1886 		buf += l;
1887 		sz -= l;
1888 	}
1889 
1890 	if (sz != bufsize)
1891 		*(buf - 1) = '\0';
1892 }
1893 
1894 static void
table_show_entry(ipfw_xtable_info * i,ipfw_obj_tentry * tent)1895 table_show_entry(ipfw_xtable_info *i, ipfw_obj_tentry *tent)
1896 {
1897 	char tbuf[128], pval[128];
1898 	const char *comma;
1899 	const u_char *mac;
1900 	void *paddr;
1901 	struct tflow_entry *tfe;
1902 
1903 	table_show_value(pval, sizeof(pval), &tent->v.value, i->vmask,
1904 	    g_co.do_value_as_ip);
1905 
1906 	switch (i->type) {
1907 	case IPFW_TABLE_ADDR:
1908 		/* IPv4 or IPv6 prefixes */
1909 		inet_ntop(tent->subtype, &tent->k, tbuf, sizeof(tbuf));
1910 		printf("%s/%u %s\n", tbuf, tent->masklen, pval);
1911 		break;
1912 	case IPFW_TABLE_MAC:
1913 		/* MAC prefixes */
1914 		mac = tent->k.mac;
1915 		printf("%02x:%02x:%02x:%02x:%02x:%02x/%u %s\n",
1916 		    mac[0], mac[1], mac[2], mac[3], mac[4], mac[5],
1917 		    tent->masklen, pval);
1918 		break;
1919 	case IPFW_TABLE_INTERFACE:
1920 		/* Interface names */
1921 		printf("%s %s\n", tent->k.iface, pval);
1922 		break;
1923 	case IPFW_TABLE_NUMBER:
1924 		/* numbers */
1925 		printf("%u %s\n", tent->k.key, pval);
1926 		break;
1927 	case IPFW_TABLE_FLOW:
1928 		/* flows */
1929 		tfe = &tent->k.flow;
1930 		comma = "";
1931 
1932 		if ((i->tflags & IPFW_TFFLAG_SRCIP) != 0) {
1933 			if (tfe->af == AF_INET)
1934 				paddr = &tfe->a.a4.sip;
1935 			else
1936 				paddr = &tfe->a.a6.sip6;
1937 
1938 			inet_ntop(tfe->af, paddr, tbuf, sizeof(tbuf));
1939 			printf("%s%s", comma, tbuf);
1940 			comma = ",";
1941 		}
1942 
1943 		if ((i->tflags & IPFW_TFFLAG_PROTO) != 0) {
1944 			printf("%s%d", comma, tfe->proto);
1945 			comma = ",";
1946 		}
1947 
1948 		if ((i->tflags & IPFW_TFFLAG_SRCPORT) != 0) {
1949 			printf("%s%d", comma, ntohs(tfe->sport));
1950 			comma = ",";
1951 		}
1952 		if ((i->tflags & IPFW_TFFLAG_DSTIP) != 0) {
1953 			if (tfe->af == AF_INET)
1954 				paddr = &tfe->a.a4.dip;
1955 			else
1956 				paddr = &tfe->a.a6.dip6;
1957 
1958 			inet_ntop(tfe->af, paddr, tbuf, sizeof(tbuf));
1959 			printf("%s%s", comma, tbuf);
1960 			comma = ",";
1961 		}
1962 
1963 		if ((i->tflags & IPFW_TFFLAG_DSTPORT) != 0) {
1964 			printf("%s%d", comma, ntohs(tfe->dport));
1965 			comma = ",";
1966 		}
1967 
1968 		printf(" %s\n", pval);
1969 	}
1970 }
1971 
1972 static int
table_do_get_stdlist(uint16_t opcode,ipfw_obj_lheader ** polh)1973 table_do_get_stdlist(uint16_t opcode, ipfw_obj_lheader **polh)
1974 {
1975 	ipfw_obj_lheader req, *olh;
1976 	size_t sz;
1977 
1978 	memset(&req, 0, sizeof(req));
1979 	sz = sizeof(req);
1980 
1981 	if (do_get3(opcode, &req.opheader, &sz) != 0)
1982 		if (errno != ENOMEM)
1983 			return (errno);
1984 
1985 	sz = req.size;
1986 	if ((olh = calloc(1, sz)) == NULL)
1987 		return (ENOMEM);
1988 
1989 	olh->size = sz;
1990 	if (do_get3(opcode, &olh->opheader, &sz) != 0) {
1991 		free(olh);
1992 		return (errno);
1993 	}
1994 
1995 	*polh = olh;
1996 	return (0);
1997 }
1998 
1999 static int
table_do_get_algolist(ipfw_obj_lheader ** polh)2000 table_do_get_algolist(ipfw_obj_lheader **polh)
2001 {
2002 
2003 	return (table_do_get_stdlist(IP_FW_TABLES_ALIST, polh));
2004 }
2005 
2006 static int
table_do_get_vlist(ipfw_obj_lheader ** polh)2007 table_do_get_vlist(ipfw_obj_lheader **polh)
2008 {
2009 
2010 	return (table_do_get_stdlist(IP_FW_TABLE_VLIST, polh));
2011 }
2012 
2013 void
ipfw_list_ta(int ac __unused,char * av[]__unused)2014 ipfw_list_ta(int ac __unused, char *av[] __unused)
2015 {
2016 	ipfw_obj_lheader *olh;
2017 	ipfw_ta_info *info;
2018 	const char *atype;
2019 	uint32_t i;
2020 	int error;
2021 
2022 	error = table_do_get_algolist(&olh);
2023 	if (error != 0)
2024 		err(EX_OSERR, "Unable to request algorithm list");
2025 
2026 	info = (ipfw_ta_info *)(olh + 1);
2027 	for (i = 0; i < olh->count; i++) {
2028 		if ((atype = match_value(tabletypes, info->type)) == NULL)
2029 			atype = "unknown";
2030 		printf("--- %s ---\n", info->algoname);
2031 		printf(" type: %s\n refcount: %u\n", atype, info->refcnt);
2032 
2033 		info = (ipfw_ta_info *)((caddr_t)info + olh->objsize);
2034 	}
2035 
2036 	free(olh);
2037 }
2038 
2039 
2040 static int
compare_values(const void * _a,const void * _b)2041 compare_values(const void *_a, const void *_b)
2042 {
2043 	const ipfw_table_value *a, *b;
2044 
2045 	a = (const ipfw_table_value *)_a;
2046 	b = (const ipfw_table_value *)_b;
2047 
2048 	if (a->kidx < b->kidx)
2049 		return (-1);
2050 	else if (a->kidx > b->kidx)
2051 		return (1);
2052 
2053 	return (0);
2054 }
2055 
2056 void
ipfw_list_values(int ac __unused,char * av[]__unused)2057 ipfw_list_values(int ac __unused, char *av[] __unused)
2058 {
2059 	char buf[128];
2060 	ipfw_obj_lheader *olh;
2061 	ipfw_table_value *v;
2062 	uint32_t i, vmask;
2063 	int error;
2064 
2065 	error = table_do_get_vlist(&olh);
2066 	if (error != 0)
2067 		err(EX_OSERR, "Unable to request value list");
2068 
2069 	vmask = 0x7FFFFFFF; /* Similar to IPFW_VTYPE_LEGACY */
2070 
2071 	table_print_valheader(buf, sizeof(buf), vmask);
2072 	printf("HEADER: %s\n", buf);
2073 	v = (ipfw_table_value *)(olh + 1);
2074 	qsort(v, olh->count, olh->objsize, compare_values);
2075 	for (i = 0; i < olh->count; i++) {
2076 		table_show_value(buf, sizeof(buf), (ipfw_table_value *)v,
2077 		    vmask, 0);
2078 		printf("[%u] refs=%lu %s\n", v->kidx, (u_long)v->refcnt, buf);
2079 		v = (ipfw_table_value *)((caddr_t)v + olh->objsize);
2080 	}
2081 
2082 	free(olh);
2083 }
2084 
2085 int
table_check_name(const char * tablename)2086 table_check_name(const char *tablename)
2087 {
2088 
2089 	if (ipfw_check_object_name(tablename) != 0)
2090 		return (EINVAL);
2091 	/* Restrict some 'special' names */
2092 	if (strcmp(tablename, "all") == 0)
2093 		return (EINVAL);
2094 	return (0);
2095 }
2096 
2097