rt2x00usb.c 19.6 KB
Newer Older
1
/*
2 3
	Copyright (C) 2010 Willow Garage <http://www.willowgarage.com>
	Copyright (C) 2004 - 2010 Ivo van Doorn <IvDoorn@gmail.com>
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
	<http://rt2x00.serialmonkey.com>

	This program is free software; you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation; either version 2 of the License, or
	(at your option) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program; if not, write to the
	Free Software Foundation, Inc.,
	59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */

/*
	Module: rt2x00usb
	Abstract: rt2x00 generic usb device routines.
 */

#include <linux/kernel.h>
#include <linux/module.h>
29
#include <linux/slab.h>
30
#include <linux/usb.h>
31
#include <linux/bug.h>
32 33 34 35 36 37 38

#include "rt2x00.h"
#include "rt2x00usb.h"

/*
 * Interfacing with the HW.
 */
A
Adam Baker 已提交
39
int rt2x00usb_vendor_request(struct rt2x00_dev *rt2x00dev,
40 41 42
			     const u8 request, const u8 requesttype,
			     const u16 offset, const u16 value,
			     void *buffer, const u16 buffer_length,
43
			     const int timeout)
44
{
45
	struct usb_device *usb_dev = to_usb_device_intf(rt2x00dev->dev);
46 47 48 49 50 51
	int status;
	unsigned int i;
	unsigned int pipe =
	    (requesttype == USB_VENDOR_REQUEST_IN) ?
	    usb_rcvctrlpipe(usb_dev, 0) : usb_sndctrlpipe(usb_dev, 0);

52 53
	if (!test_bit(DEVICE_STATE_PRESENT, &rt2x00dev->flags))
		return -ENODEV;
54

55 56 57 58 59 60 61 62
	for (i = 0; i < REGISTER_BUSY_COUNT; i++) {
		status = usb_control_msg(usb_dev, pipe, request, requesttype,
					 value, offset, buffer, buffer_length,
					 timeout);
		if (status >= 0)
			return 0;

		/*
63
		 * Check for errors
64
		 * -ENODEV: Device has disappeared, no point continuing.
65
		 * All other errors: Try again.
66
		 */
67 68
		else if (status == -ENODEV) {
			clear_bit(DEVICE_STATE_PRESENT, &rt2x00dev->flags);
69
			break;
70
		}
71 72 73 74 75 76 77 78 79 80
	}

	ERROR(rt2x00dev,
	      "Vendor Request 0x%02x failed for offset 0x%04x with error %d.\n",
	      request, offset, status);

	return status;
}
EXPORT_SYMBOL_GPL(rt2x00usb_vendor_request);

81 82 83 84
int rt2x00usb_vendor_req_buff_lock(struct rt2x00_dev *rt2x00dev,
				   const u8 request, const u8 requesttype,
				   const u16 offset, void *buffer,
				   const u16 buffer_length, const int timeout)
85 86 87
{
	int status;

88
	BUG_ON(!mutex_is_locked(&rt2x00dev->csr_mutex));
89

90 91 92
	/*
	 * Check for Cache availability.
	 */
93
	if (unlikely(!rt2x00dev->csr.cache || buffer_length > CSR_CACHE_SIZE)) {
94 95 96 97 98
		ERROR(rt2x00dev, "CSR cache not available.\n");
		return -ENOMEM;
	}

	if (requesttype == USB_VENDOR_REQUEST_OUT)
99
		memcpy(rt2x00dev->csr.cache, buffer, buffer_length);
100 101

	status = rt2x00usb_vendor_request(rt2x00dev, request, requesttype,
102
					  offset, 0, rt2x00dev->csr.cache,
103 104 105
					  buffer_length, timeout);

	if (!status && requesttype == USB_VENDOR_REQUEST_IN)
106
		memcpy(buffer, rt2x00dev->csr.cache, buffer_length);
107 108 109

	return status;
}
110 111 112 113 114 115
EXPORT_SYMBOL_GPL(rt2x00usb_vendor_req_buff_lock);

int rt2x00usb_vendor_request_buff(struct rt2x00_dev *rt2x00dev,
				  const u8 request, const u8 requesttype,
				  const u16 offset, void *buffer,
				  const u16 buffer_length, const int timeout)
