xref: /openbmc/linux/drivers/media/rc/redrat3.c (revision 293d5b43)
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);
1107