1// SPDX-License-Identifier: ISC
2/*
3 * Copyright (C) 2019 Felix Fietkau <nbd@nbd.name>
4 * Copyright (C) 2021-2022 Intel Corporation
5 */
6
7#include <net/mac80211.h>
8#include "ieee80211_i.h"
9#include "sta_info.h"
10
11#define AVG_PKT_SIZE 1024
12
13/* Number of bits for an average sized packet */
14#define MCS_NBITS (AVG_PKT_SIZE << 3)
15
16/* Number of kilo-symbols (symbols * 1024) for a packet with (bps) bits per
17 * symbol. We use k-symbols to avoid rounding in the _TIME macros below.
18 */
19#define MCS_N_KSYMS(bps) DIV_ROUND_UP(MCS_NBITS << 10, (bps))
20
21/* Transmission time (in 1024 * usec) for a packet containing (ksyms) * 1024
22 * symbols.
23 */
24#define MCS_SYMBOL_TIME(sgi, ksyms) \
25 (sgi ? \
26 ((ksyms) * 4 * 18) / 20 : /* 3.6 us per sym */ \
27 ((ksyms) * 4) /* 4.0 us per sym */ \
28 )
29
30/* Transmit duration for the raw data part of an average sized packet */
31#define MCS_DURATION(streams, sgi, bps) \
32 ((u32)MCS_SYMBOL_TIME(sgi, MCS_N_KSYMS((streams) * (bps))))
33
34#define MCS_DURATION_S(shift, streams, sgi, bps) \
35 ((u16)((MCS_DURATION(streams, sgi, bps) >> shift)))
36
37/* These should match the values in enum nl80211_he_gi */
38#define HE_GI_08 0
39#define HE_GI_16 1
40#define HE_GI_32 2
41
42/* Transmission time (1024 usec) for a packet containing (ksyms) * k-symbols */
43#define HE_SYMBOL_TIME(gi, ksyms) \
44 (gi == HE_GI_08 ? \
45 ((ksyms) * 16 * 17) / 20 : /* 13.6 us per sym */ \
46 (gi == HE_GI_16 ? \
47 ((ksyms) * 16 * 18) / 20 : /* 14.4 us per sym */ \
48 ((ksyms) * 16) /* 16.0 us per sym */ \
49 ))
50
51/* Transmit duration for the raw data part of an average sized packet */
52#define HE_DURATION(streams, gi, bps) \
53 ((u32)HE_SYMBOL_TIME(gi, MCS_N_KSYMS((streams) * (bps))))
54
55#define HE_DURATION_S(shift, streams, gi, bps) \
56 (HE_DURATION(streams, gi, bps) >> shift)
57
58/* gi in HE/EHT is identical. It matches enum nl80211_eht_gi as well */
59#define EHT_GI_08 HE_GI_08
60#define EHT_GI_16 HE_GI_16
61#define EHT_GI_32 HE_GI_32
62
63#define EHT_DURATION(streams, gi, bps) \
64 HE_DURATION(streams, gi, bps)
65#define EHT_DURATION_S(shift, streams, gi, bps) \
66 HE_DURATION_S(shift, streams, gi, bps)
67
68#define BW_20 0
69#define BW_40 1
70#define BW_80 2
71#define BW_160 3
72#define BW_320 4
73
74/*
75 * Define group sort order: HT40 -> SGI -> #streams
76 */
77#define IEEE80211_MAX_STREAMS 4
78#define IEEE80211_HT_STREAM_GROUPS 4 /* BW(=2) * SGI(=2) */
79#define IEEE80211_VHT_STREAM_GROUPS 8 /* BW(=4) * SGI(=2) */
80
81#define IEEE80211_HE_MAX_STREAMS 8
82#define IEEE80211_HE_STREAM_GROUPS 12 /* BW(=4) * GI(=3) */
83
84#define IEEE80211_EHT_MAX_STREAMS 8
85#define IEEE80211_EHT_STREAM_GROUPS 15 /* BW(=5) * GI(=3) */
86
87#define IEEE80211_HT_GROUPS_NB (IEEE80211_MAX_STREAMS * \
88 IEEE80211_HT_STREAM_GROUPS)
89#define IEEE80211_VHT_GROUPS_NB (IEEE80211_MAX_STREAMS * \
90 IEEE80211_VHT_STREAM_GROUPS)
91#define IEEE80211_HE_GROUPS_NB (IEEE80211_HE_MAX_STREAMS * \
92 IEEE80211_HE_STREAM_GROUPS)
93#define IEEE80211_EHT_GROUPS_NB (IEEE80211_EHT_MAX_STREAMS * \
94 IEEE80211_EHT_STREAM_GROUPS)
95
96#define IEEE80211_HT_GROUP_0 0
97#define IEEE80211_VHT_GROUP_0 (IEEE80211_HT_GROUP_0 + IEEE80211_HT_GROUPS_NB)
98#define IEEE80211_HE_GROUP_0 (IEEE80211_VHT_GROUP_0 + IEEE80211_VHT_GROUPS_NB)
99#define IEEE80211_EHT_GROUP_0 (IEEE80211_HE_GROUP_0 + IEEE80211_HE_GROUPS_NB)
100
101#define MCS_GROUP_RATES 14
102
103#define HT_GROUP_IDX(_streams, _sgi, _ht40) \
104 IEEE80211_HT_GROUP_0 + \
105 IEEE80211_MAX_STREAMS * 2 * _ht40 + \
106 IEEE80211_MAX_STREAMS * _sgi + \
107 _streams - 1
108
109#define _MAX(a, b) (((a)>(b))?(a):(b))
110
111#define GROUP_SHIFT(duration) \
112 _MAX(0, 16 - __builtin_clz(duration))
113
114/* MCS rate information for an MCS group */
115#define __MCS_GROUP(_streams, _sgi, _ht40, _s) \
116 [HT_GROUP_IDX(_streams, _sgi, _ht40)] = { \
117 .shift = _s, \
118 .duration = { \
119 MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 54 : 26), \
120 MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 108 : 52), \
121 MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 162 : 78), \
122 MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 216 : 104), \
123 MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 324 : 156), \
124 MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 432 : 208), \
125 MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 486 : 234), \
126 MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 540 : 260) \
127 } \
128}
129
130#define MCS_GROUP_SHIFT(_streams, _sgi, _ht40) \
131 GROUP_SHIFT(MCS_DURATION(_streams, _sgi, _ht40 ? 54 : 26))
132
133#define MCS_GROUP(_streams, _sgi, _ht40) \
134 __MCS_GROUP(_streams, _sgi, _ht40, \
135 MCS_GROUP_SHIFT(_streams, _sgi, _ht40))
136
137#define VHT_GROUP_IDX(_streams, _sgi, _bw) \
138 (IEEE80211_VHT_GROUP_0 + \
139 IEEE80211_MAX_STREAMS * 2 * (_bw) + \
140 IEEE80211_MAX_STREAMS * (_sgi) + \
141 (_streams) - 1)
142
143#define BW2VBPS(_bw, r4, r3, r2, r1) \
144 (_bw == BW_160 ? r4 : _bw == BW_80 ? r3 : _bw == BW_40 ? r2 : r1)
145
146#define __VHT_GROUP(_streams, _sgi, _bw, _s) \
147 [VHT_GROUP_IDX(_streams, _sgi, _bw)] = { \
148 .shift = _s, \
149 .duration = { \
150 MCS_DURATION_S(_s, _streams, _sgi, \
151 BW2VBPS(_bw, 234, 117, 54, 26)), \
152 MCS_DURATION_S(_s, _streams, _sgi, \
153 BW2VBPS(_bw, 468, 234, 108, 52)), \
154 MCS_DURATION_S(_s, _streams, _sgi, \
155 BW2VBPS(_bw, 702, 351, 162, 78)), \
156 MCS_DURATION_S(_s, _streams, _sgi, \
157 BW2VBPS(_bw, 936, 468, 216, 104)), \
158 MCS_DURATION_S(_s, _streams, _sgi, \
159 BW2VBPS(_bw, 1404, 702, 324, 156)), \
160 MCS_DURATION_S(_s, _streams, _sgi, \
161 BW2VBPS(_bw, 1872, 936, 432, 208)), \
162 MCS_DURATION_S(_s, _streams, _sgi, \
163 BW2VBPS(_bw, 2106, 1053, 486, 234)), \
164 MCS_DURATION_S(_s, _streams, _sgi, \
165 BW2VBPS(_bw, 2340, 1170, 540, 260)), \
166 MCS_DURATION_S(_s, _streams, _sgi, \
167 BW2VBPS(_bw, 2808, 1404, 648, 312)), \
168 MCS_DURATION_S(_s, _streams, _sgi, \
169 BW2VBPS(_bw, 3120, 1560, 720, 346)) \
170 } \
171}
172
173#define VHT_GROUP_SHIFT(_streams, _sgi, _bw) \
174 GROUP_SHIFT(MCS_DURATION(_streams, _sgi, \
175 BW2VBPS(_bw, 243, 117, 54, 26)))
176
177#define VHT_GROUP(_streams, _sgi, _bw) \
178 __VHT_GROUP(_streams, _sgi, _bw, \
179 VHT_GROUP_SHIFT(_streams, _sgi, _bw))
180
181
182#define HE_GROUP_IDX(_streams, _gi, _bw) \
183 (IEEE80211_HE_GROUP_0 + \
184 IEEE80211_HE_MAX_STREAMS * 3 * (_bw) + \
185 IEEE80211_HE_MAX_STREAMS * (_gi) + \
186 (_streams) - 1)
187
188#define __HE_GROUP(_streams, _gi, _bw, _s) \
189 [HE_GROUP_IDX(_streams, _gi, _bw)] = { \
190 .shift = _s, \
191 .duration = { \
192 HE_DURATION_S(_s, _streams, _gi, \
193 BW2VBPS(_bw, 979, 489, 230, 115)), \
194 HE_DURATION_S(_s, _streams, _gi, \
195 BW2VBPS(_bw, 1958, 979, 475, 230)), \
196 HE_DURATION_S(_s, _streams, _gi, \
197 BW2VBPS(_bw, 2937, 1468, 705, 345)), \
198 HE_DURATION_S(_s, _streams, _gi, \
199 BW2VBPS(_bw, 3916, 1958, 936, 475)), \
200 HE_DURATION_S(_s, _streams, _gi, \
201 BW2VBPS(_bw, 5875, 2937, 1411, 705)), \
202 HE_DURATION_S(_s, _streams, _gi, \
203 BW2VBPS(_bw, 7833, 3916, 1872, 936)), \
204 HE_DURATION_S(_s, _streams, _gi, \
205 BW2VBPS(_bw, 8827, 4406, 2102, 1051)), \
206 HE_DURATION_S(_s, _streams, _gi, \
207 BW2VBPS(_bw, 9806, 4896, 2347, 1166)), \
208 HE_DURATION_S(_s, _streams, _gi, \
209 BW2VBPS(_bw, 11764, 5875, 2808, 1411)), \
210 HE_DURATION_S(_s, _streams, _gi, \
211 BW2VBPS(_bw, 13060, 6523, 3124, 1555)), \
212 HE_DURATION_S(_s, _streams, _gi, \
213 BW2VBPS(_bw, 14702, 7344, 3513, 1756)), \
214 HE_DURATION_S(_s, _streams, _gi, \
215 BW2VBPS(_bw, 16329, 8164, 3902, 1944)) \
216 } \
217}
218
219#define HE_GROUP_SHIFT(_streams, _gi, _bw) \
220 GROUP_SHIFT(HE_DURATION(_streams, _gi, \
221 BW2VBPS(_bw, 979, 489, 230, 115)))
222
223#define HE_GROUP(_streams, _gi, _bw) \
224 __HE_GROUP(_streams, _gi, _bw, \
225 HE_GROUP_SHIFT(_streams, _gi, _bw))
226
227#define EHT_BW2VBPS(_bw, r5, r4, r3, r2, r1) \
228 ((_bw) == BW_320 ? r5 : BW2VBPS(_bw, r4, r3, r2, r1))
229
230#define EHT_GROUP_IDX(_streams, _gi, _bw) \
231 (IEEE80211_EHT_GROUP_0 + \
232 IEEE80211_EHT_MAX_STREAMS * 3 * (_bw) + \
233 IEEE80211_EHT_MAX_STREAMS * (_gi) + \
234 (_streams) - 1)
235
236#define __EHT_GROUP(_streams, _gi, _bw, _s) \
237 [EHT_GROUP_IDX(_streams, _gi, _bw)] = { \
238 .shift = _s, \
239 .duration = { \
240 EHT_DURATION_S(_s, _streams, _gi, \
241 EHT_BW2VBPS(_bw, 1960, 980, 490, 234, 117)), \
242 EHT_DURATION_S(_s, _streams, _gi, \
243 EHT_BW2VBPS(_bw, 3920, 1960, 980, 468, 234)), \
244 EHT_DURATION_S(_s, _streams, _gi, \
245 EHT_BW2VBPS(_bw, 5880, 2937, 1470, 702, 351)), \
246 EHT_DURATION_S(_s, _streams, _gi, \
247 EHT_BW2VBPS(_bw, 7840, 3920, 1960, 936, 468)), \
248 EHT_DURATION_S(_s, _streams, _gi, \
249 EHT_BW2VBPS(_bw, 11760, 5880, 2940, 1404, 702)), \
250 EHT_DURATION_S(_s, _streams, _gi, \
251 EHT_BW2VBPS(_bw, 15680, 7840, 3920, 1872, 936)), \
252 EHT_DURATION_S(_s, _streams, _gi, \
253 EHT_BW2VBPS(_bw, 17640, 8820, 4410, 2106, 1053)), \
254 EHT_DURATION_S(_s, _streams, _gi, \
255 EHT_BW2VBPS(_bw, 19600, 9800, 4900, 2340, 1170)), \
256 EHT_DURATION_S(_s, _streams, _gi, \
257 EHT_BW2VBPS(_bw, 23520, 11760, 5880, 2808, 1404)), \
258 EHT_DURATION_S(_s, _streams, _gi, \
259 EHT_BW2VBPS(_bw, 26133, 13066, 6533, 3120, 1560)), \
260 EHT_DURATION_S(_s, _streams, _gi, \
261 EHT_BW2VBPS(_bw, 29400, 14700, 7350, 3510, 1755)), \
262 EHT_DURATION_S(_s, _streams, _gi, \
263 EHT_BW2VBPS(_bw, 32666, 16333, 8166, 3900, 1950)), \
264 EHT_DURATION_S(_s, _streams, _gi, \
265 EHT_BW2VBPS(_bw, 35280, 17640, 8820, 4212, 2106)), \
266 EHT_DURATION_S(_s, _streams, _gi, \
267 EHT_BW2VBPS(_bw, 39200, 19600, 9800, 4680, 2340)) \
268 } \
269}
270
271#define EHT_GROUP_SHIFT(_streams, _gi, _bw) \
272 GROUP_SHIFT(EHT_DURATION(_streams, _gi, \
273 EHT_BW2VBPS(_bw, 1960, 980, 490, 234, 117)))
274
275#define EHT_GROUP(_streams, _gi, _bw) \
276 __EHT_GROUP(_streams, _gi, _bw, \
277 EHT_GROUP_SHIFT(_streams, _gi, _bw))
278
279#define EHT_GROUP_RANGE(_gi, _bw) \
280 EHT_GROUP(1, _gi, _bw), \
281 EHT_GROUP(2, _gi, _bw), \
282 EHT_GROUP(3, _gi, _bw), \
283 EHT_GROUP(4, _gi, _bw), \
284 EHT_GROUP(5, _gi, _bw), \
285 EHT_GROUP(6, _gi, _bw), \
286 EHT_GROUP(7, _gi, _bw), \
287 EHT_GROUP(8, _gi, _bw)
288
289struct mcs_group {
290 u8 shift;
291 u16 duration[MCS_GROUP_RATES];
292};
293
294static const struct mcs_group airtime_mcs_groups[] = {
295 MCS_GROUP(1, 0, BW_20),
296 MCS_GROUP(2, 0, BW_20),
297 MCS_GROUP(3, 0, BW_20),
298 MCS_GROUP(4, 0, BW_20),
299
300 MCS_GROUP(1, 1, BW_20),
301 MCS_GROUP(2, 1, BW_20),
302 MCS_GROUP(3, 1, BW_20),
303 MCS_GROUP(4, 1, BW_20),
304
305 MCS_GROUP(1, 0, BW_40),
306 MCS_GROUP(2, 0, BW_40),
307 MCS_GROUP(3, 0, BW_40),
308 MCS_GROUP(4, 0, BW_40),
309
310 MCS_GROUP(1, 1, BW_40),
311 MCS_GROUP(2, 1, BW_40),
312 MCS_GROUP(3, 1, BW_40),
313 MCS_GROUP(4, 1, BW_40),
314
315 VHT_GROUP(1, 0, BW_20),
316 VHT_GROUP(2, 0, BW_20),
317 VHT_GROUP(3, 0, BW_20),
318 VHT_GROUP(4, 0, BW_20),
319
320 VHT_GROUP(1, 1, BW_20),
321 VHT_GROUP(2, 1, BW_20),
322 VHT_GROUP(3, 1, BW_20),
323 VHT_GROUP(4, 1, BW_20),
324
325 VHT_GROUP(1, 0, BW_40),
326 VHT_GROUP(2, 0, BW_40),
327 VHT_GROUP(3, 0, BW_40),
328 VHT_GROUP(4, 0, BW_40),
329
330 VHT_GROUP(1, 1, BW_40),
331 VHT_GROUP(2, 1, BW_40),
332 VHT_GROUP(3, 1, BW_40),
333 VHT_GROUP(4, 1, BW_40),
334
335 VHT_GROUP(1, 0, BW_80),
336 VHT_GROUP(2, 0, BW_80),
337 VHT_GROUP(3, 0, BW_80),
338 VHT_GROUP(4, 0, BW_80),
339
340 VHT_GROUP(1, 1, BW_80),
341 VHT_GROUP(2, 1, BW_80),
342 VHT_GROUP(3, 1, BW_80),
343 VHT_GROUP(4, 1, BW_80),
344
345 VHT_GROUP(1, 0, BW_160),
346 VHT_GROUP(2, 0, BW_160),
347 VHT_GROUP(3, 0, BW_160),
348 VHT_GROUP(4, 0, BW_160),
349
350 VHT_GROUP(1, 1, BW_160),
351 VHT_GROUP(2, 1, BW_160),
352 VHT_GROUP(3, 1, BW_160),
353 VHT_GROUP(4, 1, BW_160),
354
355 HE_GROUP(1, HE_GI_08, BW_20),
356 HE_GROUP(2, HE_GI_08, BW_20),
357 HE_GROUP(3, HE_GI_08, BW_20),
358 HE_GROUP(4, HE_GI_08, BW_20),
359 HE_GROUP(5, HE_GI_08, BW_20),
360 HE_GROUP(6, HE_GI_08, BW_20),
361 HE_GROUP(7, HE_GI_08, BW_20),
362 HE_GROUP(8, HE_GI_08, BW_20),
363
364 HE_GROUP(1, HE_GI_16, BW_20),
365 HE_GROUP(2, HE_GI_16, BW_20),
366 HE_GROUP(3, HE_GI_16, BW_20),
367 HE_GROUP(4, HE_GI_16, BW_20),
368 HE_GROUP(5, HE_GI_16, BW_20),
369 HE_GROUP(6, HE_GI_16, BW_20),
370 HE_GROUP(7, HE_GI_16, BW_20),
371 HE_GROUP(8, HE_GI_16, BW_20),
372
373 HE_GROUP(1, HE_GI_32, BW_20),
374 HE_GROUP(2, HE_GI_32, BW_20),
375 HE_GROUP(3, HE_GI_32, BW_20),
376 HE_GROUP(4, HE_GI_32, BW_20),
377 HE_GROUP(5, HE_GI_32, BW_20),
378 HE_GROUP(6, HE_GI_32, BW_20),
379 HE_GROUP(7, HE_GI_32, BW_20),
380 HE_GROUP(8, HE_GI_32, BW_20),
381
382 HE_GROUP(1, HE_GI_08, BW_40),
383 HE_GROUP(2, HE_GI_08, BW_40),
384 HE_GROUP(3, HE_GI_08, BW_40),
385 HE_GROUP(4, HE_GI_08, BW_40),
386 HE_GROUP(5, HE_GI_08, BW_40),
387 HE_GROUP(6, HE_GI_08, BW_40),
388 HE_GROUP(7, HE_GI_08, BW_40),
389 HE_GROUP(8, HE_GI_08, BW_40),
390
391 HE_GROUP(1, HE_GI_16, BW_40),
392 HE_GROUP(2, HE_GI_16, BW_40),
393 HE_GROUP(3, HE_GI_16, BW_40),
394 HE_GROUP(4, HE_GI_16, BW_40),
395 HE_GROUP(5, HE_GI_16, BW_40),
396 HE_GROUP(6, HE_GI_16, BW_40),
397 HE_GROUP(7, HE_GI_16, BW_40),
398 HE_GROUP(8, HE_GI_16, BW_40),
399
400 HE_GROUP(1, HE_GI_32, BW_40),
401 HE_GROUP(2, HE_GI_32, BW_40),
402 HE_GROUP(3, HE_GI_32, BW_40),
403 HE_GROUP(4, HE_GI_32, BW_40),
404 HE_GROUP(5, HE_GI_32, BW_40),
405 HE_GROUP(6, HE_GI_32, BW_40),
406 HE_GROUP(7, HE_GI_32, BW_40),
407 HE_GROUP(8, HE_GI_32, BW_40),
408
409 HE_GROUP(1, HE_GI_08, BW_80),
410 HE_GROUP(2, HE_GI_08, BW_80),
411 HE_GROUP(3, HE_GI_08, BW_80),
412 HE_GROUP(4, HE_GI_08, BW_80),
413 HE_GROUP(5, HE_GI_08, BW_80),
414 HE_GROUP(6, HE_GI_08, BW_80),
415 HE_GROUP(7, HE_GI_08, BW_80),
416 HE_GROUP(8, HE_GI_08, BW_80),
417
418 HE_GROUP(1, HE_GI_16, BW_80),
419 HE_GROUP(2, HE_GI_16, BW_80),
420 HE_GROUP(3, HE_GI_16, BW_80),
421 HE_GROUP(4, HE_GI_16, BW_80),
422 HE_GROUP(5, HE_GI_16, BW_80),
423 HE_GROUP(6, HE_GI_16, BW_80),
424 HE_GROUP(7, HE_GI_16, BW_80),
425 HE_GROUP(8, HE_GI_16, BW_80),
426
427 HE_GROUP(1, HE_GI_32, BW_80),
428 HE_GROUP(2, HE_GI_32, BW_80),
429 HE_GROUP(3, HE_GI_32, BW_80),
430 HE_GROUP(4, HE_GI_32, BW_80),
431 HE_GROUP(5, HE_GI_32, BW_80),
432 HE_GROUP(6, HE_GI_32, BW_80),
433 HE_GROUP(7, HE_GI_32, BW_80),
434 HE_GROUP(8, HE_GI_32, BW_80),
435
436 HE_GROUP(1, HE_GI_08, BW_160),
437 HE_GROUP(2, HE_GI_08, BW_160),
438 HE_GROUP(3, HE_GI_08, BW_160),
439 HE_GROUP(4, HE_GI_08, BW_160),
440 HE_GROUP(5, HE_GI_08, BW_160),
441 HE_GROUP(6, HE_GI_08, BW_160),
442 HE_GROUP(7, HE_GI_08, BW_160),
443 HE_GROUP(8, HE_GI_08, BW_160),
444
445 HE_GROUP(1, HE_GI_16, BW_160),
446 HE_GROUP(2, HE_GI_16, BW_160),
447 HE_GROUP(3, HE_GI_16, BW_160),
448 HE_GROUP(4, HE_GI_16, BW_160),
449 HE_GROUP(5, HE_GI_16, BW_160),
450 HE_GROUP(6, HE_GI_16, BW_160),
451 HE_GROUP(7, HE_GI_16, BW_160),
452 HE_GROUP(8, HE_GI_16, BW_160),
453
454 HE_GROUP(1, HE_GI_32, BW_160),
455 HE_GROUP(2, HE_GI_32, BW_160),
456 HE_GROUP(3, HE_GI_32, BW_160),
457 HE_GROUP(4, HE_GI_32, BW_160),
458 HE_GROUP(5, HE_GI_32, BW_160),
459 HE_GROUP(6, HE_GI_32, BW_160),
460 HE_GROUP(7, HE_GI_32, BW_160),
461 HE_GROUP(8, HE_GI_32, BW_160),
462
463 EHT_GROUP_RANGE(EHT_GI_08, BW_20),
464 EHT_GROUP_RANGE(EHT_GI_16, BW_20),
465 EHT_GROUP_RANGE(EHT_GI_32, BW_20),
466
467 EHT_GROUP_RANGE(EHT_GI_08, BW_40),
468 EHT_GROUP_RANGE(EHT_GI_16, BW_40),
469 EHT_GROUP_RANGE(EHT_GI_32, BW_40),
470
471 EHT_GROUP_RANGE(EHT_GI_08, BW_80),
472 EHT_GROUP_RANGE(EHT_GI_16, BW_80),
473 EHT_GROUP_RANGE(EHT_GI_32, BW_80),
474
475 EHT_GROUP_RANGE(EHT_GI_08, BW_160),
476 EHT_GROUP_RANGE(EHT_GI_16, BW_160),
477 EHT_GROUP_RANGE(EHT_GI_32, BW_160),
478
479 EHT_GROUP_RANGE(EHT_GI_08, BW_320),
480 EHT_GROUP_RANGE(EHT_GI_16, BW_320),
481 EHT_GROUP_RANGE(EHT_GI_32, BW_320),
482};
483
484static u32
485ieee80211_calc_legacy_rate_duration(u16 bitrate, bool short_pre,
486 bool cck, int len)
487{
488 u32 duration;
489
490 if (cck) {
491 duration = 144 + 48; /* preamble + PLCP */
492 if (short_pre)
493 duration >>= 1;
494
495 duration += 10; /* SIFS */
496 } else {
497 duration = 20 + 16; /* premable + SIFS */
498 }
499
500 len <<= 3;
501 duration += (len * 10) / bitrate;
502
503 return duration;
504}
505
506static u32 ieee80211_get_rate_duration(struct ieee80211_hw *hw,
507 struct ieee80211_rx_status *status,
508 u32 *overhead)
509{
510 bool sgi = status->enc_flags & RX_ENC_FLAG_SHORT_GI;
511 int bw, streams;
512 int group, idx;
513 u32 duration;
514
515 switch (status->bw) {
516 case RATE_INFO_BW_20:
517 bw = BW_20;
518 break;
519 case RATE_INFO_BW_40:
520 bw = BW_40;
521 break;
522 case RATE_INFO_BW_80:
523 bw = BW_80;
524 break;
525 case RATE_INFO_BW_160:
526 bw = BW_160;
527 break;
528 case RATE_INFO_BW_320:
529 bw = BW_320;
530 break;
531 default:
532 WARN_ON_ONCE(1);
533 return 0;
534 }
535
536 switch (status->encoding) {
537 case RX_ENC_VHT:
538 streams = status->nss;
539 idx = status->rate_idx;
540 group = VHT_GROUP_IDX(streams, sgi, bw);
541 break;
542 case RX_ENC_HT:
543 streams = ((status->rate_idx >> 3) & 3) + 1;
544 idx = status->rate_idx & 7;
545 group = HT_GROUP_IDX(streams, sgi, bw);
546 break;
547 case RX_ENC_HE:
548 streams = status->nss;
549 idx = status->rate_idx;
550 group = HE_GROUP_IDX(streams, status->he_gi, bw);
551 break;
552 case RX_ENC_EHT:
553 streams = status->nss;
554 idx = status->rate_idx;
555 group = EHT_GROUP_IDX(streams, status->eht.gi, bw);
556 break;
557 default:
558 WARN_ON_ONCE(1);
559 return 0;
560 }
561
562 switch (status->encoding) {
563 case RX_ENC_EHT:
564 case RX_ENC_HE:
565 if (WARN_ON_ONCE(streams > 8))
566 return 0;
567 break;
568 default:
569 if (WARN_ON_ONCE(streams > 4))
570 return 0;
571 break;
572 }
573
574 if (idx >= MCS_GROUP_RATES)
575 return 0;
576
577 duration = airtime_mcs_groups[group].duration[idx];
578 duration <<= airtime_mcs_groups[group].shift;
579 *overhead = 36 + (streams << 2);
580
581 return duration;
582}
583
584
585u32 ieee80211_calc_rx_airtime(struct ieee80211_hw *hw,
586 struct ieee80211_rx_status *status,
587 int len)
588{
589 struct ieee80211_supported_band *sband;
590 u32 duration, overhead = 0;
591
592 if (status->encoding == RX_ENC_LEGACY) {
593 const struct ieee80211_rate *rate;
594 bool sp = status->enc_flags & RX_ENC_FLAG_SHORTPRE;
595 bool cck;
596
597 /* on 60GHz or sub-1GHz band, there are no legacy rates */
598 if (WARN_ON_ONCE(status->band == NL80211_BAND_60GHZ ||
599 status->band == NL80211_BAND_S1GHZ))
600 return 0;
601
602 sband = hw->wiphy->bands[status->band];
603 if (!sband || status->rate_idx >= sband->n_bitrates)
604 return 0;
605
606 rate = &sband->bitrates[status->rate_idx];
607 cck = rate->flags & IEEE80211_RATE_MANDATORY_B;
608
609 return ieee80211_calc_legacy_rate_duration(bitrate: rate->bitrate, short_pre: sp,
610 cck, len);
611 }
612
613 duration = ieee80211_get_rate_duration(hw, status, overhead: &overhead);
614 if (!duration)
615 return 0;
616
617 duration *= len;
618 duration /= AVG_PKT_SIZE;
619 duration /= 1024;
620
621 return duration + overhead;
622}
623EXPORT_SYMBOL_GPL(ieee80211_calc_rx_airtime);
624
625static bool ieee80211_fill_rate_info(struct ieee80211_hw *hw,
626 struct ieee80211_rx_status *stat, u8 band,
627 struct rate_info *ri)
628{
629 struct ieee80211_supported_band *sband = hw->wiphy->bands[band];
630 int i;
631
632 if (!ri || !sband)
633 return false;
634
635 stat->bw = ri->bw;
636 stat->nss = ri->nss;
637 stat->rate_idx = ri->mcs;
638
639 if (ri->flags & RATE_INFO_FLAGS_EHT_MCS)
640 stat->encoding = RX_ENC_EHT;
641 else if (ri->flags & RATE_INFO_FLAGS_HE_MCS)
642 stat->encoding = RX_ENC_HE;
643 else if (ri->flags & RATE_INFO_FLAGS_VHT_MCS)
644 stat->encoding = RX_ENC_VHT;
645 else if (ri->flags & RATE_INFO_FLAGS_MCS)
646 stat->encoding = RX_ENC_HT;
647 else
648 stat->encoding = RX_ENC_LEGACY;
649
650 if (ri->flags & RATE_INFO_FLAGS_SHORT_GI)
651 stat->enc_flags |= RX_ENC_FLAG_SHORT_GI;
652
653 switch (stat->encoding) {
654 case RX_ENC_EHT:
655 stat->eht.gi = ri->eht_gi;
656 break;
657 default:
658 stat->he_gi = ri->he_gi;
659 break;
660 }
661
662 if (stat->encoding != RX_ENC_LEGACY)
663 return true;
664
665 stat->rate_idx = 0;
666 for (i = 0; i < sband->n_bitrates; i++) {
667 if (ri->legacy != sband->bitrates[i].bitrate)
668 continue;
669
670 stat->rate_idx = i;
671 return true;
672 }
673
674 return false;
675}
676
677static int ieee80211_fill_rx_status(struct ieee80211_rx_status *stat,
678 struct ieee80211_hw *hw,
679 struct ieee80211_tx_rate *rate,
680 struct rate_info *ri, u8 band, int len)
681{
682 memset(s: stat, c: 0, n: sizeof(*stat));
683 stat->band = band;
684
685 if (ieee80211_fill_rate_info(hw, stat, band, ri))
686 return 0;
687
688 if (!ieee80211_rate_valid(rate))
689 return -1;
690
691 if (rate->flags & IEEE80211_TX_RC_160_MHZ_WIDTH)
692 stat->bw = RATE_INFO_BW_160;
693 else if (rate->flags & IEEE80211_TX_RC_80_MHZ_WIDTH)
694 stat->bw = RATE_INFO_BW_80;
695 else if (rate->flags & IEEE80211_TX_RC_40_MHZ_WIDTH)
696 stat->bw = RATE_INFO_BW_40;
697 else
698 stat->bw = RATE_INFO_BW_20;
699
700 stat->enc_flags = 0;
701 if (rate->flags & IEEE80211_TX_RC_USE_SHORT_PREAMBLE)
702 stat->enc_flags |= RX_ENC_FLAG_SHORTPRE;
703 if (rate->flags & IEEE80211_TX_RC_SHORT_GI)
704 stat->enc_flags |= RX_ENC_FLAG_SHORT_GI;
705
706 stat->rate_idx = rate->idx;
707 if (rate->flags & IEEE80211_TX_RC_VHT_MCS) {
708 stat->encoding = RX_ENC_VHT;
709 stat->rate_idx = ieee80211_rate_get_vht_mcs(rate);
710 stat->nss = ieee80211_rate_get_vht_nss(rate);
711 } else if (rate->flags & IEEE80211_TX_RC_MCS) {
712 stat->encoding = RX_ENC_HT;
713 } else {
714 stat->encoding = RX_ENC_LEGACY;
715 }
716
717 return 0;
718}
719
720static u32 ieee80211_calc_tx_airtime_rate(struct ieee80211_hw *hw,
721 struct ieee80211_tx_rate *rate,
722 struct rate_info *ri,
723 u8 band, int len)
724{
725 struct ieee80211_rx_status stat;
726
727 if (ieee80211_fill_rx_status(stat: &stat, hw, rate, ri, band, len))
728 return 0;
729
730 return ieee80211_calc_rx_airtime(hw, &stat, len);
731}
732
733u32 ieee80211_calc_tx_airtime(struct ieee80211_hw *hw,
734 struct ieee80211_tx_info *info,
735 int len)
736{
737 u32 duration = 0;
738 int i;
739
740 for (i = 0; i < ARRAY_SIZE(info->status.rates); i++) {
741 struct ieee80211_tx_rate *rate = &info->status.rates[i];
742 u32 cur_duration;
743
744 cur_duration = ieee80211_calc_tx_airtime_rate(hw, rate, NULL,
745 band: info->band, len);
746 if (!cur_duration)
747 break;
748
749 duration += cur_duration * rate->count;
750 }
751
752 return duration;
753}
754EXPORT_SYMBOL_GPL(ieee80211_calc_tx_airtime);
755
756u32 ieee80211_calc_expected_tx_airtime(struct ieee80211_hw *hw,
757 struct ieee80211_vif *vif,
758 struct ieee80211_sta *pubsta,
759 int len, bool ampdu)
760{
761 struct ieee80211_supported_band *sband;
762 struct ieee80211_chanctx_conf *conf;
763 int rateidx;
764 bool cck, short_pream;
765 u32 basic_rates;
766 u8 band = 0;
767 u16 rate;
768
769 len += 38; /* Ethernet header length */
770
771 conf = rcu_dereference(vif->bss_conf.chanctx_conf);
772 if (conf)
773 band = conf->def.chan->band;
774
775 if (pubsta) {
776 struct sta_info *sta = container_of(pubsta, struct sta_info,
777 sta);
778 struct ieee80211_rx_status stat;
779 struct ieee80211_tx_rate *tx_rate = &sta->deflink.tx_stats.last_rate;
780 struct rate_info *ri = &sta->deflink.tx_stats.last_rate_info;
781 u32 duration, overhead;
782 u8 agg_shift;
783
784 if (ieee80211_fill_rx_status(stat: &stat, hw, rate: tx_rate, ri, band, len))
785 return 0;
786
787 if (stat.encoding == RX_ENC_LEGACY || !ampdu)
788 return ieee80211_calc_rx_airtime(hw, &stat, len);
789
790 duration = ieee80211_get_rate_duration(hw, status: &stat, overhead: &overhead);
791 /*
792 * Assume that HT/VHT transmission on any AC except VO will
793 * use aggregation. Since we don't have reliable reporting
794 * of aggregation length, assume an average size based on the
795 * tx rate.
796 * This will not be very accurate, but much better than simply
797 * assuming un-aggregated tx in all cases.
798 */
799 if (duration > 400 * 1024) /* <= VHT20 MCS2 1S */
800 agg_shift = 1;
801 else if (duration > 250 * 1024) /* <= VHT20 MCS3 1S or MCS1 2S */
802 agg_shift = 2;
803 else if (duration > 150 * 1024) /* <= VHT20 MCS5 1S or MCS2 2S */
804 agg_shift = 3;
805 else if (duration > 70 * 1024) /* <= VHT20 MCS5 2S */
806 agg_shift = 4;
807 else if (stat.encoding != RX_ENC_HE ||
808 duration > 20 * 1024) /* <= HE40 MCS6 2S */
809 agg_shift = 5;
810 else
811 agg_shift = 6;
812
813 duration *= len;
814 duration /= AVG_PKT_SIZE;
815 duration /= 1024;
816 duration += (overhead >> agg_shift);
817
818 return max_t(u32, duration, 4);
819 }
820
821 if (!conf)
822 return 0;
823
824 /* No station to get latest rate from, so calculate the worst-case
825 * duration using the lowest configured basic rate.
826 */
827 sband = hw->wiphy->bands[band];
828
829 basic_rates = vif->bss_conf.basic_rates;
830 short_pream = vif->bss_conf.use_short_preamble;
831
832 rateidx = basic_rates ? ffs(basic_rates) - 1 : 0;
833 rate = sband->bitrates[rateidx].bitrate;
834 cck = sband->bitrates[rateidx].flags & IEEE80211_RATE_MANDATORY_B;
835
836 return ieee80211_calc_legacy_rate_duration(bitrate: rate, short_pre: short_pream, cck, len);
837}
838