c1e065f136baa631a9245b87f897dea167865431
[linux-2.6-block.git] / drivers / net / wireless / rtlwifi / rc.c
1 /******************************************************************************
2  *
3  * Copyright(c) 2009-2012  Realtek Corporation.
4  *
5  * This program is free software; you can redistribute it and/or modify it
6  * under the terms of version 2 of the GNU General Public License as
7  * published by the Free Software Foundation.
8  *
9  * This program is distributed in the hope that it will be useful, but WITHOUT
10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
12  * more details.
13  *
14  * You should have received a copy of the GNU General Public License along with
15  * this program; if not, write to the Free Software Foundation, Inc.,
16  * 51 Franklin Street, Fifth Floor, Boston, MA 02110, USA
17  *
18  * The full GNU General Public License is included in this distribution in the
19  * file called LICENSE.
20  *
21  * Contact Information:
22  * wlanfae <wlanfae@realtek.com>
23  * Realtek Corporation, No. 2, Innovation Road II, Hsinchu Science Park,
24  * Hsinchu 300, Taiwan.
25  *
26  * Larry Finger <Larry.Finger@lwfinger.net>
27  *
28  *****************************************************************************/
29
30 #include "wifi.h"
31 #include "base.h"
32 #include "rc.h"
33
34 /*
35  *Finds the highest rate index we can use
36  *if skb is special data like DHCP/EAPOL, we set should
37  *it to lowest rate CCK_1M, otherwise we set rate to
38  *CCK11M or OFDM_54M based on wireless mode.
39  */
40 static u8 _rtl_rc_get_highest_rix(struct rtl_priv *rtlpriv,
41                                   struct ieee80211_sta *sta,
42                                   struct sk_buff *skb, bool not_data)
43 {
44         struct rtl_mac *rtlmac = rtl_mac(rtlpriv);
45         struct rtl_hal *rtlhal = rtl_hal(rtlpriv);
46         struct rtl_phy *rtlphy = &(rtlpriv->phy);
47         struct rtl_sta_info *sta_entry = NULL;
48         u8 wireless_mode = 0;
49
50         /*
51          *this rate is no use for true rate, firmware
52          *will control rate at all it just used for
53          *1.show in iwconfig in B/G mode
54          *2.in rtl_get_tcb_desc when we check rate is
55          *      1M we will not use FW rate but user rate.
56          */
57         if (rtlmac->opmode == NL80211_IFTYPE_AP ||
58             rtlmac->opmode == NL80211_IFTYPE_ADHOC ||
59             rtlmac->opmode == NL80211_IFTYPE_MESH_POINT) {
60                 if (sta) {
61                         sta_entry = (struct rtl_sta_info *) sta->drv_priv;
62                         wireless_mode = sta_entry->wireless_mode;
63                 } else {
64                         return 0;
65                 }
66         } else {
67                 wireless_mode = rtlmac->mode;
68         }
69
70         if (rtl_is_special_data(rtlpriv->mac80211.hw, skb, true) ||
71                         not_data) {
72                 return 0;
73         } else {
74                 if (rtlhal->current_bandtype == BAND_ON_2_4G) {
75                         if (wireless_mode == WIRELESS_MODE_B) {
76                                 return B_MODE_MAX_RIX;
77                         } else if (wireless_mode == WIRELESS_MODE_G) {
78                                 return G_MODE_MAX_RIX;
79                         } else {
80                                 if (get_rf_type(rtlphy) != RF_2T2R)
81                                         return N_MODE_MCS7_RIX;
82                                 else
83                                         return N_MODE_MCS15_RIX;
84                         }
85                 } else {
86                         if (wireless_mode == WIRELESS_MODE_A) {
87                                 return A_MODE_MAX_RIX;
88                         } else {
89                                 if (get_rf_type(rtlphy) != RF_2T2R)
90                                         return N_MODE_MCS7_RIX;
91                                 else
92                                         return N_MODE_MCS15_RIX;
93                         }
94                 }
95         }
96 }
97
98 static void _rtl_rc_rate_set_series(struct rtl_priv *rtlpriv,
99                                     struct ieee80211_sta *sta,
100                                     struct ieee80211_tx_rate *rate,
101                                     struct ieee80211_tx_rate_control *txrc,
102                                     u8 tries, char rix, int rtsctsenable,
103                                     bool not_data)
104 {
105         struct rtl_mac *mac = rtl_mac(rtlpriv);
106         u8 sgi_20 = 0, sgi_40 = 0;
107
108         if (sta) {
109                 sgi_20 = sta->ht_cap.cap & IEEE80211_HT_CAP_SGI_20;
110                 sgi_40 = sta->ht_cap.cap & IEEE80211_HT_CAP_SGI_40;
111         }
112         rate->count = tries;
113         rate->idx = rix >= 0x00 ? rix : 0x00;
114
115         if (!not_data) {
116                 if (txrc->short_preamble)
117                         rate->flags |= IEEE80211_TX_RC_USE_SHORT_PREAMBLE;
118                 if (mac->opmode == NL80211_IFTYPE_AP ||
119                         mac->opmode == NL80211_IFTYPE_ADHOC) {
120                         if (sta && (sta->ht_cap.cap &
121                             IEEE80211_HT_CAP_SUP_WIDTH_20_40))
122                                 rate->flags |= IEEE80211_TX_RC_40_MHZ_WIDTH;
123                 } else {
124                         if (mac->bw_40)
125                                 rate->flags |= IEEE80211_TX_RC_40_MHZ_WIDTH;
126                 }
127                 if (sgi_20 || sgi_40)
128                         rate->flags |= IEEE80211_TX_RC_SHORT_GI;
129                 if (sta && sta->ht_cap.ht_supported)
130                         rate->flags |= IEEE80211_TX_RC_MCS;
131         }
132 }
133
134 static void rtl_get_rate(void *ppriv, struct ieee80211_sta *sta,
135                          void *priv_sta, struct ieee80211_tx_rate_control *txrc)
136 {
137         struct rtl_priv *rtlpriv = ppriv;
138         struct sk_buff *skb = txrc->skb;
139         struct ieee80211_tx_info *tx_info = IEEE80211_SKB_CB(skb);
140         struct ieee80211_tx_rate *rates = tx_info->control.rates;
141         __le16 fc = rtl_get_fc(skb);
142         u8 try_per_rate, i, rix;
143         bool not_data = !ieee80211_is_data(fc);
144
145         if (rate_control_send_low(sta, priv_sta, txrc))
146                 return;
147
148         rix = _rtl_rc_get_highest_rix(rtlpriv, sta, skb, not_data);
149         try_per_rate = 1;
150         _rtl_rc_rate_set_series(rtlpriv, sta, &rates[0], txrc,
151                                 try_per_rate, rix, 1, not_data);
152
153         if (!not_data) {
154                 for (i = 1; i < 4; i++)
155                         _rtl_rc_rate_set_series(rtlpriv, sta, &rates[i],
156                                                 txrc, i, (rix - i), 1,
157                                                 not_data);
158         }
159 }
160
161 static bool _rtl_tx_aggr_check(struct rtl_priv *rtlpriv,
162                 struct rtl_sta_info *sta_entry, u16 tid)
163 {
164         struct rtl_mac *mac = rtl_mac(rtlpriv);
165
166         if (mac->act_scanning)
167                 return false;
168
169         if (mac->opmode == NL80211_IFTYPE_STATION &&
170                 mac->cnt_after_linked < 3)
171                 return false;
172
173         if (sta_entry->tids[tid].agg.agg_state == RTL_AGG_STOP)
174                 return true;
175
176         return false;
177 }
178
179 /*mac80211 Rate Control callbacks*/
180 static void rtl_tx_status(void *ppriv,
181                           struct ieee80211_supported_band *sband,
182                           struct ieee80211_sta *sta, void *priv_sta,
183                           struct sk_buff *skb)
184 {
185         struct rtl_priv *rtlpriv = ppriv;
186         struct rtl_mac *mac = rtl_mac(rtlpriv);
187         struct ieee80211_hdr *hdr = rtl_get_hdr(skb);
188         __le16 fc = rtl_get_fc(skb);
189         struct rtl_sta_info *sta_entry;
190
191         if (!priv_sta || !ieee80211_is_data(fc))
192                 return;
193
194         if (rtl_is_special_data(mac->hw, skb, true))
195                 return;
196
197         if (is_multicast_ether_addr(ieee80211_get_DA(hdr))
198             || is_broadcast_ether_addr(ieee80211_get_DA(hdr)))
199                 return;
200
201         if (sta) {
202                 /* Check if aggregation has to be enabled for this tid */
203                 sta_entry = (struct rtl_sta_info *) sta->drv_priv;
204                 if ((sta->ht_cap.ht_supported) &&
205                                 !(skb->protocol == cpu_to_be16(ETH_P_PAE))) {
206                         if (ieee80211_is_data_qos(fc)) {
207                                 u8 tid = rtl_get_tid(skb);
208                                 if (_rtl_tx_aggr_check(rtlpriv, sta_entry,
209                                     tid)) {
210                                         sta_entry->tids[tid].agg.agg_state =
211                                                          RTL_AGG_PROGRESS;
212                                         ieee80211_start_tx_ba_session(sta,
213                                                                  tid, 5000);
214                                 }
215                         }
216                 }
217         }
218 }
219
220 static void rtl_rate_init(void *ppriv,
221                           struct ieee80211_supported_band *sband,
222                           struct ieee80211_sta *sta, void *priv_sta)
223 {
224 }
225
226 static void rtl_rate_update(void *ppriv,
227                             struct ieee80211_supported_band *sband,
228                             struct ieee80211_sta *sta, void *priv_sta,
229                             u32 changed)
230 {
231 }
232
233 static void *rtl_rate_alloc(struct ieee80211_hw *hw,
234                 struct dentry *debugfsdir)
235 {
236         struct rtl_priv *rtlpriv = rtl_priv(hw);
237         return rtlpriv;
238 }
239
240 static void rtl_rate_free(void *rtlpriv)
241 {
242         return;
243 }
244
245 static void *rtl_rate_alloc_sta(void *ppriv,
246                                 struct ieee80211_sta *sta, gfp_t gfp)
247 {
248         struct rtl_priv *rtlpriv = ppriv;
249         struct rtl_rate_priv *rate_priv;
250
251         rate_priv = kzalloc(sizeof(struct rtl_rate_priv), gfp);
252         if (!rate_priv) {
253                 RT_TRACE(rtlpriv, COMP_ERR, DBG_EMERG,
254                          "Unable to allocate private rc structure\n");
255                 return NULL;
256         }
257
258         rtlpriv->rate_priv = rate_priv;
259
260         return rate_priv;
261 }
262
263 static void rtl_rate_free_sta(void *rtlpriv,
264                               struct ieee80211_sta *sta, void *priv_sta)
265 {
266         struct rtl_rate_priv *rate_priv = priv_sta;
267         kfree(rate_priv);
268 }
269
270 static struct rate_control_ops rtl_rate_ops = {
271         .module = NULL,
272         .name = "rtl_rc",
273         .alloc = rtl_rate_alloc,
274         .free = rtl_rate_free,
275         .alloc_sta = rtl_rate_alloc_sta,
276         .free_sta = rtl_rate_free_sta,
277         .rate_init = rtl_rate_init,
278         .rate_update = rtl_rate_update,
279         .tx_status = rtl_tx_status,
280         .get_rate = rtl_get_rate,
281 };
282
283 int rtl_rate_control_register(void)
284 {
285         return ieee80211_rate_control_register(&rtl_rate_ops);
286 }
287
288 void rtl_rate_control_unregister(void)
289 {
290         ieee80211_rate_control_unregister(&rtl_rate_ops);
291 }