116 117 118 119 120
{
	int status = 0;
	unsigned char *tb;
	u16 off, len, bsize;

121
	mutex_lock(&rt2x00dev->csr_mutex);
122

I
Ivo van Doorn 已提交
123
	tb  = (char *)buffer;
124 125 126 127 128 129 130 131 132 133 134 135 136
	off = offset;
	len = buffer_length;
	while (len && !status) {
		bsize = min_t(u16, CSR_CACHE_SIZE, len);
		status = rt2x00usb_vendor_req_buff_lock(rt2x00dev, request,
							requesttype, off, tb,
							bsize, timeout);

		tb  += bsize;
		len -= bsize;
		off += bsize;
	}

137
	mutex_unlock(&rt2x00dev->csr_mutex);
138 139 140

	return status;
}
141
EXPORT_SYMBOL_GPL(rt2x00usb_vendor_request_buff);
142

143 144
int rt2x00usb_regbusy_read(struct rt2x00_dev *rt2x00dev,
			   const unsigned int offset,
145
			   const struct rt2x00_field32 field,
146 147 148 149
			   u32 *reg)
{
	unsigned int i;

150 151 152
	if (!test_bit(DEVICE_STATE_PRESENT, &rt2x00dev->flags))
		return -ENODEV;

153 154 155 156 157 158 159 160 161 162 163 164 165 166 167
	for (i = 0; i < REGISTER_BUSY_COUNT; i++) {
		rt2x00usb_register_read_lock(rt2x00dev, offset, reg);
		if (!rt2x00_get_field32(*reg, field))
			return 1;
		udelay(REGISTER_BUSY_DELAY);
	}

	ERROR(rt2x00dev, "Indirect register access failed: "
	      "offset=0x%.08x, value=0x%.08x\n", offset, *reg);
	*reg = ~0;

	return 0;
}
EXPORT_SYMBOL_GPL(rt2x00usb_regbusy_read);

168 169 170
/*
 * TX data handlers.
 */
171
static void rt2x00usb_work_txdone_entry(struct queue_entry *entry)
172 173
{
	/*
174
	 * If the transfer to hardware succeeded, it does not mean the
I
Ivo van Doorn 已提交
175 176 177 178 179
	 * frame was send out correctly. It only means the frame
	 * was succesfully pushed to the hardware, we have no
	 * way to determine the transmission status right now.
	 * (Only indirectly by looking at the failed TX counters
	 * in the register).
180
	 */
181
	if (test_bit(ENTRY_DATA_IO_FAILED, &entry->flags))
182
		rt2x00lib_txdone_noinfo(entry, TXDONE_FAILURE);
183
	else
184
		rt2x00lib_txdone_noinfo(entry, TXDONE_UNKNOWN);
185 186
}

187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210
static void rt2x00usb_work_txdone(struct work_struct *work)
{
	struct rt2x00_dev *rt2x00dev =
	    container_of(work, struct rt2x00_dev, txdone_work);
	struct data_queue *queue;
	struct queue_entry *entry;

	tx_queue_for_each(rt2x00dev, queue) {
		while (!rt2x00queue_empty(queue)) {
			entry = rt2x00queue_get_entry(queue, Q_INDEX_DONE);

			if (test_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags))
				break;

			rt2x00usb_work_txdone_entry(entry);
		}
	}
}

static void rt2x00usb_interrupt_txdone(struct urb *urb)
{
	struct queue_entry *entry = (struct queue_entry *)urb->context;
	struct rt2x00_dev *rt2x00dev = entry->queue->rt2x00dev;

211
	if (!test_and_clear_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags))
212 213
		return;

214 215 216 217 218
	/*
	 * Report the frame as DMA done
	 */
	rt2x00lib_dmadone(entry);

219 220 221 222
	/*
	 * Check if the frame was correctly uploaded
	 */
	if (urb->status)
223
		set_bit(ENTRY_DATA_IO_FAILED, &entry->flags);
224 225 226 227 228

	/*
	 * Schedule the delayed work for reading the TX status
	 * from the device.
	 */
I
Ivo van Doorn 已提交
229
	ieee80211_queue_work(rt2x00dev->hw, &rt2x00dev->txdone_work);
230 231
}

