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 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
18 
19 #include "htc.h"
20 
21 MODULE_AUTHOR("Atheros Communications");
22 MODULE_LICENSE("Dual BSD/GPL");
23 MODULE_DESCRIPTION("Atheros driver 802.11n HTC based wireless devices");
24 
25 static unsigned int ath9k_debug = ATH_DBG_DEFAULT;
26 module_param_named(debug, ath9k_debug, uint, 0);
27 MODULE_PARM_DESC(debug, "Debugging mask");
28 
29 int htc_modparam_nohwcrypt;
30 module_param_named(nohwcrypt, htc_modparam_nohwcrypt, int, 0444);
31 MODULE_PARM_DESC(nohwcrypt, "Disable hardware encryption");
32 
33 static int ath9k_htc_btcoex_enable;
34 module_param_named(btcoex_enable, ath9k_htc_btcoex_enable, int, 0444);
35 MODULE_PARM_DESC(btcoex_enable, "Enable wifi-BT coexistence");
36 
37 static int ath9k_ps_enable;
38 module_param_named(ps_enable, ath9k_ps_enable, int, 0444);
39 MODULE_PARM_DESC(ps_enable, "Enable WLAN PowerSave");
40 
41 #ifdef CONFIG_MAC80211_LEDS
42 static const struct ieee80211_tpt_blink ath9k_htc_tpt_blink[] = {
43 	{ .throughput = 0 * 1024, .blink_time = 334 },
44 	{ .throughput = 1 * 1024, .blink_time = 260 },
45 	{ .throughput = 5 * 1024, .blink_time = 220 },
46 	{ .throughput = 10 * 1024, .blink_time = 190 },
47 	{ .throughput = 20 * 1024, .blink_time = 170 },
48 	{ .throughput = 50 * 1024, .blink_time = 150 },
49 	{ .throughput = 70 * 1024, .blink_time = 130 },
50 	{ .throughput = 100 * 1024, .blink_time = 110 },
51 	{ .throughput = 200 * 1024, .blink_time = 80 },
52 	{ .throughput = 300 * 1024, .blink_time = 50 },
53 };
54 #endif
55 
56 static int ath9k_htc_wait_for_target(struct ath9k_htc_priv *priv)
57 {
58 	int time_left;
59 
60 	if (atomic_read(&priv->htc->tgt_ready) > 0) {
61 		atomic_dec(&priv->htc->tgt_ready);
62 		return 0;
63 	}
64 
65 	/* Firmware can take up to 50ms to get ready, to be safe use 1 second */
66 	time_left = wait_for_completion_timeout(&priv->htc->target_wait, HZ);
67 	if (!time_left) {
68 		dev_err(priv->dev, "ath9k_htc: Target is unresponsive\n");
69 		return -ETIMEDOUT;
70 	}
71 
72 	atomic_dec(&priv->htc->tgt_ready);
73 
74 	return 0;
75 }
76 
77 static void ath9k_deinit_priv(struct ath9k_htc_priv *priv)
78 {
79 	ath9k_hw_deinit(priv->ah);
80 	kfree(priv->ah);
81 	priv->ah = NULL;
82 }
83 
84 static void ath9k_deinit_device(struct ath9k_htc_priv *priv)
85 {
86 	struct ieee80211_hw *hw = priv->hw;
87 
88 	wiphy_rfkill_stop_polling(hw->wiphy);
89 	ath9k_deinit_leds(priv);
90 	ieee80211_unregister_hw(hw);
91 	ath9k_rx_cleanup(priv);
92 	ath9k_tx_cleanup(priv);
93 	ath9k_deinit_priv(priv);
94 }
95 
96 static inline int ath9k_htc_connect_svc(struct ath9k_htc_priv *priv,
97 					u16 service_id,
98 					void (*tx) (void *,
99 						    struct sk_buff *,
100 						    enum htc_endpoint_id,
101 						    bool txok),
102 					enum htc_endpoint_id *ep_id)
103 {
104 	struct htc_service_connreq req;
105 
106 	memset(&req, 0, sizeof(struct htc_service_connreq));
107 
108 	req.service_id = service_id;
109 	req.ep_callbacks.priv = priv;
110 	req.ep_callbacks.rx = ath9k_htc_rxep;
111 	req.ep_callbacks.tx = tx;
112 
113 	return htc_connect_service(priv->htc, &req, ep_id);
114 }
115 
116 static int ath9k_init_htc_services(struct ath9k_htc_priv *priv, u16 devid,
117 				   u32 drv_info)
118 {
119 	int ret;
120 
121 	/* WMI CMD*/
122 	ret = ath9k_wmi_connect(priv->htc, priv->wmi, &priv->wmi_cmd_ep);
123 	if (ret)
124 		goto err;
125 
126 	/* Beacon */
127 	ret = ath9k_htc_connect_svc(priv, WMI_BEACON_SVC, ath9k_htc_beaconep,
128 				    &priv->beacon_ep);
129 	if (ret)
130 		goto err;
131 
132 	/* CAB */
133 	ret = ath9k_htc_connect_svc(priv, WMI_CAB_SVC, ath9k_htc_txep,
134 				    &priv->cab_ep);
135 	if (ret)
136 		goto err;
137 
138 
139 	/* UAPSD */
140 	ret = ath9k_htc_connect_svc(priv, WMI_UAPSD_SVC, ath9k_htc_txep,
141 				    &priv->uapsd_ep);
142 	if (ret)
143 		goto err;
144 
145 	/* MGMT */
146 	ret = ath9k_htc_connect_svc(priv, WMI_MGMT_SVC, ath9k_htc_txep,
147 				    &priv->mgmt_ep);
148 	if (ret)
149 		goto err;
150 
151 	/* DATA BE */
152 	ret = ath9k_htc_connect_svc(priv, WMI_DATA_BE_SVC, ath9k_htc_txep,
153 				    &priv->data_be_ep);
154 	if (ret)
155 		goto err;
156 
157 	/* DATA BK */
158 	ret = ath9k_htc_connect_svc(priv, WMI_DATA_BK_SVC, ath9k_htc_txep,
159 				    &priv->data_bk_ep);
160 	if (ret)
161 		goto err;
162 
163 	/* DATA VI */
164 	ret = ath9k_htc_connect_svc(priv, WMI_DATA_VI_SVC, ath9k_htc_txep,
165 				    &priv->data_vi_ep);
166 	if (ret)
167 		goto err;
168 
169 	/* DATA VO */
170 	ret = ath9k_htc_connect_svc(priv, WMI_DATA_VO_SVC, ath9k_htc_txep,
171 				    &priv->data_vo_ep);
172 	if (ret)
173 		goto err;
174 
175 	/*
176 	 * Setup required credits before initializing HTC.
177 	 * This is a bit hacky, but, since queuing is done in
178 	 * the HIF layer, shouldn't matter much.
179 	 */
180 
181 	if (IS_AR7010_DEVICE(drv_info))
182 		priv->htc->credits = 45;
183 	else
184 		priv->htc->credits = 33;
185 
186 	ret = htc_init(priv->htc);
187 	if (ret)
188 		goto err;
189 
190 	dev_info(priv->dev, "ath9k_htc: HTC initialized with %d credits\n",
191 		 priv->htc->credits);
192 
193 	return 0;
194 
195 err:
196 	dev_err(priv->dev, "ath9k_htc: Unable to initialize HTC services\n");
197 	return ret;
198 }
199 
200 static void ath9k_reg_notifier(struct wiphy *wiphy,
201 			       struct regulatory_request *request)
202 {
203 	struct ieee80211_hw *hw = wiphy_to_ieee80211_hw(wiphy);
204 	struct ath9k_htc_priv *priv = hw->priv;
205 
206 	ath_reg_notifier_apply(wiphy, request,
207 			       ath9k_hw_regulatory(priv->ah));
208 }
209 
210 static unsigned int ath9k_regread(void *hw_priv, u32 reg_offset)
211 {
212 	struct ath_hw *ah = (struct ath_hw *) hw_priv;
213 	struct ath_common *common = ath9k_hw_common(ah);
214 	struct ath9k_htc_priv *priv = (struct ath9k_htc_priv *) common->priv;
215 	__be32 val, reg = cpu_to_be32(reg_offset);
216 	int r;
217 
218 	r = ath9k_wmi_cmd(priv->wmi, WMI_REG_READ_CMDID,
219 			  (u8 *) &reg, sizeof(reg),
220 			  (u8 *) &val, sizeof(val),
221 			  100);
222 	if (unlikely(r)) {
223 		ath_dbg(common, WMI, "REGISTER READ FAILED: (0x%04x, %d)\n",
224 			reg_offset, r);
225 		return -EIO;
226 	}
227 
228 	return be32_to_cpu(val);
229 }
230 
231 static void ath9k_multi_regread(void *hw_priv, u32 *addr,
232 				u32 *val, u16 count)
233 {
234 	struct ath_hw *ah = (struct ath_hw *) hw_priv;
235 	struct ath_common *common = ath9k_hw_common(ah);
236 	struct ath9k_htc_priv *priv = (struct ath9k_htc_priv *) common->priv;
237 	__be32 tmpaddr[8];
238 	__be32 tmpval[8];
239 	int i, ret;
240 
241        for (i = 0; i < count; i++) {
242 	       tmpaddr[i] = cpu_to_be32(addr[i]);
243        }
244 
245        ret = ath9k_wmi_cmd(priv->wmi, WMI_REG_READ_CMDID,
246 			   (u8 *)tmpaddr , sizeof(u32) * count,
247 			   (u8 *)tmpval, sizeof(u32) * count,
248 			   100);
249 	if (unlikely(ret)) {
250 		ath_dbg(common, WMI,
251 			"Multiple REGISTER READ FAILED (count: %d)\n", count);
252 	}
253 
254        for (i = 0; i < count; i++) {
255 	       val[i] = be32_to_cpu(tmpval[i]);
256        }
257 }
258 
259 static void ath9k_regwrite_single(void *hw_priv, u32 val, u32 reg_offset)
260 {
261 	struct ath_hw *ah = (struct ath_hw *) hw_priv;
262 	struct ath_common *common = ath9k_hw_common(ah);
263 	struct ath9k_htc_priv *priv = (struct ath9k_htc_priv *) common->priv;
264 	const __be32 buf[2] = {
265 		cpu_to_be32(reg_offset),
266 		cpu_to_be32(val),
267 	};
268 	int r;
269 
270 	r = ath9k_wmi_cmd(priv->wmi, WMI_REG_WRITE_CMDID,
271 			  (u8 *) &buf, sizeof(buf),
272 			  (u8 *) &val, sizeof(val),
273 			  100);
274 	if (unlikely(r)) {
275 		ath_dbg(common, WMI, "REGISTER WRITE FAILED:(0x%04x, %d)\n",
276 			reg_offset, r);
277 	}
278 }
279 
280 static void ath9k_regwrite_buffer(void *hw_priv, u32 val, u32 reg_offset)
281 {
282 	struct ath_hw *ah = (struct ath_hw *) hw_priv;
283 	struct ath_common *common = ath9k_hw_common(ah);
284 	struct ath9k_htc_priv *priv = (struct ath9k_htc_priv *) common->priv;
285 	u32 rsp_status;
286 	int r;
287 
288 	mutex_lock(&priv->wmi->multi_write_mutex);
289 
290 	/* Store the register/value */
291 	priv->wmi->multi_write[priv->wmi->multi_write_idx].reg =
292 		cpu_to_be32(reg_offset);
293 	priv->wmi->multi_write[priv->wmi->multi_write_idx].val =
294 		cpu_to_be32(val);
295 
296 	priv->wmi->multi_write_idx++;
297 
298 	/* If the buffer is full, send it out. */
299 	if (priv->wmi->multi_write_idx == MAX_CMD_NUMBER) {
300 		r = ath9k_wmi_cmd(priv->wmi, WMI_REG_WRITE_CMDID,
301 			  (u8 *) &priv->wmi->multi_write,
302 			  sizeof(struct register_write) * priv->wmi->multi_write_idx,
303 			  (u8 *) &rsp_status, sizeof(rsp_status),
304 			  100);
305 		if (unlikely(r)) {
306 			ath_dbg(common, WMI,
307 				"REGISTER WRITE FAILED, multi len: %d\n",
308 				priv->wmi->multi_write_idx);
309 		}
310 		priv->wmi->multi_write_idx = 0;
311 	}
312 
313 	mutex_unlock(&priv->wmi->multi_write_mutex);
314 }
315 
316 static void ath9k_regwrite(void *hw_priv, u32 val, u32 reg_offset)
317 {
318 	struct ath_hw *ah = (struct ath_hw *) hw_priv;
319 	struct ath_common *common = ath9k_hw_common(ah);
320 	struct ath9k_htc_priv *priv = (struct ath9k_htc_priv *) common->priv;
321 
322 	if (atomic_read(&priv->wmi->mwrite_cnt))
323 		ath9k_regwrite_buffer(hw_priv, val, reg_offset);
324 	else
325 		ath9k_regwrite_single(hw_priv, val, reg_offset);
326 }
327 
328 static void ath9k_enable_regwrite_buffer(void *hw_priv)
329 {
330 	struct ath_hw *ah = (struct ath_hw *) hw_priv;
331 	struct ath_common *common = ath9k_hw_common(ah);
332 	struct ath9k_htc_priv *priv = (struct ath9k_htc_priv *) common->priv;
333 
334 	atomic_inc(&priv->wmi->mwrite_cnt);
335 }
336 
337 static void ath9k_regwrite_flush(void *hw_priv)
338 {
339 	struct ath_hw *ah = (struct ath_hw *) hw_priv;
340 	struct ath_common *common = ath9k_hw_common(ah);
341 	struct ath9k_htc_priv *priv = (struct ath9k_htc_priv *) common->priv;
342 	u32 rsp_status;
343 	int r;
344 
345 	atomic_dec(&priv->wmi->mwrite_cnt);
346 
347 	mutex_lock(&priv->wmi->multi_write_mutex);
348 
349 	if (priv->wmi->multi_write_idx) {
350 		r = ath9k_wmi_cmd(priv->wmi, WMI_REG_WRITE_CMDID,
351 			  (u8 *) &priv->wmi->multi_write,
352 			  sizeof(struct register_write) * priv->wmi->multi_write_idx,
353 			  (u8 *) &rsp_status, sizeof(rsp_status),
354 			  100);
355 		if (unlikely(r)) {
356 			ath_dbg(common, WMI,
357 				"REGISTER WRITE FAILED, multi len: %d\n",
358 				priv->wmi->multi_write_idx);
359 		}
360 		priv->wmi->multi_write_idx = 0;
361 	}
362 
363 	mutex_unlock(&priv->wmi->multi_write_mutex);
364 }
365 
366 static u32 ath9k_reg_rmw(void *hw_priv, u32 reg_offset, u32 set, u32 clr)
367 {
368 	u32 val;
369 
370 	val = ath9k_regread(hw_priv, reg_offset);
371 	val &= ~clr;
372 	val |= set;
373 	ath9k_regwrite(hw_priv, val, reg_offset);
374 	return val;
375 }
376 
377 static void ath_usb_read_cachesize(struct ath_common *common, int *csz)
378 {
379 	*csz = L1_CACHE_BYTES >> 2;
380 }
381 
382 static bool ath_usb_eeprom_read(struct ath_common *common, u32 off, u16 *data)
383 {
384 	struct ath_hw *ah = (struct ath_hw *) common->ah;
385 
386 	(void)REG_READ(ah, AR5416_EEPROM_OFFSET + (off << AR5416_EEPROM_S));
387 
388 	if (!ath9k_hw_wait(ah,
389 			   AR_EEPROM_STATUS_DATA,
390 			   AR_EEPROM_STATUS_DATA_BUSY |
391 			   AR_EEPROM_STATUS_DATA_PROT_ACCESS, 0,
392 			   AH_WAIT_TIMEOUT))
393 		return false;
394 
395 	*data = MS(REG_READ(ah, AR_EEPROM_STATUS_DATA),
396 		   AR_EEPROM_STATUS_DATA_VAL);
397 
398 	return true;
399 }
400 
401 static const struct ath_bus_ops ath9k_usb_bus_ops = {
402 	.ath_bus_type = ATH_USB,
403 	.read_cachesize = ath_usb_read_cachesize,
404 	.eeprom_read = ath_usb_eeprom_read,
405 };
406 
407 static int ath9k_init_queues(struct ath9k_htc_priv *priv)
408 {
409 	struct ath_common *common = ath9k_hw_common(priv->ah);
410 	int i;
411 
412 	for (i = 0; i < ARRAY_SIZE(priv->hwq_map); i++)
413 		priv->hwq_map[i] = -1;
414 
415 	priv->beaconq = ath9k_hw_beaconq_setup(priv->ah);
416 	if (priv->beaconq == -1) {
417 		ath_err(common, "Unable to setup BEACON xmit queue\n");
418 		goto err;
419 	}
420 
421 	priv->cabq = ath9k_htc_cabq_setup(priv);
422 	if (priv->cabq == -1) {
423 		ath_err(common, "Unable to setup CAB xmit queue\n");
424 		goto err;
425 	}
426 
427 	if (!ath9k_htc_txq_setup(priv, IEEE80211_AC_BE)) {
428 		ath_err(common, "Unable to setup xmit queue for BE traffic\n");
429 		goto err;
430 	}
431 
432 	if (!ath9k_htc_txq_setup(priv, IEEE80211_AC_BK)) {
433 		ath_err(common, "Unable to setup xmit queue for BK traffic\n");
434 		goto err;
435 	}
436 	if (!ath9k_htc_txq_setup(priv, IEEE80211_AC_VI)) {
437 		ath_err(common, "Unable to setup xmit queue for VI traffic\n");
438 		goto err;
439 	}
440 	if (!ath9k_htc_txq_setup(priv, IEEE80211_AC_VO)) {
441 		ath_err(common, "Unable to setup xmit queue for VO traffic\n");
442 		goto err;
443 	}
444 
445 	return 0;
446 
447 err:
448 	return -EINVAL;
449 }
450 
451 static void ath9k_init_misc(struct ath9k_htc_priv *priv)
452 {
453 	struct ath_common *common = ath9k_hw_common(priv->ah);
454 
455 	memcpy(common->bssidmask, ath_bcast_mac, ETH_ALEN);
456 
457 	common->last_rssi = ATH_RSSI_DUMMY_MARKER;
458 	priv->ah->opmode = NL80211_IFTYPE_STATION;
459 }
460 
461 static int ath9k_init_priv(struct ath9k_htc_priv *priv,
462 			   u16 devid, char *product,
463 			   u32 drv_info)
464 {
465 	struct ath_hw *ah = NULL;
466 	struct ath_common *common;
467 	int i, ret = 0, csz = 0;
468 
469 	set_bit(OP_INVALID, &priv->op_flags);
470 
471 	ah = kzalloc(sizeof(struct ath_hw), GFP_KERNEL);
472 	if (!ah)
473 		return -ENOMEM;
474 
475 	ah->dev = priv->dev;
476 	ah->hw_version.devid = devid;
477 	ah->hw_version.usbdev = drv_info;
478 	ah->ah_flags |= AH_USE_EEPROM;
479 	ah->reg_ops.read = ath9k_regread;
480 	ah->reg_ops.multi_read = ath9k_multi_regread;
481 	ah->reg_ops.write = ath9k_regwrite;
482 	ah->reg_ops.enable_write_buffer = ath9k_enable_regwrite_buffer;
483 	ah->reg_ops.write_flush = ath9k_regwrite_flush;
484 	ah->reg_ops.rmw = ath9k_reg_rmw;
485 	priv->ah = ah;
486 
487 	common = ath9k_hw_common(ah);
488 	common->ops = &ah->reg_ops;
489 	common->bus_ops = &ath9k_usb_bus_ops;
490 	common->ah = ah;
491 	common->hw = priv->hw;
492 	common->priv = priv;
493 	common->debug_mask = ath9k_debug;
494 	common->btcoex_enabled = ath9k_htc_btcoex_enable == 1;
495 
496 	spin_lock_init(&priv->beacon_lock);
497 	spin_lock_init(&priv->tx.tx_lock);
498 	mutex_init(&priv->mutex);
499 	mutex_init(&priv->htc_pm_lock);
500 	tasklet_init(&priv->rx_tasklet, ath9k_rx_tasklet,
501 		     (unsigned long)priv);
502 	tasklet_init(&priv->tx_failed_tasklet, ath9k_tx_failed_tasklet,
503 		     (unsigned long)priv);
504 	INIT_DELAYED_WORK(&priv->ani_work, ath9k_htc_ani_work);
505 	INIT_WORK(&priv->ps_work, ath9k_ps_work);
506 	INIT_WORK(&priv->fatal_work, ath9k_fatal_work);
507 	setup_timer(&priv->tx.cleanup_timer, ath9k_htc_tx_cleanup_timer,
508 		    (unsigned long)priv);
509 
510 	/*
511 	 * Cache line size is used to size and align various
512 	 * structures used to communicate with the hardware.
513 	 */
514 	ath_read_cachesize(common, &csz);
515 	common->cachelsz = csz << 2; /* convert to bytes */
516 
517 	ret = ath9k_hw_init(ah);
518 	if (ret) {
519 		ath_err(common,
520 			"Unable to initialize hardware; initialization status: %d\n",
521 			ret);
522 		goto err_hw;
523 	}
524 
525 	ret = ath9k_init_queues(priv);
526 	if (ret)
527 		goto err_queues;
528 
529 	for (i = 0; i < ATH9K_HTC_MAX_BCN_VIF; i++)
530 		priv->cur_beacon_conf.bslot[i] = NULL;
531 
532 	ath9k_cmn_init_channels_rates(common);
533 	ath9k_cmn_init_crypto(ah);
534 	ath9k_init_misc(priv);
535 	ath9k_htc_init_btcoex(priv, product);
536 
537 	return 0;
538 
539 err_queues:
540 	ath9k_hw_deinit(ah);
541 err_hw:
542 
543 	kfree(ah);
544 	priv->ah = NULL;
545 
546 	return ret;
547 }
548 
549 static const struct ieee80211_iface_limit if_limits[] = {
550 	{ .max = 2,	.types = BIT(NL80211_IFTYPE_STATION) |
551 				 BIT(NL80211_IFTYPE_P2P_CLIENT) },
552 	{ .max = 2,	.types = BIT(NL80211_IFTYPE_AP) |
553 #ifdef CONFIG_MAC80211_MESH
554 				 BIT(NL80211_IFTYPE_MESH_POINT) |
555 #endif
556 				 BIT(NL80211_IFTYPE_P2P_GO) },
557 };
558 
559 static const struct ieee80211_iface_combination if_comb = {
560 	.limits = if_limits,
561 	.n_limits = ARRAY_SIZE(if_limits),
562 	.max_interfaces = 2,
563 	.num_different_channels = 1,
564 };
565 
566 static void ath9k_set_hw_capab(struct ath9k_htc_priv *priv,
567 			       struct ieee80211_hw *hw)
568 {
569 	struct ath_hw *ah = priv->ah;
570 	struct ath_common *common = ath9k_hw_common(priv->ah);
571 	struct base_eep_header *pBase;
572 
573 	hw->flags = IEEE80211_HW_SIGNAL_DBM |
574 		IEEE80211_HW_AMPDU_AGGREGATION |
575 		IEEE80211_HW_SPECTRUM_MGMT |
576 		IEEE80211_HW_HAS_RATE_CONTROL |
577 		IEEE80211_HW_RX_INCLUDES_FCS |
578 		IEEE80211_HW_PS_NULLFUNC_STACK |
579 		IEEE80211_HW_REPORTS_TX_ACK_STATUS |
580 		IEEE80211_HW_MFP_CAPABLE |
581 		IEEE80211_HW_HOST_BROADCAST_PS_BUFFERING;
582 
583 	if (ath9k_ps_enable)
584 		hw->flags |= IEEE80211_HW_SUPPORTS_PS;
585 
586 	hw->wiphy->interface_modes =
587 		BIT(NL80211_IFTYPE_STATION) |
588 		BIT(NL80211_IFTYPE_ADHOC) |
589 		BIT(NL80211_IFTYPE_AP) |
590 		BIT(NL80211_IFTYPE_P2P_GO) |
591 		BIT(NL80211_IFTYPE_P2P_CLIENT) |
592 		BIT(NL80211_IFTYPE_MESH_POINT);
593 
594 	hw->wiphy->iface_combinations = &if_comb;
595 	hw->wiphy->n_iface_combinations = 1;
596 
597 	hw->wiphy->flags &= ~WIPHY_FLAG_PS_ON_BY_DEFAULT;
598 
599 	hw->wiphy->flags |= WIPHY_FLAG_IBSS_RSN |
600 			    WIPHY_FLAG_HAS_REMAIN_ON_CHANNEL;
601 
602 	hw->queues = 4;
603 	hw->max_listen_interval = 1;
604 
605 	hw->vif_data_size = sizeof(struct ath9k_htc_vif);
606 	hw->sta_data_size = sizeof(struct ath9k_htc_sta);
607 
608 	/* tx_frame_hdr is larger than tx_mgmt_hdr anyway */
609 	hw->extra_tx_headroom = sizeof(struct tx_frame_hdr) +
610 		sizeof(struct htc_frame_hdr) + 4;
611 
612 	if (priv->ah->caps.hw_caps & ATH9K_HW_CAP_2GHZ)
613 		hw->wiphy->bands[IEEE80211_BAND_2GHZ] =
614 			&common->sbands[IEEE80211_BAND_2GHZ];
615 	if (priv->ah->caps.hw_caps & ATH9K_HW_CAP_5GHZ)
616 		hw->wiphy->bands[IEEE80211_BAND_5GHZ] =
617 			&common->sbands[IEEE80211_BAND_5GHZ];
618 
619 	ath9k_cmn_reload_chainmask(ah);
620 
621 	pBase = ath9k_htc_get_eeprom_base(priv);
622 	if (pBase) {
623 		hw->wiphy->available_antennas_rx = pBase->rxMask;
624 		hw->wiphy->available_antennas_tx = pBase->txMask;
625 	}
626 
627 	SET_IEEE80211_PERM_ADDR(hw, common->macaddr);
628 }
629 
630 static int ath9k_init_firmware_version(struct ath9k_htc_priv *priv)
631 {
632 	struct ieee80211_hw *hw = priv->hw;
633 	struct wmi_fw_version cmd_rsp;
634 	int ret;
635 
636 	memset(&cmd_rsp, 0, sizeof(cmd_rsp));
637 
638 	WMI_CMD(WMI_GET_FW_VERSION);
639 	if (ret)
640 		return -EINVAL;
641 
642 	priv->fw_version_major = be16_to_cpu(cmd_rsp.major);
643 	priv->fw_version_minor = be16_to_cpu(cmd_rsp.minor);
644 
645 	snprintf(hw->wiphy->fw_version, sizeof(hw->wiphy->fw_version), "%d.%d",
646 		 priv->fw_version_major,
647 		 priv->fw_version_minor);
648 
649 	dev_info(priv->dev, "ath9k_htc: FW Version: %d.%d\n",
650 		 priv->fw_version_major,
651 		 priv->fw_version_minor);
652 
653 	/*
654 	 * Check if the available FW matches the driver's
655 	 * required version.
656 	 */
657 	if (priv->fw_version_major != MAJOR_VERSION_REQ ||
658 	    priv->fw_version_minor < MINOR_VERSION_REQ) {
659 		dev_err(priv->dev, "ath9k_htc: Please upgrade to FW version %d.%d\n",
660 			MAJOR_VERSION_REQ, MINOR_VERSION_REQ);
661 		return -EINVAL;
662 	}
663 
664 	return 0;
665 }
666 
667 static int ath9k_init_device(struct ath9k_htc_priv *priv,
668 			     u16 devid, char *product, u32 drv_info)
669 {
670 	struct ieee80211_hw *hw = priv->hw;
671 	struct ath_common *common;
672 	struct ath_hw *ah;
673 	int error = 0;
674 	struct ath_regulatory *reg;
675 	char hw_name[64];
676 
677 	/* Bring up device */
678 	error = ath9k_init_priv(priv, devid, product, drv_info);
679 	if (error != 0)
680 		goto err_init;
681 
682 	ah = priv->ah;
683 	common = ath9k_hw_common(ah);
684 	ath9k_set_hw_capab(priv, hw);
685 
686 	error = ath9k_init_firmware_version(priv);
687 	if (error != 0)
688 		goto err_fw;
689 
690 	/* Initialize regulatory */
691 	error = ath_regd_init(&common->regulatory, priv->hw->wiphy,
692 			      ath9k_reg_notifier);
693 	if (error)
694 		goto err_regd;
695 
696 	reg = &common->regulatory;
697 
698 	/* Setup TX */
699 	error = ath9k_tx_init(priv);
700 	if (error != 0)
701 		goto err_tx;
702 
703 	/* Setup RX */
704 	error = ath9k_rx_init(priv);
705 	if (error != 0)
706 		goto err_rx;
707 
708 	ath9k_hw_disable(priv->ah);
709 #ifdef CONFIG_MAC80211_LEDS
710 	/* must be initialized before ieee80211_register_hw */
711 	priv->led_cdev.default_trigger = ieee80211_create_tpt_led_trigger(priv->hw,
712 		IEEE80211_TPT_LEDTRIG_FL_RADIO, ath9k_htc_tpt_blink,
713 		ARRAY_SIZE(ath9k_htc_tpt_blink));
714 #endif
715 
716 	/* Register with mac80211 */
717 	error = ieee80211_register_hw(hw);
718 	if (error)
719 		goto err_register;
720 
721 	/* Handle world regulatory */
722 	if (!ath_is_world_regd(reg)) {
723 		error = regulatory_hint(hw->wiphy, reg->alpha2);
724 		if (error)
725 			goto err_world;
726 	}
727 
728 	error = ath9k_htc_init_debug(priv->ah);
729 	if (error) {
730 		ath_err(common, "Unable to create debugfs files\n");
731 		goto err_world;
732 	}
733 
734 	ath_dbg(common, CONFIG,
735 		"WMI:%d, BCN:%d, CAB:%d, UAPSD:%d, MGMT:%d, BE:%d, BK:%d, VI:%d, VO:%d\n",
736 		priv->wmi_cmd_ep,
737 		priv->beacon_ep,
738 		priv->cab_ep,
739 		priv->uapsd_ep,
740 		priv->mgmt_ep,
741 		priv->data_be_ep,
742 		priv->data_bk_ep,
743 		priv->data_vi_ep,
744 		priv->data_vo_ep);
745 
746 	ath9k_hw_name(priv->ah, hw_name, sizeof(hw_name));
747 	wiphy_info(hw->wiphy, "%s\n", hw_name);
748 
749 	ath9k_init_leds(priv);
750 	ath9k_start_rfkill_poll(priv);
751 
752 	return 0;
753 
754 err_world:
755 	ieee80211_unregister_hw(hw);
756 err_register:
757 	ath9k_rx_cleanup(priv);
758 err_rx:
759 	ath9k_tx_cleanup(priv);
760 err_tx:
761 	/* Nothing */
762 err_regd:
763 	/* Nothing */
764 err_fw:
765 	ath9k_deinit_priv(priv);
766 err_init:
767 	return error;
768 }
769 
770 int ath9k_htc_probe_device(struct htc_target *htc_handle, struct device *dev,
771 			   u16 devid, char *product, u32 drv_info)
772 {
773 	struct ieee80211_hw *hw;
774 	struct ath9k_htc_priv *priv;
775 	int ret;
776 
777 	hw = ieee80211_alloc_hw(sizeof(struct ath9k_htc_priv), &ath9k_htc_ops);
778 	if (!hw)
779 		return -ENOMEM;
780 
781 	priv = hw->priv;
782 	priv->hw = hw;
783 	priv->htc = htc_handle;
784 	priv->dev = dev;
785 	htc_handle->drv_priv = priv;
786 	SET_IEEE80211_DEV(hw, priv->dev);
787 
788 	ret = ath9k_htc_wait_for_target(priv);
789 	if (ret)
790 		goto err_free;
791 
792 	priv->wmi = ath9k_init_wmi(priv);
793 	if (!priv->wmi) {
794 		ret = -EINVAL;
795 		goto err_free;
796 	}
797 
798 	ret = ath9k_init_htc_services(priv, devid, drv_info);
799 	if (ret)
800 		goto err_init;
801 
802 	ret = ath9k_init_device(priv, devid, product, drv_info);
803 	if (ret)
804 		goto err_init;
805 
806 	return 0;
807 
808 err_init:
809 	ath9k_deinit_wmi(priv);
810 err_free:
811 	ieee80211_free_hw(hw);
812 	return ret;
813 }
814 
815 void ath9k_htc_disconnect_device(struct htc_target *htc_handle, bool hotunplug)
816 {
817 	if (htc_handle->drv_priv) {
818 
819 		/* Check if the device has been yanked out. */
820 		if (hotunplug)
821 			htc_handle->drv_priv->ah->ah_flags |= AH_UNPLUGGED;
822 
823 		ath9k_deinit_device(htc_handle->drv_priv);
824 		ath9k_deinit_wmi(htc_handle->drv_priv);
825 		ieee80211_free_hw(htc_handle->drv_priv->hw);
826 	}
827 }
828 
829 #ifdef CONFIG_PM
830 
831 void ath9k_htc_suspend(struct htc_target *htc_handle)
832 {
833 	ath9k_htc_setpower(htc_handle->drv_priv, ATH9K_PM_FULL_SLEEP);
834 }
835 
836 int ath9k_htc_resume(struct htc_target *htc_handle)
837 {
838 	struct ath9k_htc_priv *priv = htc_handle->drv_priv;
839 	int ret;
840 
841 	ret = ath9k_htc_wait_for_target(priv);
842 	if (ret)
843 		return ret;
844 
845 	ret = ath9k_init_htc_services(priv, priv->ah->hw_version.devid,
846 				      priv->ah->hw_version.usbdev);
847 	ath9k_configure_leds(priv);
848 
849 	return ret;
850 }
851 #endif
852 
853 static int __init ath9k_htc_init(void)
854 {
855 	if (ath9k_hif_usb_init() < 0) {
856 		pr_err("No USB devices found, driver not installed\n");
857 		return -ENODEV;
858 	}
859 
860 	return 0;
861 }
862 module_init(ath9k_htc_init);
863 
864 static void __exit ath9k_htc_exit(void)
865 {
866 	ath9k_hif_usb_exit();
867 	pr_info("Driver unloaded\n");
868 }
869 module_exit(ath9k_htc_exit);
870