Commit | Line | Data |
---|---|---|
0938069f SY |
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 | ||
3404cb5c | 69 | if (tt->led_on != tt->is_led_on && tt->udev && |
0938069f SY |
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); | |
3404cb5c | 73 | if (ret) { |
0938069f SY |
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 | { | |
b83bfd1b | 124 | struct ir_raw_event rawir; |
0938069f | 125 | unsigned i, v, b; |
b83bfd1b | 126 | bool event = false; |
0938069f SY |
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; | |
b83bfd1b SY |
136 | if (ir_raw_event_store_with_filter(tt->rc, &rawir)) |
137 | event = true; | |
0938069f SY |
138 | break; |
139 | case 0: | |
140 | rawir.pulse = true; | |
141 | rawir.duration = NS_PER_BYTE; | |
b83bfd1b SY |
142 | if (ir_raw_event_store_with_filter(tt->rc, &rawir)) |
143 | event = true; | |
0938069f SY |
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); | |
b83bfd1b SY |
156 | if (ir_raw_event_store_with_filter(tt->rc, &rawir)) |
157 | event = true; | |
0938069f SY |
158 | |
159 | rawir.pulse = !rawir.pulse; | |
160 | rawir.duration = NS_PER_BIT * b; | |
b83bfd1b SY |
161 | if (ir_raw_event_store_with_filter(tt->rc, &rawir)) |
162 | event = true; | |
0938069f SY |
163 | break; |
164 | } | |
165 | } | |
166 | ||
b83bfd1b SY |
167 | /* don't wakeup when there's nothing to do */ |
168 | if (event) | |
169 | ir_raw_event_handle(tt->rc); | |
0938069f SY |
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 | ||
4c62e976 GKH |
197 | static int ttusbir_probe(struct usb_interface *intf, |
198 | const struct usb_device_id *id) | |
0938069f SY |
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 | ||
3404cb5c | 303 | tt->led.name = "ttusbir:green:power"; |
0938069f SY |
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; | |
c003ab1b | 319 | rc->allowed_protos = RC_BIT_ALL; |
0938069f SY |
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 | ||
4c62e976 | 370 | static void ttusbir_disconnect(struct usb_interface *intf) |
0938069f SY |
371 | { |
372 | struct ttusbir *tt = usb_get_intfdata(intf); | |
3404cb5c | 373 | struct usb_device *udev = tt->udev; |
0938069f SY |
374 | int i; |
375 | ||
3404cb5c SY |
376 | tt->udev = NULL; |
377 | ||
0938069f SY |
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]); | |
3404cb5c | 382 | usb_free_coherent(udev, 128, tt->urb[i]->transfer_buffer, |
0938069f SY |
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 | ||
3404cb5c SY |
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 | ||
3404cb5c | 411 | tt->is_led_on = true; |
d2008b56 | 412 | led_classdev_resume(&tt->led); |
3404cb5c SY |
413 | |
414 | for (i = 0; i < NUM_URBS; i++) { | |
415 | rc = usb_submit_urb(tt->urb[i], GFP_KERNEL); | |
416 | if (rc) { | |
417 | dev_warn(tt->dev, "failed to submit urb: %d\n", rc); | |
418 | break; | |
419 | } | |
420 | } | |
421 | ||
422 | return rc; | |
423 | } | |
424 | ||
0938069f SY |
425 | static const struct usb_device_id ttusbir_table[] = { |
426 | { USB_DEVICE(0x0b48, 0x2003) }, | |
427 | { } | |
428 | }; | |
429 | ||
430 | static struct usb_driver ttusbir_driver = { | |
431 | .name = DRIVER_NAME, | |
432 | .id_table = ttusbir_table, | |
433 | .probe = ttusbir_probe, | |
3404cb5c SY |
434 | .suspend = ttusbir_suspend, |
435 | .resume = ttusbir_resume, | |
436 | .reset_resume = ttusbir_resume, | |
4c62e976 | 437 | .disconnect = ttusbir_disconnect, |
0938069f SY |
438 | }; |
439 | ||
440 | module_usb_driver(ttusbir_driver); | |
441 | ||
442 | MODULE_DESCRIPTION(DRIVER_DESC); | |
443 | MODULE_AUTHOR("Sean Young <sean@mess.org>"); | |
444 | MODULE_LICENSE("GPL"); | |
445 | MODULE_DEVICE_TABLE(usb, ttusbir_table); | |
446 |