1 /* 2 * Copyright (c) 2010-2011 Atheros Communications Inc. 3 * 4 * Permission to use, copy, modify, and/or distribute this software for any 5 * purpose with or without fee is hereby granted, provided that the above 6 * copyright notice and this permission notice appear in all copies. 7 * 8 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES 9 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF 10 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR 11 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES 12 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN 13 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF 14 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 15 */ 16 17 #include <asm/unaligned.h> 18 #include "htc.h" 19 20 MODULE_FIRMWARE(HTC_7010_MODULE_FW); 21 MODULE_FIRMWARE(HTC_9271_MODULE_FW); 22 23 static const struct usb_device_id ath9k_hif_usb_ids[] = { 24 { USB_DEVICE(0x0cf3, 0x9271) }, /* Atheros */ 25 { USB_DEVICE(0x0cf3, 0x1006) }, /* Atheros */ 26 { USB_DEVICE(0x0846, 0x9030) }, /* Netgear N150 */ 27 { USB_DEVICE(0x07b8, 0x9271) }, /* Altai WA1011N-GU */ 28 { USB_DEVICE(0x07D1, 0x3A10) }, /* Dlink Wireless 150 */ 29 { USB_DEVICE(0x13D3, 0x3327) }, /* Azurewave */ 30 { USB_DEVICE(0x13D3, 0x3328) }, /* Azurewave */ 31 { USB_DEVICE(0x13D3, 0x3346) }, /* IMC Networks */ 32 { USB_DEVICE(0x13D3, 0x3348) }, /* Azurewave */ 33 { USB_DEVICE(0x13D3, 0x3349) }, /* Azurewave */ 34 { USB_DEVICE(0x13D3, 0x3350) }, /* Azurewave */ 35 { USB_DEVICE(0x04CA, 0x4605) }, /* Liteon */ 36 { USB_DEVICE(0x040D, 0x3801) }, /* VIA */ 37 { USB_DEVICE(0x0cf3, 0xb003) }, /* Ubiquiti WifiStation Ext */ 38 { USB_DEVICE(0x0cf3, 0xb002) }, /* Ubiquiti WifiStation */ 39 { USB_DEVICE(0x057c, 0x8403) }, /* AVM FRITZ!WLAN 11N v2 USB */ 40 { USB_DEVICE(0x0471, 0x209e) }, /* Philips (or NXP) PTA01 */ 41 { USB_DEVICE(0x1eda, 0x2315) }, /* AirTies */ 42 43 { USB_DEVICE(0x0cf3, 0x7015), 44 .driver_info = AR9287_USB }, /* Atheros */ 45 { USB_DEVICE(0x1668, 0x1200), 46 .driver_info = AR9287_USB }, /* Verizon */ 47 48 { USB_DEVICE(0x0cf3, 0x7010), 49 .driver_info = AR9280_USB }, /* Atheros */ 50 { USB_DEVICE(0x0846, 0x9018), 51 .driver_info = AR9280_USB }, /* Netgear WNDA3200 */ 52 { USB_DEVICE(0x083A, 0xA704), 53 .driver_info = AR9280_USB }, /* SMC Networks */ 54 { USB_DEVICE(0x0411, 0x017f), 55 .driver_info = AR9280_USB }, /* Sony UWA-BR100 */ 56 { USB_DEVICE(0x0411, 0x0197), 57 .driver_info = AR9280_USB }, /* Buffalo WLI-UV-AG300P */ 58 { USB_DEVICE(0x04da, 0x3904), 59 .driver_info = AR9280_USB }, 60 { USB_DEVICE(0x0930, 0x0a08), 61 .driver_info = AR9280_USB }, /* Toshiba WLM-20U2 and GN-1080 */ 62 63 { USB_DEVICE(0x0cf3, 0x20ff), 64 .driver_info = STORAGE_DEVICE }, 65 66 { }, 67 }; 68 69 MODULE_DEVICE_TABLE(usb, ath9k_hif_usb_ids); 70 71 static int __hif_usb_tx(struct hif_device_usb *hif_dev); 72 73 static void hif_usb_regout_cb(struct urb *urb) 74 { 75 struct cmd_buf *cmd = (struct cmd_buf *)urb->context; 76 77 switch (urb->status) { 78 case 0: 79 break; 80 case -ENOENT: 81 case -ECONNRESET: 82 case -ENODEV: 83 case -ESHUTDOWN: 84 goto free; 85 default: 86 break; 87 } 88 89 if (cmd) { 90 ath9k_htc_txcompletion_cb(cmd->hif_dev->htc_handle, 91 cmd->skb, true); 92 kfree(cmd); 93 } 94 95 return; 96 free: 97 kfree_skb(cmd->skb); 98 kfree(cmd); 99 } 100 101 static int hif_usb_send_regout(struct hif_device_usb *hif_dev, 102 struct sk_buff *skb) 103 { 104 struct urb *urb; 105 struct cmd_buf *cmd; 106 int ret = 0; 107 108 urb = usb_alloc_urb(0, GFP_KERNEL); 109 if (urb == NULL) 110 return -ENOMEM; 111 112 cmd = kzalloc(sizeof(*cmd), GFP_KERNEL); 113 if (cmd == NULL) { 114 usb_free_urb(urb); 115 return -ENOMEM; 116 } 117 118 cmd->skb = skb; 119 cmd->hif_dev = hif_dev; 120 121 usb_fill_int_urb(urb, hif_dev->udev, 122 usb_sndintpipe(hif_dev->udev, USB_REG_OUT_PIPE), 123 skb->data, skb->len, 124 hif_usb_regout_cb, cmd, 1); 125 126 usb_anchor_urb(urb, &hif_dev->regout_submitted); 127 ret = usb_submit_urb(urb, GFP_KERNEL); 128 if (ret) { 129 usb_unanchor_urb(urb); 130 kfree(cmd); 131 } 132 usb_free_urb(urb); 133 134 return ret; 135 } 136 137 static void hif_usb_mgmt_cb(struct urb *urb) 138 { 139 struct cmd_buf *cmd = (struct cmd_buf *)urb->context; 140 struct hif_device_usb *hif_dev; 141 unsigned long flags; 142 bool txok = true; 143 144 if (!cmd || !cmd->skb || !cmd->hif_dev) 145 return; 146 147 hif_dev = cmd->hif_dev; 148 149 switch (urb->status) { 150 case 0: 151 break; 152 case -ENOENT: 153 case -ECONNRESET: 154 case -ENODEV: 155 case -ESHUTDOWN: 156 txok = false; 157 158 /* 159 * If the URBs are being flushed, no need to complete 160 * this packet. 161 */ 162 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags); 163 if (hif_dev->tx.flags & HIF_USB_TX_FLUSH) { 164 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags); 165 dev_kfree_skb_any(cmd->skb); 166 kfree(cmd); 167 return; 168 } 169 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags); 170 171 break; 172 default: 173 txok = false; 174 break; 175 } 176 177 skb_pull(cmd->skb, 4); 178 ath9k_htc_txcompletion_cb(cmd->hif_dev->htc_handle, 179 cmd->skb, txok); 180 kfree(cmd); 181 } 182 183 static int hif_usb_send_mgmt(struct hif_device_usb *hif_dev, 184 struct sk_buff *skb) 185 { 186 struct urb *urb; 187 struct cmd_buf *cmd; 188 int ret = 0; 189 __le16 *hdr; 190 191 urb = usb_alloc_urb(0, GFP_ATOMIC); 192 if (urb == NULL) 193 return -ENOMEM; 194 195 cmd = kzalloc(sizeof(*cmd), GFP_ATOMIC); 196 if (cmd == NULL) { 197 usb_free_urb(urb); 198 return -ENOMEM; 199 } 200 201 cmd->skb = skb; 202 cmd->hif_dev = hif_dev; 203 204 hdr = skb_push(skb, 4); 205 *hdr++ = cpu_to_le16(skb->len - 4); 206 *hdr++ = cpu_to_le16(ATH_USB_TX_STREAM_MODE_TAG); 207 208 usb_fill_bulk_urb(urb, hif_dev->udev, 209 usb_sndbulkpipe(hif_dev->udev, USB_WLAN_TX_PIPE), 210 skb->data, skb->len, 211 hif_usb_mgmt_cb, cmd); 212 213 usb_anchor_urb(urb, &hif_dev->mgmt_submitted); 214 ret = usb_submit_urb(urb, GFP_ATOMIC); 215 if (ret) { 216 usb_unanchor_urb(urb); 217 kfree(cmd); 218 } 219 usb_free_urb(urb); 220 221 return ret; 222 } 223 224 static inline void ath9k_skb_queue_purge(struct hif_device_usb *hif_dev, 225 struct sk_buff_head *list) 226 { 227 struct sk_buff *skb; 228 229 while ((skb = __skb_dequeue(list)) != NULL) { 230 dev_kfree_skb_any(skb); 231 } 232 } 233 234 static inline void ath9k_skb_queue_complete(struct hif_device_usb *hif_dev, 235 struct sk_buff_head *queue, 236 bool txok) 237 { 238 struct sk_buff *skb; 239 240 while ((skb = __skb_dequeue(queue)) != NULL) { 241 #ifdef CONFIG_ATH9K_HTC_DEBUGFS 242 int ln = skb->len; 243 #endif 244 ath9k_htc_txcompletion_cb(hif_dev->htc_handle, 245 skb, txok); 246 if (txok) { 247 TX_STAT_INC(skb_success); 248 TX_STAT_ADD(skb_success_bytes, ln); 249 } 250 else 251 TX_STAT_INC(skb_failed); 252 } 253 } 254 255 static void hif_usb_tx_cb(struct urb *urb) 256 { 257 struct tx_buf *tx_buf = (struct tx_buf *) urb->context; 258 struct hif_device_usb *hif_dev; 259 bool txok = true; 260 261 if (!tx_buf || !tx_buf->hif_dev) 262 return; 263 264 hif_dev = tx_buf->hif_dev; 265 266 switch (urb->status) { 267 case 0: 268 break; 269 case -ENOENT: 270 case -ECONNRESET: 271 case -ENODEV: 272 case -ESHUTDOWN: 273 txok = false; 274 275 /* 276 * If the URBs are being flushed, no need to add this 277 * URB to the free list. 278 */ 279 spin_lock(&hif_dev->tx.tx_lock); 280 if (hif_dev->tx.flags & HIF_USB_TX_FLUSH) { 281 spin_unlock(&hif_dev->tx.tx_lock); 282 ath9k_skb_queue_purge(hif_dev, &tx_buf->skb_queue); 283 return; 284 } 285 spin_unlock(&hif_dev->tx.tx_lock); 286 287 break; 288 default: 289 txok = false; 290 break; 291 } 292 293 ath9k_skb_queue_complete(hif_dev, &tx_buf->skb_queue, txok); 294 295 /* Re-initialize the SKB queue */ 296 tx_buf->len = tx_buf->offset = 0; 297 __skb_queue_head_init(&tx_buf->skb_queue); 298 299 /* Add this TX buffer to the free list */ 300 spin_lock(&hif_dev->tx.tx_lock); 301 list_move_tail(&tx_buf->list, &hif_dev->tx.tx_buf); 302 hif_dev->tx.tx_buf_cnt++; 303 if (!(hif_dev->tx.flags & HIF_USB_TX_STOP)) 304 __hif_usb_tx(hif_dev); /* Check for pending SKBs */ 305 TX_STAT_INC(buf_completed); 306 spin_unlock(&hif_dev->tx.tx_lock); 307 } 308 309 /* TX lock has to be taken */ 310 static int __hif_usb_tx(struct hif_device_usb *hif_dev) 311 { 312 struct tx_buf *tx_buf = NULL; 313 struct sk_buff *nskb = NULL; 314 int ret = 0, i; 315 u16 tx_skb_cnt = 0; 316 u8 *buf; 317 __le16 *hdr; 318 319 if (hif_dev->tx.tx_skb_cnt == 0) 320 return 0; 321 322 /* Check if a free TX buffer is available */ 323 if (list_empty(&hif_dev->tx.tx_buf)) 324 return 0; 325 326 tx_buf = list_first_entry(&hif_dev->tx.tx_buf, struct tx_buf, list); 327 list_move_tail(&tx_buf->list, &hif_dev->tx.tx_pending); 328 hif_dev->tx.tx_buf_cnt--; 329 330 tx_skb_cnt = min_t(u16, hif_dev->tx.tx_skb_cnt, MAX_TX_AGGR_NUM); 331 332 for (i = 0; i < tx_skb_cnt; i++) { 333 nskb = __skb_dequeue(&hif_dev->tx.tx_skb_queue); 334 335 /* Should never be NULL */ 336 BUG_ON(!nskb); 337 338 hif_dev->tx.tx_skb_cnt--; 339 340 buf = tx_buf->buf; 341 buf += tx_buf->offset; 342 hdr = (__le16 *)buf; 343 *hdr++ = cpu_to_le16(nskb->len); 344 *hdr++ = cpu_to_le16(ATH_USB_TX_STREAM_MODE_TAG); 345 buf += 4; 346 memcpy(buf, nskb->data, nskb->len); 347 tx_buf->len = nskb->len + 4; 348 349 if (i < (tx_skb_cnt - 1)) 350 tx_buf->offset += (((tx_buf->len - 1) / 4) + 1) * 4; 351 352 if (i == (tx_skb_cnt - 1)) 353 tx_buf->len += tx_buf->offset; 354 355 __skb_queue_tail(&tx_buf->skb_queue, nskb); 356 TX_STAT_INC(skb_queued); 357 } 358 359 usb_fill_bulk_urb(tx_buf->urb, hif_dev->udev, 360 usb_sndbulkpipe(hif_dev->udev, USB_WLAN_TX_PIPE), 361 tx_buf->buf, tx_buf->len, 362 hif_usb_tx_cb, tx_buf); 363 364 ret = usb_submit_urb(tx_buf->urb, GFP_ATOMIC); 365 if (ret) { 366 tx_buf->len = tx_buf->offset = 0; 367 ath9k_skb_queue_complete(hif_dev, &tx_buf->skb_queue, false); 368 __skb_queue_head_init(&tx_buf->skb_queue); 369 list_move_tail(&tx_buf->list, &hif_dev->tx.tx_buf); 370 hif_dev->tx.tx_buf_cnt++; 371 } 372 373 if (!ret) 374 TX_STAT_INC(buf_queued); 375 376 return ret; 377 } 378 379 static int hif_usb_send_tx(struct hif_device_usb *hif_dev, struct sk_buff *skb) 380 { 381 struct ath9k_htc_tx_ctl *tx_ctl; 382 unsigned long flags; 383 int ret = 0; 384 385 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags); 386 387 if (hif_dev->tx.flags & HIF_USB_TX_STOP) { 388 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags); 389 return -ENODEV; 390 } 391 392 /* Check if the max queue count has been reached */ 393 if (hif_dev->tx.tx_skb_cnt > MAX_TX_BUF_NUM) { 394 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags); 395 return -ENOMEM; 396 } 397 398 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags); 399 400 tx_ctl = HTC_SKB_CB(skb); 401 402 /* Mgmt/Beacon frames don't use the TX buffer pool */ 403 if ((tx_ctl->type == ATH9K_HTC_MGMT) || 404 (tx_ctl->type == ATH9K_HTC_BEACON)) { 405 ret = hif_usb_send_mgmt(hif_dev, skb); 406 } 407 408 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags); 409 410 if ((tx_ctl->type == ATH9K_HTC_NORMAL) || 411 (tx_ctl->type == ATH9K_HTC_AMPDU)) { 412 __skb_queue_tail(&hif_dev->tx.tx_skb_queue, skb); 413 hif_dev->tx.tx_skb_cnt++; 414 } 415 416 /* Check if AMPDUs have to be sent immediately */ 417 if ((hif_dev->tx.tx_buf_cnt == MAX_TX_URB_NUM) && 418 (hif_dev->tx.tx_skb_cnt < 2)) { 419 __hif_usb_tx(hif_dev); 420 } 421 422 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags); 423 424 return ret; 425 } 426 427 static void hif_usb_start(void *hif_handle) 428 { 429 struct hif_device_usb *hif_dev = hif_handle; 430 unsigned long flags; 431 432 hif_dev->flags |= HIF_USB_START; 433 434 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags); 435 hif_dev->tx.flags &= ~HIF_USB_TX_STOP; 436 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags); 437 } 438 439 static void hif_usb_stop(void *hif_handle) 440 { 441 struct hif_device_usb *hif_dev = hif_handle; 442 struct tx_buf *tx_buf = NULL, *tx_buf_tmp = NULL; 443 unsigned long flags; 444 445 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags); 446 ath9k_skb_queue_complete(hif_dev, &hif_dev->tx.tx_skb_queue, false); 447 hif_dev->tx.tx_skb_cnt = 0; 448 hif_dev->tx.flags |= HIF_USB_TX_STOP; 449 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags); 450 451 /* The pending URBs have to be canceled. */ 452 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags); 453 list_for_each_entry_safe(tx_buf, tx_buf_tmp, 454 &hif_dev->tx.tx_pending, list) { 455 usb_get_urb(tx_buf->urb); 456 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags); 457 usb_kill_urb(tx_buf->urb); 458 list_del(&tx_buf->list); 459 usb_free_urb(tx_buf->urb); 460 kfree(tx_buf->buf); 461 kfree(tx_buf); 462 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags); 463 } 464 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags); 465 466 usb_kill_anchored_urbs(&hif_dev->mgmt_submitted); 467 } 468 469 static int hif_usb_send(void *hif_handle, u8 pipe_id, struct sk_buff *skb) 470 { 471 struct hif_device_usb *hif_dev = hif_handle; 472 int ret = 0; 473 474 switch (pipe_id) { 475 case USB_WLAN_TX_PIPE: 476 ret = hif_usb_send_tx(hif_dev, skb); 477 break; 478 case USB_REG_OUT_PIPE: 479 ret = hif_usb_send_regout(hif_dev, skb); 480 break; 481 default: 482 dev_err(&hif_dev->udev->dev, 483 "ath9k_htc: Invalid TX pipe: %d\n", pipe_id); 484 ret = -EINVAL; 485 break; 486 } 487 488 return ret; 489 } 490 491 static inline bool check_index(struct sk_buff *skb, u8 idx) 492 { 493 struct ath9k_htc_tx_ctl *tx_ctl; 494 495 tx_ctl = HTC_SKB_CB(skb); 496 497 if ((tx_ctl->type == ATH9K_HTC_AMPDU) && 498 (tx_ctl->sta_idx == idx)) 499 return true; 500 501 return false; 502 } 503 504 static void hif_usb_sta_drain(void *hif_handle, u8 idx) 505 { 506 struct hif_device_usb *hif_dev = hif_handle; 507 struct sk_buff *skb, *tmp; 508 unsigned long flags; 509 510 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags); 511 512 skb_queue_walk_safe(&hif_dev->tx.tx_skb_queue, skb, tmp) { 513 if (check_index(skb, idx)) { 514 __skb_unlink(skb, &hif_dev->tx.tx_skb_queue); 515 ath9k_htc_txcompletion_cb(hif_dev->htc_handle, 516 skb, false); 517 hif_dev->tx.tx_skb_cnt--; 518 TX_STAT_INC(skb_failed); 519 } 520 } 521 522 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags); 523 } 524 525 static struct ath9k_htc_hif hif_usb = { 526 .transport = ATH9K_HIF_USB, 527 .name = "ath9k_hif_usb", 528 529 .control_ul_pipe = USB_REG_OUT_PIPE, 530 .control_dl_pipe = USB_REG_IN_PIPE, 531 532 .start = hif_usb_start, 533 .stop = hif_usb_stop, 534 .sta_drain = hif_usb_sta_drain, 535 .send = hif_usb_send, 536 }; 537 538 static void ath9k_hif_usb_rx_stream(struct hif_device_usb *hif_dev, 539 struct sk_buff *skb) 540 { 541 struct sk_buff *nskb, *skb_pool[MAX_PKT_NUM_IN_TRANSFER]; 542 int index = 0, i, len = skb->len; 543 int rx_remain_len, rx_pkt_len; 544 u16 pool_index = 0; 545 u8 *ptr; 546 547 spin_lock(&hif_dev->rx_lock); 548 549 rx_remain_len = hif_dev->rx_remain_len; 550 rx_pkt_len = hif_dev->rx_transfer_len; 551 552 if (rx_remain_len != 0) { 553 struct sk_buff *remain_skb = hif_dev->remain_skb; 554 555 if (remain_skb) { 556 ptr = (u8 *) remain_skb->data; 557 558 index = rx_remain_len; 559 rx_remain_len -= hif_dev->rx_pad_len; 560 ptr += rx_pkt_len; 561 562 memcpy(ptr, skb->data, rx_remain_len); 563 564 rx_pkt_len += rx_remain_len; 565 hif_dev->rx_remain_len = 0; 566 skb_put(remain_skb, rx_pkt_len); 567 568 skb_pool[pool_index++] = remain_skb; 569 570 } else { 571 index = rx_remain_len; 572 } 573 } 574 575 spin_unlock(&hif_dev->rx_lock); 576 577 while (index < len) { 578 u16 pkt_len; 579 u16 pkt_tag; 580 u16 pad_len; 581 int chk_idx; 582 583 ptr = (u8 *) skb->data; 584 585 pkt_len = get_unaligned_le16(ptr + index); 586 pkt_tag = get_unaligned_le16(ptr + index + 2); 587 588 if (pkt_tag != ATH_USB_RX_STREAM_MODE_TAG) { 589 RX_STAT_INC(skb_dropped); 590 return; 591 } 592 593 if (pkt_len > 2 * MAX_RX_BUF_SIZE) { 594 dev_err(&hif_dev->udev->dev, 595 "ath9k_htc: invalid pkt_len (%x)\n", pkt_len); 596 RX_STAT_INC(skb_dropped); 597 return; 598 } 599 600 pad_len = 4 - (pkt_len & 0x3); 601 if (pad_len == 4) 602 pad_len = 0; 603 604 chk_idx = index; 605 index = index + 4 + pkt_len + pad_len; 606 607 if (index > MAX_RX_BUF_SIZE) { 608 spin_lock(&hif_dev->rx_lock); 609 hif_dev->rx_remain_len = index - MAX_RX_BUF_SIZE; 610 hif_dev->rx_transfer_len = 611 MAX_RX_BUF_SIZE - chk_idx - 4; 612 hif_dev->rx_pad_len = pad_len; 613 614 nskb = __dev_alloc_skb(pkt_len + 32, GFP_ATOMIC); 615 if (!nskb) { 616 dev_err(&hif_dev->udev->dev, 617 "ath9k_htc: RX memory allocation error\n"); 618 spin_unlock(&hif_dev->rx_lock); 619 goto err; 620 } 621 skb_reserve(nskb, 32); 622 RX_STAT_INC(skb_allocated); 623 624 memcpy(nskb->data, &(skb->data[chk_idx+4]), 625 hif_dev->rx_transfer_len); 626 627 /* Record the buffer pointer */ 628 hif_dev->remain_skb = nskb; 629 spin_unlock(&hif_dev->rx_lock); 630 } else { 631 if (pool_index == MAX_PKT_NUM_IN_TRANSFER) { 632 dev_err(&hif_dev->udev->dev, 633 "ath9k_htc: over RX MAX_PKT_NUM\n"); 634 goto err; 635 } 636 nskb = __dev_alloc_skb(pkt_len + 32, GFP_ATOMIC); 637 if (!nskb) { 638 dev_err(&hif_dev->udev->dev, 639 "ath9k_htc: RX memory allocation error\n"); 640 goto err; 641 } 642 skb_reserve(nskb, 32); 643 RX_STAT_INC(skb_allocated); 644 645 memcpy(nskb->data, &(skb->data[chk_idx+4]), pkt_len); 646 skb_put(nskb, pkt_len); 647 skb_pool[pool_index++] = nskb; 648 } 649 } 650 651 err: 652 for (i = 0; i < pool_index; i++) { 653 RX_STAT_ADD(skb_completed_bytes, skb_pool[i]->len); 654 ath9k_htc_rx_msg(hif_dev->htc_handle, skb_pool[i], 655 skb_pool[i]->len, USB_WLAN_RX_PIPE); 656 RX_STAT_INC(skb_completed); 657 } 658 } 659 660 static void ath9k_hif_usb_rx_cb(struct urb *urb) 661 { 662 struct rx_buf *rx_buf = (struct rx_buf *)urb->context; 663 struct hif_device_usb *hif_dev = rx_buf->hif_dev; 664 struct sk_buff *skb = rx_buf->skb; 665 int ret; 666 667 if (!skb) 668 return; 669 670 if (!hif_dev) 671 goto free; 672 673 switch (urb->status) { 674 case 0: 675 break; 676 case -ENOENT: 677 case -ECONNRESET: 678 case -ENODEV: 679 case -ESHUTDOWN: 680 goto free; 681 default: 682 goto resubmit; 683 } 684 685 if (likely(urb->actual_length != 0)) { 686 skb_put(skb, urb->actual_length); 687 ath9k_hif_usb_rx_stream(hif_dev, skb); 688 } 689 690 resubmit: 691 skb_reset_tail_pointer(skb); 692 skb_trim(skb, 0); 693 694 usb_anchor_urb(urb, &hif_dev->rx_submitted); 695 ret = usb_submit_urb(urb, GFP_ATOMIC); 696 if (ret) { 697 usb_unanchor_urb(urb); 698 goto free; 699 } 700 701 return; 702 free: 703 kfree_skb(skb); 704 kfree(rx_buf); 705 } 706 707 static void ath9k_hif_usb_reg_in_cb(struct urb *urb) 708 { 709 struct rx_buf *rx_buf = (struct rx_buf *)urb->context; 710 struct hif_device_usb *hif_dev = rx_buf->hif_dev; 711 struct sk_buff *skb = rx_buf->skb; 712 struct sk_buff *nskb; 713 int ret; 714 715 if (!skb) 716 return; 717 718 if (!hif_dev) 719 goto free; 720 721 switch (urb->status) { 722 case 0: 723 break; 724 case -ENOENT: 725 case -ECONNRESET: 726 case -ENODEV: 727 case -ESHUTDOWN: 728 goto free; 729 default: 730 skb_reset_tail_pointer(skb); 731 skb_trim(skb, 0); 732 733 goto resubmit; 734 } 735 736 if (likely(urb->actual_length != 0)) { 737 skb_put(skb, urb->actual_length); 738 739 /* Process the command first */ 740 ath9k_htc_rx_msg(hif_dev->htc_handle, skb, 741 skb->len, USB_REG_IN_PIPE); 742 743 744 nskb = alloc_skb(MAX_REG_IN_BUF_SIZE, GFP_ATOMIC); 745 if (!nskb) { 746 dev_err(&hif_dev->udev->dev, 747 "ath9k_htc: REG_IN memory allocation failure\n"); 748 urb->context = NULL; 749 return; 750 } 751 752 rx_buf->skb = nskb; 753 754 usb_fill_int_urb(urb, hif_dev->udev, 755 usb_rcvintpipe(hif_dev->udev, 756 USB_REG_IN_PIPE), 757 nskb->data, MAX_REG_IN_BUF_SIZE, 758 ath9k_hif_usb_reg_in_cb, rx_buf, 1); 759 } 760 761 resubmit: 762 usb_anchor_urb(urb, &hif_dev->reg_in_submitted); 763 ret = usb_submit_urb(urb, GFP_ATOMIC); 764 if (ret) { 765 usb_unanchor_urb(urb); 766 goto free; 767 } 768 769 return; 770 free: 771 kfree_skb(skb); 772 kfree(rx_buf); 773 urb->context = NULL; 774 } 775 776 static void ath9k_hif_usb_dealloc_tx_urbs(struct hif_device_usb *hif_dev) 777 { 778 struct tx_buf *tx_buf = NULL, *tx_buf_tmp = NULL; 779 unsigned long flags; 780 781 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags); 782 list_for_each_entry_safe(tx_buf, tx_buf_tmp, 783 &hif_dev->tx.tx_buf, list) { 784 usb_get_urb(tx_buf->urb); 785 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags); 786 usb_kill_urb(tx_buf->urb); 787 list_del(&tx_buf->list); 788 usb_free_urb(tx_buf->urb); 789 kfree(tx_buf->buf); 790 kfree(tx_buf); 791 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags); 792 } 793 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags); 794 795 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags); 796 hif_dev->tx.flags |= HIF_USB_TX_FLUSH; 797 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags); 798 799 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags); 800 list_for_each_entry_safe(tx_buf, tx_buf_tmp, 801 &hif_dev->tx.tx_pending, list) { 802 usb_get_urb(tx_buf->urb); 803 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags); 804 usb_kill_urb(tx_buf->urb); 805 list_del(&tx_buf->list); 806 usb_free_urb(tx_buf->urb); 807 kfree(tx_buf->buf); 808 kfree(tx_buf); 809 spin_lock_irqsave(&hif_dev->tx.tx_lock, flags); 810 } 811 spin_unlock_irqrestore(&hif_dev->tx.tx_lock, flags); 812 813 usb_kill_anchored_urbs(&hif_dev->mgmt_submitted); 814 } 815 816 static int ath9k_hif_usb_alloc_tx_urbs(struct hif_device_usb *hif_dev) 817 { 818 struct tx_buf *tx_buf; 819 int i; 820 821 INIT_LIST_HEAD(&hif_dev->tx.tx_buf); 822 INIT_LIST_HEAD(&hif_dev->tx.tx_pending); 823 spin_lock_init(&hif_dev->tx.tx_lock); 824 __skb_queue_head_init(&hif_dev->tx.tx_skb_queue); 825 init_usb_anchor(&hif_dev->mgmt_submitted); 826 827 for (i = 0; i < MAX_TX_URB_NUM; i++) { 828 tx_buf = kzalloc(sizeof(*tx_buf), GFP_KERNEL); 829 if (!tx_buf) 830 goto err; 831 832 tx_buf->buf = kzalloc(MAX_TX_BUF_SIZE, GFP_KERNEL); 833 if (!tx_buf->buf) 834 goto err; 835 836 tx_buf->urb = usb_alloc_urb(0, GFP_KERNEL); 837 if (!tx_buf->urb) 838 goto err; 839 840 tx_buf->hif_dev = hif_dev; 841 __skb_queue_head_init(&tx_buf->skb_queue); 842 843 list_add_tail(&tx_buf->list, &hif_dev->tx.tx_buf); 844 } 845 846 hif_dev->tx.tx_buf_cnt = MAX_TX_URB_NUM; 847 848 return 0; 849 err: 850 if (tx_buf) { 851 kfree(tx_buf->buf); 852 kfree(tx_buf); 853 } 854 ath9k_hif_usb_dealloc_tx_urbs(hif_dev); 855 return -ENOMEM; 856 } 857 858 static void ath9k_hif_usb_dealloc_rx_urbs(struct hif_device_usb *hif_dev) 859 { 860 usb_kill_anchored_urbs(&hif_dev->rx_submitted); 861 } 862 863 static int ath9k_hif_usb_alloc_rx_urbs(struct hif_device_usb *hif_dev) 864 { 865 struct rx_buf *rx_buf = NULL; 866 struct sk_buff *skb = NULL; 867 struct urb *urb = NULL; 868 int i, ret; 869 870 init_usb_anchor(&hif_dev->rx_submitted); 871 spin_lock_init(&hif_dev->rx_lock); 872 873 for (i = 0; i < MAX_RX_URB_NUM; i++) { 874 875 rx_buf = kzalloc(sizeof(*rx_buf), GFP_KERNEL); 876 if (!rx_buf) { 877 ret = -ENOMEM; 878 goto err_rxb; 879 } 880 881 /* Allocate URB */ 882 urb = usb_alloc_urb(0, GFP_KERNEL); 883 if (urb == NULL) { 884 ret = -ENOMEM; 885 goto err_urb; 886 } 887 888 /* Allocate buffer */ 889 skb = alloc_skb(MAX_RX_BUF_SIZE, GFP_KERNEL); 890 if (!skb) { 891 ret = -ENOMEM; 892 goto err_skb; 893 } 894 895 rx_buf->hif_dev = hif_dev; 896 rx_buf->skb = skb; 897 898 usb_fill_bulk_urb(urb, hif_dev->udev, 899 usb_rcvbulkpipe(hif_dev->udev, 900 USB_WLAN_RX_PIPE), 901 skb->data, MAX_RX_BUF_SIZE, 902 ath9k_hif_usb_rx_cb, rx_buf); 903 904 /* Anchor URB */ 905 usb_anchor_urb(urb, &hif_dev->rx_submitted); 906 907 /* Submit URB */ 908 ret = usb_submit_urb(urb, GFP_KERNEL); 909 if (ret) { 910 usb_unanchor_urb(urb); 911 goto err_submit; 912 } 913 914 /* 915 * Drop reference count. 916 * This ensures that the URB is freed when killing them. 917 */ 918 usb_free_urb(urb); 919 } 920 921 return 0; 922 923 err_submit: 924 kfree_skb(skb); 925 err_skb: 926 usb_free_urb(urb); 927 err_urb: 928 kfree(rx_buf); 929 err_rxb: 930 ath9k_hif_usb_dealloc_rx_urbs(hif_dev); 931 return ret; 932 } 933 934 static void ath9k_hif_usb_dealloc_reg_in_urbs(struct hif_device_usb *hif_dev) 935 { 936 usb_kill_anchored_urbs(&hif_dev->reg_in_submitted); 937 } 938 939 static int ath9k_hif_usb_alloc_reg_in_urbs(struct hif_device_usb *hif_dev) 940 { 941 struct rx_buf *rx_buf = NULL; 942 struct sk_buff *skb = NULL; 943 struct urb *urb = NULL; 944 int i, ret; 945 946 init_usb_anchor(&hif_dev->reg_in_submitted); 947 948 for (i = 0; i < MAX_REG_IN_URB_NUM; i++) { 949 950 rx_buf = kzalloc(sizeof(*rx_buf), GFP_KERNEL); 951 if (!rx_buf) { 952 ret = -ENOMEM; 953 goto err_rxb; 954 } 955 956 /* Allocate URB */ 957 urb = usb_alloc_urb(0, GFP_KERNEL); 958 if (urb == NULL) { 959 ret = -ENOMEM; 960 goto err_urb; 961 } 962 963 /* Allocate buffer */ 964 skb = alloc_skb(MAX_REG_IN_BUF_SIZE, GFP_KERNEL); 965 if (!skb) { 966 ret = -ENOMEM; 967 goto err_skb; 968 } 969 970 rx_buf->hif_dev = hif_dev; 971 rx_buf->skb = skb; 972 973 usb_fill_int_urb(urb, hif_dev->udev, 974 usb_rcvintpipe(hif_dev->udev, 975 USB_REG_IN_PIPE), 976 skb->data, MAX_REG_IN_BUF_SIZE, 977 ath9k_hif_usb_reg_in_cb, rx_buf, 1); 978 979 /* Anchor URB */ 980 usb_anchor_urb(urb, &hif_dev->reg_in_submitted); 981 982 /* Submit URB */ 983 ret = usb_submit_urb(urb, GFP_KERNEL); 984 if (ret) { 985 usb_unanchor_urb(urb); 986 goto err_submit; 987 } 988 989 /* 990 * Drop reference count. 991 * This ensures that the URB is freed when killing them. 992 */ 993 usb_free_urb(urb); 994 } 995 996 return 0; 997 998 err_submit: 999 kfree_skb(skb); 1000 err_skb: 1001 usb_free_urb(urb); 1002 err_urb: 1003 kfree(rx_buf); 1004 err_rxb: 1005 ath9k_hif_usb_dealloc_reg_in_urbs(hif_dev); 1006 return ret; 1007 } 1008 1009 static int ath9k_hif_usb_alloc_urbs(struct hif_device_usb *hif_dev) 1010 { 1011 /* Register Write */ 1012 init_usb_anchor(&hif_dev->regout_submitted); 1013 1014 /* TX */ 1015 if (ath9k_hif_usb_alloc_tx_urbs(hif_dev) < 0) 1016 goto err; 1017 1018 /* RX */ 1019 if (ath9k_hif_usb_alloc_rx_urbs(hif_dev) < 0) 1020 goto err_rx; 1021 1022 /* Register Read */ 1023 if (ath9k_hif_usb_alloc_reg_in_urbs(hif_dev) < 0) 1024 goto err_reg; 1025 1026 return 0; 1027 err_reg: 1028 ath9k_hif_usb_dealloc_rx_urbs(hif_dev); 1029 err_rx: 1030 ath9k_hif_usb_dealloc_tx_urbs(hif_dev); 1031 err: 1032 return -ENOMEM; 1033 } 1034 1035 void ath9k_hif_usb_dealloc_urbs(struct hif_device_usb *hif_dev) 1036 { 1037 usb_kill_anchored_urbs(&hif_dev->regout_submitted); 1038 ath9k_hif_usb_dealloc_reg_in_urbs(hif_dev); 1039 ath9k_hif_usb_dealloc_tx_urbs(hif_dev); 1040 ath9k_hif_usb_dealloc_rx_urbs(hif_dev); 1041 } 1042 1043 static int ath9k_hif_usb_download_fw(struct hif_device_usb *hif_dev) 1044 { 1045 int transfer, err; 1046 const void *data = hif_dev->fw_data; 1047 size_t len = hif_dev->fw_size; 1048 u32 addr = AR9271_FIRMWARE; 1049 u8 *buf = kzalloc(4096, GFP_KERNEL); 1050 u32 firm_offset; 1051 1052 if (!buf) 1053 return -ENOMEM; 1054 1055 while (len) { 1056 transfer = min_t(size_t, len, 4096); 1057 memcpy(buf, data, transfer); 1058 1059 err = usb_control_msg(hif_dev->udev, 1060 usb_sndctrlpipe(hif_dev->udev, 0), 1061 FIRMWARE_DOWNLOAD, 0x40 | USB_DIR_OUT, 1062 addr >> 8, 0, buf, transfer, 1063 USB_MSG_TIMEOUT); 1064 if (err < 0) { 1065 kfree(buf); 1066 return err; 1067 } 1068 1069 len -= transfer; 1070 data += transfer; 1071 addr += transfer; 1072 } 1073 kfree(buf); 1074 1075 if (IS_AR7010_DEVICE(hif_dev->usb_device_id->driver_info)) 1076 firm_offset = AR7010_FIRMWARE_TEXT; 1077 else 1078 firm_offset = AR9271_FIRMWARE_TEXT; 1079 1080 /* 1081 * Issue FW download complete command to firmware. 1082 */ 1083 err = usb_control_msg(hif_dev->udev, usb_sndctrlpipe(hif_dev->udev, 0), 1084 FIRMWARE_DOWNLOAD_COMP, 1085 0x40 | USB_DIR_OUT, 1086 firm_offset >> 8, 0, NULL, 0, USB_MSG_TIMEOUT); 1087 if (err) 1088 return -EIO; 1089 1090 dev_info(&hif_dev->udev->dev, "ath9k_htc: Transferred FW: %s, size: %ld\n", 1091 hif_dev->fw_name, (unsigned long) hif_dev->fw_size); 1092 1093 return 0; 1094 } 1095 1096 static int ath9k_hif_usb_dev_init(struct hif_device_usb *hif_dev) 1097 { 1098 int ret; 1099 1100 ret = ath9k_hif_usb_download_fw(hif_dev); 1101 if (ret) { 1102 dev_err(&hif_dev->udev->dev, 1103 "ath9k_htc: Firmware - %s download failed\n", 1104 hif_dev->fw_name); 1105 return ret; 1106 } 1107 1108 /* Alloc URBs */ 1109 ret = ath9k_hif_usb_alloc_urbs(hif_dev); 1110 if (ret) { 1111 dev_err(&hif_dev->udev->dev, 1112 "ath9k_htc: Unable to allocate URBs\n"); 1113 return ret; 1114 } 1115 1116 return 0; 1117 } 1118 1119 static void ath9k_hif_usb_dev_deinit(struct hif_device_usb *hif_dev) 1120 { 1121 ath9k_hif_usb_dealloc_urbs(hif_dev); 1122 } 1123 1124 /* 1125 * If initialization fails or the FW cannot be retrieved, 1126 * detach the device. 1127 */ 1128 static void ath9k_hif_usb_firmware_fail(struct hif_device_usb *hif_dev) 1129 { 1130 struct device *dev = &hif_dev->udev->dev; 1131 struct device *parent = dev->parent; 1132 1133 complete_all(&hif_dev->fw_done); 1134 1135 if (parent) 1136 device_lock(parent); 1137 1138 device_release_driver(dev); 1139 1140 if (parent) 1141 device_unlock(parent); 1142 } 1143 1144 static void ath9k_hif_usb_firmware_cb(const struct firmware *fw, void *context); 1145 1146 /* taken from iwlwifi */ 1147 static int ath9k_hif_request_firmware(struct hif_device_usb *hif_dev, 1148 bool first) 1149 { 1150 char index[8], *chip; 1151 int ret; 1152 1153 if (first) { 1154 if (htc_use_dev_fw) { 1155 hif_dev->fw_minor_index = FIRMWARE_MINOR_IDX_MAX + 1; 1156 sprintf(index, "%s", "dev"); 1157 } else { 1158 hif_dev->fw_minor_index = FIRMWARE_MINOR_IDX_MAX; 1159 sprintf(index, "%d", hif_dev->fw_minor_index); 1160 } 1161 } else { 1162 hif_dev->fw_minor_index--; 1163 sprintf(index, "%d", hif_dev->fw_minor_index); 1164 } 1165 1166 /* test for FW 1.3 */ 1167 if (MAJOR_VERSION_REQ == 1 && hif_dev->fw_minor_index == 3) { 1168 const char *filename; 1169 1170 if (IS_AR7010_DEVICE(hif_dev->usb_device_id->driver_info)) 1171 filename = FIRMWARE_AR7010_1_1; 1172 else 1173 filename = FIRMWARE_AR9271; 1174 1175 /* expected fw locations: 1176 * - htc_9271.fw (stable version 1.3, depricated) 1177 */ 1178 snprintf(hif_dev->fw_name, sizeof(hif_dev->fw_name), 1179 "%s", filename); 1180 1181 } else if (hif_dev->fw_minor_index < FIRMWARE_MINOR_IDX_MIN) { 1182 dev_err(&hif_dev->udev->dev, "no suitable firmware found!\n"); 1183 1184 return -ENOENT; 1185 } else { 1186 if (IS_AR7010_DEVICE(hif_dev->usb_device_id->driver_info)) 1187 chip = "7010"; 1188 else 1189 chip = "9271"; 1190 1191 /* expected fw locations: 1192 * - ath9k_htc/htc_9271-1.dev.0.fw (development version) 1193 * - ath9k_htc/htc_9271-1.4.0.fw (stable version) 1194 */ 1195 snprintf(hif_dev->fw_name, sizeof(hif_dev->fw_name), 1196 "%s/htc_%s-%d.%s.0.fw", HTC_FW_PATH, 1197 chip, MAJOR_VERSION_REQ, index); 1198 } 1199 1200 ret = request_firmware_nowait(THIS_MODULE, true, hif_dev->fw_name, 1201 &hif_dev->udev->dev, GFP_KERNEL, 1202 hif_dev, ath9k_hif_usb_firmware_cb); 1203 if (ret) { 1204 dev_err(&hif_dev->udev->dev, 1205 "ath9k_htc: Async request for firmware %s failed\n", 1206 hif_dev->fw_name); 1207 return ret; 1208 } 1209 1210 dev_info(&hif_dev->udev->dev, "ath9k_htc: Firmware %s requested\n", 1211 hif_dev->fw_name); 1212 1213 return ret; 1214 } 1215 1216 static void ath9k_hif_usb_firmware_cb(const struct firmware *fw, void *context) 1217 { 1218 struct hif_device_usb *hif_dev = context; 1219 int ret; 1220 1221 if (!fw) { 1222 ret = ath9k_hif_request_firmware(hif_dev, false); 1223 if (!ret) 1224 return; 1225 1226 dev_err(&hif_dev->udev->dev, 1227 "ath9k_htc: Failed to get firmware %s\n", 1228 hif_dev->fw_name); 1229 goto err_fw; 1230 } 1231 1232 hif_dev->htc_handle = ath9k_htc_hw_alloc(hif_dev, &hif_usb, 1233 &hif_dev->udev->dev); 1234 if (hif_dev->htc_handle == NULL) 1235 goto err_dev_alloc; 1236 1237 hif_dev->fw_data = fw->data; 1238 hif_dev->fw_size = fw->size; 1239 1240 /* Proceed with initialization */ 1241 1242 ret = ath9k_hif_usb_dev_init(hif_dev); 1243 if (ret) 1244 goto err_dev_init; 1245 1246 ret = ath9k_htc_hw_init(hif_dev->htc_handle, 1247 &hif_dev->interface->dev, 1248 hif_dev->usb_device_id->idProduct, 1249 hif_dev->udev->product, 1250 hif_dev->usb_device_id->driver_info); 1251 if (ret) { 1252 ret = -EINVAL; 1253 goto err_htc_hw_init; 1254 } 1255 1256 release_firmware(fw); 1257 hif_dev->flags |= HIF_USB_READY; 1258 complete_all(&hif_dev->fw_done); 1259 1260 return; 1261 1262 err_htc_hw_init: 1263 ath9k_hif_usb_dev_deinit(hif_dev); 1264 err_dev_init: 1265 ath9k_htc_hw_free(hif_dev->htc_handle); 1266 err_dev_alloc: 1267 release_firmware(fw); 1268 err_fw: 1269 ath9k_hif_usb_firmware_fail(hif_dev); 1270 } 1271 1272 /* 1273 * An exact copy of the function from zd1211rw. 1274 */ 1275 static int send_eject_command(struct usb_interface *interface) 1276 { 1277 struct usb_device *udev = interface_to_usbdev(interface); 1278 struct usb_host_interface *iface_desc = interface->cur_altsetting; 1279 struct usb_endpoint_descriptor *endpoint; 1280 unsigned char *cmd; 1281 u8 bulk_out_ep; 1282 int r; 1283 1284 if (iface_desc->desc.bNumEndpoints < 2) 1285 return -ENODEV; 1286 1287 /* Find bulk out endpoint */ 1288 for (r = 1; r >= 0; r--) { 1289 endpoint = &iface_desc->endpoint[r].desc; 1290 if (usb_endpoint_dir_out(endpoint) && 1291 usb_endpoint_xfer_bulk(endpoint)) { 1292 bulk_out_ep = endpoint->bEndpointAddress; 1293 break; 1294 } 1295 } 1296 if (r == -1) { 1297 dev_err(&udev->dev, 1298 "ath9k_htc: Could not find bulk out endpoint\n"); 1299 return -ENODEV; 1300 } 1301 1302 cmd = kzalloc(31, GFP_KERNEL); 1303 if (cmd == NULL) 1304 return -ENODEV; 1305 1306 /* USB bulk command block */ 1307 cmd[0] = 0x55; /* bulk command signature */ 1308 cmd[1] = 0x53; /* bulk command signature */ 1309 cmd[2] = 0x42; /* bulk command signature */ 1310 cmd[3] = 0x43; /* bulk command signature */ 1311 cmd[14] = 6; /* command length */ 1312 1313 cmd[15] = 0x1b; /* SCSI command: START STOP UNIT */ 1314 cmd[19] = 0x2; /* eject disc */ 1315 1316 dev_info(&udev->dev, "Ejecting storage device...\n"); 1317 r = usb_bulk_msg(udev, usb_sndbulkpipe(udev, bulk_out_ep), 1318 cmd, 31, NULL, 2 * USB_MSG_TIMEOUT); 1319 kfree(cmd); 1320 if (r) 1321 return r; 1322 1323 /* At this point, the device disconnects and reconnects with the real 1324 * ID numbers. */ 1325 1326 usb_set_intfdata(interface, NULL); 1327 return 0; 1328 } 1329 1330 static int ath9k_hif_usb_probe(struct usb_interface *interface, 1331 const struct usb_device_id *id) 1332 { 1333 struct usb_device *udev = interface_to_usbdev(interface); 1334 struct hif_device_usb *hif_dev; 1335 int ret = 0; 1336 1337 if (id->driver_info == STORAGE_DEVICE) 1338 return send_eject_command(interface); 1339 1340 hif_dev = kzalloc(sizeof(struct hif_device_usb), GFP_KERNEL); 1341 if (!hif_dev) { 1342 ret = -ENOMEM; 1343 goto err_alloc; 1344 } 1345 1346 usb_get_dev(udev); 1347 1348 hif_dev->udev = udev; 1349 hif_dev->interface = interface; 1350 hif_dev->usb_device_id = id; 1351 #ifdef CONFIG_PM 1352 udev->reset_resume = 1; 1353 #endif 1354 usb_set_intfdata(interface, hif_dev); 1355 1356 init_completion(&hif_dev->fw_done); 1357 1358 ret = ath9k_hif_request_firmware(hif_dev, true); 1359 if (ret) 1360 goto err_fw_req; 1361 1362 return ret; 1363 1364 err_fw_req: 1365 usb_set_intfdata(interface, NULL); 1366 kfree(hif_dev); 1367 usb_put_dev(udev); 1368 err_alloc: 1369 return ret; 1370 } 1371 1372 static void ath9k_hif_usb_reboot(struct usb_device *udev) 1373 { 1374 u32 reboot_cmd = 0xffffffff; 1375 void *buf; 1376 int ret; 1377 1378 buf = kmemdup(&reboot_cmd, 4, GFP_KERNEL); 1379 if (!buf) 1380 return; 1381 1382 ret = usb_interrupt_msg(udev, usb_sndintpipe(udev, USB_REG_OUT_PIPE), 1383 buf, 4, NULL, USB_MSG_TIMEOUT); 1384 if (ret) 1385 dev_err(&udev->dev, "ath9k_htc: USB reboot failed\n"); 1386 1387 kfree(buf); 1388 } 1389 1390 static void ath9k_hif_usb_disconnect(struct usb_interface *interface) 1391 { 1392 struct usb_device *udev = interface_to_usbdev(interface); 1393 struct hif_device_usb *hif_dev = usb_get_intfdata(interface); 1394 bool unplugged = (udev->state == USB_STATE_NOTATTACHED) ? true : false; 1395 1396 if (!hif_dev) 1397 return; 1398 1399 wait_for_completion(&hif_dev->fw_done); 1400 1401 if (hif_dev->flags & HIF_USB_READY) { 1402 ath9k_htc_hw_deinit(hif_dev->htc_handle, unplugged); 1403 ath9k_hif_usb_dev_deinit(hif_dev); 1404 ath9k_destroy_wmi(hif_dev->htc_handle->drv_priv); 1405 ath9k_htc_hw_free(hif_dev->htc_handle); 1406 } 1407 1408 usb_set_intfdata(interface, NULL); 1409 1410 /* If firmware was loaded we should drop it 1411 * go back to first stage bootloader. */ 1412 if (!unplugged && (hif_dev->flags & HIF_USB_READY)) 1413 ath9k_hif_usb_reboot(udev); 1414 1415 kfree(hif_dev); 1416 dev_info(&udev->dev, "ath9k_htc: USB layer deinitialized\n"); 1417 usb_put_dev(udev); 1418 } 1419 1420 #ifdef CONFIG_PM 1421 static int ath9k_hif_usb_suspend(struct usb_interface *interface, 1422 pm_message_t message) 1423 { 1424 struct hif_device_usb *hif_dev = usb_get_intfdata(interface); 1425 1426 /* 1427 * The device has to be set to FULLSLEEP mode in case no 1428 * interface is up. 1429 */ 1430 if (!(hif_dev->flags & HIF_USB_START)) 1431 ath9k_htc_suspend(hif_dev->htc_handle); 1432 1433 wait_for_completion(&hif_dev->fw_done); 1434 1435 if (hif_dev->flags & HIF_USB_READY) 1436 ath9k_hif_usb_dealloc_urbs(hif_dev); 1437 1438 return 0; 1439 } 1440 1441 static int ath9k_hif_usb_resume(struct usb_interface *interface) 1442 { 1443 struct hif_device_usb *hif_dev = usb_get_intfdata(interface); 1444 struct htc_target *htc_handle = hif_dev->htc_handle; 1445 int ret; 1446 const struct firmware *fw; 1447 1448 ret = ath9k_hif_usb_alloc_urbs(hif_dev); 1449 if (ret) 1450 return ret; 1451 1452 if (hif_dev->flags & HIF_USB_READY) { 1453 /* request cached firmware during suspend/resume cycle */ 1454 ret = request_firmware(&fw, hif_dev->fw_name, 1455 &hif_dev->udev->dev); 1456 if (ret) 1457 goto fail_resume; 1458 1459 hif_dev->fw_data = fw->data; 1460 hif_dev->fw_size = fw->size; 1461 ret = ath9k_hif_usb_download_fw(hif_dev); 1462 release_firmware(fw); 1463 if (ret) 1464 goto fail_resume; 1465 } else { 1466 ath9k_hif_usb_dealloc_urbs(hif_dev); 1467 return -EIO; 1468 } 1469 1470 mdelay(100); 1471 1472 ret = ath9k_htc_resume(htc_handle); 1473 1474 if (ret) 1475 goto fail_resume; 1476 1477 return 0; 1478 1479 fail_resume: 1480 ath9k_hif_usb_dealloc_urbs(hif_dev); 1481 1482 return ret; 1483 } 1484 #endif 1485 1486 static struct usb_driver ath9k_hif_usb_driver = { 1487 .name = KBUILD_MODNAME, 1488 .probe = ath9k_hif_usb_probe, 1489 .disconnect = ath9k_hif_usb_disconnect, 1490 #ifdef CONFIG_PM 1491 .suspend = ath9k_hif_usb_suspend, 1492 .resume = ath9k_hif_usb_resume, 1493 .reset_resume = ath9k_hif_usb_resume, 1494 #endif 1495 .id_table = ath9k_hif_usb_ids, 1496 .soft_unbind = 1, 1497 .disable_hub_initiated_lpm = 1, 1498 }; 1499 1500 int ath9k_hif_usb_init(void) 1501 { 1502 return usb_register(&ath9k_hif_usb_driver); 1503 } 1504 1505 void ath9k_hif_usb_exit(void) 1506 { 1507 usb_deregister(&ath9k_hif_usb_driver); 1508 } 1509