13b3a8eb9SGleb Smirnoff /*- 23b3a8eb9SGleb Smirnoff * Copyright (c) 2004 Ruslan Ermilov and Vsevolod Lobko. 33b3a8eb9SGleb Smirnoff * 43b3a8eb9SGleb Smirnoff * Redistribution and use in source and binary forms, with or without 53b3a8eb9SGleb Smirnoff * modification, are permitted provided that the following conditions 63b3a8eb9SGleb Smirnoff * are met: 73b3a8eb9SGleb Smirnoff * 1. Redistributions of source code must retain the above copyright 83b3a8eb9SGleb Smirnoff * notice, this list of conditions and the following disclaimer. 93b3a8eb9SGleb Smirnoff * 2. Redistributions in binary form must reproduce the above copyright 103b3a8eb9SGleb Smirnoff * notice, this list of conditions and the following disclaimer in the 113b3a8eb9SGleb Smirnoff * documentation and/or other materials provided with the distribution. 123b3a8eb9SGleb Smirnoff * 133b3a8eb9SGleb Smirnoff * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 143b3a8eb9SGleb Smirnoff * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 153b3a8eb9SGleb Smirnoff * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 163b3a8eb9SGleb Smirnoff * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 173b3a8eb9SGleb Smirnoff * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 183b3a8eb9SGleb Smirnoff * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 193b3a8eb9SGleb Smirnoff * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 203b3a8eb9SGleb Smirnoff * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 213b3a8eb9SGleb Smirnoff * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 223b3a8eb9SGleb Smirnoff * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 233b3a8eb9SGleb Smirnoff * SUCH DAMAGE. 243b3a8eb9SGleb Smirnoff */ 253b3a8eb9SGleb Smirnoff 263b3a8eb9SGleb Smirnoff #include <sys/cdefs.h> 273b3a8eb9SGleb Smirnoff __FBSDID("$FreeBSD$"); 283b3a8eb9SGleb Smirnoff 293b3a8eb9SGleb Smirnoff /* 30563b5ab1SAlexander V. Chernikov * Lookup table support for ipfw. 313b3a8eb9SGleb Smirnoff * 32ac35ff17SAlexander V. Chernikov * This file contains handlers for all generic tables' operations: 33563b5ab1SAlexander V. Chernikov * add/del/flush entries, list/dump tables etc.. 343b3a8eb9SGleb Smirnoff * 35563b5ab1SAlexander V. Chernikov * Table data modification is protected by both UH and runtimg lock 36563b5ab1SAlexander V. Chernikov * while reading configuration/data is protected by UH lock. 37563b5ab1SAlexander V. Chernikov * 38563b5ab1SAlexander V. Chernikov * Lookup algorithms for all table types are located in ip_fw_table_algo.c 393b3a8eb9SGleb Smirnoff */ 403b3a8eb9SGleb Smirnoff 413b3a8eb9SGleb Smirnoff #include "opt_ipfw.h" 423b3a8eb9SGleb Smirnoff 433b3a8eb9SGleb Smirnoff #include <sys/param.h> 443b3a8eb9SGleb Smirnoff #include <sys/systm.h> 453b3a8eb9SGleb Smirnoff #include <sys/malloc.h> 463b3a8eb9SGleb Smirnoff #include <sys/kernel.h> 473b3a8eb9SGleb Smirnoff #include <sys/lock.h> 483b3a8eb9SGleb Smirnoff #include <sys/rwlock.h> 493b3a8eb9SGleb Smirnoff #include <sys/socket.h> 50f1220db8SAlexander V. Chernikov #include <sys/socketvar.h> 513b3a8eb9SGleb Smirnoff #include <sys/queue.h> 523b3a8eb9SGleb Smirnoff #include <net/if.h> /* ip_fw.h requires IFNAMSIZ */ 533b3a8eb9SGleb Smirnoff 543b3a8eb9SGleb Smirnoff #include <netinet/in.h> 553b3a8eb9SGleb Smirnoff #include <netinet/ip_var.h> /* struct ipfw_rule_ref */ 563b3a8eb9SGleb Smirnoff #include <netinet/ip_fw.h> 573b3a8eb9SGleb Smirnoff 583b3a8eb9SGleb Smirnoff #include <netpfil/ipfw/ip_fw_private.h> 59ea761a5dSAlexander V. Chernikov #include <netpfil/ipfw/ip_fw_table.h> 603b3a8eb9SGleb Smirnoff 613b3a8eb9SGleb Smirnoff /* 62b074b7bbSAlexander V. Chernikov * Table has the following `type` concepts: 63b074b7bbSAlexander V. Chernikov * 649f7d47b0SAlexander V. Chernikov * `no.type` represents lookup key type (cidr, ifp, uid, etc..) 6535e1bbd0SAlexander V. Chernikov * `vtype` represents table value type (currently U32) 669490a627SAlexander V. Chernikov * `ftype` (at the moment )is pure userland field helping to properly 679490a627SAlexander V. Chernikov * format value data e.g. "value is IPv4 nexthop" or "value is DSCP" 689490a627SAlexander V. Chernikov * or "value is port". 69b074b7bbSAlexander V. Chernikov * 70b074b7bbSAlexander V. Chernikov */ 71b074b7bbSAlexander V. Chernikov struct table_config { 72b074b7bbSAlexander V. Chernikov struct named_object no; 73adf3b2b9SAlexander V. Chernikov uint8_t vtype; /* value type */ 74adf3b2b9SAlexander V. Chernikov uint8_t vftype; /* value format type */ 75914bffb6SAlexander V. Chernikov uint8_t tflags; /* type flags */ 764f43138aSAlexander V. Chernikov uint8_t locked; /* 1 if locked from changes */ 77b074b7bbSAlexander V. Chernikov uint32_t count; /* Number of records */ 784c0c07a5SAlexander V. Chernikov uint32_t limit; /* Max number of records */ 79adf3b2b9SAlexander V. Chernikov uint8_t linked; /* 1 if already linked */ 80adf3b2b9SAlexander V. Chernikov uint8_t ochanged; /* used by set swapping */ 81adf3b2b9SAlexander V. Chernikov uint16_t spare1; 82adf3b2b9SAlexander V. Chernikov uint32_t spare2; 83a73d728dSAlexander V. Chernikov uint32_t ocount; /* used by set swapping */ 84a73d728dSAlexander V. Chernikov uint64_t gencnt; /* generation count */ 85b074b7bbSAlexander V. Chernikov char tablename[64]; /* table name */ 869f7d47b0SAlexander V. Chernikov struct table_algo *ta; /* Callbacks for given algo */ 879f7d47b0SAlexander V. Chernikov void *astate; /* algorithm state */ 889f7d47b0SAlexander V. Chernikov struct table_info ti; /* data to put to table_info */ 89b074b7bbSAlexander V. Chernikov }; 90b074b7bbSAlexander V. Chernikov 91b074b7bbSAlexander V. Chernikov struct tables_config { 92b074b7bbSAlexander V. Chernikov struct namedobj_instance *namehash; 939f7d47b0SAlexander V. Chernikov int algo_count; 949f7d47b0SAlexander V. Chernikov struct table_algo *algo[256]; 9557a1cf95SAlexander V. Chernikov struct table_algo *def_algo[IPFW_TABLE_MAXTYPE + 1]; 96b074b7bbSAlexander V. Chernikov }; 97b074b7bbSAlexander V. Chernikov 98b074b7bbSAlexander V. Chernikov static struct table_config *find_table(struct namedobj_instance *ni, 99b074b7bbSAlexander V. Chernikov struct tid_info *ti); 10068394ec8SAlexander V. Chernikov static struct table_config *alloc_table_config(struct ip_fw_chain *ch, 101914bffb6SAlexander V. Chernikov struct tid_info *ti, struct table_algo *ta, char *adata, uint8_t tflags, 102914bffb6SAlexander V. Chernikov uint8_t vtype); 103b074b7bbSAlexander V. Chernikov static void free_table_config(struct namedobj_instance *ni, 104b074b7bbSAlexander V. Chernikov struct table_config *tc); 105db785d31SAlexander V. Chernikov static int create_table_internal(struct ip_fw_chain *ch, struct tid_info *ti, 1060468c5baSAlexander V. Chernikov char *aname, ipfw_xtable_info *i, struct table_config **ptc, 1070468c5baSAlexander V. Chernikov struct table_algo **pta, uint16_t *pkidx, int ref); 1080468c5baSAlexander V. Chernikov static void link_table(struct ip_fw_chain *ch, struct table_config *tc); 1090468c5baSAlexander V. Chernikov static void unlink_table(struct ip_fw_chain *ch, struct table_config *tc); 110b074b7bbSAlexander V. Chernikov static void free_table_state(void **state, void **xstate, uint8_t type); 1112d99a349SAlexander V. Chernikov static int export_tables(struct ip_fw_chain *ch, ipfw_obj_lheader *olh, 1122d99a349SAlexander V. Chernikov struct sockopt_data *sd); 113ac35ff17SAlexander V. Chernikov static void export_table_info(struct ip_fw_chain *ch, struct table_config *tc, 114ac35ff17SAlexander V. Chernikov ipfw_xtable_info *i); 11581d3153dSAlexander V. Chernikov static int dump_table_tentry(void *e, void *arg); 116f1220db8SAlexander V. Chernikov static int dump_table_xentry(void *e, void *arg); 117b074b7bbSAlexander V. Chernikov 1182d99a349SAlexander V. Chernikov static int ipfw_dump_table_v0(struct ip_fw_chain *ch, struct sockopt_data *sd); 1192d99a349SAlexander V. Chernikov static int ipfw_dump_table_v1(struct ip_fw_chain *ch, struct sockopt_data *sd); 120db785d31SAlexander V. Chernikov static int ipfw_manage_table_ent_v0(struct ip_fw_chain *ch, ip_fw3_opheader *op3, 121ac35ff17SAlexander V. Chernikov struct sockopt_data *sd); 122db785d31SAlexander V. Chernikov static int ipfw_manage_table_ent_v1(struct ip_fw_chain *ch, ip_fw3_opheader *op3, 123ac35ff17SAlexander V. Chernikov struct sockopt_data *sd); 124a73d728dSAlexander V. Chernikov static int swap_tables(struct ip_fw_chain *ch, struct tid_info *a, 12546d52008SAlexander V. Chernikov struct tid_info *b); 126ac35ff17SAlexander V. Chernikov 127b6ee846eSAlexander V. Chernikov static int check_table_space(struct ip_fw_chain *ch, struct table_config *tc, 128b6ee846eSAlexander V. Chernikov struct table_info *ti, uint32_t count); 129ac35ff17SAlexander V. Chernikov static int destroy_table(struct ip_fw_chain *ch, struct tid_info *ti); 130d3a4f924SAlexander V. Chernikov 1319f7d47b0SAlexander V. Chernikov static struct table_algo *find_table_algo(struct tables_config *tableconf, 1329490a627SAlexander V. Chernikov struct tid_info *ti, char *name); 133b074b7bbSAlexander V. Chernikov 13446d52008SAlexander V. Chernikov static void objheader_to_ti(struct _ipfw_obj_header *oh, struct tid_info *ti); 13546d52008SAlexander V. Chernikov static void ntlv_to_ti(struct _ipfw_obj_ntlv *ntlv, struct tid_info *ti); 136a73d728dSAlexander V. Chernikov static int classify_table_opcode(ipfw_insn *cmd, uint16_t *puidx, uint8_t *ptype); 13746d52008SAlexander V. Chernikov 138b074b7bbSAlexander V. Chernikov #define CHAIN_TO_TCFG(chain) ((struct tables_config *)(chain)->tblcfg) 139b074b7bbSAlexander V. Chernikov #define CHAIN_TO_NI(chain) (CHAIN_TO_TCFG(chain)->namehash) 1409f7d47b0SAlexander V. Chernikov #define KIDX_TO_TI(ch, k) (&(((struct table_info *)(ch)->tablestate)[k])) 141b074b7bbSAlexander V. Chernikov 142b6ee846eSAlexander V. Chernikov #define TA_BUF_SZ 128 /* On-stack buffer for add/delete state */ 143b6ee846eSAlexander V. Chernikov 1443a845e10SAlexander V. Chernikov /* 1453a845e10SAlexander V. Chernikov * Checks if we're able to insert/update entry @tei into table 1463a845e10SAlexander V. Chernikov * w.r.t @tc limits. 1473a845e10SAlexander V. Chernikov * May alter @tei to indicate insertion error / insert 1483a845e10SAlexander V. Chernikov * options. 1493a845e10SAlexander V. Chernikov * 1503a845e10SAlexander V. Chernikov * Returns 0 if operation can be performed/ 1513a845e10SAlexander V. Chernikov */ 1523a845e10SAlexander V. Chernikov static int 1533a845e10SAlexander V. Chernikov check_table_limit(struct table_config *tc, struct tentry_info *tei) 1543a845e10SAlexander V. Chernikov { 155b074b7bbSAlexander V. Chernikov 1563a845e10SAlexander V. Chernikov if (tc->limit == 0 || tc->count < tc->limit) 1573a845e10SAlexander V. Chernikov return (0); 1583a845e10SAlexander V. Chernikov 1593a845e10SAlexander V. Chernikov if ((tei->flags & TEI_FLAGS_UPDATE) == 0) { 1603a845e10SAlexander V. Chernikov /* Notify userland on error cause */ 1613a845e10SAlexander V. Chernikov tei->flags |= TEI_FLAGS_LIMIT; 1623a845e10SAlexander V. Chernikov return (EFBIG); 1633a845e10SAlexander V. Chernikov } 1643a845e10SAlexander V. Chernikov 1653a845e10SAlexander V. Chernikov /* 1663a845e10SAlexander V. Chernikov * We have UPDATE flag set. 1673a845e10SAlexander V. Chernikov * Permit updating record (if found), 1683a845e10SAlexander V. Chernikov * but restrict adding new one since we've 1693a845e10SAlexander V. Chernikov * already hit the limit. 1703a845e10SAlexander V. Chernikov */ 1713a845e10SAlexander V. Chernikov tei->flags |= TEI_FLAGS_DONTADD; 1723a845e10SAlexander V. Chernikov 1733a845e10SAlexander V. Chernikov return (0); 1743a845e10SAlexander V. Chernikov } 1753a845e10SAlexander V. Chernikov 1763a845e10SAlexander V. Chernikov /* 1771bc0d457SAlexander V. Chernikov * Convert algorithm callback return code into 1781bc0d457SAlexander V. Chernikov * one of pre-defined states known by userland. 1793a845e10SAlexander V. Chernikov */ 1801bc0d457SAlexander V. Chernikov static void 1811bc0d457SAlexander V. Chernikov store_tei_result(struct tentry_info *tei, int do_add, int error, uint32_t num) 1823b3a8eb9SGleb Smirnoff { 1831bc0d457SAlexander V. Chernikov int flag; 1843b3a8eb9SGleb Smirnoff 1851bc0d457SAlexander V. Chernikov flag = 0; 1861bc0d457SAlexander V. Chernikov 1871bc0d457SAlexander V. Chernikov switch (error) { 1881bc0d457SAlexander V. Chernikov case 0: 1891bc0d457SAlexander V. Chernikov if (do_add && num != 0) 1901bc0d457SAlexander V. Chernikov flag = TEI_FLAGS_ADDED; 1911bc0d457SAlexander V. Chernikov if (do_add == 0) 1921bc0d457SAlexander V. Chernikov flag = TEI_FLAGS_DELETED; 1931bc0d457SAlexander V. Chernikov break; 1941bc0d457SAlexander V. Chernikov case ENOENT: 1951bc0d457SAlexander V. Chernikov flag = TEI_FLAGS_NOTFOUND; 1961bc0d457SAlexander V. Chernikov break; 1971bc0d457SAlexander V. Chernikov case EEXIST: 1981bc0d457SAlexander V. Chernikov flag = TEI_FLAGS_EXISTS; 1991bc0d457SAlexander V. Chernikov break; 2001bc0d457SAlexander V. Chernikov default: 2011bc0d457SAlexander V. Chernikov flag = TEI_FLAGS_ERROR; 2021bc0d457SAlexander V. Chernikov } 2031bc0d457SAlexander V. Chernikov 2041bc0d457SAlexander V. Chernikov tei->flags |= flag; 2051bc0d457SAlexander V. Chernikov } 2069f7d47b0SAlexander V. Chernikov 207f99fbf96SAlexander V. Chernikov /* 208f99fbf96SAlexander V. Chernikov * Creates and references table with default parameters. 209f99fbf96SAlexander V. Chernikov * Saves table config, algo and allocated kidx info @ptc, @pta and 210f99fbf96SAlexander V. Chernikov * @pkidx if non-zero. 211f99fbf96SAlexander V. Chernikov * Used for table auto-creation to support old binaries. 212f99fbf96SAlexander V. Chernikov * 213f99fbf96SAlexander V. Chernikov * Returns 0 on success. 214f99fbf96SAlexander V. Chernikov */ 2150468c5baSAlexander V. Chernikov static int 2160468c5baSAlexander V. Chernikov create_table_compat(struct ip_fw_chain *ch, struct tid_info *ti, 2170468c5baSAlexander V. Chernikov struct table_config **ptc, struct table_algo **pta, uint16_t *pkidx) 2180468c5baSAlexander V. Chernikov { 2190468c5baSAlexander V. Chernikov ipfw_xtable_info xi; 2200468c5baSAlexander V. Chernikov int error; 2210468c5baSAlexander V. Chernikov 2220468c5baSAlexander V. Chernikov memset(&xi, 0, sizeof(xi)); 223f99fbf96SAlexander V. Chernikov /* Set u32 as default value type for legacy clients */ 2240468c5baSAlexander V. Chernikov xi.vtype = IPFW_VTYPE_U32; 2250468c5baSAlexander V. Chernikov 2260468c5baSAlexander V. Chernikov error = create_table_internal(ch, ti, NULL, &xi, ptc, pta, pkidx, 1); 2270468c5baSAlexander V. Chernikov if (error != 0) 2280468c5baSAlexander V. Chernikov return (error); 2290468c5baSAlexander V. Chernikov 2300468c5baSAlexander V. Chernikov return (0); 2310468c5baSAlexander V. Chernikov } 2320468c5baSAlexander V. Chernikov 2339f7d47b0SAlexander V. Chernikov /* 2341bc0d457SAlexander V. Chernikov * Find and reference existing table optionally 2351bc0d457SAlexander V. Chernikov * creating new one. 2361bc0d457SAlexander V. Chernikov * 2371bc0d457SAlexander V. Chernikov * Saves found table config/table algo into @ptc / @pta. 2381bc0d457SAlexander V. Chernikov * Returns 0 if table was found/created and referenced 2391bc0d457SAlexander V. Chernikov * or non-zero return code. 2409f7d47b0SAlexander V. Chernikov */ 2411bc0d457SAlexander V. Chernikov static int 2421bc0d457SAlexander V. Chernikov find_ref_table(struct ip_fw_chain *ch, struct tid_info *ti, 2431bc0d457SAlexander V. Chernikov struct tentry_info *tei, uint32_t count, int do_add, 2441bc0d457SAlexander V. Chernikov struct table_config **ptc, struct table_algo **pta) 2451bc0d457SAlexander V. Chernikov { 2461bc0d457SAlexander V. Chernikov struct namedobj_instance *ni; 2471bc0d457SAlexander V. Chernikov struct table_config *tc; 2481bc0d457SAlexander V. Chernikov struct table_algo *ta; 2491bc0d457SAlexander V. Chernikov int error; 2501bc0d457SAlexander V. Chernikov 2511bc0d457SAlexander V. Chernikov IPFW_UH_WLOCK(ch); 2521bc0d457SAlexander V. Chernikov 2531bc0d457SAlexander V. Chernikov ni = CHAIN_TO_NI(ch); 2541bc0d457SAlexander V. Chernikov tc = NULL; 2559f7d47b0SAlexander V. Chernikov ta = NULL; 2569f7d47b0SAlexander V. Chernikov if ((tc = find_table(ni, ti)) != NULL) { 2579f7d47b0SAlexander V. Chernikov /* check table type */ 2589f7d47b0SAlexander V. Chernikov if (tc->no.type != ti->type) { 2599f7d47b0SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 2603b3a8eb9SGleb Smirnoff return (EINVAL); 2613b3a8eb9SGleb Smirnoff } 2623b3a8eb9SGleb Smirnoff 2634f43138aSAlexander V. Chernikov if (tc->locked != 0) { 2644f43138aSAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 2654f43138aSAlexander V. Chernikov return (EACCES); 2664f43138aSAlexander V. Chernikov } 2674f43138aSAlexander V. Chernikov 2684c0c07a5SAlexander V. Chernikov /* Try to exit early on limit hit */ 2691bc0d457SAlexander V. Chernikov if (do_add != 0 && count == 1 && 2701bc0d457SAlexander V. Chernikov check_table_limit(tc, tei) != 0) { 2714c0c07a5SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 2724c0c07a5SAlexander V. Chernikov return (EFBIG); 2734c0c07a5SAlexander V. Chernikov } 2744c0c07a5SAlexander V. Chernikov 2759f7d47b0SAlexander V. Chernikov /* Reference and unlock */ 2769f7d47b0SAlexander V. Chernikov tc->no.refcnt++; 2779f7d47b0SAlexander V. Chernikov ta = tc->ta; 2789f7d47b0SAlexander V. Chernikov } 2799f7d47b0SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 2803b3a8eb9SGleb Smirnoff 281ac35ff17SAlexander V. Chernikov if (tc == NULL) { 2821bc0d457SAlexander V. Chernikov if (do_add == 0) 2831bc0d457SAlexander V. Chernikov return (ESRCH); 2841bc0d457SAlexander V. Chernikov 285db785d31SAlexander V. Chernikov /* Compability mode: create new table for old clients */ 286db785d31SAlexander V. Chernikov if ((tei->flags & TEI_FLAGS_COMPAT) == 0) 287db785d31SAlexander V. Chernikov return (ESRCH); 2883b3a8eb9SGleb Smirnoff 2890468c5baSAlexander V. Chernikov error = create_table_compat(ch, ti, &tc, &ta, NULL); 290db785d31SAlexander V. Chernikov if (error != 0) 291db785d31SAlexander V. Chernikov return (error); 292db785d31SAlexander V. Chernikov 2930468c5baSAlexander V. Chernikov /* OK, now we've got referenced table. */ 294db785d31SAlexander V. Chernikov } 295db785d31SAlexander V. Chernikov 2961bc0d457SAlexander V. Chernikov *ptc = tc; 2971bc0d457SAlexander V. Chernikov *pta = ta; 2981bc0d457SAlexander V. Chernikov return (0); 2991bc0d457SAlexander V. Chernikov } 3001bc0d457SAlexander V. Chernikov 3011bc0d457SAlexander V. Chernikov /* 3021bc0d457SAlexander V. Chernikov * Rolls back already @added to @tc entries using state arrat @ta_buf_m. 3031bc0d457SAlexander V. Chernikov * Assume the following layout: 3041bc0d457SAlexander V. Chernikov * 1) ADD state (ta_buf_m[0] ... t_buf_m[added - 1]) for handling update cases 3051bc0d457SAlexander V. Chernikov * 2) DEL state (ta_buf_m[count[ ... t_buf_m[count + added - 1]) 3061bc0d457SAlexander V. Chernikov * for storing deleted state 3071bc0d457SAlexander V. Chernikov */ 3081bc0d457SAlexander V. Chernikov static void 3091bc0d457SAlexander V. Chernikov rollback_added_entries(struct ip_fw_chain *ch, struct table_config *tc, 3101bc0d457SAlexander V. Chernikov struct table_info *tinfo, struct tentry_info *tei, caddr_t ta_buf_m, 3111bc0d457SAlexander V. Chernikov uint32_t count, uint32_t added) 3121bc0d457SAlexander V. Chernikov { 3131bc0d457SAlexander V. Chernikov struct table_algo *ta; 3141bc0d457SAlexander V. Chernikov struct tentry_info *ptei; 3151bc0d457SAlexander V. Chernikov caddr_t v, vv; 3161bc0d457SAlexander V. Chernikov size_t ta_buf_sz; 3171bc0d457SAlexander V. Chernikov int error, i; 3181bc0d457SAlexander V. Chernikov uint32_t num; 3191bc0d457SAlexander V. Chernikov 3201bc0d457SAlexander V. Chernikov IPFW_UH_WLOCK_ASSERT(ch); 3211bc0d457SAlexander V. Chernikov 3221bc0d457SAlexander V. Chernikov ta = tc->ta; 3233a845e10SAlexander V. Chernikov ta_buf_sz = ta->ta_buf_size; 3241bc0d457SAlexander V. Chernikov v = ta_buf_m; 3251bc0d457SAlexander V. Chernikov vv = v + count * ta_buf_sz; 3261bc0d457SAlexander V. Chernikov for (i = 0; i < added; i++, v += ta_buf_sz, vv += ta_buf_sz) { 3271bc0d457SAlexander V. Chernikov ptei = &tei[i]; 3281bc0d457SAlexander V. Chernikov if ((ptei->flags & TEI_FLAGS_UPDATED) != 0) { 3291bc0d457SAlexander V. Chernikov 3301bc0d457SAlexander V. Chernikov /* 3311bc0d457SAlexander V. Chernikov * We have old value stored by previous 3321bc0d457SAlexander V. Chernikov * call in @ptei->value. Do add once again 3331bc0d457SAlexander V. Chernikov * to restore it. 3341bc0d457SAlexander V. Chernikov */ 3351bc0d457SAlexander V. Chernikov error = ta->add(tc->astate, tinfo, ptei, v, &num); 3361bc0d457SAlexander V. Chernikov KASSERT(error == 0, ("rollback UPDATE fail")); 3371bc0d457SAlexander V. Chernikov KASSERT(num == 0, ("rollback UPDATE fail2")); 3381bc0d457SAlexander V. Chernikov continue; 3391bc0d457SAlexander V. Chernikov } 3401bc0d457SAlexander V. Chernikov 3411bc0d457SAlexander V. Chernikov error = ta->prepare_del(ch, ptei, vv); 3421bc0d457SAlexander V. Chernikov KASSERT(error == 0, ("pre-rollback INSERT failed")); 3431bc0d457SAlexander V. Chernikov error = ta->del(tc->astate, tinfo, ptei, vv, &num); 3441bc0d457SAlexander V. Chernikov KASSERT(error == 0, ("rollback INSERT failed")); 3451bc0d457SAlexander V. Chernikov tc->count -= num; 3461bc0d457SAlexander V. Chernikov } 3471bc0d457SAlexander V. Chernikov } 3481bc0d457SAlexander V. Chernikov 3491bc0d457SAlexander V. Chernikov /* 3501bc0d457SAlexander V. Chernikov * Prepares add/del state for all @count entries in @tei. 3511bc0d457SAlexander V. Chernikov * Uses either stack buffer (@ta_buf) or allocates a new one. 3521bc0d457SAlexander V. Chernikov * Stores pointer to allocated buffer back to @ta_buf. 3531bc0d457SAlexander V. Chernikov * 3541bc0d457SAlexander V. Chernikov * Returns 0 on success. 3551bc0d457SAlexander V. Chernikov */ 3561bc0d457SAlexander V. Chernikov static int 3571bc0d457SAlexander V. Chernikov prepare_batch_buffer(struct ip_fw_chain *ch, struct table_algo *ta, 3581bc0d457SAlexander V. Chernikov struct tentry_info *tei, uint32_t count, int do_add, caddr_t *ta_buf) 3591bc0d457SAlexander V. Chernikov { 3601bc0d457SAlexander V. Chernikov caddr_t ta_buf_m, v; 3611bc0d457SAlexander V. Chernikov size_t ta_buf_sz, sz; 3621bc0d457SAlexander V. Chernikov struct tentry_info *ptei; 3631bc0d457SAlexander V. Chernikov int error, i; 3641bc0d457SAlexander V. Chernikov 3651bc0d457SAlexander V. Chernikov error = 0; 3661bc0d457SAlexander V. Chernikov ta_buf_sz = ta->ta_buf_size; 3673a845e10SAlexander V. Chernikov if (count == 1) { 3681bc0d457SAlexander V. Chernikov /* Sigle add/delete, use on-stack buffer */ 3691bc0d457SAlexander V. Chernikov memset(*ta_buf, 0, TA_BUF_SZ); 3701bc0d457SAlexander V. Chernikov ta_buf_m = *ta_buf; 3713a845e10SAlexander V. Chernikov } else { 3723a845e10SAlexander V. Chernikov 3733a845e10SAlexander V. Chernikov /* 3741bc0d457SAlexander V. Chernikov * Multiple adds/deletes, allocate larger buffer 3751bc0d457SAlexander V. Chernikov * 3761bc0d457SAlexander V. Chernikov * Note we need 2xcount buffer for add case: 3771bc0d457SAlexander V. Chernikov * we have hold both ADD state 3783a845e10SAlexander V. Chernikov * and DELETE state (this may be needed 3793a845e10SAlexander V. Chernikov * if we need to rollback all changes) 3803a845e10SAlexander V. Chernikov */ 3811bc0d457SAlexander V. Chernikov sz = count * ta_buf_sz; 3821bc0d457SAlexander V. Chernikov ta_buf_m = malloc((do_add != 0) ? sz * 2 : sz, M_TEMP, 3833a845e10SAlexander V. Chernikov M_WAITOK | M_ZERO); 3843a845e10SAlexander V. Chernikov } 3851bc0d457SAlexander V. Chernikov 3863a845e10SAlexander V. Chernikov v = ta_buf_m; 3873a845e10SAlexander V. Chernikov for (i = 0; i < count; i++, v += ta_buf_sz) { 3881bc0d457SAlexander V. Chernikov ptei = &tei[i]; 3891bc0d457SAlexander V. Chernikov error = (do_add != 0) ? 3901bc0d457SAlexander V. Chernikov ta->prepare_add(ch, ptei, v) : ta->prepare_del(ch, ptei, v); 3913a845e10SAlexander V. Chernikov 3923a845e10SAlexander V. Chernikov /* 3933a845e10SAlexander V. Chernikov * Some syntax error (incorrect mask, or address, or 3943a845e10SAlexander V. Chernikov * anything). Return error regardless of atomicity 3953a845e10SAlexander V. Chernikov * settings. 3963a845e10SAlexander V. Chernikov */ 397db785d31SAlexander V. Chernikov if (error != 0) 3981bc0d457SAlexander V. Chernikov break; 3993a845e10SAlexander V. Chernikov } 4003b3a8eb9SGleb Smirnoff 4011bc0d457SAlexander V. Chernikov *ta_buf = ta_buf_m; 4021bc0d457SAlexander V. Chernikov return (error); 4031bc0d457SAlexander V. Chernikov } 4041bc0d457SAlexander V. Chernikov 4051bc0d457SAlexander V. Chernikov /* 4061bc0d457SAlexander V. Chernikov * Flushes allocated state for each @count entries in @tei. 4071bc0d457SAlexander V. Chernikov * Frees @ta_buf_m if differs from stack buffer @ta_buf. 4081bc0d457SAlexander V. Chernikov */ 4091bc0d457SAlexander V. Chernikov static void 4101bc0d457SAlexander V. Chernikov flush_batch_buffer(struct ip_fw_chain *ch, struct table_algo *ta, 4111bc0d457SAlexander V. Chernikov struct tentry_info *tei, uint32_t count, int do_add, int rollback, 4121bc0d457SAlexander V. Chernikov caddr_t ta_buf_m, caddr_t ta_buf) 4131bc0d457SAlexander V. Chernikov { 4141bc0d457SAlexander V. Chernikov caddr_t v; 4151bc0d457SAlexander V. Chernikov size_t ta_buf_sz; 4161bc0d457SAlexander V. Chernikov int i; 4171bc0d457SAlexander V. Chernikov 4181bc0d457SAlexander V. Chernikov ta_buf_sz = ta->ta_buf_size; 4191bc0d457SAlexander V. Chernikov 4201bc0d457SAlexander V. Chernikov /* Run cleaning callback anyway */ 4211bc0d457SAlexander V. Chernikov v = ta_buf_m; 4221bc0d457SAlexander V. Chernikov for (i = 0; i < count; i++, v += ta_buf_sz) 4231bc0d457SAlexander V. Chernikov ta->flush_entry(ch, &tei[i], v); 4241bc0d457SAlexander V. Chernikov 4251bc0d457SAlexander V. Chernikov /* Clean up "deleted" state in case of rollback */ 4261bc0d457SAlexander V. Chernikov if (rollback != 0) { 4271bc0d457SAlexander V. Chernikov v = ta_buf_m + count * ta_buf_sz; 4281bc0d457SAlexander V. Chernikov for (i = 0; i < count; i++, v += ta_buf_sz) 4291bc0d457SAlexander V. Chernikov ta->flush_entry(ch, &tei[i], v); 4301bc0d457SAlexander V. Chernikov } 4311bc0d457SAlexander V. Chernikov 4321bc0d457SAlexander V. Chernikov if (ta_buf_m != ta_buf) 4331bc0d457SAlexander V. Chernikov free(ta_buf_m, M_TEMP); 4341bc0d457SAlexander V. Chernikov } 4351bc0d457SAlexander V. Chernikov 4361bc0d457SAlexander V. Chernikov /* 4371bc0d457SAlexander V. Chernikov * Adds/updates one or more entries in table @ti. 4380468c5baSAlexander V. Chernikov * Function references @ti first to ensure table won't 4390468c5baSAlexander V. Chernikov * disappear or change its type. 4400468c5baSAlexander V. Chernikov * After that, prepare_add callback is called for each @tei entry. 4410468c5baSAlexander V. Chernikov * Next, we try to add each entry under UH+WHLOCK 4420468c5baSAlexander V. Chernikov * using add() callback. 4430468c5baSAlexander V. Chernikov * Finally, we free all state by calling flush_entry callback 4440468c5baSAlexander V. Chernikov * for each @tei. 4451bc0d457SAlexander V. Chernikov * 4461bc0d457SAlexander V. Chernikov * Returns 0 on success. 4471bc0d457SAlexander V. Chernikov */ 4481bc0d457SAlexander V. Chernikov int 4491bc0d457SAlexander V. Chernikov add_table_entry(struct ip_fw_chain *ch, struct tid_info *ti, 4501bc0d457SAlexander V. Chernikov struct tentry_info *tei, uint8_t flags, uint32_t count) 4511bc0d457SAlexander V. Chernikov { 4521bc0d457SAlexander V. Chernikov struct table_config *tc; 4531bc0d457SAlexander V. Chernikov struct table_algo *ta; 4541bc0d457SAlexander V. Chernikov uint16_t kidx; 4551bc0d457SAlexander V. Chernikov int error, first_error, i, rollback; 4561bc0d457SAlexander V. Chernikov uint32_t num, numadd; 4571bc0d457SAlexander V. Chernikov struct tentry_info *ptei; 4581bc0d457SAlexander V. Chernikov char ta_buf[TA_BUF_SZ]; 4591bc0d457SAlexander V. Chernikov caddr_t ta_buf_m, v; 4601bc0d457SAlexander V. Chernikov 4611bc0d457SAlexander V. Chernikov /* 4621bc0d457SAlexander V. Chernikov * Find and reference existing table. 4631bc0d457SAlexander V. Chernikov */ 4641bc0d457SAlexander V. Chernikov if ((error = find_ref_table(ch, ti, tei, count, 1, &tc, &ta)) != 0) 4651bc0d457SAlexander V. Chernikov return (error); 4661bc0d457SAlexander V. Chernikov 4671bc0d457SAlexander V. Chernikov /* Allocate memory and prepare record(s) */ 4681bc0d457SAlexander V. Chernikov rollback = 0; 4691bc0d457SAlexander V. Chernikov /* Pass stack buffer by default */ 4701bc0d457SAlexander V. Chernikov ta_buf_m = ta_buf; 4711bc0d457SAlexander V. Chernikov error = prepare_batch_buffer(ch, ta, tei, count, 1, &ta_buf_m); 4721bc0d457SAlexander V. Chernikov if (error != 0) 4731bc0d457SAlexander V. Chernikov goto cleanup; 4741bc0d457SAlexander V. Chernikov 475b074b7bbSAlexander V. Chernikov IPFW_UH_WLOCK(ch); 4763b3a8eb9SGleb Smirnoff 477b6ee846eSAlexander V. Chernikov /* 478b6ee846eSAlexander V. Chernikov * Ensure we are able to add all entries without additional 479b6ee846eSAlexander V. Chernikov * memory allocations. May release/reacquire UH_WLOCK. 480b6ee846eSAlexander V. Chernikov */ 481b6ee846eSAlexander V. Chernikov kidx = tc->no.kidx; 482b6ee846eSAlexander V. Chernikov error = check_table_space(ch, tc, KIDX_TO_TI(ch, kidx), count); 483b6ee846eSAlexander V. Chernikov if (error != 0) { 484b6ee846eSAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 4853a845e10SAlexander V. Chernikov goto cleanup; 486b6ee846eSAlexander V. Chernikov } 487b6ee846eSAlexander V. Chernikov 4889f7d47b0SAlexander V. Chernikov /* Drop reference we've used in first search */ 4899f7d47b0SAlexander V. Chernikov tc->no.refcnt--; 4903a845e10SAlexander V. Chernikov /* We've got valid table in @tc. Let's try to add data */ 4919f7d47b0SAlexander V. Chernikov kidx = tc->no.kidx; 4929f7d47b0SAlexander V. Chernikov ta = tc->ta; 4933a845e10SAlexander V. Chernikov numadd = 0; 4943a845e10SAlexander V. Chernikov first_error = 0; 4959f7d47b0SAlexander V. Chernikov 496b074b7bbSAlexander V. Chernikov IPFW_WLOCK(ch); 4973a845e10SAlexander V. Chernikov 4983a845e10SAlexander V. Chernikov v = ta_buf_m; 4991bc0d457SAlexander V. Chernikov for (i = 0; i < count; i++, v += ta->ta_buf_size) { 5003a845e10SAlexander V. Chernikov ptei = &tei[i]; 5013a845e10SAlexander V. Chernikov num = 0; 5023a845e10SAlexander V. Chernikov /* check limit before adding */ 5033a845e10SAlexander V. Chernikov if ((error = check_table_limit(tc, ptei)) == 0) { 5043a845e10SAlexander V. Chernikov error = ta->add(tc->astate, KIDX_TO_TI(ch, kidx), 5053a845e10SAlexander V. Chernikov ptei, v, &num); 5063a845e10SAlexander V. Chernikov /* Set status flag to inform userland */ 5071bc0d457SAlexander V. Chernikov store_tei_result(ptei, 1, error, num); 5083a845e10SAlexander V. Chernikov } 5093a845e10SAlexander V. Chernikov if (error == 0) { 5103a845e10SAlexander V. Chernikov /* Update number of records to ease limit checking */ 5113a845e10SAlexander V. Chernikov tc->count += num; 5123a845e10SAlexander V. Chernikov numadd += num; 5133a845e10SAlexander V. Chernikov continue; 5143a845e10SAlexander V. Chernikov } 5153a845e10SAlexander V. Chernikov 5163a845e10SAlexander V. Chernikov if (first_error == 0) 5173a845e10SAlexander V. Chernikov first_error = error; 5183a845e10SAlexander V. Chernikov 5193a845e10SAlexander V. Chernikov /* 5203a845e10SAlexander V. Chernikov * Some error have happened. Check our atomicity 5213a845e10SAlexander V. Chernikov * settings: continue if atomicity is not required, 5223a845e10SAlexander V. Chernikov * rollback changes otherwise. 5233a845e10SAlexander V. Chernikov */ 5243a845e10SAlexander V. Chernikov if ((flags & IPFW_CTF_ATOMIC) == 0) 5253a845e10SAlexander V. Chernikov continue; 5263a845e10SAlexander V. Chernikov 5271bc0d457SAlexander V. Chernikov rollback_added_entries(ch, tc, KIDX_TO_TI(ch, kidx), 5281bc0d457SAlexander V. Chernikov tei, ta_buf_m, count, i); 5293a845e10SAlexander V. Chernikov break; 5303a845e10SAlexander V. Chernikov } 5313a845e10SAlexander V. Chernikov 5323b3a8eb9SGleb Smirnoff IPFW_WUNLOCK(ch); 5339f7d47b0SAlexander V. Chernikov 534b6ee846eSAlexander V. Chernikov /* Permit post-add algorithm grow/rehash. */ 5353a845e10SAlexander V. Chernikov if (numadd != 0) 5363a845e10SAlexander V. Chernikov check_table_space(ch, tc, KIDX_TO_TI(ch, kidx), 0); 537db785d31SAlexander V. Chernikov 538b074b7bbSAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 539b074b7bbSAlexander V. Chernikov 5403a845e10SAlexander V. Chernikov /* Return first error to user, if any */ 5413a845e10SAlexander V. Chernikov error = first_error; 5423a845e10SAlexander V. Chernikov 5433a845e10SAlexander V. Chernikov cleanup: 5441bc0d457SAlexander V. Chernikov flush_batch_buffer(ch, ta, tei, count, 1, rollback, ta_buf_m, ta_buf); 545b074b7bbSAlexander V. Chernikov 5469f7d47b0SAlexander V. Chernikov return (error); 5473b3a8eb9SGleb Smirnoff } 5483b3a8eb9SGleb Smirnoff 5493a845e10SAlexander V. Chernikov /* 5503a845e10SAlexander V. Chernikov * Deletes one or more entries in table @ti. 5513a845e10SAlexander V. Chernikov * 5523a845e10SAlexander V. Chernikov * Returns 0 on success. 5533a845e10SAlexander V. Chernikov */ 5543b3a8eb9SGleb Smirnoff int 5551832a7b3SAlexander V. Chernikov del_table_entry(struct ip_fw_chain *ch, struct tid_info *ti, 5563a845e10SAlexander V. Chernikov struct tentry_info *tei, uint8_t flags, uint32_t count) 5573b3a8eb9SGleb Smirnoff { 558b074b7bbSAlexander V. Chernikov struct table_config *tc; 5599f7d47b0SAlexander V. Chernikov struct table_algo *ta; 5603a845e10SAlexander V. Chernikov struct tentry_info *ptei; 561b074b7bbSAlexander V. Chernikov uint16_t kidx; 5623a845e10SAlexander V. Chernikov int error, first_error, i; 5633a845e10SAlexander V. Chernikov uint32_t num, numdel; 564b6ee846eSAlexander V. Chernikov char ta_buf[TA_BUF_SZ]; 5653a845e10SAlexander V. Chernikov caddr_t ta_buf_m, v; 5663b3a8eb9SGleb Smirnoff 567db785d31SAlexander V. Chernikov /* 5681bc0d457SAlexander V. Chernikov * Find and reference existing table. 569db785d31SAlexander V. Chernikov */ 5701bc0d457SAlexander V. Chernikov if ((error = find_ref_table(ch, ti, tei, count, 0, &tc, &ta)) != 0) 571db785d31SAlexander V. Chernikov return (error); 5723a845e10SAlexander V. Chernikov 57335e1bbd0SAlexander V. Chernikov /* Allocate memory and prepare record(s) */ 5741bc0d457SAlexander V. Chernikov /* Pass stack buffer by default */ 5753a845e10SAlexander V. Chernikov ta_buf_m = ta_buf; 5761bc0d457SAlexander V. Chernikov error = prepare_batch_buffer(ch, ta, tei, count, 0, &ta_buf_m); 5773a845e10SAlexander V. Chernikov if (error != 0) 5783a845e10SAlexander V. Chernikov goto cleanup; 5799f7d47b0SAlexander V. Chernikov 5803a845e10SAlexander V. Chernikov IPFW_UH_WLOCK(ch); 5813a845e10SAlexander V. Chernikov 5823a845e10SAlexander V. Chernikov /* Drop reference we've used in first search */ 5833a845e10SAlexander V. Chernikov tc->no.refcnt--; 5843a845e10SAlexander V. Chernikov 585b074b7bbSAlexander V. Chernikov kidx = tc->no.kidx; 5863a845e10SAlexander V. Chernikov numdel = 0; 5873a845e10SAlexander V. Chernikov first_error = 0; 588b074b7bbSAlexander V. Chernikov 589b074b7bbSAlexander V. Chernikov IPFW_WLOCK(ch); 5903a845e10SAlexander V. Chernikov v = ta_buf_m; 5911bc0d457SAlexander V. Chernikov for (i = 0; i < count; i++, v += ta->ta_buf_size) { 5923a845e10SAlexander V. Chernikov ptei = &tei[i]; 5933a845e10SAlexander V. Chernikov num = 0; 5943a845e10SAlexander V. Chernikov error = ta->del(tc->astate, KIDX_TO_TI(ch, kidx), ptei, v, 5953a845e10SAlexander V. Chernikov &num); 5963a845e10SAlexander V. Chernikov /* Save state for userland */ 5971bc0d457SAlexander V. Chernikov store_tei_result(ptei, 0, error, num); 5983a845e10SAlexander V. Chernikov if (error != 0 && first_error == 0) 5993a845e10SAlexander V. Chernikov first_error = error; 6003a845e10SAlexander V. Chernikov tc->count -= num; 6013a845e10SAlexander V. Chernikov numdel += num; 6023a845e10SAlexander V. Chernikov } 6033b3a8eb9SGleb Smirnoff IPFW_WUNLOCK(ch); 6043b3a8eb9SGleb Smirnoff 6053a845e10SAlexander V. Chernikov if (numdel != 0) { 606b6ee846eSAlexander V. Chernikov /* Run post-del hook to permit shrinking */ 607b6ee846eSAlexander V. Chernikov error = check_table_space(ch, tc, KIDX_TO_TI(ch, kidx), 0); 608b6ee846eSAlexander V. Chernikov } 609b074b7bbSAlexander V. Chernikov 6109f7d47b0SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 6113b3a8eb9SGleb Smirnoff 6123a845e10SAlexander V. Chernikov /* Return first error to user, if any */ 6133a845e10SAlexander V. Chernikov error = first_error; 6143a845e10SAlexander V. Chernikov 6153a845e10SAlexander V. Chernikov cleanup: 6161bc0d457SAlexander V. Chernikov flush_batch_buffer(ch, ta, tei, count, 0, 0, ta_buf_m, ta_buf); 617ac35ff17SAlexander V. Chernikov 618ac35ff17SAlexander V. Chernikov return (error); 619ac35ff17SAlexander V. Chernikov } 620ac35ff17SAlexander V. Chernikov 621db785d31SAlexander V. Chernikov /* 622b6ee846eSAlexander V. Chernikov * Ensure that table @tc has enough space to add @count entries without 623b6ee846eSAlexander V. Chernikov * need for reallocation. 624db785d31SAlexander V. Chernikov * 625db785d31SAlexander V. Chernikov * Callbacks order: 626301290bcSAlexander V. Chernikov * 0) need_modify() (UH_WLOCK) - checks if @count items can be added w/o resize. 627b6ee846eSAlexander V. Chernikov * 628db785d31SAlexander V. Chernikov * 1) alloc_modify (no locks, M_WAITOK) - alloc new state based on @pflags. 629db785d31SAlexander V. Chernikov * 2) prepare_modifyt (UH_WLOCK) - copy old data into new storage 630db785d31SAlexander V. Chernikov * 3) modify (UH_WLOCK + WLOCK) - switch pointers 631b6ee846eSAlexander V. Chernikov * 4) flush_modify (UH_WLOCK) - free state, if needed 632b6ee846eSAlexander V. Chernikov * 633b6ee846eSAlexander V. Chernikov * Returns 0 on success. 634db785d31SAlexander V. Chernikov */ 635db785d31SAlexander V. Chernikov static int 636b6ee846eSAlexander V. Chernikov check_table_space(struct ip_fw_chain *ch, struct table_config *tc, 637b6ee846eSAlexander V. Chernikov struct table_info *ti, uint32_t count) 638db785d31SAlexander V. Chernikov { 639b6ee846eSAlexander V. Chernikov struct table_algo *ta; 640b6ee846eSAlexander V. Chernikov uint64_t pflags; 641b6ee846eSAlexander V. Chernikov char ta_buf[TA_BUF_SZ]; 642db785d31SAlexander V. Chernikov int error; 643db785d31SAlexander V. Chernikov 644b6ee846eSAlexander V. Chernikov IPFW_UH_WLOCK_ASSERT(ch); 645db785d31SAlexander V. Chernikov 646b6ee846eSAlexander V. Chernikov error = 0; 647b6ee846eSAlexander V. Chernikov ta = tc->ta; 648b6ee846eSAlexander V. Chernikov /* Acquire reference not to loose @tc between locks/unlocks */ 649b6ee846eSAlexander V. Chernikov tc->no.refcnt++; 650db785d31SAlexander V. Chernikov 651db785d31SAlexander V. Chernikov /* 652b6ee846eSAlexander V. Chernikov * TODO: think about avoiding race between large add/large delete 653b6ee846eSAlexander V. Chernikov * operation on algorithm which implements shrinking along with 654b6ee846eSAlexander V. Chernikov * growing. 655db785d31SAlexander V. Chernikov */ 656b6ee846eSAlexander V. Chernikov while (true) { 657b6ee846eSAlexander V. Chernikov pflags = 0; 658301290bcSAlexander V. Chernikov if (ta->need_modify(tc->astate, ti, count, &pflags) == 0) { 65935e1bbd0SAlexander V. Chernikov error = 0; 66035e1bbd0SAlexander V. Chernikov break; 661b6ee846eSAlexander V. Chernikov } 662db785d31SAlexander V. Chernikov 663b6ee846eSAlexander V. Chernikov /* We have to shrink/grow table */ 664b6ee846eSAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 665b6ee846eSAlexander V. Chernikov 666301290bcSAlexander V. Chernikov memset(&ta_buf, 0, sizeof(ta_buf)); 667b6ee846eSAlexander V. Chernikov if ((error = ta->prepare_mod(ta_buf, &pflags)) != 0) { 668b6ee846eSAlexander V. Chernikov IPFW_UH_WLOCK(ch); 669b6ee846eSAlexander V. Chernikov break; 670b6ee846eSAlexander V. Chernikov } 671b6ee846eSAlexander V. Chernikov 672b6ee846eSAlexander V. Chernikov IPFW_UH_WLOCK(ch); 673b6ee846eSAlexander V. Chernikov 674b6ee846eSAlexander V. Chernikov /* Check if we still need to alter table */ 675b6ee846eSAlexander V. Chernikov ti = KIDX_TO_TI(ch, tc->no.kidx); 676301290bcSAlexander V. Chernikov if (ta->need_modify(tc->astate, ti, count, &pflags) == 0) { 677301290bcSAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 678b6ee846eSAlexander V. Chernikov 679b6ee846eSAlexander V. Chernikov /* 68035e1bbd0SAlexander V. Chernikov * Other thread has already performed resize. 68135e1bbd0SAlexander V. Chernikov * Flush our state and return. 682b6ee846eSAlexander V. Chernikov */ 683b6ee846eSAlexander V. Chernikov ta->flush_mod(ta_buf); 684b6ee846eSAlexander V. Chernikov break; 685b6ee846eSAlexander V. Chernikov } 686b6ee846eSAlexander V. Chernikov 687b6ee846eSAlexander V. Chernikov error = ta->fill_mod(tc->astate, ti, ta_buf, &pflags); 688b6ee846eSAlexander V. Chernikov if (error == 0) { 689db785d31SAlexander V. Chernikov /* Do actual modification */ 690db785d31SAlexander V. Chernikov IPFW_WLOCK(ch); 69168394ec8SAlexander V. Chernikov ta->modify(tc->astate, ti, ta_buf, pflags); 692db785d31SAlexander V. Chernikov IPFW_WUNLOCK(ch); 693db785d31SAlexander V. Chernikov } 694db785d31SAlexander V. Chernikov 695b6ee846eSAlexander V. Chernikov /* Anyway, flush data and retry */ 696db785d31SAlexander V. Chernikov ta->flush_mod(ta_buf); 697b6ee846eSAlexander V. Chernikov } 698db785d31SAlexander V. Chernikov 699b6ee846eSAlexander V. Chernikov tc->no.refcnt--; 700db785d31SAlexander V. Chernikov return (error); 701db785d31SAlexander V. Chernikov } 702db785d31SAlexander V. Chernikov 7033a845e10SAlexander V. Chernikov /* 7043a845e10SAlexander V. Chernikov * Selects appropriate table operation handler 7053a845e10SAlexander V. Chernikov * depending on opcode version. 7063a845e10SAlexander V. Chernikov */ 707ac35ff17SAlexander V. Chernikov int 708db785d31SAlexander V. Chernikov ipfw_manage_table_ent(struct ip_fw_chain *ch, ip_fw3_opheader *op3, 709ac35ff17SAlexander V. Chernikov struct sockopt_data *sd) 710ac35ff17SAlexander V. Chernikov { 711ac35ff17SAlexander V. Chernikov int error; 712ac35ff17SAlexander V. Chernikov 713ac35ff17SAlexander V. Chernikov switch (op3->version) { 714ac35ff17SAlexander V. Chernikov case 0: 715db785d31SAlexander V. Chernikov error = ipfw_manage_table_ent_v0(ch, op3, sd); 716ac35ff17SAlexander V. Chernikov break; 717ac35ff17SAlexander V. Chernikov case 1: 718db785d31SAlexander V. Chernikov error = ipfw_manage_table_ent_v1(ch, op3, sd); 719ac35ff17SAlexander V. Chernikov break; 720ac35ff17SAlexander V. Chernikov default: 721ac35ff17SAlexander V. Chernikov error = ENOTSUP; 722ac35ff17SAlexander V. Chernikov } 723ac35ff17SAlexander V. Chernikov 724ac35ff17SAlexander V. Chernikov return (error); 725ac35ff17SAlexander V. Chernikov } 726ac35ff17SAlexander V. Chernikov 727ac35ff17SAlexander V. Chernikov /* 728ac35ff17SAlexander V. Chernikov * Adds or deletes record in table. 729ac35ff17SAlexander V. Chernikov * Data layout (v0): 730ac35ff17SAlexander V. Chernikov * Request: [ ip_fw3_opheader ipfw_table_xentry ] 731ac35ff17SAlexander V. Chernikov * 732ac35ff17SAlexander V. Chernikov * Returns 0 on success 733ac35ff17SAlexander V. Chernikov */ 734ac35ff17SAlexander V. Chernikov static int 735db785d31SAlexander V. Chernikov ipfw_manage_table_ent_v0(struct ip_fw_chain *ch, ip_fw3_opheader *op3, 736ac35ff17SAlexander V. Chernikov struct sockopt_data *sd) 737ac35ff17SAlexander V. Chernikov { 738ac35ff17SAlexander V. Chernikov ipfw_table_xentry *xent; 739ac35ff17SAlexander V. Chernikov struct tentry_info tei; 740ac35ff17SAlexander V. Chernikov struct tid_info ti; 741ac35ff17SAlexander V. Chernikov int error, hdrlen, read; 742ac35ff17SAlexander V. Chernikov 743ac35ff17SAlexander V. Chernikov hdrlen = offsetof(ipfw_table_xentry, k); 744ac35ff17SAlexander V. Chernikov 745ac35ff17SAlexander V. Chernikov /* Check minimum header size */ 746ac35ff17SAlexander V. Chernikov if (sd->valsize < (sizeof(*op3) + hdrlen)) 747ac35ff17SAlexander V. Chernikov return (EINVAL); 748ac35ff17SAlexander V. Chernikov 749ac35ff17SAlexander V. Chernikov read = sizeof(ip_fw3_opheader); 750ac35ff17SAlexander V. Chernikov 751ac35ff17SAlexander V. Chernikov /* Check if xentry len field is valid */ 752ac35ff17SAlexander V. Chernikov xent = (ipfw_table_xentry *)(op3 + 1); 753ac35ff17SAlexander V. Chernikov if (xent->len < hdrlen || xent->len + read > sd->valsize) 754ac35ff17SAlexander V. Chernikov return (EINVAL); 755ac35ff17SAlexander V. Chernikov 756ac35ff17SAlexander V. Chernikov memset(&tei, 0, sizeof(tei)); 757ac35ff17SAlexander V. Chernikov tei.paddr = &xent->k; 758ac35ff17SAlexander V. Chernikov tei.masklen = xent->masklen; 759ac35ff17SAlexander V. Chernikov tei.value = xent->value; 760ac35ff17SAlexander V. Chernikov /* Old requests compability */ 761db785d31SAlexander V. Chernikov tei.flags = TEI_FLAGS_COMPAT; 762ac35ff17SAlexander V. Chernikov if (xent->type == IPFW_TABLE_CIDR) { 763ac35ff17SAlexander V. Chernikov if (xent->len - hdrlen == sizeof(in_addr_t)) 764ac35ff17SAlexander V. Chernikov tei.subtype = AF_INET; 765ac35ff17SAlexander V. Chernikov else 766ac35ff17SAlexander V. Chernikov tei.subtype = AF_INET6; 767ac35ff17SAlexander V. Chernikov } 768ac35ff17SAlexander V. Chernikov 769ac35ff17SAlexander V. Chernikov memset(&ti, 0, sizeof(ti)); 770ac35ff17SAlexander V. Chernikov ti.uidx = xent->tbl; 771ac35ff17SAlexander V. Chernikov ti.type = xent->type; 772ac35ff17SAlexander V. Chernikov 773ac35ff17SAlexander V. Chernikov error = (op3->opcode == IP_FW_TABLE_XADD) ? 7743a845e10SAlexander V. Chernikov add_table_entry(ch, &ti, &tei, 0, 1) : 7753a845e10SAlexander V. Chernikov del_table_entry(ch, &ti, &tei, 0, 1); 776ac35ff17SAlexander V. Chernikov 777ac35ff17SAlexander V. Chernikov return (error); 778ac35ff17SAlexander V. Chernikov } 779ac35ff17SAlexander V. Chernikov 780ac35ff17SAlexander V. Chernikov /* 781ac35ff17SAlexander V. Chernikov * Adds or deletes record in table. 782ac35ff17SAlexander V. Chernikov * Data layout (v1)(current): 783db785d31SAlexander V. Chernikov * Request: [ ipfw_obj_header 784db785d31SAlexander V. Chernikov * ipfw_obj_ctlv(IPFW_TLV_TBLENT_LIST) [ ipfw_obj_tentry x N ] 785db785d31SAlexander V. Chernikov * ] 786ac35ff17SAlexander V. Chernikov * 787ac35ff17SAlexander V. Chernikov * Returns 0 on success 788ac35ff17SAlexander V. Chernikov */ 789ac35ff17SAlexander V. Chernikov static int 790db785d31SAlexander V. Chernikov ipfw_manage_table_ent_v1(struct ip_fw_chain *ch, ip_fw3_opheader *op3, 791ac35ff17SAlexander V. Chernikov struct sockopt_data *sd) 792ac35ff17SAlexander V. Chernikov { 7933a845e10SAlexander V. Chernikov ipfw_obj_tentry *tent, *ptent; 794db785d31SAlexander V. Chernikov ipfw_obj_ctlv *ctlv; 795ac35ff17SAlexander V. Chernikov ipfw_obj_header *oh; 7963a845e10SAlexander V. Chernikov struct tentry_info *ptei, tei, *tei_buf; 797ac35ff17SAlexander V. Chernikov struct tid_info ti; 7983a845e10SAlexander V. Chernikov int error, i, kidx, read; 799ac35ff17SAlexander V. Chernikov 800ac35ff17SAlexander V. Chernikov /* Check minimum header size */ 801db785d31SAlexander V. Chernikov if (sd->valsize < (sizeof(*oh) + sizeof(*ctlv))) 802ac35ff17SAlexander V. Chernikov return (EINVAL); 803ac35ff17SAlexander V. Chernikov 804ac35ff17SAlexander V. Chernikov /* Check if passed data is too long */ 805ac35ff17SAlexander V. Chernikov if (sd->valsize != sd->kavail) 806ac35ff17SAlexander V. Chernikov return (EINVAL); 807ac35ff17SAlexander V. Chernikov 808ac35ff17SAlexander V. Chernikov oh = (ipfw_obj_header *)sd->kbuf; 809ac35ff17SAlexander V. Chernikov 810ac35ff17SAlexander V. Chernikov /* Basic length checks for TLVs */ 811ac35ff17SAlexander V. Chernikov if (oh->ntlv.head.length != sizeof(oh->ntlv)) 812ac35ff17SAlexander V. Chernikov return (EINVAL); 813ac35ff17SAlexander V. Chernikov 814ac35ff17SAlexander V. Chernikov read = sizeof(*oh); 815ac35ff17SAlexander V. Chernikov 816db785d31SAlexander V. Chernikov ctlv = (ipfw_obj_ctlv *)(oh + 1); 817db785d31SAlexander V. Chernikov if (ctlv->head.length + read != sd->valsize) 818db785d31SAlexander V. Chernikov return (EINVAL); 819db785d31SAlexander V. Chernikov 820db785d31SAlexander V. Chernikov read += sizeof(*ctlv); 821db785d31SAlexander V. Chernikov tent = (ipfw_obj_tentry *)(ctlv + 1); 8223a845e10SAlexander V. Chernikov if (ctlv->count * sizeof(*tent) + read != sd->valsize) 823ac35ff17SAlexander V. Chernikov return (EINVAL); 824ac35ff17SAlexander V. Chernikov 8253a845e10SAlexander V. Chernikov if (ctlv->count == 0) 8263a845e10SAlexander V. Chernikov return (0); 827ac35ff17SAlexander V. Chernikov 8283a845e10SAlexander V. Chernikov /* 8293a845e10SAlexander V. Chernikov * Mark entire buffer as "read". 83035e1bbd0SAlexander V. Chernikov * This instructs sopt api write it back 8313a845e10SAlexander V. Chernikov * after function return. 8323a845e10SAlexander V. Chernikov */ 8333a845e10SAlexander V. Chernikov ipfw_get_sopt_header(sd, sd->valsize); 8343a845e10SAlexander V. Chernikov 8353a845e10SAlexander V. Chernikov /* Perform basic checks for each entry */ 8363a845e10SAlexander V. Chernikov ptent = tent; 8373a845e10SAlexander V. Chernikov kidx = tent->idx; 8383a845e10SAlexander V. Chernikov for (i = 0; i < ctlv->count; i++, ptent++) { 8393a845e10SAlexander V. Chernikov if (ptent->head.length != sizeof(*ptent)) 8403a845e10SAlexander V. Chernikov return (EINVAL); 8413a845e10SAlexander V. Chernikov if (ptent->idx != kidx) 8423a845e10SAlexander V. Chernikov return (ENOTSUP); 8433a845e10SAlexander V. Chernikov } 8443a845e10SAlexander V. Chernikov 8453a845e10SAlexander V. Chernikov /* Convert data into kernel request objects */ 84681d3153dSAlexander V. Chernikov objheader_to_ti(oh, &ti); 847ac35ff17SAlexander V. Chernikov ti.type = oh->ntlv.type; 8483a845e10SAlexander V. Chernikov ti.uidx = kidx; 8493a845e10SAlexander V. Chernikov 8503a845e10SAlexander V. Chernikov /* Use on-stack buffer for single add/del */ 8513a845e10SAlexander V. Chernikov if (ctlv->count == 1) { 8523a845e10SAlexander V. Chernikov memset(&tei, 0, sizeof(tei)); 8533a845e10SAlexander V. Chernikov tei_buf = &tei; 8543a845e10SAlexander V. Chernikov } else 8553a845e10SAlexander V. Chernikov tei_buf = malloc(ctlv->count * sizeof(tei), M_TEMP, 8563a845e10SAlexander V. Chernikov M_WAITOK | M_ZERO); 8573a845e10SAlexander V. Chernikov 8583a845e10SAlexander V. Chernikov ptei = tei_buf; 8593a845e10SAlexander V. Chernikov ptent = tent; 8603a845e10SAlexander V. Chernikov for (i = 0; i < ctlv->count; i++, ptent++, ptei++) { 8613a845e10SAlexander V. Chernikov ptei->paddr = &ptent->k; 8623a845e10SAlexander V. Chernikov ptei->subtype = ptent->subtype; 8633a845e10SAlexander V. Chernikov ptei->masklen = ptent->masklen; 8643a845e10SAlexander V. Chernikov if (ptent->head.flags & IPFW_TF_UPDATE) 8653a845e10SAlexander V. Chernikov ptei->flags |= TEI_FLAGS_UPDATE; 8663a845e10SAlexander V. Chernikov ptei->value = ptent->value; 8673a845e10SAlexander V. Chernikov } 868ac35ff17SAlexander V. Chernikov 869ac35ff17SAlexander V. Chernikov error = (oh->opheader.opcode == IP_FW_TABLE_XADD) ? 8703a845e10SAlexander V. Chernikov add_table_entry(ch, &ti, tei_buf, ctlv->flags, ctlv->count) : 8713a845e10SAlexander V. Chernikov del_table_entry(ch, &ti, tei_buf, ctlv->flags, ctlv->count); 8723a845e10SAlexander V. Chernikov 8733a845e10SAlexander V. Chernikov /* Translate result back to userland */ 8743a845e10SAlexander V. Chernikov ptei = tei_buf; 8753a845e10SAlexander V. Chernikov ptent = tent; 8763a845e10SAlexander V. Chernikov for (i = 0; i < ctlv->count; i++, ptent++, ptei++) { 8773a845e10SAlexander V. Chernikov if (ptei->flags & TEI_FLAGS_ADDED) 8783a845e10SAlexander V. Chernikov ptent->result = IPFW_TR_ADDED; 8793a845e10SAlexander V. Chernikov else if (ptei->flags & TEI_FLAGS_DELETED) 8803a845e10SAlexander V. Chernikov ptent->result = IPFW_TR_DELETED; 8813a845e10SAlexander V. Chernikov else if (ptei->flags & TEI_FLAGS_UPDATED) 8823a845e10SAlexander V. Chernikov ptent->result = IPFW_TR_UPDATED; 8833a845e10SAlexander V. Chernikov else if (ptei->flags & TEI_FLAGS_LIMIT) 8843a845e10SAlexander V. Chernikov ptent->result = IPFW_TR_LIMIT; 8853a845e10SAlexander V. Chernikov else if (ptei->flags & TEI_FLAGS_ERROR) 8863a845e10SAlexander V. Chernikov ptent->result = IPFW_TR_ERROR; 8873a845e10SAlexander V. Chernikov else if (ptei->flags & TEI_FLAGS_NOTFOUND) 8883a845e10SAlexander V. Chernikov ptent->result = IPFW_TR_NOTFOUND; 8893a845e10SAlexander V. Chernikov else if (ptei->flags & TEI_FLAGS_EXISTS) 8903a845e10SAlexander V. Chernikov ptent->result = IPFW_TR_EXISTS; 8913a845e10SAlexander V. Chernikov } 8923a845e10SAlexander V. Chernikov 8933a845e10SAlexander V. Chernikov if (tei_buf != &tei) 8943a845e10SAlexander V. Chernikov free(tei_buf, M_TEMP); 895ac35ff17SAlexander V. Chernikov 896ac35ff17SAlexander V. Chernikov return (error); 897ac35ff17SAlexander V. Chernikov } 898ac35ff17SAlexander V. Chernikov 89981d3153dSAlexander V. Chernikov /* 90081d3153dSAlexander V. Chernikov * Looks up an entry in given table. 90181d3153dSAlexander V. Chernikov * Data layout (v0)(current): 90281d3153dSAlexander V. Chernikov * Request: [ ipfw_obj_header ipfw_obj_tentry ] 90381d3153dSAlexander V. Chernikov * Reply: [ ipfw_obj_header ipfw_obj_tentry ] 90481d3153dSAlexander V. Chernikov * 90581d3153dSAlexander V. Chernikov * Returns 0 on success 90681d3153dSAlexander V. Chernikov */ 90781d3153dSAlexander V. Chernikov int 90881d3153dSAlexander V. Chernikov ipfw_find_table_entry(struct ip_fw_chain *ch, ip_fw3_opheader *op3, 90981d3153dSAlexander V. Chernikov struct sockopt_data *sd) 91081d3153dSAlexander V. Chernikov { 91181d3153dSAlexander V. Chernikov ipfw_obj_tentry *tent; 91281d3153dSAlexander V. Chernikov ipfw_obj_header *oh; 91381d3153dSAlexander V. Chernikov struct tid_info ti; 91481d3153dSAlexander V. Chernikov struct table_config *tc; 91581d3153dSAlexander V. Chernikov struct table_algo *ta; 91681d3153dSAlexander V. Chernikov struct table_info *kti; 91781d3153dSAlexander V. Chernikov struct namedobj_instance *ni; 918914bffb6SAlexander V. Chernikov int error; 91981d3153dSAlexander V. Chernikov size_t sz; 92081d3153dSAlexander V. Chernikov 92181d3153dSAlexander V. Chernikov /* Check minimum header size */ 92281d3153dSAlexander V. Chernikov sz = sizeof(*oh) + sizeof(*tent); 92381d3153dSAlexander V. Chernikov if (sd->valsize != sz) 92481d3153dSAlexander V. Chernikov return (EINVAL); 92581d3153dSAlexander V. Chernikov 92681d3153dSAlexander V. Chernikov oh = (struct _ipfw_obj_header *)ipfw_get_sopt_header(sd, sz); 92781d3153dSAlexander V. Chernikov tent = (ipfw_obj_tentry *)(oh + 1); 92881d3153dSAlexander V. Chernikov 92981d3153dSAlexander V. Chernikov /* Basic length checks for TLVs */ 93081d3153dSAlexander V. Chernikov if (oh->ntlv.head.length != sizeof(oh->ntlv)) 93181d3153dSAlexander V. Chernikov return (EINVAL); 93281d3153dSAlexander V. Chernikov 93381d3153dSAlexander V. Chernikov objheader_to_ti(oh, &ti); 93481d3153dSAlexander V. Chernikov ti.type = oh->ntlv.type; 93581d3153dSAlexander V. Chernikov ti.uidx = tent->idx; 93681d3153dSAlexander V. Chernikov 93781d3153dSAlexander V. Chernikov IPFW_UH_RLOCK(ch); 93881d3153dSAlexander V. Chernikov ni = CHAIN_TO_NI(ch); 93981d3153dSAlexander V. Chernikov 94081d3153dSAlexander V. Chernikov /* 94181d3153dSAlexander V. Chernikov * Find existing table and check its type . 94281d3153dSAlexander V. Chernikov */ 94381d3153dSAlexander V. Chernikov ta = NULL; 94481d3153dSAlexander V. Chernikov if ((tc = find_table(ni, &ti)) == NULL) { 94581d3153dSAlexander V. Chernikov IPFW_UH_RUNLOCK(ch); 94681d3153dSAlexander V. Chernikov return (ESRCH); 94781d3153dSAlexander V. Chernikov } 94881d3153dSAlexander V. Chernikov 94981d3153dSAlexander V. Chernikov /* check table type */ 95081d3153dSAlexander V. Chernikov if (tc->no.type != ti.type) { 95181d3153dSAlexander V. Chernikov IPFW_UH_RUNLOCK(ch); 95281d3153dSAlexander V. Chernikov return (EINVAL); 95381d3153dSAlexander V. Chernikov } 95481d3153dSAlexander V. Chernikov 95581d3153dSAlexander V. Chernikov kti = KIDX_TO_TI(ch, tc->no.kidx); 95681d3153dSAlexander V. Chernikov ta = tc->ta; 95781d3153dSAlexander V. Chernikov 958914bffb6SAlexander V. Chernikov if (ta->find_tentry == NULL) 959914bffb6SAlexander V. Chernikov return (ENOTSUP); 960914bffb6SAlexander V. Chernikov 961914bffb6SAlexander V. Chernikov error = ta->find_tentry(tc->astate, kti, tent); 96281d3153dSAlexander V. Chernikov 96381d3153dSAlexander V. Chernikov IPFW_UH_RUNLOCK(ch); 96481d3153dSAlexander V. Chernikov 96581d3153dSAlexander V. Chernikov return (error); 96681d3153dSAlexander V. Chernikov } 96781d3153dSAlexander V. Chernikov 96846d52008SAlexander V. Chernikov /* 96946d52008SAlexander V. Chernikov * Flushes all entries or destroys given table. 97046d52008SAlexander V. Chernikov * Data layout (v0)(current): 97146d52008SAlexander V. Chernikov * Request: [ ipfw_obj_header ] 97246d52008SAlexander V. Chernikov * 97346d52008SAlexander V. Chernikov * Returns 0 on success 97446d52008SAlexander V. Chernikov */ 975ac35ff17SAlexander V. Chernikov int 976ac35ff17SAlexander V. Chernikov ipfw_flush_table(struct ip_fw_chain *ch, ip_fw3_opheader *op3, 977ac35ff17SAlexander V. Chernikov struct sockopt_data *sd) 978ac35ff17SAlexander V. Chernikov { 979ac35ff17SAlexander V. Chernikov int error; 980ac35ff17SAlexander V. Chernikov struct _ipfw_obj_header *oh; 981ac35ff17SAlexander V. Chernikov struct tid_info ti; 982ac35ff17SAlexander V. Chernikov 983ac35ff17SAlexander V. Chernikov if (sd->valsize != sizeof(*oh)) 984ac35ff17SAlexander V. Chernikov return (EINVAL); 985ac35ff17SAlexander V. Chernikov 986ac35ff17SAlexander V. Chernikov oh = (struct _ipfw_obj_header *)op3; 987ac35ff17SAlexander V. Chernikov objheader_to_ti(oh, &ti); 988ac35ff17SAlexander V. Chernikov 9891832a7b3SAlexander V. Chernikov if (op3->opcode == IP_FW_TABLE_XDESTROY) 990ac35ff17SAlexander V. Chernikov error = destroy_table(ch, &ti); 9911832a7b3SAlexander V. Chernikov else if (op3->opcode == IP_FW_TABLE_XFLUSH) 992ac35ff17SAlexander V. Chernikov error = flush_table(ch, &ti); 993ac35ff17SAlexander V. Chernikov else 994ac35ff17SAlexander V. Chernikov return (ENOTSUP); 995ac35ff17SAlexander V. Chernikov 996ac35ff17SAlexander V. Chernikov return (error); 9973b3a8eb9SGleb Smirnoff } 9983b3a8eb9SGleb Smirnoff 99935e1bbd0SAlexander V. Chernikov /* 100035e1bbd0SAlexander V. Chernikov * Flushes given table. 100135e1bbd0SAlexander V. Chernikov * 100235e1bbd0SAlexander V. Chernikov * Function create new table instance with the same 100335e1bbd0SAlexander V. Chernikov * parameters, swaps it with old one and 100435e1bbd0SAlexander V. Chernikov * flushes state without holding any locks. 100535e1bbd0SAlexander V. Chernikov * 100635e1bbd0SAlexander V. Chernikov * Returns 0 on success. 100735e1bbd0SAlexander V. Chernikov */ 10081832a7b3SAlexander V. Chernikov int 1009ac35ff17SAlexander V. Chernikov flush_table(struct ip_fw_chain *ch, struct tid_info *ti) 10103b3a8eb9SGleb Smirnoff { 1011b074b7bbSAlexander V. Chernikov struct namedobj_instance *ni; 1012b074b7bbSAlexander V. Chernikov struct table_config *tc; 10139f7d47b0SAlexander V. Chernikov struct table_algo *ta; 10149f7d47b0SAlexander V. Chernikov struct table_info ti_old, ti_new, *tablestate; 10159f7d47b0SAlexander V. Chernikov void *astate_old, *astate_new; 1016914bffb6SAlexander V. Chernikov char algostate[64], *pstate; 1017b074b7bbSAlexander V. Chernikov int error; 1018b074b7bbSAlexander V. Chernikov uint16_t kidx; 1019914bffb6SAlexander V. Chernikov uint8_t tflags; 10203b3a8eb9SGleb Smirnoff 10213b3a8eb9SGleb Smirnoff /* 10229f7d47b0SAlexander V. Chernikov * Stage 1: save table algoritm. 1023b074b7bbSAlexander V. Chernikov * Reference found table to ensure it won't disappear. 10243b3a8eb9SGleb Smirnoff */ 1025b074b7bbSAlexander V. Chernikov IPFW_UH_WLOCK(ch); 1026b074b7bbSAlexander V. Chernikov ni = CHAIN_TO_NI(ch); 1027b074b7bbSAlexander V. Chernikov if ((tc = find_table(ni, ti)) == NULL) { 1028b074b7bbSAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 1029b074b7bbSAlexander V. Chernikov return (ESRCH); 1030b074b7bbSAlexander V. Chernikov } 10319f7d47b0SAlexander V. Chernikov ta = tc->ta; 1032b074b7bbSAlexander V. Chernikov tc->no.refcnt++; 103335e1bbd0SAlexander V. Chernikov /* Save startup algo parameters */ 1034daabb523SAlexander V. Chernikov if (ta->print_config != NULL) { 1035daabb523SAlexander V. Chernikov ta->print_config(tc->astate, KIDX_TO_TI(ch, tc->no.kidx), 1036daabb523SAlexander V. Chernikov algostate, sizeof(algostate)); 1037daabb523SAlexander V. Chernikov pstate = algostate; 1038daabb523SAlexander V. Chernikov } else 1039daabb523SAlexander V. Chernikov pstate = NULL; 1040914bffb6SAlexander V. Chernikov tflags = tc->tflags; 1041b074b7bbSAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 10423b3a8eb9SGleb Smirnoff 1043b074b7bbSAlexander V. Chernikov /* 10449f7d47b0SAlexander V. Chernikov * Stage 2: allocate new table instance using same algo. 1045b074b7bbSAlexander V. Chernikov */ 10469f7d47b0SAlexander V. Chernikov memset(&ti_new, 0, sizeof(struct table_info)); 1047914bffb6SAlexander V. Chernikov if ((error = ta->init(ch, &astate_new, &ti_new, pstate, tflags)) != 0) { 1048b074b7bbSAlexander V. Chernikov IPFW_UH_WLOCK(ch); 1049b074b7bbSAlexander V. Chernikov tc->no.refcnt--; 1050b074b7bbSAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 1051b074b7bbSAlexander V. Chernikov return (error); 1052b074b7bbSAlexander V. Chernikov } 1053b074b7bbSAlexander V. Chernikov 1054b074b7bbSAlexander V. Chernikov /* 1055b074b7bbSAlexander V. Chernikov * Stage 3: swap old state pointers with newly-allocated ones. 1056b074b7bbSAlexander V. Chernikov * Decrease refcount. 1057b074b7bbSAlexander V. Chernikov */ 1058b074b7bbSAlexander V. Chernikov IPFW_UH_WLOCK(ch); 1059b074b7bbSAlexander V. Chernikov 1060b074b7bbSAlexander V. Chernikov ni = CHAIN_TO_NI(ch); 1061b074b7bbSAlexander V. Chernikov kidx = tc->no.kidx; 10629f7d47b0SAlexander V. Chernikov tablestate = (struct table_info *)ch->tablestate; 1063b074b7bbSAlexander V. Chernikov 10649f7d47b0SAlexander V. Chernikov IPFW_WLOCK(ch); 10659f7d47b0SAlexander V. Chernikov ti_old = tablestate[kidx]; 10669f7d47b0SAlexander V. Chernikov tablestate[kidx] = ti_new; 10679f7d47b0SAlexander V. Chernikov IPFW_WUNLOCK(ch); 1068b074b7bbSAlexander V. Chernikov 10699f7d47b0SAlexander V. Chernikov astate_old = tc->astate; 10709f7d47b0SAlexander V. Chernikov tc->astate = astate_new; 10719f7d47b0SAlexander V. Chernikov tc->ti = ti_new; 10729f7d47b0SAlexander V. Chernikov tc->count = 0; 1073b074b7bbSAlexander V. Chernikov tc->no.refcnt--; 1074b074b7bbSAlexander V. Chernikov 1075b074b7bbSAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 10763b3a8eb9SGleb Smirnoff 1077b074b7bbSAlexander V. Chernikov /* 1078b074b7bbSAlexander V. Chernikov * Stage 4: perform real flush. 1079b074b7bbSAlexander V. Chernikov */ 10809f7d47b0SAlexander V. Chernikov ta->destroy(astate_old, &ti_old); 10813b3a8eb9SGleb Smirnoff 10823b3a8eb9SGleb Smirnoff return (0); 10833b3a8eb9SGleb Smirnoff } 10843b3a8eb9SGleb Smirnoff 1085b074b7bbSAlexander V. Chernikov /* 108646d52008SAlexander V. Chernikov * Swaps two tables. 108746d52008SAlexander V. Chernikov * Data layout (v0)(current): 108846d52008SAlexander V. Chernikov * Request: [ ipfw_obj_header ipfw_obj_ntlv ] 108946d52008SAlexander V. Chernikov * 109046d52008SAlexander V. Chernikov * Returns 0 on success 109146d52008SAlexander V. Chernikov */ 109246d52008SAlexander V. Chernikov int 109346d52008SAlexander V. Chernikov ipfw_swap_table(struct ip_fw_chain *ch, ip_fw3_opheader *op3, 109446d52008SAlexander V. Chernikov struct sockopt_data *sd) 109546d52008SAlexander V. Chernikov { 109646d52008SAlexander V. Chernikov int error; 109746d52008SAlexander V. Chernikov struct _ipfw_obj_header *oh; 109846d52008SAlexander V. Chernikov struct tid_info ti_a, ti_b; 109946d52008SAlexander V. Chernikov 110046d52008SAlexander V. Chernikov if (sd->valsize != sizeof(*oh) + sizeof(ipfw_obj_ntlv)) 110146d52008SAlexander V. Chernikov return (EINVAL); 110246d52008SAlexander V. Chernikov 110346d52008SAlexander V. Chernikov oh = (struct _ipfw_obj_header *)op3; 110446d52008SAlexander V. Chernikov ntlv_to_ti(&oh->ntlv, &ti_a); 110546d52008SAlexander V. Chernikov ntlv_to_ti((ipfw_obj_ntlv *)(oh + 1), &ti_b); 110646d52008SAlexander V. Chernikov 1107a73d728dSAlexander V. Chernikov error = swap_tables(ch, &ti_a, &ti_b); 110846d52008SAlexander V. Chernikov 110946d52008SAlexander V. Chernikov return (error); 111046d52008SAlexander V. Chernikov } 111146d52008SAlexander V. Chernikov 111235e1bbd0SAlexander V. Chernikov /* 111335e1bbd0SAlexander V. Chernikov * Swaps two tables of the same type/valtype. 111435e1bbd0SAlexander V. Chernikov * 111535e1bbd0SAlexander V. Chernikov * Checks if tables are compatible and limits 111635e1bbd0SAlexander V. Chernikov * permits swap, than actually perform swap 111735e1bbd0SAlexander V. Chernikov * by switching 111835e1bbd0SAlexander V. Chernikov * 1) runtime data (ch->tablestate) 111935e1bbd0SAlexander V. Chernikov * 2) runtime cache in @tc 112035e1bbd0SAlexander V. Chernikov * 3) algo-specific data (tc->astate) 112135e1bbd0SAlexander V. Chernikov * 4) number of items 112235e1bbd0SAlexander V. Chernikov * 112335e1bbd0SAlexander V. Chernikov * Since @ti has changed for each table, calls notification callbacks. 112435e1bbd0SAlexander V. Chernikov * 112535e1bbd0SAlexander V. Chernikov * Returns 0 on success. 112635e1bbd0SAlexander V. Chernikov */ 112746d52008SAlexander V. Chernikov static int 1128a73d728dSAlexander V. Chernikov swap_tables(struct ip_fw_chain *ch, struct tid_info *a, 112946d52008SAlexander V. Chernikov struct tid_info *b) 113046d52008SAlexander V. Chernikov { 113146d52008SAlexander V. Chernikov struct namedobj_instance *ni; 113246d52008SAlexander V. Chernikov struct table_config *tc_a, *tc_b; 113346d52008SAlexander V. Chernikov struct table_algo *ta; 113446d52008SAlexander V. Chernikov struct table_info ti, *tablestate; 113546d52008SAlexander V. Chernikov void *astate; 113646d52008SAlexander V. Chernikov uint32_t count; 113746d52008SAlexander V. Chernikov 113846d52008SAlexander V. Chernikov /* 113946d52008SAlexander V. Chernikov * Stage 1: find both tables and ensure they are of 114035e1bbd0SAlexander V. Chernikov * the same type. 114146d52008SAlexander V. Chernikov */ 114246d52008SAlexander V. Chernikov IPFW_UH_WLOCK(ch); 114346d52008SAlexander V. Chernikov ni = CHAIN_TO_NI(ch); 114446d52008SAlexander V. Chernikov if ((tc_a = find_table(ni, a)) == NULL) { 114546d52008SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 114646d52008SAlexander V. Chernikov return (ESRCH); 114746d52008SAlexander V. Chernikov } 114846d52008SAlexander V. Chernikov if ((tc_b = find_table(ni, b)) == NULL) { 114946d52008SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 115046d52008SAlexander V. Chernikov return (ESRCH); 115146d52008SAlexander V. Chernikov } 115246d52008SAlexander V. Chernikov 115346d52008SAlexander V. Chernikov /* It is very easy to swap between the same table */ 115446d52008SAlexander V. Chernikov if (tc_a == tc_b) { 115546d52008SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 115646d52008SAlexander V. Chernikov return (0); 115746d52008SAlexander V. Chernikov } 115846d52008SAlexander V. Chernikov 115946d52008SAlexander V. Chernikov /* Check type and value are the same */ 116046d52008SAlexander V. Chernikov if (tc_a->no.type != tc_b->no.type || tc_a->tflags != tc_b->tflags || 116146d52008SAlexander V. Chernikov tc_a->vtype != tc_b->vtype) { 116246d52008SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 116346d52008SAlexander V. Chernikov return (EINVAL); 116446d52008SAlexander V. Chernikov } 116546d52008SAlexander V. Chernikov 116646d52008SAlexander V. Chernikov /* Check limits before swap */ 116746d52008SAlexander V. Chernikov if ((tc_a->limit != 0 && tc_b->count > tc_a->limit) || 116846d52008SAlexander V. Chernikov (tc_b->limit != 0 && tc_a->count > tc_b->limit)) { 116946d52008SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 117046d52008SAlexander V. Chernikov return (EFBIG); 117146d52008SAlexander V. Chernikov } 117246d52008SAlexander V. Chernikov 117346d52008SAlexander V. Chernikov /* Everything is fine, prepare to swap */ 117446d52008SAlexander V. Chernikov tablestate = (struct table_info *)ch->tablestate; 117546d52008SAlexander V. Chernikov ti = tablestate[tc_a->no.kidx]; 117646d52008SAlexander V. Chernikov ta = tc_a->ta; 117746d52008SAlexander V. Chernikov astate = tc_a->astate; 117846d52008SAlexander V. Chernikov count = tc_a->count; 117946d52008SAlexander V. Chernikov 118046d52008SAlexander V. Chernikov IPFW_WLOCK(ch); 118146d52008SAlexander V. Chernikov /* a <- b */ 118246d52008SAlexander V. Chernikov tablestate[tc_a->no.kidx] = tablestate[tc_b->no.kidx]; 118346d52008SAlexander V. Chernikov tc_a->ta = tc_b->ta; 118446d52008SAlexander V. Chernikov tc_a->astate = tc_b->astate; 118546d52008SAlexander V. Chernikov tc_a->count = tc_b->count; 118646d52008SAlexander V. Chernikov /* b <- a */ 118746d52008SAlexander V. Chernikov tablestate[tc_b->no.kidx] = ti; 118846d52008SAlexander V. Chernikov tc_b->ta = ta; 118946d52008SAlexander V. Chernikov tc_b->astate = astate; 119046d52008SAlexander V. Chernikov tc_b->count = count; 119146d52008SAlexander V. Chernikov IPFW_WUNLOCK(ch); 119246d52008SAlexander V. Chernikov 119346d52008SAlexander V. Chernikov /* Ensure tc.ti copies are in sync */ 119446d52008SAlexander V. Chernikov tc_a->ti = tablestate[tc_a->no.kidx]; 119546d52008SAlexander V. Chernikov tc_b->ti = tablestate[tc_b->no.kidx]; 119646d52008SAlexander V. Chernikov 119746d52008SAlexander V. Chernikov /* Notify both tables on @ti change */ 119846d52008SAlexander V. Chernikov if (tc_a->ta->change_ti != NULL) 119946d52008SAlexander V. Chernikov tc_a->ta->change_ti(tc_a->astate, &tablestate[tc_a->no.kidx]); 120046d52008SAlexander V. Chernikov if (tc_b->ta->change_ti != NULL) 120146d52008SAlexander V. Chernikov tc_b->ta->change_ti(tc_b->astate, &tablestate[tc_b->no.kidx]); 120246d52008SAlexander V. Chernikov 120346d52008SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 120446d52008SAlexander V. Chernikov 120546d52008SAlexander V. Chernikov return (0); 120646d52008SAlexander V. Chernikov } 120746d52008SAlexander V. Chernikov 120846d52008SAlexander V. Chernikov /* 12099f7d47b0SAlexander V. Chernikov * Destroys table specified by @ti. 1210ac35ff17SAlexander V. Chernikov * Data layout (v0)(current): 1211ac35ff17SAlexander V. Chernikov * Request: [ ip_fw3_opheader ] 1212ac35ff17SAlexander V. Chernikov * 1213ac35ff17SAlexander V. Chernikov * Returns 0 on success 1214b074b7bbSAlexander V. Chernikov */ 1215ac35ff17SAlexander V. Chernikov static int 1216ac35ff17SAlexander V. Chernikov destroy_table(struct ip_fw_chain *ch, struct tid_info *ti) 1217b074b7bbSAlexander V. Chernikov { 1218b074b7bbSAlexander V. Chernikov struct namedobj_instance *ni; 1219b074b7bbSAlexander V. Chernikov struct table_config *tc; 1220b074b7bbSAlexander V. Chernikov 1221b074b7bbSAlexander V. Chernikov IPFW_UH_WLOCK(ch); 1222b074b7bbSAlexander V. Chernikov 1223b074b7bbSAlexander V. Chernikov ni = CHAIN_TO_NI(ch); 1224b074b7bbSAlexander V. Chernikov if ((tc = find_table(ni, ti)) == NULL) { 1225b074b7bbSAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 1226b074b7bbSAlexander V. Chernikov return (ESRCH); 1227b074b7bbSAlexander V. Chernikov } 1228b074b7bbSAlexander V. Chernikov 12299f7d47b0SAlexander V. Chernikov /* Do not permit destroying referenced tables */ 12309f7d47b0SAlexander V. Chernikov if (tc->no.refcnt > 0) { 1231b074b7bbSAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 1232b074b7bbSAlexander V. Chernikov return (EBUSY); 1233b074b7bbSAlexander V. Chernikov } 1234b074b7bbSAlexander V. Chernikov 1235b074b7bbSAlexander V. Chernikov IPFW_WLOCK(ch); 1236b074b7bbSAlexander V. Chernikov unlink_table(ch, tc); 1237b074b7bbSAlexander V. Chernikov IPFW_WUNLOCK(ch); 1238b074b7bbSAlexander V. Chernikov 1239b074b7bbSAlexander V. Chernikov /* Free obj index */ 1240ac35ff17SAlexander V. Chernikov if (ipfw_objhash_free_idx(ni, tc->no.kidx) != 0) 1241b074b7bbSAlexander V. Chernikov printf("Error unlinking kidx %d from table %s\n", 1242b074b7bbSAlexander V. Chernikov tc->no.kidx, tc->tablename); 1243b074b7bbSAlexander V. Chernikov 1244b074b7bbSAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 1245b074b7bbSAlexander V. Chernikov 1246b074b7bbSAlexander V. Chernikov free_table_config(ni, tc); 1247b074b7bbSAlexander V. Chernikov 1248b074b7bbSAlexander V. Chernikov return (0); 1249b074b7bbSAlexander V. Chernikov } 1250b074b7bbSAlexander V. Chernikov 1251b074b7bbSAlexander V. Chernikov static void 1252b074b7bbSAlexander V. Chernikov destroy_table_locked(struct namedobj_instance *ni, struct named_object *no, 1253b074b7bbSAlexander V. Chernikov void *arg) 1254b074b7bbSAlexander V. Chernikov { 1255b074b7bbSAlexander V. Chernikov 1256b074b7bbSAlexander V. Chernikov unlink_table((struct ip_fw_chain *)arg, (struct table_config *)no); 1257ac35ff17SAlexander V. Chernikov if (ipfw_objhash_free_idx(ni, no->kidx) != 0) 1258b074b7bbSAlexander V. Chernikov printf("Error unlinking kidx %d from table %s\n", 1259b074b7bbSAlexander V. Chernikov no->kidx, no->name); 1260b074b7bbSAlexander V. Chernikov free_table_config(ni, (struct table_config *)no); 1261b074b7bbSAlexander V. Chernikov } 1262b074b7bbSAlexander V. Chernikov 126335e1bbd0SAlexander V. Chernikov /* 126435e1bbd0SAlexander V. Chernikov * Shuts tables module down. 126535e1bbd0SAlexander V. Chernikov */ 12663b3a8eb9SGleb Smirnoff void 12673b3a8eb9SGleb Smirnoff ipfw_destroy_tables(struct ip_fw_chain *ch) 12683b3a8eb9SGleb Smirnoff { 12693b3a8eb9SGleb Smirnoff 1270b074b7bbSAlexander V. Chernikov /* Remove all tables from working set */ 1271b074b7bbSAlexander V. Chernikov IPFW_UH_WLOCK(ch); 1272b074b7bbSAlexander V. Chernikov IPFW_WLOCK(ch); 1273b074b7bbSAlexander V. Chernikov ipfw_objhash_foreach(CHAIN_TO_NI(ch), destroy_table_locked, ch); 1274b074b7bbSAlexander V. Chernikov IPFW_WUNLOCK(ch); 1275b074b7bbSAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 12763b3a8eb9SGleb Smirnoff 12773b3a8eb9SGleb Smirnoff /* Free pointers itself */ 12789f7d47b0SAlexander V. Chernikov free(ch->tablestate, M_IPFW); 12799f7d47b0SAlexander V. Chernikov 12809f7d47b0SAlexander V. Chernikov ipfw_table_algo_destroy(ch); 1281b074b7bbSAlexander V. Chernikov 1282b074b7bbSAlexander V. Chernikov ipfw_objhash_destroy(CHAIN_TO_NI(ch)); 1283b074b7bbSAlexander V. Chernikov free(CHAIN_TO_TCFG(ch), M_IPFW); 12843b3a8eb9SGleb Smirnoff } 12853b3a8eb9SGleb Smirnoff 128635e1bbd0SAlexander V. Chernikov /* 128735e1bbd0SAlexander V. Chernikov * Starts tables module. 128835e1bbd0SAlexander V. Chernikov */ 12893b3a8eb9SGleb Smirnoff int 12903b3a8eb9SGleb Smirnoff ipfw_init_tables(struct ip_fw_chain *ch) 12913b3a8eb9SGleb Smirnoff { 1292b074b7bbSAlexander V. Chernikov struct tables_config *tcfg; 1293b074b7bbSAlexander V. Chernikov 12943b3a8eb9SGleb Smirnoff /* Allocate pointers */ 12959f7d47b0SAlexander V. Chernikov ch->tablestate = malloc(V_fw_tables_max * sizeof(struct table_info), 12969f7d47b0SAlexander V. Chernikov M_IPFW, M_WAITOK | M_ZERO); 1297b074b7bbSAlexander V. Chernikov 1298b074b7bbSAlexander V. Chernikov tcfg = malloc(sizeof(struct tables_config), M_IPFW, M_WAITOK | M_ZERO); 1299b074b7bbSAlexander V. Chernikov tcfg->namehash = ipfw_objhash_create(V_fw_tables_max); 1300b074b7bbSAlexander V. Chernikov ch->tblcfg = tcfg; 1301b074b7bbSAlexander V. Chernikov 13029f7d47b0SAlexander V. Chernikov ipfw_table_algo_init(ch); 13039f7d47b0SAlexander V. Chernikov 13043b3a8eb9SGleb Smirnoff return (0); 13053b3a8eb9SGleb Smirnoff } 13063b3a8eb9SGleb Smirnoff 130735e1bbd0SAlexander V. Chernikov /* 130835e1bbd0SAlexander V. Chernikov * Grow tables index. 130935e1bbd0SAlexander V. Chernikov * 131035e1bbd0SAlexander V. Chernikov * Returns 0 on success. 131135e1bbd0SAlexander V. Chernikov */ 13123b3a8eb9SGleb Smirnoff int 13133b3a8eb9SGleb Smirnoff ipfw_resize_tables(struct ip_fw_chain *ch, unsigned int ntables) 13143b3a8eb9SGleb Smirnoff { 13153b3a8eb9SGleb Smirnoff unsigned int ntables_old, tbl; 1316b074b7bbSAlexander V. Chernikov struct namedobj_instance *ni; 13179f7d47b0SAlexander V. Chernikov void *new_idx, *old_tablestate, *tablestate; 131868394ec8SAlexander V. Chernikov struct table_info *ti; 131968394ec8SAlexander V. Chernikov struct table_config *tc; 132068394ec8SAlexander V. Chernikov int i, new_blocks; 13213b3a8eb9SGleb Smirnoff 13223b3a8eb9SGleb Smirnoff /* Check new value for validity */ 13233b3a8eb9SGleb Smirnoff if (ntables > IPFW_TABLES_MAX) 13243b3a8eb9SGleb Smirnoff ntables = IPFW_TABLES_MAX; 13253b3a8eb9SGleb Smirnoff 13263b3a8eb9SGleb Smirnoff /* Allocate new pointers */ 13279f7d47b0SAlexander V. Chernikov tablestate = malloc(ntables * sizeof(struct table_info), 13289f7d47b0SAlexander V. Chernikov M_IPFW, M_WAITOK | M_ZERO); 13299f7d47b0SAlexander V. Chernikov 1330b074b7bbSAlexander V. Chernikov ipfw_objhash_bitmap_alloc(ntables, (void *)&new_idx, &new_blocks); 13313b3a8eb9SGleb Smirnoff 13329f7d47b0SAlexander V. Chernikov IPFW_UH_WLOCK(ch); 13333b3a8eb9SGleb Smirnoff 13343b3a8eb9SGleb Smirnoff tbl = (ntables >= V_fw_tables_max) ? V_fw_tables_max : ntables; 1335b074b7bbSAlexander V. Chernikov ni = CHAIN_TO_NI(ch); 1336b074b7bbSAlexander V. Chernikov 13379f7d47b0SAlexander V. Chernikov /* Temporary restrict decreasing max_tables */ 13389f7d47b0SAlexander V. Chernikov if (ntables < V_fw_tables_max) { 13399f7d47b0SAlexander V. Chernikov 13409f7d47b0SAlexander V. Chernikov /* 13419f7d47b0SAlexander V. Chernikov * FIXME: Check if we really can shrink 13429f7d47b0SAlexander V. Chernikov */ 13439f7d47b0SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 1344b074b7bbSAlexander V. Chernikov return (EINVAL); 1345b074b7bbSAlexander V. Chernikov } 13463b3a8eb9SGleb Smirnoff 13479f7d47b0SAlexander V. Chernikov /* Copy table info/indices */ 13489f7d47b0SAlexander V. Chernikov memcpy(tablestate, ch->tablestate, sizeof(struct table_info) * tbl); 13499f7d47b0SAlexander V. Chernikov ipfw_objhash_bitmap_merge(ni, &new_idx, &new_blocks); 13503b3a8eb9SGleb Smirnoff 13519f7d47b0SAlexander V. Chernikov IPFW_WLOCK(ch); 13529f7d47b0SAlexander V. Chernikov 13539f7d47b0SAlexander V. Chernikov /* Change pointers */ 13549f7d47b0SAlexander V. Chernikov old_tablestate = ch->tablestate; 13559f7d47b0SAlexander V. Chernikov ch->tablestate = tablestate; 13569f7d47b0SAlexander V. Chernikov ipfw_objhash_bitmap_swap(ni, &new_idx, &new_blocks); 13573b3a8eb9SGleb Smirnoff 13583b3a8eb9SGleb Smirnoff ntables_old = V_fw_tables_max; 13593b3a8eb9SGleb Smirnoff V_fw_tables_max = ntables; 13603b3a8eb9SGleb Smirnoff 13613b3a8eb9SGleb Smirnoff IPFW_WUNLOCK(ch); 136268394ec8SAlexander V. Chernikov 136368394ec8SAlexander V. Chernikov /* Notify all consumers that their @ti pointer has changed */ 136468394ec8SAlexander V. Chernikov ti = (struct table_info *)ch->tablestate; 136568394ec8SAlexander V. Chernikov for (i = 0; i < tbl; i++, ti++) { 136668394ec8SAlexander V. Chernikov if (ti->lookup == NULL) 136768394ec8SAlexander V. Chernikov continue; 136868394ec8SAlexander V. Chernikov tc = (struct table_config *)ipfw_objhash_lookup_kidx(ni, i); 136968394ec8SAlexander V. Chernikov if (tc == NULL || tc->ta->change_ti == NULL) 137068394ec8SAlexander V. Chernikov continue; 137168394ec8SAlexander V. Chernikov 137268394ec8SAlexander V. Chernikov tc->ta->change_ti(tc->astate, ti); 137368394ec8SAlexander V. Chernikov } 137468394ec8SAlexander V. Chernikov 13759f7d47b0SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 13763b3a8eb9SGleb Smirnoff 13773b3a8eb9SGleb Smirnoff /* Free old pointers */ 13789f7d47b0SAlexander V. Chernikov free(old_tablestate, M_IPFW); 1379b074b7bbSAlexander V. Chernikov ipfw_objhash_bitmap_free(new_idx, new_blocks); 13803b3a8eb9SGleb Smirnoff 13813b3a8eb9SGleb Smirnoff return (0); 13823b3a8eb9SGleb Smirnoff } 13833b3a8eb9SGleb Smirnoff 1384a73d728dSAlexander V. Chernikov /* 138535e1bbd0SAlexander V. Chernikov * Switch between "set 0" and "rule's set" table binding, 1386a73d728dSAlexander V. Chernikov * Check all ruleset bindings and permits changing 1387a73d728dSAlexander V. Chernikov * IFF each binding has both rule AND table in default set (set 0). 1388a73d728dSAlexander V. Chernikov * 1389a73d728dSAlexander V. Chernikov * Returns 0 on success. 1390a73d728dSAlexander V. Chernikov */ 1391a73d728dSAlexander V. Chernikov int 1392a73d728dSAlexander V. Chernikov ipfw_switch_tables_namespace(struct ip_fw_chain *ch, unsigned int sets) 1393a73d728dSAlexander V. Chernikov { 1394a73d728dSAlexander V. Chernikov struct namedobj_instance *ni; 1395a73d728dSAlexander V. Chernikov struct named_object *no; 1396a73d728dSAlexander V. Chernikov struct ip_fw *rule; 1397a73d728dSAlexander V. Chernikov ipfw_insn *cmd; 1398a73d728dSAlexander V. Chernikov int cmdlen, i, l; 1399a73d728dSAlexander V. Chernikov uint16_t kidx; 1400a73d728dSAlexander V. Chernikov uint8_t type; 1401a73d728dSAlexander V. Chernikov 1402a73d728dSAlexander V. Chernikov IPFW_UH_WLOCK(ch); 1403a73d728dSAlexander V. Chernikov 1404a73d728dSAlexander V. Chernikov if (V_fw_tables_sets == sets) { 1405a73d728dSAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 1406a73d728dSAlexander V. Chernikov return (0); 1407a73d728dSAlexander V. Chernikov } 1408a73d728dSAlexander V. Chernikov 1409a73d728dSAlexander V. Chernikov ni = CHAIN_TO_NI(ch); 1410a73d728dSAlexander V. Chernikov 141135e1bbd0SAlexander V. Chernikov /* 141235e1bbd0SAlexander V. Chernikov * Scan all rules and examine tables opcodes. 141335e1bbd0SAlexander V. Chernikov */ 1414a73d728dSAlexander V. Chernikov for (i = 0; i < ch->n_rules; i++) { 1415a73d728dSAlexander V. Chernikov rule = ch->map[i]; 1416a73d728dSAlexander V. Chernikov 1417a73d728dSAlexander V. Chernikov l = rule->cmd_len; 1418a73d728dSAlexander V. Chernikov cmd = rule->cmd; 1419a73d728dSAlexander V. Chernikov cmdlen = 0; 1420a73d728dSAlexander V. Chernikov for ( ; l > 0 ; l -= cmdlen, cmd += cmdlen) { 1421a73d728dSAlexander V. Chernikov cmdlen = F_LEN(cmd); 1422a73d728dSAlexander V. Chernikov 1423a73d728dSAlexander V. Chernikov if (classify_table_opcode(cmd, &kidx, &type) != 0) 1424a73d728dSAlexander V. Chernikov continue; 1425a73d728dSAlexander V. Chernikov 1426a73d728dSAlexander V. Chernikov no = ipfw_objhash_lookup_kidx(ni, kidx); 1427a73d728dSAlexander V. Chernikov 142835e1bbd0SAlexander V. Chernikov /* Check if both table object and rule has the set 0 */ 1429a73d728dSAlexander V. Chernikov if (no->set != 0 || rule->set != 0) { 1430a73d728dSAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 1431a73d728dSAlexander V. Chernikov return (EBUSY); 1432a73d728dSAlexander V. Chernikov } 1433a73d728dSAlexander V. Chernikov 1434a73d728dSAlexander V. Chernikov } 1435a73d728dSAlexander V. Chernikov } 1436a73d728dSAlexander V. Chernikov V_fw_tables_sets = sets; 1437a73d728dSAlexander V. Chernikov 1438a73d728dSAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 1439a73d728dSAlexander V. Chernikov 1440a73d728dSAlexander V. Chernikov return (0); 1441a73d728dSAlexander V. Chernikov } 1442a73d728dSAlexander V. Chernikov 14433b3a8eb9SGleb Smirnoff int 14443b3a8eb9SGleb Smirnoff ipfw_lookup_table(struct ip_fw_chain *ch, uint16_t tbl, in_addr_t addr, 14453b3a8eb9SGleb Smirnoff uint32_t *val) 14463b3a8eb9SGleb Smirnoff { 14479f7d47b0SAlexander V. Chernikov struct table_info *ti; 14483b3a8eb9SGleb Smirnoff 14499f7d47b0SAlexander V. Chernikov ti = &(((struct table_info *)ch->tablestate)[tbl]); 14509f7d47b0SAlexander V. Chernikov 14519f7d47b0SAlexander V. Chernikov return (ti->lookup(ti, &addr, sizeof(in_addr_t), val)); 14523b3a8eb9SGleb Smirnoff } 14539f7d47b0SAlexander V. Chernikov 14549f7d47b0SAlexander V. Chernikov int 14559f7d47b0SAlexander V. Chernikov ipfw_lookup_table_extended(struct ip_fw_chain *ch, uint16_t tbl, uint16_t plen, 14569f7d47b0SAlexander V. Chernikov void *paddr, uint32_t *val) 14579f7d47b0SAlexander V. Chernikov { 14589f7d47b0SAlexander V. Chernikov struct table_info *ti; 14599f7d47b0SAlexander V. Chernikov 14609f7d47b0SAlexander V. Chernikov ti = &(((struct table_info *)ch->tablestate)[tbl]); 14619f7d47b0SAlexander V. Chernikov 14629f7d47b0SAlexander V. Chernikov return (ti->lookup(ti, paddr, plen, val)); 14639f7d47b0SAlexander V. Chernikov } 14649f7d47b0SAlexander V. Chernikov 14659f7d47b0SAlexander V. Chernikov /* 14669f7d47b0SAlexander V. Chernikov * Info/List/dump support for tables. 14679f7d47b0SAlexander V. Chernikov * 14689f7d47b0SAlexander V. Chernikov */ 14699f7d47b0SAlexander V. Chernikov 1470f1220db8SAlexander V. Chernikov /* 1471d3a4f924SAlexander V. Chernikov * High-level 'get' cmds sysctl handlers 1472d3a4f924SAlexander V. Chernikov */ 1473d3a4f924SAlexander V. Chernikov 1474d3a4f924SAlexander V. Chernikov /* 1475d3a4f924SAlexander V. Chernikov * Lists all tables currently available in kernel. 1476ac35ff17SAlexander V. Chernikov * Data layout (v0)(current): 1477d3a4f924SAlexander V. Chernikov * Request: [ ipfw_obj_lheader ], size = ipfw_obj_lheader.size 1478d3a4f924SAlexander V. Chernikov * Reply: [ ipfw_obj_lheader ipfw_xtable_info x N ] 1479d3a4f924SAlexander V. Chernikov * 1480d3a4f924SAlexander V. Chernikov * Returns 0 on success 1481d3a4f924SAlexander V. Chernikov */ 1482f1220db8SAlexander V. Chernikov int 14832d99a349SAlexander V. Chernikov ipfw_list_tables(struct ip_fw_chain *ch, struct sockopt_data *sd) 1484f1220db8SAlexander V. Chernikov { 1485f1220db8SAlexander V. Chernikov struct _ipfw_obj_lheader *olh; 1486f1220db8SAlexander V. Chernikov int error; 1487f1220db8SAlexander V. Chernikov 14882d99a349SAlexander V. Chernikov olh = (struct _ipfw_obj_lheader *)ipfw_get_sopt_header(sd,sizeof(*olh)); 14892d99a349SAlexander V. Chernikov if (olh == NULL) 1490d3a4f924SAlexander V. Chernikov return (EINVAL); 149168394ec8SAlexander V. Chernikov if (sd->valsize < olh->size) 149268394ec8SAlexander V. Chernikov return (EINVAL); 1493d3a4f924SAlexander V. Chernikov 1494f1220db8SAlexander V. Chernikov IPFW_UH_RLOCK(ch); 14952d99a349SAlexander V. Chernikov error = export_tables(ch, olh, sd); 1496f1220db8SAlexander V. Chernikov IPFW_UH_RUNLOCK(ch); 1497f1220db8SAlexander V. Chernikov 1498f1220db8SAlexander V. Chernikov return (error); 1499f1220db8SAlexander V. Chernikov } 1500f1220db8SAlexander V. Chernikov 1501f1220db8SAlexander V. Chernikov /* 15022d99a349SAlexander V. Chernikov * Store table info to buffer provided by @sd. 1503ac35ff17SAlexander V. Chernikov * Data layout (v0)(current): 1504d3a4f924SAlexander V. Chernikov * Request: [ ipfw_obj_header ipfw_xtable_info(empty)] 1505d3a4f924SAlexander V. Chernikov * Reply: [ ipfw_obj_header ipfw_xtable_info ] 1506d3a4f924SAlexander V. Chernikov * 1507d3a4f924SAlexander V. Chernikov * Returns 0 on success. 1508d3a4f924SAlexander V. Chernikov */ 1509d3a4f924SAlexander V. Chernikov int 15102d99a349SAlexander V. Chernikov ipfw_describe_table(struct ip_fw_chain *ch, struct sockopt_data *sd) 1511d3a4f924SAlexander V. Chernikov { 1512d3a4f924SAlexander V. Chernikov struct _ipfw_obj_header *oh; 1513d3a4f924SAlexander V. Chernikov struct table_config *tc; 1514d3a4f924SAlexander V. Chernikov struct tid_info ti; 1515d3a4f924SAlexander V. Chernikov size_t sz; 1516d3a4f924SAlexander V. Chernikov 1517d3a4f924SAlexander V. Chernikov sz = sizeof(*oh) + sizeof(ipfw_xtable_info); 15182d99a349SAlexander V. Chernikov oh = (struct _ipfw_obj_header *)ipfw_get_sopt_header(sd, sz); 15192d99a349SAlexander V. Chernikov if (oh == NULL) 1520d3a4f924SAlexander V. Chernikov return (EINVAL); 1521d3a4f924SAlexander V. Chernikov 1522d3a4f924SAlexander V. Chernikov objheader_to_ti(oh, &ti); 1523d3a4f924SAlexander V. Chernikov 1524d3a4f924SAlexander V. Chernikov IPFW_UH_RLOCK(ch); 1525d3a4f924SAlexander V. Chernikov if ((tc = find_table(CHAIN_TO_NI(ch), &ti)) == NULL) { 1526d3a4f924SAlexander V. Chernikov IPFW_UH_RUNLOCK(ch); 1527d3a4f924SAlexander V. Chernikov return (ESRCH); 1528d3a4f924SAlexander V. Chernikov } 1529d3a4f924SAlexander V. Chernikov 1530ac35ff17SAlexander V. Chernikov export_table_info(ch, tc, (ipfw_xtable_info *)(oh + 1)); 1531d3a4f924SAlexander V. Chernikov IPFW_UH_RUNLOCK(ch); 1532d3a4f924SAlexander V. Chernikov 15332d99a349SAlexander V. Chernikov return (0); 1534d3a4f924SAlexander V. Chernikov } 1535d3a4f924SAlexander V. Chernikov 1536d3a4f924SAlexander V. Chernikov /* 1537adf3b2b9SAlexander V. Chernikov * Modifies existing table. 1538adf3b2b9SAlexander V. Chernikov * Data layout (v0)(current): 1539adf3b2b9SAlexander V. Chernikov * Request: [ ipfw_obj_header ipfw_xtable_info ] 1540adf3b2b9SAlexander V. Chernikov * 1541adf3b2b9SAlexander V. Chernikov * Returns 0 on success 1542adf3b2b9SAlexander V. Chernikov */ 1543adf3b2b9SAlexander V. Chernikov int 1544adf3b2b9SAlexander V. Chernikov ipfw_modify_table(struct ip_fw_chain *ch, ip_fw3_opheader *op3, 1545adf3b2b9SAlexander V. Chernikov struct sockopt_data *sd) 1546adf3b2b9SAlexander V. Chernikov { 1547adf3b2b9SAlexander V. Chernikov struct _ipfw_obj_header *oh; 1548adf3b2b9SAlexander V. Chernikov ipfw_xtable_info *i; 1549adf3b2b9SAlexander V. Chernikov char *tname; 1550adf3b2b9SAlexander V. Chernikov struct tid_info ti; 1551adf3b2b9SAlexander V. Chernikov struct namedobj_instance *ni; 1552adf3b2b9SAlexander V. Chernikov struct table_config *tc; 1553adf3b2b9SAlexander V. Chernikov 1554adf3b2b9SAlexander V. Chernikov if (sd->valsize != sizeof(*oh) + sizeof(ipfw_xtable_info)) 1555adf3b2b9SAlexander V. Chernikov return (EINVAL); 1556adf3b2b9SAlexander V. Chernikov 1557adf3b2b9SAlexander V. Chernikov oh = (struct _ipfw_obj_header *)sd->kbuf; 1558adf3b2b9SAlexander V. Chernikov i = (ipfw_xtable_info *)(oh + 1); 1559adf3b2b9SAlexander V. Chernikov 1560adf3b2b9SAlexander V. Chernikov /* 1561adf3b2b9SAlexander V. Chernikov * Verify user-supplied strings. 1562adf3b2b9SAlexander V. Chernikov * Check for null-terminated/zero-length strings/ 1563adf3b2b9SAlexander V. Chernikov */ 1564adf3b2b9SAlexander V. Chernikov tname = oh->ntlv.name; 1565adf3b2b9SAlexander V. Chernikov if (ipfw_check_table_name(tname) != 0) 1566adf3b2b9SAlexander V. Chernikov return (EINVAL); 1567adf3b2b9SAlexander V. Chernikov 1568adf3b2b9SAlexander V. Chernikov objheader_to_ti(oh, &ti); 1569adf3b2b9SAlexander V. Chernikov ti.type = i->type; 1570adf3b2b9SAlexander V. Chernikov 1571adf3b2b9SAlexander V. Chernikov IPFW_UH_WLOCK(ch); 1572adf3b2b9SAlexander V. Chernikov ni = CHAIN_TO_NI(ch); 1573adf3b2b9SAlexander V. Chernikov if ((tc = find_table(ni, &ti)) == NULL) { 1574adf3b2b9SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 1575adf3b2b9SAlexander V. Chernikov return (ESRCH); 1576adf3b2b9SAlexander V. Chernikov } 1577adf3b2b9SAlexander V. Chernikov if ((i->mflags & IPFW_TMFLAGS_FTYPE) != 0) 1578adf3b2b9SAlexander V. Chernikov tc->vftype = i->vftype; 1579adf3b2b9SAlexander V. Chernikov if ((i->mflags & IPFW_TMFLAGS_LIMIT) != 0) 1580adf3b2b9SAlexander V. Chernikov tc->limit = i->limit; 15814f43138aSAlexander V. Chernikov if ((i->mflags & IPFW_TMFLAGS_LOCK) != 0) 15824f43138aSAlexander V. Chernikov tc->locked = ((i->flags & IPFW_TGFLAGS_LOCKED) != 0); 1583adf3b2b9SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 1584adf3b2b9SAlexander V. Chernikov 1585adf3b2b9SAlexander V. Chernikov return (0); 1586adf3b2b9SAlexander V. Chernikov } 1587adf3b2b9SAlexander V. Chernikov 1588adf3b2b9SAlexander V. Chernikov /* 15899490a627SAlexander V. Chernikov * Creates new table. 1590ac35ff17SAlexander V. Chernikov * Data layout (v0)(current): 15919490a627SAlexander V. Chernikov * Request: [ ipfw_obj_header ipfw_xtable_info ] 15929490a627SAlexander V. Chernikov * 15939490a627SAlexander V. Chernikov * Returns 0 on success 15949490a627SAlexander V. Chernikov */ 15959490a627SAlexander V. Chernikov int 1596ac35ff17SAlexander V. Chernikov ipfw_create_table(struct ip_fw_chain *ch, ip_fw3_opheader *op3, 1597ac35ff17SAlexander V. Chernikov struct sockopt_data *sd) 15989490a627SAlexander V. Chernikov { 15999490a627SAlexander V. Chernikov struct _ipfw_obj_header *oh; 16009490a627SAlexander V. Chernikov ipfw_xtable_info *i; 16019490a627SAlexander V. Chernikov char *tname, *aname; 16029490a627SAlexander V. Chernikov struct tid_info ti; 16039490a627SAlexander V. Chernikov struct namedobj_instance *ni; 16049490a627SAlexander V. Chernikov struct table_config *tc; 16059490a627SAlexander V. Chernikov 1606ac35ff17SAlexander V. Chernikov if (sd->valsize != sizeof(*oh) + sizeof(ipfw_xtable_info)) 16079490a627SAlexander V. Chernikov return (EINVAL); 16089490a627SAlexander V. Chernikov 1609ac35ff17SAlexander V. Chernikov oh = (struct _ipfw_obj_header *)sd->kbuf; 16109490a627SAlexander V. Chernikov i = (ipfw_xtable_info *)(oh + 1); 16119490a627SAlexander V. Chernikov 16129490a627SAlexander V. Chernikov /* 16139490a627SAlexander V. Chernikov * Verify user-supplied strings. 16142d99a349SAlexander V. Chernikov * Check for null-terminated/zero-length strings/ 16159490a627SAlexander V. Chernikov */ 1616ac35ff17SAlexander V. Chernikov tname = oh->ntlv.name; 16179490a627SAlexander V. Chernikov aname = i->algoname; 1618ac35ff17SAlexander V. Chernikov if (ipfw_check_table_name(tname) != 0 || 16199490a627SAlexander V. Chernikov strnlen(aname, sizeof(i->algoname)) == sizeof(i->algoname)) 16209490a627SAlexander V. Chernikov return (EINVAL); 16219490a627SAlexander V. Chernikov 16229490a627SAlexander V. Chernikov if (aname[0] == '\0') { 16239490a627SAlexander V. Chernikov /* Use default algorithm */ 16249490a627SAlexander V. Chernikov aname = NULL; 16259490a627SAlexander V. Chernikov } 16269490a627SAlexander V. Chernikov 16279490a627SAlexander V. Chernikov objheader_to_ti(oh, &ti); 1628ac35ff17SAlexander V. Chernikov ti.type = i->type; 16299490a627SAlexander V. Chernikov 16309490a627SAlexander V. Chernikov ni = CHAIN_TO_NI(ch); 16319490a627SAlexander V. Chernikov 16329490a627SAlexander V. Chernikov IPFW_UH_RLOCK(ch); 16339490a627SAlexander V. Chernikov if ((tc = find_table(ni, &ti)) != NULL) { 16349490a627SAlexander V. Chernikov IPFW_UH_RUNLOCK(ch); 16359490a627SAlexander V. Chernikov return (EEXIST); 16369490a627SAlexander V. Chernikov } 16379490a627SAlexander V. Chernikov IPFW_UH_RUNLOCK(ch); 16389490a627SAlexander V. Chernikov 16390468c5baSAlexander V. Chernikov return (create_table_internal(ch, &ti, aname, i, NULL, NULL, NULL, 0)); 1640db785d31SAlexander V. Chernikov } 1641db785d31SAlexander V. Chernikov 1642db785d31SAlexander V. Chernikov /* 1643db785d31SAlexander V. Chernikov * Creates new table based on @ti and @aname. 1644db785d31SAlexander V. Chernikov * 1645db785d31SAlexander V. Chernikov * Relies on table name checking inside find_name_tlv() 1646db785d31SAlexander V. Chernikov * Assume @aname to be checked and valid. 16470468c5baSAlexander V. Chernikov * Stores allocated table config, used algo and kidx 16480468c5baSAlexander V. Chernikov * inside @ptc, @pta and @pkidx (if non-NULL). 16490468c5baSAlexander V. Chernikov * Reference created table if @compat is non-zero. 1650db785d31SAlexander V. Chernikov * 1651db785d31SAlexander V. Chernikov * Returns 0 on success. 1652db785d31SAlexander V. Chernikov */ 1653db785d31SAlexander V. Chernikov static int 1654db785d31SAlexander V. Chernikov create_table_internal(struct ip_fw_chain *ch, struct tid_info *ti, 16550468c5baSAlexander V. Chernikov char *aname, ipfw_xtable_info *i, struct table_config **ptc, 16560468c5baSAlexander V. Chernikov struct table_algo **pta, uint16_t *pkidx, int compat) 1657db785d31SAlexander V. Chernikov { 1658db785d31SAlexander V. Chernikov struct namedobj_instance *ni; 16590468c5baSAlexander V. Chernikov struct table_config *tc, *tc_new, *tmp;; 1660db785d31SAlexander V. Chernikov struct table_algo *ta; 1661db785d31SAlexander V. Chernikov uint16_t kidx; 1662db785d31SAlexander V. Chernikov 1663db785d31SAlexander V. Chernikov ni = CHAIN_TO_NI(ch); 1664db785d31SAlexander V. Chernikov 1665db785d31SAlexander V. Chernikov ta = find_table_algo(CHAIN_TO_TCFG(ch), ti, aname); 16669490a627SAlexander V. Chernikov if (ta == NULL) 16679490a627SAlexander V. Chernikov return (ENOTSUP); 16689490a627SAlexander V. Chernikov 16694c0c07a5SAlexander V. Chernikov tc = alloc_table_config(ch, ti, ta, aname, i->tflags, i->vtype); 16704c0c07a5SAlexander V. Chernikov if (tc == NULL) 16719490a627SAlexander V. Chernikov return (ENOMEM); 16729490a627SAlexander V. Chernikov 1673adf3b2b9SAlexander V. Chernikov tc->vftype = i->vftype; 16744c0c07a5SAlexander V. Chernikov tc->limit = i->limit; 16754f43138aSAlexander V. Chernikov tc->locked = (i->flags & IPFW_TGFLAGS_LOCKED) != 0; 16764c0c07a5SAlexander V. Chernikov 16779490a627SAlexander V. Chernikov IPFW_UH_WLOCK(ch); 1678db785d31SAlexander V. Chernikov 1679db785d31SAlexander V. Chernikov /* Check if table has been already created */ 16800468c5baSAlexander V. Chernikov tc_new = find_table(ni, ti); 16810468c5baSAlexander V. Chernikov if (tc_new != NULL) { 16820468c5baSAlexander V. Chernikov 16830468c5baSAlexander V. Chernikov /* 16840468c5baSAlexander V. Chernikov * Compat: do not fail if we're 16850468c5baSAlexander V. Chernikov * requesting to create existing table 16860468c5baSAlexander V. Chernikov * which has the same type / vtype 16870468c5baSAlexander V. Chernikov */ 16880468c5baSAlexander V. Chernikov if (compat == 0 || tc_new->no.type != tc->no.type || 16890468c5baSAlexander V. Chernikov tc_new->vtype != tc->vtype) { 1690db785d31SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 1691db785d31SAlexander V. Chernikov free_table_config(ni, tc); 1692db785d31SAlexander V. Chernikov return (EEXIST); 1693db785d31SAlexander V. Chernikov } 1694db785d31SAlexander V. Chernikov 16950468c5baSAlexander V. Chernikov /* Exchange tc and tc_new for proper refcounting & freeing */ 16960468c5baSAlexander V. Chernikov tmp = tc; 16970468c5baSAlexander V. Chernikov tc = tc_new; 16980468c5baSAlexander V. Chernikov tc_new = tmp; 16990468c5baSAlexander V. Chernikov } else { 17000468c5baSAlexander V. Chernikov /* New table */ 1701ac35ff17SAlexander V. Chernikov if (ipfw_objhash_alloc_idx(ni, &kidx) != 0) { 17029490a627SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 1703db785d31SAlexander V. Chernikov printf("Unable to allocate table index." 1704db785d31SAlexander V. Chernikov " Consider increasing net.inet.ip.fw.tables_max"); 17059490a627SAlexander V. Chernikov free_table_config(ni, tc); 17069490a627SAlexander V. Chernikov return (EBUSY); 17079490a627SAlexander V. Chernikov } 17089490a627SAlexander V. Chernikov tc->no.kidx = kidx; 17099490a627SAlexander V. Chernikov 17109490a627SAlexander V. Chernikov IPFW_WLOCK(ch); 17119490a627SAlexander V. Chernikov link_table(ch, tc); 17129490a627SAlexander V. Chernikov IPFW_WUNLOCK(ch); 17130468c5baSAlexander V. Chernikov } 17140468c5baSAlexander V. Chernikov 17150468c5baSAlexander V. Chernikov if (compat != 0) 17160468c5baSAlexander V. Chernikov tc->no.refcnt++; 17170468c5baSAlexander V. Chernikov if (ptc != NULL) 17180468c5baSAlexander V. Chernikov *ptc = tc; 17190468c5baSAlexander V. Chernikov if (pta != NULL) 17200468c5baSAlexander V. Chernikov *pta = ta; 17210468c5baSAlexander V. Chernikov if (pkidx != NULL) 17220468c5baSAlexander V. Chernikov *pkidx = tc->no.kidx; 17239490a627SAlexander V. Chernikov 17249490a627SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 17259490a627SAlexander V. Chernikov 17260468c5baSAlexander V. Chernikov if (tc_new != NULL) 17270468c5baSAlexander V. Chernikov free_table_config(ni, tc_new); 17280468c5baSAlexander V. Chernikov 17299490a627SAlexander V. Chernikov return (0); 17309490a627SAlexander V. Chernikov } 17319490a627SAlexander V. Chernikov 173246d52008SAlexander V. Chernikov static void 173346d52008SAlexander V. Chernikov ntlv_to_ti(ipfw_obj_ntlv *ntlv, struct tid_info *ti) 1734d3a4f924SAlexander V. Chernikov { 1735d3a4f924SAlexander V. Chernikov 1736d3a4f924SAlexander V. Chernikov memset(ti, 0, sizeof(struct tid_info)); 173746d52008SAlexander V. Chernikov ti->set = ntlv->set; 173846d52008SAlexander V. Chernikov ti->uidx = ntlv->idx; 173946d52008SAlexander V. Chernikov ti->tlvs = ntlv; 174046d52008SAlexander V. Chernikov ti->tlen = ntlv->head.length; 174146d52008SAlexander V. Chernikov } 174246d52008SAlexander V. Chernikov 174346d52008SAlexander V. Chernikov static void 174446d52008SAlexander V. Chernikov objheader_to_ti(struct _ipfw_obj_header *oh, struct tid_info *ti) 174546d52008SAlexander V. Chernikov { 174646d52008SAlexander V. Chernikov 174746d52008SAlexander V. Chernikov ntlv_to_ti(&oh->ntlv, ti); 1748d3a4f924SAlexander V. Chernikov } 1749d3a4f924SAlexander V. Chernikov 175035e1bbd0SAlexander V. Chernikov /* 175135e1bbd0SAlexander V. Chernikov * Exports basic table info as name TLV. 175235e1bbd0SAlexander V. Chernikov * Used inside dump_static_rules() to provide info 175335e1bbd0SAlexander V. Chernikov * about all tables referenced by current ruleset. 175435e1bbd0SAlexander V. Chernikov * 175535e1bbd0SAlexander V. Chernikov * Returns 0 on success. 175635e1bbd0SAlexander V. Chernikov */ 1757563b5ab1SAlexander V. Chernikov int 1758563b5ab1SAlexander V. Chernikov ipfw_export_table_ntlv(struct ip_fw_chain *ch, uint16_t kidx, 1759563b5ab1SAlexander V. Chernikov struct sockopt_data *sd) 1760563b5ab1SAlexander V. Chernikov { 1761563b5ab1SAlexander V. Chernikov struct namedobj_instance *ni; 1762563b5ab1SAlexander V. Chernikov struct named_object *no; 1763563b5ab1SAlexander V. Chernikov ipfw_obj_ntlv *ntlv; 1764563b5ab1SAlexander V. Chernikov 1765563b5ab1SAlexander V. Chernikov ni = CHAIN_TO_NI(ch); 1766563b5ab1SAlexander V. Chernikov 1767ac35ff17SAlexander V. Chernikov no = ipfw_objhash_lookup_kidx(ni, kidx); 1768563b5ab1SAlexander V. Chernikov KASSERT(no != NULL, ("invalid table kidx passed")); 1769563b5ab1SAlexander V. Chernikov 1770563b5ab1SAlexander V. Chernikov ntlv = (ipfw_obj_ntlv *)ipfw_get_sopt_space(sd, sizeof(*ntlv)); 1771563b5ab1SAlexander V. Chernikov if (ntlv == NULL) 1772563b5ab1SAlexander V. Chernikov return (ENOMEM); 1773563b5ab1SAlexander V. Chernikov 1774563b5ab1SAlexander V. Chernikov ntlv->head.type = IPFW_TLV_TBL_NAME; 1775563b5ab1SAlexander V. Chernikov ntlv->head.length = sizeof(*ntlv); 1776563b5ab1SAlexander V. Chernikov ntlv->idx = no->kidx; 1777563b5ab1SAlexander V. Chernikov strlcpy(ntlv->name, no->name, sizeof(ntlv->name)); 1778563b5ab1SAlexander V. Chernikov 1779563b5ab1SAlexander V. Chernikov return (0); 1780563b5ab1SAlexander V. Chernikov } 1781563b5ab1SAlexander V. Chernikov 178235e1bbd0SAlexander V. Chernikov /* 1783f99fbf96SAlexander V. Chernikov * Marks every table kidx used in @rule with bit in @bmask. 1784f99fbf96SAlexander V. Chernikov * Used to generate bitmask of referenced tables for given ruleset. 1785f99fbf96SAlexander V. Chernikov * 1786f99fbf96SAlexander V. Chernikov * Returns number of newly-referenced tables. 1787f99fbf96SAlexander V. Chernikov */ 1788f99fbf96SAlexander V. Chernikov int 1789f99fbf96SAlexander V. Chernikov ipfw_mark_table_kidx(struct ip_fw_chain *chain, struct ip_fw *rule, 1790f99fbf96SAlexander V. Chernikov uint32_t *bmask) 1791f99fbf96SAlexander V. Chernikov { 1792f99fbf96SAlexander V. Chernikov int cmdlen, l, count; 1793f99fbf96SAlexander V. Chernikov ipfw_insn *cmd; 1794f99fbf96SAlexander V. Chernikov uint16_t kidx; 1795f99fbf96SAlexander V. Chernikov uint8_t type; 1796f99fbf96SAlexander V. Chernikov 1797f99fbf96SAlexander V. Chernikov l = rule->cmd_len; 1798f99fbf96SAlexander V. Chernikov cmd = rule->cmd; 1799f99fbf96SAlexander V. Chernikov cmdlen = 0; 1800f99fbf96SAlexander V. Chernikov count = 0; 1801f99fbf96SAlexander V. Chernikov for ( ; l > 0 ; l -= cmdlen, cmd += cmdlen) { 1802f99fbf96SAlexander V. Chernikov cmdlen = F_LEN(cmd); 1803f99fbf96SAlexander V. Chernikov 1804f99fbf96SAlexander V. Chernikov if (classify_table_opcode(cmd, &kidx, &type) != 0) 1805f99fbf96SAlexander V. Chernikov continue; 1806f99fbf96SAlexander V. Chernikov 1807f99fbf96SAlexander V. Chernikov if ((bmask[kidx / 32] & (1 << (kidx % 32))) == 0) 1808f99fbf96SAlexander V. Chernikov count++; 1809f99fbf96SAlexander V. Chernikov 1810f99fbf96SAlexander V. Chernikov bmask[kidx / 32] |= 1 << (kidx % 32); 1811f99fbf96SAlexander V. Chernikov } 1812f99fbf96SAlexander V. Chernikov 1813f99fbf96SAlexander V. Chernikov return (count); 1814f99fbf96SAlexander V. Chernikov } 1815f99fbf96SAlexander V. Chernikov 1816f99fbf96SAlexander V. Chernikov 1817f99fbf96SAlexander V. Chernikov /* 181835e1bbd0SAlexander V. Chernikov * Exports table @tc info into standard ipfw_xtable_info format. 181935e1bbd0SAlexander V. Chernikov */ 18209f7d47b0SAlexander V. Chernikov static void 1821ac35ff17SAlexander V. Chernikov export_table_info(struct ip_fw_chain *ch, struct table_config *tc, 1822ac35ff17SAlexander V. Chernikov ipfw_xtable_info *i) 18239f7d47b0SAlexander V. Chernikov { 1824ac35ff17SAlexander V. Chernikov struct table_info *ti; 18255f379342SAlexander V. Chernikov struct table_algo *ta; 18269f7d47b0SAlexander V. Chernikov 18279f7d47b0SAlexander V. Chernikov i->type = tc->no.type; 1828914bffb6SAlexander V. Chernikov i->tflags = tc->tflags; 1829ac35ff17SAlexander V. Chernikov i->vtype = tc->vtype; 1830adf3b2b9SAlexander V. Chernikov i->vftype = tc->vftype; 18319f7d47b0SAlexander V. Chernikov i->set = tc->no.set; 18329f7d47b0SAlexander V. Chernikov i->kidx = tc->no.kidx; 18339f7d47b0SAlexander V. Chernikov i->refcnt = tc->no.refcnt; 18349f7d47b0SAlexander V. Chernikov i->count = tc->count; 18354c0c07a5SAlexander V. Chernikov i->limit = tc->limit; 18364f43138aSAlexander V. Chernikov i->flags |= (tc->locked != 0) ? IPFW_TGFLAGS_LOCKED : 0; 183781d3153dSAlexander V. Chernikov i->size = tc->count * sizeof(ipfw_obj_tentry); 1838f1220db8SAlexander V. Chernikov i->size += sizeof(ipfw_obj_header) + sizeof(ipfw_xtable_info); 18399f7d47b0SAlexander V. Chernikov strlcpy(i->tablename, tc->tablename, sizeof(i->tablename)); 1840ac35ff17SAlexander V. Chernikov ti = KIDX_TO_TI(ch, tc->no.kidx); 18415f379342SAlexander V. Chernikov ta = tc->ta; 18425f379342SAlexander V. Chernikov if (ta->print_config != NULL) { 18435f379342SAlexander V. Chernikov /* Use algo function to print table config to string */ 18445f379342SAlexander V. Chernikov ta->print_config(tc->astate, ti, i->algoname, 1845ac35ff17SAlexander V. Chernikov sizeof(i->algoname)); 1846ac35ff17SAlexander V. Chernikov } else 18475f379342SAlexander V. Chernikov strlcpy(i->algoname, ta->name, sizeof(i->algoname)); 18485f379342SAlexander V. Chernikov /* Dump algo-specific data, if possible */ 18495f379342SAlexander V. Chernikov if (ta->dump_tinfo != NULL) { 18505f379342SAlexander V. Chernikov ta->dump_tinfo(tc->astate, ti, &i->ta_info); 18515f379342SAlexander V. Chernikov i->ta_info.flags |= IPFW_TATFLAGS_DATA; 18525f379342SAlexander V. Chernikov } 18539f7d47b0SAlexander V. Chernikov } 18549f7d47b0SAlexander V. Chernikov 1855ac35ff17SAlexander V. Chernikov struct dump_table_args { 1856ac35ff17SAlexander V. Chernikov struct ip_fw_chain *ch; 1857ac35ff17SAlexander V. Chernikov struct sockopt_data *sd; 1858ac35ff17SAlexander V. Chernikov }; 1859ac35ff17SAlexander V. Chernikov 18609f7d47b0SAlexander V. Chernikov static void 18619f7d47b0SAlexander V. Chernikov export_table_internal(struct namedobj_instance *ni, struct named_object *no, 18629f7d47b0SAlexander V. Chernikov void *arg) 18633b3a8eb9SGleb Smirnoff { 18649f7d47b0SAlexander V. Chernikov ipfw_xtable_info *i; 1865ac35ff17SAlexander V. Chernikov struct dump_table_args *dta; 18663b3a8eb9SGleb Smirnoff 1867ac35ff17SAlexander V. Chernikov dta = (struct dump_table_args *)arg; 1868ac35ff17SAlexander V. Chernikov 1869ac35ff17SAlexander V. Chernikov i = (ipfw_xtable_info *)ipfw_get_sopt_space(dta->sd, sizeof(*i)); 187068394ec8SAlexander V. Chernikov KASSERT(i != 0, ("previously checked buffer is not enough")); 18719f7d47b0SAlexander V. Chernikov 1872ac35ff17SAlexander V. Chernikov export_table_info(dta->ch, (struct table_config *)no, i); 18739f7d47b0SAlexander V. Chernikov } 18749f7d47b0SAlexander V. Chernikov 1875f1220db8SAlexander V. Chernikov /* 1876f1220db8SAlexander V. Chernikov * Export all tables as ipfw_xtable_info structures to 18772d99a349SAlexander V. Chernikov * storage provided by @sd. 187828ea4fa3SAlexander V. Chernikov * 187928ea4fa3SAlexander V. Chernikov * If supplied buffer is too small, fills in required size 188028ea4fa3SAlexander V. Chernikov * and returns ENOMEM. 1881f1220db8SAlexander V. Chernikov * Returns 0 on success. 1882f1220db8SAlexander V. Chernikov */ 1883f1220db8SAlexander V. Chernikov static int 18842d99a349SAlexander V. Chernikov export_tables(struct ip_fw_chain *ch, ipfw_obj_lheader *olh, 18852d99a349SAlexander V. Chernikov struct sockopt_data *sd) 18869f7d47b0SAlexander V. Chernikov { 18879f7d47b0SAlexander V. Chernikov uint32_t size; 18889f7d47b0SAlexander V. Chernikov uint32_t count; 1889ac35ff17SAlexander V. Chernikov struct dump_table_args dta; 18909f7d47b0SAlexander V. Chernikov 18919f7d47b0SAlexander V. Chernikov count = ipfw_objhash_count(CHAIN_TO_NI(ch)); 18929f7d47b0SAlexander V. Chernikov size = count * sizeof(ipfw_xtable_info) + sizeof(ipfw_obj_lheader); 18932d99a349SAlexander V. Chernikov 18942d99a349SAlexander V. Chernikov /* Fill in header regadless of buffer size */ 1895f1220db8SAlexander V. Chernikov olh->count = count; 1896f1220db8SAlexander V. Chernikov olh->objsize = sizeof(ipfw_xtable_info); 18972d99a349SAlexander V. Chernikov 18982d99a349SAlexander V. Chernikov if (size > olh->size) { 18992d99a349SAlexander V. Chernikov olh->size = size; 19009f7d47b0SAlexander V. Chernikov return (ENOMEM); 1901f1220db8SAlexander V. Chernikov } 190268394ec8SAlexander V. Chernikov 19039f7d47b0SAlexander V. Chernikov olh->size = size; 19042d99a349SAlexander V. Chernikov 1905ac35ff17SAlexander V. Chernikov dta.ch = ch; 1906ac35ff17SAlexander V. Chernikov dta.sd = sd; 1907ac35ff17SAlexander V. Chernikov 1908ac35ff17SAlexander V. Chernikov ipfw_objhash_foreach(CHAIN_TO_NI(ch), export_table_internal, &dta); 19099f7d47b0SAlexander V. Chernikov 19103b3a8eb9SGleb Smirnoff return (0); 19113b3a8eb9SGleb Smirnoff } 19123b3a8eb9SGleb Smirnoff 1913f99fbf96SAlexander V. Chernikov struct dump_args { 1914f99fbf96SAlexander V. Chernikov struct table_info *ti; 1915f99fbf96SAlexander V. Chernikov struct table_config *tc; 1916f99fbf96SAlexander V. Chernikov struct sockopt_data *sd; 1917f99fbf96SAlexander V. Chernikov uint32_t cnt; 1918f99fbf96SAlexander V. Chernikov uint16_t uidx; 1919f99fbf96SAlexander V. Chernikov int error; 1920f99fbf96SAlexander V. Chernikov ipfw_table_entry *ent; 1921f99fbf96SAlexander V. Chernikov uint32_t size; 1922f99fbf96SAlexander V. Chernikov ipfw_obj_tentry tent; 1923f99fbf96SAlexander V. Chernikov }; 1924f99fbf96SAlexander V. Chernikov 1925f99fbf96SAlexander V. Chernikov int 1926f99fbf96SAlexander V. Chernikov ipfw_dump_table(struct ip_fw_chain *ch, ip_fw3_opheader *op3, 1927f99fbf96SAlexander V. Chernikov struct sockopt_data *sd) 1928f99fbf96SAlexander V. Chernikov { 1929f99fbf96SAlexander V. Chernikov int error; 1930f99fbf96SAlexander V. Chernikov 1931f99fbf96SAlexander V. Chernikov switch (op3->version) { 1932f99fbf96SAlexander V. Chernikov case 0: 1933f99fbf96SAlexander V. Chernikov error = ipfw_dump_table_v0(ch, sd); 1934f99fbf96SAlexander V. Chernikov break; 1935f99fbf96SAlexander V. Chernikov case 1: 1936f99fbf96SAlexander V. Chernikov error = ipfw_dump_table_v1(ch, sd); 1937f99fbf96SAlexander V. Chernikov break; 1938f99fbf96SAlexander V. Chernikov default: 1939f99fbf96SAlexander V. Chernikov error = ENOTSUP; 1940f99fbf96SAlexander V. Chernikov } 1941f99fbf96SAlexander V. Chernikov 1942f99fbf96SAlexander V. Chernikov return (error); 1943f99fbf96SAlexander V. Chernikov } 1944f99fbf96SAlexander V. Chernikov 1945f99fbf96SAlexander V. Chernikov /* 1946f99fbf96SAlexander V. Chernikov * Dumps all table data 1947f99fbf96SAlexander V. Chernikov * Data layout (v1)(current): 1948f99fbf96SAlexander V. Chernikov * Request: [ ipfw_obj_header ], size = ipfw_xtable_info.size 1949f99fbf96SAlexander V. Chernikov * Reply: [ ipfw_obj_header ipfw_xtable_info ipfw_obj_tentry x N ] 1950f99fbf96SAlexander V. Chernikov * 1951f99fbf96SAlexander V. Chernikov * Returns 0 on success 1952f99fbf96SAlexander V. Chernikov */ 1953f99fbf96SAlexander V. Chernikov static int 1954f99fbf96SAlexander V. Chernikov ipfw_dump_table_v1(struct ip_fw_chain *ch, struct sockopt_data *sd) 1955f99fbf96SAlexander V. Chernikov { 1956f99fbf96SAlexander V. Chernikov struct _ipfw_obj_header *oh; 1957f99fbf96SAlexander V. Chernikov ipfw_xtable_info *i; 1958f99fbf96SAlexander V. Chernikov struct tid_info ti; 1959f99fbf96SAlexander V. Chernikov struct table_config *tc; 1960f99fbf96SAlexander V. Chernikov struct table_algo *ta; 1961f99fbf96SAlexander V. Chernikov struct dump_args da; 1962f99fbf96SAlexander V. Chernikov uint32_t sz; 1963f99fbf96SAlexander V. Chernikov 1964f99fbf96SAlexander V. Chernikov sz = sizeof(ipfw_obj_header) + sizeof(ipfw_xtable_info); 1965f99fbf96SAlexander V. Chernikov oh = (struct _ipfw_obj_header *)ipfw_get_sopt_header(sd, sz); 1966f99fbf96SAlexander V. Chernikov if (oh == NULL) 1967f99fbf96SAlexander V. Chernikov return (EINVAL); 1968f99fbf96SAlexander V. Chernikov 1969f99fbf96SAlexander V. Chernikov i = (ipfw_xtable_info *)(oh + 1); 1970f99fbf96SAlexander V. Chernikov objheader_to_ti(oh, &ti); 1971f99fbf96SAlexander V. Chernikov 1972f99fbf96SAlexander V. Chernikov IPFW_UH_RLOCK(ch); 1973f99fbf96SAlexander V. Chernikov if ((tc = find_table(CHAIN_TO_NI(ch), &ti)) == NULL) { 1974f99fbf96SAlexander V. Chernikov IPFW_UH_RUNLOCK(ch); 1975f99fbf96SAlexander V. Chernikov return (ESRCH); 1976f99fbf96SAlexander V. Chernikov } 1977f99fbf96SAlexander V. Chernikov export_table_info(ch, tc, i); 1978f99fbf96SAlexander V. Chernikov 1979f99fbf96SAlexander V. Chernikov if (sd->valsize < i->size) { 1980f99fbf96SAlexander V. Chernikov 1981f99fbf96SAlexander V. Chernikov /* 1982f99fbf96SAlexander V. Chernikov * Submitted buffer size is not enough. 1983f99fbf96SAlexander V. Chernikov * WE've already filled in @i structure with 1984f99fbf96SAlexander V. Chernikov * relevant table info including size, so we 1985f99fbf96SAlexander V. Chernikov * can return. Buffer will be flushed automatically. 1986f99fbf96SAlexander V. Chernikov */ 1987f99fbf96SAlexander V. Chernikov IPFW_UH_RUNLOCK(ch); 1988f99fbf96SAlexander V. Chernikov return (ENOMEM); 1989f99fbf96SAlexander V. Chernikov } 1990f99fbf96SAlexander V. Chernikov 1991f99fbf96SAlexander V. Chernikov /* 1992f99fbf96SAlexander V. Chernikov * Do the actual dump in eXtended format 1993f99fbf96SAlexander V. Chernikov */ 1994f99fbf96SAlexander V. Chernikov memset(&da, 0, sizeof(da)); 1995f99fbf96SAlexander V. Chernikov da.ti = KIDX_TO_TI(ch, tc->no.kidx); 1996f99fbf96SAlexander V. Chernikov da.tc = tc; 1997f99fbf96SAlexander V. Chernikov da.sd = sd; 1998f99fbf96SAlexander V. Chernikov 1999f99fbf96SAlexander V. Chernikov ta = tc->ta; 2000f99fbf96SAlexander V. Chernikov 2001f99fbf96SAlexander V. Chernikov ta->foreach(tc->astate, da.ti, dump_table_tentry, &da); 2002f99fbf96SAlexander V. Chernikov IPFW_UH_RUNLOCK(ch); 2003f99fbf96SAlexander V. Chernikov 2004f99fbf96SAlexander V. Chernikov return (da.error); 2005f99fbf96SAlexander V. Chernikov } 2006f99fbf96SAlexander V. Chernikov 2007f99fbf96SAlexander V. Chernikov /* 2008f99fbf96SAlexander V. Chernikov * Dumps all table data 2009f99fbf96SAlexander V. Chernikov * Data layout (version 0)(legacy): 2010f99fbf96SAlexander V. Chernikov * Request: [ ipfw_xtable ], size = IP_FW_TABLE_XGETSIZE() 2011f99fbf96SAlexander V. Chernikov * Reply: [ ipfw_xtable ipfw_table_xentry x N ] 2012f99fbf96SAlexander V. Chernikov * 2013f99fbf96SAlexander V. Chernikov * Returns 0 on success 2014f99fbf96SAlexander V. Chernikov */ 2015f99fbf96SAlexander V. Chernikov static int 2016f99fbf96SAlexander V. Chernikov ipfw_dump_table_v0(struct ip_fw_chain *ch, struct sockopt_data *sd) 2017f99fbf96SAlexander V. Chernikov { 2018f99fbf96SAlexander V. Chernikov ipfw_xtable *xtbl; 2019f99fbf96SAlexander V. Chernikov struct tid_info ti; 2020f99fbf96SAlexander V. Chernikov struct table_config *tc; 2021f99fbf96SAlexander V. Chernikov struct table_algo *ta; 2022f99fbf96SAlexander V. Chernikov struct dump_args da; 2023f99fbf96SAlexander V. Chernikov size_t sz; 2024f99fbf96SAlexander V. Chernikov 2025f99fbf96SAlexander V. Chernikov xtbl = (ipfw_xtable *)ipfw_get_sopt_header(sd, sizeof(ipfw_xtable)); 2026f99fbf96SAlexander V. Chernikov if (xtbl == NULL) 2027f99fbf96SAlexander V. Chernikov return (EINVAL); 2028f99fbf96SAlexander V. Chernikov 2029f99fbf96SAlexander V. Chernikov memset(&ti, 0, sizeof(ti)); 2030f99fbf96SAlexander V. Chernikov ti.uidx = xtbl->tbl; 2031f99fbf96SAlexander V. Chernikov 2032f99fbf96SAlexander V. Chernikov IPFW_UH_RLOCK(ch); 2033f99fbf96SAlexander V. Chernikov if ((tc = find_table(CHAIN_TO_NI(ch), &ti)) == NULL) { 2034f99fbf96SAlexander V. Chernikov IPFW_UH_RUNLOCK(ch); 2035f99fbf96SAlexander V. Chernikov return (0); 2036f99fbf96SAlexander V. Chernikov } 2037f99fbf96SAlexander V. Chernikov sz = tc->count * sizeof(ipfw_table_xentry) + sizeof(ipfw_xtable); 2038f99fbf96SAlexander V. Chernikov 2039f99fbf96SAlexander V. Chernikov xtbl->cnt = tc->count; 2040f99fbf96SAlexander V. Chernikov xtbl->size = sz; 2041f99fbf96SAlexander V. Chernikov xtbl->type = tc->no.type; 2042f99fbf96SAlexander V. Chernikov xtbl->tbl = ti.uidx; 2043f99fbf96SAlexander V. Chernikov 2044f99fbf96SAlexander V. Chernikov if (sd->valsize < sz) { 2045f99fbf96SAlexander V. Chernikov 2046f99fbf96SAlexander V. Chernikov /* 2047f99fbf96SAlexander V. Chernikov * Submitted buffer size is not enough. 2048f99fbf96SAlexander V. Chernikov * WE've already filled in @i structure with 2049f99fbf96SAlexander V. Chernikov * relevant table info including size, so we 2050f99fbf96SAlexander V. Chernikov * can return. Buffer will be flushed automatically. 2051f99fbf96SAlexander V. Chernikov */ 2052f99fbf96SAlexander V. Chernikov IPFW_UH_RUNLOCK(ch); 2053f99fbf96SAlexander V. Chernikov return (ENOMEM); 2054f99fbf96SAlexander V. Chernikov } 2055f99fbf96SAlexander V. Chernikov 2056f99fbf96SAlexander V. Chernikov /* Do the actual dump in eXtended format */ 2057f99fbf96SAlexander V. Chernikov memset(&da, 0, sizeof(da)); 2058f99fbf96SAlexander V. Chernikov da.ti = KIDX_TO_TI(ch, tc->no.kidx); 2059f99fbf96SAlexander V. Chernikov da.tc = tc; 2060f99fbf96SAlexander V. Chernikov da.sd = sd; 2061f99fbf96SAlexander V. Chernikov 2062f99fbf96SAlexander V. Chernikov ta = tc->ta; 2063f99fbf96SAlexander V. Chernikov 2064f99fbf96SAlexander V. Chernikov ta->foreach(tc->astate, da.ti, dump_table_xentry, &da); 2065f99fbf96SAlexander V. Chernikov IPFW_UH_RUNLOCK(ch); 2066f99fbf96SAlexander V. Chernikov 2067f99fbf96SAlexander V. Chernikov return (0); 2068f99fbf96SAlexander V. Chernikov } 2069f99fbf96SAlexander V. Chernikov 2070f99fbf96SAlexander V. Chernikov 207181d3153dSAlexander V. Chernikov /* 207281d3153dSAlexander V. Chernikov * Legacy IP_FW_TABLE_GETSIZE handler 207381d3153dSAlexander V. Chernikov */ 20743b3a8eb9SGleb Smirnoff int 2075b074b7bbSAlexander V. Chernikov ipfw_count_table(struct ip_fw_chain *ch, struct tid_info *ti, uint32_t *cnt) 20763b3a8eb9SGleb Smirnoff { 2077b074b7bbSAlexander V. Chernikov struct table_config *tc; 20783b3a8eb9SGleb Smirnoff 2079b074b7bbSAlexander V. Chernikov if ((tc = find_table(CHAIN_TO_NI(ch), ti)) == NULL) 2080b074b7bbSAlexander V. Chernikov return (ESRCH); 20819f7d47b0SAlexander V. Chernikov *cnt = tc->count; 20823b3a8eb9SGleb Smirnoff return (0); 20833b3a8eb9SGleb Smirnoff } 20843b3a8eb9SGleb Smirnoff 208581d3153dSAlexander V. Chernikov /* 208681d3153dSAlexander V. Chernikov * Legacy IP_FW_TABLE_XGETSIZE handler 208781d3153dSAlexander V. Chernikov */ 20889f7d47b0SAlexander V. Chernikov int 20899f7d47b0SAlexander V. Chernikov ipfw_count_xtable(struct ip_fw_chain *ch, struct tid_info *ti, uint32_t *cnt) 20903b3a8eb9SGleb Smirnoff { 20919f7d47b0SAlexander V. Chernikov struct table_config *tc; 20929f7d47b0SAlexander V. Chernikov 2093d3a4f924SAlexander V. Chernikov if ((tc = find_table(CHAIN_TO_NI(ch), ti)) == NULL) { 2094d3a4f924SAlexander V. Chernikov *cnt = 0; 20959f7d47b0SAlexander V. Chernikov return (0); /* 'table all list' requires success */ 2096d3a4f924SAlexander V. Chernikov } 20979f7d47b0SAlexander V. Chernikov *cnt = tc->count * sizeof(ipfw_table_xentry); 20989f7d47b0SAlexander V. Chernikov if (tc->count > 0) 20999f7d47b0SAlexander V. Chernikov *cnt += sizeof(ipfw_xtable); 21009f7d47b0SAlexander V. Chernikov return (0); 21019f7d47b0SAlexander V. Chernikov } 21029f7d47b0SAlexander V. Chernikov 21039f7d47b0SAlexander V. Chernikov static int 21049f7d47b0SAlexander V. Chernikov dump_table_entry(void *e, void *arg) 21059f7d47b0SAlexander V. Chernikov { 21069f7d47b0SAlexander V. Chernikov struct dump_args *da; 21079f7d47b0SAlexander V. Chernikov struct table_config *tc; 21089f7d47b0SAlexander V. Chernikov struct table_algo *ta; 21093b3a8eb9SGleb Smirnoff ipfw_table_entry *ent; 211081d3153dSAlexander V. Chernikov int error; 21113b3a8eb9SGleb Smirnoff 21129f7d47b0SAlexander V. Chernikov da = (struct dump_args *)arg; 21139f7d47b0SAlexander V. Chernikov 21149f7d47b0SAlexander V. Chernikov tc = da->tc; 21159f7d47b0SAlexander V. Chernikov ta = tc->ta; 21169f7d47b0SAlexander V. Chernikov 21179f7d47b0SAlexander V. Chernikov /* Out of memory, returning */ 2118f1220db8SAlexander V. Chernikov if (da->cnt == da->size) 21193b3a8eb9SGleb Smirnoff return (1); 2120f1220db8SAlexander V. Chernikov ent = da->ent++; 2121f1220db8SAlexander V. Chernikov ent->tbl = da->uidx; 2122f1220db8SAlexander V. Chernikov da->cnt++; 21239f7d47b0SAlexander V. Chernikov 212481d3153dSAlexander V. Chernikov error = ta->dump_tentry(tc->astate, da->ti, e, &da->tent); 212581d3153dSAlexander V. Chernikov if (error != 0) 212681d3153dSAlexander V. Chernikov return (error); 212781d3153dSAlexander V. Chernikov 212881d3153dSAlexander V. Chernikov ent->addr = da->tent.k.addr.s_addr; 212981d3153dSAlexander V. Chernikov ent->masklen = da->tent.masklen; 213081d3153dSAlexander V. Chernikov ent->value = da->tent.value; 213181d3153dSAlexander V. Chernikov 213281d3153dSAlexander V. Chernikov return (0); 21333b3a8eb9SGleb Smirnoff } 21343b3a8eb9SGleb Smirnoff 2135f1220db8SAlexander V. Chernikov /* 2136f1220db8SAlexander V. Chernikov * Dumps table in pre-8.1 legacy format. 2137f1220db8SAlexander V. Chernikov */ 21383b3a8eb9SGleb Smirnoff int 2139f1220db8SAlexander V. Chernikov ipfw_dump_table_legacy(struct ip_fw_chain *ch, struct tid_info *ti, 2140f1220db8SAlexander V. Chernikov ipfw_table *tbl) 21413b3a8eb9SGleb Smirnoff { 2142b074b7bbSAlexander V. Chernikov struct table_config *tc; 21439f7d47b0SAlexander V. Chernikov struct table_algo *ta; 21449f7d47b0SAlexander V. Chernikov struct dump_args da; 21453b3a8eb9SGleb Smirnoff 21463b3a8eb9SGleb Smirnoff tbl->cnt = 0; 21473b3a8eb9SGleb Smirnoff 2148b074b7bbSAlexander V. Chernikov if ((tc = find_table(CHAIN_TO_NI(ch), ti)) == NULL) 2149b074b7bbSAlexander V. Chernikov return (0); /* XXX: We should return ESRCH */ 21509f7d47b0SAlexander V. Chernikov 21519f7d47b0SAlexander V. Chernikov ta = tc->ta; 21529f7d47b0SAlexander V. Chernikov 215381d3153dSAlexander V. Chernikov /* This dump format supports IPv4 only */ 215481d3153dSAlexander V. Chernikov if (tc->no.type != IPFW_TABLE_CIDR) 215581d3153dSAlexander V. Chernikov return (0); 21569f7d47b0SAlexander V. Chernikov 2157d3a4f924SAlexander V. Chernikov memset(&da, 0, sizeof(da)); 21589f7d47b0SAlexander V. Chernikov da.ti = KIDX_TO_TI(ch, tc->no.kidx); 21599f7d47b0SAlexander V. Chernikov da.tc = tc; 2160f1220db8SAlexander V. Chernikov da.ent = &tbl->ent[0]; 2161f1220db8SAlexander V. Chernikov da.size = tbl->size; 21629f7d47b0SAlexander V. Chernikov 21639f7d47b0SAlexander V. Chernikov tbl->cnt = 0; 21649f7d47b0SAlexander V. Chernikov ta->foreach(tc->astate, da.ti, dump_table_entry, &da); 2165f1220db8SAlexander V. Chernikov tbl->cnt = da.cnt; 21669f7d47b0SAlexander V. Chernikov 21673b3a8eb9SGleb Smirnoff return (0); 21683b3a8eb9SGleb Smirnoff } 21693b3a8eb9SGleb Smirnoff 21709490a627SAlexander V. Chernikov /* 217181d3153dSAlexander V. Chernikov * Dumps table entry in eXtended format (v1)(current). 217281d3153dSAlexander V. Chernikov */ 217381d3153dSAlexander V. Chernikov static int 217481d3153dSAlexander V. Chernikov dump_table_tentry(void *e, void *arg) 217581d3153dSAlexander V. Chernikov { 217681d3153dSAlexander V. Chernikov struct dump_args *da; 217781d3153dSAlexander V. Chernikov struct table_config *tc; 217881d3153dSAlexander V. Chernikov struct table_algo *ta; 217981d3153dSAlexander V. Chernikov ipfw_obj_tentry *tent; 218081d3153dSAlexander V. Chernikov 218181d3153dSAlexander V. Chernikov da = (struct dump_args *)arg; 218281d3153dSAlexander V. Chernikov 218381d3153dSAlexander V. Chernikov tc = da->tc; 218481d3153dSAlexander V. Chernikov ta = tc->ta; 218581d3153dSAlexander V. Chernikov 218681d3153dSAlexander V. Chernikov tent = (ipfw_obj_tentry *)ipfw_get_sopt_space(da->sd, sizeof(*tent)); 218781d3153dSAlexander V. Chernikov /* Out of memory, returning */ 218881d3153dSAlexander V. Chernikov if (tent == NULL) { 218981d3153dSAlexander V. Chernikov da->error = ENOMEM; 219081d3153dSAlexander V. Chernikov return (1); 219181d3153dSAlexander V. Chernikov } 219281d3153dSAlexander V. Chernikov tent->head.length = sizeof(ipfw_obj_tentry); 219381d3153dSAlexander V. Chernikov tent->idx = da->uidx; 219481d3153dSAlexander V. Chernikov 219581d3153dSAlexander V. Chernikov return (ta->dump_tentry(tc->astate, da->ti, e, tent)); 219681d3153dSAlexander V. Chernikov } 219781d3153dSAlexander V. Chernikov 219881d3153dSAlexander V. Chernikov /* 219981d3153dSAlexander V. Chernikov * Dumps table entry in eXtended format (v0). 22009490a627SAlexander V. Chernikov */ 22013b3a8eb9SGleb Smirnoff static int 22029f7d47b0SAlexander V. Chernikov dump_table_xentry(void *e, void *arg) 22033b3a8eb9SGleb Smirnoff { 22049f7d47b0SAlexander V. Chernikov struct dump_args *da; 22059f7d47b0SAlexander V. Chernikov struct table_config *tc; 22069f7d47b0SAlexander V. Chernikov struct table_algo *ta; 22073b3a8eb9SGleb Smirnoff ipfw_table_xentry *xent; 220881d3153dSAlexander V. Chernikov ipfw_obj_tentry *tent; 220981d3153dSAlexander V. Chernikov int error; 22103b3a8eb9SGleb Smirnoff 22119f7d47b0SAlexander V. Chernikov da = (struct dump_args *)arg; 22129f7d47b0SAlexander V. Chernikov 22139f7d47b0SAlexander V. Chernikov tc = da->tc; 22149f7d47b0SAlexander V. Chernikov ta = tc->ta; 22159f7d47b0SAlexander V. Chernikov 22162d99a349SAlexander V. Chernikov xent = (ipfw_table_xentry *)ipfw_get_sopt_space(da->sd, sizeof(*xent)); 22173b3a8eb9SGleb Smirnoff /* Out of memory, returning */ 22182d99a349SAlexander V. Chernikov if (xent == NULL) 22193b3a8eb9SGleb Smirnoff return (1); 22203b3a8eb9SGleb Smirnoff xent->len = sizeof(ipfw_table_xentry); 2221f1220db8SAlexander V. Chernikov xent->tbl = da->uidx; 22229f7d47b0SAlexander V. Chernikov 222381d3153dSAlexander V. Chernikov memset(&da->tent, 0, sizeof(da->tent)); 222481d3153dSAlexander V. Chernikov tent = &da->tent; 222581d3153dSAlexander V. Chernikov error = ta->dump_tentry(tc->astate, da->ti, e, tent); 222681d3153dSAlexander V. Chernikov if (error != 0) 222781d3153dSAlexander V. Chernikov return (error); 222881d3153dSAlexander V. Chernikov 222981d3153dSAlexander V. Chernikov /* Convert current format to previous one */ 223081d3153dSAlexander V. Chernikov xent->masklen = tent->masklen; 223181d3153dSAlexander V. Chernikov xent->value = tent->value; 223281d3153dSAlexander V. Chernikov /* Apply some hacks */ 223381d3153dSAlexander V. Chernikov if (tc->no.type == IPFW_TABLE_CIDR && tent->subtype == AF_INET) { 223481d3153dSAlexander V. Chernikov xent->k.addr6.s6_addr32[3] = tent->k.addr.s_addr; 223581d3153dSAlexander V. Chernikov xent->flags = IPFW_TCF_INET; 223681d3153dSAlexander V. Chernikov } else 223781d3153dSAlexander V. Chernikov memcpy(&xent->k, &tent->k, sizeof(xent->k)); 223881d3153dSAlexander V. Chernikov 223981d3153dSAlexander V. Chernikov return (0); 22409f7d47b0SAlexander V. Chernikov } 22419f7d47b0SAlexander V. Chernikov 22429f7d47b0SAlexander V. Chernikov /* 22439f7d47b0SAlexander V. Chernikov * Table algorithms 22449f7d47b0SAlexander V. Chernikov */ 22453b3a8eb9SGleb Smirnoff 22469f7d47b0SAlexander V. Chernikov /* 224735e1bbd0SAlexander V. Chernikov * Finds algoritm by index, table type or supplied name. 224835e1bbd0SAlexander V. Chernikov * 224935e1bbd0SAlexander V. Chernikov * Returns pointer to algo or NULL. 22509f7d47b0SAlexander V. Chernikov */ 22519f7d47b0SAlexander V. Chernikov static struct table_algo * 22529490a627SAlexander V. Chernikov find_table_algo(struct tables_config *tcfg, struct tid_info *ti, char *name) 22539f7d47b0SAlexander V. Chernikov { 22549490a627SAlexander V. Chernikov int i, l; 22559490a627SAlexander V. Chernikov struct table_algo *ta; 22569f7d47b0SAlexander V. Chernikov 225757a1cf95SAlexander V. Chernikov if (ti->type > IPFW_TABLE_MAXTYPE) 225857a1cf95SAlexander V. Chernikov return (NULL); 225957a1cf95SAlexander V. Chernikov 22609f7d47b0SAlexander V. Chernikov /* Search by index */ 22619f7d47b0SAlexander V. Chernikov if (ti->atype != 0) { 22629f7d47b0SAlexander V. Chernikov if (ti->atype > tcfg->algo_count) 22639f7d47b0SAlexander V. Chernikov return (NULL); 22649f7d47b0SAlexander V. Chernikov return (tcfg->algo[ti->atype]); 22659f7d47b0SAlexander V. Chernikov } 22669f7d47b0SAlexander V. Chernikov 22679490a627SAlexander V. Chernikov /* Search by name if supplied */ 22689490a627SAlexander V. Chernikov if (name != NULL) { 22699490a627SAlexander V. Chernikov /* TODO: better search */ 22709490a627SAlexander V. Chernikov for (i = 1; i <= tcfg->algo_count; i++) { 22719490a627SAlexander V. Chernikov ta = tcfg->algo[i]; 22729490a627SAlexander V. Chernikov 22739490a627SAlexander V. Chernikov /* 22749490a627SAlexander V. Chernikov * One can supply additional algorithm 22759490a627SAlexander V. Chernikov * parameters so we compare only the first word 22769490a627SAlexander V. Chernikov * of supplied name: 22779490a627SAlexander V. Chernikov * 'hash_cidr hsize=32' 22789490a627SAlexander V. Chernikov * '^^^^^^^^^' 22799490a627SAlexander V. Chernikov * 22809490a627SAlexander V. Chernikov */ 22819490a627SAlexander V. Chernikov l = strlen(ta->name); 22829490a627SAlexander V. Chernikov if (strncmp(name, ta->name, l) == 0) { 22839490a627SAlexander V. Chernikov if (name[l] == '\0' || name[l] == ' ') 22849490a627SAlexander V. Chernikov return (ta); 22859490a627SAlexander V. Chernikov } 22869490a627SAlexander V. Chernikov } 22879490a627SAlexander V. Chernikov 22889490a627SAlexander V. Chernikov return (NULL); 22899490a627SAlexander V. Chernikov } 22909490a627SAlexander V. Chernikov 229157a1cf95SAlexander V. Chernikov /* Return default algorithm for given type if set */ 229257a1cf95SAlexander V. Chernikov return (tcfg->def_algo[ti->type]); 22933b3a8eb9SGleb Smirnoff } 22943b3a8eb9SGleb Smirnoff 2295b6ee846eSAlexander V. Chernikov /* 2296b6ee846eSAlexander V. Chernikov * Register new table algo @ta. 229735e1bbd0SAlexander V. Chernikov * Stores algo id inside @idx. 2298b6ee846eSAlexander V. Chernikov * 2299b6ee846eSAlexander V. Chernikov * Returns 0 on success. 2300b6ee846eSAlexander V. Chernikov */ 23010b565ac0SAlexander V. Chernikov int 23020b565ac0SAlexander V. Chernikov ipfw_add_table_algo(struct ip_fw_chain *ch, struct table_algo *ta, size_t size, 23030b565ac0SAlexander V. Chernikov int *idx) 23043b3a8eb9SGleb Smirnoff { 23059f7d47b0SAlexander V. Chernikov struct tables_config *tcfg; 23060b565ac0SAlexander V. Chernikov struct table_algo *ta_new; 2307b6ee846eSAlexander V. Chernikov size_t sz; 23080b565ac0SAlexander V. Chernikov 23090b565ac0SAlexander V. Chernikov if (size > sizeof(struct table_algo)) 23100b565ac0SAlexander V. Chernikov return (EINVAL); 23110b565ac0SAlexander V. Chernikov 2312b6ee846eSAlexander V. Chernikov /* Check for the required on-stack size for add/del */ 2313b6ee846eSAlexander V. Chernikov sz = roundup2(ta->ta_buf_size, sizeof(void *)); 2314b6ee846eSAlexander V. Chernikov if (sz > TA_BUF_SZ) 2315b6ee846eSAlexander V. Chernikov return (EINVAL); 2316b6ee846eSAlexander V. Chernikov 231757a1cf95SAlexander V. Chernikov KASSERT(ta->type >= IPFW_TABLE_MAXTYPE,("Increase IPFW_TABLE_MAXTYPE")); 231857a1cf95SAlexander V. Chernikov 231935e1bbd0SAlexander V. Chernikov /* Copy algorithm data to stable storage. */ 23200b565ac0SAlexander V. Chernikov ta_new = malloc(sizeof(struct table_algo), M_IPFW, M_WAITOK | M_ZERO); 23210b565ac0SAlexander V. Chernikov memcpy(ta_new, ta, size); 23223b3a8eb9SGleb Smirnoff 23239f7d47b0SAlexander V. Chernikov tcfg = CHAIN_TO_TCFG(ch); 2324b074b7bbSAlexander V. Chernikov 23259f7d47b0SAlexander V. Chernikov KASSERT(tcfg->algo_count < 255, ("Increase algo array size")); 23269f7d47b0SAlexander V. Chernikov 23270b565ac0SAlexander V. Chernikov tcfg->algo[++tcfg->algo_count] = ta_new; 23280b565ac0SAlexander V. Chernikov ta_new->idx = tcfg->algo_count; 23290b565ac0SAlexander V. Chernikov 233057a1cf95SAlexander V. Chernikov /* Set algorithm as default one for given type */ 233157a1cf95SAlexander V. Chernikov if ((ta_new->flags & TA_FLAG_DEFAULT) != 0 && 233257a1cf95SAlexander V. Chernikov tcfg->def_algo[ta_new->type] == NULL) 233357a1cf95SAlexander V. Chernikov tcfg->def_algo[ta_new->type] = ta_new; 233457a1cf95SAlexander V. Chernikov 23350b565ac0SAlexander V. Chernikov *idx = ta_new->idx; 23360b565ac0SAlexander V. Chernikov 23370b565ac0SAlexander V. Chernikov return (0); 23380b565ac0SAlexander V. Chernikov } 23390b565ac0SAlexander V. Chernikov 2340b6ee846eSAlexander V. Chernikov /* 2341b6ee846eSAlexander V. Chernikov * Unregisters table algo using @idx as id. 2342b6ee846eSAlexander V. Chernikov */ 23430b565ac0SAlexander V. Chernikov void 23440b565ac0SAlexander V. Chernikov ipfw_del_table_algo(struct ip_fw_chain *ch, int idx) 23450b565ac0SAlexander V. Chernikov { 23460b565ac0SAlexander V. Chernikov struct tables_config *tcfg; 23470b565ac0SAlexander V. Chernikov struct table_algo *ta; 23480b565ac0SAlexander V. Chernikov 23490b565ac0SAlexander V. Chernikov tcfg = CHAIN_TO_TCFG(ch); 23500b565ac0SAlexander V. Chernikov 2351b6ee846eSAlexander V. Chernikov KASSERT(idx <= tcfg->algo_count, ("algo idx %d out of range 1..%d", 2352b6ee846eSAlexander V. Chernikov idx, tcfg->algo_count)); 23530b565ac0SAlexander V. Chernikov 23540b565ac0SAlexander V. Chernikov ta = tcfg->algo[idx]; 23550b565ac0SAlexander V. Chernikov KASSERT(ta != NULL, ("algo idx %d is NULL", idx)); 235657a1cf95SAlexander V. Chernikov 235757a1cf95SAlexander V. Chernikov if (tcfg->def_algo[ta->type] == ta) 235857a1cf95SAlexander V. Chernikov tcfg->def_algo[ta->type] = NULL; 235957a1cf95SAlexander V. Chernikov 23600b565ac0SAlexander V. Chernikov free(ta, M_IPFW); 23613b3a8eb9SGleb Smirnoff } 23623b3a8eb9SGleb Smirnoff 23639d099b4fSAlexander V. Chernikov /* 23649d099b4fSAlexander V. Chernikov * Lists all table algorithms currently available. 23659d099b4fSAlexander V. Chernikov * Data layout (v0)(current): 23669d099b4fSAlexander V. Chernikov * Request: [ ipfw_obj_lheader ], size = ipfw_obj_lheader.size 23679d099b4fSAlexander V. Chernikov * Reply: [ ipfw_obj_lheader ipfw_ta_info x N ] 23689d099b4fSAlexander V. Chernikov * 23699d099b4fSAlexander V. Chernikov * Returns 0 on success 23709d099b4fSAlexander V. Chernikov */ 23719d099b4fSAlexander V. Chernikov int 23729d099b4fSAlexander V. Chernikov ipfw_list_table_algo(struct ip_fw_chain *ch, struct sockopt_data *sd) 23739d099b4fSAlexander V. Chernikov { 23749d099b4fSAlexander V. Chernikov struct _ipfw_obj_lheader *olh; 23759d099b4fSAlexander V. Chernikov struct tables_config *tcfg; 23769d099b4fSAlexander V. Chernikov ipfw_ta_info *i; 23779d099b4fSAlexander V. Chernikov struct table_algo *ta; 23789d099b4fSAlexander V. Chernikov uint32_t count, n, size; 23799d099b4fSAlexander V. Chernikov 23809d099b4fSAlexander V. Chernikov olh = (struct _ipfw_obj_lheader *)ipfw_get_sopt_header(sd,sizeof(*olh)); 23819d099b4fSAlexander V. Chernikov if (olh == NULL) 23829d099b4fSAlexander V. Chernikov return (EINVAL); 23839d099b4fSAlexander V. Chernikov if (sd->valsize < olh->size) 23849d099b4fSAlexander V. Chernikov return (EINVAL); 23859d099b4fSAlexander V. Chernikov 23869d099b4fSAlexander V. Chernikov IPFW_UH_RLOCK(ch); 23879d099b4fSAlexander V. Chernikov tcfg = CHAIN_TO_TCFG(ch); 23889d099b4fSAlexander V. Chernikov count = tcfg->algo_count; 23899d099b4fSAlexander V. Chernikov size = count * sizeof(ipfw_ta_info) + sizeof(ipfw_obj_lheader); 23909d099b4fSAlexander V. Chernikov 23919d099b4fSAlexander V. Chernikov /* Fill in header regadless of buffer size */ 23929d099b4fSAlexander V. Chernikov olh->count = count; 23939d099b4fSAlexander V. Chernikov olh->objsize = sizeof(ipfw_ta_info); 23949d099b4fSAlexander V. Chernikov 23959d099b4fSAlexander V. Chernikov if (size > olh->size) { 23969d099b4fSAlexander V. Chernikov olh->size = size; 23979d099b4fSAlexander V. Chernikov IPFW_UH_RUNLOCK(ch); 23989d099b4fSAlexander V. Chernikov return (ENOMEM); 23999d099b4fSAlexander V. Chernikov } 24009d099b4fSAlexander V. Chernikov olh->size = size; 24019d099b4fSAlexander V. Chernikov 24029d099b4fSAlexander V. Chernikov for (n = 1; n <= count; n++) { 24039d099b4fSAlexander V. Chernikov i = (ipfw_ta_info *)ipfw_get_sopt_space(sd, sizeof(*i)); 24049d099b4fSAlexander V. Chernikov KASSERT(i != 0, ("previously checked buffer is not enough")); 24059d099b4fSAlexander V. Chernikov ta = tcfg->algo[n]; 24069d099b4fSAlexander V. Chernikov strlcpy(i->algoname, ta->name, sizeof(i->algoname)); 24079d099b4fSAlexander V. Chernikov i->type = ta->type; 24089d099b4fSAlexander V. Chernikov i->refcnt = ta->refcnt; 24099d099b4fSAlexander V. Chernikov } 24109d099b4fSAlexander V. Chernikov 24119d099b4fSAlexander V. Chernikov IPFW_UH_RUNLOCK(ch); 24129d099b4fSAlexander V. Chernikov 24139d099b4fSAlexander V. Chernikov return (0); 24149d099b4fSAlexander V. Chernikov } 24159d099b4fSAlexander V. Chernikov 24169f7d47b0SAlexander V. Chernikov 2417b074b7bbSAlexander V. Chernikov /* 2418b074b7bbSAlexander V. Chernikov * Tables rewriting code 2419b074b7bbSAlexander V. Chernikov * 2420b074b7bbSAlexander V. Chernikov */ 2421b074b7bbSAlexander V. Chernikov 2422b074b7bbSAlexander V. Chernikov /* 2423b074b7bbSAlexander V. Chernikov * Determine table number and lookup type for @cmd. 2424b074b7bbSAlexander V. Chernikov * Fill @tbl and @type with appropriate values. 2425b074b7bbSAlexander V. Chernikov * Returns 0 for relevant opcodes, 1 otherwise. 2426b074b7bbSAlexander V. Chernikov */ 2427b074b7bbSAlexander V. Chernikov static int 2428b074b7bbSAlexander V. Chernikov classify_table_opcode(ipfw_insn *cmd, uint16_t *puidx, uint8_t *ptype) 2429b074b7bbSAlexander V. Chernikov { 2430b074b7bbSAlexander V. Chernikov ipfw_insn_if *cmdif; 2431b074b7bbSAlexander V. Chernikov int skip; 2432b074b7bbSAlexander V. Chernikov uint16_t v; 2433b074b7bbSAlexander V. Chernikov 2434b074b7bbSAlexander V. Chernikov skip = 1; 2435b074b7bbSAlexander V. Chernikov 2436b074b7bbSAlexander V. Chernikov switch (cmd->opcode) { 2437b074b7bbSAlexander V. Chernikov case O_IP_SRC_LOOKUP: 2438b074b7bbSAlexander V. Chernikov case O_IP_DST_LOOKUP: 2439b074b7bbSAlexander V. Chernikov /* Basic IPv4/IPv6 or u32 lookups */ 2440b074b7bbSAlexander V. Chernikov *puidx = cmd->arg1; 2441b074b7bbSAlexander V. Chernikov /* Assume CIDR by default */ 2442b074b7bbSAlexander V. Chernikov *ptype = IPFW_TABLE_CIDR; 2443b074b7bbSAlexander V. Chernikov skip = 0; 2444b074b7bbSAlexander V. Chernikov 2445b074b7bbSAlexander V. Chernikov if (F_LEN(cmd) > F_INSN_SIZE(ipfw_insn_u32)) { 2446b074b7bbSAlexander V. Chernikov /* 2447b074b7bbSAlexander V. Chernikov * generic lookup. The key must be 2448b074b7bbSAlexander V. Chernikov * in 32bit big-endian format. 2449b074b7bbSAlexander V. Chernikov */ 2450b074b7bbSAlexander V. Chernikov v = ((ipfw_insn_u32 *)cmd)->d[1]; 2451b074b7bbSAlexander V. Chernikov switch (v) { 2452b074b7bbSAlexander V. Chernikov case 0: 2453b074b7bbSAlexander V. Chernikov case 1: 2454b074b7bbSAlexander V. Chernikov /* IPv4 src/dst */ 2455b074b7bbSAlexander V. Chernikov break; 2456b074b7bbSAlexander V. Chernikov case 2: 2457b074b7bbSAlexander V. Chernikov case 3: 2458b074b7bbSAlexander V. Chernikov /* src/dst port */ 2459b23d5de9SAlexander V. Chernikov *ptype = IPFW_TABLE_NUMBER; 2460b074b7bbSAlexander V. Chernikov break; 2461b074b7bbSAlexander V. Chernikov case 4: 2462b074b7bbSAlexander V. Chernikov /* uid/gid */ 2463b23d5de9SAlexander V. Chernikov *ptype = IPFW_TABLE_NUMBER; 2464b23d5de9SAlexander V. Chernikov break; 2465b074b7bbSAlexander V. Chernikov case 5: 2466b074b7bbSAlexander V. Chernikov /* jid */ 2467b23d5de9SAlexander V. Chernikov *ptype = IPFW_TABLE_NUMBER; 2468b23d5de9SAlexander V. Chernikov break; 2469b074b7bbSAlexander V. Chernikov case 6: 2470b074b7bbSAlexander V. Chernikov /* dscp */ 2471b23d5de9SAlexander V. Chernikov *ptype = IPFW_TABLE_NUMBER; 2472b074b7bbSAlexander V. Chernikov break; 2473b074b7bbSAlexander V. Chernikov } 2474b074b7bbSAlexander V. Chernikov } 2475b074b7bbSAlexander V. Chernikov break; 2476b074b7bbSAlexander V. Chernikov case O_XMIT: 2477b074b7bbSAlexander V. Chernikov case O_RECV: 2478b074b7bbSAlexander V. Chernikov case O_VIA: 2479b074b7bbSAlexander V. Chernikov /* Interface table, possibly */ 2480b074b7bbSAlexander V. Chernikov cmdif = (ipfw_insn_if *)cmd; 2481b074b7bbSAlexander V. Chernikov if (cmdif->name[0] != '\1') 2482b074b7bbSAlexander V. Chernikov break; 2483b074b7bbSAlexander V. Chernikov 2484b074b7bbSAlexander V. Chernikov *ptype = IPFW_TABLE_INTERFACE; 24858bd19212SAlexander V. Chernikov *puidx = cmdif->p.glob; 2486b074b7bbSAlexander V. Chernikov skip = 0; 2487b074b7bbSAlexander V. Chernikov break; 2488914bffb6SAlexander V. Chernikov case O_IP_FLOW_LOOKUP: 2489914bffb6SAlexander V. Chernikov *puidx = cmd->arg1; 2490914bffb6SAlexander V. Chernikov *ptype = IPFW_TABLE_FLOW; 2491914bffb6SAlexander V. Chernikov skip = 0; 2492914bffb6SAlexander V. Chernikov break; 2493b074b7bbSAlexander V. Chernikov } 2494b074b7bbSAlexander V. Chernikov 2495b074b7bbSAlexander V. Chernikov return (skip); 2496b074b7bbSAlexander V. Chernikov } 2497b074b7bbSAlexander V. Chernikov 2498b074b7bbSAlexander V. Chernikov /* 2499b074b7bbSAlexander V. Chernikov * Sets new table value for given opcode. 2500b074b7bbSAlexander V. Chernikov * Assume the same opcodes as classify_table_opcode() 2501b074b7bbSAlexander V. Chernikov */ 2502b074b7bbSAlexander V. Chernikov static void 2503b074b7bbSAlexander V. Chernikov update_table_opcode(ipfw_insn *cmd, uint16_t idx) 2504b074b7bbSAlexander V. Chernikov { 2505b074b7bbSAlexander V. Chernikov ipfw_insn_if *cmdif; 2506b074b7bbSAlexander V. Chernikov 2507b074b7bbSAlexander V. Chernikov switch (cmd->opcode) { 2508b074b7bbSAlexander V. Chernikov case O_IP_SRC_LOOKUP: 2509b074b7bbSAlexander V. Chernikov case O_IP_DST_LOOKUP: 2510b074b7bbSAlexander V. Chernikov /* Basic IPv4/IPv6 or u32 lookups */ 2511b074b7bbSAlexander V. Chernikov cmd->arg1 = idx; 2512b074b7bbSAlexander V. Chernikov break; 2513b074b7bbSAlexander V. Chernikov case O_XMIT: 2514b074b7bbSAlexander V. Chernikov case O_RECV: 2515b074b7bbSAlexander V. Chernikov case O_VIA: 2516b074b7bbSAlexander V. Chernikov /* Interface table, possibly */ 2517b074b7bbSAlexander V. Chernikov cmdif = (ipfw_insn_if *)cmd; 25188bd19212SAlexander V. Chernikov cmdif->p.glob = idx; 2519b074b7bbSAlexander V. Chernikov break; 2520914bffb6SAlexander V. Chernikov case O_IP_FLOW_LOOKUP: 2521914bffb6SAlexander V. Chernikov cmd->arg1 = idx; 2522914bffb6SAlexander V. Chernikov break; 2523b074b7bbSAlexander V. Chernikov } 2524b074b7bbSAlexander V. Chernikov } 2525b074b7bbSAlexander V. Chernikov 2526ac35ff17SAlexander V. Chernikov /* 2527ac35ff17SAlexander V. Chernikov * Checks table name for validity. 2528ac35ff17SAlexander V. Chernikov * Enforce basic length checks, the rest 2529ac35ff17SAlexander V. Chernikov * should be done in userland. 2530ac35ff17SAlexander V. Chernikov * 2531ac35ff17SAlexander V. Chernikov * Returns 0 if name is considered valid. 2532ac35ff17SAlexander V. Chernikov */ 2533ac35ff17SAlexander V. Chernikov int 2534ac35ff17SAlexander V. Chernikov ipfw_check_table_name(char *name) 2535ac35ff17SAlexander V. Chernikov { 2536ac35ff17SAlexander V. Chernikov int nsize; 2537ac35ff17SAlexander V. Chernikov ipfw_obj_ntlv *ntlv = NULL; 2538ac35ff17SAlexander V. Chernikov 2539ac35ff17SAlexander V. Chernikov nsize = sizeof(ntlv->name); 2540ac35ff17SAlexander V. Chernikov 2541ac35ff17SAlexander V. Chernikov if (strnlen(name, nsize) == nsize) 2542ac35ff17SAlexander V. Chernikov return (EINVAL); 2543ac35ff17SAlexander V. Chernikov 2544ac35ff17SAlexander V. Chernikov if (name[0] == '\0') 2545ac35ff17SAlexander V. Chernikov return (EINVAL); 2546ac35ff17SAlexander V. Chernikov 2547ac35ff17SAlexander V. Chernikov /* 2548ac35ff17SAlexander V. Chernikov * TODO: do some more complicated checks 2549ac35ff17SAlexander V. Chernikov */ 2550ac35ff17SAlexander V. Chernikov 2551ac35ff17SAlexander V. Chernikov return (0); 2552ac35ff17SAlexander V. Chernikov } 2553ac35ff17SAlexander V. Chernikov 2554ac35ff17SAlexander V. Chernikov /* 2555ac35ff17SAlexander V. Chernikov * Find tablename TLV by @uid. 2556ac35ff17SAlexander V. Chernikov * Check @tlvs for valid data inside. 2557ac35ff17SAlexander V. Chernikov * 2558ac35ff17SAlexander V. Chernikov * Returns pointer to found TLV or NULL. 2559ac35ff17SAlexander V. Chernikov */ 2560ac35ff17SAlexander V. Chernikov static ipfw_obj_ntlv * 2561b074b7bbSAlexander V. Chernikov find_name_tlv(void *tlvs, int len, uint16_t uidx) 2562b074b7bbSAlexander V. Chernikov { 25639f7d47b0SAlexander V. Chernikov ipfw_obj_ntlv *ntlv; 2564b074b7bbSAlexander V. Chernikov uintptr_t pa, pe; 2565b074b7bbSAlexander V. Chernikov int l; 2566b074b7bbSAlexander V. Chernikov 2567b074b7bbSAlexander V. Chernikov pa = (uintptr_t)tlvs; 2568b074b7bbSAlexander V. Chernikov pe = pa + len; 2569b074b7bbSAlexander V. Chernikov l = 0; 2570b074b7bbSAlexander V. Chernikov for (; pa < pe; pa += l) { 25719f7d47b0SAlexander V. Chernikov ntlv = (ipfw_obj_ntlv *)pa; 2572b074b7bbSAlexander V. Chernikov l = ntlv->head.length; 2573ac35ff17SAlexander V. Chernikov 2574ac35ff17SAlexander V. Chernikov if (l != sizeof(*ntlv)) 2575ac35ff17SAlexander V. Chernikov return (NULL); 2576ac35ff17SAlexander V. Chernikov 2577563b5ab1SAlexander V. Chernikov if (ntlv->head.type != IPFW_TLV_TBL_NAME) 2578b074b7bbSAlexander V. Chernikov continue; 2579ac35ff17SAlexander V. Chernikov 2580b074b7bbSAlexander V. Chernikov if (ntlv->idx != uidx) 2581b074b7bbSAlexander V. Chernikov continue; 2582b074b7bbSAlexander V. Chernikov 2583ac35ff17SAlexander V. Chernikov if (ipfw_check_table_name(ntlv->name) != 0) 2584ac35ff17SAlexander V. Chernikov return (NULL); 2585ac35ff17SAlexander V. Chernikov 2586ac35ff17SAlexander V. Chernikov return (ntlv); 2587b074b7bbSAlexander V. Chernikov } 2588b074b7bbSAlexander V. Chernikov 2589b074b7bbSAlexander V. Chernikov return (NULL); 2590b074b7bbSAlexander V. Chernikov } 2591b074b7bbSAlexander V. Chernikov 2592ac35ff17SAlexander V. Chernikov /* 2593ac35ff17SAlexander V. Chernikov * Finds table config based on either legacy index 2594ac35ff17SAlexander V. Chernikov * or name in ntlv. 2595ac35ff17SAlexander V. Chernikov * Note @ti structure contains unchecked data from userland. 2596ac35ff17SAlexander V. Chernikov * 2597ac35ff17SAlexander V. Chernikov * Returns pointer to table_config or NULL. 2598ac35ff17SAlexander V. Chernikov */ 2599b074b7bbSAlexander V. Chernikov static struct table_config * 2600b074b7bbSAlexander V. Chernikov find_table(struct namedobj_instance *ni, struct tid_info *ti) 2601b074b7bbSAlexander V. Chernikov { 2602b074b7bbSAlexander V. Chernikov char *name, bname[16]; 2603b074b7bbSAlexander V. Chernikov struct named_object *no; 2604ac35ff17SAlexander V. Chernikov ipfw_obj_ntlv *ntlv; 2605ac35ff17SAlexander V. Chernikov uint32_t set; 2606b074b7bbSAlexander V. Chernikov 2607b074b7bbSAlexander V. Chernikov if (ti->tlvs != NULL) { 2608ac35ff17SAlexander V. Chernikov ntlv = find_name_tlv(ti->tlvs, ti->tlen, ti->uidx); 2609ac35ff17SAlexander V. Chernikov if (ntlv == NULL) 2610b074b7bbSAlexander V. Chernikov return (NULL); 2611ac35ff17SAlexander V. Chernikov name = ntlv->name; 2612ac35ff17SAlexander V. Chernikov set = ntlv->set; 2613b074b7bbSAlexander V. Chernikov } else { 2614b074b7bbSAlexander V. Chernikov snprintf(bname, sizeof(bname), "%d", ti->uidx); 2615b074b7bbSAlexander V. Chernikov name = bname; 2616ac35ff17SAlexander V. Chernikov set = 0; 2617b074b7bbSAlexander V. Chernikov } 2618b074b7bbSAlexander V. Chernikov 2619ac35ff17SAlexander V. Chernikov no = ipfw_objhash_lookup_name(ni, set, name); 2620b074b7bbSAlexander V. Chernikov 2621b074b7bbSAlexander V. Chernikov return ((struct table_config *)no); 2622b074b7bbSAlexander V. Chernikov } 2623b074b7bbSAlexander V. Chernikov 262435e1bbd0SAlexander V. Chernikov /* 262535e1bbd0SAlexander V. Chernikov * Allocate new table config structure using 262635e1bbd0SAlexander V. Chernikov * specified @algo and @aname. 262735e1bbd0SAlexander V. Chernikov * 262835e1bbd0SAlexander V. Chernikov * Returns pointer to config or NULL. 262935e1bbd0SAlexander V. Chernikov */ 2630b074b7bbSAlexander V. Chernikov static struct table_config * 263168394ec8SAlexander V. Chernikov alloc_table_config(struct ip_fw_chain *ch, struct tid_info *ti, 2632914bffb6SAlexander V. Chernikov struct table_algo *ta, char *aname, uint8_t tflags, uint8_t vtype) 2633b074b7bbSAlexander V. Chernikov { 2634b074b7bbSAlexander V. Chernikov char *name, bname[16]; 2635b074b7bbSAlexander V. Chernikov struct table_config *tc; 2636b074b7bbSAlexander V. Chernikov int error; 2637ac35ff17SAlexander V. Chernikov ipfw_obj_ntlv *ntlv; 2638ac35ff17SAlexander V. Chernikov uint32_t set; 2639b074b7bbSAlexander V. Chernikov 2640b074b7bbSAlexander V. Chernikov if (ti->tlvs != NULL) { 2641ac35ff17SAlexander V. Chernikov ntlv = find_name_tlv(ti->tlvs, ti->tlen, ti->uidx); 2642ac35ff17SAlexander V. Chernikov if (ntlv == NULL) 2643b074b7bbSAlexander V. Chernikov return (NULL); 2644ac35ff17SAlexander V. Chernikov name = ntlv->name; 2645ac35ff17SAlexander V. Chernikov set = ntlv->set; 2646b074b7bbSAlexander V. Chernikov } else { 2647b074b7bbSAlexander V. Chernikov snprintf(bname, sizeof(bname), "%d", ti->uidx); 2648b074b7bbSAlexander V. Chernikov name = bname; 2649ac35ff17SAlexander V. Chernikov set = 0; 2650b074b7bbSAlexander V. Chernikov } 2651b074b7bbSAlexander V. Chernikov 2652b074b7bbSAlexander V. Chernikov tc = malloc(sizeof(struct table_config), M_IPFW, M_WAITOK | M_ZERO); 2653b074b7bbSAlexander V. Chernikov tc->no.name = tc->tablename; 2654b074b7bbSAlexander V. Chernikov tc->no.type = ti->type; 2655ac35ff17SAlexander V. Chernikov tc->no.set = set; 2656914bffb6SAlexander V. Chernikov tc->tflags = tflags; 26579f7d47b0SAlexander V. Chernikov tc->ta = ta; 2658b074b7bbSAlexander V. Chernikov strlcpy(tc->tablename, name, sizeof(tc->tablename)); 2659db785d31SAlexander V. Chernikov tc->vtype = vtype; 2660b074b7bbSAlexander V. Chernikov 2661b074b7bbSAlexander V. Chernikov if (ti->tlvs == NULL) { 2662b074b7bbSAlexander V. Chernikov tc->no.compat = 1; 2663b074b7bbSAlexander V. Chernikov tc->no.uidx = ti->uidx; 2664b074b7bbSAlexander V. Chernikov } 2665b074b7bbSAlexander V. Chernikov 2666b074b7bbSAlexander V. Chernikov /* Preallocate data structures for new tables */ 2667914bffb6SAlexander V. Chernikov error = ta->init(ch, &tc->astate, &tc->ti, aname, tflags); 2668b074b7bbSAlexander V. Chernikov if (error != 0) { 2669b074b7bbSAlexander V. Chernikov free(tc, M_IPFW); 2670b074b7bbSAlexander V. Chernikov return (NULL); 2671b074b7bbSAlexander V. Chernikov } 2672b074b7bbSAlexander V. Chernikov 2673b074b7bbSAlexander V. Chernikov return (tc); 2674b074b7bbSAlexander V. Chernikov } 2675b074b7bbSAlexander V. Chernikov 267635e1bbd0SAlexander V. Chernikov /* 267735e1bbd0SAlexander V. Chernikov * Destroys table state and config. 267835e1bbd0SAlexander V. Chernikov */ 2679b074b7bbSAlexander V. Chernikov static void 2680b074b7bbSAlexander V. Chernikov free_table_config(struct namedobj_instance *ni, struct table_config *tc) 2681b074b7bbSAlexander V. Chernikov { 2682b074b7bbSAlexander V. Chernikov 268335e1bbd0SAlexander V. Chernikov KASSERT(tc->linked == 0, ("free() on linked config")); 2684b074b7bbSAlexander V. Chernikov 26850468c5baSAlexander V. Chernikov /* 26860468c5baSAlexander V. Chernikov * We're using ta without any locking/referencing. 26870468c5baSAlexander V. Chernikov * TODO: fix this if we're going to use unloadable algos. 26880468c5baSAlexander V. Chernikov */ 268935e1bbd0SAlexander V. Chernikov tc->ta->destroy(tc->astate, &tc->ti); 2690b074b7bbSAlexander V. Chernikov free(tc, M_IPFW); 2691b074b7bbSAlexander V. Chernikov } 2692b074b7bbSAlexander V. Chernikov 2693b074b7bbSAlexander V. Chernikov /* 2694b074b7bbSAlexander V. Chernikov * Links @tc to @chain table named instance. 2695b074b7bbSAlexander V. Chernikov * Sets appropriate type/states in @chain table info. 2696b074b7bbSAlexander V. Chernikov */ 2697b074b7bbSAlexander V. Chernikov static void 26989f7d47b0SAlexander V. Chernikov link_table(struct ip_fw_chain *ch, struct table_config *tc) 2699b074b7bbSAlexander V. Chernikov { 2700b074b7bbSAlexander V. Chernikov struct namedobj_instance *ni; 27019f7d47b0SAlexander V. Chernikov struct table_info *ti; 2702b074b7bbSAlexander V. Chernikov uint16_t kidx; 2703b074b7bbSAlexander V. Chernikov 27049f7d47b0SAlexander V. Chernikov IPFW_UH_WLOCK_ASSERT(ch); 27059f7d47b0SAlexander V. Chernikov IPFW_WLOCK_ASSERT(ch); 2706b074b7bbSAlexander V. Chernikov 27079f7d47b0SAlexander V. Chernikov ni = CHAIN_TO_NI(ch); 2708b074b7bbSAlexander V. Chernikov kidx = tc->no.kidx; 2709b074b7bbSAlexander V. Chernikov 2710b074b7bbSAlexander V. Chernikov ipfw_objhash_add(ni, &tc->no); 27119f7d47b0SAlexander V. Chernikov 27129f7d47b0SAlexander V. Chernikov ti = KIDX_TO_TI(ch, kidx); 27139f7d47b0SAlexander V. Chernikov *ti = tc->ti; 2714b074b7bbSAlexander V. Chernikov 271568394ec8SAlexander V. Chernikov /* Notify algo on real @ti address */ 271668394ec8SAlexander V. Chernikov if (tc->ta->change_ti != NULL) 271768394ec8SAlexander V. Chernikov tc->ta->change_ti(tc->astate, ti); 271868394ec8SAlexander V. Chernikov 2719b074b7bbSAlexander V. Chernikov tc->linked = 1; 27209d099b4fSAlexander V. Chernikov tc->ta->refcnt++; 2721b074b7bbSAlexander V. Chernikov } 2722b074b7bbSAlexander V. Chernikov 2723b074b7bbSAlexander V. Chernikov /* 2724b074b7bbSAlexander V. Chernikov * Unlinks @tc from @chain table named instance. 2725b074b7bbSAlexander V. Chernikov * Zeroes states in @chain and stores them in @tc. 2726b074b7bbSAlexander V. Chernikov */ 2727b074b7bbSAlexander V. Chernikov static void 27289f7d47b0SAlexander V. Chernikov unlink_table(struct ip_fw_chain *ch, struct table_config *tc) 2729b074b7bbSAlexander V. Chernikov { 2730b074b7bbSAlexander V. Chernikov struct namedobj_instance *ni; 27319f7d47b0SAlexander V. Chernikov struct table_info *ti; 2732b074b7bbSAlexander V. Chernikov uint16_t kidx; 2733b074b7bbSAlexander V. Chernikov 27349f7d47b0SAlexander V. Chernikov IPFW_UH_WLOCK_ASSERT(ch); 27359f7d47b0SAlexander V. Chernikov IPFW_WLOCK_ASSERT(ch); 2736b074b7bbSAlexander V. Chernikov 27379f7d47b0SAlexander V. Chernikov ni = CHAIN_TO_NI(ch); 2738b074b7bbSAlexander V. Chernikov kidx = tc->no.kidx; 2739b074b7bbSAlexander V. Chernikov 27409f7d47b0SAlexander V. Chernikov /* Clear state. @ti copy is already saved inside @tc */ 2741b074b7bbSAlexander V. Chernikov ipfw_objhash_del(ni, &tc->no); 27429f7d47b0SAlexander V. Chernikov ti = KIDX_TO_TI(ch, kidx); 27439f7d47b0SAlexander V. Chernikov memset(ti, 0, sizeof(struct table_info)); 2744b074b7bbSAlexander V. Chernikov tc->linked = 0; 27459d099b4fSAlexander V. Chernikov tc->ta->refcnt--; 274668394ec8SAlexander V. Chernikov 274768394ec8SAlexander V. Chernikov /* Notify algo on real @ti address */ 274868394ec8SAlexander V. Chernikov if (tc->ta->change_ti != NULL) 274968394ec8SAlexander V. Chernikov tc->ta->change_ti(tc->astate, NULL); 2750b074b7bbSAlexander V. Chernikov } 2751b074b7bbSAlexander V. Chernikov 2752a73d728dSAlexander V. Chernikov struct swap_table_args { 2753a73d728dSAlexander V. Chernikov int set; 2754a73d728dSAlexander V. Chernikov int new_set; 2755a73d728dSAlexander V. Chernikov int mv; 2756a73d728dSAlexander V. Chernikov }; 2757a73d728dSAlexander V. Chernikov 2758a73d728dSAlexander V. Chernikov /* 2759a73d728dSAlexander V. Chernikov * Change set for each matching table. 2760a73d728dSAlexander V. Chernikov * 2761a73d728dSAlexander V. Chernikov * Ensure we dispatch each table once by setting/checking ochange 2762a73d728dSAlexander V. Chernikov * fields. 2763a73d728dSAlexander V. Chernikov */ 2764a73d728dSAlexander V. Chernikov static void 2765a73d728dSAlexander V. Chernikov swap_table_set(struct namedobj_instance *ni, struct named_object *no, 2766a73d728dSAlexander V. Chernikov void *arg) 2767a73d728dSAlexander V. Chernikov { 2768a73d728dSAlexander V. Chernikov struct table_config *tc; 2769a73d728dSAlexander V. Chernikov struct swap_table_args *sta; 2770a73d728dSAlexander V. Chernikov 2771a73d728dSAlexander V. Chernikov tc = (struct table_config *)no; 2772a73d728dSAlexander V. Chernikov sta = (struct swap_table_args *)arg; 2773a73d728dSAlexander V. Chernikov 2774a73d728dSAlexander V. Chernikov if (no->set != sta->set && (no->set != sta->new_set || sta->mv != 0)) 2775a73d728dSAlexander V. Chernikov return; 2776a73d728dSAlexander V. Chernikov 2777a73d728dSAlexander V. Chernikov if (tc->ochanged != 0) 2778a73d728dSAlexander V. Chernikov return; 2779a73d728dSAlexander V. Chernikov 2780a73d728dSAlexander V. Chernikov tc->ochanged = 1; 2781a73d728dSAlexander V. Chernikov ipfw_objhash_del(ni, no); 2782a73d728dSAlexander V. Chernikov if (no->set == sta->set) 2783a73d728dSAlexander V. Chernikov no->set = sta->new_set; 2784a73d728dSAlexander V. Chernikov else 2785a73d728dSAlexander V. Chernikov no->set = sta->set; 2786a73d728dSAlexander V. Chernikov ipfw_objhash_add(ni, no); 2787a73d728dSAlexander V. Chernikov } 2788a73d728dSAlexander V. Chernikov 2789a73d728dSAlexander V. Chernikov /* 2790a73d728dSAlexander V. Chernikov * Cleans up ochange field for all tables. 2791a73d728dSAlexander V. Chernikov */ 2792a73d728dSAlexander V. Chernikov static void 2793a73d728dSAlexander V. Chernikov clean_table_set_data(struct namedobj_instance *ni, struct named_object *no, 2794a73d728dSAlexander V. Chernikov void *arg) 2795a73d728dSAlexander V. Chernikov { 2796a73d728dSAlexander V. Chernikov struct table_config *tc; 2797a73d728dSAlexander V. Chernikov struct swap_table_args *sta; 2798a73d728dSAlexander V. Chernikov 2799a73d728dSAlexander V. Chernikov tc = (struct table_config *)no; 2800a73d728dSAlexander V. Chernikov sta = (struct swap_table_args *)arg; 2801a73d728dSAlexander V. Chernikov 2802a73d728dSAlexander V. Chernikov tc->ochanged = 0; 2803a73d728dSAlexander V. Chernikov } 2804a73d728dSAlexander V. Chernikov 2805a73d728dSAlexander V. Chernikov /* 2806a73d728dSAlexander V. Chernikov * Swaps tables within two sets. 2807a73d728dSAlexander V. Chernikov */ 2808a73d728dSAlexander V. Chernikov void 2809a73d728dSAlexander V. Chernikov ipfw_swap_tables_sets(struct ip_fw_chain *ch, uint32_t set, 2810a73d728dSAlexander V. Chernikov uint32_t new_set, int mv) 2811a73d728dSAlexander V. Chernikov { 2812a73d728dSAlexander V. Chernikov struct swap_table_args sta; 2813a73d728dSAlexander V. Chernikov 2814a73d728dSAlexander V. Chernikov IPFW_UH_WLOCK_ASSERT(ch); 2815a73d728dSAlexander V. Chernikov 2816a73d728dSAlexander V. Chernikov sta.set = set; 2817a73d728dSAlexander V. Chernikov sta.new_set = new_set; 2818a73d728dSAlexander V. Chernikov sta.mv = mv; 2819a73d728dSAlexander V. Chernikov 2820a73d728dSAlexander V. Chernikov ipfw_objhash_foreach(CHAIN_TO_NI(ch), swap_table_set, &sta); 2821a73d728dSAlexander V. Chernikov ipfw_objhash_foreach(CHAIN_TO_NI(ch), clean_table_set_data, &sta); 2822a73d728dSAlexander V. Chernikov } 2823a73d728dSAlexander V. Chernikov 2824a73d728dSAlexander V. Chernikov /* 2825a73d728dSAlexander V. Chernikov * Move all tables which are reference by rules in @rr to set @new_set. 2826a73d728dSAlexander V. Chernikov * Makes sure that all relevant tables are referenced ONLLY by given rules. 2827a73d728dSAlexander V. Chernikov * 2828a73d728dSAlexander V. Chernikov * Retuns 0 on success, 2829a73d728dSAlexander V. Chernikov */ 2830a73d728dSAlexander V. Chernikov int 2831a73d728dSAlexander V. Chernikov ipfw_move_tables_sets(struct ip_fw_chain *ch, ipfw_range_tlv *rt, 2832a73d728dSAlexander V. Chernikov uint32_t new_set) 2833a73d728dSAlexander V. Chernikov { 2834a73d728dSAlexander V. Chernikov struct ip_fw *rule; 2835a73d728dSAlexander V. Chernikov struct table_config *tc; 2836a73d728dSAlexander V. Chernikov struct named_object *no; 2837a73d728dSAlexander V. Chernikov struct namedobj_instance *ni; 2838a73d728dSAlexander V. Chernikov int bad, i, l, cmdlen; 2839a73d728dSAlexander V. Chernikov uint16_t kidx; 2840a73d728dSAlexander V. Chernikov uint8_t type; 2841a73d728dSAlexander V. Chernikov ipfw_insn *cmd; 2842a73d728dSAlexander V. Chernikov 2843a73d728dSAlexander V. Chernikov IPFW_UH_WLOCK_ASSERT(ch); 2844a73d728dSAlexander V. Chernikov 2845a73d728dSAlexander V. Chernikov ni = CHAIN_TO_NI(ch); 2846a73d728dSAlexander V. Chernikov 2847a73d728dSAlexander V. Chernikov /* Stage 1: count number of references by given rules */ 2848a73d728dSAlexander V. Chernikov for (i = 0; i < ch->n_rules - 1; i++) { 2849a73d728dSAlexander V. Chernikov rule = ch->map[i]; 2850a73d728dSAlexander V. Chernikov if (ipfw_match_range(rule, rt) == 0) 2851a73d728dSAlexander V. Chernikov continue; 2852a73d728dSAlexander V. Chernikov 2853a73d728dSAlexander V. Chernikov l = rule->cmd_len; 2854a73d728dSAlexander V. Chernikov cmd = rule->cmd; 2855a73d728dSAlexander V. Chernikov cmdlen = 0; 2856a73d728dSAlexander V. Chernikov for ( ; l > 0 ; l -= cmdlen, cmd += cmdlen) { 2857a73d728dSAlexander V. Chernikov cmdlen = F_LEN(cmd); 2858a73d728dSAlexander V. Chernikov if (classify_table_opcode(cmd, &kidx, &type) != 0) 2859a73d728dSAlexander V. Chernikov continue; 2860a73d728dSAlexander V. Chernikov no = ipfw_objhash_lookup_kidx(ni, kidx); 2861a73d728dSAlexander V. Chernikov KASSERT(no != NULL, 2862a73d728dSAlexander V. Chernikov ("objhash lookup failed on index %d", kidx)); 2863a73d728dSAlexander V. Chernikov tc = (struct table_config *)no; 2864a73d728dSAlexander V. Chernikov tc->ocount++; 2865a73d728dSAlexander V. Chernikov } 2866a73d728dSAlexander V. Chernikov 2867a73d728dSAlexander V. Chernikov } 2868a73d728dSAlexander V. Chernikov 2869a73d728dSAlexander V. Chernikov /* Stage 2: verify "ownership" */ 2870a73d728dSAlexander V. Chernikov bad = 0; 2871a73d728dSAlexander V. Chernikov for (i = 0; i < ch->n_rules - 1; i++) { 2872a73d728dSAlexander V. Chernikov rule = ch->map[i]; 2873a73d728dSAlexander V. Chernikov if (ipfw_match_range(rule, rt) == 0) 2874a73d728dSAlexander V. Chernikov continue; 2875a73d728dSAlexander V. Chernikov 2876a73d728dSAlexander V. Chernikov l = rule->cmd_len; 2877a73d728dSAlexander V. Chernikov cmd = rule->cmd; 2878a73d728dSAlexander V. Chernikov cmdlen = 0; 2879a73d728dSAlexander V. Chernikov for ( ; l > 0 ; l -= cmdlen, cmd += cmdlen) { 2880a73d728dSAlexander V. Chernikov cmdlen = F_LEN(cmd); 2881a73d728dSAlexander V. Chernikov if (classify_table_opcode(cmd, &kidx, &type) != 0) 2882a73d728dSAlexander V. Chernikov continue; 2883a73d728dSAlexander V. Chernikov no = ipfw_objhash_lookup_kidx(ni, kidx); 2884a73d728dSAlexander V. Chernikov KASSERT(no != NULL, 2885a73d728dSAlexander V. Chernikov ("objhash lookup failed on index %d", kidx)); 2886a73d728dSAlexander V. Chernikov tc = (struct table_config *)no; 2887a73d728dSAlexander V. Chernikov if (tc->no.refcnt != tc->ocount) { 2888a73d728dSAlexander V. Chernikov 2889a73d728dSAlexander V. Chernikov /* 2890a73d728dSAlexander V. Chernikov * Number of references differ: 2891a73d728dSAlexander V. Chernikov * Other rule(s) are holding reference to given 2892a73d728dSAlexander V. Chernikov * table, so it is not possible to change its set. 2893a73d728dSAlexander V. Chernikov * 2894a73d728dSAlexander V. Chernikov * Note that refcnt may account 2895a73d728dSAlexander V. Chernikov * references to some going-to-be-added rules. 2896a73d728dSAlexander V. Chernikov * Since we don't know their numbers (and event 2897a73d728dSAlexander V. Chernikov * if they will be added) it is perfectly OK 2898a73d728dSAlexander V. Chernikov * to return error here. 2899a73d728dSAlexander V. Chernikov */ 2900a73d728dSAlexander V. Chernikov bad = 1; 2901a73d728dSAlexander V. Chernikov break; 2902a73d728dSAlexander V. Chernikov } 2903a73d728dSAlexander V. Chernikov } 2904a73d728dSAlexander V. Chernikov 2905a73d728dSAlexander V. Chernikov if (bad != 0) 2906a73d728dSAlexander V. Chernikov break; 2907a73d728dSAlexander V. Chernikov } 2908a73d728dSAlexander V. Chernikov 2909a73d728dSAlexander V. Chernikov /* Stage 3: change set or cleanup */ 2910a73d728dSAlexander V. Chernikov for (i = 0; i < ch->n_rules - 1; i++) { 2911a73d728dSAlexander V. Chernikov rule = ch->map[i]; 2912a73d728dSAlexander V. Chernikov if (ipfw_match_range(rule, rt) == 0) 2913a73d728dSAlexander V. Chernikov continue; 2914a73d728dSAlexander V. Chernikov 2915a73d728dSAlexander V. Chernikov l = rule->cmd_len; 2916a73d728dSAlexander V. Chernikov cmd = rule->cmd; 2917a73d728dSAlexander V. Chernikov cmdlen = 0; 2918a73d728dSAlexander V. Chernikov for ( ; l > 0 ; l -= cmdlen, cmd += cmdlen) { 2919a73d728dSAlexander V. Chernikov cmdlen = F_LEN(cmd); 2920a73d728dSAlexander V. Chernikov if (classify_table_opcode(cmd, &kidx, &type) != 0) 2921a73d728dSAlexander V. Chernikov continue; 2922a73d728dSAlexander V. Chernikov no = ipfw_objhash_lookup_kidx(ni, kidx); 2923a73d728dSAlexander V. Chernikov KASSERT(no != NULL, 2924a73d728dSAlexander V. Chernikov ("objhash lookup failed on index %d", kidx)); 2925a73d728dSAlexander V. Chernikov tc = (struct table_config *)no; 2926a73d728dSAlexander V. Chernikov 2927a73d728dSAlexander V. Chernikov tc->ocount = 0; 2928a73d728dSAlexander V. Chernikov if (bad != 0) 2929a73d728dSAlexander V. Chernikov continue; 2930a73d728dSAlexander V. Chernikov 2931a73d728dSAlexander V. Chernikov /* Actually change set. */ 2932a73d728dSAlexander V. Chernikov ipfw_objhash_del(ni, no); 2933a73d728dSAlexander V. Chernikov no->set = new_set; 2934a73d728dSAlexander V. Chernikov ipfw_objhash_add(ni, no); 2935a73d728dSAlexander V. Chernikov } 2936a73d728dSAlexander V. Chernikov } 2937a73d728dSAlexander V. Chernikov 2938a73d728dSAlexander V. Chernikov return (bad); 2939a73d728dSAlexander V. Chernikov } 2940a73d728dSAlexander V. Chernikov 2941b074b7bbSAlexander V. Chernikov /* 2942f99fbf96SAlexander V. Chernikov * Finds and bumps refcount for tables referenced by given @rule. 2943f99fbf96SAlexander V. Chernikov * Allocates new indexes for non-existing tables. 2944f99fbf96SAlexander V. Chernikov * Fills in @oib array with userland/kernel indexes. 2945f99fbf96SAlexander V. Chernikov * First free oidx pointer is saved back in @oib. 2946f99fbf96SAlexander V. Chernikov * 2947f99fbf96SAlexander V. Chernikov * Returns 0 on success. 2948f99fbf96SAlexander V. Chernikov */ 2949f99fbf96SAlexander V. Chernikov static int 2950f99fbf96SAlexander V. Chernikov bind_table_rule(struct ip_fw_chain *ch, struct ip_fw *rule, 2951f99fbf96SAlexander V. Chernikov struct rule_check_info *ci, struct obj_idx **oib, struct tid_info *ti) 2952f99fbf96SAlexander V. Chernikov { 2953f99fbf96SAlexander V. Chernikov struct table_config *tc; 2954f99fbf96SAlexander V. Chernikov struct namedobj_instance *ni; 2955f99fbf96SAlexander V. Chernikov struct named_object *no; 2956f99fbf96SAlexander V. Chernikov int cmdlen, error, l, numnew; 2957f99fbf96SAlexander V. Chernikov uint16_t kidx; 2958f99fbf96SAlexander V. Chernikov ipfw_insn *cmd; 2959f99fbf96SAlexander V. Chernikov struct obj_idx *pidx, *pidx_first, *p; 2960f99fbf96SAlexander V. Chernikov 2961f99fbf96SAlexander V. Chernikov pidx_first = *oib; 2962f99fbf96SAlexander V. Chernikov pidx = pidx_first; 2963f99fbf96SAlexander V. Chernikov l = rule->cmd_len; 2964f99fbf96SAlexander V. Chernikov cmd = rule->cmd; 2965f99fbf96SAlexander V. Chernikov cmdlen = 0; 2966f99fbf96SAlexander V. Chernikov error = 0; 2967f99fbf96SAlexander V. Chernikov numnew = 0; 2968f99fbf96SAlexander V. Chernikov 2969f99fbf96SAlexander V. Chernikov IPFW_UH_WLOCK(ch); 2970f99fbf96SAlexander V. Chernikov ni = CHAIN_TO_NI(ch); 2971f99fbf96SAlexander V. Chernikov 2972f99fbf96SAlexander V. Chernikov /* 2973f99fbf96SAlexander V. Chernikov * Increase refcount on each referenced table. 2974f99fbf96SAlexander V. Chernikov * Allocate table indexes for non-existing tables. 2975f99fbf96SAlexander V. Chernikov */ 2976f99fbf96SAlexander V. Chernikov for ( ; l > 0 ; l -= cmdlen, cmd += cmdlen) { 2977f99fbf96SAlexander V. Chernikov cmdlen = F_LEN(cmd); 2978f99fbf96SAlexander V. Chernikov 2979f99fbf96SAlexander V. Chernikov if (classify_table_opcode(cmd, &ti->uidx, &ti->type) != 0) 2980f99fbf96SAlexander V. Chernikov continue; 2981f99fbf96SAlexander V. Chernikov 2982f99fbf96SAlexander V. Chernikov pidx->uidx = ti->uidx; 2983f99fbf96SAlexander V. Chernikov pidx->type = ti->type; 2984f99fbf96SAlexander V. Chernikov 2985f99fbf96SAlexander V. Chernikov if ((tc = find_table(ni, ti)) != NULL) { 2986f99fbf96SAlexander V. Chernikov if (tc->no.type != ti->type) { 2987f99fbf96SAlexander V. Chernikov /* Incompatible types */ 2988f99fbf96SAlexander V. Chernikov error = EINVAL; 2989f99fbf96SAlexander V. Chernikov break; 2990f99fbf96SAlexander V. Chernikov } 2991f99fbf96SAlexander V. Chernikov 2992f99fbf96SAlexander V. Chernikov /* Reference found table and save kidx */ 2993f99fbf96SAlexander V. Chernikov tc->no.refcnt++; 2994f99fbf96SAlexander V. Chernikov pidx->kidx = tc->no.kidx; 2995f99fbf96SAlexander V. Chernikov pidx++; 2996f99fbf96SAlexander V. Chernikov continue; 2997f99fbf96SAlexander V. Chernikov } 2998f99fbf96SAlexander V. Chernikov 2999f99fbf96SAlexander V. Chernikov /* 3000f99fbf96SAlexander V. Chernikov * Compability stuff for old clients: 3001f99fbf96SAlexander V. Chernikov * prepare to manually create non-existing tables. 3002f99fbf96SAlexander V. Chernikov */ 3003f99fbf96SAlexander V. Chernikov pidx++; 3004f99fbf96SAlexander V. Chernikov numnew++; 3005f99fbf96SAlexander V. Chernikov } 3006f99fbf96SAlexander V. Chernikov 3007f99fbf96SAlexander V. Chernikov if (error != 0) { 3008f99fbf96SAlexander V. Chernikov /* Unref everything we have already done */ 3009f99fbf96SAlexander V. Chernikov for (p = *oib; p < pidx; p++) { 3010f99fbf96SAlexander V. Chernikov if (p->kidx == 0) 3011f99fbf96SAlexander V. Chernikov continue; 3012f99fbf96SAlexander V. Chernikov 3013f99fbf96SAlexander V. Chernikov /* Find & unref by existing idx */ 3014f99fbf96SAlexander V. Chernikov no = ipfw_objhash_lookup_kidx(ni, p->kidx); 3015f99fbf96SAlexander V. Chernikov KASSERT(no != NULL, ("Ref'd table %d disappeared", 3016f99fbf96SAlexander V. Chernikov p->kidx)); 3017f99fbf96SAlexander V. Chernikov 3018f99fbf96SAlexander V. Chernikov no->refcnt--; 3019f99fbf96SAlexander V. Chernikov } 3020f99fbf96SAlexander V. Chernikov } 3021f99fbf96SAlexander V. Chernikov 3022f99fbf96SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 3023f99fbf96SAlexander V. Chernikov 3024f99fbf96SAlexander V. Chernikov if (numnew == 0) { 3025f99fbf96SAlexander V. Chernikov *oib = pidx; 3026f99fbf96SAlexander V. Chernikov return (error); 3027f99fbf96SAlexander V. Chernikov } 3028f99fbf96SAlexander V. Chernikov 3029f99fbf96SAlexander V. Chernikov /* 3030f99fbf96SAlexander V. Chernikov * Compatibility stuff: do actual creation for non-existing, 3031f99fbf96SAlexander V. Chernikov * but referenced tables. 3032f99fbf96SAlexander V. Chernikov */ 3033f99fbf96SAlexander V. Chernikov for (p = pidx_first; p < pidx; p++) { 3034f99fbf96SAlexander V. Chernikov if (p->kidx != 0) 3035f99fbf96SAlexander V. Chernikov continue; 3036f99fbf96SAlexander V. Chernikov 3037f99fbf96SAlexander V. Chernikov ti->uidx = p->uidx; 3038f99fbf96SAlexander V. Chernikov ti->type = p->type; 3039f99fbf96SAlexander V. Chernikov ti->atype = 0; 3040f99fbf96SAlexander V. Chernikov 3041f99fbf96SAlexander V. Chernikov error = create_table_compat(ch, ti, NULL, NULL, &kidx); 3042f99fbf96SAlexander V. Chernikov if (error == 0) { 3043f99fbf96SAlexander V. Chernikov p->kidx = kidx; 3044f99fbf96SAlexander V. Chernikov continue; 3045f99fbf96SAlexander V. Chernikov } 3046f99fbf96SAlexander V. Chernikov 3047f99fbf96SAlexander V. Chernikov /* Error. We have to drop references */ 3048f99fbf96SAlexander V. Chernikov IPFW_UH_WLOCK(ch); 3049f99fbf96SAlexander V. Chernikov for (p = pidx_first; p < pidx; p++) { 3050f99fbf96SAlexander V. Chernikov if (p->kidx == 0) 3051f99fbf96SAlexander V. Chernikov continue; 3052f99fbf96SAlexander V. Chernikov 3053f99fbf96SAlexander V. Chernikov /* Find & unref by existing idx */ 3054f99fbf96SAlexander V. Chernikov no = ipfw_objhash_lookup_kidx(ni, p->kidx); 3055f99fbf96SAlexander V. Chernikov KASSERT(no != NULL, ("Ref'd table %d disappeared", 3056f99fbf96SAlexander V. Chernikov p->kidx)); 3057f99fbf96SAlexander V. Chernikov 3058f99fbf96SAlexander V. Chernikov no->refcnt--; 3059f99fbf96SAlexander V. Chernikov } 3060f99fbf96SAlexander V. Chernikov IPFW_UH_WUNLOCK(ch); 3061f99fbf96SAlexander V. Chernikov 3062f99fbf96SAlexander V. Chernikov return (error); 3063f99fbf96SAlexander V. Chernikov } 3064f99fbf96SAlexander V. Chernikov 3065f99fbf96SAlexander V. Chernikov *oib = pidx; 3066f99fbf96SAlexander V. Chernikov 3067f99fbf96SAlexander V. Chernikov return (error); 3068f99fbf96SAlexander V. Chernikov } 3069f99fbf96SAlexander V. Chernikov 3070f99fbf96SAlexander V. Chernikov /* 3071f99fbf96SAlexander V. Chernikov * Remove references from every table used in @rule. 3072f99fbf96SAlexander V. Chernikov */ 3073f99fbf96SAlexander V. Chernikov void 3074f99fbf96SAlexander V. Chernikov ipfw_unbind_table_rule(struct ip_fw_chain *chain, struct ip_fw *rule) 3075f99fbf96SAlexander V. Chernikov { 3076f99fbf96SAlexander V. Chernikov int cmdlen, l; 3077f99fbf96SAlexander V. Chernikov ipfw_insn *cmd; 3078f99fbf96SAlexander V. Chernikov struct namedobj_instance *ni; 3079f99fbf96SAlexander V. Chernikov struct named_object *no; 3080f99fbf96SAlexander V. Chernikov uint16_t kidx; 3081f99fbf96SAlexander V. Chernikov uint8_t type; 3082f99fbf96SAlexander V. Chernikov 3083f99fbf96SAlexander V. Chernikov IPFW_UH_WLOCK_ASSERT(chain); 3084f99fbf96SAlexander V. Chernikov ni = CHAIN_TO_NI(chain); 3085f99fbf96SAlexander V. Chernikov 3086f99fbf96SAlexander V. Chernikov l = rule->cmd_len; 3087f99fbf96SAlexander V. Chernikov cmd = rule->cmd; 3088f99fbf96SAlexander V. Chernikov cmdlen = 0; 3089f99fbf96SAlexander V. Chernikov for ( ; l > 0 ; l -= cmdlen, cmd += cmdlen) { 3090f99fbf96SAlexander V. Chernikov cmdlen = F_LEN(cmd); 3091f99fbf96SAlexander V. Chernikov 3092f99fbf96SAlexander V. Chernikov if (classify_table_opcode(cmd, &kidx, &type) != 0) 3093f99fbf96SAlexander V. Chernikov continue; 3094f99fbf96SAlexander V. Chernikov 3095f99fbf96SAlexander V. Chernikov no = ipfw_objhash_lookup_kidx(ni, kidx); 3096f99fbf96SAlexander V. Chernikov 3097f99fbf96SAlexander V. Chernikov KASSERT(no != NULL, ("table id %d not found", kidx)); 3098f99fbf96SAlexander V. Chernikov KASSERT(no->type == type, ("wrong type %d (%d) for table id %d", 3099f99fbf96SAlexander V. Chernikov no->type, type, kidx)); 3100f99fbf96SAlexander V. Chernikov KASSERT(no->refcnt > 0, ("refcount for table %d is %d", 3101f99fbf96SAlexander V. Chernikov kidx, no->refcnt)); 3102f99fbf96SAlexander V. Chernikov 3103f99fbf96SAlexander V. Chernikov no->refcnt--; 3104f99fbf96SAlexander V. Chernikov } 3105f99fbf96SAlexander V. Chernikov } 3106f99fbf96SAlexander V. Chernikov 3107f99fbf96SAlexander V. Chernikov /* 3108b074b7bbSAlexander V. Chernikov * Compatibility function for old ipfw(8) binaries. 3109b074b7bbSAlexander V. Chernikov * Rewrites table kernel indices with userland ones. 311035e1bbd0SAlexander V. Chernikov * Convert tables matching '/^\d+$/' to their atoi() value. 311135e1bbd0SAlexander V. Chernikov * Use number 65535 for other tables. 3112b074b7bbSAlexander V. Chernikov * 3113b074b7bbSAlexander V. Chernikov * Returns 0 on success. 3114b074b7bbSAlexander V. Chernikov */ 3115b074b7bbSAlexander V. Chernikov int 31167e767c79SAlexander V. Chernikov ipfw_rewrite_table_kidx(struct ip_fw_chain *chain, struct ip_fw_rule0 *rule) 3117b074b7bbSAlexander V. Chernikov { 31181832a7b3SAlexander V. Chernikov int cmdlen, error, l; 3119b074b7bbSAlexander V. Chernikov ipfw_insn *cmd; 31201832a7b3SAlexander V. Chernikov uint16_t kidx, uidx; 3121b074b7bbSAlexander V. Chernikov uint8_t type; 3122b074b7bbSAlexander V. Chernikov struct named_object *no; 3123b074b7bbSAlexander V. Chernikov struct namedobj_instance *ni; 3124b074b7bbSAlexander V. Chernikov 3125b074b7bbSAlexander V. Chernikov ni = CHAIN_TO_NI(chain); 31261832a7b3SAlexander V. Chernikov error = 0; 3127b074b7bbSAlexander V. Chernikov 3128b074b7bbSAlexander V. Chernikov l = rule->cmd_len; 3129b074b7bbSAlexander V. Chernikov cmd = rule->cmd; 3130b074b7bbSAlexander V. Chernikov cmdlen = 0; 3131b074b7bbSAlexander V. Chernikov for ( ; l > 0 ; l -= cmdlen, cmd += cmdlen) { 3132b074b7bbSAlexander V. Chernikov cmdlen = F_LEN(cmd); 3133b074b7bbSAlexander V. Chernikov 3134b074b7bbSAlexander V. Chernikov if (classify_table_opcode(cmd, &kidx, &type) != 0) 3135b074b7bbSAlexander V. Chernikov continue; 3136b074b7bbSAlexander V. Chernikov 3137ac35ff17SAlexander V. Chernikov if ((no = ipfw_objhash_lookup_kidx(ni, kidx)) == NULL) 3138b074b7bbSAlexander V. Chernikov return (1); 3139b074b7bbSAlexander V. Chernikov 31401832a7b3SAlexander V. Chernikov uidx = no->uidx; 31411832a7b3SAlexander V. Chernikov if (no->compat == 0) { 3142b074b7bbSAlexander V. Chernikov 31431832a7b3SAlexander V. Chernikov /* 31441832a7b3SAlexander V. Chernikov * We are called via legacy opcode. 31451832a7b3SAlexander V. Chernikov * Save error and show table as fake number 31461832a7b3SAlexander V. Chernikov * not to make ipfw(8) hang. 31471832a7b3SAlexander V. Chernikov */ 31481832a7b3SAlexander V. Chernikov uidx = 65535; 31491832a7b3SAlexander V. Chernikov error = 2; 3150b074b7bbSAlexander V. Chernikov } 3151b074b7bbSAlexander V. Chernikov 31521832a7b3SAlexander V. Chernikov update_table_opcode(cmd, uidx); 31531832a7b3SAlexander V. Chernikov } 31541832a7b3SAlexander V. Chernikov 31551832a7b3SAlexander V. Chernikov return (error); 3156b074b7bbSAlexander V. Chernikov } 3157b074b7bbSAlexander V. Chernikov 3158563b5ab1SAlexander V. Chernikov /* 3159b074b7bbSAlexander V. Chernikov * Checks is opcode is referencing table of appropriate type. 3160b074b7bbSAlexander V. Chernikov * Adds reference count for found table if true. 3161b074b7bbSAlexander V. Chernikov * Rewrites user-supplied opcode values with kernel ones. 3162b074b7bbSAlexander V. Chernikov * 3163b074b7bbSAlexander V. Chernikov * Returns 0 on success and appropriate error code otherwise. 3164b074b7bbSAlexander V. Chernikov */ 3165b074b7bbSAlexander V. Chernikov int 3166b074b7bbSAlexander V. Chernikov ipfw_rewrite_table_uidx(struct ip_fw_chain *chain, 3167b074b7bbSAlexander V. Chernikov struct rule_check_info *ci) 3168b074b7bbSAlexander V. Chernikov { 31690468c5baSAlexander V. Chernikov int cmdlen, error, l; 3170b074b7bbSAlexander V. Chernikov ipfw_insn *cmd; 3171b074b7bbSAlexander V. Chernikov uint16_t uidx; 3172b074b7bbSAlexander V. Chernikov uint8_t type; 3173b074b7bbSAlexander V. Chernikov struct namedobj_instance *ni; 31749490a627SAlexander V. Chernikov struct obj_idx *p, *pidx_first, *pidx_last; 3175b074b7bbSAlexander V. Chernikov struct tid_info ti; 3176b074b7bbSAlexander V. Chernikov 3177b074b7bbSAlexander V. Chernikov ni = CHAIN_TO_NI(chain); 3178b074b7bbSAlexander V. Chernikov 3179b074b7bbSAlexander V. Chernikov /* 3180b074b7bbSAlexander V. Chernikov * Prepare an array for storing opcode indices. 3181b074b7bbSAlexander V. Chernikov * Use stack allocation by default. 3182b074b7bbSAlexander V. Chernikov */ 3183b074b7bbSAlexander V. Chernikov if (ci->table_opcodes <= (sizeof(ci->obuf)/sizeof(ci->obuf[0]))) { 3184b074b7bbSAlexander V. Chernikov /* Stack */ 31859490a627SAlexander V. Chernikov pidx_first = ci->obuf; 3186b074b7bbSAlexander V. Chernikov } else 31879490a627SAlexander V. Chernikov pidx_first = malloc(ci->table_opcodes * sizeof(struct obj_idx), 3188b074b7bbSAlexander V. Chernikov M_IPFW, M_WAITOK | M_ZERO); 3189b074b7bbSAlexander V. Chernikov 31909490a627SAlexander V. Chernikov pidx_last = pidx_first; 3191b074b7bbSAlexander V. Chernikov error = 0; 3192b074b7bbSAlexander V. Chernikov type = 0; 3193b074b7bbSAlexander V. Chernikov memset(&ti, 0, sizeof(ti)); 31941832a7b3SAlexander V. Chernikov 31951832a7b3SAlexander V. Chernikov /* 31961832a7b3SAlexander V. Chernikov * Use default set for looking up tables (old way) or 31971832a7b3SAlexander V. Chernikov * use set rule is assigned to (new way). 31981832a7b3SAlexander V. Chernikov */ 31991832a7b3SAlexander V. Chernikov ti.set = (V_fw_tables_sets != 0) ? ci->krule->set : 0; 32006c2997ffSAlexander V. Chernikov if (ci->ctlv != NULL) { 32016c2997ffSAlexander V. Chernikov ti.tlvs = (void *)(ci->ctlv + 1); 32026c2997ffSAlexander V. Chernikov ti.tlen = ci->ctlv->head.length - sizeof(ipfw_obj_ctlv); 32036c2997ffSAlexander V. Chernikov } 3204b074b7bbSAlexander V. Chernikov 32050468c5baSAlexander V. Chernikov /* Reference all used tables */ 32069490a627SAlexander V. Chernikov error = bind_table_rule(chain, ci->krule, ci, &pidx_last, &ti); 32070468c5baSAlexander V. Chernikov if (error != 0) 32089f7d47b0SAlexander V. Chernikov goto free; 3209b074b7bbSAlexander V. Chernikov 32109f7d47b0SAlexander V. Chernikov IPFW_UH_WLOCK(chain); 32119f7d47b0SAlexander V. Chernikov 3212b074b7bbSAlexander V. Chernikov /* Perform rule rewrite */ 3213b074b7bbSAlexander V. Chernikov l = ci->krule->cmd_len; 3214b074b7bbSAlexander V. Chernikov cmd = ci->krule->cmd; 3215b074b7bbSAlexander V. Chernikov cmdlen = 0; 32169490a627SAlexander V. Chernikov p = pidx_first; 3217b074b7bbSAlexander V. Chernikov for ( ; l > 0 ; l -= cmdlen, cmd += cmdlen) { 3218b074b7bbSAlexander V. Chernikov cmdlen = F_LEN(cmd); 3219b074b7bbSAlexander V. Chernikov if (classify_table_opcode(cmd, &uidx, &type) != 0) 3220b074b7bbSAlexander V. Chernikov continue; 32219490a627SAlexander V. Chernikov update_table_opcode(cmd, p->kidx); 32229490a627SAlexander V. Chernikov p++; 3223b074b7bbSAlexander V. Chernikov } 3224b074b7bbSAlexander V. Chernikov 3225b074b7bbSAlexander V. Chernikov IPFW_UH_WUNLOCK(chain); 3226b074b7bbSAlexander V. Chernikov 3227b074b7bbSAlexander V. Chernikov free: 32289490a627SAlexander V. Chernikov if (pidx_first != ci->obuf) 32299490a627SAlexander V. Chernikov free(pidx_first, M_IPFW); 3230b074b7bbSAlexander V. Chernikov 3231b074b7bbSAlexander V. Chernikov return (error); 3232b074b7bbSAlexander V. Chernikov } 3233b074b7bbSAlexander V. Chernikov 3234