lib/libpfctl/libpfctl.c

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