1 // SPDX-License-Identifier: GPL-2.0+
2 
3 /*
4  *  HID driver for UC-Logic devices not fully compliant with HID standard
5  *
6  *  Copyright (c) 2022 José Expósito <jose.exposito89@gmail.com>
7  */
8 
9 #include <kunit/test.h>
10 #include "./hid-uclogic-params.h"
11 #include "./hid-uclogic-rdesc.h"
12 
13 #define MAX_STR_DESC_SIZE 14
14 
15 struct uclogic_parse_ugee_v2_desc_case {
16 	const char *name;
17 	int res;
18 	const __u8 str_desc[MAX_STR_DESC_SIZE];
19 	size_t str_desc_size;
20 	const s32 desc_params[UCLOGIC_RDESC_PH_ID_NUM];
21 	enum uclogic_params_frame_type frame_type;
22 };
23 
24 static struct uclogic_parse_ugee_v2_desc_case uclogic_parse_ugee_v2_desc_cases[] = {
25 	{
26 		.name = "invalid_str_desc",
27 		.res = -EINVAL,
28 		.str_desc = {},
29 		.str_desc_size = 0,
30 		.desc_params = {},
31 		.frame_type = UCLOGIC_PARAMS_FRAME_BUTTONS,
32 	},
33 	{
34 		.name = "resolution_with_value_0",
35 		.res = 0,
36 		.str_desc = {
37 			0x0E, 0x03,
38 			0x70, 0xB2,
39 			0x10, 0x77,
40 			0x08,
41 			0x00,
42 			0xFF, 0x1F,
43 			0x00, 0x00,
44 		},
45 		.str_desc_size = 12,
46 		.desc_params = {
47 			[UCLOGIC_RDESC_PEN_PH_ID_X_LM] = 0xB270,
48 			[UCLOGIC_RDESC_PEN_PH_ID_X_PM] = 0,
49 			[UCLOGIC_RDESC_PEN_PH_ID_Y_LM] = 0x7710,
50 			[UCLOGIC_RDESC_PEN_PH_ID_Y_PM] = 0,
51 			[UCLOGIC_RDESC_PEN_PH_ID_PRESSURE_LM] = 0x1FFF,
52 			[UCLOGIC_RDESC_FRAME_PH_ID_UM] = 0x08,
53 		},
54 		.frame_type = UCLOGIC_PARAMS_FRAME_BUTTONS,
55 	},
56 	/* XP-PEN Deco L str_desc: Frame with 8 buttons */
57 	{
58 		.name = "frame_type_buttons",
59 		.res = 0,
60 		.str_desc = {
61 			0x0E, 0x03,
62 			0x70, 0xB2,
63 			0x10, 0x77,
64 			0x08,
65 			0x00,
66 			0xFF, 0x1F,
67 			0xD8, 0x13,
68 		},
69 		.str_desc_size = 12,
70 		.desc_params = {
71 			[UCLOGIC_RDESC_PEN_PH_ID_X_LM] = 0xB270,
72 			[UCLOGIC_RDESC_PEN_PH_ID_X_PM] = 0x2320,
73 			[UCLOGIC_RDESC_PEN_PH_ID_Y_LM] = 0x7710,
74 			[UCLOGIC_RDESC_PEN_PH_ID_Y_PM] = 0x1770,
75 			[UCLOGIC_RDESC_PEN_PH_ID_PRESSURE_LM] = 0x1FFF,
76 			[UCLOGIC_RDESC_FRAME_PH_ID_UM] = 0x08,
77 		},
78 		.frame_type = UCLOGIC_PARAMS_FRAME_BUTTONS,
79 	},
80 	/* PARBLO A610 PRO str_desc: Frame with 9 buttons and dial */
81 	{
82 		.name = "frame_type_dial",
83 		.res = 0,
84 		.str_desc = {
85 			0x0E, 0x03,
86 			0x96, 0xC7,
87 			0xF9, 0x7C,
88 			0x09,
89 			0x01,
90 			0xFF, 0x1F,
91 			0xD8, 0x13,
92 		},
93 		.str_desc_size = 12,
94 		.desc_params = {
95 			[UCLOGIC_RDESC_PEN_PH_ID_X_LM] = 0xC796,
96 			[UCLOGIC_RDESC_PEN_PH_ID_X_PM] = 0x2749,
97 			[UCLOGIC_RDESC_PEN_PH_ID_Y_LM] = 0x7CF9,
98 			[UCLOGIC_RDESC_PEN_PH_ID_Y_PM] = 0x1899,
99 			[UCLOGIC_RDESC_PEN_PH_ID_PRESSURE_LM] = 0x1FFF,
100 			[UCLOGIC_RDESC_FRAME_PH_ID_UM] = 0x09,
101 		},
102 		.frame_type = UCLOGIC_PARAMS_FRAME_DIAL,
103 	},
104 	/* XP-PEN Deco Pro S str_desc: Frame with 8 buttons and mouse */
105 	{
106 		.name = "frame_type_mouse",
107 		.res = 0,
108 		.str_desc = {
109 			0x0E, 0x03,
110 			0xC8, 0xB3,
111 			0x34, 0x65,
112 			0x08,
113 			0x02,
114 			0xFF, 0x1F,
115 			0xD8, 0x13,
116 		},
117 		.str_desc_size = 12,
118 		.desc_params = {
119 			[UCLOGIC_RDESC_PEN_PH_ID_X_LM] = 0xB3C8,
120 			[UCLOGIC_RDESC_PEN_PH_ID_X_PM] = 0x2363,
121 			[UCLOGIC_RDESC_PEN_PH_ID_Y_LM] = 0x6534,
122 			[UCLOGIC_RDESC_PEN_PH_ID_Y_PM] = 0x13EC,
123 			[UCLOGIC_RDESC_PEN_PH_ID_PRESSURE_LM] = 0x1FFF,
124 			[UCLOGIC_RDESC_FRAME_PH_ID_UM] = 0x08,
125 		},
126 		.frame_type = UCLOGIC_PARAMS_FRAME_MOUSE,
127 	},
128 };
129 
130 static void uclogic_parse_ugee_v2_desc_case_desc(struct uclogic_parse_ugee_v2_desc_case *t,
131 						 char *desc)
132 {
133 	strscpy(desc, t->name, KUNIT_PARAM_DESC_SIZE);
134 }
135 
136 KUNIT_ARRAY_PARAM(uclogic_parse_ugee_v2_desc, uclogic_parse_ugee_v2_desc_cases,
137 		  uclogic_parse_ugee_v2_desc_case_desc);
138 
139 static void hid_test_uclogic_parse_ugee_v2_desc(struct kunit *test)
140 {
141 	int res;
142 	s32 desc_params[UCLOGIC_RDESC_PH_ID_NUM];
143 	enum uclogic_params_frame_type frame_type;
144 	const struct uclogic_parse_ugee_v2_desc_case *params = test->param_value;
145 
146 	res = uclogic_params_parse_ugee_v2_desc(params->str_desc,
147 						params->str_desc_size,
148 						desc_params,
149 						ARRAY_SIZE(desc_params),
150 						&frame_type);
151 	KUNIT_ASSERT_EQ(test, res, params->res);
152 
153 	if (res)
154 		return;
155 
156 	KUNIT_EXPECT_EQ(test,
157 			params->desc_params[UCLOGIC_RDESC_PEN_PH_ID_X_LM],
158 			desc_params[UCLOGIC_RDESC_PEN_PH_ID_X_LM]);
159 	KUNIT_EXPECT_EQ(test,
160 			params->desc_params[UCLOGIC_RDESC_PEN_PH_ID_X_PM],
161 			desc_params[UCLOGIC_RDESC_PEN_PH_ID_X_PM]);
162 	KUNIT_EXPECT_EQ(test,
163 			params->desc_params[UCLOGIC_RDESC_PEN_PH_ID_Y_LM],
164 			desc_params[UCLOGIC_RDESC_PEN_PH_ID_Y_LM]);
165 	KUNIT_EXPECT_EQ(test,
166 			params->desc_params[UCLOGIC_RDESC_PEN_PH_ID_Y_PM],
167 			desc_params[UCLOGIC_RDESC_PEN_PH_ID_Y_PM]);
168 	KUNIT_EXPECT_EQ(test,
169 			params->desc_params[UCLOGIC_RDESC_PEN_PH_ID_PRESSURE_LM],
170 			desc_params[UCLOGIC_RDESC_PEN_PH_ID_PRESSURE_LM]);
171 	KUNIT_EXPECT_EQ(test,
172 			params->desc_params[UCLOGIC_RDESC_FRAME_PH_ID_UM],
173 			desc_params[UCLOGIC_RDESC_FRAME_PH_ID_UM]);
174 	KUNIT_EXPECT_EQ(test, params->frame_type, frame_type);
175 }
176 
177 struct fake_device {
178 	unsigned long quirks;
179 };
180 
181 static void hid_test_uclogic_params_cleanup_event_hooks(struct kunit *test)
182 {
183 	int res, n;
184 	struct hid_device *hdev;
185 	struct fake_device *fake_dev;
186 	struct uclogic_params p = {0, };
187 
188 	hdev = kunit_kzalloc(test, sizeof(struct hid_device), GFP_KERNEL);
189 	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, hdev);
190 
191 	fake_dev = kunit_kzalloc(test, sizeof(struct fake_device), GFP_KERNEL);
192 	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, fake_dev);
193 
194 	hid_set_drvdata(hdev, fake_dev);
195 
196 	res = uclogic_params_ugee_v2_init_event_hooks(hdev, &p);
197 	KUNIT_ASSERT_EQ(test, res, 0);
198 
199 	/* Check that the function can be called repeatedly */
200 	for (n = 0; n < 4; n++) {
201 		uclogic_params_cleanup_event_hooks(&p);
202 		KUNIT_EXPECT_PTR_EQ(test, p.event_hooks, NULL);
203 	}
204 }
205 
206 static struct kunit_case hid_uclogic_params_test_cases[] = {
207 	KUNIT_CASE_PARAM(hid_test_uclogic_parse_ugee_v2_desc,
208 			 uclogic_parse_ugee_v2_desc_gen_params),
209 	KUNIT_CASE(hid_test_uclogic_params_cleanup_event_hooks),
210 	{}
211 };
212 
213 static struct kunit_suite hid_uclogic_params_test_suite = {
214 	.name = "hid_uclogic_params_test",
215 	.test_cases = hid_uclogic_params_test_cases,
216 };
217 
218 kunit_test_suite(hid_uclogic_params_test_suite);
219 
220 MODULE_DESCRIPTION("KUnit tests for the UC-Logic driver");
221 MODULE_LICENSE("GPL");
222 MODULE_AUTHOR("José Expósito <jose.exposito89@gmail.com>");
223