drivers/media/rc/ir_toy.c

   1 // SPDX-License-Identifier: GPL-2.0-or-later
   2
   3 /*
   4  * Infrared Toy and IR Droid RC core driver
   5  *
   6  * Copyright (C) 2020 Sean Young <sean@mess.org>
   7
   8  * This driver is based on the lirc driver which can be found here:
   9  * https://sourceforge.net/p/lirc/git/ci/master/tree/plugins/irtoy.c
  10  * Copyright (C) 2011 Peter Kooiman <pkooiman@gmail.com>
  11  */
  12
  13 #include <asm/unaligned.h>
  14 #include <linux/completion.h>
  15 #include <linux/kernel.h>
  16 #include <linux/module.h>
  17 #include <linux/usb.h>
  18 #include <linux/slab.h>
  19 #include <linux/usb/input.h>
  20
  21 #include <media/rc-core.h>
  22
  23 static const u8 COMMAND_VERSION[] = { 'v' };
  24 // End transmit and repeat reset command so we exit sump mode
  25 static const u8 COMMAND_RESET[] = { 0xff, 0xff, 0, 0, 0, 0, 0 };
  26 static const u8 COMMAND_SMODE_ENTER[] = { 's' };
  27 static const u8 COMMAND_TXSTART[] = { 0x26, 0x24, 0x25, 0x03 };
  28
  29 #define REPLY_XMITCOUNT 't'
  30 #define REPLY_XMITSUCCESS 'C'
  31 #define REPLY_VERSION 'V'
  32 #define REPLY_SAMPLEMODEPROTO 'S'
  33
  34 #define TIMEOUT 500
  35
  36 #define LEN_XMITRES 3
  37 #define LEN_VERSION 4
  38 #define LEN_SAMPLEMODEPROTO 3
  39
  40 #define MIN_FW_VERSION 20
  41 #define UNIT_NS 21333
  42 #define MAX_TIMEOUT_NS (UNIT_NS * U16_MAX)
  43
  44 #define MAX_PACKET 64
  45
  46 enum state {
  47         STATE_IRDATA,
  48         STATE_RESET,
  49         STATE_COMMAND,
  50         STATE_TX,
  51 };
  52
  53 struct irtoy {
  54         struct device *dev;
  55         struct usb_device *usbdev;
  56
  57         struct rc_dev *rc;
  58         struct urb *urb_in, *urb_out;
  59
  60         u8 *in;
  61         u8 *out;
  62         struct completion command_done;
  63
  64         bool pulse;
  65         enum state state;
  66
  67         void *tx_buf;
  68         uint tx_len;
  69
  70         uint emitted;
  71         uint hw_version;
  72         uint sw_version;
  73         uint proto_version;
  74
  75         char phys[64];
  76 };
  77
  78 static void irtoy_response(struct irtoy *irtoy, u32 len)
  79 {
  80         switch (irtoy->state) {
  81         case STATE_COMMAND:
  82                 if (len == LEN_VERSION && irtoy->in[0] == REPLY_VERSION) {
  83                         uint version;
  84
  85                         irtoy->in[LEN_VERSION] = 0;
  86
  87                         if (kstrtouint(irtoy->in + 1, 10, &version)) {
  88                                 dev_err(irtoy->dev, "invalid version %*phN. Please make sure you are using firmware v20 or higher",
  89                                         LEN_VERSION, irtoy->in);
  90                                 break;
  91                         }
  92
  93                         dev_dbg(irtoy->dev, "version %s\n", irtoy->in);
  94
  95                         irtoy->hw_version = version / 100;
  96                         irtoy->sw_version = version % 100;
  97
  98                         irtoy->state = STATE_IRDATA;
  99                         complete(&irtoy->command_done);
 100                 } else if (len == LEN_SAMPLEMODEPROTO &&
 101                            irtoy->in[0] == REPLY_SAMPLEMODEPROTO) {
 102                         uint version;
 103
 104                         irtoy->in[LEN_SAMPLEMODEPROTO] = 0;
 105
 106                         if (kstrtouint(irtoy->in + 1, 10, &version)) {
 107                                 dev_err(irtoy->dev, "invalid sample mode response %*phN",
 108                                         LEN_SAMPLEMODEPROTO, irtoy->in);
 109                                 return;
 110                         }
 111
 112                         dev_dbg(irtoy->dev, "protocol %s\n", irtoy->in);
 113
 114                         irtoy->proto_version = version;
 115
 116                         irtoy->state = STATE_IRDATA;
 117                         complete(&irtoy->command_done);
 118                 } else {
 119                         dev_err(irtoy->dev, "unexpected response to command: %*phN\n",
 120                                 len, irtoy->in);
 121                 }
 122                 break;
 123         case STATE_IRDATA: {
 124                 struct ir_raw_event rawir = { .pulse = irtoy->pulse };
 125                 __be16 *in = (__be16 *)irtoy->in;
 126                 int i;
 127
 128                 for (i = 0; i < len / sizeof(__be16); i++) {
 129                         u16 v = be16_to_cpu(in[i]);
 130
 131                         if (v == 0xffff) {
 132                                 rawir.pulse = false;
 133                         } else {
 134                                 rawir.duration = v * UNIT_NS;
 135                                 ir_raw_event_store_with_timeout(irtoy->rc,
 136                                                                 &rawir);
 137                         }
 138
 139                         rawir.pulse = !rawir.pulse;
 140                 }
 141
 142                 irtoy->pulse = rawir.pulse;
 143
 144                 ir_raw_event_handle(irtoy->rc);
 145                 break;
 146         }
 147         case STATE_TX:
 148                 if (irtoy->tx_len == 0) {
 149                         if (len == LEN_XMITRES &&
 150                             irtoy->in[0] == REPLY_XMITCOUNT) {
 151                                 u16 emitted = get_unaligned_be16(irtoy->in + 1);
 152
 153                                 dev_dbg(irtoy->dev, "emitted:%u\n", emitted);
 154
 155                                 irtoy->emitted = emitted;
 156                         } else if (len == 1 &&
 157                                    irtoy->in[0] == REPLY_XMITSUCCESS) {
 158                                 irtoy->state = STATE_IRDATA;
 159                                 complete(&irtoy->command_done);
 160                         }
 161                 } else {
 162                         // send next part of tx buffer
 163                         uint space = irtoy->in[0];
 164                         uint buf_len;
 165                         int err;
 166
 167                         if (len != 1 || space > MAX_PACKET || space == 0) {
 168                                 dev_err(irtoy->dev, "packet length expected: %*phN\n",
 169                                         len, irtoy->in);
 170                                 irtoy->state = STATE_IRDATA;
 171                                 complete(&irtoy->command_done);
 172                                 break;
 173                         }
 174
 175                         buf_len = min(space, irtoy->tx_len);
 176
 177                         dev_dbg(irtoy->dev, "remaining:%u sending:%u\n",
 178                                 irtoy->tx_len, buf_len);
 179
 180                         memcpy(irtoy->out, irtoy->tx_buf, buf_len);
 181                         irtoy->urb_out->transfer_buffer_length = buf_len;
 182                         err = usb_submit_urb(irtoy->urb_out, GFP_ATOMIC);
 183                         if (err != 0) {
 184                                 dev_err(irtoy->dev, "fail to submit tx buf urb: %d\n",
 185                                         err);
 186                                 irtoy->state = STATE_IRDATA;
 187                                 complete(&irtoy->command_done);
 188                                 break;
 189                         }
 190
 191                         irtoy->tx_buf += buf_len;
 192                         irtoy->tx_len -= buf_len;
 193                 }
 194                 break;
 195         case STATE_RESET:
 196                 dev_err(irtoy->dev, "unexpected response to reset: %*phN\n",
 197                         len, irtoy->in);
 198         }
 199 }
 200
 201 static void irtoy_out_callback(struct urb *urb)
 202 {
 203         struct irtoy *irtoy = urb->context;
 204
 205         if (urb->status == 0) {
 206                 if (irtoy->state == STATE_RESET)
 207                         complete(&irtoy->command_done);
 208         } else {
 209                 dev_warn(irtoy->dev, "out urb status: %d\n", urb->status);
 210         }
 211 }
 212
 213 static void irtoy_in_callback(struct urb *urb)
 214 {
 215         struct irtoy *irtoy = urb->context;
 216         int ret;
 217
 218         if (urb->status == 0)
 219                 irtoy_response(irtoy, urb->actual_length);
 220         else
 221                 dev_dbg(irtoy->dev, "in urb status: %d\n", urb->status);
 222
 223         ret = usb_submit_urb(urb, GFP_ATOMIC);
 224         if (ret && ret != -ENODEV)
 225                 dev_warn(irtoy->dev, "failed to resubmit urb: %d\n", ret);
 226 }
 227
 228 static int irtoy_command(struct irtoy *irtoy, const u8 *cmd, int cmd_len,
 229                          enum state state)
 230 {
 231         int err;
 232
 233         init_completion(&irtoy->command_done);
 234
 235         irtoy->state = state;
 236
 237         memcpy(irtoy->out, cmd, cmd_len);
 238         irtoy->urb_out->transfer_buffer_length = cmd_len;
 239
 240         err = usb_submit_urb(irtoy->urb_out, GFP_KERNEL);
 241         if (err != 0)
 242                 return err;
 243
 244         if (!wait_for_completion_timeout(&irtoy->command_done,
 245                                          msecs_to_jiffies(TIMEOUT))) {
 246                 usb_kill_urb(irtoy->urb_out);
 247                 return -ETIMEDOUT;
 248         }
 249
 250         return 0;
 251 }
 252
 253 static int irtoy_setup(struct irtoy *irtoy)
 254 {
 255         int err;
 256
 257         err = irtoy_command(irtoy, COMMAND_RESET, sizeof(COMMAND_RESET),
 258                             STATE_RESET);
 259         if (err != 0) {
 260                 dev_err(irtoy->dev, "could not write reset command: %d\n",
 261                         err);
 262                 return err;
 263         }
 264
 265         usleep_range(50, 50);
 266
 267         // get version
 268         err = irtoy_command(irtoy, COMMAND_VERSION, sizeof(COMMAND_VERSION),
 269                             STATE_COMMAND);
 270         if (err) {
 271                 dev_err(irtoy->dev, "could not write version command: %d\n",
 272                         err);
 273                 return err;
 274         }
 275
 276         // enter sample mode
 277         err = irtoy_command(irtoy, COMMAND_SMODE_ENTER,
 278                             sizeof(COMMAND_SMODE_ENTER), STATE_COMMAND);
 279         if (err)
 280                 dev_err(irtoy->dev, "could not write sample command: %d\n",
 281                         err);
 282
 283         return err;
 284 }
 285
 286 /*
 287  * When sending IR, it is imperative that we send the IR data as quickly
 288  * as possible to the device, so it does not run out of IR data and
 289  * introduce gaps. Allocate the buffer here, and then feed the data from
 290  * the urb callback handler.
 291  */
 292 static int irtoy_tx(struct rc_dev *rc, uint *txbuf, uint count)
 293 {
 294         struct irtoy *irtoy = rc->priv;
 295         unsigned int i, size;
 296         __be16 *buf;
 297         int err;
 298
 299         size = sizeof(u16) * (count + 1);
 300         buf = kmalloc(size, GFP_KERNEL);
 301         if (!buf)
 302                 return -ENOMEM;
 303
 304         for (i = 0; i < count; i++) {
 305                 u16 v = DIV_ROUND_CLOSEST(US_TO_NS(txbuf[i]), UNIT_NS);
 306
 307                 if (!v)
 308                         v = 1;
 309                 buf[i] = cpu_to_be16(v);
 310         }
 311
 312         buf[count] = cpu_to_be16(0xffff);
 313
 314         irtoy->tx_buf = buf;
 315         irtoy->tx_len = size;
 316         irtoy->emitted = 0;
 317
 318         err = irtoy_command(irtoy, COMMAND_TXSTART, sizeof(COMMAND_TXSTART),
 319                             STATE_TX);
 320         kfree(buf);
 321
 322         if (err) {
 323                 dev_err(irtoy->dev, "failed to send tx start command: %d\n",
 324                         err);
 325                 // not sure what state the device is in, reset it
 326                 irtoy_setup(irtoy);
 327                 return err;
 328         }
 329
 330         if (size != irtoy->emitted) {
 331                 dev_err(irtoy->dev, "expected %u emitted, got %u\n", size,
 332                         irtoy->emitted);
 333                 // not sure what state the device is in, reset it
 334                 irtoy_setup(irtoy);
 335                 return -EINVAL;
 336         }
 337
 338         return count;
 339 }
 340
 341 static int irtoy_probe(struct usb_interface *intf,
 342                        const struct usb_device_id *id)
 343 {
 344         struct usb_host_interface *idesc = intf->cur_altsetting;
 345         struct usb_device *usbdev = interface_to_usbdev(intf);
 346         struct usb_endpoint_descriptor *ep_in = NULL;
 347         struct usb_endpoint_descriptor *ep_out = NULL;
 348         struct usb_endpoint_descriptor *ep = NULL;
 349         struct irtoy *irtoy;
 350         struct rc_dev *rc;
 351         struct urb *urb;
 352         int i, pipe, err = -ENOMEM;
 353
 354         for (i = 0; i < idesc->desc.bNumEndpoints; i++) {
 355                 ep = &idesc->endpoint[i].desc;
 356
 357                 if (!ep_in && usb_endpoint_is_bulk_in(ep) &&
 358                     usb_endpoint_maxp(ep) == MAX_PACKET)
 359                         ep_in = ep;
 360
 361                 if (!ep_out && usb_endpoint_is_bulk_out(ep) &&
 362                     usb_endpoint_maxp(ep) == MAX_PACKET)
 363                         ep_out = ep;
 364         }
 365
 366         if (!ep_in || !ep_out) {
 367                 dev_err(&intf->dev, "required endpoints not found\n");
 368                 return -ENODEV;
 369         }
 370
 371         irtoy = kzalloc(sizeof(*irtoy), GFP_KERNEL);
 372         if (!irtoy)
 373                 return -ENOMEM;
 374
 375         irtoy->in = kmalloc(MAX_PACKET,  GFP_KERNEL);
 376         if (!irtoy->in)
 377                 goto free_irtoy;
 378
 379         irtoy->out = kmalloc(MAX_PACKET,  GFP_KERNEL);
 380         if (!irtoy->out)
 381                 goto free_irtoy;
 382
 383         rc = rc_allocate_device(RC_DRIVER_IR_RAW);
 384         if (!rc)
 385                 goto free_irtoy;
 386
 387         urb = usb_alloc_urb(0, GFP_KERNEL);
 388         if (!urb)
 389                 goto free_rcdev;
 390
 391         pipe = usb_rcvbulkpipe(usbdev, ep_in->bEndpointAddress);
 392         usb_fill_bulk_urb(urb, usbdev, pipe, irtoy->in, MAX_PACKET,
 393                           irtoy_in_callback, irtoy);
 394         irtoy->urb_in = urb;
 395
 396         urb = usb_alloc_urb(0, GFP_KERNEL);
 397         if (!urb)
 398                 goto free_rcdev;
 399
 400         pipe = usb_sndbulkpipe(usbdev, ep_out->bEndpointAddress);
 401         usb_fill_bulk_urb(urb, usbdev, pipe, irtoy->out, MAX_PACKET,
 402                           irtoy_out_callback, irtoy);
 403
 404         irtoy->dev = &intf->dev;
 405         irtoy->usbdev = usbdev;
 406         irtoy->rc = rc;
 407         irtoy->urb_out = urb;
 408         irtoy->pulse = true;
 409
 410         err = usb_submit_urb(irtoy->urb_in, GFP_KERNEL);
 411         if (err != 0) {
 412                 dev_err(irtoy->dev, "fail to submit in urb: %d\n", err);
 413                 return err;
 414         }
 415
 416         err = irtoy_setup(irtoy);
 417         if (err)
 418                 goto free_rcdev;
 419
 420         dev_info(irtoy->dev, "version: hardware %u, firmware %u, protocol %u",
 421                  irtoy->hw_version, irtoy->sw_version, irtoy->proto_version);
 422
 423         if (irtoy->sw_version < MIN_FW_VERSION) {
 424                 dev_err(irtoy->dev, "need firmware V%02u or higher",
 425                         MIN_FW_VERSION);
 426                 err = -ENODEV;
 427                 goto free_rcdev;
 428         }
 429
 430         usb_make_path(usbdev, irtoy->phys, sizeof(irtoy->phys));
 431
 432         rc->device_name = "Infrared Toy";
 433         rc->driver_name = KBUILD_MODNAME;
 434         rc->input_phys = irtoy->phys;
 435         usb_to_input_id(usbdev, &rc->input_id);
 436         rc->dev.parent = &intf->dev;
 437         rc->priv = irtoy;
 438         rc->tx_ir = irtoy_tx;
 439         rc->allowed_protocols = RC_PROTO_BIT_ALL_IR_DECODER;
 440         rc->map_name = RC_MAP_RC6_MCE;
 441         rc->rx_resolution = UNIT_NS;
 442         rc->timeout = IR_DEFAULT_TIMEOUT;
 443
 444         /*
 445          * end of transmission is detected by absence of a usb packet
 446          * with more pulse/spaces. However, each usb packet sent can
 447          * contain 32 pulse/spaces, which can be quite lengthy, so there
 448          * can be a delay between usb packets. For example with nec there is a
 449          * 17ms gap between packets.
 450          *
 451          * So, make timeout a largish minimum which works with most protocols.
 452          */
 453         rc->min_timeout = MS_TO_NS(40);
 454         rc->max_timeout = MAX_TIMEOUT_NS;
 455
 456         err = rc_register_device(rc);
 457         if (err)
 458                 goto free_rcdev;
 459
 460         usb_set_intfdata(intf, irtoy);
 461
 462         return 0;
 463
 464 free_rcdev:
 465         usb_kill_urb(irtoy->urb_out);
 466         usb_free_urb(irtoy->urb_out);
 467         usb_kill_urb(irtoy->urb_in);
 468         usb_free_urb(irtoy->urb_in);
 469         rc_free_device(rc);
 470 free_irtoy:
 471         kfree(irtoy->in);
 472         kfree(irtoy->out);
 473         kfree(irtoy);
 474         return err;
 475 }
 476
 477 static void irtoy_disconnect(struct usb_interface *intf)
 478 {
 479         struct irtoy *ir = usb_get_intfdata(intf);
 480
 481         rc_unregister_device(ir->rc);
 482         usb_set_intfdata(intf, NULL);
 483         usb_kill_urb(ir->urb_out);
 484         usb_free_urb(ir->urb_out);
 485         usb_kill_urb(ir->urb_in);
 486         usb_free_urb(ir->urb_in);
 487         kfree(ir->in);
 488         kfree(ir->out);
 489         kfree(ir);
 490 }
 491
 492 static const struct usb_device_id irtoy_table[] = {
 493         { USB_DEVICE_INTERFACE_CLASS(0x04d8, 0xfd08, USB_CLASS_CDC_DATA) },
 494         { }
 495 };
 496
 497 static struct usb_driver irtoy_driver = {
 498         .name = KBUILD_MODNAME,
 499         .probe = irtoy_probe,
 500         .disconnect = irtoy_disconnect,
 501         .id_table = irtoy_table,
 502 };
 503
 504 module_usb_driver(irtoy_driver);
 505
 506 MODULE_AUTHOR("Sean Young <sean@mess.org>");
 507 MODULE_DESCRIPTION("Infrared Toy and IR Droid driver");
 508 MODULE_LICENSE("GPL");
 509 MODULE_DEVICE_TABLE(usb, irtoy_table);