cdc-wdm.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877
  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 const 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_RESPONDING 7
  50. #define WDM_SUSPENDING 8
  51. #define WDM_MAX 16
  52. static DEFINE_MUTEX(wdm_mutex);
  53. /* --- method tables --- */
  54. struct wdm_device {
  55. u8 *inbuf; /* buffer for response */
  56. u8 *outbuf; /* buffer for command */
  57. u8 *sbuf; /* buffer for status */
  58. u8 *ubuf; /* buffer for copy to user space */
  59. struct urb *command;
  60. struct urb *response;
  61. struct urb *validity;
  62. struct usb_interface *intf;
  63. struct usb_ctrlrequest *orq;
  64. struct usb_ctrlrequest *irq;
  65. spinlock_t iuspin;
  66. unsigned long flags;
  67. u16 bufsize;
  68. u16 wMaxCommand;
  69. u16 wMaxPacketSize;
  70. __le16 inum;
  71. int reslength;
  72. int length;
  73. int read;
  74. int count;
  75. dma_addr_t shandle;
  76. dma_addr_t ihandle;
  77. struct mutex lock;
  78. wait_queue_head_t wait;
  79. struct work_struct rxwork;
  80. int werr;
  81. int rerr;
  82. };
  83. static struct usb_driver wdm_driver;
  84. /* --- callbacks --- */
  85. static void wdm_out_callback(struct urb *urb)
  86. {
  87. struct wdm_device *desc;
  88. desc = urb->context;
  89. spin_lock(&desc->iuspin);
  90. desc->werr = urb->status;
  91. spin_unlock(&desc->iuspin);
  92. clear_bit(WDM_IN_USE, &desc->flags);
  93. kfree(desc->outbuf);
  94. wake_up(&desc->wait);
  95. }
  96. static void wdm_in_callback(struct urb *urb)
  97. {
  98. struct wdm_device *desc = urb->context;
  99. int status = urb->status;
  100. spin_lock(&desc->iuspin);
  101. clear_bit(WDM_RESPONDING, &desc->flags);
  102. if (status) {
  103. switch (status) {
  104. case -ENOENT:
  105. dev_dbg(&desc->intf->dev,
  106. "nonzero urb status received: -ENOENT");
  107. goto skip_error;
  108. case -ECONNRESET:
  109. dev_dbg(&desc->intf->dev,
  110. "nonzero urb status received: -ECONNRESET");
  111. goto skip_error;
  112. case -ESHUTDOWN:
  113. dev_dbg(&desc->intf->dev,
  114. "nonzero urb status received: -ESHUTDOWN");
  115. goto skip_error;
  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. skip_error:
  131. wake_up(&desc->wait);
  132. set_bit(WDM_READ, &desc->flags);
  133. spin_unlock(&desc->iuspin);
  134. }
  135. static void wdm_int_callback(struct urb *urb)
  136. {
  137. int rv = 0;
  138. int status = urb->status;
  139. struct wdm_device *desc;
  140. struct usb_cdc_notification *dr;
  141. desc = urb->context;
  142. dr = (struct usb_cdc_notification *)desc->sbuf;
  143. if (status) {
  144. switch (status) {
  145. case -ESHUTDOWN:
  146. case -ENOENT:
  147. case -ECONNRESET:
  148. return; /* unplug */
  149. case -EPIPE:
  150. set_bit(WDM_INT_STALL, &desc->flags);
  151. dev_err(&desc->intf->dev, "Stall on int endpoint\n");
  152. goto sw; /* halt is cleared in work */
  153. default:
  154. dev_err(&desc->intf->dev,
  155. "nonzero urb status received: %d\n", status);
  156. break;
  157. }
  158. }
  159. if (urb->actual_length < sizeof(struct usb_cdc_notification)) {
  160. dev_err(&desc->intf->dev, "wdm_int_callback - %d bytes\n",
  161. urb->actual_length);
  162. goto exit;
  163. }
  164. switch (dr->bNotificationType) {
  165. case USB_CDC_NOTIFY_RESPONSE_AVAILABLE:
  166. dev_dbg(&desc->intf->dev,
  167. "NOTIFY_RESPONSE_AVAILABLE received: index %d len %d",
  168. dr->wIndex, dr->wLength);
  169. break;
  170. case USB_CDC_NOTIFY_NETWORK_CONNECTION:
  171. dev_dbg(&desc->intf->dev,
  172. "NOTIFY_NETWORK_CONNECTION %s network",
  173. dr->wValue ? "connected to" : "disconnected from");
  174. goto exit;
  175. default:
  176. clear_bit(WDM_POLL_RUNNING, &desc->flags);
  177. dev_err(&desc->intf->dev,
  178. "unknown notification %d received: index %d len %d\n",
  179. dr->bNotificationType, dr->wIndex, dr->wLength);
  180. goto exit;
  181. }
  182. spin_lock(&desc->iuspin);
  183. clear_bit(WDM_READ, &desc->flags);
  184. set_bit(WDM_RESPONDING, &desc->flags);
  185. if (!test_bit(WDM_DISCONNECTING, &desc->flags)
  186. && !test_bit(WDM_SUSPENDING, &desc->flags)) {
  187. rv = usb_submit_urb(desc->response, GFP_ATOMIC);
  188. dev_dbg(&desc->intf->dev, "%s: usb_submit_urb %d",
  189. __func__, rv);
  190. }
  191. spin_unlock(&desc->iuspin);
  192. if (rv < 0) {
  193. clear_bit(WDM_RESPONDING, &desc->flags);
  194. if (rv == -EPERM)
  195. return;
  196. if (rv == -ENOMEM) {
  197. sw:
  198. rv = schedule_work(&desc->rxwork);
  199. if (rv)
  200. dev_err(&desc->intf->dev,
  201. "Cannot schedule work\n");
  202. }
  203. }
  204. exit:
  205. rv = usb_submit_urb(urb, GFP_ATOMIC);
  206. if (rv)
  207. dev_err(&desc->intf->dev,
  208. "%s - usb_submit_urb failed with result %d\n",
  209. __func__, rv);
  210. }
  211. static void kill_urbs(struct wdm_device *desc)
  212. {
  213. /* the order here is essential */
  214. usb_kill_urb(desc->command);
  215. usb_kill_urb(desc->validity);
  216. usb_kill_urb(desc->response);
  217. }
  218. static void free_urbs(struct wdm_device *desc)
  219. {
  220. usb_free_urb(desc->validity);
  221. usb_free_urb(desc->response);
  222. usb_free_urb(desc->command);
  223. }
  224. static void cleanup(struct wdm_device *desc)
  225. {
  226. kfree(desc->sbuf);
  227. kfree(desc->inbuf);
  228. kfree(desc->orq);
  229. kfree(desc->irq);
  230. kfree(desc->ubuf);
  231. free_urbs(desc);
  232. kfree(desc);
  233. }
  234. static ssize_t wdm_write
  235. (struct file *file, const char __user *buffer, size_t count, loff_t *ppos)
  236. {
  237. u8 *buf;
  238. int rv = -EMSGSIZE, r, we;
  239. struct wdm_device *desc = file->private_data;
  240. struct usb_ctrlrequest *req;
  241. if (count > desc->wMaxCommand)
  242. count = desc->wMaxCommand;
  243. spin_lock_irq(&desc->iuspin);
  244. we = desc->werr;
  245. desc->werr = 0;
  246. spin_unlock_irq(&desc->iuspin);
  247. if (we < 0)
  248. return -EIO;
  249. desc->outbuf = buf = kmalloc(count, GFP_KERNEL);
  250. if (!buf) {
  251. rv = -ENOMEM;
  252. goto outnl;
  253. }
  254. r = copy_from_user(buf, buffer, count);
  255. if (r > 0) {
  256. kfree(buf);
  257. rv = -EFAULT;
  258. goto outnl;
  259. }
  260. /* concurrent writes and disconnect */
  261. r = mutex_lock_interruptible(&desc->lock);
  262. rv = -ERESTARTSYS;
  263. if (r) {
  264. kfree(buf);
  265. goto outnl;
  266. }
  267. if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
  268. kfree(buf);
  269. rv = -ENODEV;
  270. goto outnp;
  271. }
  272. r = usb_autopm_get_interface(desc->intf);
  273. if (r < 0) {
  274. kfree(buf);
  275. goto outnp;
  276. }
  277. if (!(file->f_flags & O_NONBLOCK))
  278. r = wait_event_interruptible(desc->wait, !test_bit(WDM_IN_USE,
  279. &desc->flags));
  280. else
  281. if (test_bit(WDM_IN_USE, &desc->flags))
  282. r = -EAGAIN;
  283. if (r < 0) {
  284. kfree(buf);
  285. goto out;
  286. }
  287. req = desc->orq;
  288. usb_fill_control_urb(
  289. desc->command,
  290. interface_to_usbdev(desc->intf),
  291. /* using common endpoint 0 */
  292. usb_sndctrlpipe(interface_to_usbdev(desc->intf), 0),
  293. (unsigned char *)req,
  294. buf,
  295. count,
  296. wdm_out_callback,
  297. desc
  298. );
  299. req->bRequestType = (USB_DIR_OUT | USB_TYPE_CLASS |
  300. USB_RECIP_INTERFACE);
  301. req->bRequest = USB_CDC_SEND_ENCAPSULATED_COMMAND;
  302. req->wValue = 0;
  303. req->wIndex = desc->inum;
  304. req->wLength = cpu_to_le16(count);
  305. set_bit(WDM_IN_USE, &desc->flags);
  306. rv = usb_submit_urb(desc->command, GFP_KERNEL);
  307. if (rv < 0) {
  308. kfree(buf);
  309. clear_bit(WDM_IN_USE, &desc->flags);
  310. dev_err(&desc->intf->dev, "Tx URB error: %d\n", rv);
  311. } else {
  312. dev_dbg(&desc->intf->dev, "Tx URB has been submitted index=%d",
  313. req->wIndex);
  314. }
  315. out:
  316. usb_autopm_put_interface(desc->intf);
  317. outnp:
  318. mutex_unlock(&desc->lock);
  319. outnl:
  320. return rv < 0 ? rv : count;
  321. }
  322. static ssize_t wdm_read
  323. (struct file *file, char __user *buffer, size_t count, loff_t *ppos)
  324. {
  325. int rv, cntr = 0;
  326. int i = 0;
  327. struct wdm_device *desc = file->private_data;
  328. rv = mutex_lock_interruptible(&desc->lock); /*concurrent reads */
  329. if (rv < 0)
  330. return -ERESTARTSYS;
  331. if (desc->length == 0) {
  332. desc->read = 0;
  333. retry:
  334. if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
  335. rv = -ENODEV;
  336. goto err;
  337. }
  338. i++;
  339. if (file->f_flags & O_NONBLOCK) {
  340. if (!test_bit(WDM_READ, &desc->flags)) {
  341. rv = cntr ? cntr : -EAGAIN;
  342. goto err;
  343. }
  344. rv = 0;
  345. } else {
  346. rv = wait_event_interruptible(desc->wait,
  347. test_bit(WDM_READ, &desc->flags));
  348. }
  349. /* may have happened while we slept */
  350. if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
  351. rv = -ENODEV;
  352. goto err;
  353. }
  354. usb_mark_last_busy(interface_to_usbdev(desc->intf));
  355. if (rv < 0) {
  356. rv = -ERESTARTSYS;
  357. goto err;
  358. }
  359. spin_lock_irq(&desc->iuspin);
  360. if (desc->rerr) { /* read completed, error happened */
  361. desc->rerr = 0;
  362. spin_unlock_irq(&desc->iuspin);
  363. rv = -EIO;
  364. goto err;
  365. }
  366. /*
  367. * recheck whether we've lost the race
  368. * against the completion handler
  369. */
  370. if (!test_bit(WDM_READ, &desc->flags)) { /* lost race */
  371. spin_unlock_irq(&desc->iuspin);
  372. goto retry;
  373. }
  374. if (!desc->reslength) { /* zero length read */
  375. spin_unlock_irq(&desc->iuspin);
  376. goto retry;
  377. }
  378. clear_bit(WDM_READ, &desc->flags);
  379. spin_unlock_irq(&desc->iuspin);
  380. }
  381. cntr = count > desc->length ? desc->length : count;
  382. rv = copy_to_user(buffer, desc->ubuf, cntr);
  383. if (rv > 0) {
  384. rv = -EFAULT;
  385. goto err;
  386. }
  387. for (i = 0; i < desc->length - cntr; i++)
  388. desc->ubuf[i] = desc->ubuf[i + cntr];
  389. desc->length -= cntr;
  390. /* in case we had outstanding data */
  391. if (!desc->length)
  392. clear_bit(WDM_READ, &desc->flags);
  393. rv = cntr;
  394. err:
  395. mutex_unlock(&desc->lock);
  396. return rv;
  397. }
  398. static int wdm_flush(struct file *file, fl_owner_t id)
  399. {
  400. struct wdm_device *desc = file->private_data;
  401. wait_event(desc->wait, !test_bit(WDM_IN_USE, &desc->flags));
  402. if (desc->werr < 0)
  403. dev_err(&desc->intf->dev, "Error in flush path: %d\n",
  404. desc->werr);
  405. return desc->werr;
  406. }
  407. static unsigned int wdm_poll(struct file *file, struct poll_table_struct *wait)
  408. {
  409. struct wdm_device *desc = file->private_data;
  410. unsigned long flags;
  411. unsigned int mask = 0;
  412. spin_lock_irqsave(&desc->iuspin, flags);
  413. if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
  414. mask = POLLERR;
  415. spin_unlock_irqrestore(&desc->iuspin, flags);
  416. goto desc_out;
  417. }
  418. if (test_bit(WDM_READ, &desc->flags))
  419. mask = POLLIN | POLLRDNORM;
  420. if (desc->rerr || desc->werr)
  421. mask |= POLLERR;
  422. if (!test_bit(WDM_IN_USE, &desc->flags))
  423. mask |= POLLOUT | POLLWRNORM;
  424. spin_unlock_irqrestore(&desc->iuspin, flags);
  425. poll_wait(file, &desc->wait, wait);
  426. desc_out:
  427. return mask;
  428. }
  429. static int wdm_open(struct inode *inode, struct file *file)
  430. {
  431. int minor = iminor(inode);
  432. int rv = -ENODEV;
  433. struct usb_interface *intf;
  434. struct wdm_device *desc;
  435. mutex_lock(&wdm_mutex);
  436. intf = usb_find_interface(&wdm_driver, minor);
  437. if (!intf)
  438. goto out;
  439. desc = usb_get_intfdata(intf);
  440. if (test_bit(WDM_DISCONNECTING, &desc->flags))
  441. goto out;
  442. file->private_data = desc;
  443. rv = usb_autopm_get_interface(desc->intf);
  444. if (rv < 0) {
  445. dev_err(&desc->intf->dev, "Error autopm - %d\n", rv);
  446. goto out;
  447. }
  448. intf->needs_remote_wakeup = 1;
  449. mutex_lock(&desc->lock);
  450. if (!desc->count++) {
  451. desc->werr = 0;
  452. desc->rerr = 0;
  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->lock);
  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->lock);
  473. desc->count--;
  474. mutex_unlock(&desc->lock);
  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. .llseek = noop_llseek,
  493. };
  494. static struct usb_class_driver wdm_class = {
  495. .name = "cdc-wdm%d",
  496. .fops = &wdm_fops,
  497. .minor_base = WDM_MINOR_BASE,
  498. };
  499. /* --- error handling --- */
  500. static void wdm_rxwork(struct work_struct *work)
  501. {
  502. struct wdm_device *desc = container_of(work, struct wdm_device, rxwork);
  503. unsigned long flags;
  504. int rv;
  505. spin_lock_irqsave(&desc->iuspin, flags);
  506. if (test_bit(WDM_DISCONNECTING, &desc->flags)) {
  507. spin_unlock_irqrestore(&desc->iuspin, flags);
  508. } else {
  509. spin_unlock_irqrestore(&desc->iuspin, flags);
  510. rv = usb_submit_urb(desc->response, GFP_KERNEL);
  511. if (rv < 0 && rv != -EPERM) {
  512. spin_lock_irqsave(&desc->iuspin, flags);
  513. if (!test_bit(WDM_DISCONNECTING, &desc->flags))
  514. schedule_work(&desc->rxwork);
  515. spin_unlock_irqrestore(&desc->iuspin, flags);
  516. }
  517. }
  518. }
  519. /* --- hotplug --- */
  520. static int wdm_probe(struct usb_interface *intf, const struct usb_device_id *id)
  521. {
  522. int rv = -EINVAL;
  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->lock);
  561. spin_lock_init(&desc->iuspin);
  562. init_waitqueue_head(&desc->wait);
  563. desc->wMaxCommand = maxcom;
  564. /* this will be expanded and needed in hardware endianness */
  565. desc->inum = cpu_to_le16((u16)intf->cur_altsetting->desc.bInterfaceNumber);
  566. desc->intf = intf;
  567. INIT_WORK(&desc->rxwork, wdm_rxwork);
  568. rv = -EINVAL;
  569. iface = intf->cur_altsetting;
  570. if (iface->desc.bNumEndpoints != 1)
  571. goto err;
  572. ep = &iface->endpoint[0].desc;
  573. if (!ep || !usb_endpoint_is_int_in(ep))
  574. goto err;
  575. desc->wMaxPacketSize = usb_endpoint_maxp(ep);
  576. desc->orq = kmalloc(sizeof(struct usb_ctrlrequest), GFP_KERNEL);
  577. if (!desc->orq)
  578. goto err;
  579. desc->irq = kmalloc(sizeof(struct usb_ctrlrequest), GFP_KERNEL);
  580. if (!desc->irq)
  581. goto err;
  582. desc->validity = usb_alloc_urb(0, GFP_KERNEL);
  583. if (!desc->validity)
  584. goto err;
  585. desc->response = usb_alloc_urb(0, GFP_KERNEL);
  586. if (!desc->response)
  587. goto err;
  588. desc->command = usb_alloc_urb(0, GFP_KERNEL);
  589. if (!desc->command)
  590. goto err;
  591. desc->ubuf = kmalloc(desc->wMaxCommand, GFP_KERNEL);
  592. if (!desc->ubuf)
  593. goto err;
  594. desc->sbuf = kmalloc(desc->wMaxPacketSize, GFP_KERNEL);
  595. if (!desc->sbuf)
  596. goto err;
  597. desc->inbuf = kmalloc(desc->wMaxCommand, GFP_KERNEL);
  598. if (!desc->inbuf)
  599. goto err;
  600. usb_fill_int_urb(
  601. desc->validity,
  602. interface_to_usbdev(intf),
  603. usb_rcvintpipe(interface_to_usbdev(intf), ep->bEndpointAddress),
  604. desc->sbuf,
  605. desc->wMaxPacketSize,
  606. wdm_int_callback,
  607. desc,
  608. ep->bInterval
  609. );
  610. desc->irq->bRequestType = (USB_DIR_IN | USB_TYPE_CLASS | USB_RECIP_INTERFACE);
  611. desc->irq->bRequest = USB_CDC_GET_ENCAPSULATED_RESPONSE;
  612. desc->irq->wValue = 0;
  613. desc->irq->wIndex = desc->inum;
  614. desc->irq->wLength = cpu_to_le16(desc->wMaxCommand);
  615. usb_fill_control_urb(
  616. desc->response,
  617. interface_to_usbdev(intf),
  618. /* using common endpoint 0 */
  619. usb_rcvctrlpipe(interface_to_usbdev(desc->intf), 0),
  620. (unsigned char *)desc->irq,
  621. desc->inbuf,
  622. desc->wMaxCommand,
  623. wdm_in_callback,
  624. desc
  625. );
  626. usb_set_intfdata(intf, desc);
  627. rv = usb_register_dev(intf, &wdm_class);
  628. if (rv < 0)
  629. goto err2;
  630. else
  631. dev_info(&intf->dev, "cdc-wdm%d: USB WDM device\n",
  632. intf->minor - WDM_MINOR_BASE);
  633. out:
  634. return rv;
  635. err2:
  636. usb_set_intfdata(intf, NULL);
  637. err:
  638. free_urbs(desc);
  639. kfree(desc->inbuf);
  640. kfree(desc->sbuf);
  641. kfree(desc->ubuf);
  642. kfree(desc->orq);
  643. kfree(desc->irq);
  644. kfree(desc);
  645. return rv;
  646. }
  647. static void wdm_disconnect(struct usb_interface *intf)
  648. {
  649. struct wdm_device *desc;
  650. unsigned long flags;
  651. usb_deregister_dev(intf, &wdm_class);
  652. mutex_lock(&wdm_mutex);
  653. desc = usb_get_intfdata(intf);
  654. /* the spinlock makes sure no new urbs are generated in the callbacks */
  655. spin_lock_irqsave(&desc->iuspin, flags);
  656. set_bit(WDM_DISCONNECTING, &desc->flags);
  657. set_bit(WDM_READ, &desc->flags);
  658. /* to terminate pending flushes */
  659. clear_bit(WDM_IN_USE, &desc->flags);
  660. spin_unlock_irqrestore(&desc->iuspin, flags);
  661. mutex_lock(&desc->lock);
  662. kill_urbs(desc);
  663. cancel_work_sync(&desc->rxwork);
  664. mutex_unlock(&desc->lock);
  665. wake_up_all(&desc->wait);
  666. if (!desc->count)
  667. cleanup(desc);
  668. mutex_unlock(&wdm_mutex);
  669. }
  670. #ifdef CONFIG_PM
  671. static int wdm_suspend(struct usb_interface *intf, pm_message_t message)
  672. {
  673. struct wdm_device *desc = usb_get_intfdata(intf);
  674. int rv = 0;
  675. dev_dbg(&desc->intf->dev, "wdm%d_suspend\n", intf->minor);
  676. /* if this is an autosuspend the caller does the locking */
  677. if (!PMSG_IS_AUTO(message))
  678. mutex_lock(&desc->lock);
  679. spin_lock_irq(&desc->iuspin);
  680. if (PMSG_IS_AUTO(message) &&
  681. (test_bit(WDM_IN_USE, &desc->flags)
  682. || test_bit(WDM_RESPONDING, &desc->flags))) {
  683. spin_unlock_irq(&desc->iuspin);
  684. rv = -EBUSY;
  685. } else {
  686. set_bit(WDM_SUSPENDING, &desc->flags);
  687. spin_unlock_irq(&desc->iuspin);
  688. /* callback submits work - order is essential */
  689. kill_urbs(desc);
  690. cancel_work_sync(&desc->rxwork);
  691. }
  692. if (!PMSG_IS_AUTO(message))
  693. mutex_unlock(&desc->lock);
  694. return rv;
  695. }
  696. #endif
  697. static int recover_from_urb_loss(struct wdm_device *desc)
  698. {
  699. int rv = 0;
  700. if (desc->count) {
  701. rv = usb_submit_urb(desc->validity, GFP_NOIO);
  702. if (rv < 0)
  703. dev_err(&desc->intf->dev,
  704. "Error resume submitting int urb - %d\n", rv);
  705. }
  706. return rv;
  707. }
  708. #ifdef CONFIG_PM
  709. static int wdm_resume(struct usb_interface *intf)
  710. {
  711. struct wdm_device *desc = usb_get_intfdata(intf);
  712. int rv;
  713. dev_dbg(&desc->intf->dev, "wdm%d_resume\n", intf->minor);
  714. clear_bit(WDM_SUSPENDING, &desc->flags);
  715. rv = recover_from_urb_loss(desc);
  716. return rv;
  717. }
  718. #endif
  719. static int wdm_pre_reset(struct usb_interface *intf)
  720. {
  721. struct wdm_device *desc = usb_get_intfdata(intf);
  722. mutex_lock(&desc->lock);
  723. kill_urbs(desc);
  724. /*
  725. * we notify everybody using poll of
  726. * an exceptional situation
  727. * must be done before recovery lest a spontaneous
  728. * message from the device is lost
  729. */
  730. spin_lock_irq(&desc->iuspin);
  731. desc->rerr = -EINTR;
  732. spin_unlock_irq(&desc->iuspin);
  733. wake_up_all(&desc->wait);
  734. return 0;
  735. }
  736. static int wdm_post_reset(struct usb_interface *intf)
  737. {
  738. struct wdm_device *desc = usb_get_intfdata(intf);
  739. int rv;
  740. rv = recover_from_urb_loss(desc);
  741. mutex_unlock(&desc->lock);
  742. return 0;
  743. }
  744. static struct usb_driver wdm_driver = {
  745. .name = "cdc_wdm",
  746. .probe = wdm_probe,
  747. .disconnect = wdm_disconnect,
  748. #ifdef CONFIG_PM
  749. .suspend = wdm_suspend,
  750. .resume = wdm_resume,
  751. .reset_resume = wdm_resume,
  752. #endif
  753. .pre_reset = wdm_pre_reset,
  754. .post_reset = wdm_post_reset,
  755. .id_table = wdm_ids,
  756. .supports_autosuspend = 1,
  757. };
  758. module_usb_driver(wdm_driver);
  759. MODULE_AUTHOR(DRIVER_AUTHOR);
  760. MODULE_DESCRIPTION(DRIVER_DESC);
  761. MODULE_LICENSE("GPL");