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