sys/netpfil/pf/pf_nl.c

   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
  30 #include <sys/cdefs.h>
  31
  32 #include <sys/param.h>
  33 #include <sys/malloc.h>
  34 #include <sys/mbuf.h>
  35 #include <sys/socket.h>
  36
  37 #include <net/pfvar.h>
  38
  39 #include <netlink/netlink.h>
  40 #include <netlink/netlink_ctl.h>
  41 #include <netlink/netlink_generic.h>
  42 #include <netlink/netlink_message_writer.h>
  43
  44 #include <netpfil/pf/pf_nl.h>
  45
  46 #define DEBUG_MOD_NAME  nl_pf
  47 #define DEBUG_MAX_LEVEL LOG_DEBUG3
  48 #include <netlink/netlink_debug.h>
  49 _DECLARE_DEBUG(LOG_DEBUG);
  50
  51 struct nl_parsed_state {
  52         uint8_t         version;
  53         uint32_t        id;
  54         uint32_t        creatorid;
  55         char            ifname[IFNAMSIZ];
  56         uint16_t        proto;
  57         sa_family_t     af;
  58         struct pf_addr  addr;
  59         struct pf_addr  mask;
  60 };
  61
  62 #define _IN(_field)     offsetof(struct genlmsghdr, _field)
  63 #define _OUT(_field)    offsetof(struct nl_parsed_state, _field)
  64 static const struct nlattr_parser nla_p_state[] = {
  65         { .type = PF_ST_ID, .off = _OUT(id), .cb = nlattr_get_uint32 },
  66         { .type = PF_ST_CREATORID, .off = _OUT(creatorid), .cb = nlattr_get_uint32 },
  67         { .type = PF_ST_IFNAME, .arg = (const void *)IFNAMSIZ, .off = _OUT(ifname), .cb = nlattr_get_chara },
  68         { .type = PF_ST_AF, .off = _OUT(proto), .cb = nlattr_get_uint8 },
  69         { .type = PF_ST_PROTO, .off = _OUT(proto), .cb = nlattr_get_uint16 },
  70         { .type = PF_ST_FILTER_ADDR, .off = _OUT(addr), .cb = nlattr_get_in6_addr },
  71         { .type = PF_ST_FILTER_MASK, .off = _OUT(mask), .cb = nlattr_get_in6_addr },
  72 };
  73 static const struct nlfield_parser nlf_p_generic[] = {
  74         { .off_in = _IN(version), .off_out = _OUT(version), .cb = nlf_get_u8 },
  75 };
  76 #undef _IN
  77 #undef _OUT
  78 NL_DECLARE_PARSER(state_parser, struct genlmsghdr, nlf_p_generic, nla_p_state);
  79
  80 static void
  81 dump_addr(struct nl_writer *nw, int attr, const struct pf_addr *addr, int af)
  82 {
  83         switch (af) {
  84         case AF_INET:
  85                 nlattr_add(nw, attr, 4, &addr->v4);
  86                 break;
  87         case AF_INET6:
  88                 nlattr_add(nw, attr, 16, &addr->v6);
  89                 break;
  90         };
  91 }
  92
  93 static bool
  94 dump_state_peer(struct nl_writer *nw, int attr, const struct pf_state_peer *peer)
  95 {
  96         int off = nlattr_add_nested(nw, attr);
  97         if (off == 0)
  98                 return (false);
  99
 100         nlattr_add_u32(nw, PF_STP_SEQLO, peer->seqlo);
 101         nlattr_add_u32(nw, PF_STP_SEQHI, peer->seqhi);
 102         nlattr_add_u32(nw, PF_STP_SEQDIFF, peer->seqdiff);
 103         nlattr_add_u16(nw, PF_STP_MAX_WIN, peer->max_win);
 104         nlattr_add_u16(nw, PF_STP_MSS, peer->mss);
 105         nlattr_add_u8(nw, PF_STP_STATE, peer->state);
 106         nlattr_add_u8(nw, PF_STP_WSCALE, peer->wscale);
 107
 108         if (peer->scrub != NULL) {
 109                 struct pf_state_scrub *sc = peer->scrub;
 110                 uint16_t pfss_flags = sc->pfss_flags & PFSS_TIMESTAMP;
 111
 112                 nlattr_add_u16(nw, PF_STP_PFSS_FLAGS, pfss_flags);
 113                 nlattr_add_u32(nw, PF_STP_PFSS_TS_MOD, sc->pfss_ts_mod);
 114                 nlattr_add_u8(nw, PF_STP_PFSS_TTL, sc->pfss_ttl);
 115                 nlattr_add_u8(nw, PF_STP_SCRUB_FLAG, PFSYNC_SCRUB_FLAG_VALID);
 116         }
 117         nlattr_set_len(nw, off);
 118
 119         return (true);
 120 }
 121
 122 static bool
 123 dump_state_key(struct nl_writer *nw, int attr, const struct pf_state_key *key)
 124 {
 125         int off = nlattr_add_nested(nw, attr);
 126         if (off == 0)
 127                 return (false);
 128
 129         dump_addr(nw, PF_STK_ADDR0, &key->addr[0], key->af);
 130         dump_addr(nw, PF_STK_ADDR1, &key->addr[1], key->af);
 131         nlattr_add_u16(nw, PF_STK_PORT0, key->port[0]);
 132         nlattr_add_u16(nw, PF_STK_PORT1, key->port[1]);
 133
 134         nlattr_set_len(nw, off);
 135
 136         return (true);
 137 }
 138
 139 static int
 140 dump_state(struct nlpcb *nlp, const struct nlmsghdr *hdr, struct pf_kstate *s,
 141     struct nl_pstate *npt)
 142 {
 143         struct nl_writer *nw = npt->nw;
 144         int error = 0;
 145         int af;
 146         struct pf_state_key *key;
 147
 148         if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr)))
 149                 goto enomem;
 150
 151         struct genlmsghdr *ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr);
 152         ghdr_new->cmd = PFNL_CMD_GETSTATES;
 153         ghdr_new->version = 0;
 154         ghdr_new->reserved = 0;
 155
 156         nlattr_add_u64(nw, PF_ST_VERSION, PF_STATE_VERSION);
 157
 158         key = s->key[PF_SK_WIRE];
 159         if (!dump_state_key(nw, PF_ST_KEY_WIRE, key))
 160                 goto enomem;
 161         key = s->key[PF_SK_STACK];
 162         if (!dump_state_key(nw, PF_ST_KEY_STACK, key))
 163                 goto enomem;
 164
 165         af = s->key[PF_SK_WIRE]->af;
 166         nlattr_add_u8(nw, PF_ST_PROTO, s->key[PF_SK_WIRE]->proto);
 167         nlattr_add_u8(nw, PF_ST_AF, af);
 168
 169         nlattr_add_string(nw, PF_ST_IFNAME, s->kif->pfik_name);
 170         nlattr_add_string(nw, PF_ST_ORIG_IFNAME, s->orig_kif->pfik_name);
 171         dump_addr(nw, PF_ST_RT_ADDR, &s->rt_addr, af);
 172         nlattr_add_u32(nw, PF_ST_CREATION, time_uptime - s->creation);
 173         uint32_t expire = pf_state_expires(s);
 174         if (expire > time_uptime)
 175                 expire = expire - time_uptime;
 176         nlattr_add_u32(nw, PF_ST_EXPIRE, expire);
 177         nlattr_add_u8(nw, PF_ST_DIRECTION, s->direction);
 178         nlattr_add_u8(nw, PF_ST_LOG, s->act.log);
 179         nlattr_add_u8(nw, PF_ST_TIMEOUT, s->timeout);
 180         nlattr_add_u16(nw, PF_ST_STATE_FLAGS, s->state_flags);
 181         uint8_t sync_flags = 0;
 182         if (s->src_node)
 183                 sync_flags |= PFSYNC_FLAG_SRCNODE;
 184         if (s->nat_src_node)
 185                 sync_flags |= PFSYNC_FLAG_NATSRCNODE;
 186         nlattr_add_u8(nw, PF_ST_SYNC_FLAGS, sync_flags);
 187         nlattr_add_u64(nw, PF_ST_ID, s->id);
 188         nlattr_add_u32(nw, PF_ST_CREATORID, htonl(s->creatorid));
 189
 190         nlattr_add_u32(nw, PF_ST_RULE, s->rule.ptr ? s->rule.ptr->nr : -1);
 191         nlattr_add_u32(nw, PF_ST_ANCHOR, s->anchor.ptr ? s->anchor.ptr->nr : -1);
 192         nlattr_add_u32(nw, PF_ST_NAT_RULE, s->nat_rule.ptr ? s->nat_rule.ptr->nr : -1);
 193
 194         nlattr_add_u64(nw, PF_ST_PACKETS0, s->packets[0]);
 195         nlattr_add_u64(nw, PF_ST_PACKETS1, s->packets[1]);
 196         nlattr_add_u64(nw, PF_ST_BYTES0, s->bytes[0]);
 197         nlattr_add_u64(nw, PF_ST_BYTES1, s->bytes[1]);
 198
 199         if (!dump_state_peer(nw, PF_ST_PEER_SRC, &s->src))
 200                 goto enomem;
 201         if (!dump_state_peer(nw, PF_ST_PEER_DST, &s->dst))
 202                 goto enomem;
 203
 204         if (nlmsg_end(nw))
 205                 return (0);
 206
 207 enomem:
 208         error = ENOMEM;
 209         nlmsg_abort(nw);
 210         return (error);
 211 }
 212
 213 static int
 214 handle_dumpstates(struct nlpcb *nlp, struct nl_parsed_state *attrs,
 215     struct nlmsghdr *hdr, struct nl_pstate *npt)
 216 {
 217         int error = 0;
 218
 219         hdr->nlmsg_flags |= NLM_F_MULTI;
 220
 221         for (int i = 0; i <= pf_hashmask; i++) {
 222                 struct pf_idhash *ih = &V_pf_idhash[i];
 223                 struct pf_kstate *s;
 224
 225                 if (LIST_EMPTY(&ih->states))
 226                         continue;
 227
 228                 PF_HASHROW_LOCK(ih);
 229                 LIST_FOREACH(s, &ih->states, entry) {
 230                         sa_family_t af = s->key[PF_SK_WIRE]->af;
 231
 232                         if (s->timeout == PFTM_UNLINKED)
 233                                 continue;
 234
 235                         /* Filter */
 236                         if (attrs->creatorid != 0 && s->creatorid != attrs->creatorid)
 237                                 continue;
 238                         if (attrs->ifname[0] != 0 &&
 239                             strncmp(attrs->ifname, s->kif->pfik_name, IFNAMSIZ) != 0)
 240                                 continue;
 241                         if (attrs->proto != 0 && s->key[PF_SK_WIRE]->proto != attrs->proto)
 242                                 continue;
 243                         if (attrs->af != 0 && af != attrs->af)
 244                                 continue;
 245                         if (pf_match_addr(1, &s->key[PF_SK_WIRE]->addr[0],
 246                             &attrs->mask, &attrs->addr, af) &&
 247                             pf_match_addr(1, &s->key[PF_SK_WIRE]->addr[1],
 248                             &attrs->mask, &attrs->addr, af) &&
 249                             pf_match_addr(1, &s->key[PF_SK_STACK]->addr[0],
 250                             &attrs->mask, &attrs->addr, af) &&
 251                             pf_match_addr(1, &s->key[PF_SK_STACK]->addr[1],
 252                             &attrs->mask, &attrs->addr, af))
 253                                 continue;
 254
 255                         error = dump_state(nlp, hdr, s, npt);
 256                         if (error != 0)
 257                                 break;
 258                 }
 259                 PF_HASHROW_UNLOCK(ih);
 260         }
 261
 262         if (!nlmsg_end_dump(npt->nw, error, hdr)) {
 263                 NL_LOG(LOG_DEBUG, "Unable to finalize the dump");
 264                 return (ENOMEM);
 265         }
 266
 267         return (error);
 268 }
 269
 270 static int
 271 handle_getstate(struct nlpcb *nlp, struct nl_parsed_state *attrs,
 272     struct nlmsghdr *hdr, struct nl_pstate *npt)
 273 {
 274         struct pf_kstate *s = pf_find_state_byid(attrs->id, attrs->creatorid);
 275         if (s == NULL)
 276                 return (ENOENT);
 277         return (dump_state(nlp, hdr, s, npt));
 278 }
 279
 280 static int
 281 dump_creatorid(struct nlpcb *nlp, const struct nlmsghdr *hdr, uint32_t creator,
 282     struct nl_pstate *npt)
 283 {
 284         struct nl_writer *nw = npt->nw;
 285
 286         if (!nlmsg_reply(nw, hdr, sizeof(struct genlmsghdr)))
 287                 goto enomem;
 288
 289         struct genlmsghdr *ghdr_new = nlmsg_reserve_object(nw, struct genlmsghdr);
 290         ghdr_new->cmd = PFNL_CMD_GETCREATORS;
 291         ghdr_new->version = 0;
 292         ghdr_new->reserved = 0;
 293
 294         nlattr_add_u32(nw, PF_ST_CREATORID, htonl(creator));
 295
 296         if (nlmsg_end(nw))
 297                 return (0);
 298
 299 enomem:
 300         nlmsg_abort(nw);
 301         return (ENOMEM);
 302 }
 303
 304 static int
 305 pf_handle_getstates(struct nlmsghdr *hdr, struct nl_pstate *npt)
 306 {
 307         int error;
 308
 309         struct nl_parsed_state attrs = {};
 310         error = nl_parse_nlmsg(hdr, &state_parser, npt, &attrs);
 311         if (error != 0)
 312                 return (error);
 313
 314         if (attrs.id != 0)
 315                 error = handle_getstate(npt->nlp, &attrs, hdr, npt);
 316         else
 317                 error = handle_dumpstates(npt->nlp, &attrs, hdr, npt);
 318
 319         return (error);
 320 }
 321
 322 static int
 323 pf_handle_getcreators(struct nlmsghdr *hdr, struct nl_pstate *npt)
 324 {
 325         uint32_t creators[16];
 326         int error = 0;
 327
 328         bzero(creators, sizeof(creators));
 329
 330         for (int i = 0; i < pf_hashmask; i++) {
 331                 struct pf_idhash *ih = &V_pf_idhash[i];
 332                 struct pf_kstate *s;
 333
 334                 if (LIST_EMPTY(&ih->states))
 335                         continue;
 336
 337                 PF_HASHROW_LOCK(ih);
 338                 LIST_FOREACH(s, &ih->states, entry) {
 339                         int j;
 340                         if (s->timeout == PFTM_UNLINKED)
 341                                 continue;
 342
 343                         for (j = 0; j < nitems(creators); j++) {
 344                                 if (creators[j] == s->creatorid)
 345                                         break;
 346                                 if (creators[j] == 0) {
 347                                         creators[j] = s->creatorid;
 348                                         break;
 349                                 }
 350                         }
 351                         if (j == nitems(creators))
 352                                 printf("Warning: too many creators!\n");
 353                 }
 354                 PF_HASHROW_UNLOCK(ih);
 355         }
 356
 357         hdr->nlmsg_flags |= NLM_F_MULTI;
 358         for (int i = 0; i < nitems(creators); i++) {
 359                 if (creators[i] == 0)
 360                         break;
 361                 error = dump_creatorid(npt->nlp, hdr, creators[i], npt);
 362         }
 363
 364         if (!nlmsg_end_dump(npt->nw, error, hdr)) {
 365                 NL_LOG(LOG_DEBUG, "Unable to finalize the dump");
 366                 return (ENOMEM);
 367         }
 368
 369         return (error);
 370 }
 371
 372 static int
 373 pf_handle_start(struct nlmsghdr *hdr __unused, struct nl_pstate *npt __unused)
 374 {
 375         return (pf_start());
 376 }
 377
 378 static int
 379 pf_handle_stop(struct nlmsghdr *hdr __unused, struct nl_pstate *npt __unused)
 380 {
 381         return (pf_stop());
 382 }
 383
 384 #define _OUT(_field)    offsetof(struct pf_addr_wrap, _field)
 385 static const struct nlattr_parser nla_p_addr_wrap[] = {
 386         { .type = PF_AT_ADDR, .off = _OUT(v.a.addr), .cb = nlattr_get_in6_addr },
 387         { .type = PF_AT_MASK, .off = _OUT(v.a.mask), .cb = nlattr_get_in6_addr },
 388         { .type = PF_AT_IFNAME, .off = _OUT(v.ifname), .arg = (void *)IFNAMSIZ,.cb = nlattr_get_chara },
 389         { .type = PF_AT_TABLENAME, .off = _OUT(v.tblname), .arg = (void *)PF_TABLE_NAME_SIZE, .cb = nlattr_get_chara },
 390         { .type = PF_AT_TYPE, .off = _OUT(type), .cb = nlattr_get_uint8 },
 391         { .type = PF_AT_IFLAGS, .off = _OUT(iflags), .cb = nlattr_get_uint8 },
 392 };
 393 NL_DECLARE_ATTR_PARSER(addr_wrap_parser, nla_p_addr_wrap);
 394 #undef _OUT
 395
 396 #define _OUT(_field)    offsetof(struct pf_rule_addr, _field)
 397 static const struct nlattr_parser nla_p_ruleaddr[] = {
 398         { .type = PF_RAT_ADDR, .off = _OUT(addr), .arg = &addr_wrap_parser, .cb = nlattr_get_nested },
 399         { .type = PF_RAT_SRC_PORT, .off = _OUT(port[0]), .cb = nlattr_get_uint16 },
 400         { .type = PF_RAT_DST_PORT, .off = _OUT(port[1]), .cb = nlattr_get_uint16 },
 401         { .type = PF_RAT_NEG, .off = _OUT(neg), .cb = nlattr_get_uint8 },
 402         { .type = PF_RAT_OP, .off = _OUT(port_op), .cb = nlattr_get_uint8 },
 403 };
 404 NL_DECLARE_ATTR_PARSER(rule_addr_parser, nla_p_ruleaddr);
 405 #undef _OUT
 406
 407 #define _OUT(_field)    offsetof(struct pf_mape_portset, _field)
 408 static const struct nlattr_parser nla_p_mape_portset[] = {
 409         { .type = PF_MET_OFFSET, .off = _OUT(offset), .cb = nlattr_get_uint8 },
 410         { .type = PF_MET_PSID_LEN, .off = _OUT(psidlen), .cb = nlattr_get_uint8 },
 411         {. type = PF_MET_PSID, .off = _OUT(psid), .cb = nlattr_get_uint16 },
 412 };
 413 NL_DECLARE_ATTR_PARSER(mape_portset_parser, nla_p_mape_portset);
 414 #undef _OUT
 415
 416 struct nl_parsed_labels
 417 {
 418         char            labels[PF_RULE_MAX_LABEL_COUNT][PF_RULE_LABEL_SIZE];
 419         uint32_t        i;
 420 };
 421
 422 static int
 423 nlattr_get_pf_rule_labels(struct nlattr *nla, struct nl_pstate *npt,
 424     const void *arg, void *target)
 425 {
 426         struct nl_parsed_labels *l = (struct nl_parsed_labels *)target;
 427         int ret;
 428
 429         if (l->i >= PF_RULE_MAX_LABEL_COUNT)
 430                 return (E2BIG);
 431
 432         ret = nlattr_get_chara(nla, npt, (void *)PF_RULE_LABEL_SIZE,
 433             l->labels[l->i]);
 434         if (ret == 0)
 435                 l->i++;
 436
 437         return (ret);
 438 }
 439
 440 #define _OUT(_field)    offsetof(struct nl_parsed_labels, _field)
 441 static const struct nlattr_parser nla_p_labels[] = {
 442         { .type = PF_LT_LABEL, .off = 0, .cb = nlattr_get_pf_rule_labels },
 443 };
 444 NL_DECLARE_ATTR_PARSER(rule_labels_parser, nla_p_labels);
 445 #undef _OUT
 446
 447 static int
 448 nlattr_get_nested_pf_rule_labels(struct nlattr *nla, struct nl_pstate *npt, const void *arg, void *target)
 449 {
 450         struct nl_parsed_labels parsed_labels = { };
 451         int error;
 452
 453         /* Assumes target points to the beginning of the structure */
 454         error = nl_parse_header(NLA_DATA(nla), NLA_DATA_LEN(nla), &rule_labels_parser, npt, &parsed_labels);
 455         if (error != 0)
 456                 return (error);
 457
 458         memcpy(target, parsed_labels.labels, sizeof(parsed_labels));
 459
 460         return (0);
 461 }
 462
 463 #define _OUT(_field)    offsetof(struct pf_kpool, _field)
 464 static const struct nlattr_parser nla_p_pool[] = {
 465         { .type = PF_PT_KEY, .off = _OUT(key), .arg = (void *)sizeof(struct pf_poolhashkey), .cb = nlattr_get_bytes },
 466         { .type = PF_PT_COUNTER, .off = _OUT(counter), .cb = nlattr_get_in6_addr },
 467         { .type = PF_PT_TBLIDX, .off = _OUT(tblidx), .cb = nlattr_get_uint32 },
 468         { .type = PF_PT_PROXY_SRC_PORT, .off = _OUT(proxy_port[0]), .cb = nlattr_get_uint16 },
 469         { .type = PF_PT_PROXY_DST_PORT, .off = _OUT(proxy_port[1]), .cb = nlattr_get_uint16 },
 470         { .type = PF_PT_OPTS, .off = _OUT(opts), .cb = nlattr_get_uint8 },
 471         { .type = PF_PT_MAPE, .off = _OUT(mape), .arg = &mape_portset_parser, .cb = nlattr_get_nested },
 472 };
 473 NL_DECLARE_ATTR_PARSER(pool_parser, nla_p_pool);
 474 #undef _OUT
 475
 476 #define _OUT(_field)    offsetof(struct pf_rule_uid, _field)
 477 static const struct nlattr_parser nla_p_rule_uid[] = {
 478         { .type = PF_RUT_UID_LOW, .off = _OUT(uid[0]), .cb = nlattr_get_uint32 },
 479         { .type = PF_RUT_UID_HIGH, .off = _OUT(uid[1]), .cb = nlattr_get_uint32 },
 480         { .type = PF_RUT_OP, .off = _OUT(op), .cb = nlattr_get_uint8 },
 481 };
 482 NL_DECLARE_ATTR_PARSER(rule_uid_parser, nla_p_rule_uid);
 483 #undef _OUT
 484
 485 struct nl_parsed_timeouts
 486 {
 487         uint32_t        timeouts[PFTM_MAX];
 488         uint32_t        i;
 489 };
 490
 491 static int
 492 nlattr_get_pf_timeout(struct nlattr *nla, struct nl_pstate *npt,
 493     const void *arg, void *target)
 494 {
 495         struct nl_parsed_timeouts *t = (struct nl_parsed_timeouts *)target;
 496         int ret;
 497
 498         if (t->i >= PFTM_MAX)
 499                 return (E2BIG);
 500
 501         ret = nlattr_get_uint32(nla, npt, NULL, &t->timeouts[t->i]);
 502         if (ret == 0)
 503                 t->i++;
 504
 505         return (ret);
 506 }
 507
 508 #define _OUT(_field)    offsetof(struct nl_parsed_timeout, _field)
 509 static const struct nlattr_parser nla_p_timeouts[] = {
 510         { .type = PF_TT_TIMEOUT, .off = 0, .cb = nlattr_get_pf_timeout },
 511 };
 512 NL_DECLARE_ATTR_PARSER(timeout_parser, nla_p_timeouts);
 513 #undef _OUT
 514
 515 static int
 516 nlattr_get_nested_timeouts(struct nlattr *nla, struct nl_pstate *npt, const void *arg, void *target)
 517 {
 518         struct nl_parsed_timeouts parsed_timeouts = { };
 519         int error;
 520
 521         /* Assumes target points to the beginning of the structure */
 522         error = nl_parse_header(NLA_DATA(nla), NLA_DATA_LEN(nla), &timeout_parser, npt, &parsed_timeouts);
 523         if (error != 0)
 524                 return (error);
 525
 526         memcpy(target, parsed_timeouts.timeouts, sizeof(parsed_timeouts.timeouts));
 527
 528         return (0);
 529 }
 530
 531 #define _OUT(_field)    offsetof(struct pf_krule, _field)
 532 static const struct nlattr_parser nla_p_rule[] = {
 533         { .type = PF_RT_SRC, .off = _OUT(src), .arg = &rule_addr_parser,.cb = nlattr_get_nested },
 534         { .type = PF_RT_DST, .off = _OUT(dst), .arg = &rule_addr_parser,.cb = nlattr_get_nested },
 535         { .type = PF_RT_RIDENTIFIER, .off = _OUT(ridentifier), .cb = nlattr_get_uint32 },
 536         { .type = PF_RT_LABELS, .off = _OUT(label), .arg = &rule_labels_parser,.cb = nlattr_get_nested_pf_rule_labels },
 537         { .type = PF_RT_IFNAME, .off = _OUT(ifname), .arg = (void *)IFNAMSIZ, .cb = nlattr_get_chara },
 538         { .type = PF_RT_QNAME, .off = _OUT(qname), .arg = (void *)PF_QNAME_SIZE, .cb = nlattr_get_chara },
 539         { .type = PF_RT_PQNAME, .off = _OUT(pqname), .arg = (void *)PF_QNAME_SIZE, .cb = nlattr_get_chara },
 540         { .type = PF_RT_TAGNAME, .off = _OUT(tagname), .arg = (void *)PF_TAG_NAME_SIZE, .cb = nlattr_get_chara },
 541         { .type = PF_RT_MATCH_TAGNAME, .off = _OUT(match_tagname), .arg = (void *)PF_TAG_NAME_SIZE, .cb = nlattr_get_chara },
 542         { .type = PF_RT_OVERLOAD_TBLNAME, .off = _OUT(overload_tblname), .arg = (void *)PF_TABLE_NAME_SIZE, .cb = nlattr_get_chara },
 543         { .type = PF_RT_RPOOL, .off = _OUT(rpool), .arg = &pool_parser, .cb = nlattr_get_nested },
 544         { .type = PF_RT_OS_FINGERPRINT, .off = _OUT(os_fingerprint), .cb = nlattr_get_uint32 },
 545         { .type = PF_RT_RTABLEID, .off = _OUT(rtableid), .cb = nlattr_get_uint32 },
 546         { .type = PF_RT_TIMEOUT, .off = _OUT(timeout), .arg = &timeout_parser, .cb = nlattr_get_nested_timeouts },
 547         { .type = PF_RT_MAX_STATES, .off = _OUT(max_states), .cb = nlattr_get_uint32 },
 548         { .type = PF_RT_MAX_SRC_NODES, .off = _OUT(max_src_nodes), .cb = nlattr_get_uint32 },
 549         { .type = PF_RT_MAX_SRC_STATES, .off = _OUT(max_src_states), .cb = nlattr_get_uint32 },
 550         { .type = PF_RT_MAX_SRC_CONN_RATE_LIMIT, .off = _OUT(max_src_conn_rate.limit), .cb = nlattr_get_uint32 },
 551         { .type = PF_RT_MAX_SRC_CONN_RATE_SECS, .off = _OUT(max_src_conn_rate.seconds), .cb = nlattr_get_uint32 },
 552         { .type = PF_RT_DNPIPE, .off = _OUT(dnpipe), .cb = nlattr_get_uint16 },
 553         { .type = PF_RT_DNRPIPE, .off = _OUT(dnrpipe), .cb = nlattr_get_uint16 },
 554         { .type = PF_RT_DNFLAGS, .off = _OUT(free_flags), .cb = nlattr_get_uint32 },
 555         { .type = PF_RT_NR, .off = _OUT(nr), .cb = nlattr_get_uint32 },
 556         { .type = PF_RT_PROB, .off = _OUT(prob), .cb = nlattr_get_uint32 },
 557         { .type = PF_RT_CUID, .off = _OUT(cuid), .cb = nlattr_get_uint32 },
 558         {. type = PF_RT_CPID, .off = _OUT(cpid), .cb = nlattr_get_uint32 },
 559         { .type = PF_RT_RETURN_ICMP, .off = _OUT(return_icmp), .cb = nlattr_get_uint16 },
 560         { .type = PF_RT_RETURN_ICMP6, .off = _OUT(return_icmp6), .cb = nlattr_get_uint16 },
 561         { .type = PF_RT_MAX_MSS, .off = _OUT(max_mss), .cb = nlattr_get_uint16 },
 562         { .type = PF_RT_SCRUB_FLAGS, .off = _OUT(scrub_flags), .cb = nlattr_get_uint16 },
 563         { .type = PF_RT_UID, .off = _OUT(uid), .arg = &rule_uid_parser, .cb = nlattr_get_nested },
 564         { .type = PF_RT_GID, .off = _OUT(gid), .arg = &rule_uid_parser, .cb = nlattr_get_nested },
 565         { .type = PF_RT_RULE_FLAG, .off = _OUT(rule_flag), .cb = nlattr_get_uint32 },
 566         { .type = PF_RT_ACTION, .off = _OUT(action), .cb = nlattr_get_uint8 },
 567         { .type = PF_RT_DIRECTION, .off = _OUT(direction), .cb = nlattr_get_uint8 },
 568         { .type = PF_RT_LOG, .off = _OUT(log), .cb = nlattr_get_uint8 },
 569         { .type = PF_RT_LOGIF, .off = _OUT(logif), .cb = nlattr_get_uint8 },
 570         { .type = PF_RT_QUICK, .off = _OUT(quick), .cb = nlattr_get_uint8 },
 571         { .type = PF_RT_IF_NOT, .off = _OUT(ifnot), .cb = nlattr_get_uint8 },
 572         { .type = PF_RT_MATCH_TAG_NOT, .off = _OUT(match_tag_not), .cb = nlattr_get_uint8 },
 573         { .type = PF_RT_NATPASS, .off = _OUT(natpass), .cb = nlattr_get_uint8 },
 574         { .type = PF_RT_KEEP_STATE, .off = _OUT(keep_state), .cb = nlattr_get_uint8 },
 575         { .type = PF_RT_AF, .off = _OUT(af), .cb = nlattr_get_uint8 },
 576         { .type = PF_RT_PROTO, .off = _OUT(proto), .cb = nlattr_get_uint8 },
 577         { .type = PF_RT_TYPE, .off = _OUT(type), .cb = nlattr_get_uint8 },
 578         { .type = PF_RT_CODE, .off = _OUT(code), .cb = nlattr_get_uint8 },
 579         { .type = PF_RT_FLAGS, .off = _OUT(flags), .cb = nlattr_get_uint8 },
 580         { .type = PF_RT_FLAGSET, .off = _OUT(flagset), .cb = nlattr_get_uint8 },
 581         { .type = PF_RT_MIN_TTL, .off = _OUT(min_ttl), .cb = nlattr_get_uint8 },
 582         { .type = PF_RT_ALLOW_OPTS, .off = _OUT(allow_opts), .cb = nlattr_get_uint8 },
 583         { .type = PF_RT_RT, .off = _OUT(rt), .cb = nlattr_get_uint8 },
 584         { .type = PF_RT_RETURN_TTL, .off = _OUT(return_ttl), .cb = nlattr_get_uint8 },
 585         { .type = PF_RT_TOS, .off = _OUT(tos), .cb = nlattr_get_uint8 },
 586         { .type = PF_RT_SET_TOS, .off = _OUT(set_tos), .cb = nlattr_get_uint8 },
 587         { .type = PF_RT_ANCHOR_RELATIVE, .off = _OUT(anchor_relative), .cb = nlattr_get_uint8 },
 588         { .type = PF_RT_ANCHOR_WILDCARD, .off = _OUT(anchor_wildcard), .cb = nlattr_get_uint8 },
 589         { .type = PF_RT_FLUSH, .off = _OUT(flush), .cb = nlattr_get_uint8 },
 590         { .type = PF_RT_PRIO, .off = _OUT(prio), .cb = nlattr_get_uint8 },
 591         { .type = PF_RT_SET_PRIO, .off = _OUT(set_prio[0]), .cb = nlattr_get_uint8 },
 592         { .type = PF_RT_SET_PRIO_REPLY, .off = _OUT(set_prio[1]), .cb = nlattr_get_uint8 },
 593         { .type = PF_RT_DIVERT_ADDRESS, .off = _OUT(divert.addr), .cb = nlattr_get_in6_addr },
 594         { .type = PF_RT_DIVERT_PORT, .off = _OUT(divert.port), .cb = nlattr_get_uint16 },
 595 };
 596 NL_DECLARE_ATTR_PARSER(rule_parser, nla_p_rule);
 597 #undef _OUT
 598 struct nl_parsed_addrule {
 599         struct pf_krule *rule;
 600         uint32_t         ticket;
 601         uint32_t         pool_ticket;
 602         char            *anchor;
 603         char            *anchor_call;
 604 };
 605 #define _IN(_field)     offsetof(struct genlmsghdr, _field)
 606 #define _OUT(_field)    offsetof(struct nl_parsed_addrule, _field)
 607 static const struct nlattr_parser nla_p_addrule[] = {
 608         { .type = PF_ART_TICKET, .off = _OUT(ticket), .cb = nlattr_get_uint32 },
 609         { .type = PF_ART_POOL_TICKET, .off = _OUT(pool_ticket), .cb = nlattr_get_uint32 },
 610         { .type = PF_ART_ANCHOR, .off = _OUT(anchor), .cb = nlattr_get_string },
 611         { .type = PF_ART_ANCHOR_CALL, .off = _OUT(anchor_call), .cb = nlattr_get_string },
 612         { .type = PF_ART_RULE, .off = _OUT(rule), .arg = &rule_parser, .cb = nlattr_get_nested_ptr }
 613 };
 614 static const struct nlfield_parser nlf_p_addrule[] = {
 615 };
 616 #undef _IN
 617 #undef _OUT
 618 NL_DECLARE_PARSER(addrule_parser, struct genlmsghdr, nlf_p_addrule, nla_p_addrule);
 619
 620 static int
 621 pf_handle_addrule(struct nlmsghdr *hdr, struct nl_pstate *npt)
 622 {
 623         int error;
 624         struct nl_parsed_addrule attrs = {};
 625
 626         attrs.rule = pf_krule_alloc();
 627
 628         error = nl_parse_nlmsg(hdr, &addrule_parser, npt, &attrs);
 629         if (error != 0)
 630                 return (error);
 631
 632         error = pf_ioctl_addrule(attrs.rule, attrs.ticket, attrs.pool_ticket,
 633             attrs.anchor, attrs.anchor_call, nlp_get_cred(npt->nlp)->cr_uid,
 634             hdr->nlmsg_pid);
 635
 636         if (error != 0)
 637                 pf_krule_free(attrs.rule);
 638
 639         return (error);
 640 }
 641
 642 static const struct nlhdr_parser *all_parsers[] = { &state_parser, &addrule_parser };
 643
 644 static int family_id;
 645
 646 static const struct genl_cmd pf_cmds[] = {
 647         {
 648                 .cmd_num = PFNL_CMD_GETSTATES,
 649                 .cmd_name = "GETSTATES",
 650                 .cmd_cb = pf_handle_getstates,
 651                 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL,
 652         },
 653         {
 654                 .cmd_num = PFNL_CMD_GETCREATORS,
 655                 .cmd_name = "GETCREATORS",
 656                 .cmd_cb = pf_handle_getcreators,
 657                 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL,
 658         },
 659         {
 660                 .cmd_num = PFNL_CMD_START,
 661                 .cmd_name = "START",
 662                 .cmd_cb = pf_handle_start,
 663                 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL,
 664         },
 665         {
 666                 .cmd_num = PFNL_CMD_STOP,
 667                 .cmd_name = "STOP",
 668                 .cmd_cb = pf_handle_stop,
 669                 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_HASPOL,
 670         },
 671         {
 672                 .cmd_num = PFNL_CMD_ADDRULE,
 673                 .cmd_name = "ADDRULE",
 674                 .cmd_cb = pf_handle_addrule,
 675                 .cmd_flags = GENL_CMD_CAP_DO | GENL_CMD_CAP_DUMP | GENL_CMD_CAP_HASPOL,
 676         },
 677
 678 };
 679
 680 void
 681 pf_nl_register(void)
 682 {
 683         NL_VERIFY_PARSERS(all_parsers);
 684
 685         family_id = genl_register_family(PFNL_FAMILY_NAME, 0, 2, PFNL_CMD_MAX);
 686         genl_register_cmds(PFNL_FAMILY_NAME, pf_cmds, NL_ARRAY_LEN(pf_cmds));
 687 }
 688
 689 void
 690 pf_nl_unregister(void)
 691 {
 692         genl_unregister_family(PFNL_FAMILY_NAME);
 693 }