Home | History | Annotate | Line # | Download | only in npf
npf_impl.h revision 1.32
      1 /*	$NetBSD: npf_impl.h,v 1.32 2013/09/19 01:04:46 rmind Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 2009-2013 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * This material is based upon work partially supported by The
      8  * NetBSD Foundation under a contract with Mindaugas Rasiukevicius.
      9  *
     10  * Redistribution and use in source and binary forms, with or without
     11  * modification, are permitted provided that the following conditions
     12  * are met:
     13  * 1. Redistributions of source code must retain the above copyright
     14  *    notice, this list of conditions and the following disclaimer.
     15  * 2. Redistributions in binary form must reproduce the above copyright
     16  *    notice, this list of conditions and the following disclaimer in the
     17  *    documentation and/or other materials provided with the distribution.
     18  *
     19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     29  * POSSIBILITY OF SUCH DAMAGE.
     30  */
     31 
     32 /*
     33  * Private NPF structures and interfaces.
     34  * For internal use within NPF core only.
     35  */
     36 
     37 #ifndef _NPF_IMPL_H_
     38 #define _NPF_IMPL_H_
     39 
     40 #if !defined(_KERNEL)
     41 #error "Kernel-level header only"
     42 #endif
     43 
     44 #ifdef _KERNEL_OPT
     45 /* For INET/INET6 definitions. */
     46 #include "opt_inet.h"
     47 #include "opt_inet6.h"
     48 #endif
     49 
     50 #include <sys/types.h>
     51 #include <sys/queue.h>
     52 #include <sys/hash.h>
     53 #include <sys/rbtree.h>
     54 #include <sys/ptree.h>
     55 #include <sys/rwlock.h>
     56 
     57 #include <net/bpf.h>
     58 #include <net/bpfjit.h>
     59 #include <net/if.h>
     60 
     61 #include "npf.h"
     62 #include "npf_ncode.h"
     63 
     64 #ifdef _NPF_DEBUG
     65 #define	NPF_PRINTF(x)	printf x
     66 #else
     67 #define	NPF_PRINTF(x)
     68 #endif
     69 
     70 /*
     71  * STRUCTURE DECLARATIONS.
     72  */
     73 
     74 struct npf_ruleset;
     75 struct npf_rule;
     76 struct npf_rprocset;
     77 struct npf_nat;
     78 struct npf_session;
     79 
     80 typedef struct npf_ruleset	npf_ruleset_t;
     81 typedef struct npf_rule		npf_rule_t;
     82 typedef struct npf_nat		npf_nat_t;
     83 typedef struct npf_rprocset	npf_rprocset_t;
     84 typedef struct npf_alg		npf_alg_t;
     85 typedef struct npf_natpolicy	npf_natpolicy_t;
     86 typedef struct npf_session	npf_session_t;
     87 
     88 struct npf_sehash;
     89 struct npf_table;
     90 
     91 typedef struct npf_sehash	npf_sehash_t;
     92 typedef struct npf_table	npf_table_t;
     93 
     94 typedef npf_table_t *		npf_tableset_t;
     95 
     96 /*
     97  * DEFINITIONS.
     98  */
     99 
    100 typedef bool (*npf_alg_func_t)(npf_cache_t *, nbuf_t *, npf_nat_t *, int);
    101 typedef npf_session_t *(*npf_alg_sfunc_t)(npf_cache_t *, nbuf_t *, int);
    102 typedef void (*npf_workfunc_t)(void);
    103 
    104 #define	NPF_NCODE_LIMIT		1024
    105 #define	NPF_TABLE_SLOTS		32
    106 
    107 /*
    108  * SESSION STATE STRUCTURES
    109  */
    110 
    111 #define	NPF_FLOW_FORW		0
    112 #define	NPF_FLOW_BACK		1
    113 
    114 typedef struct {
    115 	uint32_t	nst_end;
    116 	uint32_t	nst_maxend;
    117 	uint32_t	nst_maxwin;
    118 	int		nst_wscale;
    119 } npf_tcpstate_t;
    120 
    121 typedef struct {
    122 	kmutex_t	nst_lock;
    123 	int		nst_state;
    124 	npf_tcpstate_t	nst_tcpst[2];
    125 } npf_state_t;
    126 
    127 /*
    128  * INTERFACES.
    129  */
    130 
    131 /* NPF config, statistics, etc. */
    132 void		npf_config_init(void);
    133 void		npf_config_fini(void);
    134 
    135 void		npf_config_enter(void);
    136 void		npf_config_exit(void);
    137 void		npf_config_sync(void);
    138 bool		npf_config_locked_p(void);
    139 int		npf_config_read_enter(void);
    140 void		npf_config_read_exit(int);
    141 
    142 void		npf_config_reload(prop_dictionary_t, npf_ruleset_t *,
    143 		    npf_tableset_t *, npf_ruleset_t *, npf_rprocset_t *, bool);
    144 npf_ruleset_t *	npf_config_ruleset(void);
    145 npf_ruleset_t *	npf_config_natset(void);
    146 npf_tableset_t *npf_config_tableset(void);
    147 prop_dictionary_t npf_config_dict(void);
    148 bool		npf_default_pass(void);
    149 
    150 int		npf_worker_sysinit(void);
    151 void		npf_worker_sysfini(void);
    152 void		npf_worker_signal(void);
    153 void		npf_worker_register(npf_workfunc_t);
    154 void		npf_worker_unregister(npf_workfunc_t);
    155 
    156 void		npflogattach(int);
    157 void		npflogdetach(void);
    158 int		npfctl_switch(void *);
    159 int		npfctl_reload(u_long, void *);
    160 int		npfctl_getconf(u_long, void *);
    161 int		npfctl_sessions_save(u_long, void *);
    162 int		npfctl_sessions_load(u_long, void *);
    163 int		npfctl_rule(u_long, void *);
    164 int		npfctl_table(void *);
    165 
    166 void		npf_stats_inc(npf_stats_t);
    167 void		npf_stats_dec(npf_stats_t);
    168 
    169 /* Packet filter hooks. */
    170 int		npf_pfil_register(void);
    171 void		npf_pfil_unregister(void);
    172 bool		npf_pfil_registered_p(void);
    173 int		npf_packet_handler(void *, struct mbuf **, ifnet_t *, int);
    174 
    175 /* Protocol helpers. */
    176 int		npf_cache_all(npf_cache_t *, nbuf_t *);
    177 void		npf_recache(npf_cache_t *, nbuf_t *);
    178 
    179 bool		npf_rwrip(const npf_cache_t *, int, const npf_addr_t *);
    180 bool		npf_rwrport(const npf_cache_t *, int, const in_port_t);
    181 bool		npf_rwrcksum(const npf_cache_t *, const int,
    182 		    const npf_addr_t *, const in_port_t);
    183 
    184 uint16_t	npf_fixup16_cksum(uint16_t, uint16_t, uint16_t);
    185 uint16_t	npf_fixup32_cksum(uint16_t, uint32_t, uint32_t);
    186 uint16_t	npf_addr_cksum(uint16_t, int, const npf_addr_t *,
    187 		    const npf_addr_t *);
    188 uint32_t	npf_addr_sum(const int, const npf_addr_t *, const npf_addr_t *);
    189 int		npf_addr_cmp(const npf_addr_t *, const npf_netmask_t,
    190 		    const npf_addr_t *, const npf_netmask_t, const int);
    191 void		npf_addr_mask(const npf_addr_t *, const npf_netmask_t,
    192 		    const int, npf_addr_t *);
    193 
    194 int		npf_tcpsaw(const npf_cache_t *, tcp_seq *, tcp_seq *,
    195 		    uint32_t *);
    196 bool		npf_fetch_tcpopts(npf_cache_t *, nbuf_t *, uint16_t *, int *);
    197 bool		npf_return_block(npf_cache_t *, nbuf_t *, const int);
    198 
    199 /* BPF interface. */
    200 void		npf_bpf_sysinit(void);
    201 void		npf_bpf_sysfini(void);
    202 int		npf_bpf_filter(npf_cache_t *, nbuf_t *,
    203 		    const void *, bpfjit_function_t);
    204 bool		npf_bpf_validate(const void *, size_t);
    205 
    206 /* Complex instructions. */
    207 int		npf_match_ether(nbuf_t *, int, uint16_t, uint32_t *);
    208 int		npf_match_proto(const npf_cache_t *, uint32_t);
    209 int		npf_match_table(const npf_cache_t *, int, u_int);
    210 int		npf_match_ipmask(const npf_cache_t *, int,
    211 		    const npf_addr_t *, npf_netmask_t);
    212 int		npf_match_tcp_ports(const npf_cache_t *, int, uint32_t);
    213 int		npf_match_udp_ports(const npf_cache_t *, int, uint32_t);
    214 int		npf_match_icmp4(const npf_cache_t *, uint32_t);
    215 int		npf_match_icmp6(const npf_cache_t *, uint32_t);
    216 int		npf_match_tcpfl(const npf_cache_t *, uint32_t);
    217 
    218 /* Tableset interface. */
    219 void		npf_tableset_sysinit(void);
    220 void		npf_tableset_sysfini(void);
    221 
    222 extern const pt_tree_ops_t npf_table_ptree_ops;
    223 
    224 npf_tableset_t *npf_tableset_create(void);
    225 void		npf_tableset_destroy(npf_tableset_t *);
    226 int		npf_tableset_insert(npf_tableset_t *, npf_table_t *);
    227 void		npf_tableset_reload(npf_tableset_t *, npf_tableset_t *);
    228 
    229 npf_table_t *	npf_table_create(u_int, int, size_t);
    230 void		npf_table_destroy(npf_table_t *);
    231 
    232 int		npf_table_check(const npf_tableset_t *, u_int, int);
    233 int		npf_table_insert(npf_tableset_t *, u_int,
    234 		    const int, const npf_addr_t *, const npf_netmask_t);
    235 int		npf_table_remove(npf_tableset_t *, u_int,
    236 		    const int, const npf_addr_t *, const npf_netmask_t);
    237 int		npf_table_lookup(npf_tableset_t *, u_int,
    238 		    const int, const npf_addr_t *);
    239 int		npf_table_list(npf_tableset_t *, u_int, void *, size_t);
    240 int		npf_table_flush(npf_tableset_t *, u_int);
    241 
    242 /* Ruleset interface. */
    243 npf_ruleset_t *	npf_ruleset_create(size_t);
    244 void		npf_ruleset_destroy(npf_ruleset_t *);
    245 void		npf_ruleset_insert(npf_ruleset_t *, npf_rule_t *);
    246 void		npf_ruleset_reload(npf_ruleset_t *, npf_ruleset_t *);
    247 void		npf_ruleset_natreload(npf_ruleset_t *, npf_ruleset_t *);
    248 npf_rule_t *	npf_ruleset_matchnat(npf_ruleset_t *, npf_natpolicy_t *);
    249 npf_rule_t *	npf_ruleset_sharepm(npf_ruleset_t *, npf_natpolicy_t *);
    250 void		npf_ruleset_freealg(npf_ruleset_t *, npf_alg_t *);
    251 
    252 int		npf_ruleset_add(npf_ruleset_t *, const char *, npf_rule_t *);
    253 int		npf_ruleset_remove(npf_ruleset_t *, const char *, uint64_t);
    254 int		npf_ruleset_remkey(npf_ruleset_t *, const char *,
    255 		    const void *, size_t);
    256 prop_dictionary_t npf_ruleset_list(npf_ruleset_t *, const char *);
    257 int		npf_ruleset_flush(npf_ruleset_t *, const char *);
    258 void		npf_ruleset_gc(npf_ruleset_t *);
    259 
    260 npf_rule_t *	npf_ruleset_inspect(npf_cache_t *, nbuf_t *,
    261 		    const npf_ruleset_t *, const int, const int);
    262 int		npf_rule_conclude(const npf_rule_t *, int *);
    263 
    264 /* Rule interface. */
    265 npf_rule_t *	npf_rule_alloc(prop_dictionary_t);
    266 void		npf_rule_setcode(npf_rule_t *, int, void *, size_t);
    267 void		npf_rule_setrproc(npf_rule_t *, npf_rproc_t *);
    268 void		npf_rule_free(npf_rule_t *);
    269 uint64_t	npf_rule_getid(const npf_rule_t *);
    270 npf_natpolicy_t *npf_rule_getnat(const npf_rule_t *);
    271 void		npf_rule_setnat(npf_rule_t *, npf_natpolicy_t *);
    272 npf_rproc_t *	npf_rule_getrproc(npf_rule_t *);
    273 
    274 void		npf_ext_sysinit(void);
    275 void		npf_ext_sysfini(void);
    276 int		npf_ext_construct(const char *,
    277 		    npf_rproc_t *, prop_dictionary_t);
    278 
    279 npf_rprocset_t *npf_rprocset_create(void);
    280 void		npf_rprocset_destroy(npf_rprocset_t *);
    281 npf_rproc_t *	npf_rprocset_lookup(npf_rprocset_t *, const char *);
    282 void		npf_rprocset_insert(npf_rprocset_t *, npf_rproc_t *);
    283 
    284 npf_rproc_t *	npf_rproc_create(prop_dictionary_t);
    285 void		npf_rproc_acquire(npf_rproc_t *);
    286 void		npf_rproc_release(npf_rproc_t *);
    287 void		npf_rproc_run(npf_cache_t *, nbuf_t *, npf_rproc_t *, int *);
    288 
    289 /* Session handling interface. */
    290 void		npf_session_sysinit(void);
    291 void		npf_session_sysfini(void);
    292 void		npf_session_tracking(bool);
    293 
    294 npf_sehash_t *	sess_htable_create(void);
    295 void		sess_htable_destroy(npf_sehash_t *);
    296 
    297 npf_session_t *	npf_session_lookup(const npf_cache_t *, const nbuf_t *,
    298 		    const int, bool *);
    299 npf_session_t *	npf_session_inspect(npf_cache_t *, nbuf_t *, const int, int *);
    300 npf_session_t *	npf_session_establish(npf_cache_t *, nbuf_t *, const int);
    301 void		npf_session_release(npf_session_t *);
    302 void		npf_session_expire(npf_session_t *);
    303 bool		npf_session_pass(const npf_session_t *, npf_rproc_t **);
    304 void		npf_session_setpass(npf_session_t *, npf_rproc_t *);
    305 int		npf_session_setnat(npf_session_t *, npf_nat_t *, const int);
    306 npf_nat_t *	npf_session_retnat(npf_session_t *, const int, bool *);
    307 
    308 void		npf_session_load(npf_sehash_t *);
    309 int		npf_session_save(prop_array_t, prop_array_t);
    310 int		npf_session_restore(npf_sehash_t *, prop_dictionary_t);
    311 
    312 /* State handling. */
    313 bool		npf_state_init(npf_cache_t *, nbuf_t *, npf_state_t *);
    314 bool		npf_state_inspect(npf_cache_t *, nbuf_t *, npf_state_t *,
    315 		    const bool);
    316 int		npf_state_etime(const npf_state_t *, const int);
    317 void		npf_state_destroy(npf_state_t *);
    318 
    319 bool		npf_state_tcp(npf_cache_t *, nbuf_t *, npf_state_t *, int);
    320 int		npf_state_tcp_timeout(const npf_state_t *);
    321 
    322 /* NAT. */
    323 void		npf_nat_sysinit(void);
    324 void		npf_nat_sysfini(void);
    325 npf_natpolicy_t *npf_nat_newpolicy(prop_dictionary_t, npf_ruleset_t *);
    326 void		npf_nat_freepolicy(npf_natpolicy_t *);
    327 bool		npf_nat_matchpolicy(npf_natpolicy_t *, npf_natpolicy_t *);
    328 bool		npf_nat_sharepm(npf_natpolicy_t *, npf_natpolicy_t *);
    329 void		npf_nat_freealg(npf_natpolicy_t *, npf_alg_t *);
    330 
    331 int		npf_do_nat(npf_cache_t *, npf_session_t *, nbuf_t *, const int);
    332 int		npf_nat_translate(npf_cache_t *, nbuf_t *, npf_nat_t *,
    333 		    const bool, const int);
    334 void		npf_nat_expire(npf_nat_t *);
    335 void		npf_nat_getorig(npf_nat_t *, npf_addr_t **, in_port_t *);
    336 void		npf_nat_gettrans(npf_nat_t *, npf_addr_t **, in_port_t *);
    337 void		npf_nat_setalg(npf_nat_t *, npf_alg_t *, uintptr_t);
    338 
    339 int		npf_nat_save(prop_dictionary_t, prop_array_t, npf_nat_t *);
    340 npf_nat_t *	npf_nat_restore(prop_dictionary_t, npf_session_t *);
    341 
    342 /* ALG interface. */
    343 void		npf_alg_sysinit(void);
    344 void		npf_alg_sysfini(void);
    345 npf_alg_t *	npf_alg_register(const char *, npf_alg_func_t, npf_alg_func_t,
    346 		    npf_alg_sfunc_t);
    347 int		npf_alg_unregister(npf_alg_t *);
    348 npf_alg_t *	npf_alg_construct(const char *);
    349 bool		npf_alg_match(npf_cache_t *, nbuf_t *, npf_nat_t *, int);
    350 void		npf_alg_exec(npf_cache_t *, nbuf_t *, npf_nat_t *, int);
    351 npf_session_t *	npf_alg_session(npf_cache_t *, nbuf_t *, int);
    352 
    353 /* Debugging routines. */
    354 void		npf_addr_dump(const npf_addr_t *);
    355 void		npf_rulenc_dump(const npf_rule_t *);
    356 void		npf_sessions_dump(void);
    357 void		npf_state_dump(const npf_state_t *);
    358 void		npf_nat_dump(const npf_nat_t *);
    359 void		npf_state_setsampler(void (*)(npf_state_t *, bool));
    360 
    361 #endif	/* _NPF_IMPL_H_ */
    362