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