linux-stable/net/mac80211/airtime.c
Lorenzo Bianconi 3e8f7abcc3 wifi: mac8021: fix possible oob access in ieee80211_get_rate_duration
Fix possible out-of-bound access in ieee80211_get_rate_duration routine
as reported by the following UBSAN report:

UBSAN: array-index-out-of-bounds in net/mac80211/airtime.c:455:47
index 15 is out of range for type 'u16 [12]'
CPU: 2 PID: 217 Comm: kworker/u32:10 Not tainted 6.1.0-060100rc3-generic
Hardware name: Acer Aspire TC-281/Aspire TC-281, BIOS R01-A2 07/18/2017
Workqueue: mt76 mt76u_tx_status_data [mt76_usb]
Call Trace:
 <TASK>
 show_stack+0x4e/0x61
 dump_stack_lvl+0x4a/0x6f
 dump_stack+0x10/0x18
 ubsan_epilogue+0x9/0x43
 __ubsan_handle_out_of_bounds.cold+0x42/0x47
ieee80211_get_rate_duration.constprop.0+0x22f/0x2a0 [mac80211]
 ? ieee80211_tx_status_ext+0x32e/0x640 [mac80211]
 ieee80211_calc_rx_airtime+0xda/0x120 [mac80211]
 ieee80211_calc_tx_airtime+0xb4/0x100 [mac80211]
 mt76x02_send_tx_status+0x266/0x480 [mt76x02_lib]
 mt76x02_tx_status_data+0x52/0x80 [mt76x02_lib]
 mt76u_tx_status_data+0x67/0xd0 [mt76_usb]
 process_one_work+0x225/0x400
 worker_thread+0x50/0x3e0
 ? process_one_work+0x400/0x400
 kthread+0xe9/0x110
 ? kthread_complete_and_exit+0x20/0x20
 ret_from_fork+0x22/0x30

Fixes: db3e1c40cf ("mac80211: Import airtime calculation code from mt76")
Signed-off-by: Lorenzo Bianconi <lorenzo@kernel.org>
Acked-by: Toke Høiland-Jørgensen <toke@toke.dk>
Signed-off-by: Johannes Berg <johannes.berg@intel.com>
2022-11-25 12:45:53 +01:00

711 lines
19 KiB
C

