rt2x00usb.c 18.3 KB
Newer Older
1
/*
2
	Copyright (C) 2004 - 2009 Ivo van Doorn <IvDoorn@gmail.com>
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
	<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>
28
#include <linux/slab.h>
29
#include <linux/usb.h>
30
#include <linux/bug.h>
31 32 33 34 35 36 37

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

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

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

54 55 56 57 58 59 60 61
	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;

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

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

80 81 82 83
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)
84 85 86
{
	int status;

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

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

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

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

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

	return status;
}
109 110 111 112 113 114 115 116 117
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)
{
	int status;

118
	mutex_lock(&rt2x00dev->csr_mutex);
119 120 121 122 123

	status = rt2x00usb_vendor_req_buff_lock(rt2x00dev, request,
						requesttype, offset, buffer,
						buffer_length, timeout);

124
	mutex_unlock(&rt2x00dev->csr_mutex);
125 126 127

	return status;
}
128 129
EXPORT_SYMBOL_GPL(rt2x00usb_vendor_request_buff);

130 131
int rt2x00usb_vendor_request_large_buff(struct rt2x00_dev *rt2x00dev,
					const u8 request, const u8 requesttype,
I
Ivo van Doorn 已提交
132
					const u16 offset, const void *buffer,
133 134 135 136 137 138 139
					const u16 buffer_length,
					const int timeout)
{
	int status = 0;
	unsigned char *tb;
	u16 off, len, bsize;

140
	mutex_lock(&rt2x00dev->csr_mutex);
141

I
Ivo van Doorn 已提交
142
	tb  = (char *)buffer;
143 144 145 146 147 148 149 150 151 152 153 154 155
	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;
	}

156
	mutex_unlock(&rt2x00dev->csr_mutex);
157 158 159 160 161

	return status;
}
EXPORT_SYMBOL_GPL(rt2x00usb_vendor_request_large_buff);

162 163
int rt2x00usb_regbusy_read(struct rt2x00_dev *rt2x00dev,
			   const unsigned int offset,
164
			   const struct rt2x00_field32 field,
165 166 167 168
			   u32 *reg)
{
	unsigned int i;

169 170 171
	if (!test_bit(DEVICE_STATE_PRESENT, &rt2x00dev->flags))
		return -ENODEV;

172 173 174 175 176 177 178 179 180 181 182 183 184 185 186
	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);

187 188 189 190 191
/*
 * TX data handlers.
 */
static void rt2x00usb_interrupt_txdone(struct urb *urb)
{
I
Ivo van Doorn 已提交
192 193 194
	struct queue_entry *entry = (struct queue_entry *)urb->context;
	struct rt2x00_dev *rt2x00dev = entry->queue->rt2x00dev;
	struct txdone_entry_desc txdesc;
195

196
	if (!test_bit(DEVICE_STATE_ENABLED_RADIO, &rt2x00dev->flags) ||
I
Ivo van Doorn 已提交
197
	    !test_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags))
198 199
		return;

200 201 202 203 204
	/*
	 * Remove the descriptor from the front of the skb.
	 */
	skb_pull(entry->skb, entry->queue->desc_size);

205 206
	/*
	 * Obtain the status about this packet.
I
Ivo van Doorn 已提交
207 208 209 210 211 212
	 * Note that when the status is 0 it does not mean the
	 * 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).
213
	 */
214
	txdesc.flags = 0;
I
Ivo van Doorn 已提交
215 216 217 218
	if (!urb->status)
		__set_bit(TXDONE_UNKNOWN, &txdesc.flags);
	else
		__set_bit(TXDONE_FAILURE, &txdesc.flags);
I
Ivo van Doorn 已提交
219
	txdesc.retry = 0;
220

I
Ivo van Doorn 已提交
221
	rt2x00lib_txdone(entry, &txdesc);
222 223
}

224 225
int rt2x00usb_write_tx_data(struct queue_entry *entry,
			    struct txentry_desc *txdesc)
