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 (rtlhal->hw_type == HARDWARE_TYPE_RTL8812AE) {
248 		u16 mcs_map;
249 
250 		vht_cap->vht_supported = true;
251 		vht_cap->cap =
252 			IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_3895 |
253 			IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_7991 |
254 			IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454 |
255 			IEEE80211_VHT_CAP_SHORT_GI_80 |
256 			IEEE80211_VHT_CAP_TXSTBC |
257 			IEEE80211_VHT_CAP_RXSTBC_1 |
258 			IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE |
259 			IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE |
260 			IEEE80211_VHT_CAP_HTC_VHT |
261 			IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK |
262 			IEEE80211_VHT_CAP_RX_ANTENNA_PATTERN |
263 			IEEE80211_VHT_CAP_TX_ANTENNA_PATTERN |
264 			0;
265 
266 		mcs_map = IEEE80211_VHT_MCS_SUPPORT_0_9 << 0 |
267 			IEEE80211_VHT_MCS_SUPPORT_0_9 << 2 |
268 			IEEE80211_VHT_MCS_NOT_SUPPORTED << 4 |
269 			IEEE80211_VHT_MCS_NOT_SUPPORTED << 6 |
270 			IEEE80211_VHT_MCS_NOT_SUPPORTED << 8 |
271 			IEEE80211_VHT_MCS_NOT_SUPPORTED << 10 |
272 			IEEE80211_VHT_MCS_NOT_SUPPORTED << 12 |
273 			IEEE80211_VHT_MCS_NOT_SUPPORTED << 14;
274 
275 		vht_cap->vht_mcs.rx_mcs_map = cpu_to_le16(mcs_map);
276 		vht_cap->vht_mcs.rx_highest =
277 			cpu_to_le16(MAX_BIT_RATE_SHORT_GI_2NSS_80MHZ_MCS9);
278 		vht_cap->vht_mcs.tx_mcs_map = cpu_to_le16(mcs_map);
279 		vht_cap->vht_mcs.tx_highest =
280 			cpu_to_le16(MAX_BIT_RATE_SHORT_GI_2NSS_80MHZ_MCS9);
281 	} else if (rtlhal->hw_type == HARDWARE_TYPE_RTL8821AE) {
282 		u16 mcs_map;
283 
284 		vht_cap->vht_supported = true;
285 		vht_cap->cap =
286 			IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_3895 |
287 			IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_7991 |
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, CONNECTION_MONITOR);
400 	ieee80211_hw_set(hw, MFP_CAPABLE);
401 	ieee80211_hw_set(hw, REPORTS_TX_ACK_STATUS);
402 
403 	/* swlps or hwlps has been set in diff chip in init_sw_vars */
404 	if (rtlpriv->psc.swctrl_lps) {
405 		ieee80211_hw_set(hw, SUPPORTS_PS);
406 		ieee80211_hw_set(hw, PS_NULLFUNC_STACK);
407 	}
408 	if (rtlpriv->psc.fwctrl_lps) {
409 		ieee80211_hw_set(hw, SUPPORTS_PS);
410 		ieee80211_hw_set(hw, SUPPORTS_DYNAMIC_PS);
411 	}
412 	hw->wiphy->interface_modes =
413 	    BIT(NL80211_IFTYPE_AP) |
414 	    BIT(NL80211_IFTYPE_STATION) |
415 	    BIT(NL80211_IFTYPE_ADHOC) |
416 	    BIT(NL80211_IFTYPE_MESH_POINT) |
417 	    BIT(NL80211_IFTYPE_P2P_CLIENT) |
418 	    BIT(NL80211_IFTYPE_P2P_GO);
419 	hw->wiphy->flags |= WIPHY_FLAG_IBSS_RSN;
420 
421 	hw->wiphy->flags |= WIPHY_FLAG_HAS_REMAIN_ON_CHANNEL;
422 
423 	hw->wiphy->rts_threshold = 2347;
424 
425 	hw->queues = AC_MAX;
426 	hw->extra_tx_headroom = RTL_TX_HEADER_SIZE;
427 
428 	/* TODO: Correct this value for our hw */
429 	/* TODO: define these hard code value */
430 	hw->max_listen_interval = 10;
431 	hw->max_rate_tries = 4;
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 	setup_timer(&rtlpriv->works.watchdog_timer,
466 		    rtl_watch_dog_timer_callback, (unsigned long)hw);
467 	setup_timer(&rtlpriv->works.dualmac_easyconcurrent_retrytimer,
468 		    rtl_easy_concurrent_retrytimer_callback, (unsigned long)hw);
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 	spin_lock_init(&rtlpriv->locks.ips_lock);
560 	spin_lock_init(&rtlpriv->locks.irq_th_lock);
561 	spin_lock_init(&rtlpriv->locks.h2c_lock);
562 	spin_lock_init(&rtlpriv->locks.rf_ps_lock);
563 	spin_lock_init(&rtlpriv->locks.rf_lock);
564 	spin_lock_init(&rtlpriv->locks.waitq_lock);
565 	spin_lock_init(&rtlpriv->locks.entry_list_lock);
566 	spin_lock_init(&rtlpriv->locks.c2hcmd_lock);
567 	spin_lock_init(&rtlpriv->locks.scan_list_lock);
568 	spin_lock_init(&rtlpriv->locks.cck_and_rw_pagea_lock);
569 	spin_lock_init(&rtlpriv->locks.check_sendpkt_lock);
570 	spin_lock_init(&rtlpriv->locks.fw_ps_lock);
571 	spin_lock_init(&rtlpriv->locks.lps_lock);
572 	spin_lock_init(&rtlpriv->locks.iqk_lock);
573 	/* <5> init list */
574 	INIT_LIST_HEAD(&rtlpriv->entry_list);
575 	INIT_LIST_HEAD(&rtlpriv->c2hcmd_list);
576 	INIT_LIST_HEAD(&rtlpriv->scan_list.list);
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 
589 void rtl_deinit_core(struct ieee80211_hw *hw)
590 {
591 	rtl_c2hcmd_launcher(hw, 0);
592 	rtl_free_entries_from_scan_list(hw);
593 }
594 EXPORT_SYMBOL_GPL(rtl_deinit_core);
595 
596 void rtl_init_rx_config(struct ieee80211_hw *hw)
597 {
598 	struct rtl_priv *rtlpriv = rtl_priv(hw);
599 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
600 
601 	rtlpriv->cfg->ops->get_hw_reg(hw, HW_VAR_RCR, (u8 *) (&mac->rx_conf));
602 }
603 EXPORT_SYMBOL_GPL(rtl_init_rx_config);
604 
605 /*********************************************************
606  *
607  * tx information functions
608  *
609  *********************************************************/
610 static void _rtl_qurey_shortpreamble_mode(struct ieee80211_hw *hw,
611 					  struct rtl_tcb_desc *tcb_desc,
612 					  struct ieee80211_tx_info *info)
613 {
614 	struct rtl_priv *rtlpriv = rtl_priv(hw);
615 	u8 rate_flag = info->control.rates[0].flags;
616 
617 	tcb_desc->use_shortpreamble = false;
618 
619 	/* 1M can only use Long Preamble. 11B spec */
620 	if (tcb_desc->hw_rate == rtlpriv->cfg->maps[RTL_RC_CCK_RATE1M])
621 		return;
622 	else if (rate_flag & IEEE80211_TX_RC_USE_SHORT_PREAMBLE)
623 		tcb_desc->use_shortpreamble = true;
624 
625 	return;
626 }
627 
628 static void _rtl_query_shortgi(struct ieee80211_hw *hw,
629 			       struct ieee80211_sta *sta,
630 			       struct rtl_tcb_desc *tcb_desc,
631 			       struct ieee80211_tx_info *info)
632 {
633 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
634 	u8 rate_flag = info->control.rates[0].flags;
635 	u8 sgi_40 = 0, sgi_20 = 0, bw_40 = 0;
636 	u8 sgi_80 = 0, bw_80 = 0;
637 	tcb_desc->use_shortgi = false;
638 
639 	if (sta == NULL)
640 		return;
641 
642 	sgi_40 = sta->ht_cap.cap & IEEE80211_HT_CAP_SGI_40;
643 	sgi_20 = sta->ht_cap.cap & IEEE80211_HT_CAP_SGI_20;
644 	sgi_80 = sta->vht_cap.cap & IEEE80211_VHT_CAP_SHORT_GI_80;
645 
646 	if ((!sta->ht_cap.ht_supported) && (!sta->vht_cap.vht_supported))
647 		return;
648 
649 	if (!sgi_40 && !sgi_20)
650 		return;
651 
652 	if (mac->opmode == NL80211_IFTYPE_STATION) {
653 		bw_40 = mac->bw_40;
654 		bw_80 = mac->bw_80;
655 	} else if (mac->opmode == NL80211_IFTYPE_AP ||
656 		 mac->opmode == NL80211_IFTYPE_ADHOC ||
657 		 mac->opmode == NL80211_IFTYPE_MESH_POINT) {
658 		bw_40 = sta->ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40;
659 		bw_80 = sta->vht_cap.vht_supported;
660 	}
661 
662 	if (bw_80) {
663 		if (sgi_80)
664 			tcb_desc->use_shortgi = true;
665 		else
666 			tcb_desc->use_shortgi = false;
667 	} else {
668 		if (bw_40 && sgi_40)
669 			tcb_desc->use_shortgi = true;
670 		else if (!bw_40 && sgi_20)
671 			tcb_desc->use_shortgi = true;
672 	}
673 
674 	if (!(rate_flag & IEEE80211_TX_RC_SHORT_GI))
675 		tcb_desc->use_shortgi = false;
676 }
677 
678 static void _rtl_query_protection_mode(struct ieee80211_hw *hw,
679 				       struct rtl_tcb_desc *tcb_desc,
680 				       struct ieee80211_tx_info *info)
681 {
682 	struct rtl_priv *rtlpriv = rtl_priv(hw);
683 	u8 rate_flag = info->control.rates[0].flags;
684 
685 	/* Common Settings */
686 	tcb_desc->rts_stbc = false;
687 	tcb_desc->cts_enable = false;
688 	tcb_desc->rts_sc = 0;
689 	tcb_desc->rts_bw = false;
690 	tcb_desc->rts_use_shortpreamble = false;
691 	tcb_desc->rts_use_shortgi = false;
692 
693 	if (rate_flag & IEEE80211_TX_RC_USE_CTS_PROTECT) {
694 		/* Use CTS-to-SELF in protection mode. */
695 		tcb_desc->rts_enable = true;
696 		tcb_desc->cts_enable = true;
697 		tcb_desc->rts_rate = rtlpriv->cfg->maps[RTL_RC_OFDM_RATE24M];
698 	} else if (rate_flag & IEEE80211_TX_RC_USE_RTS_CTS) {
699 		/* Use RTS-CTS in protection mode. */
700 		tcb_desc->rts_enable = true;
701 		tcb_desc->rts_rate = rtlpriv->cfg->maps[RTL_RC_OFDM_RATE24M];
702 	}
703 }
704 
705 static void _rtl_txrate_selectmode(struct ieee80211_hw *hw,
706 				   struct ieee80211_sta *sta,
707 				   struct rtl_tcb_desc *tcb_desc)
708 {
709 	struct rtl_priv *rtlpriv = rtl_priv(hw);
710 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
711 	struct rtl_sta_info *sta_entry = NULL;
712 	u8 ratr_index = 7;
713 
714 	if (sta) {
715 		sta_entry = (struct rtl_sta_info *) sta->drv_priv;
716 		ratr_index = sta_entry->ratr_index;
717 	}
718 	if (!tcb_desc->disable_ratefallback || !tcb_desc->use_driver_rate) {
719 		if (mac->opmode == NL80211_IFTYPE_STATION) {
720 			tcb_desc->ratr_index = 0;
721 		} else if (mac->opmode == NL80211_IFTYPE_ADHOC ||
722 				mac->opmode == NL80211_IFTYPE_MESH_POINT) {
723 			if (tcb_desc->multicast || tcb_desc->broadcast) {
724 				tcb_desc->hw_rate =
725 				    rtlpriv->cfg->maps[RTL_RC_CCK_RATE2M];
726 				tcb_desc->use_driver_rate = 1;
727 				tcb_desc->ratr_index = RATR_INX_WIRELESS_MC;
728 			} else {
729 				tcb_desc->ratr_index = ratr_index;
730 			}
731 		} else if (mac->opmode == NL80211_IFTYPE_AP) {
732 			tcb_desc->ratr_index = ratr_index;
733 		}
734 	}
735 
736 	if (rtlpriv->dm.useramask) {
737 		tcb_desc->ratr_index = ratr_index;
738 		/* TODO we will differentiate adhoc and station future  */
739 		if (mac->opmode == NL80211_IFTYPE_STATION ||
740 		    mac->opmode == NL80211_IFTYPE_MESH_POINT) {
741 			tcb_desc->mac_id = 0;
742 
743 			if (mac->mode == WIRELESS_MODE_AC_5G)
744 				tcb_desc->ratr_index =
745 					RATR_INX_WIRELESS_AC_5N;
746 			else if (mac->mode == WIRELESS_MODE_AC_24G)
747 				tcb_desc->ratr_index =
748 					RATR_INX_WIRELESS_AC_24N;
749 			else if (mac->mode == WIRELESS_MODE_N_24G)
750 				tcb_desc->ratr_index = RATR_INX_WIRELESS_NGB;
751 			else if (mac->mode == WIRELESS_MODE_N_5G)
752 				tcb_desc->ratr_index = RATR_INX_WIRELESS_NG;
753 			else if (mac->mode & WIRELESS_MODE_G)
754 				tcb_desc->ratr_index = RATR_INX_WIRELESS_GB;
755 			else if (mac->mode & WIRELESS_MODE_B)
756 				tcb_desc->ratr_index = RATR_INX_WIRELESS_B;
757 			else if (mac->mode & WIRELESS_MODE_A)
758 				tcb_desc->ratr_index = RATR_INX_WIRELESS_G;
759 
760 		} else if (mac->opmode == NL80211_IFTYPE_AP ||
761 			mac->opmode == NL80211_IFTYPE_ADHOC) {
762 			if (NULL != sta) {
763 				if (sta->aid > 0)
764 					tcb_desc->mac_id = sta->aid + 1;
765 				else
766 					tcb_desc->mac_id = 1;
767 			} else {
768 				tcb_desc->mac_id = 0;
769 			}
770 		}
771 	}
772 }
773 
774 static void _rtl_query_bandwidth_mode(struct ieee80211_hw *hw,
775 				      struct ieee80211_sta *sta,
776 				      struct rtl_tcb_desc *tcb_desc)
777 {
778 	struct rtl_priv *rtlpriv = rtl_priv(hw);
779 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
780 
781 	tcb_desc->packet_bw = false;
782 	if (!sta)
783 		return;
784 	if (mac->opmode == NL80211_IFTYPE_AP ||
785 	    mac->opmode == NL80211_IFTYPE_ADHOC ||
786 	    mac->opmode == NL80211_IFTYPE_MESH_POINT) {
787 		if (!(sta->ht_cap.ht_supported) ||
788 		    !(sta->ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40))
789 			return;
790 	} else if (mac->opmode == NL80211_IFTYPE_STATION) {
791 		if (!mac->bw_40 || !(sta->ht_cap.ht_supported))
792 			return;
793 	}
794 	if (tcb_desc->multicast || tcb_desc->broadcast)
795 		return;
796 
797 	/*use legency rate, shall use 20MHz */
798 	if (tcb_desc->hw_rate <= rtlpriv->cfg->maps[RTL_RC_OFDM_RATE54M])
799 		return;
800 
801 	tcb_desc->packet_bw = HT_CHANNEL_WIDTH_20_40;
802 
803 	if (rtlpriv->rtlhal.hw_type == HARDWARE_TYPE_RTL8812AE ||
804 	    rtlpriv->rtlhal.hw_type == HARDWARE_TYPE_RTL8821AE) {
805 		if (mac->opmode == NL80211_IFTYPE_AP ||
806 		    mac->opmode == NL80211_IFTYPE_ADHOC ||
807 		    mac->opmode == NL80211_IFTYPE_MESH_POINT) {
808 			if (!(sta->vht_cap.vht_supported))
809 				return;
810 		} else if (mac->opmode == NL80211_IFTYPE_STATION) {
811 			if (!mac->bw_80 ||
812 			    !(sta->vht_cap.vht_supported))
813 				return;
814 		}
815 		if (tcb_desc->hw_rate <=
816 			rtlpriv->cfg->maps[RTL_RC_HT_RATEMCS15])
817 			return;
818 		tcb_desc->packet_bw = HT_CHANNEL_WIDTH_80;
819 	}
820 }
821 
822 static u8 _rtl_get_vht_highest_n_rate(struct ieee80211_hw *hw,
823 				      struct ieee80211_sta *sta)
824 {
825 	struct rtl_priv *rtlpriv = rtl_priv(hw);
826 	struct rtl_phy *rtlphy = &(rtlpriv->phy);
827 	u8 hw_rate;
828 	u16 tx_mcs_map = le16_to_cpu(sta->vht_cap.vht_mcs.tx_mcs_map);
829 
830 	if ((get_rf_type(rtlphy) == RF_2T2R) &&
831 	    (tx_mcs_map & 0x000c) != 0x000c) {
832 		if ((tx_mcs_map & 0x000c) >> 2 ==
833 			IEEE80211_VHT_MCS_SUPPORT_0_7)
834 			hw_rate =
835 			rtlpriv->cfg->maps[RTL_RC_VHT_RATE_2SS_MCS7];
836 		else if ((tx_mcs_map  & 0x000c) >> 2 ==
837 			IEEE80211_VHT_MCS_SUPPORT_0_8)
838 			hw_rate =
839 			rtlpriv->cfg->maps[RTL_RC_VHT_RATE_2SS_MCS9];
840 		else
841 			hw_rate =
842 			rtlpriv->cfg->maps[RTL_RC_VHT_RATE_2SS_MCS9];
843 	} else {
844 		if ((tx_mcs_map  & 0x0003) ==
845 			IEEE80211_VHT_MCS_SUPPORT_0_7)
846 			hw_rate =
847 			rtlpriv->cfg->maps[RTL_RC_VHT_RATE_1SS_MCS7];
848 		else if ((tx_mcs_map  & 0x0003) ==
849 			IEEE80211_VHT_MCS_SUPPORT_0_8)
850 			hw_rate =
851 			rtlpriv->cfg->maps[RTL_RC_VHT_RATE_1SS_MCS9];
852 		else
853 			hw_rate =
854 			rtlpriv->cfg->maps[RTL_RC_VHT_RATE_1SS_MCS9];
855 	}
856 
857 	return hw_rate;
858 }
859 
860 static u8 _rtl_get_highest_n_rate(struct ieee80211_hw *hw,
861 				  struct ieee80211_sta *sta)
862 {
863 	struct rtl_priv *rtlpriv = rtl_priv(hw);
864 	struct rtl_phy *rtlphy = &rtlpriv->phy;
865 	u8 hw_rate;
866 
867 	if ((get_rf_type(rtlphy) == RF_2T2R) &&
868 	    (sta->ht_cap.mcs.rx_mask[1] != 0))
869 		hw_rate = rtlpriv->cfg->maps[RTL_RC_HT_RATEMCS15];
870 	else
871 		hw_rate = rtlpriv->cfg->maps[RTL_RC_HT_RATEMCS7];
872 
873 	return hw_rate;
874 }
875 
876 /* mac80211's rate_idx is like this:
877  *
878  * 2.4G band:rx_status->band == NL80211_BAND_2GHZ
879  *
880  * B/G rate:
881  * (rx_status->flag & RX_FLAG_HT) = 0,
882  * DESC_RATE1M-->DESC_RATE54M ==> idx is 0-->11,
883  *
884  * N rate:
885  * (rx_status->flag & RX_FLAG_HT) = 1,
886  * DESC_RATEMCS0-->DESC_RATEMCS15 ==> idx is 0-->15
887  *
888  * 5G band:rx_status->band == NL80211_BAND_5GHZ
889  * A rate:
890  * (rx_status->flag & RX_FLAG_HT) = 0,
891  * DESC_RATE6M-->DESC_RATE54M ==> idx is 0-->7,
892  *
893  * N rate:
894  * (rx_status->flag & RX_FLAG_HT) = 1,
895  * DESC_RATEMCS0-->DESC_RATEMCS15 ==> idx is 0-->15
896  *
897  * VHT rates:
898  * DESC_RATEVHT1SS_MCS0-->DESC_RATEVHT1SS_MCS9 ==> idx is 0-->9
899  * DESC_RATEVHT2SS_MCS0-->DESC_RATEVHT2SS_MCS9 ==> idx is 0-->9
900  */
901 int rtlwifi_rate_mapping(struct ieee80211_hw *hw, bool isht, bool isvht,
902 			 u8 desc_rate)
903 {
904 	int rate_idx;
905 
906 	if (isvht) {
907 		switch (desc_rate) {
908 		case DESC_RATEVHT1SS_MCS0:
909 			rate_idx = 0;
910 			break;
911 		case DESC_RATEVHT1SS_MCS1:
912 			rate_idx = 1;
913 			break;
914 		case DESC_RATEVHT1SS_MCS2:
915 			rate_idx = 2;
916 			break;
917 		case DESC_RATEVHT1SS_MCS3:
918 			rate_idx = 3;
919 			break;
920 		case DESC_RATEVHT1SS_MCS4:
921 			rate_idx = 4;
922 			break;
923 		case DESC_RATEVHT1SS_MCS5:
924 			rate_idx = 5;
925 			break;
926 		case DESC_RATEVHT1SS_MCS6:
927 			rate_idx = 6;
928 			break;
929 		case DESC_RATEVHT1SS_MCS7:
930 			rate_idx = 7;
931 			break;
932 		case DESC_RATEVHT1SS_MCS8:
933 			rate_idx = 8;
934 			break;
935 		case DESC_RATEVHT1SS_MCS9:
936 			rate_idx = 9;
937 			break;
938 		case DESC_RATEVHT2SS_MCS0:
939 			rate_idx = 0;
940 			break;
941 		case DESC_RATEVHT2SS_MCS1:
942 			rate_idx = 1;
943 			break;
944 		case DESC_RATEVHT2SS_MCS2:
945 			rate_idx = 2;
946 			break;
947 		case DESC_RATEVHT2SS_MCS3:
948 			rate_idx = 3;
949 			break;
950 		case DESC_RATEVHT2SS_MCS4:
951 			rate_idx = 4;
952 			break;
953 		case DESC_RATEVHT2SS_MCS5:
954 			rate_idx = 5;
955 			break;
956 		case DESC_RATEVHT2SS_MCS6:
957 			rate_idx = 6;
958 			break;
959 		case DESC_RATEVHT2SS_MCS7:
960 			rate_idx = 7;
961 			break;
962 		case DESC_RATEVHT2SS_MCS8:
963 			rate_idx = 8;
964 			break;
965 		case DESC_RATEVHT2SS_MCS9:
966 			rate_idx = 9;
967 			break;
968 		default:
969 			rate_idx = 0;
970 			break;
971 		}
972 		return rate_idx;
973 	}
974 	if (false == isht) {
975 		if (NL80211_BAND_2GHZ == hw->conf.chandef.chan->band) {
976 			switch (desc_rate) {
977 			case DESC_RATE1M:
978 				rate_idx = 0;
979 				break;
980 			case DESC_RATE2M:
981 				rate_idx = 1;
982 				break;
983 			case DESC_RATE5_5M:
984 				rate_idx = 2;
985 				break;
986 			case DESC_RATE11M:
987 				rate_idx = 3;
988 				break;
989 			case DESC_RATE6M:
990 				rate_idx = 4;
991 				break;
992 			case DESC_RATE9M:
993 				rate_idx = 5;
994 				break;
995 			case DESC_RATE12M:
996 				rate_idx = 6;
997 				break;
998 			case DESC_RATE18M:
999 				rate_idx = 7;
1000 				break;
1001 			case DESC_RATE24M:
1002 				rate_idx = 8;
1003 				break;
1004 			case DESC_RATE36M:
1005 				rate_idx = 9;
1006 				break;
1007 			case DESC_RATE48M:
1008 				rate_idx = 10;
1009 				break;
1010 			case DESC_RATE54M:
1011 				rate_idx = 11;
1012 				break;
1013 			default:
1014 				rate_idx = 0;
1015 				break;
1016 			}
1017 		} else {
1018 			switch (desc_rate) {
1019 			case DESC_RATE6M:
1020 				rate_idx = 0;
1021 				break;
1022 			case DESC_RATE9M:
1023 				rate_idx = 1;
1024 				break;
1025 			case DESC_RATE12M:
1026 				rate_idx = 2;
1027 				break;
1028 			case DESC_RATE18M:
1029 				rate_idx = 3;
1030 				break;
1031 			case DESC_RATE24M:
1032 				rate_idx = 4;
1033 				break;
1034 			case DESC_RATE36M:
1035 				rate_idx = 5;
1036 				break;
1037 			case DESC_RATE48M:
1038 				rate_idx = 6;
1039 				break;
1040 			case DESC_RATE54M:
1041 				rate_idx = 7;
1042 				break;
1043 			default:
1044 				rate_idx = 0;
1045 				break;
1046 			}
1047 		}
1048 	} else {
1049 		switch (desc_rate) {
1050 		case DESC_RATEMCS0:
1051 			rate_idx = 0;
1052 			break;
1053 		case DESC_RATEMCS1:
1054 			rate_idx = 1;
1055 			break;
1056 		case DESC_RATEMCS2:
1057 			rate_idx = 2;
1058 			break;
1059 		case DESC_RATEMCS3:
1060 			rate_idx = 3;
1061 			break;
1062 		case DESC_RATEMCS4:
1063 			rate_idx = 4;
1064 			break;
1065 		case DESC_RATEMCS5:
1066 			rate_idx = 5;
1067 			break;
1068 		case DESC_RATEMCS6:
1069 			rate_idx = 6;
1070 			break;
1071 		case DESC_RATEMCS7:
1072 			rate_idx = 7;
1073 			break;
1074 		case DESC_RATEMCS8:
1075 			rate_idx = 8;
1076 			break;
1077 		case DESC_RATEMCS9:
1078 			rate_idx = 9;
1079 			break;
1080 		case DESC_RATEMCS10:
1081 			rate_idx = 10;
1082 			break;
1083 		case DESC_RATEMCS11:
1084 			rate_idx = 11;
1085 			break;
1086 		case DESC_RATEMCS12:
1087 			rate_idx = 12;
1088 			break;
1089 		case DESC_RATEMCS13:
1090 			rate_idx = 13;
1091 			break;
1092 		case DESC_RATEMCS14:
1093 			rate_idx = 14;
1094 			break;
1095 		case DESC_RATEMCS15:
1096 			rate_idx = 15;
1097 			break;
1098 		default:
1099 			rate_idx = 0;
1100 			break;
1101 		}
1102 	}
1103 	return rate_idx;
1104 }
1105 EXPORT_SYMBOL(rtlwifi_rate_mapping);
1106 
1107 void rtl_get_tcb_desc(struct ieee80211_hw *hw,
1108 		      struct ieee80211_tx_info *info,
1109 		      struct ieee80211_sta *sta,
1110 		      struct sk_buff *skb, struct rtl_tcb_desc *tcb_desc)
1111 {
1112 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1113 	struct rtl_mac *rtlmac = rtl_mac(rtl_priv(hw));
1114 	struct ieee80211_hdr *hdr = rtl_get_hdr(skb);
1115 	struct ieee80211_rate *txrate;
1116 	__le16 fc = rtl_get_fc(skb);
1117 
1118 	txrate = ieee80211_get_tx_rate(hw, info);
1119 	if (txrate)
1120 		tcb_desc->hw_rate = txrate->hw_value;
1121 
1122 	if (rtl_is_tx_report_skb(hw, skb))
1123 		tcb_desc->use_spe_rpt = 1;
1124 
1125 	if (ieee80211_is_data(fc)) {
1126 		/*
1127 		 *we set data rate INX 0
1128 		 *in rtl_rc.c   if skb is special data or
1129 		 *mgt which need low data rate.
1130 		 */
1131 
1132 		/*
1133 		 *So tcb_desc->hw_rate is just used for
1134 		 *special data and mgt frames
1135 		 */
1136 		if (info->control.rates[0].idx == 0 ||
1137 				ieee80211_is_nullfunc(fc)) {
1138 			tcb_desc->use_driver_rate = true;
1139 			tcb_desc->ratr_index = RATR_INX_WIRELESS_MC;
1140 
1141 			tcb_desc->disable_ratefallback = 1;
1142 		} else {
1143 			/*
1144 			 *because hw will nerver use hw_rate
1145 			 *when tcb_desc->use_driver_rate = false
1146 			 *so we never set highest N rate here,
1147 			 *and N rate will all be controlled by FW
1148 			 *when tcb_desc->use_driver_rate = false
1149 			 */
1150 			if (sta && sta->vht_cap.vht_supported) {
1151 				tcb_desc->hw_rate =
1152 				_rtl_get_vht_highest_n_rate(hw, sta);
1153 			} else {
1154 				if (sta && (sta->ht_cap.ht_supported)) {
1155 					tcb_desc->hw_rate =
1156 						_rtl_get_highest_n_rate(hw, sta);
1157 				} else {
1158 					if (rtlmac->mode == WIRELESS_MODE_B) {
1159 						tcb_desc->hw_rate =
1160 						    rtlpriv->cfg->maps[RTL_RC_CCK_RATE11M];
1161 					} else {
1162 						tcb_desc->hw_rate =
1163 						    rtlpriv->cfg->maps[RTL_RC_OFDM_RATE54M];
1164 					}
1165 				}
1166 			}
1167 		}
1168 
1169 		if (is_multicast_ether_addr(ieee80211_get_DA(hdr)))
1170 			tcb_desc->multicast = 1;
1171 		else if (is_broadcast_ether_addr(ieee80211_get_DA(hdr)))
1172 			tcb_desc->broadcast = 1;
1173 
1174 		_rtl_txrate_selectmode(hw, sta, tcb_desc);
1175 		_rtl_query_bandwidth_mode(hw, sta, tcb_desc);
1176 		_rtl_qurey_shortpreamble_mode(hw, tcb_desc, info);
1177 		_rtl_query_shortgi(hw, sta, tcb_desc, info);
1178 		_rtl_query_protection_mode(hw, tcb_desc, info);
1179 	} else {
1180 		tcb_desc->use_driver_rate = true;
1181 		tcb_desc->ratr_index = RATR_INX_WIRELESS_MC;
1182 		tcb_desc->disable_ratefallback = 1;
1183 		tcb_desc->mac_id = 0;
1184 		tcb_desc->packet_bw = false;
1185 	}
1186 }
1187 EXPORT_SYMBOL(rtl_get_tcb_desc);
1188 
1189 bool rtl_tx_mgmt_proc(struct ieee80211_hw *hw, struct sk_buff *skb)
1190 {
1191 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
1192 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1193 	__le16 fc = rtl_get_fc(skb);
1194 
1195 	if (rtlpriv->dm.supp_phymode_switch &&
1196 	    mac->link_state < MAC80211_LINKED &&
1197 	    (ieee80211_is_auth(fc) || ieee80211_is_probe_req(fc))) {
1198 		if (rtlpriv->cfg->ops->chk_switch_dmdp)
1199 			rtlpriv->cfg->ops->chk_switch_dmdp(hw);
1200 	}
1201 	if (ieee80211_is_auth(fc)) {
1202 		RT_TRACE(rtlpriv, COMP_SEND, DBG_DMESG, "MAC80211_LINKING\n");
1203 		rtl_ips_nic_on(hw);
1204 
1205 		mac->link_state = MAC80211_LINKING;
1206 		/* Dul mac */
1207 		rtlpriv->phy.need_iqk = true;
1208 
1209 	}
1210 
1211 	return true;
1212 }
1213 EXPORT_SYMBOL_GPL(rtl_tx_mgmt_proc);
1214 
1215 struct sk_buff *rtl_make_del_ba(struct ieee80211_hw *hw, u8 *sa,
1216 				u8 *bssid, u16 tid);
1217 
1218 static void process_agg_start(struct ieee80211_hw *hw,
1219 			      struct ieee80211_hdr *hdr, u16 tid)
1220 {
1221 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1222 	struct ieee80211_rx_status rx_status = { 0 };
1223 	struct sk_buff *skb_delba = NULL;
1224 
1225 	skb_delba = rtl_make_del_ba(hw, hdr->addr2, hdr->addr3, tid);
1226 	if (skb_delba) {
1227 		rx_status.freq = hw->conf.chandef.chan->center_freq;
1228 		rx_status.band = hw->conf.chandef.chan->band;
1229 		rx_status.flag |= RX_FLAG_DECRYPTED;
1230 		rx_status.flag |= RX_FLAG_MACTIME_START;
1231 		rx_status.rate_idx = 0;
1232 		rx_status.signal = 50 + 10;
1233 		memcpy(IEEE80211_SKB_RXCB(skb_delba),
1234 		       &rx_status, sizeof(rx_status));
1235 		RT_PRINT_DATA(rtlpriv, COMP_INIT, DBG_DMESG,
1236 			      "fake del\n",
1237 			      skb_delba->data,
1238 			      skb_delba->len);
1239 		ieee80211_rx_irqsafe(hw, skb_delba);
1240 	}
1241 }
1242 
1243 bool rtl_action_proc(struct ieee80211_hw *hw, struct sk_buff *skb, u8 is_tx)
1244 {
1245 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
1246 	struct ieee80211_hdr *hdr = rtl_get_hdr(skb);
1247 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1248 	__le16 fc = rtl_get_fc(skb);
1249 	u8 *act = (u8 *)(((u8 *)skb->data + MAC80211_3ADDR_LEN));
1250 	u8 category;
1251 
1252 	if (!ieee80211_is_action(fc))
1253 		return true;
1254 
1255 	category = *act;
1256 	act++;
1257 	switch (category) {
1258 	case ACT_CAT_BA:
1259 		switch (*act) {
1260 		case ACT_ADDBAREQ:
1261 			if (mac->act_scanning)
1262 				return false;
1263 
1264 			RT_TRACE(rtlpriv, (COMP_SEND | COMP_RECV), DBG_DMESG,
1265 				"%s ACT_ADDBAREQ From :%pM\n",
1266 				is_tx ? "Tx" : "Rx", hdr->addr2);
1267 			RT_PRINT_DATA(rtlpriv, COMP_INIT, DBG_DMESG, "req\n",
1268 				skb->data, skb->len);
1269 			if (!is_tx) {
1270 				struct ieee80211_sta *sta = NULL;
1271 				struct rtl_sta_info *sta_entry = NULL;
1272 				struct rtl_tid_data *tid_data;
1273 				struct ieee80211_mgmt *mgmt = (void *)skb->data;
1274 				u16 capab = 0, tid = 0;
1275 
1276 				rcu_read_lock();
1277 				sta = rtl_find_sta(hw, hdr->addr3);
1278 				if (sta == NULL) {
1279 					RT_TRACE(rtlpriv, COMP_SEND | COMP_RECV,
1280 						 DBG_DMESG, "sta is NULL\n");
1281 					rcu_read_unlock();
1282 					return true;
1283 				}
1284 
1285 				sta_entry =
1286 					(struct rtl_sta_info *)sta->drv_priv;
1287 				if (!sta_entry) {
1288 					rcu_read_unlock();
1289 					return true;
1290 				}
1291 				capab =
1292 				  le16_to_cpu(mgmt->u.action.u.addba_req.capab);
1293 				tid = (capab &
1294 				       IEEE80211_ADDBA_PARAM_TID_MASK) >> 2;
1295 				tid_data = &sta_entry->tids[tid];
1296 				if (tid_data->agg.rx_agg_state ==
1297 				    RTL_RX_AGG_START)
1298 					process_agg_start(hw, hdr, tid);
1299 				rcu_read_unlock();
1300 			}
1301 			break;
1302 		case ACT_ADDBARSP:
1303 			RT_TRACE(rtlpriv, (COMP_SEND | COMP_RECV), DBG_DMESG,
1304 				 "%s ACT_ADDBARSP From :%pM\n",
1305 				  is_tx ? "Tx" : "Rx", hdr->addr2);
1306 			break;
1307 		case ACT_DELBA:
1308 			RT_TRACE(rtlpriv, (COMP_SEND | COMP_RECV), DBG_DMESG,
1309 				 "ACT_ADDBADEL From :%pM\n", hdr->addr2);
1310 			break;
1311 		}
1312 		break;
1313 	default:
1314 		break;
1315 	}
1316 
1317 	return true;
1318 }
1319 EXPORT_SYMBOL_GPL(rtl_action_proc);
1320 
1321 static void setup_special_tx(struct rtl_priv *rtlpriv, struct rtl_ps_ctl *ppsc,
1322 			     int type)
1323 {
1324 	struct ieee80211_hw *hw = rtlpriv->hw;
1325 
1326 	rtlpriv->ra.is_special_data = true;
1327 	if (rtlpriv->cfg->ops->get_btc_status())
1328 		rtlpriv->btcoexist.btc_ops->btc_special_packet_notify(
1329 					rtlpriv, type);
1330 	rtl_lps_leave(hw);
1331 	ppsc->last_delaylps_stamp_jiffies = jiffies;
1332 }
1333 
1334 static const u8 *rtl_skb_ether_type_ptr(struct ieee80211_hw *hw,
1335 					struct sk_buff *skb, bool is_enc)
1336 {
1337 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1338 	u8 mac_hdr_len = ieee80211_get_hdrlen_from_skb(skb);
1339 	u8 encrypt_header_len = 0;
1340 	u8 offset;
1341 
1342 	switch (rtlpriv->sec.pairwise_enc_algorithm) {
1343 	case WEP40_ENCRYPTION:
1344 	case WEP104_ENCRYPTION:
1345 		encrypt_header_len = 4;/*WEP_IV_LEN*/
1346 		break;
1347 	case TKIP_ENCRYPTION:
1348 		encrypt_header_len = 8;/*TKIP_IV_LEN*/
1349 		break;
1350 	case AESCCMP_ENCRYPTION:
1351 		encrypt_header_len = 8;/*CCMP_HDR_LEN;*/
1352 		break;
1353 	default:
1354 		break;
1355 	}
1356 
1357 	offset = mac_hdr_len + SNAP_SIZE;
1358 	if (is_enc)
1359 		offset += encrypt_header_len;
1360 
1361 	return skb->data + offset;
1362 }
1363 
1364 /*should call before software enc*/
1365 u8 rtl_is_special_data(struct ieee80211_hw *hw, struct sk_buff *skb, u8 is_tx,
1366 		       bool is_enc)
1367 {
1368 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1369 	struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
1370 	__le16 fc = rtl_get_fc(skb);
1371 	u16 ether_type;
1372 	const u8 *ether_type_ptr;
1373 	const struct iphdr *ip;
1374 
1375 	if (!ieee80211_is_data(fc))
1376 		goto end;
1377 
1378 	ether_type_ptr = rtl_skb_ether_type_ptr(hw, skb, is_enc);
1379 	ether_type = be16_to_cpup((__be16 *)ether_type_ptr);
1380 
1381 	if (ETH_P_IP == ether_type) {
1382 		ip = (struct iphdr *)((u8 *)ether_type_ptr +
1383 		     PROTOC_TYPE_SIZE);
1384 		if (IPPROTO_UDP == ip->protocol) {
1385 			struct udphdr *udp = (struct udphdr *)((u8 *)ip +
1386 							       (ip->ihl << 2));
1387 			if (((((u8 *)udp)[1] == 68) &&
1388 			     (((u8 *)udp)[3] == 67)) ||
1389 			    ((((u8 *)udp)[1] == 67) &&
1390 			     (((u8 *)udp)[3] == 68))) {
1391 				/* 68 : UDP BOOTP client
1392 				 * 67 : UDP BOOTP server
1393 				 */
1394 				RT_TRACE(rtlpriv, (COMP_SEND | COMP_RECV),
1395 					 DBG_DMESG, "dhcp %s !!\n",
1396 					 (is_tx) ? "Tx" : "Rx");
1397 
1398 				if (is_tx)
1399 					setup_special_tx(rtlpriv, ppsc,
1400 							 PACKET_DHCP);
1401 
1402 				return true;
1403 			}
1404 		}
1405 	} else if (ETH_P_ARP == ether_type) {
1406 		if (is_tx)
1407 			setup_special_tx(rtlpriv, ppsc, PACKET_ARP);
1408 
1409 		return true;
1410 	} else if (ETH_P_PAE == ether_type) {
1411 		RT_TRACE(rtlpriv, (COMP_SEND | COMP_RECV), DBG_DMESG,
1412 			 "802.1X %s EAPOL pkt!!\n", (is_tx) ? "Tx" : "Rx");
1413 
1414 		if (is_tx) {
1415 			rtlpriv->ra.is_special_data = true;
1416 			rtl_lps_leave(hw);
1417 			ppsc->last_delaylps_stamp_jiffies = jiffies;
1418 
1419 			setup_special_tx(rtlpriv, ppsc, PACKET_EAPOL);
1420 		}
1421 
1422 		return true;
1423 	} else if (ETH_P_IPV6 == ether_type) {
1424 		/* TODO: Handle any IPv6 cases that need special handling.
1425 		 * For now, always return false
1426 		 */
1427 		goto end;
1428 	}
1429 
1430 end:
1431 	rtlpriv->ra.is_special_data = false;
1432 	return false;
1433 }
1434 EXPORT_SYMBOL_GPL(rtl_is_special_data);
1435 
1436 bool rtl_is_tx_report_skb(struct ieee80211_hw *hw, struct sk_buff *skb)
1437 {
1438 	u16 ether_type;
1439 	const u8 *ether_type_ptr;
1440 
1441 	ether_type_ptr = rtl_skb_ether_type_ptr(hw, skb, true);
1442 	ether_type = be16_to_cpup((__be16 *)ether_type_ptr);
1443 
1444 	/* EAPOL */
1445 	if (ether_type == ETH_P_PAE)
1446 		return true;
1447 
1448 	return false;
1449 }
1450 
1451 static u16 rtl_get_tx_report_sn(struct ieee80211_hw *hw)
1452 {
1453 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1454 	struct rtl_tx_report *tx_report = &rtlpriv->tx_report;
1455 	u16 sn;
1456 
1457 	sn = atomic_inc_return(&tx_report->sn) & 0x0FFF;
1458 
1459 	tx_report->last_sent_sn = sn;
1460 	tx_report->last_sent_time = jiffies;
1461 
1462 	RT_TRACE(rtlpriv, COMP_TX_REPORT, DBG_DMESG,
1463 		 "Send TX-Report sn=0x%X\n", sn);
1464 
1465 	return sn;
1466 }
1467 
1468 void rtl_get_tx_report(struct rtl_tcb_desc *ptcb_desc, u8 *pdesc,
1469 		       struct ieee80211_hw *hw)
1470 {
1471 	if (ptcb_desc->use_spe_rpt) {
1472 		u16 sn = rtl_get_tx_report_sn(hw);
1473 
1474 		SET_TX_DESC_SPE_RPT(pdesc, 1);
1475 		SET_TX_DESC_SW_DEFINE(pdesc, sn);
1476 	}
1477 }
1478 EXPORT_SYMBOL_GPL(rtl_get_tx_report);
1479 
1480 void rtl_tx_report_handler(struct ieee80211_hw *hw, u8 *tmp_buf, u8 c2h_cmd_len)
1481 {
1482 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1483 	struct rtl_tx_report *tx_report = &rtlpriv->tx_report;
1484 	u16 sn;
1485 
1486 	sn = ((tmp_buf[7] & 0x0F) << 8) | tmp_buf[6];
1487 
1488 	tx_report->last_recv_sn = sn;
1489 
1490 	RT_TRACE(rtlpriv, COMP_TX_REPORT, DBG_DMESG,
1491 		 "Recv TX-Report st=0x%02X sn=0x%X retry=0x%X\n",
1492 		 tmp_buf[0], sn, tmp_buf[2]);
1493 }
1494 EXPORT_SYMBOL_GPL(rtl_tx_report_handler);
1495 
1496 bool rtl_check_tx_report_acked(struct ieee80211_hw *hw)
1497 {
1498 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1499 	struct rtl_tx_report *tx_report = &rtlpriv->tx_report;
1500 
1501 	if (tx_report->last_sent_sn == tx_report->last_recv_sn)
1502 		return true;
1503 
1504 	if (time_before(tx_report->last_sent_time + 3 * HZ, jiffies)) {
1505 		RT_TRACE(rtlpriv, COMP_TX_REPORT, DBG_WARNING,
1506 			 "Check TX-Report timeout!!\n");
1507 		return true;	/* 3 sec. (timeout) seen as acked */
1508 	}
1509 
1510 	return false;
1511 }
1512 
1513 void rtl_wait_tx_report_acked(struct ieee80211_hw *hw, u32 wait_ms)
1514 {
1515 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1516 	int i;
1517 
1518 	for (i = 0; i < wait_ms; i++) {
1519 		if (rtl_check_tx_report_acked(hw))
1520 			break;
1521 		usleep_range(1000, 2000);
1522 		RT_TRACE(rtlpriv, COMP_SEC, DBG_DMESG,
1523 			 "Wait 1ms (%d/%d) to disable key.\n", i, wait_ms);
1524 	}
1525 }
1526 /*********************************************************
1527  *
1528  * functions called by core.c
1529  *
1530  *********************************************************/
1531 int rtl_tx_agg_start(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
1532 		     struct ieee80211_sta *sta, u16 tid, u16 *ssn)
1533 {
1534 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1535 	struct rtl_tid_data *tid_data;
1536 	struct rtl_sta_info *sta_entry = NULL;
1537 
1538 	if (sta == NULL)
1539 		return -EINVAL;
1540 
1541 	if (unlikely(tid >= MAX_TID_COUNT))
1542 		return -EINVAL;
1543 
1544 	sta_entry = (struct rtl_sta_info *)sta->drv_priv;
1545 	if (!sta_entry)
1546 		return -ENXIO;
1547 	tid_data = &sta_entry->tids[tid];
1548 
1549 	RT_TRACE(rtlpriv, COMP_SEND, DBG_DMESG,
1550 		 "on ra = %pM tid = %d seq:%d\n", sta->addr, tid,
1551 		 tid_data->seq_number);
1552 
1553 	*ssn = tid_data->seq_number;
1554 	tid_data->agg.agg_state = RTL_AGG_START;
1555 
1556 	ieee80211_start_tx_ba_cb_irqsafe(vif, sta->addr, tid);
1557 	return 0;
1558 }
1559 
1560 int rtl_tx_agg_stop(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
1561 		    struct ieee80211_sta *sta, u16 tid)
1562 {
1563 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1564 	struct rtl_tid_data *tid_data;
1565 	struct rtl_sta_info *sta_entry = NULL;
1566 
1567 	if (sta == NULL)
1568 		return -EINVAL;
1569 
1570 	RT_TRACE(rtlpriv, COMP_SEND, DBG_DMESG,
1571 		 "on ra = %pM tid = %d\n", sta->addr, tid);
1572 
1573 	if (unlikely(tid >= MAX_TID_COUNT))
1574 		return -EINVAL;
1575 
1576 	sta_entry = (struct rtl_sta_info *)sta->drv_priv;
1577 	tid_data = &sta_entry->tids[tid];
1578 	sta_entry->tids[tid].agg.agg_state = RTL_AGG_STOP;
1579 
1580 	ieee80211_stop_tx_ba_cb_irqsafe(vif, sta->addr, tid);
1581 	return 0;
1582 }
1583 
1584 int rtl_rx_agg_start(struct ieee80211_hw *hw,
1585 		     struct ieee80211_sta *sta, u16 tid)
1586 {
1587 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1588 	struct rtl_tid_data *tid_data;
1589 	struct rtl_sta_info *sta_entry = NULL;
1590 	u8 reject_agg;
1591 
1592 	if (sta == NULL)
1593 		return -EINVAL;
1594 
1595 	if (unlikely(tid >= MAX_TID_COUNT))
1596 		return -EINVAL;
1597 
1598 	if (rtlpriv->cfg->ops->get_btc_status()) {
1599 		rtlpriv->btcoexist.btc_ops->btc_get_ampdu_cfg(rtlpriv,
1600 							      &reject_agg,
1601 							      NULL, NULL);
1602 		if (reject_agg)
1603 			return -EINVAL;
1604 	}
1605 
1606 	sta_entry = (struct rtl_sta_info *)sta->drv_priv;
1607 	if (!sta_entry)
1608 		return -ENXIO;
1609 	tid_data = &sta_entry->tids[tid];
1610 
1611 	RT_TRACE(rtlpriv, COMP_RECV, DBG_DMESG,
1612 		 "on ra = %pM tid = %d seq:%d\n", sta->addr, tid,
1613 		 tid_data->seq_number);
1614 
1615 	tid_data->agg.rx_agg_state = RTL_RX_AGG_START;
1616 	return 0;
1617 }
1618 
1619 int rtl_rx_agg_stop(struct ieee80211_hw *hw,
1620 		    struct ieee80211_sta *sta, u16 tid)
1621 {
1622 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1623 	struct rtl_sta_info *sta_entry = NULL;
1624 
1625 	if (sta == NULL)
1626 		return -EINVAL;
1627 
1628 	RT_TRACE(rtlpriv, COMP_SEND, DBG_DMESG,
1629 		 "on ra = %pM tid = %d\n", sta->addr, tid);
1630 
1631 	if (unlikely(tid >= MAX_TID_COUNT))
1632 		return -EINVAL;
1633 
1634 	sta_entry = (struct rtl_sta_info *)sta->drv_priv;
1635 	sta_entry->tids[tid].agg.rx_agg_state = RTL_RX_AGG_STOP;
1636 
1637 	return 0;
1638 }
1639 int rtl_tx_agg_oper(struct ieee80211_hw *hw,
1640 		struct ieee80211_sta *sta, u16 tid)
1641 {
1642 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1643 	struct rtl_sta_info *sta_entry = NULL;
1644 
1645 	if (sta == NULL)
1646 		return -EINVAL;
1647 
1648 	RT_TRACE(rtlpriv, COMP_SEND, DBG_DMESG,
1649 		 "on ra = %pM tid = %d\n", sta->addr, tid);
1650 
1651 	if (unlikely(tid >= MAX_TID_COUNT))
1652 		return -EINVAL;
1653 
1654 	sta_entry = (struct rtl_sta_info *)sta->drv_priv;
1655 	sta_entry->tids[tid].agg.agg_state = RTL_AGG_OPERATIONAL;
1656 
1657 	return 0;
1658 }
1659 
1660 void rtl_rx_ampdu_apply(struct rtl_priv *rtlpriv)
1661 {
1662 	struct rtl_btc_ops *btc_ops = rtlpriv->btcoexist.btc_ops;
1663 	u8 reject_agg, ctrl_agg_size = 0, agg_size;
1664 
1665 	if (rtlpriv->cfg->ops->get_btc_status())
1666 		btc_ops->btc_get_ampdu_cfg(rtlpriv, &reject_agg,
1667 					   &ctrl_agg_size, &agg_size);
1668 
1669 	RT_TRACE(rtlpriv, COMP_BT_COEXIST, DBG_DMESG,
1670 		 "Set RX AMPDU: coex - reject=%d, ctrl_agg_size=%d, size=%d",
1671 		 reject_agg, ctrl_agg_size, agg_size);
1672 
1673 	rtlpriv->hw->max_rx_aggregation_subframes =
1674 		(ctrl_agg_size ? agg_size : IEEE80211_MAX_AMPDU_BUF);
1675 }
1676 EXPORT_SYMBOL(rtl_rx_ampdu_apply);
1677 
1678 /*********************************************************
1679  *
1680  * wq & timer callback functions
1681  *
1682  *********************************************************/
1683 /* this function is used for roaming */
1684 void rtl_beacon_statistic(struct ieee80211_hw *hw, struct sk_buff *skb)
1685 {
1686 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1687 	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
1688 
1689 	if (rtlpriv->mac80211.opmode != NL80211_IFTYPE_STATION)
1690 		return;
1691 
1692 	if (rtlpriv->mac80211.link_state < MAC80211_LINKED)
1693 		return;
1694 
1695 	/* check if this really is a beacon */
1696 	if (!ieee80211_is_beacon(hdr->frame_control) &&
1697 	    !ieee80211_is_probe_resp(hdr->frame_control))
1698 		return;
1699 
1700 	/* min. beacon length + FCS_LEN */
1701 	if (skb->len <= 40 + FCS_LEN)
1702 		return;
1703 
1704 	/* and only beacons from the associated BSSID, please */
1705 	if (!ether_addr_equal(hdr->addr3, rtlpriv->mac80211.bssid))
1706 		return;
1707 
1708 	rtlpriv->link_info.bcn_rx_inperiod++;
1709 }
1710 EXPORT_SYMBOL_GPL(rtl_beacon_statistic);
1711 
1712 static void rtl_free_entries_from_scan_list(struct ieee80211_hw *hw)
1713 {
1714 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1715 	struct rtl_bssid_entry *entry, *next;
1716 
1717 	list_for_each_entry_safe(entry, next, &rtlpriv->scan_list.list, list) {
1718 		list_del(&entry->list);
1719 		kfree(entry);
1720 		rtlpriv->scan_list.num--;
1721 	}
1722 }
1723 
1724 void rtl_scan_list_expire(struct ieee80211_hw *hw)
1725 {
1726 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1727 	struct rtl_bssid_entry *entry, *next;
1728 	unsigned long flags;
1729 
1730 	spin_lock_irqsave(&rtlpriv->locks.scan_list_lock, flags);
1731 
1732 	list_for_each_entry_safe(entry, next, &rtlpriv->scan_list.list, list) {
1733 		/* 180 seconds */
1734 		if (jiffies_to_msecs(jiffies - entry->age) < 180000)
1735 			continue;
1736 
1737 		list_del(&entry->list);
1738 		kfree(entry);
1739 		rtlpriv->scan_list.num--;
1740 
1741 		RT_TRACE(rtlpriv, COMP_SCAN, DBG_LOUD,
1742 			 "BSSID=%pM is expire in scan list (total=%d)\n",
1743 			 entry->bssid, rtlpriv->scan_list.num);
1744 	}
1745 
1746 	spin_unlock_irqrestore(&rtlpriv->locks.scan_list_lock, flags);
1747 
1748 	rtlpriv->btcoexist.btc_info.ap_num = rtlpriv->scan_list.num;
1749 }
1750 
1751 void rtl_collect_scan_list(struct ieee80211_hw *hw, struct sk_buff *skb)
1752 {
1753 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1754 	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
1755 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
1756 	unsigned long flags;
1757 
1758 	struct rtl_bssid_entry *entry;
1759 	bool entry_found = false;
1760 
1761 	/* check if it is scanning */
1762 	if (!mac->act_scanning)
1763 		return;
1764 
1765 	/* check if this really is a beacon */
1766 	if (!ieee80211_is_beacon(hdr->frame_control) &&
1767 	    !ieee80211_is_probe_resp(hdr->frame_control))
1768 		return;
1769 
1770 	spin_lock_irqsave(&rtlpriv->locks.scan_list_lock, flags);
1771 
1772 	list_for_each_entry(entry, &rtlpriv->scan_list.list, list) {
1773 		if (memcmp(entry->bssid, hdr->addr3, ETH_ALEN) == 0) {
1774 			list_del_init(&entry->list);
1775 			entry_found = true;
1776 			RT_TRACE(rtlpriv, COMP_SCAN, DBG_LOUD,
1777 				 "Update BSSID=%pM to scan list (total=%d)\n",
1778 				 hdr->addr3, rtlpriv->scan_list.num);
1779 			break;
1780 		}
1781 	}
1782 
1783 	if (!entry_found) {
1784 		entry = kmalloc(sizeof(*entry), GFP_ATOMIC);
1785 
1786 		if (!entry)
1787 			goto label_err;
1788 
1789 		memcpy(entry->bssid, hdr->addr3, ETH_ALEN);
1790 		rtlpriv->scan_list.num++;
1791 
1792 		RT_TRACE(rtlpriv, COMP_SCAN, DBG_LOUD,
1793 			 "Add BSSID=%pM to scan list (total=%d)\n",
1794 			 hdr->addr3, rtlpriv->scan_list.num);
1795 	}
1796 
1797 	entry->age = jiffies;
1798 
1799 	list_add_tail(&entry->list, &rtlpriv->scan_list.list);
1800 
1801 label_err:
1802 	spin_unlock_irqrestore(&rtlpriv->locks.scan_list_lock, flags);
1803 }
1804 EXPORT_SYMBOL(rtl_collect_scan_list);
1805 
1806 void rtl_watchdog_wq_callback(void *data)
1807 {
1808 	struct rtl_works *rtlworks = container_of_dwork_rtl(data,
1809 							    struct rtl_works,
1810 							    watchdog_wq);
1811 	struct ieee80211_hw *hw = rtlworks->hw;
1812 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1813 	struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
1814 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
1815 	bool busytraffic = false;
1816 	bool tx_busy_traffic = false;
1817 	bool rx_busy_traffic = false;
1818 	bool higher_busytraffic = false;
1819 	bool higher_busyrxtraffic = false;
1820 	u8 idx, tid;
1821 	u32 rx_cnt_inp4eriod = 0;
1822 	u32 tx_cnt_inp4eriod = 0;
1823 	u32 aver_rx_cnt_inperiod = 0;
1824 	u32 aver_tx_cnt_inperiod = 0;
1825 	u32 aver_tidtx_inperiod[MAX_TID_COUNT] = {0};
1826 	u32 tidtx_inp4eriod[MAX_TID_COUNT] = {0};
1827 
1828 	if (is_hal_stop(rtlhal))
1829 		return;
1830 
1831 	/* <1> Determine if action frame is allowed */
1832 	if (mac->link_state > MAC80211_NOLINK) {
1833 		if (mac->cnt_after_linked < 20)
1834 			mac->cnt_after_linked++;
1835 	} else {
1836 		mac->cnt_after_linked = 0;
1837 	}
1838 
1839 	/* <2> to check if traffic busy, if
1840 	 * busytraffic we don't change channel
1841 	 */
1842 	if (mac->link_state >= MAC80211_LINKED) {
1843 
1844 		/* (1) get aver_rx_cnt_inperiod & aver_tx_cnt_inperiod */
1845 		for (idx = 0; idx <= 2; idx++) {
1846 			rtlpriv->link_info.num_rx_in4period[idx] =
1847 			    rtlpriv->link_info.num_rx_in4period[idx + 1];
1848 			rtlpriv->link_info.num_tx_in4period[idx] =
1849 			    rtlpriv->link_info.num_tx_in4period[idx + 1];
1850 		}
1851 		rtlpriv->link_info.num_rx_in4period[3] =
1852 		    rtlpriv->link_info.num_rx_inperiod;
1853 		rtlpriv->link_info.num_tx_in4period[3] =
1854 		    rtlpriv->link_info.num_tx_inperiod;
1855 		for (idx = 0; idx <= 3; idx++) {
1856 			rx_cnt_inp4eriod +=
1857 			    rtlpriv->link_info.num_rx_in4period[idx];
1858 			tx_cnt_inp4eriod +=
1859 			    rtlpriv->link_info.num_tx_in4period[idx];
1860 		}
1861 		aver_rx_cnt_inperiod = rx_cnt_inp4eriod / 4;
1862 		aver_tx_cnt_inperiod = tx_cnt_inp4eriod / 4;
1863 
1864 		/* (2) check traffic busy */
1865 		if (aver_rx_cnt_inperiod > 100 || aver_tx_cnt_inperiod > 100) {
1866 			busytraffic = true;
1867 			if (aver_rx_cnt_inperiod > aver_tx_cnt_inperiod)
1868 				rx_busy_traffic = true;
1869 			else
1870 				tx_busy_traffic = false;
1871 		}
1872 
1873 		/* Higher Tx/Rx data. */
1874 		if (aver_rx_cnt_inperiod > 4000 ||
1875 		    aver_tx_cnt_inperiod > 4000) {
1876 			higher_busytraffic = true;
1877 
1878 			/* Extremely high Rx data. */
1879 			if (aver_rx_cnt_inperiod > 5000)
1880 				higher_busyrxtraffic = true;
1881 		}
1882 
1883 		/* check every tid's tx traffic */
1884 		for (tid = 0; tid <= 7; tid++) {
1885 			for (idx = 0; idx <= 2; idx++)
1886 				rtlpriv->link_info.tidtx_in4period[tid][idx] =
1887 					rtlpriv->link_info.tidtx_in4period[tid]
1888 					[idx + 1];
1889 			rtlpriv->link_info.tidtx_in4period[tid][3] =
1890 				rtlpriv->link_info.tidtx_inperiod[tid];
1891 
1892 			for (idx = 0; idx <= 3; idx++)
1893 				tidtx_inp4eriod[tid] +=
1894 				   rtlpriv->link_info.tidtx_in4period[tid][idx];
1895 			aver_tidtx_inperiod[tid] = tidtx_inp4eriod[tid] / 4;
1896 			if (aver_tidtx_inperiod[tid] > 5000)
1897 				rtlpriv->link_info.higher_busytxtraffic[tid] =
1898 									true;
1899 			else
1900 				rtlpriv->link_info.higher_busytxtraffic[tid] =
1901 									false;
1902 		}
1903 
1904 		/* PS is controlled by coex. */
1905 		if (rtlpriv->cfg->ops->get_btc_status() &&
1906 		    rtlpriv->btcoexist.btc_ops->btc_is_bt_ctrl_lps(rtlpriv))
1907 			goto label_lps_done;
1908 
1909 		if (((rtlpriv->link_info.num_rx_inperiod +
1910 		      rtlpriv->link_info.num_tx_inperiod) > 8) ||
1911 		    (rtlpriv->link_info.num_rx_inperiod > 2))
1912 			rtl_lps_leave(hw);
1913 		else
1914 			rtl_lps_enter(hw);
1915 
1916 label_lps_done:
1917 		;
1918 	}
1919 
1920 	rtlpriv->link_info.num_rx_inperiod = 0;
1921 	rtlpriv->link_info.num_tx_inperiod = 0;
1922 	for (tid = 0; tid <= 7; tid++)
1923 		rtlpriv->link_info.tidtx_inperiod[tid] = 0;
1924 
1925 	rtlpriv->link_info.busytraffic = busytraffic;
1926 	rtlpriv->link_info.higher_busytraffic = higher_busytraffic;
1927 	rtlpriv->link_info.rx_busy_traffic = rx_busy_traffic;
1928 	rtlpriv->link_info.tx_busy_traffic = tx_busy_traffic;
1929 	rtlpriv->link_info.higher_busyrxtraffic = higher_busyrxtraffic;
1930 
1931 	/* <3> DM */
1932 	if (!rtlpriv->cfg->mod_params->disable_watchdog)
1933 		rtlpriv->cfg->ops->dm_watchdog(hw);
1934 
1935 	/* <4> roaming */
1936 	if (mac->link_state == MAC80211_LINKED &&
1937 	    mac->opmode == NL80211_IFTYPE_STATION) {
1938 		if ((rtlpriv->link_info.bcn_rx_inperiod +
1939 		    rtlpriv->link_info.num_rx_inperiod) == 0) {
1940 			rtlpriv->link_info.roam_times++;
1941 			RT_TRACE(rtlpriv, COMP_ERR, DBG_DMESG,
1942 				 "AP off for %d s\n",
1943 				(rtlpriv->link_info.roam_times * 2));
1944 
1945 			/* if we can't recv beacon for 10s,
1946 			 * we should reconnect this AP
1947 			 */
1948 			if (rtlpriv->link_info.roam_times >= 5) {
1949 				pr_err("AP off, try to reconnect now\n");
1950 				rtlpriv->link_info.roam_times = 0;
1951 				ieee80211_connection_loss(
1952 					rtlpriv->mac80211.vif);
1953 			}
1954 		} else {
1955 			rtlpriv->link_info.roam_times = 0;
1956 		}
1957 	}
1958 
1959 	if (rtlpriv->cfg->ops->get_btc_status())
1960 		rtlpriv->btcoexist.btc_ops->btc_periodical(rtlpriv);
1961 
1962 	rtlpriv->link_info.bcn_rx_inperiod = 0;
1963 
1964 	/* <6> scan list */
1965 	rtl_scan_list_expire(hw);
1966 }
1967 
1968 void rtl_watch_dog_timer_callback(unsigned long data)
1969 {
1970 	struct ieee80211_hw *hw = (struct ieee80211_hw *)data;
1971 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1972 
1973 	queue_delayed_work(rtlpriv->works.rtl_wq,
1974 			   &rtlpriv->works.watchdog_wq, 0);
1975 
1976 	mod_timer(&rtlpriv->works.watchdog_timer,
1977 		  jiffies + MSECS(RTL_WATCH_DOG_TIME));
1978 }
1979 void rtl_fwevt_wq_callback(void *data)
1980 {
1981 	struct rtl_works *rtlworks =
1982 		container_of_dwork_rtl(data, struct rtl_works, fwevt_wq);
1983 	struct ieee80211_hw *hw = rtlworks->hw;
1984 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1985 
1986 	rtlpriv->cfg->ops->c2h_command_handle(hw);
1987 }
1988 
1989 void rtl_c2hcmd_enqueue(struct ieee80211_hw *hw, u8 tag, u8 len, u8 *val)
1990 {
1991 	struct rtl_priv *rtlpriv = rtl_priv(hw);
1992 	unsigned long flags;
1993 	struct rtl_c2hcmd *c2hcmd;
1994 
1995 	c2hcmd = kmalloc(sizeof(*c2hcmd),
1996 			 in_interrupt() ? GFP_ATOMIC : GFP_KERNEL);
1997 
1998 	if (!c2hcmd)
1999 		goto label_err;
2000 
2001 	c2hcmd->val = kmalloc(len,
2002 			      in_interrupt() ? GFP_ATOMIC : GFP_KERNEL);
2003 
2004 	if (!c2hcmd->val)
2005 		goto label_err2;
2006 
2007 	/* fill data */
2008 	c2hcmd->tag = tag;
2009 	c2hcmd->len = len;
2010 	memcpy(c2hcmd->val, val, len);
2011 
2012 	/* enqueue */
2013 	spin_lock_irqsave(&rtlpriv->locks.c2hcmd_lock, flags);
2014 
2015 	list_add_tail(&c2hcmd->list, &rtlpriv->c2hcmd_list);
2016 
2017 	spin_unlock_irqrestore(&rtlpriv->locks.c2hcmd_lock, flags);
2018 
2019 	/* wake up wq */
2020 	queue_delayed_work(rtlpriv->works.rtl_wq, &rtlpriv->works.c2hcmd_wq, 0);
2021 
2022 	return;
2023 
2024 label_err2:
2025 	kfree(c2hcmd);
2026 
2027 label_err:
2028 	RT_TRACE(rtlpriv, COMP_CMD, DBG_WARNING,
2029 		 "C2H cmd enqueue fail.\n");
2030 }
2031 EXPORT_SYMBOL(rtl_c2hcmd_enqueue);
2032 
2033 void rtl_c2hcmd_launcher(struct ieee80211_hw *hw, int exec)
2034 {
2035 	struct rtl_priv *rtlpriv = rtl_priv(hw);
2036 	unsigned long flags;
2037 	struct rtl_c2hcmd *c2hcmd;
2038 	int i;
2039 
2040 	for (i = 0; i < 200; i++) {
2041 		/* dequeue a task */
2042 		spin_lock_irqsave(&rtlpriv->locks.c2hcmd_lock, flags);
2043 
2044 		c2hcmd = list_first_entry_or_null(&rtlpriv->c2hcmd_list,
2045 						  struct rtl_c2hcmd, list);
2046 
2047 		if (c2hcmd)
2048 			list_del(&c2hcmd->list);
2049 
2050 		spin_unlock_irqrestore(&rtlpriv->locks.c2hcmd_lock, flags);
2051 
2052 		/* do it */
2053 		if (!c2hcmd)
2054 			break;
2055 
2056 		if (rtlpriv->cfg->ops->c2h_content_parsing && exec)
2057 			rtlpriv->cfg->ops->c2h_content_parsing(hw,
2058 					c2hcmd->tag, c2hcmd->len, c2hcmd->val);
2059 
2060 		/* free */
2061 		kfree(c2hcmd->val);
2062 
2063 		kfree(c2hcmd);
2064 	}
2065 }
2066 
2067 void rtl_c2hcmd_wq_callback(void *data)
2068 {
2069 	struct rtl_works *rtlworks = container_of_dwork_rtl(data,
2070 							    struct rtl_works,
2071 							    c2hcmd_wq);
2072 	struct ieee80211_hw *hw = rtlworks->hw;
2073 
2074 	rtl_c2hcmd_launcher(hw, 1);
2075 }
2076 
2077 void rtl_easy_concurrent_retrytimer_callback(unsigned long data)
2078 {
2079 	struct ieee80211_hw *hw = (struct ieee80211_hw *)data;
2080 	struct rtl_priv *rtlpriv = rtl_priv(hw);
2081 	struct rtl_priv *buddy_priv = rtlpriv->buddy_priv;
2082 
2083 	if (buddy_priv == NULL)
2084 		return;
2085 
2086 	rtlpriv->cfg->ops->dualmac_easy_concurrent(hw);
2087 }
2088 /*********************************************************
2089  *
2090  * frame process functions
2091  *
2092  *********************************************************/
2093 u8 *rtl_find_ie(u8 *data, unsigned int len, u8 ie)
2094 {
2095 	struct ieee80211_mgmt *mgmt = (void *)data;
2096 	u8 *pos, *end;
2097 
2098 	pos = (u8 *)mgmt->u.beacon.variable;
2099 	end = data + len;
2100 	while (pos < end) {
2101 		if (pos + 2 + pos[1] > end)
2102 			return NULL;
2103 
2104 		if (pos[0] == ie)
2105 			return pos;
2106 
2107 		pos += 2 + pos[1];
2108 	}
2109 	return NULL;
2110 }
2111 
2112 /* when we use 2 rx ants we send IEEE80211_SMPS_OFF */
2113 /* when we use 1 rx ant we send IEEE80211_SMPS_STATIC */
2114 static struct sk_buff *rtl_make_smps_action(struct ieee80211_hw *hw,
2115 				     enum ieee80211_smps_mode smps,
2116 				     u8 *da, u8 *bssid)
2117 {
2118 	struct rtl_efuse *rtlefuse = rtl_efuse(rtl_priv(hw));
2119 	struct sk_buff *skb;
2120 	struct ieee80211_mgmt *action_frame;
2121 
2122 	/* 27 = header + category + action + smps mode */
2123 	skb = dev_alloc_skb(27 + hw->extra_tx_headroom);
2124 	if (!skb)
2125 		return NULL;
2126 
2127 	skb_reserve(skb, hw->extra_tx_headroom);
2128 	action_frame = skb_put_zero(skb, 27);
2129 	memcpy(action_frame->da, da, ETH_ALEN);
2130 	memcpy(action_frame->sa, rtlefuse->dev_addr, ETH_ALEN);
2131 	memcpy(action_frame->bssid, bssid, ETH_ALEN);
2132 	action_frame->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
2133 						  IEEE80211_STYPE_ACTION);
2134 	action_frame->u.action.category = WLAN_CATEGORY_HT;
2135 	action_frame->u.action.u.ht_smps.action = WLAN_HT_ACTION_SMPS;
2136 	switch (smps) {
2137 	case IEEE80211_SMPS_AUTOMATIC:/* 0 */
2138 	case IEEE80211_SMPS_NUM_MODES:/* 4 */
2139 		WARN_ON(1);
2140 	/* Here will get a 'MISSING_BREAK' in Coverity Test, just ignore it.
2141 	 * According to Kernel Code, here is right.
2142 	 */
2143 	case IEEE80211_SMPS_OFF:/* 1 */ /*MIMO_PS_NOLIMIT*/
2144 		action_frame->u.action.u.ht_smps.smps_control =
2145 				WLAN_HT_SMPS_CONTROL_DISABLED;/* 0 */
2146 		break;
2147 	case IEEE80211_SMPS_STATIC:/* 2 */ /*MIMO_PS_STATIC*/
2148 		action_frame->u.action.u.ht_smps.smps_control =
2149 				WLAN_HT_SMPS_CONTROL_STATIC;/* 1 */
2150 		break;
2151 	case IEEE80211_SMPS_DYNAMIC:/* 3 */ /*MIMO_PS_DYNAMIC*/
2152 		action_frame->u.action.u.ht_smps.smps_control =
2153 				WLAN_HT_SMPS_CONTROL_DYNAMIC;/* 3 */
2154 		break;
2155 	}
2156 
2157 	return skb;
2158 }
2159 
2160 int rtl_send_smps_action(struct ieee80211_hw *hw,
2161 			 struct ieee80211_sta *sta,
2162 			 enum ieee80211_smps_mode smps)
2163 {
2164 	struct rtl_priv *rtlpriv = rtl_priv(hw);
2165 	struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
2166 	struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
2167 	struct sk_buff *skb = NULL;
2168 	struct rtl_tcb_desc tcb_desc;
2169 	u8 bssid[ETH_ALEN] = {0};
2170 
2171 	memset(&tcb_desc, 0, sizeof(struct rtl_tcb_desc));
2172 
2173 	if (rtlpriv->mac80211.act_scanning)
2174 		goto err_free;
2175 
2176 	if (!sta)
2177 		goto err_free;
2178 
2179 	if (unlikely(is_hal_stop(rtlhal) || ppsc->rfpwr_state != ERFON))
2180 		goto err_free;
2181 
2182 	if (!test_bit(RTL_STATUS_INTERFACE_START, &rtlpriv->status))
2183 		goto err_free;
2184 
2185 	if (rtlpriv->mac80211.opmode == NL80211_IFTYPE_AP)
2186 		memcpy(bssid, rtlpriv->efuse.dev_addr, ETH_ALEN);
2187 	else
2188 		memcpy(bssid, rtlpriv->mac80211.bssid, ETH_ALEN);
2189 
2190 	skb = rtl_make_smps_action(hw, smps, sta->addr, bssid);
2191 	/* this is a type = mgmt * stype = action frame */
2192 	if (skb) {
2193 		struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
2194 		struct rtl_sta_info *sta_entry =
2195 			(struct rtl_sta_info *) sta->drv_priv;
2196 		sta_entry->mimo_ps = smps;
2197 		/* rtlpriv->cfg->ops->update_rate_tbl(hw, sta, 0); */
2198 
2199 		info->control.rates[0].idx = 0;
2200 		info->band = hw->conf.chandef.chan->band;
2201 		rtlpriv->intf_ops->adapter_tx(hw, sta, skb, &tcb_desc);
2202 	}
2203 	return 1;
2204 
2205 err_free:
2206 	return 0;
2207 }
2208 EXPORT_SYMBOL(rtl_send_smps_action);
2209 
2210 void rtl_phy_scan_operation_backup(struct ieee80211_hw *hw, u8 operation)
2211 {
2212 	struct rtl_priv *rtlpriv = rtl_priv(hw);
2213 	struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
2214 	enum io_type iotype;
2215 
2216 	if (!is_hal_stop(rtlhal)) {
2217 		switch (operation) {
2218 		case SCAN_OPT_BACKUP:
2219 			iotype = IO_CMD_PAUSE_DM_BY_SCAN;
2220 			rtlpriv->cfg->ops->set_hw_reg(hw,
2221 						      HW_VAR_IO_CMD,
2222 						      (u8 *)&iotype);
2223 			break;
2224 		case SCAN_OPT_RESTORE:
2225 			iotype = IO_CMD_RESUME_DM_BY_SCAN;
2226 			rtlpriv->cfg->ops->set_hw_reg(hw,
2227 						      HW_VAR_IO_CMD,
2228 						      (u8 *)&iotype);
2229 			break;
2230 		default:
2231 			pr_err("Unknown Scan Backup operation.\n");
2232 			break;
2233 		}
2234 	}
2235 }
2236 EXPORT_SYMBOL(rtl_phy_scan_operation_backup);
2237 
2238 /* because mac80211 have issues when can receive del ba
2239  * so here we just make a fake del_ba if we receive a ba_req
2240  * but rx_agg was opened to let mac80211 release some ba
2241  * related resources, so please this del_ba for tx
2242  */
2243 struct sk_buff *rtl_make_del_ba(struct ieee80211_hw *hw,
2244 				u8 *sa, u8 *bssid, u16 tid)
2245 {
2246 	struct rtl_efuse *rtlefuse = rtl_efuse(rtl_priv(hw));
2247 	struct sk_buff *skb;
2248 	struct ieee80211_mgmt *action_frame;
2249 	u16 params;
2250 
2251 	/* 27 = header + category + action + smps mode */
2252 	skb = dev_alloc_skb(34 + hw->extra_tx_headroom);
2253 	if (!skb)
2254 		return NULL;
2255 
2256 	skb_reserve(skb, hw->extra_tx_headroom);
2257 	action_frame = skb_put_zero(skb, 34);
2258 	memcpy(action_frame->sa, sa, ETH_ALEN);
2259 	memcpy(action_frame->da, rtlefuse->dev_addr, ETH_ALEN);
2260 	memcpy(action_frame->bssid, bssid, ETH_ALEN);
2261 	action_frame->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
2262 						  IEEE80211_STYPE_ACTION);
2263 	action_frame->u.action.category = WLAN_CATEGORY_BACK;
2264 	action_frame->u.action.u.delba.action_code = WLAN_ACTION_DELBA;
2265 	params = (u16)(1 << 11);	/* bit 11 initiator */
2266 	params |= (u16)(tid << 12);	/* bit 15:12 TID number */
2267 
2268 	action_frame->u.action.u.delba.params = cpu_to_le16(params);
2269 	action_frame->u.action.u.delba.reason_code =
2270 		cpu_to_le16(WLAN_REASON_QSTA_TIMEOUT);
2271 
2272 	return skb;
2273 }
2274 
2275 /*********************************************************
2276  *
2277  * IOT functions
2278  *
2279  *********************************************************/
2280 static bool rtl_chk_vendor_ouisub(struct ieee80211_hw *hw,
2281 				  struct octet_string vendor_ie)
2282 {
2283 	struct rtl_priv *rtlpriv = rtl_priv(hw);
2284 	bool matched = false;
2285 	static u8 athcap_1[] = { 0x00, 0x03, 0x7F };
2286 	static u8 athcap_2[] = { 0x00, 0x13, 0x74 };
2287 	static u8 broadcap_1[] = { 0x00, 0x10, 0x18 };
2288 	static u8 broadcap_2[] = { 0x00, 0x0a, 0xf7 };
2289 	static u8 broadcap_3[] = { 0x00, 0x05, 0xb5 };
2290 	static u8 racap[] = { 0x00, 0x0c, 0x43 };
2291 	static u8 ciscocap[] = { 0x00, 0x40, 0x96 };
2292 	static u8 marvcap[] = { 0x00, 0x50, 0x43 };
2293 
2294 	if (memcmp(vendor_ie.octet, athcap_1, 3) == 0 ||
2295 		memcmp(vendor_ie.octet, athcap_2, 3) == 0) {
2296 		rtlpriv->mac80211.vendor = PEER_ATH;
2297 		matched = true;
2298 	} else if (memcmp(vendor_ie.octet, broadcap_1, 3) == 0 ||
2299 		memcmp(vendor_ie.octet, broadcap_2, 3) == 0 ||
2300 		memcmp(vendor_ie.octet, broadcap_3, 3) == 0) {
2301 		rtlpriv->mac80211.vendor = PEER_BROAD;
2302 		matched = true;
2303 	} else if (memcmp(vendor_ie.octet, racap, 3) == 0) {
2304 		rtlpriv->mac80211.vendor = PEER_RAL;
2305 		matched = true;
2306 	} else if (memcmp(vendor_ie.octet, ciscocap, 3) == 0) {
2307 		rtlpriv->mac80211.vendor = PEER_CISCO;
2308 		matched = true;
2309 	} else if (memcmp(vendor_ie.octet, marvcap, 3) == 0) {
2310 		rtlpriv->mac80211.vendor = PEER_MARV;
2311 		matched = true;
2312 	}
2313 
2314 	return matched;
2315 }
2316 
2317 static bool rtl_find_221_ie(struct ieee80211_hw *hw, u8 *data,
2318 		unsigned int len)
2319 {
2320 	struct ieee80211_mgmt *mgmt = (void *)data;
2321 	struct octet_string vendor_ie;
2322 	u8 *pos, *end;
2323 
2324 	pos = (u8 *)mgmt->u.beacon.variable;
2325 	end = data + len;
2326 	while (pos < end) {
2327 		if (pos[0] == 221) {
2328 			vendor_ie.length = pos[1];
2329 			vendor_ie.octet = &pos[2];
2330 			if (rtl_chk_vendor_ouisub(hw, vendor_ie))
2331 				return true;
2332 		}
2333 
2334 		if (pos + 2 + pos[1] > end)
2335 			return false;
2336 
2337 		pos += 2 + pos[1];
2338 	}
2339 	return false;
2340 }
2341 
2342 void rtl_recognize_peer(struct ieee80211_hw *hw, u8 *data, unsigned int len)
2343 {
2344 	struct rtl_priv *rtlpriv = rtl_priv(hw);
2345 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
2346 	struct ieee80211_hdr *hdr = (void *)data;
2347 	u32 vendor = PEER_UNKNOWN;
2348 
2349 	static u8 ap3_1[3] = { 0x00, 0x14, 0xbf };
2350 	static u8 ap3_2[3] = { 0x00, 0x1a, 0x70 };
2351 	static u8 ap3_3[3] = { 0x00, 0x1d, 0x7e };
2352 	static u8 ap4_1[3] = { 0x00, 0x90, 0xcc };
2353 	static u8 ap4_2[3] = { 0x00, 0x0e, 0x2e };
2354 	static u8 ap4_3[3] = { 0x00, 0x18, 0x02 };
2355 	static u8 ap4_4[3] = { 0x00, 0x17, 0x3f };
2356 	static u8 ap4_5[3] = { 0x00, 0x1c, 0xdf };
2357 	static u8 ap5_1[3] = { 0x00, 0x1c, 0xf0 };
2358 	static u8 ap5_2[3] = { 0x00, 0x21, 0x91 };
2359 	static u8 ap5_3[3] = { 0x00, 0x24, 0x01 };
2360 	static u8 ap5_4[3] = { 0x00, 0x15, 0xe9 };
2361 	static u8 ap5_5[3] = { 0x00, 0x17, 0x9A };
2362 	static u8 ap5_6[3] = { 0x00, 0x18, 0xE7 };
2363 	static u8 ap6_1[3] = { 0x00, 0x17, 0x94 };
2364 	static u8 ap7_1[3] = { 0x00, 0x14, 0xa4 };
2365 
2366 	if (mac->opmode != NL80211_IFTYPE_STATION)
2367 		return;
2368 
2369 	if (mac->link_state == MAC80211_NOLINK) {
2370 		mac->vendor = PEER_UNKNOWN;
2371 		return;
2372 	}
2373 
2374 	if (mac->cnt_after_linked > 2)
2375 		return;
2376 
2377 	/* check if this really is a beacon */
2378 	if (!ieee80211_is_beacon(hdr->frame_control))
2379 		return;
2380 
2381 	/* min. beacon length + FCS_LEN */
2382 	if (len <= 40 + FCS_LEN)
2383 		return;
2384 
2385 	/* and only beacons from the associated BSSID, please */
2386 	if (!ether_addr_equal_64bits(hdr->addr3, rtlpriv->mac80211.bssid))
2387 		return;
2388 
2389 	if (rtl_find_221_ie(hw, data, len))
2390 		vendor = mac->vendor;
2391 
2392 	if ((memcmp(mac->bssid, ap5_1, 3) == 0) ||
2393 		(memcmp(mac->bssid, ap5_2, 3) == 0) ||
2394 		(memcmp(mac->bssid, ap5_3, 3) == 0) ||
2395 		(memcmp(mac->bssid, ap5_4, 3) == 0) ||
2396 		(memcmp(mac->bssid, ap5_5, 3) == 0) ||
2397 		(memcmp(mac->bssid, ap5_6, 3) == 0) ||
2398 		vendor == PEER_ATH) {
2399 		vendor = PEER_ATH;
2400 		RT_TRACE(rtlpriv, COMP_MAC80211, DBG_LOUD, "=>ath find\n");
2401 	} else if ((memcmp(mac->bssid, ap4_4, 3) == 0) ||
2402 		(memcmp(mac->bssid, ap4_5, 3) == 0) ||
2403 		(memcmp(mac->bssid, ap4_1, 3) == 0) ||
2404 		(memcmp(mac->bssid, ap4_2, 3) == 0) ||
2405 		(memcmp(mac->bssid, ap4_3, 3) == 0) ||
2406 		vendor == PEER_RAL) {
2407 		RT_TRACE(rtlpriv, COMP_MAC80211, DBG_LOUD, "=>ral find\n");
2408 		vendor = PEER_RAL;
2409 	} else if (memcmp(mac->bssid, ap6_1, 3) == 0 ||
2410 		vendor == PEER_CISCO) {
2411 		vendor = PEER_CISCO;
2412 		RT_TRACE(rtlpriv, COMP_MAC80211, DBG_LOUD, "=>cisco find\n");
2413 	} else if ((memcmp(mac->bssid, ap3_1, 3) == 0) ||
2414 		(memcmp(mac->bssid, ap3_2, 3) == 0) ||
2415 		(memcmp(mac->bssid, ap3_3, 3) == 0) ||
2416 		vendor == PEER_BROAD) {
2417 		RT_TRACE(rtlpriv, COMP_MAC80211, DBG_LOUD, "=>broad find\n");
2418 		vendor = PEER_BROAD;
2419 	} else if (memcmp(mac->bssid, ap7_1, 3) == 0 ||
2420 		vendor == PEER_MARV) {
2421 		vendor = PEER_MARV;
2422 		RT_TRACE(rtlpriv, COMP_MAC80211, DBG_LOUD, "=>marv find\n");
2423 	}
2424 
2425 	mac->vendor = vendor;
2426 }
2427 EXPORT_SYMBOL_GPL(rtl_recognize_peer);
2428 
2429 MODULE_AUTHOR("lizhaoming	<chaoming_li@realsil.com.cn>");
2430 MODULE_AUTHOR("Realtek WlanFAE	<wlanfae@realtek.com>");
2431 MODULE_AUTHOR("Larry Finger	<Larry.FInger@lwfinger.net>");
2432 MODULE_LICENSE("GPL");
2433 MODULE_DESCRIPTION("Realtek 802.11n PCI wireless core");
2434 
2435 struct rtl_global_var rtl_global_var = {};
2436 EXPORT_SYMBOL_GPL(rtl_global_var);
2437 
2438 static int __init rtl_core_module_init(void)
2439 {
2440 	if (rtl_rate_control_register())
2441 		pr_err("rtl: Unable to register rtl_rc, use default RC !!\n");
2442 
2443 	/* init some global vars */
2444 	INIT_LIST_HEAD(&rtl_global_var.glb_priv_list);
2445 	spin_lock_init(&rtl_global_var.glb_list_lock);
2446 
2447 	return 0;
2448 }
2449 
2450 static void __exit rtl_core_module_exit(void)
2451 {
2452 	/*RC*/
2453 	rtl_rate_control_unregister();
2454 }
2455 
2456 module_init(rtl_core_module_init);
2457 module_exit(rtl_core_module_exit);
2458