Merge branch 'for-5.15-fixes' of git://git.kernel.org/pub/scm/linux/kernel/git/tj...
[platform/kernel/linux-starfive.git] / drivers / hid / hid-roccat-pyra.c
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * Roccat Pyra driver for Linux
4  *
5  * Copyright (c) 2010 Stefan Achatz <erazor_de@users.sourceforge.net>
6  */
7
8 /*
9  */
10
11 /*
12  * Roccat Pyra is a mobile gamer mouse which comes in wired and wireless
13  * variant. Wireless variant is not tested.
14  * Userland tools can be found at http://sourceforge.net/projects/roccat
15  */
16
17 #include <linux/device.h>
18 #include <linux/input.h>
19 #include <linux/hid.h>
20 #include <linux/module.h>
21 #include <linux/slab.h>
22 #include <linux/hid-roccat.h>
23 #include "hid-ids.h"
24 #include "hid-roccat-common.h"
25 #include "hid-roccat-pyra.h"
26
27 static uint profile_numbers[5] = {0, 1, 2, 3, 4};
28
29 /* pyra_class is used for creating sysfs attributes via roccat char device */
30 static struct class *pyra_class;
31
32 static void profile_activated(struct pyra_device *pyra,
33                 unsigned int new_profile)
34 {
35         if (new_profile >= ARRAY_SIZE(pyra->profile_settings))
36                 return;
37         pyra->actual_profile = new_profile;
38         pyra->actual_cpi = pyra->profile_settings[pyra->actual_profile].y_cpi;
39 }
40
41 static int pyra_send_control(struct usb_device *usb_dev, int value,
42                 enum pyra_control_requests request)
43 {
44         struct roccat_common2_control control;
45
46         if ((request == PYRA_CONTROL_REQUEST_PROFILE_SETTINGS ||
47                         request == PYRA_CONTROL_REQUEST_PROFILE_BUTTONS) &&
48                         (value < 0 || value > 4))
49                 return -EINVAL;
50
51         control.command = ROCCAT_COMMON_COMMAND_CONTROL;
52         control.value = value;
53         control.request = request;
54
55         return roccat_common2_send(usb_dev, ROCCAT_COMMON_COMMAND_CONTROL,
56                         &control, sizeof(struct roccat_common2_control));
57 }
58
59 static int pyra_get_profile_settings(struct usb_device *usb_dev,
60                 struct pyra_profile_settings *buf, int number)
61 {
62         int retval;
63         retval = pyra_send_control(usb_dev, number,
64                         PYRA_CONTROL_REQUEST_PROFILE_SETTINGS);
65         if (retval)
66                 return retval;
67         return roccat_common2_receive(usb_dev, PYRA_COMMAND_PROFILE_SETTINGS,
68                         buf, PYRA_SIZE_PROFILE_SETTINGS);
69 }
70
71 static int pyra_get_settings(struct usb_device *usb_dev,
72                 struct pyra_settings *buf)
73 {
74         return roccat_common2_receive(usb_dev, PYRA_COMMAND_SETTINGS,
75                         buf, PYRA_SIZE_SETTINGS);
76 }
77
78 static int pyra_set_settings(struct usb_device *usb_dev,
79                 struct pyra_settings const *settings)
80 {
81         return roccat_common2_send_with_status(usb_dev,
82                         PYRA_COMMAND_SETTINGS, settings,
83                         PYRA_SIZE_SETTINGS);
84 }
85
86 static ssize_t pyra_sysfs_read(struct file *fp, struct kobject *kobj,
87                 char *buf, loff_t off, size_t count,
88                 size_t real_size, uint command)
89 {
90         struct device *dev = kobj_to_dev(kobj)->parent->parent;
91         struct pyra_device *pyra = hid_get_drvdata(dev_get_drvdata(dev));
92         struct usb_device *usb_dev = interface_to_usbdev(to_usb_interface(dev));
93         int retval;
94
95         if (off >= real_size)
96                 return 0;
97
98         if (off != 0 || count != real_size)
99                 return -EINVAL;
100
101         mutex_lock(&pyra->pyra_lock);
102         retval = roccat_common2_receive(usb_dev, command, buf, real_size);
103         mutex_unlock(&pyra->pyra_lock);
104
105         if (retval)
106                 return retval;
107
108         return real_size;
109 }
110
111 static ssize_t pyra_sysfs_write(struct file *fp, struct kobject *kobj,
112                 void const *buf, loff_t off, size_t count,
113                 size_t real_size, uint command)
114 {
115         struct device *dev = kobj_to_dev(kobj)->parent->parent;
116         struct pyra_device *pyra = hid_get_drvdata(dev_get_drvdata(dev));
117         struct usb_device *usb_dev = interface_to_usbdev(to_usb_interface(dev));
118         int retval;
119
120         if (off != 0 || count != real_size)
121                 return -EINVAL;
122
123         mutex_lock(&pyra->pyra_lock);
124         retval = roccat_common2_send_with_status(usb_dev, command, (void *)buf, real_size);
125         mutex_unlock(&pyra->pyra_lock);
126
127         if (retval)
128                 return retval;
129
130         return real_size;
131 }
132
133 #define PYRA_SYSFS_W(thingy, THINGY) \
134 static ssize_t pyra_sysfs_write_ ## thingy(struct file *fp, \
135                 struct kobject *kobj, struct bin_attribute *attr, char *buf, \
136                 loff_t off, size_t count) \
137 { \
138         return pyra_sysfs_write(fp, kobj, buf, off, count, \
139                         PYRA_SIZE_ ## THINGY, PYRA_COMMAND_ ## THINGY); \
140 }
141
142 #define PYRA_SYSFS_R(thingy, THINGY) \
143 static ssize_t pyra_sysfs_read_ ## thingy(struct file *fp, \
144                 struct kobject *kobj, struct bin_attribute *attr, char *buf, \
145                 loff_t off, size_t count) \
146 { \
147         return pyra_sysfs_read(fp, kobj, buf, off, count, \
148                         PYRA_SIZE_ ## THINGY, PYRA_COMMAND_ ## THINGY); \
149 }
150
151 #define PYRA_SYSFS_RW(thingy, THINGY) \
152 PYRA_SYSFS_W(thingy, THINGY) \
153 PYRA_SYSFS_R(thingy, THINGY)
154
155 #define PYRA_BIN_ATTRIBUTE_RW(thingy, THINGY) \
156 PYRA_SYSFS_RW(thingy, THINGY); \
157 static struct bin_attribute bin_attr_##thingy = { \
158         .attr = { .name = #thingy, .mode = 0660 }, \
159         .size = PYRA_SIZE_ ## THINGY, \
160         .read = pyra_sysfs_read_ ## thingy, \
161         .write = pyra_sysfs_write_ ## thingy \
162 }
163
164 #define PYRA_BIN_ATTRIBUTE_R(thingy, THINGY) \
165 PYRA_SYSFS_R(thingy, THINGY); \
166 static struct bin_attribute bin_attr_##thingy = { \
167         .attr = { .name = #thingy, .mode = 0440 }, \
168         .size = PYRA_SIZE_ ## THINGY, \
169         .read = pyra_sysfs_read_ ## thingy, \
170 }
171
172 #define PYRA_BIN_ATTRIBUTE_W(thingy, THINGY) \
173 PYRA_SYSFS_W(thingy, THINGY); \
174 static struct bin_attribute bin_attr_##thingy = { \
175         .attr = { .name = #thingy, .mode = 0220 }, \
176         .size = PYRA_SIZE_ ## THINGY, \
177         .write = pyra_sysfs_write_ ## thingy \
178 }
179
180 PYRA_BIN_ATTRIBUTE_W(control, CONTROL);
181 PYRA_BIN_ATTRIBUTE_RW(info, INFO);
182 PYRA_BIN_ATTRIBUTE_RW(profile_settings, PROFILE_SETTINGS);
183 PYRA_BIN_ATTRIBUTE_RW(profile_buttons, PROFILE_BUTTONS);
184
185 static ssize_t pyra_sysfs_read_profilex_settings(struct file *fp,
186                 struct kobject *kobj, struct bin_attribute *attr, char *buf,
187                 loff_t off, size_t count)
188 {
189         struct device *dev = kobj_to_dev(kobj)->parent->parent;
190         struct usb_device *usb_dev = interface_to_usbdev(to_usb_interface(dev));
191         ssize_t retval;
192
193         retval = pyra_send_control(usb_dev, *(uint *)(attr->private),
194                         PYRA_CONTROL_REQUEST_PROFILE_SETTINGS);
195         if (retval)
196                 return retval;
197
198         return pyra_sysfs_read(fp, kobj, buf, off, count,
199                         PYRA_SIZE_PROFILE_SETTINGS,
200                         PYRA_COMMAND_PROFILE_SETTINGS);
201 }
202
203 static ssize_t pyra_sysfs_read_profilex_buttons(struct file *fp,
204                 struct kobject *kobj, struct bin_attribute *attr, char *buf,
205                 loff_t off, size_t count)
206 {
207         struct device *dev = kobj_to_dev(kobj)->parent->parent;
208         struct usb_device *usb_dev = interface_to_usbdev(to_usb_interface(dev));
209         ssize_t retval;
210
211         retval = pyra_send_control(usb_dev, *(uint *)(attr->private),
212                         PYRA_CONTROL_REQUEST_PROFILE_BUTTONS);
213         if (retval)
214                 return retval;
215
216         return pyra_sysfs_read(fp, kobj, buf, off, count,
217                         PYRA_SIZE_PROFILE_BUTTONS,
218                         PYRA_COMMAND_PROFILE_BUTTONS);
219 }
220
221 #define PROFILE_ATTR(number)                                            \
222 static struct bin_attribute bin_attr_profile##number##_settings = {     \
223         .attr = { .name = "profile" #number "_settings", .mode = 0440 },        \
224         .size = PYRA_SIZE_PROFILE_SETTINGS,                             \
225         .read = pyra_sysfs_read_profilex_settings,                      \
226         .private = &profile_numbers[number-1],                          \
227 };                                                                      \
228 static struct bin_attribute bin_attr_profile##number##_buttons = {      \
229         .attr = { .name = "profile" #number "_buttons", .mode = 0440 }, \
230         .size = PYRA_SIZE_PROFILE_BUTTONS,                              \
231         .read = pyra_sysfs_read_profilex_buttons,                       \
232         .private = &profile_numbers[number-1],                          \
233 };
234 PROFILE_ATTR(1);
235 PROFILE_ATTR(2);
236 PROFILE_ATTR(3);
237 PROFILE_ATTR(4);
238 PROFILE_ATTR(5);
239
240 static ssize_t pyra_sysfs_write_settings(struct file *fp,
241                 struct kobject *kobj, struct bin_attribute *attr, char *buf,
242                 loff_t off, size_t count)
243 {
244         struct device *dev = kobj_to_dev(kobj)->parent->parent;
245         struct pyra_device *pyra = hid_get_drvdata(dev_get_drvdata(dev));
246         struct usb_device *usb_dev = interface_to_usbdev(to_usb_interface(dev));
247         int retval = 0;
248         struct pyra_roccat_report roccat_report;
249         struct pyra_settings const *settings;
250
251         if (off != 0 || count != PYRA_SIZE_SETTINGS)
252                 return -EINVAL;
253
254         settings = (struct pyra_settings const *)buf;
255         if (settings->startup_profile >= ARRAY_SIZE(pyra->profile_settings))
256                 return -EINVAL;
257
258         mutex_lock(&pyra->pyra_lock);
259
260         retval = pyra_set_settings(usb_dev, settings);
261         if (retval) {
262                 mutex_unlock(&pyra->pyra_lock);
263                 return retval;
264         }
265
266         profile_activated(pyra, settings->startup_profile);
267
268         roccat_report.type = PYRA_MOUSE_EVENT_BUTTON_TYPE_PROFILE_2;
269         roccat_report.value = settings->startup_profile + 1;
270         roccat_report.key = 0;
271         roccat_report_event(pyra->chrdev_minor,
272                         (uint8_t const *)&roccat_report);
273
274         mutex_unlock(&pyra->pyra_lock);
275         return PYRA_SIZE_SETTINGS;
276 }
277
278 PYRA_SYSFS_R(settings, SETTINGS);
279 static struct bin_attribute bin_attr_settings =
280         __BIN_ATTR(settings, (S_IWUSR | S_IRUGO),
281                    pyra_sysfs_read_settings, pyra_sysfs_write_settings,
282                    PYRA_SIZE_SETTINGS);
283
284 static ssize_t pyra_sysfs_show_actual_cpi(struct device *dev,
285                 struct device_attribute *attr, char *buf)
286 {
287         struct pyra_device *pyra =
288                         hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
289         return snprintf(buf, PAGE_SIZE, "%d\n", pyra->actual_cpi);
290 }
291 static DEVICE_ATTR(actual_cpi, 0440, pyra_sysfs_show_actual_cpi, NULL);
292
293 static ssize_t pyra_sysfs_show_actual_profile(struct device *dev,
294                 struct device_attribute *attr, char *buf)
295 {
296         struct pyra_device *pyra =
297                         hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
298         struct usb_device *usb_dev = interface_to_usbdev(to_usb_interface(dev));
299         struct pyra_settings settings;
300
301         mutex_lock(&pyra->pyra_lock);
302         roccat_common2_receive(usb_dev, PYRA_COMMAND_SETTINGS,
303                         &settings, PYRA_SIZE_SETTINGS);
304         mutex_unlock(&pyra->pyra_lock);
305
306         return snprintf(buf, PAGE_SIZE, "%d\n", settings.startup_profile);
307 }
308 static DEVICE_ATTR(actual_profile, 0440, pyra_sysfs_show_actual_profile, NULL);
309 static DEVICE_ATTR(startup_profile, 0440, pyra_sysfs_show_actual_profile, NULL);
310
311 static ssize_t pyra_sysfs_show_firmware_version(struct device *dev,
312                 struct device_attribute *attr, char *buf)
313 {
314         struct pyra_device *pyra;
315         struct usb_device *usb_dev;
316         struct pyra_info info;
317
318         dev = dev->parent->parent;
319         pyra = hid_get_drvdata(dev_get_drvdata(dev));
320         usb_dev = interface_to_usbdev(to_usb_interface(dev));
321
322         mutex_lock(&pyra->pyra_lock);
323         roccat_common2_receive(usb_dev, PYRA_COMMAND_INFO,
324                         &info, PYRA_SIZE_INFO);
325         mutex_unlock(&pyra->pyra_lock);
326
327         return snprintf(buf, PAGE_SIZE, "%d\n", info.firmware_version);
328 }
329 static DEVICE_ATTR(firmware_version, 0440, pyra_sysfs_show_firmware_version,
330                    NULL);
331
332 static struct attribute *pyra_attrs[] = {
333         &dev_attr_actual_cpi.attr,
334         &dev_attr_actual_profile.attr,
335         &dev_attr_firmware_version.attr,
336         &dev_attr_startup_profile.attr,
337         NULL,
338 };
339
340 static struct bin_attribute *pyra_bin_attributes[] = {
341         &bin_attr_control,
342         &bin_attr_info,
343         &bin_attr_profile_settings,
344         &bin_attr_profile_buttons,
345         &bin_attr_settings,
346         &bin_attr_profile1_settings,
347         &bin_attr_profile2_settings,
348         &bin_attr_profile3_settings,
349         &bin_attr_profile4_settings,
350         &bin_attr_profile5_settings,
351         &bin_attr_profile1_buttons,
352         &bin_attr_profile2_buttons,
353         &bin_attr_profile3_buttons,
354         &bin_attr_profile4_buttons,
355         &bin_attr_profile5_buttons,
356         NULL,
357 };
358
359 static const struct attribute_group pyra_group = {
360         .attrs = pyra_attrs,
361         .bin_attrs = pyra_bin_attributes,
362 };
363
364 static const struct attribute_group *pyra_groups[] = {
365         &pyra_group,
366         NULL,
367 };
368
369 static int pyra_init_pyra_device_struct(struct usb_device *usb_dev,
370                 struct pyra_device *pyra)
371 {
372         struct pyra_settings settings;
373         int retval, i;
374
375         mutex_init(&pyra->pyra_lock);
376
377         retval = pyra_get_settings(usb_dev, &settings);
378         if (retval)
379                 return retval;
380
381         for (i = 0; i < 5; ++i) {
382                 retval = pyra_get_profile_settings(usb_dev,
383                                 &pyra->profile_settings[i], i);
384                 if (retval)
385                         return retval;
386         }
387
388         profile_activated(pyra, settings.startup_profile);
389
390         return 0;
391 }
392
393 static int pyra_init_specials(struct hid_device *hdev)
394 {
395         struct usb_interface *intf = to_usb_interface(hdev->dev.parent);
396         struct usb_device *usb_dev = interface_to_usbdev(intf);
397         struct pyra_device *pyra;
398         int retval;
399
400         if (intf->cur_altsetting->desc.bInterfaceProtocol
401                         == USB_INTERFACE_PROTOCOL_MOUSE) {
402
403                 pyra = kzalloc(sizeof(*pyra), GFP_KERNEL);
404                 if (!pyra) {
405                         hid_err(hdev, "can't alloc device descriptor\n");
406                         return -ENOMEM;
407                 }
408                 hid_set_drvdata(hdev, pyra);
409
410                 retval = pyra_init_pyra_device_struct(usb_dev, pyra);
411                 if (retval) {
412                         hid_err(hdev, "couldn't init struct pyra_device\n");
413                         goto exit_free;
414                 }
415
416                 retval = roccat_connect(pyra_class, hdev,
417                                 sizeof(struct pyra_roccat_report));
418                 if (retval < 0) {
419                         hid_err(hdev, "couldn't init char dev\n");
420                 } else {
421                         pyra->chrdev_minor = retval;
422                         pyra->roccat_claimed = 1;
423                 }
424         } else {
425                 hid_set_drvdata(hdev, NULL);
426         }
427
428         return 0;
429 exit_free:
430         kfree(pyra);
431         return retval;
432 }
433
434 static void pyra_remove_specials(struct hid_device *hdev)
435 {
436         struct usb_interface *intf = to_usb_interface(hdev->dev.parent);
437         struct pyra_device *pyra;
438
439         if (intf->cur_altsetting->desc.bInterfaceProtocol
440                         == USB_INTERFACE_PROTOCOL_MOUSE) {
441                 pyra = hid_get_drvdata(hdev);
442                 if (pyra->roccat_claimed)
443                         roccat_disconnect(pyra->chrdev_minor);
444                 kfree(hid_get_drvdata(hdev));
445         }
446 }
447
448 static int pyra_probe(struct hid_device *hdev, const struct hid_device_id *id)
449 {
450         int retval;
451
452         retval = hid_parse(hdev);
453         if (retval) {
454                 hid_err(hdev, "parse failed\n");
455                 goto exit;
456         }
457
458         retval = hid_hw_start(hdev, HID_CONNECT_DEFAULT);
459         if (retval) {
460                 hid_err(hdev, "hw start failed\n");
461                 goto exit;
462         }
463
464         retval = pyra_init_specials(hdev);
465         if (retval) {
466                 hid_err(hdev, "couldn't install mouse\n");
467                 goto exit_stop;
468         }
469         return 0;
470
471 exit_stop:
472         hid_hw_stop(hdev);
473 exit:
474         return retval;
475 }
476
477 static void pyra_remove(struct hid_device *hdev)
478 {
479         pyra_remove_specials(hdev);
480         hid_hw_stop(hdev);
481 }
482
483 static void pyra_keep_values_up_to_date(struct pyra_device *pyra,
484                 u8 const *data)
485 {
486         struct pyra_mouse_event_button const *button_event;
487
488         switch (data[0]) {
489         case PYRA_MOUSE_REPORT_NUMBER_BUTTON:
490                 button_event = (struct pyra_mouse_event_button const *)data;
491                 switch (button_event->type) {
492                 case PYRA_MOUSE_EVENT_BUTTON_TYPE_PROFILE_2:
493                         profile_activated(pyra, button_event->data1 - 1);
494                         break;
495                 case PYRA_MOUSE_EVENT_BUTTON_TYPE_CPI:
496                         pyra->actual_cpi = button_event->data1;
497                         break;
498                 }
499                 break;
500         }
501 }
502
503 static void pyra_report_to_chrdev(struct pyra_device const *pyra,
504                 u8 const *data)
505 {
506         struct pyra_roccat_report roccat_report;
507         struct pyra_mouse_event_button const *button_event;
508
509         if (data[0] != PYRA_MOUSE_REPORT_NUMBER_BUTTON)
510                 return;
511
512         button_event = (struct pyra_mouse_event_button const *)data;
513
514         switch (button_event->type) {
515         case PYRA_MOUSE_EVENT_BUTTON_TYPE_PROFILE_2:
516         case PYRA_MOUSE_EVENT_BUTTON_TYPE_CPI:
517                 roccat_report.type = button_event->type;
518                 roccat_report.value = button_event->data1;
519                 roccat_report.key = 0;
520                 roccat_report_event(pyra->chrdev_minor,
521                                 (uint8_t const *)&roccat_report);
522                 break;
523         case PYRA_MOUSE_EVENT_BUTTON_TYPE_MACRO:
524         case PYRA_MOUSE_EVENT_BUTTON_TYPE_SHORTCUT:
525         case PYRA_MOUSE_EVENT_BUTTON_TYPE_QUICKLAUNCH:
526                 if (button_event->data2 == PYRA_MOUSE_EVENT_BUTTON_PRESS) {
527                         roccat_report.type = button_event->type;
528                         roccat_report.key = button_event->data1;
529                         /*
530                          * pyra reports profile numbers with range 1-5.
531                          * Keeping this behaviour.
532                          */
533                         roccat_report.value = pyra->actual_profile + 1;
534                         roccat_report_event(pyra->chrdev_minor,
535                                         (uint8_t const *)&roccat_report);
536                 }
537                 break;
538         }
539 }
540
541 static int pyra_raw_event(struct hid_device *hdev, struct hid_report *report,
542                 u8 *data, int size)
543 {
544         struct usb_interface *intf = to_usb_interface(hdev->dev.parent);
545         struct pyra_device *pyra = hid_get_drvdata(hdev);
546
547         if (intf->cur_altsetting->desc.bInterfaceProtocol
548                         != USB_INTERFACE_PROTOCOL_MOUSE)
549                 return 0;
550
551         if (pyra == NULL)
552                 return 0;
553
554         pyra_keep_values_up_to_date(pyra, data);
555
556         if (pyra->roccat_claimed)
557                 pyra_report_to_chrdev(pyra, data);
558
559         return 0;
560 }
561
562 static const struct hid_device_id pyra_devices[] = {
563         { HID_USB_DEVICE(USB_VENDOR_ID_ROCCAT,
564                         USB_DEVICE_ID_ROCCAT_PYRA_WIRED) },
565         { HID_USB_DEVICE(USB_VENDOR_ID_ROCCAT,
566                         USB_DEVICE_ID_ROCCAT_PYRA_WIRELESS) },
567         { }
568 };
569
570 MODULE_DEVICE_TABLE(hid, pyra_devices);
571
572 static struct hid_driver pyra_driver = {
573                 .name = "pyra",
574                 .id_table = pyra_devices,
575                 .probe = pyra_probe,
576                 .remove = pyra_remove,
577                 .raw_event = pyra_raw_event
578 };
579
580 static int __init pyra_init(void)
581 {
582         int retval;
583
584         /* class name has to be same as driver name */
585         pyra_class = class_create(THIS_MODULE, "pyra");
586         if (IS_ERR(pyra_class))
587                 return PTR_ERR(pyra_class);
588         pyra_class->dev_groups = pyra_groups;
589
590         retval = hid_register_driver(&pyra_driver);
591         if (retval)
592                 class_destroy(pyra_class);
593         return retval;
594 }
595
596 static void __exit pyra_exit(void)
597 {
598         hid_unregister_driver(&pyra_driver);
599         class_destroy(pyra_class);
600 }
601
602 module_init(pyra_init);
603 module_exit(pyra_exit);
604
605 MODULE_AUTHOR("Stefan Achatz");
606 MODULE_DESCRIPTION("USB Roccat Pyra driver");
607 MODULE_LICENSE("GPL v2");