232
static void rt2x00usb_kick_tx_entry(struct queue_entry *entry)
233
{
234 235
	struct rt2x00_dev *rt2x00dev = entry->queue->rt2x00dev;
	struct usb_device *usb_dev = to_usb_device_intf(rt2x00dev->dev);
236
	struct queue_entry_priv_usb *entry_priv = entry->priv_data;
237
	u32 length;
238
	int status;
239

I
Ivo van Doorn 已提交
240 241
	if (!test_and_clear_bit(ENTRY_DATA_PENDING, &entry->flags))
		return;
242

I
Ivo van Doorn 已提交
243 244 245 246 247 248
	/*
	 * USB devices cannot blindly pass the skb->len as the
	 * length of the data to usb_fill_bulk_urb. Pass the skb
	 * to the driver to determine what the length should be.
	 */
	length = rt2x00dev->ops->lib->get_tx_data_len(entry);
249

I
Ivo van Doorn 已提交
250 251 252 253 254
	usb_fill_bulk_urb(entry_priv->urb, usb_dev,
			  usb_sndbulkpipe(usb_dev, entry->queue->usb_endpoint),
			  entry->skb->data, length,
			  rt2x00usb_interrupt_txdone, entry);

255 256 257 258
	status = usb_submit_urb(entry_priv->urb, GFP_ATOMIC);
	if (status) {
		if (status == -ENODEV)
			clear_bit(DEVICE_STATE_PRESENT, &rt2x00dev->flags);
I
Ivo van Doorn 已提交
259 260 261
		set_bit(ENTRY_DATA_IO_FAILED, &entry->flags);
		rt2x00lib_dmadone(entry);
	}
262 263
}

264
void rt2x00usb_kick_tx_queue(struct data_queue *queue)
265
{
266 267
	rt2x00queue_for_each_entry(queue, Q_INDEX_DONE, Q_INDEX,
				   rt2x00usb_kick_tx_entry);
268 269 270
}
EXPORT_SYMBOL_GPL(rt2x00usb_kick_tx_queue);

271
static void rt2x00usb_kill_entry(struct queue_entry *entry)
272
{
273 274 275
	struct rt2x00_dev *rt2x00dev = entry->queue->rt2x00dev;
	struct queue_entry_priv_usb *entry_priv = entry->priv_data;
	struct queue_entry_priv_usb_bcn *bcn_priv = entry->priv_data;
276

277 278 279 280
	if (!test_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags))
		return;

	usb_kill_urb(entry_priv->urb);
281 282

	/*
283
	 * Kill guardian urb (if required by driver).
284
	 */
285 286 287 288
	if ((entry->queue->qid == QID_BEACON) &&
	    (test_bit(DRIVER_REQUIRE_BEACON_GUARD, &rt2x00dev->flags)))
		usb_kill_urb(bcn_priv->guardian_urb);
}
289

290
void rt2x00usb_stop_queue(struct data_queue *queue)
291 292
{
	rt2x00queue_for_each_entry(queue, Q_INDEX_DONE, Q_INDEX,
293
				   rt2x00usb_kill_entry);
294
}
295
EXPORT_SYMBOL_GPL(rt2x00usb_stop_queue);
296

297
static void rt2x00usb_watchdog_tx_dma(struct data_queue *queue)
298
{
299
	struct rt2x00_dev *rt2x00dev = queue->rt2x00dev;
300 301
	unsigned short threshold = queue->threshold;

302
	WARNING(queue->rt2x00dev, "TX queue %d DMA timed out,"
303
		" invoke forced forced reset\n", queue->qid);
304 305 306 307 308 309 310 311 312

	/*
	 * Temporarily disable the TX queue, this will force mac80211
	 * to use the other queues until this queue has been restored.
	 *
	 * Set the queue threshold to the queue limit. This prevents the
	 * queue from being enabled during the txdone handler.
	 */
	queue->threshold = queue->limit;
313
	ieee80211_stop_queue(rt2x00dev->hw, queue->qid);
314 315

	/*
316 317
	 * Kill all entries in the queue, afterwards we need to
	 * wait a bit for all URBs to be cancelled.
318
	 */
319
	rt2x00usb_stop_queue(queue);
320

321 322 323 324 325
	/*
	 * In case that a driver has overriden the txdone_work
	 * function, we invoke the TX done through there.
	 */
	rt2x00dev->txdone_work.func(&rt2x00dev->txdone_work);
326

327 328 329 330 331
	/*
	 * The queue has been reset, and mac80211 is allowed to use the
	 * queue again.
	 */
	queue->threshold = threshold;
332
	ieee80211_wake_queue(rt2x00dev->hw, queue->qid);
333 334
}

