]>
Commit | Line | Data |
---|---|---|
2874c5fd | 1 | // SPDX-License-Identifier: GPL-2.0-or-later |
1da177e4 LT |
2 | /* |
3 | * net/sched/gen_estimator.c Simple rate estimator. | |
4 | * | |
1da177e4 | 5 | * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru> |
1c0d32fd | 6 | * Eric Dumazet <edumazet@google.com> |
1da177e4 LT |
7 | * |
8 | * Changes: | |
9 | * Jamal Hadi Salim - moved it to net/core and reshulfed | |
10 | * names to make it usable in general net subsystem. | |
11 | */ | |
12 | ||
7c0f6ba6 | 13 | #include <linux/uaccess.h> |
1977f032 | 14 | #include <linux/bitops.h> |
1da177e4 LT |
15 | #include <linux/module.h> |
16 | #include <linux/types.h> | |
17 | #include <linux/kernel.h> | |
18 | #include <linux/jiffies.h> | |
19 | #include <linux/string.h> | |
20 | #include <linux/mm.h> | |
21 | #include <linux/socket.h> | |
22 | #include <linux/sockios.h> | |
23 | #include <linux/in.h> | |
24 | #include <linux/errno.h> | |
25 | #include <linux/interrupt.h> | |
26 | #include <linux/netdevice.h> | |
27 | #include <linux/skbuff.h> | |
28 | #include <linux/rtnetlink.h> | |
29 | #include <linux/init.h> | |
5a0e3ad6 | 30 | #include <linux/slab.h> |
1c0d32fd | 31 | #include <linux/seqlock.h> |
1da177e4 LT |
32 | #include <net/sock.h> |
33 | #include <net/gen_stats.h> | |
34 | ||
1c0d32fd ED |
35 | /* This code is NOT intended to be used for statistics collection, |
36 | * its purpose is to provide a base for statistical multiplexing | |
37 | * for controlled load service. | |
38 | * If you need only statistics, run a user level daemon which | |
39 | * periodically reads byte counters. | |
1da177e4 LT |
40 | */ |
41 | ||
1c0d32fd | 42 | struct net_rate_estimator { |
c1a8f1f1 | 43 | struct gnet_stats_basic_packed *bstats; |
1da177e4 | 44 | spinlock_t *stats_lock; |
edb09eb1 | 45 | seqcount_t *running; |
1c0d32fd ED |
46 | struct gnet_stats_basic_cpu __percpu *cpu_bstats; |
47 | u8 ewma_log; | |
48 | u8 intvl_log; /* period : (250ms << intvl_log) */ | |
49 | ||
50 | seqcount_t seq; | |
1c8dd9cb | 51 | u64 last_packets; |
1da177e4 | 52 | u64 last_bytes; |
1c0d32fd ED |
53 | |
54 | u64 avpps; | |
511e11e3 | 55 | u64 avbps; |
1da177e4 | 56 | |
1c0d32fd ED |
57 | unsigned long next_jiffies; |
58 | struct timer_list timer; | |
59 | struct rcu_head rcu; | |
1da177e4 LT |
60 | }; |
61 | ||
1c0d32fd ED |
62 | static void est_fetch_counters(struct net_rate_estimator *e, |
63 | struct gnet_stats_basic_packed *b) | |
1da177e4 | 64 | { |
a5f7add3 | 65 | memset(b, 0, sizeof(*b)); |
1c0d32fd ED |
66 | if (e->stats_lock) |
67 | spin_lock(e->stats_lock); | |
1da177e4 | 68 | |
1c0d32fd | 69 | __gnet_stats_copy_basic(e->running, b, e->cpu_bstats, e->bstats); |
1da177e4 | 70 | |
1c0d32fd ED |
71 | if (e->stats_lock) |
72 | spin_unlock(e->stats_lock); | |
12efa1fa | 73 | |
1da177e4 LT |
74 | } |
75 | ||
e99e88a9 | 76 | static void est_timer(struct timer_list *t) |
4db0acf3 | 77 | { |
e99e88a9 | 78 | struct net_rate_estimator *est = from_timer(est, t, timer); |
1c0d32fd ED |
79 | struct gnet_stats_basic_packed b; |
80 | u64 rate, brate; | |
4db0acf3 | 81 | |
1c0d32fd | 82 | est_fetch_counters(est, &b); |
ca558e18 | 83 | brate = (b.bytes - est->last_bytes) << (10 - est->ewma_log - est->intvl_log); |
1c0d32fd | 84 | brate -= (est->avbps >> est->ewma_log); |
4db0acf3 | 85 | |
1c8dd9cb | 86 | rate = (b.packets - est->last_packets) << (10 - est->ewma_log - est->intvl_log); |
1c0d32fd | 87 | rate -= (est->avpps >> est->ewma_log); |
4db0acf3 | 88 | |
1c0d32fd ED |
89 | write_seqcount_begin(&est->seq); |
90 | est->avbps += brate; | |
91 | est->avpps += rate; | |
92 | write_seqcount_end(&est->seq); | |
4db0acf3 | 93 | |
1c0d32fd ED |
94 | est->last_bytes = b.bytes; |
95 | est->last_packets = b.packets; | |
4db0acf3 | 96 | |
1c0d32fd | 97 | est->next_jiffies += ((HZ/4) << est->intvl_log); |
4db0acf3 | 98 | |
1c0d32fd ED |
99 | if (unlikely(time_after_eq(jiffies, est->next_jiffies))) { |
100 | /* Ouch... timer was delayed. */ | |
101 | est->next_jiffies = jiffies + 1; | |
4db0acf3 | 102 | } |
1c0d32fd | 103 | mod_timer(&est->timer, est->next_jiffies); |
4db0acf3 JP |
104 | } |
105 | ||
1da177e4 LT |
106 | /** |
107 | * gen_new_estimator - create a new rate estimator | |
108 | * @bstats: basic statistics | |
e9fc2f05 | 109 | * @cpu_bstats: bstats per cpu |
1da177e4 | 110 | * @rate_est: rate estimator statistics |
51a9f5ae | 111 | * @lock: lock for statistics and control path |
edb09eb1 | 112 | * @running: qdisc running seqcount |
1da177e4 LT |
113 | * @opt: rate estimator configuration TLV |
114 | * | |
115 | * Creates a new rate estimator with &bstats as source and &rate_est | |
116 | * as destination. A new timer with the interval specified in the | |
117 | * configuration TLV is created. Upon each interval, the latest statistics | |
118 | * will be read from &bstats and the estimated rate will be stored in | |
e793c0f7 | 119 | * &rate_est with the statistics lock grabbed during this period. |
4ec93edb | 120 | * |
1da177e4 | 121 | * Returns 0 on success or a negative error code. |
0929c2dd | 122 | * |
1da177e4 | 123 | */ |
c1a8f1f1 | 124 | int gen_new_estimator(struct gnet_stats_basic_packed *bstats, |
22e0f8b9 | 125 | struct gnet_stats_basic_cpu __percpu *cpu_bstats, |
1c0d32fd | 126 | struct net_rate_estimator __rcu **rate_est, |
51a9f5ae | 127 | spinlock_t *lock, |
edb09eb1 | 128 | seqcount_t *running, |
1e90474c | 129 | struct nlattr *opt) |
1da177e4 | 130 | { |
1e90474c | 131 | struct gnet_estimator *parm = nla_data(opt); |
1c0d32fd ED |
132 | struct net_rate_estimator *old, *est; |
133 | struct gnet_stats_basic_packed b; | |
134 | int intvl_log; | |
1da177e4 | 135 | |
1e90474c | 136 | if (nla_len(opt) < sizeof(*parm)) |
1da177e4 LT |
137 | return -EINVAL; |
138 | ||
1c0d32fd ED |
139 | /* allowed timer periods are : |
140 | * -2 : 250ms, -1 : 500ms, 0 : 1 sec | |
141 | * 1 : 2 sec, 2 : 4 sec, 3 : 8 sec | |
142 | */ | |
1da177e4 LT |
143 | if (parm->interval < -2 || parm->interval > 3) |
144 | return -EINVAL; | |
145 | ||
77d04bd9 | 146 | est = kzalloc(sizeof(*est), GFP_KERNEL); |
1c0d32fd | 147 | if (!est) |
1da177e4 LT |
148 | return -ENOBUFS; |
149 | ||
1c0d32fd ED |
150 | seqcount_init(&est->seq); |
151 | intvl_log = parm->interval + 2; | |
1da177e4 | 152 | est->bstats = bstats; |
51a9f5ae | 153 | est->stats_lock = lock; |
edb09eb1 | 154 | est->running = running; |
1da177e4 | 155 | est->ewma_log = parm->ewma_log; |
1c0d32fd | 156 | est->intvl_log = intvl_log; |
22e0f8b9 | 157 | est->cpu_bstats = cpu_bstats; |
1da177e4 | 158 | |
51a9f5ae | 159 | if (lock) |
40ca54e3 | 160 | local_bh_disable(); |
1c0d32fd | 161 | est_fetch_counters(est, &b); |
51a9f5ae | 162 | if (lock) |
40ca54e3 | 163 | local_bh_enable(); |
1c0d32fd ED |
164 | est->last_bytes = b.bytes; |
165 | est->last_packets = b.packets; | |
51a9f5ae VB |
166 | |
167 | if (lock) | |
168 | spin_lock_bh(lock); | |
1c0d32fd ED |
169 | old = rcu_dereference_protected(*rate_est, 1); |
170 | if (old) { | |
171 | del_timer_sync(&old->timer); | |
172 | est->avbps = old->avbps; | |
173 | est->avpps = old->avpps; | |
1da177e4 | 174 | } |
0929c2dd | 175 | |
1c0d32fd | 176 | est->next_jiffies = jiffies + ((HZ/4) << intvl_log); |
e99e88a9 | 177 | timer_setup(&est->timer, est_timer, 0); |
1c0d32fd | 178 | mod_timer(&est->timer, est->next_jiffies); |
4db0acf3 | 179 | |
1c0d32fd | 180 | rcu_assign_pointer(*rate_est, est); |
51a9f5ae VB |
181 | if (lock) |
182 | spin_unlock_bh(lock); | |
1c0d32fd ED |
183 | if (old) |
184 | kfree_rcu(old, rcu); | |
1da177e4 LT |
185 | return 0; |
186 | } | |
c1b56878 | 187 | EXPORT_SYMBOL(gen_new_estimator); |
1da177e4 LT |
188 | |
189 | /** | |
190 | * gen_kill_estimator - remove a rate estimator | |
1c0d32fd | 191 | * @rate_est: rate estimator |
1da177e4 | 192 | * |
1c0d32fd | 193 | * Removes the rate estimator. |
0929c2dd | 194 | * |
1da177e4 | 195 | */ |
1c0d32fd | 196 | void gen_kill_estimator(struct net_rate_estimator __rcu **rate_est) |
1da177e4 | 197 | { |
1c0d32fd | 198 | struct net_rate_estimator *est; |
1da177e4 | 199 | |
1c0d32fd ED |
200 | est = xchg((__force struct net_rate_estimator **)rate_est, NULL); |
201 | if (est) { | |
202 | del_timer_sync(&est->timer); | |
203 | kfree_rcu(est, rcu); | |
1da177e4 LT |
204 | } |
205 | } | |
c1b56878 | 206 | EXPORT_SYMBOL(gen_kill_estimator); |
1da177e4 LT |
207 | |
208 | /** | |
96750162 | 209 | * gen_replace_estimator - replace rate estimator configuration |
1da177e4 | 210 | * @bstats: basic statistics |
e9fc2f05 | 211 | * @cpu_bstats: bstats per cpu |
1da177e4 | 212 | * @rate_est: rate estimator statistics |
51a9f5ae | 213 | * @lock: lock for statistics and control path |
edb09eb1 | 214 | * @running: qdisc running seqcount (might be NULL) |
1da177e4 LT |
215 | * @opt: rate estimator configuration TLV |
216 | * | |
217 | * Replaces the configuration of a rate estimator by calling | |
218 | * gen_kill_estimator() and gen_new_estimator(). | |
4ec93edb | 219 | * |
1da177e4 LT |
220 | * Returns 0 on success or a negative error code. |
221 | */ | |
c1a8f1f1 | 222 | int gen_replace_estimator(struct gnet_stats_basic_packed *bstats, |
22e0f8b9 | 223 | struct gnet_stats_basic_cpu __percpu *cpu_bstats, |
1c0d32fd | 224 | struct net_rate_estimator __rcu **rate_est, |
51a9f5ae | 225 | spinlock_t *lock, |
edb09eb1 | 226 | seqcount_t *running, struct nlattr *opt) |
1da177e4 | 227 | { |
1c0d32fd | 228 | return gen_new_estimator(bstats, cpu_bstats, rate_est, |
51a9f5ae | 229 | lock, running, opt); |
1da177e4 | 230 | } |
c1b56878 SH |
231 | EXPORT_SYMBOL(gen_replace_estimator); |
232 | ||
233 | /** | |
234 | * gen_estimator_active - test if estimator is currently in use | |
1c0d32fd | 235 | * @rate_est: rate estimator |
c1b56878 | 236 | * |
244e6c2d | 237 | * Returns true if estimator is active, and false if not. |
c1b56878 | 238 | */ |
1c0d32fd | 239 | bool gen_estimator_active(struct net_rate_estimator __rcu **rate_est) |
c1b56878 | 240 | { |
1c0d32fd ED |
241 | return !!rcu_access_pointer(*rate_est); |
242 | } | |
243 | EXPORT_SYMBOL(gen_estimator_active); | |
ae638c47 | 244 | |
1c0d32fd ED |
245 | bool gen_estimator_read(struct net_rate_estimator __rcu **rate_est, |
246 | struct gnet_stats_rate_est64 *sample) | |
247 | { | |
248 | struct net_rate_estimator *est; | |
249 | unsigned seq; | |
250 | ||
251 | rcu_read_lock(); | |
252 | est = rcu_dereference(*rate_est); | |
253 | if (!est) { | |
254 | rcu_read_unlock(); | |
255 | return false; | |
256 | } | |
1da177e4 | 257 | |
1c0d32fd ED |
258 | do { |
259 | seq = read_seqcount_begin(&est->seq); | |
260 | sample->bps = est->avbps >> 8; | |
261 | sample->pps = est->avpps >> 8; | |
262 | } while (read_seqcount_retry(&est->seq, seq)); | |
ae638c47 | 263 | |
1c0d32fd ED |
264 | rcu_read_unlock(); |
265 | return true; | |
c1b56878 | 266 | } |
1c0d32fd | 267 | EXPORT_SYMBOL(gen_estimator_read); |