Home | History | Annotate | Line # | Download | only in altq
altq_subr.c revision 1.12.4.4
      1  1.12.4.4     yamt /*	$NetBSD: altq_subr.c,v 1.12.4.4 2007/10/27 11:24:53 yamt Exp $	*/
      2  1.12.4.2     yamt /*	$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.12.4.2     yamt  * 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.12.4.4     yamt __KERNEL_RCSID(0, "$NetBSD: altq_subr.c,v 1.12.4.4 2007/10/27 11:24:53 yamt Exp $");
     32       1.1  thorpej 
     33  1.12.4.2     yamt #ifdef _KERNEL_OPT
     34       1.1  thorpej #include "opt_altq.h"
     35       1.1  thorpej #include "opt_inet.h"
     36  1.12.4.2     yamt #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.12.4.2     yamt #if NPF > 0
     66  1.12.4.2     yamt #include <net/pfvar.h>
     67  1.12.4.2     yamt #endif
     68       1.1  thorpej #include <altq/altq.h>
     69  1.12.4.2     yamt #ifdef ALTQ3_COMPAT
     70       1.1  thorpej #include <altq/altq_conf.h>
     71  1.12.4.2     yamt #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.12.4.2     yamt #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.12.4.4     yamt #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.12.4.2     yamt static void	tbr_timeout(void *);
     92  1.12.4.2     yamt int (*altq_input)(struct mbuf *, int) = NULL;
     93  1.12.4.2     yamt static int tbr_timer = 0;	/* token bucket regulator timer */
     94  1.12.4.3     yamt static struct callout tbr_callout;
     95  1.12.4.2     yamt 
     96  1.12.4.2     yamt #ifdef ALTQ3_CLFIER_COMPAT
     97  1.12.4.2     yamt static int 	extract_ports4(struct mbuf *, struct ip *, struct flowinfo_in *);
     98       1.1  thorpej #ifdef INET6
     99  1.12.4.2     yamt static int 	extract_ports6(struct mbuf *, struct ip6_hdr *,
    100  1.12.4.2     yamt 			       struct flowinfo_in6 *);
    101       1.1  thorpej #endif
    102  1.12.4.2     yamt static int	apply_filter4(u_int32_t, struct flow_filter *,
    103  1.12.4.2     yamt 			      struct flowinfo_in *);
    104  1.12.4.2     yamt static int	apply_ppfilter4(u_int32_t, struct flow_filter *,
    105  1.12.4.2     yamt 				struct flowinfo_in *);
    106       1.1  thorpej #ifdef INET6
    107  1.12.4.2     yamt static int	apply_filter6(u_int32_t, struct flow_filter6 *,
    108  1.12.4.2     yamt 			      struct flowinfo_in6 *);
    109       1.1  thorpej #endif
    110  1.12.4.2     yamt static int	apply_tosfilter4(u_int32_t, struct flow_filter *,
    111  1.12.4.2     yamt 				 struct flowinfo_in *);
    112  1.12.4.2     yamt static u_long	get_filt_handle(struct acc_classifier *, int);
    113  1.12.4.2     yamt static struct acc_filter *filth_to_filtp(struct acc_classifier *, u_long);
    114  1.12.4.2     yamt static u_int32_t filt2fibmask(struct flow_filter *);
    115  1.12.4.2     yamt 
    116  1.12.4.2     yamt static void 	ip4f_cache(struct ip *, struct flowinfo_in *);
    117  1.12.4.2     yamt static int 	ip4f_lookup(struct ip *, struct flowinfo_in *);
    118  1.12.4.2     yamt static int 	ip4f_init(void);
    119  1.12.4.2     yamt static struct ip4_frag	*ip4f_alloc(void);
    120  1.12.4.2     yamt static void 	ip4f_free(struct ip4_frag *);
    121  1.12.4.2     yamt #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.12.4.2     yamt /* look up the queue state by the interface name and the queueing type. */
    128       1.1  thorpej void *
    129  1.12.4.2     yamt 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.12.4.2     yamt altq_attach(struct ifaltq *ifq, int type, void *discipline,
    143  1.12.4.2     yamt     int (*enqueue)(struct ifaltq *, struct mbuf *, struct altq_pktattr *),
    144  1.12.4.2     yamt     struct mbuf *(*dequeue)(struct ifaltq *, int),
    145  1.12.4.2     yamt     int (*request)(struct ifaltq *, int, void *),
    146  1.12.4.2     yamt     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.12.4.2     yamt 
    151  1.12.4.2     yamt #ifdef ALTQ3_COMPAT
    152  1.12.4.2     yamt 	/*
    153  1.12.4.2     yamt 	 * pfaltq can override the existing discipline, but altq3 cannot.
    154  1.12.4.2     yamt 	 * check these if clfier is not NULL (which implies altq3).
    155  1.12.4.2     yamt 	 */
    156  1.12.4.2     yamt 	if (clfier != NULL) {
    157  1.12.4.2     yamt 		if (ALTQ_IS_ENABLED(ifq))
    158  1.12.4.2     yamt 			return EBUSY;
    159  1.12.4.2     yamt 		if (ALTQ_IS_ATTACHED(ifq))
    160  1.12.4.2     yamt 			return EEXIST;
    161  1.12.4.2     yamt 	}
    162  1.12.4.2     yamt #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.12.4.2     yamt 	ifq->altq_flags &= (ALTQF_CANTCHANGE|ALTQF_ENABLED);
    171  1.12.4.2     yamt #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.12.4.2     yamt #endif
    176       1.1  thorpej 	return 0;
    177       1.1  thorpej }
    178       1.1  thorpej 
    179       1.1  thorpej int
    180  1.12.4.2     yamt 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.12.4.2     yamt #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.12.4.2     yamt #endif
    193  1.12.4.2     yamt 
    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.12.4.2     yamt 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.12.4.2     yamt 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.12.4.2     yamt #ifdef ALTQ_DEBUG
    243       1.1  thorpej void
    244  1.12.4.2     yamt 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.12.4.2     yamt #endif
    252       1.1  thorpej 
    253       1.1  thorpej /*
    254       1.1  thorpej  * internal representation of token bucket parameters
    255  1.12.4.2     yamt  *	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.12.4.2     yamt 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.12.4.2     yamt 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.12.4.1     yamt 		free(tbr, M_DEVBUF);
    331       1.1  thorpej 		return (0);
    332       1.1  thorpej 	}
    333       1.1  thorpej 
    334  1.12.4.1     yamt 	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.12.4.1     yamt 		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.12.4.2     yamt 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.12.4.2     yamt 	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.1  thorpej 	}
    381       1.1  thorpej 	splx(s);
    382       1.1  thorpej 	if (active > 0)
    383       1.1  thorpej 		CALLOUT_RESET(&tbr_callout, 1, tbr_timeout, (void *)0);
    384       1.1  thorpej 	else
    385       1.1  thorpej 		tbr_timer = 0;	/* don't need tbr_timer anymore */
    386       1.1  thorpej #if defined(__alpha__) && !defined(ALTQ_NOPCC)
    387       1.1  thorpej 	{
    388       1.1  thorpej 		/*
    389       1.1  thorpej 		 * XXX read out the machine dependent clock once a second
    390       1.1  thorpej 		 * to detect counter wrap-around.
    391       1.1  thorpej 		 */
    392       1.1  thorpej 		static u_int cnt;
    393       1.1  thorpej 
    394       1.1  thorpej 		if (++cnt >= hz) {
    395       1.1  thorpej 			(void)read_machclk();
    396       1.1  thorpej 			cnt = 0;
    397       1.1  thorpej 		}
    398       1.1  thorpej 	}
    399  1.12.4.2     yamt #endif /* __alpha__ && !ALTQ_NOPCC */
    400  1.12.4.2     yamt }
    401  1.12.4.2     yamt 
    402  1.12.4.2     yamt /*
    403  1.12.4.2     yamt  * get token bucket regulator profile
    404  1.12.4.2     yamt  */
    405  1.12.4.2     yamt int
    406  1.12.4.2     yamt tbr_get(struct ifaltq *ifq, struct tb_profile *profile)
    407  1.12.4.2     yamt {
    408  1.12.4.2     yamt 	struct tb_regulator *tbr;
    409  1.12.4.2     yamt 
    410  1.12.4.2     yamt 	if ((tbr = ifq->altq_tbr) == NULL) {
    411  1.12.4.2     yamt 		profile->rate = 0;
    412  1.12.4.2     yamt 		profile->depth = 0;
    413  1.12.4.2     yamt 	} else {
    414  1.12.4.2     yamt 		profile->rate =
    415  1.12.4.2     yamt 		    (u_int)TBR_UNSCALE(tbr->tbr_rate * 8 * machclk_freq);
    416  1.12.4.2     yamt 		profile->depth = (u_int)TBR_UNSCALE(tbr->tbr_depth);
    417  1.12.4.2     yamt 	}
    418  1.12.4.2     yamt 	return (0);
    419  1.12.4.2     yamt }
    420  1.12.4.2     yamt 
    421  1.12.4.2     yamt #if NPF > 0
    422  1.12.4.2     yamt /*
    423  1.12.4.2     yamt  * attach a discipline to the interface.  if one already exists, it is
    424  1.12.4.2     yamt  * overridden.
    425  1.12.4.2     yamt  */
    426  1.12.4.2     yamt int
    427  1.12.4.2     yamt altq_pfattach(struct pf_altq *a)
    428  1.12.4.2     yamt {
    429  1.12.4.2     yamt 	int error = 0;
    430  1.12.4.2     yamt 
    431  1.12.4.2     yamt 	switch (a->scheduler) {
    432  1.12.4.2     yamt 	case ALTQT_NONE:
    433  1.12.4.2     yamt 		break;
    434  1.12.4.2     yamt #ifdef ALTQ_CBQ
    435  1.12.4.2     yamt 	case ALTQT_CBQ:
    436  1.12.4.2     yamt 		error = cbq_pfattach(a);
    437  1.12.4.2     yamt 		break;
    438  1.12.4.2     yamt #endif
    439  1.12.4.2     yamt #ifdef ALTQ_PRIQ
    440  1.12.4.2     yamt 	case ALTQT_PRIQ:
    441  1.12.4.2     yamt 		error = priq_pfattach(a);
    442  1.12.4.2     yamt 		break;
    443  1.12.4.2     yamt #endif
    444  1.12.4.2     yamt #ifdef ALTQ_HFSC
    445  1.12.4.2     yamt 	case ALTQT_HFSC:
    446  1.12.4.2     yamt 		error = hfsc_pfattach(a);
    447  1.12.4.2     yamt 		break;
    448  1.12.4.2     yamt #endif
    449  1.12.4.2     yamt 	default:
    450  1.12.4.2     yamt 		error = ENXIO;
    451  1.12.4.2     yamt 	}
    452  1.12.4.2     yamt 
    453  1.12.4.2     yamt 	return (error);
    454  1.12.4.2     yamt }
    455  1.12.4.2     yamt 
    456  1.12.4.2     yamt /*
    457  1.12.4.2     yamt  * detach a discipline from the interface.
    458  1.12.4.2     yamt  * it is possible that the discipline was already overridden by another
    459  1.12.4.2     yamt  * discipline.
    460  1.12.4.2     yamt  */
    461  1.12.4.2     yamt int
    462  1.12.4.2     yamt altq_pfdetach(struct pf_altq *a)
    463  1.12.4.2     yamt {
    464  1.12.4.2     yamt 	struct ifnet *ifp;
    465  1.12.4.2     yamt 	int s, error = 0;
    466  1.12.4.2     yamt 
    467  1.12.4.2     yamt 	if ((ifp = ifunit(a->ifname)) == NULL)
    468  1.12.4.2     yamt 		return (EINVAL);
    469  1.12.4.2     yamt 
    470  1.12.4.2     yamt 	/* if this discipline is no longer referenced, just return */
    471  1.12.4.2     yamt 	if (a->altq_disc == NULL || a->altq_disc != ifp->if_snd.altq_disc)
    472  1.12.4.2     yamt 		return (0);
    473  1.12.4.2     yamt 
    474  1.12.4.2     yamt 	s = splnet();
    475  1.12.4.2     yamt 	if (ALTQ_IS_ENABLED(&ifp->if_snd))
    476  1.12.4.2     yamt 		error = altq_disable(&ifp->if_snd);
    477  1.12.4.2     yamt 	if (error == 0)
    478  1.12.4.2     yamt 		error = altq_detach(&ifp->if_snd);
    479  1.12.4.2     yamt 	splx(s);
    480  1.12.4.2     yamt 
    481  1.12.4.2     yamt 	return (error);
    482  1.12.4.2     yamt }
    483  1.12.4.2     yamt 
    484  1.12.4.2     yamt /*
    485  1.12.4.2     yamt  * add a discipline or a queue
    486  1.12.4.2     yamt  */
    487  1.12.4.2     yamt int
    488  1.12.4.2     yamt altq_add(struct pf_altq *a)
    489  1.12.4.2     yamt {
    490  1.12.4.2     yamt 	int error = 0;
    491  1.12.4.2     yamt 
    492  1.12.4.2     yamt 	if (a->qname[0] != 0)
    493  1.12.4.2     yamt 		return (altq_add_queue(a));
    494  1.12.4.2     yamt 
    495  1.12.4.2     yamt 	if (machclk_freq == 0)
    496  1.12.4.2     yamt 		init_machclk();
    497  1.12.4.2     yamt 	if (machclk_freq == 0)
    498  1.12.4.2     yamt 		panic("altq_add: no CPU clock");
    499  1.12.4.2     yamt 
    500  1.12.4.2     yamt 	switch (a->scheduler) {
    501  1.12.4.2     yamt #ifdef ALTQ_CBQ
    502  1.12.4.2     yamt 	case ALTQT_CBQ:
    503  1.12.4.2     yamt 		error = cbq_add_altq(a);
    504  1.12.4.2     yamt 		break;
    505  1.12.4.2     yamt #endif
    506  1.12.4.2     yamt #ifdef ALTQ_PRIQ
    507  1.12.4.2     yamt 	case ALTQT_PRIQ:
    508  1.12.4.2     yamt 		error = priq_add_altq(a);
    509  1.12.4.2     yamt 		break;
    510  1.12.4.2     yamt #endif
    511  1.12.4.2     yamt #ifdef ALTQ_HFSC
    512  1.12.4.2     yamt 	case ALTQT_HFSC:
    513  1.12.4.2     yamt 		error = hfsc_add_altq(a);
    514  1.12.4.2     yamt 		break;
    515  1.12.4.2     yamt #endif
    516  1.12.4.2     yamt 	default:
    517  1.12.4.2     yamt 		error = ENXIO;
    518  1.12.4.2     yamt 	}
    519  1.12.4.2     yamt 
    520  1.12.4.2     yamt 	return (error);
    521  1.12.4.2     yamt }
    522  1.12.4.2     yamt 
    523  1.12.4.2     yamt /*
    524  1.12.4.2     yamt  * remove a discipline or a queue
    525  1.12.4.2     yamt  */
    526  1.12.4.2     yamt int
    527  1.12.4.2     yamt altq_remove(struct pf_altq *a)
    528  1.12.4.2     yamt {
    529  1.12.4.2     yamt 	int error = 0;
    530  1.12.4.2     yamt 
    531  1.12.4.2     yamt 	if (a->qname[0] != 0)
    532  1.12.4.2     yamt 		return (altq_remove_queue(a));
    533  1.12.4.2     yamt 
    534  1.12.4.2     yamt 	switch (a->scheduler) {
    535  1.12.4.2     yamt #ifdef ALTQ_CBQ
    536  1.12.4.2     yamt 	case ALTQT_CBQ:
    537  1.12.4.2     yamt 		error = cbq_remove_altq(a);
    538  1.12.4.2     yamt 		break;
    539  1.12.4.2     yamt #endif
    540  1.12.4.2     yamt #ifdef ALTQ_PRIQ
    541  1.12.4.2     yamt 	case ALTQT_PRIQ:
    542  1.12.4.2     yamt 		error = priq_remove_altq(a);
    543  1.12.4.2     yamt 		break;
    544  1.12.4.2     yamt #endif
    545  1.12.4.2     yamt #ifdef ALTQ_HFSC
    546  1.12.4.2     yamt 	case ALTQT_HFSC:
    547  1.12.4.2     yamt 		error = hfsc_remove_altq(a);
    548  1.12.4.2     yamt 		break;
    549  1.12.4.2     yamt #endif
    550  1.12.4.2     yamt 	default:
    551  1.12.4.2     yamt 		error = ENXIO;
    552  1.12.4.2     yamt 	}
    553  1.12.4.2     yamt 
    554  1.12.4.2     yamt 	return (error);
    555  1.12.4.2     yamt }
    556  1.12.4.2     yamt 
    557  1.12.4.2     yamt /*
    558  1.12.4.2     yamt  * add a queue to the discipline
    559  1.12.4.2     yamt  */
    560  1.12.4.2     yamt int
    561  1.12.4.2     yamt altq_add_queue(struct pf_altq *a)
    562  1.12.4.2     yamt {
    563  1.12.4.2     yamt 	int error = 0;
    564  1.12.4.2     yamt 
    565  1.12.4.2     yamt 	switch (a->scheduler) {
    566  1.12.4.2     yamt #ifdef ALTQ_CBQ
    567  1.12.4.2     yamt 	case ALTQT_CBQ:
    568  1.12.4.2     yamt 		error = cbq_add_queue(a);
    569  1.12.4.2     yamt 		break;
    570  1.12.4.2     yamt #endif
    571  1.12.4.2     yamt #ifdef ALTQ_PRIQ
    572  1.12.4.2     yamt 	case ALTQT_PRIQ:
    573  1.12.4.2     yamt 		error = priq_add_queue(a);
    574  1.12.4.2     yamt 		break;
    575  1.12.4.2     yamt #endif
    576  1.12.4.2     yamt #ifdef ALTQ_HFSC
    577  1.12.4.2     yamt 	case ALTQT_HFSC:
    578  1.12.4.2     yamt 		error = hfsc_add_queue(a);
    579  1.12.4.2     yamt 		break;
    580  1.12.4.2     yamt #endif
    581  1.12.4.2     yamt 	default:
    582  1.12.4.2     yamt 		error = ENXIO;
    583  1.12.4.2     yamt 	}
    584  1.12.4.2     yamt 
    585  1.12.4.2     yamt 	return (error);
    586  1.12.4.2     yamt }
    587  1.12.4.2     yamt 
    588  1.12.4.2     yamt /*
    589  1.12.4.2     yamt  * remove a queue from the discipline
    590  1.12.4.2     yamt  */
    591  1.12.4.2     yamt int
    592  1.12.4.2     yamt altq_remove_queue(struct pf_altq *a)
    593  1.12.4.2     yamt {
    594  1.12.4.2     yamt 	int error = 0;
    595  1.12.4.2     yamt 
    596  1.12.4.2     yamt 	switch (a->scheduler) {
    597  1.12.4.2     yamt #ifdef ALTQ_CBQ
    598  1.12.4.2     yamt 	case ALTQT_CBQ:
    599  1.12.4.2     yamt 		error = cbq_remove_queue(a);
    600  1.12.4.2     yamt 		break;
    601  1.12.4.2     yamt #endif
    602  1.12.4.2     yamt #ifdef ALTQ_PRIQ
    603  1.12.4.2     yamt 	case ALTQT_PRIQ:
    604  1.12.4.2     yamt 		error = priq_remove_queue(a);
    605  1.12.4.2     yamt 		break;
    606  1.12.4.2     yamt #endif
    607  1.12.4.2     yamt #ifdef ALTQ_HFSC
    608  1.12.4.2     yamt 	case ALTQT_HFSC:
    609  1.12.4.2     yamt 		error = hfsc_remove_queue(a);
    610  1.12.4.2     yamt 		break;
    611  1.12.4.2     yamt #endif
    612  1.12.4.2     yamt 	default:
    613  1.12.4.2     yamt 		error = ENXIO;
    614  1.12.4.2     yamt 	}
    615  1.12.4.2     yamt 
    616  1.12.4.2     yamt 	return (error);
    617  1.12.4.2     yamt }
    618  1.12.4.2     yamt 
    619  1.12.4.2     yamt /*
    620  1.12.4.2     yamt  * get queue statistics
    621  1.12.4.2     yamt  */
    622  1.12.4.2     yamt int
    623  1.12.4.2     yamt altq_getqstats(struct pf_altq *a, void *ubuf, int *nbytes)
    624  1.12.4.2     yamt {
    625  1.12.4.2     yamt 	int error = 0;
    626  1.12.4.2     yamt 
    627  1.12.4.2     yamt 	switch (a->scheduler) {
    628  1.12.4.2     yamt #ifdef ALTQ_CBQ
    629  1.12.4.2     yamt 	case ALTQT_CBQ:
    630  1.12.4.2     yamt 		error = cbq_getqstats(a, ubuf, nbytes);
    631  1.12.4.2     yamt 		break;
    632  1.12.4.2     yamt #endif
    633  1.12.4.2     yamt #ifdef ALTQ_PRIQ
    634  1.12.4.2     yamt 	case ALTQT_PRIQ:
    635  1.12.4.2     yamt 		error = priq_getqstats(a, ubuf, nbytes);
    636  1.12.4.2     yamt 		break;
    637  1.12.4.2     yamt #endif
    638  1.12.4.2     yamt #ifdef ALTQ_HFSC
    639  1.12.4.2     yamt 	case ALTQT_HFSC:
    640  1.12.4.2     yamt 		error = hfsc_getqstats(a, ubuf, nbytes);
    641  1.12.4.2     yamt 		break;
    642  1.12.4.2     yamt #endif
    643  1.12.4.2     yamt 	default:
    644  1.12.4.2     yamt 		error = ENXIO;
    645  1.12.4.2     yamt 	}
    646  1.12.4.2     yamt 
    647  1.12.4.2     yamt 	return (error);
    648  1.12.4.2     yamt }
    649  1.12.4.2     yamt #endif /* NPF > 0 */
    650  1.12.4.2     yamt 
    651  1.12.4.2     yamt /*
    652  1.12.4.2     yamt  * read and write diffserv field in IPv4 or IPv6 header
    653  1.12.4.2     yamt  */
    654  1.12.4.2     yamt u_int8_t
    655  1.12.4.2     yamt read_dsfield(struct mbuf *m, struct altq_pktattr *pktattr)
    656  1.12.4.2     yamt {
    657  1.12.4.2     yamt 	struct mbuf *m0;
    658  1.12.4.2     yamt 	u_int8_t ds_field = 0;
    659  1.12.4.2     yamt 
    660  1.12.4.2     yamt 	if (pktattr == NULL ||
    661  1.12.4.2     yamt 	    (pktattr->pattr_af != AF_INET && pktattr->pattr_af != AF_INET6))
    662  1.12.4.2     yamt 		return ((u_int8_t)0);
    663  1.12.4.2     yamt 
    664  1.12.4.2     yamt 	/* verify that pattr_hdr is within the mbuf data */
    665  1.12.4.2     yamt 	for (m0 = m; m0 != NULL; m0 = m0->m_next)
    666  1.12.4.3     yamt 		if (((char *)pktattr->pattr_hdr >= m0->m_data) &&
    667  1.12.4.3     yamt 		    ((char *)pktattr->pattr_hdr < m0->m_data + m0->m_len))
    668  1.12.4.2     yamt 			break;
    669  1.12.4.2     yamt 	if (m0 == NULL) {
    670  1.12.4.2     yamt 		/* ick, pattr_hdr is stale */
    671  1.12.4.2     yamt 		pktattr->pattr_af = AF_UNSPEC;
    672  1.12.4.2     yamt #ifdef ALTQ_DEBUG
    673  1.12.4.2     yamt 		printf("read_dsfield: can't locate header!\n");
    674  1.12.4.2     yamt #endif
    675  1.12.4.2     yamt 		return ((u_int8_t)0);
    676  1.12.4.2     yamt 	}
    677  1.12.4.2     yamt 
    678  1.12.4.2     yamt 	if (pktattr->pattr_af == AF_INET) {
    679  1.12.4.2     yamt 		struct ip *ip = (struct ip *)pktattr->pattr_hdr;
    680  1.12.4.2     yamt 
    681  1.12.4.2     yamt 		if (ip->ip_v != 4)
    682  1.12.4.2     yamt 			return ((u_int8_t)0);	/* version mismatch! */
    683  1.12.4.2     yamt 		ds_field = ip->ip_tos;
    684  1.12.4.2     yamt 	}
    685  1.12.4.2     yamt #ifdef INET6
    686  1.12.4.2     yamt 	else if (pktattr->pattr_af == AF_INET6) {
    687  1.12.4.2     yamt 		struct ip6_hdr *ip6 = (struct ip6_hdr *)pktattr->pattr_hdr;
    688  1.12.4.2     yamt 		u_int32_t flowlabel;
    689  1.12.4.2     yamt 
    690  1.12.4.2     yamt 		flowlabel = ntohl(ip6->ip6_flow);
    691  1.12.4.2     yamt 		if ((flowlabel >> 28) != 6)
    692  1.12.4.2     yamt 			return ((u_int8_t)0);	/* version mismatch! */
    693  1.12.4.2     yamt 		ds_field = (flowlabel >> 20) & 0xff;
    694  1.12.4.2     yamt 	}
    695  1.12.4.2     yamt #endif
    696  1.12.4.2     yamt 	return (ds_field);
    697  1.12.4.2     yamt }
    698  1.12.4.2     yamt 
    699  1.12.4.2     yamt void
    700  1.12.4.2     yamt write_dsfield(struct mbuf *m, struct altq_pktattr *pktattr, u_int8_t dsfield)
    701  1.12.4.2     yamt {
    702  1.12.4.2     yamt 	struct mbuf *m0;
    703  1.12.4.2     yamt 
    704  1.12.4.2     yamt 	if (pktattr == NULL ||
    705  1.12.4.2     yamt 	    (pktattr->pattr_af != AF_INET && pktattr->pattr_af != AF_INET6))
    706  1.12.4.2     yamt 		return;
    707  1.12.4.2     yamt 
    708  1.12.4.2     yamt 	/* verify that pattr_hdr is within the mbuf data */
    709  1.12.4.2     yamt 	for (m0 = m; m0 != NULL; m0 = m0->m_next)
    710  1.12.4.3     yamt 		if (((char *)pktattr->pattr_hdr >= m0->m_data) &&
    711  1.12.4.3     yamt 		    ((char *)pktattr->pattr_hdr < m0->m_data + m0->m_len))
    712  1.12.4.2     yamt 			break;
    713  1.12.4.2     yamt 	if (m0 == NULL) {
    714  1.12.4.2     yamt 		/* ick, pattr_hdr is stale */
    715  1.12.4.2     yamt 		pktattr->pattr_af = AF_UNSPEC;
    716  1.12.4.2     yamt #ifdef ALTQ_DEBUG
    717  1.12.4.2     yamt 		printf("write_dsfield: can't locate header!\n");
    718  1.12.4.2     yamt #endif
    719  1.12.4.2     yamt 		return;
    720  1.12.4.2     yamt 	}
    721  1.12.4.2     yamt 
    722  1.12.4.2     yamt 	if (pktattr->pattr_af == AF_INET) {
    723  1.12.4.2     yamt 		struct ip *ip = (struct ip *)pktattr->pattr_hdr;
    724  1.12.4.2     yamt 		u_int8_t old;
    725  1.12.4.2     yamt 		int32_t sum;
    726  1.12.4.2     yamt 
    727  1.12.4.2     yamt 		if (ip->ip_v != 4)
    728  1.12.4.2     yamt 			return;		/* version mismatch! */
    729  1.12.4.2     yamt 		old = ip->ip_tos;
    730  1.12.4.2     yamt 		dsfield |= old & 3;	/* leave CU bits */
    731  1.12.4.2     yamt 		if (old == dsfield)
    732  1.12.4.2     yamt 			return;
    733  1.12.4.2     yamt 		ip->ip_tos = dsfield;
    734  1.12.4.2     yamt 		/*
    735  1.12.4.2     yamt 		 * update checksum (from RFC1624)
    736  1.12.4.2     yamt 		 *	   HC' = ~(~HC + ~m + m')
    737  1.12.4.2     yamt 		 */
    738  1.12.4.2     yamt 		sum = ~ntohs(ip->ip_sum) & 0xffff;
    739  1.12.4.2     yamt 		sum += 0xff00 + (~old & 0xff) + dsfield;
    740  1.12.4.2     yamt 		sum = (sum >> 16) + (sum & 0xffff);
    741  1.12.4.2     yamt 		sum += (sum >> 16);  /* add carry */
    742  1.12.4.2     yamt 
    743  1.12.4.2     yamt 		ip->ip_sum = htons(~sum & 0xffff);
    744  1.12.4.2     yamt 	}
    745  1.12.4.2     yamt #ifdef INET6
    746  1.12.4.2     yamt 	else if (pktattr->pattr_af == AF_INET6) {
    747  1.12.4.2     yamt 		struct ip6_hdr *ip6 = (struct ip6_hdr *)pktattr->pattr_hdr;
    748  1.12.4.2     yamt 		u_int32_t flowlabel;
    749  1.12.4.2     yamt 
    750  1.12.4.2     yamt 		flowlabel = ntohl(ip6->ip6_flow);
    751  1.12.4.2     yamt 		if ((flowlabel >> 28) != 6)
    752  1.12.4.2     yamt 			return;		/* version mismatch! */
    753  1.12.4.2     yamt 		flowlabel = (flowlabel & 0xf03fffff) | (dsfield << 20);
    754  1.12.4.2     yamt 		ip6->ip6_flow = htonl(flowlabel);
    755  1.12.4.2     yamt 	}
    756  1.12.4.2     yamt #endif
    757  1.12.4.2     yamt 	return;
    758  1.12.4.2     yamt }
    759  1.12.4.2     yamt 
    760  1.12.4.2     yamt 
    761  1.12.4.2     yamt /*
    762  1.12.4.2     yamt  * high resolution clock support taking advantage of a machine dependent
    763  1.12.4.2     yamt  * high resolution time counter (e.g., timestamp counter of intel pentium).
    764  1.12.4.2     yamt  * we assume
    765  1.12.4.2     yamt  *  - 64-bit-long monotonically-increasing counter
    766  1.12.4.2     yamt  *  - frequency range is 100M-4GHz (CPU speed)
    767  1.12.4.2     yamt  */
    768  1.12.4.2     yamt /* if pcc is not available or disabled, emulate 256MHz using microtime() */
    769  1.12.4.2     yamt #define	MACHCLK_SHIFT	8
    770  1.12.4.2     yamt 
    771  1.12.4.2     yamt int machclk_usepcc;
    772  1.12.4.2     yamt u_int32_t machclk_freq = 0;
    773  1.12.4.2     yamt u_int32_t machclk_per_tick = 0;
    774  1.12.4.2     yamt 
    775  1.12.4.2     yamt #ifdef __alpha__
    776  1.12.4.2     yamt #ifdef __FreeBSD__
    777  1.12.4.2     yamt extern u_int32_t cycles_per_sec;	/* alpha cpu clock frequency */
    778  1.12.4.2     yamt #elif defined(__NetBSD__) || defined(__OpenBSD__)
    779  1.12.4.2     yamt extern u_int64_t cycles_per_usec;	/* alpha cpu clock frequency */
    780  1.12.4.2     yamt #endif
    781  1.12.4.2     yamt #endif /* __alpha__ */
    782  1.12.4.2     yamt 
    783  1.12.4.2     yamt void
    784  1.12.4.2     yamt init_machclk(void)
    785  1.12.4.2     yamt {
    786  1.12.4.4     yamt 
    787  1.12.4.4     yamt 	callout_init(&tbr_callout, 0);
    788  1.12.4.4     yamt 
    789  1.12.4.2     yamt 	machclk_usepcc = 1;
    790  1.12.4.2     yamt 
    791  1.12.4.2     yamt #if (!defined(__i386__) && !defined(__alpha__)) || defined(ALTQ_NOPCC)
    792  1.12.4.2     yamt 	machclk_usepcc = 0;
    793  1.12.4.2     yamt #endif
    794  1.12.4.2     yamt #if defined(__FreeBSD__) && defined(SMP)
    795  1.12.4.2     yamt 	machclk_usepcc = 0;
    796  1.12.4.2     yamt #endif
    797  1.12.4.2     yamt #if defined(__NetBSD__) && defined(MULTIPROCESSOR)
    798  1.12.4.2     yamt 	machclk_usepcc = 0;
    799  1.12.4.2     yamt #endif
    800  1.12.4.2     yamt #ifdef __i386__
    801  1.12.4.2     yamt 	/* check if TSC is available */
    802  1.12.4.2     yamt 	if (machclk_usepcc == 1 && (cpu_feature & CPUID_TSC) == 0)
    803  1.12.4.2     yamt 		machclk_usepcc = 0;
    804  1.12.4.2     yamt #endif
    805  1.12.4.2     yamt 
    806  1.12.4.2     yamt 	if (machclk_usepcc == 0) {
    807  1.12.4.2     yamt 		/* emulate 256MHz using microtime() */
    808  1.12.4.2     yamt 		machclk_freq = 1000000 << MACHCLK_SHIFT;
    809  1.12.4.2     yamt 		machclk_per_tick = machclk_freq / hz;
    810  1.12.4.2     yamt #ifdef ALTQ_DEBUG
    811  1.12.4.2     yamt 		printf("altq: emulate %uHz CPU clock\n", machclk_freq);
    812  1.12.4.2     yamt #endif
    813  1.12.4.2     yamt 		return;
    814  1.12.4.2     yamt 	}
    815  1.12.4.2     yamt 
    816  1.12.4.2     yamt 	/*
    817  1.12.4.2     yamt 	 * if the clock frequency (of Pentium TSC or Alpha PCC) is
    818  1.12.4.2     yamt 	 * accessible, just use it.
    819  1.12.4.2     yamt 	 */
    820  1.12.4.2     yamt #ifdef __i386__
    821  1.12.4.2     yamt #ifdef __FreeBSD__
    822  1.12.4.2     yamt #if (__FreeBSD_version > 300000)
    823  1.12.4.2     yamt 	machclk_freq = tsc_freq;
    824  1.12.4.2     yamt #else
    825  1.12.4.2     yamt 	machclk_freq = i586_ctr_freq;
    826  1.12.4.2     yamt #endif
    827  1.12.4.2     yamt #elif defined(__NetBSD__)
    828  1.12.4.2     yamt 	machclk_freq = (u_int32_t)curcpu()->ci_tsc_freq;
    829  1.12.4.2     yamt #elif defined(__OpenBSD__) && (defined(I586_CPU) || defined(I686_CPU))
    830  1.12.4.2     yamt 	machclk_freq = pentium_mhz * 1000000;
    831  1.12.4.2     yamt #endif
    832  1.12.4.2     yamt #elif defined(__alpha__)
    833  1.12.4.2     yamt #ifdef __FreeBSD__
    834  1.12.4.2     yamt 	machclk_freq = cycles_per_sec;
    835  1.12.4.2     yamt #elif defined(__NetBSD__) || defined(__OpenBSD__)
    836  1.12.4.2     yamt 	machclk_freq = (u_int32_t)(cycles_per_usec * 1000000);
    837  1.12.4.2     yamt #endif
    838  1.12.4.2     yamt #endif /* __alpha__ */
    839  1.12.4.2     yamt 
    840  1.12.4.2     yamt 	/*
    841  1.12.4.2     yamt 	 * if we don't know the clock frequency, measure it.
    842  1.12.4.2     yamt 	 */
    843  1.12.4.2     yamt 	if (machclk_freq == 0) {
    844  1.12.4.2     yamt 		static int	wait;
    845  1.12.4.2     yamt 		struct timeval	tv_start, tv_end;
    846  1.12.4.2     yamt 		u_int64_t	start, end, diff;
    847  1.12.4.2     yamt 		int		timo;
    848  1.12.4.2     yamt 
    849  1.12.4.2     yamt 		microtime(&tv_start);
    850  1.12.4.2     yamt 		start = read_machclk();
    851  1.12.4.2     yamt 		timo = hz;	/* 1 sec */
    852  1.12.4.2     yamt 		(void)tsleep(&wait, PWAIT | PCATCH, "init_machclk", timo);
    853  1.12.4.2     yamt 		microtime(&tv_end);
    854  1.12.4.2     yamt 		end = read_machclk();
    855  1.12.4.2     yamt 		diff = (u_int64_t)(tv_end.tv_sec - tv_start.tv_sec) * 1000000
    856  1.12.4.2     yamt 		    + tv_end.tv_usec - tv_start.tv_usec;
    857  1.12.4.2     yamt 		if (diff != 0)
    858  1.12.4.2     yamt 			machclk_freq = (u_int)((end - start) * 1000000 / diff);
    859  1.12.4.2     yamt 	}
    860  1.12.4.2     yamt 
    861  1.12.4.2     yamt 	machclk_per_tick = machclk_freq / hz;
    862  1.12.4.2     yamt 
    863  1.12.4.2     yamt #ifdef ALTQ_DEBUG
    864  1.12.4.2     yamt 	printf("altq: CPU clock: %uHz\n", machclk_freq);
    865  1.12.4.2     yamt #endif
    866       1.1  thorpej }
    867       1.1  thorpej 
    868  1.12.4.2     yamt #if defined(__OpenBSD__) && defined(__i386__)
    869  1.12.4.2     yamt static inline u_int64_t
    870  1.12.4.2     yamt rdtsc(void)
    871  1.12.4.2     yamt {
    872  1.12.4.2     yamt 	u_int64_t rv;
    873  1.12.4.2     yamt 	__asm __volatile(".byte 0x0f, 0x31" : "=A" (rv));
    874  1.12.4.2     yamt 	return (rv);
    875  1.12.4.2     yamt }
    876  1.12.4.2     yamt #endif /* __OpenBSD__ && __i386__ */
    877  1.12.4.2     yamt 
    878  1.12.4.2     yamt u_int64_t
    879  1.12.4.2     yamt read_machclk(void)
    880       1.1  thorpej {
    881  1.12.4.2     yamt 	u_int64_t val;
    882       1.1  thorpej 
    883  1.12.4.2     yamt 	if (machclk_usepcc) {
    884  1.12.4.2     yamt #if defined(__i386__)
    885  1.12.4.2     yamt 		val = rdtsc();
    886  1.12.4.2     yamt #elif defined(__alpha__)
    887  1.12.4.2     yamt 		static u_int32_t last_pcc, upper;
    888  1.12.4.2     yamt 		u_int32_t pcc;
    889  1.12.4.2     yamt 
    890  1.12.4.2     yamt 		/*
    891  1.12.4.2     yamt 		 * for alpha, make a 64bit counter value out of the 32bit
    892  1.12.4.2     yamt 		 * alpha processor cycle counter.
    893  1.12.4.2     yamt 		 * read_machclk must be called within a half of its
    894  1.12.4.2     yamt 		 * wrap-around cycle (about 5 sec for 400MHz cpu) to properly
    895  1.12.4.2     yamt 		 * detect a counter wrap-around.
    896  1.12.4.2     yamt 		 * tbr_timeout calls read_machclk once a second.
    897  1.12.4.2     yamt 		 */
    898  1.12.4.2     yamt 		pcc = (u_int32_t)alpha_rpcc();
    899  1.12.4.2     yamt 		if (pcc <= last_pcc)
    900  1.12.4.2     yamt 			upper++;
    901  1.12.4.2     yamt 		last_pcc = pcc;
    902  1.12.4.2     yamt 		val = ((u_int64_t)upper << 32) + pcc;
    903  1.12.4.2     yamt #else
    904  1.12.4.2     yamt 		panic("read_machclk");
    905  1.12.4.2     yamt #endif
    906       1.1  thorpej 	} else {
    907  1.12.4.2     yamt 		struct timeval tv;
    908  1.12.4.2     yamt 
    909  1.12.4.2     yamt 		microtime(&tv);
    910  1.12.4.2     yamt 		val = (((u_int64_t)(tv.tv_sec - boottime.tv_sec) * 1000000
    911  1.12.4.2     yamt 		    + tv.tv_usec) << MACHCLK_SHIFT);
    912       1.1  thorpej 	}
    913  1.12.4.2     yamt 	return (val);
    914       1.1  thorpej }
    915       1.1  thorpej 
    916  1.12.4.2     yamt #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.12.4.2     yamt altq_extractflow(struct mbuf *m, int af, struct flowinfo *flow,
    933  1.12.4.2     yamt     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.12.4.2     yamt 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.12.4.3     yamt 		if (((char *)ip >= m0->m_data) &&
   1051  1.12.4.3     yamt 		    ((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.12.4.3     yamt 	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.12.4.3     yamt 		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.12.4.3     yamt 			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.12.4.3     yamt 			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.12.4.2     yamt 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.12.4.3     yamt 		if (((char *)ip6 >= m0->m_data) &&
   1138  1.12.4.3     yamt 		    ((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.12.4.3     yamt 	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.12.4.3     yamt 			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.12.4.3     yamt 				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.12.4.3     yamt 			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.12.4.3     yamt 			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.12.4.2     yamt acc_add_filter(struct acc_classifier *classifier, struct flow_filter *filter,
   1223  1.12.4.2     yamt     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.12.4.1     yamt 	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.12.4.2     yamt 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.12.4.1     yamt 	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.12.4.2     yamt 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.12.4.1     yamt 					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.12.4.2     yamt 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.12.4.2     yamt apply_filter4(u_int32_t fbmask, struct flow_filter *filt,
   1488  1.12.4.2     yamt     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.12.4.2     yamt apply_ppfilter4(u_int32_t fbmask, struct flow_filter *filt,
   1521  1.12.4.2     yamt     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.12.4.2     yamt apply_tosfilter4(u_int32_t fbmask, struct flow_filter *filt,
   1540  1.12.4.2     yamt     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.12.4.2     yamt apply_filter6(u_int32_t fbmask, struct flow_filter6 *filt,
   1554  1.12.4.2     yamt     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.12.4.2     yamt 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.12.4.2     yamt 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.12.4.2     yamt 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.12.4.2     yamt 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.12.4.2     yamt 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.12.4.1     yamt 		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.12.4.2     yamt 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.12.4.2     yamt #endif /* ALTQ3_CLFIER_COMPAT */
   1812