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 <asm/unaligned.h>
18 #include "htc.h"
19
20 MODULE_FIRMWARE(HTC_7010_MODULE_FW);
21 MODULE_FIRMWARE(HTC_9271_MODULE_FW);
22
23 static const struct usb_device_id ath9k_hif_usb_ids[] = {
24 { USB_DEVICE(0x0cf3, 0x9271) }, /* Atheros */
25 { USB_DEVICE(0x0cf3, 0x1006) }, /* Atheros */
26 { USB_DEVICE(0x0846, 0x9030) }, /* Netgear N150 */
27 { USB_DEVICE(0x07b8, 0x9271) }, /* Altai WA1011N-GU */
28 { USB_DEVICE(0x07D1, 0x3A10) }, /* Dlink Wireless 150 */
29 { USB_DEVICE(0x13D3, 0x3327) }, /* Azurewave */
30 { USB_DEVICE(0x13D3, 0x3328) }, /* Azurewave */
31 { USB_DEVICE(0x13D3, 0x3346) }, /* IMC Networks */
32 { USB_DEVICE(0x13D3, 0x3348) }, /* Azurewave */
33 { USB_DEVICE(0x13D3, 0x3349) }, /* Azurewave */
34 { USB_DEVICE(0x13D3, 0x3350) }, /* Azurewave */
35 { USB_DEVICE(0x04CA, 0x4605) }, /* Liteon */
36 { USB_DEVICE(0x040D, 0x3801) }, /* VIA */
37 { USB_DEVICE(0x0cf3, 0xb003) }, /* Ubiquiti WifiStation Ext */
38 { USB_DEVICE(0x0cf3, 0xb002) }, /* Ubiquiti WifiStation */
39 { USB_DEVICE(0x057c, 0x8403) }, /* AVM FRITZ!WLAN 11N v2 USB */
40 { USB_DEVICE(0x0471, 0x209e) }, /* Philips (or NXP) PTA01 */
41 { USB_DEVICE(0x1eda, 0x2315) }, /* AirTies */
42
43 { USB_DEVICE(0x0cf3, 0x7015),
44 .driver_info = AR9287_USB }, /* Atheros */
45 { USB_DEVICE(0x1668, 0x1200),
46 .driver_info = AR9287_USB }, /* Verizon */
47
48 { USB_DEVICE(0x0cf3, 0x7010),
49 .driver_info = AR9280_USB }, /* Atheros */
50 { USB_DEVICE(0x0846, 0x9018),
51 .driver_info = AR9280_USB }, /* Netgear WNDA3200 */
52 { USB_DEVICE(0x083A, 0xA704),
53 .driver_info = AR9280_USB }, /* SMC Networks */
54 { USB_DEVICE(0x0411, 0x017f),
55 .driver_info = AR9280_USB }, /* Sony UWA-BR100 */
56 { USB_DEVICE(0x0411, 0x0197),
57 .driver_info = AR9280_USB }, /* Buffalo WLI-UV-AG300P */
58 { USB_DEVICE(0x04da, 0x3904),
59 .driver_info = AR9280_USB },
60 { USB_DEVICE(0x0930, 0x0a08),
61 .driver_info = AR9280_USB }, /* Toshiba WLM-20U2 and GN-1080 */
62
63 { USB_DEVICE(0x0cf3, 0x20ff),
64 .driver_info = STORAGE_DEVICE },
65
66 { },
67 };
68
69 MODULE_DEVICE_TABLE(usb, ath9k_hif_usb_ids);
70
71 static int __hif_usb_tx(struct hif_device_usb *hif_dev);
72
hif_usb_regout_cb(struct urb * urb)73 static void hif_usb_regout_cb(struct urb *urb)
74 {
75 struct cmd_buf *cmd = (struct cmd_buf *)urb->context;
76
77 switch (urb->status) {
78 case 0:
79 break;
80 case -ENOENT:
81 case -ECONNRESET:
82 case -ENODEV:
83 case -ESHUTDOWN:
84 goto free;
85 default:
86 break;
87 }
88
89 if (cmd) {
90 ath9k_htc_txcompletion_cb(cmd->hif_dev->htc_handle,
91 cmd->skb, true);
92 kfree(cmd);
93 }
94
95 return;
96 free:
97 kfree_skb(cmd->skb);
98 kfree(cmd);
99 }
100
hif_usb_send_regout(struct hif_device_usb * hif_dev,struct sk_buff * skb)101 static int hif_usb_send_regout(struct hif_device_usb *hif_dev,
102 struct sk_buff *skb)
103 {
104 struct urb *urb;
105 struct cmd_buf *cmd;
106 int ret = 0;
107
108 urb = usb_alloc_urb(0, GFP_KERNEL);
109 if (urb == NULL)
110 return -ENOMEM;
111
112 cmd = kzalloc(sizeof(*cmd), GFP_KERNEL);
113 if (cmd == NULL) {
114 usb_free_urb(urb);
115 return -ENOMEM;
116 }
117
118 cmd->skb = skb;
119 cmd->hif_dev = hif_dev;
120
121 usb_fill_int_urb(urb, hif_dev->udev,
122 usb_sndintpipe(hif_dev->udev, USB_REG_OUT_PIPE),
123 skb->data, skb->len,
124 hif_usb_regout_cb, cmd, 1);
125
126 usb_anchor_urb(urb, &hif_dev->regout_submitted);
127 ret = usb_submit_urb(urb, GFP_KERNEL);
128 if (ret) {
129 usb_unanchor_urb(urb);
130 kfree(cmd);
131 }
132 usb_free_urb(urb);
133
134 return ret;
135 }
136
hif_usb_mgmt_cb(struct urb * urb)137 static void hif_usb_mgmt_cb(struct urb *urb)
138 {
139 struct cmd_buf *cmd = (struct cmd_buf *)urb->context;
140 struct hif_device_usb *hif_dev;
141 unsigned long flags;
142 bool txok = true;
143
144 if (!cmd || !cmd->skb || !cmd->hif_dev)
145 return;
146
147 hif_dev = cmd->hif_dev;
148
149 switch (urb->status) {
150 case 0:
151 break;
152 case -ENOENT:
153 case -ECONNRESET:
154 case -ENODEV:
155 case -ESHUTDOWN:
156 txok = false;
157
158 /*
159 * If the URBs are being flushed, no need to complete
160 * this packet.
161 */
162 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags);
163 if (hif_dev->tx.flags & HIF_USB_TX_FLUSH) {
164 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
165 dev_kfree_skb_any(cmd->skb);
166 kfree(cmd);
167 return;
168 }
169 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
170
171 break;
172 default:
173 txok = false;
174 break;
175 }
176
177 skb_pull(cmd->skb, 4);
178 ath9k_htc_txcompletion_cb(cmd->hif_dev->htc_handle,
179 cmd->skb, txok);
180 kfree(cmd);
181 }
182
hif_usb_send_mgmt(struct hif_device_usb * hif_dev,struct sk_buff * skb)183 static int hif_usb_send_mgmt(struct hif_device_usb *hif_dev,
184 struct sk_buff *skb)
185 {
186 struct urb *urb;
187 struct cmd_buf *cmd;
188 int ret = 0;
189 __le16 *hdr;
190
191 urb = usb_alloc_urb(0, GFP_ATOMIC);
192 if (urb == NULL)
193 return -ENOMEM;
194
195 cmd = kzalloc(sizeof(*cmd), GFP_ATOMIC);
196 if (cmd == NULL) {
197 usb_free_urb(urb);
198 return -ENOMEM;
199 }
200
201 cmd->skb = skb;
202 cmd->hif_dev = hif_dev;
203
204 hdr = skb_push(skb, 4);
205 *hdr++ = cpu_to_le16(skb->len - 4);
206 *hdr++ = cpu_to_le16(ATH_USB_TX_STREAM_MODE_TAG);
207
208 usb_fill_bulk_urb(urb, hif_dev->udev,
209 usb_sndbulkpipe(hif_dev->udev, USB_WLAN_TX_PIPE),
210 skb->data, skb->len,
211 hif_usb_mgmt_cb, cmd);
212
213 usb_anchor_urb(urb, &hif_dev->mgmt_submitted);
214 ret = usb_submit_urb(urb, GFP_ATOMIC);
215 if (ret) {
216 usb_unanchor_urb(urb);
217 kfree(cmd);
218 }
219 usb_free_urb(urb);
220
221 return ret;
222 }
223
ath9k_skb_queue_purge(struct hif_device_usb * hif_dev,struct sk_buff_head * list)224 static inline void ath9k_skb_queue_purge(struct hif_device_usb *hif_dev,
225 struct sk_buff_head *list)
226 {
227 struct sk_buff *skb;
228
229 while ((skb = __skb_dequeue(list)) != NULL) {
230 dev_kfree_skb_any(skb);
231 }
232 }
233
ath9k_skb_queue_complete(struct hif_device_usb * hif_dev,struct sk_buff_head * queue,bool txok)234 static inline void ath9k_skb_queue_complete(struct hif_device_usb *hif_dev,
235 struct sk_buff_head *queue,
236 bool txok)
237 {
238 struct sk_buff *skb;
239
240 while ((skb = __skb_dequeue(queue)) != NULL) {
241 #ifdef CONFIG_ATH9K_HTC_DEBUGFS
242 int ln = skb->len;
243 #endif
244 ath9k_htc_txcompletion_cb(hif_dev->htc_handle,
245 skb, txok);
246 if (txok) {
247 TX_STAT_INC(skb_success);
248 TX_STAT_ADD(skb_success_bytes, ln);
249 }
250 else
251 TX_STAT_INC(skb_failed);
252 }
253 }
254
hif_usb_tx_cb(struct urb * urb)255 static void hif_usb_tx_cb(struct urb *urb)
256 {
257 struct tx_buf *tx_buf = (struct tx_buf *) urb->context;
258 struct hif_device_usb *hif_dev;
259 bool txok = true;
260
261 if (!tx_buf || !tx_buf->hif_dev)
262 return;
263
264 hif_dev = tx_buf->hif_dev;
265
266 switch (urb->status) {
267 case 0:
268 break;
269 case -ENOENT:
270 case -ECONNRESET:
271 case -ENODEV:
272 case -ESHUTDOWN:
273 txok = false;
274
275 /*
276 * If the URBs are being flushed, no need to add this
277 * URB to the free list.
278 */
279 spin_lock(&hif_dev->tx.tx_lock);
280 if (hif_dev->tx.flags & HIF_USB_TX_FLUSH) {
281 spin_unlock(&hif_dev->tx.tx_lock);
282 ath9k_skb_queue_purge(hif_dev, &tx_buf->skb_queue);
283 return;
284 }
285 spin_unlock(&hif_dev->tx.tx_lock);
286
287 break;
288 default:
289 txok = false;
290 break;
291 }
292
293 ath9k_skb_queue_complete(hif_dev, &tx_buf->skb_queue, txok);
294
295 /* Re-initialize the SKB queue */
296 tx_buf->len = tx_buf->offset = 0;
297 __skb_queue_head_init(&tx_buf->skb_queue);
298
299 /* Add this TX buffer to the free list */
300 spin_lock(&hif_dev->tx.tx_lock);
301 list_move_tail(&tx_buf->list, &hif_dev->tx.tx_buf);
302 hif_dev->tx.tx_buf_cnt++;
303 if (!(hif_dev->tx.flags & HIF_USB_TX_STOP))
304 __hif_usb_tx(hif_dev); /* Check for pending SKBs */
305 TX_STAT_INC(buf_completed);
306 spin_unlock(&hif_dev->tx.tx_lock);
307 }
308
309 /* TX lock has to be taken */
__hif_usb_tx(struct hif_device_usb * hif_dev)310 static int __hif_usb_tx(struct hif_device_usb *hif_dev)
311 {
312 struct tx_buf *tx_buf = NULL;
313 struct sk_buff *nskb = NULL;
314 int ret = 0, i;
315 u16 tx_skb_cnt = 0;
316 u8 *buf;
317 __le16 *hdr;
318
319 if (hif_dev->tx.tx_skb_cnt == 0)
320 return 0;
321
322 /* Check if a free TX buffer is available */
323 if (list_empty(&hif_dev->tx.tx_buf))
324 return 0;
325
326 tx_buf = list_first_entry(&hif_dev->tx.tx_buf, struct tx_buf, list);
327 list_move_tail(&tx_buf->list, &hif_dev->tx.tx_pending);
328 hif_dev->tx.tx_buf_cnt--;
329
330 tx_skb_cnt = min_t(u16, hif_dev->tx.tx_skb_cnt, MAX_TX_AGGR_NUM);
331
332 for (i = 0; i < tx_skb_cnt; i++) {
333 nskb = __skb_dequeue(&hif_dev->tx.tx_skb_queue);
334
335 /* Should never be NULL */
336 BUG_ON(!nskb);
337
338 hif_dev->tx.tx_skb_cnt--;
339
340 buf = tx_buf->buf;
341 buf += tx_buf->offset;
342 hdr = (__le16 *)buf;
343 *hdr++ = cpu_to_le16(nskb->len);
344 *hdr++ = cpu_to_le16(ATH_USB_TX_STREAM_MODE_TAG);
345 buf += 4;
346 memcpy(buf, nskb->data, nskb->len);
347 tx_buf->len = nskb->len + 4;
348
349 if (i < (tx_skb_cnt - 1))
350 tx_buf->offset += (((tx_buf->len - 1) / 4) + 1) * 4;
351
352 if (i == (tx_skb_cnt - 1))
353 tx_buf->len += tx_buf->offset;
354
355 __skb_queue_tail(&tx_buf->skb_queue, nskb);
356 TX_STAT_INC(skb_queued);
357 }
358
359 usb_fill_bulk_urb(tx_buf->urb, hif_dev->udev,
360 usb_sndbulkpipe(hif_dev->udev, USB_WLAN_TX_PIPE),
361 tx_buf->buf, tx_buf->len,
362 hif_usb_tx_cb, tx_buf);
363
364 ret = usb_submit_urb(tx_buf->urb, GFP_ATOMIC);
365 if (ret) {
366 tx_buf->len = tx_buf->offset = 0;
367 ath9k_skb_queue_complete(hif_dev, &tx_buf->skb_queue, false);
368 __skb_queue_head_init(&tx_buf->skb_queue);
369 list_move_tail(&tx_buf->list, &hif_dev->tx.tx_buf);
370 hif_dev->tx.tx_buf_cnt++;
371 }
372
373 if (!ret)
374 TX_STAT_INC(buf_queued);
375
376 return ret;
377 }
378
hif_usb_send_tx(struct hif_device_usb * hif_dev,struct sk_buff * skb)379 static int hif_usb_send_tx(struct hif_device_usb *hif_dev, struct sk_buff *skb)
380 {
381 struct ath9k_htc_tx_ctl *tx_ctl;
382 unsigned long flags;
383 int ret = 0;
384
385 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags);
386
387 if (hif_dev->tx.flags & HIF_USB_TX_STOP) {
388 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
389 return -ENODEV;
390 }
391
392 /* Check if the max queue count has been reached */
393 if (hif_dev->tx.tx_skb_cnt > MAX_TX_BUF_NUM) {
394 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
395 return -ENOMEM;
396 }
397
398 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
399
400 tx_ctl = HTC_SKB_CB(skb);
401
402 /* Mgmt/Beacon frames don't use the TX buffer pool */
403 if ((tx_ctl->type == ATH9K_HTC_MGMT) ||
404 (tx_ctl->type == ATH9K_HTC_BEACON)) {
405 ret = hif_usb_send_mgmt(hif_dev, skb);
406 }
407
408 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags);
409
410 if ((tx_ctl->type == ATH9K_HTC_NORMAL) ||
411 (tx_ctl->type == ATH9K_HTC_AMPDU)) {
412 __skb_queue_tail(&hif_dev->tx.tx_skb_queue, skb);
413 hif_dev->tx.tx_skb_cnt++;
414 }
415
416 /* Check if AMPDUs have to be sent immediately */
417 if ((hif_dev->tx.tx_buf_cnt == MAX_TX_URB_NUM) &&
418 (hif_dev->tx.tx_skb_cnt < 2)) {
419 __hif_usb_tx(hif_dev);
420 }
421
422 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
423
424 return ret;
425 }
426
hif_usb_start(void * hif_handle)427 static void hif_usb_start(void *hif_handle)
428 {
429 struct hif_device_usb *hif_dev = hif_handle;
430 unsigned long flags;
431
432 hif_dev->flags |= HIF_USB_START;
433
434 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags);
435 hif_dev->tx.flags &= ~HIF_USB_TX_STOP;
436 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
437 }
438
hif_usb_stop(void * hif_handle)439 static void hif_usb_stop(void *hif_handle)
440 {
441 struct hif_device_usb *hif_dev = hif_handle;
442 struct tx_buf *tx_buf = NULL, *tx_buf_tmp = NULL;
443 unsigned long flags;
444
445 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags);
446 ath9k_skb_queue_complete(hif_dev, &hif_dev->tx.tx_skb_queue, false);
447 hif_dev->tx.tx_skb_cnt = 0;
448 hif_dev->tx.flags |= HIF_USB_TX_STOP;
449 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
450
451 /* The pending URBs have to be canceled. */
452 list_for_each_entry_safe(tx_buf, tx_buf_tmp,
453 &hif_dev->tx.tx_pending, list) {
454 usb_kill_urb(tx_buf->urb);
455 }
456
457 usb_kill_anchored_urbs(&hif_dev->mgmt_submitted);
458 }
459
hif_usb_send(void * hif_handle,u8 pipe_id,struct sk_buff * skb)460 static int hif_usb_send(void *hif_handle, u8 pipe_id, struct sk_buff *skb)
461 {
462 struct hif_device_usb *hif_dev = hif_handle;
463 int ret = 0;
464
465 switch (pipe_id) {
466 case USB_WLAN_TX_PIPE:
467 ret = hif_usb_send_tx(hif_dev, skb);
468 break;
469 case USB_REG_OUT_PIPE:
470 ret = hif_usb_send_regout(hif_dev, skb);
471 break;
472 default:
473 dev_err(&hif_dev->udev->dev,
474 "ath9k_htc: Invalid TX pipe: %d\n", pipe_id);
475 ret = -EINVAL;
476 break;
477 }
478
479 return ret;
480 }
481
check_index(struct sk_buff * skb,u8 idx)482 static inline bool check_index(struct sk_buff *skb, u8 idx)
483 {
484 struct ath9k_htc_tx_ctl *tx_ctl;
485
486 tx_ctl = HTC_SKB_CB(skb);
487
488 if ((tx_ctl->type == ATH9K_HTC_AMPDU) &&
489 (tx_ctl->sta_idx == idx))
490 return true;
491
492 return false;
493 }
494
hif_usb_sta_drain(void * hif_handle,u8 idx)495 static void hif_usb_sta_drain(void *hif_handle, u8 idx)
496 {
497 struct hif_device_usb *hif_dev = hif_handle;
498 struct sk_buff *skb, *tmp;
499 unsigned long flags;
500
501 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags);
502
503 skb_queue_walk_safe(&hif_dev->tx.tx_skb_queue, skb, tmp) {
504 if (check_index(skb, idx)) {
505 __skb_unlink(skb, &hif_dev->tx.tx_skb_queue);
506 ath9k_htc_txcompletion_cb(hif_dev->htc_handle,
507 skb, false);
508 hif_dev->tx.tx_skb_cnt--;
509 TX_STAT_INC(skb_failed);
510 }
511 }
512
513 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
514 }
515
516 static struct ath9k_htc_hif hif_usb = {
517 .transport = ATH9K_HIF_USB,
518 .name = "ath9k_hif_usb",
519
520 .control_ul_pipe = USB_REG_OUT_PIPE,
521 .control_dl_pipe = USB_REG_IN_PIPE,
522
523 .start = hif_usb_start,
524 .stop = hif_usb_stop,
525 .sta_drain = hif_usb_sta_drain,
526 .send = hif_usb_send,
527 };
528
ath9k_hif_usb_rx_stream(struct hif_device_usb * hif_dev,struct sk_buff * skb)529 static void ath9k_hif_usb_rx_stream(struct hif_device_usb *hif_dev,
530 struct sk_buff *skb)
531 {
532 struct sk_buff *nskb, *skb_pool[MAX_PKT_NUM_IN_TRANSFER];
533 int index = 0, i, len = skb->len;
534 int rx_remain_len, rx_pkt_len;
535 u16 pool_index = 0;
536 u8 *ptr;
537
538 spin_lock(&hif_dev->rx_lock);
539
540 rx_remain_len = hif_dev->rx_remain_len;
541 rx_pkt_len = hif_dev->rx_transfer_len;
542
543 if (rx_remain_len != 0) {
544 struct sk_buff *remain_skb = hif_dev->remain_skb;
545
546 if (remain_skb) {
547 ptr = (u8 *) remain_skb->data;
548
549 index = rx_remain_len;
550 rx_remain_len -= hif_dev->rx_pad_len;
551 ptr += rx_pkt_len;
552
553 memcpy(ptr, skb->data, rx_remain_len);
554
555 rx_pkt_len += rx_remain_len;
556 hif_dev->rx_remain_len = 0;
557 skb_put(remain_skb, rx_pkt_len);
558
559 skb_pool[pool_index++] = remain_skb;
560
561 } else {
562 index = rx_remain_len;
563 }
564 }
565
566 spin_unlock(&hif_dev->rx_lock);
567
568 while (index < len) {
569 u16 pkt_len;
570 u16 pkt_tag;
571 u16 pad_len;
572 int chk_idx;
573
574 ptr = (u8 *) skb->data;
575
576 pkt_len = get_unaligned_le16(ptr + index);
577 pkt_tag = get_unaligned_le16(ptr + index + 2);
578
579 if (pkt_tag != ATH_USB_RX_STREAM_MODE_TAG) {
580 RX_STAT_INC(skb_dropped);
581 return;
582 }
583
584 pad_len = 4 - (pkt_len & 0x3);
585 if (pad_len == 4)
586 pad_len = 0;
587
588 chk_idx = index;
589 index = index + 4 + pkt_len + pad_len;
590
591 if (index > MAX_RX_BUF_SIZE) {
592 spin_lock(&hif_dev->rx_lock);
593 hif_dev->rx_remain_len = index - MAX_RX_BUF_SIZE;
594 hif_dev->rx_transfer_len =
595 MAX_RX_BUF_SIZE - chk_idx - 4;
596 hif_dev->rx_pad_len = pad_len;
597
598 nskb = __dev_alloc_skb(pkt_len + 32, GFP_ATOMIC);
599 if (!nskb) {
600 dev_err(&hif_dev->udev->dev,
601 "ath9k_htc: RX memory allocation error\n");
602 spin_unlock(&hif_dev->rx_lock);
603 goto err;
604 }
605 skb_reserve(nskb, 32);
606 RX_STAT_INC(skb_allocated);
607
608 memcpy(nskb->data, &(skb->data[chk_idx+4]),
609 hif_dev->rx_transfer_len);
610
611 /* Record the buffer pointer */
612 hif_dev->remain_skb = nskb;
613 spin_unlock(&hif_dev->rx_lock);
614 } else {
615 nskb = __dev_alloc_skb(pkt_len + 32, GFP_ATOMIC);
616 if (!nskb) {
617 dev_err(&hif_dev->udev->dev,
618 "ath9k_htc: RX memory allocation error\n");
619 goto err;
620 }
621 skb_reserve(nskb, 32);
622 RX_STAT_INC(skb_allocated);
623
624 memcpy(nskb->data, &(skb->data[chk_idx+4]), pkt_len);
625 skb_put(nskb, pkt_len);
626 skb_pool[pool_index++] = nskb;
627 }
628 }
629
630 err:
631 for (i = 0; i < pool_index; i++) {
632 RX_STAT_ADD(skb_completed_bytes, skb_pool[i]->len);
633 ath9k_htc_rx_msg(hif_dev->htc_handle, skb_pool[i],
634 skb_pool[i]->len, USB_WLAN_RX_PIPE);
635 RX_STAT_INC(skb_completed);
636 }
637 }
638
ath9k_hif_usb_rx_cb(struct urb * urb)639 static void ath9k_hif_usb_rx_cb(struct urb *urb)
640 {
641 struct sk_buff *skb = (struct sk_buff *) urb->context;
642 struct hif_device_usb *hif_dev =
643 usb_get_intfdata(usb_ifnum_to_if(urb->dev, 0));
644 int ret;
645
646 if (!skb)
647 return;
648
649 if (!hif_dev)
650 goto free;
651
652 switch (urb->status) {
653 case 0:
654 break;
655 case -ENOENT:
656 case -ECONNRESET:
657 case -ENODEV:
658 case -ESHUTDOWN:
659 goto free;
660 default:
661 goto resubmit;
662 }
663
664 if (likely(urb->actual_length != 0)) {
665 skb_put(skb, urb->actual_length);
666 ath9k_hif_usb_rx_stream(hif_dev, skb);
667 }
668
669 resubmit:
670 skb_reset_tail_pointer(skb);
671 skb_trim(skb, 0);
672
673 usb_anchor_urb(urb, &hif_dev->rx_submitted);
674 ret = usb_submit_urb(urb, GFP_ATOMIC);
675 if (ret) {
676 usb_unanchor_urb(urb);
677 goto free;
678 }
679
680 return;
681 free:
682 kfree_skb(skb);
683 }
684
ath9k_hif_usb_reg_in_cb(struct urb * urb)685 static void ath9k_hif_usb_reg_in_cb(struct urb *urb)
686 {
687 struct sk_buff *skb = (struct sk_buff *) urb->context;
688 struct sk_buff *nskb;
689 struct hif_device_usb *hif_dev =
690 usb_get_intfdata(usb_ifnum_to_if(urb->dev, 0));
691 int ret;
692
693 if (!skb)
694 return;
695
696 if (!hif_dev)
697 goto free;
698
699 switch (urb->status) {
700 case 0:
701 break;
702 case -ENOENT:
703 case -ECONNRESET:
704 case -ENODEV:
705 case -ESHUTDOWN:
706 goto free;
707 default:
708 skb_reset_tail_pointer(skb);
709 skb_trim(skb, 0);
710
711 goto resubmit;
712 }
713
714 if (likely(urb->actual_length != 0)) {
715 skb_put(skb, urb->actual_length);
716
717 /* Process the command first */
718 ath9k_htc_rx_msg(hif_dev->htc_handle, skb,
719 skb->len, USB_REG_IN_PIPE);
720
721
722 nskb = alloc_skb(MAX_REG_IN_BUF_SIZE, GFP_ATOMIC);
723 if (!nskb) {
724 dev_err(&hif_dev->udev->dev,
725 "ath9k_htc: REG_IN memory allocation failure\n");
726 urb->context = NULL;
727 return;
728 }
729
730 usb_fill_int_urb(urb, hif_dev->udev,
731 usb_rcvintpipe(hif_dev->udev,
732 USB_REG_IN_PIPE),
733 nskb->data, MAX_REG_IN_BUF_SIZE,
734 ath9k_hif_usb_reg_in_cb, nskb, 1);
735 }
736
737 resubmit:
738 usb_anchor_urb(urb, &hif_dev->reg_in_submitted);
739 ret = usb_submit_urb(urb, GFP_ATOMIC);
740 if (ret) {
741 usb_unanchor_urb(urb);
742 goto free;
743 }
744
745 return;
746 free:
747 kfree_skb(skb);
748 urb->context = NULL;
749 }
750
ath9k_hif_usb_dealloc_tx_urbs(struct hif_device_usb * hif_dev)751 static void ath9k_hif_usb_dealloc_tx_urbs(struct hif_device_usb *hif_dev)
752 {
753 struct tx_buf *tx_buf = NULL, *tx_buf_tmp = NULL;
754 unsigned long flags;
755
756 list_for_each_entry_safe(tx_buf, tx_buf_tmp,
757 &hif_dev->tx.tx_buf, list) {
758 usb_kill_urb(tx_buf->urb);
759 list_del(&tx_buf->list);
760 usb_free_urb(tx_buf->urb);
761 kfree(tx_buf->buf);
762 kfree(tx_buf);
763 }
764
765 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags);
766 hif_dev->tx.flags |= HIF_USB_TX_FLUSH;
767 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
768
769 list_for_each_entry_safe(tx_buf, tx_buf_tmp,
770 &hif_dev->tx.tx_pending, list) {
771 usb_kill_urb(tx_buf->urb);
772 list_del(&tx_buf->list);
773 usb_free_urb(tx_buf->urb);
774 kfree(tx_buf->buf);
775 kfree(tx_buf);
776 }
777
778 usb_kill_anchored_urbs(&hif_dev->mgmt_submitted);
779 }
780
ath9k_hif_usb_alloc_tx_urbs(struct hif_device_usb * hif_dev)781 static int ath9k_hif_usb_alloc_tx_urbs(struct hif_device_usb *hif_dev)
782 {
783 struct tx_buf *tx_buf;
784 int i;
785
786 INIT_LIST_HEAD(&hif_dev->tx.tx_buf);
787 INIT_LIST_HEAD(&hif_dev->tx.tx_pending);
788 spin_lock_init(&hif_dev->tx.tx_lock);
789 __skb_queue_head_init(&hif_dev->tx.tx_skb_queue);
790 init_usb_anchor(&hif_dev->mgmt_submitted);
791
792 for (i = 0; i < MAX_TX_URB_NUM; i++) {
793 tx_buf = kzalloc(sizeof(struct tx_buf), GFP_KERNEL);
794 if (!tx_buf)
795 goto err;
796
797 tx_buf->buf = kzalloc(MAX_TX_BUF_SIZE, GFP_KERNEL);
798 if (!tx_buf->buf)
799 goto err;
800
801 tx_buf->urb = usb_alloc_urb(0, GFP_KERNEL);
802 if (!tx_buf->urb)
803 goto err;
804
805 tx_buf->hif_dev = hif_dev;
806 __skb_queue_head_init(&tx_buf->skb_queue);
807
808 list_add_tail(&tx_buf->list, &hif_dev->tx.tx_buf);
809 }
810
811 hif_dev->tx.tx_buf_cnt = MAX_TX_URB_NUM;
812
813 return 0;
814 err:
815 if (tx_buf) {
816 kfree(tx_buf->buf);
817 kfree(tx_buf);
818 }
819 ath9k_hif_usb_dealloc_tx_urbs(hif_dev);
820 return -ENOMEM;
821 }
822
ath9k_hif_usb_dealloc_rx_urbs(struct hif_device_usb * hif_dev)823 static void ath9k_hif_usb_dealloc_rx_urbs(struct hif_device_usb *hif_dev)
824 {
825 usb_kill_anchored_urbs(&hif_dev->rx_submitted);
826 }
827
ath9k_hif_usb_alloc_rx_urbs(struct hif_device_usb * hif_dev)828 static int ath9k_hif_usb_alloc_rx_urbs(struct hif_device_usb *hif_dev)
829 {
830 struct urb *urb = NULL;
831 struct sk_buff *skb = NULL;
832 int i, ret;
833
834 init_usb_anchor(&hif_dev->rx_submitted);
835 spin_lock_init(&hif_dev->rx_lock);
836
837 for (i = 0; i < MAX_RX_URB_NUM; i++) {
838
839 /* Allocate URB */
840 urb = usb_alloc_urb(0, GFP_KERNEL);
841 if (urb == NULL) {
842 ret = -ENOMEM;
843 goto err_urb;
844 }
845
846 /* Allocate buffer */
847 skb = alloc_skb(MAX_RX_BUF_SIZE, GFP_KERNEL);
848 if (!skb) {
849 ret = -ENOMEM;
850 goto err_skb;
851 }
852
853 usb_fill_bulk_urb(urb, hif_dev->udev,
854 usb_rcvbulkpipe(hif_dev->udev,
855 USB_WLAN_RX_PIPE),
856 skb->data, MAX_RX_BUF_SIZE,
857 ath9k_hif_usb_rx_cb, skb);
858
859 /* Anchor URB */
860 usb_anchor_urb(urb, &hif_dev->rx_submitted);
861
862 /* Submit URB */
863 ret = usb_submit_urb(urb, GFP_KERNEL);
864 if (ret) {
865 usb_unanchor_urb(urb);
866 goto err_submit;
867 }
868
869 /*
870 * Drop reference count.
871 * This ensures that the URB is freed when killing them.
872 */
873 usb_free_urb(urb);
874 }
875
876 return 0;
877
878 err_submit:
879 kfree_skb(skb);
880 err_skb:
881 usb_free_urb(urb);
882 err_urb:
883 ath9k_hif_usb_dealloc_rx_urbs(hif_dev);
884 return ret;
885 }
886
ath9k_hif_usb_dealloc_reg_in_urbs(struct hif_device_usb * hif_dev)887 static void ath9k_hif_usb_dealloc_reg_in_urbs(struct hif_device_usb *hif_dev)
888 {
889 usb_kill_anchored_urbs(&hif_dev->reg_in_submitted);
890 }
891
ath9k_hif_usb_alloc_reg_in_urbs(struct hif_device_usb * hif_dev)892 static int ath9k_hif_usb_alloc_reg_in_urbs(struct hif_device_usb *hif_dev)
893 {
894 struct urb *urb = NULL;
895 struct sk_buff *skb = NULL;
896 int i, ret;
897
898 init_usb_anchor(&hif_dev->reg_in_submitted);
899
900 for (i = 0; i < MAX_REG_IN_URB_NUM; i++) {
901
902 /* Allocate URB */
903 urb = usb_alloc_urb(0, GFP_KERNEL);
904 if (urb == NULL) {
905 ret = -ENOMEM;
906 goto err_urb;
907 }
908
909 /* Allocate buffer */
910 skb = alloc_skb(MAX_REG_IN_BUF_SIZE, GFP_KERNEL);
911 if (!skb) {
912 ret = -ENOMEM;
913 goto err_skb;
914 }
915
916 usb_fill_int_urb(urb, hif_dev->udev,
917 usb_rcvintpipe(hif_dev->udev,
918 USB_REG_IN_PIPE),
919 skb->data, MAX_REG_IN_BUF_SIZE,
920 ath9k_hif_usb_reg_in_cb, skb, 1);
921
922 /* Anchor URB */
923 usb_anchor_urb(urb, &hif_dev->reg_in_submitted);
924
925 /* Submit URB */
926 ret = usb_submit_urb(urb, GFP_KERNEL);
927 if (ret) {
928 usb_unanchor_urb(urb);
929 goto err_submit;
930 }
931
932 /*
933 * Drop reference count.
934 * This ensures that the URB is freed when killing them.
935 */
936 usb_free_urb(urb);
937 }
938
939 return 0;
940
941 err_submit:
942 kfree_skb(skb);
943 err_skb:
944 usb_free_urb(urb);
945 err_urb:
946 ath9k_hif_usb_dealloc_reg_in_urbs(hif_dev);
947 return ret;
948 }
949
ath9k_hif_usb_alloc_urbs(struct hif_device_usb * hif_dev)950 static int ath9k_hif_usb_alloc_urbs(struct hif_device_usb *hif_dev)
951 {
952 /* Register Write */
953 init_usb_anchor(&hif_dev->regout_submitted);
954
955 /* TX */
956 if (ath9k_hif_usb_alloc_tx_urbs(hif_dev) < 0)
957 goto err;
958
959 /* RX */
960 if (ath9k_hif_usb_alloc_rx_urbs(hif_dev) < 0)
961 goto err_rx;
962
963 /* Register Read */
964 if (ath9k_hif_usb_alloc_reg_in_urbs(hif_dev) < 0)
965 goto err_reg;
966
967 return 0;
968 err_reg:
969 ath9k_hif_usb_dealloc_rx_urbs(hif_dev);
970 err_rx:
971 ath9k_hif_usb_dealloc_tx_urbs(hif_dev);
972 err:
973 return -ENOMEM;
974 }
975
ath9k_hif_usb_dealloc_urbs(struct hif_device_usb * hif_dev)976 static void ath9k_hif_usb_dealloc_urbs(struct hif_device_usb *hif_dev)
977 {
978 usb_kill_anchored_urbs(&hif_dev->regout_submitted);
979 ath9k_hif_usb_dealloc_reg_in_urbs(hif_dev);
980 ath9k_hif_usb_dealloc_tx_urbs(hif_dev);
981 ath9k_hif_usb_dealloc_rx_urbs(hif_dev);
982 }
983
ath9k_hif_usb_download_fw(struct hif_device_usb * hif_dev)984 static int ath9k_hif_usb_download_fw(struct hif_device_usb *hif_dev)
985 {
986 int transfer, err;
987 const void *data = hif_dev->fw_data;
988 size_t len = hif_dev->fw_size;
989 u32 addr = AR9271_FIRMWARE;
990 u8 *buf = kzalloc(4096, GFP_KERNEL);
991 u32 firm_offset;
992
993 if (!buf)
994 return -ENOMEM;
995
996 while (len) {
997 transfer = min_t(size_t, len, 4096);
998 memcpy(buf, data, transfer);
999
1000 err = usb_control_msg(hif_dev->udev,
1001 usb_sndctrlpipe(hif_dev->udev, 0),
1002 FIRMWARE_DOWNLOAD, 0x40 | USB_DIR_OUT,
1003 addr >> 8, 0, buf, transfer,
1004 USB_MSG_TIMEOUT);
1005 if (err < 0) {
1006 kfree(buf);
1007 return err;
1008 }
1009
1010 len -= transfer;
1011 data += transfer;
1012 addr += transfer;
1013 }
1014 kfree(buf);
1015
1016 if (IS_AR7010_DEVICE(hif_dev->usb_device_id->driver_info))
1017 firm_offset = AR7010_FIRMWARE_TEXT;
1018 else
1019 firm_offset = AR9271_FIRMWARE_TEXT;
1020
1021 /*
1022 * Issue FW download complete command to firmware.
1023 */
1024 err = usb_control_msg(hif_dev->udev, usb_sndctrlpipe(hif_dev->udev, 0),
1025 FIRMWARE_DOWNLOAD_COMP,
1026 0x40 | USB_DIR_OUT,
1027 firm_offset >> 8, 0, NULL, 0, USB_MSG_TIMEOUT);
1028 if (err)
1029 return -EIO;
1030
1031 dev_info(&hif_dev->udev->dev, "ath9k_htc: Transferred FW: %s, size: %ld\n",
1032 hif_dev->fw_name, (unsigned long) hif_dev->fw_size);
1033
1034 return 0;
1035 }
1036
ath9k_hif_usb_dev_init(struct hif_device_usb * hif_dev)1037 static int ath9k_hif_usb_dev_init(struct hif_device_usb *hif_dev)
1038 {
1039 int ret;
1040
1041 ret = ath9k_hif_usb_download_fw(hif_dev);
1042 if (ret) {
1043 dev_err(&hif_dev->udev->dev,
1044 "ath9k_htc: Firmware - %s download failed\n",
1045 hif_dev->fw_name);
1046 return ret;
1047 }
1048
1049 /* Alloc URBs */
1050 ret = ath9k_hif_usb_alloc_urbs(hif_dev);
1051 if (ret) {
1052 dev_err(&hif_dev->udev->dev,
1053 "ath9k_htc: Unable to allocate URBs\n");
1054 return ret;
1055 }
1056
1057 return 0;
1058 }
1059
ath9k_hif_usb_dev_deinit(struct hif_device_usb * hif_dev)1060 static void ath9k_hif_usb_dev_deinit(struct hif_device_usb *hif_dev)
1061 {
1062 ath9k_hif_usb_dealloc_urbs(hif_dev);
1063 }
1064
1065 /*
1066 * If initialization fails or the FW cannot be retrieved,
1067 * detach the device.
1068 */
ath9k_hif_usb_firmware_fail(struct hif_device_usb * hif_dev)1069 static void ath9k_hif_usb_firmware_fail(struct hif_device_usb *hif_dev)
1070 {
1071 struct device *dev = &hif_dev->udev->dev;
1072 struct device *parent = dev->parent;
1073
1074 complete_all(&hif_dev->fw_done);
1075
1076 if (parent)
1077 device_lock(parent);
1078
1079 device_release_driver(dev);
1080
1081 if (parent)
1082 device_unlock(parent);
1083 }
1084
1085 static void ath9k_hif_usb_firmware_cb(const struct firmware *fw, void *context);
1086
1087 /* taken from iwlwifi */
ath9k_hif_request_firmware(struct hif_device_usb * hif_dev,bool first)1088 static int ath9k_hif_request_firmware(struct hif_device_usb *hif_dev,
1089 bool first)
1090 {
1091 char index[8], *chip;
1092 int ret;
1093
1094 if (first) {
1095 if (htc_use_dev_fw) {
1096 hif_dev->fw_minor_index = FIRMWARE_MINOR_IDX_MAX + 1;
1097 sprintf(index, "%s", "dev");
1098 } else {
1099 hif_dev->fw_minor_index = FIRMWARE_MINOR_IDX_MAX;
1100 sprintf(index, "%d", hif_dev->fw_minor_index);
1101 }
1102 } else {
1103 hif_dev->fw_minor_index--;
1104 sprintf(index, "%d", hif_dev->fw_minor_index);
1105 }
1106
1107 /* test for FW 1.3 */
1108 if (MAJOR_VERSION_REQ == 1 && hif_dev->fw_minor_index == 3) {
1109 const char *filename;
1110
1111 if (IS_AR7010_DEVICE(hif_dev->usb_device_id->driver_info))
1112 filename = FIRMWARE_AR7010_1_1;
1113 else
1114 filename = FIRMWARE_AR9271;
1115
1116 /* expected fw locations:
1117 * - htc_9271.fw (stable version 1.3, depricated)
1118 */
1119 snprintf(hif_dev->fw_name, sizeof(hif_dev->fw_name),
1120 "%s", filename);
1121
1122 } else if (hif_dev->fw_minor_index < FIRMWARE_MINOR_IDX_MIN) {
1123 dev_err(&hif_dev->udev->dev, "no suitable firmware found!\n");
1124
1125 return -ENOENT;
1126 } else {
1127 if (IS_AR7010_DEVICE(hif_dev->usb_device_id->driver_info))
1128 chip = "7010";
1129 else
1130 chip = "9271";
1131
1132 /* expected fw locations:
1133 * - ath9k_htc/htc_9271-1.dev.0.fw (development version)
1134 * - ath9k_htc/htc_9271-1.4.0.fw (stable version)
1135 */
1136 snprintf(hif_dev->fw_name, sizeof(hif_dev->fw_name),
1137 "%s/htc_%s-%d.%s.0.fw", HTC_FW_PATH,
1138 chip, MAJOR_VERSION_REQ, index);
1139 }
1140
1141 ret = request_firmware_nowait(THIS_MODULE, true, hif_dev->fw_name,
1142 &hif_dev->udev->dev, GFP_KERNEL,
1143 hif_dev, ath9k_hif_usb_firmware_cb);
1144 if (ret) {
1145 dev_err(&hif_dev->udev->dev,
1146 "ath9k_htc: Async request for firmware %s failed\n",
1147 hif_dev->fw_name);
1148 return ret;
1149 }
1150
1151 dev_info(&hif_dev->udev->dev, "ath9k_htc: Firmware %s requested\n",
1152 hif_dev->fw_name);
1153
1154 return ret;
1155 }
1156
ath9k_hif_usb_firmware_cb(const struct firmware * fw,void * context)1157 static void ath9k_hif_usb_firmware_cb(const struct firmware *fw, void *context)
1158 {
1159 struct hif_device_usb *hif_dev = context;
1160 int ret;
1161
1162 if (!fw) {
1163 ret = ath9k_hif_request_firmware(hif_dev, false);
1164 if (!ret)
1165 return;
1166
1167 dev_err(&hif_dev->udev->dev,
1168 "ath9k_htc: Failed to get firmware %s\n",
1169 hif_dev->fw_name);
1170 goto err_fw;
1171 }
1172
1173 hif_dev->htc_handle = ath9k_htc_hw_alloc(hif_dev, &hif_usb,
1174 &hif_dev->udev->dev);
1175 if (hif_dev->htc_handle == NULL)
1176 goto err_dev_alloc;
1177
1178 hif_dev->fw_data = fw->data;
1179 hif_dev->fw_size = fw->size;
1180
1181 /* Proceed with initialization */
1182
1183 ret = ath9k_hif_usb_dev_init(hif_dev);
1184 if (ret)
1185 goto err_dev_init;
1186
1187 ret = ath9k_htc_hw_init(hif_dev->htc_handle,
1188 &hif_dev->interface->dev,
1189 hif_dev->usb_device_id->idProduct,
1190 hif_dev->udev->product,
1191 hif_dev->usb_device_id->driver_info);
1192 if (ret) {
1193 ret = -EINVAL;
1194 goto err_htc_hw_init;
1195 }
1196
1197 release_firmware(fw);
1198 hif_dev->flags |= HIF_USB_READY;
1199 complete_all(&hif_dev->fw_done);
1200
1201 return;
1202
1203 err_htc_hw_init:
1204 ath9k_hif_usb_dev_deinit(hif_dev);
1205 err_dev_init:
1206 ath9k_htc_hw_free(hif_dev->htc_handle);
1207 err_dev_alloc:
1208 release_firmware(fw);
1209 err_fw:
1210 ath9k_hif_usb_firmware_fail(hif_dev);
1211 }
1212
1213 /*
1214 * An exact copy of the function from zd1211rw.
1215 */
send_eject_command(struct usb_interface * interface)1216 static int send_eject_command(struct usb_interface *interface)
1217 {
1218 struct usb_device *udev = interface_to_usbdev(interface);
1219 struct usb_host_interface *iface_desc = interface->cur_altsetting;
1220 struct usb_endpoint_descriptor *endpoint;
1221 unsigned char *cmd;
1222 u8 bulk_out_ep;
1223 int r;
1224
1225 if (iface_desc->desc.bNumEndpoints < 2)
1226 return -ENODEV;
1227
1228 /* Find bulk out endpoint */
1229 for (r = 1; r >= 0; r--) {
1230 endpoint = &iface_desc->endpoint[r].desc;
1231 if (usb_endpoint_dir_out(endpoint) &&
1232 usb_endpoint_xfer_bulk(endpoint)) {
1233 bulk_out_ep = endpoint->bEndpointAddress;
1234 break;
1235 }
1236 }
1237 if (r == -1) {
1238 dev_err(&udev->dev,
1239 "ath9k_htc: Could not find bulk out endpoint\n");
1240 return -ENODEV;
1241 }
1242
1243 cmd = kzalloc(31, GFP_KERNEL);
1244 if (cmd == NULL)
1245 return -ENODEV;
1246
1247 /* USB bulk command block */
1248 cmd[0] = 0x55; /* bulk command signature */
1249 cmd[1] = 0x53; /* bulk command signature */
1250 cmd[2] = 0x42; /* bulk command signature */
1251 cmd[3] = 0x43; /* bulk command signature */
1252 cmd[14] = 6; /* command length */
1253
1254 cmd[15] = 0x1b; /* SCSI command: START STOP UNIT */
1255 cmd[19] = 0x2; /* eject disc */
1256
1257 dev_info(&udev->dev, "Ejecting storage device...\n");
1258 r = usb_bulk_msg(udev, usb_sndbulkpipe(udev, bulk_out_ep),
1259 cmd, 31, NULL, 2 * USB_MSG_TIMEOUT);
1260 kfree(cmd);
1261 if (r)
1262 return r;
1263
1264 /* At this point, the device disconnects and reconnects with the real
1265 * ID numbers. */
1266
1267 usb_set_intfdata(interface, NULL);
1268 return 0;
1269 }
1270
ath9k_hif_usb_probe(struct usb_interface * interface,const struct usb_device_id * id)1271 static int ath9k_hif_usb_probe(struct usb_interface *interface,
1272 const struct usb_device_id *id)
1273 {
1274 struct usb_device *udev = interface_to_usbdev(interface);
1275 struct hif_device_usb *hif_dev;
1276 int ret = 0;
1277
1278 if (id->driver_info == STORAGE_DEVICE)
1279 return send_eject_command(interface);
1280
1281 hif_dev = kzalloc(sizeof(struct hif_device_usb), GFP_KERNEL);
1282 if (!hif_dev) {
1283 ret = -ENOMEM;
1284 goto err_alloc;
1285 }
1286
1287 usb_get_dev(udev);
1288
1289 hif_dev->udev = udev;
1290 hif_dev->interface = interface;
1291 hif_dev->usb_device_id = id;
1292 #ifdef CONFIG_PM
1293 udev->reset_resume = 1;
1294 #endif
1295 usb_set_intfdata(interface, hif_dev);
1296
1297 init_completion(&hif_dev->fw_done);
1298
1299 ret = ath9k_hif_request_firmware(hif_dev, true);
1300 if (ret)
1301 goto err_fw_req;
1302
1303 return ret;
1304
1305 err_fw_req:
1306 usb_set_intfdata(interface, NULL);
1307 kfree(hif_dev);
1308 usb_put_dev(udev);
1309 err_alloc:
1310 return ret;
1311 }
1312
ath9k_hif_usb_reboot(struct usb_device * udev)1313 static void ath9k_hif_usb_reboot(struct usb_device *udev)
1314 {
1315 u32 reboot_cmd = 0xffffffff;
1316 void *buf;
1317 int ret;
1318
1319 buf = kmemdup(&reboot_cmd, 4, GFP_KERNEL);
1320 if (!buf)
1321 return;
1322
1323 ret = usb_interrupt_msg(udev, usb_sndintpipe(udev, USB_REG_OUT_PIPE),
1324 buf, 4, NULL, USB_MSG_TIMEOUT);
1325 if (ret)
1326 dev_err(&udev->dev, "ath9k_htc: USB reboot failed\n");
1327
1328 kfree(buf);
1329 }
1330
ath9k_hif_usb_disconnect(struct usb_interface * interface)1331 static void ath9k_hif_usb_disconnect(struct usb_interface *interface)
1332 {
1333 struct usb_device *udev = interface_to_usbdev(interface);
1334 struct hif_device_usb *hif_dev = usb_get_intfdata(interface);
1335 bool unplugged = (udev->state == USB_STATE_NOTATTACHED) ? true : false;
1336
1337 if (!hif_dev)
1338 return;
1339
1340 wait_for_completion(&hif_dev->fw_done);
1341
1342 if (hif_dev->flags & HIF_USB_READY) {
1343 ath9k_htc_hw_deinit(hif_dev->htc_handle, unplugged);
1344 ath9k_htc_hw_free(hif_dev->htc_handle);
1345 ath9k_hif_usb_dev_deinit(hif_dev);
1346 }
1347
1348 usb_set_intfdata(interface, NULL);
1349
1350 /* If firmware was loaded we should drop it
1351 * go back to first stage bootloader. */
1352 if (!unplugged && (hif_dev->flags & HIF_USB_READY))
1353 ath9k_hif_usb_reboot(udev);
1354
1355 kfree(hif_dev);
1356 dev_info(&udev->dev, "ath9k_htc: USB layer deinitialized\n");
1357 usb_put_dev(udev);
1358 }
1359
1360 #ifdef CONFIG_PM
ath9k_hif_usb_suspend(struct usb_interface * interface,pm_message_t message)1361 static int ath9k_hif_usb_suspend(struct usb_interface *interface,
1362 pm_message_t message)
1363 {
1364 struct hif_device_usb *hif_dev = usb_get_intfdata(interface);
1365
1366 /*
1367 * The device has to be set to FULLSLEEP mode in case no
1368 * interface is up.
1369 */
1370 if (!(hif_dev->flags & HIF_USB_START))
1371 ath9k_htc_suspend(hif_dev->htc_handle);
1372
1373 wait_for_completion(&hif_dev->fw_done);
1374
1375 if (hif_dev->flags & HIF_USB_READY)
1376 ath9k_hif_usb_dealloc_urbs(hif_dev);
1377
1378 return 0;
1379 }
1380
ath9k_hif_usb_resume(struct usb_interface * interface)1381 static int ath9k_hif_usb_resume(struct usb_interface *interface)
1382 {
1383 struct hif_device_usb *hif_dev = usb_get_intfdata(interface);
1384 struct htc_target *htc_handle = hif_dev->htc_handle;
1385 int ret;
1386 const struct firmware *fw;
1387
1388 ret = ath9k_hif_usb_alloc_urbs(hif_dev);
1389 if (ret)
1390 return ret;
1391
1392 if (hif_dev->flags & HIF_USB_READY) {
1393 /* request cached firmware during suspend/resume cycle */
1394 ret = request_firmware(&fw, hif_dev->fw_name,
1395 &hif_dev->udev->dev);
1396 if (ret)
1397 goto fail_resume;
1398
1399 hif_dev->fw_data = fw->data;
1400 hif_dev->fw_size = fw->size;
1401 ret = ath9k_hif_usb_download_fw(hif_dev);
1402 release_firmware(fw);
1403 if (ret)
1404 goto fail_resume;
1405 } else {
1406 ath9k_hif_usb_dealloc_urbs(hif_dev);
1407 return -EIO;
1408 }
1409
1410 mdelay(100);
1411
1412 ret = ath9k_htc_resume(htc_handle);
1413
1414 if (ret)
1415 goto fail_resume;
1416
1417 return 0;
1418
1419 fail_resume:
1420 ath9k_hif_usb_dealloc_urbs(hif_dev);
1421
1422 return ret;
1423 }
1424 #endif
1425
1426 static struct usb_driver ath9k_hif_usb_driver = {
1427 .name = KBUILD_MODNAME,
1428 .probe = ath9k_hif_usb_probe,
1429 .disconnect = ath9k_hif_usb_disconnect,
1430 #ifdef CONFIG_PM
1431 .suspend = ath9k_hif_usb_suspend,
1432 .resume = ath9k_hif_usb_resume,
1433 .reset_resume = ath9k_hif_usb_resume,
1434 #endif
1435 .id_table = ath9k_hif_usb_ids,
1436 .soft_unbind = 1,
1437 .disable_hub_initiated_lpm = 1,
1438 };
1439
ath9k_hif_usb_init(void)1440 int ath9k_hif_usb_init(void)
1441 {
1442 return usb_register(&ath9k_hif_usb_driver);
1443 }
1444
ath9k_hif_usb_exit(void)1445 void ath9k_hif_usb_exit(void)
1446 {
1447 usb_deregister(&ath9k_hif_usb_driver);
1448 }
1449