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