xiaomi-wmi.c

 1// SPDX-License-Identifier: GPL-2.0
 2/* WMI driver for Xiaomi Laptops */
 3
 4#include <linux/acpi.h>
 5#include <linux/device.h>
 6#include <linux/input.h>
 7#include <linux/module.h>
 8#include <linux/mutex.h>
 9#include <linux/wmi.h>
10
11#include <uapi/linux/input-event-codes.h>
12
13#define XIAOMI_KEY_FN_ESC_0	"A2095CCE-0491-44E7-BA27-F8ED8F88AA86"
14#define XIAOMI_KEY_FN_ESC_1	"7BBE8E39-B486-473D-BA13-66F75C5805CD"
15#define XIAOMI_KEY_FN_FN	"409B028D-F06B-4C7C-8BBB-EE133A6BD87E"
16#define XIAOMI_KEY_CAPSLOCK	"83FE7607-053A-4644-822A-21532C621FC7"
17#define XIAOMI_KEY_FN_F7	"76E9027C-95D0-4180-8692-DA6747DD1C2D"
18
19#define XIAOMI_DEVICE(guid, key)		\
20	.guid_string = (guid),			\
21	.context = &(const unsigned int){key}
22
23struct xiaomi_wmi {
24	struct input_dev *input_dev;
25	struct mutex key_lock;	/* Protects the key event sequence */
26	unsigned int key_code;
27};
28
29static int xiaomi_wmi_probe(struct wmi_device *wdev, const void *context)
30{
31	struct xiaomi_wmi *data;
32	int ret;
33
34	if (!context)
35		return -EINVAL;
36
37	data = devm_kzalloc(&wdev->dev, sizeof(struct xiaomi_wmi), GFP_KERNEL);
38	if (data == NULL)
39		return -ENOMEM;
40	dev_set_drvdata(&wdev->dev, data);
41
42	ret = devm_mutex_init(&wdev->dev, &data->key_lock);
43	if (ret < 0)
44		return ret;
45
46	data->input_dev = devm_input_allocate_device(&wdev->dev);
47	if (data->input_dev == NULL)
48		return -ENOMEM;
49	data->input_dev->name = "Xiaomi WMI keys";
50	data->input_dev->phys = "wmi/input0";
51
52	data->key_code = *((const unsigned int *)context);
53	set_bit(EV_KEY, data->input_dev->evbit);
54	set_bit(data->key_code, data->input_dev->keybit);
55
56	return input_register_device(data->input_dev);
57}
58
59static void xiaomi_wmi_notify(struct wmi_device *wdev, union acpi_object *dummy)
60{
61	struct xiaomi_wmi *data = dev_get_drvdata(&wdev->dev);
62
63	mutex_lock(&data->key_lock);
64	input_report_key(data->input_dev, data->key_code, 1);
65	input_sync(data->input_dev);
66	input_report_key(data->input_dev, data->key_code, 0);
67	input_sync(data->input_dev);
68	mutex_unlock(&data->key_lock);
69}
70
71static const struct wmi_device_id xiaomi_wmi_id_table[] = {
72	// { XIAOMI_DEVICE(XIAOMI_KEY_FN_ESC_0, KEY_FN_ESC) },
73	// { XIAOMI_DEVICE(XIAOMI_KEY_FN_ESC_1, KEY_FN_ESC) },
74	{ XIAOMI_DEVICE(XIAOMI_KEY_FN_FN, KEY_PROG1) },
75	// { XIAOMI_DEVICE(XIAOMI_KEY_CAPSLOCK, KEY_CAPSLOCK) },
76	{ XIAOMI_DEVICE(XIAOMI_KEY_FN_F7, KEY_CUT) },
77
78	/* Terminating entry */
79	{ }
80};
81
82static struct wmi_driver xiaomi_wmi_driver = {
83	.driver = {
84		.name = "xiaomi-wmi",
85	},
86	.id_table = xiaomi_wmi_id_table,
87	.probe = xiaomi_wmi_probe,
88	.notify = xiaomi_wmi_notify,
89	.no_singleton = true,
90};
91module_wmi_driver(xiaomi_wmi_driver);
92
93MODULE_DEVICE_TABLE(wmi, xiaomi_wmi_id_table);
94MODULE_AUTHOR("Mattias Jacobsson");
95MODULE_DESCRIPTION("Xiaomi WMI driver");
96MODULE_LICENSE("GPL v2");