Loading...
1/*
2 * cdc-wdm.c
3 *
4 * This driver supports USB CDC WCM Device Management.
5 *
6 * Copyright (c) 2007-2009 Oliver Neukum
7 *
8 * Some code taken from cdc-acm.c
9 *
10 * Released under the GPLv2.
11 *
12 * Many thanks to Carl Nordbeck
13 */
14#include <linux/kernel.h>
15#include <linux/errno.h>
16#include <linux/slab.h>
17#include <linux/module.h>
18#include <linux/mutex.h>
19#include <linux/uaccess.h>
20#include <linux/bitops.h>
21#include <linux/poll.h>
22#include <linux/usb.h>
23#include <linux/usb/cdc.h>
24#include <asm/byteorder.h>
25#include <asm/unaligned.h>
26
27/*
28 * Version Information
29 */
30#define DRIVER_VERSION "v0.03"
31#define DRIVER_AUTHOR "Oliver Neukum"
32#define DRIVER_DESC "USB Abstract Control Model driver for USB WCM Device Management"
33
34static const struct usb_device_id wdm_ids[] = {
35 {
36 .match_flags = USB_DEVICE_ID_MATCH_INT_CLASS |
37 USB_DEVICE_ID_MATCH_INT_SUBCLASS,
38 .bInterfaceClass = USB_CLASS_COMM,
39 .bInterfaceSubClass = USB_CDC_SUBCLASS_DMM
40 },
41 { }
42};
43
44MODULE_DEVICE_TABLE (usb, wdm_ids);
45
46#define WDM_MINOR_BASE 176
47
48
49#define WDM_IN_USE 1
50#define WDM_DISCONNECTING 2
51#define WDM_RESULT 3
52#define WDM_READ 4
53#define WDM_INT_STALL 5
54#define WDM_POLL_RUNNING 6
55#define WDM_RESPONDING 7
56#define WDM_SUSPENDING 8
57
58#define WDM_MAX 16
59
60
61static DEFINE_MUTEX(wdm_mutex);
62
63/* --- method tables --- */
64
65struct wdm_device {
66 u8 *inbuf; /* buffer for response */
67 u8 *outbuf; /* buffer for command */
68 u8 *sbuf; /* buffer for status */
69 u8 *ubuf; /* buffer for copy to user space */
70
71 struct urb *command;
72 struct urb *response;
73 struct urb *validity;
74 struct usb_interface *intf;
75 struct usb_ctrlrequest *orq;
76 struct usb_ctrlrequest *irq;
77 spinlock_t iuspin;
78
79 unsigned long flags;
80 u16 bufsize;
81 u16 wMaxCommand;
82 u16 wMaxPacketSize;
83 u16 bMaxPacketSize0;
84 __le16 inum;
85 int reslength;
86 int length;
87 int read;
88 int count;
89 dma_addr_t shandle;
90 dma_addr_t ihandle;
91 struct mutex lock;
92 wait_queue_head_t wait;
93 struct work_struct rxwork;
94 int werr;
95 int rerr;
96};
97
98static struct usb_driver wdm_driver;
99
100/* --- callbacks --- */
101static void wdm_out_callback(struct urb *urb)
102{
103 struct wdm_device *desc;
104 desc = urb->context;
105 spin_lock(&desc->iuspin);
106 desc->werr = urb->status;
107 spin_unlock(&desc->iuspin);
108 clear_bit(WDM_IN_USE, &desc->flags);
109 kfree(desc->outbuf);
110 wake_up(&desc->wait);
111}
112
113static void wdm_in_callback(struct urb *urb)
114{
115 struct wdm_device *desc = urb->context;
116 int status = urb->status;
117
118 spin_lock(&desc->iuspin);
119 clear_bit(WDM_RESPONDING, &desc->flags);
120
121 if (status) {
122 switch (status) {
123 case -ENOENT:
124 dev_dbg(&desc->intf->dev,
125 "nonzero urb status received: -ENOENT");
126 goto skip_error;
127 case -ECONNRESET:
128 dev_dbg(&desc->intf->dev,
129 "nonzero urb status received: -ECONNRESET");
130 goto skip_error;
131 case -ESHUTDOWN:
132 dev_dbg(&desc->intf->dev,
133 "nonzero urb status received: -ESHUTDOWN");
134 goto skip_error;
135 case -EPIPE:
136 dev_err(&desc->intf->dev,
137 "nonzero urb status received: -EPIPE\n");
138 break;
139 default:
140 dev_err(&desc->intf->dev,
141 "Unexpected error %d\n", status);
142 break;
143 }
144 }
145
146 desc->rerr = status;
147 desc->reslength = urb->actual_length;
148 memmove(desc->ubuf + desc->length, desc->inbuf, desc->reslength);
149 desc->length += desc->reslength;
150skip_error:
151 wake_up(&desc->wait);
152
153 set_bit(WDM_READ, &desc->flags);
154 spin_unlock(&desc->iuspin);
155}
156
157static void wdm_int_callback(struct urb *urb)
158{
159 int rv = 0;
160 int status = urb->status;
161 struct wdm_device *desc;
162 struct usb_ctrlrequest *req;
163 struct usb_cdc_notification *dr;
164
165 desc = urb->context;
166 req = desc->irq;
167 dr = (struct usb_cdc_notification *)desc->sbuf;
168
169 if (status) {
170 switch (status) {
171 case -ESHUTDOWN:
172 case -ENOENT:
173 case -ECONNRESET:
174 return; /* unplug */
175 case -EPIPE:
176 set_bit(WDM_INT_STALL, &desc->flags);
177 dev_err(&desc->intf->dev, "Stall on int endpoint\n");
178 goto sw; /* halt is cleared in work */
179 default:
180 dev_err(&desc->intf->dev,
181 "nonzero urb status received: %d\n", status);
182 break;
183 }
184 }
185
186 if (urb->actual_length < sizeof(struct usb_cdc_notification)) {
187 dev_err(&desc->intf->dev, "wdm_int_callback - %d bytes\n",
188 urb->actual_length);
189 goto exit;
190 }
191
192 switch (dr->bNotificationType) {
193 case USB_CDC_NOTIFY_RESPONSE_AVAILABLE:
194 dev_dbg(&desc->intf->dev,
195 "NOTIFY_RESPONSE_AVAILABLE received: index %d len %d",
196 dr->wIndex, dr->wLength);
197 break;
198
199 case USB_CDC_NOTIFY_NETWORK_CONNECTION:
200
201 dev_dbg(&desc->intf->dev,
202 "NOTIFY_NETWORK_CONNECTION %s network",
203 dr->wValue ? "connected to" : "disconnected from");
204 goto exit;
205 default:
206 clear_bit(WDM_POLL_RUNNING, &desc->flags);
207 dev_err(&desc->intf->dev,
208 "unknown notification %d received: index %d len %d\n",
209 dr->bNotificationType, dr->wIndex, dr->wLength);
210 goto exit;
211 }
212
213 req->bRequestType = (USB_DIR_IN | USB_TYPE_CLASS | USB_RECIP_INTERFACE);
214 req->bRequest = USB_CDC_GET_ENCAPSULATED_RESPONSE;
215 req->wValue = 0;
216 req->wIndex = desc->inum;
217 req->wLength = cpu_to_le16(desc->wMaxCommand);
218
219 usb_fill_control_urb(
220 desc->response,
221 interface_to_usbdev(desc->intf),
222 /* using common endpoint 0 */
223 usb_rcvctrlpipe(interface_to_usbdev(desc->intf), 0),
224 (unsigned char *)req,
225 desc->inbuf,
226 desc->wMaxCommand,
227 wdm_in_callback,
228 desc
229 );
230 desc->response->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
231 spin_lock(&desc->iuspin);
232 clear_bit(WDM_READ, &desc->flags);
233 set_bit(WDM_RESPONDING, &desc->flags);
234 if (!test_bit(WDM_DISCONNECTING, &desc->flags)
235 && !test_bit(WDM_SUSPENDING, &desc->flags)) {
236 rv = usb_submit_urb(desc->response, GFP_ATOMIC);
237 dev_dbg(&desc->intf->dev, "%s: usb_submit_urb %d",
238 __func__, rv);
239 }
240 spin_unlock(&desc->iuspin);
241 if (rv < 0) {
242 clear_bit(WDM_RESPONDING, &desc->flags);
243 if (rv == -EPERM)
244 return;
245 if (rv == -ENOMEM) {
246sw:
247 rv = schedule_work(&desc->rxwork);
248 if (rv)
249 dev_err(&desc->intf->dev,
250 "Cannot schedule work\n");
251 }
252 }
253exit:
254 rv = usb_submit_urb(urb, GFP_ATOMIC);
255 if (rv)
256 dev_err(&desc->intf->dev,
257 "%s - usb_submit_urb failed with result %d\n",
258 __func__, rv);
259
260}
261
262static void kill_urbs(struct wdm_device *desc)
263{
264 /* the order here is essential */
265 usb_kill_urb(desc->command);
266 usb_kill_urb(desc->validity);
267 usb_kill_urb(desc->response);
268}
269
270static void free_urbs(struct wdm_device *desc)
271{
272 usb_free_urb(desc->validity);
273 usb_free_urb(desc->response);
274 usb_free_urb(desc->command);
275}
276
277static void cleanup(struct wdm_device *desc)
278{
279 usb_free_coherent(interface_to_usbdev(desc->intf),
280 desc->wMaxPacketSize,
281 desc->sbuf,
282 desc->validity->transfer_dma);
283 usb_free_coherent(interface_to_usbdev(desc->intf),
284 desc->bMaxPacketSize0,
285 desc->inbuf,
286 desc->response->transfer_dma);
287 kfree(desc->orq);
288 kfree(desc->irq);
289 kfree(desc->ubuf);
290 free_urbs(desc);
291 kfree(desc);
292}
293
294static ssize_t wdm_write
295(struct file *file, const char __user *buffer, size_t count, loff_t *ppos)
296{
297 u8 *buf;
298 int rv = -EMSGSIZE, r, we;
299 struct wdm_device *desc = file->private_data;
300 struct usb_ctrlrequest *req;
301
302 if (count > desc->wMaxCommand)
303 count = desc->wMaxCommand;
304
305 spin_lock_irq(&desc->iuspin);
306 we = desc->werr;
307 desc->werr = 0;
308 spin_unlock_irq(&desc->iuspin);
309 if (we < 0)
310 return -EIO;
311
312 desc->outbuf = buf = kmalloc(count, GFP_KERNEL);
313 if (!buf) {
314 rv = -ENOMEM;
315 goto outnl;
316 }
317
318 r = copy_from_user(buf, buffer, count);
319 if (r > 0) {
320 kfree(buf);
321 rv = -EFAULT;
322 goto outnl;
323 }
324
325 /* concurrent writes and disconnect */
326 r = mutex_lock_interruptible(&desc->lock);
327 rv = -ERESTARTSYS;
328 if (r) {
329 kfree(buf);
330 goto outnl;
331 }
332
333 if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
334 kfree(buf);
335 rv = -ENODEV;
336 goto outnp;
337 }
338
339 r = usb_autopm_get_interface(desc->intf);
340 if (r < 0) {
341 kfree(buf);
342 goto outnp;
343 }
344
345 if (!(file->f_flags & O_NONBLOCK))
346 r = wait_event_interruptible(desc->wait, !test_bit(WDM_IN_USE,
347 &desc->flags));
348 else
349 if (test_bit(WDM_IN_USE, &desc->flags))
350 r = -EAGAIN;
351 if (r < 0) {
352 kfree(buf);
353 goto out;
354 }
355
356 req = desc->orq;
357 usb_fill_control_urb(
358 desc->command,
359 interface_to_usbdev(desc->intf),
360 /* using common endpoint 0 */
361 usb_sndctrlpipe(interface_to_usbdev(desc->intf), 0),
362 (unsigned char *)req,
363 buf,
364 count,
365 wdm_out_callback,
366 desc
367 );
368
369 req->bRequestType = (USB_DIR_OUT | USB_TYPE_CLASS |
370 USB_RECIP_INTERFACE);
371 req->bRequest = USB_CDC_SEND_ENCAPSULATED_COMMAND;
372 req->wValue = 0;
373 req->wIndex = desc->inum;
374 req->wLength = cpu_to_le16(count);
375 set_bit(WDM_IN_USE, &desc->flags);
376
377 rv = usb_submit_urb(desc->command, GFP_KERNEL);
378 if (rv < 0) {
379 kfree(buf);
380 clear_bit(WDM_IN_USE, &desc->flags);
381 dev_err(&desc->intf->dev, "Tx URB error: %d\n", rv);
382 } else {
383 dev_dbg(&desc->intf->dev, "Tx URB has been submitted index=%d",
384 req->wIndex);
385 }
386out:
387 usb_autopm_put_interface(desc->intf);
388outnp:
389 mutex_unlock(&desc->lock);
390outnl:
391 return rv < 0 ? rv : count;
392}
393
394static ssize_t wdm_read
395(struct file *file, char __user *buffer, size_t count, loff_t *ppos)
396{
397 int rv, cntr = 0;
398 int i = 0;
399 struct wdm_device *desc = file->private_data;
400
401
402 rv = mutex_lock_interruptible(&desc->lock); /*concurrent reads */
403 if (rv < 0)
404 return -ERESTARTSYS;
405
406 if (desc->length == 0) {
407 desc->read = 0;
408retry:
409 if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
410 rv = -ENODEV;
411 goto err;
412 }
413 i++;
414 if (file->f_flags & O_NONBLOCK) {
415 if (!test_bit(WDM_READ, &desc->flags)) {
416 rv = cntr ? cntr : -EAGAIN;
417 goto err;
418 }
419 rv = 0;
420 } else {
421 rv = wait_event_interruptible(desc->wait,
422 test_bit(WDM_READ, &desc->flags));
423 }
424
425 /* may have happened while we slept */
426 if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
427 rv = -ENODEV;
428 goto err;
429 }
430 usb_mark_last_busy(interface_to_usbdev(desc->intf));
431 if (rv < 0) {
432 rv = -ERESTARTSYS;
433 goto err;
434 }
435
436 spin_lock_irq(&desc->iuspin);
437
438 if (desc->rerr) { /* read completed, error happened */
439 desc->rerr = 0;
440 spin_unlock_irq(&desc->iuspin);
441 rv = -EIO;
442 goto err;
443 }
444 /*
445 * recheck whether we've lost the race
446 * against the completion handler
447 */
448 if (!test_bit(WDM_READ, &desc->flags)) { /* lost race */
449 spin_unlock_irq(&desc->iuspin);
450 goto retry;
451 }
452 if (!desc->reslength) { /* zero length read */
453 spin_unlock_irq(&desc->iuspin);
454 goto retry;
455 }
456 clear_bit(WDM_READ, &desc->flags);
457 spin_unlock_irq(&desc->iuspin);
458 }
459
460 cntr = count > desc->length ? desc->length : count;
461 rv = copy_to_user(buffer, desc->ubuf, cntr);
462 if (rv > 0) {
463 rv = -EFAULT;
464 goto err;
465 }
466
467 for (i = 0; i < desc->length - cntr; i++)
468 desc->ubuf[i] = desc->ubuf[i + cntr];
469
470 desc->length -= cntr;
471 /* in case we had outstanding data */
472 if (!desc->length)
473 clear_bit(WDM_READ, &desc->flags);
474 rv = cntr;
475
476err:
477 mutex_unlock(&desc->lock);
478 return rv;
479}
480
481static int wdm_flush(struct file *file, fl_owner_t id)
482{
483 struct wdm_device *desc = file->private_data;
484
485 wait_event(desc->wait, !test_bit(WDM_IN_USE, &desc->flags));
486 if (desc->werr < 0)
487 dev_err(&desc->intf->dev, "Error in flush path: %d\n",
488 desc->werr);
489
490 return desc->werr;
491}
492
493static unsigned int wdm_poll(struct file *file, struct poll_table_struct *wait)
494{
495 struct wdm_device *desc = file->private_data;
496 unsigned long flags;
497 unsigned int mask = 0;
498
499 spin_lock_irqsave(&desc->iuspin, flags);
500 if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
501 mask = POLLERR;
502 spin_unlock_irqrestore(&desc->iuspin, flags);
503 goto desc_out;
504 }
505 if (test_bit(WDM_READ, &desc->flags))
506 mask = POLLIN | POLLRDNORM;
507 if (desc->rerr || desc->werr)
508 mask |= POLLERR;
509 if (!test_bit(WDM_IN_USE, &desc->flags))
510 mask |= POLLOUT | POLLWRNORM;
511 spin_unlock_irqrestore(&desc->iuspin, flags);
512
513 poll_wait(file, &desc->wait, wait);
514
515desc_out:
516 return mask;
517}
518
519static int wdm_open(struct inode *inode, struct file *file)
520{
521 int minor = iminor(inode);
522 int rv = -ENODEV;
523 struct usb_interface *intf;
524 struct wdm_device *desc;
525
526 mutex_lock(&wdm_mutex);
527 intf = usb_find_interface(&wdm_driver, minor);
528 if (!intf)
529 goto out;
530
531 desc = usb_get_intfdata(intf);
532 if (test_bit(WDM_DISCONNECTING, &desc->flags))
533 goto out;
534 file->private_data = desc;
535
536 rv = usb_autopm_get_interface(desc->intf);
537 if (rv < 0) {
538 dev_err(&desc->intf->dev, "Error autopm - %d\n", rv);
539 goto out;
540 }
541 intf->needs_remote_wakeup = 1;
542
543 mutex_lock(&desc->lock);
544 if (!desc->count++) {
545 desc->werr = 0;
546 desc->rerr = 0;
547 rv = usb_submit_urb(desc->validity, GFP_KERNEL);
548 if (rv < 0) {
549 desc->count--;
550 dev_err(&desc->intf->dev,
551 "Error submitting int urb - %d\n", rv);
552 }
553 } else {
554 rv = 0;
555 }
556 mutex_unlock(&desc->lock);
557 usb_autopm_put_interface(desc->intf);
558out:
559 mutex_unlock(&wdm_mutex);
560 return rv;
561}
562
563static int wdm_release(struct inode *inode, struct file *file)
564{
565 struct wdm_device *desc = file->private_data;
566
567 mutex_lock(&wdm_mutex);
568 mutex_lock(&desc->lock);
569 desc->count--;
570 mutex_unlock(&desc->lock);
571
572 if (!desc->count) {
573 dev_dbg(&desc->intf->dev, "wdm_release: cleanup");
574 kill_urbs(desc);
575 if (!test_bit(WDM_DISCONNECTING, &desc->flags))
576 desc->intf->needs_remote_wakeup = 0;
577 }
578 mutex_unlock(&wdm_mutex);
579 return 0;
580}
581
582static const struct file_operations wdm_fops = {
583 .owner = THIS_MODULE,
584 .read = wdm_read,
585 .write = wdm_write,
586 .open = wdm_open,
587 .flush = wdm_flush,
588 .release = wdm_release,
589 .poll = wdm_poll,
590 .llseek = noop_llseek,
591};
592
593static struct usb_class_driver wdm_class = {
594 .name = "cdc-wdm%d",
595 .fops = &wdm_fops,
596 .minor_base = WDM_MINOR_BASE,
597};
598
599/* --- error handling --- */
600static void wdm_rxwork(struct work_struct *work)
601{
602 struct wdm_device *desc = container_of(work, struct wdm_device, rxwork);
603 unsigned long flags;
604 int rv;
605
606 spin_lock_irqsave(&desc->iuspin, flags);
607 if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
608 spin_unlock_irqrestore(&desc->iuspin, flags);
609 } else {
610 spin_unlock_irqrestore(&desc->iuspin, flags);
611 rv = usb_submit_urb(desc->response, GFP_KERNEL);
612 if (rv < 0 && rv != -EPERM) {
613 spin_lock_irqsave(&desc->iuspin, flags);
614 if (!test_bit(WDM_DISCONNECTING, &desc->flags))
615 schedule_work(&desc->rxwork);
616 spin_unlock_irqrestore(&desc->iuspin, flags);
617 }
618 }
619}
620
621/* --- hotplug --- */
622
623static int wdm_probe(struct usb_interface *intf, const struct usb_device_id *id)
624{
625 int rv = -EINVAL;
626 struct usb_device *udev = interface_to_usbdev(intf);
627 struct wdm_device *desc;
628 struct usb_host_interface *iface;
629 struct usb_endpoint_descriptor *ep;
630 struct usb_cdc_dmm_desc *dmhd;
631 u8 *buffer = intf->altsetting->extra;
632 int buflen = intf->altsetting->extralen;
633 u16 maxcom = 0;
634
635 if (!buffer)
636 goto out;
637
638 while (buflen > 2) {
639 if (buffer [1] != USB_DT_CS_INTERFACE) {
640 dev_err(&intf->dev, "skipping garbage\n");
641 goto next_desc;
642 }
643
644 switch (buffer [2]) {
645 case USB_CDC_HEADER_TYPE:
646 break;
647 case USB_CDC_DMM_TYPE:
648 dmhd = (struct usb_cdc_dmm_desc *)buffer;
649 maxcom = le16_to_cpu(dmhd->wMaxCommand);
650 dev_dbg(&intf->dev,
651 "Finding maximum buffer length: %d", maxcom);
652 break;
653 default:
654 dev_err(&intf->dev,
655 "Ignoring extra header, type %d, length %d\n",
656 buffer[2], buffer[0]);
657 break;
658 }
659next_desc:
660 buflen -= buffer[0];
661 buffer += buffer[0];
662 }
663
664 rv = -ENOMEM;
665 desc = kzalloc(sizeof(struct wdm_device), GFP_KERNEL);
666 if (!desc)
667 goto out;
668 mutex_init(&desc->lock);
669 spin_lock_init(&desc->iuspin);
670 init_waitqueue_head(&desc->wait);
671 desc->wMaxCommand = maxcom;
672 /* this will be expanded and needed in hardware endianness */
673 desc->inum = cpu_to_le16((u16)intf->cur_altsetting->desc.bInterfaceNumber);
674 desc->intf = intf;
675 INIT_WORK(&desc->rxwork, wdm_rxwork);
676
677 rv = -EINVAL;
678 iface = intf->cur_altsetting;
679 if (iface->desc.bNumEndpoints != 1)
680 goto err;
681 ep = &iface->endpoint[0].desc;
682 if (!ep || !usb_endpoint_is_int_in(ep))
683 goto err;
684
685 desc->wMaxPacketSize = le16_to_cpu(ep->wMaxPacketSize);
686 desc->bMaxPacketSize0 = udev->descriptor.bMaxPacketSize0;
687
688 desc->orq = kmalloc(sizeof(struct usb_ctrlrequest), GFP_KERNEL);
689 if (!desc->orq)
690 goto err;
691 desc->irq = kmalloc(sizeof(struct usb_ctrlrequest), GFP_KERNEL);
692 if (!desc->irq)
693 goto err;
694
695 desc->validity = usb_alloc_urb(0, GFP_KERNEL);
696 if (!desc->validity)
697 goto err;
698
699 desc->response = usb_alloc_urb(0, GFP_KERNEL);
700 if (!desc->response)
701 goto err;
702
703 desc->command = usb_alloc_urb(0, GFP_KERNEL);
704 if (!desc->command)
705 goto err;
706
707 desc->ubuf = kmalloc(desc->wMaxCommand, GFP_KERNEL);
708 if (!desc->ubuf)
709 goto err;
710
711 desc->sbuf = usb_alloc_coherent(interface_to_usbdev(intf),
712 desc->wMaxPacketSize,
713 GFP_KERNEL,
714 &desc->validity->transfer_dma);
715 if (!desc->sbuf)
716 goto err;
717
718 desc->inbuf = usb_alloc_coherent(interface_to_usbdev(intf),
719 desc->bMaxPacketSize0,
720 GFP_KERNEL,
721 &desc->response->transfer_dma);
722 if (!desc->inbuf)
723 goto err2;
724
725 usb_fill_int_urb(
726 desc->validity,
727 interface_to_usbdev(intf),
728 usb_rcvintpipe(interface_to_usbdev(intf), ep->bEndpointAddress),
729 desc->sbuf,
730 desc->wMaxPacketSize,
731 wdm_int_callback,
732 desc,
733 ep->bInterval
734 );
735 desc->validity->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
736
737 usb_set_intfdata(intf, desc);
738 rv = usb_register_dev(intf, &wdm_class);
739 if (rv < 0)
740 goto err3;
741 else
742 dev_info(&intf->dev, "cdc-wdm%d: USB WDM device\n",
743 intf->minor - WDM_MINOR_BASE);
744out:
745 return rv;
746err3:
747 usb_set_intfdata(intf, NULL);
748 usb_free_coherent(interface_to_usbdev(desc->intf),
749 desc->bMaxPacketSize0,
750 desc->inbuf,
751 desc->response->transfer_dma);
752err2:
753 usb_free_coherent(interface_to_usbdev(desc->intf),
754 desc->wMaxPacketSize,
755 desc->sbuf,
756 desc->validity->transfer_dma);
757err:
758 free_urbs(desc);
759 kfree(desc->ubuf);
760 kfree(desc->orq);
761 kfree(desc->irq);
762 kfree(desc);
763 return rv;
764}
765
766static void wdm_disconnect(struct usb_interface *intf)
767{
768 struct wdm_device *desc;
769 unsigned long flags;
770
771 usb_deregister_dev(intf, &wdm_class);
772 mutex_lock(&wdm_mutex);
773 desc = usb_get_intfdata(intf);
774
775 /* the spinlock makes sure no new urbs are generated in the callbacks */
776 spin_lock_irqsave(&desc->iuspin, flags);
777 set_bit(WDM_DISCONNECTING, &desc->flags);
778 set_bit(WDM_READ, &desc->flags);
779 /* to terminate pending flushes */
780 clear_bit(WDM_IN_USE, &desc->flags);
781 spin_unlock_irqrestore(&desc->iuspin, flags);
782 mutex_lock(&desc->lock);
783 kill_urbs(desc);
784 cancel_work_sync(&desc->rxwork);
785 mutex_unlock(&desc->lock);
786 wake_up_all(&desc->wait);
787 if (!desc->count)
788 cleanup(desc);
789 mutex_unlock(&wdm_mutex);
790}
791
792#ifdef CONFIG_PM
793static int wdm_suspend(struct usb_interface *intf, pm_message_t message)
794{
795 struct wdm_device *desc = usb_get_intfdata(intf);
796 int rv = 0;
797
798 dev_dbg(&desc->intf->dev, "wdm%d_suspend\n", intf->minor);
799
800 /* if this is an autosuspend the caller does the locking */
801 if (!(message.event & PM_EVENT_AUTO))
802 mutex_lock(&desc->lock);
803 spin_lock_irq(&desc->iuspin);
804
805 if ((message.event & PM_EVENT_AUTO) &&
806 (test_bit(WDM_IN_USE, &desc->flags)
807 || test_bit(WDM_RESPONDING, &desc->flags))) {
808 spin_unlock_irq(&desc->iuspin);
809 rv = -EBUSY;
810 } else {
811
812 set_bit(WDM_SUSPENDING, &desc->flags);
813 spin_unlock_irq(&desc->iuspin);
814 /* callback submits work - order is essential */
815 kill_urbs(desc);
816 cancel_work_sync(&desc->rxwork);
817 }
818 if (!(message.event & PM_EVENT_AUTO))
819 mutex_unlock(&desc->lock);
820
821 return rv;
822}
823#endif
824
825static int recover_from_urb_loss(struct wdm_device *desc)
826{
827 int rv = 0;
828
829 if (desc->count) {
830 rv = usb_submit_urb(desc->validity, GFP_NOIO);
831 if (rv < 0)
832 dev_err(&desc->intf->dev,
833 "Error resume submitting int urb - %d\n", rv);
834 }
835 return rv;
836}
837
838#ifdef CONFIG_PM
839static int wdm_resume(struct usb_interface *intf)
840{
841 struct wdm_device *desc = usb_get_intfdata(intf);
842 int rv;
843
844 dev_dbg(&desc->intf->dev, "wdm%d_resume\n", intf->minor);
845
846 clear_bit(WDM_SUSPENDING, &desc->flags);
847 rv = recover_from_urb_loss(desc);
848
849 return rv;
850}
851#endif
852
853static int wdm_pre_reset(struct usb_interface *intf)
854{
855 struct wdm_device *desc = usb_get_intfdata(intf);
856
857 mutex_lock(&desc->lock);
858 kill_urbs(desc);
859
860 /*
861 * we notify everybody using poll of
862 * an exceptional situation
863 * must be done before recovery lest a spontaneous
864 * message from the device is lost
865 */
866 spin_lock_irq(&desc->iuspin);
867 desc->rerr = -EINTR;
868 spin_unlock_irq(&desc->iuspin);
869 wake_up_all(&desc->wait);
870 return 0;
871}
872
873static int wdm_post_reset(struct usb_interface *intf)
874{
875 struct wdm_device *desc = usb_get_intfdata(intf);
876 int rv;
877
878 rv = recover_from_urb_loss(desc);
879 mutex_unlock(&desc->lock);
880 return 0;
881}
882
883static struct usb_driver wdm_driver = {
884 .name = "cdc_wdm",
885 .probe = wdm_probe,
886 .disconnect = wdm_disconnect,
887#ifdef CONFIG_PM
888 .suspend = wdm_suspend,
889 .resume = wdm_resume,
890 .reset_resume = wdm_resume,
891#endif
892 .pre_reset = wdm_pre_reset,
893 .post_reset = wdm_post_reset,
894 .id_table = wdm_ids,
895 .supports_autosuspend = 1,
896};
897
898/* --- low level module stuff --- */
899
900static int __init wdm_init(void)
901{
902 int rv;
903
904 rv = usb_register(&wdm_driver);
905
906 return rv;
907}
908
909static void __exit wdm_exit(void)
910{
911 usb_deregister(&wdm_driver);
912}
913
914module_init(wdm_init);
915module_exit(wdm_exit);
916
917MODULE_AUTHOR(DRIVER_AUTHOR);
918MODULE_DESCRIPTION(DRIVER_DESC);
919MODULE_LICENSE("GPL");
1// SPDX-License-Identifier: GPL-2.0
2/*
3 * cdc-wdm.c
4 *
5 * This driver supports USB CDC WCM Device Management.
6 *
7 * Copyright (c) 2007-2009 Oliver Neukum
8 *
9 * Some code taken from cdc-acm.c
10 *
11 * Released under the GPLv2.
12 *
13 * Many thanks to Carl Nordbeck
14 */
15#include <linux/kernel.h>
16#include <linux/errno.h>
17#include <linux/ioctl.h>
18#include <linux/slab.h>
19#include <linux/module.h>
20#include <linux/mutex.h>
21#include <linux/uaccess.h>
22#include <linux/bitops.h>
23#include <linux/poll.h>
24#include <linux/usb.h>
25#include <linux/usb/cdc.h>
26#include <asm/byteorder.h>
27#include <asm/unaligned.h>
28#include <linux/usb/cdc-wdm.h>
29
30#define DRIVER_AUTHOR "Oliver Neukum"
31#define DRIVER_DESC "USB Abstract Control Model driver for USB WCM Device Management"
32
33static const struct usb_device_id wdm_ids[] = {
34 {
35 .match_flags = USB_DEVICE_ID_MATCH_INT_CLASS |
36 USB_DEVICE_ID_MATCH_INT_SUBCLASS,
37 .bInterfaceClass = USB_CLASS_COMM,
38 .bInterfaceSubClass = USB_CDC_SUBCLASS_DMM
39 },
40 { }
41};
42
43MODULE_DEVICE_TABLE (usb, wdm_ids);
44
45#define WDM_MINOR_BASE 176
46
47
48#define WDM_IN_USE 1
49#define WDM_DISCONNECTING 2
50#define WDM_RESULT 3
51#define WDM_READ 4
52#define WDM_INT_STALL 5
53#define WDM_POLL_RUNNING 6
54#define WDM_RESPONDING 7
55#define WDM_SUSPENDING 8
56#define WDM_RESETTING 9
57#define WDM_OVERFLOW 10
58
59#define WDM_MAX 16
60
61/* CDC-WMC r1.1 requires wMaxCommand to be "at least 256 decimal (0x100)" */
62#define WDM_DEFAULT_BUFSIZE 256
63
64static DEFINE_MUTEX(wdm_mutex);
65static DEFINE_SPINLOCK(wdm_device_list_lock);
66static LIST_HEAD(wdm_device_list);
67
68/* --- method tables --- */
69
70struct wdm_device {
71 u8 *inbuf; /* buffer for response */
72 u8 *outbuf; /* buffer for command */
73 u8 *sbuf; /* buffer for status */
74 u8 *ubuf; /* buffer for copy to user space */
75
76 struct urb *command;
77 struct urb *response;
78 struct urb *validity;
79 struct usb_interface *intf;
80 struct usb_ctrlrequest *orq;
81 struct usb_ctrlrequest *irq;
82 spinlock_t iuspin;
83
84 unsigned long flags;
85 u16 bufsize;
86 u16 wMaxCommand;
87 u16 wMaxPacketSize;
88 __le16 inum;
89 int reslength;
90 int length;
91 int read;
92 int count;
93 dma_addr_t shandle;
94 dma_addr_t ihandle;
95 struct mutex wlock;
96 struct mutex rlock;
97 wait_queue_head_t wait;
98 struct work_struct rxwork;
99 int werr;
100 int rerr;
101 int resp_count;
102
103 struct list_head device_list;
104 int (*manage_power)(struct usb_interface *, int);
105};
106
107static struct usb_driver wdm_driver;
108
109/* return intfdata if we own the interface, else look up intf in the list */
110static struct wdm_device *wdm_find_device(struct usb_interface *intf)
111{
112 struct wdm_device *desc;
113
114 spin_lock(&wdm_device_list_lock);
115 list_for_each_entry(desc, &wdm_device_list, device_list)
116 if (desc->intf == intf)
117 goto found;
118 desc = NULL;
119found:
120 spin_unlock(&wdm_device_list_lock);
121
122 return desc;
123}
124
125static struct wdm_device *wdm_find_device_by_minor(int minor)
126{
127 struct wdm_device *desc;
128
129 spin_lock(&wdm_device_list_lock);
130 list_for_each_entry(desc, &wdm_device_list, device_list)
131 if (desc->intf->minor == minor)
132 goto found;
133 desc = NULL;
134found:
135 spin_unlock(&wdm_device_list_lock);
136
137 return desc;
138}
139
140/* --- callbacks --- */
141static void wdm_out_callback(struct urb *urb)
142{
143 struct wdm_device *desc;
144 desc = urb->context;
145 spin_lock(&desc->iuspin);
146 desc->werr = urb->status;
147 spin_unlock(&desc->iuspin);
148 kfree(desc->outbuf);
149 desc->outbuf = NULL;
150 clear_bit(WDM_IN_USE, &desc->flags);
151 wake_up(&desc->wait);
152}
153
154/* forward declaration */
155static int service_outstanding_interrupt(struct wdm_device *desc);
156
157static void wdm_in_callback(struct urb *urb)
158{
159 struct wdm_device *desc = urb->context;
160 int status = urb->status;
161 int length = urb->actual_length;
162
163 spin_lock(&desc->iuspin);
164 clear_bit(WDM_RESPONDING, &desc->flags);
165
166 if (status) {
167 switch (status) {
168 case -ENOENT:
169 dev_dbg(&desc->intf->dev,
170 "nonzero urb status received: -ENOENT\n");
171 goto skip_error;
172 case -ECONNRESET:
173 dev_dbg(&desc->intf->dev,
174 "nonzero urb status received: -ECONNRESET\n");
175 goto skip_error;
176 case -ESHUTDOWN:
177 dev_dbg(&desc->intf->dev,
178 "nonzero urb status received: -ESHUTDOWN\n");
179 goto skip_error;
180 case -EPIPE:
181 dev_err(&desc->intf->dev,
182 "nonzero urb status received: -EPIPE\n");
183 break;
184 default:
185 dev_err(&desc->intf->dev,
186 "Unexpected error %d\n", status);
187 break;
188 }
189 }
190
191 /*
192 * only set a new error if there is no previous error.
193 * Errors are only cleared during read/open
194 * Avoid propagating -EPIPE (stall) to userspace since it is
195 * better handled as an empty read
196 */
197 if (desc->rerr == 0 && status != -EPIPE)
198 desc->rerr = status;
199
200 if (length + desc->length > desc->wMaxCommand) {
201 /* The buffer would overflow */
202 set_bit(WDM_OVERFLOW, &desc->flags);
203 } else {
204 /* we may already be in overflow */
205 if (!test_bit(WDM_OVERFLOW, &desc->flags)) {
206 memmove(desc->ubuf + desc->length, desc->inbuf, length);
207 desc->length += length;
208 desc->reslength = length;
209 }
210 }
211skip_error:
212 set_bit(WDM_READ, &desc->flags);
213 wake_up(&desc->wait);
214
215 if (desc->rerr) {
216 /*
217 * Since there was an error, userspace may decide to not read
218 * any data after poll'ing.
219 * We should respond to further attempts from the device to send
220 * data, so that we can get unstuck.
221 */
222 service_outstanding_interrupt(desc);
223 }
224
225 spin_unlock(&desc->iuspin);
226}
227
228static void wdm_int_callback(struct urb *urb)
229{
230 int rv = 0;
231 int responding;
232 int status = urb->status;
233 struct wdm_device *desc;
234 struct usb_cdc_notification *dr;
235
236 desc = urb->context;
237 dr = (struct usb_cdc_notification *)desc->sbuf;
238
239 if (status) {
240 switch (status) {
241 case -ESHUTDOWN:
242 case -ENOENT:
243 case -ECONNRESET:
244 return; /* unplug */
245 case -EPIPE:
246 set_bit(WDM_INT_STALL, &desc->flags);
247 dev_err(&desc->intf->dev, "Stall on int endpoint\n");
248 goto sw; /* halt is cleared in work */
249 default:
250 dev_err(&desc->intf->dev,
251 "nonzero urb status received: %d\n", status);
252 break;
253 }
254 }
255
256 if (urb->actual_length < sizeof(struct usb_cdc_notification)) {
257 dev_err(&desc->intf->dev, "wdm_int_callback - %d bytes\n",
258 urb->actual_length);
259 goto exit;
260 }
261
262 switch (dr->bNotificationType) {
263 case USB_CDC_NOTIFY_RESPONSE_AVAILABLE:
264 dev_dbg(&desc->intf->dev,
265 "NOTIFY_RESPONSE_AVAILABLE received: index %d len %d\n",
266 le16_to_cpu(dr->wIndex), le16_to_cpu(dr->wLength));
267 break;
268
269 case USB_CDC_NOTIFY_NETWORK_CONNECTION:
270
271 dev_dbg(&desc->intf->dev,
272 "NOTIFY_NETWORK_CONNECTION %s network\n",
273 dr->wValue ? "connected to" : "disconnected from");
274 goto exit;
275 case USB_CDC_NOTIFY_SPEED_CHANGE:
276 dev_dbg(&desc->intf->dev, "SPEED_CHANGE received (len %u)\n",
277 urb->actual_length);
278 goto exit;
279 default:
280 clear_bit(WDM_POLL_RUNNING, &desc->flags);
281 dev_err(&desc->intf->dev,
282 "unknown notification %d received: index %d len %d\n",
283 dr->bNotificationType,
284 le16_to_cpu(dr->wIndex),
285 le16_to_cpu(dr->wLength));
286 goto exit;
287 }
288
289 spin_lock(&desc->iuspin);
290 responding = test_and_set_bit(WDM_RESPONDING, &desc->flags);
291 if (!desc->resp_count++ && !responding
292 && !test_bit(WDM_DISCONNECTING, &desc->flags)
293 && !test_bit(WDM_SUSPENDING, &desc->flags)) {
294 rv = usb_submit_urb(desc->response, GFP_ATOMIC);
295 dev_dbg(&desc->intf->dev, "submit response URB %d\n", rv);
296 }
297 spin_unlock(&desc->iuspin);
298 if (rv < 0) {
299 clear_bit(WDM_RESPONDING, &desc->flags);
300 if (rv == -EPERM)
301 return;
302 if (rv == -ENOMEM) {
303sw:
304 rv = schedule_work(&desc->rxwork);
305 if (rv)
306 dev_err(&desc->intf->dev,
307 "Cannot schedule work\n");
308 }
309 }
310exit:
311 rv = usb_submit_urb(urb, GFP_ATOMIC);
312 if (rv)
313 dev_err(&desc->intf->dev,
314 "%s - usb_submit_urb failed with result %d\n",
315 __func__, rv);
316
317}
318
319static void kill_urbs(struct wdm_device *desc)
320{
321 /* the order here is essential */
322 usb_kill_urb(desc->command);
323 usb_kill_urb(desc->validity);
324 usb_kill_urb(desc->response);
325}
326
327static void free_urbs(struct wdm_device *desc)
328{
329 usb_free_urb(desc->validity);
330 usb_free_urb(desc->response);
331 usb_free_urb(desc->command);
332}
333
334static void cleanup(struct wdm_device *desc)
335{
336 kfree(desc->sbuf);
337 kfree(desc->inbuf);
338 kfree(desc->orq);
339 kfree(desc->irq);
340 kfree(desc->ubuf);
341 free_urbs(desc);
342 kfree(desc);
343}
344
345static ssize_t wdm_write
346(struct file *file, const char __user *buffer, size_t count, loff_t *ppos)
347{
348 u8 *buf;
349 int rv = -EMSGSIZE, r, we;
350 struct wdm_device *desc = file->private_data;
351 struct usb_ctrlrequest *req;
352
353 if (count > desc->wMaxCommand)
354 count = desc->wMaxCommand;
355
356 spin_lock_irq(&desc->iuspin);
357 we = desc->werr;
358 desc->werr = 0;
359 spin_unlock_irq(&desc->iuspin);
360 if (we < 0)
361 return usb_translate_errors(we);
362
363 buf = memdup_user(buffer, count);
364 if (IS_ERR(buf))
365 return PTR_ERR(buf);
366
367 /* concurrent writes and disconnect */
368 r = mutex_lock_interruptible(&desc->wlock);
369 rv = -ERESTARTSYS;
370 if (r)
371 goto out_free_mem;
372
373 if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
374 rv = -ENODEV;
375 goto out_free_mem_lock;
376 }
377
378 r = usb_autopm_get_interface(desc->intf);
379 if (r < 0) {
380 rv = usb_translate_errors(r);
381 goto out_free_mem_lock;
382 }
383
384 if (!(file->f_flags & O_NONBLOCK))
385 r = wait_event_interruptible(desc->wait, !test_bit(WDM_IN_USE,
386 &desc->flags));
387 else
388 if (test_bit(WDM_IN_USE, &desc->flags))
389 r = -EAGAIN;
390
391 if (test_bit(WDM_RESETTING, &desc->flags))
392 r = -EIO;
393
394 if (r < 0) {
395 rv = r;
396 goto out_free_mem_pm;
397 }
398
399 req = desc->orq;
400 usb_fill_control_urb(
401 desc->command,
402 interface_to_usbdev(desc->intf),
403 /* using common endpoint 0 */
404 usb_sndctrlpipe(interface_to_usbdev(desc->intf), 0),
405 (unsigned char *)req,
406 buf,
407 count,
408 wdm_out_callback,
409 desc
410 );
411
412 req->bRequestType = (USB_DIR_OUT | USB_TYPE_CLASS |
413 USB_RECIP_INTERFACE);
414 req->bRequest = USB_CDC_SEND_ENCAPSULATED_COMMAND;
415 req->wValue = 0;
416 req->wIndex = desc->inum; /* already converted */
417 req->wLength = cpu_to_le16(count);
418 set_bit(WDM_IN_USE, &desc->flags);
419 desc->outbuf = buf;
420
421 rv = usb_submit_urb(desc->command, GFP_KERNEL);
422 if (rv < 0) {
423 desc->outbuf = NULL;
424 clear_bit(WDM_IN_USE, &desc->flags);
425 dev_err(&desc->intf->dev, "Tx URB error: %d\n", rv);
426 rv = usb_translate_errors(rv);
427 goto out_free_mem_pm;
428 } else {
429 dev_dbg(&desc->intf->dev, "Tx URB has been submitted index=%d\n",
430 le16_to_cpu(req->wIndex));
431 }
432
433 usb_autopm_put_interface(desc->intf);
434 mutex_unlock(&desc->wlock);
435 return count;
436
437out_free_mem_pm:
438 usb_autopm_put_interface(desc->intf);
439out_free_mem_lock:
440 mutex_unlock(&desc->wlock);
441out_free_mem:
442 kfree(buf);
443 return rv;
444}
445
446/*
447 * Submit the read urb if resp_count is non-zero.
448 *
449 * Called with desc->iuspin locked
450 */
451static int service_outstanding_interrupt(struct wdm_device *desc)
452{
453 int rv = 0;
454
455 /* submit read urb only if the device is waiting for it */
456 if (!desc->resp_count || !--desc->resp_count)
457 goto out;
458
459 set_bit(WDM_RESPONDING, &desc->flags);
460 spin_unlock_irq(&desc->iuspin);
461 rv = usb_submit_urb(desc->response, GFP_KERNEL);
462 spin_lock_irq(&desc->iuspin);
463 if (rv) {
464 dev_err(&desc->intf->dev,
465 "usb_submit_urb failed with result %d\n", rv);
466
467 /* make sure the next notification trigger a submit */
468 clear_bit(WDM_RESPONDING, &desc->flags);
469 desc->resp_count = 0;
470 }
471out:
472 return rv;
473}
474
475static ssize_t wdm_read
476(struct file *file, char __user *buffer, size_t count, loff_t *ppos)
477{
478 int rv, cntr;
479 int i = 0;
480 struct wdm_device *desc = file->private_data;
481
482
483 rv = mutex_lock_interruptible(&desc->rlock); /*concurrent reads */
484 if (rv < 0)
485 return -ERESTARTSYS;
486
487 cntr = READ_ONCE(desc->length);
488 if (cntr == 0) {
489 desc->read = 0;
490retry:
491 if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
492 rv = -ENODEV;
493 goto err;
494 }
495 if (test_bit(WDM_OVERFLOW, &desc->flags)) {
496 clear_bit(WDM_OVERFLOW, &desc->flags);
497 rv = -ENOBUFS;
498 goto err;
499 }
500 i++;
501 if (file->f_flags & O_NONBLOCK) {
502 if (!test_bit(WDM_READ, &desc->flags)) {
503 rv = -EAGAIN;
504 goto err;
505 }
506 rv = 0;
507 } else {
508 rv = wait_event_interruptible(desc->wait,
509 test_bit(WDM_READ, &desc->flags));
510 }
511
512 /* may have happened while we slept */
513 if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
514 rv = -ENODEV;
515 goto err;
516 }
517 if (test_bit(WDM_RESETTING, &desc->flags)) {
518 rv = -EIO;
519 goto err;
520 }
521 usb_mark_last_busy(interface_to_usbdev(desc->intf));
522 if (rv < 0) {
523 rv = -ERESTARTSYS;
524 goto err;
525 }
526
527 spin_lock_irq(&desc->iuspin);
528
529 if (desc->rerr) { /* read completed, error happened */
530 rv = usb_translate_errors(desc->rerr);
531 desc->rerr = 0;
532 spin_unlock_irq(&desc->iuspin);
533 goto err;
534 }
535 /*
536 * recheck whether we've lost the race
537 * against the completion handler
538 */
539 if (!test_bit(WDM_READ, &desc->flags)) { /* lost race */
540 spin_unlock_irq(&desc->iuspin);
541 goto retry;
542 }
543
544 if (!desc->reslength) { /* zero length read */
545 dev_dbg(&desc->intf->dev, "zero length - clearing WDM_READ\n");
546 clear_bit(WDM_READ, &desc->flags);
547 rv = service_outstanding_interrupt(desc);
548 spin_unlock_irq(&desc->iuspin);
549 if (rv < 0)
550 goto err;
551 goto retry;
552 }
553 cntr = desc->length;
554 spin_unlock_irq(&desc->iuspin);
555 }
556
557 if (cntr > count)
558 cntr = count;
559 rv = copy_to_user(buffer, desc->ubuf, cntr);
560 if (rv > 0) {
561 rv = -EFAULT;
562 goto err;
563 }
564
565 spin_lock_irq(&desc->iuspin);
566
567 for (i = 0; i < desc->length - cntr; i++)
568 desc->ubuf[i] = desc->ubuf[i + cntr];
569
570 desc->length -= cntr;
571 /* in case we had outstanding data */
572 if (!desc->length) {
573 clear_bit(WDM_READ, &desc->flags);
574 service_outstanding_interrupt(desc);
575 }
576 spin_unlock_irq(&desc->iuspin);
577 rv = cntr;
578
579err:
580 mutex_unlock(&desc->rlock);
581 return rv;
582}
583
584static int wdm_flush(struct file *file, fl_owner_t id)
585{
586 struct wdm_device *desc = file->private_data;
587
588 wait_event(desc->wait, !test_bit(WDM_IN_USE, &desc->flags));
589
590 /* cannot dereference desc->intf if WDM_DISCONNECTING */
591 if (desc->werr < 0 && !test_bit(WDM_DISCONNECTING, &desc->flags))
592 dev_err(&desc->intf->dev, "Error in flush path: %d\n",
593 desc->werr);
594
595 return usb_translate_errors(desc->werr);
596}
597
598static __poll_t wdm_poll(struct file *file, struct poll_table_struct *wait)
599{
600 struct wdm_device *desc = file->private_data;
601 unsigned long flags;
602 __poll_t mask = 0;
603
604 spin_lock_irqsave(&desc->iuspin, flags);
605 if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
606 mask = EPOLLHUP | EPOLLERR;
607 spin_unlock_irqrestore(&desc->iuspin, flags);
608 goto desc_out;
609 }
610 if (test_bit(WDM_READ, &desc->flags))
611 mask = EPOLLIN | EPOLLRDNORM;
612 if (desc->rerr || desc->werr)
613 mask |= EPOLLERR;
614 if (!test_bit(WDM_IN_USE, &desc->flags))
615 mask |= EPOLLOUT | EPOLLWRNORM;
616 spin_unlock_irqrestore(&desc->iuspin, flags);
617
618 poll_wait(file, &desc->wait, wait);
619
620desc_out:
621 return mask;
622}
623
624static int wdm_open(struct inode *inode, struct file *file)
625{
626 int minor = iminor(inode);
627 int rv = -ENODEV;
628 struct usb_interface *intf;
629 struct wdm_device *desc;
630
631 mutex_lock(&wdm_mutex);
632 desc = wdm_find_device_by_minor(minor);
633 if (!desc)
634 goto out;
635
636 intf = desc->intf;
637 if (test_bit(WDM_DISCONNECTING, &desc->flags))
638 goto out;
639 file->private_data = desc;
640
641 rv = usb_autopm_get_interface(desc->intf);
642 if (rv < 0) {
643 dev_err(&desc->intf->dev, "Error autopm - %d\n", rv);
644 goto out;
645 }
646
647 /* using write lock to protect desc->count */
648 mutex_lock(&desc->wlock);
649 if (!desc->count++) {
650 desc->werr = 0;
651 desc->rerr = 0;
652 rv = usb_submit_urb(desc->validity, GFP_KERNEL);
653 if (rv < 0) {
654 desc->count--;
655 dev_err(&desc->intf->dev,
656 "Error submitting int urb - %d\n", rv);
657 rv = usb_translate_errors(rv);
658 }
659 } else {
660 rv = 0;
661 }
662 mutex_unlock(&desc->wlock);
663 if (desc->count == 1)
664 desc->manage_power(intf, 1);
665 usb_autopm_put_interface(desc->intf);
666out:
667 mutex_unlock(&wdm_mutex);
668 return rv;
669}
670
671static int wdm_release(struct inode *inode, struct file *file)
672{
673 struct wdm_device *desc = file->private_data;
674
675 mutex_lock(&wdm_mutex);
676
677 /* using write lock to protect desc->count */
678 mutex_lock(&desc->wlock);
679 desc->count--;
680 mutex_unlock(&desc->wlock);
681
682 if (!desc->count) {
683 if (!test_bit(WDM_DISCONNECTING, &desc->flags)) {
684 dev_dbg(&desc->intf->dev, "wdm_release: cleanup\n");
685 kill_urbs(desc);
686 spin_lock_irq(&desc->iuspin);
687 desc->resp_count = 0;
688 spin_unlock_irq(&desc->iuspin);
689 desc->manage_power(desc->intf, 0);
690 } else {
691 /* must avoid dev_printk here as desc->intf is invalid */
692 pr_debug(KBUILD_MODNAME " %s: device gone - cleaning up\n", __func__);
693 cleanup(desc);
694 }
695 }
696 mutex_unlock(&wdm_mutex);
697 return 0;
698}
699
700static long wdm_ioctl(struct file *file, unsigned int cmd, unsigned long arg)
701{
702 struct wdm_device *desc = file->private_data;
703 int rv = 0;
704
705 switch (cmd) {
706 case IOCTL_WDM_MAX_COMMAND:
707 if (copy_to_user((void __user *)arg, &desc->wMaxCommand, sizeof(desc->wMaxCommand)))
708 rv = -EFAULT;
709 break;
710 default:
711 rv = -ENOTTY;
712 }
713 return rv;
714}
715
716static const struct file_operations wdm_fops = {
717 .owner = THIS_MODULE,
718 .read = wdm_read,
719 .write = wdm_write,
720 .open = wdm_open,
721 .flush = wdm_flush,
722 .release = wdm_release,
723 .poll = wdm_poll,
724 .unlocked_ioctl = wdm_ioctl,
725 .compat_ioctl = wdm_ioctl,
726 .llseek = noop_llseek,
727};
728
729static struct usb_class_driver wdm_class = {
730 .name = "cdc-wdm%d",
731 .fops = &wdm_fops,
732 .minor_base = WDM_MINOR_BASE,
733};
734
735/* --- error handling --- */
736static void wdm_rxwork(struct work_struct *work)
737{
738 struct wdm_device *desc = container_of(work, struct wdm_device, rxwork);
739 unsigned long flags;
740 int rv = 0;
741 int responding;
742
743 spin_lock_irqsave(&desc->iuspin, flags);
744 if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
745 spin_unlock_irqrestore(&desc->iuspin, flags);
746 } else {
747 responding = test_and_set_bit(WDM_RESPONDING, &desc->flags);
748 spin_unlock_irqrestore(&desc->iuspin, flags);
749 if (!responding)
750 rv = usb_submit_urb(desc->response, GFP_KERNEL);
751 if (rv < 0 && rv != -EPERM) {
752 spin_lock_irqsave(&desc->iuspin, flags);
753 clear_bit(WDM_RESPONDING, &desc->flags);
754 if (!test_bit(WDM_DISCONNECTING, &desc->flags))
755 schedule_work(&desc->rxwork);
756 spin_unlock_irqrestore(&desc->iuspin, flags);
757 }
758 }
759}
760
761/* --- hotplug --- */
762
763static int wdm_create(struct usb_interface *intf, struct usb_endpoint_descriptor *ep,
764 u16 bufsize, int (*manage_power)(struct usb_interface *, int))
765{
766 int rv = -ENOMEM;
767 struct wdm_device *desc;
768
769 desc = kzalloc(sizeof(struct wdm_device), GFP_KERNEL);
770 if (!desc)
771 goto out;
772 INIT_LIST_HEAD(&desc->device_list);
773 mutex_init(&desc->rlock);
774 mutex_init(&desc->wlock);
775 spin_lock_init(&desc->iuspin);
776 init_waitqueue_head(&desc->wait);
777 desc->wMaxCommand = bufsize;
778 /* this will be expanded and needed in hardware endianness */
779 desc->inum = cpu_to_le16((u16)intf->cur_altsetting->desc.bInterfaceNumber);
780 desc->intf = intf;
781 INIT_WORK(&desc->rxwork, wdm_rxwork);
782
783 rv = -EINVAL;
784 if (!usb_endpoint_is_int_in(ep))
785 goto err;
786
787 desc->wMaxPacketSize = usb_endpoint_maxp(ep);
788
789 desc->orq = kmalloc(sizeof(struct usb_ctrlrequest), GFP_KERNEL);
790 if (!desc->orq)
791 goto err;
792 desc->irq = kmalloc(sizeof(struct usb_ctrlrequest), GFP_KERNEL);
793 if (!desc->irq)
794 goto err;
795
796 desc->validity = usb_alloc_urb(0, GFP_KERNEL);
797 if (!desc->validity)
798 goto err;
799
800 desc->response = usb_alloc_urb(0, GFP_KERNEL);
801 if (!desc->response)
802 goto err;
803
804 desc->command = usb_alloc_urb(0, GFP_KERNEL);
805 if (!desc->command)
806 goto err;
807
808 desc->ubuf = kmalloc(desc->wMaxCommand, GFP_KERNEL);
809 if (!desc->ubuf)
810 goto err;
811
812 desc->sbuf = kmalloc(desc->wMaxPacketSize, GFP_KERNEL);
813 if (!desc->sbuf)
814 goto err;
815
816 desc->inbuf = kmalloc(desc->wMaxCommand, GFP_KERNEL);
817 if (!desc->inbuf)
818 goto err;
819
820 usb_fill_int_urb(
821 desc->validity,
822 interface_to_usbdev(intf),
823 usb_rcvintpipe(interface_to_usbdev(intf), ep->bEndpointAddress),
824 desc->sbuf,
825 desc->wMaxPacketSize,
826 wdm_int_callback,
827 desc,
828 ep->bInterval
829 );
830
831 desc->irq->bRequestType = (USB_DIR_IN | USB_TYPE_CLASS | USB_RECIP_INTERFACE);
832 desc->irq->bRequest = USB_CDC_GET_ENCAPSULATED_RESPONSE;
833 desc->irq->wValue = 0;
834 desc->irq->wIndex = desc->inum; /* already converted */
835 desc->irq->wLength = cpu_to_le16(desc->wMaxCommand);
836
837 usb_fill_control_urb(
838 desc->response,
839 interface_to_usbdev(intf),
840 /* using common endpoint 0 */
841 usb_rcvctrlpipe(interface_to_usbdev(desc->intf), 0),
842 (unsigned char *)desc->irq,
843 desc->inbuf,
844 desc->wMaxCommand,
845 wdm_in_callback,
846 desc
847 );
848
849 desc->manage_power = manage_power;
850
851 spin_lock(&wdm_device_list_lock);
852 list_add(&desc->device_list, &wdm_device_list);
853 spin_unlock(&wdm_device_list_lock);
854
855 rv = usb_register_dev(intf, &wdm_class);
856 if (rv < 0)
857 goto err;
858 else
859 dev_info(&intf->dev, "%s: USB WDM device\n", dev_name(intf->usb_dev));
860out:
861 return rv;
862err:
863 spin_lock(&wdm_device_list_lock);
864 list_del(&desc->device_list);
865 spin_unlock(&wdm_device_list_lock);
866 cleanup(desc);
867 return rv;
868}
869
870static int wdm_manage_power(struct usb_interface *intf, int on)
871{
872 /* need autopm_get/put here to ensure the usbcore sees the new value */
873 int rv = usb_autopm_get_interface(intf);
874
875 intf->needs_remote_wakeup = on;
876 if (!rv)
877 usb_autopm_put_interface(intf);
878 return 0;
879}
880
881static int wdm_probe(struct usb_interface *intf, const struct usb_device_id *id)
882{
883 int rv = -EINVAL;
884 struct usb_host_interface *iface;
885 struct usb_endpoint_descriptor *ep;
886 struct usb_cdc_parsed_header hdr;
887 u8 *buffer = intf->altsetting->extra;
888 int buflen = intf->altsetting->extralen;
889 u16 maxcom = WDM_DEFAULT_BUFSIZE;
890
891 if (!buffer)
892 goto err;
893
894 cdc_parse_cdc_header(&hdr, intf, buffer, buflen);
895
896 if (hdr.usb_cdc_dmm_desc)
897 maxcom = le16_to_cpu(hdr.usb_cdc_dmm_desc->wMaxCommand);
898
899 iface = intf->cur_altsetting;
900 if (iface->desc.bNumEndpoints != 1)
901 goto err;
902 ep = &iface->endpoint[0].desc;
903
904 rv = wdm_create(intf, ep, maxcom, &wdm_manage_power);
905
906err:
907 return rv;
908}
909
910/**
911 * usb_cdc_wdm_register - register a WDM subdriver
912 * @intf: usb interface the subdriver will associate with
913 * @ep: interrupt endpoint to monitor for notifications
914 * @bufsize: maximum message size to support for read/write
915 *
916 * Create WDM usb class character device and associate it with intf
917 * without binding, allowing another driver to manage the interface.
918 *
919 * The subdriver will manage the given interrupt endpoint exclusively
920 * and will issue control requests referring to the given intf. It
921 * will otherwise avoid interferring, and in particular not do
922 * usb_set_intfdata/usb_get_intfdata on intf.
923 *
924 * The return value is a pointer to the subdriver's struct usb_driver.
925 * The registering driver is responsible for calling this subdriver's
926 * disconnect, suspend, resume, pre_reset and post_reset methods from
927 * its own.
928 */
929struct usb_driver *usb_cdc_wdm_register(struct usb_interface *intf,
930 struct usb_endpoint_descriptor *ep,
931 int bufsize,
932 int (*manage_power)(struct usb_interface *, int))
933{
934 int rv = -EINVAL;
935
936 rv = wdm_create(intf, ep, bufsize, manage_power);
937 if (rv < 0)
938 goto err;
939
940 return &wdm_driver;
941err:
942 return ERR_PTR(rv);
943}
944EXPORT_SYMBOL(usb_cdc_wdm_register);
945
946static void wdm_disconnect(struct usb_interface *intf)
947{
948 struct wdm_device *desc;
949 unsigned long flags;
950
951 usb_deregister_dev(intf, &wdm_class);
952 desc = wdm_find_device(intf);
953 mutex_lock(&wdm_mutex);
954
955 /* the spinlock makes sure no new urbs are generated in the callbacks */
956 spin_lock_irqsave(&desc->iuspin, flags);
957 set_bit(WDM_DISCONNECTING, &desc->flags);
958 set_bit(WDM_READ, &desc->flags);
959 /* to terminate pending flushes */
960 clear_bit(WDM_IN_USE, &desc->flags);
961 spin_unlock_irqrestore(&desc->iuspin, flags);
962 wake_up_all(&desc->wait);
963 mutex_lock(&desc->rlock);
964 mutex_lock(&desc->wlock);
965 kill_urbs(desc);
966 cancel_work_sync(&desc->rxwork);
967 mutex_unlock(&desc->wlock);
968 mutex_unlock(&desc->rlock);
969
970 /* the desc->intf pointer used as list key is now invalid */
971 spin_lock(&wdm_device_list_lock);
972 list_del(&desc->device_list);
973 spin_unlock(&wdm_device_list_lock);
974
975 if (!desc->count)
976 cleanup(desc);
977 else
978 dev_dbg(&intf->dev, "%d open files - postponing cleanup\n", desc->count);
979 mutex_unlock(&wdm_mutex);
980}
981
982#ifdef CONFIG_PM
983static int wdm_suspend(struct usb_interface *intf, pm_message_t message)
984{
985 struct wdm_device *desc = wdm_find_device(intf);
986 int rv = 0;
987
988 dev_dbg(&desc->intf->dev, "wdm%d_suspend\n", intf->minor);
989
990 /* if this is an autosuspend the caller does the locking */
991 if (!PMSG_IS_AUTO(message)) {
992 mutex_lock(&desc->rlock);
993 mutex_lock(&desc->wlock);
994 }
995 spin_lock_irq(&desc->iuspin);
996
997 if (PMSG_IS_AUTO(message) &&
998 (test_bit(WDM_IN_USE, &desc->flags)
999 || test_bit(WDM_RESPONDING, &desc->flags))) {
1000 spin_unlock_irq(&desc->iuspin);
1001 rv = -EBUSY;
1002 } else {
1003
1004 set_bit(WDM_SUSPENDING, &desc->flags);
1005 spin_unlock_irq(&desc->iuspin);
1006 /* callback submits work - order is essential */
1007 kill_urbs(desc);
1008 cancel_work_sync(&desc->rxwork);
1009 }
1010 if (!PMSG_IS_AUTO(message)) {
1011 mutex_unlock(&desc->wlock);
1012 mutex_unlock(&desc->rlock);
1013 }
1014
1015 return rv;
1016}
1017#endif
1018
1019static int recover_from_urb_loss(struct wdm_device *desc)
1020{
1021 int rv = 0;
1022
1023 if (desc->count) {
1024 rv = usb_submit_urb(desc->validity, GFP_NOIO);
1025 if (rv < 0)
1026 dev_err(&desc->intf->dev,
1027 "Error resume submitting int urb - %d\n", rv);
1028 }
1029 return rv;
1030}
1031
1032#ifdef CONFIG_PM
1033static int wdm_resume(struct usb_interface *intf)
1034{
1035 struct wdm_device *desc = wdm_find_device(intf);
1036 int rv;
1037
1038 dev_dbg(&desc->intf->dev, "wdm%d_resume\n", intf->minor);
1039
1040 clear_bit(WDM_SUSPENDING, &desc->flags);
1041 rv = recover_from_urb_loss(desc);
1042
1043 return rv;
1044}
1045#endif
1046
1047static int wdm_pre_reset(struct usb_interface *intf)
1048{
1049 struct wdm_device *desc = wdm_find_device(intf);
1050
1051 /*
1052 * we notify everybody using poll of
1053 * an exceptional situation
1054 * must be done before recovery lest a spontaneous
1055 * message from the device is lost
1056 */
1057 spin_lock_irq(&desc->iuspin);
1058 set_bit(WDM_RESETTING, &desc->flags); /* inform read/write */
1059 set_bit(WDM_READ, &desc->flags); /* unblock read */
1060 clear_bit(WDM_IN_USE, &desc->flags); /* unblock write */
1061 desc->rerr = -EINTR;
1062 spin_unlock_irq(&desc->iuspin);
1063 wake_up_all(&desc->wait);
1064 mutex_lock(&desc->rlock);
1065 mutex_lock(&desc->wlock);
1066 kill_urbs(desc);
1067 cancel_work_sync(&desc->rxwork);
1068 return 0;
1069}
1070
1071static int wdm_post_reset(struct usb_interface *intf)
1072{
1073 struct wdm_device *desc = wdm_find_device(intf);
1074 int rv;
1075
1076 clear_bit(WDM_OVERFLOW, &desc->flags);
1077 clear_bit(WDM_RESETTING, &desc->flags);
1078 rv = recover_from_urb_loss(desc);
1079 mutex_unlock(&desc->wlock);
1080 mutex_unlock(&desc->rlock);
1081 return 0;
1082}
1083
1084static struct usb_driver wdm_driver = {
1085 .name = "cdc_wdm",
1086 .probe = wdm_probe,
1087 .disconnect = wdm_disconnect,
1088#ifdef CONFIG_PM
1089 .suspend = wdm_suspend,
1090 .resume = wdm_resume,
1091 .reset_resume = wdm_resume,
1092#endif
1093 .pre_reset = wdm_pre_reset,
1094 .post_reset = wdm_post_reset,
1095 .id_table = wdm_ids,
1096 .supports_autosuspend = 1,
1097 .disable_hub_initiated_lpm = 1,
1098};
1099
1100module_usb_driver(wdm_driver);
1101
1102MODULE_AUTHOR(DRIVER_AUTHOR);
1103MODULE_DESCRIPTION(DRIVER_DESC);
1104MODULE_LICENSE("GPL");