1/*
2 * ati_remote2 - ATI/Philips USB RF remote driver
3 *
4 * Copyright (C) 2005-2008 Ville Syrjala <syrjala@sci.fi>
5 * Copyright (C) 2007-2008 Peter Stokes <linux@dadeos.co.uk>
6 *
7 * This program is free software; you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License version 2
9 * as published by the Free Software Foundation.
10 */
11
12#include <linux/usb/input.h>
13#include <linux/slab.h>
14#include <linux/module.h>
15
16#define DRIVER_DESC    "ATI/Philips USB RF remote driver"
17#define DRIVER_VERSION "0.3"
18
19MODULE_DESCRIPTION(DRIVER_DESC);
20MODULE_VERSION(DRIVER_VERSION);
21MODULE_AUTHOR("Ville Syrjala <syrjala@sci.fi>");
22MODULE_LICENSE("GPL");
23
24/*
25 * ATI Remote Wonder II Channel Configuration
26 *
27 * The remote control can by assigned one of sixteen "channels" in order to facilitate
28 * the use of multiple remote controls within range of each other.
29 * A remote's "channel" may be altered by pressing and holding the "PC" button for
30 * approximately 3 seconds, after which the button will slowly flash the count of the
31 * currently configured "channel", using the numeric keypad enter a number between 1 and
32 * 16 and then press the "PC" button again, the button will slowly flash the count of the
33 * newly configured "channel".
34 */
35
36enum {
37	ATI_REMOTE2_MAX_CHANNEL_MASK = 0xFFFF,
38	ATI_REMOTE2_MAX_MODE_MASK = 0x1F,
39};
40
41static int ati_remote2_set_mask(const char *val,
42				const struct kernel_param *kp,
43				unsigned int max)
44{
45	unsigned int mask;
46	int ret;
47
48	if (!val)
49		return -EINVAL;
50
51	ret = kstrtouint(val, 0, &mask);
52	if (ret)
53		return ret;
54
55	if (mask & ~max)
56		return -EINVAL;
57
58	*(unsigned int *)kp->arg = mask;
59
60	return 0;
61}
62
63static int ati_remote2_set_channel_mask(const char *val,
64					const struct kernel_param *kp)
65{
66	pr_debug("%s()\n", __func__);
67
68	return ati_remote2_set_mask(val, kp, ATI_REMOTE2_MAX_CHANNEL_MASK);
69}
70
71static int ati_remote2_get_channel_mask(char *buffer,
72					const struct kernel_param *kp)
73{
74	pr_debug("%s()\n", __func__);
75
76	return sprintf(buffer, "0x%04x", *(unsigned int *)kp->arg);
77}
78
79static int ati_remote2_set_mode_mask(const char *val,
80				     const struct kernel_param *kp)
81{
82	pr_debug("%s()\n", __func__);
83
84	return ati_remote2_set_mask(val, kp, ATI_REMOTE2_MAX_MODE_MASK);
85}
86
87static int ati_remote2_get_mode_mask(char *buffer,
88				     const struct kernel_param *kp)
89{
90	pr_debug("%s()\n", __func__);
91
92	return sprintf(buffer, "0x%02x", *(unsigned int *)kp->arg);
93}
94
95static unsigned int channel_mask = ATI_REMOTE2_MAX_CHANNEL_MASK;
96#define param_check_channel_mask(name, p) __param_check(name, p, unsigned int)
97static const struct kernel_param_ops param_ops_channel_mask = {
98	.set = ati_remote2_set_channel_mask,
99	.get = ati_remote2_get_channel_mask,
100};
101module_param(channel_mask, channel_mask, 0644);
102MODULE_PARM_DESC(channel_mask, "Bitmask of channels to accept <15:Channel16>...<1:Channel2><0:Channel1>");
103
104static unsigned int mode_mask = ATI_REMOTE2_MAX_MODE_MASK;
105#define param_check_mode_mask(name, p) __param_check(name, p, unsigned int)
106static const struct kernel_param_ops param_ops_mode_mask = {
107	.set = ati_remote2_set_mode_mask,
108	.get = ati_remote2_get_mode_mask,
109};
110module_param(mode_mask, mode_mask, 0644);
111MODULE_PARM_DESC(mode_mask, "Bitmask of modes to accept <4:PC><3:AUX4><2:AUX3><1:AUX2><0:AUX1>");
112
113static struct usb_device_id ati_remote2_id_table[] = {
114	{ USB_DEVICE(0x0471, 0x0602) },	/* ATI Remote Wonder II */
115	{ }
116};
117MODULE_DEVICE_TABLE(usb, ati_remote2_id_table);
118
119static DEFINE_MUTEX(ati_remote2_mutex);
120
121enum {
122	ATI_REMOTE2_OPENED = 0x1,
123	ATI_REMOTE2_SUSPENDED = 0x2,
124};
125
126enum {
127	ATI_REMOTE2_AUX1,
128	ATI_REMOTE2_AUX2,
129	ATI_REMOTE2_AUX3,
130	ATI_REMOTE2_AUX4,
131	ATI_REMOTE2_PC,
132	ATI_REMOTE2_MODES,
133};
134
135static const struct {
136	u8  hw_code;
137	u16 keycode;
138} ati_remote2_key_table[] = {
139	{ 0x00, KEY_0 },
140	{ 0x01, KEY_1 },
141	{ 0x02, KEY_2 },
142	{ 0x03, KEY_3 },
143	{ 0x04, KEY_4 },
144	{ 0x05, KEY_5 },
145	{ 0x06, KEY_6 },
146	{ 0x07, KEY_7 },
147	{ 0x08, KEY_8 },
148	{ 0x09, KEY_9 },
149	{ 0x0c, KEY_POWER },
150	{ 0x0d, KEY_MUTE },
151	{ 0x10, KEY_VOLUMEUP },
152	{ 0x11, KEY_VOLUMEDOWN },
153	{ 0x20, KEY_CHANNELUP },
154	{ 0x21, KEY_CHANNELDOWN },
155	{ 0x28, KEY_FORWARD },
156	{ 0x29, KEY_REWIND },
157	{ 0x2c, KEY_PLAY },
158	{ 0x30, KEY_PAUSE },
159	{ 0x31, KEY_STOP },
160	{ 0x37, KEY_RECORD },
161	{ 0x38, KEY_DVD },
162	{ 0x39, KEY_TV },
163	{ 0x3f, KEY_PROG1 }, /* AUX1-AUX4 and PC */
164	{ 0x54, KEY_MENU },
165	{ 0x58, KEY_UP },
166	{ 0x59, KEY_DOWN },
167	{ 0x5a, KEY_LEFT },
168	{ 0x5b, KEY_RIGHT },
169	{ 0x5c, KEY_OK },
170	{ 0x78, KEY_A },
171	{ 0x79, KEY_B },
172	{ 0x7a, KEY_C },
173	{ 0x7b, KEY_D },
174	{ 0x7c, KEY_E },
175	{ 0x7d, KEY_F },
176	{ 0x82, KEY_ENTER },
177	{ 0x8e, KEY_VENDOR },
178	{ 0x96, KEY_COFFEE },
179	{ 0xa9, BTN_LEFT },
180	{ 0xaa, BTN_RIGHT },
181	{ 0xbe, KEY_QUESTION },
182	{ 0xd0, KEY_EDIT },
183	{ 0xd5, KEY_FRONT },
184	{ 0xf9, KEY_INFO },
185};
186
187struct ati_remote2 {
188	struct input_dev *idev;
189	struct usb_device *udev;
190
191	struct usb_interface *intf[2];
192	struct usb_endpoint_descriptor *ep[2];
193	struct urb *urb[2];
194	void *buf[2];
195	dma_addr_t buf_dma[2];
196
197	unsigned long jiffies;
198	int mode;
199
200	char name[64];
201	char phys[64];
202
203	/* Each mode (AUX1-AUX4 and PC) can have an independent keymap. */
204	u16 keycode[ATI_REMOTE2_MODES][ARRAY_SIZE(ati_remote2_key_table)];
205
206	unsigned int flags;
207
208	unsigned int channel_mask;
209	unsigned int mode_mask;
210};
211
212static int ati_remote2_probe(struct usb_interface *interface, const struct usb_device_id *id);
213static void ati_remote2_disconnect(struct usb_interface *interface);
214static int ati_remote2_suspend(struct usb_interface *interface, pm_message_t message);
215static int ati_remote2_resume(struct usb_interface *interface);
216static int ati_remote2_reset_resume(struct usb_interface *interface);
217static int ati_remote2_pre_reset(struct usb_interface *interface);
218static int ati_remote2_post_reset(struct usb_interface *interface);
219
220static struct usb_driver ati_remote2_driver = {
221	.name       = "ati_remote2",
222	.probe      = ati_remote2_probe,
223	.disconnect = ati_remote2_disconnect,
224	.id_table   = ati_remote2_id_table,
225	.suspend    = ati_remote2_suspend,
226	.resume     = ati_remote2_resume,
227	.reset_resume = ati_remote2_reset_resume,
228	.pre_reset  = ati_remote2_pre_reset,
229	.post_reset = ati_remote2_post_reset,
230	.supports_autosuspend = 1,
231};
232
233static int ati_remote2_submit_urbs(struct ati_remote2 *ar2)
234{
235	int r;
236
237	r = usb_submit_urb(ar2->urb[0], GFP_KERNEL);
238	if (r) {
239		dev_err(&ar2->intf[0]->dev,
240			"%s(): usb_submit_urb() = %d\n", __func__, r);
241		return r;
242	}
243	r = usb_submit_urb(ar2->urb[1], GFP_KERNEL);
244	if (r) {
245		usb_kill_urb(ar2->urb[0]);
246		dev_err(&ar2->intf[1]->dev,
247			"%s(): usb_submit_urb() = %d\n", __func__, r);
248		return r;
249	}
250
251	return 0;
252}
253
254static void ati_remote2_kill_urbs(struct ati_remote2 *ar2)
255{
256	usb_kill_urb(ar2->urb[1]);
257	usb_kill_urb(ar2->urb[0]);
258}
259
260static int ati_remote2_open(struct input_dev *idev)
261{
262	struct ati_remote2 *ar2 = input_get_drvdata(idev);
263	int r;
264
265	dev_dbg(&ar2->intf[0]->dev, "%s()\n", __func__);
266
267	r = usb_autopm_get_interface(ar2->intf[0]);
268	if (r) {
269		dev_err(&ar2->intf[0]->dev,
270			"%s(): usb_autopm_get_interface() = %d\n", __func__, r);
271		goto fail1;
272	}
273
274	mutex_lock(&ati_remote2_mutex);
275
276	if (!(ar2->flags & ATI_REMOTE2_SUSPENDED)) {
277		r = ati_remote2_submit_urbs(ar2);
278		if (r)
279			goto fail2;
280	}
281
282	ar2->flags |= ATI_REMOTE2_OPENED;
283
284	mutex_unlock(&ati_remote2_mutex);
285
286	usb_autopm_put_interface(ar2->intf[0]);
287
288	return 0;
289
290 fail2:
291	mutex_unlock(&ati_remote2_mutex);
292	usb_autopm_put_interface(ar2->intf[0]);
293 fail1:
294	return r;
295}
296
297static void ati_remote2_close(struct input_dev *idev)
298{
299	struct ati_remote2 *ar2 = input_get_drvdata(idev);
300
301	dev_dbg(&ar2->intf[0]->dev, "%s()\n", __func__);
302
303	mutex_lock(&ati_remote2_mutex);
304
305	if (!(ar2->flags & ATI_REMOTE2_SUSPENDED))
306		ati_remote2_kill_urbs(ar2);
307
308	ar2->flags &= ~ATI_REMOTE2_OPENED;
309
310	mutex_unlock(&ati_remote2_mutex);
311}
312
313static void ati_remote2_input_mouse(struct ati_remote2 *ar2)
314{
315	struct input_dev *idev = ar2->idev;
316	u8 *data = ar2->buf[0];
317	int channel, mode;
318
319	channel = data[0] >> 4;
320
321	if (!((1 << channel) & ar2->channel_mask))
322		return;
323
324	mode = data[0] & 0x0F;
325
326	if (mode > ATI_REMOTE2_PC) {
327		dev_err(&ar2->intf[0]->dev,
328			"Unknown mode byte (%02x %02x %02x %02x)\n",
329			data[3], data[2], data[1], data[0]);
330		return;
331	}
332
333	if (!((1 << mode) & ar2->mode_mask))
334		return;
335
336	input_event(idev, EV_REL, REL_X, (s8) data[1]);
337	input_event(idev, EV_REL, REL_Y, (s8) data[2]);
338	input_sync(idev);
339}
340
341static int ati_remote2_lookup(unsigned int hw_code)
342{
343	int i;
344
345	for (i = 0; i < ARRAY_SIZE(ati_remote2_key_table); i++)
346		if (ati_remote2_key_table[i].hw_code == hw_code)
347			return i;
348
349	return -1;
350}
351
352static void ati_remote2_input_key(struct ati_remote2 *ar2)
353{
354	struct input_dev *idev = ar2->idev;
355	u8 *data = ar2->buf[1];
356	int channel, mode, hw_code, index;
357
358	channel = data[0] >> 4;
359
360	if (!((1 << channel) & ar2->channel_mask))
361		return;
362
363	mode = data[0] & 0x0F;
364
365	if (mode > ATI_REMOTE2_PC) {
366		dev_err(&ar2->intf[1]->dev,
367			"Unknown mode byte (%02x %02x %02x %02x)\n",
368			data[3], data[2], data[1], data[0]);
369		return;
370	}
371
372	hw_code = data[2];
373	if (hw_code == 0x3f) {
374		/*
375		 * For some incomprehensible reason the mouse pad generates
376		 * events which look identical to the events from the last
377		 * pressed mode key. Naturally we don't want to generate key
378		 * events for the mouse pad so we filter out any subsequent
379		 * events from the same mode key.
380		 */
381		if (ar2->mode == mode)
382			return;
383
384		if (data[1] == 0)
385			ar2->mode = mode;
386	}
387
388	if (!((1 << mode) & ar2->mode_mask))
389		return;
390
391	index = ati_remote2_lookup(hw_code);
392	if (index < 0) {
393		dev_err(&ar2->intf[1]->dev,
394			"Unknown code byte (%02x %02x %02x %02x)\n",
395			data[3], data[2], data[1], data[0]);
396		return;
397	}
398
399	switch (data[1]) {
400	case 0:	/* release */
401		break;
402	case 1:	/* press */
403		ar2->jiffies = jiffies + msecs_to_jiffies(idev->rep[REP_DELAY]);
404		break;
405	case 2:	/* repeat */
406
407		/* No repeat for mouse buttons. */
408		if (ar2->keycode[mode][index] == BTN_LEFT ||
409		    ar2->keycode[mode][index] == BTN_RIGHT)
410			return;
411
412		if (!time_after_eq(jiffies, ar2->jiffies))
413			return;
414
415		ar2->jiffies = jiffies + msecs_to_jiffies(idev->rep[REP_PERIOD]);
416		break;
417	default:
418		dev_err(&ar2->intf[1]->dev,
419			"Unknown state byte (%02x %02x %02x %02x)\n",
420			data[3], data[2], data[1], data[0]);
421		return;
422	}
423
424	input_event(idev, EV_KEY, ar2->keycode[mode][index], data[1]);
425	input_sync(idev);
426}
427
428static void ati_remote2_complete_mouse(struct urb *urb)
429{
430	struct ati_remote2 *ar2 = urb->context;
431	int r;
432
433	switch (urb->status) {
434	case 0:
435		usb_mark_last_busy(ar2->udev);
436		ati_remote2_input_mouse(ar2);
437		break;
438	case -ENOENT:
439	case -EILSEQ:
440	case -ECONNRESET:
441	case -ESHUTDOWN:
442		dev_dbg(&ar2->intf[0]->dev,
443			"%s(): urb status = %d\n", __func__, urb->status);
444		return;
445	default:
446		usb_mark_last_busy(ar2->udev);
447		dev_err(&ar2->intf[0]->dev,
448			"%s(): urb status = %d\n", __func__, urb->status);
449	}
450
451	r = usb_submit_urb(urb, GFP_ATOMIC);
452	if (r)
453		dev_err(&ar2->intf[0]->dev,
454			"%s(): usb_submit_urb() = %d\n", __func__, r);
455}
456
457static void ati_remote2_complete_key(struct urb *urb)
458{
459	struct ati_remote2 *ar2 = urb->context;
460	int r;
461
462	switch (urb->status) {
463	case 0:
464		usb_mark_last_busy(ar2->udev);
465		ati_remote2_input_key(ar2);
466		break;
467	case -ENOENT:
468	case -EILSEQ:
469	case -ECONNRESET:
470	case -ESHUTDOWN:
471		dev_dbg(&ar2->intf[1]->dev,
472			"%s(): urb status = %d\n", __func__, urb->status);
473		return;
474	default:
475		usb_mark_last_busy(ar2->udev);
476		dev_err(&ar2->intf[1]->dev,
477			"%s(): urb status = %d\n", __func__, urb->status);
478	}
479
480	r = usb_submit_urb(urb, GFP_ATOMIC);
481	if (r)
482		dev_err(&ar2->intf[1]->dev,
483			"%s(): usb_submit_urb() = %d\n", __func__, r);
484}
485
486static int ati_remote2_getkeycode(struct input_dev *idev,
487				  struct input_keymap_entry *ke)
488{
489	struct ati_remote2 *ar2 = input_get_drvdata(idev);
490	unsigned int mode;
491	int offset;
492	unsigned int index;
493	unsigned int scancode;
494
495	if (ke->flags & INPUT_KEYMAP_BY_INDEX) {
496		index = ke->index;
497		if (index >= ATI_REMOTE2_MODES *
498				ARRAY_SIZE(ati_remote2_key_table))
499			return -EINVAL;
500
501		mode = ke->index / ARRAY_SIZE(ati_remote2_key_table);
502		offset = ke->index % ARRAY_SIZE(ati_remote2_key_table);
503		scancode = (mode << 8) + ati_remote2_key_table[offset].hw_code;
504	} else {
505		if (input_scancode_to_scalar(ke, &scancode))
506			return -EINVAL;
507
508		mode = scancode >> 8;
509		if (mode > ATI_REMOTE2_PC)
510			return -EINVAL;
511
512		offset = ati_remote2_lookup(scancode & 0xff);
513		if (offset < 0)
514			return -EINVAL;
515
516		index = mode * ARRAY_SIZE(ati_remote2_key_table) + offset;
517	}
518
519	ke->keycode = ar2->keycode[mode][offset];
520	ke->len = sizeof(scancode);
521	memcpy(&ke->scancode, &scancode, sizeof(scancode));
522	ke->index = index;
523
524	return 0;
525}
526
527static int ati_remote2_setkeycode(struct input_dev *idev,
528				  const struct input_keymap_entry *ke,
529				  unsigned int *old_keycode)
530{
531	struct ati_remote2 *ar2 = input_get_drvdata(idev);
532	unsigned int mode;
533	int offset;
534	unsigned int index;
535	unsigned int scancode;
536
537	if (ke->flags & INPUT_KEYMAP_BY_INDEX) {
538		if (ke->index >= ATI_REMOTE2_MODES *
539				ARRAY_SIZE(ati_remote2_key_table))
540			return -EINVAL;
541
542		mode = ke->index / ARRAY_SIZE(ati_remote2_key_table);
543		offset = ke->index % ARRAY_SIZE(ati_remote2_key_table);
544	} else {
545		if (input_scancode_to_scalar(ke, &scancode))
546			return -EINVAL;
547
548		mode = scancode >> 8;
549		if (mode > ATI_REMOTE2_PC)
550			return -EINVAL;
551
552		offset = ati_remote2_lookup(scancode & 0xff);
553		if (offset < 0)
554			return -EINVAL;
555	}
556
557	*old_keycode = ar2->keycode[mode][offset];
558	ar2->keycode[mode][offset] = ke->keycode;
559	__set_bit(ke->keycode, idev->keybit);
560
561	for (mode = 0; mode < ATI_REMOTE2_MODES; mode++) {
562		for (index = 0; index < ARRAY_SIZE(ati_remote2_key_table); index++) {
563			if (ar2->keycode[mode][index] == *old_keycode)
564				return 0;
565		}
566	}
567
568	__clear_bit(*old_keycode, idev->keybit);
569
570	return 0;
571}
572
573static int ati_remote2_input_init(struct ati_remote2 *ar2)
574{
575	struct input_dev *idev;
576	int index, mode, retval;
577
578	idev = input_allocate_device();
579	if (!idev)
580		return -ENOMEM;
581
582	ar2->idev = idev;
583	input_set_drvdata(idev, ar2);
584
585	idev->evbit[0] = BIT_MASK(EV_KEY) | BIT_MASK(EV_REP) | BIT_MASK(EV_REL);
586	idev->keybit[BIT_WORD(BTN_MOUSE)] = BIT_MASK(BTN_LEFT) |
587		BIT_MASK(BTN_RIGHT);
588	idev->relbit[0] = BIT_MASK(REL_X) | BIT_MASK(REL_Y);
589
590	for (mode = 0; mode < ATI_REMOTE2_MODES; mode++) {
591		for (index = 0; index < ARRAY_SIZE(ati_remote2_key_table); index++) {
592			ar2->keycode[mode][index] = ati_remote2_key_table[index].keycode;
593			__set_bit(ar2->keycode[mode][index], idev->keybit);
594		}
595	}
596
597	/* AUX1-AUX4 and PC generate the same scancode. */
598	index = ati_remote2_lookup(0x3f);
599	ar2->keycode[ATI_REMOTE2_AUX1][index] = KEY_PROG1;
600	ar2->keycode[ATI_REMOTE2_AUX2][index] = KEY_PROG2;
601	ar2->keycode[ATI_REMOTE2_AUX3][index] = KEY_PROG3;
602	ar2->keycode[ATI_REMOTE2_AUX4][index] = KEY_PROG4;
603	ar2->keycode[ATI_REMOTE2_PC][index] = KEY_PC;
604	__set_bit(KEY_PROG1, idev->keybit);
605	__set_bit(KEY_PROG2, idev->keybit);
606	__set_bit(KEY_PROG3, idev->keybit);
607	__set_bit(KEY_PROG4, idev->keybit);
608	__set_bit(KEY_PC, idev->keybit);
609
610	idev->rep[REP_DELAY]  = 250;
611	idev->rep[REP_PERIOD] = 33;
612
613	idev->open = ati_remote2_open;
614	idev->close = ati_remote2_close;
615
616	idev->getkeycode = ati_remote2_getkeycode;
617	idev->setkeycode = ati_remote2_setkeycode;
618
619	idev->name = ar2->name;
620	idev->phys = ar2->phys;
621
622	usb_to_input_id(ar2->udev, &idev->id);
623	idev->dev.parent = &ar2->udev->dev;
624
625	retval = input_register_device(idev);
626	if (retval)
627		input_free_device(idev);
628
629	return retval;
630}
631
632static int ati_remote2_urb_init(struct ati_remote2 *ar2)
633{
634	struct usb_device *udev = ar2->udev;
635	int i, pipe, maxp;
636
637	for (i = 0; i < 2; i++) {
638		ar2->buf[i] = usb_alloc_coherent(udev, 4, GFP_KERNEL, &ar2->buf_dma[i]);
639		if (!ar2->buf[i])
640			return -ENOMEM;
641
642		ar2->urb[i] = usb_alloc_urb(0, GFP_KERNEL);
643		if (!ar2->urb[i])
644			return -ENOMEM;
645
646		pipe = usb_rcvintpipe(udev, ar2->ep[i]->bEndpointAddress);
647		maxp = usb_maxpacket(udev, pipe, usb_pipeout(pipe));
648		maxp = maxp > 4 ? 4 : maxp;
649
650		usb_fill_int_urb(ar2->urb[i], udev, pipe, ar2->buf[i], maxp,
651				 i ? ati_remote2_complete_key : ati_remote2_complete_mouse,
652				 ar2, ar2->ep[i]->bInterval);
653		ar2->urb[i]->transfer_dma = ar2->buf_dma[i];
654		ar2->urb[i]->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
655	}
656
657	return 0;
658}
659
660static void ati_remote2_urb_cleanup(struct ati_remote2 *ar2)
661{
662	int i;
663
664	for (i = 0; i < 2; i++) {
665		usb_free_urb(ar2->urb[i]);
666		usb_free_coherent(ar2->udev, 4, ar2->buf[i], ar2->buf_dma[i]);
667	}
668}
669
670static int ati_remote2_setup(struct ati_remote2 *ar2, unsigned int ch_mask)
671{
672	int r, i, channel;
673
674	/*
675	 * Configure receiver to only accept input from remote "channel"
676	 *  channel == 0  -> Accept input from any remote channel
677	 *  channel == 1  -> Only accept input from remote channel 1
678	 *  channel == 2  -> Only accept input from remote channel 2
679	 *  ...
680	 *  channel == 16 -> Only accept input from remote channel 16
681	 */
682
683	channel = 0;
684	for (i = 0; i < 16; i++) {
685		if ((1 << i) & ch_mask) {
686			if (!(~(1 << i) & ch_mask))
687				channel = i + 1;
688			break;
689		}
690	}
691
692	r = usb_control_msg(ar2->udev, usb_sndctrlpipe(ar2->udev, 0),
693			    0x20,
694			    USB_DIR_OUT | USB_TYPE_VENDOR | USB_RECIP_INTERFACE,
695			    channel, 0x0, NULL, 0, USB_CTRL_SET_TIMEOUT);
696	if (r) {
697		dev_err(&ar2->udev->dev, "%s - failed to set channel due to error: %d\n",
698			__func__, r);
699		return r;
700	}
701
702	return 0;
703}
704
705static ssize_t ati_remote2_show_channel_mask(struct device *dev,
706					     struct device_attribute *attr,
707					     char *buf)
708{
709	struct usb_device *udev = to_usb_device(dev);
710	struct usb_interface *intf = usb_ifnum_to_if(udev, 0);
711	struct ati_remote2 *ar2 = usb_get_intfdata(intf);
712
713	return sprintf(buf, "0x%04x\n", ar2->channel_mask);
714}
715
716static ssize_t ati_remote2_store_channel_mask(struct device *dev,
717					      struct device_attribute *attr,
718					      const char *buf, size_t count)
719{
720	struct usb_device *udev = to_usb_device(dev);
721	struct usb_interface *intf = usb_ifnum_to_if(udev, 0);
722	struct ati_remote2 *ar2 = usb_get_intfdata(intf);
723	unsigned int mask;
724	int r;
725
726	r = kstrtouint(buf, 0, &mask);
727	if (r)
728		return r;
729
730	if (mask & ~ATI_REMOTE2_MAX_CHANNEL_MASK)
731		return -EINVAL;
732
733	r = usb_autopm_get_interface(ar2->intf[0]);
734	if (r) {
735		dev_err(&ar2->intf[0]->dev,
736			"%s(): usb_autopm_get_interface() = %d\n", __func__, r);
737		return r;
738	}
739
740	mutex_lock(&ati_remote2_mutex);
741
742	if (mask != ar2->channel_mask) {
743		r = ati_remote2_setup(ar2, mask);
744		if (!r)
745			ar2->channel_mask = mask;
746	}
747
748	mutex_unlock(&ati_remote2_mutex);
749
750	usb_autopm_put_interface(ar2->intf[0]);
751
752	return r ? r : count;
753}
754
755static ssize_t ati_remote2_show_mode_mask(struct device *dev,
756					  struct device_attribute *attr,
757					  char *buf)
758{
759	struct usb_device *udev = to_usb_device(dev);
760	struct usb_interface *intf = usb_ifnum_to_if(udev, 0);
761	struct ati_remote2 *ar2 = usb_get_intfdata(intf);
762
763	return sprintf(buf, "0x%02x\n", ar2->mode_mask);
764}
765
766static ssize_t ati_remote2_store_mode_mask(struct device *dev,
767					   struct device_attribute *attr,
768					   const char *buf, size_t count)
769{
770	struct usb_device *udev = to_usb_device(dev);
771	struct usb_interface *intf = usb_ifnum_to_if(udev, 0);
772	struct ati_remote2 *ar2 = usb_get_intfdata(intf);
773	unsigned int mask;
774	int err;
775
776	err = kstrtouint(buf, 0, &mask);
777	if (err)
778		return err;
779
780	if (mask & ~ATI_REMOTE2_MAX_MODE_MASK)
781		return -EINVAL;
782
783	ar2->mode_mask = mask;
784
785	return count;
786}
787
788static DEVICE_ATTR(channel_mask, 0644, ati_remote2_show_channel_mask,
789		   ati_remote2_store_channel_mask);
790
791static DEVICE_ATTR(mode_mask, 0644, ati_remote2_show_mode_mask,
792		   ati_remote2_store_mode_mask);
793
794static struct attribute *ati_remote2_attrs[] = {
795	&dev_attr_channel_mask.attr,
796	&dev_attr_mode_mask.attr,
797	NULL,
798};
799
800static struct attribute_group ati_remote2_attr_group = {
801	.attrs = ati_remote2_attrs,
802};
803
804static int ati_remote2_probe(struct usb_interface *interface, const struct usb_device_id *id)
805{
806	struct usb_device *udev = interface_to_usbdev(interface);
807	struct usb_host_interface *alt = interface->cur_altsetting;
808	struct ati_remote2 *ar2;
809	int r;
810
811	if (alt->desc.bInterfaceNumber)
812		return -ENODEV;
813
814	ar2 = kzalloc(sizeof (struct ati_remote2), GFP_KERNEL);
815	if (!ar2)
816		return -ENOMEM;
817
818	ar2->udev = udev;
819
820	/* Sanity check, first interface must have an endpoint */
821	if (alt->desc.bNumEndpoints < 1 || !alt->endpoint) {
822		dev_err(&interface->dev,
823			"%s(): interface 0 must have an endpoint\n", __func__);
824		r = -ENODEV;
825		goto fail1;
826	}
827	ar2->intf[0] = interface;
828	ar2->ep[0] = &alt->endpoint[0].desc;
829
830	/* Sanity check, the device must have two interfaces */
831	ar2->intf[1] = usb_ifnum_to_if(udev, 1);
832	if ((udev->actconfig->desc.bNumInterfaces < 2) || !ar2->intf[1]) {
833		dev_err(&interface->dev, "%s(): need 2 interfaces, found %d\n",
834			__func__, udev->actconfig->desc.bNumInterfaces);
835		r = -ENODEV;
836		goto fail1;
837	}
838
839	r = usb_driver_claim_interface(&ati_remote2_driver, ar2->intf[1], ar2);
840	if (r)
841		goto fail1;
842
843	/* Sanity check, second interface must have an endpoint */
844	alt = ar2->intf[1]->cur_altsetting;
845	if (alt->desc.bNumEndpoints < 1 || !alt->endpoint) {
846		dev_err(&interface->dev,
847			"%s(): interface 1 must have an endpoint\n", __func__);
848		r = -ENODEV;
849		goto fail2;
850	}
851	ar2->ep[1] = &alt->endpoint[0].desc;
852
853	r = ati_remote2_urb_init(ar2);
854	if (r)
855		goto fail3;
856
857	ar2->channel_mask = channel_mask;
858	ar2->mode_mask = mode_mask;
859
860	r = ati_remote2_setup(ar2, ar2->channel_mask);
861	if (r)
862		goto fail3;
863
864	usb_make_path(udev, ar2->phys, sizeof(ar2->phys));
865	strlcat(ar2->phys, "/input0", sizeof(ar2->phys));
866
867	strlcat(ar2->name, "ATI Remote Wonder II", sizeof(ar2->name));
868
869	r = sysfs_create_group(&udev->dev.kobj, &ati_remote2_attr_group);
870	if (r)
871		goto fail3;
872
873	r = ati_remote2_input_init(ar2);
874	if (r)
875		goto fail4;
876
877	usb_set_intfdata(interface, ar2);
878
879	interface->needs_remote_wakeup = 1;
880
881	return 0;
882
883 fail4:
884	sysfs_remove_group(&udev->dev.kobj, &ati_remote2_attr_group);
885 fail3:
886	ati_remote2_urb_cleanup(ar2);
887 fail2:
888	usb_driver_release_interface(&ati_remote2_driver, ar2->intf[1]);
889 fail1:
890	kfree(ar2);
891
892	return r;
893}
894
895static void ati_remote2_disconnect(struct usb_interface *interface)
896{
897	struct ati_remote2 *ar2;
898	struct usb_host_interface *alt = interface->cur_altsetting;
899
900	if (alt->desc.bInterfaceNumber)
901		return;
902
903	ar2 = usb_get_intfdata(interface);
904	usb_set_intfdata(interface, NULL);
905
906	input_unregister_device(ar2->idev);
907
908	sysfs_remove_group(&ar2->udev->dev.kobj, &ati_remote2_attr_group);
909
910	ati_remote2_urb_cleanup(ar2);
911
912	usb_driver_release_interface(&ati_remote2_driver, ar2->intf[1]);
913
914	kfree(ar2);
915}
916
917static int ati_remote2_suspend(struct usb_interface *interface,
918			       pm_message_t message)
919{
920	struct ati_remote2 *ar2;
921	struct usb_host_interface *alt = interface->cur_altsetting;
922
923	if (alt->desc.bInterfaceNumber)
924		return 0;
925
926	ar2 = usb_get_intfdata(interface);
927
928	dev_dbg(&ar2->intf[0]->dev, "%s()\n", __func__);
929
930	mutex_lock(&ati_remote2_mutex);
931
932	if (ar2->flags & ATI_REMOTE2_OPENED)
933		ati_remote2_kill_urbs(ar2);
934
935	ar2->flags |= ATI_REMOTE2_SUSPENDED;
936
937	mutex_unlock(&ati_remote2_mutex);
938
939	return 0;
940}
941
942static int ati_remote2_resume(struct usb_interface *interface)
943{
944	struct ati_remote2 *ar2;
945	struct usb_host_interface *alt = interface->cur_altsetting;
946	int r = 0;
947
948	if (alt->desc.bInterfaceNumber)
949		return 0;
950
951	ar2 = usb_get_intfdata(interface);
952
953	dev_dbg(&ar2->intf[0]->dev, "%s()\n", __func__);
954
955	mutex_lock(&ati_remote2_mutex);
956
957	if (ar2->flags & ATI_REMOTE2_OPENED)
958		r = ati_remote2_submit_urbs(ar2);
959
960	if (!r)
961		ar2->flags &= ~ATI_REMOTE2_SUSPENDED;
962
963	mutex_unlock(&ati_remote2_mutex);
964
965	return r;
966}
967
968static int ati_remote2_reset_resume(struct usb_interface *interface)
969{
970	struct ati_remote2 *ar2;
971	struct usb_host_interface *alt = interface->cur_altsetting;
972	int r = 0;
973
974	if (alt->desc.bInterfaceNumber)
975		return 0;
976
977	ar2 = usb_get_intfdata(interface);
978
979	dev_dbg(&ar2->intf[0]->dev, "%s()\n", __func__);
980
981	mutex_lock(&ati_remote2_mutex);
982
983	r = ati_remote2_setup(ar2, ar2->channel_mask);
984	if (r)
985		goto out;
986
987	if (ar2->flags & ATI_REMOTE2_OPENED)
988		r = ati_remote2_submit_urbs(ar2);
989
990	if (!r)
991		ar2->flags &= ~ATI_REMOTE2_SUSPENDED;
992
993 out:
994	mutex_unlock(&ati_remote2_mutex);
995
996	return r;
997}
998
999static int ati_remote2_pre_reset(struct usb_interface *interface)
1000{
1001	struct ati_remote2 *ar2;
1002	struct usb_host_interface *alt = interface->cur_altsetting;
1003
1004	if (alt->desc.bInterfaceNumber)
1005		return 0;
1006
1007	ar2 = usb_get_intfdata(interface);
1008
1009	dev_dbg(&ar2->intf[0]->dev, "%s()\n", __func__);
1010
1011	mutex_lock(&ati_remote2_mutex);
1012
1013	if (ar2->flags == ATI_REMOTE2_OPENED)
1014		ati_remote2_kill_urbs(ar2);
1015
1016	return 0;
1017}
1018
1019static int ati_remote2_post_reset(struct usb_interface *interface)
1020{
1021	struct ati_remote2 *ar2;
1022	struct usb_host_interface *alt = interface->cur_altsetting;
1023	int r = 0;
1024
1025	if (alt->desc.bInterfaceNumber)
1026		return 0;
1027
1028	ar2 = usb_get_intfdata(interface);
1029
1030	dev_dbg(&ar2->intf[0]->dev, "%s()\n", __func__);
1031
1032	if (ar2->flags == ATI_REMOTE2_OPENED)
1033		r = ati_remote2_submit_urbs(ar2);
1034
1035	mutex_unlock(&ati_remote2_mutex);
1036
1037	return r;
1038}
1039
1040module_usb_driver(ati_remote2_driver);
1041