drivers/net/wireless/rt2x00/rt2x00dev.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: rt2x00lib
  23         Abstract: rt2x00 generic device routines.
  24  */
  25
  26 #include <linux/kernel.h>
  27 #include <linux/module.h>
  28
  29 #include "rt2x00.h"
  30 #include "rt2x00lib.h"
  31
  32 /*
  33  * Radio control handlers.
  34  */
  35 int rt2x00lib_enable_radio(struct rt2x00_dev *rt2x00dev)
  36 {
  37         int status;
  38
  39         /*
  40          * Don't enable the radio twice.
  41          * And check if the hardware button has been disabled.
  42          */
  43         if (test_bit(DEVICE_STATE_ENABLED_RADIO, &rt2x00dev->flags))
  44                 return 0;
  45
  46         /*
  47          * Initialize all data queues.
  48          */
  49         rt2x00queue_init_queues(rt2x00dev);
  50
  51         /*
  52          * Enable radio.
  53          */
  54         status =
  55             rt2x00dev->ops->lib->set_device_state(rt2x00dev, STATE_RADIO_ON);
  56         if (status)
  57                 return status;
  58
  59         rt2x00dev->ops->lib->set_device_state(rt2x00dev, STATE_RADIO_IRQ_ON);
  60
  61         rt2x00leds_led_radio(rt2x00dev, true);
  62         rt2x00led_led_activity(rt2x00dev, true);
  63
  64         set_bit(DEVICE_STATE_ENABLED_RADIO, &rt2x00dev->flags);
  65
  66         /*
  67          * Enable RX.
  68          */
  69         rt2x00lib_toggle_rx(rt2x00dev, STATE_RADIO_RX_ON);
  70
  71         /*
  72          * Start the TX queues.
  73          */
  74         ieee80211_wake_queues(rt2x00dev->hw);
  75
  76         return 0;
  77 }
  78
  79 void rt2x00lib_disable_radio(struct rt2x00_dev *rt2x00dev)
  80 {
  81         if (!test_and_clear_bit(DEVICE_STATE_ENABLED_RADIO, &rt2x00dev->flags))
  82                 return;
  83
  84         /*
  85          * Stop the TX queues in mac80211.
  86          */
  87         ieee80211_stop_queues(rt2x00dev->hw);
  88         rt2x00queue_stop_queues(rt2x00dev);
  89
  90         /*
  91          * Disable RX.
  92          */
  93         rt2x00lib_toggle_rx(rt2x00dev, STATE_RADIO_RX_OFF);
  94
  95         /*
  96          * Disable radio.
  97          */
  98         rt2x00dev->ops->lib->set_device_state(rt2x00dev, STATE_RADIO_OFF);
  99         rt2x00dev->ops->lib->set_device_state(rt2x00dev, STATE_RADIO_IRQ_OFF);
 100         rt2x00led_led_activity(rt2x00dev, false);
 101         rt2x00leds_led_radio(rt2x00dev, false);
 102 }
 103
 104 void rt2x00lib_toggle_rx(struct rt2x00_dev *rt2x00dev, enum dev_state state)
 105 {
 106         /*
 107          * When we are disabling the RX, we should also stop the link tuner.
 108          */
 109         if (state == STATE_RADIO_RX_OFF)
 110                 rt2x00link_stop_tuner(rt2x00dev);
 111
 112         rt2x00dev->ops->lib->set_device_state(rt2x00dev, state);
 113
 114         /*
 115          * When we are enabling the RX, we should also start the link tuner.
 116          */
 117         if (state == STATE_RADIO_RX_ON)
 118                 rt2x00link_start_tuner(rt2x00dev);
 119 }
 120
 121 static void rt2x00lib_intf_scheduled_iter(void *data, u8 *mac,
 122                                           struct ieee80211_vif *vif)
 123 {
 124         struct rt2x00_dev *rt2x00dev = data;
 125         struct rt2x00_intf *intf = vif_to_intf(vif);
 126         int delayed_flags;
 127
 128         /*
 129          * Copy all data we need during this action under the protection
 130          * of a spinlock. Otherwise race conditions might occur which results
 131          * into an invalid configuration.
 132          */
 133         spin_lock(&intf->lock);
 134
 135         delayed_flags = intf->delayed_flags;
 136         intf->delayed_flags = 0;
 137
 138         spin_unlock(&intf->lock);
 139
 140         /*
 141          * It is possible the radio was disabled while the work had been
 142          * scheduled. If that happens we should return here immediately,
 143          * note that in the spinlock protected area above the delayed_flags
 144          * have been cleared correctly.
 145          */
 146         if (!test_bit(DEVICE_STATE_ENABLED_RADIO, &rt2x00dev->flags))
 147                 return;
 148
 149         if (delayed_flags & DELAYED_UPDATE_BEACON)
 150                 rt2x00queue_update_beacon(rt2x00dev, vif, true);
 151 }
 152
 153 static void rt2x00lib_intf_scheduled(struct work_struct *work)
 154 {
 155         struct rt2x00_dev *rt2x00dev =
 156             container_of(work, struct rt2x00_dev, intf_work);
 157
 158         /*
 159          * Iterate over each interface and perform the
 160          * requested configurations.
 161          */
 162         ieee80211_iterate_active_interfaces(rt2x00dev->hw,
 163                                             rt2x00lib_intf_scheduled_iter,
 164                                             rt2x00dev);
 165 }
 166
 167 /*
 168  * Interrupt context handlers.
 169  */
 170 static void rt2x00lib_beacondone_iter(void *data, u8 *mac,
 171                                       struct ieee80211_vif *vif)
 172 {
 173         struct rt2x00_intf *intf = vif_to_intf(vif);
 174
 175         if (vif->type != NL80211_IFTYPE_AP &&
 176             vif->type != NL80211_IFTYPE_ADHOC &&
 177             vif->type != NL80211_IFTYPE_MESH_POINT &&
 178             vif->type != NL80211_IFTYPE_WDS)
 179                 return;
 180
 181         spin_lock(&intf->lock);
 182         intf->delayed_flags |= DELAYED_UPDATE_BEACON;
 183         spin_unlock(&intf->lock);
 184 }
 185
 186 void rt2x00lib_beacondone(struct rt2x00_dev *rt2x00dev)
 187 {
 188         if (!test_bit(DEVICE_STATE_ENABLED_RADIO, &rt2x00dev->flags))
 189                 return;
 190
 191         ieee80211_iterate_active_interfaces_atomic(rt2x00dev->hw,
 192                                                    rt2x00lib_beacondone_iter,
 193                                                    rt2x00dev);
 194
 195         ieee80211_queue_work(rt2x00dev->hw, &rt2x00dev->intf_work);
 196 }
 197 EXPORT_SYMBOL_GPL(rt2x00lib_beacondone);
 198
 199 void rt2x00lib_txdone(struct queue_entry *entry,
 200                       struct txdone_entry_desc *txdesc)
 201 {
 202         struct rt2x00_dev *rt2x00dev = entry->queue->rt2x00dev;
 203         struct ieee80211_tx_info *tx_info = IEEE80211_SKB_CB(entry->skb);
 204         struct skb_frame_desc *skbdesc = get_skb_frame_desc(entry->skb);
 205         enum data_queue_qid qid = skb_get_queue_mapping(entry->skb);
 206         unsigned int header_length = ieee80211_get_hdrlen_from_skb(entry->skb);
 207         u8 rate_idx, rate_flags, retry_rates;
 208         u8 skbdesc_flags = skbdesc->flags;
 209         unsigned int i;
 210         bool success;
 211
 212         /*
 213          * Unmap the skb.
 214          */
 215         rt2x00queue_unmap_skb(rt2x00dev, entry->skb);
 216
 217         /*
 218          * Remove L2 padding which was added during
 219          */
 220         if (test_bit(DRIVER_REQUIRE_L2PAD, &rt2x00dev->flags))
 221                 rt2x00queue_remove_l2pad(entry->skb, header_length);
 222
 223         /*
 224          * If the IV/EIV data was stripped from the frame before it was
 225          * passed to the hardware, we should now reinsert it again because
 226          * mac80211 will expect the the same data to be present it the
 227          * frame as it was passed to us.
 228          */
 229         if (test_bit(CONFIG_SUPPORT_HW_CRYPTO, &rt2x00dev->flags))
 230                 rt2x00crypto_tx_insert_iv(entry->skb, header_length);
 231
 232         /*
 233          * Send frame to debugfs immediately, after this call is completed
 234          * we are going to overwrite the skb->cb array.
 235          */
 236         rt2x00debug_dump_frame(rt2x00dev, DUMP_FRAME_TXDONE, entry->skb);
 237
 238         /*
 239          * Determine if the frame has been successfully transmitted.
 240          */
 241         success =
 242             test_bit(TXDONE_SUCCESS, &txdesc->flags) ||
 243             test_bit(TXDONE_UNKNOWN, &txdesc->flags) ||
 244             test_bit(TXDONE_FALLBACK, &txdesc->flags);
 245
 246         /*
 247          * Update TX statistics.
 248          */
 249         rt2x00dev->link.qual.tx_success += success;
 250         rt2x00dev->link.qual.tx_failed += !success;
 251
 252         rate_idx = skbdesc->tx_rate_idx;
 253         rate_flags = skbdesc->tx_rate_flags;
 254         retry_rates = test_bit(TXDONE_FALLBACK, &txdesc->flags) ?
 255             (txdesc->retry + 1) : 1;
 256
 257         /*
 258          * Initialize TX status
 259          */
 260         memset(&tx_info->status, 0, sizeof(tx_info->status));
 261         tx_info->status.ack_signal = 0;
 262
 263         /*
 264          * Frame was send with retries, hardware tried
 265          * different rates to send out the frame, at each
 266          * retry it lowered the rate 1 step.
 267          */
 268         for (i = 0; i < retry_rates && i < IEEE80211_TX_MAX_RATES; i++) {
 269                 tx_info->status.rates[i].idx = rate_idx - i;
 270                 tx_info->status.rates[i].flags = rate_flags;
 271                 tx_info->status.rates[i].count = 1;
 272         }
 273         if (i < (IEEE80211_TX_MAX_RATES - 1))
 274                 tx_info->status.rates[i].idx = -1; /* terminate */
 275
 276         if (!(tx_info->flags & IEEE80211_TX_CTL_NO_ACK)) {
 277                 if (success)
 278                         tx_info->flags |= IEEE80211_TX_STAT_ACK;
 279                 else
 280                         rt2x00dev->low_level_stats.dot11ACKFailureCount++;
 281         }
 282
 283         if (rate_flags & IEEE80211_TX_RC_USE_RTS_CTS) {
 284                 if (success)
 285                         rt2x00dev->low_level_stats.dot11RTSSuccessCount++;
 286                 else
 287                         rt2x00dev->low_level_stats.dot11RTSFailureCount++;
 288         }
 289
 290         /*
 291          * Only send the status report to mac80211 when it's a frame
 292          * that originated in mac80211. If this was a extra frame coming
 293          * through a mac80211 library call (RTS/CTS) then we should not
 294          * send the status report back.
 295          */
 296         if (!(skbdesc_flags & SKBDESC_NOT_MAC80211))
 297                 ieee80211_tx_status_irqsafe(rt2x00dev->hw, entry->skb);
 298         else
 299                 dev_kfree_skb_irq(entry->skb);
 300
 301         /*
 302          * Make this entry available for reuse.
 303          */
 304         entry->skb = NULL;
 305         entry->flags = 0;
 306
 307         rt2x00dev->ops->lib->clear_entry(entry);
 308
 309         clear_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags);
 310         rt2x00queue_index_inc(entry->queue, Q_INDEX_DONE);
 311
 312         /*
 313          * If the data queue was below the threshold before the txdone
 314          * handler we must make sure the packet queue in the mac80211 stack
 315          * is reenabled when the txdone handler has finished.
 316          */
 317         if (!rt2x00queue_threshold(entry->queue))
 318                 ieee80211_wake_queue(rt2x00dev->hw, qid);
 319 }
 320 EXPORT_SYMBOL_GPL(rt2x00lib_txdone);
 321
 322 static int rt2x00lib_rxdone_read_signal(struct rt2x00_dev *rt2x00dev,
 323                                         struct rxdone_entry_desc *rxdesc)
 324 {
 325         struct ieee80211_supported_band *sband;
 326         const struct rt2x00_rate *rate;
 327         unsigned int i;
 328         int signal;
 329         int type;
 330
 331         /*
 332          * For non-HT rates the MCS value needs to contain the
 333          * actually used rate modulation (CCK or OFDM).
 334          */
 335         if (rxdesc->dev_flags & RXDONE_SIGNAL_MCS)
 336                 signal = RATE_MCS(rxdesc->rate_mode, rxdesc->signal);
 337         else
 338                 signal = rxdesc->signal;
 339
 340         type = (rxdesc->dev_flags & RXDONE_SIGNAL_MASK);
 341
 342         sband = &rt2x00dev->bands[rt2x00dev->curr_band];
 343         for (i = 0; i < sband->n_bitrates; i++) {
 344                 rate = rt2x00_get_rate(sband->bitrates[i].hw_value);
 345
 346                 if (((type == RXDONE_SIGNAL_PLCP) &&
 347                      (rate->plcp == signal)) ||
 348                     ((type == RXDONE_SIGNAL_BITRATE) &&
 349                       (rate->bitrate == signal)) ||
 350                     ((type == RXDONE_SIGNAL_MCS) &&
 351                       (rate->mcs == signal))) {
 352                         return i;
 353                 }
 354         }
 355
 356         WARNING(rt2x00dev, "Frame received with unrecognized signal, "
 357                 "signal=0x%.4x, type=%d.\n", signal, type);
 358         return 0;
 359 }
 360
 361 void rt2x00lib_rxdone(struct rt2x00_dev *rt2x00dev,
 362                       struct queue_entry *entry)
 363 {
 364         struct rxdone_entry_desc rxdesc;
 365         struct sk_buff *skb;
 366         struct ieee80211_rx_status *rx_status = &rt2x00dev->rx_status;
 367         unsigned int header_length;
 368         int rate_idx;
 369         /*
 370          * Allocate a new sk_buffer. If no new buffer available, drop the
 371          * received frame and reuse the existing buffer.
 372          */
 373         skb = rt2x00queue_alloc_rxskb(rt2x00dev, entry);
 374         if (!skb)
 375                 return;
 376
 377         /*
 378          * Unmap the skb.
 379          */
 380         rt2x00queue_unmap_skb(rt2x00dev, entry->skb);
 381
 382         /*
 383          * Extract the RXD details.
 384          */
 385         memset(&rxdesc, 0, sizeof(rxdesc));
 386         rt2x00dev->ops->lib->fill_rxdone(entry, &rxdesc);
 387
 388         /*
 389          * The data behind the ieee80211 header must be
 390          * aligned on a 4 byte boundary.
 391          */
 392         header_length = ieee80211_get_hdrlen_from_skb(entry->skb);
 393
 394         /*
 395          * Hardware might have stripped the IV/EIV/ICV data,
 396          * in that case it is possible that the data was
 397          * provided separately (through hardware descriptor)
 398          * in which case we should reinsert the data into the frame.
 399          */
 400         if ((rxdesc.dev_flags & RXDONE_CRYPTO_IV) &&
 401             (rxdesc.flags & RX_FLAG_IV_STRIPPED))
 402                 rt2x00crypto_rx_insert_iv(entry->skb, header_length,
 403                                           &rxdesc);
 404         else if (header_length &&
 405                  (rxdesc.size > header_length) &&
 406                  (rxdesc.dev_flags & RXDONE_L2PAD))
 407                 rt2x00queue_remove_l2pad(entry->skb, header_length);
 408         else
 409                 rt2x00queue_align_payload(entry->skb, header_length);
 410
 411         /* Trim buffer to correct size */
 412         skb_trim(entry->skb, rxdesc.size);
 413
 414         /*
 415          * Check if the frame was received using HT. In that case,
 416          * the rate is the MCS index and should be passed to mac80211
 417          * directly. Otherwise we need to translate the signal to
 418          * the correct bitrate index.
 419          */
 420         if (rxdesc.rate_mode == RATE_MODE_CCK ||
 421             rxdesc.rate_mode == RATE_MODE_OFDM) {
 422                 rate_idx = rt2x00lib_rxdone_read_signal(rt2x00dev, &rxdesc);
 423         } else {
 424                 rxdesc.flags |= RX_FLAG_HT;
 425                 rate_idx = rxdesc.signal;
 426         }
 427
 428         /*
 429          * Update extra components
 430          */
 431         rt2x00link_update_stats(rt2x00dev, entry->skb, &rxdesc);
 432         rt2x00debug_update_crypto(rt2x00dev, &rxdesc);
 433
 434         rx_status->mactime = rxdesc.timestamp;
 435         rx_status->rate_idx = rate_idx;
 436         rx_status->signal = rxdesc.rssi;
 437         rx_status->noise = rxdesc.noise;
 438         rx_status->flag = rxdesc.flags;
 439         rx_status->antenna = rt2x00dev->link.ant.active.rx;
 440
 441         /*
 442          * Send frame to mac80211 & debugfs.
 443          * mac80211 will clean up the skb structure.
 444          */
 445         rt2x00debug_dump_frame(rt2x00dev, DUMP_FRAME_RXDONE, entry->skb);
 446         memcpy(IEEE80211_SKB_RXCB(entry->skb), rx_status, sizeof(*rx_status));
 447         ieee80211_rx_irqsafe(rt2x00dev->hw, entry->skb);
 448
 449         /*
 450          * Replace the skb with the freshly allocated one.
 451          */
 452         entry->skb = skb;
 453         entry->flags = 0;
 454
 455         rt2x00dev->ops->lib->clear_entry(entry);
 456
 457         rt2x00queue_index_inc(entry->queue, Q_INDEX);
 458 }
 459 EXPORT_SYMBOL_GPL(rt2x00lib_rxdone);
 460
 461 /*
 462  * Driver initialization handlers.
 463  */
 464 const struct rt2x00_rate rt2x00_supported_rates[12] = {
 465         {
 466                 .flags = DEV_RATE_CCK,
 467                 .bitrate = 10,
 468                 .ratemask = BIT(0),
 469                 .plcp = 0x00,
 470                 .mcs = RATE_MCS(RATE_MODE_CCK, 0),
 471         },
 472         {
 473                 .flags = DEV_RATE_CCK | DEV_RATE_SHORT_PREAMBLE,
 474                 .bitrate = 20,
 475                 .ratemask = BIT(1),
 476                 .plcp = 0x01,
 477                 .mcs = RATE_MCS(RATE_MODE_CCK, 1),
 478         },
 479         {
 480                 .flags = DEV_RATE_CCK | DEV_RATE_SHORT_PREAMBLE,
 481                 .bitrate = 55,
 482                 .ratemask = BIT(2),
 483                 .plcp = 0x02,
 484                 .mcs = RATE_MCS(RATE_MODE_CCK, 2),
 485         },
 486         {
 487                 .flags = DEV_RATE_CCK | DEV_RATE_SHORT_PREAMBLE,
 488                 .bitrate = 110,
 489                 .ratemask = BIT(3),
 490                 .plcp = 0x03,
 491                 .mcs = RATE_MCS(RATE_MODE_CCK, 3),
 492         },
 493         {
 494                 .flags = DEV_RATE_OFDM,
 495                 .bitrate = 60,
 496                 .ratemask = BIT(4),
 497                 .plcp = 0x0b,
 498                 .mcs = RATE_MCS(RATE_MODE_OFDM, 0),
 499         },
 500         {
 501                 .flags = DEV_RATE_OFDM,
 502                 .bitrate = 90,
 503                 .ratemask = BIT(5),
 504                 .plcp = 0x0f,
 505                 .mcs = RATE_MCS(RATE_MODE_OFDM, 1),
 506         },
 507         {
 508                 .flags = DEV_RATE_OFDM,
 509                 .bitrate = 120,
 510                 .ratemask = BIT(6),
 511                 .plcp = 0x0a,
 512                 .mcs = RATE_MCS(RATE_MODE_OFDM, 2),
 513         },
 514         {
 515                 .flags = DEV_RATE_OFDM,
 516                 .bitrate = 180,
 517                 .ratemask = BIT(7),
 518                 .plcp = 0x0e,
 519                 .mcs = RATE_MCS(RATE_MODE_OFDM, 3),
 520         },
 521         {
 522                 .flags = DEV_RATE_OFDM,
 523                 .bitrate = 240,
 524                 .ratemask = BIT(8),
 525                 .plcp = 0x09,
 526                 .mcs = RATE_MCS(RATE_MODE_OFDM, 4),
 527         },
 528         {
 529                 .flags = DEV_RATE_OFDM,
 530                 .bitrate = 360,
 531                 .ratemask = BIT(9),
 532                 .plcp = 0x0d,
 533                 .mcs = RATE_MCS(RATE_MODE_OFDM, 5),
 534         },
 535         {
 536                 .flags = DEV_RATE_OFDM,
 537                 .bitrate = 480,
 538                 .ratemask = BIT(10),
 539                 .plcp = 0x08,
 540                 .mcs = RATE_MCS(RATE_MODE_OFDM, 6),
 541         },
 542         {
 543                 .flags = DEV_RATE_OFDM,
 544                 .bitrate = 540,
 545                 .ratemask = BIT(11),
 546                 .plcp = 0x0c,
 547                 .mcs = RATE_MCS(RATE_MODE_OFDM, 7),
 548         },
 549 };
 550
 551 static void rt2x00lib_channel(struct ieee80211_channel *entry,
 552                               const int channel, const int tx_power,
 553                               const int value)
 554 {
 555         entry->center_freq = ieee80211_channel_to_frequency(channel);
 556         entry->hw_value = value;
 557         entry->max_power = tx_power;
 558         entry->max_antenna_gain = 0xff;
 559 }
 560
 561 static void rt2x00lib_rate(struct ieee80211_rate *entry,
 562                            const u16 index, const struct rt2x00_rate *rate)
 563 {
 564         entry->flags = 0;
 565         entry->bitrate = rate->bitrate;
 566         entry->hw_value =index;
 567         entry->hw_value_short = index;
 568
 569         if (rate->flags & DEV_RATE_SHORT_PREAMBLE)
 570                 entry->flags |= IEEE80211_RATE_SHORT_PREAMBLE;
 571 }
 572
 573 static int rt2x00lib_probe_hw_modes(struct rt2x00_dev *rt2x00dev,
 574                                     struct hw_mode_spec *spec)
 575 {
 576         struct ieee80211_hw *hw = rt2x00dev->hw;
 577         struct ieee80211_channel *channels;
 578         struct ieee80211_rate *rates;
 579         unsigned int num_rates;
 580         unsigned int i;
 581
 582         num_rates = 0;
 583         if (spec->supported_rates & SUPPORT_RATE_CCK)
 584                 num_rates += 4;
 585         if (spec->supported_rates & SUPPORT_RATE_OFDM)
 586                 num_rates += 8;
 587
 588         channels = kzalloc(sizeof(*channels) * spec->num_channels, GFP_KERNEL);
 589         if (!channels)
 590                 return -ENOMEM;
 591
 592         rates = kzalloc(sizeof(*rates) * num_rates, GFP_KERNEL);
 593         if (!rates)
 594                 goto exit_free_channels;
 595
 596         /*
 597          * Initialize Rate list.
 598          */
 599         for (i = 0; i < num_rates; i++)
 600                 rt2x00lib_rate(&rates[i], i, rt2x00_get_rate(i));
 601
 602         /*
 603          * Initialize Channel list.
 604          */
 605         for (i = 0; i < spec->num_channels; i++) {
 606                 rt2x00lib_channel(&channels[i],
 607                                   spec->channels[i].channel,
 608                                   spec->channels_info[i].tx_power1, i);
 609         }
 610
 611         /*
 612          * Intitialize 802.11b, 802.11g
 613          * Rates: CCK, OFDM.
 614          * Channels: 2.4 GHz
 615          */
 616         if (spec->supported_bands & SUPPORT_BAND_2GHZ) {
 617                 rt2x00dev->bands[IEEE80211_BAND_2GHZ].n_channels = 14;
 618                 rt2x00dev->bands[IEEE80211_BAND_2GHZ].n_bitrates = num_rates;
 619                 rt2x00dev->bands[IEEE80211_BAND_2GHZ].channels = channels;
 620                 rt2x00dev->bands[IEEE80211_BAND_2GHZ].bitrates = rates;
 621                 hw->wiphy->bands[IEEE80211_BAND_2GHZ] =
 622                     &rt2x00dev->bands[IEEE80211_BAND_2GHZ];
 623                 memcpy(&rt2x00dev->bands[IEEE80211_BAND_2GHZ].ht_cap,
 624                        &spec->ht, sizeof(spec->ht));
 625         }
 626
 627         /*
 628          * Intitialize 802.11a
 629          * Rates: OFDM.
 630          * Channels: OFDM, UNII, HiperLAN2.
 631          */
 632         if (spec->supported_bands & SUPPORT_BAND_5GHZ) {
 633                 rt2x00dev->bands[IEEE80211_BAND_5GHZ].n_channels =
 634                     spec->num_channels - 14;
 635                 rt2x00dev->bands[IEEE80211_BAND_5GHZ].n_bitrates =
 636                     num_rates - 4;
 637                 rt2x00dev->bands[IEEE80211_BAND_5GHZ].channels = &channels[14];
 638                 rt2x00dev->bands[IEEE80211_BAND_5GHZ].bitrates = &rates[4];
 639                 hw->wiphy->bands[IEEE80211_BAND_5GHZ] =
 640                     &rt2x00dev->bands[IEEE80211_BAND_5GHZ];
 641                 memcpy(&rt2x00dev->bands[IEEE80211_BAND_5GHZ].ht_cap,
 642                        &spec->ht, sizeof(spec->ht));
 643         }
 644
 645         return 0;
 646
 647  exit_free_channels:
 648         kfree(channels);
 649         ERROR(rt2x00dev, "Allocation ieee80211 modes failed.\n");
 650         return -ENOMEM;
 651 }
 652
 653 static void rt2x00lib_remove_hw(struct rt2x00_dev *rt2x00dev)
 654 {
 655         if (test_bit(DEVICE_STATE_REGISTERED_HW, &rt2x00dev->flags))
 656                 ieee80211_unregister_hw(rt2x00dev->hw);
 657
 658         if (likely(rt2x00dev->hw->wiphy->bands[IEEE80211_BAND_2GHZ])) {
 659                 kfree(rt2x00dev->hw->wiphy->bands[IEEE80211_BAND_2GHZ]->channels);
 660                 kfree(rt2x00dev->hw->wiphy->bands[IEEE80211_BAND_2GHZ]->bitrates);
 661                 rt2x00dev->hw->wiphy->bands[IEEE80211_BAND_2GHZ] = NULL;
 662                 rt2x00dev->hw->wiphy->bands[IEEE80211_BAND_5GHZ] = NULL;
 663         }
 664
 665         kfree(rt2x00dev->spec.channels_info);
 666 }
 667
 668 static int rt2x00lib_probe_hw(struct rt2x00_dev *rt2x00dev)
 669 {
 670         struct hw_mode_spec *spec = &rt2x00dev->spec;
 671         int status;
 672
 673         if (test_bit(DEVICE_STATE_REGISTERED_HW, &rt2x00dev->flags))
 674                 return 0;
 675
 676         /*
 677          * Initialize HW modes.
 678          */
 679         status = rt2x00lib_probe_hw_modes(rt2x00dev, spec);
 680         if (status)
 681                 return status;
 682
 683         /*
 684          * Initialize HW fields.
 685          */
 686         rt2x00dev->hw->queues = rt2x00dev->ops->tx_queues;
 687
 688         /*
 689          * Initialize extra TX headroom required.
 690          */
 691         rt2x00dev->hw->extra_tx_headroom =
 692                 max_t(unsigned int, IEEE80211_TX_STATUS_HEADROOM,
 693                       rt2x00dev->ops->extra_tx_headroom);
 694
 695         /*
 696          * Take TX headroom required for alignment into account.
 697          */
 698         if (test_bit(DRIVER_REQUIRE_L2PAD, &rt2x00dev->flags))
 699                 rt2x00dev->hw->extra_tx_headroom += RT2X00_L2PAD_SIZE;
 700         else if (test_bit(DRIVER_REQUIRE_DMA, &rt2x00dev->flags))
 701                 rt2x00dev->hw->extra_tx_headroom += RT2X00_ALIGN_SIZE;
 702
 703         /*
 704          * Register HW.
 705          */
 706         status = ieee80211_register_hw(rt2x00dev->hw);
 707         if (status)
 708                 return status;
 709
 710         set_bit(DEVICE_STATE_REGISTERED_HW, &rt2x00dev->flags);
 711
 712         return 0;
 713 }
 714
 715 /*
 716  * Initialization/uninitialization handlers.
 717  */
 718 static void rt2x00lib_uninitialize(struct rt2x00_dev *rt2x00dev)
 719 {
 720         if (!test_and_clear_bit(DEVICE_STATE_INITIALIZED, &rt2x00dev->flags))
 721                 return;
 722
 723         /*
 724          * Unregister extra components.
 725          */
 726         rt2x00rfkill_unregister(rt2x00dev);
 727
 728         /*
 729          * Allow the HW to uninitialize.
 730          */
 731         rt2x00dev->ops->lib->uninitialize(rt2x00dev);
 732
 733         /*
 734          * Free allocated queue entries.
 735          */
 736         rt2x00queue_uninitialize(rt2x00dev);
 737 }
 738
 739 static int rt2x00lib_initialize(struct rt2x00_dev *rt2x00dev)
 740 {
 741         int status;
 742
 743         if (test_bit(DEVICE_STATE_INITIALIZED, &rt2x00dev->flags))
 744                 return 0;
 745
 746         /*
 747          * Allocate all queue entries.
 748          */
 749         status = rt2x00queue_initialize(rt2x00dev);
 750         if (status)
 751                 return status;
 752
 753         /*
 754          * Initialize the device.
 755          */
 756         status = rt2x00dev->ops->lib->initialize(rt2x00dev);
 757         if (status) {
 758                 rt2x00queue_uninitialize(rt2x00dev);
 759                 return status;
 760         }
 761
 762         set_bit(DEVICE_STATE_INITIALIZED, &rt2x00dev->flags);
 763
 764         /*
 765          * Register the extra components.
 766          */
 767         rt2x00rfkill_register(rt2x00dev);
 768
 769         return 0;
 770 }
 771
 772 int rt2x00lib_start(struct rt2x00_dev *rt2x00dev)
 773 {
 774         int retval;
 775
 776         if (test_bit(DEVICE_STATE_STARTED, &rt2x00dev->flags))
 777                 return 0;
 778
 779         /*
 780          * If this is the first interface which is added,
 781          * we should load the firmware now.
 782          */
 783         retval = rt2x00lib_load_firmware(rt2x00dev);
 784         if (retval)
 785                 return retval;
 786
 787         /*
 788          * Initialize the device.
 789          */
 790         retval = rt2x00lib_initialize(rt2x00dev);
 791         if (retval)
 792                 return retval;
 793
 794         rt2x00dev->intf_ap_count = 0;
 795         rt2x00dev->intf_sta_count = 0;
 796         rt2x00dev->intf_associated = 0;
 797
 798         /* Enable the radio */
 799         retval = rt2x00lib_enable_radio(rt2x00dev);
 800         if (retval) {
 801                 rt2x00queue_uninitialize(rt2x00dev);
 802                 return retval;
 803         }
 804
 805         set_bit(DEVICE_STATE_STARTED, &rt2x00dev->flags);
 806
 807         return 0;
 808 }
 809
 810 void rt2x00lib_stop(struct rt2x00_dev *rt2x00dev)
 811 {
 812         if (!test_and_clear_bit(DEVICE_STATE_STARTED, &rt2x00dev->flags))
 813                 return;
 814
 815         /*
 816          * Perhaps we can add something smarter here,
 817          * but for now just disabling the radio should do.
 818          */
 819         rt2x00lib_disable_radio(rt2x00dev);
 820
 821         rt2x00dev->intf_ap_count = 0;
 822         rt2x00dev->intf_sta_count = 0;
 823         rt2x00dev->intf_associated = 0;
 824 }
 825
 826 /*
 827  * driver allocation handlers.
 828  */
 829 int rt2x00lib_probe_dev(struct rt2x00_dev *rt2x00dev)
 830 {
 831         int retval = -ENOMEM;
 832
 833         mutex_init(&rt2x00dev->csr_mutex);
 834
 835         set_bit(DEVICE_STATE_PRESENT, &rt2x00dev->flags);
 836
 837         /*
 838          * Make room for rt2x00_intf inside the per-interface
 839          * structure ieee80211_vif.
 840          */
 841         rt2x00dev->hw->vif_data_size = sizeof(struct rt2x00_intf);
 842
 843         /*
 844          * Determine which operating modes are supported, all modes
 845          * which require beaconing, depend on the availability of
 846          * beacon entries.
 847          */
 848         rt2x00dev->hw->wiphy->interface_modes = BIT(NL80211_IFTYPE_STATION);
 849         if (rt2x00dev->ops->bcn->entry_num > 0)
 850                 rt2x00dev->hw->wiphy->interface_modes |=
 851                     BIT(NL80211_IFTYPE_ADHOC) |
 852                     BIT(NL80211_IFTYPE_AP) |
 853                     BIT(NL80211_IFTYPE_MESH_POINT) |
 854                     BIT(NL80211_IFTYPE_WDS);
 855
 856         /*
 857          * Let the driver probe the device to detect the capabilities.
 858          */
 859         retval = rt2x00dev->ops->lib->probe_hw(rt2x00dev);
 860         if (retval) {
 861                 ERROR(rt2x00dev, "Failed to allocate device.\n");
 862                 goto exit;
 863         }
 864
 865         /*
 866          * Initialize configuration work.
 867          */
 868         INIT_WORK(&rt2x00dev->intf_work, rt2x00lib_intf_scheduled);
 869
 870         /*
 871          * Allocate queue array.
 872          */
 873         retval = rt2x00queue_allocate(rt2x00dev);
 874         if (retval)
 875                 goto exit;
 876
 877         /*
 878          * Initialize ieee80211 structure.
 879          */
 880         retval = rt2x00lib_probe_hw(rt2x00dev);
 881         if (retval) {
 882                 ERROR(rt2x00dev, "Failed to initialize hw.\n");
 883                 goto exit;
 884         }
 885
 886         /*
 887          * Register extra components.
 888          */
 889         rt2x00link_register(rt2x00dev);
 890         rt2x00leds_register(rt2x00dev);
 891         rt2x00debug_register(rt2x00dev);
 892
 893         return 0;
 894
 895 exit:
 896         rt2x00lib_remove_dev(rt2x00dev);
 897
 898         return retval;
 899 }
 900 EXPORT_SYMBOL_GPL(rt2x00lib_probe_dev);
 901
 902 void rt2x00lib_remove_dev(struct rt2x00_dev *rt2x00dev)
 903 {
 904         clear_bit(DEVICE_STATE_PRESENT, &rt2x00dev->flags);
 905
 906         /*
 907          * Disable radio.
 908          */
 909         rt2x00lib_disable_radio(rt2x00dev);
 910
 911         /*
 912          * Stop all work.
 913          */
 914         cancel_work_sync(&rt2x00dev->intf_work);
 915
 916         /*
 917          * Uninitialize device.
 918          */
 919         rt2x00lib_uninitialize(rt2x00dev);
 920
 921         /*
 922          * Free extra components
 923          */
 924         rt2x00debug_deregister(rt2x00dev);
 925         rt2x00leds_unregister(rt2x00dev);
 926
 927         /*
 928          * Free ieee80211_hw memory.
 929          */
 930         rt2x00lib_remove_hw(rt2x00dev);
 931
 932         /*
 933          * Free firmware image.
 934          */
 935         rt2x00lib_free_firmware(rt2x00dev);
 936
 937         /*
 938          * Free queue structures.
 939          */
 940         rt2x00queue_free(rt2x00dev);
 941 }
 942 EXPORT_SYMBOL_GPL(rt2x00lib_remove_dev);
 943
 944 /*
 945  * Device state handlers
 946  */
 947 #ifdef CONFIG_PM
 948 int rt2x00lib_suspend(struct rt2x00_dev *rt2x00dev, pm_message_t state)
 949 {
 950         NOTICE(rt2x00dev, "Going to sleep.\n");
 951
 952         /*
 953          * Prevent mac80211 from accessing driver while suspended.
 954          */
 955         if (!test_and_clear_bit(DEVICE_STATE_PRESENT, &rt2x00dev->flags))
 956                 return 0;
 957
 958         /*
 959          * Cleanup as much as possible.
 960          */
 961         rt2x00lib_uninitialize(rt2x00dev);
 962
 963         /*
 964          * Suspend/disable extra components.
 965          */
 966         rt2x00leds_suspend(rt2x00dev);
 967         rt2x00debug_deregister(rt2x00dev);
 968
 969         /*
 970          * Set device mode to sleep for power management,
 971          * on some hardware this call seems to consistently fail.
 972          * From the specifications it is hard to tell why it fails,
 973          * and if this is a "bad thing".
 974          * Overall it is safe to just ignore the failure and
 975          * continue suspending. The only downside is that the
 976          * device will not be in optimal power save mode, but with
 977          * the radio and the other components already disabled the
 978          * device is as good as disabled.
 979          */
 980         if (rt2x00dev->ops->lib->set_device_state(rt2x00dev, STATE_SLEEP))
 981                 WARNING(rt2x00dev, "Device failed to enter sleep state, "
 982                         "continue suspending.\n");
 983
 984         return 0;
 985 }
 986 EXPORT_SYMBOL_GPL(rt2x00lib_suspend);
 987
 988 int rt2x00lib_resume(struct rt2x00_dev *rt2x00dev)
 989 {
 990         NOTICE(rt2x00dev, "Waking up.\n");
 991
 992         /*
 993          * Restore/enable extra components.
 994          */
 995         rt2x00debug_register(rt2x00dev);
 996         rt2x00leds_resume(rt2x00dev);
 997
 998         /*
 999          * We are ready again to receive requests from mac80211.
1000          */
1001         set_bit(DEVICE_STATE_PRESENT, &rt2x00dev->flags);
1002
1003         return 0;
1004 }
1005 EXPORT_SYMBOL_GPL(rt2x00lib_resume);
1006 #endif /* CONFIG_PM */
1007
1008 /*
1009  * rt2x00lib module information.
1010  */
1011 MODULE_AUTHOR(DRV_PROJECT);
1012 MODULE_VERSION(DRV_VERSION);
1013 MODULE_DESCRIPTION("rt2x00 library");
1014 MODULE_LICENSE("GPL");