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