xref: /freebsd/sbin/ipfw/tables.c (revision 13ad237a19b7368124483d9d1dc3258c27880fef)
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>
382530ed9eSAndrey 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);
56f1220db8SAlexander V. Chernikov 
57421c5838SAndrey V. Elsukov static int table_destroy_one(ipfw_xtable_info *i, void *arg);
58f1220db8SAlexander V. Chernikov static int table_flush_one(ipfw_xtable_info *i, void *arg);
59f1220db8SAlexander V. Chernikov static int table_show_one(ipfw_xtable_info *i, void *arg);
60720ee730SAlexander V. Chernikov static int table_do_get_list(ipfw_xtable_info *i, ipfw_obj_header **poh);
61f1220db8SAlexander V. Chernikov static void table_show_list(ipfw_obj_header *oh, int need_header);
6281d3153dSAlexander V. Chernikov static void table_show_entry(ipfw_xtable_info *i, ipfw_obj_tentry *tent);
63f1220db8SAlexander V. Chernikov 
64ac35ff17SAlexander V. Chernikov static void tentry_fill_key(ipfw_obj_header *oh, ipfw_obj_tentry *tent,
650cba2b28SAlexander V. Chernikov     char *key, int add, uint8_t *ptype, uint32_t *pvmask, ipfw_xtable_info *xi);
66ac35ff17SAlexander V. Chernikov static void tentry_fill_value(ipfw_obj_header *oh, ipfw_obj_tentry *tent,
670cba2b28SAlexander V. Chernikov     char *arg, uint8_t type, uint32_t vmask);
680cba2b28SAlexander V. Chernikov static void table_show_value(char *buf, size_t bufsize, ipfw_table_value *v,
690cba2b28SAlexander V. Chernikov     uint32_t vmask, int print_ip);
70ac35ff17SAlexander V. Chernikov 
71f1220db8SAlexander V. Chernikov typedef int (table_cb_t)(ipfw_xtable_info *i, void *arg);
72f1220db8SAlexander V. Chernikov static int tables_foreach(table_cb_t *f, void *arg, int sort);
73f1220db8SAlexander V. Chernikov 
74f1220db8SAlexander V. Chernikov #ifndef s6_addr32
75f1220db8SAlexander V. Chernikov #define s6_addr32 __u6_addr.__u6_addr32
76f1220db8SAlexander V. Chernikov #endif
77f1220db8SAlexander V. Chernikov 
78ac35ff17SAlexander V. Chernikov static struct _s_x tabletypes[] = {
79c21034b7SAlexander V. Chernikov       { "addr",		IPFW_TABLE_ADDR },
80ac35ff17SAlexander V. Chernikov       { "iface",	IPFW_TABLE_INTERFACE },
81b23d5de9SAlexander V. Chernikov       { "number",	IPFW_TABLE_NUMBER },
82914bffb6SAlexander V. Chernikov       { "flow",		IPFW_TABLE_FLOW },
83ac35ff17SAlexander V. Chernikov       { NULL, 0 }
84ac35ff17SAlexander V. Chernikov };
85ac35ff17SAlexander V. Chernikov 
86*13ad237aSAndrey V. Elsukov /* Default algorithms for various table types */
87*13ad237aSAndrey V. Elsukov static struct _s_x tablealgos[] = {
88*13ad237aSAndrey V. Elsukov       { "addr:radix",	IPFW_TABLE_ADDR },
89*13ad237aSAndrey V. Elsukov       { "flow:hash",	IPFW_TABLE_FLOW },
90*13ad237aSAndrey V. Elsukov       { "iface:array",	IPFW_TABLE_INTERFACE },
91*13ad237aSAndrey V. Elsukov       { "number:array",	IPFW_TABLE_NUMBER },
92*13ad237aSAndrey V. Elsukov       { NULL, 0 }
93*13ad237aSAndrey V. Elsukov };
94*13ad237aSAndrey V. Elsukov 
95ac35ff17SAlexander V. Chernikov static struct _s_x tablevaltypes[] = {
960cba2b28SAlexander V. Chernikov       { "skipto",	IPFW_VTYPE_SKIPTO },
970cba2b28SAlexander V. Chernikov       { "pipe",		IPFW_VTYPE_PIPE },
980cba2b28SAlexander V. Chernikov       { "fib",		IPFW_VTYPE_FIB },
990cba2b28SAlexander V. Chernikov       { "nat",		IPFW_VTYPE_NAT },
1000cba2b28SAlexander V. Chernikov       { "dscp",		IPFW_VTYPE_DSCP },
1010cba2b28SAlexander V. Chernikov       { "tag",		IPFW_VTYPE_TAG },
1020cba2b28SAlexander V. Chernikov       { "divert",	IPFW_VTYPE_DIVERT },
1030cba2b28SAlexander V. Chernikov       { "netgraph",	IPFW_VTYPE_NETGRAPH },
1040cba2b28SAlexander V. Chernikov       { "limit",	IPFW_VTYPE_LIMIT },
1050cba2b28SAlexander V. Chernikov       { "ipv4",		IPFW_VTYPE_NH4 },
1060cba2b28SAlexander V. Chernikov       { "ipv6",		IPFW_VTYPE_NH6 },
107adf3b2b9SAlexander V. Chernikov       { NULL, 0 }
108adf3b2b9SAlexander V. Chernikov };
109adf3b2b9SAlexander V. Chernikov 
110ac35ff17SAlexander V. Chernikov static struct _s_x tablecmds[] = {
111ac35ff17SAlexander V. Chernikov       { "add",		TOK_ADD },
112ac35ff17SAlexander V. Chernikov       { "delete",	TOK_DEL },
11346d52008SAlexander V. Chernikov       { "create",	TOK_CREATE },
114ac35ff17SAlexander V. Chernikov       { "destroy",	TOK_DESTROY },
115ac35ff17SAlexander V. Chernikov       { "flush",	TOK_FLUSH },
116adf3b2b9SAlexander V. Chernikov       { "modify",	TOK_MODIFY },
11746d52008SAlexander V. Chernikov       { "swap",		TOK_SWAP },
118ac35ff17SAlexander V. Chernikov       { "info",		TOK_INFO },
119358b9d09SAlexander V. Chernikov       { "detail",	TOK_DETAIL },
120ac35ff17SAlexander V. Chernikov       { "list",		TOK_LIST },
12181d3153dSAlexander V. Chernikov       { "lookup",	TOK_LOOKUP },
1223a845e10SAlexander V. Chernikov       { "atomic",	TOK_ATOMIC },
1234f43138aSAlexander V. Chernikov       { "lock",		TOK_LOCK },
1244f43138aSAlexander V. Chernikov       { "unlock",	TOK_UNLOCK },
125ac35ff17SAlexander V. Chernikov       { NULL, 0 }
126ac35ff17SAlexander V. Chernikov };
127ac35ff17SAlexander V. Chernikov 
128f1220db8SAlexander V. Chernikov static int
129f1220db8SAlexander V. Chernikov lookup_host (char *host, struct in_addr *ipaddr)
130f1220db8SAlexander V. Chernikov {
131f1220db8SAlexander V. Chernikov 	struct hostent *he;
132f1220db8SAlexander V. Chernikov 
133f1220db8SAlexander V. Chernikov 	if (!inet_aton(host, ipaddr)) {
134f1220db8SAlexander V. Chernikov 		if ((he = gethostbyname(host)) == NULL)
135f1220db8SAlexander V. Chernikov 			return(-1);
136f1220db8SAlexander V. Chernikov 		*ipaddr = *(struct in_addr *)he->h_addr_list[0];
137f1220db8SAlexander V. Chernikov 	}
138f1220db8SAlexander V. Chernikov 	return(0);
139f1220db8SAlexander V. Chernikov }
140f1220db8SAlexander V. Chernikov 
141f1220db8SAlexander V. Chernikov /*
142f1220db8SAlexander V. Chernikov  * This one handles all table-related commands
143ac35ff17SAlexander V. Chernikov  * 	ipfw table NAME create ...
144be695df9SAlexander V. Chernikov  * 	ipfw table NAME modify ...
145421c5838SAndrey V. Elsukov  * 	ipfw table {NAME | all} destroy
146be695df9SAlexander V. Chernikov  * 	ipfw table NAME swap NAME
147be695df9SAlexander V. Chernikov  * 	ipfw table NAME lock
148be695df9SAlexander V. Chernikov  * 	ipfw table NAME unlock
149ac35ff17SAlexander V. Chernikov  * 	ipfw table NAME add addr[/masklen] [value]
150be695df9SAlexander V. Chernikov  * 	ipfw table NAME add [addr[/masklen] value] [addr[/masklen] value] ..
151be695df9SAlexander V. Chernikov  * 	ipfw table NAME delete addr[/masklen] [addr[/masklen]] ..
152be695df9SAlexander V. Chernikov  * 	ipfw table NAME lookup addr
153ac35ff17SAlexander V. Chernikov  * 	ipfw table {NAME | all} flush
154ac35ff17SAlexander V. Chernikov  * 	ipfw table {NAME | all} list
155ac35ff17SAlexander V. Chernikov  * 	ipfw table {NAME | all} info
156be695df9SAlexander V. Chernikov  * 	ipfw table {NAME | all} detail
157f1220db8SAlexander V. Chernikov  */
158f1220db8SAlexander V. Chernikov void
159f1220db8SAlexander V. Chernikov ipfw_table_handler(int ac, char *av[])
160f1220db8SAlexander V. Chernikov {
161ac35ff17SAlexander V. Chernikov 	int do_add, is_all;
1623a845e10SAlexander V. Chernikov 	int atomic, error, tcmd;
163ac35ff17SAlexander V. Chernikov 	ipfw_xtable_info i;
164ac35ff17SAlexander V. Chernikov 	ipfw_obj_header oh;
165f1220db8SAlexander V. Chernikov 	char *tablename;
166d8caf56eSAndrey V. Elsukov 	uint8_t set;
167358b9d09SAlexander V. Chernikov 	void *arg;
168f1220db8SAlexander V. Chernikov 
169ac35ff17SAlexander V. Chernikov 	memset(&oh, 0, sizeof(oh));
170ac35ff17SAlexander V. Chernikov 	is_all = 0;
17156707beeSMark Johnston 	if (g_co.use_set != 0)
17256707beeSMark Johnston 		set = g_co.use_set - 1;
173ac35ff17SAlexander V. Chernikov 	else
174ac35ff17SAlexander V. Chernikov 		set = 0;
175f1220db8SAlexander V. Chernikov 
176f1220db8SAlexander V. Chernikov 	ac--; av++;
1779d099b4fSAlexander V. Chernikov 	NEED1("table needs name");
178f1220db8SAlexander V. Chernikov 	tablename = *av;
179f1220db8SAlexander V. Chernikov 
180ac35ff17SAlexander V. Chernikov 	if (table_check_name(tablename) == 0) {
181ac35ff17SAlexander V. Chernikov 		table_fill_ntlv(&oh.ntlv, *av, set, 1);
182ac35ff17SAlexander V. Chernikov 		oh.idx = 1;
183ac35ff17SAlexander V. Chernikov 	} else {
184ac35ff17SAlexander V. Chernikov 		if (strcmp(tablename, "all") == 0)
185ac35ff17SAlexander V. Chernikov 			is_all = 1;
186ac35ff17SAlexander V. Chernikov 		else
187ac35ff17SAlexander V. Chernikov 			errx(EX_USAGE, "table name %s is invalid", tablename);
188ac35ff17SAlexander V. Chernikov 	}
189ac35ff17SAlexander V. Chernikov 	ac--; av++;
1909d099b4fSAlexander V. Chernikov 	NEED1("table needs command");
191ac35ff17SAlexander V. Chernikov 
192be695df9SAlexander V. Chernikov 	tcmd = get_token(tablecmds, *av, "table command");
1933a845e10SAlexander V. Chernikov 	/* Check if atomic operation was requested */
1943a845e10SAlexander V. Chernikov 	atomic = 0;
1953a845e10SAlexander V. Chernikov 	if (tcmd == TOK_ATOMIC) {
1963a845e10SAlexander V. Chernikov 		ac--; av++;
1973a845e10SAlexander V. Chernikov 		NEED1("atomic needs command");
198be695df9SAlexander V. Chernikov 		tcmd = get_token(tablecmds, *av, "table command");
1993a845e10SAlexander V. Chernikov 		switch (tcmd) {
2003a845e10SAlexander V. Chernikov 		case TOK_ADD:
2013a845e10SAlexander V. Chernikov 			break;
2023a845e10SAlexander V. Chernikov 		default:
2033a845e10SAlexander V. Chernikov 			errx(EX_USAGE, "atomic is not compatible with %s", *av);
2043a845e10SAlexander V. Chernikov 		}
2053a845e10SAlexander V. Chernikov 		atomic = 1;
2063a845e10SAlexander V. Chernikov 	}
207ac35ff17SAlexander V. Chernikov 
208ac35ff17SAlexander V. Chernikov 	switch (tcmd) {
209ac35ff17SAlexander V. Chernikov 	case TOK_LIST:
210ac35ff17SAlexander V. Chernikov 	case TOK_INFO:
211358b9d09SAlexander V. Chernikov 	case TOK_DETAIL:
212ac35ff17SAlexander V. Chernikov 	case TOK_FLUSH:
213421c5838SAndrey V. Elsukov 	case TOK_DESTROY:
214ac35ff17SAlexander V. Chernikov 		break;
215ac35ff17SAlexander V. Chernikov 	default:
216ac35ff17SAlexander V. Chernikov 		if (is_all != 0)
217ac35ff17SAlexander V. Chernikov 			errx(EX_USAGE, "table name required");
218ac35ff17SAlexander V. Chernikov 	}
219ac35ff17SAlexander V. Chernikov 
220ac35ff17SAlexander V. Chernikov 	switch (tcmd) {
221ac35ff17SAlexander V. Chernikov 	case TOK_ADD:
222ac35ff17SAlexander V. Chernikov 	case TOK_DEL:
223f1220db8SAlexander V. Chernikov 		do_add = **av == 'a';
224f1220db8SAlexander V. Chernikov 		ac--; av++;
22556707beeSMark Johnston 		table_modify_record(&oh, ac, av, do_add, g_co.do_quiet,
22656707beeSMark Johnston 		    g_co.do_quiet, atomic);
227ac35ff17SAlexander V. Chernikov 		break;
228ac35ff17SAlexander V. Chernikov 	case TOK_CREATE:
229f1220db8SAlexander V. Chernikov 		ac--; av++;
230ac35ff17SAlexander V. Chernikov 		table_create(&oh, ac, av);
231ac35ff17SAlexander V. Chernikov 		break;
232adf3b2b9SAlexander V. Chernikov 	case TOK_MODIFY:
233adf3b2b9SAlexander V. Chernikov 		ac--; av++;
234adf3b2b9SAlexander V. Chernikov 		table_modify(&oh, ac, av);
235adf3b2b9SAlexander V. Chernikov 		break;
236ac35ff17SAlexander V. Chernikov 	case TOK_DESTROY:
237421c5838SAndrey V. Elsukov 		if (is_all == 0) {
23876d03257SAndrey V. Elsukov 			if (table_destroy(&oh) == 0)
23976d03257SAndrey V. Elsukov 				break;
24076d03257SAndrey V. Elsukov 			if (errno != ESRCH)
241421c5838SAndrey V. Elsukov 				err(EX_OSERR, "failed to destroy table %s",
242421c5838SAndrey V. Elsukov 				    tablename);
24376d03257SAndrey V. Elsukov 			/* ESRCH isn't fatal, warn if not quiet mode */
24456707beeSMark Johnston 			if (g_co.do_quiet == 0)
24576d03257SAndrey V. Elsukov 				warn("failed to destroy table %s", tablename);
246421c5838SAndrey V. Elsukov 		} else {
247421c5838SAndrey V. Elsukov 			error = tables_foreach(table_destroy_one, &oh, 1);
248421c5838SAndrey V. Elsukov 			if (error != 0)
249421c5838SAndrey V. Elsukov 				err(EX_OSERR,
250421c5838SAndrey V. Elsukov 				    "failed to destroy tables list");
251421c5838SAndrey V. Elsukov 		}
252ac35ff17SAlexander V. Chernikov 		break;
253ac35ff17SAlexander V. Chernikov 	case TOK_FLUSH:
254f1220db8SAlexander V. Chernikov 		if (is_all == 0) {
25576d03257SAndrey V. Elsukov 			if ((error = table_flush(&oh)) == 0)
25676d03257SAndrey V. Elsukov 				break;
25776d03257SAndrey V. Elsukov 			if (errno != ESRCH)
258f1220db8SAlexander V. Chernikov 				err(EX_OSERR, "failed to flush table %s info",
259f1220db8SAlexander V. Chernikov 				    tablename);
26076d03257SAndrey V. Elsukov 			/* ESRCH isn't fatal, warn if not quiet mode */
26156707beeSMark Johnston 			if (g_co.do_quiet == 0)
26276d03257SAndrey V. Elsukov 				warn("failed to flush table %s info",
26376d03257SAndrey V. Elsukov 				    tablename);
264f1220db8SAlexander V. Chernikov 		} else {
265ac35ff17SAlexander V. Chernikov 			error = tables_foreach(table_flush_one, &oh, 1);
266f1220db8SAlexander V. Chernikov 			if (error != 0)
267f1220db8SAlexander V. Chernikov 				err(EX_OSERR, "failed to flush tables list");
26876d03257SAndrey V. Elsukov 			/* XXX: we ignore errors here */
269f1220db8SAlexander V. Chernikov 		}
270ac35ff17SAlexander V. Chernikov 		break;
27146d52008SAlexander V. Chernikov 	case TOK_SWAP:
27246d52008SAlexander V. Chernikov 		ac--; av++;
27346d52008SAlexander V. Chernikov 		NEED1("second table name required");
27446d52008SAlexander V. Chernikov 		table_swap(&oh, *av);
27546d52008SAlexander V. Chernikov 		break;
2764f43138aSAlexander V. Chernikov 	case TOK_LOCK:
2774f43138aSAlexander V. Chernikov 	case TOK_UNLOCK:
2784f43138aSAlexander V. Chernikov 		table_lock(&oh, (tcmd == TOK_LOCK));
2794f43138aSAlexander V. Chernikov 		break;
280358b9d09SAlexander V. Chernikov 	case TOK_DETAIL:
281ac35ff17SAlexander V. Chernikov 	case TOK_INFO:
282358b9d09SAlexander V. Chernikov 		arg = (tcmd == TOK_DETAIL) ? (void *)1 : NULL;
283f1220db8SAlexander V. Chernikov 		if (is_all == 0) {
284ac35ff17SAlexander V. Chernikov 			if ((error = table_get_info(&oh, &i)) != 0)
285f1220db8SAlexander V. Chernikov 				err(EX_OSERR, "failed to request table info");
286358b9d09SAlexander V. Chernikov 			table_show_info(&i, arg);
287f1220db8SAlexander V. Chernikov 		} else {
288358b9d09SAlexander V. Chernikov 			error = tables_foreach(table_show_info, arg, 1);
289f1220db8SAlexander V. Chernikov 			if (error != 0)
290f1220db8SAlexander V. Chernikov 				err(EX_OSERR, "failed to request tables list");
291f1220db8SAlexander V. Chernikov 		}
292ac35ff17SAlexander V. Chernikov 		break;
293ac35ff17SAlexander V. Chernikov 	case TOK_LIST:
29410e3bebfSBryan Drewery 		arg = is_all ? (void*)1 : NULL;
295ac35ff17SAlexander V. Chernikov 		if (is_all == 0) {
296ac35ff17SAlexander V. Chernikov 			if ((error = table_get_info(&oh, &i)) != 0)
297ac35ff17SAlexander V. Chernikov 				err(EX_OSERR, "failed to request table info");
2982b3d6647SBryan Drewery 			table_show_one(&i, arg);
299f1220db8SAlexander V. Chernikov 		} else {
3002b3d6647SBryan Drewery 			error = tables_foreach(table_show_one, arg, 1);
301ac35ff17SAlexander V. Chernikov 			if (error != 0)
302ac35ff17SAlexander V. Chernikov 				err(EX_OSERR, "failed to request tables list");
303f1220db8SAlexander V. Chernikov 		}
304ac35ff17SAlexander V. Chernikov 		break;
30581d3153dSAlexander V. Chernikov 	case TOK_LOOKUP:
30681d3153dSAlexander V. Chernikov 		ac--; av++;
30781d3153dSAlexander V. Chernikov 		table_lookup(&oh, ac, av);
30881d3153dSAlexander V. Chernikov 		break;
309f1220db8SAlexander V. Chernikov 	}
310f1220db8SAlexander V. Chernikov }
311f1220db8SAlexander V. Chernikov 
312d8caf56eSAndrey V. Elsukov void
313d8caf56eSAndrey V. Elsukov table_fill_ntlv(ipfw_obj_ntlv *ntlv, const char *name, uint8_t set,
3142acdf79fSAndrey V. Elsukov     uint16_t uidx)
315f1220db8SAlexander V. Chernikov {
316f1220db8SAlexander V. Chernikov 
317563b5ab1SAlexander V. Chernikov 	ntlv->head.type = IPFW_TLV_TBL_NAME;
318f1220db8SAlexander V. Chernikov 	ntlv->head.length = sizeof(ipfw_obj_ntlv);
319f1220db8SAlexander V. Chernikov 	ntlv->idx = uidx;
320ac35ff17SAlexander V. Chernikov 	ntlv->set = set;
321f1220db8SAlexander V. Chernikov 	strlcpy(ntlv->name, name, sizeof(ntlv->name));
322f1220db8SAlexander V. Chernikov }
323f1220db8SAlexander V. Chernikov 
324f1220db8SAlexander V. Chernikov static void
325f1220db8SAlexander V. Chernikov table_fill_objheader(ipfw_obj_header *oh, ipfw_xtable_info *i)
326f1220db8SAlexander V. Chernikov {
327f1220db8SAlexander V. Chernikov 
328f1220db8SAlexander V. Chernikov 	oh->idx = 1;
32981d3153dSAlexander V. Chernikov 	table_fill_ntlv(&oh->ntlv, i->tablename, i->set, 1);
330ac35ff17SAlexander V. Chernikov }
331ac35ff17SAlexander V. Chernikov 
332ac35ff17SAlexander V. Chernikov static struct _s_x tablenewcmds[] = {
333ac35ff17SAlexander V. Chernikov       { "type",		TOK_TYPE },
334ac35ff17SAlexander V. Chernikov       { "valtype",	TOK_VALTYPE },
335ac35ff17SAlexander V. Chernikov       { "algo",		TOK_ALGO },
3364c0c07a5SAlexander V. Chernikov       { "limit",	TOK_LIMIT },
3374f43138aSAlexander V. Chernikov       { "locked",	TOK_LOCK },
33805ab1ef6SAndrey V. Elsukov       { "missing",	TOK_MISSING },
33905ab1ef6SAndrey V. Elsukov       { "or-flush",	TOK_ORFLUSH },
340ac35ff17SAlexander V. Chernikov       { NULL, 0 }
341ac35ff17SAlexander V. Chernikov };
342ac35ff17SAlexander V. Chernikov 
343914bffb6SAlexander V. Chernikov static struct _s_x flowtypecmds[] = {
344914bffb6SAlexander V. Chernikov       { "src-ip",	IPFW_TFFLAG_SRCIP },
345914bffb6SAlexander V. Chernikov       { "proto",	IPFW_TFFLAG_PROTO },
346914bffb6SAlexander V. Chernikov       { "src-port",	IPFW_TFFLAG_SRCPORT },
347914bffb6SAlexander V. Chernikov       { "dst-ip",	IPFW_TFFLAG_DSTIP },
348914bffb6SAlexander V. Chernikov       { "dst-port",	IPFW_TFFLAG_DSTPORT },
349914bffb6SAlexander V. Chernikov       { NULL, 0 }
350914bffb6SAlexander V. Chernikov };
351914bffb6SAlexander V. Chernikov 
35256707beeSMark Johnston static int
353914bffb6SAlexander V. Chernikov table_parse_type(uint8_t ttype, char *p, uint8_t *tflags)
354914bffb6SAlexander V. Chernikov {
3550cba2b28SAlexander V. Chernikov 	uint32_t fset, fclear;
3560cba2b28SAlexander V. Chernikov 	char *e;
357914bffb6SAlexander V. Chernikov 
358914bffb6SAlexander V. Chernikov 	/* Parse type options */
359914bffb6SAlexander V. Chernikov 	switch(ttype) {
360914bffb6SAlexander V. Chernikov 	case IPFW_TABLE_FLOW:
361914bffb6SAlexander V. Chernikov 		fset = fclear = 0;
3620cba2b28SAlexander V. Chernikov 		if (fill_flags(flowtypecmds, p, &e, &fset, &fclear) != 0)
3630cba2b28SAlexander V. Chernikov 			errx(EX_USAGE,
3640cba2b28SAlexander V. Chernikov 			    "unable to parse flow option %s", e);
365914bffb6SAlexander V. Chernikov 		*tflags = fset;
366914bffb6SAlexander V. Chernikov 		break;
367914bffb6SAlexander V. Chernikov 	default:
368914bffb6SAlexander V. Chernikov 		return (EX_USAGE);
369914bffb6SAlexander V. Chernikov 	}
370914bffb6SAlexander V. Chernikov 
371914bffb6SAlexander V. Chernikov 	return (0);
372914bffb6SAlexander V. Chernikov }
373914bffb6SAlexander V. Chernikov 
37456707beeSMark Johnston static void
375914bffb6SAlexander V. Chernikov table_print_type(char *tbuf, size_t size, uint8_t type, uint8_t tflags)
376914bffb6SAlexander V. Chernikov {
377914bffb6SAlexander V. Chernikov 	const char *tname;
378914bffb6SAlexander V. Chernikov 	int l;
379914bffb6SAlexander V. Chernikov 
380914bffb6SAlexander V. Chernikov 	if ((tname = match_value(tabletypes, type)) == NULL)
381914bffb6SAlexander V. Chernikov 		tname = "unknown";
382914bffb6SAlexander V. Chernikov 
383914bffb6SAlexander V. Chernikov 	l = snprintf(tbuf, size, "%s", tname);
384914bffb6SAlexander V. Chernikov 	tbuf += l;
385914bffb6SAlexander V. Chernikov 	size -= l;
386914bffb6SAlexander V. Chernikov 
387914bffb6SAlexander V. Chernikov 	switch(type) {
388914bffb6SAlexander V. Chernikov 	case IPFW_TABLE_FLOW:
389914bffb6SAlexander V. Chernikov 		if (tflags != 0) {
390914bffb6SAlexander V. Chernikov 			*tbuf++ = ':';
391914bffb6SAlexander V. Chernikov 			l--;
392914bffb6SAlexander V. Chernikov 			print_flags_buffer(tbuf, size, flowtypecmds, tflags);
393914bffb6SAlexander V. Chernikov 		}
394914bffb6SAlexander V. Chernikov 		break;
395914bffb6SAlexander V. Chernikov 	}
396914bffb6SAlexander V. Chernikov }
397914bffb6SAlexander V. Chernikov 
398ac35ff17SAlexander V. Chernikov /*
399ac35ff17SAlexander V. Chernikov  * Creates new table
400ac35ff17SAlexander V. Chernikov  *
401c21034b7SAlexander V. Chernikov  * ipfw table NAME create [ type { addr | iface | number | flow } ]
40205ab1ef6SAndrey V. Elsukov  *     [ algo algoname ] [missing] [or-flush]
403ac35ff17SAlexander V. Chernikov  */
404ac35ff17SAlexander V. Chernikov static void
405ac35ff17SAlexander V. Chernikov table_create(ipfw_obj_header *oh, int ac, char *av[])
406ac35ff17SAlexander V. Chernikov {
40705ab1ef6SAndrey V. Elsukov 	ipfw_xtable_info xi, xie;
40805ab1ef6SAndrey V. Elsukov 	int error, missing, orflush, tcmd, val;
4090cba2b28SAlexander V. Chernikov 	uint32_t fset, fclear;
4100cba2b28SAlexander V. Chernikov 	char *e, *p;
411ac35ff17SAlexander V. Chernikov 	char tbuf[128];
412ac35ff17SAlexander V. Chernikov 
41305ab1ef6SAndrey V. Elsukov 	missing = orflush = 0;
414ac35ff17SAlexander V. Chernikov 	memset(&xi, 0, sizeof(xi));
415ac35ff17SAlexander V. Chernikov 	while (ac > 0) {
416be695df9SAlexander V. Chernikov 		tcmd = get_token(tablenewcmds, *av, "option");
417ac35ff17SAlexander V. Chernikov 		ac--; av++;
418ac35ff17SAlexander V. Chernikov 
419ac35ff17SAlexander V. Chernikov 		switch (tcmd) {
4204c0c07a5SAlexander V. Chernikov 		case TOK_LIMIT:
4214c0c07a5SAlexander V. Chernikov 			NEED1("limit value required");
4224c0c07a5SAlexander V. Chernikov 			xi.limit = strtol(*av, NULL, 10);
4234c0c07a5SAlexander V. Chernikov 			ac--; av++;
4244c0c07a5SAlexander V. Chernikov 			break;
425ac35ff17SAlexander V. Chernikov 		case TOK_TYPE:
426ac35ff17SAlexander V. Chernikov 			NEED1("table type required");
427914bffb6SAlexander V. Chernikov 			/* Type may have suboptions after ':' */
428914bffb6SAlexander V. Chernikov 			if ((p = strchr(*av, ':')) != NULL)
429914bffb6SAlexander V. Chernikov 				*p++ = '\0';
430ac35ff17SAlexander V. Chernikov 			val = match_token(tabletypes, *av);
431914bffb6SAlexander V. Chernikov 			if (val == -1) {
432914bffb6SAlexander V. Chernikov 				concat_tokens(tbuf, sizeof(tbuf), tabletypes,
433914bffb6SAlexander V. Chernikov 				    ", ");
434914bffb6SAlexander V. Chernikov 				errx(EX_USAGE,
435914bffb6SAlexander V. Chernikov 				    "Unknown tabletype: %s. Supported: %s",
436ac35ff17SAlexander V. Chernikov 				    *av, tbuf);
437914bffb6SAlexander V. Chernikov 			}
438914bffb6SAlexander V. Chernikov 			xi.type = val;
439914bffb6SAlexander V. Chernikov 			if (p != NULL) {
440914bffb6SAlexander V. Chernikov 				error = table_parse_type(val, p, &xi.tflags);
441914bffb6SAlexander V. Chernikov 				if (error != 0)
442914bffb6SAlexander V. Chernikov 					errx(EX_USAGE,
443914bffb6SAlexander V. Chernikov 					    "Unsupported suboptions: %s", p);
444914bffb6SAlexander V. Chernikov 			}
445914bffb6SAlexander V. Chernikov 			ac--; av++;
446ac35ff17SAlexander V. Chernikov 			break;
447ac35ff17SAlexander V. Chernikov 		case TOK_VALTYPE:
448ac35ff17SAlexander V. Chernikov 			NEED1("table value type required");
4490cba2b28SAlexander V. Chernikov 			fset = fclear = 0;
4500cba2b28SAlexander V. Chernikov 			val = fill_flags(tablevaltypes, *av, &e, &fset, &fclear);
451ac35ff17SAlexander V. Chernikov 			if (val != -1) {
4520cba2b28SAlexander V. Chernikov 				xi.vmask = fset;
453ac35ff17SAlexander V. Chernikov 				ac--; av++;
454ac35ff17SAlexander V. Chernikov 				break;
455ac35ff17SAlexander V. Chernikov 			}
456ac35ff17SAlexander V. Chernikov 			concat_tokens(tbuf, sizeof(tbuf), tablevaltypes, ", ");
457ac35ff17SAlexander V. Chernikov 			errx(EX_USAGE, "Unknown value type: %s. Supported: %s",
4580cba2b28SAlexander V. Chernikov 			    e, tbuf);
459adf3b2b9SAlexander V. Chernikov 			break;
460ac35ff17SAlexander V. Chernikov 		case TOK_ALGO:
461ac35ff17SAlexander V. Chernikov 			NEED1("table algorithm name required");
462ac35ff17SAlexander V. Chernikov 			if (strlen(*av) > sizeof(xi.algoname))
463ac35ff17SAlexander V. Chernikov 				errx(EX_USAGE, "algorithm name too long");
464ac35ff17SAlexander V. Chernikov 			strlcpy(xi.algoname, *av, sizeof(xi.algoname));
465ac35ff17SAlexander V. Chernikov 			ac--; av++;
466ac35ff17SAlexander V. Chernikov 			break;
4674f43138aSAlexander V. Chernikov 		case TOK_LOCK:
4684f43138aSAlexander V. Chernikov 			xi.flags |= IPFW_TGFLAGS_LOCKED;
4694f43138aSAlexander V. Chernikov 			break;
47005ab1ef6SAndrey V. Elsukov 		case TOK_ORFLUSH:
47105ab1ef6SAndrey V. Elsukov 			orflush = 1;
47205ab1ef6SAndrey V. Elsukov 			/* FALLTHROUGH */
47305ab1ef6SAndrey V. Elsukov 		case TOK_MISSING:
47405ab1ef6SAndrey V. Elsukov 			missing = 1;
47505ab1ef6SAndrey V. Elsukov 			break;
476ac35ff17SAlexander V. Chernikov 		}
477ac35ff17SAlexander V. Chernikov 	}
478ac35ff17SAlexander V. Chernikov 
479463a577bSEitan Adler 	/* Set some defaults to preserve compatibility. */
480*13ad237aSAndrey V. Elsukov 	if (xi.algoname[0] == '\0') {
481*13ad237aSAndrey V. Elsukov 		const char *algo;
482*13ad237aSAndrey V. Elsukov 
483*13ad237aSAndrey V. Elsukov 		if (xi.type == 0)
484c21034b7SAlexander V. Chernikov 			xi.type = IPFW_TABLE_ADDR;
485*13ad237aSAndrey V. Elsukov 		algo = match_value(tablealgos, xi.type);
486*13ad237aSAndrey V. Elsukov 		if (algo != NULL)
487*13ad237aSAndrey V. Elsukov 			strlcpy(xi.algoname, algo, sizeof(xi.algoname));
488*13ad237aSAndrey V. Elsukov 	}
4890cba2b28SAlexander V. Chernikov 	if (xi.vmask == 0)
4900cba2b28SAlexander V. Chernikov 		xi.vmask = IPFW_VTYPE_LEGACY;
491fd0869d5SAlexander V. Chernikov 
49205ab1ef6SAndrey V. Elsukov 	error = table_do_create(oh, &xi);
49305ab1ef6SAndrey V. Elsukov 
49405ab1ef6SAndrey V. Elsukov 	if (error == 0)
49505ab1ef6SAndrey V. Elsukov 		return;
49605ab1ef6SAndrey V. Elsukov 
49705ab1ef6SAndrey V. Elsukov 	if (errno != EEXIST || missing == 0)
498ac35ff17SAlexander V. Chernikov 		err(EX_OSERR, "Table creation failed");
49905ab1ef6SAndrey V. Elsukov 
50005ab1ef6SAndrey V. Elsukov 	/* Check that existing table is the same we are trying to create */
50105ab1ef6SAndrey V. Elsukov 	if (table_get_info(oh, &xie) != 0)
50205ab1ef6SAndrey V. Elsukov 		err(EX_OSERR, "Existing table check failed");
50305ab1ef6SAndrey V. Elsukov 
50405ab1ef6SAndrey V. Elsukov 	if (xi.limit != xie.limit || xi.type != xie.type ||
50505ab1ef6SAndrey V. Elsukov 	    xi.tflags != xie.tflags || xi.vmask != xie.vmask || (
50605ab1ef6SAndrey V. Elsukov 	    xi.algoname[0] != '\0' && strcmp(xi.algoname,
50705ab1ef6SAndrey V. Elsukov 	    xie.algoname) != 0) || xi.flags != xie.flags)
50805ab1ef6SAndrey V. Elsukov 		errx(EX_DATAERR, "The existing table is not compatible "
50905ab1ef6SAndrey V. Elsukov 		    "with one you are creating.");
51005ab1ef6SAndrey V. Elsukov 
51105ab1ef6SAndrey V. Elsukov 	/* Flush existing table if instructed to do so */
51205ab1ef6SAndrey V. Elsukov 	if (orflush != 0 && table_flush(oh) != 0)
51305ab1ef6SAndrey V. Elsukov 		err(EX_OSERR, "Table flush on creation failed");
514f1220db8SAlexander V. Chernikov }
515f1220db8SAlexander V. Chernikov 
516f1220db8SAlexander V. Chernikov /*
517ac35ff17SAlexander V. Chernikov  * Creates new table
518ac35ff17SAlexander V. Chernikov  *
519ac35ff17SAlexander V. Chernikov  * Request: [ ipfw_obj_header ipfw_xtable_info ]
520ac35ff17SAlexander V. Chernikov  *
521f1220db8SAlexander V. Chernikov  * Returns 0 on success.
522f1220db8SAlexander V. Chernikov  */
523f1220db8SAlexander V. Chernikov static int
524ac35ff17SAlexander V. Chernikov table_do_create(ipfw_obj_header *oh, ipfw_xtable_info *i)
525f1220db8SAlexander V. Chernikov {
526ac35ff17SAlexander V. Chernikov 	char tbuf[sizeof(ipfw_obj_header) + sizeof(ipfw_xtable_info)];
527ac35ff17SAlexander V. Chernikov 	int error;
528f1220db8SAlexander V. Chernikov 
529ac35ff17SAlexander V. Chernikov 	memcpy(tbuf, oh, sizeof(*oh));
530ac35ff17SAlexander V. Chernikov 	memcpy(tbuf + sizeof(*oh), i, sizeof(*i));
531ac35ff17SAlexander V. Chernikov 	oh = (ipfw_obj_header *)tbuf;
532ac35ff17SAlexander V. Chernikov 
533ac35ff17SAlexander V. Chernikov 	error = do_set3(IP_FW_TABLE_XCREATE, &oh->opheader, sizeof(tbuf));
534ac35ff17SAlexander V. Chernikov 
535ac35ff17SAlexander V. Chernikov 	return (error);
536ac35ff17SAlexander V. Chernikov }
537ac35ff17SAlexander V. Chernikov 
538ac35ff17SAlexander V. Chernikov /*
539adf3b2b9SAlexander V. Chernikov  * Modifies existing table
540adf3b2b9SAlexander V. Chernikov  *
5410cba2b28SAlexander V. Chernikov  * ipfw table NAME modify [ limit number ]
542adf3b2b9SAlexander V. Chernikov  */
543adf3b2b9SAlexander V. Chernikov static void
544adf3b2b9SAlexander V. Chernikov table_modify(ipfw_obj_header *oh, int ac, char *av[])
545adf3b2b9SAlexander V. Chernikov {
546adf3b2b9SAlexander V. Chernikov 	ipfw_xtable_info xi;
547b6462881SAlexander V. Chernikov 	int tcmd;
548adf3b2b9SAlexander V. Chernikov 
549adf3b2b9SAlexander V. Chernikov 	memset(&xi, 0, sizeof(xi));
550adf3b2b9SAlexander V. Chernikov 
551adf3b2b9SAlexander V. Chernikov 	while (ac > 0) {
552be695df9SAlexander V. Chernikov 		tcmd = get_token(tablenewcmds, *av, "option");
553adf3b2b9SAlexander V. Chernikov 		ac--; av++;
554adf3b2b9SAlexander V. Chernikov 
555adf3b2b9SAlexander V. Chernikov 		switch (tcmd) {
556adf3b2b9SAlexander V. Chernikov 		case TOK_LIMIT:
557adf3b2b9SAlexander V. Chernikov 			NEED1("limit value required");
558adf3b2b9SAlexander V. Chernikov 			xi.limit = strtol(*av, NULL, 10);
559adf3b2b9SAlexander V. Chernikov 			xi.mflags |= IPFW_TMFLAGS_LIMIT;
560adf3b2b9SAlexander V. Chernikov 			ac--; av++;
561adf3b2b9SAlexander V. Chernikov 			break;
5620cba2b28SAlexander V. Chernikov 		default:
5636ef82c0aSAlan Somers 			errx(EX_USAGE, "cmd is not supported for modification");
564adf3b2b9SAlexander V. Chernikov 		}
565adf3b2b9SAlexander V. Chernikov 	}
566adf3b2b9SAlexander V. Chernikov 
567b6462881SAlexander V. Chernikov 	if (table_do_modify(oh, &xi) != 0)
568adf3b2b9SAlexander V. Chernikov 		err(EX_OSERR, "Table modification failed");
569adf3b2b9SAlexander V. Chernikov }
570adf3b2b9SAlexander V. Chernikov 
571adf3b2b9SAlexander V. Chernikov /*
572adf3b2b9SAlexander V. Chernikov  * Modifies existing table.
573adf3b2b9SAlexander V. Chernikov  *
574adf3b2b9SAlexander V. Chernikov  * Request: [ ipfw_obj_header ipfw_xtable_info ]
575adf3b2b9SAlexander V. Chernikov  *
576adf3b2b9SAlexander V. Chernikov  * Returns 0 on success.
577adf3b2b9SAlexander V. Chernikov  */
578adf3b2b9SAlexander V. Chernikov static int
579adf3b2b9SAlexander V. Chernikov table_do_modify(ipfw_obj_header *oh, ipfw_xtable_info *i)
580adf3b2b9SAlexander V. Chernikov {
581adf3b2b9SAlexander V. Chernikov 	char tbuf[sizeof(ipfw_obj_header) + sizeof(ipfw_xtable_info)];
582adf3b2b9SAlexander V. Chernikov 	int error;
583adf3b2b9SAlexander V. Chernikov 
584adf3b2b9SAlexander V. Chernikov 	memcpy(tbuf, oh, sizeof(*oh));
585adf3b2b9SAlexander V. Chernikov 	memcpy(tbuf + sizeof(*oh), i, sizeof(*i));
586adf3b2b9SAlexander V. Chernikov 	oh = (ipfw_obj_header *)tbuf;
587adf3b2b9SAlexander V. Chernikov 
588adf3b2b9SAlexander V. Chernikov 	error = do_set3(IP_FW_TABLE_XMODIFY, &oh->opheader, sizeof(tbuf));
589adf3b2b9SAlexander V. Chernikov 
590adf3b2b9SAlexander V. Chernikov 	return (error);
591adf3b2b9SAlexander V. Chernikov }
5924f43138aSAlexander V. Chernikov 
5934f43138aSAlexander V. Chernikov /*
5944f43138aSAlexander V. Chernikov  * Locks or unlocks given table
5954f43138aSAlexander V. Chernikov  */
5964f43138aSAlexander V. Chernikov static void
5974f43138aSAlexander V. Chernikov table_lock(ipfw_obj_header *oh, int lock)
5984f43138aSAlexander V. Chernikov {
5994f43138aSAlexander V. Chernikov 	ipfw_xtable_info xi;
6004f43138aSAlexander V. Chernikov 
6014f43138aSAlexander V. Chernikov 	memset(&xi, 0, sizeof(xi));
6024f43138aSAlexander V. Chernikov 
6034f43138aSAlexander V. Chernikov 	xi.mflags |= IPFW_TMFLAGS_LOCK;
6044f43138aSAlexander V. Chernikov 	xi.flags |= (lock != 0) ? IPFW_TGFLAGS_LOCKED : 0;
6054f43138aSAlexander V. Chernikov 
606b6462881SAlexander V. Chernikov 	if (table_do_modify(oh, &xi) != 0)
6074f43138aSAlexander V. Chernikov 		err(EX_OSERR, "Table %s failed", lock != 0 ? "lock" : "unlock");
6084f43138aSAlexander V. Chernikov }
6094f43138aSAlexander V. Chernikov 
610adf3b2b9SAlexander V. Chernikov /*
611ac35ff17SAlexander V. Chernikov  * Destroys given table specified by @oh->ntlv.
612ac35ff17SAlexander V. Chernikov  * Returns 0 on success.
613ac35ff17SAlexander V. Chernikov  */
614ac35ff17SAlexander V. Chernikov static int
615ac35ff17SAlexander V. Chernikov table_destroy(ipfw_obj_header *oh)
616ac35ff17SAlexander V. Chernikov {
617ac35ff17SAlexander V. Chernikov 
618ac35ff17SAlexander V. Chernikov 	if (do_set3(IP_FW_TABLE_XDESTROY, &oh->opheader, sizeof(*oh)) != 0)
619f1220db8SAlexander V. Chernikov 		return (-1);
620f1220db8SAlexander V. Chernikov 
621f1220db8SAlexander V. Chernikov 	return (0);
622f1220db8SAlexander V. Chernikov }
623f1220db8SAlexander V. Chernikov 
624421c5838SAndrey V. Elsukov static int
625421c5838SAndrey V. Elsukov table_destroy_one(ipfw_xtable_info *i, void *arg)
626421c5838SAndrey V. Elsukov {
627421c5838SAndrey V. Elsukov 	ipfw_obj_header *oh;
628421c5838SAndrey V. Elsukov 
629421c5838SAndrey V. Elsukov 	oh = (ipfw_obj_header *)arg;
630421c5838SAndrey V. Elsukov 	table_fill_ntlv(&oh->ntlv, i->tablename, i->set, 1);
631421c5838SAndrey V. Elsukov 	if (table_destroy(oh) != 0) {
63256707beeSMark Johnston 		if (g_co.do_quiet == 0)
633421c5838SAndrey V. Elsukov 			warn("failed to destroy table(%s) in set %u",
634421c5838SAndrey V. Elsukov 			    i->tablename, i->set);
635421c5838SAndrey V. Elsukov 		return (-1);
636421c5838SAndrey V. Elsukov 	}
637421c5838SAndrey V. Elsukov 	return (0);
638421c5838SAndrey V. Elsukov }
639421c5838SAndrey V. Elsukov 
640f1220db8SAlexander V. Chernikov /*
641ac35ff17SAlexander V. Chernikov  * Flushes given table specified by @oh->ntlv.
642f1220db8SAlexander V. Chernikov  * Returns 0 on success.
643f1220db8SAlexander V. Chernikov  */
644f1220db8SAlexander V. Chernikov static int
645ac35ff17SAlexander V. Chernikov table_flush(ipfw_obj_header *oh)
646f1220db8SAlexander V. Chernikov {
647f1220db8SAlexander V. Chernikov 
648ac35ff17SAlexander V. Chernikov 	if (do_set3(IP_FW_TABLE_XFLUSH, &oh->opheader, sizeof(*oh)) != 0)
649f1220db8SAlexander V. Chernikov 		return (-1);
650f1220db8SAlexander V. Chernikov 
651f1220db8SAlexander V. Chernikov 	return (0);
652f1220db8SAlexander V. Chernikov }
653f1220db8SAlexander V. Chernikov 
65446d52008SAlexander V. Chernikov static int
65546d52008SAlexander V. Chernikov table_do_swap(ipfw_obj_header *oh, char *second)
65646d52008SAlexander V. Chernikov {
65746d52008SAlexander V. Chernikov 	char tbuf[sizeof(ipfw_obj_header) + sizeof(ipfw_obj_ntlv)];
65846d52008SAlexander V. Chernikov 	int error;
65946d52008SAlexander V. Chernikov 
66046d52008SAlexander V. Chernikov 	memset(tbuf, 0, sizeof(tbuf));
66146d52008SAlexander V. Chernikov 	memcpy(tbuf, oh, sizeof(*oh));
66246d52008SAlexander V. Chernikov 	oh = (ipfw_obj_header *)tbuf;
66346d52008SAlexander V. Chernikov 	table_fill_ntlv((ipfw_obj_ntlv *)(oh + 1), second, oh->ntlv.set, 1);
66446d52008SAlexander V. Chernikov 
66546d52008SAlexander V. Chernikov 	error = do_set3(IP_FW_TABLE_XSWAP, &oh->opheader, sizeof(tbuf));
66646d52008SAlexander V. Chernikov 
66746d52008SAlexander V. Chernikov 	return (error);
66846d52008SAlexander V. Chernikov }
66946d52008SAlexander V. Chernikov 
67046d52008SAlexander V. Chernikov /*
67146d52008SAlexander V. Chernikov  * Swaps given table with @second one.
67246d52008SAlexander V. Chernikov  */
67346d52008SAlexander V. Chernikov static int
67446d52008SAlexander V. Chernikov table_swap(ipfw_obj_header *oh, char *second)
67546d52008SAlexander V. Chernikov {
67646d52008SAlexander V. Chernikov 
67746d52008SAlexander V. Chernikov 	if (table_check_name(second) != 0)
67846d52008SAlexander V. Chernikov 		errx(EX_USAGE, "table name %s is invalid", second);
67946d52008SAlexander V. Chernikov 
68076d03257SAndrey V. Elsukov 	if (table_do_swap(oh, second) == 0)
68176d03257SAndrey V. Elsukov 		return (0);
68246d52008SAlexander V. Chernikov 
68376d03257SAndrey V. Elsukov 	switch (errno) {
68446d52008SAlexander V. Chernikov 	case EINVAL:
68546d52008SAlexander V. Chernikov 		errx(EX_USAGE, "Unable to swap table: check types");
68646d52008SAlexander V. Chernikov 	case EFBIG:
68746d52008SAlexander V. Chernikov 		errx(EX_USAGE, "Unable to swap table: check limits");
68846d52008SAlexander V. Chernikov 	}
68946d52008SAlexander V. Chernikov 
69046d52008SAlexander V. Chernikov 	return (0);
69146d52008SAlexander V. Chernikov }
69246d52008SAlexander V. Chernikov 
69346d52008SAlexander V. Chernikov 
694f1220db8SAlexander V. Chernikov /*
695ac35ff17SAlexander V. Chernikov  * Retrieves table in given table specified by @oh->ntlv.
696f1220db8SAlexander V. Chernikov  * it inside @i.
697f1220db8SAlexander V. Chernikov  * Returns 0 on success.
698f1220db8SAlexander V. Chernikov  */
699f1220db8SAlexander V. Chernikov static int
700ac35ff17SAlexander V. Chernikov table_get_info(ipfw_obj_header *oh, ipfw_xtable_info *i)
701f1220db8SAlexander V. Chernikov {
702f1220db8SAlexander V. Chernikov 	char tbuf[sizeof(ipfw_obj_header) + sizeof(ipfw_xtable_info)];
703f1220db8SAlexander V. Chernikov 	size_t sz;
704f1220db8SAlexander V. Chernikov 
705f1220db8SAlexander V. Chernikov 	sz = sizeof(tbuf);
706f1220db8SAlexander V. Chernikov 	memset(tbuf, 0, sizeof(tbuf));
707ac35ff17SAlexander V. Chernikov 	memcpy(tbuf, oh, sizeof(*oh));
708f1220db8SAlexander V. Chernikov 	oh = (ipfw_obj_header *)tbuf;
709f1220db8SAlexander V. Chernikov 
710b6462881SAlexander V. Chernikov 	if (do_get3(IP_FW_TABLE_XINFO, &oh->opheader, &sz) != 0)
711b6462881SAlexander V. Chernikov 		return (errno);
712f1220db8SAlexander V. Chernikov 
713f1220db8SAlexander V. Chernikov 	if (sz < sizeof(tbuf))
714ac35ff17SAlexander V. Chernikov 		return (EINVAL);
715f1220db8SAlexander V. Chernikov 
716f1220db8SAlexander V. Chernikov 	*i = *(ipfw_xtable_info *)(oh + 1);
717f1220db8SAlexander V. Chernikov 
718f1220db8SAlexander V. Chernikov 	return (0);
719f1220db8SAlexander V. Chernikov }
720f1220db8SAlexander V. Chernikov 
7215f379342SAlexander V. Chernikov static struct _s_x tablealgoclass[] = {
7225f379342SAlexander V. Chernikov       { "hash",		IPFW_TACLASS_HASH },
7235f379342SAlexander V. Chernikov       { "array",	IPFW_TACLASS_ARRAY },
7245f379342SAlexander V. Chernikov       { "radix",	IPFW_TACLASS_RADIX },
7255f379342SAlexander V. Chernikov       { NULL, 0 }
7265f379342SAlexander V. Chernikov };
7275f379342SAlexander V. Chernikov 
7285f379342SAlexander V. Chernikov struct ta_cldata {
7295f379342SAlexander V. Chernikov 	uint8_t		taclass;
7305f379342SAlexander V. Chernikov 	uint8_t		spare4;
7315f379342SAlexander V. Chernikov 	uint16_t	itemsize;
7325f379342SAlexander V. Chernikov 	uint16_t	itemsize6;
7335f379342SAlexander V. Chernikov 	uint32_t	size;
7345f379342SAlexander V. Chernikov 	uint32_t	count;
7355f379342SAlexander V. Chernikov };
7365f379342SAlexander V. Chernikov 
7375f379342SAlexander V. Chernikov /*
7385f379342SAlexander V. Chernikov  * Print global/per-AF table @i algorithm info.
7395f379342SAlexander V. Chernikov  */
7405f379342SAlexander V. Chernikov static void
74156707beeSMark Johnston table_show_tainfo(ipfw_xtable_info *i __unused, struct ta_cldata *d,
7425f379342SAlexander V. Chernikov     const char *af, const char *taclass)
7435f379342SAlexander V. Chernikov {
7445f379342SAlexander V. Chernikov 
7455f379342SAlexander V. Chernikov 	switch (d->taclass) {
7465f379342SAlexander V. Chernikov 	case IPFW_TACLASS_HASH:
7475f379342SAlexander V. Chernikov 	case IPFW_TACLASS_ARRAY:
7485f379342SAlexander V. Chernikov 		printf(" %salgorithm %s info\n", af, taclass);
7495f379342SAlexander V. Chernikov 		if (d->itemsize == d->itemsize6)
7505f379342SAlexander V. Chernikov 			printf("  size: %u items: %u itemsize: %u\n",
7515f379342SAlexander V. Chernikov 			    d->size, d->count, d->itemsize);
7525f379342SAlexander V. Chernikov 		else
7535f379342SAlexander V. Chernikov 			printf("  size: %u items: %u "
7545f379342SAlexander V. Chernikov 			    "itemsize4: %u itemsize6: %u\n",
7555f379342SAlexander V. Chernikov 			    d->size, d->count,
7565f379342SAlexander V. Chernikov 			    d->itemsize, d->itemsize6);
7575f379342SAlexander V. Chernikov 		break;
7585f379342SAlexander V. Chernikov 	case IPFW_TACLASS_RADIX:
7595f379342SAlexander V. Chernikov 		printf(" %salgorithm %s info\n", af, taclass);
7605f379342SAlexander V. Chernikov 		if (d->itemsize == d->itemsize6)
7615f379342SAlexander V. Chernikov 			printf("  items: %u itemsize: %u\n",
7625f379342SAlexander V. Chernikov 			    d->count, d->itemsize);
7635f379342SAlexander V. Chernikov 		else
7645f379342SAlexander V. Chernikov 			printf("  items: %u "
7655f379342SAlexander V. Chernikov 			    "itemsize4: %u itemsize6: %u\n",
7665f379342SAlexander V. Chernikov 			    d->count, d->itemsize, d->itemsize6);
7675f379342SAlexander V. Chernikov 		break;
7685f379342SAlexander V. Chernikov 	default:
7695f379342SAlexander V. Chernikov 		printf(" algo class: %s\n", taclass);
7705f379342SAlexander V. Chernikov 	}
7715f379342SAlexander V. Chernikov }
7725f379342SAlexander V. Chernikov 
7730cba2b28SAlexander V. Chernikov static void
7740cba2b28SAlexander V. Chernikov table_print_valheader(char *buf, size_t bufsize, uint32_t vmask)
7750cba2b28SAlexander V. Chernikov {
7760cba2b28SAlexander V. Chernikov 
7770cba2b28SAlexander V. Chernikov 	if (vmask == IPFW_VTYPE_LEGACY) {
7780cba2b28SAlexander V. Chernikov 		snprintf(buf, bufsize, "legacy");
7790cba2b28SAlexander V. Chernikov 		return;
7800cba2b28SAlexander V. Chernikov 	}
7810cba2b28SAlexander V. Chernikov 
7828a6dbb64SAlexander V. Chernikov 	memset(buf, 0, bufsize);
7830cba2b28SAlexander V. Chernikov 	print_flags_buffer(buf, bufsize, tablevaltypes, vmask);
7840cba2b28SAlexander V. Chernikov }
7850cba2b28SAlexander V. Chernikov 
786f1220db8SAlexander V. Chernikov /*
787f1220db8SAlexander V. Chernikov  * Prints table info struct @i in human-readable form.
788f1220db8SAlexander V. Chernikov  */
789f1220db8SAlexander V. Chernikov static int
790f1220db8SAlexander V. Chernikov table_show_info(ipfw_xtable_info *i, void *arg)
791f1220db8SAlexander V. Chernikov {
7920cba2b28SAlexander V. Chernikov 	const char *vtype;
7935f379342SAlexander V. Chernikov 	ipfw_ta_tinfo *tainfo;
7945f379342SAlexander V. Chernikov 	int afdata, afitem;
7955f379342SAlexander V. Chernikov 	struct ta_cldata d;
796adf3b2b9SAlexander V. Chernikov 	char ttype[64], tvtype[64];
797f1220db8SAlexander V. Chernikov 
798914bffb6SAlexander V. Chernikov 	table_print_type(ttype, sizeof(ttype), i->type, i->tflags);
7990cba2b28SAlexander V. Chernikov 	table_print_valheader(tvtype, sizeof(tvtype), i->vmask);
800ac35ff17SAlexander V. Chernikov 
801914bffb6SAlexander V. Chernikov 	printf("--- table(%s), set(%u) ---\n", i->tablename, i->set);
8024f43138aSAlexander V. Chernikov 	if ((i->flags & IPFW_TGFLAGS_LOCKED) != 0)
8034f43138aSAlexander V. Chernikov 		printf(" kindex: %d, type: %s, locked\n", i->kidx, ttype);
8044f43138aSAlexander V. Chernikov 	else
805914bffb6SAlexander V. Chernikov 		printf(" kindex: %d, type: %s\n", i->kidx, ttype);
8060cba2b28SAlexander V. Chernikov 	printf(" references: %u, valtype: %s\n", i->refcnt, tvtype);
8079d099b4fSAlexander V. Chernikov 	printf(" algorithm: %s\n", i->algoname);
808f1220db8SAlexander V. Chernikov 	printf(" items: %u, size: %u\n", i->count, i->size);
8094c0c07a5SAlexander V. Chernikov 	if (i->limit > 0)
8104c0c07a5SAlexander V. Chernikov 		printf(" limit: %u\n", i->limit);
811f1220db8SAlexander V. Chernikov 
812358b9d09SAlexander V. Chernikov 	/* Print algo-specific info if requested & set  */
813358b9d09SAlexander V. Chernikov 	if (arg == NULL)
814358b9d09SAlexander V. Chernikov 		return (0);
815358b9d09SAlexander V. Chernikov 
8165f379342SAlexander V. Chernikov 	if ((i->ta_info.flags & IPFW_TATFLAGS_DATA) == 0)
8175f379342SAlexander V. Chernikov 		return (0);
8185f379342SAlexander V. Chernikov 	tainfo = &i->ta_info;
8195f379342SAlexander V. Chernikov 
8205f379342SAlexander V. Chernikov 	afdata = 0;
8215f379342SAlexander V. Chernikov 	afitem = 0;
8225f379342SAlexander V. Chernikov 	if (tainfo->flags & IPFW_TATFLAGS_AFDATA)
8235f379342SAlexander V. Chernikov 		afdata = 1;
8245f379342SAlexander V. Chernikov 	if (tainfo->flags & IPFW_TATFLAGS_AFITEM)
8255f379342SAlexander V. Chernikov 		afitem = 1;
8265f379342SAlexander V. Chernikov 
8275f379342SAlexander V. Chernikov 	memset(&d, 0, sizeof(d));
8285f379342SAlexander V. Chernikov 	d.taclass = tainfo->taclass4;
8295f379342SAlexander V. Chernikov 	d.size = tainfo->size4;
8305f379342SAlexander V. Chernikov 	d.count = tainfo->count4;
8315f379342SAlexander V. Chernikov 	d.itemsize = tainfo->itemsize4;
8325f379342SAlexander V. Chernikov 	if (afdata == 0 && afitem != 0)
8335f379342SAlexander V. Chernikov 		d.itemsize6 = tainfo->itemsize6;
8345f379342SAlexander V. Chernikov 	else
8355f379342SAlexander V. Chernikov 		d.itemsize6 = d.itemsize;
8365f379342SAlexander V. Chernikov 	if ((vtype = match_value(tablealgoclass, d.taclass)) == NULL)
8375f379342SAlexander V. Chernikov 		vtype = "unknown";
8385f379342SAlexander V. Chernikov 
8395f379342SAlexander V. Chernikov 	if (afdata == 0) {
8405f379342SAlexander V. Chernikov 		table_show_tainfo(i, &d, "", vtype);
8415f379342SAlexander V. Chernikov 	} else {
8425f379342SAlexander V. Chernikov 		table_show_tainfo(i, &d, "IPv4 ", vtype);
8435f379342SAlexander V. Chernikov 		memset(&d, 0, sizeof(d));
8445f379342SAlexander V. Chernikov 		d.taclass = tainfo->taclass6;
8455f379342SAlexander V. Chernikov 		if ((vtype = match_value(tablealgoclass, d.taclass)) == NULL)
8465f379342SAlexander V. Chernikov 			vtype = "unknown";
8475f379342SAlexander V. Chernikov 		d.size = tainfo->size6;
8485f379342SAlexander V. Chernikov 		d.count = tainfo->count6;
8495f379342SAlexander V. Chernikov 		d.itemsize = tainfo->itemsize6;
8505f379342SAlexander V. Chernikov 		d.itemsize6 = d.itemsize;
8515f379342SAlexander V. Chernikov 		table_show_tainfo(i, &d, "IPv6 ", vtype);
8525f379342SAlexander V. Chernikov 	}
8535f379342SAlexander V. Chernikov 
854f1220db8SAlexander V. Chernikov 	return (0);
855f1220db8SAlexander V. Chernikov }
856f1220db8SAlexander V. Chernikov 
857f1220db8SAlexander V. Chernikov 
858f1220db8SAlexander V. Chernikov /*
859f1220db8SAlexander V. Chernikov  * Function wrappers which can be used either
860f1220db8SAlexander V. Chernikov  * as is or as foreach function parameter.
861f1220db8SAlexander V. Chernikov  */
862f1220db8SAlexander V. Chernikov 
863f1220db8SAlexander V. Chernikov static int
864f1220db8SAlexander V. Chernikov table_show_one(ipfw_xtable_info *i, void *arg)
865f1220db8SAlexander V. Chernikov {
866268a20a0SAdrian Chadd 	ipfw_obj_header *oh = NULL;
86781d3153dSAlexander V. Chernikov 	int error;
86810e3bebfSBryan Drewery 	int is_all;
86910e3bebfSBryan Drewery 
87010e3bebfSBryan Drewery 	is_all = arg == NULL ? 0 : 1;
871f1220db8SAlexander V. Chernikov 
872720ee730SAlexander V. Chernikov 	if ((error = table_do_get_list(i, &oh)) != 0) {
87381d3153dSAlexander V. Chernikov 		err(EX_OSERR, "Error requesting table %s list", i->tablename);
87481d3153dSAlexander V. Chernikov 		return (error);
87581d3153dSAlexander V. Chernikov 	}
87681d3153dSAlexander V. Chernikov 
8772b3d6647SBryan Drewery 	table_show_list(oh, is_all);
878f1220db8SAlexander V. Chernikov 
879f1220db8SAlexander V. Chernikov 	free(oh);
880f1220db8SAlexander V. Chernikov 	return (0);
881f1220db8SAlexander V. Chernikov }
882f1220db8SAlexander V. Chernikov 
883f1220db8SAlexander V. Chernikov static int
884f1220db8SAlexander V. Chernikov table_flush_one(ipfw_xtable_info *i, void *arg)
885f1220db8SAlexander V. Chernikov {
886ac35ff17SAlexander V. Chernikov 	ipfw_obj_header *oh;
887f1220db8SAlexander V. Chernikov 
888ac35ff17SAlexander V. Chernikov 	oh = (ipfw_obj_header *)arg;
889ac35ff17SAlexander V. Chernikov 
890ac35ff17SAlexander V. Chernikov 	table_fill_ntlv(&oh->ntlv, i->tablename, i->set, 1);
891ac35ff17SAlexander V. Chernikov 
892ac35ff17SAlexander V. Chernikov 	return (table_flush(oh));
893f1220db8SAlexander V. Chernikov }
894f1220db8SAlexander V. Chernikov 
895ac35ff17SAlexander V. Chernikov static int
896ac35ff17SAlexander V. Chernikov table_do_modify_record(int cmd, ipfw_obj_header *oh,
8973a845e10SAlexander V. Chernikov     ipfw_obj_tentry *tent, int count, int atomic)
898ac35ff17SAlexander V. Chernikov {
899db785d31SAlexander V. Chernikov 	ipfw_obj_ctlv *ctlv;
9003a845e10SAlexander V. Chernikov 	ipfw_obj_tentry *tent_base;
9013a845e10SAlexander V. Chernikov 	caddr_t pbuf;
902db785d31SAlexander V. Chernikov 	char xbuf[sizeof(*oh) + sizeof(ipfw_obj_ctlv) + sizeof(*tent)];
9033a845e10SAlexander V. Chernikov 	int error, i;
9043a845e10SAlexander V. Chernikov 	size_t sz;
905ac35ff17SAlexander V. Chernikov 
9063a845e10SAlexander V. Chernikov 	sz = sizeof(*ctlv) + sizeof(*tent) * count;
9073a845e10SAlexander V. Chernikov 	if (count == 1) {
908ac35ff17SAlexander V. Chernikov 		memset(xbuf, 0, sizeof(xbuf));
9093a845e10SAlexander V. Chernikov 		pbuf = xbuf;
9103a845e10SAlexander V. Chernikov 	} else {
9113a845e10SAlexander V. Chernikov 		if ((pbuf = calloc(1, sizeof(*oh) + sz)) == NULL)
9123a845e10SAlexander V. Chernikov 			return (ENOMEM);
9133a845e10SAlexander V. Chernikov 	}
9143a845e10SAlexander V. Chernikov 
9153a845e10SAlexander V. Chernikov 	memcpy(pbuf, oh, sizeof(*oh));
9163a845e10SAlexander V. Chernikov 	oh = (ipfw_obj_header *)pbuf;
917ac35ff17SAlexander V. Chernikov 	oh->opheader.version = 1;
918ac35ff17SAlexander V. Chernikov 
919db785d31SAlexander V. Chernikov 	ctlv = (ipfw_obj_ctlv *)(oh + 1);
9203a845e10SAlexander V. Chernikov 	ctlv->count = count;
9213a845e10SAlexander V. Chernikov 	ctlv->head.length = sz;
9223a845e10SAlexander V. Chernikov 	if (atomic != 0)
9233a845e10SAlexander V. Chernikov 		ctlv->flags |= IPFW_CTF_ATOMIC;
924db785d31SAlexander V. Chernikov 
9253a845e10SAlexander V. Chernikov 	tent_base = tent;
9263a845e10SAlexander V. Chernikov 	memcpy(ctlv + 1, tent, sizeof(*tent) * count);
927db785d31SAlexander V. Chernikov 	tent = (ipfw_obj_tentry *)(ctlv + 1);
9283a845e10SAlexander V. Chernikov 	for (i = 0; i < count; i++, tent++) {
929ac35ff17SAlexander V. Chernikov 		tent->head.length = sizeof(ipfw_obj_tentry);
9303a845e10SAlexander V. Chernikov 		tent->idx = oh->idx;
9313a845e10SAlexander V. Chernikov 	}
932ac35ff17SAlexander V. Chernikov 
9333a845e10SAlexander V. Chernikov 	sz += sizeof(*oh);
9343a845e10SAlexander V. Chernikov 	error = do_get3(cmd, &oh->opheader, &sz);
935092f8ba3SAndrey V. Elsukov 	if (error != 0)
936092f8ba3SAndrey V. Elsukov 		error = errno;
9373a845e10SAlexander V. Chernikov 	tent = (ipfw_obj_tentry *)(ctlv + 1);
9383a845e10SAlexander V. Chernikov 	/* Copy result back to provided buffer */
9393a845e10SAlexander V. Chernikov 	memcpy(tent_base, ctlv + 1, sizeof(*tent) * count);
9403a845e10SAlexander V. Chernikov 
9413a845e10SAlexander V. Chernikov 	if (pbuf != xbuf)
9423a845e10SAlexander V. Chernikov 		free(pbuf);
943ac35ff17SAlexander V. Chernikov 
944ac35ff17SAlexander V. Chernikov 	return (error);
945ac35ff17SAlexander V. Chernikov }
946ac35ff17SAlexander V. Chernikov 
947ac35ff17SAlexander V. Chernikov static void
9483a845e10SAlexander V. Chernikov table_modify_record(ipfw_obj_header *oh, int ac, char *av[], int add,
9493a845e10SAlexander V. Chernikov     int quiet, int update, int atomic)
950ac35ff17SAlexander V. Chernikov {
9513a845e10SAlexander V. Chernikov 	ipfw_obj_tentry *ptent, tent, *tent_buf;
95281d3153dSAlexander V. Chernikov 	ipfw_xtable_info xi;
95356707beeSMark Johnston 	const char *etxt, *px, *texterr;
9540cba2b28SAlexander V. Chernikov 	uint8_t type;
9550cba2b28SAlexander V. Chernikov 	uint32_t vmask;
9563a845e10SAlexander V. Chernikov 	int cmd, count, error, i, ignored;
957ac35ff17SAlexander V. Chernikov 
958ac35ff17SAlexander V. Chernikov 	if (ac == 0)
959ac35ff17SAlexander V. Chernikov 		errx(EX_USAGE, "address required");
960ac35ff17SAlexander V. Chernikov 
961ac35ff17SAlexander V. Chernikov 	if (add != 0) {
962ac35ff17SAlexander V. Chernikov 		cmd = IP_FW_TABLE_XADD;
9634c0c07a5SAlexander V. Chernikov 		texterr = "Adding record failed";
964ac35ff17SAlexander V. Chernikov 	} else {
965ac35ff17SAlexander V. Chernikov 		cmd = IP_FW_TABLE_XDEL;
9664c0c07a5SAlexander V. Chernikov 		texterr = "Deleting record failed";
967ac35ff17SAlexander V. Chernikov 	}
968ac35ff17SAlexander V. Chernikov 
9693a845e10SAlexander V. Chernikov 	/*
9703a845e10SAlexander V. Chernikov 	 * Calculate number of entries:
9713a845e10SAlexander V. Chernikov 	 * Assume [key val] x N for add
9723a845e10SAlexander V. Chernikov 	 * and
9733a845e10SAlexander V. Chernikov 	 * key x N for delete
9743a845e10SAlexander V. Chernikov 	 */
9753a845e10SAlexander V. Chernikov 	count = (add != 0) ? ac / 2 + 1 : ac;
9763a845e10SAlexander V. Chernikov 
9773a845e10SAlexander V. Chernikov 	if (count <= 1) {
9783a845e10SAlexander V. Chernikov 		/* Adding single entry with/without value */
9793a845e10SAlexander V. Chernikov 		memset(&tent, 0, sizeof(tent));
9803a845e10SAlexander V. Chernikov 		tent_buf = &tent;
9813a845e10SAlexander V. Chernikov 	} else {
9823a845e10SAlexander V. Chernikov 
9833a845e10SAlexander V. Chernikov 		if ((tent_buf = calloc(count, sizeof(tent))) == NULL)
9843a845e10SAlexander V. Chernikov 			errx(EX_OSERR,
9853a845e10SAlexander V. Chernikov 			    "Unable to allocate memory for all entries");
9863a845e10SAlexander V. Chernikov 	}
9873a845e10SAlexander V. Chernikov 	ptent = tent_buf;
9883a845e10SAlexander V. Chernikov 
9893a845e10SAlexander V. Chernikov 	memset(&xi, 0, sizeof(xi));
9903a845e10SAlexander V. Chernikov 	count = 0;
9913a845e10SAlexander V. Chernikov 	while (ac > 0) {
9920cba2b28SAlexander V. Chernikov 		tentry_fill_key(oh, ptent, *av, add, &type, &vmask, &xi);
9933a845e10SAlexander V. Chernikov 
9943a845e10SAlexander V. Chernikov 		/*
995463a577bSEitan Adler 		 * Compatibility layer: auto-create table if not exists.
9963a845e10SAlexander V. Chernikov 		 */
9973a845e10SAlexander V. Chernikov 		if (xi.tablename[0] == '\0') {
9983a845e10SAlexander V. Chernikov 			xi.type = type;
9990cba2b28SAlexander V. Chernikov 			xi.vmask = vmask;
10003a845e10SAlexander V. Chernikov 			strlcpy(xi.tablename, oh->ntlv.name,
10013a845e10SAlexander V. Chernikov 			    sizeof(xi.tablename));
1002e62cd31fSAndrey V. Elsukov 			if (quiet == 0)
1003e62cd31fSAndrey V. Elsukov 				warnx("DEPRECATED: inserting data into "
1004e62cd31fSAndrey V. Elsukov 				    "non-existent table %s. (auto-created)",
10053a845e10SAlexander V. Chernikov 				    xi.tablename);
10063a845e10SAlexander V. Chernikov 			table_do_create(oh, &xi);
10073a845e10SAlexander V. Chernikov 		}
10083a845e10SAlexander V. Chernikov 
10093a845e10SAlexander V. Chernikov 		oh->ntlv.type = type;
10103a845e10SAlexander V. Chernikov 		ac--; av++;
10113a845e10SAlexander V. Chernikov 
10123a845e10SAlexander V. Chernikov 		if (add != 0 && ac > 0) {
10130cba2b28SAlexander V. Chernikov 			tentry_fill_value(oh, ptent, *av, type, vmask);
10143a845e10SAlexander V. Chernikov 			ac--; av++;
10153a845e10SAlexander V. Chernikov 		}
10163a845e10SAlexander V. Chernikov 
10173a845e10SAlexander V. Chernikov 		if (update != 0)
10183a845e10SAlexander V. Chernikov 			ptent->head.flags |= IPFW_TF_UPDATE;
10193a845e10SAlexander V. Chernikov 
10203a845e10SAlexander V. Chernikov 		count++;
10213a845e10SAlexander V. Chernikov 		ptent++;
10223a845e10SAlexander V. Chernikov 	}
10233a845e10SAlexander V. Chernikov 
10243a845e10SAlexander V. Chernikov 	error = table_do_modify_record(cmd, oh, tent_buf, count, atomic);
10253a845e10SAlexander V. Chernikov 
10263a845e10SAlexander V. Chernikov 	/*
10273a845e10SAlexander V. Chernikov 	 * Compatibility stuff: do not yell on duplicate keys or
10283a845e10SAlexander V. Chernikov 	 * failed deletions.
10293a845e10SAlexander V. Chernikov 	 */
10303a845e10SAlexander V. Chernikov 	if (error == 0 || (error == EEXIST && add != 0) ||
10313a845e10SAlexander V. Chernikov 	    (error == ENOENT && add == 0)) {
10323a845e10SAlexander V. Chernikov 		if (quiet != 0) {
10333a845e10SAlexander V. Chernikov 			if (tent_buf != &tent)
10343a845e10SAlexander V. Chernikov 				free(tent_buf);
10353a845e10SAlexander V. Chernikov 			return;
10363a845e10SAlexander V. Chernikov 		}
10373a845e10SAlexander V. Chernikov 	}
10383a845e10SAlexander V. Chernikov 
10393a845e10SAlexander V. Chernikov 	/* Report results back */
10403a845e10SAlexander V. Chernikov 	ptent = tent_buf;
10413a845e10SAlexander V. Chernikov 	for (i = 0; i < count; ptent++, i++) {
10423a845e10SAlexander V. Chernikov 		ignored = 0;
10433a845e10SAlexander V. Chernikov 		switch (ptent->result) {
10443a845e10SAlexander V. Chernikov 		case IPFW_TR_ADDED:
10453a845e10SAlexander V. Chernikov 			px = "added";
10463a845e10SAlexander V. Chernikov 			break;
10473a845e10SAlexander V. Chernikov 		case IPFW_TR_DELETED:
10483a845e10SAlexander V. Chernikov 			px = "deleted";
10493a845e10SAlexander V. Chernikov 			break;
10503a845e10SAlexander V. Chernikov 		case IPFW_TR_UPDATED:
10513a845e10SAlexander V. Chernikov 			px = "updated";
10523a845e10SAlexander V. Chernikov 			break;
10533a845e10SAlexander V. Chernikov 		case IPFW_TR_LIMIT:
10543a845e10SAlexander V. Chernikov 			px = "limit";
10553a845e10SAlexander V. Chernikov 			ignored = 1;
10563a845e10SAlexander V. Chernikov 			break;
10573a845e10SAlexander V. Chernikov 		case IPFW_TR_ERROR:
10583a845e10SAlexander V. Chernikov 			px = "error";
10593a845e10SAlexander V. Chernikov 			ignored = 1;
10603a845e10SAlexander V. Chernikov 			break;
10613a845e10SAlexander V. Chernikov 		case IPFW_TR_NOTFOUND:
10623a845e10SAlexander V. Chernikov 			px = "notfound";
10633a845e10SAlexander V. Chernikov 			ignored = 1;
10643a845e10SAlexander V. Chernikov 			break;
10653a845e10SAlexander V. Chernikov 		case IPFW_TR_EXISTS:
10663a845e10SAlexander V. Chernikov 			px = "exists";
10673a845e10SAlexander V. Chernikov 			ignored = 1;
10683a845e10SAlexander V. Chernikov 			break;
10693a845e10SAlexander V. Chernikov 		case IPFW_TR_IGNORED:
10703a845e10SAlexander V. Chernikov 			px = "ignored";
10713a845e10SAlexander V. Chernikov 			ignored = 1;
10723a845e10SAlexander V. Chernikov 			break;
10733a845e10SAlexander V. Chernikov 		default:
10743a845e10SAlexander V. Chernikov 			px = "unknown";
10753a845e10SAlexander V. Chernikov 			ignored = 1;
10763a845e10SAlexander V. Chernikov 		}
10773a845e10SAlexander V. Chernikov 
10783a845e10SAlexander V. Chernikov 		if (error != 0 && atomic != 0 && ignored == 0)
10793a845e10SAlexander V. Chernikov 			printf("%s(reverted): ", px);
10803a845e10SAlexander V. Chernikov 		else
10813a845e10SAlexander V. Chernikov 			printf("%s: ", px);
10823a845e10SAlexander V. Chernikov 
10833a845e10SAlexander V. Chernikov 		table_show_entry(&xi, ptent);
10843a845e10SAlexander V. Chernikov 	}
10853a845e10SAlexander V. Chernikov 
10863a845e10SAlexander V. Chernikov 	if (tent_buf != &tent)
10873a845e10SAlexander V. Chernikov 		free(tent_buf);
10883a845e10SAlexander V. Chernikov 
10893a845e10SAlexander V. Chernikov 	if (error == 0)
10904c0c07a5SAlexander V. Chernikov 		return;
109168bde59eSAlexander V. Chernikov 	/* Get real OS error */
109268bde59eSAlexander V. Chernikov 	error = errno;
10934c0c07a5SAlexander V. Chernikov 
10944c0c07a5SAlexander V. Chernikov 	/* Try to provide more human-readable error */
10954c0c07a5SAlexander V. Chernikov 	switch (error) {
10964c0c07a5SAlexander V. Chernikov 	case EEXIST:
10974c0c07a5SAlexander V. Chernikov 		etxt = "record already exists";
10984c0c07a5SAlexander V. Chernikov 		break;
10994c0c07a5SAlexander V. Chernikov 	case EFBIG:
11004c0c07a5SAlexander V. Chernikov 		etxt = "limit hit";
11014c0c07a5SAlexander V. Chernikov 		break;
11024c0c07a5SAlexander V. Chernikov 	case ESRCH:
11034c0c07a5SAlexander V. Chernikov 		etxt = "table not found";
11044c0c07a5SAlexander V. Chernikov 		break;
11054c0c07a5SAlexander V. Chernikov 	case ENOENT:
11064c0c07a5SAlexander V. Chernikov 		etxt = "record not found";
11074c0c07a5SAlexander V. Chernikov 		break;
11084f43138aSAlexander V. Chernikov 	case EACCES:
11094f43138aSAlexander V. Chernikov 		etxt = "table is locked";
11104f43138aSAlexander V. Chernikov 		break;
11114c0c07a5SAlexander V. Chernikov 	default:
11124c0c07a5SAlexander V. Chernikov 		etxt = strerror(error);
11134c0c07a5SAlexander V. Chernikov 	}
11144c0c07a5SAlexander V. Chernikov 
11154c0c07a5SAlexander V. Chernikov 	errx(EX_OSERR, "%s: %s", texterr, etxt);
1116ac35ff17SAlexander V. Chernikov }
1117ac35ff17SAlexander V. Chernikov 
111881d3153dSAlexander V. Chernikov static int
111981d3153dSAlexander V. Chernikov table_do_lookup(ipfw_obj_header *oh, char *key, ipfw_xtable_info *xi,
112081d3153dSAlexander V. Chernikov     ipfw_obj_tentry *xtent)
112181d3153dSAlexander V. Chernikov {
112281d3153dSAlexander V. Chernikov 	char xbuf[sizeof(ipfw_obj_header) + sizeof(ipfw_obj_tentry)];
112381d3153dSAlexander V. Chernikov 	ipfw_obj_tentry *tent;
11240cba2b28SAlexander V. Chernikov 	uint8_t type;
11250cba2b28SAlexander V. Chernikov 	uint32_t vmask;
112681d3153dSAlexander V. Chernikov 	size_t sz;
112781d3153dSAlexander V. Chernikov 
112881d3153dSAlexander V. Chernikov 	memcpy(xbuf, oh, sizeof(*oh));
112981d3153dSAlexander V. Chernikov 	oh = (ipfw_obj_header *)xbuf;
113081d3153dSAlexander V. Chernikov 	tent = (ipfw_obj_tentry *)(oh + 1);
113181d3153dSAlexander V. Chernikov 
113281d3153dSAlexander V. Chernikov 	memset(tent, 0, sizeof(*tent));
113381d3153dSAlexander V. Chernikov 	tent->head.length = sizeof(*tent);
113481d3153dSAlexander V. Chernikov 	tent->idx = 1;
113581d3153dSAlexander V. Chernikov 
11360cba2b28SAlexander V. Chernikov 	tentry_fill_key(oh, tent, key, 0, &type, &vmask, xi);
113781d3153dSAlexander V. Chernikov 	oh->ntlv.type = type;
113881d3153dSAlexander V. Chernikov 
113981d3153dSAlexander V. Chernikov 	sz = sizeof(xbuf);
1140b6462881SAlexander V. Chernikov 	if (do_get3(IP_FW_TABLE_XFIND, &oh->opheader, &sz) != 0)
1141b6462881SAlexander V. Chernikov 		return (errno);
114281d3153dSAlexander V. Chernikov 
114381d3153dSAlexander V. Chernikov 	if (sz < sizeof(xbuf))
114481d3153dSAlexander V. Chernikov 		return (EINVAL);
114581d3153dSAlexander V. Chernikov 
114681d3153dSAlexander V. Chernikov 	*xtent = *tent;
114781d3153dSAlexander V. Chernikov 
114881d3153dSAlexander V. Chernikov 	return (0);
114981d3153dSAlexander V. Chernikov }
115081d3153dSAlexander V. Chernikov 
115181d3153dSAlexander V. Chernikov static void
115281d3153dSAlexander V. Chernikov table_lookup(ipfw_obj_header *oh, int ac, char *av[])
115381d3153dSAlexander V. Chernikov {
115481d3153dSAlexander V. Chernikov 	ipfw_obj_tentry xtent;
115581d3153dSAlexander V. Chernikov 	ipfw_xtable_info xi;
1156914bffb6SAlexander V. Chernikov 	char key[64];
115781d3153dSAlexander V. Chernikov 	int error;
115881d3153dSAlexander V. Chernikov 
115981d3153dSAlexander V. Chernikov 	if (ac == 0)
116081d3153dSAlexander V. Chernikov 		errx(EX_USAGE, "address required");
116181d3153dSAlexander V. Chernikov 
1162914bffb6SAlexander V. Chernikov 	strlcpy(key, *av, sizeof(key));
1163914bffb6SAlexander V. Chernikov 
11643a845e10SAlexander V. Chernikov 	memset(&xi, 0, sizeof(xi));
1165914bffb6SAlexander V. Chernikov 	error = table_do_lookup(oh, key, &xi, &xtent);
116681d3153dSAlexander V. Chernikov 
116781d3153dSAlexander V. Chernikov 	switch (error) {
116881d3153dSAlexander V. Chernikov 	case 0:
116981d3153dSAlexander V. Chernikov 		break;
117081d3153dSAlexander V. Chernikov 	case ESRCH:
117181d3153dSAlexander V. Chernikov 		errx(EX_UNAVAILABLE, "Table %s not found", oh->ntlv.name);
117281d3153dSAlexander V. Chernikov 	case ENOENT:
117381d3153dSAlexander V. Chernikov 		errx(EX_UNAVAILABLE, "Entry %s not found", *av);
117481d3153dSAlexander V. Chernikov 	case ENOTSUP:
117581d3153dSAlexander V. Chernikov 		errx(EX_UNAVAILABLE, "Table %s algo does not support "
117681d3153dSAlexander V. Chernikov 		    "\"lookup\" method", oh->ntlv.name);
117781d3153dSAlexander V. Chernikov 	default:
117881d3153dSAlexander V. Chernikov 		err(EX_OSERR, "getsockopt(IP_FW_TABLE_XFIND)");
117981d3153dSAlexander V. Chernikov 	}
118081d3153dSAlexander V. Chernikov 
118181d3153dSAlexander V. Chernikov 	table_show_entry(&xi, &xtent);
118281d3153dSAlexander V. Chernikov }
1183ac35ff17SAlexander V. Chernikov 
1184ac35ff17SAlexander V. Chernikov static void
1185914bffb6SAlexander V. Chernikov tentry_fill_key_type(char *arg, ipfw_obj_tentry *tentry, uint8_t type,
1186914bffb6SAlexander V. Chernikov     uint8_t tflags)
1187ac35ff17SAlexander V. Chernikov {
1188914bffb6SAlexander V. Chernikov 	char *p, *pp;
1189ac35ff17SAlexander V. Chernikov 	int mask, af;
1190914bffb6SAlexander V. Chernikov 	struct in6_addr *paddr, tmp;
1191914bffb6SAlexander V. Chernikov 	struct tflow_entry *tfe;
1192ac35ff17SAlexander V. Chernikov 	uint32_t key, *pkey;
1193914bffb6SAlexander V. Chernikov 	uint16_t port;
1194914bffb6SAlexander V. Chernikov 	struct protoent *pent;
1195914bffb6SAlexander V. Chernikov 	struct servent *sent;
1196ac35ff17SAlexander V. Chernikov 	int masklen;
1197ac35ff17SAlexander V. Chernikov 
1198268a20a0SAdrian Chadd 	mask = masklen = 0;
1199ac35ff17SAlexander V. Chernikov 	af = 0;
1200ac35ff17SAlexander V. Chernikov 	paddr = (struct in6_addr *)&tentry->k;
1201ac35ff17SAlexander V. Chernikov 
1202ac35ff17SAlexander V. Chernikov 	switch (type) {
1203c21034b7SAlexander V. Chernikov 	case IPFW_TABLE_ADDR:
1204ac35ff17SAlexander V. Chernikov 		/* Remove / if exists */
1205ac35ff17SAlexander V. Chernikov 		if ((p = strchr(arg, '/')) != NULL) {
1206ac35ff17SAlexander V. Chernikov 			*p = '\0';
1207ac35ff17SAlexander V. Chernikov 			mask = atoi(p + 1);
1208ac35ff17SAlexander V. Chernikov 		}
1209ac35ff17SAlexander V. Chernikov 
1210ac35ff17SAlexander V. Chernikov 		if (inet_pton(AF_INET, arg, paddr) == 1) {
1211ac35ff17SAlexander V. Chernikov 			if (p != NULL && mask > 32)
1212ac35ff17SAlexander V. Chernikov 				errx(EX_DATAERR, "bad IPv4 mask width: %s",
1213ac35ff17SAlexander V. Chernikov 				    p + 1);
1214ac35ff17SAlexander V. Chernikov 
1215ac35ff17SAlexander V. Chernikov 			masklen = p ? mask : 32;
1216ac35ff17SAlexander V. Chernikov 			af = AF_INET;
1217ac35ff17SAlexander V. Chernikov 		} else if (inet_pton(AF_INET6, arg, paddr) == 1) {
1218ac35ff17SAlexander V. Chernikov 			if (IN6_IS_ADDR_V4COMPAT(paddr))
1219ac35ff17SAlexander V. Chernikov 				errx(EX_DATAERR,
1220ac35ff17SAlexander V. Chernikov 				    "Use IPv4 instead of v4-compatible");
1221ac35ff17SAlexander V. Chernikov 			if (p != NULL && mask > 128)
1222ac35ff17SAlexander V. Chernikov 				errx(EX_DATAERR, "bad IPv6 mask width: %s",
1223ac35ff17SAlexander V. Chernikov 				    p + 1);
1224ac35ff17SAlexander V. Chernikov 
1225ac35ff17SAlexander V. Chernikov 			masklen = p ? mask : 128;
1226ac35ff17SAlexander V. Chernikov 			af = AF_INET6;
1227ac35ff17SAlexander V. Chernikov 		} else {
1228ac35ff17SAlexander V. Chernikov 			/* Assume FQDN */
1229ac35ff17SAlexander V. Chernikov 			if (lookup_host(arg, (struct in_addr *)paddr) != 0)
1230ac35ff17SAlexander V. Chernikov 				errx(EX_NOHOST, "hostname ``%s'' unknown", arg);
1231ac35ff17SAlexander V. Chernikov 
1232ac35ff17SAlexander V. Chernikov 			masklen = 32;
1233c21034b7SAlexander V. Chernikov 			type = IPFW_TABLE_ADDR;
1234ac35ff17SAlexander V. Chernikov 			af = AF_INET;
1235ac35ff17SAlexander V. Chernikov 		}
1236ac35ff17SAlexander V. Chernikov 		break;
1237ac35ff17SAlexander V. Chernikov 	case IPFW_TABLE_INTERFACE:
1238ac35ff17SAlexander V. Chernikov 		/* Assume interface name. Copy significant data only */
1239ac35ff17SAlexander V. Chernikov 		mask = MIN(strlen(arg), IF_NAMESIZE - 1);
1240ac35ff17SAlexander V. Chernikov 		memcpy(paddr, arg, mask);
1241ac35ff17SAlexander V. Chernikov 		/* Set mask to exact match */
1242ac35ff17SAlexander V. Chernikov 		masklen = 8 * IF_NAMESIZE;
1243ac35ff17SAlexander V. Chernikov 		break;
1244b23d5de9SAlexander V. Chernikov 	case IPFW_TABLE_NUMBER:
1245ac35ff17SAlexander V. Chernikov 		/* Port or any other key */
1246ac35ff17SAlexander V. Chernikov 		key = strtol(arg, &p, 10);
1247ac35ff17SAlexander V. Chernikov 		if (*p != '\0')
1248ac35ff17SAlexander V. Chernikov 			errx(EX_DATAERR, "Invalid number: %s", arg);
1249ac35ff17SAlexander V. Chernikov 
1250ac35ff17SAlexander V. Chernikov 		pkey = (uint32_t *)paddr;
1251ac35ff17SAlexander V. Chernikov 		*pkey = key;
1252ac35ff17SAlexander V. Chernikov 		masklen = 32;
1253ac35ff17SAlexander V. Chernikov 		break;
1254914bffb6SAlexander V. Chernikov 	case IPFW_TABLE_FLOW:
1255914bffb6SAlexander V. Chernikov 		/* Assume [src-ip][,proto][,src-port][,dst-ip][,dst-port] */
1256914bffb6SAlexander V. Chernikov 		tfe = &tentry->k.flow;
1257914bffb6SAlexander V. Chernikov 		af = 0;
1258914bffb6SAlexander V. Chernikov 
1259914bffb6SAlexander V. Chernikov 		/* Handle <ipv4|ipv6> */
1260914bffb6SAlexander V. Chernikov 		if ((tflags & IPFW_TFFLAG_SRCIP) != 0) {
1261914bffb6SAlexander V. Chernikov 			if ((p = strchr(arg, ',')) != NULL)
1262914bffb6SAlexander V. Chernikov 				*p++ = '\0';
1263914bffb6SAlexander V. Chernikov 			/* Determine family using temporary storage */
1264914bffb6SAlexander V. Chernikov 			if (inet_pton(AF_INET, arg, &tmp) == 1) {
1265914bffb6SAlexander V. Chernikov 				if (af != 0 && af != AF_INET)
1266914bffb6SAlexander V. Chernikov 					errx(EX_DATAERR,
1267914bffb6SAlexander V. Chernikov 					    "Inconsistent address family\n");
1268914bffb6SAlexander V. Chernikov 				af = AF_INET;
1269914bffb6SAlexander V. Chernikov 				memcpy(&tfe->a.a4.sip, &tmp, 4);
1270914bffb6SAlexander V. Chernikov 			} else if (inet_pton(AF_INET6, arg, &tmp) == 1) {
1271914bffb6SAlexander V. Chernikov 				if (af != 0 && af != AF_INET6)
1272914bffb6SAlexander V. Chernikov 					errx(EX_DATAERR,
1273914bffb6SAlexander V. Chernikov 					    "Inconsistent address family\n");
1274914bffb6SAlexander V. Chernikov 				af = AF_INET6;
1275914bffb6SAlexander V. Chernikov 				memcpy(&tfe->a.a6.sip6, &tmp, 16);
1276914bffb6SAlexander V. Chernikov 			}
1277914bffb6SAlexander V. Chernikov 
1278914bffb6SAlexander V. Chernikov 			arg = p;
1279914bffb6SAlexander V. Chernikov 		}
1280914bffb6SAlexander V. Chernikov 
1281914bffb6SAlexander V. Chernikov 		/* Handle <proto-num|proto-name> */
1282914bffb6SAlexander V. Chernikov 		if ((tflags & IPFW_TFFLAG_PROTO) != 0) {
128335df97d0SAlexander V. Chernikov 			if (arg == NULL)
128435df97d0SAlexander V. Chernikov 				errx(EX_DATAERR, "invalid key: proto missing");
1285914bffb6SAlexander V. Chernikov 			if ((p = strchr(arg, ',')) != NULL)
1286914bffb6SAlexander V. Chernikov 				*p++ = '\0';
1287914bffb6SAlexander V. Chernikov 
1288914bffb6SAlexander V. Chernikov 			key = strtol(arg, &pp, 10);
1289914bffb6SAlexander V. Chernikov 			if (*pp != '\0') {
1290914bffb6SAlexander V. Chernikov 				if ((pent = getprotobyname(arg)) == NULL)
1291914bffb6SAlexander V. Chernikov 					errx(EX_DATAERR, "Unknown proto: %s",
1292914bffb6SAlexander V. Chernikov 					    arg);
1293914bffb6SAlexander V. Chernikov 				else
1294914bffb6SAlexander V. Chernikov 					key = pent->p_proto;
1295914bffb6SAlexander V. Chernikov 			}
1296914bffb6SAlexander V. Chernikov 
1297914bffb6SAlexander V. Chernikov 			if (key > 255)
1298914bffb6SAlexander V. Chernikov 				errx(EX_DATAERR, "Bad protocol number: %u",key);
1299914bffb6SAlexander V. Chernikov 
1300914bffb6SAlexander V. Chernikov 			tfe->proto = key;
1301914bffb6SAlexander V. Chernikov 
1302914bffb6SAlexander V. Chernikov 			arg = p;
1303914bffb6SAlexander V. Chernikov 		}
1304914bffb6SAlexander V. Chernikov 
1305914bffb6SAlexander V. Chernikov 		/* Handle <port-num|service-name> */
1306914bffb6SAlexander V. Chernikov 		if ((tflags & IPFW_TFFLAG_SRCPORT) != 0) {
130735df97d0SAlexander V. Chernikov 			if (arg == NULL)
130835df97d0SAlexander V. Chernikov 				errx(EX_DATAERR, "invalid key: src port missing");
1309914bffb6SAlexander V. Chernikov 			if ((p = strchr(arg, ',')) != NULL)
1310914bffb6SAlexander V. Chernikov 				*p++ = '\0';
1311914bffb6SAlexander V. Chernikov 
131252772a85SAndrey V. Elsukov 			port = htons(strtol(arg, &pp, 10));
131352772a85SAndrey V. Elsukov 			if (*pp != '\0') {
1314914bffb6SAlexander V. Chernikov 				if ((sent = getservbyname(arg, NULL)) == NULL)
1315914bffb6SAlexander V. Chernikov 					errx(EX_DATAERR, "Unknown service: %s",
1316914bffb6SAlexander V. Chernikov 					    arg);
131752772a85SAndrey V. Elsukov 				port = sent->s_port;
1318914bffb6SAlexander V. Chernikov 			}
1319914bffb6SAlexander V. Chernikov 			tfe->sport = port;
1320914bffb6SAlexander V. Chernikov 			arg = p;
1321914bffb6SAlexander V. Chernikov 		}
1322914bffb6SAlexander V. Chernikov 
1323914bffb6SAlexander V. Chernikov 		/* Handle <ipv4|ipv6>*/
1324914bffb6SAlexander V. Chernikov 		if ((tflags & IPFW_TFFLAG_DSTIP) != 0) {
132535df97d0SAlexander V. Chernikov 			if (arg == NULL)
132635df97d0SAlexander V. Chernikov 				errx(EX_DATAERR, "invalid key: dst ip missing");
1327914bffb6SAlexander V. Chernikov 			if ((p = strchr(arg, ',')) != NULL)
1328914bffb6SAlexander V. Chernikov 				*p++ = '\0';
1329914bffb6SAlexander V. Chernikov 			/* Determine family using temporary storage */
1330914bffb6SAlexander V. Chernikov 			if (inet_pton(AF_INET, arg, &tmp) == 1) {
1331914bffb6SAlexander V. Chernikov 				if (af != 0 && af != AF_INET)
1332914bffb6SAlexander V. Chernikov 					errx(EX_DATAERR,
1333914bffb6SAlexander V. Chernikov 					    "Inconsistent address family");
1334914bffb6SAlexander V. Chernikov 				af = AF_INET;
1335914bffb6SAlexander V. Chernikov 				memcpy(&tfe->a.a4.dip, &tmp, 4);
1336914bffb6SAlexander V. Chernikov 			} else if (inet_pton(AF_INET6, arg, &tmp) == 1) {
1337914bffb6SAlexander V. Chernikov 				if (af != 0 && af != AF_INET6)
1338914bffb6SAlexander V. Chernikov 					errx(EX_DATAERR,
1339914bffb6SAlexander V. Chernikov 					    "Inconsistent address family");
1340914bffb6SAlexander V. Chernikov 				af = AF_INET6;
1341914bffb6SAlexander V. Chernikov 				memcpy(&tfe->a.a6.dip6, &tmp, 16);
1342914bffb6SAlexander V. Chernikov 			}
1343914bffb6SAlexander V. Chernikov 
1344914bffb6SAlexander V. Chernikov 			arg = p;
1345914bffb6SAlexander V. Chernikov 		}
1346914bffb6SAlexander V. Chernikov 
1347914bffb6SAlexander V. Chernikov 		/* Handle <port-num|service-name> */
1348914bffb6SAlexander V. Chernikov 		if ((tflags & IPFW_TFFLAG_DSTPORT) != 0) {
134935df97d0SAlexander V. Chernikov 			if (arg == NULL)
135035df97d0SAlexander V. Chernikov 				errx(EX_DATAERR, "invalid key: dst port missing");
1351914bffb6SAlexander V. Chernikov 			if ((p = strchr(arg, ',')) != NULL)
1352914bffb6SAlexander V. Chernikov 				*p++ = '\0';
1353914bffb6SAlexander V. Chernikov 
135452772a85SAndrey V. Elsukov 			port = htons(strtol(arg, &pp, 10));
135552772a85SAndrey V. Elsukov 			if (*pp != '\0') {
1356914bffb6SAlexander V. Chernikov 				if ((sent = getservbyname(arg, NULL)) == NULL)
1357914bffb6SAlexander V. Chernikov 					errx(EX_DATAERR, "Unknown service: %s",
1358914bffb6SAlexander V. Chernikov 					    arg);
135952772a85SAndrey V. Elsukov 				port = sent->s_port;
1360914bffb6SAlexander V. Chernikov 			}
1361914bffb6SAlexander V. Chernikov 			tfe->dport = port;
1362914bffb6SAlexander V. Chernikov 			arg = p;
1363914bffb6SAlexander V. Chernikov 		}
1364914bffb6SAlexander V. Chernikov 
1365914bffb6SAlexander V. Chernikov 		tfe->af = af;
1366914bffb6SAlexander V. Chernikov 
1367914bffb6SAlexander V. Chernikov 		break;
1368914bffb6SAlexander V. Chernikov 
1369ac35ff17SAlexander V. Chernikov 	default:
1370ac35ff17SAlexander V. Chernikov 		errx(EX_DATAERR, "Unsupported table type: %d", type);
1371ac35ff17SAlexander V. Chernikov 	}
1372ac35ff17SAlexander V. Chernikov 
1373ac35ff17SAlexander V. Chernikov 	tentry->subtype = af;
1374ac35ff17SAlexander V. Chernikov 	tentry->masklen = masklen;
1375ac35ff17SAlexander V. Chernikov }
1376ac35ff17SAlexander V. Chernikov 
13778755aff6SAlexander V. Chernikov /*
13788755aff6SAlexander V. Chernikov  * Tries to guess table key type.
13798755aff6SAlexander V. Chernikov  * This procedure is used in legacy table auto-create
13808755aff6SAlexander V. Chernikov  * code AND in `ipfw -n` ruleset checking.
13818755aff6SAlexander V. Chernikov  *
13828755aff6SAlexander V. Chernikov  * Imported from old table_fill_xentry() parse code.
13838755aff6SAlexander V. Chernikov  */
13848755aff6SAlexander V. Chernikov static int
13858755aff6SAlexander V. Chernikov guess_key_type(char *key, uint8_t *ptype)
13868755aff6SAlexander V. Chernikov {
13878755aff6SAlexander V. Chernikov 	char *p;
13888755aff6SAlexander V. Chernikov 	struct in6_addr addr;
13898755aff6SAlexander V. Chernikov 	uint32_t kv;
13908755aff6SAlexander V. Chernikov 
13918755aff6SAlexander V. Chernikov 	if (ishexnumber(*key) != 0 || *key == ':') {
13928755aff6SAlexander V. Chernikov 		/* Remove / if exists */
13938755aff6SAlexander V. Chernikov 		if ((p = strchr(key, '/')) != NULL)
13948755aff6SAlexander V. Chernikov 			*p = '\0';
13958755aff6SAlexander V. Chernikov 
13968755aff6SAlexander V. Chernikov 		if ((inet_pton(AF_INET, key, &addr) == 1) ||
13978755aff6SAlexander V. Chernikov 		    (inet_pton(AF_INET6, key, &addr) == 1)) {
13988755aff6SAlexander V. Chernikov 			*ptype = IPFW_TABLE_CIDR;
13998755aff6SAlexander V. Chernikov 			if (p != NULL)
14008755aff6SAlexander V. Chernikov 				*p = '/';
14018755aff6SAlexander V. Chernikov 			return (0);
14028755aff6SAlexander V. Chernikov 		} else {
14038755aff6SAlexander V. Chernikov 			/* Port or any other key */
14048755aff6SAlexander V. Chernikov 			/* Skip non-base 10 entries like 'fa1' */
14058755aff6SAlexander V. Chernikov 			kv = strtol(key, &p, 10);
14068755aff6SAlexander V. Chernikov 			if (*p == '\0') {
14078755aff6SAlexander V. Chernikov 				*ptype = IPFW_TABLE_NUMBER;
14088755aff6SAlexander V. Chernikov 				return (0);
14098755aff6SAlexander V. Chernikov 			} else if ((p != key) && (*p == '.')) {
14108755aff6SAlexander V. Chernikov 				/*
14118755aff6SAlexander V. Chernikov 				 * Warn on IPv4 address strings
14128755aff6SAlexander V. Chernikov 				 * which are "valid" for inet_aton() but not
14138755aff6SAlexander V. Chernikov 				 * in inet_pton().
14148755aff6SAlexander V. Chernikov 				 *
14158755aff6SAlexander V. Chernikov 				 * Typical examples: '10.5' or '10.0.0.05'
14168755aff6SAlexander V. Chernikov 				 */
14178755aff6SAlexander V. Chernikov 				return (1);
14188755aff6SAlexander V. Chernikov 			}
14198755aff6SAlexander V. Chernikov 		}
14208755aff6SAlexander V. Chernikov 	}
14218755aff6SAlexander V. Chernikov 
14228755aff6SAlexander V. Chernikov 	if (strchr(key, '.') == NULL) {
14238755aff6SAlexander V. Chernikov 		*ptype = IPFW_TABLE_INTERFACE;
14248755aff6SAlexander V. Chernikov 		return (0);
14258755aff6SAlexander V. Chernikov 	}
14268755aff6SAlexander V. Chernikov 
14278755aff6SAlexander V. Chernikov 	if (lookup_host(key, (struct in_addr *)&addr) != 0)
14288755aff6SAlexander V. Chernikov 		return (1);
14298755aff6SAlexander V. Chernikov 
14308755aff6SAlexander V. Chernikov 	*ptype = IPFW_TABLE_CIDR;
14318755aff6SAlexander V. Chernikov 	return (0);
14328755aff6SAlexander V. Chernikov }
14338755aff6SAlexander V. Chernikov 
1434ac35ff17SAlexander V. Chernikov static void
1435ac35ff17SAlexander V. Chernikov tentry_fill_key(ipfw_obj_header *oh, ipfw_obj_tentry *tent, char *key,
14360cba2b28SAlexander V. Chernikov     int add, uint8_t *ptype, uint32_t *pvmask, ipfw_xtable_info *xi)
1437ac35ff17SAlexander V. Chernikov {
14380cba2b28SAlexander V. Chernikov 	uint8_t type, tflags;
14390cba2b28SAlexander V. Chernikov 	uint32_t vmask;
1440ac35ff17SAlexander V. Chernikov 	int error;
1441ac35ff17SAlexander V. Chernikov 
1442ac35ff17SAlexander V. Chernikov 	type = 0;
1443914bffb6SAlexander V. Chernikov 	tflags = 0;
14440cba2b28SAlexander V. Chernikov 	vmask = 0;
1445ac35ff17SAlexander V. Chernikov 
14463a845e10SAlexander V. Chernikov 	if (xi->tablename[0] == '\0')
144781d3153dSAlexander V. Chernikov 		error = table_get_info(oh, xi);
14483a845e10SAlexander V. Chernikov 	else
14493a845e10SAlexander V. Chernikov 		error = 0;
145081d3153dSAlexander V. Chernikov 
145181d3153dSAlexander V. Chernikov 	if (error == 0) {
145256707beeSMark Johnston 		if (g_co.test_only == 0) {
14538755aff6SAlexander V. Chernikov 			/* Table found */
145481d3153dSAlexander V. Chernikov 			type = xi->type;
1455914bffb6SAlexander V. Chernikov 			tflags = xi->tflags;
14560cba2b28SAlexander V. Chernikov 			vmask = xi->vmask;
145781d3153dSAlexander V. Chernikov 		} else {
14588755aff6SAlexander V. Chernikov 			/*
1459463a577bSEitan Adler 			 * We're running `ipfw -n`
1460463a577bSEitan Adler 			 * Compatibility layer: try to guess key type
14618755aff6SAlexander V. Chernikov 			 * before failing.
14628755aff6SAlexander V. Chernikov 			 */
14638755aff6SAlexander V. Chernikov 			if (guess_key_type(key, &type) != 0) {
14648755aff6SAlexander V. Chernikov 				/* Inknown key */
14658755aff6SAlexander V. Chernikov 				errx(EX_USAGE, "Cannot guess "
14668755aff6SAlexander V. Chernikov 				    "key '%s' type", key);
14678755aff6SAlexander V. Chernikov 			}
14688755aff6SAlexander V. Chernikov 			vmask = IPFW_VTYPE_LEGACY;
14698755aff6SAlexander V. Chernikov 		}
14708755aff6SAlexander V. Chernikov 	} else {
147181d3153dSAlexander V. Chernikov 		if (error != ESRCH)
147281d3153dSAlexander V. Chernikov 			errx(EX_OSERR, "Error requesting table %s info",
147381d3153dSAlexander V. Chernikov 			    oh->ntlv.name);
147435df97d0SAlexander V. Chernikov 		if (add == 0)
147535df97d0SAlexander V. Chernikov 			errx(EX_DATAERR, "Table %s does not exist",
147635df97d0SAlexander V. Chernikov 			    oh->ntlv.name);
1477ac35ff17SAlexander V. Chernikov 		/*
14788755aff6SAlexander V. Chernikov 		 * Table does not exist
1479463a577bSEitan Adler 		 * Compatibility layer: try to guess key type before failing.
1480ac35ff17SAlexander V. Chernikov 		 */
14818755aff6SAlexander V. Chernikov 		if (guess_key_type(key, &type) != 0) {
148281d3153dSAlexander V. Chernikov 			/* Inknown key */
148381d3153dSAlexander V. Chernikov 			errx(EX_USAGE, "Table %s does not exist, cannot guess "
1484db785d31SAlexander V. Chernikov 			    "key '%s' type", oh->ntlv.name, key);
148581d3153dSAlexander V. Chernikov 		}
14868755aff6SAlexander V. Chernikov 
14878755aff6SAlexander V. Chernikov 		vmask = IPFW_VTYPE_LEGACY;
1488ac35ff17SAlexander V. Chernikov 	}
1489ac35ff17SAlexander V. Chernikov 
1490914bffb6SAlexander V. Chernikov 	tentry_fill_key_type(key, tent, type, tflags);
1491ac35ff17SAlexander V. Chernikov 
1492ac35ff17SAlexander V. Chernikov 	*ptype = type;
14930cba2b28SAlexander V. Chernikov 	*pvmask = vmask;
14940cba2b28SAlexander V. Chernikov }
14950cba2b28SAlexander V. Chernikov 
14960cba2b28SAlexander V. Chernikov static void
14970cba2b28SAlexander V. Chernikov set_legacy_value(uint32_t val, ipfw_table_value *v)
14980cba2b28SAlexander V. Chernikov {
14990cba2b28SAlexander V. Chernikov 	v->tag = val;
15000cba2b28SAlexander V. Chernikov 	v->pipe = val;
15010cba2b28SAlexander V. Chernikov 	v->divert = val;
15020cba2b28SAlexander V. Chernikov 	v->skipto = val;
15030cba2b28SAlexander V. Chernikov 	v->netgraph = val;
15040cba2b28SAlexander V. Chernikov 	v->fib = val;
15050cba2b28SAlexander V. Chernikov 	v->nat = val;
15060cba2b28SAlexander V. Chernikov 	v->nh4 = val;
15070cba2b28SAlexander V. Chernikov 	v->dscp = (uint8_t)val;
15080cba2b28SAlexander V. Chernikov 	v->limit = val;
1509ac35ff17SAlexander V. Chernikov }
1510ac35ff17SAlexander V. Chernikov 
1511ac35ff17SAlexander V. Chernikov static void
151256707beeSMark Johnston tentry_fill_value(ipfw_obj_header *oh __unused, ipfw_obj_tentry *tent,
151356707beeSMark Johnston     char *arg, uint8_t type __unused, uint32_t vmask)
1514ac35ff17SAlexander V. Chernikov {
15152530ed9eSAndrey V. Elsukov 	struct addrinfo hints, *res;
151656707beeSMark Johnston 	struct in_addr ipaddr;
151756707beeSMark Johnston 	const char *etype;
151856707beeSMark Johnston 	char *comma, *e, *n, *p;
1519b7684f4bSMarcelo Araujo 	uint32_t a4, flag, val;
15200cba2b28SAlexander V. Chernikov 	ipfw_table_value *v;
15210cba2b28SAlexander V. Chernikov 	uint32_t i;
1522d91c61c3SAlexander V. Chernikov 	int dval;
1523ac35ff17SAlexander V. Chernikov 
15240cba2b28SAlexander V. Chernikov 	v = &tent->v.value;
15250cba2b28SAlexander V. Chernikov 
15260cba2b28SAlexander V. Chernikov 	/* Compat layer: keep old behavior for legacy value types */
15270cba2b28SAlexander V. Chernikov 	if (vmask == IPFW_VTYPE_LEGACY) {
1528adf3b2b9SAlexander V. Chernikov 		/* Try to interpret as number first */
15290cba2b28SAlexander V. Chernikov 		val = strtoul(arg, &p, 0);
15300cba2b28SAlexander V. Chernikov 		if (*p == '\0') {
15310cba2b28SAlexander V. Chernikov 			set_legacy_value(val, v);
1532adf3b2b9SAlexander V. Chernikov 			return;
15330cba2b28SAlexander V. Chernikov 		}
1534adf3b2b9SAlexander V. Chernikov 		if (inet_pton(AF_INET, arg, &val) == 1) {
15350cba2b28SAlexander V. Chernikov 			set_legacy_value(ntohl(val), v);
1536adf3b2b9SAlexander V. Chernikov 			return;
1537adf3b2b9SAlexander V. Chernikov 		}
1538adf3b2b9SAlexander V. Chernikov 		/* Try hostname */
15394e0a8b61SConrad Meyer 		if (lookup_host(arg, &ipaddr) == 0) {
15404e0a8b61SConrad Meyer 			set_legacy_value(ntohl(ipaddr.s_addr), v);
1541adf3b2b9SAlexander V. Chernikov 			return;
15420cba2b28SAlexander V. Chernikov 		}
1543adf3b2b9SAlexander V. Chernikov 		errx(EX_OSERR, "Unable to parse value %s", arg);
15440cba2b28SAlexander V. Chernikov 	}
15450cba2b28SAlexander V. Chernikov 
15460cba2b28SAlexander V. Chernikov 	/*
15470cba2b28SAlexander V. Chernikov 	 * Shorthands: handle single value if vmask consists
15480cba2b28SAlexander V. Chernikov 	 * of numbers only. e.g.:
15490cba2b28SAlexander V. Chernikov 	 * vmask = "fib,skipto" -> treat input "1" as "1,1"
15500cba2b28SAlexander V. Chernikov 	 */
15510cba2b28SAlexander V. Chernikov 
15520cba2b28SAlexander V. Chernikov 	n = arg;
15530cba2b28SAlexander V. Chernikov 	etype = NULL;
155456707beeSMark Johnston 	for (i = 1; i < (1u << 31); i *= 2) {
15550cba2b28SAlexander V. Chernikov 		if ((flag = (vmask & i)) == 0)
15560cba2b28SAlexander V. Chernikov 			continue;
15570cba2b28SAlexander V. Chernikov 		vmask &= ~flag;
15580cba2b28SAlexander V. Chernikov 
15590cba2b28SAlexander V. Chernikov 		if ((comma = strchr(n, ',')) != NULL)
15600cba2b28SAlexander V. Chernikov 			*comma = '\0';
15610cba2b28SAlexander V. Chernikov 
15620cba2b28SAlexander V. Chernikov 		switch (flag) {
15630cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_TAG:
15640cba2b28SAlexander V. Chernikov 			v->tag = strtol(n, &e, 10);
15650cba2b28SAlexander V. Chernikov 			if (*e != '\0')
15660cba2b28SAlexander V. Chernikov 				etype = "tag";
1567ac35ff17SAlexander V. Chernikov 			break;
15680cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_PIPE:
15690cba2b28SAlexander V. Chernikov 			v->pipe = strtol(n, &e, 10);
15700cba2b28SAlexander V. Chernikov 			if (*e != '\0')
15710cba2b28SAlexander V. Chernikov 				etype = "pipe";
1572ac35ff17SAlexander V. Chernikov 			break;
15730cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_DIVERT:
15740cba2b28SAlexander V. Chernikov 			v->divert = strtol(n, &e, 10);
15750cba2b28SAlexander V. Chernikov 			if (*e != '\0')
15760cba2b28SAlexander V. Chernikov 				etype = "divert";
15770cba2b28SAlexander V. Chernikov 			break;
15780cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_SKIPTO:
15790cba2b28SAlexander V. Chernikov 			v->skipto = strtol(n, &e, 10);
15800cba2b28SAlexander V. Chernikov 			if (*e != '\0')
15810cba2b28SAlexander V. Chernikov 				etype = "skipto";
15820cba2b28SAlexander V. Chernikov 			break;
15830cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_NETGRAPH:
15840cba2b28SAlexander V. Chernikov 			v->netgraph = strtol(n, &e, 10);
15850cba2b28SAlexander V. Chernikov 			if (*e != '\0')
15860cba2b28SAlexander V. Chernikov 				etype = "netgraph";
15870cba2b28SAlexander V. Chernikov 			break;
15880cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_FIB:
15890cba2b28SAlexander V. Chernikov 			v->fib = strtol(n, &e, 10);
15900cba2b28SAlexander V. Chernikov 			if (*e != '\0')
15910cba2b28SAlexander V. Chernikov 				etype = "fib";
15920cba2b28SAlexander V. Chernikov 			break;
15930cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_NAT:
15940cba2b28SAlexander V. Chernikov 			v->nat = strtol(n, &e, 10);
15950cba2b28SAlexander V. Chernikov 			if (*e != '\0')
15960cba2b28SAlexander V. Chernikov 				etype = "nat";
15970cba2b28SAlexander V. Chernikov 			break;
15980cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_LIMIT:
15990cba2b28SAlexander V. Chernikov 			v->limit = strtol(n, &e, 10);
16000cba2b28SAlexander V. Chernikov 			if (*e != '\0')
16010cba2b28SAlexander V. Chernikov 				etype = "limit";
16020cba2b28SAlexander V. Chernikov 			break;
16030cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_NH4:
16040cba2b28SAlexander V. Chernikov 			if (strchr(n, '.') != NULL &&
16050cba2b28SAlexander V. Chernikov 			    inet_pton(AF_INET, n, &a4) == 1) {
16060cba2b28SAlexander V. Chernikov 				v->nh4 = ntohl(a4);
16070cba2b28SAlexander V. Chernikov 				break;
16080cba2b28SAlexander V. Chernikov 			}
16094e0a8b61SConrad Meyer 			if (lookup_host(n, &ipaddr) == 0) {
16104e0a8b61SConrad Meyer 				v->nh4 = ntohl(ipaddr.s_addr);
16110cba2b28SAlexander V. Chernikov 				break;
16124e0a8b61SConrad Meyer 			}
16130cba2b28SAlexander V. Chernikov 			etype = "ipv4";
1614ac35ff17SAlexander V. Chernikov 			break;
1615ac35ff17SAlexander V. Chernikov 		case IPFW_VTYPE_DSCP:
16160cba2b28SAlexander V. Chernikov 			if (isalpha(*n)) {
1617d91c61c3SAlexander V. Chernikov 				if ((dval = match_token(f_ipdscp, n)) != -1) {
1618d91c61c3SAlexander V. Chernikov 					v->dscp = dval;
1619ac35ff17SAlexander V. Chernikov 					break;
1620d91c61c3SAlexander V. Chernikov 				} else
16210cba2b28SAlexander V. Chernikov 					etype = "DSCP code";
16220cba2b28SAlexander V. Chernikov 			} else {
16230cba2b28SAlexander V. Chernikov 				v->dscp = strtol(n, &e, 10);
16240cba2b28SAlexander V. Chernikov 				if (v->dscp > 63 || *e != '\0')
16250cba2b28SAlexander V. Chernikov 					etype = "DSCP value";
1626ac35ff17SAlexander V. Chernikov 			}
16270cba2b28SAlexander V. Chernikov 			break;
16280cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_NH6:
16292530ed9eSAndrey V. Elsukov 			if (strchr(n, ':') != NULL) {
16302530ed9eSAndrey V. Elsukov 				memset(&hints, 0, sizeof(hints));
16312530ed9eSAndrey V. Elsukov 				hints.ai_family = AF_INET6;
16322530ed9eSAndrey V. Elsukov 				hints.ai_flags = AI_NUMERICHOST;
16332530ed9eSAndrey V. Elsukov 				if (getaddrinfo(n, NULL, &hints, &res) == 0) {
16342530ed9eSAndrey V. Elsukov 					v->nh6 = ((struct sockaddr_in6 *)
16352530ed9eSAndrey V. Elsukov 					    res->ai_addr)->sin6_addr;
16362530ed9eSAndrey V. Elsukov 					v->zoneid = ((struct sockaddr_in6 *)
16372530ed9eSAndrey V. Elsukov 					    res->ai_addr)->sin6_scope_id;
16382530ed9eSAndrey V. Elsukov 					freeaddrinfo(res);
16390cba2b28SAlexander V. Chernikov 					break;
16402530ed9eSAndrey V. Elsukov 				}
16412530ed9eSAndrey V. Elsukov 			}
16420cba2b28SAlexander V. Chernikov 			etype = "ipv6";
16430cba2b28SAlexander V. Chernikov 			break;
16440cba2b28SAlexander V. Chernikov 		}
16450cba2b28SAlexander V. Chernikov 
16460cba2b28SAlexander V. Chernikov 		if (etype != NULL)
16470cba2b28SAlexander V. Chernikov 			errx(EX_USAGE, "Unable to parse %s as %s", n, etype);
16480cba2b28SAlexander V. Chernikov 
16490cba2b28SAlexander V. Chernikov 		if (comma != NULL)
16500cba2b28SAlexander V. Chernikov 			*comma++ = ',';
16510cba2b28SAlexander V. Chernikov 
16520cba2b28SAlexander V. Chernikov 		if ((n = comma) != NULL)
16530cba2b28SAlexander V. Chernikov 			continue;
16540cba2b28SAlexander V. Chernikov 
16550cba2b28SAlexander V. Chernikov 		/* End of input. */
16560cba2b28SAlexander V. Chernikov 		if (vmask != 0)
16570cba2b28SAlexander V. Chernikov 			errx(EX_USAGE, "Not enough fields inside value");
16580cba2b28SAlexander V. Chernikov 	}
1659ac35ff17SAlexander V. Chernikov }
1660f1220db8SAlexander V. Chernikov 
1661f1220db8SAlexander V. Chernikov /*
1662f1220db8SAlexander V. Chernikov  * Compare table names.
1663f1220db8SAlexander V. Chernikov  * Honor number comparison.
1664f1220db8SAlexander V. Chernikov  */
1665f1220db8SAlexander V. Chernikov static int
1666f1220db8SAlexander V. Chernikov tablename_cmp(const void *a, const void *b)
1667f1220db8SAlexander V. Chernikov {
166856707beeSMark Johnston 	const ipfw_xtable_info *ia, *ib;
1669f1220db8SAlexander V. Chernikov 
167056707beeSMark Johnston 	ia = (const ipfw_xtable_info *)a;
167156707beeSMark Johnston 	ib = (const ipfw_xtable_info *)b;
1672f1220db8SAlexander V. Chernikov 
167368394ec8SAlexander V. Chernikov 	return (stringnum_cmp(ia->tablename, ib->tablename));
1674f1220db8SAlexander V. Chernikov }
1675f1220db8SAlexander V. Chernikov 
1676f1220db8SAlexander V. Chernikov /*
1677f1220db8SAlexander V. Chernikov  * Retrieves table list from kernel,
1678f1220db8SAlexander V. Chernikov  * optionally sorts it and calls requested function for each table.
1679f1220db8SAlexander V. Chernikov  * Returns 0 on success.
1680f1220db8SAlexander V. Chernikov  */
1681f1220db8SAlexander V. Chernikov static int
1682f1220db8SAlexander V. Chernikov tables_foreach(table_cb_t *f, void *arg, int sort)
1683f1220db8SAlexander V. Chernikov {
168428ea4fa3SAlexander V. Chernikov 	ipfw_obj_lheader *olh;
1685f1220db8SAlexander V. Chernikov 	ipfw_xtable_info *info;
1686f1220db8SAlexander V. Chernikov 	size_t sz;
168756707beeSMark Johnston 	uint32_t i;
168856707beeSMark Johnston 	int error;
1689f1220db8SAlexander V. Chernikov 
169028ea4fa3SAlexander V. Chernikov 	/* Start with reasonable default */
169128ea4fa3SAlexander V. Chernikov 	sz = sizeof(*olh) + 16 * sizeof(ipfw_xtable_info);
1692f1220db8SAlexander V. Chernikov 
169328ea4fa3SAlexander V. Chernikov 	for (;;) {
1694f1220db8SAlexander V. Chernikov 		if ((olh = calloc(1, sz)) == NULL)
1695f1220db8SAlexander V. Chernikov 			return (ENOMEM);
1696f1220db8SAlexander V. Chernikov 
1697f1220db8SAlexander V. Chernikov 		olh->size = sz;
1698b6462881SAlexander V. Chernikov 		if (do_get3(IP_FW_TABLES_XLIST, &olh->opheader, &sz) != 0) {
169928ea4fa3SAlexander V. Chernikov 			sz = olh->size;
170068bde59eSAlexander V. Chernikov 			free(olh);
170168bde59eSAlexander V. Chernikov 			if (errno != ENOMEM)
1702b6462881SAlexander V. Chernikov 				return (errno);
170368bde59eSAlexander V. Chernikov 			continue;
1704f1220db8SAlexander V. Chernikov 		}
1705f1220db8SAlexander V. Chernikov 
1706f1220db8SAlexander V. Chernikov 		if (sort != 0)
1707e028ccdaSAndrey V. Elsukov 			qsort(olh + 1, olh->count, olh->objsize,
1708e028ccdaSAndrey V. Elsukov 			    tablename_cmp);
1709f1220db8SAlexander V. Chernikov 
1710f1220db8SAlexander V. Chernikov 		info = (ipfw_xtable_info *)(olh + 1);
1711f1220db8SAlexander V. Chernikov 		for (i = 0; i < olh->count; i++) {
171256707beeSMark Johnston 			if (g_co.use_set == 0 || info->set == g_co.use_set - 1)
1713e028ccdaSAndrey V. Elsukov 				error = f(info, arg);
1714e028ccdaSAndrey V. Elsukov 			info = (ipfw_xtable_info *)((caddr_t)info +
1715e028ccdaSAndrey V. Elsukov 			    olh->objsize);
1716f1220db8SAlexander V. Chernikov 		}
1717f1220db8SAlexander V. Chernikov 		free(olh);
171828ea4fa3SAlexander V. Chernikov 		break;
171928ea4fa3SAlexander V. Chernikov 	}
1720f1220db8SAlexander V. Chernikov 	return (0);
1721f1220db8SAlexander V. Chernikov }
1722f1220db8SAlexander V. Chernikov 
172328ea4fa3SAlexander V. Chernikov 
1724f1220db8SAlexander V. Chernikov /*
1725f1220db8SAlexander V. Chernikov  * Retrieves all entries for given table @i in
1726720ee730SAlexander V. Chernikov  * eXtended format. Allocate buffer large enough
1727720ee730SAlexander V. Chernikov  * to store result. Called needs to free it later.
1728f1220db8SAlexander V. Chernikov  *
1729f1220db8SAlexander V. Chernikov  * Returns 0 on success.
1730f1220db8SAlexander V. Chernikov  */
1731f1220db8SAlexander V. Chernikov static int
1732720ee730SAlexander V. Chernikov table_do_get_list(ipfw_xtable_info *i, ipfw_obj_header **poh)
1733f1220db8SAlexander V. Chernikov {
1734720ee730SAlexander V. Chernikov 	ipfw_obj_header *oh;
1735f1220db8SAlexander V. Chernikov 	size_t sz;
1736b6462881SAlexander V. Chernikov 	int c;
1737f1220db8SAlexander V. Chernikov 
173881d3153dSAlexander V. Chernikov 	sz = 0;
1739720ee730SAlexander V. Chernikov 	oh = NULL;
1740720ee730SAlexander V. Chernikov 	for (c = 0; c < 8; c++) {
174181d3153dSAlexander V. Chernikov 		if (sz < i->size)
1742720ee730SAlexander V. Chernikov 			sz = i->size + 44;
1743720ee730SAlexander V. Chernikov 		if (oh != NULL)
1744720ee730SAlexander V. Chernikov 			free(oh);
1745720ee730SAlexander V. Chernikov 		if ((oh = calloc(1, sz)) == NULL)
1746720ee730SAlexander V. Chernikov 			continue;
1747720ee730SAlexander V. Chernikov 		table_fill_objheader(oh, i);
1748d3a4f924SAlexander V. Chernikov 		oh->opheader.version = 1; /* Current version */
1749b6462881SAlexander V. Chernikov 		if (do_get3(IP_FW_TABLE_XLIST, &oh->opheader, &sz) == 0) {
1750720ee730SAlexander V. Chernikov 			*poh = oh;
1751720ee730SAlexander V. Chernikov 			return (0);
175281d3153dSAlexander V. Chernikov 		}
1753f1220db8SAlexander V. Chernikov 
1754b6462881SAlexander V. Chernikov 		if (errno != ENOMEM)
1755720ee730SAlexander V. Chernikov 			break;
1756720ee730SAlexander V. Chernikov 	}
1757720ee730SAlexander V. Chernikov 	free(oh);
1758720ee730SAlexander V. Chernikov 
1759b6462881SAlexander V. Chernikov 	return (errno);
1760f1220db8SAlexander V. Chernikov }
1761f1220db8SAlexander V. Chernikov 
1762f1220db8SAlexander V. Chernikov /*
1763f1220db8SAlexander V. Chernikov  * Shows all entries from @oh in human-readable format
1764f1220db8SAlexander V. Chernikov  */
1765f1220db8SAlexander V. Chernikov static void
1766f1220db8SAlexander V. Chernikov table_show_list(ipfw_obj_header *oh, int need_header)
1767f1220db8SAlexander V. Chernikov {
176881d3153dSAlexander V. Chernikov 	ipfw_obj_tentry *tent;
176981d3153dSAlexander V. Chernikov 	uint32_t count;
1770f1220db8SAlexander V. Chernikov 	ipfw_xtable_info *i;
1771f1220db8SAlexander V. Chernikov 
1772f1220db8SAlexander V. Chernikov 	i = (ipfw_xtable_info *)(oh + 1);
177381d3153dSAlexander V. Chernikov 	tent = (ipfw_obj_tentry *)(i + 1);
1774f1220db8SAlexander V. Chernikov 
1775f1220db8SAlexander V. Chernikov 	if (need_header)
1776f1220db8SAlexander V. Chernikov 		printf("--- table(%s), set(%u) ---\n", i->tablename, i->set);
1777f1220db8SAlexander V. Chernikov 
1778f1220db8SAlexander V. Chernikov 	count = i->count;
1779f1220db8SAlexander V. Chernikov 	while (count > 0) {
178081d3153dSAlexander V. Chernikov 		table_show_entry(i, tent);
178181d3153dSAlexander V. Chernikov 		tent = (ipfw_obj_tentry *)((caddr_t)tent + tent->head.length);
178281d3153dSAlexander V. Chernikov 		count--;
178381d3153dSAlexander V. Chernikov 	}
178481d3153dSAlexander V. Chernikov }
178581d3153dSAlexander V. Chernikov 
178681d3153dSAlexander V. Chernikov static void
17870cba2b28SAlexander V. Chernikov table_show_value(char *buf, size_t bufsize, ipfw_table_value *v,
17880cba2b28SAlexander V. Chernikov     uint32_t vmask, int print_ip)
17890cba2b28SAlexander V. Chernikov {
17902530ed9eSAndrey V. Elsukov 	char abuf[INET6_ADDRSTRLEN + IF_NAMESIZE + 2];
17912530ed9eSAndrey V. Elsukov 	struct sockaddr_in6 sa6;
17920cba2b28SAlexander V. Chernikov 	uint32_t flag, i, l;
17930cba2b28SAlexander V. Chernikov 	size_t sz;
17940cba2b28SAlexander V. Chernikov 	struct in_addr a4;
17950cba2b28SAlexander V. Chernikov 
17960cba2b28SAlexander V. Chernikov 	sz = bufsize;
17970cba2b28SAlexander V. Chernikov 
17980cba2b28SAlexander V. Chernikov 	/*
17990cba2b28SAlexander V. Chernikov 	 * Some shorthands for printing values:
18000cba2b28SAlexander V. Chernikov 	 * legacy assumes all values are equal, so keep the first one.
18010cba2b28SAlexander V. Chernikov 	 */
18020cba2b28SAlexander V. Chernikov 	if (vmask == IPFW_VTYPE_LEGACY) {
18030cba2b28SAlexander V. Chernikov 		if (print_ip != 0) {
18040cba2b28SAlexander V. Chernikov 			flag = htonl(v->tag);
18050cba2b28SAlexander V. Chernikov 			inet_ntop(AF_INET, &flag, buf, sz);
18060cba2b28SAlexander V. Chernikov 		} else
18070cba2b28SAlexander V. Chernikov 			snprintf(buf, sz, "%u", v->tag);
18080cba2b28SAlexander V. Chernikov 		return;
18090cba2b28SAlexander V. Chernikov 	}
18100cba2b28SAlexander V. Chernikov 
181156707beeSMark Johnston 	for (i = 1; i < (1u << 31); i *= 2) {
18120cba2b28SAlexander V. Chernikov 		if ((flag = (vmask & i)) == 0)
18130cba2b28SAlexander V. Chernikov 			continue;
18140cba2b28SAlexander V. Chernikov 		l = 0;
18150cba2b28SAlexander V. Chernikov 
18160cba2b28SAlexander V. Chernikov 		switch (flag) {
18170cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_TAG:
18180cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%u,", v->tag);
18190cba2b28SAlexander V. Chernikov 			break;
18200cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_PIPE:
18210cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%u,", v->pipe);
18220cba2b28SAlexander V. Chernikov 			break;
18230cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_DIVERT:
18240cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%d,", v->divert);
18250cba2b28SAlexander V. Chernikov 			break;
18260cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_SKIPTO:
18270cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%d,", v->skipto);
18280cba2b28SAlexander V. Chernikov 			break;
18290cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_NETGRAPH:
18300cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%u,", v->netgraph);
18310cba2b28SAlexander V. Chernikov 			break;
18320cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_FIB:
18330cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%u,", v->fib);
18340cba2b28SAlexander V. Chernikov 			break;
18350cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_NAT:
18360cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%u,", v->nat);
18370cba2b28SAlexander V. Chernikov 			break;
18380cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_LIMIT:
18390cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%u,", v->limit);
18400cba2b28SAlexander V. Chernikov 			break;
18410cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_NH4:
18420cba2b28SAlexander V. Chernikov 			a4.s_addr = htonl(v->nh4);
18430cba2b28SAlexander V. Chernikov 			inet_ntop(AF_INET, &a4, abuf, sizeof(abuf));
18440cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%s,", abuf);
18450cba2b28SAlexander V. Chernikov 			break;
18460cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_DSCP:
18470cba2b28SAlexander V. Chernikov 			l = snprintf(buf, sz, "%d,", v->dscp);
18480cba2b28SAlexander V. Chernikov 			break;
18490cba2b28SAlexander V. Chernikov 		case IPFW_VTYPE_NH6:
18502530ed9eSAndrey V. Elsukov 			sa6.sin6_family = AF_INET6;
18512530ed9eSAndrey V. Elsukov 			sa6.sin6_len = sizeof(sa6);
18522530ed9eSAndrey V. Elsukov 			sa6.sin6_addr = v->nh6;
18532530ed9eSAndrey V. Elsukov 			sa6.sin6_port = 0;
18542530ed9eSAndrey V. Elsukov 			sa6.sin6_scope_id = v->zoneid;
18552530ed9eSAndrey V. Elsukov 			if (getnameinfo((const struct sockaddr *)&sa6,
18562530ed9eSAndrey V. Elsukov 			    sa6.sin6_len, abuf, sizeof(abuf), NULL, 0,
18572530ed9eSAndrey V. Elsukov 			    NI_NUMERICHOST) == 0)
18580cba2b28SAlexander V. Chernikov 				l = snprintf(buf, sz, "%s,", abuf);
18590cba2b28SAlexander V. Chernikov 			break;
18600cba2b28SAlexander V. Chernikov 		}
18610cba2b28SAlexander V. Chernikov 
18620cba2b28SAlexander V. Chernikov 		buf += l;
18630cba2b28SAlexander V. Chernikov 		sz -= l;
18640cba2b28SAlexander V. Chernikov 	}
18650cba2b28SAlexander V. Chernikov 
18660cba2b28SAlexander V. Chernikov 	if (sz != bufsize)
18670cba2b28SAlexander V. Chernikov 		*(buf - 1) = '\0';
18680cba2b28SAlexander V. Chernikov }
18690cba2b28SAlexander V. Chernikov 
18700cba2b28SAlexander V. Chernikov static void
187181d3153dSAlexander V. Chernikov table_show_entry(ipfw_xtable_info *i, ipfw_obj_tentry *tent)
187281d3153dSAlexander V. Chernikov {
187356707beeSMark Johnston 	char tbuf[128], pval[128];
187456707beeSMark Johnston 	const char *comma;
1875914bffb6SAlexander V. Chernikov 	void *paddr;
1876914bffb6SAlexander V. Chernikov 	struct tflow_entry *tfe;
187781d3153dSAlexander V. Chernikov 
18780cba2b28SAlexander V. Chernikov 	table_show_value(pval, sizeof(pval), &tent->v.value, i->vmask,
187956707beeSMark Johnston 	    g_co.do_value_as_ip);
1880914bffb6SAlexander V. Chernikov 
1881f1220db8SAlexander V. Chernikov 	switch (i->type) {
1882c21034b7SAlexander V. Chernikov 	case IPFW_TABLE_ADDR:
1883f1220db8SAlexander V. Chernikov 		/* IPv4 or IPv6 prefixes */
188481d3153dSAlexander V. Chernikov 		inet_ntop(tent->subtype, &tent->k, tbuf, sizeof(tbuf));
1885914bffb6SAlexander V. Chernikov 		printf("%s/%u %s\n", tbuf, tent->masklen, pval);
1886f1220db8SAlexander V. Chernikov 		break;
1887f1220db8SAlexander V. Chernikov 	case IPFW_TABLE_INTERFACE:
1888f1220db8SAlexander V. Chernikov 		/* Interface names */
1889914bffb6SAlexander V. Chernikov 		printf("%s %s\n", tent->k.iface, pval);
1890b23d5de9SAlexander V. Chernikov 		break;
1891b23d5de9SAlexander V. Chernikov 	case IPFW_TABLE_NUMBER:
1892b23d5de9SAlexander V. Chernikov 		/* numbers */
1893914bffb6SAlexander V. Chernikov 		printf("%u %s\n", tent->k.key, pval);
1894b23d5de9SAlexander V. Chernikov 		break;
1895914bffb6SAlexander V. Chernikov 	case IPFW_TABLE_FLOW:
1896914bffb6SAlexander V. Chernikov 		/* flows */
1897914bffb6SAlexander V. Chernikov 		tfe = &tent->k.flow;
1898914bffb6SAlexander V. Chernikov 		comma = "";
1899914bffb6SAlexander V. Chernikov 
1900914bffb6SAlexander V. Chernikov 		if ((i->tflags & IPFW_TFFLAG_SRCIP) != 0) {
1901914bffb6SAlexander V. Chernikov 			if (tfe->af == AF_INET)
1902914bffb6SAlexander V. Chernikov 				paddr = &tfe->a.a4.sip;
1903914bffb6SAlexander V. Chernikov 			else
1904914bffb6SAlexander V. Chernikov 				paddr = &tfe->a.a6.sip6;
1905914bffb6SAlexander V. Chernikov 
1906914bffb6SAlexander V. Chernikov 			inet_ntop(tfe->af, paddr, tbuf, sizeof(tbuf));
1907914bffb6SAlexander V. Chernikov 			printf("%s%s", comma, tbuf);
1908914bffb6SAlexander V. Chernikov 			comma = ",";
1909914bffb6SAlexander V. Chernikov 		}
1910914bffb6SAlexander V. Chernikov 
1911914bffb6SAlexander V. Chernikov 		if ((i->tflags & IPFW_TFFLAG_PROTO) != 0) {
1912914bffb6SAlexander V. Chernikov 			printf("%s%d", comma, tfe->proto);
1913914bffb6SAlexander V. Chernikov 			comma = ",";
1914914bffb6SAlexander V. Chernikov 		}
1915914bffb6SAlexander V. Chernikov 
1916914bffb6SAlexander V. Chernikov 		if ((i->tflags & IPFW_TFFLAG_SRCPORT) != 0) {
1917914bffb6SAlexander V. Chernikov 			printf("%s%d", comma, ntohs(tfe->sport));
1918914bffb6SAlexander V. Chernikov 			comma = ",";
1919914bffb6SAlexander V. Chernikov 		}
1920914bffb6SAlexander V. Chernikov 		if ((i->tflags & IPFW_TFFLAG_DSTIP) != 0) {
1921914bffb6SAlexander V. Chernikov 			if (tfe->af == AF_INET)
1922914bffb6SAlexander V. Chernikov 				paddr = &tfe->a.a4.dip;
1923914bffb6SAlexander V. Chernikov 			else
1924914bffb6SAlexander V. Chernikov 				paddr = &tfe->a.a6.dip6;
1925914bffb6SAlexander V. Chernikov 
1926914bffb6SAlexander V. Chernikov 			inet_ntop(tfe->af, paddr, tbuf, sizeof(tbuf));
1927914bffb6SAlexander V. Chernikov 			printf("%s%s", comma, tbuf);
1928914bffb6SAlexander V. Chernikov 			comma = ",";
1929914bffb6SAlexander V. Chernikov 		}
1930914bffb6SAlexander V. Chernikov 
1931914bffb6SAlexander V. Chernikov 		if ((i->tflags & IPFW_TFFLAG_DSTPORT) != 0) {
1932914bffb6SAlexander V. Chernikov 			printf("%s%d", comma, ntohs(tfe->dport));
1933914bffb6SAlexander V. Chernikov 			comma = ",";
1934914bffb6SAlexander V. Chernikov 		}
1935914bffb6SAlexander V. Chernikov 
1936914bffb6SAlexander V. Chernikov 		printf(" %s\n", pval);
1937f1220db8SAlexander V. Chernikov 	}
1938f1220db8SAlexander V. Chernikov }
1939f1220db8SAlexander V. Chernikov 
19409d099b4fSAlexander V. Chernikov static int
19410cba2b28SAlexander V. Chernikov table_do_get_stdlist(uint16_t opcode, ipfw_obj_lheader **polh)
19429d099b4fSAlexander V. Chernikov {
19439d099b4fSAlexander V. Chernikov 	ipfw_obj_lheader req, *olh;
19449d099b4fSAlexander V. Chernikov 	size_t sz;
19459d099b4fSAlexander V. Chernikov 
19469d099b4fSAlexander V. Chernikov 	memset(&req, 0, sizeof(req));
19479d099b4fSAlexander V. Chernikov 	sz = sizeof(req);
19489d099b4fSAlexander V. Chernikov 
1949b6462881SAlexander V. Chernikov 	if (do_get3(opcode, &req.opheader, &sz) != 0)
1950b6462881SAlexander V. Chernikov 		if (errno != ENOMEM)
1951b6462881SAlexander V. Chernikov 			return (errno);
19529d099b4fSAlexander V. Chernikov 
19539d099b4fSAlexander V. Chernikov 	sz = req.size;
19549d099b4fSAlexander V. Chernikov 	if ((olh = calloc(1, sz)) == NULL)
19559d099b4fSAlexander V. Chernikov 		return (ENOMEM);
19569d099b4fSAlexander V. Chernikov 
19579d099b4fSAlexander V. Chernikov 	olh->size = sz;
1958b6462881SAlexander V. Chernikov 	if (do_get3(opcode, &olh->opheader, &sz) != 0) {
19599d099b4fSAlexander V. Chernikov 		free(olh);
1960b6462881SAlexander V. Chernikov 		return (errno);
19619d099b4fSAlexander V. Chernikov 	}
19629d099b4fSAlexander V. Chernikov 
19639d099b4fSAlexander V. Chernikov 	*polh = olh;
19649d099b4fSAlexander V. Chernikov 	return (0);
19659d099b4fSAlexander V. Chernikov }
19669d099b4fSAlexander V. Chernikov 
19670cba2b28SAlexander V. Chernikov static int
19680cba2b28SAlexander V. Chernikov table_do_get_algolist(ipfw_obj_lheader **polh)
19690cba2b28SAlexander V. Chernikov {
19700cba2b28SAlexander V. Chernikov 
19710cba2b28SAlexander V. Chernikov 	return (table_do_get_stdlist(IP_FW_TABLES_ALIST, polh));
19720cba2b28SAlexander V. Chernikov }
19730cba2b28SAlexander V. Chernikov 
19740cba2b28SAlexander V. Chernikov static int
19750cba2b28SAlexander V. Chernikov table_do_get_vlist(ipfw_obj_lheader **polh)
19760cba2b28SAlexander V. Chernikov {
19770cba2b28SAlexander V. Chernikov 
19780cba2b28SAlexander V. Chernikov 	return (table_do_get_stdlist(IP_FW_TABLE_VLIST, polh));
19790cba2b28SAlexander V. Chernikov }
19800cba2b28SAlexander V. Chernikov 
19819d099b4fSAlexander V. Chernikov void
198256707beeSMark Johnston ipfw_list_ta(int ac __unused, char *av[] __unused)
19839d099b4fSAlexander V. Chernikov {
19849d099b4fSAlexander V. Chernikov 	ipfw_obj_lheader *olh;
19859d099b4fSAlexander V. Chernikov 	ipfw_ta_info *info;
19869d099b4fSAlexander V. Chernikov 	const char *atype;
198756707beeSMark Johnston 	uint32_t i;
198856707beeSMark Johnston 	int error;
19899d099b4fSAlexander V. Chernikov 
19909d099b4fSAlexander V. Chernikov 	error = table_do_get_algolist(&olh);
19919d099b4fSAlexander V. Chernikov 	if (error != 0)
19929d099b4fSAlexander V. Chernikov 		err(EX_OSERR, "Unable to request algorithm list");
19939d099b4fSAlexander V. Chernikov 
19949d099b4fSAlexander V. Chernikov 	info = (ipfw_ta_info *)(olh + 1);
19959d099b4fSAlexander V. Chernikov 	for (i = 0; i < olh->count; i++) {
19969d099b4fSAlexander V. Chernikov 		if ((atype = match_value(tabletypes, info->type)) == NULL)
19979d099b4fSAlexander V. Chernikov 			atype = "unknown";
19988ce7a2bcSAlexander V. Chernikov 		printf("--- %s ---\n", info->algoname);
19998ce7a2bcSAlexander V. Chernikov 		printf(" type: %s\n refcount: %u\n", atype, info->refcnt);
20009d099b4fSAlexander V. Chernikov 
20019d099b4fSAlexander V. Chernikov 		info = (ipfw_ta_info *)((caddr_t)info + olh->objsize);
20029d099b4fSAlexander V. Chernikov 	}
20039d099b4fSAlexander V. Chernikov 
20049d099b4fSAlexander V. Chernikov 	free(olh);
20059d099b4fSAlexander V. Chernikov }
20069d099b4fSAlexander V. Chernikov 
20070cba2b28SAlexander V. Chernikov 
20080cba2b28SAlexander V. Chernikov /* Copy of current kernel table_value structure */
20090cba2b28SAlexander V. Chernikov struct _table_value {
20100cba2b28SAlexander V. Chernikov 	uint32_t	tag;		/* O_TAG/O_TAGGED */
20110cba2b28SAlexander V. Chernikov 	uint32_t	pipe;		/* O_PIPE/O_QUEUE */
20120cba2b28SAlexander V. Chernikov 	uint16_t	divert;		/* O_DIVERT/O_TEE */
20130cba2b28SAlexander V. Chernikov 	uint16_t	skipto;		/* skipto, CALLRET */
20140cba2b28SAlexander V. Chernikov 	uint32_t	netgraph;	/* O_NETGRAPH/O_NGTEE */
20150cba2b28SAlexander V. Chernikov 	uint32_t	fib;		/* O_SETFIB */
20160cba2b28SAlexander V. Chernikov 	uint32_t	nat;		/* O_NAT */
20170cba2b28SAlexander V. Chernikov 	uint32_t	nh4;
20180cba2b28SAlexander V. Chernikov 	uint8_t		dscp;
20192530ed9eSAndrey V. Elsukov 	uint8_t		spare0;
20202530ed9eSAndrey V. Elsukov 	uint16_t	spare1;
20210cba2b28SAlexander V. Chernikov 	/* -- 32 bytes -- */
20220cba2b28SAlexander V. Chernikov 	struct in6_addr	nh6;
20230cba2b28SAlexander V. Chernikov 	uint32_t	limit;		/* O_LIMIT */
20242530ed9eSAndrey V. Elsukov 	uint32_t	zoneid;
20250cba2b28SAlexander V. Chernikov 	uint64_t	refcnt;		/* Number of references */
20260cba2b28SAlexander V. Chernikov };
20270cba2b28SAlexander V. Chernikov 
202856707beeSMark Johnston static int
20290cba2b28SAlexander V. Chernikov compare_values(const void *_a, const void *_b)
20300cba2b28SAlexander V. Chernikov {
203156707beeSMark Johnston 	const struct _table_value *a, *b;
20320cba2b28SAlexander V. Chernikov 
203356707beeSMark Johnston 	a = (const struct _table_value *)_a;
203456707beeSMark Johnston 	b = (const struct _table_value *)_b;
20350cba2b28SAlexander V. Chernikov 
20360cba2b28SAlexander V. Chernikov 	if (a->spare1 < b->spare1)
20370cba2b28SAlexander V. Chernikov 		return (-1);
20380cba2b28SAlexander V. Chernikov 	else if (a->spare1 > b->spare1)
20390cba2b28SAlexander V. Chernikov 		return (1);
20400cba2b28SAlexander V. Chernikov 
20410cba2b28SAlexander V. Chernikov 	return (0);
20420cba2b28SAlexander V. Chernikov }
20430cba2b28SAlexander V. Chernikov 
20440cba2b28SAlexander V. Chernikov void
204556707beeSMark Johnston ipfw_list_values(int ac __unused, char *av[] __unused)
20460cba2b28SAlexander V. Chernikov {
204756707beeSMark Johnston 	char buf[128];
20480cba2b28SAlexander V. Chernikov 	ipfw_obj_lheader *olh;
20490cba2b28SAlexander V. Chernikov 	struct _table_value *v;
205056707beeSMark Johnston 	uint32_t i, vmask;
205156707beeSMark Johnston 	int error;
20520cba2b28SAlexander V. Chernikov 
20530cba2b28SAlexander V. Chernikov 	error = table_do_get_vlist(&olh);
20540cba2b28SAlexander V. Chernikov 	if (error != 0)
20550cba2b28SAlexander V. Chernikov 		err(EX_OSERR, "Unable to request value list");
20560cba2b28SAlexander V. Chernikov 
20570cba2b28SAlexander V. Chernikov 	vmask = 0x7FFFFFFF; /* Similar to IPFW_VTYPE_LEGACY */
20580cba2b28SAlexander V. Chernikov 
20590cba2b28SAlexander V. Chernikov 	table_print_valheader(buf, sizeof(buf), vmask);
20600cba2b28SAlexander V. Chernikov 	printf("HEADER: %s\n", buf);
20610cba2b28SAlexander V. Chernikov 	v = (struct _table_value *)(olh + 1);
20620cba2b28SAlexander V. Chernikov 	qsort(v, olh->count, olh->objsize, compare_values);
20630cba2b28SAlexander V. Chernikov 	for (i = 0; i < olh->count; i++) {
20640cba2b28SAlexander V. Chernikov 		table_show_value(buf, sizeof(buf), (ipfw_table_value *)v,
20650cba2b28SAlexander V. Chernikov 		    vmask, 0);
2066d3de6c1dSAlexander V. Chernikov 		printf("[%u] refs=%lu %s\n", v->spare1, (u_long)v->refcnt, buf);
20670cba2b28SAlexander V. Chernikov 		v = (struct _table_value *)((caddr_t)v + olh->objsize);
20680cba2b28SAlexander V. Chernikov 	}
20690cba2b28SAlexander V. Chernikov 
20700cba2b28SAlexander V. Chernikov 	free(olh);
20710cba2b28SAlexander V. Chernikov }
20720cba2b28SAlexander V. Chernikov 
20736c2997ffSAlexander V. Chernikov int
20742acdf79fSAndrey V. Elsukov table_check_name(const char *tablename)
20756c2997ffSAlexander V. Chernikov {
20766c2997ffSAlexander V. Chernikov 
20772acdf79fSAndrey V. Elsukov 	if (ipfw_check_object_name(tablename) != 0)
20786c2997ffSAlexander V. Chernikov 		return (EINVAL);
2079ac35ff17SAlexander V. Chernikov 	/* Restrict some 'special' names */
2080ac35ff17SAlexander V. Chernikov 	if (strcmp(tablename, "all") == 0)
2081ac35ff17SAlexander V. Chernikov 		return (EINVAL);
20826c2997ffSAlexander V. Chernikov 	return (0);
20836c2997ffSAlexander V. Chernikov }
20846c2997ffSAlexander V. Chernikov 
2085