drivers/media/rc/redrat3.c

   1 /*
   2  * USB RedRat3 IR Transceiver rc-core driver
   3  *
   4  * Copyright (c) 2011 by Jarod Wilson <jarod@redhat.com>
   5  *  based heavily on the work of Stephen Cox, with additional
   6  *  help from RedRat Ltd.
   7  *
   8  * This driver began life based an an old version of the first-generation
   9  * lirc_mceusb driver from the lirc 0.7.2 distribution. It was then
  10  * significantly rewritten by Stephen Cox with the aid of RedRat Ltd's
  11  * Chris Dodge.
  12  *
  13  * The driver was then ported to rc-core and significantly rewritten again,
  14  * by Jarod, using the in-kernel mceusb driver as a guide, after an initial
  15  * port effort was started by Stephen.
  16  *
  17  * TODO LIST:
  18  * - fix lirc not showing repeats properly
  19  * --
  20  *
  21  * The RedRat3 is a USB transceiver with both send & receive,
  22  * with 2 separate sensors available for receive to enable
  23  * both good long range reception for general use, and good
  24  * short range reception when required for learning a signal.
  25  *
  26  * http://www.redrat.co.uk/
  27  *
  28  * It uses its own little protocol to communicate, the required
  29  * parts of which are embedded within this driver.
  30  * --
  31  *
  32  * This program is free software; you can redistribute it and/or modify
  33  * it under the terms of the GNU General Public License as published by
  34  * the Free Software Foundation; either version 2 of the License, or
  35  * (at your option) any later version.
  36  *
  37  * This program is distributed in the hope that it will be useful,
  38  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  39  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  40  * GNU General Public License for more details.
  41  *
  42  * You should have received a copy of the GNU General Public License
  43  * along with this program; if not, write to the Free Software
  44  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  45  *
  46  */
  47
  48 #include <asm/unaligned.h>
  49 #include <linux/device.h>
  50 #include <linux/leds.h>
  51 #include <linux/module.h>
  52 #include <linux/slab.h>
  53 #include <linux/usb.h>
  54 #include <linux/usb/input.h>
  55 #include <media/rc-core.h>
  56
  57 /* Driver Information */
  58 #define DRIVER_AUTHOR "Jarod Wilson <jarod@redhat.com>"
  59 #define DRIVER_AUTHOR2 "The Dweller, Stephen Cox"
  60 #define DRIVER_DESC "RedRat3 USB IR Transceiver Driver"
  61 #define DRIVER_NAME "redrat3"
  62
  63 /* bulk data transfer types */
  64 #define RR3_ERROR               0x01
  65 #define RR3_MOD_SIGNAL_IN       0x20
  66 #define RR3_MOD_SIGNAL_OUT      0x21
  67
  68 /* Get the RR firmware version */
  69 #define RR3_FW_VERSION          0xb1
  70 #define RR3_FW_VERSION_LEN      64
  71 /* Send encoded signal bulk-sent earlier*/
  72 #define RR3_TX_SEND_SIGNAL      0xb3
  73 #define RR3_SET_IR_PARAM        0xb7
  74 #define RR3_GET_IR_PARAM        0xb8
  75 /* Blink the red LED on the device */
  76 #define RR3_BLINK_LED           0xb9
  77 /* Read serial number of device */
  78 #define RR3_READ_SER_NO         0xba
  79 #define RR3_SER_NO_LEN          4
  80 /* Start capture with the RC receiver */
  81 #define RR3_RC_DET_ENABLE       0xbb
  82 /* Stop capture with the RC receiver */
  83 #define RR3_RC_DET_DISABLE      0xbc
  84 /* Return the status of RC detector capture */
  85 #define RR3_RC_DET_STATUS       0xbd
  86 /* Reset redrat */
  87 #define RR3_RESET               0xa0
  88
  89 /* Max number of lengths in the signal. */
  90 #define RR3_IR_IO_MAX_LENGTHS   0x01
  91 /* Periods to measure mod. freq. */
  92 #define RR3_IR_IO_PERIODS_MF    0x02
  93 /* Size of memory for main signal data */
  94 #define RR3_IR_IO_SIG_MEM_SIZE  0x03
  95 /* Delta value when measuring lengths */
  96 #define RR3_IR_IO_LENGTH_FUZZ   0x04
  97 /* Timeout for end of signal detection */
  98 #define RR3_IR_IO_SIG_TIMEOUT   0x05
  99 /* Minimum value for pause recognition. */
 100 #define RR3_IR_IO_MIN_PAUSE     0x06
 101
 102 /* Clock freq. of EZ-USB chip */
 103 #define RR3_CLK                 24000000
 104 /* Clock periods per timer count */
 105 #define RR3_CLK_PER_COUNT       12
 106 /* (RR3_CLK / RR3_CLK_PER_COUNT) */
 107 #define RR3_CLK_CONV_FACTOR     2000000
 108 /* USB bulk-in IR data endpoint address */
 109 #define RR3_BULK_IN_EP_ADDR     0x82
 110
 111 /* Size of the fixed-length portion of the signal */
 112 #define RR3_DRIVER_MAXLENS      128
 113 #define RR3_MAX_SIG_SIZE        512
 114 #define RR3_TIME_UNIT           50
 115 #define RR3_END_OF_SIGNAL       0x7f
 116 #define RR3_TX_TRAILER_LEN      2
 117 #define RR3_RX_MIN_TIMEOUT      5
 118 #define RR3_RX_MAX_TIMEOUT      2000
 119
 120 /* The 8051's CPUCS Register address */
 121 #define RR3_CPUCS_REG_ADDR      0x7f92
 122
 123 #define USB_RR3USB_VENDOR_ID    0x112a
 124 #define USB_RR3USB_PRODUCT_ID   0x0001
 125 #define USB_RR3IIUSB_PRODUCT_ID 0x0005
 126
 127 struct redrat3_header {
 128         __be16 length;
 129         __be16 transfer_type;
 130 } __packed;
 131
 132 /* sending and receiving irdata */
 133 struct redrat3_irdata {
 134         struct redrat3_header header;
 135         __be32 pause;
 136         __be16 mod_freq_count;
 137         __be16 num_periods;
 138         __u8 max_lengths;
 139         __u8 no_lengths;
 140         __be16 max_sig_size;
 141         __be16 sig_size;
 142         __u8 no_repeats;
 143         __be16 lens[RR3_DRIVER_MAXLENS]; /* not aligned */
 144         __u8 sigdata[RR3_MAX_SIG_SIZE];
 145 } __packed;
 146
 147 /* firmware errors */
 148 struct redrat3_error {
 149         struct redrat3_header header;
 150         __be16 fw_error;
 151 } __packed;
 152
 153 /* table of devices that work with this driver */
 154 static struct usb_device_id redrat3_dev_table[] = {
 155         /* Original version of the RedRat3 */
 156         {USB_DEVICE(USB_RR3USB_VENDOR_ID, USB_RR3USB_PRODUCT_ID)},
 157         /* Second Version/release of the RedRat3 - RetRat3-II */
 158         {USB_DEVICE(USB_RR3USB_VENDOR_ID, USB_RR3IIUSB_PRODUCT_ID)},
 159         {}                      /* Terminating entry */
 160 };
 161
 162 /* Structure to hold all of our device specific stuff */
 163 struct redrat3_dev {
 164         /* core device bits */
 165         struct rc_dev *rc;
 166         struct device *dev;
 167
 168         /* led control */
 169         struct led_classdev led;
 170         atomic_t flash;
 171         struct usb_ctrlrequest flash_control;
 172         struct urb *flash_urb;
 173         u8 flash_in_buf;
 174
 175         /* save off the usb device pointer */
 176         struct usb_device *udev;
 177
 178         /* the receive endpoint */
 179         struct usb_endpoint_descriptor *ep_in;
 180         /* the buffer to receive data */
 181         void *bulk_in_buf;
 182         /* urb used to read ir data */
 183         struct urb *read_urb;
 184
 185         /* the send endpoint */
 186         struct usb_endpoint_descriptor *ep_out;
 187
 188         /* usb dma */
 189         dma_addr_t dma_in;
 190
 191         /* rx signal timeout */
 192         u32 hw_timeout;
 193
 194         /* Is the device currently transmitting?*/
 195         bool transmitting;
 196
 197         /* store for current packet */
 198         struct redrat3_irdata irdata;
 199         u16 bytes_read;
 200
 201         u32 carrier;
 202
 203         char name[64];
 204         char phys[64];
 205 };
 206
 207 /*
 208  * redrat3_issue_async
 209  *
 210  *  Issues an async read to the ir data in port..
 211  *  sets the callback to be redrat3_handle_async
 212  */
 213 static void redrat3_issue_async(struct redrat3_dev *rr3)
 214 {
 215         int res;
 216
 217         res = usb_submit_urb(rr3->read_urb, GFP_ATOMIC);
 218         if (res)
 219                 dev_dbg(rr3->dev,
 220                         "%s: receive request FAILED! (res %d, len %d)\n",
 221                         __func__, res, rr3->read_urb->transfer_buffer_length);
 222 }
 223
 224 static void redrat3_dump_fw_error(struct redrat3_dev *rr3, int code)
 225 {
 226         if (!rr3->transmitting && (code != 0x40))
 227                 dev_info(rr3->dev, "fw error code 0x%02x: ", code);
 228
 229         switch (code) {
 230         case 0x00:
 231                 pr_cont("No Error\n");
 232                 break;
 233
 234         /* Codes 0x20 through 0x2f are IR Firmware Errors */
 235         case 0x20:
 236                 pr_cont("Initial signal pulse not long enough "
 237                         "to measure carrier frequency\n");
 238                 break;
 239         case 0x21:
 240                 pr_cont("Not enough length values allocated for signal\n");
 241                 break;
 242         case 0x22:
 243                 pr_cont("Not enough memory allocated for signal data\n");
 244                 break;
 245         case 0x23:
 246                 pr_cont("Too many signal repeats\n");
 247                 break;
 248         case 0x28:
 249                 pr_cont("Insufficient memory available for IR signal "
 250                         "data memory allocation\n");
 251                 break;
 252         case 0x29:
 253                 pr_cont("Insufficient memory available "
 254                         "for IrDa signal data memory allocation\n");
 255                 break;
 256
 257         /* Codes 0x30 through 0x3f are USB Firmware Errors */
 258         case 0x30:
 259                 pr_cont("Insufficient memory available for bulk "
 260                         "transfer structure\n");
 261                 break;
 262
 263         /*
 264          * Other error codes... These are primarily errors that can occur in
 265          * the control messages sent to the redrat
 266          */
 267         case 0x40:
 268                 if (!rr3->transmitting)
 269                         pr_cont("Signal capture has been terminated\n");
 270                 break;
 271         case 0x41:
 272                 pr_cont("Attempt to set/get and unknown signal I/O "
 273                         "algorithm parameter\n");
 274                 break;
 275         case 0x42:
 276                 pr_cont("Signal capture already started\n");
 277                 break;
 278
 279         default:
 280                 pr_cont("Unknown Error\n");
 281                 break;
 282         }
 283 }
 284
 285 static u32 redrat3_val_to_mod_freq(struct redrat3_irdata *irdata)
 286 {
 287         u32 mod_freq = 0;
 288         u16 mod_freq_count = be16_to_cpu(irdata->mod_freq_count);
 289
 290         if (mod_freq_count != 0)
 291                 mod_freq = (RR3_CLK * be16_to_cpu(irdata->num_periods)) /
 292                         (mod_freq_count * RR3_CLK_PER_COUNT);
 293
 294         return mod_freq;
 295 }
 296
 297 /* this function scales down the figures for the same result... */
 298 static u32 redrat3_len_to_us(u32 length)
 299 {
 300         u32 biglen = length * 1000;
 301         u32 divisor = (RR3_CLK_CONV_FACTOR) / 1000;
 302         u32 result = (u32) (biglen / divisor);
 303
 304         /* don't allow zero lengths to go back, breaks lirc */
 305         return result ? result : 1;
 306 }
 307
 308 /*
 309  * convert us back into redrat3 lengths
 310  *
 311  * length * 1000   length * 1000000
 312  * ------------- = ---------------- = micro
 313  * rr3clk / 1000       rr3clk
 314
 315  * 6 * 2       4 * 3        micro * rr3clk          micro * rr3clk / 1000
 316  * ----- = 4   ----- = 6    -------------- = len    ---------------------
 317  *   3           2             1000000                    1000
 318  */
 319 static u32 redrat3_us_to_len(u32 microsec)
 320 {
 321         u32 result;
 322         u32 divisor;
 323
 324         microsec = (microsec > IR_MAX_DURATION) ? IR_MAX_DURATION : microsec;
 325         divisor = (RR3_CLK_CONV_FACTOR / 1000);
 326         result = (u32)(microsec * divisor) / 1000;
 327
 328         /* don't allow zero lengths to go back, breaks lirc */
 329         return result ? result : 1;
 330 }
 331
 332 static void redrat3_process_ir_data(struct redrat3_dev *rr3)
 333 {
 334         DEFINE_IR_RAW_EVENT(rawir);
 335         struct device *dev;
 336         unsigned int i, sig_size, single_len, offset, val;
 337         u32 mod_freq;
 338
 339         if (!rr3) {
 340                 pr_err("%s called with no context!\n", __func__);
 341                 return;
 342         }
 343
 344         dev = rr3->dev;
 345
 346         mod_freq = redrat3_val_to_mod_freq(&rr3->irdata);
 347         dev_dbg(dev, "Got mod_freq of %u\n", mod_freq);
 348
 349         /* process each rr3 encoded byte into an int */
 350         sig_size = be16_to_cpu(rr3->irdata.sig_size);
 351         for (i = 0; i < sig_size; i++) {
 352                 offset = rr3->irdata.sigdata[i];
 353                 val = get_unaligned_be16(&rr3->irdata.lens[offset]);
 354                 single_len = redrat3_len_to_us(val);
 355
 356                 /* we should always get pulse/space/pulse/space samples */
 357                 if (i % 2)
 358                         rawir.pulse = false;
 359                 else
 360                         rawir.pulse = true;
 361
 362                 rawir.duration = US_TO_NS(single_len);
 363                 /* cap the value to IR_MAX_DURATION */
 364                 rawir.duration = (rawir.duration > IR_MAX_DURATION) ?
 365                                  IR_MAX_DURATION : rawir.duration;
 366
 367                 dev_dbg(dev, "storing %s with duration %d (i: %d)\n",
 368                         rawir.pulse ? "pulse" : "space", rawir.duration, i);
 369                 ir_raw_event_store_with_filter(rr3->rc, &rawir);
 370         }
 371
 372         /* add a trailing space */
 373         rawir.pulse = false;
 374         rawir.timeout = true;
 375         rawir.duration = US_TO_NS(rr3->hw_timeout);
 376         dev_dbg(dev, "storing trailing timeout with duration %d\n",
 377                                                         rawir.duration);
 378         ir_raw_event_store_with_filter(rr3->rc, &rawir);
 379
 380         dev_dbg(dev, "calling ir_raw_event_handle\n");
 381         ir_raw_event_handle(rr3->rc);
 382 }
 383
 384 /* Util fn to send rr3 cmds */
 385 static int redrat3_send_cmd(int cmd, struct redrat3_dev *rr3)
 386 {
 387         struct usb_device *udev;
 388         u8 *data;
 389         int res;
 390
 391         data = kzalloc(sizeof(u8), GFP_KERNEL);
 392         if (!data)
 393                 return -ENOMEM;
 394
 395         udev = rr3->udev;
 396         res = usb_control_msg(udev, usb_rcvctrlpipe(udev, 0), cmd,
 397                               USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_IN,
 398                               0x0000, 0x0000, data, sizeof(u8), HZ * 10);
 399
 400         if (res < 0) {
 401                 dev_err(rr3->dev, "%s: Error sending rr3 cmd res %d, data %d",
 402                         __func__, res, *data);
 403                 res = -EIO;
 404         } else
 405                 res = data[0];
 406
 407         kfree(data);
 408
 409         return res;
 410 }
 411
 412 /* Enables the long range detector and starts async receive */
 413 static int redrat3_enable_detector(struct redrat3_dev *rr3)
 414 {
 415         struct device *dev = rr3->dev;
 416         u8 ret;
 417
 418         ret = redrat3_send_cmd(RR3_RC_DET_ENABLE, rr3);
 419         if (ret != 0)
 420                 dev_dbg(dev, "%s: unexpected ret of %d\n",
 421                         __func__, ret);
 422
 423         ret = redrat3_send_cmd(RR3_RC_DET_STATUS, rr3);
 424         if (ret != 1) {
 425                 dev_err(dev, "%s: detector status: %d, should be 1\n",
 426                         __func__, ret);
 427                 return -EIO;
 428         }
 429
 430         redrat3_issue_async(rr3);
 431
 432         return 0;
 433 }
 434
 435 static inline void redrat3_delete(struct redrat3_dev *rr3,
 436                                   struct usb_device *udev)
 437 {
 438         usb_kill_urb(rr3->read_urb);
 439         usb_kill_urb(rr3->flash_urb);
 440         usb_free_urb(rr3->read_urb);
 441         usb_free_urb(rr3->flash_urb);
 442         usb_free_coherent(udev, le16_to_cpu(rr3->ep_in->wMaxPacketSize),
 443                           rr3->bulk_in_buf, rr3->dma_in);
 444
 445         kfree(rr3);
 446 }
 447
 448 static u32 redrat3_get_timeout(struct redrat3_dev *rr3)
 449 {
 450         __be32 *tmp;
 451         u32 timeout = MS_TO_US(150); /* a sane default, if things go haywire */
 452         int len, ret, pipe;
 453
 454         len = sizeof(*tmp);
 455         tmp = kzalloc(len, GFP_KERNEL);
 456         if (!tmp) {
 457                 dev_warn(rr3->dev, "Memory allocation faillure\n");
 458                 return timeout;
 459         }
 460
 461         pipe = usb_rcvctrlpipe(rr3->udev, 0);
 462         ret = usb_control_msg(rr3->udev, pipe, RR3_GET_IR_PARAM,
 463                               USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_IN,
 464                               RR3_IR_IO_SIG_TIMEOUT, 0, tmp, len, HZ * 5);
 465         if (ret != len)
 466                 dev_warn(rr3->dev, "Failed to read timeout from hardware\n");
 467         else {
 468                 timeout = redrat3_len_to_us(be32_to_cpup(tmp));
 469
 470                 dev_dbg(rr3->dev, "Got timeout of %d ms\n", timeout / 1000);
 471         }
 472
 473         kfree(tmp);
 474
 475         return timeout;
 476 }
 477
 478 static int redrat3_set_timeout(struct rc_dev *rc_dev, unsigned int timeoutns)
 479 {
 480         struct redrat3_dev *rr3 = rc_dev->priv;
 481         struct usb_device *udev = rr3->udev;
 482         struct device *dev = rr3->dev;
 483         u32 *timeout;
 484         int ret;
 485
 486         timeout = kmalloc(sizeof(*timeout), GFP_KERNEL);
 487         if (!timeout)
 488                 return -ENOMEM;
 489
 490         *timeout = cpu_to_be32(redrat3_us_to_len(timeoutns / 1000));
 491         ret = usb_control_msg(udev, usb_sndctrlpipe(udev, 0), RR3_SET_IR_PARAM,
 492                      USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_OUT,
 493                      RR3_IR_IO_SIG_TIMEOUT, 0, timeout, sizeof(*timeout),
 494                      HZ * 25);
 495         dev_dbg(dev, "set ir parm timeout %d ret 0x%02x\n",
 496                                                 be32_to_cpu(*timeout), ret);
 497
 498         if (ret == sizeof(*timeout)) {
 499                 rr3->hw_timeout = timeoutns / 1000;
 500                 ret = 0;
 501         } else if (ret >= 0)
 502                 ret = -EIO;
 503
 504         kfree(timeout);
 505
 506         return ret;
 507 }
 508
 509 static void redrat3_reset(struct redrat3_dev *rr3)
 510 {
 511         struct usb_device *udev = rr3->udev;
 512         struct device *dev = rr3->dev;
 513         int rc, rxpipe, txpipe;
 514         u8 *val;
 515         int len = sizeof(u8);
 516
 517         rxpipe = usb_rcvctrlpipe(udev, 0);
 518         txpipe = usb_sndctrlpipe(udev, 0);
 519
 520         val = kmalloc(len, GFP_KERNEL);
 521         if (!val) {
 522                 dev_err(dev, "Memory allocation failure\n");
 523                 return;
 524         }
 525
 526         *val = 0x01;
 527         rc = usb_control_msg(udev, rxpipe, RR3_RESET,
 528                              USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_IN,
 529                              RR3_CPUCS_REG_ADDR, 0, val, len, HZ * 25);
 530         dev_dbg(dev, "reset returned 0x%02x\n", rc);
 531
 532         *val = 5;
 533         rc = usb_control_msg(udev, txpipe, RR3_SET_IR_PARAM,
 534                              USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_OUT,
 535                              RR3_IR_IO_LENGTH_FUZZ, 0, val, len, HZ * 25);
 536         dev_dbg(dev, "set ir parm len fuzz %d rc 0x%02x\n", *val, rc);
 537
 538         *val = RR3_DRIVER_MAXLENS;
 539         rc = usb_control_msg(udev, txpipe, RR3_SET_IR_PARAM,
 540                              USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_OUT,
 541                              RR3_IR_IO_MAX_LENGTHS, 0, val, len, HZ * 25);
 542         dev_dbg(dev, "set ir parm max lens %d rc 0x%02x\n", *val, rc);
 543
 544         kfree(val);
 545 }
 546
 547 static void redrat3_get_firmware_rev(struct redrat3_dev *rr3)
 548 {
 549         int rc = 0;
 550         char *buffer;
 551
 552         buffer = kzalloc(sizeof(char) * (RR3_FW_VERSION_LEN + 1), GFP_KERNEL);
 553         if (!buffer) {
 554                 dev_err(rr3->dev, "Memory allocation failure\n");
 555                 return;
 556         }
 557
 558         rc = usb_control_msg(rr3->udev, usb_rcvctrlpipe(rr3->udev, 0),
 559                              RR3_FW_VERSION,
 560                              USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_IN,
 561                              0, 0, buffer, RR3_FW_VERSION_LEN, HZ * 5);
 562
 563         if (rc >= 0)
 564                 dev_info(rr3->dev, "Firmware rev: %s", buffer);
 565         else
 566                 dev_err(rr3->dev, "Problem fetching firmware ID\n");
 567
 568         kfree(buffer);
 569 }
 570
 571 static void redrat3_read_packet_start(struct redrat3_dev *rr3, unsigned len)
 572 {
 573         struct redrat3_header *header = rr3->bulk_in_buf;
 574         unsigned pktlen, pkttype;
 575
 576         /* grab the Length and type of transfer */
 577         pktlen = be16_to_cpu(header->length);
 578         pkttype = be16_to_cpu(header->transfer_type);
 579
 580         if (pktlen > sizeof(rr3->irdata)) {
 581                 dev_warn(rr3->dev, "packet length %u too large\n", pktlen);
 582                 return;
 583         }
 584
 585         switch (pkttype) {
 586         case RR3_ERROR:
 587                 if (len >= sizeof(struct redrat3_error)) {
 588                         struct redrat3_error *error = rr3->bulk_in_buf;
 589                         unsigned fw_error = be16_to_cpu(error->fw_error);
 590                         redrat3_dump_fw_error(rr3, fw_error);
 591                 }
 592                 break;
 593
 594         case RR3_MOD_SIGNAL_IN:
 595                 memcpy(&rr3->irdata, rr3->bulk_in_buf, len);
 596                 rr3->bytes_read = len;
 597                 dev_dbg(rr3->dev, "bytes_read %d, pktlen %d\n",
 598                         rr3->bytes_read, pktlen);
 599                 break;
 600
 601         default:
 602                 dev_dbg(rr3->dev, "ignoring packet with type 0x%02x, len of %d, 0x%02x\n",
 603                                                 pkttype, len, pktlen);
 604                 break;
 605         }
 606 }
 607
 608 static void redrat3_read_packet_continue(struct redrat3_dev *rr3, unsigned len)
 609 {
 610         void *irdata = &rr3->irdata;
 611
 612         if (len + rr3->bytes_read > sizeof(rr3->irdata)) {
 613                 dev_warn(rr3->dev, "too much data for packet\n");
 614                 rr3->bytes_read = 0;
 615                 return;
 616         }
 617
 618         memcpy(irdata + rr3->bytes_read, rr3->bulk_in_buf, len);
 619
 620         rr3->bytes_read += len;
 621         dev_dbg(rr3->dev, "bytes_read %d, pktlen %d\n", rr3->bytes_read,
 622                                  be16_to_cpu(rr3->irdata.header.length));
 623 }
 624
 625 /* gather IR data from incoming urb, process it when we have enough */
 626 static int redrat3_get_ir_data(struct redrat3_dev *rr3, unsigned len)
 627 {
 628         struct device *dev = rr3->dev;
 629         unsigned pkttype;
 630         int ret = 0;
 631
 632         if (rr3->bytes_read == 0 && len >= sizeof(struct redrat3_header)) {
 633                 redrat3_read_packet_start(rr3, len);
 634         } else if (rr3->bytes_read != 0) {
 635                 redrat3_read_packet_continue(rr3, len);
 636         } else if (rr3->bytes_read == 0) {
 637                 dev_err(dev, "error: no packet data read\n");
 638                 ret = -ENODATA;
 639                 goto out;
 640         }
 641
 642         if (rr3->bytes_read < be16_to_cpu(rr3->irdata.header.length) +
 643                                                 sizeof(struct redrat3_header))
 644                 /* we're still accumulating data */
 645                 return 0;
 646
 647         /* if we get here, we've got IR data to decode */
 648         pkttype = be16_to_cpu(rr3->irdata.header.transfer_type);
 649         if (pkttype == RR3_MOD_SIGNAL_IN)
 650                 redrat3_process_ir_data(rr3);
 651         else
 652                 dev_dbg(dev, "discarding non-signal data packet (type 0x%02x)\n",
 653                                                                 pkttype);
 654
 655 out:
 656         rr3->bytes_read = 0;
 657         return ret;
 658 }
 659
 660 /* callback function from USB when async USB request has completed */
 661 static void redrat3_handle_async(struct urb *urb)
 662 {
 663         struct redrat3_dev *rr3;
 664         int ret;
 665
 666         if (!urb)
 667                 return;
 668
 669         rr3 = urb->context;
 670         if (!rr3) {
 671                 pr_err("%s called with invalid context!\n", __func__);
 672                 usb_unlink_urb(urb);
 673                 return;
 674         }
 675
 676         switch (urb->status) {
 677         case 0:
 678                 ret = redrat3_get_ir_data(rr3, urb->actual_length);
 679                 if (!ret) {
 680                         /* no error, prepare to read more */
 681                         redrat3_issue_async(rr3);
 682                 }
 683                 break;
 684
 685         case -ECONNRESET:
 686         case -ENOENT:
 687         case -ESHUTDOWN:
 688                 usb_unlink_urb(urb);
 689                 return;
 690
 691         case -EPIPE:
 692         default:
 693                 dev_warn(rr3->dev, "Error: urb status = %d\n", urb->status);
 694                 rr3->bytes_read = 0;
 695                 break;
 696         }
 697 }
 698
 699 static u16 mod_freq_to_val(unsigned int mod_freq)
 700 {
 701         int mult = 6000000;
 702
 703         /* Clk used in mod. freq. generation is CLK24/4. */
 704         return 65536 - (mult / mod_freq);
 705 }
 706
 707 static int redrat3_set_tx_carrier(struct rc_dev *rcdev, u32 carrier)
 708 {
 709         struct redrat3_dev *rr3 = rcdev->priv;
 710         struct device *dev = rr3->dev;
 711
 712         dev_dbg(dev, "Setting modulation frequency to %u", carrier);
 713         if (carrier == 0)
 714                 return -EINVAL;
 715
 716         rr3->carrier = carrier;
 717
 718         return 0;
 719 }
 720
 721 static int redrat3_transmit_ir(struct rc_dev *rcdev, unsigned *txbuf,
 722                                 unsigned count)
 723 {
 724         struct redrat3_dev *rr3 = rcdev->priv;
 725         struct device *dev = rr3->dev;
 726         struct redrat3_irdata *irdata = NULL;
 727         int ret, ret_len;
 728         int lencheck, cur_sample_len, pipe;
 729         int *sample_lens = NULL;
 730         u8 curlencheck = 0;
 731         unsigned i, sendbuf_len;
 732
 733         if (rr3->transmitting) {
 734                 dev_warn(dev, "%s: transmitter already in use\n", __func__);
 735                 return -EAGAIN;
 736         }
 737
 738         if (count > RR3_MAX_SIG_SIZE - RR3_TX_TRAILER_LEN)
 739                 return -EINVAL;
 740
 741         /* rr3 will disable rc detector on transmit */
 742         rr3->transmitting = true;
 743
 744         sample_lens = kzalloc(sizeof(int) * RR3_DRIVER_MAXLENS, GFP_KERNEL);
 745         if (!sample_lens) {
 746                 ret = -ENOMEM;
 747                 goto out;
 748         }
 749
 750         irdata = kzalloc(sizeof(*irdata), GFP_KERNEL);
 751         if (!irdata) {
 752                 ret = -ENOMEM;
 753                 goto out;
 754         }
 755
 756         for (i = 0; i < count; i++) {
 757                 cur_sample_len = redrat3_us_to_len(txbuf[i]);
 758                 if (cur_sample_len > 0xffff) {
 759                         dev_warn(dev, "transmit period of %uus truncated to %uus\n",
 760                                         txbuf[i], redrat3_len_to_us(0xffff));
 761                         cur_sample_len = 0xffff;
 762                 }
 763                 for (lencheck = 0; lencheck < curlencheck; lencheck++) {
 764                         if (sample_lens[lencheck] == cur_sample_len)
 765                                 break;
 766                 }
 767                 if (lencheck == curlencheck) {
 768                         dev_dbg(dev, "txbuf[%d]=%u, pos %d, enc %u\n",
 769                                 i, txbuf[i], curlencheck, cur_sample_len);
 770                         if (curlencheck < RR3_DRIVER_MAXLENS) {
 771                                 /* now convert the value to a proper
 772                                  * rr3 value.. */
 773                                 sample_lens[curlencheck] = cur_sample_len;
 774                                 put_unaligned_be16(cur_sample_len,
 775                                                 &irdata->lens[curlencheck]);
 776                                 curlencheck++;
 777                         } else {
 778                                 ret = -EINVAL;
 779                                 goto out;
 780                         }
 781                 }
 782                 irdata->sigdata[i] = lencheck;
 783         }
 784
 785         irdata->sigdata[count] = RR3_END_OF_SIGNAL;
 786         irdata->sigdata[count + 1] = RR3_END_OF_SIGNAL;
 787
 788         sendbuf_len = offsetof(struct redrat3_irdata,
 789                                         sigdata[count + RR3_TX_TRAILER_LEN]);
 790         /* fill in our packet header */
 791         irdata->header.length = cpu_to_be16(sendbuf_len -
 792                                                 sizeof(struct redrat3_header));
 793         irdata->header.transfer_type = cpu_to_be16(RR3_MOD_SIGNAL_OUT);
 794         irdata->pause = cpu_to_be32(redrat3_len_to_us(100));
 795         irdata->mod_freq_count = cpu_to_be16(mod_freq_to_val(rr3->carrier));
 796         irdata->no_lengths = curlencheck;
 797         irdata->sig_size = cpu_to_be16(count + RR3_TX_TRAILER_LEN);
 798
 799         pipe = usb_sndbulkpipe(rr3->udev, rr3->ep_out->bEndpointAddress);
 800         ret = usb_bulk_msg(rr3->udev, pipe, irdata,
 801                             sendbuf_len, &ret_len, 10 * HZ);
 802         dev_dbg(dev, "sent %d bytes, (ret %d)\n", ret_len, ret);
 803
 804         /* now tell the hardware to transmit what we sent it */
 805         pipe = usb_rcvctrlpipe(rr3->udev, 0);
 806         ret = usb_control_msg(rr3->udev, pipe, RR3_TX_SEND_SIGNAL,
 807                               USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_IN,
 808                               0, 0, irdata, 2, HZ * 10);
 809
 810         if (ret < 0)
 811                 dev_err(dev, "Error: control msg send failed, rc %d\n", ret);
 812         else
 813                 ret = count;
 814
 815 out:
 816         kfree(sample_lens);
 817         kfree(irdata);
 818
 819         rr3->transmitting = false;
 820         /* rr3 re-enables rc detector because it was enabled before */
 821
 822         return ret;
 823 }
 824
 825 static void redrat3_brightness_set(struct led_classdev *led_dev, enum
 826                                                 led_brightness brightness)
 827 {
 828         struct redrat3_dev *rr3 = container_of(led_dev, struct redrat3_dev,
 829                                                                         led);
 830
 831         if (brightness != LED_OFF && atomic_cmpxchg(&rr3->flash, 0, 1) == 0) {
 832                 int ret = usb_submit_urb(rr3->flash_urb, GFP_ATOMIC);
 833                 if (ret != 0) {
 834                         dev_dbg(rr3->dev, "%s: unexpected ret of %d\n",
 835                                 __func__, ret);
 836                         atomic_set(&rr3->flash, 0);
 837                 }
 838         }
 839 }
 840
 841 static void redrat3_led_complete(struct urb *urb)
 842 {
 843         struct redrat3_dev *rr3 = urb->context;
 844
 845         switch (urb->status) {
 846         case 0:
 847                 break;
 848         case -ECONNRESET:
 849         case -ENOENT:
 850         case -ESHUTDOWN:
 851                 usb_unlink_urb(urb);
 852                 return;
 853         case -EPIPE:
 854         default:
 855                 dev_dbg(rr3->dev, "Error: urb status = %d\n", urb->status);
 856                 break;
 857         }
 858
 859         rr3->led.brightness = LED_OFF;
 860         atomic_dec(&rr3->flash);
 861 }
 862
 863 static struct rc_dev *redrat3_init_rc_dev(struct redrat3_dev *rr3)
 864 {
 865         struct device *dev = rr3->dev;
 866         struct rc_dev *rc;
 867         int ret = -ENODEV;
 868         u16 prod = le16_to_cpu(rr3->udev->descriptor.idProduct);
 869
 870         rc = rc_allocate_device();
 871         if (!rc) {
 872                 dev_err(dev, "remote input dev allocation failed\n");
 873                 goto out;
 874         }
 875
 876         snprintf(rr3->name, sizeof(rr3->name), "RedRat3%s "
 877                  "Infrared Remote Transceiver (%04x:%04x)",
 878                  prod == USB_RR3IIUSB_PRODUCT_ID ? "-II" : "",
 879                  le16_to_cpu(rr3->udev->descriptor.idVendor), prod);
 880
 881         usb_make_path(rr3->udev, rr3->phys, sizeof(rr3->phys));
 882
 883         rc->input_name = rr3->name;
 884         rc->input_phys = rr3->phys;
 885         usb_to_input_id(rr3->udev, &rc->input_id);
 886         rc->dev.parent = dev;
 887         rc->priv = rr3;
 888         rc->driver_type = RC_DRIVER_IR_RAW;
 889         rc->allowed_protocols = RC_BIT_ALL;
 890         rc->min_timeout = MS_TO_NS(RR3_RX_MIN_TIMEOUT);
 891         rc->max_timeout = MS_TO_NS(RR3_RX_MAX_TIMEOUT);
 892         rc->timeout = US_TO_NS(rr3->hw_timeout);
 893         rc->s_timeout = redrat3_set_timeout;
 894         rc->tx_ir = redrat3_transmit_ir;
 895         rc->s_tx_carrier = redrat3_set_tx_carrier;
 896         rc->driver_name = DRIVER_NAME;
 897         rc->rx_resolution = US_TO_NS(2);
 898         rc->map_name = RC_MAP_HAUPPAUGE;
 899
 900         ret = rc_register_device(rc);
 901         if (ret < 0) {
 902                 dev_err(dev, "remote dev registration failed\n");
 903                 goto out;
 904         }
 905
 906         return rc;
 907
 908 out:
 909         rc_free_device(rc);
 910         return NULL;
 911 }
 912
 913 static int redrat3_dev_probe(struct usb_interface *intf,
 914                              const struct usb_device_id *id)
 915 {
 916         struct usb_device *udev = interface_to_usbdev(intf);
 917         struct device *dev = &intf->dev;
 918         struct usb_host_interface *uhi;
 919         struct redrat3_dev *rr3;
 920         struct usb_endpoint_descriptor *ep;
 921         struct usb_endpoint_descriptor *ep_in = NULL;
 922         struct usb_endpoint_descriptor *ep_out = NULL;
 923         u8 addr, attrs;
 924         int pipe, i;
 925         int retval = -ENOMEM;
 926
 927         uhi = intf->cur_altsetting;
 928
 929         /* find our bulk-in and bulk-out endpoints */
 930         for (i = 0; i < uhi->desc.bNumEndpoints; ++i) {
 931                 ep = &uhi->endpoint[i].desc;
 932                 addr = ep->bEndpointAddress;
 933                 attrs = ep->bmAttributes;
 934
 935                 if ((ep_in == NULL) &&
 936                     ((addr & USB_ENDPOINT_DIR_MASK) == USB_DIR_IN) &&
 937                     ((attrs & USB_ENDPOINT_XFERTYPE_MASK) ==
 938                      USB_ENDPOINT_XFER_BULK)) {
 939                         dev_dbg(dev, "found bulk-in endpoint at 0x%02x\n",
 940                                 ep->bEndpointAddress);
 941                         /* data comes in on 0x82, 0x81 is for other data... */
 942                         if (ep->bEndpointAddress == RR3_BULK_IN_EP_ADDR)
 943                                 ep_in = ep;
 944                 }
 945
 946                 if ((ep_out == NULL) &&
 947                     ((addr & USB_ENDPOINT_DIR_MASK) == USB_DIR_OUT) &&
 948                     ((attrs & USB_ENDPOINT_XFERTYPE_MASK) ==
 949                      USB_ENDPOINT_XFER_BULK)) {
 950                         dev_dbg(dev, "found bulk-out endpoint at 0x%02x\n",
 951                                 ep->bEndpointAddress);
 952                         ep_out = ep;
 953                 }
 954         }
 955
 956         if (!ep_in || !ep_out) {
 957                 dev_err(dev, "Couldn't find both in and out endpoints\n");
 958                 retval = -ENODEV;
 959                 goto no_endpoints;
 960         }
 961
 962         /* allocate memory for our device state and initialize it */
 963         rr3 = kzalloc(sizeof(*rr3), GFP_KERNEL);
 964         if (rr3 == NULL) {
 965                 dev_err(dev, "Memory allocation failure\n");
 966                 goto no_endpoints;
 967         }
 968
 969         rr3->dev = &intf->dev;
 970
 971         /* set up bulk-in endpoint */
 972         rr3->read_urb = usb_alloc_urb(0, GFP_KERNEL);
 973         if (!rr3->read_urb) {
 974                 dev_err(dev, "Read urb allocation failure\n");
 975                 goto error;
 976         }
 977
 978         rr3->ep_in = ep_in;
 979         rr3->bulk_in_buf = usb_alloc_coherent(udev,
 980                 le16_to_cpu(ep_in->wMaxPacketSize), GFP_KERNEL, &rr3->dma_in);
 981         if (!rr3->bulk_in_buf) {
 982                 dev_err(dev, "Read buffer allocation failure\n");
 983                 goto error;
 984         }
 985
 986         pipe = usb_rcvbulkpipe(udev, ep_in->bEndpointAddress);
 987         usb_fill_bulk_urb(rr3->read_urb, udev, pipe, rr3->bulk_in_buf,
 988                 le16_to_cpu(ep_in->wMaxPacketSize), redrat3_handle_async, rr3);
 989         rr3->read_urb->transfer_dma = rr3->dma_in;
 990         rr3->read_urb->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
 991
 992         rr3->ep_out = ep_out;
 993         rr3->udev = udev;
 994
 995         redrat3_reset(rr3);
 996         redrat3_get_firmware_rev(rr3);
 997
 998         /* might be all we need to do? */
 999         retval = redrat3_enable_detector(rr3);
1000         if (retval < 0)
1001                 goto error;
1002
1003         /* store current hardware timeout, in µs */
1004         rr3->hw_timeout = redrat3_get_timeout(rr3);
1005
1006         /* default.. will get overridden by any sends with a freq defined */
1007         rr3->carrier = 38000;
1008
1009         /* led control */
1010         rr3->led.name = "redrat3:red:feedback";
1011         rr3->led.default_trigger = "rc-feedback";
1012         rr3->led.brightness_set = redrat3_brightness_set;
1013         retval = led_classdev_register(&intf->dev, &rr3->led);
1014         if (retval)
1015                 goto error;
1016
1017         atomic_set(&rr3->flash, 0);
1018         rr3->flash_urb = usb_alloc_urb(0, GFP_KERNEL);
1019         if (!rr3->flash_urb) {
1020                 retval = -ENOMEM;
1021                 goto led_free_error;
1022         }
1023
1024         /* setup packet is 'c0 b9 0000 0000 0001' */
1025         rr3->flash_control.bRequestType = 0xc0;
1026         rr3->flash_control.bRequest = RR3_BLINK_LED;
1027         rr3->flash_control.wLength = cpu_to_le16(1);
1028
1029         usb_fill_control_urb(rr3->flash_urb, udev, usb_rcvctrlpipe(udev, 0),
1030                         (unsigned char *)&rr3->flash_control,
1031                         &rr3->flash_in_buf, sizeof(rr3->flash_in_buf),
1032                         redrat3_led_complete, rr3);
1033
1034         rr3->rc = redrat3_init_rc_dev(rr3);
1035         if (!rr3->rc) {
1036                 retval = -ENOMEM;
1037                 goto led_free_error;
1038         }
1039
1040         /* we can register the device now, as it is ready */
1041         usb_set_intfdata(intf, rr3);
1042
1043         return 0;
1044
1045 led_free_error:
1046         led_classdev_unregister(&rr3->led);
1047 error:
1048         redrat3_delete(rr3, rr3->udev);
1049
1050 no_endpoints:
1051         dev_err(dev, "%s: retval = %x", __func__, retval);
1052
1053         return retval;
1054 }
1055
1056 static void redrat3_dev_disconnect(struct usb_interface *intf)
1057 {
1058         struct usb_device *udev = interface_to_usbdev(intf);
1059         struct redrat3_dev *rr3 = usb_get_intfdata(intf);
1060
1061         if (!rr3)
1062                 return;
1063
1064         usb_set_intfdata(intf, NULL);
1065         rc_unregister_device(rr3->rc);
1066         led_classdev_unregister(&rr3->led);
1067         redrat3_delete(rr3, udev);
1068 }
1069
1070 static int redrat3_dev_suspend(struct usb_interface *intf, pm_message_t message)
1071 {
1072         struct redrat3_dev *rr3 = usb_get_intfdata(intf);
1073
1074         led_classdev_suspend(&rr3->led);
1075         usb_kill_urb(rr3->read_urb);
1076         usb_kill_urb(rr3->flash_urb);
1077         return 0;
1078 }
1079
1080 static int redrat3_dev_resume(struct usb_interface *intf)
1081 {
1082         struct redrat3_dev *rr3 = usb_get_intfdata(intf);
1083
1084         if (usb_submit_urb(rr3->read_urb, GFP_ATOMIC))
1085                 return -EIO;
1086         led_classdev_resume(&rr3->led);
1087         return 0;
1088 }
1089
1090 static struct usb_driver redrat3_dev_driver = {
1091         .name           = DRIVER_NAME,
1092         .probe          = redrat3_dev_probe,
1093         .disconnect     = redrat3_dev_disconnect,
1094         .suspend        = redrat3_dev_suspend,
1095         .resume         = redrat3_dev_resume,
1096         .reset_resume   = redrat3_dev_resume,
1097         .id_table       = redrat3_dev_table
1098 };
1099
1100 module_usb_driver(redrat3_dev_driver);
1101
1102 MODULE_DESCRIPTION(DRIVER_DESC);
1103 MODULE_AUTHOR(DRIVER_AUTHOR);
1104 MODULE_AUTHOR(DRIVER_AUTHOR2);
1105 MODULE_LICENSE("GPL");
1106 MODULE_DEVICE_TABLE(usb, redrat3_dev_table);