335 336 337
static void rt2x00usb_watchdog_tx_status(struct data_queue *queue)
{
	WARNING(queue->rt2x00dev, "TX queue %d status timed out,"
338
		" invoke forced tx handler\n", queue->qid);
339 340 341 342

	ieee80211_queue_work(queue->rt2x00dev->hw, &queue->rt2x00dev->txdone_work);
}

343 344 345 346 347
void rt2x00usb_watchdog(struct rt2x00_dev *rt2x00dev)
{
	struct data_queue *queue;

	tx_queue_for_each(rt2x00dev, queue) {
348 349 350
		if (!rt2x00queue_empty(queue)) {
			if (rt2x00queue_dma_timeout(queue))
				rt2x00usb_watchdog_tx_dma(queue);
I
Ivo van Doorn 已提交
351
			if (rt2x00queue_status_timeout(queue))
352 353
				rt2x00usb_watchdog_tx_status(queue);
		}
354 355 356 357
	}
}
EXPORT_SYMBOL_GPL(rt2x00usb_watchdog);

358 359 360
/*
 * RX data handlers.
 */
361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384
static void rt2x00usb_work_rxdone(struct work_struct *work)
{
	struct rt2x00_dev *rt2x00dev =
	    container_of(work, struct rt2x00_dev, rxdone_work);
	struct queue_entry *entry;
	struct skb_frame_desc *skbdesc;
	u8 rxd[32];

	while (!rt2x00queue_empty(rt2x00dev->rx)) {
		entry = rt2x00queue_get_entry(rt2x00dev->rx, Q_INDEX_DONE);

		if (test_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags))
			break;

		/*
		 * Fill in desc fields of the skb descriptor
		 */
		skbdesc = get_skb_frame_desc(entry->skb);
		skbdesc->desc = rxd;
		skbdesc->desc_len = entry->queue->desc_size;

		/*
		 * Send the frame to rt2x00lib for further processing.
		 */
385
		rt2x00lib_rxdone(entry);
386 387 388
	}
}

389 390
static void rt2x00usb_interrupt_rxdone(struct urb *urb)
{
I
Ivo van Doorn 已提交
391 392
	struct queue_entry *entry = (struct queue_entry *)urb->context;
	struct rt2x00_dev *rt2x00dev = entry->queue->rt2x00dev;
393

394
	if (!test_and_clear_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags))
395 396
		return;

397 398 399 400 401
	/*
	 * Report the frame as DMA done
	 */
	rt2x00lib_dmadone(entry);

402 403 404 405 406
	/*
	 * Check if the received data is simply too small
	 * to be actually valid, or if the urb is signaling
	 * a problem.
	 */
407
	if (urb->actual_length < entry->queue->desc_size || urb->status)
408
		set_bit(ENTRY_DATA_IO_FAILED, &entry->flags);
409

410
	/*
411 412
	 * Schedule the delayed work for reading the RX status
	 * from the device.
413
	 */
I
Ivo van Doorn 已提交
414
	ieee80211_queue_work(rt2x00dev->hw, &rt2x00dev->rxdone_work);
415 416 417 418 419 420 421
}

/*
 * Radio handlers
 */
void rt2x00usb_disable_radio(struct rt2x00_dev *rt2x00dev)
{
I
Ivo van Doorn 已提交
422
	rt2x00usb_vendor_request_sw(rt2x00dev, USB_RX_CONTROL, 0, 0,
423 424 425
				    REGISTER_TIMEOUT);

	/*
426
	 * The USB version of also works
427
	 * on the RX queue.
428
	 */
429
	rt2x00dev->ops->lib->kill_tx_queue(rt2x00dev->rx);
430 431 432 433 434 435
}
EXPORT_SYMBOL_GPL(rt2x00usb_disable_radio);

/*
 * Device initialization handlers.
 */
