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