1/*-
2 * Copyright (c) 2002-2003 Luigi Rizzo
3 * Copyright (c) 1996 Alex Nash, Paul Traina, Poul-Henning Kamp
4 * Copyright (c) 1994 Ugen J.S.Antsilevich
5 *
6 * Idea and grammar partially left from:
7 * Copyright (c) 1993 Daniel Boulet
8 *
9 * Redistribution and use in source forms, with and without modification,
10 * are permitted provided that this entire comment appears intact.
11 *
12 * Redistribution in binary form may occur without any restrictions.
13 * Obviously, it would be nice if you gave credit where credit is due
14 * but requiring it would be too onerous.
15 *
16 * This software is provided ``AS IS'' without any warranties of any kind.
17 *
18 * NEW command line interface for IP firewall facility
19 *
20 * $FreeBSD$
21 */
22
23#include <sys/types.h>
24#include <sys/param.h>
25#include <sys/socket.h>
26#include <sys/sockio.h>
27#include <sys/sysctl.h>
28
29#include "ipfw2.h"
30
31#include <ctype.h>
32#include <err.h>
33#include <errno.h>
34#include <grp.h>
35#include <jail.h>
36#include <netdb.h>
37#include <pwd.h>
38#include <stdio.h>
39#include <stdarg.h>
40#include <stdint.h>
41#include <stdlib.h>
42#include <string.h>
43#include <sysexits.h>
44#include <time.h>	/* ctime */
45#include <timeconv.h>	/* _long_to_time */
46#include <unistd.h>
47#include <fcntl.h>
48#include <stddef.h>	/* offsetof */
49
50#include <net/ethernet.h>
51#include <net/if.h>		/* only IFNAMSIZ */
52#include <netinet/in.h>
53#include <netinet/in_systm.h>	/* only n_short, n_long */
54#include <netinet/ip.h>
55#include <netinet/ip_icmp.h>
56#include <netinet/ip_fw.h>
57#include <netinet/tcp.h>
58#include <arpa/inet.h>
59
60struct cmdline_opts g_co;	/* global options */
61
62struct format_opts {
63	int bcwidth;
64	int pcwidth;
65	int show_counters;
66	int show_time;		/* show timestamp */
67	uint32_t set_mask;	/* enabled sets mask */
68	uint32_t flags;		/* request flags */
69	uint32_t first;		/* first rule to request */
70	uint32_t last;		/* last rule to request */
71	uint32_t dcnt;		/* number of dynamic states */
72	ipfw_obj_ctlv *tstate;	/* table state data */
73};
74
75int resvd_set_number = RESVD_SET;
76
77static int ipfw_socket = -1;
78
79#define	CHECK_LENGTH(v, len) do {				\
80	if ((v) < (len))					\
81		errx(EX_DATAERR, "Rule too long");		\
82	} while (0)
83/*
84 * Check if we have enough space in cmd buffer. Note that since
85 * first 8? u32 words are reserved by reserved header, full cmd
86 * buffer can't be used, so we need to protect from buffer overrun
87 * only. At the beginning, cblen is less than actual buffer size by
88 * size of ipfw_insn_u32 instruction + 1 u32 work. This eliminates need
89 * for checking small instructions fitting in given range.
90 * We also (ab)use the fact that ipfw_insn is always the first field
91 * for any custom instruction.
92 */
93#define	CHECK_CMDLEN	CHECK_LENGTH(cblen, F_LEN((ipfw_insn *)cmd))
94
95#define GET_UINT_ARG(arg, min, max, tok, s_x) do {			\
96	if (!av[0])							\
97		errx(EX_USAGE, "%s: missing argument", match_value(s_x, tok)); \
98	if (_substrcmp(*av, "tablearg") == 0) {				\
99		arg = IP_FW_TARG;					\
100		break;							\
101	}								\
102									\
103	{								\
104	long _xval;							\
105	char *end;							\
106									\
107	_xval = strtol(*av, &end, 10);					\
108									\
109	if (!isdigit(**av) || *end != '\0' || (_xval == 0 && errno == EINVAL)) \
110		errx(EX_DATAERR, "%s: invalid argument: %s",		\
111		    match_value(s_x, tok), *av);			\
112									\
113	if (errno == ERANGE || _xval < min || _xval > max)		\
114		errx(EX_DATAERR, "%s: argument is out of range (%u..%u): %s", \
115		    match_value(s_x, tok), min, max, *av);		\
116									\
117	if (_xval == IP_FW_TARG)					\
118		errx(EX_DATAERR, "%s: illegal argument value: %s",	\
119		    match_value(s_x, tok), *av);			\
120	arg = _xval;							\
121	}								\
122} while (0)
123
124static struct _s_x f_tcpflags[] = {
125	{ "syn", TH_SYN },
126	{ "fin", TH_FIN },
127	{ "ack", TH_ACK },
128	{ "psh", TH_PUSH },
129	{ "rst", TH_RST },
130	{ "urg", TH_URG },
131	{ "tcp flag", 0 },
132	{ NULL,	0 }
133};
134
135static struct _s_x f_tcpopts[] = {
136	{ "mss",	IP_FW_TCPOPT_MSS },
137	{ "maxseg",	IP_FW_TCPOPT_MSS },
138	{ "window",	IP_FW_TCPOPT_WINDOW },
139	{ "sack",	IP_FW_TCPOPT_SACK },
140	{ "ts",		IP_FW_TCPOPT_TS },
141	{ "timestamp",	IP_FW_TCPOPT_TS },
142	{ "cc",		IP_FW_TCPOPT_CC },
143	{ "tcp option",	0 },
144	{ NULL,	0 }
145};
146
147/*
148 * IP options span the range 0 to 255 so we need to remap them
149 * (though in fact only the low 5 bits are significant).
150 */
151static struct _s_x f_ipopts[] = {
152	{ "ssrr",	IP_FW_IPOPT_SSRR},
153	{ "lsrr",	IP_FW_IPOPT_LSRR},
154	{ "rr",		IP_FW_IPOPT_RR},
155	{ "ts",		IP_FW_IPOPT_TS},
156	{ "ip option",	0 },
157	{ NULL,	0 }
158};
159
160static struct _s_x f_iptos[] = {
161	{ "lowdelay",	IPTOS_LOWDELAY},
162	{ "throughput",	IPTOS_THROUGHPUT},
163	{ "reliability", IPTOS_RELIABILITY},
164	{ "mincost",	IPTOS_MINCOST},
165	{ "congestion",	IPTOS_ECN_CE},
166	{ "ecntransport", IPTOS_ECN_ECT0},
167	{ "ip tos option", 0},
168	{ NULL,	0 }
169};
170
171static struct _s_x f_ipoff[] = {
172	{ "rf", IP_RF >> 8 },
173	{ "df", IP_DF >> 8 },
174	{ "mf", IP_MF >> 8 },
175	{ "offset", 0x1 },
176	{ NULL, 0}
177};
178
179struct _s_x f_ipdscp[] = {
180	{ "af11", IPTOS_DSCP_AF11 >> 2 },	/* 001010 */
181	{ "af12", IPTOS_DSCP_AF12 >> 2 },	/* 001100 */
182	{ "af13", IPTOS_DSCP_AF13 >> 2 },	/* 001110 */
183	{ "af21", IPTOS_DSCP_AF21 >> 2 },	/* 010010 */
184	{ "af22", IPTOS_DSCP_AF22 >> 2 },	/* 010100 */
185	{ "af23", IPTOS_DSCP_AF23 >> 2 },	/* 010110 */
186	{ "af31", IPTOS_DSCP_AF31 >> 2 },	/* 011010 */
187	{ "af32", IPTOS_DSCP_AF32 >> 2 },	/* 011100 */
188	{ "af33", IPTOS_DSCP_AF33 >> 2 },	/* 011110 */
189	{ "af41", IPTOS_DSCP_AF41 >> 2 },	/* 100010 */
190	{ "af42", IPTOS_DSCP_AF42 >> 2 },	/* 100100 */
191	{ "af43", IPTOS_DSCP_AF43 >> 2 },	/* 100110 */
192	{ "be", IPTOS_DSCP_CS0 >> 2 }, 	/* 000000 */
193	{ "ef", IPTOS_DSCP_EF >> 2 },	/* 101110 */
194	{ "cs0", IPTOS_DSCP_CS0 >> 2 },	/* 000000 */
195	{ "cs1", IPTOS_DSCP_CS1 >> 2 },	/* 001000 */
196	{ "cs2", IPTOS_DSCP_CS2 >> 2 },	/* 010000 */
197	{ "cs3", IPTOS_DSCP_CS3 >> 2 },	/* 011000 */
198	{ "cs4", IPTOS_DSCP_CS4 >> 2 },	/* 100000 */
199	{ "cs5", IPTOS_DSCP_CS5 >> 2 },	/* 101000 */
200	{ "cs6", IPTOS_DSCP_CS6 >> 2 },	/* 110000 */
201	{ "cs7", IPTOS_DSCP_CS7 >> 2 },	/* 100000 */
202	{ NULL, 0 }
203};
204
205static struct _s_x limit_masks[] = {
206	{"all",		DYN_SRC_ADDR|DYN_SRC_PORT|DYN_DST_ADDR|DYN_DST_PORT},
207	{"src-addr",	DYN_SRC_ADDR},
208	{"src-port",	DYN_SRC_PORT},
209	{"dst-addr",	DYN_DST_ADDR},
210	{"dst-port",	DYN_DST_PORT},
211	{NULL,		0}
212};
213
214/*
215 * we use IPPROTO_ETHERTYPE as a fake protocol id to call the print routines
216 * This is only used in this code.
217 */
218#define IPPROTO_ETHERTYPE	0x1000
219static struct _s_x ether_types[] = {
220    /*
221     * Note, we cannot use "-:&/" in the names because they are field
222     * separators in the type specifications. Also, we use s = NULL as
223     * end-delimiter, because a type of 0 can be legal.
224     */
225	{ "ip",		0x0800 },
226	{ "ipv4",	0x0800 },
227	{ "ipv6",	0x86dd },
228	{ "arp",	0x0806 },
229	{ "rarp",	0x8035 },
230	{ "vlan",	0x8100 },
231	{ "loop",	0x9000 },
232	{ "trail",	0x1000 },
233	{ "at",		0x809b },
234	{ "atalk",	0x809b },
235	{ "aarp",	0x80f3 },
236	{ "pppoe_disc",	0x8863 },
237	{ "pppoe_sess",	0x8864 },
238	{ "ipx_8022",	0x00E0 },
239	{ "ipx_8023",	0x0000 },
240	{ "ipx_ii",	0x8137 },
241	{ "ipx_snap",	0x8137 },
242	{ "ipx",	0x8137 },
243	{ "ns",		0x0600 },
244	{ NULL,		0 }
245};
246
247static struct _s_x rule_eactions[] = {
248	{ "nat64clat",		TOK_NAT64CLAT },
249	{ "nat64lsn",		TOK_NAT64LSN },
250	{ "nat64stl",		TOK_NAT64STL },
251	{ "nptv6",		TOK_NPTV6 },
252	{ "tcp-setmss",		TOK_TCPSETMSS },
253	{ NULL, 0 }	/* terminator */
254};
255
256static struct _s_x rule_actions[] = {
257	{ "abort6",		TOK_ABORT6 },
258	{ "abort",		TOK_ABORT },
259	{ "accept",		TOK_ACCEPT },
260	{ "pass",		TOK_ACCEPT },
261	{ "allow",		TOK_ACCEPT },
262	{ "permit",		TOK_ACCEPT },
263	{ "count",		TOK_COUNT },
264	{ "pipe",		TOK_PIPE },
265	{ "queue",		TOK_QUEUE },
266	{ "divert",		TOK_DIVERT },
267	{ "tee",		TOK_TEE },
268	{ "netgraph",		TOK_NETGRAPH },
269	{ "ngtee",		TOK_NGTEE },
270	{ "fwd",		TOK_FORWARD },
271	{ "forward",		TOK_FORWARD },
272	{ "skipto",		TOK_SKIPTO },
273	{ "deny",		TOK_DENY },
274	{ "drop",		TOK_DENY },
275	{ "reject",		TOK_REJECT },
276	{ "reset6",		TOK_RESET6 },
277	{ "reset",		TOK_RESET },
278	{ "unreach6",		TOK_UNREACH6 },
279	{ "unreach",		TOK_UNREACH },
280	{ "check-state",	TOK_CHECKSTATE },
281	{ "//",			TOK_COMMENT },
282	{ "nat",		TOK_NAT },
283	{ "reass",		TOK_REASS },
284	{ "setfib",		TOK_SETFIB },
285	{ "setdscp",		TOK_SETDSCP },
286	{ "call",		TOK_CALL },
287	{ "return",		TOK_RETURN },
288	{ "eaction",		TOK_EACTION },
289	{ "tcp-setmss",		TOK_TCPSETMSS },
290	{ NULL, 0 }	/* terminator */
291};
292
293static struct _s_x rule_action_params[] = {
294	{ "altq",		TOK_ALTQ },
295	{ "log",		TOK_LOG },
296	{ "tag",		TOK_TAG },
297	{ "untag",		TOK_UNTAG },
298	{ NULL, 0 }	/* terminator */
299};
300
301/*
302 * The 'lookup' instruction accepts one of the following arguments.
303 * -1 is a terminator for the list.
304 * Arguments are passed as v[1] in O_DST_LOOKUP options.
305 */
306static int lookup_key[] = {
307	TOK_DSTIP, TOK_SRCIP, TOK_DSTPORT, TOK_SRCPORT,
308	TOK_UID, TOK_JAIL, TOK_DSCP, -1 };
309
310static struct _s_x rule_options[] = {
311	{ "tagged",		TOK_TAGGED },
312	{ "uid",		TOK_UID },
313	{ "gid",		TOK_GID },
314	{ "jail",		TOK_JAIL },
315	{ "in",			TOK_IN },
316	{ "limit",		TOK_LIMIT },
317	{ "set-limit",		TOK_SETLIMIT },
318	{ "keep-state",		TOK_KEEPSTATE },
319	{ "record-state",	TOK_RECORDSTATE },
320	{ "bridged",		TOK_LAYER2 },
321	{ "layer2",		TOK_LAYER2 },
322	{ "out",		TOK_OUT },
323	{ "diverted",		TOK_DIVERTED },
324	{ "diverted-loopback",	TOK_DIVERTEDLOOPBACK },
325	{ "diverted-output",	TOK_DIVERTEDOUTPUT },
326	{ "xmit",		TOK_XMIT },
327	{ "recv",		TOK_RECV },
328	{ "via",		TOK_VIA },
329	{ "fragment",		TOK_FRAG },
330	{ "frag",		TOK_FRAG },
331	{ "fib",		TOK_FIB },
332	{ "ipoptions",		TOK_IPOPTS },
333	{ "ipopts",		TOK_IPOPTS },
334	{ "iplen",		TOK_IPLEN },
335	{ "ipid",		TOK_IPID },
336	{ "ipprecedence",	TOK_IPPRECEDENCE },
337	{ "dscp",		TOK_DSCP },
338	{ "iptos",		TOK_IPTOS },
339	{ "ipttl",		TOK_IPTTL },
340	{ "ipversion",		TOK_IPVER },
341	{ "ipver",		TOK_IPVER },
342	{ "estab",		TOK_ESTAB },
343	{ "established",	TOK_ESTAB },
344	{ "setup",		TOK_SETUP },
345	{ "sockarg",		TOK_SOCKARG },
346	{ "tcpdatalen",		TOK_TCPDATALEN },
347	{ "tcpflags",		TOK_TCPFLAGS },
348	{ "tcpflgs",		TOK_TCPFLAGS },
349	{ "tcpmss",		TOK_TCPMSS },
350	{ "tcpoptions",		TOK_TCPOPTS },
351	{ "tcpopts",		TOK_TCPOPTS },
352	{ "tcpseq",		TOK_TCPSEQ },
353	{ "tcpack",		TOK_TCPACK },
354	{ "tcpwin",		TOK_TCPWIN },
355	{ "icmptype",		TOK_ICMPTYPES },
356	{ "icmptypes",		TOK_ICMPTYPES },
357	{ "dst-ip",		TOK_DSTIP },
358	{ "src-ip",		TOK_SRCIP },
359	{ "dst-port",		TOK_DSTPORT },
360	{ "src-port",		TOK_SRCPORT },
361	{ "proto",		TOK_PROTO },
362	{ "MAC",		TOK_MAC },
363	{ "mac",		TOK_MAC },
364	{ "mac-type",		TOK_MACTYPE },
365	{ "verrevpath",		TOK_VERREVPATH },
366	{ "versrcreach",	TOK_VERSRCREACH },
367	{ "antispoof",		TOK_ANTISPOOF },
368	{ "ipsec",		TOK_IPSEC },
369	{ "icmp6type",		TOK_ICMP6TYPES },
370	{ "icmp6types",		TOK_ICMP6TYPES },
371	{ "ext6hdr",		TOK_EXT6HDR},
372	{ "flow-id",		TOK_FLOWID},
373	{ "ipv6",		TOK_IPV6},
374	{ "ip6",		TOK_IPV6},
375	{ "ipv4",		TOK_IPV4},
376	{ "ip4",		TOK_IPV4},
377	{ "dst-ipv6",		TOK_DSTIP6},
378	{ "dst-ip6",		TOK_DSTIP6},
379	{ "src-ipv6",		TOK_SRCIP6},
380	{ "src-ip6",		TOK_SRCIP6},
381	{ "lookup",		TOK_LOOKUP},
382	{ "flow",		TOK_FLOW},
383	{ "defer-action",	TOK_SKIPACTION },
384	{ "defer-immediate-action",	TOK_SKIPACTION },
385	{ "//",			TOK_COMMENT },
386
387	{ "not",		TOK_NOT },		/* pseudo option */
388	{ "!", /* escape ? */	TOK_NOT },		/* pseudo option */
389	{ "or",			TOK_OR },		/* pseudo option */
390	{ "|", /* escape */	TOK_OR },		/* pseudo option */
391	{ "{",			TOK_STARTBRACE },	/* pseudo option */
392	{ "(",			TOK_STARTBRACE },	/* pseudo option */
393	{ "}",			TOK_ENDBRACE },		/* pseudo option */
394	{ ")",			TOK_ENDBRACE },		/* pseudo option */
395	{ NULL, 0 }	/* terminator */
396};
397
398void bprint_uint_arg(struct buf_pr *bp, const char *str, uint32_t arg);
399static int ipfw_get_config(struct cmdline_opts *co, struct format_opts *fo,
400    ipfw_cfg_lheader **pcfg, size_t *psize);
401static int ipfw_show_config(struct cmdline_opts *co, struct format_opts *fo,
402    ipfw_cfg_lheader *cfg, size_t sz, int ac, char **av);
403static void ipfw_list_tifaces(void);
404
405struct tidx;
406static uint16_t pack_object(struct tidx *tstate, const char *name, int otype);
407static uint16_t pack_table(struct tidx *tstate, const char *name);
408
409static char *table_search_ctlv(ipfw_obj_ctlv *ctlv, uint16_t idx);
410static void object_sort_ctlv(ipfw_obj_ctlv *ctlv);
411static char *object_search_ctlv(ipfw_obj_ctlv *ctlv, uint16_t idx,
412    uint16_t type);
413
414/*
415 * Simple string buffer API.
416 * Used to simplify buffer passing between function and for
417 * transparent overrun handling.
418 */
419
420/*
421 * Allocates new buffer of given size @sz.
422 *
423 * Returns 0 on success.
424 */
425int
426bp_alloc(struct buf_pr *b, size_t size)
427{
428	memset(b, 0, sizeof(struct buf_pr));
429
430	if ((b->buf = calloc(1, size)) == NULL)
431		return (ENOMEM);
432
433	b->ptr = b->buf;
434	b->size = size;
435	b->avail = b->size;
436
437	return (0);
438}
439
440void
441bp_free(struct buf_pr *b)
442{
443
444	free(b->buf);
445}
446
447/*
448 * Flushes buffer so new writer start from beginning.
449 */
450void
451bp_flush(struct buf_pr *b)
452{
453
454	b->ptr = b->buf;
455	b->avail = b->size;
456	b->buf[0] = '\0';
457}
458
459/*
460 * Print message specified by @format and args.
461 * Automatically manage buffer space and transparently handle
462 * buffer overruns.
463 *
464 * Returns number of bytes that should have been printed.
465 */
466int
467bprintf(struct buf_pr *b, const char *format, ...)
468{
469	va_list args;
470	int i;
471
472	va_start(args, format);
473
474	i = vsnprintf(b->ptr, b->avail, format, args);
475	va_end(args);
476
477	if (i < 0 || (size_t)i > b->avail) {
478		/* Overflow or print error */
479		b->avail = 0;
480	} else {
481		b->ptr += i;
482		b->avail -= i;
483	}
484
485	b->needed += i;
486
487	return (i);
488}
489
490/*
491 * Special values printer for tablearg-aware opcodes.
492 */
493void
494bprint_uint_arg(struct buf_pr *bp, const char *str, uint32_t arg)
495{
496
497	if (str != NULL)
498		bprintf(bp, "%s", str);
499	if (arg == IP_FW_TARG)
500		bprintf(bp, "tablearg");
501	else
502		bprintf(bp, "%u", arg);
503}
504
505/*
506 * Helper routine to print a possibly unaligned uint64_t on
507 * various platform. If width > 0, print the value with
508 * the desired width, followed by a space;
509 * otherwise, return the required width.
510 */
511int
512pr_u64(struct buf_pr *b, void *pd, int width)
513{
514#ifdef TCC
515#define U64_FMT "I64"
516#else
517#define U64_FMT "llu"
518#endif
519	uint64_t u;
520	unsigned long long d;
521
522	bcopy (pd, &u, sizeof(u));
523	d = u;
524	return (width > 0) ?
525		bprintf(b, "%*" U64_FMT " ", width, d) :
526		snprintf(NULL, 0, "%" U64_FMT, d) ;
527#undef U64_FMT
528}
529
530
531void *
532safe_calloc(size_t number, size_t size)
533{
534	void *ret = calloc(number, size);
535
536	if (ret == NULL)
537		err(EX_OSERR, "calloc");
538	return ret;
539}
540
541void *
542safe_realloc(void *ptr, size_t size)
543{
544	void *ret = realloc(ptr, size);
545
546	if (ret == NULL)
547		err(EX_OSERR, "realloc");
548	return ret;
549}
550
551/*
552 * Compare things like interface or table names.
553 */
554int
555stringnum_cmp(const char *a, const char *b)
556{
557	int la, lb;
558
559	la = strlen(a);
560	lb = strlen(b);
561
562	if (la > lb)
563		return (1);
564	else if (la < lb)
565		return (-01);
566
567	return (strcmp(a, b));
568}
569
570
571/*
572 * conditionally runs the command.
573 * Selected options or negative -> getsockopt
574 */
575int
576do_cmd(int optname, void *optval, uintptr_t optlen)
577{
578	int i;
579
580	if (g_co.test_only)
581		return 0;
582
583	if (ipfw_socket == -1)
584		ipfw_socket = socket(AF_INET, SOCK_RAW, IPPROTO_RAW);
585	if (ipfw_socket < 0)
586		err(EX_UNAVAILABLE, "socket");
587
588	if (optname == IP_FW_GET || optname == IP_DUMMYNET_GET ||
589	    optname == IP_FW_ADD || optname == IP_FW3 ||
590	    optname == IP_FW_NAT_GET_CONFIG ||
591	    optname < 0 ||
592	    optname == IP_FW_NAT_GET_LOG) {
593		if (optname < 0)
594			optname = -optname;
595		i = getsockopt(ipfw_socket, IPPROTO_IP, optname, optval,
596			(socklen_t *)optlen);
597	} else {
598		i = setsockopt(ipfw_socket, IPPROTO_IP, optname, optval, optlen);
599	}
600	return i;
601}
602
603/*
604 * do_set3 - pass ipfw control cmd to kernel
605 * @optname: option name
606 * @optval: pointer to option data
607 * @optlen: option length
608 *
609 * Assumes op3 header is already embedded.
610 * Calls setsockopt() with IP_FW3 as kernel-visible opcode.
611 * Returns 0 on success or errno otherwise.
612 */
613int
614do_set3(int optname, ip_fw3_opheader *op3, size_t optlen)
615{
616
617	if (g_co.test_only)
618		return (0);
619
620	if (ipfw_socket == -1)
621		ipfw_socket = socket(AF_INET, SOCK_RAW, IPPROTO_RAW);
622	if (ipfw_socket < 0)
623		err(EX_UNAVAILABLE, "socket");
624
625	op3->opcode = optname;
626
627	return (setsockopt(ipfw_socket, IPPROTO_IP, IP_FW3, op3, optlen));
628}
629
630/*
631 * do_get3 - pass ipfw control cmd to kernel
632 * @optname: option name
633 * @optval: pointer to option data
634 * @optlen: pointer to option length
635 *
636 * Assumes op3 header is already embedded.
637 * Calls getsockopt() with IP_FW3 as kernel-visible opcode.
638 * Returns 0 on success or errno otherwise.
639 */
640int
641do_get3(int optname, ip_fw3_opheader *op3, size_t *optlen)
642{
643	int error;
644	socklen_t len;
645
646	if (g_co.test_only)
647		return (0);
648
649	if (ipfw_socket == -1)
650		ipfw_socket = socket(AF_INET, SOCK_RAW, IPPROTO_RAW);
651	if (ipfw_socket < 0)
652		err(EX_UNAVAILABLE, "socket");
653
654	op3->opcode = optname;
655
656	len = *optlen;
657	error = getsockopt(ipfw_socket, IPPROTO_IP, IP_FW3, op3, &len);
658	*optlen = len;
659
660	return (error);
661}
662
663/**
664 * match_token takes a table and a string, returns the value associated
665 * with the string (-1 in case of failure).
666 */
667int
668match_token(struct _s_x *table, const char *string)
669{
670	struct _s_x *pt;
671	uint i = strlen(string);
672
673	for (pt = table ; i && pt->s != NULL ; pt++)
674		if (strlen(pt->s) == i && !bcmp(string, pt->s, i))
675			return pt->x;
676	return (-1);
677}
678
679/**
680 * match_token_relaxed takes a table and a string, returns the value associated
681 * with the string for the best match.
682 *
683 * Returns:
684 * value from @table for matched records
685 * -1 for non-matched records
686 * -2 if more than one records match @string.
687 */
688int
689match_token_relaxed(struct _s_x *table, const char *string)
690{
691	struct _s_x *pt, *m;
692	int i, c;
693
694	i = strlen(string);
695	c = 0;
696
697	for (pt = table ; i != 0 && pt->s != NULL ; pt++) {
698		if (strncmp(pt->s, string, i) != 0)
699			continue;
700		m = pt;
701		c++;
702	}
703
704	if (c == 1)
705		return (m->x);
706
707	return (c > 0 ? -2: -1);
708}
709
710int
711get_token(struct _s_x *table, const char *string, const char *errbase)
712{
713	int tcmd;
714
715	if ((tcmd = match_token_relaxed(table, string)) < 0)
716		errx(EX_USAGE, "%s %s %s",
717		    (tcmd == 0) ? "invalid" : "ambiguous", errbase, string);
718
719	return (tcmd);
720}
721
722/**
723 * match_value takes a table and a value, returns the string associated
724 * with the value (NULL in case of failure).
725 */
726char const *
727match_value(struct _s_x *p, int value)
728{
729	for (; p->s != NULL; p++)
730		if (p->x == value)
731			return p->s;
732	return NULL;
733}
734
735size_t
736concat_tokens(char *buf, size_t bufsize, struct _s_x *table,
737    const char *delimiter)
738{
739	struct _s_x *pt;
740	int l;
741	size_t sz;
742
743	for (sz = 0, pt = table ; pt->s != NULL; pt++) {
744		l = snprintf(buf + sz, bufsize - sz, "%s%s",
745		    (sz == 0) ? "" : delimiter, pt->s);
746		sz += l;
747		bufsize += l;
748		if (sz > bufsize)
749			return (bufsize);
750	}
751
752	return (sz);
753}
754
755/*
756 * helper function to process a set of flags and set bits in the
757 * appropriate masks.
758 */
759int
760fill_flags(struct _s_x *flags, char *p, char **e, uint32_t *set,
761    uint32_t *clear)
762{
763	char *q;	/* points to the separator */
764	int val;
765	uint32_t *which;	/* mask we are working on */
766
767	while (p && *p) {
768		if (*p == '!') {
769			p++;
770			which = clear;
771		} else
772			which = set;
773		q = strchr(p, ',');
774		if (q)
775			*q++ = '\0';
776		val = match_token(flags, p);
777		if (val <= 0) {
778			if (e != NULL)
779				*e = p;
780			return (-1);
781		}
782		*which |= (uint32_t)val;
783		p = q;
784	}
785	return (0);
786}
787
788void
789print_flags_buffer(char *buf, size_t sz, struct _s_x *list, uint32_t set)
790{
791	char const *comma = "";
792	int i, l;
793
794	for (i = 0; list[i].x != 0; i++) {
795		if ((set & list[i].x) == 0)
796			continue;
797
798		set &= ~list[i].x;
799		l = snprintf(buf, sz, "%s%s", comma, list[i].s);
800		if (l < 0 || (size_t)l >= sz)
801			return;
802		comma = ",";
803		buf += l;
804		sz -=l;
805	}
806}
807
808/*
809 * _substrcmp takes two strings and returns 1 if they do not match,
810 * and 0 if they match exactly or the first string is a sub-string
811 * of the second.  A warning is printed to stderr in the case that the
812 * first string is a sub-string of the second.
813 *
814 * This function will be removed in the future through the usual
815 * deprecation process.
816 */
817int
818_substrcmp(const char *str1, const char* str2)
819{
820
821	if (strncmp(str1, str2, strlen(str1)) != 0)
822		return 1;
823
824	if (strlen(str1) != strlen(str2))
825		warnx("DEPRECATED: '%s' matched '%s' as a sub-string",
826		    str1, str2);
827	return 0;
828}
829
830/*
831 * _substrcmp2 takes three strings and returns 1 if the first two do not match,
832 * and 0 if they match exactly or the second string is a sub-string
833 * of the first.  A warning is printed to stderr in the case that the
834 * first string does not match the third.
835 *
836 * This function exists to warn about the bizarre construction
837 * strncmp(str, "by", 2) which is used to allow people to use a shortcut
838 * for "bytes".  The problem is that in addition to accepting "by",
839 * "byt", "byte", and "bytes", it also excepts "by_rabid_dogs" and any
840 * other string beginning with "by".
841 *
842 * This function will be removed in the future through the usual
843 * deprecation process.
844 */
845int
846_substrcmp2(const char *str1, const char* str2, const char* str3)
847{
848
849	if (strncmp(str1, str2, strlen(str2)) != 0)
850		return 1;
851
852	if (strcmp(str1, str3) != 0)
853		warnx("DEPRECATED: '%s' matched '%s'",
854		    str1, str3);
855	return 0;
856}
857
858/*
859 * prints one port, symbolic or numeric
860 */
861static void
862print_port(struct buf_pr *bp, int proto, uint16_t port)
863{
864
865	if (proto == IPPROTO_ETHERTYPE) {
866		char const *s;
867
868		if (g_co.do_resolv && (s = match_value(ether_types, port)) )
869			bprintf(bp, "%s", s);
870		else
871			bprintf(bp, "0x%04x", port);
872	} else {
873		struct servent *se = NULL;
874		if (g_co.do_resolv) {
875			struct protoent *pe = getprotobynumber(proto);
876
877			se = getservbyport(htons(port), pe ? pe->p_name : NULL);
878		}
879		if (se)
880			bprintf(bp, "%s", se->s_name);
881		else
882			bprintf(bp, "%d", port);
883	}
884}
885
886static struct _s_x _port_name[] = {
887	{"dst-port",	O_IP_DSTPORT},
888	{"src-port",	O_IP_SRCPORT},
889	{"ipid",	O_IPID},
890	{"iplen",	O_IPLEN},
891	{"ipttl",	O_IPTTL},
892	{"mac-type",	O_MAC_TYPE},
893	{"tcpdatalen",	O_TCPDATALEN},
894	{"tcpmss",	O_TCPMSS},
895	{"tcpwin",	O_TCPWIN},
896	{"tagged",	O_TAGGED},
897	{NULL,		0}
898};
899
900/*
901 * Print the values in a list 16-bit items of the types above.
902 * XXX todo: add support for mask.
903 */
904static void
905print_newports(struct buf_pr *bp, const ipfw_insn_u16 *cmd, int proto, int opcode)
906{
907	const uint16_t *p = cmd->ports;
908	int i;
909	char const *sep;
910
911	if (opcode != 0) {
912		sep = match_value(_port_name, opcode);
913		if (sep == NULL)
914			sep = "???";
915		bprintf(bp, " %s", sep);
916	}
917	sep = " ";
918	for (i = F_LEN((const ipfw_insn *)cmd) - 1; i > 0; i--, p += 2) {
919		bprintf(bp, "%s", sep);
920		print_port(bp, proto, p[0]);
921		if (p[0] != p[1]) {
922			bprintf(bp, "-");
923			print_port(bp, proto, p[1]);
924		}
925		sep = ",";
926	}
927}
928
929/*
930 * Like strtol, but also translates service names into port numbers
931 * for some protocols.
932 * In particular:
933 *	proto == -1 disables the protocol check;
934 *	proto == IPPROTO_ETHERTYPE looks up an internal table
935 *	proto == <some value in /etc/protocols> matches the values there.
936 * Returns *end == s in case the parameter is not found.
937 */
938static int
939strtoport(char *s, char **end, int base, int proto)
940{
941	char *p, *buf;
942	char *s1;
943	int i;
944
945	*end = s;		/* default - not found */
946	if (*s == '\0')
947		return 0;	/* not found */
948
949	if (isdigit(*s))
950		return strtol(s, end, base);
951
952	/*
953	 * find separator. '\\' escapes the next char.
954	 */
955	for (s1 = s; *s1 && (isalnum(*s1) || *s1 == '\\' ||
956	    *s1 == '_' || *s1 == '.') ; s1++)
957		if (*s1 == '\\' && s1[1] != '\0')
958			s1++;
959
960	buf = safe_calloc(s1 - s + 1, 1);
961
962	/*
963	 * copy into a buffer skipping backslashes
964	 */
965	for (p = s, i = 0; p != s1 ; p++)
966		if (*p != '\\')
967			buf[i++] = *p;
968	buf[i++] = '\0';
969
970	if (proto == IPPROTO_ETHERTYPE) {
971		i = match_token(ether_types, buf);
972		free(buf);
973		if (i != -1) {	/* found */
974			*end = s1;
975			return i;
976		}
977	} else {
978		struct protoent *pe = NULL;
979		struct servent *se;
980
981		if (proto != 0)
982			pe = getprotobynumber(proto);
983		setservent(1);
984		se = getservbyname(buf, pe ? pe->p_name : NULL);
985		free(buf);
986		if (se != NULL) {
987			*end = s1;
988			return ntohs(se->s_port);
989		}
990	}
991	return 0;	/* not found */
992}
993
994/*
995 * Fill the body of the command with the list of port ranges.
996 */
997static int
998fill_newports(ipfw_insn_u16 *cmd, char *av, int proto, int cblen)
999{
1000	uint16_t a, b, *p = cmd->ports;
1001	int i = 0;
1002	char *s = av;
1003
1004	while (*s) {
1005		a = strtoport(av, &s, 0, proto);
1006		if (s == av) 			/* empty or invalid argument */
1007			return (0);
1008
1009		CHECK_LENGTH(cblen, i + 2);
1010
1011		switch (*s) {
1012		case '-':			/* a range */
1013			av = s + 1;
1014			b = strtoport(av, &s, 0, proto);
1015			/* Reject expressions like '1-abc' or '1-2-3'. */
1016			if (s == av || (*s != ',' && *s != '\0'))
1017				return (0);
1018			p[0] = a;
1019			p[1] = b;
1020			break;
1021		case ',':			/* comma separated list */
1022		case '\0':
1023			p[0] = p[1] = a;
1024			break;
1025		default:
1026			warnx("port list: invalid separator <%c> in <%s>",
1027				*s, av);
1028			return (0);
1029		}
1030
1031		i++;
1032		p += 2;
1033		av = s + 1;
1034	}
1035	if (i > 0) {
1036		if (i + 1 > F_LEN_MASK)
1037			errx(EX_DATAERR, "too many ports/ranges\n");
1038		cmd->o.len |= i + 1;	/* leave F_NOT and F_OR untouched */
1039	}
1040	return (i);
1041}
1042
1043/*
1044 * Fill the body of the command with the list of DiffServ codepoints.
1045 */
1046static void
1047fill_dscp(ipfw_insn *cmd, char *av, int cblen)
1048{
1049	uint32_t *low, *high;
1050	char *s = av, *a;
1051	int code;
1052
1053	cmd->opcode = O_DSCP;
1054	cmd->len |= F_INSN_SIZE(ipfw_insn_u32) + 1;
1055
1056	CHECK_CMDLEN;
1057
1058	low = (uint32_t *)(cmd + 1);
1059	high = low + 1;
1060
1061	*low = 0;
1062	*high = 0;
1063
1064	while (s != NULL) {
1065		a = strchr(s, ',');
1066
1067		if (a != NULL)
1068			*a++ = '\0';
1069
1070		if (isalpha(*s)) {
1071			if ((code = match_token(f_ipdscp, s)) == -1)
1072				errx(EX_DATAERR, "Unknown DSCP code");
1073		} else {
1074			code = strtoul(s, NULL, 10);
1075			if (code < 0 || code > 63)
1076				errx(EX_DATAERR, "Invalid DSCP value");
1077		}
1078
1079		if (code >= 32)
1080			*high |= 1 << (code - 32);
1081		else
1082			*low |= 1 << code;
1083
1084		s = a;
1085	}
1086}
1087
1088static struct _s_x icmpcodes[] = {
1089      { "net",			ICMP_UNREACH_NET },
1090      { "host",			ICMP_UNREACH_HOST },
1091      { "protocol",		ICMP_UNREACH_PROTOCOL },
1092      { "port",			ICMP_UNREACH_PORT },
1093      { "needfrag",		ICMP_UNREACH_NEEDFRAG },
1094      { "srcfail",		ICMP_UNREACH_SRCFAIL },
1095      { "net-unknown",		ICMP_UNREACH_NET_UNKNOWN },
1096      { "host-unknown",		ICMP_UNREACH_HOST_UNKNOWN },
1097      { "isolated",		ICMP_UNREACH_ISOLATED },
1098      { "net-prohib",		ICMP_UNREACH_NET_PROHIB },
1099      { "host-prohib",		ICMP_UNREACH_HOST_PROHIB },
1100      { "tosnet",		ICMP_UNREACH_TOSNET },
1101      { "toshost",		ICMP_UNREACH_TOSHOST },
1102      { "filter-prohib",	ICMP_UNREACH_FILTER_PROHIB },
1103      { "host-precedence",	ICMP_UNREACH_HOST_PRECEDENCE },
1104      { "precedence-cutoff",	ICMP_UNREACH_PRECEDENCE_CUTOFF },
1105      { NULL, 0 }
1106};
1107
1108static void
1109fill_reject_code(u_short *codep, char *str)
1110{
1111	int val;
1112	char *s;
1113
1114	val = strtoul(str, &s, 0);
1115	if (s == str || *s != '\0' || val >= 0x100)
1116		val = match_token(icmpcodes, str);
1117	if (val < 0)
1118		errx(EX_DATAERR, "unknown ICMP unreachable code ``%s''", str);
1119	*codep = val;
1120	return;
1121}
1122
1123static void
1124print_reject_code(struct buf_pr *bp, uint16_t code)
1125{
1126	char const *s;
1127
1128	if ((s = match_value(icmpcodes, code)) != NULL)
1129		bprintf(bp, "unreach %s", s);
1130	else
1131		bprintf(bp, "unreach %u", code);
1132}
1133
1134/*
1135 * Returns the number of bits set (from left) in a contiguous bitmask,
1136 * or -1 if the mask is not contiguous.
1137 * XXX this needs a proper fix.
1138 * This effectively works on masks in big-endian (network) format.
1139 * when compiled on little endian architectures.
1140 *
1141 * First bit is bit 7 of the first byte -- note, for MAC addresses,
1142 * the first bit on the wire is bit 0 of the first byte.
1143 * len is the max length in bits.
1144 */
1145int
1146contigmask(const uint8_t *p, int len)
1147{
1148	int i, n;
1149
1150	for (i=0; i<len ; i++)
1151		if ( (p[i/8] & (1 << (7 - (i%8)))) == 0) /* first bit unset */
1152			break;
1153	for (n=i+1; n < len; n++)
1154		if ( (p[n/8] & (1 << (7 - (n%8)))) != 0)
1155			return -1; /* mask not contiguous */
1156	return i;
1157}
1158
1159/*
1160 * print flags set/clear in the two bitmasks passed as parameters.
1161 * There is a specialized check for f_tcpflags.
1162 */
1163static void
1164print_flags(struct buf_pr *bp, char const *name, const ipfw_insn *cmd,
1165    struct _s_x *list)
1166{
1167	char const *comma = "";
1168	int i;
1169	uint8_t set = cmd->arg1 & 0xff;
1170	uint8_t clear = (cmd->arg1 >> 8) & 0xff;
1171
1172	if (list == f_tcpflags && set == TH_SYN && clear == TH_ACK) {
1173		bprintf(bp, " setup");
1174		return;
1175	}
1176
1177	bprintf(bp, " %s ", name);
1178	for (i=0; list[i].x != 0; i++) {
1179		if (set & list[i].x) {
1180			set &= ~list[i].x;
1181			bprintf(bp, "%s%s", comma, list[i].s);
1182			comma = ",";
1183		}
1184		if (clear & list[i].x) {
1185			clear &= ~list[i].x;
1186			bprintf(bp, "%s!%s", comma, list[i].s);
1187			comma = ",";
1188		}
1189	}
1190}
1191
1192
1193/*
1194 * Print the ip address contained in a command.
1195 */
1196static void
1197print_ip(struct buf_pr *bp, const struct format_opts *fo,
1198    const ipfw_insn_ip *cmd)
1199{
1200	struct hostent *he = NULL;
1201	const struct in_addr *ia;
1202	const uint32_t *a = ((const ipfw_insn_u32 *)cmd)->d;
1203	uint32_t len = F_LEN((const ipfw_insn *)cmd);
1204	char *t;
1205
1206	bprintf(bp, " ");
1207	if (cmd->o.opcode == O_IP_DST_LOOKUP && len > F_INSN_SIZE(ipfw_insn_u32)) {
1208		uint32_t d = a[1];
1209		const char *arg = "<invalid>";
1210
1211		if (d < sizeof(lookup_key)/sizeof(lookup_key[0]))
1212			arg = match_value(rule_options, lookup_key[d]);
1213		t = table_search_ctlv(fo->tstate,
1214		    ((const ipfw_insn *)cmd)->arg1);
1215		bprintf(bp, "lookup %s %s", arg, t);
1216		return;
1217	}
1218	if (cmd->o.opcode == O_IP_SRC_ME || cmd->o.opcode == O_IP_DST_ME) {
1219		bprintf(bp, "me");
1220		return;
1221	}
1222	if (cmd->o.opcode == O_IP_SRC_LOOKUP ||
1223	    cmd->o.opcode == O_IP_DST_LOOKUP) {
1224		t = table_search_ctlv(fo->tstate,
1225		    ((const ipfw_insn *)cmd)->arg1);
1226		bprintf(bp, "table(%s", t);
1227		if (len == F_INSN_SIZE(ipfw_insn_u32))
1228			bprintf(bp, ",%u", *a);
1229		bprintf(bp, ")");
1230		return;
1231	}
1232	if (cmd->o.opcode == O_IP_SRC_SET || cmd->o.opcode == O_IP_DST_SET) {
1233		const uint32_t *map = (const uint32_t *)&cmd->mask;
1234		struct in_addr addr;
1235		uint32_t x;
1236		int i, j;
1237		char comma = '{';
1238
1239		x = cmd->o.arg1 - 1;
1240		x = htonl(~x);
1241		addr.s_addr = htonl(cmd->addr.s_addr);
1242		bprintf(bp, "%s/%d", inet_ntoa(addr),
1243		    contigmask((uint8_t *)&x, 32));
1244		x = cmd->addr.s_addr;
1245		x &= 0xff; /* base */
1246		/*
1247		 * Print bits and ranges.
1248		 * Locate first bit set (i), then locate first bit unset (j).
1249		 * If we have 3+ consecutive bits set, then print them as a
1250		 * range, otherwise only print the initial bit and rescan.
1251		 */
1252		for (i=0; i < cmd->o.arg1; i++)
1253			if (map[i/32] & (1<<(i & 31))) {
1254				for (j=i+1; j < cmd->o.arg1; j++)
1255					if (!(map[ j/32] & (1<<(j & 31))))
1256						break;
1257				bprintf(bp, "%c%d", comma, i+x);
1258				if (j>i+2) { /* range has at least 3 elements */
1259					bprintf(bp, "-%d", j-1+x);
1260					i = j-1;
1261				}
1262				comma = ',';
1263			}
1264		bprintf(bp, "}");
1265		return;
1266	}
1267	/*
1268	 * len == 2 indicates a single IP, whereas lists of 1 or more
1269	 * addr/mask pairs have len = (2n+1). We convert len to n so we
1270	 * use that to count the number of entries.
1271	 */
1272    for (len = len / 2; len > 0; len--, a += 2) {
1273	int mb =	/* mask length */
1274	    (cmd->o.opcode == O_IP_SRC || cmd->o.opcode == O_IP_DST) ?
1275		32 : contigmask((const uint8_t *)&(a[1]), 32);
1276	if (mb == 32 && g_co.do_resolv)
1277		he = gethostbyaddr((const char *)&(a[0]), sizeof(in_addr_t),
1278		    AF_INET);
1279	if (he != NULL)		/* resolved to name */
1280		bprintf(bp, "%s", he->h_name);
1281	else if (mb == 0)	/* any */
1282		bprintf(bp, "any");
1283	else {		/* numeric IP followed by some kind of mask */
1284		ia = (const struct in_addr *)&a[0];
1285		bprintf(bp, "%s", inet_ntoa(*ia));
1286		if (mb < 0) {
1287			ia = (const struct in_addr *)&a[1];
1288			bprintf(bp, ":%s", inet_ntoa(*ia));
1289		} else if (mb < 32)
1290			bprintf(bp, "/%d", mb);
1291	}
1292	if (len > 1)
1293		bprintf(bp, ",");
1294    }
1295}
1296
1297/*
1298 * prints a MAC address/mask pair
1299 */
1300static void
1301format_mac(struct buf_pr *bp, const uint8_t *addr, const uint8_t *mask)
1302{
1303	int l = contigmask(mask, 48);
1304
1305	if (l == 0)
1306		bprintf(bp, " any");
1307	else {
1308		bprintf(bp, " %02x:%02x:%02x:%02x:%02x:%02x",
1309		    addr[0], addr[1], addr[2], addr[3], addr[4], addr[5]);
1310		if (l == -1)
1311			bprintf(bp, "&%02x:%02x:%02x:%02x:%02x:%02x",
1312			    mask[0], mask[1], mask[2],
1313			    mask[3], mask[4], mask[5]);
1314		else if (l < 48)
1315			bprintf(bp, "/%d", l);
1316	}
1317}
1318
1319static void
1320print_mac(struct buf_pr *bp, const ipfw_insn_mac *mac)
1321{
1322
1323	bprintf(bp, " MAC");
1324	format_mac(bp, mac->addr, mac->mask);
1325	format_mac(bp, mac->addr + 6, mac->mask + 6);
1326}
1327
1328static void
1329fill_icmptypes(ipfw_insn_u32 *cmd, char *av)
1330{
1331	uint8_t type;
1332
1333	cmd->d[0] = 0;
1334	while (*av) {
1335		if (*av == ',')
1336			av++;
1337
1338		type = strtoul(av, &av, 0);
1339
1340		if (*av != ',' && *av != '\0')
1341			errx(EX_DATAERR, "invalid ICMP type");
1342
1343		if (type > 31)
1344			errx(EX_DATAERR, "ICMP type out of range");
1345
1346		cmd->d[0] |= 1 << type;
1347	}
1348	cmd->o.opcode = O_ICMPTYPE;
1349	cmd->o.len |= F_INSN_SIZE(ipfw_insn_u32);
1350}
1351
1352static void
1353print_icmptypes(struct buf_pr *bp, const ipfw_insn_u32 *cmd)
1354{
1355	int i;
1356	char sep= ' ';
1357
1358	bprintf(bp, " icmptypes");
1359	for (i = 0; i < 32; i++) {
1360		if ( (cmd->d[0] & (1 << (i))) == 0)
1361			continue;
1362		bprintf(bp, "%c%d", sep, i);
1363		sep = ',';
1364	}
1365}
1366
1367static void
1368print_dscp(struct buf_pr *bp, const ipfw_insn_u32 *cmd)
1369{
1370	const uint32_t *v;
1371	const char *code;
1372	int i = 0;
1373	char sep= ' ';
1374
1375	bprintf(bp, " dscp");
1376	v = cmd->d;
1377	while (i < 64) {
1378		if (*v & (1 << i)) {
1379			if ((code = match_value(f_ipdscp, i)) != NULL)
1380				bprintf(bp, "%c%s", sep, code);
1381			else
1382				bprintf(bp, "%c%d", sep, i);
1383			sep = ',';
1384		}
1385
1386		if ((++i % 32) == 0)
1387			v++;
1388	}
1389}
1390
1391#define	insntod(cmd, type)	((const ipfw_insn_ ## type *)(cmd))
1392struct show_state {
1393	struct ip_fw_rule	*rule;
1394	const ipfw_insn		*eaction;
1395	uint8_t			*printed;
1396	int			flags;
1397#define	HAVE_PROTO		0x0001
1398#define	HAVE_SRCIP		0x0002
1399#define	HAVE_DSTIP		0x0004
1400#define	HAVE_PROBE_STATE	0x0008
1401	int			proto;
1402	int			or_block;
1403};
1404
1405static int
1406init_show_state(struct show_state *state, struct ip_fw_rule *rule)
1407{
1408
1409	state->printed = calloc(rule->cmd_len, sizeof(uint8_t));
1410	if (state->printed == NULL)
1411		return (ENOMEM);
1412	state->rule = rule;
1413	state->eaction = NULL;
1414	state->flags = 0;
1415	state->proto = 0;
1416	state->or_block = 0;
1417	return (0);
1418}
1419
1420static void
1421free_show_state(struct show_state *state)
1422{
1423
1424	free(state->printed);
1425}
1426
1427static uint8_t
1428is_printed_opcode(struct show_state *state, const ipfw_insn *cmd)
1429{
1430
1431	return (state->printed[cmd - state->rule->cmd]);
1432}
1433
1434static void
1435mark_printed(struct show_state *state, const ipfw_insn *cmd)
1436{
1437
1438	state->printed[cmd - state->rule->cmd] = 1;
1439}
1440
1441static void
1442print_limit_mask(struct buf_pr *bp, const ipfw_insn_limit *limit)
1443{
1444	struct _s_x *p = limit_masks;
1445	char const *comma = " ";
1446	uint8_t x;
1447
1448	for (x = limit->limit_mask; p->x != 0; p++) {
1449		if ((x & p->x) == p->x) {
1450			x &= ~p->x;
1451			bprintf(bp, "%s%s", comma, p->s);
1452			comma = ",";
1453		}
1454	}
1455	bprint_uint_arg(bp, " ", limit->conn_limit);
1456}
1457
1458static int
1459print_instruction(struct buf_pr *bp, const struct format_opts *fo,
1460    struct show_state *state, const ipfw_insn *cmd)
1461{
1462	struct protoent *pe;
1463	struct passwd *pwd;
1464	struct group *grp;
1465	const char *s;
1466	double d;
1467
1468	if (is_printed_opcode(state, cmd))
1469		return (0);
1470	if ((cmd->len & F_OR) != 0 && state->or_block == 0)
1471		bprintf(bp, " {");
1472	if (cmd->opcode != O_IN && (cmd->len & F_NOT) != 0)
1473		bprintf(bp, " not");
1474
1475	switch (cmd->opcode) {
1476	case O_PROB:
1477		d = 1.0 * insntod(cmd, u32)->d[0] / 0x7fffffff;
1478		bprintf(bp, "prob %f ", d);
1479		break;
1480	case O_PROBE_STATE: /* no need to print anything here */
1481		state->flags |= HAVE_PROBE_STATE;
1482		break;
1483	case O_IP_SRC:
1484	case O_IP_SRC_LOOKUP:
1485	case O_IP_SRC_MASK:
1486	case O_IP_SRC_ME:
1487	case O_IP_SRC_SET:
1488		if (state->flags & HAVE_SRCIP)
1489			bprintf(bp, " src-ip");
1490		print_ip(bp, fo, insntod(cmd, ip));
1491		break;
1492	case O_IP_DST:
1493	case O_IP_DST_LOOKUP:
1494	case O_IP_DST_MASK:
1495	case O_IP_DST_ME:
1496	case O_IP_DST_SET:
1497		if (state->flags & HAVE_DSTIP)
1498			bprintf(bp, " dst-ip");
1499		print_ip(bp, fo, insntod(cmd, ip));
1500		break;
1501	case O_IP6_SRC:
1502	case O_IP6_SRC_MASK:
1503	case O_IP6_SRC_ME:
1504		if (state->flags & HAVE_SRCIP)
1505			bprintf(bp, " src-ip6");
1506		print_ip6(bp, insntod(cmd, ip6));
1507		break;
1508	case O_IP6_DST:
1509	case O_IP6_DST_MASK:
1510	case O_IP6_DST_ME:
1511		if (state->flags & HAVE_DSTIP)
1512			bprintf(bp, " dst-ip6");
1513		print_ip6(bp, insntod(cmd, ip6));
1514		break;
1515	case O_FLOW6ID:
1516		print_flow6id(bp, insntod(cmd, u32));
1517		break;
1518	case O_IP_DSTPORT:
1519	case O_IP_SRCPORT:
1520		print_newports(bp, insntod(cmd, u16), state->proto,
1521		    (state->flags & (HAVE_SRCIP | HAVE_DSTIP)) ==
1522		    (HAVE_SRCIP | HAVE_DSTIP) ?  cmd->opcode: 0);
1523		break;
1524	case O_PROTO:
1525		pe = getprotobynumber(cmd->arg1);
1526		if (state->flags & HAVE_PROTO)
1527			bprintf(bp, " proto");
1528		if (pe != NULL)
1529			bprintf(bp, " %s", pe->p_name);
1530		else
1531			bprintf(bp, " %u", cmd->arg1);
1532		state->proto = cmd->arg1;
1533		break;
1534	case O_MACADDR2:
1535		print_mac(bp, insntod(cmd, mac));
1536		break;
1537	case O_MAC_TYPE:
1538		print_newports(bp, insntod(cmd, u16),
1539		    IPPROTO_ETHERTYPE, cmd->opcode);
1540		break;
1541	case O_FRAG:
1542		print_flags(bp, "frag", cmd, f_ipoff);
1543		break;
1544	case O_FIB:
1545		bprintf(bp, " fib %u", cmd->arg1);
1546		break;
1547	case O_SOCKARG:
1548		bprintf(bp, " sockarg");
1549		break;
1550	case O_IN:
1551		bprintf(bp, cmd->len & F_NOT ? " out" : " in");
1552		break;
1553	case O_DIVERTED:
1554		switch (cmd->arg1) {
1555		case 3:
1556			bprintf(bp, " diverted");
1557			break;
1558		case 2:
1559			bprintf(bp, " diverted-output");
1560			break;
1561		case 1:
1562			bprintf(bp, " diverted-loopback");
1563			break;
1564		default:
1565			bprintf(bp, " diverted-?<%u>", cmd->arg1);
1566			break;
1567		}
1568		break;
1569	case O_LAYER2:
1570		bprintf(bp, " layer2");
1571		break;
1572	case O_XMIT:
1573	case O_RECV:
1574	case O_VIA:
1575		if (cmd->opcode == O_XMIT)
1576			s = "xmit";
1577		else if (cmd->opcode == O_RECV)
1578			s = "recv";
1579		else /* if (cmd->opcode == O_VIA) */
1580			s = "via";
1581		switch (insntod(cmd, if)->name[0]) {
1582		case '\0':
1583			bprintf(bp, " %s %s", s,
1584			    inet_ntoa(insntod(cmd, if)->p.ip));
1585			break;
1586		case '\1':
1587			bprintf(bp, " %s table(%s)", s,
1588			    table_search_ctlv(fo->tstate,
1589			    insntod(cmd, if)->p.kidx));
1590			break;
1591		default:
1592			bprintf(bp, " %s %s", s,
1593			    insntod(cmd, if)->name);
1594		}
1595		break;
1596	case O_IP_FLOW_LOOKUP:
1597		s = table_search_ctlv(fo->tstate, cmd->arg1);
1598		bprintf(bp, " flow table(%s", s);
1599		if (F_LEN(cmd) == F_INSN_SIZE(ipfw_insn_u32))
1600			bprintf(bp, ",%u", insntod(cmd, u32)->d[0]);
1601		bprintf(bp, ")");
1602		break;
1603	case O_IPID:
1604	case O_IPTTL:
1605	case O_IPLEN:
1606	case O_TCPDATALEN:
1607	case O_TCPMSS:
1608	case O_TCPWIN:
1609		if (F_LEN(cmd) == 1) {
1610			switch (cmd->opcode) {
1611			case O_IPID:
1612				s = "ipid";
1613				break;
1614			case O_IPTTL:
1615				s = "ipttl";
1616				break;
1617			case O_IPLEN:
1618				s = "iplen";
1619				break;
1620			case O_TCPDATALEN:
1621				s = "tcpdatalen";
1622				break;
1623			case O_TCPMSS:
1624				s = "tcpmss";
1625				break;
1626			case O_TCPWIN:
1627				s = "tcpwin";
1628				break;
1629			default:
1630				s = "<unknown>";
1631				break;
1632			}
1633			bprintf(bp, " %s %u", s, cmd->arg1);
1634		} else
1635			print_newports(bp, insntod(cmd, u16), 0,
1636			    cmd->opcode);
1637		break;
1638	case O_IPVER:
1639		bprintf(bp, " ipver %u", cmd->arg1);
1640		break;
1641	case O_IPPRECEDENCE:
1642		bprintf(bp, " ipprecedence %u", cmd->arg1 >> 5);
1643		break;
1644	case O_DSCP:
1645		print_dscp(bp, insntod(cmd, u32));
1646		break;
1647	case O_IPOPT:
1648		print_flags(bp, "ipoptions", cmd, f_ipopts);
1649		break;
1650	case O_IPTOS:
1651		print_flags(bp, "iptos", cmd, f_iptos);
1652		break;
1653	case O_ICMPTYPE:
1654		print_icmptypes(bp, insntod(cmd, u32));
1655		break;
1656	case O_ESTAB:
1657		bprintf(bp, " established");
1658		break;
1659	case O_TCPFLAGS:
1660		print_flags(bp, "tcpflags", cmd, f_tcpflags);
1661		break;
1662	case O_TCPOPTS:
1663		print_flags(bp, "tcpoptions", cmd, f_tcpopts);
1664		break;
1665	case O_TCPACK:
1666		bprintf(bp, " tcpack %d",
1667		    ntohl(insntod(cmd, u32)->d[0]));
1668		break;
1669	case O_TCPSEQ:
1670		bprintf(bp, " tcpseq %d",
1671		    ntohl(insntod(cmd, u32)->d[0]));
1672		break;
1673	case O_UID:
1674		pwd = getpwuid(insntod(cmd, u32)->d[0]);
1675		if (pwd != NULL)
1676			bprintf(bp, " uid %s", pwd->pw_name);
1677		else
1678			bprintf(bp, " uid %u",
1679			    insntod(cmd, u32)->d[0]);
1680		break;
1681	case O_GID:
1682		grp = getgrgid(insntod(cmd, u32)->d[0]);
1683		if (grp != NULL)
1684			bprintf(bp, " gid %s", grp->gr_name);
1685		else
1686			bprintf(bp, " gid %u",
1687			    insntod(cmd, u32)->d[0]);
1688		break;
1689	case O_JAIL:
1690		bprintf(bp, " jail %d", insntod(cmd, u32)->d[0]);
1691		break;
1692	case O_VERREVPATH:
1693		bprintf(bp, " verrevpath");
1694		break;
1695	case O_VERSRCREACH:
1696		bprintf(bp, " versrcreach");
1697		break;
1698	case O_ANTISPOOF:
1699		bprintf(bp, " antispoof");
1700		break;
1701	case O_IPSEC:
1702		bprintf(bp, " ipsec");
1703		break;
1704	case O_NOP:
1705		bprintf(bp, " // %s", (const char *)(cmd + 1));
1706		break;
1707	case O_KEEP_STATE:
1708		if (state->flags & HAVE_PROBE_STATE)
1709			bprintf(bp, " keep-state");
1710		else
1711			bprintf(bp, " record-state");
1712		bprintf(bp, " :%s",
1713		    object_search_ctlv(fo->tstate, cmd->arg1,
1714		    IPFW_TLV_STATE_NAME));
1715		break;
1716	case O_LIMIT:
1717		if (state->flags & HAVE_PROBE_STATE)
1718			bprintf(bp, " limit");
1719		else
1720			bprintf(bp, " set-limit");
1721		print_limit_mask(bp, insntod(cmd, limit));
1722		bprintf(bp, " :%s",
1723		    object_search_ctlv(fo->tstate, cmd->arg1,
1724		    IPFW_TLV_STATE_NAME));
1725		break;
1726	case O_IP6:
1727		if (state->flags & HAVE_PROTO)
1728			bprintf(bp, " proto");
1729		bprintf(bp, " ip6");
1730		break;
1731	case O_IP4:
1732		if (state->flags & HAVE_PROTO)
1733			bprintf(bp, " proto");
1734		bprintf(bp, " ip4");
1735		break;
1736	case O_ICMP6TYPE:
1737		print_icmp6types(bp, insntod(cmd, u32));
1738		break;
1739	case O_EXT_HDR:
1740		print_ext6hdr(bp, cmd);
1741		break;
1742	case O_TAGGED:
1743		if (F_LEN(cmd) == 1)
1744			bprint_uint_arg(bp, " tagged ", cmd->arg1);
1745		else
1746			print_newports(bp, insntod(cmd, u16),
1747				    0, O_TAGGED);
1748		break;
1749	case O_SKIP_ACTION:
1750		bprintf(bp, " defer-immediate-action");
1751		break;
1752	default:
1753		bprintf(bp, " [opcode %d len %d]", cmd->opcode,
1754		    cmd->len);
1755	}
1756	if (cmd->len & F_OR) {
1757		bprintf(bp, " or");
1758		state->or_block = 1;
1759	} else if (state->or_block != 0) {
1760		bprintf(bp, " }");
1761		state->or_block = 0;
1762	}
1763	mark_printed(state, cmd);
1764
1765	return (1);
1766}
1767
1768static ipfw_insn *
1769print_opcode(struct buf_pr *bp, struct format_opts *fo,
1770    struct show_state *state, int opcode)
1771{
1772	ipfw_insn *cmd;
1773	int l;
1774
1775	for (l = state->rule->act_ofs, cmd = state->rule->cmd;
1776	    l > 0; l -= F_LEN(cmd), cmd += F_LEN(cmd)) {
1777		/* We use zero opcode to print the rest of options */
1778		if (opcode >= 0 && cmd->opcode != opcode)
1779			continue;
1780		/*
1781		 * Skip O_NOP, when we printing the rest
1782		 * of options, it will be handled separately.
1783		 */
1784		if (cmd->opcode == O_NOP && opcode != O_NOP)
1785			continue;
1786		if (!print_instruction(bp, fo, state, cmd))
1787			continue;
1788		return (cmd);
1789	}
1790	return (NULL);
1791}
1792
1793static void
1794print_fwd(struct buf_pr *bp, const ipfw_insn *cmd)
1795{
1796	char buf[INET6_ADDRSTRLEN + IF_NAMESIZE + 2];
1797	const ipfw_insn_sa6 *sa6;
1798	const ipfw_insn_sa *sa;
1799	uint16_t port;
1800
1801	if (cmd->opcode == O_FORWARD_IP) {
1802		sa = insntod(cmd, sa);
1803		port = sa->sa.sin_port;
1804		if (sa->sa.sin_addr.s_addr == INADDR_ANY)
1805			bprintf(bp, "fwd tablearg");
1806		else
1807			bprintf(bp, "fwd %s", inet_ntoa(sa->sa.sin_addr));
1808	} else {
1809		sa6 = insntod(cmd, sa6);
1810		port = sa6->sa.sin6_port;
1811		bprintf(bp, "fwd ");
1812		if (getnameinfo((const struct sockaddr *)&sa6->sa,
1813		    sizeof(struct sockaddr_in6), buf, sizeof(buf), NULL, 0,
1814		    NI_NUMERICHOST) == 0)
1815			bprintf(bp, "%s", buf);
1816	}
1817	if (port != 0)
1818		bprintf(bp, ",%u", port);
1819}
1820
1821static int
1822print_action_instruction(struct buf_pr *bp, const struct format_opts *fo,
1823    struct show_state *state, const ipfw_insn *cmd)
1824{
1825	const char *s;
1826
1827	if (is_printed_opcode(state, cmd))
1828		return (0);
1829	switch (cmd->opcode) {
1830	case O_CHECK_STATE:
1831		bprintf(bp, "check-state");
1832		if (cmd->arg1 != 0)
1833			s = object_search_ctlv(fo->tstate, cmd->arg1,
1834			    IPFW_TLV_STATE_NAME);
1835		else
1836			s = NULL;
1837		bprintf(bp, " :%s", s ? s: "any");
1838		break;
1839	case O_ACCEPT:
1840		bprintf(bp, "allow");
1841		break;
1842	case O_COUNT:
1843		bprintf(bp, "count");
1844		break;
1845	case O_DENY:
1846		bprintf(bp, "deny");
1847		break;
1848	case O_REJECT:
1849		if (cmd->arg1 == ICMP_REJECT_RST)
1850			bprintf(bp, "reset");
1851		else if (cmd->arg1 == ICMP_REJECT_ABORT)
1852			bprintf(bp, "abort");
1853		else if (cmd->arg1 == ICMP_UNREACH_HOST)
1854			bprintf(bp, "reject");
1855		else
1856			print_reject_code(bp, cmd->arg1);
1857		break;
1858	case O_UNREACH6:
1859		if (cmd->arg1 == ICMP6_UNREACH_RST)
1860			bprintf(bp, "reset6");
1861		else if (cmd->arg1 == ICMP6_UNREACH_ABORT)
1862			bprintf(bp, "abort6");
1863		else
1864			print_unreach6_code(bp, cmd->arg1);
1865		break;
1866	case O_SKIPTO:
1867		bprint_uint_arg(bp, "skipto ", cmd->arg1);
1868		break;
1869	case O_PIPE:
1870		bprint_uint_arg(bp, "pipe ", cmd->arg1);
1871		break;
1872	case O_QUEUE:
1873		bprint_uint_arg(bp, "queue ", cmd->arg1);
1874		break;
1875	case O_DIVERT:
1876		bprint_uint_arg(bp, "divert ", cmd->arg1);
1877		break;
1878	case O_TEE:
1879		bprint_uint_arg(bp, "tee ", cmd->arg1);
1880		break;
1881	case O_NETGRAPH:
1882		bprint_uint_arg(bp, "netgraph ", cmd->arg1);
1883		break;
1884	case O_NGTEE:
1885		bprint_uint_arg(bp, "ngtee ", cmd->arg1);
1886		break;
1887	case O_FORWARD_IP:
1888	case O_FORWARD_IP6:
1889		print_fwd(bp, cmd);
1890		break;
1891	case O_LOG:
1892		if (insntod(cmd, log)->max_log > 0)
1893			bprintf(bp, " log logamount %d",
1894			    insntod(cmd, log)->max_log);
1895		else
1896			bprintf(bp, " log");
1897		break;
1898	case O_ALTQ:
1899#ifndef NO_ALTQ
1900		print_altq_cmd(bp, insntod(cmd, altq));
1901#endif
1902		break;
1903	case O_TAG:
1904		bprint_uint_arg(bp, cmd->len & F_NOT ? " untag ":
1905		    " tag ", cmd->arg1);
1906		break;
1907	case O_NAT:
1908		if (cmd->arg1 != IP_FW_NAT44_GLOBAL)
1909			bprint_uint_arg(bp, "nat ", cmd->arg1);
1910		else
1911			bprintf(bp, "nat global");
1912		break;
1913	case O_SETFIB:
1914		if (cmd->arg1 == IP_FW_TARG)
1915			bprint_uint_arg(bp, "setfib ", cmd->arg1);
1916		else
1917			bprintf(bp, "setfib %u", cmd->arg1 & 0x7FFF);
1918		break;
1919	case O_EXTERNAL_ACTION:
1920		/*
1921		 * The external action can consists of two following
1922		 * each other opcodes - O_EXTERNAL_ACTION and
1923		 * O_EXTERNAL_INSTANCE. The first contains the ID of
1924		 * name of external action. The second contains the ID
1925		 * of name of external action instance.
1926		 * NOTE: in case when external action has no named
1927		 * instances support, the second opcode isn't needed.
1928		 */
1929		state->eaction = cmd;
1930		s = object_search_ctlv(fo->tstate, cmd->arg1,
1931		    IPFW_TLV_EACTION);
1932		if (match_token(rule_eactions, s) != -1)
1933			bprintf(bp, "%s", s);
1934		else
1935			bprintf(bp, "eaction %s", s);
1936		break;
1937	case O_EXTERNAL_INSTANCE:
1938		if (state->eaction == NULL)
1939			break;
1940		/*
1941		 * XXX: we need to teach ipfw(9) to rewrite opcodes
1942		 * in the user buffer on rule addition. When we add
1943		 * the rule, we specify zero TLV type for
1944		 * O_EXTERNAL_INSTANCE object. To show correct
1945		 * rule after `ipfw add` we need to search instance
1946		 * name with zero type. But when we do `ipfw show`
1947		 * we calculate TLV type using IPFW_TLV_EACTION_NAME()
1948		 * macro.
1949		 */
1950		s = object_search_ctlv(fo->tstate, cmd->arg1, 0);
1951		if (s == NULL)
1952			s = object_search_ctlv(fo->tstate,
1953			    cmd->arg1, IPFW_TLV_EACTION_NAME(
1954			    state->eaction->arg1));
1955		bprintf(bp, " %s", s);
1956		break;
1957	case O_EXTERNAL_DATA:
1958		if (state->eaction == NULL)
1959			break;
1960		/*
1961		 * Currently we support data formatting only for
1962		 * external data with datalen u16. For unknown data
1963		 * print its size in bytes.
1964		 */
1965		if (cmd->len == F_INSN_SIZE(ipfw_insn))
1966			bprintf(bp, " %u", cmd->arg1);
1967		else
1968			bprintf(bp, " %ubytes",
1969			    cmd->len * sizeof(uint32_t));
1970		break;
1971	case O_SETDSCP:
1972		if (cmd->arg1 == IP_FW_TARG) {
1973			bprintf(bp, "setdscp tablearg");
1974			break;
1975		}
1976		s = match_value(f_ipdscp, cmd->arg1 & 0x3F);
1977		if (s != NULL)
1978			bprintf(bp, "setdscp %s", s);
1979		else
1980			bprintf(bp, "setdscp %u", cmd->arg1 & 0x3F);
1981		break;
1982	case O_REASS:
1983		bprintf(bp, "reass");
1984		break;
1985	case O_CALLRETURN:
1986		if (cmd->len & F_NOT)
1987			bprintf(bp, "return");
1988		else
1989			bprint_uint_arg(bp, "call ", cmd->arg1);
1990		break;
1991	default:
1992		bprintf(bp, "** unrecognized action %d len %d ",
1993			cmd->opcode, cmd->len);
1994	}
1995	mark_printed(state, cmd);
1996
1997	return (1);
1998}
1999
2000
2001static ipfw_insn *
2002print_action(struct buf_pr *bp, struct format_opts *fo,
2003    struct show_state *state, uint8_t opcode)
2004{
2005	ipfw_insn *cmd;
2006	int l;
2007
2008	for (l = state->rule->cmd_len - state->rule->act_ofs,
2009	    cmd = ACTION_PTR(state->rule); l > 0;
2010	    l -= F_LEN(cmd), cmd += F_LEN(cmd)) {
2011		if (cmd->opcode != opcode)
2012			continue;
2013		if (!print_action_instruction(bp, fo, state, cmd))
2014			continue;
2015		return (cmd);
2016	}
2017	return (NULL);
2018}
2019
2020static void
2021print_proto(struct buf_pr *bp, struct format_opts *fo,
2022    struct show_state *state)
2023{
2024	ipfw_insn *cmd;
2025	int l, proto, ip4, ip6;
2026
2027	/* Count all O_PROTO, O_IP4, O_IP6 instructions. */
2028	proto = ip4 = ip6 = 0;
2029	for (l = state->rule->act_ofs, cmd = state->rule->cmd;
2030	    l > 0; l -= F_LEN(cmd), cmd += F_LEN(cmd)) {
2031		switch (cmd->opcode) {
2032		case O_PROTO:
2033			proto++;
2034			break;
2035		case O_IP4:
2036			ip4 = 1;
2037			if (cmd->len & F_OR)
2038				ip4++;
2039			break;
2040		case O_IP6:
2041			ip6 = 1;
2042			if (cmd->len & F_OR)
2043				ip6++;
2044			break;
2045		default:
2046			continue;
2047		}
2048	}
2049	if (proto == 0 && ip4 == 0 && ip6 == 0) {
2050		state->proto = IPPROTO_IP;
2051		state->flags |= HAVE_PROTO;
2052		bprintf(bp, " ip");
2053		return;
2054	}
2055	/* To handle the case { ip4 or ip6 }, print opcode with F_OR first */
2056	cmd = NULL;
2057	if (ip4 || ip6)
2058		cmd = print_opcode(bp, fo, state, ip4 > ip6 ? O_IP4: O_IP6);
2059	if (cmd != NULL && (cmd->len & F_OR))
2060		cmd = print_opcode(bp, fo, state, ip4 > ip6 ? O_IP6: O_IP4);
2061	if (cmd == NULL || (cmd->len & F_OR))
2062		for (l = proto; l > 0; l--) {
2063			cmd = print_opcode(bp, fo, state, O_PROTO);
2064			if (cmd == NULL || (cmd->len & F_OR) == 0)
2065				break;
2066		}
2067	/* Initialize proto, it is used by print_newports() */
2068	state->flags |= HAVE_PROTO;
2069	if (state->proto == 0 && ip6 != 0)
2070		state->proto = IPPROTO_IPV6;
2071}
2072
2073static int
2074match_opcode(int opcode, const int opcodes[], size_t nops)
2075{
2076	size_t i;
2077
2078	for (i = 0; i < nops; i++)
2079		if (opcode == opcodes[i])
2080			return (1);
2081	return (0);
2082}
2083
2084static void
2085print_address(struct buf_pr *bp, struct format_opts *fo,
2086    struct show_state *state, const int opcodes[], size_t nops, int portop,
2087    int flag)
2088{
2089	ipfw_insn *cmd;
2090	int count, l, portcnt, pf;
2091
2092	count = portcnt = 0;
2093	for (l = state->rule->act_ofs, cmd = state->rule->cmd;
2094	    l > 0; l -= F_LEN(cmd), cmd += F_LEN(cmd)) {
2095		if (match_opcode(cmd->opcode, opcodes, nops))
2096			count++;
2097		else if (cmd->opcode == portop)
2098			portcnt++;
2099	}
2100	if (count == 0)
2101		bprintf(bp, " any");
2102	for (l = state->rule->act_ofs, cmd = state->rule->cmd;
2103	    l > 0 && count > 0; l -= F_LEN(cmd), cmd += F_LEN(cmd)) {
2104		if (!match_opcode(cmd->opcode, opcodes, nops))
2105			continue;
2106		print_instruction(bp, fo, state, cmd);
2107		if ((cmd->len & F_OR) == 0)
2108			break;
2109		count--;
2110	}
2111	/*
2112	 * If several O_IP_?PORT opcodes specified, leave them to the
2113	 * options section.
2114	 */
2115	if (portcnt == 1) {
2116		for (l = state->rule->act_ofs, cmd = state->rule->cmd, pf = 0;
2117		    l > 0; l -= F_LEN(cmd), cmd += F_LEN(cmd)) {
2118			if (cmd->opcode != portop) {
2119				pf = (cmd->len & F_OR);
2120				continue;
2121			}
2122			/* Print opcode iff it is not in OR block. */
2123			if (pf == 0 && (cmd->len & F_OR) == 0)
2124				print_instruction(bp, fo, state, cmd);
2125			break;
2126		}
2127	}
2128	state->flags |= flag;
2129}
2130
2131static const int action_opcodes[] = {
2132	O_CHECK_STATE, O_ACCEPT, O_COUNT, O_DENY, O_REJECT,
2133	O_UNREACH6, O_SKIPTO, O_PIPE, O_QUEUE, O_DIVERT, O_TEE,
2134	O_NETGRAPH, O_NGTEE, O_FORWARD_IP, O_FORWARD_IP6, O_NAT,
2135	O_SETFIB, O_SETDSCP, O_REASS, O_CALLRETURN,
2136	/* keep the following opcodes at the end of the list */
2137	O_EXTERNAL_ACTION, O_EXTERNAL_INSTANCE, O_EXTERNAL_DATA
2138};
2139
2140static const int modifier_opcodes[] = {
2141	O_LOG, O_ALTQ, O_TAG
2142};
2143
2144static const int src_opcodes[] = {
2145	O_IP_SRC, O_IP_SRC_LOOKUP, O_IP_SRC_MASK, O_IP_SRC_ME,
2146	O_IP_SRC_SET, O_IP6_SRC, O_IP6_SRC_MASK, O_IP6_SRC_ME
2147};
2148
2149static const int dst_opcodes[] = {
2150	O_IP_DST, O_IP_DST_LOOKUP, O_IP_DST_MASK, O_IP_DST_ME,
2151	O_IP_DST_SET, O_IP6_DST, O_IP6_DST_MASK, O_IP6_DST_ME
2152};
2153
2154static void
2155show_static_rule(struct cmdline_opts *co, struct format_opts *fo,
2156    struct buf_pr *bp, struct ip_fw_rule *rule, struct ip_fw_bcounter *cntr)
2157{
2158	static int twidth = 0;
2159	struct show_state state;
2160	ipfw_insn *cmd;
2161	size_t i;
2162
2163	/* Print # DISABLED or skip the rule */
2164	if ((fo->set_mask & (1 << rule->set)) == 0) {
2165		/* disabled mask */
2166		if (!co->show_sets)
2167			return;
2168		else
2169			bprintf(bp, "# DISABLED ");
2170	}
2171	if (init_show_state(&state, rule) != 0) {
2172		warn("init_show_state() failed");
2173		return;
2174	}
2175	bprintf(bp, "%05u ", rule->rulenum);
2176
2177	/* Print counters if enabled */
2178	if (fo->pcwidth > 0 || fo->bcwidth > 0) {
2179		pr_u64(bp, &cntr->pcnt, fo->pcwidth);
2180		pr_u64(bp, &cntr->bcnt, fo->bcwidth);
2181	}
2182
2183	/* Print timestamp */
2184	if (co->do_time == TIMESTAMP_NUMERIC)
2185		bprintf(bp, "%10u ", cntr->timestamp);
2186	else if (co->do_time == TIMESTAMP_STRING) {
2187		char timestr[30];
2188		time_t t = (time_t)0;
2189
2190		if (twidth == 0) {
2191			strcpy(timestr, ctime(&t));
2192			*strchr(timestr, '\n') = '\0';
2193			twidth = strlen(timestr);
2194		}
2195		if (cntr->timestamp > 0) {
2196			t = _long_to_time(cntr->timestamp);
2197
2198			strcpy(timestr, ctime(&t));
2199			*strchr(timestr, '\n') = '\0';
2200			bprintf(bp, "%s ", timestr);
2201		} else {
2202			bprintf(bp, "%*s ", twidth, "");
2203		}
2204	}
2205
2206	/* Print set number */
2207	if (co->show_sets)
2208		bprintf(bp, "set %d ", rule->set);
2209
2210	/* Print the optional "match probability" */
2211	cmd = print_opcode(bp, fo, &state, O_PROB);
2212	/* Print rule action */
2213	for (i = 0; i < nitems(action_opcodes); i++) {
2214		cmd = print_action(bp, fo, &state, action_opcodes[i]);
2215		if (cmd == NULL)
2216			continue;
2217		/* Handle special cases */
2218		switch (cmd->opcode) {
2219		case O_CHECK_STATE:
2220			goto end;
2221		case O_EXTERNAL_ACTION:
2222		case O_EXTERNAL_INSTANCE:
2223			/* External action can have several instructions */
2224			continue;
2225		}
2226		break;
2227	}
2228	/* Print rule modifiers */
2229	for (i = 0; i < nitems(modifier_opcodes); i++)
2230		print_action(bp, fo, &state, modifier_opcodes[i]);
2231	/*
2232	 * Print rule body
2233	 */
2234	if (co->comment_only != 0)
2235		goto end;
2236
2237	if (rule->flags & IPFW_RULE_JUSTOPTS) {
2238		state.flags |= HAVE_PROTO | HAVE_SRCIP | HAVE_DSTIP;
2239		goto justopts;
2240	}
2241
2242	print_proto(bp, fo, &state);
2243	if (co->do_compact != 0 && (rule->flags & IPFW_RULE_NOOPT))
2244		goto justopts;
2245
2246	/* Print source */
2247	bprintf(bp, " from");
2248	print_address(bp, fo, &state, src_opcodes, nitems(src_opcodes),
2249	    O_IP_SRCPORT, HAVE_SRCIP);
2250
2251	/* Print destination */
2252	bprintf(bp, " to");
2253	print_address(bp, fo, &state, dst_opcodes, nitems(dst_opcodes),
2254	    O_IP_DSTPORT, HAVE_DSTIP);
2255
2256justopts:
2257	/* Print the rest of options */
2258	while (print_opcode(bp, fo, &state, -1))
2259		;
2260end:
2261	/* Print comment at the end */
2262	cmd = print_opcode(bp, fo, &state, O_NOP);
2263	if (co->comment_only != 0 && cmd == NULL)
2264		bprintf(bp, " // ...");
2265	bprintf(bp, "\n");
2266	free_show_state(&state);
2267}
2268
2269static void
2270show_dyn_state(struct cmdline_opts *co, struct format_opts *fo,
2271    struct buf_pr *bp, ipfw_dyn_rule *d)
2272{
2273	struct protoent *pe;
2274	struct in_addr a;
2275	uint16_t rulenum;
2276	char buf[INET6_ADDRSTRLEN];
2277
2278	if (d->expire == 0 && d->dyn_type != O_LIMIT_PARENT)
2279		return;
2280
2281	bcopy(&d->rule, &rulenum, sizeof(rulenum));
2282	bprintf(bp, "%05d", rulenum);
2283	if (fo->pcwidth > 0 || fo->bcwidth > 0) {
2284		bprintf(bp, " ");
2285		pr_u64(bp, &d->pcnt, fo->pcwidth);
2286		pr_u64(bp, &d->bcnt, fo->bcwidth);
2287		bprintf(bp, "(%ds)", d->expire);
2288	}
2289	switch (d->dyn_type) {
2290	case O_LIMIT_PARENT:
2291		bprintf(bp, " PARENT %d", d->count);
2292		break;
2293	case O_LIMIT:
2294		bprintf(bp, " LIMIT");
2295		break;
2296	case O_KEEP_STATE: /* bidir, no mask */
2297		bprintf(bp, " STATE");
2298		break;
2299	}
2300
2301	if ((pe = getprotobynumber(d->id.proto)) != NULL)
2302		bprintf(bp, " %s", pe->p_name);
2303	else
2304		bprintf(bp, " proto %u", d->id.proto);
2305
2306	if (d->id.addr_type == 4) {
2307		a.s_addr = htonl(d->id.src_ip);
2308		bprintf(bp, " %s %d", inet_ntoa(a), d->id.src_port);
2309
2310		a.s_addr = htonl(d->id.dst_ip);
2311		bprintf(bp, " <-> %s %d", inet_ntoa(a), d->id.dst_port);
2312	} else if (d->id.addr_type == 6) {
2313		bprintf(bp, " %s %d", inet_ntop(AF_INET6, &d->id.src_ip6, buf,
2314		    sizeof(buf)), d->id.src_port);
2315		bprintf(bp, " <-> %s %d", inet_ntop(AF_INET6, &d->id.dst_ip6,
2316		    buf, sizeof(buf)), d->id.dst_port);
2317	} else
2318		bprintf(bp, " UNKNOWN <-> UNKNOWN");
2319	if (d->kidx != 0)
2320		bprintf(bp, " :%s", object_search_ctlv(fo->tstate,
2321		    d->kidx, IPFW_TLV_STATE_NAME));
2322
2323#define	BOTH_SYN	(TH_SYN | (TH_SYN << 8))
2324#define	BOTH_FIN	(TH_FIN | (TH_FIN << 8))
2325	if (co->verbose) {
2326		bprintf(bp, " state 0x%08x%s", d->state,
2327		    d->state ? " ": ",");
2328		if (d->state & IPFW_DYN_ORPHANED)
2329			bprintf(bp, "ORPHANED,");
2330		if ((d->state & BOTH_SYN) == BOTH_SYN)
2331			bprintf(bp, "BOTH_SYN,");
2332		else {
2333			if (d->state & TH_SYN)
2334				bprintf(bp, "F_SYN,");
2335			if (d->state & (TH_SYN << 8))
2336				bprintf(bp, "R_SYN,");
2337		}
2338		if ((d->state & BOTH_FIN) == BOTH_FIN)
2339			bprintf(bp, "BOTH_FIN,");
2340		else {
2341			if (d->state & TH_FIN)
2342				bprintf(bp, "F_FIN,");
2343			if (d->state & (TH_FIN << 8))
2344				bprintf(bp, "R_FIN,");
2345		}
2346		bprintf(bp, " f_ack 0x%x, r_ack 0x%x", d->ack_fwd,
2347		    d->ack_rev);
2348	}
2349}
2350
2351static int
2352do_range_cmd(int cmd, ipfw_range_tlv *rt)
2353{
2354	ipfw_range_header rh;
2355	size_t sz;
2356
2357	memset(&rh, 0, sizeof(rh));
2358	memcpy(&rh.range, rt, sizeof(*rt));
2359	rh.range.head.length = sizeof(*rt);
2360	rh.range.head.type = IPFW_TLV_RANGE;
2361	sz = sizeof(rh);
2362
2363	if (do_get3(cmd, &rh.opheader, &sz) != 0)
2364		return (-1);
2365	/* Save number of matched objects */
2366	rt->new_set = rh.range.new_set;
2367	return (0);
2368}
2369
2370/*
2371 * This one handles all set-related commands
2372 * 	ipfw set { show | enable | disable }
2373 * 	ipfw set swap X Y
2374 * 	ipfw set move X to Y
2375 * 	ipfw set move rule X to Y
2376 */
2377void
2378ipfw_sets_handler(char *av[])
2379{
2380	ipfw_range_tlv rt;
2381	const char *msg;
2382	size_t size;
2383	uint32_t masks[2];
2384	int i;
2385	uint16_t rulenum;
2386	uint8_t cmd;
2387
2388	av++;
2389	memset(&rt, 0, sizeof(rt));
2390
2391	if (av[0] == NULL)
2392		errx(EX_USAGE, "set needs command");
2393	if (_substrcmp(*av, "show") == 0) {
2394		struct format_opts fo;
2395		ipfw_cfg_lheader *cfg;
2396
2397		memset(&fo, 0, sizeof(fo));
2398		if (ipfw_get_config(&g_co, &fo, &cfg, &size) != 0)
2399			err(EX_OSERR, "requesting config failed");
2400
2401		for (i = 0, msg = "disable"; i < RESVD_SET; i++)
2402			if ((cfg->set_mask & (1<<i)) == 0) {
2403				printf("%s %d", msg, i);
2404				msg = "";
2405			}
2406		msg = (cfg->set_mask != (uint32_t)-1) ? " enable" : "enable";
2407		for (i = 0; i < RESVD_SET; i++)
2408			if ((cfg->set_mask & (1<<i)) != 0) {
2409				printf("%s %d", msg, i);
2410				msg = "";
2411			}
2412		printf("\n");
2413		free(cfg);
2414	} else if (_substrcmp(*av, "swap") == 0) {
2415		av++;
2416		if ( av[0] == NULL || av[1] == NULL )
2417			errx(EX_USAGE, "set swap needs 2 set numbers\n");
2418		rt.set = atoi(av[0]);
2419		rt.new_set = atoi(av[1]);
2420		if (!isdigit(*(av[0])) || rt.set > RESVD_SET)
2421			errx(EX_DATAERR, "invalid set number %s\n", av[0]);
2422		if (!isdigit(*(av[1])) || rt.new_set > RESVD_SET)
2423			errx(EX_DATAERR, "invalid set number %s\n", av[1]);
2424		i = do_range_cmd(IP_FW_SET_SWAP, &rt);
2425	} else if (_substrcmp(*av, "move") == 0) {
2426		av++;
2427		if (av[0] && _substrcmp(*av, "rule") == 0) {
2428			rt.flags = IPFW_RCFLAG_RANGE; /* move rules to new set */
2429			cmd = IP_FW_XMOVE;
2430			av++;
2431		} else
2432			cmd = IP_FW_SET_MOVE; /* Move set to new one */
2433		if (av[0] == NULL || av[1] == NULL || av[2] == NULL ||
2434				av[3] != NULL ||  _substrcmp(av[1], "to") != 0)
2435			errx(EX_USAGE, "syntax: set move [rule] X to Y\n");
2436		rulenum = atoi(av[0]);
2437		rt.new_set = atoi(av[2]);
2438		if (cmd == IP_FW_XMOVE) {
2439			rt.start_rule = rulenum;
2440			rt.end_rule = rulenum;
2441		} else
2442			rt.set = rulenum;
2443		rt.new_set = atoi(av[2]);
2444		if (!isdigit(*(av[0])) || (cmd == 3 && rt.set > RESVD_SET) ||
2445			(cmd == 2 && rt.start_rule == IPFW_DEFAULT_RULE) )
2446			errx(EX_DATAERR, "invalid source number %s\n", av[0]);
2447		if (!isdigit(*(av[2])) || rt.new_set > RESVD_SET)
2448			errx(EX_DATAERR, "invalid dest. set %s\n", av[1]);
2449		i = do_range_cmd(cmd, &rt);
2450		if (i < 0)
2451			err(EX_OSERR, "failed to move %s",
2452			    cmd == IP_FW_SET_MOVE ? "set": "rule");
2453	} else if (_substrcmp(*av, "disable") == 0 ||
2454		   _substrcmp(*av, "enable") == 0 ) {
2455		int which = _substrcmp(*av, "enable") == 0 ? 1 : 0;
2456
2457		av++;
2458		masks[0] = masks[1] = 0;
2459
2460		while (av[0]) {
2461			if (isdigit(**av)) {
2462				i = atoi(*av);
2463				if (i < 0 || i > RESVD_SET)
2464					errx(EX_DATAERR,
2465					    "invalid set number %d\n", i);
2466				masks[which] |= (1<<i);
2467			} else if (_substrcmp(*av, "disable") == 0)
2468				which = 0;
2469			else if (_substrcmp(*av, "enable") == 0)
2470				which = 1;
2471			else
2472				errx(EX_DATAERR,
2473					"invalid set command %s\n", *av);
2474			av++;
2475		}
2476		if ( (masks[0] & masks[1]) != 0 )
2477			errx(EX_DATAERR,
2478			    "cannot enable and disable the same set\n");
2479
2480		rt.set = masks[0];
2481		rt.new_set = masks[1];
2482		i = do_range_cmd(IP_FW_SET_ENABLE, &rt);
2483		if (i)
2484			warn("set enable/disable: setsockopt(IP_FW_SET_ENABLE)");
2485	} else
2486		errx(EX_USAGE, "invalid set command %s\n", *av);
2487}
2488
2489void
2490ipfw_sysctl_handler(char *av[], int which)
2491{
2492	av++;
2493
2494	if (av[0] == NULL) {
2495		warnx("missing keyword to enable/disable\n");
2496	} else if (_substrcmp(*av, "firewall") == 0) {
2497		sysctlbyname("net.inet.ip.fw.enable", NULL, 0,
2498		    &which, sizeof(which));
2499		sysctlbyname("net.inet6.ip6.fw.enable", NULL, 0,
2500		    &which, sizeof(which));
2501	} else if (_substrcmp(*av, "one_pass") == 0) {
2502		sysctlbyname("net.inet.ip.fw.one_pass", NULL, 0,
2503		    &which, sizeof(which));
2504	} else if (_substrcmp(*av, "debug") == 0) {
2505		sysctlbyname("net.inet.ip.fw.debug", NULL, 0,
2506		    &which, sizeof(which));
2507	} else if (_substrcmp(*av, "verbose") == 0) {
2508		sysctlbyname("net.inet.ip.fw.verbose", NULL, 0,
2509		    &which, sizeof(which));
2510	} else if (_substrcmp(*av, "dyn_keepalive") == 0) {
2511		sysctlbyname("net.inet.ip.fw.dyn_keepalive", NULL, 0,
2512		    &which, sizeof(which));
2513#ifndef NO_ALTQ
2514	} else if (_substrcmp(*av, "altq") == 0) {
2515		altq_set_enabled(which);
2516#endif
2517	} else {
2518		warnx("unrecognize enable/disable keyword: %s\n", *av);
2519	}
2520}
2521
2522typedef void state_cb(struct cmdline_opts *co, struct format_opts *fo,
2523    void *arg, void *state);
2524
2525static void
2526prepare_format_dyn(struct cmdline_opts *co, struct format_opts *fo,
2527    void *arg __unused, void *_state)
2528{
2529	ipfw_dyn_rule *d;
2530	int width;
2531	uint8_t set;
2532
2533	d = (ipfw_dyn_rule *)_state;
2534	/* Count _ALL_ states */
2535	fo->dcnt++;
2536
2537	if (fo->show_counters == 0)
2538		return;
2539
2540	if (co->use_set) {
2541		/* skip states from another set */
2542		bcopy((char *)&d->rule + sizeof(uint16_t), &set,
2543		    sizeof(uint8_t));
2544		if (set != co->use_set - 1)
2545			return;
2546	}
2547
2548	width = pr_u64(NULL, &d->pcnt, 0);
2549	if (width > fo->pcwidth)
2550		fo->pcwidth = width;
2551
2552	width = pr_u64(NULL, &d->bcnt, 0);
2553	if (width > fo->bcwidth)
2554		fo->bcwidth = width;
2555}
2556
2557static int
2558foreach_state(struct cmdline_opts *co, struct format_opts *fo,
2559    caddr_t base, size_t sz, state_cb dyn_bc, void *dyn_arg)
2560{
2561	int ttype;
2562	state_cb *fptr;
2563	void *farg;
2564	ipfw_obj_tlv *tlv;
2565	ipfw_obj_ctlv *ctlv;
2566
2567	fptr = NULL;
2568	ttype = 0;
2569
2570	while (sz > 0) {
2571		ctlv = (ipfw_obj_ctlv *)base;
2572		switch (ctlv->head.type) {
2573		case IPFW_TLV_DYNSTATE_LIST:
2574			base += sizeof(*ctlv);
2575			sz -= sizeof(*ctlv);
2576			ttype = IPFW_TLV_DYN_ENT;
2577			fptr = dyn_bc;
2578			farg = dyn_arg;
2579			break;
2580		default:
2581			return (sz);
2582		}
2583
2584		while (sz > 0) {
2585			tlv = (ipfw_obj_tlv *)base;
2586			if (tlv->type != ttype)
2587				break;
2588
2589			fptr(co, fo, farg, tlv + 1);
2590			sz -= tlv->length;
2591			base += tlv->length;
2592		}
2593	}
2594
2595	return (sz);
2596}
2597
2598static void
2599prepare_format_opts(struct cmdline_opts *co, struct format_opts *fo,
2600    ipfw_obj_tlv *rtlv, int rcnt, caddr_t dynbase, size_t dynsz)
2601{
2602	int bcwidth, pcwidth, width;
2603	int n;
2604	struct ip_fw_bcounter *cntr;
2605	struct ip_fw_rule *r;
2606
2607	bcwidth = 0;
2608	pcwidth = 0;
2609	if (fo->show_counters != 0) {
2610		for (n = 0; n < rcnt; n++,
2611		    rtlv = (ipfw_obj_tlv *)((caddr_t)rtlv + rtlv->length)) {
2612			cntr = (struct ip_fw_bcounter *)(rtlv + 1);
2613			r = (struct ip_fw_rule *)((caddr_t)cntr + cntr->size);
2614			/* skip rules from another set */
2615			if (co->use_set && r->set != co->use_set - 1)
2616				continue;
2617
2618			/* packet counter */
2619			width = pr_u64(NULL, &cntr->pcnt, 0);
2620			if (width > pcwidth)
2621				pcwidth = width;
2622
2623			/* byte counter */
2624			width = pr_u64(NULL, &cntr->bcnt, 0);
2625			if (width > bcwidth)
2626				bcwidth = width;
2627		}
2628	}
2629	fo->bcwidth = bcwidth;
2630	fo->pcwidth = pcwidth;
2631
2632	fo->dcnt = 0;
2633	if (co->do_dynamic && dynsz > 0)
2634		foreach_state(co, fo, dynbase, dynsz, prepare_format_dyn, NULL);
2635}
2636
2637static int
2638list_static_range(struct cmdline_opts *co, struct format_opts *fo,
2639    struct buf_pr *bp, ipfw_obj_tlv *rtlv, int rcnt)
2640{
2641	int n, seen;
2642	struct ip_fw_rule *r;
2643	struct ip_fw_bcounter *cntr;
2644	int c = 0;
2645
2646	for (n = seen = 0; n < rcnt; n++,
2647	    rtlv = (ipfw_obj_tlv *)((caddr_t)rtlv + rtlv->length)) {
2648
2649		if ((fo->show_counters | fo->show_time) != 0) {
2650			cntr = (struct ip_fw_bcounter *)(rtlv + 1);
2651			r = (struct ip_fw_rule *)((caddr_t)cntr + cntr->size);
2652		} else {
2653			cntr = NULL;
2654			r = (struct ip_fw_rule *)(rtlv + 1);
2655		}
2656		if (r->rulenum > fo->last)
2657			break;
2658		if (co->use_set && r->set != co->use_set - 1)
2659			continue;
2660		if (r->rulenum >= fo->first && r->rulenum <= fo->last) {
2661			show_static_rule(co, fo, bp, r, cntr);
2662			printf("%s", bp->buf);
2663			c += rtlv->length;
2664			bp_flush(bp);
2665			seen++;
2666		}
2667	}
2668
2669	return (seen);
2670}
2671
2672static void
2673list_dyn_state(struct cmdline_opts *co, struct format_opts *fo,
2674    void *_arg, void *_state)
2675{
2676	uint16_t rulenum;
2677	uint8_t set;
2678	ipfw_dyn_rule *d;
2679	struct buf_pr *bp;
2680
2681	d = (ipfw_dyn_rule *)_state;
2682	bp = (struct buf_pr *)_arg;
2683
2684	bcopy(&d->rule, &rulenum, sizeof(rulenum));
2685	if (rulenum > fo->last)
2686		return;
2687	if (co->use_set) {
2688		bcopy((char *)&d->rule + sizeof(uint16_t),
2689		      &set, sizeof(uint8_t));
2690		if (set != co->use_set - 1)
2691			return;
2692	}
2693	if (rulenum >= fo->first) {
2694		show_dyn_state(co, fo, bp, d);
2695		printf("%s\n", bp->buf);
2696		bp_flush(bp);
2697	}
2698}
2699
2700static int
2701list_dyn_range(struct cmdline_opts *co, struct format_opts *fo,
2702    struct buf_pr *bp, caddr_t base, size_t sz)
2703{
2704
2705	sz = foreach_state(co, fo, base, sz, list_dyn_state, bp);
2706	return (sz);
2707}
2708
2709void
2710ipfw_list(int ac, char *av[], int show_counters)
2711{
2712	ipfw_cfg_lheader *cfg;
2713	struct format_opts sfo;
2714	size_t sz;
2715	int error;
2716	int lac;
2717	char **lav;
2718	uint32_t rnum;
2719	char *endptr;
2720
2721	if (g_co.test_only) {
2722		fprintf(stderr, "Testing only, list disabled\n");
2723		return;
2724	}
2725	if (g_co.do_pipe) {
2726		dummynet_list(ac, av, show_counters);
2727		return;
2728	}
2729
2730	ac--;
2731	av++;
2732	memset(&sfo, 0, sizeof(sfo));
2733
2734	/* Determine rule range to request */
2735	if (ac > 0) {
2736		for (lac = ac, lav = av; lac != 0; lac--) {
2737			rnum = strtoul(*lav++, &endptr, 10);
2738			if (sfo.first == 0 || rnum < sfo.first)
2739				sfo.first = rnum;
2740
2741			if (*endptr == '-')
2742				rnum = strtoul(endptr + 1, &endptr, 10);
2743			if (sfo.last == 0 || rnum > sfo.last)
2744				sfo.last = rnum;
2745		}
2746	}
2747
2748	/* get configuraion from kernel */
2749	cfg = NULL;
2750	sfo.show_counters = show_counters;
2751	sfo.show_time = g_co.do_time;
2752	if (g_co.do_dynamic != 2)
2753		sfo.flags |= IPFW_CFG_GET_STATIC;
2754	if (g_co.do_dynamic != 0)
2755		sfo.flags |= IPFW_CFG_GET_STATES;
2756	if ((sfo.show_counters | sfo.show_time) != 0)
2757		sfo.flags |= IPFW_CFG_GET_COUNTERS;
2758	if (ipfw_get_config(&g_co, &sfo, &cfg, &sz) != 0)
2759		err(EX_OSERR, "retrieving config failed");
2760
2761	error = ipfw_show_config(&g_co, &sfo, cfg, sz, ac, av);
2762
2763	free(cfg);
2764
2765	if (error != EX_OK)
2766		exit(error);
2767}
2768
2769static int
2770ipfw_show_config(struct cmdline_opts *co, struct format_opts *fo,
2771    ipfw_cfg_lheader *cfg, size_t sz, int ac, char *av[])
2772{
2773	caddr_t dynbase;
2774	size_t dynsz;
2775	int rcnt;
2776	int exitval = EX_OK;
2777	int lac;
2778	char **lav;
2779	char *endptr;
2780	size_t readsz;
2781	struct buf_pr bp;
2782	ipfw_obj_ctlv *ctlv, *tstate;
2783	ipfw_obj_tlv *rbase;
2784
2785	/*
2786	 * Handle tablenames TLV first, if any
2787	 */
2788	tstate = NULL;
2789	rbase = NULL;
2790	dynbase = NULL;
2791	dynsz = 0;
2792	readsz = sizeof(*cfg);
2793	rcnt = 0;
2794
2795	fo->set_mask = cfg->set_mask;
2796
2797	ctlv = (ipfw_obj_ctlv *)(cfg + 1);
2798	if (ctlv->head.type == IPFW_TLV_TBLNAME_LIST) {
2799		object_sort_ctlv(ctlv);
2800		fo->tstate = ctlv;
2801		readsz += ctlv->head.length;
2802		ctlv = (ipfw_obj_ctlv *)((caddr_t)ctlv + ctlv->head.length);
2803	}
2804
2805	if (cfg->flags & IPFW_CFG_GET_STATIC) {
2806		/* We've requested static rules */
2807		if (ctlv->head.type == IPFW_TLV_RULE_LIST) {
2808			rbase = (ipfw_obj_tlv *)(ctlv + 1);
2809			rcnt = ctlv->count;
2810			readsz += ctlv->head.length;
2811			ctlv = (ipfw_obj_ctlv *)((caddr_t)ctlv +
2812			    ctlv->head.length);
2813		}
2814	}
2815
2816	if ((cfg->flags & IPFW_CFG_GET_STATES) && (readsz != sz))  {
2817		/* We may have some dynamic states */
2818		dynsz = sz - readsz;
2819		/* Skip empty header */
2820		if (dynsz != sizeof(ipfw_obj_ctlv))
2821			dynbase = (caddr_t)ctlv;
2822		else
2823			dynsz = 0;
2824	}
2825
2826	prepare_format_opts(co, fo, rbase, rcnt, dynbase, dynsz);
2827	bp_alloc(&bp, 4096);
2828
2829	/* if no rule numbers were specified, list all rules */
2830	if (ac == 0) {
2831		fo->first = 0;
2832		fo->last = IPFW_DEFAULT_RULE;
2833		if (cfg->flags & IPFW_CFG_GET_STATIC)
2834			list_static_range(co, fo, &bp, rbase, rcnt);
2835
2836		if (co->do_dynamic && dynsz > 0) {
2837			printf("## Dynamic rules (%d %zu):\n", fo->dcnt,
2838			    dynsz);
2839			list_dyn_range(co, fo, &bp, dynbase, dynsz);
2840		}
2841
2842		bp_free(&bp);
2843		return (EX_OK);
2844	}
2845
2846	/* display specific rules requested on command line */
2847	for (lac = ac, lav = av; lac != 0; lac--) {
2848		/* convert command line rule # */
2849		fo->last = fo->first = strtoul(*lav++, &endptr, 10);
2850		if (*endptr == '-')
2851			fo->last = strtoul(endptr + 1, &endptr, 10);
2852		if (*endptr) {
2853			exitval = EX_USAGE;
2854			warnx("invalid rule number: %s", *(lav - 1));
2855			continue;
2856		}
2857
2858		if ((cfg->flags & IPFW_CFG_GET_STATIC) == 0)
2859			continue;
2860
2861		if (list_static_range(co, fo, &bp, rbase, rcnt) == 0) {
2862			/* give precedence to other error(s) */
2863			if (exitval == EX_OK)
2864				exitval = EX_UNAVAILABLE;
2865			if (fo->first == fo->last)
2866				warnx("rule %u does not exist", fo->first);
2867			else
2868				warnx("no rules in range %u-%u",
2869				    fo->first, fo->last);
2870		}
2871	}
2872
2873	if (co->do_dynamic && dynsz > 0) {
2874		printf("## Dynamic rules:\n");
2875		for (lac = ac, lav = av; lac != 0; lac--) {
2876			fo->last = fo->first = strtoul(*lav++, &endptr, 10);
2877			if (*endptr == '-')
2878				fo->last = strtoul(endptr+1, &endptr, 10);
2879			if (*endptr)
2880				/* already warned */
2881				continue;
2882			list_dyn_range(co, fo, &bp, dynbase, dynsz);
2883		}
2884	}
2885
2886	bp_free(&bp);
2887	return (exitval);
2888}
2889
2890
2891/*
2892 * Retrieves current ipfw configuration of given type
2893 * and stores its pointer to @pcfg.
2894 *
2895 * Caller is responsible for freeing @pcfg.
2896 *
2897 * Returns 0 on success.
2898 */
2899
2900static int
2901ipfw_get_config(struct cmdline_opts *co, struct format_opts *fo,
2902    ipfw_cfg_lheader **pcfg, size_t *psize)
2903{
2904	ipfw_cfg_lheader *cfg;
2905	size_t sz;
2906	int i;
2907
2908
2909	if (co->test_only != 0) {
2910		fprintf(stderr, "Testing only, list disabled\n");
2911		return (0);
2912	}
2913
2914	/* Start with some data size */
2915	sz = 4096;
2916	cfg = NULL;
2917
2918	for (i = 0; i < 16; i++) {
2919		if (cfg != NULL)
2920			free(cfg);
2921		if ((cfg = calloc(1, sz)) == NULL)
2922			return (ENOMEM);
2923
2924		cfg->flags = fo->flags;
2925		cfg->start_rule = fo->first;
2926		cfg->end_rule = fo->last;
2927
2928		if (do_get3(IP_FW_XGET, &cfg->opheader, &sz) != 0) {
2929			if (errno != ENOMEM) {
2930				free(cfg);
2931				return (errno);
2932			}
2933
2934			/* Buffer size is not enough. Try to increase */
2935			sz = sz * 2;
2936			if (sz < cfg->size)
2937				sz = cfg->size;
2938			continue;
2939		}
2940
2941		*pcfg = cfg;
2942		*psize = sz;
2943		return (0);
2944	}
2945
2946	free(cfg);
2947	return (ENOMEM);
2948}
2949
2950static int
2951lookup_host (char *host, struct in_addr *ipaddr)
2952{
2953	struct hostent *he;
2954
2955	if (!inet_aton(host, ipaddr)) {
2956		if ((he = gethostbyname(host)) == NULL)
2957			return(-1);
2958		*ipaddr = *(struct in_addr *)he->h_addr_list[0];
2959	}
2960	return(0);
2961}
2962
2963struct tidx {
2964	ipfw_obj_ntlv *idx;
2965	uint32_t count;
2966	uint32_t size;
2967	uint16_t counter;
2968	uint8_t set;
2969};
2970
2971int
2972ipfw_check_object_name(const char *name)
2973{
2974	int c, i, l;
2975
2976	/*
2977	 * Check that name is null-terminated and contains
2978	 * valid symbols only. Valid mask is:
2979	 * [a-zA-Z0-9\-_\.]{1,63}
2980	 */
2981	l = strlen(name);
2982	if (l == 0 || l >= 64)
2983		return (EINVAL);
2984	for (i = 0; i < l; i++) {
2985		c = name[i];
2986		if (isalpha(c) || isdigit(c) || c == '_' ||
2987		    c == '-' || c == '.')
2988			continue;
2989		return (EINVAL);
2990	}
2991	return (0);
2992}
2993
2994static const char *default_state_name = "default";
2995
2996static int
2997state_check_name(const char *name)
2998{
2999
3000	if (ipfw_check_object_name(name) != 0)
3001		return (EINVAL);
3002	if (strcmp(name, "any") == 0)
3003		return (EINVAL);
3004	return (0);
3005}
3006
3007static int
3008eaction_check_name(const char *name)
3009{
3010
3011	if (ipfw_check_object_name(name) != 0)
3012		return (EINVAL);
3013	/* Restrict some 'special' names */
3014	if (match_token(rule_actions, name) != -1 &&
3015	    match_token(rule_action_params, name) != -1)
3016		return (EINVAL);
3017	return (0);
3018}
3019
3020static uint16_t
3021pack_object(struct tidx *tstate, const char *name, int otype)
3022{
3023	ipfw_obj_ntlv *ntlv;
3024	uint32_t i;
3025
3026	for (i = 0; i < tstate->count; i++) {
3027		if (strcmp(tstate->idx[i].name, name) != 0)
3028			continue;
3029		if (tstate->idx[i].set != tstate->set)
3030			continue;
3031		if (tstate->idx[i].head.type != otype)
3032			continue;
3033
3034		return (tstate->idx[i].idx);
3035	}
3036
3037	if (tstate->count + 1 > tstate->size) {
3038		tstate->size += 4;
3039		tstate->idx = realloc(tstate->idx, tstate->size *
3040		    sizeof(ipfw_obj_ntlv));
3041		if (tstate->idx == NULL)
3042			return (0);
3043	}
3044
3045	ntlv = &tstate->idx[i];
3046	memset(ntlv, 0, sizeof(ipfw_obj_ntlv));
3047	strlcpy(ntlv->name, name, sizeof(ntlv->name));
3048	ntlv->head.type = otype;
3049	ntlv->head.length = sizeof(ipfw_obj_ntlv);
3050	ntlv->set = tstate->set;
3051	ntlv->idx = ++tstate->counter;
3052	tstate->count++;
3053
3054	return (ntlv->idx);
3055}
3056
3057static uint16_t
3058pack_table(struct tidx *tstate, const char *name)
3059{
3060
3061	if (table_check_name(name) != 0)
3062		return (0);
3063
3064	return (pack_object(tstate, name, IPFW_TLV_TBL_NAME));
3065}
3066
3067void
3068fill_table(struct _ipfw_insn *cmd, char *av, uint8_t opcode,
3069    struct tidx *tstate)
3070{
3071	uint32_t *d = ((ipfw_insn_u32 *)cmd)->d;
3072	uint16_t uidx;
3073	char *p;
3074
3075	if ((p = strchr(av + 6, ')')) == NULL)
3076		errx(EX_DATAERR, "forgotten parenthesis: '%s'", av);
3077	*p = '\0';
3078	p = strchr(av + 6, ',');
3079	if (p)
3080		*p++ = '\0';
3081
3082	if ((uidx = pack_table(tstate, av + 6)) == 0)
3083		errx(EX_DATAERR, "Invalid table name: %s", av + 6);
3084
3085	cmd->opcode = opcode;
3086	cmd->arg1 = uidx;
3087	if (p) {
3088		cmd->len |= F_INSN_SIZE(ipfw_insn_u32);
3089		d[0] = strtoul(p, NULL, 0);
3090	} else
3091		cmd->len |= F_INSN_SIZE(ipfw_insn);
3092}
3093
3094
3095/*
3096 * fills the addr and mask fields in the instruction as appropriate from av.
3097 * Update length as appropriate.
3098 * The following formats are allowed:
3099 *	me	returns O_IP_*_ME
3100 *	1.2.3.4		single IP address
3101 *	1.2.3.4:5.6.7.8	address:mask
3102 *	1.2.3.4/24	address/mask
3103 *	1.2.3.4/26{1,6,5,4,23}	set of addresses in a subnet
3104 * We can have multiple comma-separated address/mask entries.
3105 */
3106static void
3107fill_ip(ipfw_insn_ip *cmd, char *av, int cblen, struct tidx *tstate)
3108{
3109	int len = 0;
3110	uint32_t *d = ((ipfw_insn_u32 *)cmd)->d;
3111
3112	cmd->o.len &= ~F_LEN_MASK;	/* zero len */
3113
3114	if (_substrcmp(av, "any") == 0)
3115		return;
3116
3117	if (_substrcmp(av, "me") == 0) {
3118		cmd->o.len |= F_INSN_SIZE(ipfw_insn);
3119		return;
3120	}
3121
3122	if (strncmp(av, "table(", 6) == 0) {
3123		fill_table(&cmd->o, av, O_IP_DST_LOOKUP, tstate);
3124		return;
3125	}
3126
3127    while (av) {
3128	/*
3129	 * After the address we can have '/' or ':' indicating a mask,
3130	 * ',' indicating another address follows, '{' indicating a
3131	 * set of addresses of unspecified size.
3132	 */
3133	char *t = NULL, *p = strpbrk(av, "/:,{");
3134	int masklen;
3135	char md, nd = '\0';
3136
3137	CHECK_LENGTH(cblen, (int)F_INSN_SIZE(ipfw_insn) + 2 + len);
3138
3139	if (p) {
3140		md = *p;
3141		*p++ = '\0';
3142		if ((t = strpbrk(p, ",{")) != NULL) {
3143			nd = *t;
3144			*t = '\0';
3145		}
3146	} else
3147		md = '\0';
3148
3149	if (lookup_host(av, (struct in_addr *)&d[0]) != 0)
3150		errx(EX_NOHOST, "hostname ``%s'' unknown", av);
3151	switch (md) {
3152	case ':':
3153		if (!inet_aton(p, (struct in_addr *)&d[1]))
3154			errx(EX_DATAERR, "bad netmask ``%s''", p);
3155		break;
3156	case '/':
3157		masklen = atoi(p);
3158		if (masklen == 0)
3159			d[1] = htonl(0U);	/* mask */
3160		else if (masklen > 32)
3161			errx(EX_DATAERR, "bad width ``%s''", p);
3162		else
3163			d[1] = htonl(~0U << (32 - masklen));
3164		break;
3165	case '{':	/* no mask, assume /24 and put back the '{' */
3166		d[1] = htonl(~0U << (32 - 24));
3167		*(--p) = md;
3168		break;
3169
3170	case ',':	/* single address plus continuation */
3171		*(--p) = md;
3172		/* FALLTHROUGH */
3173	case 0:		/* initialization value */
3174	default:
3175		d[1] = htonl(~0U);	/* force /32 */
3176		break;
3177	}
3178	d[0] &= d[1];		/* mask base address with mask */
3179	if (t)
3180		*t = nd;
3181	/* find next separator */
3182	if (p)
3183		p = strpbrk(p, ",{");
3184	if (p && *p == '{') {
3185		/*
3186		 * We have a set of addresses. They are stored as follows:
3187		 *   arg1	is the set size (powers of 2, 2..256)
3188		 *   addr	is the base address IN HOST FORMAT
3189		 *   mask..	is an array of arg1 bits (rounded up to
3190		 *		the next multiple of 32) with bits set
3191		 *		for each host in the map.
3192		 */
3193		uint32_t *map = (uint32_t *)&cmd->mask;
3194		int low, high;
3195		int i = contigmask((uint8_t *)&(d[1]), 32);
3196
3197		if (len > 0)
3198			errx(EX_DATAERR, "address set cannot be in a list");
3199		if (i < 24 || i > 31)
3200			errx(EX_DATAERR, "invalid set with mask %d\n", i);
3201		cmd->o.arg1 = 1<<(32-i);	/* map length		*/
3202		d[0] = ntohl(d[0]);		/* base addr in host format */
3203		cmd->o.opcode = O_IP_DST_SET;	/* default */
3204		cmd->o.len |= F_INSN_SIZE(ipfw_insn_u32) + (cmd->o.arg1+31)/32;
3205		for (i = 0; i < (cmd->o.arg1+31)/32 ; i++)
3206			map[i] = 0;	/* clear map */
3207
3208		av = p + 1;
3209		low = d[0] & 0xff;
3210		high = low + cmd->o.arg1 - 1;
3211		/*
3212		 * Here, i stores the previous value when we specify a range
3213		 * of addresses within a mask, e.g. 45-63. i = -1 means we
3214		 * have no previous value.
3215		 */
3216		i = -1;	/* previous value in a range */
3217		while (isdigit(*av)) {
3218			char *s;
3219			int a = strtol(av, &s, 0);
3220
3221			if (s == av) { /* no parameter */
3222			    if (*av != '}')
3223				errx(EX_DATAERR, "set not closed\n");
3224			    if (i != -1)
3225				errx(EX_DATAERR, "incomplete range %d-", i);
3226			    break;
3227			}
3228			if (a < low || a > high)
3229			    errx(EX_DATAERR, "addr %d out of range [%d-%d]\n",
3230				a, low, high);
3231			a -= low;
3232			if (i == -1)	/* no previous in range */
3233			    i = a;
3234			else {		/* check that range is valid */
3235			    if (i > a)
3236				errx(EX_DATAERR, "invalid range %d-%d",
3237					i+low, a+low);
3238			    if (*s == '-')
3239				errx(EX_DATAERR, "double '-' in range");
3240			}
3241			for (; i <= a; i++)
3242			    map[i/32] |= 1<<(i & 31);
3243			i = -1;
3244			if (*s == '-')
3245			    i = a;
3246			else if (*s == '}')
3247			    break;
3248			av = s+1;
3249		}
3250		return;
3251	}
3252	av = p;
3253	if (av)			/* then *av must be a ',' */
3254		av++;
3255
3256	/* Check this entry */
3257	if (d[1] == 0) { /* "any", specified as x.x.x.x/0 */
3258		/*
3259		 * 'any' turns the entire list into a NOP.
3260		 * 'not any' never matches, so it is removed from the
3261		 * list unless it is the only item, in which case we
3262		 * report an error.
3263		 */
3264		if (cmd->o.len & F_NOT) {	/* "not any" never matches */
3265			if (av == NULL && len == 0) /* only this entry */
3266				errx(EX_DATAERR, "not any never matches");
3267		}
3268		/* else do nothing and skip this entry */
3269		return;
3270	}
3271	/* A single IP can be stored in an optimized format */
3272	if (d[1] == (uint32_t)~0 && av == NULL && len == 0) {
3273		cmd->o.len |= F_INSN_SIZE(ipfw_insn_u32);
3274		return;
3275	}
3276	len += 2;	/* two words... */
3277	d += 2;
3278    } /* end while */
3279    if (len + 1 > F_LEN_MASK)
3280	errx(EX_DATAERR, "address list too long");
3281    cmd->o.len |= len+1;
3282}
3283
3284
3285/* n2mask sets n bits of the mask */
3286void
3287n2mask(struct in6_addr *mask, int n)
3288{
3289	static int	minimask[9] =
3290	    { 0x00, 0x80, 0xc0, 0xe0, 0xf0, 0xf8, 0xfc, 0xfe, 0xff };
3291	u_char		*p;
3292
3293	memset(mask, 0, sizeof(struct in6_addr));
3294	p = (u_char *) mask;
3295	for (; n > 0; p++, n -= 8) {
3296		if (n >= 8)
3297			*p = 0xff;
3298		else
3299			*p = minimask[n];
3300	}
3301	return;
3302}
3303
3304static void
3305fill_flags_cmd(ipfw_insn *cmd, enum ipfw_opcodes opcode,
3306	struct _s_x *flags, char *p)
3307{
3308	char *e;
3309	uint32_t set = 0, clear = 0;
3310
3311	if (fill_flags(flags, p, &e, &set, &clear) != 0)
3312		errx(EX_DATAERR, "invalid flag %s", e);
3313
3314	cmd->opcode = opcode;
3315	cmd->len =  (cmd->len & (F_NOT | F_OR)) | 1;
3316	cmd->arg1 = (set & 0xff) | ( (clear & 0xff) << 8);
3317}
3318
3319
3320void
3321ipfw_delete(char *av[])
3322{
3323	ipfw_range_tlv rt;
3324	char *sep;
3325	int i, j;
3326	int exitval = EX_OK;
3327	int do_set = 0;
3328
3329	av++;
3330	NEED1("missing rule specification");
3331	if ( *av && _substrcmp(*av, "set") == 0) {
3332		/* Do not allow using the following syntax:
3333		 *	ipfw set N delete set M
3334		 */
3335		if (g_co.use_set)
3336			errx(EX_DATAERR, "invalid syntax");
3337		do_set = 1;	/* delete set */
3338		av++;
3339	}
3340
3341	/* Rule number */
3342	while (*av && isdigit(**av)) {
3343		i = strtol(*av, &sep, 10);
3344		j = i;
3345		if (*sep== '-')
3346			j = strtol(sep + 1, NULL, 10);
3347		av++;
3348		if (g_co.do_nat) {
3349			exitval = ipfw_delete_nat(i);
3350		} else if (g_co.do_pipe) {
3351			exitval = ipfw_delete_pipe(g_co.do_pipe, i);
3352		} else {
3353			memset(&rt, 0, sizeof(rt));
3354			if (do_set != 0) {
3355				rt.set = i & 31;
3356				rt.flags = IPFW_RCFLAG_SET;
3357			} else {
3358				rt.start_rule = i & 0xffff;
3359				rt.end_rule = j & 0xffff;
3360				if (rt.start_rule == 0 && rt.end_rule == 0)
3361					rt.flags |= IPFW_RCFLAG_ALL;
3362				else
3363					rt.flags |= IPFW_RCFLAG_RANGE;
3364				if (g_co.use_set != 0) {
3365					rt.set = g_co.use_set - 1;
3366					rt.flags |= IPFW_RCFLAG_SET;
3367				}
3368			}
3369			if (g_co.do_dynamic == 2)
3370				rt.flags |= IPFW_RCFLAG_DYNAMIC;
3371			i = do_range_cmd(IP_FW_XDEL, &rt);
3372			if (i != 0) {
3373				exitval = EX_UNAVAILABLE;
3374				if (g_co.do_quiet)
3375					continue;
3376				warn("rule %u: setsockopt(IP_FW_XDEL)",
3377				    rt.start_rule);
3378			} else if (rt.new_set == 0 && do_set == 0 &&
3379			    g_co.do_dynamic != 2) {
3380				exitval = EX_UNAVAILABLE;
3381				if (g_co.do_quiet)
3382					continue;
3383				if (rt.start_rule != rt.end_rule)
3384					warnx("no rules rules in %u-%u range",
3385					    rt.start_rule, rt.end_rule);
3386				else
3387					warnx("rule %u not found",
3388					    rt.start_rule);
3389			}
3390		}
3391	}
3392	if (exitval != EX_OK && g_co.do_force == 0)
3393		exit(exitval);
3394}
3395
3396
3397/*
3398 * fill the interface structure. We do not check the name as we can
3399 * create interfaces dynamically, so checking them at insert time
3400 * makes relatively little sense.
3401 * Interface names containing '*', '?', or '[' are assumed to be shell
3402 * patterns which match interfaces.
3403 */
3404static void
3405fill_iface(ipfw_insn_if *cmd, char *arg, int cblen, struct tidx *tstate)
3406{
3407	char *p;
3408	uint16_t uidx;
3409
3410	cmd->name[0] = '\0';
3411	cmd->o.len |= F_INSN_SIZE(ipfw_insn_if);
3412
3413	CHECK_CMDLEN;
3414
3415	/* Parse the interface or address */
3416	if (strcmp(arg, "any") == 0)
3417		cmd->o.len = 0;		/* effectively ignore this command */
3418	else if (strncmp(arg, "table(", 6) == 0) {
3419		if ((p = strchr(arg + 6, ')')) == NULL)
3420			errx(EX_DATAERR, "forgotten parenthesis: '%s'", arg);
3421		*p = '\0';
3422		p = strchr(arg + 6, ',');
3423		if (p)
3424			*p++ = '\0';
3425		if ((uidx = pack_table(tstate, arg + 6)) == 0)
3426			errx(EX_DATAERR, "Invalid table name: %s", arg + 6);
3427
3428		cmd->name[0] = '\1'; /* Special value indicating table */
3429		cmd->p.kidx = uidx;
3430	} else if (!isdigit(*arg)) {
3431		strlcpy(cmd->name, arg, sizeof(cmd->name));
3432		cmd->p.glob = strpbrk(arg, "*?[") != NULL ? 1 : 0;
3433	} else if (!inet_aton(arg, &cmd->p.ip))
3434		errx(EX_DATAERR, "bad ip address ``%s''", arg);
3435}
3436
3437static void
3438get_mac_addr_mask(const char *p, uint8_t *addr, uint8_t *mask)
3439{
3440	int i;
3441	size_t l;
3442	char *ap, *ptr, *optr;
3443	struct ether_addr *mac;
3444	const char *macset = "0123456789abcdefABCDEF:";
3445
3446	if (strcmp(p, "any") == 0) {
3447		for (i = 0; i < ETHER_ADDR_LEN; i++)
3448			addr[i] = mask[i] = 0;
3449		return;
3450	}
3451
3452	optr = ptr = strdup(p);
3453	if ((ap = strsep(&ptr, "&/")) != NULL && *ap != 0) {
3454		l = strlen(ap);
3455		if (strspn(ap, macset) != l || (mac = ether_aton(ap)) == NULL)
3456			errx(EX_DATAERR, "Incorrect MAC address");
3457		bcopy(mac, addr, ETHER_ADDR_LEN);
3458	} else
3459		errx(EX_DATAERR, "Incorrect MAC address");
3460
3461	if (ptr != NULL) { /* we have mask? */
3462		if (p[ptr - optr - 1] == '/') { /* mask len */
3463			long ml = strtol(ptr, &ap, 10);
3464			if (*ap != 0 || ml > ETHER_ADDR_LEN * 8 || ml < 0)
3465				errx(EX_DATAERR, "Incorrect mask length");
3466			for (i = 0; ml > 0 && i < ETHER_ADDR_LEN; ml -= 8, i++)
3467				mask[i] = (ml >= 8) ? 0xff: (~0) << (8 - ml);
3468		} else { /* mask */
3469			l = strlen(ptr);
3470			if (strspn(ptr, macset) != l ||
3471			    (mac = ether_aton(ptr)) == NULL)
3472				errx(EX_DATAERR, "Incorrect mask");
3473			bcopy(mac, mask, ETHER_ADDR_LEN);
3474		}
3475	} else { /* default mask: ff:ff:ff:ff:ff:ff */
3476		for (i = 0; i < ETHER_ADDR_LEN; i++)
3477			mask[i] = 0xff;
3478	}
3479	for (i = 0; i < ETHER_ADDR_LEN; i++)
3480		addr[i] &= mask[i];
3481
3482	free(optr);
3483}
3484
3485/*
3486 * helper function, updates the pointer to cmd with the length
3487 * of the current command, and also cleans up the first word of
3488 * the new command in case it has been clobbered before.
3489 */
3490static ipfw_insn *
3491next_cmd(ipfw_insn *cmd, int *len)
3492{
3493	*len -= F_LEN(cmd);
3494	CHECK_LENGTH(*len, 0);
3495	cmd += F_LEN(cmd);
3496	bzero(cmd, sizeof(*cmd));
3497	return cmd;
3498}
3499
3500/*
3501 * Takes arguments and copies them into a comment
3502 */
3503static void
3504fill_comment(ipfw_insn *cmd, char **av, int cblen)
3505{
3506	int i, l;
3507	char *p = (char *)(cmd + 1);
3508
3509	cmd->opcode = O_NOP;
3510	cmd->len =  (cmd->len & (F_NOT | F_OR));
3511
3512	/* Compute length of comment string. */
3513	for (i = 0, l = 0; av[i] != NULL; i++)
3514		l += strlen(av[i]) + 1;
3515	if (l == 0)
3516		return;
3517	if (l > 84)
3518		errx(EX_DATAERR,
3519		    "comment too long (max 80 chars)");
3520	l = 1 + (l+3)/4;
3521	cmd->len =  (cmd->len & (F_NOT | F_OR)) | l;
3522	CHECK_CMDLEN;
3523
3524	for (i = 0; av[i] != NULL; i++) {
3525		strcpy(p, av[i]);
3526		p += strlen(av[i]);
3527		*p++ = ' ';
3528	}
3529	*(--p) = '\0';
3530}
3531
3532/*
3533 * A function to fill simple commands of size 1.
3534 * Existing flags are preserved.
3535 */
3536static void
3537fill_cmd(ipfw_insn *cmd, enum ipfw_opcodes opcode, int flags, uint16_t arg)
3538{
3539	cmd->opcode = opcode;
3540	cmd->len =  ((cmd->len | flags) & (F_NOT | F_OR)) | 1;
3541	cmd->arg1 = arg;
3542}
3543
3544/*
3545 * Fetch and add the MAC address and type, with masks. This generates one or
3546 * two microinstructions, and returns the pointer to the last one.
3547 */
3548static ipfw_insn *
3549add_mac(ipfw_insn *cmd, char *av[], int cblen)
3550{
3551	ipfw_insn_mac *mac;
3552
3553	if ( ( av[0] == NULL ) || ( av[1] == NULL ) )
3554		errx(EX_DATAERR, "MAC dst src");
3555
3556	cmd->opcode = O_MACADDR2;
3557	cmd->len = (cmd->len & (F_NOT | F_OR)) | F_INSN_SIZE(ipfw_insn_mac);
3558	CHECK_CMDLEN;
3559
3560	mac = (ipfw_insn_mac *)cmd;
3561	get_mac_addr_mask(av[0], mac->addr, mac->mask);	/* dst */
3562	get_mac_addr_mask(av[1], &(mac->addr[ETHER_ADDR_LEN]),
3563	    &(mac->mask[ETHER_ADDR_LEN])); /* src */
3564	return cmd;
3565}
3566
3567static ipfw_insn *
3568add_mactype(ipfw_insn *cmd, char *av, int cblen)
3569{
3570	if (!av)
3571		errx(EX_DATAERR, "missing MAC type");
3572	if (strcmp(av, "any") != 0) { /* we have a non-null type */
3573		fill_newports((ipfw_insn_u16 *)cmd, av, IPPROTO_ETHERTYPE,
3574		    cblen);
3575		cmd->opcode = O_MAC_TYPE;
3576		return cmd;
3577	} else
3578		return NULL;
3579}
3580
3581static ipfw_insn *
3582add_proto0(ipfw_insn *cmd, char *av, u_char *protop)
3583{
3584	struct protoent *pe;
3585	char *ep;
3586	int proto;
3587
3588	proto = strtol(av, &ep, 10);
3589	if (*ep != '\0' || proto <= 0) {
3590		if ((pe = getprotobyname(av)) == NULL)
3591			return NULL;
3592		proto = pe->p_proto;
3593	}
3594
3595	fill_cmd(cmd, O_PROTO, 0, proto);
3596	*protop = proto;
3597	return cmd;
3598}
3599
3600static ipfw_insn *
3601add_proto(ipfw_insn *cmd, char *av, u_char *protop)
3602{
3603	u_char proto = IPPROTO_IP;
3604
3605	if (_substrcmp(av, "all") == 0 || strcmp(av, "ip") == 0)
3606		; /* do not set O_IP4 nor O_IP6 */
3607	else if (strcmp(av, "ip4") == 0)
3608		/* explicit "just IPv4" rule */
3609		fill_cmd(cmd, O_IP4, 0, 0);
3610	else if (strcmp(av, "ip6") == 0) {
3611		/* explicit "just IPv6" rule */
3612		proto = IPPROTO_IPV6;
3613		fill_cmd(cmd, O_IP6, 0, 0);
3614	} else
3615		return add_proto0(cmd, av, protop);
3616
3617	*protop = proto;
3618	return cmd;
3619}
3620
3621static ipfw_insn *
3622add_proto_compat(ipfw_insn *cmd, char *av, u_char *protop)
3623{
3624	u_char proto = IPPROTO_IP;
3625
3626	if (_substrcmp(av, "all") == 0 || strcmp(av, "ip") == 0)
3627		; /* do not set O_IP4 nor O_IP6 */
3628	else if (strcmp(av, "ipv4") == 0 || strcmp(av, "ip4") == 0)
3629		/* explicit "just IPv4" rule */
3630		fill_cmd(cmd, O_IP4, 0, 0);
3631	else if (strcmp(av, "ipv6") == 0 || strcmp(av, "ip6") == 0) {
3632		/* explicit "just IPv6" rule */
3633		proto = IPPROTO_IPV6;
3634		fill_cmd(cmd, O_IP6, 0, 0);
3635	} else
3636		return add_proto0(cmd, av, protop);
3637
3638	*protop = proto;
3639	return cmd;
3640}
3641
3642static ipfw_insn *
3643add_srcip(ipfw_insn *cmd, char *av, int cblen, struct tidx *tstate)
3644{
3645	fill_ip((ipfw_insn_ip *)cmd, av, cblen, tstate);
3646	if (cmd->opcode == O_IP_DST_SET)			/* set */
3647		cmd->opcode = O_IP_SRC_SET;
3648	else if (cmd->opcode == O_IP_DST_LOOKUP)		/* table */
3649		cmd->opcode = O_IP_SRC_LOOKUP;
3650	else if (F_LEN(cmd) == F_INSN_SIZE(ipfw_insn))		/* me */
3651		cmd->opcode = O_IP_SRC_ME;
3652	else if (F_LEN(cmd) == F_INSN_SIZE(ipfw_insn_u32))	/* one IP */
3653		cmd->opcode = O_IP_SRC;
3654	else							/* addr/mask */
3655		cmd->opcode = O_IP_SRC_MASK;
3656	return cmd;
3657}
3658
3659static ipfw_insn *
3660add_dstip(ipfw_insn *cmd, char *av, int cblen, struct tidx *tstate)
3661{
3662	fill_ip((ipfw_insn_ip *)cmd, av, cblen, tstate);
3663	if (cmd->opcode == O_IP_DST_SET)			/* set */
3664		;
3665	else if (cmd->opcode == O_IP_DST_LOOKUP)		/* table */
3666		;
3667	else if (F_LEN(cmd) == F_INSN_SIZE(ipfw_insn))		/* me */
3668		cmd->opcode = O_IP_DST_ME;
3669	else if (F_LEN(cmd) == F_INSN_SIZE(ipfw_insn_u32))	/* one IP */
3670		cmd->opcode = O_IP_DST;
3671	else							/* addr/mask */
3672		cmd->opcode = O_IP_DST_MASK;
3673	return cmd;
3674}
3675
3676static struct _s_x f_reserved_keywords[] = {
3677	{ "altq",	TOK_OR },
3678	{ "//",		TOK_OR },
3679	{ "diverted",	TOK_OR },
3680	{ "dst-port",	TOK_OR },
3681	{ "src-port",	TOK_OR },
3682	{ "established",	TOK_OR },
3683	{ "keep-state",	TOK_OR },
3684	{ "frag",	TOK_OR },
3685	{ "icmptypes",	TOK_OR },
3686	{ "in",		TOK_OR },
3687	{ "out",	TOK_OR },
3688	{ "ip6",	TOK_OR },
3689	{ "any",	TOK_OR },
3690	{ "to",		TOK_OR },
3691	{ "via",	TOK_OR },
3692	{ "{",		TOK_OR },
3693	{ NULL, 0 }	/* terminator */
3694};
3695
3696static ipfw_insn *
3697add_ports(ipfw_insn *cmd, char *av, u_char proto, int opcode, int cblen)
3698{
3699
3700	if (match_token(f_reserved_keywords, av) != -1)
3701		return (NULL);
3702
3703	if (fill_newports((ipfw_insn_u16 *)cmd, av, proto, cblen)) {
3704		/* XXX todo: check that we have a protocol with ports */
3705		cmd->opcode = opcode;
3706		return cmd;
3707	}
3708	return NULL;
3709}
3710
3711static ipfw_insn *
3712add_src(ipfw_insn *cmd, char *av, u_char proto, int cblen, struct tidx *tstate)
3713{
3714	struct in6_addr a;
3715	char *host, *ch, buf[INET6_ADDRSTRLEN];
3716	ipfw_insn *ret = NULL;
3717	size_t len;
3718
3719	/* Copy first address in set if needed */
3720	if ((ch = strpbrk(av, "/,")) != NULL) {
3721		len = ch - av;
3722		strlcpy(buf, av, sizeof(buf));
3723		if (len < sizeof(buf))
3724			buf[len] = '\0';
3725		host = buf;
3726	} else
3727		host = av;
3728
3729	if (proto == IPPROTO_IPV6  || strcmp(av, "me6") == 0 ||
3730	    inet_pton(AF_INET6, host, &a) == 1)
3731		ret = add_srcip6(cmd, av, cblen, tstate);
3732	/* XXX: should check for IPv4, not !IPv6 */
3733	if (ret == NULL && (proto == IPPROTO_IP || strcmp(av, "me") == 0 ||
3734	    inet_pton(AF_INET6, host, &a) != 1))
3735		ret = add_srcip(cmd, av, cblen, tstate);
3736	if (ret == NULL && strcmp(av, "any") != 0)
3737		ret = cmd;
3738
3739	return ret;
3740}
3741
3742static ipfw_insn *
3743add_dst(ipfw_insn *cmd, char *av, u_char proto, int cblen, struct tidx *tstate)
3744{
3745	struct in6_addr a;
3746	char *host, *ch, buf[INET6_ADDRSTRLEN];
3747	ipfw_insn *ret = NULL;
3748	size_t len;
3749
3750	/* Copy first address in set if needed */
3751	if ((ch = strpbrk(av, "/,")) != NULL) {
3752		len = ch - av;
3753		strlcpy(buf, av, sizeof(buf));
3754		if (len < sizeof(buf))
3755			buf[len] = '\0';
3756		host = buf;
3757	} else
3758		host = av;
3759
3760	if (proto == IPPROTO_IPV6  || strcmp(av, "me6") == 0 ||
3761	    inet_pton(AF_INET6, host, &a) == 1)
3762		ret = add_dstip6(cmd, av, cblen, tstate);
3763	/* XXX: should check for IPv4, not !IPv6 */
3764	if (ret == NULL && (proto == IPPROTO_IP || strcmp(av, "me") == 0 ||
3765	    inet_pton(AF_INET6, host, &a) != 1))
3766		ret = add_dstip(cmd, av, cblen, tstate);
3767	if (ret == NULL && strcmp(av, "any") != 0)
3768		ret = cmd;
3769
3770	return ret;
3771}
3772
3773/*
3774 * Parse arguments and assemble the microinstructions which make up a rule.
3775 * Rules are added into the 'rulebuf' and then copied in the correct order
3776 * into the actual rule.
3777 *
3778 * The syntax for a rule starts with the action, followed by
3779 * optional action parameters, and the various match patterns.
3780 * In the assembled microcode, the first opcode must be an O_PROBE_STATE
3781 * (generated if the rule includes a keep-state option), then the
3782 * various match patterns, log/altq actions, and the actual action.
3783 *
3784 */
3785static void
3786compile_rule(char *av[], uint32_t *rbuf, int *rbufsize, struct tidx *tstate)
3787{
3788	/*
3789	 * rules are added into the 'rulebuf' and then copied in
3790	 * the correct order into the actual rule.
3791	 * Some things that need to go out of order (prob, action etc.)
3792	 * go into actbuf[].
3793	 */
3794	static uint32_t actbuf[255], cmdbuf[255];
3795	int rblen, ablen, cblen;
3796
3797	ipfw_insn *src, *dst, *cmd, *action, *prev=NULL;
3798	ipfw_insn *first_cmd;	/* first match pattern */
3799
3800	struct ip_fw_rule *rule;
3801
3802	/*
3803	 * various flags used to record that we entered some fields.
3804	 */
3805	ipfw_insn *have_state = NULL;	/* any state-related option */
3806	int have_rstate = 0;
3807	ipfw_insn *have_log = NULL, *have_altq = NULL, *have_tag = NULL;
3808	ipfw_insn *have_skipcmd = NULL;
3809	size_t len;
3810
3811	int i;
3812
3813	int open_par = 0;	/* open parenthesis ( */
3814
3815	/* proto is here because it is used to fetch ports */
3816	u_char proto = IPPROTO_IP;	/* default protocol */
3817
3818	double match_prob = 1; /* match probability, default is always match */
3819
3820	bzero(actbuf, sizeof(actbuf));		/* actions go here */
3821	bzero(cmdbuf, sizeof(cmdbuf));
3822	bzero(rbuf, *rbufsize);
3823
3824	rule = (struct ip_fw_rule *)rbuf;
3825	cmd = (ipfw_insn *)cmdbuf;
3826	action = (ipfw_insn *)actbuf;
3827
3828	rblen = *rbufsize / sizeof(uint32_t);
3829	rblen -= sizeof(struct ip_fw_rule) / sizeof(uint32_t);
3830	ablen = sizeof(actbuf) / sizeof(actbuf[0]);
3831	cblen = sizeof(cmdbuf) / sizeof(cmdbuf[0]);
3832	cblen -= F_INSN_SIZE(ipfw_insn_u32) + 1;
3833
3834#define	CHECK_RBUFLEN(len)	{ CHECK_LENGTH(rblen, len); rblen -= len; }
3835#define	CHECK_ACTLEN		CHECK_LENGTH(ablen, action->len)
3836
3837	av++;
3838
3839	/* [rule N]	-- Rule number optional */
3840	if (av[0] && isdigit(**av)) {
3841		rule->rulenum = atoi(*av);
3842		av++;
3843	}
3844
3845	/* [set N]	-- set number (0..RESVD_SET), optional */
3846	if (av[0] && av[1] && _substrcmp(*av, "set") == 0) {
3847		int set = strtoul(av[1], NULL, 10);
3848		if (set < 0 || set > RESVD_SET)
3849			errx(EX_DATAERR, "illegal set %s", av[1]);
3850		rule->set = set;
3851		tstate->set = set;
3852		av += 2;
3853	}
3854
3855	/* [prob D]	-- match probability, optional */
3856	if (av[0] && av[1] && _substrcmp(*av, "prob") == 0) {
3857		match_prob = strtod(av[1], NULL);
3858
3859		if (match_prob <= 0 || match_prob > 1)
3860			errx(EX_DATAERR, "illegal match prob. %s", av[1]);
3861		av += 2;
3862	}
3863
3864	/* action	-- mandatory */
3865	NEED1("missing action");
3866	i = match_token(rule_actions, *av);
3867	av++;
3868	action->len = 1;	/* default */
3869	CHECK_ACTLEN;
3870	switch(i) {
3871	case TOK_CHECKSTATE:
3872		have_state = action;
3873		action->opcode = O_CHECK_STATE;
3874		if (*av == NULL ||
3875		    match_token(rule_options, *av) == TOK_COMMENT) {
3876			action->arg1 = pack_object(tstate,
3877			    default_state_name, IPFW_TLV_STATE_NAME);
3878			break;
3879		}
3880		if (*av[0] == ':') {
3881			if (strcmp(*av + 1, "any") == 0)
3882				action->arg1 = 0;
3883			else if (state_check_name(*av + 1) == 0)
3884				action->arg1 = pack_object(tstate, *av + 1,
3885				    IPFW_TLV_STATE_NAME);
3886			else
3887				errx(EX_DATAERR, "Invalid state name %s",
3888				    *av);
3889			av++;
3890			break;
3891		}
3892		errx(EX_DATAERR, "Invalid state name %s", *av);
3893		break;
3894
3895	case TOK_ABORT:
3896		action->opcode = O_REJECT;
3897		action->arg1 = ICMP_REJECT_ABORT;
3898		break;
3899
3900	case TOK_ABORT6:
3901		action->opcode = O_UNREACH6;
3902		action->arg1 = ICMP6_UNREACH_ABORT;
3903		break;
3904
3905	case TOK_ACCEPT:
3906		action->opcode = O_ACCEPT;
3907		break;
3908
3909	case TOK_DENY:
3910		action->opcode = O_DENY;
3911		action->arg1 = 0;
3912		break;
3913
3914	case TOK_REJECT:
3915		action->opcode = O_REJECT;
3916		action->arg1 = ICMP_UNREACH_HOST;
3917		break;
3918
3919	case TOK_RESET:
3920		action->opcode = O_REJECT;
3921		action->arg1 = ICMP_REJECT_RST;
3922		break;
3923
3924	case TOK_RESET6:
3925		action->opcode = O_UNREACH6;
3926		action->arg1 = ICMP6_UNREACH_RST;
3927		break;
3928
3929	case TOK_UNREACH:
3930		action->opcode = O_REJECT;
3931		NEED1("missing reject code");
3932		fill_reject_code(&action->arg1, *av);
3933		av++;
3934		break;
3935
3936	case TOK_UNREACH6:
3937		action->opcode = O_UNREACH6;
3938		NEED1("missing unreach code");
3939		fill_unreach6_code(&action->arg1, *av);
3940		av++;
3941		break;
3942
3943	case TOK_COUNT:
3944		action->opcode = O_COUNT;
3945		break;
3946
3947	case TOK_NAT:
3948		action->opcode = O_NAT;
3949		action->len = F_INSN_SIZE(ipfw_insn_nat);
3950		CHECK_ACTLEN;
3951		if (*av != NULL && _substrcmp(*av, "global") == 0) {
3952			action->arg1 = IP_FW_NAT44_GLOBAL;
3953			av++;
3954			break;
3955		} else
3956			goto chkarg;
3957	case TOK_QUEUE:
3958		action->opcode = O_QUEUE;
3959		goto chkarg;
3960	case TOK_PIPE:
3961		action->opcode = O_PIPE;
3962		goto chkarg;
3963	case TOK_SKIPTO:
3964		action->opcode = O_SKIPTO;
3965		goto chkarg;
3966	case TOK_NETGRAPH:
3967		action->opcode = O_NETGRAPH;
3968		goto chkarg;
3969	case TOK_NGTEE:
3970		action->opcode = O_NGTEE;
3971		goto chkarg;
3972	case TOK_DIVERT:
3973		action->opcode = O_DIVERT;
3974		goto chkarg;
3975	case TOK_TEE:
3976		action->opcode = O_TEE;
3977		goto chkarg;
3978	case TOK_CALL:
3979		action->opcode = O_CALLRETURN;
3980chkarg:
3981		if (!av[0])
3982			errx(EX_USAGE, "missing argument for %s", *(av - 1));
3983		if (isdigit(**av)) {
3984			action->arg1 = strtoul(*av, NULL, 10);
3985			if (action->arg1 <= 0 || action->arg1 >= IP_FW_TABLEARG)
3986				errx(EX_DATAERR, "illegal argument for %s",
3987				    *(av - 1));
3988		} else if (_substrcmp(*av, "tablearg") == 0) {
3989			action->arg1 = IP_FW_TARG;
3990		} else if (i == TOK_DIVERT || i == TOK_TEE) {
3991			struct servent *s;
3992			setservent(1);
3993			s = getservbyname(av[0], "divert");
3994			if (s != NULL)
3995				action->arg1 = ntohs(s->s_port);
3996			else
3997				errx(EX_DATAERR, "illegal divert/tee port");
3998		} else
3999			errx(EX_DATAERR, "illegal argument for %s", *(av - 1));
4000		av++;
4001		break;
4002
4003	case TOK_FORWARD: {
4004		/*
4005		 * Locate the address-port separator (':' or ',').
4006		 * Could be one of the following:
4007		 *	hostname:port
4008		 *	IPv4 a.b.c.d,port
4009		 *	IPv4 a.b.c.d:port
4010		 *	IPv6 w:x:y::z,port
4011		 *	IPv6 [w:x:y::z]:port
4012		 */
4013		struct sockaddr_storage result;
4014		struct addrinfo *res;
4015		char *s, *end;
4016		int family;
4017		u_short port_number = 0;
4018
4019		NEED1("missing forward address[:port]");
4020
4021		if (_substrcmp(*av, "tablearg") == 0) {
4022			family = PF_INET;
4023			((struct sockaddr_in*)&result)->sin_addr.s_addr =
4024			    INADDR_ANY;
4025		} else {
4026			/*
4027			 * Are we an bracket-enclosed IPv6 address?
4028			 */
4029			if (strchr(*av, '['))
4030				(*av)++;
4031
4032			/*
4033			 * locate the address-port separator (':' or ',')
4034			 */
4035			s = strchr(*av, ',');
4036			if (s == NULL) {
4037				s = strchr(*av, ']');
4038				/* Prevent erroneous parsing on brackets. */
4039				if (s != NULL)
4040					*(s++) = '\0';
4041				else
4042					s = *av;
4043
4044				/* Distinguish between IPv4:port and IPv6 cases. */
4045				s = strchr(s, ':');
4046				if (s && strchr(s+1, ':'))
4047					s = NULL; /* no port */
4048			}
4049
4050			if (s != NULL) {
4051				/* Terminate host portion and set s to start of port. */
4052				*(s++) = '\0';
4053				i = strtoport(s, &end, 0 /* base */, 0 /* proto */);
4054				if (s == end)
4055					errx(EX_DATAERR,
4056					    "illegal forwarding port ``%s''", s);
4057				port_number = (u_short)i;
4058			}
4059
4060			/*
4061			 * Resolve the host name or address to a family and a
4062			 * network representation of the address.
4063			 */
4064			if (getaddrinfo(*av, NULL, NULL, &res))
4065				errx(EX_DATAERR, NULL);
4066			/* Just use the first host in the answer. */
4067			family = res->ai_family;
4068			memcpy(&result, res->ai_addr, res->ai_addrlen);
4069			freeaddrinfo(res);
4070		}
4071
4072 		if (family == PF_INET) {
4073			ipfw_insn_sa *p = (ipfw_insn_sa *)action;
4074
4075			action->opcode = O_FORWARD_IP;
4076			action->len = F_INSN_SIZE(ipfw_insn_sa);
4077			CHECK_ACTLEN;
4078
4079			/*
4080			 * In the kernel we assume AF_INET and use only
4081			 * sin_port and sin_addr. Remember to set sin_len as
4082			 * the routing code seems to use it too.
4083			 */
4084			p->sa.sin_len = sizeof(struct sockaddr_in);
4085			p->sa.sin_family = AF_INET;
4086			p->sa.sin_port = port_number;
4087			p->sa.sin_addr.s_addr =
4088			     ((struct sockaddr_in *)&result)->sin_addr.s_addr;
4089		} else if (family == PF_INET6) {
4090			ipfw_insn_sa6 *p = (ipfw_insn_sa6 *)action;
4091
4092			action->opcode = O_FORWARD_IP6;
4093			action->len = F_INSN_SIZE(ipfw_insn_sa6);
4094			CHECK_ACTLEN;
4095
4096			p->sa.sin6_len = sizeof(struct sockaddr_in6);
4097			p->sa.sin6_family = AF_INET6;
4098			p->sa.sin6_port = port_number;
4099			p->sa.sin6_flowinfo = 0;
4100			p->sa.sin6_scope_id =
4101			    ((struct sockaddr_in6 *)&result)->sin6_scope_id;
4102			bcopy(&((struct sockaddr_in6*)&result)->sin6_addr,
4103			    &p->sa.sin6_addr, sizeof(p->sa.sin6_addr));
4104		} else {
4105			errx(EX_DATAERR, "Invalid address family in forward action");
4106		}
4107		av++;
4108		break;
4109	    }
4110	case TOK_COMMENT:
4111		/* pretend it is a 'count' rule followed by the comment */
4112		action->opcode = O_COUNT;
4113		av--;		/* go back... */
4114		break;
4115
4116	case TOK_SETFIB:
4117	    {
4118		int numfibs;
4119		size_t intsize = sizeof(int);
4120
4121		action->opcode = O_SETFIB;
4122		NEED1("missing fib number");
4123		if (_substrcmp(*av, "tablearg") == 0) {
4124			action->arg1 = IP_FW_TARG;
4125		} else {
4126		        action->arg1 = strtoul(*av, NULL, 10);
4127			if (sysctlbyname("net.fibs", &numfibs, &intsize,
4128			    NULL, 0) == -1)
4129				errx(EX_DATAERR, "fibs not suported.\n");
4130			if (action->arg1 >= numfibs)  /* Temporary */
4131				errx(EX_DATAERR, "fib too large.\n");
4132			/* Add high-order bit to fib to make room for tablearg*/
4133			action->arg1 |= 0x8000;
4134		}
4135		av++;
4136		break;
4137	    }
4138
4139	case TOK_SETDSCP:
4140	    {
4141		int code;
4142
4143		action->opcode = O_SETDSCP;
4144		NEED1("missing DSCP code");
4145		if (_substrcmp(*av, "tablearg") == 0) {
4146			action->arg1 = IP_FW_TARG;
4147		} else {
4148			if (isalpha(*av[0])) {
4149				if ((code = match_token(f_ipdscp, *av)) == -1)
4150					errx(EX_DATAERR, "Unknown DSCP code");
4151				action->arg1 = code;
4152			} else
4153			        action->arg1 = strtoul(*av, NULL, 10);
4154			/*
4155			 * Add high-order bit to DSCP to make room
4156			 * for tablearg
4157			 */
4158			action->arg1 |= 0x8000;
4159		}
4160		av++;
4161		break;
4162	    }
4163
4164	case TOK_REASS:
4165		action->opcode = O_REASS;
4166		break;
4167
4168	case TOK_RETURN:
4169		fill_cmd(action, O_CALLRETURN, F_NOT, 0);
4170		break;
4171
4172	case TOK_TCPSETMSS: {
4173		u_long mss;
4174		uint16_t idx;
4175
4176		idx = pack_object(tstate, "tcp-setmss", IPFW_TLV_EACTION);
4177		if (idx == 0)
4178			errx(EX_DATAERR, "pack_object failed");
4179		fill_cmd(action, O_EXTERNAL_ACTION, 0, idx);
4180		NEED1("Missing MSS value");
4181		action = next_cmd(action, &ablen);
4182		action->len = 1;
4183		CHECK_ACTLEN;
4184		mss = strtoul(*av, NULL, 10);
4185		if (mss == 0 || mss > UINT16_MAX)
4186			errx(EX_USAGE, "invalid MSS value %s", *av);
4187		fill_cmd(action, O_EXTERNAL_DATA, 0, (uint16_t)mss);
4188		av++;
4189		break;
4190	}
4191
4192	default:
4193		av--;
4194		if (match_token(rule_eactions, *av) == -1)
4195			errx(EX_DATAERR, "invalid action %s\n", *av);
4196		/*
4197		 * External actions support.
4198		 * XXX: we support only syntax with instance name.
4199		 *	For known external actions (from rule_eactions list)
4200		 *	we can handle syntax directly. But with `eaction'
4201		 *	keyword we can use only `eaction <name> <instance>'
4202		 *	syntax.
4203		 */
4204	case TOK_EACTION: {
4205		uint16_t idx;
4206
4207		NEED1("Missing eaction name");
4208		if (eaction_check_name(*av) != 0)
4209			errx(EX_DATAERR, "Invalid eaction name %s", *av);
4210		idx = pack_object(tstate, *av, IPFW_TLV_EACTION);
4211		if (idx == 0)
4212			errx(EX_DATAERR, "pack_object failed");
4213		fill_cmd(action, O_EXTERNAL_ACTION, 0, idx);
4214		av++;
4215		NEED1("Missing eaction instance name");
4216		action = next_cmd(action, &ablen);
4217		action->len = 1;
4218		CHECK_ACTLEN;
4219		if (eaction_check_name(*av) != 0)
4220			errx(EX_DATAERR, "Invalid eaction instance name %s",
4221			    *av);
4222		/*
4223		 * External action instance object has TLV type depended
4224		 * from the external action name object index. Since we
4225		 * currently don't know this index, use zero as TLV type.
4226		 */
4227		idx = pack_object(tstate, *av, 0);
4228		if (idx == 0)
4229			errx(EX_DATAERR, "pack_object failed");
4230		fill_cmd(action, O_EXTERNAL_INSTANCE, 0, idx);
4231		av++;
4232		}
4233	}
4234	action = next_cmd(action, &ablen);
4235
4236	/*
4237	 * [altq queuename] -- altq tag, optional
4238	 * [log [logamount N]]	-- log, optional
4239	 *
4240	 * If they exist, it go first in the cmdbuf, but then it is
4241	 * skipped in the copy section to the end of the buffer.
4242	 */
4243	while (av[0] != NULL && (i = match_token(rule_action_params, *av)) != -1) {
4244		av++;
4245		switch (i) {
4246		case TOK_LOG:
4247		    {
4248			ipfw_insn_log *c = (ipfw_insn_log *)cmd;
4249			int l;
4250
4251			if (have_log)
4252				errx(EX_DATAERR,
4253				    "log cannot be specified more than once");
4254			have_log = (ipfw_insn *)c;
4255			cmd->len = F_INSN_SIZE(ipfw_insn_log);
4256			CHECK_CMDLEN;
4257			cmd->opcode = O_LOG;
4258			if (av[0] && _substrcmp(*av, "logamount") == 0) {
4259				av++;
4260				NEED1("logamount requires argument");
4261				l = atoi(*av);
4262				if (l < 0)
4263					errx(EX_DATAERR,
4264					    "logamount must be positive");
4265				c->max_log = l;
4266				av++;
4267			} else {
4268				len = sizeof(c->max_log);
4269				if (sysctlbyname("net.inet.ip.fw.verbose_limit",
4270				    &c->max_log, &len, NULL, 0) == -1) {
4271					if (g_co.test_only) {
4272						c->max_log = 0;
4273						break;
4274					}
4275					errx(1, "sysctlbyname(\"%s\")",
4276					    "net.inet.ip.fw.verbose_limit");
4277				}
4278			}
4279		    }
4280			break;
4281
4282#ifndef NO_ALTQ
4283		case TOK_ALTQ:
4284		    {
4285			ipfw_insn_altq *a = (ipfw_insn_altq *)cmd;
4286
4287			NEED1("missing altq queue name");
4288			if (have_altq)
4289				errx(EX_DATAERR,
4290				    "altq cannot be specified more than once");
4291			have_altq = (ipfw_insn *)a;
4292			cmd->len = F_INSN_SIZE(ipfw_insn_altq);
4293			CHECK_CMDLEN;
4294			cmd->opcode = O_ALTQ;
4295			a->qid = altq_name_to_qid(*av);
4296			av++;
4297		    }
4298			break;
4299#endif
4300
4301		case TOK_TAG:
4302		case TOK_UNTAG: {
4303			uint16_t tag;
4304
4305			if (have_tag)
4306				errx(EX_USAGE, "tag and untag cannot be "
4307				    "specified more than once");
4308			GET_UINT_ARG(tag, IPFW_ARG_MIN, IPFW_ARG_MAX, i,
4309			   rule_action_params);
4310			have_tag = cmd;
4311			fill_cmd(cmd, O_TAG, (i == TOK_TAG) ? 0: F_NOT, tag);
4312			av++;
4313			break;
4314		}
4315
4316		default:
4317			abort();
4318		}
4319		cmd = next_cmd(cmd, &cblen);
4320	}
4321
4322	if (have_state)	{ /* must be a check-state, we are done */
4323		if (*av != NULL &&
4324		    match_token(rule_options, *av) == TOK_COMMENT) {
4325			/* check-state has a comment */
4326			av++;
4327			fill_comment(cmd, av, cblen);
4328			cmd = next_cmd(cmd, &cblen);
4329			av[0] = NULL;
4330		}
4331		goto done;
4332	}
4333
4334#define OR_START(target)					\
4335	if (av[0] && (*av[0] == '(' || *av[0] == '{')) { 	\
4336		if (open_par)					\
4337			errx(EX_USAGE, "nested \"(\" not allowed\n"); \
4338		prev = NULL;					\
4339		open_par = 1;					\
4340		if ( (av[0])[1] == '\0') {			\
4341			av++;					\
4342		} else						\
4343			(*av)++;				\
4344	}							\
4345	target:							\
4346
4347
4348#define	CLOSE_PAR						\
4349	if (open_par) {						\
4350		if (av[0] && (					\
4351		    strcmp(*av, ")") == 0 ||			\
4352		    strcmp(*av, "}") == 0)) {			\
4353			prev = NULL;				\
4354			open_par = 0;				\
4355			av++;					\
4356		} else						\
4357			errx(EX_USAGE, "missing \")\"\n");	\
4358	}
4359
4360#define NOT_BLOCK						\
4361	if (av[0] && _substrcmp(*av, "not") == 0) {		\
4362		if (cmd->len & F_NOT)				\
4363			errx(EX_USAGE, "double \"not\" not allowed\n"); \
4364		cmd->len |= F_NOT;				\
4365		av++;						\
4366	}
4367
4368#define OR_BLOCK(target)					\
4369	if (av[0] && _substrcmp(*av, "or") == 0) {		\
4370		if (prev == NULL || open_par == 0)		\
4371			errx(EX_DATAERR, "invalid OR block");	\
4372		prev->len |= F_OR;				\
4373		av++;					\
4374		goto target;					\
4375	}							\
4376	CLOSE_PAR;
4377
4378	first_cmd = cmd;
4379
4380#if 0
4381	/*
4382	 * MAC addresses, optional.
4383	 * If we have this, we skip the part "proto from src to dst"
4384	 * and jump straight to the option parsing.
4385	 */
4386	NOT_BLOCK;
4387	NEED1("missing protocol");
4388	if (_substrcmp(*av, "MAC") == 0 ||
4389	    _substrcmp(*av, "mac") == 0) {
4390		av++;			/* the "MAC" keyword */
4391		add_mac(cmd, av);	/* exits in case of errors */
4392		cmd = next_cmd(cmd);
4393		av += 2;		/* dst-mac and src-mac */
4394		NOT_BLOCK;
4395		NEED1("missing mac type");
4396		if (add_mactype(cmd, av[0]))
4397			cmd = next_cmd(cmd);
4398		av++;			/* any or mac-type */
4399		goto read_options;
4400	}
4401#endif
4402
4403	/*
4404	 * protocol, mandatory
4405	 */
4406    OR_START(get_proto);
4407	NOT_BLOCK;
4408	NEED1("missing protocol");
4409	if (add_proto_compat(cmd, *av, &proto)) {
4410		av++;
4411		if (F_LEN(cmd) != 0) {
4412			prev = cmd;
4413			cmd = next_cmd(cmd, &cblen);
4414		}
4415	} else if (first_cmd != cmd) {
4416		errx(EX_DATAERR, "invalid protocol ``%s''", *av);
4417	} else {
4418		rule->flags |= IPFW_RULE_JUSTOPTS;
4419		goto read_options;
4420	}
4421    OR_BLOCK(get_proto);
4422
4423	first_cmd = cmd; /* update pointer to use in compact form */
4424
4425	/*
4426	 * "from", mandatory
4427	 */
4428	if ((av[0] == NULL) || _substrcmp(*av, "from") != 0)
4429		errx(EX_USAGE, "missing ``from''");
4430	av++;
4431
4432	/*
4433	 * source IP, mandatory
4434	 */
4435    OR_START(source_ip);
4436	NOT_BLOCK;	/* optional "not" */
4437	NEED1("missing source address");
4438	if (add_src(cmd, *av, proto, cblen, tstate)) {
4439		av++;
4440		if (F_LEN(cmd) != 0) {	/* ! any */
4441			prev = cmd;
4442			cmd = next_cmd(cmd, &cblen);
4443		}
4444	} else
4445		errx(EX_USAGE, "bad source address %s", *av);
4446    OR_BLOCK(source_ip);
4447
4448	/*
4449	 * source ports, optional
4450	 */
4451	NOT_BLOCK;	/* optional "not" */
4452	if ( av[0] != NULL ) {
4453		if (_substrcmp(*av, "any") == 0 ||
4454		    add_ports(cmd, *av, proto, O_IP_SRCPORT, cblen)) {
4455			av++;
4456			if (F_LEN(cmd) != 0)
4457				cmd = next_cmd(cmd, &cblen);
4458		}
4459	}
4460
4461	/*
4462	 * "to", mandatory
4463	 */
4464	if ( (av[0] == NULL) || _substrcmp(*av, "to") != 0 )
4465		errx(EX_USAGE, "missing ``to''");
4466	av++;
4467
4468	/*
4469	 * destination, mandatory
4470	 */
4471    OR_START(dest_ip);
4472	NOT_BLOCK;	/* optional "not" */
4473	NEED1("missing dst address");
4474	if (add_dst(cmd, *av, proto, cblen, tstate)) {
4475		av++;
4476		if (F_LEN(cmd) != 0) {	/* ! any */
4477			prev = cmd;
4478			cmd = next_cmd(cmd, &cblen);
4479		}
4480	} else
4481		errx( EX_USAGE, "bad destination address %s", *av);
4482    OR_BLOCK(dest_ip);
4483
4484	/*
4485	 * dest. ports, optional
4486	 */
4487	NOT_BLOCK;	/* optional "not" */
4488	if (av[0]) {
4489		if (_substrcmp(*av, "any") == 0 ||
4490		    add_ports(cmd, *av, proto, O_IP_DSTPORT, cblen)) {
4491			av++;
4492			if (F_LEN(cmd) != 0)
4493				cmd = next_cmd(cmd, &cblen);
4494		}
4495	}
4496	if (first_cmd == cmd)
4497		rule->flags |= IPFW_RULE_NOOPT;
4498
4499read_options:
4500	prev = NULL;
4501	while ( av[0] != NULL ) {
4502		char *s;
4503		ipfw_insn_u32 *cmd32;	/* alias for cmd */
4504
4505		s = *av;
4506		cmd32 = (ipfw_insn_u32 *)cmd;
4507
4508		if (*s == '!') {	/* alternate syntax for NOT */
4509			if (cmd->len & F_NOT)
4510				errx(EX_USAGE, "double \"not\" not allowed\n");
4511			cmd->len = F_NOT;
4512			s++;
4513		}
4514		i = match_token(rule_options, s);
4515		av++;
4516		switch(i) {
4517		case TOK_NOT:
4518			if (cmd->len & F_NOT)
4519				errx(EX_USAGE, "double \"not\" not allowed\n");
4520			cmd->len = F_NOT;
4521			break;
4522
4523		case TOK_OR:
4524			if (open_par == 0 || prev == NULL)
4525				errx(EX_USAGE, "invalid \"or\" block\n");
4526			prev->len |= F_OR;
4527			break;
4528
4529		case TOK_STARTBRACE:
4530			if (open_par)
4531				errx(EX_USAGE, "+nested \"(\" not allowed\n");
4532			open_par = 1;
4533			break;
4534
4535		case TOK_ENDBRACE:
4536			if (!open_par)
4537				errx(EX_USAGE, "+missing \")\"\n");
4538			open_par = 0;
4539			prev = NULL;
4540			break;
4541
4542		case TOK_IN:
4543			fill_cmd(cmd, O_IN, 0, 0);
4544			break;
4545
4546		case TOK_OUT:
4547			cmd->len ^= F_NOT; /* toggle F_NOT */
4548			fill_cmd(cmd, O_IN, 0, 0);
4549			break;
4550
4551		case TOK_DIVERTED:
4552			fill_cmd(cmd, O_DIVERTED, 0, 3);
4553			break;
4554
4555		case TOK_DIVERTEDLOOPBACK:
4556			fill_cmd(cmd, O_DIVERTED, 0, 1);
4557			break;
4558
4559		case TOK_DIVERTEDOUTPUT:
4560			fill_cmd(cmd, O_DIVERTED, 0, 2);
4561			break;
4562
4563		case TOK_FRAG: {
4564			uint32_t set = 0, clear = 0;
4565
4566			if (*av != NULL && fill_flags(f_ipoff, *av, NULL,
4567			    &set, &clear) == 0)
4568				av++;
4569			else {
4570				/*
4571				 * Compatibility: no argument after "frag"
4572				 * keyword equals to "frag offset".
4573				 */
4574				set = 0x01;
4575				clear = 0;
4576			}
4577			fill_cmd(cmd, O_FRAG, 0,
4578			    (set & 0xff) | ( (clear & 0xff) << 8));
4579			break;
4580		}
4581
4582		case TOK_LAYER2:
4583			fill_cmd(cmd, O_LAYER2, 0, 0);
4584			break;
4585
4586		case TOK_XMIT:
4587		case TOK_RECV:
4588		case TOK_VIA:
4589			NEED1("recv, xmit, via require interface name"
4590				" or address");
4591			fill_iface((ipfw_insn_if *)cmd, av[0], cblen, tstate);
4592			av++;
4593			if (F_LEN(cmd) == 0)	/* not a valid address */
4594				break;
4595			if (i == TOK_XMIT)
4596				cmd->opcode = O_XMIT;
4597			else if (i == TOK_RECV)
4598				cmd->opcode = O_RECV;
4599			else if (i == TOK_VIA)
4600				cmd->opcode = O_VIA;
4601			break;
4602
4603		case TOK_ICMPTYPES:
4604			NEED1("icmptypes requires list of types");
4605			fill_icmptypes((ipfw_insn_u32 *)cmd, *av);
4606			av++;
4607			break;
4608
4609		case TOK_ICMP6TYPES:
4610			NEED1("icmptypes requires list of types");
4611			fill_icmp6types((ipfw_insn_icmp6 *)cmd, *av, cblen);
4612			av++;
4613			break;
4614
4615		case TOK_IPTTL:
4616			NEED1("ipttl requires TTL");
4617			if (strpbrk(*av, "-,")) {
4618			    if (!add_ports(cmd, *av, 0, O_IPTTL, cblen))
4619				errx(EX_DATAERR, "invalid ipttl %s", *av);
4620			} else
4621			    fill_cmd(cmd, O_IPTTL, 0, strtoul(*av, NULL, 0));
4622			av++;
4623			break;
4624
4625		case TOK_IPID:
4626			NEED1("ipid requires id");
4627			if (strpbrk(*av, "-,")) {
4628			    if (!add_ports(cmd, *av, 0, O_IPID, cblen))
4629				errx(EX_DATAERR, "invalid ipid %s", *av);
4630			} else
4631			    fill_cmd(cmd, O_IPID, 0, strtoul(*av, NULL, 0));
4632			av++;
4633			break;
4634
4635		case TOK_IPLEN:
4636			NEED1("iplen requires length");
4637			if (strpbrk(*av, "-,")) {
4638			    if (!add_ports(cmd, *av, 0, O_IPLEN, cblen))
4639				errx(EX_DATAERR, "invalid ip len %s", *av);
4640			} else
4641			    fill_cmd(cmd, O_IPLEN, 0, strtoul(*av, NULL, 0));
4642			av++;
4643			break;
4644
4645		case TOK_IPVER:
4646			NEED1("ipver requires version");
4647			fill_cmd(cmd, O_IPVER, 0, strtoul(*av, NULL, 0));
4648			av++;
4649			break;
4650
4651		case TOK_IPPRECEDENCE:
4652			NEED1("ipprecedence requires value");
4653			fill_cmd(cmd, O_IPPRECEDENCE, 0,
4654			    (strtoul(*av, NULL, 0) & 7) << 5);
4655			av++;
4656			break;
4657
4658		case TOK_DSCP:
4659			NEED1("missing DSCP code");
4660			fill_dscp(cmd, *av, cblen);
4661			av++;
4662			break;
4663
4664		case TOK_IPOPTS:
4665			NEED1("missing argument for ipoptions");
4666			fill_flags_cmd(cmd, O_IPOPT, f_ipopts, *av);
4667			av++;
4668			break;
4669
4670		case TOK_IPTOS:
4671			NEED1("missing argument for iptos");
4672			fill_flags_cmd(cmd, O_IPTOS, f_iptos, *av);
4673			av++;
4674			break;
4675
4676		case TOK_UID:
4677			NEED1("uid requires argument");
4678		    {
4679			char *end;
4680			uid_t uid;
4681			struct passwd *pwd;
4682
4683			cmd->opcode = O_UID;
4684			uid = strtoul(*av, &end, 0);
4685			pwd = (*end == '\0') ? getpwuid(uid) : getpwnam(*av);
4686			if (pwd == NULL)
4687				errx(EX_DATAERR, "uid \"%s\" nonexistent", *av);
4688			cmd32->d[0] = pwd->pw_uid;
4689			cmd->len |= F_INSN_SIZE(ipfw_insn_u32);
4690			av++;
4691		    }
4692			break;
4693
4694		case TOK_GID:
4695			NEED1("gid requires argument");
4696		    {
4697			char *end;
4698			gid_t gid;
4699			struct group *grp;
4700
4701			cmd->opcode = O_GID;
4702			gid = strtoul(*av, &end, 0);
4703			grp = (*end == '\0') ? getgrgid(gid) : getgrnam(*av);
4704			if (grp == NULL)
4705				errx(EX_DATAERR, "gid \"%s\" nonexistent", *av);
4706			cmd32->d[0] = grp->gr_gid;
4707			cmd->len |= F_INSN_SIZE(ipfw_insn_u32);
4708			av++;
4709		    }
4710			break;
4711
4712		case TOK_JAIL:
4713			NEED1("jail requires argument");
4714		    {
4715			char *end;
4716			int jid;
4717
4718			cmd->opcode = O_JAIL;
4719			/*
4720			 * If av is a number, then we'll just pass it as-is.  If
4721			 * it's a name, try to resolve that to a jid.
4722			 *
4723			 * We save the jail_getid(3) call for a fallback because
4724			 * it entails an unconditional trip to the kernel to
4725			 * either validate a jid or resolve a name to a jid.
4726			 * This specific token doesn't currently require a
4727			 * jid to be an active jail, so we save a transition
4728			 * by simply using a number that we're given.
4729			 */
4730			jid = strtoul(*av, &end, 10);
4731			if (*end != '\0') {
4732				jid = jail_getid(*av);
4733				if (jid < 0)
4734				    errx(EX_DATAERR, "%s", jail_errmsg);
4735			}
4736			cmd32->d[0] = (uint32_t)jid;
4737			cmd->len |= F_INSN_SIZE(ipfw_insn_u32);
4738			av++;
4739		    }
4740			break;
4741
4742		case TOK_ESTAB:
4743			fill_cmd(cmd, O_ESTAB, 0, 0);
4744			break;
4745
4746		case TOK_SETUP:
4747			fill_cmd(cmd, O_TCPFLAGS, 0,
4748				(TH_SYN) | ( (TH_ACK) & 0xff) <<8 );
4749			break;
4750
4751		case TOK_TCPDATALEN:
4752			NEED1("tcpdatalen requires length");
4753			if (strpbrk(*av, "-,")) {
4754			    if (!add_ports(cmd, *av, 0, O_TCPDATALEN, cblen))
4755				errx(EX_DATAERR, "invalid tcpdata len %s", *av);
4756			} else
4757			    fill_cmd(cmd, O_TCPDATALEN, 0,
4758				    strtoul(*av, NULL, 0));
4759			av++;
4760			break;
4761
4762		case TOK_TCPOPTS:
4763			NEED1("missing argument for tcpoptions");
4764			fill_flags_cmd(cmd, O_TCPOPTS, f_tcpopts, *av);
4765			av++;
4766			break;
4767
4768		case TOK_TCPSEQ:
4769		case TOK_TCPACK:
4770			NEED1("tcpseq/tcpack requires argument");
4771			cmd->len = F_INSN_SIZE(ipfw_insn_u32);
4772			cmd->opcode = (i == TOK_TCPSEQ) ? O_TCPSEQ : O_TCPACK;
4773			cmd32->d[0] = htonl(strtoul(*av, NULL, 0));
4774			av++;
4775			break;
4776
4777		case TOK_TCPMSS:
4778		case TOK_TCPWIN:
4779			NEED1("tcpmss/tcpwin requires size");
4780			if (strpbrk(*av, "-,")) {
4781				if (add_ports(cmd, *av, 0,
4782				    i == TOK_TCPWIN ? O_TCPWIN : O_TCPMSS,
4783				    cblen) == NULL)
4784					errx(EX_DATAERR, "invalid %s size %s",
4785					    s, *av);
4786			} else
4787				fill_cmd(cmd, i == TOK_TCPWIN ? O_TCPWIN :
4788				    O_TCPMSS, 0, strtoul(*av, NULL, 0));
4789			av++;
4790			break;
4791
4792		case TOK_TCPFLAGS:
4793			NEED1("missing argument for tcpflags");
4794			cmd->opcode = O_TCPFLAGS;
4795			fill_flags_cmd(cmd, O_TCPFLAGS, f_tcpflags, *av);
4796			av++;
4797			break;
4798
4799		case TOK_KEEPSTATE:
4800		case TOK_RECORDSTATE: {
4801			uint16_t uidx;
4802
4803			if (open_par)
4804				errx(EX_USAGE, "keep-state or record-state cannot be part "
4805				    "of an or block");
4806			if (have_state)
4807				errx(EX_USAGE, "only one of keep-state, record-state, "
4808					" limit and set-limit is allowed");
4809			if (*av != NULL && *av[0] == ':') {
4810				if (state_check_name(*av + 1) != 0)
4811					errx(EX_DATAERR,
4812					    "Invalid state name %s", *av);
4813				uidx = pack_object(tstate, *av + 1,
4814				    IPFW_TLV_STATE_NAME);
4815				av++;
4816			} else
4817				uidx = pack_object(tstate, default_state_name,
4818				    IPFW_TLV_STATE_NAME);
4819			have_state = cmd;
4820			have_rstate = i == TOK_RECORDSTATE;
4821			fill_cmd(cmd, O_KEEP_STATE, 0, uidx);
4822			break;
4823		}
4824
4825		case TOK_LIMIT:
4826		case TOK_SETLIMIT: {
4827			ipfw_insn_limit *c = (ipfw_insn_limit *)cmd;
4828			int val;
4829
4830			if (open_par)
4831				errx(EX_USAGE,
4832				    "limit or set-limit cannot be part of an or block");
4833			if (have_state)
4834				errx(EX_USAGE, "only one of keep-state, record-state, "
4835					" limit and set-limit is allowed");
4836			have_state = cmd;
4837			have_rstate = i == TOK_SETLIMIT;
4838
4839			cmd->len = F_INSN_SIZE(ipfw_insn_limit);
4840			CHECK_CMDLEN;
4841			cmd->opcode = O_LIMIT;
4842			c->limit_mask = c->conn_limit = 0;
4843
4844			while ( av[0] != NULL ) {
4845				if ((val = match_token(limit_masks, *av)) <= 0)
4846					break;
4847				c->limit_mask |= val;
4848				av++;
4849			}
4850
4851			if (c->limit_mask == 0)
4852				errx(EX_USAGE, "limit: missing limit mask");
4853
4854			GET_UINT_ARG(c->conn_limit, IPFW_ARG_MIN, IPFW_ARG_MAX,
4855			    TOK_LIMIT, rule_options);
4856			av++;
4857
4858			if (*av != NULL && *av[0] == ':') {
4859				if (state_check_name(*av + 1) != 0)
4860					errx(EX_DATAERR,
4861					    "Invalid state name %s", *av);
4862				cmd->arg1 = pack_object(tstate, *av + 1,
4863				    IPFW_TLV_STATE_NAME);
4864				av++;
4865			} else
4866				cmd->arg1 = pack_object(tstate,
4867				    default_state_name, IPFW_TLV_STATE_NAME);
4868			break;
4869		}
4870
4871		case TOK_PROTO:
4872			NEED1("missing protocol");
4873			if (add_proto(cmd, *av, &proto)) {
4874				av++;
4875			} else
4876				errx(EX_DATAERR, "invalid protocol ``%s''",
4877				    *av);
4878			break;
4879
4880		case TOK_SRCIP:
4881			NEED1("missing source IP");
4882			if (add_srcip(cmd, *av, cblen, tstate)) {
4883				av++;
4884			}
4885			break;
4886
4887		case TOK_DSTIP:
4888			NEED1("missing destination IP");
4889			if (add_dstip(cmd, *av, cblen, tstate)) {
4890				av++;
4891			}
4892			break;
4893
4894		case TOK_SRCIP6:
4895			NEED1("missing source IP6");
4896			if (add_srcip6(cmd, *av, cblen, tstate)) {
4897				av++;
4898			}
4899			break;
4900
4901		case TOK_DSTIP6:
4902			NEED1("missing destination IP6");
4903			if (add_dstip6(cmd, *av, cblen, tstate)) {
4904				av++;
4905			}
4906			break;
4907
4908		case TOK_SRCPORT:
4909			NEED1("missing source port");
4910			if (_substrcmp(*av, "any") == 0 ||
4911			    add_ports(cmd, *av, proto, O_IP_SRCPORT, cblen)) {
4912				av++;
4913			} else
4914				errx(EX_DATAERR, "invalid source port %s", *av);
4915			break;
4916
4917		case TOK_DSTPORT:
4918			NEED1("missing destination port");
4919			if (_substrcmp(*av, "any") == 0 ||
4920			    add_ports(cmd, *av, proto, O_IP_DSTPORT, cblen)) {
4921				av++;
4922			} else
4923				errx(EX_DATAERR, "invalid destination port %s",
4924				    *av);
4925			break;
4926
4927		case TOK_MAC:
4928			if (add_mac(cmd, av, cblen))
4929				av += 2;
4930			break;
4931
4932		case TOK_MACTYPE:
4933			NEED1("missing mac type");
4934			if (!add_mactype(cmd, *av, cblen))
4935				errx(EX_DATAERR, "invalid mac type %s", *av);
4936			av++;
4937			break;
4938
4939		case TOK_VERREVPATH:
4940			fill_cmd(cmd, O_VERREVPATH, 0, 0);
4941			break;
4942
4943		case TOK_VERSRCREACH:
4944			fill_cmd(cmd, O_VERSRCREACH, 0, 0);
4945			break;
4946
4947		case TOK_ANTISPOOF:
4948			fill_cmd(cmd, O_ANTISPOOF, 0, 0);
4949			break;
4950
4951		case TOK_IPSEC:
4952			fill_cmd(cmd, O_IPSEC, 0, 0);
4953			break;
4954
4955		case TOK_IPV6:
4956			fill_cmd(cmd, O_IP6, 0, 0);
4957			break;
4958
4959		case TOK_IPV4:
4960			fill_cmd(cmd, O_IP4, 0, 0);
4961			break;
4962
4963		case TOK_EXT6HDR:
4964			fill_ext6hdr( cmd, *av );
4965			av++;
4966			break;
4967
4968		case TOK_FLOWID:
4969			if (proto != IPPROTO_IPV6 )
4970				errx( EX_USAGE, "flow-id filter is active "
4971				    "only for ipv6 protocol\n");
4972			fill_flow6( (ipfw_insn_u32 *) cmd, *av, cblen);
4973			av++;
4974			break;
4975
4976		case TOK_COMMENT:
4977			fill_comment(cmd, av, cblen);
4978			av[0]=NULL;
4979			break;
4980
4981		case TOK_TAGGED:
4982			if (av[0] && strpbrk(*av, "-,")) {
4983				if (!add_ports(cmd, *av, 0, O_TAGGED, cblen))
4984					errx(EX_DATAERR, "tagged: invalid tag"
4985					    " list: %s", *av);
4986			}
4987			else {
4988				uint16_t tag;
4989
4990				GET_UINT_ARG(tag, IPFW_ARG_MIN, IPFW_ARG_MAX,
4991				    TOK_TAGGED, rule_options);
4992				fill_cmd(cmd, O_TAGGED, 0, tag);
4993			}
4994			av++;
4995			break;
4996
4997		case TOK_FIB:
4998			NEED1("fib requires fib number");
4999			fill_cmd(cmd, O_FIB, 0, strtoul(*av, NULL, 0));
5000			av++;
5001			break;
5002		case TOK_SOCKARG:
5003			fill_cmd(cmd, O_SOCKARG, 0, 0);
5004			break;
5005
5006		case TOK_LOOKUP: {
5007			ipfw_insn_u32 *c = (ipfw_insn_u32 *)cmd;
5008			int j;
5009
5010			if (!av[0] || !av[1])
5011				errx(EX_USAGE, "format: lookup argument tablenum");
5012			cmd->opcode = O_IP_DST_LOOKUP;
5013			cmd->len |= F_INSN_SIZE(ipfw_insn) + 2;
5014			i = match_token(rule_options, *av);
5015			for (j = 0; lookup_key[j] >= 0 ; j++) {
5016				if (i == lookup_key[j])
5017					break;
5018			}
5019			if (lookup_key[j] <= 0)
5020				errx(EX_USAGE, "format: cannot lookup on %s", *av);
5021			__PAST_END(c->d, 1) = j; // i converted to option
5022			av++;
5023
5024			if ((j = pack_table(tstate, *av)) == 0)
5025				errx(EX_DATAERR, "Invalid table name: %s", *av);
5026
5027			cmd->arg1 = j;
5028			av++;
5029		    }
5030			break;
5031		case TOK_FLOW:
5032			NEED1("missing table name");
5033			if (strncmp(*av, "table(", 6) != 0)
5034				errx(EX_DATAERR,
5035				    "enclose table name into \"table()\"");
5036			fill_table(cmd, *av, O_IP_FLOW_LOOKUP, tstate);
5037			av++;
5038			break;
5039
5040		case TOK_SKIPACTION:
5041			if (have_skipcmd)
5042				errx(EX_USAGE, "only one defer-action "
5043					"is allowed");
5044			have_skipcmd = cmd;
5045			fill_cmd(cmd, O_SKIP_ACTION, 0, 0);
5046			break;
5047
5048		default:
5049			errx(EX_USAGE, "unrecognised option [%d] %s\n", i, s);
5050		}
5051		if (F_LEN(cmd) > 0) {	/* prepare to advance */
5052			prev = cmd;
5053			cmd = next_cmd(cmd, &cblen);
5054		}
5055	}
5056
5057done:
5058
5059	if (!have_state && have_skipcmd)
5060		warnx("Rule contains \"defer-immediate-action\" "
5061			"and doesn't contain any state-related options.");
5062
5063	/*
5064	 * Now copy stuff into the rule.
5065	 * If we have a keep-state option, the first instruction
5066	 * must be a PROBE_STATE (which is generated here).
5067	 * If we have a LOG option, it was stored as the first command,
5068	 * and now must be moved to the top of the action part.
5069	 */
5070	dst = (ipfw_insn *)rule->cmd;
5071
5072	/*
5073	 * First thing to write into the command stream is the match probability.
5074	 */
5075	if (match_prob != 1) { /* 1 means always match */
5076		dst->opcode = O_PROB;
5077		dst->len = 2;
5078		*((int32_t *)(dst+1)) = (int32_t)(match_prob * 0x7fffffff);
5079		dst += dst->len;
5080	}
5081
5082	/*
5083	 * generate O_PROBE_STATE if necessary
5084	 */
5085	if (have_state && have_state->opcode != O_CHECK_STATE && !have_rstate) {
5086		fill_cmd(dst, O_PROBE_STATE, 0, have_state->arg1);
5087		dst = next_cmd(dst, &rblen);
5088	}
5089
5090	/*
5091	 * copy all commands but O_LOG, O_KEEP_STATE, O_LIMIT, O_ALTQ, O_TAG,
5092	 * O_SKIP_ACTION
5093	 */
5094	for (src = (ipfw_insn *)cmdbuf; src != cmd; src += i) {
5095		i = F_LEN(src);
5096		CHECK_RBUFLEN(i);
5097
5098		switch (src->opcode) {
5099		case O_LOG:
5100		case O_KEEP_STATE:
5101		case O_LIMIT:
5102		case O_ALTQ:
5103		case O_TAG:
5104		case O_SKIP_ACTION:
5105			break;
5106		default:
5107			bcopy(src, dst, i * sizeof(uint32_t));
5108			dst += i;
5109		}
5110	}
5111
5112	/*
5113	 * put back the have_state command as last opcode
5114	 */
5115	if (have_state && have_state->opcode != O_CHECK_STATE) {
5116		i = F_LEN(have_state);
5117		CHECK_RBUFLEN(i);
5118		bcopy(have_state, dst, i * sizeof(uint32_t));
5119		dst += i;
5120	}
5121
5122	/*
5123	 * put back the have_skipcmd command as very last opcode
5124	 */
5125	if (have_skipcmd) {
5126		i = F_LEN(have_skipcmd);
5127		CHECK_RBUFLEN(i);
5128		bcopy(have_skipcmd, dst, i * sizeof(uint32_t));
5129		dst += i;
5130	}
5131
5132	/*
5133	 * start action section
5134	 */
5135	rule->act_ofs = dst - rule->cmd;
5136
5137	/* put back O_LOG, O_ALTQ, O_TAG if necessary */
5138	if (have_log) {
5139		i = F_LEN(have_log);
5140		CHECK_RBUFLEN(i);
5141		bcopy(have_log, dst, i * sizeof(uint32_t));
5142		dst += i;
5143	}
5144	if (have_altq) {
5145		i = F_LEN(have_altq);
5146		CHECK_RBUFLEN(i);
5147		bcopy(have_altq, dst, i * sizeof(uint32_t));
5148		dst += i;
5149	}
5150	if (have_tag) {
5151		i = F_LEN(have_tag);
5152		CHECK_RBUFLEN(i);
5153		bcopy(have_tag, dst, i * sizeof(uint32_t));
5154		dst += i;
5155	}
5156
5157	/*
5158	 * copy all other actions
5159	 */
5160	for (src = (ipfw_insn *)actbuf; src != action; src += i) {
5161		i = F_LEN(src);
5162		CHECK_RBUFLEN(i);
5163		bcopy(src, dst, i * sizeof(uint32_t));
5164		dst += i;
5165	}
5166
5167	rule->cmd_len = (uint32_t *)dst - (uint32_t *)(rule->cmd);
5168	*rbufsize = (char *)dst - (char *)rule;
5169}
5170
5171static int
5172compare_ntlv(const void *_a, const void *_b)
5173{
5174	const ipfw_obj_ntlv *a, *b;
5175
5176	a = (const ipfw_obj_ntlv *)_a;
5177	b = (const ipfw_obj_ntlv *)_b;
5178
5179	if (a->set < b->set)
5180		return (-1);
5181	else if (a->set > b->set)
5182		return (1);
5183
5184	if (a->idx < b->idx)
5185		return (-1);
5186	else if (a->idx > b->idx)
5187		return (1);
5188
5189	if (a->head.type < b->head.type)
5190		return (-1);
5191	else if (a->head.type > b->head.type)
5192		return (1);
5193
5194	return (0);
5195}
5196
5197/*
5198 * Provide kernel with sorted list of referenced objects
5199 */
5200static void
5201object_sort_ctlv(ipfw_obj_ctlv *ctlv)
5202{
5203
5204	qsort(ctlv + 1, ctlv->count, ctlv->objsize, compare_ntlv);
5205}
5206
5207struct object_kt {
5208	uint16_t	uidx;
5209	uint16_t	type;
5210};
5211static int
5212compare_object_kntlv(const void *k, const void *v)
5213{
5214	const ipfw_obj_ntlv *ntlv;
5215	struct object_kt key;
5216
5217	key = *((const struct object_kt *)k);
5218	ntlv = (const ipfw_obj_ntlv *)v;
5219
5220	if (key.uidx < ntlv->idx)
5221		return (-1);
5222	else if (key.uidx > ntlv->idx)
5223		return (1);
5224
5225	if (key.type < ntlv->head.type)
5226		return (-1);
5227	else if (key.type > ntlv->head.type)
5228		return (1);
5229
5230	return (0);
5231}
5232
5233/*
5234 * Finds object name in @ctlv by @idx and @type.
5235 * Uses the following facts:
5236 * 1) All TLVs are the same size
5237 * 2) Kernel implementation provides already sorted list.
5238 *
5239 * Returns table name or NULL.
5240 */
5241static char *
5242object_search_ctlv(ipfw_obj_ctlv *ctlv, uint16_t idx, uint16_t type)
5243{
5244	ipfw_obj_ntlv *ntlv;
5245	struct object_kt key;
5246
5247	key.uidx = idx;
5248	key.type = type;
5249
5250	ntlv = bsearch(&key, (ctlv + 1), ctlv->count, ctlv->objsize,
5251	    compare_object_kntlv);
5252
5253	if (ntlv != NULL)
5254		return (ntlv->name);
5255
5256	return (NULL);
5257}
5258
5259static char *
5260table_search_ctlv(ipfw_obj_ctlv *ctlv, uint16_t idx)
5261{
5262
5263	return (object_search_ctlv(ctlv, idx, IPFW_TLV_TBL_NAME));
5264}
5265
5266/*
5267 * Adds one or more rules to ipfw chain.
5268 * Data layout:
5269 * Request:
5270 * [
5271 *   ip_fw3_opheader
5272 *   [ ipfw_obj_ctlv(IPFW_TLV_TBL_LIST) ipfw_obj_ntlv x N ] (optional *1)
5273 *   [ ipfw_obj_ctlv(IPFW_TLV_RULE_LIST) [ ip_fw_rule ip_fw_insn ] x N ] (*2) (*3)
5274 * ]
5275 * Reply:
5276 * [
5277 *   ip_fw3_opheader
5278 *   [ ipfw_obj_ctlv(IPFW_TLV_TBL_LIST) ipfw_obj_ntlv x N ] (optional)
5279 *   [ ipfw_obj_ctlv(IPFW_TLV_RULE_LIST) [ ip_fw_rule ip_fw_insn ] x N ]
5280 * ]
5281 *
5282 * Rules in reply are modified to store their actual ruleset number.
5283 *
5284 * (*1) TLVs inside IPFW_TLV_TBL_LIST needs to be sorted ascending
5285 * according to their idx field and there has to be no duplicates.
5286 * (*2) Numbered rules inside IPFW_TLV_RULE_LIST needs to be sorted ascending.
5287 * (*3) Each ip_fw structure needs to be aligned to u64 boundary.
5288 */
5289void
5290ipfw_add(char *av[])
5291{
5292	uint32_t rulebuf[1024];
5293	int rbufsize, default_off, tlen, rlen;
5294	size_t sz;
5295	struct tidx ts;
5296	struct ip_fw_rule *rule;
5297	caddr_t tbuf;
5298	ip_fw3_opheader *op3;
5299	ipfw_obj_ctlv *ctlv, *tstate;
5300
5301	rbufsize = sizeof(rulebuf);
5302	memset(rulebuf, 0, rbufsize);
5303	memset(&ts, 0, sizeof(ts));
5304
5305	/* Optimize case with no tables */
5306	default_off = sizeof(ipfw_obj_ctlv) + sizeof(ip_fw3_opheader);
5307	op3 = (ip_fw3_opheader *)rulebuf;
5308	ctlv = (ipfw_obj_ctlv *)(op3 + 1);
5309	rule = (struct ip_fw_rule *)(ctlv + 1);
5310	rbufsize -= default_off;
5311
5312	compile_rule(av, (uint32_t *)rule, &rbufsize, &ts);
5313	/* Align rule size to u64 boundary */
5314	rlen = roundup2(rbufsize, sizeof(uint64_t));
5315
5316	tbuf = NULL;
5317	sz = 0;
5318	tstate = NULL;
5319	if (ts.count != 0) {
5320		/* Some tables. We have to alloc more data */
5321		tlen = ts.count * sizeof(ipfw_obj_ntlv);
5322		sz = default_off + sizeof(ipfw_obj_ctlv) + tlen + rlen;
5323
5324		if ((tbuf = calloc(1, sz)) == NULL)
5325			err(EX_UNAVAILABLE, "malloc() failed for IP_FW_ADD");
5326		op3 = (ip_fw3_opheader *)tbuf;
5327		/* Tables first */
5328		ctlv = (ipfw_obj_ctlv *)(op3 + 1);
5329		ctlv->head.type = IPFW_TLV_TBLNAME_LIST;
5330		ctlv->head.length = sizeof(ipfw_obj_ctlv) + tlen;
5331		ctlv->count = ts.count;
5332		ctlv->objsize = sizeof(ipfw_obj_ntlv);
5333		memcpy(ctlv + 1, ts.idx, tlen);
5334		object_sort_ctlv(ctlv);
5335		tstate = ctlv;
5336		/* Rule next */
5337		ctlv = (ipfw_obj_ctlv *)((caddr_t)ctlv + ctlv->head.length);
5338		ctlv->head.type = IPFW_TLV_RULE_LIST;
5339		ctlv->head.length = sizeof(ipfw_obj_ctlv) + rlen;
5340		ctlv->count = 1;
5341		memcpy(ctlv + 1, rule, rbufsize);
5342	} else {
5343		/* Simply add header */
5344		sz = rlen + default_off;
5345		memset(ctlv, 0, sizeof(*ctlv));
5346		ctlv->head.type = IPFW_TLV_RULE_LIST;
5347		ctlv->head.length = sizeof(ipfw_obj_ctlv) + rlen;
5348		ctlv->count = 1;
5349	}
5350
5351	if (do_get3(IP_FW_XADD, op3, &sz) != 0)
5352		err(EX_UNAVAILABLE, "getsockopt(%s)", "IP_FW_XADD");
5353
5354	if (!g_co.do_quiet) {
5355		struct format_opts sfo;
5356		struct buf_pr bp;
5357		memset(&sfo, 0, sizeof(sfo));
5358		sfo.tstate = tstate;
5359		sfo.set_mask = (uint32_t)(-1);
5360		bp_alloc(&bp, 4096);
5361		show_static_rule(&g_co, &sfo, &bp, rule, NULL);
5362		printf("%s", bp.buf);
5363		bp_free(&bp);
5364	}
5365
5366	if (tbuf != NULL)
5367		free(tbuf);
5368
5369	if (ts.idx != NULL)
5370		free(ts.idx);
5371}
5372
5373/*
5374 * clear the counters or the log counters.
5375 * optname has the following values:
5376 *  0 (zero both counters and logging)
5377 *  1 (zero logging only)
5378 */
5379void
5380ipfw_zero(int ac, char *av[], int optname)
5381{
5382	ipfw_range_tlv rt;
5383	char const *errstr;
5384	char const *name = optname ? "RESETLOG" : "ZERO";
5385	uint32_t arg;
5386	int failed = EX_OK;
5387
5388	optname = optname ? IP_FW_XRESETLOG : IP_FW_XZERO;
5389	av++; ac--;
5390
5391	if (ac == 0) {
5392		/* clear all entries */
5393		memset(&rt, 0, sizeof(rt));
5394		rt.flags = IPFW_RCFLAG_ALL;
5395		if (do_range_cmd(optname, &rt) < 0)
5396			err(EX_UNAVAILABLE, "setsockopt(IP_FW_X%s)", name);
5397		if (!g_co.do_quiet)
5398			printf("%s.\n", optname == IP_FW_XZERO ?
5399			    "Accounting cleared":"Logging counts reset");
5400
5401		return;
5402	}
5403
5404	while (ac) {
5405		/* Rule number */
5406		if (isdigit(**av)) {
5407			arg = strtonum(*av, 0, 0xffff, &errstr);
5408			if (errstr)
5409				errx(EX_DATAERR,
5410				    "invalid rule number %s\n", *av);
5411			memset(&rt, 0, sizeof(rt));
5412			rt.start_rule = arg;
5413			rt.end_rule = arg;
5414			rt.flags |= IPFW_RCFLAG_RANGE;
5415			if (g_co.use_set != 0) {
5416				rt.set = g_co.use_set - 1;
5417				rt.flags |= IPFW_RCFLAG_SET;
5418			}
5419			if (do_range_cmd(optname, &rt) != 0) {
5420				warn("rule %u: setsockopt(IP_FW_X%s)",
5421				    arg, name);
5422				failed = EX_UNAVAILABLE;
5423			} else if (rt.new_set == 0) {
5424				printf("Entry %d not found\n", arg);
5425				failed = EX_UNAVAILABLE;
5426			} else if (!g_co.do_quiet)
5427				printf("Entry %d %s.\n", arg,
5428				    optname == IP_FW_XZERO ?
5429					"cleared" : "logging count reset");
5430		} else {
5431			errx(EX_USAGE, "invalid rule number ``%s''", *av);
5432		}
5433		av++; ac--;
5434	}
5435	if (failed != EX_OK)
5436		exit(failed);
5437}
5438
5439void
5440ipfw_flush(int force)
5441{
5442	ipfw_range_tlv rt;
5443
5444	if (!force && !g_co.do_quiet) { /* need to ask user */
5445		int c;
5446
5447		printf("Are you sure? [yn] ");
5448		fflush(stdout);
5449		do {
5450			c = toupper(getc(stdin));
5451			while (c != '\n' && getc(stdin) != '\n')
5452				if (feof(stdin))
5453					return; /* and do not flush */
5454		} while (c != 'Y' && c != 'N');
5455		printf("\n");
5456		if (c == 'N')	/* user said no */
5457			return;
5458	}
5459	if (g_co.do_pipe) {
5460		dummynet_flush();
5461		return;
5462	}
5463	/* `ipfw set N flush` - is the same that `ipfw delete set N` */
5464	memset(&rt, 0, sizeof(rt));
5465	if (g_co.use_set != 0) {
5466		rt.set = g_co.use_set - 1;
5467		rt.flags = IPFW_RCFLAG_SET;
5468	} else
5469		rt.flags = IPFW_RCFLAG_ALL;
5470	if (do_range_cmd(IP_FW_XDEL, &rt) != 0)
5471			err(EX_UNAVAILABLE, "setsockopt(IP_FW_XDEL)");
5472	if (!g_co.do_quiet)
5473		printf("Flushed all %s.\n", g_co.do_pipe ? "pipes" : "rules");
5474}
5475
5476static struct _s_x intcmds[] = {
5477      { "talist",	TOK_TALIST },
5478      { "iflist",	TOK_IFLIST },
5479      { "olist",	TOK_OLIST },
5480      { "vlist",	TOK_VLIST },
5481      { NULL, 0 }
5482};
5483
5484static struct _s_x otypes[] = {
5485	{ "EACTION",	IPFW_TLV_EACTION },
5486	{ "DYNSTATE",	IPFW_TLV_STATE_NAME },
5487	{ NULL, 0 }
5488};
5489
5490static const char*
5491lookup_eaction_name(ipfw_obj_ntlv *ntlv, int cnt, uint16_t type)
5492{
5493	const char *name;
5494	int i;
5495
5496	name = NULL;
5497	for (i = 0; i < cnt; i++) {
5498		if (ntlv[i].head.type != IPFW_TLV_EACTION)
5499			continue;
5500		if (IPFW_TLV_EACTION_NAME(ntlv[i].idx) != type)
5501			continue;
5502		name = ntlv[i].name;
5503		break;
5504	}
5505	return (name);
5506}
5507
5508static void
5509ipfw_list_objects(int ac __unused, char *av[] __unused)
5510{
5511	ipfw_obj_lheader req, *olh;
5512	ipfw_obj_ntlv *ntlv;
5513	const char *name;
5514	size_t sz;
5515	uint32_t i;
5516
5517	memset(&req, 0, sizeof(req));
5518	sz = sizeof(req);
5519	if (do_get3(IP_FW_DUMP_SRVOBJECTS, &req.opheader, &sz) != 0)
5520		if (errno != ENOMEM)
5521			return;
5522
5523	sz = req.size;
5524	if ((olh = calloc(1, sz)) == NULL)
5525		return;
5526
5527	olh->size = sz;
5528	if (do_get3(IP_FW_DUMP_SRVOBJECTS, &olh->opheader, &sz) != 0) {
5529		free(olh);
5530		return;
5531	}
5532
5533	if (olh->count > 0)
5534		printf("Objects list:\n");
5535	else
5536		printf("There are no objects\n");
5537	ntlv = (ipfw_obj_ntlv *)(olh + 1);
5538	for (i = 0; i < olh->count; i++) {
5539		name = match_value(otypes, ntlv->head.type);
5540		if (name == NULL)
5541			name = lookup_eaction_name(
5542			    (ipfw_obj_ntlv *)(olh + 1), olh->count,
5543			    ntlv->head.type);
5544		if (name == NULL)
5545			printf(" kidx: %4d\ttype: %10d\tname: %s\n",
5546			    ntlv->idx, ntlv->head.type, ntlv->name);
5547		else
5548			printf(" kidx: %4d\ttype: %10s\tname: %s\n",
5549			    ntlv->idx, name, ntlv->name);
5550		ntlv++;
5551	}
5552	free(olh);
5553}
5554
5555void
5556ipfw_internal_handler(int ac, char *av[])
5557{
5558	int tcmd;
5559
5560	ac--; av++;
5561	NEED1("internal cmd required");
5562
5563	if ((tcmd = match_token(intcmds, *av)) == -1)
5564		errx(EX_USAGE, "invalid internal sub-cmd: %s", *av);
5565
5566	switch (tcmd) {
5567	case TOK_IFLIST:
5568		ipfw_list_tifaces();
5569		break;
5570	case TOK_TALIST:
5571		ipfw_list_ta(ac, av);
5572		break;
5573	case TOK_OLIST:
5574		ipfw_list_objects(ac, av);
5575		break;
5576	case TOK_VLIST:
5577		ipfw_list_values(ac, av);
5578		break;
5579	}
5580}
5581
5582static int
5583ipfw_get_tracked_ifaces(ipfw_obj_lheader **polh)
5584{
5585	ipfw_obj_lheader req, *olh;
5586	size_t sz;
5587
5588	memset(&req, 0, sizeof(req));
5589	sz = sizeof(req);
5590
5591	if (do_get3(IP_FW_XIFLIST, &req.opheader, &sz) != 0) {
5592		if (errno != ENOMEM)
5593			return (errno);
5594	}
5595
5596	sz = req.size;
5597	if ((olh = calloc(1, sz)) == NULL)
5598		return (ENOMEM);
5599
5600	olh->size = sz;
5601	if (do_get3(IP_FW_XIFLIST, &olh->opheader, &sz) != 0) {
5602		free(olh);
5603		return (errno);
5604	}
5605
5606	*polh = olh;
5607	return (0);
5608}
5609
5610static int
5611ifinfo_cmp(const void *a, const void *b)
5612{
5613	const ipfw_iface_info *ia, *ib;
5614
5615	ia = (const ipfw_iface_info *)a;
5616	ib = (const ipfw_iface_info *)b;
5617
5618	return (stringnum_cmp(ia->ifname, ib->ifname));
5619}
5620
5621/*
5622 * Retrieves table list from kernel,
5623 * optionally sorts it and calls requested function for each table.
5624 * Returns 0 on success.
5625 */
5626static void
5627ipfw_list_tifaces(void)
5628{
5629	ipfw_obj_lheader *olh = NULL;
5630	ipfw_iface_info *info;
5631	uint32_t i;
5632	int error;
5633
5634	if ((error = ipfw_get_tracked_ifaces(&olh)) != 0)
5635		err(EX_OSERR, "Unable to request ipfw tracked interface list");
5636
5637	qsort(olh + 1, olh->count, olh->objsize, ifinfo_cmp);
5638
5639	info = (ipfw_iface_info *)(olh + 1);
5640	for (i = 0; i < olh->count; i++) {
5641		if (info->flags & IPFW_IFFLAG_RESOLVED)
5642			printf("%s ifindex: %d refcount: %u changes: %u\n",
5643			    info->ifname, info->ifindex, info->refcnt,
5644			    info->gencnt);
5645		else
5646			printf("%s ifindex: unresolved refcount: %u changes: %u\n",
5647			    info->ifname, info->refcnt, info->gencnt);
5648		info = (ipfw_iface_info *)((caddr_t)info + olh->objsize);
5649	}
5650
5651	free(olh);
5652}
5653