1/*
2 *  HID driver for Lenovo:
3 *  - ThinkPad USB Keyboard with TrackPoint (tpkbd)
4 *  - ThinkPad Compact Bluetooth Keyboard with TrackPoint (cptkbd)
5 *  - ThinkPad Compact USB Keyboard with TrackPoint (cptkbd)
6 *
7 *  Copyright (c) 2012 Bernhard Seibold
8 *  Copyright (c) 2014 Jamie Lentin <jm@lentin.co.uk>
9 */
10
11/*
12 * This program is free software; you can redistribute it and/or modify it
13 * under the terms of the GNU General Public License as published by the Free
14 * Software Foundation; either version 2 of the License, or (at your option)
15 * any later version.
16 */
17
18#include <linux/module.h>
19#include <linux/sysfs.h>
20#include <linux/device.h>
21#include <linux/hid.h>
22#include <linux/input.h>
23#include <linux/leds.h>
24
25#include "hid-ids.h"
26
27struct lenovo_drvdata_tpkbd {
28	int led_state;
29	struct led_classdev led_mute;
30	struct led_classdev led_micmute;
31	int press_to_select;
32	int dragging;
33	int release_to_select;
34	int select_right;
35	int sensitivity;
36	int press_speed;
37};
38
39struct lenovo_drvdata_cptkbd {
40	bool fn_lock;
41	int sensitivity;
42};
43
44#define map_key_clear(c) hid_map_usage_clear(hi, usage, bit, max, EV_KEY, (c))
45
46static int lenovo_input_mapping_tpkbd(struct hid_device *hdev,
47		struct hid_input *hi, struct hid_field *field,
48		struct hid_usage *usage, unsigned long **bit, int *max)
49{
50	if (usage->hid == (HID_UP_BUTTON | 0x0010)) {
51		/* This sub-device contains trackpoint, mark it */
52		hid_set_drvdata(hdev, (void *)1);
53		map_key_clear(KEY_MICMUTE);
54		return 1;
55	}
56	return 0;
57}
58
59static int lenovo_input_mapping_cptkbd(struct hid_device *hdev,
60		struct hid_input *hi, struct hid_field *field,
61		struct hid_usage *usage, unsigned long **bit, int *max)
62{
63	/* HID_UP_LNVENDOR = USB, HID_UP_MSVENDOR = BT */
64	if ((usage->hid & HID_USAGE_PAGE) == HID_UP_MSVENDOR ||
65	    (usage->hid & HID_USAGE_PAGE) == HID_UP_LNVENDOR) {
66		switch (usage->hid & HID_USAGE) {
67		case 0x00f1: /* Fn-F4: Mic mute */
68			map_key_clear(KEY_MICMUTE);
69			return 1;
70		case 0x00f2: /* Fn-F5: Brightness down */
71			map_key_clear(KEY_BRIGHTNESSDOWN);
72			return 1;
73		case 0x00f3: /* Fn-F6: Brightness up */
74			map_key_clear(KEY_BRIGHTNESSUP);
75			return 1;
76		case 0x00f4: /* Fn-F7: External display (projector) */
77			map_key_clear(KEY_SWITCHVIDEOMODE);
78			return 1;
79		case 0x00f5: /* Fn-F8: Wireless */
80			map_key_clear(KEY_WLAN);
81			return 1;
82		case 0x00f6: /* Fn-F9: Control panel */
83			map_key_clear(KEY_CONFIG);
84			return 1;
85		case 0x00f8: /* Fn-F11: View open applications (3 boxes) */
86			map_key_clear(KEY_SCALE);
87			return 1;
88		case 0x00f9: /* Fn-F12: Open My computer (6 boxes) USB-only */
89			/* NB: This mapping is invented in raw_event below */
90			map_key_clear(KEY_FILE);
91			return 1;
92		case 0x00fa: /* Fn-Esc: Fn-lock toggle */
93			map_key_clear(KEY_FN_ESC);
94			return 1;
95		case 0x00fb: /* Middle mouse button (in native mode) */
96			map_key_clear(BTN_MIDDLE);
97			return 1;
98		}
99	}
100
101	/* Compatibility middle/wheel mappings should be ignored */
102	if (usage->hid == HID_GD_WHEEL)
103		return -1;
104	if ((usage->hid & HID_USAGE_PAGE) == HID_UP_BUTTON &&
105			(usage->hid & HID_USAGE) == 0x003)
106		return -1;
107	if ((usage->hid & HID_USAGE_PAGE) == HID_UP_CONSUMER &&
108			(usage->hid & HID_USAGE) == 0x238)
109		return -1;
110
111	/* Map wheel emulation reports: 0xffa1 = USB, 0xff10 = BT */
112	if ((usage->hid & HID_USAGE_PAGE) == 0xff100000 ||
113	    (usage->hid & HID_USAGE_PAGE) == 0xffa10000) {
114		field->flags |= HID_MAIN_ITEM_RELATIVE | HID_MAIN_ITEM_VARIABLE;
115		field->logical_minimum = -127;
116		field->logical_maximum = 127;
117
118		switch (usage->hid & HID_USAGE) {
119		case 0x0000:
120			hid_map_usage(hi, usage, bit, max, EV_REL, 0x06);
121			return 1;
122		case 0x0001:
123			hid_map_usage(hi, usage, bit, max, EV_REL, 0x08);
124			return 1;
125		default:
126			return -1;
127		}
128	}
129
130	return 0;
131}
132
133static int lenovo_input_mapping(struct hid_device *hdev,
134		struct hid_input *hi, struct hid_field *field,
135		struct hid_usage *usage, unsigned long **bit, int *max)
136{
137	switch (hdev->product) {
138	case USB_DEVICE_ID_LENOVO_TPKBD:
139		return lenovo_input_mapping_tpkbd(hdev, hi, field,
140							usage, bit, max);
141	case USB_DEVICE_ID_LENOVO_CUSBKBD:
142	case USB_DEVICE_ID_LENOVO_CBTKBD:
143		return lenovo_input_mapping_cptkbd(hdev, hi, field,
144							usage, bit, max);
145	default:
146		return 0;
147	}
148}
149
150#undef map_key_clear
151
152/* Send a config command to the keyboard */
153static int lenovo_send_cmd_cptkbd(struct hid_device *hdev,
154			unsigned char byte2, unsigned char byte3)
155{
156	int ret;
157	unsigned char buf[] = {0x18, byte2, byte3};
158
159	switch (hdev->product) {
160	case USB_DEVICE_ID_LENOVO_CUSBKBD:
161		ret = hid_hw_raw_request(hdev, 0x13, buf, sizeof(buf),
162					HID_FEATURE_REPORT, HID_REQ_SET_REPORT);
163		break;
164	case USB_DEVICE_ID_LENOVO_CBTKBD:
165		ret = hid_hw_output_report(hdev, buf, sizeof(buf));
166		break;
167	default:
168		ret = -EINVAL;
169		break;
170	}
171
172	return ret < 0 ? ret : 0; /* BT returns 0, USB returns sizeof(buf) */
173}
174
175static void lenovo_features_set_cptkbd(struct hid_device *hdev)
176{
177	int ret;
178	struct lenovo_drvdata_cptkbd *cptkbd_data = hid_get_drvdata(hdev);
179
180	ret = lenovo_send_cmd_cptkbd(hdev, 0x05, cptkbd_data->fn_lock);
181	ret = lenovo_send_cmd_cptkbd(hdev, 0x02, cptkbd_data->sensitivity);
182	if (ret)
183		hid_err(hdev, "Fn-lock setting failed: %d\n", ret);
184}
185
186static ssize_t attr_fn_lock_show_cptkbd(struct device *dev,
187		struct device_attribute *attr,
188		char *buf)
189{
190	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
191	struct lenovo_drvdata_cptkbd *cptkbd_data = hid_get_drvdata(hdev);
192
193	return snprintf(buf, PAGE_SIZE, "%u\n", cptkbd_data->fn_lock);
194}
195
196static ssize_t attr_fn_lock_store_cptkbd(struct device *dev,
197		struct device_attribute *attr,
198		const char *buf,
199		size_t count)
200{
201	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
202	struct lenovo_drvdata_cptkbd *cptkbd_data = hid_get_drvdata(hdev);
203	int value;
204
205	if (kstrtoint(buf, 10, &value))
206		return -EINVAL;
207	if (value < 0 || value > 1)
208		return -EINVAL;
209
210	cptkbd_data->fn_lock = !!value;
211	lenovo_features_set_cptkbd(hdev);
212
213	return count;
214}
215
216static ssize_t attr_sensitivity_show_cptkbd(struct device *dev,
217		struct device_attribute *attr,
218		char *buf)
219{
220	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
221	struct lenovo_drvdata_cptkbd *cptkbd_data = hid_get_drvdata(hdev);
222
223	return snprintf(buf, PAGE_SIZE, "%u\n",
224		cptkbd_data->sensitivity);
225}
226
227static ssize_t attr_sensitivity_store_cptkbd(struct device *dev,
228		struct device_attribute *attr,
229		const char *buf,
230		size_t count)
231{
232	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
233	struct lenovo_drvdata_cptkbd *cptkbd_data = hid_get_drvdata(hdev);
234	int value;
235
236	if (kstrtoint(buf, 10, &value) || value < 1 || value > 255)
237		return -EINVAL;
238
239	cptkbd_data->sensitivity = value;
240	lenovo_features_set_cptkbd(hdev);
241
242	return count;
243}
244
245
246static struct device_attribute dev_attr_fn_lock_cptkbd =
247	__ATTR(fn_lock, S_IWUSR | S_IRUGO,
248			attr_fn_lock_show_cptkbd,
249			attr_fn_lock_store_cptkbd);
250
251static struct device_attribute dev_attr_sensitivity_cptkbd =
252	__ATTR(sensitivity, S_IWUSR | S_IRUGO,
253			attr_sensitivity_show_cptkbd,
254			attr_sensitivity_store_cptkbd);
255
256
257static struct attribute *lenovo_attributes_cptkbd[] = {
258	&dev_attr_fn_lock_cptkbd.attr,
259	&dev_attr_sensitivity_cptkbd.attr,
260	NULL
261};
262
263static const struct attribute_group lenovo_attr_group_cptkbd = {
264	.attrs = lenovo_attributes_cptkbd,
265};
266
267static int lenovo_raw_event(struct hid_device *hdev,
268			struct hid_report *report, u8 *data, int size)
269{
270	/*
271	 * Compact USB keyboard's Fn-F12 report holds down many other keys, and
272	 * its own key is outside the usage page range. Remove extra
273	 * keypresses and remap to inside usage page.
274	 */
275	if (unlikely(hdev->product == USB_DEVICE_ID_LENOVO_CUSBKBD
276			&& size == 3
277			&& data[0] == 0x15
278			&& data[1] == 0x94
279			&& data[2] == 0x01)) {
280		data[1] = 0x00;
281		data[2] = 0x01;
282	}
283
284	return 0;
285}
286
287static int lenovo_features_set_tpkbd(struct hid_device *hdev)
288{
289	struct hid_report *report;
290	struct lenovo_drvdata_tpkbd *data_pointer = hid_get_drvdata(hdev);
291
292	report = hdev->report_enum[HID_FEATURE_REPORT].report_id_hash[4];
293
294	report->field[0]->value[0]  = data_pointer->press_to_select   ? 0x01 : 0x02;
295	report->field[0]->value[0] |= data_pointer->dragging          ? 0x04 : 0x08;
296	report->field[0]->value[0] |= data_pointer->release_to_select ? 0x10 : 0x20;
297	report->field[0]->value[0] |= data_pointer->select_right      ? 0x80 : 0x40;
298	report->field[1]->value[0] = 0x03; // unknown setting, imitate windows driver
299	report->field[2]->value[0] = data_pointer->sensitivity;
300	report->field[3]->value[0] = data_pointer->press_speed;
301
302	hid_hw_request(hdev, report, HID_REQ_SET_REPORT);
303	return 0;
304}
305
306static ssize_t attr_press_to_select_show_tpkbd(struct device *dev,
307		struct device_attribute *attr,
308		char *buf)
309{
310	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
311	struct lenovo_drvdata_tpkbd *data_pointer = hid_get_drvdata(hdev);
312
313	return snprintf(buf, PAGE_SIZE, "%u\n", data_pointer->press_to_select);
314}
315
316static ssize_t attr_press_to_select_store_tpkbd(struct device *dev,
317		struct device_attribute *attr,
318		const char *buf,
319		size_t count)
320{
321	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
322	struct lenovo_drvdata_tpkbd *data_pointer = hid_get_drvdata(hdev);
323	int value;
324
325	if (kstrtoint(buf, 10, &value))
326		return -EINVAL;
327	if (value < 0 || value > 1)
328		return -EINVAL;
329
330	data_pointer->press_to_select = value;
331	lenovo_features_set_tpkbd(hdev);
332
333	return count;
334}
335
336static ssize_t attr_dragging_show_tpkbd(struct device *dev,
337		struct device_attribute *attr,
338		char *buf)
339{
340	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
341	struct lenovo_drvdata_tpkbd *data_pointer = hid_get_drvdata(hdev);
342
343	return snprintf(buf, PAGE_SIZE, "%u\n", data_pointer->dragging);
344}
345
346static ssize_t attr_dragging_store_tpkbd(struct device *dev,
347		struct device_attribute *attr,
348		const char *buf,
349		size_t count)
350{
351	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
352	struct lenovo_drvdata_tpkbd *data_pointer = hid_get_drvdata(hdev);
353	int value;
354
355	if (kstrtoint(buf, 10, &value))
356		return -EINVAL;
357	if (value < 0 || value > 1)
358		return -EINVAL;
359
360	data_pointer->dragging = value;
361	lenovo_features_set_tpkbd(hdev);
362
363	return count;
364}
365
366static ssize_t attr_release_to_select_show_tpkbd(struct device *dev,
367		struct device_attribute *attr,
368		char *buf)
369{
370	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
371	struct lenovo_drvdata_tpkbd *data_pointer = hid_get_drvdata(hdev);
372
373	return snprintf(buf, PAGE_SIZE, "%u\n", data_pointer->release_to_select);
374}
375
376static ssize_t attr_release_to_select_store_tpkbd(struct device *dev,
377		struct device_attribute *attr,
378		const char *buf,
379		size_t count)
380{
381	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
382	struct lenovo_drvdata_tpkbd *data_pointer = hid_get_drvdata(hdev);
383	int value;
384
385	if (kstrtoint(buf, 10, &value))
386		return -EINVAL;
387	if (value < 0 || value > 1)
388		return -EINVAL;
389
390	data_pointer->release_to_select = value;
391	lenovo_features_set_tpkbd(hdev);
392
393	return count;
394}
395
396static ssize_t attr_select_right_show_tpkbd(struct device *dev,
397		struct device_attribute *attr,
398		char *buf)
399{
400	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
401	struct lenovo_drvdata_tpkbd *data_pointer = hid_get_drvdata(hdev);
402
403	return snprintf(buf, PAGE_SIZE, "%u\n", data_pointer->select_right);
404}
405
406static ssize_t attr_select_right_store_tpkbd(struct device *dev,
407		struct device_attribute *attr,
408		const char *buf,
409		size_t count)
410{
411	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
412	struct lenovo_drvdata_tpkbd *data_pointer = hid_get_drvdata(hdev);
413	int value;
414
415	if (kstrtoint(buf, 10, &value))
416		return -EINVAL;
417	if (value < 0 || value > 1)
418		return -EINVAL;
419
420	data_pointer->select_right = value;
421	lenovo_features_set_tpkbd(hdev);
422
423	return count;
424}
425
426static ssize_t attr_sensitivity_show_tpkbd(struct device *dev,
427		struct device_attribute *attr,
428		char *buf)
429{
430	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
431	struct lenovo_drvdata_tpkbd *data_pointer = hid_get_drvdata(hdev);
432
433	return snprintf(buf, PAGE_SIZE, "%u\n",
434		data_pointer->sensitivity);
435}
436
437static ssize_t attr_sensitivity_store_tpkbd(struct device *dev,
438		struct device_attribute *attr,
439		const char *buf,
440		size_t count)
441{
442	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
443	struct lenovo_drvdata_tpkbd *data_pointer = hid_get_drvdata(hdev);
444	int value;
445
446	if (kstrtoint(buf, 10, &value) || value < 1 || value > 255)
447		return -EINVAL;
448
449	data_pointer->sensitivity = value;
450	lenovo_features_set_tpkbd(hdev);
451
452	return count;
453}
454
455static ssize_t attr_press_speed_show_tpkbd(struct device *dev,
456		struct device_attribute *attr,
457		char *buf)
458{
459	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
460	struct lenovo_drvdata_tpkbd *data_pointer = hid_get_drvdata(hdev);
461
462	return snprintf(buf, PAGE_SIZE, "%u\n",
463		data_pointer->press_speed);
464}
465
466static ssize_t attr_press_speed_store_tpkbd(struct device *dev,
467		struct device_attribute *attr,
468		const char *buf,
469		size_t count)
470{
471	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
472	struct lenovo_drvdata_tpkbd *data_pointer = hid_get_drvdata(hdev);
473	int value;
474
475	if (kstrtoint(buf, 10, &value) || value < 1 || value > 255)
476		return -EINVAL;
477
478	data_pointer->press_speed = value;
479	lenovo_features_set_tpkbd(hdev);
480
481	return count;
482}
483
484static struct device_attribute dev_attr_press_to_select_tpkbd =
485	__ATTR(press_to_select, S_IWUSR | S_IRUGO,
486			attr_press_to_select_show_tpkbd,
487			attr_press_to_select_store_tpkbd);
488
489static struct device_attribute dev_attr_dragging_tpkbd =
490	__ATTR(dragging, S_IWUSR | S_IRUGO,
491			attr_dragging_show_tpkbd,
492			attr_dragging_store_tpkbd);
493
494static struct device_attribute dev_attr_release_to_select_tpkbd =
495	__ATTR(release_to_select, S_IWUSR | S_IRUGO,
496			attr_release_to_select_show_tpkbd,
497			attr_release_to_select_store_tpkbd);
498
499static struct device_attribute dev_attr_select_right_tpkbd =
500	__ATTR(select_right, S_IWUSR | S_IRUGO,
501			attr_select_right_show_tpkbd,
502			attr_select_right_store_tpkbd);
503
504static struct device_attribute dev_attr_sensitivity_tpkbd =
505	__ATTR(sensitivity, S_IWUSR | S_IRUGO,
506			attr_sensitivity_show_tpkbd,
507			attr_sensitivity_store_tpkbd);
508
509static struct device_attribute dev_attr_press_speed_tpkbd =
510	__ATTR(press_speed, S_IWUSR | S_IRUGO,
511			attr_press_speed_show_tpkbd,
512			attr_press_speed_store_tpkbd);
513
514static struct attribute *lenovo_attributes_tpkbd[] = {
515	&dev_attr_press_to_select_tpkbd.attr,
516	&dev_attr_dragging_tpkbd.attr,
517	&dev_attr_release_to_select_tpkbd.attr,
518	&dev_attr_select_right_tpkbd.attr,
519	&dev_attr_sensitivity_tpkbd.attr,
520	&dev_attr_press_speed_tpkbd.attr,
521	NULL
522};
523
524static const struct attribute_group lenovo_attr_group_tpkbd = {
525	.attrs = lenovo_attributes_tpkbd,
526};
527
528static enum led_brightness lenovo_led_brightness_get_tpkbd(
529			struct led_classdev *led_cdev)
530{
531	struct device *dev = led_cdev->dev->parent;
532	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
533	struct lenovo_drvdata_tpkbd *data_pointer = hid_get_drvdata(hdev);
534	int led_nr = 0;
535
536	if (led_cdev == &data_pointer->led_micmute)
537		led_nr = 1;
538
539	return data_pointer->led_state & (1 << led_nr)
540				? LED_FULL
541				: LED_OFF;
542}
543
544static void lenovo_led_brightness_set_tpkbd(struct led_classdev *led_cdev,
545			enum led_brightness value)
546{
547	struct device *dev = led_cdev->dev->parent;
548	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
549	struct lenovo_drvdata_tpkbd *data_pointer = hid_get_drvdata(hdev);
550	struct hid_report *report;
551	int led_nr = 0;
552
553	if (led_cdev == &data_pointer->led_micmute)
554		led_nr = 1;
555
556	if (value == LED_OFF)
557		data_pointer->led_state &= ~(1 << led_nr);
558	else
559		data_pointer->led_state |= 1 << led_nr;
560
561	report = hdev->report_enum[HID_OUTPUT_REPORT].report_id_hash[3];
562	report->field[0]->value[0] = (data_pointer->led_state >> 0) & 1;
563	report->field[0]->value[1] = (data_pointer->led_state >> 1) & 1;
564	hid_hw_request(hdev, report, HID_REQ_SET_REPORT);
565}
566
567static int lenovo_probe_tpkbd(struct hid_device *hdev)
568{
569	struct device *dev = &hdev->dev;
570	struct lenovo_drvdata_tpkbd *data_pointer;
571	size_t name_sz = strlen(dev_name(dev)) + 16;
572	char *name_mute, *name_micmute;
573	int i;
574	int ret;
575
576	/*
577	 * Only register extra settings against subdevice where input_mapping
578	 * set drvdata to 1, i.e. the trackpoint.
579	 */
580	if (!hid_get_drvdata(hdev))
581		return 0;
582
583	hid_set_drvdata(hdev, NULL);
584
585	/* Validate required reports. */
586	for (i = 0; i < 4; i++) {
587		if (!hid_validate_values(hdev, HID_FEATURE_REPORT, 4, i, 1))
588			return -ENODEV;
589	}
590	if (!hid_validate_values(hdev, HID_OUTPUT_REPORT, 3, 0, 2))
591		return -ENODEV;
592
593	ret = sysfs_create_group(&hdev->dev.kobj, &lenovo_attr_group_tpkbd);
594	if (ret)
595		hid_warn(hdev, "Could not create sysfs group: %d\n", ret);
596
597	data_pointer = devm_kzalloc(&hdev->dev,
598				    sizeof(struct lenovo_drvdata_tpkbd),
599				    GFP_KERNEL);
600	if (data_pointer == NULL) {
601		hid_err(hdev, "Could not allocate memory for driver data\n");
602		return -ENOMEM;
603	}
604
605	// set same default values as windows driver
606	data_pointer->sensitivity = 0xa0;
607	data_pointer->press_speed = 0x38;
608
609	name_mute = devm_kzalloc(&hdev->dev, name_sz, GFP_KERNEL);
610	name_micmute = devm_kzalloc(&hdev->dev, name_sz, GFP_KERNEL);
611	if (name_mute == NULL || name_micmute == NULL) {
612		hid_err(hdev, "Could not allocate memory for led data\n");
613		return -ENOMEM;
614	}
615	snprintf(name_mute, name_sz, "%s:amber:mute", dev_name(dev));
616	snprintf(name_micmute, name_sz, "%s:amber:micmute", dev_name(dev));
617
618	hid_set_drvdata(hdev, data_pointer);
619
620	data_pointer->led_mute.name = name_mute;
621	data_pointer->led_mute.brightness_get = lenovo_led_brightness_get_tpkbd;
622	data_pointer->led_mute.brightness_set = lenovo_led_brightness_set_tpkbd;
623	data_pointer->led_mute.dev = dev;
624	led_classdev_register(dev, &data_pointer->led_mute);
625
626	data_pointer->led_micmute.name = name_micmute;
627	data_pointer->led_micmute.brightness_get =
628		lenovo_led_brightness_get_tpkbd;
629	data_pointer->led_micmute.brightness_set =
630		lenovo_led_brightness_set_tpkbd;
631	data_pointer->led_micmute.dev = dev;
632	led_classdev_register(dev, &data_pointer->led_micmute);
633
634	lenovo_features_set_tpkbd(hdev);
635
636	return 0;
637}
638
639static int lenovo_probe_cptkbd(struct hid_device *hdev)
640{
641	int ret;
642	struct lenovo_drvdata_cptkbd *cptkbd_data;
643
644	/* All the custom action happens on the USBMOUSE device for USB */
645	if (hdev->product == USB_DEVICE_ID_LENOVO_CUSBKBD
646			&& hdev->type != HID_TYPE_USBMOUSE) {
647		hid_dbg(hdev, "Ignoring keyboard half of device\n");
648		return 0;
649	}
650
651	cptkbd_data = devm_kzalloc(&hdev->dev,
652					sizeof(*cptkbd_data),
653					GFP_KERNEL);
654	if (cptkbd_data == NULL) {
655		hid_err(hdev, "can't alloc keyboard descriptor\n");
656		return -ENOMEM;
657	}
658	hid_set_drvdata(hdev, cptkbd_data);
659
660	/*
661	 * Tell the keyboard a driver understands it, and turn F7, F9, F11 into
662	 * regular keys
663	 */
664	ret = lenovo_send_cmd_cptkbd(hdev, 0x01, 0x03);
665	if (ret)
666		hid_warn(hdev, "Failed to switch F7/9/11 mode: %d\n", ret);
667
668	/* Switch middle button to native mode */
669	ret = lenovo_send_cmd_cptkbd(hdev, 0x09, 0x01);
670	if (ret)
671		hid_warn(hdev, "Failed to switch middle button: %d\n", ret);
672
673	/* Set keyboard settings to known state */
674	cptkbd_data->fn_lock = true;
675	cptkbd_data->sensitivity = 0x05;
676	lenovo_features_set_cptkbd(hdev);
677
678	ret = sysfs_create_group(&hdev->dev.kobj, &lenovo_attr_group_cptkbd);
679	if (ret)
680		hid_warn(hdev, "Could not create sysfs group: %d\n", ret);
681
682	return 0;
683}
684
685static int lenovo_probe(struct hid_device *hdev,
686		const struct hid_device_id *id)
687{
688	int ret;
689
690	ret = hid_parse(hdev);
691	if (ret) {
692		hid_err(hdev, "hid_parse failed\n");
693		goto err;
694	}
695
696	ret = hid_hw_start(hdev, HID_CONNECT_DEFAULT);
697	if (ret) {
698		hid_err(hdev, "hid_hw_start failed\n");
699		goto err;
700	}
701
702	switch (hdev->product) {
703	case USB_DEVICE_ID_LENOVO_TPKBD:
704		ret = lenovo_probe_tpkbd(hdev);
705		break;
706	case USB_DEVICE_ID_LENOVO_CUSBKBD:
707	case USB_DEVICE_ID_LENOVO_CBTKBD:
708		ret = lenovo_probe_cptkbd(hdev);
709		break;
710	default:
711		ret = 0;
712		break;
713	}
714	if (ret)
715		goto err_hid;
716
717	return 0;
718err_hid:
719	hid_hw_stop(hdev);
720err:
721	return ret;
722}
723
724static void lenovo_remove_tpkbd(struct hid_device *hdev)
725{
726	struct lenovo_drvdata_tpkbd *data_pointer = hid_get_drvdata(hdev);
727
728	/*
729	 * Only the trackpoint half of the keyboard has drvdata and stuff that
730	 * needs unregistering.
731	 */
732	if (data_pointer == NULL)
733		return;
734
735	sysfs_remove_group(&hdev->dev.kobj,
736			&lenovo_attr_group_tpkbd);
737
738	led_classdev_unregister(&data_pointer->led_micmute);
739	led_classdev_unregister(&data_pointer->led_mute);
740
741	hid_set_drvdata(hdev, NULL);
742}
743
744static void lenovo_remove_cptkbd(struct hid_device *hdev)
745{
746	sysfs_remove_group(&hdev->dev.kobj,
747			&lenovo_attr_group_cptkbd);
748}
749
750static void lenovo_remove(struct hid_device *hdev)
751{
752	switch (hdev->product) {
753	case USB_DEVICE_ID_LENOVO_TPKBD:
754		lenovo_remove_tpkbd(hdev);
755		break;
756	case USB_DEVICE_ID_LENOVO_CUSBKBD:
757	case USB_DEVICE_ID_LENOVO_CBTKBD:
758		lenovo_remove_cptkbd(hdev);
759		break;
760	}
761
762	hid_hw_stop(hdev);
763}
764
765static const struct hid_device_id lenovo_devices[] = {
766	{ HID_USB_DEVICE(USB_VENDOR_ID_LENOVO, USB_DEVICE_ID_LENOVO_TPKBD) },
767	{ HID_USB_DEVICE(USB_VENDOR_ID_LENOVO, USB_DEVICE_ID_LENOVO_CUSBKBD) },
768	{ HID_BLUETOOTH_DEVICE(USB_VENDOR_ID_LENOVO, USB_DEVICE_ID_LENOVO_CBTKBD) },
769	{ }
770};
771
772MODULE_DEVICE_TABLE(hid, lenovo_devices);
773
774static struct hid_driver lenovo_driver = {
775	.name = "lenovo",
776	.id_table = lenovo_devices,
777	.input_mapping = lenovo_input_mapping,
778	.probe = lenovo_probe,
779	.remove = lenovo_remove,
780	.raw_event = lenovo_raw_event,
781};
782module_hid_driver(lenovo_driver);
783
784MODULE_LICENSE("GPL");
785