1 /*- 2 * SPDX-License-Identifier: BSD-2-Clause 3 * 4 * Copyright (c) 2023 Alexander V. Chernikov <melifaro@FreeBSD.org> 5 * Copyright (c) 2023 Rubicon Communications, LLC (Netgate) 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 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 19 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 26 * SUCH DAMAGE. 27 * 28 */ 29 #include <sys/cdefs.h> 30 #include "opt_inet.h" 31 #include "opt_inet6.h" 32 33 #include <sys/param.h> 34 #include <sys/malloc.h> 35 #include <sys/mbuf.h> 36 #include <sys/priv.h> 37 #include <sys/socket.h> 38 #include <sys/ucred.h> 39 40 #include <net/pfvar.h> 41 42 #include <netlink/netlink.h> 43 #include <netlink/netlink_ctl.h> 44 #include <netlink/netlink_generic.h> 45 #include <netlink/netlink_message_writer.h> 46 47 #include <netpfil/pf/pf_nl.h> 48 49 #define DEBUG_MOD_NAME nl_pf 50 #define DEBUG_MAX_LEVEL LOG_DEBUG3 51 #include <netlink/netlink_debug.h> 52 _DECLARE_DEBUG(LOG_DEBUG); 53 54 static bool nlattr_add_pf_threshold(struct nl_writer *, int, 55 struct pf_kthreshold *); 56 57 struct nl_parsed_state { 58 uint8_t version; 59 uint32_t id; 60 uint32_t creatorid; 61 char ifname[IFNAMSIZ]; 62 uint16_t proto; 63 sa_family_t af; 64 struct pf_addr addr; 65 struct pf_addr mask; 66 }; 67 68 #define _IN(_field) offsetof(struct genlmsghdr, _field) 69 #define _OUT(_field) offsetof(struct nl_parsed_state, _field) 70 static const struct nlattr_parser nla_p_state[] = { 71 { .type = PF_ST_ID, .off = _OUT(id), .cb = nlattr_get_uint32 }, 72 { .type = PF_ST_CREATORID, .off = _OUT(creatorid), .cb = nlattr_get_uint32 }, 73 { .type = PF_ST_IFNAME, .arg = (const void *)IFNAMSIZ, .off = _OUT(ifname), .cb = nlattr_get_chara }, 74 { .type = PF_ST_AF, .off = _OUT(af), .cb = nlattr_get_uint8 }, 75 { .type = PF_ST_PROTO, .off = _OUT(proto), .cb = nlattr_get_uint16 }, 76 { .type = PF_ST_FILTER_ADDR, .off = _OUT(addr), .cb = nlattr_get_in6_addr }, 77 { .type = PF_ST_FILTER_MASK, .off = _OUT(mask), .cb = nlattr_get_in6_addr }, 78 }; 79 static const struct nlfield_parser nlf_p_generic[] = { 80 { .off_in = _IN(version), .off_out = _OUT(version), .cb = nlf_get_u8 }, 81 }; 82 #undef _IN 83 #undef _OUT 84 NL_DECLARE_PARSER(state_parser, struct genlmsghdr, nlf_p_generic, nla_p_state); 85 86 static void 87 dump_addr(struct nl_writer *nw, int attr, const struct pf_addr *addr, int af) 88 { 89 switch (af) { 90 case AF_INET: 91 nlattr_add(nw, attr, 4, &addr->v4); 92 break; 93 case AF_INET6: 94 nlattr_add(nw, attr, 16, &addr->v6); 95 break; 96 }; 97 } 98 99 static bool 100 dump_state_peer(struct nl_writer *nw, int attr, const struct pf_state_peer *peer) 101 { 102 int off = nlattr_add_nested(nw, attr); 103 if (off == 0) 104 return (false); 105 106 nlattr_add_u32(nw, PF_STP_SEQLO, peer->seqlo); 107 nlattr_add_u32(nw, PF_STP_SEQHI, peer->seqhi); 108 nlattr_add_u32(nw, PF_STP_SEQDIFF, peer->seqdiff); 109 nlattr_add_u16(nw, PF_STP_MAX_WIN, peer->max_win); 110 nlattr_add_u16(nw, PF_STP_MSS, peer->mss); 111 nlattr_add_u8(nw, PF_STP_STATE, peer->state); 112 nlattr_add_u8(nw, PF_STP_WSCALE, peer->wscale); 113 114 if (peer->scrub != NULL) { 115 struct pf_state_scrub *sc = peer->scrub; 116 uint16_t pfss_flags = sc->pfss_flags & PFSS_TIMESTAMP; 117 118 nlattr_add_u16(nw, PF_STP_PFSS_FLAGS, pfss_flags); 119 nlattr_add_u32(nw, PF_STP_PFSS_TS_MOD, sc->pfss_ts_mod); 120 nlattr_add_u8(nw, PF_STP_PFSS_TTL, sc->pfss_ttl); 121 nlattr_add_u8(nw, PF_STP_SCRUB_FLAG, PF_SCRUB_FLAG_VALID); 122 } 123 nlattr_set_len(nw, off); 124 125 return (true); 126 } 127 128 static bool 129 dump_state_key(struct nl_writer *nw, int attr, const struct pf_state_key *key) 130 { 131 int off = nlattr_add_nested(nw, attr); 132 if (off == 0) 133 return (false); 134 135 dump_addr(nw, PF_STK_ADDR0, &key->addr[0], key->af); 136 dump_addr(nw, PF_STK_ADDR1, &key->addr[1], key->af); 137 nlattr_add_u16(nw, PF_STK_PORT0, key->port[0]); 138 nlattr_add_u16(nw, PF_STK_PORT1, key->port[1]); 139 nlattr_add_u8(nw, PF_STK_AF, key->af); 140 nlattr_add_u16(nw, PF_STK_PROTO, key->proto); 141 142 nlattr_set_len(nw, off); 143 144 return (true); 145 } 146 147 static int 148 dump_state(struct nlpcb *nlp, const struct nlmsghdr *hdr, struct pf_kstate *s, 149 struct nl_pstate *npt) 150 { 151 struct nl_writer *nw = npt->nw; 152 int error = 0; 153 int af; 154 struct pf_state_key *key; 155 156 PF_STATE_LOCK_ASSERT(s); 157 158 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 159 goto enomem; 160 161 struct genlmsghdr *ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 162 ghdr_new->cmd = PFNL_CMD_GETSTATES; 163 ghdr_new->version = 0; 164 ghdr_new->reserved = 0; 165 166 nlattr_add_u64(nw, PF_ST_VERSION, PF_STATE_VERSION); 167 168 key = s->key[PF_SK_WIRE]; 169 if (!dump_state_key(nw, PF_ST_KEY_WIRE, key)) 170 goto enomem; 171 key = s->key[PF_SK_STACK]; 172 if (!dump_state_key(nw, PF_ST_KEY_STACK, key)) 173 goto enomem; 174 175 af = s->key[PF_SK_WIRE]->af; 176 nlattr_add_u8(nw, PF_ST_PROTO, s->key[PF_SK_WIRE]->proto); 177 nlattr_add_u8(nw, PF_ST_AF, af); 178 179 nlattr_add_string(nw, PF_ST_IFNAME, s->kif->pfik_name); 180 nlattr_add_string(nw, PF_ST_ORIG_IFNAME, s->orig_kif->pfik_name); 181 dump_addr(nw, PF_ST_RT_ADDR, &s->act.rt_addr, s->act.rt_af); 182 nlattr_add_u32(nw, PF_ST_CREATION, time_uptime - (s->creation / 1000)); 183 uint32_t expire = pf_state_expires(s); 184 if (expire > time_uptime) 185 expire = expire - time_uptime; 186 nlattr_add_u32(nw, PF_ST_EXPIRE, expire); 187 nlattr_add_u8(nw, PF_ST_DIRECTION, s->direction); 188 nlattr_add_u8(nw, PF_ST_LOG, s->act.log); 189 nlattr_add_u8(nw, PF_ST_TIMEOUT, s->timeout); 190 nlattr_add_u16(nw, PF_ST_STATE_FLAGS, s->state_flags); 191 uint8_t sync_flags = 0; 192 if (s->sns[PF_SN_LIMIT] != NULL) 193 sync_flags |= PFSYNC_FLAG_SRCNODE; 194 if (s->sns[PF_SN_NAT] != NULL || s->sns[PF_SN_ROUTE]) 195 sync_flags |= PFSYNC_FLAG_NATSRCNODE; 196 nlattr_add_u8(nw, PF_ST_SYNC_FLAGS, sync_flags); 197 nlattr_add_u64(nw, PF_ST_ID, s->id); 198 nlattr_add_u32(nw, PF_ST_CREATORID, htonl(s->creatorid)); 199 200 nlattr_add_u32(nw, PF_ST_RULE, s->rule ? s->rule->nr : -1); 201 nlattr_add_u32(nw, PF_ST_ANCHOR, s->anchor ? s->anchor->nr : -1); 202 nlattr_add_u32(nw, PF_ST_NAT_RULE, s->nat_rule ? s->nat_rule->nr : -1); 203 204 nlattr_add_u64(nw, PF_ST_PACKETS0, s->packets[0]); 205 nlattr_add_u64(nw, PF_ST_PACKETS1, s->packets[1]); 206 nlattr_add_u64(nw, PF_ST_BYTES0, s->bytes[0]); 207 nlattr_add_u64(nw, PF_ST_BYTES1, s->bytes[1]); 208 nlattr_add_u32(nw, PF_ST_RTABLEID, s->act.rtableid); 209 nlattr_add_u8(nw, PF_ST_MIN_TTL, s->act.min_ttl); 210 nlattr_add_u16(nw, PF_ST_MAX_MSS, s->act.max_mss); 211 nlattr_add_u16(nw, PF_ST_DNPIPE, s->act.dnpipe); 212 nlattr_add_u16(nw, PF_ST_DNRPIPE, s->act.dnrpipe); 213 nlattr_add_u8(nw, PF_ST_RT, s->act.rt); 214 if (s->act.rt_kif != NULL) 215 nlattr_add_string(nw, PF_ST_RT_IFNAME, s->act.rt_kif->pfik_name); 216 uint8_t src_node_flags = 0; 217 if (s->sns[PF_SN_LIMIT] != NULL) { 218 src_node_flags |= PFSTATE_SRC_NODE_LIMIT; 219 if (s->sns[PF_SN_LIMIT]->rule == &V_pf_default_rule) 220 src_node_flags |= PFSTATE_SRC_NODE_LIMIT_GLOBAL; 221 } 222 if (s->sns[PF_SN_NAT] != NULL) 223 src_node_flags |= PFSTATE_SRC_NODE_NAT; 224 if (s->sns[PF_SN_ROUTE] != NULL) 225 src_node_flags |= PFSTATE_SRC_NODE_ROUTE; 226 nlattr_add_u8(nw, PF_ST_SRC_NODE_FLAGS, src_node_flags); 227 nlattr_add_u8(nw, PF_ST_RT_AF, s->act.rt_af); 228 229 if (!dump_state_peer(nw, PF_ST_PEER_SRC, &s->src)) 230 goto enomem; 231 if (!dump_state_peer(nw, PF_ST_PEER_DST, &s->dst)) 232 goto enomem; 233 234 if (nlmsg_end(nw)) 235 return (0); 236 237 enomem: 238 error = ENOMEM; 239 nlmsg_abort(nw); 240 return (error); 241 } 242 243 static int 244 handle_dumpstates(struct nlpcb *nlp, struct nl_parsed_state *attrs, 245 struct nlmsghdr *hdr, struct nl_pstate *npt) 246 { 247 int error = 0; 248 249 hdr->nlmsg_flags |= NLM_F_MULTI; 250 251 for (int i = 0; i <= V_pf_hashmask; i++) { 252 struct pf_idhash *ih = &V_pf_idhash[i]; 253 struct pf_kstate *s; 254 255 if (LIST_EMPTY(&ih->states)) 256 continue; 257 258 PF_HASHROW_LOCK(ih); 259 LIST_FOREACH(s, &ih->states, entry) { 260 sa_family_t af = s->key[PF_SK_WIRE]->af; 261 262 if (s->timeout == PFTM_UNLINKED) 263 continue; 264 265 /* Filter */ 266 if (attrs->creatorid != 0 && s->creatorid != attrs->creatorid) 267 continue; 268 if (attrs->ifname[0] != 0 && 269 strncmp(attrs->ifname, s->kif->pfik_name, IFNAMSIZ) != 0) 270 continue; 271 if (attrs->proto != 0 && s->key[PF_SK_WIRE]->proto != attrs->proto) 272 continue; 273 if (attrs->af != 0 && af != attrs->af) 274 continue; 275 if (pf_match_addr(1, &s->key[PF_SK_WIRE]->addr[0], 276 &attrs->mask, &attrs->addr, af) && 277 pf_match_addr(1, &s->key[PF_SK_WIRE]->addr[1], 278 &attrs->mask, &attrs->addr, af) && 279 pf_match_addr(1, &s->key[PF_SK_STACK]->addr[0], 280 &attrs->mask, &attrs->addr, af) && 281 pf_match_addr(1, &s->key[PF_SK_STACK]->addr[1], 282 &attrs->mask, &attrs->addr, af)) 283 continue; 284 285 error = dump_state(nlp, hdr, s, npt); 286 if (error != 0) 287 break; 288 } 289 PF_HASHROW_UNLOCK(ih); 290 } 291 292 if (!nlmsg_end_dump(npt->nw, error, hdr)) { 293 NL_LOG(LOG_DEBUG, "Unable to finalize the dump"); 294 return (ENOMEM); 295 } 296 297 return (error); 298 } 299 300 static int 301 handle_getstate(struct nlpcb *nlp, struct nl_parsed_state *attrs, 302 struct nlmsghdr *hdr, struct nl_pstate *npt) 303 { 304 struct pf_kstate *s; 305 int ret; 306 307 s = pf_find_state_byid(attrs->id, attrs->creatorid); 308 if (s == NULL) 309 return (ENOENT); 310 ret = dump_state(nlp, hdr, s, npt); 311 PF_STATE_UNLOCK(s); 312 313 return (ret); 314 } 315 316 static int 317 dump_creatorid(struct nlpcb *nlp, const struct nlmsghdr *hdr, uint32_t creator, 318 struct nl_pstate *npt) 319 { 320 struct nl_writer *nw = npt->nw; 321 322 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 323 goto enomem; 324 325 struct genlmsghdr *ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 326 ghdr_new->cmd = PFNL_CMD_GETCREATORS; 327 ghdr_new->version = 0; 328 ghdr_new->reserved = 0; 329 330 nlattr_add_u32(nw, PF_ST_CREATORID, htonl(creator)); 331 332 if (nlmsg_end(nw)) 333 return (0); 334 335 enomem: 336 nlmsg_abort(nw); 337 return (ENOMEM); 338 } 339 340 static int 341 pf_handle_getstates(struct nlmsghdr *hdr, struct nl_pstate *npt) 342 { 343 int error; 344 345 struct nl_parsed_state attrs = {}; 346 error = nl_parse_nlmsg(hdr, &state_parser, npt, &attrs); 347 if (error != 0) 348 return (error); 349 350 if (attrs.id != 0) 351 error = handle_getstate(npt->nlp, &attrs, hdr, npt); 352 else 353 error = handle_dumpstates(npt->nlp, &attrs, hdr, npt); 354 355 return (error); 356 } 357 358 static int 359 pf_handle_getcreators(struct nlmsghdr *hdr, struct nl_pstate *npt) 360 { 361 uint32_t creators[16]; 362 int error = 0; 363 364 bzero(creators, sizeof(creators)); 365 366 for (int i = 0; i < V_pf_hashmask; i++) { 367 struct pf_idhash *ih = &V_pf_idhash[i]; 368 struct pf_kstate *s; 369 370 if (LIST_EMPTY(&ih->states)) 371 continue; 372 373 PF_HASHROW_LOCK(ih); 374 LIST_FOREACH(s, &ih->states, entry) { 375 int j; 376 if (s->timeout == PFTM_UNLINKED) 377 continue; 378 379 for (j = 0; j < nitems(creators); j++) { 380 if (creators[j] == s->creatorid) 381 break; 382 if (creators[j] == 0) { 383 creators[j] = s->creatorid; 384 break; 385 } 386 } 387 if (j == nitems(creators)) 388 printf("Warning: too many creators!\n"); 389 } 390 PF_HASHROW_UNLOCK(ih); 391 } 392 393 hdr->nlmsg_flags |= NLM_F_MULTI; 394 for (int i = 0; i < nitems(creators); i++) { 395 if (creators[i] == 0) 396 break; 397 error = dump_creatorid(npt->nlp, hdr, creators[i], npt); 398 } 399 400 if (!nlmsg_end_dump(npt->nw, error, hdr)) { 401 NL_LOG(LOG_DEBUG, "Unable to finalize the dump"); 402 return (ENOMEM); 403 } 404 405 return (error); 406 } 407 408 static int 409 pf_handle_start(struct nlmsghdr *hdr __unused, struct nl_pstate *npt __unused) 410 { 411 return (pf_start()); 412 } 413 414 static int 415 pf_handle_stop(struct nlmsghdr *hdr __unused, struct nl_pstate *npt __unused) 416 { 417 return (pf_stop()); 418 } 419 420 #define _OUT(_field) offsetof(struct pf_addr_wrap, _field) 421 static const struct nlattr_parser nla_p_addr_wrap[] = { 422 { .type = PF_AT_ADDR, .off = _OUT(v.a.addr), .cb = nlattr_get_in6_addr }, 423 { .type = PF_AT_MASK, .off = _OUT(v.a.mask), .cb = nlattr_get_in6_addr }, 424 { .type = PF_AT_IFNAME, .off = _OUT(v.ifname), .arg = (void *)IFNAMSIZ,.cb = nlattr_get_chara }, 425 { .type = PF_AT_TABLENAME, .off = _OUT(v.tblname), .arg = (void *)PF_TABLE_NAME_SIZE, .cb = nlattr_get_chara }, 426 { .type = PF_AT_TYPE, .off = _OUT(type), .cb = nlattr_get_uint8 }, 427 { .type = PF_AT_IFLAGS, .off = _OUT(iflags), .cb = nlattr_get_uint8 }, 428 }; 429 NL_DECLARE_ATTR_PARSER(addr_wrap_parser, nla_p_addr_wrap); 430 #undef _OUT 431 432 static bool 433 nlattr_add_addr_wrap(struct nl_writer *nw, int attrtype, struct pf_addr_wrap *a) 434 { 435 int off = nlattr_add_nested(nw, attrtype); 436 437 nlattr_add_in6_addr(nw, PF_AT_ADDR, &a->v.a.addr.v6); 438 nlattr_add_in6_addr(nw, PF_AT_MASK, &a->v.a.mask.v6); 439 nlattr_add_u8(nw, PF_AT_TYPE, a->type); 440 nlattr_add_u8(nw, PF_AT_IFLAGS, a->iflags); 441 442 if (a->type == PF_ADDR_DYNIFTL) { 443 nlattr_add_string(nw, PF_AT_IFNAME, a->v.ifname); 444 nlattr_add_u32(nw, PF_AT_DYNCNT, a->p.dyncnt); 445 } else if (a->type == PF_ADDR_TABLE) { 446 nlattr_add_string(nw, PF_AT_TABLENAME, a->v.tblname); 447 nlattr_add_u32(nw, PF_AT_TBLCNT, a->p.tblcnt); 448 } 449 450 nlattr_set_len(nw, off); 451 452 return (true); 453 } 454 455 #define _OUT(_field) offsetof(struct pf_rule_addr, _field) 456 static const struct nlattr_parser nla_p_ruleaddr[] = { 457 { .type = PF_RAT_ADDR, .off = _OUT(addr), .arg = &addr_wrap_parser, .cb = nlattr_get_nested }, 458 { .type = PF_RAT_SRC_PORT, .off = _OUT(port[0]), .cb = nlattr_get_uint16 }, 459 { .type = PF_RAT_DST_PORT, .off = _OUT(port[1]), .cb = nlattr_get_uint16 }, 460 { .type = PF_RAT_NEG, .off = _OUT(neg), .cb = nlattr_get_uint8 }, 461 { .type = PF_RAT_OP, .off = _OUT(port_op), .cb = nlattr_get_uint8 }, 462 }; 463 NL_DECLARE_ATTR_PARSER(rule_addr_parser, nla_p_ruleaddr); 464 #undef _OUT 465 466 static bool 467 nlattr_add_rule_addr(struct nl_writer *nw, int attrtype, struct pf_rule_addr *r) 468 { 469 struct pf_addr_wrap aw = {0}; 470 int off = nlattr_add_nested(nw, attrtype); 471 472 bcopy(&(r->addr), &aw, sizeof(struct pf_addr_wrap)); 473 pf_addr_copyout(&aw); 474 475 nlattr_add_addr_wrap(nw, PF_RAT_ADDR, &aw); 476 nlattr_add_u16(nw, PF_RAT_SRC_PORT, r->port[0]); 477 nlattr_add_u16(nw, PF_RAT_DST_PORT, r->port[1]); 478 nlattr_add_u8(nw, PF_RAT_NEG, r->neg); 479 nlattr_add_u8(nw, PF_RAT_OP, r->port_op); 480 481 nlattr_set_len(nw, off); 482 483 return (true); 484 } 485 486 #define _OUT(_field) offsetof(struct pf_mape_portset, _field) 487 static const struct nlattr_parser nla_p_mape_portset[] = { 488 { .type = PF_MET_OFFSET, .off = _OUT(offset), .cb = nlattr_get_uint8 }, 489 { .type = PF_MET_PSID_LEN, .off = _OUT(psidlen), .cb = nlattr_get_uint8 }, 490 {. type = PF_MET_PSID, .off = _OUT(psid), .cb = nlattr_get_uint16 }, 491 }; 492 NL_DECLARE_ATTR_PARSER(mape_portset_parser, nla_p_mape_portset); 493 #undef _OUT 494 495 static bool 496 nlattr_add_mape_portset(struct nl_writer *nw, int attrtype, const struct pf_mape_portset *m) 497 { 498 int off = nlattr_add_nested(nw, attrtype); 499 500 nlattr_add_u8(nw, PF_MET_OFFSET, m->offset); 501 nlattr_add_u8(nw, PF_MET_PSID_LEN, m->psidlen); 502 nlattr_add_u16(nw, PF_MET_PSID, m->psid); 503 504 nlattr_set_len(nw, off); 505 506 return (true); 507 } 508 509 struct nl_parsed_labels 510 { 511 char labels[PF_RULE_MAX_LABEL_COUNT][PF_RULE_LABEL_SIZE]; 512 uint32_t i; 513 }; 514 515 static int 516 nlattr_get_pf_rule_labels(struct nlattr *nla, struct nl_pstate *npt, 517 const void *arg, void *target) 518 { 519 struct nl_parsed_labels *l = (struct nl_parsed_labels *)target; 520 int ret; 521 522 if (l->i >= PF_RULE_MAX_LABEL_COUNT) 523 return (E2BIG); 524 525 ret = nlattr_get_chara(nla, npt, (void *)PF_RULE_LABEL_SIZE, 526 l->labels[l->i]); 527 if (ret == 0) 528 l->i++; 529 530 return (ret); 531 } 532 533 #define _OUT(_field) offsetof(struct nl_parsed_labels, _field) 534 static const struct nlattr_parser nla_p_labels[] = { 535 { .type = PF_LT_LABEL, .off = 0, .cb = nlattr_get_pf_rule_labels }, 536 }; 537 NL_DECLARE_ATTR_PARSER(rule_labels_parser, nla_p_labels); 538 #undef _OUT 539 540 static int 541 nlattr_get_nested_pf_rule_labels(struct nlattr *nla, struct nl_pstate *npt, const void *arg, void *target) 542 { 543 struct nl_parsed_labels parsed_labels = { }; 544 int error; 545 546 /* Assumes target points to the beginning of the structure */ 547 error = nl_parse_header(NLA_DATA(nla), NLA_DATA_LEN(nla), &rule_labels_parser, npt, &parsed_labels); 548 if (error != 0) 549 return (error); 550 551 memcpy(target, parsed_labels.labels, sizeof(parsed_labels.labels)); 552 553 return (0); 554 } 555 556 static bool 557 nlattr_add_labels(struct nl_writer *nw, int attrtype, const struct pf_krule *r) 558 { 559 int off = nlattr_add_nested(nw, attrtype); 560 int i = 0; 561 562 while (r->label[i][0] != 0 563 && i < PF_RULE_MAX_LABEL_COUNT) { 564 nlattr_add_string(nw, PF_LT_LABEL, r->label[i]); 565 i++; 566 } 567 568 nlattr_set_len(nw, off); 569 570 return (true); 571 } 572 573 #define _OUT(_field) offsetof(struct pf_kpool, _field) 574 static const struct nlattr_parser nla_p_pool[] = { 575 { .type = PF_PT_KEY, .off = _OUT(key), .arg = (void *)sizeof(struct pf_poolhashkey), .cb = nlattr_get_bytes }, 576 { .type = PF_PT_COUNTER, .off = _OUT(counter), .cb = nlattr_get_in6_addr }, 577 { .type = PF_PT_TBLIDX, .off = _OUT(tblidx), .cb = nlattr_get_uint32 }, 578 { .type = PF_PT_PROXY_SRC_PORT, .off = _OUT(proxy_port[0]), .cb = nlattr_get_uint16 }, 579 { .type = PF_PT_PROXY_DST_PORT, .off = _OUT(proxy_port[1]), .cb = nlattr_get_uint16 }, 580 { .type = PF_PT_OPTS, .off = _OUT(opts), .cb = nlattr_get_uint8 }, 581 { .type = PF_PT_MAPE, .off = _OUT(mape), .arg = &mape_portset_parser, .cb = nlattr_get_nested }, 582 }; 583 NL_DECLARE_ATTR_PARSER(pool_parser, nla_p_pool); 584 #undef _OUT 585 586 static bool 587 nlattr_add_pool(struct nl_writer *nw, int attrtype, const struct pf_kpool *pool) 588 { 589 int off = nlattr_add_nested(nw, attrtype); 590 591 nlattr_add(nw, PF_PT_KEY, sizeof(struct pf_poolhashkey), &pool->key); 592 nlattr_add_in6_addr(nw, PF_PT_COUNTER, (const struct in6_addr *)&pool->counter); 593 nlattr_add_u32(nw, PF_PT_TBLIDX, pool->tblidx); 594 nlattr_add_u16(nw, PF_PT_PROXY_SRC_PORT, pool->proxy_port[0]); 595 nlattr_add_u16(nw, PF_PT_PROXY_DST_PORT, pool->proxy_port[1]); 596 nlattr_add_u8(nw, PF_PT_OPTS, pool->opts); 597 nlattr_add_mape_portset(nw, PF_PT_MAPE, &pool->mape); 598 599 nlattr_set_len(nw, off); 600 601 return (true); 602 } 603 604 #define _OUT(_field) offsetof(struct pf_rule_uid, _field) 605 static const struct nlattr_parser nla_p_rule_uid[] = { 606 { .type = PF_RUT_UID_LOW, .off = _OUT(uid[0]), .cb = nlattr_get_uint32 }, 607 { .type = PF_RUT_UID_HIGH, .off = _OUT(uid[1]), .cb = nlattr_get_uint32 }, 608 { .type = PF_RUT_OP, .off = _OUT(op), .cb = nlattr_get_uint8 }, 609 }; 610 NL_DECLARE_ATTR_PARSER(rule_uid_parser, nla_p_rule_uid); 611 #undef _OUT 612 613 static bool 614 nlattr_add_rule_uid(struct nl_writer *nw, int attrtype, const struct pf_rule_uid *u) 615 { 616 int off = nlattr_add_nested(nw, attrtype); 617 618 nlattr_add_u32(nw, PF_RUT_UID_LOW, u->uid[0]); 619 nlattr_add_u32(nw, PF_RUT_UID_HIGH, u->uid[1]); 620 nlattr_add_u8(nw, PF_RUT_OP, u->op); 621 622 nlattr_set_len(nw, off); 623 624 return (true); 625 } 626 627 struct nl_parsed_timeouts 628 { 629 uint32_t timeouts[PFTM_MAX]; 630 uint32_t i; 631 }; 632 633 static int 634 nlattr_get_pf_timeout(struct nlattr *nla, struct nl_pstate *npt, 635 const void *arg, void *target) 636 { 637 struct nl_parsed_timeouts *t = (struct nl_parsed_timeouts *)target; 638 int ret; 639 640 if (t->i >= PFTM_MAX) 641 return (E2BIG); 642 643 ret = nlattr_get_uint32(nla, npt, NULL, &t->timeouts[t->i]); 644 if (ret == 0) 645 t->i++; 646 647 return (ret); 648 } 649 650 #define _OUT(_field) offsetof(struct nl_parsed_timeout, _field) 651 static const struct nlattr_parser nla_p_timeouts[] = { 652 { .type = PF_TT_TIMEOUT, .off = 0, .cb = nlattr_get_pf_timeout }, 653 }; 654 NL_DECLARE_ATTR_PARSER(timeout_parser, nla_p_timeouts); 655 #undef _OUT 656 657 static int 658 nlattr_get_nested_timeouts(struct nlattr *nla, struct nl_pstate *npt, const void *arg, void *target) 659 { 660 struct nl_parsed_timeouts parsed_timeouts = { }; 661 int error; 662 663 /* Assumes target points to the beginning of the structure */ 664 error = nl_parse_header(NLA_DATA(nla), NLA_DATA_LEN(nla), &timeout_parser, npt, &parsed_timeouts); 665 if (error != 0) 666 return (error); 667 668 memcpy(target, parsed_timeouts.timeouts, sizeof(parsed_timeouts.timeouts)); 669 670 return (0); 671 } 672 673 static bool 674 nlattr_add_timeout(struct nl_writer *nw, int attrtype, uint32_t *timeout) 675 { 676 int off = nlattr_add_nested(nw, attrtype); 677 678 for (int i = 0; i < PFTM_MAX; i++) 679 nlattr_add_u32(nw, PF_RT_TIMEOUT, timeout[i]); 680 681 nlattr_set_len(nw, off); 682 683 return (true); 684 } 685 686 #define _OUT(_field) offsetof(struct pf_kthreshold, _field) 687 static const struct nlattr_parser nla_p_threshold[] = { 688 { .type = PF_TH_LIMIT, .off = _OUT(limit), .cb = nlattr_get_uint32 }, 689 { .type = PF_TH_SECONDS, .off = _OUT(seconds), .cb = nlattr_get_uint32 }, 690 }; 691 NL_DECLARE_ATTR_PARSER(threshold_parser, nla_p_threshold); 692 #undef _OUT 693 694 #define _OUT(_field) offsetof(struct pf_krule, _field) 695 static const struct nlattr_parser nla_p_rule[] = { 696 { .type = PF_RT_SRC, .off = _OUT(src), .arg = &rule_addr_parser,.cb = nlattr_get_nested }, 697 { .type = PF_RT_DST, .off = _OUT(dst), .arg = &rule_addr_parser,.cb = nlattr_get_nested }, 698 { .type = PF_RT_RIDENTIFIER, .off = _OUT(ridentifier), .cb = nlattr_get_uint32 }, 699 { .type = PF_RT_LABELS, .off = _OUT(label), .arg = &rule_labels_parser,.cb = nlattr_get_nested_pf_rule_labels }, 700 { .type = PF_RT_IFNAME, .off = _OUT(ifname), .arg = (void *)IFNAMSIZ, .cb = nlattr_get_chara }, 701 { .type = PF_RT_QNAME, .off = _OUT(qname), .arg = (void *)PF_QNAME_SIZE, .cb = nlattr_get_chara }, 702 { .type = PF_RT_PQNAME, .off = _OUT(pqname), .arg = (void *)PF_QNAME_SIZE, .cb = nlattr_get_chara }, 703 { .type = PF_RT_TAGNAME, .off = _OUT(tagname), .arg = (void *)PF_TAG_NAME_SIZE, .cb = nlattr_get_chara }, 704 { .type = PF_RT_MATCH_TAGNAME, .off = _OUT(match_tagname), .arg = (void *)PF_TAG_NAME_SIZE, .cb = nlattr_get_chara }, 705 { .type = PF_RT_OVERLOAD_TBLNAME, .off = _OUT(overload_tblname), .arg = (void *)PF_TABLE_NAME_SIZE, .cb = nlattr_get_chara }, 706 { .type = PF_RT_RPOOL_RDR, .off = _OUT(rdr), .arg = &pool_parser, .cb = nlattr_get_nested }, 707 { .type = PF_RT_OS_FINGERPRINT, .off = _OUT(os_fingerprint), .cb = nlattr_get_uint32 }, 708 { .type = PF_RT_RTABLEID, .off = _OUT(rtableid), .cb = nlattr_get_uint32 }, 709 { .type = PF_RT_TIMEOUT, .off = _OUT(timeout), .arg = &timeout_parser, .cb = nlattr_get_nested_timeouts }, 710 { .type = PF_RT_MAX_STATES, .off = _OUT(max_states), .cb = nlattr_get_uint32 }, 711 { .type = PF_RT_MAX_SRC_NODES, .off = _OUT(max_src_nodes), .cb = nlattr_get_uint32 }, 712 { .type = PF_RT_MAX_SRC_STATES, .off = _OUT(max_src_states), .cb = nlattr_get_uint32 }, 713 { .type = PF_RT_MAX_SRC_CONN_RATE_LIMIT, .off = _OUT(max_src_conn_rate.limit), .cb = nlattr_get_uint32 }, 714 { .type = PF_RT_MAX_SRC_CONN_RATE_SECS, .off = _OUT(max_src_conn_rate.seconds), .cb = nlattr_get_uint32 }, 715 { .type = PF_RT_DNPIPE, .off = _OUT(dnpipe), .cb = nlattr_get_uint16 }, 716 { .type = PF_RT_DNRPIPE, .off = _OUT(dnrpipe), .cb = nlattr_get_uint16 }, 717 { .type = PF_RT_DNFLAGS, .off = _OUT(free_flags), .cb = nlattr_get_uint32 }, 718 { .type = PF_RT_NR, .off = _OUT(nr), .cb = nlattr_get_uint32 }, 719 { .type = PF_RT_PROB, .off = _OUT(prob), .cb = nlattr_get_uint32 }, 720 { .type = PF_RT_CUID, .off = _OUT(cuid), .cb = nlattr_get_uint32 }, 721 {. type = PF_RT_CPID, .off = _OUT(cpid), .cb = nlattr_get_uint32 }, 722 { .type = PF_RT_RETURN_ICMP, .off = _OUT(return_icmp), .cb = nlattr_get_uint16 }, 723 { .type = PF_RT_RETURN_ICMP6, .off = _OUT(return_icmp6), .cb = nlattr_get_uint16 }, 724 { .type = PF_RT_MAX_MSS, .off = _OUT(max_mss), .cb = nlattr_get_uint16 }, 725 { .type = PF_RT_SCRUB_FLAGS, .off = _OUT(scrub_flags), .cb = nlattr_get_uint16 }, 726 { .type = PF_RT_UID, .off = _OUT(uid), .arg = &rule_uid_parser, .cb = nlattr_get_nested }, 727 { .type = PF_RT_GID, .off = _OUT(gid), .arg = &rule_uid_parser, .cb = nlattr_get_nested }, 728 { .type = PF_RT_RULE_FLAG, .off = _OUT(rule_flag), .cb = nlattr_get_uint32 }, 729 { .type = PF_RT_ACTION, .off = _OUT(action), .cb = nlattr_get_uint8 }, 730 { .type = PF_RT_DIRECTION, .off = _OUT(direction), .cb = nlattr_get_uint8 }, 731 { .type = PF_RT_LOG, .off = _OUT(log), .cb = nlattr_get_uint8 }, 732 { .type = PF_RT_LOGIF, .off = _OUT(logif), .cb = nlattr_get_uint8 }, 733 { .type = PF_RT_QUICK, .off = _OUT(quick), .cb = nlattr_get_uint8 }, 734 { .type = PF_RT_IF_NOT, .off = _OUT(ifnot), .cb = nlattr_get_uint8 }, 735 { .type = PF_RT_MATCH_TAG_NOT, .off = _OUT(match_tag_not), .cb = nlattr_get_uint8 }, 736 { .type = PF_RT_NATPASS, .off = _OUT(natpass), .cb = nlattr_get_uint8 }, 737 { .type = PF_RT_KEEP_STATE, .off = _OUT(keep_state), .cb = nlattr_get_uint8 }, 738 { .type = PF_RT_AF, .off = _OUT(af), .cb = nlattr_get_uint8 }, 739 { .type = PF_RT_PROTO, .off = _OUT(proto), .cb = nlattr_get_uint8 }, 740 { .type = PF_RT_TYPE, .off = _OUT(type), .cb = nlattr_get_uint8 }, 741 { .type = PF_RT_CODE, .off = _OUT(code), .cb = nlattr_get_uint8 }, 742 { .type = PF_RT_FLAGS, .off = _OUT(flags), .cb = nlattr_get_uint8 }, 743 { .type = PF_RT_FLAGSET, .off = _OUT(flagset), .cb = nlattr_get_uint8 }, 744 { .type = PF_RT_MIN_TTL, .off = _OUT(min_ttl), .cb = nlattr_get_uint8 }, 745 { .type = PF_RT_ALLOW_OPTS, .off = _OUT(allow_opts), .cb = nlattr_get_uint8 }, 746 { .type = PF_RT_RT, .off = _OUT(rt), .cb = nlattr_get_uint8 }, 747 { .type = PF_RT_RETURN_TTL, .off = _OUT(return_ttl), .cb = nlattr_get_uint8 }, 748 { .type = PF_RT_TOS, .off = _OUT(tos), .cb = nlattr_get_uint8 }, 749 { .type = PF_RT_SET_TOS, .off = _OUT(set_tos), .cb = nlattr_get_uint8 }, 750 { .type = PF_RT_ANCHOR_RELATIVE, .off = _OUT(anchor_relative), .cb = nlattr_get_uint8 }, 751 { .type = PF_RT_ANCHOR_WILDCARD, .off = _OUT(anchor_wildcard), .cb = nlattr_get_uint8 }, 752 { .type = PF_RT_FLUSH, .off = _OUT(flush), .cb = nlattr_get_uint8 }, 753 { .type = PF_RT_PRIO, .off = _OUT(prio), .cb = nlattr_get_uint8 }, 754 { .type = PF_RT_SET_PRIO, .off = _OUT(set_prio[0]), .cb = nlattr_get_uint8 }, 755 { .type = PF_RT_SET_PRIO_REPLY, .off = _OUT(set_prio[1]), .cb = nlattr_get_uint8 }, 756 { .type = PF_RT_DIVERT_ADDRESS, .off = _OUT(divert.addr), .cb = nlattr_get_in6_addr }, 757 { .type = PF_RT_DIVERT_PORT, .off = _OUT(divert.port), .cb = nlattr_get_uint16 }, 758 { .type = PF_RT_RCV_IFNAME, .off = _OUT(rcv_ifname), .arg = (void *)IFNAMSIZ, .cb = nlattr_get_chara }, 759 { .type = PF_RT_MAX_SRC_CONN, .off = _OUT(max_src_conn), .cb = nlattr_get_uint32 }, 760 { .type = PF_RT_RPOOL_NAT, .off = _OUT(nat), .arg = &pool_parser, .cb = nlattr_get_nested }, 761 { .type = PF_RT_NAF, .off = _OUT(naf), .cb = nlattr_get_uint8 }, 762 { .type = PF_RT_RPOOL_RT, .off = _OUT(route), .arg = &pool_parser, .cb = nlattr_get_nested }, 763 { .type = PF_RT_RCV_IFNOT, .off = _OUT(rcvifnot), .cb = nlattr_get_bool }, 764 { .type = PF_RT_PKTRATE, .off = _OUT(pktrate), .arg = &threshold_parser, .cb = nlattr_get_nested }, 765 { .type = PF_RT_MAX_PKT_SIZE, .off = _OUT(max_pkt_size), .cb = nlattr_get_uint16 }, 766 { .type = PF_RT_TYPE_2, .off = _OUT(type), .cb = nlattr_get_uint16 }, 767 { .type = PF_RT_CODE_2, .off = _OUT(code), .cb = nlattr_get_uint16 }, 768 }; 769 NL_DECLARE_ATTR_PARSER(rule_parser, nla_p_rule); 770 #undef _OUT 771 struct nl_parsed_addrule { 772 struct pf_krule *rule; 773 uint32_t ticket; 774 uint32_t pool_ticket; 775 char *anchor; 776 char *anchor_call; 777 }; 778 #define _OUT(_field) offsetof(struct nl_parsed_addrule, _field) 779 static const struct nlattr_parser nla_p_addrule[] = { 780 { .type = PF_ART_TICKET, .off = _OUT(ticket), .cb = nlattr_get_uint32 }, 781 { .type = PF_ART_POOL_TICKET, .off = _OUT(pool_ticket), .cb = nlattr_get_uint32 }, 782 { .type = PF_ART_ANCHOR, .off = _OUT(anchor), .cb = nlattr_get_string }, 783 { .type = PF_ART_ANCHOR_CALL, .off = _OUT(anchor_call), .cb = nlattr_get_string }, 784 { .type = PF_ART_RULE, .off = _OUT(rule), .arg = &rule_parser, .cb = nlattr_get_nested_ptr } 785 }; 786 #undef _OUT 787 NL_DECLARE_PARSER(addrule_parser, struct genlmsghdr, nlf_p_empty, nla_p_addrule); 788 789 static int 790 pf_handle_addrule(struct nlmsghdr *hdr, struct nl_pstate *npt) 791 { 792 int error; 793 struct nl_parsed_addrule attrs = {}; 794 795 attrs.rule = pf_krule_alloc(); 796 797 error = nl_parse_nlmsg(hdr, &addrule_parser, npt, &attrs); 798 if (error != 0) { 799 pf_free_rule(attrs.rule); 800 return (error); 801 } 802 803 error = pf_ioctl_addrule(attrs.rule, attrs.ticket, attrs.pool_ticket, 804 attrs.anchor, attrs.anchor_call, nlp_get_cred(npt->nlp)->cr_uid, 805 hdr->nlmsg_pid); 806 807 return (error); 808 } 809 810 #define _OUT(_field) offsetof(struct pfioc_rule, _field) 811 static const struct nlattr_parser nla_p_getrules[] = { 812 { .type = PF_GR_ANCHOR, .off = _OUT(anchor), .arg = (void *)MAXPATHLEN, .cb = nlattr_get_chara }, 813 { .type = PF_GR_ACTION, .off = _OUT(rule.action), .cb = nlattr_get_uint8 }, 814 }; 815 #undef _OUT 816 NL_DECLARE_PARSER(getrules_parser, struct genlmsghdr, nlf_p_empty, nla_p_getrules); 817 818 static int 819 pf_handle_getrules(struct nlmsghdr *hdr, struct nl_pstate *npt) 820 { 821 struct pfioc_rule attrs = {}; 822 int error; 823 struct nl_writer *nw = npt->nw; 824 struct genlmsghdr *ghdr_new; 825 826 error = nl_parse_nlmsg(hdr, &getrules_parser, npt, &attrs); 827 if (error != 0) 828 return (error); 829 830 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 831 return (ENOMEM); 832 833 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 834 ghdr_new->cmd = PFNL_CMD_GETRULES; 835 ghdr_new->version = 0; 836 ghdr_new->reserved = 0; 837 838 error = pf_ioctl_getrules(&attrs); 839 if (error != 0) 840 goto out; 841 842 nlattr_add_u32(nw, PF_GR_NR, attrs.nr); 843 nlattr_add_u32(nw, PF_GR_TICKET, attrs.ticket); 844 845 if (!nlmsg_end(nw)) { 846 error = ENOMEM; 847 goto out; 848 } 849 850 return (0); 851 852 out: 853 nlmsg_abort(nw); 854 return (error); 855 } 856 857 struct nl_parsed_get_rule { 858 char anchor[MAXPATHLEN]; 859 uint8_t action; 860 uint32_t nr; 861 uint32_t ticket; 862 uint8_t clear; 863 }; 864 #define _OUT(_field) offsetof(struct nl_parsed_get_rule, _field) 865 static const struct nlattr_parser nla_p_getrule[] = { 866 { .type = PF_GR_ANCHOR, .off = _OUT(anchor), .arg = (void *)MAXPATHLEN, .cb = nlattr_get_chara }, 867 { .type = PF_GR_ACTION, .off = _OUT(action), .cb = nlattr_get_uint8 }, 868 { .type = PF_GR_NR, .off = _OUT(nr), .cb = nlattr_get_uint32 }, 869 { .type = PF_GR_TICKET, .off = _OUT(ticket), .cb = nlattr_get_uint32 }, 870 { .type = PF_GR_CLEAR, .off = _OUT(clear), .cb = nlattr_get_uint8 }, 871 }; 872 #undef _OUT 873 NL_DECLARE_PARSER(getrule_parser, struct genlmsghdr, nlf_p_empty, nla_p_getrule); 874 875 static int 876 pf_handle_getrule(struct nlmsghdr *hdr, struct nl_pstate *npt) 877 { 878 char anchor_call[MAXPATHLEN]; 879 struct nl_parsed_get_rule attrs = {}; 880 struct nl_writer *nw = npt->nw; 881 struct genlmsghdr *ghdr_new; 882 struct pf_kruleset *ruleset; 883 struct pf_krule *rule; 884 u_int64_t src_nodes_total = 0; 885 int rs_num; 886 int error; 887 888 error = nl_parse_nlmsg(hdr, &getrule_parser, npt, &attrs); 889 if (error != 0) 890 return (error); 891 892 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 893 return (ENOMEM); 894 895 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 896 ghdr_new->cmd = PFNL_CMD_GETRULE; 897 ghdr_new->version = 0; 898 ghdr_new->reserved = 0; 899 900 PF_RULES_WLOCK(); 901 ruleset = pf_find_kruleset(attrs.anchor); 902 if (ruleset == NULL) { 903 PF_RULES_WUNLOCK(); 904 error = ENOENT; 905 goto out; 906 } 907 908 rs_num = pf_get_ruleset_number(attrs.action); 909 if (rs_num >= PF_RULESET_MAX) { 910 PF_RULES_WUNLOCK(); 911 error = EINVAL; 912 goto out; 913 } 914 915 if (attrs.ticket != ruleset->rules[rs_num].active.ticket) { 916 PF_RULES_WUNLOCK(); 917 error = EBUSY; 918 goto out; 919 } 920 921 rule = TAILQ_FIRST(ruleset->rules[rs_num].active.ptr); 922 while ((rule != NULL) && (rule->nr != attrs.nr)) 923 rule = TAILQ_NEXT(rule, entries); 924 if (rule == NULL) { 925 PF_RULES_WUNLOCK(); 926 error = EBUSY; 927 goto out; 928 } 929 930 nlattr_add_rule_addr(nw, PF_RT_SRC, &rule->src); 931 nlattr_add_rule_addr(nw, PF_RT_DST, &rule->dst); 932 nlattr_add_u32(nw, PF_RT_RIDENTIFIER, rule->ridentifier); 933 nlattr_add_labels(nw, PF_RT_LABELS, rule); 934 nlattr_add_string(nw, PF_RT_IFNAME, rule->ifname); 935 nlattr_add_string(nw, PF_RT_QNAME, rule->qname); 936 nlattr_add_string(nw, PF_RT_PQNAME, rule->pqname); 937 nlattr_add_string(nw, PF_RT_TAGNAME, rule->tagname); 938 nlattr_add_string(nw, PF_RT_MATCH_TAGNAME, rule->match_tagname); 939 nlattr_add_string(nw, PF_RT_OVERLOAD_TBLNAME, rule->overload_tblname); 940 nlattr_add_pool(nw, PF_RT_RPOOL_RDR, &rule->rdr); 941 nlattr_add_pool(nw, PF_RT_RPOOL_NAT, &rule->nat); 942 nlattr_add_pool(nw, PF_RT_RPOOL_RT, &rule->route); 943 nlattr_add_u32(nw, PF_RT_OS_FINGERPRINT, rule->os_fingerprint); 944 nlattr_add_u32(nw, PF_RT_RTABLEID, rule->rtableid); 945 nlattr_add_timeout(nw, PF_RT_TIMEOUT, rule->timeout); 946 nlattr_add_u32(nw, PF_RT_MAX_STATES, rule->max_states); 947 nlattr_add_u32(nw, PF_RT_MAX_SRC_NODES, rule->max_src_nodes); 948 nlattr_add_u32(nw, PF_RT_MAX_SRC_STATES, rule->max_src_states); 949 nlattr_add_u32(nw, PF_RT_MAX_SRC_CONN, rule->max_src_conn); 950 nlattr_add_u32(nw, PF_RT_MAX_SRC_CONN_RATE_LIMIT, rule->max_src_conn_rate.limit); 951 nlattr_add_u32(nw, PF_RT_MAX_SRC_CONN_RATE_SECS, rule->max_src_conn_rate.seconds); 952 nlattr_add_u16(nw, PF_RT_MAX_PKT_SIZE, rule->max_pkt_size); 953 954 nlattr_add_u16(nw, PF_RT_DNPIPE, rule->dnpipe); 955 nlattr_add_u16(nw, PF_RT_DNRPIPE, rule->dnrpipe); 956 nlattr_add_u32(nw, PF_RT_DNFLAGS, rule->free_flags); 957 958 nlattr_add_u32(nw, PF_RT_NR, rule->nr); 959 nlattr_add_u32(nw, PF_RT_PROB, rule->prob); 960 nlattr_add_u32(nw, PF_RT_CUID, rule->cuid); 961 nlattr_add_u32(nw, PF_RT_CPID, rule->cpid); 962 963 nlattr_add_u16(nw, PF_RT_RETURN_ICMP, rule->return_icmp); 964 nlattr_add_u16(nw, PF_RT_RETURN_ICMP6, rule->return_icmp6); 965 nlattr_add_u16(nw, PF_RT_RETURN_ICMP6, rule->return_icmp6); 966 nlattr_add_u16(nw, PF_RT_MAX_MSS, rule->max_mss); 967 nlattr_add_u16(nw, PF_RT_SCRUB_FLAGS, rule->scrub_flags); 968 969 nlattr_add_rule_uid(nw, PF_RT_UID, &rule->uid); 970 nlattr_add_rule_uid(nw, PF_RT_GID, (const struct pf_rule_uid *)&rule->gid); 971 972 nlattr_add_string(nw, PF_RT_RCV_IFNAME, rule->rcv_ifname); 973 nlattr_add_bool(nw, PF_RT_RCV_IFNOT, rule->rcvifnot); 974 975 nlattr_add_u32(nw, PF_RT_RULE_FLAG, rule->rule_flag); 976 nlattr_add_u8(nw, PF_RT_ACTION, rule->action); 977 nlattr_add_u8(nw, PF_RT_DIRECTION, rule->direction); 978 nlattr_add_u8(nw, PF_RT_LOG, rule->log); 979 nlattr_add_u8(nw, PF_RT_LOGIF, rule->logif); 980 nlattr_add_u8(nw, PF_RT_QUICK, rule->quick); 981 nlattr_add_u8(nw, PF_RT_IF_NOT, rule->ifnot); 982 nlattr_add_u8(nw, PF_RT_MATCH_TAG_NOT, rule->match_tag_not); 983 nlattr_add_u8(nw, PF_RT_NATPASS, rule->natpass); 984 nlattr_add_u8(nw, PF_RT_KEEP_STATE, rule->keep_state); 985 986 nlattr_add_u8(nw, PF_RT_AF, rule->af); 987 nlattr_add_u8(nw, PF_RT_NAF, rule->naf); 988 nlattr_add_u8(nw, PF_RT_PROTO, rule->proto); 989 990 nlattr_add_u8(nw, PF_RT_TYPE, rule->type); 991 nlattr_add_u8(nw, PF_RT_CODE, rule->code); 992 nlattr_add_u16(nw, PF_RT_TYPE_2, rule->type); 993 nlattr_add_u16(nw, PF_RT_CODE_2, rule->code); 994 995 nlattr_add_u8(nw, PF_RT_FLAGS, rule->flags); 996 nlattr_add_u8(nw, PF_RT_FLAGSET, rule->flagset); 997 nlattr_add_u8(nw, PF_RT_MIN_TTL, rule->min_ttl); 998 nlattr_add_u8(nw, PF_RT_ALLOW_OPTS, rule->allow_opts); 999 nlattr_add_u8(nw, PF_RT_RT, rule->rt); 1000 nlattr_add_u8(nw, PF_RT_RETURN_TTL, rule->return_ttl); 1001 nlattr_add_u8(nw, PF_RT_TOS, rule->tos); 1002 nlattr_add_u8(nw, PF_RT_SET_TOS, rule->set_tos); 1003 nlattr_add_u8(nw, PF_RT_ANCHOR_RELATIVE, rule->anchor_relative); 1004 nlattr_add_u8(nw, PF_RT_ANCHOR_WILDCARD, rule->anchor_wildcard); 1005 nlattr_add_u8(nw, PF_RT_FLUSH, rule->flush); 1006 nlattr_add_u8(nw, PF_RT_PRIO, rule->prio); 1007 nlattr_add_u8(nw, PF_RT_SET_PRIO, rule->set_prio[0]); 1008 nlattr_add_u8(nw, PF_RT_SET_PRIO_REPLY, rule->set_prio[1]); 1009 1010 nlattr_add_in6_addr(nw, PF_RT_DIVERT_ADDRESS, &rule->divert.addr.v6); 1011 nlattr_add_u16(nw, PF_RT_DIVERT_PORT, rule->divert.port); 1012 1013 nlattr_add_u64(nw, PF_RT_PACKETS_IN, pf_counter_u64_fetch(&rule->packets[0])); 1014 nlattr_add_u64(nw, PF_RT_PACKETS_OUT, pf_counter_u64_fetch(&rule->packets[1])); 1015 nlattr_add_u64(nw, PF_RT_BYTES_IN, pf_counter_u64_fetch(&rule->bytes[0])); 1016 nlattr_add_u64(nw, PF_RT_BYTES_OUT, pf_counter_u64_fetch(&rule->bytes[1])); 1017 nlattr_add_u64(nw, PF_RT_EVALUATIONS, pf_counter_u64_fetch(&rule->evaluations)); 1018 nlattr_add_u64(nw, PF_RT_TIMESTAMP, pf_get_timestamp(rule)); 1019 nlattr_add_u64(nw, PF_RT_STATES_CUR, counter_u64_fetch(rule->states_cur)); 1020 nlattr_add_u64(nw, PF_RT_STATES_TOTAL, counter_u64_fetch(rule->states_tot)); 1021 for (pf_sn_types_t sn_type=0; sn_type<PF_SN_MAX; sn_type++) 1022 src_nodes_total += counter_u64_fetch(rule->src_nodes[sn_type]); 1023 nlattr_add_u64(nw, PF_RT_SRC_NODES, src_nodes_total); 1024 nlattr_add_u64(nw, PF_RT_SRC_NODES_LIMIT, counter_u64_fetch(rule->src_nodes[PF_SN_LIMIT])); 1025 nlattr_add_u64(nw, PF_RT_SRC_NODES_NAT, counter_u64_fetch(rule->src_nodes[PF_SN_NAT])); 1026 nlattr_add_u64(nw, PF_RT_SRC_NODES_ROUTE, counter_u64_fetch(rule->src_nodes[PF_SN_ROUTE])); 1027 nlattr_add_pf_threshold(nw, PF_RT_PKTRATE, &rule->pktrate); 1028 1029 error = pf_kanchor_copyout(ruleset, rule, anchor_call, sizeof(anchor_call)); 1030 MPASS(error == 0); 1031 1032 nlattr_add_string(nw, PF_RT_ANCHOR_CALL, anchor_call); 1033 1034 if (attrs.clear) 1035 pf_krule_clear_counters(rule); 1036 1037 PF_RULES_WUNLOCK(); 1038 1039 if (!nlmsg_end(nw)) { 1040 error = ENOMEM; 1041 goto out; 1042 } 1043 1044 return (0); 1045 out: 1046 nlmsg_abort(nw); 1047 return (error); 1048 } 1049 1050 #define _OUT(_field) offsetof(struct pf_kstate_kill, _field) 1051 static const struct nlattr_parser nla_p_clear_states[] = { 1052 { .type = PF_CS_CMP_ID, .off = _OUT(psk_pfcmp.id), .cb = nlattr_get_uint64 }, 1053 { .type = PF_CS_CMP_CREATORID, .off = _OUT(psk_pfcmp.creatorid), .cb = nlattr_get_uint32 }, 1054 { .type = PF_CS_CMP_DIR, .off = _OUT(psk_pfcmp.direction), .cb = nlattr_get_uint8 }, 1055 { .type = PF_CS_AF, .off = _OUT(psk_af), .cb = nlattr_get_uint8 }, 1056 { .type = PF_CS_PROTO, .off = _OUT(psk_proto), .cb = nlattr_get_uint8 }, 1057 { .type = PF_CS_SRC, .off = _OUT(psk_src), .arg = &rule_addr_parser, .cb = nlattr_get_nested }, 1058 { .type = PF_CS_DST, .off = _OUT(psk_dst), .arg = &rule_addr_parser, .cb = nlattr_get_nested }, 1059 { .type = PF_CS_RT_ADDR, .off = _OUT(psk_rt_addr), .arg = &rule_addr_parser, .cb = nlattr_get_nested }, 1060 { .type = PF_CS_IFNAME, .off = _OUT(psk_ifname), .arg = (void *)IFNAMSIZ, .cb = nlattr_get_chara }, 1061 { .type = PF_CS_LABEL, .off = _OUT(psk_label), .arg = (void *)PF_RULE_LABEL_SIZE, .cb = nlattr_get_chara }, 1062 { .type = PF_CS_KILL_MATCH, .off = _OUT(psk_kill_match), .cb = nlattr_get_bool }, 1063 { .type = PF_CS_NAT, .off = _OUT(psk_nat), .cb = nlattr_get_bool }, 1064 }; 1065 #undef _OUT 1066 NL_DECLARE_PARSER(clear_states_parser, struct genlmsghdr, nlf_p_empty, nla_p_clear_states); 1067 1068 static int 1069 pf_handle_killclear_states(struct nlmsghdr *hdr, struct nl_pstate *npt, int cmd) 1070 { 1071 struct pf_kstate_kill kill = {}; 1072 struct epoch_tracker et; 1073 struct nl_writer *nw = npt->nw; 1074 struct genlmsghdr *ghdr_new; 1075 int error; 1076 unsigned int killed = 0; 1077 1078 error = nl_parse_nlmsg(hdr, &clear_states_parser, npt, &kill); 1079 if (error != 0) 1080 return (error); 1081 1082 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 1083 return (ENOMEM); 1084 1085 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 1086 ghdr_new->cmd = cmd; 1087 ghdr_new->version = 0; 1088 ghdr_new->reserved = 0; 1089 1090 NET_EPOCH_ENTER(et); 1091 if (cmd == PFNL_CMD_KILLSTATES) 1092 pf_killstates(&kill, &killed); 1093 else 1094 killed = pf_clear_states(&kill); 1095 NET_EPOCH_EXIT(et); 1096 1097 nlattr_add_u32(nw, PF_CS_KILLED, killed); 1098 1099 if (! nlmsg_end(nw)) { 1100 error = ENOMEM; 1101 goto out; 1102 } 1103 1104 return (0); 1105 1106 out: 1107 nlmsg_abort(nw); 1108 return (error); 1109 } 1110 1111 static int 1112 pf_handle_clear_states(struct nlmsghdr *hdr, struct nl_pstate *npt) 1113 { 1114 return (pf_handle_killclear_states(hdr, npt, PFNL_CMD_CLRSTATES)); 1115 } 1116 1117 static int 1118 pf_handle_kill_states(struct nlmsghdr *hdr, struct nl_pstate *npt) 1119 { 1120 return (pf_handle_killclear_states(hdr, npt, PFNL_CMD_KILLSTATES)); 1121 } 1122 1123 struct nl_parsed_set_statusif { 1124 char ifname[IFNAMSIZ]; 1125 }; 1126 #define _OUT(_field) offsetof(struct nl_parsed_set_statusif, _field) 1127 static const struct nlattr_parser nla_p_set_statusif[] = { 1128 { .type = PF_SS_IFNAME, .off = _OUT(ifname), .arg = (const void *)IFNAMSIZ, .cb = nlattr_get_chara }, 1129 }; 1130 #undef _OUT 1131 NL_DECLARE_PARSER(set_statusif_parser, struct genlmsghdr, nlf_p_empty, nla_p_set_statusif); 1132 1133 static int 1134 pf_handle_set_statusif(struct nlmsghdr *hdr, struct nl_pstate *npt) 1135 { 1136 int error; 1137 struct nl_parsed_set_statusif attrs = {}; 1138 1139 error = nl_parse_nlmsg(hdr, &set_statusif_parser, npt, &attrs); 1140 if (error != 0) 1141 return (error); 1142 1143 PF_RULES_WLOCK(); 1144 strlcpy(V_pf_status.ifname, attrs.ifname, IFNAMSIZ); 1145 PF_RULES_WUNLOCK(); 1146 1147 return (0); 1148 } 1149 1150 static bool 1151 nlattr_add_counters(struct nl_writer *nw, int attr, size_t number, char **names, 1152 counter_u64_t *counters) 1153 { 1154 for (int i = 0; i < number; i++) { 1155 int off = nlattr_add_nested(nw, attr); 1156 nlattr_add_u32(nw, PF_C_ID, i); 1157 nlattr_add_string(nw, PF_C_NAME, names[i]); 1158 nlattr_add_u64(nw, PF_C_COUNTER, counter_u64_fetch(counters[i])); 1159 nlattr_set_len(nw, off); 1160 } 1161 1162 return (true); 1163 } 1164 1165 static bool 1166 nlattr_add_fcounters(struct nl_writer *nw, int attr, size_t number, char **names, 1167 struct pf_counter_u64 *counters) 1168 { 1169 for (int i = 0; i < number; i++) { 1170 int off = nlattr_add_nested(nw, attr); 1171 nlattr_add_u32(nw, PF_C_ID, i); 1172 nlattr_add_string(nw, PF_C_NAME, names[i]); 1173 nlattr_add_u64(nw, PF_C_COUNTER, pf_counter_u64_fetch(&counters[i])); 1174 nlattr_set_len(nw, off); 1175 } 1176 1177 return (true); 1178 } 1179 1180 static bool 1181 nlattr_add_u64_array(struct nl_writer *nw, int attr, size_t number, uint64_t *array) 1182 { 1183 int off = nlattr_add_nested(nw, attr); 1184 1185 for (size_t i = 0; i < number; i++) 1186 nlattr_add_u64(nw, 0, array[i]); 1187 1188 nlattr_set_len(nw, off); 1189 1190 return (true); 1191 } 1192 1193 static int 1194 pf_handle_get_status(struct nlmsghdr *hdr, struct nl_pstate *npt) 1195 { 1196 struct pf_status s; 1197 struct nl_writer *nw = npt->nw; 1198 struct genlmsghdr *ghdr_new; 1199 char *pf_reasons[PFRES_MAX+1] = PFRES_NAMES; 1200 char *pf_lcounter[KLCNT_MAX+1] = KLCNT_NAMES; 1201 char *pf_fcounter[FCNT_MAX+1] = FCNT_NAMES; 1202 time_t since; 1203 int error; 1204 1205 PF_RULES_RLOCK_TRACKER; 1206 1207 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 1208 return (ENOMEM); 1209 1210 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 1211 ghdr_new->cmd = PFNL_CMD_GET_STATUS; 1212 ghdr_new->version = 0; 1213 ghdr_new->reserved = 0; 1214 1215 since = time_second - (time_uptime - V_pf_status.since); 1216 1217 PF_RULES_RLOCK(); 1218 1219 nlattr_add_string(nw, PF_GS_IFNAME, V_pf_status.ifname); 1220 nlattr_add_bool(nw, PF_GS_RUNNING, V_pf_status.running); 1221 nlattr_add_u32(nw, PF_GS_SINCE, since); 1222 nlattr_add_u32(nw, PF_GS_DEBUG, V_pf_status.debug); 1223 nlattr_add_u32(nw, PF_GS_HOSTID, ntohl(V_pf_status.hostid)); 1224 nlattr_add_u32(nw, PF_GS_STATES, V_pf_status.states); 1225 nlattr_add_u32(nw, PF_GS_SRC_NODES, V_pf_status.src_nodes); 1226 nlattr_add_u32(nw, PF_GS_REASSEMBLE, V_pf_status.reass); 1227 nlattr_add_u32(nw, PF_GS_SYNCOOKIES_ACTIVE, V_pf_status.syncookies_active); 1228 1229 nlattr_add_counters(nw, PF_GS_COUNTERS, PFRES_MAX, pf_reasons, 1230 V_pf_status.counters); 1231 nlattr_add_counters(nw, PF_GS_LCOUNTERS, KLCNT_MAX, pf_lcounter, 1232 V_pf_status.lcounters); 1233 nlattr_add_fcounters(nw, PF_GS_FCOUNTERS, FCNT_MAX, pf_fcounter, 1234 V_pf_status.fcounters); 1235 nlattr_add_counters(nw, PF_GS_SCOUNTERS, SCNT_MAX, pf_fcounter, 1236 V_pf_status.scounters); 1237 nlattr_add_counters(nw, PF_GS_NCOUNTERS, NCNT_MAX, pf_fcounter, 1238 V_pf_status.ncounters); 1239 nlattr_add_u64(nw, PF_GS_FRAGMENTS, pf_normalize_get_frag_count()); 1240 1241 pfi_update_status(V_pf_status.ifname, &s); 1242 nlattr_add_u64_array(nw, PF_GS_BCOUNTERS, 2 * 2, (uint64_t *)s.bcounters); 1243 nlattr_add_u64_array(nw, PF_GS_PCOUNTERS, 2 * 2 * 2, (uint64_t *)s.pcounters); 1244 1245 nlattr_add(nw, PF_GS_CHKSUM, PF_MD5_DIGEST_LENGTH, V_pf_status.pf_chksum); 1246 1247 PF_RULES_RUNLOCK(); 1248 1249 if (!nlmsg_end(nw)) { 1250 error = ENOMEM; 1251 goto out; 1252 } 1253 1254 return (0); 1255 1256 out: 1257 nlmsg_abort(nw); 1258 return (error); 1259 } 1260 1261 static int 1262 pf_handle_clear_status(struct nlmsghdr *hdr, struct nl_pstate *npt) 1263 { 1264 pf_ioctl_clear_status(); 1265 1266 return (0); 1267 } 1268 1269 #define _OUT(_field) offsetof(struct pfioc_natlook, _field) 1270 static const struct nlattr_parser nla_p_natlook[] = { 1271 { .type = PF_NL_AF, .off = _OUT(af), .cb = nlattr_get_uint8 }, 1272 { .type = PF_NL_DIRECTION, .off = _OUT(direction), .cb = nlattr_get_uint8 }, 1273 { .type = PF_NL_PROTO, .off = _OUT(proto), .cb = nlattr_get_uint8 }, 1274 { .type = PF_NL_SRC_ADDR, .off = _OUT(saddr), .cb = nlattr_get_in6_addr }, 1275 { .type = PF_NL_DST_ADDR, .off = _OUT(daddr), .cb = nlattr_get_in6_addr }, 1276 { .type = PF_NL_SRC_PORT, .off = _OUT(sport), .cb = nlattr_get_uint16 }, 1277 { .type = PF_NL_DST_PORT, .off = _OUT(dport), .cb = nlattr_get_uint16 }, 1278 }; 1279 #undef _OUT 1280 NL_DECLARE_PARSER(natlook_parser, struct genlmsghdr, nlf_p_empty, nla_p_natlook); 1281 1282 static int 1283 pf_handle_natlook(struct nlmsghdr *hdr, struct nl_pstate *npt) 1284 { 1285 struct pfioc_natlook attrs = {}; 1286 struct nl_writer *nw = npt->nw; 1287 struct genlmsghdr *ghdr_new; 1288 int error; 1289 1290 error = nl_parse_nlmsg(hdr, &natlook_parser, npt, &attrs); 1291 if (error != 0) 1292 return (error); 1293 1294 error = pf_ioctl_natlook(&attrs); 1295 if (error != 0) 1296 return (error); 1297 1298 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 1299 return (ENOMEM); 1300 1301 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 1302 ghdr_new->cmd = PFNL_CMD_NATLOOK; 1303 ghdr_new->version = 0; 1304 ghdr_new->reserved = 0; 1305 1306 nlattr_add_in6_addr(nw, PF_NL_SRC_ADDR, &attrs.rsaddr.v6); 1307 nlattr_add_in6_addr(nw, PF_NL_DST_ADDR, &attrs.rdaddr.v6); 1308 nlattr_add_u16(nw, PF_NL_SRC_PORT, attrs.rsport); 1309 nlattr_add_u16(nw, PF_NL_DST_PORT, attrs.rdport); 1310 1311 if (!nlmsg_end(nw)) { 1312 nlmsg_abort(nw); 1313 return (ENOMEM); 1314 } 1315 1316 return (0); 1317 } 1318 1319 struct pf_nl_set_debug 1320 { 1321 uint32_t level; 1322 }; 1323 #define _OUT(_field) offsetof(struct pf_nl_set_debug, _field) 1324 static const struct nlattr_parser nla_p_set_debug[] = { 1325 { .type = PF_SD_LEVEL, .off = _OUT(level), .cb = nlattr_get_uint32 }, 1326 }; 1327 #undef _OUT 1328 NL_DECLARE_PARSER(set_debug_parser, struct genlmsghdr, nlf_p_empty, nla_p_set_debug); 1329 1330 static int 1331 pf_handle_set_debug(struct nlmsghdr *hdr, struct nl_pstate *npt) 1332 { 1333 struct pf_nl_set_debug attrs = {}; 1334 int error; 1335 1336 error = nl_parse_nlmsg(hdr, &set_debug_parser, npt, &attrs); 1337 if (error != 0) 1338 return (error); 1339 1340 PF_RULES_WLOCK(); 1341 V_pf_status.debug = attrs.level; 1342 PF_RULES_WUNLOCK(); 1343 1344 return (0); 1345 } 1346 1347 struct pf_nl_set_timeout 1348 { 1349 uint32_t timeout; 1350 uint32_t seconds; 1351 }; 1352 #define _OUT(_field) offsetof(struct pf_nl_set_timeout, _field) 1353 static const struct nlattr_parser nla_p_set_timeout[] = { 1354 { .type = PF_TO_TIMEOUT, .off = _OUT(timeout), .cb = nlattr_get_uint32 }, 1355 { .type = PF_TO_SECONDS, .off = _OUT(seconds), .cb = nlattr_get_uint32 }, 1356 }; 1357 #undef _OUT 1358 NL_DECLARE_PARSER(set_timeout_parser, struct genlmsghdr, nlf_p_empty, nla_p_set_timeout); 1359 1360 static int 1361 pf_handle_set_timeout(struct nlmsghdr *hdr, struct nl_pstate *npt) 1362 { 1363 struct pf_nl_set_timeout attrs = {}; 1364 int error; 1365 1366 error = nl_parse_nlmsg(hdr, &set_timeout_parser, npt, &attrs); 1367 if (error != 0) 1368 return (error); 1369 1370 return (pf_ioctl_set_timeout(attrs.timeout, attrs.seconds, NULL)); 1371 } 1372 1373 static int 1374 pf_handle_get_timeout(struct nlmsghdr *hdr, struct nl_pstate *npt) 1375 { 1376 struct pf_nl_set_timeout attrs = {}; 1377 struct nl_writer *nw = npt->nw; 1378 struct genlmsghdr *ghdr_new; 1379 int error; 1380 1381 error = nl_parse_nlmsg(hdr, &set_timeout_parser, npt, &attrs); 1382 if (error != 0) 1383 return (error); 1384 1385 error = pf_ioctl_get_timeout(attrs.timeout, &attrs.seconds); 1386 if (error != 0) 1387 return (error); 1388 1389 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 1390 return (ENOMEM); 1391 1392 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 1393 ghdr_new->cmd = PFNL_CMD_GET_TIMEOUT; 1394 ghdr_new->version = 0; 1395 ghdr_new->reserved = 0; 1396 1397 nlattr_add_u32(nw, PF_TO_SECONDS, attrs.seconds); 1398 1399 if (!nlmsg_end(nw)) { 1400 nlmsg_abort(nw); 1401 return (ENOMEM); 1402 } 1403 1404 return (0); 1405 } 1406 1407 struct pf_nl_set_limit 1408 { 1409 uint32_t index; 1410 uint32_t limit; 1411 }; 1412 #define _OUT(_field) offsetof(struct pf_nl_set_limit, _field) 1413 static const struct nlattr_parser nla_p_set_limit[] = { 1414 { .type = PF_LI_INDEX, .off = _OUT(index), .cb = nlattr_get_uint32 }, 1415 { .type = PF_LI_LIMIT, .off = _OUT(limit), .cb = nlattr_get_uint32 }, 1416 }; 1417 #undef _OUT 1418 NL_DECLARE_PARSER(set_limit_parser, struct genlmsghdr, nlf_p_empty, nla_p_set_limit); 1419 1420 static int 1421 pf_handle_set_limit(struct nlmsghdr *hdr, struct nl_pstate *npt) 1422 { 1423 struct pf_nl_set_limit attrs = {}; 1424 int error; 1425 1426 error = nl_parse_nlmsg(hdr, &set_limit_parser, npt, &attrs); 1427 if (error != 0) 1428 return (error); 1429 1430 return (pf_ioctl_set_limit(attrs.index, attrs.limit, NULL)); 1431 } 1432 1433 static int 1434 pf_handle_get_limit(struct nlmsghdr *hdr, struct nl_pstate *npt) 1435 { 1436 struct pf_nl_set_limit attrs = {}; 1437 struct nl_writer *nw = npt->nw; 1438 struct genlmsghdr *ghdr_new; 1439 int error; 1440 1441 error = nl_parse_nlmsg(hdr, &set_limit_parser, npt, &attrs); 1442 if (error != 0) 1443 return (error); 1444 1445 error = pf_ioctl_get_limit(attrs.index, &attrs.limit); 1446 if (error != 0) 1447 return (error); 1448 1449 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 1450 return (ENOMEM); 1451 1452 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 1453 ghdr_new->cmd = PFNL_CMD_GET_LIMIT; 1454 ghdr_new->version = 0; 1455 ghdr_new->reserved = 0; 1456 1457 nlattr_add_u32(nw, PF_LI_LIMIT, attrs.limit); 1458 1459 if (!nlmsg_end(nw)) { 1460 nlmsg_abort(nw); 1461 return (ENOMEM); 1462 } 1463 1464 return (0); 1465 } 1466 1467 static int 1468 pf_handle_begin_addrs(struct nlmsghdr *hdr, struct nl_pstate *npt) 1469 { 1470 struct nl_writer *nw = npt->nw; 1471 struct genlmsghdr *ghdr_new; 1472 uint32_t ticket; 1473 int error; 1474 1475 error = pf_ioctl_begin_addrs(&ticket); 1476 if (error != 0) 1477 return (error); 1478 1479 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 1480 return (ENOMEM); 1481 1482 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 1483 ghdr_new->cmd = PFNL_CMD_BEGIN_ADDRS; 1484 ghdr_new->version = 0; 1485 ghdr_new->reserved = 0; 1486 1487 nlattr_add_u32(nw, PF_BA_TICKET, ticket); 1488 1489 if (!nlmsg_end(nw)) { 1490 nlmsg_abort(nw); 1491 return (ENOMEM); 1492 } 1493 1494 return (0); 1495 } 1496 1497 static bool 1498 nlattr_add_pool_addr(struct nl_writer *nw, int attrtype, struct pf_pooladdr *a) 1499 { 1500 int off; 1501 1502 off = nlattr_add_nested(nw, attrtype); 1503 1504 nlattr_add_addr_wrap(nw, PF_PA_ADDR, &a->addr); 1505 nlattr_add_string(nw, PF_PA_IFNAME, a->ifname); 1506 1507 nlattr_set_len(nw, off); 1508 1509 return (true); 1510 } 1511 1512 #define _OUT(_field) offsetof(struct pf_pooladdr, _field) 1513 static const struct nlattr_parser nla_p_pool_addr[] = { 1514 { .type = PF_PA_ADDR, .off = _OUT(addr), .arg = &addr_wrap_parser, .cb = nlattr_get_nested }, 1515 { .type = PF_PA_IFNAME, .off = _OUT(ifname), .arg = (void *)IFNAMSIZ, .cb = nlattr_get_chara }, 1516 }; 1517 NL_DECLARE_ATTR_PARSER(pool_addr_parser, nla_p_pool_addr); 1518 #undef _OUT 1519 1520 #define _OUT(_field) offsetof(struct pf_nl_pooladdr, _field) 1521 static const struct nlattr_parser nla_p_add_addr[] = { 1522 { .type = PF_AA_ACTION, .off = _OUT(action), .cb = nlattr_get_uint32 }, 1523 { .type = PF_AA_TICKET, .off = _OUT(ticket), .cb = nlattr_get_uint32 }, 1524 { .type = PF_AA_NR, .off = _OUT(nr), .cb = nlattr_get_uint32 }, 1525 { .type = PF_AA_R_NUM, .off = _OUT(r_num), .cb = nlattr_get_uint32 }, 1526 { .type = PF_AA_R_ACTION, .off = _OUT(r_action), .cb = nlattr_get_uint8 }, 1527 { .type = PF_AA_R_LAST, .off = _OUT(r_last), .cb = nlattr_get_uint8 }, 1528 { .type = PF_AA_AF, .off = _OUT(af), .cb = nlattr_get_uint8 }, 1529 { .type = PF_AA_ANCHOR, .off = _OUT(anchor), .arg = (void *)MAXPATHLEN, .cb = nlattr_get_chara }, 1530 { .type = PF_AA_ADDR, .off = _OUT(addr), .arg = &pool_addr_parser, .cb = nlattr_get_nested }, 1531 { .type = PF_AA_WHICH, .off = _OUT(which), .cb = nlattr_get_uint32 }, 1532 }; 1533 #undef _OUT 1534 NL_DECLARE_PARSER(add_addr_parser, struct genlmsghdr, nlf_p_empty, nla_p_add_addr); 1535 1536 static int 1537 pf_handle_add_addr(struct nlmsghdr *hdr, struct nl_pstate *npt) 1538 { 1539 struct pf_nl_pooladdr attrs = { 0 }; 1540 int error; 1541 1542 error = nl_parse_nlmsg(hdr, &add_addr_parser, npt, &attrs); 1543 if (error != 0) 1544 return (error); 1545 1546 if (attrs.which == 0) 1547 attrs.which = PF_RDR; 1548 1549 error = pf_ioctl_add_addr(&attrs); 1550 1551 return (error); 1552 } 1553 1554 static int 1555 pf_handle_get_addrs(struct nlmsghdr *hdr, struct nl_pstate *npt) 1556 { 1557 struct pf_nl_pooladdr attrs = { 0 }; 1558 struct nl_writer *nw = npt->nw; 1559 struct genlmsghdr *ghdr_new; 1560 int error; 1561 1562 error = nl_parse_nlmsg(hdr, &add_addr_parser, npt, &attrs); 1563 if (error != 0) 1564 return (error); 1565 1566 if (attrs.which == 0) 1567 attrs.which = PF_RDR; 1568 1569 error = pf_ioctl_get_addrs(&attrs); 1570 if (error != 0) 1571 return (error); 1572 1573 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 1574 return (ENOMEM); 1575 1576 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 1577 ghdr_new->cmd = PFNL_CMD_GET_ADDRS; 1578 ghdr_new->version = 0; 1579 ghdr_new->reserved = 0; 1580 1581 nlattr_add_u32(nw, PF_AA_NR, attrs.nr); 1582 1583 if (!nlmsg_end(nw)) { 1584 nlmsg_abort(nw); 1585 return (ENOMEM); 1586 } 1587 1588 return (error); 1589 } 1590 1591 static int 1592 pf_handle_get_addr(struct nlmsghdr *hdr, struct nl_pstate *npt) 1593 { 1594 struct pf_nl_pooladdr attrs = { 0 }; 1595 struct nl_writer *nw = npt->nw; 1596 struct genlmsghdr *ghdr_new; 1597 int error; 1598 1599 error = nl_parse_nlmsg(hdr, &add_addr_parser, npt, &attrs); 1600 if (error != 0) 1601 return (error); 1602 1603 if (attrs.which == 0) 1604 attrs.which = PF_RDR; 1605 1606 error = pf_ioctl_get_addr(&attrs); 1607 if (error != 0) 1608 return (error); 1609 1610 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 1611 return (ENOMEM); 1612 1613 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 1614 ghdr_new->cmd = PFNL_CMD_GET_ADDR; 1615 ghdr_new->version = 0; 1616 ghdr_new->reserved = 0; 1617 1618 nlattr_add_u32(nw, PF_AA_ACTION, attrs.action); 1619 nlattr_add_u32(nw, PF_AA_TICKET, attrs.ticket); 1620 nlattr_add_u32(nw, PF_AA_NR, attrs.nr); 1621 nlattr_add_u32(nw, PF_AA_R_NUM, attrs.r_num); 1622 nlattr_add_u8(nw, PF_AA_R_ACTION, attrs.r_action); 1623 nlattr_add_u8(nw, PF_AA_R_LAST, attrs.r_last); 1624 nlattr_add_u8(nw, PF_AA_AF, attrs.af); 1625 nlattr_add_string(nw, PF_AA_ANCHOR, attrs.anchor); 1626 nlattr_add_pool_addr(nw, PF_AA_ADDR, &attrs.addr); 1627 1628 if (!nlmsg_end(nw)) { 1629 nlmsg_abort(nw); 1630 return (ENOMEM); 1631 } 1632 1633 return (0); 1634 } 1635 1636 #define _OUT(_field) offsetof(struct pfioc_ruleset, _field) 1637 static const struct nlattr_parser nla_p_ruleset[] = { 1638 { .type = PF_RS_PATH, .off = _OUT(path), .arg = (void *)MAXPATHLEN, .cb = nlattr_get_chara }, 1639 { .type = PF_RS_NR, .off = _OUT(nr), .cb = nlattr_get_uint32 }, 1640 }; 1641 NL_DECLARE_PARSER(ruleset_parser, struct genlmsghdr, nlf_p_empty, nla_p_ruleset); 1642 #undef _OUT 1643 1644 static int 1645 pf_handle_get_rulesets(struct nlmsghdr *hdr, struct nl_pstate *npt) 1646 { 1647 struct pfioc_ruleset attrs = { 0 }; 1648 struct nl_writer *nw = npt->nw; 1649 struct genlmsghdr *ghdr_new; 1650 int error; 1651 1652 error = nl_parse_nlmsg(hdr, &ruleset_parser, npt, &attrs); 1653 if (error != 0) 1654 return (error); 1655 1656 error = pf_ioctl_get_rulesets(&attrs); 1657 if (error != 0) 1658 return (error); 1659 1660 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 1661 return (ENOMEM); 1662 1663 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 1664 ghdr_new->cmd = PFNL_CMD_GET_RULESETS; 1665 ghdr_new->version = 0; 1666 ghdr_new->reserved = 0; 1667 1668 nlattr_add_u32(nw, PF_RS_NR, attrs.nr); 1669 1670 if (!nlmsg_end(nw)) { 1671 nlmsg_abort(nw); 1672 return (ENOMEM); 1673 } 1674 1675 return (0); 1676 } 1677 1678 static int 1679 pf_handle_get_ruleset(struct nlmsghdr *hdr, struct nl_pstate *npt) 1680 { 1681 struct pfioc_ruleset attrs = { 0 }; 1682 struct nl_writer *nw = npt->nw; 1683 struct genlmsghdr *ghdr_new; 1684 int error; 1685 1686 error = nl_parse_nlmsg(hdr, &ruleset_parser, npt, &attrs); 1687 if (error) 1688 return (error); 1689 1690 error = pf_ioctl_get_ruleset(&attrs); 1691 if (error != 0) 1692 return (error); 1693 1694 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 1695 return (ENOMEM); 1696 1697 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 1698 ghdr_new->cmd = PFNL_CMD_GET_RULESET; 1699 ghdr_new->version = 0; 1700 ghdr_new->reserved = 0; 1701 1702 nlattr_add_string(nw, PF_RS_NAME, attrs.name); 1703 1704 if (!nlmsg_end(nw)) { 1705 nlmsg_abort(nw); 1706 return (ENOMEM); 1707 } 1708 1709 return (0); 1710 } 1711 1712 static bool 1713 nlattr_add_pf_threshold(struct nl_writer *nw, int attrtype, 1714 struct pf_kthreshold *t) 1715 { 1716 int off = nlattr_add_nested(nw, attrtype); 1717 int conn_rate_count = 0; 1718 1719 /* Adjust the connection rate estimate. */ 1720 if (t->cr != NULL) 1721 conn_rate_count = counter_rate_get(t->cr); 1722 1723 nlattr_add_u32(nw, PF_TH_LIMIT, t->limit); 1724 nlattr_add_u32(nw, PF_TH_SECONDS, t->seconds); 1725 nlattr_add_u32(nw, PF_TH_COUNT, conn_rate_count); 1726 1727 nlattr_set_len(nw, off); 1728 1729 return (true); 1730 } 1731 1732 static int 1733 pf_handle_get_srcnodes(struct nlmsghdr *hdr, struct nl_pstate *npt) 1734 { 1735 struct nl_writer *nw = npt->nw; 1736 struct genlmsghdr *ghdr_new; 1737 struct pf_ksrc_node *n; 1738 struct pf_srchash *sh; 1739 int i; 1740 int secs; 1741 1742 hdr->nlmsg_flags |= NLM_F_MULTI; 1743 1744 for (i = 0, sh = V_pf_srchash; i <= V_pf_srchashmask; 1745 i++, sh++) { 1746 /* Avoid locking empty rows. */ 1747 if (LIST_EMPTY(&sh->nodes)) 1748 continue; 1749 1750 PF_HASHROW_LOCK(sh); 1751 secs = time_uptime; 1752 1753 LIST_FOREACH(n, &sh->nodes, entry) { 1754 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) { 1755 nlmsg_abort(nw); 1756 return (ENOMEM); 1757 } 1758 1759 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 1760 ghdr_new->cmd = PFNL_CMD_GET_SRCNODES; 1761 ghdr_new->version = 0; 1762 ghdr_new->reserved = 0; 1763 1764 nlattr_add_in6_addr(nw, PF_SN_ADDR, &n->addr.v6); 1765 nlattr_add_in6_addr(nw, PF_SN_RADDR, &n->raddr.v6); 1766 nlattr_add_u32(nw, PF_SN_RULE_NR, n->rule->nr); 1767 nlattr_add_u64(nw, PF_SN_BYTES_IN, counter_u64_fetch(n->bytes[0])); 1768 nlattr_add_u64(nw, PF_SN_BYTES_OUT, counter_u64_fetch(n->bytes[1])); 1769 nlattr_add_u64(nw, PF_SN_PACKETS_IN, counter_u64_fetch(n->packets[0])); 1770 nlattr_add_u64(nw, PF_SN_PACKETS_OUT, counter_u64_fetch(n->packets[1])); 1771 nlattr_add_u32(nw, PF_SN_STATES, n->states); 1772 nlattr_add_u32(nw, PF_SN_CONNECTIONS, n->conn); 1773 nlattr_add_u8(nw, PF_SN_AF, n->af); 1774 nlattr_add_u8(nw, PF_SN_RAF, n->raf); 1775 nlattr_add_u8(nw, PF_SN_RULE_TYPE, n->ruletype); 1776 1777 nlattr_add_u64(nw, PF_SN_CREATION, secs - n->creation); 1778 if (n->expire > secs) 1779 nlattr_add_u64(nw, PF_SN_EXPIRE, n->expire - secs); 1780 else 1781 nlattr_add_u64(nw, PF_SN_EXPIRE, 0); 1782 1783 nlattr_add_pf_threshold(nw, PF_SN_CONNECTION_RATE, 1784 &n->conn_rate); 1785 1786 nlattr_add_u8(nw, PF_SN_NODE_TYPE, n->type); 1787 1788 if (!nlmsg_end(nw)) { 1789 PF_HASHROW_UNLOCK(sh); 1790 nlmsg_abort(nw); 1791 return (ENOMEM); 1792 } 1793 } 1794 PF_HASHROW_UNLOCK(sh); 1795 } 1796 1797 return (0); 1798 } 1799 1800 #define _OUT(_field) offsetof(struct pfioc_table, _field) 1801 static const struct nlattr_parser nla_p_table[] = { 1802 { .type = PF_T_ANCHOR, .off = _OUT(pfrio_table.pfrt_anchor), .arg = (void *)MAXPATHLEN, .cb = nlattr_get_chara }, 1803 { .type = PF_T_NAME, .off = _OUT(pfrio_table.pfrt_name), .arg = (void *)PF_TABLE_NAME_SIZE, .cb = nlattr_get_chara }, 1804 { .type = PF_T_TABLE_FLAGS, .off = _OUT(pfrio_table.pfrt_flags), .cb = nlattr_get_uint32 }, 1805 { .type = PF_T_FLAGS, .off = _OUT(pfrio_flags), .cb = nlattr_get_uint32 }, 1806 }; 1807 static const struct nlfield_parser nlf_p_table[] = {}; 1808 NL_DECLARE_PARSER(table_parser, struct genlmsghdr, nlf_p_table, nla_p_table); 1809 #undef _OUT 1810 static int 1811 pf_handle_clear_tables(struct nlmsghdr *hdr, struct nl_pstate *npt) 1812 { 1813 struct pfioc_table attrs = { 0 }; 1814 struct nl_writer *nw = npt->nw; 1815 struct genlmsghdr *ghdr_new; 1816 int ndel = 0; 1817 int error; 1818 1819 error = nl_parse_nlmsg(hdr, &table_parser, npt, &attrs); 1820 if (error != 0) 1821 return (error); 1822 1823 PF_RULES_WLOCK(); 1824 error = pfr_clr_tables(&attrs.pfrio_table, &ndel, attrs.pfrio_flags | PFR_FLAG_USERIOCTL); 1825 PF_RULES_WUNLOCK(); 1826 if (error != 0) 1827 return (error); 1828 1829 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 1830 return (ENOMEM); 1831 1832 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 1833 ghdr_new->cmd = PFNL_CMD_CLEAR_TABLES; 1834 ghdr_new->version = 0; 1835 ghdr_new->reserved = 0; 1836 1837 nlattr_add_u32(nw, PF_T_NBR_DELETED, ndel); 1838 1839 if (!nlmsg_end(nw)) { 1840 nlmsg_abort(nw); 1841 return (ENOMEM); 1842 } 1843 1844 return (0); 1845 } 1846 1847 static int 1848 pf_handle_add_table(struct nlmsghdr *hdr, struct nl_pstate *npt) 1849 { 1850 struct pfioc_table attrs = { 0 }; 1851 struct nl_writer *nw = npt->nw; 1852 struct genlmsghdr *ghdr_new; 1853 int error; 1854 1855 error = nl_parse_nlmsg(hdr, &table_parser, npt, &attrs); 1856 if (error != 0) 1857 return (error); 1858 1859 PF_RULES_WLOCK(); 1860 error = pfr_add_tables(&attrs.pfrio_table, 1, &attrs.pfrio_nadd, 1861 attrs.pfrio_flags | PFR_FLAG_USERIOCTL); 1862 PF_RULES_WUNLOCK(); 1863 if (error != 0) 1864 return (error); 1865 1866 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 1867 return (ENOMEM); 1868 1869 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 1870 ghdr_new->cmd = PFNL_CMD_ADD_TABLE; 1871 ghdr_new->version = 0; 1872 ghdr_new->reserved = 0; 1873 1874 nlattr_add_u32(nw, PF_T_NBR_ADDED, attrs.pfrio_nadd); 1875 1876 if (!nlmsg_end(nw)) { 1877 nlmsg_abort(nw); 1878 return (ENOMEM); 1879 } 1880 1881 return (0); 1882 } 1883 1884 static int 1885 pf_handle_del_table(struct nlmsghdr *hdr, struct nl_pstate *npt) 1886 { 1887 struct pfioc_table attrs = { 0 }; 1888 struct nl_writer *nw = npt->nw; 1889 struct genlmsghdr *ghdr_new; 1890 int error; 1891 1892 error = nl_parse_nlmsg(hdr, &table_parser, npt, &attrs); 1893 if (error != 0) 1894 return (error); 1895 1896 PF_RULES_WLOCK(); 1897 error = pfr_del_tables(&attrs.pfrio_table, 1, &attrs.pfrio_ndel, 1898 attrs.pfrio_flags | PFR_FLAG_USERIOCTL); 1899 PF_RULES_WUNLOCK(); 1900 if (error != 0) 1901 return (error); 1902 1903 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 1904 return (ENOMEM); 1905 1906 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 1907 ghdr_new->cmd = PFNL_CMD_ADD_TABLE; 1908 ghdr_new->version = 0; 1909 ghdr_new->reserved = 0; 1910 1911 nlattr_add_u32(nw, PF_T_NBR_DELETED, attrs.pfrio_ndel); 1912 1913 if (!nlmsg_end(nw)) { 1914 nlmsg_abort(nw); 1915 return (ENOMEM); 1916 } 1917 1918 return (0); 1919 } 1920 1921 static bool 1922 nlattr_add_pfr_table(struct nl_writer *nw, int attrtype, 1923 struct pfr_table *t) 1924 { 1925 int off = nlattr_add_nested(nw, attrtype); 1926 1927 nlattr_add_string(nw, PF_T_ANCHOR, t->pfrt_anchor); 1928 nlattr_add_string(nw, PF_T_NAME, t->pfrt_name); 1929 nlattr_add_u32(nw, PF_T_TABLE_FLAGS, t->pfrt_flags); 1930 1931 nlattr_set_len(nw, off); 1932 1933 return (true); 1934 } 1935 1936 static int 1937 pf_handle_get_tstats(struct nlmsghdr *hdr, struct nl_pstate *npt) 1938 { 1939 struct pfioc_table attrs = { 0 }; 1940 struct nl_writer *nw = npt->nw; 1941 struct genlmsghdr *ghdr_new; 1942 struct pfr_tstats *pfrtstats; 1943 int error; 1944 int n; 1945 1946 PF_RULES_RLOCK_TRACKER; 1947 1948 error = nl_parse_nlmsg(hdr, &table_parser, npt, &attrs); 1949 if (error != 0) 1950 return (error); 1951 1952 PF_TABLE_STATS_LOCK(); 1953 PF_RULES_RLOCK(); 1954 1955 n = pfr_table_count(&attrs.pfrio_table, attrs.pfrio_flags); 1956 pfrtstats = mallocarray(n, 1957 sizeof(struct pfr_tstats), M_PF, M_NOWAIT | M_ZERO); 1958 1959 error = pfr_get_tstats(&attrs.pfrio_table, pfrtstats, 1960 &n, attrs.pfrio_flags | PFR_FLAG_USERIOCTL); 1961 1962 PF_RULES_RUNLOCK(); 1963 PF_TABLE_STATS_UNLOCK(); 1964 1965 if (error == 0) { 1966 hdr->nlmsg_flags |= NLM_F_MULTI; 1967 1968 for (int i = 0; i < n; i++) { 1969 uint64_t refcnt[PFR_REFCNT_MAX]; 1970 1971 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) { 1972 error = ENOMEM; 1973 break; 1974 } 1975 1976 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 1977 ghdr_new->cmd = PFNL_CMD_GET_TSTATS; 1978 ghdr_new->version = 0; 1979 ghdr_new->reserved = 0; 1980 1981 nlattr_add_pfr_table(nw, PF_TS_TABLE, 1982 &pfrtstats[i].pfrts_t); 1983 nlattr_add_u64_array(nw, PF_TS_PACKETS, 1984 PFR_DIR_MAX * PFR_OP_TABLE_MAX, 1985 (uint64_t *)pfrtstats[i].pfrts_packets); 1986 nlattr_add_u64_array(nw, PF_TS_BYTES, 1987 PFR_DIR_MAX * PFR_OP_TABLE_MAX, 1988 (uint64_t *)pfrtstats[i].pfrts_bytes); 1989 nlattr_add_u64(nw, PF_TS_MATCH, 1990 pfrtstats[i].pfrts_match); 1991 nlattr_add_u64(nw, PF_TS_NOMATCH, 1992 pfrtstats[i].pfrts_nomatch); 1993 nlattr_add_u64(nw, PF_TS_TZERO, 1994 pfrtstats[i].pfrts_tzero); 1995 nlattr_add_u64(nw, PF_TS_CNT, pfrtstats[i].pfrts_cnt); 1996 1997 for (int j = 0; j < PFR_REFCNT_MAX; j++) 1998 refcnt[j] = pfrtstats[i].pfrts_refcnt[j]; 1999 2000 nlattr_add_u64_array(nw, PF_TS_REFCNT, PFR_REFCNT_MAX, 2001 refcnt); 2002 2003 if (! nlmsg_end(nw)) { 2004 error = ENOMEM; 2005 break; 2006 } 2007 } 2008 } 2009 free(pfrtstats, M_PF); 2010 2011 if (!nlmsg_end_dump(npt->nw, error, hdr)) { 2012 NL_LOG(LOG_DEBUG, "Unable to finalize the dump"); 2013 return (ENOMEM); 2014 } 2015 2016 return (error); 2017 } 2018 2019 static int 2020 pf_handle_clear_tstats(struct nlmsghdr *hdr, struct nl_pstate *npt) 2021 { 2022 struct pfioc_table attrs = { 0 }; 2023 struct nl_writer *nw = npt->nw; 2024 struct genlmsghdr *ghdr_new; 2025 int error; 2026 int nzero; 2027 2028 PF_RULES_RLOCK_TRACKER; 2029 2030 error = nl_parse_nlmsg(hdr, &table_parser, npt, &attrs); 2031 if (error != 0) 2032 return (error); 2033 2034 PF_TABLE_STATS_LOCK(); 2035 PF_RULES_RLOCK(); 2036 error = pfr_clr_tstats(&attrs.pfrio_table, 1, 2037 &nzero, attrs.pfrio_flags | PFR_FLAG_USERIOCTL); 2038 PF_RULES_RUNLOCK(); 2039 PF_TABLE_STATS_UNLOCK(); 2040 2041 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 2042 return (ENOMEM); 2043 2044 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 2045 ghdr_new->cmd = PFNL_CMD_CLR_TSTATS; 2046 ghdr_new->version = 0; 2047 ghdr_new->reserved = 0; 2048 2049 nlattr_add_u64(nw, PF_TS_NZERO, nzero); 2050 2051 if (! nlmsg_end(nw)) 2052 error = ENOMEM; 2053 2054 return (error); 2055 } 2056 2057 static int 2058 pf_handle_clear_addrs(struct nlmsghdr *hdr, struct nl_pstate *npt) 2059 { 2060 struct pfioc_table attrs = { 0 }; 2061 struct nl_writer *nw = npt->nw; 2062 struct genlmsghdr *ghdr_new; 2063 int error; 2064 int ndel; 2065 2066 error = nl_parse_nlmsg(hdr, &table_parser, npt, &attrs); 2067 if (error != 0) 2068 return (error); 2069 2070 PF_RULES_WLOCK(); 2071 error = pfr_clr_addrs(&attrs.pfrio_table, &ndel, 2072 attrs.pfrio_flags | PFR_FLAG_USERIOCTL); 2073 PF_RULES_WUNLOCK(); 2074 2075 if (error) 2076 return (error); 2077 2078 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 2079 return (ENOMEM); 2080 2081 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 2082 ghdr_new->cmd = PFNL_CMD_CLR_ADDRS; 2083 ghdr_new->version = 0; 2084 ghdr_new->reserved = 0; 2085 2086 nlattr_add_u64(nw, PF_T_NBR_DELETED, ndel); 2087 2088 if (!nlmsg_end(nw)) 2089 return (ENOMEM); 2090 2091 return (error); 2092 } 2093 2094 TAILQ_HEAD(pfr_addrq, pfr_addr_item); 2095 struct nl_parsed_table_addrs { 2096 struct pfr_table table; 2097 uint32_t flags; 2098 struct pfr_addr addrs[256]; 2099 size_t addr_count; 2100 int nadd; 2101 int ndel; 2102 }; 2103 #define _OUT(_field) offsetof(struct pfr_addr, _field) 2104 static const struct nlattr_parser nla_p_pfr_addr[] = { 2105 { .type = PFR_A_AF, .off = _OUT(pfra_af), .cb = nlattr_get_uint8 }, 2106 { .type = PFR_A_NET, .off = _OUT(pfra_net), .cb = nlattr_get_uint8 }, 2107 { .type = PFR_A_NOT, .off = _OUT(pfra_not), .cb = nlattr_get_bool }, 2108 { .type = PFR_A_ADDR, .off = _OUT(pfra_u), .cb = nlattr_get_in6_addr }, 2109 }; 2110 #undef _OUT 2111 NL_DECLARE_ATTR_PARSER(pfra_addr_parser, nla_p_pfr_addr); 2112 2113 static int 2114 nlattr_get_pfr_addr(struct nlattr *nla, struct nl_pstate *npt, const void *arg, 2115 void *target) 2116 { 2117 struct nl_parsed_table_addrs *attrs = target; 2118 struct pfr_addr addr = { 0 }; 2119 int error; 2120 2121 if (attrs->addr_count >= nitems(attrs->addrs)) 2122 return (E2BIG); 2123 2124 error = nlattr_get_nested(nla, npt, &pfra_addr_parser, &addr); 2125 if (error != 0) 2126 return (error); 2127 2128 memcpy(&attrs->addrs[attrs->addr_count], &addr, sizeof(addr)); 2129 attrs->addr_count++; 2130 2131 return (0); 2132 } 2133 2134 NL_DECLARE_ATTR_PARSER(nested_table_parser, nla_p_table); 2135 2136 #define _OUT(_field) offsetof(struct nl_parsed_table_addrs, _field) 2137 static const struct nlattr_parser nla_p_table_addr[] = { 2138 { .type = PF_TA_TABLE, .off = _OUT(table), .arg = &nested_table_parser, .cb = nlattr_get_nested }, 2139 { .type = PF_TA_ADDR, .cb = nlattr_get_pfr_addr }, 2140 { .type = PF_TA_FLAGS, .off = _OUT(flags), .cb = nlattr_get_uint32 }, 2141 }; 2142 NL_DECLARE_PARSER(table_addr_parser, struct genlmsghdr, nlf_p_empty, nla_p_table_addr); 2143 #undef _OUT 2144 2145 static int 2146 pf_handle_table_add_addrs(struct nlmsghdr *hdr, struct nl_pstate *npt) 2147 { 2148 struct nl_parsed_table_addrs attrs = { 0 }; 2149 struct nl_writer *nw = npt->nw; 2150 struct genlmsghdr *ghdr_new; 2151 int error; 2152 2153 error = nl_parse_nlmsg(hdr, &table_addr_parser, npt, &attrs); 2154 if (error != 0) 2155 return (error); 2156 2157 PF_RULES_WLOCK(); 2158 error = pfr_add_addrs(&attrs.table, &attrs.addrs[0], 2159 attrs.addr_count, &attrs.nadd, attrs.flags | PFR_FLAG_USERIOCTL); 2160 PF_RULES_WUNLOCK(); 2161 2162 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 2163 return (ENOMEM); 2164 2165 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 2166 ghdr_new->cmd = PFNL_CMD_TABLE_ADD_ADDR; 2167 ghdr_new->version = 0; 2168 ghdr_new->reserved = 0; 2169 2170 nlattr_add_u32(nw, PF_TA_NBR_ADDED, attrs.nadd); 2171 2172 if (!nlmsg_end(nw)) 2173 return (ENOMEM); 2174 2175 return (error); 2176 } 2177 2178 static int 2179 pf_handle_table_del_addrs(struct nlmsghdr *hdr, struct nl_pstate *npt) 2180 { 2181 struct nl_parsed_table_addrs attrs = { 0 }; 2182 struct nl_writer *nw = npt->nw; 2183 struct genlmsghdr *ghdr_new; 2184 int error; 2185 2186 error = nl_parse_nlmsg(hdr, &table_addr_parser, npt, &attrs); 2187 if (error != 0) 2188 return (error); 2189 2190 PF_RULES_WLOCK(); 2191 error = pfr_del_addrs(&attrs.table, &attrs.addrs[0], 2192 attrs.addr_count, &attrs.ndel, attrs.flags | PFR_FLAG_USERIOCTL); 2193 PF_RULES_WUNLOCK(); 2194 2195 if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr))) 2196 return (ENOMEM); 2197 2198 ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr); 2199 ghdr_new->cmd = PFNL_CMD_TABLE_DEL_ADDR; 2200 ghdr_new->version = 0; 2201 ghdr_new->reserved = 0; 2202 2203 nlattr_add_u32(nw, PF_TA_NBR_DELETED, attrs.ndel); 2204 2205 if (!nlmsg_end(nw)) 2206 return (ENOMEM); 2207 2208 return (error); 2209 } 2210 2211 static const struct nlhdr_parser *all_parsers[] = { 2212 &state_parser, 2213 &addrule_parser, 2214 &getrules_parser, 2215 &clear_states_parser, 2216 &set_statusif_parser, 2217 &natlook_parser, 2218 &set_debug_parser, 2219 &set_timeout_parser, 2220 &set_limit_parser, 2221 &pool_addr_parser, 2222 &add_addr_parser, 2223 &ruleset_parser, 2224 &table_parser, 2225 &table_addr_parser, 2226 }; 2227 2228 static uint16_t family_id; 2229 2230 static const struct genl_cmd pf_cmds[] = { 2231 { 2232 .cmd_num = PFNL_CMD_GETSTATES, 2233 .cmd_name = "GETSTATES", 2234 .cmd_cb = pf_handle_getstates, 2235 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2236 .cmd_priv = PRIV_NETINET_PF, 2237 }, 2238 { 2239 .cmd_num = PFNL_CMD_GETCREATORS, 2240 .cmd_name = "GETCREATORS", 2241 .cmd_cb = pf_handle_getcreators, 2242 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2243 .cmd_priv = PRIV_NETINET_PF, 2244 }, 2245 { 2246 .cmd_num = PFNL_CMD_START, 2247 .cmd_name = "START", 2248 .cmd_cb = pf_handle_start, 2249 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL, 2250 .cmd_priv = PRIV_NETINET_PF, 2251 }, 2252 { 2253 .cmd_num = PFNL_CMD_STOP, 2254 .cmd_name = "STOP", 2255 .cmd_cb = pf_handle_stop, 2256 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL, 2257 .cmd_priv = PRIV_NETINET_PF, 2258 }, 2259 { 2260 .cmd_num = PFNL_CMD_ADDRULE, 2261 .cmd_name = "ADDRULE", 2262 .cmd_cb = pf_handle_addrule, 2263 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2264 .cmd_priv = PRIV_NETINET_PF, 2265 }, 2266 { 2267 .cmd_num = PFNL_CMD_GETRULES, 2268 .cmd_name = "GETRULES", 2269 .cmd_cb = pf_handle_getrules, 2270 .cmd_flags = GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2271 .cmd_priv = PRIV_NETINET_PF, 2272 }, 2273 { 2274 .cmd_num = PFNL_CMD_GETRULE, 2275 .cmd_name = "GETRULE", 2276 .cmd_cb = pf_handle_getrule, 2277 .cmd_flags = GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2278 .cmd_priv = PRIV_NETINET_PF, 2279 }, 2280 { 2281 .cmd_num = PFNL_CMD_CLRSTATES, 2282 .cmd_name = "CLRSTATES", 2283 .cmd_cb = pf_handle_clear_states, 2284 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2285 .cmd_priv = PRIV_NETINET_PF, 2286 }, 2287 { 2288 .cmd_num = PFNL_CMD_KILLSTATES, 2289 .cmd_name = "KILLSTATES", 2290 .cmd_cb = pf_handle_kill_states, 2291 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2292 .cmd_priv = PRIV_NETINET_PF, 2293 }, 2294 { 2295 .cmd_num = PFNL_CMD_SET_STATUSIF, 2296 .cmd_name = "SETSTATUSIF", 2297 .cmd_cb = pf_handle_set_statusif, 2298 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL, 2299 .cmd_priv = PRIV_NETINET_PF, 2300 }, 2301 { 2302 .cmd_num = PFNL_CMD_GET_STATUS, 2303 .cmd_name = "GETSTATUS", 2304 .cmd_cb = pf_handle_get_status, 2305 .cmd_flags = GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2306 .cmd_priv = PRIV_NETINET_PF, 2307 }, 2308 { 2309 .cmd_num = PFNL_CMD_CLEAR_STATUS, 2310 .cmd_name = "CLEARSTATUS", 2311 .cmd_cb = pf_handle_clear_status, 2312 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL, 2313 .cmd_priv = PRIV_NETINET_PF, 2314 }, 2315 { 2316 .cmd_num = PFNL_CMD_NATLOOK, 2317 .cmd_name = "NATLOOK", 2318 .cmd_cb = pf_handle_natlook, 2319 .cmd_flags = GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2320 .cmd_priv = PRIV_NETINET_PF, 2321 }, 2322 { 2323 .cmd_num = PFNL_CMD_SET_DEBUG, 2324 .cmd_name = "SET_DEBUG", 2325 .cmd_cb = pf_handle_set_debug, 2326 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL, 2327 .cmd_priv = PRIV_NETINET_PF, 2328 }, 2329 { 2330 .cmd_num = PFNL_CMD_SET_TIMEOUT, 2331 .cmd_name = "SET_TIMEOUT", 2332 .cmd_cb = pf_handle_set_timeout, 2333 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL, 2334 .cmd_priv = PRIV_NETINET_PF, 2335 }, 2336 { 2337 .cmd_num = PFNL_CMD_GET_TIMEOUT, 2338 .cmd_name = "GET_TIMEOUT", 2339 .cmd_cb = pf_handle_get_timeout, 2340 .cmd_flags = GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2341 .cmd_priv = PRIV_NETINET_PF, 2342 }, 2343 { 2344 .cmd_num = PFNL_CMD_SET_LIMIT, 2345 .cmd_name = "SET_LIMIT", 2346 .cmd_cb = pf_handle_set_limit, 2347 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL, 2348 .cmd_priv = PRIV_NETINET_PF, 2349 }, 2350 { 2351 .cmd_num = PFNL_CMD_GET_LIMIT, 2352 .cmd_name = "GET_LIMIT", 2353 .cmd_cb = pf_handle_get_limit, 2354 .cmd_flags = GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2355 .cmd_priv = PRIV_NETINET_PF, 2356 }, 2357 { 2358 .cmd_num = PFNL_CMD_BEGIN_ADDRS, 2359 .cmd_name = "BEGIN_ADDRS", 2360 .cmd_cb = pf_handle_begin_addrs, 2361 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2362 .cmd_priv = PRIV_NETINET_PF, 2363 }, 2364 { 2365 .cmd_num = PFNL_CMD_ADD_ADDR, 2366 .cmd_name = "ADD_ADDR", 2367 .cmd_cb = pf_handle_add_addr, 2368 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL, 2369 .cmd_priv = PRIV_NETINET_PF, 2370 }, 2371 { 2372 .cmd_num = PFNL_CMD_GET_ADDRS, 2373 .cmd_name = "GET_ADDRS", 2374 .cmd_cb = pf_handle_get_addrs, 2375 .cmd_flags = GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2376 .cmd_priv = PRIV_NETINET_PF, 2377 }, 2378 { 2379 .cmd_num = PFNL_CMD_GET_ADDR, 2380 .cmd_name = "GET_ADDRS", 2381 .cmd_cb = pf_handle_get_addr, 2382 .cmd_flags = GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2383 .cmd_priv = PRIV_NETINET_PF, 2384 }, 2385 { 2386 .cmd_num = PFNL_CMD_GET_RULESETS, 2387 .cmd_name = "GET_RULESETS", 2388 .cmd_cb = pf_handle_get_rulesets, 2389 .cmd_flags = GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2390 .cmd_priv = PRIV_NETINET_PF, 2391 }, 2392 { 2393 .cmd_num = PFNL_CMD_GET_RULESET, 2394 .cmd_name = "GET_RULESET", 2395 .cmd_cb = pf_handle_get_ruleset, 2396 .cmd_flags = GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2397 .cmd_priv = PRIV_NETINET_PF, 2398 }, 2399 { 2400 .cmd_num = PFNL_CMD_GET_SRCNODES, 2401 .cmd_name = "GET_SRCNODES", 2402 .cmd_cb = pf_handle_get_srcnodes, 2403 .cmd_flags = GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2404 .cmd_priv = PRIV_NETINET_PF, 2405 }, 2406 { 2407 .cmd_num = PFNL_CMD_CLEAR_TABLES, 2408 .cmd_name = "CLEAR_TABLES", 2409 .cmd_cb = pf_handle_clear_tables, 2410 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL, 2411 .cmd_priv = PRIV_NETINET_PF, 2412 }, 2413 { 2414 .cmd_num = PFNL_CMD_ADD_TABLE, 2415 .cmd_name = "ADD_TABLE", 2416 .cmd_cb = pf_handle_add_table, 2417 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL, 2418 .cmd_priv = PRIV_NETINET_PF, 2419 }, 2420 { 2421 .cmd_num = PFNL_CMD_DEL_TABLE, 2422 .cmd_name = "DEL_TABLE", 2423 .cmd_cb = pf_handle_del_table, 2424 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL, 2425 .cmd_priv = PRIV_NETINET_PF, 2426 }, 2427 { 2428 .cmd_num = PFNL_CMD_GET_TSTATS, 2429 .cmd_name = "GET_TSTATS", 2430 .cmd_cb = pf_handle_get_tstats, 2431 .cmd_flags = GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL, 2432 .cmd_priv = PRIV_NETINET_PF, 2433 }, 2434 { 2435 .cmd_num = PFNL_CMD_CLR_TSTATS, 2436 .cmd_name = "CLR_TSTATS", 2437 .cmd_cb = pf_handle_clear_tstats, 2438 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL, 2439 .cmd_priv = PRIV_NETINET_PF, 2440 }, 2441 { 2442 .cmd_num = PFNL_CMD_CLR_ADDRS, 2443 .cmd_name = "CRL_ADDRS", 2444 .cmd_cb = pf_handle_clear_addrs, 2445 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL, 2446 .cmd_priv = PRIV_NETINET_PF, 2447 }, 2448 { 2449 .cmd_num = PFNL_CMD_TABLE_ADD_ADDR, 2450 .cmd_name = "TABLE_ADD_ADDRS", 2451 .cmd_cb = pf_handle_table_add_addrs, 2452 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL, 2453 .cmd_priv = PRIV_NETINET_PF, 2454 }, 2455 { 2456 .cmd_num = PFNL_CMD_TABLE_DEL_ADDR, 2457 .cmd_name = "TABLE_DEL_ADDRS", 2458 .cmd_cb = pf_handle_table_del_addrs, 2459 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL, 2460 .cmd_priv = PRIV_NETINET_PF, 2461 }, 2462 }; 2463 2464 void 2465 pf_nl_register(void) 2466 { 2467 NL_VERIFY_PARSERS(all_parsers); 2468 2469 family_id = genl_register_family(PFNL_FAMILY_NAME, 0, 2, PFNL_CMD_MAX); 2470 genl_register_cmds(family_id, pf_cmds, nitems(pf_cmds)); 2471 } 2472 2473 void 2474 pf_nl_unregister(void) 2475 { 2476 genl_unregister_family(family_id); 2477 } 2478