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 static void 54 pf_nvuint_8_array(const nvlist_t *nvl, const char *name, size_t maxelems, 55 u_int8_t *numbers, size_t *nelems) 56 { 57 const uint64_t *tmp; 58 size_t elems; 59 60 tmp = nvlist_get_number_array(nvl, name, &elems); 61 assert(elems <= maxelems); 62 63 for (size_t i = 0; i < elems; i++) 64 numbers[i] = tmp[i]; 65 66 if (nelems) 67 *nelems = elems; 68 } 69 70 static void 71 pf_nvuint_16_array(const nvlist_t *nvl, const char *name, size_t maxelems, 72 u_int16_t *numbers, size_t *nelems) 73 { 74 const uint64_t *tmp; 75 size_t elems; 76 77 tmp = nvlist_get_number_array(nvl, name, &elems); 78 assert(elems <= maxelems); 79 80 for (size_t i = 0; i < elems; i++) 81 numbers[i] = tmp[i]; 82 83 if (nelems) 84 *nelems = elems; 85 } 86 87 static void 88 pf_nvuint_32_array(const nvlist_t *nvl, const char *name, size_t maxelems, 89 u_int32_t *numbers, size_t *nelems) 90 { 91 const uint64_t *tmp; 92 size_t elems; 93 94 tmp = nvlist_get_number_array(nvl, name, &elems); 95 assert(elems <= maxelems); 96 97 for (size_t i = 0; i < elems; i++) 98 numbers[i] = tmp[i]; 99 100 if (nelems) 101 *nelems = elems; 102 } 103 104 static void 105 pf_nvuint_64_array(const nvlist_t *nvl, const char *name, size_t maxelems, 106 u_int64_t *numbers, size_t *nelems) 107 { 108 const uint64_t *tmp; 109 size_t elems; 110 111 tmp = nvlist_get_number_array(nvl, name, &elems); 112 assert(elems <= maxelems); 113 114 for (size_t i = 0; i < elems; i++) 115 numbers[i] = tmp[i]; 116 117 if (nelems) 118 *nelems = elems; 119 } 120 121 static void 122 pfctl_nv_add_addr(nvlist_t *nvparent, const char *name, 123 const struct pf_addr *addr) 124 { 125 nvlist_t *nvl = nvlist_create(0); 126 127 nvlist_add_binary(nvl, "addr", addr, sizeof(*addr)); 128 129 nvlist_add_nvlist(nvparent, name, nvl); 130 } 131 132 static void 133 pf_nvaddr_to_addr(const nvlist_t *nvl, struct pf_addr *addr) 134 { 135 size_t len; 136 const void *data; 137 138 data = nvlist_get_binary(nvl, "addr", &len); 139 assert(len == sizeof(struct pf_addr)); 140 memcpy(addr, data, len); 141 } 142 143 static void 144 pfctl_nv_add_addr_wrap(nvlist_t *nvparent, const char *name, 145 const struct pf_addr_wrap *addr) 146 { 147 nvlist_t *nvl = nvlist_create(0); 148 149 nvlist_add_number(nvl, "type", addr->type); 150 nvlist_add_number(nvl, "iflags", addr->iflags); 151 if (addr->type == PF_ADDR_DYNIFTL) 152 nvlist_add_string(nvl, "ifname", addr->v.ifname); 153 if (addr->type == PF_ADDR_TABLE) 154 nvlist_add_string(nvl, "tblname", addr->v.tblname); 155 pfctl_nv_add_addr(nvl, "addr", &addr->v.a.addr); 156 pfctl_nv_add_addr(nvl, "mask", &addr->v.a.mask); 157 158 nvlist_add_nvlist(nvparent, name, nvl); 159 } 160 161 static void 162 pf_nvaddr_wrap_to_addr_wrap(const nvlist_t *nvl, struct pf_addr_wrap *addr) 163 { 164 addr->type = nvlist_get_number(nvl, "type"); 165 addr->iflags = nvlist_get_number(nvl, "iflags"); 166 if (addr->type == PF_ADDR_DYNIFTL) 167 strlcpy(addr->v.ifname, nvlist_get_string(nvl, "ifname"), 168 IFNAMSIZ); 169 if (addr->type == PF_ADDR_TABLE) 170 strlcpy(addr->v.tblname, nvlist_get_string(nvl, "tblname"), 171 PF_TABLE_NAME_SIZE); 172 173 pf_nvaddr_to_addr(nvlist_get_nvlist(nvl, "addr"), &addr->v.a.addr); 174 pf_nvaddr_to_addr(nvlist_get_nvlist(nvl, "mask"), &addr->v.a.mask); 175 } 176 177 static void 178 pfctl_nv_add_rule_addr(nvlist_t *nvparent, const char *name, 179 const struct pf_rule_addr *addr) 180 { 181 u_int64_t ports[2]; 182 nvlist_t *nvl = nvlist_create(0); 183 184 pfctl_nv_add_addr_wrap(nvl, "addr", &addr->addr); 185 ports[0] = addr->port[0]; 186 ports[1] = addr->port[1]; 187 nvlist_add_number_array(nvl, "port", ports, 2); 188 nvlist_add_number(nvl, "neg", addr->neg); 189 nvlist_add_number(nvl, "port_op", addr->port_op); 190 191 nvlist_add_nvlist(nvparent, name, nvl); 192 } 193 194 static void 195 pf_nvrule_addr_to_rule_addr(const nvlist_t *nvl, struct pf_rule_addr *addr) 196 { 197 pf_nvaddr_wrap_to_addr_wrap(nvlist_get_nvlist(nvl, "addr"), &addr->addr); 198 199 pf_nvuint_16_array(nvl, "port", 2, addr->port, NULL); 200 addr->neg = nvlist_get_number(nvl, "neg"); 201 addr->port_op = nvlist_get_number(nvl, "port_op"); 202 } 203 204 static void 205 pfctl_nv_add_mape(nvlist_t *nvparent, const char *name, 206 const struct pf_mape_portset *mape) 207 { 208 nvlist_t *nvl = nvlist_create(0); 209 210 nvlist_add_number(nvl, "offset", mape->offset); 211 nvlist_add_number(nvl, "psidlen", mape->psidlen); 212 nvlist_add_number(nvl, "psid", mape->psid); 213 nvlist_add_nvlist(nvparent, name, nvl); 214 } 215 216 static void 217 pfctl_nv_add_pool(nvlist_t *nvparent, const char *name, 218 const struct pfctl_pool *pool) 219 { 220 u_int64_t ports[2]; 221 nvlist_t *nvl = nvlist_create(0); 222 223 nvlist_add_binary(nvl, "key", &pool->key, sizeof(pool->key)); 224 pfctl_nv_add_addr(nvl, "counter", &pool->counter); 225 nvlist_add_number(nvl, "tblidx", pool->tblidx); 226 227 ports[0] = pool->proxy_port[0]; 228 ports[1] = pool->proxy_port[1]; 229 nvlist_add_number_array(nvl, "proxy_port", ports, 2); 230 nvlist_add_number(nvl, "opts", pool->opts); 231 pfctl_nv_add_mape(nvl, "mape", &pool->mape); 232 233 nvlist_add_nvlist(nvparent, name, nvl); 234 } 235 236 static void 237 pf_nvmape_to_mape(const nvlist_t *nvl, struct pf_mape_portset *mape) 238 { 239 mape->offset = nvlist_get_number(nvl, "offset"); 240 mape->psidlen = nvlist_get_number(nvl, "psidlen"); 241 mape->psid = nvlist_get_number(nvl, "psid"); 242 } 243 244 static void 245 pf_nvpool_to_pool(const nvlist_t *nvl, struct pfctl_pool *pool) 246 { 247 size_t len; 248 const void *data; 249 250 data = nvlist_get_binary(nvl, "key", &len); 251 assert(len == sizeof(pool->key)); 252 memcpy(&pool->key, data, len); 253 254 pf_nvaddr_to_addr(nvlist_get_nvlist(nvl, "counter"), &pool->counter); 255 256 pool->tblidx = nvlist_get_number(nvl, "tblidx"); 257 pf_nvuint_16_array(nvl, "proxy_port", 2, pool->proxy_port, NULL); 258 pool->opts = nvlist_get_number(nvl, "opts"); 259 260 if (nvlist_exists_nvlist(nvl, "mape")) 261 pf_nvmape_to_mape(nvlist_get_nvlist(nvl, "mape"), &pool->mape); 262 } 263 264 static void 265 pfctl_nv_add_uid(nvlist_t *nvparent, const char *name, 266 const struct pf_rule_uid *uid) 267 { 268 u_int64_t uids[2]; 269 nvlist_t *nvl = nvlist_create(0); 270 271 uids[0] = uid->uid[0]; 272 uids[1] = uid->uid[1]; 273 nvlist_add_number_array(nvl, "uid", uids, 2); 274 nvlist_add_number(nvl, "op", uid->op); 275 276 nvlist_add_nvlist(nvparent, name, nvl); 277 } 278 279 static void 280 pf_nvrule_uid_to_rule_uid(const nvlist_t *nvl, struct pf_rule_uid *uid) 281 { 282 pf_nvuint_32_array(nvl, "uid", 2, uid->uid, NULL); 283 uid->op = nvlist_get_number(nvl, "op"); 284 } 285 286 static void 287 pfctl_nv_add_divert(nvlist_t *nvparent, const char *name, 288 const struct pfctl_rule *r) 289 { 290 nvlist_t *nvl = nvlist_create(0); 291 292 pfctl_nv_add_addr(nvl, "addr", &r->divert.addr); 293 nvlist_add_number(nvl, "port", r->divert.port); 294 295 nvlist_add_nvlist(nvparent, name, nvl); 296 } 297 298 static void 299 pf_nvdivert_to_divert(const nvlist_t *nvl, struct pfctl_rule *rule) 300 { 301 pf_nvaddr_to_addr(nvlist_get_nvlist(nvl, "addr"), &rule->divert.addr); 302 rule->divert.port = nvlist_get_number(nvl, "port"); 303 } 304 305 static void 306 pf_nvrule_to_rule(const nvlist_t *nvl, struct pfctl_rule *rule) 307 { 308 const uint64_t *skip; 309 const char *const *labels; 310 size_t skipcount, labelcount; 311 312 rule->nr = nvlist_get_number(nvl, "nr"); 313 314 pf_nvrule_addr_to_rule_addr(nvlist_get_nvlist(nvl, "src"), &rule->src); 315 pf_nvrule_addr_to_rule_addr(nvlist_get_nvlist(nvl, "dst"), &rule->dst); 316 317 skip = nvlist_get_number_array(nvl, "skip", &skipcount); 318 assert(skip); 319 assert(skipcount == PF_SKIP_COUNT); 320 for (int i = 0; i < PF_SKIP_COUNT; i++) 321 rule->skip[i].nr = skip[i]; 322 323 labels = nvlist_get_string_array(nvl, "labels", &labelcount); 324 assert(labelcount <= PF_RULE_MAX_LABEL_COUNT); 325 for (size_t i = 0; i < labelcount; i++) 326 strlcpy(rule->label[i], labels[i], PF_RULE_LABEL_SIZE); 327 strlcpy(rule->ifname, nvlist_get_string(nvl, "ifname"), IFNAMSIZ); 328 strlcpy(rule->qname, nvlist_get_string(nvl, "qname"), PF_QNAME_SIZE); 329 strlcpy(rule->pqname, nvlist_get_string(nvl, "pqname"), PF_QNAME_SIZE); 330 strlcpy(rule->tagname, nvlist_get_string(nvl, "tagname"), 331 PF_TAG_NAME_SIZE); 332 strlcpy(rule->match_tagname, nvlist_get_string(nvl, "match_tagname"), 333 PF_TAG_NAME_SIZE); 334 335 strlcpy(rule->overload_tblname, nvlist_get_string(nvl, "overload_tblname"), 336 PF_TABLE_NAME_SIZE); 337 338 pf_nvpool_to_pool(nvlist_get_nvlist(nvl, "rpool"), &rule->rpool); 339 340 rule->evaluations = nvlist_get_number(nvl, "evaluations"); 341 pf_nvuint_64_array(nvl, "packets", 2, rule->packets, NULL); 342 pf_nvuint_64_array(nvl, "bytes", 2, rule->bytes, NULL); 343 344 rule->os_fingerprint = nvlist_get_number(nvl, "os_fingerprint"); 345 346 rule->rtableid = nvlist_get_number(nvl, "rtableid"); 347 pf_nvuint_32_array(nvl, "timeout", PFTM_MAX, rule->timeout, NULL); 348 rule->max_states = nvlist_get_number(nvl, "max_states"); 349 rule->max_src_nodes = nvlist_get_number(nvl, "max_src_nodes"); 350 rule->max_src_states = nvlist_get_number(nvl, "max_src_states"); 351 rule->max_src_conn = nvlist_get_number(nvl, "max_src_conn"); 352 rule->max_src_conn_rate.limit = 353 nvlist_get_number(nvl, "max_src_conn_rate.limit"); 354 rule->max_src_conn_rate.seconds = 355 nvlist_get_number(nvl, "max_src_conn_rate.seconds"); 356 rule->qid = nvlist_get_number(nvl, "qid"); 357 rule->pqid = nvlist_get_number(nvl, "pqid"); 358 rule->prob = nvlist_get_number(nvl, "prob"); 359 rule->cuid = nvlist_get_number(nvl, "cuid"); 360 rule->cpid = nvlist_get_number(nvl, "cpid"); 361 362 rule->return_icmp = nvlist_get_number(nvl, "return_icmp"); 363 rule->return_icmp6 = nvlist_get_number(nvl, "return_icmp6"); 364 rule->max_mss = nvlist_get_number(nvl, "max_mss"); 365 rule->scrub_flags = nvlist_get_number(nvl, "scrub_flags"); 366 367 pf_nvrule_uid_to_rule_uid(nvlist_get_nvlist(nvl, "uid"), &rule->uid); 368 pf_nvrule_uid_to_rule_uid(nvlist_get_nvlist(nvl, "gid"), 369 (struct pf_rule_uid *)&rule->gid); 370 371 rule->rule_flag = nvlist_get_number(nvl, "rule_flag"); 372 rule->action = nvlist_get_number(nvl, "action"); 373 rule->direction = nvlist_get_number(nvl, "direction"); 374 rule->log = nvlist_get_number(nvl, "log"); 375 rule->logif = nvlist_get_number(nvl, "logif"); 376 rule->quick = nvlist_get_number(nvl, "quick"); 377 rule->ifnot = nvlist_get_number(nvl, "ifnot"); 378 rule->match_tag_not = nvlist_get_number(nvl, "match_tag_not"); 379 rule->natpass = nvlist_get_number(nvl, "natpass"); 380 381 rule->keep_state = nvlist_get_number(nvl, "keep_state"); 382 rule->af = nvlist_get_number(nvl, "af"); 383 rule->proto = nvlist_get_number(nvl, "proto"); 384 rule->type = nvlist_get_number(nvl, "type"); 385 rule->code = nvlist_get_number(nvl, "code"); 386 rule->flags = nvlist_get_number(nvl, "flags"); 387 rule->flagset = nvlist_get_number(nvl, "flagset"); 388 rule->min_ttl = nvlist_get_number(nvl, "min_ttl"); 389 rule->allow_opts = nvlist_get_number(nvl, "allow_opts"); 390 rule->rt = nvlist_get_number(nvl, "rt"); 391 rule->return_ttl = nvlist_get_number(nvl, "return_ttl"); 392 rule->tos = nvlist_get_number(nvl, "tos"); 393 rule->set_tos = nvlist_get_number(nvl, "set_tos"); 394 rule->anchor_relative = nvlist_get_number(nvl, "anchor_relative"); 395 rule->anchor_wildcard = nvlist_get_number(nvl, "anchor_wildcard"); 396 397 rule->flush = nvlist_get_number(nvl, "flush"); 398 rule->prio = nvlist_get_number(nvl, "prio"); 399 pf_nvuint_8_array(nvl, "set_prio", 2, rule->set_prio, NULL); 400 401 pf_nvdivert_to_divert(nvlist_get_nvlist(nvl, "divert"), rule); 402 403 rule->states_cur = nvlist_get_number(nvl, "states_cur"); 404 rule->states_tot = nvlist_get_number(nvl, "states_tot"); 405 rule->src_nodes = nvlist_get_number(nvl, "src_nodes"); 406 } 407 408 int 409 pfctl_add_rule(int dev, const struct pfctl_rule *r, const char *anchor, 410 const char *anchor_call, u_int32_t ticket, u_int32_t pool_ticket) 411 { 412 struct pfioc_nv nv; 413 u_int64_t timeouts[PFTM_MAX]; 414 u_int64_t set_prio[2]; 415 nvlist_t *nvl, *nvlr; 416 size_t labelcount; 417 int ret; 418 419 nvl = nvlist_create(0); 420 nvlr = nvlist_create(0); 421 422 nvlist_add_number(nvl, "ticket", ticket); 423 nvlist_add_number(nvl, "pool_ticket", pool_ticket); 424 nvlist_add_string(nvl, "anchor", anchor); 425 nvlist_add_string(nvl, "anchor_call", anchor_call); 426 427 nvlist_add_number(nvlr, "nr", r->nr); 428 pfctl_nv_add_rule_addr(nvlr, "src", &r->src); 429 pfctl_nv_add_rule_addr(nvlr, "dst", &r->dst); 430 431 labelcount = 0; 432 while (r->label[labelcount][0] != 0 && 433 labelcount < PF_RULE_MAX_LABEL_COUNT) { 434 nvlist_append_string_array(nvlr, "labels", 435 r->label[labelcount]); 436 labelcount++; 437 } 438 439 nvlist_add_string(nvlr, "ifname", r->ifname); 440 nvlist_add_string(nvlr, "qname", r->qname); 441 nvlist_add_string(nvlr, "pqname", r->pqname); 442 nvlist_add_string(nvlr, "tagname", r->tagname); 443 nvlist_add_string(nvlr, "match_tagname", r->match_tagname); 444 nvlist_add_string(nvlr, "overload_tblname", r->overload_tblname); 445 446 pfctl_nv_add_pool(nvlr, "rpool", &r->rpool); 447 448 nvlist_add_number(nvlr, "os_fingerprint", r->os_fingerprint); 449 450 nvlist_add_number(nvlr, "rtableid", r->rtableid); 451 for (int i = 0; i < PFTM_MAX; i++) 452 timeouts[i] = r->timeout[i]; 453 nvlist_add_number_array(nvlr, "timeout", timeouts, PFTM_MAX); 454 nvlist_add_number(nvlr, "max_states", r->max_states); 455 nvlist_add_number(nvlr, "max_src_nodes", r->max_src_nodes); 456 nvlist_add_number(nvlr, "max_src_states", r->max_src_states); 457 nvlist_add_number(nvlr, "max_src_conn", r->max_src_conn); 458 nvlist_add_number(nvlr, "max_src_conn_rate.limit", 459 r->max_src_conn_rate.limit); 460 nvlist_add_number(nvlr, "max_src_conn_rate.seconds", 461 r->max_src_conn_rate.seconds); 462 nvlist_add_number(nvlr, "prob", r->prob); 463 nvlist_add_number(nvlr, "cuid", r->cuid); 464 nvlist_add_number(nvlr, "cpid", r->cpid); 465 466 nvlist_add_number(nvlr, "return_icmp", r->return_icmp); 467 nvlist_add_number(nvlr, "return_icmp6", r->return_icmp6); 468 469 nvlist_add_number(nvlr, "max_mss", r->max_mss); 470 nvlist_add_number(nvlr, "scrub_flags", r->scrub_flags); 471 472 pfctl_nv_add_uid(nvlr, "uid", &r->uid); 473 pfctl_nv_add_uid(nvlr, "gid", (const struct pf_rule_uid *)&r->gid); 474 475 nvlist_add_number(nvlr, "rule_flag", r->rule_flag); 476 nvlist_add_number(nvlr, "action", r->action); 477 nvlist_add_number(nvlr, "direction", r->direction); 478 nvlist_add_number(nvlr, "log", r->log); 479 nvlist_add_number(nvlr, "logif", r->logif); 480 nvlist_add_number(nvlr, "quick", r->quick); 481 nvlist_add_number(nvlr, "ifnot", r->ifnot); 482 nvlist_add_number(nvlr, "match_tag_not", r->match_tag_not); 483 nvlist_add_number(nvlr, "natpass", r->natpass); 484 485 nvlist_add_number(nvlr, "keep_state", r->keep_state); 486 nvlist_add_number(nvlr, "af", r->af); 487 nvlist_add_number(nvlr, "proto", r->proto); 488 nvlist_add_number(nvlr, "type", r->type); 489 nvlist_add_number(nvlr, "code", r->code); 490 nvlist_add_number(nvlr, "flags", r->flags); 491 nvlist_add_number(nvlr, "flagset", r->flagset); 492 nvlist_add_number(nvlr, "min_ttl", r->min_ttl); 493 nvlist_add_number(nvlr, "allow_opts", r->allow_opts); 494 nvlist_add_number(nvlr, "rt", r->rt); 495 nvlist_add_number(nvlr, "return_ttl", r->return_ttl); 496 nvlist_add_number(nvlr, "tos", r->tos); 497 nvlist_add_number(nvlr, "set_tos", r->set_tos); 498 nvlist_add_number(nvlr, "anchor_relative", r->anchor_relative); 499 nvlist_add_number(nvlr, "anchor_wildcard", r->anchor_wildcard); 500 501 nvlist_add_number(nvlr, "flush", r->flush); 502 503 nvlist_add_number(nvlr, "prio", r->prio); 504 set_prio[0] = r->set_prio[0]; 505 set_prio[1] = r->set_prio[1]; 506 nvlist_add_number_array(nvlr, "set_prio", set_prio, 2); 507 508 pfctl_nv_add_divert(nvlr, "divert", r); 509 510 nvlist_add_nvlist(nvl, "rule", nvlr); 511 512 /* Now do the call. */ 513 nv.data = nvlist_pack(nvl, &nv.len); 514 nv.size = nv.len; 515 516 ret = ioctl(dev, DIOCADDRULENV, &nv); 517 518 free(nv.data); 519 nvlist_destroy(nvl); 520 521 return (ret); 522 } 523 524 int 525 pfctl_get_rule(int dev, u_int32_t nr, u_int32_t ticket, const char *anchor, 526 u_int32_t ruleset, struct pfctl_rule *rule, char *anchor_call) 527 { 528 return (pfctl_get_clear_rule(dev, nr, ticket, anchor, ruleset, rule, 529 anchor_call, false)); 530 } 531 532 int pfctl_get_clear_rule(int dev, u_int32_t nr, u_int32_t ticket, 533 const char *anchor, u_int32_t ruleset, struct pfctl_rule *rule, 534 char *anchor_call, bool clear) 535 { 536 struct pfioc_nv nv; 537 nvlist_t *nvl; 538 void *nvlpacked; 539 int ret; 540 541 nvl = nvlist_create(0); 542 if (nvl == 0) 543 return (ENOMEM); 544 545 nvlist_add_number(nvl, "nr", nr); 546 nvlist_add_number(nvl, "ticket", ticket); 547 nvlist_add_string(nvl, "anchor", anchor); 548 nvlist_add_number(nvl, "ruleset", ruleset); 549 550 if (clear) 551 nvlist_add_bool(nvl, "clear_counter", true); 552 553 nvlpacked = nvlist_pack(nvl, &nv.len); 554 if (nvlpacked == NULL) { 555 nvlist_destroy(nvl); 556 return (ENOMEM); 557 } 558 nv.data = malloc(8182); 559 nv.size = 8192; 560 assert(nv.len <= nv.size); 561 memcpy(nv.data, nvlpacked, nv.len); 562 nvlist_destroy(nvl); 563 nvl = NULL; 564 free(nvlpacked); 565 566 ret = ioctl(dev, DIOCGETRULENV, &nv); 567 if (ret != 0) { 568 free(nv.data); 569 return (ret); 570 } 571 572 nvl = nvlist_unpack(nv.data, nv.len, 0); 573 if (nvl == NULL) { 574 free(nv.data); 575 return (EIO); 576 } 577 578 pf_nvrule_to_rule(nvlist_get_nvlist(nvl, "rule"), rule); 579 580 if (anchor_call) 581 strlcpy(anchor_call, nvlist_get_string(nvl, "anchor_call"), 582 MAXPATHLEN); 583 584 free(nv.data); 585 nvlist_destroy(nvl); 586 587 return (0); 588 } 589 590 int 591 pfctl_set_keepcounters(int dev, bool keep) 592 { 593 struct pfioc_nv nv; 594 nvlist_t *nvl; 595 int ret; 596 597 nvl = nvlist_create(0); 598 599 nvlist_add_bool(nvl, "keep_counters", keep); 600 601 nv.data = nvlist_pack(nvl, &nv.len); 602 nv.size = nv.len; 603 604 nvlist_destroy(nvl); 605 606 ret = ioctl(dev, DIOCKEEPCOUNTERS, &nv); 607 608 free(nv.data); 609 return (ret); 610 } 611