226
{
227
	struct rt2x00_dev *rt2x00dev = entry->queue->rt2x00dev;
228
	struct usb_device *usb_dev = to_usb_device_intf(rt2x00dev->dev);
229
	struct queue_entry_priv_usb *entry_priv = entry->priv_data;
230
	u32 length;
231 232 233 234

	/*
	 * Add the descriptor in front of the skb.
	 */
235 236
	skb_push(entry->skb, entry->queue->desc_size);
	memset(entry->skb->data, 0, entry->queue->desc_size);
237 238

	/*
239 240 241
	 * 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.
242
	 */
243
	length = rt2x00dev->ops->lib->get_tx_data_len(entry);
244

245
	usb_fill_bulk_urb(entry_priv->urb, usb_dev,
246
			  usb_sndbulkpipe(usb_dev, entry->queue->usb_endpoint),
247 248
			  entry->skb->data, length,
			  rt2x00usb_interrupt_txdone, entry);
249

250 251 252 253 254 255
	/*
	 * Call the driver's write_tx_datadesc function, if it exists.
	 */
	if (rt2x00dev->ops->lib->write_tx_datadesc)
		rt2x00dev->ops->lib->write_tx_datadesc(entry, txdesc);

256 257 258 259
	return 0;
}
EXPORT_SYMBOL_GPL(rt2x00usb_write_tx_data);

260 261 262 263
static inline void rt2x00usb_kick_tx_entry(struct queue_entry *entry)
{
	struct queue_entry_priv_usb *entry_priv = entry->priv_data;

264
	if (test_and_clear_bit(ENTRY_DATA_PENDING, &entry->flags))
265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304
		usb_submit_urb(entry_priv->urb, GFP_ATOMIC);
}

void rt2x00usb_kick_tx_queue(struct rt2x00_dev *rt2x00dev,
			     const enum data_queue_qid qid)
{
	struct data_queue *queue = rt2x00queue_get_queue(rt2x00dev, qid);
	unsigned long irqflags;
	unsigned int index;
	unsigned int index_done;
	unsigned int i;

	/*
	 * Only protect the range we are going to loop over,
	 * if during our loop a extra entry is set to pending
	 * it should not be kicked during this run, since it
	 * is part of another TX operation.
	 */
	spin_lock_irqsave(&queue->lock, irqflags);
	index = queue->index[Q_INDEX];
	index_done = queue->index[Q_INDEX_DONE];
	spin_unlock_irqrestore(&queue->lock, irqflags);

	/*
	 * Start from the TX done pointer, this guarentees that we will
	 * send out all frames in the correct order.
	 */
	if (index_done < index) {
		for (i = index_done; i < index; i++)
			rt2x00usb_kick_tx_entry(&queue->entries[i]);
	} else {
		for (i = index_done; i < queue->limit; i++)
			rt2x00usb_kick_tx_entry(&queue->entries[i]);

		for (i = 0; i < index; i++)
			rt2x00usb_kick_tx_entry(&queue->entries[i]);
	}
}
EXPORT_SYMBOL_GPL(rt2x00usb_kick_tx_queue);

305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339
void rt2x00usb_kill_tx_queue(struct rt2x00_dev *rt2x00dev,
			     const enum data_queue_qid qid)
{
	struct data_queue *queue = rt2x00queue_get_queue(rt2x00dev, qid);
	struct queue_entry_priv_usb *entry_priv;
	struct queue_entry_priv_usb_bcn *bcn_priv;
	unsigned int i;
	bool kill_guard;

	/*
	 * When killing the beacon queue, we must also kill
	 * the beacon guard byte.
	 */
	kill_guard =
	    (qid == QID_BEACON) &&
	    (test_bit(DRIVER_REQUIRE_BEACON_GUARD, &rt2x00dev->flags));

	/*
	 * Cancel all entries.
	 */
	for (i = 0; i < queue->limit; i++) {
		entry_priv = queue->entries[i].priv_data;
		usb_kill_urb(entry_priv->urb);

		/*
		 * Kill guardian urb (if required by driver).
		 */
		if (kill_guard) {
			bcn_priv = queue->entries[i].priv_data;
			usb_kill_urb(bcn_priv->guardian_urb);
		}
	}
}
EXPORT_SYMBOL_GPL(rt2x00usb_kill_tx_queue);

