Merge branch 'master' of git://git.kernel.org/pub/scm/linux/kernel/git/linville/wirel...
[linux-2.6.git] / drivers / net / wireless / ath / ar9170 / usb.c
1 /*
2  * Atheros AR9170 driver
3  *
4  * USB - frontend
5  *
6  * Copyright 2008, Johannes Berg <johannes@sipsolutions.net>
7  * Copyright 2009, Christian Lamparter <chunkeey@web.de>
8  *
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.
13  *
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.
18  *
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/.
22  *
23  * This file incorporates work covered by the following copyright and
24  * permission notice:
25  *    Copyright (c) 2007-2008 Atheros Communications, Inc.
26  *
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.
30  *
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.
38  */
39
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"
49
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");
57
58 enum ar9170_requirements {
59         AR9170_REQ_FW1_ONLY = 1,
60 };
61
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         /* Sphairon Homelink 1202 */
88         { USB_DEVICE(0x0cde, 0x0027) },
89         /* Arcadyan WN7512 */
90         { USB_DEVICE(0x083a, 0xf522) },
91         /* Planex GWUS300 */
92         { USB_DEVICE(0x2019, 0x5304) },
93         /* IO-Data WNGDNUS2 */
94         { USB_DEVICE(0x04bb, 0x093f) },
95         /* AVM FRITZ!WLAN USB Stick N */
96         { USB_DEVICE(0x057C, 0x8401) },
97         /* AVM FRITZ!WLAN USB Stick N 2.4 */
98         { USB_DEVICE(0x057C, 0x8402), .driver_info = AR9170_REQ_FW1_ONLY },
99
100         /* terminate */
101         {}
102 };
103 MODULE_DEVICE_TABLE(usb, ar9170_usb_ids);
104
105 static void ar9170_usb_submit_urb(struct ar9170_usb *aru)
106 {
107         struct urb *urb;
108         unsigned long flags;
109         int err;
110
111         if (unlikely(!IS_STARTED(&aru->common)))
112                 return ;
113
114         spin_lock_irqsave(&aru->tx_urb_lock, flags);
115         if (atomic_read(&aru->tx_submitted_urbs) >= AR9170_NUM_TX_URBS) {
116                 spin_unlock_irqrestore(&aru->tx_urb_lock, flags);
117                 return ;
118         }
119         atomic_inc(&aru->tx_submitted_urbs);
120
121         urb = usb_get_from_anchor(&aru->tx_pending);
122         if (!urb) {
123                 atomic_dec(&aru->tx_submitted_urbs);
124                 spin_unlock_irqrestore(&aru->tx_urb_lock, flags);
125
126                 return ;
127         }
128         spin_unlock_irqrestore(&aru->tx_urb_lock, flags);
129
130         aru->tx_pending_urbs--;
131         usb_anchor_urb(urb, &aru->tx_submitted);
132
133         err = usb_submit_urb(urb, GFP_ATOMIC);
134         if (unlikely(err)) {
135                 if (ar9170_nag_limiter(&aru->common))
136                         dev_err(&aru->udev->dev, "submit_urb failed (%d).\n",
137                                 err);
138
139                 usb_unanchor_urb(urb);
140                 atomic_dec(&aru->tx_submitted_urbs);
141                 ar9170_tx_callback(&aru->common, urb->context);
142         }
143
144         usb_free_urb(urb);
145 }
146
147 static void ar9170_usb_tx_urb_complete_frame(struct urb *urb)
148 {
149         struct sk_buff *skb = urb->context;
150         struct ar9170_usb *aru = (struct ar9170_usb *)
151               usb_get_intfdata(usb_ifnum_to_if(urb->dev, 0));
152
153         if (unlikely(!aru)) {
154                 dev_kfree_skb_irq(skb);
155                 return ;
156         }
157
158         atomic_dec(&aru->tx_submitted_urbs);
159
160         ar9170_tx_callback(&aru->common, skb);
161
162         ar9170_usb_submit_urb(aru);
163 }
164
165 static void ar9170_usb_tx_urb_complete(struct urb *urb)
166 {
167 }
168
169 static void ar9170_usb_irq_completed(struct urb *urb)
170 {
171         struct ar9170_usb *aru = urb->context;
172
173         switch (urb->status) {
174         /* everything is fine */
175         case 0:
176                 break;
177
178         /* disconnect */
179         case -ENOENT:
180         case -ECONNRESET:
181         case -ENODEV:
182         case -ESHUTDOWN:
183                 goto free;
184
185         default:
186                 goto resubmit;
187         }
188
189         ar9170_handle_command_response(&aru->common, urb->transfer_buffer,
190                                        urb->actual_length);
191
192 resubmit:
193         usb_anchor_urb(urb, &aru->rx_submitted);
194         if (usb_submit_urb(urb, GFP_ATOMIC)) {
195                 usb_unanchor_urb(urb);
196                 goto free;
197         }
198
199         return;
200
201 free:
202         usb_buffer_free(aru->udev, 64, urb->transfer_buffer, urb->transfer_dma);
203 }
204
205 static void ar9170_usb_rx_completed(struct urb *urb)
206 {
207         struct sk_buff *skb = urb->context;
208         struct ar9170_usb *aru = (struct ar9170_usb *)
209                 usb_get_intfdata(usb_ifnum_to_if(urb->dev, 0));
210         int err;
211
212         if (!aru)
213                 goto free;
214
215         switch (urb->status) {
216         /* everything is fine */
217         case 0:
218                 break;
219
220         /* disconnect */
221         case -ENOENT:
222         case -ECONNRESET:
223         case -ENODEV:
224         case -ESHUTDOWN:
225                 goto free;
226
227         default:
228                 goto resubmit;
229         }
230
231         skb_put(skb, urb->actual_length);
232         ar9170_rx(&aru->common, skb);
233
234 resubmit:
235         skb_reset_tail_pointer(skb);
236         skb_trim(skb, 0);
237
238         usb_anchor_urb(urb, &aru->rx_submitted);
239         err = usb_submit_urb(urb, GFP_ATOMIC);
240         if (unlikely(err)) {
241                 usb_unanchor_urb(urb);
242                 goto free;
243         }
244
245         return ;
246
247 free:
248         dev_kfree_skb_irq(skb);
249 }
250
251 static int ar9170_usb_prep_rx_urb(struct ar9170_usb *aru,
252                                   struct urb *urb, gfp_t gfp)
253 {
254         struct sk_buff *skb;
255
256         skb = __dev_alloc_skb(AR9170_MAX_RX_BUFFER_SIZE + 32, gfp);
257         if (!skb)
258                 return -ENOMEM;
259
260         /* reserve some space for mac80211's radiotap */
261         skb_reserve(skb, 32);
262
263         usb_fill_bulk_urb(urb, aru->udev,
264                           usb_rcvbulkpipe(aru->udev, AR9170_EP_RX),
265                           skb->data, min(skb_tailroom(skb),
266                           AR9170_MAX_RX_BUFFER_SIZE),
267                           ar9170_usb_rx_completed, skb);
268
269         return 0;
270 }
271
272 static int ar9170_usb_alloc_rx_irq_urb(struct ar9170_usb *aru)
273 {
274         struct urb *urb = NULL;
275         void *ibuf;
276         int err = -ENOMEM;
277
278         /* initialize interrupt endpoint */
279         urb = usb_alloc_urb(0, GFP_KERNEL);
280         if (!urb)
281                 goto out;
282
283         ibuf = usb_buffer_alloc(aru->udev, 64, GFP_KERNEL, &urb->transfer_dma);
284         if (!ibuf)
285                 goto out;
286
287         usb_fill_int_urb(urb, aru->udev,
288                          usb_rcvintpipe(aru->udev, AR9170_EP_IRQ), ibuf,
289                          64, ar9170_usb_irq_completed, aru, 1);
290         urb->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
291
292         usb_anchor_urb(urb, &aru->rx_submitted);
293         err = usb_submit_urb(urb, GFP_KERNEL);
294         if (err) {
295                 usb_unanchor_urb(urb);
296                 usb_buffer_free(aru->udev, 64, urb->transfer_buffer,
297                                 urb->transfer_dma);
298         }
299
300 out:
301         usb_free_urb(urb);
302         return err;
303 }
304
305 static int ar9170_usb_alloc_rx_bulk_urbs(struct ar9170_usb *aru)
306 {
307         struct urb *urb;
308         int i;
309         int err = -EINVAL;
310
311         for (i = 0; i < AR9170_NUM_RX_URBS; i++) {
312                 err = -ENOMEM;
313                 urb = usb_alloc_urb(0, GFP_KERNEL);
314                 if (!urb)
315                         goto err_out;
316
317                 err = ar9170_usb_prep_rx_urb(aru, urb, GFP_KERNEL);
318                 if (err) {
319                         usb_free_urb(urb);
320                         goto err_out;
321                 }
322
323                 usb_anchor_urb(urb, &aru->rx_submitted);
324                 err = usb_submit_urb(urb, GFP_KERNEL);
325                 if (err) {
326                         usb_unanchor_urb(urb);
327                         dev_kfree_skb_any((void *) urb->transfer_buffer);
328                         usb_free_urb(urb);
329                         goto err_out;
330                 }
331                 usb_free_urb(urb);
332         }
333
334         /* the device now waiting for a firmware. */
335         aru->common.state = AR9170_IDLE;
336         return 0;
337
338 err_out:
339
340         usb_kill_anchored_urbs(&aru->rx_submitted);
341         return err;
342 }
343
344 static int ar9170_usb_flush(struct ar9170 *ar)
345 {
346         struct ar9170_usb *aru = (void *) ar;
347         struct urb *urb;
348         int ret, err = 0;
349
350         if (IS_STARTED(ar))
351                 aru->common.state = AR9170_IDLE;
352
353         usb_wait_anchor_empty_timeout(&aru->tx_pending,
354                                             msecs_to_jiffies(800));
355         while ((urb = usb_get_from_anchor(&aru->tx_pending))) {
356                 ar9170_tx_callback(&aru->common, (void *) urb->context);
357                 usb_free_urb(urb);
358         }
359
360         /* lets wait a while until the tx - queues are dried out */
361         ret = usb_wait_anchor_empty_timeout(&aru->tx_submitted,
362                                             msecs_to_jiffies(100));
363         if (ret == 0)
364                 err = -ETIMEDOUT;
365
366         usb_kill_anchored_urbs(&aru->tx_submitted);
367
368         if (IS_ACCEPTING_CMD(ar))
369                 aru->common.state = AR9170_STARTED;
370
371         return err;
372 }
373
374 static void ar9170_usb_cancel_urbs(struct ar9170_usb *aru)
375 {
376         int err;
377
378         aru->common.state = AR9170_UNKNOWN_STATE;
379
380         err = ar9170_usb_flush(&aru->common);
381         if (err)
382                 dev_err(&aru->udev->dev, "stuck tx urbs!\n");
383
384         usb_poison_anchored_urbs(&aru->tx_submitted);
385         usb_poison_anchored_urbs(&aru->rx_submitted);
386 }
387
388 static int ar9170_usb_exec_cmd(struct ar9170 *ar, enum ar9170_cmd cmd,
389                                unsigned int plen, void *payload,
390                                unsigned int outlen, void *out)
391 {
392         struct ar9170_usb *aru = (void *) ar;
393         struct urb *urb = NULL;
394         unsigned long flags;
395         int err = -ENOMEM;
396
397         if (unlikely(!IS_ACCEPTING_CMD(ar)))
398                 return -EPERM;
399
400         if (WARN_ON(plen > AR9170_MAX_CMD_LEN - 4))
401                 return -EINVAL;
402
403         urb = usb_alloc_urb(0, GFP_ATOMIC);
404         if (unlikely(!urb))
405                 goto err_free;
406
407         ar->cmdbuf[0] = cpu_to_le32(plen);
408         ar->cmdbuf[0] |= cpu_to_le32(cmd << 8);
409         /* writing multiple regs fills this buffer already */
410         if (plen && payload != (u8 *)(&ar->cmdbuf[1]))
411                 memcpy(&ar->cmdbuf[1], payload, plen);
412
413         spin_lock_irqsave(&aru->common.cmdlock, flags);
414         aru->readbuf = (u8 *)out;
415         aru->readlen = outlen;
416         spin_unlock_irqrestore(&aru->common.cmdlock, flags);
417
418         usb_fill_int_urb(urb, aru->udev,
419                          usb_sndbulkpipe(aru->udev, AR9170_EP_CMD),
420                          aru->common.cmdbuf, plen + 4,
421                          ar9170_usb_tx_urb_complete, NULL, 1);
422
423         usb_anchor_urb(urb, &aru->tx_submitted);
424         err = usb_submit_urb(urb, GFP_ATOMIC);
425         if (unlikely(err)) {
426                 usb_unanchor_urb(urb);
427                 usb_free_urb(urb);
428                 goto err_unbuf;
429         }
430         usb_free_urb(urb);
431
432         err = wait_for_completion_timeout(&aru->cmd_wait, HZ);
433         if (err == 0) {
434                 err = -ETIMEDOUT;
435                 goto err_unbuf;
436         }
437
438         if (aru->readlen != outlen) {
439                 err = -EMSGSIZE;
440                 goto err_unbuf;
441         }
442
443         return 0;
444
445 err_unbuf:
446         /* Maybe the device was removed in the second we were waiting? */
447         if (IS_STARTED(ar)) {
448                 dev_err(&aru->udev->dev, "no command feedback "
449                                          "received (%d).\n", err);
450
451                 /* provide some maybe useful debug information */
452                 print_hex_dump_bytes("ar9170 cmd: ", DUMP_PREFIX_NONE,
453                                      aru->common.cmdbuf, plen + 4);
454                 dump_stack();
455         }
456
457         /* invalidate to avoid completing the next prematurely */
458         spin_lock_irqsave(&aru->common.cmdlock, flags);
459         aru->readbuf = NULL;
460         aru->readlen = 0;
461         spin_unlock_irqrestore(&aru->common.cmdlock, flags);
462
463 err_free:
464
465         return err;
466 }
467
468 static int ar9170_usb_tx(struct ar9170 *ar, struct sk_buff *skb)
469 {
470         struct ar9170_usb *aru = (struct ar9170_usb *) ar;
471         struct urb *urb;
472
473         if (unlikely(!IS_STARTED(ar))) {
474                 /* Seriously, what were you drink... err... thinking!? */
475                 return -EPERM;
476         }
477
478         urb = usb_alloc_urb(0, GFP_ATOMIC);
479         if (unlikely(!urb))
480                 return -ENOMEM;
481
482         usb_fill_bulk_urb(urb, aru->udev,
483                           usb_sndbulkpipe(aru->udev, AR9170_EP_TX),
484                           skb->data, skb->len,
485                           ar9170_usb_tx_urb_complete_frame, skb);
486         urb->transfer_flags |= URB_ZERO_PACKET;
487
488         usb_anchor_urb(urb, &aru->tx_pending);
489         aru->tx_pending_urbs++;
490
491         usb_free_urb(urb);
492
493         ar9170_usb_submit_urb(aru);
494         return 0;
495 }
496
497 static void ar9170_usb_callback_cmd(struct ar9170 *ar, u32 len , void *buffer)
498 {
499         struct ar9170_usb *aru = (void *) ar;
500         unsigned long flags;
501         u32 in, out;
502
503         if (unlikely(!buffer))
504                 return ;
505
506         in = le32_to_cpup((__le32 *)buffer);
507         out = le32_to_cpu(ar->cmdbuf[0]);
508
509         /* mask off length byte */
510         out &= ~0xFF;
511
512         if (aru->readlen >= 0) {
513                 /* add expected length */
514                 out |= aru->readlen;
515         } else {
516                 /* add obtained length */
517                 out |= in & 0xFF;
518         }
519
520         /*
521          * Some commands (e.g: AR9170_CMD_FREQUENCY) have a variable response
522          * length and we cannot predict the correct length in advance.
523          * So we only check if we provided enough space for the data.
524          */
525         if (unlikely(out < in)) {
526                 dev_warn(&aru->udev->dev, "received invalid command response "
527                                           "got %d bytes, instead of %d bytes "
528                                           "and the resp length is %d bytes\n",
529                          in, out, len);
530                 print_hex_dump_bytes("ar9170 invalid resp: ",
531                                      DUMP_PREFIX_OFFSET, buffer, len);
532                 /*
533                  * Do not complete, then the command times out,
534                  * and we get a stack trace from there.
535                  */
536                 return ;
537         }
538
539         spin_lock_irqsave(&aru->common.cmdlock, flags);
540         if (aru->readbuf && len > 0) {
541                 memcpy(aru->readbuf, buffer + 4, len - 4);
542                 aru->readbuf = NULL;
543         }
544         complete(&aru->cmd_wait);
545         spin_unlock_irqrestore(&aru->common.cmdlock, flags);
546 }
547
548 static int ar9170_usb_upload(struct ar9170_usb *aru, const void *data,
549                              size_t len, u32 addr, bool complete)
550 {
551         int transfer, err;
552         u8 *buf = kmalloc(4096, GFP_KERNEL);
553
554         if (!buf)
555                 return -ENOMEM;
556
557         while (len) {
558                 transfer = min_t(int, len, 4096);
559                 memcpy(buf, data, transfer);
560
561                 err = usb_control_msg(aru->udev, usb_sndctrlpipe(aru->udev, 0),
562                                       0x30 /* FW DL */, 0x40 | USB_DIR_OUT,
563                                       addr >> 8, 0, buf, transfer, 1000);
564
565                 if (err < 0) {
566                         kfree(buf);
567                         return err;
568                 }
569
570                 len -= transfer;
571                 data += transfer;
572                 addr += transfer;
573         }
574         kfree(buf);
575
576         if (complete) {
577                 err = usb_control_msg(aru->udev, usb_sndctrlpipe(aru->udev, 0),
578                                       0x31 /* FW DL COMPLETE */,
579                                       0x40 | USB_DIR_OUT, 0, 0, NULL, 0, 5000);
580         }
581
582         return 0;
583 }
584
585 static int ar9170_usb_request_firmware(struct ar9170_usb *aru)
586 {
587         int err = 0;
588
589         err = request_firmware(&aru->firmware, "ar9170.fw",
590                                &aru->udev->dev);
591         if (!err) {
592                 aru->init_values = NULL;
593                 return 0;
594         }
595
596         if (aru->req_one_stage_fw) {
597                 dev_err(&aru->udev->dev, "ar9170.fw firmware file "
598                         "not found and is required for this device\n");
599                 return -EINVAL;
600         }
601
602         dev_err(&aru->udev->dev, "ar9170.fw firmware file "
603                 "not found, trying old firmware...\n");
604
605         err = request_firmware(&aru->init_values, "ar9170-1.fw",
606                                &aru->udev->dev);
607         if (err) {
608                 dev_err(&aru->udev->dev, "file with init values not found.\n");
609                 return err;
610         }
611
612         err = request_firmware(&aru->firmware, "ar9170-2.fw", &aru->udev->dev);
613         if (err) {
614                 release_firmware(aru->init_values);
615                 dev_err(&aru->udev->dev, "firmware file not found.\n");
616                 return err;
617         }
618
619         return err;
620 }
621
622 static int ar9170_usb_reset(struct ar9170_usb *aru)
623 {
624         int ret, lock = (aru->intf->condition != USB_INTERFACE_BINDING);
625
626         if (lock) {
627                 ret = usb_lock_device_for_reset(aru->udev, aru->intf);
628                 if (ret < 0) {
629                         dev_err(&aru->udev->dev, "unable to lock device "
630                                 "for reset (%d).\n", ret);
631                         return ret;
632                 }
633         }
634
635         ret = usb_reset_device(aru->udev);
636         if (lock)
637                 usb_unlock_device(aru->udev);
638
639         /* let it rest - for a second - */
640         msleep(1000);
641
642         return ret;
643 }
644
645 static int ar9170_usb_upload_firmware(struct ar9170_usb *aru)
646 {
647         int err;
648
649         if (!aru->init_values)
650                 goto upload_fw_start;
651
652         /* First, upload initial values to device RAM */
653         err = ar9170_usb_upload(aru, aru->init_values->data,
654                                 aru->init_values->size, 0x102800, false);
655         if (err) {
656                 dev_err(&aru->udev->dev, "firmware part 1 "
657                         "upload failed (%d).\n", err);
658                 return err;
659         }
660
661 upload_fw_start:
662
663         /* Then, upload the firmware itself and start it */
664         return ar9170_usb_upload(aru, aru->firmware->data, aru->firmware->size,
665                                 0x200000, true);
666 }
667
668 static int ar9170_usb_init_transport(struct ar9170_usb *aru)
669 {
670         struct ar9170 *ar = (void *) &aru->common;
671         int err;
672
673         ar9170_regwrite_begin(ar);
674
675         /* Set USB Rx stream mode MAX packet number to 2 */
676         ar9170_regwrite(AR9170_USB_REG_MAX_AGG_UPLOAD, 0x4);
677
678         /* Set USB Rx stream mode timeout to 10us */
679         ar9170_regwrite(AR9170_USB_REG_UPLOAD_TIME_CTL, 0x80);
680
681         ar9170_regwrite_finish();
682
683         err = ar9170_regwrite_result();
684         if (err)
685                 dev_err(&aru->udev->dev, "USB setup failed (%d).\n", err);
686
687         return err;
688 }
689
690 static void ar9170_usb_stop(struct ar9170 *ar)
691 {
692         struct ar9170_usb *aru = (void *) ar;
693         int ret;
694
695         if (IS_ACCEPTING_CMD(ar))
696                 aru->common.state = AR9170_STOPPED;
697
698         ret = ar9170_usb_flush(ar);
699         if (ret)
700                 dev_err(&aru->udev->dev, "kill pending tx urbs.\n");
701
702         usb_poison_anchored_urbs(&aru->tx_submitted);
703
704         /*
705          * Note:
706          * So far we freed all tx urbs, but we won't dare to touch any rx urbs.
707          * Else we would end up with a unresponsive device...
708          */
709 }
710
711 static int ar9170_usb_open(struct ar9170 *ar)
712 {
713         struct ar9170_usb *aru = (void *) ar;
714         int err;
715
716         usb_unpoison_anchored_urbs(&aru->tx_submitted);
717         err = ar9170_usb_init_transport(aru);
718         if (err) {
719                 usb_poison_anchored_urbs(&aru->tx_submitted);
720                 return err;
721         }
722
723         aru->common.state = AR9170_IDLE;
724         return 0;
725 }
726
727 static int ar9170_usb_init_device(struct ar9170_usb *aru)
728 {
729         int err;
730
731         err = ar9170_usb_alloc_rx_irq_urb(aru);
732         if (err)
733                 goto err_out;
734
735         err = ar9170_usb_alloc_rx_bulk_urbs(aru);
736         if (err)
737                 goto err_unrx;
738
739         err = ar9170_usb_upload_firmware(aru);
740         if (err) {
741                 err = ar9170_echo_test(&aru->common, 0x60d43110);
742                 if (err) {
743                         /* force user invention, by disabling the device */
744                         err = usb_driver_set_configuration(aru->udev, -1);
745                         dev_err(&aru->udev->dev, "device is in a bad state. "
746                                                  "please reconnect it!\n");
747                         goto err_unrx;
748                 }
749         }
750
751         return 0;
752
753 err_unrx:
754         ar9170_usb_cancel_urbs(aru);
755
756 err_out:
757         return err;
758 }
759
760 static bool ar9170_requires_one_stage(const struct usb_device_id *id)
761 {
762         if (!id->driver_info)
763                 return false;
764         if (id->driver_info == AR9170_REQ_FW1_ONLY)
765                 return true;
766         return false;
767 }
768
769 static int ar9170_usb_probe(struct usb_interface *intf,
770                         const struct usb_device_id *id)
771 {
772         struct ar9170_usb *aru;
773         struct ar9170 *ar;
774         struct usb_device *udev;
775         int err;
776
777         aru = ar9170_alloc(sizeof(*aru));
778         if (IS_ERR(aru)) {
779                 err = PTR_ERR(aru);
780                 goto out;
781         }
782
783         udev = interface_to_usbdev(intf);
784         usb_get_dev(udev);
785         aru->udev = udev;
786         aru->intf = intf;
787         ar = &aru->common;
788
789         aru->req_one_stage_fw = ar9170_requires_one_stage(id);
790
791         usb_set_intfdata(intf, aru);
792         SET_IEEE80211_DEV(ar->hw, &intf->dev);
793
794         init_usb_anchor(&aru->rx_submitted);
795         init_usb_anchor(&aru->tx_pending);
796         init_usb_anchor(&aru->tx_submitted);
797         init_completion(&aru->cmd_wait);
798         spin_lock_init(&aru->tx_urb_lock);
799
800         aru->tx_pending_urbs = 0;
801         atomic_set(&aru->tx_submitted_urbs, 0);
802
803         aru->common.stop = ar9170_usb_stop;
804         aru->common.flush = ar9170_usb_flush;
805         aru->common.open = ar9170_usb_open;
806         aru->common.tx = ar9170_usb_tx;
807         aru->common.exec_cmd = ar9170_usb_exec_cmd;
808         aru->common.callback_cmd = ar9170_usb_callback_cmd;
809
810 #ifdef CONFIG_PM
811         udev->reset_resume = 1;
812 #endif /* CONFIG_PM */
813         err = ar9170_usb_reset(aru);
814         if (err)
815                 goto err_freehw;
816
817         err = ar9170_usb_request_firmware(aru);
818         if (err)
819                 goto err_freehw;
820
821         err = ar9170_usb_init_device(aru);
822         if (err)
823                 goto err_freefw;
824
825         err = ar9170_usb_open(ar);
826         if (err)
827                 goto err_unrx;
828
829         err = ar9170_register(ar, &udev->dev);
830
831         ar9170_usb_stop(ar);
832         if (err)
833                 goto err_unrx;
834
835         return 0;
836
837 err_unrx:
838         ar9170_usb_cancel_urbs(aru);
839
840 err_freefw:
841         release_firmware(aru->init_values);
842         release_firmware(aru->firmware);
843
844 err_freehw:
845         usb_set_intfdata(intf, NULL);
846         usb_put_dev(udev);
847         ieee80211_free_hw(ar->hw);
848 out:
849         return err;
850 }
851
852 static void ar9170_usb_disconnect(struct usb_interface *intf)
853 {
854         struct ar9170_usb *aru = usb_get_intfdata(intf);
855
856         if (!aru)
857                 return;
858
859         aru->common.state = AR9170_IDLE;
860         ar9170_unregister(&aru->common);
861         ar9170_usb_cancel_urbs(aru);
862
863         release_firmware(aru->init_values);
864         release_firmware(aru->firmware);
865
866         usb_put_dev(aru->udev);
867         usb_set_intfdata(intf, NULL);
868         ieee80211_free_hw(aru->common.hw);
869 }
870
871 #ifdef CONFIG_PM
872 static int ar9170_suspend(struct usb_interface *intf,
873                           pm_message_t  message)
874 {
875         struct ar9170_usb *aru = usb_get_intfdata(intf);
876
877         if (!aru)
878                 return -ENODEV;
879
880         aru->common.state = AR9170_IDLE;
881         ar9170_usb_cancel_urbs(aru);
882
883         return 0;
884 }
885
886 static int ar9170_resume(struct usb_interface *intf)
887 {
888         struct ar9170_usb *aru = usb_get_intfdata(intf);
889         int err;
890
891         if (!aru)
892                 return -ENODEV;
893
894         usb_unpoison_anchored_urbs(&aru->rx_submitted);
895         usb_unpoison_anchored_urbs(&aru->tx_submitted);
896
897         err = ar9170_usb_init_device(aru);
898         if (err)
899                 goto err_unrx;
900
901         err = ar9170_usb_open(&aru->common);
902         if (err)
903                 goto err_unrx;
904
905         return 0;
906
907 err_unrx:
908         aru->common.state = AR9170_IDLE;
909         ar9170_usb_cancel_urbs(aru);
910
911         return err;
912 }
913 #endif /* CONFIG_PM */
914
915 static struct usb_driver ar9170_driver = {
916         .name = "ar9170usb",
917         .probe = ar9170_usb_probe,
918         .disconnect = ar9170_usb_disconnect,
919         .id_table = ar9170_usb_ids,
920         .soft_unbind = 1,
921 #ifdef CONFIG_PM
922         .suspend = ar9170_suspend,
923         .resume = ar9170_resume,
924         .reset_resume = ar9170_resume,
925 #endif /* CONFIG_PM */
926 };
927
928 static int __init ar9170_init(void)
929 {
930         return usb_register(&ar9170_driver);
931 }
932
933 static void __exit ar9170_exit(void)
934 {
935         usb_deregister(&ar9170_driver);
936 }
937
938 module_init(ar9170_init);
939 module_exit(ar9170_exit);