1 /*-
2 * SPDX-License-Identifier: BSD-2-Clause
3 *
4 * Copyright (c) 2021 Rubicon Communications, LLC (Netgate)
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 *
11 * - Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 * - Redistributions in binary form must reproduce the above
14 * copyright notice, this list of conditions and the following
15 * disclaimer in the documentation and/or other materials provided
16 * with the distribution.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
21 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
22 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
23 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
24 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
25 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
26 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
28 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
30 *
31 * $FreeBSD$
32 */
33
34 #include <sys/cdefs.h>
35
36 #include <sys/ioctl.h>
37 #include <sys/nv.h>
38 #include <sys/queue.h>
39 #include <sys/types.h>
40
41 #include <net/if.h>
42 #include <net/pfvar.h>
43 #include <netinet/in.h>
44
45 #include <assert.h>
46 #include <err.h>
47 #include <errno.h>
48 #include <stdlib.h>
49 #include <string.h>
50
51 #include "libpfctl.h"
52
53 const char* PFCTL_SYNCOOKIES_MODE_NAMES[] = {
54 "never",
55 "always",
56 "adaptive"
57 };
58
59 static int _pfctl_clear_states(int , const struct pfctl_kill *,
60 unsigned int *, uint64_t);
61
62 static void
pf_nvuint_8_array(const nvlist_t * nvl,const char * name,size_t maxelems,uint8_t * numbers,size_t * nelems)63 pf_nvuint_8_array(const nvlist_t *nvl, const char *name, size_t maxelems,
64 uint8_t *numbers, size_t *nelems)
65 {
66 const uint64_t *tmp;
67 size_t elems;
68
69 tmp = nvlist_get_number_array(nvl, name, &elems);
70 assert(elems <= maxelems);
71
72 for (size_t i = 0; i < elems; i++)
73 numbers[i] = tmp[i];
74
75 if (nelems)
76 *nelems = elems;
77 }
78
79 static void
pf_nvuint_16_array(const nvlist_t * nvl,const char * name,size_t maxelems,uint16_t * numbers,size_t * nelems)80 pf_nvuint_16_array(const nvlist_t *nvl, const char *name, size_t maxelems,
81 uint16_t *numbers, size_t *nelems)
82 {
83 const uint64_t *tmp;
84 size_t elems;
85
86 tmp = nvlist_get_number_array(nvl, name, &elems);
87 assert(elems <= maxelems);
88
89 for (size_t i = 0; i < elems; i++)
90 numbers[i] = tmp[i];
91
92 if (nelems)
93 *nelems = elems;
94 }
95
96 static void
pf_nvuint_32_array(const nvlist_t * nvl,const char * name,size_t maxelems,uint32_t * numbers,size_t * nelems)97 pf_nvuint_32_array(const nvlist_t *nvl, const char *name, size_t maxelems,
98 uint32_t *numbers, size_t *nelems)
99 {
100 const uint64_t *tmp;
101 size_t elems;
102
103 tmp = nvlist_get_number_array(nvl, name, &elems);
104 assert(elems <= maxelems);
105
106 for (size_t i = 0; i < elems; i++)
107 numbers[i] = tmp[i];
108
109 if (nelems)
110 *nelems = elems;
111 }
112
113 static void
pf_nvuint_64_array(const nvlist_t * nvl,const char * name,size_t maxelems,uint64_t * numbers,size_t * nelems)114 pf_nvuint_64_array(const nvlist_t *nvl, const char *name, size_t maxelems,
115 uint64_t *numbers, size_t *nelems)
116 {
117 const uint64_t *tmp;
118 size_t elems;
119
120 tmp = nvlist_get_number_array(nvl, name, &elems);
121 assert(elems <= maxelems);
122
123 for (size_t i = 0; i < elems; i++)
124 numbers[i] = tmp[i];
125
126 if (nelems)
127 *nelems = elems;
128 }
129
130 static void
_pfctl_get_status_counters(const nvlist_t * nvl,struct pfctl_status_counters * counters)131 _pfctl_get_status_counters(const nvlist_t *nvl,
132 struct pfctl_status_counters *counters)
133 {
134 const uint64_t *ids, *counts;
135 const char *const *names;
136 size_t id_len, counter_len, names_len;
137
138 ids = nvlist_get_number_array(nvl, "ids", &id_len);
139 counts = nvlist_get_number_array(nvl, "counters", &counter_len);
140 names = nvlist_get_string_array(nvl, "names", &names_len);
141 assert(id_len == counter_len);
142 assert(counter_len == names_len);
143
144 TAILQ_INIT(counters);
145
146 for (size_t i = 0; i < id_len; i++) {
147 struct pfctl_status_counter *c;
148
149 c = malloc(sizeof(*c));
150
151 c->id = ids[i];
152 c->counter = counts[i];
153 c->name = strdup(names[i]);
154
155 TAILQ_INSERT_TAIL(counters, c, entry);
156 }
157 }
158
159 struct pfctl_status *
pfctl_get_status(int dev)160 pfctl_get_status(int dev)
161 {
162 struct pfioc_nv nv;
163 struct pfctl_status *status;
164 nvlist_t *nvl;
165 size_t len;
166 const void *chksum;
167
168 status = calloc(1, sizeof(*status));
169 if (status == NULL)
170 return (NULL);
171
172 nv.data = malloc(4096);
173 nv.len = nv.size = 4096;
174
175 if (ioctl(dev, DIOCGETSTATUSNV, &nv)) {
176 free(nv.data);
177 free(status);
178 return (NULL);
179 }
180
181 nvl = nvlist_unpack(nv.data, nv.len, 0);
182 free(nv.data);
183 if (nvl == NULL) {
184 free(status);
185 return (NULL);
186 }
187
188 status->running = nvlist_get_bool(nvl, "running");
189 status->since = nvlist_get_number(nvl, "since");
190 status->debug = nvlist_get_number(nvl, "debug");
191 status->hostid = ntohl(nvlist_get_number(nvl, "hostid"));
192 status->states = nvlist_get_number(nvl, "states");
193 status->src_nodes = nvlist_get_number(nvl, "src_nodes");
194
195 strlcpy(status->ifname, nvlist_get_string(nvl, "ifname"),
196 IFNAMSIZ);
197 chksum = nvlist_get_binary(nvl, "chksum", &len);
198 assert(len == PF_MD5_DIGEST_LENGTH);
199 memcpy(status->pf_chksum, chksum, len);
200
201 _pfctl_get_status_counters(nvlist_get_nvlist(nvl, "counters"),
202 &status->counters);
203 _pfctl_get_status_counters(nvlist_get_nvlist(nvl, "lcounters"),
204 &status->lcounters);
205 _pfctl_get_status_counters(nvlist_get_nvlist(nvl, "fcounters"),
206 &status->fcounters);
207 _pfctl_get_status_counters(nvlist_get_nvlist(nvl, "scounters"),
208 &status->scounters);
209
210 pf_nvuint_64_array(nvl, "pcounters", 2 * 2 * 3,
211 (uint64_t *)status->pcounters, NULL);
212 pf_nvuint_64_array(nvl, "bcounters", 2 * 2,
213 (uint64_t *)status->bcounters, NULL);
214
215 nvlist_destroy(nvl);
216
217 return (status);
218 }
219
220 void
pfctl_free_status(struct pfctl_status * status)221 pfctl_free_status(struct pfctl_status *status)
222 {
223 struct pfctl_status_counter *c, *tmp;
224
225 TAILQ_FOREACH_SAFE(c, &status->counters, entry, tmp) {
226 free(c->name);
227 free(c);
228 }
229 TAILQ_FOREACH_SAFE(c, &status->lcounters, entry, tmp) {
230 free(c->name);
231 free(c);
232 }
233 TAILQ_FOREACH_SAFE(c, &status->fcounters, entry, tmp) {
234 free(c->name);
235 free(c);
236 }
237 TAILQ_FOREACH_SAFE(c, &status->scounters, entry, tmp) {
238 free(c->name);
239 free(c);
240 }
241
242 free(status);
243 }
244
245 static void
pfctl_nv_add_addr(nvlist_t * nvparent,const char * name,const struct pf_addr * addr)246 pfctl_nv_add_addr(nvlist_t *nvparent, const char *name,
247 const struct pf_addr *addr)
248 {
249 nvlist_t *nvl = nvlist_create(0);
250
251 nvlist_add_binary(nvl, "addr", addr, sizeof(*addr));
252
253 nvlist_add_nvlist(nvparent, name, nvl);
254 nvlist_destroy(nvl);
255 }
256
257 static void
pf_nvaddr_to_addr(const nvlist_t * nvl,struct pf_addr * addr)258 pf_nvaddr_to_addr(const nvlist_t *nvl, struct pf_addr *addr)
259 {
260 size_t len;
261 const void *data;
262
263 data = nvlist_get_binary(nvl, "addr", &len);
264 assert(len == sizeof(struct pf_addr));
265 memcpy(addr, data, len);
266 }
267
268 static void
pfctl_nv_add_addr_wrap(nvlist_t * nvparent,const char * name,const struct pf_addr_wrap * addr)269 pfctl_nv_add_addr_wrap(nvlist_t *nvparent, const char *name,
270 const struct pf_addr_wrap *addr)
271 {
272 nvlist_t *nvl = nvlist_create(0);
273
274 nvlist_add_number(nvl, "type", addr->type);
275 nvlist_add_number(nvl, "iflags", addr->iflags);
276 if (addr->type == PF_ADDR_DYNIFTL)
277 nvlist_add_string(nvl, "ifname", addr->v.ifname);
278 if (addr->type == PF_ADDR_TABLE)
279 nvlist_add_string(nvl, "tblname", addr->v.tblname);
280 pfctl_nv_add_addr(nvl, "addr", &addr->v.a.addr);
281 pfctl_nv_add_addr(nvl, "mask", &addr->v.a.mask);
282
283 nvlist_add_nvlist(nvparent, name, nvl);
284 nvlist_destroy(nvl);
285 }
286
287 static void
pf_nvaddr_wrap_to_addr_wrap(const nvlist_t * nvl,struct pf_addr_wrap * addr)288 pf_nvaddr_wrap_to_addr_wrap(const nvlist_t *nvl, struct pf_addr_wrap *addr)
289 {
290 bzero(addr, sizeof(*addr));
291
292 addr->type = nvlist_get_number(nvl, "type");
293 addr->iflags = nvlist_get_number(nvl, "iflags");
294 if (addr->type == PF_ADDR_DYNIFTL) {
295 strlcpy(addr->v.ifname, nvlist_get_string(nvl, "ifname"),
296 IFNAMSIZ);
297 addr->p.dyncnt = nvlist_get_number(nvl, "dyncnt");
298 }
299 if (addr->type == PF_ADDR_TABLE) {
300 strlcpy(addr->v.tblname, nvlist_get_string(nvl, "tblname"),
301 PF_TABLE_NAME_SIZE);
302 addr->p.tblcnt = nvlist_get_number(nvl, "tblcnt");
303 }
304
305 pf_nvaddr_to_addr(nvlist_get_nvlist(nvl, "addr"), &addr->v.a.addr);
306 pf_nvaddr_to_addr(nvlist_get_nvlist(nvl, "mask"), &addr->v.a.mask);
307 }
308
309 static void
pfctl_nv_add_rule_addr(nvlist_t * nvparent,const char * name,const struct pf_rule_addr * addr)310 pfctl_nv_add_rule_addr(nvlist_t *nvparent, const char *name,
311 const struct pf_rule_addr *addr)
312 {
313 uint64_t ports[2];
314 nvlist_t *nvl = nvlist_create(0);
315
316 pfctl_nv_add_addr_wrap(nvl, "addr", &addr->addr);
317 ports[0] = addr->port[0];
318 ports[1] = addr->port[1];
319 nvlist_add_number_array(nvl, "port", ports, 2);
320 nvlist_add_number(nvl, "neg", addr->neg);
321 nvlist_add_number(nvl, "port_op", addr->port_op);
322
323 nvlist_add_nvlist(nvparent, name, nvl);
324 nvlist_destroy(nvl);
325 }
326
327 static void
pf_nvrule_addr_to_rule_addr(const nvlist_t * nvl,struct pf_rule_addr * addr)328 pf_nvrule_addr_to_rule_addr(const nvlist_t *nvl, struct pf_rule_addr *addr)
329 {
330 pf_nvaddr_wrap_to_addr_wrap(nvlist_get_nvlist(nvl, "addr"), &addr->addr);
331
332 pf_nvuint_16_array(nvl, "port", 2, addr->port, NULL);
333 addr->neg = nvlist_get_number(nvl, "neg");
334 addr->port_op = nvlist_get_number(nvl, "port_op");
335 }
336
337 static void
pfctl_nv_add_mape(nvlist_t * nvparent,const char * name,const struct pf_mape_portset * mape)338 pfctl_nv_add_mape(nvlist_t *nvparent, const char *name,
339 const struct pf_mape_portset *mape)
340 {
341 nvlist_t *nvl = nvlist_create(0);
342
343 nvlist_add_number(nvl, "offset", mape->offset);
344 nvlist_add_number(nvl, "psidlen", mape->psidlen);
345 nvlist_add_number(nvl, "psid", mape->psid);
346 nvlist_add_nvlist(nvparent, name, nvl);
347 nvlist_destroy(nvl);
348 }
349
350 static void
pfctl_nv_add_pool(nvlist_t * nvparent,const char * name,const struct pfctl_pool * pool)351 pfctl_nv_add_pool(nvlist_t *nvparent, const char *name,
352 const struct pfctl_pool *pool)
353 {
354 uint64_t ports[2];
355 nvlist_t *nvl = nvlist_create(0);
356
357 nvlist_add_binary(nvl, "key", &pool->key, sizeof(pool->key));
358 pfctl_nv_add_addr(nvl, "counter", &pool->counter);
359 nvlist_add_number(nvl, "tblidx", pool->tblidx);
360
361 ports[0] = pool->proxy_port[0];
362 ports[1] = pool->proxy_port[1];
363 nvlist_add_number_array(nvl, "proxy_port", ports, 2);
364 nvlist_add_number(nvl, "opts", pool->opts);
365 pfctl_nv_add_mape(nvl, "mape", &pool->mape);
366
367 nvlist_add_nvlist(nvparent, name, nvl);
368 nvlist_destroy(nvl);
369 }
370
371 static void
pf_nvmape_to_mape(const nvlist_t * nvl,struct pf_mape_portset * mape)372 pf_nvmape_to_mape(const nvlist_t *nvl, struct pf_mape_portset *mape)
373 {
374 mape->offset = nvlist_get_number(nvl, "offset");
375 mape->psidlen = nvlist_get_number(nvl, "psidlen");
376 mape->psid = nvlist_get_number(nvl, "psid");
377 }
378
379 static void
pf_nvpool_to_pool(const nvlist_t * nvl,struct pfctl_pool * pool)380 pf_nvpool_to_pool(const nvlist_t *nvl, struct pfctl_pool *pool)
381 {
382 size_t len;
383 const void *data;
384
385 data = nvlist_get_binary(nvl, "key", &len);
386 assert(len == sizeof(pool->key));
387 memcpy(&pool->key, data, len);
388
389 pf_nvaddr_to_addr(nvlist_get_nvlist(nvl, "counter"), &pool->counter);
390
391 pool->tblidx = nvlist_get_number(nvl, "tblidx");
392 pf_nvuint_16_array(nvl, "proxy_port", 2, pool->proxy_port, NULL);
393 pool->opts = nvlist_get_number(nvl, "opts");
394
395 if (nvlist_exists_nvlist(nvl, "mape"))
396 pf_nvmape_to_mape(nvlist_get_nvlist(nvl, "mape"), &pool->mape);
397 }
398
399 static void
pfctl_nv_add_uid(nvlist_t * nvparent,const char * name,const struct pf_rule_uid * uid)400 pfctl_nv_add_uid(nvlist_t *nvparent, const char *name,
401 const struct pf_rule_uid *uid)
402 {
403 uint64_t uids[2];
404 nvlist_t *nvl = nvlist_create(0);
405
406 uids[0] = uid->uid[0];
407 uids[1] = uid->uid[1];
408 nvlist_add_number_array(nvl, "uid", uids, 2);
409 nvlist_add_number(nvl, "op", uid->op);
410
411 nvlist_add_nvlist(nvparent, name, nvl);
412 nvlist_destroy(nvl);
413 }
414
415 static void
pf_nvrule_uid_to_rule_uid(const nvlist_t * nvl,struct pf_rule_uid * uid)416 pf_nvrule_uid_to_rule_uid(const nvlist_t *nvl, struct pf_rule_uid *uid)
417 {
418 pf_nvuint_32_array(nvl, "uid", 2, uid->uid, NULL);
419 uid->op = nvlist_get_number(nvl, "op");
420 }
421
422 static void
pfctl_nv_add_divert(nvlist_t * nvparent,const char * name,const struct pfctl_rule * r)423 pfctl_nv_add_divert(nvlist_t *nvparent, const char *name,
424 const struct pfctl_rule *r)
425 {
426 nvlist_t *nvl = nvlist_create(0);
427
428 pfctl_nv_add_addr(nvl, "addr", &r->divert.addr);
429 nvlist_add_number(nvl, "port", r->divert.port);
430
431 nvlist_add_nvlist(nvparent, name, nvl);
432 nvlist_destroy(nvl);
433 }
434
435 static void
pf_nvdivert_to_divert(const nvlist_t * nvl,struct pfctl_rule * rule)436 pf_nvdivert_to_divert(const nvlist_t *nvl, struct pfctl_rule *rule)
437 {
438 pf_nvaddr_to_addr(nvlist_get_nvlist(nvl, "addr"), &rule->divert.addr);
439 rule->divert.port = nvlist_get_number(nvl, "port");
440 }
441
442 static void
pf_nvrule_to_rule(const nvlist_t * nvl,struct pfctl_rule * rule)443 pf_nvrule_to_rule(const nvlist_t *nvl, struct pfctl_rule *rule)
444 {
445 const uint64_t *skip;
446 const char *const *labels;
447 size_t skipcount, labelcount;
448
449 rule->nr = nvlist_get_number(nvl, "nr");
450
451 pf_nvrule_addr_to_rule_addr(nvlist_get_nvlist(nvl, "src"), &rule->src);
452 pf_nvrule_addr_to_rule_addr(nvlist_get_nvlist(nvl, "dst"), &rule->dst);
453
454 skip = nvlist_get_number_array(nvl, "skip", &skipcount);
455 assert(skip);
456 assert(skipcount == PF_SKIP_COUNT);
457 for (int i = 0; i < PF_SKIP_COUNT; i++)
458 rule->skip[i].nr = skip[i];
459
460 labels = nvlist_get_string_array(nvl, "labels", &labelcount);
461 assert(labelcount <= PF_RULE_MAX_LABEL_COUNT);
462 for (size_t i = 0; i < labelcount; i++)
463 strlcpy(rule->label[i], labels[i], PF_RULE_LABEL_SIZE);
464 rule->ridentifier = nvlist_get_number(nvl, "ridentifier");
465 strlcpy(rule->ifname, nvlist_get_string(nvl, "ifname"), IFNAMSIZ);
466 strlcpy(rule->qname, nvlist_get_string(nvl, "qname"), PF_QNAME_SIZE);
467 strlcpy(rule->pqname, nvlist_get_string(nvl, "pqname"), PF_QNAME_SIZE);
468 strlcpy(rule->tagname, nvlist_get_string(nvl, "tagname"),
469 PF_TAG_NAME_SIZE);
470 strlcpy(rule->match_tagname, nvlist_get_string(nvl, "match_tagname"),
471 PF_TAG_NAME_SIZE);
472
473 strlcpy(rule->overload_tblname, nvlist_get_string(nvl, "overload_tblname"),
474 PF_TABLE_NAME_SIZE);
475
476 pf_nvpool_to_pool(nvlist_get_nvlist(nvl, "rpool"), &rule->rpool);
477
478 rule->evaluations = nvlist_get_number(nvl, "evaluations");
479 pf_nvuint_64_array(nvl, "packets", 2, rule->packets, NULL);
480 pf_nvuint_64_array(nvl, "bytes", 2, rule->bytes, NULL);
481
482 rule->os_fingerprint = nvlist_get_number(nvl, "os_fingerprint");
483
484 rule->rtableid = nvlist_get_number(nvl, "rtableid");
485 pf_nvuint_32_array(nvl, "timeout", PFTM_MAX, rule->timeout, NULL);
486 rule->max_states = nvlist_get_number(nvl, "max_states");
487 rule->max_src_nodes = nvlist_get_number(nvl, "max_src_nodes");
488 rule->max_src_states = nvlist_get_number(nvl, "max_src_states");
489 rule->max_src_conn = nvlist_get_number(nvl, "max_src_conn");
490 rule->max_src_conn_rate.limit =
491 nvlist_get_number(nvl, "max_src_conn_rate.limit");
492 rule->max_src_conn_rate.seconds =
493 nvlist_get_number(nvl, "max_src_conn_rate.seconds");
494 rule->qid = nvlist_get_number(nvl, "qid");
495 rule->pqid = nvlist_get_number(nvl, "pqid");
496 rule->prob = nvlist_get_number(nvl, "prob");
497 rule->cuid = nvlist_get_number(nvl, "cuid");
498 rule->cpid = nvlist_get_number(nvl, "cpid");
499
500 rule->return_icmp = nvlist_get_number(nvl, "return_icmp");
501 rule->return_icmp6 = nvlist_get_number(nvl, "return_icmp6");
502 rule->max_mss = nvlist_get_number(nvl, "max_mss");
503 rule->scrub_flags = nvlist_get_number(nvl, "scrub_flags");
504
505 pf_nvrule_uid_to_rule_uid(nvlist_get_nvlist(nvl, "uid"), &rule->uid);
506 pf_nvrule_uid_to_rule_uid(nvlist_get_nvlist(nvl, "gid"),
507 (struct pf_rule_uid *)&rule->gid);
508
509 rule->rule_flag = nvlist_get_number(nvl, "rule_flag");
510 rule->action = nvlist_get_number(nvl, "action");
511 rule->direction = nvlist_get_number(nvl, "direction");
512 rule->log = nvlist_get_number(nvl, "log");
513 rule->logif = nvlist_get_number(nvl, "logif");
514 rule->quick = nvlist_get_number(nvl, "quick");
515 rule->ifnot = nvlist_get_number(nvl, "ifnot");
516 rule->match_tag_not = nvlist_get_number(nvl, "match_tag_not");
517 rule->natpass = nvlist_get_number(nvl, "natpass");
518
519 rule->keep_state = nvlist_get_number(nvl, "keep_state");
520 rule->af = nvlist_get_number(nvl, "af");
521 rule->proto = nvlist_get_number(nvl, "proto");
522 rule->type = nvlist_get_number(nvl, "type");
523 rule->code = nvlist_get_number(nvl, "code");
524 rule->flags = nvlist_get_number(nvl, "flags");
525 rule->flagset = nvlist_get_number(nvl, "flagset");
526 rule->min_ttl = nvlist_get_number(nvl, "min_ttl");
527 rule->allow_opts = nvlist_get_number(nvl, "allow_opts");
528 rule->rt = nvlist_get_number(nvl, "rt");
529 rule->return_ttl = nvlist_get_number(nvl, "return_ttl");
530 rule->tos = nvlist_get_number(nvl, "tos");
531 rule->set_tos = nvlist_get_number(nvl, "set_tos");
532 rule->anchor_relative = nvlist_get_number(nvl, "anchor_relative");
533 rule->anchor_wildcard = nvlist_get_number(nvl, "anchor_wildcard");
534
535 rule->flush = nvlist_get_number(nvl, "flush");
536 rule->prio = nvlist_get_number(nvl, "prio");
537 pf_nvuint_8_array(nvl, "set_prio", 2, rule->set_prio, NULL);
538
539 pf_nvdivert_to_divert(nvlist_get_nvlist(nvl, "divert"), rule);
540
541 rule->states_cur = nvlist_get_number(nvl, "states_cur");
542 rule->states_tot = nvlist_get_number(nvl, "states_tot");
543 rule->src_nodes = nvlist_get_number(nvl, "src_nodes");
544 }
545
546 int
pfctl_add_rule(int dev,const struct pfctl_rule * r,const char * anchor,const char * anchor_call,uint32_t ticket,uint32_t pool_ticket)547 pfctl_add_rule(int dev, const struct pfctl_rule *r, const char *anchor,
548 const char *anchor_call, uint32_t ticket, uint32_t pool_ticket)
549 {
550 struct pfioc_nv nv;
551 uint64_t timeouts[PFTM_MAX];
552 uint64_t set_prio[2];
553 nvlist_t *nvl, *nvlr;
554 size_t labelcount;
555 int ret;
556
557 nvl = nvlist_create(0);
558 nvlr = nvlist_create(0);
559
560 nvlist_add_number(nvl, "ticket", ticket);
561 nvlist_add_number(nvl, "pool_ticket", pool_ticket);
562 nvlist_add_string(nvl, "anchor", anchor);
563 nvlist_add_string(nvl, "anchor_call", anchor_call);
564
565 nvlist_add_number(nvlr, "nr", r->nr);
566 pfctl_nv_add_rule_addr(nvlr, "src", &r->src);
567 pfctl_nv_add_rule_addr(nvlr, "dst", &r->dst);
568
569 labelcount = 0;
570 while (r->label[labelcount][0] != 0 &&
571 labelcount < PF_RULE_MAX_LABEL_COUNT) {
572 nvlist_append_string_array(nvlr, "labels",
573 r->label[labelcount]);
574 labelcount++;
575 }
576 nvlist_add_number(nvlr, "ridentifier", r->ridentifier);
577
578 nvlist_add_string(nvlr, "ifname", r->ifname);
579 nvlist_add_string(nvlr, "qname", r->qname);
580 nvlist_add_string(nvlr, "pqname", r->pqname);
581 nvlist_add_string(nvlr, "tagname", r->tagname);
582 nvlist_add_string(nvlr, "match_tagname", r->match_tagname);
583 nvlist_add_string(nvlr, "overload_tblname", r->overload_tblname);
584
585 pfctl_nv_add_pool(nvlr, "rpool", &r->rpool);
586
587 nvlist_add_number(nvlr, "os_fingerprint", r->os_fingerprint);
588
589 nvlist_add_number(nvlr, "rtableid", r->rtableid);
590 for (int i = 0; i < PFTM_MAX; i++)
591 timeouts[i] = r->timeout[i];
592 nvlist_add_number_array(nvlr, "timeout", timeouts, PFTM_MAX);
593 nvlist_add_number(nvlr, "max_states", r->max_states);
594 nvlist_add_number(nvlr, "max_src_nodes", r->max_src_nodes);
595 nvlist_add_number(nvlr, "max_src_states", r->max_src_states);
596 nvlist_add_number(nvlr, "max_src_conn", r->max_src_conn);
597 nvlist_add_number(nvlr, "max_src_conn_rate.limit",
598 r->max_src_conn_rate.limit);
599 nvlist_add_number(nvlr, "max_src_conn_rate.seconds",
600 r->max_src_conn_rate.seconds);
601 nvlist_add_number(nvlr, "prob", r->prob);
602 nvlist_add_number(nvlr, "cuid", r->cuid);
603 nvlist_add_number(nvlr, "cpid", r->cpid);
604
605 nvlist_add_number(nvlr, "return_icmp", r->return_icmp);
606 nvlist_add_number(nvlr, "return_icmp6", r->return_icmp6);
607
608 nvlist_add_number(nvlr, "max_mss", r->max_mss);
609 nvlist_add_number(nvlr, "scrub_flags", r->scrub_flags);
610
611 pfctl_nv_add_uid(nvlr, "uid", &r->uid);
612 pfctl_nv_add_uid(nvlr, "gid", (const struct pf_rule_uid *)&r->gid);
613
614 nvlist_add_number(nvlr, "rule_flag", r->rule_flag);
615 nvlist_add_number(nvlr, "action", r->action);
616 nvlist_add_number(nvlr, "direction", r->direction);
617 nvlist_add_number(nvlr, "log", r->log);
618 nvlist_add_number(nvlr, "logif", r->logif);
619 nvlist_add_number(nvlr, "quick", r->quick);
620 nvlist_add_number(nvlr, "ifnot", r->ifnot);
621 nvlist_add_number(nvlr, "match_tag_not", r->match_tag_not);
622 nvlist_add_number(nvlr, "natpass", r->natpass);
623
624 nvlist_add_number(nvlr, "keep_state", r->keep_state);
625 nvlist_add_number(nvlr, "af", r->af);
626 nvlist_add_number(nvlr, "proto", r->proto);
627 nvlist_add_number(nvlr, "type", r->type);
628 nvlist_add_number(nvlr, "code", r->code);
629 nvlist_add_number(nvlr, "flags", r->flags);
630 nvlist_add_number(nvlr, "flagset", r->flagset);
631 nvlist_add_number(nvlr, "min_ttl", r->min_ttl);
632 nvlist_add_number(nvlr, "allow_opts", r->allow_opts);
633 nvlist_add_number(nvlr, "rt", r->rt);
634 nvlist_add_number(nvlr, "return_ttl", r->return_ttl);
635 nvlist_add_number(nvlr, "tos", r->tos);
636 nvlist_add_number(nvlr, "set_tos", r->set_tos);
637 nvlist_add_number(nvlr, "anchor_relative", r->anchor_relative);
638 nvlist_add_number(nvlr, "anchor_wildcard", r->anchor_wildcard);
639
640 nvlist_add_number(nvlr, "flush", r->flush);
641
642 nvlist_add_number(nvlr, "prio", r->prio);
643 set_prio[0] = r->set_prio[0];
644 set_prio[1] = r->set_prio[1];
645 nvlist_add_number_array(nvlr, "set_prio", set_prio, 2);
646
647 pfctl_nv_add_divert(nvlr, "divert", r);
648
649 nvlist_add_nvlist(nvl, "rule", nvlr);
650 nvlist_destroy(nvlr);
651
652 /* Now do the call. */
653 nv.data = nvlist_pack(nvl, &nv.len);
654 nv.size = nv.len;
655
656 ret = ioctl(dev, DIOCADDRULENV, &nv);
657
658 free(nv.data);
659 nvlist_destroy(nvl);
660
661 return (ret);
662 }
663
664 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)665 pfctl_get_rule(int dev, uint32_t nr, uint32_t ticket, const char *anchor,
666 uint32_t ruleset, struct pfctl_rule *rule, char *anchor_call)
667 {
668 return (pfctl_get_clear_rule(dev, nr, ticket, anchor, ruleset, rule,
669 anchor_call, false));
670 }
671
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)672 int pfctl_get_clear_rule(int dev, uint32_t nr, uint32_t ticket,
673 const char *anchor, uint32_t ruleset, struct pfctl_rule *rule,
674 char *anchor_call, bool clear)
675 {
676 struct pfioc_nv nv;
677 nvlist_t *nvl;
678 void *nvlpacked;
679 int ret;
680
681 nvl = nvlist_create(0);
682 if (nvl == 0)
683 return (ENOMEM);
684
685 nvlist_add_number(nvl, "nr", nr);
686 nvlist_add_number(nvl, "ticket", ticket);
687 nvlist_add_string(nvl, "anchor", anchor);
688 nvlist_add_number(nvl, "ruleset", ruleset);
689
690 if (clear)
691 nvlist_add_bool(nvl, "clear_counter", true);
692
693 nvlpacked = nvlist_pack(nvl, &nv.len);
694 if (nvlpacked == NULL) {
695 nvlist_destroy(nvl);
696 return (ENOMEM);
697 }
698 nv.data = malloc(8182);
699 nv.size = 8192;
700 assert(nv.len <= nv.size);
701 memcpy(nv.data, nvlpacked, nv.len);
702 nvlist_destroy(nvl);
703 nvl = NULL;
704 free(nvlpacked);
705
706 ret = ioctl(dev, DIOCGETRULENV, &nv);
707 if (ret != 0) {
708 free(nv.data);
709 return (ret);
710 }
711
712 nvl = nvlist_unpack(nv.data, nv.len, 0);
713 if (nvl == NULL) {
714 free(nv.data);
715 return (EIO);
716 }
717
718 pf_nvrule_to_rule(nvlist_get_nvlist(nvl, "rule"), rule);
719
720 if (anchor_call)
721 strlcpy(anchor_call, nvlist_get_string(nvl, "anchor_call"),
722 MAXPATHLEN);
723
724 free(nv.data);
725 nvlist_destroy(nvl);
726
727 return (0);
728 }
729
730 int
pfctl_set_keepcounters(int dev,bool keep)731 pfctl_set_keepcounters(int dev, bool keep)
732 {
733 struct pfioc_nv nv;
734 nvlist_t *nvl;
735 int ret;
736
737 nvl = nvlist_create(0);
738
739 nvlist_add_bool(nvl, "keep_counters", keep);
740
741 nv.data = nvlist_pack(nvl, &nv.len);
742 nv.size = nv.len;
743
744 nvlist_destroy(nvl);
745
746 ret = ioctl(dev, DIOCKEEPCOUNTERS, &nv);
747
748 free(nv.data);
749 return (ret);
750 }
751
752 static void
pfctl_nv_add_state_cmp(nvlist_t * nvl,const char * name,const struct pfctl_state_cmp * cmp)753 pfctl_nv_add_state_cmp(nvlist_t *nvl, const char *name,
754 const struct pfctl_state_cmp *cmp)
755 {
756 nvlist_t *nv;
757
758 nv = nvlist_create(0);
759
760 nvlist_add_number(nv, "id", cmp->id);
761 nvlist_add_number(nv, "creatorid", htonl(cmp->creatorid));
762 nvlist_add_number(nv, "direction", cmp->direction);
763
764 nvlist_add_nvlist(nvl, name, nv);
765 nvlist_destroy(nv);
766 }
767
768 static void
pf_state_key_export_to_state_key(struct pfctl_state_key * ps,const struct pf_state_key_export * s)769 pf_state_key_export_to_state_key(struct pfctl_state_key *ps,
770 const struct pf_state_key_export *s)
771 {
772 bcopy(s->addr, ps->addr, sizeof(ps->addr[0]) * 2);
773 ps->port[0] = s->port[0];
774 ps->port[1] = s->port[1];
775 }
776
777 static void
pf_state_peer_export_to_state_peer(struct pfctl_state_peer * ps,const struct pf_state_peer_export * s)778 pf_state_peer_export_to_state_peer(struct pfctl_state_peer *ps,
779 const struct pf_state_peer_export *s)
780 {
781 /* Ignore scrub. */
782 ps->seqlo = s->seqlo;
783 ps->seqhi = s->seqhi;
784 ps->seqdiff = s->seqdiff;
785 /* Ignore max_win & mss */
786 ps->state = s->state;
787 ps->wscale = s->wscale;
788 }
789
790 static void
pf_state_export_to_state(struct pfctl_state * ps,const struct pf_state_export * s)791 pf_state_export_to_state(struct pfctl_state *ps, const struct pf_state_export *s)
792 {
793 assert(s->version >= PF_STATE_VERSION);
794
795 ps->id = s->id;
796 strlcpy(ps->ifname, s->ifname, sizeof(ps->ifname));
797 strlcpy(ps->orig_ifname, s->orig_ifname, sizeof(ps->orig_ifname));
798 pf_state_key_export_to_state_key(&ps->key[0], &s->key[0]);
799 pf_state_key_export_to_state_key(&ps->key[1], &s->key[1]);
800 pf_state_peer_export_to_state_peer(&ps->src, &s->src);
801 pf_state_peer_export_to_state_peer(&ps->dst, &s->dst);
802 bcopy(&s->rt_addr, &ps->rt_addr, sizeof(ps->rt_addr));
803 ps->rule = ntohl(s->rule);
804 ps->anchor = ntohl(s->anchor);
805 ps->nat_rule = ntohl(s->nat_rule);
806 ps->creation = ntohl(s->creation);
807 ps->expire = ntohl(s->expire);
808 ps->packets[0] = s->packets[0];
809 ps->packets[1] = s->packets[1];
810 ps->bytes[0] = s->bytes[0];
811 ps->bytes[1] = s->bytes[1];
812 ps->creatorid = ntohl(s->creatorid);
813 ps->key[0].proto = s->proto;
814 ps->key[1].proto = s->proto;
815 ps->key[0].af = s->af;
816 ps->key[1].af = s->af;
817 ps->direction = s->direction;
818 ps->state_flags = s->state_flags;
819 ps->sync_flags = s->sync_flags;
820 }
821
822 int
pfctl_get_states(int dev,struct pfctl_states * states)823 pfctl_get_states(int dev, struct pfctl_states *states)
824 {
825 struct pfioc_states_v2 ps;
826 struct pf_state_export *p;
827 char *inbuf = NULL, *newinbuf = NULL;
828 unsigned int len = 0;
829 int i, error;
830
831 bzero(&ps, sizeof(ps));
832 ps.ps_req_version = PF_STATE_VERSION;
833
834 bzero(states, sizeof(*states));
835 TAILQ_INIT(&states->states);
836
837 for (;;) {
838 ps.ps_len = len;
839 if (len) {
840 newinbuf = realloc(inbuf, len);
841 if (newinbuf == NULL)
842 return (ENOMEM);
843 ps.ps_buf = inbuf = newinbuf;
844 }
845 if ((error = ioctl(dev, DIOCGETSTATESV2, &ps)) < 0) {
846 free(inbuf);
847 return (error);
848 }
849 if (ps.ps_len + sizeof(struct pfioc_states_v2) < len)
850 break;
851 if (len == 0 && ps.ps_len == 0)
852 goto out;
853 if (len == 0 && ps.ps_len != 0)
854 len = ps.ps_len;
855 if (ps.ps_len == 0)
856 goto out; /* no states */
857 len *= 2;
858 }
859 p = ps.ps_states;
860
861 for (i = 0; i < ps.ps_len; i += sizeof(*p), p++) {
862 struct pfctl_state *s = malloc(sizeof(*s));
863 if (s == NULL) {
864 pfctl_free_states(states);
865 error = ENOMEM;
866 goto out;
867 }
868
869 pf_state_export_to_state(s, p);
870 TAILQ_INSERT_TAIL(&states->states, s, entry);
871 }
872
873 out:
874 free(inbuf);
875 return (error);
876 }
877
878 void
pfctl_free_states(struct pfctl_states * states)879 pfctl_free_states(struct pfctl_states *states)
880 {
881 struct pfctl_state *s, *tmp;
882
883 TAILQ_FOREACH_SAFE(s, &states->states, entry, tmp) {
884 free(s);
885 }
886
887 bzero(states, sizeof(*states));
888 }
889
890 static int
_pfctl_clear_states(int dev,const struct pfctl_kill * kill,unsigned int * killed,uint64_t ioctlval)891 _pfctl_clear_states(int dev, const struct pfctl_kill *kill,
892 unsigned int *killed, uint64_t ioctlval)
893 {
894 struct pfioc_nv nv;
895 nvlist_t *nvl;
896 int ret;
897
898 nvl = nvlist_create(0);
899
900 pfctl_nv_add_state_cmp(nvl, "cmp", &kill->cmp);
901 nvlist_add_number(nvl, "af", kill->af);
902 nvlist_add_number(nvl, "proto", kill->proto);
903 pfctl_nv_add_rule_addr(nvl, "src", &kill->src);
904 pfctl_nv_add_rule_addr(nvl, "dst", &kill->dst);
905 pfctl_nv_add_rule_addr(nvl, "rt_addr", &kill->rt_addr);
906 nvlist_add_string(nvl, "ifname", kill->ifname);
907 nvlist_add_string(nvl, "label", kill->label);
908 nvlist_add_bool(nvl, "kill_match", kill->kill_match);
909
910 nv.data = nvlist_pack(nvl, &nv.len);
911 nv.size = nv.len;
912 nvlist_destroy(nvl);
913 nvl = NULL;
914
915 ret = ioctl(dev, ioctlval, &nv);
916 if (ret != 0) {
917 free(nv.data);
918 return (ret);
919 }
920
921 nvl = nvlist_unpack(nv.data, nv.len, 0);
922 if (nvl == NULL) {
923 free(nv.data);
924 return (EIO);
925 }
926
927 if (killed)
928 *killed = nvlist_get_number(nvl, "killed");
929
930 nvlist_destroy(nvl);
931 free(nv.data);
932
933 return (ret);
934 }
935
936 int
pfctl_clear_states(int dev,const struct pfctl_kill * kill,unsigned int * killed)937 pfctl_clear_states(int dev, const struct pfctl_kill *kill,
938 unsigned int *killed)
939 {
940 return (_pfctl_clear_states(dev, kill, killed, DIOCCLRSTATESNV));
941 }
942
943 int
pfctl_kill_states(int dev,const struct pfctl_kill * kill,unsigned int * killed)944 pfctl_kill_states(int dev, const struct pfctl_kill *kill, unsigned int *killed)
945 {
946 return (_pfctl_clear_states(dev, kill, killed, DIOCKILLSTATESNV));
947 }
948
949 static int
pfctl_get_limit(int dev,const int index,uint * limit)950 pfctl_get_limit(int dev, const int index, uint *limit)
951 {
952 struct pfioc_limit pl;
953
954 bzero(&pl, sizeof(pl));
955 pl.index = index;
956
957 if (ioctl(dev, DIOCGETLIMIT, &pl) == -1)
958 return (errno);
959
960 *limit = pl.limit;
961
962 return (0);
963 }
964
965 int
pfctl_set_syncookies(int dev,const struct pfctl_syncookies * s)966 pfctl_set_syncookies(int dev, const struct pfctl_syncookies *s)
967 {
968 struct pfioc_nv nv;
969 nvlist_t *nvl;
970 int ret;
971 uint state_limit;
972
973 ret = pfctl_get_limit(dev, PF_LIMIT_STATES, &state_limit);
974 if (ret != 0)
975 return (ret);
976
977 nvl = nvlist_create(0);
978
979 nvlist_add_bool(nvl, "enabled", s->mode != PFCTL_SYNCOOKIES_NEVER);
980 nvlist_add_bool(nvl, "adaptive", s->mode == PFCTL_SYNCOOKIES_ADAPTIVE);
981 nvlist_add_number(nvl, "highwater", state_limit * s->highwater / 100);
982 nvlist_add_number(nvl, "lowwater", state_limit * s->lowwater / 100);
983
984 nv.data = nvlist_pack(nvl, &nv.len);
985 nv.size = nv.len;
986 nvlist_destroy(nvl);
987 nvl = NULL;
988
989 ret = ioctl(dev, DIOCSETSYNCOOKIES, &nv);
990
991 free(nv.data);
992 return (ret);
993 }
994
995 int
pfctl_get_syncookies(int dev,struct pfctl_syncookies * s)996 pfctl_get_syncookies(int dev, struct pfctl_syncookies *s)
997 {
998 struct pfioc_nv nv;
999 nvlist_t *nvl;
1000 int ret;
1001 uint state_limit;
1002 bool enabled, adaptive;
1003
1004 ret = pfctl_get_limit(dev, PF_LIMIT_STATES, &state_limit);
1005 if (ret != 0)
1006 return (ret);
1007
1008 bzero(s, sizeof(*s));
1009
1010 nv.data = malloc(256);
1011 nv.len = nv.size = 256;
1012
1013 if (ioctl(dev, DIOCGETSYNCOOKIES, &nv)) {
1014 free(nv.data);
1015 return (errno);
1016 }
1017
1018 nvl = nvlist_unpack(nv.data, nv.len, 0);
1019 free(nv.data);
1020 if (nvl == NULL) {
1021 return (EIO);
1022 }
1023
1024 enabled = nvlist_get_bool(nvl, "enabled");
1025 adaptive = nvlist_get_bool(nvl, "adaptive");
1026
1027 if (enabled) {
1028 if (adaptive)
1029 s->mode = PFCTL_SYNCOOKIES_ADAPTIVE;
1030 else
1031 s->mode = PFCTL_SYNCOOKIES_ALWAYS;
1032 } else {
1033 s->mode = PFCTL_SYNCOOKIES_NEVER;
1034 }
1035
1036 s->highwater = nvlist_get_number(nvl, "highwater") * 100 / state_limit;
1037 s->lowwater = nvlist_get_number(nvl, "lowwater") * 100 / state_limit;
1038
1039 nvlist_destroy(nvl);
1040
1041 return (0);
1042 }
1043