pf_if.c revision 1.5 1 1.5 yamt /* $NetBSD: pf_if.c,v 1.5 2004/07/26 13:46:43 yamt Exp $ */
2 1.1 itojun /* $OpenBSD: pf_if.c,v 1.11 2004/03/15 11:38:23 cedric Exp $ */
3 1.1 itojun
4 1.1 itojun /*
5 1.1 itojun * Copyright (c) 2001 Daniel Hartmeier
6 1.1 itojun * Copyright (c) 2003 Cedric Berger
7 1.1 itojun * All rights reserved.
8 1.1 itojun *
9 1.1 itojun * Redistribution and use in source and binary forms, with or without
10 1.1 itojun * modification, are permitted provided that the following conditions
11 1.1 itojun * are met:
12 1.1 itojun *
13 1.1 itojun * - Redistributions of source code must retain the above copyright
14 1.1 itojun * notice, this list of conditions and the following disclaimer.
15 1.1 itojun * - Redistributions in binary form must reproduce the above
16 1.1 itojun * copyright notice, this list of conditions and the following
17 1.1 itojun * disclaimer in the documentation and/or other materials provided
18 1.1 itojun * with the distribution.
19 1.1 itojun *
20 1.1 itojun * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21 1.1 itojun * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22 1.1 itojun * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
23 1.1 itojun * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
24 1.1 itojun * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
25 1.1 itojun * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
26 1.1 itojun * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
27 1.1 itojun * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
28 1.1 itojun * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 1.1 itojun * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
30 1.1 itojun * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
31 1.1 itojun * POSSIBILITY OF SUCH DAMAGE.
32 1.1 itojun */
33 1.1 itojun
34 1.2 itojun #ifdef _KERNEL_OPT
35 1.2 itojun #include "opt_inet.h"
36 1.2 itojun #endif
37 1.2 itojun
38 1.1 itojun #include <sys/param.h>
39 1.1 itojun #include <sys/systm.h>
40 1.1 itojun #include <sys/mbuf.h>
41 1.1 itojun #include <sys/filio.h>
42 1.1 itojun #include <sys/socket.h>
43 1.1 itojun #include <sys/socketvar.h>
44 1.1 itojun #include <sys/kernel.h>
45 1.1 itojun #include <sys/device.h>
46 1.1 itojun #include <sys/time.h>
47 1.1 itojun
48 1.1 itojun #include <net/if.h>
49 1.1 itojun #include <net/if_types.h>
50 1.1 itojun
51 1.1 itojun #include <netinet/in.h>
52 1.1 itojun #include <netinet/in_var.h>
53 1.1 itojun #include <netinet/in_systm.h>
54 1.1 itojun #include <netinet/ip.h>
55 1.1 itojun #include <netinet/ip_var.h>
56 1.1 itojun
57 1.1 itojun #include <net/pfvar.h>
58 1.1 itojun
59 1.1 itojun #ifdef INET6
60 1.1 itojun #include <netinet/ip6.h>
61 1.1 itojun #endif /* INET6 */
62 1.1 itojun
63 1.1 itojun #define ACCEPT_FLAGS(oklist) \
64 1.1 itojun do { \
65 1.1 itojun if ((flags & ~(oklist)) & \
66 1.1 itojun PFI_FLAG_ALLMASK) \
67 1.1 itojun return (EINVAL); \
68 1.1 itojun } while (0)
69 1.1 itojun
70 1.1 itojun #define senderr(e) do { rv = (e); goto _bad; } while (0)
71 1.1 itojun
72 1.1 itojun struct pfi_kif **pfi_index2kif;
73 1.1 itojun struct pfi_kif *pfi_self;
74 1.1 itojun int pfi_indexlim;
75 1.1 itojun struct pfi_ifhead pfi_ifs;
76 1.1 itojun struct pfi_statehead pfi_statehead;
77 1.1 itojun int pfi_ifcnt;
78 1.1 itojun struct pool pfi_addr_pl;
79 1.1 itojun long pfi_update = 1;
80 1.1 itojun struct pfr_addr *pfi_buffer;
81 1.1 itojun int pfi_buffer_cnt;
82 1.1 itojun int pfi_buffer_max;
83 1.1 itojun char pfi_reserved_anchor[PF_ANCHOR_NAME_SIZE] =
84 1.1 itojun PF_RESERVED_ANCHOR;
85 1.1 itojun char pfi_interface_ruleset[PF_RULESET_NAME_SIZE] =
86 1.1 itojun PF_INTERFACE_RULESET;
87 1.1 itojun
88 1.1 itojun void pfi_dynaddr_update(void *);
89 1.1 itojun void pfi_kifaddr_update(void *);
90 1.5 yamt #ifdef __NetBSD__
91 1.5 yamt void pfi_kifaddr_update_if(struct ifnet *);
92 1.5 yamt #endif
93 1.1 itojun void pfi_table_update(struct pfr_ktable *, struct pfi_kif *,
94 1.1 itojun int, int);
95 1.1 itojun void pfi_instance_add(struct ifnet *, int, int);
96 1.1 itojun void pfi_address_add(struct sockaddr *, int, int);
97 1.1 itojun int pfi_if_compare(struct pfi_kif *, struct pfi_kif *);
98 1.1 itojun struct pfi_kif *pfi_if_create(const char *, struct pfi_kif *, int);
99 1.1 itojun void pfi_copy_group(char *, const char *, int);
100 1.1 itojun void pfi_dynamic_drivers(void);
101 1.1 itojun void pfi_newgroup(const char *, int);
102 1.1 itojun int pfi_skip_if(const char *, struct pfi_kif *, int);
103 1.1 itojun int pfi_unmask(void *);
104 1.1 itojun void pfi_dohooks(struct pfi_kif *);
105 1.1 itojun
106 1.1 itojun RB_PROTOTYPE(pfi_ifhead, pfi_kif, pfik_tree, pfi_if_compare);
107 1.1 itojun RB_GENERATE(pfi_ifhead, pfi_kif, pfik_tree, pfi_if_compare);
108 1.1 itojun
109 1.1 itojun #define PFI_DYNAMIC_BUSES { "pcmcia", "cardbus", "uhub" }
110 1.1 itojun #define PFI_BUFFER_MAX 0x10000
111 1.1 itojun #define PFI_MTYPE M_IFADDR
112 1.1 itojun
113 1.2 itojun #ifdef __NetBSD__
114 1.2 itojun static void *hook_establish(struct hook_desc_head *, int, void (*)(void *),
115 1.2 itojun void *);
116 1.2 itojun static void hook_disestablish(struct hook_desc_head *, void *);
117 1.5 yamt static void dohooks(struct hook_desc_head *, int);
118 1.2 itojun
119 1.2 itojun #define HOOK_REMOVE 0x01
120 1.2 itojun #define HOOK_FREE 0x02
121 1.2 itojun #endif
122 1.2 itojun
123 1.1 itojun void
124 1.1 itojun pfi_initialize(void)
125 1.1 itojun {
126 1.1 itojun if (pfi_self != NULL) /* already initialized */
127 1.1 itojun return;
128 1.1 itojun
129 1.1 itojun TAILQ_INIT(&pfi_statehead);
130 1.1 itojun pool_init(&pfi_addr_pl, sizeof(struct pfi_dynaddr), 0, 0, 0,
131 1.1 itojun "pfiaddrpl", &pool_allocator_nointr);
132 1.1 itojun pfi_buffer_max = 64;
133 1.1 itojun pfi_buffer = malloc(pfi_buffer_max * sizeof(*pfi_buffer),
134 1.1 itojun PFI_MTYPE, M_WAITOK);
135 1.1 itojun pfi_self = pfi_if_create("self", NULL, PFI_IFLAG_GROUP);
136 1.1 itojun pfi_dynamic_drivers();
137 1.1 itojun }
138 1.1 itojun
139 1.3 itojun #ifdef _LKM
140 1.3 itojun void
141 1.3 itojun pfi_destroy(void)
142 1.3 itojun {
143 1.3 itojun struct pfi_kif *p;
144 1.3 itojun
145 1.3 itojun while ((p = TAILQ_FIRST(&pfi_statehead)) != NULL) {
146 1.3 itojun TAILQ_REMOVE(&pfi_statehead, p, pfik_w_states);
147 1.3 itojun free(p, PFI_MTYPE);
148 1.3 itojun }
149 1.3 itojun pool_destroy(&pfi_addr_pl);
150 1.3 itojun
151 1.3 itojun pfi_self = NULL;
152 1.3 itojun }
153 1.3 itojun #endif
154 1.3 itojun
155 1.1 itojun void
156 1.1 itojun pfi_attach_clone(struct if_clone *ifc)
157 1.1 itojun {
158 1.1 itojun pfi_initialize();
159 1.1 itojun pfi_newgroup(ifc->ifc_name, PFI_IFLAG_CLONABLE);
160 1.1 itojun }
161 1.1 itojun
162 1.1 itojun void
163 1.1 itojun pfi_attach_ifnet(struct ifnet *ifp)
164 1.1 itojun {
165 1.1 itojun struct pfi_kif *p, *q, key;
166 1.1 itojun int s;
167 1.1 itojun
168 1.1 itojun pfi_initialize();
169 1.1 itojun s = splsoftnet();
170 1.1 itojun pfi_update++;
171 1.1 itojun if (ifp->if_index >= pfi_indexlim) {
172 1.1 itojun /*
173 1.1 itojun * grow pfi_index2kif, similar to ifindex2ifnet code in if.c
174 1.1 itojun */
175 1.1 itojun size_t m, n, oldlim;
176 1.1 itojun struct pfi_kif **mp, **np;
177 1.1 itojun
178 1.1 itojun oldlim = pfi_indexlim;
179 1.1 itojun if (pfi_indexlim == 0)
180 1.1 itojun pfi_indexlim = 64;
181 1.1 itojun while (ifp->if_index >= pfi_indexlim)
182 1.1 itojun pfi_indexlim <<= 1;
183 1.1 itojun
184 1.1 itojun m = oldlim * sizeof(struct pfi_kif *);
185 1.1 itojun mp = pfi_index2kif;
186 1.1 itojun n = pfi_indexlim * sizeof(struct pfi_kif *);
187 1.1 itojun np = malloc(n, PFI_MTYPE, M_DONTWAIT);
188 1.1 itojun if (np == NULL)
189 1.1 itojun panic("pfi_attach_ifnet: "
190 1.1 itojun "cannot allocate translation table");
191 1.1 itojun bzero(np, n);
192 1.1 itojun if (mp != NULL)
193 1.1 itojun bcopy(mp, np, m);
194 1.1 itojun pfi_index2kif = np;
195 1.1 itojun if (mp != NULL)
196 1.1 itojun free(mp, PFI_MTYPE);
197 1.1 itojun }
198 1.1 itojun
199 1.1 itojun strlcpy(key.pfik_name, ifp->if_xname, sizeof(key.pfik_name));
200 1.1 itojun p = RB_FIND(pfi_ifhead, &pfi_ifs, &key);
201 1.1 itojun if (p == NULL) {
202 1.1 itojun /* add group */
203 1.1 itojun pfi_copy_group(key.pfik_name, ifp->if_xname,
204 1.1 itojun sizeof(key.pfik_name));
205 1.1 itojun q = RB_FIND(pfi_ifhead, &pfi_ifs, &key);
206 1.1 itojun if (q == NULL)
207 1.1 itojun q = pfi_if_create(key.pfik_name, pfi_self, PFI_IFLAG_GROUP);
208 1.1 itojun if (q == NULL)
209 1.1 itojun panic("pfi_attach_ifnet: "
210 1.1 itojun "cannot allocate '%s' group", key.pfik_name);
211 1.1 itojun
212 1.1 itojun /* add interface */
213 1.1 itojun p = pfi_if_create(ifp->if_xname, q, PFI_IFLAG_INSTANCE);
214 1.1 itojun if (p == NULL)
215 1.1 itojun panic("pfi_attach_ifnet: "
216 1.1 itojun "cannot allocate '%s' interface", ifp->if_xname);
217 1.1 itojun } else
218 1.1 itojun q = p->pfik_parent;
219 1.1 itojun p->pfik_ifp = ifp;
220 1.1 itojun p->pfik_flags |= PFI_IFLAG_ATTACHED;
221 1.2 itojun #ifdef __OpenBSD__
222 1.1 itojun p->pfik_ah_cookie =
223 1.1 itojun hook_establish(ifp->if_addrhooks, 1, pfi_kifaddr_update, p);
224 1.2 itojun #else
225 1.2 itojun p->pfik_ah_cookie =
226 1.2 itojun hook_establish(p->pfik_ifaddrhooks, 1, pfi_kifaddr_update, p);
227 1.2 itojun #endif
228 1.1 itojun pfi_index2kif[ifp->if_index] = p;
229 1.1 itojun pfi_dohooks(p);
230 1.1 itojun splx(s);
231 1.1 itojun }
232 1.1 itojun
233 1.1 itojun void
234 1.1 itojun pfi_detach_ifnet(struct ifnet *ifp)
235 1.1 itojun {
236 1.1 itojun struct pfi_kif *p, *q, key;
237 1.1 itojun int s;
238 1.1 itojun
239 1.1 itojun strlcpy(key.pfik_name, ifp->if_xname, sizeof(key.pfik_name));
240 1.1 itojun
241 1.1 itojun s = splsoftnet();
242 1.1 itojun pfi_update++;
243 1.1 itojun p = RB_FIND(pfi_ifhead, &pfi_ifs, &key);
244 1.1 itojun if (p == NULL) {
245 1.1 itojun printf("pfi_detach_ifnet: cannot find %s", ifp->if_xname);
246 1.1 itojun splx(s);
247 1.1 itojun return;
248 1.1 itojun }
249 1.2 itojun #ifdef __OpenBSD__
250 1.1 itojun hook_disestablish(p->pfik_ifp->if_addrhooks, p->pfik_ah_cookie);
251 1.2 itojun #else
252 1.2 itojun hook_disestablish(p->pfik_ifaddrhooks, p->pfik_ah_cookie);
253 1.2 itojun #endif
254 1.1 itojun q = p->pfik_parent;
255 1.1 itojun p->pfik_ifp = NULL;
256 1.1 itojun p->pfik_flags &= ~PFI_IFLAG_ATTACHED;
257 1.1 itojun pfi_index2kif[ifp->if_index] = NULL;
258 1.1 itojun pfi_dohooks(p);
259 1.1 itojun pfi_maybe_destroy(p);
260 1.1 itojun splx(s);
261 1.1 itojun }
262 1.1 itojun
263 1.1 itojun struct pfi_kif *
264 1.1 itojun pfi_lookup_create(const char *name)
265 1.1 itojun {
266 1.1 itojun struct pfi_kif *p, *q, key;
267 1.1 itojun int s;
268 1.1 itojun
269 1.1 itojun s = splsoftnet();
270 1.1 itojun p = pfi_lookup_if(name);
271 1.1 itojun if (p == NULL) {
272 1.1 itojun pfi_copy_group(key.pfik_name, name, sizeof(key.pfik_name));
273 1.1 itojun q = pfi_lookup_if(key.pfik_name);
274 1.1 itojun if (q != NULL)
275 1.1 itojun p = pfi_if_create(name, q, PFI_IFLAG_INSTANCE);
276 1.1 itojun }
277 1.1 itojun splx(s);
278 1.1 itojun return (p);
279 1.1 itojun }
280 1.1 itojun
281 1.1 itojun struct pfi_kif *
282 1.1 itojun pfi_attach_rule(const char *name)
283 1.1 itojun {
284 1.1 itojun struct pfi_kif *p;
285 1.1 itojun
286 1.1 itojun p = pfi_lookup_create(name);
287 1.1 itojun if (p != NULL)
288 1.1 itojun p->pfik_rules++;
289 1.1 itojun return (p);
290 1.1 itojun }
291 1.1 itojun
292 1.1 itojun void
293 1.1 itojun pfi_detach_rule(struct pfi_kif *p)
294 1.1 itojun {
295 1.1 itojun if (p == NULL)
296 1.1 itojun return;
297 1.1 itojun if (p->pfik_rules > 0)
298 1.1 itojun p->pfik_rules--;
299 1.1 itojun else
300 1.1 itojun printf("pfi_detach_rule: reference count at 0\n");
301 1.1 itojun pfi_maybe_destroy(p);
302 1.1 itojun }
303 1.1 itojun
304 1.1 itojun void
305 1.1 itojun pfi_attach_state(struct pfi_kif *p)
306 1.1 itojun {
307 1.1 itojun if (!p->pfik_states++)
308 1.1 itojun TAILQ_INSERT_TAIL(&pfi_statehead, p, pfik_w_states);
309 1.1 itojun }
310 1.1 itojun
311 1.1 itojun void
312 1.1 itojun pfi_detach_state(struct pfi_kif *p)
313 1.1 itojun {
314 1.1 itojun if (p == NULL)
315 1.1 itojun return;
316 1.1 itojun if (p->pfik_states <= 0) {
317 1.1 itojun printf("pfi_detach_state: reference count <= 0\n");
318 1.1 itojun return;
319 1.1 itojun }
320 1.1 itojun if (!--p->pfik_states)
321 1.1 itojun TAILQ_REMOVE(&pfi_statehead, p, pfik_w_states);
322 1.1 itojun pfi_maybe_destroy(p);
323 1.1 itojun }
324 1.1 itojun
325 1.1 itojun int
326 1.1 itojun pfi_dynaddr_setup(struct pf_addr_wrap *aw, sa_family_t af)
327 1.1 itojun {
328 1.1 itojun struct pfi_dynaddr *dyn;
329 1.1 itojun char tblname[PF_TABLE_NAME_SIZE];
330 1.1 itojun struct pf_ruleset *ruleset = NULL;
331 1.1 itojun int s, rv = 0;
332 1.1 itojun
333 1.1 itojun if (aw->type != PF_ADDR_DYNIFTL)
334 1.1 itojun return (0);
335 1.1 itojun dyn = pool_get(&pfi_addr_pl, PR_NOWAIT);
336 1.1 itojun if (dyn == NULL)
337 1.1 itojun return (1);
338 1.1 itojun bzero(dyn, sizeof(*dyn));
339 1.1 itojun
340 1.1 itojun s = splsoftnet();
341 1.1 itojun dyn->pfid_kif = pfi_attach_rule(aw->v.ifname);
342 1.1 itojun if (dyn->pfid_kif == NULL)
343 1.1 itojun senderr(1);
344 1.1 itojun
345 1.1 itojun dyn->pfid_net = pfi_unmask(&aw->v.a.mask);
346 1.1 itojun if (af == AF_INET && dyn->pfid_net == 32)
347 1.1 itojun dyn->pfid_net = 128;
348 1.1 itojun strlcpy(tblname, aw->v.ifname, sizeof(tblname));
349 1.1 itojun if (aw->iflags & PFI_AFLAG_NETWORK)
350 1.1 itojun strlcat(tblname, ":network", sizeof(tblname));
351 1.1 itojun if (aw->iflags & PFI_AFLAG_BROADCAST)
352 1.1 itojun strlcat(tblname, ":broadcast", sizeof(tblname));
353 1.1 itojun if (aw->iflags & PFI_AFLAG_PEER)
354 1.1 itojun strlcat(tblname, ":peer", sizeof(tblname));
355 1.1 itojun if (aw->iflags & PFI_AFLAG_NOALIAS)
356 1.1 itojun strlcat(tblname, ":0", sizeof(tblname));
357 1.1 itojun if (dyn->pfid_net != 128)
358 1.1 itojun snprintf(tblname + strlen(tblname),
359 1.1 itojun sizeof(tblname) - strlen(tblname), "/%d", dyn->pfid_net);
360 1.1 itojun ruleset = pf_find_or_create_ruleset(pfi_reserved_anchor,
361 1.1 itojun pfi_interface_ruleset);
362 1.1 itojun if (ruleset == NULL)
363 1.1 itojun senderr(1);
364 1.1 itojun
365 1.1 itojun dyn->pfid_kt = pfr_attach_table(ruleset, tblname);
366 1.1 itojun if (dyn->pfid_kt == NULL)
367 1.1 itojun senderr(1);
368 1.1 itojun
369 1.1 itojun dyn->pfid_kt->pfrkt_flags |= PFR_TFLAG_ACTIVE;
370 1.1 itojun dyn->pfid_iflags = aw->iflags;
371 1.1 itojun dyn->pfid_af = af;
372 1.1 itojun dyn->pfid_hook_cookie = hook_establish(dyn->pfid_kif->pfik_ah_head, 1,
373 1.1 itojun pfi_dynaddr_update, dyn);
374 1.1 itojun if (dyn->pfid_hook_cookie == NULL)
375 1.1 itojun senderr(1);
376 1.1 itojun
377 1.1 itojun aw->p.dyn = dyn;
378 1.1 itojun pfi_dynaddr_update(aw->p.dyn);
379 1.1 itojun splx(s);
380 1.1 itojun return (0);
381 1.1 itojun
382 1.1 itojun _bad:
383 1.1 itojun if (dyn->pfid_kt != NULL)
384 1.1 itojun pfr_detach_table(dyn->pfid_kt);
385 1.1 itojun if (ruleset != NULL)
386 1.1 itojun pf_remove_if_empty_ruleset(ruleset);
387 1.1 itojun if (dyn->pfid_kif != NULL)
388 1.1 itojun pfi_detach_rule(dyn->pfid_kif);
389 1.1 itojun pool_put(&pfi_addr_pl, dyn);
390 1.1 itojun splx(s);
391 1.1 itojun return (rv);
392 1.1 itojun }
393 1.1 itojun
394 1.1 itojun void
395 1.1 itojun pfi_dynaddr_update(void *p)
396 1.1 itojun {
397 1.1 itojun struct pfi_dynaddr *dyn = (struct pfi_dynaddr *)p;
398 1.1 itojun struct pfi_kif *kif = dyn->pfid_kif;
399 1.1 itojun struct pfr_ktable *kt = dyn->pfid_kt;
400 1.1 itojun
401 1.1 itojun if (dyn == NULL || kif == NULL || kt == NULL)
402 1.1 itojun panic("pfi_dynaddr_update");
403 1.1 itojun if (kt->pfrkt_larg != pfi_update) {
404 1.1 itojun /* this table needs to be brought up-to-date */
405 1.1 itojun pfi_table_update(kt, kif, dyn->pfid_net, dyn->pfid_iflags);
406 1.1 itojun kt->pfrkt_larg = pfi_update;
407 1.1 itojun }
408 1.1 itojun pfr_dynaddr_update(kt, dyn);
409 1.1 itojun }
410 1.1 itojun
411 1.1 itojun void
412 1.1 itojun pfi_table_update(struct pfr_ktable *kt, struct pfi_kif *kif, int net, int flags)
413 1.1 itojun {
414 1.1 itojun int e, size2 = 0;
415 1.1 itojun struct pfi_kif *p;
416 1.1 itojun struct pfr_table t;
417 1.1 itojun
418 1.1 itojun if ((kif->pfik_flags & PFI_IFLAG_INSTANCE) && kif->pfik_ifp == NULL) {
419 1.1 itojun pfr_clr_addrs(&kt->pfrkt_t, NULL, 0);
420 1.1 itojun return;
421 1.1 itojun }
422 1.1 itojun pfi_buffer_cnt = 0;
423 1.1 itojun if ((kif->pfik_flags & PFI_IFLAG_INSTANCE))
424 1.1 itojun pfi_instance_add(kif->pfik_ifp, net, flags);
425 1.1 itojun else if (strcmp(kif->pfik_name, "self")) {
426 1.1 itojun TAILQ_FOREACH(p, &kif->pfik_grouphead, pfik_instances)
427 1.1 itojun pfi_instance_add(p->pfik_ifp, net, flags);
428 1.1 itojun } else {
429 1.1 itojun RB_FOREACH(p, pfi_ifhead, &pfi_ifs)
430 1.1 itojun if (p->pfik_flags & PFI_IFLAG_INSTANCE)
431 1.1 itojun pfi_instance_add(p->pfik_ifp, net, flags);
432 1.1 itojun }
433 1.1 itojun t = kt->pfrkt_t;
434 1.1 itojun t.pfrt_flags = 0;
435 1.1 itojun if ((e = pfr_set_addrs(&t, pfi_buffer, pfi_buffer_cnt, &size2,
436 1.1 itojun NULL, NULL, NULL, 0)))
437 1.1 itojun printf("pfi_table_update: cannot set %d new addresses "
438 1.1 itojun "into table %s: %d\n", pfi_buffer_cnt, kt->pfrkt_name, e);
439 1.1 itojun }
440 1.1 itojun
441 1.1 itojun void
442 1.1 itojun pfi_instance_add(struct ifnet *ifp, int net, int flags)
443 1.1 itojun {
444 1.1 itojun struct ifaddr *ia;
445 1.1 itojun int got4 = 0, got6 = 0;
446 1.1 itojun int net2, af;
447 1.1 itojun
448 1.1 itojun if (ifp == NULL)
449 1.1 itojun return;
450 1.1 itojun TAILQ_FOREACH(ia, &ifp->if_addrlist, ifa_list) {
451 1.1 itojun if (ia->ifa_addr == NULL)
452 1.1 itojun continue;
453 1.1 itojun af = ia->ifa_addr->sa_family;
454 1.1 itojun if (af != AF_INET && af != AF_INET6)
455 1.1 itojun continue;
456 1.1 itojun if ((flags & PFI_AFLAG_BROADCAST) && af == AF_INET6)
457 1.1 itojun continue;
458 1.1 itojun if ((flags & PFI_AFLAG_BROADCAST) &&
459 1.1 itojun !(ifp->if_flags & IFF_BROADCAST))
460 1.1 itojun continue;
461 1.1 itojun if ((flags & PFI_AFLAG_PEER) &&
462 1.1 itojun !(ifp->if_flags & IFF_POINTOPOINT))
463 1.1 itojun continue;
464 1.1 itojun if ((flags & PFI_AFLAG_NETWORK) && af == AF_INET6 &&
465 1.1 itojun IN6_IS_ADDR_LINKLOCAL(
466 1.1 itojun &((struct sockaddr_in6 *)ia->ifa_addr)->sin6_addr))
467 1.1 itojun continue;
468 1.1 itojun if (flags & PFI_AFLAG_NOALIAS) {
469 1.1 itojun if (af == AF_INET && got4)
470 1.1 itojun continue;
471 1.1 itojun if (af == AF_INET6 && got6)
472 1.1 itojun continue;
473 1.1 itojun }
474 1.1 itojun if (af == AF_INET)
475 1.1 itojun got4 = 1;
476 1.1 itojun else
477 1.1 itojun got6 = 1;
478 1.1 itojun net2 = net;
479 1.1 itojun if (net2 == 128 && (flags & PFI_AFLAG_NETWORK)) {
480 1.1 itojun if (af == AF_INET) {
481 1.1 itojun net2 = pfi_unmask(&((struct sockaddr_in *)
482 1.1 itojun ia->ifa_netmask)->sin_addr);
483 1.1 itojun } else {
484 1.1 itojun net2 = pfi_unmask(&((struct sockaddr_in6 *)
485 1.1 itojun ia->ifa_netmask)->sin6_addr);
486 1.1 itojun }
487 1.1 itojun }
488 1.1 itojun if (af == AF_INET && net2 > 32)
489 1.1 itojun net2 = 32;
490 1.1 itojun if (flags & PFI_AFLAG_BROADCAST)
491 1.1 itojun pfi_address_add(ia->ifa_broadaddr, af, net2);
492 1.1 itojun else if (flags & PFI_AFLAG_PEER)
493 1.1 itojun pfi_address_add(ia->ifa_dstaddr, af, net2);
494 1.1 itojun else
495 1.1 itojun pfi_address_add(ia->ifa_addr, af, net2);
496 1.1 itojun }
497 1.1 itojun }
498 1.1 itojun
499 1.1 itojun void
500 1.1 itojun pfi_address_add(struct sockaddr *sa, int af, int net)
501 1.1 itojun {
502 1.1 itojun struct pfr_addr *p;
503 1.1 itojun int i;
504 1.1 itojun
505 1.1 itojun if (pfi_buffer_cnt >= pfi_buffer_max) {
506 1.1 itojun int new_max = pfi_buffer_max * 2;
507 1.1 itojun
508 1.1 itojun if (new_max > PFI_BUFFER_MAX) {
509 1.1 itojun printf("pfi_address_add: address buffer full (%d/%d)\n",
510 1.1 itojun pfi_buffer_cnt, PFI_BUFFER_MAX);
511 1.1 itojun return;
512 1.1 itojun }
513 1.1 itojun p = malloc(new_max * sizeof(*pfi_buffer), PFI_MTYPE,
514 1.1 itojun M_DONTWAIT);
515 1.1 itojun if (p == NULL) {
516 1.1 itojun printf("pfi_address_add: no memory to grow buffer "
517 1.1 itojun "(%d/%d)\n", pfi_buffer_cnt, PFI_BUFFER_MAX);
518 1.1 itojun return;
519 1.1 itojun }
520 1.1 itojun memcpy(pfi_buffer, p, pfi_buffer_cnt * sizeof(*pfi_buffer));
521 1.1 itojun /* no need to zero buffer */
522 1.1 itojun free(pfi_buffer, PFI_MTYPE);
523 1.1 itojun pfi_buffer = p;
524 1.1 itojun pfi_buffer_max = new_max;
525 1.1 itojun }
526 1.1 itojun if (af == AF_INET && net > 32)
527 1.1 itojun net = 128;
528 1.1 itojun p = pfi_buffer + pfi_buffer_cnt++;
529 1.1 itojun bzero(p, sizeof(*p));
530 1.1 itojun p->pfra_af = af;
531 1.1 itojun p->pfra_net = net;
532 1.1 itojun if (af == AF_INET)
533 1.1 itojun p->pfra_ip4addr = ((struct sockaddr_in *)sa)->sin_addr;
534 1.1 itojun if (af == AF_INET6) {
535 1.1 itojun p->pfra_ip6addr = ((struct sockaddr_in6 *)sa)->sin6_addr;
536 1.1 itojun if (IN6_IS_ADDR_LINKLOCAL(&p->pfra_ip6addr))
537 1.1 itojun p->pfra_ip6addr.s6_addr16[1] = 0;
538 1.1 itojun }
539 1.1 itojun /* mask network address bits */
540 1.1 itojun if (net < 128)
541 1.1 itojun ((caddr_t)p)[p->pfra_net/8] &= ~(0xFF >> (p->pfra_net%8));
542 1.1 itojun for (i = (p->pfra_net+7)/8; i < sizeof(p->pfra_u); i++)
543 1.1 itojun ((caddr_t)p)[i] = 0;
544 1.1 itojun }
545 1.1 itojun
546 1.1 itojun void
547 1.1 itojun pfi_dynaddr_remove(struct pf_addr_wrap *aw)
548 1.1 itojun {
549 1.1 itojun int s;
550 1.1 itojun
551 1.1 itojun if (aw->type != PF_ADDR_DYNIFTL || aw->p.dyn == NULL ||
552 1.1 itojun aw->p.dyn->pfid_kif == NULL || aw->p.dyn->pfid_kt == NULL)
553 1.1 itojun return;
554 1.1 itojun
555 1.1 itojun s = splsoftnet();
556 1.1 itojun hook_disestablish(aw->p.dyn->pfid_kif->pfik_ah_head,
557 1.1 itojun aw->p.dyn->pfid_hook_cookie);
558 1.1 itojun pfi_detach_rule(aw->p.dyn->pfid_kif);
559 1.1 itojun aw->p.dyn->pfid_kif = NULL;
560 1.1 itojun pfr_detach_table(aw->p.dyn->pfid_kt);
561 1.1 itojun aw->p.dyn->pfid_kt = NULL;
562 1.1 itojun pool_put(&pfi_addr_pl, aw->p.dyn);
563 1.1 itojun aw->p.dyn = NULL;
564 1.1 itojun splx(s);
565 1.1 itojun }
566 1.1 itojun
567 1.1 itojun void
568 1.1 itojun pfi_dynaddr_copyout(struct pf_addr_wrap *aw)
569 1.1 itojun {
570 1.1 itojun if (aw->type != PF_ADDR_DYNIFTL || aw->p.dyn == NULL ||
571 1.1 itojun aw->p.dyn->pfid_kif == NULL)
572 1.1 itojun return;
573 1.1 itojun aw->p.dyncnt = aw->p.dyn->pfid_acnt4 + aw->p.dyn->pfid_acnt6;
574 1.1 itojun }
575 1.1 itojun
576 1.1 itojun void
577 1.1 itojun pfi_kifaddr_update(void *v)
578 1.1 itojun {
579 1.1 itojun int s;
580 1.1 itojun
581 1.1 itojun s = splsoftnet();
582 1.1 itojun pfi_update++;
583 1.1 itojun pfi_dohooks(v);
584 1.1 itojun splx(s);
585 1.1 itojun }
586 1.1 itojun
587 1.5 yamt #ifdef __NetBSD__
588 1.5 yamt void
589 1.5 yamt pfi_kifaddr_update_if(struct ifnet *ifp)
590 1.5 yamt {
591 1.5 yamt struct pfi_kif *p;
592 1.5 yamt
593 1.5 yamt p = pfi_lookup_if(ifp->if_xname);
594 1.5 yamt if (p == NULL)
595 1.5 yamt panic("can't find interface");
596 1.5 yamt pfi_kifaddr_update(p);
597 1.5 yamt }
598 1.5 yamt #endif
599 1.5 yamt
600 1.1 itojun int
601 1.1 itojun pfi_if_compare(struct pfi_kif *p, struct pfi_kif *q)
602 1.1 itojun {
603 1.1 itojun return (strncmp(p->pfik_name, q->pfik_name, IFNAMSIZ));
604 1.1 itojun }
605 1.1 itojun
606 1.1 itojun struct pfi_kif *
607 1.1 itojun pfi_if_create(const char *name, struct pfi_kif *q, int flags)
608 1.1 itojun {
609 1.1 itojun struct pfi_kif *p;
610 1.1 itojun
611 1.1 itojun p = malloc(sizeof(*p), PFI_MTYPE, M_DONTWAIT);
612 1.1 itojun if (p == NULL)
613 1.1 itojun return (NULL);
614 1.1 itojun bzero(p, sizeof(*p));
615 1.1 itojun p->pfik_ah_head = malloc(sizeof(*p->pfik_ah_head), PFI_MTYPE,
616 1.1 itojun M_DONTWAIT);
617 1.1 itojun if (p->pfik_ah_head == NULL) {
618 1.1 itojun free(p, PFI_MTYPE);
619 1.1 itojun return (NULL);
620 1.1 itojun }
621 1.1 itojun bzero(p->pfik_ah_head, sizeof(*p->pfik_ah_head));
622 1.2 itojun #ifdef __NetBSD__
623 1.2 itojun p->pfik_ifaddrhooks = malloc(sizeof(*p->pfik_ifaddrhooks), PFI_MTYPE,
624 1.2 itojun M_DONTWAIT);
625 1.2 itojun if (p->pfik_ifaddrhooks == NULL) {
626 1.2 itojun free(p->pfik_ah_head, PFI_MTYPE);
627 1.2 itojun free(p, PFI_MTYPE);
628 1.2 itojun return (NULL);
629 1.2 itojun }
630 1.2 itojun bzero(p->pfik_ifaddrhooks, sizeof(*p->pfik_ifaddrhooks));
631 1.2 itojun #endif
632 1.1 itojun TAILQ_INIT(p->pfik_ah_head);
633 1.1 itojun TAILQ_INIT(&p->pfik_grouphead);
634 1.2 itojun #ifdef __NetBSD__
635 1.2 itojun TAILQ_INIT(p->pfik_ifaddrhooks);
636 1.2 itojun #endif
637 1.1 itojun strlcpy(p->pfik_name, name, sizeof(p->pfik_name));
638 1.1 itojun RB_INIT(&p->pfik_lan_ext);
639 1.1 itojun RB_INIT(&p->pfik_ext_gwy);
640 1.1 itojun p->pfik_flags = flags;
641 1.1 itojun p->pfik_parent = q;
642 1.1 itojun p->pfik_tzero = time.tv_sec;
643 1.1 itojun
644 1.1 itojun RB_INSERT(pfi_ifhead, &pfi_ifs, p);
645 1.1 itojun if (q != NULL) {
646 1.1 itojun q->pfik_addcnt++;
647 1.1 itojun TAILQ_INSERT_TAIL(&q->pfik_grouphead, p, pfik_instances);
648 1.1 itojun }
649 1.1 itojun pfi_ifcnt++;
650 1.1 itojun return (p);
651 1.1 itojun }
652 1.1 itojun
653 1.1 itojun int
654 1.1 itojun pfi_maybe_destroy(struct pfi_kif *p)
655 1.1 itojun {
656 1.1 itojun int i, j, k, s;
657 1.1 itojun struct pfi_kif *q = p->pfik_parent;
658 1.1 itojun
659 1.1 itojun if ((p->pfik_flags & (PFI_IFLAG_ATTACHED | PFI_IFLAG_GROUP)) ||
660 1.1 itojun p->pfik_rules > 0 || p->pfik_states > 0)
661 1.1 itojun return (0);
662 1.1 itojun
663 1.1 itojun s = splsoftnet();
664 1.1 itojun if (q != NULL) {
665 1.1 itojun for (i = 0; i < 2; i++)
666 1.1 itojun for (j = 0; j < 2; j++)
667 1.1 itojun for (k = 0; k < 2; k++) {
668 1.1 itojun q->pfik_bytes[i][j][k] +=
669 1.1 itojun p->pfik_bytes[i][j][k];
670 1.1 itojun q->pfik_packets[i][j][k] +=
671 1.1 itojun p->pfik_packets[i][j][k];
672 1.1 itojun }
673 1.1 itojun q->pfik_delcnt++;
674 1.1 itojun TAILQ_REMOVE(&q->pfik_grouphead, p, pfik_instances);
675 1.1 itojun }
676 1.1 itojun pfi_ifcnt--;
677 1.1 itojun RB_REMOVE(pfi_ifhead, &pfi_ifs, p);
678 1.1 itojun splx(s);
679 1.1 itojun
680 1.1 itojun free(p->pfik_ah_head, PFI_MTYPE);
681 1.1 itojun free(p, PFI_MTYPE);
682 1.1 itojun return (1);
683 1.1 itojun }
684 1.1 itojun
685 1.1 itojun void
686 1.1 itojun pfi_copy_group(char *p, const char *q, int m)
687 1.1 itojun {
688 1.1 itojun while (m > 1 && *q && !(*q >= '0' && *q <= '9')) {
689 1.1 itojun *p++ = *q++;
690 1.1 itojun m--;
691 1.1 itojun }
692 1.1 itojun if (m > 0)
693 1.1 itojun *p++ = '\0';
694 1.1 itojun }
695 1.1 itojun
696 1.1 itojun void
697 1.1 itojun pfi_dynamic_drivers(void)
698 1.1 itojun {
699 1.2 itojun #ifdef __OpenBSD__
700 1.1 itojun char *buses[] = PFI_DYNAMIC_BUSES;
701 1.1 itojun int nbuses = sizeof(buses)/sizeof(buses[0]);
702 1.1 itojun int enabled[sizeof(buses)/sizeof(buses[0])];
703 1.1 itojun struct device *dev;
704 1.1 itojun struct cfdata *cf;
705 1.1 itojun struct cfdriver *drv;
706 1.1 itojun short *p;
707 1.1 itojun int i;
708 1.1 itojun
709 1.1 itojun bzero(enabled, sizeof(enabled));
710 1.1 itojun TAILQ_FOREACH(dev, &alldevs, dv_list) {
711 1.1 itojun if (!(dev->dv_flags & DVF_ACTIVE))
712 1.1 itojun continue;
713 1.1 itojun for (i = 0; i < nbuses; i++)
714 1.1 itojun if (!enabled[i] && !strcmp(buses[i],
715 1.1 itojun dev->dv_cfdata->cf_driver->cd_name))
716 1.1 itojun enabled[i] = 1;
717 1.1 itojun }
718 1.1 itojun for (cf = cfdata; cf->cf_driver; cf++) {
719 1.1 itojun if (cf->cf_driver->cd_class != DV_IFNET)
720 1.1 itojun continue;
721 1.1 itojun for (p = cf->cf_parents; p && *p >= 0; p++) {
722 1.1 itojun if ((drv = cfdata[*p].cf_driver) == NULL)
723 1.1 itojun continue;
724 1.1 itojun for (i = 0; i < nbuses; i++)
725 1.1 itojun if (enabled[i] &&
726 1.1 itojun !strcmp(drv->cd_name, buses[i]))
727 1.1 itojun break;
728 1.1 itojun if (i < nbuses) {
729 1.1 itojun pfi_newgroup(cf->cf_driver->cd_name,
730 1.1 itojun PFI_IFLAG_DYNAMIC);
731 1.1 itojun break;
732 1.1 itojun }
733 1.1 itojun }
734 1.1 itojun }
735 1.2 itojun #else
736 1.2 itojun struct if_clone *ifc;
737 1.2 itojun extern LIST_HEAD(if_cloners, if_clone) if_cloners;
738 1.2 itojun
739 1.2 itojun LIST_FOREACH(ifc, &if_cloners, ifc_list)
740 1.2 itojun pfi_newgroup(ifc->ifc_name, PFI_IFLAG_DYNAMIC);
741 1.2 itojun #endif
742 1.1 itojun }
743 1.1 itojun
744 1.1 itojun void
745 1.1 itojun pfi_newgroup(const char *name, int flags)
746 1.1 itojun {
747 1.1 itojun struct pfi_kif *p;
748 1.1 itojun
749 1.1 itojun p = pfi_lookup_if(name);
750 1.1 itojun if (p == NULL)
751 1.1 itojun p = pfi_if_create(name, pfi_self, PFI_IFLAG_GROUP);
752 1.1 itojun if (p == NULL) {
753 1.1 itojun printf("pfi_newgroup: cannot allocate '%s' group", name);
754 1.1 itojun return;
755 1.1 itojun }
756 1.1 itojun p->pfik_flags |= flags;
757 1.1 itojun }
758 1.1 itojun
759 1.1 itojun void
760 1.1 itojun pfi_fill_oldstatus(struct pf_status *pfs)
761 1.1 itojun {
762 1.1 itojun struct pfi_kif *p, key;
763 1.1 itojun int i, j, k, s;
764 1.1 itojun
765 1.1 itojun strlcpy(key.pfik_name, pfs->ifname, sizeof(key.pfik_name));
766 1.1 itojun s = splsoftnet();
767 1.1 itojun p = RB_FIND(pfi_ifhead, &pfi_ifs, &key);
768 1.1 itojun if (p == NULL) {
769 1.1 itojun splx(s);
770 1.1 itojun return;
771 1.1 itojun }
772 1.1 itojun bzero(pfs->pcounters, sizeof(pfs->pcounters));
773 1.1 itojun bzero(pfs->bcounters, sizeof(pfs->bcounters));
774 1.1 itojun for (i = 0; i < 2; i++)
775 1.1 itojun for (j = 0; j < 2; j++)
776 1.1 itojun for (k = 0; k < 2; k++) {
777 1.1 itojun pfs->pcounters[i][j][k] =
778 1.1 itojun p->pfik_packets[i][j][k];
779 1.1 itojun pfs->bcounters[i][j] +=
780 1.1 itojun p->pfik_bytes[i][j][k];
781 1.1 itojun }
782 1.1 itojun splx(s);
783 1.1 itojun }
784 1.1 itojun
785 1.1 itojun int
786 1.1 itojun pfi_clr_istats(const char *name, int *nzero, int flags)
787 1.1 itojun {
788 1.1 itojun struct pfi_kif *p;
789 1.1 itojun int n = 0, s;
790 1.1 itojun long tzero = time.tv_sec;
791 1.1 itojun
792 1.1 itojun s = splsoftnet();
793 1.1 itojun ACCEPT_FLAGS(PFI_FLAG_GROUP|PFI_FLAG_INSTANCE);
794 1.1 itojun RB_FOREACH(p, pfi_ifhead, &pfi_ifs) {
795 1.1 itojun if (pfi_skip_if(name, p, flags))
796 1.1 itojun continue;
797 1.1 itojun bzero(p->pfik_packets, sizeof(p->pfik_packets));
798 1.1 itojun bzero(p->pfik_bytes, sizeof(p->pfik_bytes));
799 1.1 itojun p->pfik_tzero = tzero;
800 1.1 itojun n++;
801 1.1 itojun }
802 1.1 itojun splx(s);
803 1.1 itojun if (nzero != NULL)
804 1.1 itojun *nzero = n;
805 1.1 itojun return (0);
806 1.1 itojun }
807 1.1 itojun
808 1.1 itojun int
809 1.1 itojun pfi_get_ifaces(const char *name, struct pfi_if *buf, int *size, int flags)
810 1.1 itojun {
811 1.1 itojun struct pfi_kif *p;
812 1.1 itojun int s, n = 0;
813 1.1 itojun
814 1.1 itojun ACCEPT_FLAGS(PFI_FLAG_GROUP|PFI_FLAG_INSTANCE);
815 1.1 itojun s = splsoftnet();
816 1.1 itojun RB_FOREACH(p, pfi_ifhead, &pfi_ifs) {
817 1.1 itojun if (pfi_skip_if(name, p, flags))
818 1.1 itojun continue;
819 1.1 itojun if (*size > n++) {
820 1.1 itojun if (!p->pfik_tzero)
821 1.1 itojun p->pfik_tzero = boottime.tv_sec;
822 1.1 itojun if (copyout(p, buf++, sizeof(*buf))) {
823 1.1 itojun splx(s);
824 1.1 itojun return (EFAULT);
825 1.1 itojun }
826 1.1 itojun }
827 1.1 itojun }
828 1.1 itojun splx(s);
829 1.1 itojun *size = n;
830 1.1 itojun return (0);
831 1.1 itojun }
832 1.1 itojun
833 1.1 itojun struct pfi_kif *
834 1.1 itojun pfi_lookup_if(const char *name)
835 1.1 itojun {
836 1.1 itojun struct pfi_kif *p, key;
837 1.1 itojun
838 1.1 itojun strlcpy(key.pfik_name, name, sizeof(key.pfik_name));
839 1.1 itojun p = RB_FIND(pfi_ifhead, &pfi_ifs, &key);
840 1.1 itojun return (p);
841 1.1 itojun }
842 1.1 itojun
843 1.1 itojun int
844 1.1 itojun pfi_skip_if(const char *filter, struct pfi_kif *p, int f)
845 1.1 itojun {
846 1.1 itojun int n;
847 1.1 itojun
848 1.1 itojun if ((p->pfik_flags & PFI_IFLAG_GROUP) && !(f & PFI_FLAG_GROUP))
849 1.1 itojun return (1);
850 1.1 itojun if ((p->pfik_flags & PFI_IFLAG_INSTANCE) && !(f & PFI_FLAG_INSTANCE))
851 1.1 itojun return (1);
852 1.1 itojun if (filter == NULL || !*filter)
853 1.1 itojun return (0);
854 1.1 itojun if (!strcmp(p->pfik_name, filter))
855 1.1 itojun return (0); /* exact match */
856 1.1 itojun n = strlen(filter);
857 1.1 itojun if (n < 1 || n >= IFNAMSIZ)
858 1.1 itojun return (1); /* sanity check */
859 1.1 itojun if (filter[n-1] >= '0' && filter[n-1] <= '9')
860 1.1 itojun return (1); /* only do exact match in that case */
861 1.1 itojun if (strncmp(p->pfik_name, filter, n))
862 1.1 itojun return (1); /* prefix doesn't match */
863 1.1 itojun return (p->pfik_name[n] < '0' || p->pfik_name[n] > '9');
864 1.1 itojun }
865 1.1 itojun
866 1.1 itojun /* from pf_print_state.c */
867 1.1 itojun int
868 1.1 itojun pfi_unmask(void *addr)
869 1.1 itojun {
870 1.1 itojun struct pf_addr *m = addr;
871 1.1 itojun int i = 31, j = 0, b = 0;
872 1.1 itojun u_int32_t tmp;
873 1.1 itojun
874 1.1 itojun while (j < 4 && m->addr32[j] == 0xffffffff) {
875 1.1 itojun b += 32;
876 1.1 itojun j++;
877 1.1 itojun }
878 1.1 itojun if (j < 4) {
879 1.1 itojun tmp = ntohl(m->addr32[j]);
880 1.1 itojun for (i = 31; tmp & (1 << i); --i)
881 1.1 itojun b++;
882 1.1 itojun }
883 1.1 itojun return (b);
884 1.1 itojun }
885 1.1 itojun
886 1.1 itojun void
887 1.1 itojun pfi_dohooks(struct pfi_kif *p)
888 1.1 itojun {
889 1.1 itojun for (; p != NULL; p = p->pfik_parent)
890 1.1 itojun dohooks(p->pfik_ah_head, 0);
891 1.1 itojun }
892 1.1 itojun
893 1.1 itojun int
894 1.1 itojun pfi_match_addr(struct pfi_dynaddr *dyn, struct pf_addr *a, sa_family_t af)
895 1.1 itojun {
896 1.1 itojun if (af == AF_INET) {
897 1.1 itojun switch (dyn->pfid_acnt4) {
898 1.1 itojun case 0:
899 1.1 itojun return (0);
900 1.1 itojun case 1:
901 1.1 itojun return (PF_MATCHA(0, &dyn->pfid_addr4,
902 1.1 itojun &dyn->pfid_mask4, a, AF_INET));
903 1.1 itojun default:
904 1.1 itojun return (pfr_match_addr(dyn->pfid_kt, a, AF_INET));
905 1.1 itojun }
906 1.1 itojun } else {
907 1.1 itojun switch (dyn->pfid_acnt6) {
908 1.1 itojun case 0:
909 1.1 itojun return (0);
910 1.1 itojun case 1:
911 1.1 itojun return (PF_MATCHA(0, &dyn->pfid_addr6,
912 1.1 itojun &dyn->pfid_mask6, a, AF_INET6));
913 1.1 itojun default:
914 1.1 itojun return (pfr_match_addr(dyn->pfid_kt, a, AF_INET6));
915 1.1 itojun }
916 1.1 itojun }
917 1.1 itojun }
918 1.2 itojun
919 1.2 itojun /* from openbsd/sys/kern/kern_subr.c */
920 1.2 itojun #ifdef __NetBSD__
921 1.2 itojun static void *
922 1.4 yamt hook_establish(struct hook_desc_head *head, int tail, void (*fn)(void *),
923 1.4 yamt void *arg)
924 1.2 itojun {
925 1.2 itojun struct hook_desc *hdp;
926 1.2 itojun
927 1.2 itojun hdp = (struct hook_desc *)malloc(sizeof (*hdp), M_DEVBUF, M_NOWAIT);
928 1.2 itojun if (hdp == NULL)
929 1.2 itojun return (NULL);
930 1.2 itojun
931 1.2 itojun hdp->hd_fn = fn;
932 1.2 itojun hdp->hd_arg = arg;
933 1.2 itojun if (tail)
934 1.2 itojun TAILQ_INSERT_TAIL(head, hdp, hd_list);
935 1.2 itojun else
936 1.2 itojun TAILQ_INSERT_HEAD(head, hdp, hd_list);
937 1.2 itojun
938 1.2 itojun return (hdp);
939 1.2 itojun }
940 1.2 itojun
941 1.2 itojun static void
942 1.4 yamt hook_disestablish(struct hook_desc_head *head, void *vhook)
943 1.2 itojun {
944 1.2 itojun struct hook_desc *hdp;
945 1.2 itojun
946 1.2 itojun #ifdef DIAGNOSTIC
947 1.2 itojun for (hdp = TAILQ_FIRST(head); hdp != NULL;
948 1.2 itojun hdp = TAILQ_NEXT(hdp, hd_list))
949 1.2 itojun if (hdp == vhook)
950 1.2 itojun break;
951 1.2 itojun if (hdp == NULL)
952 1.2 itojun panic("hook_disestablish: hook not established");
953 1.2 itojun #endif
954 1.2 itojun hdp = vhook;
955 1.2 itojun TAILQ_REMOVE(head, hdp, hd_list);
956 1.2 itojun free(hdp, M_DEVBUF);
957 1.2 itojun }
958 1.5 yamt
959 1.5 yamt static void
960 1.5 yamt dohooks(struct hook_desc_head *head, int flags)
961 1.5 yamt {
962 1.5 yamt struct hook_desc *hdp;
963 1.5 yamt
964 1.5 yamt if ((flags & HOOK_REMOVE) == 0) {
965 1.5 yamt TAILQ_FOREACH(hdp, head, hd_list) {
966 1.5 yamt (*hdp->hd_fn)(hdp->hd_arg);
967 1.5 yamt }
968 1.5 yamt } else {
969 1.5 yamt while ((hdp = TAILQ_FIRST(head)) != NULL) {
970 1.5 yamt TAILQ_REMOVE(head, hdp, hd_list);
971 1.5 yamt (*hdp->hd_fn)(hdp->hd_arg);
972 1.5 yamt if ((flags & HOOK_FREE) != 0)
973 1.5 yamt free(hdp, M_DEVBUF);
974 1.5 yamt }
975 1.5 yamt }
976 1.5 yamt }
977 1.2 itojun #endif
978