xref: /freebsd/lib/libpfctl/libpfctl.c (revision 9ae91f59)
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 
32 #include <sys/cdefs.h>
33 
34 #include <sys/ioctl.h>
35 #include <sys/nv.h>
36 #include <sys/queue.h>
37 #include <sys/types.h>
38 
39 #include <net/if.h>
40 #include <net/pfvar.h>
41 #include <netinet/in.h>
42 
43 #include <netpfil/pf/pf_nl.h>
44 #include <netlink/netlink.h>
45 #include <netlink/netlink_generic.h>
46 #include <netlink/netlink_snl.h>
47 #include <netlink/netlink_snl_generic.h>
48 #include <netlink/netlink_snl_route.h>
49 
50 #include <assert.h>
51 #include <err.h>
52 #include <errno.h>
53 #include <fcntl.h>
54 #include <stdlib.h>
55 #include <string.h>
56 
57 #include "libpfctl.h"
58 
59 struct pfctl_handle {
60 	int fd;
61 	struct snl_state ss;
62 };
63 
64 const char* PFCTL_SYNCOOKIES_MODE_NAMES[] = {
65 	"never",
66 	"always",
67 	"adaptive"
68 };
69 
70 static int	_pfctl_clear_states(int , const struct pfctl_kill *,
71 		    unsigned int *, uint64_t);
72 
73 struct pfctl_handle *
pfctl_open(const char * pf_device)74 pfctl_open(const char *pf_device)
75 {
76 	struct pfctl_handle *h;
77 
78 	h = calloc(1, sizeof(struct pfctl_handle));
79 	h->fd = -1;
80 
81 	h->fd = open(pf_device, O_RDWR);
82 	if (h->fd < 0)
83 		goto error;
84 
85 	if (!snl_init(&h->ss, NETLINK_GENERIC))
86 		goto error;
87 
88 	return (h);
89 error:
90 	close(h->fd);
91 	snl_free(&h->ss);
92 	free(h);
93 
94 	return (NULL);
95 }
96 
97 void
pfctl_close(struct pfctl_handle * h)98 pfctl_close(struct pfctl_handle *h)
99 {
100 	close(h->fd);
101 	snl_free(&h->ss);
102 	free(h);
103 }
104 
105 int
pfctl_fd(struct pfctl_handle * h)106 pfctl_fd(struct pfctl_handle *h)
107 {
108 	return (h->fd);
109 }
110 
111 static int
pfctl_do_netlink_cmd(struct pfctl_handle * h,uint cmd)112 pfctl_do_netlink_cmd(struct pfctl_handle *h, uint cmd)
113 {
114 	struct snl_errmsg_data e = {};
115 	struct snl_writer nw;
116 	struct nlmsghdr *hdr;
117 	uint32_t seq_id;
118 	int family_id;
119 
120 	family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
121 	if (family_id == 0)
122 		return (ENOTSUP);
123 
124 	snl_init_writer(&h->ss, &nw);
125 	hdr = snl_create_genl_msg_request(&nw, family_id, cmd);
126 
127 	hdr = snl_finalize_msg(&nw);
128 	if (hdr == NULL)
129 		return (ENOMEM);
130 	seq_id = hdr->nlmsg_seq;
131 
132 	snl_send_message(&h->ss, hdr);
133 
134 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
135 	}
136 
137 	return (e.error);
138 }
139 
140 static int
pfctl_do_ioctl(int dev,uint cmd,size_t size,nvlist_t ** nvl)141 pfctl_do_ioctl(int dev, uint cmd, size_t size, nvlist_t **nvl)
142 {
143 	struct pfioc_nv nv;
144 	void *data;
145 	size_t nvlen;
146 	int ret;
147 
148 	data = nvlist_pack(*nvl, &nvlen);
149 	if (nvlen > size)
150 		size = nvlen;
151 
152 retry:
153 	nv.data = malloc(size);
154 	if (nv.data == NULL) {
155 		ret = ENOMEM;
156 		goto out;
157 	}
158 
159 	memcpy(nv.data, data, nvlen);
160 
161 	nv.len = nvlen;
162 	nv.size = size;
163 
164 	ret = ioctl(dev, cmd, &nv);
165 	if (ret == -1 && errno == ENOSPC) {
166 		size *= 2;
167 		free(nv.data);
168 		goto retry;
169 	}
170 
171 	nvlist_destroy(*nvl);
172 	*nvl = NULL;
173 
174 	if (ret == 0) {
175 		*nvl = nvlist_unpack(nv.data, nv.len, 0);
176 		if (*nvl == NULL) {
177 			ret = EIO;
178 			goto out;
179 		}
180 	} else {
181 		ret = errno;
182 	}
183 
184 out:
185 	free(data);
186 	free(nv.data);
187 
188 	return (ret);
189 }
190 
191 static void
pf_nvuint_8_array(const nvlist_t * nvl,const char * name,size_t maxelems,uint8_t * numbers,size_t * nelems)192 pf_nvuint_8_array(const nvlist_t *nvl, const char *name, size_t maxelems,
193     uint8_t *numbers, size_t *nelems)
194 {
195 	const uint64_t *tmp;
196 	size_t elems;
197 
198 	tmp = nvlist_get_number_array(nvl, name, &elems);
199 	assert(elems <= maxelems);
200 
201 	for (size_t i = 0; i < elems; i++)
202 		numbers[i] = tmp[i];
203 
204 	if (nelems)
205 		*nelems = elems;
206 }
207 
208 static void
pf_nvuint_16_array(const nvlist_t * nvl,const char * name,size_t maxelems,uint16_t * numbers,size_t * nelems)209 pf_nvuint_16_array(const nvlist_t *nvl, const char *name, size_t maxelems,
210     uint16_t *numbers, size_t *nelems)
211 {
212 	const uint64_t *tmp;
213 	size_t elems;
214 
215 	tmp = nvlist_get_number_array(nvl, name, &elems);
216 	assert(elems <= maxelems);
217 
218 	for (size_t i = 0; i < elems; i++)
219 		numbers[i] = tmp[i];
220 
221 	if (nelems)
222 		*nelems = elems;
223 }
224 
225 static void
pf_nvuint_32_array(const nvlist_t * nvl,const char * name,size_t maxelems,uint32_t * numbers,size_t * nelems)226 pf_nvuint_32_array(const nvlist_t *nvl, const char *name, size_t maxelems,
227     uint32_t *numbers, size_t *nelems)
228 {
229 	const uint64_t *tmp;
230 	size_t elems;
231 
232 	tmp = nvlist_get_number_array(nvl, name, &elems);
233 
234 	for (size_t i = 0; i < elems && i < maxelems; i++)
235 		numbers[i] = tmp[i];
236 
237 	if (nelems)
238 		*nelems = elems;
239 }
240 
241 static void
pf_nvuint_64_array(const nvlist_t * nvl,const char * name,size_t maxelems,uint64_t * numbers,size_t * nelems)242 pf_nvuint_64_array(const nvlist_t *nvl, const char *name, size_t maxelems,
243     uint64_t *numbers, size_t *nelems)
244 {
245 	const uint64_t *tmp;
246 	size_t elems;
247 
248 	tmp = nvlist_get_number_array(nvl, name, &elems);
249 	assert(elems <= maxelems);
250 
251 	for (size_t i = 0; i < elems; i++)
252 		numbers[i] = tmp[i];
253 
254 	if (nelems)
255 		*nelems = elems;
256 }
257 
258 int
pfctl_startstop(struct pfctl_handle * h,int start)259 pfctl_startstop(struct pfctl_handle *h, int start)
260 {
261 	return (pfctl_do_netlink_cmd(h, start ? PFNL_CMD_START : PFNL_CMD_STOP));
262 }
263 
264 static void
_pfctl_get_status_counters(const nvlist_t * nvl,struct pfctl_status_counters * counters)265 _pfctl_get_status_counters(const nvlist_t *nvl,
266     struct pfctl_status_counters *counters)
267 {
268 	const uint64_t		*ids, *counts;
269 	const char *const	*names;
270 	size_t id_len, counter_len, names_len;
271 
272 	ids = nvlist_get_number_array(nvl, "ids", &id_len);
273 	counts = nvlist_get_number_array(nvl, "counters", &counter_len);
274 	names = nvlist_get_string_array(nvl, "names", &names_len);
275 	assert(id_len == counter_len);
276 	assert(counter_len == names_len);
277 
278 	TAILQ_INIT(counters);
279 
280 	for (size_t i = 0; i < id_len; i++) {
281 		struct pfctl_status_counter *c;
282 
283 		c = malloc(sizeof(*c));
284 		if (c == NULL)
285 			continue;
286 
287 		c->id = ids[i];
288 		c->counter = counts[i];
289 		c->name = strdup(names[i]);
290 
291 		TAILQ_INSERT_TAIL(counters, c, entry);
292 	}
293 }
294 
295 #define	_OUT(_field)	offsetof(struct pfctl_status_counter, _field)
296 static const struct snl_attr_parser ap_counter[] = {
297 	{ .type = PF_C_COUNTER, .off = _OUT(counter), .cb = snl_attr_get_uint64 },
298 	{ .type = PF_C_NAME, .off = _OUT(name), .cb = snl_attr_get_string },
299 	{ .type = PF_C_ID, .off = _OUT(id), .cb = snl_attr_get_uint32 },
300 };
301 SNL_DECLARE_ATTR_PARSER(counter_parser, ap_counter);
302 #undef _OUT
303 
304 static bool
snl_attr_get_counters(struct snl_state * ss,struct nlattr * nla,const void * arg __unused,void * target)305 snl_attr_get_counters(struct snl_state *ss, struct nlattr *nla,
306     const void *arg __unused, void *target)
307 {
308 	struct pfctl_status_counter counter = {};
309 	struct pfctl_status_counter *c;
310 	bool error;
311 
312 	error = snl_parse_header(ss, NLA_DATA(nla), NLA_DATA_LEN(nla), &counter_parser, &counter);
313 	if (! error)
314 		return (error);
315 
316 	c = malloc(sizeof(*c));
317 	if (c == NULL)
318 		return (false);
319 
320 	c->id = counter.id;
321 	c->counter = counter.counter;
322 	c->name = strdup(counter.name);
323 
324 	TAILQ_INSERT_TAIL((struct pfctl_status_counters *)target, c, entry);
325 
326 	return (error);
327 }
328 
329 struct snl_uint64_array {
330 	uint64_t *array;
331 	size_t count;
332 	size_t max;
333 };
334 static bool
snl_attr_get_uint64_element(struct snl_state * ss,struct nlattr * nla,const void * arg,void * target)335 snl_attr_get_uint64_element(struct snl_state *ss, struct nlattr *nla,
336     const void *arg, void *target)
337 {
338 	bool error;
339 	uint64_t value;
340 	struct snl_uint64_array *t = (struct snl_uint64_array *)target;
341 
342 	if (t->count >= t->max)
343 		return (false);
344 
345 	error = snl_attr_get_uint64(ss, nla, arg, &value);
346 	if (! error)
347 		return (error);
348 
349 	t->array[t->count++] = value;
350 
351 	return (true);
352 }
353 
354 static const struct snl_attr_parser ap_array[] = {
355 	{ .cb = snl_attr_get_uint64_element },
356 };
357 SNL_DECLARE_ATTR_PARSER(array_parser, ap_array);
358 static bool
snl_attr_get_uint64_array(struct snl_state * ss,struct nlattr * nla,const void * arg,void * target)359 snl_attr_get_uint64_array(struct snl_state *ss, struct nlattr *nla,
360     const void *arg, void *target)
361 {
362 	struct snl_uint64_array a = {
363 		.array = target,
364 		.count = 0,
365 		.max = (size_t)arg,
366 	};
367 	bool error;
368 
369 	error = snl_parse_header(ss, NLA_DATA(nla), NLA_DATA_LEN(nla), &array_parser, &a);
370 	if (! error)
371 		return (error);
372 
373 	return (true);
374 }
375 
376 #define	_OUT(_field)	offsetof(struct pfctl_status, _field)
377 static const struct snl_attr_parser ap_getstatus[] = {
378 	{ .type = PF_GS_IFNAME, .off = _OUT(ifname), .arg_u32 = IFNAMSIZ, .cb = snl_attr_copy_string },
379 	{ .type = PF_GS_RUNNING, .off = _OUT(running), .cb = snl_attr_get_bool },
380 	{ .type = PF_GS_SINCE, .off = _OUT(since), .cb = snl_attr_get_uint32 },
381 	{ .type = PF_GS_DEBUG, .off = _OUT(debug), .cb = snl_attr_get_uint32 },
382 	{ .type = PF_GS_HOSTID, .off = _OUT(hostid), .cb = snl_attr_get_uint32 },
383 	{ .type = PF_GS_STATES, .off = _OUT(states), .cb = snl_attr_get_uint32 },
384 	{ .type = PF_GS_SRC_NODES, .off = _OUT(src_nodes), .cb = snl_attr_get_uint32 },
385 	{ .type = PF_GS_REASSEMBLE, .off = _OUT(reass), .cb = snl_attr_get_uint32 },
386 	{ .type = PF_GS_SYNCOOKIES_ACTIVE, .off = _OUT(syncookies_active), .cb = snl_attr_get_uint32 },
387 	{ .type = PF_GS_COUNTERS, .off = _OUT(counters), .cb = snl_attr_get_counters },
388 	{ .type = PF_GS_LCOUNTERS, .off = _OUT(lcounters), .cb = snl_attr_get_counters },
389 	{ .type = PF_GS_FCOUNTERS, .off = _OUT(fcounters), .cb = snl_attr_get_counters },
390 	{ .type = PF_GS_SCOUNTERS, .off = _OUT(scounters), .cb = snl_attr_get_counters },
391 	{ .type = PF_GS_CHKSUM, .off = _OUT(pf_chksum), .arg_u32 = PF_MD5_DIGEST_LENGTH, .cb = snl_attr_get_bytes },
392 	{ .type = PF_GS_BCOUNTERS, .off = _OUT(bcounters), .arg_u32 = 2 * 2, .cb = snl_attr_get_uint64_array },
393 	{ .type = PF_GS_PCOUNTERS, .off = _OUT(pcounters), .arg_u32 = 2 * 2 * 2, .cb = snl_attr_get_uint64_array },
394 };
395 static struct snl_field_parser fp_getstatus[] = {};
396 SNL_DECLARE_PARSER(getstatus_parser, struct genlmsghdr, fp_getstatus, ap_getstatus);
397 #undef _OUT
398 
399 struct pfctl_status *
pfctl_get_status_h(struct pfctl_handle * h __unused)400 pfctl_get_status_h(struct pfctl_handle *h __unused)
401 {
402 	struct pfctl_status	*status;
403 	struct snl_errmsg_data e = {};
404 	struct nlmsghdr *hdr;
405 	struct snl_writer nw;
406 	uint32_t seq_id;
407 	int family_id;
408 
409 	family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
410 	if (family_id == 0)
411 		return (NULL);
412 
413 	snl_init_writer(&h->ss, &nw);
414 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_GET_STATUS);
415 	hdr->nlmsg_flags |= NLM_F_DUMP;
416 
417 	hdr = snl_finalize_msg(&nw);
418 	if (hdr == NULL) {
419 		return (NULL);
420 	}
421 
422 	seq_id = hdr->nlmsg_seq;
423 	if (! snl_send_message(&h->ss, hdr))
424 		return (NULL);
425 
426 	status = calloc(1, sizeof(*status));
427 	if (status == NULL)
428 		return (NULL);
429 	TAILQ_INIT(&status->counters);
430 	TAILQ_INIT(&status->lcounters);
431 	TAILQ_INIT(&status->fcounters);
432 	TAILQ_INIT(&status->scounters);
433 
434 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
435 		if (! snl_parse_nlmsg(&h->ss, hdr, &getstatus_parser, status))
436 			continue;
437 	}
438 
439 	return (status);
440 }
441 
442 struct pfctl_status *
pfctl_get_status(int dev)443 pfctl_get_status(int dev)
444 {
445 	struct pfctl_status	*status;
446 	nvlist_t	*nvl;
447 	size_t		 len;
448 	const void	*chksum;
449 
450 	status = calloc(1, sizeof(*status));
451 	if (status == NULL)
452 		return (NULL);
453 
454 	nvl = nvlist_create(0);
455 
456 	if (pfctl_do_ioctl(dev, DIOCGETSTATUSNV, 4096, &nvl)) {
457 		nvlist_destroy(nvl);
458 		free(status);
459 		return (NULL);
460 	}
461 
462 	status->running = nvlist_get_bool(nvl, "running");
463 	status->since = nvlist_get_number(nvl, "since");
464 	status->debug = nvlist_get_number(nvl, "debug");
465 	status->hostid = ntohl(nvlist_get_number(nvl, "hostid"));
466 	status->states = nvlist_get_number(nvl, "states");
467 	status->src_nodes = nvlist_get_number(nvl, "src_nodes");
468 	status->syncookies_active = nvlist_get_bool(nvl, "syncookies_active");
469 	status->reass = nvlist_get_number(nvl, "reass");
470 
471 	strlcpy(status->ifname, nvlist_get_string(nvl, "ifname"),
472 	    IFNAMSIZ);
473 	chksum = nvlist_get_binary(nvl, "chksum", &len);
474 	assert(len == PF_MD5_DIGEST_LENGTH);
475 	memcpy(status->pf_chksum, chksum, len);
476 
477 	_pfctl_get_status_counters(nvlist_get_nvlist(nvl, "counters"),
478 	    &status->counters);
479 	_pfctl_get_status_counters(nvlist_get_nvlist(nvl, "lcounters"),
480 	    &status->lcounters);
481 	_pfctl_get_status_counters(nvlist_get_nvlist(nvl, "fcounters"),
482 	    &status->fcounters);
483 	_pfctl_get_status_counters(nvlist_get_nvlist(nvl, "scounters"),
484 	    &status->scounters);
485 
486 	pf_nvuint_64_array(nvl, "pcounters", 2 * 2 * 2,
487 	    (uint64_t *)status->pcounters, NULL);
488 	pf_nvuint_64_array(nvl, "bcounters", 2 * 2,
489 	    (uint64_t *)status->bcounters, NULL);
490 
491 	nvlist_destroy(nvl);
492 
493 	return (status);
494 }
495 int
pfctl_clear_status(struct pfctl_handle * h)496 pfctl_clear_status(struct pfctl_handle *h)
497 {
498 	return (pfctl_do_netlink_cmd(h, PFNL_CMD_CLEAR_STATUS));
499 }
500 
501 static uint64_t
_pfctl_status_counter(struct pfctl_status_counters * counters,uint64_t id)502 _pfctl_status_counter(struct pfctl_status_counters *counters, uint64_t id)
503 {
504 	struct pfctl_status_counter *c;
505 
506 	TAILQ_FOREACH(c, counters, entry) {
507 		if (c->id == id)
508 			return (c->counter);
509 	}
510 
511 	return (0);
512 }
513 
514 uint64_t
pfctl_status_counter(struct pfctl_status * status,int id)515 pfctl_status_counter(struct pfctl_status *status, int id)
516 {
517 	return (_pfctl_status_counter(&status->counters, id));
518 }
519 
520 uint64_t
pfctl_status_lcounter(struct pfctl_status * status,int id)521 pfctl_status_lcounter(struct pfctl_status *status, int id)
522 {
523 	return (_pfctl_status_counter(&status->lcounters, id));
524 }
525 
526 uint64_t
pfctl_status_fcounter(struct pfctl_status * status,int id)527 pfctl_status_fcounter(struct pfctl_status *status, int id)
528 {
529 	return (_pfctl_status_counter(&status->fcounters, id));
530 }
531 
532 uint64_t
pfctl_status_scounter(struct pfctl_status * status,int id)533 pfctl_status_scounter(struct pfctl_status *status, int id)
534 {
535 	return (_pfctl_status_counter(&status->scounters, id));
536 }
537 
538 void
pfctl_free_status(struct pfctl_status * status)539 pfctl_free_status(struct pfctl_status *status)
540 {
541 	struct pfctl_status_counter *c, *tmp;
542 
543 	if (status == NULL)
544 		return;
545 
546 	TAILQ_FOREACH_SAFE(c, &status->counters, entry, tmp) {
547 		free(c->name);
548 		free(c);
549 	}
550 	TAILQ_FOREACH_SAFE(c, &status->lcounters, entry, tmp) {
551 		free(c->name);
552 		free(c);
553 	}
554 	TAILQ_FOREACH_SAFE(c, &status->fcounters, entry, tmp) {
555 		free(c->name);
556 		free(c);
557 	}
558 	TAILQ_FOREACH_SAFE(c, &status->scounters, entry, tmp) {
559 		free(c->name);
560 		free(c);
561 	}
562 
563 	free(status);
564 }
565 
566 static void
pfctl_nv_add_addr(nvlist_t * nvparent,const char * name,const struct pf_addr * addr)567 pfctl_nv_add_addr(nvlist_t *nvparent, const char *name,
568     const struct pf_addr *addr)
569 {
570 	nvlist_t *nvl = nvlist_create(0);
571 
572 	nvlist_add_binary(nvl, "addr", addr, sizeof(*addr));
573 
574 	nvlist_add_nvlist(nvparent, name, nvl);
575 	nvlist_destroy(nvl);
576 }
577 
578 static void
pf_nvaddr_to_addr(const nvlist_t * nvl,struct pf_addr * addr)579 pf_nvaddr_to_addr(const nvlist_t *nvl, struct pf_addr *addr)
580 {
581 	size_t len;
582 	const void *data;
583 
584 	data = nvlist_get_binary(nvl, "addr", &len);
585 	assert(len == sizeof(struct pf_addr));
586 	memcpy(addr, data, len);
587 }
588 
589 static void
pfctl_nv_add_addr_wrap(nvlist_t * nvparent,const char * name,const struct pf_addr_wrap * addr)590 pfctl_nv_add_addr_wrap(nvlist_t *nvparent, const char *name,
591     const struct pf_addr_wrap *addr)
592 {
593 	nvlist_t *nvl = nvlist_create(0);
594 
595 	nvlist_add_number(nvl, "type", addr->type);
596 	nvlist_add_number(nvl, "iflags", addr->iflags);
597 	if (addr->type == PF_ADDR_DYNIFTL)
598 		nvlist_add_string(nvl, "ifname", addr->v.ifname);
599 	if (addr->type == PF_ADDR_TABLE)
600 		nvlist_add_string(nvl, "tblname", addr->v.tblname);
601 	pfctl_nv_add_addr(nvl, "addr", &addr->v.a.addr);
602 	pfctl_nv_add_addr(nvl, "mask", &addr->v.a.mask);
603 
604 	nvlist_add_nvlist(nvparent, name, nvl);
605 	nvlist_destroy(nvl);
606 }
607 
608 static void
pf_nvaddr_wrap_to_addr_wrap(const nvlist_t * nvl,struct pf_addr_wrap * addr)609 pf_nvaddr_wrap_to_addr_wrap(const nvlist_t *nvl, struct pf_addr_wrap *addr)
610 {
611 	bzero(addr, sizeof(*addr));
612 
613 	addr->type = nvlist_get_number(nvl, "type");
614 	addr->iflags = nvlist_get_number(nvl, "iflags");
615 	if (addr->type == PF_ADDR_DYNIFTL) {
616 		strlcpy(addr->v.ifname, nvlist_get_string(nvl, "ifname"),
617 		    IFNAMSIZ);
618 		addr->p.dyncnt = nvlist_get_number(nvl, "dyncnt");
619 	}
620 	if (addr->type == PF_ADDR_TABLE) {
621 		strlcpy(addr->v.tblname, nvlist_get_string(nvl, "tblname"),
622 		    PF_TABLE_NAME_SIZE);
623 		addr->p.tblcnt = nvlist_get_number(nvl, "tblcnt");
624 	}
625 
626 	pf_nvaddr_to_addr(nvlist_get_nvlist(nvl, "addr"), &addr->v.a.addr);
627 	pf_nvaddr_to_addr(nvlist_get_nvlist(nvl, "mask"), &addr->v.a.mask);
628 }
629 
630 static void
pfctl_nv_add_rule_addr(nvlist_t * nvparent,const char * name,const struct pf_rule_addr * addr)631 pfctl_nv_add_rule_addr(nvlist_t *nvparent, const char *name,
632     const struct pf_rule_addr *addr)
633 {
634 	uint64_t ports[2];
635 	nvlist_t *nvl = nvlist_create(0);
636 
637 	pfctl_nv_add_addr_wrap(nvl, "addr", &addr->addr);
638 	ports[0] = addr->port[0];
639 	ports[1] = addr->port[1];
640 	nvlist_add_number_array(nvl, "port", ports, 2);
641 	nvlist_add_number(nvl, "neg", addr->neg);
642 	nvlist_add_number(nvl, "port_op", addr->port_op);
643 
644 	nvlist_add_nvlist(nvparent, name, nvl);
645 	nvlist_destroy(nvl);
646 }
647 
648 static void
pf_nvrule_addr_to_rule_addr(const nvlist_t * nvl,struct pf_rule_addr * addr)649 pf_nvrule_addr_to_rule_addr(const nvlist_t *nvl, struct pf_rule_addr *addr)
650 {
651 	pf_nvaddr_wrap_to_addr_wrap(nvlist_get_nvlist(nvl, "addr"), &addr->addr);
652 
653 	pf_nvuint_16_array(nvl, "port", 2, addr->port, NULL);
654 	addr->neg = nvlist_get_number(nvl, "neg");
655 	addr->port_op = nvlist_get_number(nvl, "port_op");
656 }
657 
658 static void
pf_nvmape_to_mape(const nvlist_t * nvl,struct pf_mape_portset * mape)659 pf_nvmape_to_mape(const nvlist_t *nvl, struct pf_mape_portset *mape)
660 {
661 	mape->offset = nvlist_get_number(nvl, "offset");
662 	mape->psidlen = nvlist_get_number(nvl, "psidlen");
663 	mape->psid = nvlist_get_number(nvl, "psid");
664 }
665 
666 static void
pf_nvpool_to_pool(const nvlist_t * nvl,struct pfctl_pool * pool)667 pf_nvpool_to_pool(const nvlist_t *nvl, struct pfctl_pool *pool)
668 {
669 	size_t len;
670 	const void *data;
671 
672 	data = nvlist_get_binary(nvl, "key", &len);
673 	assert(len == sizeof(pool->key));
674 	memcpy(&pool->key, data, len);
675 
676 	pf_nvaddr_to_addr(nvlist_get_nvlist(nvl, "counter"), &pool->counter);
677 
678 	pool->tblidx = nvlist_get_number(nvl, "tblidx");
679 	pf_nvuint_16_array(nvl, "proxy_port", 2, pool->proxy_port, NULL);
680 	pool->opts = nvlist_get_number(nvl, "opts");
681 
682 	if (nvlist_exists_nvlist(nvl, "mape"))
683 		pf_nvmape_to_mape(nvlist_get_nvlist(nvl, "mape"), &pool->mape);
684 }
685 
686 static void
pf_nvrule_uid_to_rule_uid(const nvlist_t * nvl,struct pf_rule_uid * uid)687 pf_nvrule_uid_to_rule_uid(const nvlist_t *nvl, struct pf_rule_uid *uid)
688 {
689 	pf_nvuint_32_array(nvl, "uid", 2, uid->uid, NULL);
690 	uid->op = nvlist_get_number(nvl, "op");
691 }
692 
693 static void
pf_nvdivert_to_divert(const nvlist_t * nvl,struct pfctl_rule * rule)694 pf_nvdivert_to_divert(const nvlist_t *nvl, struct pfctl_rule *rule)
695 {
696 	pf_nvaddr_to_addr(nvlist_get_nvlist(nvl, "addr"), &rule->divert.addr);
697 	rule->divert.port = nvlist_get_number(nvl, "port");
698 }
699 
700 static void
pf_nvrule_to_rule(const nvlist_t * nvl,struct pfctl_rule * rule)701 pf_nvrule_to_rule(const nvlist_t *nvl, struct pfctl_rule *rule)
702 {
703 	const uint64_t *skip;
704 	const char *const *labels;
705 	size_t skipcount, labelcount;
706 
707 	rule->nr = nvlist_get_number(nvl, "nr");
708 
709 	pf_nvrule_addr_to_rule_addr(nvlist_get_nvlist(nvl, "src"), &rule->src);
710 	pf_nvrule_addr_to_rule_addr(nvlist_get_nvlist(nvl, "dst"), &rule->dst);
711 
712 	skip = nvlist_get_number_array(nvl, "skip", &skipcount);
713 	assert(skip);
714 	assert(skipcount == PF_SKIP_COUNT);
715 	for (int i = 0; i < PF_SKIP_COUNT; i++)
716 		rule->skip[i].nr = skip[i];
717 
718 	labels = nvlist_get_string_array(nvl, "labels", &labelcount);
719 	assert(labelcount <= PF_RULE_MAX_LABEL_COUNT);
720 	for (size_t i = 0; i < labelcount; i++)
721 		strlcpy(rule->label[i], labels[i], PF_RULE_LABEL_SIZE);
722 	rule->ridentifier = nvlist_get_number(nvl, "ridentifier");
723 	strlcpy(rule->ifname, nvlist_get_string(nvl, "ifname"), IFNAMSIZ);
724 	strlcpy(rule->qname, nvlist_get_string(nvl, "qname"), PF_QNAME_SIZE);
725 	strlcpy(rule->pqname, nvlist_get_string(nvl, "pqname"), PF_QNAME_SIZE);
726 	strlcpy(rule->tagname, nvlist_get_string(nvl, "tagname"),
727 	    PF_TAG_NAME_SIZE);
728 	strlcpy(rule->match_tagname, nvlist_get_string(nvl, "match_tagname"),
729 	    PF_TAG_NAME_SIZE);
730 
731 	strlcpy(rule->overload_tblname, nvlist_get_string(nvl, "overload_tblname"),
732 	    PF_TABLE_NAME_SIZE);
733 
734 	pf_nvpool_to_pool(nvlist_get_nvlist(nvl, "rpool"), &rule->rpool);
735 
736 	rule->evaluations = nvlist_get_number(nvl, "evaluations");
737 	pf_nvuint_64_array(nvl, "packets", 2, rule->packets, NULL);
738 	pf_nvuint_64_array(nvl, "bytes", 2, rule->bytes, NULL);
739 
740 	if (nvlist_exists_number(nvl, "timestamp")) {
741 		rule->last_active_timestamp = nvlist_get_number(nvl, "timestamp");
742 	}
743 
744 	rule->os_fingerprint = nvlist_get_number(nvl, "os_fingerprint");
745 
746 	rule->rtableid = nvlist_get_number(nvl, "rtableid");
747 	pf_nvuint_32_array(nvl, "timeout", PFTM_MAX, rule->timeout, NULL);
748 	rule->max_states = nvlist_get_number(nvl, "max_states");
749 	rule->max_src_nodes = nvlist_get_number(nvl, "max_src_nodes");
750 	rule->max_src_states = nvlist_get_number(nvl, "max_src_states");
751 	rule->max_src_conn = nvlist_get_number(nvl, "max_src_conn");
752 	rule->max_src_conn_rate.limit =
753 	    nvlist_get_number(nvl, "max_src_conn_rate.limit");
754 	rule->max_src_conn_rate.seconds =
755 	    nvlist_get_number(nvl, "max_src_conn_rate.seconds");
756 	rule->qid = nvlist_get_number(nvl, "qid");
757 	rule->pqid = nvlist_get_number(nvl, "pqid");
758 	rule->dnpipe = nvlist_get_number(nvl, "dnpipe");
759 	rule->dnrpipe = nvlist_get_number(nvl, "dnrpipe");
760 	rule->free_flags = nvlist_get_number(nvl, "dnflags");
761 	rule->prob = nvlist_get_number(nvl, "prob");
762 	rule->cuid = nvlist_get_number(nvl, "cuid");
763 	rule->cpid = nvlist_get_number(nvl, "cpid");
764 
765 	rule->return_icmp = nvlist_get_number(nvl, "return_icmp");
766 	rule->return_icmp6 = nvlist_get_number(nvl, "return_icmp6");
767 	rule->max_mss = nvlist_get_number(nvl, "max_mss");
768 	rule->scrub_flags = nvlist_get_number(nvl, "scrub_flags");
769 
770 	pf_nvrule_uid_to_rule_uid(nvlist_get_nvlist(nvl, "uid"), &rule->uid);
771 	pf_nvrule_uid_to_rule_uid(nvlist_get_nvlist(nvl, "gid"),
772 	    (struct pf_rule_uid *)&rule->gid);
773 
774 	rule->rule_flag = nvlist_get_number(nvl, "rule_flag");
775 	rule->action = nvlist_get_number(nvl, "action");
776 	rule->direction = nvlist_get_number(nvl, "direction");
777 	rule->log = nvlist_get_number(nvl, "log");
778 	rule->logif = nvlist_get_number(nvl, "logif");
779 	rule->quick = nvlist_get_number(nvl, "quick");
780 	rule->ifnot = nvlist_get_number(nvl, "ifnot");
781 	rule->match_tag_not = nvlist_get_number(nvl, "match_tag_not");
782 	rule->natpass = nvlist_get_number(nvl, "natpass");
783 
784 	rule->keep_state = nvlist_get_number(nvl, "keep_state");
785 	rule->af = nvlist_get_number(nvl, "af");
786 	rule->proto = nvlist_get_number(nvl, "proto");
787 	rule->type = nvlist_get_number(nvl, "type");
788 	rule->code = nvlist_get_number(nvl, "code");
789 	rule->flags = nvlist_get_number(nvl, "flags");
790 	rule->flagset = nvlist_get_number(nvl, "flagset");
791 	rule->min_ttl = nvlist_get_number(nvl, "min_ttl");
792 	rule->allow_opts = nvlist_get_number(nvl, "allow_opts");
793 	rule->rt = nvlist_get_number(nvl, "rt");
794 	rule->return_ttl  = nvlist_get_number(nvl, "return_ttl");
795 	rule->tos = nvlist_get_number(nvl, "tos");
796 	rule->set_tos = nvlist_get_number(nvl, "set_tos");
797 	rule->anchor_relative = nvlist_get_number(nvl, "anchor_relative");
798 	rule->anchor_wildcard = nvlist_get_number(nvl, "anchor_wildcard");
799 
800 	rule->flush = nvlist_get_number(nvl, "flush");
801 	rule->prio = nvlist_get_number(nvl, "prio");
802 	pf_nvuint_8_array(nvl, "set_prio", 2, rule->set_prio, NULL);
803 
804 	pf_nvdivert_to_divert(nvlist_get_nvlist(nvl, "divert"), rule);
805 
806 	rule->states_cur = nvlist_get_number(nvl, "states_cur");
807 	rule->states_tot = nvlist_get_number(nvl, "states_tot");
808 	rule->src_nodes = nvlist_get_number(nvl, "src_nodes");
809 }
810 
811 static void
pfctl_nveth_addr_to_eth_addr(const nvlist_t * nvl,struct pfctl_eth_addr * addr)812 pfctl_nveth_addr_to_eth_addr(const nvlist_t *nvl, struct pfctl_eth_addr *addr)
813 {
814 	static const u_int8_t EMPTY_MAC[ETHER_ADDR_LEN] = { 0 };
815 	size_t len;
816 	const void *data;
817 
818 	data = nvlist_get_binary(nvl, "addr", &len);
819 	assert(len == sizeof(addr->addr));
820 	memcpy(addr->addr, data, sizeof(addr->addr));
821 
822 	data = nvlist_get_binary(nvl, "mask", &len);
823 	assert(len == sizeof(addr->mask));
824 	memcpy(addr->mask, data, sizeof(addr->mask));
825 
826 	addr->neg = nvlist_get_bool(nvl, "neg");
827 
828 	/* To make checks for 'is this address set?' easier. */
829 	addr->isset = memcmp(addr->addr, EMPTY_MAC, ETHER_ADDR_LEN) != 0;
830 }
831 
832 static nvlist_t *
pfctl_eth_addr_to_nveth_addr(const struct pfctl_eth_addr * addr)833 pfctl_eth_addr_to_nveth_addr(const struct pfctl_eth_addr *addr)
834 {
835 	nvlist_t *nvl;
836 
837 	nvl = nvlist_create(0);
838 	if (nvl == NULL)
839 		return (NULL);
840 
841 	nvlist_add_bool(nvl, "neg", addr->neg);
842 	nvlist_add_binary(nvl, "addr", &addr->addr, ETHER_ADDR_LEN);
843 	nvlist_add_binary(nvl, "mask", &addr->mask, ETHER_ADDR_LEN);
844 
845 	return (nvl);
846 }
847 
848 static void
pfctl_nveth_rule_to_eth_rule(const nvlist_t * nvl,struct pfctl_eth_rule * rule)849 pfctl_nveth_rule_to_eth_rule(const nvlist_t *nvl, struct pfctl_eth_rule *rule)
850 {
851 	const char *const *labels;
852 	size_t labelcount, i;
853 
854 	rule->nr = nvlist_get_number(nvl, "nr");
855 	rule->quick = nvlist_get_bool(nvl, "quick");
856 	strlcpy(rule->ifname, nvlist_get_string(nvl, "ifname"), IFNAMSIZ);
857 	rule->ifnot = nvlist_get_bool(nvl, "ifnot");
858 	rule->direction = nvlist_get_number(nvl, "direction");
859 	rule->proto = nvlist_get_number(nvl, "proto");
860 	strlcpy(rule->match_tagname, nvlist_get_string(nvl, "match_tagname"),
861 	    PF_TAG_NAME_SIZE);
862 	rule->match_tag = nvlist_get_number(nvl, "match_tag");
863 	rule->match_tag_not = nvlist_get_bool(nvl, "match_tag_not");
864 
865 	labels = nvlist_get_string_array(nvl, "labels", &labelcount);
866 	assert(labelcount <= PF_RULE_MAX_LABEL_COUNT);
867 	for (i = 0; i < labelcount; i++)
868 		strlcpy(rule->label[i], labels[i], PF_RULE_LABEL_SIZE);
869 	rule->ridentifier = nvlist_get_number(nvl, "ridentifier");
870 
871 	pfctl_nveth_addr_to_eth_addr(nvlist_get_nvlist(nvl, "src"),
872 	    &rule->src);
873 	pfctl_nveth_addr_to_eth_addr(nvlist_get_nvlist(nvl, "dst"),
874 	    &rule->dst);
875 
876 	pf_nvrule_addr_to_rule_addr(nvlist_get_nvlist(nvl, "ipsrc"),
877 	    &rule->ipsrc);
878 	pf_nvrule_addr_to_rule_addr(nvlist_get_nvlist(nvl, "ipdst"),
879 	    &rule->ipdst);
880 
881 	rule->evaluations = nvlist_get_number(nvl, "evaluations");
882 	rule->packets[0] = nvlist_get_number(nvl, "packets-in");
883 	rule->packets[1] = nvlist_get_number(nvl, "packets-out");
884 	rule->bytes[0] = nvlist_get_number(nvl, "bytes-in");
885 	rule->bytes[1] = nvlist_get_number(nvl, "bytes-out");
886 
887 	if (nvlist_exists_number(nvl, "timestamp")) {
888 		rule->last_active_timestamp = nvlist_get_number(nvl, "timestamp");
889 	}
890 
891 	strlcpy(rule->qname, nvlist_get_string(nvl, "qname"), PF_QNAME_SIZE);
892 	strlcpy(rule->tagname, nvlist_get_string(nvl, "tagname"),
893 	    PF_TAG_NAME_SIZE);
894 
895 	rule->dnpipe = nvlist_get_number(nvl, "dnpipe");
896 	rule->dnflags = nvlist_get_number(nvl, "dnflags");
897 
898 	rule->anchor_relative = nvlist_get_number(nvl, "anchor_relative");
899 	rule->anchor_wildcard = nvlist_get_number(nvl, "anchor_wildcard");
900 
901 	strlcpy(rule->bridge_to, nvlist_get_string(nvl, "bridge_to"),
902 	    IFNAMSIZ);
903 
904 	rule->action = nvlist_get_number(nvl, "action");
905 }
906 
907 int
pfctl_get_eth_rulesets_info(int dev,struct pfctl_eth_rulesets_info * ri,const char * path)908 pfctl_get_eth_rulesets_info(int dev, struct pfctl_eth_rulesets_info *ri,
909     const char *path)
910 {
911 	nvlist_t *nvl;
912 	int ret;
913 
914 	bzero(ri, sizeof(*ri));
915 
916 	nvl = nvlist_create(0);
917 	nvlist_add_string(nvl, "path", path);
918 
919 	if ((ret = pfctl_do_ioctl(dev, DIOCGETETHRULESETS, 256, &nvl)) != 0)
920 		goto out;
921 
922 	ri->nr = nvlist_get_number(nvl, "nr");
923 
924 out:
925 	nvlist_destroy(nvl);
926 	return (ret);
927 }
928 
929 int
pfctl_get_eth_ruleset(int dev,const char * path,int nr,struct pfctl_eth_ruleset_info * ri)930 pfctl_get_eth_ruleset(int dev, const char *path, int nr,
931     struct pfctl_eth_ruleset_info *ri)
932 {
933 	nvlist_t *nvl;
934 	int ret;
935 
936 	bzero(ri, sizeof(*ri));
937 
938 	nvl = nvlist_create(0);
939 	nvlist_add_string(nvl, "path", path);
940 	nvlist_add_number(nvl, "nr", nr);
941 
942 	if ((ret = pfctl_do_ioctl(dev, DIOCGETETHRULESET, 1024, &nvl)) != 0)
943 		goto out;
944 
945 	ri->nr = nvlist_get_number(nvl, "nr");
946 	strlcpy(ri->path, nvlist_get_string(nvl, "path"), MAXPATHLEN);
947 	strlcpy(ri->name, nvlist_get_string(nvl, "name"),
948 	    PF_ANCHOR_NAME_SIZE);
949 
950 out:
951 	nvlist_destroy(nvl);
952 	return (ret);
953 }
954 
955 int
pfctl_get_eth_rules_info(int dev,struct pfctl_eth_rules_info * rules,const char * path)956 pfctl_get_eth_rules_info(int dev, struct pfctl_eth_rules_info *rules,
957     const char *path)
958 {
959 	nvlist_t *nvl;
960 	int ret;
961 
962 	bzero(rules, sizeof(*rules));
963 
964 	nvl = nvlist_create(0);
965 	nvlist_add_string(nvl, "anchor", path);
966 
967 	if ((ret = pfctl_do_ioctl(dev, DIOCGETETHRULES, 1024, &nvl)) != 0)
968 		goto out;
969 
970 	rules->nr = nvlist_get_number(nvl, "nr");
971 	rules->ticket = nvlist_get_number(nvl, "ticket");
972 
973 out:
974 	nvlist_destroy(nvl);
975 	return (ret);
976 }
977 
978 int
pfctl_get_eth_rule(int dev,uint32_t nr,uint32_t ticket,const char * path,struct pfctl_eth_rule * rule,bool clear,char * anchor_call)979 pfctl_get_eth_rule(int dev, uint32_t nr, uint32_t ticket,
980     const char *path, struct pfctl_eth_rule *rule, bool clear,
981     char *anchor_call)
982 {
983 	nvlist_t *nvl;
984 	int ret;
985 
986 	nvl = nvlist_create(0);
987 
988 	nvlist_add_string(nvl, "anchor", path);
989 	nvlist_add_number(nvl, "ticket", ticket);
990 	nvlist_add_number(nvl, "nr", nr);
991 	nvlist_add_bool(nvl, "clear", clear);
992 
993 	if ((ret = pfctl_do_ioctl(dev, DIOCGETETHRULE, 4096, &nvl)) != 0)
994 		goto out;
995 
996 	pfctl_nveth_rule_to_eth_rule(nvl, rule);
997 
998 	if (anchor_call)
999 		strlcpy(anchor_call, nvlist_get_string(nvl, "anchor_call"),
1000 		    MAXPATHLEN);
1001 
1002 out:
1003 	nvlist_destroy(nvl);
1004 	return (ret);
1005 }
1006 
1007 int
pfctl_add_eth_rule(int dev,const struct pfctl_eth_rule * r,const char * anchor,const char * anchor_call,uint32_t ticket)1008 pfctl_add_eth_rule(int dev, const struct pfctl_eth_rule *r, const char *anchor,
1009     const char *anchor_call, uint32_t ticket)
1010 {
1011 	struct pfioc_nv nv;
1012 	nvlist_t *nvl, *addr;
1013 	void *packed;
1014 	int error = 0;
1015 	size_t labelcount, size;
1016 
1017 	nvl = nvlist_create(0);
1018 
1019 	nvlist_add_number(nvl, "ticket", ticket);
1020 	nvlist_add_string(nvl, "anchor", anchor);
1021 	nvlist_add_string(nvl, "anchor_call", anchor_call);
1022 
1023 	nvlist_add_number(nvl, "nr", r->nr);
1024 	nvlist_add_bool(nvl, "quick", r->quick);
1025 	nvlist_add_string(nvl, "ifname", r->ifname);
1026 	nvlist_add_bool(nvl, "ifnot", r->ifnot);
1027 	nvlist_add_number(nvl, "direction", r->direction);
1028 	nvlist_add_number(nvl, "proto", r->proto);
1029 	nvlist_add_string(nvl, "match_tagname", r->match_tagname);
1030 	nvlist_add_bool(nvl, "match_tag_not", r->match_tag_not);
1031 
1032 	addr = pfctl_eth_addr_to_nveth_addr(&r->src);
1033 	if (addr == NULL) {
1034 		nvlist_destroy(nvl);
1035 		return (ENOMEM);
1036 	}
1037 	nvlist_add_nvlist(nvl, "src", addr);
1038 	nvlist_destroy(addr);
1039 
1040 	addr = pfctl_eth_addr_to_nveth_addr(&r->dst);
1041 	if (addr == NULL) {
1042 		nvlist_destroy(nvl);
1043 		return (ENOMEM);
1044 	}
1045 	nvlist_add_nvlist(nvl, "dst", addr);
1046 	nvlist_destroy(addr);
1047 
1048 	pfctl_nv_add_rule_addr(nvl, "ipsrc", &r->ipsrc);
1049 	pfctl_nv_add_rule_addr(nvl, "ipdst", &r->ipdst);
1050 
1051 	labelcount = 0;
1052 	while (labelcount < PF_RULE_MAX_LABEL_COUNT &&
1053 	    r->label[labelcount][0] != 0) {
1054 		nvlist_append_string_array(nvl, "labels",
1055 		    r->label[labelcount]);
1056 		labelcount++;
1057 	}
1058 	nvlist_add_number(nvl, "ridentifier", r->ridentifier);
1059 
1060 	nvlist_add_string(nvl, "qname", r->qname);
1061 	nvlist_add_string(nvl, "tagname", r->tagname);
1062 	nvlist_add_number(nvl, "dnpipe", r->dnpipe);
1063 	nvlist_add_number(nvl, "dnflags", r->dnflags);
1064 
1065 	nvlist_add_string(nvl, "bridge_to", r->bridge_to);
1066 
1067 	nvlist_add_number(nvl, "action", r->action);
1068 
1069 	packed = nvlist_pack(nvl, &size);
1070 	if (packed == NULL) {
1071 		nvlist_destroy(nvl);
1072 		return (ENOMEM);
1073 	}
1074 
1075 	nv.len = size;
1076 	nv.size = size;
1077 	nv.data = packed;
1078 
1079 	if (ioctl(dev, DIOCADDETHRULE, &nv) != 0)
1080 		error = errno;
1081 
1082 	free(packed);
1083 	nvlist_destroy(nvl);
1084 
1085 	return (error);
1086 }
1087 
1088 static void
snl_add_msg_attr_addr_wrap(struct snl_writer * nw,uint32_t type,const struct pf_addr_wrap * addr)1089 snl_add_msg_attr_addr_wrap(struct snl_writer *nw, uint32_t type, const struct pf_addr_wrap *addr)
1090 {
1091 	int off;
1092 
1093 	off = snl_add_msg_attr_nested(nw, type);
1094 
1095 	snl_add_msg_attr_ip6(nw, PF_AT_ADDR, &addr->v.a.addr.v6);
1096 	snl_add_msg_attr_ip6(nw, PF_AT_MASK, &addr->v.a.mask.v6);
1097 
1098 	if (addr->type == PF_ADDR_DYNIFTL)
1099 		snl_add_msg_attr_string(nw, PF_AT_IFNAME, addr->v.ifname);
1100 	if (addr->type == PF_ADDR_TABLE)
1101 		snl_add_msg_attr_string(nw, PF_AT_TABLENAME, addr->v.tblname);
1102 	snl_add_msg_attr_u8(nw, PF_AT_TYPE, addr->type);
1103 	snl_add_msg_attr_u8(nw, PF_AT_IFLAGS, addr->iflags);
1104 
1105 	snl_end_attr_nested(nw, off);
1106 }
1107 
1108 static void
snl_add_msg_attr_pool_addr(struct snl_writer * nw,uint32_t type,const struct pf_pooladdr * pa)1109 snl_add_msg_attr_pool_addr(struct snl_writer *nw, uint32_t type, const struct pf_pooladdr *pa)
1110 {
1111 	int off;
1112 
1113 	off = snl_add_msg_attr_nested(nw, type);
1114 
1115 	snl_add_msg_attr_string(nw, PF_PA_IFNAME, pa->ifname);
1116 	snl_add_msg_attr_addr_wrap(nw, PF_PA_ADDR, &pa->addr);
1117 
1118 	snl_end_attr_nested(nw, off);
1119 }
1120 
1121 static void
snl_add_msg_attr_rule_addr(struct snl_writer * nw,uint32_t type,const struct pf_rule_addr * addr)1122 snl_add_msg_attr_rule_addr(struct snl_writer *nw, uint32_t type, const struct pf_rule_addr *addr)
1123 {
1124 	int off;
1125 
1126 	off = snl_add_msg_attr_nested(nw, type);
1127 
1128 	snl_add_msg_attr_addr_wrap(nw, PF_RAT_ADDR, &addr->addr);
1129 	snl_add_msg_attr_u16(nw, PF_RAT_SRC_PORT, addr->port[0]);
1130 	snl_add_msg_attr_u16(nw, PF_RAT_DST_PORT, addr->port[1]);
1131 	snl_add_msg_attr_u8(nw, PF_RAT_NEG, addr->neg);
1132 	snl_add_msg_attr_u8(nw, PF_RAT_OP, addr->port_op);
1133 
1134 	snl_end_attr_nested(nw, off);
1135 }
1136 
1137 static void
snl_add_msg_attr_rule_labels(struct snl_writer * nw,uint32_t type,const char labels[PF_RULE_MAX_LABEL_COUNT][PF_RULE_LABEL_SIZE])1138 snl_add_msg_attr_rule_labels(struct snl_writer *nw, uint32_t type, const char labels[PF_RULE_MAX_LABEL_COUNT][PF_RULE_LABEL_SIZE])
1139 {
1140 	int off, i = 0;
1141 
1142 	off = snl_add_msg_attr_nested(nw, type);
1143 
1144 	while (i < PF_RULE_MAX_LABEL_COUNT &&
1145 	    labels[i][0] != 0) {
1146 		snl_add_msg_attr_string(nw, PF_LT_LABEL, labels[i]);
1147 		i++;
1148 	}
1149 
1150 	snl_end_attr_nested(nw, off);
1151 }
1152 
1153 static void
snl_add_msg_attr_mape(struct snl_writer * nw,uint32_t type,const struct pf_mape_portset * me)1154 snl_add_msg_attr_mape(struct snl_writer *nw, uint32_t type, const struct pf_mape_portset *me)
1155 {
1156 	int off;
1157 
1158 	off = snl_add_msg_attr_nested(nw, type);
1159 
1160 	snl_add_msg_attr_u8(nw, PF_MET_OFFSET, me->offset);
1161 	snl_add_msg_attr_u8(nw, PF_MET_PSID_LEN, me->psidlen);
1162 	snl_add_msg_attr_u16(nw, PF_MET_PSID, me->psid);
1163 
1164 	snl_end_attr_nested(nw, off);
1165 }
1166 
1167 static void
snl_add_msg_attr_rpool(struct snl_writer * nw,uint32_t type,const struct pfctl_pool * pool)1168 snl_add_msg_attr_rpool(struct snl_writer *nw, uint32_t type, const struct pfctl_pool *pool)
1169 {
1170 	int off;
1171 
1172 	off = snl_add_msg_attr_nested(nw, type);
1173 
1174 	snl_add_msg_attr(nw, PF_PT_KEY, sizeof(pool->key), &pool->key);
1175 	snl_add_msg_attr_ip6(nw, PF_PT_COUNTER, &pool->counter.v6);
1176 	snl_add_msg_attr_u32(nw, PF_PT_TBLIDX, pool->tblidx);
1177 	snl_add_msg_attr_u16(nw, PF_PT_PROXY_SRC_PORT, pool->proxy_port[0]);
1178 	snl_add_msg_attr_u16(nw, PF_PT_PROXY_DST_PORT, pool->proxy_port[1]);
1179 	snl_add_msg_attr_u8(nw, PF_PT_OPTS, pool->opts);
1180 	snl_add_msg_attr_mape(nw, PF_PT_MAPE, &pool->mape);
1181 
1182 	snl_end_attr_nested(nw, off);
1183 }
1184 
1185 static void
snl_add_msg_attr_timeouts(struct snl_writer * nw,uint32_t type,const uint32_t * timeouts)1186 snl_add_msg_attr_timeouts(struct snl_writer *nw, uint32_t type, const uint32_t *timeouts)
1187 {
1188 	int off;
1189 
1190 	off = snl_add_msg_attr_nested(nw, type);
1191 
1192 	for (int i = 0; i < PFTM_MAX; i++)
1193 		snl_add_msg_attr_u32(nw, PF_TT_TIMEOUT, timeouts[i]);
1194 
1195 	snl_end_attr_nested(nw, off);
1196 }
1197 
1198 static void
snl_add_msg_attr_uid(struct snl_writer * nw,uint32_t type,const struct pf_rule_uid * uid)1199 snl_add_msg_attr_uid(struct snl_writer *nw, uint32_t type, const struct pf_rule_uid *uid)
1200 {
1201 	int off;
1202 
1203 	off = snl_add_msg_attr_nested(nw, type);
1204 
1205 	snl_add_msg_attr_u32(nw, PF_RUT_UID_LOW, uid->uid[0]);
1206 	snl_add_msg_attr_u32(nw, PF_RUT_UID_HIGH, uid->uid[1]);
1207 	snl_add_msg_attr_u8(nw, PF_RUT_OP, uid->op);
1208 
1209 	snl_end_attr_nested(nw, off);
1210 }
1211 
1212 static void
snl_add_msg_attr_pf_rule(struct snl_writer * nw,uint32_t type,const struct pfctl_rule * r)1213 snl_add_msg_attr_pf_rule(struct snl_writer *nw, uint32_t type, const struct pfctl_rule *r)
1214 {
1215 	int off;
1216 
1217 	off = snl_add_msg_attr_nested(nw, type);
1218 
1219 	snl_add_msg_attr_rule_addr(nw, PF_RT_SRC, &r->src);
1220 	snl_add_msg_attr_rule_addr(nw, PF_RT_DST, &r->dst);
1221 	snl_add_msg_attr_rule_labels(nw, PF_RT_LABELS, r->label);
1222 	snl_add_msg_attr_u32(nw, PF_RT_RIDENTIFIER, r->ridentifier);
1223 	snl_add_msg_attr_string(nw, PF_RT_IFNAME, r->ifname);
1224 	snl_add_msg_attr_string(nw, PF_RT_QNAME, r->qname);
1225 	snl_add_msg_attr_string(nw, PF_RT_PQNAME, r->pqname);
1226 	snl_add_msg_attr_string(nw, PF_RT_TAGNAME, r->tagname);
1227 	snl_add_msg_attr_string(nw, PF_RT_MATCH_TAGNAME, r->match_tagname);
1228 	snl_add_msg_attr_string(nw, PF_RT_OVERLOAD_TBLNAME, r->overload_tblname);
1229 	snl_add_msg_attr_rpool(nw, PF_RT_RPOOL, &r->rpool);
1230 	snl_add_msg_attr_u32(nw, PF_RT_OS_FINGERPRINT, r->os_fingerprint);
1231 	snl_add_msg_attr_u32(nw, PF_RT_RTABLEID, r->rtableid);
1232 	snl_add_msg_attr_timeouts(nw, PF_RT_TIMEOUT, r->timeout);
1233 	snl_add_msg_attr_u32(nw, PF_RT_MAX_STATES, r->max_states);
1234 	snl_add_msg_attr_u32(nw, PF_RT_MAX_SRC_NODES, r->max_src_nodes);
1235 	snl_add_msg_attr_u32(nw, PF_RT_MAX_SRC_STATES, r->max_src_states);
1236 	snl_add_msg_attr_u32(nw, PF_RT_MAX_SRC_CONN_RATE_LIMIT, r->max_src_conn_rate.limit);
1237 	snl_add_msg_attr_u32(nw, PF_RT_MAX_SRC_CONN_RATE_SECS, r->max_src_conn_rate.seconds);
1238 
1239 	snl_add_msg_attr_u16(nw, PF_RT_DNPIPE, r->dnpipe);
1240 	snl_add_msg_attr_u16(nw, PF_RT_DNRPIPE, r->dnrpipe);
1241 	snl_add_msg_attr_u32(nw, PF_RT_DNFLAGS, r->free_flags);
1242 
1243 	snl_add_msg_attr_u32(nw, PF_RT_NR, r->nr);
1244 	snl_add_msg_attr_u32(nw, PF_RT_PROB, r->prob);
1245 	snl_add_msg_attr_u32(nw, PF_RT_CUID, r->cuid);
1246 	snl_add_msg_attr_u32(nw, PF_RT_CPID, r->cpid);
1247 
1248 	snl_add_msg_attr_u16(nw, PF_RT_RETURN_ICMP, r->return_icmp);
1249 	snl_add_msg_attr_u16(nw, PF_RT_RETURN_ICMP6, r->return_icmp6);
1250 	snl_add_msg_attr_u16(nw, PF_RT_MAX_MSS, r->max_mss);
1251 	snl_add_msg_attr_u16(nw, PF_RT_SCRUB_FLAGS, r->scrub_flags);
1252 
1253 	snl_add_msg_attr_uid(nw, PF_RT_UID, &r->uid);
1254 	snl_add_msg_attr_uid(nw, PF_RT_GID, (const struct pf_rule_uid *)&r->gid);
1255 
1256 	snl_add_msg_attr_u32(nw, PF_RT_RULE_FLAG, r->rule_flag);
1257 	snl_add_msg_attr_u8(nw, PF_RT_ACTION, r->action);
1258 	snl_add_msg_attr_u8(nw, PF_RT_DIRECTION, r->direction);
1259 	snl_add_msg_attr_u8(nw, PF_RT_LOG, r->log);
1260 	snl_add_msg_attr_u8(nw, PF_RT_LOGIF, r->logif);
1261 	snl_add_msg_attr_u8(nw, PF_RT_QUICK, r->quick);
1262 	snl_add_msg_attr_u8(nw, PF_RT_IF_NOT, r->ifnot);
1263 	snl_add_msg_attr_u8(nw, PF_RT_MATCH_TAG_NOT, r->match_tag_not);
1264 	snl_add_msg_attr_u8(nw, PF_RT_NATPASS, r->natpass);
1265 	snl_add_msg_attr_u8(nw, PF_RT_KEEP_STATE, r->keep_state);
1266 	snl_add_msg_attr_u8(nw, PF_RT_AF, r->af);
1267 	snl_add_msg_attr_u8(nw, PF_RT_PROTO, r->proto);
1268 	snl_add_msg_attr_u8(nw, PF_RT_TYPE, r->type);
1269 	snl_add_msg_attr_u8(nw, PF_RT_CODE, r->code);
1270 	snl_add_msg_attr_u8(nw, PF_RT_FLAGS, r->flags);
1271 	snl_add_msg_attr_u8(nw, PF_RT_FLAGSET, r->flagset);
1272 	snl_add_msg_attr_u8(nw, PF_RT_MIN_TTL, r->min_ttl);
1273 	snl_add_msg_attr_u8(nw, PF_RT_ALLOW_OPTS, r->allow_opts);
1274 	snl_add_msg_attr_u8(nw, PF_RT_RT, r->rt);
1275 	snl_add_msg_attr_u8(nw, PF_RT_RETURN_TTL, r->return_ttl);
1276 	snl_add_msg_attr_u8(nw, PF_RT_TOS, r->tos);
1277 	snl_add_msg_attr_u8(nw, PF_RT_SET_TOS, r->set_tos);
1278 
1279 	snl_add_msg_attr_u8(nw, PF_RT_ANCHOR_RELATIVE, r->anchor_relative);
1280 	snl_add_msg_attr_u8(nw, PF_RT_ANCHOR_WILDCARD, r->anchor_wildcard);
1281 	snl_add_msg_attr_u8(nw, PF_RT_FLUSH, r->flush);
1282 	snl_add_msg_attr_u8(nw, PF_RT_PRIO, r->prio);
1283 	snl_add_msg_attr_u8(nw, PF_RT_SET_PRIO, r->set_prio[0]);
1284 	snl_add_msg_attr_u8(nw, PF_RT_SET_PRIO_REPLY, r->set_prio[1]);
1285 
1286 	snl_add_msg_attr_ip6(nw, PF_RT_DIVERT_ADDRESS, &r->divert.addr.v6);
1287 	snl_add_msg_attr_u16(nw, PF_RT_DIVERT_PORT, r->divert.port);
1288 
1289 	snl_end_attr_nested(nw, off);
1290 }
1291 
1292 int
pfctl_add_rule(int dev __unused,const struct pfctl_rule * r,const char * anchor,const char * anchor_call,uint32_t ticket,uint32_t pool_ticket)1293 pfctl_add_rule(int dev __unused, const struct pfctl_rule *r, const char *anchor,
1294     const char *anchor_call, uint32_t ticket, uint32_t pool_ticket)
1295 {
1296 	struct pfctl_handle *h;
1297 	int ret;
1298 
1299 	h = pfctl_open(PF_DEVICE);
1300 	if (h == NULL)
1301 		return (ENODEV);
1302 
1303 	ret = pfctl_add_rule_h(h, r, anchor, anchor_call, ticket, pool_ticket);
1304 
1305 	pfctl_close(h);
1306 
1307 	return (ret);
1308 }
1309 
1310 int
pfctl_add_rule_h(struct pfctl_handle * h,const struct pfctl_rule * r,const char * anchor,const char * anchor_call,uint32_t ticket,uint32_t pool_ticket)1311 pfctl_add_rule_h(struct pfctl_handle *h, const struct pfctl_rule *r,
1312 	    const char *anchor, const char *anchor_call, uint32_t ticket,
1313 	    uint32_t pool_ticket)
1314 {
1315 	struct snl_writer nw;
1316 	struct snl_errmsg_data e = {};
1317 	struct nlmsghdr *hdr;
1318 	uint32_t seq_id;
1319 	int family_id;
1320 
1321 	family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
1322 	if (family_id == 0)
1323 		return (ENOTSUP);
1324 
1325 	snl_init_writer(&h->ss, &nw);
1326 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_ADDRULE);
1327 	hdr->nlmsg_flags |= NLM_F_DUMP;
1328 	snl_add_msg_attr_u32(&nw, PF_ART_TICKET, ticket);
1329 	snl_add_msg_attr_u32(&nw, PF_ART_POOL_TICKET, pool_ticket);
1330 	snl_add_msg_attr_string(&nw, PF_ART_ANCHOR, anchor);
1331 	snl_add_msg_attr_string(&nw, PF_ART_ANCHOR_CALL, anchor_call);
1332 
1333 	snl_add_msg_attr_pf_rule(&nw, PF_ART_RULE, r);
1334 
1335 	if ((hdr = snl_finalize_msg(&nw)) == NULL)
1336 		return (ENXIO);
1337 
1338 	seq_id = hdr->nlmsg_seq;
1339 
1340 	if (! snl_send_message(&h->ss, hdr))
1341 		return (ENXIO);
1342 
1343 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
1344 	}
1345 
1346 	return (e.error);
1347 }
1348 
1349 #define	_IN(_field)	offsetof(struct genlmsghdr, _field)
1350 #define	_OUT(_field)	offsetof(struct pfctl_rules_info, _field)
1351 static struct snl_attr_parser ap_getrules[] = {
1352 	{ .type = PF_GR_NR, .off = _OUT(nr), .cb = snl_attr_get_uint32 },
1353 	{ .type = PF_GR_TICKET, .off = _OUT(ticket), .cb = snl_attr_get_uint32 },
1354 };
1355 static struct snl_field_parser fp_getrules[] = {
1356 };
1357 #undef _IN
1358 #undef _OUT
1359 SNL_DECLARE_PARSER(getrules_parser, struct genlmsghdr, fp_getrules, ap_getrules);
1360 
1361 int
pfctl_get_rules_info_h(struct pfctl_handle * h,struct pfctl_rules_info * rules,uint32_t ruleset,const char * path)1362 pfctl_get_rules_info_h(struct pfctl_handle *h, struct pfctl_rules_info *rules, uint32_t ruleset,
1363     const char *path)
1364 {
1365 	struct snl_errmsg_data e = {};
1366 	struct nlmsghdr *hdr;
1367 	struct snl_writer nw;
1368 	uint32_t seq_id;
1369 	int family_id;
1370 
1371 	family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
1372 	if (family_id == 0)
1373 		return (ENOTSUP);
1374 
1375 	snl_init_writer(&h->ss, &nw);
1376 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_GETRULES);
1377 	hdr->nlmsg_flags |= NLM_F_DUMP;
1378 
1379 	snl_add_msg_attr_string(&nw, PF_GR_ANCHOR, path);
1380 	snl_add_msg_attr_u8(&nw, PF_GR_ACTION, ruleset);
1381 
1382 	hdr = snl_finalize_msg(&nw);
1383 	if (hdr == NULL)
1384 		return (ENOMEM);
1385 
1386 	seq_id = hdr->nlmsg_seq;
1387 	if (! snl_send_message(&h->ss, hdr))
1388 		return (ENXIO);
1389 
1390 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
1391 		if (! snl_parse_nlmsg(&h->ss, hdr, &getrules_parser, rules))
1392 			continue;
1393 	}
1394 
1395 	return (e.error);
1396 }
1397 
1398 int
pfctl_get_rules_info(int dev __unused,struct pfctl_rules_info * rules,uint32_t ruleset,const char * path)1399 pfctl_get_rules_info(int dev __unused, struct pfctl_rules_info *rules, uint32_t ruleset,
1400     const char *path)
1401 {
1402 	struct pfctl_handle *h;
1403 	int error;
1404 
1405 	h = pfctl_open(PF_DEVICE);
1406 	if (h == NULL)
1407 		return (ENOTSUP);
1408 	error = pfctl_get_rules_info_h(h, rules, ruleset, path);
1409 	pfctl_close(h);
1410 
1411 	return (error);
1412 }
1413 
1414 int
pfctl_get_rule_h(struct pfctl_handle * h,uint32_t nr,uint32_t ticket,const char * anchor,uint32_t ruleset,struct pfctl_rule * rule,char * anchor_call)1415 pfctl_get_rule_h(struct pfctl_handle *h, uint32_t nr, uint32_t ticket, const char *anchor,
1416     uint32_t ruleset, struct pfctl_rule *rule, char *anchor_call)
1417 {
1418 	return (pfctl_get_clear_rule_h(h, nr, ticket, anchor, ruleset, rule,
1419 	    anchor_call, false));
1420 }
1421 
1422 int
pfctl_get_rule(int dev,uint32_t nr,uint32_t ticket,const char * anchor,uint32_t ruleset,struct pfctl_rule * rule,char * anchor_call)1423 pfctl_get_rule(int dev, uint32_t nr, uint32_t ticket, const char *anchor,
1424     uint32_t ruleset, struct pfctl_rule *rule, char *anchor_call)
1425 {
1426 	return (pfctl_get_clear_rule(dev, nr, ticket, anchor, ruleset, rule,
1427 	    anchor_call, false));
1428 }
1429 
1430 #define _OUT(_field)	offsetof(struct pf_addr_wrap, _field)
1431 static const struct snl_attr_parser ap_addr_wrap[] = {
1432 	{ .type = PF_AT_ADDR, .off = _OUT(v.a.addr), .cb = snl_attr_get_in6_addr },
1433 	{ .type = PF_AT_MASK, .off = _OUT(v.a.mask), .cb = snl_attr_get_in6_addr },
1434 	{ .type = PF_AT_IFNAME, .off = _OUT(v.ifname), .arg = (void *)IFNAMSIZ,.cb = snl_attr_copy_string },
1435 	{ .type = PF_AT_TABLENAME, .off = _OUT(v.tblname), .arg = (void *)PF_TABLE_NAME_SIZE, .cb = snl_attr_copy_string },
1436 	{ .type = PF_AT_TYPE, .off = _OUT(type), .cb = snl_attr_get_uint8 },
1437 	{ .type = PF_AT_IFLAGS, .off = _OUT(iflags), .cb = snl_attr_get_uint8 },
1438 	{ .type = PF_AT_TBLCNT, .off = _OUT(p.tblcnt), .cb = snl_attr_get_uint32 },
1439 	{ .type = PF_AT_DYNCNT, .off = _OUT(p.dyncnt), .cb = snl_attr_get_uint32 },
1440 };
1441 SNL_DECLARE_ATTR_PARSER(addr_wrap_parser, ap_addr_wrap);
1442 #undef _OUT
1443 
1444 #define _OUT(_field)	offsetof(struct pf_rule_addr, _field)
1445 static struct snl_attr_parser ap_rule_addr[] = {
1446 	{ .type = PF_RAT_ADDR, .off = _OUT(addr), .arg = &addr_wrap_parser, .cb = snl_attr_get_nested },
1447 	{ .type = PF_RAT_SRC_PORT, .off = _OUT(port[0]), .cb = snl_attr_get_uint16 },
1448 	{ .type = PF_RAT_DST_PORT, .off = _OUT(port[1]), .cb = snl_attr_get_uint16 },
1449 	{ .type = PF_RAT_NEG, .off = _OUT(neg), .cb = snl_attr_get_uint8 },
1450 	{ .type = PF_RAT_OP, .off = _OUT(port_op), .cb = snl_attr_get_uint8 },
1451 };
1452 #undef _OUT
1453 SNL_DECLARE_ATTR_PARSER(rule_addr_parser, ap_rule_addr);
1454 
1455 struct snl_parsed_labels
1456 {
1457 	char		labels[PF_RULE_MAX_LABEL_COUNT][PF_RULE_LABEL_SIZE];
1458 	uint32_t	i;
1459 };
1460 
1461 static bool
snl_attr_get_pf_rule_labels(struct snl_state * ss,struct nlattr * nla,const void * arg __unused,void * target)1462 snl_attr_get_pf_rule_labels(struct snl_state *ss, struct nlattr *nla,
1463     const void *arg __unused, void *target)
1464 {
1465 	struct snl_parsed_labels *l = (struct snl_parsed_labels *)target;
1466 	bool ret;
1467 
1468 	if (l->i >= PF_RULE_MAX_LABEL_COUNT)
1469 		return (E2BIG);
1470 
1471 	ret = snl_attr_copy_string(ss, nla, (void *)PF_RULE_LABEL_SIZE,
1472 	    l->labels[l->i]);
1473 	if (ret)
1474 		l->i++;
1475 
1476 	return (ret);
1477 }
1478 
1479 #define _OUT(_field)	offsetof(struct nl_parsed_labels, _field)
1480 static const struct snl_attr_parser ap_labels[] = {
1481 	{ .type = PF_LT_LABEL, .off = 0, .cb = snl_attr_get_pf_rule_labels },
1482 };
1483 SNL_DECLARE_ATTR_PARSER(rule_labels_parser, ap_labels);
1484 #undef _OUT
1485 
1486 static bool
snl_attr_get_nested_pf_rule_labels(struct snl_state * ss,struct nlattr * nla,const void * arg __unused,void * target)1487 snl_attr_get_nested_pf_rule_labels(struct snl_state *ss, struct nlattr *nla,
1488     const void *arg __unused, void *target)
1489 {
1490 	struct snl_parsed_labels parsed_labels = { };
1491 	bool error;
1492 
1493 	/* Assumes target points to the beginning of the structure */
1494 	error = snl_parse_header(ss, NLA_DATA(nla), NLA_DATA_LEN(nla), &rule_labels_parser, &parsed_labels);
1495 	if (! error)
1496 		return (error);
1497 
1498 	memcpy(target, parsed_labels.labels, sizeof(parsed_labels.labels));
1499 
1500 	return (true);
1501 }
1502 
1503 #define _OUT(_field)	offsetof(struct pf_mape_portset, _field)
1504 static const struct snl_attr_parser ap_mape_portset[] = {
1505 	{ .type = PF_MET_OFFSET, .off = _OUT(offset), .cb = snl_attr_get_uint8 },
1506 	{ .type = PF_MET_PSID_LEN, .off = _OUT(psidlen), .cb = snl_attr_get_uint8 },
1507 	{. type = PF_MET_PSID, .off = _OUT(psid), .cb = snl_attr_get_uint16 },
1508 };
1509 SNL_DECLARE_ATTR_PARSER(mape_portset_parser, ap_mape_portset);
1510 #undef _OUT
1511 
1512 #define _OUT(_field)	offsetof(struct pfctl_pool, _field)
1513 static const struct snl_attr_parser ap_pool[] = {
1514 	{ .type = PF_PT_KEY, .off = _OUT(key), .arg = (void *)sizeof(struct pf_poolhashkey), .cb = snl_attr_get_bytes },
1515 	{ .type = PF_PT_COUNTER, .off = _OUT(counter), .cb = snl_attr_get_in6_addr },
1516 	{ .type = PF_PT_TBLIDX, .off = _OUT(tblidx), .cb = snl_attr_get_uint32 },
1517 	{ .type = PF_PT_PROXY_SRC_PORT, .off = _OUT(proxy_port[0]), .cb = snl_attr_get_uint16 },
1518 	{ .type = PF_PT_PROXY_DST_PORT, .off = _OUT(proxy_port[1]), .cb = snl_attr_get_uint16 },
1519 	{ .type = PF_PT_OPTS, .off = _OUT(opts), .cb = snl_attr_get_uint8 },
1520 	{ .type = PF_PT_MAPE, .off = _OUT(mape), .arg = &mape_portset_parser, .cb = snl_attr_get_nested },
1521 };
1522 SNL_DECLARE_ATTR_PARSER(pool_parser, ap_pool);
1523 #undef _OUT
1524 
1525 struct nl_parsed_timeouts
1526 {
1527 	uint32_t	timeouts[PFTM_MAX];
1528 	uint32_t	i;
1529 };
1530 
1531 static bool
snl_attr_get_pf_timeout(struct snl_state * ss,struct nlattr * nla,const void * arg __unused,void * target)1532 snl_attr_get_pf_timeout(struct snl_state *ss, struct nlattr *nla,
1533     const void *arg __unused, void *target)
1534 {
1535 	struct nl_parsed_timeouts *t = (struct nl_parsed_timeouts *)target;
1536 	bool ret;
1537 
1538 	if (t->i >= PFTM_MAX)
1539 		return (E2BIG);
1540 
1541 	ret = snl_attr_get_uint32(ss, nla, NULL, &t->timeouts[t->i]);
1542 	if (ret)
1543 		t->i++;
1544 
1545 	return (ret);
1546 }
1547 
1548 #define _OUT(_field)	offsetof(struct nl_parsed_timeout, _field)
1549 static const struct snl_attr_parser ap_timeouts[] = {
1550 	{ .type = PF_TT_TIMEOUT, .off = 0, .cb = snl_attr_get_pf_timeout },
1551 };
1552 SNL_DECLARE_ATTR_PARSER(timeout_parser, ap_timeouts);
1553 #undef _OUT
1554 
1555 static bool
snl_attr_get_nested_timeouts(struct snl_state * ss,struct nlattr * nla,const void * arg __unused,void * target)1556 snl_attr_get_nested_timeouts(struct snl_state *ss, struct nlattr *nla,
1557     const void *arg __unused, void *target)
1558 {
1559 	struct nl_parsed_timeouts parsed_timeouts = { };
1560 	bool error;
1561 
1562 	/* Assumes target points to the beginning of the structure */
1563 	error = snl_parse_header(ss, NLA_DATA(nla), NLA_DATA_LEN(nla), &timeout_parser, &parsed_timeouts);
1564 	if (! error)
1565 		return (error);
1566 
1567 	memcpy(target, parsed_timeouts.timeouts, sizeof(parsed_timeouts.timeouts));
1568 
1569 	return (true);
1570 }
1571 
1572 #define _OUT(_field)	offsetof(struct pf_rule_uid, _field)
1573 static const struct snl_attr_parser ap_rule_uid[] = {
1574 	{ .type = PF_RUT_UID_LOW, .off = _OUT(uid[0]), .cb = snl_attr_get_uint32 },
1575 	{ .type = PF_RUT_UID_HIGH, .off = _OUT(uid[1]), .cb = snl_attr_get_uint32 },
1576 	{ .type = PF_RUT_OP, .off = _OUT(op), .cb = snl_attr_get_uint8 },
1577 };
1578 SNL_DECLARE_ATTR_PARSER(rule_uid_parser, ap_rule_uid);
1579 #undef _OUT
1580 
1581 struct pfctl_nl_get_rule {
1582 	struct pfctl_rule r;
1583 	char anchor_call[MAXPATHLEN];
1584 };
1585 #define	_OUT(_field)	offsetof(struct pfctl_nl_get_rule, _field)
1586 static struct snl_attr_parser ap_getrule[] = {
1587 	{ .type = PF_RT_SRC, .off = _OUT(r.src), .arg = &rule_addr_parser,.cb = snl_attr_get_nested },
1588 	{ .type = PF_RT_DST, .off = _OUT(r.dst), .arg = &rule_addr_parser,.cb = snl_attr_get_nested },
1589 	{ .type = PF_RT_RIDENTIFIER, .off = _OUT(r.ridentifier), .cb = snl_attr_get_uint32 },
1590 	{ .type = PF_RT_LABELS, .off = _OUT(r.label), .arg = &rule_labels_parser,.cb = snl_attr_get_nested_pf_rule_labels },
1591 	{ .type = PF_RT_IFNAME, .off = _OUT(r.ifname), .arg = (void *)IFNAMSIZ, .cb = snl_attr_copy_string },
1592 	{ .type = PF_RT_QNAME, .off = _OUT(r.qname), .arg = (void *)PF_QNAME_SIZE, .cb = snl_attr_copy_string },
1593 	{ .type = PF_RT_PQNAME, .off = _OUT(r.pqname), .arg = (void *)PF_QNAME_SIZE, .cb = snl_attr_copy_string },
1594 	{ .type = PF_RT_TAGNAME, .off = _OUT(r.tagname), .arg = (void *)PF_TAG_NAME_SIZE, .cb = snl_attr_copy_string },
1595 	{ .type = PF_RT_MATCH_TAGNAME, .off = _OUT(r.match_tagname), .arg = (void *)PF_TAG_NAME_SIZE, .cb = snl_attr_copy_string },
1596 	{ .type = PF_RT_OVERLOAD_TBLNAME, .off = _OUT(r.overload_tblname), .arg = (void *)PF_TABLE_NAME_SIZE, .cb = snl_attr_copy_string },
1597 	{ .type = PF_RT_RPOOL, .off = _OUT(r.rpool), .arg = &pool_parser, .cb = snl_attr_get_nested },
1598 	{ .type = PF_RT_OS_FINGERPRINT, .off = _OUT(r.os_fingerprint), .cb = snl_attr_get_uint32 },
1599 	{ .type = PF_RT_RTABLEID, .off = _OUT(r.rtableid), .cb = snl_attr_get_uint32 },
1600 	{ .type = PF_RT_TIMEOUT, .off = _OUT(r.timeout), .arg = &timeout_parser, .cb = snl_attr_get_nested_timeouts },
1601 	{ .type = PF_RT_MAX_STATES, .off = _OUT(r.max_states), .cb = snl_attr_get_uint32 },
1602 	{ .type = PF_RT_MAX_SRC_NODES, .off = _OUT(r.max_src_nodes), .cb = snl_attr_get_uint32 },
1603 	{ .type = PF_RT_MAX_SRC_STATES, .off = _OUT(r.max_src_states), .cb = snl_attr_get_uint32 },
1604 	{ .type = PF_RT_MAX_SRC_CONN_RATE_LIMIT, .off = _OUT(r.max_src_conn_rate.limit), .cb = snl_attr_get_uint32 },
1605 	{ .type = PF_RT_MAX_SRC_CONN_RATE_SECS, .off = _OUT(r.max_src_conn_rate.seconds), .cb = snl_attr_get_uint32 },
1606 	{ .type = PF_RT_DNPIPE, .off = _OUT(r.dnpipe), .cb = snl_attr_get_uint16 },
1607 	{ .type = PF_RT_DNRPIPE, .off = _OUT(r.dnrpipe), .cb = snl_attr_get_uint16 },
1608 	{ .type = PF_RT_DNFLAGS, .off = _OUT(r.free_flags), .cb = snl_attr_get_uint32 },
1609 	{ .type = PF_RT_NR, .off = _OUT(r.nr), .cb = snl_attr_get_uint32 },
1610 	{ .type = PF_RT_PROB, .off = _OUT(r.prob), .cb = snl_attr_get_uint32 },
1611 	{ .type = PF_RT_CUID, .off = _OUT(r.cuid), .cb = snl_attr_get_uint32 },
1612 	{. type = PF_RT_CPID, .off = _OUT(r.cpid), .cb = snl_attr_get_uint32 },
1613 	{ .type = PF_RT_RETURN_ICMP, .off = _OUT(r.return_icmp), .cb = snl_attr_get_uint16 },
1614 	{ .type = PF_RT_RETURN_ICMP6, .off = _OUT(r.return_icmp6), .cb = snl_attr_get_uint16 },
1615 	{ .type = PF_RT_MAX_MSS, .off = _OUT(r.max_mss), .cb = snl_attr_get_uint16 },
1616 	{ .type = PF_RT_SCRUB_FLAGS, .off = _OUT(r.scrub_flags), .cb = snl_attr_get_uint16 },
1617 	{ .type = PF_RT_UID, .off = _OUT(r.uid), .arg = &rule_uid_parser, .cb = snl_attr_get_nested },
1618 	{ .type = PF_RT_GID, .off = _OUT(r.gid), .arg = &rule_uid_parser, .cb = snl_attr_get_nested },
1619 	{ .type = PF_RT_RULE_FLAG, .off = _OUT(r.rule_flag), .cb = snl_attr_get_uint32 },
1620 	{ .type = PF_RT_ACTION, .off = _OUT(r.action), .cb = snl_attr_get_uint8 },
1621 	{ .type = PF_RT_DIRECTION, .off = _OUT(r.direction), .cb = snl_attr_get_uint8 },
1622 	{ .type = PF_RT_LOG, .off = _OUT(r.log), .cb = snl_attr_get_uint8 },
1623 	{ .type = PF_RT_LOGIF, .off = _OUT(r.logif), .cb = snl_attr_get_uint8 },
1624 	{ .type = PF_RT_QUICK, .off = _OUT(r.quick), .cb = snl_attr_get_uint8 },
1625 	{ .type = PF_RT_IF_NOT, .off = _OUT(r.ifnot), .cb = snl_attr_get_uint8 },
1626 	{ .type = PF_RT_MATCH_TAG_NOT, .off = _OUT(r.match_tag_not), .cb = snl_attr_get_uint8 },
1627 	{ .type = PF_RT_NATPASS, .off = _OUT(r.natpass), .cb = snl_attr_get_uint8 },
1628 	{ .type = PF_RT_KEEP_STATE, .off = _OUT(r.keep_state), .cb = snl_attr_get_uint8 },
1629 	{ .type = PF_RT_AF, .off = _OUT(r.af), .cb = snl_attr_get_uint8 },
1630 	{ .type = PF_RT_PROTO, .off = _OUT(r.proto), .cb = snl_attr_get_uint8 },
1631 	{ .type = PF_RT_TYPE, .off = _OUT(r.type), .cb = snl_attr_get_uint8 },
1632 	{ .type = PF_RT_CODE, .off = _OUT(r.code), .cb = snl_attr_get_uint8 },
1633 	{ .type = PF_RT_FLAGS, .off = _OUT(r.flags), .cb = snl_attr_get_uint8 },
1634 	{ .type = PF_RT_FLAGSET, .off = _OUT(r.flagset), .cb = snl_attr_get_uint8 },
1635 	{ .type = PF_RT_MIN_TTL, .off = _OUT(r.min_ttl), .cb = snl_attr_get_uint8 },
1636 	{ .type = PF_RT_ALLOW_OPTS, .off = _OUT(r.allow_opts), .cb = snl_attr_get_uint8 },
1637 	{ .type = PF_RT_RT, .off = _OUT(r.rt), .cb = snl_attr_get_uint8 },
1638 	{ .type = PF_RT_RETURN_TTL, .off = _OUT(r.return_ttl), .cb = snl_attr_get_uint8 },
1639 	{ .type = PF_RT_TOS, .off = _OUT(r.tos), .cb = snl_attr_get_uint8 },
1640 	{ .type = PF_RT_SET_TOS, .off = _OUT(r.set_tos), .cb = snl_attr_get_uint8 },
1641 	{ .type = PF_RT_ANCHOR_RELATIVE, .off = _OUT(r.anchor_relative), .cb = snl_attr_get_uint8 },
1642 	{ .type = PF_RT_ANCHOR_WILDCARD, .off = _OUT(r.anchor_wildcard), .cb = snl_attr_get_uint8 },
1643 	{ .type = PF_RT_FLUSH, .off = _OUT(r.flush), .cb = snl_attr_get_uint8 },
1644 	{ .type = PF_RT_PRIO, .off = _OUT(r.prio), .cb = snl_attr_get_uint8 },
1645 	{ .type = PF_RT_SET_PRIO, .off = _OUT(r.set_prio[0]), .cb = snl_attr_get_uint8 },
1646 	{ .type = PF_RT_SET_PRIO_REPLY, .off = _OUT(r.set_prio[1]), .cb = snl_attr_get_uint8 },
1647 	{ .type = PF_RT_DIVERT_ADDRESS, .off = _OUT(r.divert.addr), .cb = snl_attr_get_in6_addr },
1648 	{ .type = PF_RT_DIVERT_PORT, .off = _OUT(r.divert.port), .cb = snl_attr_get_uint16 },
1649 	{ .type = PF_RT_PACKETS_IN, .off = _OUT(r.packets[0]), .cb = snl_attr_get_uint64 },
1650 	{ .type = PF_RT_PACKETS_OUT, .off = _OUT(r.packets[1]), .cb = snl_attr_get_uint64 },
1651 	{ .type = PF_RT_BYTES_IN, .off = _OUT(r.bytes[0]), .cb = snl_attr_get_uint64 },
1652 	{ .type = PF_RT_BYTES_OUT, .off = _OUT(r.bytes[1]), .cb = snl_attr_get_uint64 },
1653 	{ .type = PF_RT_EVALUATIONS, .off = _OUT(r.evaluations), .cb = snl_attr_get_uint64 },
1654 	{ .type = PF_RT_TIMESTAMP, .off = _OUT(r.last_active_timestamp), .cb = snl_attr_get_uint64 },
1655 	{ .type = PF_RT_STATES_CUR, .off = _OUT(r.states_cur), .cb = snl_attr_get_uint64 },
1656 	{ .type = PF_RT_STATES_TOTAL, .off = _OUT(r.states_tot), .cb = snl_attr_get_uint64 },
1657 	{ .type = PF_RT_SRC_NODES, .off = _OUT(r.src_nodes), .cb = snl_attr_get_uint64 },
1658 	{ .type = PF_RT_ANCHOR_CALL, .off = _OUT(anchor_call), .arg = (void*)MAXPATHLEN, .cb = snl_attr_copy_string },
1659 };
1660 static struct snl_field_parser fp_getrule[] = {};
1661 #undef _OUT
1662 SNL_DECLARE_PARSER(getrule_parser, struct genlmsghdr, fp_getrule, ap_getrule);
1663 
1664 int
pfctl_get_clear_rule_h(struct pfctl_handle * h,uint32_t nr,uint32_t ticket,const char * anchor,uint32_t ruleset,struct pfctl_rule * rule,char * anchor_call,bool clear)1665 pfctl_get_clear_rule_h(struct pfctl_handle *h, uint32_t nr, uint32_t ticket,
1666     const char *anchor, uint32_t ruleset, struct pfctl_rule *rule,
1667     char *anchor_call, bool clear)
1668 {
1669 	struct pfctl_nl_get_rule attrs = {};
1670 	struct snl_errmsg_data e = {};
1671 	struct nlmsghdr *hdr;
1672 	struct snl_writer nw;
1673 	uint32_t seq_id;
1674 	int family_id;
1675 
1676 	family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
1677 	if (family_id == 0)
1678 		return (ENOTSUP);
1679 
1680 	snl_init_writer(&h->ss, &nw);
1681 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_GETRULE);
1682 	hdr->nlmsg_flags |= NLM_F_DUMP;
1683 
1684 	snl_add_msg_attr_string(&nw, PF_GR_ANCHOR, anchor);
1685 	snl_add_msg_attr_u8(&nw, PF_GR_ACTION, ruleset);
1686 	snl_add_msg_attr_u32(&nw, PF_GR_NR, nr);
1687 	snl_add_msg_attr_u32(&nw, PF_GR_TICKET, ticket);
1688 	snl_add_msg_attr_u8(&nw, PF_GR_CLEAR, clear);
1689 
1690 	hdr = snl_finalize_msg(&nw);
1691 	if (hdr == NULL)
1692 		return (ENOMEM);
1693 
1694 	seq_id = hdr->nlmsg_seq;
1695 	if (! snl_send_message(&h->ss, hdr))
1696 		return (ENXIO);
1697 
1698 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
1699 		if (! snl_parse_nlmsg(&h->ss, hdr, &getrule_parser, &attrs))
1700 			continue;
1701 	}
1702 
1703 	memcpy(rule, &attrs.r, sizeof(attrs.r));
1704 	strlcpy(anchor_call, attrs.anchor_call, MAXPATHLEN);
1705 
1706 	return (e.error);
1707 }
1708 
1709 int
pfctl_get_clear_rule(int dev,uint32_t nr,uint32_t ticket,const char * anchor,uint32_t ruleset,struct pfctl_rule * rule,char * anchor_call,bool clear)1710 pfctl_get_clear_rule(int dev, uint32_t nr, uint32_t ticket,
1711 	    const char *anchor, uint32_t ruleset, struct pfctl_rule *rule,
1712 	    char *anchor_call, bool clear)
1713 {
1714 	nvlist_t *nvl;
1715 	int ret;
1716 
1717 	nvl = nvlist_create(0);
1718 	if (nvl == 0)
1719 		return (ENOMEM);
1720 
1721 	nvlist_add_number(nvl, "nr", nr);
1722 	nvlist_add_number(nvl, "ticket", ticket);
1723 	nvlist_add_string(nvl, "anchor", anchor);
1724 	nvlist_add_number(nvl, "ruleset", ruleset);
1725 
1726 	if (clear)
1727 		nvlist_add_bool(nvl, "clear_counter", true);
1728 
1729 	if ((ret = pfctl_do_ioctl(dev, DIOCGETRULENV, 8192, &nvl)) != 0)
1730 		goto out;
1731 
1732 	pf_nvrule_to_rule(nvlist_get_nvlist(nvl, "rule"), rule);
1733 
1734 	if (anchor_call)
1735 		strlcpy(anchor_call, nvlist_get_string(nvl, "anchor_call"),
1736 		    MAXPATHLEN);
1737 
1738 out:
1739 	nvlist_destroy(nvl);
1740 	return (ret);
1741 }
1742 
1743 int
pfctl_set_keepcounters(int dev,bool keep)1744 pfctl_set_keepcounters(int dev, bool keep)
1745 {
1746 	struct pfioc_nv	 nv;
1747 	nvlist_t	*nvl;
1748 	int		 ret;
1749 
1750 	nvl = nvlist_create(0);
1751 
1752 	nvlist_add_bool(nvl, "keep_counters", keep);
1753 
1754 	nv.data = nvlist_pack(nvl, &nv.len);
1755 	nv.size = nv.len;
1756 
1757 	nvlist_destroy(nvl);
1758 
1759 	ret = ioctl(dev, DIOCKEEPCOUNTERS, &nv);
1760 
1761 	free(nv.data);
1762 	return (ret);
1763 }
1764 
1765 struct pfctl_creator {
1766 	uint32_t id;
1767 };
1768 #define	_IN(_field)	offsetof(struct genlmsghdr, _field)
1769 #define	_OUT(_field)	offsetof(struct pfctl_creator, _field)
1770 static struct snl_attr_parser ap_creators[] = {
1771 	{ .type = PF_ST_CREATORID, .off = _OUT(id), .cb = snl_attr_get_uint32 },
1772 };
1773 static struct snl_field_parser fp_creators[] = {
1774 };
1775 #undef _IN
1776 #undef _OUT
1777 SNL_DECLARE_PARSER(creator_parser, struct genlmsghdr, fp_creators, ap_creators);
1778 
1779 static int
pfctl_get_creators_nl(struct snl_state * ss,uint32_t * creators,size_t * len)1780 pfctl_get_creators_nl(struct snl_state *ss, uint32_t *creators, size_t *len)
1781 {
1782 
1783 	int family_id = snl_get_genl_family(ss, PFNL_FAMILY_NAME);
1784 	size_t i = 0;
1785 
1786 	struct nlmsghdr *hdr;
1787 	struct snl_writer nw;
1788 
1789 	if (family_id == 0)
1790 		return (ENOTSUP);
1791 
1792 	snl_init_writer(ss, &nw);
1793 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_GETCREATORS);
1794 	hdr->nlmsg_flags |= NLM_F_DUMP;
1795 	hdr = snl_finalize_msg(&nw);
1796 	if (hdr == NULL)
1797 		return (ENOMEM);
1798 	uint32_t seq_id = hdr->nlmsg_seq;
1799 
1800 	snl_send_message(ss, hdr);
1801 
1802 	struct snl_errmsg_data e = {};
1803 	while ((hdr = snl_read_reply_multi(ss, seq_id, &e)) != NULL) {
1804 		struct pfctl_creator c;
1805 		bzero(&c, sizeof(c));
1806 
1807 		if (!snl_parse_nlmsg(ss, hdr, &creator_parser, &c))
1808 			continue;
1809 
1810 		creators[i] = c.id;
1811 		i++;
1812 		if (i > *len)
1813 			return (E2BIG);
1814 	}
1815 
1816 	*len = i;
1817 
1818 	return (0);
1819 }
1820 
1821 int
pfctl_get_creatorids(struct pfctl_handle * h,uint32_t * creators,size_t * len)1822 pfctl_get_creatorids(struct pfctl_handle *h, uint32_t *creators, size_t *len)
1823 {
1824 	int error;
1825 
1826 	error = pfctl_get_creators_nl(&h->ss, creators, len);
1827 
1828 	return (error);
1829 }
1830 
1831 static inline bool
snl_attr_get_pfaddr(struct snl_state * ss __unused,struct nlattr * nla,const void * arg __unused,void * target)1832 snl_attr_get_pfaddr(struct snl_state *ss __unused, struct nlattr *nla,
1833     const void *arg __unused, void *target)
1834 {
1835 	memcpy(target, NLA_DATA(nla), NLA_DATA_LEN(nla));
1836 	return (true);
1837 }
1838 
1839 static inline bool
snl_attr_store_ifname(struct snl_state * ss __unused,struct nlattr * nla,const void * arg __unused,void * target)1840 snl_attr_store_ifname(struct snl_state *ss __unused, struct nlattr *nla,
1841     const void *arg __unused, void *target)
1842 {
1843 	size_t maxlen = NLA_DATA_LEN(nla);
1844 
1845 	if (strnlen((char *)NLA_DATA(nla), maxlen) < maxlen) {
1846 		strlcpy(target, (char *)NLA_DATA(nla), maxlen);
1847 		return (true);
1848 	}
1849 	return (false);
1850 }
1851 
1852 #define	_OUT(_field)	offsetof(struct pfctl_state_peer, _field)
1853 static const struct snl_attr_parser nla_p_speer[] = {
1854 	{ .type = PF_STP_SEQLO, .off = _OUT(seqlo), .cb = snl_attr_get_uint32 },
1855 	{ .type = PF_STP_SEQHI, .off = _OUT(seqhi), .cb = snl_attr_get_uint32 },
1856 	{ .type = PF_STP_SEQDIFF, .off = _OUT(seqdiff), .cb = snl_attr_get_uint32 },
1857 	{ .type = PF_STP_STATE, .off = _OUT(state), .cb = snl_attr_get_uint8 },
1858 	{ .type = PF_STP_WSCALE, .off = _OUT(wscale), .cb = snl_attr_get_uint8 },
1859 };
1860 SNL_DECLARE_ATTR_PARSER(speer_parser, nla_p_speer);
1861 #undef _OUT
1862 
1863 #define	_OUT(_field)	offsetof(struct pf_state_key_export, _field)
1864 static const struct snl_attr_parser nla_p_skey[] = {
1865 	{ .type = PF_STK_ADDR0, .off = _OUT(addr[0]), .cb = snl_attr_get_pfaddr },
1866 	{ .type = PF_STK_ADDR1, .off = _OUT(addr[1]), .cb = snl_attr_get_pfaddr },
1867 	{ .type = PF_STK_PORT0, .off = _OUT(port[0]), .cb = snl_attr_get_uint16 },
1868 	{ .type = PF_STK_PORT1, .off = _OUT(port[1]), .cb = snl_attr_get_uint16 },
1869 };
1870 SNL_DECLARE_ATTR_PARSER(skey_parser, nla_p_skey);
1871 #undef _OUT
1872 
1873 #define	_IN(_field)	offsetof(struct genlmsghdr, _field)
1874 #define	_OUT(_field)	offsetof(struct pfctl_state, _field)
1875 static struct snl_attr_parser ap_state[] = {
1876 	{ .type = PF_ST_ID, .off = _OUT(id), .cb = snl_attr_get_uint64 },
1877 	{ .type = PF_ST_CREATORID, .off = _OUT(creatorid), .cb = snl_attr_get_uint32 },
1878 	{ .type = PF_ST_IFNAME, .off = _OUT(ifname), .cb = snl_attr_store_ifname },
1879 	{ .type = PF_ST_ORIG_IFNAME, .off = _OUT(orig_ifname), .cb = snl_attr_store_ifname },
1880 	{ .type = PF_ST_KEY_WIRE, .off = _OUT(key[0]), .arg = &skey_parser, .cb = snl_attr_get_nested },
1881 	{ .type = PF_ST_KEY_STACK, .off = _OUT(key[1]), .arg = &skey_parser, .cb = snl_attr_get_nested },
1882 	{ .type = PF_ST_PEER_SRC, .off = _OUT(src), .arg = &speer_parser, .cb = snl_attr_get_nested },
1883 	{ .type = PF_ST_PEER_DST, .off = _OUT(dst), .arg = &speer_parser, .cb = snl_attr_get_nested },
1884 	{ .type = PF_ST_RT_ADDR, .off = _OUT(rt_addr), .cb = snl_attr_get_pfaddr },
1885 	{ .type = PF_ST_RULE, .off = _OUT(rule), .cb = snl_attr_get_uint32 },
1886 	{ .type = PF_ST_ANCHOR, .off = _OUT(anchor), .cb = snl_attr_get_uint32 },
1887 	{ .type = PF_ST_NAT_RULE, .off = _OUT(nat_rule), .cb = snl_attr_get_uint32 },
1888 	{ .type = PF_ST_CREATION, .off = _OUT(creation), .cb = snl_attr_get_uint32 },
1889 	{ .type = PF_ST_EXPIRE, .off = _OUT(expire), .cb = snl_attr_get_uint32 },
1890 	{ .type = PF_ST_PACKETS0, .off = _OUT(packets[0]), .cb = snl_attr_get_uint64 },
1891 	{ .type = PF_ST_PACKETS1, .off = _OUT(packets[1]), .cb = snl_attr_get_uint64 },
1892 	{ .type = PF_ST_BYTES0, .off = _OUT(bytes[0]), .cb = snl_attr_get_uint64 },
1893 	{ .type = PF_ST_BYTES1, .off = _OUT(bytes[1]), .cb = snl_attr_get_uint64 },
1894 	{ .type = PF_ST_AF, .off = _OUT(key[0].af), .cb = snl_attr_get_uint8 },
1895 	{ .type = PF_ST_PROTO, .off = _OUT(key[0].proto), .cb = snl_attr_get_uint8 },
1896 	{ .type = PF_ST_DIRECTION, .off = _OUT(direction), .cb = snl_attr_get_uint8 },
1897 	{ .type = PF_ST_LOG, .off = _OUT(log), .cb = snl_attr_get_uint8 },
1898 	{ .type = PF_ST_STATE_FLAGS, .off = _OUT(state_flags), .cb = snl_attr_get_uint16 },
1899 	{ .type = PF_ST_SYNC_FLAGS, .off = _OUT(sync_flags), .cb = snl_attr_get_uint8 },
1900 	{ .type = PF_ST_RTABLEID, .off = _OUT(rtableid), .cb = snl_attr_get_int32 },
1901 	{ .type = PF_ST_MIN_TTL, .off = _OUT(min_ttl), .cb = snl_attr_get_uint8 },
1902 	{ .type = PF_ST_MAX_MSS, .off = _OUT(max_mss), .cb = snl_attr_get_uint16 },
1903 	{ .type = PF_ST_DNPIPE, .off = _OUT(dnpipe), .cb = snl_attr_get_uint16 },
1904 	{ .type = PF_ST_DNRPIPE, .off = _OUT(dnrpipe), .cb = snl_attr_get_uint16 },
1905 	{ .type = PF_ST_RT, .off = _OUT(rt), .cb = snl_attr_get_uint8 },
1906 	{ .type = PF_ST_RT_IFNAME, .off = _OUT(rt_ifname), .cb = snl_attr_store_ifname },
1907 };
1908 static struct snl_field_parser fp_state[] = {
1909 };
1910 #undef _IN
1911 #undef _OUT
1912 SNL_DECLARE_PARSER(state_parser, struct genlmsghdr, fp_state, ap_state);
1913 
1914 static const struct snl_hdr_parser *all_parsers[] = {
1915 	&state_parser, &skey_parser, &speer_parser,
1916 	&creator_parser, &getrules_parser
1917 };
1918 
1919 static int
pfctl_get_states_nl(struct pfctl_state_filter * filter,struct snl_state * ss,pfctl_get_state_fn f,void * arg)1920 pfctl_get_states_nl(struct pfctl_state_filter *filter, struct snl_state *ss, pfctl_get_state_fn f, void *arg)
1921 {
1922 	SNL_VERIFY_PARSERS(all_parsers);
1923 	int family_id = snl_get_genl_family(ss, PFNL_FAMILY_NAME);
1924 	int ret;
1925 
1926 	struct nlmsghdr *hdr;
1927 	struct snl_writer nw;
1928 
1929 	if (family_id == 0)
1930 		return (ENOTSUP);
1931 
1932 	snl_init_writer(ss, &nw);
1933 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_GETSTATES);
1934 	hdr->nlmsg_flags |= NLM_F_DUMP;
1935 	snl_add_msg_attr_string(&nw, PF_ST_IFNAME, filter->ifname);
1936 	snl_add_msg_attr_u16(&nw, PF_ST_PROTO, filter->proto);
1937 	snl_add_msg_attr_u8(&nw, PF_ST_AF, filter->af);
1938 	snl_add_msg_attr_ip6(&nw, PF_ST_FILTER_ADDR, &filter->addr.v6);
1939 	snl_add_msg_attr_ip6(&nw, PF_ST_FILTER_MASK, &filter->mask.v6);
1940 
1941 	hdr = snl_finalize_msg(&nw);
1942 	if (hdr == NULL)
1943 		return (ENOMEM);
1944 
1945 	uint32_t seq_id = hdr->nlmsg_seq;
1946 
1947 	snl_send_message(ss, hdr);
1948 
1949 	struct snl_errmsg_data e = {};
1950 	while ((hdr = snl_read_reply_multi(ss, seq_id, &e)) != NULL) {
1951 		struct pfctl_state s;
1952 		bzero(&s, sizeof(s));
1953 		if (!snl_parse_nlmsg(ss, hdr, &state_parser, &s))
1954 			continue;
1955 
1956 		s.key[1].af = s.key[0].af;
1957 		s.key[1].proto = s.key[0].proto;
1958 
1959 		ret = f(&s, arg);
1960 		if (ret != 0)
1961 			return (ret);
1962 	}
1963 
1964 	return (0);
1965 }
1966 
1967 int
pfctl_get_states_iter(pfctl_get_state_fn f,void * arg)1968 pfctl_get_states_iter(pfctl_get_state_fn f, void *arg)
1969 {
1970 	struct pfctl_state_filter filter = {};
1971 	return (pfctl_get_filtered_states_iter(&filter, f, arg));
1972 }
1973 
1974 int
pfctl_get_filtered_states_iter(struct pfctl_state_filter * filter,pfctl_get_state_fn f,void * arg)1975 pfctl_get_filtered_states_iter(struct pfctl_state_filter *filter, pfctl_get_state_fn f, void *arg)
1976 {
1977 	struct snl_state ss = {};
1978 	int error;
1979 
1980 	snl_init(&ss, NETLINK_GENERIC);
1981 	error = pfctl_get_states_nl(filter, &ss, f, arg);
1982 	snl_free(&ss);
1983 
1984 	return (error);
1985 }
1986 
1987 static int
pfctl_append_states(struct pfctl_state * s,void * arg)1988 pfctl_append_states(struct pfctl_state *s, void *arg)
1989 {
1990 	struct pfctl_state *new;
1991 	struct pfctl_states *states = (struct pfctl_states *)arg;
1992 
1993 	new = malloc(sizeof(*s));
1994 	if (new == NULL)
1995 		return (ENOMEM);
1996 
1997 	memcpy(new, s, sizeof(*s));
1998 
1999 	TAILQ_INSERT_TAIL(&states->states, new, entry);
2000 
2001 	return (0);
2002 }
2003 
2004 int
pfctl_get_states(int dev __unused,struct pfctl_states * states)2005 pfctl_get_states(int dev __unused, struct pfctl_states *states)
2006 {
2007 	int ret;
2008 
2009 	bzero(states, sizeof(*states));
2010 	TAILQ_INIT(&states->states);
2011 
2012 	ret = pfctl_get_states_iter(pfctl_append_states, states);
2013 	if (ret != 0) {
2014 		pfctl_free_states(states);
2015 		return (ret);
2016 	}
2017 
2018 	return (0);
2019 }
2020 
2021 void
pfctl_free_states(struct pfctl_states * states)2022 pfctl_free_states(struct pfctl_states *states)
2023 {
2024 	struct pfctl_state *s, *tmp;
2025 
2026 	TAILQ_FOREACH_SAFE(s, &states->states, entry, tmp) {
2027 		free(s);
2028 	}
2029 
2030 	bzero(states, sizeof(*states));
2031 }
2032 
2033 struct pfctl_nl_clear_states {
2034 	uint32_t killed;
2035 };
2036 #define	_OUT(_field)	offsetof(struct pfctl_nl_clear_states, _field)
2037 static struct snl_attr_parser ap_clear_states[] = {
2038 	{ .type = PF_CS_KILLED, .off = _OUT(killed), .cb = snl_attr_get_uint32 },
2039 };
2040 static struct snl_field_parser fp_clear_states[] = {};
2041 #undef _OUT
2042 SNL_DECLARE_PARSER(clear_states_parser, struct genlmsghdr, fp_clear_states, ap_clear_states);
2043 
2044 static int
_pfctl_clear_states_h(struct pfctl_handle * h,const struct pfctl_kill * kill,unsigned int * killed,int cmd)2045 _pfctl_clear_states_h(struct pfctl_handle *h, const struct pfctl_kill *kill,
2046     unsigned int *killed, int cmd)
2047 {
2048 	struct snl_writer nw;
2049 	struct snl_errmsg_data e = {};
2050 	struct pfctl_nl_clear_states attrs = {};
2051 	struct nlmsghdr *hdr;
2052 	uint32_t seq_id;
2053 	int family_id;
2054 
2055 	family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
2056 	if (family_id == 0)
2057 		return (ENOTSUP);
2058 
2059 	snl_init_writer(&h->ss, &nw);
2060 	hdr = snl_create_genl_msg_request(&nw, family_id, cmd);
2061 	hdr->nlmsg_flags |= NLM_F_DUMP;
2062 
2063 	snl_add_msg_attr_u64(&nw, PF_CS_CMP_ID, kill->cmp.id);
2064 	snl_add_msg_attr_u32(&nw, PF_CS_CMP_CREATORID, htonl(kill->cmp.creatorid));
2065 	snl_add_msg_attr_u8(&nw, PF_CS_CMP_DIR, kill->cmp.direction);
2066 	snl_add_msg_attr_u8(&nw, PF_CS_AF, kill->af);
2067 	snl_add_msg_attr_u8(&nw, PF_CS_PROTO, kill->proto);
2068 	snl_add_msg_attr_rule_addr(&nw, PF_CS_SRC, &kill->src);
2069 	snl_add_msg_attr_rule_addr(&nw, PF_CS_DST, &kill->dst);
2070 	snl_add_msg_attr_rule_addr(&nw, PF_CS_RT_ADDR, &kill->rt_addr);
2071 	snl_add_msg_attr_string(&nw, PF_CS_IFNAME, kill->ifname);
2072 	snl_add_msg_attr_string(&nw, PF_CS_LABEL, kill->label);
2073 	snl_add_msg_attr_bool(&nw, PF_CS_KILL_MATCH, kill->kill_match);
2074 	snl_add_msg_attr_bool(&nw, PF_CS_NAT, kill->nat);
2075 
2076 	if ((hdr = snl_finalize_msg(&nw)) == NULL)
2077 		return (ENXIO);
2078 
2079 	seq_id = hdr->nlmsg_seq;
2080 
2081 	if (! snl_send_message(&h->ss, hdr))
2082 		return (ENXIO);
2083 
2084 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
2085 		if (! snl_parse_nlmsg(&h->ss, hdr, &clear_states_parser, &attrs))
2086 			continue;
2087 	}
2088 
2089 	if (killed)
2090 		*killed = attrs.killed;
2091 
2092 	return (e.error);
2093 }
2094 
2095 int
pfctl_clear_states_h(struct pfctl_handle * h,const struct pfctl_kill * kill,unsigned int * killed)2096 pfctl_clear_states_h(struct pfctl_handle *h, const struct pfctl_kill *kill,
2097     unsigned int *killed)
2098 {
2099 	return(_pfctl_clear_states_h(h, kill, killed, PFNL_CMD_CLRSTATES));
2100 }
2101 
2102 int
pfctl_kill_states_h(struct pfctl_handle * h,const struct pfctl_kill * kill,unsigned int * killed)2103 pfctl_kill_states_h(struct pfctl_handle *h, const struct pfctl_kill *kill,
2104     unsigned int *killed)
2105 {
2106 	return(_pfctl_clear_states_h(h, kill, killed, PFNL_CMD_KILLSTATES));
2107 }
2108 
2109 static int
_pfctl_clear_states(int dev __unused,const struct pfctl_kill * kill,unsigned int * killed,uint64_t cmd)2110 _pfctl_clear_states(int dev __unused, const struct pfctl_kill *kill,
2111     unsigned int *killed, uint64_t cmd)
2112 {
2113 	struct pfctl_handle *h;
2114 	int ret;
2115 
2116 	h = pfctl_open(PF_DEVICE);
2117 	if (h == NULL)
2118 		return (ENODEV);
2119 
2120 	ret = _pfctl_clear_states_h(h, kill, killed, cmd);
2121 	pfctl_close(h);
2122 
2123 	return (ret);
2124 }
2125 
2126 int
pfctl_clear_states(int dev __unused,const struct pfctl_kill * kill,unsigned int * killed)2127 pfctl_clear_states(int dev __unused, const struct pfctl_kill *kill,
2128     unsigned int *killed)
2129 {
2130 	return (_pfctl_clear_states(dev, kill, killed, PFNL_CMD_CLRSTATES));
2131 }
2132 
2133 int
pfctl_kill_states(int dev __unused,const struct pfctl_kill * kill,unsigned int * killed)2134 pfctl_kill_states(int dev __unused, const struct pfctl_kill *kill, unsigned int *killed)
2135 {
2136 	return (_pfctl_clear_states(dev, kill, killed, PFNL_CMD_KILLSTATES));
2137 }
2138 
2139 int
pfctl_clear_rules(int dev,const char * anchorname)2140 pfctl_clear_rules(int dev, const char *anchorname)
2141 {
2142 	struct pfioc_trans trans;
2143 	struct pfioc_trans_e transe[2];
2144 	int ret;
2145 
2146 	bzero(&trans, sizeof(trans));
2147 	bzero(&transe, sizeof(transe));
2148 
2149 	transe[0].rs_num = PF_RULESET_SCRUB;
2150 	if (strlcpy(transe[0].anchor, anchorname, sizeof(transe[0].anchor))
2151 	    >= sizeof(transe[0].anchor))
2152 		return (E2BIG);
2153 
2154 	transe[1].rs_num = PF_RULESET_FILTER;
2155 	if (strlcpy(transe[1].anchor, anchorname, sizeof(transe[1].anchor))
2156 	    >= sizeof(transe[1].anchor))
2157 		return (E2BIG);
2158 
2159 	trans.size = 2;
2160 	trans.esize = sizeof(transe[0]);
2161 	trans.array = transe;
2162 
2163 	ret = ioctl(dev, DIOCXBEGIN, &trans);
2164 	if (ret != 0)
2165 		return (ret);
2166 	return ioctl(dev, DIOCXCOMMIT, &trans);
2167 }
2168 
2169 int
pfctl_clear_nat(int dev,const char * anchorname)2170 pfctl_clear_nat(int dev, const char *anchorname)
2171 {
2172 	struct pfioc_trans trans;
2173 	struct pfioc_trans_e transe[3];
2174 	int ret;
2175 
2176 	bzero(&trans, sizeof(trans));
2177 	bzero(&transe, sizeof(transe));
2178 
2179 	transe[0].rs_num = PF_RULESET_NAT;
2180 	if (strlcpy(transe[0].anchor, anchorname, sizeof(transe[0].anchor))
2181 	    >= sizeof(transe[0].anchor))
2182 		return (E2BIG);
2183 
2184 	transe[1].rs_num = PF_RULESET_BINAT;
2185 	if (strlcpy(transe[1].anchor, anchorname, sizeof(transe[1].anchor))
2186 	    >= sizeof(transe[0].anchor))
2187 		return (E2BIG);
2188 
2189 	transe[2].rs_num = PF_RULESET_RDR;
2190 	if (strlcpy(transe[2].anchor, anchorname, sizeof(transe[2].anchor))
2191 	    >= sizeof(transe[2].anchor))
2192 		return (E2BIG);
2193 
2194 	trans.size = 3;
2195 	trans.esize = sizeof(transe[0]);
2196 	trans.array = transe;
2197 
2198 	ret = ioctl(dev, DIOCXBEGIN, &trans);
2199 	if (ret != 0)
2200 		return (ret);
2201 	return ioctl(dev, DIOCXCOMMIT, &trans);
2202 }
2203 int
pfctl_clear_eth_rules(int dev,const char * anchorname)2204 pfctl_clear_eth_rules(int dev, const char *anchorname)
2205 {
2206 	struct pfioc_trans trans;
2207 	struct pfioc_trans_e transe;
2208 	int ret;
2209 
2210 	bzero(&trans, sizeof(trans));
2211 	bzero(&transe, sizeof(transe));
2212 
2213 	transe.rs_num = PF_RULESET_ETH;
2214 	if (strlcpy(transe.anchor, anchorname, sizeof(transe.anchor))
2215 	    >= sizeof(transe.anchor))
2216 		return (E2BIG);
2217 
2218 	trans.size = 1;
2219 	trans.esize = sizeof(transe);
2220 	trans.array = &transe;
2221 
2222 	ret = ioctl(dev, DIOCXBEGIN, &trans);
2223 	if (ret != 0)
2224 		return (ret);
2225 	return ioctl(dev, DIOCXCOMMIT, &trans);
2226 }
2227 
2228 static int
_pfctl_get_limit(int dev,const int index,uint * limit)2229 _pfctl_get_limit(int dev, const int index, uint *limit)
2230 {
2231 	struct pfioc_limit pl;
2232 
2233 	bzero(&pl, sizeof(pl));
2234 	pl.index = index;
2235 
2236 	if (ioctl(dev, DIOCGETLIMIT, &pl) == -1)
2237 		return (errno);
2238 
2239 	*limit = pl.limit;
2240 
2241 	return (0);
2242 }
2243 
2244 int
pfctl_set_syncookies(int dev,const struct pfctl_syncookies * s)2245 pfctl_set_syncookies(int dev, const struct pfctl_syncookies *s)
2246 {
2247 	struct pfioc_nv	 nv;
2248 	nvlist_t	*nvl;
2249 	int		 ret;
2250 	uint		 state_limit;
2251 	uint64_t	 lim, hi, lo;
2252 
2253 	ret = _pfctl_get_limit(dev, PF_LIMIT_STATES, &state_limit);
2254 	if (ret != 0)
2255 		return (ret);
2256 
2257 	lim = state_limit;
2258 	hi = lim * s->highwater / 100;
2259 	lo = lim * s->lowwater / 100;
2260 
2261 	if (lo == hi)
2262 		hi++;
2263 
2264 	nvl = nvlist_create(0);
2265 
2266 	nvlist_add_bool(nvl, "enabled", s->mode != PFCTL_SYNCOOKIES_NEVER);
2267 	nvlist_add_bool(nvl, "adaptive", s->mode == PFCTL_SYNCOOKIES_ADAPTIVE);
2268 	nvlist_add_number(nvl, "highwater", hi);
2269 	nvlist_add_number(nvl, "lowwater", lo);
2270 
2271 	nv.data = nvlist_pack(nvl, &nv.len);
2272 	nv.size = nv.len;
2273 	nvlist_destroy(nvl);
2274 	nvl = NULL;
2275 
2276 	ret = ioctl(dev, DIOCSETSYNCOOKIES, &nv);
2277 
2278 	free(nv.data);
2279 	return (ret);
2280 }
2281 
2282 int
pfctl_get_syncookies(int dev,struct pfctl_syncookies * s)2283 pfctl_get_syncookies(int dev, struct pfctl_syncookies *s)
2284 {
2285 	nvlist_t	*nvl;
2286 	int		 ret;
2287 	uint		 state_limit;
2288 	bool		 enabled, adaptive;
2289 
2290 	ret = _pfctl_get_limit(dev, PF_LIMIT_STATES, &state_limit);
2291 	if (ret != 0)
2292 		return (ret);
2293 
2294 	bzero(s, sizeof(*s));
2295 
2296 	nvl = nvlist_create(0);
2297 
2298 	if ((ret = pfctl_do_ioctl(dev, DIOCGETSYNCOOKIES, 256, &nvl)) != 0) {
2299 		ret = errno;
2300 		goto out;
2301 	}
2302 
2303 	enabled = nvlist_get_bool(nvl, "enabled");
2304 	adaptive = nvlist_get_bool(nvl, "adaptive");
2305 
2306 	if (enabled) {
2307 		if (adaptive)
2308 			s->mode = PFCTL_SYNCOOKIES_ADAPTIVE;
2309 		else
2310 			s->mode = PFCTL_SYNCOOKIES_ALWAYS;
2311 	} else {
2312 		s->mode = PFCTL_SYNCOOKIES_NEVER;
2313 	}
2314 
2315 	s->highwater = nvlist_get_number(nvl, "highwater") * 100 / state_limit;
2316 	s->lowwater = nvlist_get_number(nvl, "lowwater") * 100 / state_limit;
2317 	s->halfopen_states = nvlist_get_number(nvl, "halfopen_states");
2318 
2319 out:
2320 	nvlist_destroy(nvl);
2321 	return (ret);
2322 }
2323 
2324 int
pfctl_table_add_addrs(int dev,struct pfr_table * tbl,struct pfr_addr * addr,int size,int * nadd,int flags)2325 pfctl_table_add_addrs(int dev, struct pfr_table *tbl, struct pfr_addr
2326     *addr, int size, int *nadd, int flags)
2327 {
2328 	struct pfioc_table io;
2329 
2330 	if (tbl == NULL || size < 0 || (size && addr == NULL)) {
2331 		return (EINVAL);
2332 	}
2333 	bzero(&io, sizeof io);
2334 	io.pfrio_flags = flags;
2335 	io.pfrio_table = *tbl;
2336 	io.pfrio_buffer = addr;
2337 	io.pfrio_esize = sizeof(*addr);
2338 	io.pfrio_size = size;
2339 
2340 	if (ioctl(dev, DIOCRADDADDRS, &io))
2341 		return (errno);
2342 	if (nadd != NULL)
2343 		*nadd = io.pfrio_nadd;
2344 	return (0);
2345 }
2346 
2347 int
pfctl_table_del_addrs(int dev,struct pfr_table * tbl,struct pfr_addr * addr,int size,int * ndel,int flags)2348 pfctl_table_del_addrs(int dev, struct pfr_table *tbl, struct pfr_addr
2349     *addr, int size, int *ndel, int flags)
2350 {
2351 	struct pfioc_table io;
2352 
2353 	if (tbl == NULL || size < 0 || (size && addr == NULL)) {
2354 		return (EINVAL);
2355 	}
2356 	bzero(&io, sizeof io);
2357 	io.pfrio_flags = flags;
2358 	io.pfrio_table = *tbl;
2359 	io.pfrio_buffer = addr;
2360 	io.pfrio_esize = sizeof(*addr);
2361 	io.pfrio_size = size;
2362 
2363 	if (ioctl(dev, DIOCRDELADDRS, &io))
2364 		return (errno);
2365 	if (ndel != NULL)
2366 		*ndel = io.pfrio_ndel;
2367 	return (0);
2368 }
2369 
2370 int
pfctl_table_set_addrs(int dev,struct pfr_table * tbl,struct pfr_addr * addr,int size,int * size2,int * nadd,int * ndel,int * nchange,int flags)2371 pfctl_table_set_addrs(int dev, struct pfr_table *tbl, struct pfr_addr
2372     *addr, int size, int *size2, int *nadd, int *ndel, int *nchange, int flags)
2373 {
2374 	struct pfioc_table io;
2375 
2376 	if (tbl == NULL || size < 0 || (size && addr == NULL)) {
2377 		return (EINVAL);
2378 	}
2379 	bzero(&io, sizeof io);
2380 	io.pfrio_flags = flags;
2381 	io.pfrio_table = *tbl;
2382 	io.pfrio_buffer = addr;
2383 	io.pfrio_esize = sizeof(*addr);
2384 	io.pfrio_size = size;
2385 	io.pfrio_size2 = (size2 != NULL) ? *size2 : 0;
2386 	if (ioctl(dev, DIOCRSETADDRS, &io))
2387 		return (-1);
2388 	if (nadd != NULL)
2389 		*nadd = io.pfrio_nadd;
2390 	if (ndel != NULL)
2391 		*ndel = io.pfrio_ndel;
2392 	if (nchange != NULL)
2393 		*nchange = io.pfrio_nchange;
2394 	if (size2 != NULL)
2395 		*size2 = io.pfrio_size2;
2396 	return (0);
2397 }
2398 
pfctl_table_get_addrs(int dev,struct pfr_table * tbl,struct pfr_addr * addr,int * size,int flags)2399 int pfctl_table_get_addrs(int dev, struct pfr_table *tbl, struct pfr_addr *addr,
2400     int *size, int flags)
2401 {
2402 	struct pfioc_table io;
2403 
2404 	if (tbl == NULL || size == NULL || *size < 0 ||
2405 	    (*size && addr == NULL)) {
2406 		return (EINVAL);
2407 	}
2408 	bzero(&io, sizeof io);
2409 	io.pfrio_flags = flags;
2410 	io.pfrio_table = *tbl;
2411 	io.pfrio_buffer = addr;
2412 	io.pfrio_esize = sizeof(*addr);
2413 	io.pfrio_size = *size;
2414 	if (ioctl(dev, DIOCRGETADDRS, &io))
2415 		return (-1);
2416 	*size = io.pfrio_size;
2417 	return (0);
2418 }
2419 
2420 int
pfctl_set_statusif(struct pfctl_handle * h,const char * ifname)2421 pfctl_set_statusif(struct pfctl_handle *h, const char *ifname)
2422 {
2423 	struct snl_writer nw;
2424 	struct snl_errmsg_data e = {};
2425 	struct nlmsghdr *hdr;
2426 	uint32_t seq_id;
2427 	int family_id;
2428 
2429 	family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
2430 	if (family_id == 0)
2431 		return (ENOTSUP);
2432 
2433 	snl_init_writer(&h->ss, &nw);
2434 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_SET_STATUSIF);
2435 
2436 	snl_add_msg_attr_string(&nw, PF_SS_IFNAME, ifname);
2437 
2438 	if ((hdr = snl_finalize_msg(&nw)) == NULL)
2439 		return (ENXIO);
2440 
2441 	seq_id = hdr->nlmsg_seq;
2442 
2443 	if (! snl_send_message(&h->ss, hdr))
2444 		return (ENXIO);
2445 
2446 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
2447 	}
2448 
2449 	return (e.error);
2450 }
2451 
2452 #define	_IN(_field)	offsetof(struct genlmsghdr, _field)
2453 #define	_OUT(_field)	offsetof(struct pfctl_natlook, _field)
2454 static struct snl_attr_parser ap_natlook[] = {
2455 	{ .type = PF_NL_SRC_ADDR, .off = _OUT(saddr), .cb = snl_attr_get_in6_addr },
2456 	{ .type = PF_NL_DST_ADDR, .off = _OUT(daddr), .cb = snl_attr_get_in6_addr },
2457 	{ .type = PF_NL_SRC_PORT, .off = _OUT(sport), .cb = snl_attr_get_uint16 },
2458 	{ .type = PF_NL_DST_PORT, .off = _OUT(dport), .cb = snl_attr_get_uint16 },
2459 };
2460 static struct snl_field_parser fp_natlook[] = {};
2461 #undef _IN
2462 #undef _OUT
2463 SNL_DECLARE_PARSER(natlook_parser, struct genlmsghdr, fp_natlook, ap_natlook);
2464 
2465 int
pfctl_natlook(struct pfctl_handle * h,const struct pfctl_natlook_key * k,struct pfctl_natlook * r)2466 pfctl_natlook(struct pfctl_handle *h, const struct pfctl_natlook_key *k,
2467     struct pfctl_natlook *r)
2468 {
2469 	struct snl_writer nw;
2470 	struct snl_errmsg_data e = {};
2471 	struct nlmsghdr *hdr;
2472 	uint32_t seq_id;
2473 	int family_id;
2474 
2475 	family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
2476 	if (family_id == 0)
2477 		return (ENOTSUP);
2478 
2479 	snl_init_writer(&h->ss, &nw);
2480 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_NATLOOK);
2481 	hdr->nlmsg_flags |= NLM_F_DUMP;
2482 
2483 	snl_add_msg_attr_u8(&nw, PF_NL_AF, k->af);
2484 	snl_add_msg_attr_u8(&nw, PF_NL_DIRECTION, k->direction);
2485 	snl_add_msg_attr_u8(&nw, PF_NL_PROTO, k->proto);
2486 	snl_add_msg_attr_ip6(&nw, PF_NL_SRC_ADDR, &k->saddr.v6);
2487 	snl_add_msg_attr_ip6(&nw, PF_NL_DST_ADDR, &k->daddr.v6);
2488 	snl_add_msg_attr_u16(&nw, PF_NL_SRC_PORT, k->sport);
2489 	snl_add_msg_attr_u16(&nw, PF_NL_DST_PORT, k->dport);
2490 
2491 	if ((hdr = snl_finalize_msg(&nw)) == NULL)
2492 		return (ENXIO);
2493 
2494 	seq_id = hdr->nlmsg_seq;
2495 
2496 	if (! snl_send_message(&h->ss, hdr))
2497 		return (ENXIO);
2498 
2499 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
2500 		if (! snl_parse_nlmsg(&h->ss, hdr, &natlook_parser, r))
2501 			continue;
2502 	}
2503 
2504 	return (e.error);
2505 }
2506 
2507 int
pfctl_set_debug(struct pfctl_handle * h,uint32_t level)2508 pfctl_set_debug(struct pfctl_handle *h, uint32_t level)
2509 {
2510 	struct snl_writer nw;
2511 	struct snl_errmsg_data e = {};
2512 	struct nlmsghdr *hdr;
2513 	uint32_t seq_id;
2514 	int family_id;
2515 
2516 	family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
2517 	if (family_id == 0)
2518 		return (ENOTSUP);
2519 
2520 	snl_init_writer(&h->ss, &nw);
2521 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_SET_DEBUG);
2522 
2523 	snl_add_msg_attr_u32(&nw, PF_SD_LEVEL, level);
2524 
2525 	if ((hdr = snl_finalize_msg(&nw)) == NULL)
2526 		return (ENXIO);
2527 
2528 	seq_id = hdr->nlmsg_seq;
2529 
2530 	if (! snl_send_message(&h->ss, hdr))
2531 		return (ENXIO);
2532 
2533 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
2534 	}
2535 
2536 	return (e.error);
2537 }
2538 
2539 int
pfctl_set_timeout(struct pfctl_handle * h,uint32_t timeout,uint32_t seconds)2540 pfctl_set_timeout(struct pfctl_handle *h, uint32_t timeout, uint32_t seconds)
2541 {
2542 	struct snl_writer nw;
2543 	struct snl_errmsg_data e = {};
2544 	struct nlmsghdr *hdr;
2545 	uint32_t seq_id;
2546 	int family_id;
2547 
2548 	family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
2549 	if (family_id == 0)
2550 		return (ENOTSUP);
2551 
2552 	snl_init_writer(&h->ss, &nw);
2553 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_SET_TIMEOUT);
2554 
2555 	snl_add_msg_attr_u32(&nw, PF_TO_TIMEOUT, timeout);
2556 	snl_add_msg_attr_u32(&nw, PF_TO_SECONDS, seconds);
2557 
2558 	if ((hdr = snl_finalize_msg(&nw)) == NULL)
2559 		return (ENXIO);
2560 
2561 	seq_id = hdr->nlmsg_seq;
2562 
2563 	if (! snl_send_message(&h->ss, hdr))
2564 		return (ENXIO);
2565 
2566 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
2567 	}
2568 
2569 	return (e.error);
2570 }
2571 
2572 struct pfctl_nl_timeout {
2573 	uint32_t seconds;
2574 };
2575 #define	_OUT(_field)	offsetof(struct pfctl_nl_timeout, _field)
2576 static struct snl_attr_parser ap_get_timeout[] = {
2577 	{ .type = PF_TO_SECONDS, .off = _OUT(seconds), .cb = snl_attr_get_uint32 },
2578 };
2579 static struct snl_field_parser fp_get_timeout[] = {};
2580 #undef _OUT
2581 SNL_DECLARE_PARSER(get_timeout_parser, struct genlmsghdr, fp_get_timeout, ap_get_timeout);
2582 
2583 int
pfctl_get_timeout(struct pfctl_handle * h,uint32_t timeout,uint32_t * seconds)2584 pfctl_get_timeout(struct pfctl_handle *h, uint32_t timeout, uint32_t *seconds)
2585 {
2586 	struct snl_writer nw;
2587 	struct pfctl_nl_timeout to = {};
2588 	struct snl_errmsg_data e = {};
2589 	struct nlmsghdr *hdr;
2590 	uint32_t seq_id;
2591 	int family_id;
2592 
2593 	family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
2594 	if (family_id == 0)
2595 		return (ENOTSUP);
2596 
2597 	snl_init_writer(&h->ss, &nw);
2598 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_GET_TIMEOUT);
2599 	hdr->nlmsg_flags |= NLM_F_DUMP;
2600 
2601 	snl_add_msg_attr_u32(&nw, PF_TO_TIMEOUT, timeout);
2602 
2603 	if ((hdr = snl_finalize_msg(&nw)) == NULL)
2604 		return (ENXIO);
2605 
2606 	seq_id = hdr->nlmsg_seq;
2607 
2608 	if (! snl_send_message(&h->ss, hdr))
2609 		return (ENXIO);
2610 
2611 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
2612 		if (! snl_parse_nlmsg(&h->ss, hdr, &get_timeout_parser, &to))
2613 			continue;
2614 	}
2615 
2616 	if (seconds != NULL)
2617 		*seconds = to.seconds;
2618 
2619 	return (e.error);
2620 }
2621 
2622 int
pfctl_set_limit(struct pfctl_handle * h,const int index,const uint limit)2623 pfctl_set_limit(struct pfctl_handle *h, const int index, const uint limit)
2624 {
2625 	struct snl_writer nw;
2626 	struct snl_errmsg_data e = {};
2627 	struct nlmsghdr *hdr;
2628 	uint32_t seq_id;
2629 	int family_id;
2630 
2631 	family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
2632 	if (family_id == 0)
2633 		return (ENOTSUP);
2634 
2635 	snl_init_writer(&h->ss, &nw);
2636 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_SET_LIMIT);
2637 
2638 	snl_add_msg_attr_u32(&nw, PF_LI_INDEX, index);
2639 	snl_add_msg_attr_u32(&nw, PF_LI_LIMIT, limit);
2640 
2641 	if ((hdr = snl_finalize_msg(&nw)) == NULL)
2642 		return (ENXIO);
2643 
2644 	seq_id = hdr->nlmsg_seq;
2645 
2646 	if (! snl_send_message(&h->ss, hdr))
2647 		return (ENXIO);
2648 
2649 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
2650 	}
2651 
2652 	return (e.error);
2653 }
2654 
2655 struct pfctl_nl_limit {
2656 	unsigned int limit;
2657 };
2658 #define	_OUT(_field)	offsetof(struct pfctl_nl_limit, _field)
2659 static struct snl_attr_parser ap_get_limit[] = {
2660 	{ .type = PF_LI_LIMIT, .off = _OUT(limit), .cb = snl_attr_get_uint32 },
2661 };
2662 static struct snl_field_parser fp_get_limit[] = {};
2663 #undef _OUT
2664 SNL_DECLARE_PARSER(get_limit_parser, struct genlmsghdr, fp_get_limit, ap_get_limit);
2665 
2666 int
pfctl_get_limit(struct pfctl_handle * h,const int index,uint * limit)2667 pfctl_get_limit(struct pfctl_handle *h, const int index, uint *limit)
2668 {
2669 	struct snl_writer nw;
2670 	struct pfctl_nl_limit li = {};
2671 	struct snl_errmsg_data e = {};
2672 	struct nlmsghdr *hdr;
2673 	uint32_t seq_id;
2674 	int family_id;
2675 
2676 	family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
2677 	if (family_id == 0)
2678 		return (ENOTSUP);
2679 
2680 	snl_init_writer(&h->ss, &nw);
2681 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_GET_LIMIT);
2682 	hdr->nlmsg_flags |= NLM_F_DUMP;
2683 
2684 	snl_add_msg_attr_u32(&nw, PF_LI_INDEX, index);
2685 
2686 	if ((hdr = snl_finalize_msg(&nw)) == NULL)
2687 		return (ENXIO);
2688 
2689 	seq_id = hdr->nlmsg_seq;
2690 
2691 	if (! snl_send_message(&h->ss, hdr))
2692 		return (ENXIO);
2693 
2694 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
2695 		if (! snl_parse_nlmsg(&h->ss, hdr, &get_limit_parser, &li))
2696 			continue;
2697 	}
2698 
2699 	if (limit != NULL)
2700 		*limit = li.limit;
2701 
2702 	return (e.error);
2703 }
2704 
2705 struct pfctl_nl_begin_addrs {
2706 	uint32_t ticket;
2707 };
2708 #define	_OUT(_field)	offsetof(struct pfctl_nl_begin_addrs, _field)
2709 static struct snl_attr_parser ap_begin_addrs[] = {
2710 	{ .type = PF_BA_TICKET, .off = _OUT(ticket), .cb = snl_attr_get_uint32 },
2711 };
2712 static struct snl_field_parser fp_begin_addrs[] = {};
2713 #undef _OUT
2714 SNL_DECLARE_PARSER(begin_addrs_parser, struct genlmsghdr, fp_begin_addrs, ap_begin_addrs);
2715 
2716 int
pfctl_begin_addrs(struct pfctl_handle * h,uint32_t * ticket)2717 pfctl_begin_addrs(struct pfctl_handle *h, uint32_t *ticket)
2718 {
2719 	struct snl_writer nw;
2720 	struct pfctl_nl_begin_addrs attrs = {};
2721 	struct snl_errmsg_data e = {};
2722 	struct nlmsghdr *hdr;
2723 	uint32_t seq_id;
2724 	int family_id;
2725 
2726 	family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
2727 	if (family_id == 0)
2728 		return (ENOTSUP);
2729 
2730 	snl_init_writer(&h->ss, &nw);
2731 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_BEGIN_ADDRS);
2732 	hdr->nlmsg_flags |= NLM_F_DUMP;
2733 
2734 	if ((hdr = snl_finalize_msg(&nw)) == NULL)
2735 		return (ENXIO);
2736 
2737 	seq_id = hdr->nlmsg_seq;
2738 
2739 	if (! snl_send_message(&h->ss, hdr))
2740 		return (ENXIO);
2741 
2742 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
2743 		if (! snl_parse_nlmsg(&h->ss, hdr, &begin_addrs_parser, &attrs))
2744 			continue;
2745 	}
2746 
2747 	if (ticket != NULL)
2748 		*ticket = attrs.ticket;
2749 
2750 	return (e.error);
2751 }
2752 
2753 int
pfctl_add_addr(struct pfctl_handle * h,const struct pfioc_pooladdr * pa)2754 pfctl_add_addr(struct pfctl_handle *h, const struct pfioc_pooladdr *pa)
2755 {
2756 	struct snl_writer nw;
2757 	struct snl_errmsg_data e = {};
2758 	struct nlmsghdr *hdr;
2759 	uint32_t seq_id;
2760 	int family_id;
2761 
2762 	family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
2763 	if (family_id == 0)
2764 		return (ENOTSUP);
2765 
2766 	snl_init_writer(&h->ss, &nw);
2767 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_ADD_ADDR);
2768 
2769 	snl_add_msg_attr_u32(&nw, PF_AA_ACTION, pa->action);
2770 	snl_add_msg_attr_u32(&nw, PF_AA_TICKET, pa->ticket);
2771 	snl_add_msg_attr_u32(&nw, PF_AA_NR, pa->nr);
2772 	snl_add_msg_attr_u32(&nw, PF_AA_R_NUM, pa->r_num);
2773 	snl_add_msg_attr_u8(&nw, PF_AA_R_ACTION, pa->r_action);
2774 	snl_add_msg_attr_u8(&nw, PF_AA_R_LAST, pa->r_last);
2775 	snl_add_msg_attr_u8(&nw, PF_AA_AF, pa->af);
2776 	snl_add_msg_attr_string(&nw, PF_AA_ANCHOR, pa->anchor);
2777 	snl_add_msg_attr_pool_addr(&nw, PF_AA_ADDR, &pa->addr);
2778 
2779 	if ((hdr = snl_finalize_msg(&nw)) == NULL)
2780 		return (ENXIO);
2781 
2782 	seq_id = hdr->nlmsg_seq;
2783 
2784 	if (! snl_send_message(&h->ss, hdr))
2785 		return (ENXIO);
2786 
2787 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
2788 	}
2789 
2790 	return (e.error);
2791 }
2792 
2793 static const struct snl_attr_parser ap_get_addrs[] = {
2794 	{ .type = PF_AA_NR, .off = 0, .cb = snl_attr_get_uint32 },
2795 };
2796 static struct snl_field_parser fp_get_addrs[] = {};
2797 SNL_DECLARE_PARSER(get_addrs_parser, struct genlmsghdr, fp_get_addrs, ap_get_addrs);
2798 
2799 int
pfctl_get_addrs(struct pfctl_handle * h,uint32_t ticket,uint32_t r_num,uint8_t r_action,const char * anchor,uint32_t * nr)2800 pfctl_get_addrs(struct pfctl_handle *h, uint32_t ticket, uint32_t r_num,
2801     uint8_t r_action, const char *anchor, uint32_t *nr)
2802 {
2803 	struct snl_writer nw;
2804 	struct snl_errmsg_data e = {};
2805 	struct nlmsghdr *hdr;
2806 	uint32_t seq_id;
2807 	int family_id;
2808 
2809 	family_id = snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
2810 	if (family_id == 0)
2811 		return (ENOTSUP);
2812 
2813 	snl_init_writer(&h->ss, &nw);
2814 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_GET_ADDRS);
2815 
2816 	snl_add_msg_attr_u32(&nw, PF_AA_TICKET, ticket);
2817 	snl_add_msg_attr_u32(&nw, PF_AA_R_NUM, r_num);
2818 	snl_add_msg_attr_u8(&nw, PF_AA_R_ACTION, r_action);
2819 	snl_add_msg_attr_string(&nw, PF_AA_ANCHOR, anchor);
2820 
2821 	if ((hdr = snl_finalize_msg(&nw)) == NULL)
2822 		return (ENXIO);
2823 
2824 	seq_id = hdr->nlmsg_seq;
2825 
2826 	if (! snl_send_message(&h->ss, hdr))
2827 		return (ENXIO);
2828 
2829 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
2830 		if (! snl_parse_nlmsg(&h->ss, hdr, &get_addrs_parser, nr))
2831 			continue;
2832 	}
2833 
2834 	return (e.error);
2835 }
2836 
2837 #define _OUT(_field)	offsetof(struct pf_pooladdr, _field)
2838 static const struct snl_attr_parser ap_pool_addr[] = {
2839 	{ .type = PF_PA_ADDR, .off = _OUT(addr), .arg = &addr_wrap_parser, .cb = snl_attr_get_nested },
2840 	{ .type = PF_PA_IFNAME, .off = _OUT(ifname), .arg_u32 = IFNAMSIZ, .cb = snl_attr_copy_string },
2841 };
2842 SNL_DECLARE_ATTR_PARSER(pool_addr_parser, ap_pool_addr);
2843 #undef _OUT
2844 
2845 #define _OUT(_field)	offsetof(struct pfioc_pooladdr, _field)
2846 static const struct snl_attr_parser ap_get_addr[] = {
2847 	{ .type = PF_AA_ACTION, .off = _OUT(action), .cb = snl_attr_get_uint32 },
2848 	{ .type = PF_AA_TICKET, .off = _OUT(ticket), .cb = snl_attr_get_uint32 },
2849 	{ .type = PF_AA_NR, .off = _OUT(nr), .cb = snl_attr_get_uint32 },
2850 	{ .type = PF_AA_R_NUM, .off = _OUT(r_num), .cb = snl_attr_get_uint32 },
2851 	{ .type = PF_AA_R_ACTION, .off = _OUT(r_action), .cb = snl_attr_get_uint8 },
2852 	{ .type = PF_AA_R_LAST, .off = _OUT(r_last), .cb = snl_attr_get_uint8 },
2853 	{ .type = PF_AA_AF, .off = _OUT(af), .cb = snl_attr_get_uint8 },
2854 	{ .type = PF_AA_ANCHOR, .off = _OUT(anchor), .arg_u32 = MAXPATHLEN, .cb = snl_attr_copy_string },
2855 	{ .type = PF_AA_ADDR, .off = _OUT(addr), .arg = &pool_addr_parser, .cb = snl_attr_get_nested },
2856 };
2857 static struct snl_field_parser fp_get_addr[] = {};
2858 SNL_DECLARE_PARSER(get_addr_parser, struct genlmsghdr, fp_get_addr, ap_get_addr);
2859 #undef _OUT
2860 
2861 int
pfctl_get_addr(struct pfctl_handle * h,uint32_t ticket,uint32_t r_num,uint8_t r_action,const char * anchor,uint32_t nr,struct pfioc_pooladdr * pa)2862 pfctl_get_addr(struct pfctl_handle *h, uint32_t ticket, uint32_t r_num,
2863     uint8_t r_action, const char *anchor, uint32_t nr, struct pfioc_pooladdr *pa)
2864 {
2865 	struct snl_writer nw;
2866 	struct snl_errmsg_data e = {};
2867 	struct nlmsghdr *hdr;
2868 	uint32_t seq_id;
2869 	int family_id;
2870 
2871 	family_id =snl_get_genl_family(&h->ss, PFNL_FAMILY_NAME);
2872 	if (family_id == 0)
2873 		return (ENOTSUP);
2874 
2875 	snl_init_writer(&h->ss, &nw);
2876 	hdr = snl_create_genl_msg_request(&nw, family_id, PFNL_CMD_GET_ADDR);
2877 
2878 	snl_add_msg_attr_u32(&nw, PF_AA_TICKET, ticket);
2879 	snl_add_msg_attr_u32(&nw, PF_AA_R_NUM, r_num);
2880 	snl_add_msg_attr_u8(&nw, PF_AA_R_ACTION, r_action);
2881 	snl_add_msg_attr_string(&nw, PF_AA_ANCHOR, anchor);
2882 	snl_add_msg_attr_u32(&nw, PF_AA_NR, nr);
2883 
2884 	if ((hdr = snl_finalize_msg(&nw)) == NULL)
2885 		return (ENXIO);
2886 
2887 	seq_id = hdr->nlmsg_seq;
2888 
2889 	if (! snl_send_message(&h->ss, hdr))
2890 		return (ENXIO);
2891 
2892 	while ((hdr = snl_read_reply_multi(&h->ss, seq_id, &e)) != NULL) {
2893 		if (! snl_parse_nlmsg(&h->ss, hdr, &get_addr_parser, pa))
2894 			continue;
2895 	}
2896 
2897 	return (0);
2898 }
2899