2 * Copyright (C) 2010 Felix Fietkau <nbd@openwrt.org>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License version 2 as
6 * published by the Free Software Foundation.
8 #include <linux/netdevice.h>
9 #include <linux/types.h>
10 #include <linux/skbuff.h>
11 #include <linux/debugfs.h>
12 #include <linux/random.h>
13 #include <linux/ieee80211.h>
14 #include <net/mac80211.h>
16 #include "rc80211_minstrel.h"
17 #include "rc80211_minstrel_ht.h"
19 #define AVG_PKT_SIZE 1200
20 #define SAMPLE_COLUMNS 10
23 /* Number of bits for an average sized packet */
24 #define MCS_NBITS (AVG_PKT_SIZE << 3)
26 /* Number of symbols for a packet with (bps) bits per symbol */
27 #define MCS_NSYMS(bps) ((MCS_NBITS + (bps) - 1) / (bps))
29 /* Transmission time for a packet containing (syms) symbols */
30 #define MCS_SYMBOL_TIME(sgi, syms) \
32 ((syms) * 18 + 4) / 5 : /* syms * 3.6 us */ \
33 (syms) << 2 /* syms * 4 us */ \
36 /* Transmit duration for the raw data part of an average sized packet */
37 #define MCS_DURATION(streams, sgi, bps) MCS_SYMBOL_TIME(sgi, MCS_NSYMS((streams) * (bps)))
40 * Define group sort order: HT40 -> SGI -> #streams
42 #define GROUP_IDX(_streams, _sgi, _ht40) \
43 MINSTREL_MAX_STREAMS * 2 * _ht40 + \
44 MINSTREL_MAX_STREAMS * _sgi + \
47 /* MCS rate information for an MCS group */
48 #define MCS_GROUP(_streams, _sgi, _ht40) \
49 [GROUP_IDX(_streams, _sgi, _ht40)] = { \
50 .streams = _streams, \
52 (_sgi ? IEEE80211_TX_RC_SHORT_GI : 0) | \
53 (_ht40 ? IEEE80211_TX_RC_40_MHZ_WIDTH : 0), \
55 MCS_DURATION(_streams, _sgi, _ht40 ? 54 : 26), \
56 MCS_DURATION(_streams, _sgi, _ht40 ? 108 : 52), \
57 MCS_DURATION(_streams, _sgi, _ht40 ? 162 : 78), \
58 MCS_DURATION(_streams, _sgi, _ht40 ? 216 : 104), \
59 MCS_DURATION(_streams, _sgi, _ht40 ? 324 : 156), \
60 MCS_DURATION(_streams, _sgi, _ht40 ? 432 : 208), \
61 MCS_DURATION(_streams, _sgi, _ht40 ? 486 : 234), \
62 MCS_DURATION(_streams, _sgi, _ht40 ? 540 : 260) \
67 * To enable sufficiently targeted rate sampling, MCS rates are divided into
68 * groups, based on the number of streams and flags (HT40, SGI) that they
71 * Sortorder has to be fixed for GROUP_IDX macro to be applicable:
72 * HT40 -> SGI -> #streams
74 const struct mcs_group minstrel_mcs_groups
[] = {
77 #if MINSTREL_MAX_STREAMS >= 3
83 #if MINSTREL_MAX_STREAMS >= 3
89 #if MINSTREL_MAX_STREAMS >= 3
95 #if MINSTREL_MAX_STREAMS >= 3
100 static u8 sample_table
[SAMPLE_COLUMNS
][MCS_GROUP_RATES
];
103 * Perform EWMA (Exponentially Weighted Moving Average) calculation
106 minstrel_ewma(int old
, int new, int weight
)
108 return (new * (100 - weight
) + old
* weight
) / 100;
112 * Look up an MCS group index based on mac80211 rate information
115 minstrel_ht_get_group_idx(struct ieee80211_tx_rate
*rate
)
117 return GROUP_IDX((rate
->idx
/ MCS_GROUP_RATES
) + 1,
118 !!(rate
->flags
& IEEE80211_TX_RC_SHORT_GI
),
119 !!(rate
->flags
& IEEE80211_TX_RC_40_MHZ_WIDTH
));
122 static inline struct minstrel_rate_stats
*
123 minstrel_get_ratestats(struct minstrel_ht_sta
*mi
, int index
)
125 return &mi
->groups
[index
/ MCS_GROUP_RATES
].rates
[index
% MCS_GROUP_RATES
];
130 * Recalculate success probabilities and counters for a rate using EWMA
133 minstrel_calc_rate_ewma(struct minstrel_rate_stats
*mr
)
135 if (unlikely(mr
->attempts
> 0)) {
136 mr
->sample_skipped
= 0;
137 mr
->cur_prob
= MINSTREL_FRAC(mr
->success
, mr
->attempts
);
139 mr
->probability
= mr
->cur_prob
;
141 mr
->probability
= minstrel_ewma(mr
->probability
,
142 mr
->cur_prob
, EWMA_LEVEL
);
143 mr
->att_hist
+= mr
->attempts
;
144 mr
->succ_hist
+= mr
->success
;
146 mr
->sample_skipped
++;
148 mr
->last_success
= mr
->success
;
149 mr
->last_attempts
= mr
->attempts
;
155 * Calculate throughput based on the average A-MPDU length, taking into account
156 * the expected number of retransmissions and their expected length
159 minstrel_ht_calc_tp(struct minstrel_ht_sta
*mi
, int group
, int rate
)
161 struct minstrel_rate_stats
*mr
;
164 mr
= &mi
->groups
[group
].rates
[rate
];
166 if (mr
->probability
< MINSTREL_FRAC(1, 10)) {
171 usecs
= mi
->overhead
/ MINSTREL_TRUNC(mi
->avg_ampdu_len
);
172 usecs
+= minstrel_mcs_groups
[group
].duration
[rate
];
173 mr
->cur_tp
= MINSTREL_TRUNC((1000000 / usecs
) * mr
->probability
);
177 * Update rate statistics and select new primary rates
179 * Rules for rate selection:
180 * - max_prob_rate must use only one stream, as a tradeoff between delivery
181 * probability and throughput during strong fluctuations
182 * - as long as the max prob rate has a probability of more than 3/4, pick
183 * higher throughput rates, even if the probablity is a bit lower
186 minstrel_ht_update_stats(struct minstrel_priv
*mp
, struct minstrel_ht_sta
*mi
)
188 struct minstrel_mcs_group_data
*mg
;
189 struct minstrel_rate_stats
*mr
;
190 int cur_prob
, cur_prob_tp
, cur_tp
, cur_tp2
;
193 if (mi
->ampdu_packets
> 0) {
194 mi
->avg_ampdu_len
= minstrel_ewma(mi
->avg_ampdu_len
,
195 MINSTREL_FRAC(mi
->ampdu_len
, mi
->ampdu_packets
), EWMA_LEVEL
);
197 mi
->ampdu_packets
= 0;
201 mi
->sample_count
= 0;
203 mi
->max_tp_rate2
= 0;
204 mi
->max_prob_rate
= 0;
206 for (group
= 0; group
< ARRAY_SIZE(minstrel_mcs_groups
); group
++) {
212 mg
= &mi
->groups
[group
];
217 mg
->max_tp_rate2
= 0;
218 mg
->max_prob_rate
= 0;
221 for (i
= 0; i
< MCS_GROUP_RATES
; i
++) {
222 if (!(mg
->supported
& BIT(i
)))
226 mr
->retry_updated
= false;
227 index
= MCS_GROUP_RATES
* group
+ i
;
228 minstrel_calc_rate_ewma(mr
);
229 minstrel_ht_calc_tp(mi
, group
, i
);
234 /* ignore the lowest rate of each single-stream group */
235 if (!i
&& minstrel_mcs_groups
[group
].streams
== 1)
238 if ((mr
->cur_tp
> cur_prob_tp
&& mr
->probability
>
239 MINSTREL_FRAC(3, 4)) || mr
->probability
> cur_prob
) {
240 mg
->max_prob_rate
= index
;
241 cur_prob
= mr
->probability
;
242 cur_prob_tp
= mr
->cur_tp
;
245 if (mr
->cur_tp
> cur_tp
) {
246 swap(index
, mg
->max_tp_rate
);
248 mr
= minstrel_get_ratestats(mi
, index
);
251 if (index
>= mg
->max_tp_rate
)
254 if (mr
->cur_tp
> cur_tp2
) {
255 mg
->max_tp_rate2
= index
;
256 cur_tp2
= mr
->cur_tp
;
261 /* try to sample up to half of the available rates during each interval */
262 mi
->sample_count
*= 4;
268 for (group
= 0; group
< ARRAY_SIZE(minstrel_mcs_groups
); group
++) {
269 mg
= &mi
->groups
[group
];
273 mr
= minstrel_get_ratestats(mi
, mg
->max_prob_rate
);
274 if (cur_prob_tp
< mr
->cur_tp
&&
275 minstrel_mcs_groups
[group
].streams
== 1) {
276 mi
->max_prob_rate
= mg
->max_prob_rate
;
277 cur_prob
= mr
->cur_prob
;
278 cur_prob_tp
= mr
->cur_tp
;
281 mr
= minstrel_get_ratestats(mi
, mg
->max_tp_rate
);
282 if (cur_tp
< mr
->cur_tp
) {
283 mi
->max_tp_rate2
= mi
->max_tp_rate
;
285 mi
->max_tp_rate
= mg
->max_tp_rate
;
289 mr
= minstrel_get_ratestats(mi
, mg
->max_tp_rate2
);
290 if (cur_tp2
< mr
->cur_tp
) {
291 mi
->max_tp_rate2
= mg
->max_tp_rate2
;
292 cur_tp2
= mr
->cur_tp
;
296 mi
->stats_update
= jiffies
;
300 minstrel_ht_txstat_valid(struct ieee80211_tx_rate
*rate
)
308 return !!(rate
->flags
& IEEE80211_TX_RC_MCS
);
312 minstrel_next_sample_idx(struct minstrel_ht_sta
*mi
)
314 struct minstrel_mcs_group_data
*mg
;
318 mi
->sample_group
%= ARRAY_SIZE(minstrel_mcs_groups
);
319 mg
= &mi
->groups
[mi
->sample_group
];
324 if (++mg
->index
>= MCS_GROUP_RATES
) {
326 if (++mg
->column
>= ARRAY_SIZE(sample_table
))
334 minstrel_downgrade_rate(struct minstrel_ht_sta
*mi
, unsigned int *idx
,
337 int group
, orig_group
;
339 orig_group
= group
= *idx
/ MCS_GROUP_RATES
;
343 if (!mi
->groups
[group
].supported
)
346 if (minstrel_mcs_groups
[group
].streams
>
347 minstrel_mcs_groups
[orig_group
].streams
)
351 *idx
= mi
->groups
[group
].max_tp_rate
;
353 *idx
= mi
->groups
[group
].max_tp_rate2
;
359 minstrel_aggr_check(struct ieee80211_sta
*pubsta
, struct sk_buff
*skb
)
361 struct ieee80211_hdr
*hdr
= (struct ieee80211_hdr
*) skb
->data
;
362 struct sta_info
*sta
= container_of(pubsta
, struct sta_info
, sta
);
365 if (unlikely(!ieee80211_is_data_qos(hdr
->frame_control
)))
368 if (unlikely(skb
->protocol
== cpu_to_be16(ETH_P_PAE
)))
371 tid
= *ieee80211_get_qos_ctl(hdr
) & IEEE80211_QOS_CTL_TID_MASK
;
372 if (likely(sta
->ampdu_mlme
.tid_tx
[tid
]))
375 if (skb_get_queue_mapping(skb
) == IEEE80211_AC_VO
)
378 ieee80211_start_tx_ba_session(pubsta
, tid
, 5000);
382 minstrel_ht_tx_status(void *priv
, struct ieee80211_supported_band
*sband
,
383 struct ieee80211_sta
*sta
, void *priv_sta
,
386 struct minstrel_ht_sta_priv
*msp
= priv_sta
;
387 struct minstrel_ht_sta
*mi
= &msp
->ht
;
388 struct ieee80211_tx_info
*info
= IEEE80211_SKB_CB(skb
);
389 struct ieee80211_tx_rate
*ar
= info
->status
.rates
;
390 struct minstrel_rate_stats
*rate
, *rate2
;
391 struct minstrel_priv
*mp
= priv
;
397 return mac80211_minstrel
.tx_status(priv
, sband
, sta
, &msp
->legacy
, skb
);
399 /* This packet was aggregated but doesn't carry status info */
400 if ((info
->flags
& IEEE80211_TX_CTL_AMPDU
) &&
401 !(info
->flags
& IEEE80211_TX_STAT_AMPDU
))
404 if (!(info
->flags
& IEEE80211_TX_STAT_AMPDU
)) {
405 info
->status
.ampdu_ack_len
=
406 (info
->flags
& IEEE80211_TX_STAT_ACK
? 1 : 0);
407 info
->status
.ampdu_len
= 1;
411 mi
->ampdu_len
+= info
->status
.ampdu_len
;
413 if (!mi
->sample_wait
&& !mi
->sample_tries
&& mi
->sample_count
> 0) {
414 mi
->sample_wait
= 16 + 2 * MINSTREL_TRUNC(mi
->avg_ampdu_len
);
415 mi
->sample_tries
= 2;
419 if (info
->flags
& IEEE80211_TX_CTL_RATE_CTRL_PROBE
)
420 mi
->sample_packets
+= info
->status
.ampdu_len
;
422 for (i
= 0; !last
; i
++) {
423 last
= (i
== IEEE80211_TX_MAX_RATES
- 1) ||
424 !minstrel_ht_txstat_valid(&ar
[i
+ 1]);
426 if (!minstrel_ht_txstat_valid(&ar
[i
]))
429 group
= minstrel_ht_get_group_idx(&ar
[i
]);
430 rate
= &mi
->groups
[group
].rates
[ar
[i
].idx
% 8];
433 rate
->success
+= info
->status
.ampdu_ack_len
;
435 rate
->attempts
+= ar
[i
].count
* info
->status
.ampdu_len
;
439 * check for sudden death of spatial multiplexing,
440 * downgrade to a lower number of streams if necessary.
442 rate
= minstrel_get_ratestats(mi
, mi
->max_tp_rate
);
443 if (rate
->attempts
> 30 &&
444 MINSTREL_FRAC(rate
->success
, rate
->attempts
) <
445 MINSTREL_FRAC(20, 100))
446 minstrel_downgrade_rate(mi
, &mi
->max_tp_rate
, true);
448 rate2
= minstrel_get_ratestats(mi
, mi
->max_tp_rate2
);
449 if (rate2
->attempts
> 30 &&
450 MINSTREL_FRAC(rate2
->success
, rate2
->attempts
) <
451 MINSTREL_FRAC(20, 100))
452 minstrel_downgrade_rate(mi
, &mi
->max_tp_rate2
, false);
454 if (time_after(jiffies
, mi
->stats_update
+ (mp
->update_interval
/ 2 * HZ
) / 1000)) {
455 minstrel_ht_update_stats(mp
, mi
);
456 if (!(info
->flags
& IEEE80211_TX_CTL_AMPDU
))
457 minstrel_aggr_check(sta
, skb
);
462 minstrel_calc_retransmit(struct minstrel_priv
*mp
, struct minstrel_ht_sta
*mi
,
465 struct minstrel_rate_stats
*mr
;
466 const struct mcs_group
*group
;
467 unsigned int tx_time
, tx_time_rtscts
, tx_time_data
;
468 unsigned int cw
= mp
->cw_min
;
469 unsigned int ctime
= 0;
470 unsigned int t_slot
= 9; /* FIXME */
471 unsigned int ampdu_len
= MINSTREL_TRUNC(mi
->avg_ampdu_len
);
473 mr
= minstrel_get_ratestats(mi
, index
);
474 if (mr
->probability
< MINSTREL_FRAC(1, 10)) {
476 mr
->retry_count_rtscts
= 1;
481 mr
->retry_count_rtscts
= 2;
482 mr
->retry_updated
= true;
484 group
= &minstrel_mcs_groups
[index
/ MCS_GROUP_RATES
];
485 tx_time_data
= group
->duration
[index
% MCS_GROUP_RATES
] * ampdu_len
;
487 /* Contention time for first 2 tries */
488 ctime
= (t_slot
* cw
) >> 1;
489 cw
= min((cw
<< 1) | 1, mp
->cw_max
);
490 ctime
+= (t_slot
* cw
) >> 1;
491 cw
= min((cw
<< 1) | 1, mp
->cw_max
);
493 /* Total TX time for data and Contention after first 2 tries */
494 tx_time
= ctime
+ 2 * (mi
->overhead
+ tx_time_data
);
495 tx_time_rtscts
= ctime
+ 2 * (mi
->overhead_rtscts
+ tx_time_data
);
497 /* See how many more tries we can fit inside segment size */
499 /* Contention time for this try */
500 ctime
= (t_slot
* cw
) >> 1;
501 cw
= min((cw
<< 1) | 1, mp
->cw_max
);
503 /* Total TX time after this try */
504 tx_time
+= ctime
+ mi
->overhead
+ tx_time_data
;
505 tx_time_rtscts
+= ctime
+ mi
->overhead_rtscts
+ tx_time_data
;
507 if (tx_time_rtscts
< mp
->segment_size
)
508 mr
->retry_count_rtscts
++;
509 } while ((tx_time
< mp
->segment_size
) &&
510 (++mr
->retry_count
< mp
->max_retry
));
515 minstrel_ht_set_rate(struct minstrel_priv
*mp
, struct minstrel_ht_sta
*mi
,
516 struct ieee80211_tx_rate
*rate
, int index
,
517 bool sample
, bool rtscts
)
519 const struct mcs_group
*group
= &minstrel_mcs_groups
[index
/ MCS_GROUP_RATES
];
520 struct minstrel_rate_stats
*mr
;
522 mr
= minstrel_get_ratestats(mi
, index
);
523 if (!mr
->retry_updated
)
524 minstrel_calc_retransmit(mp
, mi
, index
);
528 else if (mr
->probability
< MINSTREL_FRAC(20, 100))
531 rate
->count
= mr
->retry_count_rtscts
;
533 rate
->count
= mr
->retry_count
;
535 rate
->flags
= IEEE80211_TX_RC_MCS
| group
->flags
;
537 rate
->flags
|= IEEE80211_TX_RC_USE_RTS_CTS
;
538 rate
->idx
= index
% MCS_GROUP_RATES
+ (group
->streams
- 1) * MCS_GROUP_RATES
;
542 minstrel_get_duration(int index
)
544 const struct mcs_group
*group
= &minstrel_mcs_groups
[index
/ MCS_GROUP_RATES
];
545 return group
->duration
[index
% MCS_GROUP_RATES
];
549 minstrel_get_sample_rate(struct minstrel_priv
*mp
, struct minstrel_ht_sta
*mi
)
551 struct minstrel_rate_stats
*mr
;
552 struct minstrel_mcs_group_data
*mg
;
555 if (mi
->sample_wait
> 0) {
560 if (!mi
->sample_tries
)
564 mg
= &mi
->groups
[mi
->sample_group
];
565 sample_idx
= sample_table
[mg
->column
][mg
->index
];
566 mr
= &mg
->rates
[sample_idx
];
567 sample_idx
+= mi
->sample_group
* MCS_GROUP_RATES
;
568 minstrel_next_sample_idx(mi
);
571 * Sampling might add some overhead (RTS, no aggregation)
572 * to the frame. Hence, don't use sampling for the currently
575 if (sample_idx
== mi
->max_tp_rate
)
578 * When not using MRR, do not sample if the probability is already
579 * higher than 95% to avoid wasting airtime
581 if (!mp
->has_mrr
&& (mr
->probability
> MINSTREL_FRAC(95, 100)))
585 * Make sure that lower rates get sampled only occasionally,
586 * if the link is working perfectly.
588 if (minstrel_get_duration(sample_idx
) >
589 minstrel_get_duration(mi
->max_tp_rate
)) {
590 if (mr
->sample_skipped
< 20)
593 if (mi
->sample_slow
++ > 2)
601 minstrel_ht_get_rate(void *priv
, struct ieee80211_sta
*sta
, void *priv_sta
,
602 struct ieee80211_tx_rate_control
*txrc
)
604 struct ieee80211_tx_info
*info
= IEEE80211_SKB_CB(txrc
->skb
);
605 struct ieee80211_tx_rate
*ar
= info
->status
.rates
;
606 struct minstrel_ht_sta_priv
*msp
= priv_sta
;
607 struct minstrel_ht_sta
*mi
= &msp
->ht
;
608 struct minstrel_priv
*mp
= priv
;
612 if (rate_control_send_low(sta
, priv_sta
, txrc
))
616 return mac80211_minstrel
.get_rate(priv
, sta
, &msp
->legacy
, txrc
);
618 info
->flags
|= mi
->tx_flags
;
620 /* Don't use EAPOL frames for sampling on non-mrr hw */
621 if (mp
->hw
->max_rates
== 1 &&
622 txrc
->skb
->protocol
== cpu_to_be16(ETH_P_PAE
))
625 sample_idx
= minstrel_get_sample_rate(mp
, mi
);
627 #ifdef CONFIG_MAC80211_DEBUGFS
628 /* use fixed index if set */
629 if (mp
->fixed_rate_idx
!= -1)
630 sample_idx
= mp
->fixed_rate_idx
;
633 if (sample_idx
>= 0) {
635 minstrel_ht_set_rate(mp
, mi
, &ar
[0], sample_idx
,
637 info
->flags
|= IEEE80211_TX_CTL_RATE_CTRL_PROBE
;
639 minstrel_ht_set_rate(mp
, mi
, &ar
[0], mi
->max_tp_rate
,
643 if (mp
->hw
->max_rates
>= 3) {
645 * At least 3 tx rates supported, use
646 * sample_rate -> max_tp_rate -> max_prob_rate for sampling and
647 * max_tp_rate -> max_tp_rate2 -> max_prob_rate by default.
650 minstrel_ht_set_rate(mp
, mi
, &ar
[1], mi
->max_tp_rate
,
653 minstrel_ht_set_rate(mp
, mi
, &ar
[1], mi
->max_tp_rate2
,
656 minstrel_ht_set_rate(mp
, mi
, &ar
[2], mi
->max_prob_rate
,
661 } else if (mp
->hw
->max_rates
== 2) {
663 * Only 2 tx rates supported, use
664 * sample_rate -> max_prob_rate for sampling and
665 * max_tp_rate -> max_prob_rate by default.
667 minstrel_ht_set_rate(mp
, mi
, &ar
[1], mi
->max_prob_rate
,
673 /* Not using MRR, only use the first rate */
681 if (mi
->total_packets
== ~0) {
682 mi
->total_packets
= 0;
683 mi
->sample_packets
= 0;
688 minstrel_ht_update_caps(void *priv
, struct ieee80211_supported_band
*sband
,
689 struct ieee80211_sta
*sta
, void *priv_sta
,
690 enum nl80211_channel_type oper_chan_type
)
692 struct minstrel_priv
*mp
= priv
;
693 struct minstrel_ht_sta_priv
*msp
= priv_sta
;
694 struct minstrel_ht_sta
*mi
= &msp
->ht
;
695 struct ieee80211_mcs_info
*mcs
= &sta
->ht_cap
.mcs
;
696 struct ieee80211_local
*local
= hw_to_local(mp
->hw
);
697 u16 sta_cap
= sta
->ht_cap
.cap
;
704 /* fall back to the old minstrel for legacy stations */
705 if (!sta
->ht_cap
.ht_supported
)
708 BUILD_BUG_ON(ARRAY_SIZE(minstrel_mcs_groups
) !=
709 MINSTREL_MAX_STREAMS
* MINSTREL_STREAM_GROUPS
);
712 memset(mi
, 0, sizeof(*mi
));
713 mi
->stats_update
= jiffies
;
715 ack_dur
= ieee80211_frame_duration(local
, 10, 60, 1, 1);
716 mi
->overhead
= ieee80211_frame_duration(local
, 0, 60, 1, 1) + ack_dur
;
717 mi
->overhead_rtscts
= mi
->overhead
+ 2 * ack_dur
;
719 mi
->avg_ampdu_len
= MINSTREL_FRAC(1, 1);
721 /* When using MRR, sample more on the first attempt, without delay */
723 mi
->sample_count
= 16;
726 mi
->sample_count
= 8;
729 mi
->sample_tries
= 4;
731 stbc
= (sta_cap
& IEEE80211_HT_CAP_RX_STBC
) >>
732 IEEE80211_HT_CAP_RX_STBC_SHIFT
;
733 mi
->tx_flags
|= stbc
<< IEEE80211_TX_CTL_STBC_SHIFT
;
735 if (sta_cap
& IEEE80211_HT_CAP_LDPC_CODING
)
736 mi
->tx_flags
|= IEEE80211_TX_CTL_LDPC
;
738 if (oper_chan_type
!= NL80211_CHAN_HT40MINUS
&&
739 oper_chan_type
!= NL80211_CHAN_HT40PLUS
)
740 sta_cap
&= ~IEEE80211_HT_CAP_SUP_WIDTH_20_40
;
742 smps
= (sta_cap
& IEEE80211_HT_CAP_SM_PS
) >>
743 IEEE80211_HT_CAP_SM_PS_SHIFT
;
745 for (i
= 0; i
< ARRAY_SIZE(mi
->groups
); i
++) {
748 mi
->groups
[i
].supported
= 0;
749 if (minstrel_mcs_groups
[i
].flags
& IEEE80211_TX_RC_SHORT_GI
) {
750 if (minstrel_mcs_groups
[i
].flags
& IEEE80211_TX_RC_40_MHZ_WIDTH
)
751 req
|= IEEE80211_HT_CAP_SGI_40
;
753 req
|= IEEE80211_HT_CAP_SGI_20
;
756 if (minstrel_mcs_groups
[i
].flags
& IEEE80211_TX_RC_40_MHZ_WIDTH
)
757 req
|= IEEE80211_HT_CAP_SUP_WIDTH_20_40
;
759 if ((sta_cap
& req
) != req
)
762 /* Mark MCS > 7 as unsupported if STA is in static SMPS mode */
763 if (smps
== WLAN_HT_CAP_SM_PS_STATIC
&&
764 minstrel_mcs_groups
[i
].streams
> 1)
767 mi
->groups
[i
].supported
=
768 mcs
->rx_mask
[minstrel_mcs_groups
[i
].streams
- 1];
770 if (mi
->groups
[i
].supported
)
781 memset(&msp
->legacy
, 0, sizeof(msp
->legacy
));
782 msp
->legacy
.r
= msp
->ratelist
;
783 msp
->legacy
.sample_table
= msp
->sample_table
;
784 return mac80211_minstrel
.rate_init(priv
, sband
, sta
, &msp
->legacy
);
788 minstrel_ht_rate_init(void *priv
, struct ieee80211_supported_band
*sband
,
789 struct ieee80211_sta
*sta
, void *priv_sta
)
791 struct minstrel_priv
*mp
= priv
;
793 minstrel_ht_update_caps(priv
, sband
, sta
, priv_sta
, mp
->hw
->conf
.channel_type
);
797 minstrel_ht_rate_update(void *priv
, struct ieee80211_supported_band
*sband
,
798 struct ieee80211_sta
*sta
, void *priv_sta
,
799 u32 changed
, enum nl80211_channel_type oper_chan_type
)
801 minstrel_ht_update_caps(priv
, sband
, sta
, priv_sta
, oper_chan_type
);
805 minstrel_ht_alloc_sta(void *priv
, struct ieee80211_sta
*sta
, gfp_t gfp
)
807 struct ieee80211_supported_band
*sband
;
808 struct minstrel_ht_sta_priv
*msp
;
809 struct minstrel_priv
*mp
= priv
;
810 struct ieee80211_hw
*hw
= mp
->hw
;
814 for (i
= 0; i
< IEEE80211_NUM_BANDS
; i
++) {
815 sband
= hw
->wiphy
->bands
[i
];
816 if (sband
&& sband
->n_bitrates
> max_rates
)
817 max_rates
= sband
->n_bitrates
;
820 msp
= kzalloc(sizeof(struct minstrel_ht_sta
), gfp
);
824 msp
->ratelist
= kzalloc(sizeof(struct minstrel_rate
) * max_rates
, gfp
);
828 msp
->sample_table
= kmalloc(SAMPLE_COLUMNS
* max_rates
, gfp
);
829 if (!msp
->sample_table
)
835 kfree(msp
->ratelist
);
842 minstrel_ht_free_sta(void *priv
, struct ieee80211_sta
*sta
, void *priv_sta
)
844 struct minstrel_ht_sta_priv
*msp
= priv_sta
;
846 kfree(msp
->sample_table
);
847 kfree(msp
->ratelist
);
852 minstrel_ht_alloc(struct ieee80211_hw
*hw
, struct dentry
*debugfsdir
)
854 return mac80211_minstrel
.alloc(hw
, debugfsdir
);
858 minstrel_ht_free(void *priv
)
860 mac80211_minstrel
.free(priv
);
863 static struct rate_control_ops mac80211_minstrel_ht
= {
864 .name
= "minstrel_ht",
865 .tx_status
= minstrel_ht_tx_status
,
866 .get_rate
= minstrel_ht_get_rate
,
867 .rate_init
= minstrel_ht_rate_init
,
868 .rate_update
= minstrel_ht_rate_update
,
869 .alloc_sta
= minstrel_ht_alloc_sta
,
870 .free_sta
= minstrel_ht_free_sta
,
871 .alloc
= minstrel_ht_alloc
,
872 .free
= minstrel_ht_free
,
873 #ifdef CONFIG_MAC80211_DEBUGFS
874 .add_sta_debugfs
= minstrel_ht_add_sta_debugfs
,
875 .remove_sta_debugfs
= minstrel_ht_remove_sta_debugfs
,
881 init_sample_table(void)
884 u8 rnd
[MCS_GROUP_RATES
];
886 memset(sample_table
, 0xff, sizeof(sample_table
));
887 for (col
= 0; col
< SAMPLE_COLUMNS
; col
++) {
888 for (i
= 0; i
< MCS_GROUP_RATES
; i
++) {
889 get_random_bytes(rnd
, sizeof(rnd
));
890 new_idx
= (i
+ rnd
[i
]) % MCS_GROUP_RATES
;
892 while (sample_table
[col
][new_idx
] != 0xff)
893 new_idx
= (new_idx
+ 1) % MCS_GROUP_RATES
;
895 sample_table
[col
][new_idx
] = i
;
901 rc80211_minstrel_ht_init(void)
904 return ieee80211_rate_control_register(&mac80211_minstrel_ht
);
908 rc80211_minstrel_ht_exit(void)
910 ieee80211_rate_control_unregister(&mac80211_minstrel_ht
);