drivers/net/wireless/realtek/rtlwifi/base.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /* Copyright(c) 2009-2012  Realtek Corporation.*/
   3
   4 #include "wifi.h"
   5 #include "rc.h"
   6 #include "base.h"
   7 #include "efuse.h"
   8 #include "cam.h"
   9 #include "ps.h"
  10 #include "regd.h"
  11 #include "pci.h"
  12 #include <linux/ip.h>
  13 #include <linux/module.h>
  14 #include <linux/udp.h>
  15
  16 /*
  17  *NOTICE!!!: This file will be very big, we should
  18  *keep it clear under following roles:
  19  *
  20  *This file include following parts, so, if you add new
  21  *functions into this file, please check which part it
  22  *should includes. or check if you should add new part
  23  *for this file:
  24  *
  25  *1) mac80211 init functions
  26  *2) tx information functions
  27  *3) functions called by core.c
  28  *4) wq & timer callback functions
  29  *5) frame process functions
  30  *6) IOT functions
  31  *7) sysfs functions
  32  *8) vif functions
  33  *9) ...
  34  */
  35
  36 /*********************************************************
  37  *
  38  * mac80211 init functions
  39  *
  40  *********************************************************/
  41 static struct ieee80211_channel rtl_channeltable_2g[] = {
  42         {.center_freq = 2412, .hw_value = 1,},
  43         {.center_freq = 2417, .hw_value = 2,},
  44         {.center_freq = 2422, .hw_value = 3,},
  45         {.center_freq = 2427, .hw_value = 4,},
  46         {.center_freq = 2432, .hw_value = 5,},
  47         {.center_freq = 2437, .hw_value = 6,},
  48         {.center_freq = 2442, .hw_value = 7,},
  49         {.center_freq = 2447, .hw_value = 8,},
  50         {.center_freq = 2452, .hw_value = 9,},
  51         {.center_freq = 2457, .hw_value = 10,},
  52         {.center_freq = 2462, .hw_value = 11,},
  53         {.center_freq = 2467, .hw_value = 12,},
  54         {.center_freq = 2472, .hw_value = 13,},
  55         {.center_freq = 2484, .hw_value = 14,},
  56 };
  57
  58 static struct ieee80211_channel rtl_channeltable_5g[] = {
  59         {.center_freq = 5180, .hw_value = 36,},
  60         {.center_freq = 5200, .hw_value = 40,},
  61         {.center_freq = 5220, .hw_value = 44,},
  62         {.center_freq = 5240, .hw_value = 48,},
  63         {.center_freq = 5260, .hw_value = 52,},
  64         {.center_freq = 5280, .hw_value = 56,},
  65         {.center_freq = 5300, .hw_value = 60,},
  66         {.center_freq = 5320, .hw_value = 64,},
  67         {.center_freq = 5500, .hw_value = 100,},
  68         {.center_freq = 5520, .hw_value = 104,},
  69         {.center_freq = 5540, .hw_value = 108,},
  70         {.center_freq = 5560, .hw_value = 112,},
  71         {.center_freq = 5580, .hw_value = 116,},
  72         {.center_freq = 5600, .hw_value = 120,},
  73         {.center_freq = 5620, .hw_value = 124,},
  74         {.center_freq = 5640, .hw_value = 128,},
  75         {.center_freq = 5660, .hw_value = 132,},
  76         {.center_freq = 5680, .hw_value = 136,},
  77         {.center_freq = 5700, .hw_value = 140,},
  78         {.center_freq = 5745, .hw_value = 149,},
  79         {.center_freq = 5765, .hw_value = 153,},
  80         {.center_freq = 5785, .hw_value = 157,},
  81         {.center_freq = 5805, .hw_value = 161,},
  82         {.center_freq = 5825, .hw_value = 165,},
  83 };
  84
  85 static struct ieee80211_rate rtl_ratetable_2g[] = {
  86         {.bitrate = 10, .hw_value = 0x00,},
  87         {.bitrate = 20, .hw_value = 0x01,},
  88         {.bitrate = 55, .hw_value = 0x02,},
  89         {.bitrate = 110, .hw_value = 0x03,},
  90         {.bitrate = 60, .hw_value = 0x04,},
  91         {.bitrate = 90, .hw_value = 0x05,},
  92         {.bitrate = 120, .hw_value = 0x06,},
  93         {.bitrate = 180, .hw_value = 0x07,},
  94         {.bitrate = 240, .hw_value = 0x08,},
  95         {.bitrate = 360, .hw_value = 0x09,},
  96         {.bitrate = 480, .hw_value = 0x0a,},
  97         {.bitrate = 540, .hw_value = 0x0b,},
  98 };
  99
 100 static struct ieee80211_rate rtl_ratetable_5g[] = {
 101         {.bitrate = 60, .hw_value = 0x04,},
 102         {.bitrate = 90, .hw_value = 0x05,},
 103         {.bitrate = 120, .hw_value = 0x06,},
 104         {.bitrate = 180, .hw_value = 0x07,},
 105         {.bitrate = 240, .hw_value = 0x08,},
 106         {.bitrate = 360, .hw_value = 0x09,},
 107         {.bitrate = 480, .hw_value = 0x0a,},
 108         {.bitrate = 540, .hw_value = 0x0b,},
 109 };
 110
 111 static const struct ieee80211_supported_band rtl_band_2ghz = {
 112         .band = NL80211_BAND_2GHZ,
 113
 114         .channels = rtl_channeltable_2g,
 115         .n_channels = ARRAY_SIZE(rtl_channeltable_2g),
 116
 117         .bitrates = rtl_ratetable_2g,
 118         .n_bitrates = ARRAY_SIZE(rtl_ratetable_2g),
 119
 120         .ht_cap = {0},
 121 };
 122
 123 static struct ieee80211_supported_band rtl_band_5ghz = {
 124         .band = NL80211_BAND_5GHZ,
 125
 126         .channels = rtl_channeltable_5g,
 127         .n_channels = ARRAY_SIZE(rtl_channeltable_5g),
 128
 129         .bitrates = rtl_ratetable_5g,
 130         .n_bitrates = ARRAY_SIZE(rtl_ratetable_5g),
 131
 132         .ht_cap = {0},
 133 };
 134
 135 static const u8 tid_to_ac[] = {
 136         2, /* IEEE80211_AC_BE */
 137         3, /* IEEE80211_AC_BK */
 138         3, /* IEEE80211_AC_BK */
 139         2, /* IEEE80211_AC_BE */
 140         1, /* IEEE80211_AC_VI */
 141         1, /* IEEE80211_AC_VI */
 142         0, /* IEEE80211_AC_VO */
 143         0, /* IEEE80211_AC_VO */
 144 };
 145
 146 u8 rtl_tid_to_ac(u8 tid)
 147 {
 148         return tid_to_ac[tid];
 149 }
 150 EXPORT_SYMBOL_GPL(rtl_tid_to_ac);
 151
 152 static void _rtl_init_hw_ht_capab(struct ieee80211_hw *hw,
 153                                   struct ieee80211_sta_ht_cap *ht_cap)
 154 {
 155         struct rtl_priv *rtlpriv = rtl_priv(hw);
 156         struct rtl_phy *rtlphy = &(rtlpriv->phy);
 157
 158         ht_cap->ht_supported = true;
 159         ht_cap->cap = IEEE80211_HT_CAP_SUP_WIDTH_20_40 |
 160             IEEE80211_HT_CAP_SGI_40 |
 161             IEEE80211_HT_CAP_SGI_20 |
 162             IEEE80211_HT_CAP_DSSSCCK40 | IEEE80211_HT_CAP_MAX_AMSDU;
 163
 164         if (rtlpriv->rtlhal.disable_amsdu_8k)
 165                 ht_cap->cap &= ~IEEE80211_HT_CAP_MAX_AMSDU;
 166
 167         /*
 168          *Maximum length of AMPDU that the STA can receive.
 169          *Length = 2 ^ (13 + max_ampdu_length_exp) - 1 (octets)
 170          */
 171         ht_cap->ampdu_factor = IEEE80211_HT_MAX_AMPDU_64K;
 172
 173         /*Minimum MPDU start spacing , */
 174         ht_cap->ampdu_density = IEEE80211_HT_MPDU_DENSITY_16;
 175
 176         ht_cap->mcs.tx_params = IEEE80211_HT_MCS_TX_DEFINED;
 177
 178         /*hw->wiphy->bands[NL80211_BAND_2GHZ]
 179          *base on ant_num
 180          *rx_mask: RX mask
 181          *if rx_ant = 1 rx_mask[0]= 0xff;==>MCS0-MCS7
 182          *if rx_ant = 2 rx_mask[1]= 0xff;==>MCS8-MCS15
 183          *if rx_ant >= 3 rx_mask[2]= 0xff;
 184          *if BW_40 rx_mask[4]= 0x01;
 185          *highest supported RX rate
 186          */
 187         if (rtlpriv->dm.supp_phymode_switch) {
 188                 pr_info("Support phy mode switch\n");
 189
 190                 ht_cap->mcs.rx_mask[0] = 0xFF;
 191                 ht_cap->mcs.rx_mask[1] = 0xFF;
 192                 ht_cap->mcs.rx_mask[4] = 0x01;
 193
 194                 ht_cap->mcs.rx_highest = cpu_to_le16(MAX_BIT_RATE_40MHZ_MCS15);
 195         } else {
 196                 if (get_rf_type(rtlphy) == RF_1T2R ||
 197                     get_rf_type(rtlphy) == RF_2T2R) {
 198                         RT_TRACE(rtlpriv, COMP_INIT, DBG_DMESG,
 199                                  "1T2R or 2T2R\n");
 200                         ht_cap->mcs.rx_mask[0] = 0xFF;
 201                         ht_cap->mcs.rx_mask[1] = 0xFF;
 202                         ht_cap->mcs.rx_mask[4] = 0x01;
 203
 204                         ht_cap->mcs.rx_highest =
 205                                  cpu_to_le16(MAX_BIT_RATE_40MHZ_MCS15);
 206                 } else if (get_rf_type(rtlphy) == RF_1T1R) {
 207                         RT_TRACE(rtlpriv, COMP_INIT, DBG_DMESG, "1T1R\n");
 208
 209                         ht_cap->mcs.rx_mask[0] = 0xFF;
 210                         ht_cap->mcs.rx_mask[1] = 0x00;
 211                         ht_cap->mcs.rx_mask[4] = 0x01;
 212
 213                         ht_cap->mcs.rx_highest =
 214                                  cpu_to_le16(MAX_BIT_RATE_40MHZ_MCS7);
 215                 }
 216         }
 217 }
 218
 219 static void _rtl_init_hw_vht_capab(struct ieee80211_hw *hw,
 220                                    struct ieee80211_sta_vht_cap *vht_cap)
 221 {
 222         struct rtl_priv *rtlpriv = rtl_priv(hw);
 223         struct rtl_hal *rtlhal = rtl_hal(rtlpriv);
 224
 225         if (!(rtlpriv->cfg->spec_ver & RTL_SPEC_SUPPORT_VHT))
 226                 return;
 227
 228         if (rtlhal->hw_type == HARDWARE_TYPE_RTL8812AE ||
 229             rtlhal->hw_type == HARDWARE_TYPE_RTL8822BE) {
 230                 u16 mcs_map;
 231
 232                 vht_cap->vht_supported = true;
 233                 vht_cap->cap =
 234                         IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454 |
 235                         IEEE80211_VHT_CAP_SHORT_GI_80 |
 236                         IEEE80211_VHT_CAP_TXSTBC |
 237                         IEEE80211_VHT_CAP_RXSTBC_1 |
 238                         IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE |
 239                         IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE |
 240                         IEEE80211_VHT_CAP_HTC_VHT |
 241                         IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK |
 242                         IEEE80211_VHT_CAP_RX_ANTENNA_PATTERN |
 243                         IEEE80211_VHT_CAP_TX_ANTENNA_PATTERN |
 244                         0;
 245
 246                 mcs_map = IEEE80211_VHT_MCS_SUPPORT_0_9 << 0 |
 247                         IEEE80211_VHT_MCS_SUPPORT_0_9 << 2 |
 248                         IEEE80211_VHT_MCS_NOT_SUPPORTED << 4 |
 249                         IEEE80211_VHT_MCS_NOT_SUPPORTED << 6 |
 250                         IEEE80211_VHT_MCS_NOT_SUPPORTED << 8 |
 251                         IEEE80211_VHT_MCS_NOT_SUPPORTED << 10 |
 252                         IEEE80211_VHT_MCS_NOT_SUPPORTED << 12 |
 253                         IEEE80211_VHT_MCS_NOT_SUPPORTED << 14;
 254
 255                 vht_cap->vht_mcs.rx_mcs_map = cpu_to_le16(mcs_map);
 256                 vht_cap->vht_mcs.rx_highest =
 257                         cpu_to_le16(MAX_BIT_RATE_SHORT_GI_2NSS_80MHZ_MCS9);
 258                 vht_cap->vht_mcs.tx_mcs_map = cpu_to_le16(mcs_map);
 259                 vht_cap->vht_mcs.tx_highest =
 260                         cpu_to_le16(MAX_BIT_RATE_SHORT_GI_2NSS_80MHZ_MCS9);
 261         } else if (rtlhal->hw_type == HARDWARE_TYPE_RTL8821AE) {
 262                 u16 mcs_map;
 263
 264                 vht_cap->vht_supported = true;
 265                 vht_cap->cap =
 266                         IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454 |
 267                         IEEE80211_VHT_CAP_SHORT_GI_80 |
 268                         IEEE80211_VHT_CAP_TXSTBC |
 269                         IEEE80211_VHT_CAP_RXSTBC_1 |
 270                         IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE |
 271                         IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE |
 272                         IEEE80211_VHT_CAP_HTC_VHT |
 273                         IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK |
 274                         IEEE80211_VHT_CAP_RX_ANTENNA_PATTERN |
 275                         IEEE80211_VHT_CAP_TX_ANTENNA_PATTERN |
 276                         0;
 277
 278                 mcs_map = IEEE80211_VHT_MCS_SUPPORT_0_9 << 0 |
 279                         IEEE80211_VHT_MCS_NOT_SUPPORTED << 2 |
 280                         IEEE80211_VHT_MCS_NOT_SUPPORTED << 4 |
 281                         IEEE80211_VHT_MCS_NOT_SUPPORTED << 6 |
 282                         IEEE80211_VHT_MCS_NOT_SUPPORTED << 8 |
 283                         IEEE80211_VHT_MCS_NOT_SUPPORTED << 10 |
 284                         IEEE80211_VHT_MCS_NOT_SUPPORTED << 12 |
 285                         IEEE80211_VHT_MCS_NOT_SUPPORTED << 14;
 286
 287                 vht_cap->vht_mcs.rx_mcs_map = cpu_to_le16(mcs_map);
 288                 vht_cap->vht_mcs.rx_highest =
 289                         cpu_to_le16(MAX_BIT_RATE_SHORT_GI_1NSS_80MHZ_MCS9);
 290                 vht_cap->vht_mcs.tx_mcs_map = cpu_to_le16(mcs_map);
 291                 vht_cap->vht_mcs.tx_highest =
 292                         cpu_to_le16(MAX_BIT_RATE_SHORT_GI_1NSS_80MHZ_MCS9);
 293         }
 294 }
 295
 296 static void _rtl_init_mac80211(struct ieee80211_hw *hw)
 297 {
 298         struct rtl_priv *rtlpriv = rtl_priv(hw);
 299         struct rtl_hal *rtlhal = rtl_hal(rtlpriv);
 300         struct rtl_mac *rtlmac = rtl_mac(rtl_priv(hw));
 301         struct rtl_efuse *rtlefuse = rtl_efuse(rtl_priv(hw));
 302         struct ieee80211_supported_band *sband;
 303
 304         if (rtlhal->macphymode == SINGLEMAC_SINGLEPHY &&
 305             rtlhal->bandset == BAND_ON_BOTH) {
 306                 /* 1: 2.4 G bands */
 307                 /* <1> use  mac->bands as mem for hw->wiphy->bands */
 308                 sband = &(rtlmac->bands[NL80211_BAND_2GHZ]);
 309
 310                 /* <2> set hw->wiphy->bands[NL80211_BAND_2GHZ]
 311                  * to default value(1T1R) */
 312                 memcpy(&(rtlmac->bands[NL80211_BAND_2GHZ]), &rtl_band_2ghz,
 313                                 sizeof(struct ieee80211_supported_band));
 314
 315                 /* <3> init ht cap base on ant_num */
 316                 _rtl_init_hw_ht_capab(hw, &sband->ht_cap);
 317
 318                 /* <4> set mac->sband to wiphy->sband */
 319                 hw->wiphy->bands[NL80211_BAND_2GHZ] = sband;
 320
 321                 /* 2: 5 G bands */
 322                 /* <1> use  mac->bands as mem for hw->wiphy->bands */
 323                 sband = &(rtlmac->bands[NL80211_BAND_5GHZ]);
 324
 325                 /* <2> set hw->wiphy->bands[NL80211_BAND_5GHZ]
 326                  * to default value(1T1R) */
 327                 memcpy(&(rtlmac->bands[NL80211_BAND_5GHZ]), &rtl_band_5ghz,
 328                                 sizeof(struct ieee80211_supported_band));
 329
 330                 /* <3> init ht cap base on ant_num */
 331                 _rtl_init_hw_ht_capab(hw, &sband->ht_cap);
 332
 333                 _rtl_init_hw_vht_capab(hw, &sband->vht_cap);
 334                 /* <4> set mac->sband to wiphy->sband */
 335                 hw->wiphy->bands[NL80211_BAND_5GHZ] = sband;
 336         } else {
 337                 if (rtlhal->current_bandtype == BAND_ON_2_4G) {
 338                         /* <1> use  mac->bands as mem for hw->wiphy->bands */
 339                         sband = &(rtlmac->bands[NL80211_BAND_2GHZ]);
 340
 341                         /* <2> set hw->wiphy->bands[NL80211_BAND_2GHZ]
 342                          * to default value(1T1R) */
 343                         memcpy(&(rtlmac->bands[NL80211_BAND_2GHZ]),
 344                                &rtl_band_2ghz,
 345                                sizeof(struct ieee80211_supported_band));
 346
 347                         /* <3> init ht cap base on ant_num */
 348                         _rtl_init_hw_ht_capab(hw, &sband->ht_cap);
 349
 350                         /* <4> set mac->sband to wiphy->sband */
 351                         hw->wiphy->bands[NL80211_BAND_2GHZ] = sband;
 352                 } else if (rtlhal->current_bandtype == BAND_ON_5G) {
 353                         /* <1> use  mac->bands as mem for hw->wiphy->bands */
 354                         sband = &(rtlmac->bands[NL80211_BAND_5GHZ]);
 355
 356                         /* <2> set hw->wiphy->bands[NL80211_BAND_5GHZ]
 357                          * to default value(1T1R) */
 358                         memcpy(&(rtlmac->bands[NL80211_BAND_5GHZ]),
 359                                &rtl_band_5ghz,
 360                                sizeof(struct ieee80211_supported_band));
 361
 362                         /* <3> init ht cap base on ant_num */
 363                         _rtl_init_hw_ht_capab(hw, &sband->ht_cap);
 364
 365                         _rtl_init_hw_vht_capab(hw, &sband->vht_cap);
 366                         /* <4> set mac->sband to wiphy->sband */
 367                         hw->wiphy->bands[NL80211_BAND_5GHZ] = sband;
 368                 } else {
 369                         pr_err("Err BAND %d\n",
 370                                rtlhal->current_bandtype);
 371                 }
 372         }
 373         /* <5> set hw caps */
 374         ieee80211_hw_set(hw, SIGNAL_DBM);
 375         ieee80211_hw_set(hw, RX_INCLUDES_FCS);
 376         ieee80211_hw_set(hw, AMPDU_AGGREGATION);
 377         ieee80211_hw_set(hw, MFP_CAPABLE);
 378         ieee80211_hw_set(hw, REPORTS_TX_ACK_STATUS);
 379         ieee80211_hw_set(hw, SUPPORTS_AMSDU_IN_AMPDU);
 380         ieee80211_hw_set(hw, SUPPORT_FAST_XMIT);
 381
 382         /* swlps or hwlps has been set in diff chip in init_sw_vars */
 383         if (rtlpriv->psc.swctrl_lps) {
 384                 ieee80211_hw_set(hw, SUPPORTS_PS);
 385                 ieee80211_hw_set(hw, PS_NULLFUNC_STACK);
 386         }
 387         if (rtlpriv->psc.fwctrl_lps) {
 388                 ieee80211_hw_set(hw, SUPPORTS_PS);
 389                 ieee80211_hw_set(hw, SUPPORTS_DYNAMIC_PS);
 390         }
 391         hw->wiphy->interface_modes =
 392             BIT(NL80211_IFTYPE_AP) |
 393             BIT(NL80211_IFTYPE_STATION) |
 394             BIT(NL80211_IFTYPE_ADHOC) |
 395             BIT(NL80211_IFTYPE_MESH_POINT) |
 396             BIT(NL80211_IFTYPE_P2P_CLIENT) |
 397             BIT(NL80211_IFTYPE_P2P_GO);
 398         hw->wiphy->flags |= WIPHY_FLAG_IBSS_RSN;
 399
 400         hw->wiphy->flags |= WIPHY_FLAG_HAS_REMAIN_ON_CHANNEL;
 401
 402         hw->wiphy->rts_threshold = 2347;
 403
 404         hw->queues = AC_MAX;
 405         hw->extra_tx_headroom = RTL_TX_HEADER_SIZE;
 406
 407         /* TODO: Correct this value for our hw */
 408         hw->max_listen_interval = MAX_LISTEN_INTERVAL;
 409         hw->max_rate_tries = MAX_RATE_TRIES;
 410         /* hw->max_rates = 1; */
 411         hw->sta_data_size = sizeof(struct rtl_sta_info);
 412
 413 /* wowlan is not supported by kernel if CONFIG_PM is not defined */
 414 #ifdef CONFIG_PM
 415         if (rtlpriv->psc.wo_wlan_mode) {
 416                 if (rtlpriv->psc.wo_wlan_mode & WAKE_ON_MAGIC_PACKET)
 417                         rtlpriv->wowlan.flags = WIPHY_WOWLAN_MAGIC_PKT;
 418                 if (rtlpriv->psc.wo_wlan_mode & WAKE_ON_PATTERN_MATCH) {
 419                         rtlpriv->wowlan.n_patterns =
 420                                 MAX_SUPPORT_WOL_PATTERN_NUM;
 421                         rtlpriv->wowlan.pattern_min_len = MIN_WOL_PATTERN_SIZE;
 422                         rtlpriv->wowlan.pattern_max_len = MAX_WOL_PATTERN_SIZE;
 423                 }
 424                 hw->wiphy->wowlan = &rtlpriv->wowlan;
 425         }
 426 #endif
 427
 428         /* <6> mac address */
 429         if (is_valid_ether_addr(rtlefuse->dev_addr)) {
 430                 SET_IEEE80211_PERM_ADDR(hw, rtlefuse->dev_addr);
 431         } else {
 432                 u8 rtlmac1[] = { 0x00, 0xe0, 0x4c, 0x81, 0x92, 0x00 };
 433
 434                 get_random_bytes((rtlmac1 + (ETH_ALEN - 1)), 1);
 435                 SET_IEEE80211_PERM_ADDR(hw, rtlmac1);
 436         }
 437 }
 438
 439 static void _rtl_init_deferred_work(struct ieee80211_hw *hw)
 440 {
 441         struct rtl_priv *rtlpriv = rtl_priv(hw);
 442
 443         /* <1> timer */
 444         timer_setup(&rtlpriv->works.watchdog_timer,
 445                     rtl_watch_dog_timer_callback, 0);
 446         timer_setup(&rtlpriv->works.dualmac_easyconcurrent_retrytimer,
 447                     rtl_easy_concurrent_retrytimer_callback, 0);
 448         /* <2> work queue */
 449         rtlpriv->works.hw = hw;
 450         rtlpriv->works.rtl_wq = alloc_workqueue("%s", 0, 0, rtlpriv->cfg->name);
 451         if (unlikely(!rtlpriv->works.rtl_wq)) {
 452                 pr_err("Failed to allocate work queue\n");
 453                 return;
 454         }
 455
 456         INIT_DELAYED_WORK(&rtlpriv->works.watchdog_wq,
 457                           (void *)rtl_watchdog_wq_callback);
 458         INIT_DELAYED_WORK(&rtlpriv->works.ips_nic_off_wq,
 459                           (void *)rtl_ips_nic_off_wq_callback);
 460         INIT_DELAYED_WORK(&rtlpriv->works.ps_work,
 461                           (void *)rtl_swlps_wq_callback);
 462         INIT_DELAYED_WORK(&rtlpriv->works.ps_rfon_wq,
 463                           (void *)rtl_swlps_rfon_wq_callback);
 464         INIT_DELAYED_WORK(&rtlpriv->works.fwevt_wq,
 465                           (void *)rtl_fwevt_wq_callback);
 466         INIT_DELAYED_WORK(&rtlpriv->works.c2hcmd_wq,
 467                           (void *)rtl_c2hcmd_wq_callback);
 468 }
 469
 470 void rtl_deinit_deferred_work(struct ieee80211_hw *hw, bool ips_wq)
 471 {
 472         struct rtl_priv *rtlpriv = rtl_priv(hw);
 473
 474         del_timer_sync(&rtlpriv->works.watchdog_timer);
 475
 476         cancel_delayed_work_sync(&rtlpriv->works.watchdog_wq);
 477         if (ips_wq)
 478                 cancel_delayed_work(&rtlpriv->works.ips_nic_off_wq);
 479         else
 480                 cancel_delayed_work_sync(&rtlpriv->works.ips_nic_off_wq);
 481         cancel_delayed_work_sync(&rtlpriv->works.ps_work);
 482         cancel_delayed_work_sync(&rtlpriv->works.ps_rfon_wq);
 483         cancel_delayed_work_sync(&rtlpriv->works.fwevt_wq);
 484         cancel_delayed_work_sync(&rtlpriv->works.c2hcmd_wq);
 485 }
 486 EXPORT_SYMBOL_GPL(rtl_deinit_deferred_work);
 487
 488 void rtl_init_rfkill(struct ieee80211_hw *hw)
 489 {
 490         struct rtl_priv *rtlpriv = rtl_priv(hw);
 491
 492         bool radio_state;
 493         bool blocked;
 494         u8 valid = 0;
 495
 496         /*set init state to on */
 497         rtlpriv->rfkill.rfkill_state = true;
 498         wiphy_rfkill_set_hw_state(hw->wiphy, 0);
 499
 500         radio_state = rtlpriv->cfg->ops->radio_onoff_checking(hw, &valid);
 501
 502         if (valid) {
 503                 pr_info("rtlwifi: wireless switch is %s\n",
 504                         rtlpriv->rfkill.rfkill_state ? "on" : "off");
 505
 506                 rtlpriv->rfkill.rfkill_state = radio_state;
 507
 508                 blocked = (rtlpriv->rfkill.rfkill_state == 1) ? 0 : 1;
 509                 wiphy_rfkill_set_hw_state(hw->wiphy, blocked);
 510         }
 511
 512         wiphy_rfkill_start_polling(hw->wiphy);
 513 }
 514 EXPORT_SYMBOL(rtl_init_rfkill);
 515
 516 void rtl_deinit_rfkill(struct ieee80211_hw *hw)
 517 {
 518         wiphy_rfkill_stop_polling(hw->wiphy);
 519 }
 520 EXPORT_SYMBOL_GPL(rtl_deinit_rfkill);
 521
 522 int rtl_init_core(struct ieee80211_hw *hw)
 523 {
 524         struct rtl_priv *rtlpriv = rtl_priv(hw);
 525         struct rtl_mac *rtlmac = rtl_mac(rtl_priv(hw));
 526
 527         /* <1> init mac80211 */
 528         _rtl_init_mac80211(hw);
 529         rtlmac->hw = hw;
 530
 531         /* <2> rate control register */
 532         hw->rate_control_algorithm = "rtl_rc";
 533
 534         /*
 535          * <3> init CRDA must come after init
 536          * mac80211 hw  in _rtl_init_mac80211.
 537          */
 538         if (rtl_regd_init(hw, rtl_reg_notifier)) {
 539                 pr_err("REGD init failed\n");
 540                 return 1;
 541         }
 542
 543         /* <4> locks */
 544         mutex_init(&rtlpriv->locks.conf_mutex);
 545         mutex_init(&rtlpriv->locks.ips_mutex);
 546         mutex_init(&rtlpriv->locks.lps_mutex);
 547         spin_lock_init(&rtlpriv->locks.irq_th_lock);
 548         spin_lock_init(&rtlpriv->locks.h2c_lock);
 549         spin_lock_init(&rtlpriv->locks.rf_ps_lock);
 550         spin_lock_init(&rtlpriv->locks.rf_lock);
 551         spin_lock_init(&rtlpriv->locks.waitq_lock);
 552         spin_lock_init(&rtlpriv->locks.entry_list_lock);
 553         spin_lock_init(&rtlpriv->locks.c2hcmd_lock);
 554         spin_lock_init(&rtlpriv->locks.scan_list_lock);
 555         spin_lock_init(&rtlpriv->locks.cck_and_rw_pagea_lock);
 556         spin_lock_init(&rtlpriv->locks.fw_ps_lock);
 557         spin_lock_init(&rtlpriv->locks.iqk_lock);
 558         /* <5> init list */
 559         INIT_LIST_HEAD(&rtlpriv->entry_list);
 560         INIT_LIST_HEAD(&rtlpriv->scan_list.list);
 561         skb_queue_head_init(&rtlpriv->tx_report.queue);
 562         skb_queue_head_init(&rtlpriv->c2hcmd_queue);
 563
 564         rtlmac->link_state = MAC80211_NOLINK;
 565
 566         /* <6> init deferred work */
 567         _rtl_init_deferred_work(hw);
 568
 569         return 0;
 570 }
 571 EXPORT_SYMBOL_GPL(rtl_init_core);
 572
 573 static void rtl_free_entries_from_scan_list(struct ieee80211_hw *hw);
 574 static void rtl_free_entries_from_ack_queue(struct ieee80211_hw *hw,
 575                                             bool timeout);
 576
 577 void rtl_deinit_core(struct ieee80211_hw *hw)
 578 {
 579         rtl_c2hcmd_launcher(hw, 0);
 580         rtl_free_entries_from_scan_list(hw);
 581         rtl_free_entries_from_ack_queue(hw, false);
 582 }
 583 EXPORT_SYMBOL_GPL(rtl_deinit_core);
 584
 585 void rtl_init_rx_config(struct ieee80211_hw *hw)
 586 {
 587         struct rtl_priv *rtlpriv = rtl_priv(hw);
 588         struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
 589
 590         rtlpriv->cfg->ops->get_hw_reg(hw, HW_VAR_RCR, (u8 *) (&mac->rx_conf));
 591 }
 592 EXPORT_SYMBOL_GPL(rtl_init_rx_config);
 593
 594 /*********************************************************
 595  *
 596  * tx information functions
 597  *
 598  *********************************************************/
 599 static void _rtl_qurey_shortpreamble_mode(struct ieee80211_hw *hw,
 600                                           struct rtl_tcb_desc *tcb_desc,
 601                                           struct ieee80211_tx_info *info)
 602 {
 603         struct rtl_priv *rtlpriv = rtl_priv(hw);
 604         u8 rate_flag = info->control.rates[0].flags;
 605
 606         tcb_desc->use_shortpreamble = false;
 607
 608         /* 1M can only use Long Preamble. 11B spec */
 609         if (tcb_desc->hw_rate == rtlpriv->cfg->maps[RTL_RC_CCK_RATE1M])
 610                 return;
 611         else if (rate_flag & IEEE80211_TX_RC_USE_SHORT_PREAMBLE)
 612                 tcb_desc->use_shortpreamble = true;
 613
 614         return;
 615 }
 616
 617 static void _rtl_query_shortgi(struct ieee80211_hw *hw,
 618                                struct ieee80211_sta *sta,
 619                                struct rtl_tcb_desc *tcb_desc,
 620                                struct ieee80211_tx_info *info)
 621 {
 622         struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
 623         u8 rate_flag = info->control.rates[0].flags;
 624         u8 sgi_40 = 0, sgi_20 = 0, bw_40 = 0;
 625         u8 sgi_80 = 0, bw_80 = 0;
 626
 627         tcb_desc->use_shortgi = false;
 628
 629         if (sta == NULL)
 630                 return;
 631
 632         sgi_40 = sta->ht_cap.cap & IEEE80211_HT_CAP_SGI_40;
 633         sgi_20 = sta->ht_cap.cap & IEEE80211_HT_CAP_SGI_20;
 634         sgi_80 = sta->vht_cap.cap & IEEE80211_VHT_CAP_SHORT_GI_80;
 635
 636         if ((!sta->ht_cap.ht_supported) && (!sta->vht_cap.vht_supported))
 637                 return;
 638
 639         if (!sgi_40 && !sgi_20)
 640                 return;
 641
 642         if (mac->opmode == NL80211_IFTYPE_STATION) {
 643                 bw_40 = mac->bw_40;
 644                 bw_80 = mac->bw_80;
 645         } else if (mac->opmode == NL80211_IFTYPE_AP ||
 646                  mac->opmode == NL80211_IFTYPE_ADHOC ||
 647                  mac->opmode == NL80211_IFTYPE_MESH_POINT) {
 648                 bw_40 = sta->ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40;
 649                 bw_80 = sta->vht_cap.vht_supported;
 650         }
 651
 652         if (bw_80) {
 653                 if (sgi_80)
 654                         tcb_desc->use_shortgi = true;
 655                 else
 656                         tcb_desc->use_shortgi = false;
 657         } else {
 658                 if (bw_40 && sgi_40)
 659                         tcb_desc->use_shortgi = true;
 660                 else if (!bw_40 && sgi_20)
 661                         tcb_desc->use_shortgi = true;
 662         }
 663
 664         if (!(rate_flag & IEEE80211_TX_RC_SHORT_GI))
 665                 tcb_desc->use_shortgi = false;
 666 }
 667
 668 static void _rtl_query_protection_mode(struct ieee80211_hw *hw,
 669                                        struct rtl_tcb_desc *tcb_desc,
 670                                        struct ieee80211_tx_info *info)
 671 {
 672         struct rtl_priv *rtlpriv = rtl_priv(hw);
 673         u8 rate_flag = info->control.rates[0].flags;
 674
 675         /* Common Settings */
 676         tcb_desc->rts_stbc = false;
 677         tcb_desc->cts_enable = false;
 678         tcb_desc->rts_sc = 0;
 679         tcb_desc->rts_bw = false;
 680         tcb_desc->rts_use_shortpreamble = false;
 681         tcb_desc->rts_use_shortgi = false;
 682
 683         if (rate_flag & IEEE80211_TX_RC_USE_CTS_PROTECT) {
 684                 /* Use CTS-to-SELF in protection mode. */
 685                 tcb_desc->rts_enable = true;
 686                 tcb_desc->cts_enable = true;
 687                 tcb_desc->rts_rate = rtlpriv->cfg->maps[RTL_RC_OFDM_RATE24M];
 688         } else if (rate_flag & IEEE80211_TX_RC_USE_RTS_CTS) {
 689                 /* Use RTS-CTS in protection mode. */
 690                 tcb_desc->rts_enable = true;
 691                 tcb_desc->rts_rate = rtlpriv->cfg->maps[RTL_RC_OFDM_RATE24M];
 692         }
 693 }
 694
 695 u8 rtl_mrate_idx_to_arfr_id(struct ieee80211_hw *hw, u8 rate_index,
 696                             enum wireless_mode wirelessmode)
 697 {
 698         struct rtl_priv *rtlpriv = rtl_priv(hw);
 699         struct rtl_phy *rtlphy = &rtlpriv->phy;
 700         u8 ret = 0;
 701
 702         switch (rate_index) {
 703         case RATR_INX_WIRELESS_NGB:
 704                 if (rtlphy->rf_type == RF_1T1R)
 705                         ret = RATEID_IDX_BGN_40M_1SS;
 706                 else
 707                         ret = RATEID_IDX_BGN_40M_2SS;
 708                 ; break;
 709         case RATR_INX_WIRELESS_N:
 710         case RATR_INX_WIRELESS_NG:
 711                 if (rtlphy->rf_type == RF_1T1R)
 712                         ret = RATEID_IDX_GN_N1SS;
 713                 else
 714                         ret = RATEID_IDX_GN_N2SS;
 715                 ; break;
 716         case RATR_INX_WIRELESS_NB:
 717                 if (rtlphy->rf_type == RF_1T1R)
 718                         ret = RATEID_IDX_BGN_20M_1SS_BN;
 719                 else
 720                         ret = RATEID_IDX_BGN_20M_2SS_BN;
 721                 ; break;
 722         case RATR_INX_WIRELESS_GB:
 723                 ret = RATEID_IDX_BG;
 724                 break;
 725         case RATR_INX_WIRELESS_G:
 726                 ret = RATEID_IDX_G;
 727                 break;
 728         case RATR_INX_WIRELESS_B:
 729                 ret = RATEID_IDX_B;
 730                 break;
 731         case RATR_INX_WIRELESS_MC:
 732                 if (wirelessmode == WIRELESS_MODE_B ||
 733                     wirelessmode == WIRELESS_MODE_G ||
 734                     wirelessmode == WIRELESS_MODE_N_24G ||
 735                     wirelessmode == WIRELESS_MODE_AC_24G)
 736                         ret = RATEID_IDX_BG;
 737                 else
 738                         ret = RATEID_IDX_G;
 739                 break;
 740         case RATR_INX_WIRELESS_AC_5N:
 741                 if (rtlphy->rf_type == RF_1T1R)
 742                         ret = RATEID_IDX_VHT_1SS;
 743                 else
 744                         ret = RATEID_IDX_VHT_2SS;
 745                 break;
 746         case RATR_INX_WIRELESS_AC_24N:
 747                 if (rtlphy->current_chan_bw == HT_CHANNEL_WIDTH_80) {
 748                         if (rtlphy->rf_type == RF_1T1R)
 749                                 ret = RATEID_IDX_VHT_1SS;
 750                         else
 751                                 ret = RATEID_IDX_VHT_2SS;
 752                 } else {
 753                         if (rtlphy->rf_type == RF_1T1R)
 754                                 ret = RATEID_IDX_MIX1;
 755                         else
 756                                 ret = RATEID_IDX_MIX2;
 757                 }
 758                 break;
 759         default:
 760                 ret = RATEID_IDX_BGN_40M_2SS;
 761                 break;
 762         }
 763         return ret;
 764 }
 765 EXPORT_SYMBOL(rtl_mrate_idx_to_arfr_id);
 766
 767 static void _rtl_txrate_selectmode(struct ieee80211_hw *hw,
 768                                    struct ieee80211_sta *sta,
 769                                    struct rtl_tcb_desc *tcb_desc)
 770 {
 771 #define SET_RATE_ID(rate_id)                                    \
 772         ({typeof(rate_id) _id = rate_id;                        \
 773           ((rtlpriv->cfg->spec_ver & RTL_SPEC_NEW_RATEID) ?     \
 774                 rtl_mrate_idx_to_arfr_id(hw, _id,               \
 775                         (sta_entry ? sta_entry->wireless_mode : \
 776                          WIRELESS_MODE_G)) :                    \
 777                 _id); })
 778
 779         struct rtl_priv *rtlpriv = rtl_priv(hw);
 780         struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
 781         struct rtl_sta_info *sta_entry = NULL;
 782         u8 ratr_index = SET_RATE_ID(RATR_INX_WIRELESS_MC);
 783
 784         if (sta) {
 785                 sta_entry = (struct rtl_sta_info *) sta->drv_priv;
 786                 ratr_index = sta_entry->ratr_index;
 787         }
 788         if (!tcb_desc->disable_ratefallback || !tcb_desc->use_driver_rate) {
 789                 if (mac->opmode == NL80211_IFTYPE_STATION) {
 790                         tcb_desc->ratr_index = 0;
 791                 } else if (mac->opmode == NL80211_IFTYPE_ADHOC ||
 792                                 mac->opmode == NL80211_IFTYPE_MESH_POINT) {
 793                         if (tcb_desc->multicast || tcb_desc->broadcast) {
 794                                 tcb_desc->hw_rate =
 795                                     rtlpriv->cfg->maps[RTL_RC_CCK_RATE2M];
 796                                 tcb_desc->use_driver_rate = 1;
 797                                 tcb_desc->ratr_index =
 798                                         SET_RATE_ID(RATR_INX_WIRELESS_MC);
 799                         } else {
 800                                 tcb_desc->ratr_index = ratr_index;
 801                         }
 802                 } else if (mac->opmode == NL80211_IFTYPE_AP) {
 803                         tcb_desc->ratr_index = ratr_index;
 804                 }
 805         }
 806
 807         if (rtlpriv->dm.useramask) {
 808                 tcb_desc->ratr_index = ratr_index;
 809                 /* TODO we will differentiate adhoc and station future  */
 810                 if (mac->opmode == NL80211_IFTYPE_STATION ||
 811                     mac->opmode == NL80211_IFTYPE_MESH_POINT) {
 812                         tcb_desc->mac_id = 0;
 813
 814                         if (sta &&
 815                             (rtlpriv->cfg->spec_ver & RTL_SPEC_NEW_RATEID))
 816                                 ;       /* use sta_entry->ratr_index */
 817                         else if (mac->mode == WIRELESS_MODE_AC_5G)
 818                                 tcb_desc->ratr_index =
 819                                         SET_RATE_ID(RATR_INX_WIRELESS_AC_5N);
 820                         else if (mac->mode == WIRELESS_MODE_AC_24G)
 821                                 tcb_desc->ratr_index =
 822                                         SET_RATE_ID(RATR_INX_WIRELESS_AC_24N);
 823                         else if (mac->mode == WIRELESS_MODE_N_24G)
 824                                 tcb_desc->ratr_index =
 825                                         SET_RATE_ID(RATR_INX_WIRELESS_NGB);
 826                         else if (mac->mode == WIRELESS_MODE_N_5G)
 827                                 tcb_desc->ratr_index =
 828                                         SET_RATE_ID(RATR_INX_WIRELESS_NG);
 829                         else if (mac->mode & WIRELESS_MODE_G)
 830                                 tcb_desc->ratr_index =
 831                                         SET_RATE_ID(RATR_INX_WIRELESS_GB);
 832                         else if (mac->mode & WIRELESS_MODE_B)
 833                                 tcb_desc->ratr_index =
 834                                         SET_RATE_ID(RATR_INX_WIRELESS_B);
 835                         else if (mac->mode & WIRELESS_MODE_A)
 836                                 tcb_desc->ratr_index =
 837                                         SET_RATE_ID(RATR_INX_WIRELESS_G);
 838
 839                 } else if (mac->opmode == NL80211_IFTYPE_AP ||
 840                         mac->opmode == NL80211_IFTYPE_ADHOC) {
 841                         if (NULL != sta) {
 842                                 if (sta->aid > 0)
 843                                         tcb_desc->mac_id = sta->aid + 1;
 844                                 else
 845                                         tcb_desc->mac_id = 1;
 846                         } else {
 847                                 tcb_desc->mac_id = 0;
 848                         }
 849                 }
 850         }
 851 #undef SET_RATE_ID
 852 }
 853
 854 static void _rtl_query_bandwidth_mode(struct ieee80211_hw *hw,
 855                                       struct ieee80211_sta *sta,
 856                                       struct rtl_tcb_desc *tcb_desc)
 857 {
 858         struct rtl_priv *rtlpriv = rtl_priv(hw);
 859         struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
 860
 861         tcb_desc->packet_bw = false;
 862         if (!sta)
 863                 return;
 864         if (mac->opmode == NL80211_IFTYPE_AP ||
 865             mac->opmode == NL80211_IFTYPE_ADHOC ||
 866             mac->opmode == NL80211_IFTYPE_MESH_POINT) {
 867                 if (!(sta->ht_cap.ht_supported) ||
 868                     !(sta->ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40))
 869                         return;
 870         } else if (mac->opmode == NL80211_IFTYPE_STATION) {
 871                 if (!mac->bw_40 || !(sta->ht_cap.ht_supported))
 872                         return;
 873         }
 874         if (tcb_desc->multicast || tcb_desc->broadcast)
 875                 return;
 876
 877         /*use legency rate, shall use 20MHz */
 878         if (tcb_desc->hw_rate <= rtlpriv->cfg->maps[RTL_RC_OFDM_RATE54M])
 879                 return;
 880
 881         tcb_desc->packet_bw = HT_CHANNEL_WIDTH_20_40;
 882
 883         if (rtlpriv->cfg->spec_ver & RTL_SPEC_SUPPORT_VHT) {
 884                 if (mac->opmode == NL80211_IFTYPE_AP ||
 885                     mac->opmode == NL80211_IFTYPE_ADHOC ||
 886                     mac->opmode == NL80211_IFTYPE_MESH_POINT) {
 887                         if (!(sta->vht_cap.vht_supported))
 888                                 return;
 889                 } else if (mac->opmode == NL80211_IFTYPE_STATION) {
 890                         if (!mac->bw_80 ||
 891                             !(sta->vht_cap.vht_supported))
 892                                 return;
 893                 }
 894                 if (tcb_desc->hw_rate <=
 895                         rtlpriv->cfg->maps[RTL_RC_HT_RATEMCS15])
 896                         return;
 897                 tcb_desc->packet_bw = HT_CHANNEL_WIDTH_80;
 898         }
 899 }
 900
 901 static u8 _rtl_get_vht_highest_n_rate(struct ieee80211_hw *hw,
 902                                       struct ieee80211_sta *sta)
 903 {
 904         struct rtl_priv *rtlpriv = rtl_priv(hw);
 905         struct rtl_phy *rtlphy = &(rtlpriv->phy);
 906         u8 hw_rate;
 907         u16 tx_mcs_map = le16_to_cpu(sta->vht_cap.vht_mcs.tx_mcs_map);
 908
 909         if ((get_rf_type(rtlphy) == RF_2T2R) &&
 910             (tx_mcs_map & 0x000c) != 0x000c) {
 911                 if ((tx_mcs_map & 0x000c) >> 2 ==
 912                         IEEE80211_VHT_MCS_SUPPORT_0_7)
 913                         hw_rate =
 914                         rtlpriv->cfg->maps[RTL_RC_VHT_RATE_2SS_MCS7];
 915                 else if ((tx_mcs_map  & 0x000c) >> 2 ==
 916                         IEEE80211_VHT_MCS_SUPPORT_0_8)
 917                         hw_rate =
 918                         rtlpriv->cfg->maps[RTL_RC_VHT_RATE_2SS_MCS8];
 919                 else
 920                         hw_rate =
 921                         rtlpriv->cfg->maps[RTL_RC_VHT_RATE_2SS_MCS9];
 922         } else {
 923                 if ((tx_mcs_map  & 0x0003) ==
 924                         IEEE80211_VHT_MCS_SUPPORT_0_7)
 925                         hw_rate =
 926                         rtlpriv->cfg->maps[RTL_RC_VHT_RATE_1SS_MCS7];
 927                 else if ((tx_mcs_map  & 0x0003) ==
 928                         IEEE80211_VHT_MCS_SUPPORT_0_8)
 929                         hw_rate =
 930                         rtlpriv->cfg->maps[RTL_RC_VHT_RATE_1SS_MCS8];
 931                 else
 932                         hw_rate =
 933                         rtlpriv->cfg->maps[RTL_RC_VHT_RATE_1SS_MCS9];
 934         }
 935
 936         return hw_rate;
 937 }
 938
 939 static u8 _rtl_get_highest_n_rate(struct ieee80211_hw *hw,
 940                                   struct ieee80211_sta *sta)
 941 {
 942         struct rtl_priv *rtlpriv = rtl_priv(hw);
 943         struct rtl_phy *rtlphy = &rtlpriv->phy;
 944         u8 hw_rate;
 945
 946         if (get_rf_type(rtlphy) == RF_2T2R &&
 947             sta->ht_cap.mcs.rx_mask[1] != 0)
 948                 hw_rate = rtlpriv->cfg->maps[RTL_RC_HT_RATEMCS15];
 949         else
 950                 hw_rate = rtlpriv->cfg->maps[RTL_RC_HT_RATEMCS7];
 951
 952         return hw_rate;
 953 }
 954
 955 /* mac80211's rate_idx is like this:
 956  *
 957  * 2.4G band:rx_status->band == NL80211_BAND_2GHZ
 958  *
 959  * B/G rate:
 960  * (rx_status->flag & RX_FLAG_HT) = 0,
 961  * DESC_RATE1M-->DESC_RATE54M ==> idx is 0-->11,
 962  *
 963  * N rate:
 964  * (rx_status->flag & RX_FLAG_HT) = 1,
 965  * DESC_RATEMCS0-->DESC_RATEMCS15 ==> idx is 0-->15
 966  *
 967  * 5G band:rx_status->band == NL80211_BAND_5GHZ
 968  * A rate:
 969  * (rx_status->flag & RX_FLAG_HT) = 0,
 970  * DESC_RATE6M-->DESC_RATE54M ==> idx is 0-->7,
 971  *
 972  * N rate:
 973  * (rx_status->flag & RX_FLAG_HT) = 1,
 974  * DESC_RATEMCS0-->DESC_RATEMCS15 ==> idx is 0-->15
 975  *
 976  * VHT rates:
 977  * DESC_RATEVHT1SS_MCS0-->DESC_RATEVHT1SS_MCS9 ==> idx is 0-->9
 978  * DESC_RATEVHT2SS_MCS0-->DESC_RATEVHT2SS_MCS9 ==> idx is 0-->9
 979  */
 980 int rtlwifi_rate_mapping(struct ieee80211_hw *hw, bool isht, bool isvht,
 981                          u8 desc_rate)
 982 {
 983         int rate_idx;
 984
 985         if (isvht) {
 986                 switch (desc_rate) {
 987                 case DESC_RATEVHT1SS_MCS0:
 988                         rate_idx = 0;
 989                         break;
 990                 case DESC_RATEVHT1SS_MCS1:
 991                         rate_idx = 1;
 992                         break;
 993                 case DESC_RATEVHT1SS_MCS2:
 994                         rate_idx = 2;
 995                         break;
 996                 case DESC_RATEVHT1SS_MCS3:
 997                         rate_idx = 3;
 998                         break;
 999                 case DESC_RATEVHT1SS_MCS4:
1000                         rate_idx = 4;
1001                         break;
1002                 case DESC_RATEVHT1SS_MCS5:
1003                         rate_idx = 5;
1004                         break;
1005                 case DESC_RATEVHT1SS_MCS6:
1006                         rate_idx = 6;
1007                         break;
1008                 case DESC_RATEVHT1SS_MCS7:
1009                         rate_idx = 7;
1010                         break;
1011                 case DESC_RATEVHT1SS_MCS8:
1012                         rate_idx = 8;
1013                         break;
1014                 case DESC_RATEVHT1SS_MCS9:
1015                         rate_idx = 9;
1016                         break;
1017                 case DESC_RATEVHT2SS_MCS0:
1018                         rate_idx = 0;
1019                         break;
1020                 case DESC_RATEVHT2SS_MCS1:
1021                         rate_idx = 1;
1022                         break;
1023                 case DESC_RATEVHT2SS_MCS2:
1024                         rate_idx = 2;
1025                         break;
1026                 case DESC_RATEVHT2SS_MCS3:
1027                         rate_idx = 3;
1028                         break;
1029                 case DESC_RATEVHT2SS_MCS4:
1030                         rate_idx = 4;
1031                         break;
1032                 case DESC_RATEVHT2SS_MCS5:
1033                         rate_idx = 5;
1034                         break;
1035                 case DESC_RATEVHT2SS_MCS6:
1036                         rate_idx = 6;
1037                         break;
1038                 case DESC_RATEVHT2SS_MCS7:
1039                         rate_idx = 7;
1040                         break;
1041                 case DESC_RATEVHT2SS_MCS8:
1042                         rate_idx = 8;
1043                         break;
1044                 case DESC_RATEVHT2SS_MCS9:
1045                         rate_idx = 9;
1046                         break;
1047                 default:
1048                         rate_idx = 0;
1049                         break;
1050                 }
1051                 return rate_idx;
1052         }
1053         if (false == isht) {
1054                 if (NL80211_BAND_2GHZ == hw->conf.chandef.chan->band) {
1055                         switch (desc_rate) {
1056                         case DESC_RATE1M:
1057                                 rate_idx = 0;
1058                                 break;
1059                         case DESC_RATE2M:
1060                                 rate_idx = 1;
1061                                 break;
1062                         case DESC_RATE5_5M:
1063                                 rate_idx = 2;
1064                                 break;
1065                         case DESC_RATE11M:
1066                                 rate_idx = 3;
1067                                 break;
1068                         case DESC_RATE6M:
1069                                 rate_idx = 4;
1070                                 break;
1071                         case DESC_RATE9M:
1072                                 rate_idx = 5;
1073                                 break;
1074                         case DESC_RATE12M:
1075                                 rate_idx = 6;
1076                                 break;
1077                         case DESC_RATE18M:
1078                                 rate_idx = 7;
1079                                 break;
1080                         case DESC_RATE24M:
1081                                 rate_idx = 8;
1082                                 break;
1083                         case DESC_RATE36M:
1084                                 rate_idx = 9;
1085                                 break;
1086                         case DESC_RATE48M:
1087                                 rate_idx = 10;
1088                                 break;
1089                         case DESC_RATE54M:
1090                                 rate_idx = 11;
1091                                 break;
1092                         default:
1093                                 rate_idx = 0;
1094                                 break;
1095                         }
1096                 } else {
1097                         switch (desc_rate) {
1098                         case DESC_RATE6M:
1099                                 rate_idx = 0;
1100                                 break;
1101                         case DESC_RATE9M:
1102                                 rate_idx = 1;
1103                                 break;
1104                         case DESC_RATE12M:
1105                                 rate_idx = 2;
1106                                 break;
1107                         case DESC_RATE18M:
1108                                 rate_idx = 3;
1109                                 break;
1110                         case DESC_RATE24M:
1111                                 rate_idx = 4;
1112                                 break;
1113                         case DESC_RATE36M:
1114                                 rate_idx = 5;
1115                                 break;
1116                         case DESC_RATE48M:
1117                                 rate_idx = 6;
1118                                 break;
1119                         case DESC_RATE54M:
1120                                 rate_idx = 7;
1121                                 break;
1122                         default:
1123                                 rate_idx = 0;
1124                                 break;
1125                         }
1126                 }
1127         } else {
1128                 switch (desc_rate) {
1129                 case DESC_RATEMCS0:
1130                         rate_idx = 0;
1131                         break;
1132                 case DESC_RATEMCS1:
1133                         rate_idx = 1;
1134                         break;
1135                 case DESC_RATEMCS2:
1136                         rate_idx = 2;
1137                         break;
1138                 case DESC_RATEMCS3:
1139                         rate_idx = 3;
1140                         break;
1141                 case DESC_RATEMCS4:
1142                         rate_idx = 4;
1143                         break;
1144                 case DESC_RATEMCS5:
1145                         rate_idx = 5;
1146                         break;
1147                 case DESC_RATEMCS6:
1148                         rate_idx = 6;
1149                         break;
1150                 case DESC_RATEMCS7:
1151                         rate_idx = 7;
1152                         break;
1153                 case DESC_RATEMCS8:
1154                         rate_idx = 8;
1155                         break;
1156                 case DESC_RATEMCS9:
1157                         rate_idx = 9;
1158                         break;
1159                 case DESC_RATEMCS10:
1160                         rate_idx = 10;
1161                         break;
1162                 case DESC_RATEMCS11:
1163                         rate_idx = 11;
1164                         break;
1165                 case DESC_RATEMCS12:
1166                         rate_idx = 12;
1167                         break;
1168                 case DESC_RATEMCS13:
1169                         rate_idx = 13;
1170                         break;
1171                 case DESC_RATEMCS14:
1172                         rate_idx = 14;
1173                         break;
1174                 case DESC_RATEMCS15:
1175                         rate_idx = 15;
1176                         break;
1177                 default:
1178                         rate_idx = 0;
1179                         break;
1180                 }
1181         }
1182         return rate_idx;
1183 }
1184 EXPORT_SYMBOL(rtlwifi_rate_mapping);
1185
1186 static u8 _rtl_get_tx_hw_rate(struct ieee80211_hw *hw,
1187                               struct ieee80211_tx_info *info)
1188 {
1189         struct rtl_priv *rtlpriv = rtl_priv(hw);
1190         struct ieee80211_tx_rate *r = &info->status.rates[0];
1191         struct ieee80211_rate *txrate;
1192         u8 hw_value = 0x0;
1193
1194         if (r->flags & IEEE80211_TX_RC_MCS) {
1195                 /* HT MCS0-15 */
1196                 hw_value = rtlpriv->cfg->maps[RTL_RC_HT_RATEMCS15] - 15 +
1197                            r->idx;
1198         } else if (r->flags & IEEE80211_TX_RC_VHT_MCS) {
1199                 /* VHT MCS0-9, NSS */
1200                 if (ieee80211_rate_get_vht_nss(r) == 2)
1201                         hw_value = rtlpriv->cfg->maps[RTL_RC_VHT_RATE_2SS_MCS9];
1202                 else
1203                         hw_value = rtlpriv->cfg->maps[RTL_RC_VHT_RATE_1SS_MCS9];
1204
1205                 hw_value = hw_value - 9 + ieee80211_rate_get_vht_mcs(r);
1206         } else {
1207                 /* legacy */
1208                 txrate = ieee80211_get_tx_rate(hw, info);
1209
1210                 if (txrate)
1211                         hw_value = txrate->hw_value;
1212         }
1213
1214         /* check 5G band */
1215         if (rtlpriv->rtlhal.current_bandtype == BAND_ON_5G &&
1216             hw_value < rtlpriv->cfg->maps[RTL_RC_OFDM_RATE6M])
1217                 hw_value = rtlpriv->cfg->maps[RTL_RC_OFDM_RATE6M];
1218
1219         return hw_value;
1220 }
1221
1222 void rtl_get_tcb_desc(struct ieee80211_hw *hw,
1223                       struct ieee80211_tx_info *info,
1224                       struct ieee80211_sta *sta,
1225                       struct sk_buff *skb, struct rtl_tcb_desc *tcb_desc)
1226 {
1227 #define SET_RATE_ID(rate_id)                                    \
1228         ({typeof(rate_id) _id = rate_id;                        \
1229           ((rtlpriv->cfg->spec_ver & RTL_SPEC_NEW_RATEID) ?     \
1230                 rtl_mrate_idx_to_arfr_id(hw, _id,               \
1231                         (sta_entry ? sta_entry->wireless_mode : \
1232                          WIRELESS_MODE_G)) :                    \
1233                 _id); })
1234
1235         struct rtl_priv *rtlpriv = rtl_priv(hw);
1236         struct rtl_mac *rtlmac = rtl_mac(rtl_priv(hw));
1237         struct ieee80211_hdr *hdr = rtl_get_hdr(skb);
1238         struct rtl_sta_info *sta_entry =
1239                 (sta ? (struct rtl_sta_info *)sta->drv_priv : NULL);
1240
1241         __le16 fc = rtl_get_fc(skb);
1242
1243         tcb_desc->hw_rate = _rtl_get_tx_hw_rate(hw, info);
1244
1245         if (rtl_is_tx_report_skb(hw, skb))
1246                 tcb_desc->use_spe_rpt = 1;
1247
1248         if (ieee80211_is_data(fc)) {
1249                 /*
1250                  *we set data rate INX 0
1251                  *in rtl_rc.c   if skb is special data or
1252                  *mgt which need low data rate.
1253                  */
1254
1255                 /*
1256                  *So tcb_desc->hw_rate is just used for
1257                  *special data and mgt frames
1258                  */
1259                 if (info->control.rates[0].idx == 0 ||
1260                                 ieee80211_is_nullfunc(fc)) {
1261                         tcb_desc->use_driver_rate = true;
1262                         tcb_desc->ratr_index =
1263                                         SET_RATE_ID(RATR_INX_WIRELESS_MC);
1264
1265                         tcb_desc->disable_ratefallback = 1;
1266                 } else {
1267                         /*
1268                          *because hw will nerver use hw_rate
1269                          *when tcb_desc->use_driver_rate = false
1270                          *so we never set highest N rate here,
1271                          *and N rate will all be controlled by FW
1272                          *when tcb_desc->use_driver_rate = false
1273                          */
1274                         if (sta && sta->vht_cap.vht_supported) {
1275                                 tcb_desc->hw_rate =
1276                                 _rtl_get_vht_highest_n_rate(hw, sta);
1277                         } else {
1278                                 if (sta && sta->ht_cap.ht_supported) {
1279                                         tcb_desc->hw_rate =
1280                                                 _rtl_get_highest_n_rate(hw, sta);
1281                                 } else {
1282                                         if (rtlmac->mode == WIRELESS_MODE_B) {
1283                                                 tcb_desc->hw_rate =
1284                                                     rtlpriv->cfg->maps[RTL_RC_CCK_RATE11M];
1285                                         } else {
1286                                                 tcb_desc->hw_rate =
1287                                                     rtlpriv->cfg->maps[RTL_RC_OFDM_RATE54M];
1288                                         }
1289                                 }
1290                         }
1291                 }
1292
1293                 if (is_multicast_ether_addr(hdr->addr1))
1294                         tcb_desc->multicast = 1;
1295                 else if (is_broadcast_ether_addr(hdr->addr1))
1296                         tcb_desc->broadcast = 1;
1297
1298                 _rtl_txrate_selectmode(hw, sta, tcb_desc);
1299                 _rtl_query_bandwidth_mode(hw, sta, tcb_desc);
1300                 _rtl_qurey_shortpreamble_mode(hw, tcb_desc, info);
1301                 _rtl_query_shortgi(hw, sta, tcb_desc, info);
1302                 _rtl_query_protection_mode(hw, tcb_desc, info);
1303         } else {
1304                 tcb_desc->use_driver_rate = true;
1305                 tcb_desc->ratr_index = SET_RATE_ID(RATR_INX_WIRELESS_MC);
1306                 tcb_desc->disable_ratefallback = 1;
1307                 tcb_desc->mac_id = 0;
1308                 tcb_desc->packet_bw = false;
1309         }
1310 #undef SET_RATE_ID
1311 }
1312 EXPORT_SYMBOL(rtl_get_tcb_desc);
1313
1314 bool rtl_tx_mgmt_proc(struct ieee80211_hw *hw, struct sk_buff *skb)
1315 {
1316         struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
1317         struct rtl_priv *rtlpriv = rtl_priv(hw);
1318         __le16 fc = rtl_get_fc(skb);
1319
1320         if (rtlpriv->dm.supp_phymode_switch &&
1321             mac->link_state < MAC80211_LINKED &&
1322             (ieee80211_is_auth(fc) || ieee80211_is_probe_req(fc))) {
1323                 if (rtlpriv->cfg->ops->chk_switch_dmdp)
1324                         rtlpriv->cfg->ops->chk_switch_dmdp(hw);
1325         }
1326         if (ieee80211_is_auth(fc)) {
1327                 RT_TRACE(rtlpriv, COMP_SEND, DBG_DMESG, "MAC80211_LINKING\n");
1328
1329                 mac->link_state = MAC80211_LINKING;
1330                 /* Dul mac */
1331                 rtlpriv->phy.need_iqk = true;
1332
1333         }
1334
1335         return true;
1336 }
1337 EXPORT_SYMBOL_GPL(rtl_tx_mgmt_proc);
1338
1339 struct sk_buff *rtl_make_del_ba(struct ieee80211_hw *hw, u8 *sa,
1340                                 u8 *bssid, u16 tid);
1341
1342 static void process_agg_start(struct ieee80211_hw *hw,
1343                               struct ieee80211_hdr *hdr, u16 tid)
1344 {
1345         struct rtl_priv *rtlpriv = rtl_priv(hw);
1346         struct ieee80211_rx_status rx_status = { 0 };
1347         struct sk_buff *skb_delba = NULL;
1348
1349         skb_delba = rtl_make_del_ba(hw, hdr->addr2, hdr->addr3, tid);
1350         if (skb_delba) {
1351                 rx_status.freq = hw->conf.chandef.chan->center_freq;
1352                 rx_status.band = hw->conf.chandef.chan->band;
1353                 rx_status.flag |= RX_FLAG_DECRYPTED;
1354                 rx_status.flag |= RX_FLAG_MACTIME_START;
1355                 rx_status.rate_idx = 0;
1356                 rx_status.signal = 50 + 10;
1357                 memcpy(IEEE80211_SKB_RXCB(skb_delba),
1358                        &rx_status, sizeof(rx_status));
1359                 RT_PRINT_DATA(rtlpriv, COMP_INIT, DBG_DMESG,
1360                               "fake del\n",
1361                               skb_delba->data,
1362                               skb_delba->len);
1363                 ieee80211_rx_irqsafe(hw, skb_delba);
1364         }
1365 }
1366
1367 bool rtl_action_proc(struct ieee80211_hw *hw, struct sk_buff *skb, u8 is_tx)
1368 {
1369         struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
1370         struct ieee80211_hdr *hdr = rtl_get_hdr(skb);
1371         struct rtl_priv *rtlpriv = rtl_priv(hw);
1372         __le16 fc = rtl_get_fc(skb);
1373         u8 *act = (u8 *)(((u8 *)skb->data + MAC80211_3ADDR_LEN));
1374         u8 category;
1375
1376         if (!ieee80211_is_action(fc))
1377                 return true;
1378
1379         category = *act;
1380         act++;
1381         switch (category) {
1382         case ACT_CAT_BA:
1383                 switch (*act) {
1384                 case ACT_ADDBAREQ:
1385                         if (mac->act_scanning)
1386                                 return false;
1387
1388                         RT_TRACE(rtlpriv, (COMP_SEND | COMP_RECV), DBG_DMESG,
1389                                 "%s ACT_ADDBAREQ From :%pM\n",
1390                                 is_tx ? "Tx" : "Rx", hdr->addr2);
1391                         RT_PRINT_DATA(rtlpriv, COMP_INIT, DBG_DMESG, "req\n",
1392                                 skb->data, skb->len);
1393                         if (!is_tx) {
1394                                 struct ieee80211_sta *sta = NULL;
1395                                 struct rtl_sta_info *sta_entry = NULL;
1396                                 struct rtl_tid_data *tid_data;
1397                                 struct ieee80211_mgmt *mgmt = (void *)skb->data;
1398                                 u16 capab = 0, tid = 0;
1399
1400                                 rcu_read_lock();
1401                                 sta = rtl_find_sta(hw, hdr->addr3);
1402                                 if (sta == NULL) {
1403                                         RT_TRACE(rtlpriv, COMP_SEND | COMP_RECV,
1404                                                  DBG_DMESG, "sta is NULL\n");
1405                                         rcu_read_unlock();
1406                                         return true;
1407                                 }
1408
1409                                 sta_entry =
1410                                         (struct rtl_sta_info *)sta->drv_priv;
1411                                 if (!sta_entry) {
1412                                         rcu_read_unlock();
1413                                         return true;
1414                                 }
1415                                 capab =
1416                                   le16_to_cpu(mgmt->u.action.u.addba_req.capab);
1417                                 tid = (capab &
1418                                        IEEE80211_ADDBA_PARAM_TID_MASK) >> 2;
1419                                 if (tid >= MAX_TID_COUNT) {
1420                                         rcu_read_unlock();
1421                                         return true;
1422                                 }
1423                                 tid_data = &sta_entry->tids[tid];
1424                                 if (tid_data->agg.rx_agg_state ==
1425                                     RTL_RX_AGG_START)
1426                                         process_agg_start(hw, hdr, tid);
1427                                 rcu_read_unlock();
1428                         }
1429                         break;
1430                 case ACT_ADDBARSP:
1431                         RT_TRACE(rtlpriv, (COMP_SEND | COMP_RECV), DBG_DMESG,
1432                                  "%s ACT_ADDBARSP From :%pM\n",
1433                                   is_tx ? "Tx" : "Rx", hdr->addr2);
1434                         break;
1435                 case ACT_DELBA:
1436                         RT_TRACE(rtlpriv, (COMP_SEND | COMP_RECV), DBG_DMESG,
1437                                  "ACT_ADDBADEL From :%pM\n", hdr->addr2);
1438                         break;
1439                 }
1440                 break;
1441         default:
1442                 break;
1443         }
1444
1445         return true;
1446 }
1447 EXPORT_SYMBOL_GPL(rtl_action_proc);
1448
1449 static void setup_special_tx(struct rtl_priv *rtlpriv, struct rtl_ps_ctl *ppsc,
1450                              int type)
1451 {
1452         struct ieee80211_hw *hw = rtlpriv->hw;
1453
1454         rtlpriv->ra.is_special_data = true;
1455         if (rtlpriv->cfg->ops->get_btc_status())
1456                 rtlpriv->btcoexist.btc_ops->btc_special_packet_notify(
1457                                         rtlpriv, type);
1458         rtl_lps_leave(hw);
1459         ppsc->last_delaylps_stamp_jiffies = jiffies;
1460 }
1461
1462 static const u8 *rtl_skb_ether_type_ptr(struct ieee80211_hw *hw,
1463                                         struct sk_buff *skb, bool is_enc)
1464 {
1465         struct rtl_priv *rtlpriv = rtl_priv(hw);
1466         u8 mac_hdr_len = ieee80211_get_hdrlen_from_skb(skb);
1467         u8 encrypt_header_len = 0;
1468         u8 offset;
1469
1470         switch (rtlpriv->sec.pairwise_enc_algorithm) {
1471         case WEP40_ENCRYPTION:
1472         case WEP104_ENCRYPTION:
1473                 encrypt_header_len = 4;/*WEP_IV_LEN*/
1474                 break;
1475         case TKIP_ENCRYPTION:
1476                 encrypt_header_len = 8;/*TKIP_IV_LEN*/
1477                 break;
1478         case AESCCMP_ENCRYPTION:
1479                 encrypt_header_len = 8;/*CCMP_HDR_LEN;*/
1480                 break;
1481         default:
1482                 break;
1483         }
1484
1485         offset = mac_hdr_len + SNAP_SIZE;
1486         if (is_enc)
1487                 offset += encrypt_header_len;
1488
1489         return skb->data + offset;
1490 }
1491
1492 /*should call before software enc*/
1493 u8 rtl_is_special_data(struct ieee80211_hw *hw, struct sk_buff *skb, u8 is_tx,
1494                        bool is_enc)
1495 {
1496         struct rtl_priv *rtlpriv = rtl_priv(hw);
1497         struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
1498         __le16 fc = rtl_get_fc(skb);
1499         u16 ether_type;
1500         const u8 *ether_type_ptr;
1501         const struct iphdr *ip;
1502
1503         if (!ieee80211_is_data(fc))
1504                 goto end;
1505
1506         ether_type_ptr = rtl_skb_ether_type_ptr(hw, skb, is_enc);
1507         ether_type = be16_to_cpup((__be16 *)ether_type_ptr);
1508
1509         if (ETH_P_IP == ether_type) {
1510                 ip = (struct iphdr *)((u8 *)ether_type_ptr +
1511                      PROTOC_TYPE_SIZE);
1512                 if (IPPROTO_UDP == ip->protocol) {
1513                         struct udphdr *udp = (struct udphdr *)((u8 *)ip +
1514                                                                (ip->ihl << 2));
1515                         if (((((u8 *)udp)[1] == 68) &&
1516                              (((u8 *)udp)[3] == 67)) ||
1517                             ((((u8 *)udp)[1] == 67) &&
1518                              (((u8 *)udp)[3] == 68))) {
1519                                 /* 68 : UDP BOOTP client
1520                                  * 67 : UDP BOOTP server
1521                                  */
1522                                 RT_TRACE(rtlpriv, (COMP_SEND | COMP_RECV),
1523                                          DBG_DMESG, "dhcp %s !!\n",
1524                                          (is_tx) ? "Tx" : "Rx");
1525
1526                                 if (is_tx)
1527                                         setup_special_tx(rtlpriv, ppsc,
1528                                                          PACKET_DHCP);
1529
1530                                 return true;
1531                         }
1532                 }
1533         } else if (ETH_P_ARP == ether_type) {
1534                 if (is_tx)
1535                         setup_special_tx(rtlpriv, ppsc, PACKET_ARP);
1536
1537                 return true;
1538         } else if (ETH_P_PAE == ether_type) {
1539                 /* EAPOL is seens as in-4way */
1540                 rtlpriv->btcoexist.btc_info.in_4way = true;
1541                 rtlpriv->btcoexist.btc_info.in_4way_ts = jiffies;
1542
1543                 RT_TRACE(rtlpriv, (COMP_SEND | COMP_RECV), DBG_DMESG,
1544                          "802.1X %s EAPOL pkt!!\n", (is_tx) ? "Tx" : "Rx");
1545
1546                 if (is_tx) {
1547                         rtlpriv->ra.is_special_data = true;
1548                         rtl_lps_leave(hw);
1549                         ppsc->last_delaylps_stamp_jiffies = jiffies;
1550
1551                         setup_special_tx(rtlpriv, ppsc, PACKET_EAPOL);
1552                 }
1553
1554                 return true;
1555         } else if (ETH_P_IPV6 == ether_type) {
1556                 /* TODO: Handle any IPv6 cases that need special handling.
1557                  * For now, always return false
1558                  */
1559                 goto end;
1560         }
1561
1562 end:
1563         rtlpriv->ra.is_special_data = false;
1564         return false;
1565 }
1566 EXPORT_SYMBOL_GPL(rtl_is_special_data);
1567
1568 void rtl_tx_ackqueue(struct ieee80211_hw *hw, struct sk_buff *skb)
1569 {
1570         struct rtl_priv *rtlpriv = rtl_priv(hw);
1571         struct rtl_tx_report *tx_report = &rtlpriv->tx_report;
1572
1573         __skb_queue_tail(&tx_report->queue, skb);
1574 }
1575 EXPORT_SYMBOL_GPL(rtl_tx_ackqueue);
1576
1577 static void rtl_tx_status(struct ieee80211_hw *hw, struct sk_buff *skb,
1578                           bool ack)
1579 {
1580         struct rtl_priv *rtlpriv = rtl_priv(hw);
1581         struct ieee80211_tx_info *info;
1582
1583         info = IEEE80211_SKB_CB(skb);
1584         ieee80211_tx_info_clear_status(info);
1585         if (ack) {
1586                 RT_TRACE(rtlpriv, COMP_TX_REPORT, DBG_LOUD,
1587                          "tx report: ack\n");
1588                 info->flags |= IEEE80211_TX_STAT_ACK;
1589         } else {
1590                 RT_TRACE(rtlpriv, COMP_TX_REPORT, DBG_LOUD,
1591                          "tx report: not ack\n");
1592                 info->flags &= ~IEEE80211_TX_STAT_ACK;
1593         }
1594         ieee80211_tx_status_irqsafe(hw, skb);
1595 }
1596
1597 bool rtl_is_tx_report_skb(struct ieee80211_hw *hw, struct sk_buff *skb)
1598 {
1599         u16 ether_type;
1600         const u8 *ether_type_ptr;
1601         __le16 fc = rtl_get_fc(skb);
1602
1603         ether_type_ptr = rtl_skb_ether_type_ptr(hw, skb, true);
1604         ether_type = be16_to_cpup((__be16 *)ether_type_ptr);
1605
1606         if (ether_type == ETH_P_PAE || ieee80211_is_nullfunc(fc))
1607                 return true;
1608
1609         return false;
1610 }
1611
1612 static u16 rtl_get_tx_report_sn(struct ieee80211_hw *hw,
1613                                 struct rtlwifi_tx_info *tx_info)
1614 {
1615         struct rtl_priv *rtlpriv = rtl_priv(hw);
1616         struct rtl_tx_report *tx_report = &rtlpriv->tx_report;
1617         u16 sn;
1618
1619         /* SW_DEFINE[11:8] are reserved (driver fills zeros)
1620          * SW_DEFINE[7:2] are used by driver
1621          * SW_DEFINE[1:0] are reserved for firmware (driver fills zeros)
1622          */
1623         sn = (atomic_inc_return(&tx_report->sn) & 0x003F) << 2;
1624
1625         tx_report->last_sent_sn = sn;
1626         tx_report->last_sent_time = jiffies;
1627         tx_info->sn = sn;
1628         tx_info->send_time = tx_report->last_sent_time;
1629         RT_TRACE(rtlpriv, COMP_TX_REPORT, DBG_DMESG,
1630                  "Send TX-Report sn=0x%X\n", sn);
1631
1632         return sn;
1633 }
1634
1635 void rtl_set_tx_report(struct rtl_tcb_desc *ptcb_desc, u8 *pdesc,
1636                        struct ieee80211_hw *hw, struct rtlwifi_tx_info *tx_info)
1637 {
1638         if (ptcb_desc->use_spe_rpt) {
1639                 u16 sn = rtl_get_tx_report_sn(hw, tx_info);
1640
1641                 SET_TX_DESC_SPE_RPT(pdesc, 1);
1642                 SET_TX_DESC_SW_DEFINE(pdesc, sn);
1643         }
1644 }
1645 EXPORT_SYMBOL_GPL(rtl_set_tx_report);
1646
1647 void rtl_tx_report_handler(struct ieee80211_hw *hw, u8 *tmp_buf, u8 c2h_cmd_len)
1648 {
1649         struct rtl_priv *rtlpriv = rtl_priv(hw);
1650         struct rtl_tx_report *tx_report = &rtlpriv->tx_report;
1651         struct rtlwifi_tx_info *tx_info;
1652         struct sk_buff_head *queue = &tx_report->queue;
1653         struct sk_buff *skb;
1654         u16 sn;
1655         u8 st, retry;
1656
1657         if (rtlpriv->cfg->spec_ver & RTL_SPEC_EXT_C2H) {
1658                 sn = GET_TX_REPORT_SN_V2(tmp_buf);
1659                 st = GET_TX_REPORT_ST_V2(tmp_buf);
1660                 retry = GET_TX_REPORT_RETRY_V2(tmp_buf);
1661         } else {
1662                 sn = GET_TX_REPORT_SN_V1(tmp_buf);
1663                 st = GET_TX_REPORT_ST_V1(tmp_buf);
1664                 retry = GET_TX_REPORT_RETRY_V1(tmp_buf);
1665         }
1666
1667         tx_report->last_recv_sn = sn;
1668
1669         skb_queue_walk(queue, skb) {
1670                 tx_info = rtl_tx_skb_cb_info(skb);
1671                 if (tx_info->sn == sn) {
1672                         skb_unlink(skb, queue);
1673                         rtl_tx_status(hw, skb, st == 0);
1674                         break;
1675                 }
1676         }
1677         RT_TRACE(rtlpriv, COMP_TX_REPORT, DBG_DMESG,
1678                  "Recv TX-Report st=0x%02X sn=0x%X retry=0x%X\n",
1679                  st, sn, retry);
1680 }
1681 EXPORT_SYMBOL_GPL(rtl_tx_report_handler);
1682
1683 bool rtl_check_tx_report_acked(struct ieee80211_hw *hw)
1684 {
1685         struct rtl_priv *rtlpriv = rtl_priv(hw);
1686         struct rtl_tx_report *tx_report = &rtlpriv->tx_report;
1687
1688         if (tx_report->last_sent_sn == tx_report->last_recv_sn)
1689                 return true;
1690
1691         if (time_before(tx_report->last_sent_time + 3 * HZ, jiffies)) {
1692                 RT_TRACE(rtlpriv, COMP_TX_REPORT, DBG_WARNING,
1693                          "Check TX-Report timeout!! s_sn=0x%X r_sn=0x%X\n",
1694                          tx_report->last_sent_sn, tx_report->last_recv_sn);
1695                 return true;    /* 3 sec. (timeout) seen as acked */
1696         }
1697
1698         return false;
1699 }
1700
1701 void rtl_wait_tx_report_acked(struct ieee80211_hw *hw, u32 wait_ms)
1702 {
1703         struct rtl_priv *rtlpriv = rtl_priv(hw);
1704         int i;
1705
1706         for (i = 0; i < wait_ms; i++) {
1707                 if (rtl_check_tx_report_acked(hw))
1708                         break;
1709                 usleep_range(1000, 2000);
1710                 RT_TRACE(rtlpriv, COMP_SEC, DBG_DMESG,
1711                          "Wait 1ms (%d/%d) to disable key.\n", i, wait_ms);
1712         }
1713 }
1714
1715 u32 rtl_get_hal_edca_param(struct ieee80211_hw *hw,
1716                            struct ieee80211_vif *vif,
1717                            enum wireless_mode wirelessmode,
1718                            struct ieee80211_tx_queue_params *param)
1719 {
1720         u32 reg = 0;
1721         u8 sifstime = 10;
1722         u8 slottime = 20;
1723
1724         /* AIFS = AIFSN * slot time + SIFS */
1725         switch (wirelessmode) {
1726         case WIRELESS_MODE_A:
1727         case WIRELESS_MODE_N_24G:
1728         case WIRELESS_MODE_N_5G:
1729         case WIRELESS_MODE_AC_5G:
1730         case WIRELESS_MODE_AC_24G:
1731                 sifstime = 16;
1732                 slottime = 9;
1733                 break;
1734         case WIRELESS_MODE_G:
1735                 slottime = (vif->bss_conf.use_short_slot ? 9 : 20);
1736                 break;
1737         default:
1738                 break;
1739         }
1740
1741         reg |= (param->txop & 0x7FF) << 16;
1742         reg |= (fls(param->cw_max) & 0xF) << 12;
1743         reg |= (fls(param->cw_min) & 0xF) << 8;
1744         reg |= (param->aifs & 0x0F) * slottime + sifstime;
1745
1746         return reg;
1747 }
1748 EXPORT_SYMBOL_GPL(rtl_get_hal_edca_param);
1749
1750 /*********************************************************
1751  *
1752  * functions called by core.c
1753  *
1754  *********************************************************/
1755 int rtl_tx_agg_start(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
1756                      struct ieee80211_sta *sta, u16 tid, u16 *ssn)
1757 {
1758         struct rtl_priv *rtlpriv = rtl_priv(hw);
1759         struct rtl_tid_data *tid_data;
1760         struct rtl_sta_info *sta_entry = NULL;
1761
1762         if (sta == NULL)
1763                 return -EINVAL;
1764
1765         if (unlikely(tid >= MAX_TID_COUNT))
1766                 return -EINVAL;
1767
1768         sta_entry = (struct rtl_sta_info *)sta->drv_priv;
1769         if (!sta_entry)
1770                 return -ENXIO;
1771         tid_data = &sta_entry->tids[tid];
1772
1773         RT_TRACE(rtlpriv, COMP_SEND, DBG_DMESG,
1774                  "on ra = %pM tid = %d seq:%d\n", sta->addr, tid,
1775                  *ssn);
1776
1777         tid_data->agg.agg_state = RTL_AGG_START;
1778
1779         return IEEE80211_AMPDU_TX_START_IMMEDIATE;
1780 }
1781
1782 int rtl_tx_agg_stop(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
1783                     struct ieee80211_sta *sta, u16 tid)
1784 {
1785         struct rtl_priv *rtlpriv = rtl_priv(hw);
1786         struct rtl_sta_info *sta_entry = NULL;
1787
1788         if (sta == NULL)
1789                 return -EINVAL;
1790
1791         RT_TRACE(rtlpriv, COMP_SEND, DBG_DMESG,
1792                  "on ra = %pM tid = %d\n", sta->addr, tid);
1793
1794         if (unlikely(tid >= MAX_TID_COUNT))
1795                 return -EINVAL;
1796
1797         sta_entry = (struct rtl_sta_info *)sta->drv_priv;
1798         sta_entry->tids[tid].agg.agg_state = RTL_AGG_STOP;
1799
1800         ieee80211_stop_tx_ba_cb_irqsafe(vif, sta->addr, tid);
1801         return 0;
1802 }
1803
1804 int rtl_rx_agg_start(struct ieee80211_hw *hw,
1805                      struct ieee80211_sta *sta, u16 tid)
1806 {
1807         struct rtl_priv *rtlpriv = rtl_priv(hw);
1808         struct rtl_tid_data *tid_data;
1809         struct rtl_sta_info *sta_entry = NULL;
1810         u8 reject_agg;
1811
1812         if (sta == NULL)
1813                 return -EINVAL;
1814
1815         if (unlikely(tid >= MAX_TID_COUNT))
1816                 return -EINVAL;
1817
1818         if (rtlpriv->cfg->ops->get_btc_status()) {
1819                 rtlpriv->btcoexist.btc_ops->btc_get_ampdu_cfg(rtlpriv,
1820                                                               &reject_agg,
1821                                                               NULL, NULL);
1822                 if (reject_agg)
1823                         return -EINVAL;
1824         }
1825
1826         sta_entry = (struct rtl_sta_info *)sta->drv_priv;
1827         if (!sta_entry)
1828                 return -ENXIO;
1829         tid_data = &sta_entry->tids[tid];
1830
1831         RT_TRACE(rtlpriv, COMP_RECV, DBG_DMESG,
1832                  "on ra = %pM tid = %d\n", sta->addr, tid);
1833
1834         tid_data->agg.rx_agg_state = RTL_RX_AGG_START;
1835         return 0;
1836 }
1837
1838 int rtl_rx_agg_stop(struct ieee80211_hw *hw,
1839                     struct ieee80211_sta *sta, u16 tid)
1840 {
1841         struct rtl_priv *rtlpriv = rtl_priv(hw);
1842         struct rtl_sta_info *sta_entry = NULL;
1843
1844         if (sta == NULL)
1845                 return -EINVAL;
1846
1847         RT_TRACE(rtlpriv, COMP_SEND, DBG_DMESG,
1848                  "on ra = %pM tid = %d\n", sta->addr, tid);
1849
1850         if (unlikely(tid >= MAX_TID_COUNT))
1851                 return -EINVAL;
1852
1853         sta_entry = (struct rtl_sta_info *)sta->drv_priv;
1854         sta_entry->tids[tid].agg.rx_agg_state = RTL_RX_AGG_STOP;
1855
1856         return 0;
1857 }
1858
1859 int rtl_tx_agg_oper(struct ieee80211_hw *hw,
1860                 struct ieee80211_sta *sta, u16 tid)
1861 {
1862         struct rtl_priv *rtlpriv = rtl_priv(hw);
1863         struct rtl_sta_info *sta_entry = NULL;
1864
1865         if (sta == NULL)
1866                 return -EINVAL;
1867
1868         RT_TRACE(rtlpriv, COMP_SEND, DBG_DMESG,
1869                  "on ra = %pM tid = %d\n", sta->addr, tid);
1870
1871         if (unlikely(tid >= MAX_TID_COUNT))
1872                 return -EINVAL;
1873
1874         sta_entry = (struct rtl_sta_info *)sta->drv_priv;
1875         sta_entry->tids[tid].agg.agg_state = RTL_AGG_OPERATIONAL;
1876
1877         return 0;
1878 }
1879
1880 void rtl_rx_ampdu_apply(struct rtl_priv *rtlpriv)
1881 {
1882         struct rtl_btc_ops *btc_ops = rtlpriv->btcoexist.btc_ops;
1883         u8 reject_agg = 0, ctrl_agg_size = 0, agg_size = 0;
1884
1885         if (rtlpriv->cfg->ops->get_btc_status())
1886                 btc_ops->btc_get_ampdu_cfg(rtlpriv, &reject_agg,
1887                                            &ctrl_agg_size, &agg_size);
1888
1889         RT_TRACE(rtlpriv, COMP_BT_COEXIST, DBG_DMESG,
1890                  "Set RX AMPDU: coex - reject=%d, ctrl_agg_size=%d, size=%d",
1891                  reject_agg, ctrl_agg_size, agg_size);
1892
1893         rtlpriv->hw->max_rx_aggregation_subframes =
1894                 (ctrl_agg_size ? agg_size : IEEE80211_MAX_AMPDU_BUF_HT);
1895 }
1896 EXPORT_SYMBOL(rtl_rx_ampdu_apply);
1897
1898 /*********************************************************
1899  *
1900  * wq & timer callback functions
1901  *
1902  *********************************************************/
1903 /* this function is used for roaming */
1904 void rtl_beacon_statistic(struct ieee80211_hw *hw, struct sk_buff *skb)
1905 {
1906         struct rtl_priv *rtlpriv = rtl_priv(hw);
1907         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
1908
1909         if (rtlpriv->mac80211.opmode != NL80211_IFTYPE_STATION)
1910                 return;
1911
1912         if (rtlpriv->mac80211.link_state < MAC80211_LINKED)
1913                 return;
1914
1915         /* check if this really is a beacon */
1916         if (!ieee80211_is_beacon(hdr->frame_control) &&
1917             !ieee80211_is_probe_resp(hdr->frame_control))
1918                 return;
1919
1920         /* min. beacon length + FCS_LEN */
1921         if (skb->len <= 40 + FCS_LEN)
1922                 return;
1923
1924         /* and only beacons from the associated BSSID, please */
1925         if (!ether_addr_equal(hdr->addr3, rtlpriv->mac80211.bssid))
1926                 return;
1927
1928         rtlpriv->link_info.bcn_rx_inperiod++;
1929 }
1930 EXPORT_SYMBOL_GPL(rtl_beacon_statistic);
1931
1932 static void rtl_free_entries_from_scan_list(struct ieee80211_hw *hw)
1933 {
1934         struct rtl_priv *rtlpriv = rtl_priv(hw);
1935         struct rtl_bssid_entry *entry, *next;
1936
1937         list_for_each_entry_safe(entry, next, &rtlpriv->scan_list.list, list) {
1938                 list_del(&entry->list);
1939                 kfree(entry);
1940                 rtlpriv->scan_list.num--;
1941         }
1942 }
1943
1944 static void rtl_free_entries_from_ack_queue(struct ieee80211_hw *hw,
1945                                             bool chk_timeout)
1946 {
1947         struct rtl_priv *rtlpriv = rtl_priv(hw);
1948         struct rtl_tx_report *tx_report = &rtlpriv->tx_report;
1949         struct sk_buff_head *queue = &tx_report->queue;
1950         struct sk_buff *skb, *tmp;
1951         struct rtlwifi_tx_info *tx_info;
1952
1953         skb_queue_walk_safe(queue, skb, tmp) {
1954                 tx_info = rtl_tx_skb_cb_info(skb);
1955                 if (chk_timeout &&
1956                     time_after(tx_info->send_time + HZ, jiffies))
1957                         continue;
1958                 skb_unlink(skb, queue);
1959                 rtl_tx_status(hw, skb, false);
1960         }
1961 }
1962
1963 void rtl_scan_list_expire(struct ieee80211_hw *hw)
1964 {
1965         struct rtl_priv *rtlpriv = rtl_priv(hw);
1966         struct rtl_bssid_entry *entry, *next;
1967         unsigned long flags;
1968
1969         spin_lock_irqsave(&rtlpriv->locks.scan_list_lock, flags);
1970
1971         list_for_each_entry_safe(entry, next, &rtlpriv->scan_list.list, list) {
1972                 /* 180 seconds */
1973                 if (jiffies_to_msecs(jiffies - entry->age) < 180000)
1974                         continue;
1975
1976                 list_del(&entry->list);
1977                 rtlpriv->scan_list.num--;
1978
1979                 RT_TRACE(rtlpriv, COMP_SCAN, DBG_LOUD,
1980                          "BSSID=%pM is expire in scan list (total=%d)\n",
1981                          entry->bssid, rtlpriv->scan_list.num);
1982                 kfree(entry);
1983         }
1984
1985         spin_unlock_irqrestore(&rtlpriv->locks.scan_list_lock, flags);
1986
1987         rtlpriv->btcoexist.btc_info.ap_num = rtlpriv->scan_list.num;
1988 }
1989
1990 void rtl_collect_scan_list(struct ieee80211_hw *hw, struct sk_buff *skb)
1991 {
1992         struct rtl_priv *rtlpriv = rtl_priv(hw);
1993         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
1994         struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
1995         unsigned long flags;
1996
1997         struct rtl_bssid_entry *entry;
1998         bool entry_found = false;
1999
2000         /* check if it is scanning */
2001         if (!mac->act_scanning)
2002                 return;
2003
2004         /* check if this really is a beacon */
2005         if (!ieee80211_is_beacon(hdr->frame_control) &&
2006             !ieee80211_is_probe_resp(hdr->frame_control))
2007                 return;
2008
2009         spin_lock_irqsave(&rtlpriv->locks.scan_list_lock, flags);
2010
2011         list_for_each_entry(entry, &rtlpriv->scan_list.list, list) {
2012                 if (memcmp(entry->bssid, hdr->addr3, ETH_ALEN) == 0) {
2013                         list_del_init(&entry->list);
2014                         entry_found = true;
2015                         RT_TRACE(rtlpriv, COMP_SCAN, DBG_LOUD,
2016                                  "Update BSSID=%pM to scan list (total=%d)\n",
2017                                  hdr->addr3, rtlpriv->scan_list.num);
2018                         break;
2019                 }
2020         }
2021
2022         if (!entry_found) {
2023                 entry = kmalloc(sizeof(*entry), GFP_ATOMIC);
2024
2025                 if (!entry)
2026                         goto label_err;
2027
2028                 memcpy(entry->bssid, hdr->addr3, ETH_ALEN);
2029                 rtlpriv->scan_list.num++;
2030
2031                 RT_TRACE(rtlpriv, COMP_SCAN, DBG_LOUD,
2032                          "Add BSSID=%pM to scan list (total=%d)\n",
2033                          hdr->addr3, rtlpriv->scan_list.num);
2034         }
2035
2036         entry->age = jiffies;
2037
2038         list_add_tail(&entry->list, &rtlpriv->scan_list.list);
2039
2040 label_err:
2041         spin_unlock_irqrestore(&rtlpriv->locks.scan_list_lock, flags);
2042 }
2043 EXPORT_SYMBOL(rtl_collect_scan_list);
2044
2045 void rtl_watchdog_wq_callback(void *data)
2046 {
2047         struct rtl_works *rtlworks = container_of_dwork_rtl(data,
2048                                                             struct rtl_works,
2049                                                             watchdog_wq);
2050         struct ieee80211_hw *hw = rtlworks->hw;
2051         struct rtl_priv *rtlpriv = rtl_priv(hw);
2052         struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
2053         struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
2054         bool busytraffic = false;
2055         bool tx_busy_traffic = false;
2056         bool rx_busy_traffic = false;
2057         bool higher_busytraffic = false;
2058         bool higher_busyrxtraffic = false;
2059         u8 idx, tid;
2060         u32 rx_cnt_inp4eriod = 0;
2061         u32 tx_cnt_inp4eriod = 0;
2062         u32 aver_rx_cnt_inperiod = 0;
2063         u32 aver_tx_cnt_inperiod = 0;
2064         u32 aver_tidtx_inperiod[MAX_TID_COUNT] = {0};
2065         u32 tidtx_inp4eriod[MAX_TID_COUNT] = {0};
2066
2067         if (is_hal_stop(rtlhal))
2068                 return;
2069
2070         /* <1> Determine if action frame is allowed */
2071         if (mac->link_state > MAC80211_NOLINK) {
2072                 if (mac->cnt_after_linked < 20)
2073                         mac->cnt_after_linked++;
2074         } else {
2075                 mac->cnt_after_linked = 0;
2076         }
2077
2078         /* <2> to check if traffic busy, if
2079          * busytraffic we don't change channel
2080          */
2081         if (mac->link_state >= MAC80211_LINKED) {
2082                 /* (1) get aver_rx_cnt_inperiod & aver_tx_cnt_inperiod */
2083                 for (idx = 0; idx <= 2; idx++) {
2084                         rtlpriv->link_info.num_rx_in4period[idx] =
2085                             rtlpriv->link_info.num_rx_in4period[idx + 1];
2086                         rtlpriv->link_info.num_tx_in4period[idx] =
2087                             rtlpriv->link_info.num_tx_in4period[idx + 1];
2088                 }
2089                 rtlpriv->link_info.num_rx_in4period[3] =
2090                     rtlpriv->link_info.num_rx_inperiod;
2091                 rtlpriv->link_info.num_tx_in4period[3] =
2092                     rtlpriv->link_info.num_tx_inperiod;
2093                 for (idx = 0; idx <= 3; idx++) {
2094                         rx_cnt_inp4eriod +=
2095                             rtlpriv->link_info.num_rx_in4period[idx];
2096                         tx_cnt_inp4eriod +=
2097                             rtlpriv->link_info.num_tx_in4period[idx];
2098                 }
2099                 aver_rx_cnt_inperiod = rx_cnt_inp4eriod / 4;
2100                 aver_tx_cnt_inperiod = tx_cnt_inp4eriod / 4;
2101
2102                 /* (2) check traffic busy */
2103                 if (aver_rx_cnt_inperiod > 100 || aver_tx_cnt_inperiod > 100) {
2104                         busytraffic = true;
2105                         if (aver_rx_cnt_inperiod > aver_tx_cnt_inperiod)
2106                                 rx_busy_traffic = true;
2107                         else
2108                                 tx_busy_traffic = false;
2109                 }
2110
2111                 /* Higher Tx/Rx data. */
2112                 if (aver_rx_cnt_inperiod > 4000 ||
2113                     aver_tx_cnt_inperiod > 4000) {
2114                         higher_busytraffic = true;
2115
2116                         /* Extremely high Rx data. */
2117                         if (aver_rx_cnt_inperiod > 5000)
2118                                 higher_busyrxtraffic = true;
2119                 }
2120
2121                 /* check every tid's tx traffic */
2122                 for (tid = 0; tid <= 7; tid++) {
2123                         for (idx = 0; idx <= 2; idx++)
2124                                 rtlpriv->link_info.tidtx_in4period[tid][idx] =
2125                                         rtlpriv->link_info.tidtx_in4period[tid]
2126                                         [idx + 1];
2127                         rtlpriv->link_info.tidtx_in4period[tid][3] =
2128                                 rtlpriv->link_info.tidtx_inperiod[tid];
2129
2130                         for (idx = 0; idx <= 3; idx++)
2131                                 tidtx_inp4eriod[tid] +=
2132                                    rtlpriv->link_info.tidtx_in4period[tid][idx];
2133                         aver_tidtx_inperiod[tid] = tidtx_inp4eriod[tid] / 4;
2134                         if (aver_tidtx_inperiod[tid] > 5000)
2135                                 rtlpriv->link_info.higher_busytxtraffic[tid] =
2136                                                                         true;
2137                         else
2138                                 rtlpriv->link_info.higher_busytxtraffic[tid] =
2139                                                                         false;
2140                 }
2141
2142                 /* PS is controlled by coex. */
2143                 if (rtlpriv->cfg->ops->get_btc_status() &&
2144                     rtlpriv->btcoexist.btc_ops->btc_is_bt_ctrl_lps(rtlpriv))
2145                         goto label_lps_done;
2146
2147                 if (rtlpriv->link_info.num_rx_inperiod +
2148                       rtlpriv->link_info.num_tx_inperiod > 8 ||
2149                     rtlpriv->link_info.num_rx_inperiod > 2)
2150                         rtl_lps_leave(hw);
2151                 else
2152                         rtl_lps_enter(hw);
2153
2154 label_lps_done:
2155                 ;
2156         }
2157
2158         for (tid = 0; tid <= 7; tid++)
2159                 rtlpriv->link_info.tidtx_inperiod[tid] = 0;
2160
2161         rtlpriv->link_info.busytraffic = busytraffic;
2162         rtlpriv->link_info.higher_busytraffic = higher_busytraffic;
2163         rtlpriv->link_info.rx_busy_traffic = rx_busy_traffic;
2164         rtlpriv->link_info.tx_busy_traffic = tx_busy_traffic;
2165         rtlpriv->link_info.higher_busyrxtraffic = higher_busyrxtraffic;
2166
2167         rtlpriv->stats.txbytesunicast_inperiod =
2168                 rtlpriv->stats.txbytesunicast -
2169                 rtlpriv->stats.txbytesunicast_last;
2170         rtlpriv->stats.rxbytesunicast_inperiod =
2171                 rtlpriv->stats.rxbytesunicast -
2172                 rtlpriv->stats.rxbytesunicast_last;
2173         rtlpriv->stats.txbytesunicast_last = rtlpriv->stats.txbytesunicast;
2174         rtlpriv->stats.rxbytesunicast_last = rtlpriv->stats.rxbytesunicast;
2175
2176         rtlpriv->stats.txbytesunicast_inperiod_tp =
2177                 (u32)(rtlpriv->stats.txbytesunicast_inperiod * 8 / 2 /
2178                 1024 / 1024);
2179         rtlpriv->stats.rxbytesunicast_inperiod_tp =
2180                 (u32)(rtlpriv->stats.rxbytesunicast_inperiod * 8 / 2 /
2181                 1024 / 1024);
2182
2183         /* <3> DM */
2184         if (!rtlpriv->cfg->mod_params->disable_watchdog)
2185                 rtlpriv->cfg->ops->dm_watchdog(hw);
2186
2187         /* <4> roaming */
2188         if (mac->link_state == MAC80211_LINKED &&
2189             mac->opmode == NL80211_IFTYPE_STATION) {
2190                 if ((rtlpriv->link_info.bcn_rx_inperiod +
2191                     rtlpriv->link_info.num_rx_inperiod) == 0) {
2192                         rtlpriv->link_info.roam_times++;
2193                         RT_TRACE(rtlpriv, COMP_ERR, DBG_DMESG,
2194                                  "AP off for %d s\n",
2195                                 (rtlpriv->link_info.roam_times * 2));
2196
2197                         /* if we can't recv beacon for 10s,
2198                          * we should reconnect this AP
2199                          */
2200                         if (rtlpriv->link_info.roam_times >= 5) {
2201                                 pr_err("AP off, try to reconnect now\n");
2202                                 rtlpriv->link_info.roam_times = 0;
2203                                 ieee80211_connection_loss(
2204                                         rtlpriv->mac80211.vif);
2205                         }
2206                 } else {
2207                         rtlpriv->link_info.roam_times = 0;
2208                 }
2209         }
2210
2211         if (rtlpriv->cfg->ops->get_btc_status())
2212                 rtlpriv->btcoexist.btc_ops->btc_periodical(rtlpriv);
2213
2214         if (rtlpriv->btcoexist.btc_info.in_4way) {
2215                 if (time_after(jiffies, rtlpriv->btcoexist.btc_info.in_4way_ts +
2216                                msecs_to_jiffies(IN_4WAY_TIMEOUT_TIME)))
2217                         rtlpriv->btcoexist.btc_info.in_4way = false;
2218         }
2219
2220         rtlpriv->link_info.num_rx_inperiod = 0;
2221         rtlpriv->link_info.num_tx_inperiod = 0;
2222         rtlpriv->link_info.bcn_rx_inperiod = 0;
2223
2224         /* <6> scan list */
2225         rtl_scan_list_expire(hw);
2226
2227         /* <7> check ack queue */
2228         rtl_free_entries_from_ack_queue(hw, true);
2229 }
2230
2231 void rtl_watch_dog_timer_callback(struct timer_list *t)
2232 {
2233         struct rtl_priv *rtlpriv = from_timer(rtlpriv, t, works.watchdog_timer);
2234
2235         queue_delayed_work(rtlpriv->works.rtl_wq,
2236                            &rtlpriv->works.watchdog_wq, 0);
2237
2238         mod_timer(&rtlpriv->works.watchdog_timer,
2239                   jiffies + MSECS(RTL_WATCH_DOG_TIME));
2240 }
2241
2242 void rtl_fwevt_wq_callback(void *data)
2243 {
2244         struct rtl_works *rtlworks =
2245                 container_of_dwork_rtl(data, struct rtl_works, fwevt_wq);
2246         struct ieee80211_hw *hw = rtlworks->hw;
2247         struct rtl_priv *rtlpriv = rtl_priv(hw);
2248
2249         rtlpriv->cfg->ops->c2h_command_handle(hw);
2250 }
2251
2252 static void rtl_c2h_content_parsing(struct ieee80211_hw *hw,
2253                                     struct sk_buff *skb);
2254
2255 static bool rtl_c2h_fast_cmd(struct ieee80211_hw *hw, struct sk_buff *skb)
2256 {
2257         u8 cmd_id = GET_C2H_CMD_ID(skb->data);
2258
2259         switch (cmd_id) {
2260         case C2H_BT_MP:
2261                 return true;
2262         default:
2263                 break;
2264         }
2265
2266         return false;
2267 }
2268
2269 void rtl_c2hcmd_enqueue(struct ieee80211_hw *hw, struct sk_buff *skb)
2270 {
2271         struct rtl_priv *rtlpriv = rtl_priv(hw);
2272         unsigned long flags;
2273
2274         if (rtl_c2h_fast_cmd(hw, skb)) {
2275                 rtl_c2h_content_parsing(hw, skb);
2276                 kfree_skb(skb);
2277                 return;
2278         }
2279
2280         /* enqueue */
2281         spin_lock_irqsave(&rtlpriv->locks.c2hcmd_lock, flags);
2282
2283         __skb_queue_tail(&rtlpriv->c2hcmd_queue, skb);
2284
2285         spin_unlock_irqrestore(&rtlpriv->locks.c2hcmd_lock, flags);
2286
2287         /* wake up wq */
2288         queue_delayed_work(rtlpriv->works.rtl_wq, &rtlpriv->works.c2hcmd_wq, 0);
2289 }
2290 EXPORT_SYMBOL(rtl_c2hcmd_enqueue);
2291
2292 static void rtl_c2h_content_parsing(struct ieee80211_hw *hw,
2293                                     struct sk_buff *skb)
2294 {
2295         struct rtl_priv *rtlpriv = rtl_priv(hw);
2296         struct rtl_hal_ops *hal_ops = rtlpriv->cfg->ops;
2297         const struct rtl_btc_ops *btc_ops = rtlpriv->btcoexist.btc_ops;
2298         u8 cmd_id, cmd_len;
2299         u8 *cmd_buf = NULL;
2300
2301         cmd_id = GET_C2H_CMD_ID(skb->data);
2302         cmd_len = skb->len - C2H_DATA_OFFSET;
2303         cmd_buf = GET_C2H_DATA_PTR(skb->data);
2304
2305         switch (cmd_id) {
2306         case C2H_DBG:
2307                 RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD, "[C2H], C2H_DBG!!\n");
2308                 break;
2309         case C2H_TXBF:
2310                 RT_TRACE(rtlpriv, COMP_FW, DBG_TRACE,
2311                          "[C2H], C2H_TXBF!!\n");
2312                 break;
2313         case C2H_TX_REPORT:
2314                 rtl_tx_report_handler(hw, cmd_buf, cmd_len);
2315                 break;
2316         case C2H_RA_RPT:
2317                 if (hal_ops->c2h_ra_report_handler)
2318                         hal_ops->c2h_ra_report_handler(hw, cmd_buf, cmd_len);
2319                 break;
2320         case C2H_BT_INFO:
2321                 RT_TRACE(rtlpriv, COMP_FW, DBG_TRACE,
2322                          "[C2H], C2H_BT_INFO!!\n");
2323                 if (rtlpriv->cfg->ops->get_btc_status())
2324                         btc_ops->btc_btinfo_notify(rtlpriv, cmd_buf, cmd_len);
2325                 break;
2326         case C2H_BT_MP:
2327                 RT_TRACE(rtlpriv, COMP_FW, DBG_TRACE,
2328                          "[C2H], C2H_BT_MP!!\n");
2329                 if (rtlpriv->cfg->ops->get_btc_status())
2330                         btc_ops->btc_btmpinfo_notify(rtlpriv, cmd_buf, cmd_len);
2331                 break;
2332         default:
2333                 RT_TRACE(rtlpriv, COMP_FW, DBG_TRACE,
2334                          "[C2H], Unknown packet!! cmd_id(%#X)!\n", cmd_id);
2335                 break;
2336         }
2337 }
2338
2339 void rtl_c2hcmd_launcher(struct ieee80211_hw *hw, int exec)
2340 {
2341         struct rtl_priv *rtlpriv = rtl_priv(hw);
2342         struct sk_buff *skb;
2343         unsigned long flags;
2344         int i;
2345
2346         for (i = 0; i < 200; i++) {
2347                 /* dequeue a task */
2348                 spin_lock_irqsave(&rtlpriv->locks.c2hcmd_lock, flags);
2349
2350                 skb = __skb_dequeue(&rtlpriv->c2hcmd_queue);
2351
2352                 spin_unlock_irqrestore(&rtlpriv->locks.c2hcmd_lock, flags);
2353
2354                 /* do it */
2355                 if (!skb)
2356                         break;
2357
2358                 RT_TRACE(rtlpriv, COMP_FW, DBG_DMESG, "C2H rx_desc_shift=%d\n",
2359                          *((u8 *)skb->cb));
2360                 RT_PRINT_DATA(rtlpriv, COMP_FW, DBG_DMESG,
2361                               "C2H data: ", skb->data, skb->len);
2362
2363                 if (exec)
2364                         rtl_c2h_content_parsing(hw, skb);
2365
2366                 /* free */
2367                 dev_kfree_skb_any(skb);
2368         }
2369 }
2370
2371 void rtl_c2hcmd_wq_callback(void *data)
2372 {
2373         struct rtl_works *rtlworks = container_of_dwork_rtl(data,
2374                                                             struct rtl_works,
2375                                                             c2hcmd_wq);
2376         struct ieee80211_hw *hw = rtlworks->hw;
2377
2378         rtl_c2hcmd_launcher(hw, 1);
2379 }
2380
2381 void rtl_easy_concurrent_retrytimer_callback(struct timer_list *t)
2382 {
2383         struct rtl_priv *rtlpriv =
2384                 from_timer(rtlpriv, t, works.dualmac_easyconcurrent_retrytimer);
2385         struct ieee80211_hw *hw = rtlpriv->hw;
2386         struct rtl_priv *buddy_priv = rtlpriv->buddy_priv;
2387
2388         if (buddy_priv == NULL)
2389                 return;
2390
2391         rtlpriv->cfg->ops->dualmac_easy_concurrent(hw);
2392 }
2393
2394 /*********************************************************
2395  *
2396  * frame process functions
2397  *
2398  *********************************************************/
2399 u8 *rtl_find_ie(u8 *data, unsigned int len, u8 ie)
2400 {
2401         struct ieee80211_mgmt *mgmt = (void *)data;
2402         u8 *pos, *end;
2403
2404         pos = (u8 *)mgmt->u.beacon.variable;
2405         end = data + len;
2406         while (pos < end) {
2407                 if (pos + 2 + pos[1] > end)
2408                         return NULL;
2409
2410                 if (pos[0] == ie)
2411                         return pos;
2412
2413                 pos += 2 + pos[1];
2414         }
2415         return NULL;
2416 }
2417
2418 /* when we use 2 rx ants we send IEEE80211_SMPS_OFF */
2419 /* when we use 1 rx ant we send IEEE80211_SMPS_STATIC */
2420 static struct sk_buff *rtl_make_smps_action(struct ieee80211_hw *hw,
2421                                      enum ieee80211_smps_mode smps,
2422                                      u8 *da, u8 *bssid)
2423 {
2424         struct rtl_efuse *rtlefuse = rtl_efuse(rtl_priv(hw));
2425         struct sk_buff *skb;
2426         struct ieee80211_mgmt *action_frame;
2427
2428         /* 27 = header + category + action + smps mode */
2429         skb = dev_alloc_skb(27 + hw->extra_tx_headroom);
2430         if (!skb)
2431                 return NULL;
2432
2433         skb_reserve(skb, hw->extra_tx_headroom);
2434         action_frame = skb_put_zero(skb, 27);
2435         memcpy(action_frame->da, da, ETH_ALEN);
2436         memcpy(action_frame->sa, rtlefuse->dev_addr, ETH_ALEN);
2437         memcpy(action_frame->bssid, bssid, ETH_ALEN);
2438         action_frame->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
2439                                                   IEEE80211_STYPE_ACTION);
2440         action_frame->u.action.category = WLAN_CATEGORY_HT;
2441         action_frame->u.action.u.ht_smps.action = WLAN_HT_ACTION_SMPS;
2442         switch (smps) {
2443         case IEEE80211_SMPS_AUTOMATIC:/* 0 */
2444         case IEEE80211_SMPS_NUM_MODES:/* 4 */
2445                 WARN_ON(1);
2446         /* fall through */
2447         case IEEE80211_SMPS_OFF:/* 1 */ /*MIMO_PS_NOLIMIT*/
2448                 action_frame->u.action.u.ht_smps.smps_control =
2449                                 WLAN_HT_SMPS_CONTROL_DISABLED;/* 0 */
2450                 break;
2451         case IEEE80211_SMPS_STATIC:/* 2 */ /*MIMO_PS_STATIC*/
2452                 action_frame->u.action.u.ht_smps.smps_control =
2453                                 WLAN_HT_SMPS_CONTROL_STATIC;/* 1 */
2454                 break;
2455         case IEEE80211_SMPS_DYNAMIC:/* 3 */ /*MIMO_PS_DYNAMIC*/
2456                 action_frame->u.action.u.ht_smps.smps_control =
2457                                 WLAN_HT_SMPS_CONTROL_DYNAMIC;/* 3 */
2458                 break;
2459         }
2460
2461         return skb;
2462 }
2463
2464 int rtl_send_smps_action(struct ieee80211_hw *hw,
2465                          struct ieee80211_sta *sta,
2466                          enum ieee80211_smps_mode smps)
2467 {
2468         struct rtl_priv *rtlpriv = rtl_priv(hw);
2469         struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
2470         struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
2471         struct sk_buff *skb = NULL;
2472         struct rtl_tcb_desc tcb_desc;
2473         u8 bssid[ETH_ALEN] = {0};
2474
2475         memset(&tcb_desc, 0, sizeof(struct rtl_tcb_desc));
2476
2477         if (rtlpriv->mac80211.act_scanning)
2478                 goto err_free;
2479
2480         if (!sta)
2481                 goto err_free;
2482
2483         if (unlikely(is_hal_stop(rtlhal) || ppsc->rfpwr_state != ERFON))
2484                 goto err_free;
2485
2486         if (!test_bit(RTL_STATUS_INTERFACE_START, &rtlpriv->status))
2487                 goto err_free;
2488
2489         if (rtlpriv->mac80211.opmode == NL80211_IFTYPE_AP)
2490                 memcpy(bssid, rtlpriv->efuse.dev_addr, ETH_ALEN);
2491         else
2492                 memcpy(bssid, rtlpriv->mac80211.bssid, ETH_ALEN);
2493
2494         skb = rtl_make_smps_action(hw, smps, sta->addr, bssid);
2495         /* this is a type = mgmt * stype = action frame */
2496         if (skb) {
2497                 struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
2498                 struct rtl_sta_info *sta_entry =
2499                         (struct rtl_sta_info *) sta->drv_priv;
2500                 sta_entry->mimo_ps = smps;
2501                 /* rtlpriv->cfg->ops->update_rate_tbl(hw, sta, 0, true); */
2502
2503                 info->control.rates[0].idx = 0;
2504                 info->band = hw->conf.chandef.chan->band;
2505                 rtlpriv->intf_ops->adapter_tx(hw, sta, skb, &tcb_desc);
2506         }
2507         return 1;
2508
2509 err_free:
2510         return 0;
2511 }
2512 EXPORT_SYMBOL(rtl_send_smps_action);
2513
2514 void rtl_phy_scan_operation_backup(struct ieee80211_hw *hw, u8 operation)
2515 {
2516         struct rtl_priv *rtlpriv = rtl_priv(hw);
2517         struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
2518         enum io_type iotype;
2519
2520         if (!is_hal_stop(rtlhal)) {
2521                 switch (operation) {
2522                 case SCAN_OPT_BACKUP:
2523                         iotype = IO_CMD_PAUSE_DM_BY_SCAN;
2524                         rtlpriv->cfg->ops->set_hw_reg(hw,
2525                                                       HW_VAR_IO_CMD,
2526                                                       (u8 *)&iotype);
2527                         break;
2528                 case SCAN_OPT_RESTORE:
2529                         iotype = IO_CMD_RESUME_DM_BY_SCAN;
2530                         rtlpriv->cfg->ops->set_hw_reg(hw,
2531                                                       HW_VAR_IO_CMD,
2532                                                       (u8 *)&iotype);
2533                         break;
2534                 default:
2535                         pr_err("Unknown Scan Backup operation.\n");
2536                         break;
2537                 }
2538         }
2539 }
2540 EXPORT_SYMBOL(rtl_phy_scan_operation_backup);
2541
2542 /* because mac80211 have issues when can receive del ba
2543  * so here we just make a fake del_ba if we receive a ba_req
2544  * but rx_agg was opened to let mac80211 release some ba
2545  * related resources, so please this del_ba for tx
2546  */
2547 struct sk_buff *rtl_make_del_ba(struct ieee80211_hw *hw,
2548                                 u8 *sa, u8 *bssid, u16 tid)
2549 {
2550         struct rtl_efuse *rtlefuse = rtl_efuse(rtl_priv(hw));
2551         struct sk_buff *skb;
2552         struct ieee80211_mgmt *action_frame;
2553         u16 params;
2554
2555         /* 27 = header + category + action + smps mode */
2556         skb = dev_alloc_skb(34 + hw->extra_tx_headroom);
2557         if (!skb)
2558                 return NULL;
2559
2560         skb_reserve(skb, hw->extra_tx_headroom);
2561         action_frame = skb_put_zero(skb, 34);
2562         memcpy(action_frame->sa, sa, ETH_ALEN);
2563         memcpy(action_frame->da, rtlefuse->dev_addr, ETH_ALEN);
2564         memcpy(action_frame->bssid, bssid, ETH_ALEN);
2565         action_frame->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
2566                                                   IEEE80211_STYPE_ACTION);
2567         action_frame->u.action.category = WLAN_CATEGORY_BACK;
2568         action_frame->u.action.u.delba.action_code = WLAN_ACTION_DELBA;
2569         params = (u16)(1 << 11);        /* bit 11 initiator */
2570         params |= (u16)(tid << 12);     /* bit 15:12 TID number */
2571
2572         action_frame->u.action.u.delba.params = cpu_to_le16(params);
2573         action_frame->u.action.u.delba.reason_code =
2574                 cpu_to_le16(WLAN_REASON_QSTA_TIMEOUT);
2575
2576         return skb;
2577 }
2578
2579 /*********************************************************
2580  *
2581  * IOT functions
2582  *
2583  *********************************************************/
2584 static bool rtl_chk_vendor_ouisub(struct ieee80211_hw *hw,
2585                                   struct octet_string vendor_ie)
2586 {
2587         struct rtl_priv *rtlpriv = rtl_priv(hw);
2588         bool matched = false;
2589         static u8 athcap_1[] = { 0x00, 0x03, 0x7F };
2590         static u8 athcap_2[] = { 0x00, 0x13, 0x74 };
2591         static u8 broadcap_1[] = { 0x00, 0x10, 0x18 };
2592         static u8 broadcap_2[] = { 0x00, 0x0a, 0xf7 };
2593         static u8 broadcap_3[] = { 0x00, 0x05, 0xb5 };
2594         static u8 racap[] = { 0x00, 0x0c, 0x43 };
2595         static u8 ciscocap[] = { 0x00, 0x40, 0x96 };
2596         static u8 marvcap[] = { 0x00, 0x50, 0x43 };
2597
2598         if (memcmp(vendor_ie.octet, athcap_1, 3) == 0 ||
2599                 memcmp(vendor_ie.octet, athcap_2, 3) == 0) {
2600                 rtlpriv->mac80211.vendor = PEER_ATH;
2601                 matched = true;
2602         } else if (memcmp(vendor_ie.octet, broadcap_1, 3) == 0 ||
2603                 memcmp(vendor_ie.octet, broadcap_2, 3) == 0 ||
2604                 memcmp(vendor_ie.octet, broadcap_3, 3) == 0) {
2605                 rtlpriv->mac80211.vendor = PEER_BROAD;
2606                 matched = true;
2607         } else if (memcmp(vendor_ie.octet, racap, 3) == 0) {
2608                 rtlpriv->mac80211.vendor = PEER_RAL;
2609                 matched = true;
2610         } else if (memcmp(vendor_ie.octet, ciscocap, 3) == 0) {
2611                 rtlpriv->mac80211.vendor = PEER_CISCO;
2612                 matched = true;
2613         } else if (memcmp(vendor_ie.octet, marvcap, 3) == 0) {
2614                 rtlpriv->mac80211.vendor = PEER_MARV;
2615                 matched = true;
2616         }
2617
2618         return matched;
2619 }
2620
2621 static bool rtl_find_221_ie(struct ieee80211_hw *hw, u8 *data,
2622                 unsigned int len)
2623 {
2624         struct ieee80211_mgmt *mgmt = (void *)data;
2625         struct octet_string vendor_ie;
2626         u8 *pos, *end;
2627
2628         pos = (u8 *)mgmt->u.beacon.variable;
2629         end = data + len;
2630         while (pos < end) {
2631                 if (pos[0] == 221) {
2632                         vendor_ie.length = pos[1];
2633                         vendor_ie.octet = &pos[2];
2634                         if (rtl_chk_vendor_ouisub(hw, vendor_ie))
2635                                 return true;
2636                 }
2637
2638                 if (pos + 2 + pos[1] > end)
2639                         return false;
2640
2641                 pos += 2 + pos[1];
2642         }
2643         return false;
2644 }
2645
2646 void rtl_recognize_peer(struct ieee80211_hw *hw, u8 *data, unsigned int len)
2647 {
2648         struct rtl_priv *rtlpriv = rtl_priv(hw);
2649         struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
2650         struct ieee80211_hdr *hdr = (void *)data;
2651         u32 vendor = PEER_UNKNOWN;
2652
2653         static u8 ap3_1[3] = { 0x00, 0x14, 0xbf };
2654         static u8 ap3_2[3] = { 0x00, 0x1a, 0x70 };
2655         static u8 ap3_3[3] = { 0x00, 0x1d, 0x7e };
2656         static u8 ap4_1[3] = { 0x00, 0x90, 0xcc };
2657         static u8 ap4_2[3] = { 0x00, 0x0e, 0x2e };
2658         static u8 ap4_3[3] = { 0x00, 0x18, 0x02 };
2659         static u8 ap4_4[3] = { 0x00, 0x17, 0x3f };
2660         static u8 ap4_5[3] = { 0x00, 0x1c, 0xdf };
2661         static u8 ap5_1[3] = { 0x00, 0x1c, 0xf0 };
2662         static u8 ap5_2[3] = { 0x00, 0x21, 0x91 };
2663         static u8 ap5_3[3] = { 0x00, 0x24, 0x01 };
2664         static u8 ap5_4[3] = { 0x00, 0x15, 0xe9 };
2665         static u8 ap5_5[3] = { 0x00, 0x17, 0x9A };
2666         static u8 ap5_6[3] = { 0x00, 0x18, 0xE7 };
2667         static u8 ap6_1[3] = { 0x00, 0x17, 0x94 };
2668         static u8 ap7_1[3] = { 0x00, 0x14, 0xa4 };
2669
2670         if (mac->opmode != NL80211_IFTYPE_STATION)
2671                 return;
2672
2673         if (mac->link_state == MAC80211_NOLINK) {
2674                 mac->vendor = PEER_UNKNOWN;
2675                 return;
2676         }
2677
2678         if (mac->cnt_after_linked > 2)
2679                 return;
2680
2681         /* check if this really is a beacon */
2682         if (!ieee80211_is_beacon(hdr->frame_control))
2683                 return;
2684
2685         /* min. beacon length + FCS_LEN */
2686         if (len <= 40 + FCS_LEN)
2687                 return;
2688
2689         /* and only beacons from the associated BSSID, please */
2690         if (!ether_addr_equal_64bits(hdr->addr3, rtlpriv->mac80211.bssid))
2691                 return;
2692
2693         if (rtl_find_221_ie(hw, data, len))
2694                 vendor = mac->vendor;
2695
2696         if ((memcmp(mac->bssid, ap5_1, 3) == 0) ||
2697                 (memcmp(mac->bssid, ap5_2, 3) == 0) ||
2698                 (memcmp(mac->bssid, ap5_3, 3) == 0) ||
2699                 (memcmp(mac->bssid, ap5_4, 3) == 0) ||
2700                 (memcmp(mac->bssid, ap5_5, 3) == 0) ||
2701                 (memcmp(mac->bssid, ap5_6, 3) == 0) ||
2702                 vendor == PEER_ATH) {
2703                 vendor = PEER_ATH;
2704                 RT_TRACE(rtlpriv, COMP_MAC80211, DBG_LOUD, "=>ath find\n");
2705         } else if ((memcmp(mac->bssid, ap4_4, 3) == 0) ||
2706                 (memcmp(mac->bssid, ap4_5, 3) == 0) ||
2707                 (memcmp(mac->bssid, ap4_1, 3) == 0) ||
2708                 (memcmp(mac->bssid, ap4_2, 3) == 0) ||
2709                 (memcmp(mac->bssid, ap4_3, 3) == 0) ||
2710                 vendor == PEER_RAL) {
2711                 RT_TRACE(rtlpriv, COMP_MAC80211, DBG_LOUD, "=>ral find\n");
2712                 vendor = PEER_RAL;
2713         } else if (memcmp(mac->bssid, ap6_1, 3) == 0 ||
2714                 vendor == PEER_CISCO) {
2715                 vendor = PEER_CISCO;
2716                 RT_TRACE(rtlpriv, COMP_MAC80211, DBG_LOUD, "=>cisco find\n");
2717         } else if ((memcmp(mac->bssid, ap3_1, 3) == 0) ||
2718                 (memcmp(mac->bssid, ap3_2, 3) == 0) ||
2719                 (memcmp(mac->bssid, ap3_3, 3) == 0) ||
2720                 vendor == PEER_BROAD) {
2721                 RT_TRACE(rtlpriv, COMP_MAC80211, DBG_LOUD, "=>broad find\n");
2722                 vendor = PEER_BROAD;
2723         } else if (memcmp(mac->bssid, ap7_1, 3) == 0 ||
2724                 vendor == PEER_MARV) {
2725                 vendor = PEER_MARV;
2726                 RT_TRACE(rtlpriv, COMP_MAC80211, DBG_LOUD, "=>marv find\n");
2727         }
2728
2729         mac->vendor = vendor;
2730 }
2731 EXPORT_SYMBOL_GPL(rtl_recognize_peer);
2732
2733 MODULE_AUTHOR("lizhaoming       <chaoming_li@realsil.com.cn>");
2734 MODULE_AUTHOR("Realtek WlanFAE  <wlanfae@realtek.com>");
2735 MODULE_AUTHOR("Larry Finger     <Larry.FInger@lwfinger.net>");
2736 MODULE_LICENSE("GPL");
2737 MODULE_DESCRIPTION("Realtek 802.11n PCI wireless core");
2738
2739 struct rtl_global_var rtl_global_var = {};
2740 EXPORT_SYMBOL_GPL(rtl_global_var);
2741
2742 static int __init rtl_core_module_init(void)
2743 {
2744         BUILD_BUG_ON(TX_PWR_BY_RATE_NUM_RATE < TX_PWR_BY_RATE_NUM_SECTION);
2745         BUILD_BUG_ON(MAX_RATE_SECTION_NUM != MAX_RATE_SECTION);
2746         BUILD_BUG_ON(MAX_BASE_NUM_IN_PHY_REG_PG_24G != MAX_RATE_SECTION);
2747         BUILD_BUG_ON(MAX_BASE_NUM_IN_PHY_REG_PG_5G != (MAX_RATE_SECTION - 1));
2748
2749         if (rtl_rate_control_register())
2750                 pr_err("rtl: Unable to register rtl_rc, use default RC !!\n");
2751
2752         /* add debugfs */
2753         rtl_debugfs_add_topdir();
2754
2755         /* init some global vars */
2756         INIT_LIST_HEAD(&rtl_global_var.glb_priv_list);
2757         spin_lock_init(&rtl_global_var.glb_list_lock);
2758
2759         return 0;
2760 }
2761
2762 static void __exit rtl_core_module_exit(void)
2763 {
2764         /*RC*/
2765         rtl_rate_control_unregister();
2766
2767         /* remove debugfs */
2768         rtl_debugfs_remove_topdir();
2769 }
2770
2771 module_init(rtl_core_module_init);
2772 module_exit(rtl_core_module_exit);