Linux Audio

Check our new training course

Loading...
v5.9
  1// SPDX-License-Identifier: GPL-2.0-or-later
  2/*
  3 * HID driver for Corsair devices
  4 *
  5 * Supported devices:
  6 *  - Vengeance K70 Keyboard
  7 *  - K70 RAPIDFIRE Keyboard
  8 *  - Vengeance K90 Keyboard
  9 *  - Scimitar PRO RGB Gaming Mouse
 10 *
 11 * Copyright (c) 2015 Clement Vuchener
 12 * Copyright (c) 2017 Oscar Campos
 13 * Copyright (c) 2017 Aaron Bottegal
 14 */
 15
 16/*
 17 */
 18
 19#include <linux/hid.h>
 20#include <linux/module.h>
 21#include <linux/usb.h>
 22#include <linux/leds.h>
 23
 24#include "hid-ids.h"
 25
 26#define CORSAIR_USE_K90_MACRO	(1<<0)
 27#define CORSAIR_USE_K90_BACKLIGHT	(1<<1)
 28
 29struct k90_led {
 30	struct led_classdev cdev;
 31	int brightness;
 32	struct work_struct work;
 33	bool removed;
 34};
 35
 36struct k90_drvdata {
 37	struct k90_led record_led;
 38};
 39
 40struct corsair_drvdata {
 41	unsigned long quirks;
 42	struct k90_drvdata *k90;
 43	struct k90_led *backlight;
 44};
 45
 46#define K90_GKEY_COUNT	18
 47
 48static int corsair_usage_to_gkey(unsigned int usage)
 49{
 50	/* G1 (0xd0) to G16 (0xdf) */
 51	if (usage >= 0xd0 && usage <= 0xdf)
 52		return usage - 0xd0 + 1;
 53	/* G17 (0xe8) to G18 (0xe9) */
 54	if (usage >= 0xe8 && usage <= 0xe9)
 55		return usage - 0xe8 + 17;
 56	return 0;
 57}
 58
 59static unsigned short corsair_gkey_map[K90_GKEY_COUNT] = {
 60	BTN_TRIGGER_HAPPY1,
 61	BTN_TRIGGER_HAPPY2,
 62	BTN_TRIGGER_HAPPY3,
 63	BTN_TRIGGER_HAPPY4,
 64	BTN_TRIGGER_HAPPY5,
 65	BTN_TRIGGER_HAPPY6,
 66	BTN_TRIGGER_HAPPY7,
 67	BTN_TRIGGER_HAPPY8,
 68	BTN_TRIGGER_HAPPY9,
 69	BTN_TRIGGER_HAPPY10,
 70	BTN_TRIGGER_HAPPY11,
 71	BTN_TRIGGER_HAPPY12,
 72	BTN_TRIGGER_HAPPY13,
 73	BTN_TRIGGER_HAPPY14,
 74	BTN_TRIGGER_HAPPY15,
 75	BTN_TRIGGER_HAPPY16,
 76	BTN_TRIGGER_HAPPY17,
 77	BTN_TRIGGER_HAPPY18,
 78};
 79
 80module_param_array_named(gkey_codes, corsair_gkey_map, ushort, NULL, S_IRUGO);
 81MODULE_PARM_DESC(gkey_codes, "Key codes for the G-keys");
 82
 83static unsigned short corsair_record_keycodes[2] = {
 84	BTN_TRIGGER_HAPPY19,
 85	BTN_TRIGGER_HAPPY20
 86};
 87
 88module_param_array_named(recordkey_codes, corsair_record_keycodes, ushort,
 89			 NULL, S_IRUGO);
 90MODULE_PARM_DESC(recordkey_codes, "Key codes for the MR (start and stop record) button");
 91
 92static unsigned short corsair_profile_keycodes[3] = {
 93	BTN_TRIGGER_HAPPY21,
 94	BTN_TRIGGER_HAPPY22,
 95	BTN_TRIGGER_HAPPY23
 96};
 97
 98module_param_array_named(profilekey_codes, corsair_profile_keycodes, ushort,
 99			 NULL, S_IRUGO);
