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