pfctl.c revision 261882
123678Speter/*	$OpenBSD: pfctl.c,v 1.278 2008/08/31 20:18:17 jmc Exp $ */
250476Speter
31558Srgrimes/*
41558Srgrimes * Copyright (c) 2001 Daniel Hartmeier
5152738Srodrigc * Copyright (c) 2002,2003 Henning Brauer
6152344Srodrigc * All rights reserved.
774815Sru *
81558Srgrimes * Redistribution and use in source and binary forms, with or without
91558Srgrimes * modification, are permitted provided that the following conditions
10166439Spjd * are met:
11166439Spjd *
12166439Spjd *    - Redistributions of source code must retain the above copyright
131558Srgrimes *      notice, this list of conditions and the following disclaimer.
14 *    - Redistributions in binary form must reproduce the above
15 *      copyright notice, this list of conditions and the following
16 *      disclaimer in the documentation and/or other materials provided
17 *      with the distribution.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
22 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
23 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
24 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
25 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
26 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
27 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
29 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
30 * POSSIBILITY OF SUCH DAMAGE.
31 *
32 */
33
34#include <sys/cdefs.h>
35__FBSDID("$FreeBSD: head/sbin/pfctl/pfctl.c 261882 2014-02-14 10:05:21Z glebius $");
36
37#include <sys/types.h>
38#include <sys/ioctl.h>
39#include <sys/socket.h>
40#include <sys/stat.h>
41#include <sys/endian.h>
42
43#include <net/if.h>
44#include <netinet/in.h>
45#include <net/pfvar.h>
46#include <arpa/inet.h>
47#include <altq/altq.h>
48#include <sys/sysctl.h>
49
50#include <err.h>
51#include <errno.h>
52#include <fcntl.h>
53#include <limits.h>
54#include <netdb.h>
55#include <stdio.h>
56#include <stdlib.h>
57#include <string.h>
58#include <unistd.h>
59
60#include "pfctl_parser.h"
61#include "pfctl.h"
62
63void	 usage(void);
64int	 pfctl_enable(int, int);
65int	 pfctl_disable(int, int);
66int	 pfctl_clear_stats(int, int);
67int	 pfctl_clear_interface_flags(int, int);
68int	 pfctl_clear_rules(int, int, char *);
69int	 pfctl_clear_nat(int, int, char *);
70int	 pfctl_clear_altq(int, int);
71int	 pfctl_clear_src_nodes(int, int);
72int	 pfctl_clear_states(int, const char *, int);
73void	 pfctl_addrprefix(char *, struct pf_addr *);
74int	 pfctl_kill_src_nodes(int, const char *, int);
75int	 pfctl_net_kill_states(int, const char *, int);
76int	 pfctl_label_kill_states(int, const char *, int);
77int	 pfctl_id_kill_states(int, const char *, int);
78void	 pfctl_init_options(struct pfctl *);
79int	 pfctl_load_options(struct pfctl *);
80int	 pfctl_load_limit(struct pfctl *, unsigned int, unsigned int);
81int	 pfctl_load_timeout(struct pfctl *, unsigned int, unsigned int);
82int	 pfctl_load_debug(struct pfctl *, unsigned int);
83int	 pfctl_load_logif(struct pfctl *, char *);
84int	 pfctl_load_hostid(struct pfctl *, unsigned int);
85int	 pfctl_get_pool(int, struct pf_pool *, u_int32_t, u_int32_t, int,
86	    char *);
87void	 pfctl_print_rule_counters(struct pf_rule *, int);
88int	 pfctl_show_rules(int, char *, int, enum pfctl_show, char *, int);
89int	 pfctl_show_nat(int, int, char *);
90int	 pfctl_show_src_nodes(int, int);
91int	 pfctl_show_states(int, const char *, int);
92int	 pfctl_show_status(int, int);
93int	 pfctl_show_timeouts(int, int);
94int	 pfctl_show_limits(int, int);
95void	 pfctl_debug(int, u_int32_t, int);
96int	 pfctl_test_altqsupport(int, int);
97int	 pfctl_show_anchors(int, int, char *);
98int	 pfctl_ruleset_trans(struct pfctl *, char *, struct pf_anchor *);
99int	 pfctl_load_ruleset(struct pfctl *, char *,
100		struct pf_ruleset *, int, int);
101int	 pfctl_load_rule(struct pfctl *, char *, struct pf_rule *, int);
102const char	*pfctl_lookup_option(char *, const char **);
103
104struct pf_anchor_global	 pf_anchors;
105struct pf_anchor	 pf_main_anchor;
106
107const char	*clearopt;
108char		*rulesopt;
109const char	*showopt;
110const char	*debugopt;
111char		*anchoropt;
112const char	*optiopt = NULL;
113char		*pf_device = "/dev/pf";
114char		*ifaceopt;
115char		*tableopt;
116const char	*tblcmdopt;
117int		 src_node_killers;
118char		*src_node_kill[2];
119int		 state_killers;
120char		*state_kill[2];
121int		 loadopt;
122int		 altqsupport;
123
124int		 dev = -1;
125int		 first_title = 1;
126int		 labels = 0;
127
128#define INDENT(d, o)	do {						\
129				if (o) {				\
130					int i;				\
131					for (i=0; i < d; i++)		\
132						printf("  ");		\
133				}					\
134			} while (0);					\
135
136
137static const struct {
138	const char	*name;
139	int		index;
140} pf_limits[] = {
141	{ "states",		PF_LIMIT_STATES },
142	{ "src-nodes",		PF_LIMIT_SRC_NODES },
143	{ "frags",		PF_LIMIT_FRAGS },
144	{ "table-entries",	PF_LIMIT_TABLE_ENTRIES },
145	{ NULL,			0 }
146};
147
148struct pf_hint {
149	const char	*name;
150	int		timeout;
151};
152static const struct pf_hint pf_hint_normal[] = {
153	{ "tcp.first",		2 * 60 },
154	{ "tcp.opening",	30 },
155	{ "tcp.established",	24 * 60 * 60 },
156	{ "tcp.closing",	15 * 60 },
157	{ "tcp.finwait",	45 },
158	{ "tcp.closed",		90 },
159	{ "tcp.tsdiff",		30 },
160	{ NULL,			0 }
161};
162static const struct pf_hint pf_hint_satellite[] = {
163	{ "tcp.first",		3 * 60 },
164	{ "tcp.opening",	30 + 5 },
165	{ "tcp.established",	24 * 60 * 60 },
166	{ "tcp.closing",	15 * 60 + 5 },
167	{ "tcp.finwait",	45 + 5 },
168	{ "tcp.closed",		90 + 5 },
169	{ "tcp.tsdiff",		60 },
170	{ NULL,			0 }
171};
172static const struct pf_hint pf_hint_conservative[] = {
173	{ "tcp.first",		60 * 60 },
174	{ "tcp.opening",	15 * 60 },
175	{ "tcp.established",	5 * 24 * 60 * 60 },
176	{ "tcp.closing",	60 * 60 },
177	{ "tcp.finwait",	10 * 60 },
178	{ "tcp.closed",		3 * 60 },
179	{ "tcp.tsdiff",		60 },
180	{ NULL,			0 }
181};
182static const struct pf_hint pf_hint_aggressive[] = {
183	{ "tcp.first",		30 },
184	{ "tcp.opening",	5 },
185	{ "tcp.established",	5 * 60 * 60 },
186	{ "tcp.closing",	60 },
187	{ "tcp.finwait",	30 },
188	{ "tcp.closed",		30 },
189	{ "tcp.tsdiff",		10 },
190	{ NULL,			0 }
191};
192
193static const struct {
194	const char *name;
195	const struct pf_hint *hint;
196} pf_hints[] = {
197	{ "normal",		pf_hint_normal },
198	{ "satellite",		pf_hint_satellite },
199	{ "high-latency",	pf_hint_satellite },
200	{ "conservative",	pf_hint_conservative },
201	{ "aggressive",		pf_hint_aggressive },
202	{ NULL,			NULL }
203};
204
205static const char *clearopt_list[] = {
206	"nat", "queue", "rules", "Sources",
207	"states", "info", "Tables", "osfp", "all", NULL
208};
209
210static const char *showopt_list[] = {
211	"nat", "queue", "rules", "Anchors", "Sources", "states", "info",
212	"Interfaces", "labels", "timeouts", "memory", "Tables", "osfp",
213	"all", NULL
214};
215
216static const char *tblcmdopt_list[] = {
217	"kill", "flush", "add", "delete", "load", "replace", "show",
218	"test", "zero", "expire", NULL
219};
220
221static const char *debugopt_list[] = {
222	"none", "urgent", "misc", "loud", NULL
223};
224
225static const char *optiopt_list[] = {
226	"none", "basic", "profile", NULL
227};
228
229void
230usage(void)
231{
232	extern char *__progname;
233
234	fprintf(stderr,
235"usage: %s [-AdeghmNnOPqRrvz] [-a anchor] [-D macro=value] [-F modifier]\n"
236	"\t[-f file] [-i interface] [-K host | network]\n"
237	"\t[-k host | network | label | id] [-o level] [-p device]\n"
238	"\t[-s modifier] [-t table -T command [address ...]] [-x level]\n",
239	    __progname);
240
241	exit(1);
242}
243
244int
245pfctl_enable(int dev, int opts)
246{
247	if (ioctl(dev, DIOCSTART)) {
248		if (errno == EEXIST)
249			errx(1, "pf already enabled");
250		else if (errno == ESRCH)
251			errx(1, "pfil registeration failed");
252		else
253			err(1, "DIOCSTART");
254	}
255	if ((opts & PF_OPT_QUIET) == 0)
256		fprintf(stderr, "pf enabled\n");
257
258	if (altqsupport && ioctl(dev, DIOCSTARTALTQ))
259		if (errno != EEXIST)
260			err(1, "DIOCSTARTALTQ");
261
262	return (0);
263}
264
265int
266pfctl_disable(int dev, int opts)
267{
268	if (ioctl(dev, DIOCSTOP)) {
269		if (errno == ENOENT)
270			errx(1, "pf not enabled");
271		else
272			err(1, "DIOCSTOP");
273	}
274	if ((opts & PF_OPT_QUIET) == 0)
275		fprintf(stderr, "pf disabled\n");
276
277	if (altqsupport && ioctl(dev, DIOCSTOPALTQ))
278			if (errno != ENOENT)
279				err(1, "DIOCSTOPALTQ");
280
281	return (0);
282}
283
284int
285pfctl_clear_stats(int dev, int opts)
286{
287	if (ioctl(dev, DIOCCLRSTATUS))
288		err(1, "DIOCCLRSTATUS");
289	if ((opts & PF_OPT_QUIET) == 0)
290		fprintf(stderr, "pf: statistics cleared\n");
291	return (0);
292}
293
294int
295pfctl_clear_interface_flags(int dev, int opts)
296{
297	struct pfioc_iface	pi;
298
299	if ((opts & PF_OPT_NOACTION) == 0) {
300		bzero(&pi, sizeof(pi));
301		pi.pfiio_flags = PFI_IFLAG_SKIP;
302
303		if (ioctl(dev, DIOCCLRIFFLAG, &pi))
304			err(1, "DIOCCLRIFFLAG");
305		if ((opts & PF_OPT_QUIET) == 0)
306			fprintf(stderr, "pf: interface flags reset\n");
307	}
308	return (0);
309}
310
311int
312pfctl_clear_rules(int dev, int opts, char *anchorname)
313{
314	struct pfr_buffer t;
315
316	memset(&t, 0, sizeof(t));
317	t.pfrb_type = PFRB_TRANS;
318	if (pfctl_add_trans(&t, PF_RULESET_SCRUB, anchorname) ||
319	    pfctl_add_trans(&t, PF_RULESET_FILTER, anchorname) ||
320	    pfctl_trans(dev, &t, DIOCXBEGIN, 0) ||
321	    pfctl_trans(dev, &t, DIOCXCOMMIT, 0))
322		err(1, "pfctl_clear_rules");
323	if ((opts & PF_OPT_QUIET) == 0)
324		fprintf(stderr, "rules cleared\n");
325	return (0);
326}
327
328int
329pfctl_clear_nat(int dev, int opts, char *anchorname)
330{
331	struct pfr_buffer t;
332
333	memset(&t, 0, sizeof(t));
334	t.pfrb_type = PFRB_TRANS;
335	if (pfctl_add_trans(&t, PF_RULESET_NAT, anchorname) ||
336	    pfctl_add_trans(&t, PF_RULESET_BINAT, anchorname) ||
337	    pfctl_add_trans(&t, PF_RULESET_RDR, anchorname) ||
338	    pfctl_trans(dev, &t, DIOCXBEGIN, 0) ||
339	    pfctl_trans(dev, &t, DIOCXCOMMIT, 0))
340		err(1, "pfctl_clear_nat");
341	if ((opts & PF_OPT_QUIET) == 0)
342		fprintf(stderr, "nat cleared\n");
343	return (0);
344}
345
346int
347pfctl_clear_altq(int dev, int opts)
348{
349	struct pfr_buffer t;
350
351	if (!altqsupport)
352		return (-1);
353	memset(&t, 0, sizeof(t));
354	t.pfrb_type = PFRB_TRANS;
355	if (pfctl_add_trans(&t, PF_RULESET_ALTQ, "") ||
356	    pfctl_trans(dev, &t, DIOCXBEGIN, 0) ||
357	    pfctl_trans(dev, &t, DIOCXCOMMIT, 0))
358		err(1, "pfctl_clear_altq");
359	if ((opts & PF_OPT_QUIET) == 0)
360		fprintf(stderr, "altq cleared\n");
361	return (0);
362}
363
364int
365pfctl_clear_src_nodes(int dev, int opts)
366{
367	if (ioctl(dev, DIOCCLRSRCNODES))
368		err(1, "DIOCCLRSRCNODES");
369	if ((opts & PF_OPT_QUIET) == 0)
370		fprintf(stderr, "source tracking entries cleared\n");
371	return (0);
372}
373
374int
375pfctl_clear_states(int dev, const char *iface, int opts)
376{
377	struct pfioc_state_kill psk;
378
379	memset(&psk, 0, sizeof(psk));
380	if (iface != NULL && strlcpy(psk.psk_ifname, iface,
381	    sizeof(psk.psk_ifname)) >= sizeof(psk.psk_ifname))
382		errx(1, "invalid interface: %s", iface);
383
384	if (ioctl(dev, DIOCCLRSTATES, &psk))
385		err(1, "DIOCCLRSTATES");
386	if ((opts & PF_OPT_QUIET) == 0)
387		fprintf(stderr, "%d states cleared\n", psk.psk_killed);
388	return (0);
389}
390
391void
392pfctl_addrprefix(char *addr, struct pf_addr *mask)
393{
394	char *p;
395	const char *errstr;
396	int prefix, ret_ga, q, r;
397	struct addrinfo hints, *res;
398
399	if ((p = strchr(addr, '/')) == NULL)
400		return;
401
402	*p++ = '\0';
403	prefix = strtonum(p, 0, 128, &errstr);
404	if (errstr)
405		errx(1, "prefix is %s: %s", errstr, p);
406
407	bzero(&hints, sizeof(hints));
408	/* prefix only with numeric addresses */
409	hints.ai_flags |= AI_NUMERICHOST;
410
411	if ((ret_ga = getaddrinfo(addr, NULL, &hints, &res))) {
412		errx(1, "getaddrinfo: %s", gai_strerror(ret_ga));
413		/* NOTREACHED */
414	}
415
416	if (res->ai_family == AF_INET && prefix > 32)
417		errx(1, "prefix too long for AF_INET");
418	else if (res->ai_family == AF_INET6 && prefix > 128)
419		errx(1, "prefix too long for AF_INET6");
420
421	q = prefix >> 3;
422	r = prefix & 7;
423	switch (res->ai_family) {
424	case AF_INET:
425		bzero(&mask->v4, sizeof(mask->v4));
426		mask->v4.s_addr = htonl((u_int32_t)
427		    (0xffffffffffULL << (32 - prefix)));
428		break;
429	case AF_INET6:
430		bzero(&mask->v6, sizeof(mask->v6));
431		if (q > 0)
432			memset((void *)&mask->v6, 0xff, q);
433		if (r > 0)
434			*((u_char *)&mask->v6 + q) =
435			    (0xff00 >> r) & 0xff;
436		break;
437	}
438	freeaddrinfo(res);
439}
440
441int
442pfctl_kill_src_nodes(int dev, const char *iface, int opts)
443{
444	struct pfioc_src_node_kill psnk;
445	struct addrinfo *res[2], *resp[2];
446	struct sockaddr last_src, last_dst;
447	int killed, sources, dests;
448	int ret_ga;
449
450	killed = sources = dests = 0;
451
452	memset(&psnk, 0, sizeof(psnk));
453	memset(&psnk.psnk_src.addr.v.a.mask, 0xff,
454	    sizeof(psnk.psnk_src.addr.v.a.mask));
455	memset(&last_src, 0xff, sizeof(last_src));
456	memset(&last_dst, 0xff, sizeof(last_dst));
457
458	pfctl_addrprefix(src_node_kill[0], &psnk.psnk_src.addr.v.a.mask);
459
460	if ((ret_ga = getaddrinfo(src_node_kill[0], NULL, NULL, &res[0]))) {
461		errx(1, "getaddrinfo: %s", gai_strerror(ret_ga));
462		/* NOTREACHED */
463	}
464	for (resp[0] = res[0]; resp[0]; resp[0] = resp[0]->ai_next) {
465		if (resp[0]->ai_addr == NULL)
466			continue;
467		/* We get lots of duplicates.  Catch the easy ones */
468		if (memcmp(&last_src, resp[0]->ai_addr, sizeof(last_src)) == 0)
469			continue;
470		last_src = *(struct sockaddr *)resp[0]->ai_addr;
471
472		psnk.psnk_af = resp[0]->ai_family;
473		sources++;
474
475		if (psnk.psnk_af == AF_INET)
476			psnk.psnk_src.addr.v.a.addr.v4 =
477			    ((struct sockaddr_in *)resp[0]->ai_addr)->sin_addr;
478		else if (psnk.psnk_af == AF_INET6)
479			psnk.psnk_src.addr.v.a.addr.v6 =
480			    ((struct sockaddr_in6 *)resp[0]->ai_addr)->
481			    sin6_addr;
482		else
483			errx(1, "Unknown address family %d", psnk.psnk_af);
484
485		if (src_node_killers > 1) {
486			dests = 0;
487			memset(&psnk.psnk_dst.addr.v.a.mask, 0xff,
488			    sizeof(psnk.psnk_dst.addr.v.a.mask));
489			memset(&last_dst, 0xff, sizeof(last_dst));
490			pfctl_addrprefix(src_node_kill[1],
491			    &psnk.psnk_dst.addr.v.a.mask);
492			if ((ret_ga = getaddrinfo(src_node_kill[1], NULL, NULL,
493			    &res[1]))) {
494				errx(1, "getaddrinfo: %s",
495				    gai_strerror(ret_ga));
496				/* NOTREACHED */
497			}
498			for (resp[1] = res[1]; resp[1];
499			    resp[1] = resp[1]->ai_next) {
500				if (resp[1]->ai_addr == NULL)
501					continue;
502				if (psnk.psnk_af != resp[1]->ai_family)
503					continue;
504
505				if (memcmp(&last_dst, resp[1]->ai_addr,
506				    sizeof(last_dst)) == 0)
507					continue;
508				last_dst = *(struct sockaddr *)resp[1]->ai_addr;
509
510				dests++;
511
512				if (psnk.psnk_af == AF_INET)
513					psnk.psnk_dst.addr.v.a.addr.v4 =
514					    ((struct sockaddr_in *)resp[1]->
515					    ai_addr)->sin_addr;
516				else if (psnk.psnk_af == AF_INET6)
517					psnk.psnk_dst.addr.v.a.addr.v6 =
518					    ((struct sockaddr_in6 *)resp[1]->
519					    ai_addr)->sin6_addr;
520				else
521					errx(1, "Unknown address family %d",
522					    psnk.psnk_af);
523
524				if (ioctl(dev, DIOCKILLSRCNODES, &psnk))
525					err(1, "DIOCKILLSRCNODES");
526				killed += psnk.psnk_killed;
527			}
528			freeaddrinfo(res[1]);
529		} else {
530			if (ioctl(dev, DIOCKILLSRCNODES, &psnk))
531				err(1, "DIOCKILLSRCNODES");
532			killed += psnk.psnk_killed;
533		}
534	}
535
536	freeaddrinfo(res[0]);
537
538	if ((opts & PF_OPT_QUIET) == 0)
539		fprintf(stderr, "killed %d src nodes from %d sources and %d "
540		    "destinations\n", killed, sources, dests);
541	return (0);
542}
543
544int
545pfctl_net_kill_states(int dev, const char *iface, int opts)
546{
547	struct pfioc_state_kill psk;
548	struct addrinfo *res[2], *resp[2];
549	struct sockaddr last_src, last_dst;
550	int killed, sources, dests;
551	int ret_ga;
552
553	killed = sources = dests = 0;
554
555	memset(&psk, 0, sizeof(psk));
556	memset(&psk.psk_src.addr.v.a.mask, 0xff,
557	    sizeof(psk.psk_src.addr.v.a.mask));
558	memset(&last_src, 0xff, sizeof(last_src));
559	memset(&last_dst, 0xff, sizeof(last_dst));
560	if (iface != NULL && strlcpy(psk.psk_ifname, iface,
561	    sizeof(psk.psk_ifname)) >= sizeof(psk.psk_ifname))
562		errx(1, "invalid interface: %s", iface);
563
564	pfctl_addrprefix(state_kill[0], &psk.psk_src.addr.v.a.mask);
565
566	if ((ret_ga = getaddrinfo(state_kill[0], NULL, NULL, &res[0]))) {
567		errx(1, "getaddrinfo: %s", gai_strerror(ret_ga));
568		/* NOTREACHED */
569	}
570	for (resp[0] = res[0]; resp[0]; resp[0] = resp[0]->ai_next) {
571		if (resp[0]->ai_addr == NULL)
572			continue;
573		/* We get lots of duplicates.  Catch the easy ones */
574		if (memcmp(&last_src, resp[0]->ai_addr, sizeof(last_src)) == 0)
575			continue;
576		last_src = *(struct sockaddr *)resp[0]->ai_addr;
577
578		psk.psk_af = resp[0]->ai_family;
579		sources++;
580
581		if (psk.psk_af == AF_INET)
582			psk.psk_src.addr.v.a.addr.v4 =
583			    ((struct sockaddr_in *)resp[0]->ai_addr)->sin_addr;
584		else if (psk.psk_af == AF_INET6)
585			psk.psk_src.addr.v.a.addr.v6 =
586			    ((struct sockaddr_in6 *)resp[0]->ai_addr)->
587			    sin6_addr;
588		else
589			errx(1, "Unknown address family %d", psk.psk_af);
590
591		if (state_killers > 1) {
592			dests = 0;
593			memset(&psk.psk_dst.addr.v.a.mask, 0xff,
594			    sizeof(psk.psk_dst.addr.v.a.mask));
595			memset(&last_dst, 0xff, sizeof(last_dst));
596			pfctl_addrprefix(state_kill[1],
597			    &psk.psk_dst.addr.v.a.mask);
598			if ((ret_ga = getaddrinfo(state_kill[1], NULL, NULL,
599			    &res[1]))) {
600				errx(1, "getaddrinfo: %s",
601				    gai_strerror(ret_ga));
602				/* NOTREACHED */
603			}
604			for (resp[1] = res[1]; resp[1];
605			    resp[1] = resp[1]->ai_next) {
606				if (resp[1]->ai_addr == NULL)
607					continue;
608				if (psk.psk_af != resp[1]->ai_family)
609					continue;
610
611				if (memcmp(&last_dst, resp[1]->ai_addr,
612				    sizeof(last_dst)) == 0)
613					continue;
614				last_dst = *(struct sockaddr *)resp[1]->ai_addr;
615
616				dests++;
617
618				if (psk.psk_af == AF_INET)
619					psk.psk_dst.addr.v.a.addr.v4 =
620					    ((struct sockaddr_in *)resp[1]->
621					    ai_addr)->sin_addr;
622				else if (psk.psk_af == AF_INET6)
623					psk.psk_dst.addr.v.a.addr.v6 =
624					    ((struct sockaddr_in6 *)resp[1]->
625					    ai_addr)->sin6_addr;
626				else
627					errx(1, "Unknown address family %d",
628					    psk.psk_af);
629
630				if (ioctl(dev, DIOCKILLSTATES, &psk))
631					err(1, "DIOCKILLSTATES");
632				killed += psk.psk_killed;
633			}
634			freeaddrinfo(res[1]);
635		} else {
636			if (ioctl(dev, DIOCKILLSTATES, &psk))
637				err(1, "DIOCKILLSTATES");
638			killed += psk.psk_killed;
639		}
640	}
641
642	freeaddrinfo(res[0]);
643
644	if ((opts & PF_OPT_QUIET) == 0)
645		fprintf(stderr, "killed %d states from %d sources and %d "
646		    "destinations\n", killed, sources, dests);
647	return (0);
648}
649
650int
651pfctl_label_kill_states(int dev, const char *iface, int opts)
652{
653	struct pfioc_state_kill psk;
654
655	if (state_killers != 2 || (strlen(state_kill[1]) == 0)) {
656		warnx("no label specified");
657		usage();
658	}
659	memset(&psk, 0, sizeof(psk));
660	if (iface != NULL && strlcpy(psk.psk_ifname, iface,
661	    sizeof(psk.psk_ifname)) >= sizeof(psk.psk_ifname))
662		errx(1, "invalid interface: %s", iface);
663
664	if (strlcpy(psk.psk_label, state_kill[1], sizeof(psk.psk_label)) >=
665	    sizeof(psk.psk_label))
666		errx(1, "label too long: %s", state_kill[1]);
667
668	if (ioctl(dev, DIOCKILLSTATES, &psk))
669		err(1, "DIOCKILLSTATES");
670
671	if ((opts & PF_OPT_QUIET) == 0)
672		fprintf(stderr, "killed %d states\n", psk.psk_killed);
673
674	return (0);
675}
676
677int
678pfctl_id_kill_states(int dev, const char *iface, int opts)
679{
680	struct pfioc_state_kill psk;
681
682	if (state_killers != 2 || (strlen(state_kill[1]) == 0)) {
683		warnx("no id specified");
684		usage();
685	}
686
687	memset(&psk, 0, sizeof(psk));
688	if ((sscanf(state_kill[1], "%jx/%x",
689	    &psk.psk_pfcmp.id, &psk.psk_pfcmp.creatorid)) == 2)
690		HTONL(psk.psk_pfcmp.creatorid);
691	else if ((sscanf(state_kill[1], "%jx", &psk.psk_pfcmp.id)) == 1) {
692		psk.psk_pfcmp.creatorid = 0;
693	} else {
694		warnx("wrong id format specified");
695		usage();
696	}
697	if (psk.psk_pfcmp.id == 0) {
698		warnx("cannot kill id 0");
699		usage();
700	}
701
702	psk.psk_pfcmp.id = htobe64(psk.psk_pfcmp.id);
703	if (ioctl(dev, DIOCKILLSTATES, &psk))
704		err(1, "DIOCKILLSTATES");
705
706	if ((opts & PF_OPT_QUIET) == 0)
707		fprintf(stderr, "killed %d states\n", psk.psk_killed);
708
709	return (0);
710}
711
712int
713pfctl_get_pool(int dev, struct pf_pool *pool, u_int32_t nr,
714    u_int32_t ticket, int r_action, char *anchorname)
715{
716	struct pfioc_pooladdr pp;
717	struct pf_pooladdr *pa;
718	u_int32_t pnr, mpnr;
719
720	memset(&pp, 0, sizeof(pp));
721	memcpy(pp.anchor, anchorname, sizeof(pp.anchor));
722	pp.r_action = r_action;
723	pp.r_num = nr;
724	pp.ticket = ticket;
725	if (ioctl(dev, DIOCGETADDRS, &pp)) {
726		warn("DIOCGETADDRS");
727		return (-1);
728	}
729	mpnr = pp.nr;
730	TAILQ_INIT(&pool->list);
731	for (pnr = 0; pnr < mpnr; ++pnr) {
732		pp.nr = pnr;
733		if (ioctl(dev, DIOCGETADDR, &pp)) {
734			warn("DIOCGETADDR");
735			return (-1);
736		}
737		pa = calloc(1, sizeof(struct pf_pooladdr));
738		if (pa == NULL)
739			err(1, "calloc");
740		bcopy(&pp.addr, pa, sizeof(struct pf_pooladdr));
741		TAILQ_INSERT_TAIL(&pool->list, pa, entries);
742	}
743
744	return (0);
745}
746
747void
748pfctl_move_pool(struct pf_pool *src, struct pf_pool *dst)
749{
750	struct pf_pooladdr *pa;
751
752	while ((pa = TAILQ_FIRST(&src->list)) != NULL) {
753		TAILQ_REMOVE(&src->list, pa, entries);
754		TAILQ_INSERT_TAIL(&dst->list, pa, entries);
755	}
756}
757
758void
759pfctl_clear_pool(struct pf_pool *pool)
760{
761	struct pf_pooladdr *pa;
762
763	while ((pa = TAILQ_FIRST(&pool->list)) != NULL) {
764		TAILQ_REMOVE(&pool->list, pa, entries);
765		free(pa);
766	}
767}
768
769void
770pfctl_print_rule_counters(struct pf_rule *rule, int opts)
771{
772	if (opts & PF_OPT_DEBUG) {
773		const char *t[PF_SKIP_COUNT] = { "i", "d", "f",
774		    "p", "sa", "sp", "da", "dp" };
775		int i;
776
777		printf("  [ Skip steps: ");
778		for (i = 0; i < PF_SKIP_COUNT; ++i) {
779			if (rule->skip[i].nr == rule->nr + 1)
780				continue;
781			printf("%s=", t[i]);
782			if (rule->skip[i].nr == -1)
783				printf("end ");
784			else
785				printf("%u ", rule->skip[i].nr);
786		}
787		printf("]\n");
788
789		printf("  [ queue: qname=%s qid=%u pqname=%s pqid=%u ]\n",
790		    rule->qname, rule->qid, rule->pqname, rule->pqid);
791	}
792	if (opts & PF_OPT_VERBOSE) {
793		printf("  [ Evaluations: %-8llu  Packets: %-8llu  "
794			    "Bytes: %-10llu  States: %-6lu]\n",
795			    (unsigned long long)rule->evaluations,
796			    (unsigned long long)(rule->packets[0] +
797			    rule->packets[1]),
798			    (unsigned long long)(rule->bytes[0] +
799			    rule->bytes[1]), (uint64_t)rule->states_cur);
800		if (!(opts & PF_OPT_DEBUG))
801			printf("  [ Inserted: uid %u pid %u "
802			    "State Creations: %-6lu]\n",
803			    (unsigned)rule->cuid, (unsigned)rule->cpid,
804			    (uint64_t)rule->states_tot);
805	}
806}
807
808void
809pfctl_print_title(char *title)
810{
811	if (!first_title)
812		printf("\n");
813	first_title = 0;
814	printf("%s\n", title);
815}
816
817int
818pfctl_show_rules(int dev, char *path, int opts, enum pfctl_show format,
819    char *anchorname, int depth)
820{
821	struct pfioc_rule pr;
822	u_int32_t nr, mnr, header = 0;
823	int rule_numbers = opts & (PF_OPT_VERBOSE2 | PF_OPT_DEBUG);
824	int numeric = opts & PF_OPT_NUMERIC;
825	int len = strlen(path);
826	int brace;
827	char *p;
828
829	if (path[0])
830		snprintf(&path[len], MAXPATHLEN - len, "/%s", anchorname);
831	else
832		snprintf(&path[len], MAXPATHLEN - len, "%s", anchorname);
833
834	memset(&pr, 0, sizeof(pr));
835	memcpy(pr.anchor, path, sizeof(pr.anchor));
836	if (opts & PF_OPT_SHOWALL) {
837		pr.rule.action = PF_PASS;
838		if (ioctl(dev, DIOCGETRULES, &pr)) {
839			warn("DIOCGETRULES");
840			goto error;
841		}
842		header++;
843	}
844	pr.rule.action = PF_SCRUB;
845	if (ioctl(dev, DIOCGETRULES, &pr)) {
846		warn("DIOCGETRULES");
847		goto error;
848	}
849	if (opts & PF_OPT_SHOWALL) {
850		if (format == PFCTL_SHOW_RULES && (pr.nr > 0 || header))
851			pfctl_print_title("FILTER RULES:");
852		else if (format == PFCTL_SHOW_LABELS && labels)
853			pfctl_print_title("LABEL COUNTERS:");
854	}
855	mnr = pr.nr;
856	if (opts & PF_OPT_CLRRULECTRS)
857		pr.action = PF_GET_CLR_CNTR;
858
859	for (nr = 0; nr < mnr; ++nr) {
860		pr.nr = nr;
861		if (ioctl(dev, DIOCGETRULE, &pr)) {
862			warn("DIOCGETRULE");
863			goto error;
864		}
865
866		if (pfctl_get_pool(dev, &pr.rule.rpool,
867		    nr, pr.ticket, PF_SCRUB, path) != 0)
868			goto error;
869
870		switch (format) {
871		case PFCTL_SHOW_LABELS:
872			break;
873		case PFCTL_SHOW_RULES:
874			if (pr.rule.label[0] && (opts & PF_OPT_SHOWALL))
875				labels = 1;
876			print_rule(&pr.rule, pr.anchor_call, rule_numbers, numeric);
877			printf("\n");
878			pfctl_print_rule_counters(&pr.rule, opts);
879			break;
880		case PFCTL_SHOW_NOTHING:
881			break;
882		}
883		pfctl_clear_pool(&pr.rule.rpool);
884	}
885	pr.rule.action = PF_PASS;
886	if (ioctl(dev, DIOCGETRULES, &pr)) {
887		warn("DIOCGETRULES");
888		goto error;
889	}
890	mnr = pr.nr;
891	for (nr = 0; nr < mnr; ++nr) {
892		pr.nr = nr;
893		if (ioctl(dev, DIOCGETRULE, &pr)) {
894			warn("DIOCGETRULE");
895			goto error;
896		}
897
898		if (pfctl_get_pool(dev, &pr.rule.rpool,
899		    nr, pr.ticket, PF_PASS, path) != 0)
900			goto error;
901
902		switch (format) {
903		case PFCTL_SHOW_LABELS:
904			if (pr.rule.label[0]) {
905				printf("%s %llu %llu %llu %llu"
906				    " %llu %llu %llu %llu\n",
907				    pr.rule.label,
908				    (unsigned long long)pr.rule.evaluations,
909				    (unsigned long long)(pr.rule.packets[0] +
910				    pr.rule.packets[1]),
911				    (unsigned long long)(pr.rule.bytes[0] +
912				    pr.rule.bytes[1]),
913				    (unsigned long long)pr.rule.packets[0],
914				    (unsigned long long)pr.rule.bytes[0],
915				    (unsigned long long)pr.rule.packets[1],
916				    (unsigned long long)pr.rule.bytes[1],
917				    (unsigned long long)pr.rule.states_tot);
918			}
919			break;
920		case PFCTL_SHOW_RULES:
921			brace = 0;
922			if (pr.rule.label[0] && (opts & PF_OPT_SHOWALL))
923				labels = 1;
924			INDENT(depth, !(opts & PF_OPT_VERBOSE));
925			if (pr.anchor_call[0] &&
926			   ((((p = strrchr(pr.anchor_call, '_')) != NULL) &&
927			   ((void *)p == (void *)pr.anchor_call ||
928			   *(--p) == '/')) || (opts & PF_OPT_RECURSE))) {
929				brace++;
930				if ((p = strrchr(pr.anchor_call, '/')) !=
931				    NULL)
932					p++;
933				else
934					p = &pr.anchor_call[0];
935			} else
936				p = &pr.anchor_call[0];
937
938			print_rule(&pr.rule, p, rule_numbers, numeric);
939			if (brace)
940				printf(" {\n");
941			else
942				printf("\n");
943			pfctl_print_rule_counters(&pr.rule, opts);
944			if (brace) {
945				pfctl_show_rules(dev, path, opts, format,
946				    p, depth + 1);
947				INDENT(depth, !(opts & PF_OPT_VERBOSE));
948				printf("}\n");
949			}
950			break;
951		case PFCTL_SHOW_NOTHING:
952			break;
953		}
954		pfctl_clear_pool(&pr.rule.rpool);
955	}
956	path[len] = '\0';
957	return (0);
958
959 error:
960	path[len] = '\0';
961	return (-1);
962}
963
964int
965pfctl_show_nat(int dev, int opts, char *anchorname)
966{
967	struct pfioc_rule pr;
968	u_int32_t mnr, nr;
969	static int nattype[3] = { PF_NAT, PF_RDR, PF_BINAT };
970	int i, dotitle = opts & PF_OPT_SHOWALL;
971
972	memset(&pr, 0, sizeof(pr));
973	memcpy(pr.anchor, anchorname, sizeof(pr.anchor));
974	for (i = 0; i < 3; i++) {
975		pr.rule.action = nattype[i];
976		if (ioctl(dev, DIOCGETRULES, &pr)) {
977			warn("DIOCGETRULES");
978			return (-1);
979		}
980		mnr = pr.nr;
981		for (nr = 0; nr < mnr; ++nr) {
982			pr.nr = nr;
983			if (ioctl(dev, DIOCGETRULE, &pr)) {
984				warn("DIOCGETRULE");
985				return (-1);
986			}
987			if (pfctl_get_pool(dev, &pr.rule.rpool, nr,
988			    pr.ticket, nattype[i], anchorname) != 0)
989				return (-1);
990			if (dotitle) {
991				pfctl_print_title("TRANSLATION RULES:");
992				dotitle = 0;
993			}
994			print_rule(&pr.rule, pr.anchor_call,
995			    opts & PF_OPT_VERBOSE2, opts & PF_OPT_NUMERIC);
996			printf("\n");
997			pfctl_print_rule_counters(&pr.rule, opts);
998			pfctl_clear_pool(&pr.rule.rpool);
999		}
1000	}
1001	return (0);
1002}
1003
1004int
1005pfctl_show_src_nodes(int dev, int opts)
1006{
1007	struct pfioc_src_nodes psn;
1008	struct pf_src_node *p;
1009	char *inbuf = NULL, *newinbuf = NULL;
1010	unsigned int len = 0;
1011	int i;
1012
1013	memset(&psn, 0, sizeof(psn));
1014	for (;;) {
1015		psn.psn_len = len;
1016		if (len) {
1017			newinbuf = realloc(inbuf, len);
1018			if (newinbuf == NULL)
1019				err(1, "realloc");
1020			psn.psn_buf = inbuf = newinbuf;
1021		}
1022		if (ioctl(dev, DIOCGETSRCNODES, &psn) < 0) {
1023			warn("DIOCGETSRCNODES");
1024			free(inbuf);
1025			return (-1);
1026		}
1027		if (psn.psn_len + sizeof(struct pfioc_src_nodes) < len)
1028			break;
1029		if (len == 0 && psn.psn_len == 0)
1030			goto done;
1031		if (len == 0 && psn.psn_len != 0)
1032			len = psn.psn_len;
1033		if (psn.psn_len == 0)
1034			goto done;	/* no src_nodes */
1035		len *= 2;
1036	}
1037	p = psn.psn_src_nodes;
1038	if (psn.psn_len > 0 && (opts & PF_OPT_SHOWALL))
1039		pfctl_print_title("SOURCE TRACKING NODES:");
1040	for (i = 0; i < psn.psn_len; i += sizeof(*p)) {
1041		print_src_node(p, opts);
1042		p++;
1043	}
1044done:
1045	free(inbuf);
1046	return (0);
1047}
1048
1049int
1050pfctl_show_states(int dev, const char *iface, int opts)
1051{
1052	struct pfioc_states ps;
1053	struct pfsync_state *p;
1054	char *inbuf = NULL, *newinbuf = NULL;
1055	unsigned int len = 0;
1056	int i, dotitle = (opts & PF_OPT_SHOWALL);
1057
1058	memset(&ps, 0, sizeof(ps));
1059	for (;;) {
1060		ps.ps_len = len;
1061		if (len) {
1062			newinbuf = realloc(inbuf, len);
1063			if (newinbuf == NULL)
1064				err(1, "realloc");
1065			ps.ps_buf = inbuf = newinbuf;
1066		}
1067		if (ioctl(dev, DIOCGETSTATES, &ps) < 0) {
1068			warn("DIOCGETSTATES");
1069			free(inbuf);
1070			return (-1);
1071		}
1072		if (ps.ps_len + sizeof(struct pfioc_states) < len)
1073			break;
1074		if (len == 0 && ps.ps_len == 0)
1075			goto done;
1076		if (len == 0 && ps.ps_len != 0)
1077			len = ps.ps_len;
1078		if (ps.ps_len == 0)
1079			goto done;	/* no states */
1080		len *= 2;
1081	}
1082	p = ps.ps_states;
1083	for (i = 0; i < ps.ps_len; i += sizeof(*p), p++) {
1084		if (iface != NULL && strcmp(p->ifname, iface))
1085			continue;
1086		if (dotitle) {
1087			pfctl_print_title("STATES:");
1088			dotitle = 0;
1089		}
1090		print_state(p, opts);
1091	}
1092done:
1093	free(inbuf);
1094	return (0);
1095}
1096
1097int
1098pfctl_show_status(int dev, int opts)
1099{
1100	struct pf_status status;
1101
1102	if (ioctl(dev, DIOCGETSTATUS, &status)) {
1103		warn("DIOCGETSTATUS");
1104		return (-1);
1105	}
1106	if (opts & PF_OPT_SHOWALL)
1107		pfctl_print_title("INFO:");
1108	print_status(&status, opts);
1109	return (0);
1110}
1111
1112int
1113pfctl_show_timeouts(int dev, int opts)
1114{
1115	struct pfioc_tm pt;
1116	int i;
1117
1118	if (opts & PF_OPT_SHOWALL)
1119		pfctl_print_title("TIMEOUTS:");
1120	memset(&pt, 0, sizeof(pt));
1121	for (i = 0; pf_timeouts[i].name; i++) {
1122		pt.timeout = pf_timeouts[i].timeout;
1123		if (ioctl(dev, DIOCGETTIMEOUT, &pt))
1124			err(1, "DIOCGETTIMEOUT");
1125		printf("%-20s %10d", pf_timeouts[i].name, pt.seconds);
1126		if (pf_timeouts[i].timeout >= PFTM_ADAPTIVE_START &&
1127		    pf_timeouts[i].timeout <= PFTM_ADAPTIVE_END)
1128			printf(" states");
1129		else
1130			printf("s");
1131		printf("\n");
1132	}
1133	return (0);
1134
1135}
1136
1137int
1138pfctl_show_limits(int dev, int opts)
1139{
1140	struct pfioc_limit pl;
1141	int i;
1142
1143	if (opts & PF_OPT_SHOWALL)
1144		pfctl_print_title("LIMITS:");
1145	memset(&pl, 0, sizeof(pl));
1146	for (i = 0; pf_limits[i].name; i++) {
1147		pl.index = pf_limits[i].index;
1148		if (ioctl(dev, DIOCGETLIMIT, &pl))
1149			err(1, "DIOCGETLIMIT");
1150		printf("%-13s ", pf_limits[i].name);
1151		if (pl.limit == UINT_MAX)
1152			printf("unlimited\n");
1153		else
1154			printf("hard limit %8u\n", pl.limit);
1155	}
1156	return (0);
1157}
1158
1159/* callbacks for rule/nat/rdr/addr */
1160int
1161pfctl_add_pool(struct pfctl *pf, struct pf_pool *p, sa_family_t af)
1162{
1163	struct pf_pooladdr *pa;
1164
1165	if ((pf->opts & PF_OPT_NOACTION) == 0) {
1166		if (ioctl(pf->dev, DIOCBEGINADDRS, &pf->paddr))
1167			err(1, "DIOCBEGINADDRS");
1168	}
1169
1170	pf->paddr.af = af;
1171	TAILQ_FOREACH(pa, &p->list, entries) {
1172		memcpy(&pf->paddr.addr, pa, sizeof(struct pf_pooladdr));
1173		if ((pf->opts & PF_OPT_NOACTION) == 0) {
1174			if (ioctl(pf->dev, DIOCADDADDR, &pf->paddr))
1175				err(1, "DIOCADDADDR");
1176		}
1177	}
1178	return (0);
1179}
1180
1181int
1182pfctl_add_rule(struct pfctl *pf, struct pf_rule *r, const char *anchor_call)
1183{
1184	u_int8_t		rs_num;
1185	struct pf_rule		*rule;
1186	struct pf_ruleset	*rs;
1187	char 			*p;
1188
1189	rs_num = pf_get_ruleset_number(r->action);
1190	if (rs_num == PF_RULESET_MAX)
1191		errx(1, "Invalid rule type %d", r->action);
1192
1193	rs = &pf->anchor->ruleset;
1194
1195	if (anchor_call[0] && r->anchor == NULL) {
1196		/*
1197		 * Don't make non-brace anchors part of the main anchor pool.
1198		 */
1199		if ((r->anchor = calloc(1, sizeof(*r->anchor))) == NULL)
1200			err(1, "pfctl_add_rule: calloc");
1201
1202		pf_init_ruleset(&r->anchor->ruleset);
1203		r->anchor->ruleset.anchor = r->anchor;
1204		if (strlcpy(r->anchor->path, anchor_call,
1205		    sizeof(rule->anchor->path)) >= sizeof(rule->anchor->path))
1206			errx(1, "pfctl_add_rule: strlcpy");
1207		if ((p = strrchr(anchor_call, '/')) != NULL) {
1208			if (!strlen(p))
1209				err(1, "pfctl_add_rule: bad anchor name %s",
1210				    anchor_call);
1211		} else
1212			p = (char *)anchor_call;
1213		if (strlcpy(r->anchor->name, p,
1214		    sizeof(rule->anchor->name)) >= sizeof(rule->anchor->name))
1215			errx(1, "pfctl_add_rule: strlcpy");
1216	}
1217
1218	if ((rule = calloc(1, sizeof(*rule))) == NULL)
1219		err(1, "calloc");
1220	bcopy(r, rule, sizeof(*rule));
1221	TAILQ_INIT(&rule->rpool.list);
1222	pfctl_move_pool(&r->rpool, &rule->rpool);
1223
1224	TAILQ_INSERT_TAIL(rs->rules[rs_num].active.ptr, rule, entries);
1225	return (0);
1226}
1227
1228int
1229pfctl_ruleset_trans(struct pfctl *pf, char *path, struct pf_anchor *a)
1230{
1231	int osize = pf->trans->pfrb_size;
1232
1233	if ((pf->loadopt & PFCTL_FLAG_NAT) != 0) {
1234		if (pfctl_add_trans(pf->trans, PF_RULESET_NAT, path) ||
1235		    pfctl_add_trans(pf->trans, PF_RULESET_BINAT, path) ||
1236		    pfctl_add_trans(pf->trans, PF_RULESET_RDR, path))
1237			return (1);
1238	}
1239	if (a == pf->astack[0] && ((altqsupport &&
1240	    (pf->loadopt & PFCTL_FLAG_ALTQ) != 0))) {
1241		if (pfctl_add_trans(pf->trans, PF_RULESET_ALTQ, path))
1242			return (2);
1243	}
1244	if ((pf->loadopt & PFCTL_FLAG_FILTER) != 0) {
1245		if (pfctl_add_trans(pf->trans, PF_RULESET_SCRUB, path) ||
1246		    pfctl_add_trans(pf->trans, PF_RULESET_FILTER, path))
1247			return (3);
1248	}
1249	if (pf->loadopt & PFCTL_FLAG_TABLE)
1250		if (pfctl_add_trans(pf->trans, PF_RULESET_TABLE, path))
1251			return (4);
1252	if (pfctl_trans(pf->dev, pf->trans, DIOCXBEGIN, osize))
1253		return (5);
1254
1255	return (0);
1256}
1257
1258int
1259pfctl_load_ruleset(struct pfctl *pf, char *path, struct pf_ruleset *rs,
1260    int rs_num, int depth)
1261{
1262	struct pf_rule *r;
1263	int		error, len = strlen(path);
1264	int		brace = 0;
1265
1266	pf->anchor = rs->anchor;
1267
1268	if (path[0])
1269		snprintf(&path[len], MAXPATHLEN - len, "/%s", pf->anchor->name);
1270	else
1271		snprintf(&path[len], MAXPATHLEN - len, "%s", pf->anchor->name);
1272
1273	if (depth) {
1274		if (TAILQ_FIRST(rs->rules[rs_num].active.ptr) != NULL) {
1275			brace++;
1276			if (pf->opts & PF_OPT_VERBOSE)
1277				printf(" {\n");
1278			if ((pf->opts & PF_OPT_NOACTION) == 0 &&
1279			    (error = pfctl_ruleset_trans(pf,
1280			    path, rs->anchor))) {
1281				printf("pfctl_load_rulesets: "
1282				    "pfctl_ruleset_trans %d\n", error);
1283				goto error;
1284			}
1285		} else if (pf->opts & PF_OPT_VERBOSE)
1286			printf("\n");
1287
1288	}
1289
1290	if (pf->optimize && rs_num == PF_RULESET_FILTER)
1291		pfctl_optimize_ruleset(pf, rs);
1292
1293	while ((r = TAILQ_FIRST(rs->rules[rs_num].active.ptr)) != NULL) {
1294		TAILQ_REMOVE(rs->rules[rs_num].active.ptr, r, entries);
1295		if ((error = pfctl_load_rule(pf, path, r, depth)))
1296			goto error;
1297		if (r->anchor) {
1298			if ((error = pfctl_load_ruleset(pf, path,
1299			    &r->anchor->ruleset, rs_num, depth + 1)))
1300				goto error;
1301		} else if (pf->opts & PF_OPT_VERBOSE)
1302			printf("\n");
1303		free(r);
1304	}
1305	if (brace && pf->opts & PF_OPT_VERBOSE) {
1306		INDENT(depth - 1, (pf->opts & PF_OPT_VERBOSE));
1307		printf("}\n");
1308	}
1309	path[len] = '\0';
1310	return (0);
1311
1312 error:
1313	path[len] = '\0';
1314	return (error);
1315
1316}
1317
1318int
1319pfctl_load_rule(struct pfctl *pf, char *path, struct pf_rule *r, int depth)
1320{
1321	u_int8_t		rs_num = pf_get_ruleset_number(r->action);
1322	char			*name;
1323	struct pfioc_rule	pr;
1324	int			len = strlen(path);
1325
1326	bzero(&pr, sizeof(pr));
1327	/* set up anchor before adding to path for anchor_call */
1328	if ((pf->opts & PF_OPT_NOACTION) == 0)
1329		pr.ticket = pfctl_get_ticket(pf->trans, rs_num, path);
1330	if (strlcpy(pr.anchor, path, sizeof(pr.anchor)) >= sizeof(pr.anchor))
1331		errx(1, "pfctl_load_rule: strlcpy");
1332
1333	if (r->anchor) {
1334		if (r->anchor->match) {
1335			if (path[0])
1336				snprintf(&path[len], MAXPATHLEN - len,
1337				    "/%s", r->anchor->name);
1338			else
1339				snprintf(&path[len], MAXPATHLEN - len,
1340				    "%s", r->anchor->name);
1341			name = path;
1342		} else
1343			name = r->anchor->path;
1344	} else
1345		name = "";
1346
1347	if ((pf->opts & PF_OPT_NOACTION) == 0) {
1348		if (pfctl_add_pool(pf, &r->rpool, r->af))
1349			return (1);
1350		pr.pool_ticket = pf->paddr.ticket;
1351		memcpy(&pr.rule, r, sizeof(pr.rule));
1352		if (r->anchor && strlcpy(pr.anchor_call, name,
1353		    sizeof(pr.anchor_call)) >= sizeof(pr.anchor_call))
1354			errx(1, "pfctl_load_rule: strlcpy");
1355		if (ioctl(pf->dev, DIOCADDRULE, &pr))
1356			err(1, "DIOCADDRULE");
1357	}
1358
1359	if (pf->opts & PF_OPT_VERBOSE) {
1360		INDENT(depth, !(pf->opts & PF_OPT_VERBOSE2));
1361		print_rule(r, r->anchor ? r->anchor->name : "",
1362		    pf->opts & PF_OPT_VERBOSE2,
1363		    pf->opts & PF_OPT_NUMERIC);
1364	}
1365	path[len] = '\0';
1366	pfctl_clear_pool(&r->rpool);
1367	return (0);
1368}
1369
1370int
1371pfctl_add_altq(struct pfctl *pf, struct pf_altq *a)
1372{
1373	if (altqsupport &&
1374	    (loadopt & PFCTL_FLAG_ALTQ) != 0) {
1375		memcpy(&pf->paltq->altq, a, sizeof(struct pf_altq));
1376		if ((pf->opts & PF_OPT_NOACTION) == 0) {
1377			if (ioctl(pf->dev, DIOCADDALTQ, pf->paltq)) {
1378				if (errno == ENXIO)
1379					errx(1, "qtype not configured");
1380				else if (errno == ENODEV)
1381					errx(1, "%s: driver does not support "
1382					    "altq", a->ifname);
1383				else
1384					err(1, "DIOCADDALTQ");
1385			}
1386		}
1387		pfaltq_store(&pf->paltq->altq);
1388	}
1389	return (0);
1390}
1391
1392int
1393pfctl_rules(int dev, char *filename, int opts, int optimize,
1394    char *anchorname, struct pfr_buffer *trans)
1395{
1396#define ERR(x) do { warn(x); goto _error; } while(0)
1397#define ERRX(x) do { warnx(x); goto _error; } while(0)
1398
1399	struct pfr_buffer	*t, buf;
1400	struct pfioc_altq	 pa;
1401	struct pfctl		 pf;
1402	struct pf_ruleset	*rs;
1403	struct pfr_table	 trs;
1404	char			*path;
1405	int			 osize;
1406
1407	RB_INIT(&pf_anchors);
1408	memset(&pf_main_anchor, 0, sizeof(pf_main_anchor));
1409	pf_init_ruleset(&pf_main_anchor.ruleset);
1410	pf_main_anchor.ruleset.anchor = &pf_main_anchor;
1411	if (trans == NULL) {
1412		bzero(&buf, sizeof(buf));
1413		buf.pfrb_type = PFRB_TRANS;
1414		t = &buf;
1415		osize = 0;
1416	} else {
1417		t = trans;
1418		osize = t->pfrb_size;
1419	}
1420
1421	memset(&pa, 0, sizeof(pa));
1422	memset(&pf, 0, sizeof(pf));
1423	memset(&trs, 0, sizeof(trs));
1424	if ((path = calloc(1, MAXPATHLEN)) == NULL)
1425		ERRX("pfctl_rules: calloc");
1426	if (strlcpy(trs.pfrt_anchor, anchorname,
1427	    sizeof(trs.pfrt_anchor)) >= sizeof(trs.pfrt_anchor))
1428		ERRX("pfctl_rules: strlcpy");
1429	pf.dev = dev;
1430	pf.opts = opts;
1431	pf.optimize = optimize;
1432	pf.loadopt = loadopt;
1433
1434	/* non-brace anchor, create without resolving the path */
1435	if ((pf.anchor = calloc(1, sizeof(*pf.anchor))) == NULL)
1436		ERRX("pfctl_rules: calloc");
1437	rs = &pf.anchor->ruleset;
1438	pf_init_ruleset(rs);
1439	rs->anchor = pf.anchor;
1440	if (strlcpy(pf.anchor->path, anchorname,
1441	    sizeof(pf.anchor->path)) >= sizeof(pf.anchor->path))
1442		errx(1, "pfctl_add_rule: strlcpy");
1443	if (strlcpy(pf.anchor->name, anchorname,
1444	    sizeof(pf.anchor->name)) >= sizeof(pf.anchor->name))
1445		errx(1, "pfctl_add_rule: strlcpy");
1446
1447
1448	pf.astack[0] = pf.anchor;
1449	pf.asd = 0;
1450	if (anchorname[0])
1451		pf.loadopt &= ~PFCTL_FLAG_ALTQ;
1452	pf.paltq = &pa;
1453	pf.trans = t;
1454	pfctl_init_options(&pf);
1455
1456	if ((opts & PF_OPT_NOACTION) == 0) {
1457		/*
1458		 * XXX For the time being we need to open transactions for
1459		 * the main ruleset before parsing, because tables are still
1460		 * loaded at parse time.
1461		 */
1462		if (pfctl_ruleset_trans(&pf, anchorname, pf.anchor))
1463			ERRX("pfctl_rules");
1464		if (altqsupport && (pf.loadopt & PFCTL_FLAG_ALTQ))
1465			pa.ticket =
1466			    pfctl_get_ticket(t, PF_RULESET_ALTQ, anchorname);
1467		if (pf.loadopt & PFCTL_FLAG_TABLE)
1468			pf.astack[0]->ruleset.tticket =
1469			    pfctl_get_ticket(t, PF_RULESET_TABLE, anchorname);
1470	}
1471
1472	if (parse_config(filename, &pf) < 0) {
1473		if ((opts & PF_OPT_NOACTION) == 0)
1474			ERRX("Syntax error in config file: "
1475			    "pf rules not loaded");
1476		else
1477			goto _error;
1478	}
1479
1480	if ((pf.loadopt & PFCTL_FLAG_FILTER &&
1481	    (pfctl_load_ruleset(&pf, path, rs, PF_RULESET_SCRUB, 0))) ||
1482	    (pf.loadopt & PFCTL_FLAG_NAT &&
1483	    (pfctl_load_ruleset(&pf, path, rs, PF_RULESET_NAT, 0) ||
1484	    pfctl_load_ruleset(&pf, path, rs, PF_RULESET_RDR, 0) ||
1485	    pfctl_load_ruleset(&pf, path, rs, PF_RULESET_BINAT, 0))) ||
1486	    (pf.loadopt & PFCTL_FLAG_FILTER &&
1487	    pfctl_load_ruleset(&pf, path, rs, PF_RULESET_FILTER, 0))) {
1488		if ((opts & PF_OPT_NOACTION) == 0)
1489			ERRX("Unable to load rules into kernel");
1490		else
1491			goto _error;
1492	}
1493
1494	if ((altqsupport && (pf.loadopt & PFCTL_FLAG_ALTQ) != 0))
1495		if (check_commit_altq(dev, opts) != 0)
1496			ERRX("errors in altq config");
1497
1498	/* process "load anchor" directives */
1499	if (!anchorname[0])
1500		if (pfctl_load_anchors(dev, &pf, t) == -1)
1501			ERRX("load anchors");
1502
1503	if (trans == NULL && (opts & PF_OPT_NOACTION) == 0) {
1504		if (!anchorname[0])
1505			if (pfctl_load_options(&pf))
1506				goto _error;
1507		if (pfctl_trans(dev, t, DIOCXCOMMIT, osize))
1508			ERR("DIOCXCOMMIT");
1509	}
1510	return (0);
1511
1512_error:
1513	if (trans == NULL) {	/* main ruleset */
1514		if ((opts & PF_OPT_NOACTION) == 0)
1515			if (pfctl_trans(dev, t, DIOCXROLLBACK, osize))
1516				err(1, "DIOCXROLLBACK");
1517		exit(1);
1518	} else {		/* sub ruleset */
1519		return (-1);
1520	}
1521
1522#undef ERR
1523#undef ERRX
1524}
1525
1526FILE *
1527pfctl_fopen(const char *name, const char *mode)
1528{
1529	struct stat	 st;
1530	FILE		*fp;
1531
1532	fp = fopen(name, mode);
1533	if (fp == NULL)
1534		return (NULL);
1535	if (fstat(fileno(fp), &st)) {
1536		fclose(fp);
1537		return (NULL);
1538	}
1539	if (S_ISDIR(st.st_mode)) {
1540		fclose(fp);
1541		errno = EISDIR;
1542		return (NULL);
1543	}
1544	return (fp);
1545}
1546
1547void
1548pfctl_init_options(struct pfctl *pf)
1549{
1550
1551	pf->timeout[PFTM_TCP_FIRST_PACKET] = PFTM_TCP_FIRST_PACKET_VAL;
1552	pf->timeout[PFTM_TCP_OPENING] = PFTM_TCP_OPENING_VAL;
1553	pf->timeout[PFTM_TCP_ESTABLISHED] = PFTM_TCP_ESTABLISHED_VAL;
1554	pf->timeout[PFTM_TCP_CLOSING] = PFTM_TCP_CLOSING_VAL;
1555	pf->timeout[PFTM_TCP_FIN_WAIT] = PFTM_TCP_FIN_WAIT_VAL;
1556	pf->timeout[PFTM_TCP_CLOSED] = PFTM_TCP_CLOSED_VAL;
1557	pf->timeout[PFTM_UDP_FIRST_PACKET] = PFTM_UDP_FIRST_PACKET_VAL;
1558	pf->timeout[PFTM_UDP_SINGLE] = PFTM_UDP_SINGLE_VAL;
1559	pf->timeout[PFTM_UDP_MULTIPLE] = PFTM_UDP_MULTIPLE_VAL;
1560	pf->timeout[PFTM_ICMP_FIRST_PACKET] = PFTM_ICMP_FIRST_PACKET_VAL;
1561	pf->timeout[PFTM_ICMP_ERROR_REPLY] = PFTM_ICMP_ERROR_REPLY_VAL;
1562	pf->timeout[PFTM_OTHER_FIRST_PACKET] = PFTM_OTHER_FIRST_PACKET_VAL;
1563	pf->timeout[PFTM_OTHER_SINGLE] = PFTM_OTHER_SINGLE_VAL;
1564	pf->timeout[PFTM_OTHER_MULTIPLE] = PFTM_OTHER_MULTIPLE_VAL;
1565	pf->timeout[PFTM_FRAG] = PFTM_FRAG_VAL;
1566	pf->timeout[PFTM_INTERVAL] = PFTM_INTERVAL_VAL;
1567	pf->timeout[PFTM_SRC_NODE] = PFTM_SRC_NODE_VAL;
1568	pf->timeout[PFTM_TS_DIFF] = PFTM_TS_DIFF_VAL;
1569	pf->timeout[PFTM_ADAPTIVE_START] = PFSTATE_ADAPT_START;
1570	pf->timeout[PFTM_ADAPTIVE_END] = PFSTATE_ADAPT_END;
1571
1572	pf->limit[PF_LIMIT_STATES] = PFSTATE_HIWAT;
1573	pf->limit[PF_LIMIT_FRAGS] = PFFRAG_FRENT_HIWAT;
1574	pf->limit[PF_LIMIT_SRC_NODES] = PFSNODE_HIWAT;
1575	pf->limit[PF_LIMIT_TABLE_ENTRIES] = PFR_KENTRY_HIWAT;
1576
1577	pf->debug = PF_DEBUG_URGENT;
1578}
1579
1580int
1581pfctl_load_options(struct pfctl *pf)
1582{
1583	int i, error = 0;
1584
1585	if ((loadopt & PFCTL_FLAG_OPTION) == 0)
1586		return (0);
1587
1588	/* load limits */
1589	for (i = 0; i < PF_LIMIT_MAX; i++) {
1590		if ((pf->opts & PF_OPT_MERGE) && !pf->limit_set[i])
1591			continue;
1592		if (pfctl_load_limit(pf, i, pf->limit[i]))
1593			error = 1;
1594	}
1595
1596	/*
1597	 * If we've set the limit, but haven't explicitly set adaptive
1598	 * timeouts, do it now with a start of 60% and end of 120%.
1599	 */
1600	if (pf->limit_set[PF_LIMIT_STATES] &&
1601	    !pf->timeout_set[PFTM_ADAPTIVE_START] &&
1602	    !pf->timeout_set[PFTM_ADAPTIVE_END]) {
1603		pf->timeout[PFTM_ADAPTIVE_START] =
1604			(pf->limit[PF_LIMIT_STATES] / 10) * 6;
1605		pf->timeout_set[PFTM_ADAPTIVE_START] = 1;
1606		pf->timeout[PFTM_ADAPTIVE_END] =
1607			(pf->limit[PF_LIMIT_STATES] / 10) * 12;
1608		pf->timeout_set[PFTM_ADAPTIVE_END] = 1;
1609	}
1610
1611	/* load timeouts */
1612	for (i = 0; i < PFTM_MAX; i++) {
1613		if ((pf->opts & PF_OPT_MERGE) && !pf->timeout_set[i])
1614			continue;
1615		if (pfctl_load_timeout(pf, i, pf->timeout[i]))
1616			error = 1;
1617	}
1618
1619	/* load debug */
1620	if (!(pf->opts & PF_OPT_MERGE) || pf->debug_set)
1621		if (pfctl_load_debug(pf, pf->debug))
1622			error = 1;
1623
1624	/* load logif */
1625	if (!(pf->opts & PF_OPT_MERGE) || pf->ifname_set)
1626		if (pfctl_load_logif(pf, pf->ifname))
1627			error = 1;
1628
1629	/* load hostid */
1630	if (!(pf->opts & PF_OPT_MERGE) || pf->hostid_set)
1631		if (pfctl_load_hostid(pf, pf->hostid))
1632			error = 1;
1633
1634	return (error);
1635}
1636
1637int
1638pfctl_set_limit(struct pfctl *pf, const char *opt, unsigned int limit)
1639{
1640	int i;
1641
1642
1643	for (i = 0; pf_limits[i].name; i++) {
1644		if (strcasecmp(opt, pf_limits[i].name) == 0) {
1645			pf->limit[pf_limits[i].index] = limit;
1646			pf->limit_set[pf_limits[i].index] = 1;
1647			break;
1648		}
1649	}
1650	if (pf_limits[i].name == NULL) {
1651		warnx("Bad pool name.");
1652		return (1);
1653	}
1654
1655	if (pf->opts & PF_OPT_VERBOSE)
1656		printf("set limit %s %d\n", opt, limit);
1657
1658	return (0);
1659}
1660
1661int
1662pfctl_load_limit(struct pfctl *pf, unsigned int index, unsigned int limit)
1663{
1664	struct pfioc_limit pl;
1665
1666	memset(&pl, 0, sizeof(pl));
1667	pl.index = index;
1668	pl.limit = limit;
1669	if (ioctl(pf->dev, DIOCSETLIMIT, &pl)) {
1670		if (errno == EBUSY)
1671			warnx("Current pool size exceeds requested hard limit");
1672		else
1673			warnx("DIOCSETLIMIT");
1674		return (1);
1675	}
1676	return (0);
1677}
1678
1679int
1680pfctl_set_timeout(struct pfctl *pf, const char *opt, int seconds, int quiet)
1681{
1682	int i;
1683
1684	if ((loadopt & PFCTL_FLAG_OPTION) == 0)
1685		return (0);
1686
1687	for (i = 0; pf_timeouts[i].name; i++) {
1688		if (strcasecmp(opt, pf_timeouts[i].name) == 0) {
1689			pf->timeout[pf_timeouts[i].timeout] = seconds;
1690			pf->timeout_set[pf_timeouts[i].timeout] = 1;
1691			break;
1692		}
1693	}
1694
1695	if (pf_timeouts[i].name == NULL) {
1696		warnx("Bad timeout name.");
1697		return (1);
1698	}
1699
1700
1701	if (pf->opts & PF_OPT_VERBOSE && ! quiet)
1702		printf("set timeout %s %d\n", opt, seconds);
1703
1704	return (0);
1705}
1706
1707int
1708pfctl_load_timeout(struct pfctl *pf, unsigned int timeout, unsigned int seconds)
1709{
1710	struct pfioc_tm pt;
1711
1712	memset(&pt, 0, sizeof(pt));
1713	pt.timeout = timeout;
1714	pt.seconds = seconds;
1715	if (ioctl(pf->dev, DIOCSETTIMEOUT, &pt)) {
1716		warnx("DIOCSETTIMEOUT");
1717		return (1);
1718	}
1719	return (0);
1720}
1721
1722int
1723pfctl_set_optimization(struct pfctl *pf, const char *opt)
1724{
1725	const struct pf_hint *hint;
1726	int i, r;
1727
1728	if ((loadopt & PFCTL_FLAG_OPTION) == 0)
1729		return (0);
1730
1731	for (i = 0; pf_hints[i].name; i++)
1732		if (strcasecmp(opt, pf_hints[i].name) == 0)
1733			break;
1734
1735	hint = pf_hints[i].hint;
1736	if (hint == NULL) {
1737		warnx("invalid state timeouts optimization");
1738		return (1);
1739	}
1740
1741	for (i = 0; hint[i].name; i++)
1742		if ((r = pfctl_set_timeout(pf, hint[i].name,
1743		    hint[i].timeout, 1)))
1744			return (r);
1745
1746	if (pf->opts & PF_OPT_VERBOSE)
1747		printf("set optimization %s\n", opt);
1748
1749	return (0);
1750}
1751
1752int
1753pfctl_set_logif(struct pfctl *pf, char *ifname)
1754{
1755
1756	if ((loadopt & PFCTL_FLAG_OPTION) == 0)
1757		return (0);
1758
1759	if (!strcmp(ifname, "none")) {
1760		free(pf->ifname);
1761		pf->ifname = NULL;
1762	} else {
1763		pf->ifname = strdup(ifname);
1764		if (!pf->ifname)
1765			errx(1, "pfctl_set_logif: strdup");
1766	}
1767	pf->ifname_set = 1;
1768
1769	if (pf->opts & PF_OPT_VERBOSE)
1770		printf("set loginterface %s\n", ifname);
1771
1772	return (0);
1773}
1774
1775int
1776pfctl_load_logif(struct pfctl *pf, char *ifname)
1777{
1778	struct pfioc_if pi;
1779
1780	memset(&pi, 0, sizeof(pi));
1781	if (ifname && strlcpy(pi.ifname, ifname,
1782	    sizeof(pi.ifname)) >= sizeof(pi.ifname)) {
1783		warnx("pfctl_load_logif: strlcpy");
1784		return (1);
1785	}
1786	if (ioctl(pf->dev, DIOCSETSTATUSIF, &pi)) {
1787		warnx("DIOCSETSTATUSIF");
1788		return (1);
1789	}
1790	return (0);
1791}
1792
1793int
1794pfctl_set_hostid(struct pfctl *pf, u_int32_t hostid)
1795{
1796	if ((loadopt & PFCTL_FLAG_OPTION) == 0)
1797		return (0);
1798
1799	HTONL(hostid);
1800
1801	pf->hostid = hostid;
1802	pf->hostid_set = 1;
1803
1804	if (pf->opts & PF_OPT_VERBOSE)
1805		printf("set hostid 0x%08x\n", ntohl(hostid));
1806
1807	return (0);
1808}
1809
1810int
1811pfctl_load_hostid(struct pfctl *pf, u_int32_t hostid)
1812{
1813	if (ioctl(dev, DIOCSETHOSTID, &hostid)) {
1814		warnx("DIOCSETHOSTID");
1815		return (1);
1816	}
1817	return (0);
1818}
1819
1820int
1821pfctl_set_debug(struct pfctl *pf, char *d)
1822{
1823	u_int32_t	level;
1824
1825	if ((loadopt & PFCTL_FLAG_OPTION) == 0)
1826		return (0);
1827
1828	if (!strcmp(d, "none"))
1829		pf->debug = PF_DEBUG_NONE;
1830	else if (!strcmp(d, "urgent"))
1831		pf->debug = PF_DEBUG_URGENT;
1832	else if (!strcmp(d, "misc"))
1833		pf->debug = PF_DEBUG_MISC;
1834	else if (!strcmp(d, "loud"))
1835		pf->debug = PF_DEBUG_NOISY;
1836	else {
1837		warnx("unknown debug level \"%s\"", d);
1838		return (-1);
1839	}
1840
1841	pf->debug_set = 1;
1842
1843	if ((pf->opts & PF_OPT_NOACTION) == 0)
1844		if (ioctl(dev, DIOCSETDEBUG, &level))
1845			err(1, "DIOCSETDEBUG");
1846
1847	if (pf->opts & PF_OPT_VERBOSE)
1848		printf("set debug %s\n", d);
1849
1850	return (0);
1851}
1852
1853int
1854pfctl_load_debug(struct pfctl *pf, unsigned int level)
1855{
1856	if (ioctl(pf->dev, DIOCSETDEBUG, &level)) {
1857		warnx("DIOCSETDEBUG");
1858		return (1);
1859	}
1860	return (0);
1861}
1862
1863int
1864pfctl_set_interface_flags(struct pfctl *pf, char *ifname, int flags, int how)
1865{
1866	struct pfioc_iface	pi;
1867
1868	if ((loadopt & PFCTL_FLAG_OPTION) == 0)
1869		return (0);
1870
1871	bzero(&pi, sizeof(pi));
1872
1873	pi.pfiio_flags = flags;
1874
1875	if (strlcpy(pi.pfiio_name, ifname, sizeof(pi.pfiio_name)) >=
1876	    sizeof(pi.pfiio_name))
1877		errx(1, "pfctl_set_interface_flags: strlcpy");
1878
1879	if ((pf->opts & PF_OPT_NOACTION) == 0) {
1880		if (how == 0) {
1881			if (ioctl(pf->dev, DIOCCLRIFFLAG, &pi))
1882				err(1, "DIOCCLRIFFLAG");
1883		} else {
1884			if (ioctl(pf->dev, DIOCSETIFFLAG, &pi))
1885				err(1, "DIOCSETIFFLAG");
1886		}
1887	}
1888	return (0);
1889}
1890
1891void
1892pfctl_debug(int dev, u_int32_t level, int opts)
1893{
1894	if (ioctl(dev, DIOCSETDEBUG, &level))
1895		err(1, "DIOCSETDEBUG");
1896	if ((opts & PF_OPT_QUIET) == 0) {
1897		fprintf(stderr, "debug level set to '");
1898		switch (level) {
1899		case PF_DEBUG_NONE:
1900			fprintf(stderr, "none");
1901			break;
1902		case PF_DEBUG_URGENT:
1903			fprintf(stderr, "urgent");
1904			break;
1905		case PF_DEBUG_MISC:
1906			fprintf(stderr, "misc");
1907			break;
1908		case PF_DEBUG_NOISY:
1909			fprintf(stderr, "loud");
1910			break;
1911		default:
1912			fprintf(stderr, "<invalid>");
1913			break;
1914		}
1915		fprintf(stderr, "'\n");
1916	}
1917}
1918
1919int
1920pfctl_test_altqsupport(int dev, int opts)
1921{
1922	struct pfioc_altq pa;
1923
1924	if (ioctl(dev, DIOCGETALTQS, &pa)) {
1925		if (errno == ENODEV) {
1926			if (!(opts & PF_OPT_QUIET))
1927				fprintf(stderr, "No ALTQ support in kernel\n"
1928				    "ALTQ related functions disabled\n");
1929			return (0);
1930		} else
1931			err(1, "DIOCGETALTQS");
1932	}
1933	return (1);
1934}
1935
1936int
1937pfctl_show_anchors(int dev, int opts, char *anchorname)
1938{
1939	struct pfioc_ruleset	 pr;
1940	u_int32_t		 mnr, nr;
1941
1942	memset(&pr, 0, sizeof(pr));
1943	memcpy(pr.path, anchorname, sizeof(pr.path));
1944	if (ioctl(dev, DIOCGETRULESETS, &pr)) {
1945		if (errno == EINVAL)
1946			fprintf(stderr, "Anchor '%s' not found.\n",
1947			    anchorname);
1948		else
1949			err(1, "DIOCGETRULESETS");
1950		return (-1);
1951	}
1952	mnr = pr.nr;
1953	for (nr = 0; nr < mnr; ++nr) {
1954		char sub[MAXPATHLEN];
1955
1956		pr.nr = nr;
1957		if (ioctl(dev, DIOCGETRULESET, &pr))
1958			err(1, "DIOCGETRULESET");
1959		if (!strcmp(pr.name, PF_RESERVED_ANCHOR))
1960			continue;
1961		sub[0] = 0;
1962		if (pr.path[0]) {
1963			strlcat(sub, pr.path, sizeof(sub));
1964			strlcat(sub, "/", sizeof(sub));
1965		}
1966		strlcat(sub, pr.name, sizeof(sub));
1967		if (sub[0] != '_' || (opts & PF_OPT_VERBOSE))
1968			printf("  %s\n", sub);
1969		if ((opts & PF_OPT_VERBOSE) && pfctl_show_anchors(dev, opts, sub))
1970			return (-1);
1971	}
1972	return (0);
1973}
1974
1975const char *
1976pfctl_lookup_option(char *cmd, const char **list)
1977{
1978	if (cmd != NULL && *cmd)
1979		for (; *list; list++)
1980			if (!strncmp(cmd, *list, strlen(cmd)))
1981				return (*list);
1982	return (NULL);
1983}
1984
1985int
1986main(int argc, char *argv[])
1987{
1988	int	 error = 0;
1989	int	 ch;
1990	int	 mode = O_RDONLY;
1991	int	 opts = 0;
1992	int	 optimize = PF_OPTIMIZE_BASIC;
1993	char	 anchorname[MAXPATHLEN];
1994	char	*path;
1995
1996	if (argc < 2)
1997		usage();
1998
1999	while ((ch = getopt(argc, argv,
2000	    "a:AdD:eqf:F:ghi:k:K:mnNOo:Pp:rRs:t:T:vx:z")) != -1) {
2001		switch (ch) {
2002		case 'a':
2003			anchoropt = optarg;
2004			break;
2005		case 'd':
2006			opts |= PF_OPT_DISABLE;
2007			mode = O_RDWR;
2008			break;
2009		case 'D':
2010			if (pfctl_cmdline_symset(optarg) < 0)
2011				warnx("could not parse macro definition %s",
2012				    optarg);
2013			break;
2014		case 'e':
2015			opts |= PF_OPT_ENABLE;
2016			mode = O_RDWR;
2017			break;
2018		case 'q':
2019			opts |= PF_OPT_QUIET;
2020			break;
2021		case 'F':
2022			clearopt = pfctl_lookup_option(optarg, clearopt_list);
2023			if (clearopt == NULL) {
2024				warnx("Unknown flush modifier '%s'", optarg);
2025				usage();
2026			}
2027			mode = O_RDWR;
2028			break;
2029		case 'i':
2030			ifaceopt = optarg;
2031			break;
2032		case 'k':
2033			if (state_killers >= 2) {
2034				warnx("can only specify -k twice");
2035				usage();
2036				/* NOTREACHED */
2037			}
2038			state_kill[state_killers++] = optarg;
2039			mode = O_RDWR;
2040			break;
2041		case 'K':
2042			if (src_node_killers >= 2) {
2043				warnx("can only specify -K twice");
2044				usage();
2045				/* NOTREACHED */
2046			}
2047			src_node_kill[src_node_killers++] = optarg;
2048			mode = O_RDWR;
2049			break;
2050		case 'm':
2051			opts |= PF_OPT_MERGE;
2052			break;
2053		case 'n':
2054			opts |= PF_OPT_NOACTION;
2055			break;
2056		case 'N':
2057			loadopt |= PFCTL_FLAG_NAT;
2058			break;
2059		case 'r':
2060			opts |= PF_OPT_USEDNS;
2061			break;
2062		case 'f':
2063			rulesopt = optarg;
2064			mode = O_RDWR;
2065			break;
2066		case 'g':
2067			opts |= PF_OPT_DEBUG;
2068			break;
2069		case 'A':
2070			loadopt |= PFCTL_FLAG_ALTQ;
2071			break;
2072		case 'R':
2073			loadopt |= PFCTL_FLAG_FILTER;
2074			break;
2075		case 'o':
2076			optiopt = pfctl_lookup_option(optarg, optiopt_list);
2077			if (optiopt == NULL) {
2078				warnx("Unknown optimization '%s'", optarg);
2079				usage();
2080			}
2081			opts |= PF_OPT_OPTIMIZE;
2082			break;
2083		case 'O':
2084			loadopt |= PFCTL_FLAG_OPTION;
2085			break;
2086		case 'p':
2087			pf_device = optarg;
2088			break;
2089		case 'P':
2090			opts |= PF_OPT_NUMERIC;
2091			break;
2092		case 's':
2093			showopt = pfctl_lookup_option(optarg, showopt_list);
2094			if (showopt == NULL) {
2095				warnx("Unknown show modifier '%s'", optarg);
2096				usage();
2097			}
2098			break;
2099		case 't':
2100			tableopt = optarg;
2101			break;
2102		case 'T':
2103			tblcmdopt = pfctl_lookup_option(optarg, tblcmdopt_list);
2104			if (tblcmdopt == NULL) {
2105				warnx("Unknown table command '%s'", optarg);
2106				usage();
2107			}
2108			break;
2109		case 'v':
2110			if (opts & PF_OPT_VERBOSE)
2111				opts |= PF_OPT_VERBOSE2;
2112			opts |= PF_OPT_VERBOSE;
2113			break;
2114		case 'x':
2115			debugopt = pfctl_lookup_option(optarg, debugopt_list);
2116			if (debugopt == NULL) {
2117				warnx("Unknown debug level '%s'", optarg);
2118				usage();
2119			}
2120			mode = O_RDWR;
2121			break;
2122		case 'z':
2123			opts |= PF_OPT_CLRRULECTRS;
2124			mode = O_RDWR;
2125			break;
2126		case 'h':
2127			/* FALLTHROUGH */
2128		default:
2129			usage();
2130			/* NOTREACHED */
2131		}
2132	}
2133
2134	if (tblcmdopt != NULL) {
2135		argc -= optind;
2136		argv += optind;
2137		ch = *tblcmdopt;
2138		if (ch == 'l') {
2139			loadopt |= PFCTL_FLAG_TABLE;
2140			tblcmdopt = NULL;
2141		} else
2142			mode = strchr("acdefkrz", ch) ? O_RDWR : O_RDONLY;
2143	} else if (argc != optind) {
2144		warnx("unknown command line argument: %s ...", argv[optind]);
2145		usage();
2146		/* NOTREACHED */
2147	}
2148	if (loadopt == 0)
2149		loadopt = ~0;
2150
2151	if ((path = calloc(1, MAXPATHLEN)) == NULL)
2152		errx(1, "pfctl: calloc");
2153	memset(anchorname, 0, sizeof(anchorname));
2154	if (anchoropt != NULL) {
2155		int len = strlen(anchoropt);
2156
2157		if (anchoropt[len - 1] == '*') {
2158			if (len >= 2 && anchoropt[len - 2] == '/')
2159				anchoropt[len - 2] = '\0';
2160			else
2161				anchoropt[len - 1] = '\0';
2162			opts |= PF_OPT_RECURSE;
2163		}
2164		if (strlcpy(anchorname, anchoropt,
2165		    sizeof(anchorname)) >= sizeof(anchorname))
2166			errx(1, "anchor name '%s' too long",
2167			    anchoropt);
2168		loadopt &= PFCTL_FLAG_FILTER|PFCTL_FLAG_NAT|PFCTL_FLAG_TABLE;
2169	}
2170
2171	if ((opts & PF_OPT_NOACTION) == 0) {
2172		dev = open(pf_device, mode);
2173		if (dev == -1)
2174			err(1, "%s", pf_device);
2175		altqsupport = pfctl_test_altqsupport(dev, opts);
2176	} else {
2177		dev = open(pf_device, O_RDONLY);
2178		if (dev >= 0)
2179			opts |= PF_OPT_DUMMYACTION;
2180		/* turn off options */
2181		opts &= ~ (PF_OPT_DISABLE | PF_OPT_ENABLE);
2182		clearopt = showopt = debugopt = NULL;
2183#if !defined(ENABLE_ALTQ)
2184		altqsupport = 0;
2185#else
2186		altqsupport = 1;
2187#endif
2188	}
2189
2190	if (opts & PF_OPT_DISABLE)
2191		if (pfctl_disable(dev, opts))
2192			error = 1;
2193
2194	if (showopt != NULL) {
2195		switch (*showopt) {
2196		case 'A':
2197			pfctl_show_anchors(dev, opts, anchorname);
2198			break;
2199		case 'r':
2200			pfctl_load_fingerprints(dev, opts);
2201			pfctl_show_rules(dev, path, opts, PFCTL_SHOW_RULES,
2202			    anchorname, 0);
2203			break;
2204		case 'l':
2205			pfctl_load_fingerprints(dev, opts);
2206			pfctl_show_rules(dev, path, opts, PFCTL_SHOW_LABELS,
2207			    anchorname, 0);
2208			break;
2209		case 'n':
2210			pfctl_load_fingerprints(dev, opts);
2211			pfctl_show_nat(dev, opts, anchorname);
2212			break;
2213		case 'q':
2214			pfctl_show_altq(dev, ifaceopt, opts,
2215			    opts & PF_OPT_VERBOSE2);
2216			break;
2217		case 's':
2218			pfctl_show_states(dev, ifaceopt, opts);
2219			break;
2220		case 'S':
2221			pfctl_show_src_nodes(dev, opts);
2222			break;
2223		case 'i':
2224			pfctl_show_status(dev, opts);
2225			break;
2226		case 't':
2227			pfctl_show_timeouts(dev, opts);
2228			break;
2229		case 'm':
2230			pfctl_show_limits(dev, opts);
2231			break;
2232		case 'a':
2233			opts |= PF_OPT_SHOWALL;
2234			pfctl_load_fingerprints(dev, opts);
2235
2236			pfctl_show_nat(dev, opts, anchorname);
2237			pfctl_show_rules(dev, path, opts, 0, anchorname, 0);
2238			pfctl_show_altq(dev, ifaceopt, opts, 0);
2239			pfctl_show_states(dev, ifaceopt, opts);
2240			pfctl_show_src_nodes(dev, opts);
2241			pfctl_show_status(dev, opts);
2242			pfctl_show_rules(dev, path, opts, 1, anchorname, 0);
2243			pfctl_show_timeouts(dev, opts);
2244			pfctl_show_limits(dev, opts);
2245			pfctl_show_tables(anchorname, opts);
2246			pfctl_show_fingerprints(opts);
2247			break;
2248		case 'T':
2249			pfctl_show_tables(anchorname, opts);
2250			break;
2251		case 'o':
2252			pfctl_load_fingerprints(dev, opts);
2253			pfctl_show_fingerprints(opts);
2254			break;
2255		case 'I':
2256			pfctl_show_ifaces(ifaceopt, opts);
2257			break;
2258		}
2259	}
2260
2261	if ((opts & PF_OPT_CLRRULECTRS) && showopt == NULL)
2262		pfctl_show_rules(dev, path, opts, PFCTL_SHOW_NOTHING,
2263		    anchorname, 0);
2264
2265	if (clearopt != NULL) {
2266		if (anchorname[0] == '_' || strstr(anchorname, "/_") != NULL)
2267			errx(1, "anchor names beginning with '_' cannot "
2268			    "be modified from the command line");
2269
2270		switch (*clearopt) {
2271		case 'r':
2272			pfctl_clear_rules(dev, opts, anchorname);
2273			break;
2274		case 'n':
2275			pfctl_clear_nat(dev, opts, anchorname);
2276			break;
2277		case 'q':
2278			pfctl_clear_altq(dev, opts);
2279			break;
2280		case 's':
2281			pfctl_clear_states(dev, ifaceopt, opts);
2282			break;
2283		case 'S':
2284			pfctl_clear_src_nodes(dev, opts);
2285			break;
2286		case 'i':
2287			pfctl_clear_stats(dev, opts);
2288			break;
2289		case 'a':
2290			pfctl_clear_rules(dev, opts, anchorname);
2291			pfctl_clear_nat(dev, opts, anchorname);
2292			pfctl_clear_tables(anchorname, opts);
2293			if (!*anchorname) {
2294				pfctl_clear_altq(dev, opts);
2295				pfctl_clear_states(dev, ifaceopt, opts);
2296				pfctl_clear_src_nodes(dev, opts);
2297				pfctl_clear_stats(dev, opts);
2298				pfctl_clear_fingerprints(dev, opts);
2299				pfctl_clear_interface_flags(dev, opts);
2300			}
2301			break;
2302		case 'o':
2303			pfctl_clear_fingerprints(dev, opts);
2304			break;
2305		case 'T':
2306			pfctl_clear_tables(anchorname, opts);
2307			break;
2308		}
2309	}
2310	if (state_killers) {
2311		if (!strcmp(state_kill[0], "label"))
2312			pfctl_label_kill_states(dev, ifaceopt, opts);
2313		else if (!strcmp(state_kill[0], "id"))
2314			pfctl_id_kill_states(dev, ifaceopt, opts);
2315		else
2316			pfctl_net_kill_states(dev, ifaceopt, opts);
2317	}
2318
2319	if (src_node_killers)
2320		pfctl_kill_src_nodes(dev, ifaceopt, opts);
2321
2322	if (tblcmdopt != NULL) {
2323		error = pfctl_command_tables(argc, argv, tableopt,
2324		    tblcmdopt, rulesopt, anchorname, opts);
2325		rulesopt = NULL;
2326	}
2327	if (optiopt != NULL) {
2328		switch (*optiopt) {
2329		case 'n':
2330			optimize = 0;
2331			break;
2332		case 'b':
2333			optimize |= PF_OPTIMIZE_BASIC;
2334			break;
2335		case 'o':
2336		case 'p':
2337			optimize |= PF_OPTIMIZE_PROFILE;
2338			break;
2339		}
2340	}
2341
2342	if ((rulesopt != NULL) && (loadopt & PFCTL_FLAG_OPTION) &&
2343	    !anchorname[0])
2344		if (pfctl_clear_interface_flags(dev, opts | PF_OPT_QUIET))
2345			error = 1;
2346
2347	if (rulesopt != NULL && !(opts & (PF_OPT_MERGE|PF_OPT_NOACTION)) &&
2348	    !anchorname[0] && (loadopt & PFCTL_FLAG_OPTION))
2349		if (pfctl_file_fingerprints(dev, opts, PF_OSFP_FILE))
2350			error = 1;
2351
2352	if (rulesopt != NULL) {
2353		if (anchorname[0] == '_' || strstr(anchorname, "/_") != NULL)
2354			errx(1, "anchor names beginning with '_' cannot "
2355			    "be modified from the command line");
2356		if (pfctl_rules(dev, rulesopt, opts, optimize,
2357		    anchorname, NULL))
2358			error = 1;
2359		else if (!(opts & PF_OPT_NOACTION) &&
2360		    (loadopt & PFCTL_FLAG_TABLE))
2361			warn_namespace_collision(NULL);
2362	}
2363
2364	if (opts & PF_OPT_ENABLE)
2365		if (pfctl_enable(dev, opts))
2366			error = 1;
2367
2368	if (debugopt != NULL) {
2369		switch (*debugopt) {
2370		case 'n':
2371			pfctl_debug(dev, PF_DEBUG_NONE, opts);
2372			break;
2373		case 'u':
2374			pfctl_debug(dev, PF_DEBUG_URGENT, opts);
2375			break;
2376		case 'm':
2377			pfctl_debug(dev, PF_DEBUG_MISC, opts);
2378			break;
2379		case 'l':
2380			pfctl_debug(dev, PF_DEBUG_NOISY, opts);
2381			break;
2382		}
2383	}
2384
2385	exit(error);
2386}
2387