// SPDX-License-Identifier: ISC
/*
* Copyright (C) 2019 Felix Fietkau <nbd@nbd.name>
* Copyright (C) 2021-2022 Intel Corporation
*/
#include <net/mac80211.h>
#include "ieee80211_i.h"
#include "sta_info.h"
#define AVG_PKT_SIZE 1024
/* Number of bits for an average sized packet */
#define MCS_NBITS (AVG_PKT_SIZE << 3)
/* Number of kilo-symbols (symbols * 1024) for a packet with (bps) bits per
* symbol. We use k-symbols to avoid rounding in the _TIME macros below.
*/
#define MCS_N_KSYMS(bps) DIV_ROUND_UP(MCS_NBITS << 10, (bps))
/* Transmission time (in 1024 * usec) for a packet containing (ksyms) * 1024
* symbols.
*/
#define MCS_SYMBOL_TIME(sgi, ksyms) \
(sgi ? \
((ksyms) * 4 * 18) / 20 : /* 3.6 us per sym */ \
((ksyms) * 4) /* 4.0 us per sym */ \
)
/* Transmit duration for the raw data part of an average sized packet */
#define MCS_DURATION(streams, sgi, bps) \
((u32)MCS_SYMBOL_TIME(sgi, MCS_N_KSYMS((streams) * (bps))))
#define MCS_DURATION_S(shift, streams, sgi, bps) \
((u16)((MCS_DURATION(streams, sgi, bps) >> shift)))
/* These should match the values in enum nl80211_he_gi */
#define HE_GI_08 0
#define HE_GI_16 1
#define HE_GI_32 2
/* Transmission time (1024 usec) for a packet containing (ksyms) * k-symbols */
#define HE_SYMBOL_TIME(gi, ksyms) \
(gi == HE_GI_08 ? \
((ksyms) * 16 * 17) / 20 : /* 13.6 us per sym */ \
(gi == HE_GI_16 ? \
((ksyms) * 16 * 18) / 20 : /* 14.4 us per sym */ \
((ksyms) * 16) /* 16.0 us per sym */ \
))
/* Transmit duration for the raw data part of an average sized packet */
#define HE_DURATION(streams, gi, bps) \
((u32)HE_SYMBOL_TIME(gi, MCS_N_KSYMS((streams) * (bps))))
#define HE_DURATION_S(shift, streams, gi, bps) \
(HE_DURATION(streams, gi, bps) >> shift)
#define BW_20 0
#define BW_40 1
#define BW_80 2
#define BW_160 3
/*
* Define group sort order: HT40 -> SGI -> #streams
*/
#define IEEE80211_MAX_STREAMS 4
#define IEEE80211_HT_STREAM_GROUPS 4 /* BW(=2) * SGI(=2) */
#define IEEE80211_VHT_STREAM_GROUPS 8 /* BW(=4) * SGI(=2) */
#define IEEE80211_HE_MAX_STREAMS 8
#define IEEE80211_HT_GROUPS_NB (IEEE80211_MAX_STREAMS * \
IEEE80211_HT_STREAM_GROUPS)
#define IEEE80211_VHT_GROUPS_NB (IEEE80211_MAX_STREAMS * \
IEEE80211_VHT_STREAM_GROUPS)
#define IEEE80211_HT_GROUP_0 0
#define IEEE80211_VHT_GROUP_0 (IEEE80211_HT_GROUP_0 + IEEE80211_HT_GROUPS_NB)
#define IEEE80211_HE_GROUP_0 (IEEE80211_VHT_GROUP_0 + IEEE80211_VHT_GROUPS_NB)
#define MCS_GROUP_RATES 12
#define HT_GROUP_IDX(_streams, _sgi, _ht40) \
IEEE80211_HT_GROUP_0 + \
IEEE80211_MAX_STREAMS * 2 * _ht40 + \
IEEE80211_MAX_STREAMS * _sgi + \
_streams - 1
#define _MAX(a, b) (((a)>(b))?(a):(b))
#define GROUP_SHIFT(duration) \
_MAX(0, 16 - __builtin_clz(duration))
/* MCS rate information for an MCS group */
#define __MCS_GROUP(_streams, _sgi, _ht40, _s) \
[HT_GROUP_IDX(_streams, _sgi, _ht40)] = { \
.shift = _s, \
.duration = { \
MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 54 : 26), \
MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 108 : 52), \
MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 162 : 78), \
MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 216 : 104), \
MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 324 : 156), \
MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 432 : 208), \
MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 486 : 234), \
MCS_DURATION_S(_s, _streams, _sgi, _ht40 ? 540 : 260) \
} \
}
#define MCS_GROUP_SHIFT(_streams, _sgi, _ht40) \
GROUP_SHIFT(MCS_DURATION(_streams, _sgi, _ht40 ? 54 : 26))
#define MCS_GROUP(_streams, _sgi, _ht40) \
__MCS_GROUP(_streams, _sgi, _ht40, \
MCS_GROUP_SHIFT(_streams, _sgi, _ht40))
#define VHT_GROUP_IDX(_streams, _sgi, _bw) \
(IEEE80211_VHT_GROUP_0 + \
IEEE80211_MAX_STREAMS * 2 * (_bw) + \
IEEE80211_MAX_STREAMS * (_sgi) + \
(_streams) - 1)
#define BW2VBPS(_bw, r4, r3, r2, r1) \
(_bw == BW_160 ? r4 : _bw == BW_80 ? r3 : _bw == BW_40 ? r2 : r1)
#define __VHT_GROUP(_streams, _sgi, _bw, _s) \
[VHT_GROUP_IDX(_streams, _sgi, _bw)] = { \
.shift = _s, \
.duration = { \
MCS_DURATION_S(_s, _streams, _sgi, \
BW2VBPS(_bw, 234, 117, 54, 26)), \
MCS_DURATION_S(_s, _streams, _sgi, \
BW2VBPS(_bw, 468, 234, 108, 52)), \
MCS_DURATION_S(_s, _streams, _sgi, \
BW2VBPS(_bw, 702, 351, 162, 78)), \
MCS_DURATION_S(_s, _streams, _sgi, \
BW2VBPS(_bw, 936, 468, 216, 104)), \
MCS_DURATION_S(_s, _streams, _sgi, \
BW2VBPS(_bw, 1404, 702, 324, 156)), \
MCS_DURATION_S(_s, _streams, _sgi, \
BW2VBPS(_bw, 1872, 936, 432, 208)), \
MCS_DURATION_S(_s, _streams, _sgi, \
BW2VBPS(_bw, 2106, 1053, 486, 234)), \
MCS_DURATION_S(_s, _streams, _sgi, \
BW2VBPS(_bw, 2340, 1170, 540, 260)), \
MCS_DURATION_S(_s, _streams, _sgi, \
BW2VBPS(_bw, 2808, 1404, 648, 312)), \
MCS_DURATION_S(_s, _streams, _sgi, \
BW2VBPS(_bw, 3120, 1560, 720, 346)) \
} \
}
#define VHT_GROUP_SHIFT(_streams, _sgi, _bw) \
GROUP_SHIFT(MCS_DURATION(_streams, _sgi, \
BW2VBPS(_bw, 243, 117, 54, 26)))
#define VHT_GROUP(_streams, _sgi, _bw) \
__VHT_GROUP(_streams, _sgi, _bw, \
VHT_GROUP_SHIFT(_streams, _sgi, _bw))
#define HE_GROUP_IDX(_streams, _gi, _bw) \
(IEEE80211_HE_GROUP_0 + \
IEEE80211_HE_MAX_STREAMS * 3 * (_bw) + \
IEEE80211_HE_MAX_STREAMS * (_gi) + \
(_streams) - 1)
#define __HE_GROUP(_streams, _gi, _bw, _s) \
[HE_GROUP_IDX(_streams, _gi, _bw)] = { \
.shift = _s, \
.duration = { \
HE_DURATION_S(_s, _streams, _gi, \
BW2VBPS(_bw, 979, 489, 230, 115)), \
HE_DURATION_S(_s, _streams, _gi, \
BW2VBPS(_bw, 1958, 979, 475, 230)), \
HE_DURATION_S(_s, _streams, _gi, \
BW2VBPS(_bw, 2937, 1468, 705, 345)), \
HE_DURATION_S(_s, _streams, _gi, \
BW2VBPS(_bw, 3916, 1958, 936, 475)), \
HE_DURATION_S(_s, _streams, _gi, \
BW2VBPS(_bw, 5875, 2937, 1411, 705)), \
HE_DURATION_S(_s, _streams, _gi, \
BW2VBPS(_bw, 7833, 3916, 1872, 936)), \
HE_DURATION_S(_s, _streams, _gi, \
BW2VBPS(_bw, 8827, 4406, 2102, 1051)), \
HE_DURATION_S(_s, _streams, _gi, \
BW2VBPS(_bw, 9806, 4896, 2347, 1166)), \
HE_DURATION_S(_s, _streams, _gi, \
BW2VBPS(_bw, 11764, 5875, 2808, 1411)), \
HE_DURATION_S(_s, _streams, _gi, \
BW2VBPS(_bw, 13060, 6523, 3124, 1555)), \
HE_DURATION_S(_s, _streams, _gi, \
BW2VBPS(_bw, 14702, 7344, 3513, 1756)), \
HE_DURATION_S(_s, _streams, _gi, \
BW2VBPS(_bw, 16329, 8164, 3902, 1944)) \
} \
}
#define HE_GROUP_SHIFT(_streams, _gi, _bw) \
GROUP_SHIFT(HE_DURATION(_streams, _gi, \
BW2VBPS(_bw, 979, 489, 230, 115)))
#define HE_GROUP(_streams, _gi, _bw) \
__HE_GROUP(_streams, _gi, _bw, \
HE_GROUP_SHIFT(_streams, _gi, _bw))
struct mcs_group {
u8 shift;
u16 duration[MCS_GROUP_RATES];
};
static const struct mcs_group airtime_mcs_groups[] = {
MCS_GROUP(1, 0, BW_20),
MCS_GROUP(2, 0, BW_20),
MCS_GROUP(3, 0, BW_20),
MCS_GROUP(4, 0, BW_20),
MCS_GROUP(1, 1, BW_20),
MCS_GROUP(2, 1, BW_20),
MCS_GROUP(3, 1, BW_20),
MCS_GROUP(4, 1, BW_20),
MCS_GROUP(1, 0, BW_40),
MCS_GROUP(2, 0, BW_40),
MCS_GROUP(3, 0, BW_40),
MCS_GROUP(4, 0, BW_40),
MCS_GROUP(1, 1, BW_40),
MCS_GROUP(2, 1, BW_40),
MCS_GROUP(3, 1, BW_40),
MCS_GROUP(4, 1, BW_40),
VHT_GROUP(1, 0, BW_20),
VHT_GROUP(2, 0, BW_20),
VHT_GROUP(3, 0, BW_20),
VHT_GROUP(4, 0, BW_20),
VHT_GROUP(1, 1, BW_20),
VHT_GROUP(2, 1, BW_20),
VHT_GROUP(3, 1, BW_20),
VHT_GROUP(4, 1, BW_20),
VHT_GROUP(1, 0, BW_40),
VHT_GROUP(2, 0, BW_40),
VHT_GROUP(3, 0, BW_40),
VHT_GROUP(4, 0, BW_40),
VHT_GROUP(1, 1, BW_40),
VHT_GROUP(2, 1, BW_40),
VHT_GROUP(3, 1, BW_40),
VHT_GROUP(4, 1, BW_40),
VHT_GROUP(1, 0, BW_80),
VHT_GROUP(2, 0, BW_80),
VHT_GROUP(3, 0, BW_80),
VHT_GROUP(4, 0, BW_80),
VHT_GROUP(1, 1, BW_80),
VHT_GROUP(2, 1, BW_80),
VHT_GROUP(3, 1, BW_80),
VHT_GROUP(4, 1, BW_80),
VHT_GROUP(1, 0, BW_160),
VHT_GROUP(2, 0, BW_160),
VHT_GROUP(3, 0, BW_160),
VHT_GROUP(4, 0, BW_160),
VHT_GROUP(1, 1, BW_160),
VHT_GROUP(2, 1, BW_160),
VHT_GROUP(3, 1, BW_160),
VHT_GROUP(4, 1, BW_160),
HE_GROUP(1, HE_GI_08, BW_20),
HE_GROUP(2, HE_GI_08, BW_20),
HE_GROUP(3, HE_GI_08, BW_20),
HE_GROUP(4, HE_GI_08, BW_20),
HE_GROUP(5, HE_GI_08, BW_20),
HE_GROUP(6, HE_GI_08, BW_20),
HE_GROUP(7, HE_GI_08, BW_20),
HE_GROUP(8, HE_GI_08, BW_20),
HE_GROUP(1, HE_GI_16, BW_20),
HE_GROUP(2, HE_GI_16, BW_20),
HE_GROUP(3, HE_GI_16, BW_20),
HE_GROUP(4, HE_GI_16, BW_20),
HE_GROUP(5, HE_GI_16, BW_20),
HE_GROUP(6, HE_GI_16, BW_20),
HE_GROUP(7, HE_GI_16, BW_20),
HE_GROUP(8, HE_GI_16, BW_20),
HE_GROUP(1, HE_GI_32, BW_20),
HE_GROUP(2, HE_GI_32, BW_20),
HE_GROUP(3, HE_GI_32, BW_20),
HE_GROUP(4, HE_GI_32, BW_20),
HE_GROUP(5, HE_GI_32, BW_20),
HE_GROUP(6, HE_GI_32, BW_20),
HE_GROUP(7, HE_GI_32, BW_20),
HE_GROUP(8, HE_GI_32, BW_20),
HE_GROUP(1, HE_GI_08, BW_40),
HE_GROUP(2, HE_GI_08, BW_40),
HE_GROUP(3, HE_GI_08, BW_40),
HE_GROUP(4, HE_GI_08, BW_40),
HE_GROUP(5, HE_GI_08, BW_40),
HE_GROUP(6, HE_GI_08, BW_40),
HE_GROUP(7, HE_GI_08, BW_40),
HE_GROUP(8, HE_GI_08, BW_40),
HE_GROUP(1, HE_GI_16, BW_40),
HE_GROUP(2, HE_GI_16, BW_40),
HE_GROUP(3, HE_GI_16, BW_40),
HE_GROUP(4, HE_GI_16, BW_40),
HE_GROUP(5, HE_GI_16, BW_40),
HE_GROUP(6, HE_GI_16, BW_40),
HE_GROUP(7, HE_GI_16, BW_40),
HE_GROUP(8, HE_GI_16, BW_40),
HE_GROUP(1, HE_GI_32, BW_40),
HE_GROUP(2, HE_GI_32, BW_40),
HE_GROUP(3, HE_GI_32, BW_40),
HE_GROUP(4, HE_GI_32, BW_40),
HE_GROUP(5, HE_GI_32, BW_40),
HE_GROUP(6, HE_GI_32, BW_40),
HE_GROUP(7, HE_GI_32, BW_40),
HE_GROUP(8, HE_GI_32, BW_40),
HE_GROUP(1, HE_GI_08, BW_80),
HE_GROUP(2, HE_GI_08, BW_80),
HE_GROUP(3, HE_GI_08, BW_80),
HE_GROUP(4, HE_GI_08, BW_80),
HE_GROUP(5, HE_GI_08, BW_80),
HE_GROUP(6, HE_GI_08, BW_80),
HE_GROUP(7, HE_GI_08, BW_80),
HE_GROUP(8, HE_GI_08, BW_80),
HE_GROUP(1, HE_GI_16, BW_80),
HE_GROUP(2, HE_GI_16, BW_80),
HE_GROUP(3, HE_GI_16, BW_80),
HE_GROUP(4, HE_GI_16, BW_80),
HE_GROUP(5, HE_GI_16, BW_80),
HE_GROUP(6, HE_GI_16, BW_80),
HE_GROUP(7, HE_GI_16, BW_80),
HE_GROUP(8, HE_GI_16, BW_80),
HE_GROUP(1, HE_GI_32, BW_80),
HE_GROUP(2, HE_GI_32, BW_80),
HE_GROUP(3, HE_GI_32, BW_80),
HE_GROUP(4, HE_GI_32, BW_80),
HE_GROUP(5, HE_GI_32, BW_80),
HE_GROUP(6, HE_GI_32, BW_80),
HE_GROUP(7, HE_GI_32, BW_80),
HE_GROUP(8, HE_GI_32, BW_80),
HE_GROUP(1, HE_GI_08, BW_160),
HE_GROUP(2, HE_GI_08, BW_160),
HE_GROUP(3, HE_GI_08, BW_160),
HE_GROUP(4, HE_GI_08, BW_160),
HE_GROUP(5, HE_GI_08, BW_160),
HE_GROUP(6, HE_GI_08, BW_160),
HE_GROUP(7, HE_GI_08, BW_160),
HE_GROUP(8, HE_GI_08, BW_160),
HE_GROUP(1, HE_GI_16, BW_160),
HE_GROUP(2, HE_GI_16, BW_160),
HE_GROUP(3, HE_GI_16, BW_160),
HE_GROUP(4, HE_GI_16, BW_160),
HE_GROUP(5, HE_GI_16, BW_160),
HE_GROUP(6, HE_GI_16, BW_160),
HE_GROUP(7, HE_GI_16, BW_160),
HE_GROUP(8, HE_GI_16, BW_160),
HE_GROUP(1, HE_GI_32, BW_160),
HE_GROUP(2, HE_GI_32, BW_160),
HE_GROUP(3, HE_GI_32, BW_160),
HE_GROUP(4, HE_GI_32, BW_160),
HE_GROUP(5, HE_GI_32, BW_160),
HE_GROUP(6, HE_GI_32, BW_160),
HE_GROUP(7, HE_GI_32, BW_160),
HE_GROUP(8, HE_GI_32, BW_160),
};
static u32
ieee80211_calc_legacy_rate_duration(u16 bitrate, bool short_pre,
bool cck, int len)
{
u32 duration;
if (cck) {
duration = 144 + 48; /* preamble + PLCP */
if (short_pre)
duration >>= 1;
duration += 10; /* SIFS */
} else {
duration = 20 + 16; /* premable + SIFS */
}
len <<= 3;
duration += (len * 10) / bitrate;
return duration;
}
static u32 ieee80211_get_rate_duration(struct ieee80211_hw *hw,
struct ieee80211_rx_status *status,
u32 *overhead)
{
bool sgi = status->enc_flags & RX_ENC_FLAG_SHORT_GI;
int bw, streams;
int group, idx;
u32 duration;
switch (status->bw) {
case RATE_INFO_BW_20:
bw = BW_20;
break;
case RATE_INFO_BW_40:
bw = BW_40;
break;
case RATE_INFO_BW_80:
bw = BW_80;
break;
case RATE_INFO_BW_160:
bw = BW_160;
break;
default:
WARN_ON_ONCE(1);
return 0;
}
switch (status->encoding) {
case RX_ENC_VHT:
streams = status->nss;
idx = status->rate_idx;
group = VHT_GROUP_IDX(streams, sgi, bw);
break;
case RX_ENC_HT:
streams = ((status->rate_idx >> 3) & 3) + 1;
idx = status->rate_idx & 7;
group = HT_GROUP_IDX(streams, sgi, bw);
break;
case RX_ENC_HE:
streams = status->nss;
idx = status->rate_idx;
group = HE_GROUP_IDX(streams, status->he_gi, bw);
break;
default:
WARN_ON_ONCE(1);
return 0;
}
if (WARN_ON_ONCE((status->encoding != RX_ENC_HE && streams > 4) ||
(status->encoding == RX_ENC_HE && streams > 8)))
return 0;
if (idx >= MCS_GROUP_RATES)
return 0;
duration = airtime_mcs_groups[group].duration[idx];
duration <<= airtime_mcs_groups[group].shift;
*overhead = 36 + (streams << 2);
return duration;
}
u32 ieee80211_calc_rx_airtime(struct ieee80211_hw *hw,
struct ieee80211_rx_status *status,
int len)
{
struct ieee80211_supported_band *sband;
u32 duration, overhead = 0;
if (status->encoding == RX_ENC_LEGACY) {
const struct ieee80211_rate *rate;
bool sp = status->enc_flags & RX_ENC_FLAG_SHORTPRE;
bool cck;
/* on 60GHz or sub-1GHz band, there are no legacy rates */
if (WARN_ON_ONCE(status->band == NL80211_BAND_60GHZ ||
status->band == NL80211_BAND_S1GHZ))
return 0;
sband = hw->wiphy->bands[status->band];
if (!sband || status->rate_idx >= sband->n_bitrates)
return 0;
rate = &sband->bitrates[status->rate_idx];
cck = rate->flags & IEEE80211_RATE_MANDATORY_B;
return ieee80211_calc_legacy_rate_duration(rate->bitrate, sp,
cck, len);
}
duration = ieee80211_get_rate_duration(hw, status, &overhead);
if (!duration)
return 0;
duration *= len;
duration /= AVG_PKT_SIZE;
duration /= 1024;
return duration + overhead;
}
EXPORT_SYMBOL_GPL(ieee80211_calc_rx_airtime);
static bool ieee80211_fill_rate_info(struct ieee80211_hw *hw,
struct ieee80211_rx_status *stat, u8 band,
struct rate_info *ri)
{
struct ieee80211_supported_band *sband = hw->wiphy->bands[band];
int i;
if (!ri || !sband)
return false;
stat->bw = ri->bw;
stat->nss = ri->nss;
stat->rate_idx = ri->mcs;
if (ri->flags & RATE_INFO_FLAGS_HE_MCS)
stat->encoding = RX_ENC_HE;
else if (ri->flags & RATE_INFO_FLAGS_VHT_MCS)
stat->encoding = RX_ENC_VHT;
else if (ri->flags & RATE_INFO_FLAGS_MCS)
stat->encoding = RX_ENC_HT;
else
stat->encoding = RX_ENC_LEGACY;
if (ri->flags & RATE_INFO_FLAGS_SHORT_GI)
stat->enc_flags |= RX_ENC_FLAG_SHORT_GI;
stat->he_gi = ri->he_gi;
if (stat->encoding != RX_ENC_LEGACY)
return true;
stat->rate_idx = 0;
for (i = 0; i < sband->n_bitrates; i++) {
if (ri->legacy != sband->bitrates[i].bitrate)
continue;
stat->rate_idx = i;
return true;
}
return false;
}
static int 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)
{
memset(stat, 0, sizeof(*stat));
stat->band = band;
if (ieee80211_fill_rate_info(hw, stat, band, ri))
return 0;
if (rate->idx < 0 || !rate->count)
return -1;
if (rate->flags & IEEE80211_TX_RC_160_MHZ_WIDTH)
stat->bw = RATE_INFO_BW_160;
else if (rate->flags & IEEE80211_TX_RC_80_MHZ_WIDTH)
stat->bw = RATE_INFO_BW_80;
else if (rate->flags & IEEE80211_TX_RC_40_MHZ_WIDTH)
stat->bw = RATE_INFO_BW_40;
else
stat->bw = RATE_INFO_BW_20;
stat->enc_flags = 0;
if (rate->flags & IEEE80211_TX_RC_USE_SHORT_PREAMBLE)
stat->enc_flags |= RX_ENC_FLAG_SHORTPRE;
if (rate->flags & IEEE80211_TX_RC_SHORT_GI)
stat->enc_flags |= RX_ENC_FLAG_SHORT_GI;
stat->rate_idx = rate->idx;
if (rate->flags & IEEE80211_TX_RC_VHT_MCS) {
stat->encoding = RX_ENC_VHT;
stat->rate_idx = ieee80211_rate_get_vht_mcs(rate);
stat->nss = ieee80211_rate_get_vht_nss(rate);
} else if (rate->flags & IEEE80211_TX_RC_MCS) {
stat->encoding = RX_ENC_HT;
} else {
stat->encoding = RX_ENC_LEGACY;
}
return 0;
}
static u32 ieee80211_calc_tx_airtime_rate(struct ieee80211_hw *hw,
struct ieee80211_tx_rate *rate,
struct rate_info *ri,
u8 band, int len)
{
struct ieee80211_rx_status stat;
if (ieee80211_fill_rx_status(&stat, hw, rate, ri, band, len))
return 0;
return ieee80211_calc_rx_airtime(hw, &stat, len);
}
u32 ieee80211_calc_tx_airtime(struct ieee80211_hw *hw,
struct ieee80211_tx_info *info,
int len)
{
u32 duration = 0;
int i;
for (i = 0; i < ARRAY_SIZE(info->status.rates); i++) {
struct ieee80211_tx_rate *rate = &info->status.rates[i];
u32 cur_duration;
cur_duration = ieee80211_calc_tx_airtime_rate(hw, rate, NULL,
info->band, len);
if (!cur_duration)
break;
duration += cur_duration * rate->count;
}
return duration;
}
EXPORT_SYMBOL_GPL(ieee80211_calc_tx_airtime);
u32 ieee80211_calc_expected_tx_airtime(struct ieee80211_hw *hw,
struct ieee80211_vif *vif,
struct ieee80211_sta *pubsta,
int len, bool ampdu)
{
struct ieee80211_supported_band *sband;
struct ieee80211_chanctx_conf *conf;
int rateidx, shift = 0;
bool cck, short_pream;
u32 basic_rates;
u8 band = 0;
u16 rate;
len += 38; /* Ethernet header length */
conf = rcu_dereference(vif->bss_conf.chanctx_conf);
if (conf) {
band = conf->def.chan->band;
shift = ieee80211_chandef_get_shift(&conf->def);
}
if (pubsta) {
struct sta_info *sta = container_of(pubsta, struct sta_info,
sta);
struct ieee80211_rx_status stat;
struct ieee80211_tx_rate *tx_rate = &sta->deflink.tx_stats.last_rate;
struct rate_info *ri = &sta->deflink.tx_stats.last_rate_info;
u32 duration, overhead;
u8 agg_shift;
if (ieee80211_fill_rx_status(&stat, hw, tx_rate, ri, band, len))
return 0;
if (stat.encoding == RX_ENC_LEGACY || !ampdu)
return ieee80211_calc_rx_airtime(hw, &stat, len);
duration = ieee80211_get_rate_duration(hw, &stat, &overhead);
/*
* Assume that HT/VHT transmission on any AC except VO will
* use aggregation. Since we don't have reliable reporting
* of aggregation length, assume an average size based on the
* tx rate.
* This will not be very accurate, but much better than simply
* assuming un-aggregated tx in all cases.
*/
if (duration > 400 * 1024) /* <= VHT20 MCS2 1S */
agg_shift = 1;
else if (duration > 250 * 1024) /* <= VHT20 MCS3 1S or MCS1 2S */
agg_shift = 2;
else if (duration > 150 * 1024) /* <= VHT20 MCS5 1S or MCS2 2S */
agg_shift = 3;
else if (duration > 70 * 1024) /* <= VHT20 MCS5 2S */
agg_shift = 4;
else if (stat.encoding != RX_ENC_HE ||
duration > 20 * 1024) /* <= HE40 MCS6 2S */
agg_shift = 5;
else
agg_shift = 6;
duration *= len;
duration /= AVG_PKT_SIZE;
duration /= 1024;
duration += (overhead >> agg_shift);
return max_t(u32, duration, 4);
}
if (!conf)
return 0;
/* No station to get latest rate from, so calculate the worst-case
* duration using the lowest configured basic rate.
*/
sband = hw->wiphy->bands[band];
basic_rates = vif->bss_conf.basic_rates;
short_pream = vif->bss_conf.use_short_preamble;
rateidx = basic_rates ? ffs(basic_rates) - 1 : 0;
rate = sband->bitrates[rateidx].bitrate << shift;
cck = sband->bitrates[rateidx].flags & IEEE80211_RATE_MANDATORY_B;
return ieee80211_calc_legacy_rate_duration(rate, short_pream, cck, len);
}