cdc-wdm.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866
  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. * Version Information
  28. */
  29. #define DRIVER_VERSION "v0.03"
  30. #define DRIVER_AUTHOR "Oliver Neukum"
  31. #define DRIVER_DESC "USB Abstract Control Model driver for USB WCM Device Management"
  32. static struct usb_device_id wdm_ids[] = {
  33. {
  34. .match_flags = USB_DEVICE_ID_MATCH_INT_CLASS |
  35. USB_DEVICE_ID_MATCH_INT_SUBCLASS,
  36. .bInterfaceClass = USB_CLASS_COMM,
  37. .bInterfaceSubClass = USB_CDC_SUBCLASS_DMM
  38. },
  39. { }
  40. };
  41. MODULE_DEVICE_TABLE (usb, wdm_ids);
  42. #define WDM_MINOR_BASE 176
  43. #define WDM_IN_USE 1
  44. #define WDM_DISCONNECTING 2
  45. #define WDM_RESULT 3
  46. #define WDM_READ 4
  47. #define WDM_INT_STALL 5
  48. #define WDM_POLL_RUNNING 6
  49. #define WDM_MAX 16
  50. static DEFINE_MUTEX(wdm_mutex);
  51. /* --- method tables --- */
  52. struct wdm_device {
  53. u8 *inbuf; /* buffer for response */
  54. u8 *outbuf; /* buffer for command */
  55. u8 *sbuf; /* buffer for status */
  56. u8 *ubuf; /* buffer for copy to user space */
  57. struct urb *command;
  58. struct urb *response;
  59. struct urb *validity;
  60. struct usb_interface *intf;
  61. struct usb_ctrlrequest *orq;
  62. struct usb_ctrlrequest *irq;
  63. spinlock_t iuspin;
  64. unsigned long flags;
  65. u16 bufsize;
  66. u16 wMaxCommand;
  67. u16 wMaxPacketSize;
  68. u16 bMaxPacketSize0;
  69. __le16 inum;
  70. int reslength;
  71. int length;
  72. int read;
  73. int count;
  74. dma_addr_t shandle;
  75. dma_addr_t ihandle;
  76. struct mutex wlock;
  77. struct mutex rlock;
  78. struct mutex plock;
  79. wait_queue_head_t wait;
  80. struct work_struct rxwork;
  81. int werr;
  82. int rerr;
  83. };
  84. static struct usb_driver wdm_driver;
  85. /* --- callbacks --- */
  86. static void wdm_out_callback(struct urb *urb)
  87. {
  88. struct wdm_device *desc;
  89. desc = urb->context;
  90. spin_lock(&desc->iuspin);
  91. desc->werr = urb->status;
  92. spin_unlock(&desc->iuspin);
  93. clear_bit(WDM_IN_USE, &desc->flags);
  94. kfree(desc->outbuf);
  95. wake_up(&desc->wait);
  96. }
  97. static void wdm_in_callback(struct urb *urb)
  98. {
  99. struct wdm_device *desc = urb->context;
  100. int status = urb->status;
  101. spin_lock(&desc->iuspin);
  102. if (status) {
  103. switch (status) {
  104. case -ENOENT:
  105. dev_dbg(&desc->intf->dev,
  106. "nonzero urb status received: -ENOENT");
  107. break;
  108. case -ECONNRESET:
  109. dev_dbg(&desc->intf->dev,
  110. "nonzero urb status received: -ECONNRESET");
  111. break;
  112. case -ESHUTDOWN:
  113. dev_dbg(&desc->intf->dev,
  114. "nonzero urb status received: -ESHUTDOWN");
  115. break;
  116. case -EPIPE:
  117. dev_err(&desc->intf->dev,
  118. "nonzero urb status received: -EPIPE\n");
  119. break;
  120. default:
  121. dev_err(&desc->intf->dev,
  122. "Unexpected error %d\n", status);
  123. break;
  124. }
  125. }
  126. desc->rerr = status;
  127. desc->reslength = urb->actual_length;
  128. memmove(desc->ubuf + desc->length, desc->inbuf, desc->reslength);
  129. desc->length += desc->reslength;
  130. wake_up(&desc->wait);
  131. set_bit(WDM_READ, &desc->flags);
  132. spin_unlock(&desc->iuspin);
  133. }
  134. static void wdm_int_callback(struct urb *urb)
  135. {
  136. int rv = 0;
  137. int status = urb->status;
  138. struct wdm_device *desc;
  139. struct usb_ctrlrequest *req;
  140. struct usb_cdc_notification *dr;
  141. desc = urb->context;
  142. req = desc->irq;
  143. dr = (struct usb_cdc_notification *)desc->sbuf;
  144. if (status) {
  145. switch (status) {
  146. case -ESHUTDOWN:
  147. case -ENOENT:
  148. case -ECONNRESET:
  149. return; /* unplug */
  150. case -EPIPE:
  151. set_bit(WDM_INT_STALL, &desc->flags);
  152. dev_err(&desc->intf->dev, "Stall on int endpoint\n");
  153. goto sw; /* halt is cleared in work */
  154. default:
  155. dev_err(&desc->intf->dev,
  156. "nonzero urb status received: %d\n", status);
  157. break;
  158. }
  159. }
  160. if (urb->actual_length < sizeof(struct usb_cdc_notification)) {
  161. dev_err(&desc->intf->dev, "wdm_int_callback - %d bytes\n",
  162. urb->actual_length);
  163. goto exit;
  164. }
  165. switch (dr->bNotificationType) {
  166. case USB_CDC_NOTIFY_RESPONSE_AVAILABLE:
  167. dev_dbg(&desc->intf->dev,
  168. "NOTIFY_RESPONSE_AVAILABLE received: index %d len %d",
  169. dr->wIndex, dr->wLength);
  170. break;
  171. case USB_CDC_NOTIFY_NETWORK_CONNECTION:
  172. dev_dbg(&desc->intf->dev,
  173. "NOTIFY_NETWORK_CONNECTION %s network",
  174. dr->wValue ? "connected to" : "disconnected from");
  175. goto exit;
  176. default:
  177. clear_bit(WDM_POLL_RUNNING, &desc->flags);
  178. dev_err(&desc->intf->dev,
  179. "unknown notification %d received: index %d len %d\n",
  180. dr->bNotificationType, dr->wIndex, dr->wLength);
  181. goto exit;
  182. }
  183. req->bRequestType = (USB_DIR_IN | USB_TYPE_CLASS | USB_RECIP_INTERFACE);
  184. req->bRequest = USB_CDC_GET_ENCAPSULATED_RESPONSE;
  185. req->wValue = 0;
  186. req->wIndex = desc->inum;
  187. req->wLength = cpu_to_le16(desc->wMaxCommand);
  188. usb_fill_control_urb(
  189. desc->response,
  190. interface_to_usbdev(desc->intf),
  191. /* using common endpoint 0 */
  192. usb_rcvctrlpipe(interface_to_usbdev(desc->intf), 0),
  193. (unsigned char *)req,
  194. desc->inbuf,
  195. desc->wMaxCommand,
  196. wdm_in_callback,
  197. desc
  198. );
  199. desc->response->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
  200. spin_lock(&desc->iuspin);
  201. clear_bit(WDM_READ, &desc->flags);
  202. if (!test_bit(WDM_DISCONNECTING, &desc->flags)) {
  203. rv = usb_submit_urb(desc->response, GFP_ATOMIC);
  204. dev_dbg(&desc->intf->dev, "%s: usb_submit_urb %d",
  205. __func__, rv);
  206. }
  207. spin_unlock(&desc->iuspin);
  208. if (rv < 0) {
  209. if (rv == -EPERM)
  210. return;
  211. if (rv == -ENOMEM) {
  212. sw:
  213. rv = schedule_work(&desc->rxwork);
  214. if (rv)
  215. dev_err(&desc->intf->dev,
  216. "Cannot schedule work\n");
  217. }
  218. }
  219. exit:
  220. rv = usb_submit_urb(urb, GFP_ATOMIC);
  221. if (rv)
  222. dev_err(&desc->intf->dev,
  223. "%s - usb_submit_urb failed with result %d\n",
  224. __func__, rv);
  225. }
  226. static void kill_urbs(struct wdm_device *desc)
  227. {
  228. /* the order here is essential */
  229. usb_kill_urb(desc->command);
  230. usb_kill_urb(desc->validity);
  231. usb_kill_urb(desc->response);
  232. }
  233. static void free_urbs(struct wdm_device *desc)
  234. {
  235. usb_free_urb(desc->validity);
  236. usb_free_urb(desc->response);
  237. usb_free_urb(desc->command);
  238. }
  239. static void cleanup(struct wdm_device *desc)
  240. {
  241. usb_buffer_free(interface_to_usbdev(desc->intf),
  242. desc->wMaxPacketSize,
  243. desc->sbuf,
  244. desc->validity->transfer_dma);
  245. usb_buffer_free(interface_to_usbdev(desc->intf),
  246. desc->wMaxCommand,
  247. desc->inbuf,
  248. desc->response->transfer_dma);
  249. kfree(desc->orq);
  250. kfree(desc->irq);
  251. kfree(desc->ubuf);
  252. free_urbs(desc);
  253. kfree(desc);
  254. }
  255. static ssize_t wdm_write
  256. (struct file *file, const char __user *buffer, size_t count, loff_t *ppos)
  257. {
  258. u8 *buf;
  259. int rv = -EMSGSIZE, r, we;
  260. struct wdm_device *desc = file->private_data;
  261. struct usb_ctrlrequest *req;
  262. if (count > desc->wMaxCommand)
  263. count = desc->wMaxCommand;
  264. spin_lock_irq(&desc->iuspin);
  265. we = desc->werr;
  266. desc->werr = 0;
  267. spin_unlock_irq(&desc->iuspin);
  268. if (we < 0)
  269. return -EIO;
  270. r = mutex_lock_interruptible(&desc->wlock); /* concurrent writes */
  271. rv = -ERESTARTSYS;
  272. if (r)
  273. goto outnl;
  274. r = usb_autopm_get_interface(desc->intf);
  275. if (r < 0)
  276. goto outnp;
  277. r = wait_event_interruptible(desc->wait, !test_bit(WDM_IN_USE,
  278. &desc->flags));
  279. if (r < 0)
  280. goto out;
  281. if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
  282. rv = -ENODEV;
  283. goto out;
  284. }
  285. desc->outbuf = buf = kmalloc(count, GFP_KERNEL);
  286. if (!buf) {
  287. rv = -ENOMEM;
  288. goto out;
  289. }
  290. r = copy_from_user(buf, buffer, count);
  291. if (r > 0) {
  292. kfree(buf);
  293. rv = -EFAULT;
  294. goto out;
  295. }
  296. req = desc->orq;
  297. usb_fill_control_urb(
  298. desc->command,
  299. interface_to_usbdev(desc->intf),
  300. /* using common endpoint 0 */
  301. usb_sndctrlpipe(interface_to_usbdev(desc->intf), 0),
  302. (unsigned char *)req,
  303. buf,
  304. count,
  305. wdm_out_callback,
  306. desc
  307. );
  308. req->bRequestType = (USB_DIR_OUT | USB_TYPE_CLASS |
  309. USB_RECIP_INTERFACE);
  310. req->bRequest = USB_CDC_SEND_ENCAPSULATED_COMMAND;
  311. req->wValue = 0;
  312. req->wIndex = desc->inum;
  313. req->wLength = cpu_to_le16(count);
  314. set_bit(WDM_IN_USE, &desc->flags);
  315. rv = usb_submit_urb(desc->command, GFP_KERNEL);
  316. if (rv < 0) {
  317. kfree(buf);
  318. clear_bit(WDM_IN_USE, &desc->flags);
  319. dev_err(&desc->intf->dev, "Tx URB error: %d\n", rv);
  320. } else {
  321. dev_dbg(&desc->intf->dev, "Tx URB has been submitted index=%d",
  322. req->wIndex);
  323. }
  324. out:
  325. usb_autopm_put_interface(desc->intf);
  326. outnp:
  327. mutex_unlock(&desc->wlock);
  328. outnl:
  329. return rv < 0 ? rv : count;
  330. }
  331. static ssize_t wdm_read
  332. (struct file *file, char __user *buffer, size_t count, loff_t *ppos)
  333. {
  334. int rv, cntr;
  335. int i = 0;
  336. struct wdm_device *desc = file->private_data;
  337. rv = mutex_lock_interruptible(&desc->rlock); /*concurrent reads */
  338. if (rv < 0)
  339. return -ERESTARTSYS;
  340. if (desc->length == 0) {
  341. desc->read = 0;
  342. retry:
  343. i++;
  344. rv = wait_event_interruptible(desc->wait,
  345. test_bit(WDM_READ, &desc->flags));
  346. if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
  347. rv = -ENODEV;
  348. goto err;
  349. }
  350. usb_mark_last_busy(interface_to_usbdev(desc->intf));
  351. if (rv < 0) {
  352. rv = -ERESTARTSYS;
  353. goto err;
  354. }
  355. spin_lock_irq(&desc->iuspin);
  356. if (desc->rerr) { /* read completed, error happened */
  357. int t = desc->rerr;
  358. desc->rerr = 0;
  359. spin_unlock_irq(&desc->iuspin);
  360. dev_err(&desc->intf->dev,
  361. "reading had resulted in %d\n", t);
  362. rv = -EIO;
  363. goto err;
  364. }
  365. /*
  366. * recheck whether we've lost the race
  367. * against the completion handler
  368. */
  369. if (!test_bit(WDM_READ, &desc->flags)) { /* lost race */
  370. spin_unlock_irq(&desc->iuspin);
  371. goto retry;
  372. }
  373. if (!desc->reslength) { /* zero length read */
  374. spin_unlock_irq(&desc->iuspin);
  375. goto retry;
  376. }
  377. clear_bit(WDM_READ, &desc->flags);
  378. spin_unlock_irq(&desc->iuspin);
  379. }
  380. cntr = count > desc->length ? desc->length : count;
  381. rv = copy_to_user(buffer, desc->ubuf, cntr);
  382. if (rv > 0) {
  383. rv = -EFAULT;
  384. goto err;
  385. }
  386. for (i = 0; i < desc->length - cntr; i++)
  387. desc->ubuf[i] = desc->ubuf[i + cntr];
  388. desc->length -= cntr;
  389. /* in case we had outstanding data */
  390. if (!desc->length)
  391. clear_bit(WDM_READ, &desc->flags);
  392. rv = cntr;
  393. err:
  394. mutex_unlock(&desc->rlock);
  395. if (rv < 0)
  396. dev_err(&desc->intf->dev, "wdm_read: exit error\n");
  397. return rv;
  398. }
  399. static int wdm_flush(struct file *file, fl_owner_t id)
  400. {
  401. struct wdm_device *desc = file->private_data;
  402. wait_event(desc->wait, !test_bit(WDM_IN_USE, &desc->flags));
  403. if (desc->werr < 0)
  404. dev_err(&desc->intf->dev, "Error in flush path: %d\n",
  405. desc->werr);
  406. return desc->werr;
  407. }
  408. static unsigned int wdm_poll(struct file *file, struct poll_table_struct *wait)
  409. {
  410. struct wdm_device *desc = file->private_data;
  411. unsigned long flags;
  412. unsigned int mask = 0;
  413. spin_lock_irqsave(&desc->iuspin, flags);
  414. if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
  415. mask = POLLERR;
  416. spin_unlock_irqrestore(&desc->iuspin, flags);
  417. goto desc_out;
  418. }
  419. if (test_bit(WDM_READ, &desc->flags))
  420. mask = POLLIN | POLLRDNORM;
  421. if (desc->rerr || desc->werr)
  422. mask |= POLLERR;
  423. if (!test_bit(WDM_IN_USE, &desc->flags))
  424. mask |= POLLOUT | POLLWRNORM;
  425. spin_unlock_irqrestore(&desc->iuspin, flags);
  426. poll_wait(file, &desc->wait, wait);
  427. desc_out:
  428. return mask;
  429. }
  430. static int wdm_open(struct inode *inode, struct file *file)
  431. {
  432. int minor = iminor(inode);
  433. int rv = -ENODEV;
  434. struct usb_interface *intf;
  435. struct wdm_device *desc;
  436. mutex_lock(&wdm_mutex);
  437. intf = usb_find_interface(&wdm_driver, minor);
  438. if (!intf)
  439. goto out;
  440. desc = usb_get_intfdata(intf);
  441. if (test_bit(WDM_DISCONNECTING, &desc->flags))
  442. goto out;
  443. ;
  444. file->private_data = desc;
  445. rv = usb_autopm_get_interface(desc->intf);
  446. if (rv < 0) {
  447. dev_err(&desc->intf->dev, "Error autopm - %d\n", rv);
  448. goto out;
  449. }
  450. intf->needs_remote_wakeup = 1;
  451. mutex_lock(&desc->plock);
  452. if (!desc->count++) {
  453. rv = usb_submit_urb(desc->validity, GFP_KERNEL);
  454. if (rv < 0) {
  455. desc->count--;
  456. dev_err(&desc->intf->dev,
  457. "Error submitting int urb - %d\n", rv);
  458. }
  459. } else {
  460. rv = 0;
  461. }
  462. mutex_unlock(&desc->plock);
  463. usb_autopm_put_interface(desc->intf);
  464. out:
  465. mutex_unlock(&wdm_mutex);
  466. return rv;
  467. }
  468. static int wdm_release(struct inode *inode, struct file *file)
  469. {
  470. struct wdm_device *desc = file->private_data;
  471. mutex_lock(&wdm_mutex);
  472. mutex_lock(&desc->plock);
  473. desc->count--;
  474. mutex_unlock(&desc->plock);
  475. if (!desc->count) {
  476. dev_dbg(&desc->intf->dev, "wdm_release: cleanup");
  477. kill_urbs(desc);
  478. if (!test_bit(WDM_DISCONNECTING, &desc->flags))
  479. desc->intf->needs_remote_wakeup = 0;
  480. }
  481. mutex_unlock(&wdm_mutex);
  482. return 0;
  483. }
  484. static const struct file_operations wdm_fops = {
  485. .owner = THIS_MODULE,
  486. .read = wdm_read,
  487. .write = wdm_write,
  488. .open = wdm_open,
  489. .flush = wdm_flush,
  490. .release = wdm_release,
  491. .poll = wdm_poll
  492. };
  493. static struct usb_class_driver wdm_class = {
  494. .name = "cdc-wdm%d",
  495. .fops = &wdm_fops,
  496. .minor_base = WDM_MINOR_BASE,
  497. };
  498. /* --- error handling --- */
  499. static void wdm_rxwork(struct work_struct *work)
  500. {
  501. struct wdm_device *desc = container_of(work, struct wdm_device, rxwork);
  502. unsigned long flags;
  503. int rv;
  504. spin_lock_irqsave(&desc->iuspin, flags);
  505. if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
  506. spin_unlock_irqrestore(&desc->iuspin, flags);
  507. } else {
  508. spin_unlock_irqrestore(&desc->iuspin, flags);
  509. rv = usb_submit_urb(desc->response, GFP_KERNEL);
  510. if (rv < 0 && rv != -EPERM) {
  511. spin_lock_irqsave(&desc->iuspin, flags);
  512. if (!test_bit(WDM_DISCONNECTING, &desc->flags))
  513. schedule_work(&desc->rxwork);
  514. spin_unlock_irqrestore(&desc->iuspin, flags);
  515. }
  516. }
  517. }
  518. /* --- hotplug --- */
  519. static int wdm_probe(struct usb_interface *intf, const struct usb_device_id *id)
  520. {
  521. int rv = -EINVAL;
  522. struct usb_device *udev = interface_to_usbdev(intf);
  523. struct wdm_device *desc;
  524. struct usb_host_interface *iface;
  525. struct usb_endpoint_descriptor *ep;
  526. struct usb_cdc_dmm_desc *dmhd;
  527. u8 *buffer = intf->altsetting->extra;
  528. int buflen = intf->altsetting->extralen;
  529. u16 maxcom = 0;
  530. if (!buffer)
  531. goto out;
  532. while (buflen > 2) {
  533. if (buffer [1] != USB_DT_CS_INTERFACE) {
  534. dev_err(&intf->dev, "skipping garbage\n");
  535. goto next_desc;
  536. }
  537. switch (buffer [2]) {
  538. case USB_CDC_HEADER_TYPE:
  539. break;
  540. case USB_CDC_DMM_TYPE:
  541. dmhd = (struct usb_cdc_dmm_desc *)buffer;
  542. maxcom = le16_to_cpu(dmhd->wMaxCommand);
  543. dev_dbg(&intf->dev,
  544. "Finding maximum buffer length: %d", maxcom);
  545. break;
  546. default:
  547. dev_err(&intf->dev,
  548. "Ignoring extra header, type %d, length %d\n",
  549. buffer[2], buffer[0]);
  550. break;
  551. }
  552. next_desc:
  553. buflen -= buffer[0];
  554. buffer += buffer[0];
  555. }
  556. rv = -ENOMEM;
  557. desc = kzalloc(sizeof(struct wdm_device), GFP_KERNEL);
  558. if (!desc)
  559. goto out;
  560. mutex_init(&desc->wlock);
  561. mutex_init(&desc->rlock);
  562. mutex_init(&desc->plock);
  563. spin_lock_init(&desc->iuspin);
  564. init_waitqueue_head(&desc->wait);
  565. desc->wMaxCommand = maxcom;
  566. /* this will be expanded and needed in hardware endianness */
  567. desc->inum = cpu_to_le16((u16)intf->cur_altsetting->desc.bInterfaceNumber);
  568. desc->intf = intf;
  569. INIT_WORK(&desc->rxwork, wdm_rxwork);
  570. rv = -EINVAL;
  571. iface = intf->cur_altsetting;
  572. if (iface->desc.bNumEndpoints != 1)
  573. goto err;
  574. ep = &iface->endpoint[0].desc;
  575. if (!ep || !usb_endpoint_is_int_in(ep))
  576. goto err;
  577. desc->wMaxPacketSize = le16_to_cpu(ep->wMaxPacketSize);
  578. desc->bMaxPacketSize0 = udev->descriptor.bMaxPacketSize0;
  579. desc->orq = kmalloc(sizeof(struct usb_ctrlrequest), GFP_KERNEL);
  580. if (!desc->orq)
  581. goto err;
  582. desc->irq = kmalloc(sizeof(struct usb_ctrlrequest), GFP_KERNEL);
  583. if (!desc->irq)
  584. goto err;
  585. desc->validity = usb_alloc_urb(0, GFP_KERNEL);
  586. if (!desc->validity)
  587. goto err;
  588. desc->response = usb_alloc_urb(0, GFP_KERNEL);
  589. if (!desc->response)
  590. goto err;
  591. desc->command = usb_alloc_urb(0, GFP_KERNEL);
  592. if (!desc->command)
  593. goto err;
  594. desc->ubuf = kmalloc(desc->wMaxCommand, GFP_KERNEL);
  595. if (!desc->ubuf)
  596. goto err;
  597. desc->sbuf = usb_buffer_alloc(interface_to_usbdev(intf),
  598. desc->wMaxPacketSize,
  599. GFP_KERNEL,
  600. &desc->validity->transfer_dma);
  601. if (!desc->sbuf)
  602. goto err;
  603. desc->inbuf = usb_buffer_alloc(interface_to_usbdev(intf),
  604. desc->bMaxPacketSize0,
  605. GFP_KERNEL,
  606. &desc->response->transfer_dma);
  607. if (!desc->inbuf)
  608. goto err2;
  609. usb_fill_int_urb(
  610. desc->validity,
  611. interface_to_usbdev(intf),
  612. usb_rcvintpipe(interface_to_usbdev(intf), ep->bEndpointAddress),
  613. desc->sbuf,
  614. desc->wMaxPacketSize,
  615. wdm_int_callback,
  616. desc,
  617. ep->bInterval
  618. );
  619. desc->validity->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
  620. usb_set_intfdata(intf, desc);
  621. rv = usb_register_dev(intf, &wdm_class);
  622. if (rv < 0)
  623. goto err3;
  624. else
  625. dev_info(&intf->dev, "cdc-wdm%d: USB WDM device\n",
  626. intf->minor - WDM_MINOR_BASE);
  627. out:
  628. return rv;
  629. err3:
  630. usb_set_intfdata(intf, NULL);
  631. usb_buffer_free(interface_to_usbdev(desc->intf),
  632. desc->bMaxPacketSize0,
  633. desc->inbuf,
  634. desc->response->transfer_dma);
  635. err2:
  636. usb_buffer_free(interface_to_usbdev(desc->intf),
  637. desc->wMaxPacketSize,
  638. desc->sbuf,
  639. desc->validity->transfer_dma);
  640. err:
  641. free_urbs(desc);
  642. kfree(desc->ubuf);
  643. kfree(desc->orq);
  644. kfree(desc->irq);
  645. kfree(desc);
  646. return rv;
  647. }
  648. static void wdm_disconnect(struct usb_interface *intf)
  649. {
  650. struct wdm_device *desc;
  651. unsigned long flags;
  652. usb_deregister_dev(intf, &wdm_class);
  653. mutex_lock(&wdm_mutex);
  654. desc = usb_get_intfdata(intf);
  655. /* the spinlock makes sure no new urbs are generated in the callbacks */
  656. spin_lock_irqsave(&desc->iuspin, flags);
  657. set_bit(WDM_DISCONNECTING, &desc->flags);
  658. set_bit(WDM_READ, &desc->flags);
  659. /* to terminate pending flushes */
  660. clear_bit(WDM_IN_USE, &desc->flags);
  661. spin_unlock_irqrestore(&desc->iuspin, flags);
  662. cancel_work_sync(&desc->rxwork);
  663. kill_urbs(desc);
  664. wake_up_all(&desc->wait);
  665. if (!desc->count)
  666. cleanup(desc);
  667. mutex_unlock(&wdm_mutex);
  668. }
  669. static int wdm_suspend(struct usb_interface *intf, pm_message_t message)
  670. {
  671. struct wdm_device *desc = usb_get_intfdata(intf);
  672. int rv = 0;
  673. dev_dbg(&desc->intf->dev, "wdm%d_suspend\n", intf->minor);
  674. mutex_lock(&desc->plock);
  675. #ifdef CONFIG_PM
  676. if ((message.event & PM_EVENT_AUTO) &&
  677. test_bit(WDM_IN_USE, &desc->flags)) {
  678. rv = -EBUSY;
  679. } else {
  680. #endif
  681. cancel_work_sync(&desc->rxwork);
  682. kill_urbs(desc);
  683. #ifdef CONFIG_PM
  684. }
  685. #endif
  686. mutex_unlock(&desc->plock);
  687. return rv;
  688. }
  689. static int recover_from_urb_loss(struct wdm_device *desc)
  690. {
  691. int rv = 0;
  692. if (desc->count) {
  693. rv = usb_submit_urb(desc->validity, GFP_NOIO);
  694. if (rv < 0)
  695. dev_err(&desc->intf->dev,
  696. "Error resume submitting int urb - %d\n", rv);
  697. }
  698. return rv;
  699. }
  700. static int wdm_resume(struct usb_interface *intf)
  701. {
  702. struct wdm_device *desc = usb_get_intfdata(intf);
  703. int rv;
  704. dev_dbg(&desc->intf->dev, "wdm%d_resume\n", intf->minor);
  705. mutex_lock(&desc->plock);
  706. rv = recover_from_urb_loss(desc);
  707. mutex_unlock(&desc->plock);
  708. return rv;
  709. }
  710. static int wdm_pre_reset(struct usb_interface *intf)
  711. {
  712. struct wdm_device *desc = usb_get_intfdata(intf);
  713. mutex_lock(&desc->plock);
  714. return 0;
  715. }
  716. static int wdm_post_reset(struct usb_interface *intf)
  717. {
  718. struct wdm_device *desc = usb_get_intfdata(intf);
  719. int rv;
  720. rv = recover_from_urb_loss(desc);
  721. mutex_unlock(&desc->plock);
  722. return 0;
  723. }
  724. static struct usb_driver wdm_driver = {
  725. .name = "cdc_wdm",
  726. .probe = wdm_probe,
  727. .disconnect = wdm_disconnect,
  728. .suspend = wdm_suspend,
  729. .resume = wdm_resume,
  730. .reset_resume = wdm_resume,
  731. .pre_reset = wdm_pre_reset,
  732. .post_reset = wdm_post_reset,
  733. .id_table = wdm_ids,
  734. .supports_autosuspend = 1,
  735. };
  736. /* --- low level module stuff --- */
  737. static int __init wdm_init(void)
  738. {
  739. int rv;
  740. rv = usb_register(&wdm_driver);
  741. return rv;
  742. }
  743. static void __exit wdm_exit(void)
  744. {
  745. usb_deregister(&wdm_driver);
  746. }
  747. module_init(wdm_init);
  748. module_exit(wdm_exit);
  749. MODULE_AUTHOR(DRIVER_AUTHOR);
  750. MODULE_DESCRIPTION(DRIVER_DESC);
  751. MODULE_LICENSE("GPL");