drivers/usb/misc/yurex.c

   1 /*
   2  * Driver for Meywa-Denki & KAYAC YUREX
   3  *
   4  * Copyright (C) 2010 Tomoki Sekiyama (tomoki.sekiyama@gmail.com)
   5  *
   6  *      This program is free software; you can redistribute it and/or
   7  *      modify it under the terms of the GNU General Public License as
   8  *      published by the Free Software Foundation, version 2.
   9  *
  10  */
  11
  12 #include <linux/kernel.h>
  13 #include <linux/errno.h>
  14 #include <linux/slab.h>
  15 #include <linux/module.h>
  16 #include <linux/kref.h>
  17 #include <linux/mutex.h>
  18 #include <linux/uaccess.h>
  19 #include <linux/usb.h>
  20 #include <linux/hid.h>
  21
  22 #define DRIVER_AUTHOR "Tomoki Sekiyama"
  23 #define DRIVER_DESC "Driver for Meywa-Denki & KAYAC YUREX"
  24
  25 #define YUREX_VENDOR_ID         0x0c45
  26 #define YUREX_PRODUCT_ID        0x1010
  27
  28 #define CMD_ACK         '!'
  29 #define CMD_ANIMATE     'A'
  30 #define CMD_COUNT       'C'
  31 #define CMD_LED         'L'
  32 #define CMD_READ        'R'
  33 #define CMD_SET         'S'
  34 #define CMD_VERSION     'V'
  35 #define CMD_EOF         0x0d
  36 #define CMD_PADDING     0xff
  37
  38 #define YUREX_BUF_SIZE          8
  39 #define YUREX_WRITE_TIMEOUT     (HZ*2)
  40
  41 /* table of devices that work with this driver */
  42 static struct usb_device_id yurex_table[] = {
  43         { USB_DEVICE(YUREX_VENDOR_ID, YUREX_PRODUCT_ID) },
  44         { }                                     /* Terminating entry */
  45 };
  46 MODULE_DEVICE_TABLE(usb, yurex_table);
  47
  48 #ifdef CONFIG_USB_DYNAMIC_MINORS
  49 #define YUREX_MINOR_BASE        0
  50 #else
  51 #define YUREX_MINOR_BASE        192
  52 #endif
  53
  54 /* Structure to hold all of our device specific stuff */
  55 struct usb_yurex {
  56         struct usb_device       *udev;
  57         struct usb_interface    *interface;
  58         __u8                    int_in_endpointAddr;
  59         struct urb              *urb;           /* URB for interrupt in */
  60         unsigned char           *int_buffer;    /* buffer for intterupt in */
  61         struct urb              *cntl_urb;      /* URB for control msg */
  62         struct usb_ctrlrequest  *cntl_req;      /* req for control msg */
  63         unsigned char           *cntl_buffer;   /* buffer for control msg */
  64
  65         struct kref             kref;
  66         struct mutex            io_mutex;
  67         unsigned long           disconnected:1;
  68         struct fasync_struct    *async_queue;
  69         wait_queue_head_t       waitq;
  70
  71         spinlock_t              lock;
  72         __s64                   bbu;            /* BBU from device */
  73 };
  74 #define to_yurex_dev(d) container_of(d, struct usb_yurex, kref)
  75
  76 static struct usb_driver yurex_driver;
  77 static const struct file_operations yurex_fops;
  78
  79
  80 static void yurex_control_callback(struct urb *urb)
  81 {
  82         struct usb_yurex *dev = urb->context;
  83         int status = urb->status;
  84
  85         if (status) {
  86                 dev_err(&urb->dev->dev, "%s - control failed: %d\n",
  87                         __func__, status);
  88                 wake_up_interruptible(&dev->waitq);
  89                 return;
  90         }
  91         /* on success, sender woken up by CMD_ACK int in, or timeout */
  92 }
  93
  94 static void yurex_delete(struct kref *kref)
  95 {
  96         struct usb_yurex *dev = to_yurex_dev(kref);
  97
  98         dev_dbg(&dev->interface->dev, "%s\n", __func__);
  99
 100         if (dev->cntl_urb) {
 101                 usb_kill_urb(dev->cntl_urb);
 102                 kfree(dev->cntl_req);
 103                 if (dev->cntl_buffer)
 104                         usb_free_coherent(dev->udev, YUREX_BUF_SIZE,
 105                                 dev->cntl_buffer, dev->cntl_urb->transfer_dma);
 106                 usb_free_urb(dev->cntl_urb);
 107         }
 108         if (dev->urb) {
 109                 usb_kill_urb(dev->urb);
 110                 if (dev->int_buffer)
 111                         usb_free_coherent(dev->udev, YUREX_BUF_SIZE,
 112                                 dev->int_buffer, dev->urb->transfer_dma);
 113                 usb_free_urb(dev->urb);
 114         }
 115         usb_put_intf(dev->interface);
 116         usb_put_dev(dev->udev);
 117         kfree(dev);
 118 }
 119
 120 /*
 121  * usb class driver info in order to get a minor number from the usb core,
 122  * and to have the device registered with the driver core
 123  */
 124 static struct usb_class_driver yurex_class = {
 125         .name =         "yurex%d",
 126         .fops =         &yurex_fops,
 127         .minor_base =   YUREX_MINOR_BASE,
 128 };
 129
 130 static void yurex_interrupt(struct urb *urb)
 131 {
 132         struct usb_yurex *dev = urb->context;
 133         unsigned char *buf = dev->int_buffer;
 134         int status = urb->status;
 135         unsigned long flags;
 136         int retval, i;
 137
 138         switch (status) {
 139         case 0: /*success*/
 140                 break;
 141         /* The device is terminated or messed up, give up */
 142         case -EOVERFLOW:
 143                 dev_err(&dev->interface->dev,
 144                         "%s - overflow with length %d, actual length is %d\n",
 145                         __func__, YUREX_BUF_SIZE, dev->urb->actual_length);
 146         case -ECONNRESET:
 147         case -ENOENT:
 148         case -ESHUTDOWN:
 149         case -EILSEQ:
 150         case -EPROTO:
 151         case -ETIME:
 152                 return;
 153         default:
 154                 dev_err(&dev->interface->dev,
 155                         "%s - unknown status received: %d\n", __func__, status);
 156                 return;
 157         }
 158
 159         /* handle received message */
 160         switch (buf[0]) {
 161         case CMD_COUNT:
 162         case CMD_READ:
 163                 if (buf[6] == CMD_EOF) {
 164                         spin_lock_irqsave(&dev->lock, flags);
 165                         dev->bbu = 0;
 166                         for (i = 1; i < 6; i++) {
 167                                 dev->bbu += buf[i];
 168                                 if (i != 5)
 169                                         dev->bbu <<= 8;
 170                         }
 171                         dev_dbg(&dev->interface->dev, "%s count: %lld\n",
 172                                 __func__, dev->bbu);
 173                         spin_unlock_irqrestore(&dev->lock, flags);
 174
 175                         kill_fasync(&dev->async_queue, SIGIO, POLL_IN);
 176                 }
 177                 else
 178                         dev_dbg(&dev->interface->dev,
 179                                 "data format error - no EOF\n");
 180                 break;
 181         case CMD_ACK:
 182                 dev_dbg(&dev->interface->dev, "%s ack: %c\n",
 183                         __func__, buf[1]);
 184                 wake_up_interruptible(&dev->waitq);
 185                 break;
 186         }
 187
 188         retval = usb_submit_urb(dev->urb, GFP_ATOMIC);
 189         if (retval) {
 190                 dev_err(&dev->interface->dev, "%s - usb_submit_urb failed: %d\n",
 191                         __func__, retval);
 192         }
 193 }
 194
 195 static int yurex_probe(struct usb_interface *interface, const struct usb_device_id *id)
 196 {
 197         struct usb_yurex *dev;
 198         struct usb_host_interface *iface_desc;
 199         struct usb_endpoint_descriptor *endpoint;
 200         int retval = -ENOMEM;
 201         int i;
 202         DEFINE_WAIT(wait);
 203
 204         /* allocate memory for our device state and initialize it */
 205         dev = kzalloc(sizeof(*dev), GFP_KERNEL);
 206         if (!dev) {
 207                 dev_err(&interface->dev, "Out of memory\n");
 208                 goto error;
 209         }
 210         kref_init(&dev->kref);
 211         mutex_init(&dev->io_mutex);
 212         spin_lock_init(&dev->lock);
 213         init_waitqueue_head(&dev->waitq);
 214
 215         dev->udev = usb_get_dev(interface_to_usbdev(interface));
 216         dev->interface = usb_get_intf(interface);
 217
 218         /* set up the endpoint information */
 219         iface_desc = interface->cur_altsetting;
 220         for (i = 0; i < iface_desc->desc.bNumEndpoints; i++) {
 221                 endpoint = &iface_desc->endpoint[i].desc;
 222
 223                 if (usb_endpoint_is_int_in(endpoint)) {
 224                         dev->int_in_endpointAddr = endpoint->bEndpointAddress;
 225                         break;
 226                 }
 227         }
 228         if (!dev->int_in_endpointAddr) {
 229                 retval = -ENODEV;
 230                 dev_err(&interface->dev, "Could not find endpoints\n");
 231                 goto error;
 232         }
 233
 234
 235         /* allocate control URB */
 236         dev->cntl_urb = usb_alloc_urb(0, GFP_KERNEL);
 237         if (!dev->cntl_urb) {
 238                 dev_err(&interface->dev, "Could not allocate control URB\n");
 239                 goto error;
 240         }
 241
 242         /* allocate buffer for control req */
 243         dev->cntl_req = kmalloc(YUREX_BUF_SIZE, GFP_KERNEL);
 244         if (!dev->cntl_req) {
 245                 dev_err(&interface->dev, "Could not allocate cntl_req\n");
 246                 goto error;
 247         }
 248
 249         /* allocate buffer for control msg */
 250         dev->cntl_buffer = usb_alloc_coherent(dev->udev, YUREX_BUF_SIZE,
 251                                               GFP_KERNEL,
 252                                               &dev->cntl_urb->transfer_dma);
 253         if (!dev->cntl_buffer) {
 254                 dev_err(&interface->dev, "Could not allocate cntl_buffer\n");
 255                 goto error;
 256         }
 257
 258         /* configure control URB */
 259         dev->cntl_req->bRequestType = USB_DIR_OUT | USB_TYPE_CLASS |
 260                                       USB_RECIP_INTERFACE;
 261         dev->cntl_req->bRequest = HID_REQ_SET_REPORT;
 262         dev->cntl_req->wValue   = cpu_to_le16((HID_OUTPUT_REPORT + 1) << 8);
 263         dev->cntl_req->wIndex   = cpu_to_le16(iface_desc->desc.bInterfaceNumber);
 264         dev->cntl_req->wLength  = cpu_to_le16(YUREX_BUF_SIZE);
 265
 266         usb_fill_control_urb(dev->cntl_urb, dev->udev,
 267                              usb_sndctrlpipe(dev->udev, 0),
 268                              (void *)dev->cntl_req, dev->cntl_buffer,
 269                              YUREX_BUF_SIZE, yurex_control_callback, dev);
 270         dev->cntl_urb->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
 271
 272
 273         /* allocate interrupt URB */
 274         dev->urb = usb_alloc_urb(0, GFP_KERNEL);
 275         if (!dev->urb) {
 276                 dev_err(&interface->dev, "Could not allocate URB\n");
 277                 goto error;
 278         }
 279
 280         /* allocate buffer for interrupt in */
 281         dev->int_buffer = usb_alloc_coherent(dev->udev, YUREX_BUF_SIZE,
 282                                         GFP_KERNEL, &dev->urb->transfer_dma);
 283         if (!dev->int_buffer) {
 284                 dev_err(&interface->dev, "Could not allocate int_buffer\n");
 285                 goto error;
 286         }
 287
 288         /* configure interrupt URB */
 289         usb_fill_int_urb(dev->urb, dev->udev,
 290                          usb_rcvintpipe(dev->udev, dev->int_in_endpointAddr),
 291                          dev->int_buffer, YUREX_BUF_SIZE, yurex_interrupt,
 292                          dev, 1);
 293         dev->urb->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
 294         if (usb_submit_urb(dev->urb, GFP_KERNEL)) {
 295                 retval = -EIO;
 296                 dev_err(&interface->dev, "Could not submitting URB\n");
 297                 goto error;
 298         }
 299
 300         /* save our data pointer in this interface device */
 301         usb_set_intfdata(interface, dev);
 302         dev->bbu = -1;
 303
 304         /* we can register the device now, as it is ready */
 305         retval = usb_register_dev(interface, &yurex_class);
 306         if (retval) {
 307                 dev_err(&interface->dev,
 308                         "Not able to get a minor for this device.\n");
 309                 usb_set_intfdata(interface, NULL);
 310                 goto error;
 311         }
 312
 313         dev_info(&interface->dev,
 314                  "USB YUREX device now attached to Yurex #%d\n",
 315                  interface->minor);
 316
 317         return 0;
 318
 319 error:
 320         if (dev)
 321                 /* this frees allocated memory */
 322                 kref_put(&dev->kref, yurex_delete);
 323         return retval;
 324 }
 325
 326 static void yurex_disconnect(struct usb_interface *interface)
 327 {
 328         struct usb_yurex *dev;
 329         int minor = interface->minor;
 330
 331         dev = usb_get_intfdata(interface);
 332         usb_set_intfdata(interface, NULL);
 333
 334         /* give back our minor */
 335         usb_deregister_dev(interface, &yurex_class);
 336
 337         /* prevent more I/O from starting */
 338         usb_poison_urb(dev->urb);
 339         usb_poison_urb(dev->cntl_urb);
 340         mutex_lock(&dev->io_mutex);
 341         dev->disconnected = 1;
 342         mutex_unlock(&dev->io_mutex);
 343
 344         /* wakeup waiters */
 345         kill_fasync(&dev->async_queue, SIGIO, POLL_IN);
 346         wake_up_interruptible(&dev->waitq);
 347
 348         /* decrement our usage count */
 349         kref_put(&dev->kref, yurex_delete);
 350
 351         dev_info(&interface->dev, "USB YUREX #%d now disconnected\n", minor);
 352 }
 353
 354 static struct usb_driver yurex_driver = {
 355         .name =         "yurex",
 356         .probe =        yurex_probe,
 357         .disconnect =   yurex_disconnect,
 358         .id_table =     yurex_table,
 359 };
 360
 361
 362 static int yurex_fasync(int fd, struct file *file, int on)
 363 {
 364         struct usb_yurex *dev;
 365
 366         dev = file->private_data;
 367         return fasync_helper(fd, file, on, &dev->async_queue);
 368 }
 369
 370 static int yurex_open(struct inode *inode, struct file *file)
 371 {
 372         struct usb_yurex *dev;
 373         struct usb_interface *interface;
 374         int subminor;
 375         int retval = 0;
 376
 377         subminor = iminor(inode);
 378
 379         interface = usb_find_interface(&yurex_driver, subminor);
 380         if (!interface) {
 381                 printk(KERN_ERR "%s - error, can't find device for minor %d",
 382                        __func__, subminor);
 383                 retval = -ENODEV;
 384                 goto exit;
 385         }
 386
 387         dev = usb_get_intfdata(interface);
 388         if (!dev) {
 389                 retval = -ENODEV;
 390                 goto exit;
 391         }
 392
 393         /* increment our usage count for the device */
 394         kref_get(&dev->kref);
 395
 396         /* save our object in the file's private structure */
 397         mutex_lock(&dev->io_mutex);
 398         file->private_data = dev;
 399         mutex_unlock(&dev->io_mutex);
 400
 401 exit:
 402         return retval;
 403 }
 404
 405 static int yurex_release(struct inode *inode, struct file *file)
 406 {
 407         struct usb_yurex *dev;
 408
 409         dev = file->private_data;
 410         if (dev == NULL)
 411                 return -ENODEV;
 412
 413         /* decrement the count on our device */
 414         kref_put(&dev->kref, yurex_delete);
 415         return 0;
 416 }
 417
 418 static ssize_t yurex_read(struct file *file, char __user *buffer, size_t count,
 419                           loff_t *ppos)
 420 {
 421         struct usb_yurex *dev;
 422         int len = 0;
 423         char in_buffer[20];
 424         unsigned long flags;
 425
 426         dev = file->private_data;
 427
 428         mutex_lock(&dev->io_mutex);
 429         if (dev->disconnected) {                /* already disconnected */
 430                 mutex_unlock(&dev->io_mutex);
 431                 return -ENODEV;
 432         }
 433
 434         spin_lock_irqsave(&dev->lock, flags);
 435         len = snprintf(in_buffer, 20, "%lld\n", dev->bbu);
 436         spin_unlock_irqrestore(&dev->lock, flags);
 437         mutex_unlock(&dev->io_mutex);
 438
 439         if (WARN_ON_ONCE(len >= sizeof(in_buffer)))
 440                 return -EIO;
 441
 442         return simple_read_from_buffer(buffer, count, ppos, in_buffer, len);
 443 }
 444
 445 static ssize_t yurex_write(struct file *file, const char __user *user_buffer,
 446                            size_t count, loff_t *ppos)
 447 {
 448         struct usb_yurex *dev;
 449         int i, set = 0, retval = 0;
 450         char buffer[16 + 1];
 451         char *data = buffer;
 452         unsigned long long c, c2 = 0;
 453         signed long timeout = 0;
 454         DEFINE_WAIT(wait);
 455
 456         count = min(sizeof(buffer) - 1, count);
 457         dev = file->private_data;
 458
 459         /* verify that we actually have some data to write */
 460         if (count == 0)
 461                 goto error;
 462
 463         mutex_lock(&dev->io_mutex);
 464         if (dev->disconnected) {                /* already disconnected */
 465                 mutex_unlock(&dev->io_mutex);
 466                 retval = -ENODEV;
 467                 goto error;
 468         }
 469
 470         if (copy_from_user(buffer, user_buffer, count)) {
 471                 mutex_unlock(&dev->io_mutex);
 472                 retval = -EFAULT;
 473                 goto error;
 474         }
 475         buffer[count] = 0;
 476         memset(dev->cntl_buffer, CMD_PADDING, YUREX_BUF_SIZE);
 477
 478         switch (buffer[0]) {
 479         case CMD_ANIMATE:
 480         case CMD_LED:
 481                 dev->cntl_buffer[0] = buffer[0];
 482                 dev->cntl_buffer[1] = buffer[1];
 483                 dev->cntl_buffer[2] = CMD_EOF;
 484                 break;
 485         case CMD_READ:
 486         case CMD_VERSION:
 487                 dev->cntl_buffer[0] = buffer[0];
 488                 dev->cntl_buffer[1] = 0x00;
 489                 dev->cntl_buffer[2] = CMD_EOF;
 490                 break;
 491         case CMD_SET:
 492                 data++;
 493                 /* FALL THROUGH */
 494         case '0' ... '9':
 495                 set = 1;
 496                 c = c2 = simple_strtoull(data, NULL, 0);
 497                 dev->cntl_buffer[0] = CMD_SET;
 498                 for (i = 1; i < 6; i++) {
 499                         dev->cntl_buffer[i] = (c>>32) & 0xff;
 500                         c <<= 8;
 501                 }
 502                 buffer[6] = CMD_EOF;
 503                 break;
 504         default:
 505                 mutex_unlock(&dev->io_mutex);
 506                 return -EINVAL;
 507         }
 508
 509         /* send the data as the control msg */
 510         prepare_to_wait(&dev->waitq, &wait, TASK_INTERRUPTIBLE);
 511         dev_dbg(&dev->interface->dev, "%s - submit %c\n", __func__,
 512                 dev->cntl_buffer[0]);
 513         retval = usb_submit_urb(dev->cntl_urb, GFP_ATOMIC);
 514         if (retval >= 0)
 515                 timeout = schedule_timeout(YUREX_WRITE_TIMEOUT);
 516         finish_wait(&dev->waitq, &wait);
 517
 518         /* make sure URB is idle after timeout or (spurious) CMD_ACK */
 519         usb_kill_urb(dev->cntl_urb);
 520
 521         mutex_unlock(&dev->io_mutex);
 522
 523         if (retval < 0) {
 524                 dev_err(&dev->interface->dev,
 525                         "%s - failed to send bulk msg, error %d\n",
 526                         __func__, retval);
 527                 goto error;
 528         }
 529         if (set && timeout)
 530                 dev->bbu = c2;
 531         return timeout ? count : -EIO;
 532
 533 error:
 534         return retval;
 535 }
 536
 537 static const struct file_operations yurex_fops = {
 538         .owner =        THIS_MODULE,
 539         .read =         yurex_read,
 540         .write =        yurex_write,
 541         .open =         yurex_open,
 542         .release =      yurex_release,
 543         .fasync =       yurex_fasync,
 544         .llseek =       default_llseek,
 545 };
 546
 547 module_usb_driver(yurex_driver);
 548
 549 MODULE_LICENSE("GPL");