436
void rt2x00usb_clear_entry(struct queue_entry *entry)
437
{
438 439
	struct usb_device *usb_dev =
	    to_usb_device_intf(entry->queue->rt2x00dev->dev);
440
	struct queue_entry_priv_usb *entry_priv = entry->priv_data;
441
	int pipe;
442
	int status;
443

444 445
	entry->flags = 0;

446
	if (entry->queue->qid == QID_RX) {
447 448
		pipe = usb_rcvbulkpipe(usb_dev, entry->queue->usb_endpoint);
		usb_fill_bulk_urb(entry_priv->urb, usb_dev, pipe,
449 450
				entry->skb->data, entry->skb->len,
				rt2x00usb_interrupt_rxdone, entry);
451

452
		set_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags);
453 454 455 456 457 458

		status = usb_submit_urb(entry_priv->urb, GFP_ATOMIC);
		if (status) {
			if (status == -ENODEV)
				clear_bit(DEVICE_STATE_PRESENT,
					  &entry->queue->rt2x00dev->flags);
I
Ivo van Doorn 已提交
459 460 461
			set_bit(ENTRY_DATA_IO_FAILED, &entry->flags);
			rt2x00lib_dmadone(entry);
		}
462
	}
463
}
464
EXPORT_SYMBOL_GPL(rt2x00usb_clear_entry);
465

466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505
static void rt2x00usb_assign_endpoint(struct data_queue *queue,
				      struct usb_endpoint_descriptor *ep_desc)
{
	struct usb_device *usb_dev = to_usb_device_intf(queue->rt2x00dev->dev);
	int pipe;

	queue->usb_endpoint = usb_endpoint_num(ep_desc);

	if (queue->qid == QID_RX) {
		pipe = usb_rcvbulkpipe(usb_dev, queue->usb_endpoint);
		queue->usb_maxpacket = usb_maxpacket(usb_dev, pipe, 0);
	} else {
		pipe = usb_sndbulkpipe(usb_dev, queue->usb_endpoint);
		queue->usb_maxpacket = usb_maxpacket(usb_dev, pipe, 1);
	}

	if (!queue->usb_maxpacket)
		queue->usb_maxpacket = 1;
}

static int rt2x00usb_find_endpoints(struct rt2x00_dev *rt2x00dev)
{
	struct usb_interface *intf = to_usb_interface(rt2x00dev->dev);
	struct usb_host_interface *intf_desc = intf->cur_altsetting;
	struct usb_endpoint_descriptor *ep_desc;
	struct data_queue *queue = rt2x00dev->tx;
	struct usb_endpoint_descriptor *tx_ep_desc = NULL;
	unsigned int i;

	/*
	 * Walk through all available endpoints to search for "bulk in"
	 * and "bulk out" endpoints. When we find such endpoints collect
	 * the information we need from the descriptor and assign it
	 * to the queue.
	 */
	for (i = 0; i < intf_desc->desc.bNumEndpoints; i++) {
		ep_desc = &intf_desc->endpoint[i].desc;

		if (usb_endpoint_is_bulk_in(ep_desc)) {
			rt2x00usb_assign_endpoint(rt2x00dev->rx, ep_desc);
I
Ivo van Doorn 已提交
506 507
		} else if (usb_endpoint_is_bulk_out(ep_desc) &&
			   (queue != queue_end(rt2x00dev))) {
508
			rt2x00usb_assign_endpoint(queue, ep_desc);
I
Ivo van Doorn 已提交
509
			queue = queue_next(queue);
510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535

			tx_ep_desc = ep_desc;
		}
	}

	/*
	 * At least 1 endpoint for RX and 1 endpoint for TX must be available.
	 */
	if (!rt2x00dev->rx->usb_endpoint || !rt2x00dev->tx->usb_endpoint) {
		ERROR(rt2x00dev, "Bulk-in/Bulk-out endpoints not found\n");
		return -EPIPE;
	}

	/*
	 * It might be possible not all queues have a dedicated endpoint.
	 * Loop through all TX queues and copy the endpoint information
	 * which we have gathered from already assigned endpoints.
	 */
	txall_queue_for_each(rt2x00dev, queue) {
		if (!queue->usb_endpoint)
			rt2x00usb_assign_endpoint(queue, tx_ep_desc);
	}

	return 0;
}

