1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * drivers/input/tablet/wacom_sys.c
4  *
5  *  USB Wacom tablet support - system specific code
6  */
7 
8 /*
9  */
10 
11 #include "wacom_wac.h"
12 #include "wacom.h"
13 #include <linux/input/mt.h>
14 
15 #define WAC_MSG_RETRIES		5
16 #define WAC_CMD_RETRIES		10
17 
18 #define DEV_ATTR_RW_PERM (S_IRUGO | S_IWUSR | S_IWGRP)
19 #define DEV_ATTR_WO_PERM (S_IWUSR | S_IWGRP)
20 #define DEV_ATTR_RO_PERM (S_IRUSR | S_IRGRP)
21 
wacom_get_report(struct hid_device * hdev,u8 type,u8 * buf,size_t size,unsigned int retries)22 static int wacom_get_report(struct hid_device *hdev, u8 type, u8 *buf,
23 			    size_t size, unsigned int retries)
24 {
25 	int retval;
26 
27 	do {
28 		retval = hid_hw_raw_request(hdev, buf[0], buf, size, type,
29 				HID_REQ_GET_REPORT);
30 	} while ((retval == -ETIMEDOUT || retval == -EAGAIN) && --retries);
31 
32 	if (retval < 0)
33 		hid_err(hdev, "wacom_get_report: ran out of retries "
34 			"(last error = %d)\n", retval);
35 
36 	return retval;
37 }
38 
wacom_set_report(struct hid_device * hdev,u8 type,u8 * buf,size_t size,unsigned int retries)39 static int wacom_set_report(struct hid_device *hdev, u8 type, u8 *buf,
40 			    size_t size, unsigned int retries)
41 {
42 	int retval;
43 
44 	do {
45 		retval = hid_hw_raw_request(hdev, buf[0], buf, size, type,
46 				HID_REQ_SET_REPORT);
47 	} while ((retval == -ETIMEDOUT || retval == -EAGAIN) && --retries);
48 
49 	if (retval < 0)
50 		hid_err(hdev, "wacom_set_report: ran out of retries "
51 			"(last error = %d)\n", retval);
52 
53 	return retval;
54 }
55 
wacom_wac_queue_insert(struct hid_device * hdev,struct kfifo_rec_ptr_2 * fifo,u8 * raw_data,int size)56 static void wacom_wac_queue_insert(struct hid_device *hdev,
57 				   struct kfifo_rec_ptr_2 *fifo,
58 				   u8 *raw_data, int size)
59 {
60 	bool warned = false;
61 
62 	while (kfifo_avail(fifo) < size) {
63 		if (!warned)
64 			hid_warn(hdev, "%s: kfifo has filled, starting to drop events\n", __func__);
65 		warned = true;
66 
67 		kfifo_skip(fifo);
68 	}
69 
70 	kfifo_in(fifo, raw_data, size);
71 }
72 
wacom_wac_queue_flush(struct hid_device * hdev,struct kfifo_rec_ptr_2 * fifo)73 static void wacom_wac_queue_flush(struct hid_device *hdev,
74 				  struct kfifo_rec_ptr_2 *fifo)
75 {
76 	while (!kfifo_is_empty(fifo)) {
77 		u8 buf[WACOM_PKGLEN_MAX];
78 		int size;
79 		int err;
80 
81 		size = kfifo_out(fifo, buf, sizeof(buf));
82 		err = hid_report_raw_event(hdev, HID_INPUT_REPORT, buf, size, false);
83 		if (err) {
84 			hid_warn(hdev, "%s: unable to flush event due to error %d\n",
85 				 __func__, err);
86 		}
87 	}
88 }
89 
wacom_wac_pen_serial_enforce(struct hid_device * hdev,struct hid_report * report,u8 * raw_data,int report_size)90 static int wacom_wac_pen_serial_enforce(struct hid_device *hdev,
91 		struct hid_report *report, u8 *raw_data, int report_size)
92 {
93 	struct wacom *wacom = hid_get_drvdata(hdev);
94 	struct wacom_wac *wacom_wac = &wacom->wacom_wac;
95 	struct wacom_features *features = &wacom_wac->features;
96 	bool flush = false;
97 	bool insert = false;
98 	int i, j;
99 
100 	if (wacom_wac->serial[0] || !(features->quirks & WACOM_QUIRK_TOOLSERIAL))
101 		return 0;
102 
103 	/* Queue events which have invalid tool type or serial number */
104 	for (i = 0; i < report->maxfield; i++) {
105 		for (j = 0; j < report->field[i]->maxusage; j++) {
106 			struct hid_field *field = report->field[i];
107 			struct hid_usage *usage = &field->usage[j];
108 			unsigned int equivalent_usage = wacom_equivalent_usage(usage->hid);
109 			unsigned int offset;
110 			unsigned int size;
111 			unsigned int value;
112 
113 			if (equivalent_usage != HID_DG_INRANGE &&
114 			    equivalent_usage != HID_DG_TOOLSERIALNUMBER &&
115 			    equivalent_usage != WACOM_HID_WD_SERIALHI &&
116 			    equivalent_usage != WACOM_HID_WD_TOOLTYPE)
117 				continue;
118 
119 			offset = field->report_offset;
120 			size = field->report_size;
121 			value = hid_field_extract(hdev, raw_data+1, offset + j * size, size);
122 
123 			/* If we go out of range, we need to flush the queue ASAP */
124 			if (equivalent_usage == HID_DG_INRANGE)
125 				value = !value;
126 
127 			if (value) {
128 				flush = true;
129 				switch (equivalent_usage) {
130 				case HID_DG_TOOLSERIALNUMBER:
131 					wacom_wac->serial[0] = value;
132 					break;
133 
134 				case WACOM_HID_WD_SERIALHI:
135 					wacom_wac->serial[0] |= ((__u64)value) << 32;
136 					break;
137 
138 				case WACOM_HID_WD_TOOLTYPE:
139 					wacom_wac->id[0] = value;
140 					break;
141 				}
142 			}
143 			else {
144 				insert = true;
145 			}
146 		}
147 	}
148 
149 	if (flush)
150 		wacom_wac_queue_flush(hdev, wacom_wac->pen_fifo);
151 	else if (insert)
152 		wacom_wac_queue_insert(hdev, wacom_wac->pen_fifo,
153 				       raw_data, report_size);
154 
155 	return insert && !flush;
156 }
157 
wacom_raw_event(struct hid_device * hdev,struct hid_report * report,u8 * raw_data,int size)158 static int wacom_raw_event(struct hid_device *hdev, struct hid_report *report,
159 		u8 *raw_data, int size)
160 {
161 	struct wacom *wacom = hid_get_drvdata(hdev);
162 
163 	if (size > WACOM_PKGLEN_MAX)
164 		return 1;
165 
166 	if (wacom_wac_pen_serial_enforce(hdev, report, raw_data, size))
167 		return -1;
168 
169 	memcpy(wacom->wacom_wac.data, raw_data, size);
170 
171 	wacom_wac_irq(&wacom->wacom_wac, size);
172 
173 	return 0;
174 }
175 
wacom_open(struct input_dev * dev)176 static int wacom_open(struct input_dev *dev)
177 {
178 	struct wacom *wacom = input_get_drvdata(dev);
179 
180 	return hid_hw_open(wacom->hdev);
181 }
182 
wacom_close(struct input_dev * dev)183 static void wacom_close(struct input_dev *dev)
184 {
185 	struct wacom *wacom = input_get_drvdata(dev);
186 
187 	/*
188 	 * wacom->hdev should never be null, but surprisingly, I had the case
189 	 * once while unplugging the Wacom Wireless Receiver.
190 	 */
191 	if (wacom->hdev)
192 		hid_hw_close(wacom->hdev);
193 }
194 
195 /*
196  * Calculate the resolution of the X or Y axis using hidinput_calc_abs_res.
197  */
wacom_calc_hid_res(int logical_extents,int physical_extents,unsigned unit,int exponent)198 static int wacom_calc_hid_res(int logical_extents, int physical_extents,
199 			       unsigned unit, int exponent)
200 {
201 	struct hid_field field = {
202 		.logical_maximum = logical_extents,
203 		.physical_maximum = physical_extents,
204 		.unit = unit,
205 		.unit_exponent = exponent,
206 	};
207 
208 	return hidinput_calc_abs_res(&field, ABS_X);
209 }
210 
wacom_hid_usage_quirk(struct hid_device * hdev,struct hid_field * field,struct hid_usage * usage)211 static void wacom_hid_usage_quirk(struct hid_device *hdev,
212 		struct hid_field *field, struct hid_usage *usage)
213 {
214 	struct wacom *wacom = hid_get_drvdata(hdev);
215 	struct wacom_features *features = &wacom->wacom_wac.features;
216 	unsigned int equivalent_usage = wacom_equivalent_usage(usage->hid);
217 
218 	/*
219 	 * The Dell Canvas 27 needs to be switched to its vendor-defined
220 	 * report to provide the best resolution.
221 	 */
222 	if (hdev->vendor == USB_VENDOR_ID_WACOM &&
223 	    hdev->product == 0x4200 &&
224 	    field->application == HID_UP_MSVENDOR) {
225 		wacom->wacom_wac.mode_report = field->report->id;
226 		wacom->wacom_wac.mode_value = 2;
227 	}
228 
229 	/*
230 	 * ISDv4 devices which predate HID's adoption of the
231 	 * HID_DG_BARELSWITCH2 usage use 0x000D0000 in its
232 	 * position instead. We can accurately detect if a
233 	 * usage with that value should be HID_DG_BARRELSWITCH2
234 	 * based on the surrounding usages, which have remained
235 	 * constant across generations.
236 	 */
237 	if (features->type == HID_GENERIC &&
238 	    usage->hid == 0x000D0000 &&
239 	    field->application == HID_DG_PEN &&
240 	    field->physical == HID_DG_STYLUS) {
241 		int i = usage->usage_index;
242 
243 		if (i-4 >= 0 && i+1 < field->maxusage &&
244 		    field->usage[i-4].hid == HID_DG_TIPSWITCH &&
245 		    field->usage[i-3].hid == HID_DG_BARRELSWITCH &&
246 		    field->usage[i-2].hid == HID_DG_ERASER &&
247 		    field->usage[i-1].hid == HID_DG_INVERT &&
248 		    field->usage[i+1].hid == HID_DG_INRANGE) {
249 			usage->hid = HID_DG_BARRELSWITCH2;
250 		}
251 	}
252 
253 	/*
254 	 * Wacom's AES devices use different vendor-defined usages to
255 	 * report serial number information compared to their branded
256 	 * hardware. The usages are also sometimes ill-defined and do
257 	 * not have the correct logical min/max values set. Lets patch
258 	 * the descriptor to use the branded usage convention and fix
259 	 * the errors.
260 	 */
261 	if (usage->hid == WACOM_HID_WT_SERIALNUMBER &&
262 	    field->report_size == 16 &&
263 	    field->index + 2 < field->report->maxfield) {
264 		struct hid_field *a = field->report->field[field->index + 1];
265 		struct hid_field *b = field->report->field[field->index + 2];
266 
267 		if (a->maxusage > 0 &&
268 		    a->usage[0].hid == HID_DG_TOOLSERIALNUMBER &&
269 		    a->report_size == 32 &&
270 		    b->maxusage > 0 &&
271 		    b->usage[0].hid == 0xFF000000 &&
272 		    b->report_size == 8) {
273 			features->quirks |= WACOM_QUIRK_AESPEN;
274 			usage->hid = WACOM_HID_WD_TOOLTYPE;
275 			field->logical_minimum = S16_MIN;
276 			field->logical_maximum = S16_MAX;
277 			a->logical_minimum = S32_MIN;
278 			a->logical_maximum = S32_MAX;
279 			b->usage[0].hid = WACOM_HID_WD_SERIALHI;
280 			b->logical_minimum = 0;
281 			b->logical_maximum = U8_MAX;
282 		}
283 	}
284 
285 	/* 2nd-generation Intuos Pro Large has incorrect Y maximum */
286 	if (hdev->vendor == USB_VENDOR_ID_WACOM &&
287 	    hdev->product == 0x0358 &&
288 	    WACOM_PEN_FIELD(field) &&
289 	    equivalent_usage == HID_GD_Y) {
290 		field->logical_maximum = 43200;
291 	}
292 }
293 
wacom_feature_mapping(struct hid_device * hdev,struct hid_field * field,struct hid_usage * usage)294 static void wacom_feature_mapping(struct hid_device *hdev,
295 		struct hid_field *field, struct hid_usage *usage)
296 {
297 	struct wacom *wacom = hid_get_drvdata(hdev);
298 	struct wacom_features *features = &wacom->wacom_wac.features;
299 	struct hid_data *hid_data = &wacom->wacom_wac.hid_data;
300 	unsigned int equivalent_usage = wacom_equivalent_usage(usage->hid);
301 	u8 *data;
302 	int ret;
303 	u32 n;
304 
305 	wacom_hid_usage_quirk(hdev, field, usage);
306 
307 	switch (equivalent_usage) {
308 	case WACOM_HID_WD_TOUCH_RING_SETTING:
309 		wacom->generic_has_leds = true;
310 		break;
311 	case HID_DG_CONTACTMAX:
312 		/* leave touch_max as is if predefined */
313 		if (!features->touch_max) {
314 			/* read manually */
315 			n = hid_report_len(field->report);
316 			data = hid_alloc_report_buf(field->report, GFP_KERNEL);
317 			if (!data)
318 				break;
319 			data[0] = field->report->id;
320 			ret = wacom_get_report(hdev, HID_FEATURE_REPORT,
321 					       data, n, WAC_CMD_RETRIES);
322 			if (ret == n && features->type == HID_GENERIC) {
323 				ret = hid_report_raw_event(hdev,
324 					HID_FEATURE_REPORT, data, n, 0);
325 			} else if (ret == 2 && features->type != HID_GENERIC) {
326 				features->touch_max = data[1];
327 			} else {
328 				features->touch_max = 16;
329 				hid_warn(hdev, "wacom_feature_mapping: "
330 					 "could not get HID_DG_CONTACTMAX, "
331 					 "defaulting to %d\n",
332 					  features->touch_max);
333 			}
334 			kfree(data);
335 		}
336 		break;
337 	case HID_DG_INPUTMODE:
338 		/* Ignore if value index is out of bounds. */
339 		if (usage->usage_index >= field->report_count) {
340 			dev_err(&hdev->dev, "HID_DG_INPUTMODE out of range\n");
341 			break;
342 		}
343 
344 		hid_data->inputmode = field->report->id;
345 		hid_data->inputmode_index = usage->usage_index;
346 		break;
347 
348 	case HID_UP_DIGITIZER:
349 		if (field->report->id == 0x0B &&
350 		    (field->application == WACOM_HID_G9_PEN ||
351 		     field->application == WACOM_HID_G11_PEN)) {
352 			wacom->wacom_wac.mode_report = field->report->id;
353 			wacom->wacom_wac.mode_value = 0;
354 		}
355 		break;
356 
357 	case WACOM_HID_WD_DATAMODE:
358 		wacom->wacom_wac.mode_report = field->report->id;
359 		wacom->wacom_wac.mode_value = 2;
360 		break;
361 
362 	case WACOM_HID_UP_G9:
363 	case WACOM_HID_UP_G11:
364 		if (field->report->id == 0x03 &&
365 		    (field->application == WACOM_HID_G9_TOUCHSCREEN ||
366 		     field->application == WACOM_HID_G11_TOUCHSCREEN)) {
367 			wacom->wacom_wac.mode_report = field->report->id;
368 			wacom->wacom_wac.mode_value = 0;
369 		}
370 		break;
371 	case WACOM_HID_WD_OFFSETLEFT:
372 	case WACOM_HID_WD_OFFSETTOP:
373 	case WACOM_HID_WD_OFFSETRIGHT:
374 	case WACOM_HID_WD_OFFSETBOTTOM:
375 		/* read manually */
376 		n = hid_report_len(field->report);
377 		data = hid_alloc_report_buf(field->report, GFP_KERNEL);
378 		if (!data)
379 			break;
380 		data[0] = field->report->id;
381 		ret = wacom_get_report(hdev, HID_FEATURE_REPORT,
382 					data, n, WAC_CMD_RETRIES);
383 		if (ret == n) {
384 			ret = hid_report_raw_event(hdev, HID_FEATURE_REPORT,
385 						   data, n, 0);
386 		} else {
387 			hid_warn(hdev, "%s: could not retrieve sensor offsets\n",
388 				 __func__);
389 		}
390 		kfree(data);
391 		break;
392 	}
393 }
394 
395 /*
396  * Interface Descriptor of wacom devices can be incomplete and
397  * inconsistent so wacom_features table is used to store stylus
398  * device's packet lengths, various maximum values, and tablet
399  * resolution based on product ID's.
400  *
401  * For devices that contain 2 interfaces, wacom_features table is
402  * inaccurate for the touch interface.  Since the Interface Descriptor
403  * for touch interfaces has pretty complete data, this function exists
404  * to query tablet for this missing information instead of hard coding in
405  * an additional table.
406  *
407  * A typical Interface Descriptor for a stylus will contain a
408  * boot mouse application collection that is not of interest and this
409  * function will ignore it.
410  *
411  * It also contains a digitizer application collection that also is not
412  * of interest since any information it contains would be duplicate
413  * of what is in wacom_features. Usually it defines a report of an array
414  * of bytes that could be used as max length of the stylus packet returned.
415  * If it happens to define a Digitizer-Stylus Physical Collection then
416  * the X and Y logical values contain valid data but it is ignored.
417  *
418  * A typical Interface Descriptor for a touch interface will contain a
419  * Digitizer-Finger Physical Collection which will define both logical
420  * X/Y maximum as well as the physical size of tablet. Since touch
421  * interfaces haven't supported pressure or distance, this is enough
422  * information to override invalid values in the wacom_features table.
423  *
424  * Intuos5 touch interface and 3rd gen Bamboo Touch do not contain useful
425  * data. We deal with them after returning from this function.
426  */
wacom_usage_mapping(struct hid_device * hdev,struct hid_field * field,struct hid_usage * usage)427 static void wacom_usage_mapping(struct hid_device *hdev,
428 		struct hid_field *field, struct hid_usage *usage)
429 {
430 	struct wacom *wacom = hid_get_drvdata(hdev);
431 	struct wacom_features *features = &wacom->wacom_wac.features;
432 	bool finger = WACOM_FINGER_FIELD(field);
433 	bool pen = WACOM_PEN_FIELD(field);
434 	unsigned equivalent_usage = wacom_equivalent_usage(usage->hid);
435 
436 	/*
437 	* Requiring Stylus Usage will ignore boot mouse
438 	* X/Y values and some cases of invalid Digitizer X/Y
439 	* values commonly reported.
440 	*/
441 	if (pen)
442 		features->device_type |= WACOM_DEVICETYPE_PEN;
443 	else if (finger)
444 		features->device_type |= WACOM_DEVICETYPE_TOUCH;
445 	else
446 		return;
447 
448 	wacom_hid_usage_quirk(hdev, field, usage);
449 
450 	switch (equivalent_usage) {
451 	case HID_GD_X:
452 		features->x_max = field->logical_maximum;
453 		if (finger) {
454 			features->x_phy = field->physical_maximum;
455 			if ((features->type != BAMBOO_PT) &&
456 			    (features->type != BAMBOO_TOUCH)) {
457 				features->unit = field->unit;
458 				features->unitExpo = field->unit_exponent;
459 			}
460 		}
461 		break;
462 	case HID_GD_Y:
463 		features->y_max = field->logical_maximum;
464 		if (finger) {
465 			features->y_phy = field->physical_maximum;
466 			if ((features->type != BAMBOO_PT) &&
467 			    (features->type != BAMBOO_TOUCH)) {
468 				features->unit = field->unit;
469 				features->unitExpo = field->unit_exponent;
470 			}
471 		}
472 		break;
473 	case HID_DG_TIPPRESSURE:
474 		if (pen)
475 			features->pressure_max = field->logical_maximum;
476 		break;
477 	}
478 
479 	if (features->type == HID_GENERIC)
480 		wacom_wac_usage_mapping(hdev, field, usage);
481 }
482 
wacom_post_parse_hid(struct hid_device * hdev,struct wacom_features * features)483 static void wacom_post_parse_hid(struct hid_device *hdev,
484 				 struct wacom_features *features)
485 {
486 	struct wacom *wacom = hid_get_drvdata(hdev);
487 	struct wacom_wac *wacom_wac = &wacom->wacom_wac;
488 
489 	if (features->type == HID_GENERIC) {
490 		/* Any last-minute generic device setup */
491 		if (wacom_wac->has_mode_change) {
492 			if (wacom_wac->is_direct_mode)
493 				features->device_type |= WACOM_DEVICETYPE_DIRECT;
494 			else
495 				features->device_type &= ~WACOM_DEVICETYPE_DIRECT;
496 		}
497 
498 		if (features->touch_max > 1) {
499 			if (features->device_type & WACOM_DEVICETYPE_DIRECT)
500 				input_mt_init_slots(wacom_wac->touch_input,
501 						    wacom_wac->features.touch_max,
502 						    INPUT_MT_DIRECT);
503 			else
504 				input_mt_init_slots(wacom_wac->touch_input,
505 						    wacom_wac->features.touch_max,
506 						    INPUT_MT_POINTER);
507 		}
508 	}
509 }
510 
wacom_parse_hid(struct hid_device * hdev,struct wacom_features * features)511 static void wacom_parse_hid(struct hid_device *hdev,
512 			   struct wacom_features *features)
513 {
514 	struct hid_report_enum *rep_enum;
515 	struct hid_report *hreport;
516 	int i, j;
517 
518 	/* check features first */
519 	rep_enum = &hdev->report_enum[HID_FEATURE_REPORT];
520 	list_for_each_entry(hreport, &rep_enum->report_list, list) {
521 		for (i = 0; i < hreport->maxfield; i++) {
522 			/* Ignore if report count is out of bounds. */
523 			if (hreport->field[i]->report_count < 1)
524 				continue;
525 
526 			for (j = 0; j < hreport->field[i]->maxusage; j++) {
527 				wacom_feature_mapping(hdev, hreport->field[i],
528 						hreport->field[i]->usage + j);
529 			}
530 		}
531 	}
532 
533 	/* now check the input usages */
534 	rep_enum = &hdev->report_enum[HID_INPUT_REPORT];
535 	list_for_each_entry(hreport, &rep_enum->report_list, list) {
536 
537 		if (!hreport->maxfield)
538 			continue;
539 
540 		for (i = 0; i < hreport->maxfield; i++)
541 			for (j = 0; j < hreport->field[i]->maxusage; j++)
542 				wacom_usage_mapping(hdev, hreport->field[i],
543 						hreport->field[i]->usage + j);
544 	}
545 
546 	wacom_post_parse_hid(hdev, features);
547 }
548 
wacom_hid_set_device_mode(struct hid_device * hdev)549 static int wacom_hid_set_device_mode(struct hid_device *hdev)
550 {
551 	struct wacom *wacom = hid_get_drvdata(hdev);
552 	struct hid_data *hid_data = &wacom->wacom_wac.hid_data;
553 	struct hid_report *r;
554 	struct hid_report_enum *re;
555 
556 	if (hid_data->inputmode < 0)
557 		return 0;
558 
559 	re = &(hdev->report_enum[HID_FEATURE_REPORT]);
560 	r = re->report_id_hash[hid_data->inputmode];
561 	if (r) {
562 		r->field[0]->value[hid_data->inputmode_index] = 2;
563 		hid_hw_request(hdev, r, HID_REQ_SET_REPORT);
564 	}
565 	return 0;
566 }
567 
wacom_set_device_mode(struct hid_device * hdev,struct wacom_wac * wacom_wac)568 static int wacom_set_device_mode(struct hid_device *hdev,
569 				 struct wacom_wac *wacom_wac)
570 {
571 	u8 *rep_data;
572 	struct hid_report *r;
573 	struct hid_report_enum *re;
574 	u32 length;
575 	int error = -ENOMEM, limit = 0;
576 
577 	if (wacom_wac->mode_report < 0)
578 		return 0;
579 
580 	re = &(hdev->report_enum[HID_FEATURE_REPORT]);
581 	r = re->report_id_hash[wacom_wac->mode_report];
582 	if (!r)
583 		return -EINVAL;
584 
585 	rep_data = hid_alloc_report_buf(r, GFP_KERNEL);
586 	if (!rep_data)
587 		return -ENOMEM;
588 
589 	length = hid_report_len(r);
590 
591 	do {
592 		rep_data[0] = wacom_wac->mode_report;
593 		rep_data[1] = wacom_wac->mode_value;
594 
595 		error = wacom_set_report(hdev, HID_FEATURE_REPORT, rep_data,
596 					 length, 1);
597 		if (error >= 0)
598 			error = wacom_get_report(hdev, HID_FEATURE_REPORT,
599 			                         rep_data, length, 1);
600 	} while (error >= 0 &&
601 		 rep_data[1] != wacom_wac->mode_report &&
602 		 limit++ < WAC_MSG_RETRIES);
603 
604 	kfree(rep_data);
605 
606 	return error < 0 ? error : 0;
607 }
608 
wacom_bt_query_tablet_data(struct hid_device * hdev,u8 speed,struct wacom_features * features)609 static int wacom_bt_query_tablet_data(struct hid_device *hdev, u8 speed,
610 		struct wacom_features *features)
611 {
612 	struct wacom *wacom = hid_get_drvdata(hdev);
613 	int ret;
614 	u8 rep_data[2];
615 
616 	switch (features->type) {
617 	case GRAPHIRE_BT:
618 		rep_data[0] = 0x03;
619 		rep_data[1] = 0x00;
620 		ret = wacom_set_report(hdev, HID_FEATURE_REPORT, rep_data, 2,
621 					3);
622 
623 		if (ret >= 0) {
624 			rep_data[0] = speed == 0 ? 0x05 : 0x06;
625 			rep_data[1] = 0x00;
626 
627 			ret = wacom_set_report(hdev, HID_FEATURE_REPORT,
628 						rep_data, 2, 3);
629 
630 			if (ret >= 0) {
631 				wacom->wacom_wac.bt_high_speed = speed;
632 				return 0;
633 			}
634 		}
635 
636 		/*
637 		 * Note that if the raw queries fail, it's not a hard failure
638 		 * and it is safe to continue
639 		 */
640 		hid_warn(hdev, "failed to poke device, command %d, err %d\n",
641 			 rep_data[0], ret);
642 		break;
643 	case INTUOS4WL:
644 		if (speed == 1)
645 			wacom->wacom_wac.bt_features &= ~0x20;
646 		else
647 			wacom->wacom_wac.bt_features |= 0x20;
648 
649 		rep_data[0] = 0x03;
650 		rep_data[1] = wacom->wacom_wac.bt_features;
651 
652 		ret = wacom_set_report(hdev, HID_FEATURE_REPORT, rep_data, 2,
653 					1);
654 		if (ret >= 0)
655 			wacom->wacom_wac.bt_high_speed = speed;
656 		break;
657 	}
658 
659 	return 0;
660 }
661 
662 /*
663  * Switch the tablet into its most-capable mode. Wacom tablets are
664  * typically configured to power-up in a mode which sends mouse-like
665  * reports to the OS. To get absolute position, pressure data, etc.
666  * from the tablet, it is necessary to switch the tablet out of this
667  * mode and into one which sends the full range of tablet data.
668  */
_wacom_query_tablet_data(struct wacom * wacom)669 static int _wacom_query_tablet_data(struct wacom *wacom)
670 {
671 	struct hid_device *hdev = wacom->hdev;
672 	struct wacom_wac *wacom_wac = &wacom->wacom_wac;
673 	struct wacom_features *features = &wacom_wac->features;
674 
675 	if (hdev->bus == BUS_BLUETOOTH)
676 		return wacom_bt_query_tablet_data(hdev, 1, features);
677 
678 	if (features->type != HID_GENERIC) {
679 		if (features->device_type & WACOM_DEVICETYPE_TOUCH) {
680 			if (features->type > TABLETPC) {
681 				/* MT Tablet PC touch */
682 				wacom_wac->mode_report = 3;
683 				wacom_wac->mode_value = 4;
684 			} else if (features->type == WACOM_24HDT) {
685 				wacom_wac->mode_report = 18;
686 				wacom_wac->mode_value = 2;
687 			} else if (features->type == WACOM_27QHDT) {
688 				wacom_wac->mode_report = 131;
689 				wacom_wac->mode_value = 2;
690 			} else if (features->type == BAMBOO_PAD) {
691 				wacom_wac->mode_report = 2;
692 				wacom_wac->mode_value = 2;
693 			}
694 		} else if (features->device_type & WACOM_DEVICETYPE_PEN) {
695 			if (features->type <= BAMBOO_PT) {
696 				wacom_wac->mode_report = 2;
697 				wacom_wac->mode_value = 2;
698 			}
699 		}
700 	}
701 
702 	wacom_set_device_mode(hdev, wacom_wac);
703 
704 	if (features->type == HID_GENERIC)
705 		return wacom_hid_set_device_mode(hdev);
706 
707 	return 0;
708 }
709 
wacom_retrieve_hid_descriptor(struct hid_device * hdev,struct wacom_features * features)710 static void wacom_retrieve_hid_descriptor(struct hid_device *hdev,
711 					 struct wacom_features *features)
712 {
713 	struct wacom *wacom = hid_get_drvdata(hdev);
714 	struct usb_interface *intf = wacom->intf;
715 
716 	/* default features */
717 	features->x_fuzz = 4;
718 	features->y_fuzz = 4;
719 	features->pressure_fuzz = 0;
720 	features->distance_fuzz = 1;
721 	features->tilt_fuzz = 1;
722 
723 	/*
724 	 * The wireless device HID is basic and layout conflicts with
725 	 * other tablets (monitor and touch interface can look like pen).
726 	 * Skip the query for this type and modify defaults based on
727 	 * interface number.
728 	 */
729 	if (features->type == WIRELESS && intf) {
730 		if (intf->cur_altsetting->desc.bInterfaceNumber == 0)
731 			features->device_type = WACOM_DEVICETYPE_WL_MONITOR;
732 		else
733 			features->device_type = WACOM_DEVICETYPE_NONE;
734 		return;
735 	}
736 
737 	wacom_parse_hid(hdev, features);
738 }
739 
740 struct wacom_hdev_data {
741 	struct list_head list;
742 	struct kref kref;
743 	struct hid_device *dev;
744 	struct wacom_shared shared;
745 };
746 
747 static LIST_HEAD(wacom_udev_list);
748 static DEFINE_MUTEX(wacom_udev_list_lock);
749 
wacom_are_sibling(struct hid_device * hdev,struct hid_device * sibling)750 static bool wacom_are_sibling(struct hid_device *hdev,
751 		struct hid_device *sibling)
752 {
753 	struct wacom *wacom = hid_get_drvdata(hdev);
754 	struct wacom_features *features = &wacom->wacom_wac.features;
755 	struct wacom *sibling_wacom = hid_get_drvdata(sibling);
756 	struct wacom_features *sibling_features = &sibling_wacom->wacom_wac.features;
757 	__u32 oVid = features->oVid ? features->oVid : hdev->vendor;
758 	__u32 oPid = features->oPid ? features->oPid : hdev->product;
759 
760 	/* The defined oVid/oPid must match that of the sibling */
761 	if (features->oVid != HID_ANY_ID && sibling->vendor != oVid)
762 		return false;
763 	if (features->oPid != HID_ANY_ID && sibling->product != oPid)
764 		return false;
765 
766 	/*
767 	 * Devices with the same VID/PID must share the same physical
768 	 * device path, while those with different VID/PID must share
769 	 * the same physical parent device path.
770 	 */
771 	if (hdev->vendor == sibling->vendor && hdev->product == sibling->product) {
772 		if (!hid_compare_device_paths(hdev, sibling, '/'))
773 			return false;
774 	} else {
775 		if (!hid_compare_device_paths(hdev, sibling, '.'))
776 			return false;
777 	}
778 
779 	/* Skip the remaining heuristics unless you are a HID_GENERIC device */
780 	if (features->type != HID_GENERIC)
781 		return true;
782 
783 	/*
784 	 * Direct-input devices may not be siblings of indirect-input
785 	 * devices.
786 	 */
787 	if ((features->device_type & WACOM_DEVICETYPE_DIRECT) &&
788 	    !(sibling_features->device_type & WACOM_DEVICETYPE_DIRECT))
789 		return false;
790 
791 	/*
792 	 * Indirect-input devices may not be siblings of direct-input
793 	 * devices.
794 	 */
795 	if (!(features->device_type & WACOM_DEVICETYPE_DIRECT) &&
796 	    (sibling_features->device_type & WACOM_DEVICETYPE_DIRECT))
797 		return false;
798 
799 	/* Pen devices may only be siblings of touch devices */
800 	if ((features->device_type & WACOM_DEVICETYPE_PEN) &&
801 	    !(sibling_features->device_type & WACOM_DEVICETYPE_TOUCH))
802 		return false;
803 
804 	/* Touch devices may only be siblings of pen devices */
805 	if ((features->device_type & WACOM_DEVICETYPE_TOUCH) &&
806 	    !(sibling_features->device_type & WACOM_DEVICETYPE_PEN))
807 		return false;
808 
809 	/*
810 	 * No reason could be found for these two devices to NOT be
811 	 * siblings, so there's a good chance they ARE siblings
812 	 */
813 	return true;
814 }
815 
wacom_get_hdev_data(struct hid_device * hdev)816 static struct wacom_hdev_data *wacom_get_hdev_data(struct hid_device *hdev)
817 {
818 	struct wacom_hdev_data *data;
819 
820 	/* Try to find an already-probed interface from the same device */
821 	list_for_each_entry(data, &wacom_udev_list, list) {
822 		if (hid_compare_device_paths(hdev, data->dev, '/')) {
823 			kref_get(&data->kref);
824 			return data;
825 		}
826 	}
827 
828 	/* Fallback to finding devices that appear to be "siblings" */
829 	list_for_each_entry(data, &wacom_udev_list, list) {
830 		if (wacom_are_sibling(hdev, data->dev)) {
831 			kref_get(&data->kref);
832 			return data;
833 		}
834 	}
835 
836 	return NULL;
837 }
838 
wacom_release_shared_data(struct kref * kref)839 static void wacom_release_shared_data(struct kref *kref)
840 {
841 	struct wacom_hdev_data *data =
842 		container_of(kref, struct wacom_hdev_data, kref);
843 
844 	mutex_lock(&wacom_udev_list_lock);
845 	list_del(&data->list);
846 	mutex_unlock(&wacom_udev_list_lock);
847 
848 	kfree(data);
849 }
850 
wacom_remove_shared_data(void * res)851 static void wacom_remove_shared_data(void *res)
852 {
853 	struct wacom *wacom = res;
854 	struct wacom_hdev_data *data;
855 	struct wacom_wac *wacom_wac = &wacom->wacom_wac;
856 
857 	if (wacom_wac->shared) {
858 		data = container_of(wacom_wac->shared, struct wacom_hdev_data,
859 				    shared);
860 
861 		if (wacom_wac->shared->touch == wacom->hdev)
862 			wacom_wac->shared->touch = NULL;
863 		else if (wacom_wac->shared->pen == wacom->hdev)
864 			wacom_wac->shared->pen = NULL;
865 
866 		kref_put(&data->kref, wacom_release_shared_data);
867 		wacom_wac->shared = NULL;
868 	}
869 }
870 
wacom_add_shared_data(struct hid_device * hdev)871 static int wacom_add_shared_data(struct hid_device *hdev)
872 {
873 	struct wacom *wacom = hid_get_drvdata(hdev);
874 	struct wacom_wac *wacom_wac = &wacom->wacom_wac;
875 	struct wacom_hdev_data *data;
876 	int retval = 0;
877 
878 	mutex_lock(&wacom_udev_list_lock);
879 
880 	data = wacom_get_hdev_data(hdev);
881 	if (!data) {
882 		data = kzalloc(sizeof(struct wacom_hdev_data), GFP_KERNEL);
883 		if (!data) {
884 			mutex_unlock(&wacom_udev_list_lock);
885 			return -ENOMEM;
886 		}
887 
888 		kref_init(&data->kref);
889 		data->dev = hdev;
890 		list_add_tail(&data->list, &wacom_udev_list);
891 	}
892 
893 	mutex_unlock(&wacom_udev_list_lock);
894 
895 	wacom_wac->shared = &data->shared;
896 
897 	retval = devm_add_action_or_reset(&hdev->dev, wacom_remove_shared_data, wacom);
898 	if (retval)
899 		return retval;
900 
901 	if (wacom_wac->features.device_type & WACOM_DEVICETYPE_TOUCH)
902 		wacom_wac->shared->touch = hdev;
903 	else if (wacom_wac->features.device_type & WACOM_DEVICETYPE_PEN)
904 		wacom_wac->shared->pen = hdev;
905 
906 	return retval;
907 }
908 
wacom_led_control(struct wacom * wacom)909 static int wacom_led_control(struct wacom *wacom)
910 {
911 	unsigned char *buf;
912 	int retval;
913 	unsigned char report_id = WAC_CMD_LED_CONTROL;
914 	int buf_size = 9;
915 
916 	if (!wacom->led.groups)
917 		return -ENOTSUPP;
918 
919 	if (wacom->wacom_wac.features.type == REMOTE)
920 		return -ENOTSUPP;
921 
922 	if (wacom->wacom_wac.pid) { /* wireless connected */
923 		report_id = WAC_CMD_WL_LED_CONTROL;
924 		buf_size = 13;
925 	}
926 	else if (wacom->wacom_wac.features.type == INTUOSP2_BT) {
927 		report_id = WAC_CMD_WL_INTUOSP2;
928 		buf_size = 51;
929 	}
930 	buf = kzalloc(buf_size, GFP_KERNEL);
931 	if (!buf)
932 		return -ENOMEM;
933 
934 	if (wacom->wacom_wac.features.type == HID_GENERIC) {
935 		buf[0] = WAC_CMD_LED_CONTROL_GENERIC;
936 		buf[1] = wacom->led.llv;
937 		buf[2] = wacom->led.groups[0].select & 0x03;
938 
939 	} else if ((wacom->wacom_wac.features.type >= INTUOS5S &&
940 	    wacom->wacom_wac.features.type <= INTUOSPL)) {
941 		/*
942 		 * Touch Ring and crop mark LED luminance may take on
943 		 * one of four values:
944 		 *    0 = Low; 1 = Medium; 2 = High; 3 = Off
945 		 */
946 		int ring_led = wacom->led.groups[0].select & 0x03;
947 		int ring_lum = (((wacom->led.llv & 0x60) >> 5) - 1) & 0x03;
948 		int crop_lum = 0;
949 		unsigned char led_bits = (crop_lum << 4) | (ring_lum << 2) | (ring_led);
950 
951 		buf[0] = report_id;
952 		if (wacom->wacom_wac.pid) {
953 			wacom_get_report(wacom->hdev, HID_FEATURE_REPORT,
954 					 buf, buf_size, WAC_CMD_RETRIES);
955 			buf[0] = report_id;
956 			buf[4] = led_bits;
957 		} else
958 			buf[1] = led_bits;
959 	}
960 	else if (wacom->wacom_wac.features.type == INTUOSP2_BT) {
961 		buf[0] = report_id;
962 		buf[4] = 100; // Power Connection LED (ORANGE)
963 		buf[5] = 100; // BT Connection LED (BLUE)
964 		buf[6] = 100; // Paper Mode (RED?)
965 		buf[7] = 100; // Paper Mode (GREEN?)
966 		buf[8] = 100; // Paper Mode (BLUE?)
967 		buf[9] = wacom->led.llv;
968 		buf[10] = wacom->led.groups[0].select & 0x03;
969 	}
970 	else {
971 		int led = wacom->led.groups[0].select | 0x4;
972 
973 		if (wacom->wacom_wac.features.type == WACOM_21UX2 ||
974 		    wacom->wacom_wac.features.type == WACOM_24HD)
975 			led |= (wacom->led.groups[1].select << 4) | 0x40;
976 
977 		buf[0] = report_id;
978 		buf[1] = led;
979 		buf[2] = wacom->led.llv;
980 		buf[3] = wacom->led.hlv;
981 		buf[4] = wacom->led.img_lum;
982 	}
983 
984 	retval = wacom_set_report(wacom->hdev, HID_FEATURE_REPORT, buf, buf_size,
985 				  WAC_CMD_RETRIES);
986 	kfree(buf);
987 
988 	return retval;
989 }
990 
wacom_led_putimage(struct wacom * wacom,int button_id,u8 xfer_id,const unsigned len,const void * img)991 static int wacom_led_putimage(struct wacom *wacom, int button_id, u8 xfer_id,
992 		const unsigned len, const void *img)
993 {
994 	unsigned char *buf;
995 	int i, retval;
996 	const unsigned chunk_len = len / 4; /* 4 chunks are needed to be sent */
997 
998 	buf = kzalloc(chunk_len + 3 , GFP_KERNEL);
999 	if (!buf)
1000 		return -ENOMEM;
1001 
1002 	/* Send 'start' command */
1003 	buf[0] = WAC_CMD_ICON_START;
1004 	buf[1] = 1;
1005 	retval = wacom_set_report(wacom->hdev, HID_FEATURE_REPORT, buf, 2,
1006 				  WAC_CMD_RETRIES);
1007 	if (retval < 0)
1008 		goto out;
1009 
1010 	buf[0] = xfer_id;
1011 	buf[1] = button_id & 0x07;
1012 	for (i = 0; i < 4; i++) {
1013 		buf[2] = i;
1014 		memcpy(buf + 3, img + i * chunk_len, chunk_len);
1015 
1016 		retval = wacom_set_report(wacom->hdev, HID_FEATURE_REPORT,
1017 					  buf, chunk_len + 3, WAC_CMD_RETRIES);
1018 		if (retval < 0)
1019 			break;
1020 	}
1021 
1022 	/* Send 'stop' */
1023 	buf[0] = WAC_CMD_ICON_START;
1024 	buf[1] = 0;
1025 	wacom_set_report(wacom->hdev, HID_FEATURE_REPORT, buf, 2,
1026 			 WAC_CMD_RETRIES);
1027 
1028 out:
1029 	kfree(buf);
1030 	return retval;
1031 }
1032 
wacom_led_select_store(struct device * dev,int set_id,const char * buf,size_t count)1033 static ssize_t wacom_led_select_store(struct device *dev, int set_id,
1034 				      const char *buf, size_t count)
1035 {
1036 	struct hid_device *hdev = to_hid_device(dev);
1037 	struct wacom *wacom = hid_get_drvdata(hdev);
1038 	unsigned int id;
1039 	int err;
1040 
1041 	err = kstrtouint(buf, 10, &id);
1042 	if (err)
1043 		return err;
1044 
1045 	mutex_lock(&wacom->lock);
1046 
1047 	wacom->led.groups[set_id].select = id & 0x3;
1048 	err = wacom_led_control(wacom);
1049 
1050 	mutex_unlock(&wacom->lock);
1051 
1052 	return err < 0 ? err : count;
1053 }
1054 
1055 #define DEVICE_LED_SELECT_ATTR(SET_ID)					\
1056 static ssize_t wacom_led##SET_ID##_select_store(struct device *dev,	\
1057 	struct device_attribute *attr, const char *buf, size_t count)	\
1058 {									\
1059 	return wacom_led_select_store(dev, SET_ID, buf, count);		\
1060 }									\
1061 static ssize_t wacom_led##SET_ID##_select_show(struct device *dev,	\
1062 	struct device_attribute *attr, char *buf)			\
1063 {									\
1064 	struct hid_device *hdev = to_hid_device(dev);\
1065 	struct wacom *wacom = hid_get_drvdata(hdev);			\
1066 	return scnprintf(buf, PAGE_SIZE, "%d\n",			\
1067 			 wacom->led.groups[SET_ID].select);		\
1068 }									\
1069 static DEVICE_ATTR(status_led##SET_ID##_select, DEV_ATTR_RW_PERM,	\
1070 		    wacom_led##SET_ID##_select_show,			\
1071 		    wacom_led##SET_ID##_select_store)
1072 
1073 DEVICE_LED_SELECT_ATTR(0);
1074 DEVICE_LED_SELECT_ATTR(1);
1075 
wacom_luminance_store(struct wacom * wacom,u8 * dest,const char * buf,size_t count)1076 static ssize_t wacom_luminance_store(struct wacom *wacom, u8 *dest,
1077 				     const char *buf, size_t count)
1078 {
1079 	unsigned int value;
1080 	int err;
1081 
1082 	err = kstrtouint(buf, 10, &value);
1083 	if (err)
1084 		return err;
1085 
1086 	mutex_lock(&wacom->lock);
1087 
1088 	*dest = value & 0x7f;
1089 	err = wacom_led_control(wacom);
1090 
1091 	mutex_unlock(&wacom->lock);
1092 
1093 	return err < 0 ? err : count;
1094 }
1095 
1096 #define DEVICE_LUMINANCE_ATTR(name, field)				\
1097 static ssize_t wacom_##name##_luminance_store(struct device *dev,	\
1098 	struct device_attribute *attr, const char *buf, size_t count)	\
1099 {									\
1100 	struct hid_device *hdev = to_hid_device(dev);\
1101 	struct wacom *wacom = hid_get_drvdata(hdev);			\
1102 									\
1103 	return wacom_luminance_store(wacom, &wacom->led.field,		\
1104 				     buf, count);			\
1105 }									\
1106 static ssize_t wacom_##name##_luminance_show(struct device *dev,	\
1107 	struct device_attribute *attr, char *buf)			\
1108 {									\
1109 	struct wacom *wacom = dev_get_drvdata(dev);			\
1110 	return scnprintf(buf, PAGE_SIZE, "%d\n", wacom->led.field);	\
1111 }									\
1112 static DEVICE_ATTR(name##_luminance, DEV_ATTR_RW_PERM,			\
1113 		   wacom_##name##_luminance_show,			\
1114 		   wacom_##name##_luminance_store)
1115 
1116 DEVICE_LUMINANCE_ATTR(status0, llv);
1117 DEVICE_LUMINANCE_ATTR(status1, hlv);
1118 DEVICE_LUMINANCE_ATTR(buttons, img_lum);
1119 
wacom_button_image_store(struct device * dev,int button_id,const char * buf,size_t count)1120 static ssize_t wacom_button_image_store(struct device *dev, int button_id,
1121 					const char *buf, size_t count)
1122 {
1123 	struct hid_device *hdev = to_hid_device(dev);
1124 	struct wacom *wacom = hid_get_drvdata(hdev);
1125 	int err;
1126 	unsigned len;
1127 	u8 xfer_id;
1128 
1129 	if (hdev->bus == BUS_BLUETOOTH) {
1130 		len = 256;
1131 		xfer_id = WAC_CMD_ICON_BT_XFER;
1132 	} else {
1133 		len = 1024;
1134 		xfer_id = WAC_CMD_ICON_XFER;
1135 	}
1136 
1137 	if (count != len)
1138 		return -EINVAL;
1139 
1140 	mutex_lock(&wacom->lock);
1141 
1142 	err = wacom_led_putimage(wacom, button_id, xfer_id, len, buf);
1143 
1144 	mutex_unlock(&wacom->lock);
1145 
1146 	return err < 0 ? err : count;
1147 }
1148 
1149 #define DEVICE_BTNIMG_ATTR(BUTTON_ID)					\
1150 static ssize_t wacom_btnimg##BUTTON_ID##_store(struct device *dev,	\
1151 	struct device_attribute *attr, const char *buf, size_t count)	\
1152 {									\
1153 	return wacom_button_image_store(dev, BUTTON_ID, buf, count);	\
1154 }									\
1155 static DEVICE_ATTR(button##BUTTON_ID##_rawimg, DEV_ATTR_WO_PERM,	\
1156 		   NULL, wacom_btnimg##BUTTON_ID##_store)
1157 
1158 DEVICE_BTNIMG_ATTR(0);
1159 DEVICE_BTNIMG_ATTR(1);
1160 DEVICE_BTNIMG_ATTR(2);
1161 DEVICE_BTNIMG_ATTR(3);
1162 DEVICE_BTNIMG_ATTR(4);
1163 DEVICE_BTNIMG_ATTR(5);
1164 DEVICE_BTNIMG_ATTR(6);
1165 DEVICE_BTNIMG_ATTR(7);
1166 
1167 static struct attribute *cintiq_led_attrs[] = {
1168 	&dev_attr_status_led0_select.attr,
1169 	&dev_attr_status_led1_select.attr,
1170 	NULL
1171 };
1172 
1173 static const struct attribute_group cintiq_led_attr_group = {
1174 	.name = "wacom_led",
1175 	.attrs = cintiq_led_attrs,
1176 };
1177 
1178 static struct attribute *intuos4_led_attrs[] = {
1179 	&dev_attr_status0_luminance.attr,
1180 	&dev_attr_status1_luminance.attr,
1181 	&dev_attr_status_led0_select.attr,
1182 	&dev_attr_buttons_luminance.attr,
1183 	&dev_attr_button0_rawimg.attr,
1184 	&dev_attr_button1_rawimg.attr,
1185 	&dev_attr_button2_rawimg.attr,
1186 	&dev_attr_button3_rawimg.attr,
1187 	&dev_attr_button4_rawimg.attr,
1188 	&dev_attr_button5_rawimg.attr,
1189 	&dev_attr_button6_rawimg.attr,
1190 	&dev_attr_button7_rawimg.attr,
1191 	NULL
1192 };
1193 
1194 static const struct attribute_group intuos4_led_attr_group = {
1195 	.name = "wacom_led",
1196 	.attrs = intuos4_led_attrs,
1197 };
1198 
1199 static struct attribute *intuos5_led_attrs[] = {
1200 	&dev_attr_status0_luminance.attr,
1201 	&dev_attr_status_led0_select.attr,
1202 	NULL
1203 };
1204 
1205 static const struct attribute_group intuos5_led_attr_group = {
1206 	.name = "wacom_led",
1207 	.attrs = intuos5_led_attrs,
1208 };
1209 
1210 static struct attribute *generic_led_attrs[] = {
1211 	&dev_attr_status0_luminance.attr,
1212 	&dev_attr_status_led0_select.attr,
1213 	NULL
1214 };
1215 
1216 static const struct attribute_group generic_led_attr_group = {
1217 	.name = "wacom_led",
1218 	.attrs = generic_led_attrs,
1219 };
1220 
1221 struct wacom_sysfs_group_devres {
1222 	const struct attribute_group *group;
1223 	struct kobject *root;
1224 };
1225 
wacom_devm_sysfs_group_release(struct device * dev,void * res)1226 static void wacom_devm_sysfs_group_release(struct device *dev, void *res)
1227 {
1228 	struct wacom_sysfs_group_devres *devres = res;
1229 	struct kobject *kobj = devres->root;
1230 
1231 	dev_dbg(dev, "%s: dropping reference to %s\n",
1232 		__func__, devres->group->name);
1233 	sysfs_remove_group(kobj, devres->group);
1234 }
1235 
__wacom_devm_sysfs_create_group(struct wacom * wacom,struct kobject * root,const struct attribute_group * group)1236 static int __wacom_devm_sysfs_create_group(struct wacom *wacom,
1237 					   struct kobject *root,
1238 					   const struct attribute_group *group)
1239 {
1240 	struct wacom_sysfs_group_devres *devres;
1241 	int error;
1242 
1243 	devres = devres_alloc(wacom_devm_sysfs_group_release,
1244 			      sizeof(struct wacom_sysfs_group_devres),
1245 			      GFP_KERNEL);
1246 	if (!devres)
1247 		return -ENOMEM;
1248 
1249 	devres->group = group;
1250 	devres->root = root;
1251 
1252 	error = sysfs_create_group(devres->root, group);
1253 	if (error) {
1254 		devres_free(devres);
1255 		return error;
1256 	}
1257 
1258 	devres_add(&wacom->hdev->dev, devres);
1259 
1260 	return 0;
1261 }
1262 
wacom_devm_sysfs_create_group(struct wacom * wacom,const struct attribute_group * group)1263 static int wacom_devm_sysfs_create_group(struct wacom *wacom,
1264 					 const struct attribute_group *group)
1265 {
1266 	return __wacom_devm_sysfs_create_group(wacom, &wacom->hdev->dev.kobj,
1267 					       group);
1268 }
1269 
wacom_devm_kfifo_release(struct device * dev,void * res)1270 static void wacom_devm_kfifo_release(struct device *dev, void *res)
1271 {
1272 	struct kfifo_rec_ptr_2 *devres = res;
1273 
1274 	kfifo_free(devres);
1275 }
1276 
wacom_devm_kfifo_alloc(struct wacom * wacom)1277 static int wacom_devm_kfifo_alloc(struct wacom *wacom)
1278 {
1279 	struct wacom_wac *wacom_wac = &wacom->wacom_wac;
1280 	struct kfifo_rec_ptr_2 *pen_fifo;
1281 	int error;
1282 
1283 	pen_fifo = devres_alloc(wacom_devm_kfifo_release,
1284 			      sizeof(struct kfifo_rec_ptr_2),
1285 			      GFP_KERNEL);
1286 
1287 	if (!pen_fifo)
1288 		return -ENOMEM;
1289 
1290 	error = kfifo_alloc(pen_fifo, WACOM_PKGLEN_MAX, GFP_KERNEL);
1291 	if (error) {
1292 		devres_free(pen_fifo);
1293 		return error;
1294 	}
1295 
1296 	devres_add(&wacom->hdev->dev, pen_fifo);
1297 	wacom_wac->pen_fifo = pen_fifo;
1298 
1299 	return 0;
1300 }
1301 
wacom_leds_brightness_get(struct wacom_led * led)1302 enum led_brightness wacom_leds_brightness_get(struct wacom_led *led)
1303 {
1304 	struct wacom *wacom = led->wacom;
1305 
1306 	if (wacom->led.max_hlv)
1307 		return led->hlv * LED_FULL / wacom->led.max_hlv;
1308 
1309 	if (wacom->led.max_llv)
1310 		return led->llv * LED_FULL / wacom->led.max_llv;
1311 
1312 	/* device doesn't support brightness tuning */
1313 	return LED_FULL;
1314 }
1315 
__wacom_led_brightness_get(struct led_classdev * cdev)1316 static enum led_brightness __wacom_led_brightness_get(struct led_classdev *cdev)
1317 {
1318 	struct wacom_led *led = container_of(cdev, struct wacom_led, cdev);
1319 	struct wacom *wacom = led->wacom;
1320 
1321 	if (wacom->led.groups[led->group].select != led->id)
1322 		return LED_OFF;
1323 
1324 	return wacom_leds_brightness_get(led);
1325 }
1326 
wacom_led_brightness_set(struct led_classdev * cdev,enum led_brightness brightness)1327 static int wacom_led_brightness_set(struct led_classdev *cdev,
1328 				    enum led_brightness brightness)
1329 {
1330 	struct wacom_led *led = container_of(cdev, struct wacom_led, cdev);
1331 	struct wacom *wacom = led->wacom;
1332 	int error;
1333 
1334 	mutex_lock(&wacom->lock);
1335 
1336 	if (!wacom->led.groups || (brightness == LED_OFF &&
1337 	    wacom->led.groups[led->group].select != led->id)) {
1338 		error = 0;
1339 		goto out;
1340 	}
1341 
1342 	led->llv = wacom->led.llv = wacom->led.max_llv * brightness / LED_FULL;
1343 	led->hlv = wacom->led.hlv = wacom->led.max_hlv * brightness / LED_FULL;
1344 
1345 	wacom->led.groups[led->group].select = led->id;
1346 
1347 	error = wacom_led_control(wacom);
1348 
1349 out:
1350 	mutex_unlock(&wacom->lock);
1351 
1352 	return error;
1353 }
1354 
wacom_led_readonly_brightness_set(struct led_classdev * cdev,enum led_brightness brightness)1355 static void wacom_led_readonly_brightness_set(struct led_classdev *cdev,
1356 					       enum led_brightness brightness)
1357 {
1358 }
1359 
wacom_led_register_one(struct device * dev,struct wacom * wacom,struct wacom_led * led,unsigned int group,unsigned int id,bool read_only)1360 static int wacom_led_register_one(struct device *dev, struct wacom *wacom,
1361 				  struct wacom_led *led, unsigned int group,
1362 				  unsigned int id, bool read_only)
1363 {
1364 	int error;
1365 	char *name;
1366 
1367 	name = devm_kasprintf(dev, GFP_KERNEL,
1368 			      "%s::wacom-%d.%d",
1369 			      dev_name(dev),
1370 			      group,
1371 			      id);
1372 	if (!name)
1373 		return -ENOMEM;
1374 
1375 	if (!read_only) {
1376 		led->trigger.name = name;
1377 		error = devm_led_trigger_register(dev, &led->trigger);
1378 		if (error) {
1379 			hid_err(wacom->hdev,
1380 				"failed to register LED trigger %s: %d\n",
1381 				led->cdev.name, error);
1382 			return error;
1383 		}
1384 	}
1385 
1386 	led->group = group;
1387 	led->id = id;
1388 	led->wacom = wacom;
1389 	led->llv = wacom->led.llv;
1390 	led->hlv = wacom->led.hlv;
1391 	led->cdev.name = name;
1392 	led->cdev.max_brightness = LED_FULL;
1393 	led->cdev.flags = LED_HW_PLUGGABLE;
1394 	led->cdev.brightness_get = __wacom_led_brightness_get;
1395 	if (!read_only) {
1396 		led->cdev.brightness_set_blocking = wacom_led_brightness_set;
1397 		led->cdev.default_trigger = led->cdev.name;
1398 	} else {
1399 		led->cdev.brightness_set = wacom_led_readonly_brightness_set;
1400 	}
1401 
1402 	error = devm_led_classdev_register(dev, &led->cdev);
1403 	if (error) {
1404 		hid_err(wacom->hdev,
1405 			"failed to register LED %s: %d\n",
1406 			led->cdev.name, error);
1407 		led->cdev.name = NULL;
1408 		return error;
1409 	}
1410 
1411 	return 0;
1412 }
1413 
wacom_led_groups_release_one(void * data)1414 static void wacom_led_groups_release_one(void *data)
1415 {
1416 	struct wacom_group_leds *group = data;
1417 
1418 	devres_release_group(group->dev, group);
1419 }
1420 
wacom_led_groups_alloc_and_register_one(struct device * dev,struct wacom * wacom,int group_id,int count,bool read_only)1421 static int wacom_led_groups_alloc_and_register_one(struct device *dev,
1422 						   struct wacom *wacom,
1423 						   int group_id, int count,
1424 						   bool read_only)
1425 {
1426 	struct wacom_led *leds;
1427 	int i, error;
1428 
1429 	if (group_id >= wacom->led.count || count <= 0)
1430 		return -EINVAL;
1431 
1432 	if (!devres_open_group(dev, &wacom->led.groups[group_id], GFP_KERNEL))
1433 		return -ENOMEM;
1434 
1435 	leds = devm_kcalloc(dev, count, sizeof(struct wacom_led), GFP_KERNEL);
1436 	if (!leds) {
1437 		error = -ENOMEM;
1438 		goto err;
1439 	}
1440 
1441 	wacom->led.groups[group_id].leds = leds;
1442 	wacom->led.groups[group_id].count = count;
1443 
1444 	for (i = 0; i < count; i++) {
1445 		error = wacom_led_register_one(dev, wacom, &leds[i],
1446 					       group_id, i, read_only);
1447 		if (error)
1448 			goto err;
1449 	}
1450 
1451 	wacom->led.groups[group_id].dev = dev;
1452 
1453 	devres_close_group(dev, &wacom->led.groups[group_id]);
1454 
1455 	/*
1456 	 * There is a bug (?) in devm_led_classdev_register() in which its
1457 	 * increments the refcount of the parent. If the parent is an input
1458 	 * device, that means the ref count never reaches 0 when
1459 	 * devm_input_device_release() gets called.
1460 	 * This means that the LEDs are still there after disconnect.
1461 	 * Manually force the release of the group so that the leds are released
1462 	 * once we are done using them.
1463 	 */
1464 	error = devm_add_action_or_reset(&wacom->hdev->dev,
1465 					 wacom_led_groups_release_one,
1466 					 &wacom->led.groups[group_id]);
1467 	if (error)
1468 		return error;
1469 
1470 	return 0;
1471 
1472 err:
1473 	devres_release_group(dev, &wacom->led.groups[group_id]);
1474 	return error;
1475 }
1476 
wacom_led_find(struct wacom * wacom,unsigned int group_id,unsigned int id)1477 struct wacom_led *wacom_led_find(struct wacom *wacom, unsigned int group_id,
1478 				 unsigned int id)
1479 {
1480 	struct wacom_group_leds *group;
1481 
1482 	if (group_id >= wacom->led.count)
1483 		return NULL;
1484 
1485 	group = &wacom->led.groups[group_id];
1486 
1487 	if (!group->leds)
1488 		return NULL;
1489 
1490 	id %= group->count;
1491 
1492 	return &group->leds[id];
1493 }
1494 
1495 /*
1496  * wacom_led_next: gives the next available led with a wacom trigger.
1497  *
1498  * returns the next available struct wacom_led which has its default trigger
1499  * or the current one if none is available.
1500  */
wacom_led_next(struct wacom * wacom,struct wacom_led * cur)1501 struct wacom_led *wacom_led_next(struct wacom *wacom, struct wacom_led *cur)
1502 {
1503 	struct wacom_led *next_led;
1504 	int group, next;
1505 
1506 	if (!wacom || !cur)
1507 		return NULL;
1508 
1509 	group = cur->group;
1510 	next = cur->id;
1511 
1512 	do {
1513 		next_led = wacom_led_find(wacom, group, ++next);
1514 		if (!next_led || next_led == cur)
1515 			return next_led;
1516 	} while (next_led->cdev.trigger != &next_led->trigger);
1517 
1518 	return next_led;
1519 }
1520 
wacom_led_groups_release(void * data)1521 static void wacom_led_groups_release(void *data)
1522 {
1523 	struct wacom *wacom = data;
1524 
1525 	wacom->led.groups = NULL;
1526 	wacom->led.count = 0;
1527 }
1528 
wacom_led_groups_allocate(struct wacom * wacom,int count)1529 static int wacom_led_groups_allocate(struct wacom *wacom, int count)
1530 {
1531 	struct device *dev = &wacom->hdev->dev;
1532 	struct wacom_group_leds *groups;
1533 	int error;
1534 
1535 	groups = devm_kcalloc(dev, count, sizeof(struct wacom_group_leds),
1536 			      GFP_KERNEL);
1537 	if (!groups)
1538 		return -ENOMEM;
1539 
1540 	error = devm_add_action_or_reset(dev, wacom_led_groups_release, wacom);
1541 	if (error)
1542 		return error;
1543 
1544 	wacom->led.groups = groups;
1545 	wacom->led.count = count;
1546 
1547 	return 0;
1548 }
1549 
wacom_leds_alloc_and_register(struct wacom * wacom,int group_count,int led_per_group,bool read_only)1550 static int wacom_leds_alloc_and_register(struct wacom *wacom, int group_count,
1551 					 int led_per_group, bool read_only)
1552 {
1553 	struct device *dev;
1554 	int i, error;
1555 
1556 	if (!wacom->wacom_wac.pad_input)
1557 		return -EINVAL;
1558 
1559 	dev = &wacom->wacom_wac.pad_input->dev;
1560 
1561 	error = wacom_led_groups_allocate(wacom, group_count);
1562 	if (error)
1563 		return error;
1564 
1565 	for (i = 0; i < group_count; i++) {
1566 		error = wacom_led_groups_alloc_and_register_one(dev, wacom, i,
1567 								led_per_group,
1568 								read_only);
1569 		if (error)
1570 			return error;
1571 	}
1572 
1573 	return 0;
1574 }
1575 
wacom_initialize_leds(struct wacom * wacom)1576 int wacom_initialize_leds(struct wacom *wacom)
1577 {
1578 	int error;
1579 
1580 	if (!(wacom->wacom_wac.features.device_type & WACOM_DEVICETYPE_PAD))
1581 		return 0;
1582 
1583 	/* Initialize default values */
1584 	switch (wacom->wacom_wac.features.type) {
1585 	case HID_GENERIC:
1586 		if (!wacom->generic_has_leds)
1587 			return 0;
1588 		wacom->led.llv = 100;
1589 		wacom->led.max_llv = 100;
1590 
1591 		error = wacom_leds_alloc_and_register(wacom, 1, 4, false);
1592 		if (error) {
1593 			hid_err(wacom->hdev,
1594 				"cannot create leds err: %d\n", error);
1595 			return error;
1596 		}
1597 
1598 		error = wacom_devm_sysfs_create_group(wacom,
1599 						      &generic_led_attr_group);
1600 		break;
1601 
1602 	case INTUOS4S:
1603 	case INTUOS4:
1604 	case INTUOS4WL:
1605 	case INTUOS4L:
1606 		wacom->led.llv = 10;
1607 		wacom->led.hlv = 20;
1608 		wacom->led.max_llv = 127;
1609 		wacom->led.max_hlv = 127;
1610 		wacom->led.img_lum = 10;
1611 
1612 		error = wacom_leds_alloc_and_register(wacom, 1, 4, false);
1613 		if (error) {
1614 			hid_err(wacom->hdev,
1615 				"cannot create leds err: %d\n", error);
1616 			return error;
1617 		}
1618 
1619 		error = wacom_devm_sysfs_create_group(wacom,
1620 						      &intuos4_led_attr_group);
1621 		break;
1622 
1623 	case WACOM_24HD:
1624 	case WACOM_21UX2:
1625 		wacom->led.llv = 0;
1626 		wacom->led.hlv = 0;
1627 		wacom->led.img_lum = 0;
1628 
1629 		error = wacom_leds_alloc_and_register(wacom, 2, 4, false);
1630 		if (error) {
1631 			hid_err(wacom->hdev,
1632 				"cannot create leds err: %d\n", error);
1633 			return error;
1634 		}
1635 
1636 		error = wacom_devm_sysfs_create_group(wacom,
1637 						      &cintiq_led_attr_group);
1638 		break;
1639 
1640 	case INTUOS5S:
1641 	case INTUOS5:
1642 	case INTUOS5L:
1643 	case INTUOSPS:
1644 	case INTUOSPM:
1645 	case INTUOSPL:
1646 		wacom->led.llv = 32;
1647 		wacom->led.max_llv = 96;
1648 
1649 		error = wacom_leds_alloc_and_register(wacom, 1, 4, false);
1650 		if (error) {
1651 			hid_err(wacom->hdev,
1652 				"cannot create leds err: %d\n", error);
1653 			return error;
1654 		}
1655 
1656 		error = wacom_devm_sysfs_create_group(wacom,
1657 						      &intuos5_led_attr_group);
1658 		break;
1659 
1660 	case INTUOSP2_BT:
1661 		wacom->led.llv = 50;
1662 		wacom->led.max_llv = 100;
1663 		error = wacom_leds_alloc_and_register(wacom, 1, 4, false);
1664 		if (error) {
1665 			hid_err(wacom->hdev,
1666 				"cannot create leds err: %d\n", error);
1667 			return error;
1668 		}
1669 		return 0;
1670 
1671 	case REMOTE:
1672 		wacom->led.llv = 255;
1673 		wacom->led.max_llv = 255;
1674 		error = wacom_led_groups_allocate(wacom, 5);
1675 		if (error) {
1676 			hid_err(wacom->hdev,
1677 				"cannot create leds err: %d\n", error);
1678 			return error;
1679 		}
1680 		return 0;
1681 
1682 	default:
1683 		return 0;
1684 	}
1685 
1686 	if (error) {
1687 		hid_err(wacom->hdev,
1688 			"cannot create sysfs group err: %d\n", error);
1689 		return error;
1690 	}
1691 
1692 	return 0;
1693 }
1694 
wacom_init_work(struct work_struct * work)1695 static void wacom_init_work(struct work_struct *work)
1696 {
1697 	struct wacom *wacom = container_of(work, struct wacom, init_work.work);
1698 
1699 	_wacom_query_tablet_data(wacom);
1700 	wacom_led_control(wacom);
1701 }
1702 
wacom_query_tablet_data(struct wacom * wacom)1703 static void wacom_query_tablet_data(struct wacom *wacom)
1704 {
1705 	schedule_delayed_work(&wacom->init_work, msecs_to_jiffies(1000));
1706 }
1707 
1708 static enum power_supply_property wacom_battery_props[] = {
1709 	POWER_SUPPLY_PROP_MODEL_NAME,
1710 	POWER_SUPPLY_PROP_PRESENT,
1711 	POWER_SUPPLY_PROP_STATUS,
1712 	POWER_SUPPLY_PROP_SCOPE,
1713 	POWER_SUPPLY_PROP_CAPACITY
1714 };
1715 
wacom_battery_get_property(struct power_supply * psy,enum power_supply_property psp,union power_supply_propval * val)1716 static int wacom_battery_get_property(struct power_supply *psy,
1717 				      enum power_supply_property psp,
1718 				      union power_supply_propval *val)
1719 {
1720 	struct wacom_battery *battery = power_supply_get_drvdata(psy);
1721 	int ret = 0;
1722 
1723 	switch (psp) {
1724 		case POWER_SUPPLY_PROP_MODEL_NAME:
1725 			val->strval = battery->wacom->wacom_wac.name;
1726 			break;
1727 		case POWER_SUPPLY_PROP_PRESENT:
1728 			val->intval = battery->bat_connected;
1729 			break;
1730 		case POWER_SUPPLY_PROP_SCOPE:
1731 			val->intval = POWER_SUPPLY_SCOPE_DEVICE;
1732 			break;
1733 		case POWER_SUPPLY_PROP_CAPACITY:
1734 			val->intval = battery->battery_capacity;
1735 			break;
1736 		case POWER_SUPPLY_PROP_STATUS:
1737 			if (battery->bat_status != WACOM_POWER_SUPPLY_STATUS_AUTO)
1738 				val->intval = battery->bat_status;
1739 			else if (battery->bat_charging)
1740 				val->intval = POWER_SUPPLY_STATUS_CHARGING;
1741 			else if (battery->battery_capacity == 100 &&
1742 				    battery->ps_connected)
1743 				val->intval = POWER_SUPPLY_STATUS_FULL;
1744 			else if (battery->ps_connected)
1745 				val->intval = POWER_SUPPLY_STATUS_NOT_CHARGING;
1746 			else
1747 				val->intval = POWER_SUPPLY_STATUS_DISCHARGING;
1748 			break;
1749 		default:
1750 			ret = -EINVAL;
1751 			break;
1752 	}
1753 
1754 	return ret;
1755 }
1756 
__wacom_initialize_battery(struct wacom * wacom,struct wacom_battery * battery)1757 static int __wacom_initialize_battery(struct wacom *wacom,
1758 				      struct wacom_battery *battery)
1759 {
1760 	static atomic_t battery_no = ATOMIC_INIT(0);
1761 	struct device *dev = &wacom->hdev->dev;
1762 	struct power_supply_config psy_cfg = { .drv_data = battery, };
1763 	struct power_supply *ps_bat;
1764 	struct power_supply_desc *bat_desc = &battery->bat_desc;
1765 	unsigned long n;
1766 	int error;
1767 
1768 	if (!devres_open_group(dev, bat_desc, GFP_KERNEL))
1769 		return -ENOMEM;
1770 
1771 	battery->wacom = wacom;
1772 
1773 	n = atomic_inc_return(&battery_no) - 1;
1774 
1775 	bat_desc->properties = wacom_battery_props;
1776 	bat_desc->num_properties = ARRAY_SIZE(wacom_battery_props);
1777 	bat_desc->get_property = wacom_battery_get_property;
1778 	sprintf(battery->bat_name, "wacom_battery_%ld", n);
1779 	bat_desc->name = battery->bat_name;
1780 	bat_desc->type = POWER_SUPPLY_TYPE_BATTERY;
1781 	bat_desc->use_for_apm = 0;
1782 
1783 	ps_bat = devm_power_supply_register(dev, bat_desc, &psy_cfg);
1784 	if (IS_ERR(ps_bat)) {
1785 		error = PTR_ERR(ps_bat);
1786 		goto err;
1787 	}
1788 
1789 	power_supply_powers(ps_bat, &wacom->hdev->dev);
1790 
1791 	battery->battery = ps_bat;
1792 
1793 	devres_close_group(dev, bat_desc);
1794 	return 0;
1795 
1796 err:
1797 	devres_release_group(dev, bat_desc);
1798 	return error;
1799 }
1800 
wacom_initialize_battery(struct wacom * wacom)1801 static int wacom_initialize_battery(struct wacom *wacom)
1802 {
1803 	if (wacom->wacom_wac.features.quirks & WACOM_QUIRK_BATTERY)
1804 		return __wacom_initialize_battery(wacom, &wacom->battery);
1805 
1806 	return 0;
1807 }
1808 
wacom_destroy_battery(struct wacom * wacom)1809 static void wacom_destroy_battery(struct wacom *wacom)
1810 {
1811 	if (wacom->battery.battery) {
1812 		devres_release_group(&wacom->hdev->dev,
1813 				     &wacom->battery.bat_desc);
1814 		wacom->battery.battery = NULL;
1815 	}
1816 }
1817 
wacom_show_speed(struct device * dev,struct device_attribute * attr,char * buf)1818 static ssize_t wacom_show_speed(struct device *dev,
1819 				struct device_attribute
1820 				*attr, char *buf)
1821 {
1822 	struct hid_device *hdev = to_hid_device(dev);
1823 	struct wacom *wacom = hid_get_drvdata(hdev);
1824 
1825 	return sysfs_emit(buf, "%i\n", wacom->wacom_wac.bt_high_speed);
1826 }
1827 
wacom_store_speed(struct device * dev,struct device_attribute * attr,const char * buf,size_t count)1828 static ssize_t wacom_store_speed(struct device *dev,
1829 				struct device_attribute *attr,
1830 				const char *buf, size_t count)
1831 {
1832 	struct hid_device *hdev = to_hid_device(dev);
1833 	struct wacom *wacom = hid_get_drvdata(hdev);
1834 	u8 new_speed;
1835 
1836 	if (kstrtou8(buf, 0, &new_speed))
1837 		return -EINVAL;
1838 
1839 	if (new_speed != 0 && new_speed != 1)
1840 		return -EINVAL;
1841 
1842 	wacom_bt_query_tablet_data(hdev, new_speed, &wacom->wacom_wac.features);
1843 
1844 	return count;
1845 }
1846 
1847 static DEVICE_ATTR(speed, DEV_ATTR_RW_PERM,
1848 		wacom_show_speed, wacom_store_speed);
1849 
1850 
wacom_show_remote_mode(struct kobject * kobj,struct kobj_attribute * kattr,char * buf,int index)1851 static ssize_t wacom_show_remote_mode(struct kobject *kobj,
1852 				      struct kobj_attribute *kattr,
1853 				      char *buf, int index)
1854 {
1855 	struct device *dev = kobj_to_dev(kobj->parent);
1856 	struct hid_device *hdev = to_hid_device(dev);
1857 	struct wacom *wacom = hid_get_drvdata(hdev);
1858 	u8 mode;
1859 
1860 	mode = wacom->led.groups[index].select;
1861 	return sprintf(buf, "%d\n", mode < 3 ? mode : -1);
1862 }
1863 
1864 #define DEVICE_EKR_ATTR_GROUP(SET_ID)					\
1865 static ssize_t wacom_show_remote##SET_ID##_mode(struct kobject *kobj,	\
1866 			       struct kobj_attribute *kattr, char *buf)	\
1867 {									\
1868 	return wacom_show_remote_mode(kobj, kattr, buf, SET_ID);	\
1869 }									\
1870 static struct kobj_attribute remote##SET_ID##_mode_attr = {		\
1871 	.attr = {.name = "remote_mode",					\
1872 		.mode = DEV_ATTR_RO_PERM},				\
1873 	.show = wacom_show_remote##SET_ID##_mode,			\
1874 };									\
1875 static struct attribute *remote##SET_ID##_serial_attrs[] = {		\
1876 	&remote##SET_ID##_mode_attr.attr,				\
1877 	NULL								\
1878 };									\
1879 static const struct attribute_group remote##SET_ID##_serial_group = {	\
1880 	.name = NULL,							\
1881 	.attrs = remote##SET_ID##_serial_attrs,				\
1882 }
1883 
1884 DEVICE_EKR_ATTR_GROUP(0);
1885 DEVICE_EKR_ATTR_GROUP(1);
1886 DEVICE_EKR_ATTR_GROUP(2);
1887 DEVICE_EKR_ATTR_GROUP(3);
1888 DEVICE_EKR_ATTR_GROUP(4);
1889 
wacom_remote_create_attr_group(struct wacom * wacom,__u32 serial,int index)1890 static int wacom_remote_create_attr_group(struct wacom *wacom, __u32 serial,
1891 					  int index)
1892 {
1893 	int error = 0;
1894 	struct wacom_remote *remote = wacom->remote;
1895 
1896 	remote->remotes[index].group.name = devm_kasprintf(&wacom->hdev->dev,
1897 							  GFP_KERNEL,
1898 							  "%d", serial);
1899 	if (!remote->remotes[index].group.name)
1900 		return -ENOMEM;
1901 
1902 	error = __wacom_devm_sysfs_create_group(wacom, remote->remote_dir,
1903 						&remote->remotes[index].group);
1904 	if (error) {
1905 		remote->remotes[index].group.name = NULL;
1906 		hid_err(wacom->hdev,
1907 			"cannot create sysfs group err: %d\n", error);
1908 		return error;
1909 	}
1910 
1911 	return 0;
1912 }
1913 
wacom_cmd_unpair_remote(struct wacom * wacom,unsigned char selector)1914 static int wacom_cmd_unpair_remote(struct wacom *wacom, unsigned char selector)
1915 {
1916 	const size_t buf_size = 2;
1917 	unsigned char *buf;
1918 	int retval;
1919 
1920 	buf = kzalloc(buf_size, GFP_KERNEL);
1921 	if (!buf)
1922 		return -ENOMEM;
1923 
1924 	buf[0] = WAC_CMD_DELETE_PAIRING;
1925 	buf[1] = selector;
1926 
1927 	retval = wacom_set_report(wacom->hdev, HID_OUTPUT_REPORT, buf,
1928 				  buf_size, WAC_CMD_RETRIES);
1929 	kfree(buf);
1930 
1931 	return retval;
1932 }
1933 
wacom_store_unpair_remote(struct kobject * kobj,struct kobj_attribute * attr,const char * buf,size_t count)1934 static ssize_t wacom_store_unpair_remote(struct kobject *kobj,
1935 					 struct kobj_attribute *attr,
1936 					 const char *buf, size_t count)
1937 {
1938 	unsigned char selector = 0;
1939 	struct device *dev = kobj_to_dev(kobj->parent);
1940 	struct hid_device *hdev = to_hid_device(dev);
1941 	struct wacom *wacom = hid_get_drvdata(hdev);
1942 	int err;
1943 
1944 	if (!strncmp(buf, "*\n", 2)) {
1945 		selector = WAC_CMD_UNPAIR_ALL;
1946 	} else {
1947 		hid_info(wacom->hdev, "remote: unrecognized unpair code: %s\n",
1948 			 buf);
1949 		return -1;
1950 	}
1951 
1952 	mutex_lock(&wacom->lock);
1953 
1954 	err = wacom_cmd_unpair_remote(wacom, selector);
1955 	mutex_unlock(&wacom->lock);
1956 
1957 	return err < 0 ? err : count;
1958 }
1959 
1960 static struct kobj_attribute unpair_remote_attr = {
1961 	.attr = {.name = "unpair_remote", .mode = 0200},
1962 	.store = wacom_store_unpair_remote,
1963 };
1964 
1965 static const struct attribute *remote_unpair_attrs[] = {
1966 	&unpair_remote_attr.attr,
1967 	NULL
1968 };
1969 
wacom_remotes_destroy(void * data)1970 static void wacom_remotes_destroy(void *data)
1971 {
1972 	struct wacom *wacom = data;
1973 	struct wacom_remote *remote = wacom->remote;
1974 
1975 	if (!remote)
1976 		return;
1977 
1978 	kobject_put(remote->remote_dir);
1979 	kfifo_free(&remote->remote_fifo);
1980 	wacom->remote = NULL;
1981 }
1982 
wacom_initialize_remotes(struct wacom * wacom)1983 static int wacom_initialize_remotes(struct wacom *wacom)
1984 {
1985 	int error = 0;
1986 	struct wacom_remote *remote;
1987 	int i;
1988 
1989 	if (wacom->wacom_wac.features.type != REMOTE)
1990 		return 0;
1991 
1992 	remote = devm_kzalloc(&wacom->hdev->dev, sizeof(*wacom->remote),
1993 			      GFP_KERNEL);
1994 	if (!remote)
1995 		return -ENOMEM;
1996 
1997 	wacom->remote = remote;
1998 
1999 	spin_lock_init(&remote->remote_lock);
2000 
2001 	error = kfifo_alloc(&remote->remote_fifo,
2002 			5 * sizeof(struct wacom_remote_data),
2003 			GFP_KERNEL);
2004 	if (error) {
2005 		hid_err(wacom->hdev, "failed allocating remote_fifo\n");
2006 		return -ENOMEM;
2007 	}
2008 
2009 	remote->remotes[0].group = remote0_serial_group;
2010 	remote->remotes[1].group = remote1_serial_group;
2011 	remote->remotes[2].group = remote2_serial_group;
2012 	remote->remotes[3].group = remote3_serial_group;
2013 	remote->remotes[4].group = remote4_serial_group;
2014 
2015 	remote->remote_dir = kobject_create_and_add("wacom_remote",
2016 						    &wacom->hdev->dev.kobj);
2017 	if (!remote->remote_dir)
2018 		return -ENOMEM;
2019 
2020 	error = sysfs_create_files(remote->remote_dir, remote_unpair_attrs);
2021 
2022 	if (error) {
2023 		hid_err(wacom->hdev,
2024 			"cannot create sysfs group err: %d\n", error);
2025 		return error;
2026 	}
2027 
2028 	for (i = 0; i < WACOM_MAX_REMOTES; i++) {
2029 		wacom->led.groups[i].select = WACOM_STATUS_UNKNOWN;
2030 		remote->remotes[i].serial = 0;
2031 	}
2032 
2033 	error = devm_add_action_or_reset(&wacom->hdev->dev,
2034 					 wacom_remotes_destroy, wacom);
2035 	if (error)
2036 		return error;
2037 
2038 	return 0;
2039 }
2040 
wacom_allocate_input(struct wacom * wacom)2041 static struct input_dev *wacom_allocate_input(struct wacom *wacom)
2042 {
2043 	struct input_dev *input_dev;
2044 	struct hid_device *hdev = wacom->hdev;
2045 	struct wacom_wac *wacom_wac = &(wacom->wacom_wac);
2046 
2047 	input_dev = devm_input_allocate_device(&hdev->dev);
2048 	if (!input_dev)
2049 		return NULL;
2050 
2051 	input_dev->name = wacom_wac->features.name;
2052 	input_dev->phys = hdev->phys;
2053 	input_dev->dev.parent = &hdev->dev;
2054 	input_dev->open = wacom_open;
2055 	input_dev->close = wacom_close;
2056 	input_dev->uniq = hdev->uniq;
2057 	input_dev->id.bustype = hdev->bus;
2058 	input_dev->id.vendor  = hdev->vendor;
2059 	input_dev->id.product = wacom_wac->pid ? wacom_wac->pid : hdev->product;
2060 	input_dev->id.version = hdev->version;
2061 	input_set_drvdata(input_dev, wacom);
2062 
2063 	return input_dev;
2064 }
2065 
wacom_allocate_inputs(struct wacom * wacom)2066 static int wacom_allocate_inputs(struct wacom *wacom)
2067 {
2068 	struct wacom_wac *wacom_wac = &(wacom->wacom_wac);
2069 
2070 	wacom_wac->pen_input = wacom_allocate_input(wacom);
2071 	wacom_wac->touch_input = wacom_allocate_input(wacom);
2072 	wacom_wac->pad_input = wacom_allocate_input(wacom);
2073 	if (!wacom_wac->pen_input ||
2074 	    !wacom_wac->touch_input ||
2075 	    !wacom_wac->pad_input)
2076 		return -ENOMEM;
2077 
2078 	wacom_wac->pen_input->name = wacom_wac->pen_name;
2079 	wacom_wac->touch_input->name = wacom_wac->touch_name;
2080 	wacom_wac->pad_input->name = wacom_wac->pad_name;
2081 
2082 	return 0;
2083 }
2084 
wacom_register_inputs(struct wacom * wacom)2085 static int wacom_register_inputs(struct wacom *wacom)
2086 {
2087 	struct input_dev *pen_input_dev, *touch_input_dev, *pad_input_dev;
2088 	struct wacom_wac *wacom_wac = &(wacom->wacom_wac);
2089 	int error = 0;
2090 
2091 	pen_input_dev = wacom_wac->pen_input;
2092 	touch_input_dev = wacom_wac->touch_input;
2093 	pad_input_dev = wacom_wac->pad_input;
2094 
2095 	if (!pen_input_dev || !touch_input_dev || !pad_input_dev)
2096 		return -EINVAL;
2097 
2098 	error = wacom_setup_pen_input_capabilities(pen_input_dev, wacom_wac);
2099 	if (error) {
2100 		/* no pen in use on this interface */
2101 		input_free_device(pen_input_dev);
2102 		wacom_wac->pen_input = NULL;
2103 		pen_input_dev = NULL;
2104 	} else {
2105 		error = input_register_device(pen_input_dev);
2106 		if (error)
2107 			goto fail;
2108 	}
2109 
2110 	error = wacom_setup_touch_input_capabilities(touch_input_dev, wacom_wac);
2111 	if (error) {
2112 		/* no touch in use on this interface */
2113 		input_free_device(touch_input_dev);
2114 		wacom_wac->touch_input = NULL;
2115 		touch_input_dev = NULL;
2116 	} else {
2117 		error = input_register_device(touch_input_dev);
2118 		if (error)
2119 			goto fail;
2120 	}
2121 
2122 	error = wacom_setup_pad_input_capabilities(pad_input_dev, wacom_wac);
2123 	if (error) {
2124 		/* no pad events using this interface */
2125 		input_free_device(pad_input_dev);
2126 		wacom_wac->pad_input = NULL;
2127 		pad_input_dev = NULL;
2128 	} else {
2129 		error = input_register_device(pad_input_dev);
2130 		if (error)
2131 			goto fail;
2132 	}
2133 
2134 	return 0;
2135 
2136 fail:
2137 	wacom_wac->pad_input = NULL;
2138 	wacom_wac->touch_input = NULL;
2139 	wacom_wac->pen_input = NULL;
2140 	return error;
2141 }
2142 
2143 /*
2144  * Not all devices report physical dimensions from HID.
2145  * Compute the default from hardcoded logical dimension
2146  * and resolution before driver overwrites them.
2147  */
wacom_set_default_phy(struct wacom_features * features)2148 static void wacom_set_default_phy(struct wacom_features *features)
2149 {
2150 	if (features->x_resolution) {
2151 		features->x_phy = (features->x_max * 100) /
2152 					features->x_resolution;
2153 		features->y_phy = (features->y_max * 100) /
2154 					features->y_resolution;
2155 	}
2156 }
2157 
wacom_calculate_res(struct wacom_features * features)2158 static void wacom_calculate_res(struct wacom_features *features)
2159 {
2160 	/* set unit to "100th of a mm" for devices not reported by HID */
2161 	if (!features->unit) {
2162 		features->unit = 0x11;
2163 		features->unitExpo = -3;
2164 	}
2165 
2166 	features->x_resolution = wacom_calc_hid_res(features->x_max,
2167 						    features->x_phy,
2168 						    features->unit,
2169 						    features->unitExpo);
2170 	features->y_resolution = wacom_calc_hid_res(features->y_max,
2171 						    features->y_phy,
2172 						    features->unit,
2173 						    features->unitExpo);
2174 }
2175 
wacom_battery_work(struct work_struct * work)2176 void wacom_battery_work(struct work_struct *work)
2177 {
2178 	struct wacom *wacom = container_of(work, struct wacom, battery_work);
2179 
2180 	if ((wacom->wacom_wac.features.quirks & WACOM_QUIRK_BATTERY) &&
2181 	     !wacom->battery.battery) {
2182 		wacom_initialize_battery(wacom);
2183 	}
2184 	else if (!(wacom->wacom_wac.features.quirks & WACOM_QUIRK_BATTERY) &&
2185 		 wacom->battery.battery) {
2186 		wacom_destroy_battery(wacom);
2187 	}
2188 }
2189 
wacom_compute_pktlen(struct hid_device * hdev)2190 static size_t wacom_compute_pktlen(struct hid_device *hdev)
2191 {
2192 	struct hid_report_enum *report_enum;
2193 	struct hid_report *report;
2194 	size_t size = 0;
2195 
2196 	report_enum = hdev->report_enum + HID_INPUT_REPORT;
2197 
2198 	list_for_each_entry(report, &report_enum->report_list, list) {
2199 		size_t report_size = hid_report_len(report);
2200 		if (report_size > size)
2201 			size = report_size;
2202 	}
2203 
2204 	return size;
2205 }
2206 
wacom_update_name(struct wacom * wacom,const char * suffix)2207 static void wacom_update_name(struct wacom *wacom, const char *suffix)
2208 {
2209 	struct wacom_wac *wacom_wac = &wacom->wacom_wac;
2210 	struct wacom_features *features = &wacom_wac->features;
2211 	char name[WACOM_NAME_MAX - 20]; /* Leave some room for suffixes */
2212 
2213 	/* Generic devices name unspecified */
2214 	if ((features->type == HID_GENERIC) && !strcmp("Wacom HID", features->name)) {
2215 		char *product_name = wacom->hdev->name;
2216 
2217 		if (hid_is_usb(wacom->hdev)) {
2218 			struct usb_interface *intf = to_usb_interface(wacom->hdev->dev.parent);
2219 			struct usb_device *dev = interface_to_usbdev(intf);
2220 			product_name = dev->product;
2221 		}
2222 
2223 		if (wacom->hdev->bus == BUS_I2C) {
2224 			snprintf(name, sizeof(name), "%s %X",
2225 				 features->name, wacom->hdev->product);
2226 		} else if (strstr(product_name, "Wacom") ||
2227 			   strstr(product_name, "wacom") ||
2228 			   strstr(product_name, "WACOM")) {
2229 			strlcpy(name, product_name, sizeof(name));
2230 		} else {
2231 			snprintf(name, sizeof(name), "Wacom %s", product_name);
2232 		}
2233 
2234 		/* strip out excess whitespaces */
2235 		while (1) {
2236 			char *gap = strstr(name, "  ");
2237 			if (gap == NULL)
2238 				break;
2239 			/* shift everything including the terminator */
2240 			memmove(gap, gap+1, strlen(gap));
2241 		}
2242 
2243 		/* get rid of trailing whitespace */
2244 		if (name[strlen(name)-1] == ' ')
2245 			name[strlen(name)-1] = '\0';
2246 	} else {
2247 		strlcpy(name, features->name, sizeof(name));
2248 	}
2249 
2250 	snprintf(wacom_wac->name, sizeof(wacom_wac->name), "%s%s",
2251 		 name, suffix);
2252 
2253 	/* Append the device type to the name */
2254 	snprintf(wacom_wac->pen_name, sizeof(wacom_wac->pen_name),
2255 		"%s%s Pen", name, suffix);
2256 	snprintf(wacom_wac->touch_name, sizeof(wacom_wac->touch_name),
2257 		"%s%s Finger", name, suffix);
2258 	snprintf(wacom_wac->pad_name, sizeof(wacom_wac->pad_name),
2259 		"%s%s Pad", name, suffix);
2260 }
2261 
wacom_release_resources(struct wacom * wacom)2262 static void wacom_release_resources(struct wacom *wacom)
2263 {
2264 	struct hid_device *hdev = wacom->hdev;
2265 
2266 	if (!wacom->resources)
2267 		return;
2268 
2269 	devres_release_group(&hdev->dev, wacom);
2270 
2271 	wacom->resources = false;
2272 
2273 	wacom->wacom_wac.pen_input = NULL;
2274 	wacom->wacom_wac.touch_input = NULL;
2275 	wacom->wacom_wac.pad_input = NULL;
2276 }
2277 
wacom_set_shared_values(struct wacom_wac * wacom_wac)2278 static void wacom_set_shared_values(struct wacom_wac *wacom_wac)
2279 {
2280 	if (wacom_wac->features.device_type & WACOM_DEVICETYPE_TOUCH) {
2281 		wacom_wac->shared->type = wacom_wac->features.type;
2282 		wacom_wac->shared->touch_input = wacom_wac->touch_input;
2283 	}
2284 
2285 	if (wacom_wac->has_mute_touch_switch) {
2286 		wacom_wac->shared->has_mute_touch_switch = true;
2287 		/* Hardware touch switch may be off. Wait until
2288 		 * we know the switch state to decide is_touch_on.
2289 		 * Softkey state should be initialized to "on" to
2290 		 * match historic default.
2291 		 */
2292 		if (wacom_wac->is_soft_touch_switch)
2293 			wacom_wac->shared->is_touch_on = true;
2294 	}
2295 
2296 	if (wacom_wac->shared->has_mute_touch_switch &&
2297 	    wacom_wac->shared->touch_input) {
2298 		set_bit(EV_SW, wacom_wac->shared->touch_input->evbit);
2299 		input_set_capability(wacom_wac->shared->touch_input, EV_SW,
2300 				     SW_MUTE_DEVICE);
2301 	}
2302 }
2303 
wacom_parse_and_register(struct wacom * wacom,bool wireless)2304 static int wacom_parse_and_register(struct wacom *wacom, bool wireless)
2305 {
2306 	struct wacom_wac *wacom_wac = &wacom->wacom_wac;
2307 	struct wacom_features *features = &wacom_wac->features;
2308 	struct hid_device *hdev = wacom->hdev;
2309 	int error;
2310 	unsigned int connect_mask = HID_CONNECT_HIDRAW;
2311 
2312 	features->pktlen = wacom_compute_pktlen(hdev);
2313 	if (features->pktlen > WACOM_PKGLEN_MAX)
2314 		return -EINVAL;
2315 
2316 	if (!devres_open_group(&hdev->dev, wacom, GFP_KERNEL))
2317 		return -ENOMEM;
2318 
2319 	wacom->resources = true;
2320 
2321 	error = wacom_allocate_inputs(wacom);
2322 	if (error)
2323 		goto fail;
2324 
2325 	/*
2326 	 * Bamboo Pad has a generic hid handling for the Pen, and we switch it
2327 	 * into debug mode for the touch part.
2328 	 * We ignore the other interfaces.
2329 	 */
2330 	if (features->type == BAMBOO_PAD) {
2331 		if (features->pktlen == WACOM_PKGLEN_PENABLED) {
2332 			features->type = HID_GENERIC;
2333 		} else if ((features->pktlen != WACOM_PKGLEN_BPAD_TOUCH) &&
2334 			   (features->pktlen != WACOM_PKGLEN_BPAD_TOUCH_USB)) {
2335 			error = -ENODEV;
2336 			goto fail;
2337 		}
2338 	}
2339 
2340 	/* set the default size in case we do not get them from hid */
2341 	wacom_set_default_phy(features);
2342 
2343 	/* Retrieve the physical and logical size for touch devices */
2344 	wacom_retrieve_hid_descriptor(hdev, features);
2345 	wacom_setup_device_quirks(wacom);
2346 
2347 	if (features->device_type == WACOM_DEVICETYPE_NONE &&
2348 	    features->type != WIRELESS) {
2349 		error = features->type == HID_GENERIC ? -ENODEV : 0;
2350 
2351 		dev_warn(&hdev->dev, "Unknown device_type for '%s'. %s.",
2352 			 hdev->name,
2353 			 error ? "Ignoring" : "Assuming pen");
2354 
2355 		if (error)
2356 			goto fail;
2357 
2358 		features->device_type |= WACOM_DEVICETYPE_PEN;
2359 	}
2360 
2361 	wacom_calculate_res(features);
2362 
2363 	wacom_update_name(wacom, wireless ? " (WL)" : "");
2364 
2365 	/* pen only Bamboo neither support touch nor pad */
2366 	if ((features->type == BAMBOO_PEN) &&
2367 	    ((features->device_type & WACOM_DEVICETYPE_TOUCH) ||
2368 	    (features->device_type & WACOM_DEVICETYPE_PAD))) {
2369 		error = -ENODEV;
2370 		goto fail;
2371 	}
2372 
2373 	error = wacom_add_shared_data(hdev);
2374 	if (error)
2375 		goto fail;
2376 
2377 	if (!(features->device_type & WACOM_DEVICETYPE_WL_MONITOR) &&
2378 	     (features->quirks & WACOM_QUIRK_BATTERY)) {
2379 		error = wacom_initialize_battery(wacom);
2380 		if (error)
2381 			goto fail;
2382 	}
2383 
2384 	error = wacom_register_inputs(wacom);
2385 	if (error)
2386 		goto fail;
2387 
2388 	if (wacom->wacom_wac.features.device_type & WACOM_DEVICETYPE_PAD) {
2389 		error = wacom_initialize_leds(wacom);
2390 		if (error)
2391 			goto fail;
2392 
2393 		error = wacom_initialize_remotes(wacom);
2394 		if (error)
2395 			goto fail;
2396 	}
2397 
2398 	if (features->type == HID_GENERIC)
2399 		connect_mask |= HID_CONNECT_DRIVER;
2400 
2401 	/* Regular HID work starts now */
2402 	error = hid_hw_start(hdev, connect_mask);
2403 	if (error) {
2404 		hid_err(hdev, "hw start failed\n");
2405 		goto fail;
2406 	}
2407 
2408 	if (!wireless) {
2409 		/* Note that if query fails it is not a hard failure */
2410 		wacom_query_tablet_data(wacom);
2411 	}
2412 
2413 	/* touch only Bamboo doesn't support pen */
2414 	if ((features->type == BAMBOO_TOUCH) &&
2415 	    (features->device_type & WACOM_DEVICETYPE_PEN)) {
2416 		cancel_delayed_work_sync(&wacom->init_work);
2417 		_wacom_query_tablet_data(wacom);
2418 		error = -ENODEV;
2419 		goto fail_quirks;
2420 	}
2421 
2422 	if (features->device_type & WACOM_DEVICETYPE_WL_MONITOR)
2423 		error = hid_hw_open(hdev);
2424 
2425 	wacom_set_shared_values(wacom_wac);
2426 	devres_close_group(&hdev->dev, wacom);
2427 
2428 	return 0;
2429 
2430 fail_quirks:
2431 	hid_hw_stop(hdev);
2432 fail:
2433 	wacom_release_resources(wacom);
2434 	return error;
2435 }
2436 
wacom_wireless_work(struct work_struct * work)2437 static void wacom_wireless_work(struct work_struct *work)
2438 {
2439 	struct wacom *wacom = container_of(work, struct wacom, wireless_work);
2440 	struct usb_device *usbdev = wacom->usbdev;
2441 	struct wacom_wac *wacom_wac = &wacom->wacom_wac;
2442 	struct hid_device *hdev1, *hdev2;
2443 	struct wacom *wacom1, *wacom2;
2444 	struct wacom_wac *wacom_wac1, *wacom_wac2;
2445 	int error;
2446 
2447 	/*
2448 	 * Regardless if this is a disconnect or a new tablet,
2449 	 * remove any existing input and battery devices.
2450 	 */
2451 
2452 	wacom_destroy_battery(wacom);
2453 
2454 	if (!usbdev)
2455 		return;
2456 
2457 	/* Stylus interface */
2458 	hdev1 = usb_get_intfdata(usbdev->config->interface[1]);
2459 	wacom1 = hid_get_drvdata(hdev1);
2460 	wacom_wac1 = &(wacom1->wacom_wac);
2461 	wacom_release_resources(wacom1);
2462 
2463 	/* Touch interface */
2464 	hdev2 = usb_get_intfdata(usbdev->config->interface[2]);
2465 	wacom2 = hid_get_drvdata(hdev2);
2466 	wacom_wac2 = &(wacom2->wacom_wac);
2467 	wacom_release_resources(wacom2);
2468 
2469 	if (wacom_wac->pid == 0) {
2470 		hid_info(wacom->hdev, "wireless tablet disconnected\n");
2471 	} else {
2472 		const struct hid_device_id *id = wacom_ids;
2473 
2474 		hid_info(wacom->hdev, "wireless tablet connected with PID %x\n",
2475 			 wacom_wac->pid);
2476 
2477 		while (id->bus) {
2478 			if (id->vendor == USB_VENDOR_ID_WACOM &&
2479 			    id->product == wacom_wac->pid)
2480 				break;
2481 			id++;
2482 		}
2483 
2484 		if (!id->bus) {
2485 			hid_info(wacom->hdev, "ignoring unknown PID.\n");
2486 			return;
2487 		}
2488 
2489 		/* Stylus interface */
2490 		wacom_wac1->features =
2491 			*((struct wacom_features *)id->driver_data);
2492 
2493 		wacom_wac1->pid = wacom_wac->pid;
2494 		hid_hw_stop(hdev1);
2495 		error = wacom_parse_and_register(wacom1, true);
2496 		if (error)
2497 			goto fail;
2498 
2499 		/* Touch interface */
2500 		if (wacom_wac1->features.touch_max ||
2501 		    (wacom_wac1->features.type >= INTUOSHT &&
2502 		    wacom_wac1->features.type <= BAMBOO_PT)) {
2503 			wacom_wac2->features =
2504 				*((struct wacom_features *)id->driver_data);
2505 			wacom_wac2->pid = wacom_wac->pid;
2506 			hid_hw_stop(hdev2);
2507 			error = wacom_parse_and_register(wacom2, true);
2508 			if (error)
2509 				goto fail;
2510 		}
2511 
2512 		strlcpy(wacom_wac->name, wacom_wac1->name,
2513 			sizeof(wacom_wac->name));
2514 		error = wacom_initialize_battery(wacom);
2515 		if (error)
2516 			goto fail;
2517 	}
2518 
2519 	return;
2520 
2521 fail:
2522 	wacom_release_resources(wacom1);
2523 	wacom_release_resources(wacom2);
2524 	return;
2525 }
2526 
wacom_remote_destroy_one(struct wacom * wacom,unsigned int index)2527 static void wacom_remote_destroy_one(struct wacom *wacom, unsigned int index)
2528 {
2529 	struct wacom_remote *remote = wacom->remote;
2530 	u32 serial = remote->remotes[index].serial;
2531 	int i;
2532 	unsigned long flags;
2533 
2534 	for (i = 0; i < WACOM_MAX_REMOTES; i++) {
2535 		if (remote->remotes[i].serial == serial) {
2536 
2537 			spin_lock_irqsave(&remote->remote_lock, flags);
2538 			remote->remotes[i].registered = false;
2539 			spin_unlock_irqrestore(&remote->remote_lock, flags);
2540 
2541 			if (remote->remotes[i].battery.battery)
2542 				devres_release_group(&wacom->hdev->dev,
2543 						     &remote->remotes[i].battery.bat_desc);
2544 
2545 			if (remote->remotes[i].group.name)
2546 				devres_release_group(&wacom->hdev->dev,
2547 						     &remote->remotes[i]);
2548 
2549 			remote->remotes[i].serial = 0;
2550 			remote->remotes[i].group.name = NULL;
2551 			remote->remotes[i].battery.battery = NULL;
2552 			wacom->led.groups[i].select = WACOM_STATUS_UNKNOWN;
2553 		}
2554 	}
2555 }
2556 
wacom_remote_create_one(struct wacom * wacom,u32 serial,unsigned int index)2557 static int wacom_remote_create_one(struct wacom *wacom, u32 serial,
2558 				   unsigned int index)
2559 {
2560 	struct wacom_remote *remote = wacom->remote;
2561 	struct device *dev = &wacom->hdev->dev;
2562 	int error, k;
2563 
2564 	/* A remote can pair more than once with an EKR,
2565 	 * check to make sure this serial isn't already paired.
2566 	 */
2567 	for (k = 0; k < WACOM_MAX_REMOTES; k++) {
2568 		if (remote->remotes[k].serial == serial)
2569 			break;
2570 	}
2571 
2572 	if (k < WACOM_MAX_REMOTES) {
2573 		remote->remotes[index].serial = serial;
2574 		return 0;
2575 	}
2576 
2577 	if (!devres_open_group(dev, &remote->remotes[index], GFP_KERNEL))
2578 		return -ENOMEM;
2579 
2580 	error = wacom_remote_create_attr_group(wacom, serial, index);
2581 	if (error)
2582 		goto fail;
2583 
2584 	remote->remotes[index].input = wacom_allocate_input(wacom);
2585 	if (!remote->remotes[index].input) {
2586 		error = -ENOMEM;
2587 		goto fail;
2588 	}
2589 	remote->remotes[index].input->uniq = remote->remotes[index].group.name;
2590 	remote->remotes[index].input->name = wacom->wacom_wac.pad_name;
2591 
2592 	if (!remote->remotes[index].input->name) {
2593 		error = -EINVAL;
2594 		goto fail;
2595 	}
2596 
2597 	error = wacom_setup_pad_input_capabilities(remote->remotes[index].input,
2598 						   &wacom->wacom_wac);
2599 	if (error)
2600 		goto fail;
2601 
2602 	remote->remotes[index].serial = serial;
2603 
2604 	error = input_register_device(remote->remotes[index].input);
2605 	if (error)
2606 		goto fail;
2607 
2608 	error = wacom_led_groups_alloc_and_register_one(
2609 					&remote->remotes[index].input->dev,
2610 					wacom, index, 3, true);
2611 	if (error)
2612 		goto fail;
2613 
2614 	remote->remotes[index].registered = true;
2615 
2616 	devres_close_group(dev, &remote->remotes[index]);
2617 	return 0;
2618 
2619 fail:
2620 	devres_release_group(dev, &remote->remotes[index]);
2621 	remote->remotes[index].serial = 0;
2622 	return error;
2623 }
2624 
wacom_remote_attach_battery(struct wacom * wacom,int index)2625 static int wacom_remote_attach_battery(struct wacom *wacom, int index)
2626 {
2627 	struct wacom_remote *remote = wacom->remote;
2628 	int error;
2629 
2630 	if (!remote->remotes[index].registered)
2631 		return 0;
2632 
2633 	if (remote->remotes[index].battery.battery)
2634 		return 0;
2635 
2636 	if (wacom->led.groups[index].select == WACOM_STATUS_UNKNOWN)
2637 		return 0;
2638 
2639 	error = __wacom_initialize_battery(wacom,
2640 					&wacom->remote->remotes[index].battery);
2641 	if (error)
2642 		return error;
2643 
2644 	return 0;
2645 }
2646 
wacom_remote_work(struct work_struct * work)2647 static void wacom_remote_work(struct work_struct *work)
2648 {
2649 	struct wacom *wacom = container_of(work, struct wacom, remote_work);
2650 	struct wacom_remote *remote = wacom->remote;
2651 	struct wacom_remote_data data;
2652 	unsigned long flags;
2653 	unsigned int count;
2654 	u32 serial;
2655 	int i;
2656 
2657 	spin_lock_irqsave(&remote->remote_lock, flags);
2658 
2659 	count = kfifo_out(&remote->remote_fifo, &data, sizeof(data));
2660 
2661 	if (count != sizeof(data)) {
2662 		hid_err(wacom->hdev,
2663 			"workitem triggered without status available\n");
2664 		spin_unlock_irqrestore(&remote->remote_lock, flags);
2665 		return;
2666 	}
2667 
2668 	if (!kfifo_is_empty(&remote->remote_fifo))
2669 		wacom_schedule_work(&wacom->wacom_wac, WACOM_WORKER_REMOTE);
2670 
2671 	spin_unlock_irqrestore(&remote->remote_lock, flags);
2672 
2673 	for (i = 0; i < WACOM_MAX_REMOTES; i++) {
2674 		serial = data.remote[i].serial;
2675 		if (data.remote[i].connected) {
2676 
2677 			if (remote->remotes[i].serial == serial) {
2678 				wacom_remote_attach_battery(wacom, i);
2679 				continue;
2680 			}
2681 
2682 			if (remote->remotes[i].serial)
2683 				wacom_remote_destroy_one(wacom, i);
2684 
2685 			wacom_remote_create_one(wacom, serial, i);
2686 
2687 		} else if (remote->remotes[i].serial) {
2688 			wacom_remote_destroy_one(wacom, i);
2689 		}
2690 	}
2691 }
2692 
wacom_mode_change_work(struct work_struct * work)2693 static void wacom_mode_change_work(struct work_struct *work)
2694 {
2695 	struct wacom *wacom = container_of(work, struct wacom, mode_change_work);
2696 	struct wacom_shared *shared = wacom->wacom_wac.shared;
2697 	struct wacom *wacom1 = NULL;
2698 	struct wacom *wacom2 = NULL;
2699 	bool is_direct = wacom->wacom_wac.is_direct_mode;
2700 	int error = 0;
2701 
2702 	if (shared->pen) {
2703 		wacom1 = hid_get_drvdata(shared->pen);
2704 		wacom_release_resources(wacom1);
2705 		hid_hw_stop(wacom1->hdev);
2706 		wacom1->wacom_wac.has_mode_change = true;
2707 		wacom1->wacom_wac.is_direct_mode = is_direct;
2708 	}
2709 
2710 	if (shared->touch) {
2711 		wacom2 = hid_get_drvdata(shared->touch);
2712 		wacom_release_resources(wacom2);
2713 		hid_hw_stop(wacom2->hdev);
2714 		wacom2->wacom_wac.has_mode_change = true;
2715 		wacom2->wacom_wac.is_direct_mode = is_direct;
2716 	}
2717 
2718 	if (wacom1) {
2719 		error = wacom_parse_and_register(wacom1, false);
2720 		if (error)
2721 			return;
2722 	}
2723 
2724 	if (wacom2) {
2725 		error = wacom_parse_and_register(wacom2, false);
2726 		if (error)
2727 			return;
2728 	}
2729 
2730 	return;
2731 }
2732 
wacom_probe(struct hid_device * hdev,const struct hid_device_id * id)2733 static int wacom_probe(struct hid_device *hdev,
2734 		const struct hid_device_id *id)
2735 {
2736 	struct wacom *wacom;
2737 	struct wacom_wac *wacom_wac;
2738 	struct wacom_features *features;
2739 	int error;
2740 
2741 	if (!id->driver_data)
2742 		return -EINVAL;
2743 
2744 	hdev->quirks |= HID_QUIRK_NO_INIT_REPORTS;
2745 
2746 	/* hid-core sets this quirk for the boot interface */
2747 	hdev->quirks &= ~HID_QUIRK_NOGET;
2748 
2749 	wacom = devm_kzalloc(&hdev->dev, sizeof(struct wacom), GFP_KERNEL);
2750 	if (!wacom)
2751 		return -ENOMEM;
2752 
2753 	hid_set_drvdata(hdev, wacom);
2754 	wacom->hdev = hdev;
2755 
2756 	wacom_wac = &wacom->wacom_wac;
2757 	wacom_wac->features = *((struct wacom_features *)id->driver_data);
2758 	features = &wacom_wac->features;
2759 
2760 	if (features->check_for_hid_type && features->hid_type != hdev->type)
2761 		return -ENODEV;
2762 
2763 	error = wacom_devm_kfifo_alloc(wacom);
2764 	if (error)
2765 		return error;
2766 
2767 	wacom_wac->hid_data.inputmode = -1;
2768 	wacom_wac->mode_report = -1;
2769 
2770 	if (hid_is_usb(hdev)) {
2771 		struct usb_interface *intf = to_usb_interface(hdev->dev.parent);
2772 		struct usb_device *dev = interface_to_usbdev(intf);
2773 
2774 		wacom->usbdev = dev;
2775 		wacom->intf = intf;
2776 	}
2777 
2778 	mutex_init(&wacom->lock);
2779 	INIT_DELAYED_WORK(&wacom->init_work, wacom_init_work);
2780 	INIT_WORK(&wacom->wireless_work, wacom_wireless_work);
2781 	INIT_WORK(&wacom->battery_work, wacom_battery_work);
2782 	INIT_WORK(&wacom->remote_work, wacom_remote_work);
2783 	INIT_WORK(&wacom->mode_change_work, wacom_mode_change_work);
2784 
2785 	/* ask for the report descriptor to be loaded by HID */
2786 	error = hid_parse(hdev);
2787 	if (error) {
2788 		hid_err(hdev, "parse failed\n");
2789 		return error;
2790 	}
2791 
2792 	error = wacom_parse_and_register(wacom, false);
2793 	if (error)
2794 		return error;
2795 
2796 	if (hdev->bus == BUS_BLUETOOTH) {
2797 		error = device_create_file(&hdev->dev, &dev_attr_speed);
2798 		if (error)
2799 			hid_warn(hdev,
2800 				 "can't create sysfs speed attribute err: %d\n",
2801 				 error);
2802 	}
2803 
2804 	wacom_wac->probe_complete = true;
2805 	return 0;
2806 }
2807 
wacom_remove(struct hid_device * hdev)2808 static void wacom_remove(struct hid_device *hdev)
2809 {
2810 	struct wacom *wacom = hid_get_drvdata(hdev);
2811 	struct wacom_wac *wacom_wac = &wacom->wacom_wac;
2812 	struct wacom_features *features = &wacom_wac->features;
2813 
2814 	if (features->device_type & WACOM_DEVICETYPE_WL_MONITOR)
2815 		hid_hw_close(hdev);
2816 
2817 	hid_hw_stop(hdev);
2818 
2819 	cancel_delayed_work_sync(&wacom->init_work);
2820 	cancel_work_sync(&wacom->wireless_work);
2821 	cancel_work_sync(&wacom->battery_work);
2822 	cancel_work_sync(&wacom->remote_work);
2823 	cancel_work_sync(&wacom->mode_change_work);
2824 	if (hdev->bus == BUS_BLUETOOTH)
2825 		device_remove_file(&hdev->dev, &dev_attr_speed);
2826 
2827 	/* make sure we don't trigger the LEDs */
2828 	wacom_led_groups_release(wacom);
2829 
2830 	if (wacom->wacom_wac.features.type != REMOTE)
2831 		wacom_release_resources(wacom);
2832 }
2833 
2834 #ifdef CONFIG_PM
wacom_resume(struct hid_device * hdev)2835 static int wacom_resume(struct hid_device *hdev)
2836 {
2837 	struct wacom *wacom = hid_get_drvdata(hdev);
2838 
2839 	mutex_lock(&wacom->lock);
2840 
2841 	/* switch to wacom mode first */
2842 	_wacom_query_tablet_data(wacom);
2843 	wacom_led_control(wacom);
2844 
2845 	mutex_unlock(&wacom->lock);
2846 
2847 	return 0;
2848 }
2849 
wacom_reset_resume(struct hid_device * hdev)2850 static int wacom_reset_resume(struct hid_device *hdev)
2851 {
2852 	return wacom_resume(hdev);
2853 }
2854 #endif /* CONFIG_PM */
2855 
2856 static struct hid_driver wacom_driver = {
2857 	.name =		"wacom",
2858 	.id_table =	wacom_ids,
2859 	.probe =	wacom_probe,
2860 	.remove =	wacom_remove,
2861 	.report =	wacom_wac_report,
2862 #ifdef CONFIG_PM
2863 	.resume =	wacom_resume,
2864 	.reset_resume =	wacom_reset_resume,
2865 #endif
2866 	.raw_event =	wacom_raw_event,
2867 };
2868 module_hid_driver(wacom_driver);
2869 
2870 MODULE_VERSION(DRIVER_VERSION);
2871 MODULE_AUTHOR(DRIVER_AUTHOR);
2872 MODULE_DESCRIPTION(DRIVER_DESC);
2873 MODULE_LICENSE("GPL");
2874