xref: /linux/drivers/hid/hid-magicmouse.c (revision 3f0a50f345f78183f6e9b39c2f45ca5dcaa511ca)
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  *   Apple "Magic" Wireless Mouse driver
4  *
5  *   Copyright (c) 2010 Michael Poole <mdpoole@troilus.org>
6  *   Copyright (c) 2010 Chase Douglas <chase.douglas@canonical.com>
7  */
8 
9 /*
10  */
11 
12 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
13 
14 #include <linux/device.h>
15 #include <linux/hid.h>
16 #include <linux/input/mt.h>
17 #include <linux/module.h>
18 #include <linux/slab.h>
19 #include <linux/workqueue.h>
20 
21 #include "hid-ids.h"
22 
23 static bool emulate_3button = true;
24 module_param(emulate_3button, bool, 0644);
25 MODULE_PARM_DESC(emulate_3button, "Emulate a middle button");
26 
27 static int middle_button_start = -350;
28 static int middle_button_stop = +350;
29 
30 static bool emulate_scroll_wheel = true;
31 module_param(emulate_scroll_wheel, bool, 0644);
32 MODULE_PARM_DESC(emulate_scroll_wheel, "Emulate a scroll wheel");
33 
34 static unsigned int scroll_speed = 32;
35 static int param_set_scroll_speed(const char *val,
36 				  const struct kernel_param *kp) {
37 	unsigned long speed;
38 	if (!val || kstrtoul(val, 0, &speed) || speed > 63)
39 		return -EINVAL;
40 	scroll_speed = speed;
41 	return 0;
42 }
43 module_param_call(scroll_speed, param_set_scroll_speed, param_get_uint, &scroll_speed, 0644);
44 MODULE_PARM_DESC(scroll_speed, "Scroll speed, value from 0 (slow) to 63 (fast)");
45 
46 static bool scroll_acceleration = false;
47 module_param(scroll_acceleration, bool, 0644);
48 MODULE_PARM_DESC(scroll_acceleration, "Accelerate sequential scroll events");
49 
50 static bool report_undeciphered;
51 module_param(report_undeciphered, bool, 0644);
52 MODULE_PARM_DESC(report_undeciphered, "Report undeciphered multi-touch state field using a MSC_RAW event");
53 
54 #define TRACKPAD2_2021_BT_VERSION 0x110
55 
56 #define TRACKPAD_REPORT_ID 0x28
57 #define TRACKPAD2_USB_REPORT_ID 0x02
58 #define TRACKPAD2_BT_REPORT_ID 0x31
59 #define MOUSE_REPORT_ID    0x29
60 #define MOUSE2_REPORT_ID   0x12
61 #define DOUBLE_REPORT_ID   0xf7
62 #define USB_BATTERY_TIMEOUT_MS 60000
63 
64 /* These definitions are not precise, but they're close enough.  (Bits
65  * 0x03 seem to indicate the aspect ratio of the touch, bits 0x70 seem
66  * to be some kind of bit mask -- 0x20 may be a near-field reading,
67  * and 0x40 is actual contact, and 0x10 may be a start/stop or change
68  * indication.)
69  */
70 #define TOUCH_STATE_MASK  0xf0
71 #define TOUCH_STATE_NONE  0x00
72 #define TOUCH_STATE_START 0x30
73 #define TOUCH_STATE_DRAG  0x40
74 
75 /* Number of high-resolution events for each low-resolution detent. */
76 #define SCROLL_HR_STEPS 10
77 #define SCROLL_HR_MULT (120 / SCROLL_HR_STEPS)
78 #define SCROLL_HR_THRESHOLD 90 /* units */
79 #define SCROLL_ACCEL_DEFAULT 7
80 
81 /* Touch surface information. Dimension is in hundredths of a mm, min and max
82  * are in units. */
83 #define MOUSE_DIMENSION_X (float)9056
84 #define MOUSE_MIN_X -1100
85 #define MOUSE_MAX_X 1258
86 #define MOUSE_RES_X ((MOUSE_MAX_X - MOUSE_MIN_X) / (MOUSE_DIMENSION_X / 100))
87 #define MOUSE_DIMENSION_Y (float)5152
88 #define MOUSE_MIN_Y -1589
89 #define MOUSE_MAX_Y 2047
90 #define MOUSE_RES_Y ((MOUSE_MAX_Y - MOUSE_MIN_Y) / (MOUSE_DIMENSION_Y / 100))
91 
92 #define TRACKPAD_DIMENSION_X (float)13000
93 #define TRACKPAD_MIN_X -2909
94 #define TRACKPAD_MAX_X 3167
95 #define TRACKPAD_RES_X \
96 	((TRACKPAD_MAX_X - TRACKPAD_MIN_X) / (TRACKPAD_DIMENSION_X / 100))
97 #define TRACKPAD_DIMENSION_Y (float)11000
98 #define TRACKPAD_MIN_Y -2456
99 #define TRACKPAD_MAX_Y 2565
100 #define TRACKPAD_RES_Y \
101 	((TRACKPAD_MAX_Y - TRACKPAD_MIN_Y) / (TRACKPAD_DIMENSION_Y / 100))
102 
103 #define TRACKPAD2_DIMENSION_X (float)16000
104 #define TRACKPAD2_MIN_X -3678
105 #define TRACKPAD2_MAX_X 3934
106 #define TRACKPAD2_RES_X \
107 	((TRACKPAD2_MAX_X - TRACKPAD2_MIN_X) / (TRACKPAD2_DIMENSION_X / 100))
108 #define TRACKPAD2_DIMENSION_Y (float)11490
109 #define TRACKPAD2_MIN_Y -2478
110 #define TRACKPAD2_MAX_Y 2587
111 #define TRACKPAD2_RES_Y \
112 	((TRACKPAD2_MAX_Y - TRACKPAD2_MIN_Y) / (TRACKPAD2_DIMENSION_Y / 100))
113 
114 /**
115  * struct magicmouse_sc - Tracks Magic Mouse-specific data.
116  * @input: Input device through which we report events.
117  * @quirks: Currently unused.
118  * @ntouches: Number of touches in most recent touch report.
119  * @scroll_accel: Number of consecutive scroll motions.
120  * @scroll_jiffies: Time of last scroll motion.
121  * @touches: Most recent data for a touch, indexed by tracking ID.
122  * @tracking_ids: Mapping of current touch input data to @touches.
123  */
124 struct magicmouse_sc {
125 	struct input_dev *input;
126 	unsigned long quirks;
127 
128 	int ntouches;
129 	int scroll_accel;
130 	unsigned long scroll_jiffies;
131 
132 	struct {
133 		short x;
134 		short y;
135 		short scroll_x;
136 		short scroll_y;
137 		short scroll_x_hr;
138 		short scroll_y_hr;
139 		u8 size;
140 		bool scroll_x_active;
141 		bool scroll_y_active;
142 	} touches[16];
143 	int tracking_ids[16];
144 
145 	struct hid_device *hdev;
146 	struct delayed_work work;
147 	struct timer_list battery_timer;
148 };
149 
150 static int magicmouse_firm_touch(struct magicmouse_sc *msc)
151 {
152 	int touch = -1;
153 	int ii;
154 
155 	/* If there is only one "firm" touch, set touch to its
156 	 * tracking ID.
157 	 */
158 	for (ii = 0; ii < msc->ntouches; ii++) {
159 		int idx = msc->tracking_ids[ii];
160 		if (msc->touches[idx].size < 8) {
161 			/* Ignore this touch. */
162 		} else if (touch >= 0) {
163 			touch = -1;
164 			break;
165 		} else {
166 			touch = idx;
167 		}
168 	}
169 
170 	return touch;
171 }
172 
173 static void magicmouse_emit_buttons(struct magicmouse_sc *msc, int state)
174 {
175 	int last_state = test_bit(BTN_LEFT, msc->input->key) << 0 |
176 		test_bit(BTN_RIGHT, msc->input->key) << 1 |
177 		test_bit(BTN_MIDDLE, msc->input->key) << 2;
178 
179 	if (emulate_3button) {
180 		int id;
181 
182 		/* If some button was pressed before, keep it held
183 		 * down.  Otherwise, if there's exactly one firm
184 		 * touch, use that to override the mouse's guess.
185 		 */
186 		if (state == 0) {
187 			/* The button was released. */
188 		} else if (last_state != 0) {
189 			state = last_state;
190 		} else if ((id = magicmouse_firm_touch(msc)) >= 0) {
191 			int x = msc->touches[id].x;
192 			if (x < middle_button_start)
193 				state = 1;
194 			else if (x > middle_button_stop)
195 				state = 2;
196 			else
197 				state = 4;
198 		} /* else: we keep the mouse's guess */
199 
200 		input_report_key(msc->input, BTN_MIDDLE, state & 4);
201 	}
202 
203 	input_report_key(msc->input, BTN_LEFT, state & 1);
204 	input_report_key(msc->input, BTN_RIGHT, state & 2);
205 
206 	if (state != last_state)
207 		msc->scroll_accel = SCROLL_ACCEL_DEFAULT;
208 }
209 
210 static void magicmouse_emit_touch(struct magicmouse_sc *msc, int raw_id, u8 *tdata)
211 {
212 	struct input_dev *input = msc->input;
213 	int id, x, y, size, orientation, touch_major, touch_minor, state, down;
214 	int pressure = 0;
215 
216 	if (input->id.product == USB_DEVICE_ID_APPLE_MAGICMOUSE ||
217 	    input->id.product == USB_DEVICE_ID_APPLE_MAGICMOUSE2) {
218 		id = (tdata[6] << 2 | tdata[5] >> 6) & 0xf;
219 		x = (tdata[1] << 28 | tdata[0] << 20) >> 20;
220 		y = -((tdata[2] << 24 | tdata[1] << 16) >> 20);
221 		size = tdata[5] & 0x3f;
222 		orientation = (tdata[6] >> 2) - 32;
223 		touch_major = tdata[3];
224 		touch_minor = tdata[4];
225 		state = tdata[7] & TOUCH_STATE_MASK;
226 		down = state != TOUCH_STATE_NONE;
227 	} else if (input->id.product == USB_DEVICE_ID_APPLE_MAGICTRACKPAD2) {
228 		id = tdata[8] & 0xf;
229 		x = (tdata[1] << 27 | tdata[0] << 19) >> 19;
230 		y = -((tdata[3] << 30 | tdata[2] << 22 | tdata[1] << 14) >> 19);
231 		size = tdata[6];
232 		orientation = (tdata[8] >> 5) - 4;
233 		touch_major = tdata[4];
234 		touch_minor = tdata[5];
235 		pressure = tdata[7];
236 		state = tdata[3] & 0xC0;
237 		down = state == 0x80;
238 	} else { /* USB_DEVICE_ID_APPLE_MAGICTRACKPAD */
239 		id = (tdata[7] << 2 | tdata[6] >> 6) & 0xf;
240 		x = (tdata[1] << 27 | tdata[0] << 19) >> 19;
241 		y = -((tdata[3] << 30 | tdata[2] << 22 | tdata[1] << 14) >> 19);
242 		size = tdata[6] & 0x3f;
243 		orientation = (tdata[7] >> 2) - 32;
244 		touch_major = tdata[4];
245 		touch_minor = tdata[5];
246 		state = tdata[8] & TOUCH_STATE_MASK;
247 		down = state != TOUCH_STATE_NONE;
248 	}
249 
250 	/* Store tracking ID and other fields. */
251 	msc->tracking_ids[raw_id] = id;
252 	msc->touches[id].x = x;
253 	msc->touches[id].y = y;
254 	msc->touches[id].size = size;
255 
256 	/* If requested, emulate a scroll wheel by detecting small
257 	 * vertical touch motions.
258 	 */
259 	if (emulate_scroll_wheel && (input->id.product !=
260 			USB_DEVICE_ID_APPLE_MAGICTRACKPAD2)) {
261 		unsigned long now = jiffies;
262 		int step_x = msc->touches[id].scroll_x - x;
263 		int step_y = msc->touches[id].scroll_y - y;
264 		int step_hr =
265 			max_t(int,
266 			      ((64 - (int)scroll_speed) * msc->scroll_accel) /
267 					SCROLL_HR_STEPS,
268 			      1);
269 		int step_x_hr = msc->touches[id].scroll_x_hr - x;
270 		int step_y_hr = msc->touches[id].scroll_y_hr - y;
271 
272 		/* Calculate and apply the scroll motion. */
273 		switch (state) {
274 		case TOUCH_STATE_START:
275 			msc->touches[id].scroll_x = x;
276 			msc->touches[id].scroll_y = y;
277 			msc->touches[id].scroll_x_hr = x;
278 			msc->touches[id].scroll_y_hr = y;
279 			msc->touches[id].scroll_x_active = false;
280 			msc->touches[id].scroll_y_active = false;
281 
282 			/* Reset acceleration after half a second. */
283 			if (scroll_acceleration && time_before(now,
284 						msc->scroll_jiffies + HZ / 2))
285 				msc->scroll_accel = max_t(int,
286 						msc->scroll_accel - 1, 1);
287 			else
288 				msc->scroll_accel = SCROLL_ACCEL_DEFAULT;
289 
290 			break;
291 		case TOUCH_STATE_DRAG:
292 			step_x /= (64 - (int)scroll_speed) * msc->scroll_accel;
293 			if (step_x != 0) {
294 				msc->touches[id].scroll_x -= step_x *
295 					(64 - scroll_speed) * msc->scroll_accel;
296 				msc->scroll_jiffies = now;
297 				input_report_rel(input, REL_HWHEEL, -step_x);
298 			}
299 
300 			step_y /= (64 - (int)scroll_speed) * msc->scroll_accel;
301 			if (step_y != 0) {
302 				msc->touches[id].scroll_y -= step_y *
303 					(64 - scroll_speed) * msc->scroll_accel;
304 				msc->scroll_jiffies = now;
305 				input_report_rel(input, REL_WHEEL, step_y);
306 			}
307 
308 			if (!msc->touches[id].scroll_x_active &&
309 			    abs(step_x_hr) > SCROLL_HR_THRESHOLD) {
310 				msc->touches[id].scroll_x_active = true;
311 				msc->touches[id].scroll_x_hr = x;
312 				step_x_hr = 0;
313 			}
314 
315 			step_x_hr /= step_hr;
316 			if (step_x_hr != 0 &&
317 			    msc->touches[id].scroll_x_active) {
318 				msc->touches[id].scroll_x_hr -= step_x_hr *
319 					step_hr;
320 				input_report_rel(input,
321 						 REL_HWHEEL_HI_RES,
322 						 -step_x_hr * SCROLL_HR_MULT);
323 			}
324 
325 			if (!msc->touches[id].scroll_y_active &&
326 			    abs(step_y_hr) > SCROLL_HR_THRESHOLD) {
327 				msc->touches[id].scroll_y_active = true;
328 				msc->touches[id].scroll_y_hr = y;
329 				step_y_hr = 0;
330 			}
331 
332 			step_y_hr /= step_hr;
333 			if (step_y_hr != 0 &&
334 			    msc->touches[id].scroll_y_active) {
335 				msc->touches[id].scroll_y_hr -= step_y_hr *
336 					step_hr;
337 				input_report_rel(input,
338 						 REL_WHEEL_HI_RES,
339 						 step_y_hr * SCROLL_HR_MULT);
340 			}
341 			break;
342 		}
343 	}
344 
345 	if (down)
346 		msc->ntouches++;
347 
348 	input_mt_slot(input, id);
349 	input_mt_report_slot_state(input, MT_TOOL_FINGER, down);
350 
351 	/* Generate the input events for this touch. */
352 	if (down) {
353 		input_report_abs(input, ABS_MT_TOUCH_MAJOR, touch_major << 2);
354 		input_report_abs(input, ABS_MT_TOUCH_MINOR, touch_minor << 2);
355 		input_report_abs(input, ABS_MT_ORIENTATION, -orientation);
356 		input_report_abs(input, ABS_MT_POSITION_X, x);
357 		input_report_abs(input, ABS_MT_POSITION_Y, y);
358 
359 		if (input->id.product == USB_DEVICE_ID_APPLE_MAGICTRACKPAD2)
360 			input_report_abs(input, ABS_MT_PRESSURE, pressure);
361 
362 		if (report_undeciphered) {
363 			if (input->id.product == USB_DEVICE_ID_APPLE_MAGICMOUSE ||
364 			    input->id.product == USB_DEVICE_ID_APPLE_MAGICMOUSE2)
365 				input_event(input, EV_MSC, MSC_RAW, tdata[7]);
366 			else if (input->id.product !=
367 					USB_DEVICE_ID_APPLE_MAGICTRACKPAD2)
368 				input_event(input, EV_MSC, MSC_RAW, tdata[8]);
369 		}
370 	}
371 }
372 
373 static int magicmouse_raw_event(struct hid_device *hdev,
374 		struct hid_report *report, u8 *data, int size)
375 {
376 	struct magicmouse_sc *msc = hid_get_drvdata(hdev);
377 	struct input_dev *input = msc->input;
378 	int x = 0, y = 0, ii, clicks = 0, npoints;
379 
380 	switch (data[0]) {
381 	case TRACKPAD_REPORT_ID:
382 	case TRACKPAD2_BT_REPORT_ID:
383 		/* Expect four bytes of prefix, and N*9 bytes of touch data. */
384 		if (size < 4 || ((size - 4) % 9) != 0)
385 			return 0;
386 		npoints = (size - 4) / 9;
387 		if (npoints > 15) {
388 			hid_warn(hdev, "invalid size value (%d) for TRACKPAD_REPORT_ID\n",
389 					size);
390 			return 0;
391 		}
392 		msc->ntouches = 0;
393 		for (ii = 0; ii < npoints; ii++)
394 			magicmouse_emit_touch(msc, ii, data + ii * 9 + 4);
395 
396 		clicks = data[1];
397 
398 		/* The following bits provide a device specific timestamp. They
399 		 * are unused here.
400 		 *
401 		 * ts = data[1] >> 6 | data[2] << 2 | data[3] << 10;
402 		 */
403 		break;
404 	case TRACKPAD2_USB_REPORT_ID:
405 		/* Expect twelve bytes of prefix and N*9 bytes of touch data. */
406 		if (size < 12 || ((size - 12) % 9) != 0)
407 			return 0;
408 		npoints = (size - 12) / 9;
409 		if (npoints > 15) {
410 			hid_warn(hdev, "invalid size value (%d) for TRACKPAD2_USB_REPORT_ID\n",
411 					size);
412 			return 0;
413 		}
414 		msc->ntouches = 0;
415 		for (ii = 0; ii < npoints; ii++)
416 			magicmouse_emit_touch(msc, ii, data + ii * 9 + 12);
417 
418 		clicks = data[1];
419 		break;
420 	case MOUSE_REPORT_ID:
421 		/* Expect six bytes of prefix, and N*8 bytes of touch data. */
422 		if (size < 6 || ((size - 6) % 8) != 0)
423 			return 0;
424 		npoints = (size - 6) / 8;
425 		if (npoints > 15) {
426 			hid_warn(hdev, "invalid size value (%d) for MOUSE_REPORT_ID\n",
427 					size);
428 			return 0;
429 		}
430 		msc->ntouches = 0;
431 		for (ii = 0; ii < npoints; ii++)
432 			magicmouse_emit_touch(msc, ii, data + ii * 8 + 6);
433 
434 		/* When emulating three-button mode, it is important
435 		 * to have the current touch information before
436 		 * generating a click event.
437 		 */
438 		x = (int)(((data[3] & 0x0c) << 28) | (data[1] << 22)) >> 22;
439 		y = (int)(((data[3] & 0x30) << 26) | (data[2] << 22)) >> 22;
440 		clicks = data[3];
441 
442 		/* The following bits provide a device specific timestamp. They
443 		 * are unused here.
444 		 *
445 		 * ts = data[3] >> 6 | data[4] << 2 | data[5] << 10;
446 		 */
447 		break;
448 	case MOUSE2_REPORT_ID:
449 		/* Size is either 8 or (14 + 8 * N) */
450 		if (size != 8 && (size < 14 || (size - 14) % 8 != 0))
451 			return 0;
452 		npoints = (size - 14) / 8;
453 		if (npoints > 15) {
454 			hid_warn(hdev, "invalid size value (%d) for MOUSE2_REPORT_ID\n",
455 					size);
456 			return 0;
457 		}
458 		msc->ntouches = 0;
459 		for (ii = 0; ii < npoints; ii++)
460 			magicmouse_emit_touch(msc, ii, data + ii * 8 + 14);
461 
462 		/* When emulating three-button mode, it is important
463 		 * to have the current touch information before
464 		 * generating a click event.
465 		 */
466 		x = (int)((data[3] << 24) | (data[2] << 16)) >> 16;
467 		y = (int)((data[5] << 24) | (data[4] << 16)) >> 16;
468 		clicks = data[1];
469 
470 		/* The following bits provide a device specific timestamp. They
471 		 * are unused here.
472 		 *
473 		 * ts = data[11] >> 6 | data[12] << 2 | data[13] << 10;
474 		 */
475 		break;
476 	case DOUBLE_REPORT_ID:
477 		/* Sometimes the trackpad sends two touch reports in one
478 		 * packet.
479 		 */
480 		magicmouse_raw_event(hdev, report, data + 2, data[1]);
481 		magicmouse_raw_event(hdev, report, data + 2 + data[1],
482 			size - 2 - data[1]);
483 		break;
484 	default:
485 		return 0;
486 	}
487 
488 	if (input->id.product == USB_DEVICE_ID_APPLE_MAGICMOUSE ||
489 	    input->id.product == USB_DEVICE_ID_APPLE_MAGICMOUSE2) {
490 		magicmouse_emit_buttons(msc, clicks & 3);
491 		input_report_rel(input, REL_X, x);
492 		input_report_rel(input, REL_Y, y);
493 	} else if (input->id.product == USB_DEVICE_ID_APPLE_MAGICTRACKPAD2) {
494 		input_mt_sync_frame(input);
495 		input_report_key(input, BTN_MOUSE, clicks & 1);
496 	} else { /* USB_DEVICE_ID_APPLE_MAGICTRACKPAD */
497 		input_report_key(input, BTN_MOUSE, clicks & 1);
498 		input_mt_report_pointer_emulation(input, true);
499 	}
500 
501 	input_sync(input);
502 	return 1;
503 }
504 
505 static int magicmouse_event(struct hid_device *hdev, struct hid_field *field,
506 		struct hid_usage *usage, __s32 value)
507 {
508 	struct magicmouse_sc *msc = hid_get_drvdata(hdev);
509 	if (msc->input->id.product == USB_DEVICE_ID_APPLE_MAGICMOUSE2 &&
510 	    field->report->id == MOUSE2_REPORT_ID) {
511 		/*
512 		 * magic_mouse_raw_event has done all the work. Skip hidinput.
513 		 *
514 		 * Specifically, hidinput may modify BTN_LEFT and BTN_RIGHT,
515 		 * breaking emulate_3button.
516 		 */
517 		return 1;
518 	}
519 	return 0;
520 }
521 
522 static int magicmouse_setup_input(struct input_dev *input, struct hid_device *hdev)
523 {
524 	int error;
525 	int mt_flags = 0;
526 
527 	__set_bit(EV_KEY, input->evbit);
528 
529 	if (input->id.product == USB_DEVICE_ID_APPLE_MAGICMOUSE ||
530 	    input->id.product == USB_DEVICE_ID_APPLE_MAGICMOUSE2) {
531 		__set_bit(BTN_LEFT, input->keybit);
532 		__set_bit(BTN_RIGHT, input->keybit);
533 		if (emulate_3button)
534 			__set_bit(BTN_MIDDLE, input->keybit);
535 
536 		__set_bit(EV_REL, input->evbit);
537 		__set_bit(REL_X, input->relbit);
538 		__set_bit(REL_Y, input->relbit);
539 		if (emulate_scroll_wheel) {
540 			__set_bit(REL_WHEEL, input->relbit);
541 			__set_bit(REL_HWHEEL, input->relbit);
542 			__set_bit(REL_WHEEL_HI_RES, input->relbit);
543 			__set_bit(REL_HWHEEL_HI_RES, input->relbit);
544 		}
545 	} else if (input->id.product == USB_DEVICE_ID_APPLE_MAGICTRACKPAD2) {
546 		/* If the trackpad has been connected to a Mac, the name is
547 		 * automatically personalized, e.g., "José Expósito's Trackpad".
548 		 * When connected through Bluetooth, the personalized name is
549 		 * reported, however, when connected through USB the generic
550 		 * name is reported.
551 		 * Set the device name to ensure the same driver settings get
552 		 * loaded, whether connected through bluetooth or USB.
553 		 */
554 		if (hdev->vendor == BT_VENDOR_ID_APPLE) {
555 			if (input->id.version == TRACKPAD2_2021_BT_VERSION)
556 				input->name = "Apple Inc. Magic Trackpad";
557 			else
558 				input->name = "Apple Inc. Magic Trackpad 2";
559 		} else { /* USB_VENDOR_ID_APPLE */
560 			input->name = hdev->name;
561 		}
562 
563 		__clear_bit(EV_MSC, input->evbit);
564 		__clear_bit(BTN_0, input->keybit);
565 		__clear_bit(BTN_RIGHT, input->keybit);
566 		__clear_bit(BTN_MIDDLE, input->keybit);
567 		__set_bit(BTN_MOUSE, input->keybit);
568 		__set_bit(INPUT_PROP_BUTTONPAD, input->propbit);
569 		__set_bit(BTN_TOOL_FINGER, input->keybit);
570 
571 		mt_flags = INPUT_MT_POINTER | INPUT_MT_DROP_UNUSED |
572 				INPUT_MT_TRACK;
573 	} else { /* USB_DEVICE_ID_APPLE_MAGICTRACKPAD */
574 		/* input->keybit is initialized with incorrect button info
575 		 * for Magic Trackpad. There really is only one physical
576 		 * button (BTN_LEFT == BTN_MOUSE). Make sure we don't
577 		 * advertise buttons that don't exist...
578 		 */
579 		__clear_bit(BTN_RIGHT, input->keybit);
580 		__clear_bit(BTN_MIDDLE, input->keybit);
581 		__set_bit(BTN_MOUSE, input->keybit);
582 		__set_bit(BTN_TOOL_FINGER, input->keybit);
583 		__set_bit(BTN_TOOL_DOUBLETAP, input->keybit);
584 		__set_bit(BTN_TOOL_TRIPLETAP, input->keybit);
585 		__set_bit(BTN_TOOL_QUADTAP, input->keybit);
586 		__set_bit(BTN_TOOL_QUINTTAP, input->keybit);
587 		__set_bit(BTN_TOUCH, input->keybit);
588 		__set_bit(INPUT_PROP_POINTER, input->propbit);
589 		__set_bit(INPUT_PROP_BUTTONPAD, input->propbit);
590 	}
591 
592 
593 	__set_bit(EV_ABS, input->evbit);
594 
595 	error = input_mt_init_slots(input, 16, mt_flags);
596 	if (error)
597 		return error;
598 	input_set_abs_params(input, ABS_MT_TOUCH_MAJOR, 0, 255 << 2,
599 			     4, 0);
600 	input_set_abs_params(input, ABS_MT_TOUCH_MINOR, 0, 255 << 2,
601 			     4, 0);
602 
603 	/* Note: Touch Y position from the device is inverted relative
604 	 * to how pointer motion is reported (and relative to how USB
605 	 * HID recommends the coordinates work).  This driver keeps
606 	 * the origin at the same position, and just uses the additive
607 	 * inverse of the reported Y.
608 	 */
609 	if (input->id.product == USB_DEVICE_ID_APPLE_MAGICMOUSE ||
610 	    input->id.product == USB_DEVICE_ID_APPLE_MAGICMOUSE2) {
611 		input_set_abs_params(input, ABS_MT_ORIENTATION, -31, 32, 1, 0);
612 		input_set_abs_params(input, ABS_MT_POSITION_X,
613 				     MOUSE_MIN_X, MOUSE_MAX_X, 4, 0);
614 		input_set_abs_params(input, ABS_MT_POSITION_Y,
615 				     MOUSE_MIN_Y, MOUSE_MAX_Y, 4, 0);
616 
617 		input_abs_set_res(input, ABS_MT_POSITION_X,
618 				  MOUSE_RES_X);
619 		input_abs_set_res(input, ABS_MT_POSITION_Y,
620 				  MOUSE_RES_Y);
621 	} else if (input->id.product ==  USB_DEVICE_ID_APPLE_MAGICTRACKPAD2) {
622 		input_set_abs_params(input, ABS_MT_PRESSURE, 0, 253, 0, 0);
623 		input_set_abs_params(input, ABS_PRESSURE, 0, 253, 0, 0);
624 		input_set_abs_params(input, ABS_MT_ORIENTATION, -3, 4, 0, 0);
625 		input_set_abs_params(input, ABS_X, TRACKPAD2_MIN_X,
626 				     TRACKPAD2_MAX_X, 0, 0);
627 		input_set_abs_params(input, ABS_Y, TRACKPAD2_MIN_Y,
628 				     TRACKPAD2_MAX_Y, 0, 0);
629 		input_set_abs_params(input, ABS_MT_POSITION_X,
630 				     TRACKPAD2_MIN_X, TRACKPAD2_MAX_X, 0, 0);
631 		input_set_abs_params(input, ABS_MT_POSITION_Y,
632 				     TRACKPAD2_MIN_Y, TRACKPAD2_MAX_Y, 0, 0);
633 
634 		input_abs_set_res(input, ABS_X, TRACKPAD2_RES_X);
635 		input_abs_set_res(input, ABS_Y, TRACKPAD2_RES_Y);
636 		input_abs_set_res(input, ABS_MT_POSITION_X, TRACKPAD2_RES_X);
637 		input_abs_set_res(input, ABS_MT_POSITION_Y, TRACKPAD2_RES_Y);
638 	} else { /* USB_DEVICE_ID_APPLE_MAGICTRACKPAD */
639 		input_set_abs_params(input, ABS_MT_ORIENTATION, -31, 32, 1, 0);
640 		input_set_abs_params(input, ABS_X, TRACKPAD_MIN_X,
641 				     TRACKPAD_MAX_X, 4, 0);
642 		input_set_abs_params(input, ABS_Y, TRACKPAD_MIN_Y,
643 				     TRACKPAD_MAX_Y, 4, 0);
644 		input_set_abs_params(input, ABS_MT_POSITION_X,
645 				     TRACKPAD_MIN_X, TRACKPAD_MAX_X, 4, 0);
646 		input_set_abs_params(input, ABS_MT_POSITION_Y,
647 				     TRACKPAD_MIN_Y, TRACKPAD_MAX_Y, 4, 0);
648 
649 		input_abs_set_res(input, ABS_X, TRACKPAD_RES_X);
650 		input_abs_set_res(input, ABS_Y, TRACKPAD_RES_Y);
651 		input_abs_set_res(input, ABS_MT_POSITION_X,
652 				  TRACKPAD_RES_X);
653 		input_abs_set_res(input, ABS_MT_POSITION_Y,
654 				  TRACKPAD_RES_Y);
655 	}
656 
657 	input_set_events_per_packet(input, 60);
658 
659 	if (report_undeciphered &&
660 	    input->id.product != USB_DEVICE_ID_APPLE_MAGICTRACKPAD2) {
661 		__set_bit(EV_MSC, input->evbit);
662 		__set_bit(MSC_RAW, input->mscbit);
663 	}
664 
665 	/*
666 	 * hid-input may mark device as using autorepeat, but neither
667 	 * the trackpad, nor the mouse actually want it.
668 	 */
669 	__clear_bit(EV_REP, input->evbit);
670 
671 	return 0;
672 }
673 
674 static int magicmouse_input_mapping(struct hid_device *hdev,
675 		struct hid_input *hi, struct hid_field *field,
676 		struct hid_usage *usage, unsigned long **bit, int *max)
677 {
678 	struct magicmouse_sc *msc = hid_get_drvdata(hdev);
679 
680 	if (!msc->input)
681 		msc->input = hi->input;
682 
683 	/* Magic Trackpad does not give relative data after switching to MT */
684 	if ((hi->input->id.product == USB_DEVICE_ID_APPLE_MAGICTRACKPAD ||
685 	     hi->input->id.product == USB_DEVICE_ID_APPLE_MAGICTRACKPAD2) &&
686 	    field->flags & HID_MAIN_ITEM_RELATIVE)
687 		return -1;
688 
689 	return 0;
690 }
691 
692 static int magicmouse_input_configured(struct hid_device *hdev,
693 		struct hid_input *hi)
694 
695 {
696 	struct magicmouse_sc *msc = hid_get_drvdata(hdev);
697 	int ret;
698 
699 	ret = magicmouse_setup_input(msc->input, hdev);
700 	if (ret) {
701 		hid_err(hdev, "magicmouse setup input failed (%d)\n", ret);
702 		/* clean msc->input to notify probe() of the failure */
703 		msc->input = NULL;
704 		return ret;
705 	}
706 
707 	return 0;
708 }
709 
710 static int magicmouse_enable_multitouch(struct hid_device *hdev)
711 {
712 	const u8 *feature;
713 	const u8 feature_mt[] = { 0xD7, 0x01 };
714 	const u8 feature_mt_mouse2[] = { 0xF1, 0x02, 0x01 };
715 	const u8 feature_mt_trackpad2_usb[] = { 0x02, 0x01 };
716 	const u8 feature_mt_trackpad2_bt[] = { 0xF1, 0x02, 0x01 };
717 	u8 *buf;
718 	int ret;
719 	int feature_size;
720 
721 	if (hdev->product == USB_DEVICE_ID_APPLE_MAGICTRACKPAD2) {
722 		if (hdev->vendor == BT_VENDOR_ID_APPLE) {
723 			feature_size = sizeof(feature_mt_trackpad2_bt);
724 			feature = feature_mt_trackpad2_bt;
725 		} else { /* USB_VENDOR_ID_APPLE */
726 			feature_size = sizeof(feature_mt_trackpad2_usb);
727 			feature = feature_mt_trackpad2_usb;
728 		}
729 	} else if (hdev->product == USB_DEVICE_ID_APPLE_MAGICMOUSE2) {
730 		feature_size = sizeof(feature_mt_mouse2);
731 		feature = feature_mt_mouse2;
732 	} else {
733 		feature_size = sizeof(feature_mt);
734 		feature = feature_mt;
735 	}
736 
737 	buf = kmemdup(feature, feature_size, GFP_KERNEL);
738 	if (!buf)
739 		return -ENOMEM;
740 
741 	ret = hid_hw_raw_request(hdev, buf[0], buf, feature_size,
742 				HID_FEATURE_REPORT, HID_REQ_SET_REPORT);
743 	kfree(buf);
744 	return ret;
745 }
746 
747 static void magicmouse_enable_mt_work(struct work_struct *work)
748 {
749 	struct magicmouse_sc *msc =
750 		container_of(work, struct magicmouse_sc, work.work);
751 	int ret;
752 
753 	ret = magicmouse_enable_multitouch(msc->hdev);
754 	if (ret < 0)
755 		hid_err(msc->hdev, "unable to request touch data (%d)\n", ret);
756 }
757 
758 static int magicmouse_fetch_battery(struct hid_device *hdev)
759 {
760 #ifdef CONFIG_HID_BATTERY_STRENGTH
761 	struct hid_report_enum *report_enum;
762 	struct hid_report *report;
763 
764 	if (!hdev->battery || hdev->vendor != USB_VENDOR_ID_APPLE ||
765 	    (hdev->product != USB_DEVICE_ID_APPLE_MAGICMOUSE2 &&
766 	     hdev->product != USB_DEVICE_ID_APPLE_MAGICTRACKPAD2))
767 		return -1;
768 
769 	report_enum = &hdev->report_enum[hdev->battery_report_type];
770 	report = report_enum->report_id_hash[hdev->battery_report_id];
771 
772 	if (!report || report->maxfield < 1)
773 		return -1;
774 
775 	if (hdev->battery_capacity == hdev->battery_max)
776 		return -1;
777 
778 	hid_hw_request(hdev, report, HID_REQ_GET_REPORT);
779 	return 0;
780 #else
781 	return -1;
782 #endif
783 }
784 
785 static void magicmouse_battery_timer_tick(struct timer_list *t)
786 {
787 	struct magicmouse_sc *msc = from_timer(msc, t, battery_timer);
788 	struct hid_device *hdev = msc->hdev;
789 
790 	if (magicmouse_fetch_battery(hdev) == 0) {
791 		mod_timer(&msc->battery_timer,
792 			  jiffies + msecs_to_jiffies(USB_BATTERY_TIMEOUT_MS));
793 	}
794 }
795 
796 static int magicmouse_probe(struct hid_device *hdev,
797 	const struct hid_device_id *id)
798 {
799 	struct magicmouse_sc *msc;
800 	struct hid_report *report;
801 	int ret;
802 
803 	msc = devm_kzalloc(&hdev->dev, sizeof(*msc), GFP_KERNEL);
804 	if (msc == NULL) {
805 		hid_err(hdev, "can't alloc magicmouse descriptor\n");
806 		return -ENOMEM;
807 	}
808 
809 	msc->scroll_accel = SCROLL_ACCEL_DEFAULT;
810 	msc->hdev = hdev;
811 	INIT_DEFERRABLE_WORK(&msc->work, magicmouse_enable_mt_work);
812 
813 	msc->quirks = id->driver_data;
814 	hid_set_drvdata(hdev, msc);
815 
816 	ret = hid_parse(hdev);
817 	if (ret) {
818 		hid_err(hdev, "magicmouse hid parse failed\n");
819 		return ret;
820 	}
821 
822 	ret = hid_hw_start(hdev, HID_CONNECT_DEFAULT);
823 	if (ret) {
824 		hid_err(hdev, "magicmouse hw start failed\n");
825 		return ret;
826 	}
827 
828 	timer_setup(&msc->battery_timer, magicmouse_battery_timer_tick, 0);
829 	mod_timer(&msc->battery_timer,
830 		  jiffies + msecs_to_jiffies(USB_BATTERY_TIMEOUT_MS));
831 	magicmouse_fetch_battery(hdev);
832 
833 	if (id->vendor == USB_VENDOR_ID_APPLE &&
834 	    (id->product == USB_DEVICE_ID_APPLE_MAGICMOUSE2 ||
835 	     (id->product == USB_DEVICE_ID_APPLE_MAGICTRACKPAD2 && hdev->type != HID_TYPE_USBMOUSE)))
836 		return 0;
837 
838 	if (!msc->input) {
839 		hid_err(hdev, "magicmouse input not registered\n");
840 		ret = -ENOMEM;
841 		goto err_stop_hw;
842 	}
843 
844 	if (id->product == USB_DEVICE_ID_APPLE_MAGICMOUSE)
845 		report = hid_register_report(hdev, HID_INPUT_REPORT,
846 			MOUSE_REPORT_ID, 0);
847 	else if (id->product == USB_DEVICE_ID_APPLE_MAGICMOUSE2)
848 		report = hid_register_report(hdev, HID_INPUT_REPORT,
849 			MOUSE2_REPORT_ID, 0);
850 	else if (id->product == USB_DEVICE_ID_APPLE_MAGICTRACKPAD2) {
851 		if (id->vendor == BT_VENDOR_ID_APPLE)
852 			report = hid_register_report(hdev, HID_INPUT_REPORT,
853 				TRACKPAD2_BT_REPORT_ID, 0);
854 		else /* USB_VENDOR_ID_APPLE */
855 			report = hid_register_report(hdev, HID_INPUT_REPORT,
856 				TRACKPAD2_USB_REPORT_ID, 0);
857 	} else { /* USB_DEVICE_ID_APPLE_MAGICTRACKPAD */
858 		report = hid_register_report(hdev, HID_INPUT_REPORT,
859 			TRACKPAD_REPORT_ID, 0);
860 		report = hid_register_report(hdev, HID_INPUT_REPORT,
861 			DOUBLE_REPORT_ID, 0);
862 	}
863 
864 	if (!report) {
865 		hid_err(hdev, "unable to register touch report\n");
866 		ret = -ENOMEM;
867 		goto err_stop_hw;
868 	}
869 	report->size = 6;
870 
871 	/*
872 	 * Some devices repond with 'invalid report id' when feature
873 	 * report switching it into multitouch mode is sent to it.
874 	 *
875 	 * This results in -EIO from the _raw low-level transport callback,
876 	 * but there seems to be no other way of switching the mode.
877 	 * Thus the super-ugly hacky success check below.
878 	 */
879 	ret = magicmouse_enable_multitouch(hdev);
880 	if (ret != -EIO && ret < 0) {
881 		hid_err(hdev, "unable to request touch data (%d)\n", ret);
882 		goto err_stop_hw;
883 	}
884 	if (ret == -EIO && id->product == USB_DEVICE_ID_APPLE_MAGICMOUSE2) {
885 		schedule_delayed_work(&msc->work, msecs_to_jiffies(500));
886 	}
887 
888 	return 0;
889 err_stop_hw:
890 	del_timer_sync(&msc->battery_timer);
891 	hid_hw_stop(hdev);
892 	return ret;
893 }
894 
895 static void magicmouse_remove(struct hid_device *hdev)
896 {
897 	struct magicmouse_sc *msc = hid_get_drvdata(hdev);
898 
899 	if (msc) {
900 		cancel_delayed_work_sync(&msc->work);
901 		del_timer_sync(&msc->battery_timer);
902 	}
903 
904 	hid_hw_stop(hdev);
905 }
906 
907 static __u8 *magicmouse_report_fixup(struct hid_device *hdev, __u8 *rdesc,
908 				     unsigned int *rsize)
909 {
910 	/*
911 	 * Change the usage from:
912 	 *   0x06, 0x00, 0xff, // Usage Page (Vendor Defined Page 1)  0
913 	 *   0x09, 0x0b,       // Usage (Vendor Usage 0x0b)           3
914 	 * To:
915 	 *   0x05, 0x01,       // Usage Page (Generic Desktop)        0
916 	 *   0x09, 0x02,       // Usage (Mouse)                       2
917 	 */
918 	if (hdev->vendor == USB_VENDOR_ID_APPLE &&
919 	    (hdev->product == USB_DEVICE_ID_APPLE_MAGICMOUSE2 ||
920 	     hdev->product == USB_DEVICE_ID_APPLE_MAGICTRACKPAD2) &&
921 	    *rsize == 83 && rdesc[46] == 0x84 && rdesc[58] == 0x85) {
922 		hid_info(hdev,
923 			 "fixing up magicmouse battery report descriptor\n");
924 		*rsize = *rsize - 1;
925 		rdesc = kmemdup(rdesc + 1, *rsize, GFP_KERNEL);
926 		if (!rdesc)
927 			return NULL;
928 
929 		rdesc[0] = 0x05;
930 		rdesc[1] = 0x01;
931 		rdesc[2] = 0x09;
932 		rdesc[3] = 0x02;
933 	}
934 
935 	return rdesc;
936 }
937 
938 static const struct hid_device_id magic_mice[] = {
939 	{ HID_BLUETOOTH_DEVICE(USB_VENDOR_ID_APPLE,
940 		USB_DEVICE_ID_APPLE_MAGICMOUSE), .driver_data = 0 },
941 	{ HID_BLUETOOTH_DEVICE(BT_VENDOR_ID_APPLE,
942 		USB_DEVICE_ID_APPLE_MAGICMOUSE2), .driver_data = 0 },
943 	{ HID_USB_DEVICE(USB_VENDOR_ID_APPLE,
944 		USB_DEVICE_ID_APPLE_MAGICMOUSE2), .driver_data = 0 },
945 	{ HID_BLUETOOTH_DEVICE(USB_VENDOR_ID_APPLE,
946 		USB_DEVICE_ID_APPLE_MAGICTRACKPAD), .driver_data = 0 },
947 	{ HID_BLUETOOTH_DEVICE(BT_VENDOR_ID_APPLE,
948 		USB_DEVICE_ID_APPLE_MAGICTRACKPAD2), .driver_data = 0 },
949 	{ HID_USB_DEVICE(USB_VENDOR_ID_APPLE,
950 		USB_DEVICE_ID_APPLE_MAGICTRACKPAD2), .driver_data = 0 },
951 	{ }
952 };
953 MODULE_DEVICE_TABLE(hid, magic_mice);
954 
955 static struct hid_driver magicmouse_driver = {
956 	.name = "magicmouse",
957 	.id_table = magic_mice,
958 	.probe = magicmouse_probe,
959 	.remove = magicmouse_remove,
960 	.report_fixup = magicmouse_report_fixup,
961 	.raw_event = magicmouse_raw_event,
962 	.event = magicmouse_event,
963 	.input_mapping = magicmouse_input_mapping,
964 	.input_configured = magicmouse_input_configured,
965 };
966 module_hid_driver(magicmouse_driver);
967 
968 MODULE_LICENSE("GPL");
969