]> git.saurik.com Git - apple/xnu.git/blame - bsd/net/classq/classq_sfb.c
xnu-6153.141.1.tar.gz
[apple/xnu.git] / bsd / net / classq / classq_sfb.c
CommitLineData
316670eb 1/*
cb323159 2 * Copyright (c) 2011-2019 Apple Inc. All rights reserved.
316670eb
A
3 *
4 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
5 *
6 * This file contains Original Code and/or Modifications of Original Code
7 * as defined in and that are subject to the Apple Public Source License
8 * Version 2.0 (the 'License'). You may not use this file except in
9 * compliance with the License. The rights granted to you under the License
10 * may not be used to create, or enable the creation or redistribution of,
11 * unlawful or unlicensed copies of an Apple operating system, or to
12 * circumvent, violate, or enable the circumvention or violation of, any
13 * terms of an Apple operating system software license agreement.
14 *
15 * Please obtain a copy of the License at
16 * http://www.opensource.apple.com/apsl/ and read it before using this file.
17 *
18 * The Original Code and all software distributed under the License are
19 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
20 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
22 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23 * Please see the License for the specific language governing rights and
24 * limitations under the License.
25 *
26 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
27 */
28
29#include <sys/cdefs.h>
30#include <sys/param.h>
316670eb
A
31#include <sys/mbuf.h>
32#include <sys/socket.h>
33#include <sys/sockio.h>
34#include <sys/systm.h>
35#include <sys/sysctl.h>
36#include <sys/syslog.h>
37#include <sys/proc.h>
38#include <sys/errno.h>
39#include <sys/kernel.h>
40#include <sys/kauth.h>
41
42#include <kern/zalloc.h>
43
44#include <net/if.h>
45#include <net/if_var.h>
46#include <net/if_types.h>
47#include <net/dlil.h>
39236c6e 48#include <net/flowadv.h>
316670eb
A
49
50#include <netinet/in.h>
51#include <netinet/in_systm.h>
52#include <netinet/ip.h>
53#if INET6
54#include <netinet/ip6.h>
55#endif
56
57#include <net/classq/classq_sfb.h>
58#include <net/flowhash.h>
59#include <net/net_osdep.h>
39236c6e 60#include <dev/random/randomdev.h>
316670eb
A
61
62/*
63 * Stochastic Fair Blue
64 *
65 * Wu-chang Feng, Dilip D. Kandlur, Debanjan Saha, Kang G. Shin
66 * http://www.thefengs.com/wuchang/blue/CSE-TR-387-99.pdf
67 *
68 * Based on the NS code with the following parameters:
69 *
70 * bytes: false
71 * decrement: 0.001
72 * increment: 0.005
73 * hold-time: 10ms-50ms (randomized)
74 * algorithm: 0
75 * pbox: 1
76 * pbox-time: 50-100ms (randomized)
77 * hinterval: 11-23 (randomized)
78 *
79 * This implementation uses L = 2 and N = 32 for 2 sets of:
80 *
81 * B[L][N]: L x N array of bins (L levels, N bins per level)
82 *
83 * Each set effectively creates 32^2 virtual buckets (bin combinations)
84 * while using only O(32*2) states.
85 *
86 * Given a 32-bit hash value, we divide it such that octets [0,1,2,3] are
87 * used as index for the bins across the 2 levels, where level 1 uses [0,2]
88 * and level 2 uses [1,3]. The 2 values per level correspond to the indices
89 * for the current and warm-up sets (section 4.4. in the SFB paper regarding
90 * Moving Hash Functions explains the purposes of these 2 sets.)
91 */
92
93/*
94 * Use Murmur3A_x86_32 for hash function. It seems to perform consistently
95 * across platforms for 1-word key (32-bit flowhash value). See flowhash.h
96 * for other alternatives. We only need 16-bit hash output.
97 */
0a7de745
A
98#define SFB_HASH net_flowhash_mh3_x86_32
99#define SFB_HASHMASK HASHMASK(16)
316670eb 100
0a7de745 101#define SFB_BINMASK(_x) \
316670eb
A
102 ((_x) & HASHMASK(SFB_BINS_SHIFT))
103
0a7de745 104#define SFB_BINST(_sp, _l, _n, _c) \
316670eb
A
105 (&(*(_sp)->sfb_bins)[_c].stats[_l][_n])
106
0a7de745 107#define SFB_BINFT(_sp, _l, _n, _c) \
316670eb
A
108 (&(*(_sp)->sfb_bins)[_c].freezetime[_l][_n])
109
0a7de745 110#define SFB_FC_LIST(_sp, _n) \
316670eb
A
111 (&(*(_sp)->sfb_fc_lists)[_n])
112
113/*
114 * The holdtime parameter determines the minimum time interval between
115 * two successive updates of the marking probability. In the event the
116 * uplink speed is not known, a default value is chosen and is randomized
117 * to be within the following range.
118 */
0a7de745
A
119#define HOLDTIME_BASE (100ULL * 1000 * 1000) /* 100ms */
120#define HOLDTIME_MIN (10ULL * 1000 * 1000) /* 10ms */
121#define HOLDTIME_MAX (100ULL * 1000 * 1000) /* 100ms */
316670eb
A
122
123/*
124 * The pboxtime parameter determines the bandwidth allocated for rogue
125 * flows, i.e. the rate limiting bandwidth. In the event the uplink speed
126 * is not known, a default value is chosen and is randomized to be within
127 * the following range.
128 */
0a7de745
A
129#define PBOXTIME_BASE (300ULL * 1000 * 1000) /* 300ms */
130#define PBOXTIME_MIN (30ULL * 1000 * 1000) /* 30ms */
131#define PBOXTIME_MAX (300ULL * 1000 * 1000) /* 300ms */
316670eb 132
fe8ab488
A
133/*
134 * Target queueing delay is the amount of extra delay that can be added
135 * to accommodate variations in the link bandwidth. The queue should be
136 * large enough to induce this much delay and nothing more than that.
137 */
0a7de745
A
138#define TARGET_QDELAY_BASE (10ULL * 1000 * 1000) /* 10ms */
139#define TARGET_QDELAY_MIN (10ULL * 1000) /* 10us */
140#define TARGET_QDELAY_MAX (20ULL * 1000 * 1000 * 1000) /* 20s */
fe8ab488
A
141
142/*
143 * Update interval for checking the extra delay added by the queue. This
39037602 144 * should be 90-95 percentile of RTT experienced by any TCP connection
fe8ab488
A
145 * so that it will take care of the burst traffic.
146 */
0a7de745
A
147#define UPDATE_INTERVAL_BASE (100ULL * 1000 * 1000) /* 100ms */
148#define UPDATE_INTERVAL_MIN (100ULL * 1000 * 1000) /* 100ms */
149#define UPDATE_INTERVAL_MAX (10ULL * 1000 * 1000 * 1000) /* 10s */
fe8ab488 150
0a7de745 151#define SFB_RANDOM(sp, tmin, tmax) ((sfb_random(sp) % (tmax)) + (tmin))
316670eb 152
0a7de745 153#define SFB_PKT_PBOX 0x1 /* in penalty box */
316670eb
A
154
155/* The following mantissa values are in SFB_FP_SHIFT Q format */
0a7de745 156#define SFB_MAX_PMARK (1 << SFB_FP_SHIFT) /* Q14 representation of 1.00 */
316670eb
A
157
158/*
159 * These are d1 (increment) and d2 (decrement) parameters, used to determine
160 * the amount by which the marking probability is incremented when the queue
161 * overflows, or is decremented when the link is idle. d1 is set higher than
162 * d2, because link underutilization can occur when congestion management is
163 * either too conservative or too aggressive, but packet loss occurs only
164 * when congestion management is too conservative. By weighing heavily
165 * against packet loss, it can quickly reach to a substantial increase in
166 * traffic load.
167 */
0a7de745
A
168#define SFB_INCREMENT 82 /* Q14 representation of 0.005 */
169#define SFB_DECREMENT 16 /* Q14 representation of 0.001 */
316670eb 170
0a7de745
A
171#define SFB_PMARK_TH 16056 /* Q14 representation of 0.98 */
172#define SFB_PMARK_WARM 3276 /* Q14 representation of 0.2 */
316670eb 173
0a7de745
A
174#define SFB_PMARK_INC(_bin) do { \
175 (_bin)->pmark += sfb_increment; \
176 if ((_bin)->pmark > SFB_MAX_PMARK) \
177 (_bin)->pmark = SFB_MAX_PMARK; \
316670eb
A
178} while (0)
179
0a7de745
A
180#define SFB_PMARK_DEC(_bin) do { \
181 if ((_bin)->pmark > 0) { \
182 (_bin)->pmark -= sfb_decrement; \
183 if ((_bin)->pmark < 0) \
184 (_bin)->pmark = 0; \
185 } \
316670eb
A
186} while (0)
187
fe8ab488 188/* Minimum nuber of bytes in queue to get flow controlled */
0a7de745 189#define SFB_MIN_FC_THRESHOLD_BYTES 7500
fe8ab488 190
0a7de745
A
191#define SFB_SET_DELAY_HIGH(_sp_, _q_) do { \
192 (_sp_)->sfb_flags |= SFBF_DELAYHIGH; \
193 (_sp_)->sfb_fc_threshold = max(SFB_MIN_FC_THRESHOLD_BYTES, \
194 (qsize((_q_)) >> 3)); \
fe8ab488
A
195} while (0)
196
0a7de745
A
197#define SFB_QUEUE_DELAYBASED(_sp_) ((_sp_)->sfb_flags & SFBF_DELAYBASED)
198#define SFB_IS_DELAYHIGH(_sp_) ((_sp_)->sfb_flags & SFBF_DELAYHIGH)
199#define SFB_QUEUE_DELAYBASED_MAXSIZE 2048 /* max pkts */
fe8ab488 200
0a7de745
A
201#define HINTERVAL_MIN (10) /* 10 seconds */
202#define HINTERVAL_MAX (20) /* 20 seconds */
203#define SFB_HINTERVAL(sp) ((sfb_random(sp) % HINTERVAL_MAX) + HINTERVAL_MIN)
316670eb 204
0a7de745
A
205#define DEQUEUE_DECAY 7 /* ilog2 of EWMA decay rate, (128) */
206#define DEQUEUE_SPIKE(_new, _old) \
316670eb
A
207 ((u_int64_t)ABS((int64_t)(_new) - (int64_t)(_old)) > ((_old) << 11))
208
0a7de745
A
209#define SFB_ZONE_MAX 32 /* maximum elements in zone */
210#define SFB_ZONE_NAME "classq_sfb" /* zone name */
39236c6e 211
0a7de745
A
212#define SFB_BINS_ZONE_MAX 32 /* maximum elements in zone */
213#define SFB_BINS_ZONE_NAME "classq_sfb_bins" /* zone name */
39236c6e 214
0a7de745
A
215#define SFB_FCL_ZONE_MAX 32 /* maximum elements in zone */
216#define SFB_FCL_ZONE_NAME "classq_sfb_fcl" /* zone name */
316670eb
A
217
218/* Place the flow control entries in current bin on level 0 */
0a7de745 219#define SFB_FC_LEVEL 0
316670eb 220
0a7de745
A
221static unsigned int sfb_size; /* size of zone element */
222static struct zone *sfb_zone; /* zone for sfb */
316670eb 223
0a7de745
A
224static unsigned int sfb_bins_size; /* size of zone element */
225static struct zone *sfb_bins_zone; /* zone for sfb_bins */
39236c6e 226
0a7de745
A
227static unsigned int sfb_fcl_size; /* size of zone element */
228static struct zone *sfb_fcl_zone; /* zone for sfb_fc_lists */
39236c6e 229
316670eb
A
230/* internal function prototypes */
231static u_int32_t sfb_random(struct sfb *);
5ba3f43e
A
232static void *sfb_getq_flow(struct sfb *, class_queue_t *, u_int32_t, boolean_t,
233 pktsched_pkt_t *);
316670eb
A
234static void sfb_resetq(struct sfb *, cqev_t);
235static void sfb_calc_holdtime(struct sfb *, u_int64_t);
236static void sfb_calc_pboxtime(struct sfb *, u_int64_t);
237static void sfb_calc_hinterval(struct sfb *, u_int64_t *);
fe8ab488 238static void sfb_calc_update_interval(struct sfb *, u_int64_t);
316670eb 239static void sfb_swap_bins(struct sfb *, u_int32_t);
5ba3f43e
A
240static inline int sfb_pcheck(struct sfb *, uint32_t);
241static int sfb_penalize(struct sfb *, uint32_t, uint32_t *, struct timespec *);
316670eb
A
242static void sfb_adjust_bin(struct sfb *, struct sfbbinstats *,
243 struct timespec *, struct timespec *, boolean_t);
244static void sfb_decrement_bin(struct sfb *, struct sfbbinstats *,
245 struct timespec *, struct timespec *);
246static void sfb_increment_bin(struct sfb *, struct sfbbinstats *,
247 struct timespec *, struct timespec *);
5ba3f43e 248static inline void sfb_dq_update_bins(struct sfb *, uint32_t, uint32_t,
fe8ab488 249 struct timespec *, u_int32_t qsize);
5ba3f43e
A
250static inline void sfb_eq_update_bins(struct sfb *, uint32_t, uint32_t);
251static int sfb_drop_early(struct sfb *, uint32_t, u_int16_t *,
316670eb 252 struct timespec *);
5ba3f43e
A
253static boolean_t sfb_bin_addfcentry(struct sfb *, pktsched_pkt_t *,
254 uint32_t, uint8_t, uint32_t);
39236c6e 255static void sfb_fclist_append(struct sfb *, struct sfb_fcl *);
316670eb 256static void sfb_fclists_clean(struct sfb *sp);
fe8ab488
A
257static int sfb_bin_mark_or_drop(struct sfb *sp, struct sfbbinstats *bin);
258static void sfb_detect_dequeue_stall(struct sfb *sp, class_queue_t *,
259 struct timespec *);
316670eb 260
0a7de745 261SYSCTL_NODE(_net_classq, OID_AUTO, sfb, CTLFLAG_RW | CTLFLAG_LOCKED, 0, "SFB");
316670eb 262
0a7de745
A
263static u_int64_t sfb_holdtime = 0; /* 0 indicates "automatic" */
264SYSCTL_QUAD(_net_classq_sfb, OID_AUTO, holdtime, CTLFLAG_RW | CTLFLAG_LOCKED,
316670eb
A
265 &sfb_holdtime, "SFB freeze time in nanoseconds");
266
0a7de745
A
267static u_int64_t sfb_pboxtime = 0; /* 0 indicates "automatic" */
268SYSCTL_QUAD(_net_classq_sfb, OID_AUTO, pboxtime, CTLFLAG_RW | CTLFLAG_LOCKED,
316670eb
A
269 &sfb_pboxtime, "SFB penalty box time in nanoseconds");
270
271static u_int64_t sfb_hinterval;
0a7de745 272SYSCTL_QUAD(_net_classq_sfb, OID_AUTO, hinterval, CTLFLAG_RW | CTLFLAG_LOCKED,
316670eb
A
273 &sfb_hinterval, "SFB hash interval in nanoseconds");
274
275static u_int32_t sfb_increment = SFB_INCREMENT;
0a7de745 276SYSCTL_UINT(_net_classq_sfb, OID_AUTO, increment, CTLFLAG_RW | CTLFLAG_LOCKED,
316670eb
A
277 &sfb_increment, SFB_INCREMENT, "SFB increment [d1]");
278
279static u_int32_t sfb_decrement = SFB_DECREMENT;
0a7de745 280SYSCTL_UINT(_net_classq_sfb, OID_AUTO, decrement, CTLFLAG_RW | CTLFLAG_LOCKED,
316670eb
A
281 &sfb_decrement, SFB_DECREMENT, "SFB decrement [d2]");
282
0a7de745
A
283static u_int32_t sfb_allocation = 0; /* 0 means "automatic" */
284SYSCTL_UINT(_net_classq_sfb, OID_AUTO, allocation, CTLFLAG_RW | CTLFLAG_LOCKED,
316670eb
A
285 &sfb_allocation, 0, "SFB bin allocation");
286
287static u_int32_t sfb_ratelimit = 0;
0a7de745
A
288SYSCTL_UINT(_net_classq_sfb, OID_AUTO, ratelimit, CTLFLAG_RW | CTLFLAG_LOCKED,
289 &sfb_ratelimit, 0, "SFB rate limit");
316670eb 290
0a7de745
A
291#define KBPS (1ULL * 1000) /* 1 Kbits per second */
292#define MBPS (1ULL * 1000 * 1000) /* 1 Mbits per second */
293#define GBPS (MBPS * 1000) /* 1 Gbits per second */
316670eb
A
294
295struct sfb_time_tbl {
0a7de745
A
296 u_int64_t speed; /* uplink speed */
297 u_int64_t holdtime; /* hold time */
298 u_int64_t pboxtime; /* penalty box time */
316670eb
A
299};
300
301static struct sfb_time_tbl sfb_ttbl[] = {
cb323159
A
302 { .speed = 1 * MBPS, .holdtime = HOLDTIME_BASE * 1000, .pboxtime = PBOXTIME_BASE * 1000},
303 { .speed = 10 * MBPS, .holdtime = HOLDTIME_BASE * 100, .pboxtime = PBOXTIME_BASE * 100 },
304 { .speed = 100 * MBPS, .holdtime = HOLDTIME_BASE * 10, .pboxtime = PBOXTIME_BASE * 10 },
305 { .speed = 1 * GBPS, .holdtime = HOLDTIME_BASE, .pboxtime = PBOXTIME_BASE },
306 { .speed = 10 * GBPS, .holdtime = HOLDTIME_BASE / 10, .pboxtime = PBOXTIME_BASE / 10 },
307 { .speed = 100 * GBPS, .holdtime = HOLDTIME_BASE / 100, .pboxtime = PBOXTIME_BASE / 100 },
308 { .speed = 0, .holdtime = 0, .pboxtime = 0 }
316670eb
A
309};
310
311void
312sfb_init(void)
313{
314 _CASSERT(SFBF_ECN4 == CLASSQF_ECN4);
315 _CASSERT(SFBF_ECN6 == CLASSQF_ECN6);
316
0a7de745 317 sfb_size = sizeof(struct sfb);
316670eb
A
318 sfb_zone = zinit(sfb_size, SFB_ZONE_MAX * sfb_size,
319 0, SFB_ZONE_NAME);
320 if (sfb_zone == NULL) {
321 panic("%s: failed allocating %s", __func__, SFB_ZONE_NAME);
322 /* NOTREACHED */
323 }
324 zone_change(sfb_zone, Z_EXPAND, TRUE);
325 zone_change(sfb_zone, Z_CALLERACCT, TRUE);
39236c6e 326
cb323159 327 sfb_bins_size = sizeof(struct sfb_bins);
39236c6e
A
328 sfb_bins_zone = zinit(sfb_bins_size, SFB_BINS_ZONE_MAX * sfb_bins_size,
329 0, SFB_BINS_ZONE_NAME);
330 if (sfb_bins_zone == NULL) {
331 panic("%s: failed allocating %s", __func__, SFB_BINS_ZONE_NAME);
332 /* NOTREACHED */
333 }
334 zone_change(sfb_bins_zone, Z_EXPAND, TRUE);
335 zone_change(sfb_bins_zone, Z_CALLERACCT, TRUE);
336
cb323159 337 sfb_fcl_size = sizeof(struct sfb_fcl);
39236c6e
A
338 sfb_fcl_zone = zinit(sfb_fcl_size, SFB_FCL_ZONE_MAX * sfb_fcl_size,
339 0, SFB_FCL_ZONE_NAME);
340 if (sfb_fcl_zone == NULL) {
341 panic("%s: failed allocating %s", __func__, SFB_FCL_ZONE_NAME);
342 /* NOTREACHED */
343 }
344 zone_change(sfb_fcl_zone, Z_EXPAND, TRUE);
345 zone_change(sfb_fcl_zone, Z_CALLERACCT, TRUE);
316670eb
A
346}
347
348static u_int32_t
349sfb_random(struct sfb *sp)
350{
351 IFCQ_CONVERT_LOCK(&sp->sfb_ifp->if_snd);
0a7de745 352 return RandomULong();
316670eb
A
353}
354
355static void
356sfb_calc_holdtime(struct sfb *sp, u_int64_t outbw)
357{
358 u_int64_t holdtime;
359
360 if (sfb_holdtime != 0) {
361 holdtime = sfb_holdtime;
362 } else if (outbw == 0) {
363 holdtime = SFB_RANDOM(sp, HOLDTIME_MIN, HOLDTIME_MAX);
364 } else {
365 unsigned int n, i;
366
367 n = sfb_ttbl[0].holdtime;
368 for (i = 0; sfb_ttbl[i].speed != 0; i++) {
0a7de745 369 if (outbw < sfb_ttbl[i].speed) {
316670eb 370 break;
0a7de745 371 }
316670eb
A
372 n = sfb_ttbl[i].holdtime;
373 }
374 holdtime = n;
375 }
376 net_nsectimer(&holdtime, &sp->sfb_holdtime);
377}
378
379static void
380sfb_calc_pboxtime(struct sfb *sp, u_int64_t outbw)
381{
382 u_int64_t pboxtime;
383
384 if (sfb_pboxtime != 0) {
385 pboxtime = sfb_pboxtime;
386 } else if (outbw == 0) {
387 pboxtime = SFB_RANDOM(sp, PBOXTIME_MIN, PBOXTIME_MAX);
388 } else {
389 unsigned int n, i;
390
391 n = sfb_ttbl[0].pboxtime;
392 for (i = 0; sfb_ttbl[i].speed != 0; i++) {
0a7de745 393 if (outbw < sfb_ttbl[i].speed) {
316670eb 394 break;
0a7de745 395 }
316670eb
A
396 n = sfb_ttbl[i].pboxtime;
397 }
398 pboxtime = n;
399 }
400 net_nsectimer(&pboxtime, &sp->sfb_pboxtime);
401 net_timerclear(&sp->sfb_pboxfreeze);
402}
403
404static void
405sfb_calc_hinterval(struct sfb *sp, u_int64_t *t)
406{
5ba3f43e 407 u_int64_t hinterval = 0;
316670eb
A
408 struct timespec now;
409
410 if (t != NULL) {
411 /*
412 * TODO adi@apple.com: use dq_avg to derive hinterval.
413 */
414 hinterval = *t;
415 }
416
0a7de745 417 if (sfb_hinterval != 0) {
316670eb 418 hinterval = sfb_hinterval;
0a7de745 419 } else if (t == NULL || hinterval == 0) {
316670eb 420 hinterval = ((u_int64_t)SFB_HINTERVAL(sp) * NSEC_PER_SEC);
0a7de745 421 }
316670eb
A
422
423 net_nsectimer(&hinterval, &sp->sfb_hinterval);
424
425 nanouptime(&now);
426 net_timeradd(&now, &sp->sfb_hinterval, &sp->sfb_nextreset);
427}
428
fe8ab488
A
429static void
430sfb_calc_update_interval(struct sfb *sp, u_int64_t out_bw)
431{
432#pragma unused(out_bw)
433 u_int64_t update_interval = 0;
39037602 434 ifclassq_calc_update_interval(&update_interval);
fe8ab488
A
435 net_nsectimer(&update_interval, &sp->sfb_update_interval);
436}
437
316670eb
A
438/*
439 * sfb support routines
440 */
441struct sfb *
442sfb_alloc(struct ifnet *ifp, u_int32_t qid, u_int32_t qlim, u_int32_t flags)
443{
444 struct sfb *sp;
39236c6e 445 int i;
316670eb
A
446
447 VERIFY(ifp != NULL && qlim > 0);
448
449 sp = zalloc(sfb_zone);
450 if (sp == NULL) {
451 log(LOG_ERR, "%s: SFB unable to allocate\n", if_name(ifp));
0a7de745 452 return NULL;
316670eb 453 }
316670eb 454 bzero(sp, sfb_size);
39236c6e
A
455
456 if ((sp->sfb_bins = zalloc(sfb_bins_zone)) == NULL) {
316670eb
A
457 log(LOG_ERR, "%s: SFB unable to allocate bins\n", if_name(ifp));
458 sfb_destroy(sp);
0a7de745 459 return NULL;
316670eb 460 }
39236c6e 461 bzero(sp->sfb_bins, sfb_bins_size);
316670eb 462
39236c6e 463 if ((sp->sfb_fc_lists = zalloc(sfb_fcl_zone)) == NULL) {
316670eb
A
464 log(LOG_ERR, "%s: SFB unable to allocate flow control lists\n",
465 if_name(ifp));
466 sfb_destroy(sp);
0a7de745 467 return NULL;
316670eb 468 }
39236c6e
A
469 bzero(sp->sfb_fc_lists, sfb_fcl_size);
470
0a7de745 471 for (i = 0; i < SFB_BINS; ++i) {
39236c6e 472 STAILQ_INIT(&SFB_FC_LIST(sp, i)->fclist);
0a7de745 473 }
316670eb 474
316670eb
A
475 sp->sfb_ifp = ifp;
476 sp->sfb_qlim = qlim;
477 sp->sfb_qid = qid;
39236c6e
A
478 sp->sfb_flags = (flags & SFBF_USERFLAGS);
479#if !PF_ECN
480 if (sp->sfb_flags & SFBF_ECN) {
481 sp->sfb_flags &= ~SFBF_ECN;
482 log(LOG_ERR, "%s: SFB qid=%d, ECN not available; ignoring "
483 "SFBF_ECN flag!\n", if_name(ifp), sp->sfb_qid);
484 }
485#endif /* !PF_ECN */
316670eb 486
5ba3f43e 487 sfb_resetq(sp, CLASSQ_EV_INIT);
316670eb 488
0a7de745 489 return sp;
316670eb
A
490}
491
492static void
39236c6e 493sfb_fclist_append(struct sfb *sp, struct sfb_fcl *fcl)
316670eb
A
494{
495 IFCQ_CONVERT_LOCK(&sp->sfb_ifp->if_snd);
39236c6e
A
496 VERIFY(STAILQ_EMPTY(&fcl->fclist) || fcl->cnt > 0);
497 sp->sfb_stats.flow_feedback += fcl->cnt;
498 fcl->cnt = 0;
499
500 flowadv_add(&fcl->fclist);
501 VERIFY(fcl->cnt == 0 && STAILQ_EMPTY(&fcl->fclist));
316670eb
A
502}
503
504static void
505sfb_fclists_clean(struct sfb *sp)
506{
507 int i;
508
39236c6e 509 /* Move all the flow control entries to the flowadv list */
316670eb 510 for (i = 0; i < SFB_BINS; ++i) {
39236c6e 511 struct sfb_fcl *fcl = SFB_FC_LIST(sp, i);
0a7de745 512 if (!STAILQ_EMPTY(&fcl->fclist)) {
316670eb 513 sfb_fclist_append(sp, fcl);
0a7de745 514 }
316670eb
A
515 }
516}
517
518void
519sfb_destroy(struct sfb *sp)
520{
521 sfb_fclists_clean(sp);
522 if (sp->sfb_bins != NULL) {
39236c6e 523 zfree(sfb_bins_zone, sp->sfb_bins);
316670eb
A
524 sp->sfb_bins = NULL;
525 }
526 if (sp->sfb_fc_lists != NULL) {
39236c6e 527 zfree(sfb_fcl_zone, sp->sfb_fc_lists);
316670eb
A
528 sp->sfb_fc_lists = NULL;
529 }
530 zfree(sfb_zone, sp);
531}
532
533static void
534sfb_resetq(struct sfb *sp, cqev_t ev)
535{
536 struct ifnet *ifp = sp->sfb_ifp;
537 u_int64_t eff_rate;
538
539 VERIFY(ifp != NULL);
540
541 if (ev != CLASSQ_EV_LINK_DOWN) {
542 (*sp->sfb_bins)[0].fudge = sfb_random(sp);
543 (*sp->sfb_bins)[1].fudge = sfb_random(sp);
544 sp->sfb_allocation = ((sfb_allocation == 0) ?
545 (sp->sfb_qlim / 3) : sfb_allocation);
546 sp->sfb_drop_thresh = sp->sfb_allocation +
547 (sp->sfb_allocation >> 1);
548 }
549
550 sp->sfb_clearpkts = 0;
551 sp->sfb_current = 0;
552
553 eff_rate = ifnet_output_linkrate(ifp);
554 sp->sfb_eff_rate = eff_rate;
555
556 sfb_calc_holdtime(sp, eff_rate);
557 sfb_calc_pboxtime(sp, eff_rate);
558 sfb_calc_hinterval(sp, NULL);
39037602 559 ifclassq_calc_target_qdelay(ifp, &sp->sfb_target_qdelay);
fe8ab488 560 sfb_calc_update_interval(sp, eff_rate);
316670eb
A
561
562 if (ev == CLASSQ_EV_LINK_DOWN ||
0a7de745 563 ev == CLASSQ_EV_LINK_UP) {
316670eb 564 sfb_fclists_clean(sp);
0a7de745 565 }
316670eb 566
0a7de745
A
567 bzero(sp->sfb_bins, sizeof(*sp->sfb_bins));
568 bzero(&sp->sfb_stats, sizeof(sp->sfb_stats));
316670eb 569
0a7de745 570 if (ev == CLASSQ_EV_LINK_DOWN || !classq_verbose) {
316670eb 571 return;
0a7de745 572 }
316670eb
A
573
574 log(LOG_DEBUG, "%s: SFB qid=%d, holdtime=%llu nsec, "
575 "pboxtime=%llu nsec, allocation=%d, drop_thresh=%d, "
fe8ab488
A
576 "hinterval=%d sec, sfb_bins=%d bytes, eff_rate=%llu bps"
577 "target_qdelay= %llu nsec "
578 "update_interval=%llu sec %llu nsec flags=0x%x\n",
316670eb
A
579 if_name(ifp), sp->sfb_qid, (u_int64_t)sp->sfb_holdtime.tv_nsec,
580 (u_int64_t)sp->sfb_pboxtime.tv_nsec,
581 (u_int32_t)sp->sfb_allocation, (u_int32_t)sp->sfb_drop_thresh,
0a7de745 582 (int)sp->sfb_hinterval.tv_sec, (int)sizeof(*sp->sfb_bins),
fe8ab488
A
583 eff_rate, (u_int64_t)sp->sfb_target_qdelay,
584 (u_int64_t)sp->sfb_update_interval.tv_sec,
585 (u_int64_t)sp->sfb_update_interval.tv_nsec, sp->sfb_flags);
316670eb
A
586}
587
588void
589sfb_getstats(struct sfb *sp, struct sfb_stats *sps)
590{
591 sps->allocation = sp->sfb_allocation;
592 sps->dropthresh = sp->sfb_drop_thresh;
593 sps->clearpkts = sp->sfb_clearpkts;
594 sps->current = sp->sfb_current;
fe8ab488
A
595 sps->target_qdelay = sp->sfb_target_qdelay;
596 sps->min_estdelay = sp->sfb_min_qdelay;
597 sps->delay_fcthreshold = sp->sfb_fc_threshold;
598 sps->flags = sp->sfb_flags;
316670eb
A
599
600 net_timernsec(&sp->sfb_holdtime, &sp->sfb_stats.hold_time);
601 net_timernsec(&sp->sfb_pboxtime, &sp->sfb_stats.pbox_time);
602 net_timernsec(&sp->sfb_hinterval, &sp->sfb_stats.rehash_intval);
fe8ab488 603 net_timernsec(&sp->sfb_update_interval, &sps->update_interval);
316670eb
A
604 *(&(sps->sfbstats)) = *(&(sp->sfb_stats));
605
0a7de745
A
606 _CASSERT(sizeof((*sp->sfb_bins)[0].stats) ==
607 sizeof(sps->binstats[0].stats));
316670eb
A
608
609 bcopy(&(*sp->sfb_bins)[0].stats, &sps->binstats[0].stats,
0a7de745 610 sizeof(sps->binstats[0].stats));
316670eb 611 bcopy(&(*sp->sfb_bins)[1].stats, &sps->binstats[1].stats,
0a7de745 612 sizeof(sps->binstats[1].stats));
316670eb
A
613}
614
615static void
616sfb_swap_bins(struct sfb *sp, u_int32_t len)
617{
618 int i, j, s;
619
0a7de745 620 if (sp->sfb_flags & SFBF_SUSPENDED) {
316670eb 621 return;
0a7de745 622 }
316670eb
A
623
624 s = sp->sfb_current;
625 VERIFY((s + (s ^ 1)) == 1);
626
627 (*sp->sfb_bins)[s].fudge = sfb_random(sp); /* recompute perturbation */
628 sp->sfb_clearpkts = len;
629 sp->sfb_stats.num_rehash++;
630
0a7de745 631 s = (sp->sfb_current ^= 1); /* flip the bit (swap current) */
316670eb
A
632
633 if (classq_verbose) {
634 log(LOG_DEBUG, "%s: SFB qid=%d, set %d is now current, "
635 "qlen=%d\n", if_name(sp->sfb_ifp), sp->sfb_qid, s, len);
636 }
637
638 /* clear freezetime for all current bins */
639 bzero(&(*sp->sfb_bins)[s].freezetime,
0a7de745 640 sizeof((*sp->sfb_bins)[s].freezetime));
316670eb
A
641
642 /* clear/adjust bin statistics and flow control lists */
643 for (i = 0; i < SFB_BINS; i++) {
39236c6e 644 struct sfb_fcl *fcl = SFB_FC_LIST(sp, i);
316670eb 645
0a7de745 646 if (!STAILQ_EMPTY(&fcl->fclist)) {
316670eb 647 sfb_fclist_append(sp, fcl);
0a7de745 648 }
316670eb
A
649
650 for (j = 0; j < SFB_LEVELS; j++) {
651 struct sfbbinstats *cbin, *wbin;
652
0a7de745
A
653 cbin = SFB_BINST(sp, j, i, s); /* current */
654 wbin = SFB_BINST(sp, j, i, s ^ 1); /* warm-up */
316670eb
A
655
656 cbin->pkts = 0;
fe8ab488 657 cbin->bytes = 0;
0a7de745 658 if (cbin->pmark > SFB_MAX_PMARK) {
316670eb 659 cbin->pmark = SFB_MAX_PMARK;
0a7de745
A
660 }
661 if (cbin->pmark < 0) {
316670eb 662 cbin->pmark = 0;
0a7de745 663 }
316670eb
A
664
665 /*
666 * Keep pmark from before to identify
667 * non-responsives immediately.
668 */
0a7de745 669 if (wbin->pmark > SFB_PMARK_WARM) {
316670eb 670 wbin->pmark = SFB_PMARK_WARM;
0a7de745 671 }
316670eb
A
672 }
673 }
674}
675
676static inline int
5ba3f43e 677sfb_pcheck(struct sfb *sp, uint32_t pkt_sfb_hash)
316670eb
A
678{
679#if SFB_LEVELS != 2
680 int i, n;
681#endif /* SFB_LEVELS != 2 */
5ba3f43e 682 uint8_t *pkt_sfb_hash8 = (uint8_t *)&pkt_sfb_hash;
316670eb
A
683 int s;
684
685 s = sp->sfb_current;
686 VERIFY((s + (s ^ 1)) == 1);
687
688 /*
689 * For current bins, returns 1 if all pmark >= SFB_PMARK_TH,
690 * 0 otherwise; optimize for SFB_LEVELS=2.
691 */
692#if SFB_LEVELS == 2
693 /*
694 * Level 0: bin index at [0] for set 0; [2] for set 1
695 * Level 1: bin index at [1] for set 0; [3] for set 1
696 */
5ba3f43e 697 if (SFB_BINST(sp, 0, SFB_BINMASK(pkt_sfb_hash8[(s << 1)]),
316670eb 698 s)->pmark < SFB_PMARK_TH ||
5ba3f43e 699 SFB_BINST(sp, 1, SFB_BINMASK(pkt_sfb_hash8[(s << 1) + 1]),
0a7de745
A
700 s)->pmark < SFB_PMARK_TH) {
701 return 0;
702 }
316670eb
A
703#else /* SFB_LEVELS != 2 */
704 for (i = 0; i < SFB_LEVELS; i++) {
0a7de745 705 if (s == 0) { /* set 0, bin index [0,1] */
5ba3f43e 706 n = SFB_BINMASK(pkt_sfb_hash8[i]);
0a7de745 707 } else { /* set 1, bin index [2,3] */
5ba3f43e 708 n = SFB_BINMASK(pkt_sfb_hash8[i + 2]);
0a7de745 709 }
316670eb 710
0a7de745
A
711 if (SFB_BINST(sp, i, n, s)->pmark < SFB_PMARK_TH) {
712 return 0;
713 }
316670eb
A
714 }
715#endif /* SFB_LEVELS != 2 */
0a7de745 716 return 1;
316670eb
A
717}
718
719static int
5ba3f43e
A
720sfb_penalize(struct sfb *sp, uint32_t pkt_sfb_hash, uint32_t *pkt_sfb_flags,
721 struct timespec *now)
316670eb 722{
cb323159 723 struct timespec delta = { .tv_sec = 0, .tv_nsec = 0 };
5ba3f43e 724 uint8_t *pkt_sfb_hash8 = (uint8_t *)&pkt_sfb_hash;
316670eb
A
725
726 /* If minimum pmark of current bins is < SFB_PMARK_TH, we're done */
0a7de745
A
727 if (!sfb_ratelimit || !sfb_pcheck(sp, pkt_sfb_hash)) {
728 return 0;
729 }
316670eb
A
730
731 net_timersub(now, &sp->sfb_pboxfreeze, &delta);
732 if (net_timercmp(&delta, &sp->sfb_pboxtime, <)) {
733#if SFB_LEVELS != 2
734 int i;
735#endif /* SFB_LEVELS != 2 */
736 struct sfbbinstats *bin;
737 int n, w;
738
739 w = sp->sfb_current ^ 1;
740 VERIFY((w + (w ^ 1)) == 1);
741
742 /*
743 * Update warm-up bins; optimize for SFB_LEVELS=2
744 */
745#if SFB_LEVELS == 2
746 /* Level 0: bin index at [0] for set 0; [2] for set 1 */
5ba3f43e 747 n = SFB_BINMASK(pkt_sfb_hash8[(w << 1)]);
316670eb 748 bin = SFB_BINST(sp, 0, n, w);
0a7de745 749 if (bin->pkts >= sp->sfb_allocation) {
316670eb 750 sfb_increment_bin(sp, bin, SFB_BINFT(sp, 0, n, w), now);
0a7de745 751 }
316670eb
A
752
753 /* Level 0: bin index at [1] for set 0; [3] for set 1 */
5ba3f43e 754 n = SFB_BINMASK(pkt_sfb_hash8[(w << 1) + 1]);
316670eb 755 bin = SFB_BINST(sp, 1, n, w);
0a7de745 756 if (bin->pkts >= sp->sfb_allocation) {
316670eb 757 sfb_increment_bin(sp, bin, SFB_BINFT(sp, 1, n, w), now);
0a7de745 758 }
316670eb
A
759#else /* SFB_LEVELS != 2 */
760 for (i = 0; i < SFB_LEVELS; i++) {
0a7de745 761 if (w == 0) { /* set 0, bin index [0,1] */
5ba3f43e 762 n = SFB_BINMASK(pkt_sfb_hash8[i]);
0a7de745 763 } else { /* set 1, bin index [2,3] */
5ba3f43e 764 n = SFB_BINMASK(pkt_sfb_hash8[i + 2]);
0a7de745 765 }
316670eb
A
766
767 bin = SFB_BINST(sp, i, n, w);
768 if (bin->pkts >= sp->sfb_allocation) {
769 sfb_increment_bin(sp, bin,
770 SFB_BINFT(sp, i, n, w), now);
771 }
772 }
773#endif /* SFB_LEVELS != 2 */
0a7de745 774 return 1;
316670eb
A
775 }
776
777 /* non-conformant or else misclassified flow; queue it anyway */
5ba3f43e 778 *pkt_sfb_flags |= SFB_PKT_PBOX;
316670eb
A
779 *(&sp->sfb_pboxfreeze) = *now;
780
0a7de745 781 return 0;
316670eb
A
782}
783
784static void
785sfb_adjust_bin(struct sfb *sp, struct sfbbinstats *bin, struct timespec *ft,
786 struct timespec *now, boolean_t inc)
787{
788 struct timespec delta;
789
790 net_timersub(now, ft, &delta);
791 if (net_timercmp(&delta, &sp->sfb_holdtime, <)) {
792 if (classq_verbose > 1) {
793 log(LOG_DEBUG, "%s: SFB qid=%d, %s update frozen "
794 "(delta=%llu nsec)\n", if_name(sp->sfb_ifp),
795 sp->sfb_qid, inc ? "increment" : "decrement",
796 (u_int64_t)delta.tv_nsec);
797 }
798 return;
799 }
800
801 /* increment/decrement marking probability */
802 *ft = *now;
0a7de745 803 if (inc) {
316670eb 804 SFB_PMARK_INC(bin);
0a7de745 805 } else {
316670eb 806 SFB_PMARK_DEC(bin);
0a7de745 807 }
316670eb
A
808}
809
810static void
811sfb_decrement_bin(struct sfb *sp, struct sfbbinstats *bin, struct timespec *ft,
812 struct timespec *now)
813{
0a7de745 814 return sfb_adjust_bin(sp, bin, ft, now, FALSE);
316670eb
A
815}
816
817static void
818sfb_increment_bin(struct sfb *sp, struct sfbbinstats *bin, struct timespec *ft,
819 struct timespec *now)
820{
0a7de745 821 return sfb_adjust_bin(sp, bin, ft, now, TRUE);
316670eb
A
822}
823
824static inline void
5ba3f43e 825sfb_dq_update_bins(struct sfb *sp, uint32_t pkt_sfb_hash, uint32_t pkt_len,
fe8ab488 826 struct timespec *now, u_int32_t qsize)
316670eb
A
827{
828#if SFB_LEVELS != 2 || SFB_FC_LEVEL != 0
829 int i;
830#endif /* SFB_LEVELS != 2 || SFB_FC_LEVEL != 0 */
831 struct sfbbinstats *bin;
832 int s, n;
39236c6e 833 struct sfb_fcl *fcl = NULL;
5ba3f43e 834 uint8_t *pkt_sfb_hash8 = (uint8_t *)&pkt_sfb_hash;
316670eb
A
835
836 s = sp->sfb_current;
837 VERIFY((s + (s ^ 1)) == 1);
838
839 /*
840 * Update current bins; optimize for SFB_LEVELS=2 and SFB_FC_LEVEL=0
841 */
842#if SFB_LEVELS == 2 && SFB_FC_LEVEL == 0
843 /* Level 0: bin index at [0] for set 0; [2] for set 1 */
5ba3f43e 844 n = SFB_BINMASK(pkt_sfb_hash8[(s << 1)]);
316670eb
A
845 bin = SFB_BINST(sp, 0, n, s);
846
5ba3f43e 847 VERIFY(bin->pkts > 0 && bin->bytes >= pkt_len);
fe8ab488 848 bin->pkts--;
5ba3f43e 849 bin->bytes -= pkt_len;
fe8ab488 850
0a7de745 851 if (bin->pkts == 0) {
316670eb 852 sfb_decrement_bin(sp, bin, SFB_BINFT(sp, 0, n, s), now);
0a7de745 853 }
fe8ab488
A
854
855 /* Deliver flow control feedback to the sockets */
856 if (SFB_QUEUE_DELAYBASED(sp)) {
857 if (!(SFB_IS_DELAYHIGH(sp)) ||
858 bin->bytes <= sp->sfb_fc_threshold ||
0a7de745 859 bin->pkts == 0 || qsize == 0) {
fe8ab488 860 fcl = SFB_FC_LIST(sp, n);
0a7de745 861 }
fe8ab488 862 } else if (bin->pkts <= (sp->sfb_allocation >> 2)) {
0a7de745 863 fcl = SFB_FC_LIST(sp, n);
316670eb 864 }
fe8ab488 865
0a7de745 866 if (fcl != NULL && !STAILQ_EMPTY(&fcl->fclist)) {
fe8ab488 867 sfb_fclist_append(sp, fcl);
0a7de745 868 }
fe8ab488 869 fcl = NULL;
316670eb
A
870
871 /* Level 1: bin index at [1] for set 0; [3] for set 1 */
5ba3f43e 872 n = SFB_BINMASK(pkt_sfb_hash8[(s << 1) + 1]);
316670eb
A
873 bin = SFB_BINST(sp, 1, n, s);
874
5ba3f43e 875 VERIFY(bin->pkts > 0 && bin->bytes >= (u_int64_t)pkt_len);
fe8ab488 876 bin->pkts--;
5ba3f43e 877 bin->bytes -= pkt_len;
0a7de745 878 if (bin->pkts == 0) {
316670eb 879 sfb_decrement_bin(sp, bin, SFB_BINFT(sp, 1, n, s), now);
0a7de745 880 }
316670eb
A
881#else /* SFB_LEVELS != 2 || SFB_FC_LEVEL != 0 */
882 for (i = 0; i < SFB_LEVELS; i++) {
0a7de745 883 if (s == 0) { /* set 0, bin index [0,1] */
5ba3f43e 884 n = SFB_BINMASK(pkt_sfb_hash8[i]);
0a7de745 885 } else { /* set 1, bin index [2,3] */
5ba3f43e 886 n = SFB_BINMASK(pkt_sfb_hash8[i + 2]);
0a7de745 887 }
316670eb
A
888
889 bin = SFB_BINST(sp, i, n, s);
890
5ba3f43e 891 VERIFY(bin->pkts > 0 && bin->bytes >= pkt_len);
fe8ab488 892 bin->pkts--;
5ba3f43e 893 bin->bytes -= pkt_len;
0a7de745 894 if (bin->pkts == 0) {
316670eb
A
895 sfb_decrement_bin(sp, bin,
896 SFB_BINFT(sp, i, n, s), now);
0a7de745
A
897 }
898 if (i != SFB_FC_LEVEL) {
fe8ab488 899 continue;
0a7de745 900 }
fe8ab488
A
901 if (SFB_QUEUE_DELAYBASED(sp)) {
902 if (!(SFB_IS_DELAYHIGH(sp)) ||
0a7de745 903 bin->bytes <= sp->sfb_fc_threshold) {
316670eb 904 fcl = SFB_FC_LIST(sp, n);
0a7de745 905 }
fe8ab488
A
906 } else if (bin->pkts <= (sp->sfb_allocation >> 2)) {
907 fcl = SFB_FC_LIST(sp, n);
316670eb 908 }
0a7de745 909 if (fcl != NULL && !STAILQ_EMPTY(&fcl->fclist)) {
fe8ab488 910 sfb_fclist_append(sp, fcl);
0a7de745 911 }
fe8ab488 912 fcl = NULL;
316670eb
A
913 }
914#endif /* SFB_LEVELS != 2 || SFB_FC_LEVEL != 0 */
915}
916
917static inline void
5ba3f43e 918sfb_eq_update_bins(struct sfb *sp, uint32_t pkt_sfb_hash, uint32_t pkt_len)
316670eb
A
919{
920#if SFB_LEVELS != 2
921 int i, n;
922#endif /* SFB_LEVELS != 2 */
923 int s;
fe8ab488 924 struct sfbbinstats *bin;
5ba3f43e 925 uint8_t *pkt_sfb_hash8 = (uint8_t *)&pkt_sfb_hash;
316670eb
A
926 s = sp->sfb_current;
927 VERIFY((s + (s ^ 1)) == 1);
928
929 /*
930 * Update current bins; optimize for SFB_LEVELS=2
931 */
932#if SFB_LEVELS == 2
933 /* Level 0: bin index at [0] for set 0; [2] for set 1 */
fe8ab488 934 bin = SFB_BINST(sp, 0,
5ba3f43e 935 SFB_BINMASK(pkt_sfb_hash8[(s << 1)]), s);
fe8ab488 936 bin->pkts++;
5ba3f43e 937 bin->bytes += pkt_len;
316670eb
A
938
939 /* Level 1: bin index at [1] for set 0; [3] for set 1 */
fe8ab488 940 bin = SFB_BINST(sp, 1,
5ba3f43e 941 SFB_BINMASK(pkt_sfb_hash8[(s << 1) + 1]), s);
fe8ab488 942 bin->pkts++;
5ba3f43e 943 bin->bytes += pkt_len;
fe8ab488 944
316670eb
A
945#else /* SFB_LEVELS != 2 */
946 for (i = 0; i < SFB_LEVELS; i++) {
0a7de745 947 if (s == 0) { /* set 0, bin index [0,1] */
5ba3f43e 948 n = SFB_BINMASK(pkt_sfb_hash8[i]);
0a7de745 949 } else { /* set 1, bin index [2,3] */
5ba3f43e 950 n = SFB_BINMASK(pkt_sfb_hash8[i + 2]);
0a7de745 951 }
316670eb 952
fe8ab488
A
953 bin = SFB_BINST(sp, i, n, s);
954 bin->pkts++;
5ba3f43e 955 bin->bytes += pkt_len;
316670eb
A
956 }
957#endif /* SFB_LEVELS != 2 */
958}
959
960static boolean_t
5ba3f43e
A
961sfb_bin_addfcentry(struct sfb *sp, pktsched_pkt_t *pkt, uint32_t pkt_sfb_hash,
962 uint8_t flowsrc, uint32_t flowid)
316670eb 963{
39236c6e 964 struct flowadv_fcentry *fce;
39236c6e 965 struct sfb_fcl *fcl;
316670eb 966 int s;
5ba3f43e 967 uint8_t *pkt_sfb_hash8 = (uint8_t *)&pkt_sfb_hash;
316670eb
A
968
969 s = sp->sfb_current;
970 VERIFY((s + (s ^ 1)) == 1);
971
39236c6e
A
972 if (flowid == 0) {
973 sp->sfb_stats.null_flowid++;
0a7de745 974 return FALSE;
316670eb
A
975 }
976
977 /*
978 * Use value at index 0 for set 0 and
979 * value at index 2 for set 1
980 */
5ba3f43e 981 fcl = SFB_FC_LIST(sp, SFB_BINMASK(pkt_sfb_hash8[(s << 1)]));
39236c6e 982 STAILQ_FOREACH(fce, &fcl->fclist, fce_link) {
5ba3f43e 983 if ((uint8_t)fce->fce_flowsrc_type == flowsrc &&
39236c6e 984 fce->fce_flowid == flowid) {
316670eb 985 /* Already on flow control list; just return */
0a7de745 986 return TRUE;
316670eb
A
987 }
988 }
989
990 IFCQ_CONVERT_LOCK(&sp->sfb_ifp->if_snd);
5ba3f43e 991 fce = pktsched_alloc_fcentry(pkt, sp->sfb_ifp, M_WAITOK);
316670eb 992 if (fce != NULL) {
39236c6e
A
993 STAILQ_INSERT_TAIL(&fcl->fclist, fce, fce_link);
994 fcl->cnt++;
316670eb
A
995 sp->sfb_stats.flow_controlled++;
996 }
997
0a7de745 998 return fce != NULL;
316670eb
A
999}
1000
fe8ab488
A
1001/*
1002 * check if this flow needs to be flow-controlled or if this
1003 * packet needs to be dropped.
1004 */
1005static int
1006sfb_bin_mark_or_drop(struct sfb *sp, struct sfbbinstats *bin)
1007{
1008 int ret = 0;
1009 if (SFB_QUEUE_DELAYBASED(sp)) {
1010 /*
1011 * Mark or drop if this bin has more
1012 * bytes than the flowcontrol threshold.
1013 */
1014 if (SFB_IS_DELAYHIGH(sp) &&
0a7de745 1015 bin->bytes >= (sp->sfb_fc_threshold << 1)) {
fe8ab488 1016 ret = 1;
0a7de745 1017 }
fe8ab488
A
1018 } else {
1019 if (bin->pkts >= sp->sfb_allocation &&
0a7de745
A
1020 bin->pkts >= sp->sfb_drop_thresh) {
1021 ret = 1; /* drop or mark */
1022 }
fe8ab488 1023 }
0a7de745 1024 return ret;
fe8ab488
A
1025}
1026
316670eb
A
1027/*
1028 * early-drop probability is kept in pmark of each bin of the flow
1029 */
1030static int
5ba3f43e 1031sfb_drop_early(struct sfb *sp, uint32_t pkt_sfb_hash, u_int16_t *pmin,
316670eb
A
1032 struct timespec *now)
1033{
1034#if SFB_LEVELS != 2
1035 int i;
1036#endif /* SFB_LEVELS != 2 */
1037 struct sfbbinstats *bin;
1038 int s, n, ret = 0;
5ba3f43e 1039 uint8_t *pkt_sfb_hash8 = (uint8_t *)&pkt_sfb_hash;
316670eb
A
1040
1041 s = sp->sfb_current;
1042 VERIFY((s + (s ^ 1)) == 1);
1043
1044 *pmin = (u_int16_t)-1;
1045
1046 /*
1047 * Update current bins; optimize for SFB_LEVELS=2
1048 */
1049#if SFB_LEVELS == 2
1050 /* Level 0: bin index at [0] for set 0; [2] for set 1 */
5ba3f43e 1051 n = SFB_BINMASK(pkt_sfb_hash8[(s << 1)]);
316670eb 1052 bin = SFB_BINST(sp, 0, n, s);
0a7de745 1053 if (*pmin > (u_int16_t)bin->pmark) {
316670eb 1054 *pmin = (u_int16_t)bin->pmark;
0a7de745 1055 }
316670eb 1056
fe8ab488
A
1057
1058 /* Update SFB probability */
0a7de745 1059 if (bin->pkts >= sp->sfb_allocation) {
316670eb 1060 sfb_increment_bin(sp, bin, SFB_BINFT(sp, 0, n, s), now);
0a7de745 1061 }
fe8ab488
A
1062
1063 ret = sfb_bin_mark_or_drop(sp, bin);
316670eb
A
1064
1065 /* Level 1: bin index at [1] for set 0; [3] for set 1 */
5ba3f43e 1066 n = SFB_BINMASK(pkt_sfb_hash8[(s << 1) + 1]);
316670eb 1067 bin = SFB_BINST(sp, 1, n, s);
0a7de745 1068 if (*pmin > (u_int16_t)bin->pmark) {
316670eb 1069 *pmin = (u_int16_t)bin->pmark;
0a7de745 1070 }
316670eb 1071
0a7de745 1072 if (bin->pkts >= sp->sfb_allocation) {
316670eb 1073 sfb_increment_bin(sp, bin, SFB_BINFT(sp, 1, n, s), now);
0a7de745 1074 }
316670eb
A
1075#else /* SFB_LEVELS != 2 */
1076 for (i = 0; i < SFB_LEVELS; i++) {
0a7de745 1077 if (s == 0) { /* set 0, bin index [0,1] */
5ba3f43e 1078 n = SFB_BINMASK(pkt_sfb_hash8[i]);
0a7de745 1079 } else { /* set 1, bin index [2,3] */
5ba3f43e 1080 n = SFB_BINMASK(pkt_sfb_hash8[i + 2]);
0a7de745 1081 }
316670eb
A
1082
1083 bin = SFB_BINST(sp, i, n, s);
0a7de745 1084 if (*pmin > (u_int16_t)bin->pmark) {
316670eb 1085 *pmin = (u_int16_t)bin->pmark;
0a7de745 1086 }
316670eb 1087
0a7de745 1088 if (bin->pkts >= sp->sfb_allocation) {
316670eb
A
1089 sfb_increment_bin(sp, bin,
1090 SFB_BINFT(sp, i, n, s), now);
0a7de745
A
1091 }
1092 if (i == SFB_FC_LEVEL) {
fe8ab488 1093 ret = sfb_bin_mark_or_drop(sp, bin);
0a7de745 1094 }
316670eb
A
1095 }
1096#endif /* SFB_LEVELS != 2 */
1097
0a7de745
A
1098 if (sp->sfb_flags & SFBF_SUSPENDED) {
1099 ret = 1; /* drop or mark */
1100 }
1101 return ret;
316670eb
A
1102}
1103
fe8ab488
A
1104void
1105sfb_detect_dequeue_stall(struct sfb *sp, class_queue_t *q,
1106 struct timespec *now)
1107{
1108 struct timespec max_getqtime;
1109
1110 if (!SFB_QUEUE_DELAYBASED(sp) || SFB_IS_DELAYHIGH(sp) ||
1111 qsize(q) <= SFB_MIN_FC_THRESHOLD_BYTES ||
0a7de745 1112 !net_timerisset(&sp->sfb_getqtime)) {
fe8ab488 1113 return;
0a7de745 1114 }
fe8ab488
A
1115
1116 net_timeradd(&sp->sfb_getqtime, &sp->sfb_update_interval,
1117 &max_getqtime);
1118 if (net_timercmp(now, &max_getqtime, >)) {
1119 /*
1120 * No packets have been dequeued in an update interval
1121 * worth of time. It means that the queue is stalled
1122 */
1123 SFB_SET_DELAY_HIGH(sp, q);
1124 sp->sfb_stats.dequeue_stall++;
1125 }
1126}
1127
0a7de745
A
1128#define DTYPE_NODROP 0 /* no drop */
1129#define DTYPE_FORCED 1 /* a "forced" drop */
1130#define DTYPE_EARLY 2 /* an "unforced" (early) drop */
316670eb
A
1131
1132int
5ba3f43e
A
1133sfb_addq(struct sfb *sp, class_queue_t *q, pktsched_pkt_t *pkt,
1134 struct pf_mtag *t)
316670eb 1135{
39236c6e
A
1136#if !PF_ECN
1137#pragma unused(t)
1138#endif /* !PF_ECN */
316670eb
A
1139 struct timespec now;
1140 int droptype, s;
5ba3f43e 1141 uint16_t pmin;
316670eb
A
1142 int fc_adv = 0;
1143 int ret = CLASSQEQ_SUCCESS;
5ba3f43e
A
1144 uint32_t maxqsize = 0;
1145 uint64_t *pkt_timestamp;
1146 uint32_t *pkt_sfb_hash;
1147 uint16_t *pkt_sfb_hash16;
1148 uint32_t *pkt_sfb_flags;
1149 uint32_t pkt_flowid;
cb323159 1150 volatile uint32_t *pkt_flags;
5ba3f43e 1151 uint8_t pkt_proto, pkt_flowsrc;
316670eb
A
1152
1153 s = sp->sfb_current;
1154 VERIFY((s + (s ^ 1)) == 1);
1155
5ba3f43e
A
1156 pktsched_get_pkt_vars(pkt, &pkt_flags, &pkt_timestamp, &pkt_flowid,
1157 &pkt_flowsrc, &pkt_proto, NULL);
1158 pkt_sfb_hash = pktsched_get_pkt_sfb_vars(pkt, &pkt_sfb_flags);
1159 pkt_sfb_hash16 = (uint16_t *)pkt_sfb_hash;
fe8ab488 1160
cb323159
A
1161 switch (pkt->pktsched_ptype) {
1162 case QP_MBUF:
5ba3f43e
A
1163 /* See comments in <rdar://problem/14040693> */
1164 VERIFY(!(*pkt_flags & PKTF_PRIV_GUARDED));
1165 *pkt_flags |= PKTF_PRIV_GUARDED;
cb323159
A
1166 break;
1167 default:
1168 VERIFY(0);
1169 /* NOTREACHED */
1170 __builtin_unreachable();
5ba3f43e
A
1171 }
1172
1173 if (*pkt_timestamp > 0) {
1174 net_nsectimer(pkt_timestamp, &now);
3e170ce0
A
1175 } else {
1176 nanouptime(&now);
5ba3f43e 1177 net_timernsec(&now, pkt_timestamp);
3e170ce0
A
1178 }
1179
316670eb
A
1180 /* time to swap the bins? */
1181 if (net_timercmp(&now, &sp->sfb_nextreset, >=)) {
1182 net_timeradd(&now, &sp->sfb_hinterval, &sp->sfb_nextreset);
1183 sfb_swap_bins(sp, qlen(q));
1184 s = sp->sfb_current;
1185 VERIFY((s + (s ^ 1)) == 1);
1186 }
1187
fe8ab488
A
1188 if (!net_timerisset(&sp->sfb_update_time)) {
1189 net_timeradd(&now, &sp->sfb_update_interval,
1190 &sp->sfb_update_time);
1191 }
1192
3e170ce0
A
1193 /*
1194 * If getq time is not set because this is the first packet
1195 * or after idle time, set it now so that we can detect a stall.
1196 */
0a7de745 1197 if (qsize(q) == 0 && !net_timerisset(&sp->sfb_getqtime)) {
3e170ce0 1198 *(&sp->sfb_getqtime) = *(&now);
0a7de745 1199 }
3e170ce0 1200
5ba3f43e
A
1201 *pkt_sfb_flags = 0;
1202 pkt_sfb_hash16[s] =
0a7de745 1203 (SFB_HASH(&pkt_flowid, sizeof(pkt_flowid),
316670eb 1204 (*sp->sfb_bins)[s].fudge) & SFB_HASHMASK);
5ba3f43e 1205 pkt_sfb_hash16[s ^ 1] =
0a7de745 1206 (SFB_HASH(&pkt_flowid, sizeof(pkt_flowid),
316670eb
A
1207 (*sp->sfb_bins)[s ^ 1].fudge) & SFB_HASHMASK);
1208
fe8ab488
A
1209 /* check if the queue has been stalled */
1210 sfb_detect_dequeue_stall(sp, q, &now);
1211
316670eb
A
1212 /* see if we drop early */
1213 droptype = DTYPE_NODROP;
5ba3f43e 1214 if (sfb_drop_early(sp, *pkt_sfb_hash, &pmin, &now)) {
316670eb
A
1215 /* flow control, mark or drop by sfb */
1216 if ((sp->sfb_flags & SFBF_FLOWCTL) &&
5ba3f43e 1217 (*pkt_flags & PKTF_FLOW_ADV)) {
316670eb
A
1218 fc_adv = 1;
1219 /* drop all during suspension or for non-TCP */
1220 if ((sp->sfb_flags & SFBF_SUSPENDED) ||
5ba3f43e 1221 pkt_proto != IPPROTO_TCP) {
316670eb
A
1222 droptype = DTYPE_EARLY;
1223 sp->sfb_stats.drop_early++;
1224 }
39236c6e
A
1225 }
1226#if PF_ECN
5ba3f43e 1227 /* XXX: only supported for mbuf */
39236c6e 1228 else if ((sp->sfb_flags & SFBF_ECN) &&
5ba3f43e
A
1229 (pkt->pktsched_ptype == QP_MBUF) &&
1230 (pkt_proto == IPPROTO_TCP) && /* only for TCP */
316670eb
A
1231 ((sfb_random(sp) & SFB_MAX_PMARK) <= pmin) &&
1232 mark_ecn(m, t, sp->sfb_flags) &&
1233 !(sp->sfb_flags & SFBF_SUSPENDED)) {
1234 /* successfully marked; do not drop. */
1235 sp->sfb_stats.marked_packets++;
39236c6e
A
1236 }
1237#endif /* PF_ECN */
1238 else {
316670eb
A
1239 /* unforced drop by sfb */
1240 droptype = DTYPE_EARLY;
1241 sp->sfb_stats.drop_early++;
1242 }
1243 }
1244
1245 /* non-responsive flow penalty? */
5ba3f43e
A
1246 if (droptype == DTYPE_NODROP && sfb_penalize(sp, *pkt_sfb_hash,
1247 pkt_sfb_flags, &now)) {
316670eb
A
1248 droptype = DTYPE_FORCED;
1249 sp->sfb_stats.drop_pbox++;
1250 }
1251
0a7de745 1252 if (SFB_QUEUE_DELAYBASED(sp)) {
3e170ce0 1253 maxqsize = SFB_QUEUE_DELAYBASED_MAXSIZE;
0a7de745 1254 } else {
3e170ce0 1255 maxqsize = qlimit(q);
0a7de745 1256 }
fe8ab488
A
1257
1258 /*
3e170ce0
A
1259 * When the queue length hits the queue limit, make it a forced
1260 * drop
fe8ab488 1261 */
3e170ce0 1262 if (droptype == DTYPE_NODROP && qlen(q) >= maxqsize) {
5ba3f43e 1263 if (pkt_proto == IPPROTO_TCP &&
4bd07ac2 1264 qlen(q) < (maxqsize + (maxqsize >> 1)) &&
5ba3f43e 1265 ((*pkt_flags & PKTF_TCP_REXMT) ||
3e170ce0
A
1266 (sp->sfb_flags & SFBF_LAST_PKT_DROPPED))) {
1267 /*
1268 * At some level, dropping packets will make the
1269 * flows backoff and will keep memory requirements
1270 * under control. But we should not cause a tail
1271 * drop because it can take a long time for a
1272 * TCP flow to recover. We should try to drop
1273 * alternate packets instead.
1274 */
1275 sp->sfb_flags &= ~SFBF_LAST_PKT_DROPPED;
1276 } else {
1277 droptype = DTYPE_FORCED;
1278 sp->sfb_stats.drop_queue++;
1279 sp->sfb_flags |= SFBF_LAST_PKT_DROPPED;
1280 }
316670eb
A
1281 }
1282
1283 if (fc_adv == 1 && droptype != DTYPE_FORCED &&
5ba3f43e
A
1284 sfb_bin_addfcentry(sp, pkt, *pkt_sfb_hash, pkt_flowsrc,
1285 pkt_flowid)) {
316670eb
A
1286 /* deliver flow control advisory error */
1287 if (droptype == DTYPE_NODROP) {
1288 ret = CLASSQEQ_SUCCESS_FC;
1289 VERIFY(!(sp->sfb_flags & SFBF_SUSPENDED));
1290 } else if (sp->sfb_flags & SFBF_SUSPENDED) {
5ba3f43e
A
1291 /* drop due to suspension */
1292 ret = CLASSQEQ_DROP_SP;
316670eb 1293 } else {
5ba3f43e
A
1294 /* drop due to flow-control */
1295 ret = CLASSQEQ_DROP_FC;
316670eb
A
1296 }
1297 }
316670eb
A
1298 /* if successful enqueue this packet, else drop it */
1299 if (droptype == DTYPE_NODROP) {
5ba3f43e 1300 VERIFY(pkt->pktsched_ptype == qptype(q));
cb323159 1301 _addq(q, &pkt->pktsched_pkt);
316670eb
A
1302 } else {
1303 IFCQ_CONVERT_LOCK(&sp->sfb_ifp->if_snd);
0a7de745 1304 return (ret != CLASSQEQ_SUCCESS) ? ret : CLASSQEQ_DROP;
316670eb
A
1305 }
1306
0a7de745 1307 if (!(*pkt_sfb_flags & SFB_PKT_PBOX)) {
5ba3f43e
A
1308 sfb_eq_update_bins(sp, *pkt_sfb_hash,
1309 pktsched_get_pkt_len(pkt));
0a7de745 1310 } else {
316670eb 1311 sp->sfb_stats.pbox_packets++;
0a7de745 1312 }
316670eb
A
1313
1314 /* successfully queued */
0a7de745 1315 return ret;
316670eb
A
1316}
1317
5ba3f43e
A
1318static void *
1319sfb_getq_flow(struct sfb *sp, class_queue_t *q, u_int32_t flow, boolean_t purge,
1320 pktsched_pkt_t *pkt)
316670eb
A
1321{
1322 struct timespec now;
5ba3f43e 1323 uint64_t *pkt_timestamp;
cb323159 1324 volatile uint32_t *pkt_flags;
5ba3f43e
A
1325 uint32_t *pkt_sfb_flags;
1326 uint32_t *pkt_sfb_hash;
cb323159 1327 classq_pkt_t p = CLASSQ_PKT_INITIALIZER(p);
316670eb 1328
0a7de745
A
1329 if (!purge && (sp->sfb_flags & SFBF_SUSPENDED)) {
1330 return NULL;
1331 }
316670eb
A
1332
1333 nanouptime(&now);
1334
1335 /* flow of 0 means head of queue */
cb323159
A
1336 if (flow == 0) {
1337 _getq(q, &p);
1338 } else {
1339 _getq_flow(q, &p, flow);
1340 }
1341
1342 if (p.cp_ptype == QP_INVALID) {
0a7de745 1343 if (!purge) {
316670eb 1344 net_timerclear(&sp->sfb_getqtime);
0a7de745
A
1345 }
1346 return NULL;
316670eb
A
1347 }
1348
cb323159 1349 pktsched_pkt_encap(pkt, &p);
5ba3f43e
A
1350 pktsched_get_pkt_vars(pkt, &pkt_flags, &pkt_timestamp, NULL,
1351 NULL, NULL, NULL);
1352 pkt_sfb_hash = pktsched_get_pkt_sfb_vars(pkt, &pkt_sfb_flags);
316670eb 1353
5ba3f43e 1354 /* See comments in <rdar://problem/14040693> */
cb323159
A
1355 switch (p.cp_ptype) {
1356 case QP_MBUF:
5ba3f43e 1357 VERIFY(*pkt_flags & PKTF_PRIV_GUARDED);
cb323159
A
1358 break;
1359 default:
1360 VERIFY(0);
1361 /* NOTREACHED */
1362 __builtin_unreachable();
0a7de745 1363 }
316670eb
A
1364
1365 if (!purge) {
1366 /* calculate EWMA of dequeues */
1367 if (net_timerisset(&sp->sfb_getqtime)) {
1368 struct timespec delta;
1369 u_int64_t avg, new;
316670eb
A
1370 net_timersub(&now, &sp->sfb_getqtime, &delta);
1371 net_timernsec(&delta, &new);
1372 avg = sp->sfb_stats.dequeue_avg;
1373 if (avg > 0) {
1374 int decay = DEQUEUE_DECAY;
1375 /*
1376 * If the time since last dequeue is
1377 * significantly greater than the current
fe8ab488 1378 * average, weigh the average more against
316670eb
A
1379 * the old value.
1380 */
0a7de745 1381 if (DEQUEUE_SPIKE(new, avg)) {
316670eb 1382 decay += 5;
0a7de745 1383 }
316670eb
A
1384 avg = (((avg << decay) - avg) + new) >> decay;
1385 } else {
1386 avg = new;
1387 }
1388 sp->sfb_stats.dequeue_avg = avg;
1389 }
1390 *(&sp->sfb_getqtime) = *(&now);
1391 }
1392
fe8ab488
A
1393 if (!purge && SFB_QUEUE_DELAYBASED(sp)) {
1394 u_int64_t dequeue_ns, queue_delay = 0;
1395 net_timernsec(&now, &dequeue_ns);
0a7de745 1396 if (dequeue_ns > *pkt_timestamp) {
5ba3f43e 1397 queue_delay = dequeue_ns - *pkt_timestamp;
0a7de745 1398 }
fe8ab488
A
1399
1400 if (sp->sfb_min_qdelay == 0 ||
0a7de745 1401 (queue_delay > 0 && queue_delay < sp->sfb_min_qdelay)) {
fe8ab488 1402 sp->sfb_min_qdelay = queue_delay;
0a7de745 1403 }
fe8ab488
A
1404 if (net_timercmp(&now, &sp->sfb_update_time, >=)) {
1405 if (sp->sfb_min_qdelay > sp->sfb_target_qdelay) {
0a7de745 1406 if (!SFB_IS_DELAYHIGH(sp)) {
fe8ab488 1407 SFB_SET_DELAY_HIGH(sp, q);
0a7de745 1408 }
fe8ab488
A
1409 } else {
1410 sp->sfb_flags &= ~(SFBF_DELAYHIGH);
1411 sp->sfb_fc_threshold = 0;
fe8ab488
A
1412 }
1413 net_timeradd(&now, &sp->sfb_update_interval,
1414 &sp->sfb_update_time);
1415 sp->sfb_min_qdelay = 0;
1416 }
1417 }
5ba3f43e 1418 *pkt_timestamp = 0;
fe8ab488 1419
316670eb
A
1420 /*
1421 * Clearpkts are the ones which were in the queue when the hash
1422 * function was perturbed. Since the perturbation value (fudge),
1423 * and thus bin information for these packets is not known, we do
1424 * not change accounting information while dequeuing these packets.
1425 * It is important not to set the hash interval too small due to
1426 * this reason. A rule of thumb is to set it to K*D, where D is
1427 * the time taken to drain queue.
1428 */
5ba3f43e
A
1429 if (*pkt_sfb_flags & SFB_PKT_PBOX) {
1430 *pkt_sfb_flags &= ~SFB_PKT_PBOX;
0a7de745 1431 if (sp->sfb_clearpkts > 0) {
316670eb 1432 sp->sfb_clearpkts--;
0a7de745 1433 }
316670eb
A
1434 } else if (sp->sfb_clearpkts > 0) {
1435 sp->sfb_clearpkts--;
1436 } else {
5ba3f43e
A
1437 sfb_dq_update_bins(sp, *pkt_sfb_hash, pktsched_get_pkt_len(pkt),
1438 &now, qsize(q));
fe8ab488
A
1439 }
1440
cb323159
A
1441 switch (p.cp_ptype) {
1442 case QP_MBUF:
1443 /* See comments in <rdar://problem/14040693> */
5ba3f43e 1444 *pkt_flags &= ~PKTF_PRIV_GUARDED;
cb323159
A
1445 break;
1446 default:
1447 VERIFY(0);
1448 /* NOTREACHED */
1449 __builtin_unreachable();
0a7de745 1450 }
fe8ab488
A
1451
1452 /*
1453 * If the queue becomes empty before the update interval, reset
1454 * the flow control threshold
1455 */
1456 if (qsize(q) == 0) {
1457 sp->sfb_flags &= ~SFBF_DELAYHIGH;
1458 sp->sfb_min_qdelay = 0;
1459 sp->sfb_fc_threshold = 0;
1460 net_timerclear(&sp->sfb_update_time);
3e170ce0 1461 net_timerclear(&sp->sfb_getqtime);
316670eb 1462 }
cb323159 1463 return pkt->pktsched_pkt_mbuf;
316670eb
A
1464}
1465
5ba3f43e
A
1466void
1467sfb_getq(struct sfb *sp, class_queue_t *q, pktsched_pkt_t *pkt)
316670eb 1468{
5ba3f43e 1469 sfb_getq_flow(sp, q, 0, FALSE, pkt);
316670eb
A
1470}
1471
1472void
1473sfb_purgeq(struct sfb *sp, class_queue_t *q, u_int32_t flow, u_int32_t *packets,
1474 u_int32_t *bytes)
1475{
1476 u_int32_t cnt = 0, len = 0;
5ba3f43e 1477 pktsched_pkt_t pkt;
316670eb
A
1478
1479 IFCQ_CONVERT_LOCK(&sp->sfb_ifp->if_snd);
5ba3f43e 1480 while (sfb_getq_flow(sp, q, flow, TRUE, &pkt) != NULL) {
316670eb 1481 cnt++;
5ba3f43e
A
1482 len += pktsched_get_pkt_len(&pkt);
1483 pktsched_free_pkt(&pkt);
316670eb
A
1484 }
1485
0a7de745 1486 if (packets != NULL) {
316670eb 1487 *packets = cnt;
0a7de745
A
1488 }
1489 if (bytes != NULL) {
316670eb 1490 *bytes = len;
0a7de745 1491 }
316670eb
A
1492}
1493
1494void
1495sfb_updateq(struct sfb *sp, cqev_t ev)
1496{
1497 struct ifnet *ifp = sp->sfb_ifp;
1498
1499 VERIFY(ifp != NULL);
1500
1501 switch (ev) {
39236c6e 1502 case CLASSQ_EV_LINK_BANDWIDTH: {
316670eb
A
1503 u_int64_t eff_rate = ifnet_output_linkrate(ifp);
1504
1505 /* update parameters only if rate has changed */
0a7de745 1506 if (eff_rate == sp->sfb_eff_rate) {
316670eb 1507 break;
0a7de745 1508 }
316670eb
A
1509
1510 if (classq_verbose) {
1511 log(LOG_DEBUG, "%s: SFB qid=%d, adapting to new "
1512 "eff_rate=%llu bps\n", if_name(ifp), sp->sfb_qid,
1513 eff_rate);
1514 }
1515 sfb_calc_holdtime(sp, eff_rate);
1516 sfb_calc_pboxtime(sp, eff_rate);
39037602 1517 ifclassq_calc_target_qdelay(ifp, &sp->sfb_target_qdelay);
fe8ab488 1518 sfb_calc_update_interval(sp, eff_rate);
316670eb
A
1519 break;
1520 }
1521
1522 case CLASSQ_EV_LINK_UP:
1523 case CLASSQ_EV_LINK_DOWN:
1524 if (classq_verbose) {
1525 log(LOG_DEBUG, "%s: SFB qid=%d, resetting due to "
1526 "link %s\n", if_name(ifp), sp->sfb_qid,
1527 (ev == CLASSQ_EV_LINK_UP) ? "UP" : "DOWN");
1528 }
1529 sfb_resetq(sp, ev);
1530 break;
1531
39236c6e 1532 case CLASSQ_EV_LINK_LATENCY:
316670eb
A
1533 case CLASSQ_EV_LINK_MTU:
1534 default:
1535 break;
1536 }
1537}
1538
1539int
1540sfb_suspendq(struct sfb *sp, class_queue_t *q, boolean_t on)
1541{
1542#pragma unused(q)
1543 struct ifnet *ifp = sp->sfb_ifp;
1544
1545 VERIFY(ifp != NULL);
1546
1547 if ((on && (sp->sfb_flags & SFBF_SUSPENDED)) ||
0a7de745
A
1548 (!on && !(sp->sfb_flags & SFBF_SUSPENDED))) {
1549 return 0;
1550 }
316670eb
A
1551
1552 if (!(sp->sfb_flags & SFBF_FLOWCTL)) {
1553 log(LOG_ERR, "%s: SFB qid=%d, unable to %s queue since "
1554 "flow-control is not enabled", if_name(ifp), sp->sfb_qid,
1555 (on ? "suspend" : "resume"));
0a7de745 1556 return ENOTSUP;
316670eb
A
1557 }
1558
1559 if (classq_verbose) {
1560 log(LOG_DEBUG, "%s: SFB qid=%d, setting state to %s",
1561 if_name(ifp), sp->sfb_qid, (on ? "SUSPENDED" : "RUNNING"));
1562 }
1563
1564 if (on) {
1565 sp->sfb_flags |= SFBF_SUSPENDED;
1566 } else {
1567 sp->sfb_flags &= ~SFBF_SUSPENDED;
1568 sfb_swap_bins(sp, qlen(q));
1569 }
1570
0a7de745 1571 return 0;
316670eb 1572}