Merge tag 'block-5.11-2021-01-10' of git://git.kernel.dk/linux-block
[linux/fpc-iii.git] / drivers / net / wireless / realtek / rtlwifi / usb.c
blobd62b87f010c9f0b4e3d07d15383611f094ca7ae1
1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright(c) 2009-2012 Realtek Corporation.*/
4 #include "wifi.h"
5 #include "core.h"
6 #include "usb.h"
7 #include "base.h"
8 #include "ps.h"
9 #include "rtl8192c/fw_common.h"
10 #include <linux/export.h>
11 #include <linux/module.h>
13 MODULE_AUTHOR("lizhaoming <chaoming_li@realsil.com.cn>");
14 MODULE_AUTHOR("Realtek WlanFAE <wlanfae@realtek.com>");
15 MODULE_AUTHOR("Larry Finger <Larry.FInger@lwfinger.net>");
16 MODULE_LICENSE("GPL");
17 MODULE_DESCRIPTION("USB basic driver for rtlwifi");
19 #define REALTEK_USB_VENQT_READ 0xC0
20 #define REALTEK_USB_VENQT_WRITE 0x40
21 #define REALTEK_USB_VENQT_CMD_REQ 0x05
22 #define REALTEK_USB_VENQT_CMD_IDX 0x00
24 #define MAX_USBCTRL_VENDORREQ_TIMES 10
26 static void usbctrl_async_callback(struct urb *urb)
28 if (urb) {
29 /* free dr */
30 kfree(urb->setup_packet);
31 /* free databuf */
32 kfree(urb->transfer_buffer);
36 static int _usbctrl_vendorreq_async_write(struct usb_device *udev, u8 request,
37 u16 value, u16 index, void *pdata,
38 u16 len)
40 int rc;
41 unsigned int pipe;
42 u8 reqtype;
43 struct usb_ctrlrequest *dr;
44 struct urb *urb;
45 const u16 databuf_maxlen = REALTEK_USB_VENQT_MAX_BUF_SIZE;
46 u8 *databuf;
48 if (WARN_ON_ONCE(len > databuf_maxlen))
49 len = databuf_maxlen;
51 pipe = usb_sndctrlpipe(udev, 0); /* write_out */
52 reqtype = REALTEK_USB_VENQT_WRITE;
54 dr = kzalloc(sizeof(*dr), GFP_ATOMIC);
55 if (!dr)
56 return -ENOMEM;
58 databuf = kzalloc(databuf_maxlen, GFP_ATOMIC);
59 if (!databuf) {
60 kfree(dr);
61 return -ENOMEM;
64 urb = usb_alloc_urb(0, GFP_ATOMIC);
65 if (!urb) {
66 kfree(databuf);
67 kfree(dr);
68 return -ENOMEM;
71 dr->bRequestType = reqtype;
72 dr->bRequest = request;
73 dr->wValue = cpu_to_le16(value);
74 dr->wIndex = cpu_to_le16(index);
75 dr->wLength = cpu_to_le16(len);
76 /* data are already in little-endian order */
77 memcpy(databuf, pdata, len);
78 usb_fill_control_urb(urb, udev, pipe,
79 (unsigned char *)dr, databuf, len,
80 usbctrl_async_callback, NULL);
81 rc = usb_submit_urb(urb, GFP_ATOMIC);
82 if (rc < 0) {
83 kfree(databuf);
84 kfree(dr);
86 usb_free_urb(urb);
87 return rc;
90 static int _usbctrl_vendorreq_sync_read(struct usb_device *udev, u8 request,
91 u16 value, u16 index, void *pdata,
92 u16 len)
94 unsigned int pipe;
95 int status;
96 u8 reqtype;
97 int vendorreq_times = 0;
98 static int count;
100 pipe = usb_rcvctrlpipe(udev, 0); /* read_in */
101 reqtype = REALTEK_USB_VENQT_READ;
103 do {
104 status = usb_control_msg(udev, pipe, request, reqtype, value,
105 index, pdata, len, 1000);
106 if (status < 0) {
107 /* firmware download is checksumed, don't retry */
108 if ((value >= FW_8192C_START_ADDRESS &&
109 value <= FW_8192C_END_ADDRESS))
110 break;
111 } else {
112 break;
114 } while (++vendorreq_times < MAX_USBCTRL_VENDORREQ_TIMES);
116 if (status < 0 && count++ < 4)
117 pr_err("reg 0x%x, usbctrl_vendorreq TimeOut! status:0x%x value=0x%x\n",
118 value, status, *(u32 *)pdata);
119 return status;
122 static u32 _usb_read_sync(struct rtl_priv *rtlpriv, u32 addr, u16 len)
124 struct device *dev = rtlpriv->io.dev;
125 struct usb_device *udev = to_usb_device(dev);
126 u8 request;
127 u16 wvalue;
128 u16 index;
129 __le32 *data;
130 unsigned long flags;
132 spin_lock_irqsave(&rtlpriv->locks.usb_lock, flags);
133 if (++rtlpriv->usb_data_index >= RTL_USB_MAX_RX_COUNT)
134 rtlpriv->usb_data_index = 0;
135 data = &rtlpriv->usb_data[rtlpriv->usb_data_index];
136 spin_unlock_irqrestore(&rtlpriv->locks.usb_lock, flags);
137 request = REALTEK_USB_VENQT_CMD_REQ;
138 index = REALTEK_USB_VENQT_CMD_IDX; /* n/a */
140 wvalue = (u16)addr;
141 _usbctrl_vendorreq_sync_read(udev, request, wvalue, index, data, len);
142 return le32_to_cpu(*data);
145 static u8 _usb_read8_sync(struct rtl_priv *rtlpriv, u32 addr)
147 return (u8)_usb_read_sync(rtlpriv, addr, 1);
150 static u16 _usb_read16_sync(struct rtl_priv *rtlpriv, u32 addr)
152 return (u16)_usb_read_sync(rtlpriv, addr, 2);
155 static u32 _usb_read32_sync(struct rtl_priv *rtlpriv, u32 addr)
157 return _usb_read_sync(rtlpriv, addr, 4);
160 static void _usb_write_async(struct usb_device *udev, u32 addr, u32 val,
161 u16 len)
163 u8 request;
164 u16 wvalue;
165 u16 index;
166 __le32 data;
168 request = REALTEK_USB_VENQT_CMD_REQ;
169 index = REALTEK_USB_VENQT_CMD_IDX; /* n/a */
170 wvalue = (u16)(addr&0x0000ffff);
171 data = cpu_to_le32(val);
172 _usbctrl_vendorreq_async_write(udev, request, wvalue, index, &data,
173 len);
176 static void _usb_write8_async(struct rtl_priv *rtlpriv, u32 addr, u8 val)
178 struct device *dev = rtlpriv->io.dev;
180 _usb_write_async(to_usb_device(dev), addr, val, 1);
183 static void _usb_write16_async(struct rtl_priv *rtlpriv, u32 addr, u16 val)
185 struct device *dev = rtlpriv->io.dev;
187 _usb_write_async(to_usb_device(dev), addr, val, 2);
190 static void _usb_write32_async(struct rtl_priv *rtlpriv, u32 addr, u32 val)
192 struct device *dev = rtlpriv->io.dev;
194 _usb_write_async(to_usb_device(dev), addr, val, 4);
197 static void _usb_writen_sync(struct rtl_priv *rtlpriv, u32 addr, void *data,
198 u16 len)
200 struct device *dev = rtlpriv->io.dev;
201 struct usb_device *udev = to_usb_device(dev);
202 u8 request = REALTEK_USB_VENQT_CMD_REQ;
203 u8 reqtype = REALTEK_USB_VENQT_WRITE;
204 u16 wvalue;
205 u16 index = REALTEK_USB_VENQT_CMD_IDX;
206 int pipe = usb_sndctrlpipe(udev, 0); /* write_out */
207 u8 *buffer;
209 wvalue = (u16)(addr & 0x0000ffff);
210 buffer = kmemdup(data, len, GFP_ATOMIC);
211 if (!buffer)
212 return;
213 usb_control_msg(udev, pipe, request, reqtype, wvalue,
214 index, buffer, len, 50);
216 kfree(buffer);
219 static void _rtl_usb_io_handler_init(struct device *dev,
220 struct ieee80211_hw *hw)
222 struct rtl_priv *rtlpriv = rtl_priv(hw);
224 rtlpriv->io.dev = dev;
225 mutex_init(&rtlpriv->io.bb_mutex);
226 rtlpriv->io.write8_async = _usb_write8_async;
227 rtlpriv->io.write16_async = _usb_write16_async;
228 rtlpriv->io.write32_async = _usb_write32_async;
229 rtlpriv->io.read8_sync = _usb_read8_sync;
230 rtlpriv->io.read16_sync = _usb_read16_sync;
231 rtlpriv->io.read32_sync = _usb_read32_sync;
232 rtlpriv->io.writen_sync = _usb_writen_sync;
235 static void _rtl_usb_io_handler_release(struct ieee80211_hw *hw)
237 struct rtl_priv __maybe_unused *rtlpriv = rtl_priv(hw);
239 mutex_destroy(&rtlpriv->io.bb_mutex);
242 /* Default aggregation handler. Do nothing and just return the oldest skb. */
243 static struct sk_buff *_none_usb_tx_aggregate_hdl(struct ieee80211_hw *hw,
244 struct sk_buff_head *list)
246 return skb_dequeue(list);
249 #define IS_HIGH_SPEED_USB(udev) \
250 ((USB_SPEED_HIGH == (udev)->speed) ? true : false)
252 static int _rtl_usb_init_tx(struct ieee80211_hw *hw)
254 u32 i;
255 struct rtl_priv *rtlpriv = rtl_priv(hw);
256 struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
258 rtlusb->max_bulk_out_size = IS_HIGH_SPEED_USB(rtlusb->udev)
259 ? USB_HIGH_SPEED_BULK_SIZE
260 : USB_FULL_SPEED_BULK_SIZE;
262 rtl_dbg(rtlpriv, COMP_INIT, DBG_DMESG, "USB Max Bulk-out Size=%d\n",
263 rtlusb->max_bulk_out_size);
265 for (i = 0; i < __RTL_TXQ_NUM; i++) {
266 u32 ep_num = rtlusb->ep_map.ep_mapping[i];
268 if (!ep_num) {
269 rtl_dbg(rtlpriv, COMP_INIT, DBG_DMESG,
270 "Invalid endpoint map setting!\n");
271 return -EINVAL;
275 rtlusb->usb_tx_post_hdl =
276 rtlpriv->cfg->usb_interface_cfg->usb_tx_post_hdl;
277 rtlusb->usb_tx_cleanup =
278 rtlpriv->cfg->usb_interface_cfg->usb_tx_cleanup;
279 rtlusb->usb_tx_aggregate_hdl =
280 (rtlpriv->cfg->usb_interface_cfg->usb_tx_aggregate_hdl)
281 ? rtlpriv->cfg->usb_interface_cfg->usb_tx_aggregate_hdl
282 : &_none_usb_tx_aggregate_hdl;
284 init_usb_anchor(&rtlusb->tx_submitted);
285 for (i = 0; i < RTL_USB_MAX_EP_NUM; i++) {
286 skb_queue_head_init(&rtlusb->tx_skb_queue[i]);
287 init_usb_anchor(&rtlusb->tx_pending[i]);
289 return 0;
292 static void _rtl_rx_work(struct tasklet_struct *t);
294 static int _rtl_usb_init_rx(struct ieee80211_hw *hw)
296 struct rtl_priv *rtlpriv = rtl_priv(hw);
297 struct rtl_usb_priv *usb_priv = rtl_usbpriv(hw);
298 struct rtl_usb *rtlusb = rtl_usbdev(usb_priv);
300 rtlusb->rx_max_size = rtlpriv->cfg->usb_interface_cfg->rx_max_size;
301 rtlusb->rx_urb_num = rtlpriv->cfg->usb_interface_cfg->rx_urb_num;
302 rtlusb->in_ep = rtlpriv->cfg->usb_interface_cfg->in_ep_num;
303 rtlusb->usb_rx_hdl = rtlpriv->cfg->usb_interface_cfg->usb_rx_hdl;
304 rtlusb->usb_rx_segregate_hdl =
305 rtlpriv->cfg->usb_interface_cfg->usb_rx_segregate_hdl;
307 pr_info("rx_max_size %d, rx_urb_num %d, in_ep %d\n",
308 rtlusb->rx_max_size, rtlusb->rx_urb_num, rtlusb->in_ep);
309 init_usb_anchor(&rtlusb->rx_submitted);
310 init_usb_anchor(&rtlusb->rx_cleanup_urbs);
312 skb_queue_head_init(&rtlusb->rx_queue);
313 rtlusb->rx_work_tasklet.func = (void(*))_rtl_rx_work;
314 rtlusb->rx_work_tasklet.data = (unsigned long)&rtlusb->rx_work_tasklet;
316 return 0;
319 static int _rtl_usb_init(struct ieee80211_hw *hw)
321 struct rtl_priv *rtlpriv = rtl_priv(hw);
322 struct rtl_usb_priv *usb_priv = rtl_usbpriv(hw);
323 struct rtl_usb *rtlusb = rtl_usbdev(usb_priv);
324 int err;
325 u8 epidx;
326 struct usb_interface *usb_intf = rtlusb->intf;
327 u8 epnums = usb_intf->cur_altsetting->desc.bNumEndpoints;
329 rtlusb->out_ep_nums = rtlusb->in_ep_nums = 0;
330 for (epidx = 0; epidx < epnums; epidx++) {
331 struct usb_endpoint_descriptor *pep_desc;
333 pep_desc = &usb_intf->cur_altsetting->endpoint[epidx].desc;
335 if (usb_endpoint_dir_in(pep_desc))
336 rtlusb->in_ep_nums++;
337 else if (usb_endpoint_dir_out(pep_desc))
338 rtlusb->out_ep_nums++;
340 rtl_dbg(rtlpriv, COMP_INIT, DBG_DMESG,
341 "USB EP(0x%02x), MaxPacketSize=%d, Interval=%d\n",
342 pep_desc->bEndpointAddress, pep_desc->wMaxPacketSize,
343 pep_desc->bInterval);
345 if (rtlusb->in_ep_nums < rtlpriv->cfg->usb_interface_cfg->in_ep_num) {
346 pr_err("Too few input end points found\n");
347 return -EINVAL;
349 if (rtlusb->out_ep_nums == 0) {
350 pr_err("No output end points found\n");
351 return -EINVAL;
353 /* usb endpoint mapping */
354 err = rtlpriv->cfg->usb_interface_cfg->usb_endpoint_mapping(hw);
355 rtlusb->usb_mq_to_hwq = rtlpriv->cfg->usb_interface_cfg->usb_mq_to_hwq;
356 _rtl_usb_init_tx(hw);
357 _rtl_usb_init_rx(hw);
358 return err;
361 static void rtl_usb_init_sw(struct ieee80211_hw *hw)
363 struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
364 struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
365 struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
366 struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
368 rtlhal->hw = hw;
369 ppsc->inactiveps = false;
370 ppsc->leisure_ps = false;
371 ppsc->fwctrl_lps = false;
372 ppsc->reg_fwctrl_lps = 3;
373 ppsc->reg_max_lps_awakeintvl = 5;
374 ppsc->fwctrl_psmode = FW_PS_DTIM_MODE;
376 /* IBSS */
377 mac->beacon_interval = 100;
379 /* AMPDU */
380 mac->min_space_cfg = 0;
381 mac->max_mss_density = 0;
383 /* set sane AMPDU defaults */
384 mac->current_ampdu_density = 7;
385 mac->current_ampdu_factor = 3;
387 /* QOS */
388 rtlusb->acm_method = EACMWAY2_SW;
390 /* IRQ */
391 /* HIMR - turn all on */
392 rtlusb->irq_mask[0] = 0xFFFFFFFF;
393 /* HIMR_EX - turn all on */
394 rtlusb->irq_mask[1] = 0xFFFFFFFF;
395 rtlusb->disablehwsm = true;
398 static void _rtl_rx_completed(struct urb *urb);
400 static int _rtl_prep_rx_urb(struct ieee80211_hw *hw, struct rtl_usb *rtlusb,
401 struct urb *urb, gfp_t gfp_mask)
403 void *buf;
405 buf = usb_alloc_coherent(rtlusb->udev, rtlusb->rx_max_size, gfp_mask,
406 &urb->transfer_dma);
407 if (!buf) {
408 pr_err("Failed to usb_alloc_coherent!!\n");
409 return -ENOMEM;
412 usb_fill_bulk_urb(urb, rtlusb->udev,
413 usb_rcvbulkpipe(rtlusb->udev, rtlusb->in_ep),
414 buf, rtlusb->rx_max_size, _rtl_rx_completed, rtlusb);
415 urb->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
417 return 0;
420 static void _rtl_usb_rx_process_agg(struct ieee80211_hw *hw,
421 struct sk_buff *skb)
423 struct rtl_priv *rtlpriv = rtl_priv(hw);
424 u8 *rxdesc = skb->data;
425 struct ieee80211_hdr *hdr;
426 bool unicast = false;
427 __le16 fc;
428 struct ieee80211_rx_status rx_status = {0};
429 struct rtl_stats stats = {
430 .signal = 0,
431 .rate = 0,
434 skb_pull(skb, RTL_RX_DESC_SIZE);
435 rtlpriv->cfg->ops->query_rx_desc(hw, &stats, &rx_status, rxdesc, skb);
436 skb_pull(skb, (stats.rx_drvinfo_size + stats.rx_bufshift));
437 hdr = (struct ieee80211_hdr *)(skb->data);
438 fc = hdr->frame_control;
439 if (!stats.crc) {
440 memcpy(IEEE80211_SKB_RXCB(skb), &rx_status, sizeof(rx_status));
442 if (is_broadcast_ether_addr(hdr->addr1)) {
443 /*TODO*/;
444 } else if (is_multicast_ether_addr(hdr->addr1)) {
445 /*TODO*/
446 } else {
447 unicast = true;
448 rtlpriv->stats.rxbytesunicast += skb->len;
451 if (ieee80211_is_data(fc)) {
452 rtlpriv->cfg->ops->led_control(hw, LED_CTL_RX);
454 if (unicast)
455 rtlpriv->link_info.num_rx_inperiod++;
457 /* static bcn for roaming */
458 rtl_beacon_statistic(hw, skb);
462 static void _rtl_usb_rx_process_noagg(struct ieee80211_hw *hw,
463 struct sk_buff *skb)
465 struct rtl_priv *rtlpriv = rtl_priv(hw);
466 u8 *rxdesc = skb->data;
467 struct ieee80211_hdr *hdr;
468 bool unicast = false;
469 __le16 fc;
470 struct ieee80211_rx_status rx_status = {0};
471 struct rtl_stats stats = {
472 .signal = 0,
473 .rate = 0,
476 skb_pull(skb, RTL_RX_DESC_SIZE);
477 rtlpriv->cfg->ops->query_rx_desc(hw, &stats, &rx_status, rxdesc, skb);
478 skb_pull(skb, (stats.rx_drvinfo_size + stats.rx_bufshift));
479 hdr = (struct ieee80211_hdr *)(skb->data);
480 fc = hdr->frame_control;
481 if (!stats.crc) {
482 memcpy(IEEE80211_SKB_RXCB(skb), &rx_status, sizeof(rx_status));
484 if (is_broadcast_ether_addr(hdr->addr1)) {
485 /*TODO*/;
486 } else if (is_multicast_ether_addr(hdr->addr1)) {
487 /*TODO*/
488 } else {
489 unicast = true;
490 rtlpriv->stats.rxbytesunicast += skb->len;
493 if (ieee80211_is_data(fc)) {
494 rtlpriv->cfg->ops->led_control(hw, LED_CTL_RX);
496 if (unicast)
497 rtlpriv->link_info.num_rx_inperiod++;
500 /* static bcn for roaming */
501 rtl_beacon_statistic(hw, skb);
503 if (likely(rtl_action_proc(hw, skb, false)))
504 ieee80211_rx(hw, skb);
505 else
506 dev_kfree_skb_any(skb);
507 } else {
508 dev_kfree_skb_any(skb);
512 static void _rtl_rx_pre_process(struct ieee80211_hw *hw, struct sk_buff *skb)
514 struct sk_buff *_skb;
515 struct sk_buff_head rx_queue;
516 struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
518 skb_queue_head_init(&rx_queue);
519 if (rtlusb->usb_rx_segregate_hdl)
520 rtlusb->usb_rx_segregate_hdl(hw, skb, &rx_queue);
521 WARN_ON(skb_queue_empty(&rx_queue));
522 while (!skb_queue_empty(&rx_queue)) {
523 _skb = skb_dequeue(&rx_queue);
524 _rtl_usb_rx_process_agg(hw, _skb);
525 ieee80211_rx(hw, _skb);
529 #define __RX_SKB_MAX_QUEUED 64
531 static void _rtl_rx_work(struct tasklet_struct *t)
533 struct rtl_usb *rtlusb = from_tasklet(rtlusb, t, rx_work_tasklet);
534 struct ieee80211_hw *hw = usb_get_intfdata(rtlusb->intf);
535 struct sk_buff *skb;
537 while ((skb = skb_dequeue(&rtlusb->rx_queue))) {
538 if (unlikely(IS_USB_STOP(rtlusb))) {
539 dev_kfree_skb_any(skb);
540 continue;
543 if (likely(!rtlusb->usb_rx_segregate_hdl)) {
544 _rtl_usb_rx_process_noagg(hw, skb);
545 } else {
546 /* TO DO */
547 _rtl_rx_pre_process(hw, skb);
548 pr_err("rx agg not supported\n");
553 static unsigned int _rtl_rx_get_padding(struct ieee80211_hdr *hdr,
554 unsigned int len)
556 #if NET_IP_ALIGN != 0
557 unsigned int padding = 0;
558 #endif
560 /* make function no-op when possible */
561 if (NET_IP_ALIGN == 0 || len < sizeof(*hdr))
562 return 0;
564 #if NET_IP_ALIGN != 0
565 /* alignment calculation as in lbtf_rx() / carl9170_rx_copy_data() */
566 /* TODO: deduplicate common code, define helper function instead? */
568 if (ieee80211_is_data_qos(hdr->frame_control)) {
569 u8 *qc = ieee80211_get_qos_ctl(hdr);
571 padding ^= NET_IP_ALIGN;
573 /* Input might be invalid, avoid accessing memory outside
574 * the buffer.
576 if ((unsigned long)qc - (unsigned long)hdr < len &&
577 *qc & IEEE80211_QOS_CTL_A_MSDU_PRESENT)
578 padding ^= NET_IP_ALIGN;
581 if (ieee80211_has_a4(hdr->frame_control))
582 padding ^= NET_IP_ALIGN;
584 return padding;
585 #endif
588 #define __RADIO_TAP_SIZE_RSV 32
590 static void _rtl_rx_completed(struct urb *_urb)
592 struct rtl_usb *rtlusb = (struct rtl_usb *)_urb->context;
593 int err = 0;
595 if (unlikely(IS_USB_STOP(rtlusb)))
596 goto free;
598 if (likely(0 == _urb->status)) {
599 unsigned int padding;
600 struct sk_buff *skb;
601 unsigned int qlen;
602 unsigned int size = _urb->actual_length;
603 struct ieee80211_hdr *hdr;
605 if (size < RTL_RX_DESC_SIZE + sizeof(struct ieee80211_hdr)) {
606 pr_err("Too short packet from bulk IN! (len: %d)\n",
607 size);
608 goto resubmit;
611 qlen = skb_queue_len(&rtlusb->rx_queue);
612 if (qlen >= __RX_SKB_MAX_QUEUED) {
613 pr_err("Pending RX skbuff queue full! (qlen: %d)\n",
614 qlen);
615 goto resubmit;
618 hdr = (void *)(_urb->transfer_buffer + RTL_RX_DESC_SIZE);
619 padding = _rtl_rx_get_padding(hdr, size - RTL_RX_DESC_SIZE);
621 skb = dev_alloc_skb(size + __RADIO_TAP_SIZE_RSV + padding);
622 if (!skb) {
623 pr_err("Can't allocate skb for bulk IN!\n");
624 goto resubmit;
627 _rtl_install_trx_info(rtlusb, skb, rtlusb->in_ep);
629 /* Make sure the payload data is 4 byte aligned. */
630 skb_reserve(skb, padding);
632 /* reserve some space for mac80211's radiotap */
633 skb_reserve(skb, __RADIO_TAP_SIZE_RSV);
635 skb_put_data(skb, _urb->transfer_buffer, size);
637 skb_queue_tail(&rtlusb->rx_queue, skb);
638 tasklet_schedule(&rtlusb->rx_work_tasklet);
640 goto resubmit;
643 switch (_urb->status) {
644 /* disconnect */
645 case -ENOENT:
646 case -ECONNRESET:
647 case -ENODEV:
648 case -ESHUTDOWN:
649 goto free;
650 default:
651 break;
654 resubmit:
655 usb_anchor_urb(_urb, &rtlusb->rx_submitted);
656 err = usb_submit_urb(_urb, GFP_ATOMIC);
657 if (unlikely(err)) {
658 usb_unanchor_urb(_urb);
659 goto free;
661 return;
663 free:
664 /* On some architectures, usb_free_coherent must not be called from
665 * hardirq context. Queue urb to cleanup list.
667 usb_anchor_urb(_urb, &rtlusb->rx_cleanup_urbs);
670 #undef __RADIO_TAP_SIZE_RSV
672 static void _rtl_usb_cleanup_rx(struct ieee80211_hw *hw)
674 struct rtl_priv *rtlpriv = rtl_priv(hw);
675 struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
676 struct urb *urb;
678 usb_kill_anchored_urbs(&rtlusb->rx_submitted);
680 tasklet_kill(&rtlusb->rx_work_tasklet);
681 cancel_work_sync(&rtlpriv->works.lps_change_work);
683 if (rtlpriv->works.rtl_wq) {
684 destroy_workqueue(rtlpriv->works.rtl_wq);
685 rtlpriv->works.rtl_wq = NULL;
688 skb_queue_purge(&rtlusb->rx_queue);
690 while ((urb = usb_get_from_anchor(&rtlusb->rx_cleanup_urbs))) {
691 usb_free_coherent(urb->dev, urb->transfer_buffer_length,
692 urb->transfer_buffer, urb->transfer_dma);
693 usb_free_urb(urb);
697 static int _rtl_usb_receive(struct ieee80211_hw *hw)
699 struct urb *urb;
700 int err;
701 int i;
702 struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
704 WARN_ON(0 == rtlusb->rx_urb_num);
705 /* 1600 == 1514 + max WLAN header + rtk info */
706 WARN_ON(rtlusb->rx_max_size < 1600);
708 for (i = 0; i < rtlusb->rx_urb_num; i++) {
709 err = -ENOMEM;
710 urb = usb_alloc_urb(0, GFP_KERNEL);
711 if (!urb)
712 goto err_out;
714 err = _rtl_prep_rx_urb(hw, rtlusb, urb, GFP_KERNEL);
715 if (err < 0) {
716 pr_err("Failed to prep_rx_urb!!\n");
717 usb_free_urb(urb);
718 goto err_out;
721 usb_anchor_urb(urb, &rtlusb->rx_submitted);
722 err = usb_submit_urb(urb, GFP_KERNEL);
723 if (err) {
724 usb_unanchor_urb(urb);
725 usb_free_urb(urb);
726 goto err_out;
728 usb_free_urb(urb);
730 return 0;
732 err_out:
733 usb_kill_anchored_urbs(&rtlusb->rx_submitted);
734 return err;
737 static int rtl_usb_start(struct ieee80211_hw *hw)
739 int err;
740 struct rtl_priv *rtlpriv = rtl_priv(hw);
741 struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
742 struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
744 err = rtlpriv->cfg->ops->hw_init(hw);
745 if (!err) {
746 rtl_init_rx_config(hw);
748 /* Enable software */
749 SET_USB_START(rtlusb);
750 /* should after adapter start and interrupt enable. */
751 set_hal_start(rtlhal);
753 /* Start bulk IN */
754 err = _rtl_usb_receive(hw);
757 return err;
760 /*======================= tx =========================================*/
761 static void rtl_usb_cleanup(struct ieee80211_hw *hw)
763 u32 i;
764 struct sk_buff *_skb;
765 struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
766 struct ieee80211_tx_info *txinfo;
768 /* clean up rx stuff. */
769 _rtl_usb_cleanup_rx(hw);
771 /* clean up tx stuff */
772 for (i = 0; i < RTL_USB_MAX_EP_NUM; i++) {
773 while ((_skb = skb_dequeue(&rtlusb->tx_skb_queue[i]))) {
774 rtlusb->usb_tx_cleanup(hw, _skb);
775 txinfo = IEEE80211_SKB_CB(_skb);
776 ieee80211_tx_info_clear_status(txinfo);
777 txinfo->flags |= IEEE80211_TX_STAT_ACK;
778 ieee80211_tx_status_irqsafe(hw, _skb);
780 usb_kill_anchored_urbs(&rtlusb->tx_pending[i]);
782 usb_kill_anchored_urbs(&rtlusb->tx_submitted);
785 /* We may add some struct into struct rtl_usb later. Do deinit here. */
786 static void rtl_usb_deinit(struct ieee80211_hw *hw)
788 rtl_usb_cleanup(hw);
791 static void rtl_usb_stop(struct ieee80211_hw *hw)
793 struct rtl_priv *rtlpriv = rtl_priv(hw);
794 struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
795 struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
796 struct urb *urb;
798 /* should after adapter start and interrupt enable. */
799 set_hal_stop(rtlhal);
800 cancel_work_sync(&rtlpriv->works.fill_h2c_cmd);
801 /* Enable software */
802 SET_USB_STOP(rtlusb);
804 /* free pre-allocated URBs from rtl_usb_start() */
805 usb_kill_anchored_urbs(&rtlusb->rx_submitted);
807 tasklet_kill(&rtlusb->rx_work_tasklet);
808 cancel_work_sync(&rtlpriv->works.lps_change_work);
810 flush_workqueue(rtlpriv->works.rtl_wq);
812 skb_queue_purge(&rtlusb->rx_queue);
814 while ((urb = usb_get_from_anchor(&rtlusb->rx_cleanup_urbs))) {
815 usb_free_coherent(urb->dev, urb->transfer_buffer_length,
816 urb->transfer_buffer, urb->transfer_dma);
817 usb_free_urb(urb);
820 rtlpriv->cfg->ops->hw_disable(hw);
823 static void _rtl_submit_tx_urb(struct ieee80211_hw *hw, struct urb *_urb)
825 int err;
826 struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
828 usb_anchor_urb(_urb, &rtlusb->tx_submitted);
829 err = usb_submit_urb(_urb, GFP_ATOMIC);
830 if (err < 0) {
831 struct sk_buff *skb;
833 pr_err("Failed to submit urb\n");
834 usb_unanchor_urb(_urb);
835 skb = (struct sk_buff *)_urb->context;
836 kfree_skb(skb);
838 usb_free_urb(_urb);
841 static int _usb_tx_post(struct ieee80211_hw *hw, struct urb *urb,
842 struct sk_buff *skb)
844 struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
845 struct ieee80211_tx_info *txinfo;
847 rtlusb->usb_tx_post_hdl(hw, urb, skb);
848 skb_pull(skb, RTL_TX_HEADER_SIZE);
849 txinfo = IEEE80211_SKB_CB(skb);
850 ieee80211_tx_info_clear_status(txinfo);
851 txinfo->flags |= IEEE80211_TX_STAT_ACK;
853 if (urb->status) {
854 pr_err("Urb has error status 0x%X\n", urb->status);
855 goto out;
857 /* TODO: statistics */
858 out:
859 ieee80211_tx_status_irqsafe(hw, skb);
860 return urb->status;
863 static void _rtl_tx_complete(struct urb *urb)
865 struct sk_buff *skb = (struct sk_buff *)urb->context;
866 struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
867 struct rtl_usb *rtlusb = (struct rtl_usb *)info->rate_driver_data[0];
868 struct ieee80211_hw *hw = usb_get_intfdata(rtlusb->intf);
869 int err;
871 if (unlikely(IS_USB_STOP(rtlusb)))
872 return;
873 err = _usb_tx_post(hw, urb, skb);
874 if (err) {
875 /* Ignore error and keep issuiing other urbs */
876 return;
880 static struct urb *_rtl_usb_tx_urb_setup(struct ieee80211_hw *hw,
881 struct sk_buff *skb, u32 ep_num)
883 struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
884 struct urb *_urb;
886 WARN_ON(NULL == skb);
887 _urb = usb_alloc_urb(0, GFP_ATOMIC);
888 if (!_urb)
889 return NULL;
890 _rtl_install_trx_info(rtlusb, skb, ep_num);
891 usb_fill_bulk_urb(_urb, rtlusb->udev, usb_sndbulkpipe(rtlusb->udev,
892 ep_num), skb->data, skb->len, _rtl_tx_complete, skb);
893 _urb->transfer_flags |= URB_ZERO_PACKET;
894 return _urb;
897 static void _rtl_usb_transmit(struct ieee80211_hw *hw, struct sk_buff *skb,
898 enum rtl_txq qnum)
900 struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
901 u32 ep_num;
902 struct urb *_urb = NULL;
904 WARN_ON(NULL == rtlusb->usb_tx_aggregate_hdl);
905 if (unlikely(IS_USB_STOP(rtlusb))) {
906 pr_err("USB device is stopping...\n");
907 kfree_skb(skb);
908 return;
910 ep_num = rtlusb->ep_map.ep_mapping[qnum];
911 _urb = _rtl_usb_tx_urb_setup(hw, skb, ep_num);
912 if (unlikely(!_urb)) {
913 pr_err("Can't allocate urb. Drop skb!\n");
914 kfree_skb(skb);
915 return;
917 _rtl_submit_tx_urb(hw, _urb);
920 static void _rtl_usb_tx_preprocess(struct ieee80211_hw *hw,
921 struct ieee80211_sta *sta,
922 struct sk_buff *skb,
923 u16 hw_queue)
925 struct rtl_priv *rtlpriv = rtl_priv(hw);
926 struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
927 struct rtl_tx_desc *pdesc = NULL;
928 struct rtl_tcb_desc tcb_desc;
929 struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)(skb->data);
930 __le16 fc = hdr->frame_control;
931 u8 *pda_addr = hdr->addr1;
933 memset(&tcb_desc, 0, sizeof(struct rtl_tcb_desc));
934 if (ieee80211_is_auth(fc)) {
935 rtl_dbg(rtlpriv, COMP_SEND, DBG_DMESG, "MAC80211_LINKING\n");
938 if (rtlpriv->psc.sw_ps_enabled) {
939 if (ieee80211_is_data(fc) && !ieee80211_is_nullfunc(fc) &&
940 !ieee80211_has_pm(fc))
941 hdr->frame_control |= cpu_to_le16(IEEE80211_FCTL_PM);
944 rtl_action_proc(hw, skb, true);
945 if (is_multicast_ether_addr(pda_addr))
946 rtlpriv->stats.txbytesmulticast += skb->len;
947 else if (is_broadcast_ether_addr(pda_addr))
948 rtlpriv->stats.txbytesbroadcast += skb->len;
949 else
950 rtlpriv->stats.txbytesunicast += skb->len;
951 rtlpriv->cfg->ops->fill_tx_desc(hw, hdr, (u8 *)pdesc, NULL, info, sta, skb,
952 hw_queue, &tcb_desc);
953 if (ieee80211_is_data(fc))
954 rtlpriv->cfg->ops->led_control(hw, LED_CTL_TX);
957 static int rtl_usb_tx(struct ieee80211_hw *hw,
958 struct ieee80211_sta *sta,
959 struct sk_buff *skb,
960 struct rtl_tcb_desc *dummy)
962 struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
963 struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
964 struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)(skb->data);
965 __le16 fc = hdr->frame_control;
966 u16 hw_queue;
968 if (unlikely(is_hal_stop(rtlhal)))
969 goto err_free;
970 hw_queue = rtlusb->usb_mq_to_hwq(fc, skb_get_queue_mapping(skb));
971 _rtl_usb_tx_preprocess(hw, sta, skb, hw_queue);
972 _rtl_usb_transmit(hw, skb, hw_queue);
973 return NETDEV_TX_OK;
975 err_free:
976 dev_kfree_skb_any(skb);
977 return NETDEV_TX_OK;
980 static bool rtl_usb_tx_chk_waitq_insert(struct ieee80211_hw *hw,
981 struct ieee80211_sta *sta,
982 struct sk_buff *skb)
984 return false;
987 static void rtl_fill_h2c_cmd_work_callback(struct work_struct *work)
989 struct rtl_works *rtlworks =
990 container_of(work, struct rtl_works, fill_h2c_cmd);
991 struct ieee80211_hw *hw = rtlworks->hw;
992 struct rtl_priv *rtlpriv = rtl_priv(hw);
994 rtlpriv->cfg->ops->fill_h2c_cmd(hw, H2C_RA_MASK, 5, rtlpriv->rate_mask);
997 static const struct rtl_intf_ops rtl_usb_ops = {
998 .adapter_start = rtl_usb_start,
999 .adapter_stop = rtl_usb_stop,
1000 .adapter_tx = rtl_usb_tx,
1001 .waitq_insert = rtl_usb_tx_chk_waitq_insert,
1004 int rtl_usb_probe(struct usb_interface *intf,
1005 const struct usb_device_id *id,
1006 struct rtl_hal_cfg *rtl_hal_cfg)
1008 int err;
1009 struct ieee80211_hw *hw = NULL;
1010 struct rtl_priv *rtlpriv = NULL;
1011 struct usb_device *udev;
1012 struct rtl_usb_priv *usb_priv;
1014 hw = ieee80211_alloc_hw(sizeof(struct rtl_priv) +
1015 sizeof(struct rtl_usb_priv), &rtl_ops);
1016 if (!hw) {
1017 WARN_ONCE(true, "rtl_usb: ieee80211 alloc failed\n");
1018 return -ENOMEM;
1020 rtlpriv = hw->priv;
1021 rtlpriv->hw = hw;
1022 rtlpriv->usb_data = kcalloc(RTL_USB_MAX_RX_COUNT, sizeof(u32),
1023 GFP_KERNEL);
1024 if (!rtlpriv->usb_data) {
1025 ieee80211_free_hw(hw);
1026 return -ENOMEM;
1029 /* this spin lock must be initialized early */
1030 spin_lock_init(&rtlpriv->locks.usb_lock);
1031 INIT_WORK(&rtlpriv->works.fill_h2c_cmd,
1032 rtl_fill_h2c_cmd_work_callback);
1033 INIT_WORK(&rtlpriv->works.lps_change_work,
1034 rtl_lps_change_work_callback);
1036 rtlpriv->usb_data_index = 0;
1037 init_completion(&rtlpriv->firmware_loading_complete);
1038 SET_IEEE80211_DEV(hw, &intf->dev);
1039 udev = interface_to_usbdev(intf);
1040 usb_get_dev(udev);
1041 usb_priv = rtl_usbpriv(hw);
1042 memset(usb_priv, 0, sizeof(*usb_priv));
1043 usb_priv->dev.intf = intf;
1044 usb_priv->dev.udev = udev;
1045 usb_set_intfdata(intf, hw);
1046 /* init cfg & intf_ops */
1047 rtlpriv->rtlhal.interface = INTF_USB;
1048 rtlpriv->cfg = rtl_hal_cfg;
1049 rtlpriv->intf_ops = &rtl_usb_ops;
1050 /* Init IO handler */
1051 _rtl_usb_io_handler_init(&udev->dev, hw);
1052 rtlpriv->cfg->ops->read_chip_version(hw);
1053 /*like read eeprom and so on */
1054 rtlpriv->cfg->ops->read_eeprom_info(hw);
1055 err = _rtl_usb_init(hw);
1056 if (err)
1057 goto error_out2;
1058 rtl_usb_init_sw(hw);
1059 /* Init mac80211 sw */
1060 err = rtl_init_core(hw);
1061 if (err) {
1062 pr_err("Can't allocate sw for mac80211\n");
1063 goto error_out2;
1065 if (rtlpriv->cfg->ops->init_sw_vars(hw)) {
1066 pr_err("Can't init_sw_vars\n");
1067 goto error_out;
1069 rtlpriv->cfg->ops->init_sw_leds(hw);
1071 err = ieee80211_register_hw(hw);
1072 if (err) {
1073 pr_err("Can't register mac80211 hw.\n");
1074 err = -ENODEV;
1075 goto error_out;
1077 rtlpriv->mac80211.mac80211_registered = 1;
1079 set_bit(RTL_STATUS_INTERFACE_START, &rtlpriv->status);
1080 return 0;
1082 error_out:
1083 rtl_deinit_core(hw);
1084 error_out2:
1085 _rtl_usb_io_handler_release(hw);
1086 usb_put_dev(udev);
1087 complete(&rtlpriv->firmware_loading_complete);
1088 kfree(rtlpriv->usb_data);
1089 ieee80211_free_hw(hw);
1090 return -ENODEV;
1092 EXPORT_SYMBOL(rtl_usb_probe);
1094 void rtl_usb_disconnect(struct usb_interface *intf)
1096 struct ieee80211_hw *hw = usb_get_intfdata(intf);
1097 struct rtl_priv *rtlpriv = rtl_priv(hw);
1098 struct rtl_mac *rtlmac = rtl_mac(rtl_priv(hw));
1099 struct rtl_usb *rtlusb = rtl_usbdev(rtl_usbpriv(hw));
1101 if (unlikely(!rtlpriv))
1102 return;
1103 /* just in case driver is removed before firmware callback */
1104 wait_for_completion(&rtlpriv->firmware_loading_complete);
1105 clear_bit(RTL_STATUS_INTERFACE_START, &rtlpriv->status);
1106 /*ieee80211_unregister_hw will call ops_stop */
1107 if (rtlmac->mac80211_registered == 1) {
1108 ieee80211_unregister_hw(hw);
1109 rtlmac->mac80211_registered = 0;
1110 } else {
1111 rtl_deinit_deferred_work(hw, false);
1112 rtlpriv->intf_ops->adapter_stop(hw);
1114 /*deinit rfkill */
1115 /* rtl_deinit_rfkill(hw); */
1116 rtl_usb_deinit(hw);
1117 rtl_deinit_core(hw);
1118 kfree(rtlpriv->usb_data);
1119 rtlpriv->cfg->ops->deinit_sw_leds(hw);
1120 rtlpriv->cfg->ops->deinit_sw_vars(hw);
1121 _rtl_usb_io_handler_release(hw);
1122 usb_put_dev(rtlusb->udev);
1123 usb_set_intfdata(intf, NULL);
1124 ieee80211_free_hw(hw);
1126 EXPORT_SYMBOL(rtl_usb_disconnect);
1128 int rtl_usb_suspend(struct usb_interface *pusb_intf, pm_message_t message)
1130 return 0;
1132 EXPORT_SYMBOL(rtl_usb_suspend);
1134 int rtl_usb_resume(struct usb_interface *pusb_intf)
1136 return 0;
1138 EXPORT_SYMBOL(rtl_usb_resume);