ipfw2.c revision 112250
198943Sluigi/*
298943Sluigi * Copyright (c) 2002 Luigi Rizzo
398943Sluigi * Copyright (c) 1996 Alex Nash, Paul Traina, Poul-Henning Kamp
498943Sluigi * Copyright (c) 1994 Ugen J.S.Antsilevich
598943Sluigi *
698943Sluigi * Idea and grammar partially left from:
798943Sluigi * Copyright (c) 1993 Daniel Boulet
898943Sluigi *
998943Sluigi * Redistribution and use in source forms, with and without modification,
1098943Sluigi * are permitted provided that this entire comment appears intact.
1198943Sluigi *
1298943Sluigi * Redistribution in binary form may occur without any restrictions.
1398943Sluigi * Obviously, it would be nice if you gave credit where credit is due
1498943Sluigi * but requiring it would be too onerous.
1598943Sluigi *
1698943Sluigi * This software is provided ``AS IS'' without any warranties of any kind.
1798943Sluigi *
1898943Sluigi * NEW command line interface for IP firewall facility
1998943Sluigi *
2098943Sluigi * $FreeBSD: head/sbin/ipfw/ipfw2.c 112250 2003-03-15 01:13:00Z cjc $
2198943Sluigi */
2298943Sluigi
2398943Sluigi#include <sys/param.h>
2498943Sluigi#include <sys/mbuf.h>
2598943Sluigi#include <sys/socket.h>
2698943Sluigi#include <sys/sockio.h>
2798943Sluigi#include <sys/sysctl.h>
2898943Sluigi#include <sys/time.h>
2998943Sluigi#include <sys/wait.h>
3098943Sluigi
3198943Sluigi#include <ctype.h>
3298943Sluigi#include <err.h>
3398943Sluigi#include <errno.h>
3498943Sluigi#include <grp.h>
3598943Sluigi#include <limits.h>
3698943Sluigi#include <netdb.h>
3798943Sluigi#include <pwd.h>
3898943Sluigi#include <signal.h>
3998943Sluigi#include <stdio.h>
4098943Sluigi#include <stdlib.h>
4198943Sluigi#include <stdarg.h>
4298943Sluigi#include <string.h>
4399603Sbde#include <timeconv.h>
4498943Sluigi#include <unistd.h>
4598943Sluigi#include <sysexits.h>
4698943Sluigi
4798943Sluigi#include <net/if.h>
4898943Sluigi#include <netinet/in.h>
4998943Sluigi#include <netinet/in_systm.h>
5098943Sluigi#include <netinet/ip.h>
5198943Sluigi#include <netinet/ip_icmp.h>
5298943Sluigi#include <netinet/ip_fw.h>
5398943Sluigi#include <net/route.h> /* def. of struct route */
5498943Sluigi#include <netinet/ip_dummynet.h>
5598943Sluigi#include <netinet/tcp.h>
5698943Sluigi#include <arpa/inet.h>
5798943Sluigi
5898943Sluigiint		s,			/* main RAW socket */
5998943Sluigi		do_resolv,		/* Would try to resolve all */
6098943Sluigi		do_acct,		/* Show packet/byte count */
6198943Sluigi		do_time,		/* Show time stamps */
6298943Sluigi		do_quiet,		/* Be quiet in add and flush */
6398943Sluigi		do_force,		/* Don't ask for confirmation */
6498943Sluigi		do_pipe,		/* this cmd refers to a pipe */
6598943Sluigi		do_sort,		/* field to sort results (0 = no) */
6698943Sluigi		do_dynamic,		/* display dynamic rules */
6798943Sluigi		do_expired,		/* display expired dynamic rules */
68102098Sluigi		do_compact,		/* show rules in compact mode */
69101628Sluigi		show_sets,		/* display rule sets */
7098943Sluigi		verbose;
7198943Sluigi
7298943Sluigi#define	IP_MASK_ALL	0xffffffff
7398943Sluigi
7498943Sluigi/*
7598943Sluigi * structure to hold flag names and associated values to be
7698943Sluigi * set in the appropriate masks.
7798943Sluigi * A NULL string terminates the array.
7898943Sluigi * Often, an element with 0 value contains an error string.
7998943Sluigi *
8098943Sluigi */
8198943Sluigistruct _s_x {
8298943Sluigi	char *s;
8398943Sluigi	int x;
8498943Sluigi};
8598943Sluigi
8698943Sluigistatic struct _s_x f_tcpflags[] = {
8798943Sluigi	{ "syn", TH_SYN },
8898943Sluigi	{ "fin", TH_FIN },
8998943Sluigi	{ "ack", TH_ACK },
9098943Sluigi	{ "psh", TH_PUSH },
9198943Sluigi	{ "rst", TH_RST },
9298943Sluigi	{ "urg", TH_URG },
9398943Sluigi	{ "tcp flag", 0 },
9498943Sluigi	{ NULL,	0 }
9598943Sluigi};
9698943Sluigi
9798943Sluigistatic struct _s_x f_tcpopts[] = {
9898943Sluigi	{ "mss",	IP_FW_TCPOPT_MSS },
9998943Sluigi	{ "maxseg",	IP_FW_TCPOPT_MSS },
10098943Sluigi	{ "window",	IP_FW_TCPOPT_WINDOW },
10198943Sluigi	{ "sack",	IP_FW_TCPOPT_SACK },
10298943Sluigi	{ "ts",		IP_FW_TCPOPT_TS },
10398943Sluigi	{ "timestamp",	IP_FW_TCPOPT_TS },
10498943Sluigi	{ "cc",		IP_FW_TCPOPT_CC },
10598943Sluigi	{ "tcp option",	0 },
10698943Sluigi	{ NULL,	0 }
10798943Sluigi};
10898943Sluigi
10998943Sluigi/*
11098943Sluigi * IP options span the range 0 to 255 so we need to remap them
11198943Sluigi * (though in fact only the low 5 bits are significant).
11298943Sluigi */
11398943Sluigistatic struct _s_x f_ipopts[] = {
11498943Sluigi	{ "ssrr",	IP_FW_IPOPT_SSRR},
11598943Sluigi	{ "lsrr",	IP_FW_IPOPT_LSRR},
11698943Sluigi	{ "rr",		IP_FW_IPOPT_RR},
11798943Sluigi	{ "ts",		IP_FW_IPOPT_TS},
11898943Sluigi	{ "ip option",	0 },
11998943Sluigi	{ NULL,	0 }
12098943Sluigi};
12198943Sluigi
12298943Sluigistatic struct _s_x f_iptos[] = {
12398943Sluigi	{ "lowdelay",	IPTOS_LOWDELAY},
12498943Sluigi	{ "throughput",	IPTOS_THROUGHPUT},
12598943Sluigi	{ "reliability", IPTOS_RELIABILITY},
12698943Sluigi	{ "mincost",	IPTOS_MINCOST},
12798943Sluigi	{ "congestion",	IPTOS_CE},
12898943Sluigi	{ "ecntransport", IPTOS_ECT},
12998943Sluigi	{ "ip tos option", 0},
13098943Sluigi	{ NULL,	0 }
13198943Sluigi};
13298943Sluigi
13398943Sluigistatic struct _s_x limit_masks[] = {
13498943Sluigi	{"all",		DYN_SRC_ADDR|DYN_SRC_PORT|DYN_DST_ADDR|DYN_DST_PORT},
13598943Sluigi	{"src-addr",	DYN_SRC_ADDR},
13698943Sluigi	{"src-port",	DYN_SRC_PORT},
13798943Sluigi	{"dst-addr",	DYN_DST_ADDR},
13898943Sluigi	{"dst-port",	DYN_DST_PORT},
13998943Sluigi	{NULL,		0}
14098943Sluigi};
14198943Sluigi
14298943Sluigi/*
14398943Sluigi * we use IPPROTO_ETHERTYPE as a fake protocol id to call the print routines
14498943Sluigi * This is only used in this code.
14598943Sluigi */
14698943Sluigi#define IPPROTO_ETHERTYPE	0x1000
14798943Sluigistatic struct _s_x ether_types[] = {
14898943Sluigi    /*
14998943Sluigi     * Note, we cannot use "-:&/" in the names because they are field
15098943Sluigi     * separators in the type specifications. Also, we use s = NULL as
15198943Sluigi     * end-delimiter, because a type of 0 can be legal.
15298943Sluigi     */
15398943Sluigi	{ "ip",		0x0800 },
15498943Sluigi	{ "ipv4",	0x0800 },
15598943Sluigi	{ "ipv6",	0x86dd },
15698943Sluigi	{ "arp",	0x0806 },
15798943Sluigi	{ "rarp",	0x8035 },
15898943Sluigi	{ "vlan",	0x8100 },
15998943Sluigi	{ "loop",	0x9000 },
16098943Sluigi	{ "trail",	0x1000 },
16198943Sluigi	{ "at",		0x809b },
16298943Sluigi	{ "atalk",	0x809b },
16398943Sluigi	{ "aarp",	0x80f3 },
16498943Sluigi	{ "pppoe_disc",	0x8863 },
16598943Sluigi	{ "pppoe_sess",	0x8864 },
16698943Sluigi	{ "ipx_8022",	0x00E0 },
16798943Sluigi	{ "ipx_8023",	0x0000 },
16898943Sluigi	{ "ipx_ii",	0x8137 },
16998943Sluigi	{ "ipx_snap",	0x8137 },
17098943Sluigi	{ "ipx",	0x8137 },
17198943Sluigi	{ "ns",		0x0600 },
17298943Sluigi	{ NULL,		0 }
17398943Sluigi};
17498943Sluigi
17598943Sluigistatic void show_usage(void);
17698943Sluigi
17798943Sluigienum tokens {
17898943Sluigi	TOK_NULL=0,
17998943Sluigi
18098943Sluigi	TOK_OR,
18198943Sluigi	TOK_NOT,
182101641Sluigi	TOK_STARTBRACE,
183101641Sluigi	TOK_ENDBRACE,
18498943Sluigi
18598943Sluigi	TOK_ACCEPT,
18698943Sluigi	TOK_COUNT,
18798943Sluigi	TOK_PIPE,
18898943Sluigi	TOK_QUEUE,
18998943Sluigi	TOK_DIVERT,
19098943Sluigi	TOK_TEE,
19198943Sluigi	TOK_FORWARD,
19298943Sluigi	TOK_SKIPTO,
19398943Sluigi	TOK_DENY,
19498943Sluigi	TOK_REJECT,
19598943Sluigi	TOK_RESET,
19698943Sluigi	TOK_UNREACH,
19798943Sluigi	TOK_CHECKSTATE,
19898943Sluigi
19998943Sluigi	TOK_UID,
20098943Sluigi	TOK_GID,
20198943Sluigi	TOK_IN,
20298943Sluigi	TOK_LIMIT,
20398943Sluigi	TOK_KEEPSTATE,
20498943Sluigi	TOK_LAYER2,
20598943Sluigi	TOK_OUT,
20698943Sluigi	TOK_XMIT,
20798943Sluigi	TOK_RECV,
20898943Sluigi	TOK_VIA,
20998943Sluigi	TOK_FRAG,
21098943Sluigi	TOK_IPOPTS,
21198943Sluigi	TOK_IPLEN,
21298943Sluigi	TOK_IPID,
21398943Sluigi	TOK_IPPRECEDENCE,
21498943Sluigi	TOK_IPTOS,
21598943Sluigi	TOK_IPTTL,
21698943Sluigi	TOK_IPVER,
21798943Sluigi	TOK_ESTAB,
21898943Sluigi	TOK_SETUP,
21998943Sluigi	TOK_TCPFLAGS,
22098943Sluigi	TOK_TCPOPTS,
22198943Sluigi	TOK_TCPSEQ,
22298943Sluigi	TOK_TCPACK,
22398943Sluigi	TOK_TCPWIN,
22498943Sluigi	TOK_ICMPTYPES,
225102087Sluigi	TOK_MAC,
226102087Sluigi	TOK_MACTYPE,
227112250Scjc	TOK_VERREVPATH,
22898943Sluigi
22998943Sluigi	TOK_PLR,
230101978Sluigi	TOK_NOERROR,
23198943Sluigi	TOK_BUCKETS,
23298943Sluigi	TOK_DSTIP,
23398943Sluigi	TOK_SRCIP,
23498943Sluigi	TOK_DSTPORT,
23598943Sluigi	TOK_SRCPORT,
23698943Sluigi	TOK_ALL,
23798943Sluigi	TOK_MASK,
23898943Sluigi	TOK_BW,
23998943Sluigi	TOK_DELAY,
24098943Sluigi	TOK_RED,
24198943Sluigi	TOK_GRED,
24298943Sluigi	TOK_DROPTAIL,
24398943Sluigi	TOK_PROTO,
24498943Sluigi	TOK_WEIGHT,
24598943Sluigi};
24698943Sluigi
24798943Sluigistruct _s_x dummynet_params[] = {
24898943Sluigi	{ "plr",		TOK_PLR },
249101978Sluigi	{ "noerror",		TOK_NOERROR },
25098943Sluigi	{ "buckets",		TOK_BUCKETS },
25198943Sluigi	{ "dst-ip",		TOK_DSTIP },
25298943Sluigi	{ "src-ip",		TOK_SRCIP },
25398943Sluigi	{ "dst-port",		TOK_DSTPORT },
25498943Sluigi	{ "src-port",		TOK_SRCPORT },
25598943Sluigi	{ "proto",		TOK_PROTO },
25698943Sluigi	{ "weight",		TOK_WEIGHT },
25798943Sluigi	{ "all",		TOK_ALL },
25898943Sluigi	{ "mask",		TOK_MASK },
25998943Sluigi	{ "droptail",		TOK_DROPTAIL },
26098943Sluigi	{ "red",		TOK_RED },
26198943Sluigi	{ "gred",		TOK_GRED },
26298943Sluigi	{ "bw",			TOK_BW },
26398943Sluigi	{ "bandwidth",		TOK_BW },
26498943Sluigi	{ "delay",		TOK_DELAY },
26599475Sluigi	{ "pipe",		TOK_PIPE },
26698943Sluigi	{ "queue",		TOK_QUEUE },
26798943Sluigi	{ "dummynet-params",	TOK_NULL },
26898943Sluigi	{ NULL, 0 }
26998943Sluigi};
27098943Sluigi
27198943Sluigistruct _s_x rule_actions[] = {
27298943Sluigi	{ "accept",		TOK_ACCEPT },
27398943Sluigi	{ "pass",		TOK_ACCEPT },
27498943Sluigi	{ "allow",		TOK_ACCEPT },
27598943Sluigi	{ "permit",		TOK_ACCEPT },
27698943Sluigi	{ "count",		TOK_COUNT },
27798943Sluigi	{ "pipe",		TOK_PIPE },
27898943Sluigi	{ "queue",		TOK_QUEUE },
27998943Sluigi	{ "divert",		TOK_DIVERT },
28098943Sluigi	{ "tee",		TOK_TEE },
28198943Sluigi	{ "fwd",		TOK_FORWARD },
28298943Sluigi	{ "forward",		TOK_FORWARD },
28398943Sluigi	{ "skipto",		TOK_SKIPTO },
28498943Sluigi	{ "deny",		TOK_DENY },
28598943Sluigi	{ "drop",		TOK_DENY },
28698943Sluigi	{ "reject",		TOK_REJECT },
28798943Sluigi	{ "reset",		TOK_RESET },
28899475Sluigi	{ "unreach",		TOK_UNREACH },
28998943Sluigi	{ "check-state",	TOK_CHECKSTATE },
29098943Sluigi	{ NULL,			TOK_NULL },
29198943Sluigi	{ NULL, 0 }
29298943Sluigi};
29398943Sluigi
29498943Sluigistruct _s_x rule_options[] = {
29598943Sluigi	{ "uid",		TOK_UID },
29698943Sluigi	{ "gid",		TOK_GID },
29798943Sluigi	{ "in",			TOK_IN },
29898943Sluigi	{ "limit",		TOK_LIMIT },
29998943Sluigi	{ "keep-state",		TOK_KEEPSTATE },
30098943Sluigi	{ "bridged",		TOK_LAYER2 },
30198943Sluigi	{ "layer2",		TOK_LAYER2 },
30298943Sluigi	{ "out",		TOK_OUT },
30398943Sluigi	{ "xmit",		TOK_XMIT },
30498943Sluigi	{ "recv",		TOK_RECV },
30598943Sluigi	{ "via",		TOK_VIA },
30698943Sluigi	{ "fragment",		TOK_FRAG },
30798943Sluigi	{ "frag",		TOK_FRAG },
30898943Sluigi	{ "ipoptions",		TOK_IPOPTS },
30998943Sluigi	{ "ipopts",		TOK_IPOPTS },
31098943Sluigi	{ "iplen",		TOK_IPLEN },
31198943Sluigi	{ "ipid",		TOK_IPID },
31298943Sluigi	{ "ipprecedence",	TOK_IPPRECEDENCE },
31398943Sluigi	{ "iptos",		TOK_IPTOS },
31498943Sluigi	{ "ipttl",		TOK_IPTTL },
31598943Sluigi	{ "ipversion",		TOK_IPVER },
31698943Sluigi	{ "ipver",		TOK_IPVER },
31798943Sluigi	{ "estab",		TOK_ESTAB },
31898943Sluigi	{ "established",	TOK_ESTAB },
31998943Sluigi	{ "setup",		TOK_SETUP },
32098943Sluigi	{ "tcpflags",		TOK_TCPFLAGS },
32198943Sluigi	{ "tcpflgs",		TOK_TCPFLAGS },
32298943Sluigi	{ "tcpoptions",		TOK_TCPOPTS },
32398943Sluigi	{ "tcpopts",		TOK_TCPOPTS },
32498943Sluigi	{ "tcpseq",		TOK_TCPSEQ },
32598943Sluigi	{ "tcpack",		TOK_TCPACK },
32698943Sluigi	{ "tcpwin",		TOK_TCPWIN },
32799909Sluigi	{ "icmptype",		TOK_ICMPTYPES },
32898943Sluigi	{ "icmptypes",		TOK_ICMPTYPES },
329102087Sluigi	{ "dst-ip",		TOK_DSTIP },
330102087Sluigi	{ "src-ip",		TOK_SRCIP },
331102087Sluigi	{ "dst-port",		TOK_DSTPORT },
332102087Sluigi	{ "src-port",		TOK_SRCPORT },
333102087Sluigi	{ "proto",		TOK_PROTO },
334102087Sluigi	{ "MAC",		TOK_MAC },
335102087Sluigi	{ "mac",		TOK_MAC },
336102087Sluigi	{ "mac-type",		TOK_MACTYPE },
337112250Scjc	{ "verrevpath",		TOK_VERREVPATH },
33898943Sluigi
33998943Sluigi	{ "not",		TOK_NOT },		/* pseudo option */
34098943Sluigi	{ "!", /* escape ? */	TOK_NOT },		/* pseudo option */
34198943Sluigi	{ "or",			TOK_OR },		/* pseudo option */
34298943Sluigi	{ "|", /* escape */	TOK_OR },		/* pseudo option */
343101641Sluigi	{ "{",			TOK_STARTBRACE },	/* pseudo option */
344101641Sluigi	{ "(",			TOK_STARTBRACE },	/* pseudo option */
345101641Sluigi	{ "}",			TOK_ENDBRACE },		/* pseudo option */
346101641Sluigi	{ ")",			TOK_ENDBRACE },		/* pseudo option */
34798943Sluigi	{ NULL,			TOK_NULL },
34898943Sluigi	{ NULL, 0 }
34998943Sluigi};
35098943Sluigi
35198943Sluigi/**
35298943Sluigi * match_token takes a table and a string, returns the value associated
35398943Sluigi * with the string (0 meaning an error in most cases)
35498943Sluigi */
35598943Sluigistatic int
35698943Sluigimatch_token(struct _s_x *table, char *string)
35798943Sluigi{
35898943Sluigi	struct _s_x *pt;
35998943Sluigi	int i = strlen(string);
36098943Sluigi
36198943Sluigi	for (pt = table ; i && pt->s != NULL ; pt++)
36298943Sluigi		if (strlen(pt->s) == i && !bcmp(string, pt->s, i))
36398943Sluigi			return pt->x;
36498943Sluigi	return -1;
36598943Sluigi};
36698943Sluigi
36798943Sluigistatic char *
36898943Sluigimatch_value(struct _s_x *p, u_int32_t value)
36998943Sluigi{
37098943Sluigi	for (; p->s != NULL; p++)
37198943Sluigi		if (p->x == value)
37298943Sluigi			return p->s;
37398943Sluigi	return NULL;
37498943Sluigi}
37598943Sluigi
37698943Sluigi/*
37798943Sluigi * prints one port, symbolic or numeric
37898943Sluigi */
37998943Sluigistatic void
38098943Sluigiprint_port(int proto, u_int16_t port)
38198943Sluigi{
38298943Sluigi
38398943Sluigi	if (proto == IPPROTO_ETHERTYPE) {
38498943Sluigi		char *s;
38598943Sluigi
38698943Sluigi		if (do_resolv && (s = match_value(ether_types, port)) )
38798943Sluigi			printf("%s", s);
38898943Sluigi		else
38998943Sluigi			printf("0x%04x", port);
39098943Sluigi	} else {
39198943Sluigi		struct servent *se = NULL;
39298943Sluigi		if (do_resolv) {
39398943Sluigi			struct protoent *pe = getprotobynumber(proto);
39498943Sluigi
39598943Sluigi			se = getservbyport(htons(port), pe ? pe->p_name : NULL);
39698943Sluigi		}
39798943Sluigi		if (se)
39898943Sluigi			printf("%s", se->s_name);
39998943Sluigi		else
40098943Sluigi			printf("%d", port);
40198943Sluigi	}
40298943Sluigi}
40398943Sluigi
40498943Sluigi/*
40598943Sluigi * print the values in a list of ports
40698943Sluigi * XXX todo: add support for mask.
40798943Sluigi */
40898943Sluigistatic void
409102087Sluigiprint_newports(ipfw_insn_u16 *cmd, int proto, int opcode)
41098943Sluigi{
41198943Sluigi	u_int16_t *p = cmd->ports;
41298943Sluigi	int i;
41398943Sluigi	char *sep= " ";
41498943Sluigi
41598943Sluigi	if (cmd->o.len & F_NOT)
41698943Sluigi		printf(" not");
417102087Sluigi	if (opcode != 0)
418102087Sluigi		printf ("%s", opcode == O_MAC_TYPE ? " mac-type" :
419102087Sluigi		    (opcode == O_IP_DSTPORT ? " dst-port" : " src-port"));
42098943Sluigi	for (i = F_LEN((ipfw_insn *)cmd) - 1; i > 0; i--, p += 2) {
42198943Sluigi		printf(sep);
42298943Sluigi		print_port(proto, p[0]);
42398943Sluigi		if (p[0] != p[1]) {
42498943Sluigi			printf("-");
42598943Sluigi			print_port(proto, p[1]);
42698943Sluigi		}
42798943Sluigi		sep = ",";
42898943Sluigi	}
42998943Sluigi}
43098943Sluigi
43198943Sluigi/*
43298943Sluigi * Like strtol, but also translates service names into port numbers
43398943Sluigi * for some protocols.
43498943Sluigi * In particular:
43598943Sluigi *	proto == -1 disables the protocol check;
43698943Sluigi *	proto == IPPROTO_ETHERTYPE looks up an internal table
43798943Sluigi *	proto == <some value in /etc/protocols> matches the values there.
438101628Sluigi * Returns *end == s in case the parameter is not found.
43998943Sluigi */
44098943Sluigistatic int
44198943Sluigistrtoport(char *s, char **end, int base, int proto)
44298943Sluigi{
443101628Sluigi	char *p, *buf;
444101628Sluigi	char *s1;
44598943Sluigi	int i;
44698943Sluigi
447101628Sluigi	*end = s;		/* default - not found */
44898943Sluigi	if ( *s == '\0')
449101628Sluigi		return 0;	/* not found */
450106505Smaxim
45198943Sluigi	if (isdigit(*s))
45298943Sluigi		return strtol(s, end, base);
45398943Sluigi
45498943Sluigi	/*
455101628Sluigi	 * find separator. '\\' escapes the next char.
45698943Sluigi	 */
457101628Sluigi	for (s1 = s; *s1 && (isalnum(*s1) || *s1 == '\\') ; s1++)
458101628Sluigi		if (*s1 == '\\' && s1[1] != '\0')
459101628Sluigi			s1++;
46098943Sluigi
461101628Sluigi	buf = malloc(s1 - s + 1);
462101628Sluigi	if (buf == NULL)
463101628Sluigi		return 0;
464101628Sluigi
465101628Sluigi	/*
466101628Sluigi	 * copy into a buffer skipping backslashes
467101628Sluigi	 */
468101628Sluigi	for (p = s, i = 0; p != s1 ; p++)
469101628Sluigi		if ( *p != '\\')
470101628Sluigi			buf[i++] = *p;
471101628Sluigi	buf[i++] = '\0';
472101628Sluigi
47398943Sluigi	if (proto == IPPROTO_ETHERTYPE) {
474101628Sluigi		i = match_token(ether_types, buf);
475101628Sluigi		free(buf);
476101628Sluigi		if (i != -1) {	/* found */
47798943Sluigi			*end = s1;
47898943Sluigi			return i;
47998943Sluigi		}
48098943Sluigi	} else {
48198943Sluigi		struct protoent *pe = NULL;
48298943Sluigi		struct servent *se;
48398943Sluigi
48498943Sluigi		if (proto != 0)
48598943Sluigi			pe = getprotobynumber(proto);
48698943Sluigi		setservent(1);
487101628Sluigi		se = getservbyname(buf, pe ? pe->p_name : NULL);
488101628Sluigi		free(buf);
48998943Sluigi		if (se != NULL) {
49098943Sluigi			*end = s1;
49198943Sluigi			return ntohs(se->s_port);
49298943Sluigi		}
49398943Sluigi	}
494101628Sluigi	return 0;	/* not found */
49598943Sluigi}
49698943Sluigi
49798943Sluigi/*
49898943Sluigi * fill the body of the command with the list of port ranges.
49998943Sluigi * At the moment it only understands numeric ranges.
50098943Sluigi */
50198943Sluigistatic int
50298943Sluigifill_newports(ipfw_insn_u16 *cmd, char *av, int proto)
50398943Sluigi{
50498943Sluigi	u_int16_t *p = cmd->ports;
50598943Sluigi	int i = 0;
506102087Sluigi	char *s = av;
50798943Sluigi
508102087Sluigi	while (*s) {
50998943Sluigi		u_int16_t a, b;
51098943Sluigi
51198943Sluigi		a = strtoport(av, &s, 0, proto);
51298943Sluigi		if (s == av) /* no parameter */
51398943Sluigi			break;
51498943Sluigi		if (*s == '-') { /* a range */
51598943Sluigi			av = s+1;
51698943Sluigi			b = strtoport(av, &s, 0, proto);
51798943Sluigi			if (s == av) /* no parameter */
51898943Sluigi				break;
51998943Sluigi			p[0] = a;
52098943Sluigi			p[1] = b;
52198943Sluigi		} else if (*s == ',' || *s == '\0' ) {
52298943Sluigi			p[0] = p[1] = a;
523101978Sluigi		} else {	/* invalid separator */
524101978Sluigi			errx(EX_DATAERR, "invalid separator <%c> in <%s>\n",
525101978Sluigi				*s, av);
526101978Sluigi		}
527102087Sluigi		i++;
528102087Sluigi		p += 2;
52998943Sluigi		av = s+1;
53098943Sluigi	}
53198943Sluigi	if (i > 0) {
53298943Sluigi		if (i+1 > F_LEN_MASK)
533102087Sluigi			errx(EX_DATAERR, "too many ports/ranges\n");
53498943Sluigi		cmd->o.len |= i+1; /* leave F_NOT and F_OR untouched */
53598943Sluigi	}
53698943Sluigi	return i;
53798943Sluigi}
53898943Sluigi
53998943Sluigistatic struct _s_x icmpcodes[] = {
54098943Sluigi      { "net",			ICMP_UNREACH_NET },
54198943Sluigi      { "host",			ICMP_UNREACH_HOST },
54298943Sluigi      { "protocol",		ICMP_UNREACH_PROTOCOL },
54398943Sluigi      { "port",			ICMP_UNREACH_PORT },
54498943Sluigi      { "needfrag",		ICMP_UNREACH_NEEDFRAG },
54598943Sluigi      { "srcfail",		ICMP_UNREACH_SRCFAIL },
54698943Sluigi      { "net-unknown",		ICMP_UNREACH_NET_UNKNOWN },
54798943Sluigi      { "host-unknown",		ICMP_UNREACH_HOST_UNKNOWN },
54898943Sluigi      { "isolated",		ICMP_UNREACH_ISOLATED },
54998943Sluigi      { "net-prohib",		ICMP_UNREACH_NET_PROHIB },
55098943Sluigi      { "host-prohib",		ICMP_UNREACH_HOST_PROHIB },
55198943Sluigi      { "tosnet",		ICMP_UNREACH_TOSNET },
55298943Sluigi      { "toshost",		ICMP_UNREACH_TOSHOST },
55398943Sluigi      { "filter-prohib",	ICMP_UNREACH_FILTER_PROHIB },
55498943Sluigi      { "host-precedence",	ICMP_UNREACH_HOST_PRECEDENCE },
55598943Sluigi      { "precedence-cutoff",	ICMP_UNREACH_PRECEDENCE_CUTOFF },
55698943Sluigi      { NULL, 0 }
55798943Sluigi};
55898943Sluigi
55998943Sluigistatic void
56098943Sluigifill_reject_code(u_short *codep, char *str)
56198943Sluigi{
56298943Sluigi	int val;
56398943Sluigi	char *s;
56498943Sluigi
56598943Sluigi	val = strtoul(str, &s, 0);
56698943Sluigi	if (s == str || *s != '\0' || val >= 0x100)
56798943Sluigi		val = match_token(icmpcodes, str);
568102087Sluigi	if (val < 0)
56998943Sluigi		errx(EX_DATAERR, "unknown ICMP unreachable code ``%s''", str);
57098943Sluigi	*codep = val;
57198943Sluigi	return;
57298943Sluigi}
57398943Sluigi
57498943Sluigistatic void
57599475Sluigiprint_reject_code(u_int16_t code)
57698943Sluigi{
57798943Sluigi	char *s = match_value(icmpcodes, code);
57898943Sluigi
57998943Sluigi	if (s != NULL)
58099475Sluigi		printf("unreach %s", s);
58198943Sluigi	else
58299475Sluigi		printf("unreach %u", code);
58398943Sluigi}
58498943Sluigi
58598943Sluigi/*
58698943Sluigi * Returns the number of bits set (from left) in a contiguous bitmask,
58798943Sluigi * or -1 if the mask is not contiguous.
58898943Sluigi * XXX this needs a proper fix.
58998943Sluigi * This effectively works on masks in big-endian (network) format.
59098943Sluigi * when compiled on little endian architectures.
59198943Sluigi *
59298943Sluigi * First bit is bit 7 of the first byte -- note, for MAC addresses,
59398943Sluigi * the first bit on the wire is bit 0 of the first byte.
59498943Sluigi * len is the max length in bits.
59598943Sluigi */
59698943Sluigistatic int
59798943Sluigicontigmask(u_char *p, int len)
59898943Sluigi{
59998943Sluigi	int i, n;
60098943Sluigi	for (i=0; i<len ; i++)
60198943Sluigi		if ( (p[i/8] & (1 << (7 - (i%8)))) == 0) /* first bit unset */
60298943Sluigi			break;
60398943Sluigi	for (n=i+1; n < len; n++)
60498943Sluigi		if ( (p[n/8] & (1 << (7 - (n%8)))) != 0)
60598943Sluigi			return -1; /* mask not contiguous */
60698943Sluigi	return i;
60798943Sluigi}
60898943Sluigi
60998943Sluigi/*
61098943Sluigi * print flags set/clear in the two bitmasks passed as parameters.
61198943Sluigi * There is a specialized check for f_tcpflags.
61298943Sluigi */
61398943Sluigistatic void
61498943Sluigiprint_flags(char *name, ipfw_insn *cmd, struct _s_x *list)
61598943Sluigi{
61698943Sluigi	char *comma="";
61798943Sluigi	int i;
61898943Sluigi	u_char set = cmd->arg1 & 0xff;
61998943Sluigi	u_char clear = (cmd->arg1 >> 8) & 0xff;
62098943Sluigi
62198943Sluigi	if (list == f_tcpflags && set == TH_SYN && clear == TH_ACK) {
62298943Sluigi		printf(" setup");
62398943Sluigi		return;
62498943Sluigi	}
62598943Sluigi
62698943Sluigi	printf(" %s ", name);
62798943Sluigi	for (i=0; list[i].x != 0; i++) {
62898943Sluigi		if (set & list[i].x) {
62998943Sluigi			set &= ~list[i].x;
63098943Sluigi			printf("%s%s", comma, list[i].s);
63198943Sluigi			comma = ",";
63298943Sluigi		}
63398943Sluigi		if (clear & list[i].x) {
63498943Sluigi			clear &= ~list[i].x;
63598943Sluigi			printf("%s!%s", comma, list[i].s);
63698943Sluigi			comma = ",";
63798943Sluigi		}
63898943Sluigi	}
63998943Sluigi}
64098943Sluigi
64198943Sluigi/*
64298943Sluigi * Print the ip address contained in a command.
64398943Sluigi */
64498943Sluigistatic void
645102087Sluigiprint_ip(ipfw_insn_ip *cmd, char *s)
64698943Sluigi{
64798943Sluigi	struct hostent *he = NULL;
64898943Sluigi	int mb;
64998943Sluigi
650102087Sluigi	printf("%s%s ", cmd->o.len & F_NOT ? " not": "", s);
65198943Sluigi
65298943Sluigi	if (cmd->o.opcode == O_IP_SRC_ME || cmd->o.opcode == O_IP_DST_ME) {
65398943Sluigi		printf("me");
65498943Sluigi		return;
65598943Sluigi	}
65698943Sluigi	if (cmd->o.opcode == O_IP_SRC_SET || cmd->o.opcode == O_IP_DST_SET) {
65798943Sluigi		u_int32_t x, *d;
65898943Sluigi		int i;
65998943Sluigi		char comma = '{';
66098943Sluigi
66198943Sluigi		x = cmd->o.arg1 - 1;
66298943Sluigi		x = htonl( ~x );
66398943Sluigi		cmd->addr.s_addr = htonl(cmd->addr.s_addr);
66498943Sluigi		printf("%s/%d", inet_ntoa(cmd->addr),
66598943Sluigi			contigmask((u_char *)&x, 32));
66698943Sluigi		x = cmd->addr.s_addr = htonl(cmd->addr.s_addr);
66798943Sluigi		x &= 0xff; /* base */
66898943Sluigi		d = (u_int32_t *)&(cmd->mask);
66998943Sluigi		for (i=0; i < cmd->o.arg1; i++)
67098943Sluigi			if (d[ i/32] & (1<<(i & 31))) {
67198943Sluigi				printf("%c%d", comma, i+x);
67298943Sluigi				comma = ',';
67398943Sluigi			}
67498943Sluigi		printf("}");
67598943Sluigi		return;
67698943Sluigi	}
67798943Sluigi	if (cmd->o.opcode == O_IP_SRC || cmd->o.opcode == O_IP_DST)
67898943Sluigi		mb = 32;
67998943Sluigi	else
68098943Sluigi		mb = contigmask((u_char *)&(cmd->mask.s_addr), 32);
68198943Sluigi	if (mb == 32 && do_resolv)
68298943Sluigi		he = gethostbyaddr((char *)&(cmd->addr.s_addr),
68398943Sluigi		    sizeof(u_long), AF_INET);
68498943Sluigi	if (he != NULL)		/* resolved to name */
68598943Sluigi		printf("%s", he->h_name);
68698943Sluigi	else if (mb == 0)	/* any */
68798943Sluigi		printf("any");
68898943Sluigi	else {		/* numeric IP followed by some kind of mask */
68998943Sluigi		printf("%s", inet_ntoa(cmd->addr));
69098943Sluigi		if (mb < 0)
69198943Sluigi			printf(":%s", inet_ntoa(cmd->mask));
69298943Sluigi		else if (mb < 32)
69398943Sluigi			printf("/%d", mb);
69498943Sluigi	}
69598943Sluigi}
69698943Sluigi
69798943Sluigi/*
69898943Sluigi * prints a MAC address/mask pair
69998943Sluigi */
70098943Sluigistatic void
70198943Sluigiprint_mac(u_char *addr, u_char *mask)
70298943Sluigi{
70398943Sluigi	int l = contigmask(mask, 48);
70498943Sluigi
70598943Sluigi	if (l == 0)
70698943Sluigi		printf(" any");
70798943Sluigi	else {
70898943Sluigi		printf(" %02x:%02x:%02x:%02x:%02x:%02x",
70998943Sluigi		    addr[0], addr[1], addr[2], addr[3], addr[4], addr[5]);
71098943Sluigi		if (l == -1)
71198943Sluigi			printf("&%02x:%02x:%02x:%02x:%02x:%02x",
71298943Sluigi			    mask[0], mask[1], mask[2],
71398943Sluigi			    mask[3], mask[4], mask[5]);
71498943Sluigi		else if (l < 48)
71598943Sluigi			printf("/%d", l);
71698943Sluigi	}
71798943Sluigi}
71898943Sluigi
71999475Sluigistatic void
72099475Sluigifill_icmptypes(ipfw_insn_u32 *cmd, char *av)
72199475Sluigi{
72299475Sluigi	u_int8_t type;
72398943Sluigi
72499475Sluigi	cmd->d[0] = 0;
72599475Sluigi	while (*av) {
72699475Sluigi		if (*av == ',')
72799475Sluigi			av++;
72899475Sluigi
72999475Sluigi		type = strtoul(av, &av, 0);
73099475Sluigi
73199475Sluigi		if (*av != ',' && *av != '\0')
73299475Sluigi			errx(EX_DATAERR, "invalid ICMP type");
73399475Sluigi
73499475Sluigi		if (type > 31)
73599475Sluigi			errx(EX_DATAERR, "ICMP type out of range");
73699475Sluigi
73799475Sluigi		cmd->d[0] |= 1 << type;
73899475Sluigi	}
73999475Sluigi	cmd->o.opcode = O_ICMPTYPE;
74099475Sluigi	cmd->o.len |= F_INSN_SIZE(ipfw_insn_u32);
74199475Sluigi}
74299475Sluigi
74399475Sluigistatic void
74499475Sluigiprint_icmptypes(ipfw_insn_u32 *cmd)
74599475Sluigi{
74699475Sluigi	int i;
74799475Sluigi	char sep= ' ';
74899475Sluigi
74999475Sluigi	printf(" icmptypes");
75099475Sluigi	for (i = 0; i < 32; i++) {
75199475Sluigi		if ( (cmd->d[0] & (1 << (i))) == 0)
75299475Sluigi			continue;
75399475Sluigi		printf("%c%d", sep, i);
75499475Sluigi		sep = ',';
75599475Sluigi	}
75699475Sluigi}
75799475Sluigi
75898943Sluigi/*
75998943Sluigi * show_ipfw() prints the body of an ipfw rule.
76098943Sluigi * Because the standard rule has at least proto src_ip dst_ip, we use
76198943Sluigi * a helper function to produce these entries if not provided explicitly.
762102087Sluigi * The first argument is the list of fields we have, the second is
763102087Sluigi * the list of fields we want to be printed.
764101978Sluigi *
765102087Sluigi * Special cases if we have provided a MAC header:
766102087Sluigi *   + if the rule does not contain IP addresses/ports, do not print them;
767102087Sluigi *   + if the rule does not contain an IP proto, print "all" instead of "ip";
768102087Sluigi *
769102087Sluigi * Once we have 'have_options', IP header fields are printed as options.
77098943Sluigi */
771101978Sluigi#define	HAVE_PROTO	0x0001
772101978Sluigi#define	HAVE_SRCIP	0x0002
773101978Sluigi#define	HAVE_DSTIP	0x0004
774101978Sluigi#define	HAVE_MAC	0x0008
775101978Sluigi#define	HAVE_MACTYPE	0x0010
776102087Sluigi#define	HAVE_OPTIONS	0x8000
77798943Sluigi
778101978Sluigi#define	HAVE_IP		(HAVE_PROTO | HAVE_SRCIP | HAVE_DSTIP)
77998943Sluigistatic void
780102087Sluigishow_prerequisites(int *flags, int want, int cmd)
78198943Sluigi{
782102087Sluigi	if ( (*flags & HAVE_IP) == HAVE_IP)
783102087Sluigi		*flags |= HAVE_OPTIONS;
784102087Sluigi
785102087Sluigi	if ( (*flags & (HAVE_MAC|HAVE_MACTYPE|HAVE_OPTIONS)) == HAVE_MAC &&
786102087Sluigi	     cmd != O_MAC_TYPE) {
787102087Sluigi		/*
788102087Sluigi		 * mac-type was optimized out by the compiler,
789102087Sluigi		 * restore it
790102087Sluigi		 */
791102087Sluigi		printf(" any");
792102087Sluigi		*flags |= HAVE_MACTYPE | HAVE_OPTIONS;
793102087Sluigi		return;
794101978Sluigi	}
795102087Sluigi	if ( !(*flags & HAVE_OPTIONS)) {
796102087Sluigi		if ( !(*flags & HAVE_PROTO) && (want & HAVE_PROTO))
797102087Sluigi			printf(" ip");
798102087Sluigi		if ( !(*flags & HAVE_SRCIP) && (want & HAVE_SRCIP))
799102087Sluigi			printf(" from any");
800102087Sluigi		if ( !(*flags & HAVE_DSTIP) && (want & HAVE_DSTIP))
801102087Sluigi			printf(" to any");
802102087Sluigi	}
80398943Sluigi	*flags |= want;
80498943Sluigi}
80598943Sluigi
80698943Sluigistatic void
807112189Smaximshow_ipfw(struct ip_fw *rule, int pcwidth, int bcwidth)
80898943Sluigi{
809107291Skeramida	static int twidth = 0;
81098943Sluigi	int l;
81198943Sluigi	ipfw_insn *cmd;
81298943Sluigi	int proto = 0;		/* default */
81398943Sluigi	int flags = 0;	/* prerequisites */
81498943Sluigi	ipfw_insn_log *logptr = NULL; /* set if we find an O_LOG */
81598943Sluigi	int or_block = 0;	/* we are in an or block */
81698943Sluigi
817105887Smux	u_int32_t set_disable = rule->set_disable;
818101628Sluigi
819101628Sluigi	if (set_disable & (1 << rule->set)) { /* disabled */
820101628Sluigi		if (!show_sets)
821101628Sluigi			return;
822101628Sluigi		else
823101628Sluigi			printf("# DISABLED ");
824101628Sluigi	}
82598943Sluigi	printf("%05u ", rule->rulenum);
82698943Sluigi
82798943Sluigi	if (do_acct)
828112189Smaxim		printf("%*llu %*llu ", pcwidth, rule->pcnt, bcwidth,
829112189Smaxim		    rule->bcnt);
83098943Sluigi
83198943Sluigi	if (do_time) {
832107291Skeramida		char timestr[30];
833107291Skeramida		time_t t = (time_t)0;
834107291Skeramida
835107291Skeramida		if (twidth == 0) {
836107291Skeramida			strcpy(timestr, ctime(&t));
837107291Skeramida			*strchr(timestr, '\n') = '\0';
838107291Skeramida			twidth = strlen(timestr);
839107291Skeramida		}
84098943Sluigi		if (rule->timestamp) {
841107291Skeramida			t = _long_to_time(rule->timestamp);
84298943Sluigi
84398943Sluigi			strcpy(timestr, ctime(&t));
84498943Sluigi			*strchr(timestr, '\n') = '\0';
84598943Sluigi			printf("%s ", timestr);
84698943Sluigi		} else {
847107291Skeramida			printf("%*s", twidth, " ");
84898943Sluigi		}
84998943Sluigi	}
85098943Sluigi
851101628Sluigi	if (show_sets)
852101628Sluigi		printf("set %d ", rule->set);
853101628Sluigi
85498943Sluigi	/*
855107289Sluigi	 * print the optional "match probability"
856107289Sluigi	 */
857107289Sluigi	if (rule->cmd_len > 0) {
858107289Sluigi		cmd = rule->cmd ;
859107289Sluigi		if (cmd->opcode == O_PROB) {
860107289Sluigi			ipfw_insn_u32 *p = (ipfw_insn_u32 *)cmd;
861107289Sluigi			double d = 1.0 * p->d[0];
862107289Sluigi
863107289Sluigi			d = (d / 0x7fffffff);
864107289Sluigi			printf("prob %f ", d);
865107289Sluigi		}
866107289Sluigi	}
867107289Sluigi
868107289Sluigi	/*
86998943Sluigi	 * first print actions
87098943Sluigi	 */
87198943Sluigi        for (l = rule->cmd_len - rule->act_ofs, cmd = ACTION_PTR(rule);
87298943Sluigi			l > 0 ; l -= F_LEN(cmd), cmd += F_LEN(cmd)) {
87398943Sluigi		switch(cmd->opcode) {
87498943Sluigi		case O_CHECK_STATE:
87598943Sluigi			printf("check-state");
876102087Sluigi			flags = HAVE_IP; /* avoid printing anything else */
87798943Sluigi			break;
87898943Sluigi
87998943Sluigi		case O_ACCEPT:
88098943Sluigi			printf("allow");
88198943Sluigi			break;
88298943Sluigi
88398943Sluigi		case O_COUNT:
88498943Sluigi			printf("count");
88598943Sluigi			break;
88698943Sluigi
88798943Sluigi		case O_DENY:
88898943Sluigi			printf("deny");
88998943Sluigi			break;
89098943Sluigi
89199475Sluigi		case O_REJECT:
89299475Sluigi			if (cmd->arg1 == ICMP_REJECT_RST)
89399475Sluigi				printf("reset");
89499475Sluigi			else if (cmd->arg1 == ICMP_UNREACH_HOST)
89599475Sluigi				printf("reject");
89699475Sluigi			else
89799475Sluigi				print_reject_code(cmd->arg1);
89899475Sluigi			break;
89999475Sluigi
90098943Sluigi		case O_SKIPTO:
90198943Sluigi			printf("skipto %u", cmd->arg1);
90298943Sluigi			break;
90398943Sluigi
90498943Sluigi		case O_PIPE:
90598943Sluigi			printf("pipe %u", cmd->arg1);
90698943Sluigi			break;
90798943Sluigi
90898943Sluigi		case O_QUEUE:
90998943Sluigi			printf("queue %u", cmd->arg1);
91098943Sluigi			break;
91198943Sluigi
91298943Sluigi		case O_DIVERT:
91398943Sluigi			printf("divert %u", cmd->arg1);
91498943Sluigi			break;
91598943Sluigi
91698943Sluigi		case O_TEE:
91798943Sluigi			printf("tee %u", cmd->arg1);
91898943Sluigi			break;
91998943Sluigi
92098943Sluigi		case O_FORWARD_IP:
92198943Sluigi		    {
92298943Sluigi			ipfw_insn_sa *s = (ipfw_insn_sa *)cmd;
92398943Sluigi
92498943Sluigi			printf("fwd %s", inet_ntoa(s->sa.sin_addr));
92598943Sluigi			if (s->sa.sin_port)
926103241Sluigi				printf(",%d", s->sa.sin_port);
92798943Sluigi		    }
92898943Sluigi			break;
92998943Sluigi
93098943Sluigi		case O_LOG: /* O_LOG is printed last */
93198943Sluigi			logptr = (ipfw_insn_log *)cmd;
93298943Sluigi			break;
93398943Sluigi
93498943Sluigi		default:
93598943Sluigi			printf("** unrecognized action %d len %d",
93698943Sluigi				cmd->opcode, cmd->len);
93798943Sluigi		}
93898943Sluigi	}
93998943Sluigi	if (logptr) {
94098943Sluigi		if (logptr->max_log > 0)
94199909Sluigi			printf(" log logamount %d", logptr->max_log);
94298943Sluigi		else
94399909Sluigi			printf(" log");
94498943Sluigi	}
945102087Sluigi
94698943Sluigi	/*
947102087Sluigi	 * then print the body.
94898943Sluigi	 */
949102087Sluigi	if (rule->_pad & 1) {	/* empty rules before options */
950102098Sluigi		if (!do_compact)
951102098Sluigi			printf(" ip from any to any");
952102087Sluigi		flags |= HAVE_IP | HAVE_OPTIONS;
953102087Sluigi	}
954102087Sluigi
95598943Sluigi        for (l = rule->act_ofs, cmd = rule->cmd ;
95698943Sluigi			l > 0 ; l -= F_LEN(cmd) , cmd += F_LEN(cmd)) {
95799475Sluigi		/* useful alias */
95899475Sluigi		ipfw_insn_u32 *cmd32 = (ipfw_insn_u32 *)cmd;
95998943Sluigi
960102087Sluigi		show_prerequisites(&flags, 0, cmd->opcode);
961102087Sluigi
96298943Sluigi		switch(cmd->opcode) {
963107289Sluigi		case O_PROB:
964107289Sluigi			break;	/* done already */
965107289Sluigi
96698943Sluigi		case O_PROBE_STATE:
96798943Sluigi			break; /* no need to print anything here */
96898943Sluigi
96998943Sluigi		case O_MACADDR2: {
97098943Sluigi			ipfw_insn_mac *m = (ipfw_insn_mac *)cmd;
971102087Sluigi
972102087Sluigi			if ((cmd->len & F_OR) && !or_block)
973102087Sluigi				printf(" {");
97498943Sluigi			if (cmd->len & F_NOT)
97598943Sluigi				printf(" not");
976102087Sluigi			printf(" MAC");
977102087Sluigi			flags |= HAVE_MAC;
97898943Sluigi			print_mac( m->addr, m->mask);
97998943Sluigi			print_mac( m->addr + 6, m->mask + 6);
98098943Sluigi			}
98198943Sluigi			break;
98298943Sluigi
98398943Sluigi		case O_MAC_TYPE:
984102087Sluigi			if ((cmd->len & F_OR) && !or_block)
985102087Sluigi				printf(" {");
986102087Sluigi			print_newports((ipfw_insn_u16 *)cmd, IPPROTO_ETHERTYPE,
987102087Sluigi				(flags & HAVE_OPTIONS) ? cmd->opcode : 0);
988102087Sluigi			flags |= HAVE_MAC | HAVE_MACTYPE | HAVE_OPTIONS;
98998943Sluigi			break;
99098943Sluigi
99198943Sluigi		case O_IP_SRC:
99298943Sluigi		case O_IP_SRC_MASK:
99398943Sluigi		case O_IP_SRC_ME:
99498943Sluigi		case O_IP_SRC_SET:
995102087Sluigi			show_prerequisites(&flags, HAVE_PROTO, 0);
99698943Sluigi			if (!(flags & HAVE_SRCIP))
99798943Sluigi				printf(" from");
99898943Sluigi			if ((cmd->len & F_OR) && !or_block)
99998943Sluigi				printf(" {");
1000102087Sluigi			print_ip((ipfw_insn_ip *)cmd,
1001102087Sluigi				(flags & HAVE_OPTIONS) ? " src-ip" : "");
100298943Sluigi			flags |= HAVE_SRCIP;
100398943Sluigi			break;
100498943Sluigi
100598943Sluigi		case O_IP_DST:
100698943Sluigi		case O_IP_DST_MASK:
100798943Sluigi		case O_IP_DST_ME:
100898943Sluigi		case O_IP_DST_SET:
1009102087Sluigi			show_prerequisites(&flags, HAVE_PROTO|HAVE_SRCIP, 0);
101098943Sluigi			if (!(flags & HAVE_DSTIP))
101198943Sluigi				printf(" to");
101298943Sluigi			if ((cmd->len & F_OR) && !or_block)
101398943Sluigi				printf(" {");
1014102087Sluigi			print_ip((ipfw_insn_ip *)cmd,
1015102087Sluigi				(flags & HAVE_OPTIONS) ? " dst-ip" : "");
101698943Sluigi			flags |= HAVE_DSTIP;
101798943Sluigi			break;
101898943Sluigi
101998943Sluigi		case O_IP_DSTPORT:
1020102087Sluigi			show_prerequisites(&flags, HAVE_IP, 0);
102198943Sluigi		case O_IP_SRCPORT:
1022102087Sluigi			show_prerequisites(&flags, HAVE_PROTO|HAVE_SRCIP, 0);
1023101641Sluigi			if ((cmd->len & F_OR) && !or_block)
1024101641Sluigi				printf(" {");
1025102087Sluigi			print_newports((ipfw_insn_u16 *)cmd, proto,
1026102087Sluigi				(flags & HAVE_OPTIONS) ? cmd->opcode : 0);
102798943Sluigi			break;
102898943Sluigi
102998943Sluigi		case O_PROTO: {
103098943Sluigi			struct protoent *pe;
103198943Sluigi
103298943Sluigi			if ((cmd->len & F_OR) && !or_block)
103398943Sluigi				printf(" {");
103498943Sluigi			if (cmd->len & F_NOT)
103598943Sluigi				printf(" not");
103698943Sluigi			proto = cmd->arg1;
103798943Sluigi			pe = getprotobynumber(cmd->arg1);
1038102087Sluigi			if (flags & HAVE_OPTIONS)
1039102087Sluigi				printf(" proto");
104098943Sluigi			if (pe)
104198943Sluigi				printf(" %s", pe->p_name);
104298943Sluigi			else
104398943Sluigi				printf(" %u", cmd->arg1);
104498943Sluigi			}
104598943Sluigi			flags |= HAVE_PROTO;
104698943Sluigi			break;
1047106505Smaxim
104898943Sluigi		default: /*options ... */
1049102087Sluigi			show_prerequisites(&flags, HAVE_IP | HAVE_OPTIONS, 0);
105098943Sluigi			if ((cmd->len & F_OR) && !or_block)
105198943Sluigi				printf(" {");
105298943Sluigi			if (cmd->len & F_NOT && cmd->opcode != O_IN)
105398943Sluigi				printf(" not");
105498943Sluigi			switch(cmd->opcode) {
105598943Sluigi			case O_FRAG:
105698943Sluigi				printf(" frag");
105798943Sluigi				break;
105898943Sluigi
105998943Sluigi			case O_IN:
106098943Sluigi				printf(cmd->len & F_NOT ? " out" : " in");
106198943Sluigi				break;
106298943Sluigi
106398943Sluigi			case O_LAYER2:
106498943Sluigi				printf(" layer2");
106598943Sluigi				break;
106698943Sluigi			case O_XMIT:
106798943Sluigi			case O_RECV:
106898943Sluigi			case O_VIA: {
106998943Sluigi				char *s;
107098943Sluigi				ipfw_insn_if *cmdif = (ipfw_insn_if *)cmd;
107198943Sluigi
107298943Sluigi				if (cmd->opcode == O_XMIT)
107398943Sluigi					s = "xmit";
107498943Sluigi				else if (cmd->opcode == O_RECV)
107598943Sluigi					s = "recv";
107698943Sluigi				else if (cmd->opcode == O_VIA)
107798943Sluigi					s = "via";
107898943Sluigi				if (cmdif->name[0] == '\0')
107999475Sluigi					printf(" %s %s", s,
108099475Sluigi					    inet_ntoa(cmdif->p.ip));
108198943Sluigi				else if (cmdif->p.unit == -1)
108298943Sluigi					printf(" %s %s*", s, cmdif->name);
108398943Sluigi				else
108499475Sluigi					printf(" %s %s%d", s, cmdif->name,
108599475Sluigi					    cmdif->p.unit);
108698943Sluigi				}
108798943Sluigi				break;
108898943Sluigi
108998943Sluigi			case O_IPID:
109098943Sluigi				printf(" ipid %u", cmd->arg1 );
109198943Sluigi				break;
109298943Sluigi
109398943Sluigi			case O_IPTTL:
109498943Sluigi				printf(" ipttl %u", cmd->arg1 );
109598943Sluigi				break;
109698943Sluigi
109798943Sluigi			case O_IPVER:
109898943Sluigi				printf(" ipver %u", cmd->arg1 );
109998943Sluigi				break;
110098943Sluigi
110199475Sluigi			case O_IPPRECEDENCE:
110299475Sluigi				printf(" ipprecedence %u", (cmd->arg1) >> 5 );
110399475Sluigi				break;
110499475Sluigi
110598943Sluigi			case O_IPLEN:
110698943Sluigi				printf(" iplen %u", cmd->arg1 );
110798943Sluigi				break;
110898943Sluigi
1109101116Sluigi			case O_IPOPT:
111098943Sluigi				print_flags("ipoptions", cmd, f_ipopts);
111198943Sluigi				break;
111298943Sluigi
111399475Sluigi			case O_IPTOS:
111499475Sluigi				print_flags("iptos", cmd, f_iptos);
111599475Sluigi				break;
111699475Sluigi
111799475Sluigi			case O_ICMPTYPE:
111899475Sluigi				print_icmptypes((ipfw_insn_u32 *)cmd);
111999475Sluigi				break;
112099475Sluigi
112198943Sluigi			case O_ESTAB:
112298943Sluigi				printf(" established");
112398943Sluigi				break;
112498943Sluigi
112598943Sluigi			case O_TCPFLAGS:
112698943Sluigi				print_flags("tcpflags", cmd, f_tcpflags);
112798943Sluigi				break;
112898943Sluigi
112998943Sluigi			case O_TCPOPTS:
113098943Sluigi				print_flags("tcpoptions", cmd, f_tcpopts);
113198943Sluigi				break;
113298943Sluigi
113398943Sluigi			case O_TCPWIN:
113498943Sluigi				printf(" tcpwin %d", ntohs(cmd->arg1));
113598943Sluigi				break;
113698943Sluigi
113798943Sluigi			case O_TCPACK:
113898943Sluigi				printf(" tcpack %d", ntohl(cmd32->d[0]));
113998943Sluigi				break;
114098943Sluigi
114198943Sluigi			case O_TCPSEQ:
114298943Sluigi				printf(" tcpseq %d", ntohl(cmd32->d[0]));
114398943Sluigi				break;
114498943Sluigi
114598943Sluigi			case O_UID:
114698943Sluigi			    {
114798943Sluigi				struct passwd *pwd = getpwuid(cmd32->d[0]);
114898943Sluigi
114998943Sluigi				if (pwd)
115098943Sluigi					printf(" uid %s", pwd->pw_name);
115198943Sluigi				else
115298943Sluigi					printf(" uid %u", cmd32->d[0]);
115398943Sluigi			    }
115498943Sluigi				break;
115598943Sluigi
115698943Sluigi			case O_GID:
115798943Sluigi			    {
115898943Sluigi				struct group *grp = getgrgid(cmd32->d[0]);
115998943Sluigi
116098943Sluigi				if (grp)
116198943Sluigi					printf(" gid %s", grp->gr_name);
116298943Sluigi				else
116398943Sluigi					printf(" gid %u", cmd32->d[0]);
116498943Sluigi			    }
116598943Sluigi				break;
116698943Sluigi
1167112250Scjc			case O_VERREVPATH:
1168112250Scjc				printf(" verrevpath");
1169112250Scjc				break;
1170112250Scjc
117198943Sluigi			case O_KEEP_STATE:
117298943Sluigi				printf(" keep-state");
117398943Sluigi				break;
117498943Sluigi
117598943Sluigi			case O_LIMIT:
117698943Sluigi			    {
117798943Sluigi				struct _s_x *p = limit_masks;
117898943Sluigi				ipfw_insn_limit *c = (ipfw_insn_limit *)cmd;
117998943Sluigi				u_int8_t x = c->limit_mask;
118098943Sluigi				char *comma = " ";
118198943Sluigi
118298943Sluigi				printf(" limit");
1183106505Smaxim				for ( ; p->x != 0 ; p++)
118499909Sluigi					if ((x & p->x) == p->x) {
118598943Sluigi						x &= ~p->x;
118698943Sluigi						printf("%s%s", comma, p->s);
118798943Sluigi						comma = ",";
118898943Sluigi					}
118998943Sluigi				printf(" %d", c->conn_limit);
119098943Sluigi			    }
119198943Sluigi				break;
119298943Sluigi
119398943Sluigi			default:
119498943Sluigi				printf(" [opcode %d len %d]",
119598943Sluigi				    cmd->opcode, cmd->len);
119698943Sluigi			}
119798943Sluigi		}
119898943Sluigi		if (cmd->len & F_OR) {
119998943Sluigi			printf(" or");
120098943Sluigi			or_block = 1;
120198943Sluigi		} else if (or_block) {
120298943Sluigi			printf(" }");
120398943Sluigi			or_block = 0;
120498943Sluigi		}
120598943Sluigi	}
1206102087Sluigi	show_prerequisites(&flags, HAVE_IP, 0);
120798943Sluigi
120898943Sluigi	printf("\n");
120998943Sluigi}
121098943Sluigi
121198943Sluigistatic void
1212112189Smaximshow_dyn_ipfw(ipfw_dyn_rule *d, int pcwidth, int bcwidth)
121398943Sluigi{
121498943Sluigi	struct protoent *pe;
121598943Sluigi	struct in_addr a;
121698943Sluigi
121798943Sluigi	if (!do_expired) {
121898943Sluigi		if (!d->expire && !(d->dyn_type == O_LIMIT_PARENT))
121998943Sluigi			return;
122098943Sluigi	}
122198943Sluigi
1222112189Smaxim	printf("%05d %*llu %*llu (%ds)", d->rulenum, pcwidth, d->pcnt, bcwidth,
1223112189Smaxim	    d->bcnt, d->expire);
122498943Sluigi	switch (d->dyn_type) {
122598943Sluigi	case O_LIMIT_PARENT:
122698943Sluigi		printf(" PARENT %d", d->count);
122798943Sluigi		break;
122898943Sluigi	case O_LIMIT:
122998943Sluigi		printf(" LIMIT");
123098943Sluigi		break;
123198943Sluigi	case O_KEEP_STATE: /* bidir, no mask */
1232106505Smaxim		printf(" STATE");
123398943Sluigi		break;
123498943Sluigi	}
123598943Sluigi
123698943Sluigi	if ((pe = getprotobynumber(d->id.proto)) != NULL)
123798943Sluigi		printf(" %s", pe->p_name);
123898943Sluigi	else
123998943Sluigi		printf(" proto %u", d->id.proto);
124098943Sluigi
124198943Sluigi	a.s_addr = htonl(d->id.src_ip);
124298943Sluigi	printf(" %s %d", inet_ntoa(a), d->id.src_port);
124398943Sluigi
124498943Sluigi	a.s_addr = htonl(d->id.dst_ip);
124598943Sluigi	printf(" <-> %s %d", inet_ntoa(a), d->id.dst_port);
124698943Sluigi	printf("\n");
124798943Sluigi}
124898943Sluigi
124998943Sluigiint
125098943Sluigisort_q(const void *pa, const void *pb)
125198943Sluigi{
125298943Sluigi	int rev = (do_sort < 0);
125398943Sluigi	int field = rev ? -do_sort : do_sort;
125498943Sluigi	long long res = 0;
125598943Sluigi	const struct dn_flow_queue *a = pa;
125698943Sluigi	const struct dn_flow_queue *b = pb;
125798943Sluigi
125898943Sluigi	switch (field) {
125998943Sluigi	case 1: /* pkts */
126098943Sluigi		res = a->len - b->len;
126198943Sluigi		break;
126298943Sluigi	case 2: /* bytes */
126398943Sluigi		res = a->len_bytes - b->len_bytes;
126498943Sluigi		break;
126598943Sluigi
126698943Sluigi	case 3: /* tot pkts */
126798943Sluigi		res = a->tot_pkts - b->tot_pkts;
126898943Sluigi		break;
126998943Sluigi
127098943Sluigi	case 4: /* tot bytes */
127198943Sluigi		res = a->tot_bytes - b->tot_bytes;
127298943Sluigi		break;
127398943Sluigi	}
127498943Sluigi	if (res < 0)
127598943Sluigi		res = -1;
127698943Sluigi	if (res > 0)
127798943Sluigi		res = 1;
127898943Sluigi	return (int)(rev ? res : -res);
127998943Sluigi}
128098943Sluigi
128198943Sluigistatic void
128298943Sluigilist_queues(struct dn_flow_set *fs, struct dn_flow_queue *q)
128398943Sluigi{
128498943Sluigi	int l;
128598943Sluigi
128698943Sluigi	printf("    mask: 0x%02x 0x%08x/0x%04x -> 0x%08x/0x%04x\n",
128798943Sluigi	    fs->flow_mask.proto,
128898943Sluigi	    fs->flow_mask.src_ip, fs->flow_mask.src_port,
128998943Sluigi	    fs->flow_mask.dst_ip, fs->flow_mask.dst_port);
129098943Sluigi	if (fs->rq_elements == 0)
129198943Sluigi		return;
129298943Sluigi
129398943Sluigi	printf("BKT Prot ___Source IP/port____ "
129498943Sluigi	    "____Dest. IP/port____ Tot_pkt/bytes Pkt/Byte Drp\n");
129598943Sluigi	if (do_sort != 0)
129698943Sluigi		heapsort(q, fs->rq_elements, sizeof *q, sort_q);
129798943Sluigi	for (l = 0; l < fs->rq_elements; l++) {
129898943Sluigi		struct in_addr ina;
129998943Sluigi		struct protoent *pe;
130098943Sluigi
130198943Sluigi		ina.s_addr = htonl(q[l].id.src_ip);
130298943Sluigi		printf("%3d ", q[l].hash_slot);
130398943Sluigi		pe = getprotobynumber(q[l].id.proto);
130498943Sluigi		if (pe)
130598943Sluigi			printf("%-4s ", pe->p_name);
130698943Sluigi		else
130798943Sluigi			printf("%4u ", q[l].id.proto);
130898943Sluigi		printf("%15s/%-5d ",
130998943Sluigi		    inet_ntoa(ina), q[l].id.src_port);
131098943Sluigi		ina.s_addr = htonl(q[l].id.dst_ip);
131198943Sluigi		printf("%15s/%-5d ",
131298943Sluigi		    inet_ntoa(ina), q[l].id.dst_port);
131398943Sluigi		printf("%4qu %8qu %2u %4u %3u\n",
131498943Sluigi		    q[l].tot_pkts, q[l].tot_bytes,
131598943Sluigi		    q[l].len, q[l].len_bytes, q[l].drops);
131698943Sluigi		if (verbose)
131798943Sluigi			printf("   S %20qd  F %20qd\n",
131898943Sluigi			    q[l].S, q[l].F);
131998943Sluigi	}
132098943Sluigi}
132198943Sluigi
132298943Sluigistatic void
132398943Sluigiprint_flowset_parms(struct dn_flow_set *fs, char *prefix)
132498943Sluigi{
132598943Sluigi	int l;
132698943Sluigi	char qs[30];
132798943Sluigi	char plr[30];
132898943Sluigi	char red[90];	/* Display RED parameters */
132998943Sluigi
133098943Sluigi	l = fs->qsize;
133198943Sluigi	if (fs->flags_fs & DN_QSIZE_IS_BYTES) {
133298943Sluigi		if (l >= 8192)
133398943Sluigi			sprintf(qs, "%d KB", l / 1024);
133498943Sluigi		else
133598943Sluigi			sprintf(qs, "%d B", l);
133698943Sluigi	} else
133798943Sluigi		sprintf(qs, "%3d sl.", l);
133898943Sluigi	if (fs->plr)
133998943Sluigi		sprintf(plr, "plr %f", 1.0 * fs->plr / (double)(0x7fffffff));
134098943Sluigi	else
134198943Sluigi		plr[0] = '\0';
134298943Sluigi	if (fs->flags_fs & DN_IS_RED)	/* RED parameters */
134398943Sluigi		sprintf(red,
134498943Sluigi		    "\n\t  %cRED w_q %f min_th %d max_th %d max_p %f",
134598943Sluigi		    (fs->flags_fs & DN_IS_GENTLE_RED) ? 'G' : ' ',
134698943Sluigi		    1.0 * fs->w_q / (double)(1 << SCALE_RED),
134798943Sluigi		    SCALE_VAL(fs->min_th),
134898943Sluigi		    SCALE_VAL(fs->max_th),
134998943Sluigi		    1.0 * fs->max_p / (double)(1 << SCALE_RED));
135098943Sluigi	else
135198943Sluigi		sprintf(red, "droptail");
135298943Sluigi
135398943Sluigi	printf("%s %s%s %d queues (%d buckets) %s\n",
135498943Sluigi	    prefix, qs, plr, fs->rq_elements, fs->rq_size, red);
135598943Sluigi}
135698943Sluigi
135798943Sluigistatic void
135898943Sluigilist_pipes(void *data, int nbytes, int ac, char *av[])
135998943Sluigi{
136098943Sluigi	u_long rulenum;
136198943Sluigi	void *next = data;
136298943Sluigi	struct dn_pipe *p = (struct dn_pipe *) data;
136398943Sluigi	struct dn_flow_set *fs;
136498943Sluigi	struct dn_flow_queue *q;
136598943Sluigi	int l;
136698943Sluigi
136798943Sluigi	if (ac > 0)
136898943Sluigi		rulenum = strtoul(*av++, NULL, 10);
136998943Sluigi	else
137098943Sluigi		rulenum = 0;
137198943Sluigi	for (; nbytes >= sizeof *p; p = (struct dn_pipe *)next) {
137298943Sluigi		double b = p->bandwidth;
137398943Sluigi		char buf[30];
137498943Sluigi		char prefix[80];
137598943Sluigi
137698943Sluigi		if (p->next != (struct dn_pipe *)DN_IS_PIPE)
137798943Sluigi			break;	/* done with pipes, now queues */
137898943Sluigi
137998943Sluigi		/*
138098943Sluigi		 * compute length, as pipe have variable size
138198943Sluigi		 */
138298943Sluigi		l = sizeof(*p) + p->fs.rq_elements * sizeof(*q);
138398943Sluigi		next = (void *)p + l;
138498943Sluigi		nbytes -= l;
138598943Sluigi
138698943Sluigi		if (rulenum != 0 && rulenum != p->pipe_nr)
138798943Sluigi			continue;
138898943Sluigi
138998943Sluigi		/*
139098943Sluigi		 * Print rate (or clocking interface)
139198943Sluigi		 */
139298943Sluigi		if (p->if_name[0] != '\0')
139398943Sluigi			sprintf(buf, "%s", p->if_name);
139498943Sluigi		else if (b == 0)
139598943Sluigi			sprintf(buf, "unlimited");
139698943Sluigi		else if (b >= 1000000)
139798943Sluigi			sprintf(buf, "%7.3f Mbit/s", b/1000000);
139898943Sluigi		else if (b >= 1000)
139998943Sluigi			sprintf(buf, "%7.3f Kbit/s", b/1000);
140098943Sluigi		else
140198943Sluigi			sprintf(buf, "%7.3f bit/s ", b);
140298943Sluigi
140398943Sluigi		sprintf(prefix, "%05d: %s %4d ms ",
140498943Sluigi		    p->pipe_nr, buf, p->delay);
140598943Sluigi		print_flowset_parms(&(p->fs), prefix);
140698943Sluigi		if (verbose)
140798943Sluigi			printf("   V %20qd\n", p->V >> MY_M);
1408106505Smaxim
140998943Sluigi		q = (struct dn_flow_queue *)(p+1);
141098943Sluigi		list_queues(&(p->fs), q);
141198943Sluigi	}
141298943Sluigi	for (fs = next; nbytes >= sizeof *fs; fs = next) {
141398943Sluigi		char prefix[80];
141498943Sluigi
141598943Sluigi		if (fs->next != (struct dn_flow_set *)DN_IS_QUEUE)
141698943Sluigi			break;
141798943Sluigi		l = sizeof(*fs) + fs->rq_elements * sizeof(*q);
141898943Sluigi		next = (void *)fs + l;
141998943Sluigi		nbytes -= l;
142098943Sluigi		q = (struct dn_flow_queue *)(fs+1);
142198943Sluigi		sprintf(prefix, "q%05d: weight %d pipe %d ",
142298943Sluigi		    fs->fs_nr, fs->weight, fs->parent_nr);
142398943Sluigi		print_flowset_parms(fs, prefix);
142498943Sluigi		list_queues(fs, q);
142598943Sluigi	}
142698943Sluigi}
142798943Sluigi
1428101978Sluigi/*
1429101978Sluigi * This one handles all set-related commands
1430101978Sluigi * 	ipfw set { show | enable | disable }
1431101978Sluigi * 	ipfw set swap X Y
1432101978Sluigi * 	ipfw set move X to Y
1433101978Sluigi * 	ipfw set move rule X to Y
1434101978Sluigi */
143598943Sluigistatic void
1436101978Sluigisets_handler(int ac, char *av[])
1437101978Sluigi{
1438101978Sluigi	u_int32_t set_disable, masks[2];
1439101978Sluigi	int i, nbytes;
1440101978Sluigi	u_int16_t rulenum;
1441101978Sluigi	u_int8_t cmd, new_set;
1442101978Sluigi
1443101978Sluigi	ac--;
1444101978Sluigi	av++;
1445101978Sluigi
1446101978Sluigi	if (!ac)
1447101978Sluigi		errx(EX_USAGE, "set needs command");
1448101978Sluigi	if (!strncmp(*av, "show", strlen(*av)) ) {
1449101978Sluigi		void *data;
1450101978Sluigi		char *msg;
1451101978Sluigi
1452101978Sluigi		nbytes = sizeof(struct ip_fw);
1453101978Sluigi		if ((data = malloc(nbytes)) == NULL)
1454101978Sluigi			err(EX_OSERR, "malloc");
1455101978Sluigi		if (getsockopt(s, IPPROTO_IP, IP_FW_GET, data, &nbytes) < 0)
1456101978Sluigi			err(EX_OSERR, "getsockopt(IP_FW_GET)");
1457105887Smux		set_disable = ((struct ip_fw *)data)->set_disable;
1458101978Sluigi
1459101978Sluigi		for (i = 0, msg = "disable" ; i < 31; i++)
1460101978Sluigi			if (  (set_disable & (1<<i))) {
1461101978Sluigi				printf("%s %d", msg, i);
1462101978Sluigi				msg = "";
1463101978Sluigi			}
1464101978Sluigi		msg = (set_disable) ? " enable" : "enable";
1465101978Sluigi		for (i = 0; i < 31; i++)
1466101978Sluigi			if ( !(set_disable & (1<<i))) {
1467101978Sluigi				printf("%s %d", msg, i);
1468101978Sluigi				msg = "";
1469101978Sluigi			}
1470101978Sluigi		printf("\n");
1471101978Sluigi	} else if (!strncmp(*av, "swap", strlen(*av))) {
1472101978Sluigi		ac--; av++;
1473101978Sluigi		if (ac != 2)
1474101978Sluigi			errx(EX_USAGE, "set swap needs 2 set numbers\n");
1475101978Sluigi		rulenum = atoi(av[0]);
1476101978Sluigi		new_set = atoi(av[1]);
1477101978Sluigi		if (!isdigit(*(av[0])) || rulenum > 30)
1478101978Sluigi			errx(EX_DATAERR, "invalid set number %s\n", av[0]);
1479101978Sluigi		if (!isdigit(*(av[1])) || new_set > 30)
1480101978Sluigi			errx(EX_DATAERR, "invalid set number %s\n", av[1]);
1481101978Sluigi		masks[0] = (4 << 24) | (new_set << 16) | (rulenum);
1482101978Sluigi		i = setsockopt(s, IPPROTO_IP, IP_FW_DEL,
1483101978Sluigi			masks, sizeof(u_int32_t));
1484101978Sluigi	} else if (!strncmp(*av, "move", strlen(*av))) {
1485101978Sluigi		ac--; av++;
1486101978Sluigi		if (ac && !strncmp(*av, "rule", strlen(*av))) {
1487101978Sluigi			cmd = 2;
1488101978Sluigi			ac--; av++;
1489101978Sluigi		} else
1490101978Sluigi			cmd = 3;
1491101978Sluigi		if (ac != 3 || strncmp(av[1], "to", strlen(*av)))
1492101978Sluigi			errx(EX_USAGE, "syntax: set move [rule] X to Y\n");
1493101978Sluigi		rulenum = atoi(av[0]);
1494101978Sluigi		new_set = atoi(av[2]);
1495101978Sluigi		if (!isdigit(*(av[0])) || (cmd == 3 && rulenum > 30) ||
1496101978Sluigi			(cmd == 2 && rulenum == 65535) )
1497101978Sluigi			errx(EX_DATAERR, "invalid source number %s\n", av[0]);
1498101978Sluigi		if (!isdigit(*(av[2])) || new_set > 30)
1499101978Sluigi			errx(EX_DATAERR, "invalid dest. set %s\n", av[1]);
1500101978Sluigi		masks[0] = (cmd << 24) | (new_set << 16) | (rulenum);
1501101978Sluigi		i = setsockopt(s, IPPROTO_IP, IP_FW_DEL,
1502101978Sluigi			masks, sizeof(u_int32_t));
1503101978Sluigi	} else if (!strncmp(*av, "disable", strlen(*av)) ||
1504101978Sluigi		   !strncmp(*av, "enable",  strlen(*av)) ) {
1505101978Sluigi		int which = !strncmp(*av, "enable",  strlen(*av)) ? 1 : 0;
1506101978Sluigi
1507101978Sluigi		ac--; av++;
1508101978Sluigi		masks[0] = masks[1] = 0;
1509101978Sluigi
1510101978Sluigi		while (ac) {
1511101978Sluigi			if (isdigit(**av)) {
1512101978Sluigi				i = atoi(*av);
1513101978Sluigi				if (i < 0 || i > 30)
1514101978Sluigi					errx(EX_DATAERR,
1515101978Sluigi					    "invalid set number %d\n", i);
1516101978Sluigi				masks[which] |= (1<<i);
1517101978Sluigi			} else if (!strncmp(*av, "disable", strlen(*av)))
1518101978Sluigi				which = 0;
1519101978Sluigi			else if (!strncmp(*av, "enable", strlen(*av)))
1520101978Sluigi				which = 1;
1521101978Sluigi			else
1522101978Sluigi				errx(EX_DATAERR,
1523101978Sluigi					"invalid set command %s\n", *av);
1524101978Sluigi			av++; ac--;
1525101978Sluigi		}
1526101978Sluigi		if ( (masks[0] & masks[1]) != 0 )
1527101978Sluigi			errx(EX_DATAERR,
1528101978Sluigi			    "cannot enable and disable the same set\n");
1529101978Sluigi
1530101978Sluigi		i = setsockopt(s, IPPROTO_IP, IP_FW_DEL, masks, sizeof(masks));
1531101978Sluigi		if (i)
1532101978Sluigi			warn("set enable/disable: setsockopt(IP_FW_DEL)");
1533101978Sluigi	} else
1534101978Sluigi		errx(EX_USAGE, "invalid set command %s\n", *av);
1535101978Sluigi}
1536101978Sluigi
1537101978Sluigistatic void
1538109126Sdillonsysctl_handler(int ac, char *av[], int which)
1539109126Sdillon{
1540109126Sdillon	ac--;
1541109126Sdillon	av++;
1542109126Sdillon
1543109126Sdillon	if (*av == NULL) {
1544109126Sdillon		warnx("missing keyword to enable/disable\n");
1545109126Sdillon	} else if (strncmp(*av, "firewall", strlen(*av)) == 0) {
1546109126Sdillon		sysctlbyname("net.inet.ip.fw.enable", NULL, 0, &which, sizeof(which));
1547109126Sdillon	} else if (strncmp(*av, "one_pass", strlen(*av)) == 0) {
1548109126Sdillon		sysctlbyname("net.inet.ip.fw.one_pass", NULL, 0, &which, sizeof(which));
1549109126Sdillon	} else if (strncmp(*av, "debug", strlen(*av)) == 0) {
1550109126Sdillon		sysctlbyname("net.inet.ip.fw.debug", NULL, 0, &which, sizeof(which));
1551109126Sdillon	} else if (strncmp(*av, "verbose", strlen(*av)) == 0) {
1552109126Sdillon		sysctlbyname("net.inet.ip.fw.verbose", NULL, 0, &which, sizeof(which));
1553109126Sdillon	} else if (strncmp(*av, "dyn_keepalive", strlen(*av)) == 0) {
1554109126Sdillon		sysctlbyname("net.inet.ip.fw.dyn_keepalive", NULL, 0, &which, sizeof(which));
1555109126Sdillon	} else {
1556109126Sdillon		warnx("unrecognize enable/disable keyword: %s\n", *av);
1557109126Sdillon	}
1558109126Sdillon}
1559109126Sdillon
1560109126Sdillonstatic void
156198943Sluigilist(int ac, char *av[])
156298943Sluigi{
156398943Sluigi	struct ip_fw *r;
156498943Sluigi	ipfw_dyn_rule *dynrules, *d;
156598943Sluigi
156698943Sluigi	void *lim, *data = NULL;
1567112189Smaxim	int bcwidth, n, nbytes, nstat, ndyn, pcwidth, width;
156898943Sluigi	int exitval = EX_OK;
156998943Sluigi	int lac;
157098943Sluigi	char **lav;
157198943Sluigi	u_long rnum;
157298943Sluigi	char *endptr;
157398943Sluigi	int seen = 0;
157498943Sluigi
157598943Sluigi	const int ocmd = do_pipe ? IP_DUMMYNET_GET : IP_FW_GET;
157698943Sluigi	int nalloc = 1024;	/* start somewhere... */
157798943Sluigi
157898943Sluigi	ac--;
157998943Sluigi	av++;
158098943Sluigi
158198943Sluigi	/* get rules or pipes from kernel, resizing array as necessary */
158298943Sluigi	nbytes = nalloc;
158398943Sluigi
158498943Sluigi	while (nbytes >= nalloc) {
158598943Sluigi		nalloc = nalloc * 2 + 200;
158698943Sluigi		nbytes = nalloc;
158798943Sluigi		if ((data = realloc(data, nbytes)) == NULL)
158898943Sluigi			err(EX_OSERR, "realloc");
158998943Sluigi		if (getsockopt(s, IPPROTO_IP, ocmd, data, &nbytes) < 0)
159098943Sluigi			err(EX_OSERR, "getsockopt(IP_%s_GET)",
159198943Sluigi				do_pipe ? "DUMMYNET" : "FW");
159298943Sluigi	}
159398943Sluigi
159498943Sluigi	if (do_pipe) {
159598943Sluigi		list_pipes(data, nbytes, ac, av);
159698943Sluigi		goto done;
159798943Sluigi	}
159898943Sluigi
159998943Sluigi	/*
160098943Sluigi	 * Count static rules. They have variable size so we
160198943Sluigi	 * need to scan the list to count them.
160298943Sluigi	 */
160398943Sluigi	for (nstat = 1, r = data, lim = data + nbytes;
160498943Sluigi		    r->rulenum < 65535 && (void *)r < lim;
160598943Sluigi		    ++nstat, r = (void *)r + RULESIZE(r) )
160698943Sluigi		; /* nothing */
160798943Sluigi
160898943Sluigi	/*
160998943Sluigi	 * Count dynamic rules. This is easier as they have
161098943Sluigi	 * fixed size.
161198943Sluigi	 */
161298943Sluigi	r = (void *)r + RULESIZE(r);
161398943Sluigi	dynrules = (ipfw_dyn_rule *)r ;
161498943Sluigi	n = (void *)r - data;
161598943Sluigi	ndyn = (nbytes - n) / sizeof *dynrules;
161698943Sluigi
1617112189Smaxim	/* if showing stats, figure out column widths ahead of time */
1618112189Smaxim	bcwidth = pcwidth = 0;
1619112189Smaxim	if (do_acct) {
1620112189Smaxim		for (n = 0, r = data; n < nstat;
1621112189Smaxim		    n++, r = (void *)r + RULESIZE(r)) {
1622112189Smaxim			/* packet counter */
1623112189Smaxim			width = snprintf(NULL, 0, "%llu", r->pcnt);
1624112189Smaxim			if (width > pcwidth)
1625112189Smaxim				pcwidth = width;
1626112189Smaxim
1627112189Smaxim			/* byte counter */
1628112189Smaxim			width = snprintf(NULL, 0, "%llu", r->bcnt);
1629112189Smaxim			if (width > bcwidth)
1630112189Smaxim				bcwidth = width;
1631112189Smaxim		}
1632112189Smaxim	}
1633112189Smaxim	if (do_dynamic && ndyn) {
1634112189Smaxim		for (n = 0, d = dynrules; n < ndyn; n++, d++) {
1635112189Smaxim			width = snprintf(NULL, 0, "%llu", d->pcnt);
1636112189Smaxim			if (width > pcwidth)
1637112189Smaxim				pcwidth = width;
1638112189Smaxim
1639112189Smaxim			width = snprintf(NULL, 0, "%llu", d->bcnt);
1640112189Smaxim			if (width > bcwidth)
1641112189Smaxim				bcwidth = width;
1642112189Smaxim		}
1643112189Smaxim	}
164498943Sluigi	/* if no rule numbers were specified, list all rules */
164598943Sluigi	if (ac == 0) {
164698943Sluigi		for (n = 0, r = data; n < nstat;
164798943Sluigi		    n++, r = (void *)r + RULESIZE(r) )
1648112189Smaxim			show_ipfw(r, pcwidth, bcwidth);
164998943Sluigi
165098943Sluigi		if (do_dynamic && ndyn) {
165198943Sluigi			printf("## Dynamic rules (%d):\n", ndyn);
165298943Sluigi			for (n = 0, d = dynrules; n < ndyn; n++, d++)
1653112189Smaxim				show_dyn_ipfw(d, pcwidth, bcwidth);
165498943Sluigi		}
165598943Sluigi		goto done;
165698943Sluigi	}
165798943Sluigi
165898943Sluigi	/* display specific rules requested on command line */
165998943Sluigi
166098943Sluigi	for (lac = ac, lav = av; lac != 0; lac--) {
166198943Sluigi		/* convert command line rule # */
166298943Sluigi		rnum = strtoul(*lav++, &endptr, 10);
166398943Sluigi		if (*endptr) {
166498943Sluigi			exitval = EX_USAGE;
166598943Sluigi			warnx("invalid rule number: %s", *(lav - 1));
166698943Sluigi			continue;
166798943Sluigi		}
166898943Sluigi		for (n = seen = 0, r = data; n < nstat;
166998943Sluigi		    n++, r = (void *)r + RULESIZE(r) ) {
167098943Sluigi			if (r->rulenum > rnum)
167198943Sluigi				break;
167298943Sluigi			if (r->rulenum == rnum) {
1673112189Smaxim				show_ipfw(r, pcwidth, bcwidth);
167498943Sluigi				seen = 1;
167598943Sluigi			}
167698943Sluigi		}
167798943Sluigi		if (!seen) {
167898943Sluigi			/* give precedence to other error(s) */
167998943Sluigi			if (exitval == EX_OK)
168098943Sluigi				exitval = EX_UNAVAILABLE;
168198943Sluigi			warnx("rule %lu does not exist", rnum);
168298943Sluigi		}
168398943Sluigi	}
168498943Sluigi
168598943Sluigi	if (do_dynamic && ndyn) {
168698943Sluigi		printf("## Dynamic rules:\n");
168798943Sluigi		for (lac = ac, lav = av; lac != 0; lac--) {
168898943Sluigi			rnum = strtoul(*lav++, &endptr, 10);
168998943Sluigi			if (*endptr)
169098943Sluigi				/* already warned */
169198943Sluigi				continue;
169298943Sluigi			for (n = 0, d = dynrules; n < ndyn; n++, d++) {
1693105887Smux				if (d->rulenum > rnum)
169498943Sluigi					break;
1695105887Smux				if (d->rulenum == rnum)
1696112189Smaxim					show_dyn_ipfw(d, pcwidth, bcwidth);
169798943Sluigi			}
169898943Sluigi		}
169998943Sluigi	}
170098943Sluigi
170198943Sluigi	ac = 0;
170298943Sluigi
170398943Sluigidone:
170498943Sluigi	free(data);
170598943Sluigi
170698943Sluigi	if (exitval != EX_OK)
170798943Sluigi		exit(exitval);
170898943Sluigi}
170998943Sluigi
171098943Sluigistatic void
171198943Sluigishow_usage(void)
171298943Sluigi{
171398943Sluigi	fprintf(stderr, "usage: ipfw [options]\n"
171498943Sluigi"    add [number] rule\n"
171598943Sluigi"    pipe number config [pipeconfig]\n"
171698943Sluigi"    queue number config [queueconfig]\n"
171798943Sluigi"    [pipe] flush\n"
171898943Sluigi"    [pipe] delete number ...\n"
171998943Sluigi"    [pipe] {list|show} [number ...]\n"
172098943Sluigi"    {zero|resetlog} [number ...]\n"
172198943Sluigi"do \"ipfw -h\" or see ipfw manpage for details\n"
172298943Sluigi);
172398943Sluigi
172498943Sluigi	exit(EX_USAGE);
172598943Sluigi}
172698943Sluigi
172798943Sluigistatic void
172898943Sluigihelp(void)
172998943Sluigi{
1730106505Smaxim
173198943Sluigi	fprintf(stderr, "ipfw syntax summary:\n"
173298943Sluigi"ipfw add [N] [prob {0..1}] ACTION [log [logamount N]] ADDR OPTIONS\n"
173398943Sluigi"ipfw {pipe|queue} N config BODY\n"
173498943Sluigi"ipfw [pipe] {zero|delete|show} [N{,N}]\n"
173598943Sluigi"\n"
173698943Sluigi"RULE:		[1..] [PROB] BODY\n"
173798943Sluigi"RULENUM:	INTEGER(1..65534)\n"
173898943Sluigi"PROB:		prob REAL(0..1)\n"
173998943Sluigi"BODY:		check-state [LOG] (no body) |\n"
174098943Sluigi"		ACTION [LOG] MATCH_ADDR [OPTION_LIST]\n"
174198943Sluigi"ACTION:	check-state | allow | count | deny | reject | skipto N |\n"
174298943Sluigi"		{divert|tee} PORT | forward ADDR | pipe N | queue N\n"
174398943Sluigi"ADDR:		[ MAC dst src ether_type ] \n"
174498943Sluigi"		[ from IPLIST [ PORT ] to IPLIST [ PORTLIST ] ]\n"
174598943Sluigi"IPLIST:	IPADDR | ( IPADDR or ... or IPADDR )\n"
174698943Sluigi"IPADDR:	[not] { any | me | ip | ip/bits | ip:mask | ip/bits{x,y,z} }\n"
174798943Sluigi"OPTION_LIST:	OPTION [,OPTION_LIST]\n"
174898943Sluigi);
174998943Sluigiexit(0);
175098943Sluigi}
175198943Sluigi
175298943Sluigi
175398943Sluigistatic int
175498943Sluigilookup_host (char *host, struct in_addr *ipaddr)
175598943Sluigi{
175698943Sluigi	struct hostent *he;
175798943Sluigi
175898943Sluigi	if (!inet_aton(host, ipaddr)) {
175998943Sluigi		if ((he = gethostbyname(host)) == NULL)
176098943Sluigi			return(-1);
176198943Sluigi		*ipaddr = *(struct in_addr *)he->h_addr_list[0];
176298943Sluigi	}
176398943Sluigi	return(0);
176498943Sluigi}
176598943Sluigi
176698943Sluigi/*
176798943Sluigi * fills the addr and mask fields in the instruction as appropriate from av.
176898943Sluigi * Update length as appropriate.
176998943Sluigi * The following formats are allowed:
177098943Sluigi *	any	matches any IP. Actually returns an empty instruction.
177198943Sluigi *	me	returns O_IP_*_ME
177298943Sluigi *	1.2.3.4		single IP address
177398943Sluigi *	1.2.3.4:5.6.7.8	address:mask
177498943Sluigi *	1.2.3.4/24	address/mask
177598943Sluigi *	1.2.3.4/26{1,6,5,4,23}	set of addresses in a subnet
177698943Sluigi */
177798943Sluigistatic void
177898943Sluigifill_ip(ipfw_insn_ip *cmd, char *av)
177998943Sluigi{
178098943Sluigi	char *p = 0, md = 0;
178198943Sluigi	u_int32_t i;
178298943Sluigi
178398943Sluigi	cmd->o.len &= ~F_LEN_MASK;	/* zero len */
178498943Sluigi
178598943Sluigi	if (!strncmp(av, "any", strlen(av)))
178698943Sluigi		return;
178798943Sluigi
178898943Sluigi	if (!strncmp(av, "me", strlen(av))) {
178998943Sluigi		cmd->o.len |= F_INSN_SIZE(ipfw_insn);
179098943Sluigi		return;
179198943Sluigi	}
179298943Sluigi
179398943Sluigi	p = strchr(av, '/');
179498943Sluigi	if (!p)
179598943Sluigi		p = strchr(av, ':');
179698943Sluigi	if (p) {
179798943Sluigi		md = *p;
179898943Sluigi		*p++ = '\0';
179998943Sluigi	}
180098943Sluigi
180198943Sluigi	if (lookup_host(av, &cmd->addr) != 0)
180298943Sluigi		errx(EX_NOHOST, "hostname ``%s'' unknown", av);
180398943Sluigi	switch (md) {
180498943Sluigi	case ':':
180598943Sluigi		if (!inet_aton(p, &cmd->mask))
180698943Sluigi			errx(EX_DATAERR, "bad netmask ``%s''", p);
180798943Sluigi		break;
180898943Sluigi	case '/':
180998943Sluigi		i = atoi(p);
181098943Sluigi		if (i == 0)
181198943Sluigi			cmd->mask.s_addr = htonl(0);
181298943Sluigi		else if (i > 32)
181398943Sluigi			errx(EX_DATAERR, "bad width ``%s''", p);
181498943Sluigi		else
181598943Sluigi			cmd->mask.s_addr = htonl(~0 << (32 - i));
181698943Sluigi		break;
181798943Sluigi	default:
181898943Sluigi		cmd->mask.s_addr = htonl(~0);
181998943Sluigi		break;
182098943Sluigi	}
182198943Sluigi	cmd->addr.s_addr &= cmd->mask.s_addr;
182298943Sluigi	/*
182398943Sluigi	 * now look if we have a set of addresses. They are stored as follows:
182498943Sluigi	 *   arg1	is the set size (powers of 2, 2..256)
182598943Sluigi	 *   addr	is the base address IN HOST FORMAT
182698943Sluigi	 *   mask..	is an array of u_int32_t with bits set.
182798943Sluigi	 */
182898943Sluigi	if (p)
182998943Sluigi		p = strchr(p, '{');
183098943Sluigi	if (p) {	/* fetch addresses */
183198943Sluigi		u_int32_t *d;
183298943Sluigi		int low, high;
183398943Sluigi		int i = contigmask((u_char *)&(cmd->mask), 32);
183498943Sluigi
183598943Sluigi		if (i < 24 || i > 31) {
183698943Sluigi			fprintf(stderr, "invalid set with mask %d\n",
183798943Sluigi				i);
183898943Sluigi			exit(0);
183998943Sluigi		}
184098943Sluigi		cmd->o.arg1 = 1<<(32-i);
184198943Sluigi		cmd->addr.s_addr = ntohl(cmd->addr.s_addr);
184298943Sluigi		d = (u_int32_t *)&cmd->mask;
184398943Sluigi		cmd->o.opcode = O_IP_DST_SET;	/* default */
184498943Sluigi		cmd->o.len |= F_INSN_SIZE(ipfw_insn_u32) + (cmd->o.arg1+31)/32;
1845101117Sluigi		for (i = 0; i < (cmd->o.arg1+31)/32 ; i++)
184698943Sluigi			d[i] = 0;	/* clear masks */
184798943Sluigi
184898943Sluigi		av = p+1;
184998943Sluigi		low = cmd->addr.s_addr & 0xff;
185098943Sluigi		high = low + cmd->o.arg1 - 1;
185198943Sluigi		while (isdigit(*av)) {
185298943Sluigi			char *s;
185398943Sluigi			u_int16_t a = strtol(av, &s, 0);
185498943Sluigi
185598943Sluigi			if (s == av) /* no parameter */
185698943Sluigi				break;
185798943Sluigi			if (a < low || a > high) {
185898943Sluigi			    fprintf(stderr, "addr %d out of range [%d-%d]\n",
185998943Sluigi				a, low, high);
186098943Sluigi			    exit(0);
186198943Sluigi			}
186298943Sluigi			a -= low;
186398943Sluigi			d[ a/32] |= 1<<(a & 31);
186498943Sluigi			if (*s != ',')
186598943Sluigi				break;
186698943Sluigi			av = s+1;
186798943Sluigi		}
186898943Sluigi		return;
186998943Sluigi	}
187098943Sluigi
187198943Sluigi	if (cmd->mask.s_addr == 0) { /* any */
187298943Sluigi		if (cmd->o.len & F_NOT)
187398943Sluigi			errx(EX_DATAERR, "not any never matches");
187498943Sluigi		else	/* useless, nuke it */
187598943Sluigi			return;
187698943Sluigi	} else if (cmd->mask.s_addr ==  IP_MASK_ALL)	/* one IP */
187798943Sluigi		cmd->o.len |= F_INSN_SIZE(ipfw_insn_u32);
187898943Sluigi	else						/* addr/mask */
187998943Sluigi		cmd->o.len |= F_INSN_SIZE(ipfw_insn_ip);
188098943Sluigi}
188198943Sluigi
188298943Sluigi
188398943Sluigi/*
188498943Sluigi * helper function to process a set of flags and set bits in the
188598943Sluigi * appropriate masks.
188698943Sluigi */
188798943Sluigistatic void
188898943Sluigifill_flags(ipfw_insn *cmd, enum ipfw_opcodes opcode,
188998943Sluigi	struct _s_x *flags, char *p)
189098943Sluigi{
189198943Sluigi	u_int8_t set=0, clear=0;
189298943Sluigi
189398943Sluigi	while (p && *p) {
189498943Sluigi		char *q;	/* points to the separator */
189598943Sluigi		int val;
189698943Sluigi		u_int8_t *which;	/* mask we are working on */
189798943Sluigi
189898943Sluigi		if (*p == '!') {
189998943Sluigi			p++;
190098943Sluigi			which = &clear;
190198943Sluigi		} else
190298943Sluigi			which = &set;
190398943Sluigi		q = strchr(p, ',');
190498943Sluigi		if (q)
190598943Sluigi			*q++ = '\0';
190698943Sluigi		val = match_token(flags, p);
190798943Sluigi		if (val <= 0)
190898943Sluigi			errx(EX_DATAERR, "invalid flag %s", p);
190998943Sluigi		*which |= (u_int8_t)val;
191098943Sluigi		p = q;
191198943Sluigi	}
191298943Sluigi        cmd->opcode = opcode;
191398943Sluigi        cmd->len =  (cmd->len & (F_NOT | F_OR)) | 1;
191498943Sluigi        cmd->arg1 = (set & 0xff) | ( (clear & 0xff) << 8);
191598943Sluigi}
191698943Sluigi
191798943Sluigi
191898943Sluigistatic void
191998943Sluigidelete(int ac, char *av[])
192098943Sluigi{
1921101978Sluigi	u_int32_t rulenum;
192298943Sluigi	struct dn_pipe pipe;
192398943Sluigi	int i;
192498943Sluigi	int exitval = EX_OK;
1925101628Sluigi	int do_set = 0;
192698943Sluigi
192798943Sluigi	memset(&pipe, 0, sizeof pipe);
192898943Sluigi
192998943Sluigi	av++; ac--;
1930101641Sluigi	if (ac > 0 && !strncmp(*av, "set", strlen(*av))) {
1931101978Sluigi		do_set = 1;	/* delete set */
1932101628Sluigi		ac--; av++;
1933101978Sluigi	}
193498943Sluigi
193598943Sluigi	/* Rule number */
193698943Sluigi	while (ac && isdigit(**av)) {
193798943Sluigi		i = atoi(*av); av++; ac--;
193898943Sluigi		if (do_pipe) {
193998943Sluigi			if (do_pipe == 1)
194098943Sluigi				pipe.pipe_nr = i;
194198943Sluigi			else
194298943Sluigi				pipe.fs.fs_nr = i;
194398943Sluigi			i = setsockopt(s, IPPROTO_IP, IP_DUMMYNET_DEL,
194498943Sluigi			    &pipe, sizeof pipe);
194598943Sluigi			if (i) {
194698943Sluigi				exitval = 1;
194798943Sluigi				warn("rule %u: setsockopt(IP_DUMMYNET_DEL)",
194898943Sluigi				    do_pipe == 1 ? pipe.pipe_nr :
194998943Sluigi				    pipe.fs.fs_nr);
195098943Sluigi			}
195198943Sluigi		} else {
1952101978Sluigi			rulenum =  (i & 0xffff) | (do_set << 24);
195398943Sluigi			i = setsockopt(s, IPPROTO_IP, IP_FW_DEL, &rulenum,
195498943Sluigi			    sizeof rulenum);
195598943Sluigi			if (i) {
195698943Sluigi				exitval = EX_UNAVAILABLE;
195798943Sluigi				warn("rule %u: setsockopt(IP_FW_DEL)",
195898943Sluigi				    rulenum);
195998943Sluigi			}
196098943Sluigi		}
196198943Sluigi	}
196298943Sluigi	if (exitval != EX_OK)
196398943Sluigi		exit(exitval);
196498943Sluigi}
196598943Sluigi
196698943Sluigi
196798943Sluigi/*
196898943Sluigi * fill the interface structure. We do not check the name as we can
196998943Sluigi * create interfaces dynamically, so checking them at insert time
197098943Sluigi * makes relatively little sense.
197198943Sluigi * A '*' following the name means any unit.
197298943Sluigi */
197398943Sluigistatic void
197498943Sluigifill_iface(ipfw_insn_if *cmd, char *arg)
197598943Sluigi{
197698943Sluigi	cmd->name[0] = '\0';
197798943Sluigi	cmd->o.len |= F_INSN_SIZE(ipfw_insn_if);
197898943Sluigi
197998943Sluigi	/* Parse the interface or address */
198098943Sluigi	if (!strcmp(arg, "any"))
198198943Sluigi		cmd->o.len = 0;		/* effectively ignore this command */
198298943Sluigi	else if (!isdigit(*arg)) {
198398943Sluigi		char *q;
198498943Sluigi
198598943Sluigi		strncpy(cmd->name, arg, sizeof(cmd->name));
198698943Sluigi		cmd->name[sizeof(cmd->name) - 1] = '\0';
198798943Sluigi		/* find first digit or wildcard */
198898943Sluigi		for (q = cmd->name; *q && !isdigit(*q) && *q != '*'; q++)
198998943Sluigi			continue;
199098943Sluigi		cmd->p.unit = (*q == '*') ? -1 : atoi(q);
199198943Sluigi		*q = '\0';
199298943Sluigi	} else if (!inet_aton(arg, &cmd->p.ip))
199398943Sluigi		errx(EX_DATAERR, "bad ip address ``%s''", arg);
199498943Sluigi}
199598943Sluigi
199698943Sluigi/*
199798943Sluigi * the following macro returns an error message if we run out of
199898943Sluigi * arguments.
199998943Sluigi */
200098943Sluigi#define	NEED1(msg)	{if (!ac) errx(EX_USAGE, msg);}
200198943Sluigi
200298943Sluigistatic void
200398943Sluigiconfig_pipe(int ac, char **av)
200498943Sluigi{
200598943Sluigi	struct dn_pipe pipe;
200698943Sluigi	int i;
200798943Sluigi	char *end;
200898943Sluigi	u_int32_t a;
200998943Sluigi	void *par = NULL;
201098943Sluigi
201198943Sluigi	memset(&pipe, 0, sizeof pipe);
201298943Sluigi
201398943Sluigi	av++; ac--;
201498943Sluigi	/* Pipe number */
201598943Sluigi	if (ac && isdigit(**av)) {
201698943Sluigi		i = atoi(*av); av++; ac--;
201798943Sluigi		if (do_pipe == 1)
201898943Sluigi			pipe.pipe_nr = i;
201998943Sluigi		else
202098943Sluigi			pipe.fs.fs_nr = i;
202198943Sluigi	}
202299475Sluigi	while (ac > 0) {
202398943Sluigi		double d;
202498943Sluigi		int tok = match_token(dummynet_params, *av);
202598943Sluigi		ac--; av++;
202698943Sluigi
202798943Sluigi		switch(tok) {
2028101978Sluigi		case TOK_NOERROR:
2029101978Sluigi			pipe.fs.flags_fs |= DN_NOERROR;
2030101978Sluigi			break;
2031101978Sluigi
203298943Sluigi		case TOK_PLR:
203398943Sluigi			NEED1("plr needs argument 0..1\n");
203498943Sluigi			d = strtod(av[0], NULL);
203598943Sluigi			if (d > 1)
203698943Sluigi				d = 1;
203798943Sluigi			else if (d < 0)
203898943Sluigi				d = 0;
203998943Sluigi			pipe.fs.plr = (int)(d*0x7fffffff);
204098943Sluigi			ac--; av++;
204198943Sluigi			break;
204298943Sluigi
204398943Sluigi		case TOK_QUEUE:
204498943Sluigi			NEED1("queue needs queue size\n");
204598943Sluigi			end = NULL;
204698943Sluigi			pipe.fs.qsize = strtoul(av[0], &end, 0);
204798943Sluigi			if (*end == 'K' || *end == 'k') {
204898943Sluigi				pipe.fs.flags_fs |= DN_QSIZE_IS_BYTES;
204998943Sluigi				pipe.fs.qsize *= 1024;
205098943Sluigi			} else if (*end == 'B' || !strncmp(end, "by", 2)) {
205198943Sluigi				pipe.fs.flags_fs |= DN_QSIZE_IS_BYTES;
205298943Sluigi			}
205398943Sluigi			ac--; av++;
205498943Sluigi			break;
205598943Sluigi
205698943Sluigi		case TOK_BUCKETS:
205798943Sluigi			NEED1("buckets needs argument\n");
205898943Sluigi			pipe.fs.rq_size = strtoul(av[0], NULL, 0);
205998943Sluigi			ac--; av++;
206098943Sluigi			break;
206198943Sluigi
206298943Sluigi		case TOK_MASK:
206398943Sluigi			NEED1("mask needs mask specifier\n");
206498943Sluigi			/*
206598943Sluigi			 * per-flow queue, mask is dst_ip, dst_port,
206698943Sluigi			 * src_ip, src_port, proto measured in bits
206798943Sluigi			 */
206898943Sluigi			par = NULL;
206998943Sluigi
207098943Sluigi			pipe.fs.flow_mask.dst_ip = 0;
207198943Sluigi			pipe.fs.flow_mask.src_ip = 0;
207298943Sluigi			pipe.fs.flow_mask.dst_port = 0;
207398943Sluigi			pipe.fs.flow_mask.src_port = 0;
207498943Sluigi			pipe.fs.flow_mask.proto = 0;
207598943Sluigi			end = NULL;
207698943Sluigi
207798943Sluigi			while (ac >= 1) {
207898943Sluigi			    u_int32_t *p32 = NULL;
207998943Sluigi			    u_int16_t *p16 = NULL;
208098943Sluigi
208198943Sluigi			    tok = match_token(dummynet_params, *av);
208298943Sluigi			    ac--; av++;
208398943Sluigi			    switch(tok) {
208498943Sluigi			    case TOK_ALL:
208598943Sluigi				    /*
208698943Sluigi				     * special case, all bits significant
208798943Sluigi				     */
208898943Sluigi				    pipe.fs.flow_mask.dst_ip = ~0;
208998943Sluigi				    pipe.fs.flow_mask.src_ip = ~0;
209098943Sluigi				    pipe.fs.flow_mask.dst_port = ~0;
209198943Sluigi				    pipe.fs.flow_mask.src_port = ~0;
209298943Sluigi				    pipe.fs.flow_mask.proto = ~0;
209398943Sluigi				    pipe.fs.flags_fs |= DN_HAVE_FLOW_MASK;
209498943Sluigi				    goto end_mask;
209598943Sluigi
209698943Sluigi			    case TOK_DSTIP:
209798943Sluigi				    p32 = &pipe.fs.flow_mask.dst_ip;
209898943Sluigi				    break;
209998943Sluigi
210098943Sluigi			    case TOK_SRCIP:
210198943Sluigi				    p32 = &pipe.fs.flow_mask.src_ip;
210298943Sluigi				    break;
210398943Sluigi
210498943Sluigi			    case TOK_DSTPORT:
210598943Sluigi				    p16 = &pipe.fs.flow_mask.dst_port;
210698943Sluigi				    break;
210798943Sluigi
210898943Sluigi			    case TOK_SRCPORT:
210998943Sluigi				    p16 = &pipe.fs.flow_mask.src_port;
211098943Sluigi				    break;
211198943Sluigi
211298943Sluigi			    case TOK_PROTO:
211398943Sluigi				    break;
211498943Sluigi
211598943Sluigi			    default:
211698943Sluigi				    ac++; av--; /* backtrack */
211798943Sluigi				    goto end_mask;
211898943Sluigi			    }
211998943Sluigi			    if (ac < 1)
212098943Sluigi				    errx(EX_USAGE, "mask: value missing");
212198943Sluigi			    if (*av[0] == '/') {
212298943Sluigi				    a = strtoul(av[0]+1, &end, 0);
212398943Sluigi				    a = (a == 32) ? ~0 : (1 << a) - 1;
2124106505Smaxim			    } else
212599909Sluigi				    a = strtoul(av[0], &end, 0);
212698943Sluigi			    if (p32 != NULL)
212798943Sluigi				    *p32 = a;
212898943Sluigi			    else if (p16 != NULL) {
212998943Sluigi				    if (a > 65535)
213098943Sluigi					    errx(EX_DATAERR,
213198943Sluigi						"mask: must be 16 bit");
213298943Sluigi				    *p16 = (u_int16_t)a;
213398943Sluigi			    } else {
213498943Sluigi				    if (a > 255)
213598943Sluigi					    errx(EX_DATAERR,
213698943Sluigi						"mask: must be 8 bit");
213798943Sluigi				    pipe.fs.flow_mask.proto = (u_int8_t)a;
213898943Sluigi			    }
213998943Sluigi			    if (a != 0)
214098943Sluigi				    pipe.fs.flags_fs |= DN_HAVE_FLOW_MASK;
214198943Sluigi			    ac--; av++;
214298943Sluigi			} /* end while, config masks */
214398943Sluigiend_mask:
214498943Sluigi			break;
214598943Sluigi
214698943Sluigi		case TOK_RED:
214798943Sluigi		case TOK_GRED:
214898943Sluigi			NEED1("red/gred needs w_q/min_th/max_th/max_p\n");
214998943Sluigi			pipe.fs.flags_fs |= DN_IS_RED;
215098943Sluigi			if (tok == TOK_GRED)
215198943Sluigi				pipe.fs.flags_fs |= DN_IS_GENTLE_RED;
215298943Sluigi			/*
215398943Sluigi			 * the format for parameters is w_q/min_th/max_th/max_p
215498943Sluigi			 */
215598943Sluigi			if ((end = strsep(&av[0], "/"))) {
215698943Sluigi			    double w_q = strtod(end, NULL);
215798943Sluigi			    if (w_q > 1 || w_q <= 0)
215898943Sluigi				errx(EX_DATAERR, "0 < w_q <= 1");
215998943Sluigi			    pipe.fs.w_q = (int) (w_q * (1 << SCALE_RED));
216098943Sluigi			}
216198943Sluigi			if ((end = strsep(&av[0], "/"))) {
216298943Sluigi			    pipe.fs.min_th = strtoul(end, &end, 0);
216398943Sluigi			    if (*end == 'K' || *end == 'k')
216498943Sluigi				pipe.fs.min_th *= 1024;
216598943Sluigi			}
216698943Sluigi			if ((end = strsep(&av[0], "/"))) {
216798943Sluigi			    pipe.fs.max_th = strtoul(end, &end, 0);
216898943Sluigi			    if (*end == 'K' || *end == 'k')
216998943Sluigi				pipe.fs.max_th *= 1024;
217098943Sluigi			}
217198943Sluigi			if ((end = strsep(&av[0], "/"))) {
217298943Sluigi			    double max_p = strtod(end, NULL);
217398943Sluigi			    if (max_p > 1 || max_p <= 0)
217498943Sluigi				errx(EX_DATAERR, "0 < max_p <= 1");
217598943Sluigi			    pipe.fs.max_p = (int)(max_p * (1 << SCALE_RED));
217698943Sluigi			}
217798943Sluigi			ac--; av++;
217898943Sluigi			break;
217998943Sluigi
218098943Sluigi		case TOK_DROPTAIL:
218198943Sluigi			pipe.fs.flags_fs &= ~(DN_IS_RED|DN_IS_GENTLE_RED);
218298943Sluigi			break;
2183106505Smaxim
218498943Sluigi		case TOK_BW:
218598943Sluigi			NEED1("bw needs bandwidth or interface\n");
218698943Sluigi			if (do_pipe != 1)
218798943Sluigi			    errx(EX_DATAERR, "bandwidth only valid for pipes");
218898943Sluigi			/*
218998943Sluigi			 * set clocking interface or bandwidth value
219098943Sluigi			 */
219198943Sluigi			if (av[0][0] >= 'a' && av[0][0] <= 'z') {
219298943Sluigi			    int l = sizeof(pipe.if_name)-1;
219398943Sluigi			    /* interface name */
219498943Sluigi			    strncpy(pipe.if_name, av[0], l);
219598943Sluigi			    pipe.if_name[l] = '\0';
219698943Sluigi			    pipe.bandwidth = 0;
219798943Sluigi			} else {
219898943Sluigi			    pipe.if_name[0] = '\0';
219998943Sluigi			    pipe.bandwidth = strtoul(av[0], &end, 0);
220098943Sluigi			    if (*end == 'K' || *end == 'k') {
220198943Sluigi				end++;
220298943Sluigi				pipe.bandwidth *= 1000;
220398943Sluigi			    } else if (*end == 'M') {
220498943Sluigi				end++;
220598943Sluigi				pipe.bandwidth *= 1000000;
220698943Sluigi			    }
220798943Sluigi			    if (*end == 'B' || !strncmp(end, "by", 2))
220898943Sluigi				pipe.bandwidth *= 8;
220998943Sluigi			    if (pipe.bandwidth < 0)
221098943Sluigi				errx(EX_DATAERR, "bandwidth too large");
221198943Sluigi			}
221298943Sluigi			ac--; av++;
221398943Sluigi			break;
221498943Sluigi
221598943Sluigi		case TOK_DELAY:
221698943Sluigi			if (do_pipe != 1)
221798943Sluigi				errx(EX_DATAERR, "delay only valid for pipes");
221898943Sluigi			NEED1("delay needs argument 0..10000ms\n");
221998943Sluigi			pipe.delay = strtoul(av[0], NULL, 0);
222098943Sluigi			ac--; av++;
222198943Sluigi			break;
222298943Sluigi
222398943Sluigi		case TOK_WEIGHT:
222498943Sluigi			if (do_pipe == 1)
222598943Sluigi				errx(EX_DATAERR,"weight only valid for queues");
222698943Sluigi			NEED1("weight needs argument 0..100\n");
222798943Sluigi			pipe.fs.weight = strtoul(av[0], &end, 0);
222898943Sluigi			ac--; av++;
222998943Sluigi			break;
223098943Sluigi
223198943Sluigi		case TOK_PIPE:
223298943Sluigi			if (do_pipe == 1)
223398943Sluigi				errx(EX_DATAERR,"pipe only valid for queues");
223498943Sluigi			NEED1("pipe needs pipe_number\n");
223598943Sluigi			pipe.fs.parent_nr = strtoul(av[0], &end, 0);
223698943Sluigi			ac--; av++;
223798943Sluigi			break;
223898943Sluigi
223998943Sluigi		default:
224098943Sluigi			errx(EX_DATAERR, "unrecognised option ``%s''", *av);
224198943Sluigi		}
224298943Sluigi	}
224398943Sluigi	if (do_pipe == 1) {
224498943Sluigi		if (pipe.pipe_nr == 0)
224598943Sluigi			errx(EX_DATAERR, "pipe_nr must be > 0");
224698943Sluigi		if (pipe.delay > 10000)
224798943Sluigi			errx(EX_DATAERR, "delay must be < 10000");
224898943Sluigi	} else { /* do_pipe == 2, queue */
224998943Sluigi		if (pipe.fs.parent_nr == 0)
225098943Sluigi			errx(EX_DATAERR, "pipe must be > 0");
225198943Sluigi		if (pipe.fs.weight >100)
225298943Sluigi			errx(EX_DATAERR, "weight must be <= 100");
225398943Sluigi	}
225498943Sluigi	if (pipe.fs.flags_fs & DN_QSIZE_IS_BYTES) {
225598943Sluigi		if (pipe.fs.qsize > 1024*1024)
225698943Sluigi			errx(EX_DATAERR, "queue size must be < 1MB");
225798943Sluigi	} else {
225898943Sluigi		if (pipe.fs.qsize > 100)
225998943Sluigi			errx(EX_DATAERR, "2 <= queue size <= 100");
226098943Sluigi	}
226198943Sluigi	if (pipe.fs.flags_fs & DN_IS_RED) {
226298943Sluigi		size_t len;
226398943Sluigi		int lookup_depth, avg_pkt_size;
226498943Sluigi		double s, idle, weight, w_q;
226598943Sluigi		struct clockinfo clock;
226698943Sluigi		int t;
226798943Sluigi
226898943Sluigi		if (pipe.fs.min_th >= pipe.fs.max_th)
226998943Sluigi		    errx(EX_DATAERR, "min_th %d must be < than max_th %d",
227098943Sluigi			pipe.fs.min_th, pipe.fs.max_th);
227198943Sluigi		if (pipe.fs.max_th == 0)
227298943Sluigi		    errx(EX_DATAERR, "max_th must be > 0");
227398943Sluigi
227498943Sluigi		len = sizeof(int);
227598943Sluigi		if (sysctlbyname("net.inet.ip.dummynet.red_lookup_depth",
227698943Sluigi			&lookup_depth, &len, NULL, 0) == -1)
227798943Sluigi
227898943Sluigi		    errx(1, "sysctlbyname(\"%s\")",
227998943Sluigi			"net.inet.ip.dummynet.red_lookup_depth");
228098943Sluigi		if (lookup_depth == 0)
228198943Sluigi		    errx(EX_DATAERR, "net.inet.ip.dummynet.red_lookup_depth"
228298943Sluigi			" must be greater than zero");
228398943Sluigi
228498943Sluigi		len = sizeof(int);
228598943Sluigi		if (sysctlbyname("net.inet.ip.dummynet.red_avg_pkt_size",
228698943Sluigi			&avg_pkt_size, &len, NULL, 0) == -1)
228798943Sluigi
228898943Sluigi		    errx(1, "sysctlbyname(\"%s\")",
228998943Sluigi			"net.inet.ip.dummynet.red_avg_pkt_size");
229098943Sluigi		if (avg_pkt_size == 0)
229198943Sluigi			errx(EX_DATAERR,
229298943Sluigi			    "net.inet.ip.dummynet.red_avg_pkt_size must"
229398943Sluigi			    " be greater than zero");
229498943Sluigi
229598943Sluigi		len = sizeof(struct clockinfo);
229698943Sluigi		if (sysctlbyname("kern.clockrate", &clock, &len, NULL, 0) == -1)
229798943Sluigi			errx(1, "sysctlbyname(\"%s\")", "kern.clockrate");
229898943Sluigi
229998943Sluigi		/*
230098943Sluigi		 * Ticks needed for sending a medium-sized packet.
230198943Sluigi		 * Unfortunately, when we are configuring a WF2Q+ queue, we
230298943Sluigi		 * do not have bandwidth information, because that is stored
230398943Sluigi		 * in the parent pipe, and also we have multiple queues
230498943Sluigi		 * competing for it. So we set s=0, which is not very
230598943Sluigi		 * correct. But on the other hand, why do we want RED with
230698943Sluigi		 * WF2Q+ ?
230798943Sluigi		 */
230898943Sluigi		if (pipe.bandwidth==0) /* this is a WF2Q+ queue */
230998943Sluigi			s = 0;
231098943Sluigi		else
231198943Sluigi			s = clock.hz * avg_pkt_size * 8 / pipe.bandwidth;
231298943Sluigi
231398943Sluigi		/*
231498943Sluigi		 * max idle time (in ticks) before avg queue size becomes 0.
231598943Sluigi		 * NOTA:  (3/w_q) is approx the value x so that
231698943Sluigi		 * (1-w_q)^x < 10^-3.
231798943Sluigi		 */
231898943Sluigi		w_q = ((double)pipe.fs.w_q) / (1 << SCALE_RED);
231998943Sluigi		idle = s * 3. / w_q;
232098943Sluigi		pipe.fs.lookup_step = (int)idle / lookup_depth;
232198943Sluigi		if (!pipe.fs.lookup_step)
232298943Sluigi			pipe.fs.lookup_step = 1;
232398943Sluigi		weight = 1 - w_q;
232498943Sluigi		for (t = pipe.fs.lookup_step; t > 0; --t)
232598943Sluigi			weight *= weight;
232698943Sluigi		pipe.fs.lookup_weight = (int)(weight * (1 << SCALE_RED));
232798943Sluigi	}
232898943Sluigi	i = setsockopt(s, IPPROTO_IP, IP_DUMMYNET_CONFIGURE, &pipe,
232998943Sluigi			    sizeof pipe);
233098943Sluigi	if (i)
233198943Sluigi		err(1, "setsockopt(%s)", "IP_DUMMYNET_CONFIGURE");
233298943Sluigi}
233398943Sluigi
233498943Sluigistatic void
233598943Sluigiget_mac_addr_mask(char *p, u_char *addr, u_char *mask)
233698943Sluigi{
233798943Sluigi	int i, l;
233898943Sluigi
233998943Sluigi	for (i=0; i<6; i++)
234098943Sluigi		addr[i] = mask[i] = 0;
234198943Sluigi	if (!strcmp(p, "any"))
234298943Sluigi		return;
234398943Sluigi
234498943Sluigi	for (i=0; *p && i<6;i++, p++) {
234598943Sluigi		addr[i] = strtol(p, &p, 16);
234698943Sluigi		if (*p != ':') /* we start with the mask */
234798943Sluigi			break;
234898943Sluigi	}
234998943Sluigi	if (*p == '/') { /* mask len */
235098943Sluigi		l = strtol(p+1, &p, 0);
235198943Sluigi		for (i=0; l>0; l -=8, i++)
235298943Sluigi			mask[i] = (l >=8) ? 0xff : (~0) << (8-l);
235398943Sluigi	} else if (*p == '&') { /* mask */
235498943Sluigi		for (i=0, p++; *p && i<6;i++, p++) {
235598943Sluigi			mask[i] = strtol(p, &p, 16);
235698943Sluigi			if (*p != ':')
235798943Sluigi				break;
235898943Sluigi		}
235998943Sluigi	} else if (*p == '\0') {
236098943Sluigi		for (i=0; i<6; i++)
236198943Sluigi			mask[i] = 0xff;
236298943Sluigi	}
236398943Sluigi	for (i=0; i<6; i++)
236498943Sluigi		addr[i] &= mask[i];
236598943Sluigi}
236698943Sluigi
236798943Sluigi/*
236898943Sluigi * helper function, updates the pointer to cmd with the length
236998943Sluigi * of the current command, and also cleans up the first word of
237098943Sluigi * the new command in case it has been clobbered before.
237198943Sluigi */
237298943Sluigistatic ipfw_insn *
237398943Sluiginext_cmd(ipfw_insn *cmd)
237498943Sluigi{
237598943Sluigi	cmd += F_LEN(cmd);
237698943Sluigi	bzero(cmd, sizeof(*cmd));
237798943Sluigi	return cmd;
237898943Sluigi}
237998943Sluigi
238098943Sluigi/*
238198943Sluigi * A function to fill simple commands of size 1.
238298943Sluigi * Existing flags are preserved.
238398943Sluigi */
238498943Sluigistatic void
238598943Sluigifill_cmd(ipfw_insn *cmd, enum ipfw_opcodes opcode, int flags, u_int16_t arg)
238698943Sluigi{
238798943Sluigi	cmd->opcode = opcode;
238898943Sluigi	cmd->len =  ((cmd->len | flags) & (F_NOT | F_OR)) | 1;
238998943Sluigi	cmd->arg1 = arg;
239098943Sluigi}
239198943Sluigi
239298943Sluigi/*
239398943Sluigi * Fetch and add the MAC address and type, with masks. This generates one or
239498943Sluigi * two microinstructions, and returns the pointer to the last one.
239598943Sluigi */
239698943Sluigistatic ipfw_insn *
239798943Sluigiadd_mac(ipfw_insn *cmd, int ac, char *av[])
239898943Sluigi{
2399102087Sluigi	ipfw_insn_mac *mac;
240098943Sluigi
2401102087Sluigi	if (ac < 2)
2402102098Sluigi		errx(EX_DATAERR, "MAC dst src");
240398943Sluigi
240498943Sluigi	cmd->opcode = O_MACADDR2;
240598943Sluigi	cmd->len = (cmd->len & (F_NOT | F_OR)) | F_INSN_SIZE(ipfw_insn_mac);
240698943Sluigi
240798943Sluigi	mac = (ipfw_insn_mac *)cmd;
2408101978Sluigi	get_mac_addr_mask(av[0], mac->addr, mac->mask);	/* dst */
240998943Sluigi	get_mac_addr_mask(av[1], &(mac->addr[6]), &(mac->mask[6])); /* src */
2410102087Sluigi	return cmd;
2411102087Sluigi}
241298943Sluigi
2413102087Sluigistatic ipfw_insn *
2414102087Sluigiadd_mactype(ipfw_insn *cmd, int ac, char *av)
2415102087Sluigi{
2416102087Sluigi	if (ac < 1)
2417102087Sluigi		errx(EX_DATAERR, "missing MAC type");
2418102087Sluigi	if (strcmp(av, "any") != 0) { /* we have a non-null type */
2419102087Sluigi		fill_newports((ipfw_insn_u16 *)cmd, av, IPPROTO_ETHERTYPE);
242098943Sluigi		cmd->opcode = O_MAC_TYPE;
2421102087Sluigi		return cmd;
2422102087Sluigi	} else
2423102087Sluigi		return NULL;
2424102087Sluigi}
242598943Sluigi
2426102087Sluigistatic ipfw_insn *
2427102087Sluigiadd_proto(ipfw_insn *cmd, char *av)
2428102087Sluigi{
2429102087Sluigi	struct protoent *pe;
2430102087Sluigi	u_char proto = 0;
2431102087Sluigi
2432102087Sluigi	if (!strncmp(av, "all", strlen(av)))
2433102087Sluigi		; /* same as "ip" */
2434102087Sluigi	else if ((proto = atoi(av)) > 0)
2435102087Sluigi		; /* all done! */
2436102087Sluigi	else if ((pe = getprotobyname(av)) != NULL)
2437102087Sluigi		proto = pe->p_proto;
2438102087Sluigi	else
2439102098Sluigi		return NULL;
2440102087Sluigi	if (proto != IPPROTO_IP)
2441102087Sluigi		fill_cmd(cmd, O_PROTO, 0, proto);
244298943Sluigi	return cmd;
244398943Sluigi}
244498943Sluigi
2445102087Sluigistatic ipfw_insn *
2446102087Sluigiadd_srcip(ipfw_insn *cmd, char *av)
2447102087Sluigi{
2448102087Sluigi	fill_ip((ipfw_insn_ip *)cmd, av);
2449102087Sluigi	if (cmd->opcode == O_IP_DST_SET)			/* set */
2450102087Sluigi		cmd->opcode = O_IP_SRC_SET;
2451102087Sluigi	else if (F_LEN(cmd) == F_INSN_SIZE(ipfw_insn))		/* me */
2452102087Sluigi		cmd->opcode = O_IP_SRC_ME;
2453102087Sluigi	else if (F_LEN(cmd) == F_INSN_SIZE(ipfw_insn_u32))	/* one IP */
2454102087Sluigi		cmd->opcode = O_IP_SRC;
2455102087Sluigi	else if (F_LEN(cmd) == F_INSN_SIZE(ipfw_insn_ip))	/* addr/mask */
2456102087Sluigi		cmd->opcode = O_IP_SRC_MASK;
2457102087Sluigi	return cmd;
2458102087Sluigi}
2459102087Sluigi
2460102087Sluigistatic ipfw_insn *
2461102087Sluigiadd_dstip(ipfw_insn *cmd, char *av)
2462102087Sluigi{
2463102087Sluigi	fill_ip((ipfw_insn_ip *)cmd, av);
2464102087Sluigi	if (cmd->opcode == O_IP_DST_SET)			/* set */
2465102087Sluigi		;
2466102087Sluigi	else if (F_LEN(cmd) == F_INSN_SIZE(ipfw_insn))		/* me */
2467102087Sluigi		cmd->opcode = O_IP_DST_ME;
2468102087Sluigi	else if (F_LEN(cmd) == F_INSN_SIZE(ipfw_insn_u32))	/* one IP */
2469102087Sluigi		cmd->opcode = O_IP_DST;
2470102087Sluigi	else if (F_LEN(cmd) == F_INSN_SIZE(ipfw_insn_ip))	/* addr/mask */
2471102087Sluigi		cmd->opcode = O_IP_DST_MASK;
2472102087Sluigi	return cmd;
2473102087Sluigi}
2474102087Sluigi
2475102087Sluigistatic ipfw_insn *
2476102087Sluigiadd_ports(ipfw_insn *cmd, char *av, u_char proto, int opcode)
2477102087Sluigi{
2478102087Sluigi	if (!strncmp(av, "any", strlen(av))) {
2479102087Sluigi		return NULL;
2480102087Sluigi	} else if (fill_newports((ipfw_insn_u16 *)cmd, av, proto)) {
2481102087Sluigi		/* XXX todo: check that we have a protocol with ports */
2482102087Sluigi		cmd->opcode = opcode;
2483102087Sluigi		return cmd;
2484102087Sluigi	}
2485102087Sluigi	return NULL;
2486102087Sluigi}
2487102087Sluigi
248898943Sluigi/*
248998943Sluigi * Parse arguments and assemble the microinstructions which make up a rule.
249098943Sluigi * Rules are added into the 'rulebuf' and then copied in the correct order
249198943Sluigi * into the actual rule.
249298943Sluigi *
249398943Sluigi * The syntax for a rule starts with the action, followed by an
249498943Sluigi * optional log action, and the various match patterns.
2495108533Sschweikh * In the assembled microcode, the first opcode must be an O_PROBE_STATE
249698943Sluigi * (generated if the rule includes a keep-state option), then the
249798943Sluigi * various match patterns, the "log" action, and the actual action.
2498106505Smaxim *
249998943Sluigi */
250098943Sluigistatic void
250198943Sluigiadd(int ac, char *av[])
250298943Sluigi{
250398943Sluigi	/*
250498943Sluigi	 * rules are added into the 'rulebuf' and then copied in
250598943Sluigi	 * the correct order into the actual rule.
250698943Sluigi	 * Some things that need to go out of order (prob, action etc.)
250798943Sluigi	 * go into actbuf[].
250898943Sluigi	 */
250998943Sluigi	static u_int32_t rulebuf[255], actbuf[255], cmdbuf[255];
251098943Sluigi
251198943Sluigi	ipfw_insn *src, *dst, *cmd, *action, *prev;
2512102087Sluigi	ipfw_insn *first_cmd;	/* first match pattern */
251398943Sluigi
251498943Sluigi	struct ip_fw *rule;
251598943Sluigi
251698943Sluigi	/*
251798943Sluigi	 * various flags used to record that we entered some fields.
251898943Sluigi	 */
2519101116Sluigi	ipfw_insn *have_state = NULL;	/* check-state or keep-state */
252098943Sluigi
252198943Sluigi	int i;
252298943Sluigi
252398943Sluigi	int open_par = 0;	/* open parenthesis ( */
252498943Sluigi
252598943Sluigi	/* proto is here because it is used to fetch ports */
252698943Sluigi	u_char proto = IPPROTO_IP;	/* default protocol */
252798943Sluigi
2528107289Sluigi	double match_prob = 1; /* match probability, default is always match */
2529107289Sluigi
253098943Sluigi	bzero(actbuf, sizeof(actbuf));		/* actions go here */
253198943Sluigi	bzero(cmdbuf, sizeof(cmdbuf));
253298943Sluigi	bzero(rulebuf, sizeof(rulebuf));
253398943Sluigi
253498943Sluigi	rule = (struct ip_fw *)rulebuf;
253598943Sluigi	cmd = (ipfw_insn *)cmdbuf;
253698943Sluigi	action = (ipfw_insn *)actbuf;
253798943Sluigi
253898943Sluigi	av++; ac--;
253998943Sluigi
254098943Sluigi	/* [rule N]	-- Rule number optional */
254198943Sluigi	if (ac && isdigit(**av)) {
254298943Sluigi		rule->rulenum = atoi(*av);
254398943Sluigi		av++;
254498943Sluigi		ac--;
254598943Sluigi	}
254698943Sluigi
2547101628Sluigi	/* [set N]	-- set number (0..30), optional */
2548101628Sluigi	if (ac > 1 && !strncmp(*av, "set", strlen(*av))) {
2549101628Sluigi		int set = strtoul(av[1], NULL, 10);
2550101628Sluigi		if (set < 0 || set > 30)
2551101628Sluigi			errx(EX_DATAERR, "illegal set %s", av[1]);
2552101628Sluigi		rule->set = set;
2553101628Sluigi		av += 2; ac -= 2;
2554101628Sluigi	}
2555101628Sluigi
255698943Sluigi	/* [prob D]	-- match probability, optional */
255798943Sluigi	if (ac > 1 && !strncmp(*av, "prob", strlen(*av))) {
2558107289Sluigi		match_prob = strtod(av[1], NULL);
255998943Sluigi
2560107289Sluigi		if (match_prob <= 0 || match_prob > 1)
256198943Sluigi			errx(EX_DATAERR, "illegal match prob. %s", av[1]);
256298943Sluigi		av += 2; ac -= 2;
256398943Sluigi	}
256498943Sluigi
256598943Sluigi	/* action	-- mandatory */
256698943Sluigi	NEED1("missing action");
256798943Sluigi	i = match_token(rule_actions, *av);
256898943Sluigi	ac--; av++;
256998943Sluigi	action->len = 1;	/* default */
257098943Sluigi	switch(i) {
257198943Sluigi	case TOK_CHECKSTATE:
2572101116Sluigi		have_state = action;
257398943Sluigi		action->opcode = O_CHECK_STATE;
257498943Sluigi		break;
257598943Sluigi
257698943Sluigi	case TOK_ACCEPT:
257798943Sluigi		action->opcode = O_ACCEPT;
257898943Sluigi		break;
257998943Sluigi
258098943Sluigi	case TOK_DENY:
258198943Sluigi		action->opcode = O_DENY;
258299475Sluigi		action->arg1 = 0;
258398943Sluigi		break;
258498943Sluigi
258599475Sluigi	case TOK_REJECT:
258699475Sluigi		action->opcode = O_REJECT;
258799475Sluigi		action->arg1 = ICMP_UNREACH_HOST;
258899475Sluigi		break;
258999475Sluigi
259099475Sluigi	case TOK_RESET:
259199475Sluigi		action->opcode = O_REJECT;
259299475Sluigi		action->arg1 = ICMP_REJECT_RST;
259399475Sluigi		break;
259499475Sluigi
259599475Sluigi	case TOK_UNREACH:
259699475Sluigi		action->opcode = O_REJECT;
259799475Sluigi		NEED1("missing reject code");
259899475Sluigi		fill_reject_code(&action->arg1, *av);
259999475Sluigi		ac--; av++;
260099475Sluigi		break;
260199475Sluigi
260298943Sluigi	case TOK_COUNT:
260398943Sluigi		action->opcode = O_COUNT;
260498943Sluigi		break;
260598943Sluigi
260698943Sluigi	case TOK_QUEUE:
260798943Sluigi	case TOK_PIPE:
260898943Sluigi		action->len = F_INSN_SIZE(ipfw_insn_pipe);
260998943Sluigi	case TOK_SKIPTO:
261098943Sluigi		if (i == TOK_QUEUE)
261198943Sluigi			action->opcode = O_QUEUE;
261298943Sluigi		else if (i == TOK_PIPE)
261398943Sluigi			action->opcode = O_PIPE;
261498943Sluigi		else if (i == TOK_SKIPTO)
261598943Sluigi			action->opcode = O_SKIPTO;
261698943Sluigi		NEED1("missing skipto/pipe/queue number");
261798943Sluigi		action->arg1 = strtoul(*av, NULL, 10);
261898943Sluigi		av++; ac--;
261998943Sluigi		break;
262098943Sluigi
262198943Sluigi	case TOK_DIVERT:
262298943Sluigi	case TOK_TEE:
262398943Sluigi		action->opcode = (i == TOK_DIVERT) ? O_DIVERT : O_TEE;
262498943Sluigi		NEED1("missing divert/tee port");
262598943Sluigi		action->arg1 = strtoul(*av, NULL, 0);
262698943Sluigi		if (action->arg1 == 0) {
262798943Sluigi			struct servent *s;
262898943Sluigi			setservent(1);
262998943Sluigi			s = getservbyname(av[0], "divert");
263098943Sluigi			if (s != NULL)
263198943Sluigi				action->arg1 = ntohs(s->s_port);
263298943Sluigi			else
263398943Sluigi				errx(EX_DATAERR, "illegal divert/tee port");
263498943Sluigi		}
263598943Sluigi		ac--; av++;
263698943Sluigi		break;
263798943Sluigi
263898943Sluigi	case TOK_FORWARD: {
263998943Sluigi		ipfw_insn_sa *p = (ipfw_insn_sa *)action;
264098943Sluigi		char *s, *end;
264198943Sluigi
264298943Sluigi		NEED1("missing forward address[:port]");
264398943Sluigi
264498943Sluigi		action->opcode = O_FORWARD_IP;
264598943Sluigi		action->len = F_INSN_SIZE(ipfw_insn_sa);
264698943Sluigi
264798943Sluigi		p->sa.sin_len = sizeof(struct sockaddr_in);
264898943Sluigi		p->sa.sin_family = AF_INET;
264998943Sluigi		p->sa.sin_port = 0;
265098943Sluigi		/*
265198943Sluigi		 * locate the address-port separator (':' or ',')
265298943Sluigi		 */
265398943Sluigi		s = strchr(*av, ':');
265498943Sluigi		if (s == NULL)
265598943Sluigi			s = strchr(*av, ',');
265698943Sluigi		if (s != NULL) {
265798943Sluigi			*(s++) = '\0';
265898943Sluigi			i = strtoport(s, &end, 0 /* base */, 0 /* proto */);
265998943Sluigi			if (s == end)
266098943Sluigi				errx(EX_DATAERR,
266198943Sluigi				    "illegal forwarding port ``%s''", s);
2662103241Sluigi			p->sa.sin_port = (u_short)i;
266398943Sluigi		}
266498943Sluigi		lookup_host(*av, &(p->sa.sin_addr));
266598943Sluigi		}
266698943Sluigi		ac--; av++;
266798943Sluigi		break;
266898943Sluigi
266998943Sluigi	default:
2670102087Sluigi		errx(EX_DATAERR, "invalid action %s\n", av[-1]);
267198943Sluigi	}
267298943Sluigi	action = next_cmd(action);
267398943Sluigi
267498943Sluigi	/*
267598943Sluigi	 * [log [logamount N]]	-- log, optional
267698943Sluigi	 *
267798943Sluigi	 * If exists, it goes first in the cmdbuf, but then it is
267898943Sluigi	 * skipped in the copy section to the end of the buffer.
267998943Sluigi	 */
268098943Sluigi	if (ac && !strncmp(*av, "log", strlen(*av))) {
268198943Sluigi		ipfw_insn_log *c = (ipfw_insn_log *)cmd;
268298943Sluigi
268398943Sluigi		cmd->len = F_INSN_SIZE(ipfw_insn_log);
268498943Sluigi		cmd->opcode = O_LOG;
268598943Sluigi		av++; ac--;
268698943Sluigi		if (ac && !strncmp(*av, "logamount", strlen(*av))) {
268798943Sluigi			ac--; av++;
268898943Sluigi			NEED1("logamount requires argument");
268998943Sluigi			c->max_log = atoi(*av);
269098943Sluigi			if (c->max_log < 0)
269198943Sluigi				errx(EX_DATAERR, "logamount must be positive");
269298943Sluigi			ac--; av++;
269398943Sluigi		}
269498943Sluigi		cmd = next_cmd(cmd);
269598943Sluigi	}
269698943Sluigi
2697101116Sluigi	if (have_state)	/* must be a check-state, we are done */
269898943Sluigi		goto done;
269998943Sluigi
270098943Sluigi#define OR_START(target)					\
270198943Sluigi	if (ac && (*av[0] == '(' || *av[0] == '{')) {		\
270298943Sluigi		if (open_par)					\
270398943Sluigi			errx(EX_USAGE, "nested \"(\" not allowed\n"); \
2704101641Sluigi		prev = NULL;					\
270598943Sluigi		open_par = 1;					\
270698943Sluigi		if ( (av[0])[1] == '\0') {			\
270798943Sluigi			ac--; av++;				\
270898943Sluigi		} else						\
270998943Sluigi			(*av)++;				\
271098943Sluigi	}							\
271198943Sluigi	target:							\
271298943Sluigi
271398943Sluigi
271498943Sluigi#define	CLOSE_PAR						\
271598943Sluigi	if (open_par) {						\
271698943Sluigi		if (ac && (					\
271798943Sluigi		    !strncmp(*av, ")", strlen(*av)) ||		\
271898943Sluigi		    !strncmp(*av, "}", strlen(*av)) )) {	\
2719101641Sluigi			prev = NULL;				\
272098943Sluigi			open_par = 0;				\
272198943Sluigi			ac--; av++;				\
272298943Sluigi		} else						\
272398943Sluigi			errx(EX_USAGE, "missing \")\"\n");	\
272498943Sluigi	}
2725106505Smaxim
272698943Sluigi#define NOT_BLOCK						\
272798943Sluigi	if (ac && !strncmp(*av, "not", strlen(*av))) {		\
272898943Sluigi		if (cmd->len & F_NOT)				\
272998943Sluigi			errx(EX_USAGE, "double \"not\" not allowed\n"); \
273098943Sluigi		cmd->len |= F_NOT;				\
273198943Sluigi		ac--; av++;					\
273298943Sluigi	}
273398943Sluigi
273498943Sluigi#define OR_BLOCK(target)					\
273598943Sluigi	if (ac && !strncmp(*av, "or", strlen(*av))) {		\
273698943Sluigi		if (prev == NULL || open_par == 0)		\
273798943Sluigi			errx(EX_DATAERR, "invalid OR block");	\
273898943Sluigi		prev->len |= F_OR;				\
273998943Sluigi		ac--; av++;					\
274098943Sluigi		goto target;					\
274198943Sluigi	}							\
274298943Sluigi	CLOSE_PAR;
274398943Sluigi
2744102087Sluigi	first_cmd = cmd;
2745102098Sluigi
2746102098Sluigi#if 0
274798943Sluigi	/*
2748102087Sluigi	 * MAC addresses, optional.
2749102087Sluigi	 * If we have this, we skip the part "proto from src to dst"
2750102087Sluigi	 * and jump straight to the option parsing.
2751102087Sluigi	 */
2752102087Sluigi	NOT_BLOCK;
2753102087Sluigi	NEED1("missing protocol");
2754102087Sluigi	if (!strncmp(*av, "MAC", strlen(*av)) ||
2755102087Sluigi	    !strncmp(*av, "mac", strlen(*av))) {
2756102087Sluigi		ac--; av++;	/* the "MAC" keyword */
2757102087Sluigi		add_mac(cmd, ac, av); /* exits in case of errors */
2758102087Sluigi		cmd = next_cmd(cmd);
2759102087Sluigi		ac -= 2; av += 2;	/* dst-mac and src-mac */
2760102087Sluigi		NOT_BLOCK;
2761102087Sluigi		NEED1("missing mac type");
2762102087Sluigi		if (add_mactype(cmd, ac, av[0]))
2763102087Sluigi			cmd = next_cmd(cmd);
2764102087Sluigi		ac--; av++;	/* any or mac-type */
2765102087Sluigi		goto read_options;
2766102087Sluigi	}
2767102098Sluigi#endif
2768102087Sluigi
2769102087Sluigi	/*
277098943Sluigi	 * protocol, mandatory
277198943Sluigi	 */
277298943Sluigi    OR_START(get_proto);
277398943Sluigi	NOT_BLOCK;
277498943Sluigi	NEED1("missing protocol");
2775102087Sluigi	if (add_proto(cmd, *av)) {
2776102087Sluigi		av++; ac--;
2777102087Sluigi		if (F_LEN(cmd) == 0)	/* plain IP */
2778102087Sluigi			proto = 0;
2779102087Sluigi		else {
2780102087Sluigi			proto = cmd->arg1;
2781102087Sluigi			prev = cmd;
2782102087Sluigi			cmd = next_cmd(cmd);
2783102087Sluigi		}
2784102098Sluigi	} else if (first_cmd != cmd) {
2785102098Sluigi		errx(EX_DATAERR, "invalid protocol ``%s''", av);
2786102098Sluigi	} else
2787102098Sluigi		goto read_options;
278898943Sluigi    OR_BLOCK(get_proto);
278998943Sluigi
279098943Sluigi	/*
2791102087Sluigi	 * "from", mandatory
279298943Sluigi	 */
2793102087Sluigi	if (!ac || strncmp(*av, "from", strlen(*av)))
279498943Sluigi		errx(EX_USAGE, "missing ``from''");
279598943Sluigi	ac--; av++;
279698943Sluigi
279798943Sluigi	/*
279898943Sluigi	 * source IP, mandatory
279998943Sluigi	 */
280098943Sluigi    OR_START(source_ip);
280198943Sluigi	NOT_BLOCK;	/* optional "not" */
280298943Sluigi	NEED1("missing source address");
2803102087Sluigi	if (add_srcip(cmd, *av)) {
2804102087Sluigi		ac--; av++;
2805102087Sluigi		if (F_LEN(cmd) != 0) {	/* ! any */
2806102087Sluigi			prev = cmd;
2807102087Sluigi			cmd = next_cmd(cmd);
2808102087Sluigi		}
2809102087Sluigi	}
281098943Sluigi    OR_BLOCK(source_ip);
281198943Sluigi
281298943Sluigi	/*
281398943Sluigi	 * source ports, optional
281498943Sluigi	 */
281598943Sluigi	NOT_BLOCK;	/* optional "not" */
2816101641Sluigi	if (ac) {
2817102087Sluigi		if (!strncmp(*av, "any", strlen(*av)) ||
2818102087Sluigi		    add_ports(cmd, *av, proto, O_IP_SRCPORT)) {
2819102087Sluigi			ac--; av++;
2820102087Sluigi			if (F_LEN(cmd) != 0)
2821102087Sluigi				cmd = next_cmd(cmd);
2822101641Sluigi		}
282398943Sluigi	}
282498943Sluigi
282598943Sluigi	/*
2826102087Sluigi	 * "to", mandatory
282798943Sluigi	 */
2828102087Sluigi	if (!ac || strncmp(*av, "to", strlen(*av)))
282998943Sluigi		errx(EX_USAGE, "missing ``to''");
283098943Sluigi	av++; ac--;
283198943Sluigi
283298943Sluigi	/*
283398943Sluigi	 * destination, mandatory
283498943Sluigi	 */
283598943Sluigi    OR_START(dest_ip);
283698943Sluigi	NOT_BLOCK;	/* optional "not" */
283798943Sluigi	NEED1("missing dst address");
2838102087Sluigi	if (add_dstip(cmd, *av)) {
2839102087Sluigi		ac--; av++;
2840102087Sluigi		if (F_LEN(cmd) != 0) {	/* ! any */
2841102087Sluigi			prev = cmd;
2842102087Sluigi			cmd = next_cmd(cmd);
2843102087Sluigi		}
2844102087Sluigi	}
284598943Sluigi    OR_BLOCK(dest_ip);
284698943Sluigi
284798943Sluigi	/*
284898943Sluigi	 * dest. ports, optional
284998943Sluigi	 */
285098943Sluigi	NOT_BLOCK;	/* optional "not" */
2851101641Sluigi	if (ac) {
2852102087Sluigi		if (!strncmp(*av, "any", strlen(*av)) ||
2853102087Sluigi		    add_ports(cmd, *av, proto, O_IP_DSTPORT)) {
2854102087Sluigi			ac--; av++;
2855102087Sluigi			if (F_LEN(cmd) != 0)
2856102087Sluigi				cmd = next_cmd(cmd);
2857101641Sluigi		}
285898943Sluigi	}
285998943Sluigi
286098943Sluigiread_options:
2861102087Sluigi	if (ac && first_cmd == cmd) {
2862102087Sluigi		/*
2863102087Sluigi		 * nothing specified so far, store in the rule to ease
2864102087Sluigi		 * printout later.
2865102087Sluigi		 */
2866102087Sluigi		 rule->_pad = 1;
2867102087Sluigi	}
286898943Sluigi	prev = NULL;
286998943Sluigi	while (ac) {
2870101641Sluigi		char *s;
2871101641Sluigi		ipfw_insn_u32 *cmd32;	/* alias for cmd */
287298943Sluigi
2873101641Sluigi		s = *av;
2874101641Sluigi		cmd32 = (ipfw_insn_u32 *)cmd;
2875101641Sluigi
287698943Sluigi		if (*s == '!') {	/* alternate syntax for NOT */
287798943Sluigi			if (cmd->len & F_NOT)
287898943Sluigi				errx(EX_USAGE, "double \"not\" not allowed\n");
287998943Sluigi			cmd->len = F_NOT;
288098943Sluigi			s++;
288198943Sluigi		}
288298943Sluigi		i = match_token(rule_options, s);
288398943Sluigi		ac--; av++;
288498943Sluigi		switch(i) {
288598943Sluigi		case TOK_NOT:
288698943Sluigi			if (cmd->len & F_NOT)
288798943Sluigi				errx(EX_USAGE, "double \"not\" not allowed\n");
288898943Sluigi			cmd->len = F_NOT;
288998943Sluigi			break;
289098943Sluigi
289198943Sluigi		case TOK_OR:
2892101641Sluigi			if (open_par == 0 || prev == NULL)
289398943Sluigi				errx(EX_USAGE, "invalid \"or\" block\n");
289498943Sluigi			prev->len |= F_OR;
289598943Sluigi			break;
2896101641Sluigi
2897101641Sluigi		case TOK_STARTBRACE:
2898101641Sluigi			if (open_par)
2899101641Sluigi				errx(EX_USAGE, "+nested \"(\" not allowed\n");
2900101641Sluigi			open_par = 1;
2901101641Sluigi			break;
2902101641Sluigi
2903101641Sluigi		case TOK_ENDBRACE:
2904101641Sluigi			if (!open_par)
2905101641Sluigi				errx(EX_USAGE, "+missing \")\"\n");
2906101641Sluigi			open_par = 0;
2907102087Sluigi			prev = NULL;
2908101641Sluigi        		break;
2909101641Sluigi
291098943Sluigi		case TOK_IN:
291198943Sluigi			fill_cmd(cmd, O_IN, 0, 0);
291298943Sluigi			break;
291398943Sluigi
291498943Sluigi		case TOK_OUT:
291598943Sluigi			cmd->len ^= F_NOT; /* toggle F_NOT */
291698943Sluigi			fill_cmd(cmd, O_IN, 0, 0);
291798943Sluigi			break;
291898943Sluigi
291998943Sluigi		case TOK_FRAG:
292098943Sluigi			fill_cmd(cmd, O_FRAG, 0, 0);
292198943Sluigi			break;
292298943Sluigi
292398943Sluigi		case TOK_LAYER2:
292498943Sluigi			fill_cmd(cmd, O_LAYER2, 0, 0);
292598943Sluigi			break;
292698943Sluigi
292798943Sluigi		case TOK_XMIT:
292898943Sluigi		case TOK_RECV:
292998943Sluigi		case TOK_VIA:
293098943Sluigi			NEED1("recv, xmit, via require interface name"
293198943Sluigi				" or address");
293298943Sluigi			fill_iface((ipfw_insn_if *)cmd, av[0]);
293398943Sluigi			ac--; av++;
293498943Sluigi			if (F_LEN(cmd) == 0)	/* not a valid address */
293598943Sluigi				break;
293698943Sluigi			if (i == TOK_XMIT)
293798943Sluigi				cmd->opcode = O_XMIT;
293898943Sluigi			else if (i == TOK_RECV)
293998943Sluigi				cmd->opcode = O_RECV;
294098943Sluigi			else if (i == TOK_VIA)
294198943Sluigi				cmd->opcode = O_VIA;
294298943Sluigi			break;
294398943Sluigi
294499475Sluigi		case TOK_ICMPTYPES:
294599475Sluigi			NEED1("icmptypes requires list of types");
294699475Sluigi			fill_icmptypes((ipfw_insn_u32 *)cmd, *av);
294799475Sluigi			av++; ac--;
294899475Sluigi			break;
294999475Sluigi
295098943Sluigi		case TOK_IPTTL:
295198943Sluigi			NEED1("ipttl requires TTL");
295298943Sluigi			fill_cmd(cmd, O_IPTTL, 0, strtoul(*av, NULL, 0));
295398943Sluigi			ac--; av++;
295498943Sluigi			break;
295598943Sluigi
295698943Sluigi		case TOK_IPID:
295798943Sluigi			NEED1("ipid requires length");
295898943Sluigi			fill_cmd(cmd, O_IPID, 0, strtoul(*av, NULL, 0));
295998943Sluigi			ac--; av++;
296098943Sluigi			break;
296198943Sluigi
296298943Sluigi		case TOK_IPLEN:
296398943Sluigi			NEED1("iplen requires length");
296498943Sluigi			fill_cmd(cmd, O_IPLEN, 0, strtoul(*av, NULL, 0));
296598943Sluigi			ac--; av++;
296698943Sluigi			break;
296798943Sluigi
296898943Sluigi		case TOK_IPVER:
296998943Sluigi			NEED1("ipver requires version");
297098943Sluigi			fill_cmd(cmd, O_IPVER, 0, strtoul(*av, NULL, 0));
297198943Sluigi			ac--; av++;
297298943Sluigi			break;
297398943Sluigi
297499475Sluigi		case TOK_IPPRECEDENCE:
297599475Sluigi			NEED1("ipprecedence requires value");
297699475Sluigi			fill_cmd(cmd, O_IPPRECEDENCE, 0,
297799475Sluigi			    (strtoul(*av, NULL, 0) & 7) << 5);
297899475Sluigi			ac--; av++;
297999475Sluigi			break;
298099475Sluigi
298198943Sluigi		case TOK_IPOPTS:
298298943Sluigi			NEED1("missing argument for ipoptions");
2983101116Sluigi			fill_flags(cmd, O_IPOPT, f_ipopts, *av);
298498943Sluigi			ac--; av++;
298598943Sluigi			break;
298698943Sluigi
298799475Sluigi		case TOK_IPTOS:
298899475Sluigi			NEED1("missing argument for iptos");
2989101116Sluigi			fill_flags(cmd, O_IPTOS, f_iptos, *av);
299099475Sluigi			ac--; av++;
299199475Sluigi			break;
299299475Sluigi
299398943Sluigi		case TOK_UID:
299498943Sluigi			NEED1("uid requires argument");
299598943Sluigi		    {
299698943Sluigi			char *end;
299798943Sluigi			uid_t uid;
299898943Sluigi			struct passwd *pwd;
299998943Sluigi
300098943Sluigi			cmd->opcode = O_UID;
300198943Sluigi			uid = strtoul(*av, &end, 0);
300298943Sluigi			pwd = (*end == '\0') ? getpwuid(uid) : getpwnam(*av);
300398943Sluigi			if (pwd == NULL)
300498943Sluigi				errx(EX_DATAERR, "uid \"%s\" nonexistent", *av);
3005106504Smaxim			cmd32->d[0] = pwd->pw_uid;
300698943Sluigi			cmd->len = F_INSN_SIZE(ipfw_insn_u32);
300798943Sluigi			ac--; av++;
300898943Sluigi		    }
300998943Sluigi			break;
301098943Sluigi
301198943Sluigi		case TOK_GID:
301298943Sluigi			NEED1("gid requires argument");
301398943Sluigi		    {
301498943Sluigi			char *end;
301598943Sluigi			gid_t gid;
301698943Sluigi			struct group *grp;
301798943Sluigi
301898943Sluigi			cmd->opcode = O_GID;
301998943Sluigi			gid = strtoul(*av, &end, 0);
302098943Sluigi			grp = (*end == '\0') ? getgrgid(gid) : getgrnam(*av);
302198943Sluigi			if (grp == NULL)
302298943Sluigi				errx(EX_DATAERR, "gid \"%s\" nonexistent", *av);
3023106504Smaxim			cmd32->d[0] = grp->gr_gid;
302498943Sluigi			cmd->len = F_INSN_SIZE(ipfw_insn_u32);
302598943Sluigi			ac--; av++;
302698943Sluigi		    }
302798943Sluigi			break;
302898943Sluigi
302998943Sluigi		case TOK_ESTAB:
303098943Sluigi			fill_cmd(cmd, O_ESTAB, 0, 0);
303198943Sluigi			break;
303298943Sluigi
303398943Sluigi		case TOK_SETUP:
303498943Sluigi			fill_cmd(cmd, O_TCPFLAGS, 0,
303598943Sluigi				(TH_SYN) | ( (TH_ACK) & 0xff) <<8 );
303698943Sluigi			break;
303798943Sluigi
303898943Sluigi		case TOK_TCPOPTS:
303998943Sluigi			NEED1("missing argument for tcpoptions");
304098943Sluigi			fill_flags(cmd, O_TCPOPTS, f_tcpopts, *av);
304198943Sluigi			ac--; av++;
304298943Sluigi			break;
304398943Sluigi
304498943Sluigi		case TOK_TCPSEQ:
304598943Sluigi		case TOK_TCPACK:
304698943Sluigi			NEED1("tcpseq/tcpack requires argument");
304798943Sluigi			cmd->len = F_INSN_SIZE(ipfw_insn_u32);
304898943Sluigi			cmd->opcode = (i == TOK_TCPSEQ) ? O_TCPSEQ : O_TCPACK;
304998943Sluigi			cmd32->d[0] = htonl(strtoul(*av, NULL, 0));
305098943Sluigi			ac--; av++;
305198943Sluigi			break;
305298943Sluigi
305398943Sluigi		case TOK_TCPWIN:
305498943Sluigi			NEED1("tcpwin requires length");
305598943Sluigi			fill_cmd(cmd, O_TCPWIN, 0,
305698943Sluigi			    htons(strtoul(*av, NULL, 0)));
305798943Sluigi			ac--; av++;
305898943Sluigi			break;
305998943Sluigi
306098943Sluigi		case TOK_TCPFLAGS:
306198943Sluigi			NEED1("missing argument for tcpflags");
306298943Sluigi			cmd->opcode = O_TCPFLAGS;
306398943Sluigi			fill_flags(cmd, O_TCPFLAGS, f_tcpflags, *av);
306498943Sluigi			ac--; av++;
306598943Sluigi			break;
306698943Sluigi
306798943Sluigi		case TOK_KEEPSTATE:
3068101641Sluigi			if (open_par)
3069101641Sluigi				errx(EX_USAGE, "keep-state cannot be part "
3070101641Sluigi				    "of an or block");
307199909Sluigi			if (have_state)
3072101116Sluigi				errx(EX_USAGE, "only one of keep-state "
307399909Sluigi					"and limit is allowed");
3074101116Sluigi			have_state = cmd;
307598943Sluigi			fill_cmd(cmd, O_KEEP_STATE, 0, 0);
307698943Sluigi			break;
307798943Sluigi
307898943Sluigi		case TOK_LIMIT:
3079101641Sluigi			if (open_par)
3080101641Sluigi				errx(EX_USAGE, "limit cannot be part "
3081101641Sluigi				    "of an or block");
308299909Sluigi			if (have_state)
3083101116Sluigi				errx(EX_USAGE, "only one of keep-state "
308499909Sluigi					"and limit is allowed");
3085101641Sluigi			NEED1("limit needs mask and # of connections");
3086101116Sluigi			have_state = cmd;
308798943Sluigi		    {
308898943Sluigi			ipfw_insn_limit *c = (ipfw_insn_limit *)cmd;
308998943Sluigi
309098943Sluigi			cmd->len = F_INSN_SIZE(ipfw_insn_limit);
309198943Sluigi			cmd->opcode = O_LIMIT;
309298943Sluigi			c->limit_mask = 0;
309398943Sluigi			c->conn_limit = 0;
309498943Sluigi			for (; ac >1 ;) {
309598943Sluigi				int val;
309698943Sluigi
309798943Sluigi				val = match_token(limit_masks, *av);
309898943Sluigi				if (val <= 0)
309998943Sluigi					break;
310098943Sluigi				c->limit_mask |= val;
310198943Sluigi				ac--; av++;
310298943Sluigi			}
310398943Sluigi			c->conn_limit = atoi(*av);
310498943Sluigi			if (c->conn_limit == 0)
310598943Sluigi				errx(EX_USAGE, "limit: limit must be >0");
310698943Sluigi			if (c->limit_mask == 0)
310798943Sluigi				errx(EX_USAGE, "missing limit mask");
310898943Sluigi			ac--; av++;
310998943Sluigi		    }
311098943Sluigi			break;
311198943Sluigi
3112102087Sluigi		case TOK_PROTO:
3113102087Sluigi			NEED1("missing protocol");
3114102087Sluigi			if (add_proto(cmd, *av)) {
3115102087Sluigi				proto = cmd->arg1;
3116102087Sluigi				ac--; av++;
3117102098Sluigi			} else
3118102098Sluigi				errx(EX_DATAERR, "invalid protocol ``%s''", av);
3119102087Sluigi			break;
3120106505Smaxim
3121102087Sluigi		case TOK_SRCIP:
3122102087Sluigi			NEED1("missing source IP");
3123102087Sluigi			if (add_srcip(cmd, *av)) {
3124102087Sluigi				ac--; av++;
3125102087Sluigi			}
3126102087Sluigi			break;
3127102087Sluigi
3128102087Sluigi		case TOK_DSTIP:
3129102087Sluigi			NEED1("missing destination IP");
3130102087Sluigi			if (add_dstip(cmd, *av)) {
3131102087Sluigi				ac--; av++;
3132102087Sluigi			}
3133102087Sluigi			break;
3134102087Sluigi
3135102087Sluigi		case TOK_SRCPORT:
3136102087Sluigi			NEED1("missing source port");
3137102087Sluigi			if (!strncmp(*av, "any", strlen(*av)) ||
3138102087Sluigi			    add_ports(cmd, *av, proto, O_IP_SRCPORT)) {
3139102087Sluigi				ac--; av++;
3140102087Sluigi			} else
3141102087Sluigi				errx(EX_DATAERR, "invalid source port %s", *av);
3142102087Sluigi			break;
3143102087Sluigi
3144102087Sluigi		case TOK_DSTPORT:
3145102087Sluigi			NEED1("missing destination port");
3146102087Sluigi			if (!strncmp(*av, "any", strlen(*av)) ||
3147102087Sluigi			    add_ports(cmd, *av, proto, O_IP_DSTPORT)) {
3148102087Sluigi				ac--; av++;
3149102087Sluigi			} else
3150102087Sluigi				errx(EX_DATAERR, "invalid destination port %s",
3151102087Sluigi				    *av);
3152102087Sluigi			break;
3153102087Sluigi
3154102087Sluigi		case TOK_MAC:
3155102087Sluigi			if (ac < 2)
3156102087Sluigi				errx(EX_USAGE, "MAC dst-mac src-mac");
3157102087Sluigi			if (add_mac(cmd, ac, av)) {
3158102087Sluigi				ac -= 2; av += 2;
3159102087Sluigi			}
3160102087Sluigi			break;
3161102087Sluigi
3162102087Sluigi		case TOK_MACTYPE:
3163102087Sluigi			NEED1("missing mac type");
3164102087Sluigi			if (!add_mactype(cmd, ac, *av))
3165102087Sluigi				errx(EX_DATAERR, "invalid mac type %s", av);
3166102087Sluigi			ac--; av++;
3167102087Sluigi			break;
3168102087Sluigi
3169112250Scjc		case TOK_VERREVPATH:
3170112250Scjc			fill_cmd(cmd, O_VERREVPATH, 0, 0);
3171112250Scjc			break;
3172112250Scjc
317398943Sluigi		default:
317498943Sluigi			errx(EX_USAGE, "unrecognised option [%d] %s\n", i, s);
317598943Sluigi		}
317698943Sluigi		if (F_LEN(cmd) > 0) {	/* prepare to advance */
317798943Sluigi			prev = cmd;
317898943Sluigi			cmd = next_cmd(cmd);
317998943Sluigi		}
318098943Sluigi	}
318198943Sluigi
318298943Sluigidone:
318398943Sluigi	/*
318498943Sluigi	 * Now copy stuff into the rule.
318598943Sluigi	 * If we have a keep-state option, the first instruction
318698943Sluigi	 * must be a PROBE_STATE (which is generated here).
318798943Sluigi	 * If we have a LOG option, it was stored as the first command,
318898943Sluigi	 * and now must be moved to the top of the action part.
318998943Sluigi	 */
319098943Sluigi	dst = (ipfw_insn *)rule->cmd;
319198943Sluigi
319298943Sluigi	/*
3193107289Sluigi	 * First thing to write into the command stream is the match probability.
3194107289Sluigi	 */
3195107289Sluigi	if (match_prob != 1) { /* 1 means always match */
3196107289Sluigi		dst->opcode = O_PROB;
3197107289Sluigi		dst->len = 2;
3198107289Sluigi		*((int32_t *)(dst+1)) = (int32_t)(match_prob * 0x7fffffff);
3199107289Sluigi		dst += dst->len;
3200107289Sluigi	}
3201107289Sluigi
3202107289Sluigi	/*
320398943Sluigi	 * generate O_PROBE_STATE if necessary
320498943Sluigi	 */
3205101116Sluigi	if (have_state && have_state->opcode != O_CHECK_STATE) {
320698943Sluigi		fill_cmd(dst, O_PROBE_STATE, 0, 0);
320798943Sluigi		dst = next_cmd(dst);
320898943Sluigi	}
320998943Sluigi	/*
3210101116Sluigi	 * copy all commands but O_LOG, O_KEEP_STATE, O_LIMIT
321198943Sluigi	 */
321298943Sluigi	for (src = (ipfw_insn *)cmdbuf; src != cmd; src += i) {
321398943Sluigi		i = F_LEN(src);
321498943Sluigi
3215101116Sluigi		switch (src->opcode) {
3216101116Sluigi		case O_LOG:
3217101116Sluigi		case O_KEEP_STATE:
3218101116Sluigi		case O_LIMIT:
3219101116Sluigi			break;
3220101116Sluigi		default:
322198943Sluigi			bcopy(src, dst, i * sizeof(u_int32_t));
322298943Sluigi			dst += i;
322398943Sluigi		}
322498943Sluigi	}
322598943Sluigi
322698943Sluigi	/*
3227101116Sluigi	 * put back the have_state command as last opcode
3228101116Sluigi	 */
3229101295Sluigi	if (have_state && have_state->opcode != O_CHECK_STATE) {
3230101116Sluigi		i = F_LEN(have_state);
3231101116Sluigi		bcopy(have_state, dst, i * sizeof(u_int32_t));
3232101116Sluigi		dst += i;
3233101116Sluigi	}
3234101116Sluigi	/*
323598943Sluigi	 * start action section
323698943Sluigi	 */
323798943Sluigi	rule->act_ofs = dst - rule->cmd;
323898943Sluigi
323998943Sluigi	/*
324098943Sluigi	 * put back O_LOG if necessary
324198943Sluigi	 */
324298943Sluigi	src = (ipfw_insn *)cmdbuf;
324398943Sluigi	if ( src->opcode == O_LOG ) {
324498943Sluigi		i = F_LEN(src);
324598943Sluigi		bcopy(src, dst, i * sizeof(u_int32_t));
324698943Sluigi		dst += i;
324798943Sluigi	}
324898943Sluigi	/*
324998943Sluigi	 * copy all other actions
325098943Sluigi	 */
325198943Sluigi	for (src = (ipfw_insn *)actbuf; src != action; src += i) {
325298943Sluigi		i = F_LEN(src);
325398943Sluigi		bcopy(src, dst, i * sizeof(u_int32_t));
325498943Sluigi		dst += i;
325598943Sluigi	}
325698943Sluigi
325798943Sluigi	rule->cmd_len = (u_int32_t *)dst - (u_int32_t *)(rule->cmd);
325898943Sluigi	i = (void *)dst - (void *)rule;
325998943Sluigi	if (getsockopt(s, IPPROTO_IP, IP_FW_ADD, rule, &i) == -1)
326098943Sluigi		err(EX_UNAVAILABLE, "getsockopt(%s)", "IP_FW_ADD");
326198943Sluigi	if (!do_quiet)
3262112189Smaxim		show_ipfw(rule, 10, 10);
326398943Sluigi}
326498943Sluigi
326598943Sluigistatic void
3266101978Sluigizero(int ac, char *av[])
326798943Sluigi{
326898943Sluigi	int rulenum;
326998943Sluigi	int failed = EX_OK;
327098943Sluigi
327198943Sluigi	av++; ac--;
327298943Sluigi
327398943Sluigi	if (!ac) {
327498943Sluigi		/* clear all entries */
327598943Sluigi		if (setsockopt(s, IPPROTO_IP, IP_FW_ZERO, NULL, 0) < 0)
327698943Sluigi			err(EX_UNAVAILABLE, "setsockopt(%s)", "IP_FW_ZERO");
327798943Sluigi		if (!do_quiet)
327898943Sluigi			printf("Accounting cleared.\n");
327998943Sluigi
328098943Sluigi		return;
328198943Sluigi	}
328298943Sluigi
328398943Sluigi	while (ac) {
328498943Sluigi		/* Rule number */
328598943Sluigi		if (isdigit(**av)) {
328698943Sluigi			rulenum = atoi(*av);
328798943Sluigi			av++;
328898943Sluigi			ac--;
328998943Sluigi			if (setsockopt(s, IPPROTO_IP,
329098943Sluigi			    IP_FW_ZERO, &rulenum, sizeof rulenum)) {
329198943Sluigi				warn("rule %u: setsockopt(IP_FW_ZERO)",
329298943Sluigi				    rulenum);
329398943Sluigi				failed = EX_UNAVAILABLE;
329498943Sluigi			} else if (!do_quiet)
329598943Sluigi				printf("Entry %d cleared\n", rulenum);
329698943Sluigi		} else {
329798943Sluigi			errx(EX_USAGE, "invalid rule number ``%s''", *av);
329898943Sluigi		}
329998943Sluigi	}
330098943Sluigi	if (failed != EX_OK)
330198943Sluigi		exit(failed);
330298943Sluigi}
330398943Sluigi
330498943Sluigistatic void
3305101978Sluigiresetlog(int ac, char *av[])
330698943Sluigi{
330798943Sluigi	int rulenum;
330898943Sluigi	int failed = EX_OK;
330998943Sluigi
331098943Sluigi	av++; ac--;
331198943Sluigi
331298943Sluigi	if (!ac) {
331398943Sluigi		/* clear all entries */
331498943Sluigi		if (setsockopt(s, IPPROTO_IP, IP_FW_RESETLOG, NULL, 0) < 0)
331598943Sluigi			err(EX_UNAVAILABLE, "setsockopt(IP_FW_RESETLOG)");
331698943Sluigi		if (!do_quiet)
331798943Sluigi			printf("Logging counts reset.\n");
331898943Sluigi
331998943Sluigi		return;
332098943Sluigi	}
332198943Sluigi
332298943Sluigi	while (ac) {
332398943Sluigi		/* Rule number */
332498943Sluigi		if (isdigit(**av)) {
332598943Sluigi			rulenum = atoi(*av);
332698943Sluigi			av++;
332798943Sluigi			ac--;
332898943Sluigi			if (setsockopt(s, IPPROTO_IP,
332998943Sluigi			    IP_FW_RESETLOG, &rulenum, sizeof rulenum)) {
333098943Sluigi				warn("rule %u: setsockopt(IP_FW_RESETLOG)",
333198943Sluigi				    rulenum);
333298943Sluigi				failed = EX_UNAVAILABLE;
333398943Sluigi			} else if (!do_quiet)
333498943Sluigi				printf("Entry %d logging count reset\n",
333598943Sluigi				    rulenum);
333698943Sluigi		} else {
333798943Sluigi			errx(EX_DATAERR, "invalid rule number ``%s''", *av);
333898943Sluigi		}
333998943Sluigi	}
334098943Sluigi	if (failed != EX_OK)
334198943Sluigi		exit(failed);
334298943Sluigi}
334398943Sluigi
334498943Sluigistatic void
334598943Sluigiflush()
334698943Sluigi{
334798943Sluigi	int cmd = do_pipe ? IP_DUMMYNET_FLUSH : IP_FW_FLUSH;
334898943Sluigi
334998943Sluigi	if (!do_force && !do_quiet) { /* need to ask user */
335098943Sluigi		int c;
335198943Sluigi
335298943Sluigi		printf("Are you sure? [yn] ");
335398943Sluigi		fflush(stdout);
335498943Sluigi		do {
335598943Sluigi			c = toupper(getc(stdin));
335698943Sluigi			while (c != '\n' && getc(stdin) != '\n')
335798943Sluigi				if (feof(stdin))
335898943Sluigi					return; /* and do not flush */
335998943Sluigi		} while (c != 'Y' && c != 'N');
336098943Sluigi		printf("\n");
336198943Sluigi		if (c == 'N')	/* user said no */
336298943Sluigi			return;
336398943Sluigi	}
336498943Sluigi	if (setsockopt(s, IPPROTO_IP, cmd, NULL, 0) < 0)
336598943Sluigi		err(EX_UNAVAILABLE, "setsockopt(IP_%s_FLUSH)",
336698943Sluigi		    do_pipe ? "DUMMYNET" : "FW");
336798943Sluigi	if (!do_quiet)
336898943Sluigi		printf("Flushed all %s.\n", do_pipe ? "pipes" : "rules");
336998943Sluigi}
337098943Sluigi
337198943Sluigistatic int
337298943Sluigiipfw_main(int ac, char **av)
337398943Sluigi{
337498943Sluigi	int ch;
337598943Sluigi
337698943Sluigi	if (ac == 1)
337798943Sluigi		show_usage();
337898943Sluigi
337998943Sluigi	/* Set the force flag for non-interactive processes */
338098943Sluigi	do_force = !isatty(STDIN_FILENO);
338198943Sluigi
338298943Sluigi	optind = optreset = 1;
3383102098Sluigi	while ((ch = getopt(ac, av, "hs:acdefNqStv")) != -1)
338498943Sluigi		switch (ch) {
338598943Sluigi		case 'h': /* help */
338698943Sluigi			help();
338798943Sluigi			break;	/* NOTREACHED */
338898943Sluigi
338998943Sluigi		case 's': /* sort */
339098943Sluigi			do_sort = atoi(optarg);
339198943Sluigi			break;
339298943Sluigi		case 'a':
339398943Sluigi			do_acct = 1;
339498943Sluigi			break;
3395102098Sluigi		case 'c':
3396102098Sluigi			do_compact = 1;
3397102098Sluigi			break;
339898943Sluigi		case 'd':
339998943Sluigi			do_dynamic = 1;
340098943Sluigi			break;
340198943Sluigi		case 'e':
340298943Sluigi			do_expired = 1;
340398943Sluigi			break;
340498943Sluigi		case 'f':
340598943Sluigi			do_force = 1;
340698943Sluigi			break;
340798943Sluigi		case 'N':
340898943Sluigi			do_resolv = 1;
340998943Sluigi			break;
341098943Sluigi		case 'q':
341198943Sluigi			do_quiet = 1;
341298943Sluigi			break;
3413101628Sluigi		case 'S':
3414101628Sluigi			show_sets = 1;
3415101628Sluigi			break;
341698943Sluigi		case 't':
341798943Sluigi			do_time = 1;
341898943Sluigi			break;
341998943Sluigi		case 'v': /* verbose */
342098943Sluigi			verbose++;
342198943Sluigi			break;
342298943Sluigi		default:
342398943Sluigi			show_usage();
342498943Sluigi		}
342598943Sluigi
342698943Sluigi	ac -= optind;
342798943Sluigi	av += optind;
342898943Sluigi	NEED1("bad arguments, for usage summary ``ipfw''");
342998943Sluigi
343098943Sluigi	/*
343198943Sluigi	 * optional: pipe or queue
343298943Sluigi	 */
343398943Sluigi	if (!strncmp(*av, "pipe", strlen(*av))) {
343498943Sluigi		do_pipe = 1;
343598943Sluigi		ac--;
343698943Sluigi		av++;
343798943Sluigi	} else if (!strncmp(*av, "queue", strlen(*av))) {
343898943Sluigi		do_pipe = 2;
343998943Sluigi		ac--;
344098943Sluigi		av++;
344198943Sluigi	}
344298943Sluigi	NEED1("missing command");
344398943Sluigi
344498943Sluigi	/*
344598943Sluigi	 * for pipes and queues we normally say 'pipe NN config'
344698943Sluigi	 * but the code is easier to parse as 'pipe config NN'
344798943Sluigi	 * so we swap the two arguments.
344898943Sluigi	 */
344998943Sluigi	if (do_pipe > 0 && ac > 1 && *av[0] >= '0' && *av[0] <= '9') {
345098943Sluigi		char *p = av[0];
345198943Sluigi		av[0] = av[1];
345298943Sluigi		av[1] = p;
345398943Sluigi	}
345498943Sluigi	if (!strncmp(*av, "add", strlen(*av)))
345598943Sluigi		add(ac, av);
345698943Sluigi	else if (do_pipe && !strncmp(*av, "config", strlen(*av)))
345798943Sluigi		config_pipe(ac, av);
3458101978Sluigi	else if (!strncmp(*av, "delete", strlen(*av)))
345998943Sluigi		delete(ac, av);
346098943Sluigi	else if (!strncmp(*av, "flush", strlen(*av)))
346198943Sluigi		flush();
346298943Sluigi	else if (!strncmp(*av, "zero", strlen(*av)))
346398943Sluigi		zero(ac, av);
346498943Sluigi	else if (!strncmp(*av, "resetlog", strlen(*av)))
346598943Sluigi		resetlog(ac, av);
346698943Sluigi	else if (!strncmp(*av, "print", strlen(*av)) ||
346798943Sluigi	         !strncmp(*av, "list", strlen(*av)))
346898943Sluigi		list(ac, av);
3469101978Sluigi	else if (!strncmp(*av, "set", strlen(*av)))
3470101978Sluigi		sets_handler(ac, av);
3471109126Sdillon	else if (!strncmp(*av, "enable", strlen(*av)))
3472109126Sdillon		sysctl_handler(ac, av, 1);
3473109126Sdillon	else if (!strncmp(*av, "disable", strlen(*av)))
3474109126Sdillon		sysctl_handler(ac, av, 0);
347598943Sluigi	else if (!strncmp(*av, "show", strlen(*av))) {
347698943Sluigi		do_acct++;
347798943Sluigi		list(ac, av);
347898943Sluigi	} else
347998943Sluigi		errx(EX_USAGE, "bad command `%s'", *av);
348098943Sluigi	return 0;
348198943Sluigi}
348298943Sluigi
348398943Sluigi
348498943Sluigistatic void
3485106505Smaximipfw_readfile(int ac, char *av[])
348698943Sluigi{
348798943Sluigi#define MAX_ARGS	32
348898943Sluigi#define WHITESP		" \t\f\v\n\r"
348998943Sluigi	char	buf[BUFSIZ];
349098943Sluigi	char	*a, *p, *args[MAX_ARGS], *cmd = NULL;
349198943Sluigi	char	linename[10];
349298943Sluigi	int	i=0, lineno=0, qflag=0, pflag=0, status;
349398943Sluigi	FILE	*f = NULL;
349498943Sluigi	pid_t	preproc = 0;
349598943Sluigi	int	c;
349698943Sluigi
3497108231Skbyanc	while ((c = getopt(ac, av, "p:q")) != -1) {
349898943Sluigi		switch(c) {
349998943Sluigi		case 'p':
350098943Sluigi			pflag = 1;
350198943Sluigi			cmd = optarg;
350298943Sluigi			args[0] = cmd;
350398943Sluigi			i = 1;
350498943Sluigi			break;
350598943Sluigi
350698943Sluigi		case 'q':
350798943Sluigi			qflag = 1;
350898943Sluigi			break;
350998943Sluigi
351098943Sluigi		default:
351198943Sluigi			errx(EX_USAGE, "bad arguments, for usage"
351298943Sluigi			     " summary ``ipfw''");
351398943Sluigi		}
351498943Sluigi
3515108231Skbyanc		if (pflag)
3516108231Skbyanc			break;
3517108231Skbyanc	}
3518108231Skbyanc
3519108231Skbyanc	if (pflag) {
3520108231Skbyanc		/* Pass all but the last argument to the preprocessor. */
3521108231Skbyanc		while (optind < ac - 1) {
3522108231Skbyanc			if (i >= MAX_ARGS)
3523108231Skbyanc				errx(EX_USAGE, "too many preprocessor options");
3524108231Skbyanc			args[i++] = av[optind++];
3525108231Skbyanc		}
3526108231Skbyanc	}
3527108231Skbyanc
352898943Sluigi	av += optind;
352998943Sluigi	ac -= optind;
353098943Sluigi	if (ac != 1)
353198943Sluigi		errx(EX_USAGE, "extraneous filename arguments");
353298943Sluigi
353398943Sluigi	if ((f = fopen(av[0], "r")) == NULL)
353498943Sluigi		err(EX_UNAVAILABLE, "fopen: %s", av[0]);
353598943Sluigi
353698943Sluigi	if (pflag) {
353798943Sluigi		/* pipe through preprocessor (cpp or m4) */
353898943Sluigi		int pipedes[2];
353998943Sluigi
354098943Sluigi		args[i] = 0;
354198943Sluigi
354298943Sluigi		if (pipe(pipedes) == -1)
354398943Sluigi			err(EX_OSERR, "cannot create pipe");
354498943Sluigi
354598943Sluigi		switch((preproc = fork())) {
354698943Sluigi		case -1:
354798943Sluigi			err(EX_OSERR, "cannot fork");
354898943Sluigi
354998943Sluigi		case 0:
355098943Sluigi			/* child */
355198943Sluigi			if (dup2(fileno(f), 0) == -1
355298943Sluigi			    || dup2(pipedes[1], 1) == -1)
355398943Sluigi				err(EX_OSERR, "dup2()");
355498943Sluigi			fclose(f);
355598943Sluigi			close(pipedes[1]);
355698943Sluigi			close(pipedes[0]);
355798943Sluigi			execvp(cmd, args);
355898943Sluigi			err(EX_OSERR, "execvp(%s) failed", cmd);
355998943Sluigi
356098943Sluigi		default:
356198943Sluigi			/* parent */
356298943Sluigi			fclose(f);
356398943Sluigi			close(pipedes[1]);
356498943Sluigi			if ((f = fdopen(pipedes[0], "r")) == NULL) {
356598943Sluigi				int savederrno = errno;
356698943Sluigi
356798943Sluigi				(void)kill(preproc, SIGTERM);
356898943Sluigi				errno = savederrno;
356998943Sluigi				err(EX_OSERR, "fdopen()");
357098943Sluigi			}
357198943Sluigi		}
357298943Sluigi	}
357398943Sluigi
357498943Sluigi	while (fgets(buf, BUFSIZ, f)) {
357598943Sluigi		lineno++;
357698943Sluigi		sprintf(linename, "Line %d", lineno);
357798943Sluigi		args[0] = linename;
357898943Sluigi
357998943Sluigi		if (*buf == '#')
358098943Sluigi			continue;
358198943Sluigi		if ((p = strchr(buf, '#')) != NULL)
358298943Sluigi			*p = '\0';
358398943Sluigi		i = 1;
358498943Sluigi		if (qflag)
358598943Sluigi			args[i++] = "-q";
358698943Sluigi		for (a = strtok(buf, WHITESP);
358798943Sluigi		    a && i < MAX_ARGS; a = strtok(NULL, WHITESP), i++)
358898943Sluigi			args[i] = a;
358998943Sluigi		if (i == (qflag? 2: 1))
359098943Sluigi			continue;
359198943Sluigi		if (i == MAX_ARGS)
359298943Sluigi			errx(EX_USAGE, "%s: too many arguments",
359398943Sluigi			    linename);
359498943Sluigi		args[i] = NULL;
359598943Sluigi
359698943Sluigi		ipfw_main(i, args);
359798943Sluigi	}
359898943Sluigi	fclose(f);
359998943Sluigi	if (pflag) {
360098943Sluigi		if (waitpid(preproc, &status, 0) == -1)
360198943Sluigi			errx(EX_OSERR, "waitpid()");
360298943Sluigi		if (WIFEXITED(status) && WEXITSTATUS(status) != EX_OK)
360398943Sluigi			errx(EX_UNAVAILABLE,
360498943Sluigi			    "preprocessor exited with status %d",
360598943Sluigi			    WEXITSTATUS(status));
360698943Sluigi		else if (WIFSIGNALED(status))
360798943Sluigi			errx(EX_UNAVAILABLE,
360898943Sluigi			    "preprocessor exited with signal %d",
360998943Sluigi			    WTERMSIG(status));
361098943Sluigi	}
361198943Sluigi}
361298943Sluigi
361398943Sluigiint
361498943Sluigimain(int ac, char *av[])
361598943Sluigi{
361698943Sluigi	s = socket(AF_INET, SOCK_RAW, IPPROTO_RAW);
361798943Sluigi	if (s < 0)
361898943Sluigi		err(EX_UNAVAILABLE, "socket");
361998943Sluigi
362098943Sluigi	/*
362198943Sluigi	 * If the last argument is an absolute pathname, interpret it
362298943Sluigi	 * as a file to be preprocessed.
362398943Sluigi	 */
362498943Sluigi
362598943Sluigi	if (ac > 1 && av[ac - 1][0] == '/' && access(av[ac - 1], R_OK) == 0)
362698943Sluigi		ipfw_readfile(ac, av);
362798943Sluigi	else
362898943Sluigi		ipfw_main(ac, av);
362998943Sluigi	return EX_OK;
363098943Sluigi}
3631