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 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags);
453 list_for_each_entry_safe(tx_buf, tx_buf_tmp,
454 &hif_dev->tx.tx_pending, list) {
455 usb_get_urb(tx_buf->urb);
456 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
457 usb_kill_urb(tx_buf->urb);
458 list_del(&tx_buf->list);
459 usb_free_urb(tx_buf->urb);
460 kfree(tx_buf->buf);
461 kfree(tx_buf);
462 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags);
463 }
464 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
465
466 usb_kill_anchored_urbs(&hif_dev->mgmt_submitted);
467 }
468
hif_usb_send(void * hif_handle,u8 pipe_id,struct sk_buff * skb)469 static int hif_usb_send(void *hif_handle, u8 pipe_id, struct sk_buff *skb)
470 {
471 struct hif_device_usb *hif_dev = hif_handle;
472 int ret = 0;
473
474 switch (pipe_id) {
475 case USB_WLAN_TX_PIPE:
476 ret = hif_usb_send_tx(hif_dev, skb);
477 break;
478 case USB_REG_OUT_PIPE:
479 ret = hif_usb_send_regout(hif_dev, skb);
480 break;
481 default:
482 dev_err(&hif_dev->udev->dev,
483 "ath9k_htc: Invalid TX pipe: %d\n", pipe_id);
484 ret = -EINVAL;
485 break;
486 }
487
488 return ret;
489 }
490
check_index(struct sk_buff * skb,u8 idx)491 static inline bool check_index(struct sk_buff *skb, u8 idx)
492 {
493 struct ath9k_htc_tx_ctl *tx_ctl;
494
495 tx_ctl = HTC_SKB_CB(skb);
496
497 if ((tx_ctl->type == ATH9K_HTC_AMPDU) &&
498 (tx_ctl->sta_idx == idx))
499 return true;
500
501 return false;
502 }
503
hif_usb_sta_drain(void * hif_handle,u8 idx)504 static void hif_usb_sta_drain(void *hif_handle, u8 idx)
505 {
506 struct hif_device_usb *hif_dev = hif_handle;
507 struct sk_buff *skb, *tmp;
508 unsigned long flags;
509
510 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags);
511
512 skb_queue_walk_safe(&hif_dev->tx.tx_skb_queue, skb, tmp) {
513 if (check_index(skb, idx)) {
514 __skb_unlink(skb, &hif_dev->tx.tx_skb_queue);
515 ath9k_htc_txcompletion_cb(hif_dev->htc_handle,
516 skb, false);
517 hif_dev->tx.tx_skb_cnt--;
518 TX_STAT_INC(skb_failed);
519 }
520 }
521
522 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
523 }
524
525 static struct ath9k_htc_hif hif_usb = {
526 .transport = ATH9K_HIF_USB,
527 .name = "ath9k_hif_usb",
528
529 .control_ul_pipe = USB_REG_OUT_PIPE,
530 .control_dl_pipe = USB_REG_IN_PIPE,
531
532 .start = hif_usb_start,
533 .stop = hif_usb_stop,
534 .sta_drain = hif_usb_sta_drain,
535 .send = hif_usb_send,
536 };
537
ath9k_hif_usb_rx_stream(struct hif_device_usb * hif_dev,struct sk_buff * skb)538 static void ath9k_hif_usb_rx_stream(struct hif_device_usb *hif_dev,
539 struct sk_buff *skb)
540 {
541 struct sk_buff *nskb, *skb_pool[MAX_PKT_NUM_IN_TRANSFER];
542 int index = 0, i, len = skb->len;
543 int rx_remain_len, rx_pkt_len;
544 u16 pool_index = 0;
545 u8 *ptr;
546
547 spin_lock(&hif_dev->rx_lock);
548
549 rx_remain_len = hif_dev->rx_remain_len;
550 rx_pkt_len = hif_dev->rx_transfer_len;
551
552 if (rx_remain_len != 0) {
553 struct sk_buff *remain_skb = hif_dev->remain_skb;
554
555 if (remain_skb) {
556 ptr = (u8 *) remain_skb->data;
557
558 index = rx_remain_len;
559 rx_remain_len -= hif_dev->rx_pad_len;
560 ptr += rx_pkt_len;
561
562 memcpy(ptr, skb->data, rx_remain_len);
563
564 rx_pkt_len += rx_remain_len;
565 hif_dev->rx_remain_len = 0;
566 skb_put(remain_skb, rx_pkt_len);
567
568 skb_pool[pool_index++] = remain_skb;
569
570 } else {
571 index = rx_remain_len;
572 }
573 }
574
575 spin_unlock(&hif_dev->rx_lock);
576
577 while (index < len) {
578 u16 pkt_len;
579 u16 pkt_tag;
580 u16 pad_len;
581 int chk_idx;
582
583 ptr = (u8 *) skb->data;
584
585 pkt_len = get_unaligned_le16(ptr + index);
586 pkt_tag = get_unaligned_le16(ptr + index + 2);
587
588 if (pkt_tag != ATH_USB_RX_STREAM_MODE_TAG) {
589 RX_STAT_INC(skb_dropped);
590 return;
591 }
592
593 if (pkt_len > 2 * MAX_RX_BUF_SIZE) {
594 dev_err(&hif_dev->udev->dev,
595 "ath9k_htc: invalid pkt_len (%x)\n", pkt_len);
596 RX_STAT_INC(skb_dropped);
597 return;
598 }
599
600 pad_len = 4 - (pkt_len & 0x3);
601 if (pad_len == 4)
602 pad_len = 0;
603
604 chk_idx = index;
605 index = index + 4 + pkt_len + pad_len;
606
607 if (index > MAX_RX_BUF_SIZE) {
608 spin_lock(&hif_dev->rx_lock);
609 hif_dev->rx_remain_len = index - MAX_RX_BUF_SIZE;
610 hif_dev->rx_transfer_len =
611 MAX_RX_BUF_SIZE - chk_idx - 4;
612 hif_dev->rx_pad_len = pad_len;
613
614 nskb = __dev_alloc_skb(pkt_len + 32, GFP_ATOMIC);
615 if (!nskb) {
616 dev_err(&hif_dev->udev->dev,
617 "ath9k_htc: RX memory allocation error\n");
618 spin_unlock(&hif_dev->rx_lock);
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]),
625 hif_dev->rx_transfer_len);
626
627 /* Record the buffer pointer */
628 hif_dev->remain_skb = nskb;
629 spin_unlock(&hif_dev->rx_lock);
630 } else {
631 if (pool_index == MAX_PKT_NUM_IN_TRANSFER) {
632 dev_err(&hif_dev->udev->dev,
633 "ath9k_htc: over RX MAX_PKT_NUM\n");
634 goto err;
635 }
636 nskb = __dev_alloc_skb(pkt_len + 32, GFP_ATOMIC);
637 if (!nskb) {
638 dev_err(&hif_dev->udev->dev,
639 "ath9k_htc: RX memory allocation error\n");
640 goto err;
641 }
642 skb_reserve(nskb, 32);
643 RX_STAT_INC(skb_allocated);
644
645 memcpy(nskb->data, &(skb->data[chk_idx+4]), pkt_len);
646 skb_put(nskb, pkt_len);
647 skb_pool[pool_index++] = nskb;
648 }
649 }
650
651 err:
652 for (i = 0; i < pool_index; i++) {
653 RX_STAT_ADD(skb_completed_bytes, skb_pool[i]->len);
654 ath9k_htc_rx_msg(hif_dev->htc_handle, skb_pool[i],
655 skb_pool[i]->len, USB_WLAN_RX_PIPE);
656 RX_STAT_INC(skb_completed);
657 }
658 }
659
ath9k_hif_usb_rx_cb(struct urb * urb)660 static void ath9k_hif_usb_rx_cb(struct urb *urb)
661 {
662 struct rx_buf *rx_buf = (struct rx_buf *)urb->context;
663 struct hif_device_usb *hif_dev = rx_buf->hif_dev;
664 struct sk_buff *skb = rx_buf->skb;
665 int ret;
666
667 if (!skb)
668 return;
669
670 if (!hif_dev)
671 goto free;
672
673 switch (urb->status) {
674 case 0:
675 break;
676 case -ENOENT:
677 case -ECONNRESET:
678 case -ENODEV:
679 case -ESHUTDOWN:
680 goto free;
681 default:
682 goto resubmit;
683 }
684
685 if (likely(urb->actual_length != 0)) {
686 skb_put(skb, urb->actual_length);
687 ath9k_hif_usb_rx_stream(hif_dev, skb);
688 }
689
690 resubmit:
691 skb_reset_tail_pointer(skb);
692 skb_trim(skb, 0);
693
694 usb_anchor_urb(urb, &hif_dev->rx_submitted);
695 ret = usb_submit_urb(urb, GFP_ATOMIC);
696 if (ret) {
697 usb_unanchor_urb(urb);
698 goto free;
699 }
700
701 return;
702 free:
703 kfree_skb(skb);
704 kfree(rx_buf);
705 }
706
ath9k_hif_usb_reg_in_cb(struct urb * urb)707 static void ath9k_hif_usb_reg_in_cb(struct urb *urb)
708 {
709 struct rx_buf *rx_buf = (struct rx_buf *)urb->context;
710 struct hif_device_usb *hif_dev = rx_buf->hif_dev;
711 struct sk_buff *skb = rx_buf->skb;
712 int ret;
713
714 if (!skb)
715 return;
716
717 if (!hif_dev)
718 goto free_skb;
719
720 switch (urb->status) {
721 case 0:
722 break;
723 case -ENOENT:
724 case -ECONNRESET:
725 case -ENODEV:
726 case -ESHUTDOWN:
727 goto free_skb;
728 default:
729 skb_reset_tail_pointer(skb);
730 skb_trim(skb, 0);
731
732 goto resubmit;
733 }
734
735 if (likely(urb->actual_length != 0)) {
736 skb_put(skb, urb->actual_length);
737
738 /*
739 * Process the command first.
740 * skb is either freed here or passed to be
741 * managed to another callback function.
742 */
743 ath9k_htc_rx_msg(hif_dev->htc_handle, skb,
744 skb->len, USB_REG_IN_PIPE);
745
746 skb = alloc_skb(MAX_REG_IN_BUF_SIZE, GFP_ATOMIC);
747 if (!skb) {
748 dev_err(&hif_dev->udev->dev,
749 "ath9k_htc: REG_IN memory allocation failure\n");
750 goto free_rx_buf;
751 }
752
753 rx_buf->skb = skb;
754
755 usb_fill_int_urb(urb, hif_dev->udev,
756 usb_rcvintpipe(hif_dev->udev,
757 USB_REG_IN_PIPE),
758 skb->data, MAX_REG_IN_BUF_SIZE,
759 ath9k_hif_usb_reg_in_cb, rx_buf, 1);
760 }
761
762 resubmit:
763 usb_anchor_urb(urb, &hif_dev->reg_in_submitted);
764 ret = usb_submit_urb(urb, GFP_ATOMIC);
765 if (ret) {
766 usb_unanchor_urb(urb);
767 goto free_skb;
768 }
769
770 return;
771 free_skb:
772 kfree_skb(skb);
773 free_rx_buf:
774 kfree(rx_buf);
775 urb->context = NULL;
776 }
777
ath9k_hif_usb_dealloc_tx_urbs(struct hif_device_usb * hif_dev)778 static void ath9k_hif_usb_dealloc_tx_urbs(struct hif_device_usb *hif_dev)
779 {
780 struct tx_buf *tx_buf = NULL, *tx_buf_tmp = NULL;
781 unsigned long flags;
782
783 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags);
784 list_for_each_entry_safe(tx_buf, tx_buf_tmp,
785 &hif_dev->tx.tx_buf, list) {
786 list_del(&tx_buf->list);
787 usb_free_urb(tx_buf->urb);
788 kfree(tx_buf->buf);
789 kfree(tx_buf);
790 }
791 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
792
793 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags);
794 hif_dev->tx.flags |= HIF_USB_TX_FLUSH;
795 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
796
797 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags);
798 list_for_each_entry_safe(tx_buf, tx_buf_tmp,
799 &hif_dev->tx.tx_pending, list) {
800 usb_get_urb(tx_buf->urb);
801 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
802 usb_kill_urb(tx_buf->urb);
803 list_del(&tx_buf->list);
804 usb_free_urb(tx_buf->urb);
805 kfree(tx_buf->buf);
806 kfree(tx_buf);
807 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags);
808 }
809 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags);
810
811 usb_kill_anchored_urbs(&hif_dev->mgmt_submitted);
812 }
813
ath9k_hif_usb_alloc_tx_urbs(struct hif_device_usb * hif_dev)814 static int ath9k_hif_usb_alloc_tx_urbs(struct hif_device_usb *hif_dev)
815 {
816 struct tx_buf *tx_buf;
817 int i;
818
819 INIT_LIST_HEAD(&hif_dev->tx.tx_buf);
820 INIT_LIST_HEAD(&hif_dev->tx.tx_pending);
821 spin_lock_init(&hif_dev->tx.tx_lock);
822 __skb_queue_head_init(&hif_dev->tx.tx_skb_queue);
823 init_usb_anchor(&hif_dev->mgmt_submitted);
824
825 for (i = 0; i < MAX_TX_URB_NUM; i++) {
826 tx_buf = kzalloc(sizeof(*tx_buf), GFP_KERNEL);
827 if (!tx_buf)
828 goto err;
829
830 tx_buf->buf = kzalloc(MAX_TX_BUF_SIZE, GFP_KERNEL);
831 if (!tx_buf->buf)
832 goto err;
833
834 tx_buf->urb = usb_alloc_urb(0, GFP_KERNEL);
835 if (!tx_buf->urb)
836 goto err;
837
838 tx_buf->hif_dev = hif_dev;
839 __skb_queue_head_init(&tx_buf->skb_queue);
840
841 list_add_tail(&tx_buf->list, &hif_dev->tx.tx_buf);
842 }
843
844 hif_dev->tx.tx_buf_cnt = MAX_TX_URB_NUM;
845
846 return 0;
847 err:
848 if (tx_buf) {
849 kfree(tx_buf->buf);
850 kfree(tx_buf);
851 }
852 ath9k_hif_usb_dealloc_tx_urbs(hif_dev);
853 return -ENOMEM;
854 }
855
ath9k_hif_usb_dealloc_rx_urbs(struct hif_device_usb * hif_dev)856 static void ath9k_hif_usb_dealloc_rx_urbs(struct hif_device_usb *hif_dev)
857 {
858 usb_kill_anchored_urbs(&hif_dev->rx_submitted);
859 }
860
ath9k_hif_usb_alloc_rx_urbs(struct hif_device_usb * hif_dev)861 static int ath9k_hif_usb_alloc_rx_urbs(struct hif_device_usb *hif_dev)
862 {
863 struct rx_buf *rx_buf = NULL;
864 struct sk_buff *skb = NULL;
865 struct urb *urb = NULL;
866 int i, ret;
867
868 init_usb_anchor(&hif_dev->rx_submitted);
869 spin_lock_init(&hif_dev->rx_lock);
870
871 for (i = 0; i < MAX_RX_URB_NUM; i++) {
872
873 rx_buf = kzalloc(sizeof(*rx_buf), GFP_KERNEL);
874 if (!rx_buf) {
875 ret = -ENOMEM;
876 goto err_rxb;
877 }
878
879 /* Allocate URB */
880 urb = usb_alloc_urb(0, GFP_KERNEL);
881 if (urb == NULL) {
882 ret = -ENOMEM;
883 goto err_urb;
884 }
885
886 /* Allocate buffer */
887 skb = alloc_skb(MAX_RX_BUF_SIZE, GFP_KERNEL);
888 if (!skb) {
889 ret = -ENOMEM;
890 goto err_skb;
891 }
892
893 rx_buf->hif_dev = hif_dev;
894 rx_buf->skb = skb;
895
896 usb_fill_bulk_urb(urb, hif_dev->udev,
897 usb_rcvbulkpipe(hif_dev->udev,
898 USB_WLAN_RX_PIPE),
899 skb->data, MAX_RX_BUF_SIZE,
900 ath9k_hif_usb_rx_cb, rx_buf);
901
902 /* Anchor URB */
903 usb_anchor_urb(urb, &hif_dev->rx_submitted);
904
905 /* Submit URB */
906 ret = usb_submit_urb(urb, GFP_KERNEL);
907 if (ret) {
908 usb_unanchor_urb(urb);
909 goto err_submit;
910 }
911
912 /*
913 * Drop reference count.
914 * This ensures that the URB is freed when killing them.
915 */
916 usb_free_urb(urb);
917 }
918
919 return 0;
920
921 err_submit:
922 kfree_skb(skb);
923 err_skb:
924 usb_free_urb(urb);
925 err_urb:
926 kfree(rx_buf);
927 err_rxb:
928 ath9k_hif_usb_dealloc_rx_urbs(hif_dev);
929 return ret;
930 }
931
ath9k_hif_usb_dealloc_reg_in_urbs(struct hif_device_usb * hif_dev)932 static void ath9k_hif_usb_dealloc_reg_in_urbs(struct hif_device_usb *hif_dev)
933 {
934 usb_kill_anchored_urbs(&hif_dev->reg_in_submitted);
935 }
936
ath9k_hif_usb_alloc_reg_in_urbs(struct hif_device_usb * hif_dev)937 static int ath9k_hif_usb_alloc_reg_in_urbs(struct hif_device_usb *hif_dev)
938 {
939 struct rx_buf *rx_buf = NULL;
940 struct sk_buff *skb = NULL;
941 struct urb *urb = NULL;
942 int i, ret;
943
944 init_usb_anchor(&hif_dev->reg_in_submitted);
945
946 for (i = 0; i < MAX_REG_IN_URB_NUM; i++) {
947
948 rx_buf = kzalloc(sizeof(*rx_buf), GFP_KERNEL);
949 if (!rx_buf) {
950 ret = -ENOMEM;
951 goto err_rxb;
952 }
953
954 /* Allocate URB */
955 urb = usb_alloc_urb(0, GFP_KERNEL);
956 if (urb == NULL) {
957 ret = -ENOMEM;
958 goto err_urb;
959 }
960
961 /* Allocate buffer */
962 skb = alloc_skb(MAX_REG_IN_BUF_SIZE, GFP_KERNEL);
963 if (!skb) {
964 ret = -ENOMEM;
965 goto err_skb;
966 }
967
968 rx_buf->hif_dev = hif_dev;
969 rx_buf->skb = skb;
970
971 usb_fill_int_urb(urb, hif_dev->udev,
972 usb_rcvintpipe(hif_dev->udev,
973 USB_REG_IN_PIPE),
974 skb->data, MAX_REG_IN_BUF_SIZE,
975 ath9k_hif_usb_reg_in_cb, rx_buf, 1);
976
977 /* Anchor URB */
978 usb_anchor_urb(urb, &hif_dev->reg_in_submitted);
979
980 /* Submit URB */
981 ret = usb_submit_urb(urb, GFP_KERNEL);
982 if (ret) {
983 usb_unanchor_urb(urb);
984 goto err_submit;
985 }
986
987 /*
988 * Drop reference count.
989 * This ensures that the URB is freed when killing them.
990 */
991 usb_free_urb(urb);
992 }
993
994 return 0;
995
996 err_submit:
997 kfree_skb(skb);
998 err_skb:
999 usb_free_urb(urb);
1000 err_urb:
1001 kfree(rx_buf);
1002 err_rxb:
1003 ath9k_hif_usb_dealloc_reg_in_urbs(hif_dev);
1004 return ret;
1005 }
1006
ath9k_hif_usb_alloc_urbs(struct hif_device_usb * hif_dev)1007 static int ath9k_hif_usb_alloc_urbs(struct hif_device_usb *hif_dev)
1008 {
1009 /* Register Write */
1010 init_usb_anchor(&hif_dev->regout_submitted);
1011
1012 /* TX */
1013 if (ath9k_hif_usb_alloc_tx_urbs(hif_dev) < 0)
1014 goto err;
1015
1016 /* RX */
1017 if (ath9k_hif_usb_alloc_rx_urbs(hif_dev) < 0)
1018 goto err_rx;
1019
1020 /* Register Read */
1021 if (ath9k_hif_usb_alloc_reg_in_urbs(hif_dev) < 0)
1022 goto err_reg;
1023
1024 return 0;
1025 err_reg:
1026 ath9k_hif_usb_dealloc_rx_urbs(hif_dev);
1027 err_rx:
1028 ath9k_hif_usb_dealloc_tx_urbs(hif_dev);
1029 err:
1030 return -ENOMEM;
1031 }
1032
ath9k_hif_usb_dealloc_urbs(struct hif_device_usb * hif_dev)1033 void ath9k_hif_usb_dealloc_urbs(struct hif_device_usb *hif_dev)
1034 {
1035 usb_kill_anchored_urbs(&hif_dev->regout_submitted);
1036 ath9k_hif_usb_dealloc_reg_in_urbs(hif_dev);
1037 ath9k_hif_usb_dealloc_tx_urbs(hif_dev);
1038 ath9k_hif_usb_dealloc_rx_urbs(hif_dev);
1039 }
1040
ath9k_hif_usb_download_fw(struct hif_device_usb * hif_dev)1041 static int ath9k_hif_usb_download_fw(struct hif_device_usb *hif_dev)
1042 {
1043 int transfer, err;
1044 const void *data = hif_dev->fw_data;
1045 size_t len = hif_dev->fw_size;
1046 u32 addr = AR9271_FIRMWARE;
1047 u8 *buf = kzalloc(4096, GFP_KERNEL);
1048 u32 firm_offset;
1049
1050 if (!buf)
1051 return -ENOMEM;
1052
1053 while (len) {
1054 transfer = min_t(size_t, len, 4096);
1055 memcpy(buf, data, transfer);
1056
1057 err = usb_control_msg(hif_dev->udev,
1058 usb_sndctrlpipe(hif_dev->udev, 0),
1059 FIRMWARE_DOWNLOAD, 0x40 | USB_DIR_OUT,
1060 addr >> 8, 0, buf, transfer,
1061 USB_MSG_TIMEOUT);
1062 if (err < 0) {
1063 kfree(buf);
1064 return err;
1065 }
1066
1067 len -= transfer;
1068 data += transfer;
1069 addr += transfer;
1070 }
1071 kfree(buf);
1072
1073 if (IS_AR7010_DEVICE(hif_dev->usb_device_id->driver_info))
1074 firm_offset = AR7010_FIRMWARE_TEXT;
1075 else
1076 firm_offset = AR9271_FIRMWARE_TEXT;
1077
1078 /*
1079 * Issue FW download complete command to firmware.
1080 */
1081 err = usb_control_msg(hif_dev->udev, usb_sndctrlpipe(hif_dev->udev, 0),
1082 FIRMWARE_DOWNLOAD_COMP,
1083 0x40 | USB_DIR_OUT,
1084 firm_offset >> 8, 0, NULL, 0, USB_MSG_TIMEOUT);
1085 if (err)
1086 return -EIO;
1087
1088 dev_info(&hif_dev->udev->dev, "ath9k_htc: Transferred FW: %s, size: %ld\n",
1089 hif_dev->fw_name, (unsigned long) hif_dev->fw_size);
1090
1091 return 0;
1092 }
1093
ath9k_hif_usb_dev_init(struct hif_device_usb * hif_dev)1094 static int ath9k_hif_usb_dev_init(struct hif_device_usb *hif_dev)
1095 {
1096 int ret;
1097
1098 ret = ath9k_hif_usb_download_fw(hif_dev);
1099 if (ret) {
1100 dev_err(&hif_dev->udev->dev,
1101 "ath9k_htc: Firmware - %s download failed\n",
1102 hif_dev->fw_name);
1103 return ret;
1104 }
1105
1106 /* Alloc URBs */
1107 ret = ath9k_hif_usb_alloc_urbs(hif_dev);
1108 if (ret) {
1109 dev_err(&hif_dev->udev->dev,
1110 "ath9k_htc: Unable to allocate URBs\n");
1111 return ret;
1112 }
1113
1114 return 0;
1115 }
1116
ath9k_hif_usb_dev_deinit(struct hif_device_usb * hif_dev)1117 static void ath9k_hif_usb_dev_deinit(struct hif_device_usb *hif_dev)
1118 {
1119 ath9k_hif_usb_dealloc_urbs(hif_dev);
1120 }
1121
1122 /*
1123 * If initialization fails or the FW cannot be retrieved,
1124 * detach the device.
1125 */
ath9k_hif_usb_firmware_fail(struct hif_device_usb * hif_dev)1126 static void ath9k_hif_usb_firmware_fail(struct hif_device_usb *hif_dev)
1127 {
1128 struct device *dev = &hif_dev->udev->dev;
1129 struct device *parent = dev->parent;
1130
1131 complete_all(&hif_dev->fw_done);
1132
1133 if (parent)
1134 device_lock(parent);
1135
1136 device_release_driver(dev);
1137
1138 if (parent)
1139 device_unlock(parent);
1140 }
1141
1142 static void ath9k_hif_usb_firmware_cb(const struct firmware *fw, void *context);
1143
1144 /* taken from iwlwifi */
ath9k_hif_request_firmware(struct hif_device_usb * hif_dev,bool first)1145 static int ath9k_hif_request_firmware(struct hif_device_usb *hif_dev,
1146 bool first)
1147 {
1148 char index[8], *chip;
1149 int ret;
1150
1151 if (first) {
1152 if (htc_use_dev_fw) {
1153 hif_dev->fw_minor_index = FIRMWARE_MINOR_IDX_MAX + 1;
1154 sprintf(index, "%s", "dev");
1155 } else {
1156 hif_dev->fw_minor_index = FIRMWARE_MINOR_IDX_MAX;
1157 sprintf(index, "%d", hif_dev->fw_minor_index);
1158 }
1159 } else {
1160 hif_dev->fw_minor_index--;
1161 sprintf(index, "%d", hif_dev->fw_minor_index);
1162 }
1163
1164 /* test for FW 1.3 */
1165 if (MAJOR_VERSION_REQ == 1 && hif_dev->fw_minor_index == 3) {
1166 const char *filename;
1167
1168 if (IS_AR7010_DEVICE(hif_dev->usb_device_id->driver_info))
1169 filename = FIRMWARE_AR7010_1_1;
1170 else
1171 filename = FIRMWARE_AR9271;
1172
1173 /* expected fw locations:
1174 * - htc_9271.fw (stable version 1.3, depricated)
1175 */
1176 snprintf(hif_dev->fw_name, sizeof(hif_dev->fw_name),
1177 "%s", filename);
1178
1179 } else if (hif_dev->fw_minor_index < FIRMWARE_MINOR_IDX_MIN) {
1180 dev_err(&hif_dev->udev->dev, "no suitable firmware found!\n");
1181
1182 return -ENOENT;
1183 } else {
1184 if (IS_AR7010_DEVICE(hif_dev->usb_device_id->driver_info))
1185 chip = "7010";
1186 else
1187 chip = "9271";
1188
1189 /* expected fw locations:
1190 * - ath9k_htc/htc_9271-1.dev.0.fw (development version)
1191 * - ath9k_htc/htc_9271-1.4.0.fw (stable version)
1192 */
1193 snprintf(hif_dev->fw_name, sizeof(hif_dev->fw_name),
1194 "%s/htc_%s-%d.%s.0.fw", HTC_FW_PATH,
1195 chip, MAJOR_VERSION_REQ, index);
1196 }
1197
1198 ret = request_firmware_nowait(THIS_MODULE, true, hif_dev->fw_name,
1199 &hif_dev->udev->dev, GFP_KERNEL,
1200 hif_dev, ath9k_hif_usb_firmware_cb);
1201 if (ret) {
1202 dev_err(&hif_dev->udev->dev,
1203 "ath9k_htc: Async request for firmware %s failed\n",
1204 hif_dev->fw_name);
1205 return ret;
1206 }
1207
1208 dev_info(&hif_dev->udev->dev, "ath9k_htc: Firmware %s requested\n",
1209 hif_dev->fw_name);
1210
1211 return ret;
1212 }
1213
ath9k_hif_usb_firmware_cb(const struct firmware * fw,void * context)1214 static void ath9k_hif_usb_firmware_cb(const struct firmware *fw, void *context)
1215 {
1216 struct hif_device_usb *hif_dev = context;
1217 int ret;
1218
1219 if (!fw) {
1220 ret = ath9k_hif_request_firmware(hif_dev, false);
1221 if (!ret)
1222 return;
1223
1224 dev_err(&hif_dev->udev->dev,
1225 "ath9k_htc: Failed to get firmware %s\n",
1226 hif_dev->fw_name);
1227 goto err_fw;
1228 }
1229
1230 hif_dev->htc_handle = ath9k_htc_hw_alloc(hif_dev, &hif_usb,
1231 &hif_dev->udev->dev);
1232 if (hif_dev->htc_handle == NULL)
1233 goto err_dev_alloc;
1234
1235 hif_dev->fw_data = fw->data;
1236 hif_dev->fw_size = fw->size;
1237
1238 /* Proceed with initialization */
1239
1240 ret = ath9k_hif_usb_dev_init(hif_dev);
1241 if (ret)
1242 goto err_dev_init;
1243
1244 ret = ath9k_htc_hw_init(hif_dev->htc_handle,
1245 &hif_dev->interface->dev,
1246 hif_dev->usb_device_id->idProduct,
1247 hif_dev->udev->product,
1248 hif_dev->usb_device_id->driver_info);
1249 if (ret) {
1250 ret = -EINVAL;
1251 goto err_htc_hw_init;
1252 }
1253
1254 release_firmware(fw);
1255 hif_dev->flags |= HIF_USB_READY;
1256 complete_all(&hif_dev->fw_done);
1257
1258 return;
1259
1260 err_htc_hw_init:
1261 ath9k_hif_usb_dev_deinit(hif_dev);
1262 err_dev_init:
1263 ath9k_htc_hw_free(hif_dev->htc_handle);
1264 err_dev_alloc:
1265 release_firmware(fw);
1266 err_fw:
1267 ath9k_hif_usb_firmware_fail(hif_dev);
1268 }
1269
1270 /*
1271 * An exact copy of the function from zd1211rw.
1272 */
send_eject_command(struct usb_interface * interface)1273 static int send_eject_command(struct usb_interface *interface)
1274 {
1275 struct usb_device *udev = interface_to_usbdev(interface);
1276 struct usb_host_interface *iface_desc = interface->cur_altsetting;
1277 struct usb_endpoint_descriptor *endpoint;
1278 unsigned char *cmd;
1279 u8 bulk_out_ep;
1280 int r;
1281
1282 if (iface_desc->desc.bNumEndpoints < 2)
1283 return -ENODEV;
1284
1285 /* Find bulk out endpoint */
1286 for (r = 1; r >= 0; r--) {
1287 endpoint = &iface_desc->endpoint[r].desc;
1288 if (usb_endpoint_dir_out(endpoint) &&
1289 usb_endpoint_xfer_bulk(endpoint)) {
1290 bulk_out_ep = endpoint->bEndpointAddress;
1291 break;
1292 }
1293 }
1294 if (r == -1) {
1295 dev_err(&udev->dev,
1296 "ath9k_htc: Could not find bulk out endpoint\n");
1297 return -ENODEV;
1298 }
1299
1300 cmd = kzalloc(31, GFP_KERNEL);
1301 if (cmd == NULL)
1302 return -ENODEV;
1303
1304 /* USB bulk command block */
1305 cmd[0] = 0x55; /* bulk command signature */
1306 cmd[1] = 0x53; /* bulk command signature */
1307 cmd[2] = 0x42; /* bulk command signature */
1308 cmd[3] = 0x43; /* bulk command signature */
1309 cmd[14] = 6; /* command length */
1310
1311 cmd[15] = 0x1b; /* SCSI command: START STOP UNIT */
1312 cmd[19] = 0x2; /* eject disc */
1313
1314 dev_info(&udev->dev, "Ejecting storage device...\n");
1315 r = usb_bulk_msg(udev, usb_sndbulkpipe(udev, bulk_out_ep),
1316 cmd, 31, NULL, 2 * USB_MSG_TIMEOUT);
1317 kfree(cmd);
1318 if (r)
1319 return r;
1320
1321 /* At this point, the device disconnects and reconnects with the real
1322 * ID numbers. */
1323
1324 usb_set_intfdata(interface, NULL);
1325 return 0;
1326 }
1327
ath9k_hif_usb_probe(struct usb_interface * interface,const struct usb_device_id * id)1328 static int ath9k_hif_usb_probe(struct usb_interface *interface,
1329 const struct usb_device_id *id)
1330 {
1331 struct usb_endpoint_descriptor *bulk_in, *bulk_out, *int_in, *int_out;
1332 struct usb_device *udev = interface_to_usbdev(interface);
1333 struct usb_host_interface *alt;
1334 struct hif_device_usb *hif_dev;
1335 int ret = 0;
1336
1337 /* Verify the expected endpoints are present */
1338 alt = interface->cur_altsetting;
1339 if (usb_find_common_endpoints(alt, &bulk_in, &bulk_out, &int_in, &int_out) < 0 ||
1340 usb_endpoint_num(bulk_in) != USB_WLAN_RX_PIPE ||
1341 usb_endpoint_num(bulk_out) != USB_WLAN_TX_PIPE ||
1342 usb_endpoint_num(int_in) != USB_REG_IN_PIPE ||
1343 usb_endpoint_num(int_out) != USB_REG_OUT_PIPE) {
1344 dev_err(&udev->dev,
1345 "ath9k_htc: Device endpoint numbers are not the expected ones\n");
1346 return -ENODEV;
1347 }
1348
1349 if (id->driver_info == STORAGE_DEVICE)
1350 return send_eject_command(interface);
1351
1352 hif_dev = kzalloc(sizeof(struct hif_device_usb), GFP_KERNEL);
1353 if (!hif_dev) {
1354 ret = -ENOMEM;
1355 goto err_alloc;
1356 }
1357
1358 usb_get_dev(udev);
1359
1360 hif_dev->udev = udev;
1361 hif_dev->interface = interface;
1362 hif_dev->usb_device_id = id;
1363 #ifdef CONFIG_PM
1364 udev->reset_resume = 1;
1365 #endif
1366 usb_set_intfdata(interface, hif_dev);
1367
1368 init_completion(&hif_dev->fw_done);
1369
1370 ret = ath9k_hif_request_firmware(hif_dev, true);
1371 if (ret)
1372 goto err_fw_req;
1373
1374 return ret;
1375
1376 err_fw_req:
1377 usb_set_intfdata(interface, NULL);
1378 kfree(hif_dev);
1379 usb_put_dev(udev);
1380 err_alloc:
1381 return ret;
1382 }
1383
ath9k_hif_usb_reboot(struct usb_device * udev)1384 static void ath9k_hif_usb_reboot(struct usb_device *udev)
1385 {
1386 u32 reboot_cmd = 0xffffffff;
1387 void *buf;
1388 int ret;
1389
1390 buf = kmemdup(&reboot_cmd, 4, GFP_KERNEL);
1391 if (!buf)
1392 return;
1393
1394 ret = usb_interrupt_msg(udev, usb_sndintpipe(udev, USB_REG_OUT_PIPE),
1395 buf, 4, NULL, USB_MSG_TIMEOUT);
1396 if (ret)
1397 dev_err(&udev->dev, "ath9k_htc: USB reboot failed\n");
1398
1399 kfree(buf);
1400 }
1401
ath9k_hif_usb_disconnect(struct usb_interface * interface)1402 static void ath9k_hif_usb_disconnect(struct usb_interface *interface)
1403 {
1404 struct usb_device *udev = interface_to_usbdev(interface);
1405 struct hif_device_usb *hif_dev = usb_get_intfdata(interface);
1406 bool unplugged = (udev->state == USB_STATE_NOTATTACHED) ? true : false;
1407
1408 if (!hif_dev)
1409 return;
1410
1411 wait_for_completion(&hif_dev->fw_done);
1412
1413 if (hif_dev->flags & HIF_USB_READY) {
1414 ath9k_htc_hw_deinit(hif_dev->htc_handle, unplugged);
1415 ath9k_hif_usb_dev_deinit(hif_dev);
1416 ath9k_destroy_wmi(hif_dev->htc_handle->drv_priv);
1417 ath9k_htc_hw_free(hif_dev->htc_handle);
1418 }
1419
1420 usb_set_intfdata(interface, NULL);
1421
1422 /* If firmware was loaded we should drop it
1423 * go back to first stage bootloader. */
1424 if (!unplugged && (hif_dev->flags & HIF_USB_READY))
1425 ath9k_hif_usb_reboot(udev);
1426
1427 kfree(hif_dev);
1428 dev_info(&udev->dev, "ath9k_htc: USB layer deinitialized\n");
1429 usb_put_dev(udev);
1430 }
1431
1432 #ifdef CONFIG_PM
ath9k_hif_usb_suspend(struct usb_interface * interface,pm_message_t message)1433 static int ath9k_hif_usb_suspend(struct usb_interface *interface,
1434 pm_message_t message)
1435 {
1436 struct hif_device_usb *hif_dev = usb_get_intfdata(interface);
1437
1438 /*
1439 * The device has to be set to FULLSLEEP mode in case no
1440 * interface is up.
1441 */
1442 if (!(hif_dev->flags & HIF_USB_START))
1443 ath9k_htc_suspend(hif_dev->htc_handle);
1444
1445 wait_for_completion(&hif_dev->fw_done);
1446
1447 if (hif_dev->flags & HIF_USB_READY)
1448 ath9k_hif_usb_dealloc_urbs(hif_dev);
1449
1450 return 0;
1451 }
1452
ath9k_hif_usb_resume(struct usb_interface * interface)1453 static int ath9k_hif_usb_resume(struct usb_interface *interface)
1454 {
1455 struct hif_device_usb *hif_dev = usb_get_intfdata(interface);
1456 struct htc_target *htc_handle = hif_dev->htc_handle;
1457 int ret;
1458 const struct firmware *fw;
1459
1460 ret = ath9k_hif_usb_alloc_urbs(hif_dev);
1461 if (ret)
1462 return ret;
1463
1464 if (hif_dev->flags & HIF_USB_READY) {
1465 /* request cached firmware during suspend/resume cycle */
1466 ret = request_firmware(&fw, hif_dev->fw_name,
1467 &hif_dev->udev->dev);
1468 if (ret)
1469 goto fail_resume;
1470
1471 hif_dev->fw_data = fw->data;
1472 hif_dev->fw_size = fw->size;
1473 ret = ath9k_hif_usb_download_fw(hif_dev);
1474 release_firmware(fw);
1475 if (ret)
1476 goto fail_resume;
1477 } else {
1478 ath9k_hif_usb_dealloc_urbs(hif_dev);
1479 return -EIO;
1480 }
1481
1482 mdelay(100);
1483
1484 ret = ath9k_htc_resume(htc_handle);
1485
1486 if (ret)
1487 goto fail_resume;
1488
1489 return 0;
1490
1491 fail_resume:
1492 ath9k_hif_usb_dealloc_urbs(hif_dev);
1493
1494 return ret;
1495 }
1496 #endif
1497
1498 static struct usb_driver ath9k_hif_usb_driver = {
1499 .name = KBUILD_MODNAME,
1500 .probe = ath9k_hif_usb_probe,
1501 .disconnect = ath9k_hif_usb_disconnect,
1502 #ifdef CONFIG_PM
1503 .suspend = ath9k_hif_usb_suspend,
1504 .resume = ath9k_hif_usb_resume,
1505 .reset_resume = ath9k_hif_usb_resume,
1506 #endif
1507 .id_table = ath9k_hif_usb_ids,
1508 .soft_unbind = 1,
1509 .disable_hub_initiated_lpm = 1,
1510 };
1511
ath9k_hif_usb_init(void)1512 int ath9k_hif_usb_init(void)
1513 {
1514 return usb_register(&ath9k_hif_usb_driver);
1515 }
1516
ath9k_hif_usb_exit(void)1517 void ath9k_hif_usb_exit(void)
1518 {
1519 usb_deregister(&ath9k_hif_usb_driver);
1520 }
1521