340 341 342 343 344
/*
 * RX data handlers.
 */
static void rt2x00usb_interrupt_rxdone(struct urb *urb)
{
I
Ivo van Doorn 已提交
345 346
	struct queue_entry *entry = (struct queue_entry *)urb->context;
	struct rt2x00_dev *rt2x00dev = entry->queue->rt2x00dev;
347
	struct skb_frame_desc *skbdesc = get_skb_frame_desc(entry->skb);
348
	u8 rxd[32];
349

350
	if (!test_bit(DEVICE_STATE_ENABLED_RADIO, &rt2x00dev->flags) ||
I
Ivo van Doorn 已提交
351
	    !test_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags))
352 353 354 355 356 357 358
		return;

	/*
	 * Check if the received data is simply too small
	 * to be actually valid, or if the urb is signaling
	 * a problem.
	 */
I
Ivo van Doorn 已提交
359
	if (urb->actual_length < entry->queue->desc_size || urb->status) {
360
		set_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags);
I
Ivo van Doorn 已提交
361 362 363
		usb_submit_urb(urb, GFP_ATOMIC);
		return;
	}
364

365
	/*
366
	 * Fill in desc fields of the skb descriptor
367
	 */
368 369
	skbdesc->desc = rxd;
	skbdesc->desc_len = entry->queue->desc_size;
370

371 372 373
	/*
	 * Send the frame to rt2x00lib for further processing.
	 */
374
	rt2x00lib_rxdone(rt2x00dev, entry);
375 376 377 378 379 380 381
}

/*
 * Radio handlers
 */
void rt2x00usb_disable_radio(struct rt2x00_dev *rt2x00dev)
{
I
Ivo van Doorn 已提交
382
	rt2x00usb_vendor_request_sw(rt2x00dev, USB_RX_CONTROL, 0, 0,
383 384 385
				    REGISTER_TIMEOUT);

	/*
386 387
	 * The USB version of kill_tx_queue also works
	 * on the RX queue.
388
	 */
389
	rt2x00dev->ops->lib->kill_tx_queue(rt2x00dev, QID_RX);
390 391 392 393 394 395
}
EXPORT_SYMBOL_GPL(rt2x00usb_disable_radio);

/*
 * Device initialization handlers.
 */
396
void rt2x00usb_clear_entry(struct queue_entry *entry)
397
{
398 399
	struct usb_device *usb_dev =
	    to_usb_device_intf(entry->queue->rt2x00dev->dev);
400
	struct queue_entry_priv_usb *entry_priv = entry->priv_data;
401
	int pipe;
402

403
	if (entry->queue->qid == QID_RX) {
404 405
		pipe = usb_rcvbulkpipe(usb_dev, entry->queue->usb_endpoint);
		usb_fill_bulk_urb(entry_priv->urb, usb_dev, pipe,
406 407
				entry->skb->data, entry->skb->len,
				rt2x00usb_interrupt_rxdone, entry);
408

409 410 411 412 413
		set_bit(ENTRY_OWNER_DEVICE_DATA, &entry->flags);
		usb_submit_urb(entry_priv->urb, GFP_ATOMIC);
	} else {
		entry->flags = 0;
	}
414
}
415
EXPORT_SYMBOL_GPL(rt2x00usb_clear_entry);
416

