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