Linux-libre 5.4.48-gnu
[librecmc/linux-libre.git] / drivers / input / tablet / hanwang.c
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  *  USB Hanwang tablet support
4  *
5  *  Copyright (c) 2010 Xing Wei <weixing@hanwang.com.cn>
6  */
7
8 /*
9  */
10
11 #include <linux/types.h>
12 #include <linux/kernel.h>
13 #include <linux/slab.h>
14 #include <linux/module.h>
15 #include <linux/usb/input.h>
16
17 MODULE_AUTHOR("Xing Wei <weixing@hanwang.com.cn>");
18 MODULE_DESCRIPTION("USB Hanwang tablet driver");
19 MODULE_LICENSE("GPL");
20
21 #define USB_VENDOR_ID_HANWANG           0x0b57
22 #define HANWANG_TABLET_INT_CLASS        0x0003
23 #define HANWANG_TABLET_INT_SUB_CLASS    0x0001
24 #define HANWANG_TABLET_INT_PROTOCOL     0x0002
25
26 #define ART_MASTER_PKGLEN_MAX   10
27
28 /* device IDs */
29 #define STYLUS_DEVICE_ID        0x02
30 #define TOUCH_DEVICE_ID         0x03
31 #define CURSOR_DEVICE_ID        0x06
32 #define ERASER_DEVICE_ID        0x0A
33 #define PAD_DEVICE_ID           0x0F
34
35 /* match vendor and interface info  */
36 #define HANWANG_TABLET_DEVICE(vend, cl, sc, pr) \
37         .match_flags = USB_DEVICE_ID_MATCH_VENDOR \
38                 | USB_DEVICE_ID_MATCH_INT_INFO, \
39         .idVendor = (vend), \
40         .bInterfaceClass = (cl), \
41         .bInterfaceSubClass = (sc), \
42         .bInterfaceProtocol = (pr)
43
44 enum hanwang_tablet_type {
45         HANWANG_ART_MASTER_III,
46         HANWANG_ART_MASTER_HD,
47         HANWANG_ART_MASTER_II,
48 };
49
50 struct hanwang {
51         unsigned char *data;
52         dma_addr_t data_dma;
53         struct input_dev *dev;
54         struct usb_device *usbdev;
55         struct urb *irq;
56         const struct hanwang_features *features;
57         unsigned int current_tool;
58         unsigned int current_id;
59         char name[64];
60         char phys[32];
61 };
62
63 struct hanwang_features {
64         unsigned short pid;
65         char *name;
66         enum hanwang_tablet_type type;
67         int pkg_len;
68         int max_x;
69         int max_y;
70         int max_tilt_x;
71         int max_tilt_y;
72         int max_pressure;
73 };
74
75 static const struct hanwang_features features_array[] = {
76         { 0x8528, "Hanwang Art Master III 0906", HANWANG_ART_MASTER_III,
77           ART_MASTER_PKGLEN_MAX, 0x5757, 0x3692, 0x3f, 0x7f, 2048 },
78         { 0x8529, "Hanwang Art Master III 0604", HANWANG_ART_MASTER_III,
79           ART_MASTER_PKGLEN_MAX, 0x3d84, 0x2672, 0x3f, 0x7f, 2048 },
80         { 0x852a, "Hanwang Art Master III 1308", HANWANG_ART_MASTER_III,
81           ART_MASTER_PKGLEN_MAX, 0x7f00, 0x4f60, 0x3f, 0x7f, 2048 },
82         { 0x8401, "Hanwang Art Master HD 5012", HANWANG_ART_MASTER_HD,
83           ART_MASTER_PKGLEN_MAX, 0x678e, 0x4150, 0x3f, 0x7f, 1024 },
84         { 0x8503, "Hanwang Art Master II", HANWANG_ART_MASTER_II,
85           ART_MASTER_PKGLEN_MAX, 0x27de, 0x1cfe, 0x3f, 0x7f, 1024 },
86 };
87
88 static const int hw_eventtypes[] = {
89         EV_KEY, EV_ABS, EV_MSC,
90 };
91
92 static const int hw_absevents[] = {
93         ABS_X, ABS_Y, ABS_TILT_X, ABS_TILT_Y, ABS_WHEEL,
94         ABS_RX, ABS_RY, ABS_PRESSURE, ABS_MISC,
95 };
96
97 static const int hw_btnevents[] = {
98         BTN_STYLUS, BTN_STYLUS2, BTN_TOOL_PEN, BTN_TOOL_RUBBER,
99         BTN_TOOL_MOUSE, BTN_TOOL_FINGER,
100         BTN_0, BTN_1, BTN_2, BTN_3, BTN_4, BTN_5, BTN_6, BTN_7, BTN_8,
101 };
102
103 static const int hw_mscevents[] = {
104         MSC_SERIAL,
105 };
106
107 static void hanwang_parse_packet(struct hanwang *hanwang)
108 {
109         unsigned char *data = hanwang->data;
110         struct input_dev *input_dev = hanwang->dev;
111         struct usb_device *dev = hanwang->usbdev;
112         enum hanwang_tablet_type type = hanwang->features->type;
113         int i;
114         u16 p;
115
116         if (type == HANWANG_ART_MASTER_II) {
117                 hanwang->current_tool = BTN_TOOL_PEN;
118                 hanwang->current_id = STYLUS_DEVICE_ID;
119         }
120
121         switch (data[0]) {
122         case 0x02:      /* data packet */
123                 switch (data[1]) {
124                 case 0x80:      /* tool prox out */
125                         if (type != HANWANG_ART_MASTER_II) {
126                                 hanwang->current_id = 0;
127                                 input_report_key(input_dev,
128                                                  hanwang->current_tool, 0);
129                         }
130                         break;
131
132                 case 0x00:      /* artmaster ii pen leave */
133                         if (type == HANWANG_ART_MASTER_II) {
134                                 hanwang->current_id = 0;
135                                 input_report_key(input_dev,
136                                                  hanwang->current_tool, 0);
137                         }
138                         break;
139
140                 case 0xc2:      /* first time tool prox in */
141                         switch (data[3] & 0xf0) {
142                         case 0x20:      /* art_master III */
143                         case 0x30:      /* art_master_HD */
144                                 hanwang->current_id = STYLUS_DEVICE_ID;
145                                 hanwang->current_tool = BTN_TOOL_PEN;
146                                 input_report_key(input_dev, BTN_TOOL_PEN, 1);
147                                 break;
148                         case 0xa0:      /* art_master III */
149                         case 0xb0:      /* art_master_HD */
150                                 hanwang->current_id = ERASER_DEVICE_ID;
151                                 hanwang->current_tool = BTN_TOOL_RUBBER;
152                                 input_report_key(input_dev, BTN_TOOL_RUBBER, 1);
153                                 break;
154                         default:
155                                 hanwang->current_id = 0;
156                                 dev_dbg(&dev->dev,
157                                         "unknown tablet tool %02x\n", data[0]);
158                                 break;
159                         }
160                         break;
161
162                 default:        /* tool data packet */
163                         switch (type) {
164                         case HANWANG_ART_MASTER_III:
165                                 p = (data[6] << 3) |
166                                     ((data[7] & 0xc0) >> 5) |
167                                     (data[1] & 0x01);
168                                 break;
169
170                         case HANWANG_ART_MASTER_HD:
171                         case HANWANG_ART_MASTER_II:
172                                 p = (data[7] >> 6) | (data[6] << 2);
173                                 break;
174
175                         default:
176                                 p = 0;
177                                 break;
178                         }
179
180                         input_report_abs(input_dev, ABS_X,
181                                          be16_to_cpup((__be16 *)&data[2]));
182                         input_report_abs(input_dev, ABS_Y,
183                                          be16_to_cpup((__be16 *)&data[4]));
184                         input_report_abs(input_dev, ABS_PRESSURE, p);
185                         input_report_abs(input_dev, ABS_TILT_X, data[7] & 0x3f);
186                         input_report_abs(input_dev, ABS_TILT_Y, data[8] & 0x7f);
187                         input_report_key(input_dev, BTN_STYLUS, data[1] & 0x02);
188
189                         if (type != HANWANG_ART_MASTER_II)
190                                 input_report_key(input_dev, BTN_STYLUS2,
191                                                  data[1] & 0x04);
192                         else
193                                 input_report_key(input_dev, BTN_TOOL_PEN, 1);
194
195                         break;
196                 }
197
198                 input_report_abs(input_dev, ABS_MISC, hanwang->current_id);
199                 input_event(input_dev, EV_MSC, MSC_SERIAL,
200                                 hanwang->features->pid);
201                 break;
202
203         case 0x0c:
204                 /* roll wheel */
205                 hanwang->current_id = PAD_DEVICE_ID;
206
207                 switch (type) {
208                 case HANWANG_ART_MASTER_III:
209                         input_report_key(input_dev, BTN_TOOL_FINGER,
210                                          data[1] || data[2] || data[3]);
211                         input_report_abs(input_dev, ABS_WHEEL, data[1]);
212                         input_report_key(input_dev, BTN_0, data[2]);
213                         for (i = 0; i < 8; i++)
214                                 input_report_key(input_dev,
215                                          BTN_1 + i, data[3] & (1 << i));
216                         break;
217
218                 case HANWANG_ART_MASTER_HD:
219                         input_report_key(input_dev, BTN_TOOL_FINGER, data[1] ||
220                                         data[2] || data[3] || data[4] ||
221                                         data[5] || data[6]);
222                         input_report_abs(input_dev, ABS_RX,
223                                         ((data[1] & 0x1f) << 8) | data[2]);
224                         input_report_abs(input_dev, ABS_RY,
225                                         ((data[3] & 0x1f) << 8) | data[4]);
226                         input_report_key(input_dev, BTN_0, data[5] & 0x01);
227                         for (i = 0; i < 4; i++) {
228                                 input_report_key(input_dev,
229                                          BTN_1 + i, data[5] & (1 << i));
230                                 input_report_key(input_dev,
231                                          BTN_5 + i, data[6] & (1 << i));
232                         }
233                         break;
234
235                 case HANWANG_ART_MASTER_II:
236                         dev_dbg(&dev->dev, "error packet  %02x\n", data[0]);
237                         return;
238                 }
239
240                 input_report_abs(input_dev, ABS_MISC, hanwang->current_id);
241                 input_event(input_dev, EV_MSC, MSC_SERIAL, 0xffffffff);
242                 break;
243
244         default:
245                 dev_dbg(&dev->dev, "error packet  %02x\n", data[0]);
246                 break;
247         }
248
249         input_sync(input_dev);
250 }
251
252 static void hanwang_irq(struct urb *urb)
253 {
254         struct hanwang *hanwang = urb->context;
255         struct usb_device *dev = hanwang->usbdev;
256         int retval;
257
258         switch (urb->status) {
259         case 0:
260                 /* success */;
261                 hanwang_parse_packet(hanwang);
262                 break;
263         case -ECONNRESET:
264         case -ENOENT:
265         case -ESHUTDOWN:
266                 /* this urb is terminated, clean up */
267                 dev_err(&dev->dev, "%s - urb shutting down with status: %d",
268                         __func__, urb->status);
269                 return;
270         default:
271                 dev_err(&dev->dev, "%s - nonzero urb status received: %d",
272                         __func__, urb->status);
273                 break;
274         }
275
276         retval = usb_submit_urb(urb, GFP_ATOMIC);
277         if (retval)
278                 dev_err(&dev->dev, "%s - usb_submit_urb failed with result %d",
279                         __func__, retval);
280 }
281
282 static int hanwang_open(struct input_dev *dev)
283 {
284         struct hanwang *hanwang = input_get_drvdata(dev);
285
286         hanwang->irq->dev = hanwang->usbdev;
287         if (usb_submit_urb(hanwang->irq, GFP_KERNEL))
288                 return -EIO;
289
290         return 0;
291 }
292
293 static void hanwang_close(struct input_dev *dev)
294 {
295         struct hanwang *hanwang = input_get_drvdata(dev);
296
297         usb_kill_urb(hanwang->irq);
298 }
299
300 static bool get_features(struct usb_device *dev, struct hanwang *hanwang)
301 {
302         int i;
303
304         for (i = 0; i < ARRAY_SIZE(features_array); i++) {
305                 if (le16_to_cpu(dev->descriptor.idProduct) ==
306                                 features_array[i].pid) {
307                         hanwang->features = &features_array[i];
308                         return true;
309                 }
310         }
311
312         return false;
313 }
314
315
316 static int hanwang_probe(struct usb_interface *intf, const struct usb_device_id *id)
317 {
318         struct usb_device *dev = interface_to_usbdev(intf);
319         struct usb_endpoint_descriptor *endpoint;
320         struct hanwang *hanwang;
321         struct input_dev *input_dev;
322         int error;
323         int i;
324
325         if (intf->cur_altsetting->desc.bNumEndpoints < 1)
326                 return -ENODEV;
327
328         hanwang = kzalloc(sizeof(struct hanwang), GFP_KERNEL);
329         input_dev = input_allocate_device();
330         if (!hanwang || !input_dev) {
331                 error = -ENOMEM;
332                 goto fail1;
333         }
334
335         if (!get_features(dev, hanwang)) {
336                 error = -ENXIO;
337                 goto fail1;
338         }
339
340         hanwang->data = usb_alloc_coherent(dev, hanwang->features->pkg_len,
341                                         GFP_KERNEL, &hanwang->data_dma);
342         if (!hanwang->data) {
343                 error = -ENOMEM;
344                 goto fail1;
345         }
346
347         hanwang->irq = usb_alloc_urb(0, GFP_KERNEL);
348         if (!hanwang->irq) {
349                 error = -ENOMEM;
350                 goto fail2;
351         }
352
353         hanwang->usbdev = dev;
354         hanwang->dev = input_dev;
355
356         usb_make_path(dev, hanwang->phys, sizeof(hanwang->phys));
357         strlcat(hanwang->phys, "/input0", sizeof(hanwang->phys));
358
359         strlcpy(hanwang->name, hanwang->features->name, sizeof(hanwang->name));
360         input_dev->name = hanwang->name;
361         input_dev->phys = hanwang->phys;
362         usb_to_input_id(dev, &input_dev->id);
363         input_dev->dev.parent = &intf->dev;
364
365         input_set_drvdata(input_dev, hanwang);
366
367         input_dev->open = hanwang_open;
368         input_dev->close = hanwang_close;
369
370         for (i = 0; i < ARRAY_SIZE(hw_eventtypes); ++i)
371                 __set_bit(hw_eventtypes[i], input_dev->evbit);
372
373         for (i = 0; i < ARRAY_SIZE(hw_absevents); ++i)
374                 __set_bit(hw_absevents[i], input_dev->absbit);
375
376         for (i = 0; i < ARRAY_SIZE(hw_btnevents); ++i)
377                 __set_bit(hw_btnevents[i], input_dev->keybit);
378
379         for (i = 0; i < ARRAY_SIZE(hw_mscevents); ++i)
380                 __set_bit(hw_mscevents[i], input_dev->mscbit);
381
382         input_set_abs_params(input_dev, ABS_X,
383                              0, hanwang->features->max_x, 4, 0);
384         input_set_abs_params(input_dev, ABS_Y,
385                              0, hanwang->features->max_y, 4, 0);
386         input_set_abs_params(input_dev, ABS_TILT_X,
387                              0, hanwang->features->max_tilt_x, 0, 0);
388         input_set_abs_params(input_dev, ABS_TILT_Y,
389                              0, hanwang->features->max_tilt_y, 0, 0);
390         input_set_abs_params(input_dev, ABS_PRESSURE,
391                              0, hanwang->features->max_pressure, 0, 0);
392
393         endpoint = &intf->cur_altsetting->endpoint[0].desc;
394         usb_fill_int_urb(hanwang->irq, dev,
395                         usb_rcvintpipe(dev, endpoint->bEndpointAddress),
396                         hanwang->data, hanwang->features->pkg_len,
397                         hanwang_irq, hanwang, endpoint->bInterval);
398         hanwang->irq->transfer_dma = hanwang->data_dma;
399         hanwang->irq->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
400
401         error = input_register_device(hanwang->dev);
402         if (error)
403                 goto fail3;
404
405         usb_set_intfdata(intf, hanwang);
406
407         return 0;
408
409  fail3: usb_free_urb(hanwang->irq);
410  fail2: usb_free_coherent(dev, hanwang->features->pkg_len,
411                         hanwang->data, hanwang->data_dma);
412  fail1: input_free_device(input_dev);
413         kfree(hanwang);
414         return error;
415
416 }
417
418 static void hanwang_disconnect(struct usb_interface *intf)
419 {
420         struct hanwang *hanwang = usb_get_intfdata(intf);
421
422         input_unregister_device(hanwang->dev);
423         usb_free_urb(hanwang->irq);
424         usb_free_coherent(interface_to_usbdev(intf),
425                         hanwang->features->pkg_len, hanwang->data,
426                         hanwang->data_dma);
427         kfree(hanwang);
428         usb_set_intfdata(intf, NULL);
429 }
430
431 static const struct usb_device_id hanwang_ids[] = {
432         { HANWANG_TABLET_DEVICE(USB_VENDOR_ID_HANWANG, HANWANG_TABLET_INT_CLASS,
433                 HANWANG_TABLET_INT_SUB_CLASS, HANWANG_TABLET_INT_PROTOCOL) },
434         {}
435 };
436
437 MODULE_DEVICE_TABLE(usb, hanwang_ids);
438
439 static struct usb_driver hanwang_driver = {
440         .name           = "hanwang",
441         .probe          = hanwang_probe,
442         .disconnect     = hanwang_disconnect,
443         .id_table       = hanwang_ids,
444 };
445
446 module_usb_driver(hanwang_driver);