1 /*
2  * Copyright (c) 2010-2011 Atheros Communications Inc.
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
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include "htc.h"
18 
19 MODULE_AUTHOR("Atheros Communications");
20 MODULE_LICENSE("Dual BSD/GPL");
21 MODULE_DESCRIPTION("Atheros driver 802.11n HTC based wireless devices");
22 
23 static unsigned int ath9k_debug = ATH_DBG_DEFAULT;
24 module_param_named(debug, ath9k_debug, uint, 0);
25 MODULE_PARM_DESC(debug, "Debugging mask");
26 
27 int htc_modparam_nohwcrypt;
28 module_param_named(nohwcrypt, htc_modparam_nohwcrypt, int, 0444);
29 MODULE_PARM_DESC(nohwcrypt, "Disable hardware encryption");
30 
31 #define CHAN2G(_freq, _idx)  { \
32 	.center_freq = (_freq), \
33 	.hw_value = (_idx), \
34 	.max_power = 20, \
35 }
36 
37 #define CHAN5G(_freq, _idx) { \
38 	.band = IEEE80211_BAND_5GHZ, \
39 	.center_freq = (_freq), \
40 	.hw_value = (_idx), \
41 	.max_power = 20, \
42 }
43 
44 #define ATH_HTC_BTCOEX_PRODUCT_ID "wb193"
45 
46 static struct ieee80211_channel ath9k_2ghz_channels[] = {
47 	CHAN2G(2412, 0), /* Channel 1 */
48 	CHAN2G(2417, 1), /* Channel 2 */
49 	CHAN2G(2422, 2), /* Channel 3 */
50 	CHAN2G(2427, 3), /* Channel 4 */
51 	CHAN2G(2432, 4), /* Channel 5 */
52 	CHAN2G(2437, 5), /* Channel 6 */
53 	CHAN2G(2442, 6), /* Channel 7 */
54 	CHAN2G(2447, 7), /* Channel 8 */
55 	CHAN2G(2452, 8), /* Channel 9 */
56 	CHAN2G(2457, 9), /* Channel 10 */
57 	CHAN2G(2462, 10), /* Channel 11 */
58 	CHAN2G(2467, 11), /* Channel 12 */
59 	CHAN2G(2472, 12), /* Channel 13 */
60 	CHAN2G(2484, 13), /* Channel 14 */
61 };
62 
63 static struct ieee80211_channel ath9k_5ghz_channels[] = {
64 	/* _We_ call this UNII 1 */
65 	CHAN5G(5180, 14), /* Channel 36 */
66 	CHAN5G(5200, 15), /* Channel 40 */
67 	CHAN5G(5220, 16), /* Channel 44 */
68 	CHAN5G(5240, 17), /* Channel 48 */
69 	/* _We_ call this UNII 2 */
70 	CHAN5G(5260, 18), /* Channel 52 */
71 	CHAN5G(5280, 19), /* Channel 56 */
72 	CHAN5G(5300, 20), /* Channel 60 */
73 	CHAN5G(5320, 21), /* Channel 64 */
74 	/* _We_ call this "Middle band" */
75 	CHAN5G(5500, 22), /* Channel 100 */
76 	CHAN5G(5520, 23), /* Channel 104 */
77 	CHAN5G(5540, 24), /* Channel 108 */
78 	CHAN5G(5560, 25), /* Channel 112 */
79 	CHAN5G(5580, 26), /* Channel 116 */
80 	CHAN5G(5600, 27), /* Channel 120 */
81 	CHAN5G(5620, 28), /* Channel 124 */
82 	CHAN5G(5640, 29), /* Channel 128 */
83 	CHAN5G(5660, 30), /* Channel 132 */
84 	CHAN5G(5680, 31), /* Channel 136 */
85 	CHAN5G(5700, 32), /* Channel 140 */
86 	/* _We_ call this UNII 3 */
87 	CHAN5G(5745, 33), /* Channel 149 */
88 	CHAN5G(5765, 34), /* Channel 153 */
89 	CHAN5G(5785, 35), /* Channel 157 */
90 	CHAN5G(5805, 36), /* Channel 161 */
91 	CHAN5G(5825, 37), /* Channel 165 */
92 };
93 
94 /* Atheros hardware rate code addition for short premble */
95 #define SHPCHECK(__hw_rate, __flags) \
96 	((__flags & IEEE80211_RATE_SHORT_PREAMBLE) ? (__hw_rate | 0x04) : 0)
97 
98 #define RATE(_bitrate, _hw_rate, _flags) {		\
99 	.bitrate	= (_bitrate),			\
100 	.flags		= (_flags),			\
101 	.hw_value	= (_hw_rate),			\
102 	.hw_value_short = (SHPCHECK(_hw_rate, _flags))	\
103 }
104 
105 static struct ieee80211_rate ath9k_legacy_rates[] = {
106 	RATE(10, 0x1b, 0),
107 	RATE(20, 0x1a, IEEE80211_RATE_SHORT_PREAMBLE), /* shortp : 0x1e */
108 	RATE(55, 0x19, IEEE80211_RATE_SHORT_PREAMBLE), /* shortp: 0x1d */
109 	RATE(110, 0x18, IEEE80211_RATE_SHORT_PREAMBLE), /* short: 0x1c */
110 	RATE(60, 0x0b, 0),
111 	RATE(90, 0x0f, 0),
112 	RATE(120, 0x0a, 0),
113 	RATE(180, 0x0e, 0),
114 	RATE(240, 0x09, 0),
115 	RATE(360, 0x0d, 0),
116 	RATE(480, 0x08, 0),
117 	RATE(540, 0x0c, 0),
118 };
119 
120 #ifdef CONFIG_MAC80211_LEDS
121 static const struct ieee80211_tpt_blink ath9k_htc_tpt_blink[] = {
122 	{ .throughput = 0 * 1024, .blink_time = 334 },
123 	{ .throughput = 1 * 1024, .blink_time = 260 },
124 	{ .throughput = 5 * 1024, .blink_time = 220 },
125 	{ .throughput = 10 * 1024, .blink_time = 190 },
126 	{ .throughput = 20 * 1024, .blink_time = 170 },
127 	{ .throughput = 50 * 1024, .blink_time = 150 },
128 	{ .throughput = 70 * 1024, .blink_time = 130 },
129 	{ .throughput = 100 * 1024, .blink_time = 110 },
130 	{ .throughput = 200 * 1024, .blink_time = 80 },
131 	{ .throughput = 300 * 1024, .blink_time = 50 },
132 };
133 #endif
134 
135 static int ath9k_htc_wait_for_target(struct ath9k_htc_priv *priv)
136 {
137 	int time_left;
138 
139 	if (atomic_read(&priv->htc->tgt_ready) > 0) {
140 		atomic_dec(&priv->htc->tgt_ready);
141 		return 0;
142 	}
143 
144 	/* Firmware can take up to 50ms to get ready, to be safe use 1 second */
145 	time_left = wait_for_completion_timeout(&priv->htc->target_wait, HZ);
146 	if (!time_left) {
147 		dev_err(priv->dev, "ath9k_htc: Target is unresponsive\n");
148 		return -ETIMEDOUT;
149 	}
150 
151 	atomic_dec(&priv->htc->tgt_ready);
152 
153 	return 0;
154 }
155 
156 static void ath9k_deinit_priv(struct ath9k_htc_priv *priv)
157 {
158 	ath9k_hw_deinit(priv->ah);
159 	kfree(priv->ah);
160 	priv->ah = NULL;
161 }
162 
163 static void ath9k_deinit_device(struct ath9k_htc_priv *priv)
164 {
165 	struct ieee80211_hw *hw = priv->hw;
166 
167 	wiphy_rfkill_stop_polling(hw->wiphy);
168 	ath9k_deinit_leds(priv);
169 	ieee80211_unregister_hw(hw);
170 	ath9k_rx_cleanup(priv);
171 	ath9k_tx_cleanup(priv);
172 	ath9k_deinit_priv(priv);
173 }
174 
175 static inline int ath9k_htc_connect_svc(struct ath9k_htc_priv *priv,
176 					u16 service_id,
177 					void (*tx) (void *,
178 						    struct sk_buff *,
179 						    enum htc_endpoint_id,
180 						    bool txok),
181 					enum htc_endpoint_id *ep_id)
182 {
183 	struct htc_service_connreq req;
184 
185 	memset(&req, 0, sizeof(struct htc_service_connreq));
186 
187 	req.service_id = service_id;
188 	req.ep_callbacks.priv = priv;
189 	req.ep_callbacks.rx = ath9k_htc_rxep;
190 	req.ep_callbacks.tx = tx;
191 
192 	return htc_connect_service(priv->htc, &req, ep_id);
193 }
194 
195 static int ath9k_init_htc_services(struct ath9k_htc_priv *priv, u16 devid,
196 				   u32 drv_info)
197 {
198 	int ret;
199 
200 	/* WMI CMD*/
201 	ret = ath9k_wmi_connect(priv->htc, priv->wmi, &priv->wmi_cmd_ep);
202 	if (ret)
203 		goto err;
204 
205 	/* Beacon */
206 	ret = ath9k_htc_connect_svc(priv, WMI_BEACON_SVC, ath9k_htc_beaconep,
207 				    &priv->beacon_ep);
208 	if (ret)
209 		goto err;
210 
211 	/* CAB */
212 	ret = ath9k_htc_connect_svc(priv, WMI_CAB_SVC, ath9k_htc_txep,
213 				    &priv->cab_ep);
214 	if (ret)
215 		goto err;
216 
217 
218 	/* UAPSD */
219 	ret = ath9k_htc_connect_svc(priv, WMI_UAPSD_SVC, ath9k_htc_txep,
220 				    &priv->uapsd_ep);
221 	if (ret)
222 		goto err;
223 
224 	/* MGMT */
225 	ret = ath9k_htc_connect_svc(priv, WMI_MGMT_SVC, ath9k_htc_txep,
226 				    &priv->mgmt_ep);
227 	if (ret)
228 		goto err;
229 
230 	/* DATA BE */
231 	ret = ath9k_htc_connect_svc(priv, WMI_DATA_BE_SVC, ath9k_htc_txep,
232 				    &priv->data_be_ep);
233 	if (ret)
234 		goto err;
235 
236 	/* DATA BK */
237 	ret = ath9k_htc_connect_svc(priv, WMI_DATA_BK_SVC, ath9k_htc_txep,
238 				    &priv->data_bk_ep);
239 	if (ret)
240 		goto err;
241 
242 	/* DATA VI */
243 	ret = ath9k_htc_connect_svc(priv, WMI_DATA_VI_SVC, ath9k_htc_txep,
244 				    &priv->data_vi_ep);
245 	if (ret)
246 		goto err;
247 
248 	/* DATA VO */
249 	ret = ath9k_htc_connect_svc(priv, WMI_DATA_VO_SVC, ath9k_htc_txep,
250 				    &priv->data_vo_ep);
251 	if (ret)
252 		goto err;
253 
254 	/*
255 	 * Setup required credits before initializing HTC.
256 	 * This is a bit hacky, but, since queuing is done in
257 	 * the HIF layer, shouldn't matter much.
258 	 */
259 
260 	if (IS_AR7010_DEVICE(drv_info))
261 		priv->htc->credits = 45;
262 	else
263 		priv->htc->credits = 33;
264 
265 	ret = htc_init(priv->htc);
266 	if (ret)
267 		goto err;
268 
269 	dev_info(priv->dev, "ath9k_htc: HTC initialized with %d credits\n",
270 		 priv->htc->credits);
271 
272 	return 0;
273 
274 err:
275 	dev_err(priv->dev, "ath9k_htc: Unable to initialize HTC services\n");
276 	return ret;
277 }
278 
279 static int ath9k_reg_notifier(struct wiphy *wiphy,
280 			      struct regulatory_request *request)
281 {
282 	struct ieee80211_hw *hw = wiphy_to_ieee80211_hw(wiphy);
283 	struct ath9k_htc_priv *priv = hw->priv;
284 
285 	return ath_reg_notifier_apply(wiphy, request,
286 				      ath9k_hw_regulatory(priv->ah));
287 }
288 
289 static unsigned int ath9k_regread(void *hw_priv, u32 reg_offset)
290 {
291 	struct ath_hw *ah = (struct ath_hw *) hw_priv;
292 	struct ath_common *common = ath9k_hw_common(ah);
293 	struct ath9k_htc_priv *priv = (struct ath9k_htc_priv *) common->priv;
294 	__be32 val, reg = cpu_to_be32(reg_offset);
295 	int r;
296 
297 	r = ath9k_wmi_cmd(priv->wmi, WMI_REG_READ_CMDID,
298 			  (u8 *) &reg, sizeof(reg),
299 			  (u8 *) &val, sizeof(val),
300 			  100);
301 	if (unlikely(r)) {
302 		ath_dbg(common, WMI, "REGISTER READ FAILED: (0x%04x, %d)\n",
303 			reg_offset, r);
304 		return -EIO;
305 	}
306 
307 	return be32_to_cpu(val);
308 }
309 
310 static void ath9k_multi_regread(void *hw_priv, u32 *addr,
311 				u32 *val, u16 count)
312 {
313 	struct ath_hw *ah = (struct ath_hw *) hw_priv;
314 	struct ath_common *common = ath9k_hw_common(ah);
315 	struct ath9k_htc_priv *priv = (struct ath9k_htc_priv *) common->priv;
316 	__be32 tmpaddr[8];
317 	__be32 tmpval[8];
318 	int i, ret;
319 
320        for (i = 0; i < count; i++) {
321 	       tmpaddr[i] = cpu_to_be32(addr[i]);
322        }
323 
324        ret = ath9k_wmi_cmd(priv->wmi, WMI_REG_READ_CMDID,
325 			   (u8 *)tmpaddr , sizeof(u32) * count,
326 			   (u8 *)tmpval, sizeof(u32) * count,
327 			   100);
328 	if (unlikely(ret)) {
329 		ath_dbg(common, WMI,
330 			"Multiple REGISTER READ FAILED (count: %d)\n", count);
331 	}
332 
333        for (i = 0; i < count; i++) {
334 	       val[i] = be32_to_cpu(tmpval[i]);
335        }
336 }
337 
338 static void ath9k_regwrite_single(void *hw_priv, u32 val, u32 reg_offset)
339 {
340 	struct ath_hw *ah = (struct ath_hw *) hw_priv;
341 	struct ath_common *common = ath9k_hw_common(ah);
342 	struct ath9k_htc_priv *priv = (struct ath9k_htc_priv *) common->priv;
343 	const __be32 buf[2] = {
344 		cpu_to_be32(reg_offset),
345 		cpu_to_be32(val),
346 	};
347 	int r;
348 
349 	r = ath9k_wmi_cmd(priv->wmi, WMI_REG_WRITE_CMDID,
350 			  (u8 *) &buf, sizeof(buf),
351 			  (u8 *) &val, sizeof(val),
352 			  100);
353 	if (unlikely(r)) {
354 		ath_dbg(common, WMI, "REGISTER WRITE FAILED:(0x%04x, %d)\n",
355 			reg_offset, r);
356 	}
357 }
358 
359 static void ath9k_regwrite_buffer(void *hw_priv, u32 val, u32 reg_offset)
360 {
361 	struct ath_hw *ah = (struct ath_hw *) hw_priv;
362 	struct ath_common *common = ath9k_hw_common(ah);
363 	struct ath9k_htc_priv *priv = (struct ath9k_htc_priv *) common->priv;
364 	u32 rsp_status;
365 	int r;
366 
367 	mutex_lock(&priv->wmi->multi_write_mutex);
368 
369 	/* Store the register/value */
370 	priv->wmi->multi_write[priv->wmi->multi_write_idx].reg =
371 		cpu_to_be32(reg_offset);
372 	priv->wmi->multi_write[priv->wmi->multi_write_idx].val =
373 		cpu_to_be32(val);
374 
375 	priv->wmi->multi_write_idx++;
376 
377 	/* If the buffer is full, send it out. */
378 	if (priv->wmi->multi_write_idx == MAX_CMD_NUMBER) {
379 		r = ath9k_wmi_cmd(priv->wmi, WMI_REG_WRITE_CMDID,
380 			  (u8 *) &priv->wmi->multi_write,
381 			  sizeof(struct register_write) * priv->wmi->multi_write_idx,
382 			  (u8 *) &rsp_status, sizeof(rsp_status),
383 			  100);
384 		if (unlikely(r)) {
385 			ath_dbg(common, WMI,
386 				"REGISTER WRITE FAILED, multi len: %d\n",
387 				priv->wmi->multi_write_idx);
388 		}
389 		priv->wmi->multi_write_idx = 0;
390 	}
391 
392 	mutex_unlock(&priv->wmi->multi_write_mutex);
393 }
394 
395 static void ath9k_regwrite(void *hw_priv, u32 val, u32 reg_offset)
396 {
397 	struct ath_hw *ah = (struct ath_hw *) hw_priv;
398 	struct ath_common *common = ath9k_hw_common(ah);
399 	struct ath9k_htc_priv *priv = (struct ath9k_htc_priv *) common->priv;
400 
401 	if (atomic_read(&priv->wmi->mwrite_cnt))
402 		ath9k_regwrite_buffer(hw_priv, val, reg_offset);
403 	else
404 		ath9k_regwrite_single(hw_priv, val, reg_offset);
405 }
406 
407 static void ath9k_enable_regwrite_buffer(void *hw_priv)
408 {
409 	struct ath_hw *ah = (struct ath_hw *) hw_priv;
410 	struct ath_common *common = ath9k_hw_common(ah);
411 	struct ath9k_htc_priv *priv = (struct ath9k_htc_priv *) common->priv;
412 
413 	atomic_inc(&priv->wmi->mwrite_cnt);
414 }
415 
416 static void ath9k_regwrite_flush(void *hw_priv)
417 {
418 	struct ath_hw *ah = (struct ath_hw *) hw_priv;
419 	struct ath_common *common = ath9k_hw_common(ah);
420 	struct ath9k_htc_priv *priv = (struct ath9k_htc_priv *) common->priv;
421 	u32 rsp_status;
422 	int r;
423 
424 	atomic_dec(&priv->wmi->mwrite_cnt);
425 
426 	mutex_lock(&priv->wmi->multi_write_mutex);
427 
428 	if (priv->wmi->multi_write_idx) {
429 		r = ath9k_wmi_cmd(priv->wmi, WMI_REG_WRITE_CMDID,
430 			  (u8 *) &priv->wmi->multi_write,
431 			  sizeof(struct register_write) * priv->wmi->multi_write_idx,
432 			  (u8 *) &rsp_status, sizeof(rsp_status),
433 			  100);
434 		if (unlikely(r)) {
435 			ath_dbg(common, WMI,
436 				"REGISTER WRITE FAILED, multi len: %d\n",
437 				priv->wmi->multi_write_idx);
438 		}
439 		priv->wmi->multi_write_idx = 0;
440 	}
441 
442 	mutex_unlock(&priv->wmi->multi_write_mutex);
443 }
444 
445 static u32 ath9k_reg_rmw(void *hw_priv, u32 reg_offset, u32 set, u32 clr)
446 {
447 	u32 val;
448 
449 	val = ath9k_regread(hw_priv, reg_offset);
450 	val &= ~clr;
451 	val |= set;
452 	ath9k_regwrite(hw_priv, val, reg_offset);
453 	return val;
454 }
455 
456 static void ath_usb_read_cachesize(struct ath_common *common, int *csz)
457 {
458 	*csz = L1_CACHE_BYTES >> 2;
459 }
460 
461 static bool ath_usb_eeprom_read(struct ath_common *common, u32 off, u16 *data)
462 {
463 	struct ath_hw *ah = (struct ath_hw *) common->ah;
464 
465 	(void)REG_READ(ah, AR5416_EEPROM_OFFSET + (off << AR5416_EEPROM_S));
466 
467 	if (!ath9k_hw_wait(ah,
468 			   AR_EEPROM_STATUS_DATA,
469 			   AR_EEPROM_STATUS_DATA_BUSY |
470 			   AR_EEPROM_STATUS_DATA_PROT_ACCESS, 0,
471 			   AH_WAIT_TIMEOUT))
472 		return false;
473 
474 	*data = MS(REG_READ(ah, AR_EEPROM_STATUS_DATA),
475 		   AR_EEPROM_STATUS_DATA_VAL);
476 
477 	return true;
478 }
479 
480 static const struct ath_bus_ops ath9k_usb_bus_ops = {
481 	.ath_bus_type = ATH_USB,
482 	.read_cachesize = ath_usb_read_cachesize,
483 	.eeprom_read = ath_usb_eeprom_read,
484 };
485 
486 static void setup_ht_cap(struct ath9k_htc_priv *priv,
487 			 struct ieee80211_sta_ht_cap *ht_info)
488 {
489 	struct ath_common *common = ath9k_hw_common(priv->ah);
490 	u8 tx_streams, rx_streams;
491 	int i;
492 
493 	ht_info->ht_supported = true;
494 	ht_info->cap = IEEE80211_HT_CAP_SUP_WIDTH_20_40 |
495 		       IEEE80211_HT_CAP_SM_PS |
496 		       IEEE80211_HT_CAP_SGI_40 |
497 		       IEEE80211_HT_CAP_DSSSCCK40;
498 
499 	if (priv->ah->caps.hw_caps & ATH9K_HW_CAP_SGI_20)
500 		ht_info->cap |= IEEE80211_HT_CAP_SGI_20;
501 
502 	ht_info->cap |= (1 << IEEE80211_HT_CAP_RX_STBC_SHIFT);
503 
504 	ht_info->ampdu_factor = IEEE80211_HT_MAX_AMPDU_64K;
505 	ht_info->ampdu_density = IEEE80211_HT_MPDU_DENSITY_8;
506 
507 	memset(&ht_info->mcs, 0, sizeof(ht_info->mcs));
508 
509 	/* ath9k_htc supports only 1 or 2 stream devices */
510 	tx_streams = ath9k_cmn_count_streams(priv->ah->txchainmask, 2);
511 	rx_streams = ath9k_cmn_count_streams(priv->ah->rxchainmask, 2);
512 
513 	ath_dbg(common, CONFIG, "TX streams %d, RX streams: %d\n",
514 		tx_streams, rx_streams);
515 
516 	if (tx_streams != rx_streams) {
517 		ht_info->mcs.tx_params |= IEEE80211_HT_MCS_TX_RX_DIFF;
518 		ht_info->mcs.tx_params |= ((tx_streams - 1) <<
519 					   IEEE80211_HT_MCS_TX_MAX_STREAMS_SHIFT);
520 	}
521 
522 	for (i = 0; i < rx_streams; i++)
523 		ht_info->mcs.rx_mask[i] = 0xff;
524 
525 	ht_info->mcs.tx_params |= IEEE80211_HT_MCS_TX_DEFINED;
526 }
527 
528 static int ath9k_init_queues(struct ath9k_htc_priv *priv)
529 {
530 	struct ath_common *common = ath9k_hw_common(priv->ah);
531 	int i;
532 
533 	for (i = 0; i < ARRAY_SIZE(priv->hwq_map); i++)
534 		priv->hwq_map[i] = -1;
535 
536 	priv->beaconq = ath9k_hw_beaconq_setup(priv->ah);
537 	if (priv->beaconq == -1) {
538 		ath_err(common, "Unable to setup BEACON xmit queue\n");
539 		goto err;
540 	}
541 
542 	priv->cabq = ath9k_htc_cabq_setup(priv);
543 	if (priv->cabq == -1) {
544 		ath_err(common, "Unable to setup CAB xmit queue\n");
545 		goto err;
546 	}
547 
548 	if (!ath9k_htc_txq_setup(priv, WME_AC_BE)) {
549 		ath_err(common, "Unable to setup xmit queue for BE traffic\n");
550 		goto err;
551 	}
552 
553 	if (!ath9k_htc_txq_setup(priv, WME_AC_BK)) {
554 		ath_err(common, "Unable to setup xmit queue for BK traffic\n");
555 		goto err;
556 	}
557 	if (!ath9k_htc_txq_setup(priv, WME_AC_VI)) {
558 		ath_err(common, "Unable to setup xmit queue for VI traffic\n");
559 		goto err;
560 	}
561 	if (!ath9k_htc_txq_setup(priv, WME_AC_VO)) {
562 		ath_err(common, "Unable to setup xmit queue for VO traffic\n");
563 		goto err;
564 	}
565 
566 	return 0;
567 
568 err:
569 	return -EINVAL;
570 }
571 
572 static void ath9k_init_channels_rates(struct ath9k_htc_priv *priv)
573 {
574 	if (priv->ah->caps.hw_caps & ATH9K_HW_CAP_2GHZ) {
575 		priv->sbands[IEEE80211_BAND_2GHZ].channels =
576 			ath9k_2ghz_channels;
577 		priv->sbands[IEEE80211_BAND_2GHZ].band = IEEE80211_BAND_2GHZ;
578 		priv->sbands[IEEE80211_BAND_2GHZ].n_channels =
579 			ARRAY_SIZE(ath9k_2ghz_channels);
580 		priv->sbands[IEEE80211_BAND_2GHZ].bitrates = ath9k_legacy_rates;
581 		priv->sbands[IEEE80211_BAND_2GHZ].n_bitrates =
582 			ARRAY_SIZE(ath9k_legacy_rates);
583 	}
584 
585 	if (priv->ah->caps.hw_caps & ATH9K_HW_CAP_5GHZ) {
586 		priv->sbands[IEEE80211_BAND_5GHZ].channels = ath9k_5ghz_channels;
587 		priv->sbands[IEEE80211_BAND_5GHZ].band = IEEE80211_BAND_5GHZ;
588 		priv->sbands[IEEE80211_BAND_5GHZ].n_channels =
589 			ARRAY_SIZE(ath9k_5ghz_channels);
590 		priv->sbands[IEEE80211_BAND_5GHZ].bitrates =
591 			ath9k_legacy_rates + 4;
592 		priv->sbands[IEEE80211_BAND_5GHZ].n_bitrates =
593 			ARRAY_SIZE(ath9k_legacy_rates) - 4;
594 	}
595 }
596 
597 static void ath9k_init_misc(struct ath9k_htc_priv *priv)
598 {
599 	struct ath_common *common = ath9k_hw_common(priv->ah);
600 
601 	memcpy(common->bssidmask, ath_bcast_mac, ETH_ALEN);
602 
603 	priv->ah->opmode = NL80211_IFTYPE_STATION;
604 }
605 
606 static void ath9k_init_btcoex(struct ath9k_htc_priv *priv)
607 {
608 	int qnum;
609 
610 	switch (ath9k_hw_get_btcoex_scheme(priv->ah)) {
611 	case ATH_BTCOEX_CFG_NONE:
612 		break;
613 	case ATH_BTCOEX_CFG_3WIRE:
614 		priv->ah->btcoex_hw.btactive_gpio = 7;
615 		priv->ah->btcoex_hw.btpriority_gpio = 6;
616 		priv->ah->btcoex_hw.wlanactive_gpio = 8;
617 		priv->btcoex.bt_stomp_type = ATH_BTCOEX_STOMP_LOW;
618 		ath9k_hw_btcoex_init_3wire(priv->ah);
619 		ath_htc_init_btcoex_work(priv);
620 		qnum = priv->hwq_map[WME_AC_BE];
621 		ath9k_hw_init_btcoex_hw(priv->ah, qnum);
622 		break;
623 	default:
624 		WARN_ON(1);
625 		break;
626 	}
627 }
628 
629 static int ath9k_init_priv(struct ath9k_htc_priv *priv,
630 			   u16 devid, char *product,
631 			   u32 drv_info)
632 {
633 	struct ath_hw *ah = NULL;
634 	struct ath_common *common;
635 	int i, ret = 0, csz = 0;
636 
637 	priv->op_flags |= OP_INVALID;
638 
639 	ah = kzalloc(sizeof(struct ath_hw), GFP_KERNEL);
640 	if (!ah)
641 		return -ENOMEM;
642 
643 	ah->hw_version.devid = devid;
644 	ah->hw_version.usbdev = drv_info;
645 	ah->ah_flags |= AH_USE_EEPROM;
646 	ah->reg_ops.read = ath9k_regread;
647 	ah->reg_ops.multi_read = ath9k_multi_regread;
648 	ah->reg_ops.write = ath9k_regwrite;
649 	ah->reg_ops.enable_write_buffer = ath9k_enable_regwrite_buffer;
650 	ah->reg_ops.write_flush = ath9k_regwrite_flush;
651 	ah->reg_ops.rmw = ath9k_reg_rmw;
652 	priv->ah = ah;
653 
654 	common = ath9k_hw_common(ah);
655 	common->ops = &ah->reg_ops;
656 	common->bus_ops = &ath9k_usb_bus_ops;
657 	common->ah = ah;
658 	common->hw = priv->hw;
659 	common->priv = priv;
660 	common->debug_mask = ath9k_debug;
661 
662 	spin_lock_init(&priv->beacon_lock);
663 	spin_lock_init(&priv->tx.tx_lock);
664 	mutex_init(&priv->mutex);
665 	mutex_init(&priv->htc_pm_lock);
666 	tasklet_init(&priv->rx_tasklet, ath9k_rx_tasklet,
667 		     (unsigned long)priv);
668 	tasklet_init(&priv->tx_failed_tasklet, ath9k_tx_failed_tasklet,
669 		     (unsigned long)priv);
670 	INIT_DELAYED_WORK(&priv->ani_work, ath9k_htc_ani_work);
671 	INIT_WORK(&priv->ps_work, ath9k_ps_work);
672 	INIT_WORK(&priv->fatal_work, ath9k_fatal_work);
673 	setup_timer(&priv->tx.cleanup_timer, ath9k_htc_tx_cleanup_timer,
674 		    (unsigned long)priv);
675 
676 	/*
677 	 * Cache line size is used to size and align various
678 	 * structures used to communicate with the hardware.
679 	 */
680 	ath_read_cachesize(common, &csz);
681 	common->cachelsz = csz << 2; /* convert to bytes */
682 
683 	ret = ath9k_hw_init(ah);
684 	if (ret) {
685 		ath_err(common,
686 			"Unable to initialize hardware; initialization status: %d\n",
687 			ret);
688 		goto err_hw;
689 	}
690 
691 	ret = ath9k_init_queues(priv);
692 	if (ret)
693 		goto err_queues;
694 
695 	for (i = 0; i < ATH9K_HTC_MAX_BCN_VIF; i++)
696 		priv->cur_beacon_conf.bslot[i] = NULL;
697 
698 	ath9k_cmn_init_crypto(ah);
699 	ath9k_init_channels_rates(priv);
700 	ath9k_init_misc(priv);
701 
702 	if (product && strncmp(product, ATH_HTC_BTCOEX_PRODUCT_ID, 5) == 0) {
703 		ah->btcoex_hw.scheme = ATH_BTCOEX_CFG_3WIRE;
704 		if (ath9k_hw_get_btcoex_scheme(ah) != ATH_BTCOEX_CFG_NONE)
705 			ath9k_init_btcoex(priv);
706 	}
707 
708 	return 0;
709 
710 err_queues:
711 	ath9k_hw_deinit(ah);
712 err_hw:
713 
714 	kfree(ah);
715 	priv->ah = NULL;
716 
717 	return ret;
718 }
719 
720 static void ath9k_set_hw_capab(struct ath9k_htc_priv *priv,
721 			       struct ieee80211_hw *hw)
722 {
723 	struct ath_common *common = ath9k_hw_common(priv->ah);
724 
725 	hw->flags = IEEE80211_HW_SIGNAL_DBM |
726 		IEEE80211_HW_AMPDU_AGGREGATION |
727 		IEEE80211_HW_SPECTRUM_MGMT |
728 		IEEE80211_HW_HAS_RATE_CONTROL |
729 		IEEE80211_HW_RX_INCLUDES_FCS |
730 		IEEE80211_HW_SUPPORTS_PS |
731 		IEEE80211_HW_PS_NULLFUNC_STACK |
732 		IEEE80211_HW_REPORTS_TX_ACK_STATUS |
733 		IEEE80211_HW_HOST_BROADCAST_PS_BUFFERING;
734 
735 	hw->wiphy->interface_modes =
736 		BIT(NL80211_IFTYPE_STATION) |
737 		BIT(NL80211_IFTYPE_ADHOC) |
738 		BIT(NL80211_IFTYPE_AP) |
739 		BIT(NL80211_IFTYPE_P2P_GO) |
740 		BIT(NL80211_IFTYPE_P2P_CLIENT);
741 
742 	hw->wiphy->flags &= ~WIPHY_FLAG_PS_ON_BY_DEFAULT;
743 
744 	hw->queues = 4;
745 	hw->channel_change_time = 5000;
746 	hw->max_listen_interval = 10;
747 
748 	hw->vif_data_size = sizeof(struct ath9k_htc_vif);
749 	hw->sta_data_size = sizeof(struct ath9k_htc_sta);
750 
751 	/* tx_frame_hdr is larger than tx_mgmt_hdr anyway */
752 	hw->extra_tx_headroom = sizeof(struct tx_frame_hdr) +
753 		sizeof(struct htc_frame_hdr) + 4;
754 
755 	if (priv->ah->caps.hw_caps & ATH9K_HW_CAP_2GHZ)
756 		hw->wiphy->bands[IEEE80211_BAND_2GHZ] =
757 			&priv->sbands[IEEE80211_BAND_2GHZ];
758 	if (priv->ah->caps.hw_caps & ATH9K_HW_CAP_5GHZ)
759 		hw->wiphy->bands[IEEE80211_BAND_5GHZ] =
760 			&priv->sbands[IEEE80211_BAND_5GHZ];
761 
762 	if (priv->ah->caps.hw_caps & ATH9K_HW_CAP_HT) {
763 		if (priv->ah->caps.hw_caps & ATH9K_HW_CAP_2GHZ)
764 			setup_ht_cap(priv,
765 				     &priv->sbands[IEEE80211_BAND_2GHZ].ht_cap);
766 		if (priv->ah->caps.hw_caps & ATH9K_HW_CAP_5GHZ)
767 			setup_ht_cap(priv,
768 				     &priv->sbands[IEEE80211_BAND_5GHZ].ht_cap);
769 	}
770 
771 	SET_IEEE80211_PERM_ADDR(hw, common->macaddr);
772 }
773 
774 static int ath9k_init_firmware_version(struct ath9k_htc_priv *priv)
775 {
776 	struct ieee80211_hw *hw = priv->hw;
777 	struct wmi_fw_version cmd_rsp;
778 	int ret;
779 
780 	memset(&cmd_rsp, 0, sizeof(cmd_rsp));
781 
782 	WMI_CMD(WMI_GET_FW_VERSION);
783 	if (ret)
784 		return -EINVAL;
785 
786 	priv->fw_version_major = be16_to_cpu(cmd_rsp.major);
787 	priv->fw_version_minor = be16_to_cpu(cmd_rsp.minor);
788 
789 	snprintf(hw->wiphy->fw_version, ETHTOOL_BUSINFO_LEN, "%d.%d",
790 		 priv->fw_version_major,
791 		 priv->fw_version_minor);
792 
793 	dev_info(priv->dev, "ath9k_htc: FW Version: %d.%d\n",
794 		 priv->fw_version_major,
795 		 priv->fw_version_minor);
796 
797 	/*
798 	 * Check if the available FW matches the driver's
799 	 * required version.
800 	 */
801 	if (priv->fw_version_major != MAJOR_VERSION_REQ ||
802 	    priv->fw_version_minor != MINOR_VERSION_REQ) {
803 		dev_err(priv->dev, "ath9k_htc: Please upgrade to FW version %d.%d\n",
804 			MAJOR_VERSION_REQ, MINOR_VERSION_REQ);
805 		return -EINVAL;
806 	}
807 
808 	return 0;
809 }
810 
811 static int ath9k_init_device(struct ath9k_htc_priv *priv,
812 			     u16 devid, char *product, u32 drv_info)
813 {
814 	struct ieee80211_hw *hw = priv->hw;
815 	struct ath_common *common;
816 	struct ath_hw *ah;
817 	int error = 0;
818 	struct ath_regulatory *reg;
819 	char hw_name[64];
820 
821 	/* Bring up device */
822 	error = ath9k_init_priv(priv, devid, product, drv_info);
823 	if (error != 0)
824 		goto err_init;
825 
826 	ah = priv->ah;
827 	common = ath9k_hw_common(ah);
828 	ath9k_set_hw_capab(priv, hw);
829 
830 	error = ath9k_init_firmware_version(priv);
831 	if (error != 0)
832 		goto err_fw;
833 
834 	/* Initialize regulatory */
835 	error = ath_regd_init(&common->regulatory, priv->hw->wiphy,
836 			      ath9k_reg_notifier);
837 	if (error)
838 		goto err_regd;
839 
840 	reg = &common->regulatory;
841 
842 	/* Setup TX */
843 	error = ath9k_tx_init(priv);
844 	if (error != 0)
845 		goto err_tx;
846 
847 	/* Setup RX */
848 	error = ath9k_rx_init(priv);
849 	if (error != 0)
850 		goto err_rx;
851 
852 #ifdef CONFIG_MAC80211_LEDS
853 	/* must be initialized before ieee80211_register_hw */
854 	priv->led_cdev.default_trigger = ieee80211_create_tpt_led_trigger(priv->hw,
855 		IEEE80211_TPT_LEDTRIG_FL_RADIO, ath9k_htc_tpt_blink,
856 		ARRAY_SIZE(ath9k_htc_tpt_blink));
857 #endif
858 
859 	/* Register with mac80211 */
860 	error = ieee80211_register_hw(hw);
861 	if (error)
862 		goto err_register;
863 
864 	/* Handle world regulatory */
865 	if (!ath_is_world_regd(reg)) {
866 		error = regulatory_hint(hw->wiphy, reg->alpha2);
867 		if (error)
868 			goto err_world;
869 	}
870 
871 	error = ath9k_htc_init_debug(priv->ah);
872 	if (error) {
873 		ath_err(common, "Unable to create debugfs files\n");
874 		goto err_world;
875 	}
876 
877 	ath_dbg(common, CONFIG,
878 		"WMI:%d, BCN:%d, CAB:%d, UAPSD:%d, MGMT:%d, BE:%d, BK:%d, VI:%d, VO:%d\n",
879 		priv->wmi_cmd_ep,
880 		priv->beacon_ep,
881 		priv->cab_ep,
882 		priv->uapsd_ep,
883 		priv->mgmt_ep,
884 		priv->data_be_ep,
885 		priv->data_bk_ep,
886 		priv->data_vi_ep,
887 		priv->data_vo_ep);
888 
889 	ath9k_hw_name(priv->ah, hw_name, sizeof(hw_name));
890 	wiphy_info(hw->wiphy, "%s\n", hw_name);
891 
892 	ath9k_init_leds(priv);
893 	ath9k_start_rfkill_poll(priv);
894 
895 	return 0;
896 
897 err_world:
898 	ieee80211_unregister_hw(hw);
899 err_register:
900 	ath9k_rx_cleanup(priv);
901 err_rx:
902 	ath9k_tx_cleanup(priv);
903 err_tx:
904 	/* Nothing */
905 err_regd:
906 	/* Nothing */
907 err_fw:
908 	ath9k_deinit_priv(priv);
909 err_init:
910 	return error;
911 }
912 
913 int ath9k_htc_probe_device(struct htc_target *htc_handle, struct device *dev,
914 			   u16 devid, char *product, u32 drv_info)
915 {
916 	struct ieee80211_hw *hw;
917 	struct ath9k_htc_priv *priv;
918 	int ret;
919 
920 	hw = ieee80211_alloc_hw(sizeof(struct ath9k_htc_priv), &ath9k_htc_ops);
921 	if (!hw)
922 		return -ENOMEM;
923 
924 	priv = hw->priv;
925 	priv->hw = hw;
926 	priv->htc = htc_handle;
927 	priv->dev = dev;
928 	htc_handle->drv_priv = priv;
929 	SET_IEEE80211_DEV(hw, priv->dev);
930 
931 	ret = ath9k_htc_wait_for_target(priv);
932 	if (ret)
933 		goto err_free;
934 
935 	priv->wmi = ath9k_init_wmi(priv);
936 	if (!priv->wmi) {
937 		ret = -EINVAL;
938 		goto err_free;
939 	}
940 
941 	ret = ath9k_init_htc_services(priv, devid, drv_info);
942 	if (ret)
943 		goto err_init;
944 
945 	ret = ath9k_init_device(priv, devid, product, drv_info);
946 	if (ret)
947 		goto err_init;
948 
949 	return 0;
950 
951 err_init:
952 	ath9k_deinit_wmi(priv);
953 err_free:
954 	ieee80211_free_hw(hw);
955 	return ret;
956 }
957 
958 void ath9k_htc_disconnect_device(struct htc_target *htc_handle, bool hotunplug)
959 {
960 	if (htc_handle->drv_priv) {
961 
962 		/* Check if the device has been yanked out. */
963 		if (hotunplug)
964 			htc_handle->drv_priv->ah->ah_flags |= AH_UNPLUGGED;
965 
966 		ath9k_deinit_device(htc_handle->drv_priv);
967 		ath9k_deinit_wmi(htc_handle->drv_priv);
968 		ieee80211_free_hw(htc_handle->drv_priv->hw);
969 	}
970 }
971 
972 #ifdef CONFIG_PM
973 
974 void ath9k_htc_suspend(struct htc_target *htc_handle)
975 {
976 	ath9k_htc_setpower(htc_handle->drv_priv, ATH9K_PM_FULL_SLEEP);
977 }
978 
979 int ath9k_htc_resume(struct htc_target *htc_handle)
980 {
981 	struct ath9k_htc_priv *priv = htc_handle->drv_priv;
982 	int ret;
983 
984 	ret = ath9k_htc_wait_for_target(priv);
985 	if (ret)
986 		return ret;
987 
988 	ret = ath9k_init_htc_services(priv, priv->ah->hw_version.devid,
989 				      priv->ah->hw_version.usbdev);
990 	return ret;
991 }
992 #endif
993 
994 static int __init ath9k_htc_init(void)
995 {
996 	if (ath9k_hif_usb_init() < 0) {
997 		printk(KERN_ERR
998 			"ath9k_htc: No USB devices found,"
999 			" driver not installed.\n");
1000 		return -ENODEV;
1001 	}
1002 
1003 	return 0;
1004 }
1005 module_init(ath9k_htc_init);
1006 
1007 static void __exit ath9k_htc_exit(void)
1008 {
1009 	ath9k_hif_usb_exit();
1010 	printk(KERN_INFO "ath9k_htc: Driver unloaded\n");
1011 }
1012 module_exit(ath9k_htc_exit);
1013