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