417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456
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 已提交
457 458
		} else if (usb_endpoint_is_bulk_out(ep_desc) &&
			   (queue != queue_end(rt2x00dev))) {
459
			rt2x00usb_assign_endpoint(queue, ep_desc);
I
Ivo van Doorn 已提交
460
			queue = queue_next(queue);
461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486

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

487
static int rt2x00usb_alloc_urb(struct rt2x00_dev *rt2x00dev,
I
Ivo van Doorn 已提交
488
			       struct data_queue *queue)
489
{
490 491
	struct queue_entry_priv_usb *entry_priv;
	struct queue_entry_priv_usb_bcn *bcn_priv;
492 493
	unsigned int i;

494 495 496 497 498 499 500
	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;
	}

501
	/*
502 503 504
	 * If this is not the beacon queue or
	 * no guardian byte was required for the beacon,
	 * then we are done.
505
	 */
506 507 508 509
	if (rt2x00dev->bcn != queue ||
	    !test_bit(DRIVER_REQUIRE_BEACON_GUARD, &rt2x00dev->flags))
		return 0;

I
Ivo van Doorn 已提交
510
	for (i = 0; i < queue->limit; i++) {
511 512 513
		bcn_priv = queue->entries[i].priv_data;
		bcn_priv->guardian_urb = usb_alloc_urb(0, GFP_KERNEL);
		if (!bcn_priv->guardian_urb)
514 515 516 517 518 519 520
			return -ENOMEM;
	}

	return 0;
}

static void rt2x00usb_free_urb(struct rt2x00_dev *rt2x00dev,
I
Ivo van Doorn 已提交
521
			       struct data_queue *queue)
522
{
523 524
	struct queue_entry_priv_usb *entry_priv;
	struct queue_entry_priv_usb_bcn *bcn_priv;
525 526
	unsigned int i;

I
Ivo van Doorn 已提交
527
	if (!queue->entries)
528 529
		return;

I
Ivo van Doorn 已提交
530
	for (i = 0; i < queue->limit; i++) {
531 532 533
		entry_priv = queue->entries[i].priv_data;
		usb_kill_urb(entry_priv->urb);
		usb_free_urb(entry_priv->urb);
534
	}
535 536 537 538 539 540 541 542 543 544 545 546 547 548 549

	/*
	 * If this is not the beacon queue or
	 * no guardian byte was required for the beacon,
	 * then we are done.
	 */
	if (rt2x00dev->bcn != queue ||
	    !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);
	}
550 551 552 553
}

int rt2x00usb_initialize(struct rt2x00_dev *rt2x00dev)
{
I
Ivo van Doorn 已提交
554
	struct data_queue *queue;
555
	int status;
556

557 558 559 560 561 562 563
	/*
	 * Find endpoints for each queue
	 */
	status = rt2x00usb_find_endpoints(rt2x00dev);
	if (status)
		goto exit;

564 565 566
	/*
	 * Allocate DMA
	 */
I
Ivo van Doorn 已提交
567 568
	queue_for_each(rt2x00dev, queue) {
		status = rt2x00usb_alloc_urb(rt2x00dev, queue);
569 570 571 572 573 574 575 576 577 578 579 580 581 582 583
		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 已提交
584
	struct data_queue *queue;
585

I
Ivo van Doorn 已提交
586 587
	queue_for_each(rt2x00dev, queue)
		rt2x00usb_free_urb(rt2x00dev, queue);
588 589 590 591 592 593 594 595 596 597 598 599 600 601
}
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;

602 603
	kfree(rt2x00dev->csr.cache);
	rt2x00dev->csr.cache = NULL;
604 605 606 607
}

static int rt2x00usb_alloc_reg(struct rt2x00_dev *rt2x00dev)
{
608 609
	rt2x00dev->csr.cache = kzalloc(CSR_CACHE_SIZE, GFP_KERNEL);
	if (!rt2x00dev->csr.cache)
610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650
		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;
651
	rt2x00dev->dev = &usb_intf->dev;
652 653 654
	rt2x00dev->ops = ops;
	rt2x00dev->hw = hw;

655 656
	rt2x00_set_chip_intf(rt2x00dev, RT2X00_CHIP_INTF_USB);

657 658 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 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728
	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));

729
	return rt2x00lib_resume(rt2x00dev);
730 731 732 733 734
}
EXPORT_SYMBOL_GPL(rt2x00usb_resume);
#endif /* CONFIG_PM */

/*
I
Ivo van Doorn 已提交
735
 * rt2x00usb module information.
736 737 738
 */
MODULE_AUTHOR(DRV_PROJECT);
MODULE_VERSION(DRV_VERSION);
I
Ivo van Doorn 已提交
739
MODULE_DESCRIPTION("rt2x00 usb library");
740
MODULE_LICENSE("GPL");