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