Home | History | Annotate | Line # | Download | only in npf
npf_ctl.c revision 1.33.2.1
      1  1.33.2.1       tls /*	$NetBSD: npf_ctl.c,v 1.33.2.1 2014/08/10 06:56:16 tls Exp $	*/
      2       1.1     rmind 
      3       1.1     rmind /*-
      4      1.33     rmind  * Copyright (c) 2009-2014 The NetBSD Foundation, Inc.
      5       1.1     rmind  * All rights reserved.
      6       1.1     rmind  *
      7       1.1     rmind  * This material is based upon work partially supported by The
      8       1.1     rmind  * NetBSD Foundation under a contract with Mindaugas Rasiukevicius.
      9       1.1     rmind  *
     10       1.1     rmind  * Redistribution and use in source and binary forms, with or without
     11       1.1     rmind  * modification, are permitted provided that the following conditions
     12       1.1     rmind  * are met:
     13       1.1     rmind  * 1. Redistributions of source code must retain the above copyright
     14       1.1     rmind  *    notice, this list of conditions and the following disclaimer.
     15       1.1     rmind  * 2. Redistributions in binary form must reproduce the above copyright
     16       1.1     rmind  *    notice, this list of conditions and the following disclaimer in the
     17       1.1     rmind  *    documentation and/or other materials provided with the distribution.
     18       1.1     rmind  *
     19       1.1     rmind  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     20       1.1     rmind  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     21       1.1     rmind  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     22       1.1     rmind  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     23       1.1     rmind  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     24       1.1     rmind  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     25       1.1     rmind  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     26       1.1     rmind  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     27       1.1     rmind  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     28       1.1     rmind  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     29       1.1     rmind  * POSSIBILITY OF SUCH DAMAGE.
     30       1.1     rmind  */
     31       1.1     rmind 
     32       1.1     rmind /*
     33       1.1     rmind  * NPF device control.
     34       1.1     rmind  *
     35       1.1     rmind  * Implementation of (re)loading, construction of tables and rules.
     36       1.1     rmind  * NPF proplib(9) dictionary consumer.
     37       1.1     rmind  */
     38       1.1     rmind 
     39       1.1     rmind #include <sys/cdefs.h>
     40  1.33.2.1       tls __KERNEL_RCSID(0, "$NetBSD: npf_ctl.c,v 1.33.2.1 2014/08/10 06:56:16 tls Exp $");
     41       1.1     rmind 
     42       1.1     rmind #include <sys/param.h>
     43       1.1     rmind #include <sys/conf.h>
     44      1.21     rmind #include <sys/kmem.h>
     45      1.21     rmind #include <net/bpf.h>
     46       1.1     rmind 
     47       1.1     rmind #include <prop/proplib.h>
     48       1.1     rmind 
     49       1.1     rmind #include "npf_impl.h"
     50  1.33.2.1       tls #include "npf_conn.h"
     51       1.1     rmind 
     52      1.12     rmind #if defined(DEBUG) || defined(DIAGNOSTIC)
     53      1.12     rmind #define	NPF_ERR_DEBUG(e) \
     54      1.12     rmind 	prop_dictionary_set_cstring_nocopy((e), "source-file", __FILE__); \
     55      1.12     rmind 	prop_dictionary_set_uint32((e), "source-line", __LINE__);
     56      1.12     rmind #else
     57      1.12     rmind #define	NPF_ERR_DEBUG(e)
     58      1.12     rmind #endif
     59      1.12     rmind 
     60       1.1     rmind /*
     61       1.1     rmind  * npfctl_switch: enable or disable packet inspection.
     62       1.1     rmind  */
     63       1.1     rmind int
     64       1.1     rmind npfctl_switch(void *data)
     65       1.1     rmind {
     66       1.1     rmind 	const bool onoff = *(int *)data ? true : false;
     67       1.1     rmind 	int error;
     68       1.1     rmind 
     69       1.1     rmind 	if (onoff) {
     70       1.1     rmind 		/* Enable: add pfil hooks. */
     71      1.31     rmind 		error = npf_pfil_register(false);
     72       1.1     rmind 	} else {
     73       1.1     rmind 		/* Disable: remove pfil hooks. */
     74      1.31     rmind 		npf_pfil_unregister(false);
     75       1.1     rmind 		error = 0;
     76       1.1     rmind 	}
     77       1.1     rmind 	return error;
     78       1.1     rmind }
     79       1.1     rmind 
     80       1.6     rmind static int __noinline
     81      1.33     rmind npf_mk_table_entries(npf_table_t *t, prop_array_t entries)
     82      1.33     rmind {
     83      1.33     rmind 	prop_object_iterator_t eit;
     84      1.33     rmind 	prop_dictionary_t ent;
     85      1.33     rmind 	int error = 0;
     86      1.33     rmind 
     87      1.33     rmind 	/* Fill all the entries. */
     88      1.33     rmind 	eit = prop_array_iterator(entries);
     89      1.33     rmind 	while ((ent = prop_object_iterator_next(eit)) != NULL) {
     90      1.33     rmind 		const npf_addr_t *addr;
     91      1.33     rmind 		npf_netmask_t mask;
     92      1.33     rmind 		int alen;
     93      1.33     rmind 
     94      1.33     rmind 		/* Get address and mask.  Add a table entry. */
     95      1.33     rmind 		prop_object_t obj = prop_dictionary_get(ent, "addr");
     96      1.33     rmind 		addr = (const npf_addr_t *)prop_data_data_nocopy(obj);
     97      1.33     rmind 		prop_dictionary_get_uint8(ent, "mask", &mask);
     98      1.33     rmind 		alen = prop_data_size(obj);
     99      1.33     rmind 
    100      1.33     rmind 		error = npf_table_insert(t, alen, addr, mask);
    101      1.33     rmind 		if (error)
    102      1.33     rmind 			break;
    103      1.33     rmind 	}
    104      1.33     rmind 	prop_object_iterator_release(eit);
    105      1.33     rmind 	return error;
    106      1.33     rmind }
    107      1.33     rmind 
    108      1.33     rmind static int __noinline
    109      1.12     rmind npf_mk_tables(npf_tableset_t *tblset, prop_array_t tables,
    110      1.12     rmind     prop_dictionary_t errdict)
    111       1.1     rmind {
    112       1.1     rmind 	prop_object_iterator_t it;
    113       1.1     rmind 	prop_dictionary_t tbldict;
    114       1.1     rmind 	int error = 0;
    115       1.1     rmind 
    116       1.1     rmind 	/* Tables - array. */
    117      1.12     rmind 	if (prop_object_type(tables) != PROP_TYPE_ARRAY) {
    118      1.12     rmind 		NPF_ERR_DEBUG(errdict);
    119       1.1     rmind 		return EINVAL;
    120      1.12     rmind 	}
    121       1.1     rmind 
    122       1.1     rmind 	it = prop_array_iterator(tables);
    123       1.1     rmind 	while ((tbldict = prop_object_iterator_next(it)) != NULL) {
    124      1.32     rmind 		const char *name;
    125       1.1     rmind 		npf_table_t *t;
    126       1.1     rmind 		u_int tid;
    127       1.1     rmind 		int type;
    128       1.1     rmind 
    129       1.1     rmind 		/* Table - dictionary. */
    130       1.1     rmind 		if (prop_object_type(tbldict) != PROP_TYPE_DICTIONARY) {
    131      1.12     rmind 			NPF_ERR_DEBUG(errdict);
    132       1.1     rmind 			error = EINVAL;
    133       1.1     rmind 			break;
    134       1.1     rmind 		}
    135       1.1     rmind 
    136      1.32     rmind 		/* Table name, ID and type.  Validate them. */
    137      1.32     rmind 		if (!prop_dictionary_get_cstring_nocopy(tbldict, "name", &name)) {
    138      1.32     rmind 			NPF_ERR_DEBUG(errdict);
    139      1.32     rmind 			error = EINVAL;
    140      1.32     rmind 			break;
    141      1.32     rmind 		}
    142       1.6     rmind 		prop_dictionary_get_uint32(tbldict, "id", &tid);
    143       1.6     rmind 		prop_dictionary_get_int32(tbldict, "type", &type);
    144      1.32     rmind 		error = npf_table_check(tblset, name, tid, type);
    145      1.32     rmind 		if (error) {
    146      1.32     rmind 			NPF_ERR_DEBUG(errdict);
    147       1.1     rmind 			break;
    148      1.32     rmind 		}
    149       1.1     rmind 
    150      1.33     rmind 		/* Get the entries or binary data. */
    151      1.33     rmind 		prop_array_t entries = prop_dictionary_get(tbldict, "entries");
    152      1.33     rmind 		if (prop_object_type(entries) != PROP_TYPE_ARRAY) {
    153      1.33     rmind 			NPF_ERR_DEBUG(errdict);
    154      1.33     rmind 			error = EINVAL;
    155      1.33     rmind 			break;
    156      1.33     rmind 		}
    157      1.33     rmind 		prop_object_t obj = prop_dictionary_get(tbldict, "data");
    158      1.33     rmind 		void *blob = prop_data_data(obj);
    159      1.33     rmind 		size_t size = prop_data_size(obj);
    160      1.33     rmind 
    161      1.33     rmind 		if (type == NPF_TABLE_CDB && (blob == NULL || size == 0)) {
    162      1.33     rmind 			NPF_ERR_DEBUG(errdict);
    163      1.33     rmind 			error = EINVAL;
    164      1.33     rmind 			break;
    165      1.33     rmind 		}
    166  1.33.2.1       tls 		if (type == NPF_TABLE_HASH) {
    167  1.33.2.1       tls 			size = 1024; /* XXX */
    168  1.33.2.1       tls 		}
    169      1.33     rmind 
    170       1.1     rmind 		/* Create and insert the table. */
    171      1.33     rmind 		t = npf_table_create(name, tid, type, blob, size);
    172       1.1     rmind 		if (t == NULL) {
    173      1.12     rmind 			NPF_ERR_DEBUG(errdict);
    174       1.1     rmind 			error = ENOMEM;
    175       1.1     rmind 			break;
    176       1.1     rmind 		}
    177       1.1     rmind 		error = npf_tableset_insert(tblset, t);
    178       1.1     rmind 		KASSERT(error == 0);
    179       1.1     rmind 
    180      1.33     rmind 		if ((error = npf_mk_table_entries(t, entries)) != 0) {
    181      1.12     rmind 			NPF_ERR_DEBUG(errdict);
    182       1.1     rmind 			break;
    183       1.1     rmind 		}
    184  1.33.2.1       tls 		prop_dictionary_remove(tbldict, "entries");
    185       1.1     rmind 	}
    186       1.1     rmind 	prop_object_iterator_release(it);
    187       1.1     rmind 	/*
    188       1.4     rmind 	 * Note: in a case of error, caller will free the tableset.
    189       1.1     rmind 	 */
    190       1.1     rmind 	return error;
    191       1.1     rmind }
    192       1.1     rmind 
    193       1.5     rmind static npf_rproc_t *
    194      1.21     rmind npf_mk_singlerproc(prop_dictionary_t rpdict)
    195       1.5     rmind {
    196       1.5     rmind 	prop_object_iterator_t it;
    197      1.21     rmind 	prop_dictionary_t extdict;
    198      1.18     rmind 	prop_array_t extlist;
    199       1.5     rmind 	npf_rproc_t *rp;
    200      1.18     rmind 
    201      1.18     rmind 	extlist = prop_dictionary_get(rpdict, "extcalls");
    202      1.18     rmind 	if (prop_object_type(extlist) != PROP_TYPE_ARRAY) {
    203      1.18     rmind 		return NULL;
    204      1.18     rmind 	}
    205      1.18     rmind 
    206      1.18     rmind 	rp = npf_rproc_create(rpdict);
    207      1.21     rmind 	if (rp == NULL) {
    208      1.18     rmind 		return NULL;
    209      1.18     rmind 	}
    210      1.21     rmind 
    211      1.18     rmind 	it = prop_array_iterator(extlist);
    212      1.18     rmind 	while ((extdict = prop_object_iterator_next(it)) != NULL) {
    213      1.21     rmind 		const char *name;
    214      1.21     rmind 
    215      1.18     rmind 		if (!prop_dictionary_get_cstring_nocopy(extdict,
    216      1.18     rmind 		    "name", &name) || npf_ext_construct(name, rp, extdict)) {
    217      1.18     rmind 			npf_rproc_release(rp);
    218      1.18     rmind 			rp = NULL;
    219      1.18     rmind 			break;
    220      1.18     rmind 		}
    221      1.18     rmind 	}
    222      1.18     rmind 	prop_object_iterator_release(it);
    223      1.21     rmind 	return rp;
    224      1.21     rmind }
    225      1.21     rmind 
    226      1.21     rmind static int __noinline
    227      1.21     rmind npf_mk_rprocs(npf_rprocset_t *rpset, prop_array_t rprocs,
    228      1.21     rmind     prop_dictionary_t errdict)
    229      1.21     rmind {
    230      1.21     rmind 	prop_object_iterator_t it;
    231      1.21     rmind 	prop_dictionary_t rpdict;
    232      1.21     rmind 	int error = 0;
    233      1.21     rmind 
    234      1.21     rmind 	it = prop_array_iterator(rprocs);
    235      1.21     rmind 	while ((rpdict = prop_object_iterator_next(it)) != NULL) {
    236      1.21     rmind 		npf_rproc_t *rp;
    237      1.18     rmind 
    238      1.21     rmind 		if ((rp = npf_mk_singlerproc(rpdict)) == NULL) {
    239      1.21     rmind 			NPF_ERR_DEBUG(errdict);
    240      1.21     rmind 			error = EINVAL;
    241      1.21     rmind 			break;
    242      1.21     rmind 		}
    243      1.21     rmind 		npf_rprocset_insert(rpset, rp);
    244       1.5     rmind 	}
    245      1.21     rmind 	prop_object_iterator_release(it);
    246      1.21     rmind 	return error;
    247       1.5     rmind }
    248       1.5     rmind 
    249      1.24  christos static npf_alg_t *
    250      1.24  christos npf_mk_singlealg(prop_dictionary_t aldict)
    251      1.24  christos {
    252      1.24  christos 	const char *name;
    253      1.24  christos 
    254      1.24  christos 	if (!prop_dictionary_get_cstring_nocopy(aldict, "name", &name))
    255      1.24  christos 		return NULL;
    256      1.24  christos 	return npf_alg_construct(name);
    257      1.24  christos }
    258      1.24  christos 
    259      1.24  christos static int __noinline
    260      1.24  christos npf_mk_algs(prop_array_t alglist, prop_dictionary_t errdict)
    261      1.24  christos {
    262      1.24  christos 	prop_object_iterator_t it;
    263      1.24  christos 	prop_dictionary_t nadict;
    264      1.24  christos 	int error = 0;
    265      1.24  christos 
    266      1.24  christos 	it = prop_array_iterator(alglist);
    267      1.24  christos 	while ((nadict = prop_object_iterator_next(it)) != NULL) {
    268      1.24  christos 		if (npf_mk_singlealg(nadict) == NULL) {
    269      1.24  christos 			NPF_ERR_DEBUG(errdict);
    270      1.24  christos 			error = EINVAL;
    271      1.24  christos 			break;
    272      1.24  christos 		}
    273      1.24  christos 	}
    274      1.24  christos 	prop_object_iterator_release(it);
    275      1.24  christos 	return error;
    276      1.24  christos }
    277      1.24  christos 
    278       1.6     rmind static int __noinline
    279      1.21     rmind npf_mk_code(prop_object_t obj, int type, void **code, size_t *csize,
    280      1.12     rmind     prop_dictionary_t errdict)
    281      1.12     rmind {
    282      1.21     rmind 	const void *cptr;
    283      1.21     rmind 	size_t clen;
    284      1.21     rmind 	void *bc;
    285      1.12     rmind 
    286      1.29     rmind 	if (type != NPF_CODE_BPF) {
    287      1.29     rmind 		return ENOTSUP;
    288      1.29     rmind 	}
    289      1.21     rmind 	cptr = prop_data_data_nocopy(obj);
    290      1.21     rmind 	if (cptr == NULL || (clen = prop_data_size(obj)) == 0) {
    291      1.12     rmind 		NPF_ERR_DEBUG(errdict);
    292      1.21     rmind 		return EINVAL;
    293      1.12     rmind 	}
    294      1.29     rmind 	if (!npf_bpf_validate(cptr, clen)) {
    295      1.29     rmind 		NPF_ERR_DEBUG(errdict);
    296      1.29     rmind 		return EINVAL;
    297      1.12     rmind 	}
    298      1.21     rmind 	bc = kmem_alloc(clen, KM_SLEEP);
    299      1.21     rmind 	memcpy(bc, cptr, clen);
    300      1.21     rmind 
    301      1.21     rmind 	*code = bc;
    302      1.21     rmind 	*csize = clen;
    303      1.12     rmind 	return 0;
    304      1.12     rmind }
    305      1.12     rmind 
    306      1.12     rmind static int __noinline
    307      1.21     rmind npf_mk_singlerule(prop_dictionary_t rldict, npf_rprocset_t *rpset,
    308      1.21     rmind     npf_rule_t **rlret, prop_dictionary_t errdict)
    309       1.1     rmind {
    310      1.21     rmind 	npf_rule_t *rl;
    311      1.21     rmind 	const char *rname;
    312       1.1     rmind 	prop_object_t obj;
    313      1.21     rmind 	int p, error = 0;
    314       1.1     rmind 
    315       1.1     rmind 	/* Rule - dictionary. */
    316      1.12     rmind 	if (prop_object_type(rldict) != PROP_TYPE_DICTIONARY) {
    317      1.12     rmind 		NPF_ERR_DEBUG(errdict);
    318       1.1     rmind 		return EINVAL;
    319      1.12     rmind 	}
    320      1.21     rmind 	if ((rl = npf_rule_alloc(rldict)) == NULL) {
    321      1.21     rmind 		NPF_ERR_DEBUG(errdict);
    322      1.21     rmind 		return EINVAL;
    323      1.21     rmind 	}
    324       1.1     rmind 
    325      1.21     rmind 	/* Assign rule procedure, if any. */
    326      1.21     rmind 	if (prop_dictionary_get_cstring_nocopy(rldict, "rproc", &rname)) {
    327      1.21     rmind 		npf_rproc_t *rp;
    328      1.21     rmind 
    329      1.21     rmind 		if (rpset == NULL) {
    330      1.21     rmind 			error = EINVAL;
    331      1.21     rmind 			goto err;
    332      1.21     rmind 		}
    333      1.21     rmind 		if ((rp = npf_rprocset_lookup(rpset, rname)) == NULL) {
    334      1.18     rmind 			NPF_ERR_DEBUG(errdict);
    335      1.18     rmind 			error = EINVAL;
    336      1.18     rmind 			goto err;
    337      1.18     rmind 		}
    338      1.21     rmind 		npf_rule_setrproc(rl, rp);
    339      1.18     rmind 	}
    340      1.18     rmind 
    341      1.21     rmind 	/* Filter code (binary data). */
    342      1.21     rmind 	if ((obj = prop_dictionary_get(rldict, "code")) != NULL) {
    343      1.21     rmind 		int type;
    344      1.21     rmind 		size_t len;
    345      1.21     rmind 		void *code;
    346      1.21     rmind 
    347      1.21     rmind 		prop_dictionary_get_int32(rldict, "code-type", &type);
    348      1.21     rmind 		error = npf_mk_code(obj, type, &code, &len, errdict);
    349      1.12     rmind 		if (error) {
    350      1.12     rmind 			goto err;
    351       1.4     rmind 		}
    352      1.21     rmind 		npf_rule_setcode(rl, type, code, len);
    353       1.1     rmind 	}
    354       1.1     rmind 
    355      1.21     rmind 	*rlret = rl;
    356       1.6     rmind 	return 0;
    357      1.12     rmind err:
    358      1.21     rmind 	npf_rule_free(rl);
    359      1.12     rmind 	prop_dictionary_get_int32(rldict, "priority", &p); /* XXX */
    360      1.12     rmind 	prop_dictionary_set_int32(errdict, "id", p);
    361      1.12     rmind 	return error;
    362       1.6     rmind }
    363       1.6     rmind 
    364       1.6     rmind static int __noinline
    365      1.21     rmind npf_mk_rules(npf_ruleset_t *rlset, prop_array_t rules, npf_rprocset_t *rpset,
    366      1.12     rmind     prop_dictionary_t errdict)
    367       1.6     rmind {
    368       1.6     rmind 	prop_object_iterator_t it;
    369       1.6     rmind 	prop_dictionary_t rldict;
    370      1.21     rmind 	int error;
    371       1.6     rmind 
    372       1.6     rmind 	if (prop_object_type(rules) != PROP_TYPE_ARRAY) {
    373      1.12     rmind 		NPF_ERR_DEBUG(errdict);
    374       1.6     rmind 		return EINVAL;
    375       1.6     rmind 	}
    376       1.5     rmind 
    377       1.4     rmind 	error = 0;
    378       1.1     rmind 	it = prop_array_iterator(rules);
    379       1.1     rmind 	while ((rldict = prop_object_iterator_next(it)) != NULL) {
    380      1.21     rmind 		npf_rule_t *rl = NULL;
    381       1.1     rmind 
    382       1.1     rmind 		/* Generate a single rule. */
    383      1.21     rmind 		error = npf_mk_singlerule(rldict, rpset, &rl, errdict);
    384       1.6     rmind 		if (error) {
    385       1.1     rmind 			break;
    386       1.6     rmind 		}
    387       1.6     rmind 		npf_ruleset_insert(rlset, rl);
    388       1.1     rmind 	}
    389       1.1     rmind 	prop_object_iterator_release(it);
    390       1.1     rmind 	/*
    391       1.4     rmind 	 * Note: in a case of error, caller will free the ruleset.
    392       1.1     rmind 	 */
    393       1.1     rmind 	return error;
    394       1.1     rmind }
    395       1.1     rmind 
    396       1.6     rmind static int __noinline
    397      1.12     rmind npf_mk_natlist(npf_ruleset_t *nset, prop_array_t natlist,
    398      1.12     rmind     prop_dictionary_t errdict)
    399       1.1     rmind {
    400       1.1     rmind 	prop_object_iterator_t it;
    401       1.1     rmind 	prop_dictionary_t natdict;
    402       1.1     rmind 	int error;
    403       1.1     rmind 
    404       1.1     rmind 	/* NAT policies - array. */
    405      1.12     rmind 	if (prop_object_type(natlist) != PROP_TYPE_ARRAY) {
    406      1.12     rmind 		NPF_ERR_DEBUG(errdict);
    407       1.1     rmind 		return EINVAL;
    408      1.12     rmind 	}
    409       1.1     rmind 
    410       1.4     rmind 	error = 0;
    411       1.1     rmind 	it = prop_array_iterator(natlist);
    412       1.1     rmind 	while ((natdict = prop_object_iterator_next(it)) != NULL) {
    413      1.21     rmind 		npf_rule_t *rl = NULL;
    414       1.1     rmind 		npf_natpolicy_t *np;
    415       1.1     rmind 
    416       1.1     rmind 		/* NAT policy - dictionary. */
    417       1.1     rmind 		if (prop_object_type(natdict) != PROP_TYPE_DICTIONARY) {
    418      1.12     rmind 			NPF_ERR_DEBUG(errdict);
    419       1.1     rmind 			error = EINVAL;
    420       1.1     rmind 			break;
    421       1.1     rmind 		}
    422       1.1     rmind 
    423       1.1     rmind 		/*
    424       1.1     rmind 		 * NAT policies are standard rules, plus additional
    425       1.1     rmind 		 * information for translation.  Make a rule.
    426       1.1     rmind 		 */
    427      1.12     rmind 		error = npf_mk_singlerule(natdict, NULL, &rl, errdict);
    428       1.6     rmind 		if (error) {
    429       1.1     rmind 			break;
    430       1.6     rmind 		}
    431       1.6     rmind 		npf_ruleset_insert(nset, rl);
    432       1.6     rmind 
    433       1.6     rmind 		/* If rule is named, it is a group with NAT policies. */
    434       1.6     rmind 		if (prop_dictionary_get(natdict, "name") &&
    435       1.6     rmind 		    prop_dictionary_get(natdict, "subrules")) {
    436       1.6     rmind 			continue;
    437       1.6     rmind 		}
    438       1.1     rmind 
    439       1.1     rmind 		/* Allocate a new NAT policy and assign to the rule. */
    440       1.5     rmind 		np = npf_nat_newpolicy(natdict, nset);
    441       1.1     rmind 		if (np == NULL) {
    442      1.12     rmind 			NPF_ERR_DEBUG(errdict);
    443       1.1     rmind 			error = ENOMEM;
    444       1.1     rmind 			break;
    445       1.1     rmind 		}
    446       1.1     rmind 		npf_rule_setnat(rl, np);
    447       1.1     rmind 	}
    448       1.1     rmind 	prop_object_iterator_release(it);
    449       1.1     rmind 	/*
    450       1.1     rmind 	 * Note: in a case of error, caller will free entire NAT ruleset
    451       1.1     rmind 	 * with assigned NAT policies.
    452       1.1     rmind 	 */
    453       1.1     rmind 	return error;
    454       1.1     rmind }
    455       1.1     rmind 
    456       1.1     rmind /*
    457  1.33.2.1       tls  * npf_mk_connlist: import a list of connections and load them.
    458  1.33.2.1       tls  */
    459  1.33.2.1       tls static int __noinline
    460  1.33.2.1       tls npf_mk_connlist(prop_array_t conlist, npf_ruleset_t *natlist,
    461  1.33.2.1       tls     npf_conndb_t **conndb, prop_dictionary_t errdict)
    462  1.33.2.1       tls {
    463  1.33.2.1       tls 	prop_dictionary_t condict;
    464  1.33.2.1       tls 	prop_object_iterator_t it;
    465  1.33.2.1       tls 	npf_conndb_t *cd;
    466  1.33.2.1       tls 	int error;
    467  1.33.2.1       tls 
    468  1.33.2.1       tls 	/* Connection list - array */
    469  1.33.2.1       tls 	if (prop_object_type(conlist) != PROP_TYPE_ARRAY) {
    470  1.33.2.1       tls 		NPF_ERR_DEBUG(errdict);
    471  1.33.2.1       tls 		return EINVAL;
    472  1.33.2.1       tls 	}
    473  1.33.2.1       tls 
    474  1.33.2.1       tls 	/* Create a connection database. */
    475  1.33.2.1       tls 	cd = npf_conndb_create();
    476  1.33.2.1       tls 
    477  1.33.2.1       tls 	error = 0;
    478  1.33.2.1       tls 	it = prop_array_iterator(conlist);
    479  1.33.2.1       tls 	while ((condict = prop_object_iterator_next(it)) != NULL) {
    480  1.33.2.1       tls 		/* Connection - dictionary. */
    481  1.33.2.1       tls 		if (prop_object_type(condict) != PROP_TYPE_DICTIONARY) {
    482  1.33.2.1       tls 			NPF_ERR_DEBUG(errdict);
    483  1.33.2.1       tls 			error = EINVAL;
    484  1.33.2.1       tls 			break;
    485  1.33.2.1       tls 		}
    486  1.33.2.1       tls 		/* Construct and insert real connection structure. */
    487  1.33.2.1       tls 		error = npf_conn_import(cd, condict, natlist);
    488  1.33.2.1       tls 		if (error) {
    489  1.33.2.1       tls 			NPF_ERR_DEBUG(errdict);
    490  1.33.2.1       tls 			break;
    491  1.33.2.1       tls 		}
    492  1.33.2.1       tls 	}
    493  1.33.2.1       tls 	prop_object_iterator_release(it);
    494  1.33.2.1       tls 	if (error) {
    495  1.33.2.1       tls 		npf_conn_gc(cd, true, false);
    496  1.33.2.1       tls 		npf_conndb_destroy(cd);
    497  1.33.2.1       tls 	} else {
    498  1.33.2.1       tls 		*conndb = cd;
    499  1.33.2.1       tls 	}
    500  1.33.2.1       tls 	return error;
    501  1.33.2.1       tls }
    502  1.33.2.1       tls 
    503  1.33.2.1       tls /*
    504  1.33.2.1       tls  * npfctl_load: store passed data i.e. update settings, create passed
    505       1.1     rmind  * tables, rules and atomically activate all them.
    506       1.1     rmind  */
    507       1.1     rmind int
    508  1.33.2.1       tls npfctl_load(u_long cmd, void *data)
    509       1.1     rmind {
    510      1.12     rmind 	struct plistref *pref = data;
    511      1.14     rmind 	prop_dictionary_t npf_dict, errdict;
    512  1.33.2.1       tls 	prop_array_t alglist, natlist, tables, rprocs, rules, conlist;
    513       1.1     rmind 	npf_tableset_t *tblset = NULL;
    514      1.21     rmind 	npf_rprocset_t *rpset = NULL;
    515       1.1     rmind 	npf_ruleset_t *rlset = NULL;
    516       1.1     rmind 	npf_ruleset_t *nset = NULL;
    517  1.33.2.1       tls 	npf_conndb_t *conndb = NULL;
    518      1.20     rmind 	uint32_t ver = 0;
    519      1.21     rmind 	size_t nitems;
    520      1.11     rmind 	bool flush;
    521       1.1     rmind 	int error;
    522       1.1     rmind 
    523       1.1     rmind 	/* Retrieve the dictionary. */
    524      1.15     rmind #ifndef _NPF_TESTING
    525      1.14     rmind 	error = prop_dictionary_copyin_ioctl(pref, cmd, &npf_dict);
    526       1.1     rmind 	if (error)
    527       1.1     rmind 		return error;
    528       1.1     rmind #else
    529      1.15     rmind 	npf_dict = (prop_dictionary_t)pref;
    530       1.1     rmind #endif
    531      1.15     rmind 
    532      1.20     rmind 	/* Dictionary for error reporting and version check. */
    533      1.12     rmind 	errdict = prop_dictionary_create();
    534      1.20     rmind 	prop_dictionary_get_uint32(npf_dict, "version", &ver);
    535      1.20     rmind 	if (ver != NPF_VERSION) {
    536      1.20     rmind 		error = EPROGMISMATCH;
    537      1.20     rmind 		goto fail;
    538      1.20     rmind 	}
    539      1.12     rmind 
    540      1.24  christos 	/* ALGs. */
    541      1.24  christos 	alglist = prop_dictionary_get(npf_dict, "algs");
    542      1.24  christos 	error = npf_mk_algs(alglist, errdict);
    543      1.24  christos 	if (error) {
    544      1.24  christos 		goto fail;
    545      1.24  christos 	}
    546      1.24  christos 
    547       1.1     rmind 	/* NAT policies. */
    548      1.14     rmind 	natlist = prop_dictionary_get(npf_dict, "translation");
    549      1.30     rmind 	if ((nitems = prop_array_count(natlist)) > NPF_MAX_RULES) {
    550      1.30     rmind 		goto fail;
    551      1.30     rmind 	}
    552      1.21     rmind 
    553      1.21     rmind 	nset = npf_ruleset_create(nitems);
    554      1.12     rmind 	error = npf_mk_natlist(nset, natlist, errdict);
    555      1.10     rmind 	if (error) {
    556       1.1     rmind 		goto fail;
    557      1.10     rmind 	}
    558       1.1     rmind 
    559       1.1     rmind 	/* Tables. */
    560      1.14     rmind 	tables = prop_dictionary_get(npf_dict, "tables");
    561      1.32     rmind 	if ((nitems = prop_array_count(tables)) > NPF_MAX_TABLES) {
    562      1.32     rmind 		goto fail;
    563      1.32     rmind 	}
    564      1.32     rmind 	tblset = npf_tableset_create(nitems);
    565      1.12     rmind 	error = npf_mk_tables(tblset, tables, errdict);
    566      1.10     rmind 	if (error) {
    567       1.1     rmind 		goto fail;
    568      1.10     rmind 	}
    569       1.1     rmind 
    570      1.21     rmind 	/* Rule procedures. */
    571      1.32     rmind 	rprocs = prop_dictionary_get(npf_dict, "rprocs");
    572      1.32     rmind 	if ((nitems = prop_array_count(rprocs)) > NPF_MAX_RPROCS) {
    573      1.32     rmind 		goto fail;
    574      1.32     rmind 	}
    575      1.21     rmind 	rpset = npf_rprocset_create();
    576      1.21     rmind 	error = npf_mk_rprocs(rpset, rprocs, errdict);
    577      1.21     rmind 	if (error) {
    578      1.21     rmind 		goto fail;
    579      1.21     rmind 	}
    580      1.21     rmind 
    581      1.21     rmind 	/* Rules. */
    582      1.14     rmind 	rules = prop_dictionary_get(npf_dict, "rules");
    583      1.30     rmind 	if ((nitems = prop_array_count(rules)) > NPF_MAX_RULES) {
    584      1.30     rmind 		goto fail;
    585      1.30     rmind 	}
    586      1.21     rmind 
    587      1.21     rmind 	rlset = npf_ruleset_create(nitems);
    588      1.21     rmind 	error = npf_mk_rules(rlset, rules, rpset, errdict);
    589      1.10     rmind 	if (error) {
    590       1.1     rmind 		goto fail;
    591      1.10     rmind 	}
    592       1.1     rmind 
    593  1.33.2.1       tls 	/* Connections (if loading any). */
    594  1.33.2.1       tls 	if ((conlist = prop_dictionary_get(npf_dict, "conn-list")) != NULL) {
    595  1.33.2.1       tls 		error = npf_mk_connlist(conlist, nset, &conndb, errdict);
    596  1.33.2.1       tls 		if (error) {
    597  1.33.2.1       tls 			goto fail;
    598  1.33.2.1       tls 		}
    599  1.33.2.1       tls 		prop_dictionary_remove(npf_dict, "conn-list");
    600  1.33.2.1       tls 	}
    601  1.33.2.1       tls 
    602      1.11     rmind 	flush = false;
    603      1.14     rmind 	prop_dictionary_get_bool(npf_dict, "flush", &flush);
    604      1.11     rmind 
    605       1.1     rmind 	/*
    606  1.33.2.1       tls 	 * Finally - perform the load.
    607       1.1     rmind 	 */
    608  1.33.2.1       tls 	npf_config_load(npf_dict, rlset, tblset, nset, rpset, conndb, flush);
    609      1.11     rmind 
    610       1.1     rmind 	/* Done.  Since data is consumed now, we shall not destroy it. */
    611       1.1     rmind 	tblset = NULL;
    612      1.21     rmind 	rpset = NULL;
    613       1.1     rmind 	rlset = NULL;
    614       1.1     rmind 	nset = NULL;
    615       1.1     rmind fail:
    616       1.1     rmind 	/*
    617       1.1     rmind 	 * Note: destroy rulesets first, to drop references to the tableset.
    618       1.1     rmind 	 */
    619      1.21     rmind 	KASSERT(error == 0 || (nset || rpset || rlset || tblset));
    620       1.1     rmind 	if (nset) {
    621       1.1     rmind 		npf_ruleset_destroy(nset);
    622       1.1     rmind 	}
    623       1.1     rmind 	if (rlset) {
    624       1.1     rmind 		npf_ruleset_destroy(rlset);
    625       1.1     rmind 	}
    626      1.21     rmind 	if (rpset) {
    627      1.21     rmind 		npf_rprocset_destroy(rpset);
    628      1.21     rmind 	}
    629       1.1     rmind 	if (tblset) {
    630       1.1     rmind 		npf_tableset_destroy(tblset);
    631       1.1     rmind 	}
    632      1.14     rmind 	if (error) {
    633      1.14     rmind 		prop_object_release(npf_dict);
    634      1.14     rmind 	}
    635      1.12     rmind 
    636      1.12     rmind 	/* Error report. */
    637      1.18     rmind #ifndef _NPF_TESTING
    638      1.12     rmind 	prop_dictionary_set_int32(errdict, "errno", error);
    639      1.12     rmind 	prop_dictionary_copyout_ioctl(pref, cmd, errdict);
    640      1.18     rmind 	prop_object_release(errdict);
    641      1.18     rmind 	error = 0;
    642      1.13     rmind #endif
    643      1.18     rmind 	return error;
    644       1.6     rmind }
    645       1.6     rmind 
    646      1.21     rmind /*
    647  1.33.2.1       tls  * npfctl_save: export the config dictionary as it was submitted,
    648  1.33.2.1       tls  * including the current snapshot of the connections.  Additionally,
    649  1.33.2.1       tls  * indicate whether the ruleset is currently active.
    650  1.33.2.1       tls  */
    651  1.33.2.1       tls int
    652  1.33.2.1       tls npfctl_save(u_long cmd, void *data)
    653  1.33.2.1       tls {
    654  1.33.2.1       tls 	struct plistref *pref = data;
    655  1.33.2.1       tls 	prop_dictionary_t npf_dict;
    656  1.33.2.1       tls 	prop_array_t conlist;
    657  1.33.2.1       tls 	int error;
    658  1.33.2.1       tls 
    659  1.33.2.1       tls 	npf_config_enter();
    660  1.33.2.1       tls 	conlist = prop_array_create();
    661  1.33.2.1       tls 
    662  1.33.2.1       tls 	/* Serialise the connections. */
    663  1.33.2.1       tls 	error = npf_conn_export(conlist);
    664  1.33.2.1       tls 	if (error) {
    665  1.33.2.1       tls 		prop_object_release(conlist);
    666  1.33.2.1       tls 		goto out;
    667  1.33.2.1       tls 	}
    668  1.33.2.1       tls 
    669  1.33.2.1       tls 	npf_dict = npf_config_dict();
    670  1.33.2.1       tls 	prop_dictionary_set_bool(npf_dict, "active", npf_pfil_registered_p());
    671  1.33.2.1       tls 	prop_dictionary_set_and_rel(npf_dict, "conn-list", conlist);
    672  1.33.2.1       tls 
    673  1.33.2.1       tls 	error = prop_dictionary_copyout_ioctl(pref, cmd, npf_dict);
    674  1.33.2.1       tls out:
    675  1.33.2.1       tls 	npf_config_exit();
    676  1.33.2.1       tls 	return error;
    677  1.33.2.1       tls }
    678  1.33.2.1       tls 
    679  1.33.2.1       tls /*
    680      1.21     rmind  * npfctl_rule: add or remove dynamic rules in the specified ruleset.
    681      1.21     rmind  */
    682      1.14     rmind int
    683      1.21     rmind npfctl_rule(u_long cmd, void *data)
    684      1.14     rmind {
    685      1.14     rmind 	struct plistref *pref = data;
    686      1.21     rmind 	prop_dictionary_t npf_rule, retdict = NULL;
    687      1.21     rmind 	npf_ruleset_t *rlset;
    688      1.21     rmind 	npf_rule_t *rl = NULL;
    689      1.21     rmind 	const char *ruleset_name;
    690      1.21     rmind 	uint32_t rcmd = 0;
    691      1.14     rmind 	int error;
    692      1.14     rmind 
    693      1.21     rmind 	error = prop_dictionary_copyin_ioctl(pref, cmd, &npf_rule);
    694      1.21     rmind 	if (error) {
    695      1.21     rmind 		return error;
    696      1.21     rmind 	}
    697      1.21     rmind 	prop_dictionary_get_uint32(npf_rule, "command", &rcmd);
    698      1.21     rmind 	if (!prop_dictionary_get_cstring_nocopy(npf_rule,
    699      1.21     rmind 	    "ruleset-name", &ruleset_name)) {
    700      1.27     rmind 		error = EINVAL;
    701      1.27     rmind 		goto out;
    702      1.21     rmind 	}
    703      1.21     rmind 
    704      1.21     rmind 	if (rcmd == NPF_CMD_RULE_ADD) {
    705      1.27     rmind 		retdict = prop_dictionary_create();
    706      1.27     rmind 		if (npf_mk_singlerule(npf_rule, NULL, &rl, retdict) != 0) {
    707      1.27     rmind 			error = EINVAL;
    708      1.27     rmind 			goto out;
    709      1.21     rmind 		}
    710      1.21     rmind 	}
    711      1.21     rmind 
    712      1.21     rmind 	npf_config_enter();
    713      1.21     rmind 	rlset = npf_config_ruleset();
    714      1.21     rmind 
    715      1.21     rmind 	switch (rcmd) {
    716      1.21     rmind 	case NPF_CMD_RULE_ADD: {
    717      1.21     rmind 		if ((error = npf_ruleset_add(rlset, ruleset_name, rl)) == 0) {
    718      1.21     rmind 			/* Success. */
    719      1.23     rmind 			uint64_t id = npf_rule_getid(rl);
    720      1.23     rmind 			prop_dictionary_set_uint64(retdict, "id", id);
    721      1.21     rmind 			rl = NULL;
    722      1.21     rmind 		}
    723      1.21     rmind 		break;
    724      1.21     rmind 	}
    725      1.21     rmind 	case NPF_CMD_RULE_REMOVE: {
    726      1.23     rmind 		uint64_t id;
    727      1.21     rmind 
    728      1.23     rmind 		if (!prop_dictionary_get_uint64(npf_rule, "id", &id)) {
    729      1.21     rmind 			error = EINVAL;
    730      1.21     rmind 			break;
    731      1.21     rmind 		}
    732      1.23     rmind 		error = npf_ruleset_remove(rlset, ruleset_name, id);
    733      1.21     rmind 		break;
    734      1.21     rmind 	}
    735      1.21     rmind 	case NPF_CMD_RULE_REMKEY: {
    736      1.21     rmind 		prop_object_t obj = prop_dictionary_get(npf_rule, "key");
    737      1.21     rmind 		const void *key = prop_data_data_nocopy(obj);
    738      1.21     rmind 		size_t len = prop_data_size(obj);
    739      1.21     rmind 
    740      1.21     rmind 		if (len == 0 || len > NPF_RULE_MAXKEYLEN) {
    741      1.21     rmind 			error = EINVAL;
    742      1.21     rmind 			break;
    743      1.21     rmind 		}
    744      1.22     rmind 		error = npf_ruleset_remkey(rlset, ruleset_name, key, len);
    745      1.22     rmind 		break;
    746      1.22     rmind 	}
    747      1.22     rmind 	case NPF_CMD_RULE_LIST: {
    748      1.22     rmind 		retdict = npf_ruleset_list(rlset, ruleset_name);
    749      1.22     rmind 		break;
    750      1.22     rmind 	}
    751      1.22     rmind 	case NPF_CMD_RULE_FLUSH: {
    752      1.22     rmind 		error = npf_ruleset_flush(rlset, ruleset_name);
    753      1.21     rmind 		break;
    754      1.21     rmind 	}
    755      1.21     rmind 	default:
    756      1.21     rmind 		error = EINVAL;
    757      1.21     rmind 		break;
    758      1.21     rmind 	}
    759      1.22     rmind 
    760      1.22     rmind 	/* Destroy any removed rules. */
    761      1.22     rmind 	if (!error && rcmd != NPF_CMD_RULE_ADD && rcmd != NPF_CMD_RULE_LIST) {
    762      1.22     rmind 		npf_config_sync();
    763      1.22     rmind 		npf_ruleset_gc(rlset);
    764      1.22     rmind 	}
    765      1.21     rmind 	npf_config_exit();
    766      1.14     rmind 
    767      1.21     rmind 	if (rl) {
    768      1.21     rmind 		npf_rule_free(rl);
    769      1.21     rmind 	}
    770      1.27     rmind out:
    771      1.21     rmind 	if (retdict) {
    772      1.21     rmind 		prop_object_release(npf_rule);
    773      1.21     rmind 		prop_dictionary_copyout_ioctl(pref, cmd, retdict);
    774      1.21     rmind 		prop_object_release(retdict);
    775      1.21     rmind 	}
    776      1.14     rmind 	return error;
    777      1.14     rmind }
    778      1.14     rmind 
    779       1.6     rmind /*
    780       1.2     rmind  * npfctl_table: add, remove or query entries in the specified table.
    781       1.1     rmind  *
    782       1.1     rmind  * For maximum performance, interface is avoiding proplib(3)'s overhead.
    783       1.1     rmind  */
    784       1.1     rmind int
    785       1.1     rmind npfctl_table(void *data)
    786       1.1     rmind {
    787      1.19     rmind 	const npf_ioctl_table_t *nct = data;
    788      1.32     rmind 	char tname[NPF_TABLE_MAXNAMELEN];
    789      1.32     rmind 	npf_tableset_t *ts;
    790      1.32     rmind 	npf_table_t *t;
    791      1.32     rmind 	int s, error;
    792      1.32     rmind 
    793      1.32     rmind 	error = copyinstr(nct->nct_name, tname, sizeof(tname), NULL);
    794      1.32     rmind 	if (error) {
    795      1.32     rmind 		return error;
    796      1.32     rmind 	}
    797       1.1     rmind 
    798      1.32     rmind 	s = npf_config_read_enter(); /* XXX */
    799      1.32     rmind 	ts = npf_config_tableset();
    800      1.32     rmind 	if ((t = npf_tableset_getbyname(ts, tname)) == NULL) {
    801      1.32     rmind 		npf_config_read_exit(s);
    802      1.32     rmind 		return EINVAL;
    803      1.32     rmind 	}
    804      1.21     rmind 
    805      1.21     rmind 	switch (nct->nct_cmd) {
    806      1.21     rmind 	case NPF_CMD_TABLE_LOOKUP:
    807      1.32     rmind 		error = npf_table_lookup(t, nct->nct_data.ent.alen,
    808      1.32     rmind 		    &nct->nct_data.ent.addr);
    809      1.20     rmind 		break;
    810      1.21     rmind 	case NPF_CMD_TABLE_ADD:
    811      1.32     rmind 		error = npf_table_insert(t, nct->nct_data.ent.alen,
    812      1.32     rmind 		    &nct->nct_data.ent.addr, nct->nct_data.ent.mask);
    813       1.1     rmind 		break;
    814      1.21     rmind 	case NPF_CMD_TABLE_REMOVE:
    815      1.32     rmind 		error = npf_table_remove(t, nct->nct_data.ent.alen,
    816      1.32     rmind 		    &nct->nct_data.ent.addr, nct->nct_data.ent.mask);
    817      1.19     rmind 		break;
    818      1.21     rmind 	case NPF_CMD_TABLE_LIST:
    819      1.32     rmind 		error = npf_table_list(t, nct->nct_data.buf.buf,
    820      1.32     rmind 		    nct->nct_data.buf.len);
    821       1.1     rmind 		break;
    822      1.25     rmind 	case NPF_CMD_TABLE_FLUSH:
    823      1.32     rmind 		error = npf_table_flush(t);
    824      1.25     rmind 		break;
    825       1.1     rmind 	default:
    826      1.19     rmind 		error = EINVAL;
    827      1.19     rmind 		break;
    828       1.1     rmind 	}
    829      1.32     rmind 	npf_config_read_exit(s);
    830      1.21     rmind 
    831       1.1     rmind 	return error;
    832       1.1     rmind }
    833