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