536
static int rt2x00usb_alloc_entries(struct data_queue *queue)
537
{
538
	struct rt2x00_dev *rt2x00dev = queue->rt2x00dev;
539 540
	struct queue_entry_priv_usb *entry_priv;
	struct queue_entry_priv_usb_bcn *bcn_priv;
541 542
	unsigned int i;

543 544 545 546 547 548 549
	for (i = 0; i < queue->limit; i++) {
		entry_priv = queue->entries[i].priv_data;
		entry_priv->urb = usb_alloc_urb(0, GFP_KERNEL);
		if (!entry_priv->urb)
			return -ENOMEM;
	}

550
	/*
551 552 553
	 * If this is not the beacon queue or
	 * no guardian byte was required for the beacon,
	 * then we are done.
554
	 */
555
	if (queue->qid != QID_BEACON ||
556 557 558
	    !test_bit(DRIVER_REQUIRE_BEACON_GUARD, &rt2x00dev->flags))
		return 0;

I
Ivo van Doorn 已提交
559
	for (i = 0; i < queue->limit; i++) {
560 561 562
		bcn_priv = queue->entries[i].priv_data;
		bcn_priv->guardian_urb = usb_alloc_urb(0, GFP_KERNEL);
		if (!bcn_priv->guardian_urb)
563 564 565 566 567 568
			return -ENOMEM;
	}

	return 0;
}

569
static void rt2x00usb_free_entries(struct data_queue *queue)
570
{
571
	struct rt2x00_dev *rt2x00dev = queue->rt2x00dev;
572 573
	struct queue_entry_priv_usb *entry_priv;
	struct queue_entry_priv_usb_bcn *bcn_priv;
574 575
	unsigned int i;

I
Ivo van Doorn 已提交
576
	if (!queue->entries)
577 578
		return;

I
Ivo van Doorn 已提交
579
	for (i = 0; i < queue->limit; i++) {
580 581 582
		entry_priv = queue->entries[i].priv_data;
		usb_kill_urb(entry_priv->urb);
		usb_free_urb(entry_priv->urb);
583
	}
584 585 586 587 588 589

	/*
	 * If this is not the beacon queue or
	 * no guardian byte was required for the beacon,
	 * then we are done.
	 */
590
	if (queue->qid != QID_BEACON ||
591 592 593 594 595 596 597 598
	    !test_bit(DRIVER_REQUIRE_BEACON_GUARD, &rt2x00dev->flags))
		return;

	for (i = 0; i < queue->limit; i++) {
		bcn_priv = queue->entries[i].priv_data;
		usb_kill_urb(bcn_priv->guardian_urb);
		usb_free_urb(bcn_priv->guardian_urb);
	}
599 600 601 602
}

int rt2x00usb_initialize(struct rt2x00_dev *rt2x00dev)
{
I
Ivo van Doorn 已提交
603
	struct data_queue *queue;
604
	int status;
605

606 607 608 609 610 611 612
	/*
	 * Find endpoints for each queue
	 */
	status = rt2x00usb_find_endpoints(rt2x00dev);
	if (status)
		goto exit;

613 614 615
	/*
	 * Allocate DMA
	 */
I
Ivo van Doorn 已提交
616
	queue_for_each(rt2x00dev, queue) {
617
		status = rt2x00usb_alloc_entries(queue);
618 619 620 621 622 623 624 625 626 627 628 629 630 631 632
		if (status)
			goto exit;
	}

	return 0;

exit:
	rt2x00usb_uninitialize(rt2x00dev);

	return status;
}
EXPORT_SYMBOL_GPL(rt2x00usb_initialize);

void rt2x00usb_uninitialize(struct rt2x00_dev *rt2x00dev)
{
I
Ivo van Doorn 已提交
633
	struct data_queue *queue;
634

I
Ivo van Doorn 已提交
635
	queue_for_each(rt2x00dev, queue)
636
		rt2x00usb_free_entries(queue);
637 638 639 640 641 642 643 644 645 646 647 648 649 650
}
EXPORT_SYMBOL_GPL(rt2x00usb_uninitialize);

/*
 * USB driver handlers.
 */
static void rt2x00usb_free_reg(struct rt2x00_dev *rt2x00dev)
{
	kfree(rt2x00dev->rf);
	rt2x00dev->rf = NULL;

	kfree(rt2x00dev->eeprom);
	rt2x00dev->eeprom = NULL;

651 652
	kfree(rt2x00dev->csr.cache);
	rt2x00dev->csr.cache = NULL;
653 654 655 656
}

static int rt2x00usb_alloc_reg(struct rt2x00_dev *rt2x00dev)
{
657 658
	rt2x00dev->csr.cache = kzalloc(CSR_CACHE_SIZE, GFP_KERNEL);
	if (!rt2x00dev->csr.cache)
659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699
		goto exit;

	rt2x00dev->eeprom = kzalloc(rt2x00dev->ops->eeprom_size, GFP_KERNEL);
	if (!rt2x00dev->eeprom)
		goto exit;

	rt2x00dev->rf = kzalloc(rt2x00dev->ops->rf_size, GFP_KERNEL);
	if (!rt2x00dev->rf)
		goto exit;

	return 0;

exit:
	ERROR_PROBE("Failed to allocate registers.\n");

	rt2x00usb_free_reg(rt2x00dev);

	return -ENOMEM;
}

