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