legousbtower.c 29 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087
  1. /*
  2. * LEGO USB Tower driver
  3. *
  4. * Copyright (C) 2003 David Glance <davidgsf@sourceforge.net>
  5. * 2001-2004 Juergen Stuber <starblue@users.sourceforge.net>
  6. *
  7. * This program is free software; you can redistribute it and/or
  8. * modify it under the terms of the GNU General Public License as
  9. * published by the Free Software Foundation; either version 2 of
  10. * the License, or (at your option) any later version.
  11. *
  12. * derived from USB Skeleton driver - 0.5
  13. * Copyright (C) 2001 Greg Kroah-Hartman (greg@kroah.com)
  14. *
  15. * History:
  16. *
  17. * 2001-10-13 - 0.1 js
  18. * - first version
  19. * 2001-11-03 - 0.2 js
  20. * - simplified buffering, one-shot URBs for writing
  21. * 2001-11-10 - 0.3 js
  22. * - removed IOCTL (setting power/mode is more complicated, postponed)
  23. * 2001-11-28 - 0.4 js
  24. * - added vendor commands for mode of operation and power level in open
  25. * 2001-12-04 - 0.5 js
  26. * - set IR mode by default (by oversight 0.4 set VLL mode)
  27. * 2002-01-11 - 0.5? pcchan
  28. * - make read buffer reusable and work around bytes_to_write issue between
  29. * uhci and legusbtower
  30. * 2002-09-23 - 0.52 david (david@csse.uwa.edu.au)
  31. * - imported into lejos project
  32. * - changed wake_up to wake_up_interruptible
  33. * - changed to use lego0 rather than tower0
  34. * - changed dbg() to use __func__ rather than deprecated __FUNCTION__
  35. * 2003-01-12 - 0.53 david (david@csse.uwa.edu.au)
  36. * - changed read and write to write everything or
  37. * timeout (from a patch by Chris Riesen and Brett Thaeler driver)
  38. * - added ioctl functionality to set timeouts
  39. * 2003-07-18 - 0.54 davidgsf (david@csse.uwa.edu.au)
  40. * - initial import into LegoUSB project
  41. * - merge of existing LegoUSB.c driver
  42. * 2003-07-18 - 0.56 davidgsf (david@csse.uwa.edu.au)
  43. * - port to 2.6 style driver
  44. * 2004-02-29 - 0.6 Juergen Stuber <starblue@users.sourceforge.net>
  45. * - fix locking
  46. * - unlink read URBs which are no longer needed
  47. * - allow increased buffer size, eliminates need for timeout on write
  48. * - have read URB running continuously
  49. * - added poll
  50. * - forbid seeking
  51. * - added nonblocking I/O
  52. * - changed back __func__ to __FUNCTION__
  53. * - read and log tower firmware version
  54. * - reset tower on probe, avoids failure of first write
  55. * 2004-03-09 - 0.7 Juergen Stuber <starblue@users.sourceforge.net>
  56. * - timeout read now only after inactivity, shorten default accordingly
  57. * 2004-03-11 - 0.8 Juergen Stuber <starblue@users.sourceforge.net>
  58. * - log major, minor instead of possibly confusing device filename
  59. * - whitespace cleanup
  60. * 2004-03-12 - 0.9 Juergen Stuber <starblue@users.sourceforge.net>
  61. * - normalize whitespace in debug messages
  62. * - take care about endianness in control message responses
  63. * 2004-03-13 - 0.91 Juergen Stuber <starblue@users.sourceforge.net>
  64. * - make default intervals longer to accommodate current EHCI driver
  65. * 2004-03-19 - 0.92 Juergen Stuber <starblue@users.sourceforge.net>
  66. * - replaced atomic_t by memory barriers
  67. * 2004-04-21 - 0.93 Juergen Stuber <starblue@users.sourceforge.net>
  68. * - wait for completion of write urb in release (needed for remotecontrol)
  69. * - corrected poll for write direction (missing negation)
  70. * 2004-04-22 - 0.94 Juergen Stuber <starblue@users.sourceforge.net>
  71. * - make device locking interruptible
  72. * 2004-04-30 - 0.95 Juergen Stuber <starblue@users.sourceforge.net>
  73. * - check for valid udev on resubmitting and unlinking urbs
  74. * 2004-08-03 - 0.96 Juergen Stuber <starblue@users.sourceforge.net>
  75. * - move reset into open to clean out spurious data
  76. */
  77. #include <linux/config.h>
  78. #include <linux/kernel.h>
  79. #include <linux/errno.h>
  80. #include <linux/init.h>
  81. #include <linux/slab.h>
  82. #include <linux/module.h>
  83. #include <linux/smp_lock.h>
  84. #include <linux/completion.h>
  85. #include <linux/mutex.h>
  86. #include <asm/uaccess.h>
  87. #include <linux/usb.h>
  88. #include <linux/poll.h>
  89. #ifdef CONFIG_USB_DEBUG
  90. static int debug = 4;
  91. #else
  92. static int debug = 0;
  93. #endif
  94. /* Use our own dbg macro */
  95. #undef dbg
  96. #define dbg(lvl, format, arg...) do { if (debug >= lvl) printk(KERN_DEBUG __FILE__ ": " format "\n", ## arg); } while (0)
  97. /* Version Information */
  98. #define DRIVER_VERSION "v0.96"
  99. #define DRIVER_AUTHOR "Juergen Stuber <starblue@sourceforge.net>"
  100. #define DRIVER_DESC "LEGO USB Tower Driver"
  101. /* Module parameters */
  102. module_param(debug, int, S_IRUGO | S_IWUSR);
  103. MODULE_PARM_DESC(debug, "Debug enabled or not");
  104. /* The defaults are chosen to work with the latest versions of leJOS and NQC.
  105. */
  106. /* Some legacy software likes to receive packets in one piece.
  107. * In this case read_buffer_size should exceed the maximal packet length
  108. * (417 for datalog uploads), and packet_timeout should be set.
  109. */
  110. static int read_buffer_size = 480;
  111. module_param(read_buffer_size, int, 0);
  112. MODULE_PARM_DESC(read_buffer_size, "Read buffer size");
  113. /* Some legacy software likes to send packets in one piece.
  114. * In this case write_buffer_size should exceed the maximal packet length
  115. * (417 for firmware and program downloads).
  116. * A problem with long writes is that the following read may time out
  117. * if the software is not prepared to wait long enough.
  118. */
  119. static int write_buffer_size = 480;
  120. module_param(write_buffer_size, int, 0);
  121. MODULE_PARM_DESC(write_buffer_size, "Write buffer size");
  122. /* Some legacy software expects reads to contain whole LASM packets.
  123. * To achieve this, characters which arrive before a packet timeout
  124. * occurs will be returned in a single read operation.
  125. * A problem with long reads is that the software may time out
  126. * if it is not prepared to wait long enough.
  127. * The packet timeout should be greater than the time between the
  128. * reception of subsequent characters, which should arrive about
  129. * every 5ms for the standard 2400 baud.
  130. * Set it to 0 to disable.
  131. */
  132. static int packet_timeout = 50;
  133. module_param(packet_timeout, int, 0);
  134. MODULE_PARM_DESC(packet_timeout, "Packet timeout in ms");
  135. /* Some legacy software expects blocking reads to time out.
  136. * Timeout occurs after the specified time of read and write inactivity.
  137. * Set it to 0 to disable.
  138. */
  139. static int read_timeout = 200;
  140. module_param(read_timeout, int, 0);
  141. MODULE_PARM_DESC(read_timeout, "Read timeout in ms");
  142. /* As of kernel version 2.6.4 ehci-hcd uses an
  143. * "only one interrupt transfer per frame" shortcut
  144. * to simplify the scheduling of periodic transfers.
  145. * This conflicts with our standard 1ms intervals for in and out URBs.
  146. * We use default intervals of 2ms for in and 8ms for out transfers,
  147. * which is fast enough for 2400 baud and allows a small additional load.
  148. * Increase the interval to allow more devices that do interrupt transfers,
  149. * or set to 0 to use the standard interval from the endpoint descriptors.
  150. */
  151. static int interrupt_in_interval = 2;
  152. module_param(interrupt_in_interval, int, 0);
  153. MODULE_PARM_DESC(interrupt_in_interval, "Interrupt in interval in ms");
  154. static int interrupt_out_interval = 8;
  155. module_param(interrupt_out_interval, int, 0);
  156. MODULE_PARM_DESC(interrupt_out_interval, "Interrupt out interval in ms");
  157. /* Define these values to match your device */
  158. #define LEGO_USB_TOWER_VENDOR_ID 0x0694
  159. #define LEGO_USB_TOWER_PRODUCT_ID 0x0001
  160. /* Vendor requests */
  161. #define LEGO_USB_TOWER_REQUEST_RESET 0x04
  162. #define LEGO_USB_TOWER_REQUEST_GET_VERSION 0xFD
  163. struct tower_reset_reply {
  164. __le16 size; /* little-endian */
  165. __u8 err_code;
  166. __u8 spare;
  167. } __attribute__ ((packed));
  168. struct tower_get_version_reply {
  169. __le16 size; /* little-endian */
  170. __u8 err_code;
  171. __u8 spare;
  172. __u8 major;
  173. __u8 minor;
  174. __le16 build_no; /* little-endian */
  175. } __attribute__ ((packed));
  176. /* table of devices that work with this driver */
  177. static struct usb_device_id tower_table [] = {
  178. { USB_DEVICE(LEGO_USB_TOWER_VENDOR_ID, LEGO_USB_TOWER_PRODUCT_ID) },
  179. { } /* Terminating entry */
  180. };
  181. MODULE_DEVICE_TABLE (usb, tower_table);
  182. #define LEGO_USB_TOWER_MINOR_BASE 160
  183. /* Structure to hold all of our device specific stuff */
  184. struct lego_usb_tower {
  185. struct semaphore sem; /* locks this structure */
  186. struct usb_device* udev; /* save off the usb device pointer */
  187. unsigned char minor; /* the starting minor number for this device */
  188. int open_count; /* number of times this port has been opened */
  189. char* read_buffer;
  190. size_t read_buffer_length; /* this much came in */
  191. size_t read_packet_length; /* this much will be returned on read */
  192. spinlock_t read_buffer_lock;
  193. int packet_timeout_jiffies;
  194. unsigned long read_last_arrival;
  195. wait_queue_head_t read_wait;
  196. wait_queue_head_t write_wait;
  197. char* interrupt_in_buffer;
  198. struct usb_endpoint_descriptor* interrupt_in_endpoint;
  199. struct urb* interrupt_in_urb;
  200. int interrupt_in_interval;
  201. int interrupt_in_running;
  202. int interrupt_in_done;
  203. char* interrupt_out_buffer;
  204. struct usb_endpoint_descriptor* interrupt_out_endpoint;
  205. struct urb* interrupt_out_urb;
  206. int interrupt_out_interval;
  207. int interrupt_out_busy;
  208. };
  209. /* local function prototypes */
  210. static ssize_t tower_read (struct file *file, char __user *buffer, size_t count, loff_t *ppos);
  211. static ssize_t tower_write (struct file *file, const char __user *buffer, size_t count, loff_t *ppos);
  212. static inline void tower_delete (struct lego_usb_tower *dev);
  213. static int tower_open (struct inode *inode, struct file *file);
  214. static int tower_release (struct inode *inode, struct file *file);
  215. static unsigned int tower_poll (struct file *file, poll_table *wait);
  216. static loff_t tower_llseek (struct file *file, loff_t off, int whence);
  217. static void tower_abort_transfers (struct lego_usb_tower *dev);
  218. static void tower_check_for_read_packet (struct lego_usb_tower *dev);
  219. static void tower_interrupt_in_callback (struct urb *urb, struct pt_regs *regs);
  220. static void tower_interrupt_out_callback (struct urb *urb, struct pt_regs *regs);
  221. static int tower_probe (struct usb_interface *interface, const struct usb_device_id *id);
  222. static void tower_disconnect (struct usb_interface *interface);
  223. /* prevent races between open() and disconnect */
  224. static DEFINE_MUTEX (disconnect_mutex);
  225. /* file operations needed when we register this driver */
  226. static struct file_operations tower_fops = {
  227. .owner = THIS_MODULE,
  228. .read = tower_read,
  229. .write = tower_write,
  230. .open = tower_open,
  231. .release = tower_release,
  232. .poll = tower_poll,
  233. .llseek = tower_llseek,
  234. };
  235. /*
  236. * usb class driver info in order to get a minor number from the usb core,
  237. * and to have the device registered with the driver core
  238. */
  239. static struct usb_class_driver tower_class = {
  240. .name = "legousbtower%d",
  241. .fops = &tower_fops,
  242. .minor_base = LEGO_USB_TOWER_MINOR_BASE,
  243. };
  244. /* usb specific object needed to register this driver with the usb subsystem */
  245. static struct usb_driver tower_driver = {
  246. .name = "legousbtower",
  247. .probe = tower_probe,
  248. .disconnect = tower_disconnect,
  249. .id_table = tower_table,
  250. };
  251. /**
  252. * lego_usb_tower_debug_data
  253. */
  254. static inline void lego_usb_tower_debug_data (int level, const char *function, int size, const unsigned char *data)
  255. {
  256. int i;
  257. if (debug < level)
  258. return;
  259. printk (KERN_DEBUG __FILE__": %s - length = %d, data = ", function, size);
  260. for (i = 0; i < size; ++i) {
  261. printk ("%.2x ", data[i]);
  262. }
  263. printk ("\n");
  264. }
  265. /**
  266. * tower_delete
  267. */
  268. static inline void tower_delete (struct lego_usb_tower *dev)
  269. {
  270. dbg(2, "%s: enter", __FUNCTION__);
  271. tower_abort_transfers (dev);
  272. /* free data structures */
  273. if (dev->interrupt_in_urb != NULL) {
  274. usb_free_urb (dev->interrupt_in_urb);
  275. }
  276. if (dev->interrupt_out_urb != NULL) {
  277. usb_free_urb (dev->interrupt_out_urb);
  278. }
  279. kfree (dev->read_buffer);
  280. kfree (dev->interrupt_in_buffer);
  281. kfree (dev->interrupt_out_buffer);
  282. kfree (dev);
  283. dbg(2, "%s: leave", __FUNCTION__);
  284. }
  285. /**
  286. * tower_open
  287. */
  288. static int tower_open (struct inode *inode, struct file *file)
  289. {
  290. struct lego_usb_tower *dev = NULL;
  291. int subminor;
  292. int retval = 0;
  293. struct usb_interface *interface;
  294. struct tower_reset_reply reset_reply;
  295. int result;
  296. dbg(2, "%s: enter", __FUNCTION__);
  297. nonseekable_open(inode, file);
  298. subminor = iminor(inode);
  299. mutex_lock (&disconnect_mutex);
  300. interface = usb_find_interface (&tower_driver, subminor);
  301. if (!interface) {
  302. err ("%s - error, can't find device for minor %d",
  303. __FUNCTION__, subminor);
  304. retval = -ENODEV;
  305. goto unlock_disconnect_exit;
  306. }
  307. dev = usb_get_intfdata(interface);
  308. if (!dev) {
  309. retval = -ENODEV;
  310. goto unlock_disconnect_exit;
  311. }
  312. /* lock this device */
  313. if (down_interruptible (&dev->sem)) {
  314. retval = -ERESTARTSYS;
  315. goto unlock_disconnect_exit;
  316. }
  317. /* allow opening only once */
  318. if (dev->open_count) {
  319. retval = -EBUSY;
  320. goto unlock_exit;
  321. }
  322. dev->open_count = 1;
  323. /* reset the tower */
  324. result = usb_control_msg (dev->udev,
  325. usb_rcvctrlpipe(dev->udev, 0),
  326. LEGO_USB_TOWER_REQUEST_RESET,
  327. USB_TYPE_VENDOR | USB_DIR_IN | USB_RECIP_DEVICE,
  328. 0,
  329. 0,
  330. &reset_reply,
  331. sizeof(reset_reply),
  332. 1000);
  333. if (result < 0) {
  334. err("LEGO USB Tower reset control request failed");
  335. retval = result;
  336. goto unlock_exit;
  337. }
  338. /* initialize in direction */
  339. dev->read_buffer_length = 0;
  340. dev->read_packet_length = 0;
  341. usb_fill_int_urb (dev->interrupt_in_urb,
  342. dev->udev,
  343. usb_rcvintpipe(dev->udev, dev->interrupt_in_endpoint->bEndpointAddress),
  344. dev->interrupt_in_buffer,
  345. le16_to_cpu(dev->interrupt_in_endpoint->wMaxPacketSize),
  346. tower_interrupt_in_callback,
  347. dev,
  348. dev->interrupt_in_interval);
  349. dev->interrupt_in_running = 1;
  350. dev->interrupt_in_done = 0;
  351. mb();
  352. retval = usb_submit_urb (dev->interrupt_in_urb, GFP_KERNEL);
  353. if (retval) {
  354. err("Couldn't submit interrupt_in_urb %d", retval);
  355. dev->interrupt_in_running = 0;
  356. dev->open_count = 0;
  357. goto unlock_exit;
  358. }
  359. /* save device in the file's private structure */
  360. file->private_data = dev;
  361. unlock_exit:
  362. up (&dev->sem);
  363. unlock_disconnect_exit:
  364. mutex_unlock (&disconnect_mutex);
  365. dbg(2, "%s: leave, return value %d ", __FUNCTION__, retval);
  366. return retval;
  367. }
  368. /**
  369. * tower_release
  370. */
  371. static int tower_release (struct inode *inode, struct file *file)
  372. {
  373. struct lego_usb_tower *dev;
  374. int retval = 0;
  375. dbg(2, "%s: enter", __FUNCTION__);
  376. dev = (struct lego_usb_tower *)file->private_data;
  377. if (dev == NULL) {
  378. dbg(1, "%s: object is NULL", __FUNCTION__);
  379. retval = -ENODEV;
  380. goto exit;
  381. }
  382. if (down_interruptible (&dev->sem)) {
  383. retval = -ERESTARTSYS;
  384. goto exit;
  385. }
  386. if (dev->open_count != 1) {
  387. dbg(1, "%s: device not opened exactly once", __FUNCTION__);
  388. retval = -ENODEV;
  389. goto unlock_exit;
  390. }
  391. if (dev->udev == NULL) {
  392. /* the device was unplugged before the file was released */
  393. up (&dev->sem); /* unlock here as tower_delete frees dev */
  394. tower_delete (dev);
  395. goto exit;
  396. }
  397. /* wait until write transfer is finished */
  398. if (dev->interrupt_out_busy) {
  399. wait_event_interruptible_timeout (dev->write_wait, !dev->interrupt_out_busy, 2 * HZ);
  400. }
  401. tower_abort_transfers (dev);
  402. dev->open_count = 0;
  403. unlock_exit:
  404. up (&dev->sem);
  405. exit:
  406. dbg(2, "%s: leave, return value %d", __FUNCTION__, retval);
  407. return retval;
  408. }
  409. /**
  410. * tower_abort_transfers
  411. * aborts transfers and frees associated data structures
  412. */
  413. static void tower_abort_transfers (struct lego_usb_tower *dev)
  414. {
  415. dbg(2, "%s: enter", __FUNCTION__);
  416. if (dev == NULL) {
  417. dbg(1, "%s: dev is null", __FUNCTION__);
  418. goto exit;
  419. }
  420. /* shutdown transfer */
  421. if (dev->interrupt_in_running) {
  422. dev->interrupt_in_running = 0;
  423. mb();
  424. if (dev->interrupt_in_urb != NULL && dev->udev) {
  425. usb_kill_urb (dev->interrupt_in_urb);
  426. }
  427. }
  428. if (dev->interrupt_out_busy) {
  429. if (dev->interrupt_out_urb != NULL && dev->udev) {
  430. usb_kill_urb (dev->interrupt_out_urb);
  431. }
  432. }
  433. exit:
  434. dbg(2, "%s: leave", __FUNCTION__);
  435. }
  436. /**
  437. * tower_check_for_read_packet
  438. *
  439. * To get correct semantics for signals and non-blocking I/O
  440. * with packetizing we pretend not to see any data in the read buffer
  441. * until it has been there unchanged for at least
  442. * dev->packet_timeout_jiffies, or until the buffer is full.
  443. */
  444. static void tower_check_for_read_packet (struct lego_usb_tower *dev)
  445. {
  446. spin_lock_irq (&dev->read_buffer_lock);
  447. if (!packet_timeout
  448. || time_after(jiffies, dev->read_last_arrival + dev->packet_timeout_jiffies)
  449. || dev->read_buffer_length == read_buffer_size) {
  450. dev->read_packet_length = dev->read_buffer_length;
  451. }
  452. dev->interrupt_in_done = 0;
  453. spin_unlock_irq (&dev->read_buffer_lock);
  454. }
  455. /**
  456. * tower_poll
  457. */
  458. static unsigned int tower_poll (struct file *file, poll_table *wait)
  459. {
  460. struct lego_usb_tower *dev;
  461. unsigned int mask = 0;
  462. dbg(2, "%s: enter", __FUNCTION__);
  463. dev = file->private_data;
  464. poll_wait(file, &dev->read_wait, wait);
  465. poll_wait(file, &dev->write_wait, wait);
  466. tower_check_for_read_packet(dev);
  467. if (dev->read_packet_length > 0) {
  468. mask |= POLLIN | POLLRDNORM;
  469. }
  470. if (!dev->interrupt_out_busy) {
  471. mask |= POLLOUT | POLLWRNORM;
  472. }
  473. dbg(2, "%s: leave, mask = %d", __FUNCTION__, mask);
  474. return mask;
  475. }
  476. /**
  477. * tower_llseek
  478. */
  479. static loff_t tower_llseek (struct file *file, loff_t off, int whence)
  480. {
  481. return -ESPIPE; /* unseekable */
  482. }
  483. /**
  484. * tower_read
  485. */
  486. static ssize_t tower_read (struct file *file, char __user *buffer, size_t count, loff_t *ppos)
  487. {
  488. struct lego_usb_tower *dev;
  489. size_t bytes_to_read;
  490. int i;
  491. int retval = 0;
  492. unsigned long timeout = 0;
  493. dbg(2, "%s: enter, count = %Zd", __FUNCTION__, count);
  494. dev = (struct lego_usb_tower *)file->private_data;
  495. /* lock this object */
  496. if (down_interruptible (&dev->sem)) {
  497. retval = -ERESTARTSYS;
  498. goto exit;
  499. }
  500. /* verify that the device wasn't unplugged */
  501. if (dev->udev == NULL) {
  502. retval = -ENODEV;
  503. err("No device or device unplugged %d", retval);
  504. goto unlock_exit;
  505. }
  506. /* verify that we actually have some data to read */
  507. if (count == 0) {
  508. dbg(1, "%s: read request of 0 bytes", __FUNCTION__);
  509. goto unlock_exit;
  510. }
  511. if (read_timeout) {
  512. timeout = jiffies + read_timeout * HZ / 1000;
  513. }
  514. /* wait for data */
  515. tower_check_for_read_packet (dev);
  516. while (dev->read_packet_length == 0) {
  517. if (file->f_flags & O_NONBLOCK) {
  518. retval = -EAGAIN;
  519. goto unlock_exit;
  520. }
  521. retval = wait_event_interruptible_timeout(dev->read_wait, dev->interrupt_in_done, dev->packet_timeout_jiffies);
  522. if (retval < 0) {
  523. goto unlock_exit;
  524. }
  525. /* reset read timeout during read or write activity */
  526. if (read_timeout
  527. && (dev->read_buffer_length || dev->interrupt_out_busy)) {
  528. timeout = jiffies + read_timeout * HZ / 1000;
  529. }
  530. /* check for read timeout */
  531. if (read_timeout && time_after (jiffies, timeout)) {
  532. retval = -ETIMEDOUT;
  533. goto unlock_exit;
  534. }
  535. tower_check_for_read_packet (dev);
  536. }
  537. /* copy the data from read_buffer into userspace */
  538. bytes_to_read = min(count, dev->read_packet_length);
  539. if (copy_to_user (buffer, dev->read_buffer, bytes_to_read)) {
  540. retval = -EFAULT;
  541. goto unlock_exit;
  542. }
  543. spin_lock_irq (&dev->read_buffer_lock);
  544. dev->read_buffer_length -= bytes_to_read;
  545. dev->read_packet_length -= bytes_to_read;
  546. for (i=0; i<dev->read_buffer_length; i++) {
  547. dev->read_buffer[i] = dev->read_buffer[i+bytes_to_read];
  548. }
  549. spin_unlock_irq (&dev->read_buffer_lock);
  550. retval = bytes_to_read;
  551. unlock_exit:
  552. /* unlock the device */
  553. up (&dev->sem);
  554. exit:
  555. dbg(2, "%s: leave, return value %d", __FUNCTION__, retval);
  556. return retval;
  557. }
  558. /**
  559. * tower_write
  560. */
  561. static ssize_t tower_write (struct file *file, const char __user *buffer, size_t count, loff_t *ppos)
  562. {
  563. struct lego_usb_tower *dev;
  564. size_t bytes_to_write;
  565. int retval = 0;
  566. dbg(2, "%s: enter, count = %Zd", __FUNCTION__, count);
  567. dev = (struct lego_usb_tower *)file->private_data;
  568. /* lock this object */
  569. if (down_interruptible (&dev->sem)) {
  570. retval = -ERESTARTSYS;
  571. goto exit;
  572. }
  573. /* verify that the device wasn't unplugged */
  574. if (dev->udev == NULL) {
  575. retval = -ENODEV;
  576. err("No device or device unplugged %d", retval);
  577. goto unlock_exit;
  578. }
  579. /* verify that we actually have some data to write */
  580. if (count == 0) {
  581. dbg(1, "%s: write request of 0 bytes", __FUNCTION__);
  582. goto unlock_exit;
  583. }
  584. /* wait until previous transfer is finished */
  585. while (dev->interrupt_out_busy) {
  586. if (file->f_flags & O_NONBLOCK) {
  587. retval = -EAGAIN;
  588. goto unlock_exit;
  589. }
  590. retval = wait_event_interruptible (dev->write_wait, !dev->interrupt_out_busy);
  591. if (retval) {
  592. goto unlock_exit;
  593. }
  594. }
  595. /* write the data into interrupt_out_buffer from userspace */
  596. bytes_to_write = min_t(int, count, write_buffer_size);
  597. dbg(4, "%s: count = %Zd, bytes_to_write = %Zd", __FUNCTION__, count, bytes_to_write);
  598. if (copy_from_user (dev->interrupt_out_buffer, buffer, bytes_to_write)) {
  599. retval = -EFAULT;
  600. goto unlock_exit;
  601. }
  602. /* send off the urb */
  603. usb_fill_int_urb(dev->interrupt_out_urb,
  604. dev->udev,
  605. usb_sndintpipe(dev->udev, dev->interrupt_out_endpoint->bEndpointAddress),
  606. dev->interrupt_out_buffer,
  607. bytes_to_write,
  608. tower_interrupt_out_callback,
  609. dev,
  610. dev->interrupt_out_interval);
  611. dev->interrupt_out_busy = 1;
  612. wmb();
  613. retval = usb_submit_urb (dev->interrupt_out_urb, GFP_KERNEL);
  614. if (retval) {
  615. dev->interrupt_out_busy = 0;
  616. err("Couldn't submit interrupt_out_urb %d", retval);
  617. goto unlock_exit;
  618. }
  619. retval = bytes_to_write;
  620. unlock_exit:
  621. /* unlock the device */
  622. up (&dev->sem);
  623. exit:
  624. dbg(2, "%s: leave, return value %d", __FUNCTION__, retval);
  625. return retval;
  626. }
  627. /**
  628. * tower_interrupt_in_callback
  629. */
  630. static void tower_interrupt_in_callback (struct urb *urb, struct pt_regs *regs)
  631. {
  632. struct lego_usb_tower *dev = (struct lego_usb_tower *)urb->context;
  633. int retval;
  634. dbg(4, "%s: enter, status %d", __FUNCTION__, urb->status);
  635. lego_usb_tower_debug_data(5, __FUNCTION__, urb->actual_length, urb->transfer_buffer);
  636. if (urb->status) {
  637. if (urb->status == -ENOENT ||
  638. urb->status == -ECONNRESET ||
  639. urb->status == -ESHUTDOWN) {
  640. goto exit;
  641. } else {
  642. dbg(1, "%s: nonzero status received: %d", __FUNCTION__, urb->status);
  643. goto resubmit; /* maybe we can recover */
  644. }
  645. }
  646. if (urb->actual_length > 0) {
  647. spin_lock (&dev->read_buffer_lock);
  648. if (dev->read_buffer_length + urb->actual_length < read_buffer_size) {
  649. memcpy (dev->read_buffer + dev->read_buffer_length,
  650. dev->interrupt_in_buffer,
  651. urb->actual_length);
  652. dev->read_buffer_length += urb->actual_length;
  653. dev->read_last_arrival = jiffies;
  654. dbg(3, "%s: received %d bytes", __FUNCTION__, urb->actual_length);
  655. } else {
  656. printk(KERN_WARNING "%s: read_buffer overflow, %d bytes dropped", __FUNCTION__, urb->actual_length);
  657. }
  658. spin_unlock (&dev->read_buffer_lock);
  659. }
  660. resubmit:
  661. /* resubmit if we're still running */
  662. if (dev->interrupt_in_running && dev->udev) {
  663. retval = usb_submit_urb (dev->interrupt_in_urb, GFP_ATOMIC);
  664. if (retval) {
  665. err("%s: usb_submit_urb failed (%d)", __FUNCTION__, retval);
  666. }
  667. }
  668. exit:
  669. dev->interrupt_in_done = 1;
  670. wake_up_interruptible (&dev->read_wait);
  671. lego_usb_tower_debug_data(5, __FUNCTION__, urb->actual_length, urb->transfer_buffer);
  672. dbg(4, "%s: leave, status %d", __FUNCTION__, urb->status);
  673. }
  674. /**
  675. * tower_interrupt_out_callback
  676. */
  677. static void tower_interrupt_out_callback (struct urb *urb, struct pt_regs *regs)
  678. {
  679. struct lego_usb_tower *dev = (struct lego_usb_tower *)urb->context;
  680. dbg(4, "%s: enter, status %d", __FUNCTION__, urb->status);
  681. lego_usb_tower_debug_data(5, __FUNCTION__, urb->actual_length, urb->transfer_buffer);
  682. /* sync/async unlink faults aren't errors */
  683. if (urb->status && !(urb->status == -ENOENT ||
  684. urb->status == -ECONNRESET ||
  685. urb->status == -ESHUTDOWN)) {
  686. dbg(1, "%s - nonzero write bulk status received: %d",
  687. __FUNCTION__, urb->status);
  688. }
  689. dev->interrupt_out_busy = 0;
  690. wake_up_interruptible(&dev->write_wait);
  691. lego_usb_tower_debug_data(5, __FUNCTION__, urb->actual_length, urb->transfer_buffer);
  692. dbg(4, "%s: leave, status %d", __FUNCTION__, urb->status);
  693. }
  694. /**
  695. * tower_probe
  696. *
  697. * Called by the usb core when a new device is connected that it thinks
  698. * this driver might be interested in.
  699. */
  700. static int tower_probe (struct usb_interface *interface, const struct usb_device_id *id)
  701. {
  702. struct usb_device *udev = interface_to_usbdev(interface);
  703. struct lego_usb_tower *dev = NULL;
  704. struct usb_host_interface *iface_desc;
  705. struct usb_endpoint_descriptor* endpoint;
  706. struct tower_get_version_reply get_version_reply;
  707. int i;
  708. int retval = -ENOMEM;
  709. int result;
  710. dbg(2, "%s: enter", __FUNCTION__);
  711. if (udev == NULL) {
  712. info ("udev is NULL.");
  713. }
  714. /* allocate memory for our device state and initialize it */
  715. dev = kmalloc (sizeof(struct lego_usb_tower), GFP_KERNEL);
  716. if (dev == NULL) {
  717. err ("Out of memory");
  718. goto exit;
  719. }
  720. init_MUTEX (&dev->sem);
  721. dev->udev = udev;
  722. dev->open_count = 0;
  723. dev->read_buffer = NULL;
  724. dev->read_buffer_length = 0;
  725. dev->read_packet_length = 0;
  726. spin_lock_init (&dev->read_buffer_lock);
  727. dev->packet_timeout_jiffies = packet_timeout * HZ / 1000;
  728. dev->read_last_arrival = jiffies;
  729. init_waitqueue_head (&dev->read_wait);
  730. init_waitqueue_head (&dev->write_wait);
  731. dev->interrupt_in_buffer = NULL;
  732. dev->interrupt_in_endpoint = NULL;
  733. dev->interrupt_in_urb = NULL;
  734. dev->interrupt_in_running = 0;
  735. dev->interrupt_in_done = 0;
  736. dev->interrupt_out_buffer = NULL;
  737. dev->interrupt_out_endpoint = NULL;
  738. dev->interrupt_out_urb = NULL;
  739. dev->interrupt_out_busy = 0;
  740. iface_desc = interface->cur_altsetting;
  741. /* set up the endpoint information */
  742. for (i = 0; i < iface_desc->desc.bNumEndpoints; ++i) {
  743. endpoint = &iface_desc->endpoint[i].desc;
  744. if (((endpoint->bEndpointAddress & USB_ENDPOINT_DIR_MASK) == USB_DIR_IN) &&
  745. ((endpoint->bmAttributes & USB_ENDPOINT_XFERTYPE_MASK) == USB_ENDPOINT_XFER_INT)) {
  746. dev->interrupt_in_endpoint = endpoint;
  747. }
  748. if (((endpoint->bEndpointAddress & USB_ENDPOINT_DIR_MASK) == USB_DIR_OUT) &&
  749. ((endpoint->bmAttributes & USB_ENDPOINT_XFERTYPE_MASK) == USB_ENDPOINT_XFER_INT)) {
  750. dev->interrupt_out_endpoint = endpoint;
  751. }
  752. }
  753. if(dev->interrupt_in_endpoint == NULL) {
  754. err("interrupt in endpoint not found");
  755. goto error;
  756. }
  757. if (dev->interrupt_out_endpoint == NULL) {
  758. err("interrupt out endpoint not found");
  759. goto error;
  760. }
  761. dev->read_buffer = kmalloc (read_buffer_size, GFP_KERNEL);
  762. if (!dev->read_buffer) {
  763. err("Couldn't allocate read_buffer");
  764. goto error;
  765. }
  766. dev->interrupt_in_buffer = kmalloc (le16_to_cpu(dev->interrupt_in_endpoint->wMaxPacketSize), GFP_KERNEL);
  767. if (!dev->interrupt_in_buffer) {
  768. err("Couldn't allocate interrupt_in_buffer");
  769. goto error;
  770. }
  771. dev->interrupt_in_urb = usb_alloc_urb(0, GFP_KERNEL);
  772. if (!dev->interrupt_in_urb) {
  773. err("Couldn't allocate interrupt_in_urb");
  774. goto error;
  775. }
  776. dev->interrupt_out_buffer = kmalloc (write_buffer_size, GFP_KERNEL);
  777. if (!dev->interrupt_out_buffer) {
  778. err("Couldn't allocate interrupt_out_buffer");
  779. goto error;
  780. }
  781. dev->interrupt_out_urb = usb_alloc_urb(0, GFP_KERNEL);
  782. if (!dev->interrupt_out_urb) {
  783. err("Couldn't allocate interrupt_out_urb");
  784. goto error;
  785. }
  786. dev->interrupt_in_interval = interrupt_in_interval ? interrupt_in_interval : dev->interrupt_in_endpoint->bInterval;
  787. dev->interrupt_out_interval = interrupt_out_interval ? interrupt_out_interval : dev->interrupt_out_endpoint->bInterval;
  788. /* we can register the device now, as it is ready */
  789. usb_set_intfdata (interface, dev);
  790. retval = usb_register_dev (interface, &tower_class);
  791. if (retval) {
  792. /* something prevented us from registering this driver */
  793. err ("Not able to get a minor for this device.");
  794. usb_set_intfdata (interface, NULL);
  795. goto error;
  796. }
  797. dev->minor = interface->minor;
  798. /* let the user know what node this device is now attached to */
  799. info ("LEGO USB Tower #%d now attached to major %d minor %d", (dev->minor - LEGO_USB_TOWER_MINOR_BASE), USB_MAJOR, dev->minor);
  800. /* get the firmware version and log it */
  801. result = usb_control_msg (udev,
  802. usb_rcvctrlpipe(udev, 0),
  803. LEGO_USB_TOWER_REQUEST_GET_VERSION,
  804. USB_TYPE_VENDOR | USB_DIR_IN | USB_RECIP_DEVICE,
  805. 0,
  806. 0,
  807. &get_version_reply,
  808. sizeof(get_version_reply),
  809. 1000);
  810. if (result < 0) {
  811. err("LEGO USB Tower get version control request failed");
  812. retval = result;
  813. goto error;
  814. }
  815. info("LEGO USB Tower firmware version is %d.%d build %d",
  816. get_version_reply.major,
  817. get_version_reply.minor,
  818. le16_to_cpu(get_version_reply.build_no));
  819. exit:
  820. dbg(2, "%s: leave, return value 0x%.8lx (dev)", __FUNCTION__, (long) dev);
  821. return retval;
  822. error:
  823. tower_delete(dev);
  824. return retval;
  825. }
  826. /**
  827. * tower_disconnect
  828. *
  829. * Called by the usb core when the device is removed from the system.
  830. */
  831. static void tower_disconnect (struct usb_interface *interface)
  832. {
  833. struct lego_usb_tower *dev;
  834. int minor;
  835. dbg(2, "%s: enter", __FUNCTION__);
  836. mutex_lock (&disconnect_mutex);
  837. dev = usb_get_intfdata (interface);
  838. usb_set_intfdata (interface, NULL);
  839. down (&dev->sem);
  840. minor = dev->minor;
  841. /* give back our minor */
  842. usb_deregister_dev (interface, &tower_class);
  843. /* if the device is not opened, then we clean up right now */
  844. if (!dev->open_count) {
  845. up (&dev->sem);
  846. tower_delete (dev);
  847. } else {
  848. dev->udev = NULL;
  849. up (&dev->sem);
  850. }
  851. mutex_unlock (&disconnect_mutex);
  852. info("LEGO USB Tower #%d now disconnected", (minor - LEGO_USB_TOWER_MINOR_BASE));
  853. dbg(2, "%s: leave", __FUNCTION__);
  854. }
  855. /**
  856. * lego_usb_tower_init
  857. */
  858. static int __init lego_usb_tower_init(void)
  859. {
  860. int result;
  861. int retval = 0;
  862. dbg(2, "%s: enter", __FUNCTION__);
  863. /* register this driver with the USB subsystem */
  864. result = usb_register(&tower_driver);
  865. if (result < 0) {
  866. err("usb_register failed for the "__FILE__" driver. Error number %d", result);
  867. retval = -1;
  868. goto exit;
  869. }
  870. info(DRIVER_DESC " " DRIVER_VERSION);
  871. exit:
  872. dbg(2, "%s: leave, return value %d", __FUNCTION__, retval);
  873. return retval;
  874. }
  875. /**
  876. * lego_usb_tower_exit
  877. */
  878. static void __exit lego_usb_tower_exit(void)
  879. {
  880. dbg(2, "%s: enter", __FUNCTION__);
  881. /* deregister this driver with the USB subsystem */
  882. usb_deregister (&tower_driver);
  883. dbg(2, "%s: leave", __FUNCTION__);
  884. }
  885. module_init (lego_usb_tower_init);
  886. module_exit (lego_usb_tower_exit);
  887. MODULE_AUTHOR(DRIVER_AUTHOR);
  888. MODULE_DESCRIPTION(DRIVER_DESC);
  889. #ifdef MODULE_LICENSE
  890. MODULE_LICENSE("GPL");
  891. #endif