1 /*
2  * Roccat Kone driver for Linux
3  *
4  * Copyright (c) 2010 Stefan Achatz <erazor_de@users.sourceforge.net>
5  */
6 
7 /*
8  * This program is free software; you can redistribute it and/or modify it
9  * under the terms of the GNU General Public License as published by the Free
10  * Software Foundation; either version 2 of the License, or (at your option)
11  * any later version.
12  */
13 
14 /*
15  * Roccat Kone is a gamer mouse which consists of a mouse part and a keyboard
16  * part. The keyboard part enables the mouse to execute stored macros with mixed
17  * key- and button-events.
18  *
19  * TODO implement on-the-fly polling-rate change
20  *      The windows driver has the ability to change the polling rate of the
21  *      device on the press of a mousebutton.
22  *      Is it possible to remove and reinstall the urb in raw-event- or any
23  *      other handler, or to defer this action to be executed somewhere else?
24  *
25  * TODO is it possible to overwrite group for sysfs attributes via udev?
26  */
27 
28 #include <linux/device.h>
29 #include <linux/input.h>
30 #include <linux/hid.h>
31 #include <linux/module.h>
32 #include <linux/slab.h>
33 #include <linux/hid-roccat.h>
34 #include "hid-ids.h"
35 #include "hid-roccat-common.h"
36 #include "hid-roccat-kone.h"
37 
38 static uint profile_numbers[5] = {0, 1, 2, 3, 4};
39 
kone_profile_activated(struct kone_device * kone,uint new_profile)40 static void kone_profile_activated(struct kone_device *kone, uint new_profile)
41 {
42 	kone->actual_profile = new_profile;
43 	kone->actual_dpi = kone->profiles[new_profile - 1].startup_dpi;
44 }
45 
kone_profile_report(struct kone_device * kone,uint new_profile)46 static void kone_profile_report(struct kone_device *kone, uint new_profile)
47 {
48 	struct kone_roccat_report roccat_report;
49 
50 	roccat_report.event = kone_mouse_event_switch_profile;
51 	roccat_report.value = new_profile;
52 	roccat_report.key = 0;
53 	roccat_report_event(kone->chrdev_minor, (uint8_t *)&roccat_report);
54 }
55 
kone_receive(struct usb_device * usb_dev,uint usb_command,void * data,uint size)56 static int kone_receive(struct usb_device *usb_dev, uint usb_command,
57 		void *data, uint size)
58 {
59 	char *buf;
60 	int len;
61 
62 	buf = kmalloc(size, GFP_KERNEL);
63 	if (buf == NULL)
64 		return -ENOMEM;
65 
66 	len = usb_control_msg(usb_dev, usb_rcvctrlpipe(usb_dev, 0),
67 			HID_REQ_GET_REPORT,
68 			USB_TYPE_CLASS | USB_RECIP_INTERFACE | USB_DIR_IN,
69 			usb_command, 0, buf, size, USB_CTRL_SET_TIMEOUT);
70 
71 	memcpy(data, buf, size);
72 	kfree(buf);
73 	return ((len < 0) ? len : ((len != size) ? -EIO : 0));
74 }
75 
kone_send(struct usb_device * usb_dev,uint usb_command,void const * data,uint size)76 static int kone_send(struct usb_device *usb_dev, uint usb_command,
77 		void const *data, uint size)
78 {
79 	char *buf;
80 	int len;
81 
82 	buf = kmemdup(data, size, GFP_KERNEL);
83 	if (buf == NULL)
84 		return -ENOMEM;
85 
86 	len = usb_control_msg(usb_dev, usb_sndctrlpipe(usb_dev, 0),
87 			HID_REQ_SET_REPORT,
88 			USB_TYPE_CLASS | USB_RECIP_INTERFACE | USB_DIR_OUT,
89 			usb_command, 0, buf, size, USB_CTRL_SET_TIMEOUT);
90 
91 	kfree(buf);
92 	return ((len < 0) ? len : ((len != size) ? -EIO : 0));
93 }
94 
95 /* kone_class is used for creating sysfs attributes via roccat char device */
96 static struct class *kone_class;
97 
kone_set_settings_checksum(struct kone_settings * settings)98 static void kone_set_settings_checksum(struct kone_settings *settings)
99 {
100 	uint16_t checksum = 0;
101 	unsigned char *address = (unsigned char *)settings;
102 	int i;
103 
104 	for (i = 0; i < sizeof(struct kone_settings) - 2; ++i, ++address)
105 		checksum += *address;
106 	settings->checksum = cpu_to_le16(checksum);
107 }
108 
109 /*
110  * Checks success after writing data to mouse
111  * On success returns 0
112  * On failure returns errno
113  */
kone_check_write(struct usb_device * usb_dev)114 static int kone_check_write(struct usb_device *usb_dev)
115 {
116 	int retval;
117 	uint8_t data;
118 
119 	do {
120 		/*
121 		 * Mouse needs 50 msecs until it says ok, but there are
122 		 * 30 more msecs needed for next write to work.
123 		 */
124 		msleep(80);
125 
126 		retval = kone_receive(usb_dev,
127 				kone_command_confirm_write, &data, 1);
128 		if (retval)
129 			return retval;
130 
131 		/*
132 		 * value of 3 seems to mean something like
133 		 * "not finished yet, but it looks good"
134 		 * So check again after a moment.
135 		 */
136 	} while (data == 3);
137 
138 	if (data == 1) /* everything alright */
139 		return 0;
140 
141 	/* unknown answer */
142 	dev_err(&usb_dev->dev, "got retval %d when checking write\n", data);
143 	return -EIO;
144 }
145 
146 /*
147  * Reads settings from mouse and stores it in @buf
148  * On success returns 0
149  * On failure returns errno
150  */
kone_get_settings(struct usb_device * usb_dev,struct kone_settings * buf)151 static int kone_get_settings(struct usb_device *usb_dev,
152 		struct kone_settings *buf)
153 {
154 	return kone_receive(usb_dev, kone_command_settings, buf,
155 			sizeof(struct kone_settings));
156 }
157 
158 /*
159  * Writes settings from @buf to mouse
160  * On success returns 0
161  * On failure returns errno
162  */
kone_set_settings(struct usb_device * usb_dev,struct kone_settings const * settings)163 static int kone_set_settings(struct usb_device *usb_dev,
164 		struct kone_settings const *settings)
165 {
166 	int retval;
167 
168 	retval = kone_send(usb_dev, kone_command_settings,
169 			settings, sizeof(struct kone_settings));
170 	if (retval)
171 		return retval;
172 	return kone_check_write(usb_dev);
173 }
174 
175 /*
176  * Reads profile data from mouse and stores it in @buf
177  * @number: profile number to read
178  * On success returns 0
179  * On failure returns errno
180  */
kone_get_profile(struct usb_device * usb_dev,struct kone_profile * buf,int number)181 static int kone_get_profile(struct usb_device *usb_dev,
182 		struct kone_profile *buf, int number)
183 {
184 	int len;
185 
186 	if (number < 1 || number > 5)
187 		return -EINVAL;
188 
189 	len = usb_control_msg(usb_dev, usb_rcvctrlpipe(usb_dev, 0),
190 			USB_REQ_CLEAR_FEATURE,
191 			USB_TYPE_CLASS | USB_RECIP_INTERFACE | USB_DIR_IN,
192 			kone_command_profile, number, buf,
193 			sizeof(struct kone_profile), USB_CTRL_SET_TIMEOUT);
194 
195 	if (len != sizeof(struct kone_profile))
196 		return -EIO;
197 
198 	return 0;
199 }
200 
201 /*
202  * Writes profile data to mouse.
203  * @number: profile number to write
204  * On success returns 0
205  * On failure returns errno
206  */
kone_set_profile(struct usb_device * usb_dev,struct kone_profile const * profile,int number)207 static int kone_set_profile(struct usb_device *usb_dev,
208 		struct kone_profile const *profile, int number)
209 {
210 	int len;
211 
212 	if (number < 1 || number > 5)
213 		return -EINVAL;
214 
215 	len = usb_control_msg(usb_dev, usb_sndctrlpipe(usb_dev, 0),
216 			USB_REQ_SET_CONFIGURATION,
217 			USB_TYPE_CLASS | USB_RECIP_INTERFACE | USB_DIR_OUT,
218 			kone_command_profile, number, (void *)profile,
219 			sizeof(struct kone_profile),
220 			USB_CTRL_SET_TIMEOUT);
221 
222 	if (len != sizeof(struct kone_profile))
223 		return len;
224 
225 	if (kone_check_write(usb_dev))
226 		return -EIO;
227 
228 	return 0;
229 }
230 
231 /*
232  * Reads value of "fast-clip-weight" and stores it in @result
233  * On success returns 0
234  * On failure returns errno
235  */
kone_get_weight(struct usb_device * usb_dev,int * result)236 static int kone_get_weight(struct usb_device *usb_dev, int *result)
237 {
238 	int retval;
239 	uint8_t data;
240 
241 	retval = kone_receive(usb_dev, kone_command_weight, &data, 1);
242 
243 	if (retval)
244 		return retval;
245 
246 	*result = (int)data;
247 	return 0;
248 }
249 
250 /*
251  * Reads firmware_version of mouse and stores it in @result
252  * On success returns 0
253  * On failure returns errno
254  */
kone_get_firmware_version(struct usb_device * usb_dev,int * result)255 static int kone_get_firmware_version(struct usb_device *usb_dev, int *result)
256 {
257 	int retval;
258 	uint16_t data;
259 
260 	retval = kone_receive(usb_dev, kone_command_firmware_version,
261 			&data, 2);
262 	if (retval)
263 		return retval;
264 
265 	*result = le16_to_cpu(data);
266 	return 0;
267 }
268 
kone_sysfs_read_settings(struct file * fp,struct kobject * kobj,struct bin_attribute * attr,char * buf,loff_t off,size_t count)269 static ssize_t kone_sysfs_read_settings(struct file *fp, struct kobject *kobj,
270 		struct bin_attribute *attr, char *buf,
271 		loff_t off, size_t count) {
272 	struct device *dev = kobj_to_dev(kobj)->parent->parent;
273 	struct kone_device *kone = hid_get_drvdata(dev_get_drvdata(dev));
274 
275 	if (off >= sizeof(struct kone_settings))
276 		return 0;
277 
278 	if (off + count > sizeof(struct kone_settings))
279 		count = sizeof(struct kone_settings) - off;
280 
281 	mutex_lock(&kone->kone_lock);
282 	memcpy(buf, ((char const *)&kone->settings) + off, count);
283 	mutex_unlock(&kone->kone_lock);
284 
285 	return count;
286 }
287 
288 /*
289  * Writing settings automatically activates startup_profile.
290  * This function keeps values in kone_device up to date and assumes that in
291  * case of error the old data is still valid
292  */
kone_sysfs_write_settings(struct file * fp,struct kobject * kobj,struct bin_attribute * attr,char * buf,loff_t off,size_t count)293 static ssize_t kone_sysfs_write_settings(struct file *fp, struct kobject *kobj,
294 		struct bin_attribute *attr, char *buf,
295 		loff_t off, size_t count) {
296 	struct device *dev = kobj_to_dev(kobj)->parent->parent;
297 	struct kone_device *kone = hid_get_drvdata(dev_get_drvdata(dev));
298 	struct usb_device *usb_dev = interface_to_usbdev(to_usb_interface(dev));
299 	int retval = 0, difference, old_profile;
300 
301 	/* I need to get my data in one piece */
302 	if (off != 0 || count != sizeof(struct kone_settings))
303 		return -EINVAL;
304 
305 	mutex_lock(&kone->kone_lock);
306 	difference = memcmp(buf, &kone->settings, sizeof(struct kone_settings));
307 	if (difference) {
308 		retval = kone_set_settings(usb_dev,
309 				(struct kone_settings const *)buf);
310 		if (retval) {
311 			mutex_unlock(&kone->kone_lock);
312 			return retval;
313 		}
314 
315 		old_profile = kone->settings.startup_profile;
316 		memcpy(&kone->settings, buf, sizeof(struct kone_settings));
317 
318 		kone_profile_activated(kone, kone->settings.startup_profile);
319 
320 		if (kone->settings.startup_profile != old_profile)
321 			kone_profile_report(kone, kone->settings.startup_profile);
322 	}
323 	mutex_unlock(&kone->kone_lock);
324 
325 	return sizeof(struct kone_settings);
326 }
327 static BIN_ATTR(settings, 0660, kone_sysfs_read_settings,
328 		kone_sysfs_write_settings, sizeof(struct kone_settings));
329 
kone_sysfs_read_profilex(struct file * fp,struct kobject * kobj,struct bin_attribute * attr,char * buf,loff_t off,size_t count)330 static ssize_t kone_sysfs_read_profilex(struct file *fp,
331 		struct kobject *kobj, struct bin_attribute *attr,
332 		char *buf, loff_t off, size_t count) {
333 	struct device *dev = kobj_to_dev(kobj)->parent->parent;
334 	struct kone_device *kone = hid_get_drvdata(dev_get_drvdata(dev));
335 
336 	if (off >= sizeof(struct kone_profile))
337 		return 0;
338 
339 	if (off + count > sizeof(struct kone_profile))
340 		count = sizeof(struct kone_profile) - off;
341 
342 	mutex_lock(&kone->kone_lock);
343 	memcpy(buf, ((char const *)&kone->profiles[*(uint *)(attr->private)]) + off, count);
344 	mutex_unlock(&kone->kone_lock);
345 
346 	return count;
347 }
348 
349 /* Writes data only if different to stored data */
kone_sysfs_write_profilex(struct file * fp,struct kobject * kobj,struct bin_attribute * attr,char * buf,loff_t off,size_t count)350 static ssize_t kone_sysfs_write_profilex(struct file *fp,
351 		struct kobject *kobj, struct bin_attribute *attr,
352 		char *buf, loff_t off, size_t count) {
353 	struct device *dev = kobj_to_dev(kobj)->parent->parent;
354 	struct kone_device *kone = hid_get_drvdata(dev_get_drvdata(dev));
355 	struct usb_device *usb_dev = interface_to_usbdev(to_usb_interface(dev));
356 	struct kone_profile *profile;
357 	int retval = 0, difference;
358 
359 	/* I need to get my data in one piece */
360 	if (off != 0 || count != sizeof(struct kone_profile))
361 		return -EINVAL;
362 
363 	profile = &kone->profiles[*(uint *)(attr->private)];
364 
365 	mutex_lock(&kone->kone_lock);
366 	difference = memcmp(buf, profile, sizeof(struct kone_profile));
367 	if (difference) {
368 		retval = kone_set_profile(usb_dev,
369 				(struct kone_profile const *)buf,
370 				*(uint *)(attr->private) + 1);
371 		if (!retval)
372 			memcpy(profile, buf, sizeof(struct kone_profile));
373 	}
374 	mutex_unlock(&kone->kone_lock);
375 
376 	if (retval)
377 		return retval;
378 
379 	return sizeof(struct kone_profile);
380 }
381 #define PROFILE_ATTR(number)					\
382 static struct bin_attribute bin_attr_profile##number = {	\
383 	.attr = { .name = "profile" #number, .mode = 0660 },	\
384 	.size = sizeof(struct kone_profile),			\
385 	.read = kone_sysfs_read_profilex,			\
386 	.write = kone_sysfs_write_profilex,			\
387 	.private = &profile_numbers[number-1],			\
388 }
389 PROFILE_ATTR(1);
390 PROFILE_ATTR(2);
391 PROFILE_ATTR(3);
392 PROFILE_ATTR(4);
393 PROFILE_ATTR(5);
394 
kone_sysfs_show_actual_profile(struct device * dev,struct device_attribute * attr,char * buf)395 static ssize_t kone_sysfs_show_actual_profile(struct device *dev,
396 		struct device_attribute *attr, char *buf)
397 {
398 	struct kone_device *kone =
399 			hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
400 	return snprintf(buf, PAGE_SIZE, "%d\n", kone->actual_profile);
401 }
402 static DEVICE_ATTR(actual_profile, 0440, kone_sysfs_show_actual_profile, NULL);
403 
kone_sysfs_show_actual_dpi(struct device * dev,struct device_attribute * attr,char * buf)404 static ssize_t kone_sysfs_show_actual_dpi(struct device *dev,
405 		struct device_attribute *attr, char *buf)
406 {
407 	struct kone_device *kone =
408 			hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
409 	return snprintf(buf, PAGE_SIZE, "%d\n", kone->actual_dpi);
410 }
411 static DEVICE_ATTR(actual_dpi, 0440, kone_sysfs_show_actual_dpi, NULL);
412 
413 /* weight is read each time, since we don't get informed when it's changed */
kone_sysfs_show_weight(struct device * dev,struct device_attribute * attr,char * buf)414 static ssize_t kone_sysfs_show_weight(struct device *dev,
415 		struct device_attribute *attr, char *buf)
416 {
417 	struct kone_device *kone;
418 	struct usb_device *usb_dev;
419 	int weight = 0;
420 	int retval;
421 
422 	dev = dev->parent->parent;
423 	kone = hid_get_drvdata(dev_get_drvdata(dev));
424 	usb_dev = interface_to_usbdev(to_usb_interface(dev));
425 
426 	mutex_lock(&kone->kone_lock);
427 	retval = kone_get_weight(usb_dev, &weight);
428 	mutex_unlock(&kone->kone_lock);
429 
430 	if (retval)
431 		return retval;
432 	return snprintf(buf, PAGE_SIZE, "%d\n", weight);
433 }
434 static DEVICE_ATTR(weight, 0440, kone_sysfs_show_weight, NULL);
435 
kone_sysfs_show_firmware_version(struct device * dev,struct device_attribute * attr,char * buf)436 static ssize_t kone_sysfs_show_firmware_version(struct device *dev,
437 		struct device_attribute *attr, char *buf)
438 {
439 	struct kone_device *kone =
440 			hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
441 	return snprintf(buf, PAGE_SIZE, "%d\n", kone->firmware_version);
442 }
443 static DEVICE_ATTR(firmware_version, 0440, kone_sysfs_show_firmware_version,
444 		   NULL);
445 
kone_sysfs_show_tcu(struct device * dev,struct device_attribute * attr,char * buf)446 static ssize_t kone_sysfs_show_tcu(struct device *dev,
447 		struct device_attribute *attr, char *buf)
448 {
449 	struct kone_device *kone =
450 			hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
451 	return snprintf(buf, PAGE_SIZE, "%d\n", kone->settings.tcu);
452 }
453 
kone_tcu_command(struct usb_device * usb_dev,int number)454 static int kone_tcu_command(struct usb_device *usb_dev, int number)
455 {
456 	unsigned char value;
457 
458 	value = number;
459 	return kone_send(usb_dev, kone_command_calibrate, &value, 1);
460 }
461 
462 /*
463  * Calibrating the tcu is the only action that changes settings data inside the
464  * mouse, so this data needs to be reread
465  */
kone_sysfs_set_tcu(struct device * dev,struct device_attribute * attr,char const * buf,size_t size)466 static ssize_t kone_sysfs_set_tcu(struct device *dev,
467 		struct device_attribute *attr, char const *buf, size_t size)
468 {
469 	struct kone_device *kone;
470 	struct usb_device *usb_dev;
471 	int retval;
472 	unsigned long state;
473 
474 	dev = dev->parent->parent;
475 	kone = hid_get_drvdata(dev_get_drvdata(dev));
476 	usb_dev = interface_to_usbdev(to_usb_interface(dev));
477 
478 	retval = kstrtoul(buf, 10, &state);
479 	if (retval)
480 		return retval;
481 
482 	if (state != 0 && state != 1)
483 		return -EINVAL;
484 
485 	mutex_lock(&kone->kone_lock);
486 
487 	if (state == 1) { /* state activate */
488 		retval = kone_tcu_command(usb_dev, 1);
489 		if (retval)
490 			goto exit_unlock;
491 		retval = kone_tcu_command(usb_dev, 2);
492 		if (retval)
493 			goto exit_unlock;
494 		ssleep(5); /* tcu needs this time for calibration */
495 		retval = kone_tcu_command(usb_dev, 3);
496 		if (retval)
497 			goto exit_unlock;
498 		retval = kone_tcu_command(usb_dev, 0);
499 		if (retval)
500 			goto exit_unlock;
501 		retval = kone_tcu_command(usb_dev, 4);
502 		if (retval)
503 			goto exit_unlock;
504 		/*
505 		 * Kone needs this time to settle things.
506 		 * Reading settings too early will result in invalid data.
507 		 * Roccat's driver waits 1 sec, maybe this time could be
508 		 * shortened.
509 		 */
510 		ssleep(1);
511 	}
512 
513 	/* calibration changes values in settings, so reread */
514 	retval = kone_get_settings(usb_dev, &kone->settings);
515 	if (retval)
516 		goto exit_no_settings;
517 
518 	/* only write settings back if activation state is different */
519 	if (kone->settings.tcu != state) {
520 		kone->settings.tcu = state;
521 		kone_set_settings_checksum(&kone->settings);
522 
523 		retval = kone_set_settings(usb_dev, &kone->settings);
524 		if (retval) {
525 			dev_err(&usb_dev->dev, "couldn't set tcu state\n");
526 			/*
527 			 * try to reread valid settings into buffer overwriting
528 			 * first error code
529 			 */
530 			retval = kone_get_settings(usb_dev, &kone->settings);
531 			if (retval)
532 				goto exit_no_settings;
533 			goto exit_unlock;
534 		}
535 		/* calibration resets profile */
536 		kone_profile_activated(kone, kone->settings.startup_profile);
537 	}
538 
539 	retval = size;
540 exit_no_settings:
541 	dev_err(&usb_dev->dev, "couldn't read settings\n");
542 exit_unlock:
543 	mutex_unlock(&kone->kone_lock);
544 	return retval;
545 }
546 static DEVICE_ATTR(tcu, 0660, kone_sysfs_show_tcu, kone_sysfs_set_tcu);
547 
kone_sysfs_show_startup_profile(struct device * dev,struct device_attribute * attr,char * buf)548 static ssize_t kone_sysfs_show_startup_profile(struct device *dev,
549 		struct device_attribute *attr, char *buf)
550 {
551 	struct kone_device *kone =
552 			hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
553 	return snprintf(buf, PAGE_SIZE, "%d\n", kone->settings.startup_profile);
554 }
555 
kone_sysfs_set_startup_profile(struct device * dev,struct device_attribute * attr,char const * buf,size_t size)556 static ssize_t kone_sysfs_set_startup_profile(struct device *dev,
557 		struct device_attribute *attr, char const *buf, size_t size)
558 {
559 	struct kone_device *kone;
560 	struct usb_device *usb_dev;
561 	int retval;
562 	unsigned long new_startup_profile;
563 
564 	dev = dev->parent->parent;
565 	kone = hid_get_drvdata(dev_get_drvdata(dev));
566 	usb_dev = interface_to_usbdev(to_usb_interface(dev));
567 
568 	retval = kstrtoul(buf, 10, &new_startup_profile);
569 	if (retval)
570 		return retval;
571 
572 	if (new_startup_profile  < 1 || new_startup_profile > 5)
573 		return -EINVAL;
574 
575 	mutex_lock(&kone->kone_lock);
576 
577 	kone->settings.startup_profile = new_startup_profile;
578 	kone_set_settings_checksum(&kone->settings);
579 
580 	retval = kone_set_settings(usb_dev, &kone->settings);
581 	if (retval) {
582 		mutex_unlock(&kone->kone_lock);
583 		return retval;
584 	}
585 
586 	/* changing the startup profile immediately activates this profile */
587 	kone_profile_activated(kone, new_startup_profile);
588 	kone_profile_report(kone, new_startup_profile);
589 
590 	mutex_unlock(&kone->kone_lock);
591 	return size;
592 }
593 static DEVICE_ATTR(startup_profile, 0660, kone_sysfs_show_startup_profile,
594 		   kone_sysfs_set_startup_profile);
595 
596 static struct attribute *kone_attrs[] = {
597 	/*
598 	 * Read actual dpi settings.
599 	 * Returns raw value for further processing. Refer to enum
600 	 * kone_polling_rates to get real value.
601 	 */
602 	&dev_attr_actual_dpi.attr,
603 	&dev_attr_actual_profile.attr,
604 
605 	/*
606 	 * The mouse can be equipped with one of four supplied weights from 5
607 	 * to 20 grams which are recognized and its value can be read out.
608 	 * This returns the raw value reported by the mouse for easy evaluation
609 	 * by software. Refer to enum kone_weights to get corresponding real
610 	 * weight.
611 	 */
612 	&dev_attr_weight.attr,
613 
614 	/*
615 	 * Prints firmware version stored in mouse as integer.
616 	 * The raw value reported by the mouse is returned for easy evaluation,
617 	 * to get the real version number the decimal point has to be shifted 2
618 	 * positions to the left. E.g. a value of 138 means 1.38.
619 	 */
620 	&dev_attr_firmware_version.attr,
621 
622 	/*
623 	 * Prints state of Tracking Control Unit as number where 0 = off and
624 	 * 1 = on. Writing 0 deactivates tcu and writing 1 calibrates and
625 	 * activates the tcu
626 	 */
627 	&dev_attr_tcu.attr,
628 
629 	/* Prints and takes the number of the profile the mouse starts with */
630 	&dev_attr_startup_profile.attr,
631 	NULL,
632 };
633 
634 static struct bin_attribute *kone_bin_attributes[] = {
635 	&bin_attr_settings,
636 	&bin_attr_profile1,
637 	&bin_attr_profile2,
638 	&bin_attr_profile3,
639 	&bin_attr_profile4,
640 	&bin_attr_profile5,
641 	NULL,
642 };
643 
644 static const struct attribute_group kone_group = {
645 	.attrs = kone_attrs,
646 	.bin_attrs = kone_bin_attributes,
647 };
648 
649 static const struct attribute_group *kone_groups[] = {
650 	&kone_group,
651 	NULL,
652 };
653 
kone_init_kone_device_struct(struct usb_device * usb_dev,struct kone_device * kone)654 static int kone_init_kone_device_struct(struct usb_device *usb_dev,
655 		struct kone_device *kone)
656 {
657 	uint i;
658 	int retval;
659 
660 	mutex_init(&kone->kone_lock);
661 
662 	for (i = 0; i < 5; ++i) {
663 		retval = kone_get_profile(usb_dev, &kone->profiles[i], i + 1);
664 		if (retval)
665 			return retval;
666 	}
667 
668 	retval = kone_get_settings(usb_dev, &kone->settings);
669 	if (retval)
670 		return retval;
671 
672 	retval = kone_get_firmware_version(usb_dev, &kone->firmware_version);
673 	if (retval)
674 		return retval;
675 
676 	kone_profile_activated(kone, kone->settings.startup_profile);
677 
678 	return 0;
679 }
680 
681 /*
682  * Since IGNORE_MOUSE quirk moved to hid-apple, there is no way to bind only to
683  * mousepart if usb_hid is compiled into the kernel and kone is compiled as
684  * module.
685  * Secial behaviour is bound only to mousepart since only mouseevents contain
686  * additional notifications.
687  */
kone_init_specials(struct hid_device * hdev)688 static int kone_init_specials(struct hid_device *hdev)
689 {
690 	struct usb_interface *intf = to_usb_interface(hdev->dev.parent);
691 	struct usb_device *usb_dev = interface_to_usbdev(intf);
692 	struct kone_device *kone;
693 	int retval;
694 
695 	if (intf->cur_altsetting->desc.bInterfaceProtocol
696 			== USB_INTERFACE_PROTOCOL_MOUSE) {
697 
698 		kone = kzalloc(sizeof(*kone), GFP_KERNEL);
699 		if (!kone)
700 			return -ENOMEM;
701 		hid_set_drvdata(hdev, kone);
702 
703 		retval = kone_init_kone_device_struct(usb_dev, kone);
704 		if (retval) {
705 			hid_err(hdev, "couldn't init struct kone_device\n");
706 			goto exit_free;
707 		}
708 
709 		retval = roccat_connect(kone_class, hdev,
710 				sizeof(struct kone_roccat_report));
711 		if (retval < 0) {
712 			hid_err(hdev, "couldn't init char dev\n");
713 			/* be tolerant about not getting chrdev */
714 		} else {
715 			kone->roccat_claimed = 1;
716 			kone->chrdev_minor = retval;
717 		}
718 	} else {
719 		hid_set_drvdata(hdev, NULL);
720 	}
721 
722 	return 0;
723 exit_free:
724 	kfree(kone);
725 	return retval;
726 }
727 
kone_remove_specials(struct hid_device * hdev)728 static void kone_remove_specials(struct hid_device *hdev)
729 {
730 	struct usb_interface *intf = to_usb_interface(hdev->dev.parent);
731 	struct kone_device *kone;
732 
733 	if (intf->cur_altsetting->desc.bInterfaceProtocol
734 			== USB_INTERFACE_PROTOCOL_MOUSE) {
735 		kone = hid_get_drvdata(hdev);
736 		if (kone->roccat_claimed)
737 			roccat_disconnect(kone->chrdev_minor);
738 		kfree(hid_get_drvdata(hdev));
739 	}
740 }
741 
kone_probe(struct hid_device * hdev,const struct hid_device_id * id)742 static int kone_probe(struct hid_device *hdev, const struct hid_device_id *id)
743 {
744 	int retval;
745 
746 	retval = hid_parse(hdev);
747 	if (retval) {
748 		hid_err(hdev, "parse failed\n");
749 		goto exit;
750 	}
751 
752 	retval = hid_hw_start(hdev, HID_CONNECT_DEFAULT);
753 	if (retval) {
754 		hid_err(hdev, "hw start failed\n");
755 		goto exit;
756 	}
757 
758 	retval = kone_init_specials(hdev);
759 	if (retval) {
760 		hid_err(hdev, "couldn't install mouse\n");
761 		goto exit_stop;
762 	}
763 
764 	return 0;
765 
766 exit_stop:
767 	hid_hw_stop(hdev);
768 exit:
769 	return retval;
770 }
771 
kone_remove(struct hid_device * hdev)772 static void kone_remove(struct hid_device *hdev)
773 {
774 	kone_remove_specials(hdev);
775 	hid_hw_stop(hdev);
776 }
777 
778 /* handle special events and keep actual profile and dpi values up to date */
kone_keep_values_up_to_date(struct kone_device * kone,struct kone_mouse_event const * event)779 static void kone_keep_values_up_to_date(struct kone_device *kone,
780 		struct kone_mouse_event const *event)
781 {
782 	switch (event->event) {
783 	case kone_mouse_event_switch_profile:
784 		kone->actual_dpi = kone->profiles[event->value - 1].
785 				startup_dpi;
786 	case kone_mouse_event_osd_profile:
787 		kone->actual_profile = event->value;
788 		break;
789 	case kone_mouse_event_switch_dpi:
790 	case kone_mouse_event_osd_dpi:
791 		kone->actual_dpi = event->value;
792 		break;
793 	}
794 }
795 
kone_report_to_chrdev(struct kone_device const * kone,struct kone_mouse_event const * event)796 static void kone_report_to_chrdev(struct kone_device const *kone,
797 		struct kone_mouse_event const *event)
798 {
799 	struct kone_roccat_report roccat_report;
800 
801 	switch (event->event) {
802 	case kone_mouse_event_switch_profile:
803 	case kone_mouse_event_switch_dpi:
804 	case kone_mouse_event_osd_profile:
805 	case kone_mouse_event_osd_dpi:
806 		roccat_report.event = event->event;
807 		roccat_report.value = event->value;
808 		roccat_report.key = 0;
809 		roccat_report_event(kone->chrdev_minor,
810 				(uint8_t *)&roccat_report);
811 		break;
812 	case kone_mouse_event_call_overlong_macro:
813 	case kone_mouse_event_multimedia:
814 		if (event->value == kone_keystroke_action_press) {
815 			roccat_report.event = event->event;
816 			roccat_report.value = kone->actual_profile;
817 			roccat_report.key = event->macro_key;
818 			roccat_report_event(kone->chrdev_minor,
819 					(uint8_t *)&roccat_report);
820 		}
821 		break;
822 	}
823 
824 }
825 
826 /*
827  * Is called for keyboard- and mousepart.
828  * Only mousepart gets informations about special events in its extended event
829  * structure.
830  */
kone_raw_event(struct hid_device * hdev,struct hid_report * report,u8 * data,int size)831 static int kone_raw_event(struct hid_device *hdev, struct hid_report *report,
832 		u8 *data, int size)
833 {
834 	struct kone_device *kone = hid_get_drvdata(hdev);
835 	struct kone_mouse_event *event = (struct kone_mouse_event *)data;
836 
837 	/* keyboard events are always processed by default handler */
838 	if (size != sizeof(struct kone_mouse_event))
839 		return 0;
840 
841 	if (kone == NULL)
842 		return 0;
843 
844 	/*
845 	 * Firmware 1.38 introduced new behaviour for tilt and special buttons.
846 	 * Pressed button is reported in each movement event.
847 	 * Workaround sends only one event per press.
848 	 */
849 	if (memcmp(&kone->last_mouse_event.tilt, &event->tilt, 5))
850 		memcpy(&kone->last_mouse_event, event,
851 				sizeof(struct kone_mouse_event));
852 	else
853 		memset(&event->tilt, 0, 5);
854 
855 	kone_keep_values_up_to_date(kone, event);
856 
857 	if (kone->roccat_claimed)
858 		kone_report_to_chrdev(kone, event);
859 
860 	return 0; /* always do further processing */
861 }
862 
863 static const struct hid_device_id kone_devices[] = {
864 	{ HID_USB_DEVICE(USB_VENDOR_ID_ROCCAT, USB_DEVICE_ID_ROCCAT_KONE) },
865 	{ }
866 };
867 
868 MODULE_DEVICE_TABLE(hid, kone_devices);
869 
870 static struct hid_driver kone_driver = {
871 		.name = "kone",
872 		.id_table = kone_devices,
873 		.probe = kone_probe,
874 		.remove = kone_remove,
875 		.raw_event = kone_raw_event
876 };
877 
kone_init(void)878 static int __init kone_init(void)
879 {
880 	int retval;
881 
882 	/* class name has to be same as driver name */
883 	kone_class = class_create(THIS_MODULE, "kone");
884 	if (IS_ERR(kone_class))
885 		return PTR_ERR(kone_class);
886 	kone_class->dev_groups = kone_groups;
887 
888 	retval = hid_register_driver(&kone_driver);
889 	if (retval)
890 		class_destroy(kone_class);
891 	return retval;
892 }
893 
kone_exit(void)894 static void __exit kone_exit(void)
895 {
896 	hid_unregister_driver(&kone_driver);
897 	class_destroy(kone_class);
898 }
899 
900 module_init(kone_init);
901 module_exit(kone_exit);
902 
903 MODULE_AUTHOR("Stefan Achatz");
904 MODULE_DESCRIPTION("USB Roccat Kone driver");
905 MODULE_LICENSE("GPL v2");
906