1db3e1c40SToke Høiland-Jørgensen // SPDX-License-Identifier: ISC
2db3e1c40SToke Høiland-Jørgensen /*
3db3e1c40SToke Høiland-Jørgensen * Copyright (C) 2019 Felix Fietkau <nbd@nbd.name>
4d0a9123eSJohannes Berg * Copyright (C) 2021-2022 Intel Corporation
5db3e1c40SToke Høiland-Jørgensen */
6db3e1c40SToke Høiland-Jørgensen
7db3e1c40SToke Høiland-Jørgensen #include <net/mac80211.h>
8db3e1c40SToke Høiland-Jørgensen #include "ieee80211_i.h"
9db3e1c40SToke Høiland-Jørgensen #include "sta_info.h"
10db3e1c40SToke Høiland-Jørgensen
11db3e1c40SToke Høiland-Jørgensen #define AVG_PKT_SIZE 1024
12db3e1c40SToke Høiland-Jørgensen
13db3e1c40SToke Høiland-Jørgensen /* Number of bits for an average sized packet */
14db3e1c40SToke Høiland-Jørgensen #define MCS_NBITS (AVG_PKT_SIZE << 3)
15db3e1c40SToke Høiland-Jørgensen
16db3e1c40SToke Høiland-Jørgensen /* Number of kilo-symbols (symbols * 1024) for a packet with (bps) bits per
17db3e1c40SToke Høiland-Jørgensen * symbol. We use k-symbols to avoid rounding in the _TIME macros below.
18db3e1c40SToke Høiland-Jørgensen */
19db3e1c40SToke Høiland-Jørgensen #define MCS_N_KSYMS(bps) DIV_ROUND_UP(MCS_NBITS << 10, (bps))
20db3e1c40SToke Høiland-Jørgensen
21db3e1c40SToke Høiland-Jørgensen /* Transmission time (in 1024 * usec) for a packet containing (ksyms) * 1024
22db3e1c40SToke Høiland-Jørgensen * symbols.
23db3e1c40SToke Høiland-Jørgensen */
24db3e1c40SToke Høiland-Jørgensen #define MCS_SYMBOL_TIME(sgi, ksyms) \
25db3e1c40SToke Høiland-Jørgensen (sgi ? \
26db3e1c40SToke Høiland-Jørgensen ((ksyms) * 4 * 18) / 20 : /* 3.6 us per sym */ \
27db3e1c40SToke Høiland-Jørgensen ((ksyms) * 4) /* 4.0 us per sym */ \
28db3e1c40SToke Høiland-Jørgensen )
29db3e1c40SToke Høiland-Jørgensen
30db3e1c40SToke Høiland-Jørgensen /* Transmit duration for the raw data part of an average sized packet */
31db3e1c40SToke Høiland-Jørgensen #define MCS_DURATION(streams, sgi, bps) \
32db3e1c40SToke Høiland-Jørgensen ((u32)MCS_SYMBOL_TIME(sgi, MCS_N_KSYMS((streams) * (bps))))
33db3e1c40SToke Høiland-Jørgensen
34db3e1c40SToke Høiland-Jørgensen #define MCS_DURATION_S(shift, streams, sgi, bps) \
35db3e1c40SToke Høiland-Jørgensen ((u16)((MCS_DURATION(streams, sgi, bps) >> shift)))
36db3e1c40SToke Høiland-Jørgensen
37db3e1c40SToke Høiland-Jørgensen /* These should match the values in enum nl80211_he_gi */
38db3e1c40SToke Høiland-Jørgensen #define HE_GI_08 0
39db3e1c40SToke Høiland-Jørgensen #define HE_GI_16 1
40db3e1c40SToke Høiland-Jørgensen #define HE_GI_32 2
41db3e1c40SToke Høiland-Jørgensen
42db3e1c40SToke Høiland-Jørgensen /* Transmission time (1024 usec) for a packet containing (ksyms) * k-symbols */
43db3e1c40SToke Høiland-Jørgensen #define HE_SYMBOL_TIME(gi, ksyms) \
44db3e1c40SToke Høiland-Jørgensen (gi == HE_GI_08 ? \
45db3e1c40SToke Høiland-Jørgensen ((ksyms) * 16 * 17) / 20 : /* 13.6 us per sym */ \
46db3e1c40SToke Høiland-Jørgensen (gi == HE_GI_16 ? \
47db3e1c40SToke Høiland-Jørgensen ((ksyms) * 16 * 18) / 20 : /* 14.4 us per sym */ \
48db3e1c40SToke Høiland-Jørgensen ((ksyms) * 16) /* 16.0 us per sym */ \
49db3e1c40SToke Høiland-Jørgensen ))
50db3e1c40SToke Høiland-Jørgensen
51db3e1c40SToke Høiland-Jørgensen /* Transmit duration for the raw data part of an average sized packet */
52db3e1c40SToke Høiland-Jørgensen #define HE_DURATION(streams, gi, bps) \
53db3e1c40SToke Høiland-Jørgensen ((u32)HE_SYMBOL_TIME(gi, MCS_N_KSYMS((streams) * (bps))))
54db3e1c40SToke Høiland-Jørgensen
55db3e1c40SToke Høiland-Jørgensen #define HE_DURATION_S(shift, streams, gi, bps) \
56db3e1c40SToke Høiland-Jørgensen (HE_DURATION(streams, gi, bps) >> shift)
57db3e1c40SToke Høiland-Jørgensen
58db3e1c40SToke Høiland-Jørgensen #define BW_20 0
59db3e1c40SToke Høiland-Jørgensen #define BW_40 1
60db3e1c40SToke Høiland-Jørgensen #define BW_80 2
61db3e1c40SToke Høiland-Jørgensen #define BW_160 3
62db3e1c40SToke Høiland-Jørgensen
63db3e1c40SToke Høiland-Jørgensen /*
64db3e1c40SToke Høiland-Jørgensen * Define group sort order: HT40 -> SGI -> #streams
65db3e1c40SToke Høiland-Jørgensen */
66db3e1c40SToke Høiland-Jørgensen #define IEEE80211_MAX_STREAMS 4
67db3e1c40SToke Høiland-Jørgensen #define IEEE80211_HT_STREAM_GROUPS 4 /* BW(=2) * SGI(=2) */
68db3e1c40SToke Høiland-Jørgensen #define IEEE80211_VHT_STREAM_GROUPS 8 /* BW(=4) * SGI(=2) */
69db3e1c40SToke Høiland-Jørgensen
70db3e1c40SToke Høiland-Jørgensen #define IEEE80211_HE_MAX_STREAMS 8
71db3e1c40SToke Høiland-Jørgensen
72db3e1c40SToke Høiland-Jørgensen #define IEEE80211_HT_GROUPS_NB (IEEE80211_MAX_STREAMS * \
73db3e1c40SToke Høiland-Jørgensen IEEE80211_HT_STREAM_GROUPS)
74db3e1c40SToke Høiland-Jørgensen #define IEEE80211_VHT_GROUPS_NB (IEEE80211_MAX_STREAMS * \
75db3e1c40SToke Høiland-Jørgensen IEEE80211_VHT_STREAM_GROUPS)
76db3e1c40SToke Høiland-Jørgensen
77db3e1c40SToke Høiland-Jørgensen #define IEEE80211_HT_GROUP_0 0
78db3e1c40SToke Høiland-Jørgensen #define IEEE80211_VHT_GROUP_0 (IEEE80211_HT_GROUP_0 + IEEE80211_HT_GROUPS_NB)
79db3e1c40SToke Høiland-Jørgensen #define IEEE80211_HE_GROUP_0 (IEEE80211_VHT_GROUP_0 + IEEE80211_VHT_GROUPS_NB)
80db3e1c40SToke Høiland-Jørgensen
81db3e1c40SToke Høiland-Jørgensen #define MCS_GROUP_RATES 12
82db3e1c40SToke Høiland-Jørgensen
83db3e1c40SToke Høiland-Jørgensen #define HT_GROUP_IDX(_streams, _sgi, _ht40) \
84db3e1c40SToke Høiland-Jørgensen IEEE80211_HT_GROUP_0 + \
85db3e1c40SToke Høiland-Jørgensen IEEE80211_MAX_STREAMS * 2 * _ht40 + \
86db3e1c40SToke Høiland-Jørgensen IEEE80211_MAX_STREAMS * _sgi + \
87db3e1c40SToke Høiland-Jørgensen _streams - 1
88db3e1c40SToke Høiland-Jørgensen
89db3e1c40SToke Høiland-Jørgensen #define _MAX(a, b) (((a)>(b))?(a):(b))
90db3e1c40SToke Høiland-Jørgensen
91db3e1c40SToke Høiland-Jørgensen #define GROUP_SHIFT(duration) \
92db3e1c40SToke Høiland-Jørgensen _MAX(0, 16 - __builtin_clz(duration))
93db3e1c40SToke Høiland-Jørgensen
94db3e1c40SToke Høiland-Jørgensen /* MCS rate information for an MCS group */
95db3e1c40SToke Høiland-Jørgensen #define __MCS_GROUP(_streams, _sgi, _ht40, _s) \
96db3e1c40SToke Høiland-Jørgensen [HT_GROUP_IDX(_streams, _sgi, _ht40)] = { \
97db3e1c40SToke Høiland-Jørgensen .shift = _s, \
98db3e1c40SToke Høiland-Jørgensen .duration = { \
99db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 54 : 26), \
100db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 108 : 52), \
101db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 162 : 78), \
102db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 216 : 104), \
103db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 324 : 156), \
104db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 432 : 208), \
105db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 486 : 234), \
106db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 540 : 260) \
107db3e1c40SToke Høiland-Jørgensen } \
108db3e1c40SToke Høiland-Jørgensen }
109db3e1c40SToke Høiland-Jørgensen
110db3e1c40SToke Høiland-Jørgensen #define MCS_GROUP_SHIFT(_streams, _sgi, _ht40) \
111db3e1c40SToke Høiland-Jørgensen GROUP_SHIFT(MCS_DURATION(_streams, _sgi, _ht40 ? 54 : 26))
112db3e1c40SToke Høiland-Jørgensen
113db3e1c40SToke Høiland-Jørgensen #define MCS_GROUP(_streams, _sgi, _ht40) \
114db3e1c40SToke Høiland-Jørgensen __MCS_GROUP(_streams, _sgi, _ht40, \
115db3e1c40SToke Høiland-Jørgensen MCS_GROUP_SHIFT(_streams, _sgi, _ht40))
116db3e1c40SToke Høiland-Jørgensen
117db3e1c40SToke Høiland-Jørgensen #define VHT_GROUP_IDX(_streams, _sgi, _bw) \
118db3e1c40SToke Høiland-Jørgensen (IEEE80211_VHT_GROUP_0 + \
119db3e1c40SToke Høiland-Jørgensen IEEE80211_MAX_STREAMS * 2 * (_bw) + \
120db3e1c40SToke Høiland-Jørgensen IEEE80211_MAX_STREAMS * (_sgi) + \
121db3e1c40SToke Høiland-Jørgensen (_streams) - 1)
122db3e1c40SToke Høiland-Jørgensen
123db3e1c40SToke Høiland-Jørgensen #define BW2VBPS(_bw, r4, r3, r2, r1) \
124db3e1c40SToke Høiland-Jørgensen (_bw == BW_160 ? r4 : _bw == BW_80 ? r3 : _bw == BW_40 ? r2 : r1)
125db3e1c40SToke Høiland-Jørgensen
126db3e1c40SToke Høiland-Jørgensen #define __VHT_GROUP(_streams, _sgi, _bw, _s) \
127db3e1c40SToke Høiland-Jørgensen [VHT_GROUP_IDX(_streams, _sgi, _bw)] = { \
128db3e1c40SToke Høiland-Jørgensen .shift = _s, \
129db3e1c40SToke Høiland-Jørgensen .duration = { \
130db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, \
131db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 234, 117, 54, 26)), \
132db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, \
133db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 468, 234, 108, 52)), \
134db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, \
135db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 702, 351, 162, 78)), \
136db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, \
137db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 936, 468, 216, 104)), \
138db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, \
139db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 1404, 702, 324, 156)), \
140db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, \
141db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 1872, 936, 432, 208)), \
142db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, \
143db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 2106, 1053, 486, 234)), \
144db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, \
145db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 2340, 1170, 540, 260)), \
146db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, \
147db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 2808, 1404, 648, 312)), \
148db3e1c40SToke Høiland-Jørgensen MCS_DURATION_S(_s, _streams, _sgi, \
149db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 3120, 1560, 720, 346)) \
150db3e1c40SToke Høiland-Jørgensen } \
151db3e1c40SToke Høiland-Jørgensen }
152db3e1c40SToke Høiland-Jørgensen
153db3e1c40SToke Høiland-Jørgensen #define VHT_GROUP_SHIFT(_streams, _sgi, _bw) \
154db3e1c40SToke Høiland-Jørgensen GROUP_SHIFT(MCS_DURATION(_streams, _sgi, \
155db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 243, 117, 54, 26)))
156db3e1c40SToke Høiland-Jørgensen
157db3e1c40SToke Høiland-Jørgensen #define VHT_GROUP(_streams, _sgi, _bw) \
158db3e1c40SToke Høiland-Jørgensen __VHT_GROUP(_streams, _sgi, _bw, \
159db3e1c40SToke Høiland-Jørgensen VHT_GROUP_SHIFT(_streams, _sgi, _bw))
160db3e1c40SToke Høiland-Jørgensen
161db3e1c40SToke Høiland-Jørgensen
162db3e1c40SToke Høiland-Jørgensen #define HE_GROUP_IDX(_streams, _gi, _bw) \
163db3e1c40SToke Høiland-Jørgensen (IEEE80211_HE_GROUP_0 + \
164db3e1c40SToke Høiland-Jørgensen IEEE80211_HE_MAX_STREAMS * 3 * (_bw) + \
165db3e1c40SToke Høiland-Jørgensen IEEE80211_HE_MAX_STREAMS * (_gi) + \
166db3e1c40SToke Høiland-Jørgensen (_streams) - 1)
167db3e1c40SToke Høiland-Jørgensen
168db3e1c40SToke Høiland-Jørgensen #define __HE_GROUP(_streams, _gi, _bw, _s) \
169db3e1c40SToke Høiland-Jørgensen [HE_GROUP_IDX(_streams, _gi, _bw)] = { \
170db3e1c40SToke Høiland-Jørgensen .shift = _s, \
171db3e1c40SToke Høiland-Jørgensen .duration = { \
172db3e1c40SToke Høiland-Jørgensen HE_DURATION_S(_s, _streams, _gi, \
173db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 979, 489, 230, 115)), \
174db3e1c40SToke Høiland-Jørgensen HE_DURATION_S(_s, _streams, _gi, \
175db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 1958, 979, 475, 230)), \
176db3e1c40SToke Høiland-Jørgensen HE_DURATION_S(_s, _streams, _gi, \
177db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 2937, 1468, 705, 345)), \
178db3e1c40SToke Høiland-Jørgensen HE_DURATION_S(_s, _streams, _gi, \
179db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 3916, 1958, 936, 475)), \
180db3e1c40SToke Høiland-Jørgensen HE_DURATION_S(_s, _streams, _gi, \
181db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 5875, 2937, 1411, 705)), \
182db3e1c40SToke Høiland-Jørgensen HE_DURATION_S(_s, _streams, _gi, \
183db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 7833, 3916, 1872, 936)), \
184db3e1c40SToke Høiland-Jørgensen HE_DURATION_S(_s, _streams, _gi, \
185db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 8827, 4406, 2102, 1051)), \
186db3e1c40SToke Høiland-Jørgensen HE_DURATION_S(_s, _streams, _gi, \
187db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 9806, 4896, 2347, 1166)), \
188db3e1c40SToke Høiland-Jørgensen HE_DURATION_S(_s, _streams, _gi, \
189db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 11764, 5875, 2808, 1411)), \
190db3e1c40SToke Høiland-Jørgensen HE_DURATION_S(_s, _streams, _gi, \
191db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 13060, 6523, 3124, 1555)), \
192db3e1c40SToke Høiland-Jørgensen HE_DURATION_S(_s, _streams, _gi, \
193db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 14702, 7344, 3513, 1756)), \
194db3e1c40SToke Høiland-Jørgensen HE_DURATION_S(_s, _streams, _gi, \
195db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 16329, 8164, 3902, 1944)) \
196db3e1c40SToke Høiland-Jørgensen } \
197db3e1c40SToke Høiland-Jørgensen }
198db3e1c40SToke Høiland-Jørgensen
199db3e1c40SToke Høiland-Jørgensen #define HE_GROUP_SHIFT(_streams, _gi, _bw) \
200db3e1c40SToke Høiland-Jørgensen GROUP_SHIFT(HE_DURATION(_streams, _gi, \
201db3e1c40SToke Høiland-Jørgensen BW2VBPS(_bw, 979, 489, 230, 115)))
202db3e1c40SToke Høiland-Jørgensen
203db3e1c40SToke Høiland-Jørgensen #define HE_GROUP(_streams, _gi, _bw) \
204db3e1c40SToke Høiland-Jørgensen __HE_GROUP(_streams, _gi, _bw, \
205db3e1c40SToke Høiland-Jørgensen HE_GROUP_SHIFT(_streams, _gi, _bw))
206db3e1c40SToke Høiland-Jørgensen struct mcs_group {
207db3e1c40SToke Høiland-Jørgensen u8 shift;
208db3e1c40SToke Høiland-Jørgensen u16 duration[MCS_GROUP_RATES];
209db3e1c40SToke Høiland-Jørgensen };
210db3e1c40SToke Høiland-Jørgensen
211db3e1c40SToke Høiland-Jørgensen static const struct mcs_group airtime_mcs_groups[] = {
212db3e1c40SToke Høiland-Jørgensen MCS_GROUP(1, 0, BW_20),
213db3e1c40SToke Høiland-Jørgensen MCS_GROUP(2, 0, BW_20),
214db3e1c40SToke Høiland-Jørgensen MCS_GROUP(3, 0, BW_20),
215db3e1c40SToke Høiland-Jørgensen MCS_GROUP(4, 0, BW_20),
216db3e1c40SToke Høiland-Jørgensen
217db3e1c40SToke Høiland-Jørgensen MCS_GROUP(1, 1, BW_20),
218db3e1c40SToke Høiland-Jørgensen MCS_GROUP(2, 1, BW_20),
219db3e1c40SToke Høiland-Jørgensen MCS_GROUP(3, 1, BW_20),
220db3e1c40SToke Høiland-Jørgensen MCS_GROUP(4, 1, BW_20),
221db3e1c40SToke Høiland-Jørgensen
222db3e1c40SToke Høiland-Jørgensen MCS_GROUP(1, 0, BW_40),
223db3e1c40SToke Høiland-Jørgensen MCS_GROUP(2, 0, BW_40),
224db3e1c40SToke Høiland-Jørgensen MCS_GROUP(3, 0, BW_40),
225db3e1c40SToke Høiland-Jørgensen MCS_GROUP(4, 0, BW_40),
226db3e1c40SToke Høiland-Jørgensen
227db3e1c40SToke Høiland-Jørgensen MCS_GROUP(1, 1, BW_40),
228db3e1c40SToke Høiland-Jørgensen MCS_GROUP(2, 1, BW_40),
229db3e1c40SToke Høiland-Jørgensen MCS_GROUP(3, 1, BW_40),
230db3e1c40SToke Høiland-Jørgensen MCS_GROUP(4, 1, BW_40),
231db3e1c40SToke Høiland-Jørgensen
232db3e1c40SToke Høiland-Jørgensen VHT_GROUP(1, 0, BW_20),
233db3e1c40SToke Høiland-Jørgensen VHT_GROUP(2, 0, BW_20),
234db3e1c40SToke Høiland-Jørgensen VHT_GROUP(3, 0, BW_20),
235db3e1c40SToke Høiland-Jørgensen VHT_GROUP(4, 0, BW_20),
236db3e1c40SToke Høiland-Jørgensen
237db3e1c40SToke Høiland-Jørgensen VHT_GROUP(1, 1, BW_20),
238db3e1c40SToke Høiland-Jørgensen VHT_GROUP(2, 1, BW_20),
239db3e1c40SToke Høiland-Jørgensen VHT_GROUP(3, 1, BW_20),
240db3e1c40SToke Høiland-Jørgensen VHT_GROUP(4, 1, BW_20),
241db3e1c40SToke Høiland-Jørgensen
242db3e1c40SToke Høiland-Jørgensen VHT_GROUP(1, 0, BW_40),
243db3e1c40SToke Høiland-Jørgensen VHT_GROUP(2, 0, BW_40),
244db3e1c40SToke Høiland-Jørgensen VHT_GROUP(3, 0, BW_40),
245db3e1c40SToke Høiland-Jørgensen VHT_GROUP(4, 0, BW_40),
246db3e1c40SToke Høiland-Jørgensen
247db3e1c40SToke Høiland-Jørgensen VHT_GROUP(1, 1, BW_40),
248db3e1c40SToke Høiland-Jørgensen VHT_GROUP(2, 1, BW_40),
249db3e1c40SToke Høiland-Jørgensen VHT_GROUP(3, 1, BW_40),
250db3e1c40SToke Høiland-Jørgensen VHT_GROUP(4, 1, BW_40),
251db3e1c40SToke Høiland-Jørgensen
252db3e1c40SToke Høiland-Jørgensen VHT_GROUP(1, 0, BW_80),
253db3e1c40SToke Høiland-Jørgensen VHT_GROUP(2, 0, BW_80),
254db3e1c40SToke Høiland-Jørgensen VHT_GROUP(3, 0, BW_80),
255db3e1c40SToke Høiland-Jørgensen VHT_GROUP(4, 0, BW_80),
256db3e1c40SToke Høiland-Jørgensen
257db3e1c40SToke Høiland-Jørgensen VHT_GROUP(1, 1, BW_80),
258db3e1c40SToke Høiland-Jørgensen VHT_GROUP(2, 1, BW_80),
259db3e1c40SToke Høiland-Jørgensen VHT_GROUP(3, 1, BW_80),
260db3e1c40SToke Høiland-Jørgensen VHT_GROUP(4, 1, BW_80),
261db3e1c40SToke Høiland-Jørgensen
262db3e1c40SToke Høiland-Jørgensen VHT_GROUP(1, 0, BW_160),
263db3e1c40SToke Høiland-Jørgensen VHT_GROUP(2, 0, BW_160),
264db3e1c40SToke Høiland-Jørgensen VHT_GROUP(3, 0, BW_160),
265db3e1c40SToke Høiland-Jørgensen VHT_GROUP(4, 0, BW_160),
266db3e1c40SToke Høiland-Jørgensen
267db3e1c40SToke Høiland-Jørgensen VHT_GROUP(1, 1, BW_160),
268db3e1c40SToke Høiland-Jørgensen VHT_GROUP(2, 1, BW_160),
269db3e1c40SToke Høiland-Jørgensen VHT_GROUP(3, 1, BW_160),
270db3e1c40SToke Høiland-Jørgensen VHT_GROUP(4, 1, BW_160),
271db3e1c40SToke Høiland-Jørgensen
272db3e1c40SToke Høiland-Jørgensen HE_GROUP(1, HE_GI_08, BW_20),
273db3e1c40SToke Høiland-Jørgensen HE_GROUP(2, HE_GI_08, BW_20),
274db3e1c40SToke Høiland-Jørgensen HE_GROUP(3, HE_GI_08, BW_20),
275db3e1c40SToke Høiland-Jørgensen HE_GROUP(4, HE_GI_08, BW_20),
276db3e1c40SToke Høiland-Jørgensen HE_GROUP(5, HE_GI_08, BW_20),
277db3e1c40SToke Høiland-Jørgensen HE_GROUP(6, HE_GI_08, BW_20),
278db3e1c40SToke Høiland-Jørgensen HE_GROUP(7, HE_GI_08, BW_20),
279db3e1c40SToke Høiland-Jørgensen HE_GROUP(8, HE_GI_08, BW_20),
280db3e1c40SToke Høiland-Jørgensen
281db3e1c40SToke Høiland-Jørgensen HE_GROUP(1, HE_GI_16, BW_20),
282db3e1c40SToke Høiland-Jørgensen HE_GROUP(2, HE_GI_16, BW_20),
283db3e1c40SToke Høiland-Jørgensen HE_GROUP(3, HE_GI_16, BW_20),
284db3e1c40SToke Høiland-Jørgensen HE_GROUP(4, HE_GI_16, BW_20),
285db3e1c40SToke Høiland-Jørgensen HE_GROUP(5, HE_GI_16, BW_20),
286db3e1c40SToke Høiland-Jørgensen HE_GROUP(6, HE_GI_16, BW_20),
287db3e1c40SToke Høiland-Jørgensen HE_GROUP(7, HE_GI_16, BW_20),
288db3e1c40SToke Høiland-Jørgensen HE_GROUP(8, HE_GI_16, BW_20),
289db3e1c40SToke Høiland-Jørgensen
290db3e1c40SToke Høiland-Jørgensen HE_GROUP(1, HE_GI_32, BW_20),
291db3e1c40SToke Høiland-Jørgensen HE_GROUP(2, HE_GI_32, BW_20),
292db3e1c40SToke Høiland-Jørgensen HE_GROUP(3, HE_GI_32, BW_20),
293db3e1c40SToke Høiland-Jørgensen HE_GROUP(4, HE_GI_32, BW_20),
294db3e1c40SToke Høiland-Jørgensen HE_GROUP(5, HE_GI_32, BW_20),
295db3e1c40SToke Høiland-Jørgensen HE_GROUP(6, HE_GI_32, BW_20),
296db3e1c40SToke Høiland-Jørgensen HE_GROUP(7, HE_GI_32, BW_20),
297db3e1c40SToke Høiland-Jørgensen HE_GROUP(8, HE_GI_32, BW_20),
298db3e1c40SToke Høiland-Jørgensen
299db3e1c40SToke Høiland-Jørgensen HE_GROUP(1, HE_GI_08, BW_40),
300db3e1c40SToke Høiland-Jørgensen HE_GROUP(2, HE_GI_08, BW_40),
301db3e1c40SToke Høiland-Jørgensen HE_GROUP(3, HE_GI_08, BW_40),
302db3e1c40SToke Høiland-Jørgensen HE_GROUP(4, HE_GI_08, BW_40),
303db3e1c40SToke Høiland-Jørgensen HE_GROUP(5, HE_GI_08, BW_40),
304db3e1c40SToke Høiland-Jørgensen HE_GROUP(6, HE_GI_08, BW_40),
305db3e1c40SToke Høiland-Jørgensen HE_GROUP(7, HE_GI_08, BW_40),
306db3e1c40SToke Høiland-Jørgensen HE_GROUP(8, HE_GI_08, BW_40),
307db3e1c40SToke Høiland-Jørgensen
308db3e1c40SToke Høiland-Jørgensen HE_GROUP(1, HE_GI_16, BW_40),
309db3e1c40SToke Høiland-Jørgensen HE_GROUP(2, HE_GI_16, BW_40),
310db3e1c40SToke Høiland-Jørgensen HE_GROUP(3, HE_GI_16, BW_40),
311db3e1c40SToke Høiland-Jørgensen HE_GROUP(4, HE_GI_16, BW_40),
312db3e1c40SToke Høiland-Jørgensen HE_GROUP(5, HE_GI_16, BW_40),
313db3e1c40SToke Høiland-Jørgensen HE_GROUP(6, HE_GI_16, BW_40),
314db3e1c40SToke Høiland-Jørgensen HE_GROUP(7, HE_GI_16, BW_40),
315db3e1c40SToke Høiland-Jørgensen HE_GROUP(8, HE_GI_16, BW_40),
316db3e1c40SToke Høiland-Jørgensen
317db3e1c40SToke Høiland-Jørgensen HE_GROUP(1, HE_GI_32, BW_40),
318db3e1c40SToke Høiland-Jørgensen HE_GROUP(2, HE_GI_32, BW_40),
319db3e1c40SToke Høiland-Jørgensen HE_GROUP(3, HE_GI_32, BW_40),
320db3e1c40SToke Høiland-Jørgensen HE_GROUP(4, HE_GI_32, BW_40),
321db3e1c40SToke Høiland-Jørgensen HE_GROUP(5, HE_GI_32, BW_40),
322db3e1c40SToke Høiland-Jørgensen HE_GROUP(6, HE_GI_32, BW_40),
323db3e1c40SToke Høiland-Jørgensen HE_GROUP(7, HE_GI_32, BW_40),
324db3e1c40SToke Høiland-Jørgensen HE_GROUP(8, HE_GI_32, BW_40),
325db3e1c40SToke Høiland-Jørgensen
326db3e1c40SToke Høiland-Jørgensen HE_GROUP(1, HE_GI_08, BW_80),
327db3e1c40SToke Høiland-Jørgensen HE_GROUP(2, HE_GI_08, BW_80),
328db3e1c40SToke Høiland-Jørgensen HE_GROUP(3, HE_GI_08, BW_80),
329db3e1c40SToke Høiland-Jørgensen HE_GROUP(4, HE_GI_08, BW_80),
330db3e1c40SToke Høiland-Jørgensen HE_GROUP(5, HE_GI_08, BW_80),
331db3e1c40SToke Høiland-Jørgensen HE_GROUP(6, HE_GI_08, BW_80),
332db3e1c40SToke Høiland-Jørgensen HE_GROUP(7, HE_GI_08, BW_80),
333db3e1c40SToke Høiland-Jørgensen HE_GROUP(8, HE_GI_08, BW_80),
334db3e1c40SToke Høiland-Jørgensen
335db3e1c40SToke Høiland-Jørgensen HE_GROUP(1, HE_GI_16, BW_80),
336db3e1c40SToke Høiland-Jørgensen HE_GROUP(2, HE_GI_16, BW_80),
337db3e1c40SToke Høiland-Jørgensen HE_GROUP(3, HE_GI_16, BW_80),
338db3e1c40SToke Høiland-Jørgensen HE_GROUP(4, HE_GI_16, BW_80),
339db3e1c40SToke Høiland-Jørgensen HE_GROUP(5, HE_GI_16, BW_80),
340db3e1c40SToke Høiland-Jørgensen HE_GROUP(6, HE_GI_16, BW_80),
341db3e1c40SToke Høiland-Jørgensen HE_GROUP(7, HE_GI_16, BW_80),
342db3e1c40SToke Høiland-Jørgensen HE_GROUP(8, HE_GI_16, BW_80),
343db3e1c40SToke Høiland-Jørgensen
344db3e1c40SToke Høiland-Jørgensen HE_GROUP(1, HE_GI_32, BW_80),
345db3e1c40SToke Høiland-Jørgensen HE_GROUP(2, HE_GI_32, BW_80),
346db3e1c40SToke Høiland-Jørgensen HE_GROUP(3, HE_GI_32, BW_80),
347db3e1c40SToke Høiland-Jørgensen HE_GROUP(4, HE_GI_32, BW_80),
348db3e1c40SToke Høiland-Jørgensen HE_GROUP(5, HE_GI_32, BW_80),
349db3e1c40SToke Høiland-Jørgensen HE_GROUP(6, HE_GI_32, BW_80),
350db3e1c40SToke Høiland-Jørgensen HE_GROUP(7, HE_GI_32, BW_80),
351db3e1c40SToke Høiland-Jørgensen HE_GROUP(8, HE_GI_32, BW_80),
352db3e1c40SToke Høiland-Jørgensen
353db3e1c40SToke Høiland-Jørgensen HE_GROUP(1, HE_GI_08, BW_160),
354db3e1c40SToke Høiland-Jørgensen HE_GROUP(2, HE_GI_08, BW_160),
355db3e1c40SToke Høiland-Jørgensen HE_GROUP(3, HE_GI_08, BW_160),
356db3e1c40SToke Høiland-Jørgensen HE_GROUP(4, HE_GI_08, BW_160),
357db3e1c40SToke Høiland-Jørgensen HE_GROUP(5, HE_GI_08, BW_160),
358db3e1c40SToke Høiland-Jørgensen HE_GROUP(6, HE_GI_08, BW_160),
359db3e1c40SToke Høiland-Jørgensen HE_GROUP(7, HE_GI_08, BW_160),
360db3e1c40SToke Høiland-Jørgensen HE_GROUP(8, HE_GI_08, BW_160),
361db3e1c40SToke Høiland-Jørgensen
362db3e1c40SToke Høiland-Jørgensen HE_GROUP(1, HE_GI_16, BW_160),
363db3e1c40SToke Høiland-Jørgensen HE_GROUP(2, HE_GI_16, BW_160),
364db3e1c40SToke Høiland-Jørgensen HE_GROUP(3, HE_GI_16, BW_160),
365db3e1c40SToke Høiland-Jørgensen HE_GROUP(4, HE_GI_16, BW_160),
366db3e1c40SToke Høiland-Jørgensen HE_GROUP(5, HE_GI_16, BW_160),
367db3e1c40SToke Høiland-Jørgensen HE_GROUP(6, HE_GI_16, BW_160),
368db3e1c40SToke Høiland-Jørgensen HE_GROUP(7, HE_GI_16, BW_160),
369db3e1c40SToke Høiland-Jørgensen HE_GROUP(8, HE_GI_16, BW_160),
370db3e1c40SToke Høiland-Jørgensen
371db3e1c40SToke Høiland-Jørgensen HE_GROUP(1, HE_GI_32, BW_160),
372db3e1c40SToke Høiland-Jørgensen HE_GROUP(2, HE_GI_32, BW_160),
373db3e1c40SToke Høiland-Jørgensen HE_GROUP(3, HE_GI_32, BW_160),
374db3e1c40SToke Høiland-Jørgensen HE_GROUP(4, HE_GI_32, BW_160),
375db3e1c40SToke Høiland-Jørgensen HE_GROUP(5, HE_GI_32, BW_160),
376db3e1c40SToke Høiland-Jørgensen HE_GROUP(6, HE_GI_32, BW_160),
377db3e1c40SToke Høiland-Jørgensen HE_GROUP(7, HE_GI_32, BW_160),
378db3e1c40SToke Høiland-Jørgensen HE_GROUP(8, HE_GI_32, BW_160),
379db3e1c40SToke Høiland-Jørgensen };
380db3e1c40SToke Høiland-Jørgensen
381db3e1c40SToke Høiland-Jørgensen static u32
ieee80211_calc_legacy_rate_duration(u16 bitrate,bool short_pre,bool cck,int len)382db3e1c40SToke Høiland-Jørgensen ieee80211_calc_legacy_rate_duration(u16 bitrate, bool short_pre,
383db3e1c40SToke Høiland-Jørgensen bool cck, int len)
384db3e1c40SToke Høiland-Jørgensen {
385db3e1c40SToke Høiland-Jørgensen u32 duration;
386db3e1c40SToke Høiland-Jørgensen
387db3e1c40SToke Høiland-Jørgensen if (cck) {
388db3e1c40SToke Høiland-Jørgensen duration = 144 + 48; /* preamble + PLCP */
389db3e1c40SToke Høiland-Jørgensen if (short_pre)
390db3e1c40SToke Høiland-Jørgensen duration >>= 1;
391db3e1c40SToke Høiland-Jørgensen
392db3e1c40SToke Høiland-Jørgensen duration += 10; /* SIFS */
393db3e1c40SToke Høiland-Jørgensen } else {
394db3e1c40SToke Høiland-Jørgensen duration = 20 + 16; /* premable + SIFS */
395db3e1c40SToke Høiland-Jørgensen }
396db3e1c40SToke Høiland-Jørgensen
397db3e1c40SToke Høiland-Jørgensen len <<= 3;
398db3e1c40SToke Høiland-Jørgensen duration += (len * 10) / bitrate;
399db3e1c40SToke Høiland-Jørgensen
400db3e1c40SToke Høiland-Jørgensen return duration;
401db3e1c40SToke Høiland-Jørgensen }
402db3e1c40SToke Høiland-Jørgensen
ieee80211_get_rate_duration(struct ieee80211_hw * hw,struct ieee80211_rx_status * status,u32 * overhead)40343cd72c5SFelix Fietkau static u32 ieee80211_get_rate_duration(struct ieee80211_hw *hw,
404db3e1c40SToke Høiland-Jørgensen struct ieee80211_rx_status *status,
40543cd72c5SFelix Fietkau u32 *overhead)
406db3e1c40SToke Høiland-Jørgensen {
407db3e1c40SToke Høiland-Jørgensen bool sgi = status->enc_flags & RX_ENC_FLAG_SHORT_GI;
408db3e1c40SToke Høiland-Jørgensen int bw, streams;
409db3e1c40SToke Høiland-Jørgensen int group, idx;
410db3e1c40SToke Høiland-Jørgensen u32 duration;
411db3e1c40SToke Høiland-Jørgensen
412db3e1c40SToke Høiland-Jørgensen switch (status->bw) {
413db3e1c40SToke Høiland-Jørgensen case RATE_INFO_BW_20:
414db3e1c40SToke Høiland-Jørgensen bw = BW_20;
415db3e1c40SToke Høiland-Jørgensen break;
416db3e1c40SToke Høiland-Jørgensen case RATE_INFO_BW_40:
417db3e1c40SToke Høiland-Jørgensen bw = BW_40;
418db3e1c40SToke Høiland-Jørgensen break;
419db3e1c40SToke Høiland-Jørgensen case RATE_INFO_BW_80:
420db3e1c40SToke Høiland-Jørgensen bw = BW_80;
421db3e1c40SToke Høiland-Jørgensen break;
422db3e1c40SToke Høiland-Jørgensen case RATE_INFO_BW_160:
423db3e1c40SToke Høiland-Jørgensen bw = BW_160;
424db3e1c40SToke Høiland-Jørgensen break;
425db3e1c40SToke Høiland-Jørgensen default:
426db3e1c40SToke Høiland-Jørgensen WARN_ON_ONCE(1);
427db3e1c40SToke Høiland-Jørgensen return 0;
428db3e1c40SToke Høiland-Jørgensen }
429db3e1c40SToke Høiland-Jørgensen
430db3e1c40SToke Høiland-Jørgensen switch (status->encoding) {
431db3e1c40SToke Høiland-Jørgensen case RX_ENC_VHT:
432db3e1c40SToke Høiland-Jørgensen streams = status->nss;
433db3e1c40SToke Høiland-Jørgensen idx = status->rate_idx;
434db3e1c40SToke Høiland-Jørgensen group = VHT_GROUP_IDX(streams, sgi, bw);
435db3e1c40SToke Høiland-Jørgensen break;
436db3e1c40SToke Høiland-Jørgensen case RX_ENC_HT:
437db3e1c40SToke Høiland-Jørgensen streams = ((status->rate_idx >> 3) & 3) + 1;
438db3e1c40SToke Høiland-Jørgensen idx = status->rate_idx & 7;
439db3e1c40SToke Høiland-Jørgensen group = HT_GROUP_IDX(streams, sgi, bw);
440db3e1c40SToke Høiland-Jørgensen break;
441db3e1c40SToke Høiland-Jørgensen case RX_ENC_HE:
442db3e1c40SToke Høiland-Jørgensen streams = status->nss;
443db3e1c40SToke Høiland-Jørgensen idx = status->rate_idx;
444db3e1c40SToke Høiland-Jørgensen group = HE_GROUP_IDX(streams, status->he_gi, bw);
445db3e1c40SToke Høiland-Jørgensen break;
446db3e1c40SToke Høiland-Jørgensen default:
447db3e1c40SToke Høiland-Jørgensen WARN_ON_ONCE(1);
448db3e1c40SToke Høiland-Jørgensen return 0;
449db3e1c40SToke Høiland-Jørgensen }
450db3e1c40SToke Høiland-Jørgensen
451db3e1c40SToke Høiland-Jørgensen if (WARN_ON_ONCE((status->encoding != RX_ENC_HE && streams > 4) ||
452db3e1c40SToke Høiland-Jørgensen (status->encoding == RX_ENC_HE && streams > 8)))
453db3e1c40SToke Høiland-Jørgensen return 0;
454db3e1c40SToke Høiland-Jørgensen
455*3e8f7abcSLorenzo Bianconi if (idx >= MCS_GROUP_RATES)
456*3e8f7abcSLorenzo Bianconi return 0;
457*3e8f7abcSLorenzo Bianconi
458db3e1c40SToke Høiland-Jørgensen duration = airtime_mcs_groups[group].duration[idx];
459db3e1c40SToke Høiland-Jørgensen duration <<= airtime_mcs_groups[group].shift;
46043cd72c5SFelix Fietkau *overhead = 36 + (streams << 2);
46143cd72c5SFelix Fietkau
46243cd72c5SFelix Fietkau return duration;
46343cd72c5SFelix Fietkau }
46443cd72c5SFelix Fietkau
46543cd72c5SFelix Fietkau
ieee80211_calc_rx_airtime(struct ieee80211_hw * hw,struct ieee80211_rx_status * status,int len)46643cd72c5SFelix Fietkau u32 ieee80211_calc_rx_airtime(struct ieee80211_hw *hw,
46743cd72c5SFelix Fietkau struct ieee80211_rx_status *status,
46843cd72c5SFelix Fietkau int len)
46943cd72c5SFelix Fietkau {
47043cd72c5SFelix Fietkau struct ieee80211_supported_band *sband;
47143cd72c5SFelix Fietkau u32 duration, overhead = 0;
47243cd72c5SFelix Fietkau
47343cd72c5SFelix Fietkau if (status->encoding == RX_ENC_LEGACY) {
47443cd72c5SFelix Fietkau const struct ieee80211_rate *rate;
47543cd72c5SFelix Fietkau bool sp = status->enc_flags & RX_ENC_FLAG_SHORTPRE;
47643cd72c5SFelix Fietkau bool cck;
47743cd72c5SFelix Fietkau
478022143d0SMeiChia Chiu /* on 60GHz or sub-1GHz band, there are no legacy rates */
479022143d0SMeiChia Chiu if (WARN_ON_ONCE(status->band == NL80211_BAND_60GHZ ||
480022143d0SMeiChia Chiu status->band == NL80211_BAND_S1GHZ))
48143cd72c5SFelix Fietkau return 0;
48243cd72c5SFelix Fietkau
48343cd72c5SFelix Fietkau sband = hw->wiphy->bands[status->band];
48443cd72c5SFelix Fietkau if (!sband || status->rate_idx >= sband->n_bitrates)
48543cd72c5SFelix Fietkau return 0;
48643cd72c5SFelix Fietkau
48743cd72c5SFelix Fietkau rate = &sband->bitrates[status->rate_idx];
48843cd72c5SFelix Fietkau cck = rate->flags & IEEE80211_RATE_MANDATORY_B;
48943cd72c5SFelix Fietkau
49043cd72c5SFelix Fietkau return ieee80211_calc_legacy_rate_duration(rate->bitrate, sp,
49143cd72c5SFelix Fietkau cck, len);
49243cd72c5SFelix Fietkau }
49343cd72c5SFelix Fietkau
49443cd72c5SFelix Fietkau duration = ieee80211_get_rate_duration(hw, status, &overhead);
49543cd72c5SFelix Fietkau if (!duration)
49643cd72c5SFelix Fietkau return 0;
49743cd72c5SFelix Fietkau
498db3e1c40SToke Høiland-Jørgensen duration *= len;
499db3e1c40SToke Høiland-Jørgensen duration /= AVG_PKT_SIZE;
500db3e1c40SToke Høiland-Jørgensen duration /= 1024;
501db3e1c40SToke Høiland-Jørgensen
50243cd72c5SFelix Fietkau return duration + overhead;
503db3e1c40SToke Høiland-Jørgensen }
504db3e1c40SToke Høiland-Jørgensen EXPORT_SYMBOL_GPL(ieee80211_calc_rx_airtime);
505db3e1c40SToke Høiland-Jørgensen
ieee80211_fill_rate_info(struct ieee80211_hw * hw,struct ieee80211_rx_status * stat,u8 band,struct rate_info * ri)5068ed37e79SFelix Fietkau static bool ieee80211_fill_rate_info(struct ieee80211_hw *hw,
5078ed37e79SFelix Fietkau struct ieee80211_rx_status *stat, u8 band,
5088ed37e79SFelix Fietkau struct rate_info *ri)
5098ed37e79SFelix Fietkau {
5108ed37e79SFelix Fietkau struct ieee80211_supported_band *sband = hw->wiphy->bands[band];
5118ed37e79SFelix Fietkau int i;
5128ed37e79SFelix Fietkau
5138ed37e79SFelix Fietkau if (!ri || !sband)
5148ed37e79SFelix Fietkau return false;
5158ed37e79SFelix Fietkau
5168ed37e79SFelix Fietkau stat->bw = ri->bw;
5178ed37e79SFelix Fietkau stat->nss = ri->nss;
5188ed37e79SFelix Fietkau stat->rate_idx = ri->mcs;
5198ed37e79SFelix Fietkau
5208ed37e79SFelix Fietkau if (ri->flags & RATE_INFO_FLAGS_HE_MCS)
5218ed37e79SFelix Fietkau stat->encoding = RX_ENC_HE;
5228ed37e79SFelix Fietkau else if (ri->flags & RATE_INFO_FLAGS_VHT_MCS)
5238ed37e79SFelix Fietkau stat->encoding = RX_ENC_VHT;
5248ed37e79SFelix Fietkau else if (ri->flags & RATE_INFO_FLAGS_MCS)
5258ed37e79SFelix Fietkau stat->encoding = RX_ENC_HT;
5268ed37e79SFelix Fietkau else
5278ed37e79SFelix Fietkau stat->encoding = RX_ENC_LEGACY;
5288ed37e79SFelix Fietkau
5298ed37e79SFelix Fietkau if (ri->flags & RATE_INFO_FLAGS_SHORT_GI)
5308ed37e79SFelix Fietkau stat->enc_flags |= RX_ENC_FLAG_SHORT_GI;
5318ed37e79SFelix Fietkau
5328ed37e79SFelix Fietkau stat->he_gi = ri->he_gi;
5338ed37e79SFelix Fietkau
5348ed37e79SFelix Fietkau if (stat->encoding != RX_ENC_LEGACY)
5358ed37e79SFelix Fietkau return true;
5368ed37e79SFelix Fietkau
5378ed37e79SFelix Fietkau stat->rate_idx = 0;
5388ed37e79SFelix Fietkau for (i = 0; i < sband->n_bitrates; i++) {
5398ed37e79SFelix Fietkau if (ri->legacy != sband->bitrates[i].bitrate)
5408ed37e79SFelix Fietkau continue;
5418ed37e79SFelix Fietkau
5428ed37e79SFelix Fietkau stat->rate_idx = i;
5438ed37e79SFelix Fietkau return true;
5448ed37e79SFelix Fietkau }
5458ed37e79SFelix Fietkau
5468ed37e79SFelix Fietkau return false;
5478ed37e79SFelix Fietkau }
5488ed37e79SFelix Fietkau
ieee80211_fill_rx_status(struct ieee80211_rx_status * stat,struct ieee80211_hw * hw,struct ieee80211_tx_rate * rate,struct rate_info * ri,u8 band,int len)54943cd72c5SFelix Fietkau static int ieee80211_fill_rx_status(struct ieee80211_rx_status *stat,
55043cd72c5SFelix Fietkau struct ieee80211_hw *hw,
55143cd72c5SFelix Fietkau struct ieee80211_tx_rate *rate,
55243cd72c5SFelix Fietkau struct rate_info *ri, u8 band, int len)
55343cd72c5SFelix Fietkau {
55443cd72c5SFelix Fietkau memset(stat, 0, sizeof(*stat));
55543cd72c5SFelix Fietkau stat->band = band;
55643cd72c5SFelix Fietkau
55743cd72c5SFelix Fietkau if (ieee80211_fill_rate_info(hw, stat, band, ri))
55843cd72c5SFelix Fietkau return 0;
55943cd72c5SFelix Fietkau
56043cd72c5SFelix Fietkau if (rate->idx < 0 || !rate->count)
56143cd72c5SFelix Fietkau return -1;
56243cd72c5SFelix Fietkau
5638e280369SFelix Fietkau if (rate->flags & IEEE80211_TX_RC_160_MHZ_WIDTH)
5648e280369SFelix Fietkau stat->bw = RATE_INFO_BW_160;
5658e280369SFelix Fietkau else if (rate->flags & IEEE80211_TX_RC_80_MHZ_WIDTH)
56643cd72c5SFelix Fietkau stat->bw = RATE_INFO_BW_80;
56743cd72c5SFelix Fietkau else if (rate->flags & IEEE80211_TX_RC_40_MHZ_WIDTH)
56843cd72c5SFelix Fietkau stat->bw = RATE_INFO_BW_40;
56943cd72c5SFelix Fietkau else
57043cd72c5SFelix Fietkau stat->bw = RATE_INFO_BW_20;
57143cd72c5SFelix Fietkau
57243cd72c5SFelix Fietkau stat->enc_flags = 0;
57343cd72c5SFelix Fietkau if (rate->flags & IEEE80211_TX_RC_USE_SHORT_PREAMBLE)
57443cd72c5SFelix Fietkau stat->enc_flags |= RX_ENC_FLAG_SHORTPRE;
57543cd72c5SFelix Fietkau if (rate->flags & IEEE80211_TX_RC_SHORT_GI)
57643cd72c5SFelix Fietkau stat->enc_flags |= RX_ENC_FLAG_SHORT_GI;
57743cd72c5SFelix Fietkau
57843cd72c5SFelix Fietkau stat->rate_idx = rate->idx;
57943cd72c5SFelix Fietkau if (rate->flags & IEEE80211_TX_RC_VHT_MCS) {
58043cd72c5SFelix Fietkau stat->encoding = RX_ENC_VHT;
58143cd72c5SFelix Fietkau stat->rate_idx = ieee80211_rate_get_vht_mcs(rate);
58243cd72c5SFelix Fietkau stat->nss = ieee80211_rate_get_vht_nss(rate);
58343cd72c5SFelix Fietkau } else if (rate->flags & IEEE80211_TX_RC_MCS) {
58443cd72c5SFelix Fietkau stat->encoding = RX_ENC_HT;
58543cd72c5SFelix Fietkau } else {
58643cd72c5SFelix Fietkau stat->encoding = RX_ENC_LEGACY;
58743cd72c5SFelix Fietkau }
58843cd72c5SFelix Fietkau
58943cd72c5SFelix Fietkau return 0;
59043cd72c5SFelix Fietkau }
59143cd72c5SFelix Fietkau
ieee80211_calc_tx_airtime_rate(struct ieee80211_hw * hw,struct ieee80211_tx_rate * rate,struct rate_info * ri,u8 band,int len)592db3e1c40SToke Høiland-Jørgensen static u32 ieee80211_calc_tx_airtime_rate(struct ieee80211_hw *hw,
593db3e1c40SToke Høiland-Jørgensen struct ieee80211_tx_rate *rate,
5948ed37e79SFelix Fietkau struct rate_info *ri,
595db3e1c40SToke Høiland-Jørgensen u8 band, int len)
596db3e1c40SToke Høiland-Jørgensen {
59743cd72c5SFelix Fietkau struct ieee80211_rx_status stat;
598db3e1c40SToke Høiland-Jørgensen
59943cd72c5SFelix Fietkau if (ieee80211_fill_rx_status(&stat, hw, rate, ri, band, len))
600db3e1c40SToke Høiland-Jørgensen return 0;
601db3e1c40SToke Høiland-Jørgensen
602db3e1c40SToke Høiland-Jørgensen return ieee80211_calc_rx_airtime(hw, &stat, len);
603db3e1c40SToke Høiland-Jørgensen }
604db3e1c40SToke Høiland-Jørgensen
ieee80211_calc_tx_airtime(struct ieee80211_hw * hw,struct ieee80211_tx_info * info,int len)605db3e1c40SToke Høiland-Jørgensen u32 ieee80211_calc_tx_airtime(struct ieee80211_hw *hw,
606db3e1c40SToke Høiland-Jørgensen struct ieee80211_tx_info *info,
607db3e1c40SToke Høiland-Jørgensen int len)
608db3e1c40SToke Høiland-Jørgensen {
609db3e1c40SToke Høiland-Jørgensen u32 duration = 0;
610db3e1c40SToke Høiland-Jørgensen int i;
611db3e1c40SToke Høiland-Jørgensen
612db3e1c40SToke Høiland-Jørgensen for (i = 0; i < ARRAY_SIZE(info->status.rates); i++) {
613db3e1c40SToke Høiland-Jørgensen struct ieee80211_tx_rate *rate = &info->status.rates[i];
614db3e1c40SToke Høiland-Jørgensen u32 cur_duration;
615db3e1c40SToke Høiland-Jørgensen
6168ed37e79SFelix Fietkau cur_duration = ieee80211_calc_tx_airtime_rate(hw, rate, NULL,
617db3e1c40SToke Høiland-Jørgensen info->band, len);
618db3e1c40SToke Høiland-Jørgensen if (!cur_duration)
619db3e1c40SToke Høiland-Jørgensen break;
620db3e1c40SToke Høiland-Jørgensen
621db3e1c40SToke Høiland-Jørgensen duration += cur_duration * rate->count;
622db3e1c40SToke Høiland-Jørgensen }
623db3e1c40SToke Høiland-Jørgensen
624db3e1c40SToke Høiland-Jørgensen return duration;
625db3e1c40SToke Høiland-Jørgensen }
626db3e1c40SToke Høiland-Jørgensen EXPORT_SYMBOL_GPL(ieee80211_calc_tx_airtime);
627db3e1c40SToke Høiland-Jørgensen
ieee80211_calc_expected_tx_airtime(struct ieee80211_hw * hw,struct ieee80211_vif * vif,struct ieee80211_sta * pubsta,int len,bool ampdu)628db3e1c40SToke Høiland-Jørgensen u32 ieee80211_calc_expected_tx_airtime(struct ieee80211_hw *hw,
629db3e1c40SToke Høiland-Jørgensen struct ieee80211_vif *vif,
630db3e1c40SToke Høiland-Jørgensen struct ieee80211_sta *pubsta,
6313ff901cbSFelix Fietkau int len, bool ampdu)
632db3e1c40SToke Høiland-Jørgensen {
633db3e1c40SToke Høiland-Jørgensen struct ieee80211_supported_band *sband;
634db3e1c40SToke Høiland-Jørgensen struct ieee80211_chanctx_conf *conf;
635db3e1c40SToke Høiland-Jørgensen int rateidx, shift = 0;
636db3e1c40SToke Høiland-Jørgensen bool cck, short_pream;
637db3e1c40SToke Høiland-Jørgensen u32 basic_rates;
638db3e1c40SToke Høiland-Jørgensen u8 band = 0;
639db3e1c40SToke Høiland-Jørgensen u16 rate;
640db3e1c40SToke Høiland-Jørgensen
641db3e1c40SToke Høiland-Jørgensen len += 38; /* Ethernet header length */
642db3e1c40SToke Høiland-Jørgensen
643d0a9123eSJohannes Berg conf = rcu_dereference(vif->bss_conf.chanctx_conf);
644db3e1c40SToke Høiland-Jørgensen if (conf) {
645db3e1c40SToke Høiland-Jørgensen band = conf->def.chan->band;
646db3e1c40SToke Høiland-Jørgensen shift = ieee80211_chandef_get_shift(&conf->def);
647db3e1c40SToke Høiland-Jørgensen }
648db3e1c40SToke Høiland-Jørgensen
649db3e1c40SToke Høiland-Jørgensen if (pubsta) {
650db3e1c40SToke Høiland-Jørgensen struct sta_info *sta = container_of(pubsta, struct sta_info,
651db3e1c40SToke Høiland-Jørgensen sta);
652f01cfbafSFelix Fietkau struct ieee80211_rx_status stat;
653046d2e7cSSriram R struct ieee80211_tx_rate *tx_rate = &sta->deflink.tx_stats.last_rate;
654046d2e7cSSriram R struct rate_info *ri = &sta->deflink.tx_stats.last_rate_info;
655f01cfbafSFelix Fietkau u32 duration, overhead;
656f01cfbafSFelix Fietkau u8 agg_shift;
657db3e1c40SToke Høiland-Jørgensen
658ae962e5fSJohannes Berg if (ieee80211_fill_rx_status(&stat, hw, tx_rate, ri, band, len))
659f01cfbafSFelix Fietkau return 0;
6603ff901cbSFelix Fietkau
661f01cfbafSFelix Fietkau if (stat.encoding == RX_ENC_LEGACY || !ampdu)
662f01cfbafSFelix Fietkau return ieee80211_calc_rx_airtime(hw, &stat, len);
663f01cfbafSFelix Fietkau
664f01cfbafSFelix Fietkau duration = ieee80211_get_rate_duration(hw, &stat, &overhead);
6653ff901cbSFelix Fietkau /*
6663ff901cbSFelix Fietkau * Assume that HT/VHT transmission on any AC except VO will
6673ff901cbSFelix Fietkau * use aggregation. Since we don't have reliable reporting
668f01cfbafSFelix Fietkau * of aggregation length, assume an average size based on the
669f01cfbafSFelix Fietkau * tx rate.
6703ff901cbSFelix Fietkau * This will not be very accurate, but much better than simply
671f01cfbafSFelix Fietkau * assuming un-aggregated tx in all cases.
6723ff901cbSFelix Fietkau */
6731f38b8c5SFelix Fietkau if (duration > 400 * 1024) /* <= VHT20 MCS2 1S */
674f01cfbafSFelix Fietkau agg_shift = 1;
6751f38b8c5SFelix Fietkau else if (duration > 250 * 1024) /* <= VHT20 MCS3 1S or MCS1 2S */
676f01cfbafSFelix Fietkau agg_shift = 2;
6771f38b8c5SFelix Fietkau else if (duration > 150 * 1024) /* <= VHT20 MCS5 1S or MCS2 2S */
678f01cfbafSFelix Fietkau agg_shift = 3;
6791f38b8c5SFelix Fietkau else if (duration > 70 * 1024) /* <= VHT20 MCS5 2S */
680f01cfbafSFelix Fietkau agg_shift = 4;
6811f38b8c5SFelix Fietkau else if (stat.encoding != RX_ENC_HE ||
6821f38b8c5SFelix Fietkau duration > 20 * 1024) /* <= HE40 MCS6 2S */
6831f38b8c5SFelix Fietkau agg_shift = 5;
6841f38b8c5SFelix Fietkau else
6851f38b8c5SFelix Fietkau agg_shift = 6;
6863ff901cbSFelix Fietkau
687f01cfbafSFelix Fietkau duration *= len;
688f01cfbafSFelix Fietkau duration /= AVG_PKT_SIZE;
689f01cfbafSFelix Fietkau duration /= 1024;
6901f38b8c5SFelix Fietkau duration += (overhead >> agg_shift);
691f01cfbafSFelix Fietkau
6921f38b8c5SFelix Fietkau return max_t(u32, duration, 4);
693db3e1c40SToke Høiland-Jørgensen }
694db3e1c40SToke Høiland-Jørgensen
695db3e1c40SToke Høiland-Jørgensen if (!conf)
696db3e1c40SToke Høiland-Jørgensen return 0;
697db3e1c40SToke Høiland-Jørgensen
698db3e1c40SToke Høiland-Jørgensen /* No station to get latest rate from, so calculate the worst-case
699db3e1c40SToke Høiland-Jørgensen * duration using the lowest configured basic rate.
700db3e1c40SToke Høiland-Jørgensen */
701db3e1c40SToke Høiland-Jørgensen sband = hw->wiphy->bands[band];
702db3e1c40SToke Høiland-Jørgensen
703db3e1c40SToke Høiland-Jørgensen basic_rates = vif->bss_conf.basic_rates;
704db3e1c40SToke Høiland-Jørgensen short_pream = vif->bss_conf.use_short_preamble;
705db3e1c40SToke Høiland-Jørgensen
706db3e1c40SToke Høiland-Jørgensen rateidx = basic_rates ? ffs(basic_rates) - 1 : 0;
707db3e1c40SToke Høiland-Jørgensen rate = sband->bitrates[rateidx].bitrate << shift;
708db3e1c40SToke Høiland-Jørgensen cck = sband->bitrates[rateidx].flags & IEEE80211_RATE_MANDATORY_B;
709db3e1c40SToke Høiland-Jørgensen
710db3e1c40SToke Høiland-Jørgensen return ieee80211_calc_legacy_rate_duration(rate, short_pream, cck, len);
711db3e1c40SToke Høiland-Jørgensen }
712