adutux.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835
  1. /*
  2. * adutux - driver for ADU devices from Ontrak Control Systems
  3. * This is an experimental driver. Use at your own risk.
  4. * This driver is not supported by Ontrak Control Systems.
  5. *
  6. * Copyright (c) 2003 John Homppi (SCO, leave this notice here)
  7. *
  8. * This program is free software; you can redistribute it and/or
  9. * modify it under the terms of the GNU General Public License as
  10. * published by the Free Software Foundation; either version 2 of
  11. * the License, or (at your option) any later version.
  12. *
  13. * derived from the Lego USB Tower driver 0.56:
  14. * Copyright (c) 2003 David Glance <davidgsf@sourceforge.net>
  15. * 2001 Juergen Stuber <stuber@loria.fr>
  16. * that was derived from USB Skeleton driver - 0.5
  17. * Copyright (c) 2001 Greg Kroah-Hartman (greg@kroah.com)
  18. *
  19. */
  20. #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
  21. #include <linux/kernel.h>
  22. #include <linux/errno.h>
  23. #include <linux/slab.h>
  24. #include <linux/module.h>
  25. #include <linux/usb.h>
  26. #include <linux/mutex.h>
  27. #include <linux/uaccess.h>
  28. /* Version Information */
  29. #define DRIVER_VERSION "v0.0.13"
  30. #define DRIVER_AUTHOR "John Homppi"
  31. #define DRIVER_DESC "adutux (see www.ontrak.net)"
  32. /* Define these values to match your device */
  33. #define ADU_VENDOR_ID 0x0a07
  34. #define ADU_PRODUCT_ID 0x0064
  35. /* table of devices that work with this driver */
  36. static const struct usb_device_id device_table[] = {
  37. { USB_DEVICE(ADU_VENDOR_ID, ADU_PRODUCT_ID) }, /* ADU100 */
  38. { USB_DEVICE(ADU_VENDOR_ID, ADU_PRODUCT_ID+20) }, /* ADU120 */
  39. { USB_DEVICE(ADU_VENDOR_ID, ADU_PRODUCT_ID+30) }, /* ADU130 */
  40. { USB_DEVICE(ADU_VENDOR_ID, ADU_PRODUCT_ID+100) }, /* ADU200 */
  41. { USB_DEVICE(ADU_VENDOR_ID, ADU_PRODUCT_ID+108) }, /* ADU208 */
  42. { USB_DEVICE(ADU_VENDOR_ID, ADU_PRODUCT_ID+118) }, /* ADU218 */
  43. { } /* Terminating entry */
  44. };
  45. MODULE_DEVICE_TABLE(usb, device_table);
  46. #ifdef CONFIG_USB_DYNAMIC_MINORS
  47. #define ADU_MINOR_BASE 0
  48. #else
  49. #define ADU_MINOR_BASE 67
  50. #endif
  51. /* we can have up to this number of device plugged in at once */
  52. #define MAX_DEVICES 16
  53. #define COMMAND_TIMEOUT (2*HZ) /* 60 second timeout for a command */
  54. /*
  55. * The locking scheme is a vanilla 3-lock:
  56. * adu_device.buflock: A spinlock, covers what IRQs touch.
  57. * adutux_mutex: A Static lock to cover open_count. It would also cover
  58. * any globals, but we don't have them in 2.6.
  59. * adu_device.mtx: A mutex to hold across sleepers like copy_from_user.
  60. * It covers all of adu_device, except the open_count
  61. * and what .buflock covers.
  62. */
  63. /* Structure to hold all of our device specific stuff */
  64. struct adu_device {
  65. struct mutex mtx;
  66. struct usb_device *udev; /* save off the usb device pointer */
  67. struct usb_interface *interface;
  68. unsigned int minor; /* the starting minor number for this device */
  69. char serial_number[8];
  70. int open_count; /* number of times this port has been opened */
  71. char *read_buffer_primary;
  72. int read_buffer_length;
  73. char *read_buffer_secondary;
  74. int secondary_head;
  75. int secondary_tail;
  76. spinlock_t buflock;
  77. wait_queue_head_t read_wait;
  78. wait_queue_head_t write_wait;
  79. char *interrupt_in_buffer;
  80. struct usb_endpoint_descriptor *interrupt_in_endpoint;
  81. struct urb *interrupt_in_urb;
  82. int read_urb_finished;
  83. char *interrupt_out_buffer;
  84. struct usb_endpoint_descriptor *interrupt_out_endpoint;
  85. struct urb *interrupt_out_urb;
  86. int out_urb_finished;
  87. };
  88. static DEFINE_MUTEX(adutux_mutex);
  89. static struct usb_driver adu_driver;
  90. static inline void adu_debug_data(struct device *dev, const char *function,
  91. int size, const unsigned char *data)
  92. {
  93. dev_dbg(dev, "%s - length = %d, data = %*ph\n",
  94. function, size, size, data);
  95. }
  96. /**
  97. * adu_abort_transfers
  98. * aborts transfers and frees associated data structures
  99. */
  100. static void adu_abort_transfers(struct adu_device *dev)
  101. {
  102. unsigned long flags;
  103. if (dev->udev == NULL)
  104. return;
  105. /* shutdown transfer */
  106. /* XXX Anchor these instead */
  107. spin_lock_irqsave(&dev->buflock, flags);
  108. if (!dev->read_urb_finished) {
  109. spin_unlock_irqrestore(&dev->buflock, flags);
  110. usb_kill_urb(dev->interrupt_in_urb);
  111. } else
  112. spin_unlock_irqrestore(&dev->buflock, flags);
  113. spin_lock_irqsave(&dev->buflock, flags);
  114. if (!dev->out_urb_finished) {
  115. spin_unlock_irqrestore(&dev->buflock, flags);
  116. usb_kill_urb(dev->interrupt_out_urb);
  117. } else
  118. spin_unlock_irqrestore(&dev->buflock, flags);
  119. }
  120. static void adu_delete(struct adu_device *dev)
  121. {
  122. /* free data structures */
  123. usb_free_urb(dev->interrupt_in_urb);
  124. usb_free_urb(dev->interrupt_out_urb);
  125. kfree(dev->read_buffer_primary);
  126. kfree(dev->read_buffer_secondary);
  127. kfree(dev->interrupt_in_buffer);
  128. kfree(dev->interrupt_out_buffer);
  129. kfree(dev);
  130. }
  131. static void adu_interrupt_in_callback(struct urb *urb)
  132. {
  133. struct adu_device *dev = urb->context;
  134. int status = urb->status;
  135. adu_debug_data(&dev->udev->dev, __func__,
  136. urb->actual_length, urb->transfer_buffer);
  137. spin_lock(&dev->buflock);
  138. if (status != 0) {
  139. if ((status != -ENOENT) && (status != -ECONNRESET) &&
  140. (status != -ESHUTDOWN)) {
  141. dev_dbg(&dev->udev->dev,
  142. "%s : nonzero status received: %d\n",
  143. __func__, status);
  144. }
  145. goto exit;
  146. }
  147. if (urb->actual_length > 0 && dev->interrupt_in_buffer[0] != 0x00) {
  148. if (dev->read_buffer_length <
  149. (4 * usb_endpoint_maxp(dev->interrupt_in_endpoint)) -
  150. (urb->actual_length)) {
  151. memcpy (dev->read_buffer_primary +
  152. dev->read_buffer_length,
  153. dev->interrupt_in_buffer, urb->actual_length);
  154. dev->read_buffer_length += urb->actual_length;
  155. dev_dbg(&dev->udev->dev,"%s reading %d\n", __func__,
  156. urb->actual_length);
  157. } else {
  158. dev_dbg(&dev->udev->dev,"%s : read_buffer overflow\n",
  159. __func__);
  160. }
  161. }
  162. exit:
  163. dev->read_urb_finished = 1;
  164. spin_unlock(&dev->buflock);
  165. /* always wake up so we recover from errors */
  166. wake_up_interruptible(&dev->read_wait);
  167. }
  168. static void adu_interrupt_out_callback(struct urb *urb)
  169. {
  170. struct adu_device *dev = urb->context;
  171. int status = urb->status;
  172. adu_debug_data(&dev->udev->dev, __func__,
  173. urb->actual_length, urb->transfer_buffer);
  174. if (status != 0) {
  175. if ((status != -ENOENT) &&
  176. (status != -ECONNRESET)) {
  177. dev_dbg(&dev->udev->dev,
  178. "%s :nonzero status received: %d\n", __func__,
  179. status);
  180. }
  181. return;
  182. }
  183. spin_lock(&dev->buflock);
  184. dev->out_urb_finished = 1;
  185. wake_up(&dev->write_wait);
  186. spin_unlock(&dev->buflock);
  187. }
  188. static int adu_open(struct inode *inode, struct file *file)
  189. {
  190. struct adu_device *dev = NULL;
  191. struct usb_interface *interface;
  192. int subminor;
  193. int retval;
  194. subminor = iminor(inode);
  195. retval = mutex_lock_interruptible(&adutux_mutex);
  196. if (retval)
  197. goto exit_no_lock;
  198. interface = usb_find_interface(&adu_driver, subminor);
  199. if (!interface) {
  200. pr_err("%s - error, can't find device for minor %d\n",
  201. __func__, subminor);
  202. retval = -ENODEV;
  203. goto exit_no_device;
  204. }
  205. dev = usb_get_intfdata(interface);
  206. if (!dev || !dev->udev) {
  207. retval = -ENODEV;
  208. goto exit_no_device;
  209. }
  210. /* check that nobody else is using the device */
  211. if (dev->open_count) {
  212. retval = -EBUSY;
  213. goto exit_no_device;
  214. }
  215. ++dev->open_count;
  216. dev_dbg(&dev->udev->dev, "%s: open count %d\n", __func__,
  217. dev->open_count);
  218. /* save device in the file's private structure */
  219. file->private_data = dev;
  220. /* initialize in direction */
  221. dev->read_buffer_length = 0;
  222. /* fixup first read by having urb waiting for it */
  223. usb_fill_int_urb(dev->interrupt_in_urb, dev->udev,
  224. usb_rcvintpipe(dev->udev,
  225. dev->interrupt_in_endpoint->bEndpointAddress),
  226. dev->interrupt_in_buffer,
  227. usb_endpoint_maxp(dev->interrupt_in_endpoint),
  228. adu_interrupt_in_callback, dev,
  229. dev->interrupt_in_endpoint->bInterval);
  230. dev->read_urb_finished = 0;
  231. if (usb_submit_urb(dev->interrupt_in_urb, GFP_KERNEL))
  232. dev->read_urb_finished = 1;
  233. /* we ignore failure */
  234. /* end of fixup for first read */
  235. /* initialize out direction */
  236. dev->out_urb_finished = 1;
  237. retval = 0;
  238. exit_no_device:
  239. mutex_unlock(&adutux_mutex);
  240. exit_no_lock:
  241. return retval;
  242. }
  243. static void adu_release_internal(struct adu_device *dev)
  244. {
  245. /* decrement our usage count for the device */
  246. --dev->open_count;
  247. dev_dbg(&dev->udev->dev, "%s : open count %d\n", __func__,
  248. dev->open_count);
  249. if (dev->open_count <= 0) {
  250. adu_abort_transfers(dev);
  251. dev->open_count = 0;
  252. }
  253. }
  254. static int adu_release(struct inode *inode, struct file *file)
  255. {
  256. struct adu_device *dev;
  257. int retval = 0;
  258. if (file == NULL) {
  259. retval = -ENODEV;
  260. goto exit;
  261. }
  262. dev = file->private_data;
  263. if (dev == NULL) {
  264. retval = -ENODEV;
  265. goto exit;
  266. }
  267. mutex_lock(&adutux_mutex); /* not interruptible */
  268. if (dev->open_count <= 0) {
  269. dev_dbg(&dev->udev->dev, "%s : device not opened\n", __func__);
  270. retval = -ENODEV;
  271. goto unlock;
  272. }
  273. adu_release_internal(dev);
  274. if (dev->udev == NULL) {
  275. /* the device was unplugged before the file was released */
  276. if (!dev->open_count) /* ... and we're the last user */
  277. adu_delete(dev);
  278. }
  279. unlock:
  280. mutex_unlock(&adutux_mutex);
  281. exit:
  282. return retval;
  283. }
  284. static ssize_t adu_read(struct file *file, __user char *buffer, size_t count,
  285. loff_t *ppos)
  286. {
  287. struct adu_device *dev;
  288. size_t bytes_read = 0;
  289. size_t bytes_to_read = count;
  290. int i;
  291. int retval = 0;
  292. int timeout = 0;
  293. int should_submit = 0;
  294. unsigned long flags;
  295. DECLARE_WAITQUEUE(wait, current);
  296. dev = file->private_data;
  297. if (mutex_lock_interruptible(&dev->mtx))
  298. return -ERESTARTSYS;
  299. /* verify that the device wasn't unplugged */
  300. if (dev->udev == NULL) {
  301. retval = -ENODEV;
  302. pr_err("No device or device unplugged %d\n", retval);
  303. goto exit;
  304. }
  305. /* verify that some data was requested */
  306. if (count == 0) {
  307. dev_dbg(&dev->udev->dev, "%s : read request of 0 bytes\n",
  308. __func__);
  309. goto exit;
  310. }
  311. timeout = COMMAND_TIMEOUT;
  312. dev_dbg(&dev->udev->dev, "%s : about to start looping\n", __func__);
  313. while (bytes_to_read) {
  314. int data_in_secondary = dev->secondary_tail - dev->secondary_head;
  315. dev_dbg(&dev->udev->dev,
  316. "%s : while, data_in_secondary=%d, status=%d\n",
  317. __func__, data_in_secondary,
  318. dev->interrupt_in_urb->status);
  319. if (data_in_secondary) {
  320. /* drain secondary buffer */
  321. int amount = bytes_to_read < data_in_secondary ? bytes_to_read : data_in_secondary;
  322. i = copy_to_user(buffer, dev->read_buffer_secondary+dev->secondary_head, amount);
  323. if (i) {
  324. retval = -EFAULT;
  325. goto exit;
  326. }
  327. dev->secondary_head += (amount - i);
  328. bytes_read += (amount - i);
  329. bytes_to_read -= (amount - i);
  330. if (i) {
  331. retval = bytes_read ? bytes_read : -EFAULT;
  332. goto exit;
  333. }
  334. } else {
  335. /* we check the primary buffer */
  336. spin_lock_irqsave (&dev->buflock, flags);
  337. if (dev->read_buffer_length) {
  338. /* we secure access to the primary */
  339. char *tmp;
  340. dev_dbg(&dev->udev->dev,
  341. "%s : swap, read_buffer_length = %d\n",
  342. __func__, dev->read_buffer_length);
  343. tmp = dev->read_buffer_secondary;
  344. dev->read_buffer_secondary = dev->read_buffer_primary;
  345. dev->read_buffer_primary = tmp;
  346. dev->secondary_head = 0;
  347. dev->secondary_tail = dev->read_buffer_length;
  348. dev->read_buffer_length = 0;
  349. spin_unlock_irqrestore(&dev->buflock, flags);
  350. /* we have a free buffer so use it */
  351. should_submit = 1;
  352. } else {
  353. /* even the primary was empty - we may need to do IO */
  354. if (!dev->read_urb_finished) {
  355. /* somebody is doing IO */
  356. spin_unlock_irqrestore(&dev->buflock, flags);
  357. dev_dbg(&dev->udev->dev,
  358. "%s : submitted already\n",
  359. __func__);
  360. } else {
  361. /* we must initiate input */
  362. dev_dbg(&dev->udev->dev,
  363. "%s : initiate input\n",
  364. __func__);
  365. dev->read_urb_finished = 0;
  366. spin_unlock_irqrestore(&dev->buflock, flags);
  367. usb_fill_int_urb(dev->interrupt_in_urb, dev->udev,
  368. usb_rcvintpipe(dev->udev,
  369. dev->interrupt_in_endpoint->bEndpointAddress),
  370. dev->interrupt_in_buffer,
  371. usb_endpoint_maxp(dev->interrupt_in_endpoint),
  372. adu_interrupt_in_callback,
  373. dev,
  374. dev->interrupt_in_endpoint->bInterval);
  375. retval = usb_submit_urb(dev->interrupt_in_urb, GFP_KERNEL);
  376. if (retval) {
  377. dev->read_urb_finished = 1;
  378. if (retval == -ENOMEM) {
  379. retval = bytes_read ? bytes_read : -ENOMEM;
  380. }
  381. dev_dbg(&dev->udev->dev,
  382. "%s : submit failed\n",
  383. __func__);
  384. goto exit;
  385. }
  386. }
  387. /* we wait for I/O to complete */
  388. set_current_state(TASK_INTERRUPTIBLE);
  389. add_wait_queue(&dev->read_wait, &wait);
  390. spin_lock_irqsave(&dev->buflock, flags);
  391. if (!dev->read_urb_finished) {
  392. spin_unlock_irqrestore(&dev->buflock, flags);
  393. timeout = schedule_timeout(COMMAND_TIMEOUT);
  394. } else {
  395. spin_unlock_irqrestore(&dev->buflock, flags);
  396. set_current_state(TASK_RUNNING);
  397. }
  398. remove_wait_queue(&dev->read_wait, &wait);
  399. if (timeout <= 0) {
  400. dev_dbg(&dev->udev->dev,
  401. "%s : timeout\n", __func__);
  402. retval = bytes_read ? bytes_read : -ETIMEDOUT;
  403. goto exit;
  404. }
  405. if (signal_pending(current)) {
  406. dev_dbg(&dev->udev->dev,
  407. "%s : signal pending\n",
  408. __func__);
  409. retval = bytes_read ? bytes_read : -EINTR;
  410. goto exit;
  411. }
  412. }
  413. }
  414. }
  415. retval = bytes_read;
  416. /* if the primary buffer is empty then use it */
  417. spin_lock_irqsave(&dev->buflock, flags);
  418. if (should_submit && dev->read_urb_finished) {
  419. dev->read_urb_finished = 0;
  420. spin_unlock_irqrestore(&dev->buflock, flags);
  421. usb_fill_int_urb(dev->interrupt_in_urb, dev->udev,
  422. usb_rcvintpipe(dev->udev,
  423. dev->interrupt_in_endpoint->bEndpointAddress),
  424. dev->interrupt_in_buffer,
  425. usb_endpoint_maxp(dev->interrupt_in_endpoint),
  426. adu_interrupt_in_callback,
  427. dev,
  428. dev->interrupt_in_endpoint->bInterval);
  429. if (usb_submit_urb(dev->interrupt_in_urb, GFP_KERNEL) != 0)
  430. dev->read_urb_finished = 1;
  431. /* we ignore failure */
  432. } else {
  433. spin_unlock_irqrestore(&dev->buflock, flags);
  434. }
  435. exit:
  436. /* unlock the device */
  437. mutex_unlock(&dev->mtx);
  438. return retval;
  439. }
  440. static ssize_t adu_write(struct file *file, const __user char *buffer,
  441. size_t count, loff_t *ppos)
  442. {
  443. DECLARE_WAITQUEUE(waita, current);
  444. struct adu_device *dev;
  445. size_t bytes_written = 0;
  446. size_t bytes_to_write;
  447. size_t buffer_size;
  448. unsigned long flags;
  449. int retval;
  450. dev = file->private_data;
  451. retval = mutex_lock_interruptible(&dev->mtx);
  452. if (retval)
  453. goto exit_nolock;
  454. /* verify that the device wasn't unplugged */
  455. if (dev->udev == NULL) {
  456. retval = -ENODEV;
  457. pr_err("No device or device unplugged %d\n", retval);
  458. goto exit;
  459. }
  460. /* verify that we actually have some data to write */
  461. if (count == 0) {
  462. dev_dbg(&dev->udev->dev, "%s : write request of 0 bytes\n",
  463. __func__);
  464. goto exit;
  465. }
  466. while (count > 0) {
  467. add_wait_queue(&dev->write_wait, &waita);
  468. set_current_state(TASK_INTERRUPTIBLE);
  469. spin_lock_irqsave(&dev->buflock, flags);
  470. if (!dev->out_urb_finished) {
  471. spin_unlock_irqrestore(&dev->buflock, flags);
  472. mutex_unlock(&dev->mtx);
  473. if (signal_pending(current)) {
  474. dev_dbg(&dev->udev->dev, "%s : interrupted\n",
  475. __func__);
  476. set_current_state(TASK_RUNNING);
  477. retval = -EINTR;
  478. goto exit_onqueue;
  479. }
  480. if (schedule_timeout(COMMAND_TIMEOUT) == 0) {
  481. dev_dbg(&dev->udev->dev,
  482. "%s - command timed out.\n", __func__);
  483. retval = -ETIMEDOUT;
  484. goto exit_onqueue;
  485. }
  486. remove_wait_queue(&dev->write_wait, &waita);
  487. retval = mutex_lock_interruptible(&dev->mtx);
  488. if (retval) {
  489. retval = bytes_written ? bytes_written : retval;
  490. goto exit_nolock;
  491. }
  492. dev_dbg(&dev->udev->dev,
  493. "%s : in progress, count = %Zd\n",
  494. __func__, count);
  495. } else {
  496. spin_unlock_irqrestore(&dev->buflock, flags);
  497. set_current_state(TASK_RUNNING);
  498. remove_wait_queue(&dev->write_wait, &waita);
  499. dev_dbg(&dev->udev->dev, "%s : sending, count = %Zd\n",
  500. __func__, count);
  501. /* write the data into interrupt_out_buffer from userspace */
  502. buffer_size = usb_endpoint_maxp(dev->interrupt_out_endpoint);
  503. bytes_to_write = count > buffer_size ? buffer_size : count;
  504. dev_dbg(&dev->udev->dev,
  505. "%s : buffer_size = %Zd, count = %Zd, bytes_to_write = %Zd\n",
  506. __func__, buffer_size, count, bytes_to_write);
  507. if (copy_from_user(dev->interrupt_out_buffer, buffer, bytes_to_write) != 0) {
  508. retval = -EFAULT;
  509. goto exit;
  510. }
  511. /* send off the urb */
  512. usb_fill_int_urb(
  513. dev->interrupt_out_urb,
  514. dev->udev,
  515. usb_sndintpipe(dev->udev, dev->interrupt_out_endpoint->bEndpointAddress),
  516. dev->interrupt_out_buffer,
  517. bytes_to_write,
  518. adu_interrupt_out_callback,
  519. dev,
  520. dev->interrupt_out_endpoint->bInterval);
  521. dev->interrupt_out_urb->actual_length = bytes_to_write;
  522. dev->out_urb_finished = 0;
  523. retval = usb_submit_urb(dev->interrupt_out_urb, GFP_KERNEL);
  524. if (retval < 0) {
  525. dev->out_urb_finished = 1;
  526. dev_err(&dev->udev->dev, "Couldn't submit "
  527. "interrupt_out_urb %d\n", retval);
  528. goto exit;
  529. }
  530. buffer += bytes_to_write;
  531. count -= bytes_to_write;
  532. bytes_written += bytes_to_write;
  533. }
  534. }
  535. mutex_unlock(&dev->mtx);
  536. return bytes_written;
  537. exit:
  538. mutex_unlock(&dev->mtx);
  539. exit_nolock:
  540. return retval;
  541. exit_onqueue:
  542. remove_wait_queue(&dev->write_wait, &waita);
  543. return retval;
  544. }
  545. /* file operations needed when we register this driver */
  546. static const struct file_operations adu_fops = {
  547. .owner = THIS_MODULE,
  548. .read = adu_read,
  549. .write = adu_write,
  550. .open = adu_open,
  551. .release = adu_release,
  552. .llseek = noop_llseek,
  553. };
  554. /*
  555. * usb class driver info in order to get a minor number from the usb core,
  556. * and to have the device registered with devfs and the driver core
  557. */
  558. static struct usb_class_driver adu_class = {
  559. .name = "usb/adutux%d",
  560. .fops = &adu_fops,
  561. .minor_base = ADU_MINOR_BASE,
  562. };
  563. /**
  564. * adu_probe
  565. *
  566. * Called by the usb core when a new device is connected that it thinks
  567. * this driver might be interested in.
  568. */
  569. static int adu_probe(struct usb_interface *interface,
  570. const struct usb_device_id *id)
  571. {
  572. struct usb_device *udev = interface_to_usbdev(interface);
  573. struct adu_device *dev = NULL;
  574. struct usb_host_interface *iface_desc;
  575. struct usb_endpoint_descriptor *endpoint;
  576. int retval = -ENODEV;
  577. int in_end_size;
  578. int out_end_size;
  579. int i;
  580. if (udev == NULL) {
  581. dev_err(&interface->dev, "udev is NULL.\n");
  582. goto exit;
  583. }
  584. /* allocate memory for our device state and initialize it */
  585. dev = kzalloc(sizeof(struct adu_device), GFP_KERNEL);
  586. if (dev == NULL) {
  587. dev_err(&interface->dev, "Out of memory\n");
  588. retval = -ENOMEM;
  589. goto exit;
  590. }
  591. mutex_init(&dev->mtx);
  592. spin_lock_init(&dev->buflock);
  593. dev->udev = udev;
  594. init_waitqueue_head(&dev->read_wait);
  595. init_waitqueue_head(&dev->write_wait);
  596. iface_desc = &interface->altsetting[0];
  597. /* set up the endpoint information */
  598. for (i = 0; i < iface_desc->desc.bNumEndpoints; ++i) {
  599. endpoint = &iface_desc->endpoint[i].desc;
  600. if (usb_endpoint_is_int_in(endpoint))
  601. dev->interrupt_in_endpoint = endpoint;
  602. if (usb_endpoint_is_int_out(endpoint))
  603. dev->interrupt_out_endpoint = endpoint;
  604. }
  605. if (dev->interrupt_in_endpoint == NULL) {
  606. dev_err(&interface->dev, "interrupt in endpoint not found\n");
  607. goto error;
  608. }
  609. if (dev->interrupt_out_endpoint == NULL) {
  610. dev_err(&interface->dev, "interrupt out endpoint not found\n");
  611. goto error;
  612. }
  613. in_end_size = usb_endpoint_maxp(dev->interrupt_in_endpoint);
  614. out_end_size = usb_endpoint_maxp(dev->interrupt_out_endpoint);
  615. dev->read_buffer_primary = kmalloc((4 * in_end_size), GFP_KERNEL);
  616. if (!dev->read_buffer_primary) {
  617. dev_err(&interface->dev, "Couldn't allocate read_buffer_primary\n");
  618. retval = -ENOMEM;
  619. goto error;
  620. }
  621. /* debug code prime the buffer */
  622. memset(dev->read_buffer_primary, 'a', in_end_size);
  623. memset(dev->read_buffer_primary + in_end_size, 'b', in_end_size);
  624. memset(dev->read_buffer_primary + (2 * in_end_size), 'c', in_end_size);
  625. memset(dev->read_buffer_primary + (3 * in_end_size), 'd', in_end_size);
  626. dev->read_buffer_secondary = kmalloc((4 * in_end_size), GFP_KERNEL);
  627. if (!dev->read_buffer_secondary) {
  628. dev_err(&interface->dev, "Couldn't allocate read_buffer_secondary\n");
  629. retval = -ENOMEM;
  630. goto error;
  631. }
  632. /* debug code prime the buffer */
  633. memset(dev->read_buffer_secondary, 'e', in_end_size);
  634. memset(dev->read_buffer_secondary + in_end_size, 'f', in_end_size);
  635. memset(dev->read_buffer_secondary + (2 * in_end_size), 'g', in_end_size);
  636. memset(dev->read_buffer_secondary + (3 * in_end_size), 'h', in_end_size);
  637. dev->interrupt_in_buffer = kmalloc(in_end_size, GFP_KERNEL);
  638. if (!dev->interrupt_in_buffer) {
  639. dev_err(&interface->dev, "Couldn't allocate interrupt_in_buffer\n");
  640. goto error;
  641. }
  642. /* debug code prime the buffer */
  643. memset(dev->interrupt_in_buffer, 'i', in_end_size);
  644. dev->interrupt_in_urb = usb_alloc_urb(0, GFP_KERNEL);
  645. if (!dev->interrupt_in_urb) {
  646. dev_err(&interface->dev, "Couldn't allocate interrupt_in_urb\n");
  647. goto error;
  648. }
  649. dev->interrupt_out_buffer = kmalloc(out_end_size, GFP_KERNEL);
  650. if (!dev->interrupt_out_buffer) {
  651. dev_err(&interface->dev, "Couldn't allocate interrupt_out_buffer\n");
  652. goto error;
  653. }
  654. dev->interrupt_out_urb = usb_alloc_urb(0, GFP_KERNEL);
  655. if (!dev->interrupt_out_urb) {
  656. dev_err(&interface->dev, "Couldn't allocate interrupt_out_urb\n");
  657. goto error;
  658. }
  659. if (!usb_string(udev, udev->descriptor.iSerialNumber, dev->serial_number,
  660. sizeof(dev->serial_number))) {
  661. dev_err(&interface->dev, "Could not retrieve serial number\n");
  662. goto error;
  663. }
  664. dev_dbg(&interface->dev,"serial_number=%s", dev->serial_number);
  665. /* we can register the device now, as it is ready */
  666. usb_set_intfdata(interface, dev);
  667. retval = usb_register_dev(interface, &adu_class);
  668. if (retval) {
  669. /* something prevented us from registering this driver */
  670. dev_err(&interface->dev, "Not able to get a minor for this device.\n");
  671. usb_set_intfdata(interface, NULL);
  672. goto error;
  673. }
  674. dev->minor = interface->minor;
  675. /* let the user know what node this device is now attached to */
  676. dev_info(&interface->dev, "ADU%d %s now attached to /dev/usb/adutux%d\n",
  677. le16_to_cpu(udev->descriptor.idProduct), dev->serial_number,
  678. (dev->minor - ADU_MINOR_BASE));
  679. exit:
  680. return retval;
  681. error:
  682. adu_delete(dev);
  683. return retval;
  684. }
  685. /**
  686. * adu_disconnect
  687. *
  688. * Called by the usb core when the device is removed from the system.
  689. */
  690. static void adu_disconnect(struct usb_interface *interface)
  691. {
  692. struct adu_device *dev;
  693. int minor;
  694. dev = usb_get_intfdata(interface);
  695. mutex_lock(&dev->mtx); /* not interruptible */
  696. dev->udev = NULL; /* poison */
  697. minor = dev->minor;
  698. usb_deregister_dev(interface, &adu_class);
  699. mutex_unlock(&dev->mtx);
  700. mutex_lock(&adutux_mutex);
  701. usb_set_intfdata(interface, NULL);
  702. /* if the device is not opened, then we clean up right now */
  703. if (!dev->open_count)
  704. adu_delete(dev);
  705. mutex_unlock(&adutux_mutex);
  706. }
  707. /* usb specific object needed to register this driver with the usb subsystem */
  708. static struct usb_driver adu_driver = {
  709. .name = "adutux",
  710. .probe = adu_probe,
  711. .disconnect = adu_disconnect,
  712. .id_table = device_table,
  713. };
  714. module_usb_driver(adu_driver);
  715. MODULE_AUTHOR(DRIVER_AUTHOR);
  716. MODULE_DESCRIPTION(DRIVER_DESC);
  717. MODULE_LICENSE("GPL");