1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26#include "wifi.h"
27#include "base.h"
28#include "rc.h"
29
30
31
32
33
34
35
36
37static u8 _rtl_rc_get_highest_rix(struct rtl_priv *rtlpriv,
38 struct ieee80211_sta *sta,
39 struct sk_buff *skb, bool not_data)
40{
41 struct rtl_hal *rtlhal = rtl_hal(rtlpriv);
42 struct rtl_phy *rtlphy = &(rtlpriv->phy);
43 struct rtl_sta_info *sta_entry = NULL;
44 u16 wireless_mode = 0;
45 u8 nss;
46 struct ieee80211_tx_rate rate;
47
48 switch (get_rf_type(rtlphy)) {
49 case RF_4T4R:
50 nss = 4;
51 break;
52 case RF_3T3R:
53 nss = 3;
54 break;
55 case RF_2T2R:
56 nss = 2;
57 break;
58 default:
59 nss = 1;
60 break;
61 }
62
63
64
65
66
67
68
69
70
71 if (sta) {
72 sta_entry = (struct rtl_sta_info *)sta->drv_priv;
73 wireless_mode = sta_entry->wireless_mode;
74 }
75
76 if (rtl_is_special_data(rtlpriv->mac80211.hw, skb, true, false) ||
77 not_data) {
78 return 0;
79 } else {
80 if (rtlhal->current_bandtype == BAND_ON_2_4G) {
81 if (wireless_mode == WIRELESS_MODE_B) {
82 return B_MODE_MAX_RIX;
83 } else if (wireless_mode == WIRELESS_MODE_G) {
84 return G_MODE_MAX_RIX;
85 } else if (wireless_mode == WIRELESS_MODE_N_24G) {
86 if (nss == 1)
87 return N_MODE_MCS7_RIX;
88 else
89 return N_MODE_MCS15_RIX;
90 } else if (wireless_mode == WIRELESS_MODE_AC_24G) {
91 if (sta->bandwidth == IEEE80211_STA_RX_BW_20) {
92 ieee80211_rate_set_vht(&rate,
93 AC_MODE_MCS8_RIX,
94 nss);
95 goto out;
96 } else {
97 ieee80211_rate_set_vht(&rate,
98 AC_MODE_MCS9_RIX,
99 nss);
100 goto out;
101 }
102 }
103 return 0;
104 } else {
105 if (wireless_mode == WIRELESS_MODE_A) {
106 return A_MODE_MAX_RIX;
107 } else if (wireless_mode == WIRELESS_MODE_N_5G) {
108 if (nss == 1)
109 return N_MODE_MCS7_RIX;
110 else
111 return N_MODE_MCS15_RIX;
112 } else if (wireless_mode == WIRELESS_MODE_AC_5G) {
113 if (sta->bandwidth == IEEE80211_STA_RX_BW_20) {
114 ieee80211_rate_set_vht(&rate,
115 AC_MODE_MCS8_RIX,
116 nss);
117 goto out;
118 } else {
119 ieee80211_rate_set_vht(&rate,
120 AC_MODE_MCS9_RIX,
121 nss);
122 goto out;
123 }
124 }
125 return 0;
126 }
127 }
128
129out:
130 return rate.idx;
131}
132
133static void _rtl_rc_rate_set_series(struct rtl_priv *rtlpriv,
134 struct ieee80211_sta *sta,
135 struct ieee80211_tx_rate *rate,
136 struct ieee80211_tx_rate_control *txrc,
137 u8 tries, s8 rix, int rtsctsenable,
138 bool not_data)
139{
140 struct rtl_mac *mac = rtl_mac(rtlpriv);
141 struct rtl_sta_info *sta_entry = NULL;
142 u16 wireless_mode = 0;
143 u8 sgi_20 = 0, sgi_40 = 0, sgi_80 = 0;
144
145 if (sta) {
146 sgi_20 = sta->ht_cap.cap & IEEE80211_HT_CAP_SGI_20;
147 sgi_40 = sta->ht_cap.cap & IEEE80211_HT_CAP_SGI_40;
148 sgi_80 = sta->vht_cap.cap & IEEE80211_VHT_CAP_SHORT_GI_80;
149 sta_entry = (struct rtl_sta_info *)sta->drv_priv;
150 wireless_mode = sta_entry->wireless_mode;
151 }
152 rate->count = tries;
153 rate->idx = rix >= 0x00 ? rix : 0x00;
154
155 if (!not_data) {
156 if (txrc->short_preamble)
157 rate->flags |= IEEE80211_TX_RC_USE_SHORT_PREAMBLE;
158 if (mac->opmode == NL80211_IFTYPE_AP ||
159 mac->opmode == NL80211_IFTYPE_ADHOC) {
160 if (sta && (sta->ht_cap.cap &
161 IEEE80211_HT_CAP_SUP_WIDTH_20_40))
162 rate->flags |= IEEE80211_TX_RC_40_MHZ_WIDTH;
163 if (sta && sta->vht_cap.vht_supported)
164 rate->flags |= IEEE80211_TX_RC_80_MHZ_WIDTH;
165 } else {
166 if (mac->bw_80)
167 rate->flags |= IEEE80211_TX_RC_80_MHZ_WIDTH;
168 else if (mac->bw_40)
169 rate->flags |= IEEE80211_TX_RC_40_MHZ_WIDTH;
170 }
171
172 if (sgi_20 || sgi_40 || sgi_80)
173 rate->flags |= IEEE80211_TX_RC_SHORT_GI;
174 if (sta && sta->ht_cap.ht_supported &&
175 (wireless_mode == WIRELESS_MODE_N_5G ||
176 wireless_mode == WIRELESS_MODE_N_24G))
177 rate->flags |= IEEE80211_TX_RC_MCS;
178 if (sta && sta->vht_cap.vht_supported &&
179 (wireless_mode == WIRELESS_MODE_AC_5G ||
180 wireless_mode == WIRELESS_MODE_AC_24G ||
181 wireless_mode == WIRELESS_MODE_AC_ONLY))
182 rate->flags |= IEEE80211_TX_RC_VHT_MCS;
183 }
184}
185
186static void rtl_get_rate(void *ppriv, struct ieee80211_sta *sta,
187 void *priv_sta,
188 struct ieee80211_tx_rate_control *txrc)
189{
190 struct rtl_priv *rtlpriv = ppriv;
191 struct sk_buff *skb = txrc->skb;
192 struct ieee80211_tx_info *tx_info = IEEE80211_SKB_CB(skb);
193 struct ieee80211_tx_rate *rates = tx_info->control.rates;
194 __le16 fc = rtl_get_fc(skb);
195 u8 try_per_rate, i, rix;
196 bool not_data = !ieee80211_is_data(fc);
197
198 if (rate_control_send_low(sta, priv_sta, txrc))
199 return;
200
201 rix = _rtl_rc_get_highest_rix(rtlpriv, sta, skb, not_data);
202 try_per_rate = 1;
203 _rtl_rc_rate_set_series(rtlpriv, sta, &rates[0], txrc,
204 try_per_rate, rix, 1, not_data);
205
206 if (!not_data) {
207 for (i = 1; i < 4; i++)
208 _rtl_rc_rate_set_series(rtlpriv, sta, &rates[i],
209 txrc, i, (rix - i), 1,
210 not_data);
211 }
212}
213
214static bool _rtl_tx_aggr_check(struct rtl_priv *rtlpriv,
215 struct rtl_sta_info *sta_entry, u16 tid)
216{
217 struct rtl_mac *mac = rtl_mac(rtlpriv);
218
219 if (mac->act_scanning)
220 return false;
221
222 if (mac->opmode == NL80211_IFTYPE_STATION &&
223 mac->cnt_after_linked < 3)
224 return false;
225
226 if (sta_entry->tids[tid].agg.agg_state == RTL_AGG_STOP)
227 return true;
228
229 return false;
230}
231
232
233static void rtl_tx_status(void *ppriv,
234 struct ieee80211_supported_band *sband,
235 struct ieee80211_sta *sta, void *priv_sta,
236 struct sk_buff *skb)
237{
238 struct rtl_priv *rtlpriv = ppriv;
239 struct rtl_mac *mac = rtl_mac(rtlpriv);
240 struct ieee80211_hdr *hdr = rtl_get_hdr(skb);
241 __le16 fc = rtl_get_fc(skb);
242 struct rtl_sta_info *sta_entry;
243
244 if (!priv_sta || !ieee80211_is_data(fc))
245 return;
246
247 if (rtl_is_special_data(mac->hw, skb, true, true))
248 return;
249
250 if (is_multicast_ether_addr(ieee80211_get_DA(hdr)) ||
251 is_broadcast_ether_addr(ieee80211_get_DA(hdr)))
252 return;
253
254 if (sta) {
255
256 sta_entry = (struct rtl_sta_info *)sta->drv_priv;
257 if (sta->ht_cap.ht_supported &&
258 !(skb->protocol == cpu_to_be16(ETH_P_PAE))) {
259 if (ieee80211_is_data_qos(fc)) {
260 u8 tid = rtl_get_tid(skb);
261 if (_rtl_tx_aggr_check(rtlpriv, sta_entry,
262 tid)) {
263 sta_entry->tids[tid].agg.agg_state =
264 RTL_AGG_PROGRESS;
265 ieee80211_start_tx_ba_session(sta, tid,
266 5000);
267 }
268 }
269 }
270 }
271}
272
273static void rtl_rate_init(void *ppriv,
274 struct ieee80211_supported_band *sband,
275 struct cfg80211_chan_def *chandef,
276 struct ieee80211_sta *sta, void *priv_sta)
277{
278}
279
280static void rtl_rate_update(void *ppriv,
281 struct ieee80211_supported_band *sband,
282 struct cfg80211_chan_def *chandef,
283 struct ieee80211_sta *sta, void *priv_sta,
284 u32 changed)
285{
286}
287
288static void *rtl_rate_alloc(struct ieee80211_hw *hw, struct dentry *debugfsdir)
289{
290 struct rtl_priv *rtlpriv = rtl_priv(hw);
291 return rtlpriv;
292}
293
294static void rtl_rate_free(void *rtlpriv)
295{
296 return;
297}
298
299static void *rtl_rate_alloc_sta(void *ppriv,
300 struct ieee80211_sta *sta, gfp_t gfp)
301{
302 struct rtl_priv *rtlpriv = ppriv;
303 struct rtl_rate_priv *rate_priv;
304
305 rate_priv = kzalloc(sizeof(*rate_priv), gfp);
306 if (!rate_priv)
307 return NULL;
308
309 rtlpriv->rate_priv = rate_priv;
310
311 return rate_priv;
312}
313
314static void rtl_rate_free_sta(void *rtlpriv,
315 struct ieee80211_sta *sta, void *priv_sta)
316{
317 struct rtl_rate_priv *rate_priv = priv_sta;
318 kfree(rate_priv);
319}
320
321static const struct rate_control_ops rtl_rate_ops = {
322 .name = "rtl_rc",
323 .alloc = rtl_rate_alloc,
324 .free = rtl_rate_free,
325 .alloc_sta = rtl_rate_alloc_sta,
326 .free_sta = rtl_rate_free_sta,
327 .rate_init = rtl_rate_init,
328 .rate_update = rtl_rate_update,
329 .tx_status = rtl_tx_status,
330 .get_rate = rtl_get_rate,
331};
332
333int rtl_rate_control_register(void)
334{
335 return ieee80211_rate_control_register(&rtl_rate_ops);
336}
337
338void rtl_rate_control_unregister(void)
339{
340 ieee80211_rate_control_unregister(&rtl_rate_ops);
341}
342
