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