drivers/staging/media/lirc/lirc_sasem.c

   1 /*
   2  * lirc_sasem.c - USB remote support for LIRC
   3  * Version 0.5
   4  *
   5  * Copyright (C) 2004-2005 Oliver Stabel <oliver.stabel@gmx.de>
   6  *                       Tim Davies <tim@opensystems.net.au>
   7  *
   8  * This driver was derived from:
   9  *   Venky Raju <dev@venky.ws>
  10  *      "lirc_imon - "LIRC/VFD driver for Ahanix/Soundgraph IMON IR/VFD"
  11  *   Paul Miller <pmiller9@users.sourceforge.net>'s 2003-2004
  12  *      "lirc_atiusb - USB remote support for LIRC"
  13  *   Culver Consulting Services <henry@culcon.com>'s 2003
  14  *      "Sasem OnAir VFD/IR USB driver"
  15  *
  16  *
  17  * NOTE - The LCDproc iMon driver should work with this module.  More info at
  18  *      http://www.frogstorm.info/sasem
  19  */
  20
  21 /*
  22  *  This program is free software; you can redistribute it and/or modify
  23  *  it under the terms of the GNU General Public License as published by
  24  *  the Free Software Foundation; either version 2 of the License, or
  25  *  (at your option) any later version.
  26  *
  27  *  This program is distributed in the hope that it will be useful,
  28  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  29  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  30  *  GNU General Public License for more details.
  31  *
  32  *  You should have received a copy of the GNU General Public License
  33  *  along with this program; if not, write to the Free Software
  34  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  35  */
  36
  37 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
  38
  39 #include <linux/errno.h>
  40 #include <linux/kernel.h>
  41 #include <linux/module.h>
  42 #include <linux/slab.h>
  43 #include <linux/uaccess.h>
  44 #include <linux/usb.h>
  45
  46 #include <media/lirc.h>
  47 #include <media/lirc_dev.h>
  48
  49
  50 #define MOD_AUTHOR      "Oliver Stabel <oliver.stabel@gmx.de>, " \
  51                         "Tim Davies <tim@opensystems.net.au>"
  52 #define MOD_DESC        "USB Driver for Sasem Remote Controller V1.1"
  53 #define MOD_NAME        "lirc_sasem"
  54 #define MOD_VERSION     "0.5"
  55
  56 #define VFD_MINOR_BASE  144     /* Same as LCD */
  57 #define DEVICE_NAME     "lcd%d"
  58
  59 #define BUF_CHUNK_SIZE  8
  60 #define BUF_SIZE        128
  61
  62 #define IOCTL_LCD_CONTRAST 1
  63
  64 /*** P R O T O T Y P E S ***/
  65
  66 /* USB Callback prototypes */
  67 static int sasem_probe(struct usb_interface *interface,
  68                         const struct usb_device_id *id);
  69 static void sasem_disconnect(struct usb_interface *interface);
  70 static void usb_rx_callback(struct urb *urb);
  71 static void usb_tx_callback(struct urb *urb);
  72
  73 /* VFD file_operations function prototypes */
  74 static int vfd_open(struct inode *inode, struct file *file);
  75 static long vfd_ioctl(struct file *file, unsigned cmd, unsigned long arg);
  76 static int vfd_close(struct inode *inode, struct file *file);
  77 static ssize_t vfd_write(struct file *file, const char __user *buf,
  78                                 size_t n_bytes, loff_t *pos);
  79
  80 /* LIRC driver function prototypes */
  81 static int ir_open(void *data);
  82 static void ir_close(void *data);
  83
  84 /*** G L O B A L S ***/
  85 #define SASEM_DATA_BUF_SZ       32
  86
  87 struct sasem_context {
  88
  89         struct usb_device *dev;
  90         int vfd_isopen;                 /* VFD port has been opened */
  91         unsigned int vfd_contrast;      /* VFD contrast */
  92         int ir_isopen;                  /* IR port has been opened */
  93         int dev_present;                /* USB device presence */
  94         struct mutex ctx_lock;          /* to lock this object */
  95         wait_queue_head_t remove_ok;    /* For unexpected USB disconnects */
  96
  97         struct lirc_driver *driver;
  98         struct usb_endpoint_descriptor *rx_endpoint;
  99         struct usb_endpoint_descriptor *tx_endpoint;
 100         struct urb *rx_urb;
 101         struct urb *tx_urb;
 102         unsigned char usb_rx_buf[8];
 103         unsigned char usb_tx_buf[8];
 104
 105         struct tx_t {
 106                 unsigned char data_buf[SASEM_DATA_BUF_SZ]; /* user data
 107                                                             * buffer */
 108                 struct completion finished;  /* wait for write to finish  */
 109                 atomic_t busy;               /* write in progress */
 110                 int status;                  /* status of tx completion */
 111         } tx;
 112
 113         /* for dealing with repeat codes (wish there was a toggle bit!) */
 114         struct timeval presstime;
 115         char lastcode[8];
 116         int codesaved;
 117 };
 118
 119 /* VFD file operations */
 120 static const struct file_operations vfd_fops = {
 121         .owner          = THIS_MODULE,
 122         .open           = &vfd_open,
 123         .write          = vfd_write,
 124         .unlocked_ioctl = &vfd_ioctl,
 125         .release        = &vfd_close,
 126         .llseek         = noop_llseek,
 127 };
 128
 129 /* USB Device ID for Sasem USB Control Board */
 130 static struct usb_device_id sasem_usb_id_table[] = {
 131         /* Sasem USB Control Board */
 132         { USB_DEVICE(0x11ba, 0x0101) },
 133         /* Terminating entry */
 134         {}
 135 };
 136
 137 /* USB Device data */
 138 static struct usb_driver sasem_driver = {
 139         .name           = MOD_NAME,
 140         .probe          = sasem_probe,
 141         .disconnect     = sasem_disconnect,
 142         .id_table       = sasem_usb_id_table,
 143 };
 144
 145 static struct usb_class_driver sasem_class = {
 146         .name           = DEVICE_NAME,
 147         .fops           = &vfd_fops,
 148         .minor_base     = VFD_MINOR_BASE,
 149 };
 150
 151 /* to prevent races between open() and disconnect() */
 152 static DEFINE_MUTEX(disconnect_lock);
 153
 154 static int debug;
 155
 156
 157 /*** M O D U L E   C O D E ***/
 158
 159 MODULE_AUTHOR(MOD_AUTHOR);
 160 MODULE_DESCRIPTION(MOD_DESC);
 161 MODULE_LICENSE("GPL");
 162 module_param(debug, int, S_IRUGO | S_IWUSR);
 163 MODULE_PARM_DESC(debug, "Debug messages: 0=no, 1=yes (default: no)");
 164
 165 static void delete_context(struct sasem_context *context)
 166 {
 167         usb_free_urb(context->tx_urb);  /* VFD */
 168         usb_free_urb(context->rx_urb);  /* IR */
 169         lirc_buffer_free(context->driver->rbuf);
 170         kfree(context->driver->rbuf);
 171         kfree(context->driver);
 172         kfree(context);
 173
 174         if (debug)
 175                 pr_info("%s: context deleted\n", __func__);
 176 }
 177
 178 static void deregister_from_lirc(struct sasem_context *context)
 179 {
 180         int retval;
 181         int minor = context->driver->minor;
 182
 183         retval = lirc_unregister_driver(minor);
 184         if (retval)
 185                 pr_err("%s: unable to deregister from lirc (%d)\n",
 186                        __func__, retval);
 187         else
 188                 pr_info("Deregistered Sasem driver (minor:%d)\n", minor);
 189
 190 }
 191
 192 /**
 193  * Called when the VFD device (e.g. /dev/usb/lcd)
 194  * is opened by the application.
 195  */
 196 static int vfd_open(struct inode *inode, struct file *file)
 197 {
 198         struct usb_interface *interface;
 199         struct sasem_context *context = NULL;
 200         int subminor;
 201         int retval = 0;
 202
 203         /* prevent races with disconnect */
 204         mutex_lock(&disconnect_lock);
 205
 206         subminor = iminor(inode);
 207         interface = usb_find_interface(&sasem_driver, subminor);
 208         if (!interface) {
 209                 pr_err("%s: could not find interface for minor %d\n",
 210                        __func__, subminor);
 211                 retval = -ENODEV;
 212                 goto exit;
 213         }
 214         context = usb_get_intfdata(interface);
 215
 216         if (!context) {
 217                 dev_err(&interface->dev,
 218                         "%s: no context found for minor %d\n",
 219                         __func__, subminor);
 220                 retval = -ENODEV;
 221                 goto exit;
 222         }
 223
 224         mutex_lock(&context->ctx_lock);
 225
 226         if (context->vfd_isopen) {
 227                 dev_err(&interface->dev,
 228                         "%s: VFD port is already open", __func__);
 229                 retval = -EBUSY;
 230         } else {
 231                 context->vfd_isopen = 1;
 232                 file->private_data = context;
 233                 dev_info(&interface->dev, "VFD port opened\n");
 234         }
 235
 236         mutex_unlock(&context->ctx_lock);
 237
 238 exit:
 239         mutex_unlock(&disconnect_lock);
 240         return retval;
 241 }
 242
 243 /**
 244  * Called when the VFD device (e.g. /dev/usb/lcd)
 245  * is closed by the application.
 246  */
 247 static long vfd_ioctl(struct file *file, unsigned cmd, unsigned long arg)
 248 {
 249         struct sasem_context *context = NULL;
 250
 251         context = (struct sasem_context *) file->private_data;
 252
 253         if (!context) {
 254                 pr_err("%s: no context for device\n", __func__);
 255                 return -ENODEV;
 256         }
 257
 258         mutex_lock(&context->ctx_lock);
 259
 260         switch (cmd) {
 261         case IOCTL_LCD_CONTRAST:
 262                 if (arg > 1000)
 263                         arg = 1000;
 264                 context->vfd_contrast = (unsigned int)arg;
 265                 break;
 266         default:
 267                 pr_info("Unknown IOCTL command\n");
 268                 mutex_unlock(&context->ctx_lock);
 269                 return -ENOIOCTLCMD;  /* not supported */
 270         }
 271
 272         mutex_unlock(&context->ctx_lock);
 273         return 0;
 274 }
 275
 276 /**
 277  * Called when the VFD device (e.g. /dev/usb/lcd)
 278  * is closed by the application.
 279  */
 280 static int vfd_close(struct inode *inode, struct file *file)
 281 {
 282         struct sasem_context *context = NULL;
 283         int retval = 0;
 284
 285         context = (struct sasem_context *) file->private_data;
 286
 287         if (!context) {
 288                 pr_err("%s: no context for device\n", __func__);
 289                 return -ENODEV;
 290         }
 291
 292         mutex_lock(&context->ctx_lock);
 293
 294         if (!context->vfd_isopen) {
 295                 dev_err(&context->dev->dev, "%s: VFD is not open\n", __func__);
 296                 retval = -EIO;
 297         } else {
 298                 context->vfd_isopen = 0;
 299                 dev_info(&context->dev->dev, "VFD port closed\n");
 300                 if (!context->dev_present && !context->ir_isopen) {
 301
 302                         /* Device disconnected before close and IR port is
 303                          * not open. If IR port is open, context will be
 304                          * deleted by ir_close. */
 305                         mutex_unlock(&context->ctx_lock);
 306                         delete_context(context);
 307                         return retval;
 308                 }
 309         }
 310
 311         mutex_unlock(&context->ctx_lock);
 312         return retval;
 313 }
 314
 315 /**
 316  * Sends a packet to the VFD.
 317  */
 318 static int send_packet(struct sasem_context *context)
 319 {
 320         unsigned int pipe;
 321         int interval = 0;
 322         int retval = 0;
 323
 324         pipe = usb_sndintpipe(context->dev,
 325                         context->tx_endpoint->bEndpointAddress);
 326         interval = context->tx_endpoint->bInterval;
 327
 328         usb_fill_int_urb(context->tx_urb, context->dev, pipe,
 329                 context->usb_tx_buf, sizeof(context->usb_tx_buf),
 330                 usb_tx_callback, context, interval);
 331
 332         context->tx_urb->actual_length = 0;
 333
 334         init_completion(&context->tx.finished);
 335         atomic_set(&(context->tx.busy), 1);
 336
 337         retval =  usb_submit_urb(context->tx_urb, GFP_KERNEL);
 338         if (retval) {
 339                 atomic_set(&(context->tx.busy), 0);
 340                 dev_err(&context->dev->dev, "%s: error submitting urb (%d)\n",
 341                         __func__, retval);
 342         } else {
 343                 /* Wait for transmission to complete (or abort) */
 344                 mutex_unlock(&context->ctx_lock);
 345                 wait_for_completion(&context->tx.finished);
 346                 mutex_lock(&context->ctx_lock);
 347
 348                 retval = context->tx.status;
 349                 if (retval)
 350                         dev_err(&context->dev->dev,
 351                                 "%s: packet tx failed (%d)\n",
 352                                 __func__, retval);
 353         }
 354
 355         return retval;
 356 }
 357
 358 /**
 359  * Writes data to the VFD.  The Sasem VFD is 2x16 characters
 360  * and requires data in 9 consecutive USB interrupt packets,
 361  * each packet carrying 8 bytes.
 362  */
 363 static ssize_t vfd_write(struct file *file, const char __user *buf,
 364                                 size_t n_bytes, loff_t *pos)
 365 {
 366         int i;
 367         int retval = 0;
 368         struct sasem_context *context;
 369         int *data_buf = NULL;
 370
 371         context = (struct sasem_context *) file->private_data;
 372         if (!context) {
 373                 pr_err("%s: no context for device\n", __func__);
 374                 return -ENODEV;
 375         }
 376
 377         mutex_lock(&context->ctx_lock);
 378
 379         if (!context->dev_present) {
 380                 pr_err("%s: no Sasem device present\n", __func__);
 381                 retval = -ENODEV;
 382                 goto exit;
 383         }
 384
 385         if (n_bytes <= 0 || n_bytes > SASEM_DATA_BUF_SZ) {
 386                 dev_err(&context->dev->dev, "%s: invalid payload size\n",
 387                         __func__);
 388                 retval = -EINVAL;
 389                 goto exit;
 390         }
 391
 392         data_buf = memdup_user((void const __user *)buf, n_bytes);
 393         if (IS_ERR(data_buf)) {
 394                 retval = PTR_ERR(data_buf);
 395                 data_buf = NULL;
 396                 goto exit;
 397         }
 398
 399         memcpy(context->tx.data_buf, data_buf, n_bytes);
 400
 401         /* Pad with spaces */
 402         for (i = n_bytes; i < SASEM_DATA_BUF_SZ; ++i)
 403                 context->tx.data_buf[i] = ' ';
 404
 405         /* Nine 8 byte packets to be sent */
 406         /* NOTE: "\x07\x01\0\0\0\0\0\0" or "\x0c\0\0\0\0\0\0\0"
 407          *       will clear the VFD */
 408         for (i = 0; i < 9; i++) {
 409                 switch (i) {
 410                 case 0:
 411                         memcpy(context->usb_tx_buf, "\x07\0\0\0\0\0\0\0", 8);
 412                         context->usb_tx_buf[1] = (context->vfd_contrast) ?
 413                                 (0x2B - (context->vfd_contrast - 1) / 250)
 414                                 : 0x2B;
 415                         break;
 416                 case 1:
 417                         memcpy(context->usb_tx_buf, "\x09\x01\0\0\0\0\0\0", 8);
 418                         break;
 419                 case 2:
 420                         memcpy(context->usb_tx_buf, "\x0b\x01\0\0\0\0\0\0", 8);
 421                         break;
 422                 case 3:
 423                         memcpy(context->usb_tx_buf, context->tx.data_buf, 8);
 424                         break;
 425                 case 4:
 426                         memcpy(context->usb_tx_buf,
 427                                context->tx.data_buf + 8, 8);
 428                         break;
 429                 case 5:
 430                         memcpy(context->usb_tx_buf, "\x09\x01\0\0\0\0\0\0", 8);
 431                         break;
 432                 case 6:
 433                         memcpy(context->usb_tx_buf, "\x0b\x02\0\0\0\0\0\0", 8);
 434                         break;
 435                 case 7:
 436                         memcpy(context->usb_tx_buf,
 437                                context->tx.data_buf + 16, 8);
 438                         break;
 439                 case 8:
 440                         memcpy(context->usb_tx_buf,
 441                                context->tx.data_buf + 24, 8);
 442                         break;
 443                 }
 444                 retval = send_packet(context);
 445                 if (retval) {
 446                         dev_err(&context->dev->dev,
 447                                 "%s: send packet failed for packet #%d\n",
 448                                 __func__, i);
 449                         goto exit;
 450                 }
 451         }
 452 exit:
 453
 454         mutex_unlock(&context->ctx_lock);
 455         kfree(data_buf);
 456
 457         return (!retval) ? n_bytes : retval;
 458 }
 459
 460 /**
 461  * Callback function for USB core API: transmit data
 462  */
 463 static void usb_tx_callback(struct urb *urb)
 464 {
 465         struct sasem_context *context;
 466
 467         if (!urb)
 468                 return;
 469         context = (struct sasem_context *) urb->context;
 470         if (!context)
 471                 return;
 472
 473         context->tx.status = urb->status;
 474
 475         /* notify waiters that write has finished */
 476         atomic_set(&context->tx.busy, 0);
 477         complete(&context->tx.finished);
 478 }
 479
 480 /**
 481  * Called by lirc_dev when the application opens /dev/lirc
 482  */
 483 static int ir_open(void *data)
 484 {
 485         int retval = 0;
 486         struct sasem_context *context;
 487
 488         /* prevent races with disconnect */
 489         mutex_lock(&disconnect_lock);
 490
 491         context = data;
 492
 493         mutex_lock(&context->ctx_lock);
 494
 495         if (context->ir_isopen) {
 496                 dev_err(&context->dev->dev, "%s: IR port is already open\n",
 497                         __func__);
 498                 retval = -EBUSY;
 499                 goto exit;
 500         }
 501
 502         usb_fill_int_urb(context->rx_urb, context->dev,
 503                 usb_rcvintpipe(context->dev,
 504                                 context->rx_endpoint->bEndpointAddress),
 505                 context->usb_rx_buf, sizeof(context->usb_rx_buf),
 506                 usb_rx_callback, context, context->rx_endpoint->bInterval);
 507
 508         retval = usb_submit_urb(context->rx_urb, GFP_KERNEL);
 509
 510         if (retval)
 511                 dev_err(&context->dev->dev,
 512                         "%s: usb_submit_urb failed for ir_open (%d)\n",
 513                         __func__, retval);
 514         else {
 515                 context->ir_isopen = 1;
 516                 dev_info(&context->dev->dev, "IR port opened\n");
 517         }
 518
 519 exit:
 520         mutex_unlock(&context->ctx_lock);
 521
 522         mutex_unlock(&disconnect_lock);
 523         return retval;
 524 }
 525
 526 /**
 527  * Called by lirc_dev when the application closes /dev/lirc
 528  */
 529 static void ir_close(void *data)
 530 {
 531         struct sasem_context *context;
 532
 533         context = data;
 534         if (!context) {
 535                 pr_err("%s: no context for device\n", __func__);
 536                 return;
 537         }
 538
 539         mutex_lock(&context->ctx_lock);
 540
 541         usb_kill_urb(context->rx_urb);
 542         context->ir_isopen = 0;
 543         pr_info("IR port closed\n");
 544
 545         if (!context->dev_present) {
 546
 547                 /*
 548                  * Device disconnected while IR port was
 549                  * still open. Driver was not deregistered
 550                  * at disconnect time, so do it now.
 551                  */
 552                 deregister_from_lirc(context);
 553
 554                 if (!context->vfd_isopen) {
 555
 556                         mutex_unlock(&context->ctx_lock);
 557                         delete_context(context);
 558                         return;
 559                 }
 560                 /* If VFD port is open, context will be deleted by vfd_close */
 561         }
 562
 563         mutex_unlock(&context->ctx_lock);
 564 }
 565
 566 /**
 567  * Process the incoming packet
 568  */
 569 static void incoming_packet(struct sasem_context *context,
 570                                    struct urb *urb)
 571 {
 572         int len = urb->actual_length;
 573         unsigned char *buf = urb->transfer_buffer;
 574         long ms;
 575         struct timeval tv;
 576
 577         if (len != 8) {
 578                 dev_warn(&context->dev->dev,
 579                          "%s: invalid incoming packet size (%d)\n",
 580                          __func__, len);
 581                 return;
 582         }
 583
 584         if (debug)
 585                 dev_info(&context->dev->dev, "Incoming data: %*ph\n", len, buf);
 586         /*
 587          * Lirc could deal with the repeat code, but we really need to block it
 588          * if it arrives too late.  Otherwise we could repeat the wrong code.
 589          */
 590
 591         /* get the time since the last button press */
 592         do_gettimeofday(&tv);
 593         ms = (tv.tv_sec - context->presstime.tv_sec) * 1000 +
 594              (tv.tv_usec - context->presstime.tv_usec) / 1000;
 595
 596         if (memcmp(buf, "\x08\0\0\0\0\0\0\0", 8) == 0) {
 597                 /*
 598                  * the repeat code is being sent, so we copy
 599                  * the old code to LIRC
 600                  */
 601
 602                 /*
 603                  * NOTE: Only if the last code was less than 250ms ago
 604                  * - no one should be able to push another (undetected) button
 605                  *   in that time and then get a false repeat of the previous
 606                  *   press but it is long enough for a genuine repeat
 607                  */
 608                 if ((ms < 250) && (context->codesaved != 0)) {
 609                         memcpy(buf, &context->lastcode, 8);
 610                         context->presstime.tv_sec = tv.tv_sec;
 611                         context->presstime.tv_usec = tv.tv_usec;
 612                 }
 613         } else {
 614                 /* save the current valid code for repeats */
 615                 memcpy(&context->lastcode, buf, 8);
 616                 /*
 617                  * set flag to signal a valid code was save;
 618                  * just for safety reasons
 619                  */
 620                 context->codesaved = 1;
 621                 context->presstime.tv_sec = tv.tv_sec;
 622                 context->presstime.tv_usec = tv.tv_usec;
 623         }
 624
 625         lirc_buffer_write(context->driver->rbuf, buf);
 626         wake_up(&context->driver->rbuf->wait_poll);
 627 }
 628
 629 /**
 630  * Callback function for USB core API: receive data
 631  */
 632 static void usb_rx_callback(struct urb *urb)
 633 {
 634         struct sasem_context *context;
 635
 636         if (!urb)
 637                 return;
 638         context = (struct sasem_context *) urb->context;
 639         if (!context)
 640                 return;
 641
 642         switch (urb->status) {
 643
 644         case -ENOENT:           /* usbcore unlink successful! */
 645                 return;
 646
 647         case 0:
 648                 if (context->ir_isopen)
 649                         incoming_packet(context, urb);
 650                 break;
 651
 652         default:
 653                 dev_warn(&urb->dev->dev, "%s: status (%d): ignored",
 654                          __func__, urb->status);
 655                 break;
 656         }
 657
 658         usb_submit_urb(context->rx_urb, GFP_ATOMIC);
 659 }
 660
 661
 662
 663 /**
 664  * Callback function for USB core API: Probe
 665  */
 666 static int sasem_probe(struct usb_interface *interface,
 667                         const struct usb_device_id *id)
 668 {
 669         struct usb_device *dev = NULL;
 670         struct usb_host_interface *iface_desc = NULL;
 671         struct usb_endpoint_descriptor *rx_endpoint = NULL;
 672         struct usb_endpoint_descriptor *tx_endpoint = NULL;
 673         struct urb *rx_urb = NULL;
 674         struct urb *tx_urb = NULL;
 675         struct lirc_driver *driver = NULL;
 676         struct lirc_buffer *rbuf = NULL;
 677         int lirc_minor = 0;
 678         int num_endpoints;
 679         int retval = 0;
 680         int vfd_ep_found;
 681         int ir_ep_found;
 682         int alloc_status;
 683         struct sasem_context *context = NULL;
 684         int i;
 685
 686         dev_info(&interface->dev, "%s: found Sasem device\n", __func__);
 687
 688
 689         dev = usb_get_dev(interface_to_usbdev(interface));
 690         iface_desc = interface->cur_altsetting;
 691         num_endpoints = iface_desc->desc.bNumEndpoints;
 692
 693         /*
 694          * Scan the endpoint list and set:
 695          *      first input endpoint = IR endpoint
 696          *      first output endpoint = VFD endpoint
 697          */
 698
 699         ir_ep_found = 0;
 700         vfd_ep_found = 0;
 701
 702         for (i = 0; i < num_endpoints && !(ir_ep_found && vfd_ep_found); ++i) {
 703
 704                 struct usb_endpoint_descriptor *ep;
 705                 int ep_dir;
 706                 int ep_type;
 707
 708                 ep = &iface_desc->endpoint [i].desc;
 709                 ep_dir = ep->bEndpointAddress & USB_ENDPOINT_DIR_MASK;
 710                 ep_type = ep->bmAttributes & USB_ENDPOINT_XFERTYPE_MASK;
 711
 712                 if (!ir_ep_found &&
 713                         ep_dir == USB_DIR_IN &&
 714                         ep_type == USB_ENDPOINT_XFER_INT) {
 715
 716                         rx_endpoint = ep;
 717                         ir_ep_found = 1;
 718                         if (debug)
 719                                 dev_info(&interface->dev,
 720                                         "%s: found IR endpoint\n", __func__);
 721
 722                 } else if (!vfd_ep_found &&
 723                         ep_dir == USB_DIR_OUT &&
 724                         ep_type == USB_ENDPOINT_XFER_INT) {
 725
 726                         tx_endpoint = ep;
 727                         vfd_ep_found = 1;
 728                         if (debug)
 729                                 dev_info(&interface->dev,
 730                                         "%s: found VFD endpoint\n", __func__);
 731                 }
 732         }
 733
 734         /* Input endpoint is mandatory */
 735         if (!ir_ep_found) {
 736                 dev_err(&interface->dev,
 737                         "%s: no valid input (IR) endpoint found.\n", __func__);
 738                 retval = -ENODEV;
 739                 goto exit;
 740         }
 741
 742         if (!vfd_ep_found)
 743                 dev_info(&interface->dev,
 744                         "%s: no valid output (VFD) endpoint found.\n",
 745                         __func__);
 746
 747
 748         /* Allocate memory */
 749         alloc_status = 0;
 750
 751         context = kzalloc(sizeof(struct sasem_context), GFP_KERNEL);
 752         if (!context) {
 753                 alloc_status = 1;
 754                 goto alloc_status_switch;
 755         }
 756         driver = kzalloc(sizeof(struct lirc_driver), GFP_KERNEL);
 757         if (!driver) {
 758                 alloc_status = 2;
 759                 goto alloc_status_switch;
 760         }
 761         rbuf = kmalloc(sizeof(struct lirc_buffer), GFP_KERNEL);
 762         if (!rbuf) {
 763                 alloc_status = 3;
 764                 goto alloc_status_switch;
 765         }
 766         if (lirc_buffer_init(rbuf, BUF_CHUNK_SIZE, BUF_SIZE)) {
 767                 dev_err(&interface->dev,
 768                         "%s: lirc_buffer_init failed\n", __func__);
 769                 alloc_status = 4;
 770                 goto alloc_status_switch;
 771         }
 772         rx_urb = usb_alloc_urb(0, GFP_KERNEL);
 773         if (!rx_urb) {
 774                 dev_err(&interface->dev,
 775                         "%s: usb_alloc_urb failed for IR urb\n", __func__);
 776                 alloc_status = 5;
 777                 goto alloc_status_switch;
 778         }
 779         if (vfd_ep_found) {
 780                 tx_urb = usb_alloc_urb(0, GFP_KERNEL);
 781                 if (!tx_urb) {
 782                         dev_err(&interface->dev,
 783                                 "%s: usb_alloc_urb failed for VFD urb",
 784                                 __func__);
 785                         alloc_status = 6;
 786                         goto alloc_status_switch;
 787                 }
 788         }
 789
 790         mutex_init(&context->ctx_lock);
 791
 792         strcpy(driver->name, MOD_NAME);
 793         driver->minor = -1;
 794         driver->code_length = 64;
 795         driver->sample_rate = 0;
 796         driver->features = LIRC_CAN_REC_LIRCCODE;
 797         driver->data = context;
 798         driver->rbuf = rbuf;
 799         driver->set_use_inc = ir_open;
 800         driver->set_use_dec = ir_close;
 801         driver->dev   = &interface->dev;
 802         driver->owner = THIS_MODULE;
 803
 804         mutex_lock(&context->ctx_lock);
 805
 806         lirc_minor = lirc_register_driver(driver);
 807         if (lirc_minor < 0) {
 808                 dev_err(&interface->dev,
 809                         "%s: lirc_register_driver failed\n", __func__);
 810                 alloc_status = 7;
 811                 retval = lirc_minor;
 812                 goto unlock;
 813         } else
 814                 dev_info(&interface->dev,
 815                          "%s: Registered Sasem driver (minor:%d)\n",
 816                          __func__, lirc_minor);
 817
 818         /* Needed while unregistering! */
 819         driver->minor = lirc_minor;
 820
 821         context->dev = dev;
 822         context->dev_present = 1;
 823         context->rx_endpoint = rx_endpoint;
 824         context->rx_urb = rx_urb;
 825         if (vfd_ep_found) {
 826                 context->tx_endpoint = tx_endpoint;
 827                 context->tx_urb = tx_urb;
 828                 context->vfd_contrast = 1000;   /* range 0 - 1000 */
 829         }
 830         context->driver = driver;
 831
 832         usb_set_intfdata(interface, context);
 833
 834         if (vfd_ep_found) {
 835
 836                 if (debug)
 837                         dev_info(&interface->dev,
 838                                  "Registering VFD with sysfs\n");
 839                 if (usb_register_dev(interface, &sasem_class))
 840                         /* Not a fatal error, so ignore */
 841                         dev_info(&interface->dev,
 842                                  "%s: could not get a minor number for VFD\n",
 843                                  __func__);
 844         }
 845
 846         dev_info(&interface->dev,
 847                  "%s: Sasem device on usb<%d:%d> initialized\n",
 848                  __func__, dev->bus->busnum, dev->devnum);
 849 unlock:
 850         mutex_unlock(&context->ctx_lock);
 851
 852 alloc_status_switch:
 853         switch (alloc_status) {
 854
 855         case 7:
 856                 if (vfd_ep_found)
 857                         usb_free_urb(tx_urb);
 858         case 6:
 859                 usb_free_urb(rx_urb);
 860                 /* fall-through */
 861         case 5:
 862                 lirc_buffer_free(rbuf);
 863                 /* fall-through */
 864         case 4:
 865                 kfree(rbuf);
 866                 /* fall-through */
 867         case 3:
 868                 kfree(driver);
 869                 /* fall-through */
 870         case 2:
 871                 kfree(context);
 872                 context = NULL;
 873                 /* fall-through */
 874         case 1:
 875                 if (retval == 0)
 876                         retval = -ENOMEM;
 877         }
 878
 879 exit:
 880         return retval;
 881 }
 882
 883 /**
 884  * Callback function for USB core API: disconnect
 885  */
 886 static void sasem_disconnect(struct usb_interface *interface)
 887 {
 888         struct sasem_context *context;
 889
 890         /* prevent races with ir_open()/vfd_open() */
 891         mutex_lock(&disconnect_lock);
 892
 893         context = usb_get_intfdata(interface);
 894         mutex_lock(&context->ctx_lock);
 895
 896         dev_info(&interface->dev, "%s: Sasem device disconnected\n",
 897                  __func__);
 898
 899         usb_set_intfdata(interface, NULL);
 900         context->dev_present = 0;
 901
 902         /* Stop reception */
 903         usb_kill_urb(context->rx_urb);
 904
 905         /* Abort ongoing write */
 906         if (atomic_read(&context->tx.busy)) {
 907
 908                 usb_kill_urb(context->tx_urb);
 909                 wait_for_completion(&context->tx.finished);
 910         }
 911
 912         /* De-register from lirc_dev if IR port is not open */
 913         if (!context->ir_isopen)
 914                 deregister_from_lirc(context);
 915
 916         usb_deregister_dev(interface, &sasem_class);
 917
 918         mutex_unlock(&context->ctx_lock);
 919
 920         if (!context->ir_isopen && !context->vfd_isopen)
 921                 delete_context(context);
 922
 923         mutex_unlock(&disconnect_lock);
 924 }
 925
 926 module_usb_driver(sasem_driver);