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