xref: /linux/drivers/hid/hid-sony.c (revision cb54b53adae70701bdd77d848cea4b9b39b61cf9)
1 /*
2  *  HID driver for Sony / PS2 / PS3 BD devices.
3  *
4  *  Copyright (c) 1999 Andreas Gal
5  *  Copyright (c) 2000-2005 Vojtech Pavlik <vojtech@suse.cz>
6  *  Copyright (c) 2005 Michael Haboustak <mike-@cinci.rr.com> for Concept2, Inc
7  *  Copyright (c) 2008 Jiri Slaby
8  *  Copyright (c) 2012 David Dillow <dave@thedillows.org>
9  *  Copyright (c) 2006-2013 Jiri Kosina
10  *  Copyright (c) 2013 Colin Leitner <colin.leitner@gmail.com>
11  */
12 
13 /*
14  * This program is free software; you can redistribute it and/or modify it
15  * under the terms of the GNU General Public License as published by the Free
16  * Software Foundation; either version 2 of the License, or (at your option)
17  * any later version.
18  */
19 
20 /* NOTE: in order for the Sony PS3 BD Remote Control to be found by
21  * a Bluetooth host, the key combination Start+Enter has to be kept pressed
22  * for about 7 seconds with the Bluetooth Host Controller in discovering mode.
23  *
24  * There will be no PIN request from the device.
25  */
26 
27 #include <linux/device.h>
28 #include <linux/hid.h>
29 #include <linux/module.h>
30 #include <linux/slab.h>
31 #include <linux/usb.h>
32 #include <linux/leds.h>
33 
34 #include "hid-ids.h"
35 
36 #define VAIO_RDESC_CONSTANT     (1 << 0)
37 #define SIXAXIS_CONTROLLER_USB  (1 << 1)
38 #define SIXAXIS_CONTROLLER_BT   (1 << 2)
39 #define BUZZ_CONTROLLER         (1 << 3)
40 #define PS3REMOTE		(1 << 4)
41 
42 static const u8 sixaxis_rdesc_fixup[] = {
43 	0x95, 0x13, 0x09, 0x01, 0x81, 0x02, 0x95, 0x0C,
44 	0x81, 0x01, 0x75, 0x10, 0x95, 0x04, 0x26, 0xFF,
45 	0x03, 0x46, 0xFF, 0x03, 0x09, 0x01, 0x81, 0x02
46 };
47 
48 static const u8 sixaxis_rdesc_fixup2[] = {
49 	0x05, 0x01, 0x09, 0x04, 0xa1, 0x01, 0xa1, 0x02,
50 	0x85, 0x01, 0x75, 0x08, 0x95, 0x01, 0x15, 0x00,
51 	0x26, 0xff, 0x00, 0x81, 0x03, 0x75, 0x01, 0x95,
52 	0x13, 0x15, 0x00, 0x25, 0x01, 0x35, 0x00, 0x45,
53 	0x01, 0x05, 0x09, 0x19, 0x01, 0x29, 0x13, 0x81,
54 	0x02, 0x75, 0x01, 0x95, 0x0d, 0x06, 0x00, 0xff,
55 	0x81, 0x03, 0x15, 0x00, 0x26, 0xff, 0x00, 0x05,
56 	0x01, 0x09, 0x01, 0xa1, 0x00, 0x75, 0x08, 0x95,
57 	0x04, 0x35, 0x00, 0x46, 0xff, 0x00, 0x09, 0x30,
58 	0x09, 0x31, 0x09, 0x32, 0x09, 0x35, 0x81, 0x02,
59 	0xc0, 0x05, 0x01, 0x95, 0x13, 0x09, 0x01, 0x81,
60 	0x02, 0x95, 0x0c, 0x81, 0x01, 0x75, 0x10, 0x95,
61 	0x04, 0x26, 0xff, 0x03, 0x46, 0xff, 0x03, 0x09,
62 	0x01, 0x81, 0x02, 0xc0, 0xa1, 0x02, 0x85, 0x02,
63 	0x75, 0x08, 0x95, 0x30, 0x09, 0x01, 0xb1, 0x02,
64 	0xc0, 0xa1, 0x02, 0x85, 0xee, 0x75, 0x08, 0x95,
65 	0x30, 0x09, 0x01, 0xb1, 0x02, 0xc0, 0xa1, 0x02,
66 	0x85, 0xef, 0x75, 0x08, 0x95, 0x30, 0x09, 0x01,
67 	0xb1, 0x02, 0xc0, 0xc0,
68 };
69 
70 static __u8 ps3remote_rdesc[] = {
71 	0x05, 0x01,          /* GUsagePage Generic Desktop */
72 	0x09, 0x05,          /* LUsage 0x05 [Game Pad] */
73 	0xA1, 0x01,          /* MCollection Application (mouse, keyboard) */
74 
75 	 /* Use collection 1 for joypad buttons */
76 	 0xA1, 0x02,         /* MCollection Logical (interrelated data) */
77 
78 	  /* Ignore the 1st byte, maybe it is used for a controller
79 	   * number but it's not needed for correct operation */
80 	  0x75, 0x08,        /* GReportSize 0x08 [8] */
81 	  0x95, 0x01,        /* GReportCount 0x01 [1] */
82 	  0x81, 0x01,        /* MInput 0x01 (Const[0] Arr[1] Abs[2]) */
83 
84 	  /* Bytes from 2nd to 4th are a bitmap for joypad buttons, for these
85 	   * buttons multiple keypresses are allowed */
86 	  0x05, 0x09,        /* GUsagePage Button */
87 	  0x19, 0x01,        /* LUsageMinimum 0x01 [Button 1 (primary/trigger)] */
88 	  0x29, 0x18,        /* LUsageMaximum 0x18 [Button 24] */
89 	  0x14,              /* GLogicalMinimum [0] */
90 	  0x25, 0x01,        /* GLogicalMaximum 0x01 [1] */
91 	  0x75, 0x01,        /* GReportSize 0x01 [1] */
92 	  0x95, 0x18,        /* GReportCount 0x18 [24] */
93 	  0x81, 0x02,        /* MInput 0x02 (Data[0] Var[1] Abs[2]) */
94 
95 	  0xC0,              /* MEndCollection */
96 
97 	 /* Use collection 2 for remote control buttons */
98 	 0xA1, 0x02,         /* MCollection Logical (interrelated data) */
99 
100 	  /* 5th byte is used for remote control buttons */
101 	  0x05, 0x09,        /* GUsagePage Button */
102 	  0x18,              /* LUsageMinimum [No button pressed] */
103 	  0x29, 0xFE,        /* LUsageMaximum 0xFE [Button 254] */
104 	  0x14,              /* GLogicalMinimum [0] */
105 	  0x26, 0xFE, 0x00,  /* GLogicalMaximum 0x00FE [254] */
106 	  0x75, 0x08,        /* GReportSize 0x08 [8] */
107 	  0x95, 0x01,        /* GReportCount 0x01 [1] */
108 	  0x80,              /* MInput  */
109 
110 	  /* Ignore bytes from 6th to 11th, 6th to 10th are always constant at
111 	   * 0xff and 11th is for press indication */
112 	  0x75, 0x08,        /* GReportSize 0x08 [8] */
113 	  0x95, 0x06,        /* GReportCount 0x06 [6] */
114 	  0x81, 0x01,        /* MInput 0x01 (Const[0] Arr[1] Abs[2]) */
115 
116 	  /* 12th byte is for battery strength */
117 	  0x05, 0x06,        /* GUsagePage Generic Device Controls */
118 	  0x09, 0x20,        /* LUsage 0x20 [Battery Strength] */
119 	  0x14,              /* GLogicalMinimum [0] */
120 	  0x25, 0x05,        /* GLogicalMaximum 0x05 [5] */
121 	  0x75, 0x08,        /* GReportSize 0x08 [8] */
122 	  0x95, 0x01,        /* GReportCount 0x01 [1] */
123 	  0x81, 0x02,        /* MInput 0x02 (Data[0] Var[1] Abs[2]) */
124 
125 	  0xC0,              /* MEndCollection */
126 
127 	 0xC0                /* MEndCollection [Game Pad] */
128 };
129 
130 static const unsigned int ps3remote_keymap_joypad_buttons[] = {
131 	[0x01] = KEY_SELECT,
132 	[0x02] = BTN_THUMBL,		/* L3 */
133 	[0x03] = BTN_THUMBR,		/* R3 */
134 	[0x04] = BTN_START,
135 	[0x05] = KEY_UP,
136 	[0x06] = KEY_RIGHT,
137 	[0x07] = KEY_DOWN,
138 	[0x08] = KEY_LEFT,
139 	[0x09] = BTN_TL2,		/* L2 */
140 	[0x0a] = BTN_TR2,		/* R2 */
141 	[0x0b] = BTN_TL,		/* L1 */
142 	[0x0c] = BTN_TR,		/* R1 */
143 	[0x0d] = KEY_OPTION,		/* options/triangle */
144 	[0x0e] = KEY_BACK,		/* back/circle */
145 	[0x0f] = BTN_0,			/* cross */
146 	[0x10] = KEY_SCREEN,		/* view/square */
147 	[0x11] = KEY_HOMEPAGE,		/* PS button */
148 	[0x14] = KEY_ENTER,
149 };
150 static const unsigned int ps3remote_keymap_remote_buttons[] = {
151 	[0x00] = KEY_1,
152 	[0x01] = KEY_2,
153 	[0x02] = KEY_3,
154 	[0x03] = KEY_4,
155 	[0x04] = KEY_5,
156 	[0x05] = KEY_6,
157 	[0x06] = KEY_7,
158 	[0x07] = KEY_8,
159 	[0x08] = KEY_9,
160 	[0x09] = KEY_0,
161 	[0x0e] = KEY_ESC,		/* return */
162 	[0x0f] = KEY_CLEAR,
163 	[0x16] = KEY_EJECTCD,
164 	[0x1a] = KEY_MENU,		/* top menu */
165 	[0x28] = KEY_TIME,
166 	[0x30] = KEY_PREVIOUS,
167 	[0x31] = KEY_NEXT,
168 	[0x32] = KEY_PLAY,
169 	[0x33] = KEY_REWIND,		/* scan back */
170 	[0x34] = KEY_FORWARD,		/* scan forward */
171 	[0x38] = KEY_STOP,
172 	[0x39] = KEY_PAUSE,
173 	[0x40] = KEY_CONTEXT_MENU,	/* pop up/menu */
174 	[0x60] = KEY_FRAMEBACK,		/* slow/step back */
175 	[0x61] = KEY_FRAMEFORWARD,	/* slow/step forward */
176 	[0x63] = KEY_SUBTITLE,
177 	[0x64] = KEY_AUDIO,
178 	[0x65] = KEY_ANGLE,
179 	[0x70] = KEY_INFO,		/* display */
180 	[0x80] = KEY_BLUE,
181 	[0x81] = KEY_RED,
182 	[0x82] = KEY_GREEN,
183 	[0x83] = KEY_YELLOW,
184 };
185 
186 static const unsigned int buzz_keymap[] = {
187 	/* The controller has 4 remote buzzers, each with one LED and 5
188 	 * buttons.
189 	 *
190 	 * We use the mapping chosen by the controller, which is:
191 	 *
192 	 * Key          Offset
193 	 * -------------------
194 	 * Buzz              1
195 	 * Blue              5
196 	 * Orange            4
197 	 * Green             3
198 	 * Yellow            2
199 	 *
200 	 * So, for example, the orange button on the third buzzer is mapped to
201 	 * BTN_TRIGGER_HAPPY14
202 	 */
203 	[ 1] = BTN_TRIGGER_HAPPY1,
204 	[ 2] = BTN_TRIGGER_HAPPY2,
205 	[ 3] = BTN_TRIGGER_HAPPY3,
206 	[ 4] = BTN_TRIGGER_HAPPY4,
207 	[ 5] = BTN_TRIGGER_HAPPY5,
208 	[ 6] = BTN_TRIGGER_HAPPY6,
209 	[ 7] = BTN_TRIGGER_HAPPY7,
210 	[ 8] = BTN_TRIGGER_HAPPY8,
211 	[ 9] = BTN_TRIGGER_HAPPY9,
212 	[10] = BTN_TRIGGER_HAPPY10,
213 	[11] = BTN_TRIGGER_HAPPY11,
214 	[12] = BTN_TRIGGER_HAPPY12,
215 	[13] = BTN_TRIGGER_HAPPY13,
216 	[14] = BTN_TRIGGER_HAPPY14,
217 	[15] = BTN_TRIGGER_HAPPY15,
218 	[16] = BTN_TRIGGER_HAPPY16,
219 	[17] = BTN_TRIGGER_HAPPY17,
220 	[18] = BTN_TRIGGER_HAPPY18,
221 	[19] = BTN_TRIGGER_HAPPY19,
222 	[20] = BTN_TRIGGER_HAPPY20,
223 };
224 
225 struct sony_sc {
226 	unsigned long quirks;
227 
228 	void *extra;
229 };
230 
231 struct buzz_extra {
232 	int led_state;
233 	struct led_classdev *leds[4];
234 };
235 
236 static __u8 *ps3remote_fixup(struct hid_device *hdev, __u8 *rdesc,
237 			     unsigned int *rsize)
238 {
239 	*rsize = sizeof(ps3remote_rdesc);
240 	return ps3remote_rdesc;
241 }
242 
243 static int ps3remote_mapping(struct hid_device *hdev, struct hid_input *hi,
244 			     struct hid_field *field, struct hid_usage *usage,
245 			     unsigned long **bit, int *max)
246 {
247 	unsigned int key = usage->hid & HID_USAGE;
248 
249 	if ((usage->hid & HID_USAGE_PAGE) != HID_UP_BUTTON)
250 		return -1;
251 
252 	switch (usage->collection_index) {
253 	case 1:
254 		if (key >= ARRAY_SIZE(ps3remote_keymap_joypad_buttons))
255 			return -1;
256 
257 		key = ps3remote_keymap_joypad_buttons[key];
258 		if (!key)
259 			return -1;
260 		break;
261 	case 2:
262 		if (key >= ARRAY_SIZE(ps3remote_keymap_remote_buttons))
263 			return -1;
264 
265 		key = ps3remote_keymap_remote_buttons[key];
266 		if (!key)
267 			return -1;
268 		break;
269 	default:
270 		return -1;
271 	}
272 
273 	hid_map_usage_clear(hi, usage, bit, max, EV_KEY, key);
274 	return 1;
275 }
276 
277 
278 /* Sony Vaio VGX has wrongly mouse pointer declared as constant */
279 static __u8 *sony_report_fixup(struct hid_device *hdev, __u8 *rdesc,
280 		unsigned int *rsize)
281 {
282 	struct sony_sc *sc = hid_get_drvdata(hdev);
283 
284 	/*
285 	 * Some Sony RF receivers wrongly declare the mouse pointer as a
286 	 * a constant non-data variable.
287 	 */
288 	if ((sc->quirks & VAIO_RDESC_CONSTANT) && *rsize >= 56 &&
289 	    /* usage page: generic desktop controls */
290 	    /* rdesc[0] == 0x05 && rdesc[1] == 0x01 && */
291 	    /* usage: mouse */
292 	    rdesc[2] == 0x09 && rdesc[3] == 0x02 &&
293 	    /* input (usage page for x,y axes): constant, variable, relative */
294 	    rdesc[54] == 0x81 && rdesc[55] == 0x07) {
295 		hid_info(hdev, "Fixing up Sony RF Receiver report descriptor\n");
296 		/* input: data, variable, relative */
297 		rdesc[55] = 0x06;
298 	}
299 
300 	/* The HID descriptor exposed over BT has a trailing zero byte */
301 	if ((((sc->quirks & SIXAXIS_CONTROLLER_USB) && *rsize == 148) ||
302 			((sc->quirks & SIXAXIS_CONTROLLER_BT) && *rsize == 149)) &&
303 			rdesc[83] == 0x75) {
304 		hid_info(hdev, "Fixing up Sony Sixaxis report descriptor\n");
305 		memcpy((void *)&rdesc[83], (void *)&sixaxis_rdesc_fixup,
306 			sizeof(sixaxis_rdesc_fixup));
307 	} else if (sc->quirks & SIXAXIS_CONTROLLER_USB &&
308 		   *rsize > sizeof(sixaxis_rdesc_fixup2)) {
309 		hid_info(hdev, "Sony Sixaxis clone detected. Using original report descriptor (size: %d clone; %d new)\n",
310 			 *rsize, (int)sizeof(sixaxis_rdesc_fixup2));
311 		*rsize = sizeof(sixaxis_rdesc_fixup2);
312 		memcpy(rdesc, &sixaxis_rdesc_fixup2, *rsize);
313 	}
314 
315 	if (sc->quirks & PS3REMOTE)
316 		return ps3remote_fixup(hdev, rdesc, rsize);
317 
318 	return rdesc;
319 }
320 
321 static int sony_raw_event(struct hid_device *hdev, struct hid_report *report,
322 		__u8 *rd, int size)
323 {
324 	struct sony_sc *sc = hid_get_drvdata(hdev);
325 
326 	/* Sixaxis HID report has acclerometers/gyro with MSByte first, this
327 	 * has to be BYTE_SWAPPED before passing up to joystick interface
328 	 */
329 	if ((sc->quirks & (SIXAXIS_CONTROLLER_USB | SIXAXIS_CONTROLLER_BT)) &&
330 			rd[0] == 0x01 && size == 49) {
331 		swap(rd[41], rd[42]);
332 		swap(rd[43], rd[44]);
333 		swap(rd[45], rd[46]);
334 		swap(rd[47], rd[48]);
335 	}
336 
337 	return 0;
338 }
339 
340 static int sony_mapping(struct hid_device *hdev, struct hid_input *hi,
341 			struct hid_field *field, struct hid_usage *usage,
342 			unsigned long **bit, int *max)
343 {
344 	struct sony_sc *sc = hid_get_drvdata(hdev);
345 
346 	if (sc->quirks & BUZZ_CONTROLLER) {
347 		unsigned int key = usage->hid & HID_USAGE;
348 
349 		if ((usage->hid & HID_USAGE_PAGE) != HID_UP_BUTTON)
350 			return -1;
351 
352 		switch (usage->collection_index) {
353 		case 1:
354 			if (key >= ARRAY_SIZE(buzz_keymap))
355 				return -1;
356 
357 			key = buzz_keymap[key];
358 			if (!key)
359 				return -1;
360 			break;
361 		default:
362 			return -1;
363 		}
364 
365 		hid_map_usage_clear(hi, usage, bit, max, EV_KEY, key);
366 		return 1;
367 	}
368 
369 	if (sc->quirks & PS3REMOTE)
370 		return ps3remote_mapping(hdev, hi, field, usage, bit, max);
371 
372 	return -1;
373 }
374 
375 /*
376  * The Sony Sixaxis does not handle HID Output Reports on the Interrupt EP
377  * like it should according to usbhid/hid-core.c::usbhid_output_raw_report()
378  * so we need to override that forcing HID Output Reports on the Control EP.
379  *
380  * There is also another issue about HID Output Reports via USB, the Sixaxis
381  * does not want the report_id as part of the data packet, so we have to
382  * discard buf[0] when sending the actual control message, even for numbered
383  * reports, humpf!
384  */
385 static int sixaxis_usb_output_raw_report(struct hid_device *hid, __u8 *buf,
386 		size_t count, unsigned char report_type)
387 {
388 	struct usb_interface *intf = to_usb_interface(hid->dev.parent);
389 	struct usb_device *dev = interface_to_usbdev(intf);
390 	struct usb_host_interface *interface = intf->cur_altsetting;
391 	int report_id = buf[0];
392 	int ret;
393 
394 	if (report_type == HID_OUTPUT_REPORT) {
395 		/* Don't send the Report ID */
396 		buf++;
397 		count--;
398 	}
399 
400 	ret = usb_control_msg(dev, usb_sndctrlpipe(dev, 0),
401 		HID_REQ_SET_REPORT,
402 		USB_DIR_OUT | USB_TYPE_CLASS | USB_RECIP_INTERFACE,
403 		((report_type + 1) << 8) | report_id,
404 		interface->desc.bInterfaceNumber, buf, count,
405 		USB_CTRL_SET_TIMEOUT);
406 
407 	/* Count also the Report ID, in case of an Output report. */
408 	if (ret > 0 && report_type == HID_OUTPUT_REPORT)
409 		ret++;
410 
411 	return ret;
412 }
413 
414 /*
415  * Sending HID_REQ_GET_REPORT changes the operation mode of the ps3 controller
416  * to "operational".  Without this, the ps3 controller will not report any
417  * events.
418  */
419 static int sixaxis_set_operational_usb(struct hid_device *hdev)
420 {
421 	struct usb_interface *intf = to_usb_interface(hdev->dev.parent);
422 	struct usb_device *dev = interface_to_usbdev(intf);
423 	__u16 ifnum = intf->cur_altsetting->desc.bInterfaceNumber;
424 	int ret;
425 	char *buf = kmalloc(18, GFP_KERNEL);
426 
427 	if (!buf)
428 		return -ENOMEM;
429 
430 	ret = usb_control_msg(dev, usb_rcvctrlpipe(dev, 0),
431 				 HID_REQ_GET_REPORT,
432 				 USB_DIR_IN | USB_TYPE_CLASS |
433 				 USB_RECIP_INTERFACE,
434 				 (3 << 8) | 0xf2, ifnum, buf, 17,
435 				 USB_CTRL_GET_TIMEOUT);
436 	if (ret < 0)
437 		hid_err(hdev, "can't set operational mode\n");
438 
439 	kfree(buf);
440 
441 	return ret;
442 }
443 
444 static int sixaxis_set_operational_bt(struct hid_device *hdev)
445 {
446 	unsigned char buf[] = { 0xf4,  0x42, 0x03, 0x00, 0x00 };
447 	return hdev->hid_output_raw_report(hdev, buf, sizeof(buf), HID_FEATURE_REPORT);
448 }
449 
450 static void buzz_set_leds(struct hid_device *hdev, int leds)
451 {
452 	struct list_head *report_list =
453 		&hdev->report_enum[HID_OUTPUT_REPORT].report_list;
454 	struct hid_report *report = list_entry(report_list->next,
455 		struct hid_report, list);
456 	__s32 *value = report->field[0]->value;
457 
458 	value[0] = 0x00;
459 	value[1] = (leds & 1) ? 0xff : 0x00;
460 	value[2] = (leds & 2) ? 0xff : 0x00;
461 	value[3] = (leds & 4) ? 0xff : 0x00;
462 	value[4] = (leds & 8) ? 0xff : 0x00;
463 	value[5] = 0x00;
464 	value[6] = 0x00;
465 	hid_hw_request(hdev, report, HID_REQ_SET_REPORT);
466 }
467 
468 static void buzz_led_set_brightness(struct led_classdev *led,
469 				    enum led_brightness value)
470 {
471 	struct device *dev = led->dev->parent;
472 	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
473 	struct sony_sc *drv_data;
474 	struct buzz_extra *buzz;
475 
476 	int n;
477 
478 	drv_data = hid_get_drvdata(hdev);
479 	if (!drv_data || !drv_data->extra) {
480 		hid_err(hdev, "No device data\n");
481 		return;
482 	}
483 	buzz = drv_data->extra;
484 
485 	for (n = 0; n < 4; n++) {
486 		if (led == buzz->leds[n]) {
487 			int on = !! (buzz->led_state & (1 << n));
488 			if (value == LED_OFF && on) {
489 				buzz->led_state &= ~(1 << n);
490 				buzz_set_leds(hdev, buzz->led_state);
491 			} else if (value != LED_OFF && !on) {
492 				buzz->led_state |= (1 << n);
493 				buzz_set_leds(hdev, buzz->led_state);
494 			}
495 			break;
496 		}
497 	}
498 }
499 
500 static enum led_brightness buzz_led_get_brightness(struct led_classdev *led)
501 {
502 	struct device *dev = led->dev->parent;
503 	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
504 	struct sony_sc *drv_data;
505 	struct buzz_extra *buzz;
506 
507 	int n;
508 	int on = 0;
509 
510 	drv_data = hid_get_drvdata(hdev);
511 	if (!drv_data || !drv_data->extra) {
512 		hid_err(hdev, "No device data\n");
513 		return LED_OFF;
514 	}
515 	buzz = drv_data->extra;
516 
517 	for (n = 0; n < 4; n++) {
518 		if (led == buzz->leds[n]) {
519 			on = !! (buzz->led_state & (1 << n));
520 			break;
521 		}
522 	}
523 
524 	return on ? LED_FULL : LED_OFF;
525 }
526 
527 static int buzz_init(struct hid_device *hdev)
528 {
529 	struct sony_sc *drv_data;
530 	struct buzz_extra *buzz;
531 	int n, ret = 0;
532 	struct led_classdev *led;
533 	size_t name_sz;
534 	char *name;
535 
536 	drv_data = hid_get_drvdata(hdev);
537 	BUG_ON(!(drv_data->quirks & BUZZ_CONTROLLER));
538 
539 	buzz = kzalloc(sizeof(*buzz), GFP_KERNEL);
540 	if (!buzz) {
541 		hid_err(hdev, "Insufficient memory, cannot allocate driver data\n");
542 		return -ENOMEM;
543 	}
544 	drv_data->extra = buzz;
545 
546 	/* Clear LEDs as we have no way of reading their initial state. This is
547 	 * only relevant if the driver is loaded after somebody actively set the
548 	 * LEDs to on */
549 	buzz_set_leds(hdev, 0x00);
550 
551 	name_sz = strlen(dev_name(&hdev->dev)) + strlen("::buzz#") + 1;
552 
553 	for (n = 0; n < 4; n++) {
554 		led = kzalloc(sizeof(struct led_classdev) + name_sz, GFP_KERNEL);
555 		if (!led) {
556 			hid_err(hdev, "Couldn't allocate memory for LED %d\n", n);
557 			goto error_leds;
558 		}
559 
560 		name = (void *)(&led[1]);
561 		snprintf(name, name_sz, "%s::buzz%d", dev_name(&hdev->dev), n + 1);
562 		led->name = name;
563 		led->brightness = 0;
564 		led->max_brightness = 1;
565 		led->brightness_get = buzz_led_get_brightness;
566 		led->brightness_set = buzz_led_set_brightness;
567 
568 		if (led_classdev_register(&hdev->dev, led)) {
569 			hid_err(hdev, "Failed to register LED %d\n", n);
570 			kfree(led);
571 			goto error_leds;
572 		}
573 
574 		buzz->leds[n] = led;
575 	}
576 
577 	return ret;
578 
579 error_leds:
580 	for (n = 0; n < 4; n++) {
581 		led = buzz->leds[n];
582 		buzz->leds[n] = NULL;
583 		if (!led)
584 			continue;
585 		led_classdev_unregister(led);
586 		kfree(led);
587 	}
588 
589 	kfree(drv_data->extra);
590 	drv_data->extra = NULL;
591 	return ret;
592 }
593 
594 static void buzz_remove(struct hid_device *hdev)
595 {
596 	struct sony_sc *drv_data;
597 	struct buzz_extra *buzz;
598 	struct led_classdev *led;
599 	int n;
600 
601 	drv_data = hid_get_drvdata(hdev);
602 	BUG_ON(!(drv_data->quirks & BUZZ_CONTROLLER));
603 
604 	buzz = drv_data->extra;
605 
606 	for (n = 0; n < 4; n++) {
607 		led = buzz->leds[n];
608 		buzz->leds[n] = NULL;
609 		if (!led)
610 			continue;
611 		led_classdev_unregister(led);
612 		kfree(led);
613 	}
614 
615 	kfree(drv_data->extra);
616 	drv_data->extra = NULL;
617 }
618 
619 static int sony_probe(struct hid_device *hdev, const struct hid_device_id *id)
620 {
621 	int ret;
622 	unsigned long quirks = id->driver_data;
623 	struct sony_sc *sc;
624 	unsigned int connect_mask = HID_CONNECT_DEFAULT;
625 
626 	sc = kzalloc(sizeof(*sc), GFP_KERNEL);
627 	if (sc == NULL) {
628 		hid_err(hdev, "can't alloc sony descriptor\n");
629 		return -ENOMEM;
630 	}
631 
632 	sc->quirks = quirks;
633 	hid_set_drvdata(hdev, sc);
634 
635 	ret = hid_parse(hdev);
636 	if (ret) {
637 		hid_err(hdev, "parse failed\n");
638 		goto err_free;
639 	}
640 
641 	if (sc->quirks & VAIO_RDESC_CONSTANT)
642 		connect_mask |= HID_CONNECT_HIDDEV_FORCE;
643 	else if (sc->quirks & SIXAXIS_CONTROLLER_USB)
644 		connect_mask |= HID_CONNECT_HIDDEV_FORCE;
645 	else if (sc->quirks & SIXAXIS_CONTROLLER_BT)
646 		connect_mask |= HID_CONNECT_HIDDEV_FORCE;
647 
648 	ret = hid_hw_start(hdev, connect_mask);
649 	if (ret) {
650 		hid_err(hdev, "hw start failed\n");
651 		goto err_free;
652 	}
653 
654 	if (sc->quirks & SIXAXIS_CONTROLLER_USB) {
655 		hdev->hid_output_raw_report = sixaxis_usb_output_raw_report;
656 		ret = sixaxis_set_operational_usb(hdev);
657 	}
658 	else if (sc->quirks & SIXAXIS_CONTROLLER_BT)
659 		ret = sixaxis_set_operational_bt(hdev);
660 	else if (sc->quirks & BUZZ_CONTROLLER)
661 		ret = buzz_init(hdev);
662 	else
663 		ret = 0;
664 
665 	if (ret < 0)
666 		goto err_stop;
667 
668 	return 0;
669 err_stop:
670 	hid_hw_stop(hdev);
671 err_free:
672 	kfree(sc);
673 	return ret;
674 }
675 
676 static void sony_remove(struct hid_device *hdev)
677 {
678 	struct sony_sc *sc = hid_get_drvdata(hdev);
679 
680 	if (sc->quirks & BUZZ_CONTROLLER)
681 		buzz_remove(hdev);
682 
683 	hid_hw_stop(hdev);
684 	kfree(sc);
685 }
686 
687 static const struct hid_device_id sony_devices[] = {
688 	{ HID_USB_DEVICE(USB_VENDOR_ID_SONY, USB_DEVICE_ID_SONY_PS3_CONTROLLER),
689 		.driver_data = SIXAXIS_CONTROLLER_USB },
690 	{ HID_USB_DEVICE(USB_VENDOR_ID_SONY, USB_DEVICE_ID_SONY_NAVIGATION_CONTROLLER),
691 		.driver_data = SIXAXIS_CONTROLLER_USB },
692 	{ HID_BLUETOOTH_DEVICE(USB_VENDOR_ID_SONY, USB_DEVICE_ID_SONY_PS3_CONTROLLER),
693 		.driver_data = SIXAXIS_CONTROLLER_BT },
694 	{ HID_USB_DEVICE(USB_VENDOR_ID_SONY, USB_DEVICE_ID_SONY_VAIO_VGX_MOUSE),
695 		.driver_data = VAIO_RDESC_CONSTANT },
696 	{ HID_USB_DEVICE(USB_VENDOR_ID_SONY, USB_DEVICE_ID_SONY_VAIO_VGP_MOUSE),
697 		.driver_data = VAIO_RDESC_CONSTANT },
698 	/* Wired Buzz Controller. Reported as Sony Hub from its USB ID and as
699 	 * Logitech joystick from the device descriptor. */
700 	{ HID_USB_DEVICE(USB_VENDOR_ID_SONY, USB_DEVICE_ID_SONY_BUZZ_CONTROLLER),
701 		.driver_data = BUZZ_CONTROLLER },
702 	{ HID_USB_DEVICE(USB_VENDOR_ID_SONY, USB_DEVICE_ID_SONY_WIRELESS_BUZZ_CONTROLLER),
703 		.driver_data = BUZZ_CONTROLLER },
704 	/* PS3 BD Remote Control */
705 	{ HID_BLUETOOTH_DEVICE(USB_VENDOR_ID_SONY, USB_DEVICE_ID_SONY_PS3_BDREMOTE),
706 		.driver_data = PS3REMOTE },
707 	/* Logitech Harmony Adapter for PS3 */
708 	{ HID_BLUETOOTH_DEVICE(USB_VENDOR_ID_LOGITECH, USB_DEVICE_ID_LOGITECH_HARMONY_PS3),
709 		.driver_data = PS3REMOTE },
710 	{ }
711 };
712 MODULE_DEVICE_TABLE(hid, sony_devices);
713 
714 static struct hid_driver sony_driver = {
715 	.name          = "sony",
716 	.id_table      = sony_devices,
717 	.input_mapping = sony_mapping,
718 	.probe         = sony_probe,
719 	.remove        = sony_remove,
720 	.report_fixup  = sony_report_fixup,
721 	.raw_event     = sony_raw_event
722 };
723 module_hid_driver(sony_driver);
724 
725 MODULE_LICENSE("GPL");
726