17e642978f8dce60d7dbcfdfcfef14b79336691a
[pandora-kernel.git] / drivers / net / wireless / rtlwifi / usb.c
1 /******************************************************************************
2  *
3  * Copyright(c) 2009-2011  Realtek Corporation. All rights reserved.
4  *
5  * This program is free software; you can redistribute it and/or modify it
6  * under the terms of version 2 of the GNU General Public License as
7  * published by the Free Software Foundation.
8  *
9  * This program is distributed in the hope that it will be useful, but WITHOUT
10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
12  * more details.
13  *
14  * You should have received a copy of the GNU General Public License along with
15  * this program; if not, write to the Free Software Foundation, Inc.,
16  * 51 Franklin Street, Fifth Floor, Boston, MA 02110, USA
17  *
18  * The full GNU General Public License is included in this distribution in the
19  * file called LICENSE.
20  *
21  * Contact Information:
22  * wlanfae <wlanfae@realtek.com>
23  * Realtek Corporation, No. 2, Innovation Road II, Hsinchu Science Park,
24  * Hsinchu 300, Taiwan.
25  *
26  *****************************************************************************/
27
28 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
29
30 #include <linux/usb.h>
31 #include <linux/export.h>
32 #include "core.h"
33 #include "wifi.h"
34 #include "usb.h"
35 #include "base.h"
36 #include "ps.h"
37
38 #define REALTEK_USB_VENQT_READ                  0xC0
39 #define REALTEK_USB_VENQT_WRITE                 0x40
40 #define REALTEK_USB_VENQT_CMD_REQ               0x05
41 #define REALTEK_USB_VENQT_CMD_IDX               0x00
42
43 #define REALTEK_USB_VENQT_MAX_BUF_SIZE          254
44
45 static void usbctrl_async_callback(struct urb *urb)
46 {
47         if (urb) {
48                 /* free dr */
49                 kfree(urb->setup_packet);
50                 /* free databuf */
51                 kfree(urb->transfer_buffer);
52         }
53 }
54
55 static int _usbctrl_vendorreq_async_write(struct usb_device *udev, u8 request,
56                                           u16 value, u16 index, void *pdata,
57                                           u16 len)
58 {
59         int rc;
60         unsigned int pipe;
61         u8 reqtype;
62         struct usb_ctrlrequest *dr;
63         struct urb *urb;
64         const u16 databuf_maxlen = REALTEK_USB_VENQT_MAX_BUF_SIZE;
65         u8 *databuf;
66
67         if (WARN_ON_ONCE(len > databuf_maxlen))
68                 len = databuf_maxlen;
69
70         pipe = usb_sndctrlpipe(udev, 0); /* write_out */
71         reqtype =  REALTEK_USB_VENQT_WRITE;
72
73         dr = kmalloc(sizeof(*dr), GFP_ATOMIC);
74         if (!dr)
75                 return -ENOMEM;
76
77         databuf = kmalloc(databuf_maxlen, GFP_ATOMIC);
78         if (!databuf) {
79                 kfree(dr);
80                 return -ENOMEM;
81         }
82
83         urb = usb_alloc_urb(0, GFP_ATOMIC);
84         if (!urb) {
85                 kfree(databuf);
86                 kfree(dr);
87                 return -ENOMEM;
88         }
89
90         dr->bRequestType = reqtype;
91         dr->bRequest = request;
92         dr->wValue = cpu_to_le16(value);
93         dr->wIndex = cpu_to_le16(index);
94         dr->wLength = cpu_to_le16(len);
95         memcpy(databuf, pdata, len);
96         usb_fill_control_urb(urb, udev, pipe,
97                              (unsigned char *)dr, databuf, len,
98                              usbctrl_async_callback, NULL);
99         rc = usb_submit_urb(urb, GFP_ATOMIC);
100         if (rc < 0) {
101                 kfree(databuf);
102                 kfree(dr);
103         }
104         usb_free_urb(urb);
105         return rc;
106 }
107
108 static int _usbctrl_vendorreq_sync_read(struct usb_device *udev, u8 request,
109                                         u16 value, u16 index, void *pdata,
110                                         u16 len)
111 {
112         unsigned int pipe;
113         int status;
114         u8 reqtype;
115
116         pipe = usb_rcvctrlpipe(udev, 0); /* read_in */
117         reqtype =  REALTEK_USB_VENQT_READ;
118
119         status = usb_control_msg(udev, pipe, request, reqtype, value, index,
120                                  pdata, len, 1000);
121
122         if (status < 0)
123                 pr_err("reg 0x%x, usbctrl_vendorreq TimeOut! status:0x%x value=0x%x\n",
124                        value, status, *(u32 *)pdata);
125         return status;
126 }
127
128 static u32 _usb_read_sync(struct rtl_priv *rtlpriv, u32 addr, u16 len)
129 {
130         struct device *dev = rtlpriv->io.dev;
131         struct usb_device *udev = to_usb_device(dev);
132         u8 request;
133         u16 wvalue;
134         u16 index;
135         __le32 *data;
136         unsigned long flags;
137
138         spin_lock_irqsave(&rtlpriv->locks.usb_lock, flags);
139         if (++rtlpriv->usb_data_index >= RTL_USB_MAX_RX_COUNT)
140                 rtlpriv->usb_data_index = 0;
141         data = &rtlpriv->usb_data[rtlpriv->usb_data_index];
142         spin_unlock_irqrestore(&rtlpriv->locks.usb_lock, flags);
143         request = REALTEK_USB_VENQT_CMD_REQ;
144         index = REALTEK_USB_VENQT_CMD_IDX; /* n/a */
145
146         wvalue = (u16)addr;
147         _usbctrl_vendorreq_sync_read(udev, request, wvalue, index, data, len);
148         return le32_to_cpu(*data);
149 }
150
151 static u8 _usb_read8_sync(struct rtl_priv *rtlpriv, u32 addr)
152 {
153         return (u8)_usb_read_sync(rtlpriv, addr, 1);
154 }
155
156 static u16 _usb_read16_sync(struct rtl_priv *rtlpriv, u32 addr)
157 {
158         return (u16)_usb_read_sync(rtlpriv, addr, 2);
159 }
160
161 static u32 _usb_read32_sync(struct rtl_priv *rtlpriv, u32 addr)
162 {
163         return _usb_read_sync(rtlpriv, addr, 4);
164 }
165
166 static void _usb_write_async(struct usb_device *udev, u32 addr, u32 val,
167                              u16 len)
168 {
169         u8 request;
170         u16 wvalue;
171         u16 index;
172         u32 data;
173
174         request = REALTEK_USB_VENQT_CMD_REQ;
175         index = REALTEK_USB_VENQT_CMD_IDX; /* n/a */
176         wvalue = (u16)(addr&0x0000ffff);
177         data = val;
178         _usbctrl_vendorreq_async_write(udev, request, wvalue, index, &data,
179                                        len);
180 }
181
182 static void _usb_write8_async(struct rtl_priv *rtlpriv, u32 addr, u8 val)
183 {
184         struct device *dev = rtlpriv->io.dev;
185
186         _usb_write_async(to_usb_device(dev), addr, val, 1);
187 }
188
189 static void _usb_write16_async(struct rtl_priv *rtlpriv, u32 addr, u16 val)
190 {
191         struct device *dev = rtlpriv->io.dev;
192
193         _usb_write_async(to_usb_device(dev), addr, val, 2);
194 }
195
196 static void _usb_write32_async(struct rtl_priv *rtlpriv, u32 addr, u32 val)
197 {
198         struct device *dev = rtlpriv->io.dev;
199
200         _usb_write_async(to_usb_device(dev), addr, val, 4);
201 }
202
203 static void _rtl_usb_io_handler_init(struct device *dev,
204                                      struct ieee80211_hw *hw)
205 {
206         struct rtl_priv *rtlpriv = rtl_priv(hw);
207
208         rtlpriv->io.dev = dev;
209         mutex_init(&rtlpriv->io.bb_mutex);
210         rtlpriv->io.write8_async        = _usb_write8_async;
211         rtlpriv->io.write16_async       = _usb_write16_async;
212         rtlpriv->io.write32_async       = _usb_write32_async;
213         rtlpriv->io.read8_sync          = _usb_read8_sync;
214         rtlpriv->io.read16_sync         = _usb_read16_sync;
215         rtlpriv->io.read32_sync         = _usb_read32_sync;
216 }
217
218 static void _rtl_usb_io_handler_release(struct ieee80211_hw *hw)
219 {
220         struct rtl_priv __maybe_unused *rtlpriv = rtl_priv(hw);
221
222         mutex_destroy(&rtlpriv->io.bb_mutex);
223 }
224
225 /**
226  *
227  *      Default aggregation handler. Do nothing and just return the oldest skb.
228  */
229 static struct sk_buff *_none_usb_tx_aggregate_hdl(struct ieee80211_hw *hw,
230                                                   struct sk_buff_head *list)
231 {
232         return skb_dequeue(list);
233 }
234
235 #define IS_HIGH_SPEED_USB(udev) \
236                 ((USB_SPEED_HIGH == (udev)->speed) ? true : false)
237
238 static int _rtl_usb_init_tx(struct ieee80211_hw *hw)
239 {
240         u32 i;
241         struct rtl_priv *rtlpriv = rtl_priv(hw);
242         struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
243
244         rtlusb->max_bulk_out_size = IS_HIGH_SPEED_USB(rtlusb->udev)
245                                                     ? USB_HIGH_SPEED_BULK_SIZE
246                                                     : USB_FULL_SPEED_BULK_SIZE;
247
248         RT_TRACE(rtlpriv, COMP_INIT, DBG_DMESG, ("USB Max Bulk-out Size=%d\n",
249                  rtlusb->max_bulk_out_size));
250
251         for (i = 0; i < __RTL_TXQ_NUM; i++) {
252                 u32 ep_num = rtlusb->ep_map.ep_mapping[i];
253                 if (!ep_num) {
254                         RT_TRACE(rtlpriv, COMP_INIT, DBG_DMESG,
255                                  ("Invalid endpoint map setting!\n"));
256                         return -EINVAL;
257                 }
258         }
259
260         rtlusb->usb_tx_post_hdl =
261                  rtlpriv->cfg->usb_interface_cfg->usb_tx_post_hdl;
262         rtlusb->usb_tx_cleanup  =
263                  rtlpriv->cfg->usb_interface_cfg->usb_tx_cleanup;
264         rtlusb->usb_tx_aggregate_hdl =
265                  (rtlpriv->cfg->usb_interface_cfg->usb_tx_aggregate_hdl)
266                  ? rtlpriv->cfg->usb_interface_cfg->usb_tx_aggregate_hdl
267                  : &_none_usb_tx_aggregate_hdl;
268
269         init_usb_anchor(&rtlusb->tx_submitted);
270         for (i = 0; i < RTL_USB_MAX_EP_NUM; i++) {
271                 skb_queue_head_init(&rtlusb->tx_skb_queue[i]);
272                 init_usb_anchor(&rtlusb->tx_pending[i]);
273         }
274         return 0;
275 }
276
277 static int _rtl_usb_init_rx(struct ieee80211_hw *hw)
278 {
279         struct rtl_priv *rtlpriv = rtl_priv(hw);
280         struct rtl_usb_priv *usb_priv = rtl_usbpriv(hw);
281         struct rtl_usb *rtlusb = rtl_usbdev(usb_priv);
282
283         rtlusb->rx_max_size = rtlpriv->cfg->usb_interface_cfg->rx_max_size;
284         rtlusb->rx_urb_num = rtlpriv->cfg->usb_interface_cfg->rx_urb_num;
285         rtlusb->in_ep = rtlpriv->cfg->usb_interface_cfg->in_ep_num;
286         rtlusb->usb_rx_hdl = rtlpriv->cfg->usb_interface_cfg->usb_rx_hdl;
287         rtlusb->usb_rx_segregate_hdl =
288                 rtlpriv->cfg->usb_interface_cfg->usb_rx_segregate_hdl;
289
290         pr_info("rx_max_size %d, rx_urb_num %d, in_ep %d\n",
291                 rtlusb->rx_max_size, rtlusb->rx_urb_num, rtlusb->in_ep);
292         init_usb_anchor(&rtlusb->rx_submitted);
293         return 0;
294 }
295
296 static int _rtl_usb_init(struct ieee80211_hw *hw)
297 {
298         struct rtl_priv *rtlpriv = rtl_priv(hw);
299         struct rtl_usb_priv *usb_priv = rtl_usbpriv(hw);
300         struct rtl_usb *rtlusb = rtl_usbdev(usb_priv);
301         int err;
302         u8 epidx;
303         struct usb_interface    *usb_intf = rtlusb->intf;
304         u8 epnums = usb_intf->cur_altsetting->desc.bNumEndpoints;
305
306         rtlusb->out_ep_nums = rtlusb->in_ep_nums = 0;
307         for (epidx = 0; epidx < epnums; epidx++) {
308                 struct usb_endpoint_descriptor *pep_desc;
309                 pep_desc = &usb_intf->cur_altsetting->endpoint[epidx].desc;
310
311                 if (usb_endpoint_dir_in(pep_desc))
312                         rtlusb->in_ep_nums++;
313                 else if (usb_endpoint_dir_out(pep_desc))
314                         rtlusb->out_ep_nums++;
315
316                 RT_TRACE(rtlpriv, COMP_INIT, DBG_DMESG,
317                          ("USB EP(0x%02x), MaxPacketSize=%d ,Interval=%d.\n",
318                          pep_desc->bEndpointAddress, pep_desc->wMaxPacketSize,
319                          pep_desc->bInterval));
320         }
321         if (rtlusb->in_ep_nums <  rtlpriv->cfg->usb_interface_cfg->in_ep_num)
322                 return -EINVAL ;
323
324         /* usb endpoint mapping */
325         err = rtlpriv->cfg->usb_interface_cfg->usb_endpoint_mapping(hw);
326         rtlusb->usb_mq_to_hwq =  rtlpriv->cfg->usb_interface_cfg->usb_mq_to_hwq;
327         _rtl_usb_init_tx(hw);
328         _rtl_usb_init_rx(hw);
329         return err;
330 }
331
332 static int _rtl_usb_init_sw(struct ieee80211_hw *hw)
333 {
334         struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
335         struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
336         struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
337         struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
338
339         rtlhal->hw = hw;
340         ppsc->inactiveps = false;
341         ppsc->leisure_ps = false;
342         ppsc->fwctrl_lps = false;
343         ppsc->reg_fwctrl_lps = 3;
344         ppsc->reg_max_lps_awakeintvl = 5;
345         ppsc->fwctrl_psmode = FW_PS_DTIM_MODE;
346
347          /* IBSS */
348         mac->beacon_interval = 100;
349
350          /* AMPDU */
351         mac->min_space_cfg = 0;
352         mac->max_mss_density = 0;
353
354         /* set sane AMPDU defaults */
355         mac->current_ampdu_density = 7;
356         mac->current_ampdu_factor = 3;
357
358         /* QOS */
359         rtlusb->acm_method = eAcmWay2_SW;
360
361         /* IRQ */
362         /* HIMR - turn all on */
363         rtlusb->irq_mask[0] = 0xFFFFFFFF;
364         /* HIMR_EX - turn all on */
365         rtlusb->irq_mask[1] = 0xFFFFFFFF;
366         rtlusb->disableHWSM =  true;
367         return 0;
368 }
369
370 #define __RADIO_TAP_SIZE_RSV    32
371
372 static void _rtl_rx_completed(struct urb *urb);
373
374 static struct sk_buff *_rtl_prep_rx_urb(struct ieee80211_hw *hw,
375                                         struct rtl_usb *rtlusb,
376                                         struct urb *urb,
377                                         gfp_t gfp_mask)
378 {
379         struct sk_buff *skb;
380         struct rtl_priv *rtlpriv = rtl_priv(hw);
381
382         skb = __dev_alloc_skb((rtlusb->rx_max_size + __RADIO_TAP_SIZE_RSV),
383                                gfp_mask);
384         if (!skb) {
385                 RT_TRACE(rtlpriv, COMP_USB, DBG_EMERG,
386                          ("Failed to __dev_alloc_skb!!\n"))
387                 return ERR_PTR(-ENOMEM);
388         }
389
390         /* reserve some space for mac80211's radiotap */
391         skb_reserve(skb, __RADIO_TAP_SIZE_RSV);
392         usb_fill_bulk_urb(urb, rtlusb->udev,
393                           usb_rcvbulkpipe(rtlusb->udev, rtlusb->in_ep),
394                           skb->data, min(skb_tailroom(skb),
395                           (int)rtlusb->rx_max_size),
396                           _rtl_rx_completed, skb);
397
398         _rtl_install_trx_info(rtlusb, skb, rtlusb->in_ep);
399         return skb;
400 }
401
402 #undef __RADIO_TAP_SIZE_RSV
403
404 static void _rtl_usb_rx_process_agg(struct ieee80211_hw *hw,
405                                     struct sk_buff *skb)
406 {
407         struct rtl_priv *rtlpriv = rtl_priv(hw);
408         u8 *rxdesc = skb->data;
409         struct ieee80211_hdr *hdr;
410         bool unicast = false;
411         __le16 fc;
412         struct ieee80211_rx_status rx_status = {0};
413         struct rtl_stats stats = {
414                 .signal = 0,
415                 .noise = -98,
416                 .rate = 0,
417         };
418
419         skb_pull(skb, RTL_RX_DESC_SIZE);
420         rtlpriv->cfg->ops->query_rx_desc(hw, &stats, &rx_status, rxdesc, skb);
421         skb_pull(skb, (stats.rx_drvinfo_size + stats.rx_bufshift));
422         hdr = (struct ieee80211_hdr *)(skb->data);
423         fc = hdr->frame_control;
424         if (!stats.crc) {
425                 memcpy(IEEE80211_SKB_RXCB(skb), &rx_status, sizeof(rx_status));
426
427                 if (is_broadcast_ether_addr(hdr->addr1)) {
428                         /*TODO*/;
429                 } else if (is_multicast_ether_addr(hdr->addr1)) {
430                         /*TODO*/
431                 } else {
432                         unicast = true;
433                         rtlpriv->stats.rxbytesunicast +=  skb->len;
434                 }
435
436                 rtl_is_special_data(hw, skb, false);
437
438                 if (ieee80211_is_data(fc)) {
439                         rtlpriv->cfg->ops->led_control(hw, LED_CTL_RX);
440
441                         if (unicast)
442                                 rtlpriv->link_info.num_rx_inperiod++;
443                 }
444         }
445 }
446
447 static void _rtl_usb_rx_process_noagg(struct ieee80211_hw *hw,
448                                       struct sk_buff *skb)
449 {
450         struct rtl_priv *rtlpriv = rtl_priv(hw);
451         u8 *rxdesc = skb->data;
452         struct ieee80211_hdr *hdr;
453         bool unicast = false;
454         __le16 fc;
455         struct ieee80211_rx_status rx_status = {0};
456         struct rtl_stats stats = {
457                 .signal = 0,
458                 .noise = -98,
459                 .rate = 0,
460         };
461
462         skb_pull(skb, RTL_RX_DESC_SIZE);
463         rtlpriv->cfg->ops->query_rx_desc(hw, &stats, &rx_status, rxdesc, skb);
464         skb_pull(skb, (stats.rx_drvinfo_size + stats.rx_bufshift));
465         hdr = (struct ieee80211_hdr *)(skb->data);
466         fc = hdr->frame_control;
467         if (!stats.crc) {
468                 memcpy(IEEE80211_SKB_RXCB(skb), &rx_status, sizeof(rx_status));
469
470                 if (is_broadcast_ether_addr(hdr->addr1)) {
471                         /*TODO*/;
472                 } else if (is_multicast_ether_addr(hdr->addr1)) {
473                         /*TODO*/
474                 } else {
475                         unicast = true;
476                         rtlpriv->stats.rxbytesunicast +=  skb->len;
477                 }
478
479                 rtl_is_special_data(hw, skb, false);
480
481                 if (ieee80211_is_data(fc)) {
482                         rtlpriv->cfg->ops->led_control(hw, LED_CTL_RX);
483
484                         if (unicast)
485                                 rtlpriv->link_info.num_rx_inperiod++;
486                 }
487                 if (likely(rtl_action_proc(hw, skb, false))) {
488                         struct sk_buff *uskb = NULL;
489                         u8 *pdata;
490
491                         uskb = dev_alloc_skb(skb->len + 128);
492                         if (uskb) {     /* drop packet on allocation failure */
493                                 memcpy(IEEE80211_SKB_RXCB(uskb), &rx_status,
494                                        sizeof(rx_status));
495                                 pdata = (u8 *)skb_put(uskb, skb->len);
496                                 memcpy(pdata, skb->data, skb->len);
497                                 ieee80211_rx_irqsafe(hw, uskb);
498                         }
499                         dev_kfree_skb_any(skb);
500                 } else {
501                         dev_kfree_skb_any(skb);
502                 }
503         }
504 }
505
506 static void _rtl_rx_pre_process(struct ieee80211_hw *hw, struct sk_buff *skb)
507 {
508         struct sk_buff *_skb;
509         struct sk_buff_head rx_queue;
510         struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
511
512         skb_queue_head_init(&rx_queue);
513         if (rtlusb->usb_rx_segregate_hdl)
514                 rtlusb->usb_rx_segregate_hdl(hw, skb, &rx_queue);
515         WARN_ON(skb_queue_empty(&rx_queue));
516         while (!skb_queue_empty(&rx_queue)) {
517                 _skb = skb_dequeue(&rx_queue);
518                 _rtl_usb_rx_process_agg(hw, _skb);
519                 ieee80211_rx_irqsafe(hw, _skb);
520         }
521 }
522
523 static void _rtl_rx_completed(struct urb *_urb)
524 {
525         struct sk_buff *skb = (struct sk_buff *)_urb->context;
526         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
527         struct rtl_usb *rtlusb = (struct rtl_usb *)info->rate_driver_data[0];
528         struct ieee80211_hw *hw = usb_get_intfdata(rtlusb->intf);
529         struct rtl_priv *rtlpriv = rtl_priv(hw);
530         int err = 0;
531
532         if (unlikely(IS_USB_STOP(rtlusb)))
533                 goto free;
534
535         if (likely(0 == _urb->status)) {
536                 /* If this code were moved to work queue, would CPU
537                  * utilization be improved?  NOTE: We shall allocate another skb
538                  * and reuse the original one.
539                  */
540                 skb_put(skb, _urb->actual_length);
541
542                 if (likely(!rtlusb->usb_rx_segregate_hdl)) {
543                         struct sk_buff *_skb;
544                         _rtl_usb_rx_process_noagg(hw, skb);
545                         _skb = _rtl_prep_rx_urb(hw, rtlusb, _urb, GFP_ATOMIC);
546                         if (IS_ERR(_skb)) {
547                                 err = PTR_ERR(_skb);
548                                 RT_TRACE(rtlpriv, COMP_USB, DBG_EMERG,
549                                         ("Can't allocate skb for bulk IN!\n"));
550                                 return;
551                         }
552                         skb = _skb;
553                 } else{
554                         /* TO DO */
555                         _rtl_rx_pre_process(hw, skb);
556                         pr_err("rx agg not supported\n");
557                 }
558                 goto resubmit;
559         }
560
561         switch (_urb->status) {
562         /* disconnect */
563         case -ENOENT:
564         case -ECONNRESET:
565         case -ENODEV:
566         case -ESHUTDOWN:
567                 goto free;
568         default:
569                 break;
570         }
571
572 resubmit:
573         skb_reset_tail_pointer(skb);
574         skb_trim(skb, 0);
575
576         usb_anchor_urb(_urb, &rtlusb->rx_submitted);
577         err = usb_submit_urb(_urb, GFP_ATOMIC);
578         if (unlikely(err)) {
579                 usb_unanchor_urb(_urb);
580                 goto free;
581         }
582         return;
583
584 free:
585         dev_kfree_skb_irq(skb);
586 }
587
588 static int _rtl_usb_receive(struct ieee80211_hw *hw)
589 {
590         struct urb *urb;
591         struct sk_buff *skb;
592         int err;
593         int i;
594         struct rtl_priv *rtlpriv = rtl_priv(hw);
595         struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
596
597         WARN_ON(0 == rtlusb->rx_urb_num);
598         /* 1600 == 1514 + max WLAN header + rtk info */
599         WARN_ON(rtlusb->rx_max_size < 1600);
600
601         for (i = 0; i < rtlusb->rx_urb_num; i++) {
602                 err = -ENOMEM;
603                 urb = usb_alloc_urb(0, GFP_KERNEL);
604                 if (!urb) {
605                         RT_TRACE(rtlpriv, COMP_USB, DBG_EMERG,
606                                  ("Failed to alloc URB!!\n"))
607                         goto err_out;
608                 }
609
610                 skb = _rtl_prep_rx_urb(hw, rtlusb, urb, GFP_KERNEL);
611                 if (IS_ERR(skb)) {
612                         RT_TRACE(rtlpriv, COMP_USB, DBG_EMERG,
613                                  ("Failed to prep_rx_urb!!\n"))
614                         err = PTR_ERR(skb);
615                         goto err_out;
616                 }
617
618                 usb_anchor_urb(urb, &rtlusb->rx_submitted);
619                 err = usb_submit_urb(urb, GFP_KERNEL);
620                 if (err)
621                         goto err_out;
622                 usb_free_urb(urb);
623         }
624         return 0;
625
626 err_out:
627         usb_kill_anchored_urbs(&rtlusb->rx_submitted);
628         return err;
629 }
630
631 static int rtl_usb_start(struct ieee80211_hw *hw)
632 {
633         int err;
634         struct rtl_priv *rtlpriv = rtl_priv(hw);
635         struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
636         struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
637
638         err = rtlpriv->cfg->ops->hw_init(hw);
639         rtl_init_rx_config(hw);
640
641         /* Enable software */
642         SET_USB_START(rtlusb);
643         /* should after adapter start and interrupt enable. */
644         set_hal_start(rtlhal);
645
646         /* Start bulk IN */
647         _rtl_usb_receive(hw);
648
649         return err;
650 }
651 /**
652  *
653  *
654  */
655
656 /*=======================  tx =========================================*/
657 static void rtl_usb_cleanup(struct ieee80211_hw *hw)
658 {
659         u32 i;
660         struct sk_buff *_skb;
661         struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
662         struct ieee80211_tx_info *txinfo;
663
664         SET_USB_STOP(rtlusb);
665
666         /* clean up rx stuff. */
667         usb_kill_anchored_urbs(&rtlusb->rx_submitted);
668
669         /* clean up tx stuff */
670         for (i = 0; i < RTL_USB_MAX_EP_NUM; i++) {
671                 while ((_skb = skb_dequeue(&rtlusb->tx_skb_queue[i]))) {
672                         rtlusb->usb_tx_cleanup(hw, _skb);
673                         txinfo = IEEE80211_SKB_CB(_skb);
674                         ieee80211_tx_info_clear_status(txinfo);
675                         txinfo->flags |= IEEE80211_TX_STAT_ACK;
676                         ieee80211_tx_status_irqsafe(hw, _skb);
677                 }
678                 usb_kill_anchored_urbs(&rtlusb->tx_pending[i]);
679         }
680         usb_kill_anchored_urbs(&rtlusb->tx_submitted);
681 }
682
683 /**
684  *
685  * We may add some struct into struct rtl_usb later. Do deinit here.
686  *
687  */
688 static void rtl_usb_deinit(struct ieee80211_hw *hw)
689 {
690         rtl_usb_cleanup(hw);
691 }
692
693 static void rtl_usb_stop(struct ieee80211_hw *hw)
694 {
695         struct rtl_priv *rtlpriv = rtl_priv(hw);
696         struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
697         struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
698
699         /* should after adapter start and interrupt enable. */
700         set_hal_stop(rtlhal);
701         /* Enable software */
702         SET_USB_STOP(rtlusb);
703         rtl_usb_deinit(hw);
704         rtlpriv->cfg->ops->hw_disable(hw);
705 }
706
707 static void _rtl_submit_tx_urb(struct ieee80211_hw *hw, struct urb *_urb)
708 {
709         int err;
710         struct rtl_priv *rtlpriv = rtl_priv(hw);
711         struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
712
713         usb_anchor_urb(_urb, &rtlusb->tx_submitted);
714         err = usb_submit_urb(_urb, GFP_ATOMIC);
715         if (err < 0) {
716                 struct sk_buff *skb;
717
718                 RT_TRACE(rtlpriv, COMP_USB, DBG_EMERG,
719                          ("Failed to submit urb.\n"));
720                 usb_unanchor_urb(_urb);
721                 skb = (struct sk_buff *)_urb->context;
722                 kfree_skb(skb);
723         }
724         usb_free_urb(_urb);
725 }
726
727 static int _usb_tx_post(struct ieee80211_hw *hw, struct urb *urb,
728                         struct sk_buff *skb)
729 {
730         struct rtl_priv *rtlpriv = rtl_priv(hw);
731         struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
732         struct ieee80211_tx_info *txinfo;
733
734         rtlusb->usb_tx_post_hdl(hw, urb, skb);
735         skb_pull(skb, RTL_TX_HEADER_SIZE);
736         txinfo = IEEE80211_SKB_CB(skb);
737         ieee80211_tx_info_clear_status(txinfo);
738         txinfo->flags |= IEEE80211_TX_STAT_ACK;
739
740         if (urb->status) {
741                 RT_TRACE(rtlpriv, COMP_USB, DBG_EMERG,
742                          ("Urb has error status 0x%X\n", urb->status));
743                 goto out;
744         }
745         /*  TODO:       statistics */
746 out:
747         ieee80211_tx_status_irqsafe(hw, skb);
748         return urb->status;
749 }
750
751 static void _rtl_tx_complete(struct urb *urb)
752 {
753         struct sk_buff *skb = (struct sk_buff *)urb->context;
754         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
755         struct rtl_usb *rtlusb = (struct rtl_usb *)info->rate_driver_data[0];
756         struct ieee80211_hw *hw = usb_get_intfdata(rtlusb->intf);
757         int err;
758
759         if (unlikely(IS_USB_STOP(rtlusb)))
760                 return;
761         err = _usb_tx_post(hw, urb, skb);
762         if (err) {
763                 /* Ignore error and keep issuiing other urbs */
764                 return;
765         }
766 }
767
768 static struct urb *_rtl_usb_tx_urb_setup(struct ieee80211_hw *hw,
769                                 struct sk_buff *skb, u32 ep_num)
770 {
771         struct rtl_priv *rtlpriv = rtl_priv(hw);
772         struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
773         struct urb *_urb;
774
775         WARN_ON(NULL == skb);
776         _urb = usb_alloc_urb(0, GFP_ATOMIC);
777         if (!_urb) {
778                 RT_TRACE(rtlpriv, COMP_USB, DBG_EMERG,
779                          ("Can't allocate URB for bulk out!\n"));
780                 kfree_skb(skb);
781                 return NULL;
782         }
783         _rtl_install_trx_info(rtlusb, skb, ep_num);
784         usb_fill_bulk_urb(_urb, rtlusb->udev, usb_sndbulkpipe(rtlusb->udev,
785                           ep_num), skb->data, skb->len, _rtl_tx_complete, skb);
786         _urb->transfer_flags |= URB_ZERO_PACKET;
787         return _urb;
788 }
789
790 static void _rtl_usb_transmit(struct ieee80211_hw *hw, struct sk_buff *skb,
791                        enum rtl_txq qnum)
792 {
793         struct rtl_priv *rtlpriv = rtl_priv(hw);
794         struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
795         u32 ep_num;
796         struct urb *_urb = NULL;
797         struct sk_buff *_skb = NULL;
798         struct sk_buff_head *skb_list;
799         struct usb_anchor *urb_list;
800
801         WARN_ON(NULL == rtlusb->usb_tx_aggregate_hdl);
802         if (unlikely(IS_USB_STOP(rtlusb))) {
803                 RT_TRACE(rtlpriv, COMP_USB, DBG_EMERG,
804                          ("USB device is stopping...\n"));
805                 kfree_skb(skb);
806                 return;
807         }
808         ep_num = rtlusb->ep_map.ep_mapping[qnum];
809         skb_list = &rtlusb->tx_skb_queue[ep_num];
810         _skb = skb;
811         _urb = _rtl_usb_tx_urb_setup(hw, _skb, ep_num);
812         if (unlikely(!_urb)) {
813                 RT_TRACE(rtlpriv, COMP_ERR, DBG_EMERG,
814                          ("Can't allocate urb. Drop skb!\n"));
815                 kfree_skb(skb);
816                 return;
817         }
818         urb_list = &rtlusb->tx_pending[ep_num];
819         _rtl_submit_tx_urb(hw, _urb);
820 }
821
822 static void _rtl_usb_tx_preprocess(struct ieee80211_hw *hw, struct sk_buff *skb,
823                             u16 hw_queue)
824 {
825         struct rtl_priv *rtlpriv = rtl_priv(hw);
826         struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
827         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
828         struct rtl_tx_desc *pdesc = NULL;
829         struct rtl_tcb_desc tcb_desc;
830         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)(skb->data);
831         __le16 fc = hdr->frame_control;
832         u8 *pda_addr = hdr->addr1;
833         /* ssn */
834         u8 *qc = NULL;
835         u8 tid = 0;
836         u16 seq_number = 0;
837
838         memset(&tcb_desc, 0, sizeof(struct rtl_tcb_desc));
839         if (ieee80211_is_auth(fc)) {
840                 RT_TRACE(rtlpriv, COMP_SEND, DBG_DMESG, ("MAC80211_LINKING\n"));
841                 rtl_ips_nic_on(hw);
842         }
843
844         if (rtlpriv->psc.sw_ps_enabled) {
845                 if (ieee80211_is_data(fc) && !ieee80211_is_nullfunc(fc) &&
846                     !ieee80211_has_pm(fc))
847                         hdr->frame_control |= cpu_to_le16(IEEE80211_FCTL_PM);
848         }
849
850         rtl_action_proc(hw, skb, true);
851         if (is_multicast_ether_addr(pda_addr))
852                 rtlpriv->stats.txbytesmulticast += skb->len;
853         else if (is_broadcast_ether_addr(pda_addr))
854                 rtlpriv->stats.txbytesbroadcast += skb->len;
855         else
856                 rtlpriv->stats.txbytesunicast += skb->len;
857         if (ieee80211_is_data_qos(fc)) {
858                 qc = ieee80211_get_qos_ctl(hdr);
859                 tid = qc[0] & IEEE80211_QOS_CTL_TID_MASK;
860                 seq_number = (le16_to_cpu(hdr->seq_ctrl) &
861                              IEEE80211_SCTL_SEQ) >> 4;
862                 seq_number += 1;
863                 seq_number <<= 4;
864         }
865         rtlpriv->cfg->ops->fill_tx_desc(hw, hdr, (u8 *)pdesc, info, skb,
866                                         hw_queue, &tcb_desc);
867         if (!ieee80211_has_morefrags(hdr->frame_control)) {
868                 if (qc)
869                         mac->tids[tid].seq_number = seq_number;
870         }
871         if (ieee80211_is_data(fc))
872                 rtlpriv->cfg->ops->led_control(hw, LED_CTL_TX);
873 }
874
875 static int rtl_usb_tx(struct ieee80211_hw *hw, struct sk_buff *skb,
876                       struct rtl_tcb_desc *dummy)
877 {
878         struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
879         struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
880         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)(skb->data);
881         __le16 fc = hdr->frame_control;
882         u16 hw_queue;
883
884         if (unlikely(is_hal_stop(rtlhal)))
885                 goto err_free;
886         hw_queue = rtlusb->usb_mq_to_hwq(fc, skb_get_queue_mapping(skb));
887         _rtl_usb_tx_preprocess(hw, skb, hw_queue);
888         _rtl_usb_transmit(hw, skb, hw_queue);
889         return NETDEV_TX_OK;
890
891 err_free:
892         dev_kfree_skb_any(skb);
893         return NETDEV_TX_OK;
894 }
895
896 static bool rtl_usb_tx_chk_waitq_insert(struct ieee80211_hw *hw,
897                                         struct sk_buff *skb)
898 {
899         return false;
900 }
901
902 static struct rtl_intf_ops rtl_usb_ops = {
903         .adapter_start = rtl_usb_start,
904         .adapter_stop = rtl_usb_stop,
905         .adapter_tx = rtl_usb_tx,
906         .waitq_insert = rtl_usb_tx_chk_waitq_insert,
907 };
908
909 int __devinit rtl_usb_probe(struct usb_interface *intf,
910                             const struct usb_device_id *id,
911                             struct rtl_hal_cfg *rtl_hal_cfg)
912 {
913         int err;
914         struct ieee80211_hw *hw = NULL;
915         struct rtl_priv *rtlpriv = NULL;
916         struct usb_device       *udev;
917         struct rtl_usb_priv *usb_priv;
918
919         hw = ieee80211_alloc_hw(sizeof(struct rtl_priv) +
920                                 sizeof(struct rtl_usb_priv), &rtl_ops);
921         if (!hw) {
922                 RT_ASSERT(false, ("%s : ieee80211 alloc failed\n", __func__));
923                 return -ENOMEM;
924         }
925         rtlpriv = hw->priv;
926         rtlpriv->usb_data = kzalloc(RTL_USB_MAX_RX_COUNT * sizeof(u32),
927                                     GFP_KERNEL);
928         if (!rtlpriv->usb_data)
929                 return -ENOMEM;
930
931         /* this spin lock must be initialized early */
932         spin_lock_init(&rtlpriv->locks.usb_lock);
933
934         rtlpriv->usb_data_index = 0;
935         SET_IEEE80211_DEV(hw, &intf->dev);
936         udev = interface_to_usbdev(intf);
937         usb_get_dev(udev);
938         usb_priv = rtl_usbpriv(hw);
939         memset(usb_priv, 0, sizeof(*usb_priv));
940         usb_priv->dev.intf = intf;
941         usb_priv->dev.udev = udev;
942         usb_set_intfdata(intf, hw);
943         /* init cfg & intf_ops */
944         rtlpriv->rtlhal.interface = INTF_USB;
945         rtlpriv->cfg = rtl_hal_cfg;
946         rtlpriv->intf_ops = &rtl_usb_ops;
947         rtl_dbgp_flag_init(hw);
948         /* Init IO handler */
949         _rtl_usb_io_handler_init(&udev->dev, hw);
950         rtlpriv->cfg->ops->read_chip_version(hw);
951         /*like read eeprom and so on */
952         rtlpriv->cfg->ops->read_eeprom_info(hw);
953         if (rtlpriv->cfg->ops->init_sw_vars(hw)) {
954                 RT_TRACE(rtlpriv, COMP_ERR, DBG_EMERG,
955                          ("Can't init_sw_vars.\n"));
956                 goto error_out;
957         }
958         rtlpriv->cfg->ops->init_sw_leds(hw);
959         err = _rtl_usb_init(hw);
960         err = _rtl_usb_init_sw(hw);
961         /* Init mac80211 sw */
962         err = rtl_init_core(hw);
963         if (err) {
964                 RT_TRACE(rtlpriv, COMP_ERR, DBG_EMERG,
965                          ("Can't allocate sw for mac80211.\n"));
966                 goto error_out;
967         }
968
969         /*init rfkill */
970         /* rtl_init_rfkill(hw); */
971
972         err = ieee80211_register_hw(hw);
973         if (err) {
974                 RT_TRACE(rtlpriv, COMP_INIT, DBG_EMERG,
975                          ("Can't register mac80211 hw.\n"));
976                 goto error_out;
977         } else {
978                 rtlpriv->mac80211.mac80211_registered = 1;
979         }
980         set_bit(RTL_STATUS_INTERFACE_START, &rtlpriv->status);
981         return 0;
982 error_out:
983         rtl_deinit_core(hw);
984         _rtl_usb_io_handler_release(hw);
985         ieee80211_free_hw(hw);
986         usb_put_dev(udev);
987         return -ENODEV;
988 }
989 EXPORT_SYMBOL(rtl_usb_probe);
990
991 void rtl_usb_disconnect(struct usb_interface *intf)
992 {
993         struct ieee80211_hw *hw = usb_get_intfdata(intf);
994         struct rtl_priv *rtlpriv = rtl_priv(hw);
995         struct rtl_mac *rtlmac = rtl_mac(rtl_priv(hw));
996         struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
997
998         if (unlikely(!rtlpriv))
999                 return;
1000         /*ieee80211_unregister_hw will call ops_stop */
1001         if (rtlmac->mac80211_registered == 1) {
1002                 ieee80211_unregister_hw(hw);
1003                 rtlmac->mac80211_registered = 0;
1004         } else {
1005                 rtl_deinit_deferred_work(hw);
1006                 rtlpriv->intf_ops->adapter_stop(hw);
1007         }
1008         /*deinit rfkill */
1009         /* rtl_deinit_rfkill(hw); */
1010         rtl_usb_deinit(hw);
1011         rtl_deinit_core(hw);
1012         kfree(rtlpriv->usb_data);
1013         rtlpriv->cfg->ops->deinit_sw_leds(hw);
1014         rtlpriv->cfg->ops->deinit_sw_vars(hw);
1015         _rtl_usb_io_handler_release(hw);
1016         usb_put_dev(rtlusb->udev);
1017         usb_set_intfdata(intf, NULL);
1018         ieee80211_free_hw(hw);
1019 }
1020 EXPORT_SYMBOL(rtl_usb_disconnect);
1021
1022 int rtl_usb_suspend(struct usb_interface *pusb_intf, pm_message_t message)
1023 {
1024         return 0;
1025 }
1026 EXPORT_SYMBOL(rtl_usb_suspend);
1027
1028 int rtl_usb_resume(struct usb_interface *pusb_intf)
1029 {
1030         return 0;
1031 }
1032 EXPORT_SYMBOL(rtl_usb_resume);