summaryrefslogtreecommitdiffstats
path: root/drivers/hid/hid-razer.c
blob: 740df148b0ef8a9aad88eaeab94a2a9c1866b052 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
// SPDX-License-Identifier: GPL-2.0+
/*
 *  HID driver for gaming keys on Razer Blackwidow gaming keyboards
 *  Macro Key Keycodes: M1 = 191, M2 = 192, M3 = 193, M4 = 194, M5 = 195
 *
 *  Copyright (c) 2021 Jelle van der Waa <jvanderwaa@redhat.com>
 */

#include <linux/device.h>
#include <linux/hid.h>
#include <linux/module.h>
#include <linux/random.h>
#include <linux/sched.h>
#include <linux/usb.h>
#include <linux/wait.h>

#include "hid-ids.h"

#define map_key_clear(c) hid_map_usage_clear(hi, usage, bit, max, EV_KEY, (c))

#define RAZER_BLACKWIDOW_TRANSFER_BUF_SIZE	91

static bool macro_key_remapping = 1;
module_param(macro_key_remapping, bool, 0644);
MODULE_PARM_DESC(macro_key_remapping, " on (Y) off (N)");


static unsigned char blackwidow_init[RAZER_BLACKWIDOW_TRANSFER_BUF_SIZE] = {
	0x00,
	0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0x00, 0x04,
	0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
	0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
	0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
	0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
	0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
	0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
	0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
	0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
	0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
	0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
	0x04, 0x00
};

static int razer_input_mapping(struct hid_device *hdev,
		struct hid_input *hi, struct hid_field *field,
		struct hid_usage *usage, unsigned long **bit, int *max)
{

	if (!macro_key_remapping)
		return 0;

	if ((usage->hid & HID_UP_KEYBOARD) != HID_UP_KEYBOARD)
		return 0;

	switch (usage->hid & ~HID_UP_KEYBOARD) {
	case 0x68:
		map_key_clear(KEY_MACRO1);
		return 1;
	case 0x69:
		map_key_clear(KEY_MACRO2);
		return 1;
	case 0x6a:
		map_key_clear(KEY_MACRO3);
		return 1;
	case 0x6b:
		map_key_clear(KEY_MACRO4);
		return 1;
	case 0x6c:
		map_key_clear(KEY_MACRO5);
		return 1;
	}

	return 0;
}

static int razer_probe(struct hid_device *hdev, const struct hid_device_id *id)
{
	char *buf;
	int ret = 0;

	ret = hid_parse(hdev);
	if (ret)
		return ret;

	/*
	 * Only send the enable macro keys command for the third device
	 * identified as mouse input.
	 */
	if (hdev->type == HID_TYPE_USBMOUSE) {
		buf = kmemdup(blackwidow_init, RAZER_BLACKWIDOW_TRANSFER_BUF_SIZE, GFP_KERNEL);
		if (buf == NULL)
			return -ENOMEM;

		ret = hid_hw_raw_request(hdev, 0, buf, RAZER_BLACKWIDOW_TRANSFER_BUF_SIZE,
				HID_FEATURE_REPORT, HID_REQ_SET_REPORT);
		if (ret != RAZER_BLACKWIDOW_TRANSFER_BUF_SIZE)
			hid_err(hdev, "failed to enable macro keys: %d\n", ret);

		kfree(buf);
	}

	return hid_hw_start(hdev, HID_CONNECT_DEFAULT);
}

static const struct hid_device_id razer_devices[] = {
	{ HID_USB_DEVICE(USB_VENDOR_ID_RAZER,
		USB_DEVICE_ID_RAZER_BLACKWIDOW) },
	{ HID_USB_DEVICE(USB_VENDOR_ID_RAZER,
		USB_DEVICE_ID_RAZER_BLACKWIDOW_CLASSIC) },
	{ HID_USB_DEVICE(USB_VENDOR_ID_RAZER,
		USB_DEVICE_ID_RAZER_BLACKWIDOW_ULTIMATE) },
	{ }
};
MODULE_DEVICE_TABLE(hid, razer_devices);

static struct hid_driver razer_driver = {
	.name = "razer",
	.id_table = razer_devices,
	.input_mapping = razer_input_mapping,
	.probe = razer_probe,
};
module_hid_driver(razer_driver);

MODULE_AUTHOR("Jelle van der Waa <jvanderwaa@redhat.com>");
MODULE_LICENSE("GPL");