100MODULE_PARM_DESC(profilekey_codes, "Key codes for the profile buttons");
101
102#define CORSAIR_USAGE_SPECIAL_MIN 0xf0
103#define CORSAIR_USAGE_SPECIAL_MAX 0xff
104
105#define CORSAIR_USAGE_MACRO_RECORD_START 0xf6
106#define CORSAIR_USAGE_MACRO_RECORD_STOP 0xf7
107
108#define CORSAIR_USAGE_PROFILE 0xf1
109#define CORSAIR_USAGE_M1 0xf1
110#define CORSAIR_USAGE_M2 0xf2
111#define CORSAIR_USAGE_M3 0xf3
112#define CORSAIR_USAGE_PROFILE_MAX 0xf3
113
114#define CORSAIR_USAGE_META_OFF 0xf4
115#define CORSAIR_USAGE_META_ON  0xf5
116
117#define CORSAIR_USAGE_LIGHT 0xfa
118#define CORSAIR_USAGE_LIGHT_OFF 0xfa
119#define CORSAIR_USAGE_LIGHT_DIM 0xfb
120#define CORSAIR_USAGE_LIGHT_MEDIUM 0xfc
121#define CORSAIR_USAGE_LIGHT_BRIGHT 0xfd
122#define CORSAIR_USAGE_LIGHT_MAX 0xfd
123
124/* USB control protocol */
125
126#define K90_REQUEST_BRIGHTNESS 49
127#define K90_REQUEST_MACRO_MODE 2
128#define K90_REQUEST_STATUS 4
129#define K90_REQUEST_GET_MODE 5
130#define K90_REQUEST_PROFILE 20
131
132#define K90_MACRO_MODE_SW 0x0030
133#define K90_MACRO_MODE_HW 0x0001
134
135#define K90_MACRO_LED_ON  0x0020
136#define K90_MACRO_LED_OFF 0x0040
137
138/*
139 * LED class devices
140 */
141
142#define K90_BACKLIGHT_LED_SUFFIX "::backlight"
143#define K90_RECORD_LED_SUFFIX "::record"
144
145static enum led_brightness k90_backlight_get(struct led_classdev *led_cdev)
146{
147	int ret;
148	struct k90_led *led = container_of(led_cdev, struct k90_led, cdev);
149	struct device *dev = led->cdev.dev->parent;
150	struct usb_interface *usbif = to_usb_interface(dev->parent);
151	struct usb_device *usbdev = interface_to_usbdev(usbif);
152	int brightness;
153	char *data;
154
155	data = kmalloc(8, GFP_KERNEL);
156	if (!data)
157		return -ENOMEM;
158
159	ret = usb_control_msg(usbdev, usb_rcvctrlpipe(usbdev, 0),
160			      K90_REQUEST_STATUS,
161			      USB_DIR_IN | USB_TYPE_VENDOR |
162			      USB_RECIP_DEVICE, 0, 0, data, 8,
163			      USB_CTRL_SET_TIMEOUT);
164	if (ret < 5) {
165		dev_warn(dev, "Failed to get K90 initial state (error %d).\n",
166			 ret);
167		ret = -EIO;
168		goto out;
169	}
170	brightness = data[4];
171	if (brightness < 0 || brightness > 3) {
172		dev_warn(dev,
173			 "Read invalid backlight brightness: %02hhx.\n",
174			 data[4]);
175		ret = -EIO;
176		goto out;
177	}
178	ret = brightness;
179out:
180	kfree(data);
181
182	return ret;
183}
184
185static enum led_brightness k90_record_led_get(struct led_classdev *led_cdev)
186{
187	struct k90_led *led = container_of(led_cdev, struct k90_led, cdev);
188
189	return led->brightness;
190}
191
192static void k90_brightness_set(struct led_classdev *led_cdev,
193			       enum led_brightness brightness)
194{
195	struct k90_led *led = container_of(led_cdev, struct k90_led, cdev);
196
197	led->brightness = brightness;
198	schedule_work(&led->work);
199}
200
201static void k90_backlight_work(struct work_struct *work)
202{
203	int ret;
204	struct k90_led *led = container_of(work, struct k90_led, work);
205	struct device *dev;
206	struct usb_interface *usbif;
207	struct usb_device *usbdev;
208
209	if (led->removed)
210		return;
211
212	dev = led->cdev.dev->parent;
213	usbif = to_usb_interface(dev->parent);
214	usbdev = interface_to_usbdev(usbif);
215
216	ret = usb_control_msg(usbdev, usb_sndctrlpipe(usbdev, 0),
217			      K90_REQUEST_BRIGHTNESS,
218			      USB_DIR_OUT | USB_TYPE_VENDOR |
219			      USB_RECIP_DEVICE, led->brightness, 0,
220			      NULL, 0, USB_CTRL_SET_TIMEOUT);
221	if (ret != 0)
222		dev_warn(dev, "Failed to set backlight brightness (error: %d).\n",
223			 ret);
224}
225
226static void k90_record_led_work(struct work_struct *work)
227{
228	int ret;
229	struct k90_led *led = container_of(work, struct k90_led, work);
230	struct device *dev;
231	struct usb_interface *usbif;
232	struct usb_device *usbdev;
233	int value;
234
235	if (led->removed)
236		return;
237
238	dev = led->cdev.dev->parent;
239	usbif = to_usb_interface(dev->parent);
240	usbdev = interface_to_usbdev(usbif);
241
242	if (led->brightness > 0)
243		value = K90_MACRO_LED_ON;
244	else
245		value = K90_MACRO_LED_OFF;
246
247	ret = usb_control_msg(usbdev, usb_sndctrlpipe(usbdev, 0),
248			      K90_REQUEST_MACRO_MODE,
249			      USB_DIR_OUT | USB_TYPE_VENDOR |
250			      USB_RECIP_DEVICE, value, 0, NULL, 0,
251			      USB_CTRL_SET_TIMEOUT);
252	if (ret != 0)
253		dev_warn(dev, "Failed to set record LED state (error: %d).\n",
254			 ret);
255}
256
257/*
258 * Keyboard attributes
259 */
260
261static ssize_t k90_show_macro_mode(struct device *dev,
262				   struct device_attribute *attr, char *buf)
263{
264	int ret;
265	struct usb_interface *usbif = to_usb_interface(dev->parent);
266	struct usb_device *usbdev = interface_to_usbdev(usbif);
267	const char *macro_mode;
268	char *data;
269
270	data = kmalloc(2, GFP_KERNEL);
271	if (!data)
272		return -ENOMEM;
273
274	ret = usb_control_msg(usbdev, usb_rcvctrlpipe(usbdev, 0),
275			      K90_REQUEST_GET_MODE,
276			      USB_DIR_IN | USB_TYPE_VENDOR |
277			      USB_RECIP_DEVICE, 0, 0, data, 2,
278			      USB_CTRL_SET_TIMEOUT);
279	if (ret < 1) {
280		dev_warn(dev, "Failed to get K90 initial mode (error %d).\n",
281			 ret);
282		ret = -EIO;
283		goto out;
284	}
285
286	switch (data[0]) {
287	case K90_MACRO_MODE_HW:
288		macro_mode = "HW";
289		break;
290
291	case K90_MACRO_MODE_SW:
292		macro_mode = "SW";
293		break;
294	default:
295		dev_warn(dev, "K90 in unknown mode: %02hhx.\n",
296			 data[0]);
297		ret = -EIO;
298		goto out;
299	}
300
301	ret = snprintf(buf, PAGE_SIZE, "%s\n", macro_mode);
302out:
303	kfree(data);
304
305	return ret;
306}
307
308static ssize_t k90_store_macro_mode(struct device *dev,
309				    struct device_attribute *attr,
310				    const char *buf, size_t count)
311{
312	int ret;
313	struct usb_interface *usbif = to_usb_interface(dev->parent);
314	struct usb_device *usbdev = interface_to_usbdev(usbif);
315	__u16 value;
316
317	if (strncmp(buf, "SW", 2) == 0)
318		value = K90_MACRO_MODE_SW;
319	else if (strncmp(buf, "HW", 2) == 0)
320		value = K90_MACRO_MODE_HW;
321	else
322		return -EINVAL;
323
324	ret = usb_control_msg(usbdev, usb_sndctrlpipe(usbdev, 0),
325			      K90_REQUEST_MACRO_MODE,
326			      USB_DIR_OUT | USB_TYPE_VENDOR |
327			      USB_RECIP_DEVICE, value, 0, NULL, 0,
328			      USB_CTRL_SET_TIMEOUT);
329	if (ret != 0) {
330		dev_warn(dev, "Failed to set macro mode.\n");
331		return ret;
332	}
333
334	return count;
335}
336
337static ssize_t k90_show_current_profile(struct device *dev,
338					struct device_attribute *attr,
339					char *buf)
340{
341	int ret;
342	struct usb_interface *usbif = to_usb_interface(dev->parent);
343	struct usb_device *usbdev = interface_to_usbdev(usbif);
344	int current_profile;
345	char *data;
346
347	data = kmalloc(8, GFP_KERNEL);
348	if (!data)
349		return -ENOMEM;
350
351	ret = usb_control_msg(usbdev, usb_rcvctrlpipe(usbdev, 0),
352			      K90_REQUEST_STATUS,
353			      USB_DIR_IN | USB_TYPE_VENDOR |
354			      USB_RECIP_DEVICE, 0, 0, data, 8,
355			      USB_CTRL_SET_TIMEOUT);
356	if (ret < 8) {
357		dev_warn(dev, "Failed to get K90 initial state (error %d).\n",
358			 ret);
359		ret = -EIO;
360		goto out;
361	}
362	current_profile = data[7];
363	if (current_profile < 1 || current_profile > 3) {
364		dev_warn(dev, "Read invalid current profile: %02hhx.\n",
365			 data[7]);
366		ret = -EIO;
367		goto out;
368	}
369
370	ret = snprintf(buf, PAGE_SIZE, "%d\n", current_profile);
371out:
372	kfree(data);
373
374	return ret;
375}
376
377static ssize_t k90_store_current_profile(struct device *dev,
378					 struct device_attribute *attr,
379					 const char *buf, size_t count)
380{
381	int ret;
382	struct usb_interface *usbif = to_usb_interface(dev->parent);
383	struct usb_device *usbdev = interface_to_usbdev(usbif);
384	int profile;
385
386	if (kstrtoint(buf, 10, &profile))
387		return -EINVAL;
388	if (profile < 1 || profile > 3)
389		return -EINVAL;
390
391	ret = usb_control_msg(usbdev, usb_sndctrlpipe(usbdev, 0),
392			      K90_REQUEST_PROFILE,
393			      USB_DIR_OUT | USB_TYPE_VENDOR |
394			      USB_RECIP_DEVICE, profile, 0, NULL, 0,
395			      USB_CTRL_SET_TIMEOUT);
396	if (ret != 0) {
397		dev_warn(dev, "Failed to change current profile (error %d).\n",
398			 ret);
399		return ret;
400	}
401
402	return count;
403}
404
405static DEVICE_ATTR(macro_mode, 0644, k90_show_macro_mode, k90_store_macro_mode);
406static DEVICE_ATTR(current_profile, 0644, k90_show_current_profile,
407		   k90_store_current_profile);
408
409static struct attribute *k90_attrs[] = {
410	&dev_attr_macro_mode.attr,
411	&dev_attr_current_profile.attr,
412	NULL
413};
414
415static const struct attribute_group k90_attr_group = {
416	.attrs = k90_attrs,
417};
418
419/*
420 * Driver functions
421 */
422
423static int k90_init_backlight(struct hid_device *dev)
424{
425	int ret;
426	struct corsair_drvdata *drvdata = hid_get_drvdata(dev);
427	size_t name_sz;
428	char *name;
429
430	drvdata->backlight = kzalloc(sizeof(struct k90_led), GFP_KERNEL);
431	if (!drvdata->backlight) {
432		ret = -ENOMEM;
433		goto fail_backlight_alloc;
434	}
435
436	name_sz =
437	    strlen(dev_name(&dev->dev)) + sizeof(K90_BACKLIGHT_LED_SUFFIX);
438	name = kzalloc(name_sz, GFP_KERNEL);
439	if (!name) {
440		ret = -ENOMEM;
441		goto fail_name_alloc;
442	}
443	snprintf(name, name_sz, "%s" K90_BACKLIGHT_LED_SUFFIX,
444		 dev_name(&dev->dev));
445	drvdata->backlight->removed = false;
446	drvdata->backlight->cdev.name = name;
447	drvdata->backlight->cdev.max_brightness = 3;
448	drvdata->backlight->cdev.brightness_set = k90_brightness_set;
449	drvdata->backlight->cdev.brightness_get = k90_backlight_get;
450	INIT_WORK(&drvdata->backlight->work, k90_backlight_work);
451	ret = led_classdev_register(&dev->dev, &drvdata->backlight->cdev);
452	if (ret != 0)
453		goto fail_register_cdev;
454
455	return 0;
456
457fail_register_cdev:
458	kfree(drvdata->backlight->cdev.name);
459fail_name_alloc:
460	kfree(drvdata->backlight);
461	drvdata->backlight = NULL;
462fail_backlight_alloc:
463	return ret;
464}
465
466static int k90_init_macro_functions(struct hid_device *dev)
467{
468	int ret;
469	struct corsair_drvdata *drvdata = hid_get_drvdata(dev);
470	struct k90_drvdata *k90;
471	size_t name_sz;
472	char *name;
473
474	k90 = kzalloc(sizeof(struct k90_drvdata), GFP_KERNEL);
475	if (!k90) {
476		ret = -ENOMEM;
477		goto fail_drvdata;
478	}
479	drvdata->k90 = k90;
480
481	/* Init LED device for record LED */
482	name_sz = strlen(dev_name(&dev->dev)) + sizeof(K90_RECORD_LED_SUFFIX);
483	name = kzalloc(name_sz, GFP_KERNEL);
484	if (!name) {
485		ret = -ENOMEM;
486		goto fail_record_led_alloc;
487	}
488	snprintf(name, name_sz, "%s" K90_RECORD_LED_SUFFIX,
489		 dev_name(&dev->dev));
490	k90->record_led.removed = false;
491	k90->record_led.cdev.name = name;
492	k90->record_led.cdev.max_brightness = 1;
493	k90->record_led.cdev.brightness_set = k90_brightness_set;
494	k90->record_led.cdev.brightness_get = k90_record_led_get;
495	INIT_WORK(&k90->record_led.work, k90_record_led_work);
496	k90->record_led.brightness = 0;
497	ret = led_classdev_register(&dev->dev, &k90->record_led.cdev);
498	if (ret != 0)
499		goto fail_record_led;
500
501	/* Init attributes */
502	ret = sysfs_create_group(&dev->dev.kobj, &k90_attr_group);
503	if (ret != 0)
504		goto fail_sysfs;
505
506	return 0;
507
508fail_sysfs:
509	k90->record_led.removed = true;
510	led_classdev_unregister(&k90->record_led.cdev);
511	cancel_work_sync(&k90->record_led.work);
512fail_record_led:
513	kfree(k90->record_led.cdev.name);
514fail_record_led_alloc:
515	kfree(k90);
516fail_drvdata:
517	drvdata->k90 = NULL;
518	return ret;
519}
520
521static void k90_cleanup_backlight(struct hid_device *dev)
522{
523	struct corsair_drvdata *drvdata = hid_get_drvdata(dev);
524
525	if (drvdata->backlight) {
526		drvdata->backlight->removed = true;
527		led_classdev_unregister(&drvdata->backlight->cdev);
528		cancel_work_sync(&drvdata->backlight->work);
529		kfree(drvdata->backlight->cdev.name);
530		kfree(drvdata->backlight);
531	}
532}
533
534static void k90_cleanup_macro_functions(struct hid_device *dev)
535{
536	struct corsair_drvdata *drvdata = hid_get_drvdata(dev);
537	struct k90_drvdata *k90 = drvdata->k90;
538
539	if (k90) {
540		sysfs_remove_group(&dev->dev.kobj, &k90_attr_group);
541
542		k90->record_led.removed = true;
543		led_classdev_unregister(&k90->record_led.cdev);
544		cancel_work_sync(&k90->record_led.work);
545		kfree(k90->record_led.cdev.name);
546
547		kfree(k90);
548	}
549}
550
551static int corsair_probe(struct hid_device *dev, const struct hid_device_id *id)
552{
553	int ret;
554	unsigned long quirks = id->driver_data;
555	struct corsair_drvdata *drvdata;
556	struct usb_interface *usbif = to_usb_interface(dev->dev.parent);
557
558	drvdata = devm_kzalloc(&dev->dev, sizeof(struct corsair_drvdata),
559			       GFP_KERNEL);
560	if (drvdata == NULL)
561		return -ENOMEM;
562	drvdata->quirks = quirks;
563	hid_set_drvdata(dev, drvdata);
564
565	ret = hid_parse(dev);
566	if (ret != 0) {
567		hid_err(dev, "parse failed\n");
568		return ret;
569	}
570	ret = hid_hw_start(dev, HID_CONNECT_DEFAULT);
571	if (ret != 0) {
572		hid_err(dev, "hw start failed\n");
573		return ret;
574	}
575
576	if (usbif->cur_altsetting->desc.bInterfaceNumber == 0) {
577		if (quirks & CORSAIR_USE_K90_MACRO) {
578			ret = k90_init_macro_functions(dev);
579			if (ret != 0)
580				hid_warn(dev, "Failed to initialize K90 macro functions.\n");
581		}
582		if (quirks & CORSAIR_USE_K90_BACKLIGHT) {
583			ret = k90_init_backlight(dev);
584			if (ret != 0)
585				hid_warn(dev, "Failed to initialize K90 backlight.\n");
586		}
587	}
588
589	return 0;
590}
591
592static void corsair_remove(struct hid_device *dev)
593{
594	k90_cleanup_macro_functions(dev);
595	k90_cleanup_backlight(dev);
596
597	hid_hw_stop(dev);
598}
599
600static int corsair_event(struct hid_device *dev, struct hid_field *field,
601			 struct hid_usage *usage, __s32 value)
602{
603	struct corsair_drvdata *drvdata = hid_get_drvdata(dev);
604
605	if (!drvdata->k90)
606		return 0;
607
608	switch (usage->hid & HID_USAGE) {
609	case CORSAIR_USAGE_MACRO_RECORD_START:
610		drvdata->k90->record_led.brightness = 1;
611		break;
612	case CORSAIR_USAGE_MACRO_RECORD_STOP:
613		drvdata->k90->record_led.brightness = 0;
614		break;
615	default:
616		break;
617	}
618
619	return 0;
620}
621
622static int corsair_input_mapping(struct hid_device *dev,
623				 struct hid_input *input,
624				 struct hid_field *field,
625				 struct hid_usage *usage, unsigned long **bit,
626				 int *max)
627{
628	int gkey;
629
630	if ((usage->hid & HID_USAGE_PAGE) != HID_UP_KEYBOARD)
631		return 0;
632
633	gkey = corsair_usage_to_gkey(usage->hid & HID_USAGE);
634	if (gkey != 0) {
635		hid_map_usage_clear(input, usage, bit, max, EV_KEY,
636				    corsair_gkey_map[gkey - 1]);
637		return 1;
638	}
639	if ((usage->hid & HID_USAGE) >= CORSAIR_USAGE_SPECIAL_MIN &&
640	    (usage->hid & HID_USAGE) <= CORSAIR_USAGE_SPECIAL_MAX) {
641		switch (usage->hid & HID_USAGE) {
642		case CORSAIR_USAGE_MACRO_RECORD_START:
643			hid_map_usage_clear(input, usage, bit, max, EV_KEY,
644					    corsair_record_keycodes[0]);
645			return 1;
646
647		case CORSAIR_USAGE_MACRO_RECORD_STOP:
648			hid_map_usage_clear(input, usage, bit, max, EV_KEY,
649					    corsair_record_keycodes[1]);
650			return 1;
651
652		case CORSAIR_USAGE_M1:
653			hid_map_usage_clear(input, usage, bit, max, EV_KEY,
654					    corsair_profile_keycodes[0]);
655			return 1;
656
657		case CORSAIR_USAGE_M2:
658			hid_map_usage_clear(input, usage, bit, max, EV_KEY,
659					    corsair_profile_keycodes[1]);
660			return 1;
661
662		case CORSAIR_USAGE_M3:
663			hid_map_usage_clear(input, usage, bit, max, EV_KEY,
664					    corsair_profile_keycodes[2]);
665			return 1;
666
667		default:
668			return -1;
669		}
670	}
671
672	return 0;
673}
674
675/*
676 * The report descriptor of some of the Corsair gaming mice is
677 * non parseable as they define two consecutive Logical Minimum for
678 * the Usage Page (Consumer) in rdescs bytes 75 and 77 being 77 0x16
679 * that should be obviousy 0x26 for Logical Magimum of 16 bits. This
680 * prevents poper parsing of the report descriptor due Logical
681 * Minimum being larger than Logical Maximum.
682 *
683 * This driver fixes the report descriptor for:
684 * - USB ID 1b1c:1b34, sold as GLAIVE RGB Gaming mouse
685 * - USB ID 1b1c:1b3e, sold as Scimitar RGB Pro Gaming mouse
686 */
687
688static __u8 *corsair_mouse_report_fixup(struct hid_device *hdev, __u8 *rdesc,
689        unsigned int *rsize)
690{
691	struct usb_interface *intf = to_usb_interface(hdev->dev.parent);
692
693	if (intf->cur_altsetting->desc.bInterfaceNumber == 1) {
694		/*
695		 * Corsair GLAIVE RGB and Scimitar RGB Pro report descriptor is
696		 * broken and defines two different Logical Minimum for the
697		 * Consumer Application. The byte 77 should be a 0x26 defining
698		 * a 16 bits integer for the Logical Maximum but it is a 0x16
699		 * instead (Logical Minimum)
700		 */
701		switch (hdev->product) {
702		case USB_DEVICE_ID_CORSAIR_GLAIVE_RGB:
703		case USB_DEVICE_ID_CORSAIR_SCIMITAR_PRO_RGB:
704			if (*rsize >= 172 && rdesc[75] == 0x15 && rdesc[77] == 0x16
705			&& rdesc[78] == 0xff && rdesc[79] == 0x0f) {
706				hid_info(hdev, "Fixing up report descriptor\n");
707				rdesc[77] = 0x26;
708			}
709			break;
710		}
711
712	}
713	return rdesc;
714}
715
716static const struct hid_device_id corsair_devices[] = {
717	{ HID_USB_DEVICE(USB_VENDOR_ID_CORSAIR, USB_DEVICE_ID_CORSAIR_K90),
718		.driver_data = CORSAIR_USE_K90_MACRO |
719			       CORSAIR_USE_K90_BACKLIGHT },
720	{ HID_USB_DEVICE(USB_VENDOR_ID_CORSAIR,
721            USB_DEVICE_ID_CORSAIR_GLAIVE_RGB) },
722	{ HID_USB_DEVICE(USB_VENDOR_ID_CORSAIR,
723            USB_DEVICE_ID_CORSAIR_SCIMITAR_PRO_RGB) },
724	/*
725	 * Vengeance K70 and K70 RAPIDFIRE share product IDs.
726	 */
727	{ HID_USB_DEVICE(USB_VENDOR_ID_CORSAIR,
728            USB_DEVICE_ID_CORSAIR_K70R) },
729	{}
730};
731
732MODULE_DEVICE_TABLE(hid, corsair_devices);
733
734static struct hid_driver corsair_driver = {
735	.name = "corsair",
736	.id_table = corsair_devices,
737	.probe = corsair_probe,
738	.event = corsair_event,
739	.remove = corsair_remove,
740	.input_mapping = corsair_input_mapping,
741	.report_fixup = corsair_mouse_report_fixup,
742};
743
744module_hid_driver(corsair_driver);
745
746MODULE_LICENSE("GPL");
747/* Original K90 driver author */
748MODULE_AUTHOR("Clement Vuchener");
749/* Scimitar PRO RGB driver author */
750MODULE_AUTHOR("Oscar Campos");
751MODULE_DESCRIPTION("HID driver for Corsair devices");
v5.4
  1// SPDX-License-Identifier: GPL-2.0-or-later
  2/*
  3 * HID driver for Corsair devices
  4 *
  5 * Supported devices:
  6 *  - Vengeance K70 Keyboard
  7 *  - K70 RAPIDFIRE Keyboard
  8 *  - Vengeance K90 Keyboard
  9 *  - Scimitar PRO RGB Gaming Mouse
 10 *
 11 * Copyright (c) 2015 Clement Vuchener
 12 * Copyright (c) 2017 Oscar Campos
 13 * Copyright (c) 2017 Aaron Bottegal
 14 */
 15
 16/*
 17 */
 18
 19#include <linux/hid.h>
 20#include <linux/module.h>
 21#include <linux/usb.h>
 22#include <linux/leds.h>
 23
 24#include "hid-ids.h"
 25
 26#define CORSAIR_USE_K90_MACRO	(1<<0)
 27#define CORSAIR_USE_K90_BACKLIGHT	(1<<1)
 28
 29struct k90_led {
 30	struct led_classdev cdev;
 31	int brightness;
 32	struct work_struct work;
 33	bool removed;
 34};
 35
 36struct k90_drvdata {
 37	struct k90_led record_led;
 38};
 39
 40struct corsair_drvdata {
 41	unsigned long quirks;
 42	struct k90_drvdata *k90;
 43	struct k90_led *backlight;
 44};
 45
 46#define K90_GKEY_COUNT	18
 47
 48static int corsair_usage_to_gkey(unsigned int usage)
 49{
 50	/* G1 (0xd0) to G16 (0xdf) */
 51	if (usage >= 0xd0 && usage <= 0xdf)
 52		return usage - 0xd0 + 1;
 53	/* G17 (0xe8) to G18 (0xe9) */
 54	if (usage >= 0xe8 && usage <= 0xe9)
 55		return usage - 0xe8 + 17;
 56	return 0;
 57}
 58
 59static unsigned short corsair_gkey_map[K90_GKEY_COUNT] = {
 60	BTN_TRIGGER_HAPPY1,
 61	BTN_TRIGGER_HAPPY2,
 62	BTN_TRIGGER_HAPPY3,
 63	BTN_TRIGGER_HAPPY4,
 64	BTN_TRIGGER_HAPPY5,
 65	BTN_TRIGGER_HAPPY6,
 66	BTN_TRIGGER_HAPPY7,
 67	BTN_TRIGGER_HAPPY8,
 68	BTN_TRIGGER_HAPPY9,
 69	BTN_TRIGGER_HAPPY10,
 70	BTN_TRIGGER_HAPPY11,
 71	BTN_TRIGGER_HAPPY12,
 72	BTN_TRIGGER_HAPPY13,
 73	BTN_TRIGGER_HAPPY14,
 74	BTN_TRIGGER_HAPPY15,
 75	BTN_TRIGGER_HAPPY16,
 76	BTN_TRIGGER_HAPPY17,
 77	BTN_TRIGGER_HAPPY18,
 78};
 79
 80module_param_array_named(gkey_codes, corsair_gkey_map, ushort, NULL, S_IRUGO);
 81MODULE_PARM_DESC(gkey_codes, "Key codes for the G-keys");
 82
 83static unsigned short corsair_record_keycodes[2] = {
 84	BTN_TRIGGER_HAPPY19,
 85	BTN_TRIGGER_HAPPY20
 86};
 87
 88module_param_array_named(recordkey_codes, corsair_record_keycodes, ushort,
 89			 NULL, S_IRUGO);
 90MODULE_PARM_DESC(recordkey_codes, "Key codes for the MR (start and stop record) button");
 91
 92static unsigned short corsair_profile_keycodes[3] = {
 93	BTN_TRIGGER_HAPPY21,
 94	BTN_TRIGGER_HAPPY22,
 95	BTN_TRIGGER_HAPPY23
 96};
 97
 98module_param_array_named(profilekey_codes, corsair_profile_keycodes, ushort,
 99			 NULL, S_IRUGO);
