ar9170: Add support for D-Link DWA 160 A2
[linux-2.6/btrfs-unstable.git] / drivers / net / wireless / ath / ar9170 / usb.c
blobe0799d92405773206f9e2128b5a5cddeb6765d06
1 /*
2 * Atheros AR9170 driver
4 * USB - frontend
6 * Copyright 2008, Johannes Berg <johannes@sipsolutions.net>
7 * Copyright 2009, Christian Lamparter <chunkeey@web.de>
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
19 * You should have received a copy of the GNU General Public License
20 * along with this program; see the file COPYING. If not, see
21 * http://www.gnu.org/licenses/.
23 * This file incorporates work covered by the following copyright and
24 * permission notice:
25 * Copyright (c) 2007-2008 Atheros Communications, Inc.
27 * Permission to use, copy, modify, and/or distribute this software for any
28 * purpose with or without fee is hereby granted, provided that the above
29 * copyright notice and this permission notice appear in all copies.
31 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
32 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
33 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
34 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
35 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
36 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
37 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
40 #include <linux/module.h>
41 #include <linux/usb.h>
42 #include <linux/firmware.h>
43 #include <linux/etherdevice.h>
44 #include <net/mac80211.h>
45 #include "ar9170.h"
46 #include "cmd.h"
47 #include "hw.h"
48 #include "usb.h"
50 MODULE_AUTHOR("Johannes Berg <johannes@sipsolutions.net>");
51 MODULE_AUTHOR("Christian Lamparter <chunkeey@web.de>");
52 MODULE_LICENSE("GPL");
53 MODULE_DESCRIPTION("Atheros AR9170 802.11n USB wireless");
54 MODULE_FIRMWARE("ar9170.fw");
55 MODULE_FIRMWARE("ar9170-1.fw");
56 MODULE_FIRMWARE("ar9170-2.fw");
58 enum ar9170_requirements {
59 AR9170_REQ_FW1_ONLY = 1,
62 static struct usb_device_id ar9170_usb_ids[] = {
63 /* Atheros 9170 */
64 { USB_DEVICE(0x0cf3, 0x9170) },
65 /* Atheros TG121N */
66 { USB_DEVICE(0x0cf3, 0x1001) },
67 /* TP-Link TL-WN821N v2 */
68 { USB_DEVICE(0x0cf3, 0x1002) },
69 /* Cace Airpcap NX */
70 { USB_DEVICE(0xcace, 0x0300) },
71 /* D-Link DWA 160 A1 */
72 { USB_DEVICE(0x07d1, 0x3c10) },
73 /* D-Link DWA 160 A2 */
74 { USB_DEVICE(0x07d1, 0x3a09) },
75 /* Netgear WNDA3100 */
76 { USB_DEVICE(0x0846, 0x9010) },
77 /* Netgear WN111 v2 */
78 { USB_DEVICE(0x0846, 0x9001) },
79 /* Zydas ZD1221 */
80 { USB_DEVICE(0x0ace, 0x1221) },
81 /* ZyXEL NWD271N */
82 { USB_DEVICE(0x0586, 0x3417) },
83 /* Z-Com UB81 BG */
84 { USB_DEVICE(0x0cde, 0x0023) },
85 /* Z-Com UB82 ABG */
86 { USB_DEVICE(0x0cde, 0x0026) },
87 /* Arcadyan WN7512 */
88 { USB_DEVICE(0x083a, 0xf522) },
89 /* Planex GWUS300 */
90 { USB_DEVICE(0x2019, 0x5304) },
91 /* IO-Data WNGDNUS2 */
92 { USB_DEVICE(0x04bb, 0x093f) },
93 /* AVM FRITZ!WLAN USB Stick N */
94 { USB_DEVICE(0x057C, 0x8401) },
95 /* AVM FRITZ!WLAN USB Stick N 2.4 */
96 { USB_DEVICE(0x057C, 0x8402), .driver_info = AR9170_REQ_FW1_ONLY },
98 /* terminate */
101 MODULE_DEVICE_TABLE(usb, ar9170_usb_ids);
103 static void ar9170_usb_submit_urb(struct ar9170_usb *aru)
105 struct urb *urb;
106 unsigned long flags;
107 int err;
109 if (unlikely(!IS_STARTED(&aru->common)))
110 return ;
112 spin_lock_irqsave(&aru->tx_urb_lock, flags);
113 if (atomic_read(&aru->tx_submitted_urbs) >= AR9170_NUM_TX_URBS) {
114 spin_unlock_irqrestore(&aru->tx_urb_lock, flags);
115 return ;
117 atomic_inc(&aru->tx_submitted_urbs);
119 urb = usb_get_from_anchor(&aru->tx_pending);
120 if (!urb) {
121 atomic_dec(&aru->tx_submitted_urbs);
122 spin_unlock_irqrestore(&aru->tx_urb_lock, flags);
124 return ;
126 spin_unlock_irqrestore(&aru->tx_urb_lock, flags);
128 aru->tx_pending_urbs--;
129 usb_anchor_urb(urb, &aru->tx_submitted);
131 err = usb_submit_urb(urb, GFP_ATOMIC);
132 if (unlikely(err)) {
133 if (ar9170_nag_limiter(&aru->common))
134 dev_err(&aru->udev->dev, "submit_urb failed (%d).\n",
135 err);
137 usb_unanchor_urb(urb);
138 atomic_dec(&aru->tx_submitted_urbs);
139 ar9170_tx_callback(&aru->common, urb->context);
142 usb_free_urb(urb);
145 static void ar9170_usb_tx_urb_complete_frame(struct urb *urb)
147 struct sk_buff *skb = urb->context;
148 struct ar9170_usb *aru = (struct ar9170_usb *)
149 usb_get_intfdata(usb_ifnum_to_if(urb->dev, 0));
151 if (unlikely(!aru)) {
152 dev_kfree_skb_irq(skb);
153 return ;
156 atomic_dec(&aru->tx_submitted_urbs);
158 ar9170_tx_callback(&aru->common, skb);
160 ar9170_usb_submit_urb(aru);
163 static void ar9170_usb_tx_urb_complete(struct urb *urb)
167 static void ar9170_usb_irq_completed(struct urb *urb)
169 struct ar9170_usb *aru = urb->context;
171 switch (urb->status) {
172 /* everything is fine */
173 case 0:
174 break;
176 /* disconnect */
177 case -ENOENT:
178 case -ECONNRESET:
179 case -ENODEV:
180 case -ESHUTDOWN:
181 goto free;
183 default:
184 goto resubmit;
187 ar9170_handle_command_response(&aru->common, urb->transfer_buffer,
188 urb->actual_length);
190 resubmit:
191 usb_anchor_urb(urb, &aru->rx_submitted);
192 if (usb_submit_urb(urb, GFP_ATOMIC)) {
193 usb_unanchor_urb(urb);
194 goto free;
197 return;
199 free:
200 usb_buffer_free(aru->udev, 64, urb->transfer_buffer, urb->transfer_dma);
203 static void ar9170_usb_rx_completed(struct urb *urb)
205 struct sk_buff *skb = urb->context;
206 struct ar9170_usb *aru = (struct ar9170_usb *)
207 usb_get_intfdata(usb_ifnum_to_if(urb->dev, 0));
208 int err;
210 if (!aru)
211 goto free;
213 switch (urb->status) {
214 /* everything is fine */
215 case 0:
216 break;
218 /* disconnect */
219 case -ENOENT:
220 case -ECONNRESET:
221 case -ENODEV:
222 case -ESHUTDOWN:
223 goto free;
225 default:
226 goto resubmit;
229 skb_put(skb, urb->actual_length);
230 ar9170_rx(&aru->common, skb);
232 resubmit:
233 skb_reset_tail_pointer(skb);
234 skb_trim(skb, 0);
236 usb_anchor_urb(urb, &aru->rx_submitted);
237 err = usb_submit_urb(urb, GFP_ATOMIC);
238 if (unlikely(err)) {
239 usb_unanchor_urb(urb);
240 goto free;
243 return ;
245 free:
246 dev_kfree_skb_irq(skb);
249 static int ar9170_usb_prep_rx_urb(struct ar9170_usb *aru,
250 struct urb *urb, gfp_t gfp)
252 struct sk_buff *skb;
254 skb = __dev_alloc_skb(AR9170_MAX_RX_BUFFER_SIZE + 32, gfp);
255 if (!skb)
256 return -ENOMEM;
258 /* reserve some space for mac80211's radiotap */
259 skb_reserve(skb, 32);
261 usb_fill_bulk_urb(urb, aru->udev,
262 usb_rcvbulkpipe(aru->udev, AR9170_EP_RX),
263 skb->data, min(skb_tailroom(skb),
264 AR9170_MAX_RX_BUFFER_SIZE),
265 ar9170_usb_rx_completed, skb);
267 return 0;
270 static int ar9170_usb_alloc_rx_irq_urb(struct ar9170_usb *aru)
272 struct urb *urb = NULL;
273 void *ibuf;
274 int err = -ENOMEM;
276 /* initialize interrupt endpoint */
277 urb = usb_alloc_urb(0, GFP_KERNEL);
278 if (!urb)
279 goto out;
281 ibuf = usb_buffer_alloc(aru->udev, 64, GFP_KERNEL, &urb->transfer_dma);
282 if (!ibuf)
283 goto out;
285 usb_fill_int_urb(urb, aru->udev,
286 usb_rcvintpipe(aru->udev, AR9170_EP_IRQ), ibuf,
287 64, ar9170_usb_irq_completed, aru, 1);
288 urb->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
290 usb_anchor_urb(urb, &aru->rx_submitted);
291 err = usb_submit_urb(urb, GFP_KERNEL);
292 if (err) {
293 usb_unanchor_urb(urb);
294 usb_buffer_free(aru->udev, 64, urb->transfer_buffer,
295 urb->transfer_dma);
298 out:
299 usb_free_urb(urb);
300 return err;
303 static int ar9170_usb_alloc_rx_bulk_urbs(struct ar9170_usb *aru)
305 struct urb *urb;
306 int i;
307 int err = -EINVAL;
309 for (i = 0; i < AR9170_NUM_RX_URBS; i++) {
310 err = -ENOMEM;
311 urb = usb_alloc_urb(0, GFP_KERNEL);
312 if (!urb)
313 goto err_out;
315 err = ar9170_usb_prep_rx_urb(aru, urb, GFP_KERNEL);
316 if (err) {
317 usb_free_urb(urb);
318 goto err_out;
321 usb_anchor_urb(urb, &aru->rx_submitted);
322 err = usb_submit_urb(urb, GFP_KERNEL);
323 if (err) {
324 usb_unanchor_urb(urb);
325 dev_kfree_skb_any((void *) urb->transfer_buffer);
326 usb_free_urb(urb);
327 goto err_out;
329 usb_free_urb(urb);
332 /* the device now waiting for a firmware. */
333 aru->common.state = AR9170_IDLE;
334 return 0;
336 err_out:
338 usb_kill_anchored_urbs(&aru->rx_submitted);
339 return err;
342 static int ar9170_usb_flush(struct ar9170 *ar)
344 struct ar9170_usb *aru = (void *) ar;
345 struct urb *urb;
346 int ret, err = 0;
348 if (IS_STARTED(ar))
349 aru->common.state = AR9170_IDLE;
351 usb_wait_anchor_empty_timeout(&aru->tx_pending,
352 msecs_to_jiffies(800));
353 while ((urb = usb_get_from_anchor(&aru->tx_pending))) {
354 ar9170_tx_callback(&aru->common, (void *) urb->context);
355 usb_free_urb(urb);
358 /* lets wait a while until the tx - queues are dried out */
359 ret = usb_wait_anchor_empty_timeout(&aru->tx_submitted,
360 msecs_to_jiffies(100));
361 if (ret == 0)
362 err = -ETIMEDOUT;
364 usb_kill_anchored_urbs(&aru->tx_submitted);
366 if (IS_ACCEPTING_CMD(ar))
367 aru->common.state = AR9170_STARTED;
369 return err;
372 static void ar9170_usb_cancel_urbs(struct ar9170_usb *aru)
374 int err;
376 aru->common.state = AR9170_UNKNOWN_STATE;
378 err = ar9170_usb_flush(&aru->common);
379 if (err)
380 dev_err(&aru->udev->dev, "stuck tx urbs!\n");
382 usb_poison_anchored_urbs(&aru->tx_submitted);
383 usb_poison_anchored_urbs(&aru->rx_submitted);
386 static int ar9170_usb_exec_cmd(struct ar9170 *ar, enum ar9170_cmd cmd,
387 unsigned int plen, void *payload,
388 unsigned int outlen, void *out)
390 struct ar9170_usb *aru = (void *) ar;
391 struct urb *urb = NULL;
392 unsigned long flags;
393 int err = -ENOMEM;
395 if (unlikely(!IS_ACCEPTING_CMD(ar)))
396 return -EPERM;
398 if (WARN_ON(plen > AR9170_MAX_CMD_LEN - 4))
399 return -EINVAL;
401 urb = usb_alloc_urb(0, GFP_ATOMIC);
402 if (unlikely(!urb))
403 goto err_free;
405 ar->cmdbuf[0] = cpu_to_le32(plen);
406 ar->cmdbuf[0] |= cpu_to_le32(cmd << 8);
407 /* writing multiple regs fills this buffer already */
408 if (plen && payload != (u8 *)(&ar->cmdbuf[1]))
409 memcpy(&ar->cmdbuf[1], payload, plen);
411 spin_lock_irqsave(&aru->common.cmdlock, flags);
412 aru->readbuf = (u8 *)out;
413 aru->readlen = outlen;
414 spin_unlock_irqrestore(&aru->common.cmdlock, flags);
416 usb_fill_int_urb(urb, aru->udev,
417 usb_sndbulkpipe(aru->udev, AR9170_EP_CMD),
418 aru->common.cmdbuf, plen + 4,
419 ar9170_usb_tx_urb_complete, NULL, 1);
421 usb_anchor_urb(urb, &aru->tx_submitted);
422 err = usb_submit_urb(urb, GFP_ATOMIC);
423 if (unlikely(err)) {
424 usb_unanchor_urb(urb);
425 usb_free_urb(urb);
426 goto err_unbuf;
428 usb_free_urb(urb);
430 err = wait_for_completion_timeout(&aru->cmd_wait, HZ);
431 if (err == 0) {
432 err = -ETIMEDOUT;
433 goto err_unbuf;
436 if (aru->readlen != outlen) {
437 err = -EMSGSIZE;
438 goto err_unbuf;
441 return 0;
443 err_unbuf:
444 /* Maybe the device was removed in the second we were waiting? */
445 if (IS_STARTED(ar)) {
446 dev_err(&aru->udev->dev, "no command feedback "
447 "received (%d).\n", err);
449 /* provide some maybe useful debug information */
450 print_hex_dump_bytes("ar9170 cmd: ", DUMP_PREFIX_NONE,
451 aru->common.cmdbuf, plen + 4);
452 dump_stack();
455 /* invalidate to avoid completing the next prematurely */
456 spin_lock_irqsave(&aru->common.cmdlock, flags);
457 aru->readbuf = NULL;
458 aru->readlen = 0;
459 spin_unlock_irqrestore(&aru->common.cmdlock, flags);
461 err_free:
463 return err;
466 static int ar9170_usb_tx(struct ar9170 *ar, struct sk_buff *skb)
468 struct ar9170_usb *aru = (struct ar9170_usb *) ar;
469 struct urb *urb;
471 if (unlikely(!IS_STARTED(ar))) {
472 /* Seriously, what were you drink... err... thinking!? */
473 return -EPERM;
476 urb = usb_alloc_urb(0, GFP_ATOMIC);
477 if (unlikely(!urb))
478 return -ENOMEM;
480 usb_fill_bulk_urb(urb, aru->udev,
481 usb_sndbulkpipe(aru->udev, AR9170_EP_TX),
482 skb->data, skb->len,
483 ar9170_usb_tx_urb_complete_frame, skb);
484 urb->transfer_flags |= URB_ZERO_PACKET;
486 usb_anchor_urb(urb, &aru->tx_pending);
487 aru->tx_pending_urbs++;
489 usb_free_urb(urb);
491 ar9170_usb_submit_urb(aru);
492 return 0;
495 static void ar9170_usb_callback_cmd(struct ar9170 *ar, u32 len , void *buffer)
497 struct ar9170_usb *aru = (void *) ar;
498 unsigned long flags;
499 u32 in, out;
501 if (unlikely(!buffer))
502 return ;
504 in = le32_to_cpup((__le32 *)buffer);
505 out = le32_to_cpu(ar->cmdbuf[0]);
507 /* mask off length byte */
508 out &= ~0xFF;
510 if (aru->readlen >= 0) {
511 /* add expected length */
512 out |= aru->readlen;
513 } else {
514 /* add obtained length */
515 out |= in & 0xFF;
519 * Some commands (e.g: AR9170_CMD_FREQUENCY) have a variable response
520 * length and we cannot predict the correct length in advance.
521 * So we only check if we provided enough space for the data.
523 if (unlikely(out < in)) {
524 dev_warn(&aru->udev->dev, "received invalid command response "
525 "got %d bytes, instead of %d bytes "
526 "and the resp length is %d bytes\n",
527 in, out, len);
528 print_hex_dump_bytes("ar9170 invalid resp: ",
529 DUMP_PREFIX_OFFSET, buffer, len);
531 * Do not complete, then the command times out,
532 * and we get a stack trace from there.
534 return ;
537 spin_lock_irqsave(&aru->common.cmdlock, flags);
538 if (aru->readbuf && len > 0) {
539 memcpy(aru->readbuf, buffer + 4, len - 4);
540 aru->readbuf = NULL;
542 complete(&aru->cmd_wait);
543 spin_unlock_irqrestore(&aru->common.cmdlock, flags);
546 static int ar9170_usb_upload(struct ar9170_usb *aru, const void *data,
547 size_t len, u32 addr, bool complete)
549 int transfer, err;
550 u8 *buf = kmalloc(4096, GFP_KERNEL);
552 if (!buf)
553 return -ENOMEM;
555 while (len) {
556 transfer = min_t(int, len, 4096);
557 memcpy(buf, data, transfer);
559 err = usb_control_msg(aru->udev, usb_sndctrlpipe(aru->udev, 0),
560 0x30 /* FW DL */, 0x40 | USB_DIR_OUT,
561 addr >> 8, 0, buf, transfer, 1000);
563 if (err < 0) {
564 kfree(buf);
565 return err;
568 len -= transfer;
569 data += transfer;
570 addr += transfer;
572 kfree(buf);
574 if (complete) {
575 err = usb_control_msg(aru->udev, usb_sndctrlpipe(aru->udev, 0),
576 0x31 /* FW DL COMPLETE */,
577 0x40 | USB_DIR_OUT, 0, 0, NULL, 0, 5000);
580 return 0;
583 static int ar9170_usb_request_firmware(struct ar9170_usb *aru)
585 int err = 0;
587 err = request_firmware(&aru->firmware, "ar9170.fw",
588 &aru->udev->dev);
589 if (!err) {
590 aru->init_values = NULL;
591 return 0;
594 if (aru->req_one_stage_fw) {
595 dev_err(&aru->udev->dev, "ar9170.fw firmware file "
596 "not found and is required for this device\n");
597 return -EINVAL;
600 dev_err(&aru->udev->dev, "ar9170.fw firmware file "
601 "not found, trying old firmware...\n");
603 err = request_firmware(&aru->init_values, "ar9170-1.fw",
604 &aru->udev->dev);
605 if (err) {
606 dev_err(&aru->udev->dev, "file with init values not found.\n");
607 return err;
610 err = request_firmware(&aru->firmware, "ar9170-2.fw", &aru->udev->dev);
611 if (err) {
612 release_firmware(aru->init_values);
613 dev_err(&aru->udev->dev, "firmware file not found.\n");
614 return err;
617 return err;
620 static int ar9170_usb_reset(struct ar9170_usb *aru)
622 int ret, lock = (aru->intf->condition != USB_INTERFACE_BINDING);
624 if (lock) {
625 ret = usb_lock_device_for_reset(aru->udev, aru->intf);
626 if (ret < 0) {
627 dev_err(&aru->udev->dev, "unable to lock device "
628 "for reset (%d).\n", ret);
629 return ret;
633 ret = usb_reset_device(aru->udev);
634 if (lock)
635 usb_unlock_device(aru->udev);
637 /* let it rest - for a second - */
638 msleep(1000);
640 return ret;
643 static int ar9170_usb_upload_firmware(struct ar9170_usb *aru)
645 int err;
647 if (!aru->init_values)
648 goto upload_fw_start;
650 /* First, upload initial values to device RAM */
651 err = ar9170_usb_upload(aru, aru->init_values->data,
652 aru->init_values->size, 0x102800, false);
653 if (err) {
654 dev_err(&aru->udev->dev, "firmware part 1 "
655 "upload failed (%d).\n", err);
656 return err;
659 upload_fw_start:
661 /* Then, upload the firmware itself and start it */
662 return ar9170_usb_upload(aru, aru->firmware->data, aru->firmware->size,
663 0x200000, true);
666 static int ar9170_usb_init_transport(struct ar9170_usb *aru)
668 struct ar9170 *ar = (void *) &aru->common;
669 int err;
671 ar9170_regwrite_begin(ar);
673 /* Set USB Rx stream mode MAX packet number to 2 */
674 ar9170_regwrite(AR9170_USB_REG_MAX_AGG_UPLOAD, 0x4);
676 /* Set USB Rx stream mode timeout to 10us */
677 ar9170_regwrite(AR9170_USB_REG_UPLOAD_TIME_CTL, 0x80);
679 ar9170_regwrite_finish();
681 err = ar9170_regwrite_result();
682 if (err)
683 dev_err(&aru->udev->dev, "USB setup failed (%d).\n", err);
685 return err;
688 static void ar9170_usb_stop(struct ar9170 *ar)
690 struct ar9170_usb *aru = (void *) ar;
691 int ret;
693 if (IS_ACCEPTING_CMD(ar))
694 aru->common.state = AR9170_STOPPED;
696 ret = ar9170_usb_flush(ar);
697 if (ret)
698 dev_err(&aru->udev->dev, "kill pending tx urbs.\n");
700 usb_poison_anchored_urbs(&aru->tx_submitted);
703 * Note:
704 * So far we freed all tx urbs, but we won't dare to touch any rx urbs.
705 * Else we would end up with a unresponsive device...
709 static int ar9170_usb_open(struct ar9170 *ar)
711 struct ar9170_usb *aru = (void *) ar;
712 int err;
714 usb_unpoison_anchored_urbs(&aru->tx_submitted);
715 err = ar9170_usb_init_transport(aru);
716 if (err) {
717 usb_poison_anchored_urbs(&aru->tx_submitted);
718 return err;
721 aru->common.state = AR9170_IDLE;
722 return 0;
725 static int ar9170_usb_init_device(struct ar9170_usb *aru)
727 int err;
729 err = ar9170_usb_alloc_rx_irq_urb(aru);
730 if (err)
731 goto err_out;
733 err = ar9170_usb_alloc_rx_bulk_urbs(aru);
734 if (err)
735 goto err_unrx;
737 err = ar9170_usb_upload_firmware(aru);
738 if (err) {
739 err = ar9170_echo_test(&aru->common, 0x60d43110);
740 if (err) {
741 /* force user invention, by disabling the device */
742 err = usb_driver_set_configuration(aru->udev, -1);
743 dev_err(&aru->udev->dev, "device is in a bad state. "
744 "please reconnect it!\n");
745 goto err_unrx;
749 return 0;
751 err_unrx:
752 ar9170_usb_cancel_urbs(aru);
754 err_out:
755 return err;
758 static bool ar9170_requires_one_stage(const struct usb_device_id *id)
760 if (!id->driver_info)
761 return false;
762 if (id->driver_info == AR9170_REQ_FW1_ONLY)
763 return true;
764 return false;
767 static int ar9170_usb_probe(struct usb_interface *intf,
768 const struct usb_device_id *id)
770 struct ar9170_usb *aru;
771 struct ar9170 *ar;
772 struct usb_device *udev;
773 int err;
775 aru = ar9170_alloc(sizeof(*aru));
776 if (IS_ERR(aru)) {
777 err = PTR_ERR(aru);
778 goto out;
781 udev = interface_to_usbdev(intf);
782 usb_get_dev(udev);
783 aru->udev = udev;
784 aru->intf = intf;
785 ar = &aru->common;
787 aru->req_one_stage_fw = ar9170_requires_one_stage(id);
789 usb_set_intfdata(intf, aru);
790 SET_IEEE80211_DEV(ar->hw, &intf->dev);
792 init_usb_anchor(&aru->rx_submitted);
793 init_usb_anchor(&aru->tx_pending);
794 init_usb_anchor(&aru->tx_submitted);
795 init_completion(&aru->cmd_wait);
796 spin_lock_init(&aru->tx_urb_lock);
798 aru->tx_pending_urbs = 0;
799 atomic_set(&aru->tx_submitted_urbs, 0);
801 aru->common.stop = ar9170_usb_stop;
802 aru->common.flush = ar9170_usb_flush;
803 aru->common.open = ar9170_usb_open;
804 aru->common.tx = ar9170_usb_tx;
805 aru->common.exec_cmd = ar9170_usb_exec_cmd;
806 aru->common.callback_cmd = ar9170_usb_callback_cmd;
808 #ifdef CONFIG_PM
809 udev->reset_resume = 1;
810 #endif /* CONFIG_PM */
811 err = ar9170_usb_reset(aru);
812 if (err)
813 goto err_freehw;
815 err = ar9170_usb_request_firmware(aru);
816 if (err)
817 goto err_freehw;
819 err = ar9170_usb_init_device(aru);
820 if (err)
821 goto err_freefw;
823 err = ar9170_usb_open(ar);
824 if (err)
825 goto err_unrx;
827 err = ar9170_register(ar, &udev->dev);
829 ar9170_usb_stop(ar);
830 if (err)
831 goto err_unrx;
833 return 0;
835 err_unrx:
836 ar9170_usb_cancel_urbs(aru);
838 err_freefw:
839 release_firmware(aru->init_values);
840 release_firmware(aru->firmware);
842 err_freehw:
843 usb_set_intfdata(intf, NULL);
844 usb_put_dev(udev);
845 ieee80211_free_hw(ar->hw);
846 out:
847 return err;
850 static void ar9170_usb_disconnect(struct usb_interface *intf)
852 struct ar9170_usb *aru = usb_get_intfdata(intf);
854 if (!aru)
855 return;
857 aru->common.state = AR9170_IDLE;
858 ar9170_unregister(&aru->common);
859 ar9170_usb_cancel_urbs(aru);
861 release_firmware(aru->init_values);
862 release_firmware(aru->firmware);
864 usb_put_dev(aru->udev);
865 usb_set_intfdata(intf, NULL);
866 ieee80211_free_hw(aru->common.hw);
869 #ifdef CONFIG_PM
870 static int ar9170_suspend(struct usb_interface *intf,
871 pm_message_t message)
873 struct ar9170_usb *aru = usb_get_intfdata(intf);
875 if (!aru)
876 return -ENODEV;
878 aru->common.state = AR9170_IDLE;
879 ar9170_usb_cancel_urbs(aru);
881 return 0;
884 static int ar9170_resume(struct usb_interface *intf)
886 struct ar9170_usb *aru = usb_get_intfdata(intf);
887 int err;
889 if (!aru)
890 return -ENODEV;
892 usb_unpoison_anchored_urbs(&aru->rx_submitted);
893 usb_unpoison_anchored_urbs(&aru->tx_submitted);
895 err = ar9170_usb_init_device(aru);
896 if (err)
897 goto err_unrx;
899 err = ar9170_usb_open(&aru->common);
900 if (err)
901 goto err_unrx;
903 return 0;
905 err_unrx:
906 aru->common.state = AR9170_IDLE;
907 ar9170_usb_cancel_urbs(aru);
909 return err;
911 #endif /* CONFIG_PM */
913 static struct usb_driver ar9170_driver = {
914 .name = "ar9170usb",
915 .probe = ar9170_usb_probe,
916 .disconnect = ar9170_usb_disconnect,
917 .id_table = ar9170_usb_ids,
918 .soft_unbind = 1,
919 #ifdef CONFIG_PM
920 .suspend = ar9170_suspend,
921 .resume = ar9170_resume,
922 .reset_resume = ar9170_resume,
923 #endif /* CONFIG_PM */
926 static int __init ar9170_init(void)
928 return usb_register(&ar9170_driver);
931 static void __exit ar9170_exit(void)
933 usb_deregister(&ar9170_driver);
936 module_init(ar9170_init);
937 module_exit(ar9170_exit);