npf_impl.h revision 1.40 1 /* $NetBSD: npf_impl.h,v 1.40 2013/11/16 01:18:58 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
63 #ifdef _NPF_DEBUG
64 #define NPF_PRINTF(x) printf x
65 #else
66 #define NPF_PRINTF(x)
67 #endif
68
69 /*
70 * STRUCTURE DECLARATIONS.
71 */
72
73 struct npf_ruleset;
74 struct npf_rule;
75 struct npf_rprocset;
76 struct npf_nat;
77 struct npf_session;
78
79 typedef struct npf_ruleset npf_ruleset_t;
80 typedef struct npf_rule npf_rule_t;
81 typedef struct npf_nat npf_nat_t;
82 typedef struct npf_rprocset npf_rprocset_t;
83 typedef struct npf_alg npf_alg_t;
84 typedef struct npf_natpolicy npf_natpolicy_t;
85 typedef struct npf_session npf_session_t;
86
87 struct npf_sehash;
88 struct npf_table;
89 struct npf_tableset;
90
91 typedef struct npf_sehash npf_sehash_t;
92 typedef struct npf_table npf_table_t;
93 typedef struct npf_tableset npf_tableset_t;
94
95 /*
96 * DEFINITIONS.
97 */
98
99 typedef bool (*npf_alg_func_t)(npf_cache_t *, nbuf_t *, npf_nat_t *, int);
100 typedef npf_session_t *(*npf_alg_sfunc_t)(npf_cache_t *, nbuf_t *, int);
101 typedef void (*npf_workfunc_t)(void);
102
103 /* Some artificial limits. */
104 #define NPF_MAX_RULES (1024 * 1024)
105 #define NPF_MAX_TABLES 128
106 #define NPF_MAX_RPROCS 128
107 #define NPF_MAX_IFMAP 64
108
109 /*
110 * SESSION STATE STRUCTURES
111 */
112
113 #define NPF_FLOW_FORW 0
114 #define NPF_FLOW_BACK 1
115
116 typedef struct {
117 uint32_t nst_end;
118 uint32_t nst_maxend;
119 uint32_t nst_maxwin;
120 int nst_wscale;
121 } npf_tcpstate_t;
122
123 typedef struct {
124 kmutex_t nst_lock;
125 u_int nst_state;
126 npf_tcpstate_t nst_tcpst[2];
127 } npf_state_t;
128
129 /*
130 * INTERFACES.
131 */
132
133 /* NPF config, statistics, etc. */
134 void npf_config_init(void);
135 void npf_config_fini(void);
136
137 void npf_config_enter(void);
138 void npf_config_exit(void);
139 void npf_config_sync(void);
140 bool npf_config_locked_p(void);
141 int npf_config_read_enter(void);
142 void npf_config_read_exit(int);
143
144 void npf_config_reload(prop_dictionary_t, npf_ruleset_t *,
145 npf_tableset_t *, npf_ruleset_t *, npf_rprocset_t *, bool);
146 npf_ruleset_t * npf_config_ruleset(void);
147 npf_ruleset_t * npf_config_natset(void);
148 npf_tableset_t *npf_config_tableset(void);
149 prop_dictionary_t npf_config_dict(void);
150 bool npf_default_pass(void);
151
152 int npf_worker_sysinit(void);
153 void npf_worker_sysfini(void);
154 void npf_worker_signal(void);
155 void npf_worker_register(npf_workfunc_t);
156 void npf_worker_unregister(npf_workfunc_t);
157
158 void npflogattach(int);
159 void npflogdetach(void);
160 int npfctl_switch(void *);
161 int npfctl_reload(u_long, void *);
162 int npfctl_getconf(u_long, void *);
163 int npfctl_sessions_save(u_long, void *);
164 int npfctl_sessions_load(u_long, void *);
165 int npfctl_rule(u_long, void *);
166 int npfctl_table(void *);
167
168 void npf_stats_inc(npf_stats_t);
169 void npf_stats_dec(npf_stats_t);
170
171 u_int npf_ifmap_register(const char *);
172 void npf_ifmap_flush(void);
173 void npf_ifmap_attach(ifnet_t *);
174 void npf_ifmap_detach(ifnet_t *);
175 u_int npf_ifmap_id(const ifnet_t *);
176
177 /* Packet filter hooks. */
178 int npf_pfil_register(bool);
179 void npf_pfil_unregister(bool);
180 bool npf_pfil_registered_p(void);
181 int npf_packet_handler(void *, struct mbuf **, ifnet_t *, int);
182
183 /* Protocol helpers. */
184 int npf_cache_all(npf_cache_t *, nbuf_t *);
185 void npf_recache(npf_cache_t *, nbuf_t *);
186
187 bool npf_rwrip(const npf_cache_t *, int, const npf_addr_t *);
188 bool npf_rwrport(const npf_cache_t *, int, const in_port_t);
189 bool npf_rwrcksum(const npf_cache_t *, const int,
190 const npf_addr_t *, const in_port_t);
191
192 uint16_t npf_fixup16_cksum(uint16_t, uint16_t, uint16_t);
193 uint16_t npf_fixup32_cksum(uint16_t, uint32_t, uint32_t);
194 uint16_t npf_addr_cksum(uint16_t, int, const npf_addr_t *,
195 const npf_addr_t *);
196 uint32_t npf_addr_sum(const int, const npf_addr_t *, const npf_addr_t *);
197 int npf_addr_cmp(const npf_addr_t *, const npf_netmask_t,
198 const npf_addr_t *, const npf_netmask_t, const int);
199 void npf_addr_mask(const npf_addr_t *, const npf_netmask_t,
200 const int, npf_addr_t *);
201
202 int npf_tcpsaw(const npf_cache_t *, tcp_seq *, tcp_seq *,
203 uint32_t *);
204 bool npf_fetch_tcpopts(npf_cache_t *, nbuf_t *, uint16_t *, int *);
205 bool npf_return_block(npf_cache_t *, nbuf_t *, const int);
206
207 /* BPF interface. */
208 void npf_bpf_sysinit(void);
209 void npf_bpf_sysfini(void);
210 int npf_bpf_filter(npf_cache_t *, nbuf_t *,
211 const void *, bpfjit_func_t);
212 void * npf_bpf_compile(void *, size_t);
213 bool npf_bpf_validate(const void *, size_t);
214
215 /* Tableset interface. */
216 void npf_tableset_sysinit(void);
217 void npf_tableset_sysfini(void);
218
219 extern const pt_tree_ops_t npf_table_ptree_ops;
220
221 npf_tableset_t *npf_tableset_create(u_int);
222 void npf_tableset_destroy(npf_tableset_t *);
223 int npf_tableset_insert(npf_tableset_t *, npf_table_t *);
224 npf_table_t * npf_tableset_getbyname(npf_tableset_t *, const char *);
225 npf_table_t * npf_tableset_getbyid(npf_tableset_t *, u_int);
226 void npf_tableset_reload(npf_tableset_t *, npf_tableset_t *);
227
228 npf_table_t * npf_table_create(const char *, u_int, int, size_t);
229 void npf_table_destroy(npf_table_t *);
230
231 int npf_table_check(npf_tableset_t *, const char *, u_int, int);
232 int npf_table_insert(npf_table_t *, const int,
233 const npf_addr_t *, const npf_netmask_t);
234 int npf_table_remove(npf_table_t *, const int,
235 const npf_addr_t *, const npf_netmask_t);
236 int npf_table_lookup(npf_table_t *, const int, const npf_addr_t *);
237 int npf_table_list(npf_table_t *, void *, size_t);
238 int npf_table_flush(npf_table_t *);
239
240 /* Ruleset interface. */
241 npf_ruleset_t * npf_ruleset_create(size_t);
242 void npf_ruleset_destroy(npf_ruleset_t *);
243 void npf_ruleset_insert(npf_ruleset_t *, npf_rule_t *);
244 void npf_ruleset_reload(npf_ruleset_t *, npf_ruleset_t *);
245 void npf_ruleset_natreload(npf_ruleset_t *, npf_ruleset_t *);
246 npf_rule_t * npf_ruleset_matchnat(npf_ruleset_t *, npf_natpolicy_t *);
247 npf_rule_t * npf_ruleset_sharepm(npf_ruleset_t *, npf_natpolicy_t *);
248 void npf_ruleset_freealg(npf_ruleset_t *, npf_alg_t *);
249
250 int npf_ruleset_add(npf_ruleset_t *, const char *, npf_rule_t *);
251 int npf_ruleset_remove(npf_ruleset_t *, const char *, uint64_t);
252 int npf_ruleset_remkey(npf_ruleset_t *, const char *,
253 const void *, size_t);
254 prop_dictionary_t npf_ruleset_list(npf_ruleset_t *, const char *);
255 int npf_ruleset_flush(npf_ruleset_t *, const char *);
256 void npf_ruleset_gc(npf_ruleset_t *);
257
258 npf_rule_t * npf_ruleset_inspect(npf_cache_t *, nbuf_t *,
259 const npf_ruleset_t *, const int, const int);
260 int npf_rule_conclude(const npf_rule_t *, int *);
261
262 /* Rule interface. */
263 npf_rule_t * npf_rule_alloc(prop_dictionary_t);
264 void npf_rule_setcode(npf_rule_t *, int, void *, size_t);
265 void npf_rule_setrproc(npf_rule_t *, npf_rproc_t *);
266 void npf_rule_free(npf_rule_t *);
267 uint64_t npf_rule_getid(const npf_rule_t *);
268 npf_natpolicy_t *npf_rule_getnat(const npf_rule_t *);
269 void npf_rule_setnat(npf_rule_t *, npf_natpolicy_t *);
270 npf_rproc_t * npf_rule_getrproc(npf_rule_t *);
271
272 void npf_ext_sysinit(void);
273 void npf_ext_sysfini(void);
274 int npf_ext_construct(const char *,
275 npf_rproc_t *, prop_dictionary_t);
276
277 npf_rprocset_t *npf_rprocset_create(void);
278 void npf_rprocset_destroy(npf_rprocset_t *);
279 npf_rproc_t * npf_rprocset_lookup(npf_rprocset_t *, const char *);
280 void npf_rprocset_insert(npf_rprocset_t *, npf_rproc_t *);
281
282 npf_rproc_t * npf_rproc_create(prop_dictionary_t);
283 void npf_rproc_acquire(npf_rproc_t *);
284 void npf_rproc_release(npf_rproc_t *);
285 void npf_rproc_run(npf_cache_t *, nbuf_t *, npf_rproc_t *, int *);
286
287 /* Session handling interface. */
288 void npf_session_sysinit(void);
289 void npf_session_sysfini(void);
290 void npf_session_tracking(bool);
291
292 npf_sehash_t * sess_htable_create(void);
293 void sess_htable_destroy(npf_sehash_t *);
294
295 npf_session_t * npf_session_lookup(const npf_cache_t *, const nbuf_t *,
296 const int, bool *);
297 npf_session_t * npf_session_inspect(npf_cache_t *, nbuf_t *, const int, int *);
298 npf_session_t * npf_session_establish(npf_cache_t *, nbuf_t *, const int);
299 void npf_session_release(npf_session_t *);
300 void npf_session_expire(npf_session_t *);
301 bool npf_session_pass(const npf_session_t *, npf_rproc_t **);
302 void npf_session_setpass(npf_session_t *, npf_rproc_t *);
303 int npf_session_setnat(npf_session_t *, npf_nat_t *, u_int);
304 npf_nat_t * npf_session_retnat(npf_session_t *, const int, bool *);
305
306 void npf_session_load(npf_sehash_t *);
307 int npf_session_save(prop_array_t, prop_array_t);
308 int npf_session_restore(npf_sehash_t *, prop_dictionary_t);
309
310 /* State handling. */
311 bool npf_state_init(npf_cache_t *, nbuf_t *, npf_state_t *);
312 bool npf_state_inspect(npf_cache_t *, nbuf_t *, npf_state_t *,
313 const bool);
314 int npf_state_etime(const npf_state_t *, const int);
315 void npf_state_destroy(npf_state_t *);
316
317 bool npf_state_tcp(npf_cache_t *, nbuf_t *, npf_state_t *, int);
318 int npf_state_tcp_timeout(const npf_state_t *);
319
320 /* NAT. */
321 void npf_nat_sysinit(void);
322 void npf_nat_sysfini(void);
323 npf_natpolicy_t *npf_nat_newpolicy(prop_dictionary_t, npf_ruleset_t *);
324 void npf_nat_freepolicy(npf_natpolicy_t *);
325 bool npf_nat_matchpolicy(npf_natpolicy_t *, npf_natpolicy_t *);
326 bool npf_nat_sharepm(npf_natpolicy_t *, npf_natpolicy_t *);
327 void npf_nat_freealg(npf_natpolicy_t *, npf_alg_t *);
328
329 int npf_do_nat(npf_cache_t *, npf_session_t *, nbuf_t *, const int);
330 int npf_nat_translate(npf_cache_t *, nbuf_t *, npf_nat_t *,
331 const bool, const int);
332 void npf_nat_expire(npf_nat_t *);
333 void npf_nat_getorig(npf_nat_t *, npf_addr_t **, in_port_t *);
334 void npf_nat_gettrans(npf_nat_t *, npf_addr_t **, in_port_t *);
335 void npf_nat_setalg(npf_nat_t *, npf_alg_t *, uintptr_t);
336
337 int npf_nat_save(prop_dictionary_t, prop_array_t, npf_nat_t *);
338 npf_nat_t * npf_nat_restore(prop_dictionary_t, npf_session_t *);
339
340 /* ALG interface. */
341 void npf_alg_sysinit(void);
342 void npf_alg_sysfini(void);
343 npf_alg_t * npf_alg_register(const char *, npf_alg_func_t, npf_alg_func_t,
344 npf_alg_sfunc_t);
345 int npf_alg_unregister(npf_alg_t *);
346 npf_alg_t * npf_alg_construct(const char *);
347 bool npf_alg_match(npf_cache_t *, nbuf_t *, npf_nat_t *, int);
348 void npf_alg_exec(npf_cache_t *, nbuf_t *, npf_nat_t *, int);
349 npf_session_t * npf_alg_session(npf_cache_t *, nbuf_t *, int);
350
351 /* Debugging routines. */
352 void npf_addr_dump(const npf_addr_t *);
353 void npf_sessions_dump(void);
354 void npf_state_dump(const npf_state_t *);
355 void npf_nat_dump(const npf_nat_t *);
356 void npf_state_setsampler(void (*)(npf_state_t *, bool));
357
358 #endif /* _NPF_IMPL_H_ */
359