/*
   * at76c503/at76c505 USB driver
   *
   * Copyright (c) 2002 - 2003 Oliver Kurth
   * Copyright (c) 2004 Joerg Albert <joerg.albert@gmx.de>
   * Copyright (c) 2004 Nick Jones
   * Copyright (c) 2004 Balint Seeber <n0_5p4m_p13453@hotmail.com>
   * Copyright (c) 2007 Guido Guenther <agx@sigxcpu.org>
   * Copyright (c) 2007 Kalle Valo <kalle.valo@iki.fi>
   * Copyright (c) 2010 Sebastian Smolorz <sesmo@gmx.net>
   *
   * 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 file is part of the Berlios driver for WLAN USB devices based on the
   * Atmel AT76C503A/505/505A.
   *
   * Some iw_handler code was taken from airo.c, (C) 1999 Benjamin Reed
   *
   * TODO list is at the wiki:
   *
   * http://wireless.kernel.org/en/users/Drivers/at76c50x-usb#TODO
   *
   */
  
  #include <linux/init.h>
  #include <linux/kernel.h>
  #include <linux/sched.h>
  #include <linux/errno.h>
  #include <linux/slab.h>
  #include <linux/module.h>
  #include <linux/spinlock.h>
  #include <linux/list.h>
  #include <linux/usb.h>
  #include <linux/netdevice.h>
  #include <linux/if_arp.h>
  #include <linux/etherdevice.h>
  #include <linux/ethtool.h>
  #include <linux/wireless.h>
  #include <net/iw_handler.h>
  #include <net/ieee80211_radiotap.h>
  #include <linux/firmware.h>
  #include <linux/leds.h>
  #include <net/mac80211.h>
  
  #include "at76c50x-usb.h"
  
  /* Version information */
  #define DRIVER_NAME "at76c50x-usb"
  #define DRIVER_VERSION	"0.17"
  #define DRIVER_DESC "Atmel at76x USB Wireless LAN Driver"
  
  /* at76_debug bits */
  #define DBG_PROGRESS		0x00000001	/* authentication/accociation */
  #define DBG_BSS_TABLE		0x00000002	/* show BSS table after scans */
  #define DBG_IOCTL		0x00000004	/* ioctl calls / settings */
  #define DBG_MAC_STATE		0x00000008	/* MAC state transitions */
  #define DBG_TX_DATA		0x00000010	/* tx header */
  #define DBG_TX_DATA_CONTENT	0x00000020	/* tx content */
  #define DBG_TX_MGMT		0x00000040	/* tx management */
  #define DBG_RX_DATA		0x00000080	/* rx data header */
  #define DBG_RX_DATA_CONTENT	0x00000100	/* rx data content */
  #define DBG_RX_MGMT		0x00000200	/* rx mgmt frame headers */
  #define DBG_RX_BEACON		0x00000400	/* rx beacon */
  #define DBG_RX_CTRL		0x00000800	/* rx control */
  #define DBG_RX_MGMT_CONTENT	0x00001000	/* rx mgmt content */
  #define DBG_RX_FRAGS		0x00002000	/* rx data fragment handling */
  #define DBG_DEVSTART		0x00004000	/* fw download, device start */
  #define DBG_URB			0x00008000	/* rx urb status, ... */
  #define DBG_RX_ATMEL_HDR	0x00010000	/* Atmel-specific Rx headers */
  #define DBG_PROC_ENTRY		0x00020000	/* procedure entries/exits */
  #define DBG_PM			0x00040000	/* power management settings */
  #define DBG_BSS_MATCH		0x00080000	/* BSS match failures */
  #define DBG_PARAMS		0x00100000	/* show configured parameters */
  #define DBG_WAIT_COMPLETE	0x00200000	/* command completion */
  #define DBG_RX_FRAGS_SKB	0x00400000	/* skb header of Rx fragments */
  #define DBG_BSS_TABLE_RM	0x00800000	/* purging bss table entries */
  #define DBG_MONITOR_MODE	0x01000000	/* monitor mode */
  #define DBG_MIB			0x02000000	/* dump all MIBs on startup */
  #define DBG_MGMT_TIMER		0x04000000	/* dump mgmt_timer ops */
  #define DBG_WE_EVENTS		0x08000000	/* dump wireless events */
  #define DBG_FW			0x10000000	/* firmware download */
  #define DBG_DFU			0x20000000	/* device firmware upgrade */
  #define DBG_CMD			0x40000000
  #define DBG_MAC80211		0x80000000
  
  #define DBG_DEFAULTS		0
  
  /* Use our own dbg macro */
  #define at76_dbg(bits, format, arg...)					\
  do {									\
  	if (at76_debug & (bits))					\
  		printk(KERN_DEBUG DRIVER_NAME ": " format "
  ", ##arg);	\
  } while (0)
  
  #define at76_dbg_dump(bits, buf, len, format, arg...)			\
  do {									\
  	if (at76_debug & (bits)) {					\
  		printk(KERN_DEBUG DRIVER_NAME ": " format "
  ", ##arg);	\
  		print_hex_dump_bytes("", DUMP_PREFIX_OFFSET, buf, len);	\
  	}								\
  } while (0)
  
  static uint at76_debug = DBG_DEFAULTS;
  
  /* Protect against concurrent firmware loading and parsing */
  static struct mutex fw_mutex;
  
  static struct fwentry firmwares[] = {
  	[0] = { "" },
  	[BOARD_503_ISL3861] = { "atmel_at76c503-i3861.bin" },
  	[BOARD_503_ISL3863] = { "atmel_at76c503-i3863.bin" },
  	[BOARD_503] = { "atmel_at76c503-rfmd.bin" },
  	[BOARD_503_ACC] = { "atmel_at76c503-rfmd-acc.bin" },
  	[BOARD_505] = { "atmel_at76c505-rfmd.bin" },
  	[BOARD_505_2958] = { "atmel_at76c505-rfmd2958.bin" },
  	[BOARD_505A] = { "atmel_at76c505a-rfmd2958.bin" },
  	[BOARD_505AMX] = { "atmel_at76c505amx-rfmd.bin" },
  };
  MODULE_FIRMWARE("atmel_at76c503-i3861.bin");
  MODULE_FIRMWARE("atmel_at76c503-i3863.bin");
  MODULE_FIRMWARE("atmel_at76c503-rfmd.bin");
  MODULE_FIRMWARE("atmel_at76c503-rfmd-acc.bin");
  MODULE_FIRMWARE("atmel_at76c505-rfmd.bin");
  MODULE_FIRMWARE("atmel_at76c505-rfmd2958.bin");
  MODULE_FIRMWARE("atmel_at76c505a-rfmd2958.bin");
  MODULE_FIRMWARE("atmel_at76c505amx-rfmd.bin");
  
  #define USB_DEVICE_DATA(__ops)	.driver_info = (kernel_ulong_t)(__ops)
  
  static struct usb_device_id dev_table[] = {
  	/*
  	 * at76c503-i3861
  	 */
  	/* Generic AT76C503/3861 device */
  	{ USB_DEVICE(0x03eb, 0x7603), USB_DEVICE_DATA(BOARD_503_ISL3861) },
  	/* Linksys WUSB11 v2.1/v2.6 */
  	{ USB_DEVICE(0x066b, 0x2211), USB_DEVICE_DATA(BOARD_503_ISL3861) },
  	/* Netgear MA101 rev. A */
  	{ USB_DEVICE(0x0864, 0x4100), USB_DEVICE_DATA(BOARD_503_ISL3861) },
  	/* Tekram U300C / Allnet ALL0193 */
  	{ USB_DEVICE(0x0b3b, 0x1612), USB_DEVICE_DATA(BOARD_503_ISL3861) },
  	/* HP HN210W J7801A */
  	{ USB_DEVICE(0x03f0, 0x011c), USB_DEVICE_DATA(BOARD_503_ISL3861) },
  	/* Sitecom/Z-Com/Zyxel M4Y-750 */
  	{ USB_DEVICE(0x0cde, 0x0001), USB_DEVICE_DATA(BOARD_503_ISL3861) },
  	/* Dynalink/Askey WLL013 (intersil) */
  	{ USB_DEVICE(0x069a, 0x0320), USB_DEVICE_DATA(BOARD_503_ISL3861) },
  	/* EZ connect 11Mpbs Wireless USB Adapter SMC2662W v1 */
  	{ USB_DEVICE(0x0d5c, 0xa001), USB_DEVICE_DATA(BOARD_503_ISL3861) },
  	/* BenQ AWL300 */
  	{ USB_DEVICE(0x04a5, 0x9000), USB_DEVICE_DATA(BOARD_503_ISL3861) },
  	/* Addtron AWU-120, Compex WLU11 */
  	{ USB_DEVICE(0x05dd, 0xff31), USB_DEVICE_DATA(BOARD_503_ISL3861) },
  	/* Intel AP310 AnyPoint II USB */
  	{ USB_DEVICE(0x8086, 0x0200), USB_DEVICE_DATA(BOARD_503_ISL3861) },
  	/* Dynalink L11U */
  	{ USB_DEVICE(0x0d8e, 0x7100), USB_DEVICE_DATA(BOARD_503_ISL3861) },
  	/* Arescom WL-210, FCC id 07J-GL2411USB */
  	{ USB_DEVICE(0x0d8e, 0x7110), USB_DEVICE_DATA(BOARD_503_ISL3861) },
  	/* I-O DATA WN-B11/USB */
  	{ USB_DEVICE(0x04bb, 0x0919), USB_DEVICE_DATA(BOARD_503_ISL3861) },
  	/* BT Voyager 1010 */
  	{ USB_DEVICE(0x069a, 0x0821), USB_DEVICE_DATA(BOARD_503_ISL3861) },
  	/*
  	 * at76c503-i3863
  	 */
  	/* Generic AT76C503/3863 device */
  	{ USB_DEVICE(0x03eb, 0x7604), USB_DEVICE_DATA(BOARD_503_ISL3863) },
  	/* Samsung SWL-2100U */
  	{ USB_DEVICE(0x055d, 0xa000), USB_DEVICE_DATA(BOARD_503_ISL3863) },
  	/*
  	 * at76c503-rfmd
  	 */
  	/* Generic AT76C503/RFMD device */
  	{ USB_DEVICE(0x03eb, 0x7605), USB_DEVICE_DATA(BOARD_503) },
  	/* Dynalink/Askey WLL013 (rfmd) */
  	{ USB_DEVICE(0x069a, 0x0321), USB_DEVICE_DATA(BOARD_503) },
  	/* Linksys WUSB11 v2.6 */
  	{ USB_DEVICE(0x077b, 0x2219), USB_DEVICE_DATA(BOARD_503) },
  	/* Network Everywhere NWU11B */
  	{ USB_DEVICE(0x077b, 0x2227), USB_DEVICE_DATA(BOARD_503) },
  	/* Netgear MA101 rev. B */
  	{ USB_DEVICE(0x0864, 0x4102), USB_DEVICE_DATA(BOARD_503) },
  	/* D-Link DWL-120 rev. E */
  	{ USB_DEVICE(0x2001, 0x3200), USB_DEVICE_DATA(BOARD_503) },
  	/* Actiontec 802UAT1, HWU01150-01UK */
  	{ USB_DEVICE(0x1668, 0x7605), USB_DEVICE_DATA(BOARD_503) },
  	/* AirVast W-Buddie WN210 */
  	{ USB_DEVICE(0x03eb, 0x4102), USB_DEVICE_DATA(BOARD_503) },
  	/* Dick Smith Electronics XH1153 802.11b USB adapter */
  	{ USB_DEVICE(0x1371, 0x5743), USB_DEVICE_DATA(BOARD_503) },
  	/* CNet CNUSB611 */
  	{ USB_DEVICE(0x1371, 0x0001), USB_DEVICE_DATA(BOARD_503) },
  	/* FiberLine FL-WL200U */
  	{ USB_DEVICE(0x1371, 0x0002), USB_DEVICE_DATA(BOARD_503) },
  	/* BenQ AWL400 USB stick */
  	{ USB_DEVICE(0x04a5, 0x9001), USB_DEVICE_DATA(BOARD_503) },
  	/* 3Com 3CRSHEW696 */
  	{ USB_DEVICE(0x0506, 0x0a01), USB_DEVICE_DATA(BOARD_503) },
  	/* Siemens Santis ADSL WLAN USB adapter WLL 013 */
  	{ USB_DEVICE(0x0681, 0x001b), USB_DEVICE_DATA(BOARD_503) },
  	/* Belkin F5D6050, version 2 */
  	{ USB_DEVICE(0x050d, 0x0050), USB_DEVICE_DATA(BOARD_503) },
  	/* iBlitzz, BWU613 (not *B or *SB) */
  	{ USB_DEVICE(0x07b8, 0xb000), USB_DEVICE_DATA(BOARD_503) },
  	/* Gigabyte GN-WLBM101 */
  	{ USB_DEVICE(0x1044, 0x8003), USB_DEVICE_DATA(BOARD_503) },
  	/* Planex GW-US11S */
  	{ USB_DEVICE(0x2019, 0x3220), USB_DEVICE_DATA(BOARD_503) },
  	/* Internal WLAN adapter in h5[4,5]xx series iPAQs */
  	{ USB_DEVICE(0x049f, 0x0032), USB_DEVICE_DATA(BOARD_503) },
  	/* Corega Wireless LAN USB-11 mini */
  	{ USB_DEVICE(0x07aa, 0x0011), USB_DEVICE_DATA(BOARD_503) },
  	/* Corega Wireless LAN USB-11 mini2 */
  	{ USB_DEVICE(0x07aa, 0x0018), USB_DEVICE_DATA(BOARD_503) },
  	/* Uniden PCW100 */
  	{ USB_DEVICE(0x05dd, 0xff35), USB_DEVICE_DATA(BOARD_503) },
  	/*
  	 * at76c503-rfmd-acc
  	 */
  	/* SMC2664W */
  	{ USB_DEVICE(0x083a, 0x3501), USB_DEVICE_DATA(BOARD_503_ACC) },
  	/* Belkin F5D6050, SMC2662W v2, SMC2662W-AR */
  	{ USB_DEVICE(0x0d5c, 0xa002), USB_DEVICE_DATA(BOARD_503_ACC) },
  	/*
  	 * at76c505-rfmd
  	 */
  	/* Generic AT76C505/RFMD */
  	{ USB_DEVICE(0x03eb, 0x7606), USB_DEVICE_DATA(BOARD_505) },
  	/*
  	 * at76c505-rfmd2958
  	 */
  	/* Generic AT76C505/RFMD, OvisLink WL-1130USB */
  	{ USB_DEVICE(0x03eb, 0x7613), USB_DEVICE_DATA(BOARD_505_2958) },
  	/* Fiberline FL-WL240U */
  	{ USB_DEVICE(0x1371, 0x0014), USB_DEVICE_DATA(BOARD_505_2958) },
  	/* CNet CNUSB-611G */
  	{ USB_DEVICE(0x1371, 0x0013), USB_DEVICE_DATA(BOARD_505_2958) },
  	/* Linksys WUSB11 v2.8 */
  	{ USB_DEVICE(0x1915, 0x2233), USB_DEVICE_DATA(BOARD_505_2958) },
  	/* Xterasys XN-2122B, IBlitzz BWU613B/BWU613SB */
  	{ USB_DEVICE(0x12fd, 0x1001), USB_DEVICE_DATA(BOARD_505_2958) },
  	/* Corega WLAN USB Stick 11 */
  	{ USB_DEVICE(0x07aa, 0x7613), USB_DEVICE_DATA(BOARD_505_2958) },
  	/* Microstar MSI Box MS6978 */
  	{ USB_DEVICE(0x0db0, 0x1020), USB_DEVICE_DATA(BOARD_505_2958) },
  	/*
  	 * at76c505a-rfmd2958
  	 */
  	/* Generic AT76C505A device */
  	{ USB_DEVICE(0x03eb, 0x7614), USB_DEVICE_DATA(BOARD_505A) },
  	/* Generic AT76C505AS device */
  	{ USB_DEVICE(0x03eb, 0x7617), USB_DEVICE_DATA(BOARD_505A) },
  	/* Siemens Gigaset USB WLAN Adapter 11 */
  	{ USB_DEVICE(0x1690, 0x0701), USB_DEVICE_DATA(BOARD_505A) },
  	/* OQO Model 01+ Internal Wi-Fi */
  	{ USB_DEVICE(0x1557, 0x0002), USB_DEVICE_DATA(BOARD_505A) },
  	/*
  	 * at76c505amx-rfmd
  	 */
  	/* Generic AT76C505AMX device */
  	{ USB_DEVICE(0x03eb, 0x7615), USB_DEVICE_DATA(BOARD_505AMX) },
  	{ }
  };
  
  MODULE_DEVICE_TABLE(usb, dev_table);
  
  /* Supported rates of this hardware, bit 7 marks basic rates */
  static const u8 hw_rates[] = { 0x82, 0x84, 0x0b, 0x16 };
  
  static const char *const preambles[] = { "long", "short", "auto" };
  
  /* Firmware download */
  /* DFU states */
  #define STATE_IDLE			0x00
  #define STATE_DETACH			0x01
  #define STATE_DFU_IDLE			0x02
  #define STATE_DFU_DOWNLOAD_SYNC		0x03
  #define STATE_DFU_DOWNLOAD_BUSY		0x04
  #define STATE_DFU_DOWNLOAD_IDLE		0x05
  #define STATE_DFU_MANIFEST_SYNC		0x06
  #define STATE_DFU_MANIFEST		0x07
  #define STATE_DFU_MANIFEST_WAIT_RESET	0x08
  #define STATE_DFU_UPLOAD_IDLE		0x09
  #define STATE_DFU_ERROR			0x0a
  
  /* DFU commands */
  #define DFU_DETACH			0
  #define DFU_DNLOAD			1
  #define DFU_UPLOAD			2
  #define DFU_GETSTATUS			3
  #define DFU_CLRSTATUS			4
  #define DFU_GETSTATE			5
  #define DFU_ABORT			6
  
  #define FW_BLOCK_SIZE 1024
  
  struct dfu_status {
  	unsigned char status;
  	unsigned char poll_timeout[3];
  	unsigned char state;
  	unsigned char string;
  } __packed;
  
  static inline int at76_is_intersil(enum board_type board)
  {
  	return (board == BOARD_503_ISL3861 || board == BOARD_503_ISL3863);
  }
  
  static inline int at76_is_503rfmd(enum board_type board)
  {
  	return (board == BOARD_503 || board == BOARD_503_ACC);
  }
  
  static inline int at76_is_505a(enum board_type board)
  {
  	return (board == BOARD_505A || board == BOARD_505AMX);
  }
  
  /* Load a block of the first (internal) part of the firmware */
  static int at76_load_int_fw_block(struct usb_device *udev, int blockno,
  				  void *block, int size)
  {
  	return usb_control_msg(udev, usb_sndctrlpipe(udev, 0), DFU_DNLOAD,
  			       USB_TYPE_CLASS | USB_DIR_OUT |
  			       USB_RECIP_INTERFACE, blockno, 0, block, size,
  			       USB_CTRL_GET_TIMEOUT);
  }
  
  static int at76_dfu_get_status(struct usb_device *udev,
  			       struct dfu_status *status)
  {
  	int ret;
  
  	ret = usb_control_msg(udev, usb_rcvctrlpipe(udev, 0), DFU_GETSTATUS,
  			      USB_TYPE_CLASS | USB_DIR_IN | USB_RECIP_INTERFACE,
  			      0, 0, status, sizeof(struct dfu_status),
  			      USB_CTRL_GET_TIMEOUT);
  	return ret;
  }
  
  static int at76_dfu_get_state(struct usb_device *udev, u8 *state)
  {
  	int ret;
  
  	ret = usb_control_msg(udev, usb_rcvctrlpipe(udev, 0), DFU_GETSTATE,
  			      USB_TYPE_CLASS | USB_DIR_IN | USB_RECIP_INTERFACE,
  			      0, 0, state, 1, USB_CTRL_GET_TIMEOUT);
  	return ret;
  }
  
  /* Convert timeout from the DFU status to jiffies */
  static inline unsigned long at76_get_timeout(struct dfu_status *s)
  {
  	return msecs_to_jiffies((s->poll_timeout[2] << 16)
  				| (s->poll_timeout[1] << 8)
  				| (s->poll_timeout[0]));
  }
  
  /* Load internal firmware from the buffer.  If manifest_sync_timeout > 0, use
   * its value in jiffies in the MANIFEST_SYNC state.  */
  static int at76_usbdfu_download(struct usb_device *udev, u8 *buf, u32 size,
  				int manifest_sync_timeout)
  {
  	int ret = 0;
  	int need_dfu_state = 1;
  	int is_done = 0;
  	u32 dfu_timeout = 0;
  	int bsize = 0;
  	int blockno = 0;
  	struct dfu_status *dfu_stat_buf = NULL;
  	u8 *dfu_state = NULL;
  	u8 *block = NULL;
  
  	at76_dbg(DBG_DFU, "%s( %p, %u, %d)", __func__, buf, size,
  		 manifest_sync_timeout);
  
  	if (!size) {
  		dev_err(&udev->dev, "FW buffer length invalid!
  ");
  		return -EINVAL;
  	}
  
  	dfu_stat_buf = kmalloc(sizeof(struct dfu_status), GFP_KERNEL);
  	if (!dfu_stat_buf) {
  		ret = -ENOMEM;
  		goto exit;
  	}
  
  	block = kmalloc(FW_BLOCK_SIZE, GFP_KERNEL);
  	if (!block) {
  		ret = -ENOMEM;
  		goto exit;
  	}
  
  	dfu_state = kmalloc(sizeof(u8), GFP_KERNEL);
  	if (!dfu_state) {
  		ret = -ENOMEM;
  		goto exit;
  	}
  	*dfu_state = 0;
  
  	do {
  		if (need_dfu_state) {
  			ret = at76_dfu_get_state(udev, dfu_state);
  			if (ret < 0) {
  				dev_err(&udev->dev,
  					"cannot get DFU state: %d
  ", ret);
  				goto exit;
  			}
  			need_dfu_state = 0;
  		}
  
  		switch (*dfu_state) {
  		case STATE_DFU_DOWNLOAD_SYNC:
  			at76_dbg(DBG_DFU, "STATE_DFU_DOWNLOAD_SYNC");
  			ret = at76_dfu_get_status(udev, dfu_stat_buf);
  			if (ret >= 0) {
  				*dfu_state = dfu_stat_buf->state;
  				dfu_timeout = at76_get_timeout(dfu_stat_buf);
  				need_dfu_state = 0;
  			} else
  				dev_err(&udev->dev,
  					"at76_dfu_get_status returned %d
  ",
  					ret);
  			break;
  
  		case STATE_DFU_DOWNLOAD_BUSY:
  			at76_dbg(DBG_DFU, "STATE_DFU_DOWNLOAD_BUSY");
  			need_dfu_state = 1;
  
  			at76_dbg(DBG_DFU, "DFU: Resetting device");
  			schedule_timeout_interruptible(dfu_timeout);
  			break;
  
  		case STATE_DFU_DOWNLOAD_IDLE:
  			at76_dbg(DBG_DFU, "DOWNLOAD...");
  			/* fall through */
  		case STATE_DFU_IDLE:
  			at76_dbg(DBG_DFU, "DFU IDLE");
  
  			bsize = min_t(int, size, FW_BLOCK_SIZE);
  			memcpy(block, buf, bsize);
  			at76_dbg(DBG_DFU, "int fw, size left = %5d, "
  				 "bsize = %4d, blockno = %2d", size, bsize,
  				 blockno);
  			ret =
  			    at76_load_int_fw_block(udev, blockno, block, bsize);
  			buf += bsize;
  			size -= bsize;
  			blockno++;
  
  			if (ret != bsize)
  				dev_err(&udev->dev,
  					"at76_load_int_fw_block returned %d
  ",
  					ret);
  			need_dfu_state = 1;
  			break;
  
  		case STATE_DFU_MANIFEST_SYNC:
  			at76_dbg(DBG_DFU, "STATE_DFU_MANIFEST_SYNC");
  
  			ret = at76_dfu_get_status(udev, dfu_stat_buf);
  			if (ret < 0)
  				break;
  
  			*dfu_state = dfu_stat_buf->state;
  			dfu_timeout = at76_get_timeout(dfu_stat_buf);
  			need_dfu_state = 0;
  
  			/* override the timeout from the status response,
  			   needed for AT76C505A */
  			if (manifest_sync_timeout > 0)
  				dfu_timeout = manifest_sync_timeout;
  
  			at76_dbg(DBG_DFU, "DFU: Waiting for manifest phase");
  			schedule_timeout_interruptible(dfu_timeout);
  			break;
  
  		case STATE_DFU_MANIFEST:
  			at76_dbg(DBG_DFU, "STATE_DFU_MANIFEST");
  			is_done = 1;
  			break;
  
  		case STATE_DFU_MANIFEST_WAIT_RESET:
  			at76_dbg(DBG_DFU, "STATE_DFU_MANIFEST_WAIT_RESET");
  			is_done = 1;
  			break;
  
  		case STATE_DFU_UPLOAD_IDLE:
  			at76_dbg(DBG_DFU, "STATE_DFU_UPLOAD_IDLE");
  			break;
  
  		case STATE_DFU_ERROR:
  			at76_dbg(DBG_DFU, "STATE_DFU_ERROR");
  			ret = -EPIPE;
  			break;
  
  		default:
  			at76_dbg(DBG_DFU, "DFU UNKNOWN STATE (%d)", *dfu_state);
  			ret = -EINVAL;
  			break;
  		}
  	} while (!is_done && (ret >= 0));
  
  exit:
  	kfree(dfu_state);
  	kfree(block);
  	kfree(dfu_stat_buf);
  
  	if (ret >= 0)
  		ret = 0;
  
  	return ret;
  }
  
  /* LED trigger */
  static int tx_activity;
  static void at76_ledtrig_tx_timerfunc(unsigned long data);
  static DEFINE_TIMER(ledtrig_tx_timer, at76_ledtrig_tx_timerfunc, 0, 0);
  DEFINE_LED_TRIGGER(ledtrig_tx);
  
  static void at76_ledtrig_tx_timerfunc(unsigned long data)
  {
  	static int tx_lastactivity;
  
  	if (tx_lastactivity != tx_activity) {
  		tx_lastactivity = tx_activity;
  		led_trigger_event(ledtrig_tx, LED_FULL);
  		mod_timer(&ledtrig_tx_timer, jiffies + HZ / 4);
  	} else
  		led_trigger_event(ledtrig_tx, LED_OFF);
  }
  
  static void at76_ledtrig_tx_activity(void)
  {
  	tx_activity++;
  	if (!timer_pending(&ledtrig_tx_timer))
  		mod_timer(&ledtrig_tx_timer, jiffies + HZ / 4);
  }
  
  static int at76_remap(struct usb_device *udev)
  {
  	int ret;
  	ret = usb_control_msg(udev, usb_sndctrlpipe(udev, 0), 0x0a,
  			      USB_TYPE_VENDOR | USB_DIR_OUT |
  			      USB_RECIP_INTERFACE, 0, 0, NULL, 0,
  			      USB_CTRL_GET_TIMEOUT);
  	if (ret < 0)
  		return ret;
  	return 0;
  }
  
  static int at76_get_op_mode(struct usb_device *udev)
  {
  	int ret;
  	u8 saved;
  	u8 *op_mode;
  
  	op_mode = kmalloc(1, GFP_NOIO);
  	if (!op_mode)
  		return -ENOMEM;
  	ret = usb_control_msg(udev, usb_rcvctrlpipe(udev, 0), 0x33,
  			      USB_TYPE_VENDOR | USB_DIR_IN |
  			      USB_RECIP_INTERFACE, 0x01, 0, op_mode, 1,
  			      USB_CTRL_GET_TIMEOUT);
  	saved = *op_mode;
  	kfree(op_mode);
  
  	if (ret < 0)
  		return ret;
  	else if (ret < 1)
  		return -EIO;
  	else
  		return saved;
  }
  
  /* Load a block of the second ("external") part of the firmware */
  static inline int at76_load_ext_fw_block(struct usb_device *udev, int blockno,
  					 void *block, int size)
  {
  	return usb_control_msg(udev, usb_sndctrlpipe(udev, 0), 0x0e,
  			       USB_TYPE_VENDOR | USB_DIR_OUT | USB_RECIP_DEVICE,
  			       0x0802, blockno, block, size,
  			       USB_CTRL_GET_TIMEOUT);
  }
  
  static inline int at76_get_hw_cfg(struct usb_device *udev,
  				  union at76_hwcfg *buf, int buf_size)
  {
  	return usb_control_msg(udev, usb_rcvctrlpipe(udev, 0), 0x33,
  			       USB_TYPE_VENDOR | USB_DIR_IN |
  			       USB_RECIP_INTERFACE, 0x0a02, 0,
  			       buf, buf_size, USB_CTRL_GET_TIMEOUT);
  }
  
  /* Intersil boards use a different "value" for GetHWConfig requests */
  static inline int at76_get_hw_cfg_intersil(struct usb_device *udev,
  					   union at76_hwcfg *buf, int buf_size)
  {
  	return usb_control_msg(udev, usb_rcvctrlpipe(udev, 0), 0x33,
  			       USB_TYPE_VENDOR | USB_DIR_IN |
  			       USB_RECIP_INTERFACE, 0x0902, 0,
  			       buf, buf_size, USB_CTRL_GET_TIMEOUT);
  }
  
  /* Get the hardware configuration for the adapter and put it to the appropriate
   * fields of 'priv' (the GetHWConfig request and interpretation of the result
   * depends on the board type) */
  static int at76_get_hw_config(struct at76_priv *priv)
  {
  	int ret;
  	union at76_hwcfg *hwcfg = kmalloc(sizeof(*hwcfg), GFP_KERNEL);
  
  	if (!hwcfg)
  		return -ENOMEM;
  
  	if (at76_is_intersil(priv->board_type)) {
  		ret = at76_get_hw_cfg_intersil(priv->udev, hwcfg,
  					       sizeof(hwcfg->i));
  		if (ret < 0)
  			goto exit;
  		memcpy(priv->mac_addr, hwcfg->i.mac_addr, ETH_ALEN);
  		priv->regulatory_domain = hwcfg->i.regulatory_domain;
  	} else if (at76_is_503rfmd(priv->board_type)) {
  		ret = at76_get_hw_cfg(priv->udev, hwcfg, sizeof(hwcfg->r3));
  		if (ret < 0)
  			goto exit;
  		memcpy(priv->mac_addr, hwcfg->r3.mac_addr, ETH_ALEN);
  		priv->regulatory_domain = hwcfg->r3.regulatory_domain;
  	} else {
  		ret = at76_get_hw_cfg(priv->udev, hwcfg, sizeof(hwcfg->r5));
  		if (ret < 0)
  			goto exit;
  		memcpy(priv->mac_addr, hwcfg->r5.mac_addr, ETH_ALEN);
  		priv->regulatory_domain = hwcfg->r5.regulatory_domain;
  	}
  
  exit:
  	kfree(hwcfg);
  	if (ret < 0)
  		wiphy_err(priv->hw->wiphy, "cannot get HW Config (error %d)
  ",
  			  ret);
  
  	return ret;
  }
  
  static struct reg_domain const *at76_get_reg_domain(u16 code)
  {
  	int i;
  	static struct reg_domain const fd_tab[] = {
  		{ 0x10, "FCC (USA)", 0x7ff },	/* ch 1-11 */
  		{ 0x20, "IC (Canada)", 0x7ff },	/* ch 1-11 */
  		{ 0x30, "ETSI (most of Europe)", 0x1fff },	/* ch 1-13 */
  		{ 0x31, "Spain", 0x600 },	/* ch 10-11 */
  		{ 0x32, "France", 0x1e00 },	/* ch 10-13 */
  		{ 0x40, "MKK (Japan)", 0x2000 },	/* ch 14 */
  		{ 0x41, "MKK1 (Japan)", 0x3fff },	/* ch 1-14 */
  		{ 0x50, "Israel", 0x3fc },	/* ch 3-9 */
  		{ 0x00, "<unknown>", 0xffffffff }	/* ch 1-32 */
  	};
  
  	/* Last entry is fallback for unknown domain code */
  	for (i = 0; i < ARRAY_SIZE(fd_tab) - 1; i++)
  		if (code == fd_tab[i].code)
  			break;
  
  	return &fd_tab[i];
  }
  
  static inline int at76_get_mib(struct usb_device *udev, u16 mib, void *buf,
  			       int buf_size)
  {
  	int ret;
  
  	ret = usb_control_msg(udev, usb_rcvctrlpipe(udev, 0), 0x33,
  			      USB_TYPE_VENDOR | USB_DIR_IN |
  			      USB_RECIP_INTERFACE, mib << 8, 0, buf, buf_size,
  			      USB_CTRL_GET_TIMEOUT);
  	if (ret >= 0 && ret != buf_size)
  		return -EIO;
  	return ret;
  }
  
  /* Return positive number for status, negative for an error */
  static inline int at76_get_cmd_status(struct usb_device *udev, u8 cmd)
  {
  	u8 *stat_buf;
  	int ret;
  
  	stat_buf = kmalloc(40, GFP_NOIO);
  	if (!stat_buf)
  		return -ENOMEM;
  
  	ret = usb_control_msg(udev, usb_rcvctrlpipe(udev, 0), 0x22,
  			USB_TYPE_VENDOR | USB_DIR_IN |
  			USB_RECIP_INTERFACE, cmd, 0, stat_buf,
  			40, USB_CTRL_GET_TIMEOUT);
  	if (ret >= 0)
  		ret = stat_buf[5];
  	kfree(stat_buf);
  
  	return ret;
  }
  
  #define MAKE_CMD_CASE(c) case (c): return #c
  static const char *at76_get_cmd_string(u8 cmd_status)
  {
  	switch (cmd_status) {
  		MAKE_CMD_CASE(CMD_SET_MIB);
  		MAKE_CMD_CASE(CMD_GET_MIB);
  		MAKE_CMD_CASE(CMD_SCAN);
  		MAKE_CMD_CASE(CMD_JOIN);
  		MAKE_CMD_CASE(CMD_START_IBSS);
  		MAKE_CMD_CASE(CMD_RADIO_ON);
  		MAKE_CMD_CASE(CMD_RADIO_OFF);
  		MAKE_CMD_CASE(CMD_STARTUP);
  	}
  
  	return "UNKNOWN";
  }
  
  static int at76_set_card_command(struct usb_device *udev, u8 cmd, void *buf,
  				 int buf_size)
  {
  	int ret;
  	struct at76_command *cmd_buf = kmalloc(sizeof(struct at76_command) +
  					       buf_size, GFP_KERNEL);
  
  	if (!cmd_buf)
  		return -ENOMEM;
  
  	cmd_buf->cmd = cmd;
  	cmd_buf->reserved = 0;
  	cmd_buf->size = cpu_to_le16(buf_size);
  	memcpy(cmd_buf->data, buf, buf_size);
  
  	at76_dbg_dump(DBG_CMD, cmd_buf, sizeof(struct at76_command) + buf_size,
  		      "issuing command %s (0x%02x)",
  		      at76_get_cmd_string(cmd), cmd);
  
  	ret = usb_control_msg(udev, usb_sndctrlpipe(udev, 0), 0x0e,
  			      USB_TYPE_VENDOR | USB_DIR_OUT | USB_RECIP_DEVICE,
  			      0, 0, cmd_buf,
  			      sizeof(struct at76_command) + buf_size,
  			      USB_CTRL_GET_TIMEOUT);
  	kfree(cmd_buf);
  	return ret;
  }
  
  #define MAKE_CMD_STATUS_CASE(c)	case (c): return #c
  static const char *at76_get_cmd_status_string(u8 cmd_status)
  {
  	switch (cmd_status) {
  		MAKE_CMD_STATUS_CASE(CMD_STATUS_IDLE);
  		MAKE_CMD_STATUS_CASE(CMD_STATUS_COMPLETE);
  		MAKE_CMD_STATUS_CASE(CMD_STATUS_UNKNOWN);
  		MAKE_CMD_STATUS_CASE(CMD_STATUS_INVALID_PARAMETER);
  		MAKE_CMD_STATUS_CASE(CMD_STATUS_FUNCTION_NOT_SUPPORTED);
  		MAKE_CMD_STATUS_CASE(CMD_STATUS_TIME_OUT);
  		MAKE_CMD_STATUS_CASE(CMD_STATUS_IN_PROGRESS);
  		MAKE_CMD_STATUS_CASE(CMD_STATUS_HOST_FAILURE);
  		MAKE_CMD_STATUS_CASE(CMD_STATUS_SCAN_FAILED);
  	}
  
  	return "UNKNOWN";
  }
  
  /* Wait until the command is completed */
  static int at76_wait_completion(struct at76_priv *priv, int cmd)
  {
  	int status = 0;
  	unsigned long timeout = jiffies + CMD_COMPLETION_TIMEOUT;
  
  	do {
  		status = at76_get_cmd_status(priv->udev, cmd);
  		if (status < 0) {
  			wiphy_err(priv->hw->wiphy,
  				  "at76_get_cmd_status failed: %d
  ",
  				  status);
  			break;
  		}
  
  		at76_dbg(DBG_WAIT_COMPLETE,
  			 "%s: Waiting on cmd %d, status = %d (%s)",
  			 wiphy_name(priv->hw->wiphy), cmd, status,
  			 at76_get_cmd_status_string(status));
  
  		if (status != CMD_STATUS_IN_PROGRESS
  		    && status != CMD_STATUS_IDLE)
  			break;
  
  		schedule_timeout_interruptible(HZ / 10);	/* 100 ms */
  		if (time_after(jiffies, timeout)) {
  			wiphy_err(priv->hw->wiphy,
  				  "completion timeout for command %d
  ", cmd);
  			status = -ETIMEDOUT;
  			break;
  		}
  	} while (1);
  
  	return status;
  }
  
  static int at76_set_mib(struct at76_priv *priv, struct set_mib_buffer *buf)
  {
  	int ret;
  
  	ret = at76_set_card_command(priv->udev, CMD_SET_MIB, buf,
  				    offsetof(struct set_mib_buffer,
  					     data) + buf->size);
  	if (ret < 0)
  		return ret;
  
  	ret = at76_wait_completion(priv, CMD_SET_MIB);
  	if (ret != CMD_STATUS_COMPLETE) {
  		wiphy_info(priv->hw->wiphy,
  			   "set_mib: at76_wait_completion failed with %d
  ",
  			   ret);
  		ret = -EIO;
  	}
  
  	return ret;
  }
  
  /* Return < 0 on error, == 0 if no command sent, == 1 if cmd sent */
  static int at76_set_radio(struct at76_priv *priv, int enable)
  {
  	int ret;
  	int cmd;
  
  	if (priv->radio_on == enable)
  		return 0;
  
  	cmd = enable ? CMD_RADIO_ON : CMD_RADIO_OFF;
  
  	ret = at76_set_card_command(priv->udev, cmd, NULL, 0);
  	if (ret < 0)
  		wiphy_err(priv->hw->wiphy,
  			  "at76_set_card_command(%d) failed: %d
  ", cmd, ret);
  	else
  		ret = 1;
  
  	priv->radio_on = enable;
  	return ret;
  }
  
  /* Set current power save mode (AT76_PM_OFF/AT76_PM_ON/AT76_PM_SMART) */
  static int at76_set_pm_mode(struct at76_priv *priv)
  {
  	int ret = 0;
  
  	priv->mib_buf.type = MIB_MAC_MGMT;
  	priv->mib_buf.size = 1;
  	priv->mib_buf.index = offsetof(struct mib_mac_mgmt, power_mgmt_mode);
  	priv->mib_buf.data.byte = priv->pm_mode;
  
  	ret = at76_set_mib(priv, &priv->mib_buf);
  	if (ret < 0)
  		wiphy_err(priv->hw->wiphy, "set_mib (pm_mode) failed: %d
  ",
  			  ret);
  
  	return ret;
  }
  
  static int at76_set_preamble(struct at76_priv *priv, u8 type)
  {
  	int ret = 0;
  
  	priv->mib_buf.type = MIB_LOCAL;
  	priv->mib_buf.size = 1;
  	priv->mib_buf.index = offsetof(struct mib_local, preamble_type);
  	priv->mib_buf.data.byte = type;
  
  	ret = at76_set_mib(priv, &priv->mib_buf);
  	if (ret < 0)
  		wiphy_err(priv->hw->wiphy, "set_mib (preamble) failed: %d
  ",
  			  ret);
  
  	return ret;
  }
  
  static int at76_set_frag(struct at76_priv *priv, u16 size)
  {
  	int ret = 0;
  
  	priv->mib_buf.type = MIB_MAC;
  	priv->mib_buf.size = 2;
  	priv->mib_buf.index = offsetof(struct mib_mac, frag_threshold);
  	priv->mib_buf.data.word = cpu_to_le16(size);
  
  	ret = at76_set_mib(priv, &priv->mib_buf);
  	if (ret < 0)
  		wiphy_err(priv->hw->wiphy,
  			  "set_mib (frag threshold) failed: %d
  ", ret);
  
  	return ret;
  }
  
  static int at76_set_rts(struct at76_priv *priv, u16 size)
  {
  	int ret = 0;
  
  	priv->mib_buf.type = MIB_MAC;
  	priv->mib_buf.size = 2;
  	priv->mib_buf.index = offsetof(struct mib_mac, rts_threshold);
  	priv->mib_buf.data.word = cpu_to_le16(size);
  
  	ret = at76_set_mib(priv, &priv->mib_buf);
  	if (ret < 0)
  		wiphy_err(priv->hw->wiphy, "set_mib (rts) failed: %d
  ", ret);
  
  	return ret;
  }
  
  static int at76_set_autorate_fallback(struct at76_priv *priv, int onoff)
  {
  	int ret = 0;
  
  	priv->mib_buf.type = MIB_LOCAL;
  	priv->mib_buf.size = 1;
  	priv->mib_buf.index = offsetof(struct mib_local, txautorate_fallback);
  	priv->mib_buf.data.byte = onoff;
  
  	ret = at76_set_mib(priv, &priv->mib_buf);
  	if (ret < 0)
  		wiphy_err(priv->hw->wiphy,
  			  "set_mib (autorate fallback) failed: %d
  ", ret);
  
  	return ret;
  }
  
  static void at76_dump_mib_mac_addr(struct at76_priv *priv)
  {
  	int i;
  	int ret;
  	struct mib_mac_addr *m = kmalloc(sizeof(struct mib_mac_addr),
  					 GFP_KERNEL);
  
  	if (!m)
  		return;
  
  	ret = at76_get_mib(priv->udev, MIB_MAC_ADDR, m,
  			   sizeof(struct mib_mac_addr));
  	if (ret < 0) {
  		wiphy_err(priv->hw->wiphy,
  			  "at76_get_mib (MAC_ADDR) failed: %d
  ", ret);
  		goto exit;
  	}
  
  	at76_dbg(DBG_MIB, "%s: MIB MAC_ADDR: mac_addr %pM res 0x%x 0x%x",
  		 wiphy_name(priv->hw->wiphy),
  		 m->mac_addr, m->res[0], m->res[1]);
  	for (i = 0; i < ARRAY_SIZE(m->group_addr); i++)
  		at76_dbg(DBG_MIB, "%s: MIB MAC_ADDR: group addr %d: %pM, "
  			 "status %d", wiphy_name(priv->hw->wiphy), i,
  			 m->group_addr[i], m->group_addr_status[i]);
  exit:
  	kfree(m);
  }
  
  static void at76_dump_mib_mac_wep(struct at76_priv *priv)
  {
  	int i;
  	int ret;
  	int key_len;
  	struct mib_mac_wep *m = kmalloc(sizeof(struct mib_mac_wep), GFP_KERNEL);
  
  	if (!m)
  		return;
  
  	ret = at76_get_mib(priv->udev, MIB_MAC_WEP, m,
  			   sizeof(struct mib_mac_wep));
  	if (ret < 0) {
  		wiphy_err(priv->hw->wiphy,
  			  "at76_get_mib (MAC_WEP) failed: %d
  ", ret);
  		goto exit;
  	}
  
  	at76_dbg(DBG_MIB, "%s: MIB MAC_WEP: priv_invoked %u def_key_id %u "
  		 "key_len %u excl_unencr %u wep_icv_err %u wep_excluded %u "
  		 "encr_level %u key %d", wiphy_name(priv->hw->wiphy),
  		 m->privacy_invoked, m->wep_default_key_id,
  		 m->wep_key_mapping_len, m->exclude_unencrypted,
  		 le32_to_cpu(m->wep_icv_error_count),
  		 le32_to_cpu(m->wep_excluded_count), m->encryption_level,
  		 m->wep_default_key_id);
  
  	key_len = (m->encryption_level == 1) ?
  	    WEP_SMALL_KEY_LEN : WEP_LARGE_KEY_LEN;
  
  	for (i = 0; i < WEP_KEYS; i++)
  		at76_dbg(DBG_MIB, "%s: MIB MAC_WEP: key %d: %*phD",
  			 wiphy_name(priv->hw->wiphy), i,
  			 key_len, m->wep_default_keyvalue[i]);
  exit:
  	kfree(m);
  }
  
  static void at76_dump_mib_mac_mgmt(struct at76_priv *priv)
  {
  	int ret;
  	struct mib_mac_mgmt *m = kmalloc(sizeof(struct mib_mac_mgmt),
  					 GFP_KERNEL);
  
  	if (!m)
  		return;
  
  	ret = at76_get_mib(priv->udev, MIB_MAC_MGMT, m,
  			   sizeof(struct mib_mac_mgmt));
  	if (ret < 0) {
  		wiphy_err(priv->hw->wiphy,
  			  "at76_get_mib (MAC_MGMT) failed: %d
  ", ret);
  		goto exit;
  	}
  
  	at76_dbg(DBG_MIB, "%s: MIB MAC_MGMT: beacon_period %d CFP_max_duration "
  		 "%d medium_occupancy_limit %d station_id 0x%x ATIM_window %d "
  		 "CFP_mode %d privacy_opt_impl %d DTIM_period %d CFP_period %d "
  		 "current_bssid %pM current_essid %*phD current_bss_type %d "
  		 "pm_mode %d ibss_change %d res %d "
  		 "multi_domain_capability_implemented %d "
  		 "international_roaming %d country_string %.3s",
  		 wiphy_name(priv->hw->wiphy), le16_to_cpu(m->beacon_period),
  		 le16_to_cpu(m->CFP_max_duration),
  		 le16_to_cpu(m->medium_occupancy_limit),
  		 le16_to_cpu(m->station_id), le16_to_cpu(m->ATIM_window),
  		 m->CFP_mode, m->privacy_option_implemented, m->DTIM_period,
  		 m->CFP_period, m->current_bssid,
  		 IW_ESSID_MAX_SIZE, m->current_essid,
  		 m->current_bss_type, m->power_mgmt_mode, m->ibss_change,
  		 m->res, m->multi_domain_capability_implemented,
  		 m->multi_domain_capability_enabled, m->country_string);
  exit:
  	kfree(m);
  }
  
  static void at76_dump_mib_mac(struct at76_priv *priv)
  {
  	int ret;
  	struct mib_mac *m = kmalloc(sizeof(struct mib_mac), GFP_KERNEL);
  
  	if (!m)
  		return;
  
  	ret = at76_get_mib(priv->udev, MIB_MAC, m, sizeof(struct mib_mac));
  	if (ret < 0) {
  		wiphy_err(priv->hw->wiphy,
  			  "at76_get_mib (MAC) failed: %d
  ", ret);
  		goto exit;
  	}
  
  	at76_dbg(DBG_MIB, "%s: MIB MAC: max_tx_msdu_lifetime %d "
  		 "max_rx_lifetime %d frag_threshold %d rts_threshold %d "
  		 "cwmin %d cwmax %d short_retry_time %d long_retry_time %d "
  		 "scan_type %d scan_channel %d probe_delay %u "
  		 "min_channel_time %d max_channel_time %d listen_int %d "
  		 "desired_ssid %*phD desired_bssid %pM desired_bsstype %d",
  		 wiphy_name(priv->hw->wiphy),
  		 le32_to_cpu(m->max_tx_msdu_lifetime),
  		 le32_to_cpu(m->max_rx_lifetime),
  		 le16_to_cpu(m->frag_threshold), le16_to_cpu(m->rts_threshold),
  		 le16_to_cpu(m->cwmin), le16_to_cpu(m->cwmax),
  		 m->short_retry_time, m->long_retry_time, m->scan_type,
  		 m->scan_channel, le16_to_cpu(m->probe_delay),
  		 le16_to_cpu(m->min_channel_time),
  		 le16_to_cpu(m->max_channel_time),
  		 le16_to_cpu(m->listen_interval),
  		 IW_ESSID_MAX_SIZE, m->desired_ssid,
  		 m->desired_bssid, m->desired_bsstype);
  exit:
  	kfree(m);
  }
  
  static void at76_dump_mib_phy(struct at76_priv *priv)
  {
  	int ret;
  	struct mib_phy *m = kmalloc(sizeof(struct mib_phy), GFP_KERNEL);
  
  	if (!m)
  		return;
  
  	ret = at76_get_mib(priv->udev, MIB_PHY, m, sizeof(struct mib_phy));
  	if (ret < 0) {
  		wiphy_err(priv->hw->wiphy,
  			  "at76_get_mib (PHY) failed: %d
  ", ret);
  		goto exit;
  	}
  
  	at76_dbg(DBG_MIB, "%s: MIB PHY: ed_threshold %d slot_time %d "
  		 "sifs_time %d preamble_length %d plcp_header_length %d "
  		 "mpdu_max_length %d cca_mode_supported %d operation_rate_set "
  		 "0x%x 0x%x 0x%x 0x%x channel_id %d current_cca_mode %d "
  		 "phy_type %d current_reg_domain %d",
  		 wiphy_name(priv->hw->wiphy), le32_to_cpu(m->ed_threshold),
  		 le16_to_cpu(m->slot_time), le16_to_cpu(m->sifs_time),
  		 le16_to_cpu(m->preamble_length),
  		 le16_to_cpu(m->plcp_header_length),
  		 le16_to_cpu(m->mpdu_max_length),
  		 le16_to_cpu(m->cca_mode_supported), m->operation_rate_set[0],
  		 m->operation_rate_set[1], m->operation_rate_set[2],
  		 m->operation_rate_set[3], m->channel_id, m->current_cca_mode,
  		 m->phy_type, m->current_reg_domain);
  exit:
  	kfree(m);
  }
  
  static void at76_dump_mib_local(struct at76_priv *priv)
  {
  	int ret;
  	struct mib_local *m = kmalloc(sizeof(*m), GFP_KERNEL);
  
  	if (!m)
  		return;
  
  	ret = at76_get_mib(priv->udev, MIB_LOCAL, m, sizeof(*m));
  	if (ret < 0) {
  		wiphy_err(priv->hw->wiphy,
  			  "at76_get_mib (LOCAL) failed: %d
  ", ret);
  		goto exit;
  	}
  
  	at76_dbg(DBG_MIB, "%s: MIB LOCAL: beacon_enable %d "
  		 "txautorate_fallback %d ssid_size %d promiscuous_mode %d "
  		 "preamble_type %d", wiphy_name(priv->hw->wiphy),
  		 m->beacon_enable,
  		 m->txautorate_fallback, m->ssid_size, m->promiscuous_mode,
  		 m->preamble_type);
  exit:
  	kfree(m);
  }
  
  static void at76_dump_mib_mdomain(struct at76_priv *priv)
  {
  	int ret;
  	struct mib_mdomain *m = kmalloc(sizeof(struct mib_mdomain), GFP_KERNEL);
  
  	if (!m)
  		return;
  
  	ret = at76_get_mib(priv->udev, MIB_MDOMAIN, m,
  			   sizeof(struct mib_mdomain));
  	if (ret < 0) {
  		wiphy_err(priv->hw->wiphy,
  			  "at76_get_mib (MDOMAIN) failed: %d
  ", ret);
  		goto exit;
  	}
  
  	at76_dbg(DBG_MIB, "%s: MIB MDOMAIN: channel_list %*phD",
  		 wiphy_name(priv->hw->wiphy),
  		 (int)sizeof(m->channel_list), m->channel_list);
  
  	at76_dbg(DBG_MIB, "%s: MIB MDOMAIN: tx_powerlevel %*phD",
  		 wiphy_name(priv->hw->wiphy),
  		 (int)sizeof(m->tx_powerlevel), m->tx_powerlevel);
  exit:
  	kfree(m);
  }
  
  /* Enable monitor mode */
  static int at76_start_monitor(struct at76_priv *priv)
  {
  	struct at76_req_scan scan;
  	int ret;
  
  	memset(&scan, 0, sizeof(struct at76_req_scan));
  	eth_broadcast_addr(scan.bssid);
  
  	scan.channel = priv->channel;
  	scan.scan_type = SCAN_TYPE_PASSIVE;
  	scan.international_scan = 0;
  	scan.min_channel_time = cpu_to_le16(priv->scan_min_time);
  	scan.max_channel_time = cpu_to_le16(priv->scan_max_time);
  	scan.probe_delay = cpu_to_le16(0);
  
  	ret = at76_set_card_command(priv->udev, CMD_SCAN, &scan, sizeof(scan));
  	if (ret >= 0)
  		ret = at76_get_cmd_status(priv->udev, CMD_SCAN);
  
  	return ret;
  }
  
  /* Calculate padding from txbuf->wlength (which excludes the USB TX header),
     likely to compensate a flaw in the AT76C503A USB part ... */
  static inline int at76_calc_padding(int wlen)
  {
  	/* add the USB TX header */
  	wlen += AT76_TX_HDRLEN;
  
  	wlen = wlen % 64;
  
  	if (wlen < 50)
  		return 50 - wlen;
  
  	if (wlen >= 61)
  		return 64 + 50 - wlen;
  
  	return 0;
  }
  
  static void at76_rx_callback(struct urb *urb)
  {
  	struct at76_priv *priv = urb->context;
  
  	priv->rx_tasklet.data = (unsigned long)urb;
  	tasklet_schedule(&priv->rx_tasklet);
  }
  
  static int at76_submit_rx_urb(struct at76_priv *priv)
  {
  	int ret;
  	int size;
  	struct sk_buff *skb = priv->rx_skb;
  
  	if (!priv->rx_urb) {
  		wiphy_err(priv->hw->wiphy, "%s: priv->rx_urb is NULL
  ",
  			  __func__);
  		return -EFAULT;
  	}
  
  	if (!skb) {
  		skb = dev_alloc_skb(sizeof(struct at76_rx_buffer));
  		if (!skb) {
  			wiphy_err(priv->hw->wiphy,
  				  "cannot allocate rx skbuff
  ");
  			ret = -ENOMEM;
  			goto exit;
  		}
  		priv->rx_skb = skb;
  	} else {
  		skb_push(skb, skb_headroom(skb));
  		skb_trim(skb, 0);
  	}
  
  	size = skb_tailroom(skb);
  	usb_fill_bulk_urb(priv->rx_urb, priv->udev, priv->rx_pipe,
  			  skb_put(skb, size), size, at76_rx_callback, priv);
  	ret = usb_submit_urb(priv->rx_urb, GFP_ATOMIC);
  	if (ret < 0) {
  		if (ret == -ENODEV)
  			at76_dbg(DBG_DEVSTART,
  				 "usb_submit_urb returned -ENODEV");
  		else
  			wiphy_err(priv->hw->wiphy,
  				  "rx, usb_submit_urb failed: %d
  ", ret);
  	}
  
  exit:
  	if (ret < 0 && ret != -ENODEV)
  		wiphy_err(priv->hw->wiphy,
  			  "cannot submit rx urb - please unload the driver and/or power cycle the device
  ");
  
  	return ret;
  }
  
  /* Download external firmware */
  static int at76_load_external_fw(struct usb_device *udev, struct fwentry *fwe)
  {
  	int ret;
  	int op_mode;
  	int blockno = 0;
  	int bsize;
  	u8 *block;
  	u8 *buf = fwe->extfw;
  	int size = fwe->extfw_size;
  
  	if (!buf || !size)
  		return -ENOENT;
  
  	op_mode = at76_get_op_mode(udev);
  	at76_dbg(DBG_DEVSTART, "opmode %d", op_mode);
  
  	if (op_mode != OPMODE_NORMAL_NIC_WITHOUT_FLASH) {
  		dev_err(&udev->dev, "unexpected opmode %d
  ", op_mode);
  		return -EINVAL;
  	}
  
  	block = kmalloc(FW_BLOCK_SIZE, GFP_KERNEL);
  	if (!block)
  		return -ENOMEM;
  
  	at76_dbg(DBG_DEVSTART, "downloading external firmware");
  
  	/* for fw >= 0.100, the device needs an extra empty block */
  	do {
  		bsize = min_t(int, size, FW_BLOCK_SIZE);
  		memcpy(block, buf, bsize);
  		at76_dbg(DBG_DEVSTART,
  			 "ext fw, size left = %5d, bsize = %4d, blockno = %2d",
  			 size, bsize, blockno);
  		ret = at76_load_ext_fw_block(udev, blockno, block, bsize);
  		if (ret != bsize) {
  			dev_err(&udev->dev,
  				"loading %dth firmware block failed: %d
  ",
  				blockno, ret);
  			ret = -EIO;
  			goto exit;
  		}
  		buf += bsize;
  		size -= bsize;
  		blockno++;
  	} while (bsize > 0);
  
  	if (at76_is_505a(fwe->board_type)) {
  		at76_dbg(DBG_DEVSTART, "200 ms delay for 505a");
  		schedule_timeout_interruptible(HZ / 5 + 1);
  	}
  
  exit:
  	kfree(block);
  	if (ret < 0)
  		dev_err(&udev->dev,
  			"downloading external firmware failed: %d
  ", ret);
  	return ret;
  }
  
  /* Download internal firmware */
  static int at76_load_internal_fw(struct usb_device *udev, struct fwentry *fwe)
  {
  	int ret;
  	int need_remap = !at76_is_505a(fwe->board_type);
  
  	ret = at76_usbdfu_download(udev, fwe->intfw, fwe->intfw_size,
  				   need_remap ? 0 : 2 * HZ);
  
  	if (ret < 0) {
  		dev_err(&udev->dev,
  			"downloading internal fw failed with %d
  ", ret);
  		goto exit;
  	}
  
  	at76_dbg(DBG_DEVSTART, "sending REMAP");
  
  	/* no REMAP for 505A (see SF driver) */
  	if (need_remap) {
  		ret = at76_remap(udev);
  		if (ret < 0) {
  			dev_err(&udev->dev,
  				"sending REMAP failed with %d
  ", ret);
  			goto exit;
  		}
  	}
  
  	at76_dbg(DBG_DEVSTART, "sleeping for 2 seconds");
  	schedule_timeout_interruptible(2 * HZ + 1);
  	usb_reset_device(udev);
  
  exit:
  	return ret;
  }
  
  static int at76_startup_device(struct at76_priv *priv)
  {
  	struct at76_card_config *ccfg = &priv->card_config;
  	int ret;
  
  	at76_dbg(DBG_PARAMS,
  		 "%s param: ssid %.*s (%*phD) mode %s ch %d wep %s key %d "
  		 "keylen %d", wiphy_name(priv->hw->wiphy), priv->essid_size,
  		 priv->essid, IW_ESSID_MAX_SIZE, priv->essid,
  		 priv->iw_mode == IW_MODE_ADHOC ? "adhoc" : "infra",
  		 priv->channel, priv->wep_enabled ? "enabled" : "disabled",
  		 priv->wep_key_id, priv->wep_keys_len[priv->wep_key_id]);
  	at76_dbg(DBG_PARAMS,
  		 "%s param: preamble %s rts %d retry %d frag %d "
  		 "txrate %s auth_mode %d", wiphy_name(priv->hw->wiphy),
  		 preambles[priv->preamble_type], priv->rts_threshold,
  		 priv->short_retry_limit, priv->frag_threshold,
  		 priv->txrate == TX_RATE_1MBIT ? "1MBit" : priv->txrate ==
  		 TX_RATE_2MBIT ? "2MBit" : priv->txrate ==
  		 TX_RATE_5_5MBIT ? "5.5MBit" : priv->txrate ==
  		 TX_RATE_11MBIT ? "11MBit" : priv->txrate ==
  		 TX_RATE_AUTO ? "auto" : "<invalid>", priv->auth_mode);
  	at76_dbg(DBG_PARAMS,
  		 "%s param: pm_mode %d pm_period %d auth_mode %s "
  		 "scan_times %d %d scan_mode %s",
  		 wiphy_name(priv->hw->wiphy), priv->pm_mode, priv->pm_period,
  		 priv->auth_mode == WLAN_AUTH_OPEN ? "open" : "shared_secret",
  		 priv->scan_min_time, priv->scan_max_time,
  		 priv->scan_mode == SCAN_TYPE_ACTIVE ? "active" : "passive");
  
  	memset(ccfg, 0, sizeof(struct at76_card_config));
  	ccfg->promiscuous_mode = 0;
  	ccfg->short_retry_limit = priv->short_retry_limit;
  
  	if (priv->wep_enabled) {
  		if (priv->wep_keys_len[priv->wep_key_id] > WEP_SMALL_KEY_LEN)
  			ccfg->encryption_type = 2;
  		else
  			ccfg->encryption_type = 1;
  
  		/* jal: always exclude unencrypted if WEP is active */
  		ccfg->exclude_unencrypted = 1;
  	} else {
  		ccfg->exclude_unencrypted = 0;
  		ccfg->encryption_type = 0;
  	}
  
  	ccfg->rts_threshold = cpu_to_le16(priv->rts_threshold);
  	ccfg->fragmentation_threshold = cpu_to_le16(priv->frag_threshold);
  
  	memcpy(ccfg->basic_rate_set, hw_rates, 4);
  	/* jal: really needed, we do a set_mib for autorate later ??? */
  	ccfg->auto_rate_fallback = (priv->txrate == TX_RATE_AUTO ? 1 : 0);
  	ccfg->channel = priv->channel;
  	ccfg->privacy_invoked = priv->wep_enabled;
  	memcpy(ccfg->current_ssid, priv->essid, IW_ESSID_MAX_SIZE);
  	ccfg->ssid_len = priv->essid_size;
  
  	ccfg->wep_default_key_id = priv->wep_key_id;
  	memcpy(ccfg->wep_default_key_value, priv->wep_keys,
  	       sizeof(priv->wep_keys));
  
  	ccfg->short_preamble = priv->preamble_type;
  	ccfg->beacon_period = cpu_to_le16(priv->beacon_period);
  
  	ret = at76_set_card_command(priv->udev, CMD_STARTUP, &priv->card_config,
  				    sizeof(struct at76_card_config));
  	if (ret < 0) {
  		wiphy_err(priv->hw->wiphy, "at76_set_card_command failed: %d
  ",
  			  ret);
  		return ret;
  	}
  
  	at76_wait_completion(priv, CMD_STARTUP);
  
  	/* remove BSSID from previous run */
  	eth_zero_addr(priv->bssid);
  
  	priv->scanning = false;
  
  	if (at76_set_radio(priv, 1) == 1)
  		at76_wait_completion(priv, CMD_RADIO_ON);
  
  	ret = at76_set_preamble(priv, priv->preamble_type);
  	if (ret < 0)
  		return ret;
  
  	ret = at76_set_frag(priv, priv->frag_threshold);
  	if (ret < 0)
  		return ret;
  
  	ret = at76_set_rts(priv, priv->rts_threshold);
  	if (ret < 0)
  		return ret;
  
  	ret = at76_set_autorate_fallback(priv,
  					 priv->txrate == TX_RATE_AUTO ? 1 : 0);
  	if (ret < 0)
  		return ret;
  
  	ret = at76_set_pm_mode(priv);
  	if (ret < 0)
  		return ret;
  
  	if (at76_debug & DBG_MIB) {
  		at76_dump_mib_mac(priv);
  		at76_dump_mib_mac_addr(priv);
  		at76_dump_mib_mac_mgmt(priv);
  		at76_dump_mib_mac_wep(priv);
  		at76_dump_mib_mdomain(priv);
  		at76_dump_mib_phy(priv);
  		at76_dump_mib_local(priv);
  	}
  
  	return 0;
  }
  
  /* Enable or disable promiscuous mode */
  static void at76_work_set_promisc(struct work_struct *work)
  {
  	struct at76_priv *priv = container_of(work, struct at76_priv,
  					      work_set_promisc);
  	int ret = 0;
  
  	if (priv->device_unplugged)
  		return;
  
  	mutex_lock(&priv->mtx);
  
  	priv->mib_buf.type = MIB_LOCAL;
  	priv->mib_buf.size = 1;
  	priv->mib_buf.index = offsetof(struct mib_local, promiscuous_mode);
  	priv->mib_buf.data.byte = priv->promisc ? 1 : 0;
  
  	ret = at76_set_mib(priv, &priv->mib_buf);
  	if (ret < 0)
  		wiphy_err(priv->hw->wiphy,
  			  "set_mib (promiscuous_mode) failed: %d
  ", ret);
  
  	mutex_unlock(&priv->mtx);
  }
  
  /* Submit Rx urb back to the device */
  static void at76_work_submit_rx(struct work_struct *work)
  {
  	struct at76_priv *priv = container_of(work, struct at76_priv,
  					      work_submit_rx);
  
  	mutex_lock(&priv->mtx);
  	at76_submit_rx_urb(priv);
  	mutex_unlock(&priv->mtx);
  }
  
  /* This is a workaround to make scan working:
   * currently mac80211 does not process frames with no frequency
   * information.
   * However during scan the HW performs a sweep by itself, and we
   * are unable to know where the radio is actually tuned.
   * This function tries to do its best to guess this information..
   * During scan, If the current frame is a beacon or a probe response,
   * the channel information is extracted from it.
   * When not scanning, for other frames, or if it happens that for
   * whatever reason we fail to parse beacons and probe responses, this
   * function returns the priv->channel information, that should be correct
   * at least when we are not scanning.
   */
  static inline int at76_guess_freq(struct at76_priv *priv)
  {
  	size_t el_off;
  	const u8 *el;
  	int channel = priv->channel;
  	int len = priv->rx_skb->len;
  	struct ieee80211_hdr *hdr = (void *)priv->rx_skb->data;
  
  	if (!priv->scanning)
  		goto exit;
  
  	if (len < 24)
  		goto exit;
  
  	if (ieee80211_is_probe_resp(hdr->frame_control)) {
  		el_off = offsetof(struct ieee80211_mgmt, u.probe_resp.variable);
  		el = ((struct ieee80211_mgmt *)hdr)->u.probe_resp.variable;
  	} else if (ieee80211_is_beacon(hdr->frame_control)) {
  		el_off = offsetof(struct ieee80211_mgmt, u.beacon.variable);
  		el = ((struct ieee80211_mgmt *)hdr)->u.beacon.variable;
  	} else {
  		goto exit;
  	}
  	len -= el_off;
  
  	el = cfg80211_find_ie(WLAN_EID_DS_PARAMS, el, len);
  	if (el && el[1] > 0)
  		channel = el[2];
  
  exit:
  	return ieee80211_channel_to_frequency(channel, IEEE80211_BAND_2GHZ);
  }
  
  static void at76_rx_tasklet(unsigned long param)
  {
  	struct urb *urb = (struct urb *)param;
  	struct at76_priv *priv = urb->context;
  	struct at76_rx_buffer *buf;
  	struct ieee80211_rx_status rx_status = { 0 };
  
  	if (priv->device_unplugged) {
  		at76_dbg(DBG_DEVSTART, "device unplugged");
  		at76_dbg(DBG_DEVSTART, "urb status %d", urb->status);
  		return;
  	}
  
  	if (!priv->rx_skb || !priv->rx_skb->data)
  		return;
  
  	buf = (struct at76_rx_buffer *)priv->rx_skb->data;
  
  	if (urb->status != 0) {
  		if (urb->status != -ENOENT && urb->status != -ECONNRESET)
  			at76_dbg(DBG_URB,
  				 "%s %s: - nonzero Rx bulk status received: %d",
  				 __func__, wiphy_name(priv->hw->wiphy),
  				 urb->status);
  		return;
  	}
  
  	at76_dbg(DBG_RX_ATMEL_HDR,
  		 "%s: rx frame: rate %d rssi %d noise %d link %d",
  		 wiphy_name(priv->hw->wiphy), buf->rx_rate, buf->rssi,
  		 buf->noise_level, buf->link_quality);
  
  	skb_pull(priv->rx_skb, AT76_RX_HDRLEN);
  	skb_trim(priv->rx_skb, le16_to_cpu(buf->wlength));
  	at76_dbg_dump(DBG_RX_DATA, priv->rx_skb->data,
  		      priv->rx_skb->len, "RX: len=%d", priv->rx_skb->len);
  
  	rx_status.signal = buf->rssi;
  	rx_status.flag |= RX_FLAG_DECRYPTED;
  	rx_status.flag |= RX_FLAG_IV_STRIPPED;
  	rx_status.band = IEEE80211_BAND_2GHZ;
  	rx_status.freq = at76_guess_freq(priv);
  
  	at76_dbg(DBG_MAC80211, "calling ieee80211_rx_irqsafe(): %d/%d",
  		 priv->rx_skb->len, priv->rx_skb->data_len);
  	memcpy(IEEE80211_SKB_RXCB(priv->rx_skb), &rx_status, sizeof(rx_status));
  	ieee80211_rx_irqsafe(priv->hw, priv->rx_skb);
  
  	/* Use a new skb for the next receive */
  	priv->rx_skb = NULL;
  
  	at76_submit_rx_urb(priv);
  }
  
  /* Load firmware into kernel memory and parse it */
  static struct fwentry *at76_load_firmware(struct usb_device *udev,
  					  enum board_type board_type)
  {
  	int ret;
  	char *str;
  	struct at76_fw_header *fwh;
  	struct fwentry *fwe = &firmwares[board_type];
  
  	mutex_lock(&fw_mutex);
  
  	if (fwe->loaded) {
  		at76_dbg(DBG_FW, "re-using previously loaded fw");
  		goto exit;
  	}
  
  	at76_dbg(DBG_FW, "downloading firmware %s", fwe->fwname);
  	ret = request_firmware(&fwe->fw, fwe->fwname, &udev->dev);
  	if (ret < 0) {
  		dev_err(&udev->dev, "firmware %s not found!
  ",
  			fwe->fwname);
  		dev_err(&udev->dev,
  			"you may need to download the firmware from http://developer.berlios.de/projects/at76c503a/
  ");
  		goto exit;
  	}
  
  	at76_dbg(DBG_FW, "got it.");
  	fwh = (struct at76_fw_header *)(fwe->fw->data);
  
  	if (fwe->fw->size <= sizeof(*fwh)) {
  		dev_err(&udev->dev,
  			"firmware is too short (0x%zx)
  ", fwe->fw->size);
  		goto exit;
  	}
  
  	/* CRC currently not checked */
  	fwe->board_type = le32_to_cpu(fwh->board_type);
  	if (fwe->board_type != board_type) {
  		dev_err(&udev->dev,
  			"board type mismatch, requested %u, got %u
  ",
  			board_type, fwe->board_type);
  		goto exit;
  	}
  
  	fwe->fw_version.major = fwh->major;
  	fwe->fw_version.minor = fwh->minor;
  	fwe->fw_version.patch = fwh->patch;
  	fwe->fw_version.build = fwh->build;
  
  	str = (char *)fwh + le32_to_cpu(fwh->str_offset);
  	fwe->intfw = (u8 *)fwh + le32_to_cpu(fwh->int_fw_offset);
  	fwe->intfw_size = le32_to_cpu(fwh->int_fw_len);
  	fwe->extfw = (u8 *)fwh + le32_to_cpu(fwh->ext_fw_offset);
  	fwe->extfw_size = le32_to_cpu(fwh->ext_fw_len);
  
  	fwe->loaded = 1;
  
  	dev_printk(KERN_DEBUG, &udev->dev,
  		   "using firmware %s (version %d.%d.%d-%d)
  ",
  		   fwe->fwname, fwh->major, fwh->minor, fwh->patch, fwh->build);
  
  	at76_dbg(DBG_DEVSTART, "board %u, int %d:%d, ext %d:%d", board_type,
  		 le32_to_cpu(fwh->int_fw_offset), le32_to_cpu(fwh->int_fw_len),
  		 le32_to_cpu(fwh->ext_fw_offset), le32_to_cpu(fwh->ext_fw_len));
  	at76_dbg(DBG_DEVSTART, "firmware id %s", str);
  
  exit:
  	mutex_unlock(&fw_mutex);
  
  	if (fwe->loaded)
  		return fwe;
  	else
  		return NULL;
  }
  
  static int at76_join(struct at76_priv *priv)
  {
  	struct at76_req_join join;
  	int ret;
  
  	memset(&join, 0, sizeof(struct at76_req_join));
  	memcpy(join.essid, priv->essid, priv->essid_size);
  	join.essid_size = priv->essid_size;
  	memcpy(join.bssid, priv->bssid, ETH_ALEN);
  	join.bss_type = INFRASTRUCTURE_MODE;
  	join.channel = priv->channel;
  	join.timeout = cpu_to_le16(2000);
  
  	at76_dbg(DBG_MAC80211, "%s: sending CMD_JOIN", __func__);
  	ret = at76_set_card_command(priv->udev, CMD_JOIN, &join,
  				    sizeof(struct at76_req_join));
  
  	if (ret < 0) {
  		wiphy_err(priv->hw->wiphy, "at76_set_card_command failed: %d
  ",
  			  ret);
  		return 0;
  	}
  
  	ret = at76_wait_completion(priv, CMD_JOIN);
  	at76_dbg(DBG_MAC80211, "%s: CMD_JOIN returned: 0x%02x", __func__, ret);
  	if (ret != CMD_STATUS_COMPLETE) {
  		wiphy_err(priv->hw->wiphy, "at76_wait_completion failed: %d
  ",
  			  ret);
  		return 0;
  	}
  
  	at76_set_pm_mode(priv);
  
  	return 0;
  }
  
  static void at76_work_join_bssid(struct work_struct *work)
  {
  	struct at76_priv *priv = container_of(work, struct at76_priv,
  					      work_join_bssid);
  
  	if (priv->device_unplugged)
  		return;
  
  	mutex_lock(&priv->mtx);
  
  	if (is_valid_ether_addr(priv->bssid))
  		at76_join(priv);
  
  	mutex_unlock(&priv->mtx);
  }
  
  static void at76_mac80211_tx_callback(struct urb *urb)
  {
  	struct at76_priv *priv = urb->context;
  	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(priv->tx_skb);
  
  	at76_dbg(DBG_MAC80211, "%s()", __func__);
  
  	switch (urb->status) {
  	case 0:
  		/* success */
  		info->flags |= IEEE80211_TX_STAT_ACK;
  		break;
  	case -ENOENT:
  	case -ECONNRESET:
  		/* fail, urb has been unlinked */
  		/* FIXME: add error message */
  		break;
  	default:
  		at76_dbg(DBG_URB, "%s - nonzero tx status received: %d",
  			 __func__, urb->status);
  		break;
  	}
  
  	memset(&info->status, 0, sizeof(info->status));
  
  	ieee80211_tx_status_irqsafe(priv->hw, priv->tx_skb);
  
  	priv->tx_skb = NULL;
  
  	ieee80211_wake_queues(priv->hw);
  }
  
  static void at76_mac80211_tx(struct ieee80211_hw *hw,
  			     struct ieee80211_tx_control *control,
  			     struct sk_buff *skb)
  {
  	struct at76_priv *priv = hw->priv;
  	struct at76_tx_buffer *tx_buffer = priv->bulk_out_buffer;
  	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
  	struct ieee80211_mgmt *mgmt = (struct ieee80211_mgmt *)skb->data;
  	int padding, submit_len, ret;
  
  	at76_dbg(DBG_MAC80211, "%s()", __func__);
  
  	if (priv->tx_urb->status == -EINPROGRESS) {
  		wiphy_err(priv->hw->wiphy,
  			  "%s called while tx urb is pending
  ", __func__);
  		dev_kfree_skb_any(skb);
  		return;
  	}
  
  	/* The following code lines are important when the device is going to
  	 * authenticate with a new bssid. The driver must send CMD_JOIN before
  	 * an authentication frame is transmitted. For this to succeed, the
  	 * correct bssid of the AP must be known. As mac80211 does not inform
  	 * drivers about the bssid prior to the authentication process the
  	 * following workaround is necessary. If the TX frame is an
  	 * authentication frame extract the bssid and send the CMD_JOIN. */
  	if (mgmt->frame_control & cpu_to_le16(IEEE80211_STYPE_AUTH)) {
  		if (!ether_addr_equal_64bits(priv->bssid, mgmt->bssid)) {
  			memcpy(priv->bssid, mgmt->bssid, ETH_ALEN);
  			ieee80211_queue_work(hw, &priv->work_join_bssid);
  			dev_kfree_skb_any(skb);
  			return;
  		}
  	}
  
  	ieee80211_stop_queues(hw);
  
  	at76_ledtrig_tx_activity();	/* tell ledtrigger we send a packet */
  
  	WARN_ON(priv->tx_skb != NULL);
  
  	priv->tx_skb = skb;
  	padding = at76_calc_padding(skb->len);
  	submit_len = AT76_TX_HDRLEN + skb->len + padding;
  
  	/* setup 'Atmel' header */
  	memset(tx_buffer, 0, sizeof(*tx_buffer));
  	tx_buffer->padding = padding;
  	tx_buffer->wlength = cpu_to_le16(skb->len);
  	tx_buffer->tx_rate = ieee80211_get_tx_rate(hw, info)->hw_value;
  	memset(tx_buffer->reserved, 0, sizeof(tx_buffer->reserved));
  	memcpy(tx_buffer->packet, skb->data, skb->len);
  
  	at76_dbg(DBG_TX_DATA, "%s tx: wlen 0x%x pad 0x%x rate %d hdr",
  		 wiphy_name(priv->hw->wiphy), le16_to_cpu(tx_buffer->wlength),
  		 tx_buffer->padding, tx_buffer->tx_rate);
  
  	/* send stuff */
  	at76_dbg_dump(DBG_TX_DATA_CONTENT, tx_buffer, submit_len,
  		      "%s(): tx_buffer %d bytes:", __func__, submit_len);
  	usb_fill_bulk_urb(priv->tx_urb, priv->udev, priv->tx_pipe, tx_buffer,
  			  submit_len, at76_mac80211_tx_callback, priv);
  	ret = usb_submit_urb(priv->tx_urb, GFP_ATOMIC);
  	if (ret) {
  		wiphy_err(priv->hw->wiphy, "error in tx submit urb: %d
  ", ret);
  		if (ret == -EINVAL)
  			wiphy_err(priv->hw->wiphy,
  				  "-EINVAL: tx urb %p hcpriv %p complete %p
  ",
  				  priv->tx_urb,
  				  priv->tx_urb->hcpriv, priv->tx_urb->complete);
  	}
  }
  
  static int at76_mac80211_start(struct ieee80211_hw *hw)
  {
  	struct at76_priv *priv = hw->priv;
  	int ret;
  
  	at76_dbg(DBG_MAC80211, "%s()", __func__);
  
  	mutex_lock(&priv->mtx);
  
  	ret = at76_submit_rx_urb(priv);
  	if (ret < 0) {
  		wiphy_err(priv->hw->wiphy, "open: submit_rx_urb failed: %d
  ",
  			  ret);
  		goto error;
  	}
  
  	at76_startup_device(priv);
  
  	at76_start_monitor(priv);
  
  error:
  	mutex_unlock(&priv->mtx);
  
  	return 0;
  }
  
  static void at76_mac80211_stop(struct ieee80211_hw *hw)
  {
  	struct at76_priv *priv = hw->priv;
  
  	at76_dbg(DBG_MAC80211, "%s()", __func__);
  
  	cancel_delayed_work(&priv->dwork_hw_scan);
  	cancel_work_sync(&priv->work_join_bssid);
  	cancel_work_sync(&priv->work_set_promisc);
  
  	mutex_lock(&priv->mtx);
  
  	if (!priv->device_unplugged) {
  		/* We are called by "ifconfig ethX down", not because the
  		 * device is not available anymore. */
  		at76_set_radio(priv, 0);
  
  		/* We unlink rx_urb because at76_open() re-submits it.
  		 * If unplugged, at76_delete_device() takes care of it. */
  		usb_kill_urb(priv->rx_urb);
  	}
  
  	mutex_unlock(&priv->mtx);
  }
  
  static int at76_add_interface(struct ieee80211_hw *hw,
  			      struct ieee80211_vif *vif)
  {
  	struct at76_priv *priv = hw->priv;
  	int ret = 0;
  
  	at76_dbg(DBG_MAC80211, "%s()", __func__);
  
  	mutex_lock(&priv->mtx);
  
  	switch (vif->type) {
  	case NL80211_IFTYPE_STATION:
  		priv->iw_mode = IW_MODE_INFRA;
  		break;
  	default:
  		ret = -EOPNOTSUPP;
  		goto exit;
  	}
  
  exit:
  	mutex_unlock(&priv->mtx);
  
  	return ret;
  }
  
  static void at76_remove_interface(struct ieee80211_hw *hw,
  				  struct ieee80211_vif *vif)
  {
  	at76_dbg(DBG_MAC80211, "%s()", __func__);
  }
  
  static void at76_dwork_hw_scan(struct work_struct *work)
  {
  	struct at76_priv *priv = container_of(work, struct at76_priv,
  					      dwork_hw_scan.work);
  	int ret;
  
  	if (priv->device_unplugged)
  		return;
  
  	mutex_lock(&priv->mtx);
  
  	ret = at76_get_cmd_status(priv->udev, CMD_SCAN);
  	at76_dbg(DBG_MAC80211, "%s: CMD_SCAN status 0x%02x", __func__, ret);
  
  	/* FIXME: add maximum time for scan to complete */
  
  	if (ret != CMD_STATUS_COMPLETE) {
  		ieee80211_queue_delayed_work(priv->hw, &priv->dwork_hw_scan,
  					     SCAN_POLL_INTERVAL);
  		mutex_unlock(&priv->mtx);
  		return;
  	}
  
  	if (is_valid_ether_addr(priv->bssid))
  		at76_join(priv);
  
  	priv->scanning = false;
  
  	mutex_unlock(&priv->mtx);
  
  	ieee80211_scan_completed(priv->hw, false);
  
  	ieee80211_wake_queues(priv->hw);
  }
  
  static int at76_hw_scan(struct ieee80211_hw *hw,
  			struct ieee80211_vif *vif,
  			struct ieee80211_scan_request *hw_req)
  {
  	struct cfg80211_scan_request *req = &hw_req->req;
  	struct at76_priv *priv = hw->priv;
  	struct at76_req_scan scan;
  	u8 *ssid = NULL;
  	int ret, len = 0;
  
  	at76_dbg(DBG_MAC80211, "%s():", __func__);
  
  	if (priv->device_unplugged)
  		return 0;
  
  	mutex_lock(&priv->mtx);
  
  	ieee80211_stop_queues(hw);
  
  	memset(&scan, 0, sizeof(struct at76_req_scan));
  	eth_broadcast_addr(scan.bssid);
  
  	if (req->n_ssids) {
  		scan.scan_type = SCAN_TYPE_ACTIVE;
  		ssid = req->ssids[0].ssid;
  		len = req->ssids[0].ssid_len;
  	} else {
  		scan.scan_type = SCAN_TYPE_PASSIVE;
  	}
  
  	if (len) {
  		memcpy(scan.essid, ssid, len);
  		scan.essid_size = len;
  	}
  
  	scan.min_channel_time = cpu_to_le16(priv->scan_min_time);
  	scan.max_channel_time = cpu_to_le16(priv->scan_max_time);
  	scan.probe_delay = cpu_to_le16(priv->scan_min_time * 1000);
  	scan.international_scan = 0;
  
  	at76_dbg(DBG_MAC80211, "%s: sending CMD_SCAN", __func__);
  	ret = at76_set_card_command(priv->udev, CMD_SCAN, &scan, sizeof(scan));
  
  	if (ret < 0) {
  		wiphy_err(priv->hw->wiphy, "CMD_SCAN failed: %d
  ", ret);
  		goto exit;
  	}
  
  	priv->scanning = true;
  	ieee80211_queue_delayed_work(priv->hw, &priv->dwork_hw_scan,
  				     SCAN_POLL_INTERVAL);
  
  exit:
  	mutex_unlock(&priv->mtx);
  
  	return 0;
  }
  
  static int at76_config(struct ieee80211_hw *hw, u32 changed)
  {
  	struct at76_priv *priv = hw->priv;
  
  	at76_dbg(DBG_MAC80211, "%s(): channel %d",
  		 __func__, hw->conf.chandef.chan->hw_value);
  	at76_dbg_dump(DBG_MAC80211, priv->bssid, ETH_ALEN, "bssid:");
  
  	mutex_lock(&priv->mtx);
  
  	priv->channel = hw->conf.chandef.chan->hw_value;
  
  	if (is_valid_ether_addr(priv->bssid))
  		at76_join(priv);
  	else
  		at76_start_monitor(priv);
  
  	mutex_unlock(&priv->mtx);
  
  	return 0;
  }
  
  static void at76_bss_info_changed(struct ieee80211_hw *hw,
  				  struct ieee80211_vif *vif,
  				  struct ieee80211_bss_conf *conf,
  				  u32 changed)
  {
  	struct at76_priv *priv = hw->priv;
  
  	at76_dbg(DBG_MAC80211, "%s():", __func__);
  
  	if (!(changed & BSS_CHANGED_BSSID))
  		return;
  
  	at76_dbg_dump(DBG_MAC80211, conf->bssid, ETH_ALEN, "bssid:");
  
  	mutex_lock(&priv->mtx);
  
  	memcpy(priv->bssid, conf->bssid, ETH_ALEN);
  
  	if (is_valid_ether_addr(priv->bssid))
  		/* mac80211 is joining a bss */
  		at76_join(priv);
  
  	mutex_unlock(&priv->mtx);
  }
  
  /* must be atomic */
  static void at76_configure_filter(struct ieee80211_hw *hw,
  				  unsigned int changed_flags,
  				  unsigned int *total_flags, u64 multicast)
  {
  	struct at76_priv *priv = hw->priv;
  	int flags;
  
  	at76_dbg(DBG_MAC80211, "%s(): changed_flags=0x%08x "
  		 "total_flags=0x%08x",
  		 __func__, changed_flags, *total_flags);
  
  	flags = changed_flags & AT76_SUPPORTED_FILTERS;
  	*total_flags = AT76_SUPPORTED_FILTERS;
  
  	/* Bail out after updating flags to prevent a WARN_ON in mac80211. */
  	if (priv->device_unplugged)
  		return;
  
  	/* FIXME: access to priv->promisc should be protected with
  	 * priv->mtx, but it's impossible because this function needs to be
  	 * atomic */
  
  	if (flags && !priv->promisc) {
  		/* mac80211 wants us to enable promiscuous mode */
  		priv->promisc = 1;
  	} else if (!flags && priv->promisc) {
  		/* we need to disable promiscuous mode */
  		priv->promisc = 0;
  	} else
  		return;
  
  	ieee80211_queue_work(hw, &priv->work_set_promisc);
  }
  
  static int at76_set_wep(struct at76_priv *priv)
  {
  	int ret = 0;
  	struct mib_mac_wep *mib_data = &priv->mib_buf.data.wep_mib;
  
  	priv->mib_buf.type = MIB_MAC_WEP;
  	priv->mib_buf.size = sizeof(struct mib_mac_wep);
  	priv->mib_buf.index = 0;
  
  	memset(mib_data, 0, sizeof(*mib_data));
  
  	if (priv->wep_enabled) {
  		if (priv->wep_keys_len[priv->wep_key_id] > WEP_SMALL_KEY_LEN)
  			mib_data->encryption_level = 2;
  		else
  			mib_data->encryption_level = 1;
  
  		/* always exclude unencrypted if WEP is active */
  		mib_data->exclude_unencrypted = 1;
  	} else {
  		mib_data->exclude_unencrypted = 0;
  		mib_data->encryption_level = 0;
  	}
  
  	mib_data->privacy_invoked = priv->wep_enabled;
  	mib_data->wep_default_key_id = priv->wep_key_id;
  	memcpy(mib_data->wep_default_keyvalue, priv->wep_keys,
  	       sizeof(priv->wep_keys));
  
  	ret = at76_set_mib(priv, &priv->mib_buf);
  
  	if (ret < 0)
  		wiphy_err(priv->hw->wiphy,
  			  "set_mib (wep) failed: %d
  ", ret);
  
  	return ret;
  }
  
  static int at76_set_key(struct ieee80211_hw *hw, enum set_key_cmd cmd,
  			struct ieee80211_vif *vif, struct ieee80211_sta *sta,
  			struct ieee80211_key_conf *key)
  {
  	struct at76_priv *priv = hw->priv;
  
  	int i;
  
  	at76_dbg(DBG_MAC80211, "%s(): cmd %d key->cipher %d key->keyidx %d "
  		 "key->keylen %d",
  		 __func__, cmd, key->cipher, key->keyidx, key->keylen);
  
  	if ((key->cipher != WLAN_CIPHER_SUITE_WEP40) &&
  	    (key->cipher != WLAN_CIPHER_SUITE_WEP104))
  		return -EOPNOTSUPP;
  
  	key->hw_key_idx = key->keyidx;
  
  	mutex_lock(&priv->mtx);
  
  	switch (cmd) {
  	case SET_KEY:
  		memcpy(priv->wep_keys[key->keyidx], key->key, key->keylen);
  		priv->wep_keys_len[key->keyidx] = key->keylen;
  
  		/* FIXME: find out how to do this properly */
  		priv->wep_key_id = key->keyidx;
  
  		break;
  	case DISABLE_KEY:
  	default:
  		priv->wep_keys_len[key->keyidx] = 0;
  		break;
  	}
  
  	priv->wep_enabled = 0;
  
  	for (i = 0; i < WEP_KEYS; i++) {
  		if (priv->wep_keys_len[i] != 0)
  			priv->wep_enabled = 1;
  	}
  
  	at76_set_wep(priv);
  
  	mutex_unlock(&priv->mtx);
  
  	return 0;
  }
  
  static const struct ieee80211_ops at76_ops = {
  	.tx = at76_mac80211_tx,
  	.add_interface = at76_add_interface,
  	.remove_interface = at76_remove_interface,
  	.config = at76_config,
  	.bss_info_changed = at76_bss_info_changed,
  	.configure_filter = at76_configure_filter,
  	.start = at76_mac80211_start,
  	.stop = at76_mac80211_stop,
  	.hw_scan = at76_hw_scan,
  	.set_key = at76_set_key,
  };
  
  /* Allocate network device and initialize private data */
  static struct at76_priv *at76_alloc_new_device(struct usb_device *udev)
  {
  	struct ieee80211_hw *hw;
  	struct at76_priv *priv;
  
  	hw = ieee80211_alloc_hw(sizeof(struct at76_priv), &at76_ops);
  	if (!hw) {
  		printk(KERN_ERR DRIVER_NAME ": could not register"
  		       " ieee80211_hw
  ");
  		return NULL;
  	}
  
  	priv = hw->priv;
  	priv->hw = hw;
  
  	priv->udev = udev;
  
  	mutex_init(&priv->mtx);
  	INIT_WORK(&priv->work_set_promisc, at76_work_set_promisc);
  	INIT_WORK(&priv->work_submit_rx, at76_work_submit_rx);
  	INIT_WORK(&priv->work_join_bssid, at76_work_join_bssid);
  	INIT_DELAYED_WORK(&priv->dwork_hw_scan, at76_dwork_hw_scan);
  
  	tasklet_init(&priv->rx_tasklet, at76_rx_tasklet, 0);
  
  	priv->pm_mode = AT76_PM_OFF;
  	priv->pm_period = 0;
  
  	/* unit us */
  
  	return priv;
  }
  
  static int at76_alloc_urbs(struct at76_priv *priv,
  			   struct usb_interface *interface)
  {
  	struct usb_endpoint_descriptor *endpoint, *ep_in, *ep_out;
  	int i;
  	int buffer_size;
  	struct usb_host_interface *iface_desc;
  
  	at76_dbg(DBG_PROC_ENTRY, "%s: ENTER", __func__);
  
  	at76_dbg(DBG_URB, "%s: NumEndpoints %d ", __func__,
  		 interface->altsetting[0].desc.bNumEndpoints);
  
  	ep_in = NULL;
  	ep_out = NULL;
  	iface_desc = interface->cur_altsetting;
  	for (i = 0; i < iface_desc->desc.bNumEndpoints; i++) {
  		endpoint = &iface_desc->endpoint[i].desc;
  
  		at76_dbg(DBG_URB, "%s: %d. endpoint: addr 0x%x attr 0x%x",
  			 __func__, i, endpoint->bEndpointAddress,
  			 endpoint->bmAttributes);
  
  		if (!ep_in && usb_endpoint_is_bulk_in(endpoint))
  			ep_in = endpoint;
  
  		if (!ep_out && usb_endpoint_is_bulk_out(endpoint))
  			ep_out = endpoint;
  	}
  
  	if (!ep_in || !ep_out) {
  		dev_err(&interface->dev, "bulk endpoints missing
  ");
  		return -ENXIO;
  	}
  
  	priv->rx_pipe = usb_rcvbulkpipe(priv->udev, ep_in->bEndpointAddress);
  	priv->tx_pipe = usb_sndbulkpipe(priv->udev, ep_out->bEndpointAddress);
  
  	priv->rx_urb = usb_alloc_urb(0, GFP_KERNEL);
  	priv->tx_urb = usb_alloc_urb(0, GFP_KERNEL);
  	if (!priv->rx_urb || !priv->tx_urb) {
  		dev_err(&interface->dev, "cannot allocate URB
  ");
  		return -ENOMEM;
  	}
  
  	buffer_size = sizeof(struct at76_tx_buffer) + MAX_PADDING_SIZE;
  	priv->bulk_out_buffer = kmalloc(buffer_size, GFP_KERNEL);
  	if (!priv->bulk_out_buffer)
  		return -ENOMEM;
  
  	at76_dbg(DBG_PROC_ENTRY, "%s: EXIT", __func__);
  
  	return 0;
  }
  
  static struct ieee80211_rate at76_rates[] = {
  	{ .bitrate = 10, .hw_value = TX_RATE_1MBIT, },
  	{ .bitrate = 20, .hw_value = TX_RATE_2MBIT, },
  	{ .bitrate = 55, .hw_value = TX_RATE_5_5MBIT, },
  	{ .bitrate = 110, .hw_value = TX_RATE_11MBIT, },
  };
  
  static struct ieee80211_channel at76_channels[] = {
  	{ .center_freq = 2412, .hw_value = 1 },
  	{ .center_freq = 2417, .hw_value = 2 },
  	{ .center_freq = 2422, .hw_value = 3 },
  	{ .center_freq = 2427, .hw_value = 4 },
  	{ .center_freq = 2432, .hw_value = 5 },
  	{ .center_freq = 2437, .hw_value = 6 },
  	{ .center_freq = 2442, .hw_value = 7 },
  	{ .center_freq = 2447, .hw_value = 8 },
  	{ .center_freq = 2452, .hw_value = 9 },
  	{ .center_freq = 2457, .hw_value = 10 },
  	{ .center_freq = 2462, .hw_value = 11 },
  	{ .center_freq = 2467, .hw_value = 12 },
  	{ .center_freq = 2472, .hw_value = 13 },
  	{ .center_freq = 2484, .hw_value = 14 }
  };
  
  static struct ieee80211_supported_band at76_supported_band = {
  	.channels = at76_channels,
  	.n_channels = ARRAY_SIZE(at76_channels),
  	.bitrates = at76_rates,
  	.n_bitrates = ARRAY_SIZE(at76_rates),
  };
  
  /* Register network device and initialize the hardware */
  static int at76_init_new_device(struct at76_priv *priv,
  				struct usb_interface *interface)
  {
  	struct wiphy *wiphy;
  	size_t len;
  	int ret;
  
  	/* set up the endpoint information */
  	/* check out the endpoints */
  
  	at76_dbg(DBG_DEVSTART, "USB interface: %d endpoints",
  		 interface->cur_altsetting->desc.bNumEndpoints);
  
  	ret = at76_alloc_urbs(priv, interface);
  	if (ret < 0)
  		goto exit;
  
  	/* MAC address */
  	ret = at76_get_hw_config(priv);
  	if (ret < 0) {
  		dev_err(&interface->dev, "cannot get MAC address
  ");
  		goto exit;
  	}
  
  	priv->domain = at76_get_reg_domain(priv->regulatory_domain);
  
  	priv->channel = DEF_CHANNEL;
  	priv->iw_mode = IW_MODE_INFRA;
  	priv->rts_threshold = DEF_RTS_THRESHOLD;
  	priv->frag_threshold = DEF_FRAG_THRESHOLD;
  	priv->short_retry_limit = DEF_SHORT_RETRY_LIMIT;
  	priv->txrate = TX_RATE_AUTO;
  	priv->preamble_type = PREAMBLE_TYPE_LONG;
  	priv->beacon_period = 100;
  	priv->auth_mode = WLAN_AUTH_OPEN;
  	priv->scan_min_time = DEF_SCAN_MIN_TIME;
  	priv->scan_max_time = DEF_SCAN_MAX_TIME;
  	priv->scan_mode = SCAN_TYPE_ACTIVE;
  	priv->device_unplugged = 0;
  
  	/* mac80211 initialisation */
  	wiphy = priv->hw->wiphy;
  	priv->hw->wiphy->max_scan_ssids = 1;
  	priv->hw->wiphy->max_scan_ie_len = 0;
  	priv->hw->wiphy->interface_modes = BIT(NL80211_IFTYPE_STATION);
  	priv->hw->wiphy->bands[IEEE80211_BAND_2GHZ] = &at76_supported_band;
  	ieee80211_hw_set(priv->hw, RX_INCLUDES_FCS);
  	ieee80211_hw_set(priv->hw, SIGNAL_UNSPEC);
  	priv->hw->max_signal = 100;
  
  	SET_IEEE80211_DEV(priv->hw, &interface->dev);
  	SET_IEEE80211_PERM_ADDR(priv->hw, priv->mac_addr);
  
  	len = sizeof(wiphy->fw_version);
  	snprintf(wiphy->fw_version, len, "%d.%d.%d-%d",
  		 priv->fw_version.major, priv->fw_version.minor,
  		 priv->fw_version.patch, priv->fw_version.build);
  
  	wiphy->hw_version = priv->board_type;
  
  	ret = ieee80211_register_hw(priv->hw);
  	if (ret) {
  		printk(KERN_ERR "cannot register mac80211 hw (status %d)!
  ",
  		       ret);
  		goto exit;
  	}
  
  	priv->mac80211_registered = 1;
  
  	wiphy_info(priv->hw->wiphy, "USB %s, MAC %pM, firmware %d.%d.%d-%d
  ",
  		   dev_name(&interface->dev), priv->mac_addr,
  		   priv->fw_version.major, priv->fw_version.minor,
  		   priv->fw_version.patch, priv->fw_version.build);
  	wiphy_info(priv->hw->wiphy, "regulatory domain 0x%02x: %s
  ",
  		   priv->regulatory_domain, priv->domain->name);
  
  exit:
  	return ret;
  }
  
  static void at76_delete_device(struct at76_priv *priv)
  {
  	at76_dbg(DBG_PROC_ENTRY, "%s: ENTER", __func__);
  
  	/* The device is gone, don't bother turning it off */
  	priv->device_unplugged = 1;
  
  	tasklet_kill(&priv->rx_tasklet);
  
  	if (priv->mac80211_registered)
  		ieee80211_unregister_hw(priv->hw);
  
  	if (priv->tx_urb) {
  		usb_kill_urb(priv->tx_urb);
  		usb_free_urb(priv->tx_urb);
  	}
  	if (priv->rx_urb) {
  		usb_kill_urb(priv->rx_urb);
  		usb_free_urb(priv->rx_urb);
  	}
  
  	at76_dbg(DBG_PROC_ENTRY, "%s: unlinked urbs", __func__);
  
  	kfree(priv->bulk_out_buffer);
  
  	del_timer_sync(&ledtrig_tx_timer);
  
  	kfree_skb(priv->rx_skb);
  
  	at76_dbg(DBG_PROC_ENTRY, "%s: before freeing priv/ieee80211_hw",
  		 __func__);
  	ieee80211_free_hw(priv->hw);
  
  	at76_dbg(DBG_PROC_ENTRY, "%s: EXIT", __func__);
  }
  
  static int at76_probe(struct usb_interface *interface,
  		      const struct usb_device_id *id)
  {
  	int ret;
  	struct at76_priv *priv;
  	struct fwentry *fwe;
  	struct usb_device *udev;
  	int op_mode;
  	int need_ext_fw = 0;
  	struct mib_fw_version *fwv = NULL;
  	int board_type = (int)id->driver_info;
  
  	udev = usb_get_dev(interface_to_usbdev(interface));
  
  	fwv = kmalloc(sizeof(*fwv), GFP_KERNEL);
  	if (!fwv) {
  		ret = -ENOMEM;
  		goto exit;
  	}
  
  	/* Load firmware into kernel memory */
  	fwe = at76_load_firmware(udev, board_type);
  	if (!fwe) {
  		ret = -ENOENT;
  		goto exit;
  	}
  
  	op_mode = at76_get_op_mode(udev);
  
  	at76_dbg(DBG_DEVSTART, "opmode %d", op_mode);
  
  	/* we get OPMODE_NONE with 2.4.23, SMC2662W-AR ???
  	   we get 204 with 2.4.23, Fiberline FL-WL240u (505A+RFMD2958) ??? */
  
  	if (op_mode == OPMODE_HW_CONFIG_MODE) {
  		dev_err(&interface->dev,
  			"cannot handle a device in HW_CONFIG_MODE
  ");
  		ret = -EBUSY;
  		goto exit;
  	}
  
  	if (op_mode != OPMODE_NORMAL_NIC_WITH_FLASH
  	    && op_mode != OPMODE_NORMAL_NIC_WITHOUT_FLASH) {
  		/* download internal firmware part */
  		dev_printk(KERN_DEBUG, &interface->dev,
  			   "downloading internal firmware
  ");
  		ret = at76_load_internal_fw(udev, fwe);
  		if (ret < 0) {
  			dev_err(&interface->dev,
  				"error %d downloading internal firmware
  ",
  				ret);
  			goto exit;
  		}
  		usb_put_dev(udev);
  		goto exit;
  	}
  
  	/* Internal firmware already inside the device.  Get firmware
  	 * version to test if external firmware is loaded.
  	 * This works only for newer firmware, e.g. the Intersil 0.90.x
  	 * says "control timeout on ep0in" and subsequent
  	 * at76_get_op_mode() fail too :-( */
  
  	/* if version >= 0.100.x.y or device with built-in flash we can
  	 * query the device for the fw version */
  	if ((fwe->fw_version.major > 0 || fwe->fw_version.minor >= 100)
  	    || (op_mode == OPMODE_NORMAL_NIC_WITH_FLASH)) {
  		ret = at76_get_mib(udev, MIB_FW_VERSION, fwv, sizeof(*fwv));
  		if (ret < 0 || (fwv->major | fwv->minor) == 0)
  			need_ext_fw = 1;
  	} else
  		/* No way to check firmware version, reload to be sure */
  		need_ext_fw = 1;
  
  	if (need_ext_fw) {
  		dev_printk(KERN_DEBUG, &interface->dev,
  			   "downloading external firmware
  ");
  
  		ret = at76_load_external_fw(udev, fwe);
  		if (ret < 0)
  			goto exit;
  
  		/* Re-check firmware version */
  		ret = at76_get_mib(udev, MIB_FW_VERSION, fwv, sizeof(*fwv));
  		if (ret < 0) {
  			dev_err(&interface->dev,
  				"error %d getting firmware version
  ", ret);
  			goto exit;
  		}
  	}
  
  	priv = at76_alloc_new_device(udev);
  	if (!priv) {
  		ret = -ENOMEM;
  		goto exit;
  	}
  
  	usb_set_intfdata(interface, priv);
  
  	memcpy(&priv->fw_version, fwv, sizeof(struct mib_fw_version));
  	priv->board_type = board_type;
  
  	ret = at76_init_new_device(priv, interface);
  	if (ret < 0)
  		at76_delete_device(priv);
  
  exit:
  	kfree(fwv);
  	if (ret < 0)
  		usb_put_dev(udev);
  	return ret;
  }
  
  static void at76_disconnect(struct usb_interface *interface)
  {
  	struct at76_priv *priv;
  
  	priv = usb_get_intfdata(interface);
  	usb_set_intfdata(interface, NULL);
  
  	/* Disconnect after loading internal firmware */
  	if (!priv)
  		return;
  
  	wiphy_info(priv->hw->wiphy, "disconnecting
  ");
  	at76_delete_device(priv);
  	usb_put_dev(priv->udev);
  	dev_info(&interface->dev, "disconnected
  ");
  }
  
  /* Structure for registering this driver with the USB subsystem */
  static struct usb_driver at76_driver = {
  	.name = DRIVER_NAME,
  	.probe = at76_probe,
  	.disconnect = at76_disconnect,
  	.id_table = dev_table,
  	.disable_hub_initiated_lpm = 1,
  };
  
  static int __init at76_mod_init(void)
  {
  	int result;
  
  	printk(KERN_INFO DRIVER_DESC " " DRIVER_VERSION " loading
  ");
  
  	mutex_init(&fw_mutex);
  
  	/* register this driver with the USB subsystem */
  	result = usb_register(&at76_driver);
  	if (result < 0)
  		printk(KERN_ERR DRIVER_NAME
  		       ": usb_register failed (status %d)
  ", result);
  
  	led_trigger_register_simple("at76_usb-tx", &ledtrig_tx);
  	return result;
  }
  
  static void __exit at76_mod_exit(void)
  {
  	int i;
  
  	printk(KERN_INFO DRIVER_DESC " " DRIVER_VERSION " unloading
  ");
  	usb_deregister(&at76_driver);
  	for (i = 0; i < ARRAY_SIZE(firmwares); i++)
  		release_firmware(firmwares[i].fw);
  	led_trigger_unregister_simple(ledtrig_tx);
  }
  
  module_param_named(debug, at76_debug, uint, 0600);
  MODULE_PARM_DESC(debug, "Debugging level");
  
  module_init(at76_mod_init);
  module_exit(at76_mod_exit);
  
  MODULE_AUTHOR("Oliver Kurth <oku@masqmail.cx>");
  MODULE_AUTHOR("Joerg Albert <joerg.albert@gmx.de>");
  MODULE_AUTHOR("Alex <alex@foogod.com>");
  MODULE_AUTHOR("Nick Jones");
  MODULE_AUTHOR("Balint Seeber <n0_5p4m_p13453@hotmail.com>");
  MODULE_AUTHOR("Pavel Roskin <proski@gnu.org>");
  MODULE_AUTHOR("Guido Guenther <agx@sigxcpu.org>");
  MODULE_AUTHOR("Kalle Valo <kalle.valo@iki.fi>");
  MODULE_AUTHOR("Sebastian Smolorz <sesmo@gmx.net>");
  MODULE_DESCRIPTION(DRIVER_DESC);
  MODULE_LICENSE("GPL");