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
127struct redrat3_header {
128	__be16 length;
129	__be16 transfer_type;
130} __packed;
131
132/* sending and receiving irdata */
133struct 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 */
148struct redrat3_error {
149	struct redrat3_header header;
150	__be16 fw_error;
151} __packed;
152
153/* table of devices that work with this driver */
154static 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 */
163struct 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 timer */
192	struct timer_list rx_timeout;
193	u32 hw_timeout;
194
195	/* Is the device currently transmitting?*/
196	bool transmitting;
197
198	/* store for current packet */
199	struct redrat3_irdata irdata;
200	u16 bytes_read;
201
202	u32 carrier;
203
204	char name[64];
205	char phys[64];
206};
207
208/*
209 * redrat3_issue_async
210 *
211 *  Issues an async read to the ir data in port..
212 *  sets the callback to be redrat3_handle_async
213 */
214static void redrat3_issue_async(struct redrat3_dev *rr3)
215{
216	int res;
217
218	res = usb_submit_urb(rr3->read_urb, GFP_ATOMIC);
219	if (res)
220		dev_dbg(rr3->dev,
221			"%s: receive request FAILED! (res %d, len %d)\n",
222			__func__, res, rr3->read_urb->transfer_buffer_length);
223}
224
225static void redrat3_dump_fw_error(struct redrat3_dev *rr3, int code)
226{
227	if (!rr3->transmitting && (code != 0x40))
228		dev_info(rr3->dev, "fw error code 0x%02x: ", code);
229
230	switch (code) {
231	case 0x00:
232		pr_cont("No Error\n");
233		break;
234
235	/* Codes 0x20 through 0x2f are IR Firmware Errors */
236	case 0x20:
237		pr_cont("Initial signal pulse not long enough "
238			"to measure carrier frequency\n");
239		break;
240	case 0x21:
241		pr_cont("Not enough length values allocated for signal\n");
242		break;
243	case 0x22:
244		pr_cont("Not enough memory allocated for signal data\n");
245		break;
246	case 0x23:
247		pr_cont("Too many signal repeats\n");
248		break;
249	case 0x28:
250		pr_cont("Insufficient memory available for IR signal "
251			"data memory allocation\n");
252		break;
253	case 0x29:
254		pr_cont("Insufficient memory available "
255			"for IrDa signal data memory allocation\n");
256		break;
257
258	/* Codes 0x30 through 0x3f are USB Firmware Errors */
259	case 0x30:
260		pr_cont("Insufficient memory available for bulk "
261			"transfer structure\n");
262		break;
263
264	/*
265	 * Other error codes... These are primarily errors that can occur in
266	 * the control messages sent to the redrat
267	 */
268	case 0x40:
269		if (!rr3->transmitting)
270			pr_cont("Signal capture has been terminated\n");
271		break;
272	case 0x41:
273		pr_cont("Attempt to set/get and unknown signal I/O "
274			"algorithm parameter\n");
275		break;
276	case 0x42:
277		pr_cont("Signal capture already started\n");
278		break;
279
280	default:
281		pr_cont("Unknown Error\n");
282		break;
283	}
284}
285
286static u32 redrat3_val_to_mod_freq(struct redrat3_irdata *irdata)
287{
288	u32 mod_freq = 0;
289	u16 mod_freq_count = be16_to_cpu(irdata->mod_freq_count);
290
291	if (mod_freq_count != 0)
292		mod_freq = (RR3_CLK * be16_to_cpu(irdata->num_periods)) /
293			(mod_freq_count * RR3_CLK_PER_COUNT);
294
295	return mod_freq;
296}
297
298/* this function scales down the figures for the same result... */
299static u32 redrat3_len_to_us(u32 length)
300{
301	u32 biglen = length * 1000;
302	u32 divisor = (RR3_CLK_CONV_FACTOR) / 1000;
303	u32 result = (u32) (biglen / divisor);
304
305	/* don't allow zero lengths to go back, breaks lirc */
306	return result ? result : 1;
307}
308
309/*
310 * convert us back into redrat3 lengths
311 *
312 * length * 1000   length * 1000000
313 * ------------- = ---------------- = micro
314 * rr3clk / 1000       rr3clk
315
316 * 6 * 2       4 * 3        micro * rr3clk          micro * rr3clk / 1000
317 * ----- = 4   ----- = 6    -------------- = len    ---------------------
318 *   3           2             1000000                    1000
319 */
320static u32 redrat3_us_to_len(u32 microsec)
321{
322	u32 result;
323	u32 divisor;
324
325	microsec = (microsec > IR_MAX_DURATION) ? IR_MAX_DURATION : microsec;
326	divisor = (RR3_CLK_CONV_FACTOR / 1000);
327	result = (u32)(microsec * divisor) / 1000;
328
329	/* don't allow zero lengths to go back, breaks lirc */
330	return result ? result : 1;
331}
332
333/* timer callback to send reset event */
334static void redrat3_rx_timeout(unsigned long data)
335{
336	struct redrat3_dev *rr3 = (struct redrat3_dev *)data;
337
338	dev_dbg(rr3->dev, "calling ir_raw_event_reset\n");
339	ir_raw_event_reset(rr3->rc);
340}
341
342static void redrat3_process_ir_data(struct redrat3_dev *rr3)
343{
344	DEFINE_IR_RAW_EVENT(rawir);
345	struct device *dev;
346	unsigned i, trailer = 0;
347	unsigned sig_size, single_len, offset, val;
348	unsigned long delay;
349	u32 mod_freq;
350
351	if (!rr3) {
352		pr_err("%s called with no context!\n", __func__);
353		return;
354	}
355
356	dev = rr3->dev;
357
358	/* Make sure we reset the IR kfifo after a bit of inactivity */
359	delay = usecs_to_jiffies(rr3->hw_timeout);
360	mod_timer(&rr3->rx_timeout, jiffies + delay);
361
362	mod_freq = redrat3_val_to_mod_freq(&rr3->irdata);
363	dev_dbg(dev, "Got mod_freq of %u\n", mod_freq);
364
365	/* process each rr3 encoded byte into an int */
366	sig_size = be16_to_cpu(rr3->irdata.sig_size);
367	for (i = 0; i < sig_size; i++) {
368		offset = rr3->irdata.sigdata[i];
369		val = get_unaligned_be16(&rr3->irdata.lens[offset]);
370		single_len = redrat3_len_to_us(val);
371
372		/* we should always get pulse/space/pulse/space samples */
373		if (i % 2)
374			rawir.pulse = false;
375		else
376			rawir.pulse = true;
377
378		rawir.duration = US_TO_NS(single_len);
379		/* Save initial pulse length to fudge trailer */
380		if (i == 0)
381			trailer = rawir.duration;
382		/* cap the value to IR_MAX_DURATION */
383		rawir.duration = (rawir.duration > IR_MAX_DURATION) ?
384				 IR_MAX_DURATION : rawir.duration;
385
386		dev_dbg(dev, "storing %s with duration %d (i: %d)\n",
387			rawir.pulse ? "pulse" : "space", rawir.duration, i);
388		ir_raw_event_store_with_filter(rr3->rc, &rawir);
389	}
390
391	/* add a trailing space, if need be */
392	if (i % 2) {
393		rawir.pulse = false;
394		/* this duration is made up, and may not be ideal... */
395		if (trailer < US_TO_NS(1000))
396			rawir.duration = US_TO_NS(2800);
397		else
398			rawir.duration = trailer;
399		dev_dbg(dev, "storing trailing space with duration %d\n",
400			rawir.duration);
401		ir_raw_event_store_with_filter(rr3->rc, &rawir);
402	}
403
404	dev_dbg(dev, "calling ir_raw_event_handle\n");
405	ir_raw_event_handle(rr3->rc);
406}
407
408/* Util fn to send rr3 cmds */
409static int redrat3_send_cmd(int cmd, struct redrat3_dev *rr3)
410{
411	struct usb_device *udev;
412	u8 *data;
413	int res;
414
415	data = kzalloc(sizeof(u8), GFP_KERNEL);
416	if (!data)
417		return -ENOMEM;
418
419	udev = rr3->udev;
420	res = usb_control_msg(udev, usb_rcvctrlpipe(udev, 0), cmd,
421			      USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_IN,
422			      0x0000, 0x0000, data, sizeof(u8), HZ * 10);
423
424	if (res < 0) {
425		dev_err(rr3->dev, "%s: Error sending rr3 cmd res %d, data %d",
426			__func__, res, *data);
427		res = -EIO;
428	} else
429		res = data[0];
430
431	kfree(data);
432
433	return res;
434}
435
436/* Enables the long range detector and starts async receive */
437static int redrat3_enable_detector(struct redrat3_dev *rr3)
438{
439	struct device *dev = rr3->dev;
440	u8 ret;
441
442	ret = redrat3_send_cmd(RR3_RC_DET_ENABLE, rr3);
443	if (ret != 0)
444		dev_dbg(dev, "%s: unexpected ret of %d\n",
445			__func__, ret);
446
447	ret = redrat3_send_cmd(RR3_RC_DET_STATUS, rr3);
448	if (ret != 1) {
449		dev_err(dev, "%s: detector status: %d, should be 1\n",
450			__func__, ret);
451		return -EIO;
452	}
453
454	redrat3_issue_async(rr3);
455
456	return 0;
457}
458
459static inline void redrat3_delete(struct redrat3_dev *rr3,
460				  struct usb_device *udev)
461{
462	usb_kill_urb(rr3->read_urb);
463	usb_kill_urb(rr3->flash_urb);
464	usb_free_urb(rr3->read_urb);
465	usb_free_urb(rr3->flash_urb);
466	usb_free_coherent(udev, le16_to_cpu(rr3->ep_in->wMaxPacketSize),
467			  rr3->bulk_in_buf, rr3->dma_in);
468
469	kfree(rr3);
470}
471
472static u32 redrat3_get_timeout(struct redrat3_dev *rr3)
473{
474	__be32 *tmp;
475	u32 timeout = MS_TO_US(150); /* a sane default, if things go haywire */
476	int len, ret, pipe;
477
478	len = sizeof(*tmp);
479	tmp = kzalloc(len, GFP_KERNEL);
480	if (!tmp) {
481		dev_warn(rr3->dev, "Memory allocation faillure\n");
482		return timeout;
483	}
484
485	pipe = usb_rcvctrlpipe(rr3->udev, 0);
486	ret = usb_control_msg(rr3->udev, pipe, RR3_GET_IR_PARAM,
487			      USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_IN,
488			      RR3_IR_IO_SIG_TIMEOUT, 0, tmp, len, HZ * 5);
489	if (ret != len)
490		dev_warn(rr3->dev, "Failed to read timeout from hardware\n");
491	else {
492		timeout = redrat3_len_to_us(be32_to_cpup(tmp));
493
494		dev_dbg(rr3->dev, "Got timeout of %d ms\n", timeout / 1000);
495	}
496
497	kfree(tmp);
498
499	return timeout;
500}
501
502static void redrat3_reset(struct redrat3_dev *rr3)
503{
504	struct usb_device *udev = rr3->udev;
505	struct device *dev = rr3->dev;
506	int rc, rxpipe, txpipe;
507	u8 *val;
508	int len = sizeof(u8);
509
510	rxpipe = usb_rcvctrlpipe(udev, 0);
511	txpipe = usb_sndctrlpipe(udev, 0);
512
513	val = kmalloc(len, GFP_KERNEL);
514	if (!val) {
515		dev_err(dev, "Memory allocation failure\n");
516		return;
517	}
518
519	*val = 0x01;
520	rc = usb_control_msg(udev, rxpipe, RR3_RESET,
521			     USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_IN,
522			     RR3_CPUCS_REG_ADDR, 0, val, len, HZ * 25);
523	dev_dbg(dev, "reset returned 0x%02x\n", rc);
524
525	*val = 5;
526	rc = usb_control_msg(udev, txpipe, RR3_SET_IR_PARAM,
527			     USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_OUT,
528			     RR3_IR_IO_LENGTH_FUZZ, 0, val, len, HZ * 25);
529	dev_dbg(dev, "set ir parm len fuzz %d rc 0x%02x\n", *val, rc);
530
531	*val = RR3_DRIVER_MAXLENS;
532	rc = usb_control_msg(udev, txpipe, RR3_SET_IR_PARAM,
533			     USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_OUT,
534			     RR3_IR_IO_MAX_LENGTHS, 0, val, len, HZ * 25);
535	dev_dbg(dev, "set ir parm max lens %d rc 0x%02x\n", *val, rc);
536
537	kfree(val);
538}
539
540static void redrat3_get_firmware_rev(struct redrat3_dev *rr3)
541{
542	int rc = 0;
543	char *buffer;
544
545	buffer = kzalloc(sizeof(char) * (RR3_FW_VERSION_LEN + 1), GFP_KERNEL);
546	if (!buffer) {
547		dev_err(rr3->dev, "Memory allocation failure\n");
548		return;
549	}
550
551	rc = usb_control_msg(rr3->udev, usb_rcvctrlpipe(rr3->udev, 0),
552			     RR3_FW_VERSION,
553			     USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_IN,
554			     0, 0, buffer, RR3_FW_VERSION_LEN, HZ * 5);
555
556	if (rc >= 0)
557		dev_info(rr3->dev, "Firmware rev: %s", buffer);
558	else
559		dev_err(rr3->dev, "Problem fetching firmware ID\n");
560
561	kfree(buffer);
562}
563
564static void redrat3_read_packet_start(struct redrat3_dev *rr3, unsigned len)
565{
566	struct redrat3_header *header = rr3->bulk_in_buf;
567	unsigned pktlen, pkttype;
568
569	/* grab the Length and type of transfer */
570	pktlen = be16_to_cpu(header->length);
571	pkttype = be16_to_cpu(header->transfer_type);
572
573	if (pktlen > sizeof(rr3->irdata)) {
574		dev_warn(rr3->dev, "packet length %u too large\n", pktlen);
575		return;
576	}
577
578	switch (pkttype) {
579	case RR3_ERROR:
580		if (len >= sizeof(struct redrat3_error)) {
581			struct redrat3_error *error = rr3->bulk_in_buf;
582			unsigned fw_error = be16_to_cpu(error->fw_error);
583			redrat3_dump_fw_error(rr3, fw_error);
584		}
585		break;
586
587	case RR3_MOD_SIGNAL_IN:
588		memcpy(&rr3->irdata, rr3->bulk_in_buf, len);
589		rr3->bytes_read = len;
590		dev_dbg(rr3->dev, "bytes_read %d, pktlen %d\n",
591			rr3->bytes_read, pktlen);
592		break;
593
594	default:
595		dev_dbg(rr3->dev, "ignoring packet with type 0x%02x, len of %d, 0x%02x\n",
596						pkttype, len, pktlen);
597		break;
598	}
599}
600
601static void redrat3_read_packet_continue(struct redrat3_dev *rr3, unsigned len)
602{
603	void *irdata = &rr3->irdata;
604
605	if (len + rr3->bytes_read > sizeof(rr3->irdata)) {
606		dev_warn(rr3->dev, "too much data for packet\n");
607		rr3->bytes_read = 0;
608		return;
609	}
610
611	memcpy(irdata + rr3->bytes_read, rr3->bulk_in_buf, len);
612
613	rr3->bytes_read += len;
614	dev_dbg(rr3->dev, "bytes_read %d, pktlen %d\n", rr3->bytes_read,
615				 be16_to_cpu(rr3->irdata.header.length));
616}
617
618/* gather IR data from incoming urb, process it when we have enough */
619static int redrat3_get_ir_data(struct redrat3_dev *rr3, unsigned len)
620{
621	struct device *dev = rr3->dev;
622	unsigned pkttype;
623	int ret = 0;
624
625	if (rr3->bytes_read == 0 && len >= sizeof(struct redrat3_header)) {
626		redrat3_read_packet_start(rr3, len);
627	} else if (rr3->bytes_read != 0) {
628		redrat3_read_packet_continue(rr3, len);
629	} else if (rr3->bytes_read == 0) {
630		dev_err(dev, "error: no packet data read\n");
631		ret = -ENODATA;
632		goto out;
633	}
634
635	if (rr3->bytes_read < be16_to_cpu(rr3->irdata.header.length) +
636						sizeof(struct redrat3_header))
637		/* we're still accumulating data */
638		return 0;
639
640	/* if we get here, we've got IR data to decode */
641	pkttype = be16_to_cpu(rr3->irdata.header.transfer_type);
642	if (pkttype == RR3_MOD_SIGNAL_IN)
643		redrat3_process_ir_data(rr3);
644	else
645		dev_dbg(dev, "discarding non-signal data packet (type 0x%02x)\n",
646								pkttype);
647
648out:
649	rr3->bytes_read = 0;
650	return ret;
651}
652
653/* callback function from USB when async USB request has completed */
654static void redrat3_handle_async(struct urb *urb)
655{
656	struct redrat3_dev *rr3;
657	int ret;
658
659	if (!urb)
660		return;
661
662	rr3 = urb->context;
663	if (!rr3) {
664		pr_err("%s called with invalid context!\n", __func__);
665		usb_unlink_urb(urb);
666		return;
667	}
668
669	switch (urb->status) {
670	case 0:
671		ret = redrat3_get_ir_data(rr3, urb->actual_length);
672		if (!ret) {
673			/* no error, prepare to read more */
674			redrat3_issue_async(rr3);
675		}
676		break;
677
678	case -ECONNRESET:
679	case -ENOENT:
680	case -ESHUTDOWN:
681		usb_unlink_urb(urb);
682		return;
683
684	case -EPIPE:
685	default:
686		dev_warn(rr3->dev, "Error: urb status = %d\n", urb->status);
687		rr3->bytes_read = 0;
688		break;
689	}
690}
691
692static u16 mod_freq_to_val(unsigned int mod_freq)
693{
694	int mult = 6000000;
695
696	/* Clk used in mod. freq. generation is CLK24/4. */
697	return 65536 - (mult / mod_freq);
698}
699
700static int redrat3_set_tx_carrier(struct rc_dev *rcdev, u32 carrier)
701{
702	struct redrat3_dev *rr3 = rcdev->priv;
703	struct device *dev = rr3->dev;
704
705	dev_dbg(dev, "Setting modulation frequency to %u", carrier);
706	if (carrier == 0)
707		return -EINVAL;
708
709	rr3->carrier = carrier;
710
711	return carrier;
712}
713
714static int redrat3_transmit_ir(struct rc_dev *rcdev, unsigned *txbuf,
715				unsigned count)
716{
717	struct redrat3_dev *rr3 = rcdev->priv;
718	struct device *dev = rr3->dev;
719	struct redrat3_irdata *irdata = NULL;
720	int ret, ret_len;
721	int lencheck, cur_sample_len, pipe;
722	int *sample_lens = NULL;
723	u8 curlencheck = 0;
724	unsigned i, sendbuf_len;
725
726	if (rr3->transmitting) {
727		dev_warn(dev, "%s: transmitter already in use\n", __func__);
728		return -EAGAIN;
729	}
730
731	if (count > RR3_MAX_SIG_SIZE - RR3_TX_TRAILER_LEN)
732		return -EINVAL;
733
734	/* rr3 will disable rc detector on transmit */
735	rr3->transmitting = true;
736
737	sample_lens = kzalloc(sizeof(int) * RR3_DRIVER_MAXLENS, GFP_KERNEL);
738	if (!sample_lens) {
739		ret = -ENOMEM;
740		goto out;
741	}
742
743	irdata = kzalloc(sizeof(*irdata), GFP_KERNEL);
744	if (!irdata) {
745		ret = -ENOMEM;
746		goto out;
747	}
748
749	for (i = 0; i < count; i++) {
750		cur_sample_len = redrat3_us_to_len(txbuf[i]);
751		if (cur_sample_len > 0xffff) {
752			dev_warn(dev, "transmit period of %uus truncated to %uus\n",
753					txbuf[i], redrat3_len_to_us(0xffff));
754			cur_sample_len = 0xffff;
755		}
756		for (lencheck = 0; lencheck < curlencheck; lencheck++) {
757			if (sample_lens[lencheck] == cur_sample_len)
758				break;
759		}
760		if (lencheck == curlencheck) {
761			dev_dbg(dev, "txbuf[%d]=%u, pos %d, enc %u\n",
762				i, txbuf[i], curlencheck, cur_sample_len);
763			if (curlencheck < RR3_DRIVER_MAXLENS) {
764				/* now convert the value to a proper
765				 * rr3 value.. */
766				sample_lens[curlencheck] = cur_sample_len;
767				put_unaligned_be16(cur_sample_len,
768						&irdata->lens[curlencheck]);
769				curlencheck++;
770			} else {
771				ret = -EINVAL;
772				goto out;
773			}
774		}
775		irdata->sigdata[i] = lencheck;
776	}
777
778	irdata->sigdata[count] = RR3_END_OF_SIGNAL;
779	irdata->sigdata[count + 1] = RR3_END_OF_SIGNAL;
780
781	sendbuf_len = offsetof(struct redrat3_irdata,
782					sigdata[count + RR3_TX_TRAILER_LEN]);
783	/* fill in our packet header */
784	irdata->header.length = cpu_to_be16(sendbuf_len -
785						sizeof(struct redrat3_header));
786	irdata->header.transfer_type = cpu_to_be16(RR3_MOD_SIGNAL_OUT);
787	irdata->pause = cpu_to_be32(redrat3_len_to_us(100));
788	irdata->mod_freq_count = cpu_to_be16(mod_freq_to_val(rr3->carrier));
789	irdata->no_lengths = curlencheck;
790	irdata->sig_size = cpu_to_be16(count + RR3_TX_TRAILER_LEN);
791
792	pipe = usb_sndbulkpipe(rr3->udev, rr3->ep_out->bEndpointAddress);
793	ret = usb_bulk_msg(rr3->udev, pipe, irdata,
794			    sendbuf_len, &ret_len, 10 * HZ);
795	dev_dbg(dev, "sent %d bytes, (ret %d)\n", ret_len, ret);
796
797	/* now tell the hardware to transmit what we sent it */
798	pipe = usb_rcvctrlpipe(rr3->udev, 0);
799	ret = usb_control_msg(rr3->udev, pipe, RR3_TX_SEND_SIGNAL,
800			      USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_IN,
801			      0, 0, irdata, 2, HZ * 10);
802
803	if (ret < 0)
804		dev_err(dev, "Error: control msg send failed, rc %d\n", ret);
805	else
806		ret = count;
807
808out:
809	kfree(sample_lens);
810	kfree(irdata);
811
812	rr3->transmitting = false;
813	/* rr3 re-enables rc detector because it was enabled before */
814
815	return ret;
816}
817
818static void redrat3_brightness_set(struct led_classdev *led_dev, enum
819						led_brightness brightness)
820{
821	struct redrat3_dev *rr3 = container_of(led_dev, struct redrat3_dev,
822									led);
823
824	if (brightness != LED_OFF && atomic_cmpxchg(&rr3->flash, 0, 1) == 0) {
825		int ret = usb_submit_urb(rr3->flash_urb, GFP_ATOMIC);
826		if (ret != 0) {
827			dev_dbg(rr3->dev, "%s: unexpected ret of %d\n",
828				__func__, ret);
829			atomic_set(&rr3->flash, 0);
830		}
831	}
832}
833
834static void redrat3_led_complete(struct urb *urb)
835{
836	struct redrat3_dev *rr3 = urb->context;
837
838	switch (urb->status) {
839	case 0:
840		break;
841	case -ECONNRESET:
842	case -ENOENT:
843	case -ESHUTDOWN:
844		usb_unlink_urb(urb);
845		return;
846	case -EPIPE:
847	default:
848		dev_dbg(rr3->dev, "Error: urb status = %d\n", urb->status);
849		break;
850	}
851
852	rr3->led.brightness = LED_OFF;
853	atomic_dec(&rr3->flash);
854}
855
856static struct rc_dev *redrat3_init_rc_dev(struct redrat3_dev *rr3)
857{
858	struct device *dev = rr3->dev;
859	struct rc_dev *rc;
860	int ret = -ENODEV;
861	u16 prod = le16_to_cpu(rr3->udev->descriptor.idProduct);
862
863	rc = rc_allocate_device();
864	if (!rc) {
865		dev_err(dev, "remote input dev allocation failed\n");
866		goto out;
867	}
868
869	snprintf(rr3->name, sizeof(rr3->name), "RedRat3%s "
870		 "Infrared Remote Transceiver (%04x:%04x)",
871		 prod == USB_RR3IIUSB_PRODUCT_ID ? "-II" : "",
872		 le16_to_cpu(rr3->udev->descriptor.idVendor), prod);
873
874	usb_make_path(rr3->udev, rr3->phys, sizeof(rr3->phys));
875
876	rc->input_name = rr3->name;
877	rc->input_phys = rr3->phys;
878	usb_to_input_id(rr3->udev, &rc->input_id);
879	rc->dev.parent = dev;
880	rc->priv = rr3;
881	rc->driver_type = RC_DRIVER_IR_RAW;
882	rc->allowed_protocols = RC_BIT_ALL;
883	rc->timeout = US_TO_NS(2750);
884	rc->tx_ir = redrat3_transmit_ir;
885	rc->s_tx_carrier = redrat3_set_tx_carrier;
886	rc->driver_name = DRIVER_NAME;
887	rc->rx_resolution = US_TO_NS(2);
888	rc->map_name = RC_MAP_HAUPPAUGE;
889
890	ret = rc_register_device(rc);
891	if (ret < 0) {
892		dev_err(dev, "remote dev registration failed\n");
893		goto out;
894	}
895
896	return rc;
897
898out:
899	rc_free_device(rc);
900	return NULL;
901}
902
903static int redrat3_dev_probe(struct usb_interface *intf,
904			     const struct usb_device_id *id)
905{
906	struct usb_device *udev = interface_to_usbdev(intf);
907	struct device *dev = &intf->dev;
908	struct usb_host_interface *uhi;
909	struct redrat3_dev *rr3;
910	struct usb_endpoint_descriptor *ep;
911	struct usb_endpoint_descriptor *ep_in = NULL;
912	struct usb_endpoint_descriptor *ep_out = NULL;
913	u8 addr, attrs;
914	int pipe, i;
915	int retval = -ENOMEM;
916
917	uhi = intf->cur_altsetting;
918
919	/* find our bulk-in and bulk-out endpoints */
920	for (i = 0; i < uhi->desc.bNumEndpoints; ++i) {
921		ep = &uhi->endpoint[i].desc;
922		addr = ep->bEndpointAddress;
923		attrs = ep->bmAttributes;
924
925		if ((ep_in == NULL) &&
926		    ((addr & USB_ENDPOINT_DIR_MASK) == USB_DIR_IN) &&
927		    ((attrs & USB_ENDPOINT_XFERTYPE_MASK) ==
928		     USB_ENDPOINT_XFER_BULK)) {
929			dev_dbg(dev, "found bulk-in endpoint at 0x%02x\n",
930				ep->bEndpointAddress);
931			/* data comes in on 0x82, 0x81 is for other data... */
932			if (ep->bEndpointAddress == RR3_BULK_IN_EP_ADDR)
933				ep_in = ep;
934		}
935
936		if ((ep_out == NULL) &&
937		    ((addr & USB_ENDPOINT_DIR_MASK) == USB_DIR_OUT) &&
938		    ((attrs & USB_ENDPOINT_XFERTYPE_MASK) ==
939		     USB_ENDPOINT_XFER_BULK)) {
940			dev_dbg(dev, "found bulk-out endpoint at 0x%02x\n",
941				ep->bEndpointAddress);
942			ep_out = ep;
943		}
944	}
945
946	if (!ep_in || !ep_out) {
947		dev_err(dev, "Couldn't find both in and out endpoints\n");
948		retval = -ENODEV;
949		goto no_endpoints;
950	}
951
952	/* allocate memory for our device state and initialize it */
953	rr3 = kzalloc(sizeof(*rr3), GFP_KERNEL);
954	if (rr3 == NULL) {
955		dev_err(dev, "Memory allocation failure\n");
956		goto no_endpoints;
957	}
958
959	rr3->dev = &intf->dev;
960
961	/* set up bulk-in endpoint */
962	rr3->read_urb = usb_alloc_urb(0, GFP_KERNEL);
963	if (!rr3->read_urb) {
964		dev_err(dev, "Read urb allocation failure\n");
965		goto error;
966	}
967
968	rr3->ep_in = ep_in;
969	rr3->bulk_in_buf = usb_alloc_coherent(udev,
970		le16_to_cpu(ep_in->wMaxPacketSize), GFP_KERNEL, &rr3->dma_in);
971	if (!rr3->bulk_in_buf) {
972		dev_err(dev, "Read buffer allocation failure\n");
973		goto error;
974	}
975
976	pipe = usb_rcvbulkpipe(udev, ep_in->bEndpointAddress);
977	usb_fill_bulk_urb(rr3->read_urb, udev, pipe, rr3->bulk_in_buf,
978		le16_to_cpu(ep_in->wMaxPacketSize), redrat3_handle_async, rr3);
979	rr3->read_urb->transfer_dma = rr3->dma_in;
980	rr3->read_urb->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
981
982	rr3->ep_out = ep_out;
983	rr3->udev = udev;
984
985	redrat3_reset(rr3);
986	redrat3_get_firmware_rev(rr3);
987
988	/* might be all we need to do? */
989	retval = redrat3_enable_detector(rr3);
990	if (retval < 0)
991		goto error;
992
993	/* store current hardware timeout, in us, will use for kfifo resets */
994	rr3->hw_timeout = redrat3_get_timeout(rr3);
995
996	/* default.. will get overridden by any sends with a freq defined */
997	rr3->carrier = 38000;
998
999	/* led control */
1000	rr3->led.name = "redrat3:red:feedback";
1001	rr3->led.default_trigger = "rc-feedback";
1002	rr3->led.brightness_set = redrat3_brightness_set;
1003	retval = led_classdev_register(&intf->dev, &rr3->led);
1004	if (retval)
1005		goto error;
1006
1007	atomic_set(&rr3->flash, 0);
1008	rr3->flash_urb = usb_alloc_urb(0, GFP_KERNEL);
1009	if (!rr3->flash_urb) {
1010		retval = -ENOMEM;
1011		goto led_free_error;
1012	}
1013
1014	/* setup packet is 'c0 b9 0000 0000 0001' */
1015	rr3->flash_control.bRequestType = 0xc0;
1016	rr3->flash_control.bRequest = RR3_BLINK_LED;
1017	rr3->flash_control.wLength = cpu_to_le16(1);
1018
1019	usb_fill_control_urb(rr3->flash_urb, udev, usb_rcvctrlpipe(udev, 0),
1020			(unsigned char *)&rr3->flash_control,
1021			&rr3->flash_in_buf, sizeof(rr3->flash_in_buf),
1022			redrat3_led_complete, rr3);
1023
1024	rr3->rc = redrat3_init_rc_dev(rr3);
1025	if (!rr3->rc) {
1026		retval = -ENOMEM;
1027		goto led_free_error;
1028	}
1029	setup_timer(&rr3->rx_timeout, redrat3_rx_timeout, (unsigned long)rr3);
1030
1031	/* we can register the device now, as it is ready */
1032	usb_set_intfdata(intf, rr3);
1033
1034	return 0;
1035
1036led_free_error:
1037	led_classdev_unregister(&rr3->led);
1038error:
1039	redrat3_delete(rr3, rr3->udev);
1040
1041no_endpoints:
1042	dev_err(dev, "%s: retval = %x", __func__, retval);
1043
1044	return retval;
1045}
1046
1047static void redrat3_dev_disconnect(struct usb_interface *intf)
1048{
1049	struct usb_device *udev = interface_to_usbdev(intf);
1050	struct redrat3_dev *rr3 = usb_get_intfdata(intf);
1051
1052	if (!rr3)
1053		return;
1054
1055	usb_set_intfdata(intf, NULL);
1056	rc_unregister_device(rr3->rc);
1057	led_classdev_unregister(&rr3->led);
1058	del_timer_sync(&rr3->rx_timeout);
1059	redrat3_delete(rr3, udev);
1060}
1061
1062static int redrat3_dev_suspend(struct usb_interface *intf, pm_message_t message)
1063{
1064	struct redrat3_dev *rr3 = usb_get_intfdata(intf);
1065
1066	led_classdev_suspend(&rr3->led);
1067	usb_kill_urb(rr3->read_urb);
1068	usb_kill_urb(rr3->flash_urb);
1069	return 0;
1070}
1071
1072static int redrat3_dev_resume(struct usb_interface *intf)
1073{
1074	struct redrat3_dev *rr3 = usb_get_intfdata(intf);
1075
1076	if (usb_submit_urb(rr3->read_urb, GFP_ATOMIC))
1077		return -EIO;
1078	led_classdev_resume(&rr3->led);
1079	return 0;
1080}
1081
1082static struct usb_driver redrat3_dev_driver = {
1083	.name		= DRIVER_NAME,
1084	.probe		= redrat3_dev_probe,
1085	.disconnect	= redrat3_dev_disconnect,
1086	.suspend	= redrat3_dev_suspend,
1087	.resume		= redrat3_dev_resume,
1088	.reset_resume	= redrat3_dev_resume,
1089	.id_table	= redrat3_dev_table
1090};
1091
1092module_usb_driver(redrat3_dev_driver);
1093
1094MODULE_DESCRIPTION(DRIVER_DESC);
1095MODULE_AUTHOR(DRIVER_AUTHOR);
1096MODULE_AUTHOR(DRIVER_AUTHOR2);
1097MODULE_LICENSE("GPL");
1098MODULE_DEVICE_TABLE(usb, redrat3_dev_table);
1099