cdc-wdm.c 18 KB

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