rt2x00usb.c 19.7 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_queue(struct data_queue *queue)
265
{
266 267 268 269 270 271 272 273 274 275 276 277
	switch (queue->qid) {
	case QID_AC_BE:
	case QID_AC_BK:
	case QID_AC_VI:
	case QID_AC_VO:
		if (!rt2x00queue_empty(queue))
			rt2x00queue_for_each_entry(queue, Q_INDEX_DONE, Q_INDEX,
						   rt2x00usb_kick_tx_entry);
		break;
	default:
		break;
	}
278
}
279
EXPORT_SYMBOL_GPL(rt2x00usb_kick_queue);
280

281
static void rt2x00usb_kill_entry(struct queue_entry *entry)
282
{
283 284 285
	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;
286

287 288 289 290
	if (!test_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags))
		return;

	usb_kill_urb(entry_priv->urb);
291 292

	/*
293
	 * Kill guardian urb (if required by driver).
294
	 */
295 296 297 298
	if ((entry->queue->qid == QID_BEACON) &&
	    (test_bit(DRIVER_REQUIRE_BEACON_GUARD, &rt2x00dev->flags)))
		usb_kill_urb(bcn_priv->guardian_urb);
}
299

300
void rt2x00usb_stop_queue(struct data_queue *queue)
301 302
{
	rt2x00queue_for_each_entry(queue, Q_INDEX_DONE, Q_INDEX,
303
				   rt2x00usb_kill_entry);
304
}
305
EXPORT_SYMBOL_GPL(rt2x00usb_stop_queue);
306

307
static void rt2x00usb_watchdog_tx_dma(struct data_queue *queue)
308
{
309
	struct rt2x00_dev *rt2x00dev = queue->rt2x00dev;
310 311
	unsigned short threshold = queue->threshold;

312
	WARNING(queue->rt2x00dev, "TX queue %d DMA timed out,"
313
		" invoke forced forced reset\n", queue->qid);
314 315 316 317 318 319 320 321 322

	/*
	 * 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;
323
	ieee80211_stop_queue(rt2x00dev->hw, queue->qid);
324 325

	/*
326 327
	 * Kill all entries in the queue, afterwards we need to
	 * wait a bit for all URBs to be cancelled.
328
	 */
329
	rt2x00usb_stop_queue(queue);
330

331 332 333 334 335
	/*
	 * 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);
336

337 338 339 340 341
	/*
	 * The queue has been reset, and mac80211 is allowed to use the
	 * queue again.
	 */
	queue->threshold = threshold;
342
	ieee80211_wake_queue(rt2x00dev->hw, queue->qid);
343 344
}

345 346 347
static void rt2x00usb_watchdog_tx_status(struct data_queue *queue)
{
	WARNING(queue->rt2x00dev, "TX queue %d status timed out,"
348
		" invoke forced tx handler\n", queue->qid);
349 350 351 352

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

353 354 355 356 357
void rt2x00usb_watchdog(struct rt2x00_dev *rt2x00dev)
{
	struct data_queue *queue;

	tx_queue_for_each(rt2x00dev, queue) {
358 359 360
		if (!rt2x00queue_empty(queue)) {
			if (rt2x00queue_dma_timeout(queue))
				rt2x00usb_watchdog_tx_dma(queue);
I
Ivo van Doorn 已提交
361
			if (rt2x00queue_status_timeout(queue))
362 363
				rt2x00usb_watchdog_tx_status(queue);
		}
364 365 366 367
	}
}
EXPORT_SYMBOL_GPL(rt2x00usb_watchdog);

368 369 370
/*
 * RX data handlers.
 */
371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394
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.
		 */
395
		rt2x00lib_rxdone(entry);
396 397 398
	}
}

399 400
static void rt2x00usb_interrupt_rxdone(struct urb *urb)
{
I
Ivo van Doorn 已提交
401 402
	struct queue_entry *entry = (struct queue_entry *)urb->context;
	struct rt2x00_dev *rt2x00dev = entry->queue->rt2x00dev;
403

404
	if (!test_and_clear_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags))
405 406
		return;

407 408 409 410 411
	/*
	 * Report the frame as DMA done
	 */
	rt2x00lib_dmadone(entry);

412 413 414 415 416
	/*
	 * Check if the received data is simply too small
	 * to be actually valid, or if the urb is signaling
	 * a problem.
	 */
417
	if (urb->actual_length < entry->queue->desc_size || urb->status)
418
		set_bit(ENTRY_DATA_IO_FAILED, &entry->flags);
419

420
	/*
421 422
	 * Schedule the delayed work for reading the RX status
	 * from the device.
423
	 */
I
Ivo van Doorn 已提交
424
	ieee80211_queue_work(rt2x00dev->hw, &rt2x00dev->rxdone_work);
425 426 427 428 429 430 431
}

/*
 * Radio handlers
 */
void rt2x00usb_disable_radio(struct rt2x00_dev *rt2x00dev)
{
I
Ivo van Doorn 已提交
432
	rt2x00usb_vendor_request_sw(rt2x00dev, USB_RX_CONTROL, 0, 0,
433 434
				    REGISTER_TIMEOUT);

435
	rt2x00dev->ops->lib->stop_queue(rt2x00dev->rx);
436 437 438 439 440 441
}
EXPORT_SYMBOL_GPL(rt2x00usb_disable_radio);

/*
 * Device initialization handlers.
 */
