These changes are the raw update to linux-4.4.6-rt14. Kernel sources
[kvmfornfv.git] / kernel / 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, 0x02dd, "Microsoft X-Box One pad (Covert Forces)", 0, XTYPE_XBOXONE },
129         { 0x045e, 0x0291, "Xbox 360 Wireless Receiver (XBOX)", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX360W },
130         { 0x045e, 0x0719, "Xbox 360 Wireless Receiver", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX360W },
131         { 0x044f, 0x0f07, "Thrustmaster, Inc. Controller", 0, XTYPE_XBOX },
132         { 0x044f, 0xb326, "Thrustmaster Gamepad GP XID", 0, XTYPE_XBOX360 },
133         { 0x046d, 0xc21d, "Logitech Gamepad F310", 0, XTYPE_XBOX360 },
134         { 0x046d, 0xc21e, "Logitech Gamepad F510", 0, XTYPE_XBOX360 },
135         { 0x046d, 0xc21f, "Logitech Gamepad F710", 0, XTYPE_XBOX360 },
136         { 0x046d, 0xc242, "Logitech Chillstream Controller", 0, XTYPE_XBOX360 },
137         { 0x046d, 0xca84, "Logitech Xbox Cordless Controller", 0, XTYPE_XBOX },
138         { 0x046d, 0xca88, "Logitech Compact Controller for Xbox", 0, XTYPE_XBOX },
139         { 0x05fd, 0x1007, "Mad Catz Controller (unverified)", 0, XTYPE_XBOX },
140         { 0x05fd, 0x107a, "InterAct 'PowerPad Pro' X-Box pad (Germany)", 0, XTYPE_XBOX },
141         { 0x0738, 0x4516, "Mad Catz Control Pad", 0, XTYPE_XBOX },
142         { 0x0738, 0x4522, "Mad Catz LumiCON", 0, XTYPE_XBOX },
143         { 0x0738, 0x4526, "Mad Catz Control Pad Pro", 0, XTYPE_XBOX },
144         { 0x0738, 0x4536, "Mad Catz MicroCON", 0, XTYPE_XBOX },
145         { 0x0738, 0x4540, "Mad Catz Beat Pad", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX },
146         { 0x0738, 0x4556, "Mad Catz Lynx Wireless Controller", 0, XTYPE_XBOX },
147         { 0x0738, 0x4716, "Mad Catz Wired Xbox 360 Controller", 0, XTYPE_XBOX360 },
148         { 0x0738, 0x4718, "Mad Catz Street Fighter IV FightStick SE", 0, XTYPE_XBOX360 },
149         { 0x0738, 0x4726, "Mad Catz Xbox 360 Controller", 0, XTYPE_XBOX360 },
150         { 0x0738, 0x4728, "Mad Catz Street Fighter IV FightPad", MAP_TRIGGERS_TO_BUTTONS, XTYPE_XBOX360 },
151         { 0x0738, 0x4738, "Mad Catz Wired Xbox 360 Controller (SFIV)", MAP_TRIGGERS_TO_BUTTONS, XTYPE_XBOX360 },
152         { 0x0738, 0x4740, "Mad Catz Beat Pad", 0, XTYPE_XBOX360 },
153         { 0x0738, 0x6040, "Mad Catz Beat Pad Pro", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX },
154         { 0x0738, 0xb726, "Mad Catz Xbox controller - MW2", 0, XTYPE_XBOX360 },
155         { 0x0738, 0xbeef, "Mad Catz JOYTECH NEO SE Advanced GamePad", XTYPE_XBOX360 },
156         { 0x0738, 0xcb02, "Saitek Cyborg Rumble Pad - PC/Xbox 360", 0, XTYPE_XBOX360 },
157         { 0x0738, 0xcb03, "Saitek P3200 Rumble Pad - PC/Xbox 360", 0, XTYPE_XBOX360 },
158         { 0x0738, 0xf738, "Super SFIV FightStick TE S", 0, XTYPE_XBOX360 },
159         { 0x0c12, 0x8802, "Zeroplus Xbox Controller", 0, XTYPE_XBOX },
160         { 0x0c12, 0x8809, "RedOctane Xbox Dance Pad", DANCEPAD_MAP_CONFIG, XTYPE_XBOX },
161         { 0x0c12, 0x880a, "Pelican Eclipse PL-2023", 0, XTYPE_XBOX },
162         { 0x0c12, 0x8810, "Zeroplus Xbox Controller", 0, XTYPE_XBOX },
163         { 0x0c12, 0x9902, "HAMA VibraX - *FAULTY HARDWARE*", 0, XTYPE_XBOX },
164         { 0x0d2f, 0x0002, "Andamiro Pump It Up pad", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX },
165         { 0x0e4c, 0x1097, "Radica Gamester Controller", 0, XTYPE_XBOX },
166         { 0x0e4c, 0x2390, "Radica Games Jtech Controller", 0, XTYPE_XBOX },
167         { 0x0e6f, 0x0003, "Logic3 Freebird wireless Controller", 0, XTYPE_XBOX },
168         { 0x0e6f, 0x0005, "Eclipse wireless Controller", 0, XTYPE_XBOX },
169         { 0x0e6f, 0x0006, "Edge wireless Controller", 0, XTYPE_XBOX },
170         { 0x0e6f, 0x0105, "HSM3 Xbox360 dancepad", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX360 },
171         { 0x0e6f, 0x0113, "Afterglow AX.1 Gamepad for Xbox 360", 0, XTYPE_XBOX360 },
172         { 0x0e6f, 0x0201, "Pelican PL-3601 'TSZ' Wired Xbox 360 Controller", 0, XTYPE_XBOX360 },
173         { 0x0e6f, 0x0213, "Afterglow Gamepad for Xbox 360", 0, XTYPE_XBOX360 },
174         { 0x0e6f, 0x021f, "Rock Candy Gamepad for Xbox 360", 0, XTYPE_XBOX360 },
175         { 0x0e6f, 0x0301, "Logic3 Controller", 0, XTYPE_XBOX360 },
176         { 0x0e6f, 0x0401, "Logic3 Controller", 0, XTYPE_XBOX360 },
177         { 0x0e8f, 0x0201, "SmartJoy Frag Xpad/PS2 adaptor", 0, XTYPE_XBOX },
178         { 0x0e8f, 0x3008, "Generic xbox control (dealextreme)", 0, XTYPE_XBOX },
179         { 0x0f0d, 0x000a, "Hori Co. DOA4 FightStick", 0, XTYPE_XBOX360 },
180         { 0x0f0d, 0x000d, "Hori Fighting Stick EX2", MAP_TRIGGERS_TO_BUTTONS, XTYPE_XBOX360 },
181         { 0x0f0d, 0x0016, "Hori Real Arcade Pro.EX", MAP_TRIGGERS_TO_BUTTONS, XTYPE_XBOX360 },
182         { 0x0f30, 0x0202, "Joytech Advanced Controller", 0, XTYPE_XBOX },
183         { 0x0f30, 0x8888, "BigBen XBMiniPad Controller", 0, XTYPE_XBOX },
184         { 0x102c, 0xff0c, "Joytech Wireless Advanced Controller", 0, XTYPE_XBOX },
185         { 0x12ab, 0x0004, "Honey Bee Xbox360 dancepad", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX360 },
186         { 0x12ab, 0x0301, "PDP AFTERGLOW AX.1", 0, XTYPE_XBOX360 },
187         { 0x12ab, 0x8809, "Xbox DDR dancepad", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX },
188         { 0x1430, 0x4748, "RedOctane Guitar Hero X-plorer", 0, XTYPE_XBOX360 },
189         { 0x1430, 0x8888, "TX6500+ Dance Pad (first generation)", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX },
190         { 0x146b, 0x0601, "BigBen Interactive XBOX 360 Controller", 0, XTYPE_XBOX360 },
191         { 0x1532, 0x0037, "Razer Sabertooth", 0, XTYPE_XBOX360 },
192         { 0x15e4, 0x3f00, "Power A Mini Pro Elite", 0, XTYPE_XBOX360 },
193         { 0x15e4, 0x3f0a, "Xbox Airflo wired controller", 0, XTYPE_XBOX360 },
194         { 0x15e4, 0x3f10, "Batarang Xbox 360 controller", 0, XTYPE_XBOX360 },
195         { 0x162e, 0xbeef, "Joytech Neo-Se Take2", 0, XTYPE_XBOX360 },
196         { 0x1689, 0xfd00, "Razer Onza Tournament Edition", 0, XTYPE_XBOX360 },
197         { 0x1689, 0xfd01, "Razer Onza Classic Edition", 0, XTYPE_XBOX360 },
198         { 0x24c6, 0x5d04, "Razer Sabertooth", 0, XTYPE_XBOX360 },
199         { 0x1bad, 0x0002, "Harmonix Rock Band Guitar", 0, XTYPE_XBOX360 },
200         { 0x1bad, 0x0003, "Harmonix Rock Band Drumkit", MAP_DPAD_TO_BUTTONS, XTYPE_XBOX360 },
201         { 0x1bad, 0xf016, "Mad Catz Xbox 360 Controller", 0, XTYPE_XBOX360 },
202         { 0x1bad, 0xf023, "MLG Pro Circuit Controller (Xbox)", 0, XTYPE_XBOX360 },
203         { 0x1bad, 0xf028, "Street Fighter IV FightPad", 0, XTYPE_XBOX360 },
204         { 0x1bad, 0xf038, "Street Fighter IV FightStick TE", 0, XTYPE_XBOX360 },
205         { 0x1bad, 0xf900, "Harmonix Xbox 360 Controller", 0, XTYPE_XBOX360 },
206         { 0x1bad, 0xf901, "Gamestop Xbox 360 Controller", 0, XTYPE_XBOX360 },
207         { 0x1bad, 0xf903, "Tron Xbox 360 controller", 0, XTYPE_XBOX360 },
208         { 0x24c6, 0x5000, "Razer Atrox Arcade Stick", MAP_TRIGGERS_TO_BUTTONS, XTYPE_XBOX360 },
209         { 0x24c6, 0x5300, "PowerA MINI PROEX Controller", 0, XTYPE_XBOX360 },
210         { 0x24c6, 0x5303, "Xbox Airflo wired controller", 0, XTYPE_XBOX360 },
211         { 0x24c6, 0x5500, "Hori XBOX 360 EX 2 with Turbo", 0, XTYPE_XBOX360 },
212         { 0x24c6, 0x5501, "Hori Real Arcade Pro VX-SA", 0, XTYPE_XBOX360 },
213         { 0x24c6, 0x5506, "Hori SOULCALIBUR V Stick", 0, XTYPE_XBOX360 },
214         { 0x24c6, 0x5b02, "Thrustmaster, Inc. GPX Controller", 0, XTYPE_XBOX360 },
215         { 0x24c6, 0x5b03, "Thrustmaster Ferrari 458 Racing Wheel", 0, XTYPE_XBOX360 },
216         { 0xffff, 0xffff, "Chinese-made Xbox Controller", 0, XTYPE_XBOX },
217         { 0x0000, 0x0000, "Generic X-Box pad", 0, XTYPE_UNKNOWN }
218 };
219
220 /* buttons shared with xbox and xbox360 */
221 static const signed short xpad_common_btn[] = {
222         BTN_A, BTN_B, BTN_X, BTN_Y,                     /* "analog" buttons */
223         BTN_START, BTN_SELECT, BTN_THUMBL, BTN_THUMBR,  /* start/back/sticks */
224         -1                                              /* terminating entry */
225 };
226
227 /* original xbox controllers only */
228 static const signed short xpad_btn[] = {
229         BTN_C, BTN_Z,           /* "analog" buttons */
230         -1                      /* terminating entry */
231 };
232
233 /* used when dpad is mapped to buttons */
234 static const signed short xpad_btn_pad[] = {
235         BTN_TRIGGER_HAPPY1, BTN_TRIGGER_HAPPY2,         /* d-pad left, right */
236         BTN_TRIGGER_HAPPY3, BTN_TRIGGER_HAPPY4,         /* d-pad up, down */
237         -1                              /* terminating entry */
238 };
239
240 /* used when triggers are mapped to buttons */
241 static const signed short xpad_btn_triggers[] = {
242         BTN_TL2, BTN_TR2,               /* triggers left/right */
243         -1
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 *irq_out;            /* urb for interrupt out report */
332         unsigned char *odata;           /* output data */
333         dma_addr_t odata_dma;
334         struct mutex odata_mutex;
335
336 #if defined(CONFIG_JOYSTICK_XPAD_LEDS)
337         struct xpad_led *led;
338 #endif
339
340         char phys[64];                  /* physical device path */
341
342         int mapping;                    /* map d-pad to buttons or to axes */
343         int xtype;                      /* type of xbox device */
344         int pad_nr;                     /* the order x360 pads were attached */
345         const char *name;               /* name of the device */
346 };
347
348 /*
349  *      xpad_process_packet
350  *
351  *      Completes a request by converting the data into events for the
352  *      input subsystem.
353  *
354  *      The used report descriptor was taken from ITO Takayukis website:
355  *       http://euc.jp/periphs/xbox-controller.ja.html
356  */
357 static void xpad_process_packet(struct usb_xpad *xpad, u16 cmd, unsigned char *data)
358 {
359         struct input_dev *dev = xpad->dev;
360
361         if (!(xpad->mapping & MAP_STICKS_TO_NULL)) {
362                 /* left stick */
363                 input_report_abs(dev, ABS_X,
364                                  (__s16) le16_to_cpup((__le16 *)(data + 12)));
365                 input_report_abs(dev, ABS_Y,
366                                  ~(__s16) le16_to_cpup((__le16 *)(data + 14)));
367
368                 /* right stick */
369                 input_report_abs(dev, ABS_RX,
370                                  (__s16) le16_to_cpup((__le16 *)(data + 16)));
371                 input_report_abs(dev, ABS_RY,
372                                  ~(__s16) le16_to_cpup((__le16 *)(data + 18)));
373         }
374
375         /* triggers left/right */
376         if (xpad->mapping & MAP_TRIGGERS_TO_BUTTONS) {
377                 input_report_key(dev, BTN_TL2, data[10]);
378                 input_report_key(dev, BTN_TR2, data[11]);
379         } else {
380                 input_report_abs(dev, ABS_Z, data[10]);
381                 input_report_abs(dev, ABS_RZ, data[11]);
382         }
383
384         /* digital pad */
385         if (xpad->mapping & MAP_DPAD_TO_BUTTONS) {
386                 /* dpad as buttons (left, right, up, down) */
387                 input_report_key(dev, BTN_TRIGGER_HAPPY1, data[2] & 0x04);
388                 input_report_key(dev, BTN_TRIGGER_HAPPY2, data[2] & 0x08);
389                 input_report_key(dev, BTN_TRIGGER_HAPPY3, data[2] & 0x01);
390                 input_report_key(dev, BTN_TRIGGER_HAPPY4, data[2] & 0x02);
391         } else {
392                 input_report_abs(dev, ABS_HAT0X,
393                                  !!(data[2] & 0x08) - !!(data[2] & 0x04));
394                 input_report_abs(dev, ABS_HAT0Y,
395                                  !!(data[2] & 0x02) - !!(data[2] & 0x01));
396         }
397
398         /* start/back buttons and stick press left/right */
399         input_report_key(dev, BTN_START,  data[2] & 0x10);
400         input_report_key(dev, BTN_SELECT, data[2] & 0x20);
401         input_report_key(dev, BTN_THUMBL, data[2] & 0x40);
402         input_report_key(dev, BTN_THUMBR, data[2] & 0x80);
403
404         /* "analog" buttons A, B, X, Y */
405         input_report_key(dev, BTN_A, data[4]);
406         input_report_key(dev, BTN_B, data[5]);
407         input_report_key(dev, BTN_X, data[6]);
408         input_report_key(dev, BTN_Y, data[7]);
409
410         /* "analog" buttons black, white */
411         input_report_key(dev, BTN_C, data[8]);
412         input_report_key(dev, BTN_Z, data[9]);
413
414         input_sync(dev);
415 }
416
417 /*
418  *      xpad360_process_packet
419  *
420  *      Completes a request by converting the data into events for the
421  *      input subsystem. It is version for xbox 360 controller
422  *
423  *      The used report descriptor was taken from:
424  *              http://www.free60.org/wiki/Gamepad
425  */
426
427 static void xpad360_process_packet(struct usb_xpad *xpad,
428                                    u16 cmd, unsigned char *data)
429 {
430         struct input_dev *dev = xpad->dev;
431
432         /* digital pad */
433         if (xpad->mapping & MAP_DPAD_TO_BUTTONS) {
434                 /* dpad as buttons (left, right, up, down) */
435                 input_report_key(dev, BTN_TRIGGER_HAPPY1, data[2] & 0x04);
436                 input_report_key(dev, BTN_TRIGGER_HAPPY2, data[2] & 0x08);
437                 input_report_key(dev, BTN_TRIGGER_HAPPY3, data[2] & 0x01);
438                 input_report_key(dev, BTN_TRIGGER_HAPPY4, data[2] & 0x02);
439         }
440
441         /*
442          * This should be a simple else block. However historically
443          * xbox360w has mapped DPAD to buttons while xbox360 did not. This
444          * made no sense, but now we can not just switch back and have to
445          * support both behaviors.
446          */
447         if (!(xpad->mapping & MAP_DPAD_TO_BUTTONS) ||
448             xpad->xtype == XTYPE_XBOX360W) {
449                 input_report_abs(dev, ABS_HAT0X,
450                                  !!(data[2] & 0x08) - !!(data[2] & 0x04));
451                 input_report_abs(dev, ABS_HAT0Y,
452                                  !!(data[2] & 0x02) - !!(data[2] & 0x01));
453         }
454
455         /* start/back buttons */
456         input_report_key(dev, BTN_START,  data[2] & 0x10);
457         input_report_key(dev, BTN_SELECT, data[2] & 0x20);
458
459         /* stick press left/right */
460         input_report_key(dev, BTN_THUMBL, data[2] & 0x40);
461         input_report_key(dev, BTN_THUMBR, data[2] & 0x80);
462
463         /* buttons A,B,X,Y,TL,TR and MODE */
464         input_report_key(dev, BTN_A,    data[3] & 0x10);
465         input_report_key(dev, BTN_B,    data[3] & 0x20);
466         input_report_key(dev, BTN_X,    data[3] & 0x40);
467         input_report_key(dev, BTN_Y,    data[3] & 0x80);
468         input_report_key(dev, BTN_TL,   data[3] & 0x01);
469         input_report_key(dev, BTN_TR,   data[3] & 0x02);
470         input_report_key(dev, BTN_MODE, data[3] & 0x04);
471
472         if (!(xpad->mapping & MAP_STICKS_TO_NULL)) {
473                 /* left stick */
474                 input_report_abs(dev, ABS_X,
475                                  (__s16) le16_to_cpup((__le16 *)(data + 6)));
476                 input_report_abs(dev, ABS_Y,
477                                  ~(__s16) le16_to_cpup((__le16 *)(data + 8)));
478
479                 /* right stick */
480                 input_report_abs(dev, ABS_RX,
481                                  (__s16) le16_to_cpup((__le16 *)(data + 10)));
482                 input_report_abs(dev, ABS_RY,
483                                  ~(__s16) le16_to_cpup((__le16 *)(data + 12)));
484         }
485
486         /* triggers left/right */
487         if (xpad->mapping & MAP_TRIGGERS_TO_BUTTONS) {
488                 input_report_key(dev, BTN_TL2, data[4]);
489                 input_report_key(dev, BTN_TR2, data[5]);
490         } else {
491                 input_report_abs(dev, ABS_Z, data[4]);
492                 input_report_abs(dev, ABS_RZ, data[5]);
493         }
494
495         input_sync(dev);
496 }
497
498 static void xpad_identify_controller(struct usb_xpad *xpad);
499
500 /*
501  * xpad360w_process_packet
502  *
503  * Completes a request by converting the data into events for the
504  * input subsystem. It is version for xbox 360 wireless controller.
505  *
506  * Byte.Bit
507  * 00.1 - Status change: The controller or headset has connected/disconnected
508  *                       Bits 01.7 and 01.6 are valid
509  * 01.7 - Controller present
510  * 01.6 - Headset present
511  * 01.1 - Pad state (Bytes 4+) valid
512  *
513  */
514 static void xpad360w_process_packet(struct usb_xpad *xpad, u16 cmd, unsigned char *data)
515 {
516         /* Presence change */
517         if (data[0] & 0x08) {
518                 if (data[1] & 0x80) {
519                         xpad->pad_present = 1;
520                         /*
521                          * Light up the segment corresponding to
522                          * controller number.
523                          */
524                         xpad_identify_controller(xpad);
525                 } else
526                         xpad->pad_present = 0;
527         }
528
529         /* Valid pad data */
530         if (!(data[1] & 0x1))
531                 return;
532
533         xpad360_process_packet(xpad, cmd, &data[4]);
534 }
535
536 /*
537  *      xpadone_process_buttons
538  *
539  *      Process a button update packet from an Xbox one controller.
540  */
541 static void xpadone_process_buttons(struct usb_xpad *xpad,
542                                 struct input_dev *dev,
543                                 unsigned char *data)
544 {
545         /* menu/view buttons */
546         input_report_key(dev, BTN_START,  data[4] & 0x04);
547         input_report_key(dev, BTN_SELECT, data[4] & 0x08);
548
549         /* buttons A,B,X,Y */
550         input_report_key(dev, BTN_A,    data[4] & 0x10);
551         input_report_key(dev, BTN_B,    data[4] & 0x20);
552         input_report_key(dev, BTN_X,    data[4] & 0x40);
553         input_report_key(dev, BTN_Y,    data[4] & 0x80);
554
555         /* digital pad */
556         if (xpad->mapping & MAP_DPAD_TO_BUTTONS) {
557                 /* dpad as buttons (left, right, up, down) */
558                 input_report_key(dev, BTN_TRIGGER_HAPPY1, data[5] & 0x04);
559                 input_report_key(dev, BTN_TRIGGER_HAPPY2, data[5] & 0x08);
560                 input_report_key(dev, BTN_TRIGGER_HAPPY3, data[5] & 0x01);
561                 input_report_key(dev, BTN_TRIGGER_HAPPY4, data[5] & 0x02);
562         } else {
563                 input_report_abs(dev, ABS_HAT0X,
564                                  !!(data[5] & 0x08) - !!(data[5] & 0x04));
565                 input_report_abs(dev, ABS_HAT0Y,
566                                  !!(data[5] & 0x02) - !!(data[5] & 0x01));
567         }
568
569         /* TL/TR */
570         input_report_key(dev, BTN_TL,   data[5] & 0x10);
571         input_report_key(dev, BTN_TR,   data[5] & 0x20);
572
573         /* stick press left/right */
574         input_report_key(dev, BTN_THUMBL, data[5] & 0x40);
575         input_report_key(dev, BTN_THUMBR, data[5] & 0x80);
576
577         if (!(xpad->mapping & MAP_STICKS_TO_NULL)) {
578                 /* left stick */
579                 input_report_abs(dev, ABS_X,
580                                  (__s16) le16_to_cpup((__le16 *)(data + 10)));
581                 input_report_abs(dev, ABS_Y,
582                                  ~(__s16) le16_to_cpup((__le16 *)(data + 12)));
583
584                 /* right stick */
585                 input_report_abs(dev, ABS_RX,
586                                  (__s16) le16_to_cpup((__le16 *)(data + 14)));
587                 input_report_abs(dev, ABS_RY,
588                                  ~(__s16) le16_to_cpup((__le16 *)(data + 16)));
589         }
590
591         /* triggers left/right */
592         if (xpad->mapping & MAP_TRIGGERS_TO_BUTTONS) {
593                 input_report_key(dev, BTN_TL2,
594                                  (__u16) le16_to_cpup((__le16 *)(data + 6)));
595                 input_report_key(dev, BTN_TR2,
596                                  (__u16) le16_to_cpup((__le16 *)(data + 8)));
597         } else {
598                 input_report_abs(dev, ABS_Z,
599                                  (__u16) le16_to_cpup((__le16 *)(data + 6)));
600                 input_report_abs(dev, ABS_RZ,
601                                  (__u16) le16_to_cpup((__le16 *)(data + 8)));
602         }
603
604         input_sync(dev);
605 }
606
607 /*
608  *      xpadone_process_packet
609  *
610  *      Completes a request by converting the data into events for the
611  *      input subsystem. This version is for the Xbox One controller.
612  *
613  *      The report format was gleaned from
614  *      https://github.com/kylelemons/xbox/blob/master/xbox.go
615  */
616
617 static void xpadone_process_packet(struct usb_xpad *xpad,
618                                 u16 cmd, unsigned char *data)
619 {
620         struct input_dev *dev = xpad->dev;
621
622         switch (data[0]) {
623         case 0x20:
624                 xpadone_process_buttons(xpad, dev, data);
625                 break;
626
627         case 0x07:
628                 /* the xbox button has its own special report */
629                 input_report_key(dev, BTN_MODE, data[4] & 0x01);
630                 input_sync(dev);
631                 break;
632         }
633 }
634
635 static void xpad_irq_in(struct urb *urb)
636 {
637         struct usb_xpad *xpad = urb->context;
638         struct device *dev = &xpad->intf->dev;
639         int retval, status;
640
641         status = urb->status;
642
643         switch (status) {
644         case 0:
645                 /* success */
646                 break;
647         case -ECONNRESET:
648         case -ENOENT:
649         case -ESHUTDOWN:
650                 /* this urb is terminated, clean up */
651                 dev_dbg(dev, "%s - urb shutting down with status: %d\n",
652                         __func__, status);
653                 return;
654         default:
655                 dev_dbg(dev, "%s - nonzero urb status received: %d\n",
656                         __func__, status);
657                 goto exit;
658         }
659
660         switch (xpad->xtype) {
661         case XTYPE_XBOX360:
662                 xpad360_process_packet(xpad, 0, xpad->idata);
663                 break;
664         case XTYPE_XBOX360W:
665                 xpad360w_process_packet(xpad, 0, xpad->idata);
666                 break;
667         case XTYPE_XBOXONE:
668                 xpadone_process_packet(xpad, 0, xpad->idata);
669                 break;
670         default:
671                 xpad_process_packet(xpad, 0, xpad->idata);
672         }
673
674 exit:
675         retval = usb_submit_urb(urb, GFP_ATOMIC);
676         if (retval)
677                 dev_err(dev, "%s - usb_submit_urb failed with result %d\n",
678                         __func__, retval);
679 }
680
681 static void xpad_irq_out(struct urb *urb)
682 {
683         struct usb_xpad *xpad = urb->context;
684         struct device *dev = &xpad->intf->dev;
685         int retval, status;
686
687         status = urb->status;
688
689         switch (status) {
690         case 0:
691                 /* success */
692                 return;
693
694         case -ECONNRESET:
695         case -ENOENT:
696         case -ESHUTDOWN:
697                 /* this urb is terminated, clean up */
698                 dev_dbg(dev, "%s - urb shutting down with status: %d\n",
699                         __func__, status);
700                 return;
701
702         default:
703                 dev_dbg(dev, "%s - nonzero urb status received: %d\n",
704                         __func__, status);
705                 goto exit;
706         }
707
708 exit:
709         retval = usb_submit_urb(urb, GFP_ATOMIC);
710         if (retval)
711                 dev_err(dev, "%s - usb_submit_urb failed with result %d\n",
712                         __func__, retval);
713 }
714
715 static int xpad_init_output(struct usb_interface *intf, struct usb_xpad *xpad)
716 {
717         struct usb_endpoint_descriptor *ep_irq_out;
718         int ep_irq_out_idx;
719         int error;
720
721         if (xpad->xtype == XTYPE_UNKNOWN)
722                 return 0;
723
724         xpad->odata = usb_alloc_coherent(xpad->udev, XPAD_PKT_LEN,
725                                          GFP_KERNEL, &xpad->odata_dma);
726         if (!xpad->odata) {
727                 error = -ENOMEM;
728                 goto fail1;
729         }
730
731         mutex_init(&xpad->odata_mutex);
732
733         xpad->irq_out = usb_alloc_urb(0, GFP_KERNEL);
734         if (!xpad->irq_out) {
735                 error = -ENOMEM;
736                 goto fail2;
737         }
738
739         /* Xbox One controller has in/out endpoints swapped. */
740         ep_irq_out_idx = xpad->xtype == XTYPE_XBOXONE ? 0 : 1;
741         ep_irq_out = &intf->cur_altsetting->endpoint[ep_irq_out_idx].desc;
742
743         usb_fill_int_urb(xpad->irq_out, xpad->udev,
744                          usb_sndintpipe(xpad->udev, ep_irq_out->bEndpointAddress),
745                          xpad->odata, XPAD_PKT_LEN,
746                          xpad_irq_out, xpad, ep_irq_out->bInterval);
747         xpad->irq_out->transfer_dma = xpad->odata_dma;
748         xpad->irq_out->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
749
750         return 0;
751
752  fail2: usb_free_coherent(xpad->udev, XPAD_PKT_LEN, xpad->odata, xpad->odata_dma);
753  fail1: return error;
754 }
755
756 static void xpad_stop_output(struct usb_xpad *xpad)
757 {
758         if (xpad->xtype != XTYPE_UNKNOWN)
759                 usb_kill_urb(xpad->irq_out);
760 }
761
762 static void xpad_deinit_output(struct usb_xpad *xpad)
763 {
764         if (xpad->xtype != XTYPE_UNKNOWN) {
765                 usb_free_urb(xpad->irq_out);
766                 usb_free_coherent(xpad->udev, XPAD_PKT_LEN,
767                                 xpad->odata, xpad->odata_dma);
768         }
769 }
770
771 static int xpad_inquiry_pad_presence(struct usb_xpad *xpad)
772 {
773         int retval;
774
775         mutex_lock(&xpad->odata_mutex);
776
777         xpad->odata[0] = 0x08;
778         xpad->odata[1] = 0x00;
779         xpad->odata[2] = 0x0F;
780         xpad->odata[3] = 0xC0;
781         xpad->odata[4] = 0x00;
782         xpad->odata[5] = 0x00;
783         xpad->odata[6] = 0x00;
784         xpad->odata[7] = 0x00;
785         xpad->odata[8] = 0x00;
786         xpad->odata[9] = 0x00;
787         xpad->odata[10] = 0x00;
788         xpad->odata[11] = 0x00;
789         xpad->irq_out->transfer_buffer_length = 12;
790
791         retval = usb_submit_urb(xpad->irq_out, GFP_KERNEL);
792
793         mutex_unlock(&xpad->odata_mutex);
794
795         return retval;
796 }
797
798 #ifdef CONFIG_JOYSTICK_XPAD_FF
799 static int xpad_play_effect(struct input_dev *dev, void *data, struct ff_effect *effect)
800 {
801         struct usb_xpad *xpad = input_get_drvdata(dev);
802         __u16 strong;
803         __u16 weak;
804
805         if (effect->type != FF_RUMBLE)
806                 return 0;
807
808         strong = effect->u.rumble.strong_magnitude;
809         weak = effect->u.rumble.weak_magnitude;
810
811         switch (xpad->xtype) {
812         case XTYPE_XBOX:
813                 xpad->odata[0] = 0x00;
814                 xpad->odata[1] = 0x06;
815                 xpad->odata[2] = 0x00;
816                 xpad->odata[3] = strong / 256;  /* left actuator */
817                 xpad->odata[4] = 0x00;
818                 xpad->odata[5] = weak / 256;    /* right actuator */
819                 xpad->irq_out->transfer_buffer_length = 6;
820                 break;
821
822         case XTYPE_XBOX360:
823                 xpad->odata[0] = 0x00;
824                 xpad->odata[1] = 0x08;
825                 xpad->odata[2] = 0x00;
826                 xpad->odata[3] = strong / 256;  /* left actuator? */
827                 xpad->odata[4] = weak / 256;    /* right actuator? */
828                 xpad->odata[5] = 0x00;
829                 xpad->odata[6] = 0x00;
830                 xpad->odata[7] = 0x00;
831                 xpad->irq_out->transfer_buffer_length = 8;
832                 break;
833
834         case XTYPE_XBOX360W:
835                 xpad->odata[0] = 0x00;
836                 xpad->odata[1] = 0x01;
837                 xpad->odata[2] = 0x0F;
838                 xpad->odata[3] = 0xC0;
839                 xpad->odata[4] = 0x00;
840                 xpad->odata[5] = strong / 256;
841                 xpad->odata[6] = weak / 256;
842                 xpad->odata[7] = 0x00;
843                 xpad->odata[8] = 0x00;
844                 xpad->odata[9] = 0x00;
845                 xpad->odata[10] = 0x00;
846                 xpad->odata[11] = 0x00;
847                 xpad->irq_out->transfer_buffer_length = 12;
848                 break;
849
850         case XTYPE_XBOXONE:
851                 xpad->odata[0] = 0x09; /* activate rumble */
852                 xpad->odata[1] = 0x08;
853                 xpad->odata[2] = 0x00;
854                 xpad->odata[3] = 0x08; /* continuous effect */
855                 xpad->odata[4] = 0x00; /* simple rumble mode */
856                 xpad->odata[5] = 0x03; /* L and R actuator only */
857                 xpad->odata[6] = 0x00; /* TODO: LT actuator */
858                 xpad->odata[7] = 0x00; /* TODO: RT actuator */
859                 xpad->odata[8] = strong / 256;  /* left actuator */
860                 xpad->odata[9] = weak / 256;    /* right actuator */
861                 xpad->odata[10] = 0x80; /* length of pulse */
862                 xpad->odata[11] = 0x00; /* stop period of pulse */
863                 xpad->irq_out->transfer_buffer_length = 12;
864                 break;
865
866         default:
867                 dev_dbg(&xpad->dev->dev,
868                         "%s - rumble command sent to unsupported xpad type: %d\n",
869                         __func__, xpad->xtype);
870                 return -EINVAL;
871         }
872
873         return usb_submit_urb(xpad->irq_out, GFP_ATOMIC);
874 }
875
876 static int xpad_init_ff(struct usb_xpad *xpad)
877 {
878         if (xpad->xtype == XTYPE_UNKNOWN)
879                 return 0;
880
881         input_set_capability(xpad->dev, EV_FF, FF_RUMBLE);
882
883         return input_ff_create_memless(xpad->dev, NULL, xpad_play_effect);
884 }
885
886 #else
887 static int xpad_init_ff(struct usb_xpad *xpad) { return 0; }
888 #endif
889
890 #if defined(CONFIG_JOYSTICK_XPAD_LEDS)
891 #include <linux/leds.h>
892 #include <linux/idr.h>
893
894 static DEFINE_IDA(xpad_pad_seq);
895
896 struct xpad_led {
897         char name[16];
898         struct led_classdev led_cdev;
899         struct usb_xpad *xpad;
900 };
901
902 /**
903  * set the LEDs on Xbox360 / Wireless Controllers
904  * @param command
905  *  0: off
906  *  1: all blink, then previous setting
907  *  2: 1/top-left blink, then on
908  *  3: 2/top-right blink, then on
909  *  4: 3/bottom-left blink, then on
910  *  5: 4/bottom-right blink, then on
911  *  6: 1/top-left on
912  *  7: 2/top-right on
913  *  8: 3/bottom-left on
914  *  9: 4/bottom-right on
915  * 10: rotate
916  * 11: blink, based on previous setting
917  * 12: slow blink, based on previous setting
918  * 13: rotate with two lights
919  * 14: persistent slow all blink
920  * 15: blink once, then previous setting
921  */
922 static void xpad_send_led_command(struct usb_xpad *xpad, int command)
923 {
924         command %= 16;
925
926         mutex_lock(&xpad->odata_mutex);
927
928         switch (xpad->xtype) {
929         case XTYPE_XBOX360:
930                 xpad->odata[0] = 0x01;
931                 xpad->odata[1] = 0x03;
932                 xpad->odata[2] = command;
933                 xpad->irq_out->transfer_buffer_length = 3;
934                 break;
935         case XTYPE_XBOX360W:
936                 xpad->odata[0] = 0x00;
937                 xpad->odata[1] = 0x00;
938                 xpad->odata[2] = 0x08;
939                 xpad->odata[3] = 0x40 + command;
940                 xpad->odata[4] = 0x00;
941                 xpad->odata[5] = 0x00;
942                 xpad->odata[6] = 0x00;
943                 xpad->odata[7] = 0x00;
944                 xpad->odata[8] = 0x00;
945                 xpad->odata[9] = 0x00;
946                 xpad->odata[10] = 0x00;
947                 xpad->odata[11] = 0x00;
948                 xpad->irq_out->transfer_buffer_length = 12;
949                 break;
950         }
951
952         usb_submit_urb(xpad->irq_out, GFP_KERNEL);
953         mutex_unlock(&xpad->odata_mutex);
954 }
955
956 /*
957  * Light up the segment corresponding to the pad number on
958  * Xbox 360 Controllers.
959  */
960 static void xpad_identify_controller(struct usb_xpad *xpad)
961 {
962         xpad_send_led_command(xpad, (xpad->pad_nr % 4) + 2);
963 }
964
965 static void xpad_led_set(struct led_classdev *led_cdev,
966                          enum led_brightness value)
967 {
968         struct xpad_led *xpad_led = container_of(led_cdev,
969                                                  struct xpad_led, led_cdev);
970
971         xpad_send_led_command(xpad_led->xpad, value);
972 }
973
974 static int xpad_led_probe(struct usb_xpad *xpad)
975 {
976         struct xpad_led *led;
977         struct led_classdev *led_cdev;
978         int error;
979
980         if (xpad->xtype != XTYPE_XBOX360 && xpad->xtype != XTYPE_XBOX360W)
981                 return 0;
982
983         xpad->led = led = kzalloc(sizeof(struct xpad_led), GFP_KERNEL);
984         if (!led)
985                 return -ENOMEM;
986
987         xpad->pad_nr = ida_simple_get(&xpad_pad_seq, 0, 0, GFP_KERNEL);
988         if (xpad->pad_nr < 0) {
989                 error = xpad->pad_nr;
990                 goto err_free_mem;
991         }
992
993         snprintf(led->name, sizeof(led->name), "xpad%d", xpad->pad_nr);
994         led->xpad = xpad;
995
996         led_cdev = &led->led_cdev;
997         led_cdev->name = led->name;
998         led_cdev->brightness_set = xpad_led_set;
999
1000         error = led_classdev_register(&xpad->udev->dev, led_cdev);
1001         if (error)
1002                 goto err_free_id;
1003
1004         if (xpad->xtype == XTYPE_XBOX360) {
1005                 /*
1006                  * Light up the segment corresponding to controller
1007                  * number on wired devices. On wireless we'll do that
1008                  * when they respond to "presence" packet.
1009                  */
1010                 xpad_identify_controller(xpad);
1011         }
1012
1013         return 0;
1014
1015 err_free_id:
1016         ida_simple_remove(&xpad_pad_seq, xpad->pad_nr);
1017 err_free_mem:
1018         kfree(led);
1019         xpad->led = NULL;
1020         return error;
1021 }
1022
1023 static void xpad_led_disconnect(struct usb_xpad *xpad)
1024 {
1025         struct xpad_led *xpad_led = xpad->led;
1026
1027         if (xpad_led) {
1028                 led_classdev_unregister(&xpad_led->led_cdev);
1029                 ida_simple_remove(&xpad_pad_seq, xpad->pad_nr);
1030                 kfree(xpad_led);
1031         }
1032 }
1033 #else
1034 static int xpad_led_probe(struct usb_xpad *xpad) { return 0; }
1035 static void xpad_led_disconnect(struct usb_xpad *xpad) { }
1036 static void xpad_identify_controller(struct usb_xpad *xpad) { }
1037 #endif
1038
1039 static int xpad_open(struct input_dev *dev)
1040 {
1041         struct usb_xpad *xpad = input_get_drvdata(dev);
1042
1043         /* URB was submitted in probe */
1044         if (xpad->xtype == XTYPE_XBOX360W)
1045                 return 0;
1046
1047         xpad->irq_in->dev = xpad->udev;
1048         if (usb_submit_urb(xpad->irq_in, GFP_KERNEL))
1049                 return -EIO;
1050
1051         if (xpad->xtype == XTYPE_XBOXONE) {
1052                 /* Xbox one controller needs to be initialized. */
1053                 xpad->odata[0] = 0x05;
1054                 xpad->odata[1] = 0x20;
1055                 xpad->irq_out->transfer_buffer_length = 2;
1056                 return usb_submit_urb(xpad->irq_out, GFP_KERNEL);
1057         }
1058
1059         return 0;
1060 }
1061
1062 static void xpad_close(struct input_dev *dev)
1063 {
1064         struct usb_xpad *xpad = input_get_drvdata(dev);
1065
1066         if (xpad->xtype != XTYPE_XBOX360W)
1067                 usb_kill_urb(xpad->irq_in);
1068
1069         xpad_stop_output(xpad);
1070 }
1071
1072 static void xpad_set_up_abs(struct input_dev *input_dev, signed short abs)
1073 {
1074         struct usb_xpad *xpad = input_get_drvdata(input_dev);
1075         set_bit(abs, input_dev->absbit);
1076
1077         switch (abs) {
1078         case ABS_X:
1079         case ABS_Y:
1080         case ABS_RX:
1081         case ABS_RY:    /* the two sticks */
1082                 input_set_abs_params(input_dev, abs, -32768, 32767, 16, 128);
1083                 break;
1084         case ABS_Z:
1085         case ABS_RZ:    /* the triggers (if mapped to axes) */
1086                 if (xpad->xtype == XTYPE_XBOXONE)
1087                         input_set_abs_params(input_dev, abs, 0, 1023, 0, 0);
1088                 else
1089                         input_set_abs_params(input_dev, abs, 0, 255, 0, 0);
1090                 break;
1091         case ABS_HAT0X:
1092         case ABS_HAT0Y: /* the d-pad (only if dpad is mapped to axes */
1093                 input_set_abs_params(input_dev, abs, -1, 1, 0, 0);
1094                 break;
1095         }
1096 }
1097
1098 static void xpad_deinit_input(struct usb_xpad *xpad)
1099 {
1100         xpad_led_disconnect(xpad);
1101         input_unregister_device(xpad->dev);
1102 }
1103
1104 static int xpad_init_input(struct usb_xpad *xpad)
1105 {
1106         struct input_dev *input_dev;
1107         int i, error;
1108
1109         input_dev = input_allocate_device();
1110         if (!input_dev)
1111                 return -ENOMEM;
1112
1113         xpad->dev = input_dev;
1114         input_dev->name = xpad->name;
1115         input_dev->phys = xpad->phys;
1116         usb_to_input_id(xpad->udev, &input_dev->id);
1117         input_dev->dev.parent = &xpad->intf->dev;
1118
1119         input_set_drvdata(input_dev, xpad);
1120
1121         input_dev->open = xpad_open;
1122         input_dev->close = xpad_close;
1123
1124         __set_bit(EV_KEY, input_dev->evbit);
1125
1126         if (!(xpad->mapping & MAP_STICKS_TO_NULL)) {
1127                 __set_bit(EV_ABS, input_dev->evbit);
1128                 /* set up axes */
1129                 for (i = 0; xpad_abs[i] >= 0; i++)
1130                         xpad_set_up_abs(input_dev, xpad_abs[i]);
1131         }
1132
1133         /* set up standard buttons */
1134         for (i = 0; xpad_common_btn[i] >= 0; i++)
1135                 __set_bit(xpad_common_btn[i], input_dev->keybit);
1136
1137         /* set up model-specific ones */
1138         if (xpad->xtype == XTYPE_XBOX360 || xpad->xtype == XTYPE_XBOX360W ||
1139             xpad->xtype == XTYPE_XBOXONE) {
1140                 for (i = 0; xpad360_btn[i] >= 0; i++)
1141                         __set_bit(xpad360_btn[i], input_dev->keybit);
1142         } else {
1143                 for (i = 0; xpad_btn[i] >= 0; i++)
1144                         __set_bit(xpad_btn[i], input_dev->keybit);
1145         }
1146
1147         if (xpad->mapping & MAP_DPAD_TO_BUTTONS) {
1148                 for (i = 0; xpad_btn_pad[i] >= 0; i++)
1149                         __set_bit(xpad_btn_pad[i], input_dev->keybit);
1150         }
1151
1152         /*
1153          * This should be a simple else block. However historically
1154          * xbox360w has mapped DPAD to buttons while xbox360 did not. This
1155          * made no sense, but now we can not just switch back and have to
1156          * support both behaviors.
1157          */
1158         if (!(xpad->mapping & MAP_DPAD_TO_BUTTONS) ||
1159             xpad->xtype == XTYPE_XBOX360W) {
1160                 for (i = 0; xpad_abs_pad[i] >= 0; i++)
1161                         xpad_set_up_abs(input_dev, xpad_abs_pad[i]);
1162         }
1163
1164         if (xpad->mapping & MAP_TRIGGERS_TO_BUTTONS) {
1165                 for (i = 0; xpad_btn_triggers[i] >= 0; i++)
1166                         __set_bit(xpad_btn_triggers[i], input_dev->keybit);
1167         } else {
1168                 for (i = 0; xpad_abs_triggers[i] >= 0; i++)
1169                         xpad_set_up_abs(input_dev, xpad_abs_triggers[i]);
1170         }
1171
1172         error = xpad_init_ff(xpad);
1173         if (error)
1174                 goto err_free_input;
1175
1176         error = xpad_led_probe(xpad);
1177         if (error)
1178                 goto err_destroy_ff;
1179
1180         error = input_register_device(xpad->dev);
1181         if (error)
1182                 goto err_disconnect_led;
1183
1184         return 0;
1185
1186 err_disconnect_led:
1187         xpad_led_disconnect(xpad);
1188 err_destroy_ff:
1189         input_ff_destroy(input_dev);
1190 err_free_input:
1191         input_free_device(input_dev);
1192         return error;
1193 }
1194
1195 static int xpad_probe(struct usb_interface *intf, const struct usb_device_id *id)
1196 {
1197         struct usb_device *udev = interface_to_usbdev(intf);
1198         struct usb_xpad *xpad;
1199         struct usb_endpoint_descriptor *ep_irq_in;
1200         int ep_irq_in_idx;
1201         int i, error;
1202
1203         for (i = 0; xpad_device[i].idVendor; i++) {
1204                 if ((le16_to_cpu(udev->descriptor.idVendor) == xpad_device[i].idVendor) &&
1205                     (le16_to_cpu(udev->descriptor.idProduct) == xpad_device[i].idProduct))
1206                         break;
1207         }
1208
1209         if (xpad_device[i].xtype == XTYPE_XBOXONE &&
1210             intf->cur_altsetting->desc.bInterfaceNumber != 0) {
1211                 /*
1212                  * The Xbox One controller lists three interfaces all with the
1213                  * same interface class, subclass and protocol. Differentiate by
1214                  * interface number.
1215                  */
1216                 return -ENODEV;
1217         }
1218
1219         xpad = kzalloc(sizeof(struct usb_xpad), GFP_KERNEL);
1220         if (!xpad)
1221                 return -ENOMEM;
1222
1223         usb_make_path(udev, xpad->phys, sizeof(xpad->phys));
1224         strlcat(xpad->phys, "/input0", sizeof(xpad->phys));
1225
1226         xpad->idata = usb_alloc_coherent(udev, XPAD_PKT_LEN,
1227                                          GFP_KERNEL, &xpad->idata_dma);
1228         if (!xpad->idata) {
1229                 error = -ENOMEM;
1230                 goto err_free_mem;
1231         }
1232
1233         xpad->irq_in = usb_alloc_urb(0, GFP_KERNEL);
1234         if (!xpad->irq_in) {
1235                 error = -ENOMEM;
1236                 goto err_free_idata;
1237         }
1238
1239         xpad->udev = udev;
1240         xpad->intf = intf;
1241         xpad->mapping = xpad_device[i].mapping;
1242         xpad->xtype = xpad_device[i].xtype;
1243         xpad->name = xpad_device[i].name;
1244
1245         if (xpad->xtype == XTYPE_UNKNOWN) {
1246                 if (intf->cur_altsetting->desc.bInterfaceClass == USB_CLASS_VENDOR_SPEC) {
1247                         if (intf->cur_altsetting->desc.bInterfaceProtocol == 129)
1248                                 xpad->xtype = XTYPE_XBOX360W;
1249                         else
1250                                 xpad->xtype = XTYPE_XBOX360;
1251                 } else {
1252                         xpad->xtype = XTYPE_XBOX;
1253                 }
1254
1255                 if (dpad_to_buttons)
1256                         xpad->mapping |= MAP_DPAD_TO_BUTTONS;
1257                 if (triggers_to_buttons)
1258                         xpad->mapping |= MAP_TRIGGERS_TO_BUTTONS;
1259                 if (sticks_to_null)
1260                         xpad->mapping |= MAP_STICKS_TO_NULL;
1261         }
1262
1263         error = xpad_init_output(intf, xpad);
1264         if (error)
1265                 goto err_free_in_urb;
1266
1267         /* Xbox One controller has in/out endpoints swapped. */
1268         ep_irq_in_idx = xpad->xtype == XTYPE_XBOXONE ? 1 : 0;
1269         ep_irq_in = &intf->cur_altsetting->endpoint[ep_irq_in_idx].desc;
1270
1271         usb_fill_int_urb(xpad->irq_in, udev,
1272                          usb_rcvintpipe(udev, ep_irq_in->bEndpointAddress),
1273                          xpad->idata, XPAD_PKT_LEN, xpad_irq_in,
1274                          xpad, ep_irq_in->bInterval);
1275         xpad->irq_in->transfer_dma = xpad->idata_dma;
1276         xpad->irq_in->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
1277
1278         usb_set_intfdata(intf, xpad);
1279
1280         error = xpad_init_input(xpad);
1281         if (error)
1282                 goto err_deinit_output;
1283
1284         if (xpad->xtype == XTYPE_XBOX360W) {
1285                 /*
1286                  * Submit the int URB immediately rather than waiting for open
1287                  * because we get status messages from the device whether
1288                  * or not any controllers are attached.  In fact, it's
1289                  * exactly the message that a controller has arrived that
1290                  * we're waiting for.
1291                  */
1292                 xpad->irq_in->dev = xpad->udev;
1293                 error = usb_submit_urb(xpad->irq_in, GFP_KERNEL);
1294                 if (error)
1295                         goto err_deinit_input;
1296
1297                 /*
1298                  * Send presence packet.
1299                  * This will force the controller to resend connection packets.
1300                  * This is useful in the case we activate the module after the
1301                  * adapter has been plugged in, as it won't automatically
1302                  * send us info about the controllers.
1303                  */
1304                 error = xpad_inquiry_pad_presence(xpad);
1305                 if (error)
1306                         goto err_kill_in_urb;
1307         }
1308         return 0;
1309
1310 err_kill_in_urb:
1311         usb_kill_urb(xpad->irq_in);
1312 err_deinit_input:
1313         xpad_deinit_input(xpad);
1314 err_deinit_output:
1315         xpad_deinit_output(xpad);
1316 err_free_in_urb:
1317         usb_free_urb(xpad->irq_in);
1318 err_free_idata:
1319         usb_free_coherent(udev, XPAD_PKT_LEN, xpad->idata, xpad->idata_dma);
1320 err_free_mem:
1321         kfree(xpad);
1322         return error;
1323
1324 }
1325
1326 static void xpad_disconnect(struct usb_interface *intf)
1327 {
1328         struct usb_xpad *xpad = usb_get_intfdata (intf);
1329
1330         xpad_deinit_input(xpad);
1331         xpad_deinit_output(xpad);
1332
1333         if (xpad->xtype == XTYPE_XBOX360W) {
1334                 usb_kill_urb(xpad->irq_in);
1335         }
1336
1337         usb_free_urb(xpad->irq_in);
1338         usb_free_coherent(xpad->udev, XPAD_PKT_LEN,
1339                         xpad->idata, xpad->idata_dma);
1340
1341         kfree(xpad);
1342
1343         usb_set_intfdata(intf, NULL);
1344 }
1345
1346 static struct usb_driver xpad_driver = {
1347         .name           = "xpad",
1348         .probe          = xpad_probe,
1349         .disconnect     = xpad_disconnect,
1350         .id_table       = xpad_table,
1351 };
1352
1353 module_usb_driver(xpad_driver);
1354
1355 MODULE_AUTHOR(DRIVER_AUTHOR);
1356 MODULE_DESCRIPTION(DRIVER_DESC);
1357 MODULE_LICENSE("GPL");