Merge /spare/repo/linux-2.6/
[linux-2.6] / drivers / usb / net / zd1201.c
1 /*
2  *      Driver for ZyDAS zd1201 based wireless USB devices.
3  *
4  *      Copyright (c) 2004, 2005 Jeroen Vreeken (pe1rxq@amsat.org)
5  *
6  *      This program is free software; you can redistribute it and/or
7  *      modify it under the terms of the GNU General Public License
8  *      version 2 as published by the Free Software Foundation.
9  *
10  *      Parts of this driver have been derived from a wlan-ng version
11  *      modified by ZyDAS. They also made documentation available, thanks!
12  *      Copyright (C) 1999 AbsoluteValue Systems, Inc.  All Rights Reserved.
13  */
14
15 #include <linux/module.h>
16 #include <linux/usb.h>
17 #include <linux/netdevice.h>
18 #include <linux/etherdevice.h>
19 #include <linux/wireless.h>
20 #include <net/iw_handler.h>
21 #include <linux/string.h>
22 #include <linux/if_arp.h>
23 #include <linux/firmware.h>
24 #include <net/ieee80211.h>
25 #include "zd1201.h"
26
27 static struct usb_device_id zd1201_table[] = {
28         {USB_DEVICE(0x0586, 0x3400)}, /* Peabird Wireless USB Adapter */
29         {USB_DEVICE(0x0ace, 0x1201)}, /* ZyDAS ZD1201 Wireless USB Adapter */
30         {USB_DEVICE(0x050d, 0x6051)}, /* Belkin F5D6051 usb  adapter */
31         {USB_DEVICE(0x0db0, 0x6823)}, /* MSI UB11B usb  adapter */
32         {USB_DEVICE(0x1044, 0x8005)}, /* GIGABYTE GN-WLBZ201 usb adapter */
33         {}
34 };
35
36 static int ap = 0;      /* Are we an AP or a normal station? */
37
38 #define ZD1201_VERSION  "0.15"
39
40 MODULE_AUTHOR("Jeroen Vreeken <pe1rxq@amsat.org>");
41 MODULE_DESCRIPTION("Driver for ZyDAS ZD1201 based USB Wireless adapters");
42 MODULE_VERSION(ZD1201_VERSION);
43 MODULE_LICENSE("GPL");
44 module_param(ap, int, 0);
45 MODULE_PARM_DESC(ap, "If non-zero Access Point firmware will be loaded");
46 MODULE_DEVICE_TABLE(usb, zd1201_table);
47
48
49 static int zd1201_fw_upload(struct usb_device *dev, int apfw)
50 {
51         const struct firmware *fw_entry;
52         char* data;
53         unsigned long len;
54         int err;
55         unsigned char ret;
56         char *buf;
57         char *fwfile;
58
59         if (apfw)
60                 fwfile = "zd1201-ap.fw";
61         else
62                 fwfile = "zd1201.fw";
63
64         err = request_firmware(&fw_entry, fwfile, &dev->dev);
65         if (err) {
66                 dev_err(&dev->dev, "Failed to load %s firmware file!\n", fwfile);
67                 dev_err(&dev->dev, "Make sure the hotplug firmware loader is installed.\n");
68                 dev_err(&dev->dev, "Goto http://linux-lc100020.sourceforge.net for more info\n");
69                 return err;
70         }
71
72         data = fw_entry->data;
73         len = fw_entry->size;
74
75         buf = kmalloc(1024, GFP_ATOMIC);
76         if (!buf)
77                 goto exit;
78         
79         while (len > 0) {
80                 int translen = (len > 1024) ? 1024 : len;
81                 memcpy(buf, data, translen);
82
83                 err = usb_control_msg(dev, usb_sndctrlpipe(dev, 0), 0,
84                     USB_DIR_OUT | 0x40, 0, 0, buf, translen,
85                     ZD1201_FW_TIMEOUT);
86                 if (err < 0)
87                         goto exit;
88
89                 len -= translen;
90                 data += translen;
91         }
92                                         
93         err = usb_control_msg(dev, usb_sndctrlpipe(dev, 0), 0x2,
94             USB_DIR_OUT | 0x40, 0, 0, NULL, 0, ZD1201_FW_TIMEOUT);
95         if (err < 0)
96                 goto exit;
97                                                                                                                                                                 
98         err = usb_control_msg(dev, usb_rcvctrlpipe(dev, 0), 0x4,
99             USB_DIR_IN | 0x40, 0,0, &ret, sizeof(ret), ZD1201_FW_TIMEOUT);
100         if (err < 0)
101                 goto exit;
102                                                                                                                                                                                                                                                                                         
103         if (ret & 0x80) {
104                 err = -EIO;
105                 goto exit;
106         }
107
108         err = 0;
109 exit:
110         kfree(buf);
111         release_firmware(fw_entry);
112         return err;
113 }
114
115 static void zd1201_usbfree(struct urb *urb, struct pt_regs *regs)
116 {
117         struct zd1201 *zd = urb->context;
118
119         switch(urb->status) {
120                 case -EILSEQ:
121                 case -ENODEV:
122                 case -ETIMEDOUT:
123                 case -ENOENT:
124                 case -EPIPE:
125                 case -EOVERFLOW:
126                 case -ESHUTDOWN:
127                         dev_warn(&zd->usb->dev, "%s: urb failed: %d\n", 
128                             zd->dev->name, urb->status);
129         }
130
131         kfree(urb->transfer_buffer);
132         usb_free_urb(urb);
133         return;
134 }
135
136 /* cmdreq message: 
137         u32 type
138         u16 cmd
139         u16 parm0
140         u16 parm1
141         u16 parm2
142         u8  pad[4]
143
144         total: 4 + 2 + 2 + 2 + 2 + 4 = 16
145 */
146 static int zd1201_docmd(struct zd1201 *zd, int cmd, int parm0,
147                         int parm1, int parm2)
148 {
149         unsigned char *command;
150         int ret;
151         struct urb *urb;
152
153         command = kmalloc(16, GFP_ATOMIC);
154         if (!command)
155                 return -ENOMEM;
156
157         *((__le32*)command) = cpu_to_le32(ZD1201_USB_CMDREQ);
158         *((__le16*)&command[4]) = cpu_to_le16(cmd);
159         *((__le16*)&command[6]) = cpu_to_le16(parm0);
160         *((__le16*)&command[8]) = cpu_to_le16(parm1);
161         *((__le16*)&command[10])= cpu_to_le16(parm2);
162
163         urb = usb_alloc_urb(0, GFP_ATOMIC);
164         if (!urb) {
165                 kfree(command);
166                 return -ENOMEM;
167         }
168         usb_fill_bulk_urb(urb, zd->usb, usb_sndbulkpipe(zd->usb, zd->endp_out2),
169              command, 16, zd1201_usbfree, zd);
170         ret = usb_submit_urb(urb, GFP_ATOMIC);
171         if (ret) {
172                 kfree(command);
173                 usb_free_urb(urb);
174         }
175         
176         return ret;
177 }
178
179 /* Callback after sending out a packet */
180 static void zd1201_usbtx(struct urb *urb, struct pt_regs *regs)
181 {
182         struct zd1201 *zd = urb->context;
183         netif_wake_queue(zd->dev);
184         return;
185 }
186
187 /* Incoming data */
188 static void zd1201_usbrx(struct urb *urb, struct pt_regs *regs)
189 {
190         struct zd1201 *zd = urb->context;
191         int free = 0;
192         unsigned char *data = urb->transfer_buffer;
193         struct sk_buff *skb;
194         unsigned char type;
195
196         if (!zd) {
197                 free = 1;
198                 goto exit;
199         }
200
201         switch(urb->status) {
202                 case -EILSEQ:
203                 case -ENODEV:
204                 case -ETIMEDOUT:
205                 case -ENOENT:
206                 case -EPIPE:
207                 case -EOVERFLOW:
208                 case -ESHUTDOWN:
209                         dev_warn(&zd->usb->dev, "%s: rx urb failed: %d\n",
210                             zd->dev->name, urb->status);
211                         free = 1;
212                         goto exit;
213         }
214         
215         if (urb->status != 0 || urb->actual_length == 0)
216                 goto resubmit;
217
218         type = data[0];
219         if (type == ZD1201_PACKET_EVENTSTAT || type == ZD1201_PACKET_RESOURCE) {
220                 memcpy(zd->rxdata, data, urb->actual_length);
221                 zd->rxlen = urb->actual_length;
222                 zd->rxdatas = 1;
223                 wake_up(&zd->rxdataq);
224         }
225         /* Info frame */
226         if (type == ZD1201_PACKET_INQUIRE) {
227                 int i = 0;
228                 unsigned short infotype, framelen, copylen;
229                 framelen = le16_to_cpu(*(__le16*)&data[4]);
230                 infotype = le16_to_cpu(*(__le16*)&data[6]);
231
232                 if (infotype == ZD1201_INF_LINKSTATUS) {
233                         short linkstatus;
234
235                         linkstatus = le16_to_cpu(*(__le16*)&data[8]);
236                         switch(linkstatus) {
237                                 case 1:
238                                         netif_carrier_on(zd->dev);
239                                         break;
240                                 case 2:
241                                         netif_carrier_off(zd->dev);
242                                         break;
243                                 case 3:
244                                         netif_carrier_off(zd->dev);
245                                         break;
246                                 case 4:
247                                         netif_carrier_on(zd->dev);
248                                         break;
249                                 default:
250                                         netif_carrier_off(zd->dev);
251                         }
252                         goto resubmit;
253                 }
254                 if (infotype == ZD1201_INF_ASSOCSTATUS) {
255                         short status = le16_to_cpu(*(__le16*)(data+8));
256                         int event;
257                         union iwreq_data wrqu;
258
259                         switch (status) {
260                                 case ZD1201_ASSOCSTATUS_STAASSOC:
261                                 case ZD1201_ASSOCSTATUS_REASSOC:
262                                         event = IWEVREGISTERED;
263                                         break;
264                                 case ZD1201_ASSOCSTATUS_DISASSOC:
265                                 case ZD1201_ASSOCSTATUS_ASSOCFAIL:
266                                 case ZD1201_ASSOCSTATUS_AUTHFAIL:
267                                 default:
268                                         event = IWEVEXPIRED;
269                         }
270                         memcpy(wrqu.addr.sa_data, data+10, ETH_ALEN);
271                         wrqu.addr.sa_family = ARPHRD_ETHER;
272
273                         /* Send event to user space */
274                         wireless_send_event(zd->dev, event, &wrqu, NULL);
275
276                         goto resubmit;
277                 }
278                 if (infotype == ZD1201_INF_AUTHREQ) {
279                         union iwreq_data wrqu;
280
281                         memcpy(wrqu.addr.sa_data, data+8, ETH_ALEN);
282                         wrqu.addr.sa_family = ARPHRD_ETHER;
283                         /* There isn't a event that trully fits this request.
284                            We assume that userspace will be smart enough to
285                            see a new station being expired and sends back a
286                            authstation ioctl to authorize it. */
287                         wireless_send_event(zd->dev, IWEVEXPIRED, &wrqu, NULL);
288                         goto resubmit;
289                 }
290                 /* Other infotypes are handled outside this handler */
291                 zd->rxlen = 0;
292                 while (i < urb->actual_length) {
293                         copylen = le16_to_cpu(*(__le16*)&data[i+2]);
294                         /* Sanity check, sometimes we get junk */
295                         if (copylen+zd->rxlen > sizeof(zd->rxdata))
296                                 break;
297                         memcpy(zd->rxdata+zd->rxlen, data+i+4, copylen);
298                         zd->rxlen += copylen;
299                         i += 64;
300                 }
301                 if (i >= urb->actual_length) {
302                         zd->rxdatas = 1;
303                         wake_up(&zd->rxdataq);
304                 }
305                 goto  resubmit;
306         }
307         /* Actual data */
308         if (data[urb->actual_length-1] == ZD1201_PACKET_RXDATA) {
309                 int datalen = urb->actual_length-1;
310                 unsigned short len, fc, seq;
311                 struct hlist_node *node;
312
313                 len = ntohs(*(__be16 *)&data[datalen-2]);
314                 if (len>datalen)
315                         len=datalen;
316                 fc = le16_to_cpu(*(__le16 *)&data[datalen-16]);
317                 seq = le16_to_cpu(*(__le16 *)&data[datalen-24]);
318
319                 if(zd->monitor) {
320                         if (datalen < 24)
321                                 goto resubmit;
322                         if (!(skb = dev_alloc_skb(datalen+24)))
323                                 goto resubmit;
324                         
325                         memcpy(skb_put(skb, 2), &data[datalen-16], 2);
326                         memcpy(skb_put(skb, 2), &data[datalen-2], 2);
327                         memcpy(skb_put(skb, 6), &data[datalen-14], 6);
328                         memcpy(skb_put(skb, 6), &data[datalen-22], 6);
329                         memcpy(skb_put(skb, 6), &data[datalen-8], 6);
330                         memcpy(skb_put(skb, 2), &data[datalen-24], 2);
331                         memcpy(skb_put(skb, len), data, len);
332                         skb->dev = zd->dev;
333                         skb->dev->last_rx = jiffies;
334                         skb->protocol = eth_type_trans(skb, zd->dev);
335                         zd->stats.rx_packets++;
336                         zd->stats.rx_bytes += skb->len;
337                         netif_rx(skb);
338                         goto resubmit;
339                 }
340                         
341                 if ((seq & IEEE80211_SCTL_FRAG) ||
342                     (fc & IEEE80211_FCTL_MOREFRAGS)) {
343                         struct zd1201_frag *frag = NULL;
344                         char *ptr;
345
346                         if (datalen<14)
347                                 goto resubmit;
348                         if ((seq & IEEE80211_SCTL_FRAG) == 0) {
349                                 frag = kmalloc(sizeof(*frag), GFP_ATOMIC);
350                                 if (!frag)
351                                         goto resubmit;
352                                 skb = dev_alloc_skb(IEEE80211_DATA_LEN +14+2);
353                                 if (!skb) {
354                                         kfree(frag);
355                                         goto resubmit;
356                                 }
357                                 frag->skb = skb;
358                                 frag->seq = seq & IEEE80211_SCTL_SEQ;
359                                 skb_reserve(skb, 2);
360                                 memcpy(skb_put(skb, 12), &data[datalen-14], 12);
361                                 memcpy(skb_put(skb, 2), &data[6], 2);
362                                 memcpy(skb_put(skb, len), data+8, len);
363                                 hlist_add_head(&frag->fnode, &zd->fraglist);
364                                 goto resubmit;
365                         }
366                         hlist_for_each_entry(frag, node, &zd->fraglist, fnode)
367                                 if(frag->seq == (seq&IEEE80211_SCTL_SEQ))
368                                         break;
369                         if (!frag)
370                                 goto resubmit;
371                         skb = frag->skb;
372                         ptr = skb_put(skb, len);
373                         if (ptr)
374                                 memcpy(ptr, data+8, len);
375                         if (fc & IEEE80211_FCTL_MOREFRAGS)
376                                 goto resubmit;
377                         hlist_del_init(&frag->fnode);
378                         kfree(frag);
379                         /* Fallthrough */
380                 } else {
381                         if (datalen<14)
382                                 goto resubmit;
383                         skb = dev_alloc_skb(len + 14 + 2);
384                         if (!skb)
385                                 goto resubmit;
386                         skb_reserve(skb, 2);
387                         memcpy(skb_put(skb, 12), &data[datalen-14], 12);
388                         memcpy(skb_put(skb, 2), &data[6], 2);
389                         memcpy(skb_put(skb, len), data+8, len);
390                 }
391                 skb->dev = zd->dev;
392                 skb->dev->last_rx = jiffies;
393                 skb->protocol = eth_type_trans(skb, zd->dev);
394                 zd->stats.rx_packets++;
395                 zd->stats.rx_bytes += skb->len;
396                 netif_rx(skb);
397         }
398 resubmit:
399         memset(data, 0, ZD1201_RXSIZE);
400
401         urb->status = 0;
402         urb->dev = zd->usb;
403         if(usb_submit_urb(urb, GFP_ATOMIC))
404                 free = 1;
405
406 exit:
407         if (free) {
408                 zd->rxlen = 0;
409                 zd->rxdatas = 1;
410                 wake_up(&zd->rxdataq);
411                 kfree(urb->transfer_buffer);
412         }
413         return;
414 }
415
416 static int zd1201_getconfig(struct zd1201 *zd, int rid, void *riddata,
417         unsigned int riddatalen)
418 {
419         int err;
420         int i = 0;
421         int code;
422         int rid_fid;
423         int length;
424         unsigned char *pdata;
425         
426         zd->rxdatas = 0;
427         err = zd1201_docmd(zd, ZD1201_CMDCODE_ACCESS, rid, 0, 0);
428         if (err)
429                 return err;
430
431         wait_event_interruptible(zd->rxdataq, zd->rxdatas);
432         if (!zd->rxlen)
433                 return -EIO;
434
435         code = le16_to_cpu(*(__le16*)(&zd->rxdata[4]));
436         rid_fid = le16_to_cpu(*(__le16*)(&zd->rxdata[6]));
437         length = le16_to_cpu(*(__le16*)(&zd->rxdata[8]));
438         if (length > zd->rxlen)
439                 length = zd->rxlen-6;
440
441         /* If access bit is not on, then error */
442         if ((code & ZD1201_ACCESSBIT) != ZD1201_ACCESSBIT || rid_fid != rid )
443                 return -EINVAL;
444
445         /* Not enough buffer for allocating data */
446         if (riddatalen != (length - 4)) {
447                 dev_dbg(&zd->usb->dev, "riddatalen mismatches, expected=%u, (packet=%u) length=%u, rid=0x%04X, rid_fid=0x%04X\n",
448                     riddatalen, zd->rxlen, length, rid, rid_fid);
449                 return -ENODATA;
450         }
451
452         zd->rxdatas = 0;
453         /* Issue SetRxRid commnd */                     
454         err = zd1201_docmd(zd, ZD1201_CMDCODE_SETRXRID, rid, 0, length);
455         if (err)
456                 return err;
457
458         /* Receive RID record from resource packets */
459         wait_event_interruptible(zd->rxdataq, zd->rxdatas);
460         if (!zd->rxlen)
461                 return -EIO;
462
463         if (zd->rxdata[zd->rxlen - 1] != ZD1201_PACKET_RESOURCE) {
464                 dev_dbg(&zd->usb->dev, "Packet type mismatch: 0x%x not 0x3\n",
465                     zd->rxdata[zd->rxlen-1]);
466                 return -EINVAL;
467         }
468
469         /* Set the data pointer and received data length */
470         pdata = zd->rxdata;
471         length = zd->rxlen;
472
473         do {
474                 int  actual_length;
475
476                 actual_length = (length > 64) ? 64 : length;
477
478                 if(pdata[0] != 0x3) {
479                         dev_dbg(&zd->usb->dev, "Rx Resource packet type error: %02X\n",
480                             pdata[0]);
481                         return -EINVAL;
482                 }
483
484                 if (actual_length != 64) {
485                         /* Trim the last packet type byte */
486                         actual_length--;
487                 }
488
489                 /* Skip the 4 bytes header (RID length and RID) */
490                 if(i == 0) {
491                         pdata += 8;
492                         actual_length -= 8;
493                 }
494                 else {
495                         pdata += 4;
496                         actual_length -= 4;
497                 }
498                 
499                 memcpy(riddata, pdata, actual_length);
500                 riddata += actual_length;
501                 pdata += actual_length;
502                 length -= 64;
503                 i++;
504         } while (length > 0);
505
506         return 0;
507 }
508
509 /*
510  *      resreq:
511  *              byte    type
512  *              byte    sequence
513  *              u16     reserved
514  *              byte    data[12]
515  *      total: 16
516  */
517 static int zd1201_setconfig(struct zd1201 *zd, int rid, void *buf, int len, int wait)
518 {
519         int err;
520         unsigned char *request;
521         int reqlen;
522         char seq=0;
523         struct urb *urb;
524         unsigned int gfp_mask = wait ? GFP_NOIO : GFP_ATOMIC;
525
526         len += 4;                       /* first 4 are for header */
527
528         zd->rxdatas = 0;
529         zd->rxlen = 0;
530         for (seq=0; len > 0; seq++) {
531                 request = kmalloc(16, gfp_mask);
532                 if (!request)
533                         return -ENOMEM;
534                 urb = usb_alloc_urb(0, gfp_mask);
535                 if (!urb) {
536                         kfree(request);
537                         return -ENOMEM;
538                 }
539                 memset(request, 0, 16);
540                 reqlen = len>12 ? 12 : len;
541                 request[0] = ZD1201_USB_RESREQ;
542                 request[1] = seq;
543                 request[2] = 0;
544                 request[3] = 0;
545                 if (request[1] == 0) {
546                         /* add header */
547                         *(__le16*)&request[4] = cpu_to_le16((len-2+1)/2);
548                         *(__le16*)&request[6] = cpu_to_le16(rid);
549                         memcpy(request+8, buf, reqlen-4);
550                         buf += reqlen-4;
551                 } else {
552                         memcpy(request+4, buf, reqlen);
553                         buf += reqlen;
554                 }
555
556                 len -= reqlen;
557
558                 usb_fill_bulk_urb(urb, zd->usb, usb_sndbulkpipe(zd->usb,
559                     zd->endp_out2), request, 16, zd1201_usbfree, zd);
560                 err = usb_submit_urb(urb, gfp_mask);
561                 if (err)
562                         goto err;
563         }
564
565         request = kmalloc(16, gfp_mask);
566         if (!request)
567                 return -ENOMEM;
568         urb = usb_alloc_urb(0, gfp_mask);
569         if (!urb) {
570                 kfree(request);
571                 return -ENOMEM;
572         }
573         *((__le32*)request) = cpu_to_le32(ZD1201_USB_CMDREQ);
574         *((__le16*)&request[4]) = 
575             cpu_to_le16(ZD1201_CMDCODE_ACCESS|ZD1201_ACCESSBIT);
576         *((__le16*)&request[6]) = cpu_to_le16(rid);
577         *((__le16*)&request[8]) = cpu_to_le16(0);
578         *((__le16*)&request[10]) = cpu_to_le16(0);
579         usb_fill_bulk_urb(urb, zd->usb, usb_sndbulkpipe(zd->usb, zd->endp_out2),
580              request, 16, zd1201_usbfree, zd);
581         err = usb_submit_urb(urb, gfp_mask);
582         if (err)
583                 goto err;
584         
585         if (wait) {
586                 wait_event_interruptible(zd->rxdataq, zd->rxdatas);
587                 if (!zd->rxlen || le16_to_cpu(*(__le16*)&zd->rxdata[6]) != rid) {
588                         dev_dbg(&zd->usb->dev, "wrong or no RID received\n");
589                 }
590         }
591
592         return 0;
593 err:
594         kfree(request);
595         usb_free_urb(urb);
596         return err;
597 }
598
599 static inline int zd1201_getconfig16(struct zd1201 *zd, int rid, short *val)
600 {
601         int err;
602         __le16 zdval;
603
604         err = zd1201_getconfig(zd, rid, &zdval, sizeof(__le16));
605         if (err)
606                 return err;
607         *val = le16_to_cpu(zdval);
608         return 0;
609 }
610
611 static inline int zd1201_setconfig16(struct zd1201 *zd, int rid, short val)
612 {
613         __le16 zdval = cpu_to_le16(val);
614         return (zd1201_setconfig(zd, rid, &zdval, sizeof(__le16), 1));
615 }
616
617 static int zd1201_drvr_start(struct zd1201 *zd)
618 {
619         int err, i;
620         short max;
621         __le16 zdmax;
622         unsigned char *buffer;
623         
624         buffer = kmalloc(ZD1201_RXSIZE, GFP_KERNEL);
625         if (!buffer)
626                 return -ENOMEM;
627         memset(buffer, 0, ZD1201_RXSIZE);
628
629         usb_fill_bulk_urb(zd->rx_urb, zd->usb, 
630             usb_rcvbulkpipe(zd->usb, zd->endp_in), buffer, ZD1201_RXSIZE,
631             zd1201_usbrx, zd);
632
633         err = usb_submit_urb(zd->rx_urb, GFP_KERNEL);
634         if (err)
635                 goto err_buffer;
636         
637         err = zd1201_docmd(zd, ZD1201_CMDCODE_INIT, 0, 0, 0);
638         if (err)
639                 goto err_urb;
640
641         err = zd1201_getconfig(zd, ZD1201_RID_CNFMAXTXBUFFERNUMBER, &zdmax,
642             sizeof(__le16));
643         if (err)
644                 goto err_urb;
645
646         max = le16_to_cpu(zdmax);
647         for (i=0; i<max; i++) {
648                 err = zd1201_docmd(zd, ZD1201_CMDCODE_ALLOC, 1514, 0, 0);
649                 if (err)
650                         goto err_urb;
651         }
652
653         return 0;
654
655 err_urb:
656         usb_kill_urb(zd->rx_urb);
657         return err;
658 err_buffer:
659         kfree(buffer);
660         return err;
661 }
662
663 /*      Magic alert: The firmware doesn't seem to like the MAC state being
664  *      toggled in promisc (aka monitor) mode.
665  *      (It works a number of times, but will halt eventually)
666  *      So we turn it of before disabling and on after enabling if needed.
667  */
668 static int zd1201_enable(struct zd1201 *zd)
669 {
670         int err;
671
672         if (zd->mac_enabled)
673                 return 0;
674
675         err = zd1201_docmd(zd, ZD1201_CMDCODE_ENABLE, 0, 0, 0);
676         if (!err)
677                 zd->mac_enabled = 1;
678
679         if (zd->monitor)
680                 err = zd1201_setconfig16(zd, ZD1201_RID_PROMISCUOUSMODE, 1);
681
682         return err;
683 }
684
685 static int zd1201_disable(struct zd1201 *zd)
686 {
687         int err;
688         
689         if (!zd->mac_enabled)
690                 return 0;
691         if (zd->monitor) {
692                 err = zd1201_setconfig16(zd, ZD1201_RID_PROMISCUOUSMODE, 0);
693                 if (err)
694                         return err;
695         }
696
697         err = zd1201_docmd(zd, ZD1201_CMDCODE_DISABLE, 0, 0, 0);
698         if (!err)
699                 zd->mac_enabled = 0;
700         return err;
701 }
702
703 static int zd1201_mac_reset(struct zd1201 *zd)
704 {
705         if (!zd->mac_enabled)
706                 return 0;
707         zd1201_disable(zd);
708         return zd1201_enable(zd);
709 }
710
711 static int zd1201_join(struct zd1201 *zd, char *essid, int essidlen)
712 {
713         int err, val;
714         char buf[IW_ESSID_MAX_SIZE+2];
715
716         err = zd1201_disable(zd);
717         if (err)
718                 return err;
719
720         val = ZD1201_CNFAUTHENTICATION_OPENSYSTEM;
721         val |= ZD1201_CNFAUTHENTICATION_SHAREDKEY;
722         err = zd1201_setconfig16(zd, ZD1201_RID_CNFAUTHENTICATION, val);
723         if (err)
724                 return err;
725
726         *(__le16 *)buf = cpu_to_le16(essidlen);
727         memcpy(buf+2, essid, essidlen);
728         if (!zd->ap) {  /* Normal station */
729                 err = zd1201_setconfig(zd, ZD1201_RID_CNFDESIREDSSID, buf,
730                     IW_ESSID_MAX_SIZE+2, 1);
731                 if (err)
732                         return err;
733         } else {        /* AP */
734                 err = zd1201_setconfig(zd, ZD1201_RID_CNFOWNSSID, buf,
735                     IW_ESSID_MAX_SIZE+2, 1);
736                 if (err)
737                         return err;
738         }
739
740         err = zd1201_setconfig(zd, ZD1201_RID_CNFOWNMACADDR, 
741             zd->dev->dev_addr, zd->dev->addr_len, 1);
742         if (err)
743                 return err;
744
745         err = zd1201_enable(zd);
746         if (err)
747                 return err;
748
749         msleep(100);
750         return 0;
751 }
752
753 static int zd1201_net_open(struct net_device *dev)
754 {
755         struct zd1201 *zd = (struct zd1201 *)dev->priv;
756
757         /* Start MAC with wildcard if no essid set */
758         if (!zd->mac_enabled)
759                 zd1201_join(zd, zd->essid, zd->essidlen);
760         netif_start_queue(dev);
761
762         return 0;
763 }
764
765 static int zd1201_net_stop(struct net_device *dev)
766 {
767         netif_stop_queue(dev);
768         
769         return 0;
770 }
771
772 /*
773         RFC 1042 encapsulates Ethernet frames in 802.11 frames
774         by prefixing them with 0xaa, 0xaa, 0x03) followed by a SNAP OID of 0
775         (0x00, 0x00, 0x00). Zd requires an additional padding, copy
776         of ethernet addresses, length of the standard RFC 1042 packet
777         and a command byte (which is nul for tx).
778         
779         tx frame (from Wlan NG):
780         RFC 1042:
781                 llc             0xAA 0xAA 0x03 (802.2 LLC)
782                 snap            0x00 0x00 0x00 (Ethernet encapsulated)
783                 type            2 bytes, Ethernet type field
784                 payload         (minus eth header)
785         Zydas specific:
786                 padding         1B if (skb->len+8+1)%64==0
787                 Eth MAC addr    12 bytes, Ethernet MAC addresses
788                 length          2 bytes, RFC 1042 packet length 
789                                 (llc+snap+type+payload)
790                 zd              1 null byte, zd1201 packet type
791  */
792 static int zd1201_hard_start_xmit(struct sk_buff *skb, struct net_device *dev)
793 {
794         struct zd1201 *zd = (struct zd1201 *)dev->priv;
795         unsigned char *txbuf = zd->txdata;
796         int txbuflen, pad = 0, err;
797         struct urb *urb = zd->tx_urb;
798
799         if (!zd->mac_enabled || zd->monitor) {
800                 zd->stats.tx_dropped++;
801                 kfree_skb(skb);
802                 return 0;
803         }
804         netif_stop_queue(dev);
805
806         txbuflen = skb->len + 8 + 1;
807         if (txbuflen%64 == 0) {
808                 pad = 1;
809                 txbuflen++;
810         }
811         txbuf[0] = 0xAA;
812         txbuf[1] = 0xAA;
813         txbuf[2] = 0x03;
814         txbuf[3] = 0x00;        /* rfc1042 */
815         txbuf[4] = 0x00;
816         txbuf[5] = 0x00;
817
818         memcpy(txbuf+6, skb->data+12, skb->len-12);
819         if (pad)
820                 txbuf[skb->len-12+6]=0;
821         memcpy(txbuf+skb->len-12+6+pad, skb->data, 12);
822         *(__be16*)&txbuf[skb->len+6+pad] = htons(skb->len-12+6);
823         txbuf[txbuflen-1] = 0;
824
825         usb_fill_bulk_urb(urb, zd->usb, usb_sndbulkpipe(zd->usb, zd->endp_out),
826             txbuf, txbuflen, zd1201_usbtx, zd);
827
828         err = usb_submit_urb(zd->tx_urb, GFP_ATOMIC);
829         if (err) {
830                 zd->stats.tx_errors++;
831                 netif_start_queue(dev);
832                 return err;
833         }
834         zd->stats.tx_packets++;
835         zd->stats.tx_bytes += skb->len;
836         dev->trans_start = jiffies;
837         kfree_skb(skb);
838
839         return 0;
840 }
841
842 static void zd1201_tx_timeout(struct net_device *dev)
843 {
844         struct zd1201 *zd = (struct zd1201 *)dev->priv;
845
846         if (!zd)
847                 return;
848         dev_warn(&zd->usb->dev, "%s: TX timeout, shooting down urb\n",
849             dev->name);
850         usb_unlink_urb(zd->tx_urb);
851         zd->stats.tx_errors++;
852         /* Restart the timeout to quiet the watchdog: */
853         dev->trans_start = jiffies;
854 }
855
856 static int zd1201_set_mac_address(struct net_device *dev, void *p)
857 {
858         struct sockaddr *addr = p;
859         struct zd1201 *zd = (struct zd1201 *)dev->priv;
860         int err;
861
862         if (!zd)
863                 return -ENODEV;
864
865         err = zd1201_setconfig(zd, ZD1201_RID_CNFOWNMACADDR, 
866             addr->sa_data, dev->addr_len, 1);
867         if (err)
868                 return err;
869         memcpy(dev->dev_addr, addr->sa_data, dev->addr_len);
870
871         return zd1201_mac_reset(zd);
872 }
873
874 static struct net_device_stats *zd1201_get_stats(struct net_device *dev)
875 {
876         struct zd1201 *zd = (struct zd1201 *)dev->priv;
877
878         return &zd->stats;
879 }
880
881 static struct iw_statistics *zd1201_get_wireless_stats(struct net_device *dev)
882 {
883         struct zd1201 *zd = (struct zd1201 *)dev->priv;
884
885         return &zd->iwstats;
886 }
887
888 static void zd1201_set_multicast(struct net_device *dev)
889 {
890         struct zd1201 *zd = (struct zd1201 *)dev->priv;
891         struct dev_mc_list *mc = dev->mc_list;
892         unsigned char reqbuf[ETH_ALEN*ZD1201_MAXMULTI];
893         int i;
894
895         if (dev->mc_count > ZD1201_MAXMULTI)
896                 return;
897
898         for (i=0; i<dev->mc_count; i++) {
899                 memcpy(reqbuf+i*ETH_ALEN, mc->dmi_addr, ETH_ALEN);
900                 mc = mc->next;
901         }
902         zd1201_setconfig(zd, ZD1201_RID_CNFGROUPADDRESS, reqbuf,
903             dev->mc_count*ETH_ALEN, 0);
904         
905 }
906
907 static int zd1201_config_commit(struct net_device *dev, 
908     struct iw_request_info *info, struct iw_point *data, char *essid)
909 {
910         struct zd1201 *zd = (struct zd1201 *)dev->priv;
911
912         return zd1201_mac_reset(zd);
913 }
914
915 static int zd1201_get_name(struct net_device *dev,
916     struct iw_request_info *info, char *name, char *extra)
917 {
918         strcpy(name, "IEEE 802.11b");
919
920         return 0;
921 }
922
923 static int zd1201_set_freq(struct net_device *dev,
924     struct iw_request_info *info, struct iw_freq *freq, char *extra)
925 {
926         struct zd1201 *zd = (struct zd1201 *)dev->priv;
927         short channel = 0;
928         int err;
929
930         if (freq->e == 0)
931                 channel = freq->m;
932         else {
933                 if (freq->m >= 2482)
934                         channel = 14;
935                 if (freq->m >= 2407)
936                         channel = (freq->m-2407)/5;
937         }
938
939         err = zd1201_setconfig16(zd, ZD1201_RID_CNFOWNCHANNEL, channel);
940         if (err)
941                 return err;
942
943         zd1201_mac_reset(zd);
944
945         return 0;
946 }
947
948 static int zd1201_get_freq(struct net_device *dev,
949     struct iw_request_info *info, struct iw_freq *freq, char *extra)
950 {
951         struct zd1201 *zd = (struct zd1201 *)dev->priv;
952         short channel;
953         int err;
954
955         err = zd1201_getconfig16(zd, ZD1201_RID_CNFOWNCHANNEL, &channel);
956         if (err)
957                 return err;
958         freq->e = 0;
959         freq->m = channel;
960
961         return 0;
962 }
963
964 static int zd1201_set_mode(struct net_device *dev,
965     struct iw_request_info *info, __u32 *mode, char *extra)
966 {
967         struct zd1201 *zd = (struct zd1201 *)dev->priv;
968         short porttype, monitor = 0;
969         unsigned char buffer[IW_ESSID_MAX_SIZE+2];
970         int err;
971
972         if (zd->ap) {
973                 if (*mode != IW_MODE_MASTER)
974                         return -EINVAL;
975                 return 0;
976         }
977
978         err = zd1201_setconfig16(zd, ZD1201_RID_PROMISCUOUSMODE, 0);
979         if (err)
980                 return err;
981         zd->dev->type = ARPHRD_ETHER;
982         switch(*mode) {
983                 case IW_MODE_MONITOR:
984                         monitor = 1;
985                         zd->dev->type = ARPHRD_IEEE80211;
986                         /* Make sure we are no longer associated with by
987                            setting an 'impossible' essid.
988                            (otherwise we mess up firmware)
989                          */
990                         zd1201_join(zd, "\0-*#\0", 5);
991                         /* Put port in pIBSS */
992                 case 8: /* No pseudo-IBSS in wireless extensions (yet) */
993                         porttype = ZD1201_PORTTYPE_PSEUDOIBSS;
994                         break;
995                 case IW_MODE_ADHOC:
996                         porttype = ZD1201_PORTTYPE_IBSS;
997                         break;
998                 case IW_MODE_INFRA:
999                         porttype = ZD1201_PORTTYPE_BSS;
1000                         break;
1001                 default:
1002                         return -EINVAL;
1003         }
1004
1005         err = zd1201_setconfig16(zd, ZD1201_RID_CNFPORTTYPE, porttype);
1006         if (err)
1007                 return err;
1008         if (zd->monitor && !monitor) {
1009                         zd1201_disable(zd);
1010                         *(__le16 *)buffer = cpu_to_le16(zd->essidlen);
1011                         memcpy(buffer+2, zd->essid, zd->essidlen);
1012                         err = zd1201_setconfig(zd, ZD1201_RID_CNFDESIREDSSID,
1013                             buffer, IW_ESSID_MAX_SIZE+2, 1);
1014                         if (err)
1015                                 return err;
1016         }
1017         zd->monitor=monitor;
1018         /* If monitor mode is set we don't actually turn it on here since it
1019          * is done during mac reset anyway (see zd1201_mac_enable).
1020          */
1021
1022         zd1201_mac_reset(zd);
1023
1024         return 0;
1025 }
1026
1027 static int zd1201_get_mode(struct net_device *dev,
1028     struct iw_request_info *info, __u32 *mode, char *extra)
1029 {
1030         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1031         short porttype;
1032         int err;
1033
1034         err = zd1201_getconfig16(zd, ZD1201_RID_CNFPORTTYPE, &porttype);
1035         if (err)
1036                 return err;
1037         switch(porttype) {
1038                 case ZD1201_PORTTYPE_IBSS:
1039                         *mode = IW_MODE_ADHOC;
1040                         break;
1041                 case ZD1201_PORTTYPE_BSS:
1042                         *mode = IW_MODE_INFRA;
1043                         break;
1044                 case ZD1201_PORTTYPE_WDS:
1045                         *mode = IW_MODE_REPEAT;
1046                         break;
1047                 case ZD1201_PORTTYPE_PSEUDOIBSS:
1048                         *mode = 8;/* No Pseudo-IBSS... */
1049                         break;
1050                 case ZD1201_PORTTYPE_AP:
1051                         *mode = IW_MODE_MASTER;
1052                         break;
1053                 default:
1054                         dev_dbg(&zd->usb->dev, "Unknown porttype: %d\n",
1055                             porttype);
1056                         *mode = IW_MODE_AUTO;
1057         }
1058         if (zd->monitor)
1059                 *mode = IW_MODE_MONITOR;
1060
1061         return 0;
1062 }
1063
1064 static int zd1201_get_range(struct net_device *dev,
1065     struct iw_request_info *info, struct iw_point *wrq, char *extra)
1066 {
1067         struct iw_range *range = (struct iw_range *)extra;
1068
1069         wrq->length = sizeof(struct iw_range);
1070         memset(range, 0, sizeof(struct iw_range));
1071         range->we_version_compiled = WIRELESS_EXT;
1072         range->we_version_source = WIRELESS_EXT;
1073
1074         range->max_qual.qual = 128;
1075         range->max_qual.level = 128;
1076         range->max_qual.noise = 128;
1077         range->max_qual.updated = 7;
1078
1079         range->encoding_size[0] = 5;
1080         range->encoding_size[1] = 13;
1081         range->num_encoding_sizes = 2;
1082         range->max_encoding_tokens = ZD1201_NUMKEYS;
1083
1084         range->num_bitrates = 4;
1085         range->bitrate[0] = 1000000;
1086         range->bitrate[1] = 2000000;
1087         range->bitrate[2] = 5500000;
1088         range->bitrate[3] = 11000000;
1089
1090         range->min_rts = 0;
1091         range->min_frag = ZD1201_FRAGMIN;
1092         range->max_rts = ZD1201_RTSMAX;
1093         range->min_frag = ZD1201_FRAGMAX;
1094
1095         return 0;
1096 }
1097
1098 /*      Little bit of magic here: we only get the quality if we poll
1099  *      for it, and we never get an actual request to trigger such
1100  *      a poll. Therefore we 'assume' that the user will soon ask for
1101  *      the stats after asking the bssid.
1102  */
1103 static int zd1201_get_wap(struct net_device *dev,
1104     struct iw_request_info *info, struct sockaddr *ap_addr, char *extra)
1105 {
1106         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1107         unsigned char buffer[6];
1108
1109         if (!zd1201_getconfig(zd, ZD1201_RID_COMMSQUALITY, buffer, 6)) {
1110                 /* Unfortunately the quality and noise reported is useless.
1111                    they seem to be accumulators that increase until you
1112                    read them, unless we poll on a fixed interval we can't
1113                    use them
1114                  */
1115                 /*zd->iwstats.qual.qual = le16_to_cpu(((__le16 *)buffer)[0]);*/
1116                 zd->iwstats.qual.level = le16_to_cpu(((__le16 *)buffer)[1]);
1117                 /*zd->iwstats.qual.noise = le16_to_cpu(((__le16 *)buffer)[2]);*/
1118                 zd->iwstats.qual.updated = 2;
1119         }
1120
1121         return zd1201_getconfig(zd,ZD1201_RID_CURRENTBSSID,ap_addr->sa_data,6);
1122 }
1123
1124 static int zd1201_set_scan(struct net_device *dev,
1125     struct iw_request_info *info, struct iw_point *srq, char *extra)
1126 {
1127         /* We do everything in get_scan */
1128         return 0;
1129 }
1130
1131 static int zd1201_get_scan(struct net_device *dev,
1132     struct iw_request_info *info, struct iw_point *srq, char *extra)
1133 {
1134         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1135         int err, i, j, enabled_save;
1136         struct iw_event iwe;
1137         char *cev = extra;
1138         char *end_buf = extra + IW_SCAN_MAX_DATA;
1139
1140         /* No scanning in AP mode */
1141         if (zd->ap)
1142                 return -EOPNOTSUPP;
1143
1144         /* Scan doesn't seem to work if disabled */
1145         enabled_save = zd->mac_enabled;
1146         zd1201_enable(zd);
1147
1148         zd->rxdatas = 0;
1149         err = zd1201_docmd(zd, ZD1201_CMDCODE_INQUIRE, 
1150              ZD1201_INQ_SCANRESULTS, 0, 0);
1151         if (err)
1152                 return err;
1153
1154         wait_event_interruptible(zd->rxdataq, zd->rxdatas);
1155         if (!zd->rxlen)
1156                 return -EIO;
1157
1158         if (le16_to_cpu(*(__le16*)&zd->rxdata[2]) != ZD1201_INQ_SCANRESULTS)
1159                 return -EIO;
1160
1161         for(i=8; i<zd->rxlen; i+=62) {
1162                 iwe.cmd = SIOCGIWAP;
1163                 iwe.u.ap_addr.sa_family = ARPHRD_ETHER;
1164                 memcpy(iwe.u.ap_addr.sa_data, zd->rxdata+i+6, 6);
1165                 cev = iwe_stream_add_event(cev, end_buf, &iwe, IW_EV_ADDR_LEN);
1166
1167                 iwe.cmd = SIOCGIWESSID;
1168                 iwe.u.data.length = zd->rxdata[i+16];
1169                 iwe.u.data.flags = 1;
1170                 cev = iwe_stream_add_point(cev, end_buf, &iwe, zd->rxdata+i+18);
1171
1172                 iwe.cmd = SIOCGIWMODE;
1173                 if (zd->rxdata[i+14]&0x01)
1174                         iwe.u.mode = IW_MODE_MASTER;
1175                 else
1176                         iwe.u.mode = IW_MODE_ADHOC;
1177                 cev = iwe_stream_add_event(cev, end_buf, &iwe, IW_EV_UINT_LEN);
1178                 
1179                 iwe.cmd = SIOCGIWFREQ;
1180                 iwe.u.freq.m = zd->rxdata[i+0];
1181                 iwe.u.freq.e = 0;
1182                 cev = iwe_stream_add_event(cev, end_buf, &iwe, IW_EV_FREQ_LEN);
1183                 
1184                 iwe.cmd = SIOCGIWRATE;
1185                 iwe.u.bitrate.fixed = 0;
1186                 iwe.u.bitrate.disabled = 0;
1187                 for (j=0; j<10; j++) if (zd->rxdata[i+50+j]) {
1188                         iwe.u.bitrate.value = (zd->rxdata[i+50+j]&0x7f)*500000;
1189                         cev=iwe_stream_add_event(cev, end_buf, &iwe,
1190                             IW_EV_PARAM_LEN);
1191                 }
1192                 
1193                 iwe.cmd = SIOCGIWENCODE;
1194                 iwe.u.data.length = 0;
1195                 if (zd->rxdata[i+14]&0x10)
1196                         iwe.u.data.flags = IW_ENCODE_ENABLED;
1197                 else
1198                         iwe.u.data.flags = IW_ENCODE_DISABLED;
1199                 cev = iwe_stream_add_point(cev, end_buf, &iwe, NULL);
1200                 
1201                 iwe.cmd = IWEVQUAL;
1202                 iwe.u.qual.qual = zd->rxdata[i+4];
1203                 iwe.u.qual.noise= zd->rxdata[i+2]/10-100;
1204                 iwe.u.qual.level = (256+zd->rxdata[i+4]*100)/255-100;
1205                 iwe.u.qual.updated = 7;
1206                 cev = iwe_stream_add_event(cev, end_buf, &iwe, IW_EV_QUAL_LEN);
1207         }
1208
1209         if (!enabled_save)
1210                 zd1201_disable(zd);
1211
1212         srq->length = cev - extra;
1213         srq->flags = 0;
1214
1215         return 0;
1216 }
1217
1218 static int zd1201_set_essid(struct net_device *dev,
1219     struct iw_request_info *info, struct iw_point *data, char *essid)
1220 {
1221         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1222
1223         if (data->length > IW_ESSID_MAX_SIZE)
1224                 return -EINVAL;
1225         if (data->length < 1)
1226                 data->length = 1;
1227         zd->essidlen = data->length-1;
1228         memset(zd->essid, 0, IW_ESSID_MAX_SIZE+1);
1229         memcpy(zd->essid, essid, data->length);
1230         return zd1201_join(zd, zd->essid, zd->essidlen);
1231 }
1232
1233 static int zd1201_get_essid(struct net_device *dev,
1234     struct iw_request_info *info, struct iw_point *data, char *essid)
1235 {
1236         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1237
1238         memcpy(essid, zd->essid, zd->essidlen);
1239         data->flags = 1;
1240         data->length = zd->essidlen;
1241
1242         return 0;
1243 }
1244
1245 static int zd1201_get_nick(struct net_device *dev, struct iw_request_info *info,
1246     struct iw_point *data, char *nick)
1247 {
1248         strcpy(nick, "zd1201");
1249         data->flags = 1;
1250         data->length = strlen(nick);
1251         return 0;
1252 }
1253
1254 static int zd1201_set_rate(struct net_device *dev,
1255     struct iw_request_info *info, struct iw_param *rrq, char *extra)
1256 {
1257         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1258         short rate;
1259         int err;
1260
1261         switch (rrq->value) {
1262                 case 1000000:
1263                         rate = ZD1201_RATEB1;
1264                         break;
1265                 case 2000000:
1266                         rate = ZD1201_RATEB2;
1267                         break;
1268                 case 5500000:
1269                         rate = ZD1201_RATEB5;
1270                         break;
1271                 case 11000000:
1272                 default:
1273                         rate = ZD1201_RATEB11;
1274                         break;
1275         }
1276         if (!rrq->fixed) { /* Also enable all lower bitrates */
1277                 rate |= rate-1;
1278         }
1279         
1280         err = zd1201_setconfig16(zd, ZD1201_RID_TXRATECNTL, rate);
1281         if (err)
1282                 return err;
1283
1284         return zd1201_mac_reset(zd);
1285 }
1286
1287 static int zd1201_get_rate(struct net_device *dev,
1288     struct iw_request_info *info, struct iw_param *rrq, char *extra)
1289 {
1290         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1291         short rate;
1292         int err;
1293
1294         err = zd1201_getconfig16(zd, ZD1201_RID_CURRENTTXRATE, &rate);
1295         if (err)
1296                 return err;
1297
1298         switch(rate) {
1299                 case 1:
1300                         rrq->value = 1000000;
1301                         break;
1302                 case 2:
1303                         rrq->value = 2000000;
1304                         break;
1305                 case 5:
1306                         rrq->value = 5500000;
1307                         break;
1308                 case 11:
1309                         rrq->value = 11000000;
1310                         break;
1311                 default:
1312                         rrq->value = 0;
1313         }
1314         rrq->fixed = 0;
1315         rrq->disabled = 0;
1316
1317         return 0;
1318 }
1319
1320 static int zd1201_set_rts(struct net_device *dev, struct iw_request_info *info,
1321     struct iw_param *rts, char *extra)
1322 {
1323         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1324         int err;
1325         short val = rts->value;
1326
1327         if (rts->disabled || !rts->fixed)
1328                 val = ZD1201_RTSMAX;
1329         if (val > ZD1201_RTSMAX)
1330                 return -EINVAL;
1331         if (val < 0)
1332                 return -EINVAL;
1333
1334         err = zd1201_setconfig16(zd, ZD1201_RID_CNFRTSTHRESHOLD, val);
1335         if (err)
1336                 return err;
1337         return zd1201_mac_reset(zd);
1338 }
1339
1340 static int zd1201_get_rts(struct net_device *dev, struct iw_request_info *info,
1341     struct iw_param *rts, char *extra)
1342 {
1343         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1344         short rtst;
1345         int err;
1346
1347         err = zd1201_getconfig16(zd, ZD1201_RID_CNFRTSTHRESHOLD, &rtst);
1348         if (err)
1349                 return err;
1350         rts->value = rtst;
1351         rts->disabled = (rts->value == ZD1201_RTSMAX);
1352         rts->fixed = 1;
1353
1354         return 0;
1355 }
1356
1357 static int zd1201_set_frag(struct net_device *dev, struct iw_request_info *info,
1358     struct iw_param *frag, char *extra)
1359 {
1360         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1361         int err;
1362         short val = frag->value;
1363
1364         if (frag->disabled || !frag->fixed)
1365                 val = ZD1201_FRAGMAX;
1366         if (val > ZD1201_FRAGMAX)
1367                 return -EINVAL;
1368         if (val < ZD1201_FRAGMIN)
1369                 return -EINVAL;
1370         if (val & 1)
1371                 return -EINVAL;
1372         err = zd1201_setconfig16(zd, ZD1201_RID_CNFFRAGTHRESHOLD, val);
1373         if (err)
1374                 return err;
1375         return zd1201_mac_reset(zd);
1376 }
1377
1378 static int zd1201_get_frag(struct net_device *dev, struct iw_request_info *info,
1379     struct iw_param *frag, char *extra)
1380 {
1381         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1382         short fragt;
1383         int err;
1384
1385         err = zd1201_getconfig16(zd, ZD1201_RID_CNFFRAGTHRESHOLD, &fragt);
1386         if (err)
1387                 return err;
1388         frag->value = fragt;
1389         frag->disabled = (frag->value == ZD1201_FRAGMAX);
1390         frag->fixed = 1;
1391
1392         return 0;
1393 }
1394
1395 static int zd1201_set_retry(struct net_device *dev,
1396     struct iw_request_info *info, struct iw_param *rrq, char *extra)
1397 {
1398         return 0;
1399 }
1400
1401 static int zd1201_get_retry(struct net_device *dev,
1402     struct iw_request_info *info, struct iw_param *rrq, char *extra)
1403 {
1404         return 0;
1405 }
1406
1407 static int zd1201_set_encode(struct net_device *dev,
1408     struct iw_request_info *info, struct iw_point *erq, char *key)
1409 {
1410         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1411         short i;
1412         int err, rid;
1413
1414         if (erq->length > ZD1201_MAXKEYLEN)
1415                 return -EINVAL;
1416
1417         i = (erq->flags & IW_ENCODE_INDEX)-1;
1418         if (i == -1) {
1419                 err = zd1201_getconfig16(zd,ZD1201_RID_CNFDEFAULTKEYID,&i);
1420                 if (err)
1421                         return err;
1422         } else {
1423                 err = zd1201_setconfig16(zd, ZD1201_RID_CNFDEFAULTKEYID, i);
1424                 if (err)
1425                         return err;
1426         }
1427
1428         if (i < 0 || i >= ZD1201_NUMKEYS)
1429                 return -EINVAL;
1430
1431         rid = ZD1201_RID_CNFDEFAULTKEY0 + i;
1432         err = zd1201_setconfig(zd, rid, key, erq->length, 1);
1433         if (err)
1434                 return err;
1435         zd->encode_keylen[i] = erq->length;
1436         memcpy(zd->encode_keys[i], key, erq->length);
1437
1438         i=0;
1439         if (!(erq->flags & IW_ENCODE_DISABLED & IW_ENCODE_MODE)) {
1440                 i |= 0x01;
1441                 zd->encode_enabled = 1;
1442         } else
1443                 zd->encode_enabled = 0;
1444         if (erq->flags & IW_ENCODE_RESTRICTED & IW_ENCODE_MODE) {
1445                 i |= 0x02;
1446                 zd->encode_restricted = 1;
1447         } else
1448                 zd->encode_restricted = 0;
1449         err = zd1201_setconfig16(zd, ZD1201_RID_CNFWEBFLAGS, i);
1450         if (err)
1451                 return err;
1452
1453         if (zd->encode_enabled)
1454                 i = ZD1201_CNFAUTHENTICATION_SHAREDKEY;
1455         else
1456                 i = ZD1201_CNFAUTHENTICATION_OPENSYSTEM;
1457         err = zd1201_setconfig16(zd, ZD1201_RID_CNFAUTHENTICATION, i);
1458         if (err)
1459                 return err;
1460
1461         return zd1201_mac_reset(zd);
1462 }
1463
1464 static int zd1201_get_encode(struct net_device *dev,
1465     struct iw_request_info *info, struct iw_point *erq, char *key)
1466 {
1467         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1468         short i;
1469         int err;
1470
1471         if (zd->encode_enabled)
1472                 erq->flags = IW_ENCODE_ENABLED;
1473         else
1474                 erq->flags = IW_ENCODE_DISABLED;
1475         if (zd->encode_restricted)
1476                 erq->flags |= IW_ENCODE_RESTRICTED;
1477         else
1478                 erq->flags |= IW_ENCODE_OPEN;
1479
1480         i = (erq->flags & IW_ENCODE_INDEX) -1;
1481         if (i == -1) {
1482                 err = zd1201_getconfig16(zd, ZD1201_RID_CNFDEFAULTKEYID, &i);
1483                 if (err)
1484                         return err;
1485         }
1486         if (i<0 || i>= ZD1201_NUMKEYS)
1487                 return -EINVAL;
1488
1489         erq->flags |= i+1;
1490         
1491         erq->length = zd->encode_keylen[i];
1492         memcpy(key, zd->encode_keys[i], erq->length);
1493
1494         return 0;
1495 }
1496
1497 static int zd1201_set_power(struct net_device *dev, 
1498     struct iw_request_info *info, struct iw_param *vwrq, char *extra)
1499 {
1500         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1501         short enabled, duration, level;
1502         int err;
1503
1504         enabled = vwrq->disabled ? 0 : 1;
1505         if (enabled) {
1506                 if (vwrq->flags & IW_POWER_PERIOD) {
1507                         duration = vwrq->value;
1508                         err = zd1201_setconfig16(zd, 
1509                             ZD1201_RID_CNFMAXSLEEPDURATION, duration);
1510                         if (err)
1511                                 return err;
1512                         goto out;
1513                 }
1514                 if (vwrq->flags & IW_POWER_TIMEOUT) {
1515                         err = zd1201_getconfig16(zd, 
1516                             ZD1201_RID_CNFMAXSLEEPDURATION, &duration);
1517                         if (err)
1518                                 return err;
1519                         level = vwrq->value * 4 / duration;
1520                         if (level > 4)
1521                                 level = 4;
1522                         if (level < 0)
1523                                 level = 0;
1524                         err = zd1201_setconfig16(zd, ZD1201_RID_CNFPMEPS,
1525                             level);
1526                         if (err)
1527                                 return err;
1528                         goto out;
1529                 }
1530                 return -EINVAL;
1531         }
1532 out:
1533         err = zd1201_setconfig16(zd, ZD1201_RID_CNFPMENABLED, enabled);
1534         if (err)
1535                 return err;
1536
1537         return 0;
1538 }
1539
1540 static int zd1201_get_power(struct net_device *dev,
1541     struct iw_request_info *info, struct iw_param *vwrq, char *extra)
1542 {
1543         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1544         short enabled, level, duration;
1545         int err;
1546
1547         err = zd1201_getconfig16(zd, ZD1201_RID_CNFPMENABLED, &enabled);
1548         if (err)
1549                 return err;
1550         err = zd1201_getconfig16(zd, ZD1201_RID_CNFPMEPS, &level);
1551         if (err)
1552                 return err;
1553         err = zd1201_getconfig16(zd, ZD1201_RID_CNFMAXSLEEPDURATION, &duration);
1554         if (err)
1555                 return err;
1556         vwrq->disabled = enabled ? 0 : 1;
1557         if (vwrq->flags & IW_POWER_TYPE) {
1558                 if (vwrq->flags & IW_POWER_PERIOD) {
1559                         vwrq->value = duration;
1560                         vwrq->flags = IW_POWER_PERIOD;
1561                 } else {
1562                         vwrq->value = duration * level / 4;
1563                         vwrq->flags = IW_POWER_TIMEOUT;
1564                 }
1565         }
1566         if (vwrq->flags & IW_POWER_MODE) {
1567                 if (enabled && level)
1568                         vwrq->flags = IW_POWER_UNICAST_R;
1569                 else
1570                         vwrq->flags = IW_POWER_ALL_R;
1571         }
1572
1573         return 0;
1574 }
1575
1576
1577 static const iw_handler zd1201_iw_handler[] =
1578 {
1579         (iw_handler) zd1201_config_commit,      /* SIOCSIWCOMMIT */
1580         (iw_handler) zd1201_get_name,           /* SIOCGIWNAME */
1581         (iw_handler) NULL,                      /* SIOCSIWNWID */
1582         (iw_handler) NULL,                      /* SIOCGIWNWID */
1583         (iw_handler) zd1201_set_freq,           /* SIOCSIWFREQ */
1584         (iw_handler) zd1201_get_freq,           /* SIOCGIWFREQ */
1585         (iw_handler) zd1201_set_mode,           /* SIOCSIWMODE */
1586         (iw_handler) zd1201_get_mode,           /* SIOCGIWMODE */
1587         (iw_handler) NULL,                      /* SIOCSIWSENS */
1588         (iw_handler) NULL,                      /* SIOCGIWSENS */
1589         (iw_handler) NULL,                      /* SIOCSIWRANGE */
1590         (iw_handler) zd1201_get_range,           /* SIOCGIWRANGE */
1591         (iw_handler) NULL,                      /* SIOCSIWPRIV */
1592         (iw_handler) NULL,                      /* SIOCGIWPRIV */
1593         (iw_handler) NULL,                      /* SIOCSIWSTATS */
1594         (iw_handler) NULL,                      /* SIOCGIWSTATS */
1595         (iw_handler) NULL,                      /* SIOCSIWSPY */
1596         (iw_handler) NULL,                      /* SIOCGIWSPY */
1597         (iw_handler) NULL,                      /* -- hole -- */
1598         (iw_handler) NULL,                      /* -- hole -- */
1599         (iw_handler) NULL/*zd1201_set_wap*/,            /* SIOCSIWAP */
1600         (iw_handler) zd1201_get_wap,            /* SIOCGIWAP */
1601         (iw_handler) NULL,                      /* -- hole -- */
1602         (iw_handler) NULL,                      /* SIOCGIWAPLIST */
1603         (iw_handler) zd1201_set_scan,           /* SIOCSIWSCAN */
1604         (iw_handler) zd1201_get_scan,           /* SIOCGIWSCAN */
1605         (iw_handler) zd1201_set_essid,          /* SIOCSIWESSID */
1606         (iw_handler) zd1201_get_essid,          /* SIOCGIWESSID */
1607         (iw_handler) NULL,                      /* SIOCSIWNICKN */
1608         (iw_handler) zd1201_get_nick,           /* SIOCGIWNICKN */
1609         (iw_handler) NULL,                      /* -- hole -- */
1610         (iw_handler) NULL,                      /* -- hole -- */
1611         (iw_handler) zd1201_set_rate,           /* SIOCSIWRATE */
1612         (iw_handler) zd1201_get_rate,           /* SIOCGIWRATE */
1613         (iw_handler) zd1201_set_rts,            /* SIOCSIWRTS */
1614         (iw_handler) zd1201_get_rts,            /* SIOCGIWRTS */
1615         (iw_handler) zd1201_set_frag,           /* SIOCSIWFRAG */
1616         (iw_handler) zd1201_get_frag,           /* SIOCGIWFRAG */
1617         (iw_handler) NULL,                      /* SIOCSIWTXPOW */
1618         (iw_handler) NULL,                      /* SIOCGIWTXPOW */
1619         (iw_handler) zd1201_set_retry,          /* SIOCSIWRETRY */
1620         (iw_handler) zd1201_get_retry,          /* SIOCGIWRETRY */
1621         (iw_handler) zd1201_set_encode,         /* SIOCSIWENCODE */
1622         (iw_handler) zd1201_get_encode,         /* SIOCGIWENCODE */
1623         (iw_handler) zd1201_set_power,          /* SIOCSIWPOWER */
1624         (iw_handler) zd1201_get_power,          /* SIOCGIWPOWER */
1625 };
1626
1627 static int zd1201_set_hostauth(struct net_device *dev,
1628     struct iw_request_info *info, struct iw_param *rrq, char *extra)
1629 {
1630         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1631         int err;
1632
1633         if (!zd->ap)
1634                 return -EOPNOTSUPP;
1635
1636         err = zd1201_setconfig16(zd, ZD1201_RID_CNFHOSTAUTH, rrq->value);
1637         if (err)
1638                 return err;
1639         return 0;
1640 }
1641
1642 static int zd1201_get_hostauth(struct net_device *dev,
1643     struct iw_request_info *info, struct iw_param *rrq, char *extra)
1644 {
1645         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1646         short hostauth;
1647         int err;
1648
1649         if (!zd->ap)
1650                 return -EOPNOTSUPP;
1651
1652         err = zd1201_getconfig16(zd, ZD1201_RID_CNFHOSTAUTH, &hostauth);
1653         if (err)
1654                 return err;
1655         rrq->value = hostauth;
1656         rrq->fixed = 1;
1657
1658         return 0;
1659 }
1660
1661 static int zd1201_auth_sta(struct net_device *dev,
1662     struct iw_request_info *info, struct sockaddr *sta, char *extra)
1663 {
1664         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1665         unsigned char buffer[10];
1666
1667         if (!zd->ap)
1668                 return -EOPNOTSUPP;
1669
1670         memcpy(buffer, sta->sa_data, ETH_ALEN);
1671         *(short*)(buffer+6) = 0;        /* 0==success, 1==failure */
1672         *(short*)(buffer+8) = 0;
1673
1674         return zd1201_setconfig(zd, ZD1201_RID_AUTHENTICATESTA, buffer, 10, 1);
1675 }
1676
1677 static int zd1201_set_maxassoc(struct net_device *dev,
1678     struct iw_request_info *info, struct iw_param *rrq, char *extra)
1679 {
1680         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1681         int err;
1682
1683         if (!zd->ap)
1684                 return -EOPNOTSUPP;
1685
1686         err = zd1201_setconfig16(zd, ZD1201_RID_CNFMAXASSOCSTATIONS, rrq->value);
1687         if (err)
1688                 return err;
1689         return 0;
1690 }
1691
1692 static int zd1201_get_maxassoc(struct net_device *dev,
1693     struct iw_request_info *info, struct iw_param *rrq, char *extra)
1694 {
1695         struct zd1201 *zd = (struct zd1201 *)dev->priv;
1696         short maxassoc;
1697         int err;
1698
1699         if (!zd->ap)
1700                 return -EOPNOTSUPP;
1701
1702         err = zd1201_getconfig16(zd, ZD1201_RID_CNFMAXASSOCSTATIONS, &maxassoc);
1703         if (err)
1704                 return err;
1705         rrq->value = maxassoc;
1706         rrq->fixed = 1;
1707
1708         return 0;
1709 }
1710
1711 static const iw_handler zd1201_private_handler[] = {
1712         (iw_handler) zd1201_set_hostauth,       /* ZD1201SIWHOSTAUTH */
1713         (iw_handler) zd1201_get_hostauth,       /* ZD1201GIWHOSTAUTH */
1714         (iw_handler) zd1201_auth_sta,           /* ZD1201SIWAUTHSTA */
1715         (iw_handler) NULL,                      /* nothing to get */
1716         (iw_handler) zd1201_set_maxassoc,       /* ZD1201SIMAXASSOC */
1717         (iw_handler) zd1201_get_maxassoc,       /* ZD1201GIMAXASSOC */
1718 };
1719
1720 static const struct iw_priv_args zd1201_private_args[] = {
1721         { ZD1201SIWHOSTAUTH, IW_PRIV_TYPE_INT | IW_PRIV_SIZE_FIXED | 1,
1722             IW_PRIV_TYPE_NONE, "sethostauth" },
1723         { ZD1201GIWHOSTAUTH, IW_PRIV_TYPE_NONE,
1724             IW_PRIV_TYPE_INT | IW_PRIV_SIZE_FIXED | 1, "gethostauth" },
1725         { ZD1201SIWAUTHSTA, IW_PRIV_TYPE_ADDR | IW_PRIV_SIZE_FIXED | 1, 
1726             IW_PRIV_TYPE_NONE, "authstation" },
1727         { ZD1201SIWMAXASSOC, IW_PRIV_TYPE_INT | IW_PRIV_SIZE_FIXED | 1,
1728             IW_PRIV_TYPE_NONE, "setmaxassoc" },
1729         { ZD1201GIWMAXASSOC, IW_PRIV_TYPE_NONE,
1730             IW_PRIV_TYPE_INT | IW_PRIV_SIZE_FIXED | 1, "getmaxassoc" },
1731 };
1732
1733 static const struct iw_handler_def zd1201_iw_handlers = {
1734         .num_standard           = sizeof(zd1201_iw_handler)/sizeof(iw_handler),
1735         .num_private            = sizeof(zd1201_private_handler)/sizeof(iw_handler),
1736         .num_private_args       = sizeof(zd1201_private_args)/sizeof(struct iw_priv_args),
1737         .standard               = (iw_handler *)zd1201_iw_handler,
1738         .private                = (iw_handler *)zd1201_private_handler,
1739         .private_args           = (struct iw_priv_args *) zd1201_private_args,
1740 };
1741
1742 static int zd1201_probe(struct usb_interface *interface,
1743                         const struct usb_device_id *id)
1744 {
1745         struct zd1201 *zd;
1746         struct usb_device *usb;
1747         int i, err;
1748         short porttype;
1749         char buf[IW_ESSID_MAX_SIZE+2];
1750
1751         usb = interface_to_usbdev(interface);
1752
1753         zd = kmalloc(sizeof(struct zd1201), GFP_KERNEL);
1754         if (!zd) {
1755                 return -ENOMEM;
1756         }
1757         memset(zd, 0, sizeof(struct zd1201));
1758         zd->ap = ap;
1759         zd->usb = usb;
1760         zd->removed = 0;
1761         init_waitqueue_head(&zd->rxdataq);
1762         INIT_HLIST_HEAD(&zd->fraglist);
1763         
1764         err = zd1201_fw_upload(usb, zd->ap);
1765         if (err) {
1766                 dev_err(&usb->dev, "zd1201 firmware upload failed: %d\n", err);
1767                 goto err_zd;
1768         }
1769         
1770         zd->endp_in = 1;
1771         zd->endp_out = 1;
1772         zd->endp_out2 = 2;
1773         zd->rx_urb = usb_alloc_urb(0, GFP_KERNEL);
1774         zd->tx_urb = usb_alloc_urb(0, GFP_KERNEL);
1775         if (!zd->rx_urb || !zd->tx_urb)
1776                 goto err_zd;
1777
1778         for(i = 0; i<100; i++)
1779                 udelay(1000);
1780
1781         err = zd1201_drvr_start(zd);
1782         if (err)
1783                 goto err_zd;
1784
1785         err = zd1201_setconfig16(zd, ZD1201_RID_CNFMAXDATALEN, 2312);
1786         if (err)
1787                 goto err_start;
1788
1789         err = zd1201_setconfig16(zd, ZD1201_RID_TXRATECNTL,
1790             ZD1201_RATEB1 | ZD1201_RATEB2 | ZD1201_RATEB5 | ZD1201_RATEB11);
1791         if (err)
1792                 goto err_start;
1793
1794         zd->dev = alloc_etherdev(0);
1795         if (!zd->dev)
1796                 goto err_start;
1797
1798         zd->dev->priv = zd;
1799         zd->dev->open = zd1201_net_open;
1800         zd->dev->stop = zd1201_net_stop;
1801         zd->dev->get_stats = zd1201_get_stats;
1802         zd->dev->get_wireless_stats = zd1201_get_wireless_stats;
1803         zd->dev->wireless_handlers =
1804             (struct iw_handler_def *)&zd1201_iw_handlers;
1805         zd->dev->hard_start_xmit = zd1201_hard_start_xmit;
1806         zd->dev->watchdog_timeo = ZD1201_TX_TIMEOUT;
1807         zd->dev->tx_timeout = zd1201_tx_timeout;
1808         zd->dev->set_multicast_list = zd1201_set_multicast;
1809         zd->dev->set_mac_address = zd1201_set_mac_address;
1810         strcpy(zd->dev->name, "wlan%d");
1811
1812         err = zd1201_getconfig(zd, ZD1201_RID_CNFOWNMACADDR, 
1813             zd->dev->dev_addr, zd->dev->addr_len);
1814         if (err)
1815                 goto err_net;
1816
1817         /* Set wildcard essid to match zd->essid */
1818         *(__le16 *)buf = cpu_to_le16(0);
1819         err = zd1201_setconfig(zd, ZD1201_RID_CNFDESIREDSSID, buf,
1820             IW_ESSID_MAX_SIZE+2, 1);
1821         if (err)
1822                 goto err_net;
1823
1824         if (zd->ap)
1825                 porttype = ZD1201_PORTTYPE_AP;
1826         else
1827                 porttype = ZD1201_PORTTYPE_BSS;
1828         err = zd1201_setconfig16(zd, ZD1201_RID_CNFPORTTYPE, porttype);
1829         if (err)
1830                 goto err_net;
1831
1832         err = register_netdev(zd->dev);
1833         if (err)
1834                 goto err_net;
1835         dev_info(&usb->dev, "%s: ZD1201 USB Wireless interface\n",
1836             zd->dev->name);
1837         
1838         usb_set_intfdata(interface, zd);
1839         return 0;
1840
1841 err_net:
1842         free_netdev(zd->dev);
1843 err_start:
1844         /* Leave the device in reset state */
1845         zd1201_docmd(zd, ZD1201_CMDCODE_INIT, 0, 0, 0);
1846 err_zd:
1847         if (zd->tx_urb)
1848                 usb_free_urb(zd->tx_urb);
1849         if (zd->rx_urb)
1850                 usb_free_urb(zd->rx_urb);
1851         kfree(zd);
1852         return err;
1853 }
1854
1855 static void zd1201_disconnect(struct usb_interface *interface)
1856 {
1857         struct zd1201 *zd=(struct zd1201 *)usb_get_intfdata(interface);
1858         struct hlist_node *node, *node2;
1859         struct zd1201_frag *frag;
1860
1861         if (!zd)
1862                 return;
1863         usb_set_intfdata(interface, NULL);
1864         if (zd->dev) {
1865                 unregister_netdev(zd->dev);
1866                 free_netdev(zd->dev);
1867         }
1868
1869         hlist_for_each_entry_safe(frag, node, node2, &zd->fraglist, fnode) {
1870                 hlist_del_init(&frag->fnode);
1871                 kfree_skb(frag->skb);
1872                 kfree(frag);
1873         }
1874
1875         if (zd->tx_urb) {
1876                 usb_kill_urb(zd->tx_urb);
1877                 usb_free_urb(zd->tx_urb);
1878         }
1879         if (zd->rx_urb) {
1880                 usb_kill_urb(zd->rx_urb);
1881                 usb_free_urb(zd->rx_urb);
1882         }
1883         kfree(zd);
1884 }
1885
1886 #ifdef CONFIG_PM
1887
1888 static int zd1201_suspend(struct usb_interface *interface,
1889                            pm_message_t message)
1890 {
1891         struct zd1201 *zd = usb_get_intfdata(interface);
1892
1893         netif_device_detach(zd->dev);
1894
1895         zd->was_enabled = zd->mac_enabled;
1896
1897         if (zd->was_enabled)
1898                 return zd1201_disable(zd);
1899         else
1900                 return 0;
1901 }
1902
1903 static int zd1201_resume(struct usb_interface *interface)
1904 {
1905         struct zd1201 *zd = usb_get_intfdata(interface);
1906
1907         if (!zd || !zd->dev)
1908                 return -ENODEV;
1909
1910         netif_device_attach(zd->dev);
1911
1912         if (zd->was_enabled)
1913                 return zd1201_enable(zd);
1914         else
1915                 return 0;
1916 }
1917
1918 #else
1919
1920 #define zd1201_suspend NULL
1921 #define zd1201_resume  NULL
1922
1923 #endif
1924
1925 static struct usb_driver zd1201_usb = {
1926         .owner = THIS_MODULE,
1927         .name = "zd1201",
1928         .probe = zd1201_probe,
1929         .disconnect = zd1201_disconnect,
1930         .id_table = zd1201_table,
1931         .suspend = zd1201_suspend,
1932         .resume = zd1201_resume,
1933 };
1934
1935 static int __init zd1201_init(void)
1936 {
1937         return usb_register(&zd1201_usb);
1938 }
1939
1940 static void __exit zd1201_cleanup(void)
1941 {
1942         usb_deregister(&zd1201_usb);
1943 }
1944
1945 module_init(zd1201_init);
1946 module_exit(zd1201_cleanup);