Input: xpad - add support for wireless xbox360 controllers
[safe/jmp/linux-2.6] / drivers / input / joystick / xpad.c
1 /*
2  * X-Box gamepad driver
3  *
4  * Copyright (c) 2002 Marko Friedemann <mfr@bmx-chemnitz.de>
5  *               2004 Oliver Schwartz <Oliver.Schwartz@gmx.de>,
6  *                    Steven Toth <steve@toth.demon.co.uk>,
7  *                    Franz Lehner <franz@caos.at>,
8  *                    Ivan Hawkes <blackhawk@ivanhawkes.com>
9  *               2005 Dominic Cerquetti <binary1230@yahoo.com>
10  *               2006 Adam Buchbinder <adam.buchbinder@gmail.com>
11  *               2007 Jan Kratochvil <honza@jikos.cz>
12  *
13  * This program is free software; you can redistribute it and/or
14  * modify it under the terms of the GNU General Public License as
15  * published by the Free Software Foundation; either version 2 of
16  * the License, or (at your option) any later version.
17  *
18  * This program is distributed in the hope that it will be useful,
19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
21  * GNU General Public License for more details.
22  *
23  * You should have received a copy of the GNU General Public License
24  * along with this program; if not, write to the Free Software
25  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
26  *
27  *
28  * This driver is based on:
29  *  - information from     http://euc.jp/periphs/xbox-controller.ja.html
30  *  - the iForce driver    drivers/char/joystick/iforce.c
31  *  - the skeleton-driver  drivers/usb/usb-skeleton.c
32  *  - Xbox 360 information http://www.free60.org/wiki/Gamepad
33  *
34  * Thanks to:
35  *  - ITO Takayuki for providing essential xpad information on his website
36  *  - Vojtech Pavlik     - iforce driver / input subsystem
37  *  - Greg Kroah-Hartman - usb-skeleton driver
38  *  - XBOX Linux project - extra USB id's
39  *
40  * TODO:
41  *  - fine tune axes (especially trigger axes)
42  *  - fix "analog" buttons (reported as digital now)
43  *  - get rumble working
44  *  - need USB IDs for other dance pads
45  *
46  * History:
47  *
48  * 2002-06-27 - 0.0.1 : first version, just said "XBOX HID controller"
49  *
50  * 2002-07-02 - 0.0.2 : basic working version
51  *  - all axes and 9 of the 10 buttons work (german InterAct device)
52  *  - the black button does not work
53  *
54  * 2002-07-14 - 0.0.3 : rework by Vojtech Pavlik
55  *  - indentation fixes
56  *  - usb + input init sequence fixes
57  *
58  * 2002-07-16 - 0.0.4 : minor changes, merge with Vojtech's v0.0.3
59  *  - verified the lack of HID and report descriptors
60  *  - verified that ALL buttons WORK
61  *  - fixed d-pad to axes mapping
62  *
63  * 2002-07-17 - 0.0.5 : simplified d-pad handling
64  *
65  * 2004-10-02 - 0.0.6 : DDR pad support
66  *  - borrowed from the XBOX linux kernel
67  *  - USB id's for commonly used dance pads are present
68  *  - dance pads will map D-PAD to buttons, not axes
69  *  - pass the module paramater 'dpad_to_buttons' to force
70  *    the D-PAD to map to buttons if your pad is not detected
71  *
72  * Later changes can be tracked in SCM.
73  */
74
75 #include <linux/kernel.h>
76 #include <linux/init.h>
77 #include <linux/slab.h>
78 #include <linux/stat.h>
79 #include <linux/module.h>
80 #include <linux/usb/input.h>
81
82 #define DRIVER_AUTHOR "Marko Friedemann <mfr@bmx-chemnitz.de>"
83 #define DRIVER_DESC "X-Box pad driver"
84
85 #define XPAD_PKT_LEN 32
86
87 /* xbox d-pads should map to buttons, as is required for DDR pads
88    but we map them to axes when possible to simplify things */
89 #define MAP_DPAD_TO_BUTTONS    0
90 #define MAP_DPAD_TO_AXES       1
91 #define MAP_DPAD_UNKNOWN       2
92
93 #define XTYPE_XBOX        0
94 #define XTYPE_XBOX360     1
95 #define XTYPE_XBOX360W    2
96 #define XTYPE_UNKNOWN     3
97
98 static int dpad_to_buttons;
99 module_param(dpad_to_buttons, bool, S_IRUGO);
100 MODULE_PARM_DESC(dpad_to_buttons, "Map D-PAD to buttons rather than axes for unknown pads");
101
102 static const struct xpad_device {
103         u16 idVendor;
104         u16 idProduct;
105         char *name;
106         u8 dpad_mapping;
107         u8 xtype;
108 } xpad_device[] = {
109         { 0x045e, 0x0202, "Microsoft X-Box pad v1 (US)", MAP_DPAD_TO_AXES, XTYPE_XBOX },
110         { 0x045e, 0x0289, "Microsoft X-Box pad v2 (US)", MAP_DPAD_TO_AXES, XTYPE_XBOX },
111         { 0x045e, 0x0285, "Microsoft X-Box pad (Japan)", MAP_DPAD_TO_AXES, XTYPE_XBOX },
112         { 0x045e, 0x0287, "Microsoft Xbox Controller S", MAP_DPAD_TO_AXES, XTYPE_XBOX },
113         { 0x045e, 0x0719, "Xbox 360 Wireless Receiver", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX360W },
114         { 0x0c12, 0x8809, "RedOctane Xbox Dance Pad", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX },
115         { 0x044f, 0x0f07, "Thrustmaster, Inc. Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX },
116         { 0x046d, 0xc242, "Logitech Chillstream Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX360 },
117         { 0x046d, 0xca84, "Logitech Xbox Cordless Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX },
118         { 0x046d, 0xca88, "Logitech Compact Controller for Xbox", MAP_DPAD_TO_AXES, XTYPE_XBOX },
119         { 0x05fd, 0x1007, "Mad Catz Controller (unverified)", MAP_DPAD_TO_AXES, XTYPE_XBOX },
120         { 0x05fd, 0x107a, "InterAct 'PowerPad Pro' X-Box pad (Germany)", MAP_DPAD_TO_AXES, XTYPE_XBOX },
121         { 0x0738, 0x4516, "Mad Catz Control Pad", MAP_DPAD_TO_AXES, XTYPE_XBOX },
122         { 0x0738, 0x4522, "Mad Catz LumiCON", MAP_DPAD_TO_AXES, XTYPE_XBOX },
123         { 0x0738, 0x4526, "Mad Catz Control Pad Pro", MAP_DPAD_TO_AXES, XTYPE_XBOX },
124         { 0x0738, 0x4536, "Mad Catz MicroCON", MAP_DPAD_TO_AXES, XTYPE_XBOX },
125         { 0x0738, 0x4540, "Mad Catz Beat Pad", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX },
126         { 0x0738, 0x4556, "Mad Catz Lynx Wireless Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX },
127         { 0x0738, 0x4716, "Mad Catz Wired Xbox 360 Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX360 },
128         { 0x0738, 0x6040, "Mad Catz Beat Pad Pro", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX },
129         { 0x0c12, 0x8802, "Zeroplus Xbox Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX },
130         { 0x0c12, 0x8810, "Zeroplus Xbox Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX },
131         { 0x0c12, 0x9902, "HAMA VibraX - *FAULTY HARDWARE*", MAP_DPAD_TO_AXES, XTYPE_XBOX },
132         { 0x0e4c, 0x1097, "Radica Gamester Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX },
133         { 0x0e4c, 0x2390, "Radica Games Jtech Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX },
134         { 0x0e6f, 0x0003, "Logic3 Freebird wireless Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX },
135         { 0x0e6f, 0x0005, "Eclipse wireless Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX },
136         { 0x0e6f, 0x0006, "Edge wireless Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX },
137         { 0x0e6f, 0x0006, "Pelican 'TSZ' Wired Xbox 360 Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX360 },
138         { 0x0e8f, 0x0201, "SmartJoy Frag Xpad/PS2 adaptor", MAP_DPAD_TO_AXES, XTYPE_XBOX },
139         { 0x0f30, 0x0202, "Joytech Advanced Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX },
140         { 0x0f30, 0x8888, "BigBen XBMiniPad Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX },
141         { 0x102c, 0xff0c, "Joytech Wireless Advanced Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX },
142         { 0x12ab, 0x8809, "Xbox DDR dancepad", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX },
143         { 0x1430, 0x4748, "RedOctane Guitar Hero X-plorer", MAP_DPAD_TO_AXES, XTYPE_XBOX360 },
144         { 0x1430, 0x8888, "TX6500+ Dance Pad (first generation)", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX },
145         { 0x045e, 0x028e, "Microsoft X-Box 360 pad", MAP_DPAD_TO_AXES, XTYPE_XBOX360 },
146         { 0xffff, 0xffff, "Chinese-made Xbox Controller", MAP_DPAD_TO_AXES, XTYPE_XBOX },
147         { 0x0000, 0x0000, "Generic X-Box pad", MAP_DPAD_UNKNOWN, XTYPE_UNKNOWN }
148 };
149
150 /* buttons shared with xbox and xbox360 */
151 static const signed short xpad_common_btn[] = {
152         BTN_A, BTN_B, BTN_X, BTN_Y,                     /* "analog" buttons */
153         BTN_START, BTN_BACK, BTN_THUMBL, BTN_THUMBR,    /* start/back/sticks */
154         -1                                              /* terminating entry */
155 };
156
157 /* original xbox controllers only */
158 static const signed short xpad_btn[] = {
159         BTN_C, BTN_Z,           /* "analog" buttons */
160         -1                      /* terminating entry */
161 };
162
163 /* only used if MAP_DPAD_TO_BUTTONS */
164 static const signed short xpad_btn_pad[] = {
165         BTN_LEFT, BTN_RIGHT,            /* d-pad left, right */
166         BTN_0, BTN_1,                   /* d-pad up, down (XXX names??) */
167         -1                              /* terminating entry */
168 };
169
170 static const signed short xpad360_btn[] = {  /* buttons for x360 controller */
171         BTN_TL, BTN_TR,         /* Button LB/RB */
172         BTN_MODE,               /* The big X button */
173         -1
174 };
175
176 static const signed short xpad_abs[] = {
177         ABS_X, ABS_Y,           /* left stick */
178         ABS_RX, ABS_RY,         /* right stick */
179         ABS_Z, ABS_RZ,          /* triggers left/right */
180         -1                      /* terminating entry */
181 };
182
183 /* only used if MAP_DPAD_TO_AXES */
184 static const signed short xpad_abs_pad[] = {
185         ABS_HAT0X, ABS_HAT0Y,   /* d-pad axes */
186         -1                      /* terminating entry */
187 };
188
189 /* Xbox 360 has a vendor-specific class, so we cannot match it with only
190  * USB_INTERFACE_INFO (also specifically refused by USB subsystem), so we
191  * match against vendor id as well. Wired Xbox 360 devices have protocol 1,
192  * wireless controllers have protocol 129. */
193 #define XPAD_XBOX360_VENDOR_PROTOCOL(vend,pr) \
194         .match_flags = USB_DEVICE_ID_MATCH_VENDOR | USB_DEVICE_ID_MATCH_INT_INFO, \
195         .idVendor = (vend), \
196         .bInterfaceClass = USB_CLASS_VENDOR_SPEC, \
197         .bInterfaceSubClass = 93, \
198         .bInterfaceProtocol = (pr)
199 #define XPAD_XBOX360_VENDOR(vend) \
200         { XPAD_XBOX360_VENDOR_PROTOCOL(vend,1) }, \
201         { XPAD_XBOX360_VENDOR_PROTOCOL(vend,129) }
202
203 static struct usb_device_id xpad_table [] = {
204         { USB_INTERFACE_INFO('X', 'B', 0) },    /* X-Box USB-IF not approved class */
205         XPAD_XBOX360_VENDOR(0x045e),            /* Microsoft X-Box 360 controllers */
206         XPAD_XBOX360_VENDOR(0x046d),            /* Logitech X-Box 360 style controllers */
207         XPAD_XBOX360_VENDOR(0x0738),            /* Mad Catz X-Box 360 controllers */
208         XPAD_XBOX360_VENDOR(0x0e6f),            /* 0x0e6f X-Box 360 controllers */
209         XPAD_XBOX360_VENDOR(0x1430),            /* RedOctane X-Box 360 controllers */
210         { }
211 };
212
213 MODULE_DEVICE_TABLE (usb, xpad_table);
214
215 struct usb_xpad {
216         struct input_dev *dev;          /* input device interface */
217         struct usb_device *udev;        /* usb device */
218
219         int pad_present;
220
221         struct urb *irq_in;             /* urb for interrupt in report */
222         unsigned char *idata;           /* input data */
223         dma_addr_t idata_dma;
224
225         struct urb *bulk_out;
226         unsigned char *bdata;
227
228 #if defined(CONFIG_JOYSTICK_XPAD_FF) || defined(CONFIG_JOYSTICK_XPAD_LEDS)
229         struct urb *irq_out;            /* urb for interrupt out report */
230         unsigned char *odata;           /* output data */
231         dma_addr_t odata_dma;
232         struct mutex odata_mutex;
233 #endif
234
235 #if defined(CONFIG_JOYSTICK_XPAD_LEDS)
236         struct xpad_led *led;
237 #endif
238
239         char phys[64];                  /* physical device path */
240
241         int dpad_mapping;               /* map d-pad to buttons or to axes */
242         int xtype;                      /* type of xbox device */
243 };
244
245 /*
246  *      xpad_process_packet
247  *
248  *      Completes a request by converting the data into events for the
249  *      input subsystem.
250  *
251  *      The used report descriptor was taken from ITO Takayukis website:
252  *       http://euc.jp/periphs/xbox-controller.ja.html
253  */
254
255 static void xpad_process_packet(struct usb_xpad *xpad, u16 cmd, unsigned char *data)
256 {
257         struct input_dev *dev = xpad->dev;
258
259         /* left stick */
260         input_report_abs(dev, ABS_X,
261                          (__s16) le16_to_cpup((__le16 *)(data + 12)));
262         input_report_abs(dev, ABS_Y,
263                          ~(__s16) le16_to_cpup((__le16 *)(data + 14)));
264
265         /* right stick */
266         input_report_abs(dev, ABS_RX,
267                          (__s16) le16_to_cpup((__le16 *)(data + 16)));
268         input_report_abs(dev, ABS_RY,
269                          ~(__s16) le16_to_cpup((__le16 *)(data + 18)));
270
271         /* triggers left/right */
272         input_report_abs(dev, ABS_Z, data[10]);
273         input_report_abs(dev, ABS_RZ, data[11]);
274
275         /* digital pad */
276         if (xpad->dpad_mapping == MAP_DPAD_TO_AXES) {
277                 input_report_abs(dev, ABS_HAT0X,
278                                  !!(data[2] & 0x08) - !!(data[2] & 0x04));
279                 input_report_abs(dev, ABS_HAT0Y,
280                                  !!(data[2] & 0x02) - !!(data[2] & 0x01));
281         } else /* xpad->dpad_mapping == MAP_DPAD_TO_BUTTONS */ {
282                 input_report_key(dev, BTN_LEFT,  data[2] & 0x04);
283                 input_report_key(dev, BTN_RIGHT, data[2] & 0x08);
284                 input_report_key(dev, BTN_0,     data[2] & 0x01); /* up */
285                 input_report_key(dev, BTN_1,     data[2] & 0x02); /* down */
286         }
287
288         /* start/back buttons and stick press left/right */
289         input_report_key(dev, BTN_START,  data[2] & 0x10);
290         input_report_key(dev, BTN_BACK,   data[2] & 0x20);
291         input_report_key(dev, BTN_THUMBL, data[2] & 0x40);
292         input_report_key(dev, BTN_THUMBR, data[2] & 0x80);
293
294         /* "analog" buttons A, B, X, Y */
295         input_report_key(dev, BTN_A, data[4]);
296         input_report_key(dev, BTN_B, data[5]);
297         input_report_key(dev, BTN_X, data[6]);
298         input_report_key(dev, BTN_Y, data[7]);
299
300         /* "analog" buttons black, white */
301         input_report_key(dev, BTN_C, data[8]);
302         input_report_key(dev, BTN_Z, data[9]);
303
304         input_sync(dev);
305 }
306
307 /*
308  *      xpad360_process_packet
309  *
310  *      Completes a request by converting the data into events for the
311  *      input subsystem. It is version for xbox 360 controller
312  *
313  *      The used report descriptor was taken from:
314  *              http://www.free60.org/wiki/Gamepad
315  */
316
317 static void xpad360_process_packet(struct usb_xpad *xpad,
318                                    u16 cmd, unsigned char *data)
319 {
320         struct input_dev *dev = xpad->dev;
321
322         /* digital pad */
323         if (xpad->dpad_mapping == MAP_DPAD_TO_AXES) {
324                 input_report_abs(dev, ABS_HAT0X,
325                                  !!(data[2] & 0x08) - !!(data[2] & 0x04));
326                 input_report_abs(dev, ABS_HAT0Y,
327                                  !!(data[2] & 0x02) - !!(data[2] & 0x01));
328         } else if (xpad->dpad_mapping == MAP_DPAD_TO_BUTTONS) {
329                 /* dpad as buttons (right, left, down, up) */
330                 input_report_key(dev, BTN_LEFT, data[2] & 0x04);
331                 input_report_key(dev, BTN_RIGHT, data[2] & 0x08);
332                 input_report_key(dev, BTN_0, data[2] & 0x01);   /* up */
333                 input_report_key(dev, BTN_1, data[2] & 0x02);   /* down */
334         }
335
336         /* start/back buttons */
337         input_report_key(dev, BTN_START,  data[2] & 0x10);
338         input_report_key(dev, BTN_BACK,   data[2] & 0x20);
339
340         /* stick press left/right */
341         input_report_key(dev, BTN_THUMBL, data[2] & 0x40);
342         input_report_key(dev, BTN_THUMBR, data[2] & 0x80);
343
344         /* buttons A,B,X,Y,TL,TR and MODE */
345         input_report_key(dev, BTN_A,    data[3] & 0x10);
346         input_report_key(dev, BTN_B,    data[3] & 0x20);
347         input_report_key(dev, BTN_X,    data[3] & 0x40);
348         input_report_key(dev, BTN_Y,    data[3] & 0x80);
349         input_report_key(dev, BTN_TL,   data[3] & 0x01);
350         input_report_key(dev, BTN_TR,   data[3] & 0x02);
351         input_report_key(dev, BTN_MODE, data[3] & 0x04);
352
353         /* left stick */
354         input_report_abs(dev, ABS_X,
355                          (__s16) le16_to_cpup((__le16 *)(data + 6)));
356         input_report_abs(dev, ABS_Y,
357                          ~(__s16) le16_to_cpup((__le16 *)(data + 8)));
358
359         /* right stick */
360         input_report_abs(dev, ABS_RX,
361                          (__s16) le16_to_cpup((__le16 *)(data + 10)));
362         input_report_abs(dev, ABS_RY,
363                          ~(__s16) le16_to_cpup((__le16 *)(data + 12)));
364
365         /* triggers left/right */
366         input_report_abs(dev, ABS_Z, data[4]);
367         input_report_abs(dev, ABS_RZ, data[5]);
368
369         input_sync(dev);
370 }
371
372 /*
373  * xpad360w_process_packet
374  *
375  * Completes a request by converting the data into events for the
376  * input subsystem. It is version for xbox 360 wireless controller.
377  *
378  * Byte.Bit
379  * 00.1 - Status change: The controller or headset has connected/disconnected
380  *                       Bits 01.7 and 01.6 are valid
381  * 01.7 - Controller present
382  * 01.6 - Headset present
383  * 01.1 - Pad state (Bytes 4+) valid
384  *
385  */
386
387 static void xpad360w_process_packet(struct usb_xpad *xpad, u16 cmd, unsigned char *data)
388 {
389         /* Presence change */
390         if (data[0] & 0x08) {
391                 if (data[1] & 0x80) {
392                         xpad->pad_present = 1;
393                         usb_submit_urb(xpad->bulk_out, GFP_ATOMIC);
394                 } else
395                         xpad->pad_present = 0;
396         }
397
398         /* Valid pad data */
399         if (!(data[1] & 0x1))
400                 return;
401
402         xpad360_process_packet(xpad, cmd, &data[4]);
403 }
404
405 static void xpad_irq_in(struct urb *urb)
406 {
407         struct usb_xpad *xpad = urb->context;
408         int retval;
409
410         switch (urb->status) {
411         case 0:
412                 /* success */
413                 break;
414         case -ECONNRESET:
415         case -ENOENT:
416         case -ESHUTDOWN:
417                 /* this urb is terminated, clean up */
418                 dbg("%s - urb shutting down with status: %d",
419                         __FUNCTION__, urb->status);
420                 return;
421         default:
422                 dbg("%s - nonzero urb status received: %d",
423                         __FUNCTION__, urb->status);
424                 goto exit;
425         }
426
427         switch (xpad->xtype) {
428         case XTYPE_XBOX360:
429                 xpad360_process_packet(xpad, 0, xpad->idata);
430                 break;
431         case XTYPE_XBOX360W:
432                 xpad360w_process_packet(xpad, 0, xpad->idata);
433                 break;
434         default:
435                 xpad_process_packet(xpad, 0, xpad->idata);
436         }
437
438 exit:
439         retval = usb_submit_urb (urb, GFP_ATOMIC);
440         if (retval)
441                 err ("%s - usb_submit_urb failed with result %d",
442                      __FUNCTION__, retval);
443 }
444
445 #if defined(CONFIG_JOYSTICK_XPAD_FF) || defined(CONFIG_JOYSTICK_XPAD_LEDS)
446 static void xpad_irq_out(struct urb *urb)
447 {
448         int retval;
449
450         switch (urb->status) {
451                 case 0:
452                 /* success */
453                 break;
454                 case -ECONNRESET:
455                 case -ENOENT:
456                 case -ESHUTDOWN:
457                         /* this urb is terminated, clean up */
458                         dbg("%s - urb shutting down with status: %d",
459                                 __FUNCTION__, urb->status);
460                         return;
461                 default:
462                         dbg("%s - nonzero urb status received: %d",
463                                 __FUNCTION__, urb->status);
464                         goto exit;
465         }
466
467 exit:
468         retval = usb_submit_urb(urb, GFP_ATOMIC);
469         if (retval)
470                 err("%s - usb_submit_urb failed with result %d",
471                     __FUNCTION__, retval);
472 }
473
474 static void xpad_bulk_out(struct urb *urb)
475 {
476         switch (urb->status) {
477         case 0:
478                 /* success */
479                 break;
480         case -ECONNRESET:
481         case -ENOENT:
482         case -ESHUTDOWN:
483                 /* this urb is terminated, clean up */
484                 dbg("%s - urb shutting down with status: %d", __FUNCTION__, urb->status);
485                 break;
486         default:
487                 dbg("%s - nonzero urb status received: %d", __FUNCTION__, urb->status);
488         }
489 }
490
491 static int xpad_init_output(struct usb_interface *intf, struct usb_xpad *xpad)
492 {
493         struct usb_endpoint_descriptor *ep_irq_out;
494         int error = -ENOMEM;
495
496         if (xpad->xtype != XTYPE_XBOX360)
497                 return 0;
498
499         xpad->odata = usb_buffer_alloc(xpad->udev, XPAD_PKT_LEN,
500                                        GFP_ATOMIC, &xpad->odata_dma );
501         if (!xpad->odata)
502                 goto fail1;
503
504         mutex_init(&xpad->odata_mutex);
505
506         xpad->irq_out = usb_alloc_urb(0, GFP_KERNEL);
507         if (!xpad->irq_out)
508                 goto fail2;
509
510         ep_irq_out = &intf->cur_altsetting->endpoint[1].desc;
511         usb_fill_int_urb(xpad->irq_out, xpad->udev,
512                          usb_sndintpipe(xpad->udev, ep_irq_out->bEndpointAddress),
513                          xpad->odata, XPAD_PKT_LEN,
514                          xpad_irq_out, xpad, ep_irq_out->bInterval);
515         xpad->irq_out->transfer_dma = xpad->odata_dma;
516         xpad->irq_out->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
517
518         return 0;
519
520  fail2: usb_buffer_free(xpad->udev, XPAD_PKT_LEN, xpad->odata, xpad->odata_dma);
521  fail1: return error;
522 }
523
524 static void xpad_stop_output(struct usb_xpad *xpad)
525 {
526         if (xpad->xtype == XTYPE_XBOX360)
527                 usb_kill_urb(xpad->irq_out);
528 }
529
530 static void xpad_deinit_output(struct usb_xpad *xpad)
531 {
532         if (xpad->xtype == XTYPE_XBOX360) {
533                 usb_free_urb(xpad->irq_out);
534                 usb_buffer_free(xpad->udev, XPAD_PKT_LEN,
535                                 xpad->odata, xpad->odata_dma);
536         }
537 }
538 #else
539 static int xpad_init_output(struct usb_interface *intf, struct usb_xpad *xpad) { return 0; }
540 static void xpad_deinit_output(struct usb_xpad *xpad) {}
541 static void xpad_stop_output(struct usb_xpad *xpad) {}
542 #endif
543
544 #ifdef CONFIG_JOYSTICK_XPAD_FF
545 static int xpad_play_effect(struct input_dev *dev, void *data,
546                             struct ff_effect *effect)
547 {
548         struct usb_xpad *xpad = input_get_drvdata(dev);
549
550         if (effect->type == FF_RUMBLE) {
551                 __u16 strong = effect->u.rumble.strong_magnitude;
552                 __u16 weak = effect->u.rumble.weak_magnitude;
553                 xpad->odata[0] = 0x00;
554                 xpad->odata[1] = 0x08;
555                 xpad->odata[2] = 0x00;
556                 xpad->odata[3] = strong / 256;
557                 xpad->odata[4] = weak / 256;
558                 xpad->odata[5] = 0x00;
559                 xpad->odata[6] = 0x00;
560                 xpad->odata[7] = 0x00;
561                 usb_submit_urb(xpad->irq_out, GFP_KERNEL);
562         }
563
564         return 0;
565 }
566
567 static int xpad_init_ff(struct usb_xpad *xpad)
568 {
569         if (xpad->xtype != XTYPE_XBOX360)
570                 return 0;
571
572         input_set_capability(xpad->dev, EV_FF, FF_RUMBLE);
573
574         return input_ff_create_memless(xpad->dev, NULL, xpad_play_effect);
575 }
576
577 #else
578 static int xpad_init_ff(struct usb_xpad *xpad) { return 0; }
579 #endif
580
581 #if defined(CONFIG_JOYSTICK_XPAD_LEDS)
582 #include <linux/leds.h>
583
584 struct xpad_led {
585         char name[16];
586         struct led_classdev led_cdev;
587         struct usb_xpad *xpad;
588 };
589
590 static void xpad_send_led_command(struct usb_xpad *xpad, int command)
591 {
592         if (command >= 0 && command < 14) {
593                 mutex_lock(&xpad->odata_mutex);
594                 xpad->odata[0] = 0x01;
595                 xpad->odata[1] = 0x03;
596                 xpad->odata[2] = command;
597                 usb_submit_urb(xpad->irq_out, GFP_KERNEL);
598                 mutex_unlock(&xpad->odata_mutex);
599         }
600 }
601
602 static void xpad_led_set(struct led_classdev *led_cdev,
603                          enum led_brightness value)
604 {
605         struct xpad_led *xpad_led = container_of(led_cdev,
606                                                  struct xpad_led, led_cdev);
607
608         xpad_send_led_command(xpad_led->xpad, value);
609 }
610
611 static int xpad_led_probe(struct usb_xpad *xpad)
612 {
613         static atomic_t led_seq = ATOMIC_INIT(0);
614         long led_no;
615         struct xpad_led *led;
616         struct led_classdev *led_cdev;
617         int error;
618
619         if (xpad->xtype != XTYPE_XBOX360)
620                 return 0;
621
622         xpad->led = led = kzalloc(sizeof(struct xpad_led), GFP_KERNEL);
623         if (!led)
624                 return -ENOMEM;
625
626         led_no = (long)atomic_inc_return(&led_seq) - 1;
627
628         snprintf(led->name, sizeof(led->name), "xpad%ld", led_no);
629         led->xpad = xpad;
630
631         led_cdev = &led->led_cdev;
632         led_cdev->name = led->name;
633         led_cdev->brightness_set = xpad_led_set;
634
635         error = led_classdev_register(&xpad->udev->dev, led_cdev);
636         if (error) {
637                 kfree(led);
638                 xpad->led = NULL;
639                 return error;
640         }
641
642         /*
643          * Light up the segment corresponding to controller number
644          */
645         xpad_send_led_command(xpad, (led_no % 4) + 2);
646
647         return 0;
648 }
649
650 static void xpad_led_disconnect(struct usb_xpad *xpad)
651 {
652         struct xpad_led *xpad_led = xpad->led;
653
654         if (xpad_led) {
655                 led_classdev_unregister(&xpad_led->led_cdev);
656                 kfree(xpad_led->name);
657         }
658 }
659 #else
660 static int xpad_led_probe(struct usb_xpad *xpad) { return 0; }
661 static void xpad_led_disconnect(struct usb_xpad *xpad) { }
662 #endif
663
664
665 static int xpad_open(struct input_dev *dev)
666 {
667         struct usb_xpad *xpad = input_get_drvdata(dev);
668
669         /* URB was submitted in probe */
670         if(xpad->xtype == XTYPE_XBOX360W)
671                 return 0;
672
673         xpad->irq_in->dev = xpad->udev;
674         if (usb_submit_urb(xpad->irq_in, GFP_KERNEL))
675                 return -EIO;
676
677         return 0;
678 }
679
680 static void xpad_close(struct input_dev *dev)
681 {
682         struct usb_xpad *xpad = input_get_drvdata(dev);
683
684         if(xpad->xtype != XTYPE_XBOX360W)
685                 usb_kill_urb(xpad->irq_in);
686         xpad_stop_output(xpad);
687 }
688
689 static void xpad_set_up_abs(struct input_dev *input_dev, signed short abs)
690 {
691         set_bit(abs, input_dev->absbit);
692
693         switch (abs) {
694         case ABS_X:
695         case ABS_Y:
696         case ABS_RX:
697         case ABS_RY:    /* the two sticks */
698                 input_set_abs_params(input_dev, abs, -32768, 32767, 16, 128);
699                 break;
700         case ABS_Z:
701         case ABS_RZ:    /* the triggers */
702                 input_set_abs_params(input_dev, abs, 0, 255, 0, 0);
703                 break;
704         case ABS_HAT0X:
705         case ABS_HAT0Y: /* the d-pad (only if MAP_DPAD_TO_AXES) */
706                 input_set_abs_params(input_dev, abs, -1, 1, 0, 0);
707                 break;
708         }
709 }
710
711 static int xpad_probe(struct usb_interface *intf, const struct usb_device_id *id)
712 {
713         struct usb_device *udev = interface_to_usbdev(intf);
714         struct usb_xpad *xpad;
715         struct input_dev *input_dev;
716         struct usb_endpoint_descriptor *ep_irq_in;
717         int i;
718         int error = -ENOMEM;
719
720         for (i = 0; xpad_device[i].idVendor; i++) {
721                 if ((le16_to_cpu(udev->descriptor.idVendor) == xpad_device[i].idVendor) &&
722                     (le16_to_cpu(udev->descriptor.idProduct) == xpad_device[i].idProduct))
723                         break;
724         }
725
726         xpad = kzalloc(sizeof(struct usb_xpad), GFP_KERNEL);
727         input_dev = input_allocate_device();
728         if (!xpad || !input_dev)
729                 goto fail1;
730
731         xpad->idata = usb_buffer_alloc(udev, XPAD_PKT_LEN,
732                                        GFP_ATOMIC, &xpad->idata_dma);
733         if (!xpad->idata)
734                 goto fail1;
735
736         xpad->irq_in = usb_alloc_urb(0, GFP_KERNEL);
737         if (!xpad->irq_in)
738                 goto fail2;
739
740         xpad->udev = udev;
741         xpad->dpad_mapping = xpad_device[i].dpad_mapping;
742         xpad->xtype = xpad_device[i].xtype;
743         if (xpad->dpad_mapping == MAP_DPAD_UNKNOWN)
744                 xpad->dpad_mapping = !dpad_to_buttons;
745         if (xpad->xtype == XTYPE_UNKNOWN) {
746                 if (intf->cur_altsetting->desc.bInterfaceClass == USB_CLASS_VENDOR_SPEC) {
747                         if (intf->cur_altsetting->desc.bInterfaceProtocol == 129)
748                                 xpad->xtype = XTYPE_XBOX360W;
749                         else
750                                 xpad->xtype = XTYPE_XBOX360;
751                 } else
752                         xpad->xtype = XTYPE_XBOX;
753         }
754         xpad->dev = input_dev;
755         usb_make_path(udev, xpad->phys, sizeof(xpad->phys));
756         strlcat(xpad->phys, "/input0", sizeof(xpad->phys));
757
758         input_dev->name = xpad_device[i].name;
759         input_dev->phys = xpad->phys;
760         usb_to_input_id(udev, &input_dev->id);
761         input_dev->dev.parent = &intf->dev;
762
763         input_set_drvdata(input_dev, xpad);
764
765         input_dev->open = xpad_open;
766         input_dev->close = xpad_close;
767
768         input_dev->evbit[0] = BIT_MASK(EV_KEY) | BIT_MASK(EV_ABS);
769
770         /* set up buttons */
771         for (i = 0; xpad_common_btn[i] >= 0; i++)
772                 set_bit(xpad_common_btn[i], input_dev->keybit);
773         if ((xpad->xtype == XTYPE_XBOX360) || (xpad->xtype == XTYPE_XBOX360W))
774                 for (i = 0; xpad360_btn[i] >= 0; i++)
775                         set_bit(xpad360_btn[i], input_dev->keybit);
776         else
777                 for (i = 0; xpad_btn[i] >= 0; i++)
778                         set_bit(xpad_btn[i], input_dev->keybit);
779         if (xpad->dpad_mapping == MAP_DPAD_TO_BUTTONS)
780                 for (i = 0; xpad_btn_pad[i] >= 0; i++)
781                         set_bit(xpad_btn_pad[i], input_dev->keybit);
782
783         /* set up axes */
784         for (i = 0; xpad_abs[i] >= 0; i++)
785                 xpad_set_up_abs(input_dev, xpad_abs[i]);
786         if (xpad->dpad_mapping == MAP_DPAD_TO_AXES)
787                 for (i = 0; xpad_abs_pad[i] >= 0; i++)
788                     xpad_set_up_abs(input_dev, xpad_abs_pad[i]);
789
790         error = xpad_init_output(intf, xpad);
791         if (error)
792                 goto fail2;
793
794         error = xpad_init_ff(xpad);
795         if (error)
796                 goto fail3;
797
798         error = xpad_led_probe(xpad);
799         if (error)
800                 goto fail3;
801
802         ep_irq_in = &intf->cur_altsetting->endpoint[0].desc;
803         usb_fill_int_urb(xpad->irq_in, udev,
804                          usb_rcvintpipe(udev, ep_irq_in->bEndpointAddress),
805                          xpad->idata, XPAD_PKT_LEN, xpad_irq_in,
806                          xpad, ep_irq_in->bInterval);
807         xpad->irq_in->transfer_dma = xpad->idata_dma;
808         xpad->irq_in->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
809
810         error = input_register_device(xpad->dev);
811         if (error)
812                 goto fail4;
813
814         usb_set_intfdata(intf, xpad);
815
816         /*
817          * Submit the int URB immediatly rather than waiting for open
818          * because we get status messages from the device whether
819          * or not any controllers are attached.  In fact, it's
820          * exactly the message that a controller has arrived that
821          * we're waiting for.
822          */
823         if (xpad->xtype == XTYPE_XBOX360W) {
824                 xpad->irq_in->dev = xpad->udev;
825                 error = usb_submit_urb(xpad->irq_in, GFP_KERNEL);
826                 if (error)
827                         goto fail4;
828
829                 /*
830                  * Setup the message to set the LEDs on the
831                  * controller when it shows up
832                  */
833                 xpad->bulk_out = usb_alloc_urb(0, GFP_KERNEL);
834                 if(!xpad->bulk_out)
835                         goto fail5;
836
837                 xpad->bdata = kzalloc(XPAD_PKT_LEN, GFP_KERNEL);
838                 if(!xpad->bdata)
839                         goto fail6;
840
841                 xpad->bdata[2] = 0x08;
842                 switch (intf->cur_altsetting->desc.bInterfaceNumber) {
843                 case 0:
844                         xpad->bdata[3] = 0x42;
845                         break;
846                 case 2:
847                         xpad->bdata[3] = 0x43;
848                         break;
849                 case 4:
850                         xpad->bdata[3] = 0x44;
851                         break;
852                 case 6:
853                         xpad->bdata[3] = 0x45;
854                 }
855
856                 ep_irq_in = &intf->cur_altsetting->endpoint[1].desc;
857                 usb_fill_bulk_urb(xpad->bulk_out, udev,
858                                 usb_sndbulkpipe(udev, ep_irq_in->bEndpointAddress),
859                                 xpad->bdata, XPAD_PKT_LEN, xpad_bulk_out, xpad);
860         }
861
862         return 0;
863
864  fail6: usb_free_urb(xpad->bulk_out);
865  fail5: usb_kill_urb(xpad->irq_in);
866  fail4: usb_free_urb(xpad->irq_in);
867  fail3: xpad_deinit_output(xpad);
868  fail2: usb_buffer_free(udev, XPAD_PKT_LEN, xpad->idata, xpad->idata_dma);
869  fail1: input_free_device(input_dev);
870         kfree(xpad);
871         return error;
872
873 }
874
875 static void xpad_disconnect(struct usb_interface *intf)
876 {
877         struct usb_xpad *xpad = usb_get_intfdata (intf);
878
879         usb_set_intfdata(intf, NULL);
880         if (xpad) {
881                 xpad_led_disconnect(xpad);
882                 input_unregister_device(xpad->dev);
883                 xpad_deinit_output(xpad);
884                 if (xpad->xtype == XTYPE_XBOX360W) {
885                         usb_kill_urb(xpad->bulk_out);
886                         usb_free_urb(xpad->bulk_out);
887                         usb_kill_urb(xpad->irq_in);
888                 }
889                 usb_free_urb(xpad->irq_in);
890                 usb_buffer_free(xpad->udev, XPAD_PKT_LEN,
891                                 xpad->idata, xpad->idata_dma);
892                 kfree(xpad);
893         }
894 }
895
896 static struct usb_driver xpad_driver = {
897         .name           = "xpad",
898         .probe          = xpad_probe,
899         .disconnect     = xpad_disconnect,
900         .id_table       = xpad_table,
901 };
902
903 static int __init usb_xpad_init(void)
904 {
905         int result = usb_register(&xpad_driver);
906         if (result == 0)
907                 info(DRIVER_DESC);
908         return result;
909 }
910
911 static void __exit usb_xpad_exit(void)
912 {
913         usb_deregister(&xpad_driver);
914 }
915
916 module_init(usb_xpad_init);
917 module_exit(usb_xpad_exit);
918
919 MODULE_AUTHOR(DRIVER_AUTHOR);
920 MODULE_DESCRIPTION(DRIVER_DESC);
921 MODULE_LICENSE("GPL");