zig

fork of https://codeberg.org/ziglang/zig
Log | Files | Refs | README | LICENSE

ifq.h (11643B) - Raw


      1 /*-
      2  * SPDX-License-Identifier: BSD-3-Clause
      3  *
      4  * Copyright (c) 1982, 1986, 1989, 1993
      5  *	The Regents of the University of California.  All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  * 3. Neither the name of the University nor the names of its contributors
     16  *    may be used to endorse or promote products derived from this software
     17  *    without specific prior written permission.
     18  *
     19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     29  * SUCH DAMAGE.
     30  *
     31  *	From: @(#)if.h	8.1 (Berkeley) 6/10/93
     32  */
     33 
     34 #ifndef	_NET_IFQ_H_
     35 #define	_NET_IFQ_H_
     36 
     37 #ifdef _KERNEL
     38 #include <sys/mbuf.h>		/* ifqueue only? */
     39 #include <sys/buf_ring.h>
     40 #include <net/vnet.h>
     41 #endif /* _KERNEL */
     42 #include <sys/lock.h>		/* XXX */
     43 #include <sys/mutex.h>		/* struct ifqueue */
     44 
     45 /*
     46  * Couple of ugly extra definitions that are required since ifq.h
     47  * is splitted from if_var.h.
     48  */
     49 #define	IF_DUNIT_NONE	-1
     50 
     51 #include <net/altq/if_altq.h>
     52 
     53 /*
     54  * Structure defining a queue for a network interface.
     55  */
     56 struct	ifqueue {
     57 	struct	mbuf *ifq_head;
     58 	struct	mbuf *ifq_tail;
     59 	int	ifq_len;
     60 	int	ifq_maxlen;
     61 	struct	mtx ifq_mtx;
     62 };
     63 
     64 #ifdef _KERNEL
     65 /*
     66  * Output queues (ifp->if_snd) and slow device input queues (*ifp->if_slowq)
     67  * are queues of messages stored on ifqueue structures
     68  * (defined above).  Entries are added to and deleted from these structures
     69  * by these macros.
     70  */
     71 #define IF_LOCK(ifq)		mtx_lock(&(ifq)->ifq_mtx)
     72 #define IF_UNLOCK(ifq)		mtx_unlock(&(ifq)->ifq_mtx)
     73 #define	IF_LOCK_ASSERT(ifq)	mtx_assert(&(ifq)->ifq_mtx, MA_OWNED)
     74 #define	_IF_QFULL(ifq)		((ifq)->ifq_len >= (ifq)->ifq_maxlen)
     75 #define	_IF_QLEN(ifq)		((ifq)->ifq_len)
     76 
     77 #define	_IF_ENQUEUE(ifq, m) do { 				\
     78 	(m)->m_nextpkt = NULL;					\
     79 	if ((ifq)->ifq_tail == NULL) 				\
     80 		(ifq)->ifq_head = m; 				\
     81 	else 							\
     82 		(ifq)->ifq_tail->m_nextpkt = m; 		\
     83 	(ifq)->ifq_tail = m; 					\
     84 	(ifq)->ifq_len++; 					\
     85 } while (0)
     86 
     87 #define IF_ENQUEUE(ifq, m) do {					\
     88 	IF_LOCK(ifq); 						\
     89 	_IF_ENQUEUE(ifq, m); 					\
     90 	IF_UNLOCK(ifq); 					\
     91 } while (0)
     92 
     93 #define	_IF_PREPEND(ifq, m) do {				\
     94 	(m)->m_nextpkt = (ifq)->ifq_head; 			\
     95 	if ((ifq)->ifq_tail == NULL) 				\
     96 		(ifq)->ifq_tail = (m); 				\
     97 	(ifq)->ifq_head = (m); 					\
     98 	(ifq)->ifq_len++; 					\
     99 } while (0)
    100 
    101 #define IF_PREPEND(ifq, m) do {		 			\
    102 	IF_LOCK(ifq); 						\
    103 	_IF_PREPEND(ifq, m); 					\
    104 	IF_UNLOCK(ifq); 					\
    105 } while (0)
    106 
    107 #define	_IF_DEQUEUE(ifq, m) do { 				\
    108 	(m) = (ifq)->ifq_head; 					\
    109 	if (m) { 						\
    110 		if (((ifq)->ifq_head = (m)->m_nextpkt) == NULL)	\
    111 			(ifq)->ifq_tail = NULL; 		\
    112 		(m)->m_nextpkt = NULL; 				\
    113 		(ifq)->ifq_len--; 				\
    114 	} 							\
    115 } while (0)
    116 
    117 #define IF_DEQUEUE(ifq, m) do { 				\
    118 	IF_LOCK(ifq); 						\
    119 	_IF_DEQUEUE(ifq, m); 					\
    120 	IF_UNLOCK(ifq); 					\
    121 } while (0)
    122 
    123 #define	_IF_DEQUEUE_ALL(ifq, m) do {				\
    124 	(m) = (ifq)->ifq_head;					\
    125 	(ifq)->ifq_head = (ifq)->ifq_tail = NULL;		\
    126 	(ifq)->ifq_len = 0;					\
    127 } while (0)
    128 
    129 #define	IF_DEQUEUE_ALL(ifq, m) do {				\
    130 	IF_LOCK(ifq); 						\
    131 	_IF_DEQUEUE_ALL(ifq, m);				\
    132 	IF_UNLOCK(ifq); 					\
    133 } while (0)
    134 
    135 #define	_IF_POLL(ifq, m)	((m) = (ifq)->ifq_head)
    136 #define	IF_POLL(ifq, m)		_IF_POLL(ifq, m)
    137 
    138 #define _IF_DRAIN(ifq) do { 					\
    139 	struct mbuf *m; 					\
    140 	for (;;) { 						\
    141 		_IF_DEQUEUE(ifq, m); 				\
    142 		if (m == NULL) 					\
    143 			break; 					\
    144 		m_freem(m); 					\
    145 	} 							\
    146 } while (0)
    147 
    148 #define IF_DRAIN(ifq) do {					\
    149 	IF_LOCK(ifq);						\
    150 	_IF_DRAIN(ifq);						\
    151 	IF_UNLOCK(ifq);						\
    152 } while(0)
    153 
    154 int	if_handoff(struct ifqueue *ifq, struct mbuf *m, struct ifnet *ifp,
    155 	    int adjust);
    156 #define	IF_HANDOFF(ifq, m, ifp)			\
    157 	if_handoff((struct ifqueue *)ifq, m, ifp, 0)
    158 #define	IF_HANDOFF_ADJ(ifq, m, ifp, adj)	\
    159 	if_handoff((struct ifqueue *)ifq, m, ifp, adj)
    160 
    161 void	if_start(struct ifnet *);
    162 
    163 #define	IFQ_ENQUEUE(ifq, m, err)					\
    164 do {									\
    165 	IF_LOCK(ifq);							\
    166 	if (ALTQ_IS_ENABLED(ifq))					\
    167 		ALTQ_ENQUEUE(ifq, m, NULL, err);			\
    168 	else {								\
    169 		if (_IF_QFULL(ifq)) {					\
    170 			m_freem(m);					\
    171 			(err) = ENOBUFS;				\
    172 		} else {						\
    173 			_IF_ENQUEUE(ifq, m);				\
    174 			(err) = 0;					\
    175 		}							\
    176 	}								\
    177 	IF_UNLOCK(ifq);							\
    178 } while (0)
    179 
    180 #define	IFQ_DEQUEUE_NOLOCK(ifq, m)					\
    181 do {									\
    182 	if (TBR_IS_ENABLED(ifq))					\
    183 		(m) = tbr_dequeue_ptr(ifq, ALTDQ_REMOVE);		\
    184 	else if (ALTQ_IS_ENABLED(ifq))					\
    185 		ALTQ_DEQUEUE(ifq, m);					\
    186 	else								\
    187 		_IF_DEQUEUE(ifq, m);					\
    188 } while (0)
    189 
    190 #define	IFQ_DEQUEUE(ifq, m)						\
    191 do {									\
    192 	IF_LOCK(ifq);							\
    193 	IFQ_DEQUEUE_NOLOCK(ifq, m);					\
    194 	IF_UNLOCK(ifq);							\
    195 } while (0)
    196 
    197 #define	IFQ_POLL_NOLOCK(ifq, m)						\
    198 do {									\
    199 	if (TBR_IS_ENABLED(ifq))					\
    200 		(m) = tbr_dequeue_ptr(ifq, ALTDQ_POLL);			\
    201 	else if (ALTQ_IS_ENABLED(ifq))					\
    202 		ALTQ_POLL(ifq, m);					\
    203 	else								\
    204 		_IF_POLL(ifq, m);					\
    205 } while (0)
    206 
    207 #define	IFQ_POLL(ifq, m)						\
    208 do {									\
    209 	IF_LOCK(ifq);							\
    210 	IFQ_POLL_NOLOCK(ifq, m);					\
    211 	IF_UNLOCK(ifq);							\
    212 } while (0)
    213 
    214 #define	IFQ_PURGE_NOLOCK(ifq)						\
    215 do {									\
    216 	if (ALTQ_IS_ENABLED(ifq)) {					\
    217 		ALTQ_PURGE(ifq);					\
    218 	} else								\
    219 		_IF_DRAIN(ifq);						\
    220 } while (0)
    221 
    222 #define	IFQ_PURGE(ifq)							\
    223 do {									\
    224 	IF_LOCK(ifq);							\
    225 	IFQ_PURGE_NOLOCK(ifq);						\
    226 	IF_UNLOCK(ifq);							\
    227 } while (0)
    228 
    229 #define	IFQ_SET_READY(ifq)						\
    230 	do { ((ifq)->altq_flags |= ALTQF_READY); } while (0)
    231 
    232 #define	IFQ_LOCK(ifq)			IF_LOCK(ifq)
    233 #define	IFQ_UNLOCK(ifq)			IF_UNLOCK(ifq)
    234 #define	IFQ_LOCK_ASSERT(ifq)		IF_LOCK_ASSERT(ifq)
    235 #define	IFQ_IS_EMPTY(ifq)		((ifq)->ifq_len == 0)
    236 #define	IFQ_INC_LEN(ifq)		((ifq)->ifq_len++)
    237 #define	IFQ_DEC_LEN(ifq)		(--(ifq)->ifq_len)
    238 #define	IFQ_SET_MAXLEN(ifq, len)	((ifq)->ifq_maxlen = (len))
    239 
    240 /*
    241  * The IFF_DRV_OACTIVE test should really occur in the device driver, not in
    242  * the handoff logic, as that flag is locked by the device driver.
    243  */
    244 #define	IFQ_HANDOFF_ADJ(ifp, m, adj, err)				\
    245 do {									\
    246 	int len;							\
    247 	short mflags;							\
    248 									\
    249 	len = (m)->m_pkthdr.len;					\
    250 	mflags = (m)->m_flags;						\
    251 	IFQ_ENQUEUE(&(ifp)->if_snd, m, err);				\
    252 	if ((err) == 0) {						\
    253 		if_inc_counter((ifp), IFCOUNTER_OBYTES, len + (adj));	\
    254 		if (mflags & M_MCAST)					\
    255 			if_inc_counter((ifp), IFCOUNTER_OMCASTS, 1);	\
    256 		if (((ifp)->if_drv_flags & IFF_DRV_OACTIVE) == 0)	\
    257 			if_start(ifp);					\
    258 	} else								\
    259 		if_inc_counter((ifp), IFCOUNTER_OQDROPS, 1);		\
    260 } while (0)
    261 
    262 #define	IFQ_HANDOFF(ifp, m, err)					\
    263 	IFQ_HANDOFF_ADJ(ifp, m, 0, err)
    264 
    265 #define	IFQ_DRV_DEQUEUE(ifq, m)						\
    266 do {									\
    267 	(m) = (ifq)->ifq_drv_head;					\
    268 	if (m) {							\
    269 		if (((ifq)->ifq_drv_head = (m)->m_nextpkt) == NULL)	\
    270 			(ifq)->ifq_drv_tail = NULL;			\
    271 		(m)->m_nextpkt = NULL;					\
    272 		(ifq)->ifq_drv_len--;					\
    273 	} else {							\
    274 		IFQ_LOCK(ifq);						\
    275 		IFQ_DEQUEUE_NOLOCK(ifq, m);				\
    276 		while ((ifq)->ifq_drv_len < (ifq)->ifq_drv_maxlen) {	\
    277 			struct mbuf *m0;				\
    278 			IFQ_DEQUEUE_NOLOCK(ifq, m0);			\
    279 			if (m0 == NULL)					\
    280 				break;					\
    281 			m0->m_nextpkt = NULL;				\
    282 			if ((ifq)->ifq_drv_tail == NULL)		\
    283 				(ifq)->ifq_drv_head = m0;		\
    284 			else						\
    285 				(ifq)->ifq_drv_tail->m_nextpkt = m0;	\
    286 			(ifq)->ifq_drv_tail = m0;			\
    287 			(ifq)->ifq_drv_len++;				\
    288 		}							\
    289 		IFQ_UNLOCK(ifq);					\
    290 	}								\
    291 } while (0)
    292 
    293 #define	IFQ_DRV_PREPEND(ifq, m)						\
    294 do {									\
    295 	(m)->m_nextpkt = (ifq)->ifq_drv_head;				\
    296 	if ((ifq)->ifq_drv_tail == NULL)				\
    297 		(ifq)->ifq_drv_tail = (m);				\
    298 	(ifq)->ifq_drv_head = (m);					\
    299 	(ifq)->ifq_drv_len++;						\
    300 } while (0)
    301 
    302 #define	IFQ_DRV_IS_EMPTY(ifq)						\
    303 	(((ifq)->ifq_drv_len == 0) && ((ifq)->ifq_len == 0))
    304 
    305 #define	IFQ_DRV_PURGE(ifq)						\
    306 do {									\
    307 	struct mbuf *m, *n = (ifq)->ifq_drv_head;			\
    308 	while((m = n) != NULL) {					\
    309 		n = m->m_nextpkt;					\
    310 		m_freem(m);						\
    311 	}								\
    312 	(ifq)->ifq_drv_head = (ifq)->ifq_drv_tail = NULL;		\
    313 	(ifq)->ifq_drv_len = 0;						\
    314 	IFQ_PURGE(ifq);							\
    315 } while (0)
    316 
    317 #ifdef ALTQ
    318 int drbr_enqueue(struct ifnet *ifp, struct buf_ring *br, struct mbuf *m);
    319 void drbr_putback(struct ifnet *ifp, struct buf_ring *br, struct mbuf *m_new);
    320 struct mbuf *drbr_peek(struct ifnet *ifp, struct buf_ring *br);
    321 void drbr_flush(struct ifnet *ifp, struct buf_ring *br);
    322 struct mbuf *drbr_dequeue(struct ifnet *ifp, struct buf_ring *br);
    323 void	drbr_advance(struct ifnet *ifp, struct buf_ring *br);
    324 struct mbuf *drbr_dequeue_cond(struct ifnet *ifp, struct buf_ring *br,
    325     int (*func) (struct mbuf *, void *), void *arg);
    326 int	drbr_empty(struct ifnet *ifp, struct buf_ring *br);
    327 int	drbr_needs_enqueue(struct ifnet *ifp, struct buf_ring *br);
    328 int	drbr_inuse(struct ifnet *ifp, struct buf_ring *br);
    329 
    330 #else /* !ALTQ */
    331 static __inline int
    332 drbr_enqueue(struct ifnet *ifp, struct buf_ring *br, struct mbuf *m)
    333 {	
    334 	int error = 0;
    335 
    336 	error = buf_ring_enqueue(br, m);
    337 	if (error)
    338 		m_freem(m);
    339 
    340 	return (error);
    341 }
    342 
    343 static __inline void
    344 drbr_putback(struct ifnet *ifp, struct buf_ring *br, struct mbuf *m_new)
    345 {
    346 	/*
    347 	 * The top of the list needs to be swapped 
    348 	 * for this one.
    349 	 */
    350 	buf_ring_putback_sc(br, m_new);
    351 }
    352 
    353 static __inline struct mbuf *
    354 drbr_peek(struct ifnet *ifp, struct buf_ring *br)
    355 {
    356 	return ((struct mbuf *)buf_ring_peek_clear_sc(br));
    357 }
    358 
    359 static __inline void
    360 drbr_flush(struct ifnet *ifp, struct buf_ring *br)
    361 {
    362 	struct mbuf *m;
    363 
    364 	while ((m = (struct mbuf *)buf_ring_dequeue_sc(br)) != NULL)
    365 		m_freem(m);
    366 }
    367 
    368 static __inline struct mbuf *
    369 drbr_dequeue(struct ifnet *ifp, struct buf_ring *br)
    370 {
    371 	return ((struct mbuf *)buf_ring_dequeue_sc(br));
    372 }
    373 
    374 static __inline void
    375 drbr_advance(struct ifnet *ifp, struct buf_ring *br)
    376 {
    377 	return (buf_ring_advance_sc(br));
    378 }
    379 
    380 static __inline struct mbuf *
    381 drbr_dequeue_cond(struct ifnet *ifp, struct buf_ring *br,
    382     int (*func) (struct mbuf *, void *), void *arg) 
    383 {
    384 	struct mbuf *m;
    385 	m = (struct mbuf *)buf_ring_peek(br);
    386 	if (m == NULL || func(m, arg) == 0)
    387 		return (NULL);
    388 
    389 	return ((struct mbuf *)buf_ring_dequeue_sc(br));
    390 }
    391 
    392 static __inline int
    393 drbr_empty(struct ifnet *ifp, struct buf_ring *br)
    394 {
    395 	return (buf_ring_empty(br));
    396 }
    397 
    398 static __inline int
    399 drbr_needs_enqueue(struct ifnet *ifp, struct buf_ring *br)
    400 {
    401 	return (!buf_ring_empty(br));
    402 }
    403 
    404 static __inline int
    405 drbr_inuse(struct ifnet *ifp, struct buf_ring *br)
    406 {
    407 	return (buf_ring_count(br));
    408 }
    409 #endif /* ALTQ */
    410 
    411 static __inline void
    412 drbr_free(struct buf_ring *br, struct malloc_type *type)
    413 {
    414 
    415 	drbr_flush(NULL, br);
    416 	buf_ring_free(br, type);
    417 }
    418 
    419 extern	int ifqmaxlen;
    420 
    421 void	if_qflush(struct ifnet *);
    422 void	ifq_init(struct ifaltq *, struct ifnet *ifp);
    423 void	ifq_delete(struct ifaltq *);
    424 
    425 #endif /* _KERNEL */
    426 #endif /* !_NET_IFQ_H_ */