Home | History | Annotate | Line # | Download | only in altq
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