]> git.kernelconcepts.de Git - karo-tx-linux.git/blob - drivers/input/joystick/xpad.c
Merge remote-tracking branch 'usb-chipidea-fixes/ci-for-usb-stable'
[karo-tx-linux.git] / 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  *               2010 Christoph Fritz <chf.fritz@googlemail.com>
13  *
14  * This program is free software; you can redistribute it and/or
15  * modify it under the terms of the GNU General Public License as
16  * published by the Free Software Foundation; either version 2 of
17  * the License, or (at your option) any later version.
18  *
19  * This program is distributed in the hope that it will be useful,
20  * but WITHOUT ANY WARRANTY; without even the implied warranty of
21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
22  * GNU General Public License for more details.
23  *
24  * You should have received a copy of the GNU General Public License
25  * along with this program; if not, write to the Free Software
26  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
27  *
28  *
29  * This driver is based on:
30  *  - information from     http://euc.jp/periphs/xbox-controller.ja.html
31  *  - the iForce driver    drivers/char/joystick/iforce.c
32  *  - the skeleton-driver  drivers/usb/usb-skeleton.c
33  *  - Xbox 360 information http://www.free60.org/wiki/Gamepad
34  *  - Xbox One information https://github.com/quantus/xbox-one-controller-protocol
35  *
36  * Thanks to:
37  *  - ITO Takayuki for providing essential xpad information on his website
38  *  - Vojtech Pavlik     - iforce driver / input subsystem
39  *  - Greg Kroah-Hartman - usb-skeleton driver
40  *  - XBOX Linux project - extra USB id's
41  *  - Pekka Pöyry (quantus) - Xbox One controller reverse engineering
42  *
43  * TODO:
44  *  - fine tune axes (especially trigger axes)
45  *  - fix "analog" buttons (reported as digital now)
46  *  - get rumble working
47  *  - need USB IDs for other dance pads
48  *
49  * History:
50  *
51  * 2002-06-27 - 0.0.1 : first version, just said "XBOX HID controller"
52  *
53  * 2002-07-02 - 0.0.2 : basic working version
54  *  - all axes and 9 of the 10 buttons work (german InterAct device)
55  *  - the black button does not work
56  *
57  * 2002-07-14 - 0.0.3 : rework by Vojtech Pavlik
58  *  - indentation fixes
59  *  - usb + input init sequence fixes
60  *
61  * 2002-07-16 - 0.0.4 : minor changes, merge with Vojtech's v0.0.3
62  *  - verified the lack of HID and report descriptors
63  *  - verified that ALL buttons WORK
64  *  - fixed d-pad to axes mapping
65  *
66  * 2002-07-17 - 0.0.5 : simplified d-pad handling
67  *
68  * 2004-10-02 - 0.0.6 : DDR pad support
69  *  - borrowed from the XBOX linux kernel
70  *  - USB id's for commonly used dance pads are present
71  *  - dance pads will map D-PAD to buttons, not axes
72  *  - pass the module paramater 'dpad_to_buttons' to force
73  *    the D-PAD to map to buttons if your pad is not detected
74  *
75  * Later changes can be tracked in SCM.
76  */
77
78 #include <linux/kernel.h>
79 #include <linux/slab.h>
80 #include <linux/stat.h>
81 #include <linux/module.h>
82 #include <linux/usb/input.h>
83
84 #define DRIVER_AUTHOR "Marko Friedemann <mfr@bmx-chemnitz.de>"
85 #define DRIVER_DESC "X-Box pad driver"
86
87 #define XPAD_PKT_LEN 32
88
89 /* xbox d-pads should map to buttons, as is required for DDR pads
90    but we map them to axes when possible to simplify things */
91 #define MAP_DPAD_TO_BUTTONS             (1 << 0)
92 #define MAP_TRIGGERS_TO_BUTTONS         (1 << 1)
93 #define MAP_STICKS_TO_NULL              (1 << 2)
94 #define DANCEPAD_MAP_CONFIG     (MAP_DPAD_TO_BUTTONS |                  \
95                                 MAP_TRIGGERS_TO_BUTTONS | MAP_STICKS_TO_NULL)
96
97 #define XTYPE_XBOX        0
98 #define XTYPE_XBOX360     1
99 #define XTYPE_XBOX360W    2
100 #define XTYPE_XBOXONE     3
101 #define XTYPE_UNKNOWN     4
102
103 static bool dpad_to_buttons;
104 module_param(dpad_to_buttons, bool, S_IRUGO);
105 MODULE_PARM_DESC(dpad_to_buttons, "Map D-PAD to buttons rather than axes for unknown pads");
106
107 static bool triggers_to_buttons;
108 module_param(triggers_to_buttons, bool, S_IRUGO);
109 MODULE_PARM_DESC(triggers_to_buttons, "Map triggers to buttons rather than axes for unknown pads");
110
111 static bool sticks_to_null;
112 module_param(sticks_to_null, bool, S_IRUGO);
113 MODULE_PARM_DESC(sticks_to_null, "Do not map sticks at all for unknown pads");
114
115 static const struct xpad_device {
116         u16 idVendor;
117         u16 idProduct;
118         char *name;
119         u8 mapping;
120         u8 xtype;
121 } xpad_device[] = {
122         { 0x045e, 0x0202, "Microsoft X-Box pad v1 (US)", 0, XTYPE_XBOX },
123         { 0x045e, 0x0285, "Microsoft X-Box pad (Japan)", 0, XTYPE_XBOX },
124         { 0x045e, 0x0287, "Microsoft Xbox Controller S", 0, XTYPE_XBOX },
125         { 0x045e, 0x0289, "Microsoft X-Box pad v2 (US)", 0, XTYPE_XBOX },
126         { 0x045e, 0x028e, "Microsoft X-Box 360 pad", 0, XTYPE_XBOX360 },
127         { 0x045e, 0x02d1, "Microsoft X-Box One pad", 0, XTYPE_XBOXONE },
128         { 0x045e, 0x0291, "Xbox 360 Wireless Receiver (XBOX)", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX360W },
129         { 0x045e, 0x0719, "Xbox 360 Wireless Receiver", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX360W },
130         { 0x044f, 0x0f07, "Thrustmaster, Inc. Controller", 0, XTYPE_XBOX },
131         { 0x044f, 0xb326, "Thrustmaster Gamepad GP XID", 0, XTYPE_XBOX360 },
132         { 0x046d, 0xc21d, "Logitech Gamepad F310", 0, XTYPE_XBOX360 },
133         { 0x046d, 0xc21e, "Logitech Gamepad F510", 0, XTYPE_XBOX360 },
134         { 0x046d, 0xc21f, "Logitech Gamepad F710", 0, XTYPE_XBOX360 },
135         { 0x046d, 0xc242, "Logitech Chillstream Controller", 0, XTYPE_XBOX360 },
136         { 0x046d, 0xca84, "Logitech Xbox Cordless Controller", 0, XTYPE_XBOX },
137         { 0x046d, 0xca88, "Logitech Compact Controller for Xbox", 0, XTYPE_XBOX },
138         { 0x05fd, 0x1007, "Mad Catz Controller (unverified)", 0, XTYPE_XBOX },
139         { 0x05fd, 0x107a, "InterAct 'PowerPad Pro' X-Box pad (Germany)", 0, XTYPE_XBOX },
140         { 0x0738, 0x4516, "Mad Catz Control Pad", 0, XTYPE_XBOX },
141         { 0x0738, 0x4522, "Mad Catz LumiCON", 0, XTYPE_XBOX },
142         { 0x0738, 0x4526, "Mad Catz Control Pad Pro", 0, XTYPE_XBOX },
143         { 0x0738, 0x4536, "Mad Catz MicroCON", 0, XTYPE_XBOX },
144         { 0x0738, 0x4540, "Mad Catz Beat Pad", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX },
145         { 0x0738, 0x4556, "Mad Catz Lynx Wireless Controller", 0, XTYPE_XBOX },
146         { 0x0738, 0x4716, "Mad Catz Wired Xbox 360 Controller", 0, XTYPE_XBOX360 },
147         { 0x0738, 0x4718, "Mad Catz Street Fighter IV FightStick SE", 0, XTYPE_XBOX360 },
148         { 0x0738, 0x4726, "Mad Catz Xbox 360 Controller", 0, XTYPE_XBOX360 },
149         { 0x0738, 0x4728, "Mad Catz Street Fighter IV FightPad", MAP_TRIGGERS_TO_BUTTONS, XTYPE_XBOX360 },
150         { 0x0738, 0x4738, "Mad Catz Wired Xbox 360 Controller (SFIV)", MAP_TRIGGERS_TO_BUTTONS, XTYPE_XBOX360 },
151         { 0x0738, 0x4740, "Mad Catz Beat Pad", 0, XTYPE_XBOX360 },
152         { 0x0738, 0x6040, "Mad Catz Beat Pad Pro", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX },
153         { 0x0738, 0xb726, "Mad Catz Xbox controller - MW2", 0, XTYPE_XBOX360 },
154         { 0x0738, 0xbeef, "Mad Catz JOYTECH NEO SE Advanced GamePad", XTYPE_XBOX360 },
155         { 0x0738, 0xcb02, "Saitek Cyborg Rumble Pad - PC/Xbox 360", 0, XTYPE_XBOX360 },
156         { 0x0738, 0xcb03, "Saitek P3200 Rumble Pad - PC/Xbox 360", 0, XTYPE_XBOX360 },
157         { 0x0738, 0xf738, "Super SFIV FightStick TE S", 0, XTYPE_XBOX360 },
158         { 0x0c12, 0x8802, "Zeroplus Xbox Controller", 0, XTYPE_XBOX },
159         { 0x0c12, 0x8809, "RedOctane Xbox Dance Pad", DANCEPAD_MAP_CONFIG, XTYPE_XBOX },
160         { 0x0c12, 0x880a, "Pelican Eclipse PL-2023", 0, XTYPE_XBOX },
161         { 0x0c12, 0x8810, "Zeroplus Xbox Controller", 0, XTYPE_XBOX },
162         { 0x0c12, 0x9902, "HAMA VibraX - *FAULTY HARDWARE*", 0, XTYPE_XBOX },
163         { 0x0d2f, 0x0002, "Andamiro Pump It Up pad", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX },
164         { 0x0e4c, 0x1097, "Radica Gamester Controller", 0, XTYPE_XBOX },
165         { 0x0e4c, 0x2390, "Radica Games Jtech Controller", 0, XTYPE_XBOX },
166         { 0x0e6f, 0x0003, "Logic3 Freebird wireless Controller", 0, XTYPE_XBOX },
167         { 0x0e6f, 0x0005, "Eclipse wireless Controller", 0, XTYPE_XBOX },
168         { 0x0e6f, 0x0006, "Edge wireless Controller", 0, XTYPE_XBOX },
169         { 0x0e6f, 0x0105, "HSM3 Xbox360 dancepad", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX360 },
170         { 0x0e6f, 0x0113, "Afterglow AX.1 Gamepad for Xbox 360", 0, XTYPE_XBOX360 },
171         { 0x0e6f, 0x0201, "Pelican PL-3601 'TSZ' Wired Xbox 360 Controller", 0, XTYPE_XBOX360 },
172         { 0x0e6f, 0x0213, "Afterglow Gamepad for Xbox 360", 0, XTYPE_XBOX360 },
173         { 0x0e6f, 0x021f, "Rock Candy Gamepad for Xbox 360", 0, XTYPE_XBOX360 },
174         { 0x0e6f, 0x0301, "Logic3 Controller", 0, XTYPE_XBOX360 },
175         { 0x0e6f, 0x0401, "Logic3 Controller", 0, XTYPE_XBOX360 },
176         { 0x0e8f, 0x0201, "SmartJoy Frag Xpad/PS2 adaptor", 0, XTYPE_XBOX },
177         { 0x0e8f, 0x3008, "Generic xbox control (dealextreme)", 0, XTYPE_XBOX },
178         { 0x0f0d, 0x000a, "Hori Co. DOA4 FightStick", 0, XTYPE_XBOX360 },
179         { 0x0f0d, 0x000d, "Hori Fighting Stick EX2", MAP_TRIGGERS_TO_BUTTONS, XTYPE_XBOX360 },
180         { 0x0f0d, 0x0016, "Hori Real Arcade Pro.EX", MAP_TRIGGERS_TO_BUTTONS, XTYPE_XBOX360 },
181         { 0x0f30, 0x0202, "Joytech Advanced Controller", 0, XTYPE_XBOX },
182         { 0x0f30, 0x8888, "BigBen XBMiniPad Controller", 0, XTYPE_XBOX },
183         { 0x102c, 0xff0c, "Joytech Wireless Advanced Controller", 0, XTYPE_XBOX },
184         { 0x12ab, 0x0004, "Honey Bee Xbox360 dancepad", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX360 },
185         { 0x12ab, 0x0301, "PDP AFTERGLOW AX.1", 0, XTYPE_XBOX360 },
186         { 0x12ab, 0x8809, "Xbox DDR dancepad", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX },
187         { 0x1430, 0x4748, "RedOctane Guitar Hero X-plorer", 0, XTYPE_XBOX360 },
188         { 0x1430, 0x8888, "TX6500+ Dance Pad (first generation)", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX },
189         { 0x146b, 0x0601, "BigBen Interactive XBOX 360 Controller", 0, XTYPE_XBOX360 },
190         { 0x1532, 0x0037, "Razer Sabertooth", 0, XTYPE_XBOX360 },
191         { 0x15e4, 0x3f00, "Power A Mini Pro Elite", 0, XTYPE_XBOX360 },
192         { 0x15e4, 0x3f0a, "Xbox Airflo wired controller", 0, XTYPE_XBOX360 },
193         { 0x15e4, 0x3f10, "Batarang Xbox 360 controller", 0, XTYPE_XBOX360 },
194         { 0x162e, 0xbeef, "Joytech Neo-Se Take2", 0, XTYPE_XBOX360 },
195         { 0x1689, 0xfd00, "Razer Onza Tournament Edition", 0, XTYPE_XBOX360 },
196         { 0x1689, 0xfd01, "Razer Onza Classic Edition", 0, XTYPE_XBOX360 },
197         { 0x24c6, 0x5d04, "Razer Sabertooth", 0, XTYPE_XBOX360 },
198         { 0x1bad, 0x0002, "Harmonix Rock Band Guitar", 0, XTYPE_XBOX360 },
199         { 0x1bad, 0x0003, "Harmonix Rock Band Drumkit", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX360 },
200         { 0x1bad, 0xf016, "Mad Catz Xbox 360 Controller", 0, XTYPE_XBOX360 },
201         { 0x1bad, 0xf023, "MLG Pro Circuit Controller (Xbox)", 0, XTYPE_XBOX360 },
202         { 0x1bad, 0xf028, "Street Fighter IV FightPad", 0, XTYPE_XBOX360 },
203         { 0x1bad, 0xf038, "Street Fighter IV FightStick TE", 0, XTYPE_XBOX360 },
204         { 0x1bad, 0xf900, "Harmonix Xbox 360 Controller", 0, XTYPE_XBOX360 },
205         { 0x1bad, 0xf901, "Gamestop Xbox 360 Controller", 0, XTYPE_XBOX360 },
206         { 0x1bad, 0xf903, "Tron Xbox 360 controller", 0, XTYPE_XBOX360 },
207         { 0x24c6, 0x5000, "Razer Atrox Arcade Stick", 0, XTYPE_XBOX360 },
208         { 0x24c6, 0x5300, "PowerA MINI PROEX Controller", 0, XTYPE_XBOX360 },
209         { 0x24c6, 0x5303, "Xbox Airflo wired controller", 0, XTYPE_XBOX360 },
210         { 0x24c6, 0x5500, "Hori XBOX 360 EX 2 with Turbo", 0, XTYPE_XBOX360 },
211         { 0x24c6, 0x5501, "Hori Real Arcade Pro VX-SA", 0, XTYPE_XBOX360 },
212         { 0x24c6, 0x5506, "Hori SOULCALIBUR V Stick", 0, XTYPE_XBOX360 },
213         { 0x24c6, 0x5b02, "Thrustmaster, Inc. GPX Controller", 0, XTYPE_XBOX360 },
214         { 0x24c6, 0x5b03, "Thrustmaster Ferrari 458 Racing Wheel", 0, XTYPE_XBOX360 },
215         { 0xffff, 0xffff, "Chinese-made Xbox Controller", 0, XTYPE_XBOX },
216         { 0x0000, 0x0000, "Generic X-Box pad", 0, XTYPE_UNKNOWN }
217 };
218
219 /* buttons shared with xbox and xbox360 */
220 static const signed short xpad_common_btn[] = {
221         BTN_A, BTN_B, BTN_X, BTN_Y,                     /* "analog" buttons */
222         BTN_START, BTN_SELECT, BTN_THUMBL, BTN_THUMBR,  /* start/back/sticks */
223         -1                                              /* terminating entry */
224 };
225
226 /* original xbox controllers only */
227 static const signed short xpad_btn[] = {
228         BTN_C, BTN_Z,           /* "analog" buttons */
229         -1                      /* terminating entry */
230 };
231
232 /* used when dpad is mapped to buttons */
233 static const signed short xpad_btn_pad[] = {
234         BTN_TRIGGER_HAPPY1, BTN_TRIGGER_HAPPY2,         /* d-pad left, right */
235         BTN_TRIGGER_HAPPY3, BTN_TRIGGER_HAPPY4,         /* d-pad up, down */
236         -1                              /* terminating entry */
237 };
238
239 /* used when triggers are mapped to buttons */
240 static const signed short xpad_btn_triggers[] = {
241         BTN_TL2, BTN_TR2,               /* triggers left/right */
242         -1
243 };
244
245
246 static const signed short xpad360_btn[] = {  /* buttons for x360 controller */
247         BTN_TL, BTN_TR,         /* Button LB/RB */
248         BTN_MODE,               /* The big X button */
249         -1
250 };
251
252 static const signed short xpad_abs[] = {
253         ABS_X, ABS_Y,           /* left stick */
254         ABS_RX, ABS_RY,         /* right stick */
255         -1                      /* terminating entry */
256 };
257
258 /* used when dpad is mapped to axes */
259 static const signed short xpad_abs_pad[] = {
260         ABS_HAT0X, ABS_HAT0Y,   /* d-pad axes */
261         -1                      /* terminating entry */
262 };
263
264 /* used when triggers are mapped to axes */
265 static const signed short xpad_abs_triggers[] = {
266         ABS_Z, ABS_RZ,          /* triggers left/right */
267         -1
268 };
269
270 /*
271  * Xbox 360 has a vendor-specific class, so we cannot match it with only
272  * USB_INTERFACE_INFO (also specifically refused by USB subsystem), so we
273  * match against vendor id as well. Wired Xbox 360 devices have protocol 1,
274  * wireless controllers have protocol 129.
275  */
276 #define XPAD_XBOX360_VENDOR_PROTOCOL(vend,pr) \
277         .match_flags = USB_DEVICE_ID_MATCH_VENDOR | USB_DEVICE_ID_MATCH_INT_INFO, \
278         .idVendor = (vend), \
279         .bInterfaceClass = USB_CLASS_VENDOR_SPEC, \
280         .bInterfaceSubClass = 93, \
281         .bInterfaceProtocol = (pr)
282 #define XPAD_XBOX360_VENDOR(vend) \
283         { XPAD_XBOX360_VENDOR_PROTOCOL(vend,1) }, \
284         { XPAD_XBOX360_VENDOR_PROTOCOL(vend,129) }
285
286 /* The Xbox One controller uses subclass 71 and protocol 208. */
287 #define XPAD_XBOXONE_VENDOR_PROTOCOL(vend, pr) \
288         .match_flags = USB_DEVICE_ID_MATCH_VENDOR | USB_DEVICE_ID_MATCH_INT_INFO, \
289         .idVendor = (vend), \
290         .bInterfaceClass = USB_CLASS_VENDOR_SPEC, \
291         .bInterfaceSubClass = 71, \
292         .bInterfaceProtocol = (pr)
293 #define XPAD_XBOXONE_VENDOR(vend) \
294         { XPAD_XBOXONE_VENDOR_PROTOCOL(vend, 208) }
295
296 static struct usb_device_id xpad_table[] = {
297         { USB_INTERFACE_INFO('X', 'B', 0) },    /* X-Box USB-IF not approved class */
298         XPAD_XBOX360_VENDOR(0x044f),            /* Thrustmaster X-Box 360 controllers */
299         XPAD_XBOX360_VENDOR(0x045e),            /* Microsoft X-Box 360 controllers */
300         XPAD_XBOXONE_VENDOR(0x045e),            /* Microsoft X-Box One controllers */
301         XPAD_XBOX360_VENDOR(0x046d),            /* Logitech X-Box 360 style controllers */
302         XPAD_XBOX360_VENDOR(0x0738),            /* Mad Catz X-Box 360 controllers */
303         { USB_DEVICE(0x0738, 0x4540) },         /* Mad Catz Beat Pad */
304         XPAD_XBOX360_VENDOR(0x0e6f),            /* 0x0e6f X-Box 360 controllers */
305         XPAD_XBOX360_VENDOR(0x12ab),            /* X-Box 360 dance pads */
306         XPAD_XBOX360_VENDOR(0x1430),            /* RedOctane X-Box 360 controllers */
307         XPAD_XBOX360_VENDOR(0x146b),            /* BigBen Interactive Controllers */
308         XPAD_XBOX360_VENDOR(0x1bad),            /* Harminix Rock Band Guitar and Drums */
309         XPAD_XBOX360_VENDOR(0x0f0d),            /* Hori Controllers */
310         XPAD_XBOX360_VENDOR(0x1689),            /* Razer Onza */
311         XPAD_XBOX360_VENDOR(0x24c6),            /* PowerA Controllers */
312         XPAD_XBOX360_VENDOR(0x1532),            /* Razer Sabertooth */
313         XPAD_XBOX360_VENDOR(0x15e4),            /* Numark X-Box 360 controllers */
314         XPAD_XBOX360_VENDOR(0x162e),            /* Joytech X-Box 360 controllers */
315         { }
316 };
317
318 MODULE_DEVICE_TABLE(usb, xpad_table);
319
320 struct usb_xpad {
321         struct input_dev *dev;          /* input device interface */
322         struct usb_device *udev;        /* usb device */
323         struct usb_interface *intf;     /* usb interface */
324
325         int pad_present;
326
327         struct urb *irq_in;             /* urb for interrupt in report */
328         unsigned char *idata;           /* input data */
329         dma_addr_t idata_dma;
330
331         struct urb *bulk_out;
332         unsigned char *bdata;
333
334         struct urb *irq_out;            /* urb for interrupt out report */
335         unsigned char *odata;           /* output data */
336         dma_addr_t odata_dma;
337         struct mutex odata_mutex;
338
339 #if defined(CONFIG_JOYSTICK_XPAD_LEDS)
340         struct xpad_led *led;
341 #endif
342
343         char phys[64];                  /* physical device path */
344
345         int mapping;                    /* map d-pad to buttons or to axes */
346         int xtype;                      /* type of xbox device */
347         unsigned long led_no;           /* led to lit on xbox360 controllers */
348 };
349
350 /*
351  *      xpad_process_packet
352  *
353  *      Completes a request by converting the data into events for the
354  *      input subsystem.
355  *
356  *      The used report descriptor was taken from ITO Takayukis website:
357  *       http://euc.jp/periphs/xbox-controller.ja.html
358  */
359
360 static void xpad_process_packet(struct usb_xpad *xpad, u16 cmd, unsigned char *data)
361 {
362         struct input_dev *dev = xpad->dev;
363
364         if (!(xpad->mapping & MAP_STICKS_TO_NULL)) {
365                 /* left stick */
366                 input_report_abs(dev, ABS_X,
367                                  (__s16) le16_to_cpup((__le16 *)(data + 12)));
368                 input_report_abs(dev, ABS_Y,
369                                  ~(__s16) le16_to_cpup((__le16 *)(data + 14)));
370
371                 /* right stick */
372                 input_report_abs(dev, ABS_RX,
373                                  (__s16) le16_to_cpup((__le16 *)(data + 16)));
374                 input_report_abs(dev, ABS_RY,
375                                  ~(__s16) le16_to_cpup((__le16 *)(data + 18)));
376         }
377
378         /* triggers left/right */
379         if (xpad->mapping & MAP_TRIGGERS_TO_BUTTONS) {
380                 input_report_key(dev, BTN_TL2, data[10]);
381                 input_report_key(dev, BTN_TR2, data[11]);
382         } else {
383                 input_report_abs(dev, ABS_Z, data[10]);
384                 input_report_abs(dev, ABS_RZ, data[11]);
385         }
386
387         /* digital pad */
388         if (xpad->mapping & MAP_DPAD_TO_BUTTONS) {
389                 /* dpad as buttons (left, right, up, down) */
390                 input_report_key(dev, BTN_TRIGGER_HAPPY1, data[2] & 0x04);
391                 input_report_key(dev, BTN_TRIGGER_HAPPY2, data[2] & 0x08);
392                 input_report_key(dev, BTN_TRIGGER_HAPPY3, data[2] & 0x01);
393                 input_report_key(dev, BTN_TRIGGER_HAPPY4, data[2] & 0x02);
394         } else {
395                 input_report_abs(dev, ABS_HAT0X,
396                                  !!(data[2] & 0x08) - !!(data[2] & 0x04));
397                 input_report_abs(dev, ABS_HAT0Y,
398                                  !!(data[2] & 0x02) - !!(data[2] & 0x01));
399         }
400
401         /* start/back buttons and stick press left/right */
402         input_report_key(dev, BTN_START,  data[2] & 0x10);
403         input_report_key(dev, BTN_SELECT, data[2] & 0x20);
404         input_report_key(dev, BTN_THUMBL, data[2] & 0x40);
405         input_report_key(dev, BTN_THUMBR, data[2] & 0x80);
406
407         /* "analog" buttons A, B, X, Y */
408         input_report_key(dev, BTN_A, data[4]);
409         input_report_key(dev, BTN_B, data[5]);
410         input_report_key(dev, BTN_X, data[6]);
411         input_report_key(dev, BTN_Y, data[7]);
412
413         /* "analog" buttons black, white */
414         input_report_key(dev, BTN_C, data[8]);
415         input_report_key(dev, BTN_Z, data[9]);
416
417         input_sync(dev);
418 }
419
420 /*
421  *      xpad360_process_packet
422  *
423  *      Completes a request by converting the data into events for the
424  *      input subsystem. It is version for xbox 360 controller
425  *
426  *      The used report descriptor was taken from:
427  *              http://www.free60.org/wiki/Gamepad
428  */
429
430 static void xpad360_process_packet(struct usb_xpad *xpad,
431                                    u16 cmd, unsigned char *data)
432 {
433         struct input_dev *dev = xpad->dev;
434
435         /* digital pad */
436         if (xpad->mapping & MAP_DPAD_TO_BUTTONS) {
437                 /* dpad as buttons (left, right, up, down) */
438                 input_report_key(dev, BTN_TRIGGER_HAPPY1, data[2] & 0x04);
439                 input_report_key(dev, BTN_TRIGGER_HAPPY2, data[2] & 0x08);
440                 input_report_key(dev, BTN_TRIGGER_HAPPY3, data[2] & 0x01);
441                 input_report_key(dev, BTN_TRIGGER_HAPPY4, data[2] & 0x02);
442         } else {
443                 input_report_abs(dev, ABS_HAT0X,
444                                  !!(data[2] & 0x08) - !!(data[2] & 0x04));
445                 input_report_abs(dev, ABS_HAT0Y,
446                                  !!(data[2] & 0x02) - !!(data[2] & 0x01));
447         }
448
449         /* start/back buttons */
450         input_report_key(dev, BTN_START,  data[2] & 0x10);
451         input_report_key(dev, BTN_SELECT, data[2] & 0x20);
452
453         /* stick press left/right */
454         input_report_key(dev, BTN_THUMBL, data[2] & 0x40);
455         input_report_key(dev, BTN_THUMBR, data[2] & 0x80);
456
457         /* buttons A,B,X,Y,TL,TR and MODE */
458         input_report_key(dev, BTN_A,    data[3] & 0x10);
459         input_report_key(dev, BTN_B,    data[3] & 0x20);
460         input_report_key(dev, BTN_X,    data[3] & 0x40);
461         input_report_key(dev, BTN_Y,    data[3] & 0x80);
462         input_report_key(dev, BTN_TL,   data[3] & 0x01);
463         input_report_key(dev, BTN_TR,   data[3] & 0x02);
464         input_report_key(dev, BTN_MODE, data[3] & 0x04);
465
466         if (!(xpad->mapping & MAP_STICKS_TO_NULL)) {
467                 /* left stick */
468                 input_report_abs(dev, ABS_X,
469                                  (__s16) le16_to_cpup((__le16 *)(data + 6)));
470                 input_report_abs(dev, ABS_Y,
471                                  ~(__s16) le16_to_cpup((__le16 *)(data + 8)));
472
473                 /* right stick */
474                 input_report_abs(dev, ABS_RX,
475                                  (__s16) le16_to_cpup((__le16 *)(data + 10)));
476                 input_report_abs(dev, ABS_RY,
477                                  ~(__s16) le16_to_cpup((__le16 *)(data + 12)));
478         }
479
480         /* triggers left/right */
481         if (xpad->mapping & MAP_TRIGGERS_TO_BUTTONS) {
482                 input_report_key(dev, BTN_TL2, data[4]);
483                 input_report_key(dev, BTN_TR2, data[5]);
484         } else {
485                 input_report_abs(dev, ABS_Z, data[4]);
486                 input_report_abs(dev, ABS_RZ, data[5]);
487         }
488
489         input_sync(dev);
490 }
491
492 static void xpad_identify_controller(struct usb_xpad *xpad);
493
494 /*
495  * xpad360w_process_packet
496  *
497  * Completes a request by converting the data into events for the
498  * input subsystem. It is version for xbox 360 wireless controller.
499  *
500  * Byte.Bit
501  * 00.1 - Status change: The controller or headset has connected/disconnected
502  *                       Bits 01.7 and 01.6 are valid
503  * 01.7 - Controller present
504  * 01.6 - Headset present
505  * 01.1 - Pad state (Bytes 4+) valid
506  *
507  */
508
509 static void xpad360w_process_packet(struct usb_xpad *xpad, u16 cmd, unsigned char *data)
510 {
511         /* Presence change */
512         if (data[0] & 0x08) {
513                 if (data[1] & 0x80) {
514                         xpad->pad_present = 1;
515                         usb_submit_urb(xpad->bulk_out, GFP_ATOMIC);
516                         /*
517                          * Light up the segment corresponding to
518                          * controller number.
519                          */
520                         xpad_identify_controller(xpad);
521                 } else
522                         xpad->pad_present = 0;
523         }
524
525         /* Valid pad data */
526         if (!(data[1] & 0x1))
527                 return;
528
529         xpad360_process_packet(xpad, cmd, &data[4]);
530 }
531
532 /*
533  *      xpadone_process_buttons
534  *
535  *      Process a button update packet from an Xbox one controller.
536  */
537 static void xpadone_process_buttons(struct usb_xpad *xpad,
538                                 struct input_dev *dev,
539                                 unsigned char *data)
540 {
541         /* menu/view buttons */
542         input_report_key(dev, BTN_START,  data[4] & 0x04);
543         input_report_key(dev, BTN_SELECT, data[4] & 0x08);
544
545         /* buttons A,B,X,Y */
546         input_report_key(dev, BTN_A,    data[4] & 0x10);
547         input_report_key(dev, BTN_B,    data[4] & 0x20);
548         input_report_key(dev, BTN_X,    data[4] & 0x40);
549         input_report_key(dev, BTN_Y,    data[4] & 0x80);
550
551         /* digital pad */
552         if (xpad->mapping & MAP_DPAD_TO_BUTTONS) {
553                 /* dpad as buttons (left, right, up, down) */
554                 input_report_key(dev, BTN_TRIGGER_HAPPY1, data[5] & 0x04);
555                 input_report_key(dev, BTN_TRIGGER_HAPPY2, data[5] & 0x08);
556                 input_report_key(dev, BTN_TRIGGER_HAPPY3, data[5] & 0x01);
557                 input_report_key(dev, BTN_TRIGGER_HAPPY4, data[5] & 0x02);
558         } else {
559                 input_report_abs(dev, ABS_HAT0X,
560                                  !!(data[5] & 0x08) - !!(data[5] & 0x04));
561                 input_report_abs(dev, ABS_HAT0Y,
562                                  !!(data[5] & 0x02) - !!(data[5] & 0x01));
563         }
564
565         /* TL/TR */
566         input_report_key(dev, BTN_TL,   data[5] & 0x10);
567         input_report_key(dev, BTN_TR,   data[5] & 0x20);
568
569         /* stick press left/right */
570         input_report_key(dev, BTN_THUMBL, data[5] & 0x40);
571         input_report_key(dev, BTN_THUMBR, data[5] & 0x80);
572
573         if (!(xpad->mapping & MAP_STICKS_TO_NULL)) {
574                 /* left stick */
575                 input_report_abs(dev, ABS_X,
576                                  (__s16) le16_to_cpup((__le16 *)(data + 10)));
577                 input_report_abs(dev, ABS_Y,
578                                  ~(__s16) le16_to_cpup((__le16 *)(data + 12)));
579
580                 /* right stick */
581                 input_report_abs(dev, ABS_RX,
582                                  (__s16) le16_to_cpup((__le16 *)(data + 14)));
583                 input_report_abs(dev, ABS_RY,
584                                  ~(__s16) le16_to_cpup((__le16 *)(data + 16)));
585         }
586
587         /* triggers left/right */
588         if (xpad->mapping & MAP_TRIGGERS_TO_BUTTONS) {
589                 input_report_key(dev, BTN_TL2,
590                                  (__u16) le16_to_cpup((__le16 *)(data + 6)));
591                 input_report_key(dev, BTN_TR2,
592                                  (__u16) le16_to_cpup((__le16 *)(data + 8)));
593         } else {
594                 input_report_abs(dev, ABS_Z,
595                                  (__u16) le16_to_cpup((__le16 *)(data + 6)));
596                 input_report_abs(dev, ABS_RZ,
597                                  (__u16) le16_to_cpup((__le16 *)(data + 8)));
598         }
599
600         input_sync(dev);
601 }
602
603 /*
604  *      xpadone_process_packet
605  *
606  *      Completes a request by converting the data into events for the
607  *      input subsystem. This version is for the Xbox One controller.
608  *
609  *      The report format was gleaned from
610  *      https://github.com/kylelemons/xbox/blob/master/xbox.go
611  */
612
613 static void xpadone_process_packet(struct usb_xpad *xpad,
614                                 u16 cmd, unsigned char *data)
615 {
616         struct input_dev *dev = xpad->dev;
617
618         switch (data[0]) {
619         case 0x20:
620                 xpadone_process_buttons(xpad, dev, data);
621                 break;
622
623         case 0x07:
624                 /* the xbox button has its own special report */
625                 input_report_key(dev, BTN_MODE, data[4] & 0x01);
626                 input_sync(dev);
627                 break;
628         }
629 }
630
631 static void xpad_irq_in(struct urb *urb)
632 {
633         struct usb_xpad *xpad = urb->context;
634         struct device *dev = &xpad->intf->dev;
635         int retval, status;
636
637         status = urb->status;
638
639         switch (status) {
640         case 0:
641                 /* success */
642                 break;
643         case -ECONNRESET:
644         case -ENOENT:
645         case -ESHUTDOWN:
646                 /* this urb is terminated, clean up */
647                 dev_dbg(dev, "%s - urb shutting down with status: %d\n",
648                         __func__, status);
649                 return;
650         default:
651                 dev_dbg(dev, "%s - nonzero urb status received: %d\n",
652                         __func__, status);
653                 goto exit;
654         }
655
656         switch (xpad->xtype) {
657         case XTYPE_XBOX360:
658                 xpad360_process_packet(xpad, 0, xpad->idata);
659                 break;
660         case XTYPE_XBOX360W:
661                 xpad360w_process_packet(xpad, 0, xpad->idata);
662                 break;
663         case XTYPE_XBOXONE:
664                 xpadone_process_packet(xpad, 0, xpad->idata);
665                 break;
666         default:
667                 xpad_process_packet(xpad, 0, xpad->idata);
668         }
669
670 exit:
671         retval = usb_submit_urb(urb, GFP_ATOMIC);
672         if (retval)
673                 dev_err(dev, "%s - usb_submit_urb failed with result %d\n",
674                         __func__, retval);
675 }
676
677 static void xpad_bulk_out(struct urb *urb)
678 {
679         struct usb_xpad *xpad = urb->context;
680         struct device *dev = &xpad->intf->dev;
681
682         switch (urb->status) {
683         case 0:
684                 /* success */
685                 break;
686         case -ECONNRESET:
687         case -ENOENT:
688         case -ESHUTDOWN:
689                 /* this urb is terminated, clean up */
690                 dev_dbg(dev, "%s - urb shutting down with status: %d\n",
691                         __func__, urb->status);
692                 break;
693         default:
694                 dev_dbg(dev, "%s - nonzero urb status received: %d\n",
695                         __func__, urb->status);
696         }
697 }
698
699 static void xpad_irq_out(struct urb *urb)
700 {
701         struct usb_xpad *xpad = urb->context;
702         struct device *dev = &xpad->intf->dev;
703         int retval, status;
704
705         status = urb->status;
706
707         switch (status) {
708         case 0:
709                 /* success */
710                 return;
711
712         case -ECONNRESET:
713         case -ENOENT:
714         case -ESHUTDOWN:
715                 /* this urb is terminated, clean up */
716                 dev_dbg(dev, "%s - urb shutting down with status: %d\n",
717                         __func__, status);
718                 return;
719
720         default:
721                 dev_dbg(dev, "%s - nonzero urb status received: %d\n",
722                         __func__, status);
723                 goto exit;
724         }
725
726 exit:
727         retval = usb_submit_urb(urb, GFP_ATOMIC);
728         if (retval)
729                 dev_err(dev, "%s - usb_submit_urb failed with result %d\n",
730                         __func__, retval);
731 }
732
733 static int xpad_init_output(struct usb_interface *intf, struct usb_xpad *xpad)
734 {
735         struct usb_endpoint_descriptor *ep_irq_out;
736         int ep_irq_out_idx;
737         int error;
738
739         if (xpad->xtype == XTYPE_UNKNOWN)
740                 return 0;
741
742         xpad->odata = usb_alloc_coherent(xpad->udev, XPAD_PKT_LEN,
743                                          GFP_KERNEL, &xpad->odata_dma);
744         if (!xpad->odata) {
745                 error = -ENOMEM;
746                 goto fail1;
747         }
748
749         mutex_init(&xpad->odata_mutex);
750
751         xpad->irq_out = usb_alloc_urb(0, GFP_KERNEL);
752         if (!xpad->irq_out) {
753                 error = -ENOMEM;
754                 goto fail2;
755         }
756
757         /* Xbox One controller has in/out endpoints swapped. */
758         ep_irq_out_idx = xpad->xtype == XTYPE_XBOXONE ? 0 : 1;
759         ep_irq_out = &intf->cur_altsetting->endpoint[ep_irq_out_idx].desc;
760
761         usb_fill_int_urb(xpad->irq_out, xpad->udev,
762                          usb_sndintpipe(xpad->udev, ep_irq_out->bEndpointAddress),
763                          xpad->odata, XPAD_PKT_LEN,
764                          xpad_irq_out, xpad, ep_irq_out->bInterval);
765         xpad->irq_out->transfer_dma = xpad->odata_dma;
766         xpad->irq_out->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
767
768         return 0;
769
770  fail2: usb_free_coherent(xpad->udev, XPAD_PKT_LEN, xpad->odata, xpad->odata_dma);
771  fail1: return error;
772 }
773
774 static void xpad_stop_output(struct usb_xpad *xpad)
775 {
776         if (xpad->xtype != XTYPE_UNKNOWN)
777                 usb_kill_urb(xpad->irq_out);
778 }
779
780 static void xpad_deinit_output(struct usb_xpad *xpad)
781 {
782         if (xpad->xtype != XTYPE_UNKNOWN) {
783                 usb_free_urb(xpad->irq_out);
784                 usb_free_coherent(xpad->udev, XPAD_PKT_LEN,
785                                 xpad->odata, xpad->odata_dma);
786         }
787 }
788
789 #ifdef CONFIG_JOYSTICK_XPAD_FF
790 static int xpad_play_effect(struct input_dev *dev, void *data, struct ff_effect *effect)
791 {
792         struct usb_xpad *xpad = input_get_drvdata(dev);
793
794         if (effect->type == FF_RUMBLE) {
795                 __u16 strong = effect->u.rumble.strong_magnitude;
796                 __u16 weak = effect->u.rumble.weak_magnitude;
797
798                 switch (xpad->xtype) {
799
800                 case XTYPE_XBOX:
801                         xpad->odata[0] = 0x00;
802                         xpad->odata[1] = 0x06;
803                         xpad->odata[2] = 0x00;
804                         xpad->odata[3] = strong / 256;  /* left actuator */
805                         xpad->odata[4] = 0x00;
806                         xpad->odata[5] = weak / 256;    /* right actuator */
807                         xpad->irq_out->transfer_buffer_length = 6;
808
809                         return usb_submit_urb(xpad->irq_out, GFP_ATOMIC);
810
811                 case XTYPE_XBOX360:
812                         xpad->odata[0] = 0x00;
813                         xpad->odata[1] = 0x08;
814                         xpad->odata[2] = 0x00;
815                         xpad->odata[3] = strong / 256;  /* left actuator? */
816                         xpad->odata[4] = weak / 256;    /* right actuator? */
817                         xpad->odata[5] = 0x00;
818                         xpad->odata[6] = 0x00;
819                         xpad->odata[7] = 0x00;
820                         xpad->irq_out->transfer_buffer_length = 8;
821
822                         return usb_submit_urb(xpad->irq_out, GFP_ATOMIC);
823
824                 case XTYPE_XBOX360W:
825                         xpad->odata[0] = 0x00;
826                         xpad->odata[1] = 0x01;
827                         xpad->odata[2] = 0x0F;
828                         xpad->odata[3] = 0xC0;
829                         xpad->odata[4] = 0x00;
830                         xpad->odata[5] = strong / 256;
831                         xpad->odata[6] = weak / 256;
832                         xpad->odata[7] = 0x00;
833                         xpad->odata[8] = 0x00;
834                         xpad->odata[9] = 0x00;
835                         xpad->odata[10] = 0x00;
836                         xpad->odata[11] = 0x00;
837                         xpad->irq_out->transfer_buffer_length = 12;
838
839                         return usb_submit_urb(xpad->irq_out, GFP_ATOMIC);
840
841                 case XTYPE_XBOXONE:
842                         xpad->odata[0] = 0x09; /* activate rumble */
843                         xpad->odata[1] = 0x08;
844                         xpad->odata[2] = 0x00;
845                         xpad->odata[3] = 0x08; /* continuous effect */
846                         xpad->odata[4] = 0x00; /* simple rumble mode */
847                         xpad->odata[5] = 0x03; /* L and R actuator only */
848                         xpad->odata[6] = 0x00; /* TODO: LT actuator */
849                         xpad->odata[7] = 0x00; /* TODO: RT actuator */
850                         xpad->odata[8] = strong / 256;  /* left actuator */
851                         xpad->odata[9] = weak / 256;    /* right actuator */
852                         xpad->odata[10] = 0x80; /* length of pulse */
853                         xpad->odata[11] = 0x00; /* stop period of pulse */
854                         xpad->irq_out->transfer_buffer_length = 12;
855
856                         return usb_submit_urb(xpad->irq_out, GFP_ATOMIC);
857
858                 default:
859                         dev_dbg(&xpad->dev->dev,
860                                 "%s - rumble command sent to unsupported xpad type: %d\n",
861                                 __func__, xpad->xtype);
862                         return -1;
863                 }
864         }
865
866         return 0;
867 }
868
869 static int xpad_init_ff(struct usb_xpad *xpad)
870 {
871         if (xpad->xtype == XTYPE_UNKNOWN)
872                 return 0;
873
874         input_set_capability(xpad->dev, EV_FF, FF_RUMBLE);
875
876         return input_ff_create_memless(xpad->dev, NULL, xpad_play_effect);
877 }
878
879 #else
880 static int xpad_init_ff(struct usb_xpad *xpad) { return 0; }
881 #endif
882
883 #if defined(CONFIG_JOYSTICK_XPAD_LEDS)
884 #include <linux/leds.h>
885
886 struct xpad_led {
887         char name[16];
888         struct led_classdev led_cdev;
889         struct usb_xpad *xpad;
890 };
891
892 /**
893  * @param command
894  *  0: off
895  *  1: all blink, then previous setting
896  *  2: 1/top-left blink, then on
897  *  3: 2/top-right blink, then on
898  *  4: 3/bottom-left blink, then on
899  *  5: 4/bottom-right blink, then on
900  *  6: 1/top-left on
901  *  7: 2/top-right on
902  *  8: 3/bottom-left on
903  *  9: 4/bottom-right on
904  * 10: rotate
905  * 11: blink, based on previous setting
906  * 12: slow blink, based on previous setting
907  * 13: rotate with two lights
908  * 14: persistent slow all blink
909  * 15: blink once, then previous setting
910  */
911 static void xpad_send_led_command(struct usb_xpad *xpad, int command)
912 {
913         command %= 16;
914
915         mutex_lock(&xpad->odata_mutex);
916
917         switch (xpad->xtype) {
918         case XTYPE_XBOX360:
919                 xpad->odata[0] = 0x01;
920                 xpad->odata[1] = 0x03;
921                 xpad->odata[2] = command;
922                 xpad->irq_out->transfer_buffer_length = 3;
923                 break;
924         case XTYPE_XBOX360W:
925                 xpad->odata[0] = 0x00;
926                 xpad->odata[1] = 0x00;
927                 xpad->odata[2] = 0x08;
928                 xpad->odata[3] = 0x40 + command;
929                 xpad->odata[4] = 0x00;
930                 xpad->odata[5] = 0x00;
931                 xpad->odata[6] = 0x00;
932                 xpad->odata[7] = 0x00;
933                 xpad->odata[8] = 0x00;
934                 xpad->odata[9] = 0x00;
935                 xpad->odata[10] = 0x00;
936                 xpad->odata[11] = 0x00;
937                 xpad->irq_out->transfer_buffer_length = 12;
938                 break;
939         }
940
941         usb_submit_urb(xpad->irq_out, GFP_KERNEL);
942         mutex_unlock(&xpad->odata_mutex);
943 }
944
945 static void xpad_identify_controller(struct usb_xpad *xpad)
946 {
947         /* Light up the segment corresponding to controller number */
948         xpad_send_led_command(xpad, (xpad->led_no % 4) + 2);
949 }
950
951 static void xpad_led_set(struct led_classdev *led_cdev,
952                          enum led_brightness value)
953 {
954         struct xpad_led *xpad_led = container_of(led_cdev,
955                                                  struct xpad_led, led_cdev);
956
957         xpad_send_led_command(xpad_led->xpad, value);
958 }
959
960 static int xpad_led_probe(struct usb_xpad *xpad)
961 {
962         static atomic_t led_seq = ATOMIC_INIT(-1);
963         struct xpad_led *led;
964         struct led_classdev *led_cdev;
965         int error;
966
967         if (xpad->xtype != XTYPE_XBOX360 && xpad->xtype != XTYPE_XBOX360W)
968                 return 0;
969
970         xpad->led = led = kzalloc(sizeof(struct xpad_led), GFP_KERNEL);
971         if (!led)
972                 return -ENOMEM;
973
974         xpad->led_no = atomic_inc_return(&led_seq);
975
976         snprintf(led->name, sizeof(led->name), "xpad%lu", xpad->led_no);
977         led->xpad = xpad;
978
979         led_cdev = &led->led_cdev;
980         led_cdev->name = led->name;
981         led_cdev->brightness_set = xpad_led_set;
982
983         error = led_classdev_register(&xpad->udev->dev, led_cdev);
984         if (error) {
985                 kfree(led);
986                 xpad->led = NULL;
987                 return error;
988         }
989
990         /* Light up the segment corresponding to controller number */
991         xpad_identify_controller(xpad);
992
993         return 0;
994 }
995
996 static void xpad_led_disconnect(struct usb_xpad *xpad)
997 {
998         struct xpad_led *xpad_led = xpad->led;
999
1000         if (xpad_led) {
1001                 led_classdev_unregister(&xpad_led->led_cdev);
1002                 kfree(xpad_led);
1003         }
1004 }
1005 #else
1006 static int xpad_led_probe(struct usb_xpad *xpad) { return 0; }
1007 static void xpad_led_disconnect(struct usb_xpad *xpad) { }
1008 static void xpad_identify_controller(struct usb_xpad *xpad) { }
1009 #endif
1010
1011
1012 static int xpad_open(struct input_dev *dev)
1013 {
1014         struct usb_xpad *xpad = input_get_drvdata(dev);
1015
1016         /* URB was submitted in probe */
1017         if (xpad->xtype == XTYPE_XBOX360W)
1018                 return 0;
1019
1020         xpad->irq_in->dev = xpad->udev;
1021         if (usb_submit_urb(xpad->irq_in, GFP_KERNEL))
1022                 return -EIO;
1023
1024         if (xpad->xtype == XTYPE_XBOXONE) {
1025                 /* Xbox one controller needs to be initialized. */
1026                 xpad->odata[0] = 0x05;
1027                 xpad->odata[1] = 0x20;
1028                 xpad->irq_out->transfer_buffer_length = 2;
1029                 return usb_submit_urb(xpad->irq_out, GFP_KERNEL);
1030         }
1031
1032         return 0;
1033 }
1034
1035 static void xpad_close(struct input_dev *dev)
1036 {
1037         struct usb_xpad *xpad = input_get_drvdata(dev);
1038
1039         if (xpad->xtype != XTYPE_XBOX360W)
1040                 usb_kill_urb(xpad->irq_in);
1041
1042         xpad_stop_output(xpad);
1043 }
1044
1045 static void xpad_set_up_abs(struct input_dev *input_dev, signed short abs)
1046 {
1047         struct usb_xpad *xpad = input_get_drvdata(input_dev);
1048         set_bit(abs, input_dev->absbit);
1049
1050         switch (abs) {
1051         case ABS_X:
1052         case ABS_Y:
1053         case ABS_RX:
1054         case ABS_RY:    /* the two sticks */
1055                 input_set_abs_params(input_dev, abs, -32768, 32767, 16, 128);
1056                 break;
1057         case ABS_Z:
1058         case ABS_RZ:    /* the triggers (if mapped to axes) */
1059                 if (xpad->xtype == XTYPE_XBOXONE)
1060                         input_set_abs_params(input_dev, abs, 0, 1023, 0, 0);
1061                 else
1062                         input_set_abs_params(input_dev, abs, 0, 255, 0, 0);
1063                 break;
1064         case ABS_HAT0X:
1065         case ABS_HAT0Y: /* the d-pad (only if dpad is mapped to axes */
1066                 input_set_abs_params(input_dev, abs, -1, 1, 0, 0);
1067                 break;
1068         }
1069 }
1070
1071 static int xpad_probe(struct usb_interface *intf, const struct usb_device_id *id)
1072 {
1073         struct usb_device *udev = interface_to_usbdev(intf);
1074         struct usb_xpad *xpad;
1075         struct input_dev *input_dev;
1076         struct usb_endpoint_descriptor *ep_irq_in;
1077         int ep_irq_in_idx;
1078         int i, error;
1079
1080         for (i = 0; xpad_device[i].idVendor; i++) {
1081                 if ((le16_to_cpu(udev->descriptor.idVendor) == xpad_device[i].idVendor) &&
1082                     (le16_to_cpu(udev->descriptor.idProduct) == xpad_device[i].idProduct))
1083                         break;
1084         }
1085
1086         if (xpad_device[i].xtype == XTYPE_XBOXONE &&
1087             intf->cur_altsetting->desc.bInterfaceNumber != 0) {
1088                 /*
1089                  * The Xbox One controller lists three interfaces all with the
1090                  * same interface class, subclass and protocol. Differentiate by
1091                  * interface number.
1092                  */
1093                 return -ENODEV;
1094         }
1095
1096         xpad = kzalloc(sizeof(struct usb_xpad), GFP_KERNEL);
1097         input_dev = input_allocate_device();
1098         if (!xpad || !input_dev) {
1099                 error = -ENOMEM;
1100                 goto fail1;
1101         }
1102
1103         xpad->idata = usb_alloc_coherent(udev, XPAD_PKT_LEN,
1104                                          GFP_KERNEL, &xpad->idata_dma);
1105         if (!xpad->idata) {
1106                 error = -ENOMEM;
1107                 goto fail1;
1108         }
1109
1110         xpad->irq_in = usb_alloc_urb(0, GFP_KERNEL);
1111         if (!xpad->irq_in) {
1112                 error = -ENOMEM;
1113                 goto fail2;
1114         }
1115
1116         xpad->udev = udev;
1117         xpad->intf = intf;
1118         xpad->mapping = xpad_device[i].mapping;
1119         xpad->xtype = xpad_device[i].xtype;
1120
1121         if (xpad->xtype == XTYPE_UNKNOWN) {
1122                 if (intf->cur_altsetting->desc.bInterfaceClass == USB_CLASS_VENDOR_SPEC) {
1123                         if (intf->cur_altsetting->desc.bInterfaceProtocol == 129)
1124                                 xpad->xtype = XTYPE_XBOX360W;
1125                         else
1126                                 xpad->xtype = XTYPE_XBOX360;
1127                 } else
1128                         xpad->xtype = XTYPE_XBOX;
1129
1130                 if (dpad_to_buttons)
1131                         xpad->mapping |= MAP_DPAD_TO_BUTTONS;
1132                 if (triggers_to_buttons)
1133                         xpad->mapping |= MAP_TRIGGERS_TO_BUTTONS;
1134                 if (sticks_to_null)
1135                         xpad->mapping |= MAP_STICKS_TO_NULL;
1136         }
1137
1138         xpad->dev = input_dev;
1139         usb_make_path(udev, xpad->phys, sizeof(xpad->phys));
1140         strlcat(xpad->phys, "/input0", sizeof(xpad->phys));
1141
1142         input_dev->name = xpad_device[i].name;
1143         input_dev->phys = xpad->phys;
1144         usb_to_input_id(udev, &input_dev->id);
1145         input_dev->dev.parent = &intf->dev;
1146
1147         input_set_drvdata(input_dev, xpad);
1148
1149         input_dev->open = xpad_open;
1150         input_dev->close = xpad_close;
1151
1152         input_dev->evbit[0] = BIT_MASK(EV_KEY);
1153
1154         if (!(xpad->mapping & MAP_STICKS_TO_NULL)) {
1155                 input_dev->evbit[0] |= BIT_MASK(EV_ABS);
1156                 /* set up axes */
1157                 for (i = 0; xpad_abs[i] >= 0; i++)
1158                         xpad_set_up_abs(input_dev, xpad_abs[i]);
1159         }
1160
1161         /* set up standard buttons */
1162         for (i = 0; xpad_common_btn[i] >= 0; i++)
1163                 __set_bit(xpad_common_btn[i], input_dev->keybit);
1164
1165         /* set up model-specific ones */
1166         if (xpad->xtype == XTYPE_XBOX360 || xpad->xtype == XTYPE_XBOX360W ||
1167             xpad->xtype == XTYPE_XBOXONE) {
1168                 for (i = 0; xpad360_btn[i] >= 0; i++)
1169                         __set_bit(xpad360_btn[i], input_dev->keybit);
1170         } else {
1171                 for (i = 0; xpad_btn[i] >= 0; i++)
1172                         __set_bit(xpad_btn[i], input_dev->keybit);
1173         }
1174
1175         if (xpad->mapping & MAP_DPAD_TO_BUTTONS) {
1176                 for (i = 0; xpad_btn_pad[i] >= 0; i++)
1177                         __set_bit(xpad_btn_pad[i], input_dev->keybit);
1178         } else {
1179                 for (i = 0; xpad_abs_pad[i] >= 0; i++)
1180                         xpad_set_up_abs(input_dev, xpad_abs_pad[i]);
1181         }
1182
1183         if (xpad->mapping & MAP_TRIGGERS_TO_BUTTONS) {
1184                 for (i = 0; xpad_btn_triggers[i] >= 0; i++)
1185                         __set_bit(xpad_btn_triggers[i], input_dev->keybit);
1186         } else {
1187                 for (i = 0; xpad_abs_triggers[i] >= 0; i++)
1188                         xpad_set_up_abs(input_dev, xpad_abs_triggers[i]);
1189         }
1190
1191         error = xpad_init_output(intf, xpad);
1192         if (error)
1193                 goto fail3;
1194
1195         error = xpad_init_ff(xpad);
1196         if (error)
1197                 goto fail4;
1198
1199         error = xpad_led_probe(xpad);
1200         if (error)
1201                 goto fail5;
1202
1203         /* Xbox One controller has in/out endpoints swapped. */
1204         ep_irq_in_idx = xpad->xtype == XTYPE_XBOXONE ? 1 : 0;
1205         ep_irq_in = &intf->cur_altsetting->endpoint[ep_irq_in_idx].desc;
1206
1207         usb_fill_int_urb(xpad->irq_in, udev,
1208                          usb_rcvintpipe(udev, ep_irq_in->bEndpointAddress),
1209                          xpad->idata, XPAD_PKT_LEN, xpad_irq_in,
1210                          xpad, ep_irq_in->bInterval);
1211         xpad->irq_in->transfer_dma = xpad->idata_dma;
1212         xpad->irq_in->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
1213
1214         error = input_register_device(xpad->dev);
1215         if (error)
1216                 goto fail6;
1217
1218         usb_set_intfdata(intf, xpad);
1219
1220         if (xpad->xtype == XTYPE_XBOX360W) {
1221                 /*
1222                  * Setup the message to set the LEDs on the
1223                  * controller when it shows up
1224                  */
1225                 xpad->bulk_out = usb_alloc_urb(0, GFP_KERNEL);
1226                 if (!xpad->bulk_out) {
1227                         error = -ENOMEM;
1228                         goto fail7;
1229                 }
1230
1231                 xpad->bdata = kzalloc(XPAD_PKT_LEN, GFP_KERNEL);
1232                 if (!xpad->bdata) {
1233                         error = -ENOMEM;
1234                         goto fail8;
1235                 }
1236
1237                 xpad->bdata[2] = 0x08;
1238                 switch (intf->cur_altsetting->desc.bInterfaceNumber) {
1239                 case 0:
1240                         xpad->bdata[3] = 0x42;
1241                         break;
1242                 case 2:
1243                         xpad->bdata[3] = 0x43;
1244                         break;
1245                 case 4:
1246                         xpad->bdata[3] = 0x44;
1247                         break;
1248                 case 6:
1249                         xpad->bdata[3] = 0x45;
1250                 }
1251
1252                 ep_irq_in = &intf->cur_altsetting->endpoint[1].desc;
1253                 if (usb_endpoint_is_bulk_out(ep_irq_in)) {
1254                         usb_fill_bulk_urb(xpad->bulk_out, udev,
1255                                           usb_sndbulkpipe(udev,
1256                                                           ep_irq_in->bEndpointAddress),
1257                                           xpad->bdata, XPAD_PKT_LEN,
1258                                           xpad_bulk_out, xpad);
1259                 } else {
1260                         usb_fill_int_urb(xpad->bulk_out, udev,
1261                                          usb_sndintpipe(udev,
1262                                                         ep_irq_in->bEndpointAddress),
1263                                          xpad->bdata, XPAD_PKT_LEN,
1264                                          xpad_bulk_out, xpad, 0);
1265                 }
1266
1267                 /*
1268                  * Submit the int URB immediately rather than waiting for open
1269                  * because we get status messages from the device whether
1270                  * or not any controllers are attached.  In fact, it's
1271                  * exactly the message that a controller has arrived that
1272                  * we're waiting for.
1273                  */
1274                 xpad->irq_in->dev = xpad->udev;
1275                 error = usb_submit_urb(xpad->irq_in, GFP_KERNEL);
1276                 if (error)
1277                         goto fail9;
1278         }
1279
1280         return 0;
1281
1282  fail9: kfree(xpad->bdata);
1283  fail8: usb_free_urb(xpad->bulk_out);
1284  fail7: input_unregister_device(input_dev);
1285         input_dev = NULL;
1286  fail6: xpad_led_disconnect(xpad);
1287  fail5: if (input_dev)
1288                 input_ff_destroy(input_dev);
1289  fail4: xpad_deinit_output(xpad);
1290  fail3: usb_free_urb(xpad->irq_in);
1291  fail2: usb_free_coherent(udev, XPAD_PKT_LEN, xpad->idata, xpad->idata_dma);
1292  fail1: input_free_device(input_dev);
1293         kfree(xpad);
1294         return error;
1295
1296 }
1297
1298 static void xpad_disconnect(struct usb_interface *intf)
1299 {
1300         struct usb_xpad *xpad = usb_get_intfdata (intf);
1301
1302         xpad_led_disconnect(xpad);
1303         input_unregister_device(xpad->dev);
1304         xpad_deinit_output(xpad);
1305
1306         if (xpad->xtype == XTYPE_XBOX360W) {
1307                 usb_kill_urb(xpad->bulk_out);
1308                 usb_free_urb(xpad->bulk_out);
1309                 usb_kill_urb(xpad->irq_in);
1310         }
1311
1312         usb_free_urb(xpad->irq_in);
1313         usb_free_coherent(xpad->udev, XPAD_PKT_LEN,
1314                         xpad->idata, xpad->idata_dma);
1315
1316         kfree(xpad->bdata);
1317         kfree(xpad);
1318
1319         usb_set_intfdata(intf, NULL);
1320 }
1321
1322 static struct usb_driver xpad_driver = {
1323         .name           = "xpad",
1324         .probe          = xpad_probe,
1325         .disconnect     = xpad_disconnect,
1326         .id_table       = xpad_table,
1327 };
1328
1329 module_usb_driver(xpad_driver);
1330
1331 MODULE_AUTHOR(DRIVER_AUTHOR);
1332 MODULE_DESCRIPTION(DRIVER_DESC);
1333 MODULE_LICENSE("GPL");