altq_subr.c revision 1.23.4.1 1 1.23.4.1 bouyer /* $NetBSD: altq_subr.c,v 1.23.4.1 2007/10/25 22:35:29 bouyer Exp $ */
2 1.16 peter /* $KAME: altq_subr.c,v 1.24 2005/04/13 03:44:25 suz Exp $ */
3 1.1 thorpej
4 1.1 thorpej /*
5 1.16 peter * Copyright (C) 1997-2003
6 1.1 thorpej * Sony Computer Science Laboratories Inc. All rights reserved.
7 1.1 thorpej *
8 1.1 thorpej * Redistribution and use in source and binary forms, with or without
9 1.1 thorpej * modification, are permitted provided that the following conditions
10 1.1 thorpej * are met:
11 1.1 thorpej * 1. Redistributions of source code must retain the above copyright
12 1.1 thorpej * notice, this list of conditions and the following disclaimer.
13 1.1 thorpej * 2. Redistributions in binary form must reproduce the above copyright
14 1.1 thorpej * notice, this list of conditions and the following disclaimer in the
15 1.1 thorpej * documentation and/or other materials provided with the distribution.
16 1.1 thorpej *
17 1.1 thorpej * THIS SOFTWARE IS PROVIDED BY SONY CSL AND CONTRIBUTORS ``AS IS'' AND
18 1.1 thorpej * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 1.1 thorpej * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 1.1 thorpej * ARE DISCLAIMED. IN NO EVENT SHALL SONY CSL OR CONTRIBUTORS BE LIABLE
21 1.1 thorpej * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 1.1 thorpej * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 1.1 thorpej * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 1.1 thorpej * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 1.1 thorpej * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 1.1 thorpej * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 1.1 thorpej * SUCH DAMAGE.
28 1.1 thorpej */
29 1.7 lukem
30 1.7 lukem #include <sys/cdefs.h>
31 1.23.4.1 bouyer __KERNEL_RCSID(0, "$NetBSD: altq_subr.c,v 1.23.4.1 2007/10/25 22:35:29 bouyer Exp $");
32 1.1 thorpej
33 1.16 peter #ifdef _KERNEL_OPT
34 1.1 thorpej #include "opt_altq.h"
35 1.1 thorpej #include "opt_inet.h"
36 1.17 peter #include "pf.h"
37 1.1 thorpej #endif
38 1.1 thorpej
39 1.1 thorpej #include <sys/param.h>
40 1.1 thorpej #include <sys/malloc.h>
41 1.1 thorpej #include <sys/mbuf.h>
42 1.1 thorpej #include <sys/systm.h>
43 1.1 thorpej #include <sys/proc.h>
44 1.1 thorpej #include <sys/socket.h>
45 1.1 thorpej #include <sys/socketvar.h>
46 1.1 thorpej #include <sys/kernel.h>
47 1.1 thorpej #include <sys/errno.h>
48 1.1 thorpej #include <sys/syslog.h>
49 1.1 thorpej #include <sys/sysctl.h>
50 1.1 thorpej #include <sys/queue.h>
51 1.1 thorpej
52 1.1 thorpej #include <net/if.h>
53 1.1 thorpej #include <net/if_dl.h>
54 1.1 thorpej #include <net/if_types.h>
55 1.1 thorpej
56 1.1 thorpej #include <netinet/in.h>
57 1.1 thorpej #include <netinet/in_systm.h>
58 1.1 thorpej #include <netinet/ip.h>
59 1.1 thorpej #ifdef INET6
60 1.1 thorpej #include <netinet/ip6.h>
61 1.1 thorpej #endif
62 1.1 thorpej #include <netinet/tcp.h>
63 1.1 thorpej #include <netinet/udp.h>
64 1.1 thorpej
65 1.17 peter #if NPF > 0
66 1.16 peter #include <net/pfvar.h>
67 1.17 peter #endif
68 1.1 thorpej #include <altq/altq.h>
69 1.16 peter #ifdef ALTQ3_COMPAT
70 1.1 thorpej #include <altq/altq_conf.h>
71 1.16 peter #endif
72 1.1 thorpej
73 1.8 itojun /* machine dependent clock related includes */
74 1.1 thorpej #ifdef __FreeBSD__
75 1.1 thorpej #include "opt_cpu.h" /* for FreeBSD-2.2.8 to get i586_ctr_freq */
76 1.1 thorpej #include <machine/clock.h>
77 1.1 thorpej #endif
78 1.8 itojun #if defined(__i386__)
79 1.16 peter #include <machine/cpufunc.h> /* for pentium tsc */
80 1.8 itojun #include <machine/specialreg.h> /* for CPUID_TSC */
81 1.8 itojun #ifdef __FreeBSD__
82 1.8 itojun #include <machine/md_var.h> /* for cpu_feature */
83 1.8 itojun #elif defined(__NetBSD__) || defined(__OpenBSD__)
84 1.23.4.1 bouyer #include <sys/cpu.h> /* for cpu_feature */
85 1.8 itojun #endif
86 1.8 itojun #endif /* __i386__ */
87 1.1 thorpej
88 1.1 thorpej /*
89 1.1 thorpej * internal function prototypes
90 1.1 thorpej */
91 1.16 peter static void tbr_timeout(void *);
92 1.16 peter int (*altq_input)(struct mbuf *, int) = NULL;
93 1.16 peter static int tbr_timer = 0; /* token bucket regulator timer */
94 1.22 ad static struct callout tbr_callout;
95 1.16 peter
96 1.16 peter #ifdef ALTQ3_CLFIER_COMPAT
97 1.16 peter static int extract_ports4(struct mbuf *, struct ip *, struct flowinfo_in *);
98 1.1 thorpej #ifdef INET6
99 1.16 peter static int extract_ports6(struct mbuf *, struct ip6_hdr *,
100 1.16 peter struct flowinfo_in6 *);
101 1.1 thorpej #endif
102 1.16 peter static int apply_filter4(u_int32_t, struct flow_filter *,
103 1.16 peter struct flowinfo_in *);
104 1.16 peter static int apply_ppfilter4(u_int32_t, struct flow_filter *,
105 1.16 peter struct flowinfo_in *);
106 1.1 thorpej #ifdef INET6
107 1.16 peter static int apply_filter6(u_int32_t, struct flow_filter6 *,
108 1.16 peter struct flowinfo_in6 *);
109 1.1 thorpej #endif
110 1.16 peter static int apply_tosfilter4(u_int32_t, struct flow_filter *,
111 1.16 peter struct flowinfo_in *);
112 1.16 peter static u_long get_filt_handle(struct acc_classifier *, int);
113 1.16 peter static struct acc_filter *filth_to_filtp(struct acc_classifier *, u_long);
114 1.16 peter static u_int32_t filt2fibmask(struct flow_filter *);
115 1.16 peter
116 1.16 peter static void ip4f_cache(struct ip *, struct flowinfo_in *);
117 1.16 peter static int ip4f_lookup(struct ip *, struct flowinfo_in *);
118 1.16 peter static int ip4f_init(void);
119 1.16 peter static struct ip4_frag *ip4f_alloc(void);
120 1.16 peter static void ip4f_free(struct ip4_frag *);
121 1.16 peter #endif /* ALTQ3_CLFIER_COMPAT */
122 1.1 thorpej
123 1.1 thorpej /*
124 1.1 thorpej * alternate queueing support routines
125 1.1 thorpej */
126 1.1 thorpej
127 1.16 peter /* look up the queue state by the interface name and the queueing type. */
128 1.1 thorpej void *
129 1.16 peter altq_lookup(char *name, int type)
130 1.1 thorpej {
131 1.1 thorpej struct ifnet *ifp;
132 1.1 thorpej
133 1.1 thorpej if ((ifp = ifunit(name)) != NULL) {
134 1.1 thorpej if (type != ALTQT_NONE && ifp->if_snd.altq_type == type)
135 1.1 thorpej return (ifp->if_snd.altq_disc);
136 1.1 thorpej }
137 1.1 thorpej
138 1.1 thorpej return NULL;
139 1.1 thorpej }
140 1.1 thorpej
141 1.1 thorpej int
142 1.16 peter altq_attach(struct ifaltq *ifq, int type, void *discipline,
143 1.16 peter int (*enqueue)(struct ifaltq *, struct mbuf *, struct altq_pktattr *),
144 1.16 peter struct mbuf *(*dequeue)(struct ifaltq *, int),
145 1.16 peter int (*request)(struct ifaltq *, int, void *),
146 1.16 peter void *clfier, void *(*classify)(void *, struct mbuf *, int))
147 1.1 thorpej {
148 1.1 thorpej if (!ALTQ_IS_READY(ifq))
149 1.1 thorpej return ENXIO;
150 1.16 peter
151 1.16 peter #ifdef ALTQ3_COMPAT
152 1.16 peter /*
153 1.16 peter * pfaltq can override the existing discipline, but altq3 cannot.
154 1.16 peter * check these if clfier is not NULL (which implies altq3).
155 1.16 peter */
156 1.16 peter if (clfier != NULL) {
157 1.16 peter if (ALTQ_IS_ENABLED(ifq))
158 1.16 peter return EBUSY;
159 1.16 peter if (ALTQ_IS_ATTACHED(ifq))
160 1.16 peter return EEXIST;
161 1.16 peter }
162 1.16 peter #endif
163 1.1 thorpej ifq->altq_type = type;
164 1.1 thorpej ifq->altq_disc = discipline;
165 1.1 thorpej ifq->altq_enqueue = enqueue;
166 1.1 thorpej ifq->altq_dequeue = dequeue;
167 1.1 thorpej ifq->altq_request = request;
168 1.1 thorpej ifq->altq_clfier = clfier;
169 1.1 thorpej ifq->altq_classify = classify;
170 1.16 peter ifq->altq_flags &= (ALTQF_CANTCHANGE|ALTQF_ENABLED);
171 1.16 peter #ifdef ALTQ3_COMPAT
172 1.1 thorpej #ifdef ALTQ_KLD
173 1.1 thorpej altq_module_incref(type);
174 1.1 thorpej #endif
175 1.16 peter #endif
176 1.1 thorpej return 0;
177 1.1 thorpej }
178 1.1 thorpej
179 1.1 thorpej int
180 1.16 peter altq_detach(struct ifaltq *ifq)
181 1.1 thorpej {
182 1.1 thorpej if (!ALTQ_IS_READY(ifq))
183 1.1 thorpej return ENXIO;
184 1.1 thorpej if (ALTQ_IS_ENABLED(ifq))
185 1.1 thorpej return EBUSY;
186 1.1 thorpej if (!ALTQ_IS_ATTACHED(ifq))
187 1.1 thorpej return (0);
188 1.16 peter #ifdef ALTQ3_COMPAT
189 1.1 thorpej #ifdef ALTQ_KLD
190 1.1 thorpej altq_module_declref(ifq->altq_type);
191 1.1 thorpej #endif
192 1.16 peter #endif
193 1.16 peter
194 1.1 thorpej ifq->altq_type = ALTQT_NONE;
195 1.1 thorpej ifq->altq_disc = NULL;
196 1.1 thorpej ifq->altq_enqueue = NULL;
197 1.1 thorpej ifq->altq_dequeue = NULL;
198 1.1 thorpej ifq->altq_request = NULL;
199 1.1 thorpej ifq->altq_clfier = NULL;
200 1.1 thorpej ifq->altq_classify = NULL;
201 1.1 thorpej ifq->altq_flags &= ALTQF_CANTCHANGE;
202 1.1 thorpej return 0;
203 1.1 thorpej }
204 1.1 thorpej
205 1.1 thorpej int
206 1.16 peter altq_enable(struct ifaltq *ifq)
207 1.1 thorpej {
208 1.1 thorpej int s;
209 1.12 perry
210 1.1 thorpej if (!ALTQ_IS_READY(ifq))
211 1.1 thorpej return ENXIO;
212 1.1 thorpej if (ALTQ_IS_ENABLED(ifq))
213 1.1 thorpej return 0;
214 1.1 thorpej
215 1.5 thorpej s = splnet();
216 1.1 thorpej IFQ_PURGE(ifq);
217 1.1 thorpej ASSERT(ifq->ifq_len == 0);
218 1.1 thorpej ifq->altq_flags |= ALTQF_ENABLED;
219 1.1 thorpej if (ifq->altq_clfier != NULL)
220 1.1 thorpej ifq->altq_flags |= ALTQF_CLASSIFY;
221 1.1 thorpej splx(s);
222 1.1 thorpej
223 1.1 thorpej return 0;
224 1.1 thorpej }
225 1.1 thorpej
226 1.1 thorpej int
227 1.16 peter altq_disable(struct ifaltq *ifq)
228 1.1 thorpej {
229 1.1 thorpej int s;
230 1.12 perry
231 1.1 thorpej if (!ALTQ_IS_ENABLED(ifq))
232 1.1 thorpej return 0;
233 1.1 thorpej
234 1.5 thorpej s = splnet();
235 1.1 thorpej IFQ_PURGE(ifq);
236 1.1 thorpej ASSERT(ifq->ifq_len == 0);
237 1.1 thorpej ifq->altq_flags &= ~(ALTQF_ENABLED|ALTQF_CLASSIFY);
238 1.1 thorpej splx(s);
239 1.1 thorpej return 0;
240 1.1 thorpej }
241 1.1 thorpej
242 1.16 peter #ifdef ALTQ_DEBUG
243 1.1 thorpej void
244 1.16 peter altq_assert(const char *file, int line, const char *failedexpr)
245 1.1 thorpej {
246 1.1 thorpej (void)printf("altq assertion \"%s\" failed: file \"%s\", line %d\n",
247 1.1 thorpej failedexpr, file, line);
248 1.1 thorpej panic("altq assertion");
249 1.1 thorpej /* NOTREACHED */
250 1.1 thorpej }
251 1.16 peter #endif
252 1.1 thorpej
253 1.1 thorpej /*
254 1.1 thorpej * internal representation of token bucket parameters
255 1.16 peter * rate: byte_per_unittime << 32
256 1.1 thorpej * (((bits_per_sec) / 8) << 32) / machclk_freq
257 1.1 thorpej * depth: byte << 32
258 1.1 thorpej *
259 1.1 thorpej */
260 1.1 thorpej #define TBR_SHIFT 32
261 1.1 thorpej #define TBR_SCALE(x) ((int64_t)(x) << TBR_SHIFT)
262 1.1 thorpej #define TBR_UNSCALE(x) ((x) >> TBR_SHIFT)
263 1.1 thorpej
264 1.1 thorpej struct mbuf *
265 1.16 peter tbr_dequeue(struct ifaltq *ifq, int op)
266 1.1 thorpej {
267 1.1 thorpej struct tb_regulator *tbr;
268 1.1 thorpej struct mbuf *m;
269 1.1 thorpej int64_t interval;
270 1.1 thorpej u_int64_t now;
271 1.1 thorpej
272 1.1 thorpej tbr = ifq->altq_tbr;
273 1.1 thorpej if (op == ALTDQ_REMOVE && tbr->tbr_lastop == ALTDQ_POLL) {
274 1.1 thorpej /* if this is a remove after poll, bypass tbr check */
275 1.1 thorpej } else {
276 1.1 thorpej /* update token only when it is negative */
277 1.1 thorpej if (tbr->tbr_token <= 0) {
278 1.1 thorpej now = read_machclk();
279 1.1 thorpej interval = now - tbr->tbr_last;
280 1.1 thorpej if (interval >= tbr->tbr_filluptime)
281 1.1 thorpej tbr->tbr_token = tbr->tbr_depth;
282 1.1 thorpej else {
283 1.1 thorpej tbr->tbr_token += interval * tbr->tbr_rate;
284 1.1 thorpej if (tbr->tbr_token > tbr->tbr_depth)
285 1.1 thorpej tbr->tbr_token = tbr->tbr_depth;
286 1.1 thorpej }
287 1.1 thorpej tbr->tbr_last = now;
288 1.1 thorpej }
289 1.1 thorpej /* if token is still negative, don't allow dequeue */
290 1.1 thorpej if (tbr->tbr_token <= 0)
291 1.1 thorpej return (NULL);
292 1.1 thorpej }
293 1.1 thorpej
294 1.1 thorpej if (ALTQ_IS_ENABLED(ifq))
295 1.1 thorpej m = (*ifq->altq_dequeue)(ifq, op);
296 1.1 thorpej else {
297 1.1 thorpej if (op == ALTDQ_POLL)
298 1.1 thorpej IF_POLL(ifq, m);
299 1.1 thorpej else
300 1.1 thorpej IF_DEQUEUE(ifq, m);
301 1.1 thorpej }
302 1.1 thorpej
303 1.1 thorpej if (m != NULL && op == ALTDQ_REMOVE)
304 1.1 thorpej tbr->tbr_token -= TBR_SCALE(m_pktlen(m));
305 1.1 thorpej tbr->tbr_lastop = op;
306 1.1 thorpej return (m);
307 1.1 thorpej }
308 1.1 thorpej
309 1.1 thorpej /*
310 1.1 thorpej * set a token bucket regulator.
311 1.1 thorpej * if the specified rate is zero, the token bucket regulator is deleted.
312 1.1 thorpej */
313 1.1 thorpej int
314 1.16 peter tbr_set(struct ifaltq *ifq, struct tb_profile *profile)
315 1.1 thorpej {
316 1.1 thorpej struct tb_regulator *tbr, *otbr;
317 1.1 thorpej
318 1.1 thorpej if (machclk_freq == 0)
319 1.1 thorpej init_machclk();
320 1.1 thorpej if (machclk_freq == 0) {
321 1.11 wiz printf("tbr_set: no CPU clock available!\n");
322 1.1 thorpej return (ENXIO);
323 1.1 thorpej }
324 1.12 perry
325 1.1 thorpej if (profile->rate == 0) {
326 1.1 thorpej /* delete this tbr */
327 1.1 thorpej if ((tbr = ifq->altq_tbr) == NULL)
328 1.1 thorpej return (ENOENT);
329 1.1 thorpej ifq->altq_tbr = NULL;
330 1.14 christos free(tbr, M_DEVBUF);
331 1.1 thorpej return (0);
332 1.1 thorpej }
333 1.1 thorpej
334 1.14 christos tbr = malloc(sizeof(struct tb_regulator), M_DEVBUF, M_WAITOK|M_ZERO);
335 1.1 thorpej if (tbr == NULL)
336 1.1 thorpej return (ENOMEM);
337 1.1 thorpej
338 1.1 thorpej tbr->tbr_rate = TBR_SCALE(profile->rate / 8) / machclk_freq;
339 1.1 thorpej tbr->tbr_depth = TBR_SCALE(profile->depth);
340 1.1 thorpej if (tbr->tbr_rate > 0)
341 1.1 thorpej tbr->tbr_filluptime = tbr->tbr_depth / tbr->tbr_rate;
342 1.1 thorpej else
343 1.1 thorpej tbr->tbr_filluptime = 0xffffffffffffffffLL;
344 1.1 thorpej tbr->tbr_token = tbr->tbr_depth;
345 1.1 thorpej tbr->tbr_last = read_machclk();
346 1.1 thorpej tbr->tbr_lastop = ALTDQ_REMOVE;
347 1.1 thorpej
348 1.1 thorpej otbr = ifq->altq_tbr;
349 1.1 thorpej ifq->altq_tbr = tbr; /* set the new tbr */
350 1.1 thorpej
351 1.1 thorpej if (otbr != NULL)
352 1.14 christos free(otbr, M_DEVBUF);
353 1.1 thorpej else {
354 1.1 thorpej if (tbr_timer == 0) {
355 1.1 thorpej CALLOUT_RESET(&tbr_callout, 1, tbr_timeout, (void *)0);
356 1.1 thorpej tbr_timer = 1;
357 1.1 thorpej }
358 1.1 thorpej }
359 1.1 thorpej return (0);
360 1.1 thorpej }
361 1.1 thorpej
362 1.1 thorpej /*
363 1.1 thorpej * tbr_timeout goes through the interface list, and kicks the drivers
364 1.1 thorpej * if necessary.
365 1.1 thorpej */
366 1.1 thorpej static void
367 1.19 christos tbr_timeout(void *arg)
368 1.1 thorpej {
369 1.1 thorpej struct ifnet *ifp;
370 1.1 thorpej int active, s;
371 1.1 thorpej
372 1.1 thorpej active = 0;
373 1.5 thorpej s = splnet();
374 1.16 peter for (ifp = TAILQ_FIRST(&ifnet); ifp; ifp = TAILQ_NEXT(ifp, if_list)) {
375 1.1 thorpej if (!TBR_IS_ENABLED(&ifp->if_snd))
376 1.1 thorpej continue;
377 1.1 thorpej active++;
378 1.1 thorpej if (!IFQ_IS_EMPTY(&ifp->if_snd) && ifp->if_start != NULL)
379 1.1 thorpej (*ifp->if_start)(ifp);
380 1.16 peter }
381 1.16 peter splx(s);
382 1.16 peter if (active > 0)
383 1.16 peter CALLOUT_RESET(&tbr_callout, 1, tbr_timeout, (void *)0);
384 1.16 peter else
385 1.16 peter tbr_timer = 0; /* don't need tbr_timer anymore */
386 1.16 peter #if defined(__alpha__) && !defined(ALTQ_NOPCC)
387 1.16 peter {
388 1.16 peter /*
389 1.16 peter * XXX read out the machine dependent clock once a second
390 1.16 peter * to detect counter wrap-around.
391 1.16 peter */
392 1.16 peter static u_int cnt;
393 1.16 peter
394 1.16 peter if (++cnt >= hz) {
395 1.16 peter (void)read_machclk();
396 1.16 peter cnt = 0;
397 1.16 peter }
398 1.16 peter }
399 1.16 peter #endif /* __alpha__ && !ALTQ_NOPCC */
400 1.16 peter }
401 1.16 peter
402 1.16 peter /*
403 1.16 peter * get token bucket regulator profile
404 1.16 peter */
405 1.16 peter int
406 1.16 peter tbr_get(struct ifaltq *ifq, struct tb_profile *profile)
407 1.16 peter {
408 1.16 peter struct tb_regulator *tbr;
409 1.16 peter
410 1.16 peter if ((tbr = ifq->altq_tbr) == NULL) {
411 1.16 peter profile->rate = 0;
412 1.16 peter profile->depth = 0;
413 1.16 peter } else {
414 1.16 peter profile->rate =
415 1.16 peter (u_int)TBR_UNSCALE(tbr->tbr_rate * 8 * machclk_freq);
416 1.16 peter profile->depth = (u_int)TBR_UNSCALE(tbr->tbr_depth);
417 1.16 peter }
418 1.16 peter return (0);
419 1.16 peter }
420 1.16 peter
421 1.17 peter #if NPF > 0
422 1.16 peter /*
423 1.16 peter * attach a discipline to the interface. if one already exists, it is
424 1.16 peter * overridden.
425 1.16 peter */
426 1.16 peter int
427 1.16 peter altq_pfattach(struct pf_altq *a)
428 1.16 peter {
429 1.18 peter int error = 0;
430 1.16 peter
431 1.16 peter switch (a->scheduler) {
432 1.16 peter case ALTQT_NONE:
433 1.16 peter break;
434 1.16 peter #ifdef ALTQ_CBQ
435 1.16 peter case ALTQT_CBQ:
436 1.16 peter error = cbq_pfattach(a);
437 1.16 peter break;
438 1.16 peter #endif
439 1.16 peter #ifdef ALTQ_PRIQ
440 1.16 peter case ALTQT_PRIQ:
441 1.16 peter error = priq_pfattach(a);
442 1.16 peter break;
443 1.16 peter #endif
444 1.16 peter #ifdef ALTQ_HFSC
445 1.16 peter case ALTQT_HFSC:
446 1.16 peter error = hfsc_pfattach(a);
447 1.16 peter break;
448 1.16 peter #endif
449 1.16 peter default:
450 1.16 peter error = ENXIO;
451 1.16 peter }
452 1.16 peter
453 1.16 peter return (error);
454 1.16 peter }
455 1.16 peter
456 1.16 peter /*
457 1.16 peter * detach a discipline from the interface.
458 1.16 peter * it is possible that the discipline was already overridden by another
459 1.16 peter * discipline.
460 1.16 peter */
461 1.16 peter int
462 1.16 peter altq_pfdetach(struct pf_altq *a)
463 1.16 peter {
464 1.16 peter struct ifnet *ifp;
465 1.16 peter int s, error = 0;
466 1.16 peter
467 1.16 peter if ((ifp = ifunit(a->ifname)) == NULL)
468 1.16 peter return (EINVAL);
469 1.16 peter
470 1.16 peter /* if this discipline is no longer referenced, just return */
471 1.16 peter if (a->altq_disc == NULL || a->altq_disc != ifp->if_snd.altq_disc)
472 1.16 peter return (0);
473 1.16 peter
474 1.16 peter s = splnet();
475 1.16 peter if (ALTQ_IS_ENABLED(&ifp->if_snd))
476 1.16 peter error = altq_disable(&ifp->if_snd);
477 1.16 peter if (error == 0)
478 1.16 peter error = altq_detach(&ifp->if_snd);
479 1.16 peter splx(s);
480 1.16 peter
481 1.16 peter return (error);
482 1.16 peter }
483 1.16 peter
484 1.16 peter /*
485 1.16 peter * add a discipline or a queue
486 1.16 peter */
487 1.16 peter int
488 1.16 peter altq_add(struct pf_altq *a)
489 1.16 peter {
490 1.16 peter int error = 0;
491 1.16 peter
492 1.16 peter if (a->qname[0] != 0)
493 1.16 peter return (altq_add_queue(a));
494 1.16 peter
495 1.16 peter if (machclk_freq == 0)
496 1.16 peter init_machclk();
497 1.16 peter if (machclk_freq == 0)
498 1.16 peter panic("altq_add: no CPU clock");
499 1.16 peter
500 1.16 peter switch (a->scheduler) {
501 1.16 peter #ifdef ALTQ_CBQ
502 1.16 peter case ALTQT_CBQ:
503 1.16 peter error = cbq_add_altq(a);
504 1.16 peter break;
505 1.16 peter #endif
506 1.16 peter #ifdef ALTQ_PRIQ
507 1.16 peter case ALTQT_PRIQ:
508 1.16 peter error = priq_add_altq(a);
509 1.16 peter break;
510 1.16 peter #endif
511 1.16 peter #ifdef ALTQ_HFSC
512 1.16 peter case ALTQT_HFSC:
513 1.16 peter error = hfsc_add_altq(a);
514 1.16 peter break;
515 1.16 peter #endif
516 1.16 peter default:
517 1.16 peter error = ENXIO;
518 1.16 peter }
519 1.16 peter
520 1.16 peter return (error);
521 1.16 peter }
522 1.16 peter
523 1.16 peter /*
524 1.16 peter * remove a discipline or a queue
525 1.16 peter */
526 1.16 peter int
527 1.16 peter altq_remove(struct pf_altq *a)
528 1.16 peter {
529 1.16 peter int error = 0;
530 1.16 peter
531 1.16 peter if (a->qname[0] != 0)
532 1.16 peter return (altq_remove_queue(a));
533 1.16 peter
534 1.16 peter switch (a->scheduler) {
535 1.16 peter #ifdef ALTQ_CBQ
536 1.16 peter case ALTQT_CBQ:
537 1.16 peter error = cbq_remove_altq(a);
538 1.16 peter break;
539 1.16 peter #endif
540 1.16 peter #ifdef ALTQ_PRIQ
541 1.16 peter case ALTQT_PRIQ:
542 1.16 peter error = priq_remove_altq(a);
543 1.16 peter break;
544 1.16 peter #endif
545 1.16 peter #ifdef ALTQ_HFSC
546 1.16 peter case ALTQT_HFSC:
547 1.16 peter error = hfsc_remove_altq(a);
548 1.16 peter break;
549 1.16 peter #endif
550 1.16 peter default:
551 1.16 peter error = ENXIO;
552 1.16 peter }
553 1.16 peter
554 1.16 peter return (error);
555 1.16 peter }
556 1.16 peter
557 1.16 peter /*
558 1.16 peter * add a queue to the discipline
559 1.16 peter */
560 1.16 peter int
561 1.16 peter altq_add_queue(struct pf_altq *a)
562 1.16 peter {
563 1.16 peter int error = 0;
564 1.16 peter
565 1.16 peter switch (a->scheduler) {
566 1.16 peter #ifdef ALTQ_CBQ
567 1.16 peter case ALTQT_CBQ:
568 1.16 peter error = cbq_add_queue(a);
569 1.16 peter break;
570 1.16 peter #endif
571 1.16 peter #ifdef ALTQ_PRIQ
572 1.16 peter case ALTQT_PRIQ:
573 1.16 peter error = priq_add_queue(a);
574 1.16 peter break;
575 1.16 peter #endif
576 1.16 peter #ifdef ALTQ_HFSC
577 1.16 peter case ALTQT_HFSC:
578 1.16 peter error = hfsc_add_queue(a);
579 1.16 peter break;
580 1.16 peter #endif
581 1.16 peter default:
582 1.16 peter error = ENXIO;
583 1.16 peter }
584 1.16 peter
585 1.16 peter return (error);
586 1.16 peter }
587 1.16 peter
588 1.16 peter /*
589 1.16 peter * remove a queue from the discipline
590 1.16 peter */
591 1.16 peter int
592 1.16 peter altq_remove_queue(struct pf_altq *a)
593 1.16 peter {
594 1.16 peter int error = 0;
595 1.16 peter
596 1.16 peter switch (a->scheduler) {
597 1.16 peter #ifdef ALTQ_CBQ
598 1.16 peter case ALTQT_CBQ:
599 1.16 peter error = cbq_remove_queue(a);
600 1.16 peter break;
601 1.16 peter #endif
602 1.16 peter #ifdef ALTQ_PRIQ
603 1.16 peter case ALTQT_PRIQ:
604 1.16 peter error = priq_remove_queue(a);
605 1.16 peter break;
606 1.16 peter #endif
607 1.16 peter #ifdef ALTQ_HFSC
608 1.16 peter case ALTQT_HFSC:
609 1.16 peter error = hfsc_remove_queue(a);
610 1.16 peter break;
611 1.16 peter #endif
612 1.16 peter default:
613 1.16 peter error = ENXIO;
614 1.16 peter }
615 1.16 peter
616 1.16 peter return (error);
617 1.16 peter }
618 1.16 peter
619 1.16 peter /*
620 1.16 peter * get queue statistics
621 1.16 peter */
622 1.16 peter int
623 1.16 peter altq_getqstats(struct pf_altq *a, void *ubuf, int *nbytes)
624 1.16 peter {
625 1.16 peter int error = 0;
626 1.16 peter
627 1.16 peter switch (a->scheduler) {
628 1.16 peter #ifdef ALTQ_CBQ
629 1.16 peter case ALTQT_CBQ:
630 1.16 peter error = cbq_getqstats(a, ubuf, nbytes);
631 1.16 peter break;
632 1.16 peter #endif
633 1.16 peter #ifdef ALTQ_PRIQ
634 1.16 peter case ALTQT_PRIQ:
635 1.16 peter error = priq_getqstats(a, ubuf, nbytes);
636 1.16 peter break;
637 1.16 peter #endif
638 1.16 peter #ifdef ALTQ_HFSC
639 1.16 peter case ALTQT_HFSC:
640 1.16 peter error = hfsc_getqstats(a, ubuf, nbytes);
641 1.16 peter break;
642 1.16 peter #endif
643 1.16 peter default:
644 1.16 peter error = ENXIO;
645 1.16 peter }
646 1.16 peter
647 1.16 peter return (error);
648 1.16 peter }
649 1.17 peter #endif /* NPF > 0 */
650 1.16 peter
651 1.16 peter /*
652 1.16 peter * read and write diffserv field in IPv4 or IPv6 header
653 1.16 peter */
654 1.16 peter u_int8_t
655 1.16 peter read_dsfield(struct mbuf *m, struct altq_pktattr *pktattr)
656 1.16 peter {
657 1.16 peter struct mbuf *m0;
658 1.16 peter u_int8_t ds_field = 0;
659 1.16 peter
660 1.16 peter if (pktattr == NULL ||
661 1.16 peter (pktattr->pattr_af != AF_INET && pktattr->pattr_af != AF_INET6))
662 1.16 peter return ((u_int8_t)0);
663 1.16 peter
664 1.16 peter /* verify that pattr_hdr is within the mbuf data */
665 1.16 peter for (m0 = m; m0 != NULL; m0 = m0->m_next)
666 1.20 christos if (((char *)pktattr->pattr_hdr >= m0->m_data) &&
667 1.20 christos ((char *)pktattr->pattr_hdr < m0->m_data + m0->m_len))
668 1.16 peter break;
669 1.16 peter if (m0 == NULL) {
670 1.16 peter /* ick, pattr_hdr is stale */
671 1.16 peter pktattr->pattr_af = AF_UNSPEC;
672 1.16 peter #ifdef ALTQ_DEBUG
673 1.16 peter printf("read_dsfield: can't locate header!\n");
674 1.16 peter #endif
675 1.16 peter return ((u_int8_t)0);
676 1.16 peter }
677 1.16 peter
678 1.16 peter if (pktattr->pattr_af == AF_INET) {
679 1.16 peter struct ip *ip = (struct ip *)pktattr->pattr_hdr;
680 1.16 peter
681 1.16 peter if (ip->ip_v != 4)
682 1.16 peter return ((u_int8_t)0); /* version mismatch! */
683 1.16 peter ds_field = ip->ip_tos;
684 1.16 peter }
685 1.16 peter #ifdef INET6
686 1.16 peter else if (pktattr->pattr_af == AF_INET6) {
687 1.16 peter struct ip6_hdr *ip6 = (struct ip6_hdr *)pktattr->pattr_hdr;
688 1.16 peter u_int32_t flowlabel;
689 1.16 peter
690 1.16 peter flowlabel = ntohl(ip6->ip6_flow);
691 1.16 peter if ((flowlabel >> 28) != 6)
692 1.16 peter return ((u_int8_t)0); /* version mismatch! */
693 1.16 peter ds_field = (flowlabel >> 20) & 0xff;
694 1.16 peter }
695 1.16 peter #endif
696 1.16 peter return (ds_field);
697 1.16 peter }
698 1.16 peter
699 1.16 peter void
700 1.16 peter write_dsfield(struct mbuf *m, struct altq_pktattr *pktattr, u_int8_t dsfield)
701 1.16 peter {
702 1.16 peter struct mbuf *m0;
703 1.16 peter
704 1.16 peter if (pktattr == NULL ||
705 1.16 peter (pktattr->pattr_af != AF_INET && pktattr->pattr_af != AF_INET6))
706 1.16 peter return;
707 1.16 peter
708 1.16 peter /* verify that pattr_hdr is within the mbuf data */
709 1.16 peter for (m0 = m; m0 != NULL; m0 = m0->m_next)
710 1.20 christos if (((char *)pktattr->pattr_hdr >= m0->m_data) &&
711 1.20 christos ((char *)pktattr->pattr_hdr < m0->m_data + m0->m_len))
712 1.16 peter break;
713 1.16 peter if (m0 == NULL) {
714 1.16 peter /* ick, pattr_hdr is stale */
715 1.16 peter pktattr->pattr_af = AF_UNSPEC;
716 1.16 peter #ifdef ALTQ_DEBUG
717 1.16 peter printf("write_dsfield: can't locate header!\n");
718 1.16 peter #endif
719 1.16 peter return;
720 1.16 peter }
721 1.16 peter
722 1.16 peter if (pktattr->pattr_af == AF_INET) {
723 1.16 peter struct ip *ip = (struct ip *)pktattr->pattr_hdr;
724 1.16 peter u_int8_t old;
725 1.16 peter int32_t sum;
726 1.16 peter
727 1.16 peter if (ip->ip_v != 4)
728 1.16 peter return; /* version mismatch! */
729 1.16 peter old = ip->ip_tos;
730 1.16 peter dsfield |= old & 3; /* leave CU bits */
731 1.16 peter if (old == dsfield)
732 1.16 peter return;
733 1.16 peter ip->ip_tos = dsfield;
734 1.16 peter /*
735 1.16 peter * update checksum (from RFC1624)
736 1.16 peter * HC' = ~(~HC + ~m + m')
737 1.16 peter */
738 1.16 peter sum = ~ntohs(ip->ip_sum) & 0xffff;
739 1.16 peter sum += 0xff00 + (~old & 0xff) + dsfield;
740 1.16 peter sum = (sum >> 16) + (sum & 0xffff);
741 1.16 peter sum += (sum >> 16); /* add carry */
742 1.16 peter
743 1.16 peter ip->ip_sum = htons(~sum & 0xffff);
744 1.16 peter }
745 1.16 peter #ifdef INET6
746 1.16 peter else if (pktattr->pattr_af == AF_INET6) {
747 1.16 peter struct ip6_hdr *ip6 = (struct ip6_hdr *)pktattr->pattr_hdr;
748 1.16 peter u_int32_t flowlabel;
749 1.16 peter
750 1.16 peter flowlabel = ntohl(ip6->ip6_flow);
751 1.16 peter if ((flowlabel >> 28) != 6)
752 1.16 peter return; /* version mismatch! */
753 1.16 peter flowlabel = (flowlabel & 0xf03fffff) | (dsfield << 20);
754 1.16 peter ip6->ip6_flow = htonl(flowlabel);
755 1.16 peter }
756 1.16 peter #endif
757 1.16 peter return;
758 1.16 peter }
759 1.16 peter
760 1.16 peter
761 1.16 peter /*
762 1.16 peter * high resolution clock support taking advantage of a machine dependent
763 1.16 peter * high resolution time counter (e.g., timestamp counter of intel pentium).
764 1.16 peter * we assume
765 1.16 peter * - 64-bit-long monotonically-increasing counter
766 1.16 peter * - frequency range is 100M-4GHz (CPU speed)
767 1.16 peter */
768 1.16 peter /* if pcc is not available or disabled, emulate 256MHz using microtime() */
769 1.16 peter #define MACHCLK_SHIFT 8
770 1.16 peter
771 1.16 peter int machclk_usepcc;
772 1.16 peter u_int32_t machclk_freq = 0;
773 1.16 peter u_int32_t machclk_per_tick = 0;
774 1.16 peter
775 1.16 peter #ifdef __alpha__
776 1.16 peter #ifdef __FreeBSD__
777 1.16 peter extern u_int32_t cycles_per_sec; /* alpha cpu clock frequency */
778 1.16 peter #elif defined(__NetBSD__) || defined(__OpenBSD__)
779 1.16 peter extern u_int64_t cycles_per_usec; /* alpha cpu clock frequency */
780 1.16 peter #endif
781 1.16 peter #endif /* __alpha__ */
782 1.16 peter
783 1.16 peter void
784 1.16 peter init_machclk(void)
785 1.16 peter {
786 1.23 pooka
787 1.23 pooka callout_init(&tbr_callout, 0);
788 1.23 pooka
789 1.16 peter machclk_usepcc = 1;
790 1.16 peter
791 1.16 peter #if (!defined(__i386__) && !defined(__alpha__)) || defined(ALTQ_NOPCC)
792 1.16 peter machclk_usepcc = 0;
793 1.16 peter #endif
794 1.16 peter #if defined(__FreeBSD__) && defined(SMP)
795 1.16 peter machclk_usepcc = 0;
796 1.16 peter #endif
797 1.16 peter #if defined(__NetBSD__) && defined(MULTIPROCESSOR)
798 1.16 peter machclk_usepcc = 0;
799 1.16 peter #endif
800 1.16 peter #ifdef __i386__
801 1.16 peter /* check if TSC is available */
802 1.16 peter if (machclk_usepcc == 1 && (cpu_feature & CPUID_TSC) == 0)
803 1.16 peter machclk_usepcc = 0;
804 1.16 peter #endif
805 1.16 peter
806 1.16 peter if (machclk_usepcc == 0) {
807 1.16 peter /* emulate 256MHz using microtime() */
808 1.16 peter machclk_freq = 1000000 << MACHCLK_SHIFT;
809 1.16 peter machclk_per_tick = machclk_freq / hz;
810 1.16 peter #ifdef ALTQ_DEBUG
811 1.16 peter printf("altq: emulate %uHz CPU clock\n", machclk_freq);
812 1.16 peter #endif
813 1.16 peter return;
814 1.16 peter }
815 1.16 peter
816 1.16 peter /*
817 1.16 peter * if the clock frequency (of Pentium TSC or Alpha PCC) is
818 1.16 peter * accessible, just use it.
819 1.16 peter */
820 1.16 peter #ifdef __i386__
821 1.16 peter #ifdef __FreeBSD__
822 1.16 peter #if (__FreeBSD_version > 300000)
823 1.16 peter machclk_freq = tsc_freq;
824 1.16 peter #else
825 1.16 peter machclk_freq = i586_ctr_freq;
826 1.16 peter #endif
827 1.16 peter #elif defined(__NetBSD__)
828 1.16 peter machclk_freq = (u_int32_t)curcpu()->ci_tsc_freq;
829 1.16 peter #elif defined(__OpenBSD__) && (defined(I586_CPU) || defined(I686_CPU))
830 1.16 peter machclk_freq = pentium_mhz * 1000000;
831 1.16 peter #endif
832 1.16 peter #elif defined(__alpha__)
833 1.16 peter #ifdef __FreeBSD__
834 1.16 peter machclk_freq = cycles_per_sec;
835 1.16 peter #elif defined(__NetBSD__) || defined(__OpenBSD__)
836 1.16 peter machclk_freq = (u_int32_t)(cycles_per_usec * 1000000);
837 1.16 peter #endif
838 1.16 peter #endif /* __alpha__ */
839 1.16 peter
840 1.16 peter /*
841 1.16 peter * if we don't know the clock frequency, measure it.
842 1.16 peter */
843 1.16 peter if (machclk_freq == 0) {
844 1.16 peter static int wait;
845 1.16 peter struct timeval tv_start, tv_end;
846 1.16 peter u_int64_t start, end, diff;
847 1.16 peter int timo;
848 1.16 peter
849 1.16 peter microtime(&tv_start);
850 1.16 peter start = read_machclk();
851 1.16 peter timo = hz; /* 1 sec */
852 1.16 peter (void)tsleep(&wait, PWAIT | PCATCH, "init_machclk", timo);
853 1.16 peter microtime(&tv_end);
854 1.16 peter end = read_machclk();
855 1.16 peter diff = (u_int64_t)(tv_end.tv_sec - tv_start.tv_sec) * 1000000
856 1.16 peter + tv_end.tv_usec - tv_start.tv_usec;
857 1.16 peter if (diff != 0)
858 1.16 peter machclk_freq = (u_int)((end - start) * 1000000 / diff);
859 1.16 peter }
860 1.16 peter
861 1.16 peter machclk_per_tick = machclk_freq / hz;
862 1.16 peter
863 1.16 peter #ifdef ALTQ_DEBUG
864 1.16 peter printf("altq: CPU clock: %uHz\n", machclk_freq);
865 1.16 peter #endif
866 1.16 peter }
867 1.1 thorpej
868 1.16 peter #if defined(__OpenBSD__) && defined(__i386__)
869 1.16 peter static inline u_int64_t
870 1.16 peter rdtsc(void)
871 1.16 peter {
872 1.16 peter u_int64_t rv;
873 1.16 peter __asm __volatile(".byte 0x0f, 0x31" : "=A" (rv));
874 1.16 peter return (rv);
875 1.1 thorpej }
876 1.16 peter #endif /* __OpenBSD__ && __i386__ */
877 1.1 thorpej
878 1.16 peter u_int64_t
879 1.16 peter read_machclk(void)
880 1.1 thorpej {
881 1.16 peter u_int64_t val;
882 1.16 peter
883 1.16 peter if (machclk_usepcc) {
884 1.16 peter #if defined(__i386__)
885 1.16 peter val = rdtsc();
886 1.16 peter #elif defined(__alpha__)
887 1.16 peter static u_int32_t last_pcc, upper;
888 1.16 peter u_int32_t pcc;
889 1.1 thorpej
890 1.16 peter /*
891 1.16 peter * for alpha, make a 64bit counter value out of the 32bit
892 1.16 peter * alpha processor cycle counter.
893 1.16 peter * read_machclk must be called within a half of its
894 1.16 peter * wrap-around cycle (about 5 sec for 400MHz cpu) to properly
895 1.16 peter * detect a counter wrap-around.
896 1.16 peter * tbr_timeout calls read_machclk once a second.
897 1.16 peter */
898 1.16 peter pcc = (u_int32_t)alpha_rpcc();
899 1.16 peter if (pcc <= last_pcc)
900 1.16 peter upper++;
901 1.16 peter last_pcc = pcc;
902 1.16 peter val = ((u_int64_t)upper << 32) + pcc;
903 1.16 peter #else
904 1.16 peter panic("read_machclk");
905 1.16 peter #endif
906 1.1 thorpej } else {
907 1.16 peter struct timeval tv;
908 1.16 peter
909 1.16 peter microtime(&tv);
910 1.16 peter val = (((u_int64_t)(tv.tv_sec - boottime.tv_sec) * 1000000
911 1.16 peter + tv.tv_usec) << MACHCLK_SHIFT);
912 1.1 thorpej }
913 1.16 peter return (val);
914 1.1 thorpej }
915 1.1 thorpej
916 1.16 peter #ifdef ALTQ3_CLFIER_COMPAT
917 1.1 thorpej
918 1.1 thorpej #ifndef IPPROTO_ESP
919 1.1 thorpej #define IPPROTO_ESP 50 /* encapsulating security payload */
920 1.1 thorpej #endif
921 1.1 thorpej #ifndef IPPROTO_AH
922 1.1 thorpej #define IPPROTO_AH 51 /* authentication header */
923 1.1 thorpej #endif
924 1.1 thorpej
925 1.12 perry /*
926 1.1 thorpej * extract flow information from a given packet.
927 1.1 thorpej * filt_mask shows flowinfo fields required.
928 1.1 thorpej * we assume the ip header is in one mbuf, and addresses and ports are
929 1.1 thorpej * in network byte order.
930 1.1 thorpej */
931 1.12 perry int
932 1.16 peter altq_extractflow(struct mbuf *m, int af, struct flowinfo *flow,
933 1.16 peter u_int32_t filt_bmask)
934 1.1 thorpej {
935 1.1 thorpej
936 1.1 thorpej switch (af) {
937 1.1 thorpej case PF_INET: {
938 1.1 thorpej struct flowinfo_in *fin;
939 1.1 thorpej struct ip *ip;
940 1.1 thorpej
941 1.1 thorpej ip = mtod(m, struct ip *);
942 1.1 thorpej
943 1.1 thorpej if (ip->ip_v != 4)
944 1.1 thorpej break;
945 1.8 itojun
946 1.1 thorpej fin = (struct flowinfo_in *)flow;
947 1.1 thorpej fin->fi_len = sizeof(struct flowinfo_in);
948 1.1 thorpej fin->fi_family = AF_INET;
949 1.1 thorpej
950 1.1 thorpej fin->fi_proto = ip->ip_p;
951 1.1 thorpej fin->fi_tos = ip->ip_tos;
952 1.1 thorpej
953 1.1 thorpej fin->fi_src.s_addr = ip->ip_src.s_addr;
954 1.1 thorpej fin->fi_dst.s_addr = ip->ip_dst.s_addr;
955 1.8 itojun
956 1.1 thorpej if (filt_bmask & FIMB4_PORTS)
957 1.1 thorpej /* if port info is required, extract port numbers */
958 1.1 thorpej extract_ports4(m, ip, fin);
959 1.1 thorpej else {
960 1.1 thorpej fin->fi_sport = 0;
961 1.1 thorpej fin->fi_dport = 0;
962 1.1 thorpej fin->fi_gpi = 0;
963 1.1 thorpej }
964 1.1 thorpej return (1);
965 1.1 thorpej }
966 1.8 itojun
967 1.1 thorpej #ifdef INET6
968 1.1 thorpej case PF_INET6: {
969 1.1 thorpej struct flowinfo_in6 *fin6;
970 1.1 thorpej struct ip6_hdr *ip6;
971 1.1 thorpej
972 1.1 thorpej ip6 = mtod(m, struct ip6_hdr *);
973 1.1 thorpej /* should we check the ip version? */
974 1.8 itojun
975 1.1 thorpej fin6 = (struct flowinfo_in6 *)flow;
976 1.1 thorpej fin6->fi6_len = sizeof(struct flowinfo_in6);
977 1.1 thorpej fin6->fi6_family = AF_INET6;
978 1.1 thorpej
979 1.1 thorpej fin6->fi6_proto = ip6->ip6_nxt;
980 1.1 thorpej fin6->fi6_tclass = (ntohl(ip6->ip6_flow) >> 20) & 0xff;
981 1.1 thorpej
982 1.1 thorpej fin6->fi6_flowlabel = ip6->ip6_flow & htonl(0x000fffff);
983 1.1 thorpej fin6->fi6_src = ip6->ip6_src;
984 1.1 thorpej fin6->fi6_dst = ip6->ip6_dst;
985 1.1 thorpej
986 1.1 thorpej if ((filt_bmask & FIMB6_PORTS) ||
987 1.1 thorpej ((filt_bmask & FIMB6_PROTO)
988 1.1 thorpej && ip6->ip6_nxt > IPPROTO_IPV6))
989 1.1 thorpej /*
990 1.1 thorpej * if port info is required, or proto is required
991 1.1 thorpej * but there are option headers, extract port
992 1.1 thorpej * and protocol numbers.
993 1.1 thorpej */
994 1.1 thorpej extract_ports6(m, ip6, fin6);
995 1.1 thorpej else {
996 1.1 thorpej fin6->fi6_sport = 0;
997 1.1 thorpej fin6->fi6_dport = 0;
998 1.1 thorpej fin6->fi6_gpi = 0;
999 1.1 thorpej }
1000 1.1 thorpej return (1);
1001 1.1 thorpej }
1002 1.1 thorpej #endif /* INET6 */
1003 1.1 thorpej
1004 1.1 thorpej default:
1005 1.1 thorpej break;
1006 1.1 thorpej }
1007 1.1 thorpej
1008 1.1 thorpej /* failed */
1009 1.1 thorpej flow->fi_len = sizeof(struct flowinfo);
1010 1.1 thorpej flow->fi_family = AF_UNSPEC;
1011 1.1 thorpej return (0);
1012 1.1 thorpej }
1013 1.1 thorpej
1014 1.1 thorpej /*
1015 1.1 thorpej * helper routine to extract port numbers
1016 1.1 thorpej */
1017 1.1 thorpej /* structure for ipsec and ipv6 option header template */
1018 1.1 thorpej struct _opt6 {
1019 1.1 thorpej u_int8_t opt6_nxt; /* next header */
1020 1.1 thorpej u_int8_t opt6_hlen; /* header extension length */
1021 1.1 thorpej u_int16_t _pad;
1022 1.1 thorpej u_int32_t ah_spi; /* security parameter index
1023 1.1 thorpej for authentication header */
1024 1.1 thorpej };
1025 1.1 thorpej
1026 1.1 thorpej /*
1027 1.1 thorpej * extract port numbers from a ipv4 packet.
1028 1.1 thorpej */
1029 1.1 thorpej static int
1030 1.16 peter extract_ports4(struct mbuf *m, struct ip *ip, struct flowinfo_in *fin)
1031 1.1 thorpej {
1032 1.1 thorpej struct mbuf *m0;
1033 1.1 thorpej u_short ip_off;
1034 1.1 thorpej u_int8_t proto;
1035 1.1 thorpej int off;
1036 1.8 itojun
1037 1.1 thorpej fin->fi_sport = 0;
1038 1.1 thorpej fin->fi_dport = 0;
1039 1.1 thorpej fin->fi_gpi = 0;
1040 1.8 itojun
1041 1.1 thorpej ip_off = ntohs(ip->ip_off);
1042 1.1 thorpej /* if it is a fragment, try cached fragment info */
1043 1.1 thorpej if (ip_off & IP_OFFMASK) {
1044 1.1 thorpej ip4f_lookup(ip, fin);
1045 1.1 thorpej return (1);
1046 1.1 thorpej }
1047 1.1 thorpej
1048 1.1 thorpej /* locate the mbuf containing the protocol header */
1049 1.1 thorpej for (m0 = m; m0 != NULL; m0 = m0->m_next)
1050 1.20 christos if (((char *)ip >= m0->m_data) &&
1051 1.20 christos ((char *)ip < m0->m_data + m0->m_len))
1052 1.1 thorpej break;
1053 1.1 thorpej if (m0 == NULL) {
1054 1.1 thorpej #ifdef ALTQ_DEBUG
1055 1.1 thorpej printf("extract_ports4: can't locate header! ip=%p\n", ip);
1056 1.1 thorpej #endif
1057 1.1 thorpej return (0);
1058 1.1 thorpej }
1059 1.20 christos off = ((char *)ip - m0->m_data) + (ip->ip_hl << 2);
1060 1.1 thorpej proto = ip->ip_p;
1061 1.1 thorpej
1062 1.1 thorpej #ifdef ALTQ_IPSEC
1063 1.1 thorpej again:
1064 1.1 thorpej #endif
1065 1.1 thorpej while (off >= m0->m_len) {
1066 1.1 thorpej off -= m0->m_len;
1067 1.1 thorpej m0 = m0->m_next;
1068 1.8 itojun if (m0 == NULL)
1069 1.8 itojun return (0); /* bogus ip_hl! */
1070 1.1 thorpej }
1071 1.8 itojun if (m0->m_len < off + 4)
1072 1.8 itojun return (0);
1073 1.1 thorpej
1074 1.1 thorpej switch (proto) {
1075 1.1 thorpej case IPPROTO_TCP:
1076 1.1 thorpej case IPPROTO_UDP: {
1077 1.1 thorpej struct udphdr *udp;
1078 1.8 itojun
1079 1.20 christos udp = (struct udphdr *)(mtod(m0, char *) + off);
1080 1.1 thorpej fin->fi_sport = udp->uh_sport;
1081 1.1 thorpej fin->fi_dport = udp->uh_dport;
1082 1.1 thorpej fin->fi_proto = proto;
1083 1.1 thorpej }
1084 1.1 thorpej break;
1085 1.1 thorpej
1086 1.1 thorpej #ifdef ALTQ_IPSEC
1087 1.1 thorpej case IPPROTO_ESP:
1088 1.1 thorpej if (fin->fi_gpi == 0){
1089 1.1 thorpej u_int32_t *gpi;
1090 1.8 itojun
1091 1.21 he gpi = (u_int32_t *)(mtod(m0, char *) + off);
1092 1.1 thorpej fin->fi_gpi = *gpi;
1093 1.1 thorpej }
1094 1.1 thorpej fin->fi_proto = proto;
1095 1.1 thorpej break;
1096 1.1 thorpej
1097 1.1 thorpej case IPPROTO_AH: {
1098 1.1 thorpej /* get next header and header length */
1099 1.1 thorpej struct _opt6 *opt6;
1100 1.1 thorpej
1101 1.21 he opt6 = (struct _opt6 *)(mtod(m0, char *) + off);
1102 1.1 thorpej proto = opt6->opt6_nxt;
1103 1.1 thorpej off += 8 + (opt6->opt6_hlen * 4);
1104 1.8 itojun if (fin->fi_gpi == 0 && m0->m_len >= off + 8)
1105 1.1 thorpej fin->fi_gpi = opt6->ah_spi;
1106 1.1 thorpej }
1107 1.1 thorpej /* goto the next header */
1108 1.1 thorpej goto again;
1109 1.1 thorpej #endif /* ALTQ_IPSEC */
1110 1.1 thorpej
1111 1.1 thorpej default:
1112 1.1 thorpej fin->fi_proto = proto;
1113 1.1 thorpej return (0);
1114 1.1 thorpej }
1115 1.1 thorpej
1116 1.1 thorpej /* if this is a first fragment, cache it. */
1117 1.1 thorpej if (ip_off & IP_MF)
1118 1.1 thorpej ip4f_cache(ip, fin);
1119 1.1 thorpej
1120 1.1 thorpej return (1);
1121 1.1 thorpej }
1122 1.1 thorpej
1123 1.1 thorpej #ifdef INET6
1124 1.1 thorpej static int
1125 1.16 peter extract_ports6(struct mbuf *m, struct ip6_hdr *ip6, struct flowinfo_in6 *fin6)
1126 1.1 thorpej {
1127 1.1 thorpej struct mbuf *m0;
1128 1.1 thorpej int off;
1129 1.1 thorpej u_int8_t proto;
1130 1.8 itojun
1131 1.1 thorpej fin6->fi6_gpi = 0;
1132 1.1 thorpej fin6->fi6_sport = 0;
1133 1.1 thorpej fin6->fi6_dport = 0;
1134 1.8 itojun
1135 1.1 thorpej /* locate the mbuf containing the protocol header */
1136 1.1 thorpej for (m0 = m; m0 != NULL; m0 = m0->m_next)
1137 1.20 christos if (((char *)ip6 >= m0->m_data) &&
1138 1.20 christos ((char *)ip6 < m0->m_data + m0->m_len))
1139 1.1 thorpej break;
1140 1.1 thorpej if (m0 == NULL) {
1141 1.1 thorpej #ifdef ALTQ_DEBUG
1142 1.1 thorpej printf("extract_ports6: can't locate header! ip6=%p\n", ip6);
1143 1.1 thorpej #endif
1144 1.1 thorpej return (0);
1145 1.1 thorpej }
1146 1.20 christos off = ((char *)ip6 - m0->m_data) + sizeof(struct ip6_hdr);
1147 1.1 thorpej
1148 1.1 thorpej proto = ip6->ip6_nxt;
1149 1.1 thorpej do {
1150 1.1 thorpej while (off >= m0->m_len) {
1151 1.1 thorpej off -= m0->m_len;
1152 1.1 thorpej m0 = m0->m_next;
1153 1.8 itojun if (m0 == NULL)
1154 1.8 itojun return (0);
1155 1.1 thorpej }
1156 1.8 itojun if (m0->m_len < off + 4)
1157 1.8 itojun return (0);
1158 1.1 thorpej
1159 1.1 thorpej switch (proto) {
1160 1.1 thorpej case IPPROTO_TCP:
1161 1.1 thorpej case IPPROTO_UDP: {
1162 1.1 thorpej struct udphdr *udp;
1163 1.8 itojun
1164 1.20 christos udp = (struct udphdr *)(mtod(m0, char *) + off);
1165 1.1 thorpej fin6->fi6_sport = udp->uh_sport;
1166 1.1 thorpej fin6->fi6_dport = udp->uh_dport;
1167 1.1 thorpej fin6->fi6_proto = proto;
1168 1.1 thorpej }
1169 1.1 thorpej return (1);
1170 1.8 itojun
1171 1.1 thorpej case IPPROTO_ESP:
1172 1.1 thorpej if (fin6->fi6_gpi == 0) {
1173 1.1 thorpej u_int32_t *gpi;
1174 1.8 itojun
1175 1.20 christos gpi = (u_int32_t *)(mtod(m0, char *) + off);
1176 1.1 thorpej fin6->fi6_gpi = *gpi;
1177 1.1 thorpej }
1178 1.1 thorpej fin6->fi6_proto = proto;
1179 1.1 thorpej return (1);
1180 1.1 thorpej
1181 1.1 thorpej case IPPROTO_AH: {
1182 1.1 thorpej /* get next header and header length */
1183 1.1 thorpej struct _opt6 *opt6;
1184 1.1 thorpej
1185 1.20 christos opt6 = (struct _opt6 *)(mtod(m0, char *) + off);
1186 1.8 itojun if (fin6->fi6_gpi == 0 && m0->m_len >= off + 8)
1187 1.1 thorpej fin6->fi6_gpi = opt6->ah_spi;
1188 1.1 thorpej proto = opt6->opt6_nxt;
1189 1.1 thorpej off += 8 + (opt6->opt6_hlen * 4);
1190 1.1 thorpej /* goto the next header */
1191 1.1 thorpej break;
1192 1.1 thorpej }
1193 1.1 thorpej
1194 1.1 thorpej case IPPROTO_HOPOPTS:
1195 1.1 thorpej case IPPROTO_ROUTING:
1196 1.1 thorpej case IPPROTO_DSTOPTS: {
1197 1.1 thorpej /* get next header and header length */
1198 1.1 thorpej struct _opt6 *opt6;
1199 1.1 thorpej
1200 1.20 christos opt6 = (struct _opt6 *)(mtod(m0, char *) + off);
1201 1.1 thorpej proto = opt6->opt6_nxt;
1202 1.1 thorpej off += (opt6->opt6_hlen + 1) * 8;
1203 1.1 thorpej /* goto the next header */
1204 1.1 thorpej break;
1205 1.1 thorpej }
1206 1.8 itojun
1207 1.1 thorpej case IPPROTO_FRAGMENT:
1208 1.1 thorpej /* ipv6 fragmentations are not supported yet */
1209 1.1 thorpej default:
1210 1.1 thorpej fin6->fi6_proto = proto;
1211 1.1 thorpej return (0);
1212 1.1 thorpej }
1213 1.1 thorpej } while (1);
1214 1.1 thorpej /*NOTREACHED*/
1215 1.1 thorpej }
1216 1.1 thorpej #endif /* INET6 */
1217 1.1 thorpej
1218 1.1 thorpej /*
1219 1.1 thorpej * altq common classifier
1220 1.1 thorpej */
1221 1.1 thorpej int
1222 1.16 peter acc_add_filter(struct acc_classifier *classifier, struct flow_filter *filter,
1223 1.16 peter void *class, u_long *phandle)
1224 1.1 thorpej {
1225 1.1 thorpej struct acc_filter *afp, *prev, *tmp;
1226 1.1 thorpej int i, s;
1227 1.1 thorpej
1228 1.1 thorpej #ifdef INET6
1229 1.1 thorpej if (filter->ff_flow.fi_family != AF_INET &&
1230 1.1 thorpej filter->ff_flow.fi_family != AF_INET6)
1231 1.1 thorpej return (EINVAL);
1232 1.1 thorpej #else
1233 1.1 thorpej if (filter->ff_flow.fi_family != AF_INET)
1234 1.1 thorpej return (EINVAL);
1235 1.1 thorpej #endif
1236 1.8 itojun
1237 1.14 christos afp = malloc(sizeof(struct acc_filter), M_DEVBUF, M_WAITOK|M_ZERO);
1238 1.1 thorpej if (afp == NULL)
1239 1.1 thorpej return (ENOMEM);
1240 1.1 thorpej
1241 1.1 thorpej afp->f_filter = *filter;
1242 1.1 thorpej afp->f_class = class;
1243 1.1 thorpej
1244 1.1 thorpej i = ACC_WILDCARD_INDEX;
1245 1.1 thorpej if (filter->ff_flow.fi_family == AF_INET) {
1246 1.1 thorpej struct flow_filter *filter4 = &afp->f_filter;
1247 1.8 itojun
1248 1.1 thorpej /*
1249 1.1 thorpej * if address is 0, it's a wildcard. if address mask
1250 1.1 thorpej * isn't set, use full mask.
1251 1.1 thorpej */
1252 1.1 thorpej if (filter4->ff_flow.fi_dst.s_addr == 0)
1253 1.1 thorpej filter4->ff_mask.mask_dst.s_addr = 0;
1254 1.1 thorpej else if (filter4->ff_mask.mask_dst.s_addr == 0)
1255 1.1 thorpej filter4->ff_mask.mask_dst.s_addr = 0xffffffff;
1256 1.1 thorpej if (filter4->ff_flow.fi_src.s_addr == 0)
1257 1.1 thorpej filter4->ff_mask.mask_src.s_addr = 0;
1258 1.1 thorpej else if (filter4->ff_mask.mask_src.s_addr == 0)
1259 1.1 thorpej filter4->ff_mask.mask_src.s_addr = 0xffffffff;
1260 1.1 thorpej
1261 1.1 thorpej /* clear extra bits in addresses */
1262 1.1 thorpej filter4->ff_flow.fi_dst.s_addr &=
1263 1.1 thorpej filter4->ff_mask.mask_dst.s_addr;
1264 1.1 thorpej filter4->ff_flow.fi_src.s_addr &=
1265 1.1 thorpej filter4->ff_mask.mask_src.s_addr;
1266 1.1 thorpej
1267 1.1 thorpej /*
1268 1.1 thorpej * if dst address is a wildcard, use hash-entry
1269 1.1 thorpej * ACC_WILDCARD_INDEX.
1270 1.1 thorpej */
1271 1.1 thorpej if (filter4->ff_mask.mask_dst.s_addr != 0xffffffff)
1272 1.1 thorpej i = ACC_WILDCARD_INDEX;
1273 1.1 thorpej else
1274 1.1 thorpej i = ACC_GET_HASH_INDEX(filter4->ff_flow.fi_dst.s_addr);
1275 1.1 thorpej }
1276 1.1 thorpej #ifdef INET6
1277 1.1 thorpej else if (filter->ff_flow.fi_family == AF_INET6) {
1278 1.1 thorpej struct flow_filter6 *filter6 =
1279 1.1 thorpej (struct flow_filter6 *)&afp->f_filter;
1280 1.1 thorpej #ifndef IN6MASK0 /* taken from kame ipv6 */
1281 1.1 thorpej #define IN6MASK0 {{{ 0, 0, 0, 0 }}}
1282 1.1 thorpej #define IN6MASK128 {{{ 0xffffffff, 0xffffffff, 0xffffffff, 0xffffffff }}}
1283 1.1 thorpej const struct in6_addr in6mask0 = IN6MASK0;
1284 1.1 thorpej const struct in6_addr in6mask128 = IN6MASK128;
1285 1.1 thorpej #endif
1286 1.1 thorpej
1287 1.1 thorpej if (IN6_IS_ADDR_UNSPECIFIED(&filter6->ff_flow6.fi6_dst))
1288 1.1 thorpej filter6->ff_mask6.mask6_dst = in6mask0;
1289 1.1 thorpej else if (IN6_IS_ADDR_UNSPECIFIED(&filter6->ff_mask6.mask6_dst))
1290 1.1 thorpej filter6->ff_mask6.mask6_dst = in6mask128;
1291 1.1 thorpej if (IN6_IS_ADDR_UNSPECIFIED(&filter6->ff_flow6.fi6_src))
1292 1.1 thorpej filter6->ff_mask6.mask6_src = in6mask0;
1293 1.1 thorpej else if (IN6_IS_ADDR_UNSPECIFIED(&filter6->ff_mask6.mask6_src))
1294 1.1 thorpej filter6->ff_mask6.mask6_src = in6mask128;
1295 1.1 thorpej
1296 1.1 thorpej /* clear extra bits in addresses */
1297 1.1 thorpej for (i = 0; i < 16; i++)
1298 1.1 thorpej filter6->ff_flow6.fi6_dst.s6_addr[i] &=
1299 1.1 thorpej filter6->ff_mask6.mask6_dst.s6_addr[i];
1300 1.1 thorpej for (i = 0; i < 16; i++)
1301 1.1 thorpej filter6->ff_flow6.fi6_src.s6_addr[i] &=
1302 1.1 thorpej filter6->ff_mask6.mask6_src.s6_addr[i];
1303 1.8 itojun
1304 1.1 thorpej if (filter6->ff_flow6.fi6_flowlabel == 0)
1305 1.1 thorpej i = ACC_WILDCARD_INDEX;
1306 1.1 thorpej else
1307 1.1 thorpej i = ACC_GET_HASH_INDEX(filter6->ff_flow6.fi6_flowlabel);
1308 1.1 thorpej }
1309 1.1 thorpej #endif /* INET6 */
1310 1.1 thorpej
1311 1.1 thorpej afp->f_handle = get_filt_handle(classifier, i);
1312 1.1 thorpej
1313 1.1 thorpej /* update filter bitmask */
1314 1.1 thorpej afp->f_fbmask = filt2fibmask(filter);
1315 1.1 thorpej classifier->acc_fbmask |= afp->f_fbmask;
1316 1.1 thorpej
1317 1.1 thorpej /*
1318 1.1 thorpej * add this filter to the filter list.
1319 1.1 thorpej * filters are ordered from the highest rule number.
1320 1.1 thorpej */
1321 1.5 thorpej s = splnet();
1322 1.1 thorpej prev = NULL;
1323 1.1 thorpej LIST_FOREACH(tmp, &classifier->acc_filters[i], f_chain) {
1324 1.1 thorpej if (tmp->f_filter.ff_ruleno > afp->f_filter.ff_ruleno)
1325 1.1 thorpej prev = tmp;
1326 1.1 thorpej else
1327 1.1 thorpej break;
1328 1.1 thorpej }
1329 1.1 thorpej if (prev == NULL)
1330 1.1 thorpej LIST_INSERT_HEAD(&classifier->acc_filters[i], afp, f_chain);
1331 1.1 thorpej else
1332 1.1 thorpej LIST_INSERT_AFTER(prev, afp, f_chain);
1333 1.1 thorpej splx(s);
1334 1.1 thorpej
1335 1.1 thorpej *phandle = afp->f_handle;
1336 1.1 thorpej return (0);
1337 1.1 thorpej }
1338 1.1 thorpej
1339 1.1 thorpej int
1340 1.16 peter acc_delete_filter(struct acc_classifier *classifier, u_long handle)
1341 1.1 thorpej {
1342 1.1 thorpej struct acc_filter *afp;
1343 1.1 thorpej int s;
1344 1.1 thorpej
1345 1.1 thorpej if ((afp = filth_to_filtp(classifier, handle)) == NULL)
1346 1.1 thorpej return (EINVAL);
1347 1.1 thorpej
1348 1.5 thorpej s = splnet();
1349 1.1 thorpej LIST_REMOVE(afp, f_chain);
1350 1.1 thorpej splx(s);
1351 1.1 thorpej
1352 1.14 christos free(afp, M_DEVBUF);
1353 1.1 thorpej
1354 1.1 thorpej /* todo: update filt_bmask */
1355 1.1 thorpej
1356 1.1 thorpej return (0);
1357 1.1 thorpej }
1358 1.1 thorpej
1359 1.1 thorpej /*
1360 1.1 thorpej * delete filters referencing to the specified class.
1361 1.1 thorpej * if the all flag is not 0, delete all the filters.
1362 1.1 thorpej */
1363 1.1 thorpej int
1364 1.16 peter acc_discard_filters(struct acc_classifier *classifier, void *class, int all)
1365 1.1 thorpej {
1366 1.1 thorpej struct acc_filter *afp;
1367 1.1 thorpej int i, s;
1368 1.1 thorpej
1369 1.5 thorpej s = splnet();
1370 1.1 thorpej for (i = 0; i < ACC_FILTER_TABLESIZE; i++) {
1371 1.1 thorpej do {
1372 1.1 thorpej LIST_FOREACH(afp, &classifier->acc_filters[i], f_chain)
1373 1.1 thorpej if (all || afp->f_class == class) {
1374 1.1 thorpej LIST_REMOVE(afp, f_chain);
1375 1.14 christos free(afp, M_DEVBUF);
1376 1.1 thorpej /* start again from the head */
1377 1.1 thorpej break;
1378 1.1 thorpej }
1379 1.1 thorpej } while (afp != NULL);
1380 1.1 thorpej }
1381 1.1 thorpej splx(s);
1382 1.1 thorpej
1383 1.1 thorpej if (all)
1384 1.1 thorpej classifier->acc_fbmask = 0;
1385 1.1 thorpej
1386 1.1 thorpej return (0);
1387 1.1 thorpej }
1388 1.1 thorpej
1389 1.1 thorpej void *
1390 1.16 peter acc_classify(void *clfier, struct mbuf *m, int af)
1391 1.1 thorpej {
1392 1.1 thorpej struct acc_classifier *classifier;
1393 1.1 thorpej struct flowinfo flow;
1394 1.1 thorpej struct acc_filter *afp;
1395 1.1 thorpej int i;
1396 1.1 thorpej
1397 1.1 thorpej classifier = (struct acc_classifier *)clfier;
1398 1.1 thorpej altq_extractflow(m, af, &flow, classifier->acc_fbmask);
1399 1.1 thorpej
1400 1.1 thorpej if (flow.fi_family == AF_INET) {
1401 1.1 thorpej struct flowinfo_in *fp = (struct flowinfo_in *)&flow;
1402 1.8 itojun
1403 1.1 thorpej if ((classifier->acc_fbmask & FIMB4_ALL) == FIMB4_TOS) {
1404 1.1 thorpej /* only tos is used */
1405 1.1 thorpej LIST_FOREACH(afp,
1406 1.1 thorpej &classifier->acc_filters[ACC_WILDCARD_INDEX],
1407 1.1 thorpej f_chain)
1408 1.1 thorpej if (apply_tosfilter4(afp->f_fbmask,
1409 1.1 thorpej &afp->f_filter, fp))
1410 1.1 thorpej /* filter matched */
1411 1.1 thorpej return (afp->f_class);
1412 1.1 thorpej } else if ((classifier->acc_fbmask &
1413 1.1 thorpej (~(FIMB4_PROTO|FIMB4_SPORT|FIMB4_DPORT) & FIMB4_ALL))
1414 1.1 thorpej == 0) {
1415 1.1 thorpej /* only proto and ports are used */
1416 1.1 thorpej LIST_FOREACH(afp,
1417 1.1 thorpej &classifier->acc_filters[ACC_WILDCARD_INDEX],
1418 1.1 thorpej f_chain)
1419 1.1 thorpej if (apply_ppfilter4(afp->f_fbmask,
1420 1.1 thorpej &afp->f_filter, fp))
1421 1.1 thorpej /* filter matched */
1422 1.1 thorpej return (afp->f_class);
1423 1.1 thorpej } else {
1424 1.1 thorpej /* get the filter hash entry from its dest address */
1425 1.1 thorpej i = ACC_GET_HASH_INDEX(fp->fi_dst.s_addr);
1426 1.1 thorpej do {
1427 1.1 thorpej /*
1428 1.1 thorpej * go through this loop twice. first for dst
1429 1.1 thorpej * hash, second for wildcards.
1430 1.1 thorpej */
1431 1.1 thorpej LIST_FOREACH(afp, &classifier->acc_filters[i],
1432 1.1 thorpej f_chain)
1433 1.1 thorpej if (apply_filter4(afp->f_fbmask,
1434 1.1 thorpej &afp->f_filter, fp))
1435 1.1 thorpej /* filter matched */
1436 1.1 thorpej return (afp->f_class);
1437 1.8 itojun
1438 1.1 thorpej /*
1439 1.1 thorpej * check again for filters with a dst addr
1440 1.1 thorpej * wildcard.
1441 1.1 thorpej * (daddr == 0 || dmask != 0xffffffff).
1442 1.1 thorpej */
1443 1.1 thorpej if (i != ACC_WILDCARD_INDEX)
1444 1.1 thorpej i = ACC_WILDCARD_INDEX;
1445 1.1 thorpej else
1446 1.1 thorpej break;
1447 1.1 thorpej } while (1);
1448 1.1 thorpej }
1449 1.1 thorpej }
1450 1.1 thorpej #ifdef INET6
1451 1.1 thorpej else if (flow.fi_family == AF_INET6) {
1452 1.1 thorpej struct flowinfo_in6 *fp6 = (struct flowinfo_in6 *)&flow;
1453 1.8 itojun
1454 1.1 thorpej /* get the filter hash entry from its flow ID */
1455 1.1 thorpej if (fp6->fi6_flowlabel != 0)
1456 1.1 thorpej i = ACC_GET_HASH_INDEX(fp6->fi6_flowlabel);
1457 1.1 thorpej else
1458 1.1 thorpej /* flowlable can be zero */
1459 1.1 thorpej i = ACC_WILDCARD_INDEX;
1460 1.1 thorpej
1461 1.1 thorpej /* go through this loop twice. first for flow hash, second
1462 1.1 thorpej for wildcards. */
1463 1.1 thorpej do {
1464 1.1 thorpej LIST_FOREACH(afp, &classifier->acc_filters[i], f_chain)
1465 1.1 thorpej if (apply_filter6(afp->f_fbmask,
1466 1.1 thorpej (struct flow_filter6 *)&afp->f_filter,
1467 1.1 thorpej fp6))
1468 1.1 thorpej /* filter matched */
1469 1.1 thorpej return (afp->f_class);
1470 1.1 thorpej
1471 1.1 thorpej /*
1472 1.1 thorpej * check again for filters with a wildcard.
1473 1.1 thorpej */
1474 1.1 thorpej if (i != ACC_WILDCARD_INDEX)
1475 1.1 thorpej i = ACC_WILDCARD_INDEX;
1476 1.1 thorpej else
1477 1.1 thorpej break;
1478 1.1 thorpej } while (1);
1479 1.1 thorpej }
1480 1.1 thorpej #endif /* INET6 */
1481 1.1 thorpej
1482 1.1 thorpej /* no filter matched */
1483 1.1 thorpej return (NULL);
1484 1.1 thorpej }
1485 1.1 thorpej
1486 1.1 thorpej static int
1487 1.16 peter apply_filter4(u_int32_t fbmask, struct flow_filter *filt,
1488 1.16 peter struct flowinfo_in *pkt)
1489 1.1 thorpej {
1490 1.1 thorpej if (filt->ff_flow.fi_family != AF_INET)
1491 1.1 thorpej return (0);
1492 1.1 thorpej if ((fbmask & FIMB4_SPORT) && filt->ff_flow.fi_sport != pkt->fi_sport)
1493 1.1 thorpej return (0);
1494 1.1 thorpej if ((fbmask & FIMB4_DPORT) && filt->ff_flow.fi_dport != pkt->fi_dport)
1495 1.1 thorpej return (0);
1496 1.1 thorpej if ((fbmask & FIMB4_DADDR) &&
1497 1.1 thorpej filt->ff_flow.fi_dst.s_addr !=
1498 1.1 thorpej (pkt->fi_dst.s_addr & filt->ff_mask.mask_dst.s_addr))
1499 1.1 thorpej return (0);
1500 1.1 thorpej if ((fbmask & FIMB4_SADDR) &&
1501 1.1 thorpej filt->ff_flow.fi_src.s_addr !=
1502 1.1 thorpej (pkt->fi_src.s_addr & filt->ff_mask.mask_src.s_addr))
1503 1.1 thorpej return (0);
1504 1.1 thorpej if ((fbmask & FIMB4_PROTO) && filt->ff_flow.fi_proto != pkt->fi_proto)
1505 1.1 thorpej return (0);
1506 1.1 thorpej if ((fbmask & FIMB4_TOS) && filt->ff_flow.fi_tos !=
1507 1.1 thorpej (pkt->fi_tos & filt->ff_mask.mask_tos))
1508 1.1 thorpej return (0);
1509 1.1 thorpej if ((fbmask & FIMB4_GPI) && filt->ff_flow.fi_gpi != (pkt->fi_gpi))
1510 1.1 thorpej return (0);
1511 1.1 thorpej /* match */
1512 1.1 thorpej return (1);
1513 1.1 thorpej }
1514 1.1 thorpej
1515 1.1 thorpej /*
1516 1.1 thorpej * filter matching function optimized for a common case that checks
1517 1.1 thorpej * only protocol and port numbers
1518 1.1 thorpej */
1519 1.1 thorpej static int
1520 1.16 peter apply_ppfilter4(u_int32_t fbmask, struct flow_filter *filt,
1521 1.16 peter struct flowinfo_in *pkt)
1522 1.1 thorpej {
1523 1.1 thorpej if (filt->ff_flow.fi_family != AF_INET)
1524 1.1 thorpej return (0);
1525 1.1 thorpej if ((fbmask & FIMB4_SPORT) && filt->ff_flow.fi_sport != pkt->fi_sport)
1526 1.1 thorpej return (0);
1527 1.1 thorpej if ((fbmask & FIMB4_DPORT) && filt->ff_flow.fi_dport != pkt->fi_dport)
1528 1.1 thorpej return (0);
1529 1.1 thorpej if ((fbmask & FIMB4_PROTO) && filt->ff_flow.fi_proto != pkt->fi_proto)
1530 1.1 thorpej return (0);
1531 1.1 thorpej /* match */
1532 1.1 thorpej return (1);
1533 1.1 thorpej }
1534 1.1 thorpej
1535 1.1 thorpej /*
1536 1.1 thorpej * filter matching function only for tos field.
1537 1.1 thorpej */
1538 1.1 thorpej static int
1539 1.16 peter apply_tosfilter4(u_int32_t fbmask, struct flow_filter *filt,
1540 1.16 peter struct flowinfo_in *pkt)
1541 1.1 thorpej {
1542 1.1 thorpej if (filt->ff_flow.fi_family != AF_INET)
1543 1.1 thorpej return (0);
1544 1.1 thorpej if ((fbmask & FIMB4_TOS) && filt->ff_flow.fi_tos !=
1545 1.1 thorpej (pkt->fi_tos & filt->ff_mask.mask_tos))
1546 1.1 thorpej return (0);
1547 1.1 thorpej /* match */
1548 1.1 thorpej return (1);
1549 1.1 thorpej }
1550 1.1 thorpej
1551 1.1 thorpej #ifdef INET6
1552 1.1 thorpej static int
1553 1.16 peter apply_filter6(u_int32_t fbmask, struct flow_filter6 *filt,
1554 1.16 peter struct flowinfo_in6 *pkt)
1555 1.1 thorpej {
1556 1.1 thorpej int i;
1557 1.1 thorpej
1558 1.1 thorpej if (filt->ff_flow6.fi6_family != AF_INET6)
1559 1.1 thorpej return (0);
1560 1.1 thorpej if ((fbmask & FIMB6_FLABEL) &&
1561 1.1 thorpej filt->ff_flow6.fi6_flowlabel != pkt->fi6_flowlabel)
1562 1.1 thorpej return (0);
1563 1.1 thorpej if ((fbmask & FIMB6_PROTO) &&
1564 1.1 thorpej filt->ff_flow6.fi6_proto != pkt->fi6_proto)
1565 1.1 thorpej return (0);
1566 1.1 thorpej if ((fbmask & FIMB6_SPORT) &&
1567 1.1 thorpej filt->ff_flow6.fi6_sport != pkt->fi6_sport)
1568 1.1 thorpej return (0);
1569 1.1 thorpej if ((fbmask & FIMB6_DPORT) &&
1570 1.1 thorpej filt->ff_flow6.fi6_dport != pkt->fi6_dport)
1571 1.1 thorpej return (0);
1572 1.1 thorpej if (fbmask & FIMB6_SADDR) {
1573 1.1 thorpej for (i = 0; i < 4; i++)
1574 1.1 thorpej if (filt->ff_flow6.fi6_src.s6_addr32[i] !=
1575 1.1 thorpej (pkt->fi6_src.s6_addr32[i] &
1576 1.1 thorpej filt->ff_mask6.mask6_src.s6_addr32[i]))
1577 1.1 thorpej return (0);
1578 1.1 thorpej }
1579 1.1 thorpej if (fbmask & FIMB6_DADDR) {
1580 1.1 thorpej for (i = 0; i < 4; i++)
1581 1.1 thorpej if (filt->ff_flow6.fi6_dst.s6_addr32[i] !=
1582 1.1 thorpej (pkt->fi6_dst.s6_addr32[i] &
1583 1.1 thorpej filt->ff_mask6.mask6_dst.s6_addr32[i]))
1584 1.1 thorpej return (0);
1585 1.1 thorpej }
1586 1.1 thorpej if ((fbmask & FIMB6_TCLASS) &&
1587 1.1 thorpej filt->ff_flow6.fi6_tclass !=
1588 1.1 thorpej (pkt->fi6_tclass & filt->ff_mask6.mask6_tclass))
1589 1.1 thorpej return (0);
1590 1.1 thorpej if ((fbmask & FIMB6_GPI) &&
1591 1.1 thorpej filt->ff_flow6.fi6_gpi != pkt->fi6_gpi)
1592 1.1 thorpej return (0);
1593 1.1 thorpej /* match */
1594 1.1 thorpej return (1);
1595 1.1 thorpej }
1596 1.1 thorpej #endif /* INET6 */
1597 1.1 thorpej
1598 1.1 thorpej /*
1599 1.1 thorpej * filter handle:
1600 1.1 thorpej * bit 20-28: index to the filter hash table
1601 1.1 thorpej * bit 0-19: unique id in the hash bucket.
1602 1.1 thorpej */
1603 1.1 thorpej static u_long
1604 1.16 peter get_filt_handle(struct acc_classifier *classifier, int i)
1605 1.1 thorpej {
1606 1.1 thorpej static u_long handle_number = 1;
1607 1.1 thorpej u_long handle;
1608 1.1 thorpej struct acc_filter *afp;
1609 1.1 thorpej
1610 1.1 thorpej while (1) {
1611 1.1 thorpej handle = handle_number++ & 0x000fffff;
1612 1.1 thorpej
1613 1.1 thorpej if (LIST_EMPTY(&classifier->acc_filters[i]))
1614 1.1 thorpej break;
1615 1.1 thorpej
1616 1.1 thorpej LIST_FOREACH(afp, &classifier->acc_filters[i], f_chain)
1617 1.1 thorpej if ((afp->f_handle & 0x000fffff) == handle)
1618 1.1 thorpej break;
1619 1.1 thorpej if (afp == NULL)
1620 1.1 thorpej break;
1621 1.1 thorpej /* this handle is already used, try again */
1622 1.1 thorpej }
1623 1.1 thorpej
1624 1.1 thorpej return ((i << 20) | handle);
1625 1.1 thorpej }
1626 1.1 thorpej
1627 1.1 thorpej /* convert filter handle to filter pointer */
1628 1.1 thorpej static struct acc_filter *
1629 1.16 peter filth_to_filtp(struct acc_classifier *classifier, u_long handle)
1630 1.1 thorpej {
1631 1.1 thorpej struct acc_filter *afp;
1632 1.1 thorpej int i;
1633 1.1 thorpej
1634 1.1 thorpej i = ACC_GET_HINDEX(handle);
1635 1.1 thorpej
1636 1.1 thorpej LIST_FOREACH(afp, &classifier->acc_filters[i], f_chain)
1637 1.1 thorpej if (afp->f_handle == handle)
1638 1.1 thorpej return (afp);
1639 1.1 thorpej
1640 1.1 thorpej return (NULL);
1641 1.1 thorpej }
1642 1.1 thorpej
1643 1.1 thorpej /* create flowinfo bitmask */
1644 1.1 thorpej static u_int32_t
1645 1.16 peter filt2fibmask(struct flow_filter *filt)
1646 1.1 thorpej {
1647 1.1 thorpej u_int32_t mask = 0;
1648 1.1 thorpej #ifdef INET6
1649 1.1 thorpej struct flow_filter6 *filt6;
1650 1.1 thorpej #endif
1651 1.1 thorpej
1652 1.1 thorpej switch (filt->ff_flow.fi_family) {
1653 1.1 thorpej case AF_INET:
1654 1.1 thorpej if (filt->ff_flow.fi_proto != 0)
1655 1.1 thorpej mask |= FIMB4_PROTO;
1656 1.1 thorpej if (filt->ff_flow.fi_tos != 0)
1657 1.1 thorpej mask |= FIMB4_TOS;
1658 1.1 thorpej if (filt->ff_flow.fi_dst.s_addr != 0)
1659 1.1 thorpej mask |= FIMB4_DADDR;
1660 1.1 thorpej if (filt->ff_flow.fi_src.s_addr != 0)
1661 1.1 thorpej mask |= FIMB4_SADDR;
1662 1.1 thorpej if (filt->ff_flow.fi_sport != 0)
1663 1.1 thorpej mask |= FIMB4_SPORT;
1664 1.1 thorpej if (filt->ff_flow.fi_dport != 0)
1665 1.1 thorpej mask |= FIMB4_DPORT;
1666 1.1 thorpej if (filt->ff_flow.fi_gpi != 0)
1667 1.1 thorpej mask |= FIMB4_GPI;
1668 1.1 thorpej break;
1669 1.1 thorpej #ifdef INET6
1670 1.1 thorpej case AF_INET6:
1671 1.1 thorpej filt6 = (struct flow_filter6 *)filt;
1672 1.1 thorpej
1673 1.1 thorpej if (filt6->ff_flow6.fi6_proto != 0)
1674 1.1 thorpej mask |= FIMB6_PROTO;
1675 1.1 thorpej if (filt6->ff_flow6.fi6_tclass != 0)
1676 1.1 thorpej mask |= FIMB6_TCLASS;
1677 1.1 thorpej if (!IN6_IS_ADDR_UNSPECIFIED(&filt6->ff_flow6.fi6_dst))
1678 1.1 thorpej mask |= FIMB6_DADDR;
1679 1.1 thorpej if (!IN6_IS_ADDR_UNSPECIFIED(&filt6->ff_flow6.fi6_src))
1680 1.1 thorpej mask |= FIMB6_SADDR;
1681 1.1 thorpej if (filt6->ff_flow6.fi6_sport != 0)
1682 1.1 thorpej mask |= FIMB6_SPORT;
1683 1.1 thorpej if (filt6->ff_flow6.fi6_dport != 0)
1684 1.1 thorpej mask |= FIMB6_DPORT;
1685 1.1 thorpej if (filt6->ff_flow6.fi6_gpi != 0)
1686 1.1 thorpej mask |= FIMB6_GPI;
1687 1.1 thorpej if (filt6->ff_flow6.fi6_flowlabel != 0)
1688 1.1 thorpej mask |= FIMB6_FLABEL;
1689 1.1 thorpej break;
1690 1.1 thorpej #endif /* INET6 */
1691 1.1 thorpej }
1692 1.1 thorpej return (mask);
1693 1.1 thorpej }
1694 1.1 thorpej
1695 1.1 thorpej
1696 1.1 thorpej /*
1697 1.1 thorpej * helper functions to handle IPv4 fragments.
1698 1.1 thorpej * currently only in-sequence fragments are handled.
1699 1.1 thorpej * - fragment info is cached in a LRU list.
1700 1.1 thorpej * - when a first fragment is found, cache its flow info.
1701 1.1 thorpej * - when a non-first fragment is found, lookup the cache.
1702 1.1 thorpej */
1703 1.1 thorpej
1704 1.1 thorpej struct ip4_frag {
1705 1.1 thorpej TAILQ_ENTRY(ip4_frag) ip4f_chain;
1706 1.1 thorpej char ip4f_valid;
1707 1.1 thorpej u_short ip4f_id;
1708 1.1 thorpej struct flowinfo_in ip4f_info;
1709 1.1 thorpej };
1710 1.1 thorpej
1711 1.1 thorpej static TAILQ_HEAD(ip4f_list, ip4_frag) ip4f_list; /* IPv4 fragment cache */
1712 1.1 thorpej
1713 1.1 thorpej #define IP4F_TABSIZE 16 /* IPv4 fragment cache size */
1714 1.1 thorpej
1715 1.1 thorpej
1716 1.1 thorpej static void
1717 1.16 peter ip4f_cache(struct ip *ip, struct flowinfo_in *fin)
1718 1.1 thorpej {
1719 1.1 thorpej struct ip4_frag *fp;
1720 1.1 thorpej
1721 1.1 thorpej if (TAILQ_EMPTY(&ip4f_list)) {
1722 1.1 thorpej /* first time call, allocate fragment cache entries. */
1723 1.1 thorpej if (ip4f_init() < 0)
1724 1.1 thorpej /* allocation failed! */
1725 1.1 thorpej return;
1726 1.1 thorpej }
1727 1.1 thorpej
1728 1.1 thorpej fp = ip4f_alloc();
1729 1.1 thorpej fp->ip4f_id = ip->ip_id;
1730 1.6 itojun fp->ip4f_info.fi_proto = ip->ip_p;
1731 1.6 itojun fp->ip4f_info.fi_src.s_addr = ip->ip_src.s_addr;
1732 1.6 itojun fp->ip4f_info.fi_dst.s_addr = ip->ip_dst.s_addr;
1733 1.1 thorpej
1734 1.1 thorpej /* save port numbers */
1735 1.1 thorpej fp->ip4f_info.fi_sport = fin->fi_sport;
1736 1.1 thorpej fp->ip4f_info.fi_dport = fin->fi_dport;
1737 1.1 thorpej fp->ip4f_info.fi_gpi = fin->fi_gpi;
1738 1.1 thorpej }
1739 1.1 thorpej
1740 1.1 thorpej static int
1741 1.16 peter ip4f_lookup(struct ip *ip, struct flowinfo_in *fin)
1742 1.1 thorpej {
1743 1.1 thorpej struct ip4_frag *fp;
1744 1.1 thorpej
1745 1.1 thorpej for (fp = TAILQ_FIRST(&ip4f_list); fp != NULL && fp->ip4f_valid;
1746 1.1 thorpej fp = TAILQ_NEXT(fp, ip4f_chain))
1747 1.1 thorpej if (ip->ip_id == fp->ip4f_id &&
1748 1.1 thorpej ip->ip_src.s_addr == fp->ip4f_info.fi_src.s_addr &&
1749 1.1 thorpej ip->ip_dst.s_addr == fp->ip4f_info.fi_dst.s_addr &&
1750 1.1 thorpej ip->ip_p == fp->ip4f_info.fi_proto) {
1751 1.1 thorpej
1752 1.1 thorpej /* found the matching entry */
1753 1.1 thorpej fin->fi_sport = fp->ip4f_info.fi_sport;
1754 1.1 thorpej fin->fi_dport = fp->ip4f_info.fi_dport;
1755 1.1 thorpej fin->fi_gpi = fp->ip4f_info.fi_gpi;
1756 1.1 thorpej
1757 1.1 thorpej if ((ntohs(ip->ip_off) & IP_MF) == 0)
1758 1.1 thorpej /* this is the last fragment,
1759 1.1 thorpej release the entry. */
1760 1.1 thorpej ip4f_free(fp);
1761 1.1 thorpej
1762 1.1 thorpej return (1);
1763 1.1 thorpej }
1764 1.1 thorpej
1765 1.1 thorpej /* no matching entry found */
1766 1.1 thorpej return (0);
1767 1.1 thorpej }
1768 1.1 thorpej
1769 1.1 thorpej static int
1770 1.1 thorpej ip4f_init(void)
1771 1.1 thorpej {
1772 1.1 thorpej struct ip4_frag *fp;
1773 1.1 thorpej int i;
1774 1.8 itojun
1775 1.1 thorpej TAILQ_INIT(&ip4f_list);
1776 1.1 thorpej for (i=0; i<IP4F_TABSIZE; i++) {
1777 1.14 christos fp = malloc(sizeof(struct ip4_frag), M_DEVBUF, M_NOWAIT);
1778 1.1 thorpej if (fp == NULL) {
1779 1.1 thorpej printf("ip4f_init: can't alloc %dth entry!\n", i);
1780 1.1 thorpej if (i == 0)
1781 1.1 thorpej return (-1);
1782 1.1 thorpej return (0);
1783 1.1 thorpej }
1784 1.1 thorpej fp->ip4f_valid = 0;
1785 1.1 thorpej TAILQ_INSERT_TAIL(&ip4f_list, fp, ip4f_chain);
1786 1.1 thorpej }
1787 1.1 thorpej return (0);
1788 1.1 thorpej }
1789 1.1 thorpej
1790 1.1 thorpej static struct ip4_frag *
1791 1.1 thorpej ip4f_alloc(void)
1792 1.1 thorpej {
1793 1.1 thorpej struct ip4_frag *fp;
1794 1.1 thorpej
1795 1.1 thorpej /* reclaim an entry at the tail, put it at the head */
1796 1.1 thorpej fp = TAILQ_LAST(&ip4f_list, ip4f_list);
1797 1.1 thorpej TAILQ_REMOVE(&ip4f_list, fp, ip4f_chain);
1798 1.1 thorpej fp->ip4f_valid = 1;
1799 1.1 thorpej TAILQ_INSERT_HEAD(&ip4f_list, fp, ip4f_chain);
1800 1.1 thorpej return (fp);
1801 1.1 thorpej }
1802 1.1 thorpej
1803 1.1 thorpej static void
1804 1.16 peter ip4f_free(struct ip4_frag *fp)
1805 1.1 thorpej {
1806 1.1 thorpej TAILQ_REMOVE(&ip4f_list, fp, ip4f_chain);
1807 1.1 thorpej fp->ip4f_valid = 0;
1808 1.1 thorpej TAILQ_INSERT_TAIL(&ip4f_list, fp, ip4f_chain);
1809 1.1 thorpej }
1810 1.1 thorpej
1811 1.16 peter #endif /* ALTQ3_CLFIER_COMPAT */
1812