xref: /freebsd/sbin/ipfw/tables.c (revision 2530ed9e7009590ef585f627a8a14ad7cee796e8)
1f1220db8SAlexander V. Chernikov /*
21a33e799SAlexander V. Chernikov  * Copyright (c) 2014 Yandex LLC
31a33e799SAlexander V. Chernikov  * Copyright (c) 2014 Alexander V. Chernikov
4f1220db8SAlexander V. Chernikov  *
5f1220db8SAlexander V. Chernikov  * Redistribution and use in source forms, with and without modification,
6f1220db8SAlexander V. Chernikov  * are permitted provided that this entire comment appears intact.
7f1220db8SAlexander V. Chernikov  *
8f1220db8SAlexander V. Chernikov  * Redistribution in binary form may occur without any restrictions.
9f1220db8SAlexander V. Chernikov  * Obviously, it would be nice if you gave credit where credit is due
10f1220db8SAlexander V. Chernikov  * but requiring it would be too onerous.
11f1220db8SAlexander V. Chernikov  *
12f1220db8SAlexander V. Chernikov  * This software is provided ``AS IS'' without any warranties of any kind.
13f1220db8SAlexander V. Chernikov  *
141a33e799SAlexander V. Chernikov  * in-kernel ipfw tables support.
15f1220db8SAlexander V. Chernikov  *
169fe15d06SAlexander V. Chernikov  * $FreeBSD$
17f1220db8SAlexander V. Chernikov  */
18f1220db8SAlexander V. Chernikov 
19f1220db8SAlexander V. Chernikov 
20f1220db8SAlexander V. Chernikov #include <sys/types.h>
21f1220db8SAlexander V. Chernikov #include <sys/param.h>
22f1220db8SAlexander V. Chernikov #include <sys/socket.h>
23f1220db8SAlexander V. Chernikov #include <sys/sysctl.h>
24f1220db8SAlexander V. Chernikov 
25f1220db8SAlexander V. Chernikov #include <ctype.h>
26f1220db8SAlexander V. Chernikov #include <err.h>
27f1220db8SAlexander V. Chernikov #include <errno.h>
28f1220db8SAlexander V. Chernikov #include <netdb.h>
29f1220db8SAlexander V. Chernikov #include <stdio.h>
30f1220db8SAlexander V. Chernikov #include <stdlib.h>
31f1220db8SAlexander V. Chernikov #include <string.h>
32f1220db8SAlexander V. Chernikov #include <sysexits.h>
33f1220db8SAlexander V. Chernikov 
34f1220db8SAlexander V. Chernikov #include <net/if.h>
35f1220db8SAlexander V. Chernikov #include <netinet/in.h>
36f1220db8SAlexander V. Chernikov #include <netinet/ip_fw.h>
37f1220db8SAlexander V. Chernikov #include <arpa/inet.h>
38*2530ed9eSAndrey V. Elsukov #include <netdb.h>
39f1220db8SAlexander V. Chernikov 
40f1220db8SAlexander V. Chernikov #include "ipfw2.h"
41f1220db8SAlexander V. Chernikov 
42ac35ff17SAlexander V. Chernikov static void table_modify_record(ipfw_obj_header *oh, int ac, char *av[],
433a845e10SAlexander V. Chernikov     int add, int quiet, int update, int atomic);
44ac35ff17SAlexander V. Chernikov static int table_flush(ipfw_obj_header *oh);
45ac35ff17SAlexander V. Chernikov static int table_destroy(ipfw_obj_header *oh);
46ac35ff17SAlexander V. Chernikov static int table_do_create(ipfw_obj_header *oh, ipfw_xtable_info *i);
47adf3b2b9SAlexander V. Chernikov static int table_do_modify(ipfw_obj_header *oh, ipfw_xtable_info *i);
4846d52008SAlexander V. Chernikov static int table_do_swap(ipfw_obj_header *oh, char *second);
49adf3b2b9SAlexander V. Chernikov static void table_create(ipfw_obj_header *oh, int ac, char *av[]);
50adf3b2b9SAlexander V. Chernikov static void table_modify(ipfw_obj_header *oh, int ac, char *av[]);
51adf3b2b9SAlexander V. Chernikov static void table_lookup(ipfw_obj_header *oh, int ac, char *av[]);
524f43138aSAlexander V. Chernikov static void table_lock(ipfw_obj_header *oh, int lock);
5346d52008SAlexander V. Chernikov static int table_swap(ipfw_obj_header *oh, char *second);
54ac35ff17SAlexander V. Chernikov static int table_get_info(ipfw_obj_header *oh, ipfw_xtable_info *i);
55f1220db8SAlexander V. Chernikov static int table_show_info(ipfw_xtable_info *i, void *arg);
56ac35ff17SAlexander V. Chernikov static void table_fill_ntlv(ipfw_obj_ntlv *ntlv, char *name, uint32_t set,
57ac35ff17SAlexander V. Chernikov     uint16_t uidx);
58f1220db8SAlexander V. Chernikov 
59f1220db8SAlexander V. Chernikov static int table_flush_one(ipfw_xtable_info *i, void *arg);
60f1220db8SAlexander V. Chernikov static int table_show_one(ipfw_xtable_info *i, void *arg);
61720ee730SAlexander V. Chernikov static int table_do_get_list(ipfw_xtable_info *i, ipfw_obj_header **poh);
62f1220db8SAlexander V. Chernikov static void table_show_list(ipfw_obj_header *oh, int need_header);
6381d3153dSAlexander V. Chernikov static void table_show_entry(ipfw_xtable_info *i, ipfw_obj_tentry *tent);
64f1220db8SAlexander V. Chernikov 
65ac35ff17SAlexander V. Chernikov static void tentry_fill_key(ipfw_obj_header *oh, ipfw_obj_tentry *tent,
660cba2b28SAlexander V. Chernikov     char *key, int add, uint8_t *ptype, uint32_t *pvmask, ipfw_xtable_info *xi);
67ac35ff17SAlexander V. Chernikov static void tentry_fill_value(ipfw_obj_header *oh, ipfw_obj_tentry *tent,
680cba2b28SAlexander V. Chernikov     char *arg, uint8_t type, uint32_t vmask);
690cba2b28SAlexander V. Chernikov static void table_show_value(char *buf, size_t bufsize, ipfw_table_value *v,
700cba2b28SAlexander V. Chernikov     uint32_t vmask, int print_ip);
71ac35ff17SAlexander V. Chernikov 
72f1220db8SAlexander V. Chernikov typedef int (table_cb_t)(ipfw_xtable_info *i, void *arg);
73f1220db8SAlexander V. Chernikov static int tables_foreach(table_cb_t *f, void *arg, int sort);
74f1220db8SAlexander V. Chernikov 
75f1220db8SAlexander V. Chernikov #ifndef s6_addr32
76f1220db8SAlexander V. Chernikov #define s6_addr32 __u6_addr.__u6_addr32
77f1220db8SAlexander V. Chernikov #endif
78f1220db8SAlexander V. Chernikov 
79ac35ff17SAlexander V. Chernikov static struct _s_x tabletypes[] = {
80c21034b7SAlexander V. Chernikov       { "addr",		IPFW_TABLE_ADDR },
81ac35ff17SAlexander V. Chernikov       { "iface",	IPFW_TABLE_INTERFACE },
82b23d5de9SAlexander V. Chernikov       { "number",	IPFW_TABLE_NUMBER },
83914bffb6SAlexander V. Chernikov       { "flow",		IPFW_TABLE_FLOW },
84ac35ff17SAlexander V. Chernikov       { NULL, 0 }
85ac35ff17SAlexander V. Chernikov };
86ac35ff17SAlexander V. Chernikov 
87ac35ff17SAlexander V. Chernikov static struct _s_x tablevaltypes[] = {
880cba2b28SAlexander V. Chernikov       { "skipto",	IPFW_VTYPE_SKIPTO },
890cba2b28SAlexander V. Chernikov       { "pipe",		IPFW_VTYPE_PIPE },
900cba2b28SAlexander V. Chernikov       { "fib",		IPFW_VTYPE_FIB },
910cba2b28SAlexander V. Chernikov       { "nat",		IPFW_VTYPE_NAT },
920cba2b28SAlexander V. Chernikov       { "dscp",		IPFW_VTYPE_DSCP },
930cba2b28SAlexander V. Chernikov       { "tag",		IPFW_VTYPE_TAG },
940cba2b28SAlexander V. Chernikov       { "divert",	IPFW_VTYPE_DIVERT },
950cba2b28SAlexander V. Chernikov       { "netgraph",	IPFW_VTYPE_NETGRAPH },
960cba2b28SAlexander V. Chernikov       { "limit",	IPFW_VTYPE_LIMIT },
970cba2b28SAlexander V. Chernikov       { "ipv4",		IPFW_VTYPE_NH4 },
980cba2b28SAlexander V. Chernikov       { "ipv6",		IPFW_VTYPE_NH6 },
99adf3b2b9SAlexander V. Chernikov       { NULL, 0 }
100adf3b2b9SAlexander V. Chernikov };
101adf3b2b9SAlexander V. Chernikov 
102ac35ff17SAlexander V. Chernikov static struct _s_x tablecmds[] = {
103ac35ff17SAlexander V. Chernikov       { "add",		TOK_ADD },
104ac35ff17SAlexander V. Chernikov       { "delete",	TOK_DEL },
10546d52008SAlexander V. Chernikov       { "create",	TOK_CREATE },
106ac35ff17SAlexander V. Chernikov       { "destroy",	TOK_DESTROY },
107ac35ff17SAlexander V. Chernikov       { "flush",	TOK_FLUSH },
108adf3b2b9SAlexander V. Chernikov       { "modify",	TOK_MODIFY },
10946d52008SAlexander V. Chernikov       { "swap",		TOK_SWAP },
110ac35ff17SAlexander V. Chernikov       { "info",		TOK_INFO },
111358b9d09SAlexander V. Chernikov       { "detail",	TOK_DETAIL },
112ac35ff17SAlexander V. Chernikov       { "list",		TOK_LIST },
11381d3153dSAlexander V. Chernikov       { "lookup",	TOK_LOOKUP },
1143a845e10SAlexander V. Chernikov       { "atomic",	TOK_ATOMIC },
1154f43138aSAlexander V. Chernikov       { "lock",		TOK_LOCK },
1164f43138aSAlexander V. Chernikov       { "unlock",	TOK_UNLOCK },
117ac35ff17SAlexander V. Chernikov       { NULL, 0 }
118ac35ff17SAlexander V. Chernikov };
119ac35ff17SAlexander V. Chernikov 
120f1220db8SAlexander V. Chernikov static int
121f1220db8SAlexander V. Chernikov lookup_host (char *host, struct in_addr *ipaddr)
122f1220db8SAlexander V. Chernikov {
123f1220db8SAlexander V. Chernikov 	struct hostent *he;
124f1220db8SAlexander V. Chernikov 
125f1220db8SAlexander V. Chernikov 	if (!inet_aton(host, ipaddr)) {
126f1220db8SAlexander V. Chernikov 		if ((he = gethostbyname(host)) == NULL)
127f1220db8SAlexander V. Chernikov 			return(-1);
128f1220db8SAlexander V. Chernikov 		*ipaddr = *(struct in_addr *)he->h_addr_list[0];
129f1220db8SAlexander V. Chernikov 	}
130f1220db8SAlexander V. Chernikov 	return(0);
131f1220db8SAlexander V. Chernikov }
132f1220db8SAlexander V. Chernikov 
133be695df9SAlexander V. Chernikov static int
134be695df9SAlexander V. Chernikov get_token(struct _s_x *table, char *string, char *errbase)
135be695df9SAlexander V. Chernikov {
136be695df9SAlexander V. Chernikov 	int tcmd;
137be695df9SAlexander V. Chernikov 
138be695df9SAlexander V. Chernikov 	if ((tcmd = match_token_relaxed(table, string)) < 0)
139be695df9SAlexander V. Chernikov 		errx(EX_USAGE, "%s %s %s",
140be695df9SAlexander V. Chernikov 		    (tcmd == 0) ? "invalid" : "ambiguous", errbase, string);
141be695df9SAlexander V. Chernikov 
142be695df9SAlexander V. Chernikov 	return (tcmd);
143be695df9SAlexander V. Chernikov }
144be695df9SAlexander V. Chernikov 
145f1220db8SAlexander V. Chernikov /*
146f1220db8SAlexander V. Chernikov  * This one handles all table-related commands
147ac35ff17SAlexander V. Chernikov  * 	ipfw table NAME create ...
148be695df9SAlexander V. Chernikov  * 	ipfw table NAME modify ...
149ac35ff17SAlexander V. Chernikov  * 	ipfw table NAME destroy
150be695df9SAlexander V. Chernikov  * 	ipfw table NAME swap NAME
151be695df9SAlexander V. Chernikov  * 	ipfw table NAME lock
152be695df9SAlexander V. Chernikov  * 	ipfw table NAME unlock
153ac35ff17SAlexander V. Chernikov  * 	ipfw table NAME add addr[/masklen] [value]
154be695df9SAlexander V. Chernikov  * 	ipfw table NAME add [addr[/masklen] value] [addr[/masklen] value] ..
155be695df9SAlexander V. Chernikov  * 	ipfw table NAME delete addr[/masklen] [addr[/masklen]] ..
156be695df9SAlexander V. Chernikov  * 	ipfw table NAME lookup addr
157ac35ff17SAlexander V. Chernikov  * 	ipfw table {NAME | all} flush
158ac35ff17SAlexander V. Chernikov  * 	ipfw table {NAME | all} list
159ac35ff17SAlexander V. Chernikov  * 	ipfw table {NAME | all} info
160be695df9SAlexander V. Chernikov  * 	ipfw table {NAME | all} detail
161f1220db8SAlexander V. Chernikov  */
162f1220db8SAlexander V. Chernikov void
163f1220db8SAlexander V. Chernikov ipfw_table_handler(int ac, char *av[])
164f1220db8SAlexander V. Chernikov {
165ac35ff17SAlexander V. Chernikov 	int do_add, is_all;
1663a845e10SAlexander V. Chernikov 	int atomic, error, tcmd;
167ac35ff17SAlexander V. Chernikov 	ipfw_xtable_info i;
168ac35ff17SAlexander V. Chernikov 	ipfw_obj_header oh;
169f1220db8SAlexander V. Chernikov 	char *tablename;
170ac35ff17SAlexander V. Chernikov 	uint32_t set;
171358b9d09SAlexander V. Chernikov 	void *arg;
172f1220db8SAlexander V. Chernikov 
173ac35ff17SAlexander V. Chernikov 	memset(&oh, 0, sizeof(oh));
174ac35ff17SAlexander V. Chernikov 	is_all = 0;
175ac35ff17SAlexander V. Chernikov 	if (co.use_set != 0)
176ac35ff17SAlexander V. Chernikov 		set = co.use_set - 1;
177ac35ff17SAlexander V. Chernikov 	else
178ac35ff17SAlexander V. Chernikov 		set = 0;
179f1220db8SAlexander V. Chernikov 
180f1220db8SAlexander V. Chernikov 	ac--; av++;
1819d099b4fSAlexander V. Chernikov 	NEED1("table needs name");
182f1220db8SAlexander V. Chernikov 	tablename = *av;
183f1220db8SAlexander V. Chernikov 
184ac35ff17SAlexander V. Chernikov 	if (table_check_name(tablename) == 0) {
185ac35ff17SAlexander V. Chernikov 		table_fill_ntlv(&oh.ntlv, *av, set, 1);
186ac35ff17SAlexander V. Chernikov 		oh.idx = 1;
187ac35ff17SAlexander V. Chernikov 	} else {
188ac35ff17SAlexander V. Chernikov 		if (strcmp(tablename, "all") == 0)
189ac35ff17SAlexander V. Chernikov 			is_all = 1;
190ac35ff17SAlexander V. Chernikov 		else
191ac35ff17SAlexander V. Chernikov 			errx(EX_USAGE, "table name %s is invalid", tablename);
192ac35ff17SAlexander V. Chernikov 	}
193ac35ff17SAlexander V. Chernikov 	ac--; av++;
1949d099b4fSAlexander V. Chernikov 	NEED1("table needs command");
195ac35ff17SAlexander V. Chernikov 
196be695df9SAlexander V. Chernikov 	tcmd = get_token(tablecmds, *av, "table command");
1973a845e10SAlexander V. Chernikov 	/* Check if atomic operation was requested */
1983a845e10SAlexander V. Chernikov 	atomic = 0;
1993a845e10SAlexander V. Chernikov 	if (tcmd == TOK_ATOMIC) {
2003a845e10SAlexander V. Chernikov 		ac--; av++;
2013a845e10SAlexander V. Chernikov 		NEED1("atomic needs command");
202be695df9SAlexander V. Chernikov 		tcmd = get_token(tablecmds, *av, "table command");
2033a845e10SAlexander V. Chernikov 		switch (tcmd) {
2043a845e10SAlexander V. Chernikov 		case TOK_ADD:
2053a845e10SAlexander V. Chernikov 			break;
2063a845e10SAlexander V. Chernikov 		default:
2073a845e10SAlexander V. Chernikov 			errx(EX_USAGE, "atomic is not compatible with %s", *av);
2083a845e10SAlexander V. Chernikov 		}
2093a845e10SAlexander V. Chernikov 		atomic = 1;
2103a845e10SAlexander V. Chernikov 	}
211ac35ff17SAlexander V. Chernikov 
212ac35ff17SAlexander V. Chernikov 	switch (tcmd) {
213ac35ff17SAlexander V. Chernikov 	case TOK_LIST:
214ac35ff17SAlexander V. Chernikov 	case TOK_INFO:
215358b9d09SAlexander V. Chernikov 	case TOK_DETAIL:
216ac35ff17SAlexander V. Chernikov 	case TOK_FLUSH:
217ac35ff17SAlexander V. Chernikov 		break;
218ac35ff17SAlexander V. Chernikov 	default:
219ac35ff17SAlexander V. Chernikov 		if (is_all != 0)
220ac35ff17SAlexander V. Chernikov 			errx(EX_USAGE, "table name required");
221ac35ff17SAlexander V. Chernikov 	}
222ac35ff17SAlexander V. Chernikov 
223ac35ff17SAlexander V. Chernikov 	switch (tcmd) {
224ac35ff17SAlexander V. Chernikov 	case TOK_ADD:
225ac35ff17SAlexander V. Chernikov 	case TOK_DEL:
226f1220db8SAlexander V. Chernikov 		do_add = **av == 'a';
227f1220db8SAlexander V. Chernikov 		ac--; av++;
2283a845e10SAlexander V. Chernikov 		table_modify_record(&oh, ac, av, do_add, co.do_quiet,
2293a845e10SAlexander V. Chernikov 		    co.do_quiet, atomic);
230ac35ff17SAlexander V. Chernikov 		break;
231ac35ff17SAlexander V. Chernikov 	case TOK_CREATE:
232f1220db8SAlexander V. Chernikov 		ac--; av++;
233ac35ff17SAlexander V. Chernikov 		table_create(&oh, ac, av);
234ac35ff17SAlexander V. Chernikov 		break;
235adf3b2b9SAlexander V. Chernikov 	case TOK_MODIFY:
236adf3b2b9SAlexander V. Chernikov 		ac--; av++;
237adf3b2b9SAlexander V. Chernikov 		table_modify(&oh, ac, av);
238adf3b2b9SAlexander V. Chernikov 		break;
239ac35ff17SAlexander V. Chernikov 	case TOK_DESTROY:
240ac35ff17SAlexander V. Chernikov 		if (table_destroy(&oh) != 0)
241ac35ff17SAlexander V. Chernikov 			err(EX_OSERR, "failed to destroy table %s", tablename);
242ac35ff17SAlexander V. Chernikov 		break;
243ac35ff17SAlexander V. Chernikov 	case TOK_FLUSH:
244f1220db8SAlexander V. Chernikov 		if (is_all == 0) {
245ac35ff17SAlexander V. Chernikov 			if ((error = table_flush(&oh)) != 0)
246f1220db8SAlexander V. Chernikov 				err(EX_OSERR, "failed to flush table %s info",
247f1220db8SAlexander V. Chernikov 				    tablename);
248f1220db8SAlexander V. Chernikov 		} else {
249ac35ff17SAlexander V. Chernikov 			error = tables_foreach(table_flush_one, &oh, 1);
250f1220db8SAlexander V. Chernikov 			if (error != 0)
251f1220db8SAlexander V. Chernikov 				err(EX_OSERR, "failed to flush tables list");
252f1220db8SAlexander V. Chernikov 		}
253ac35ff17SAlexander V. Chernikov 		break;
25446d52008SAlexander V. Chernikov 	case TOK_SWAP:
25546d52008SAlexander V. Chernikov 		ac--; av++;
25646d52008SAlexander V. Chernikov 		NEED1("second table name required");
25746d52008SAlexander V. Chernikov 		table_swap(&oh, *av);
25846d52008SAlexander V. Chernikov 		break;
2594f43138aSAlexander V. Chernikov 	case TOK_LOCK:
2604f43138aSAlexander V. Chernikov 	case TOK_UNLOCK:
2614f43138aSAlexander V. Chernikov 		table_lock(&oh, (tcmd == TOK_LOCK));
2624f43138aSAlexander V. Chernikov 		break;
263358b9d09SAlexander V. Chernikov 	case TOK_DETAIL:
264ac35ff17SAlexander V. Chernikov 	case TOK_INFO:
265358b9d09SAlexander V. Chernikov 		arg = (tcmd == TOK_DETAIL) ? (void *)1 : NULL;
266f1220db8SAlexander V. Chernikov 		if (is_all == 0) {
267ac35ff17SAlexander V. Chernikov 			if ((error = table_get_info(&oh, &i)) != 0)
268f1220db8SAlexander V. Chernikov 				err(EX_OSERR, "failed to request table info");
269358b9d09SAlexander V. Chernikov 			table_show_info(&i, arg);
270f1220db8SAlexander V. Chernikov 		} else {
271358b9d09SAlexander V. Chernikov 			error = tables_foreach(table_show_info, arg, 1);
272f1220db8SAlexander V. Chernikov 			if (error != 0)
273f1220db8SAlexander V. Chernikov 				err(EX_OSERR, "failed to request tables list");
274f1220db8SAlexander V. Chernikov 		}
275ac35ff17SAlexander V. Chernikov 		break;
276ac35ff17SAlexander V. Chernikov 	case TOK_LIST:
277ac35ff17SAlexander V. Chernikov 		if (is_all == 0) {
278ac35ff17SAlexander V. Chernikov 			ipfw_xtable_info i;
279ac35ff17SAlexander V. Chernikov 			if ((error = table_get_info(&oh, &i)) != 0)
280ac35ff17SAlexander V. Chernikov 				err(EX_OSERR, "failed to request table info");
281ac35ff17SAlexander V. Chernikov 			table_show_one(&i, NULL);
282f1220db8SAlexander V. Chernikov 		} else {
283ac35ff17SAlexander V. Chernikov 			error = tables_foreach(table_show_one, NULL, 1);
284ac35ff17SAlexander V. Chernikov 			if (error != 0)
285ac35ff17SAlexander V. Chernikov 				err(EX_OSERR, "failed to request tables list");
286f1220db8SAlexander V. Chernikov 		}
287ac35ff17SAlexander V. Chernikov 		break;
28881d3153dSAlexander V. Chernikov 	case TOK_LOOKUP:
28981d3153dSAlexander V. Chernikov 		ac--; av++;
29081d3153dSAlexander V. Chernikov 		table_lookup(&oh, ac, av);
29181d3153dSAlexander V. Chernikov 		break;
292f1220db8SAlexander V. Chernikov 	}
293f1220db8SAlexander V. Chernikov }
294f1220db8SAlexander V. Chernikov 
295f1220db8SAlexander V. Chernikov static void
296ac35ff17SAlexander V. Chernikov table_fill_ntlv(ipfw_obj_ntlv *ntlv, char *name, uint32_t set, uint16_t uidx)
297f1220db8SAlexander V. Chernikov {
298f1220db8SAlexander V. Chernikov 
299563b5ab1SAlexander V. Chernikov 	ntlv->head.type = IPFW_TLV_TBL_NAME;
300f1220db8SAlexander V. Chernikov 	ntlv->head.length = sizeof(ipfw_obj_ntlv);
301f1220db8SAlexander V. Chernikov 	ntlv->idx = uidx;
302ac35ff17SAlexander V. Chernikov 	ntlv->set = set;
303f1220db8SAlexander V. Chernikov 	strlcpy(ntlv->name, name, sizeof(ntlv->name));
304f1220db8SAlexander V. Chernikov }
305f1220db8SAlexander V. Chernikov 
306f1220db8SAlexander V. Chernikov static void
307f1220db8SAlexander V. Chernikov table_fill_objheader(ipfw_obj_header *oh, ipfw_xtable_info *i)
308f1220db8SAlexander V. Chernikov {
309f1220db8SAlexander V. Chernikov 
310f1220db8SAlexander V. Chernikov 	oh->idx = 1;
31181d3153dSAlexander V. Chernikov 	table_fill_ntlv(&oh->ntlv, i->tablename, i->set, 1);
312ac35ff17SAlexander V. Chernikov }
313ac35ff17SAlexander V. Chernikov 
314ac35ff17SAlexander V. Chernikov static struct _s_x tablenewcmds[] = {
315ac35ff17SAlexander V. Chernikov       { "type",		TOK_TYPE },
316ac35ff17SAlexander V. Chernikov       { "valtype",	TOK_VALTYPE },
317ac35ff17SAlexander V. Chernikov       { "algo",		TOK_ALGO },
3184c0c07a5SAlexander V. Chernikov       { "limit",	TOK_LIMIT },
3194f43138aSAlexander V. Chernikov       { "locked",	TOK_LOCK },
320ac35ff17SAlexander V. Chernikov       { NULL, 0 }
321ac35ff17SAlexander V. Chernikov };
322ac35ff17SAlexander V. Chernikov 
323914bffb6SAlexander V. Chernikov static struct _s_x flowtypecmds[] = {
324914bffb6SAlexander V. Chernikov       { "src-ip",	IPFW_TFFLAG_SRCIP },
325914bffb6SAlexander V. Chernikov       { "proto",	IPFW_TFFLAG_PROTO },
326914bffb6SAlexander V. Chernikov       { "src-port",	IPFW_TFFLAG_SRCPORT },
327914bffb6SAlexander V. Chernikov       { "dst-ip",	IPFW_TFFLAG_DSTIP },
328914bffb6SAlexander V. Chernikov       { "dst-port",	IPFW_TFFLAG_DSTPORT },
329914bffb6SAlexander V. Chernikov       { NULL, 0 }
330914bffb6SAlexander V. Chernikov };
331914bffb6SAlexander V. Chernikov 
332914bffb6SAlexander V. Chernikov int
333914bffb6SAlexander V. Chernikov table_parse_type(uint8_t ttype, char *p, uint8_t *tflags)
334914bffb6SAlexander V. Chernikov {
3350cba2b28SAlexander V. Chernikov 	uint32_t fset, fclear;
3360cba2b28SAlexander V. Chernikov 	char *e;
337914bffb6SAlexander V. Chernikov 
338914bffb6SAlexander V. Chernikov 	/* Parse type options */
339914bffb6SAlexander V. Chernikov 	switch(ttype) {
340914bffb6SAlexander V. Chernikov 	case IPFW_TABLE_FLOW:
341914bffb6SAlexander V. Chernikov 		fset = fclear = 0;
3420cba2b28SAlexander V. Chernikov 		if (fill_flags(flowtypecmds, p, &e, &fset, &fclear) != 0)
3430cba2b28SAlexander V. Chernikov 			errx(EX_USAGE,
3440cba2b28SAlexander V. Chernikov 			    "unable to parse flow option %s", e);
345914bffb6SAlexander V. Chernikov 		*tflags = fset;
346914bffb6SAlexander V. Chernikov 		break;
347914bffb6SAlexander V. Chernikov 	default:
348914bffb6SAlexander V. Chernikov 		return (EX_USAGE);
349914bffb6SAlexander V. Chernikov 	}
350914bffb6SAlexander V. Chernikov 
351914bffb6SAlexander V. Chernikov 	return (0);
352914bffb6SAlexander V. Chernikov }
353914bffb6SAlexander V. Chernikov 
354914bffb6SAlexander V. Chernikov void
355914bffb6SAlexander V. Chernikov table_print_type(char *tbuf, size_t size, uint8_t type, uint8_t tflags)
356914bffb6SAlexander V. Chernikov {
357914bffb6SAlexander V. Chernikov 	const char *tname;
358914bffb6SAlexander V. Chernikov 	int l;
359914bffb6SAlexander V. Chernikov 
360914bffb6SAlexander V. Chernikov 	if ((tname = match_value(tabletypes, type)) == NULL)
361914bffb6SAlexander V. Chernikov 		tname = "unknown";
362914bffb6SAlexander V. Chernikov 
363914bffb6SAlexander V. Chernikov 	l = snprintf(tbuf, size, "%s", tname);
364914bffb6SAlexander V. Chernikov 	tbuf += l;
365914bffb6SAlexander V. Chernikov 	size -= l;
366914bffb6SAlexander V. Chernikov 
367914bffb6SAlexander V. Chernikov 	switch(type) {
368914bffb6SAlexander V. Chernikov 	case IPFW_TABLE_FLOW:
369914bffb6SAlexander V. Chernikov 		if (tflags != 0) {
370914bffb6SAlexander V. Chernikov 			*tbuf++ = ':';
371914bffb6SAlexander V. Chernikov 			l--;
372914bffb6SAlexander V. Chernikov 			print_flags_buffer(tbuf, size, flowtypecmds, tflags);
373914bffb6SAlexander V. Chernikov 		}
374914bffb6SAlexander V. Chernikov 		break;
375914bffb6SAlexander V. Chernikov 	}
376914bffb6SAlexander V. Chernikov }
377914bffb6SAlexander V. Chernikov 
378ac35ff17SAlexander V. Chernikov /*
379ac35ff17SAlexander V. Chernikov  * Creates new table
380ac35ff17SAlexander V. Chernikov  *
381c21034b7SAlexander V. Chernikov  * ipfw table NAME create [ type { addr | iface | number | flow } ]
382ac35ff17SAlexander V. Chernikov  *     [ algo algoname ]
383ac35ff17SAlexander V. Chernikov  */
384ac35ff17SAlexander V. Chernikov static void
385ac35ff17SAlexander V. Chernikov table_create(ipfw_obj_header *oh, int ac, char *av[])
386ac35ff17SAlexander V. Chernikov {
387ac35ff17SAlexander V. Chernikov 	ipfw_xtable_info xi;
388ac35ff17SAlexander V. Chernikov 	int error, tcmd, val;
3890cba2b28SAlexander V. Chernikov 	uint32_t fset, fclear;
390ac35ff17SAlexander V. Chernikov 	size_t sz;
3910cba2b28SAlexander V. Chernikov 	char *e, *p;
392ac35ff17SAlexander V. Chernikov 	char tbuf[128];
393ac35ff17SAlexander V. Chernikov 
394ac35ff17SAlexander V. Chernikov 	sz = sizeof(tbuf);
395ac35ff17SAlexander V. Chernikov 	memset(&xi, 0, sizeof(xi));
396ac35ff17SAlexander V. Chernikov 
397ac35ff17SAlexander V. Chernikov 	while (ac > 0) {
398be695df9SAlexander V. Chernikov 		tcmd = get_token(tablenewcmds, *av, "option");
399ac35ff17SAlexander V. Chernikov 		ac--; av++;
400ac35ff17SAlexander V. Chernikov 
401ac35ff17SAlexander V. Chernikov 		switch (tcmd) {
4024c0c07a5SAlexander V. Chernikov 		case TOK_LIMIT:
4034c0c07a5SAlexander V. Chernikov 			NEED1("limit value required");
4044c0c07a5SAlexander V. Chernikov 			xi.limit = strtol(*av, NULL, 10);
4054c0c07a5SAlexander V. Chernikov 			ac--; av++;
4064c0c07a5SAlexander V. Chernikov 			break;
407ac35ff17SAlexander V. Chernikov 		case TOK_TYPE:
408ac35ff17SAlexander V. Chernikov 			NEED1("table type required");
409914bffb6SAlexander V. Chernikov 			/* Type may have suboptions after ':' */
410914bffb6SAlexander V. Chernikov 			if ((p = strchr(*av, ':')) != NULL)
411914bffb6SAlexander V. Chernikov 				*p++ = '\0';
412ac35ff17SAlexander V. Chernikov 			val = match_token(tabletypes, *av);
413914bffb6SAlexander V. Chernikov 			if (val == -1) {
414914bffb6SAlexander V. Chernikov 				concat_tokens(tbuf, sizeof(tbuf), tabletypes,
415914bffb6SAlexander V. Chernikov 				    ", ");
416914bffb6SAlexander V. Chernikov 				errx(EX_USAGE,
417914bffb6SAlexander V. Chernikov 				    "Unknown tabletype: %s. Supported: %s",
418ac35ff17SAlexander V. Chernikov 				    *av, tbuf);
419914bffb6SAlexander V. Chernikov 			}
420914bffb6SAlexander V. Chernikov 			xi.type = val;
421914bffb6SAlexander V. Chernikov 			if (p != NULL) {
422914bffb6SAlexander V. Chernikov 				error = table_parse_type(val, p, &xi.tflags);
423914bffb6SAlexander V. Chernikov 				if (error != 0)
424914bffb6SAlexander V. Chernikov 					errx(EX_USAGE,
425914bffb6SAlexander V. Chernikov 					    "Unsupported suboptions: %s", p);
426914bffb6SAlexander V. Chernikov 			}
427914bffb6SAlexander V. Chernikov 			ac--; av++;
428ac35ff17SAlexander V. Chernikov 			break;
429ac35ff17SAlexander V. Chernikov 		case TOK_VALTYPE:
430ac35ff17SAlexander V. Chernikov 			NEED1("table value type required");
4310cba2b28SAlexander V. Chernikov 			fset = fclear = 0;
4320cba2b28SAlexander V. Chernikov 			val = fill_flags(tablevaltypes, *av, &e, &fset, &fclear);
433ac35ff17SAlexander V. Chernikov 			if (val != -1) {
4340cba2b28SAlexander V. Chernikov 				xi.vmask = fset;
435ac35ff17SAlexander V. Chernikov 				ac--; av++;
436ac35ff17SAlexander V. Chernikov 				break;
437ac35ff17SAlexander V. Chernikov 			}
438ac35ff17SAlexander V. Chernikov 			concat_tokens(tbuf, sizeof(tbuf), tablevaltypes, ", ");
439ac35ff17SAlexander V. Chernikov 			errx(EX_USAGE, "Unknown value type: %s. Supported: %s",
4400cba2b28SAlexander V. Chernikov 			    e, tbuf);
441adf3b2b9SAlexander V. Chernikov 			break;
442ac35ff17SAlexander V. Chernikov 		case TOK_ALGO:
443ac35ff17SAlexander V. Chernikov 			NEED1("table algorithm name required");
444ac35ff17SAlexander V. Chernikov 			if (strlen(*av) > sizeof(xi.algoname))
445ac35ff17SAlexander V. Chernikov 				errx(EX_USAGE, "algorithm name too long");
446ac35ff17SAlexander V. Chernikov 			strlcpy(xi.algoname, *av, sizeof(xi.algoname));
447ac35ff17SAlexander V. Chernikov 			ac--; av++;
448ac35ff17SAlexander V. Chernikov 			break;
4494f43138aSAlexander V. Chernikov 		case TOK_LOCK:
4504f43138aSAlexander V. Chernikov 			xi.flags |= IPFW_TGFLAGS_LOCKED;
4514f43138aSAlexander V. Chernikov 			break;
452ac35ff17SAlexander V. Chernikov 		}
453ac35ff17SAlexander V. Chernikov 	}
454ac35ff17SAlexander V. Chernikov 
455fd0869d5SAlexander V. Chernikov 	/* Set some defaults to preserve compability */
456fd0869d5SAlexander V. Chernikov 	if (xi.algoname[0] == '\0' && xi.type == 0)
457c21034b7SAlexander V. Chernikov 		xi.type = IPFW_TABLE_ADDR;
4580cba2b28SAlexander V. Chernikov 	if (xi.vmask == 0)
4590cba2b28SAlexander V. Chernikov 		xi.vmask = IPFW_VTYPE_LEGACY;
460fd0869d5SAlexander V. Chernikov 
461ac35ff17SAlexander V. Chernikov 	if ((error = table_do_create(oh, &xi)) != 0)
462ac35ff17SAlexander V. Chernikov 		err(EX_OSERR, "Table creation failed");
463f1220db8SAlexander V. Chernikov }
464f1220db8SAlexander V. Chernikov 
465f1220db8SAlexander V. Chernikov /*
466ac35ff17SAlexander V. Chernikov  * Creates new table
467ac35ff17SAlexander V. Chernikov  *
468ac35ff17SAlexander V. Chernikov  * Request: [ ipfw_obj_header ipfw_xtable_info ]
469ac35ff17SAlexander V. Chernikov  *
470f1220db8SAlexander V. Chernikov  * Returns 0 on success.
471f1220db8SAlexander V. Chernikov  */
472f1220db8SAlexander V. Chernikov static int
473ac35ff17SAlexander V. Chernikov table_do_create(ipfw_obj_header *oh, ipfw_xtable_info *i)
474f1220db8SAlexander V. Chernikov {
475ac35ff17SAlexander V. Chernikov 	char tbuf[sizeof(ipfw_obj_header) + sizeof(ipfw_xtable_info)];
476ac35ff17SAlexander V. Chernikov 	int error;
477f1220db8SAlexander V. Chernikov 
478ac35ff17SAlexander V. Chernikov 	memcpy(tbuf, oh, sizeof(*oh));
479ac35ff17SAlexander V. Chernikov 	memcpy(tbuf + sizeof(*oh), i, sizeof(*i));
480ac35ff17SAlexander V. Chernikov 	oh = (ipfw_obj_header *)tbuf;
481ac35ff17SAlexander V. Chernikov 
482ac35ff17SAlexander V. Chernikov 	error = do_set3(IP_FW_TABLE_XCREATE, &oh->opheader, sizeof(tbuf));
483ac35ff17SAlexander V. Chernikov 
484ac35ff17SAlexander V. Chernikov 	return (error);
485ac35ff17SAlexander V. Chernikov }
486ac35ff17SAlexander V. Chernikov 
487ac35ff17SAlexander V. Chernikov /*
488adf3b2b9SAlexander V. Chernikov  * Modifies existing table
489adf3b2b9SAlexander V. Chernikov  *
4900cba2b28SAlexander V. Chernikov  * ipfw table NAME modify [ limit number ]
491adf3b2b9SAlexander V. Chernikov  */
492adf3b2b9SAlexander V. Chernikov static void
493adf3b2b9SAlexander V. Chernikov table_modify(ipfw_obj_header *oh, int ac, char *av[])
494adf3b2b9SAlexander V. Chernikov {
495adf3b2b9SAlexander V. Chernikov 	ipfw_xtable_info xi;
496b6462881SAlexander V. Chernikov 	int tcmd;
497adf3b2b9SAlexander V. Chernikov 	size_t sz;
498adf3b2b9SAlexander V. Chernikov 	char tbuf[128];
499adf3b2b9SAlexander V. Chernikov 
500adf3b2b9SAlexander V. Chernikov 	sz = sizeof(tbuf);
501adf3b2b9SAlexander V. Chernikov 	memset(&xi, 0, sizeof(xi));
502adf3b2b9SAlexander V. Chernikov 
503adf3b2b9SAlexander V. Chernikov 	while (ac > 0) {
504be695df9SAlexander V. Chernikov 		tcmd = get_token(tablenewcmds, *av, "option");
505adf3b2b9SAlexander V. Chernikov 		ac--; av++;
506adf3b2b9SAlexander V. Chernikov 
507adf3b2b9SAlexander V. Chernikov 		switch (tcmd) {
508adf3b2b9SAlexander V. Chernikov 		case TOK_LIMIT:
509adf3b2b9SAlexander V. Chernikov 			NEED1("limit value required");
510adf3b2b9SAlexander V. Chernikov 			xi.limit = strtol(*av, NULL, 10);
511adf3b2b9SAlexander V. Chernikov 			xi.mflags |= IPFW_TMFLAGS_LIMIT;
512adf3b2b9SAlexander V. Chernikov 			ac--; av++;
513adf3b2b9SAlexander V. Chernikov 			break;
5140cba2b28SAlexander V. Chernikov 		default:
5150cba2b28SAlexander V. Chernikov 			errx(EX_USAGE, "cmd is not supported for modificatiob");
516adf3b2b9SAlexander V. Chernikov 		}
517adf3b2b9SAlexander V. Chernikov 	}
518adf3b2b9SAlexander V. Chernikov 
519b6462881SAlexander V. Chernikov 	if (table_do_modify(oh, &xi) != 0)
520adf3b2b9SAlexander V. Chernikov 		err(EX_OSERR, "Table modification failed");
521adf3b2b9SAlexander V. Chernikov }
522adf3b2b9SAlexander V. Chernikov 
523adf3b2b9SAlexander V. Chernikov /*
524adf3b2b9SAlexander V. Chernikov  * Modifies existing table.
525adf3b2b9SAlexander V. Chernikov  *
526adf3b2b9SAlexander V. Chernikov  * Request: [ ipfw_obj_header ipfw_xtable_info ]
527adf3b2b9SAlexander V. Chernikov  *
528adf3b2b9SAlexander V. Chernikov  * Returns 0 on success.
529adf3b2b9SAlexander V. Chernikov  */
530adf3b2b9SAlexander V. Chernikov static int
531adf3b2b9SAlexander V. Chernikov table_do_modify(ipfw_obj_header *oh, ipfw_xtable_info *i)
532adf3b2b9SAlexander V. Chernikov {
533adf3b2b9SAlexander V. Chernikov 	char tbuf[sizeof(ipfw_obj_header) + sizeof(ipfw_xtable_info)];
534adf3b2b9SAlexander V. Chernikov 	int error;
535adf3b2b9SAlexander V. Chernikov 
536adf3b2b9SAlexander V. Chernikov 	memcpy(tbuf, oh, sizeof(*oh));
537adf3b2b9SAlexander V. Chernikov 	memcpy(tbuf + sizeof(*oh), i, sizeof(*i));
538adf3b2b9SAlexander V. Chernikov 	oh = (ipfw_obj_header *)tbuf;
539adf3b2b9SAlexander V. Chernikov 
540adf3b2b9SAlexander V. Chernikov 	error = do_set3(IP_FW_TABLE_XMODIFY, &oh->opheader, sizeof(tbuf));
541adf3b2b9SAlexander V. Chernikov 
542adf3b2b9SAlexander V. Chernikov 	return (error);
543adf3b2b9SAlexander V. Chernikov }
5444f43138aSAlexander V. Chernikov 
5454f43138aSAlexander V. Chernikov /*
5464f43138aSAlexander V. Chernikov  * Locks or unlocks given table
5474f43138aSAlexander V. Chernikov  */
5484f43138aSAlexander V. Chernikov static void
5494f43138aSAlexander V. Chernikov table_lock(ipfw_obj_header *oh, int lock)
5504f43138aSAlexander V. Chernikov {
5514f43138aSAlexander V. Chernikov 	ipfw_xtable_info xi;
5524f43138aSAlexander V. Chernikov 
5534f43138aSAlexander V. Chernikov 	memset(&xi, 0, sizeof(xi));
5544f43138aSAlexander V. Chernikov 
5554f43138aSAlexander V. Chernikov 	xi.mflags |= IPFW_TMFLAGS_LOCK;
5564f43138aSAlexander V. Chernikov 	xi.flags |= (lock != 0) ? IPFW_TGFLAGS_LOCKED : 0;
5574f43138aSAlexander V. Chernikov 
558b6462881SAlexander V. Chernikov 	if (table_do_modify(oh, &xi) != 0)
5594f43138aSAlexander V. Chernikov 		err(EX_OSERR, "Table %s failed", lock != 0 ? "lock" : "unlock");
5604f43138aSAlexander V. Chernikov }
5614f43138aSAlexander V. Chernikov 
562adf3b2b9SAlexander V. Chernikov /*
563ac35ff17SAlexander V. Chernikov  * Destroys given table specified by @oh->ntlv.
564ac35ff17SAlexander V. Chernikov  * Returns 0 on success.
565ac35ff17SAlexander V. Chernikov  */
566ac35ff17SAlexander V. Chernikov static int
567ac35ff17SAlexander V. Chernikov table_destroy(ipfw_obj_header *oh)
568ac35ff17SAlexander V. Chernikov {
569ac35ff17SAlexander V. Chernikov 
570ac35ff17SAlexander V. Chernikov 	if (do_set3(IP_FW_TABLE_XDESTROY, &oh->opheader, sizeof(*oh)) != 0)
571f1220db8SAlexander V. Chernikov 		return (-1);
572f1220db8SAlexander V. Chernikov 
573f1220db8SAlexander V. Chernikov 	return (0);
574f1220db8SAlexander V. Chernikov }
575f1220db8SAlexander V. Chernikov 
576f1220db8SAlexander V. Chernikov /*
577ac35ff17SAlexander V. Chernikov  * Flushes given table specified by @oh->ntlv.
578f1220db8SAlexander V. Chernikov  * Returns 0 on success.
579f1220db8SAlexander V. Chernikov  */
580f1220db8SAlexander V. Chernikov static int
581ac35ff17SAlexander V. Chernikov table_flush(ipfw_obj_header *oh)
582f1220db8SAlexander V. Chernikov {
583f1220db8SAlexander V. Chernikov 
584ac35ff17SAlexander V. Chernikov 	if (do_set3(IP_FW_TABLE_XFLUSH, &oh->opheader, sizeof(*oh)) != 0)
585f1220db8SAlexander V. Chernikov 		return (-1);
586f1220db8SAlexander V. Chernikov 
587f1220db8SAlexander V. Chernikov 	return (0);
588f1220db8SAlexander V. Chernikov }
589f1220db8SAlexander V. Chernikov 
59046d52008SAlexander V. Chernikov static int
59146d52008SAlexander V. Chernikov table_do_swap(ipfw_obj_header *oh, char *second)
59246d52008SAlexander V. Chernikov {
59346d52008SAlexander V. Chernikov 	char tbuf[sizeof(ipfw_obj_header) + sizeof(ipfw_obj_ntlv)];
59446d52008SAlexander V. Chernikov 	int error;
59546d52008SAlexander V. Chernikov 
59646d52008SAlexander V. Chernikov 	memset(tbuf, 0, sizeof(tbuf));
59746d52008SAlexander V. Chernikov 	memcpy(tbuf, oh, sizeof(*oh));
59846d52008SAlexander V. Chernikov 	oh = (ipfw_obj_header *)tbuf;
59946d52008SAlexander V. Chernikov 	table_fill_ntlv((ipfw_obj_ntlv *)(oh + 1), second, oh->ntlv.set, 1);
60046d52008SAlexander V. Chernikov 
60146d52008SAlexander V. Chernikov 	error = do_set3(IP_FW_TABLE_XSWAP, &oh->opheader, sizeof(tbuf));
60246d52008SAlexander V. Chernikov 
60346d52008SAlexander V. Chernikov 	return (error);
60446d52008SAlexander V. Chernikov }
60546d52008SAlexander V. Chernikov 
60646d52008SAlexander V. Chernikov /*
60746d52008SAlexander V. Chernikov  * Swaps given table with @second one.
60846d52008SAlexander V. Chernikov  */
60946d52008SAlexander V. Chernikov static int
61046d52008SAlexander V. Chernikov table_swap(ipfw_obj_header *oh, char *second)
61146d52008SAlexander V. Chernikov {
61246d52008SAlexander V. Chernikov 	int error;
61346d52008SAlexander V. Chernikov 
61446d52008SAlexander V. Chernikov 	if (table_check_name(second) != 0)
61546d52008SAlexander V. Chernikov 		errx(EX_USAGE, "table name %s is invalid", second);
61646d52008SAlexander V. Chernikov 
61746d52008SAlexander V. Chernikov 	error = table_do_swap(oh, second);
61846d52008SAlexander V. Chernikov 
61946d52008SAlexander V. Chernikov 	switch (error) {
62046d52008SAlexander V. Chernikov 	case EINVAL:
62146d52008SAlexander V. Chernikov 		errx(EX_USAGE, "Unable to swap table: check types");
62246d52008SAlexander V. Chernikov 	case EFBIG:
62346d52008SAlexander V. Chernikov 		errx(EX_USAGE, "Unable to swap table: check limits");
62446d52008SAlexander V. Chernikov 	}
62546d52008SAlexander V. Chernikov 
62646d52008SAlexander V. Chernikov 	return (0);
62746d52008SAlexander V. Chernikov }
62846d52008SAlexander V. Chernikov 
62946d52008SAlexander V. Chernikov 
630f1220db8SAlexander V. Chernikov /*
631ac35ff17SAlexander V. Chernikov  * Retrieves table in given table specified by @oh->ntlv.
632f1220db8SAlexander V. Chernikov  * it inside @i.
633f1220db8SAlexander V. Chernikov  * Returns 0 on success.
634f1220db8SAlexander V. Chernikov  */
635f1220db8SAlexander V. Chernikov static int
636ac35ff17SAlexander V. Chernikov table_get_info(ipfw_obj_header *oh, ipfw_xtable_info *i)
637f1220db8SAlexander V. Chernikov {
638f1220db8SAlexander V. Chernikov 	char tbuf[sizeof(ipfw_obj_header) + sizeof(ipfw_xtable_info)];
639f1220db8SAlexander V. Chernikov 	size_t sz;
640f1220db8SAlexander V. Chernikov 
641f1220db8SAlexander V. Chernikov 	sz = sizeof(tbuf);
642f1220db8SAlexander V. Chernikov 	memset(tbuf, 0, sizeof(tbuf));
643ac35ff17SAlexander V. Chernikov 	memcpy(tbuf, oh, sizeof(*oh));
644f1220db8SAlexander V. Chernikov 	oh = (ipfw_obj_header *)tbuf;
645f1220db8SAlexander V. Chernikov 
646b6462881SAlexander V. Chernikov 	if (do_get3(IP_FW_TABLE_XINFO, &oh->opheader, &sz) != 0)
647b6462881SAlexander V. Chernikov 		return (errno);
648f1220db8SAlexander V. Chernikov 
649f1220db8SAlexander V. Chernikov 	if (sz < sizeof(tbuf))
650ac35ff17SAlexander V. Chernikov 		return (EINVAL);
651f1220db8SAlexander V. Chernikov 
652f1220db8SAlexander V. Chernikov 	*i = *(ipfw_xtable_info *)(oh + 1);
653f1220db8SAlexander V. Chernikov 
654f1220db8SAlexander V. Chernikov 	return (0);
655f1220db8SAlexander V. Chernikov }
656f1220db8SAlexander V. Chernikov 
6575f379342SAlexander V. Chernikov static struct _s_x tablealgoclass[] = {
6585f379342SAlexander V. Chernikov       { "hash",		IPFW_TACLASS_HASH },
6595f379342SAlexander V. Chernikov       { "array",	IPFW_TACLASS_ARRAY },
6605f379342SAlexander V. Chernikov       { "radix",	IPFW_TACLASS_RADIX },
6615f379342SAlexander V. Chernikov       { NULL, 0 }
6625f379342SAlexander V. Chernikov };
6635f379342SAlexander V. Chernikov 
6645f379342SAlexander V. Chernikov struct ta_cldata {
6655f379342SAlexander V. Chernikov 	uint8_t		taclass;
6665f379342SAlexander V. Chernikov 	uint8_t		spare4;
6675f379342SAlexander V. Chernikov 	uint16_t	itemsize;
6685f379342SAlexander V. Chernikov 	uint16_t	itemsize6;
6695f379342SAlexander V. Chernikov 	uint32_t	size;
6705f379342SAlexander V. Chernikov 	uint32_t	count;
6715f379342SAlexander V. Chernikov };
6725f379342SAlexander V. Chernikov 
6735f379342SAlexander V. Chernikov /*
6745f379342SAlexander V. Chernikov  * Print global/per-AF table @i algorithm info.
6755f379342SAlexander V. Chernikov  */
6765f379342SAlexander V. Chernikov static void
6775f379342SAlexander V. Chernikov table_show_tainfo(ipfw_xtable_info *i, struct ta_cldata *d,
6785f379342SAlexander V. Chernikov     const char *af, const char *taclass)
6795f379342SAlexander V. Chernikov {
6805f379342SAlexander V. Chernikov 
6815f379342SAlexander V. Chernikov 	switch (d->taclass) {
6825f379342SAlexander V. Chernikov 	case IPFW_TACLASS_HASH:
6835f379342SAlexander V. Chernikov 	case IPFW_TACLASS_ARRAY:
6845f379342SAlexander V. Chernikov 		printf(" %salgorithm %s info\n", af, taclass);
6855f379342SAlexander V. Chernikov 		if (d->itemsize == d->itemsize6)
6865f379342SAlexander V. Chernikov 			printf("  size: %u items: %u itemsize: %u\n",
6875f379342SAlexander V. Chernikov 			    d->size, d->count, d->itemsize);
6885f379342SAlexander V. Chernikov 		else
6895f379342SAlexander V. Chernikov 			printf("  size: %u items: %u "
6905f379342SAlexander V. Chernikov 			    "itemsize4: %u itemsize6: %u\n",
6915f379342SAlexander V. Chernikov 			    d->size, d->count,
6925f379342SAlexander V. Chernikov 			    d->itemsize, d->itemsize6);
6935f379342SAlexander V. Chernikov 		break;
6945f379342SAlexander V. Chernikov 	case IPFW_TACLASS_RADIX:
6955f379342SAlexander V. Chernikov 		printf(" %salgorithm %s info\n", af, taclass);
6965f379342SAlexander V. Chernikov 		if (d->itemsize == d->itemsize6)
6975f379342SAlexander V. Chernikov 			printf("  items: %u itemsize: %u\n",
6985f379342SAlexander V. Chernikov 			    d->count, d->itemsize);
6995f379342SAlexander V. Chernikov 		else
7005f379342SAlexander V. Chernikov 			printf("  items: %u "
7015f379342SAlexander V. Chernikov 			    "itemsize4: %u itemsize6: %u\n",
7025f379342SAlexander V. Chernikov 			    d->count, d->itemsize, d->itemsize6);
7035f379342SAlexander V. Chernikov 		break;
7045f379342SAlexander V. Chernikov 	default:
7055f379342SAlexander V. Chernikov 		printf(" algo class: %s\n", taclass);
7065f379342SAlexander V. Chernikov 	}
7075f379342SAlexander V. Chernikov }
7085f379342SAlexander V. Chernikov 
7090cba2b28SAlexander V. Chernikov static void
7100cba2b28SAlexander V. Chernikov table_print_valheader(char *buf, size_t bufsize, uint32_t vmask)
7110cba2b28SAlexander V. Chernikov {
7120cba2b28SAlexander V. Chernikov 
7130cba2b28SAlexander V. Chernikov 	if (vmask == IPFW_VTYPE_LEGACY) {
7140cba2b28SAlexander V. Chernikov 		snprintf(buf, bufsize, "legacy");
7150cba2b28SAlexander V. Chernikov 		return;
7160cba2b28SAlexander V. Chernikov 	}
7170cba2b28SAlexander V. Chernikov 
7180cba2b28SAlexander V. Chernikov 	print_flags_buffer(buf, bufsize, tablevaltypes, vmask);
7190cba2b28SAlexander V. Chernikov }
7200cba2b28SAlexander V. Chernikov 
721f1220db8SAlexander V. Chernikov /*
722f1220db8SAlexander V. Chernikov  * Prints table info struct @i in human-readable form.
723f1220db8SAlexander V. Chernikov  */
724f1220db8SAlexander V. Chernikov static int
725f1220db8SAlexander V. Chernikov table_show_info(ipfw_xtable_info *i, void *arg)
726f1220db8SAlexander V. Chernikov {
7270cba2b28SAlexander V. Chernikov 	const char *vtype;
7285f379342SAlexander V. Chernikov 	ipfw_ta_tinfo *tainfo;
7295f379342SAlexander V. Chernikov 	int afdata, afitem;
7305f379342SAlexander V. Chernikov 	struct ta_cldata d;
731adf3b2b9SAlexander V. Chernikov 	char ttype[64], tvtype[64];
732f1220db8SAlexander V. Chernikov 
733914bffb6SAlexander V. Chernikov 	table_print_type(ttype, sizeof(ttype), i->type, i->tflags);
7340cba2b28SAlexander V. Chernikov 	table_print_valheader(tvtype, sizeof(tvtype), i->vmask);
735ac35ff17SAlexander V. Chernikov 
736914bffb6SAlexander V. Chernikov 	printf("--- table(%s), set(%u) ---\n", i->tablename, i->set);
7374f43138aSAlexander V. Chernikov 	if ((i->flags & IPFW_TGFLAGS_LOCKED) != 0)
7384f43138aSAlexander V. Chernikov 		printf(" kindex: %d, type: %s, locked\n", i->kidx, ttype);
7394f43138aSAlexander V. Chernikov 	else
740914bffb6SAlexander V. Chernikov 		printf(" kindex: %d, type: %s\n", i->kidx, ttype);
7410cba2b28SAlexander V. Chernikov 	printf(" references: %u, valtype: %s\n", i->refcnt, tvtype);
7429d099b4fSAlexander V. Chernikov 	printf(" algorithm: %s\n", i->algoname);
743f1220db8SAlexander V. Chernikov 	printf(" items: %u, size: %u\n", i->count, i->size);
7444c0c07a5SAlexander V. Chernikov 	if (i->limit > 0)
7454c0c07a5SAlexander V. Chernikov 		printf(" limit: %u\n", i->limit);
746f1220db8SAlexander V. Chernikov 
747358b9d09SAlexander V. Chernikov 	/* Print algo-specific info if requested & set  */
748358b9d09SAlexander V. Chernikov 	if (arg == NULL)
749358b9d09SAlexander V. Chernikov 		return (0);
750358b9d09SAlexander V. Chernikov 
7515f379342SAlexander V. Chernikov 	if ((i->ta_info.flags & IPFW_TATFLAGS_DATA) == 0)
7525f379342SAlexander V. Chernikov 		return (0);
7535f379342SAlexander V. Chernikov 	tainfo = &i->ta_info;
7545f379342SAlexander V. Chernikov 
7555f379342SAlexander V. Chernikov 	afdata = 0;
7565f379342SAlexander V. Chernikov 	afitem = 0;
7575f379342SAlexander V. Chernikov 	if (tainfo->flags & IPFW_TATFLAGS_AFDATA)
7585f379342SAlexander V. Chernikov 		afdata = 1;
7595f379342SAlexander V. Chernikov 	if (tainfo->flags & IPFW_TATFLAGS_AFITEM)
7605f379342SAlexander V. Chernikov 		afitem = 1;
7615f379342SAlexander V. Chernikov 
7625f379342SAlexander V. Chernikov 	memset(&d, 0, sizeof(d));
7635f379342SAlexander V. Chernikov 	d.taclass = tainfo->taclass4;
7645f379342SAlexander V. Chernikov 	d.size = tainfo->size4;
7655f379342SAlexander V. Chernikov 	d.count = tainfo->count4;
7665f379342SAlexander V. Chernikov 	d.itemsize = tainfo->itemsize4;
7675f379342SAlexander V. Chernikov 	if (afdata == 0 && afitem != 0)
7685f379342SAlexander V. Chernikov 		d.itemsize6 = tainfo->itemsize6;
7695f379342SAlexander V. Chernikov 	else
7705f379342SAlexander V. Chernikov 		d.itemsize6 = d.itemsize;
7715f379342SAlexander V. Chernikov 	if ((vtype = match_value(tablealgoclass, d.taclass)) == NULL)
7725f379342SAlexander V. Chernikov 		vtype = "unknown";
7735f379342SAlexander V. Chernikov 
7745f379342SAlexander V. Chernikov 	if (afdata == 0) {
7755f379342SAlexander V. Chernikov 		table_show_tainfo(i, &d, "", vtype);
7765f379342SAlexander V. Chernikov 	} else {
7775f379342SAlexander V. Chernikov 		table_show_tainfo(i, &d, "IPv4 ", vtype);
7785f379342SAlexander V. Chernikov 		memset(&d, 0, sizeof(d));
7795f379342SAlexander V. Chernikov 		d.taclass = tainfo->taclass6;
7805f379342SAlexander V. Chernikov 		if ((vtype = match_value(tablealgoclass, d.taclass)) == NULL)
7815f379342SAlexander V. Chernikov 			vtype = "unknown";
7825f379342SAlexander V. Chernikov 		d.size = tainfo->size6;
7835f379342SAlexander V. Chernikov 		d.count = tainfo->count6;
7845f379342SAlexander V. Chernikov 		d.itemsize = tainfo->itemsize6;
7855f379342SAlexander V. Chernikov 		d.itemsize6 = d.itemsize;
7865f379342SAlexander V. Chernikov 		table_show_tainfo(i, &d, "IPv6 ", vtype);
7875f379342SAlexander V. Chernikov 	}
7885f379342SAlexander V. Chernikov 
789f1220db8SAlexander V. Chernikov 	return (0);
790f1220db8SAlexander V. Chernikov }
791f1220db8SAlexander V. Chernikov 
792f1220db8SAlexander V. Chernikov 
793f1220db8SAlexander V. Chernikov /*
794f1220db8SAlexander V. Chernikov  * Function wrappers which can be used either
795f1220db8SAlexander V. Chernikov  * as is or as foreach function parameter.
796f1220db8SAlexander V. Chernikov  */
797f1220db8SAlexander V. Chernikov 
798f1220db8SAlexander V. Chernikov static int
799f1220db8SAlexander V. Chernikov table_show_one(ipfw_xtable_info *i, void *arg)
800f1220db8SAlexander V. Chernikov {
801f1220db8SAlexander V. Chernikov 	ipfw_obj_header *oh;
80281d3153dSAlexander V. Chernikov 	int error;
803f1220db8SAlexander V. Chernikov 
804720ee730SAlexander V. Chernikov 	if ((error = table_do_get_list(i, &oh)) != 0) {
80581d3153dSAlexander V. Chernikov 		err(EX_OSERR, "Error requesting table %s list", i->tablename);
80681d3153dSAlexander V. Chernikov 		return (error);
80781d3153dSAlexander V. Chernikov 	}
80881d3153dSAlexander V. Chernikov 
809f1220db8SAlexander V. Chernikov 	table_show_list(oh, 1);
810f1220db8SAlexander V. Chernikov 
811f1220db8SAlexander V. Chernikov 	free(oh);
812f1220db8SAlexander V. Chernikov 	return (0);
813f1220db8SAlexander V. Chernikov }
814f1220db8SAlexander V. Chernikov 
815f1220db8SAlexander V. Chernikov static int
816f1220db8SAlexander V. Chernikov table_flush_one(ipfw_xtable_info *i, void *arg)
817f1220db8SAlexander V. Chernikov {
818ac35ff17SAlexander V. Chernikov 	ipfw_obj_header *oh;
819f1220db8SAlexander V. Chernikov 
820ac35ff17SAlexander V. Chernikov 	oh = (ipfw_obj_header *)arg;
821ac35ff17SAlexander V. Chernikov 
822ac35ff17SAlexander V. Chernikov 	table_fill_ntlv(&oh->ntlv, i->tablename, i->set, 1);
823ac35ff17SAlexander V. Chernikov 
824ac35ff17SAlexander V. Chernikov 	return (table_flush(oh));
825f1220db8SAlexander V. Chernikov }
826f1220db8SAlexander V. Chernikov 
827ac35ff17SAlexander V. Chernikov static int
828ac35ff17SAlexander V. Chernikov table_do_modify_record(int cmd, ipfw_obj_header *oh,
8293a845e10SAlexander V. Chernikov     ipfw_obj_tentry *tent, int count, int atomic)
830ac35ff17SAlexander V. Chernikov {
831db785d31SAlexander V. Chernikov 	ipfw_obj_ctlv *ctlv;
8323a845e10SAlexander V. Chernikov 	ipfw_obj_tentry *tent_base;
8333a845e10SAlexander V. Chernikov 	caddr_t pbuf;
834db785d31SAlexander V. Chernikov 	char xbuf[sizeof(*oh) + sizeof(ipfw_obj_ctlv) + sizeof(*tent)];
8353a845e10SAlexander V. Chernikov 	int error, i;
8363a845e10SAlexander V. Chernikov 	size_t sz;
837ac35ff17SAlexander V. Chernikov 
8383a845e10SAlexander V. Chernikov 	sz = sizeof(*ctlv) + sizeof(*tent) * count;
8393a845e10SAlexander V. Chernikov 	if (count == 1) {
840ac35ff17SAlexander V. Chernikov 		memset(xbuf, 0, sizeof(xbuf));
8413a845e10SAlexander V. Chernikov 		pbuf = xbuf;
8423a845e10SAlexander V. Chernikov 	} else {
8433a845e10SAlexander V. Chernikov 		if ((pbuf = calloc(1, sizeof(*oh) + sz)) == NULL)
8443a845e10SAlexander V. Chernikov 			return (ENOMEM);
8453a845e10SAlexander V. Chernikov 	}
8463a845e10SAlexander V. Chernikov 
8473a845e10SAlexander V. Chernikov 	memcpy(pbuf, oh, sizeof(*oh));
8483a845e10SAlexander V. Chernikov 	oh = (ipfw_obj_header *)pbuf;
849ac35ff17SAlexander V. Chernikov 	oh->opheader.version = 1;
850ac35ff17SAlexander V. Chernikov 
851db785d31SAlexander V. Chernikov 	ctlv = (ipfw_obj_ctlv *)(oh + 1);
8523a845e10SAlexander V. Chernikov 	ctlv->count = count;
8533a845e10SAlexander V. Chernikov 	ctlv->head.length = sz;
8543a845e10SAlexander V. Chernikov 	if (atomic != 0)
8553a845e10SAlexander V. Chernikov 		ctlv->flags |= IPFW_CTF_ATOMIC;
856db785d31SAlexander V. Chernikov 
8573a845e10SAlexander V. Chernikov 	tent_base = tent;
8583a845e10SAlexander V. Chernikov 	memcpy(ctlv + 1, tent, sizeof(*tent) * count);
859db785d31SAlexander V. Chernikov 	tent = (ipfw_obj_tentry *)(ctlv + 1);
8603a845e10SAlexander V. Chernikov 	for (i = 0; i < count; i++, tent++) {
861ac35ff17SAlexander V. Chernikov 		tent->head.length = sizeof(ipfw_obj_tentry);
8623a845e10SAlexander V. Chernikov 		tent->idx = oh->idx;
8633a845e10SAlexander V. Chernikov 	}
864ac35ff17SAlexander V. Chernikov 
8653a845e10SAlexander V. Chernikov 	sz += sizeof(*oh);
8663a845e10SAlexander V. Chernikov 	error = do_get3(cmd, &oh->opheader, &sz);
8673a845e10SAlexander V. Chernikov 	tent = (ipfw_obj_tentry *)(ctlv + 1);
8683a845e10SAlexander V. Chernikov 	/* Copy result back to provided buffer */
8693a845e10SAlexander V. Chernikov 	memcpy(tent_base, ctlv + 1, sizeof(*tent) * count);
8703a845e10SAlexander V. Chernikov 
8713a845e10SAlexander V. Chernikov 	if (pbuf != xbuf)
8723a845e10SAlexander V. Chernikov 		free(pbuf);
873ac35ff17SAlexander V. Chernikov 
874ac35ff17SAlexander V. Chernikov 	return (error);
875ac35ff17SAlexander V. Chernikov }
876ac35ff17SAlexander V. Chernikov 
877ac35ff17SAlexander V. Chernikov static void
8783a845e10SAlexander V. Chernikov table_modify_record(ipfw_obj_header *oh, int ac, char *av[], int add,
8793a845e10SAlexander V. Chernikov     int quiet, int update, int atomic)
880ac35ff17SAlexander V. Chernikov {
8813a845e10SAlexander V. Chernikov 	ipfw_obj_tentry *ptent, tent, *tent_buf;
88281d3153dSAlexander V. Chernikov 	ipfw_xtable_info xi;
8830cba2b28SAlexander V. Chernikov 	uint8_t type;
8840cba2b28SAlexander V. Chernikov 	uint32_t vmask;
8853a845e10SAlexander V. Chernikov 	int cmd, count, error, i, ignored;
8863a845e10SAlexander V. Chernikov 	char *texterr, *etxt, *px;
887ac35ff17SAlexander V. Chernikov 
888ac35ff17SAlexander V. Chernikov 	if (ac == 0)
889ac35ff17SAlexander V. Chernikov 		errx(EX_USAGE, "address required");
890ac35ff17SAlexander V. Chernikov 
891ac35ff17SAlexander V. Chernikov 	if (add != 0) {
892ac35ff17SAlexander V. Chernikov 		cmd = IP_FW_TABLE_XADD;
8934c0c07a5SAlexander V. Chernikov 		texterr = "Adding record failed";
894ac35ff17SAlexander V. Chernikov 	} else {
895ac35ff17SAlexander V. Chernikov 		cmd = IP_FW_TABLE_XDEL;
8964c0c07a5SAlexander V. Chernikov 		texterr = "Deleting record failed";
897ac35ff17SAlexander V. Chernikov 	}
898ac35ff17SAlexander V. Chernikov 
8993a845e10SAlexander V. Chernikov 	/*
9003a845e10SAlexander V. Chernikov 	 * Calculate number of entries:
9013a845e10SAlexander V. Chernikov 	 * Assume [key val] x N for add
9023a845e10SAlexander V. Chernikov 	 * and
9033a845e10SAlexander V. Chernikov 	 * key x N for delete
9043a845e10SAlexander V. Chernikov 	 */
9053a845e10SAlexander V. Chernikov 	count = (add != 0) ? ac / 2 + 1 : ac;
9063a845e10SAlexander V. Chernikov 
9073a845e10SAlexander V. Chernikov 	if (count <= 1) {
9083a845e10SAlexander V. Chernikov 		/* Adding single entry with/without value */
9093a845e10SAlexander V. Chernikov 		memset(&tent, 0, sizeof(tent));
9103a845e10SAlexander V. Chernikov 		tent_buf = &tent;
9113a845e10SAlexander V. Chernikov 	} else {
9123a845e10SAlexander V. Chernikov 
9133a845e10SAlexander V. Chernikov 		if ((tent_buf = calloc(count, sizeof(tent))) == NULL)
9143a845e10SAlexander V. Chernikov 			errx(EX_OSERR,
9153a845e10SAlexander V. Chernikov 			    "Unable to allocate memory for all entries");
9163a845e10SAlexander V. Chernikov 	}
9173a845e10SAlexander V. Chernikov 	ptent = tent_buf;
9183a845e10SAlexander V. Chernikov 
9193a845e10SAlexander V. Chernikov 	memset(&xi, 0, sizeof(xi));
9203a845e10SAlexander V. Chernikov 	count = 0;
9213a845e10SAlexander V. Chernikov 	while (ac > 0) {
9220cba2b28SAlexander V. Chernikov 		tentry_fill_key(oh, ptent, *av, add, &type, &vmask, &xi);
9233a845e10SAlexander V. Chernikov 
9243a845e10SAlexander V. Chernikov 		/*
9253a845e10SAlexander V. Chernikov 		 * compability layer: auto-create table if not exists
9263a845e10SAlexander V. Chernikov 		 */
9273a845e10SAlexander V. Chernikov 		if (xi.tablename[0] == '\0') {
9283a845e10SAlexander V. Chernikov 			xi.type = type;
9290cba2b28SAlexander V. Chernikov 			xi.vmask = vmask;
9303a845e10SAlexander V. Chernikov 			strlcpy(xi.tablename, oh->ntlv.name,
9313a845e10SAlexander V. Chernikov 			    sizeof(xi.tablename));
9323a845e10SAlexander V. Chernikov 			fprintf(stderr, "DEPRECATED: inserting data info "
9333a845e10SAlexander V. Chernikov 			    "non-existent table %s. (auto-created)\n",
9343a845e10SAlexander V. Chernikov 			    xi.tablename);
9353a845e10SAlexander V. Chernikov 			table_do_create(oh, &xi);
9363a845e10SAlexander V. Chernikov 		}
9373a845e10SAlexander V. Chernikov 
9383a845e10SAlexander V. Chernikov 		oh->ntlv.type = type;
9393a845e10SAlexander V. Chernikov 		ac--; av++;
9403a845e10SAlexander V. Chernikov 
9413a845e10SAlexander V. Chernikov 		if (add != 0 && ac > 0) {
9420cba2b28SAlexander V. Chernikov 			tentry_fill_value(oh, ptent, *av, type, vmask);
9433a845e10SAlexander V. Chernikov 			ac--; av++;
9443a845e10SAlexander V. Chernikov 		}
9453a845e10SAlexander V. Chernikov 
9463a845e10SAlexander V. Chernikov 		if (update != 0)
9473a845e10SAlexander V. Chernikov 			ptent->head.flags |= IPFW_TF_UPDATE;
9483a845e10SAlexander V. Chernikov 
9493a845e10SAlexander V. Chernikov 		count++;
9503a845e10SAlexander V. Chernikov 		ptent++;
9513a845e10SAlexander V. Chernikov 	}
9523a845e10SAlexander V. Chernikov 
9533a845e10SAlexander V. Chernikov 	error = table_do_modify_record(cmd, oh, tent_buf, count, atomic);
9543a845e10SAlexander V. Chernikov 
9550cba2b28SAlexander V. Chernikov 	quiet = 0;
9560cba2b28SAlexander V. Chernikov 
9573a845e10SAlexander V. Chernikov 	/*
9583a845e10SAlexander V. Chernikov 	 * Compatibility stuff: do not yell on duplicate keys or
9593a845e10SAlexander V. Chernikov 	 * failed deletions.
9603a845e10SAlexander V. Chernikov 	 */
9613a845e10SAlexander V. Chernikov 	if (error == 0 || (error == EEXIST && add != 0) ||
9623a845e10SAlexander V. Chernikov 	    (error == ENOENT && add == 0)) {
9633a845e10SAlexander V. Chernikov 		if (quiet != 0) {
9643a845e10SAlexander V. Chernikov 			if (tent_buf != &tent)
9653a845e10SAlexander V. Chernikov 				free(tent_buf);
9663a845e10SAlexander V. Chernikov 			return;
9673a845e10SAlexander V. Chernikov 		}
9683a845e10SAlexander V. Chernikov 	}
9693a845e10SAlexander V. Chernikov 
9703a845e10SAlexander V. Chernikov 	/* Report results back */
9713a845e10SAlexander V. Chernikov 	ptent = tent_buf;
9723a845e10SAlexander V. Chernikov 	for (i = 0; i < count; ptent++, i++) {
9733a845e10SAlexander V. Chernikov 		ignored = 0;
9743a845e10SAlexander V. Chernikov 		switch (ptent->result) {
9753a845e10SAlexander V. Chernikov 		case IPFW_TR_ADDED:
9763a845e10SAlexander V. Chernikov 			px = "added";
9773a845e10SAlexander V. Chernikov 			break;
9783a845e10SAlexander V. Chernikov 		case IPFW_TR_DELETED:
9793a845e10SAlexander V. Chernikov 			px = "deleted";
9803a845e10SAlexander V. Chernikov 			break;
9813a845e10SAlexander V. Chernikov 		case IPFW_TR_UPDATED:
9823a845e10SAlexander V. Chernikov 			px = "updated";
9833a845e10SAlexander V. Chernikov 			break;
9843a845e10SAlexander V. Chernikov 		case IPFW_TR_LIMIT:
9853a845e10SAlexander V. Chernikov 			px = "limit";
9863a845e10SAlexander V. Chernikov 			ignored = 1;
9873a845e10SAlexander V. Chernikov 			break;
9883a845e10SAlexander V. Chernikov 		case IPFW_TR_ERROR:
9893a845e10SAlexander V. Chernikov 			px = "error";
9903a845e10SAlexander V. Chernikov 			ignored = 1;
9913a845e10SAlexander V. Chernikov 			break;
9923a845e10SAlexander V. Chernikov 		case IPFW_TR_NOTFOUND:
9933a845e10SAlexander V. Chernikov 			px = "notfound";
9943a845e10SAlexander V. Chernikov 			ignored = 1;
9953a845e10SAlexander V. Chernikov 			break;
9963a845e10SAlexander V. Chernikov 		case IPFW_TR_EXISTS:
9973a845e10SAlexander V. Chernikov 			px = "exists";
9983a845e10SAlexander V. Chernikov 			ignored = 1;
9993a845e10SAlexander V. Chernikov 			break;
10003a845e10SAlexander V. Chernikov 		case IPFW_TR_IGNORED:
10013a845e10SAlexander V. Chernikov 			px = "ignored";
10023a845e10SAlexander V. Chernikov 			ignored = 1;
10033a845e10SAlexander V. Chernikov 			break;
10043a845e10SAlexander V. Chernikov 		default:
10053a845e10SAlexander V. Chernikov 			px = "unknown";
10063a845e10SAlexander V. Chernikov 			ignored = 1;
10073a845e10SAlexander V. Chernikov 		}
10083a845e10SAlexander V. Chernikov 
10093a845e10SAlexander V. Chernikov 		if (error != 0 && atomic != 0 && ignored == 0)
10103a845e10SAlexander V. Chernikov 			printf("%s(reverted): ", px);
10113a845e10SAlexander V. Chernikov 		else
10123a845e10SAlexander V. Chernikov 			printf("%s: ", px);
10133a845e10SAlexander V. Chernikov 
10143a845e10SAlexander V. Chernikov 		table_show_entry(&xi, ptent);
10153a845e10SAlexander V. Chernikov 	}
10163a845e10SAlexander V. Chernikov 
10173a845e10SAlexander V. Chernikov 	if (tent_buf != &tent)
10183a845e10SAlexander V. Chernikov 		free(tent_buf);
10193a845e10SAlexander V. Chernikov 
10203a845e10SAlexander V. Chernikov 	if (error == 0)
10214c0c07a5SAlexander V. Chernikov 		return;
102268bde59eSAlexander V. Chernikov 	/* Get real OS error */
102368bde59eSAlexander V. Chernikov 	error = errno;
10244c0c07a5SAlexander V. Chernikov 
10254c0c07a5SAlexander V. Chernikov 	/* Try to provide more human-readable error */
10264c0c07a5SAlexander V. Chernikov 	switch (error) {
10274c0c07a5SAlexander V. Chernikov 	case EEXIST:
10284c0c07a5SAlexander V. Chernikov 		etxt = "record already exists";
10294c0c07a5SAlexander V. Chernikov 		break;
10304c0c07a5SAlexander V. Chernikov 	case EFBIG:
10314c0c07a5SAlexander V. Chernikov 		etxt = "limit hit";
10324c0c07a5SAlexander V. Chernikov 		break;
10334c0c07a5SAlexander V. Chernikov 	case ESRCH:
10344c0c07a5SAlexander V. Chernikov 		etxt = "table not found";
10354c0c07a5SAlexander V. Chernikov 		break;
10364c0c07a5SAlexander V. Chernikov 	case ENOENT:
10374c0c07a5SAlexander V. Chernikov 		etxt = "record not found";
10384c0c07a5SAlexander V. Chernikov 		break;
10394f43138aSAlexander V. Chernikov 	case EACCES:
10404f43138aSAlexander V. Chernikov 		etxt = "table is locked";
10414f43138aSAlexander V. Chernikov 		break;
10424c0c07a5SAlexander V. Chernikov 	default:
10434c0c07a5SAlexander V. Chernikov 		etxt = strerror(error);
10444c0c07a5SAlexander V. Chernikov 	}
10454c0c07a5SAlexander V. Chernikov 
10464c0c07a5SAlexander V. Chernikov 	errx(EX_OSERR, "%s: %s", texterr, etxt);
1047ac35ff17SAlexander V. Chernikov }
1048ac35ff17SAlexander V. Chernikov 
104981d3153dSAlexander V. Chernikov static int
105081d3153dSAlexander V. Chernikov table_do_lookup(ipfw_obj_header *oh, char *key, ipfw_xtable_info *xi,
105181d3153dSAlexander V. Chernikov     ipfw_obj_tentry *xtent)
105281d3153dSAlexander V. Chernikov {
105381d3153dSAlexander V. Chernikov 	char xbuf[sizeof(ipfw_obj_header) + sizeof(ipfw_obj_tentry)];
105481d3153dSAlexander V. Chernikov 	ipfw_obj_tentry *tent;
10550cba2b28SAlexander V. Chernikov 	uint8_t type;
10560cba2b28SAlexander V. Chernikov 	uint32_t vmask;
105781d3153dSAlexander V. Chernikov 	size_t sz;
105881d3153dSAlexander V. Chernikov 
105981d3153dSAlexander V. Chernikov 	memcpy(xbuf, oh, sizeof(*oh));
106081d3153dSAlexander V. Chernikov 	oh = (ipfw_obj_header *)xbuf;
106181d3153dSAlexander V. Chernikov 	tent = (ipfw_obj_tentry *)(oh + 1);
106281d3153dSAlexander V. Chernikov 
106381d3153dSAlexander V. Chernikov 	memset(tent, 0, sizeof(*tent));
106481d3153dSAlexander V. Chernikov 	tent->head.length = sizeof(*tent);
106581d3153dSAlexander V. Chernikov 	tent->idx = 1;
106681d3153dSAlexander V. Chernikov 
10670cba2b28SAlexander V. Chernikov 	tentry_fill_key(oh, tent, key, 0, &type, &vmask, xi);
106881d3153dSAlexander V. Chernikov 	oh->ntlv.type = type;
106981d3153dSAlexander V. Chernikov 
107081d3153dSAlexander V. Chernikov 	sz = sizeof(xbuf);
1071b6462881SAlexander V. Chernikov 	if (do_get3(IP_FW_TABLE_XFIND, &oh->opheader, &sz) != 0)
1072b6462881SAlexander V. Chernikov 		return (errno);
107381d3153dSAlexander V. Chernikov 
107481d3153dSAlexander V. Chernikov 	if (sz < sizeof(xbuf))
107581d3153dSAlexander V. Chernikov 		return (EINVAL);
107681d3153dSAlexander V. Chernikov 
107781d3153dSAlexander V. Chernikov 	*xtent = *tent;
107881d3153dSAlexander V. Chernikov 
107981d3153dSAlexander V. Chernikov 	return (0);
108081d3153dSAlexander V. Chernikov }
108181d3153dSAlexander V. Chernikov 
108281d3153dSAlexander V. Chernikov static void
108381d3153dSAlexander V. Chernikov table_lookup(ipfw_obj_header *oh, int ac, char *av[])
108481d3153dSAlexander V. Chernikov {
108581d3153dSAlexander V. Chernikov 	ipfw_obj_tentry xtent;
108681d3153dSAlexander V. Chernikov 	ipfw_xtable_info xi;
1087914bffb6SAlexander V. Chernikov 	char key[64];
108881d3153dSAlexander V. Chernikov 	int error;
108981d3153dSAlexander V. Chernikov 
109081d3153dSAlexander V. Chernikov 	if (ac == 0)
109181d3153dSAlexander V. Chernikov 		errx(EX_USAGE, "address required");
109281d3153dSAlexander V. Chernikov 
1093914bffb6SAlexander V. Chernikov 	strlcpy(key, *av, sizeof(key));
1094914bffb6SAlexander V. Chernikov 
10953a845e10SAlexander V. Chernikov 	memset(&xi, 0, sizeof(xi));
1096914bffb6SAlexander V. Chernikov 	error = table_do_lookup(oh, key, &xi, &xtent);
109781d3153dSAlexander V. Chernikov 
109881d3153dSAlexander V. Chernikov 	switch (error) {
109981d3153dSAlexander V. Chernikov 	case 0:
110081d3153dSAlexander V. Chernikov 		break;
110181d3153dSAlexander V. Chernikov 	case ESRCH:
110281d3153dSAlexander V. Chernikov 		errx(EX_UNAVAILABLE, "Table %s not found", oh->ntlv.name);
110381d3153dSAlexander V. Chernikov 	case ENOENT:
110481d3153dSAlexander V. Chernikov 		errx(EX_UNAVAILABLE, "Entry %s not found", *av);
110581d3153dSAlexander V. Chernikov 	case ENOTSUP:
110681d3153dSAlexander V. Chernikov 		errx(EX_UNAVAILABLE, "Table %s algo does not support "
110781d3153dSAlexander V. Chernikov 		    "\"lookup\" method", oh->ntlv.name);
110881d3153dSAlexander V. Chernikov 	default:
110981d3153dSAlexander V. Chernikov 		err(EX_OSERR, "getsockopt(IP_FW_TABLE_XFIND)");
111081d3153dSAlexander V. Chernikov 	}
111181d3153dSAlexander V. Chernikov 
111281d3153dSAlexander V. Chernikov 	table_show_entry(&xi, &xtent);
111381d3153dSAlexander V. Chernikov }
1114ac35ff17SAlexander V. Chernikov 
1115ac35ff17SAlexander V. Chernikov static void
1116914bffb6SAlexander V. Chernikov tentry_fill_key_type(char *arg, ipfw_obj_tentry *tentry, uint8_t type,
1117914bffb6SAlexander V. Chernikov     uint8_t tflags)
1118ac35ff17SAlexander V. Chernikov {
1119914bffb6SAlexander V. Chernikov 	char *p, *pp;
1120ac35ff17SAlexander V. Chernikov 	int mask, af;
1121914bffb6SAlexander V. Chernikov 	struct in6_addr *paddr, tmp;
1122914bffb6SAlexander V. Chernikov 	struct tflow_entry *tfe;
1123ac35ff17SAlexander V. Chernikov 	uint32_t key, *pkey;
1124914bffb6SAlexander V. Chernikov 	uint16_t port;
1125914bffb6SAlexander V. Chernikov 	struct protoent *pent;
1126914bffb6SAlexander V. Chernikov 	struct servent *sent;
1127ac35ff17SAlexander V. Chernikov 	int masklen;
1128ac35ff17SAlexander V. Chernikov 
1129ac35ff17SAlexander V. Chernikov 	masklen = 0;
1130ac35ff17SAlexander V. Chernikov 	af = 0;
1131ac35ff17SAlexander V. Chernikov 	paddr = (struct in6_addr *)&tentry->k;
1132ac35ff17SAlexander V. Chernikov 
1133ac35ff17SAlexander V. Chernikov 	switch (type) {
1134c21034b7SAlexander V. Chernikov 	case IPFW_TABLE_ADDR:
1135ac35ff17SAlexander V. Chernikov 		/* Remove / if exists */
1136ac35ff17SAlexander V. Chernikov 		if ((p = strchr(arg, '/')) != NULL) {
1137ac35ff17SAlexander V. Chernikov 			*p = '\0';
1138ac35ff17SAlexander V. Chernikov 			mask = atoi(p + 1);
1139ac35ff17SAlexander V. Chernikov 		}
1140ac35ff17SAlexander V. Chernikov 
1141ac35ff17SAlexander V. Chernikov 		if (inet_pton(AF_INET, arg, paddr) == 1) {
1142ac35ff17SAlexander V. Chernikov 			if (p != NULL && mask > 32)
1143ac35ff17SAlexander V. Chernikov 				errx(EX_DATAERR, "bad IPv4 mask width: %s",
1144ac35ff17SAlexander V. Chernikov 				    p + 1);
1145ac35ff17SAlexander V. Chernikov 
1146ac35ff17SAlexander V. Chernikov 			masklen = p ? mask : 32;
1147ac35ff17SAlexander V. Chernikov 			af = AF_INET;
1148ac35ff17SAlexander V. Chernikov 		} else if (inet_pton(AF_INET6, arg, paddr) == 1) {
1149ac35ff17SAlexander V. Chernikov 			if (IN6_IS_ADDR_V4COMPAT(paddr))
1150ac35ff17SAlexander V. Chernikov 				errx(EX_DATAERR,
1151ac35ff17SAlexander V. Chernikov 				    "Use IPv4 instead of v4-compatible");
1152ac35ff17SAlexander V. Chernikov 			if (p != NULL && mask > 128)
1153ac35ff17SAlexander V. Chernikov 				errx(EX_DATAERR, "bad IPv6 mask width: %s",
1154ac35ff17SAlexander V. Chernikov 				    p + 1);
1155ac35ff17SAlexander V. Chernikov 
1156ac35ff17SAlexander V. Chernikov 			masklen = p ? mask : 128;
1157ac35ff17SAlexander V. Chernikov 			af = AF_INET6;
1158ac35ff17SAlexander V. Chernikov 		} else {
1159ac35ff17SAlexander V. Chernikov 			/* Assume FQDN */
1160ac35ff17SAlexander V. Chernikov 			if (lookup_host(arg, (struct in_addr *)paddr) != 0)
1161ac35ff17SAlexander V. Chernikov 				errx(EX_NOHOST, "hostname ``%s'' unknown", arg);
1162ac35ff17SAlexander V. Chernikov 
1163ac35ff17SAlexander V. Chernikov 			masklen = 32;
1164c21034b7SAlexander V. Chernikov 			type = IPFW_TABLE_ADDR;
1165ac35ff17SAlexander V. Chernikov 			af = AF_INET;
1166ac35ff17SAlexander V. Chernikov 		}
1167ac35ff17SAlexander V. Chernikov 		break;
1168ac35ff17SAlexander V. Chernikov 	case IPFW_TABLE_INTERFACE:
1169ac35ff17SAlexander V. Chernikov 		/* Assume interface name. Copy significant data only */
1170ac35ff17SAlexander V. Chernikov 		mask = MIN(strlen(arg), IF_NAMESIZE - 1);
1171ac35ff17SAlexander V. Chernikov 		memcpy(paddr, arg, mask);
1172ac35ff17SAlexander V. Chernikov 		/* Set mask to exact match */
1173ac35ff17SAlexander V. Chernikov 		masklen = 8 * IF_NAMESIZE;
1174ac35ff17SAlexander V. Chernikov 		break;
1175b23d5de9SAlexander V. Chernikov 	case IPFW_TABLE_NUMBER:
1176ac35ff17SAlexander V. Chernikov 		/* Port or any other key */
1177ac35ff17SAlexander V. Chernikov 		key = strtol(arg, &p, 10);
1178ac35ff17SAlexander V. Chernikov 		if (*p != '\0')
1179ac35ff17SAlexander V. Chernikov 			errx(EX_DATAERR, "Invalid number: %s", arg);
1180ac35ff17SAlexander V. Chernikov 
1181ac35ff17SAlexander V. Chernikov 		pkey = (uint32_t *)paddr;
1182ac35ff17SAlexander V. Chernikov 		*pkey = key;
1183ac35ff17SAlexander V. Chernikov 		masklen = 32;
1184ac35ff17SAlexander V. Chernikov 		break;
1185914bffb6SAlexander V. Chernikov 	case IPFW_TABLE_FLOW:
1186914bffb6SAlexander V. Chernikov 		/* Assume [src-ip][,proto][,src-port][,dst-ip][,dst-port] */
1187914bffb6SAlexander V. Chernikov 		tfe = &tentry->k.flow;
1188914bffb6SAlexander V. Chernikov 		af = 0;
1189914bffb6SAlexander V. Chernikov 
1190914bffb6SAlexander V. Chernikov 		/* Handle <ipv4|ipv6> */
1191914bffb6SAlexander V. Chernikov 		if ((tflags & IPFW_TFFLAG_SRCIP) != 0) {
1192914bffb6SAlexander V. Chernikov 			if ((p = strchr(arg, ',')) != NULL)
1193914bffb6SAlexander V. Chernikov 				*p++ = '\0';
1194914bffb6SAlexander V. Chernikov 			/* Determine family using temporary storage */
1195914bffb6SAlexander V. Chernikov 			if (inet_pton(AF_INET, arg, &tmp) == 1) {
1196914bffb6SAlexander V. Chernikov 				if (af != 0 && af != AF_INET)
1197914bffb6SAlexander V. Chernikov 					errx(EX_DATAERR,
1198914bffb6SAlexander V. Chernikov 					    "Inconsistent address family\n");
1199914bffb6SAlexander V. Chernikov 				af = AF_INET;
1200914bffb6SAlexander V. Chernikov 				memcpy(&tfe->a.a4.sip, &tmp, 4);
1201914bffb6SAlexander V. Chernikov 			} else if (inet_pton(AF_INET6, arg, &tmp) == 1) {
1202914bffb6SAlexander V. Chernikov 				if (af != 0 && af != AF_INET6)
1203914bffb6SAlexander V. Chernikov 					errx(EX_DATAERR,
1204914bffb6SAlexander V. Chernikov 					    "Inconsistent address family\n");
1205914bffb6SAlexander V. Chernikov 				af = AF_INET6;
1206914bffb6SAlexander V. Chernikov 				memcpy(&tfe->a.a6.sip6, &tmp, 16);
1207914bffb6SAlexander V. Chernikov 			}
1208914bffb6SAlexander V. Chernikov 
1209914bffb6SAlexander V. Chernikov 			arg = p;
1210914bffb6SAlexander V. Chernikov 		}
1211914bffb6SAlexander V. Chernikov 
1212914bffb6SAlexander V. Chernikov 		/* Handle <proto-num|proto-name> */
1213914bffb6SAlexander V. Chernikov 		if ((tflags & IPFW_TFFLAG_PROTO) != 0) {
121435df97d0SAlexander V. Chernikov 			if (arg == NULL)
121535df97d0SAlexander V. Chernikov 				errx(EX_DATAERR, "invalid key: proto missing");
1216914bffb6SAlexander V. Chernikov 			if ((p = strchr(arg, ',')) != NULL)
1217914bffb6SAlexander V. Chernikov 				*p++ = '\0';
1218914bffb6SAlexander V. Chernikov 
1219914bffb6SAlexander V. Chernikov 			key = strtol(arg, &pp, 10);
1220914bffb6SAlexander V. Chernikov 			if (*pp != '\0') {
1221914bffb6SAlexander V. Chernikov 				if ((pent = getprotobyname(arg)) == NULL)
1222914bffb6SAlexander V. Chernikov 					errx(EX_DATAERR, "Unknown proto: %s",
1223914bffb6SAlexander V. Chernikov 					    arg);
1224914bffb6SAlexander V. Chernikov 				else
1225914bffb6SAlexander V. Chernikov 					key = pent->p_proto;
1226914bffb6SAlexander V. Chernikov 			}
1227914bffb6SAlexander V. Chernikov 
1228914bffb6SAlexander V. Chernikov 			if (key > 255)
1229914bffb6SAlexander V. Chernikov 				errx(EX_DATAERR, "Bad protocol number: %u",key);
1230914bffb6SAlexander V. Chernikov 
1231914bffb6SAlexander V. Chernikov 			tfe->proto = key;
1232914bffb6SAlexander V. Chernikov 
1233914bffb6SAlexander V. Chernikov 			arg = p;
1234914bffb6SAlexander V. Chernikov 		}
1235914bffb6SAlexander V. Chernikov 
1236914bffb6SAlexander V. Chernikov 		/* Handle <port-num|service-name> */
1237914bffb6SAlexander V. Chernikov 		if ((tflags & IPFW_TFFLAG_SRCPORT) != 0) {
123835df97d0SAlexander V. Chernikov 			if (arg == NULL)
123935df97d0SAlexander V. Chernikov 				errx(EX_DATAERR, "invalid key: src port missing");
1240914bffb6SAlexander V. Chernikov 			if ((p = strchr(arg, ',')) != NULL)
1241914bffb6SAlexander V. Chernikov 				*p++ = '\0';
1242914bffb6SAlexander V. Chernikov 
1243914bffb6SAlexander V. Chernikov 			if ((port = htons(strtol(arg, NULL, 10))) == 0) {
1244914bffb6SAlexander V. Chernikov 				if ((sent = getservbyname(arg, NULL)) == NULL)
1245914bffb6SAlexander V. Chernikov 					errx(EX_DATAERR, "Unknown service: %s",
1246914bffb6SAlexander V. Chernikov 					    arg);
1247914bffb6SAlexander V. Chernikov 				else
1248914bffb6SAlexander V. Chernikov 					key = sent->s_port;
1249914bffb6SAlexander V. Chernikov 			}
1250914bffb6SAlexander V. Chernikov 
1251914bffb6SAlexander V. Chernikov 			tfe->sport = port;
1252914bffb6SAlexander V. Chernikov 
1253914bffb6SAlexander V. Chernikov 			arg = p;
1254914bffb6SAlexander V. Chernikov 		}
1255914bffb6SAlexander V. Chernikov 
1256914bffb6SAlexander V. Chernikov 		/* Handle <ipv4|ipv6>*/
1257914bffb6SAlexander V. Chernikov 		if ((tflags & IPFW_TFFLAG_DSTIP) != 0) {
125835df97d0SAlexander V. Chernikov 			if (arg == NULL)
125935df97d0SAlexander V. Chernikov 				errx(EX_DATAERR, "invalid key: dst ip missing");
1260914bffb6SAlexander V. Chernikov 			if ((p = strchr(arg, ',')) != NULL)
1261914bffb6SAlexander V. Chernikov 				*p++ = '\0';
1262914bffb6SAlexander V. Chernikov 			/* Determine family using temporary storage */
1263914bffb6SAlexander V. Chernikov 			if (inet_pton(AF_INET, arg, &tmp) == 1) {
1264914bffb6SAlexander V. Chernikov 				if (af != 0 && af != AF_INET)
1265914bffb6SAlexander V. Chernikov 					errx(EX_DATAERR,
1266914bffb6SAlexander V. Chernikov 					    "Inconsistent address family");
1267914bffb6SAlexander V. Chernikov 				af = AF_INET;
1268914bffb6SAlexander V. Chernikov 				memcpy(&tfe->a.a4.dip, &tmp, 4);
1269914bffb6SAlexander V. Chernikov 			} else if (inet_pton(AF_INET6, arg, &tmp) == 1) {
1270914bffb6SAlexander V. Chernikov 				if (af != 0 && af != AF_INET6)
1271914bffb6SAlexander V. Chernikov 					errx(EX_DATAERR,
1272914bffb6SAlexander V. Chernikov 					    "Inconsistent address family");
1273914bffb6SAlexander V. Chernikov 				af = AF_INET6;
1274914bffb6SAlexander V. Chernikov 				memcpy(&tfe->a.a6.dip6, &tmp, 16);
1275914bffb6SAlexander V. Chernikov 			}
1276914bffb6SAlexander V. Chernikov 
1277914bffb6SAlexander V. Chernikov 			arg = p;
1278914bffb6SAlexander V. Chernikov 		}
1279914bffb6SAlexander V. Chernikov 
1280914bffb6SAlexander V. Chernikov 		/* Handle <port-num|service-name> */
1281914bffb6SAlexander V. Chernikov 		if ((tflags & IPFW_TFFLAG_DSTPORT) != 0) {
128235df97d0SAlexander V. Chernikov 			if (arg == NULL)
128335df97d0SAlexander V. Chernikov 				errx(EX_DATAERR, "invalid key: dst port missing");
1284914bffb6SAlexander V. Chernikov 			if ((p = strchr(arg, ',')) != NULL)
1285914bffb6SAlexander V. Chernikov 				*p++ = '\0';
1286914bffb6SAlexander V. Chernikov 
1287914bffb6SAlexander V. Chernikov 			if ((port = htons(strtol(arg, NULL, 10))) == 0) {
1288914bffb6SAlexander V. Chernikov 				if ((sent = getservbyname(arg, NULL)) == NULL)
1289914bffb6SAlexander V. Chernikov 					errx(EX_DATAERR, "Unknown service: %s",
1290914bffb6SAlexander V. Chernikov 					    arg);
1291914bffb6SAlexander V. Chernikov 				else
1292914bffb6SAlexander V. Chernikov 					key = sent->s_port;
1293914bffb6SAlexander V. Chernikov 			}
1294914bffb6SAlexander V. Chernikov 
1295914bffb6SAlexander V. Chernikov 			tfe->dport = port;
1296914bffb6SAlexander V. Chernikov 
1297914bffb6SAlexander V. Chernikov 			arg = p;
1298914bffb6SAlexander V. Chernikov 		}
1299914bffb6SAlexander V. Chernikov 
1300914bffb6SAlexander V. Chernikov 		tfe->af = af;
1301914bffb6SAlexander V. Chernikov 
1302914bffb6SAlexander V. Chernikov 		break;
1303914bffb6SAlexander V. Chernikov 
1304ac35ff17SAlexander V. Chernikov 	default:
1305ac35ff17SAlexander V. Chernikov 		errx(EX_DATAERR, "Unsupported table type: %d", type);
1306ac35ff17SAlexander V. Chernikov 	}
1307ac35ff17SAlexander V. Chernikov 
1308ac35ff17SAlexander V. Chernikov 	tentry->subtype = af;
1309ac35ff17SAlexander V. Chernikov 	tentry->masklen = masklen;
1310ac35ff17SAlexander V. Chernikov }
1311ac35ff17SAlexander V. Chernikov 
1312ac35ff17SAlexander V. Chernikov static void
1313ac35ff17SAlexander V. Chernikov tentry_fill_key(ipfw_obj_header *oh, ipfw_obj_tentry *tent, char *key,
13140cba2b28SAlexander V. Chernikov     int add, uint8_t *ptype, uint32_t *pvmask, ipfw_xtable_info *xi)
1315ac35ff17SAlexander V. Chernikov {
13160cba2b28SAlexander V. Chernikov 	uint8_t type, tflags;
13170cba2b28SAlexander V. Chernikov 	uint32_t vmask;
1318ac35ff17SAlexander V. Chernikov 	int error;
1319db785d31SAlexander V. Chernikov 	char *del;
1320ac35ff17SAlexander V. Chernikov 
1321ac35ff17SAlexander V. Chernikov 	type = 0;
1322914bffb6SAlexander V. Chernikov 	tflags = 0;
13230cba2b28SAlexander V. Chernikov 	vmask = 0;
1324ac35ff17SAlexander V. Chernikov 
13253a845e10SAlexander V. Chernikov 	if (xi->tablename[0] == '\0')
132681d3153dSAlexander V. Chernikov 		error = table_get_info(oh, xi);
13273a845e10SAlexander V. Chernikov 	else
13283a845e10SAlexander V. Chernikov 		error = 0;
132981d3153dSAlexander V. Chernikov 
133081d3153dSAlexander V. Chernikov 	if (error == 0) {
133181d3153dSAlexander V. Chernikov 		/* Table found. */
133281d3153dSAlexander V. Chernikov 		type = xi->type;
1333914bffb6SAlexander V. Chernikov 		tflags = xi->tflags;
13340cba2b28SAlexander V. Chernikov 		vmask = xi->vmask;
133581d3153dSAlexander V. Chernikov 	} else {
133681d3153dSAlexander V. Chernikov 		if (error != ESRCH)
133781d3153dSAlexander V. Chernikov 			errx(EX_OSERR, "Error requesting table %s info",
133881d3153dSAlexander V. Chernikov 			    oh->ntlv.name);
133935df97d0SAlexander V. Chernikov 		if (add == 0)
134035df97d0SAlexander V. Chernikov 			errx(EX_DATAERR, "Table %s does not exist",
134135df97d0SAlexander V. Chernikov 			    oh->ntlv.name);
1342ac35ff17SAlexander V. Chernikov 		/*
134381d3153dSAlexander V. Chernikov 		 * Table does not exist.
1344c21034b7SAlexander V. Chernikov 		 * Compability layer: try to interpret data as ADDR
134581d3153dSAlexander V. Chernikov 		 * before failing.
1346ac35ff17SAlexander V. Chernikov 		 */
1347db785d31SAlexander V. Chernikov 		if ((del = strchr(key, '/')) != NULL)
1348db785d31SAlexander V. Chernikov 			*del = '\0';
1349ac35ff17SAlexander V. Chernikov 		if (inet_pton(AF_INET, key, &tent->k.addr6) == 1 ||
1350ac35ff17SAlexander V. Chernikov 		    inet_pton(AF_INET6, key, &tent->k.addr6) == 1) {
1351ac35ff17SAlexander V. Chernikov 			/* OK Prepare and send */
1352c21034b7SAlexander V. Chernikov 			type = IPFW_TABLE_ADDR;
13530cba2b28SAlexander V. Chernikov 			vmask = IPFW_VTYPE_LEGACY;
135481d3153dSAlexander V. Chernikov 		} else {
135581d3153dSAlexander V. Chernikov 			/* Inknown key */
135681d3153dSAlexander V. Chernikov 			errx(EX_USAGE, "Table %s does not exist, cannot guess "
1357db785d31SAlexander V. Chernikov 			    "key '%s' type", oh->ntlv.name, key);
135881d3153dSAlexander V. Chernikov 		}
1359db785d31SAlexander V. Chernikov 		if (del != NULL)
1360db785d31SAlexander V. Chernikov 			*del = '/';
1361ac35ff17SAlexander V. Chernikov 	}
1362ac35ff17SAlexander V. Chernikov 
1363914bffb6SAlexander V. Chernikov 	tentry_fill_key_type(key, tent, type, tflags);
1364ac35ff17SAlexander V. Chernikov 
1365ac35ff17SAlexander V. Chernikov 	*ptype = type;
13660cba2b28SAlexander V. Chernikov 	*pvmask = vmask;
13670cba2b28SAlexander V. Chernikov }
13680cba2b28SAlexander V. Chernikov 
13690cba2b28SAlexander V. Chernikov static void
13700cba2b28SAlexander V. Chernikov set_legacy_value(uint32_t val, ipfw_table_value *v)
13710cba2b28SAlexander V. Chernikov {
13720cba2b28SAlexander V. Chernikov 	v->tag = val;
13730cba2b28SAlexander V. Chernikov 	v->pipe = val;
13740cba2b28SAlexander V. Chernikov 	v->divert = val;
13750cba2b28SAlexander V. Chernikov 	v->skipto = val;
13760cba2b28SAlexander V. Chernikov 	v->netgraph = val;
13770cba2b28SAlexander V. Chernikov 	v->fib = val;
13780cba2b28SAlexander V. Chernikov 	v->nat = val;
13790cba2b28SAlexander V. Chernikov 	v->nh4 = val;
13800cba2b28SAlexander V. Chernikov 	v->dscp = (uint8_t)val;
13810cba2b28SAlexander V. Chernikov 	v->limit = val;
1382ac35ff17SAlexander V. Chernikov }
1383ac35ff17SAlexander V. Chernikov 
1384ac35ff17SAlexander V. Chernikov static void
1385ac35ff17SAlexander V. Chernikov tentry_fill_value(ipfw_obj_header *oh, ipfw_obj_tentry *tent, char *arg,
13860cba2b28SAlexander V. Chernikov     uint8_t type, uint32_t vmask)
1387ac35ff17SAlexander V. Chernikov {
1388*2530ed9eSAndrey V. Elsukov 	struct addrinfo hints, *res;
13890cba2b28SAlexander V. Chernikov 	uint32_t a4, flag, val, vm;
13900cba2b28SAlexander V. Chernikov 	ipfw_table_value *v;
13910cba2b28SAlexander V. Chernikov 	uint32_t i;
1392d91c61c3SAlexander V. Chernikov 	int dval;
13930cba2b28SAlexander V. Chernikov 	char *comma, *e, *etype, *n, *p;
1394ac35ff17SAlexander V. Chernikov 
13950cba2b28SAlexander V. Chernikov 	v = &tent->v.value;
13960cba2b28SAlexander V. Chernikov 	vm = vmask;
13970cba2b28SAlexander V. Chernikov 
13980cba2b28SAlexander V. Chernikov 	/* Compat layer: keep old behavior for legacy value types */
13990cba2b28SAlexander V. Chernikov 	if (vmask == IPFW_VTYPE_LEGACY) {
1400adf3b2b9SAlexander V. Chernikov 		/* Try to interpret as number first */
14010cba2b28SAlexander V. Chernikov 		val = strtoul(arg, &p, 0);
14020cba2b28SAlexander V. Chernikov 		if (*p == '\0') {
14030cba2b28SAlexander V. Chernikov 			set_legacy_value(val, v);
1404adf3b2b9SAlexander V. Chernikov 			return;
14050cba2b28SAlexander V. Chernikov 		}
1406adf3b2b9SAlexander V. Chernikov 		if (inet_pton(AF_INET, arg, &val) == 1) {
14070cba2b28SAlexander V. Chernikov 			set_legacy_value(ntohl(val), v);
1408adf3b2b9SAlexander V. Chernikov 			return;
1409adf3b2b9SAlexander V. Chernikov 		}
1410adf3b2b9SAlexander V. Chernikov 		/* Try hostname */
14110cba2b28SAlexander V. Chernikov 		if (lookup_host(arg, (struct in_addr *)&val) == 0) {
14120cba2b28SAlexander V. Chernikov 			set_legacy_value(val, v);
1413adf3b2b9SAlexander V. Chernikov 			return;
14140cba2b28SAlexander V. Chernikov 		}
1415adf3b2b9SAlexander V. Chernikov 		errx(EX_OSERR, "Unable to parse value %s", arg);
14160cba2b28SAlexander V. Chernikov 	}
14170cba2b28SAlexander V. Chernikov 
14180cba2b28SAlexander V. Chernikov 	/*
14190cba2b28SAlexander V. Chernikov 	 * Shorthands: handle single value if vmask consists
14200cba2b28SAlexander V. Chernikov 	 * of numbers only. e.g.:
14210cba2b28SAlexander V. Chernikov 	 * vmask = "fib,skipto" -> treat input "1" as "1,1"
14220cba2b28SAlexander V. Chernikov 	 */
14230cba2b28SAlexander V. Chernikov 
14240cba2b28SAlexander V. Chernikov 	n = arg;
14250cba2b28SAlexander V. Chernikov 	etype = NULL;
14260cba2b28SAlexander V. Chernikov 	for (i = 1; i < (1 << 31); i *= 2) {
14270cba2b28SAlexander V. Chernikov 		if ((flag = (vmask & i)) == 0)
14280cba2b28SAlexander V. Chernikov 			continue;
14290cba2b28SAlexander V. Chernikov 		vmask &= ~flag;
14300cba2b28SAlexander V. Chernikov 
14310cba2b28SAlexander V. Chernikov 		if ((comma = strchr(n, ',')) != NULL)
14320cba2b28SAlexander V. Chernikov 			*comma = '\0';
14330cba2b28SAlexander V. Chernikov 
14340cba2b28SAlexander V. Chernikov 		switch (flag) {
14350cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_TAG:
14360cba2b28SAlexander V. Chernikov 			v->tag = strtol(n, &e, 10);
14370cba2b28SAlexander V. Chernikov 			if (*e != '\0')
14380cba2b28SAlexander V. Chernikov 				etype = "tag";
1439ac35ff17SAlexander V. Chernikov 			break;
14400cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_PIPE:
14410cba2b28SAlexander V. Chernikov 			v->pipe = strtol(n, &e, 10);
14420cba2b28SAlexander V. Chernikov 			if (*e != '\0')
14430cba2b28SAlexander V. Chernikov 				etype = "pipe";
1444ac35ff17SAlexander V. Chernikov 			break;
14450cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_DIVERT:
14460cba2b28SAlexander V. Chernikov 			v->divert = strtol(n, &e, 10);
14470cba2b28SAlexander V. Chernikov 			if (*e != '\0')
14480cba2b28SAlexander V. Chernikov 				etype = "divert";
14490cba2b28SAlexander V. Chernikov 			break;
14500cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_SKIPTO:
14510cba2b28SAlexander V. Chernikov 			v->skipto = strtol(n, &e, 10);
14520cba2b28SAlexander V. Chernikov 			if (*e != '\0')
14530cba2b28SAlexander V. Chernikov 				etype = "skipto";
14540cba2b28SAlexander V. Chernikov 			break;
14550cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_NETGRAPH:
14560cba2b28SAlexander V. Chernikov 			v->netgraph = strtol(n, &e, 10);
14570cba2b28SAlexander V. Chernikov 			if (*e != '\0')
14580cba2b28SAlexander V. Chernikov 				etype = "netgraph";
14590cba2b28SAlexander V. Chernikov 			break;
14600cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_FIB:
14610cba2b28SAlexander V. Chernikov 			v->fib = strtol(n, &e, 10);
14620cba2b28SAlexander V. Chernikov 			if (*e != '\0')
14630cba2b28SAlexander V. Chernikov 				etype = "fib";
14640cba2b28SAlexander V. Chernikov 			break;
14650cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_NAT:
14660cba2b28SAlexander V. Chernikov 			v->nat = strtol(n, &e, 10);
14670cba2b28SAlexander V. Chernikov 			if (*e != '\0')
14680cba2b28SAlexander V. Chernikov 				etype = "nat";
14690cba2b28SAlexander V. Chernikov 			break;
14700cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_LIMIT:
14710cba2b28SAlexander V. Chernikov 			v->limit = strtol(n, &e, 10);
14720cba2b28SAlexander V. Chernikov 			if (*e != '\0')
14730cba2b28SAlexander V. Chernikov 				etype = "limit";
14740cba2b28SAlexander V. Chernikov 			break;
14750cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_NH4:
14760cba2b28SAlexander V. Chernikov 			if (strchr(n, '.') != NULL &&
14770cba2b28SAlexander V. Chernikov 			    inet_pton(AF_INET, n, &a4) == 1) {
14780cba2b28SAlexander V. Chernikov 				v->nh4 = ntohl(a4);
14790cba2b28SAlexander V. Chernikov 				break;
14800cba2b28SAlexander V. Chernikov 			}
14810cba2b28SAlexander V. Chernikov 			if (lookup_host(n, (struct in_addr *)&v->nh4) == 0)
14820cba2b28SAlexander V. Chernikov 				break;
14830cba2b28SAlexander V. Chernikov 			etype = "ipv4";
1484ac35ff17SAlexander V. Chernikov 			break;
1485ac35ff17SAlexander V. Chernikov 		case IPFW_VTYPE_DSCP:
14860cba2b28SAlexander V. Chernikov 			if (isalpha(*n)) {
1487d91c61c3SAlexander V. Chernikov 				if ((dval = match_token(f_ipdscp, n)) != -1) {
1488d91c61c3SAlexander V. Chernikov 					v->dscp = dval;
1489ac35ff17SAlexander V. Chernikov 					break;
1490d91c61c3SAlexander V. Chernikov 				} else
14910cba2b28SAlexander V. Chernikov 					etype = "DSCP code";
14920cba2b28SAlexander V. Chernikov 			} else {
14930cba2b28SAlexander V. Chernikov 				v->dscp = strtol(n, &e, 10);
14940cba2b28SAlexander V. Chernikov 				if (v->dscp > 63 || *e != '\0')
14950cba2b28SAlexander V. Chernikov 					etype = "DSCP value";
1496ac35ff17SAlexander V. Chernikov 			}
14970cba2b28SAlexander V. Chernikov 			break;
14980cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_NH6:
1499*2530ed9eSAndrey V. Elsukov 			if (strchr(n, ':') != NULL) {
1500*2530ed9eSAndrey V. Elsukov 				memset(&hints, 0, sizeof(hints));
1501*2530ed9eSAndrey V. Elsukov 				hints.ai_family = AF_INET6;
1502*2530ed9eSAndrey V. Elsukov 				hints.ai_flags = AI_NUMERICHOST;
1503*2530ed9eSAndrey V. Elsukov 				if (getaddrinfo(n, NULL, &hints, &res) == 0) {
1504*2530ed9eSAndrey V. Elsukov 					v->nh6 = ((struct sockaddr_in6 *)
1505*2530ed9eSAndrey V. Elsukov 					    res->ai_addr)->sin6_addr;
1506*2530ed9eSAndrey V. Elsukov 					v->zoneid = ((struct sockaddr_in6 *)
1507*2530ed9eSAndrey V. Elsukov 					    res->ai_addr)->sin6_scope_id;
1508*2530ed9eSAndrey V. Elsukov 					freeaddrinfo(res);
15090cba2b28SAlexander V. Chernikov 					break;
1510*2530ed9eSAndrey V. Elsukov 				}
1511*2530ed9eSAndrey V. Elsukov 			}
15120cba2b28SAlexander V. Chernikov 			etype = "ipv6";
15130cba2b28SAlexander V. Chernikov 			break;
15140cba2b28SAlexander V. Chernikov 		}
15150cba2b28SAlexander V. Chernikov 
15160cba2b28SAlexander V. Chernikov 		if (etype != NULL)
15170cba2b28SAlexander V. Chernikov 			errx(EX_USAGE, "Unable to parse %s as %s", n, etype);
15180cba2b28SAlexander V. Chernikov 
15190cba2b28SAlexander V. Chernikov 		if (comma != NULL)
15200cba2b28SAlexander V. Chernikov 			*comma++ = ',';
15210cba2b28SAlexander V. Chernikov 
15220cba2b28SAlexander V. Chernikov 		if ((n = comma) != NULL)
15230cba2b28SAlexander V. Chernikov 			continue;
15240cba2b28SAlexander V. Chernikov 
15250cba2b28SAlexander V. Chernikov 		/* End of input. */
15260cba2b28SAlexander V. Chernikov 		if (vmask != 0)
15270cba2b28SAlexander V. Chernikov 			errx(EX_USAGE, "Not enough fields inside value");
15280cba2b28SAlexander V. Chernikov 	}
1529ac35ff17SAlexander V. Chernikov }
1530f1220db8SAlexander V. Chernikov 
1531f1220db8SAlexander V. Chernikov /*
1532f1220db8SAlexander V. Chernikov  * Compare table names.
1533f1220db8SAlexander V. Chernikov  * Honor number comparison.
1534f1220db8SAlexander V. Chernikov  */
1535f1220db8SAlexander V. Chernikov static int
1536f1220db8SAlexander V. Chernikov tablename_cmp(const void *a, const void *b)
1537f1220db8SAlexander V. Chernikov {
1538f1220db8SAlexander V. Chernikov 	ipfw_xtable_info *ia, *ib;
1539f1220db8SAlexander V. Chernikov 
1540f1220db8SAlexander V. Chernikov 	ia = (ipfw_xtable_info *)a;
1541f1220db8SAlexander V. Chernikov 	ib = (ipfw_xtable_info *)b;
1542f1220db8SAlexander V. Chernikov 
154368394ec8SAlexander V. Chernikov 	return (stringnum_cmp(ia->tablename, ib->tablename));
1544f1220db8SAlexander V. Chernikov }
1545f1220db8SAlexander V. Chernikov 
1546f1220db8SAlexander V. Chernikov /*
1547f1220db8SAlexander V. Chernikov  * Retrieves table list from kernel,
1548f1220db8SAlexander V. Chernikov  * optionally sorts it and calls requested function for each table.
1549f1220db8SAlexander V. Chernikov  * Returns 0 on success.
1550f1220db8SAlexander V. Chernikov  */
1551f1220db8SAlexander V. Chernikov static int
1552f1220db8SAlexander V. Chernikov tables_foreach(table_cb_t *f, void *arg, int sort)
1553f1220db8SAlexander V. Chernikov {
155428ea4fa3SAlexander V. Chernikov 	ipfw_obj_lheader *olh;
1555f1220db8SAlexander V. Chernikov 	ipfw_xtable_info *info;
1556f1220db8SAlexander V. Chernikov 	size_t sz;
1557f1220db8SAlexander V. Chernikov 	int i, error;
1558f1220db8SAlexander V. Chernikov 
155928ea4fa3SAlexander V. Chernikov 	/* Start with reasonable default */
156028ea4fa3SAlexander V. Chernikov 	sz = sizeof(*olh) + 16 * sizeof(ipfw_xtable_info);
1561f1220db8SAlexander V. Chernikov 
156228ea4fa3SAlexander V. Chernikov 	for (;;) {
1563f1220db8SAlexander V. Chernikov 		if ((olh = calloc(1, sz)) == NULL)
1564f1220db8SAlexander V. Chernikov 			return (ENOMEM);
1565f1220db8SAlexander V. Chernikov 
1566f1220db8SAlexander V. Chernikov 		olh->size = sz;
1567b6462881SAlexander V. Chernikov 		if (do_get3(IP_FW_TABLES_XLIST, &olh->opheader, &sz) != 0) {
156828ea4fa3SAlexander V. Chernikov 			sz = olh->size;
156968bde59eSAlexander V. Chernikov 			free(olh);
157068bde59eSAlexander V. Chernikov 			if (errno != ENOMEM)
1571b6462881SAlexander V. Chernikov 				return (errno);
157268bde59eSAlexander V. Chernikov 			continue;
1573f1220db8SAlexander V. Chernikov 		}
1574f1220db8SAlexander V. Chernikov 
1575f1220db8SAlexander V. Chernikov 		if (sort != 0)
1576f1220db8SAlexander V. Chernikov 			qsort(olh + 1, olh->count, olh->objsize, tablename_cmp);
1577f1220db8SAlexander V. Chernikov 
1578f1220db8SAlexander V. Chernikov 		info = (ipfw_xtable_info *)(olh + 1);
1579f1220db8SAlexander V. Chernikov 		for (i = 0; i < olh->count; i++) {
1580f1220db8SAlexander V. Chernikov 			error = f(info, arg); /* Ignore errors for now */
1581f1220db8SAlexander V. Chernikov 			info = (ipfw_xtable_info *)((caddr_t)info + olh->objsize);
1582f1220db8SAlexander V. Chernikov 		}
1583f1220db8SAlexander V. Chernikov 
1584f1220db8SAlexander V. Chernikov 		free(olh);
158528ea4fa3SAlexander V. Chernikov 		break;
158628ea4fa3SAlexander V. Chernikov 	}
1587f1220db8SAlexander V. Chernikov 
1588f1220db8SAlexander V. Chernikov 	return (0);
1589f1220db8SAlexander V. Chernikov }
1590f1220db8SAlexander V. Chernikov 
159128ea4fa3SAlexander V. Chernikov 
1592f1220db8SAlexander V. Chernikov /*
1593f1220db8SAlexander V. Chernikov  * Retrieves all entries for given table @i in
1594720ee730SAlexander V. Chernikov  * eXtended format. Allocate buffer large enough
1595720ee730SAlexander V. Chernikov  * to store result. Called needs to free it later.
1596f1220db8SAlexander V. Chernikov  *
1597f1220db8SAlexander V. Chernikov  * Returns 0 on success.
1598f1220db8SAlexander V. Chernikov  */
1599f1220db8SAlexander V. Chernikov static int
1600720ee730SAlexander V. Chernikov table_do_get_list(ipfw_xtable_info *i, ipfw_obj_header **poh)
1601f1220db8SAlexander V. Chernikov {
1602720ee730SAlexander V. Chernikov 	ipfw_obj_header *oh;
1603f1220db8SAlexander V. Chernikov 	size_t sz;
1604b6462881SAlexander V. Chernikov 	int c;
1605f1220db8SAlexander V. Chernikov 
160681d3153dSAlexander V. Chernikov 	sz = 0;
1607720ee730SAlexander V. Chernikov 	oh = NULL;
1608720ee730SAlexander V. Chernikov 	for (c = 0; c < 8; c++) {
160981d3153dSAlexander V. Chernikov 		if (sz < i->size)
1610720ee730SAlexander V. Chernikov 			sz = i->size + 44;
1611720ee730SAlexander V. Chernikov 		if (oh != NULL)
1612720ee730SAlexander V. Chernikov 			free(oh);
1613720ee730SAlexander V. Chernikov 		if ((oh = calloc(1, sz)) == NULL)
1614720ee730SAlexander V. Chernikov 			continue;
1615720ee730SAlexander V. Chernikov 		table_fill_objheader(oh, i);
1616d3a4f924SAlexander V. Chernikov 		oh->opheader.version = 1; /* Current version */
1617b6462881SAlexander V. Chernikov 		if (do_get3(IP_FW_TABLE_XLIST, &oh->opheader, &sz) == 0) {
1618720ee730SAlexander V. Chernikov 			*poh = oh;
1619720ee730SAlexander V. Chernikov 			return (0);
162081d3153dSAlexander V. Chernikov 		}
1621f1220db8SAlexander V. Chernikov 
1622b6462881SAlexander V. Chernikov 		if (errno != ENOMEM)
1623720ee730SAlexander V. Chernikov 			break;
1624720ee730SAlexander V. Chernikov 	}
1625720ee730SAlexander V. Chernikov 	free(oh);
1626720ee730SAlexander V. Chernikov 
1627b6462881SAlexander V. Chernikov 	return (errno);
1628f1220db8SAlexander V. Chernikov }
1629f1220db8SAlexander V. Chernikov 
1630f1220db8SAlexander V. Chernikov /*
1631f1220db8SAlexander V. Chernikov  * Shows all entries from @oh in human-readable format
1632f1220db8SAlexander V. Chernikov  */
1633f1220db8SAlexander V. Chernikov static void
1634f1220db8SAlexander V. Chernikov table_show_list(ipfw_obj_header *oh, int need_header)
1635f1220db8SAlexander V. Chernikov {
163681d3153dSAlexander V. Chernikov 	ipfw_obj_tentry *tent;
163781d3153dSAlexander V. Chernikov 	uint32_t count;
1638f1220db8SAlexander V. Chernikov 	ipfw_xtable_info *i;
1639f1220db8SAlexander V. Chernikov 
1640f1220db8SAlexander V. Chernikov 	i = (ipfw_xtable_info *)(oh + 1);
164181d3153dSAlexander V. Chernikov 	tent = (ipfw_obj_tentry *)(i + 1);
1642f1220db8SAlexander V. Chernikov 
1643f1220db8SAlexander V. Chernikov 	if (need_header)
1644f1220db8SAlexander V. Chernikov 		printf("--- table(%s), set(%u) ---\n", i->tablename, i->set);
1645f1220db8SAlexander V. Chernikov 
1646f1220db8SAlexander V. Chernikov 	count = i->count;
1647f1220db8SAlexander V. Chernikov 	while (count > 0) {
164881d3153dSAlexander V. Chernikov 		table_show_entry(i, tent);
164981d3153dSAlexander V. Chernikov 		tent = (ipfw_obj_tentry *)((caddr_t)tent + tent->head.length);
165081d3153dSAlexander V. Chernikov 		count--;
165181d3153dSAlexander V. Chernikov 	}
165281d3153dSAlexander V. Chernikov }
165381d3153dSAlexander V. Chernikov 
165481d3153dSAlexander V. Chernikov static void
16550cba2b28SAlexander V. Chernikov table_show_value(char *buf, size_t bufsize, ipfw_table_value *v,
16560cba2b28SAlexander V. Chernikov     uint32_t vmask, int print_ip)
16570cba2b28SAlexander V. Chernikov {
1658*2530ed9eSAndrey V. Elsukov 	char abuf[INET6_ADDRSTRLEN + IF_NAMESIZE + 2];
1659*2530ed9eSAndrey V. Elsukov 	struct sockaddr_in6 sa6;
16600cba2b28SAlexander V. Chernikov 	uint32_t flag, i, l;
16610cba2b28SAlexander V. Chernikov 	size_t sz;
16620cba2b28SAlexander V. Chernikov 	struct in_addr a4;
16630cba2b28SAlexander V. Chernikov 
16640cba2b28SAlexander V. Chernikov 	sz = bufsize;
16650cba2b28SAlexander V. Chernikov 
16660cba2b28SAlexander V. Chernikov 	/*
16670cba2b28SAlexander V. Chernikov 	 * Some shorthands for printing values:
16680cba2b28SAlexander V. Chernikov 	 * legacy assumes all values are equal, so keep the first one.
16690cba2b28SAlexander V. Chernikov 	 */
16700cba2b28SAlexander V. Chernikov 	if (vmask == IPFW_VTYPE_LEGACY) {
16710cba2b28SAlexander V. Chernikov 		if (print_ip != 0) {
16720cba2b28SAlexander V. Chernikov 			flag = htonl(v->tag);
16730cba2b28SAlexander V. Chernikov 			inet_ntop(AF_INET, &flag, buf, sz);
16740cba2b28SAlexander V. Chernikov 		} else
16750cba2b28SAlexander V. Chernikov 			snprintf(buf, sz, "%u", v->tag);
16760cba2b28SAlexander V. Chernikov 		return;
16770cba2b28SAlexander V. Chernikov 	}
16780cba2b28SAlexander V. Chernikov 
16790cba2b28SAlexander V. Chernikov 	for (i = 1; i < (1 << 31); i *= 2) {
16800cba2b28SAlexander V. Chernikov 		if ((flag = (vmask & i)) == 0)
16810cba2b28SAlexander V. Chernikov 			continue;
16820cba2b28SAlexander V. Chernikov 		l = 0;
16830cba2b28SAlexander V. Chernikov 
16840cba2b28SAlexander V. Chernikov 		switch (flag) {
16850cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_TAG:
16860cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%u,", v->tag);
16870cba2b28SAlexander V. Chernikov 			break;
16880cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_PIPE:
16890cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%u,", v->pipe);
16900cba2b28SAlexander V. Chernikov 			break;
16910cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_DIVERT:
16920cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%d,", v->divert);
16930cba2b28SAlexander V. Chernikov 			break;
16940cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_SKIPTO:
16950cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%d,", v->skipto);
16960cba2b28SAlexander V. Chernikov 			break;
16970cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_NETGRAPH:
16980cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%u,", v->netgraph);
16990cba2b28SAlexander V. Chernikov 			break;
17000cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_FIB:
17010cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%u,", v->fib);
17020cba2b28SAlexander V. Chernikov 			break;
17030cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_NAT:
17040cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%u,", v->nat);
17050cba2b28SAlexander V. Chernikov 			break;
17060cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_LIMIT:
17070cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%u,", v->limit);
17080cba2b28SAlexander V. Chernikov 			break;
17090cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_NH4:
17100cba2b28SAlexander V. Chernikov 			a4.s_addr = htonl(v->nh4);
17110cba2b28SAlexander V. Chernikov 			inet_ntop(AF_INET, &a4, abuf, sizeof(abuf));
17120cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%s,", abuf);
17130cba2b28SAlexander V. Chernikov 			break;
17140cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_DSCP:
17150cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%d,", v->dscp);
17160cba2b28SAlexander V. Chernikov 			break;
17170cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_NH6:
1718*2530ed9eSAndrey V. Elsukov 			sa6.sin6_family = AF_INET6;
1719*2530ed9eSAndrey V. Elsukov 			sa6.sin6_len = sizeof(sa6);
1720*2530ed9eSAndrey V. Elsukov 			sa6.sin6_addr = v->nh6;
1721*2530ed9eSAndrey V. Elsukov 			sa6.sin6_port = 0;
1722*2530ed9eSAndrey V. Elsukov 			sa6.sin6_scope_id = v->zoneid;
1723*2530ed9eSAndrey V. Elsukov 			if (getnameinfo((const struct sockaddr *)&sa6,
1724*2530ed9eSAndrey V. Elsukov 			    sa6.sin6_len, abuf, sizeof(abuf), NULL, 0,
1725*2530ed9eSAndrey V. Elsukov 			    NI_NUMERICHOST) == 0)
17260cba2b28SAlexander V. Chernikov 				l = snprintf(buf, sz, "%s,", abuf);
17270cba2b28SAlexander V. Chernikov 			break;
17280cba2b28SAlexander V. Chernikov 		}
17290cba2b28SAlexander V. Chernikov 
17300cba2b28SAlexander V. Chernikov 		buf += l;
17310cba2b28SAlexander V. Chernikov 		sz -= l;
17320cba2b28SAlexander V. Chernikov 	}
17330cba2b28SAlexander V. Chernikov 
17340cba2b28SAlexander V. Chernikov 	if (sz != bufsize)
17350cba2b28SAlexander V. Chernikov 		*(buf - 1) = '\0';
17360cba2b28SAlexander V. Chernikov }
17370cba2b28SAlexander V. Chernikov 
17380cba2b28SAlexander V. Chernikov static void
173981d3153dSAlexander V. Chernikov table_show_entry(ipfw_xtable_info *i, ipfw_obj_tentry *tent)
174081d3153dSAlexander V. Chernikov {
17410cba2b28SAlexander V. Chernikov 	char *comma, tbuf[128], pval[128];
1742914bffb6SAlexander V. Chernikov 	void *paddr;
1743914bffb6SAlexander V. Chernikov 	struct tflow_entry *tfe;
174481d3153dSAlexander V. Chernikov 
17450cba2b28SAlexander V. Chernikov 	table_show_value(pval, sizeof(pval), &tent->v.value, i->vmask,
17460cba2b28SAlexander V. Chernikov 	    co.do_value_as_ip);
1747914bffb6SAlexander V. Chernikov 
1748f1220db8SAlexander V. Chernikov 	switch (i->type) {
1749c21034b7SAlexander V. Chernikov 	case IPFW_TABLE_ADDR:
1750f1220db8SAlexander V. Chernikov 		/* IPv4 or IPv6 prefixes */
175181d3153dSAlexander V. Chernikov 		inet_ntop(tent->subtype, &tent->k, tbuf, sizeof(tbuf));
1752914bffb6SAlexander V. Chernikov 		printf("%s/%u %s\n", tbuf, tent->masklen, pval);
1753f1220db8SAlexander V. Chernikov 		break;
1754f1220db8SAlexander V. Chernikov 	case IPFW_TABLE_INTERFACE:
1755f1220db8SAlexander V. Chernikov 		/* Interface names */
1756914bffb6SAlexander V. Chernikov 		printf("%s %s\n", tent->k.iface, pval);
1757b23d5de9SAlexander V. Chernikov 		break;
1758b23d5de9SAlexander V. Chernikov 	case IPFW_TABLE_NUMBER:
1759b23d5de9SAlexander V. Chernikov 		/* numbers */
1760914bffb6SAlexander V. Chernikov 		printf("%u %s\n", tent->k.key, pval);
1761b23d5de9SAlexander V. Chernikov 		break;
1762914bffb6SAlexander V. Chernikov 	case IPFW_TABLE_FLOW:
1763914bffb6SAlexander V. Chernikov 		/* flows */
1764914bffb6SAlexander V. Chernikov 		tfe = &tent->k.flow;
1765914bffb6SAlexander V. Chernikov 		comma = "";
1766914bffb6SAlexander V. Chernikov 
1767914bffb6SAlexander V. Chernikov 		if ((i->tflags & IPFW_TFFLAG_SRCIP) != 0) {
1768914bffb6SAlexander V. Chernikov 			if (tfe->af == AF_INET)
1769914bffb6SAlexander V. Chernikov 				paddr = &tfe->a.a4.sip;
1770914bffb6SAlexander V. Chernikov 			else
1771914bffb6SAlexander V. Chernikov 				paddr = &tfe->a.a6.sip6;
1772914bffb6SAlexander V. Chernikov 
1773914bffb6SAlexander V. Chernikov 			inet_ntop(tfe->af, paddr, tbuf, sizeof(tbuf));
1774914bffb6SAlexander V. Chernikov 			printf("%s%s", comma, tbuf);
1775914bffb6SAlexander V. Chernikov 			comma = ",";
1776914bffb6SAlexander V. Chernikov 		}
1777914bffb6SAlexander V. Chernikov 
1778914bffb6SAlexander V. Chernikov 		if ((i->tflags & IPFW_TFFLAG_PROTO) != 0) {
1779914bffb6SAlexander V. Chernikov 			printf("%s%d", comma, tfe->proto);
1780914bffb6SAlexander V. Chernikov 			comma = ",";
1781914bffb6SAlexander V. Chernikov 		}
1782914bffb6SAlexander V. Chernikov 
1783914bffb6SAlexander V. Chernikov 		if ((i->tflags & IPFW_TFFLAG_SRCPORT) != 0) {
1784914bffb6SAlexander V. Chernikov 			printf("%s%d", comma, ntohs(tfe->sport));
1785914bffb6SAlexander V. Chernikov 			comma = ",";
1786914bffb6SAlexander V. Chernikov 		}
1787914bffb6SAlexander V. Chernikov 		if ((i->tflags & IPFW_TFFLAG_DSTIP) != 0) {
1788914bffb6SAlexander V. Chernikov 			if (tfe->af == AF_INET)
1789914bffb6SAlexander V. Chernikov 				paddr = &tfe->a.a4.dip;
1790914bffb6SAlexander V. Chernikov 			else
1791914bffb6SAlexander V. Chernikov 				paddr = &tfe->a.a6.dip6;
1792914bffb6SAlexander V. Chernikov 
1793914bffb6SAlexander V. Chernikov 			inet_ntop(tfe->af, paddr, tbuf, sizeof(tbuf));
1794914bffb6SAlexander V. Chernikov 			printf("%s%s", comma, tbuf);
1795914bffb6SAlexander V. Chernikov 			comma = ",";
1796914bffb6SAlexander V. Chernikov 		}
1797914bffb6SAlexander V. Chernikov 
1798914bffb6SAlexander V. Chernikov 		if ((i->tflags & IPFW_TFFLAG_DSTPORT) != 0) {
1799914bffb6SAlexander V. Chernikov 			printf("%s%d", comma, ntohs(tfe->dport));
1800914bffb6SAlexander V. Chernikov 			comma = ",";
1801914bffb6SAlexander V. Chernikov 		}
1802914bffb6SAlexander V. Chernikov 
1803914bffb6SAlexander V. Chernikov 		printf(" %s\n", pval);
1804f1220db8SAlexander V. Chernikov 	}
1805f1220db8SAlexander V. Chernikov }
1806f1220db8SAlexander V. Chernikov 
18079d099b4fSAlexander V. Chernikov static int
18080cba2b28SAlexander V. Chernikov table_do_get_stdlist(uint16_t opcode, ipfw_obj_lheader **polh)
18099d099b4fSAlexander V. Chernikov {
18109d099b4fSAlexander V. Chernikov 	ipfw_obj_lheader req, *olh;
18119d099b4fSAlexander V. Chernikov 	size_t sz;
18129d099b4fSAlexander V. Chernikov 
18139d099b4fSAlexander V. Chernikov 	memset(&req, 0, sizeof(req));
18149d099b4fSAlexander V. Chernikov 	sz = sizeof(req);
18159d099b4fSAlexander V. Chernikov 
1816b6462881SAlexander V. Chernikov 	if (do_get3(opcode, &req.opheader, &sz) != 0)
1817b6462881SAlexander V. Chernikov 		if (errno != ENOMEM)
1818b6462881SAlexander V. Chernikov 			return (errno);
18199d099b4fSAlexander V. Chernikov 
18209d099b4fSAlexander V. Chernikov 	sz = req.size;
18219d099b4fSAlexander V. Chernikov 	if ((olh = calloc(1, sz)) == NULL)
18229d099b4fSAlexander V. Chernikov 		return (ENOMEM);
18239d099b4fSAlexander V. Chernikov 
18249d099b4fSAlexander V. Chernikov 	olh->size = sz;
1825b6462881SAlexander V. Chernikov 	if (do_get3(opcode, &olh->opheader, &sz) != 0) {
18269d099b4fSAlexander V. Chernikov 		free(olh);
1827b6462881SAlexander V. Chernikov 		return (errno);
18289d099b4fSAlexander V. Chernikov 	}
18299d099b4fSAlexander V. Chernikov 
18309d099b4fSAlexander V. Chernikov 	*polh = olh;
18319d099b4fSAlexander V. Chernikov 	return (0);
18329d099b4fSAlexander V. Chernikov }
18339d099b4fSAlexander V. Chernikov 
18340cba2b28SAlexander V. Chernikov static int
18350cba2b28SAlexander V. Chernikov table_do_get_algolist(ipfw_obj_lheader **polh)
18360cba2b28SAlexander V. Chernikov {
18370cba2b28SAlexander V. Chernikov 
18380cba2b28SAlexander V. Chernikov 	return (table_do_get_stdlist(IP_FW_TABLES_ALIST, polh));
18390cba2b28SAlexander V. Chernikov }
18400cba2b28SAlexander V. Chernikov 
18410cba2b28SAlexander V. Chernikov static int
18420cba2b28SAlexander V. Chernikov table_do_get_vlist(ipfw_obj_lheader **polh)
18430cba2b28SAlexander V. Chernikov {
18440cba2b28SAlexander V. Chernikov 
18450cba2b28SAlexander V. Chernikov 	return (table_do_get_stdlist(IP_FW_TABLE_VLIST, polh));
18460cba2b28SAlexander V. Chernikov }
18470cba2b28SAlexander V. Chernikov 
18489d099b4fSAlexander V. Chernikov void
18499d099b4fSAlexander V. Chernikov ipfw_list_ta(int ac, char *av[])
18509d099b4fSAlexander V. Chernikov {
18519d099b4fSAlexander V. Chernikov 	ipfw_obj_lheader *olh;
18529d099b4fSAlexander V. Chernikov 	ipfw_ta_info *info;
18539d099b4fSAlexander V. Chernikov 	int error, i;
18549d099b4fSAlexander V. Chernikov 	const char *atype;
18559d099b4fSAlexander V. Chernikov 
18569d099b4fSAlexander V. Chernikov 	error = table_do_get_algolist(&olh);
18579d099b4fSAlexander V. Chernikov 	if (error != 0)
18589d099b4fSAlexander V. Chernikov 		err(EX_OSERR, "Unable to request algorithm list");
18599d099b4fSAlexander V. Chernikov 
18609d099b4fSAlexander V. Chernikov 	info = (ipfw_ta_info *)(olh + 1);
18619d099b4fSAlexander V. Chernikov 	for (i = 0; i < olh->count; i++) {
18629d099b4fSAlexander V. Chernikov 		if ((atype = match_value(tabletypes, info->type)) == NULL)
18639d099b4fSAlexander V. Chernikov 			atype = "unknown";
18648ce7a2bcSAlexander V. Chernikov 		printf("--- %s ---\n", info->algoname);
18658ce7a2bcSAlexander V. Chernikov 		printf(" type: %s\n refcount: %u\n", atype, info->refcnt);
18669d099b4fSAlexander V. Chernikov 
18679d099b4fSAlexander V. Chernikov 		info = (ipfw_ta_info *)((caddr_t)info + olh->objsize);
18689d099b4fSAlexander V. Chernikov 	}
18699d099b4fSAlexander V. Chernikov 
18709d099b4fSAlexander V. Chernikov 	free(olh);
18719d099b4fSAlexander V. Chernikov }
18729d099b4fSAlexander V. Chernikov 
18730cba2b28SAlexander V. Chernikov 
18740cba2b28SAlexander V. Chernikov /* Copy of current kernel table_value structure */
18750cba2b28SAlexander V. Chernikov struct _table_value {
18760cba2b28SAlexander V. Chernikov 	uint32_t	tag;		/* O_TAG/O_TAGGED */
18770cba2b28SAlexander V. Chernikov 	uint32_t	pipe;		/* O_PIPE/O_QUEUE */
18780cba2b28SAlexander V. Chernikov 	uint16_t	divert;		/* O_DIVERT/O_TEE */
18790cba2b28SAlexander V. Chernikov 	uint16_t	skipto;		/* skipto, CALLRET */
18800cba2b28SAlexander V. Chernikov 	uint32_t	netgraph;	/* O_NETGRAPH/O_NGTEE */
18810cba2b28SAlexander V. Chernikov 	uint32_t	fib;		/* O_SETFIB */
18820cba2b28SAlexander V. Chernikov 	uint32_t	nat;		/* O_NAT */
18830cba2b28SAlexander V. Chernikov 	uint32_t	nh4;
18840cba2b28SAlexander V. Chernikov 	uint8_t		dscp;
1885*2530ed9eSAndrey V. Elsukov 	uint8_t		spare0;
1886*2530ed9eSAndrey V. Elsukov 	uint16_t	spare1;
18870cba2b28SAlexander V. Chernikov 	/* -- 32 bytes -- */
18880cba2b28SAlexander V. Chernikov 	struct in6_addr	nh6;
18890cba2b28SAlexander V. Chernikov 	uint32_t	limit;		/* O_LIMIT */
1890*2530ed9eSAndrey V. Elsukov 	uint32_t	zoneid;
18910cba2b28SAlexander V. Chernikov 	uint64_t	refcnt;		/* Number of references */
18920cba2b28SAlexander V. Chernikov };
18930cba2b28SAlexander V. Chernikov 
18940cba2b28SAlexander V. Chernikov int
18950cba2b28SAlexander V. Chernikov compare_values(const void *_a, const void *_b)
18960cba2b28SAlexander V. Chernikov {
18970cba2b28SAlexander V. Chernikov 	struct _table_value *a, *b;
18980cba2b28SAlexander V. Chernikov 
18990cba2b28SAlexander V. Chernikov 	a = (struct _table_value *)_a;
19000cba2b28SAlexander V. Chernikov 	b = (struct _table_value *)_b;
19010cba2b28SAlexander V. Chernikov 
19020cba2b28SAlexander V. Chernikov 	if (a->spare1 < b->spare1)
19030cba2b28SAlexander V. Chernikov 		return (-1);
19040cba2b28SAlexander V. Chernikov 	else if (a->spare1 > b->spare1)
19050cba2b28SAlexander V. Chernikov 		return (1);
19060cba2b28SAlexander V. Chernikov 
19070cba2b28SAlexander V. Chernikov 	return (0);
19080cba2b28SAlexander V. Chernikov }
19090cba2b28SAlexander V. Chernikov 
19100cba2b28SAlexander V. Chernikov void
19110cba2b28SAlexander V. Chernikov ipfw_list_values(int ac, char *av[])
19120cba2b28SAlexander V. Chernikov {
19130cba2b28SAlexander V. Chernikov 	ipfw_obj_lheader *olh;
19140cba2b28SAlexander V. Chernikov 	struct _table_value *v;
19150cba2b28SAlexander V. Chernikov 	int error, i;
19160cba2b28SAlexander V. Chernikov 	uint32_t vmask;
19170cba2b28SAlexander V. Chernikov 	char buf[128];
19180cba2b28SAlexander V. Chernikov 
19190cba2b28SAlexander V. Chernikov 	error = table_do_get_vlist(&olh);
19200cba2b28SAlexander V. Chernikov 	if (error != 0)
19210cba2b28SAlexander V. Chernikov 		err(EX_OSERR, "Unable to request value list");
19220cba2b28SAlexander V. Chernikov 
19230cba2b28SAlexander V. Chernikov 	vmask = 0x7FFFFFFF; /* Similar to IPFW_VTYPE_LEGACY */
19240cba2b28SAlexander V. Chernikov 
19250cba2b28SAlexander V. Chernikov 	table_print_valheader(buf, sizeof(buf), vmask);
19260cba2b28SAlexander V. Chernikov 	printf("HEADER: %s\n", buf);
19270cba2b28SAlexander V. Chernikov 	v = (struct _table_value *)(olh + 1);
19280cba2b28SAlexander V. Chernikov 	qsort(v, olh->count, olh->objsize, compare_values);
19290cba2b28SAlexander V. Chernikov 	for (i = 0; i < olh->count; i++) {
19300cba2b28SAlexander V. Chernikov 		table_show_value(buf, sizeof(buf), (ipfw_table_value *)v,
19310cba2b28SAlexander V. Chernikov 		    vmask, 0);
1932d3de6c1dSAlexander V. Chernikov 		printf("[%u] refs=%lu %s\n", v->spare1, (u_long)v->refcnt, buf);
19330cba2b28SAlexander V. Chernikov 		v = (struct _table_value *)((caddr_t)v + olh->objsize);
19340cba2b28SAlexander V. Chernikov 	}
19350cba2b28SAlexander V. Chernikov 
19360cba2b28SAlexander V. Chernikov 	free(olh);
19370cba2b28SAlexander V. Chernikov }
19380cba2b28SAlexander V. Chernikov 
19396c2997ffSAlexander V. Chernikov int
19406c2997ffSAlexander V. Chernikov compare_ntlv(const void *_a, const void *_b)
19416c2997ffSAlexander V. Chernikov {
19426c2997ffSAlexander V. Chernikov 	ipfw_obj_ntlv *a, *b;
19436c2997ffSAlexander V. Chernikov 
19446c2997ffSAlexander V. Chernikov 	a = (ipfw_obj_ntlv *)_a;
19456c2997ffSAlexander V. Chernikov 	b = (ipfw_obj_ntlv *)_b;
19466c2997ffSAlexander V. Chernikov 
19476c2997ffSAlexander V. Chernikov 	if (a->set < b->set)
19486c2997ffSAlexander V. Chernikov 		return (-1);
19496c2997ffSAlexander V. Chernikov 	else if (a->set > b->set)
19506c2997ffSAlexander V. Chernikov 		return (1);
19516c2997ffSAlexander V. Chernikov 
19526c2997ffSAlexander V. Chernikov 	if (a->idx < b->idx)
19536c2997ffSAlexander V. Chernikov 		return (-1);
19546c2997ffSAlexander V. Chernikov 	else if (a->idx > b->idx)
19556c2997ffSAlexander V. Chernikov 		return (1);
19566c2997ffSAlexander V. Chernikov 
19576c2997ffSAlexander V. Chernikov 	return (0);
19586c2997ffSAlexander V. Chernikov }
1959563b5ab1SAlexander V. Chernikov 
1960563b5ab1SAlexander V. Chernikov int
19616c2997ffSAlexander V. Chernikov compare_kntlv(const void *k, const void *v)
1962563b5ab1SAlexander V. Chernikov {
1963563b5ab1SAlexander V. Chernikov 	ipfw_obj_ntlv *ntlv;
1964563b5ab1SAlexander V. Chernikov 	uint16_t key;
1965563b5ab1SAlexander V. Chernikov 
1966563b5ab1SAlexander V. Chernikov 	key = *((uint16_t *)k);
1967563b5ab1SAlexander V. Chernikov 	ntlv = (ipfw_obj_ntlv *)v;
1968563b5ab1SAlexander V. Chernikov 
1969563b5ab1SAlexander V. Chernikov 	if (key < ntlv->idx)
1970563b5ab1SAlexander V. Chernikov 		return (-1);
1971563b5ab1SAlexander V. Chernikov 	else if (key > ntlv->idx)
1972563b5ab1SAlexander V. Chernikov 		return (1);
1973563b5ab1SAlexander V. Chernikov 
1974563b5ab1SAlexander V. Chernikov 	return (0);
1975563b5ab1SAlexander V. Chernikov }
1976563b5ab1SAlexander V. Chernikov 
1977563b5ab1SAlexander V. Chernikov /*
1978563b5ab1SAlexander V. Chernikov  * Finds table name in @ctlv by @idx.
1979563b5ab1SAlexander V. Chernikov  * Uses the following facts:
1980563b5ab1SAlexander V. Chernikov  * 1) All TLVs are the same size
1981563b5ab1SAlexander V. Chernikov  * 2) Kernel implementation provides already sorted list.
1982563b5ab1SAlexander V. Chernikov  *
1983563b5ab1SAlexander V. Chernikov  * Returns table name or NULL.
1984563b5ab1SAlexander V. Chernikov  */
1985563b5ab1SAlexander V. Chernikov char *
1986563b5ab1SAlexander V. Chernikov table_search_ctlv(ipfw_obj_ctlv *ctlv, uint16_t idx)
1987563b5ab1SAlexander V. Chernikov {
1988563b5ab1SAlexander V. Chernikov 	ipfw_obj_ntlv *ntlv;
1989563b5ab1SAlexander V. Chernikov 
1990563b5ab1SAlexander V. Chernikov 	ntlv = bsearch(&idx, (ctlv + 1), ctlv->count, ctlv->objsize,
19916c2997ffSAlexander V. Chernikov 	    compare_kntlv);
1992563b5ab1SAlexander V. Chernikov 
1993563b5ab1SAlexander V. Chernikov 	if (ntlv != 0)
1994563b5ab1SAlexander V. Chernikov 		return (ntlv->name);
1995563b5ab1SAlexander V. Chernikov 
1996563b5ab1SAlexander V. Chernikov 	return (NULL);
1997563b5ab1SAlexander V. Chernikov }
1998563b5ab1SAlexander V. Chernikov 
19996c2997ffSAlexander V. Chernikov void
20006c2997ffSAlexander V. Chernikov table_sort_ctlv(ipfw_obj_ctlv *ctlv)
20016c2997ffSAlexander V. Chernikov {
20026c2997ffSAlexander V. Chernikov 
20036c2997ffSAlexander V. Chernikov 	qsort(ctlv + 1, ctlv->count, ctlv->objsize, compare_ntlv);
20046c2997ffSAlexander V. Chernikov }
20056c2997ffSAlexander V. Chernikov 
20066c2997ffSAlexander V. Chernikov int
20076c2997ffSAlexander V. Chernikov table_check_name(char *tablename)
20086c2997ffSAlexander V. Chernikov {
20096c2997ffSAlexander V. Chernikov 	int c, i, l;
20106c2997ffSAlexander V. Chernikov 
20116c2997ffSAlexander V. Chernikov 	/*
20126c2997ffSAlexander V. Chernikov 	 * Check if tablename is null-terminated and contains
20136c2997ffSAlexander V. Chernikov 	 * valid symbols only. Valid mask is:
2014ac35ff17SAlexander V. Chernikov 	 * [a-zA-Z0-9\-_\.]{1,63}
20156c2997ffSAlexander V. Chernikov 	 */
20166c2997ffSAlexander V. Chernikov 	l = strlen(tablename);
20176c2997ffSAlexander V. Chernikov 	if (l == 0 || l >= 64)
20186c2997ffSAlexander V. Chernikov 		return (EINVAL);
20196c2997ffSAlexander V. Chernikov 	for (i = 0; i < l; i++) {
20206c2997ffSAlexander V. Chernikov 		c = tablename[i];
20216c2997ffSAlexander V. Chernikov 		if (isalpha(c) || isdigit(c) || c == '_' ||
20226c2997ffSAlexander V. Chernikov 		    c == '-' || c == '.')
20236c2997ffSAlexander V. Chernikov 			continue;
20246c2997ffSAlexander V. Chernikov 		return (EINVAL);
20256c2997ffSAlexander V. Chernikov 	}
20266c2997ffSAlexander V. Chernikov 
2027ac35ff17SAlexander V. Chernikov 	/* Restrict some 'special' names */
2028ac35ff17SAlexander V. Chernikov 	if (strcmp(tablename, "all") == 0)
2029ac35ff17SAlexander V. Chernikov 		return (EINVAL);
2030ac35ff17SAlexander V. Chernikov 
20316c2997ffSAlexander V. Chernikov 	return (0);
20326c2997ffSAlexander V. Chernikov }
20336c2997ffSAlexander V. Chernikov 
2034