sys/netpfil/ipfw/ip_fw_private.h

   1 /*-
   2  * Copyright (c) 2002-2009 Luigi Rizzo, Universita` di Pisa
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  14  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  16  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  17  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  18  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  19  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  20  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  21  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  22  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  23  * SUCH DAMAGE.
  24  *
  25  * $FreeBSD$
  26  */
  27
  28 #ifndef _IPFW2_PRIVATE_H
  29 #define _IPFW2_PRIVATE_H
  30
  31 /*
  32  * Internal constants and data structures used by ipfw components
  33  * and not meant to be exported outside the kernel.
  34  */
  35
  36 #ifdef _KERNEL
  37
  38 /*
  39  * For platforms that do not have SYSCTL support, we wrap the
  40  * SYSCTL_* into a function (one per file) to collect the values
  41  * into an array at module initialization. The wrapping macros,
  42  * SYSBEGIN() and SYSEND, are empty in the default case.
  43  */
  44 #ifndef SYSBEGIN
  45 #define SYSBEGIN(x)
  46 #endif
  47 #ifndef SYSEND
  48 #define SYSEND
  49 #endif
  50
  51 /* Return values from ipfw_chk() */
  52 enum {
  53         IP_FW_PASS = 0,
  54         IP_FW_DENY,
  55         IP_FW_DIVERT,
  56         IP_FW_TEE,
  57         IP_FW_DUMMYNET,
  58         IP_FW_NETGRAPH,
  59         IP_FW_NGTEE,
  60         IP_FW_NAT,
  61         IP_FW_REASS,
  62 };
  63
  64 /*
  65  * Structure for collecting parameters to dummynet for ip6_output forwarding
  66  */
  67 struct _ip6dn_args {
  68        struct ip6_pktopts *opt_or;
  69        int flags_or;
  70        struct ip6_moptions *im6o_or;
  71        struct ifnet *origifp_or;
  72        struct ifnet *ifp_or;
  73        struct sockaddr_in6 dst_or;
  74        u_long mtu_or;
  75 };
  76
  77
  78 /*
  79  * Arguments for calling ipfw_chk() and dummynet_io(). We put them
  80  * all into a structure because this way it is easier and more
  81  * efficient to pass variables around and extend the interface.
  82  */
  83 struct ip_fw_args {
  84         struct mbuf     *m;             /* the mbuf chain               */
  85         struct ifnet    *oif;           /* output interface             */
  86         struct sockaddr_in *next_hop;   /* forward address              */
  87         struct sockaddr_in6 *next_hop6; /* ipv6 forward address         */
  88
  89         /*
  90          * On return, it points to the matching rule.
  91          * On entry, rule.slot > 0 means the info is valid and
  92          * contains the starting rule for an ipfw search.
  93          * If chain_id == chain->id && slot >0 then jump to that slot.
  94          * Otherwise, we locate the first rule >= rulenum:rule_id
  95          */
  96         struct ipfw_rule_ref rule;      /* match/restart info           */
  97
  98         struct ether_header *eh;        /* for bridged packets          */
  99
 100         struct ipfw_flow_id f_id;       /* grabbed from IP header       */
 101         //uint32_t      cookie;         /* a cookie depending on rule action */
 102         struct inpcb    *inp;
 103
 104         struct _ip6dn_args      dummypar; /* dummynet->ip6_output */
 105         union {         /* store here if cannot use a pointer */
 106                 struct sockaddr_in hopstore;
 107                 struct sockaddr_in6 hopstore6;
 108         };
 109 };
 110
 111 MALLOC_DECLARE(M_IPFW);
 112
 113 /*
 114  * Hooks sometime need to know the direction of the packet
 115  * (divert, dummynet, netgraph, ...)
 116  * We use a generic definition here, with bit0-1 indicating the
 117  * direction, bit 2 indicating layer2 or 3, bit 3-4 indicating the
 118  * specific protocol
 119  * indicating the protocol (if necessary)
 120  */
 121 enum {
 122         DIR_MASK =      0x3,
 123         DIR_OUT =       0,
 124         DIR_IN =        1,
 125         DIR_FWD =       2,
 126         DIR_DROP =      3,
 127         PROTO_LAYER2 =  0x4, /* set for layer 2 */
 128         /* PROTO_DEFAULT = 0, */
 129         PROTO_IPV4 =    0x08,
 130         PROTO_IPV6 =    0x10,
 131         PROTO_IFB =     0x0c, /* layer2 + ifbridge */
 132    /*   PROTO_OLDBDG =  0x14, unused, old bridge */
 133 };
 134
 135 /* wrapper for freeing a packet, in case we need to do more work */
 136 #ifndef FREE_PKT
 137 #if defined(__linux__) || defined(_WIN32)
 138 #define FREE_PKT(m)     netisr_dispatch(-1, m)
 139 #else
 140 #define FREE_PKT(m)     m_freem(m)
 141 #endif
 142 #endif /* !FREE_PKT */
 143
 144 /*
 145  * Function definitions.
 146  */
 147
 148 /* attach (arg = 1) or detach (arg = 0) hooks */
 149 int ipfw_attach_hooks(int);
 150 #ifdef NOTYET
 151 void ipfw_nat_destroy(void);
 152 #endif
 153
 154 /* In ip_fw_log.c */
 155 struct ip;
 156 struct ip_fw_chain;
 157 void ipfw_log_bpf(int);
 158 void ipfw_log(struct ip_fw_chain *chain, struct ip_fw *f, u_int hlen,
 159     struct ip_fw_args *args, struct mbuf *m, struct ifnet *oif,
 160     u_short offset, uint32_t tablearg, struct ip *ip);
 161 VNET_DECLARE(u_int64_t, norule_counter);
 162 #define V_norule_counter        VNET(norule_counter)
 163 VNET_DECLARE(int, verbose_limit);
 164 #define V_verbose_limit         VNET(verbose_limit)
 165
 166 /* In ip_fw_dynamic.c */
 167
 168 enum { /* result for matching dynamic rules */
 169         MATCH_REVERSE = 0,
 170         MATCH_FORWARD,
 171         MATCH_NONE,
 172         MATCH_UNKNOWN,
 173 };
 174
 175 /*
 176  * The lock for dynamic rules is only used once outside the file,
 177  * and only to release the result of lookup_dyn_rule().
 178  * Eventually we may implement it with a callback on the function.
 179  */
 180 struct ip_fw_chain;
 181 struct sockopt_data;
 182 int ipfw_is_dyn_rule(struct ip_fw *rule);
 183 void ipfw_expire_dyn_rules(struct ip_fw_chain *, ipfw_range_tlv *);
 184 void ipfw_dyn_unlock(ipfw_dyn_rule *q);
 185
 186 struct tcphdr;
 187 struct mbuf *ipfw_send_pkt(struct mbuf *, struct ipfw_flow_id *,
 188     u_int32_t, u_int32_t, int);
 189 int ipfw_install_state(struct ip_fw_chain *chain, struct ip_fw *rule,
 190     ipfw_insn_limit *cmd, struct ip_fw_args *args, uint32_t tablearg);
 191 ipfw_dyn_rule *ipfw_lookup_dyn_rule(struct ipfw_flow_id *pkt,
 192         int *match_direction, struct tcphdr *tcp);
 193 void ipfw_remove_dyn_children(struct ip_fw *rule);
 194 void ipfw_get_dynamic(struct ip_fw_chain *chain, char **bp, const char *ep);
 195 int ipfw_dump_states(struct ip_fw_chain *chain, struct sockopt_data *sd);
 196
 197 void ipfw_dyn_init(struct ip_fw_chain *);       /* per-vnet initialization */
 198 void ipfw_dyn_uninit(int);      /* per-vnet deinitialization */
 199 int ipfw_dyn_len(void);
 200 int ipfw_dyn_get_count(void);
 201
 202 /* common variables */
 203 VNET_DECLARE(int, fw_one_pass);
 204 #define V_fw_one_pass           VNET(fw_one_pass)
 205
 206 VNET_DECLARE(int, fw_verbose);
 207 #define V_fw_verbose            VNET(fw_verbose)
 208
 209 VNET_DECLARE(struct ip_fw_chain, layer3_chain);
 210 #define V_layer3_chain          VNET(layer3_chain)
 211
 212 VNET_DECLARE(int, ipfw_vnet_ready);
 213 #define V_ipfw_vnet_ready       VNET(ipfw_vnet_ready)
 214
 215 VNET_DECLARE(u_int32_t, set_disable);
 216 #define V_set_disable           VNET(set_disable)
 217
 218 VNET_DECLARE(int, autoinc_step);
 219 #define V_autoinc_step          VNET(autoinc_step)
 220
 221 VNET_DECLARE(unsigned int, fw_tables_max);
 222 #define V_fw_tables_max         VNET(fw_tables_max)
 223
 224 VNET_DECLARE(unsigned int, fw_tables_sets);
 225 #define V_fw_tables_sets        VNET(fw_tables_sets)
 226
 227 struct tables_config;
 228
 229 #ifdef _KERNEL
 230 /*
 231  * Here we have the structure representing an ipfw rule.
 232  *
 233  * It starts with a general area
 234  * followed by an array of one or more instructions, which the code
 235  * accesses as an array of 32-bit values.
 236  *
 237  * Given a rule pointer  r:
 238  *
 239  *  r->cmd              is the start of the first instruction.
 240  *  ACTION_PTR(r)       is the start of the first action (things to do
 241  *                      once a rule matched).
 242  */
 243
 244 struct ip_fw {
 245         uint16_t        act_ofs;        /* offset of action in 32-bit units */
 246         uint16_t        cmd_len;        /* # of 32-bit words in cmd     */
 247         uint16_t        rulenum;        /* rule number                  */
 248         uint8_t         set;            /* rule set (0..31)             */
 249         uint8_t         flags;          /* currently unused             */
 250         counter_u64_t   cntr;           /* Pointer to rule counters     */
 251         uint32_t        timestamp;      /* tv_sec of last match         */
 252         uint32_t        id;             /* rule id                      */
 253         uint32_t        cached_id;      /* used by jump_fast            */
 254         uint32_t        cached_pos;     /* used by jump_fast            */
 255
 256         ipfw_insn       cmd[1];         /* storage for commands         */
 257 };
 258
 259 #define IPFW_RULE_CNTR_SIZE     (2 * sizeof(counter_u64_t))
 260
 261 #endif
 262
 263 struct ip_fw_chain {
 264         struct ip_fw    **map;          /* array of rule ptrs to ease lookup */
 265         uint32_t        id;             /* ruleset id */
 266         int             n_rules;        /* number of static rules */
 267         void            *tablestate;    /* runtime table info */
 268         void            *valuestate;    /* runtime table value info */
 269         int             *idxmap;        /* skipto array of rules */
 270         void            **srvstate;     /* runtime service mappings */
 271 #if defined( __linux__ ) || defined( _WIN32 )
 272         spinlock_t rwmtx;
 273 #else
 274         struct rmlock   rwmtx;
 275 #endif
 276         int             static_len;     /* total len of static rules (v0) */
 277         uint32_t        gencnt;         /* NAT generation count */
 278         LIST_HEAD(nat_list, cfg_nat) nat;       /* list of nat entries */
 279         struct ip_fw    *default_rule;
 280         struct tables_config *tblcfg;   /* tables module data */
 281         void            *ifcfg;         /* interface module data */
 282         int             *idxmap_back;   /* standby skipto array of rules */
 283         struct namedobj_instance        *srvmap; /* cfg name->number mappings */
 284 #if defined( __linux__ ) || defined( _WIN32 )
 285         spinlock_t uh_lock;
 286 #else
 287         struct rwlock   uh_lock;        /* lock for upper half */
 288 #endif
 289 };
 290
 291 /* 64-byte structure representing multi-field table value */
 292 struct table_value {
 293         uint32_t        tag;            /* O_TAG/O_TAGGED */
 294         uint32_t        pipe;           /* O_PIPE/O_QUEUE */
 295         uint16_t        divert;         /* O_DIVERT/O_TEE */
 296         uint16_t        skipto;         /* skipto, CALLRET */
 297         uint32_t        netgraph;       /* O_NETGRAPH/O_NGTEE */
 298         uint32_t        fib;            /* O_SETFIB */
 299         uint32_t        nat;            /* O_NAT */
 300         uint32_t        nh4;
 301         uint8_t         dscp;
 302         uint8_t         spare0;
 303         uint16_t        spare1;
 304         /* -- 32 bytes -- */
 305         struct in6_addr nh6;
 306         uint32_t        limit;          /* O_LIMIT */
 307         uint32_t        zoneid;         /* scope zone id for nh6 */
 308         uint64_t        refcnt;         /* Number of references */
 309 };
 310
 311
 312 struct named_object {
 313         TAILQ_ENTRY(named_object)       nn_next;        /* namehash */
 314         TAILQ_ENTRY(named_object)       nv_next;        /* valuehash */
 315         char                    *name;  /* object name */
 316         uint8_t                 subtype;        /* object subtype within class */
 317         uint8_t                 etlv;   /* Export TLV id */
 318         uint16_t                spare[2];
 319         uint16_t                kidx;   /* object kernel index */
 320         uint32_t                set;    /* set object belongs to */
 321         uint32_t                refcnt; /* number of references */
 322 };
 323 TAILQ_HEAD(namedobjects_head, named_object);
 324
 325 struct sockopt; /* used by tcp_var.h */
 326 struct sockopt_data {
 327         caddr_t         kbuf;           /* allocated buffer */
 328         size_t          ksize;          /* given buffer size */
 329         size_t          koff;           /* data already used */
 330         size_t          kavail;         /* number of bytes available */
 331         size_t          ktotal;         /* total bytes pushed */
 332         struct sockopt  *sopt;          /* socket data */
 333         caddr_t         sopt_val;       /* sopt user buffer */
 334         size_t          valsize;        /* original data size */
 335 };
 336
 337 struct ipfw_ifc;
 338
 339 typedef void (ipfw_ifc_cb)(struct ip_fw_chain *ch, void *cbdata,
 340     uint16_t ifindex);
 341
 342 struct ipfw_iface {
 343         struct named_object     no;
 344         char ifname[64];
 345         int resolved;
 346         uint16_t ifindex;
 347         uint16_t spare;
 348         uint64_t gencnt;
 349         TAILQ_HEAD(, ipfw_ifc)  consumers;
 350 };
 351
 352 struct ipfw_ifc {
 353         TAILQ_ENTRY(ipfw_ifc)   next;
 354         struct ipfw_iface       *iface;
 355         ipfw_ifc_cb             *cb;
 356         void                    *cbdata;
 357 };
 358
 359 /* Macro for working with various counters */
 360 #define IPFW_INC_RULE_COUNTER(_cntr, _bytes)    do {    \
 361         counter_u64_add((_cntr)->cntr, 1);              \
 362         counter_u64_add((_cntr)->cntr + 1, _bytes);     \
 363         if ((_cntr)->timestamp != time_uptime)          \
 364                 (_cntr)->timestamp = time_uptime;       \
 365         } while (0)
 366
 367 #define IPFW_INC_DYN_COUNTER(_cntr, _bytes)     do {            \
 368         (_cntr)->pcnt++;                                \
 369         (_cntr)->bcnt += _bytes;                        \
 370         } while (0)
 371
 372 #define IPFW_ZERO_RULE_COUNTER(_cntr) do {              \
 373         counter_u64_zero((_cntr)->cntr);                \
 374         counter_u64_zero((_cntr)->cntr + 1);            \
 375         (_cntr)->timestamp = 0;                         \
 376         } while (0)
 377
 378 #define IPFW_ZERO_DYN_COUNTER(_cntr) do {               \
 379         (_cntr)->pcnt = 0;                              \
 380         (_cntr)->bcnt = 0;                              \
 381         } while (0)
 382
 383 #define TARG_VAL(ch, k, f)      ((struct table_value *)((ch)->valuestate))[k].f
 384 #define IP_FW_ARG_TABLEARG(ch, a, f)    \
 385         (((a) == IP_FW_TARG) ? TARG_VAL(ch, tablearg, f) : (a))
 386 /*
 387  * The lock is heavily used by ip_fw2.c (the main file) and ip_fw_nat.c
 388  * so the variable and the macros must be here.
 389  */
 390
 391 #if defined( __linux__ ) || defined( _WIN32 )
 392 #define IPFW_LOCK_INIT(_chain) do {                     \
 393         rw_init(&(_chain)->rwmtx, "IPFW static rules"); \
 394         rw_init(&(_chain)->uh_lock, "IPFW UH lock");    \
 395         } while (0)
 396
 397 #define IPFW_LOCK_DESTROY(_chain) do {                  \
 398         rw_destroy(&(_chain)->rwmtx);                   \
 399         rw_destroy(&(_chain)->uh_lock);                 \
 400         } while (0)
 401
 402 #define IPFW_RLOCK_ASSERT(_chain)       rw_assert(&(_chain)->rwmtx, RA_RLOCKED)
 403 #define IPFW_WLOCK_ASSERT(_chain)       rw_assert(&(_chain)->rwmtx, RA_WLOCKED)
 404
 405 #define IPFW_RLOCK_TRACKER
 406 #define IPFW_RLOCK(p)                   rw_rlock(&(p)->rwmtx)
 407 #define IPFW_RUNLOCK(p)                 rw_runlock(&(p)->rwmtx)
 408 #define IPFW_WLOCK(p)                   rw_wlock(&(p)->rwmtx)
 409 #define IPFW_WUNLOCK(p)                 rw_wunlock(&(p)->rwmtx)
 410 #define IPFW_PF_RLOCK(p)                IPFW_RLOCK(p)
 411 #define IPFW_PF_RUNLOCK(p)              IPFW_RUNLOCK(p)
 412 #else /* FreeBSD */
 413 #define IPFW_LOCK_INIT(_chain) do {                     \
 414         rm_init(&(_chain)->rwmtx, "IPFW static rules"); \
 415         rw_init(&(_chain)->uh_lock, "IPFW UH lock");    \
 416         } while (0)
 417
 418 #define IPFW_LOCK_DESTROY(_chain) do {                  \
 419         rm_destroy(&(_chain)->rwmtx);                   \
 420         rw_destroy(&(_chain)->uh_lock);                 \
 421         } while (0)
 422
 423 #define IPFW_RLOCK_ASSERT(_chain)       rm_assert(&(_chain)->rwmtx, RA_RLOCKED)
 424 #define IPFW_WLOCK_ASSERT(_chain)       rm_assert(&(_chain)->rwmtx, RA_WLOCKED)
 425
 426 #define IPFW_RLOCK_TRACKER              struct rm_priotracker _tracker
 427 #define IPFW_RLOCK(p)                   rm_rlock(&(p)->rwmtx, &_tracker)
 428 #define IPFW_RUNLOCK(p)                 rm_runlock(&(p)->rwmtx, &_tracker)
 429 #define IPFW_WLOCK(p)                   rm_wlock(&(p)->rwmtx)
 430 #define IPFW_WUNLOCK(p)                 rm_wunlock(&(p)->rwmtx)
 431 #define IPFW_PF_RLOCK(p)                IPFW_RLOCK(p)
 432 #define IPFW_PF_RUNLOCK(p)              IPFW_RUNLOCK(p)
 433 #endif
 434
 435 #define IPFW_UH_RLOCK_ASSERT(_chain)    rw_assert(&(_chain)->uh_lock, RA_RLOCKED)
 436 #define IPFW_UH_WLOCK_ASSERT(_chain)    rw_assert(&(_chain)->uh_lock, RA_WLOCKED)
 437 #define IPFW_UH_UNLOCK_ASSERT(_chain)   rw_assert(&(_chain)->uh_lock, RA_UNLOCKED)
 438
 439 #define IPFW_UH_RLOCK(p) rw_rlock(&(p)->uh_lock)
 440 #define IPFW_UH_RUNLOCK(p) rw_runlock(&(p)->uh_lock)
 441 #define IPFW_UH_WLOCK(p) rw_wlock(&(p)->uh_lock)
 442 #define IPFW_UH_WUNLOCK(p) rw_wunlock(&(p)->uh_lock)
 443
 444 struct obj_idx {
 445         uint16_t        uidx;   /* internal index supplied by userland */
 446         uint16_t        kidx;   /* kernel object index */
 447         uint16_t        off;    /* tlv offset from rule end in 4-byte words */
 448         uint8_t         spare;
 449         uint8_t         type;   /* object type within its category */
 450 };
 451
 452 struct rule_check_info {
 453         uint16_t        flags;          /* rule-specific check flags */
 454         uint16_t        object_opcodes; /* num of opcodes referencing objects */
 455         uint16_t        urule_numoff;   /* offset of rulenum in bytes */
 456         uint8_t         version;        /* rule version */
 457         uint8_t         spare;
 458         ipfw_obj_ctlv   *ctlv;          /* name TLV containter */
 459         struct ip_fw    *krule;         /* resulting rule pointer */
 460         caddr_t         urule;          /* original rule pointer */
 461         struct obj_idx  obuf[8];        /* table references storage */
 462 };
 463
 464 /* Legacy interface support */
 465 /*
 466  * FreeBSD 8 export rule format
 467  */
 468 struct ip_fw_rule0 {
 469         struct ip_fw    *x_next;        /* linked list of rules         */
 470         struct ip_fw    *next_rule;     /* ptr to next [skipto] rule    */
 471         /* 'next_rule' is used to pass up 'set_disable' status          */
 472
 473         uint16_t        act_ofs;        /* offset of action in 32-bit units */
 474         uint16_t        cmd_len;        /* # of 32-bit words in cmd     */
 475         uint16_t        rulenum;        /* rule number                  */
 476         uint8_t         set;            /* rule set (0..31)             */
 477         uint8_t         _pad;           /* padding                      */
 478         uint32_t        id;             /* rule id */
 479
 480         /* These fields are present in all rules.                       */
 481         uint64_t        pcnt;           /* Packet counter               */
 482         uint64_t        bcnt;           /* Byte counter                 */
 483         uint32_t        timestamp;      /* tv_sec of last match         */
 484
 485         ipfw_insn       cmd[1];         /* storage for commands         */
 486 };
 487
 488 struct ip_fw_bcounter0 {
 489         uint64_t        pcnt;           /* Packet counter               */
 490         uint64_t        bcnt;           /* Byte counter                 */
 491         uint32_t        timestamp;      /* tv_sec of last match         */
 492 };
 493
 494 /* Kernel rule length */
 495 /*
 496  * RULE _K_ SIZE _V_ ->
 497  * get kernel size from userland rool version _V_.
 498  * RULE _U_ SIZE _V_ ->
 499  * get user size version _V_ from kernel rule
 500  * RULESIZE _V_ ->
 501  * get user size rule length
 502  */
 503 /* FreeBSD8 <> current kernel format */
 504 #define RULEUSIZE0(r)   (sizeof(struct ip_fw_rule0) + (r)->cmd_len * 4 - 4)
 505 #define RULEKSIZE0(r)   roundup2((sizeof(struct ip_fw) + (r)->cmd_len*4 - 4), 8)
 506 /* FreeBSD11 <> current kernel format */
 507 #define RULEUSIZE1(r)   (roundup2(sizeof(struct ip_fw_rule) + \
 508     (r)->cmd_len * 4 - 4, 8))
 509 #define RULEKSIZE1(r)   roundup2((sizeof(struct ip_fw) + (r)->cmd_len*4 - 4), 8)
 510
 511 /*
 512  * Tables/Objects index rewriting code
 513  */
 514
 515 /* Default and maximum number of ipfw tables/objects. */
 516 #define IPFW_TABLES_MAX         65536
 517 #define IPFW_TABLES_DEFAULT     128
 518 #define IPFW_OBJECTS_MAX        65536
 519 #define IPFW_OBJECTS_DEFAULT    128
 520
 521 #define CHAIN_TO_SRV(ch)        ((ch)->srvmap)
 522
 523 struct tid_info {
 524         uint32_t        set;    /* table set */
 525         uint16_t        uidx;   /* table index */
 526         uint8_t         type;   /* table type */
 527         uint8_t         atype;
 528         uint8_t         spare;
 529         int             tlen;   /* Total TLV size block */
 530         void            *tlvs;  /* Pointer to first TLV */
 531 };
 532
 533 /*
 534  * Classifier callback. Checks if @cmd opcode contains kernel object reference.
 535  * If true, returns its index and type.
 536  * Returns 0 if match is found, 1 overwise.
 537  */
 538 typedef int (ipfw_obj_rw_cl)(ipfw_insn *cmd, uint16_t *puidx, uint8_t *ptype);
 539 /*
 540  * Updater callback. Sets kernel object reference index to @puidx
 541  */
 542 typedef void (ipfw_obj_rw_upd)(ipfw_insn *cmd, uint16_t puidx);
 543 /*
 544  * Finder callback. Tries to find named object by name (specified via @ti).
 545  * Stores found named object pointer in @pno.
 546  * If object was not found, NULL is stored.
 547  *
 548  * Return 0 if input data was valid.
 549  */
 550 typedef int (ipfw_obj_fname_cb)(struct ip_fw_chain *ch,
 551     struct tid_info *ti, struct named_object **pno);
 552 /*
 553  * Another finder callback. Tries to findex named object by kernel index.
 554  *
 555  * Returns pointer to named object or NULL.
 556  */
 557 typedef struct named_object *(ipfw_obj_fidx_cb)(struct ip_fw_chain *ch,
 558     uint16_t kidx);
 559 /*
 560  * Object creator callback. Tries to create object specified by @ti.
 561  * Stores newly-allocated object index in @pkidx.
 562  *
 563  * Returns 0 on success.
 564  */
 565 typedef int (ipfw_obj_create_cb)(struct ip_fw_chain *ch, struct tid_info *ti,
 566     uint16_t *pkidx);
 567
 568
 569 struct opcode_obj_rewrite {
 570         uint32_t                opcode;         /* Opcode to act upon */
 571         uint32_t                etlv;           /* Relevant export TLV id  */
 572         ipfw_obj_rw_cl          *classifier;    /* Check if rewrite is needed */
 573         ipfw_obj_rw_upd         *update;        /* update cmd with new value */
 574         ipfw_obj_fname_cb       *find_byname;   /* Find named object by name */
 575         ipfw_obj_fidx_cb        *find_bykidx;   /* Find named object by kidx */
 576         ipfw_obj_create_cb      *create_object; /* Create named object */
 577 };
 578
 579 #define IPFW_ADD_OBJ_REWRITER(f, c)     do {    \
 580         if ((f) != 0)                           \
 581                 ipfw_add_obj_rewriter(c,        \
 582                     sizeof(c) / sizeof(c[0]));  \
 583         } while(0)
 584 #define IPFW_DEL_OBJ_REWRITER(l, c)     do {    \
 585         if ((l) != 0)                           \
 586                 ipfw_del_obj_rewriter(c,        \
 587                     sizeof(c) / sizeof(c[0]));  \
 588         } while(0)
 589
 590 /* In ip_fw_iface.c */
 591 int ipfw_iface_init(void);
 592 void ipfw_iface_destroy(void);
 593 void vnet_ipfw_iface_destroy(struct ip_fw_chain *ch);
 594 int ipfw_iface_ref(struct ip_fw_chain *ch, char *name,
 595     struct ipfw_ifc *ic);
 596 void ipfw_iface_unref(struct ip_fw_chain *ch, struct ipfw_ifc *ic);
 597 void ipfw_iface_add_notify(struct ip_fw_chain *ch, struct ipfw_ifc *ic);
 598 void ipfw_iface_del_notify(struct ip_fw_chain *ch, struct ipfw_ifc *ic);
 599
 600 /* In ip_fw_sockopt.c */
 601 void ipfw_init_skipto_cache(struct ip_fw_chain *chain);
 602 void ipfw_destroy_skipto_cache(struct ip_fw_chain *chain);
 603 int ipfw_find_rule(struct ip_fw_chain *chain, uint32_t key, uint32_t id);
 604 int ipfw_ctl3(struct sockopt *sopt);
 605 int ipfw_chk(struct ip_fw_args *args);
 606 void ipfw_reap_add(struct ip_fw_chain *chain, struct ip_fw **head,
 607     struct ip_fw *rule);
 608 void ipfw_reap_rules(struct ip_fw *head);
 609 void ipfw_init_counters(void);
 610 void ipfw_destroy_counters(void);
 611 struct ip_fw *ipfw_alloc_rule(struct ip_fw_chain *chain, size_t rulesize);
 612 int ipfw_match_range(struct ip_fw *rule, ipfw_range_tlv *rt);
 613
 614 typedef int (sopt_handler_f)(struct ip_fw_chain *ch,
 615     ip_fw3_opheader *op3, struct sockopt_data *sd);
 616 struct ipfw_sopt_handler {
 617         uint16_t        opcode;
 618         uint8_t         version;
 619         uint8_t         dir;
 620         sopt_handler_f  *handler;
 621         uint64_t        refcnt;
 622 };
 623 #define HDIR_SET        0x01    /* Handler is used to set some data */
 624 #define HDIR_GET        0x02    /* Handler is used to retrieve data */
 625 #define HDIR_BOTH       HDIR_GET|HDIR_SET
 626
 627 void ipfw_init_sopt_handler(void);
 628 void ipfw_destroy_sopt_handler(void);
 629 void ipfw_add_sopt_handler(struct ipfw_sopt_handler *sh, size_t count);
 630 int ipfw_del_sopt_handler(struct ipfw_sopt_handler *sh, size_t count);
 631 caddr_t ipfw_get_sopt_space(struct sockopt_data *sd, size_t needed);
 632 caddr_t ipfw_get_sopt_header(struct sockopt_data *sd, size_t needed);
 633 #define IPFW_ADD_SOPT_HANDLER(f, c)     do {    \
 634         if ((f) != 0)                           \
 635                 ipfw_add_sopt_handler(c,        \
 636                     sizeof(c) / sizeof(c[0]));  \
 637         } while(0)
 638 #define IPFW_DEL_SOPT_HANDLER(l, c)     do {    \
 639         if ((l) != 0)                           \
 640                 ipfw_del_sopt_handler(c,        \
 641                     sizeof(c) / sizeof(c[0]));  \
 642         } while(0)
 643
 644 struct namedobj_instance;
 645 typedef void (objhash_cb_t)(struct namedobj_instance *ni, struct named_object *,
 646     void *arg);
 647 typedef uint32_t (objhash_hash_f)(struct namedobj_instance *ni, void *key,
 648     uint32_t kopt);
 649 typedef int (objhash_cmp_f)(struct named_object *no, void *key, uint32_t kopt);
 650 struct namedobj_instance *ipfw_objhash_create(uint32_t items);
 651 void ipfw_objhash_destroy(struct namedobj_instance *);
 652 void ipfw_objhash_bitmap_alloc(uint32_t items, void **idx, int *pblocks);
 653 void ipfw_objhash_bitmap_merge(struct namedobj_instance *ni,
 654     void **idx, int *blocks);
 655 void ipfw_objhash_bitmap_swap(struct namedobj_instance *ni,
 656     void **idx, int *blocks);
 657 void ipfw_objhash_bitmap_free(void *idx, int blocks);
 658 void ipfw_objhash_set_hashf(struct namedobj_instance *ni, objhash_hash_f *f);
 659 struct named_object *ipfw_objhash_lookup_name(struct namedobj_instance *ni,
 660     uint32_t set, char *name);
 661 struct named_object *ipfw_objhash_lookup_name_type(struct namedobj_instance *ni,
 662     uint32_t set, uint32_t type, char *name);
 663 struct named_object *ipfw_objhash_lookup_kidx(struct namedobj_instance *ni,
 664     uint16_t idx);
 665 int ipfw_objhash_same_name(struct namedobj_instance *ni, struct named_object *a,
 666     struct named_object *b);
 667 void ipfw_objhash_add(struct namedobj_instance *ni, struct named_object *no);
 668 void ipfw_objhash_del(struct namedobj_instance *ni, struct named_object *no);
 669 uint32_t ipfw_objhash_count(struct namedobj_instance *ni);
 670 void ipfw_objhash_foreach(struct namedobj_instance *ni, objhash_cb_t *f,
 671     void *arg);
 672 int ipfw_objhash_free_idx(struct namedobj_instance *ni, uint16_t idx);
 673 int ipfw_objhash_alloc_idx(void *n, uint16_t *pidx);
 674 void ipfw_objhash_set_funcs(struct namedobj_instance *ni,
 675     objhash_hash_f *hash_f, objhash_cmp_f *cmp_f);
 676 void ipfw_init_obj_rewriter(void);
 677 void ipfw_destroy_obj_rewriter(void);
 678 void ipfw_add_obj_rewriter(struct opcode_obj_rewrite *rw, size_t count);
 679 int ipfw_del_obj_rewriter(struct opcode_obj_rewrite *rw, size_t count);
 680
 681 int ipfw_rewrite_rule_uidx(struct ip_fw_chain *chain,
 682     struct rule_check_info *ci);
 683 int ipfw_mark_object_kidx(struct ip_fw_chain *chain, struct ip_fw *rule,
 684     uint32_t *bmask);
 685 int ref_opcode_object(struct ip_fw_chain *ch, ipfw_insn *cmd, struct tid_info *ti,
 686     struct obj_idx *pidx, int *found, int *unresolved);
 687 void unref_oib_objects(struct ip_fw_chain *ch, ipfw_insn *cmd,
 688     struct obj_idx *oib, struct obj_idx *end);
 689 int create_objects_compat(struct ip_fw_chain *ch, ipfw_insn *cmd,
 690     struct obj_idx *oib, struct obj_idx *pidx, struct tid_info *ti);
 691 void update_opcode_kidx(ipfw_insn *cmd, uint16_t idx);
 692 int classify_opcode_kidx(ipfw_insn *cmd, uint16_t *puidx);
 693 void ipfw_init_srv(struct ip_fw_chain *ch);
 694 void ipfw_destroy_srv(struct ip_fw_chain *ch);
 695
 696 /* In ip_fw_table.c */
 697 struct table_info;
 698
 699 typedef int (table_lookup_t)(struct table_info *ti, void *key, uint32_t keylen,
 700     uint32_t *val);
 701
 702 int ipfw_lookup_table(struct ip_fw_chain *ch, uint16_t tbl, in_addr_t addr,
 703     uint32_t *val);
 704 int ipfw_lookup_table_extended(struct ip_fw_chain *ch, uint16_t tbl, uint16_t plen,
 705     void *paddr, uint32_t *val);
 706 int ipfw_init_tables(struct ip_fw_chain *ch, int first);
 707 int ipfw_resize_tables(struct ip_fw_chain *ch, unsigned int ntables);
 708 int ipfw_switch_tables_namespace(struct ip_fw_chain *ch, unsigned int nsets);
 709 void ipfw_destroy_tables(struct ip_fw_chain *ch, int last);
 710
 711 /* In ip_fw_nat.c -- XXX to be moved to ip_var.h */
 712
 713 extern struct cfg_nat *(*lookup_nat_ptr)(struct nat_list *, int);
 714
 715 typedef int ipfw_nat_t(struct ip_fw_args *, struct cfg_nat *, struct mbuf *);
 716 typedef int ipfw_nat_cfg_t(struct sockopt *);
 717
 718 VNET_DECLARE(int, ipfw_nat_ready);
 719 #define V_ipfw_nat_ready        VNET(ipfw_nat_ready)
 720 #define IPFW_NAT_LOADED (V_ipfw_nat_ready)
 721
 722 extern ipfw_nat_t *ipfw_nat_ptr;
 723 extern ipfw_nat_cfg_t *ipfw_nat_cfg_ptr;
 724 extern ipfw_nat_cfg_t *ipfw_nat_del_ptr;
 725 extern ipfw_nat_cfg_t *ipfw_nat_get_cfg_ptr;
 726 extern ipfw_nat_cfg_t *ipfw_nat_get_log_ptr;
 727
 728 /* Helper functions for IP checksum adjustment */
 729 static __inline uint16_t
 730 cksum_add(uint16_t sum, uint16_t a)
 731 {
 732         uint16_t res;
 733
 734         res = sum + a;
 735         return (res + (res < a));
 736 }
 737
 738 static __inline uint16_t
 739 cksum_adjust(uint16_t oldsum, uint16_t old, uint16_t new)
 740 {
 741
 742         return (~cksum_add(cksum_add(~oldsum, ~old), new));
 743 }
 744
 745 #endif /* _KERNEL */
 746 #endif /* _IPFW2_PRIVATE_H */