442
void rt2x00usb_clear_entry(struct queue_entry *entry)
443
{
444 445
	struct usb_device *usb_dev =
	    to_usb_device_intf(entry->queue->rt2x00dev->dev);
446
	struct queue_entry_priv_usb *entry_priv = entry->priv_data;
447
	int pipe;
448
	int status;
449

450 451
	entry->flags = 0;

452
	if (entry->queue->qid == QID_RX) {
453 454
		pipe = usb_rcvbulkpipe(usb_dev, entry->queue->usb_endpoint);
		usb_fill_bulk_urb(entry_priv->urb, usb_dev, pipe,
455 456
				entry->skb->data, entry->skb->len,
				rt2x00usb_interrupt_rxdone, entry);
457

458
		set_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags);
459 460 461 462 463 464

		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 已提交
465 466 467
			set_bit(ENTRY_DATA_IO_FAILED, &entry->flags);
			rt2x00lib_dmadone(entry);
		}
468
	}
469
}
470
EXPORT_SYMBOL_GPL(rt2x00usb_clear_entry);
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 506 507 508 509 510 511
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 已提交
512 513
		} else if (usb_endpoint_is_bulk_out(ep_desc) &&
			   (queue != queue_end(rt2x00dev))) {
514
			rt2x00usb_assign_endpoint(queue, ep_desc);
I
Ivo van Doorn 已提交
515
			queue = queue_next(queue);
516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541

			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;
}

542
static int rt2x00usb_alloc_entries(struct data_queue *queue)
543
{
544
	struct rt2x00_dev *rt2x00dev = queue->rt2x00dev;
545 546
	struct queue_entry_priv_usb *entry_priv;
	struct queue_entry_priv_usb_bcn *bcn_priv;
547 548
	unsigned int i;

549 550 551 552 553 554 555
	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;
	}

556
	/*
557 558 559
	 * If this is not the beacon queue or
	 * no guardian byte was required for the beacon,
	 * then we are done.
560
	 */
561
	if (queue->qid != QID_BEACON ||
562 563 564
	    !test_bit(DRIVER_REQUIRE_BEACON_GUARD, &rt2x00dev->flags))
		return 0;

I
Ivo van Doorn 已提交
565
	for (i = 0; i < queue->limit; i++) {
566 567 568
		bcn_priv = queue->entries[i].priv_data;
		bcn_priv->guardian_urb = usb_alloc_urb(0, GFP_KERNEL);
		if (!bcn_priv->guardian_urb)
569 570 571 572 573 574
			return -ENOMEM;
	}

	return 0;
}

575
static void rt2x00usb_free_entries(struct data_queue *queue)
576
{
577
	struct rt2x00_dev *rt2x00dev = queue->rt2x00dev;
578 579
	struct queue_entry_priv_usb *entry_priv;
	struct queue_entry_priv_usb_bcn *bcn_priv;
580 581
	unsigned int i;

I
Ivo van Doorn 已提交
582
	if (!queue->entries)
583 584
		return;

I
Ivo van Doorn 已提交
585
	for (i = 0; i < queue->limit; i++) {
586 587 588
		entry_priv = queue->entries[i].priv_data;
		usb_kill_urb(entry_priv->urb);
		usb_free_urb(entry_priv->urb);
589
	}
590 591 592 593 594 595

	/*
	 * If this is not the beacon queue or
	 * no guardian byte was required for the beacon,
	 * then we are done.
	 */
596
	if (queue->qid != QID_BEACON ||
597 598 599 600 601 602 603 604
	    !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);
	}
605 606 607 608
}

int rt2x00usb_initialize(struct rt2x00_dev *rt2x00dev)
{
I
Ivo van Doorn 已提交
609
	struct data_queue *queue;
610
	int status;
611

612 613 614 615 616 617 618
	/*
	 * Find endpoints for each queue
	 */
	status = rt2x00usb_find_endpoints(rt2x00dev);
	if (status)
		goto exit;

619 620 621
	/*
	 * Allocate DMA
	 */
I
Ivo van Doorn 已提交
622
	queue_for_each(rt2x00dev, queue) {
623
		status = rt2x00usb_alloc_entries(queue);
624 625 626 627 628 629 630 631 632 633 634 635 636 637 638
		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 已提交
639
	struct data_queue *queue;
640

I
Ivo van Doorn 已提交
641
	queue_for_each(rt2x00dev, queue)
642
		rt2x00usb_free_entries(queue);
643 644 645 646 647 648 649 650 651 652 653 654 655 656
}
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;

657 658
	kfree(rt2x00dev->csr.cache);
	rt2x00dev->csr.cache = NULL;
659 660 661 662
}

static int rt2x00usb_alloc_reg(struct rt2x00_dev *rt2x00dev)
{
663 664
	rt2x00dev->csr.cache = kzalloc(CSR_CACHE_SIZE, GFP_KERNEL);
	if (!rt2x00dev->csr.cache)
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 700 701 702 703 704 705
		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;
706
	rt2x00dev->dev = &usb_intf->dev;
707 708 709
	rt2x00dev->ops = ops;
	rt2x00dev->hw = hw;

710 711
	rt2x00_set_chip_intf(rt2x00dev, RT2X00_CHIP_INTF_USB);

712 713 714
	INIT_WORK(&rt2x00dev->rxdone_work, rt2x00usb_work_rxdone);
	INIT_WORK(&rt2x00dev->txdone_work, rt2x00usb_work_txdone);

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 781 782 783 784 785 786
	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));

787
	return rt2x00lib_resume(rt2x00dev);
788 789 790 791 792
}
EXPORT_SYMBOL_GPL(rt2x00usb_resume);
#endif /* CONFIG_PM */

/*
I
Ivo van Doorn 已提交
793
 * rt2x00usb module information.
794 795 796
 */
MODULE_AUTHOR(DRV_PROJECT);
MODULE_VERSION(DRV_VERSION);
I
Ivo van Doorn 已提交
797
MODULE_DESCRIPTION("rt2x00 usb library");
798
MODULE_LICENSE("GPL");