Merge git://git.kernel.org/pub/scm/linux/kernel/git/gregkh/staging-2.6
[linux-2.6.git] / drivers / usb / gadget / printer.c
1 /*
2  * printer.c -- Printer gadget driver
3  *
4  * Copyright (C) 2003-2005 David Brownell
5  * Copyright (C) 2006 Craig W. Nadler
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 2 of the License, or
10  * (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, write to the Free Software
19  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
20  */
21
22 #include <linux/module.h>
23 #include <linux/kernel.h>
24 #include <linux/delay.h>
25 #include <linux/ioport.h>
26 #include <linux/sched.h>
27 #include <linux/slab.h>
28 #include <linux/smp_lock.h>
29 #include <linux/errno.h>
30 #include <linux/init.h>
31 #include <linux/timer.h>
32 #include <linux/list.h>
33 #include <linux/interrupt.h>
34 #include <linux/utsname.h>
35 #include <linux/device.h>
36 #include <linux/moduleparam.h>
37 #include <linux/fs.h>
38 #include <linux/poll.h>
39 #include <linux/types.h>
40 #include <linux/ctype.h>
41 #include <linux/cdev.h>
42
43 #include <asm/byteorder.h>
44 #include <linux/io.h>
45 #include <linux/irq.h>
46 #include <asm/system.h>
47 #include <linux/uaccess.h>
48 #include <asm/unaligned.h>
49
50 #include <linux/usb/ch9.h>
51 #include <linux/usb/gadget.h>
52 #include <linux/usb/g_printer.h>
53
54 #include "gadget_chips.h"
55
56
57 /*
58  * Kbuild is not very cooperative with respect to linking separately
59  * compiled library objects into one module.  So for now we won't use
60  * separate compilation ... ensuring init/exit sections work to shrink
61  * the runtime footprint, and giving us at least some parts of what
62  * a "gcc --combine ... part1.c part2.c part3.c ... " build would.
63  */
64 #include "usbstring.c"
65 #include "config.c"
66 #include "epautoconf.c"
67
68 /*-------------------------------------------------------------------------*/
69
70 #define DRIVER_DESC             "Printer Gadget"
71 #define DRIVER_VERSION          "2007 OCT 06"
72
73 static const char shortname [] = "printer";
74 static const char driver_desc [] = DRIVER_DESC;
75
76 static dev_t g_printer_devno;
77
78 static struct class *usb_gadget_class;
79
80 /*-------------------------------------------------------------------------*/
81
82 struct printer_dev {
83         spinlock_t              lock;           /* lock this structure */
84         /* lock buffer lists during read/write calls */
85         struct mutex            lock_printer_io;
86         struct usb_gadget       *gadget;
87         struct usb_request      *req;           /* for control responses */
88         u8                      config;
89         s8                      interface;
90         struct usb_ep           *in_ep, *out_ep;
91         const struct usb_endpoint_descriptor
92                                 *in, *out;
93         struct list_head        rx_reqs;        /* List of free RX structs */
94         struct list_head        rx_reqs_active; /* List of Active RX xfers */
95         struct list_head        rx_buffers;     /* List of completed xfers */
96         /* wait until there is data to be read. */
97         wait_queue_head_t       rx_wait;
98         struct list_head        tx_reqs;        /* List of free TX structs */
99         struct list_head        tx_reqs_active; /* List of Active TX xfers */
100         /* Wait until there are write buffers available to use. */
101         wait_queue_head_t       tx_wait;
102         /* Wait until all write buffers have been sent. */
103         wait_queue_head_t       tx_flush_wait;
104         struct usb_request      *current_rx_req;
105         size_t                  current_rx_bytes;
106         u8                      *current_rx_buf;
107         u8                      printer_status;
108         u8                      reset_printer;
109         struct cdev             printer_cdev;
110         struct device           *pdev;
111         u8                      printer_cdev_open;
112         wait_queue_head_t       wait;
113 };
114
115 static struct printer_dev usb_printer_gadget;
116
117 /*-------------------------------------------------------------------------*/
118
119 /* DO NOT REUSE THESE IDs with a protocol-incompatible driver!!  Ever!!
120  * Instead:  allocate your own, using normal USB-IF procedures.
121  */
122
123 /* Thanks to NetChip Technologies for donating this product ID.
124  */
125 #define PRINTER_VENDOR_NUM      0x0525          /* NetChip */
126 #define PRINTER_PRODUCT_NUM     0xa4a8          /* Linux-USB Printer Gadget */
127
128 /* Some systems will want different product identifers published in the
129  * device descriptor, either numbers or strings or both.  These string
130  * parameters are in UTF-8 (superset of ASCII's 7 bit characters).
131  */
132
133 static ushort __initdata idVendor;
134 module_param(idVendor, ushort, S_IRUGO);
135 MODULE_PARM_DESC(idVendor, "USB Vendor ID");
136
137 static ushort __initdata idProduct;
138 module_param(idProduct, ushort, S_IRUGO);
139 MODULE_PARM_DESC(idProduct, "USB Product ID");
140
141 static ushort __initdata bcdDevice;
142 module_param(bcdDevice, ushort, S_IRUGO);
143 MODULE_PARM_DESC(bcdDevice, "USB Device version (BCD)");
144
145 static char *__initdata iManufacturer;
146 module_param(iManufacturer, charp, S_IRUGO);
147 MODULE_PARM_DESC(iManufacturer, "USB Manufacturer string");
148
149 static char *__initdata iProduct;
150 module_param(iProduct, charp, S_IRUGO);
151 MODULE_PARM_DESC(iProduct, "USB Product string");
152
153 static char *__initdata iSerialNum;
154 module_param(iSerialNum, charp, S_IRUGO);
155 MODULE_PARM_DESC(iSerialNum, "1");
156
157 static char *__initdata iPNPstring;
158 module_param(iPNPstring, charp, S_IRUGO);
159 MODULE_PARM_DESC(iPNPstring, "MFG:linux;MDL:g_printer;CLS:PRINTER;SN:1;");
160
161 /* Number of requests to allocate per endpoint, not used for ep0. */
162 static unsigned qlen = 10;
163 module_param(qlen, uint, S_IRUGO|S_IWUSR);
164
165 #define QLEN    qlen
166
167 #ifdef CONFIG_USB_GADGET_DUALSPEED
168 #define DEVSPEED        USB_SPEED_HIGH
169 #else   /* full speed (low speed doesn't do bulk) */
170 #define DEVSPEED        USB_SPEED_FULL
171 #endif
172
173 /*-------------------------------------------------------------------------*/
174
175 #define xprintk(d, level, fmt, args...) \
176         printk(level "%s: " fmt, DRIVER_DESC, ## args)
177
178 #ifdef DEBUG
179 #define DBG(dev, fmt, args...) \
180         xprintk(dev, KERN_DEBUG, fmt, ## args)
181 #else
182 #define DBG(dev, fmt, args...) \
183         do { } while (0)
184 #endif /* DEBUG */
185
186 #ifdef VERBOSE
187 #define VDBG(dev, fmt, args...) \
188         xprintk(dev, KERN_DEBUG, fmt, ## args)
189 #else
190 #define VDBG(dev, fmt, args...) \
191         do { } while (0)
192 #endif /* VERBOSE */
193
194 #define ERROR(dev, fmt, args...) \
195         xprintk(dev, KERN_ERR, fmt, ## args)
196 #define WARNING(dev, fmt, args...) \
197         xprintk(dev, KERN_WARNING, fmt, ## args)
198 #define INFO(dev, fmt, args...) \
199         xprintk(dev, KERN_INFO, fmt, ## args)
200
201 /*-------------------------------------------------------------------------*/
202
203 /* USB DRIVER HOOKUP (to the hardware driver, below us), mostly
204  * ep0 implementation:  descriptors, config management, setup().
205  * also optional class-specific notification interrupt transfer.
206  */
207
208 /*
209  * DESCRIPTORS ... most are static, but strings and (full) configuration
210  * descriptors are built on demand.
211  */
212
213 #define STRING_MANUFACTURER             1
214 #define STRING_PRODUCT                  2
215 #define STRING_SERIALNUM                3
216
217 /* holds our biggest descriptor */
218 #define USB_DESC_BUFSIZE                256
219 #define USB_BUFSIZE                     8192
220
221 /* This device advertises one configuration. */
222 #define DEV_CONFIG_VALUE                1
223 #define PRINTER_INTERFACE               0
224
225 static struct usb_device_descriptor device_desc = {
226         .bLength =              sizeof device_desc,
227         .bDescriptorType =      USB_DT_DEVICE,
228         .bcdUSB =               cpu_to_le16(0x0200),
229         .bDeviceClass =         USB_CLASS_PER_INTERFACE,
230         .bDeviceSubClass =      0,
231         .bDeviceProtocol =      0,
232         .idVendor =             cpu_to_le16(PRINTER_VENDOR_NUM),
233         .idProduct =            cpu_to_le16(PRINTER_PRODUCT_NUM),
234         .iManufacturer =        STRING_MANUFACTURER,
235         .iProduct =             STRING_PRODUCT,
236         .iSerialNumber =        STRING_SERIALNUM,
237         .bNumConfigurations =   1
238 };
239
240 static struct usb_otg_descriptor otg_desc = {
241         .bLength =              sizeof otg_desc,
242         .bDescriptorType =      USB_DT_OTG,
243         .bmAttributes =         USB_OTG_SRP
244 };
245
246 static struct usb_config_descriptor config_desc = {
247         .bLength =              sizeof config_desc,
248         .bDescriptorType =      USB_DT_CONFIG,
249
250         /* compute wTotalLength on the fly */
251         .bNumInterfaces =       1,
252         .bConfigurationValue =  DEV_CONFIG_VALUE,
253         .iConfiguration =       0,
254         .bmAttributes =         USB_CONFIG_ATT_ONE | USB_CONFIG_ATT_SELFPOWER,
255         .bMaxPower =            CONFIG_USB_GADGET_VBUS_DRAW / 2,
256 };
257
258 static struct usb_interface_descriptor intf_desc = {
259         .bLength =              sizeof intf_desc,
260         .bDescriptorType =      USB_DT_INTERFACE,
261         .bInterfaceNumber =     PRINTER_INTERFACE,
262         .bNumEndpoints =        2,
263         .bInterfaceClass =      USB_CLASS_PRINTER,
264         .bInterfaceSubClass =   1,      /* Printer Sub-Class */
265         .bInterfaceProtocol =   2,      /* Bi-Directional */
266         .iInterface =           0
267 };
268
269 static struct usb_endpoint_descriptor fs_ep_in_desc = {
270         .bLength =              USB_DT_ENDPOINT_SIZE,
271         .bDescriptorType =      USB_DT_ENDPOINT,
272         .bEndpointAddress =     USB_DIR_IN,
273         .bmAttributes =         USB_ENDPOINT_XFER_BULK
274 };
275
276 static struct usb_endpoint_descriptor fs_ep_out_desc = {
277         .bLength =              USB_DT_ENDPOINT_SIZE,
278         .bDescriptorType =      USB_DT_ENDPOINT,
279         .bEndpointAddress =     USB_DIR_OUT,
280         .bmAttributes =         USB_ENDPOINT_XFER_BULK
281 };
282
283 static const struct usb_descriptor_header *fs_printer_function [11] = {
284         (struct usb_descriptor_header *) &otg_desc,
285         (struct usb_descriptor_header *) &intf_desc,
286         (struct usb_descriptor_header *) &fs_ep_in_desc,
287         (struct usb_descriptor_header *) &fs_ep_out_desc,
288         NULL
289 };
290
291 #ifdef  CONFIG_USB_GADGET_DUALSPEED
292
293 /*
294  * usb 2.0 devices need to expose both high speed and full speed
295  * descriptors, unless they only run at full speed.
296  */
297
298 static struct usb_endpoint_descriptor hs_ep_in_desc = {
299         .bLength =              USB_DT_ENDPOINT_SIZE,
300         .bDescriptorType =      USB_DT_ENDPOINT,
301         .bmAttributes =         USB_ENDPOINT_XFER_BULK,
302         .wMaxPacketSize =       cpu_to_le16(512)
303 };
304
305 static struct usb_endpoint_descriptor hs_ep_out_desc = {
306         .bLength =              USB_DT_ENDPOINT_SIZE,
307         .bDescriptorType =      USB_DT_ENDPOINT,
308         .bmAttributes =         USB_ENDPOINT_XFER_BULK,
309         .wMaxPacketSize =       cpu_to_le16(512)
310 };
311
312 static struct usb_qualifier_descriptor dev_qualifier = {
313         .bLength =              sizeof dev_qualifier,
314         .bDescriptorType =      USB_DT_DEVICE_QUALIFIER,
315         .bcdUSB =               cpu_to_le16(0x0200),
316         .bDeviceClass =         USB_CLASS_PRINTER,
317         .bNumConfigurations =   1
318 };
319
320 static const struct usb_descriptor_header *hs_printer_function [11] = {
321         (struct usb_descriptor_header *) &otg_desc,
322         (struct usb_descriptor_header *) &intf_desc,
323         (struct usb_descriptor_header *) &hs_ep_in_desc,
324         (struct usb_descriptor_header *) &hs_ep_out_desc,
325         NULL
326 };
327
328 /* maxpacket and other transfer characteristics vary by speed. */
329 #define ep_desc(g, hs, fs) (((g)->speed == USB_SPEED_HIGH)?(hs):(fs))
330
331 #else
332
333 /* if there's no high speed support, maxpacket doesn't change. */
334 #define ep_desc(g, hs, fs) (((void)(g)), (fs))
335
336 #endif  /* !CONFIG_USB_GADGET_DUALSPEED */
337
338 /*-------------------------------------------------------------------------*/
339
340 /* descriptors that are built on-demand */
341
342 static char                             manufacturer [50];
343 static char                             product_desc [40] = DRIVER_DESC;
344 static char                             serial_num [40] = "1";
345 static char                             pnp_string [1024] =
346         "XXMFG:linux;MDL:g_printer;CLS:PRINTER;SN:1;";
347
348 /* static strings, in UTF-8 */
349 static struct usb_string                strings [] = {
350         { STRING_MANUFACTURER,  manufacturer, },
351         { STRING_PRODUCT,       product_desc, },
352         { STRING_SERIALNUM,     serial_num, },
353         {  }            /* end of list */
354 };
355
356 static struct usb_gadget_strings        stringtab = {
357         .language       = 0x0409,       /* en-us */
358         .strings        = strings,
359 };
360
361 /*-------------------------------------------------------------------------*/
362
363 static struct usb_request *
364 printer_req_alloc(struct usb_ep *ep, unsigned len, gfp_t gfp_flags)
365 {
366         struct usb_request      *req;
367
368         req = usb_ep_alloc_request(ep, gfp_flags);
369
370         if (req != NULL) {
371                 req->length = len;
372                 req->buf = kmalloc(len, gfp_flags);
373                 if (req->buf == NULL) {
374                         usb_ep_free_request(ep, req);
375                         return NULL;
376                 }
377         }
378
379         return req;
380 }
381
382 static void
383 printer_req_free(struct usb_ep *ep, struct usb_request *req)
384 {
385         if (ep != NULL && req != NULL) {
386                 kfree(req->buf);
387                 usb_ep_free_request(ep, req);
388         }
389 }
390
391 /*-------------------------------------------------------------------------*/
392
393 static void rx_complete(struct usb_ep *ep, struct usb_request *req)
394 {
395         struct printer_dev      *dev = ep->driver_data;
396         int                     status = req->status;
397         unsigned long           flags;
398
399         spin_lock_irqsave(&dev->lock, flags);
400
401         list_del_init(&req->list);      /* Remode from Active List */
402
403         switch (status) {
404
405         /* normal completion */
406         case 0:
407                 if (req->actual > 0) {
408                         list_add_tail(&req->list, &dev->rx_buffers);
409                         DBG(dev, "G_Printer : rx length %d\n", req->actual);
410                 } else {
411                         list_add(&req->list, &dev->rx_reqs);
412                 }
413                 break;
414
415         /* software-driven interface shutdown */
416         case -ECONNRESET:               /* unlink */
417         case -ESHUTDOWN:                /* disconnect etc */
418                 VDBG(dev, "rx shutdown, code %d\n", status);
419                 list_add(&req->list, &dev->rx_reqs);
420                 break;
421
422         /* for hardware automagic (such as pxa) */
423         case -ECONNABORTED:             /* endpoint reset */
424                 DBG(dev, "rx %s reset\n", ep->name);
425                 list_add(&req->list, &dev->rx_reqs);
426                 break;
427
428         /* data overrun */
429         case -EOVERFLOW:
430                 /* FALLTHROUGH */
431
432         default:
433                 DBG(dev, "rx status %d\n", status);
434                 list_add(&req->list, &dev->rx_reqs);
435                 break;
436         }
437
438         wake_up_interruptible(&dev->rx_wait);
439         spin_unlock_irqrestore(&dev->lock, flags);
440 }
441
442 static void tx_complete(struct usb_ep *ep, struct usb_request *req)
443 {
444         struct printer_dev      *dev = ep->driver_data;
445
446         switch (req->status) {
447         default:
448                 VDBG(dev, "tx err %d\n", req->status);
449                 /* FALLTHROUGH */
450         case -ECONNRESET:               /* unlink */
451         case -ESHUTDOWN:                /* disconnect etc */
452                 break;
453         case 0:
454                 break;
455         }
456
457         spin_lock(&dev->lock);
458         /* Take the request struct off the active list and put it on the
459          * free list.
460          */
461         list_del_init(&req->list);
462         list_add(&req->list, &dev->tx_reqs);
463         wake_up_interruptible(&dev->tx_wait);
464         if (likely(list_empty(&dev->tx_reqs_active)))
465                 wake_up_interruptible(&dev->tx_flush_wait);
466
467         spin_unlock(&dev->lock);
468 }
469
470 /*-------------------------------------------------------------------------*/
471
472 static int
473 printer_open(struct inode *inode, struct file *fd)
474 {
475         struct printer_dev      *dev;
476         unsigned long           flags;
477         int                     ret = -EBUSY;
478
479         lock_kernel();
480         dev = container_of(inode->i_cdev, struct printer_dev, printer_cdev);
481
482         spin_lock_irqsave(&dev->lock, flags);
483
484         if (!dev->printer_cdev_open) {
485                 dev->printer_cdev_open = 1;
486                 fd->private_data = dev;
487                 ret = 0;
488                 /* Change the printer status to show that it's on-line. */
489                 dev->printer_status |= PRINTER_SELECTED;
490         }
491
492         spin_unlock_irqrestore(&dev->lock, flags);
493
494         DBG(dev, "printer_open returned %x\n", ret);
495         unlock_kernel();
496         return ret;
497 }
498
499 static int
500 printer_close(struct inode *inode, struct file *fd)
501 {
502         struct printer_dev      *dev = fd->private_data;
503         unsigned long           flags;
504
505         spin_lock_irqsave(&dev->lock, flags);
506         dev->printer_cdev_open = 0;
507         fd->private_data = NULL;
508         /* Change printer status to show that the printer is off-line. */
509         dev->printer_status &= ~PRINTER_SELECTED;
510         spin_unlock_irqrestore(&dev->lock, flags);
511
512         DBG(dev, "printer_close\n");
513
514         return 0;
515 }
516
517 /* This function must be called with interrupts turned off. */
518 static void
519 setup_rx_reqs(struct printer_dev *dev)
520 {
521         struct usb_request              *req;
522
523         while (likely(!list_empty(&dev->rx_reqs))) {
524                 int error;
525
526                 req = container_of(dev->rx_reqs.next,
527                                 struct usb_request, list);
528                 list_del_init(&req->list);
529
530                 /* The USB Host sends us whatever amount of data it wants to
531                  * so we always set the length field to the full USB_BUFSIZE.
532                  * If the amount of data is more than the read() caller asked
533                  * for it will be stored in the request buffer until it is
534                  * asked for by read().
535                  */
536                 req->length = USB_BUFSIZE;
537                 req->complete = rx_complete;
538
539                 error = usb_ep_queue(dev->out_ep, req, GFP_ATOMIC);
540                 if (error) {
541                         DBG(dev, "rx submit --> %d\n", error);
542                         list_add(&req->list, &dev->rx_reqs);
543                         break;
544                 } else {
545                         list_add(&req->list, &dev->rx_reqs_active);
546                 }
547         }
548 }
549
550 static ssize_t
551 printer_read(struct file *fd, char __user *buf, size_t len, loff_t *ptr)
552 {
553         struct printer_dev              *dev = fd->private_data;
554         unsigned long                   flags;
555         size_t                          size;
556         size_t                          bytes_copied;
557         struct usb_request              *req;
558         /* This is a pointer to the current USB rx request. */
559         struct usb_request              *current_rx_req;
560         /* This is the number of bytes in the current rx buffer. */
561         size_t                          current_rx_bytes;
562         /* This is a pointer to the current rx buffer. */
563         u8                              *current_rx_buf;
564
565         if (len == 0)
566                 return -EINVAL;
567
568         DBG(dev, "printer_read trying to read %d bytes\n", (int)len);
569
570         mutex_lock(&dev->lock_printer_io);
571         spin_lock_irqsave(&dev->lock, flags);
572
573         /* We will use this flag later to check if a printer reset happened
574          * after we turn interrupts back on.
575          */
576         dev->reset_printer = 0;
577
578         setup_rx_reqs(dev);
579
580         bytes_copied = 0;
581         current_rx_req = dev->current_rx_req;
582         current_rx_bytes = dev->current_rx_bytes;
583         current_rx_buf = dev->current_rx_buf;
584         dev->current_rx_req = NULL;
585         dev->current_rx_bytes = 0;
586         dev->current_rx_buf = NULL;
587
588         /* Check if there is any data in the read buffers. Please note that
589          * current_rx_bytes is the number of bytes in the current rx buffer.
590          * If it is zero then check if there are any other rx_buffers that
591          * are on the completed list. We are only out of data if all rx
592          * buffers are empty.
593          */
594         if ((current_rx_bytes == 0) &&
595                         (likely(list_empty(&dev->rx_buffers)))) {
596                 /* Turn interrupts back on before sleeping. */
597                 spin_unlock_irqrestore(&dev->lock, flags);
598
599                 /*
600                  * If no data is available check if this is a NON-Blocking
601                  * call or not.
602                  */
603                 if (fd->f_flags & (O_NONBLOCK|O_NDELAY)) {
604                         mutex_unlock(&dev->lock_printer_io);
605                         return -EAGAIN;
606                 }
607
608                 /* Sleep until data is available */
609                 wait_event_interruptible(dev->rx_wait,
610                                 (likely(!list_empty(&dev->rx_buffers))));
611                 spin_lock_irqsave(&dev->lock, flags);
612         }
613
614         /* We have data to return then copy it to the caller's buffer.*/
615         while ((current_rx_bytes || likely(!list_empty(&dev->rx_buffers)))
616                         && len) {
617                 if (current_rx_bytes == 0) {
618                         req = container_of(dev->rx_buffers.next,
619                                         struct usb_request, list);
620                         list_del_init(&req->list);
621
622                         if (req->actual && req->buf) {
623                                 current_rx_req = req;
624                                 current_rx_bytes = req->actual;
625                                 current_rx_buf = req->buf;
626                         } else {
627                                 list_add(&req->list, &dev->rx_reqs);
628                                 continue;
629                         }
630                 }
631
632                 /* Don't leave irqs off while doing memory copies */
633                 spin_unlock_irqrestore(&dev->lock, flags);
634
635                 if (len > current_rx_bytes)
636                         size = current_rx_bytes;
637                 else
638                         size = len;
639
640                 size -= copy_to_user(buf, current_rx_buf, size);
641                 bytes_copied += size;
642                 len -= size;
643                 buf += size;
644
645                 spin_lock_irqsave(&dev->lock, flags);
646
647                 /* We've disconnected or reset so return. */
648                 if (dev->reset_printer) {
649                         list_add(&current_rx_req->list, &dev->rx_reqs);
650                         spin_unlock_irqrestore(&dev->lock, flags);
651                         mutex_unlock(&dev->lock_printer_io);
652                         return -EAGAIN;
653                 }
654
655                 /* If we not returning all the data left in this RX request
656                  * buffer then adjust the amount of data left in the buffer.
657                  * Othewise if we are done with this RX request buffer then
658                  * requeue it to get any incoming data from the USB host.
659                  */
660                 if (size < current_rx_bytes) {
661                         current_rx_bytes -= size;
662                         current_rx_buf += size;
663                 } else {
664                         list_add(&current_rx_req->list, &dev->rx_reqs);
665                         current_rx_bytes = 0;
666                         current_rx_buf = NULL;
667                         current_rx_req = NULL;
668                 }
669         }
670
671         dev->current_rx_req = current_rx_req;
672         dev->current_rx_bytes = current_rx_bytes;
673         dev->current_rx_buf = current_rx_buf;
674
675         spin_unlock_irqrestore(&dev->lock, flags);
676         mutex_unlock(&dev->lock_printer_io);
677
678         DBG(dev, "printer_read returned %d bytes\n", (int)bytes_copied);
679
680         if (bytes_copied)
681                 return bytes_copied;
682         else
683                 return -EAGAIN;
684 }
685
686 static ssize_t
687 printer_write(struct file *fd, const char __user *buf, size_t len, loff_t *ptr)
688 {
689         struct printer_dev      *dev = fd->private_data;
690         unsigned long           flags;
691         size_t                  size;   /* Amount of data in a TX request. */
692         size_t                  bytes_copied = 0;
693         struct usb_request      *req;
694
695         DBG(dev, "printer_write trying to send %d bytes\n", (int)len);
696
697         if (len == 0)
698                 return -EINVAL;
699
700         mutex_lock(&dev->lock_printer_io);
701         spin_lock_irqsave(&dev->lock, flags);
702
703         /* Check if a printer reset happens while we have interrupts on */
704         dev->reset_printer = 0;
705
706         /* Check if there is any available write buffers */
707         if (likely(list_empty(&dev->tx_reqs))) {
708                 /* Turn interrupts back on before sleeping. */
709                 spin_unlock_irqrestore(&dev->lock, flags);
710
711                 /*
712                  * If write buffers are available check if this is
713                  * a NON-Blocking call or not.
714                  */
715                 if (fd->f_flags & (O_NONBLOCK|O_NDELAY)) {
716                         mutex_unlock(&dev->lock_printer_io);
717                         return -EAGAIN;
718                 }
719
720                 /* Sleep until a write buffer is available */
721                 wait_event_interruptible(dev->tx_wait,
722                                 (likely(!list_empty(&dev->tx_reqs))));
723                 spin_lock_irqsave(&dev->lock, flags);
724         }
725
726         while (likely(!list_empty(&dev->tx_reqs)) && len) {
727
728                 if (len > USB_BUFSIZE)
729                         size = USB_BUFSIZE;
730                 else
731                         size = len;
732
733                 req = container_of(dev->tx_reqs.next, struct usb_request,
734                                 list);
735                 list_del_init(&req->list);
736
737                 req->complete = tx_complete;
738                 req->length = size;
739
740                 /* Check if we need to send a zero length packet. */
741                 if (len > size)
742                         /* They will be more TX requests so no yet. */
743                         req->zero = 0;
744                 else
745                         /* If the data amount is not a multple of the
746                          * maxpacket size then send a zero length packet.
747                          */
748                         req->zero = ((len % dev->in_ep->maxpacket) == 0);
749
750                 /* Don't leave irqs off while doing memory copies */
751                 spin_unlock_irqrestore(&dev->lock, flags);
752
753                 if (copy_from_user(req->buf, buf, size)) {
754                         list_add(&req->list, &dev->tx_reqs);
755                         mutex_unlock(&dev->lock_printer_io);
756                         return bytes_copied;
757                 }
758
759                 bytes_copied += size;
760                 len -= size;
761                 buf += size;
762
763                 spin_lock_irqsave(&dev->lock, flags);
764
765                 /* We've disconnected or reset so free the req and buffer */
766                 if (dev->reset_printer) {
767                         list_add(&req->list, &dev->tx_reqs);
768                         spin_unlock_irqrestore(&dev->lock, flags);
769                         mutex_unlock(&dev->lock_printer_io);
770                         return -EAGAIN;
771                 }
772
773                 if (usb_ep_queue(dev->in_ep, req, GFP_ATOMIC)) {
774                         list_add(&req->list, &dev->tx_reqs);
775                         spin_unlock_irqrestore(&dev->lock, flags);
776                         mutex_unlock(&dev->lock_printer_io);
777                         return -EAGAIN;
778                 }
779
780                 list_add(&req->list, &dev->tx_reqs_active);
781
782         }
783
784         spin_unlock_irqrestore(&dev->lock, flags);
785         mutex_unlock(&dev->lock_printer_io);
786
787         DBG(dev, "printer_write sent %d bytes\n", (int)bytes_copied);
788
789         if (bytes_copied) {
790                 return bytes_copied;
791         } else {
792                 return -EAGAIN;
793         }
794 }
795
796 static int
797 printer_fsync(struct file *fd, int datasync)
798 {
799         struct printer_dev      *dev = fd->private_data;
800         unsigned long           flags;
801         int                     tx_list_empty;
802
803         spin_lock_irqsave(&dev->lock, flags);
804         tx_list_empty = (likely(list_empty(&dev->tx_reqs)));
805         spin_unlock_irqrestore(&dev->lock, flags);
806
807         if (!tx_list_empty) {
808                 /* Sleep until all data has been sent */
809                 wait_event_interruptible(dev->tx_flush_wait,
810                                 (likely(list_empty(&dev->tx_reqs_active))));
811         }
812
813         return 0;
814 }
815
816 static unsigned int
817 printer_poll(struct file *fd, poll_table *wait)
818 {
819         struct printer_dev      *dev = fd->private_data;
820         unsigned long           flags;
821         int                     status = 0;
822
823         mutex_lock(&dev->lock_printer_io);
824         spin_lock_irqsave(&dev->lock, flags);
825         setup_rx_reqs(dev);
826         spin_unlock_irqrestore(&dev->lock, flags);
827         mutex_unlock(&dev->lock_printer_io);
828
829         poll_wait(fd, &dev->rx_wait, wait);
830         poll_wait(fd, &dev->tx_wait, wait);
831
832         spin_lock_irqsave(&dev->lock, flags);
833         if (likely(!list_empty(&dev->tx_reqs)))
834                 status |= POLLOUT | POLLWRNORM;
835
836         if (likely(dev->current_rx_bytes) ||
837                         likely(!list_empty(&dev->rx_buffers)))
838                 status |= POLLIN | POLLRDNORM;
839
840         spin_unlock_irqrestore(&dev->lock, flags);
841
842         return status;
843 }
844
845 static long
846 printer_ioctl(struct file *fd, unsigned int code, unsigned long arg)
847 {
848         struct printer_dev      *dev = fd->private_data;
849         unsigned long           flags;
850         int                     status = 0;
851
852         DBG(dev, "printer_ioctl: cmd=0x%4.4x, arg=%lu\n", code, arg);
853
854         /* handle ioctls */
855
856         spin_lock_irqsave(&dev->lock, flags);
857
858         switch (code) {
859         case GADGET_GET_PRINTER_STATUS:
860                 status = (int)dev->printer_status;
861                 break;
862         case GADGET_SET_PRINTER_STATUS:
863                 dev->printer_status = (u8)arg;
864                 break;
865         default:
866                 /* could not handle ioctl */
867                 DBG(dev, "printer_ioctl: ERROR cmd=0x%4.4xis not supported\n",
868                                 code);
869                 status = -ENOTTY;
870         }
871
872         spin_unlock_irqrestore(&dev->lock, flags);
873
874         return status;
875 }
876
877 /* used after endpoint configuration */
878 static const struct file_operations printer_io_operations = {
879         .owner =        THIS_MODULE,
880         .open =         printer_open,
881         .read =         printer_read,
882         .write =        printer_write,
883         .fsync =        printer_fsync,
884         .poll =         printer_poll,
885         .unlocked_ioctl = printer_ioctl,
886         .release =      printer_close
887 };
888
889 /*-------------------------------------------------------------------------*/
890
891 static int
892 set_printer_interface(struct printer_dev *dev)
893 {
894         int                     result = 0;
895
896         dev->in = ep_desc(dev->gadget, &hs_ep_in_desc, &fs_ep_in_desc);
897         dev->in_ep->driver_data = dev;
898
899         dev->out = ep_desc(dev->gadget, &hs_ep_out_desc, &fs_ep_out_desc);
900         dev->out_ep->driver_data = dev;
901
902         result = usb_ep_enable(dev->in_ep, dev->in);
903         if (result != 0) {
904                 DBG(dev, "enable %s --> %d\n", dev->in_ep->name, result);
905                 goto done;
906         }
907
908         result = usb_ep_enable(dev->out_ep, dev->out);
909         if (result != 0) {
910                 DBG(dev, "enable %s --> %d\n", dev->in_ep->name, result);
911                 goto done;
912         }
913
914 done:
915         /* on error, disable any endpoints  */
916         if (result != 0) {
917                 (void) usb_ep_disable(dev->in_ep);
918                 (void) usb_ep_disable(dev->out_ep);
919                 dev->in = NULL;
920                 dev->out = NULL;
921         }
922
923         /* caller is responsible for cleanup on error */
924         return result;
925 }
926
927 static void printer_reset_interface(struct printer_dev *dev)
928 {
929         if (dev->interface < 0)
930                 return;
931
932         DBG(dev, "%s\n", __func__);
933
934         if (dev->in)
935                 usb_ep_disable(dev->in_ep);
936
937         if (dev->out)
938                 usb_ep_disable(dev->out_ep);
939
940         dev->interface = -1;
941 }
942
943 /* change our operational config.  must agree with the code
944  * that returns config descriptors, and altsetting code.
945  */
946 static int
947 printer_set_config(struct printer_dev *dev, unsigned number)
948 {
949         int                     result = 0;
950         struct usb_gadget       *gadget = dev->gadget;
951
952         switch (number) {
953         case DEV_CONFIG_VALUE:
954                 result = 0;
955                 break;
956         default:
957                 result = -EINVAL;
958                 /* FALL THROUGH */
959         case 0:
960                 break;
961         }
962
963         if (result) {
964                 usb_gadget_vbus_draw(dev->gadget,
965                                 dev->gadget->is_otg ? 8 : 100);
966         } else {
967                 char *speed;
968                 unsigned power;
969
970                 power = 2 * config_desc.bMaxPower;
971                 usb_gadget_vbus_draw(dev->gadget, power);
972
973                 switch (gadget->speed) {
974                 case USB_SPEED_FULL:    speed = "full"; break;
975 #ifdef CONFIG_USB_GADGET_DUALSPEED
976                 case USB_SPEED_HIGH:    speed = "high"; break;
977 #endif
978                 default:                speed = "?"; break;
979                 }
980
981                 dev->config = number;
982                 INFO(dev, "%s speed config #%d: %d mA, %s\n",
983                                 speed, number, power, driver_desc);
984         }
985         return result;
986 }
987
988 static int
989 config_buf(enum usb_device_speed speed, u8 *buf, u8 type, unsigned index,
990                 int is_otg)
991 {
992         int                                     len;
993         const struct usb_descriptor_header      **function;
994 #ifdef CONFIG_USB_GADGET_DUALSPEED
995         int                                     hs = (speed == USB_SPEED_HIGH);
996
997         if (type == USB_DT_OTHER_SPEED_CONFIG)
998                 hs = !hs;
999
1000         if (hs) {
1001                 function = hs_printer_function;
1002         } else {
1003                 function = fs_printer_function;
1004         }
1005 #else
1006         function = fs_printer_function;
1007 #endif
1008
1009         if (index >= device_desc.bNumConfigurations)
1010                 return -EINVAL;
1011
1012         /* for now, don't advertise srp-only devices */
1013         if (!is_otg)
1014                 function++;
1015
1016         len = usb_gadget_config_buf(&config_desc, buf, USB_DESC_BUFSIZE,
1017                         function);
1018         if (len < 0)
1019                 return len;
1020         ((struct usb_config_descriptor *) buf)->bDescriptorType = type;
1021         return len;
1022 }
1023
1024 /* Change our operational Interface. */
1025 static int
1026 set_interface(struct printer_dev *dev, unsigned number)
1027 {
1028         int                     result = 0;
1029
1030         /* Free the current interface */
1031         switch (dev->interface) {
1032         case PRINTER_INTERFACE:
1033                 printer_reset_interface(dev);
1034                 break;
1035         }
1036
1037         switch (number) {
1038         case PRINTER_INTERFACE:
1039                 result = set_printer_interface(dev);
1040                 if (result) {
1041                         printer_reset_interface(dev);
1042                 } else {
1043                         dev->interface = PRINTER_INTERFACE;
1044                 }
1045                 break;
1046         default:
1047                 result = -EINVAL;
1048                 /* FALL THROUGH */
1049         }
1050
1051         if (!result)
1052                 INFO(dev, "Using interface %x\n", number);
1053
1054         return result;
1055 }
1056
1057 static void printer_setup_complete(struct usb_ep *ep, struct usb_request *req)
1058 {
1059         if (req->status || req->actual != req->length)
1060                 DBG((struct printer_dev *) ep->driver_data,
1061                                 "setup complete --> %d, %d/%d\n",
1062                                 req->status, req->actual, req->length);
1063 }
1064
1065 static void printer_soft_reset(struct printer_dev *dev)
1066 {
1067         struct usb_request      *req;
1068
1069         INFO(dev, "Received Printer Reset Request\n");
1070
1071         if (usb_ep_disable(dev->in_ep))
1072                 DBG(dev, "Failed to disable USB in_ep\n");
1073         if (usb_ep_disable(dev->out_ep))
1074                 DBG(dev, "Failed to disable USB out_ep\n");
1075
1076         if (dev->current_rx_req != NULL) {
1077                 list_add(&dev->current_rx_req->list, &dev->rx_reqs);
1078                 dev->current_rx_req = NULL;
1079         }
1080         dev->current_rx_bytes = 0;
1081         dev->current_rx_buf = NULL;
1082         dev->reset_printer = 1;
1083
1084         while (likely(!(list_empty(&dev->rx_buffers)))) {
1085                 req = container_of(dev->rx_buffers.next, struct usb_request,
1086                                 list);
1087                 list_del_init(&req->list);
1088                 list_add(&req->list, &dev->rx_reqs);
1089         }
1090
1091         while (likely(!(list_empty(&dev->rx_reqs_active)))) {
1092                 req = container_of(dev->rx_buffers.next, struct usb_request,
1093                                 list);
1094                 list_del_init(&req->list);
1095                 list_add(&req->list, &dev->rx_reqs);
1096         }
1097
1098         while (likely(!(list_empty(&dev->tx_reqs_active)))) {
1099                 req = container_of(dev->tx_reqs_active.next,
1100                                 struct usb_request, list);
1101                 list_del_init(&req->list);
1102                 list_add(&req->list, &dev->tx_reqs);
1103         }
1104
1105         if (usb_ep_enable(dev->in_ep, dev->in))
1106                 DBG(dev, "Failed to enable USB in_ep\n");
1107         if (usb_ep_enable(dev->out_ep, dev->out))
1108                 DBG(dev, "Failed to enable USB out_ep\n");
1109
1110         wake_up_interruptible(&dev->rx_wait);
1111         wake_up_interruptible(&dev->tx_wait);
1112         wake_up_interruptible(&dev->tx_flush_wait);
1113 }
1114
1115 /*-------------------------------------------------------------------------*/
1116
1117 /*
1118  * The setup() callback implements all the ep0 functionality that's not
1119  * handled lower down.
1120  */
1121 static int
1122 printer_setup(struct usb_gadget *gadget, const struct usb_ctrlrequest *ctrl)
1123 {
1124         struct printer_dev      *dev = get_gadget_data(gadget);
1125         struct usb_request      *req = dev->req;
1126         int                     value = -EOPNOTSUPP;
1127         u16                     wIndex = le16_to_cpu(ctrl->wIndex);
1128         u16                     wValue = le16_to_cpu(ctrl->wValue);
1129         u16                     wLength = le16_to_cpu(ctrl->wLength);
1130
1131         DBG(dev, "ctrl req%02x.%02x v%04x i%04x l%d\n",
1132                 ctrl->bRequestType, ctrl->bRequest, wValue, wIndex, wLength);
1133
1134         req->complete = printer_setup_complete;
1135
1136         switch (ctrl->bRequestType&USB_TYPE_MASK) {
1137
1138         case USB_TYPE_STANDARD:
1139                 switch (ctrl->bRequest) {
1140
1141                 case USB_REQ_GET_DESCRIPTOR:
1142                         if (ctrl->bRequestType != USB_DIR_IN)
1143                                 break;
1144                         switch (wValue >> 8) {
1145
1146                         case USB_DT_DEVICE:
1147                                 value = min(wLength, (u16) sizeof device_desc);
1148                                 memcpy(req->buf, &device_desc, value);
1149                                 break;
1150 #ifdef CONFIG_USB_GADGET_DUALSPEED
1151                         case USB_DT_DEVICE_QUALIFIER:
1152                                 if (!gadget->is_dualspeed)
1153                                         break;
1154                                 value = min(wLength,
1155                                                 (u16) sizeof dev_qualifier);
1156                                 memcpy(req->buf, &dev_qualifier, value);
1157                                 break;
1158
1159                         case USB_DT_OTHER_SPEED_CONFIG:
1160                                 if (!gadget->is_dualspeed)
1161                                         break;
1162                                 /* FALLTHROUGH */
1163 #endif /* CONFIG_USB_GADGET_DUALSPEED */
1164                         case USB_DT_CONFIG:
1165                                 value = config_buf(gadget->speed, req->buf,
1166                                                 wValue >> 8,
1167                                                 wValue & 0xff,
1168                                                 gadget->is_otg);
1169                                 if (value >= 0)
1170                                         value = min(wLength, (u16) value);
1171                                 break;
1172
1173                         case USB_DT_STRING:
1174                                 value = usb_gadget_get_string(&stringtab,
1175                                                 wValue & 0xff, req->buf);
1176                                 if (value >= 0)
1177                                         value = min(wLength, (u16) value);
1178                                 break;
1179                         }
1180                         break;
1181
1182                 case USB_REQ_SET_CONFIGURATION:
1183                         if (ctrl->bRequestType != 0)
1184                                 break;
1185                         if (gadget->a_hnp_support)
1186                                 DBG(dev, "HNP available\n");
1187                         else if (gadget->a_alt_hnp_support)
1188                                 DBG(dev, "HNP needs a different root port\n");
1189                         value = printer_set_config(dev, wValue);
1190                         break;
1191                 case USB_REQ_GET_CONFIGURATION:
1192                         if (ctrl->bRequestType != USB_DIR_IN)
1193                                 break;
1194                         *(u8 *)req->buf = dev->config;
1195                         value = min(wLength, (u16) 1);
1196                         break;
1197
1198                 case USB_REQ_SET_INTERFACE:
1199                         if (ctrl->bRequestType != USB_RECIP_INTERFACE ||
1200                                         !dev->config)
1201                                 break;
1202
1203                         value = set_interface(dev, PRINTER_INTERFACE);
1204                         break;
1205                 case USB_REQ_GET_INTERFACE:
1206                         if (ctrl->bRequestType !=
1207                                         (USB_DIR_IN|USB_RECIP_INTERFACE)
1208                                         || !dev->config)
1209                                 break;
1210
1211                         *(u8 *)req->buf = dev->interface;
1212                         value = min(wLength, (u16) 1);
1213                         break;
1214
1215                 default:
1216                         goto unknown;
1217                 }
1218                 break;
1219
1220         case USB_TYPE_CLASS:
1221                 switch (ctrl->bRequest) {
1222                 case 0: /* Get the IEEE-1284 PNP String */
1223                         /* Only one printer interface is supported. */
1224                         if ((wIndex>>8) != PRINTER_INTERFACE)
1225                                 break;
1226
1227                         value = (pnp_string[0]<<8)|pnp_string[1];
1228                         memcpy(req->buf, pnp_string, value);
1229                         DBG(dev, "1284 PNP String: %x %s\n", value,
1230                                         &pnp_string[2]);
1231                         break;
1232
1233                 case 1: /* Get Port Status */
1234                         /* Only one printer interface is supported. */
1235                         if (wIndex != PRINTER_INTERFACE)
1236                                 break;
1237
1238                         *(u8 *)req->buf = dev->printer_status;
1239                         value = min(wLength, (u16) 1);
1240                         break;
1241
1242                 case 2: /* Soft Reset */
1243                         /* Only one printer interface is supported. */
1244                         if (wIndex != PRINTER_INTERFACE)
1245                                 break;
1246
1247                         printer_soft_reset(dev);
1248
1249                         value = 0;
1250                         break;
1251
1252                 default:
1253                         goto unknown;
1254                 }
1255                 break;
1256
1257         default:
1258 unknown:
1259                 VDBG(dev,
1260                         "unknown ctrl req%02x.%02x v%04x i%04x l%d\n",
1261                         ctrl->bRequestType, ctrl->bRequest,
1262                         wValue, wIndex, wLength);
1263                 break;
1264         }
1265
1266         /* respond with data transfer before status phase? */
1267         if (value >= 0) {
1268                 req->length = value;
1269                 req->zero = value < wLength;
1270                 value = usb_ep_queue(gadget->ep0, req, GFP_ATOMIC);
1271                 if (value < 0) {
1272                         DBG(dev, "ep_queue --> %d\n", value);
1273                         req->status = 0;
1274                         printer_setup_complete(gadget->ep0, req);
1275                 }
1276         }
1277
1278         /* host either stalls (value < 0) or reports success */
1279         return value;
1280 }
1281
1282 static void
1283 printer_disconnect(struct usb_gadget *gadget)
1284 {
1285         struct printer_dev      *dev = get_gadget_data(gadget);
1286         unsigned long           flags;
1287
1288         DBG(dev, "%s\n", __func__);
1289
1290         spin_lock_irqsave(&dev->lock, flags);
1291
1292         printer_reset_interface(dev);
1293
1294         spin_unlock_irqrestore(&dev->lock, flags);
1295 }
1296
1297 static void
1298 printer_unbind(struct usb_gadget *gadget)
1299 {
1300         struct printer_dev      *dev = get_gadget_data(gadget);
1301         struct usb_request      *req;
1302
1303
1304         DBG(dev, "%s\n", __func__);
1305
1306         /* Remove sysfs files */
1307         device_destroy(usb_gadget_class, g_printer_devno);
1308
1309         /* Remove Character Device */
1310         cdev_del(&dev->printer_cdev);
1311
1312         /* we must already have been disconnected ... no i/o may be active */
1313         WARN_ON(!list_empty(&dev->tx_reqs_active));
1314         WARN_ON(!list_empty(&dev->rx_reqs_active));
1315
1316         /* Free all memory for this driver. */
1317         while (!list_empty(&dev->tx_reqs)) {
1318                 req = container_of(dev->tx_reqs.next, struct usb_request,
1319                                 list);
1320                 list_del(&req->list);
1321                 printer_req_free(dev->in_ep, req);
1322         }
1323
1324         if (dev->current_rx_req != NULL)
1325                 printer_req_free(dev->out_ep, dev->current_rx_req);
1326
1327         while (!list_empty(&dev->rx_reqs)) {
1328                 req = container_of(dev->rx_reqs.next,
1329                                 struct usb_request, list);
1330                 list_del(&req->list);
1331                 printer_req_free(dev->out_ep, req);
1332         }
1333
1334         while (!list_empty(&dev->rx_buffers)) {
1335                 req = container_of(dev->rx_buffers.next,
1336                                 struct usb_request, list);
1337                 list_del(&req->list);
1338                 printer_req_free(dev->out_ep, req);
1339         }
1340
1341         if (dev->req) {
1342                 printer_req_free(gadget->ep0, dev->req);
1343                 dev->req = NULL;
1344         }
1345
1346         set_gadget_data(gadget, NULL);
1347 }
1348
1349 static int __init
1350 printer_bind(struct usb_gadget *gadget)
1351 {
1352         struct printer_dev      *dev;
1353         struct usb_ep           *in_ep, *out_ep;
1354         int                     status = -ENOMEM;
1355         int                     gcnum;
1356         size_t                  len;
1357         u32                     i;
1358         struct usb_request      *req;
1359
1360         dev = &usb_printer_gadget;
1361
1362
1363         /* Setup the sysfs files for the printer gadget. */
1364         dev->pdev = device_create(usb_gadget_class, NULL, g_printer_devno,
1365                                   NULL, "g_printer");
1366         if (IS_ERR(dev->pdev)) {
1367                 ERROR(dev, "Failed to create device: g_printer\n");
1368                 goto fail;
1369         }
1370
1371         /*
1372          * Register a character device as an interface to a user mode
1373          * program that handles the printer specific functionality.
1374          */
1375         cdev_init(&dev->printer_cdev, &printer_io_operations);
1376         dev->printer_cdev.owner = THIS_MODULE;
1377         status = cdev_add(&dev->printer_cdev, g_printer_devno, 1);
1378         if (status) {
1379                 ERROR(dev, "Failed to open char device\n");
1380                 goto fail;
1381         }
1382
1383         gcnum = usb_gadget_controller_number(gadget);
1384         if (gcnum >= 0) {
1385                 device_desc.bcdDevice = cpu_to_le16(0x0200 + gcnum);
1386         } else {
1387                 dev_warn(&gadget->dev, "controller '%s' not recognized\n",
1388                         gadget->name);
1389                 /* unrecognized, but safe unless bulk is REALLY quirky */
1390                 device_desc.bcdDevice =
1391                         cpu_to_le16(0xFFFF);
1392         }
1393         snprintf(manufacturer, sizeof(manufacturer), "%s %s with %s",
1394                 init_utsname()->sysname, init_utsname()->release,
1395                 gadget->name);
1396
1397         device_desc.idVendor =
1398                 cpu_to_le16(PRINTER_VENDOR_NUM);
1399         device_desc.idProduct =
1400                 cpu_to_le16(PRINTER_PRODUCT_NUM);
1401
1402         /* support optional vendor/distro customization */
1403         if (idVendor) {
1404                 if (!idProduct) {
1405                         dev_err(&gadget->dev, "idVendor needs idProduct!\n");
1406                         return -ENODEV;
1407                 }
1408                 device_desc.idVendor = cpu_to_le16(idVendor);
1409                 device_desc.idProduct = cpu_to_le16(idProduct);
1410                 if (bcdDevice)
1411                         device_desc.bcdDevice = cpu_to_le16(bcdDevice);
1412         }
1413
1414         if (iManufacturer)
1415                 strlcpy(manufacturer, iManufacturer, sizeof manufacturer);
1416
1417         if (iProduct)
1418                 strlcpy(product_desc, iProduct, sizeof product_desc);
1419
1420         if (iSerialNum)
1421                 strlcpy(serial_num, iSerialNum, sizeof serial_num);
1422
1423         if (iPNPstring)
1424                 strlcpy(&pnp_string[2], iPNPstring, (sizeof pnp_string)-2);
1425
1426         len = strlen(pnp_string);
1427         pnp_string[0] = (len >> 8) & 0xFF;
1428         pnp_string[1] = len & 0xFF;
1429
1430         /* all we really need is bulk IN/OUT */
1431         usb_ep_autoconfig_reset(gadget);
1432         in_ep = usb_ep_autoconfig(gadget, &fs_ep_in_desc);
1433         if (!in_ep) {
1434 autoconf_fail:
1435                 dev_err(&gadget->dev, "can't autoconfigure on %s\n",
1436                         gadget->name);
1437                 return -ENODEV;
1438         }
1439         in_ep->driver_data = in_ep;     /* claim */
1440
1441         out_ep = usb_ep_autoconfig(gadget, &fs_ep_out_desc);
1442         if (!out_ep)
1443                 goto autoconf_fail;
1444         out_ep->driver_data = out_ep;   /* claim */
1445
1446 #ifdef  CONFIG_USB_GADGET_DUALSPEED
1447         /* assumes ep0 uses the same value for both speeds ... */
1448         dev_qualifier.bMaxPacketSize0 = device_desc.bMaxPacketSize0;
1449
1450         /* and that all endpoints are dual-speed */
1451         hs_ep_in_desc.bEndpointAddress = fs_ep_in_desc.bEndpointAddress;
1452         hs_ep_out_desc.bEndpointAddress = fs_ep_out_desc.bEndpointAddress;
1453 #endif  /* DUALSPEED */
1454
1455         device_desc.bMaxPacketSize0 = gadget->ep0->maxpacket;
1456         usb_gadget_set_selfpowered(gadget);
1457
1458         if (gadget->is_otg) {
1459                 otg_desc.bmAttributes |= USB_OTG_HNP,
1460                 config_desc.bmAttributes |= USB_CONFIG_ATT_WAKEUP;
1461         }
1462
1463         spin_lock_init(&dev->lock);
1464         mutex_init(&dev->lock_printer_io);
1465         INIT_LIST_HEAD(&dev->tx_reqs);
1466         INIT_LIST_HEAD(&dev->tx_reqs_active);
1467         INIT_LIST_HEAD(&dev->rx_reqs);
1468         INIT_LIST_HEAD(&dev->rx_reqs_active);
1469         INIT_LIST_HEAD(&dev->rx_buffers);
1470         init_waitqueue_head(&dev->rx_wait);
1471         init_waitqueue_head(&dev->tx_wait);
1472         init_waitqueue_head(&dev->tx_flush_wait);
1473
1474         dev->config = 0;
1475         dev->interface = -1;
1476         dev->printer_cdev_open = 0;
1477         dev->printer_status = PRINTER_NOT_ERROR;
1478         dev->current_rx_req = NULL;
1479         dev->current_rx_bytes = 0;
1480         dev->current_rx_buf = NULL;
1481
1482         dev->in_ep = in_ep;
1483         dev->out_ep = out_ep;
1484
1485         /* preallocate control message data and buffer */
1486         dev->req = printer_req_alloc(gadget->ep0, USB_DESC_BUFSIZE,
1487                         GFP_KERNEL);
1488         if (!dev->req) {
1489                 status = -ENOMEM;
1490                 goto fail;
1491         }
1492
1493         for (i = 0; i < QLEN; i++) {
1494                 req = printer_req_alloc(dev->in_ep, USB_BUFSIZE, GFP_KERNEL);
1495                 if (!req) {
1496                         while (!list_empty(&dev->tx_reqs)) {
1497                                 req = container_of(dev->tx_reqs.next,
1498                                                 struct usb_request, list);
1499                                 list_del(&req->list);
1500                                 printer_req_free(dev->in_ep, req);
1501                         }
1502                         return -ENOMEM;
1503                 }
1504                 list_add(&req->list, &dev->tx_reqs);
1505         }
1506
1507         for (i = 0; i < QLEN; i++) {
1508                 req = printer_req_alloc(dev->out_ep, USB_BUFSIZE, GFP_KERNEL);
1509                 if (!req) {
1510                         while (!list_empty(&dev->rx_reqs)) {
1511                                 req = container_of(dev->rx_reqs.next,
1512                                                 struct usb_request, list);
1513                                 list_del(&req->list);
1514                                 printer_req_free(dev->out_ep, req);
1515                         }
1516                         return -ENOMEM;
1517                 }
1518                 list_add(&req->list, &dev->rx_reqs);
1519         }
1520
1521         dev->req->complete = printer_setup_complete;
1522
1523         /* finish hookup to lower layer ... */
1524         dev->gadget = gadget;
1525         set_gadget_data(gadget, dev);
1526         gadget->ep0->driver_data = dev;
1527
1528         INFO(dev, "%s, version: " DRIVER_VERSION "\n", driver_desc);
1529         INFO(dev, "using %s, OUT %s IN %s\n", gadget->name, out_ep->name,
1530                         in_ep->name);
1531
1532         return 0;
1533
1534 fail:
1535         printer_unbind(gadget);
1536         return status;
1537 }
1538
1539 /*-------------------------------------------------------------------------*/
1540
1541 static struct usb_gadget_driver printer_driver = {
1542         .speed          = DEVSPEED,
1543
1544         .function       = (char *) driver_desc,
1545         .bind           = printer_bind,
1546         .unbind         = printer_unbind,
1547
1548         .setup          = printer_setup,
1549         .disconnect     = printer_disconnect,
1550
1551         .driver         = {
1552                 .name           = (char *) shortname,
1553                 .owner          = THIS_MODULE,
1554         },
1555 };
1556
1557 MODULE_DESCRIPTION(DRIVER_DESC);
1558 MODULE_AUTHOR("Craig Nadler");
1559 MODULE_LICENSE("GPL");
1560
1561 static int __init
1562 init(void)
1563 {
1564         int status;
1565
1566         usb_gadget_class = class_create(THIS_MODULE, "usb_printer_gadget");
1567         if (IS_ERR(usb_gadget_class)) {
1568                 status = PTR_ERR(usb_gadget_class);
1569                 ERROR(dev, "unable to create usb_gadget class %d\n", status);
1570                 return status;
1571         }
1572
1573         status = alloc_chrdev_region(&g_printer_devno, 0, 1,
1574                         "USB printer gadget");
1575         if (status) {
1576                 ERROR(dev, "alloc_chrdev_region %d\n", status);
1577                 class_destroy(usb_gadget_class);
1578                 return status;
1579         }
1580
1581         status = usb_gadget_register_driver(&printer_driver);
1582         if (status) {
1583                 class_destroy(usb_gadget_class);
1584                 unregister_chrdev_region(g_printer_devno, 1);
1585                 DBG(dev, "usb_gadget_register_driver %x\n", status);
1586         }
1587
1588         return status;
1589 }
1590 module_init(init);
1591
1592 static void __exit
1593 cleanup(void)
1594 {
1595         int status;
1596
1597         mutex_lock(&usb_printer_gadget.lock_printer_io);
1598         class_destroy(usb_gadget_class);
1599         unregister_chrdev_region(g_printer_devno, 2);
1600
1601         status = usb_gadget_unregister_driver(&printer_driver);
1602         if (status)
1603                 ERROR(dev, "usb_gadget_unregister_driver %x\n", status);
1604
1605         mutex_unlock(&usb_printer_gadget.lock_printer_io);
1606 }
1607 module_exit(cleanup);