1 /*
2  * Copyright (c) 2013 Eugene Krasnikov <k.eugene.e@gmail.com>
3  *
4  * Permission to use, copy, modify, and/or distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
11  * SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
13  * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
14  * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
18 
19 #include <linux/module.h>
20 #include <linux/firmware.h>
21 #include <linux/platform_device.h>
22 #include <linux/of_address.h>
23 #include <linux/of_device.h>
24 #include <linux/of_irq.h>
25 #include <linux/rpmsg.h>
26 #include <linux/soc/qcom/smem_state.h>
27 #include <linux/soc/qcom/wcnss_ctrl.h>
28 #include "wcn36xx.h"
29 
30 unsigned int wcn36xx_dbg_mask;
31 module_param_named(debug_mask, wcn36xx_dbg_mask, uint, 0644);
32 MODULE_PARM_DESC(debug_mask, "Debugging mask");
33 
34 #define CHAN2G(_freq, _idx) { \
35 	.band = NL80211_BAND_2GHZ, \
36 	.center_freq = (_freq), \
37 	.hw_value = (_idx), \
38 	.max_power = 25, \
39 }
40 
41 #define CHAN5G(_freq, _idx) { \
42 	.band = NL80211_BAND_5GHZ, \
43 	.center_freq = (_freq), \
44 	.hw_value = (_idx), \
45 	.max_power = 25, \
46 }
47 
48 /* The wcn firmware expects channel values to matching
49  * their mnemonic values. So use these for .hw_value. */
50 static struct ieee80211_channel wcn_2ghz_channels[] = {
51 	CHAN2G(2412, 1), /* Channel 1 */
52 	CHAN2G(2417, 2), /* Channel 2 */
53 	CHAN2G(2422, 3), /* Channel 3 */
54 	CHAN2G(2427, 4), /* Channel 4 */
55 	CHAN2G(2432, 5), /* Channel 5 */
56 	CHAN2G(2437, 6), /* Channel 6 */
57 	CHAN2G(2442, 7), /* Channel 7 */
58 	CHAN2G(2447, 8), /* Channel 8 */
59 	CHAN2G(2452, 9), /* Channel 9 */
60 	CHAN2G(2457, 10), /* Channel 10 */
61 	CHAN2G(2462, 11), /* Channel 11 */
62 	CHAN2G(2467, 12), /* Channel 12 */
63 	CHAN2G(2472, 13), /* Channel 13 */
64 	CHAN2G(2484, 14)  /* Channel 14 */
65 
66 };
67 
68 static struct ieee80211_channel wcn_5ghz_channels[] = {
69 	CHAN5G(5180, 36),
70 	CHAN5G(5200, 40),
71 	CHAN5G(5220, 44),
72 	CHAN5G(5240, 48),
73 	CHAN5G(5260, 52),
74 	CHAN5G(5280, 56),
75 	CHAN5G(5300, 60),
76 	CHAN5G(5320, 64),
77 	CHAN5G(5500, 100),
78 	CHAN5G(5520, 104),
79 	CHAN5G(5540, 108),
80 	CHAN5G(5560, 112),
81 	CHAN5G(5580, 116),
82 	CHAN5G(5600, 120),
83 	CHAN5G(5620, 124),
84 	CHAN5G(5640, 128),
85 	CHAN5G(5660, 132),
86 	CHAN5G(5700, 140),
87 	CHAN5G(5745, 149),
88 	CHAN5G(5765, 153),
89 	CHAN5G(5785, 157),
90 	CHAN5G(5805, 161),
91 	CHAN5G(5825, 165)
92 };
93 
94 #define RATE(_bitrate, _hw_rate, _flags) { \
95 	.bitrate        = (_bitrate),                   \
96 	.flags          = (_flags),                     \
97 	.hw_value       = (_hw_rate),                   \
98 	.hw_value_short = (_hw_rate)  \
99 }
100 
101 static struct ieee80211_rate wcn_2ghz_rates[] = {
102 	RATE(10, HW_RATE_INDEX_1MBPS, 0),
103 	RATE(20, HW_RATE_INDEX_2MBPS, IEEE80211_RATE_SHORT_PREAMBLE),
104 	RATE(55, HW_RATE_INDEX_5_5MBPS, IEEE80211_RATE_SHORT_PREAMBLE),
105 	RATE(110, HW_RATE_INDEX_11MBPS, IEEE80211_RATE_SHORT_PREAMBLE),
106 	RATE(60, HW_RATE_INDEX_6MBPS, 0),
107 	RATE(90, HW_RATE_INDEX_9MBPS, 0),
108 	RATE(120, HW_RATE_INDEX_12MBPS, 0),
109 	RATE(180, HW_RATE_INDEX_18MBPS, 0),
110 	RATE(240, HW_RATE_INDEX_24MBPS, 0),
111 	RATE(360, HW_RATE_INDEX_36MBPS, 0),
112 	RATE(480, HW_RATE_INDEX_48MBPS, 0),
113 	RATE(540, HW_RATE_INDEX_54MBPS, 0)
114 };
115 
116 static struct ieee80211_rate wcn_5ghz_rates[] = {
117 	RATE(60, HW_RATE_INDEX_6MBPS, 0),
118 	RATE(90, HW_RATE_INDEX_9MBPS, 0),
119 	RATE(120, HW_RATE_INDEX_12MBPS, 0),
120 	RATE(180, HW_RATE_INDEX_18MBPS, 0),
121 	RATE(240, HW_RATE_INDEX_24MBPS, 0),
122 	RATE(360, HW_RATE_INDEX_36MBPS, 0),
123 	RATE(480, HW_RATE_INDEX_48MBPS, 0),
124 	RATE(540, HW_RATE_INDEX_54MBPS, 0)
125 };
126 
127 static struct ieee80211_supported_band wcn_band_2ghz = {
128 	.channels	= wcn_2ghz_channels,
129 	.n_channels	= ARRAY_SIZE(wcn_2ghz_channels),
130 	.bitrates	= wcn_2ghz_rates,
131 	.n_bitrates	= ARRAY_SIZE(wcn_2ghz_rates),
132 	.ht_cap		= {
133 		.cap =	IEEE80211_HT_CAP_GRN_FLD |
134 			IEEE80211_HT_CAP_SGI_20 |
135 			IEEE80211_HT_CAP_DSSSCCK40 |
136 			IEEE80211_HT_CAP_LSIG_TXOP_PROT,
137 		.ht_supported = true,
138 		.ampdu_factor = IEEE80211_HT_MAX_AMPDU_64K,
139 		.ampdu_density = IEEE80211_HT_MPDU_DENSITY_16,
140 		.mcs = {
141 			.rx_mask = { 0xff, 0, 0, 0, 0, 0, 0, 0, 0, 0, },
142 			.rx_highest = cpu_to_le16(72),
143 			.tx_params = IEEE80211_HT_MCS_TX_DEFINED,
144 		}
145 	}
146 };
147 
148 static struct ieee80211_supported_band wcn_band_5ghz = {
149 	.channels	= wcn_5ghz_channels,
150 	.n_channels	= ARRAY_SIZE(wcn_5ghz_channels),
151 	.bitrates	= wcn_5ghz_rates,
152 	.n_bitrates	= ARRAY_SIZE(wcn_5ghz_rates),
153 	.ht_cap		= {
154 		.cap =	IEEE80211_HT_CAP_GRN_FLD |
155 			IEEE80211_HT_CAP_SGI_20 |
156 			IEEE80211_HT_CAP_DSSSCCK40 |
157 			IEEE80211_HT_CAP_LSIG_TXOP_PROT |
158 			IEEE80211_HT_CAP_SGI_40 |
159 			IEEE80211_HT_CAP_SUP_WIDTH_20_40,
160 		.ht_supported = true,
161 		.ampdu_factor = IEEE80211_HT_MAX_AMPDU_64K,
162 		.ampdu_density = IEEE80211_HT_MPDU_DENSITY_16,
163 		.mcs = {
164 			.rx_mask = { 0xff, 0, 0, 0, 0, 0, 0, 0, 0, 0, },
165 			.rx_highest = cpu_to_le16(72),
166 			.tx_params = IEEE80211_HT_MCS_TX_DEFINED,
167 		}
168 	}
169 };
170 
171 #ifdef CONFIG_PM
172 
173 static const struct wiphy_wowlan_support wowlan_support = {
174 	.flags = WIPHY_WOWLAN_ANY
175 };
176 
177 #endif
178 
179 static inline u8 get_sta_index(struct ieee80211_vif *vif,
180 			       struct wcn36xx_sta *sta_priv)
181 {
182 	return NL80211_IFTYPE_STATION == vif->type ?
183 	       sta_priv->bss_sta_index :
184 	       sta_priv->sta_index;
185 }
186 
187 static const char * const wcn36xx_caps_names[] = {
188 	"MCC",				/* 0 */
189 	"P2P",				/* 1 */
190 	"DOT11AC",			/* 2 */
191 	"SLM_SESSIONIZATION",		/* 3 */
192 	"DOT11AC_OPMODE",		/* 4 */
193 	"SAP32STA",			/* 5 */
194 	"TDLS",				/* 6 */
195 	"P2P_GO_NOA_DECOUPLE_INIT_SCAN",/* 7 */
196 	"WLANACTIVE_OFFLOAD",		/* 8 */
197 	"BEACON_OFFLOAD",		/* 9 */
198 	"SCAN_OFFLOAD",			/* 10 */
199 	"ROAM_OFFLOAD",			/* 11 */
200 	"BCN_MISS_OFFLOAD",		/* 12 */
201 	"STA_POWERSAVE",		/* 13 */
202 	"STA_ADVANCED_PWRSAVE",		/* 14 */
203 	"AP_UAPSD",			/* 15 */
204 	"AP_DFS",			/* 16 */
205 	"BLOCKACK",			/* 17 */
206 	"PHY_ERR",			/* 18 */
207 	"BCN_FILTER",			/* 19 */
208 	"RTT",				/* 20 */
209 	"RATECTRL",			/* 21 */
210 	"WOW",				/* 22 */
211 	"WLAN_ROAM_SCAN_OFFLOAD",	/* 23 */
212 	"SPECULATIVE_PS_POLL",		/* 24 */
213 	"SCAN_SCH",			/* 25 */
214 	"IBSS_HEARTBEAT_OFFLOAD",	/* 26 */
215 	"WLAN_SCAN_OFFLOAD",		/* 27 */
216 	"WLAN_PERIODIC_TX_PTRN",	/* 28 */
217 	"ADVANCE_TDLS",			/* 29 */
218 	"BATCH_SCAN",			/* 30 */
219 	"FW_IN_TX_PATH",		/* 31 */
220 	"EXTENDED_NSOFFLOAD_SLOT",	/* 32 */
221 	"CH_SWITCH_V1",			/* 33 */
222 	"HT40_OBSS_SCAN",		/* 34 */
223 	"UPDATE_CHANNEL_LIST",		/* 35 */
224 	"WLAN_MCADDR_FLT",		/* 36 */
225 	"WLAN_CH144",			/* 37 */
226 	"NAN",				/* 38 */
227 	"TDLS_SCAN_COEXISTENCE",	/* 39 */
228 	"LINK_LAYER_STATS_MEAS",	/* 40 */
229 	"MU_MIMO",			/* 41 */
230 	"EXTENDED_SCAN",		/* 42 */
231 	"DYNAMIC_WMM_PS",		/* 43 */
232 	"MAC_SPOOFED_SCAN",		/* 44 */
233 	"BMU_ERROR_GENERIC_RECOVERY",	/* 45 */
234 	"DISA",				/* 46 */
235 	"FW_STATS",			/* 47 */
236 	"WPS_PRBRSP_TMPL",		/* 48 */
237 	"BCN_IE_FLT_DELTA",		/* 49 */
238 	"TDLS_OFF_CHANNEL",		/* 51 */
239 	"RTT3",				/* 52 */
240 	"MGMT_FRAME_LOGGING",		/* 53 */
241 	"ENHANCED_TXBD_COMPLETION",	/* 54 */
242 	"LOGGING_ENHANCEMENT",		/* 55 */
243 	"EXT_SCAN_ENHANCED",		/* 56 */
244 	"MEMORY_DUMP_SUPPORTED",	/* 57 */
245 	"PER_PKT_STATS_SUPPORTED",	/* 58 */
246 	"EXT_LL_STAT",			/* 60 */
247 	"WIFI_CONFIG",			/* 61 */
248 	"ANTENNA_DIVERSITY_SELECTION",	/* 62 */
249 };
250 
251 static const char *wcn36xx_get_cap_name(enum place_holder_in_cap_bitmap x)
252 {
253 	if (x >= ARRAY_SIZE(wcn36xx_caps_names))
254 		return "UNKNOWN";
255 	return wcn36xx_caps_names[x];
256 }
257 
258 static void wcn36xx_feat_caps_info(struct wcn36xx *wcn)
259 {
260 	int i;
261 
262 	for (i = 0; i < MAX_FEATURE_SUPPORTED; i++) {
263 		if (get_feat_caps(wcn->fw_feat_caps, i))
264 			wcn36xx_info("FW Cap %s\n", wcn36xx_get_cap_name(i));
265 	}
266 }
267 
268 static int wcn36xx_start(struct ieee80211_hw *hw)
269 {
270 	struct wcn36xx *wcn = hw->priv;
271 	int ret;
272 
273 	wcn36xx_dbg(WCN36XX_DBG_MAC, "mac start\n");
274 
275 	/* SMD initialization */
276 	ret = wcn36xx_smd_open(wcn);
277 	if (ret) {
278 		wcn36xx_err("Failed to open smd channel: %d\n", ret);
279 		goto out_err;
280 	}
281 
282 	/* Allocate memory pools for Mgmt BD headers and Data BD headers */
283 	ret = wcn36xx_dxe_allocate_mem_pools(wcn);
284 	if (ret) {
285 		wcn36xx_err("Failed to alloc DXE mempool: %d\n", ret);
286 		goto out_smd_close;
287 	}
288 
289 	ret = wcn36xx_dxe_alloc_ctl_blks(wcn);
290 	if (ret) {
291 		wcn36xx_err("Failed to alloc DXE ctl blocks: %d\n", ret);
292 		goto out_free_dxe_pool;
293 	}
294 
295 	wcn->hal_buf = kmalloc(WCN36XX_HAL_BUF_SIZE, GFP_KERNEL);
296 	if (!wcn->hal_buf) {
297 		wcn36xx_err("Failed to allocate smd buf\n");
298 		ret = -ENOMEM;
299 		goto out_free_dxe_ctl;
300 	}
301 
302 	ret = wcn36xx_smd_load_nv(wcn);
303 	if (ret) {
304 		wcn36xx_err("Failed to push NV to chip\n");
305 		goto out_free_smd_buf;
306 	}
307 
308 	ret = wcn36xx_smd_start(wcn);
309 	if (ret) {
310 		wcn36xx_err("Failed to start chip\n");
311 		goto out_free_smd_buf;
312 	}
313 
314 	if (!wcn36xx_is_fw_version(wcn, 1, 2, 2, 24)) {
315 		ret = wcn36xx_smd_feature_caps_exchange(wcn);
316 		if (ret)
317 			wcn36xx_warn("Exchange feature caps failed\n");
318 		else
319 			wcn36xx_feat_caps_info(wcn);
320 	}
321 
322 	/* DMA channel initialization */
323 	ret = wcn36xx_dxe_init(wcn);
324 	if (ret) {
325 		wcn36xx_err("DXE init failed\n");
326 		goto out_smd_stop;
327 	}
328 
329 	wcn36xx_debugfs_init(wcn);
330 
331 	INIT_LIST_HEAD(&wcn->vif_list);
332 	spin_lock_init(&wcn->dxe_lock);
333 
334 	return 0;
335 
336 out_smd_stop:
337 	wcn36xx_smd_stop(wcn);
338 out_free_smd_buf:
339 	kfree(wcn->hal_buf);
340 out_free_dxe_ctl:
341 	wcn36xx_dxe_free_ctl_blks(wcn);
342 out_free_dxe_pool:
343 	wcn36xx_dxe_free_mem_pools(wcn);
344 out_smd_close:
345 	wcn36xx_smd_close(wcn);
346 out_err:
347 	return ret;
348 }
349 
350 static void wcn36xx_stop(struct ieee80211_hw *hw)
351 {
352 	struct wcn36xx *wcn = hw->priv;
353 
354 	wcn36xx_dbg(WCN36XX_DBG_MAC, "mac stop\n");
355 
356 	wcn36xx_debugfs_exit(wcn);
357 	wcn36xx_smd_stop(wcn);
358 	wcn36xx_dxe_deinit(wcn);
359 	wcn36xx_smd_close(wcn);
360 
361 	wcn36xx_dxe_free_mem_pools(wcn);
362 	wcn36xx_dxe_free_ctl_blks(wcn);
363 
364 	kfree(wcn->hal_buf);
365 }
366 
367 static int wcn36xx_config(struct ieee80211_hw *hw, u32 changed)
368 {
369 	struct wcn36xx *wcn = hw->priv;
370 	struct ieee80211_vif *vif = NULL;
371 	struct wcn36xx_vif *tmp;
372 
373 	wcn36xx_dbg(WCN36XX_DBG_MAC, "mac config changed 0x%08x\n", changed);
374 
375 	mutex_lock(&wcn->conf_mutex);
376 
377 	if (changed & IEEE80211_CONF_CHANGE_CHANNEL) {
378 		int ch = WCN36XX_HW_CHANNEL(wcn);
379 		wcn36xx_dbg(WCN36XX_DBG_MAC, "wcn36xx_config channel switch=%d\n",
380 			    ch);
381 		list_for_each_entry(tmp, &wcn->vif_list, list) {
382 			vif = wcn36xx_priv_to_vif(tmp);
383 			wcn36xx_smd_switch_channel(wcn, vif, ch);
384 		}
385 	}
386 
387 	if (changed & IEEE80211_CONF_CHANGE_PS) {
388 		list_for_each_entry(tmp, &wcn->vif_list, list) {
389 			vif = wcn36xx_priv_to_vif(tmp);
390 			if (hw->conf.flags & IEEE80211_CONF_PS) {
391 				if (vif->bss_conf.ps) /* ps allowed ? */
392 					wcn36xx_pmc_enter_bmps_state(wcn, vif);
393 			} else {
394 				wcn36xx_pmc_exit_bmps_state(wcn, vif);
395 			}
396 		}
397 	}
398 
399 	mutex_unlock(&wcn->conf_mutex);
400 
401 	return 0;
402 }
403 
404 static void wcn36xx_configure_filter(struct ieee80211_hw *hw,
405 				     unsigned int changed,
406 				     unsigned int *total, u64 multicast)
407 {
408 	struct wcn36xx_hal_rcv_flt_mc_addr_list_type *fp;
409 	struct wcn36xx *wcn = hw->priv;
410 	struct wcn36xx_vif *tmp;
411 	struct ieee80211_vif *vif = NULL;
412 
413 	wcn36xx_dbg(WCN36XX_DBG_MAC, "mac configure filter\n");
414 
415 	mutex_lock(&wcn->conf_mutex);
416 
417 	*total &= FIF_ALLMULTI;
418 
419 	fp = (void *)(unsigned long)multicast;
420 	list_for_each_entry(tmp, &wcn->vif_list, list) {
421 		vif = wcn36xx_priv_to_vif(tmp);
422 
423 		/* FW handles MC filtering only when connected as STA */
424 		if (*total & FIF_ALLMULTI)
425 			wcn36xx_smd_set_mc_list(wcn, vif, NULL);
426 		else if (NL80211_IFTYPE_STATION == vif->type && tmp->sta_assoc)
427 			wcn36xx_smd_set_mc_list(wcn, vif, fp);
428 	}
429 
430 	mutex_unlock(&wcn->conf_mutex);
431 	kfree(fp);
432 }
433 
434 static u64 wcn36xx_prepare_multicast(struct ieee80211_hw *hw,
435 				     struct netdev_hw_addr_list *mc_list)
436 {
437 	struct wcn36xx_hal_rcv_flt_mc_addr_list_type *fp;
438 	struct netdev_hw_addr *ha;
439 
440 	wcn36xx_dbg(WCN36XX_DBG_MAC, "mac prepare multicast list\n");
441 	fp = kzalloc(sizeof(*fp), GFP_ATOMIC);
442 	if (!fp) {
443 		wcn36xx_err("Out of memory setting filters.\n");
444 		return 0;
445 	}
446 
447 	fp->mc_addr_count = 0;
448 	/* update multicast filtering parameters */
449 	if (netdev_hw_addr_list_count(mc_list) <=
450 	    WCN36XX_HAL_MAX_NUM_MULTICAST_ADDRESS) {
451 		netdev_hw_addr_list_for_each(ha, mc_list) {
452 			memcpy(fp->mc_addr[fp->mc_addr_count],
453 					ha->addr, ETH_ALEN);
454 			fp->mc_addr_count++;
455 		}
456 	}
457 
458 	return (u64)(unsigned long)fp;
459 }
460 
461 static void wcn36xx_tx(struct ieee80211_hw *hw,
462 		       struct ieee80211_tx_control *control,
463 		       struct sk_buff *skb)
464 {
465 	struct wcn36xx *wcn = hw->priv;
466 	struct wcn36xx_sta *sta_priv = NULL;
467 
468 	if (control->sta)
469 		sta_priv = wcn36xx_sta_to_priv(control->sta);
470 
471 	if (wcn36xx_start_tx(wcn, sta_priv, skb))
472 		ieee80211_free_txskb(wcn->hw, skb);
473 }
474 
475 static int wcn36xx_set_key(struct ieee80211_hw *hw, enum set_key_cmd cmd,
476 			   struct ieee80211_vif *vif,
477 			   struct ieee80211_sta *sta,
478 			   struct ieee80211_key_conf *key_conf)
479 {
480 	struct wcn36xx *wcn = hw->priv;
481 	struct wcn36xx_vif *vif_priv = wcn36xx_vif_to_priv(vif);
482 	struct wcn36xx_sta *sta_priv = wcn36xx_sta_to_priv(sta);
483 	int ret = 0;
484 	u8 key[WLAN_MAX_KEY_LEN];
485 
486 	wcn36xx_dbg(WCN36XX_DBG_MAC, "mac80211 set key\n");
487 	wcn36xx_dbg(WCN36XX_DBG_MAC, "Key: cmd=0x%x algo:0x%x, id:%d, len:%d flags 0x%x\n",
488 		    cmd, key_conf->cipher, key_conf->keyidx,
489 		    key_conf->keylen, key_conf->flags);
490 	wcn36xx_dbg_dump(WCN36XX_DBG_MAC, "KEY: ",
491 			 key_conf->key,
492 			 key_conf->keylen);
493 
494 	mutex_lock(&wcn->conf_mutex);
495 
496 	switch (key_conf->cipher) {
497 	case WLAN_CIPHER_SUITE_WEP40:
498 		vif_priv->encrypt_type = WCN36XX_HAL_ED_WEP40;
499 		break;
500 	case WLAN_CIPHER_SUITE_WEP104:
501 		vif_priv->encrypt_type = WCN36XX_HAL_ED_WEP40;
502 		break;
503 	case WLAN_CIPHER_SUITE_CCMP:
504 		vif_priv->encrypt_type = WCN36XX_HAL_ED_CCMP;
505 		break;
506 	case WLAN_CIPHER_SUITE_TKIP:
507 		vif_priv->encrypt_type = WCN36XX_HAL_ED_TKIP;
508 		break;
509 	default:
510 		wcn36xx_err("Unsupported key type 0x%x\n",
511 			      key_conf->cipher);
512 		ret = -EOPNOTSUPP;
513 		goto out;
514 	}
515 
516 	switch (cmd) {
517 	case SET_KEY:
518 		if (WCN36XX_HAL_ED_TKIP == vif_priv->encrypt_type) {
519 			/*
520 			 * Supplicant is sending key in the wrong order:
521 			 * Temporal Key (16 b) - TX MIC (8 b) - RX MIC (8 b)
522 			 * but HW expects it to be in the order as described in
523 			 * IEEE 802.11 spec (see chapter 11.7) like this:
524 			 * Temporal Key (16 b) - RX MIC (8 b) - TX MIC (8 b)
525 			 */
526 			memcpy(key, key_conf->key, 16);
527 			memcpy(key + 16, key_conf->key + 24, 8);
528 			memcpy(key + 24, key_conf->key + 16, 8);
529 		} else {
530 			memcpy(key, key_conf->key, key_conf->keylen);
531 		}
532 
533 		if (IEEE80211_KEY_FLAG_PAIRWISE & key_conf->flags) {
534 			sta_priv->is_data_encrypted = true;
535 			/* Reconfigure bss with encrypt_type */
536 			if (NL80211_IFTYPE_STATION == vif->type)
537 				wcn36xx_smd_config_bss(wcn,
538 						       vif,
539 						       sta,
540 						       sta->addr,
541 						       true);
542 
543 			wcn36xx_smd_set_stakey(wcn,
544 				vif_priv->encrypt_type,
545 				key_conf->keyidx,
546 				key_conf->keylen,
547 				key,
548 				get_sta_index(vif, sta_priv));
549 		} else {
550 			wcn36xx_smd_set_bsskey(wcn,
551 				vif_priv->encrypt_type,
552 				key_conf->keyidx,
553 				key_conf->keylen,
554 				key);
555 			if ((WLAN_CIPHER_SUITE_WEP40 == key_conf->cipher) ||
556 			    (WLAN_CIPHER_SUITE_WEP104 == key_conf->cipher)) {
557 				sta_priv->is_data_encrypted = true;
558 				wcn36xx_smd_set_stakey(wcn,
559 					vif_priv->encrypt_type,
560 					key_conf->keyidx,
561 					key_conf->keylen,
562 					key,
563 					get_sta_index(vif, sta_priv));
564 			}
565 		}
566 		break;
567 	case DISABLE_KEY:
568 		if (!(IEEE80211_KEY_FLAG_PAIRWISE & key_conf->flags)) {
569 			vif_priv->encrypt_type = WCN36XX_HAL_ED_NONE;
570 			wcn36xx_smd_remove_bsskey(wcn,
571 				vif_priv->encrypt_type,
572 				key_conf->keyidx);
573 		} else {
574 			sta_priv->is_data_encrypted = false;
575 			/* do not remove key if disassociated */
576 			if (sta_priv->aid)
577 				wcn36xx_smd_remove_stakey(wcn,
578 					vif_priv->encrypt_type,
579 					key_conf->keyidx,
580 					get_sta_index(vif, sta_priv));
581 		}
582 		break;
583 	default:
584 		wcn36xx_err("Unsupported key cmd 0x%x\n", cmd);
585 		ret = -EOPNOTSUPP;
586 		goto out;
587 	}
588 
589 out:
590 	mutex_unlock(&wcn->conf_mutex);
591 
592 	return ret;
593 }
594 
595 static void wcn36xx_hw_scan_worker(struct work_struct *work)
596 {
597 	struct wcn36xx *wcn = container_of(work, struct wcn36xx, scan_work);
598 	struct cfg80211_scan_request *req = wcn->scan_req;
599 	u8 channels[WCN36XX_HAL_PNO_MAX_NETW_CHANNELS_EX];
600 	struct cfg80211_scan_info scan_info = {};
601 	bool aborted = false;
602 	int i;
603 
604 	wcn36xx_dbg(WCN36XX_DBG_MAC, "mac80211 scan %d channels worker\n", req->n_channels);
605 
606 	for (i = 0; i < req->n_channels; i++)
607 		channels[i] = req->channels[i]->hw_value;
608 
609 	wcn36xx_smd_update_scan_params(wcn, channels, req->n_channels);
610 
611 	wcn36xx_smd_init_scan(wcn, HAL_SYS_MODE_SCAN);
612 	for (i = 0; i < req->n_channels; i++) {
613 		mutex_lock(&wcn->scan_lock);
614 		aborted = wcn->scan_aborted;
615 		mutex_unlock(&wcn->scan_lock);
616 
617 		if (aborted)
618 			break;
619 
620 		wcn->scan_freq = req->channels[i]->center_freq;
621 		wcn->scan_band = req->channels[i]->band;
622 
623 		wcn36xx_smd_start_scan(wcn, req->channels[i]->hw_value);
624 		msleep(30);
625 		wcn36xx_smd_end_scan(wcn, req->channels[i]->hw_value);
626 
627 		wcn->scan_freq = 0;
628 	}
629 	wcn36xx_smd_finish_scan(wcn, HAL_SYS_MODE_SCAN);
630 
631 	scan_info.aborted = aborted;
632 	ieee80211_scan_completed(wcn->hw, &scan_info);
633 
634 	mutex_lock(&wcn->scan_lock);
635 	wcn->scan_req = NULL;
636 	mutex_unlock(&wcn->scan_lock);
637 }
638 
639 static int wcn36xx_hw_scan(struct ieee80211_hw *hw,
640 			   struct ieee80211_vif *vif,
641 			   struct ieee80211_scan_request *hw_req)
642 {
643 	struct wcn36xx *wcn = hw->priv;
644 	mutex_lock(&wcn->scan_lock);
645 	if (wcn->scan_req) {
646 		mutex_unlock(&wcn->scan_lock);
647 		return -EBUSY;
648 	}
649 
650 	wcn->scan_aborted = false;
651 	wcn->scan_req = &hw_req->req;
652 
653 	mutex_unlock(&wcn->scan_lock);
654 
655 	if (!get_feat_caps(wcn->fw_feat_caps, SCAN_OFFLOAD)) {
656 		/* legacy manual/sw scan */
657 		schedule_work(&wcn->scan_work);
658 		return 0;
659 	}
660 
661 	return wcn36xx_smd_start_hw_scan(wcn, vif, &hw_req->req);
662 }
663 
664 static void wcn36xx_cancel_hw_scan(struct ieee80211_hw *hw,
665 				   struct ieee80211_vif *vif)
666 {
667 	struct wcn36xx *wcn = hw->priv;
668 
669 	if (!wcn36xx_smd_stop_hw_scan(wcn)) {
670 		struct cfg80211_scan_info scan_info = { .aborted = true };
671 
672 		ieee80211_scan_completed(wcn->hw, &scan_info);
673 	}
674 
675 	mutex_lock(&wcn->scan_lock);
676 	wcn->scan_aborted = true;
677 	mutex_unlock(&wcn->scan_lock);
678 
679 	cancel_work_sync(&wcn->scan_work);
680 }
681 
682 static void wcn36xx_update_allowed_rates(struct ieee80211_sta *sta,
683 					 enum nl80211_band band)
684 {
685 	int i, size;
686 	u16 *rates_table;
687 	struct wcn36xx_sta *sta_priv = wcn36xx_sta_to_priv(sta);
688 	u32 rates = sta->supp_rates[band];
689 
690 	memset(&sta_priv->supported_rates, 0,
691 		sizeof(sta_priv->supported_rates));
692 	sta_priv->supported_rates.op_rate_mode = STA_11n;
693 
694 	size = ARRAY_SIZE(sta_priv->supported_rates.dsss_rates);
695 	rates_table = sta_priv->supported_rates.dsss_rates;
696 	if (band == NL80211_BAND_2GHZ) {
697 		for (i = 0; i < size; i++) {
698 			if (rates & 0x01) {
699 				rates_table[i] = wcn_2ghz_rates[i].hw_value;
700 				rates = rates >> 1;
701 			}
702 		}
703 	}
704 
705 	size = ARRAY_SIZE(sta_priv->supported_rates.ofdm_rates);
706 	rates_table = sta_priv->supported_rates.ofdm_rates;
707 	for (i = 0; i < size; i++) {
708 		if (rates & 0x01) {
709 			rates_table[i] = wcn_5ghz_rates[i].hw_value;
710 			rates = rates >> 1;
711 		}
712 	}
713 
714 	if (sta->ht_cap.ht_supported) {
715 		BUILD_BUG_ON(sizeof(sta->ht_cap.mcs.rx_mask) >
716 			sizeof(sta_priv->supported_rates.supported_mcs_set));
717 		memcpy(sta_priv->supported_rates.supported_mcs_set,
718 		       sta->ht_cap.mcs.rx_mask,
719 		       sizeof(sta->ht_cap.mcs.rx_mask));
720 	}
721 }
722 void wcn36xx_set_default_rates(struct wcn36xx_hal_supported_rates *rates)
723 {
724 	u16 ofdm_rates[WCN36XX_HAL_NUM_OFDM_RATES] = {
725 		HW_RATE_INDEX_6MBPS,
726 		HW_RATE_INDEX_9MBPS,
727 		HW_RATE_INDEX_12MBPS,
728 		HW_RATE_INDEX_18MBPS,
729 		HW_RATE_INDEX_24MBPS,
730 		HW_RATE_INDEX_36MBPS,
731 		HW_RATE_INDEX_48MBPS,
732 		HW_RATE_INDEX_54MBPS
733 	};
734 	u16 dsss_rates[WCN36XX_HAL_NUM_DSSS_RATES] = {
735 		HW_RATE_INDEX_1MBPS,
736 		HW_RATE_INDEX_2MBPS,
737 		HW_RATE_INDEX_5_5MBPS,
738 		HW_RATE_INDEX_11MBPS
739 	};
740 
741 	rates->op_rate_mode = STA_11n;
742 	memcpy(rates->dsss_rates, dsss_rates,
743 		sizeof(*dsss_rates) * WCN36XX_HAL_NUM_DSSS_RATES);
744 	memcpy(rates->ofdm_rates, ofdm_rates,
745 		sizeof(*ofdm_rates) * WCN36XX_HAL_NUM_OFDM_RATES);
746 	rates->supported_mcs_set[0] = 0xFF;
747 }
748 static void wcn36xx_bss_info_changed(struct ieee80211_hw *hw,
749 				     struct ieee80211_vif *vif,
750 				     struct ieee80211_bss_conf *bss_conf,
751 				     u32 changed)
752 {
753 	struct wcn36xx *wcn = hw->priv;
754 	struct sk_buff *skb = NULL;
755 	u16 tim_off, tim_len;
756 	enum wcn36xx_hal_link_state link_state;
757 	struct wcn36xx_vif *vif_priv = wcn36xx_vif_to_priv(vif);
758 
759 	wcn36xx_dbg(WCN36XX_DBG_MAC, "mac bss info changed vif %p changed 0x%08x\n",
760 		    vif, changed);
761 
762 	mutex_lock(&wcn->conf_mutex);
763 
764 	if (changed & BSS_CHANGED_BEACON_INFO) {
765 		wcn36xx_dbg(WCN36XX_DBG_MAC,
766 			    "mac bss changed dtim period %d\n",
767 			    bss_conf->dtim_period);
768 
769 		vif_priv->dtim_period = bss_conf->dtim_period;
770 	}
771 
772 	if (changed & BSS_CHANGED_BSSID) {
773 		wcn36xx_dbg(WCN36XX_DBG_MAC, "mac bss changed_bssid %pM\n",
774 			    bss_conf->bssid);
775 
776 		if (!is_zero_ether_addr(bss_conf->bssid)) {
777 			vif_priv->is_joining = true;
778 			vif_priv->bss_index = WCN36XX_HAL_BSS_INVALID_IDX;
779 			wcn36xx_smd_join(wcn, bss_conf->bssid,
780 					 vif->addr, WCN36XX_HW_CHANNEL(wcn));
781 			wcn36xx_smd_config_bss(wcn, vif, NULL,
782 					       bss_conf->bssid, false);
783 		} else {
784 			vif_priv->is_joining = false;
785 			wcn36xx_smd_delete_bss(wcn, vif);
786 			vif_priv->encrypt_type = WCN36XX_HAL_ED_NONE;
787 		}
788 	}
789 
790 	if (changed & BSS_CHANGED_SSID) {
791 		wcn36xx_dbg(WCN36XX_DBG_MAC,
792 			    "mac bss changed ssid\n");
793 		wcn36xx_dbg_dump(WCN36XX_DBG_MAC, "ssid ",
794 				 bss_conf->ssid, bss_conf->ssid_len);
795 
796 		vif_priv->ssid.length = bss_conf->ssid_len;
797 		memcpy(&vif_priv->ssid.ssid,
798 		       bss_conf->ssid,
799 		       bss_conf->ssid_len);
800 	}
801 
802 	if (changed & BSS_CHANGED_ASSOC) {
803 		vif_priv->is_joining = false;
804 		if (bss_conf->assoc) {
805 			struct ieee80211_sta *sta;
806 			struct wcn36xx_sta *sta_priv;
807 
808 			wcn36xx_dbg(WCN36XX_DBG_MAC,
809 				    "mac assoc bss %pM vif %pM AID=%d\n",
810 				     bss_conf->bssid,
811 				     vif->addr,
812 				     bss_conf->aid);
813 
814 			vif_priv->sta_assoc = true;
815 
816 			/*
817 			 * Holding conf_mutex ensures mutal exclusion with
818 			 * wcn36xx_sta_remove() and as such ensures that sta
819 			 * won't be freed while we're operating on it. As such
820 			 * we do not need to hold the rcu_read_lock().
821 			 */
822 			sta = ieee80211_find_sta(vif, bss_conf->bssid);
823 			if (!sta) {
824 				wcn36xx_err("sta %pM is not found\n",
825 					      bss_conf->bssid);
826 				goto out;
827 			}
828 			sta_priv = wcn36xx_sta_to_priv(sta);
829 
830 			wcn36xx_update_allowed_rates(sta, WCN36XX_BAND(wcn));
831 
832 			wcn36xx_smd_set_link_st(wcn, bss_conf->bssid,
833 				vif->addr,
834 				WCN36XX_HAL_LINK_POSTASSOC_STATE);
835 			wcn36xx_smd_config_bss(wcn, vif, sta,
836 					       bss_conf->bssid,
837 					       true);
838 			sta_priv->aid = bss_conf->aid;
839 			/*
840 			 * config_sta must be called from  because this is the
841 			 * place where AID is available.
842 			 */
843 			wcn36xx_smd_config_sta(wcn, vif, sta);
844 		} else {
845 			wcn36xx_dbg(WCN36XX_DBG_MAC,
846 				    "disassociated bss %pM vif %pM AID=%d\n",
847 				    bss_conf->bssid,
848 				    vif->addr,
849 				    bss_conf->aid);
850 			vif_priv->sta_assoc = false;
851 			wcn36xx_smd_set_link_st(wcn,
852 						bss_conf->bssid,
853 						vif->addr,
854 						WCN36XX_HAL_LINK_IDLE_STATE);
855 		}
856 	}
857 
858 	if (changed & BSS_CHANGED_AP_PROBE_RESP) {
859 		wcn36xx_dbg(WCN36XX_DBG_MAC, "mac bss changed ap probe resp\n");
860 		skb = ieee80211_proberesp_get(hw, vif);
861 		if (!skb) {
862 			wcn36xx_err("failed to alloc probereq skb\n");
863 			goto out;
864 		}
865 
866 		wcn36xx_smd_update_proberesp_tmpl(wcn, vif, skb);
867 		dev_kfree_skb(skb);
868 	}
869 
870 	if (changed & BSS_CHANGED_BEACON_ENABLED ||
871 	    changed & BSS_CHANGED_BEACON) {
872 		wcn36xx_dbg(WCN36XX_DBG_MAC,
873 			    "mac bss changed beacon enabled %d\n",
874 			    bss_conf->enable_beacon);
875 
876 		if (bss_conf->enable_beacon) {
877 			vif_priv->dtim_period = bss_conf->dtim_period;
878 			vif_priv->bss_index = WCN36XX_HAL_BSS_INVALID_IDX;
879 			wcn36xx_smd_config_bss(wcn, vif, NULL,
880 					       vif->addr, false);
881 			skb = ieee80211_beacon_get_tim(hw, vif, &tim_off,
882 						       &tim_len);
883 			if (!skb) {
884 				wcn36xx_err("failed to alloc beacon skb\n");
885 				goto out;
886 			}
887 			wcn36xx_smd_send_beacon(wcn, vif, skb, tim_off, 0);
888 			dev_kfree_skb(skb);
889 
890 			if (vif->type == NL80211_IFTYPE_ADHOC ||
891 			    vif->type == NL80211_IFTYPE_MESH_POINT)
892 				link_state = WCN36XX_HAL_LINK_IBSS_STATE;
893 			else
894 				link_state = WCN36XX_HAL_LINK_AP_STATE;
895 
896 			wcn36xx_smd_set_link_st(wcn, vif->addr, vif->addr,
897 						link_state);
898 		} else {
899 			wcn36xx_smd_delete_bss(wcn, vif);
900 			wcn36xx_smd_set_link_st(wcn, vif->addr, vif->addr,
901 						WCN36XX_HAL_LINK_IDLE_STATE);
902 		}
903 	}
904 out:
905 
906 	mutex_unlock(&wcn->conf_mutex);
907 
908 	return;
909 }
910 
911 /* this is required when using IEEE80211_HW_HAS_RATE_CONTROL */
912 static int wcn36xx_set_rts_threshold(struct ieee80211_hw *hw, u32 value)
913 {
914 	struct wcn36xx *wcn = hw->priv;
915 	wcn36xx_dbg(WCN36XX_DBG_MAC, "mac set RTS threshold %d\n", value);
916 
917 	mutex_lock(&wcn->conf_mutex);
918 	wcn36xx_smd_update_cfg(wcn, WCN36XX_HAL_CFG_RTS_THRESHOLD, value);
919 	mutex_unlock(&wcn->conf_mutex);
920 
921 	return 0;
922 }
923 
924 static void wcn36xx_remove_interface(struct ieee80211_hw *hw,
925 				     struct ieee80211_vif *vif)
926 {
927 	struct wcn36xx *wcn = hw->priv;
928 	struct wcn36xx_vif *vif_priv = wcn36xx_vif_to_priv(vif);
929 	wcn36xx_dbg(WCN36XX_DBG_MAC, "mac remove interface vif %p\n", vif);
930 
931 	mutex_lock(&wcn->conf_mutex);
932 
933 	list_del(&vif_priv->list);
934 	wcn36xx_smd_delete_sta_self(wcn, vif->addr);
935 
936 	mutex_unlock(&wcn->conf_mutex);
937 }
938 
939 static int wcn36xx_add_interface(struct ieee80211_hw *hw,
940 				 struct ieee80211_vif *vif)
941 {
942 	struct wcn36xx *wcn = hw->priv;
943 	struct wcn36xx_vif *vif_priv = wcn36xx_vif_to_priv(vif);
944 
945 	wcn36xx_dbg(WCN36XX_DBG_MAC, "mac add interface vif %p type %d\n",
946 		    vif, vif->type);
947 
948 	if (!(NL80211_IFTYPE_STATION == vif->type ||
949 	      NL80211_IFTYPE_AP == vif->type ||
950 	      NL80211_IFTYPE_ADHOC == vif->type ||
951 	      NL80211_IFTYPE_MESH_POINT == vif->type)) {
952 		wcn36xx_warn("Unsupported interface type requested: %d\n",
953 			     vif->type);
954 		return -EOPNOTSUPP;
955 	}
956 
957 	mutex_lock(&wcn->conf_mutex);
958 
959 	list_add(&vif_priv->list, &wcn->vif_list);
960 	wcn36xx_smd_add_sta_self(wcn, vif);
961 
962 	mutex_unlock(&wcn->conf_mutex);
963 
964 	return 0;
965 }
966 
967 static int wcn36xx_sta_add(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
968 			   struct ieee80211_sta *sta)
969 {
970 	struct wcn36xx *wcn = hw->priv;
971 	struct wcn36xx_vif *vif_priv = wcn36xx_vif_to_priv(vif);
972 	struct wcn36xx_sta *sta_priv = wcn36xx_sta_to_priv(sta);
973 	wcn36xx_dbg(WCN36XX_DBG_MAC, "mac sta add vif %p sta %pM\n",
974 		    vif, sta->addr);
975 
976 	mutex_lock(&wcn->conf_mutex);
977 
978 	spin_lock_init(&sta_priv->ampdu_lock);
979 	sta_priv->vif = vif_priv;
980 	/*
981 	 * For STA mode HW will be configured on BSS_CHANGED_ASSOC because
982 	 * at this stage AID is not available yet.
983 	 */
984 	if (NL80211_IFTYPE_STATION != vif->type) {
985 		wcn36xx_update_allowed_rates(sta, WCN36XX_BAND(wcn));
986 		sta_priv->aid = sta->aid;
987 		wcn36xx_smd_config_sta(wcn, vif, sta);
988 	}
989 
990 	mutex_unlock(&wcn->conf_mutex);
991 
992 	return 0;
993 }
994 
995 static int wcn36xx_sta_remove(struct ieee80211_hw *hw,
996 			      struct ieee80211_vif *vif,
997 			      struct ieee80211_sta *sta)
998 {
999 	struct wcn36xx *wcn = hw->priv;
1000 	struct wcn36xx_sta *sta_priv = wcn36xx_sta_to_priv(sta);
1001 
1002 	wcn36xx_dbg(WCN36XX_DBG_MAC, "mac sta remove vif %p sta %pM index %d\n",
1003 		    vif, sta->addr, sta_priv->sta_index);
1004 
1005 	mutex_lock(&wcn->conf_mutex);
1006 
1007 	wcn36xx_smd_delete_sta(wcn, sta_priv->sta_index);
1008 	sta_priv->vif = NULL;
1009 
1010 	mutex_unlock(&wcn->conf_mutex);
1011 
1012 	return 0;
1013 }
1014 
1015 #ifdef CONFIG_PM
1016 
1017 static int wcn36xx_suspend(struct ieee80211_hw *hw, struct cfg80211_wowlan *wow)
1018 {
1019 	struct wcn36xx *wcn = hw->priv;
1020 
1021 	wcn36xx_dbg(WCN36XX_DBG_MAC, "mac suspend\n");
1022 
1023 	flush_workqueue(wcn->hal_ind_wq);
1024 	wcn36xx_smd_set_power_params(wcn, true);
1025 	return 0;
1026 }
1027 
1028 static int wcn36xx_resume(struct ieee80211_hw *hw)
1029 {
1030 	struct wcn36xx *wcn = hw->priv;
1031 
1032 	wcn36xx_dbg(WCN36XX_DBG_MAC, "mac resume\n");
1033 
1034 	flush_workqueue(wcn->hal_ind_wq);
1035 	wcn36xx_smd_set_power_params(wcn, false);
1036 	return 0;
1037 }
1038 
1039 #endif
1040 
1041 static int wcn36xx_ampdu_action(struct ieee80211_hw *hw,
1042 		    struct ieee80211_vif *vif,
1043 		    struct ieee80211_ampdu_params *params)
1044 {
1045 	struct wcn36xx *wcn = hw->priv;
1046 	struct wcn36xx_sta *sta_priv = wcn36xx_sta_to_priv(params->sta);
1047 	struct ieee80211_sta *sta = params->sta;
1048 	enum ieee80211_ampdu_mlme_action action = params->action;
1049 	u16 tid = params->tid;
1050 	u16 *ssn = &params->ssn;
1051 
1052 	wcn36xx_dbg(WCN36XX_DBG_MAC, "mac ampdu action action %d tid %d\n",
1053 		    action, tid);
1054 
1055 	mutex_lock(&wcn->conf_mutex);
1056 
1057 	switch (action) {
1058 	case IEEE80211_AMPDU_RX_START:
1059 		sta_priv->tid = tid;
1060 		wcn36xx_smd_add_ba_session(wcn, sta, tid, ssn, 0,
1061 			get_sta_index(vif, sta_priv));
1062 		wcn36xx_smd_add_ba(wcn);
1063 		wcn36xx_smd_trigger_ba(wcn, get_sta_index(vif, sta_priv));
1064 		break;
1065 	case IEEE80211_AMPDU_RX_STOP:
1066 		wcn36xx_smd_del_ba(wcn, tid, get_sta_index(vif, sta_priv));
1067 		break;
1068 	case IEEE80211_AMPDU_TX_START:
1069 		spin_lock_bh(&sta_priv->ampdu_lock);
1070 		sta_priv->ampdu_state[tid] = WCN36XX_AMPDU_START;
1071 		spin_unlock_bh(&sta_priv->ampdu_lock);
1072 
1073 		ieee80211_start_tx_ba_cb_irqsafe(vif, sta->addr, tid);
1074 		break;
1075 	case IEEE80211_AMPDU_TX_OPERATIONAL:
1076 		spin_lock_bh(&sta_priv->ampdu_lock);
1077 		sta_priv->ampdu_state[tid] = WCN36XX_AMPDU_OPERATIONAL;
1078 		spin_unlock_bh(&sta_priv->ampdu_lock);
1079 
1080 		wcn36xx_smd_add_ba_session(wcn, sta, tid, ssn, 1,
1081 			get_sta_index(vif, sta_priv));
1082 		break;
1083 	case IEEE80211_AMPDU_TX_STOP_FLUSH:
1084 	case IEEE80211_AMPDU_TX_STOP_FLUSH_CONT:
1085 	case IEEE80211_AMPDU_TX_STOP_CONT:
1086 		spin_lock_bh(&sta_priv->ampdu_lock);
1087 		sta_priv->ampdu_state[tid] = WCN36XX_AMPDU_NONE;
1088 		spin_unlock_bh(&sta_priv->ampdu_lock);
1089 
1090 		ieee80211_stop_tx_ba_cb_irqsafe(vif, sta->addr, tid);
1091 		break;
1092 	default:
1093 		wcn36xx_err("Unknown AMPDU action\n");
1094 	}
1095 
1096 	mutex_unlock(&wcn->conf_mutex);
1097 
1098 	return 0;
1099 }
1100 
1101 static const struct ieee80211_ops wcn36xx_ops = {
1102 	.start			= wcn36xx_start,
1103 	.stop			= wcn36xx_stop,
1104 	.add_interface		= wcn36xx_add_interface,
1105 	.remove_interface	= wcn36xx_remove_interface,
1106 #ifdef CONFIG_PM
1107 	.suspend		= wcn36xx_suspend,
1108 	.resume			= wcn36xx_resume,
1109 #endif
1110 	.config			= wcn36xx_config,
1111 	.prepare_multicast	= wcn36xx_prepare_multicast,
1112 	.configure_filter       = wcn36xx_configure_filter,
1113 	.tx			= wcn36xx_tx,
1114 	.set_key		= wcn36xx_set_key,
1115 	.hw_scan		= wcn36xx_hw_scan,
1116 	.cancel_hw_scan		= wcn36xx_cancel_hw_scan,
1117 	.bss_info_changed	= wcn36xx_bss_info_changed,
1118 	.set_rts_threshold	= wcn36xx_set_rts_threshold,
1119 	.sta_add		= wcn36xx_sta_add,
1120 	.sta_remove		= wcn36xx_sta_remove,
1121 	.ampdu_action		= wcn36xx_ampdu_action,
1122 };
1123 
1124 static int wcn36xx_init_ieee80211(struct wcn36xx *wcn)
1125 {
1126 	int ret = 0;
1127 
1128 	static const u32 cipher_suites[] = {
1129 		WLAN_CIPHER_SUITE_WEP40,
1130 		WLAN_CIPHER_SUITE_WEP104,
1131 		WLAN_CIPHER_SUITE_TKIP,
1132 		WLAN_CIPHER_SUITE_CCMP,
1133 	};
1134 
1135 	ieee80211_hw_set(wcn->hw, TIMING_BEACON_ONLY);
1136 	ieee80211_hw_set(wcn->hw, AMPDU_AGGREGATION);
1137 	ieee80211_hw_set(wcn->hw, CONNECTION_MONITOR);
1138 	ieee80211_hw_set(wcn->hw, SUPPORTS_PS);
1139 	ieee80211_hw_set(wcn->hw, SIGNAL_DBM);
1140 	ieee80211_hw_set(wcn->hw, HAS_RATE_CONTROL);
1141 	ieee80211_hw_set(wcn->hw, SINGLE_SCAN_ON_ALL_BANDS);
1142 
1143 	wcn->hw->wiphy->interface_modes = BIT(NL80211_IFTYPE_STATION) |
1144 		BIT(NL80211_IFTYPE_AP) |
1145 		BIT(NL80211_IFTYPE_ADHOC) |
1146 		BIT(NL80211_IFTYPE_MESH_POINT);
1147 
1148 	wcn->hw->wiphy->bands[NL80211_BAND_2GHZ] = &wcn_band_2ghz;
1149 	if (wcn->rf_id != RF_IRIS_WCN3620)
1150 		wcn->hw->wiphy->bands[NL80211_BAND_5GHZ] = &wcn_band_5ghz;
1151 
1152 	wcn->hw->wiphy->max_scan_ssids = WCN36XX_MAX_SCAN_SSIDS;
1153 	wcn->hw->wiphy->max_scan_ie_len = WCN36XX_MAX_SCAN_IE_LEN;
1154 
1155 	wcn->hw->wiphy->cipher_suites = cipher_suites;
1156 	wcn->hw->wiphy->n_cipher_suites = ARRAY_SIZE(cipher_suites);
1157 
1158 	wcn->hw->wiphy->flags |= WIPHY_FLAG_AP_PROBE_RESP_OFFLOAD;
1159 
1160 #ifdef CONFIG_PM
1161 	wcn->hw->wiphy->wowlan = &wowlan_support;
1162 #endif
1163 
1164 	wcn->hw->max_listen_interval = 200;
1165 
1166 	wcn->hw->queues = 4;
1167 
1168 	SET_IEEE80211_DEV(wcn->hw, wcn->dev);
1169 
1170 	wcn->hw->sta_data_size = sizeof(struct wcn36xx_sta);
1171 	wcn->hw->vif_data_size = sizeof(struct wcn36xx_vif);
1172 
1173 	wiphy_ext_feature_set(wcn->hw->wiphy,
1174 			      NL80211_EXT_FEATURE_CQM_RSSI_LIST);
1175 
1176 	return ret;
1177 }
1178 
1179 static int wcn36xx_platform_get_resources(struct wcn36xx *wcn,
1180 					  struct platform_device *pdev)
1181 {
1182 	struct device_node *mmio_node;
1183 	struct device_node *iris_node;
1184 	struct resource *res;
1185 	int index;
1186 	int ret;
1187 
1188 	/* Set TX IRQ */
1189 	res = platform_get_resource_byname(pdev, IORESOURCE_IRQ, "tx");
1190 	if (!res) {
1191 		wcn36xx_err("failed to get tx_irq\n");
1192 		return -ENOENT;
1193 	}
1194 	wcn->tx_irq = res->start;
1195 
1196 	/* Set RX IRQ */
1197 	res = platform_get_resource_byname(pdev, IORESOURCE_IRQ, "rx");
1198 	if (!res) {
1199 		wcn36xx_err("failed to get rx_irq\n");
1200 		return -ENOENT;
1201 	}
1202 	wcn->rx_irq = res->start;
1203 
1204 	/* Acquire SMSM tx enable handle */
1205 	wcn->tx_enable_state = qcom_smem_state_get(&pdev->dev,
1206 			"tx-enable", &wcn->tx_enable_state_bit);
1207 	if (IS_ERR(wcn->tx_enable_state)) {
1208 		wcn36xx_err("failed to get tx-enable state\n");
1209 		return PTR_ERR(wcn->tx_enable_state);
1210 	}
1211 
1212 	/* Acquire SMSM tx rings empty handle */
1213 	wcn->tx_rings_empty_state = qcom_smem_state_get(&pdev->dev,
1214 			"tx-rings-empty", &wcn->tx_rings_empty_state_bit);
1215 	if (IS_ERR(wcn->tx_rings_empty_state)) {
1216 		wcn36xx_err("failed to get tx-rings-empty state\n");
1217 		return PTR_ERR(wcn->tx_rings_empty_state);
1218 	}
1219 
1220 	mmio_node = of_parse_phandle(pdev->dev.parent->of_node, "qcom,mmio", 0);
1221 	if (!mmio_node) {
1222 		wcn36xx_err("failed to acquire qcom,mmio reference\n");
1223 		return -EINVAL;
1224 	}
1225 
1226 	wcn->is_pronto = !!of_device_is_compatible(mmio_node, "qcom,pronto");
1227 
1228 	/* Map the CCU memory */
1229 	index = of_property_match_string(mmio_node, "reg-names", "ccu");
1230 	wcn->ccu_base = of_iomap(mmio_node, index);
1231 	if (!wcn->ccu_base) {
1232 		wcn36xx_err("failed to map ccu memory\n");
1233 		ret = -ENOMEM;
1234 		goto put_mmio_node;
1235 	}
1236 
1237 	/* Map the DXE memory */
1238 	index = of_property_match_string(mmio_node, "reg-names", "dxe");
1239 	wcn->dxe_base = of_iomap(mmio_node, index);
1240 	if (!wcn->dxe_base) {
1241 		wcn36xx_err("failed to map dxe memory\n");
1242 		ret = -ENOMEM;
1243 		goto unmap_ccu;
1244 	}
1245 
1246 	/* External RF module */
1247 	iris_node = of_get_child_by_name(mmio_node, "iris");
1248 	if (iris_node) {
1249 		if (of_device_is_compatible(iris_node, "qcom,wcn3620"))
1250 			wcn->rf_id = RF_IRIS_WCN3620;
1251 		of_node_put(iris_node);
1252 	}
1253 
1254 	of_node_put(mmio_node);
1255 	return 0;
1256 
1257 unmap_ccu:
1258 	iounmap(wcn->ccu_base);
1259 put_mmio_node:
1260 	of_node_put(mmio_node);
1261 	return ret;
1262 }
1263 
1264 static int wcn36xx_probe(struct platform_device *pdev)
1265 {
1266 	struct ieee80211_hw *hw;
1267 	struct wcn36xx *wcn;
1268 	void *wcnss;
1269 	int ret;
1270 	const u8 *addr;
1271 
1272 	wcn36xx_dbg(WCN36XX_DBG_MAC, "platform probe\n");
1273 
1274 	wcnss = dev_get_drvdata(pdev->dev.parent);
1275 
1276 	hw = ieee80211_alloc_hw(sizeof(struct wcn36xx), &wcn36xx_ops);
1277 	if (!hw) {
1278 		wcn36xx_err("failed to alloc hw\n");
1279 		ret = -ENOMEM;
1280 		goto out_err;
1281 	}
1282 	platform_set_drvdata(pdev, hw);
1283 	wcn = hw->priv;
1284 	wcn->hw = hw;
1285 	wcn->dev = &pdev->dev;
1286 	mutex_init(&wcn->conf_mutex);
1287 	mutex_init(&wcn->hal_mutex);
1288 	mutex_init(&wcn->scan_lock);
1289 
1290 	INIT_WORK(&wcn->scan_work, wcn36xx_hw_scan_worker);
1291 
1292 	wcn->smd_channel = qcom_wcnss_open_channel(wcnss, "WLAN_CTRL", wcn36xx_smd_rsp_process, hw);
1293 	if (IS_ERR(wcn->smd_channel)) {
1294 		wcn36xx_err("failed to open WLAN_CTRL channel\n");
1295 		ret = PTR_ERR(wcn->smd_channel);
1296 		goto out_wq;
1297 	}
1298 
1299 	addr = of_get_property(pdev->dev.of_node, "local-mac-address", &ret);
1300 	if (addr && ret != ETH_ALEN) {
1301 		wcn36xx_err("invalid local-mac-address\n");
1302 		ret = -EINVAL;
1303 		goto out_wq;
1304 	} else if (addr) {
1305 		wcn36xx_info("mac address: %pM\n", addr);
1306 		SET_IEEE80211_PERM_ADDR(wcn->hw, addr);
1307 	}
1308 
1309 	ret = wcn36xx_platform_get_resources(wcn, pdev);
1310 	if (ret)
1311 		goto out_wq;
1312 
1313 	wcn36xx_init_ieee80211(wcn);
1314 	ret = ieee80211_register_hw(wcn->hw);
1315 	if (ret)
1316 		goto out_unmap;
1317 
1318 	return 0;
1319 
1320 out_unmap:
1321 	iounmap(wcn->ccu_base);
1322 	iounmap(wcn->dxe_base);
1323 out_wq:
1324 	ieee80211_free_hw(hw);
1325 out_err:
1326 	return ret;
1327 }
1328 
1329 static int wcn36xx_remove(struct platform_device *pdev)
1330 {
1331 	struct ieee80211_hw *hw = platform_get_drvdata(pdev);
1332 	struct wcn36xx *wcn = hw->priv;
1333 	wcn36xx_dbg(WCN36XX_DBG_MAC, "platform remove\n");
1334 
1335 	release_firmware(wcn->nv);
1336 
1337 	ieee80211_unregister_hw(hw);
1338 
1339 	qcom_smem_state_put(wcn->tx_enable_state);
1340 	qcom_smem_state_put(wcn->tx_rings_empty_state);
1341 
1342 	rpmsg_destroy_ept(wcn->smd_channel);
1343 
1344 	iounmap(wcn->dxe_base);
1345 	iounmap(wcn->ccu_base);
1346 
1347 	mutex_destroy(&wcn->hal_mutex);
1348 	ieee80211_free_hw(hw);
1349 
1350 	return 0;
1351 }
1352 
1353 static const struct of_device_id wcn36xx_of_match[] = {
1354 	{ .compatible = "qcom,wcnss-wlan" },
1355 	{}
1356 };
1357 MODULE_DEVICE_TABLE(of, wcn36xx_of_match);
1358 
1359 static struct platform_driver wcn36xx_driver = {
1360 	.probe      = wcn36xx_probe,
1361 	.remove     = wcn36xx_remove,
1362 	.driver         = {
1363 		.name   = "wcn36xx",
1364 		.of_match_table = wcn36xx_of_match,
1365 	},
1366 };
1367 
1368 module_platform_driver(wcn36xx_driver);
1369 
1370 MODULE_LICENSE("Dual BSD/GPL");
1371 MODULE_AUTHOR("Eugene Krasnikov k.eugene.e@gmail.com");
1372 MODULE_FIRMWARE(WLAN_NV_FILE);
1373