int rt2x00usb_probe(struct usb_interface *usb_intf,
		    const struct usb_device_id *id)
{
	struct usb_device *usb_dev = interface_to_usbdev(usb_intf);
	struct rt2x00_ops *ops = (struct rt2x00_ops *)id->driver_info;
	struct ieee80211_hw *hw;
	struct rt2x00_dev *rt2x00dev;
	int retval;

	usb_dev = usb_get_dev(usb_dev);

	hw = ieee80211_alloc_hw(sizeof(struct rt2x00_dev), ops->hw);
	if (!hw) {
		ERROR_PROBE("Failed to allocate hardware.\n");
		retval = -ENOMEM;
		goto exit_put_device;
	}

	usb_set_intfdata(usb_intf, hw);

	rt2x00dev = hw->priv;
700
	rt2x00dev->dev = &usb_intf->dev;
701 702 703
	rt2x00dev->ops = ops;
	rt2x00dev->hw = hw;

704 705
	rt2x00_set_chip_intf(rt2x00dev, RT2X00_CHIP_INTF_USB);

706 707 708
	INIT_WORK(&rt2x00dev->rxdone_work, rt2x00usb_work_rxdone);
	INIT_WORK(&rt2x00dev->txdone_work, rt2x00usb_work_txdone);

709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780
	retval = rt2x00usb_alloc_reg(rt2x00dev);
	if (retval)
		goto exit_free_device;

	retval = rt2x00lib_probe_dev(rt2x00dev);
	if (retval)
		goto exit_free_reg;

	return 0;

exit_free_reg:
	rt2x00usb_free_reg(rt2x00dev);

exit_free_device:
	ieee80211_free_hw(hw);

exit_put_device:
	usb_put_dev(usb_dev);

	usb_set_intfdata(usb_intf, NULL);

	return retval;
}
EXPORT_SYMBOL_GPL(rt2x00usb_probe);

void rt2x00usb_disconnect(struct usb_interface *usb_intf)
{
	struct ieee80211_hw *hw = usb_get_intfdata(usb_intf);
	struct rt2x00_dev *rt2x00dev = hw->priv;

	/*
	 * Free all allocated data.
	 */
	rt2x00lib_remove_dev(rt2x00dev);
	rt2x00usb_free_reg(rt2x00dev);
	ieee80211_free_hw(hw);

	/*
	 * Free the USB device data.
	 */
	usb_set_intfdata(usb_intf, NULL);
	usb_put_dev(interface_to_usbdev(usb_intf));
}
EXPORT_SYMBOL_GPL(rt2x00usb_disconnect);

#ifdef CONFIG_PM
int rt2x00usb_suspend(struct usb_interface *usb_intf, pm_message_t state)
{
	struct ieee80211_hw *hw = usb_get_intfdata(usb_intf);
	struct rt2x00_dev *rt2x00dev = hw->priv;
	int retval;

	retval = rt2x00lib_suspend(rt2x00dev, state);
	if (retval)
		return retval;

	/*
	 * Decrease usbdev refcount.
	 */
	usb_put_dev(interface_to_usbdev(usb_intf));

	return 0;
}
EXPORT_SYMBOL_GPL(rt2x00usb_suspend);

int rt2x00usb_resume(struct usb_interface *usb_intf)
{
	struct ieee80211_hw *hw = usb_get_intfdata(usb_intf);
	struct rt2x00_dev *rt2x00dev = hw->priv;

	usb_get_dev(interface_to_usbdev(usb_intf));

781
	return rt2x00lib_resume(rt2x00dev);
782 783 784 785 786
}
EXPORT_SYMBOL_GPL(rt2x00usb_resume);
#endif /* CONFIG_PM */

/*
I
Ivo van Doorn 已提交
787
 * rt2x00usb module information.
788 789 790
 */
MODULE_AUTHOR(DRV_PROJECT);
MODULE_VERSION(DRV_VERSION);
I
Ivo van Doorn 已提交
791
MODULE_DESCRIPTION("rt2x00 usb library");
792
MODULE_LICENSE("GPL");