xref: /freebsd/lib/libpfctl/libpfctl.c (revision 20777f79b896f3be37086fc7e2c6cbb8cdb865c5)
1 /*-
2  * SPDX-License-Identifier: BSD-2-Clause
3  *
4  * Copyright (c) 2021 Rubicon Communications, LLC (Netgate)
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  *
11  *    - Redistributions of source code must retain the above copyright
12  *      notice, this list of conditions and the following disclaimer.
13  *    - Redistributions in binary form must reproduce the above
14  *      copyright notice, this list of conditions and the following
15  *      disclaimer in the documentation and/or other materials provided
16  *      with the distribution.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
21  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
22  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
23  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
24  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
25  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
26  * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
28  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29  * POSSIBILITY OF SUCH DAMAGE.
30  *
31  * $FreeBSD$
32  */
33 
34 #include <sys/cdefs.h>
35 
36 #include <sys/ioctl.h>
37 #include <sys/nv.h>
38 #include <sys/queue.h>
39 #include <sys/types.h>
40 
41 #include <net/if.h>
42 #include <net/pfvar.h>
43 #include <netinet/in.h>
44 
45 #include <assert.h>
46 #include <err.h>
47 #include <errno.h>
48 #include <stdlib.h>
49 #include <string.h>
50 
51 #include "libpfctl.h"
52 
53 const char* PFCTL_SYNCOOKIES_MODE_NAMES[] = {
54 	"never",
55 	"always",
56 	"adaptive"
57 };
58 
59 static int	_pfctl_clear_states(int , const struct pfctl_kill *,
60 		    unsigned int *, uint64_t);
61 
62 static int
63 pfctl_do_ioctl(int dev, uint cmd, size_t size, nvlist_t **nvl)
64 {
65 	struct pfioc_nv nv;
66 	void *data;
67 	size_t nvlen;
68 	int ret;
69 
70 	data = nvlist_pack(*nvl, &nvlen);
71 
72 retry:
73 	nv.data = malloc(size);
74 	memcpy(nv.data, data, nvlen);
75 	free(data);
76 
77 	nv.len = nvlen;
78 	nv.size = size;
79 
80 	ret = ioctl(dev, cmd, &nv);
81 	if (ret == -1 && errno == ENOSPC) {
82 		size *= 2;
83 		free(nv.data);
84 		goto retry;
85 	}
86 
87 	nvlist_destroy(*nvl);
88 	*nvl = NULL;
89 
90 	if (ret == 0) {
91 		*nvl = nvlist_unpack(nv.data, nv.len, 0);
92 		if (*nvl == NULL) {
93 			free(nv.data);
94 			return (EIO);
95 		}
96 	} else {
97 		ret = errno;
98 	}
99 
100 	free(nv.data);
101 
102 	return (ret);
103 }
104 
105 static void
106 pf_nvuint_8_array(const nvlist_t *nvl, const char *name, size_t maxelems,
107     uint8_t *numbers, size_t *nelems)
108 {
109 	const uint64_t *tmp;
110 	size_t elems;
111 
112 	tmp = nvlist_get_number_array(nvl, name, &elems);
113 	assert(elems <= maxelems);
114 
115 	for (size_t i = 0; i < elems; i++)
116 		numbers[i] = tmp[i];
117 
118 	if (nelems)
119 		*nelems = elems;
120 }
121 
122 static void
123 pf_nvuint_16_array(const nvlist_t *nvl, const char *name, size_t maxelems,
124     uint16_t *numbers, size_t *nelems)
125 {
126 	const uint64_t *tmp;
127 	size_t elems;
128 
129 	tmp = nvlist_get_number_array(nvl, name, &elems);
130 	assert(elems <= maxelems);
131 
132 	for (size_t i = 0; i < elems; i++)
133 		numbers[i] = tmp[i];
134 
135 	if (nelems)
136 		*nelems = elems;
137 }
138 
139 static void
140 pf_nvuint_32_array(const nvlist_t *nvl, const char *name, size_t maxelems,
141     uint32_t *numbers, size_t *nelems)
142 {
143 	const uint64_t *tmp;
144 	size_t elems;
145 
146 	tmp = nvlist_get_number_array(nvl, name, &elems);
147 	assert(elems <= maxelems);
148 
149 	for (size_t i = 0; i < elems; i++)
150 		numbers[i] = tmp[i];
151 
152 	if (nelems)
153 		*nelems = elems;
154 }
155 
156 static void
157 pf_nvuint_64_array(const nvlist_t *nvl, const char *name, size_t maxelems,
158     uint64_t *numbers, size_t *nelems)
159 {
160 	const uint64_t *tmp;
161 	size_t elems;
162 
163 	tmp = nvlist_get_number_array(nvl, name, &elems);
164 	assert(elems <= maxelems);
165 
166 	for (size_t i = 0; i < elems; i++)
167 		numbers[i] = tmp[i];
168 
169 	if (nelems)
170 		*nelems = elems;
171 }
172 
173 static void
174 _pfctl_get_status_counters(const nvlist_t *nvl,
175     struct pfctl_status_counters *counters)
176 {
177 	const uint64_t		*ids, *counts;
178 	const char *const	*names;
179 	size_t id_len, counter_len, names_len;
180 
181 	ids = nvlist_get_number_array(nvl, "ids", &id_len);
182 	counts = nvlist_get_number_array(nvl, "counters", &counter_len);
183 	names = nvlist_get_string_array(nvl, "names", &names_len);
184 	assert(id_len == counter_len);
185 	assert(counter_len == names_len);
186 
187 	TAILQ_INIT(counters);
188 
189 	for (size_t i = 0; i < id_len; i++) {
190 		struct pfctl_status_counter *c;
191 
192 		c = malloc(sizeof(*c));
193 
194 		c->id = ids[i];
195 		c->counter = counts[i];
196 		c->name = strdup(names[i]);
197 
198 		TAILQ_INSERT_TAIL(counters, c, entry);
199 	}
200 }
201 
202 struct pfctl_status *
203 pfctl_get_status(int dev)
204 {
205 	struct pfctl_status	*status;
206 	nvlist_t	*nvl;
207 	size_t		 len;
208 	const void	*chksum;
209 
210 	status = calloc(1, sizeof(*status));
211 	if (status == NULL)
212 		return (NULL);
213 
214 	nvl = nvlist_create(0);
215 
216 	if (pfctl_do_ioctl(dev, DIOCGETSTATUSNV, 4096, &nvl)) {
217 		free(status);
218 		return (NULL);
219 	}
220 
221 	status->running = nvlist_get_bool(nvl, "running");
222 	status->since = nvlist_get_number(nvl, "since");
223 	status->debug = nvlist_get_number(nvl, "debug");
224 	status->hostid = ntohl(nvlist_get_number(nvl, "hostid"));
225 	status->states = nvlist_get_number(nvl, "states");
226 	status->src_nodes = nvlist_get_number(nvl, "src_nodes");
227 	status->syncookies_active = nvlist_get_bool(nvl, "syncookies_active");
228 
229 	strlcpy(status->ifname, nvlist_get_string(nvl, "ifname"),
230 	    IFNAMSIZ);
231 	chksum = nvlist_get_binary(nvl, "chksum", &len);
232 	assert(len == PF_MD5_DIGEST_LENGTH);
233 	memcpy(status->pf_chksum, chksum, len);
234 
235 	_pfctl_get_status_counters(nvlist_get_nvlist(nvl, "counters"),
236 	    &status->counters);
237 	_pfctl_get_status_counters(nvlist_get_nvlist(nvl, "lcounters"),
238 	    &status->lcounters);
239 	_pfctl_get_status_counters(nvlist_get_nvlist(nvl, "fcounters"),
240 	    &status->fcounters);
241 	_pfctl_get_status_counters(nvlist_get_nvlist(nvl, "scounters"),
242 	    &status->scounters);
243 
244 	pf_nvuint_64_array(nvl, "pcounters", 2 * 2 * 3,
245 	    (uint64_t *)status->pcounters, NULL);
246 	pf_nvuint_64_array(nvl, "bcounters", 2 * 2,
247 	    (uint64_t *)status->bcounters, NULL);
248 
249 	nvlist_destroy(nvl);
250 
251 	return (status);
252 }
253 
254 void
255 pfctl_free_status(struct pfctl_status *status)
256 {
257 	struct pfctl_status_counter *c, *tmp;
258 
259 	TAILQ_FOREACH_SAFE(c, &status->counters, entry, tmp) {
260 		free(c->name);
261 		free(c);
262 	}
263 	TAILQ_FOREACH_SAFE(c, &status->lcounters, entry, tmp) {
264 		free(c->name);
265 		free(c);
266 	}
267 	TAILQ_FOREACH_SAFE(c, &status->fcounters, entry, tmp) {
268 		free(c->name);
269 		free(c);
270 	}
271 	TAILQ_FOREACH_SAFE(c, &status->scounters, entry, tmp) {
272 		free(c->name);
273 		free(c);
274 	}
275 
276 	free(status);
277 }
278 
279 static void
280 pfctl_nv_add_addr(nvlist_t *nvparent, const char *name,
281     const struct pf_addr *addr)
282 {
283 	nvlist_t *nvl = nvlist_create(0);
284 
285 	nvlist_add_binary(nvl, "addr", addr, sizeof(*addr));
286 
287 	nvlist_add_nvlist(nvparent, name, nvl);
288 	nvlist_destroy(nvl);
289 }
290 
291 static void
292 pf_nvaddr_to_addr(const nvlist_t *nvl, struct pf_addr *addr)
293 {
294 	size_t len;
295 	const void *data;
296 
297 	data = nvlist_get_binary(nvl, "addr", &len);
298 	assert(len == sizeof(struct pf_addr));
299 	memcpy(addr, data, len);
300 }
301 
302 static void
303 pfctl_nv_add_addr_wrap(nvlist_t *nvparent, const char *name,
304     const struct pf_addr_wrap *addr)
305 {
306 	nvlist_t *nvl = nvlist_create(0);
307 
308 	nvlist_add_number(nvl, "type", addr->type);
309 	nvlist_add_number(nvl, "iflags", addr->iflags);
310 	if (addr->type == PF_ADDR_DYNIFTL)
311 		nvlist_add_string(nvl, "ifname", addr->v.ifname);
312 	if (addr->type == PF_ADDR_TABLE)
313 		nvlist_add_string(nvl, "tblname", addr->v.tblname);
314 	pfctl_nv_add_addr(nvl, "addr", &addr->v.a.addr);
315 	pfctl_nv_add_addr(nvl, "mask", &addr->v.a.mask);
316 
317 	nvlist_add_nvlist(nvparent, name, nvl);
318 	nvlist_destroy(nvl);
319 }
320 
321 static void
322 pf_nvaddr_wrap_to_addr_wrap(const nvlist_t *nvl, struct pf_addr_wrap *addr)
323 {
324 	bzero(addr, sizeof(*addr));
325 
326 	addr->type = nvlist_get_number(nvl, "type");
327 	addr->iflags = nvlist_get_number(nvl, "iflags");
328 	if (addr->type == PF_ADDR_DYNIFTL) {
329 		strlcpy(addr->v.ifname, nvlist_get_string(nvl, "ifname"),
330 		    IFNAMSIZ);
331 		addr->p.dyncnt = nvlist_get_number(nvl, "dyncnt");
332 	}
333 	if (addr->type == PF_ADDR_TABLE) {
334 		strlcpy(addr->v.tblname, nvlist_get_string(nvl, "tblname"),
335 		    PF_TABLE_NAME_SIZE);
336 		addr->p.tblcnt = nvlist_get_number(nvl, "tblcnt");
337 	}
338 
339 	pf_nvaddr_to_addr(nvlist_get_nvlist(nvl, "addr"), &addr->v.a.addr);
340 	pf_nvaddr_to_addr(nvlist_get_nvlist(nvl, "mask"), &addr->v.a.mask);
341 }
342 
343 static void
344 pfctl_nv_add_rule_addr(nvlist_t *nvparent, const char *name,
345     const struct pf_rule_addr *addr)
346 {
347 	uint64_t ports[2];
348 	nvlist_t *nvl = nvlist_create(0);
349 
350 	pfctl_nv_add_addr_wrap(nvl, "addr", &addr->addr);
351 	ports[0] = addr->port[0];
352 	ports[1] = addr->port[1];
353 	nvlist_add_number_array(nvl, "port", ports, 2);
354 	nvlist_add_number(nvl, "neg", addr->neg);
355 	nvlist_add_number(nvl, "port_op", addr->port_op);
356 
357 	nvlist_add_nvlist(nvparent, name, nvl);
358 	nvlist_destroy(nvl);
359 }
360 
361 static void
362 pf_nvrule_addr_to_rule_addr(const nvlist_t *nvl, struct pf_rule_addr *addr)
363 {
364 	pf_nvaddr_wrap_to_addr_wrap(nvlist_get_nvlist(nvl, "addr"), &addr->addr);
365 
366 	pf_nvuint_16_array(nvl, "port", 2, addr->port, NULL);
367 	addr->neg = nvlist_get_number(nvl, "neg");
368 	addr->port_op = nvlist_get_number(nvl, "port_op");
369 }
370 
371 static void
372 pfctl_nv_add_mape(nvlist_t *nvparent, const char *name,
373     const struct pf_mape_portset *mape)
374 {
375 	nvlist_t *nvl = nvlist_create(0);
376 
377 	nvlist_add_number(nvl, "offset", mape->offset);
378 	nvlist_add_number(nvl, "psidlen", mape->psidlen);
379 	nvlist_add_number(nvl, "psid", mape->psid);
380 	nvlist_add_nvlist(nvparent, name, nvl);
381 	nvlist_destroy(nvl);
382 }
383 
384 static void
385 pfctl_nv_add_pool(nvlist_t *nvparent, const char *name,
386     const struct pfctl_pool *pool)
387 {
388 	uint64_t ports[2];
389 	nvlist_t *nvl = nvlist_create(0);
390 
391 	nvlist_add_binary(nvl, "key", &pool->key, sizeof(pool->key));
392 	pfctl_nv_add_addr(nvl, "counter", &pool->counter);
393 	nvlist_add_number(nvl, "tblidx", pool->tblidx);
394 
395 	ports[0] = pool->proxy_port[0];
396 	ports[1] = pool->proxy_port[1];
397 	nvlist_add_number_array(nvl, "proxy_port", ports, 2);
398 	nvlist_add_number(nvl, "opts", pool->opts);
399 	pfctl_nv_add_mape(nvl, "mape", &pool->mape);
400 
401 	nvlist_add_nvlist(nvparent, name, nvl);
402 	nvlist_destroy(nvl);
403 }
404 
405 static void
406 pf_nvmape_to_mape(const nvlist_t *nvl, struct pf_mape_portset *mape)
407 {
408 	mape->offset = nvlist_get_number(nvl, "offset");
409 	mape->psidlen = nvlist_get_number(nvl, "psidlen");
410 	mape->psid = nvlist_get_number(nvl, "psid");
411 }
412 
413 static void
414 pf_nvpool_to_pool(const nvlist_t *nvl, struct pfctl_pool *pool)
415 {
416 	size_t len;
417 	const void *data;
418 
419 	data = nvlist_get_binary(nvl, "key", &len);
420 	assert(len == sizeof(pool->key));
421 	memcpy(&pool->key, data, len);
422 
423 	pf_nvaddr_to_addr(nvlist_get_nvlist(nvl, "counter"), &pool->counter);
424 
425 	pool->tblidx = nvlist_get_number(nvl, "tblidx");
426 	pf_nvuint_16_array(nvl, "proxy_port", 2, pool->proxy_port, NULL);
427 	pool->opts = nvlist_get_number(nvl, "opts");
428 
429 	if (nvlist_exists_nvlist(nvl, "mape"))
430 		pf_nvmape_to_mape(nvlist_get_nvlist(nvl, "mape"), &pool->mape);
431 }
432 
433 static void
434 pfctl_nv_add_uid(nvlist_t *nvparent, const char *name,
435     const struct pf_rule_uid *uid)
436 {
437 	uint64_t uids[2];
438 	nvlist_t *nvl = nvlist_create(0);
439 
440 	uids[0] = uid->uid[0];
441 	uids[1] = uid->uid[1];
442 	nvlist_add_number_array(nvl, "uid", uids, 2);
443 	nvlist_add_number(nvl, "op", uid->op);
444 
445 	nvlist_add_nvlist(nvparent, name, nvl);
446 	nvlist_destroy(nvl);
447 }
448 
449 static void
450 pf_nvrule_uid_to_rule_uid(const nvlist_t *nvl, struct pf_rule_uid *uid)
451 {
452 	pf_nvuint_32_array(nvl, "uid", 2, uid->uid, NULL);
453 	uid->op = nvlist_get_number(nvl, "op");
454 }
455 
456 static void
457 pfctl_nv_add_divert(nvlist_t *nvparent, const char *name,
458     const struct pfctl_rule *r)
459 {
460 	nvlist_t *nvl = nvlist_create(0);
461 
462 	pfctl_nv_add_addr(nvl, "addr", &r->divert.addr);
463 	nvlist_add_number(nvl, "port", r->divert.port);
464 
465 	nvlist_add_nvlist(nvparent, name, nvl);
466 	nvlist_destroy(nvl);
467 }
468 
469 static void
470 pf_nvdivert_to_divert(const nvlist_t *nvl, struct pfctl_rule *rule)
471 {
472 	pf_nvaddr_to_addr(nvlist_get_nvlist(nvl, "addr"), &rule->divert.addr);
473 	rule->divert.port = nvlist_get_number(nvl, "port");
474 }
475 
476 static void
477 pf_nvrule_to_rule(const nvlist_t *nvl, struct pfctl_rule *rule)
478 {
479 	const uint64_t *skip;
480 	const char *const *labels;
481 	size_t skipcount, labelcount;
482 
483 	rule->nr = nvlist_get_number(nvl, "nr");
484 
485 	pf_nvrule_addr_to_rule_addr(nvlist_get_nvlist(nvl, "src"), &rule->src);
486 	pf_nvrule_addr_to_rule_addr(nvlist_get_nvlist(nvl, "dst"), &rule->dst);
487 
488 	skip = nvlist_get_number_array(nvl, "skip", &skipcount);
489 	assert(skip);
490 	assert(skipcount == PF_SKIP_COUNT);
491 	for (int i = 0; i < PF_SKIP_COUNT; i++)
492 		rule->skip[i].nr = skip[i];
493 
494 	labels = nvlist_get_string_array(nvl, "labels", &labelcount);
495 	assert(labelcount <= PF_RULE_MAX_LABEL_COUNT);
496 	for (size_t i = 0; i < labelcount; i++)
497 		strlcpy(rule->label[i], labels[i], PF_RULE_LABEL_SIZE);
498 	rule->ridentifier = nvlist_get_number(nvl, "ridentifier");
499 	strlcpy(rule->ifname, nvlist_get_string(nvl, "ifname"), IFNAMSIZ);
500 	strlcpy(rule->qname, nvlist_get_string(nvl, "qname"), PF_QNAME_SIZE);
501 	strlcpy(rule->pqname, nvlist_get_string(nvl, "pqname"), PF_QNAME_SIZE);
502 	strlcpy(rule->tagname, nvlist_get_string(nvl, "tagname"),
503 	    PF_TAG_NAME_SIZE);
504 	strlcpy(rule->match_tagname, nvlist_get_string(nvl, "match_tagname"),
505 	    PF_TAG_NAME_SIZE);
506 
507 	strlcpy(rule->overload_tblname, nvlist_get_string(nvl, "overload_tblname"),
508 	    PF_TABLE_NAME_SIZE);
509 
510 	pf_nvpool_to_pool(nvlist_get_nvlist(nvl, "rpool"), &rule->rpool);
511 
512 	rule->evaluations = nvlist_get_number(nvl, "evaluations");
513 	pf_nvuint_64_array(nvl, "packets", 2, rule->packets, NULL);
514 	pf_nvuint_64_array(nvl, "bytes", 2, rule->bytes, NULL);
515 
516 	if (nvlist_exists_number(nvl, "timestamp")) {
517 		rule->last_active_timestamp = nvlist_get_number(nvl, "timestamp");
518 	}
519 
520 	rule->os_fingerprint = nvlist_get_number(nvl, "os_fingerprint");
521 
522 	rule->rtableid = nvlist_get_number(nvl, "rtableid");
523 	pf_nvuint_32_array(nvl, "timeout", PFTM_MAX, rule->timeout, NULL);
524 	rule->max_states = nvlist_get_number(nvl, "max_states");
525 	rule->max_src_nodes = nvlist_get_number(nvl, "max_src_nodes");
526 	rule->max_src_states = nvlist_get_number(nvl, "max_src_states");
527 	rule->max_src_conn = nvlist_get_number(nvl, "max_src_conn");
528 	rule->max_src_conn_rate.limit =
529 	    nvlist_get_number(nvl, "max_src_conn_rate.limit");
530 	rule->max_src_conn_rate.seconds =
531 	    nvlist_get_number(nvl, "max_src_conn_rate.seconds");
532 	rule->qid = nvlist_get_number(nvl, "qid");
533 	rule->pqid = nvlist_get_number(nvl, "pqid");
534 	rule->dnpipe = nvlist_get_number(nvl, "dnpipe");
535 	rule->dnrpipe = nvlist_get_number(nvl, "dnrpipe");
536 	rule->free_flags = nvlist_get_number(nvl, "dnflags");
537 	rule->prob = nvlist_get_number(nvl, "prob");
538 	rule->cuid = nvlist_get_number(nvl, "cuid");
539 	rule->cpid = nvlist_get_number(nvl, "cpid");
540 
541 	rule->return_icmp = nvlist_get_number(nvl, "return_icmp");
542 	rule->return_icmp6 = nvlist_get_number(nvl, "return_icmp6");
543 	rule->max_mss = nvlist_get_number(nvl, "max_mss");
544 	rule->scrub_flags = nvlist_get_number(nvl, "scrub_flags");
545 
546 	pf_nvrule_uid_to_rule_uid(nvlist_get_nvlist(nvl, "uid"), &rule->uid);
547 	pf_nvrule_uid_to_rule_uid(nvlist_get_nvlist(nvl, "gid"),
548 	    (struct pf_rule_uid *)&rule->gid);
549 
550 	rule->rule_flag = nvlist_get_number(nvl, "rule_flag");
551 	rule->action = nvlist_get_number(nvl, "action");
552 	rule->direction = nvlist_get_number(nvl, "direction");
553 	rule->log = nvlist_get_number(nvl, "log");
554 	rule->logif = nvlist_get_number(nvl, "logif");
555 	rule->quick = nvlist_get_number(nvl, "quick");
556 	rule->ifnot = nvlist_get_number(nvl, "ifnot");
557 	rule->match_tag_not = nvlist_get_number(nvl, "match_tag_not");
558 	rule->natpass = nvlist_get_number(nvl, "natpass");
559 
560 	rule->keep_state = nvlist_get_number(nvl, "keep_state");
561 	rule->af = nvlist_get_number(nvl, "af");
562 	rule->proto = nvlist_get_number(nvl, "proto");
563 	rule->type = nvlist_get_number(nvl, "type");
564 	rule->code = nvlist_get_number(nvl, "code");
565 	rule->flags = nvlist_get_number(nvl, "flags");
566 	rule->flagset = nvlist_get_number(nvl, "flagset");
567 	rule->min_ttl = nvlist_get_number(nvl, "min_ttl");
568 	rule->allow_opts = nvlist_get_number(nvl, "allow_opts");
569 	rule->rt = nvlist_get_number(nvl, "rt");
570 	rule->return_ttl  = nvlist_get_number(nvl, "return_ttl");
571 	rule->tos = nvlist_get_number(nvl, "tos");
572 	rule->set_tos = nvlist_get_number(nvl, "set_tos");
573 	rule->anchor_relative = nvlist_get_number(nvl, "anchor_relative");
574 	rule->anchor_wildcard = nvlist_get_number(nvl, "anchor_wildcard");
575 
576 	rule->flush = nvlist_get_number(nvl, "flush");
577 	rule->prio = nvlist_get_number(nvl, "prio");
578 	pf_nvuint_8_array(nvl, "set_prio", 2, rule->set_prio, NULL);
579 
580 	pf_nvdivert_to_divert(nvlist_get_nvlist(nvl, "divert"), rule);
581 
582 	rule->states_cur = nvlist_get_number(nvl, "states_cur");
583 	rule->states_tot = nvlist_get_number(nvl, "states_tot");
584 	rule->src_nodes = nvlist_get_number(nvl, "src_nodes");
585 }
586 
587 static void
588 pfctl_nveth_addr_to_eth_addr(const nvlist_t *nvl, struct pfctl_eth_addr *addr)
589 {
590 	static const u_int8_t EMPTY_MAC[ETHER_ADDR_LEN] = { 0 };
591 	size_t len;
592 	const void *data;
593 
594 	data = nvlist_get_binary(nvl, "addr", &len);
595 	assert(len == sizeof(addr->addr));
596 	memcpy(addr->addr, data, sizeof(addr->addr));
597 
598 	data = nvlist_get_binary(nvl, "mask", &len);
599 	assert(len == sizeof(addr->mask));
600 	memcpy(addr->mask, data, sizeof(addr->mask));
601 
602 	addr->neg = nvlist_get_bool(nvl, "neg");
603 
604 	/* To make checks for 'is this address set?' easier. */
605 	addr->isset = memcmp(addr->addr, EMPTY_MAC, ETHER_ADDR_LEN) != 0;
606 }
607 
608 static nvlist_t *
609 pfctl_eth_addr_to_nveth_addr(const struct pfctl_eth_addr *addr)
610 {
611 	nvlist_t *nvl;
612 
613 	nvl = nvlist_create(0);
614 	if (nvl == NULL)
615 		return (NULL);
616 
617 	nvlist_add_bool(nvl, "neg", addr->neg);
618 	nvlist_add_binary(nvl, "addr", &addr->addr, ETHER_ADDR_LEN);
619 	nvlist_add_binary(nvl, "mask", &addr->mask, ETHER_ADDR_LEN);
620 
621 	return (nvl);
622 }
623 
624 static void
625 pfctl_nveth_rule_to_eth_rule(const nvlist_t *nvl, struct pfctl_eth_rule *rule)
626 {
627 	rule->nr = nvlist_get_number(nvl, "nr");
628 	rule->quick = nvlist_get_bool(nvl, "quick");
629 	strlcpy(rule->ifname, nvlist_get_string(nvl, "ifname"), IFNAMSIZ);
630 	rule->ifnot = nvlist_get_bool(nvl, "ifnot");
631 	rule->direction = nvlist_get_number(nvl, "direction");
632 	rule->proto = nvlist_get_number(nvl, "proto");
633 	strlcpy(rule->match_tagname, nvlist_get_string(nvl, "match_tagname"),
634 	    PF_TAG_NAME_SIZE);
635 	rule->match_tag = nvlist_get_number(nvl, "match_tag");
636 	rule->match_tag_not = nvlist_get_bool(nvl, "match_tag_not");
637 
638 	pfctl_nveth_addr_to_eth_addr(nvlist_get_nvlist(nvl, "src"),
639 	    &rule->src);
640 	pfctl_nveth_addr_to_eth_addr(nvlist_get_nvlist(nvl, "dst"),
641 	    &rule->dst);
642 
643 	pf_nvrule_addr_to_rule_addr(nvlist_get_nvlist(nvl, "ipsrc"),
644 	    &rule->ipsrc);
645 	pf_nvrule_addr_to_rule_addr(nvlist_get_nvlist(nvl, "ipdst"),
646 	    &rule->ipdst);
647 
648 	rule->evaluations = nvlist_get_number(nvl, "evaluations");
649 	rule->packets[0] = nvlist_get_number(nvl, "packets-in");
650 	rule->packets[1] = nvlist_get_number(nvl, "packets-out");
651 	rule->bytes[0] = nvlist_get_number(nvl, "bytes-in");
652 	rule->bytes[1] = nvlist_get_number(nvl, "bytes-out");
653 
654 	if (nvlist_exists_number(nvl, "timestamp")) {
655 		rule->last_active_timestamp = nvlist_get_number(nvl, "timestamp");
656 	}
657 
658 	strlcpy(rule->qname, nvlist_get_string(nvl, "qname"), PF_QNAME_SIZE);
659 	strlcpy(rule->tagname, nvlist_get_string(nvl, "tagname"),
660 	    PF_TAG_NAME_SIZE);
661 
662 	rule->dnpipe = nvlist_get_number(nvl, "dnpipe");
663 	rule->dnflags = nvlist_get_number(nvl, "dnflags");
664 
665 	rule->anchor_relative = nvlist_get_number(nvl, "anchor_relative");
666 	rule->anchor_wildcard = nvlist_get_number(nvl, "anchor_wildcard");
667 
668 	rule->action = nvlist_get_number(nvl, "action");
669 }
670 
671 int
672 pfctl_get_eth_rulesets_info(int dev, struct pfctl_eth_rulesets_info *ri,
673     const char *path)
674 {
675 	nvlist_t *nvl;
676 	int ret;
677 
678 	bzero(ri, sizeof(*ri));
679 
680 	nvl = nvlist_create(0);
681 	nvlist_add_string(nvl, "path", path);
682 
683 	if ((ret = pfctl_do_ioctl(dev, DIOCGETETHRULESETS, 256, &nvl)) != 0)
684 		return (ret);
685 
686 	ri->nr = nvlist_get_number(nvl, "nr");
687 
688 	nvlist_destroy(nvl);
689 	return (0);
690 }
691 
692 int
693 pfctl_get_eth_ruleset(int dev, const char *path, int nr,
694     struct pfctl_eth_ruleset_info *ri)
695 {
696 	nvlist_t *nvl;
697 	int ret;
698 
699 	bzero(ri, sizeof(*ri));
700 
701 	nvl = nvlist_create(0);
702 	nvlist_add_string(nvl, "path", path);
703 	nvlist_add_number(nvl, "nr", nr);
704 
705 	if ((ret = pfctl_do_ioctl(dev, DIOCGETETHRULESET, 1024, &nvl)) != 0)
706 		return (ret);
707 
708 	ri->nr = nvlist_get_number(nvl, "nr");
709 	strlcpy(ri->path, nvlist_get_string(nvl, "path"), MAXPATHLEN);
710 	strlcpy(ri->name, nvlist_get_string(nvl, "name"),
711 	    PF_ANCHOR_NAME_SIZE);
712 
713 	return (0);
714 }
715 
716 int
717 pfctl_get_eth_rules_info(int dev, struct pfctl_eth_rules_info *rules,
718     const char *path)
719 {
720 	nvlist_t *nvl;
721 	int ret;
722 
723 	bzero(rules, sizeof(*rules));
724 
725 	nvl = nvlist_create(0);
726 	nvlist_add_string(nvl, "anchor", path);
727 
728 	if ((ret = pfctl_do_ioctl(dev, DIOCGETETHRULES, 1024, &nvl)) != 0)
729 		return (ret);
730 
731 	rules->nr = nvlist_get_number(nvl, "nr");
732 	rules->ticket = nvlist_get_number(nvl, "ticket");
733 
734 	nvlist_destroy(nvl);
735 	return (0);
736 }
737 
738 int
739 pfctl_get_eth_rule(int dev, uint32_t nr, uint32_t ticket,
740     const char *path, struct pfctl_eth_rule *rule, bool clear,
741     char *anchor_call)
742 {
743 	nvlist_t *nvl;
744 	int ret;
745 
746 	nvl = nvlist_create(0);
747 
748 	nvlist_add_string(nvl, "anchor", path);
749 	nvlist_add_number(nvl, "ticket", ticket);
750 	nvlist_add_number(nvl, "nr", nr);
751 	nvlist_add_bool(nvl, "clear", clear);
752 
753 	if ((ret = pfctl_do_ioctl(dev, DIOCGETETHRULE, 4096, &nvl)) != 0)
754 		return (ret);
755 
756 	pfctl_nveth_rule_to_eth_rule(nvl, rule);
757 
758 	if (anchor_call)
759 		strlcpy(anchor_call, nvlist_get_string(nvl, "anchor_call"),
760 		    MAXPATHLEN);
761 
762 	nvlist_destroy(nvl);
763 	return (0);
764 }
765 
766 int
767 pfctl_add_eth_rule(int dev, const struct pfctl_eth_rule *r, const char *anchor,
768     const char *anchor_call, uint32_t ticket)
769 {
770 	struct pfioc_nv nv;
771 	nvlist_t *nvl, *addr;
772 	void *packed;
773 	int error = 0;
774 	size_t size;
775 
776 	nvl = nvlist_create(0);
777 
778 	nvlist_add_number(nvl, "ticket", ticket);
779 	nvlist_add_string(nvl, "anchor", anchor);
780 	nvlist_add_string(nvl, "anchor_call", anchor_call);
781 
782 	nvlist_add_number(nvl, "nr", r->nr);
783 	nvlist_add_bool(nvl, "quick", r->quick);
784 	nvlist_add_string(nvl, "ifname", r->ifname);
785 	nvlist_add_bool(nvl, "ifnot", r->ifnot);
786 	nvlist_add_number(nvl, "direction", r->direction);
787 	nvlist_add_number(nvl, "proto", r->proto);
788 	nvlist_add_string(nvl, "match_tagname", r->match_tagname);
789 	nvlist_add_bool(nvl, "match_tag_not", r->match_tag_not);
790 
791 	addr = pfctl_eth_addr_to_nveth_addr(&r->src);
792 	if (addr == NULL) {
793 		nvlist_destroy(nvl);
794 		return (ENOMEM);
795 	}
796 	nvlist_add_nvlist(nvl, "src", addr);
797 	nvlist_destroy(addr);
798 
799 	addr = pfctl_eth_addr_to_nveth_addr(&r->dst);
800 	if (addr == NULL) {
801 		nvlist_destroy(nvl);
802 		return (ENOMEM);
803 	}
804 	nvlist_add_nvlist(nvl, "dst", addr);
805 	nvlist_destroy(addr);
806 
807 	pfctl_nv_add_rule_addr(nvl, "ipsrc", &r->ipsrc);
808 	pfctl_nv_add_rule_addr(nvl, "ipdst", &r->ipdst);
809 
810 	nvlist_add_string(nvl, "qname", r->qname);
811 	nvlist_add_string(nvl, "tagname", r->tagname);
812 	nvlist_add_number(nvl, "dnpipe", r->dnpipe);
813 	nvlist_add_number(nvl, "dnflags", r->dnflags);
814 
815 	nvlist_add_number(nvl, "action", r->action);
816 
817 	packed = nvlist_pack(nvl, &size);
818 	if (packed == NULL) {
819 		nvlist_destroy(nvl);
820 		return (ENOMEM);
821 	}
822 
823 	nv.len = size;
824 	nv.size = size;
825 	nv.data = packed;
826 
827 	if (ioctl(dev, DIOCADDETHRULE, &nv) != 0)
828 		error = errno;
829 
830 	free(packed);
831 	nvlist_destroy(nvl);
832 
833 	return (error);
834 }
835 
836 int
837 pfctl_add_rule(int dev, const struct pfctl_rule *r, const char *anchor,
838     const char *anchor_call, uint32_t ticket, uint32_t pool_ticket)
839 {
840 	struct pfioc_nv nv;
841 	uint64_t timeouts[PFTM_MAX];
842 	uint64_t set_prio[2];
843 	nvlist_t *nvl, *nvlr;
844 	size_t labelcount;
845 	int ret;
846 
847 	nvl = nvlist_create(0);
848 	nvlr = nvlist_create(0);
849 
850 	nvlist_add_number(nvl, "ticket", ticket);
851 	nvlist_add_number(nvl, "pool_ticket", pool_ticket);
852 	nvlist_add_string(nvl, "anchor", anchor);
853 	nvlist_add_string(nvl, "anchor_call", anchor_call);
854 
855 	nvlist_add_number(nvlr, "nr", r->nr);
856 	pfctl_nv_add_rule_addr(nvlr, "src", &r->src);
857 	pfctl_nv_add_rule_addr(nvlr, "dst", &r->dst);
858 
859 	labelcount = 0;
860 	while (r->label[labelcount][0] != 0 &&
861 	    labelcount < PF_RULE_MAX_LABEL_COUNT) {
862 		nvlist_append_string_array(nvlr, "labels",
863 		    r->label[labelcount]);
864 		labelcount++;
865 	}
866 	nvlist_add_number(nvlr, "ridentifier", r->ridentifier);
867 
868 	nvlist_add_string(nvlr, "ifname", r->ifname);
869 	nvlist_add_string(nvlr, "qname", r->qname);
870 	nvlist_add_string(nvlr, "pqname", r->pqname);
871 	nvlist_add_string(nvlr, "tagname", r->tagname);
872 	nvlist_add_string(nvlr, "match_tagname", r->match_tagname);
873 	nvlist_add_string(nvlr, "overload_tblname", r->overload_tblname);
874 
875 	pfctl_nv_add_pool(nvlr, "rpool", &r->rpool);
876 
877 	nvlist_add_number(nvlr, "os_fingerprint", r->os_fingerprint);
878 
879 	nvlist_add_number(nvlr, "rtableid", r->rtableid);
880 	for (int i = 0; i < PFTM_MAX; i++)
881 		timeouts[i] = r->timeout[i];
882 	nvlist_add_number_array(nvlr, "timeout", timeouts, PFTM_MAX);
883 	nvlist_add_number(nvlr, "max_states", r->max_states);
884 	nvlist_add_number(nvlr, "max_src_nodes", r->max_src_nodes);
885 	nvlist_add_number(nvlr, "max_src_states", r->max_src_states);
886 	nvlist_add_number(nvlr, "max_src_conn", r->max_src_conn);
887 	nvlist_add_number(nvlr, "max_src_conn_rate.limit",
888 	    r->max_src_conn_rate.limit);
889 	nvlist_add_number(nvlr, "max_src_conn_rate.seconds",
890 	    r->max_src_conn_rate.seconds);
891 	nvlist_add_number(nvlr, "dnpipe", r->dnpipe);
892 	nvlist_add_number(nvlr, "dnrpipe", r->dnrpipe);
893 	nvlist_add_number(nvlr, "dnflags", r->free_flags);
894 	nvlist_add_number(nvlr, "prob", r->prob);
895 	nvlist_add_number(nvlr, "cuid", r->cuid);
896 	nvlist_add_number(nvlr, "cpid", r->cpid);
897 
898 	nvlist_add_number(nvlr, "return_icmp", r->return_icmp);
899 	nvlist_add_number(nvlr, "return_icmp6", r->return_icmp6);
900 
901 	nvlist_add_number(nvlr, "max_mss", r->max_mss);
902 	nvlist_add_number(nvlr, "scrub_flags", r->scrub_flags);
903 
904 	pfctl_nv_add_uid(nvlr, "uid", &r->uid);
905 	pfctl_nv_add_uid(nvlr, "gid", (const struct pf_rule_uid *)&r->gid);
906 
907 	nvlist_add_number(nvlr, "rule_flag", r->rule_flag);
908 	nvlist_add_number(nvlr, "action", r->action);
909 	nvlist_add_number(nvlr, "direction", r->direction);
910 	nvlist_add_number(nvlr, "log", r->log);
911 	nvlist_add_number(nvlr, "logif", r->logif);
912 	nvlist_add_number(nvlr, "quick", r->quick);
913 	nvlist_add_number(nvlr, "ifnot", r->ifnot);
914 	nvlist_add_number(nvlr, "match_tag_not", r->match_tag_not);
915 	nvlist_add_number(nvlr, "natpass", r->natpass);
916 
917 	nvlist_add_number(nvlr, "keep_state", r->keep_state);
918 	nvlist_add_number(nvlr, "af", r->af);
919 	nvlist_add_number(nvlr, "proto", r->proto);
920 	nvlist_add_number(nvlr, "type", r->type);
921 	nvlist_add_number(nvlr, "code", r->code);
922 	nvlist_add_number(nvlr, "flags", r->flags);
923 	nvlist_add_number(nvlr, "flagset", r->flagset);
924 	nvlist_add_number(nvlr, "min_ttl", r->min_ttl);
925 	nvlist_add_number(nvlr, "allow_opts", r->allow_opts);
926 	nvlist_add_number(nvlr, "rt", r->rt);
927 	nvlist_add_number(nvlr, "return_ttl", r->return_ttl);
928 	nvlist_add_number(nvlr, "tos", r->tos);
929 	nvlist_add_number(nvlr, "set_tos", r->set_tos);
930 	nvlist_add_number(nvlr, "anchor_relative", r->anchor_relative);
931 	nvlist_add_number(nvlr, "anchor_wildcard", r->anchor_wildcard);
932 
933 	nvlist_add_number(nvlr, "flush", r->flush);
934 
935 	nvlist_add_number(nvlr, "prio", r->prio);
936 	set_prio[0] = r->set_prio[0];
937 	set_prio[1] = r->set_prio[1];
938 	nvlist_add_number_array(nvlr, "set_prio", set_prio, 2);
939 
940 	pfctl_nv_add_divert(nvlr, "divert", r);
941 
942 	nvlist_add_nvlist(nvl, "rule", nvlr);
943 	nvlist_destroy(nvlr);
944 
945 	/* Now do the call. */
946 	nv.data = nvlist_pack(nvl, &nv.len);
947 	nv.size = nv.len;
948 
949 	ret = ioctl(dev, DIOCADDRULENV, &nv);
950 	if (ret == -1)
951 		ret = errno;
952 
953 	free(nv.data);
954 	nvlist_destroy(nvl);
955 
956 	return (ret);
957 }
958 
959 int
960 pfctl_get_rules_info(int dev, struct pfctl_rules_info *rules, uint32_t ruleset,
961     const char *path)
962 {
963 	struct pfioc_rule pr;
964 	int ret;
965 
966 	bzero(&pr, sizeof(pr));
967 	if (strlcpy(pr.anchor, path, sizeof(pr.anchor)) >= sizeof(pr.anchor))
968 		return (E2BIG);
969 
970 	pr.rule.action = ruleset;
971 	ret = ioctl(dev, DIOCGETRULES, &pr);
972 	if (ret != 0)
973 		return (ret);
974 
975 	rules->nr = pr.nr;
976 	rules->ticket = pr.ticket;
977 
978 	return (0);
979 }
980 
981 int
982 pfctl_get_rule(int dev, uint32_t nr, uint32_t ticket, const char *anchor,
983     uint32_t ruleset, struct pfctl_rule *rule, char *anchor_call)
984 {
985 	return (pfctl_get_clear_rule(dev, nr, ticket, anchor, ruleset, rule,
986 	    anchor_call, false));
987 }
988 
989 int	pfctl_get_clear_rule(int dev, uint32_t nr, uint32_t ticket,
990 	    const char *anchor, uint32_t ruleset, struct pfctl_rule *rule,
991 	    char *anchor_call, bool clear)
992 {
993 	nvlist_t *nvl;
994 	int ret;
995 
996 	nvl = nvlist_create(0);
997 	if (nvl == 0)
998 		return (ENOMEM);
999 
1000 	nvlist_add_number(nvl, "nr", nr);
1001 	nvlist_add_number(nvl, "ticket", ticket);
1002 	nvlist_add_string(nvl, "anchor", anchor);
1003 	nvlist_add_number(nvl, "ruleset", ruleset);
1004 
1005 	if (clear)
1006 		nvlist_add_bool(nvl, "clear_counter", true);
1007 
1008 	if ((ret = pfctl_do_ioctl(dev, DIOCGETRULENV, 8192, &nvl)) != 0)
1009 		return (ret);
1010 
1011 	pf_nvrule_to_rule(nvlist_get_nvlist(nvl, "rule"), rule);
1012 
1013 	if (anchor_call)
1014 		strlcpy(anchor_call, nvlist_get_string(nvl, "anchor_call"),
1015 		    MAXPATHLEN);
1016 
1017 	nvlist_destroy(nvl);
1018 
1019 	return (0);
1020 }
1021 
1022 int
1023 pfctl_set_keepcounters(int dev, bool keep)
1024 {
1025 	struct pfioc_nv	 nv;
1026 	nvlist_t	*nvl;
1027 	int		 ret;
1028 
1029 	nvl = nvlist_create(0);
1030 
1031 	nvlist_add_bool(nvl, "keep_counters", keep);
1032 
1033 	nv.data = nvlist_pack(nvl, &nv.len);
1034 	nv.size = nv.len;
1035 
1036 	nvlist_destroy(nvl);
1037 
1038 	ret = ioctl(dev, DIOCKEEPCOUNTERS, &nv);
1039 
1040 	free(nv.data);
1041 	return (ret);
1042 }
1043 
1044 static void
1045 pfctl_nv_add_state_cmp(nvlist_t *nvl, const char *name,
1046     const struct pfctl_state_cmp *cmp)
1047 {
1048 	nvlist_t	*nv;
1049 
1050 	nv = nvlist_create(0);
1051 
1052 	nvlist_add_number(nv, "id", cmp->id);
1053 	nvlist_add_number(nv, "creatorid", htonl(cmp->creatorid));
1054 	nvlist_add_number(nv, "direction", cmp->direction);
1055 
1056 	nvlist_add_nvlist(nvl, name, nv);
1057 	nvlist_destroy(nv);
1058 }
1059 
1060 static void
1061 pf_state_key_export_to_state_key(struct pfctl_state_key *ps,
1062     const struct pf_state_key_export *s)
1063 {
1064 	bcopy(s->addr, ps->addr, sizeof(ps->addr[0]) * 2);
1065 	ps->port[0] = s->port[0];
1066 	ps->port[1] = s->port[1];
1067 }
1068 
1069 static void
1070 pf_state_peer_export_to_state_peer(struct pfctl_state_peer *ps,
1071     const struct pf_state_peer_export *s)
1072 {
1073 	/* Ignore scrub. */
1074 	ps->seqlo = s->seqlo;
1075 	ps->seqhi = s->seqhi;
1076 	ps->seqdiff = s->seqdiff;
1077 	/* Ignore max_win & mss */
1078 	ps->state = s->state;
1079 	ps->wscale = s->wscale;
1080 }
1081 
1082 static void
1083 pf_state_export_to_state(struct pfctl_state *ps, const struct pf_state_export *s)
1084 {
1085 	assert(s->version >= PF_STATE_VERSION);
1086 
1087 	ps->id = s->id;
1088 	strlcpy(ps->ifname, s->ifname, sizeof(ps->ifname));
1089 	strlcpy(ps->orig_ifname, s->orig_ifname, sizeof(ps->orig_ifname));
1090 	pf_state_key_export_to_state_key(&ps->key[0], &s->key[0]);
1091 	pf_state_key_export_to_state_key(&ps->key[1], &s->key[1]);
1092 	pf_state_peer_export_to_state_peer(&ps->src, &s->src);
1093 	pf_state_peer_export_to_state_peer(&ps->dst, &s->dst);
1094 	bcopy(&s->rt_addr, &ps->rt_addr, sizeof(ps->rt_addr));
1095 	ps->rule = ntohl(s->rule);
1096 	ps->anchor = ntohl(s->anchor);
1097 	ps->nat_rule = ntohl(s->nat_rule);
1098 	ps->creation = ntohl(s->creation);
1099 	ps->expire = ntohl(s->expire);
1100 	ps->packets[0] = s->packets[0];
1101 	ps->packets[1] = s->packets[1];
1102 	ps->bytes[0] = s->bytes[0];
1103 	ps->bytes[1] = s->bytes[1];
1104 	ps->creatorid = ntohl(s->creatorid);
1105 	ps->key[0].proto = s->proto;
1106 	ps->key[1].proto = s->proto;
1107 	ps->key[0].af = s->af;
1108 	ps->key[1].af = s->af;
1109 	ps->direction = s->direction;
1110 	ps->state_flags = s->state_flags;
1111 	ps->sync_flags = s->sync_flags;
1112 }
1113 
1114 int
1115 pfctl_get_states(int dev, struct pfctl_states *states)
1116 {
1117 	struct pfioc_states_v2 ps;
1118 	struct pf_state_export *p;
1119 	char *inbuf = NULL, *newinbuf = NULL;
1120 	unsigned int len = 0;
1121 	int i, error;
1122 
1123 	bzero(&ps, sizeof(ps));
1124 	ps.ps_req_version = PF_STATE_VERSION;
1125 
1126 	bzero(states, sizeof(*states));
1127 	TAILQ_INIT(&states->states);
1128 
1129 	for (;;) {
1130 		ps.ps_len = len;
1131 		if (len) {
1132 			newinbuf = realloc(inbuf, len);
1133 			if (newinbuf == NULL)
1134 				return (ENOMEM);
1135 			ps.ps_buf = inbuf = newinbuf;
1136 		}
1137 		if ((error = ioctl(dev, DIOCGETSTATESV2, &ps)) < 0) {
1138 			free(inbuf);
1139 			return (error);
1140 		}
1141 		if (ps.ps_len + sizeof(struct pfioc_states_v2) < len)
1142 			break;
1143 		if (len == 0 && ps.ps_len == 0)
1144 			goto out;
1145 		if (len == 0 && ps.ps_len != 0)
1146 			len = ps.ps_len;
1147 		if (ps.ps_len == 0)
1148 			goto out;      /* no states */
1149 		len *= 2;
1150 	}
1151 	p = ps.ps_states;
1152 
1153 	for (i = 0; i < ps.ps_len; i += sizeof(*p), p++) {
1154 		struct pfctl_state *s = malloc(sizeof(*s));
1155 		if (s == NULL) {
1156 			pfctl_free_states(states);
1157 			error = ENOMEM;
1158 			goto out;
1159 		}
1160 
1161 		pf_state_export_to_state(s, p);
1162 		TAILQ_INSERT_TAIL(&states->states, s, entry);
1163 	}
1164 
1165 out:
1166 	free(inbuf);
1167 	return (error);
1168 }
1169 
1170 void
1171 pfctl_free_states(struct pfctl_states *states)
1172 {
1173 	struct pfctl_state *s, *tmp;
1174 
1175 	TAILQ_FOREACH_SAFE(s, &states->states, entry, tmp) {
1176 		free(s);
1177 	}
1178 
1179 	bzero(states, sizeof(*states));
1180 }
1181 
1182 static int
1183 _pfctl_clear_states(int dev, const struct pfctl_kill *kill,
1184     unsigned int *killed, uint64_t ioctlval)
1185 {
1186 	struct pfioc_nv	 nv;
1187 	nvlist_t	*nvl;
1188 	int		 ret;
1189 
1190 	nvl = nvlist_create(0);
1191 
1192 	pfctl_nv_add_state_cmp(nvl, "cmp", &kill->cmp);
1193 	nvlist_add_number(nvl, "af", kill->af);
1194 	nvlist_add_number(nvl, "proto", kill->proto);
1195 	pfctl_nv_add_rule_addr(nvl, "src", &kill->src);
1196 	pfctl_nv_add_rule_addr(nvl, "dst", &kill->dst);
1197 	pfctl_nv_add_rule_addr(nvl, "rt_addr", &kill->rt_addr);
1198 	nvlist_add_string(nvl, "ifname", kill->ifname);
1199 	nvlist_add_string(nvl, "label", kill->label);
1200 	nvlist_add_bool(nvl, "kill_match", kill->kill_match);
1201 
1202 	if ((ret = pfctl_do_ioctl(dev, ioctlval, 1024, &nvl)) != 0)
1203 		return (ret);
1204 
1205 	if (killed)
1206 		*killed = nvlist_get_number(nvl, "killed");
1207 
1208 	nvlist_destroy(nvl);
1209 	free(nv.data);
1210 
1211 	return (ret);
1212 }
1213 
1214 int
1215 pfctl_clear_states(int dev, const struct pfctl_kill *kill,
1216     unsigned int *killed)
1217 {
1218 	return (_pfctl_clear_states(dev, kill, killed, DIOCCLRSTATESNV));
1219 }
1220 
1221 int
1222 pfctl_kill_states(int dev, const struct pfctl_kill *kill, unsigned int *killed)
1223 {
1224 	return (_pfctl_clear_states(dev, kill, killed, DIOCKILLSTATESNV));
1225 }
1226 
1227 int
1228 pfctl_clear_rules(int dev, const char *anchorname)
1229 {
1230 	struct pfioc_trans trans;
1231 	struct pfioc_trans_e transe[2];
1232 	int ret;
1233 
1234 	bzero(&trans, sizeof(trans));
1235 	bzero(&transe, sizeof(transe));
1236 
1237 	transe[0].rs_num = PF_RULESET_SCRUB;
1238 	if (strlcpy(transe[0].anchor, anchorname, sizeof(transe[0].anchor))
1239 	    >= sizeof(transe[0].anchor))
1240 		return (E2BIG);
1241 
1242 	transe[1].rs_num = PF_RULESET_FILTER;
1243 	if (strlcpy(transe[1].anchor, anchorname, sizeof(transe[1].anchor))
1244 	    >= sizeof(transe[1].anchor))
1245 		return (E2BIG);
1246 
1247 	trans.size = 2;
1248 	trans.esize = sizeof(transe[0]);
1249 	trans.array = transe;
1250 
1251 	ret = ioctl(dev, DIOCXBEGIN, &trans);
1252 	if (ret != 0)
1253 		return (ret);
1254 	return ioctl(dev, DIOCXCOMMIT, &trans);
1255 }
1256 
1257 int
1258 pfctl_clear_nat(int dev, const char *anchorname)
1259 {
1260 	struct pfioc_trans trans;
1261 	struct pfioc_trans_e transe[3];
1262 	int ret;
1263 
1264 	bzero(&trans, sizeof(trans));
1265 	bzero(&transe, sizeof(transe));
1266 
1267 	transe[0].rs_num = PF_RULESET_NAT;
1268 	if (strlcpy(transe[0].anchor, anchorname, sizeof(transe[0].anchor))
1269 	    >= sizeof(transe[0].anchor))
1270 		return (E2BIG);
1271 
1272 	transe[1].rs_num = PF_RULESET_BINAT;
1273 	if (strlcpy(transe[1].anchor, anchorname, sizeof(transe[1].anchor))
1274 	    >= sizeof(transe[0].anchor))
1275 		return (E2BIG);
1276 
1277 	transe[2].rs_num = PF_RULESET_RDR;
1278 	if (strlcpy(transe[2].anchor, anchorname, sizeof(transe[2].anchor))
1279 	    >= sizeof(transe[2].anchor))
1280 		return (E2BIG);
1281 
1282 	trans.size = 3;
1283 	trans.esize = sizeof(transe[0]);
1284 	trans.array = transe;
1285 
1286 	ret = ioctl(dev, DIOCXBEGIN, &trans);
1287 	if (ret != 0)
1288 		return (ret);
1289 	return ioctl(dev, DIOCXCOMMIT, &trans);
1290 }
1291 int
1292 pfctl_clear_eth_rules(int dev, const char *anchorname)
1293 {
1294 	struct pfioc_trans trans;
1295 	struct pfioc_trans_e transe;
1296 	int ret;
1297 
1298 	bzero(&trans, sizeof(trans));
1299 	bzero(&transe, sizeof(transe));
1300 
1301 	transe.rs_num = PF_RULESET_ETH;
1302 	if (strlcpy(transe.anchor, anchorname, sizeof(transe.anchor))
1303 	    >= sizeof(transe.anchor))
1304 		return (E2BIG);
1305 
1306 	trans.size = 1;
1307 	trans.esize = sizeof(transe);
1308 	trans.array = &transe;
1309 
1310 	ret = ioctl(dev, DIOCXBEGIN, &trans);
1311 	if (ret != 0)
1312 		return (ret);
1313 	return ioctl(dev, DIOCXCOMMIT, &trans);
1314 }
1315 
1316 static int
1317 pfctl_get_limit(int dev, const int index, uint *limit)
1318 {
1319 	struct pfioc_limit pl;
1320 
1321 	bzero(&pl, sizeof(pl));
1322 	pl.index = index;
1323 
1324 	if (ioctl(dev, DIOCGETLIMIT, &pl) == -1)
1325 		return (errno);
1326 
1327 	*limit = pl.limit;
1328 
1329 	return (0);
1330 }
1331 
1332 int
1333 pfctl_set_syncookies(int dev, const struct pfctl_syncookies *s)
1334 {
1335 	struct pfioc_nv	 nv;
1336 	nvlist_t	*nvl;
1337 	int		 ret;
1338 	uint		 state_limit;
1339 	uint64_t	 lim, hi, lo;
1340 
1341 	ret = pfctl_get_limit(dev, PF_LIMIT_STATES, &state_limit);
1342 	if (ret != 0)
1343 		return (ret);
1344 
1345 	lim = state_limit;
1346 	hi = lim * s->highwater / 100;
1347 	lo = lim * s->lowwater / 100;
1348 
1349 	if (lo == hi)
1350 		hi++;
1351 
1352 	nvl = nvlist_create(0);
1353 
1354 	nvlist_add_bool(nvl, "enabled", s->mode != PFCTL_SYNCOOKIES_NEVER);
1355 	nvlist_add_bool(nvl, "adaptive", s->mode == PFCTL_SYNCOOKIES_ADAPTIVE);
1356 	nvlist_add_number(nvl, "highwater", hi);
1357 	nvlist_add_number(nvl, "lowwater", lo);
1358 
1359 	nv.data = nvlist_pack(nvl, &nv.len);
1360 	nv.size = nv.len;
1361 	nvlist_destroy(nvl);
1362 	nvl = NULL;
1363 
1364 	ret = ioctl(dev, DIOCSETSYNCOOKIES, &nv);
1365 
1366 	free(nv.data);
1367 	return (ret);
1368 }
1369 
1370 int
1371 pfctl_get_syncookies(int dev, struct pfctl_syncookies *s)
1372 {
1373 	nvlist_t	*nvl;
1374 	int		 ret;
1375 	uint		 state_limit;
1376 	bool		 enabled, adaptive;
1377 
1378 	ret = pfctl_get_limit(dev, PF_LIMIT_STATES, &state_limit);
1379 	if (ret != 0)
1380 		return (ret);
1381 
1382 	bzero(s, sizeof(*s));
1383 
1384 	nvl = nvlist_create(0);
1385 
1386 	if ((ret = pfctl_do_ioctl(dev, DIOCGETSYNCOOKIES, 256, &nvl)) != 0)
1387 		return (errno);
1388 
1389 	enabled = nvlist_get_bool(nvl, "enabled");
1390 	adaptive = nvlist_get_bool(nvl, "adaptive");
1391 
1392 	if (enabled) {
1393 		if (adaptive)
1394 			s->mode = PFCTL_SYNCOOKIES_ADAPTIVE;
1395 		else
1396 			s->mode = PFCTL_SYNCOOKIES_ALWAYS;
1397 	} else {
1398 		s->mode = PFCTL_SYNCOOKIES_NEVER;
1399 	}
1400 
1401 	s->highwater = nvlist_get_number(nvl, "highwater") * 100 / state_limit;
1402 	s->lowwater = nvlist_get_number(nvl, "lowwater") * 100 / state_limit;
1403 
1404 	nvlist_destroy(nvl);
1405 
1406 	return (0);
1407 }
1408 
1409 int
1410 pfctl_table_add_addrs(int dev, struct pfr_table *tbl, struct pfr_addr
1411     *addr, int size, int *nadd, int flags)
1412 {
1413 	struct pfioc_table io;
1414 
1415 	if (tbl == NULL || size < 0 || (size && addr == NULL)) {
1416 		return (EINVAL);
1417 	}
1418 	bzero(&io, sizeof io);
1419 	io.pfrio_flags = flags;
1420 	io.pfrio_table = *tbl;
1421 	io.pfrio_buffer = addr;
1422 	io.pfrio_esize = sizeof(*addr);
1423 	io.pfrio_size = size;
1424 
1425 	if (ioctl(dev, DIOCRADDADDRS, &io))
1426 		return (errno);
1427 	if (nadd != NULL)
1428 		*nadd = io.pfrio_nadd;
1429 	return (0);
1430 }
1431 
1432 int
1433 pfctl_table_del_addrs(int dev, struct pfr_table *tbl, struct pfr_addr
1434     *addr, int size, int *ndel, int flags)
1435 {
1436 	struct pfioc_table io;
1437 
1438 	if (tbl == NULL || size < 0 || (size && addr == NULL)) {
1439 		return (EINVAL);
1440 	}
1441 	bzero(&io, sizeof io);
1442 	io.pfrio_flags = flags;
1443 	io.pfrio_table = *tbl;
1444 	io.pfrio_buffer = addr;
1445 	io.pfrio_esize = sizeof(*addr);
1446 	io.pfrio_size = size;
1447 
1448 	if (ioctl(dev, DIOCRDELADDRS, &io))
1449 		return (errno);
1450 	if (ndel != NULL)
1451 		*ndel = io.pfrio_ndel;
1452 	return (0);
1453 }
1454 
1455 int
1456 pfctl_table_set_addrs(int dev, struct pfr_table *tbl, struct pfr_addr
1457     *addr, int size, int *size2, int *nadd, int *ndel, int *nchange, int flags)
1458 {
1459 	struct pfioc_table io;
1460 
1461 	if (tbl == NULL || size < 0 || (size && addr == NULL)) {
1462 		return (EINVAL);
1463 	}
1464 	bzero(&io, sizeof io);
1465 	io.pfrio_flags = flags;
1466 	io.pfrio_table = *tbl;
1467 	io.pfrio_buffer = addr;
1468 	io.pfrio_esize = sizeof(*addr);
1469 	io.pfrio_size = size;
1470 	io.pfrio_size2 = (size2 != NULL) ? *size2 : 0;
1471 	if (ioctl(dev, DIOCRSETADDRS, &io))
1472 		return (-1);
1473 	if (nadd != NULL)
1474 		*nadd = io.pfrio_nadd;
1475 	if (ndel != NULL)
1476 		*ndel = io.pfrio_ndel;
1477 	if (nchange != NULL)
1478 		*nchange = io.pfrio_nchange;
1479 	if (size2 != NULL)
1480 		*size2 = io.pfrio_size2;
1481 	return (0);
1482 }
1483 
1484 int pfctl_table_get_addrs(int dev, struct pfr_table *tbl, struct pfr_addr *addr,
1485     int *size, int flags)
1486 {
1487 	struct pfioc_table io;
1488 
1489 	if (tbl == NULL || size == NULL || *size < 0 ||
1490 	    (*size && addr == NULL)) {
1491 		return (EINVAL);
1492 	}
1493 	bzero(&io, sizeof io);
1494 	io.pfrio_flags = flags;
1495 	io.pfrio_table = *tbl;
1496 	io.pfrio_buffer = addr;
1497 	io.pfrio_esize = sizeof(*addr);
1498 	io.pfrio_size = *size;
1499 	if (ioctl(dev, DIOCRGETADDRS, &io))
1500 		return (-1);
1501 	*size = io.pfrio_size;
1502 	return (0);
1503 }
1504