100MODULE_PARM_DESC(profilekey_codes, "Key codes for the profile buttons");
101
102#define CORSAIR_USAGE_SPECIAL_MIN 0xf0
103#define CORSAIR_USAGE_SPECIAL_MAX 0xff
104
105#define CORSAIR_USAGE_MACRO_RECORD_START 0xf6
106#define CORSAIR_USAGE_MACRO_RECORD_STOP 0xf7
107
108#define CORSAIR_USAGE_PROFILE 0xf1
109#define CORSAIR_USAGE_M1 0xf1
110#define CORSAIR_USAGE_M2 0xf2
111#define CORSAIR_USAGE_M3 0xf3
112#define CORSAIR_USAGE_PROFILE_MAX 0xf3
113
114#define CORSAIR_USAGE_META_OFF 0xf4
115#define CORSAIR_USAGE_META_ON  0xf5
116
117#define CORSAIR_USAGE_LIGHT 0xfa
118#define CORSAIR_USAGE_LIGHT_OFF 0xfa
119#define CORSAIR_USAGE_LIGHT_DIM 0xfb
120#define CORSAIR_USAGE_LIGHT_MEDIUM 0xfc
121#define CORSAIR_USAGE_LIGHT_BRIGHT 0xfd
122#define CORSAIR_USAGE_LIGHT_MAX 0xfd
123
124/* USB control protocol */
125
126#define K90_REQUEST_BRIGHTNESS 49
127#define K90_REQUEST_MACRO_MODE 2
128#define K90_REQUEST_STATUS 4
129#define K90_REQUEST_GET_MODE 5
130#define K90_REQUEST_PROFILE 20
131
132#define K90_MACRO_MODE_SW 0x0030
133#define K90_MACRO_MODE_HW 0x0001
134
135#define K90_MACRO_LED_ON  0x0020
136#define K90_MACRO_LED_OFF 0x0040
137
138/*
139 * LED class devices
140 */
141
142#define K90_BACKLIGHT_LED_SUFFIX "::backlight"
143#define K90_RECORD_LED_SUFFIX "::record"
144
145static enum led_brightness k90_backlight_get(struct led_classdev *led_cdev)
146{
147	int ret;
148	struct k90_led *led = container_of(led_cdev, struct k90_led, cdev);
149	struct device *dev = led->cdev.dev->parent;
150	struct usb_interface *usbif = to_usb_interface(dev->parent);
151	struct usb_device *usbdev = interface_to_usbdev(usbif);
152	int brightness;
153	char *data;
154
155	data = kmalloc(8, GFP_KERNEL);
156	if (!data)
157		return -ENOMEM;
158
159	ret = usb_control_msg(usbdev, usb_rcvctrlpipe(usbdev, 0),
160			      K90_REQUEST_STATUS,
161			      USB_DIR_IN | USB_TYPE_VENDOR |
162			      USB_RECIP_DEVICE, 0, 0, data, 8,
163			      USB_CTRL_SET_TIMEOUT);
164	if (ret < 5) {
165		dev_warn(dev, "Failed to get K90 initial state (error %d).\n",
166			 ret);
167		ret = -EIO;
168		goto out;
169	}
170	brightness = data[4];
171	if (brightness < 0 || brightness > 3) {
172		dev_warn(dev,
173			 "Read invalid backlight brightness: %02hhx.\n",
174			 data[4]);
175		ret = -EIO;
176		goto out;
177	}
178	ret = brightness;
179out:
180	kfree(data);
181
182	return ret;
183}
184
185static enum led_brightness k90_record_led_get(struct led_classdev *led_cdev)
186{
187	struct k90_led *led = container_of(led_cdev, struct k90_led, cdev);
188
189	return led->brightness;
190}
191
192static void k90_brightness_set(struct led_classdev *led_cdev,
193			       enum led_brightness brightness)
194{
195	struct k90_led *led = container_of(led_cdev, struct k90_led, cdev);
196
197	led->brightness = brightness;
198	schedule_work(&led->work);
199}
200
201static void k90_backlight_work(struct work_struct *work)
202{
203	int ret;
204	struct k90_led *led = container_of(work, struct k90_led, work);
205	struct device *dev;
206	struct usb_interface *usbif;
207	struct usb_device *usbdev;
208
209	if (led->removed)
210		return;
211
212	dev = led->cdev.dev->parent;
213	usbif = to_usb_interface(dev->parent);
214	usbdev = interface_to_usbdev(usbif);
215
216	ret = usb_control_msg(usbdev, usb_sndctrlpipe(usbdev, 0),
217			      K90_REQUEST_BRIGHTNESS,
218			      USB_DIR_OUT | USB_TYPE_VENDOR |
219			      USB_RECIP_DEVICE, led->brightness, 0,
220			      NULL, 0, USB_CTRL_SET_TIMEOUT);
221	if (ret != 0)
222		dev_warn(dev, "Failed to set backlight brightness (error: %d).\n",
223			 ret);
224}
225
226static void k90_record_led_work(struct work_struct *work)
227{
228	int ret;
229	struct k90_led *led = container_of(work, struct k90_led, work);
230	struct device *dev;
231	struct usb_interface *usbif;
232	struct usb_device *usbdev;
233	int value;
234
235	if (led->removed)
236		return;
237
238	dev = led->cdev.dev->parent;
239	usbif = to_usb_interface(dev->parent);
240	usbdev = interface_to_usbdev(usbif);
241
242	if (led->brightness > 0)
243		value = K90_MACRO_LED_ON;
244	else
245		value = K90_MACRO_LED_OFF;
246
247	ret = usb_control_msg(usbdev, usb_sndctrlpipe(usbdev, 0),
248			      K90_REQUEST_MACRO_MODE,
249			      USB_DIR_OUT | USB_TYPE_VENDOR |
250			      USB_RECIP_DEVICE, value, 0, NULL, 0,
251			      USB_CTRL_SET_TIMEOUT);
252	if (ret != 0)
253		dev_warn(dev, "Failed to set record LED state (error: %d).\n",
254			 ret);
255}
256
257/*
258 * Keyboard attributes
259 */
260
261static ssize_t k90_show_macro_mode(struct device *dev,
262				   struct device_attribute *attr, char *buf)
263{
264	int ret;
265	struct usb_interface *usbif = to_usb_interface(dev->parent);
266	struct usb_device *usbdev = interface_to_usbdev(usbif);
267	const char *macro_mode;
268	char *data;
269
270	data = kmalloc(2, GFP_KERNEL);
271	if (!data)
272		return -ENOMEM;
273
274	ret = usb_control_msg(usbdev, usb_rcvctrlpipe(usbdev, 0),
275			      K90_REQUEST_GET_MODE,
276			      USB_DIR_IN | USB_TYPE_VENDOR |
277			      USB_RECIP_DEVICE, 0, 0, data, 2,
278			      USB_CTRL_SET_TIMEOUT);
279	if (ret < 1) {
280		dev_warn(dev, "Failed to get K90 initial mode (error %d).\n",
281			 ret);
282		ret = -EIO;
283		goto out;
284	}
285
286	switch (data[0]) {
287	case K90_MACRO_MODE_HW:
288		macro_mode = "HW";
289		break;
290
291	case K90_MACRO_MODE_SW:
292		macro_mode = "SW";
293		break;
294	default:
295		dev_warn(dev, "K90 in unknown mode: %02hhx.\n",
296			 data[0]);
297		ret = -EIO;
298		goto out;
299	}
300
301	ret = snprintf(buf, PAGE_SIZE, "%s\n", macro_mode);
302out:
303	kfree(data);
304
305	return ret;
306}
307
308static ssize_t k90_store_macro_mode(struct device *dev,
309				    struct device_attribute *attr,
310				    const char *buf, size_t count)
311{
312	int ret;
313	struct usb_interface *usbif = to_usb_interface(dev->parent);
314	struct usb_device *usbdev = interface_to_usbdev(usbif);
315	__u16 value;
316
317	if (strncmp(buf, "SW", 2) == 0)
318		value = K90_MACRO_MODE_SW;
319	else if (strncmp(buf, "HW", 2) == 0)
320		value = K90_MACRO_MODE_HW;
321	else
322		return -EINVAL;
323
324	ret = usb_control_msg(usbdev, usb_sndctrlpipe(usbdev, 0),
325			      K90_REQUEST_MACRO_MODE,
326			      USB_DIR_OUT | USB_TYPE_VENDOR |
327			      USB_RECIP_DEVICE, value, 0, NULL, 0,
328			      USB_CTRL_SET_TIMEOUT);
329	if (ret != 0) {
330		dev_warn(dev, "Failed to set macro mode.\n");
331		return ret;
332	}
333
334	return count;
335}
336
337static ssize_t k90_show_current_profile(struct device *dev,
338					struct device_attribute *attr,
339					char *buf)
340{
341	int ret;
342	struct usb_interface *usbif = to_usb_interface(dev->parent);
343	struct usb_device *usbdev = interface_to_usbdev(usbif);
344	int current_profile;
345	char *data;
346
347	data = kmalloc(8, GFP_KERNEL);
348	if (!data)
349		return -ENOMEM;
350
351	ret = usb_control_msg(usbdev, usb_rcvctrlpipe(usbdev, 0),
352			      K90_REQUEST_STATUS,
353			      USB_DIR_IN | USB_TYPE_VENDOR |
354			      USB_RECIP_DEVICE, 0, 0, data, 8,
355			      USB_CTRL_SET_TIMEOUT);
356	if (ret < 8) {
357		dev_warn(dev, "Failed to get K90 initial state (error %d).\n",
358			 ret);
359		ret = -EIO;
360		goto out;
361	}
362	current_profile = data[7];
363	if (current_profile < 1 || current_profile > 3) {
364		dev_warn(dev, "Read invalid current profile: %02hhx.\n",
365			 data[7]);
366		ret = -EIO;
367		goto out;
368	}
369
370	ret = snprintf(buf, PAGE_SIZE, "%d\n", current_profile);
371out:
372	kfree(data);
373
374	return ret;
375}
376
377static ssize_t k90_store_current_profile(struct device *dev,
378					 struct device_attribute *attr,
379					 const char *buf, size_t count)
380{
381	int ret;
382	struct usb_interface *usbif = to_usb_interface(dev->parent);
383	struct usb_device *usbdev = interface_to_usbdev(usbif);
384	int profile;
385
386	if (kstrtoint(buf, 10, &profile))
387		return -EINVAL;
388	if (profile < 1 || profile > 3)
389		return -EINVAL;
390
391	ret = usb_control_msg(usbdev, usb_sndctrlpipe(usbdev, 0),
392			      K90_REQUEST_PROFILE,
393			      USB_DIR_OUT | USB_TYPE_VENDOR |
394			      USB_RECIP_DEVICE, profile, 0, NULL, 0,
395			      USB_CTRL_SET_TIMEOUT);
396	if (ret != 0) {
397		dev_warn(dev, "Failed to change current profile (error %d).\n",
398			 ret);
399		return ret;
400	}
401
402	return count;
403}
404
405static DEVICE_ATTR(macro_mode, 0644, k90_show_macro_mode, k90_store_macro_mode);
406static DEVICE_ATTR(current_profile, 0644, k90_show_current_profile,
407		   k90_store_current_profile);
408
409static struct attribute *k90_attrs[] = {
410	&dev_attr_macro_mode.attr,
411	&dev_attr_current_profile.attr,
412	NULL
413};
414
415static const struct attribute_group k90_attr_group = {
416	.attrs = k90_attrs,
417};
418
419/*
420 * Driver functions
421 */
422
423static int k90_init_backlight(struct hid_device *dev)
424{
425	int ret;
426	struct corsair_drvdata *drvdata = hid_get_drvdata(dev);
427	size_t name_sz;
428	char *name;
429
430	drvdata->backlight = kzalloc(sizeof(struct k90_led), GFP_KERNEL);
431	if (!drvdata->backlight) {
432		ret = -ENOMEM;
433		goto fail_backlight_alloc;
434	}
435
436	name_sz =
437	    strlen(dev_name(&dev->dev)) + sizeof(K90_BACKLIGHT_LED_SUFFIX);
438	name = kzalloc(name_sz, GFP_KERNEL);
439	if (!name) {
440		ret = -ENOMEM;
441		goto fail_name_alloc;
442	}
443	snprintf(name, name_sz, "%s" K90_BACKLIGHT_LED_SUFFIX,
444		 dev_name(&dev->dev));
445	drvdata->backlight->removed = false;
446	drvdata->backlight->cdev.name = name;
447	drvdata->backlight->cdev.max_brightness = 3;
448	drvdata->backlight->cdev.brightness_set = k90_brightness_set;
449	drvdata->backlight->cdev.brightness_get = k90_backlight_get;
450	INIT_WORK(&drvdata->backlight->work, k90_backlight_work);
451	ret = led_classdev_register(&dev->dev, &drvdata->backlight->cdev);
452	if (ret != 0)
453		goto fail_register_cdev;
454
455	return 0;
456
457fail_register_cdev:
458	kfree(drvdata->backlight->cdev.name);
459fail_name_alloc:
460	kfree(drvdata->backlight);
461	drvdata->backlight = NULL;
462fail_backlight_alloc:
463	return ret;
464}
465
466static int k90_init_macro_functions(struct hid_device *dev)
467{
468	int ret;
469	struct corsair_drvdata *drvdata = hid_get_drvdata(dev);
470	struct k90_drvdata *k90;
471	size_t name_sz;
472	char *name;
473
474	k90 = kzalloc(sizeof(struct k90_drvdata), GFP_KERNEL);
475	if (!k90) {
476		ret = -ENOMEM;
477		goto fail_drvdata;
478	}
479	drvdata->k90 = k90;
480
481	/* Init LED device for record LED */
482	name_sz = strlen(dev_name(&dev->dev)) + sizeof(K90_RECORD_LED_SUFFIX);
483	name = kzalloc(name_sz, GFP_KERNEL);
484	if (!name) {
485		ret = -ENOMEM;
486		goto fail_record_led_alloc;
487	}
488	snprintf(name, name_sz, "%s" K90_RECORD_LED_SUFFIX,
489		 dev_name(&dev->dev));
490	k90->record_led.removed = false;
491	k90->record_led.cdev.name = name;
492	k90->record_led.cdev.max_brightness = 1;
493	k90->record_led.cdev.brightness_set = k90_brightness_set;
494	k90->record_led.cdev.brightness_get = k90_record_led_get;
495	INIT_WORK(&k90->record_led.work, k90_record_led_work);
496	k90->record_led.brightness = 0;
497	ret = led_classdev_register(&dev->dev, &k90->record_led.cdev);
498	if (ret != 0)
499		goto fail_record_led;
500
501	/* Init attributes */
502	ret = sysfs_create_group(&dev->dev.kobj, &k90_attr_group);
503	if (ret != 0)
504		goto fail_sysfs;
505
506	return 0;
507
508fail_sysfs:
509	k90->record_led.removed = true;
510	led_classdev_unregister(&k90->record_led.cdev);
511	cancel_work_sync(&k90->record_led.work);
512fail_record_led:
513	kfree(k90->record_led.cdev.name);
514fail_record_led_alloc:
515	kfree(k90);
516fail_drvdata:
517	drvdata->k90 = NULL;
518	return ret;
519}
520
521static void k90_cleanup_backlight(struct hid_device *dev)
522{
523	struct corsair_drvdata *drvdata = hid_get_drvdata(dev);
524
525	if (drvdata->backlight) {
526		drvdata->backlight->removed = true;
527		led_classdev_unregister(&drvdata->backlight->cdev);
528		cancel_work_sync(&drvdata->backlight->work);
529		kfree(drvdata->backlight->cdev.name);
530		kfree(drvdata->backlight);
531	}
532}
533
534static void k90_cleanup_macro_functions(struct hid_device *dev)
535{
536	struct corsair_drvdata *drvdata = hid_get_drvdata(dev);
537	struct k90_drvdata *k90 = drvdata->k90;
538
539	if (k90) {
540		sysfs_remove_group(&dev->dev.kobj, &k90_attr_group);
541
542		k90->record_led.removed = true;
543		led_classdev_unregister(&k90->record_led.cdev);
544		cancel_work_sync(&k90->record_led.work);
545		kfree(k90->record_led.cdev.name);
546
547		kfree(k90);
548	}
549}
550
551static int corsair_probe(struct hid_device *dev, const struct hid_device_id *id)
552{
553	int ret;
554	unsigned long quirks = id->driver_data;
555	struct corsair_drvdata *drvdata;
556	struct usb_interface *usbif = to_usb_interface(dev->dev.parent);
557
558	drvdata = devm_kzalloc(&dev->dev, sizeof(struct corsair_drvdata),
559			       GFP_KERNEL);
560	if (drvdata == NULL)
561		return -ENOMEM;
562	drvdata->quirks = quirks;
563	hid_set_drvdata(dev, drvdata);
564
565	ret = hid_parse(dev);
566	if (ret != 0) {
567		hid_err(dev, "parse failed\n");
568		return ret;
569	}
570	ret = hid_hw_start(dev, HID_CONNECT_DEFAULT);
571	if (ret != 0) {
572		hid_err(dev, "hw start failed\n");
573		return ret;
574	}
575
576	if (usbif->cur_altsetting->desc.bInterfaceNumber == 0) {
577		if (quirks & CORSAIR_USE_K90_MACRO) {
578			ret = k90_init_macro_functions(dev);
579			if (ret != 0)
580				hid_warn(dev, "Failed to initialize K90 macro functions.\n");
581		}
582		if (quirks & CORSAIR_USE_K90_BACKLIGHT) {
583			ret = k90_init_backlight(dev);
584			if (ret != 0)
585				hid_warn(dev, "Failed to initialize K90 backlight.\n");
586		}
587	}
588
589	return 0;
590}
591
592static void corsair_remove(struct hid_device *dev)
593{
594	k90_cleanup_macro_functions(dev);
595	k90_cleanup_backlight(dev);
596
597	hid_hw_stop(dev);
598}
599
600static int corsair_event(struct hid_device *dev, struct hid_field *field,
601			 struct hid_usage *usage, __s32 value)
602{
603	struct corsair_drvdata *drvdata = hid_get_drvdata(dev);
604
605	if (!drvdata->k90)
606		return 0;
607
608	switch (usage->hid & HID_USAGE) {
609	case CORSAIR_USAGE_MACRO_RECORD_START:
610		drvdata->k90->record_led.brightness = 1;
611		break;
612	case CORSAIR_USAGE_MACRO_RECORD_STOP:
613		drvdata->k90->record_led.brightness = 0;
614		break;
615	default:
616		break;
617	}
618
619	return 0;
620}
621
622static int corsair_input_mapping(struct hid_device *dev,
623				 struct hid_input *input,
624				 struct hid_field *field,
625				 struct hid_usage *usage, unsigned long **bit,
626				 int *max)
627{
628	int gkey;
629
630	if ((usage->hid & HID_USAGE_PAGE) != HID_UP_KEYBOARD)
631		return 0;
632
633	gkey = corsair_usage_to_gkey(usage->hid & HID_USAGE);
634	if (gkey != 0) {
635		hid_map_usage_clear(input, usage, bit, max, EV_KEY,
636				    corsair_gkey_map[gkey - 1]);
637		return 1;
638	}
639	if ((usage->hid & HID_USAGE) >= CORSAIR_USAGE_SPECIAL_MIN &&
640	    (usage->hid & HID_USAGE) <= CORSAIR_USAGE_SPECIAL_MAX) {
641		switch (usage->hid & HID_USAGE) {
642		case CORSAIR_USAGE_MACRO_RECORD_START:
643			hid_map_usage_clear(input, usage, bit, max, EV_KEY,
644					    corsair_record_keycodes[0]);
645			return 1;
646
647		case CORSAIR_USAGE_MACRO_RECORD_STOP:
648			hid_map_usage_clear(input, usage, bit, max, EV_KEY,
649					    corsair_record_keycodes[1]);
650			return 1;
651
652		case CORSAIR_USAGE_M1:
653			hid_map_usage_clear(input, usage, bit, max, EV_KEY,
654					    corsair_profile_keycodes[0]);
655			return 1;
656
657		case CORSAIR_USAGE_M2:
658			hid_map_usage_clear(input, usage, bit, max, EV_KEY,
659					    corsair_profile_keycodes[1]);
660			return 1;
661
662		case CORSAIR_USAGE_M3:
663			hid_map_usage_clear(input, usage, bit, max, EV_KEY,
664					    corsair_profile_keycodes[2]);
665			return 1;
666
667		default:
668			return -1;
669		}
670	}
671
672	return 0;
673}
674
675/*
676 * The report descriptor of some of the Corsair gaming mice is
677 * non parseable as they define two consecutive Logical Minimum for
678 * the Usage Page (Consumer) in rdescs bytes 75 and 77 being 77 0x16
679 * that should be obviousy 0x26 for Logical Magimum of 16 bits. This
680 * prevents poper parsing of the report descriptor due Logical
681 * Minimum being larger than Logical Maximum.
682 *
683 * This driver fixes the report descriptor for:
684 * - USB ID 1b1c:1b34, sold as GLAIVE RGB Gaming mouse
685 * - USB ID 1b1c:1b3e, sold as Scimitar RGB Pro Gaming mouse
686 */
687
688static __u8 *corsair_mouse_report_fixup(struct hid_device *hdev, __u8 *rdesc,
689        unsigned int *rsize)
690{
691	struct usb_interface *intf = to_usb_interface(hdev->dev.parent);
692
693	if (intf->cur_altsetting->desc.bInterfaceNumber == 1) {
694		/*
695		 * Corsair GLAIVE RGB and Scimitar RGB Pro report descriptor is
696		 * broken and defines two different Logical Minimum for the
697		 * Consumer Application. The byte 77 should be a 0x26 defining
698		 * a 16 bits integer for the Logical Maximum but it is a 0x16
699		 * instead (Logical Minimum)
700		 */
701		switch (hdev->product) {
702		case USB_DEVICE_ID_CORSAIR_GLAIVE_RGB:
703		case USB_DEVICE_ID_CORSAIR_SCIMITAR_PRO_RGB:
704			if (*rsize >= 172 && rdesc[75] == 0x15 && rdesc[77] == 0x16
705			&& rdesc[78] == 0xff && rdesc[79] == 0x0f) {
706				hid_info(hdev, "Fixing up report descriptor\n");
707				rdesc[77] = 0x26;
708			}
709			break;
710		}
711
712	}
713	return rdesc;
714}
715
716static const struct hid_device_id corsair_devices[] = {
717	{ HID_USB_DEVICE(USB_VENDOR_ID_CORSAIR, USB_DEVICE_ID_CORSAIR_K90),
718		.driver_data = CORSAIR_USE_K90_MACRO |
719			       CORSAIR_USE_K90_BACKLIGHT },
720	{ HID_USB_DEVICE(USB_VENDOR_ID_CORSAIR,
721            USB_DEVICE_ID_CORSAIR_GLAIVE_RGB) },
722	{ HID_USB_DEVICE(USB_VENDOR_ID_CORSAIR,
723            USB_DEVICE_ID_CORSAIR_SCIMITAR_PRO_RGB) },
724	/*
725	 * Vengeance K70 and K70 RAPIDFIRE share product IDs.
726	 */
727	{ HID_USB_DEVICE(USB_VENDOR_ID_CORSAIR,
728            USB_DEVICE_ID_CORSAIR_K70R) },
729	{}
730};
731
732MODULE_DEVICE_TABLE(hid, corsair_devices);
733
734static struct hid_driver corsair_driver = {
735	.name = "corsair",
736	.id_table = corsair_devices,
737	.probe = corsair_probe,
738	.event = corsair_event,
739	.remove = corsair_remove,
740	.input_mapping = corsair_input_mapping,
741	.report_fixup = corsair_mouse_report_fixup,
742};
743
744module_hid_driver(corsair_driver);
745
746MODULE_LICENSE("GPL");
747/* Original K90 driver author */
748MODULE_AUTHOR("Clement Vuchener");
749/* Scimitar PRO RGB driver author */
750MODULE_AUTHOR("Oscar Campos");
751MODULE_DESCRIPTION("HID driver for Corsair devices");