1 /* 2 * TechnoTrend USB IR Receiver 3 * 4 * Copyright (C) 2012 Sean Young <sean@mess.org> 5 * 6 * This program is free software; you can redistribute it and/or modify 7 * it under the terms of the GNU General Public License as published by 8 * the Free Software Foundation; either version 2 of the License, or 9 * (at your option) any later version. 10 * 11 * This program is distributed in the hope that it will be useful, 12 * but WITHOUT ANY WARRANTY; without even the implied warranty of 13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14 * GNU General Public License for more details. 15 * 16 * You should have received a copy of the GNU General Public License 17 * along with this program; if not, write to the Free Software 18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 19 */ 20 21 #include <linux/module.h> 22 #include <linux/usb.h> 23 #include <linux/usb/input.h> 24 #include <linux/slab.h> 25 #include <linux/leds.h> 26 #include <media/rc-core.h> 27 28 #define DRIVER_NAME "ttusbir" 29 #define DRIVER_DESC "TechnoTrend USB IR Receiver" 30 /* 31 * The Windows driver uses 8 URBS, the original lirc drivers has a 32 * configurable amount (2 default, 4 max). This device generates about 125 33 * messages per second (!), whether IR is idle or not. 34 */ 35 #define NUM_URBS 4 36 #define NS_PER_BYTE 62500 37 #define NS_PER_BIT (NS_PER_BYTE/8) 38 39 struct ttusbir { 40 struct rc_dev *rc; 41 struct device *dev; 42 struct usb_device *udev; 43 44 struct urb *urb[NUM_URBS]; 45 46 struct led_classdev led; 47 struct urb *bulk_urb; 48 uint8_t bulk_buffer[5]; 49 int bulk_out_endp, iso_in_endp; 50 bool led_on, is_led_on; 51 atomic_t led_complete; 52 53 char phys[64]; 54 }; 55 56 static enum led_brightness ttusbir_brightness_get(struct led_classdev *led_dev) 57 { 58 struct ttusbir *tt = container_of(led_dev, struct ttusbir, led); 59 60 return tt->led_on ? LED_FULL : LED_OFF; 61 } 62 63 static void ttusbir_set_led(struct ttusbir *tt) 64 { 65 int ret; 66 67 smp_mb(); 68 69 if (tt->led_on != tt->is_led_on && tt->udev && 70 atomic_add_unless(&tt->led_complete, 1, 1)) { 71 tt->bulk_buffer[4] = tt->is_led_on = tt->led_on; 72 ret = usb_submit_urb(tt->bulk_urb, GFP_ATOMIC); 73 if (ret) { 74 dev_warn(tt->dev, "failed to submit bulk urb: %d\n", 75 ret); 76 atomic_dec(&tt->led_complete); 77 } 78 } 79 } 80 81 static void ttusbir_brightness_set(struct led_classdev *led_dev, enum 82 led_brightness brightness) 83 { 84 struct ttusbir *tt = container_of(led_dev, struct ttusbir, led); 85 86 tt->led_on = brightness != LED_OFF; 87 88 ttusbir_set_led(tt); 89 } 90 91 /* 92 * The urb cannot be reused until the urb completes 93 */ 94 static void ttusbir_bulk_complete(struct urb *urb) 95 { 96 struct ttusbir *tt = urb->context; 97 98 atomic_dec(&tt->led_complete); 99 100 switch (urb->status) { 101 case 0: 102 break; 103 case -ECONNRESET: 104 case -ENOENT: 105 case -ESHUTDOWN: 106 usb_unlink_urb(urb); 107 return; 108 case -EPIPE: 109 default: 110 dev_dbg(tt->dev, "Error: urb status = %d\n", urb->status); 111 break; 112 } 113 114 ttusbir_set_led(tt); 115 } 116 117 /* 118 * The data is one bit per sample, a set bit signifying silence and samples 119 * being MSB first. Bit 0 can contain garbage so take it to be whatever 120 * bit 1 is, so we don't have unexpected edges. 121 */ 122 static void ttusbir_process_ir_data(struct ttusbir *tt, uint8_t *buf) 123 { 124 struct ir_raw_event rawir; 125 unsigned i, v, b; 126 bool event = false; 127 128 init_ir_raw_event(&rawir); 129 130 for (i = 0; i < 128; i++) { 131 v = buf[i] & 0xfe; 132 switch (v) { 133 case 0xfe: 134 rawir.pulse = false; 135 rawir.duration = NS_PER_BYTE; 136 if (ir_raw_event_store_with_filter(tt->rc, &rawir)) 137 event = true; 138 break; 139 case 0: 140 rawir.pulse = true; 141 rawir.duration = NS_PER_BYTE; 142 if (ir_raw_event_store_with_filter(tt->rc, &rawir)) 143 event = true; 144 break; 145 default: 146 /* one edge per byte */ 147 if (v & 2) { 148 b = ffz(v | 1); 149 rawir.pulse = true; 150 } else { 151 b = ffs(v) - 1; 152 rawir.pulse = false; 153 } 154 155 rawir.duration = NS_PER_BIT * (8 - b); 156 if (ir_raw_event_store_with_filter(tt->rc, &rawir)) 157 event = true; 158 159 rawir.pulse = !rawir.pulse; 160 rawir.duration = NS_PER_BIT * b; 161 if (ir_raw_event_store_with_filter(tt->rc, &rawir)) 162 event = true; 163 break; 164 } 165 } 166 167 /* don't wakeup when there's nothing to do */ 168 if (event) 169 ir_raw_event_handle(tt->rc); 170 } 171 172 static void ttusbir_urb_complete(struct urb *urb) 173 { 174 struct ttusbir *tt = urb->context; 175 int rc; 176 177 switch (urb->status) { 178 case 0: 179 ttusbir_process_ir_data(tt, urb->transfer_buffer); 180 break; 181 case -ECONNRESET: 182 case -ENOENT: 183 case -ESHUTDOWN: 184 usb_unlink_urb(urb); 185 return; 186 case -EPIPE: 187 default: 188 dev_dbg(tt->dev, "Error: urb status = %d\n", urb->status); 189 break; 190 } 191 192 rc = usb_submit_urb(urb, GFP_ATOMIC); 193 if (rc && rc != -ENODEV) 194 dev_warn(tt->dev, "failed to resubmit urb: %d\n", rc); 195 } 196 197 static int ttusbir_probe(struct usb_interface *intf, 198 const struct usb_device_id *id) 199 { 200 struct ttusbir *tt; 201 struct usb_interface_descriptor *idesc; 202 struct usb_endpoint_descriptor *desc; 203 struct rc_dev *rc; 204 int i, j, ret; 205 int altsetting = -1; 206 207 tt = kzalloc(sizeof(*tt), GFP_KERNEL); 208 rc = rc_allocate_device(); 209 if (!tt || !rc) { 210 ret = -ENOMEM; 211 goto out; 212 } 213 214 /* find the correct alt setting */ 215 for (i = 0; i < intf->num_altsetting && altsetting == -1; i++) { 216 int bulk_out_endp = -1, iso_in_endp = -1; 217 218 idesc = &intf->altsetting[i].desc; 219 220 for (j = 0; j < idesc->bNumEndpoints; j++) { 221 desc = &intf->altsetting[i].endpoint[j].desc; 222 if (usb_endpoint_dir_in(desc) && 223 usb_endpoint_xfer_isoc(desc) && 224 desc->wMaxPacketSize == 0x10) 225 iso_in_endp = j; 226 else if (usb_endpoint_dir_out(desc) && 227 usb_endpoint_xfer_bulk(desc) && 228 desc->wMaxPacketSize == 0x20) 229 bulk_out_endp = j; 230 231 if (bulk_out_endp != -1 && iso_in_endp != -1) { 232 tt->bulk_out_endp = bulk_out_endp; 233 tt->iso_in_endp = iso_in_endp; 234 altsetting = i; 235 break; 236 } 237 } 238 } 239 240 if (altsetting == -1) { 241 dev_err(&intf->dev, "cannot find expected altsetting\n"); 242 ret = -ENODEV; 243 goto out; 244 } 245 246 tt->dev = &intf->dev; 247 tt->udev = interface_to_usbdev(intf); 248 tt->rc = rc; 249 250 ret = usb_set_interface(tt->udev, 0, altsetting); 251 if (ret) 252 goto out; 253 254 for (i = 0; i < NUM_URBS; i++) { 255 struct urb *urb = usb_alloc_urb(8, GFP_KERNEL); 256 void *buffer; 257 258 if (!urb) { 259 ret = -ENOMEM; 260 goto out; 261 } 262 263 urb->dev = tt->udev; 264 urb->context = tt; 265 urb->pipe = usb_rcvisocpipe(tt->udev, tt->iso_in_endp); 266 urb->interval = 1; 267 buffer = usb_alloc_coherent(tt->udev, 128, GFP_KERNEL, 268 &urb->transfer_dma); 269 if (!buffer) { 270 usb_free_urb(urb); 271 ret = -ENOMEM; 272 goto out; 273 } 274 urb->transfer_flags = URB_NO_TRANSFER_DMA_MAP | URB_ISO_ASAP; 275 urb->transfer_buffer = buffer; 276 urb->complete = ttusbir_urb_complete; 277 urb->number_of_packets = 8; 278 urb->transfer_buffer_length = 128; 279 280 for (j = 0; j < 8; j++) { 281 urb->iso_frame_desc[j].offset = j * 16; 282 urb->iso_frame_desc[j].length = 16; 283 } 284 285 tt->urb[i] = urb; 286 } 287 288 tt->bulk_urb = usb_alloc_urb(0, GFP_KERNEL); 289 if (!tt->bulk_urb) { 290 ret = -ENOMEM; 291 goto out; 292 } 293 294 tt->bulk_buffer[0] = 0xaa; 295 tt->bulk_buffer[1] = 0x01; 296 tt->bulk_buffer[2] = 0x05; 297 tt->bulk_buffer[3] = 0x01; 298 299 usb_fill_bulk_urb(tt->bulk_urb, tt->udev, usb_sndbulkpipe(tt->udev, 300 tt->bulk_out_endp), tt->bulk_buffer, sizeof(tt->bulk_buffer), 301 ttusbir_bulk_complete, tt); 302 303 tt->led.name = "ttusbir:green:power"; 304 tt->led.brightness_set = ttusbir_brightness_set; 305 tt->led.brightness_get = ttusbir_brightness_get; 306 tt->is_led_on = tt->led_on = true; 307 atomic_set(&tt->led_complete, 0); 308 ret = led_classdev_register(&intf->dev, &tt->led); 309 if (ret) 310 goto out; 311 312 usb_make_path(tt->udev, tt->phys, sizeof(tt->phys)); 313 314 rc->input_name = DRIVER_DESC; 315 rc->input_phys = tt->phys; 316 usb_to_input_id(tt->udev, &rc->input_id); 317 rc->dev.parent = &intf->dev; 318 rc->driver_type = RC_DRIVER_IR_RAW; 319 rc->allowed_protos = RC_BIT_ALL; 320 rc->priv = tt; 321 rc->driver_name = DRIVER_NAME; 322 rc->map_name = RC_MAP_TT_1500; 323 rc->timeout = MS_TO_NS(100); 324 /* 325 * The precision is NS_PER_BIT, but since every 8th bit can be 326 * overwritten with garbage the accuracy is at best 2 * NS_PER_BIT. 327 */ 328 rc->rx_resolution = NS_PER_BIT; 329 330 ret = rc_register_device(rc); 331 if (ret) { 332 dev_err(&intf->dev, "failed to register rc device %d\n", ret); 333 goto out2; 334 } 335 336 usb_set_intfdata(intf, tt); 337 338 for (i = 0; i < NUM_URBS; i++) { 339 ret = usb_submit_urb(tt->urb[i], GFP_KERNEL); 340 if (ret) { 341 dev_err(tt->dev, "failed to submit urb %d\n", ret); 342 goto out3; 343 } 344 } 345 346 return 0; 347 out3: 348 rc_unregister_device(rc); 349 out2: 350 led_classdev_unregister(&tt->led); 351 out: 352 if (tt) { 353 for (i = 0; i < NUM_URBS && tt->urb[i]; i++) { 354 struct urb *urb = tt->urb[i]; 355 356 usb_kill_urb(urb); 357 usb_free_coherent(tt->udev, 128, urb->transfer_buffer, 358 urb->transfer_dma); 359 usb_free_urb(urb); 360 } 361 usb_kill_urb(tt->bulk_urb); 362 usb_free_urb(tt->bulk_urb); 363 kfree(tt); 364 } 365 rc_free_device(rc); 366 367 return ret; 368 } 369 370 static void ttusbir_disconnect(struct usb_interface *intf) 371 { 372 struct ttusbir *tt = usb_get_intfdata(intf); 373 struct usb_device *udev = tt->udev; 374 int i; 375 376 tt->udev = NULL; 377 378 rc_unregister_device(tt->rc); 379 led_classdev_unregister(&tt->led); 380 for (i = 0; i < NUM_URBS; i++) { 381 usb_kill_urb(tt->urb[i]); 382 usb_free_coherent(udev, 128, tt->urb[i]->transfer_buffer, 383 tt->urb[i]->transfer_dma); 384 usb_free_urb(tt->urb[i]); 385 } 386 usb_kill_urb(tt->bulk_urb); 387 usb_free_urb(tt->bulk_urb); 388 usb_set_intfdata(intf, NULL); 389 kfree(tt); 390 } 391 392 static int ttusbir_suspend(struct usb_interface *intf, pm_message_t message) 393 { 394 struct ttusbir *tt = usb_get_intfdata(intf); 395 int i; 396 397 for (i = 0; i < NUM_URBS; i++) 398 usb_kill_urb(tt->urb[i]); 399 400 led_classdev_suspend(&tt->led); 401 usb_kill_urb(tt->bulk_urb); 402 403 return 0; 404 } 405 406 static int ttusbir_resume(struct usb_interface *intf) 407 { 408 struct ttusbir *tt = usb_get_intfdata(intf); 409 int i, rc; 410 411 led_classdev_resume(&tt->led); 412 tt->is_led_on = true; 413 ttusbir_set_led(tt); 414 415 for (i = 0; i < NUM_URBS; i++) { 416 rc = usb_submit_urb(tt->urb[i], GFP_KERNEL); 417 if (rc) { 418 dev_warn(tt->dev, "failed to submit urb: %d\n", rc); 419 break; 420 } 421 } 422 423 return rc; 424 } 425 426 static const struct usb_device_id ttusbir_table[] = { 427 { USB_DEVICE(0x0b48, 0x2003) }, 428 { } 429 }; 430 431 static struct usb_driver ttusbir_driver = { 432 .name = DRIVER_NAME, 433 .id_table = ttusbir_table, 434 .probe = ttusbir_probe, 435 .suspend = ttusbir_suspend, 436 .resume = ttusbir_resume, 437 .reset_resume = ttusbir_resume, 438 .disconnect = ttusbir_disconnect, 439 }; 440 441 module_usb_driver(ttusbir_driver); 442 443 MODULE_DESCRIPTION(DRIVER_DESC); 444 MODULE_AUTHOR("Sean Young <sean@mess.org>"); 445 MODULE_LICENSE("GPL"); 446 MODULE_DEVICE_TABLE(usb, ttusbir_table); 447 448