drivers/net/wireless/rt2x00/rt2x00usb.c

   1 /*
   2         Copyright (C) 2004 - 2009 Ivo van Doorn <IvDoorn@gmail.com>
   3         <http://rt2x00.serialmonkey.com>
   4
   5         This program is free software; you can redistribute it and/or modify
   6         it under the terms of the GNU General Public License as published by
   7         the Free Software Foundation; either version 2 of the License, or
   8         (at your option) any later version.
   9
  10         This program is distributed in the hope that it will be useful,
  11         but WITHOUT ANY WARRANTY; without even the implied warranty of
  12         MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13         GNU General Public License for more details.
  14
  15         You should have received a copy of the GNU General Public License
  16         along with this program; if not, write to the
  17         Free Software Foundation, Inc.,
  18         59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  19  */
  20
  21 /*
  22         Module: rt2x00usb
  23         Abstract: rt2x00 generic usb device routines.
  24  */
  25
  26 #include <linux/kernel.h>
  27 #include <linux/module.h>
  28 #include <linux/usb.h>
  29 #include <linux/bug.h>
  30
  31 #include "rt2x00.h"
  32 #include "rt2x00usb.h"
  33
  34 /*
  35  * Interfacing with the HW.
  36  */
  37 int rt2x00usb_vendor_request(struct rt2x00_dev *rt2x00dev,
  38                              const u8 request, const u8 requesttype,
  39                              const u16 offset, const u16 value,
  40                              void *buffer, const u16 buffer_length,
  41                              const int timeout)
  42 {
  43         struct usb_device *usb_dev = to_usb_device_intf(rt2x00dev->dev);
  44         int status;
  45         unsigned int i;
  46         unsigned int pipe =
  47             (requesttype == USB_VENDOR_REQUEST_IN) ?
  48             usb_rcvctrlpipe(usb_dev, 0) : usb_sndctrlpipe(usb_dev, 0);
  49
  50         if (!test_bit(DEVICE_STATE_PRESENT, &rt2x00dev->flags))
  51                 return -ENODEV;
  52
  53         for (i = 0; i < REGISTER_BUSY_COUNT; i++) {
  54                 status = usb_control_msg(usb_dev, pipe, request, requesttype,
  55                                          value, offset, buffer, buffer_length,
  56                                          timeout);
  57                 if (status >= 0)
  58                         return 0;
  59
  60                 /*
  61                  * Check for errors
  62                  * -ENODEV: Device has disappeared, no point continuing.
  63                  * All other errors: Try again.
  64                  */
  65                 else if (status == -ENODEV) {
  66                         clear_bit(DEVICE_STATE_PRESENT, &rt2x00dev->flags);
  67                         break;
  68                 }
  69         }
  70
  71         ERROR(rt2x00dev,
  72               "Vendor Request 0x%02x failed for offset 0x%04x with error %d.\n",
  73               request, offset, status);
  74
  75         return status;
  76 }
  77 EXPORT_SYMBOL_GPL(rt2x00usb_vendor_request);
  78
  79 int rt2x00usb_vendor_req_buff_lock(struct rt2x00_dev *rt2x00dev,
  80                                    const u8 request, const u8 requesttype,
  81                                    const u16 offset, void *buffer,
  82                                    const u16 buffer_length, const int timeout)
  83 {
  84         int status;
  85
  86         BUG_ON(!mutex_is_locked(&rt2x00dev->csr_mutex));
  87
  88         /*
  89          * Check for Cache availability.
  90          */
  91         if (unlikely(!rt2x00dev->csr.cache || buffer_length > CSR_CACHE_SIZE)) {
  92                 ERROR(rt2x00dev, "CSR cache not available.\n");
  93                 return -ENOMEM;
  94         }
  95
  96         if (requesttype == USB_VENDOR_REQUEST_OUT)
  97                 memcpy(rt2x00dev->csr.cache, buffer, buffer_length);
  98
  99         status = rt2x00usb_vendor_request(rt2x00dev, request, requesttype,
 100                                           offset, 0, rt2x00dev->csr.cache,
 101                                           buffer_length, timeout);
 102
 103         if (!status && requesttype == USB_VENDOR_REQUEST_IN)
 104                 memcpy(buffer, rt2x00dev->csr.cache, buffer_length);
 105
 106         return status;
 107 }
 108 EXPORT_SYMBOL_GPL(rt2x00usb_vendor_req_buff_lock);
 109
 110 int rt2x00usb_vendor_request_buff(struct rt2x00_dev *rt2x00dev,
 111                                   const u8 request, const u8 requesttype,
 112                                   const u16 offset, void *buffer,
 113                                   const u16 buffer_length, const int timeout)
 114 {
 115         int status;
 116
 117         mutex_lock(&rt2x00dev->csr_mutex);
 118
 119         status = rt2x00usb_vendor_req_buff_lock(rt2x00dev, request,
 120                                                 requesttype, offset, buffer,
 121                                                 buffer_length, timeout);
 122
 123         mutex_unlock(&rt2x00dev->csr_mutex);
 124
 125         return status;
 126 }
 127 EXPORT_SYMBOL_GPL(rt2x00usb_vendor_request_buff);
 128
 129 int rt2x00usb_vendor_request_large_buff(struct rt2x00_dev *rt2x00dev,
 130                                         const u8 request, const u8 requesttype,
 131                                         const u16 offset, const void *buffer,
 132                                         const u16 buffer_length,
 133                                         const int timeout)
 134 {
 135         int status = 0;
 136         unsigned char *tb;
 137         u16 off, len, bsize;
 138
 139         mutex_lock(&rt2x00dev->csr_mutex);
 140
 141         tb  = (char *)buffer;
 142         off = offset;
 143         len = buffer_length;
 144         while (len && !status) {
 145                 bsize = min_t(u16, CSR_CACHE_SIZE, len);
 146                 status = rt2x00usb_vendor_req_buff_lock(rt2x00dev, request,
 147                                                         requesttype, off, tb,
 148                                                         bsize, timeout);
 149
 150                 tb  += bsize;
 151                 len -= bsize;
 152                 off += bsize;
 153         }
 154
 155         mutex_unlock(&rt2x00dev->csr_mutex);
 156
 157         return status;
 158 }
 159 EXPORT_SYMBOL_GPL(rt2x00usb_vendor_request_large_buff);
 160
 161 int rt2x00usb_regbusy_read(struct rt2x00_dev *rt2x00dev,
 162                            const unsigned int offset,
 163                            const struct rt2x00_field32 field,
 164                            u32 *reg)
 165 {
 166         unsigned int i;
 167
 168         if (!test_bit(DEVICE_STATE_PRESENT, &rt2x00dev->flags))
 169                 return -ENODEV;
 170
 171         for (i = 0; i < REGISTER_BUSY_COUNT; i++) {
 172                 rt2x00usb_register_read_lock(rt2x00dev, offset, reg);
 173                 if (!rt2x00_get_field32(*reg, field))
 174                         return 1;
 175                 udelay(REGISTER_BUSY_DELAY);
 176         }
 177
 178         ERROR(rt2x00dev, "Indirect register access failed: "
 179               "offset=0x%.08x, value=0x%.08x\n", offset, *reg);
 180         *reg = ~0;
 181
 182         return 0;
 183 }
 184 EXPORT_SYMBOL_GPL(rt2x00usb_regbusy_read);
 185
 186 /*
 187  * TX data handlers.
 188  */
 189 static void rt2x00usb_interrupt_txdone(struct urb *urb)
 190 {
 191         struct queue_entry *entry = (struct queue_entry *)urb->context;
 192         struct rt2x00_dev *rt2x00dev = entry->queue->rt2x00dev;
 193         struct txdone_entry_desc txdesc;
 194
 195         if (!test_bit(DEVICE_STATE_ENABLED_RADIO, &rt2x00dev->flags) ||
 196             !test_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags))
 197                 return;
 198
 199         /*
 200          * Obtain the status about this packet.
 201          * Note that when the status is 0 it does not mean the
 202          * frame was send out correctly. It only means the frame
 203          * was succesfully pushed to the hardware, we have no
 204          * way to determine the transmission status right now.
 205          * (Only indirectly by looking at the failed TX counters
 206          * in the register).
 207          */
 208         txdesc.flags = 0;
 209         if (!urb->status)
 210                 __set_bit(TXDONE_UNKNOWN, &txdesc.flags);
 211         else
 212                 __set_bit(TXDONE_FAILURE, &txdesc.flags);
 213         txdesc.retry = 0;
 214
 215         rt2x00lib_txdone(entry, &txdesc);
 216 }
 217
 218 int rt2x00usb_write_tx_data(struct queue_entry *entry,
 219                             struct txentry_desc *txdesc)
 220 {
 221         struct rt2x00_dev *rt2x00dev = entry->queue->rt2x00dev;
 222         struct usb_device *usb_dev = to_usb_device_intf(rt2x00dev->dev);
 223         struct queue_entry_priv_usb *entry_priv = entry->priv_data;
 224         struct skb_frame_desc *skbdesc;
 225         u32 length;
 226
 227         /*
 228          * Add the descriptor in front of the skb.
 229          */
 230         skb_push(entry->skb, entry->queue->desc_size);
 231         memset(entry->skb->data, 0, entry->queue->desc_size);
 232
 233         /*
 234          * Fill in skb descriptor
 235          */
 236         skbdesc = get_skb_frame_desc(entry->skb);
 237         skbdesc->desc = entry->skb->data;
 238         skbdesc->desc_len = entry->queue->desc_size;
 239
 240         /*
 241          * USB devices cannot blindly pass the skb->len as the
 242          * length of the data to usb_fill_bulk_urb. Pass the skb
 243          * to the driver to determine what the length should be.
 244          */
 245         length = rt2x00dev->ops->lib->get_tx_data_len(entry);
 246
 247         usb_fill_bulk_urb(entry_priv->urb, usb_dev,
 248                           usb_sndbulkpipe(usb_dev, entry->queue->usb_endpoint),
 249                           entry->skb->data, length,
 250                           rt2x00usb_interrupt_txdone, entry);
 251
 252         /*
 253          * Make sure the skb->data pointer points to the frame, not the
 254          * descriptor.
 255          */
 256         skb_pull(entry->skb, entry->queue->desc_size);
 257
 258         return 0;
 259 }
 260 EXPORT_SYMBOL_GPL(rt2x00usb_write_tx_data);
 261
 262 static inline void rt2x00usb_kick_tx_entry(struct queue_entry *entry)
 263 {
 264         struct queue_entry_priv_usb *entry_priv = entry->priv_data;
 265
 266         if (test_and_clear_bit(ENTRY_DATA_PENDING, &entry->flags))
 267                 usb_submit_urb(entry_priv->urb, GFP_ATOMIC);
 268 }
 269
 270 void rt2x00usb_kick_tx_queue(struct rt2x00_dev *rt2x00dev,
 271                              const enum data_queue_qid qid)
 272 {
 273         struct data_queue *queue = rt2x00queue_get_queue(rt2x00dev, qid);
 274         unsigned long irqflags;
 275         unsigned int index;
 276         unsigned int index_done;
 277         unsigned int i;
 278
 279         /*
 280          * Only protect the range we are going to loop over,
 281          * if during our loop a extra entry is set to pending
 282          * it should not be kicked during this run, since it
 283          * is part of another TX operation.
 284          */
 285         spin_lock_irqsave(&queue->lock, irqflags);
 286         index = queue->index[Q_INDEX];
 287         index_done = queue->index[Q_INDEX_DONE];
 288         spin_unlock_irqrestore(&queue->lock, irqflags);
 289
 290         /*
 291          * Start from the TX done pointer, this guarentees that we will
 292          * send out all frames in the correct order.
 293          */
 294         if (index_done < index) {
 295                 for (i = index_done; i < index; i++)
 296                         rt2x00usb_kick_tx_entry(&queue->entries[i]);
 297         } else {
 298                 for (i = index_done; i < queue->limit; i++)
 299                         rt2x00usb_kick_tx_entry(&queue->entries[i]);
 300
 301                 for (i = 0; i < index; i++)
 302                         rt2x00usb_kick_tx_entry(&queue->entries[i]);
 303         }
 304 }
 305 EXPORT_SYMBOL_GPL(rt2x00usb_kick_tx_queue);
 306
 307 void rt2x00usb_kill_tx_queue(struct rt2x00_dev *rt2x00dev,
 308                              const enum data_queue_qid qid)
 309 {
 310         struct data_queue *queue = rt2x00queue_get_queue(rt2x00dev, qid);
 311         struct queue_entry_priv_usb *entry_priv;
 312         struct queue_entry_priv_usb_bcn *bcn_priv;
 313         unsigned int i;
 314         bool kill_guard;
 315
 316         /*
 317          * When killing the beacon queue, we must also kill
 318          * the beacon guard byte.
 319          */
 320         kill_guard =
 321             (qid == QID_BEACON) &&
 322             (test_bit(DRIVER_REQUIRE_BEACON_GUARD, &rt2x00dev->flags));
 323
 324         /*
 325          * Cancel all entries.
 326          */
 327         for (i = 0; i < queue->limit; i++) {
 328                 entry_priv = queue->entries[i].priv_data;
 329                 usb_kill_urb(entry_priv->urb);
 330
 331                 /*
 332                  * Kill guardian urb (if required by driver).
 333                  */
 334                 if (kill_guard) {
 335                         bcn_priv = queue->entries[i].priv_data;
 336                         usb_kill_urb(bcn_priv->guardian_urb);
 337                 }
 338         }
 339 }
 340 EXPORT_SYMBOL_GPL(rt2x00usb_kill_tx_queue);
 341
 342 /*
 343  * RX data handlers.
 344  */
 345 static void rt2x00usb_interrupt_rxdone(struct urb *urb)
 346 {
 347         struct queue_entry *entry = (struct queue_entry *)urb->context;
 348         struct rt2x00_dev *rt2x00dev = entry->queue->rt2x00dev;
 349         struct skb_frame_desc *skbdesc = get_skb_frame_desc(entry->skb);
 350         u8 rxd[32];
 351
 352         if (!test_bit(DEVICE_STATE_ENABLED_RADIO, &rt2x00dev->flags) ||
 353             !test_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags))
 354                 return;
 355
 356         /*
 357          * Check if the received data is simply too small
 358          * to be actually valid, or if the urb is signaling
 359          * a problem.
 360          */
 361         if (urb->actual_length < entry->queue->desc_size || urb->status) {
 362                 set_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags);
 363                 usb_submit_urb(urb, GFP_ATOMIC);
 364                 return;
 365         }
 366
 367         /*
 368          * Fill in desc fields of the skb descriptor
 369          */
 370         skbdesc->desc = rxd;
 371         skbdesc->desc_len = entry->queue->desc_size;
 372
 373         /*
 374          * Send the frame to rt2x00lib for further processing.
 375          */
 376         rt2x00lib_rxdone(rt2x00dev, entry);
 377 }
 378
 379 /*
 380  * Radio handlers
 381  */
 382 void rt2x00usb_disable_radio(struct rt2x00_dev *rt2x00dev)
 383 {
 384         rt2x00usb_vendor_request_sw(rt2x00dev, USB_RX_CONTROL, 0, 0,
 385                                     REGISTER_TIMEOUT);
 386
 387         /*
 388          * The USB version of kill_tx_queue also works
 389          * on the RX queue.
 390          */
 391         rt2x00dev->ops->lib->kill_tx_queue(rt2x00dev, QID_RX);
 392 }
 393 EXPORT_SYMBOL_GPL(rt2x00usb_disable_radio);
 394
 395 /*
 396  * Device initialization handlers.
 397  */
 398 void rt2x00usb_clear_entry(struct queue_entry *entry)
 399 {
 400         struct usb_device *usb_dev =
 401             to_usb_device_intf(entry->queue->rt2x00dev->dev);
 402         struct queue_entry_priv_usb *entry_priv = entry->priv_data;
 403         int pipe;
 404
 405         if (entry->queue->qid == QID_RX) {
 406                 pipe = usb_rcvbulkpipe(usb_dev, entry->queue->usb_endpoint);
 407                 usb_fill_bulk_urb(entry_priv->urb, usb_dev, pipe,
 408                                 entry->skb->data, entry->skb->len,
 409                                 rt2x00usb_interrupt_rxdone, entry);
 410
 411                 set_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags);
 412                 usb_submit_urb(entry_priv->urb, GFP_ATOMIC);
 413         } else {
 414                 entry->flags = 0;
 415         }
 416 }
 417 EXPORT_SYMBOL_GPL(rt2x00usb_clear_entry);
 418
 419 static void rt2x00usb_assign_endpoint(struct data_queue *queue,
 420                                       struct usb_endpoint_descriptor *ep_desc)
 421 {
 422         struct usb_device *usb_dev = to_usb_device_intf(queue->rt2x00dev->dev);
 423         int pipe;
 424
 425         queue->usb_endpoint = usb_endpoint_num(ep_desc);
 426
 427         if (queue->qid == QID_RX) {
 428                 pipe = usb_rcvbulkpipe(usb_dev, queue->usb_endpoint);
 429                 queue->usb_maxpacket = usb_maxpacket(usb_dev, pipe, 0);
 430         } else {
 431                 pipe = usb_sndbulkpipe(usb_dev, queue->usb_endpoint);
 432                 queue->usb_maxpacket = usb_maxpacket(usb_dev, pipe, 1);
 433         }
 434
 435         if (!queue->usb_maxpacket)
 436                 queue->usb_maxpacket = 1;
 437 }
 438
 439 static int rt2x00usb_find_endpoints(struct rt2x00_dev *rt2x00dev)
 440 {
 441         struct usb_interface *intf = to_usb_interface(rt2x00dev->dev);
 442         struct usb_host_interface *intf_desc = intf->cur_altsetting;
 443         struct usb_endpoint_descriptor *ep_desc;
 444         struct data_queue *queue = rt2x00dev->tx;
 445         struct usb_endpoint_descriptor *tx_ep_desc = NULL;
 446         unsigned int i;
 447
 448         /*
 449          * Walk through all available endpoints to search for "bulk in"
 450          * and "bulk out" endpoints. When we find such endpoints collect
 451          * the information we need from the descriptor and assign it
 452          * to the queue.
 453          */
 454         for (i = 0; i < intf_desc->desc.bNumEndpoints; i++) {
 455                 ep_desc = &intf_desc->endpoint[i].desc;
 456
 457                 if (usb_endpoint_is_bulk_in(ep_desc)) {
 458                         rt2x00usb_assign_endpoint(rt2x00dev->rx, ep_desc);
 459                 } else if (usb_endpoint_is_bulk_out(ep_desc) &&
 460                            (queue != queue_end(rt2x00dev))) {
 461                         rt2x00usb_assign_endpoint(queue, ep_desc);
 462                         queue = queue_next(queue);
 463
 464                         tx_ep_desc = ep_desc;
 465                 }
 466         }
 467
 468         /*
 469          * At least 1 endpoint for RX and 1 endpoint for TX must be available.
 470          */
 471         if (!rt2x00dev->rx->usb_endpoint || !rt2x00dev->tx->usb_endpoint) {
 472                 ERROR(rt2x00dev, "Bulk-in/Bulk-out endpoints not found\n");
 473                 return -EPIPE;
 474         }
 475
 476         /*
 477          * It might be possible not all queues have a dedicated endpoint.
 478          * Loop through all TX queues and copy the endpoint information
 479          * which we have gathered from already assigned endpoints.
 480          */
 481         txall_queue_for_each(rt2x00dev, queue) {
 482                 if (!queue->usb_endpoint)
 483                         rt2x00usb_assign_endpoint(queue, tx_ep_desc);
 484         }
 485
 486         return 0;
 487 }
 488
 489 static int rt2x00usb_alloc_urb(struct rt2x00_dev *rt2x00dev,
 490                                struct data_queue *queue)
 491 {
 492         struct queue_entry_priv_usb *entry_priv;
 493         struct queue_entry_priv_usb_bcn *bcn_priv;
 494         unsigned int i;
 495
 496         for (i = 0; i < queue->limit; i++) {
 497                 entry_priv = queue->entries[i].priv_data;
 498                 entry_priv->urb = usb_alloc_urb(0, GFP_KERNEL);
 499                 if (!entry_priv->urb)
 500                         return -ENOMEM;
 501         }
 502
 503         /*
 504          * If this is not the beacon queue or
 505          * no guardian byte was required for the beacon,
 506          * then we are done.
 507          */
 508         if (rt2x00dev->bcn != queue ||
 509             !test_bit(DRIVER_REQUIRE_BEACON_GUARD, &rt2x00dev->flags))
 510                 return 0;
 511
 512         for (i = 0; i < queue->limit; i++) {
 513                 bcn_priv = queue->entries[i].priv_data;
 514                 bcn_priv->guardian_urb = usb_alloc_urb(0, GFP_KERNEL);
 515                 if (!bcn_priv->guardian_urb)
 516                         return -ENOMEM;
 517         }
 518
 519         return 0;
 520 }
 521
 522 static void rt2x00usb_free_urb(struct rt2x00_dev *rt2x00dev,
 523                                struct data_queue *queue)
 524 {
 525         struct queue_entry_priv_usb *entry_priv;
 526         struct queue_entry_priv_usb_bcn *bcn_priv;
 527         unsigned int i;
 528
 529         if (!queue->entries)
 530                 return;
 531
 532         for (i = 0; i < queue->limit; i++) {
 533                 entry_priv = queue->entries[i].priv_data;
 534                 usb_kill_urb(entry_priv->urb);
 535                 usb_free_urb(entry_priv->urb);
 536         }
 537
 538         /*
 539          * If this is not the beacon queue or
 540          * no guardian byte was required for the beacon,
 541          * then we are done.
 542          */
 543         if (rt2x00dev->bcn != queue ||
 544             !test_bit(DRIVER_REQUIRE_BEACON_GUARD, &rt2x00dev->flags))
 545                 return;
 546
 547         for (i = 0; i < queue->limit; i++) {
 548                 bcn_priv = queue->entries[i].priv_data;
 549                 usb_kill_urb(bcn_priv->guardian_urb);
 550                 usb_free_urb(bcn_priv->guardian_urb);
 551         }
 552 }
 553
 554 int rt2x00usb_initialize(struct rt2x00_dev *rt2x00dev)
 555 {
 556         struct data_queue *queue;
 557         int status;
 558
 559         /*
 560          * Find endpoints for each queue
 561          */
 562         status = rt2x00usb_find_endpoints(rt2x00dev);
 563         if (status)
 564                 goto exit;
 565
 566         /*
 567          * Allocate DMA
 568          */
 569         queue_for_each(rt2x00dev, queue) {
 570                 status = rt2x00usb_alloc_urb(rt2x00dev, queue);
 571                 if (status)
 572                         goto exit;
 573         }
 574
 575         return 0;
 576
 577 exit:
 578         rt2x00usb_uninitialize(rt2x00dev);
 579
 580         return status;
 581 }
 582 EXPORT_SYMBOL_GPL(rt2x00usb_initialize);
 583
 584 void rt2x00usb_uninitialize(struct rt2x00_dev *rt2x00dev)
 585 {
 586         struct data_queue *queue;
 587
 588         queue_for_each(rt2x00dev, queue)
 589                 rt2x00usb_free_urb(rt2x00dev, queue);
 590 }
 591 EXPORT_SYMBOL_GPL(rt2x00usb_uninitialize);
 592
 593 /*
 594  * USB driver handlers.
 595  */
 596 static void rt2x00usb_free_reg(struct rt2x00_dev *rt2x00dev)
 597 {
 598         kfree(rt2x00dev->rf);
 599         rt2x00dev->rf = NULL;
 600
 601         kfree(rt2x00dev->eeprom);
 602         rt2x00dev->eeprom = NULL;
 603
 604         kfree(rt2x00dev->csr.cache);
 605         rt2x00dev->csr.cache = NULL;
 606 }
 607
 608 static int rt2x00usb_alloc_reg(struct rt2x00_dev *rt2x00dev)
 609 {
 610         rt2x00dev->csr.cache = kzalloc(CSR_CACHE_SIZE, GFP_KERNEL);
 611         if (!rt2x00dev->csr.cache)
 612                 goto exit;
 613
 614         rt2x00dev->eeprom = kzalloc(rt2x00dev->ops->eeprom_size, GFP_KERNEL);
 615         if (!rt2x00dev->eeprom)
 616                 goto exit;
 617
 618         rt2x00dev->rf = kzalloc(rt2x00dev->ops->rf_size, GFP_KERNEL);
 619         if (!rt2x00dev->rf)
 620                 goto exit;
 621
 622         return 0;
 623
 624 exit:
 625         ERROR_PROBE("Failed to allocate registers.\n");
 626
 627         rt2x00usb_free_reg(rt2x00dev);
 628
 629         return -ENOMEM;
 630 }
 631
 632 int rt2x00usb_probe(struct usb_interface *usb_intf,
 633                     const struct usb_device_id *id)
 634 {
 635         struct usb_device *usb_dev = interface_to_usbdev(usb_intf);
 636         struct rt2x00_ops *ops = (struct rt2x00_ops *)id->driver_info;
 637         struct ieee80211_hw *hw;
 638         struct rt2x00_dev *rt2x00dev;
 639         int retval;
 640
 641         usb_dev = usb_get_dev(usb_dev);
 642
 643         hw = ieee80211_alloc_hw(sizeof(struct rt2x00_dev), ops->hw);
 644         if (!hw) {
 645                 ERROR_PROBE("Failed to allocate hardware.\n");
 646                 retval = -ENOMEM;
 647                 goto exit_put_device;
 648         }
 649
 650         usb_set_intfdata(usb_intf, hw);
 651
 652         rt2x00dev = hw->priv;
 653         rt2x00dev->dev = &usb_intf->dev;
 654         rt2x00dev->ops = ops;
 655         rt2x00dev->hw = hw;
 656
 657         rt2x00_set_chip_intf(rt2x00dev, RT2X00_CHIP_INTF_USB);
 658
 659         retval = rt2x00usb_alloc_reg(rt2x00dev);
 660         if (retval)
 661                 goto exit_free_device;
 662
 663         retval = rt2x00lib_probe_dev(rt2x00dev);
 664         if (retval)
 665                 goto exit_free_reg;
 666
 667         return 0;
 668
 669 exit_free_reg:
 670         rt2x00usb_free_reg(rt2x00dev);
 671
 672 exit_free_device:
 673         ieee80211_free_hw(hw);
 674
 675 exit_put_device:
 676         usb_put_dev(usb_dev);
 677
 678         usb_set_intfdata(usb_intf, NULL);
 679
 680         return retval;
 681 }
 682 EXPORT_SYMBOL_GPL(rt2x00usb_probe);
 683
 684 void rt2x00usb_disconnect(struct usb_interface *usb_intf)
 685 {
 686         struct ieee80211_hw *hw = usb_get_intfdata(usb_intf);
 687         struct rt2x00_dev *rt2x00dev = hw->priv;
 688
 689         /*
 690          * Free all allocated data.
 691          */
 692         rt2x00lib_remove_dev(rt2x00dev);
 693         rt2x00usb_free_reg(rt2x00dev);
 694         ieee80211_free_hw(hw);
 695
 696         /*
 697          * Free the USB device data.
 698          */
 699         usb_set_intfdata(usb_intf, NULL);
 700         usb_put_dev(interface_to_usbdev(usb_intf));
 701 }
 702 EXPORT_SYMBOL_GPL(rt2x00usb_disconnect);
 703
 704 #ifdef CONFIG_PM
 705 int rt2x00usb_suspend(struct usb_interface *usb_intf, pm_message_t state)
 706 {
 707         struct ieee80211_hw *hw = usb_get_intfdata(usb_intf);
 708         struct rt2x00_dev *rt2x00dev = hw->priv;
 709         int retval;
 710
 711         retval = rt2x00lib_suspend(rt2x00dev, state);
 712         if (retval)
 713                 return retval;
 714
 715         /*
 716          * Decrease usbdev refcount.
 717          */
 718         usb_put_dev(interface_to_usbdev(usb_intf));
 719
 720         return 0;
 721 }
 722 EXPORT_SYMBOL_GPL(rt2x00usb_suspend);
 723
 724 int rt2x00usb_resume(struct usb_interface *usb_intf)
 725 {
 726         struct ieee80211_hw *hw = usb_get_intfdata(usb_intf);
 727         struct rt2x00_dev *rt2x00dev = hw->priv;
 728
 729         usb_get_dev(interface_to_usbdev(usb_intf));
 730
 731         return rt2x00lib_resume(rt2x00dev);
 732 }
 733 EXPORT_SYMBOL_GPL(rt2x00usb_resume);
 734 #endif /* CONFIG_PM */
 735
 736 /*
 737  * rt2x00usb module information.
 738  */
 739 MODULE_AUTHOR(DRV_PROJECT);
 740 MODULE_VERSION(DRV_VERSION);
 741 MODULE_DESCRIPTION("rt2x00 usb library");
 742 MODULE_LICENSE("GPL");