GNU Linux-libre 4.14.266-gnu1
[releases.git] / drivers / input / joydev.c
1 /*
2  * Joystick device driver for the input driver suite.
3  *
4  * Copyright (c) 1999-2002 Vojtech Pavlik
5  * Copyright (c) 1999 Colin Van Dyke
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 2 of the License, or
10  * (at your option) any later version.
11  */
12
13 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
14
15 #include <asm/io.h>
16 #include <linux/delay.h>
17 #include <linux/errno.h>
18 #include <linux/joystick.h>
19 #include <linux/input.h>
20 #include <linux/kernel.h>
21 #include <linux/major.h>
22 #include <linux/sched.h>
23 #include <linux/slab.h>
24 #include <linux/mm.h>
25 #include <linux/module.h>
26 #include <linux/poll.h>
27 #include <linux/init.h>
28 #include <linux/device.h>
29 #include <linux/cdev.h>
30
31 MODULE_AUTHOR("Vojtech Pavlik <vojtech@ucw.cz>");
32 MODULE_DESCRIPTION("Joystick device interfaces");
33 MODULE_SUPPORTED_DEVICE("input/js");
34 MODULE_LICENSE("GPL");
35
36 #define JOYDEV_MINOR_BASE       0
37 #define JOYDEV_MINORS           16
38 #define JOYDEV_BUFFER_SIZE      64
39
40 struct joydev {
41         int open;
42         struct input_handle handle;
43         wait_queue_head_t wait;
44         struct list_head client_list;
45         spinlock_t client_lock; /* protects client_list */
46         struct mutex mutex;
47         struct device dev;
48         struct cdev cdev;
49         bool exist;
50
51         struct js_corr corr[ABS_CNT];
52         struct JS_DATA_SAVE_TYPE glue;
53         int nabs;
54         int nkey;
55         __u16 keymap[KEY_MAX - BTN_MISC + 1];
56         __u16 keypam[KEY_MAX - BTN_MISC + 1];
57         __u8 absmap[ABS_CNT];
58         __u8 abspam[ABS_CNT];
59         __s16 abs[ABS_CNT];
60 };
61
62 struct joydev_client {
63         struct js_event buffer[JOYDEV_BUFFER_SIZE];
64         int head;
65         int tail;
66         int startup;
67         spinlock_t buffer_lock; /* protects access to buffer, head and tail */
68         struct fasync_struct *fasync;
69         struct joydev *joydev;
70         struct list_head node;
71 };
72
73 static int joydev_correct(int value, struct js_corr *corr)
74 {
75         switch (corr->type) {
76
77         case JS_CORR_NONE:
78                 break;
79
80         case JS_CORR_BROKEN:
81                 value = value > corr->coef[0] ? (value < corr->coef[1] ? 0 :
82                         ((corr->coef[3] * (value - corr->coef[1])) >> 14)) :
83                         ((corr->coef[2] * (value - corr->coef[0])) >> 14);
84                 break;
85
86         default:
87                 return 0;
88         }
89
90         return clamp(value, -32767, 32767);
91 }
92
93 static void joydev_pass_event(struct joydev_client *client,
94                               struct js_event *event)
95 {
96         struct joydev *joydev = client->joydev;
97
98         /*
99          * IRQs already disabled, just acquire the lock
100          */
101         spin_lock(&client->buffer_lock);
102
103         client->buffer[client->head] = *event;
104
105         if (client->startup == joydev->nabs + joydev->nkey) {
106                 client->head++;
107                 client->head &= JOYDEV_BUFFER_SIZE - 1;
108                 if (client->tail == client->head)
109                         client->startup = 0;
110         }
111
112         spin_unlock(&client->buffer_lock);
113
114         kill_fasync(&client->fasync, SIGIO, POLL_IN);
115 }
116
117 static void joydev_event(struct input_handle *handle,
118                          unsigned int type, unsigned int code, int value)
119 {
120         struct joydev *joydev = handle->private;
121         struct joydev_client *client;
122         struct js_event event;
123
124         switch (type) {
125
126         case EV_KEY:
127                 if (code < BTN_MISC || value == 2)
128                         return;
129                 event.type = JS_EVENT_BUTTON;
130                 event.number = joydev->keymap[code - BTN_MISC];
131                 event.value = value;
132                 break;
133
134         case EV_ABS:
135                 event.type = JS_EVENT_AXIS;
136                 event.number = joydev->absmap[code];
137                 event.value = joydev_correct(value,
138                                         &joydev->corr[event.number]);
139                 if (event.value == joydev->abs[event.number])
140                         return;
141                 joydev->abs[event.number] = event.value;
142                 break;
143
144         default:
145                 return;
146         }
147
148         event.time = jiffies_to_msecs(jiffies);
149
150         rcu_read_lock();
151         list_for_each_entry_rcu(client, &joydev->client_list, node)
152                 joydev_pass_event(client, &event);
153         rcu_read_unlock();
154
155         wake_up_interruptible(&joydev->wait);
156 }
157
158 static int joydev_fasync(int fd, struct file *file, int on)
159 {
160         struct joydev_client *client = file->private_data;
161
162         return fasync_helper(fd, file, on, &client->fasync);
163 }
164
165 static void joydev_free(struct device *dev)
166 {
167         struct joydev *joydev = container_of(dev, struct joydev, dev);
168
169         input_put_device(joydev->handle.dev);
170         kfree(joydev);
171 }
172
173 static void joydev_attach_client(struct joydev *joydev,
174                                  struct joydev_client *client)
175 {
176         spin_lock(&joydev->client_lock);
177         list_add_tail_rcu(&client->node, &joydev->client_list);
178         spin_unlock(&joydev->client_lock);
179 }
180
181 static void joydev_detach_client(struct joydev *joydev,
182                                  struct joydev_client *client)
183 {
184         spin_lock(&joydev->client_lock);
185         list_del_rcu(&client->node);
186         spin_unlock(&joydev->client_lock);
187         synchronize_rcu();
188 }
189
190 static void joydev_refresh_state(struct joydev *joydev)
191 {
192         struct input_dev *dev = joydev->handle.dev;
193         int i, val;
194
195         for (i = 0; i < joydev->nabs; i++) {
196                 val = input_abs_get_val(dev, joydev->abspam[i]);
197                 joydev->abs[i] = joydev_correct(val, &joydev->corr[i]);
198         }
199 }
200
201 static int joydev_open_device(struct joydev *joydev)
202 {
203         int retval;
204
205         retval = mutex_lock_interruptible(&joydev->mutex);
206         if (retval)
207                 return retval;
208
209         if (!joydev->exist)
210                 retval = -ENODEV;
211         else if (!joydev->open++) {
212                 retval = input_open_device(&joydev->handle);
213                 if (retval)
214                         joydev->open--;
215                 else
216                         joydev_refresh_state(joydev);
217         }
218
219         mutex_unlock(&joydev->mutex);
220         return retval;
221 }
222
223 static void joydev_close_device(struct joydev *joydev)
224 {
225         mutex_lock(&joydev->mutex);
226
227         if (joydev->exist && !--joydev->open)
228                 input_close_device(&joydev->handle);
229
230         mutex_unlock(&joydev->mutex);
231 }
232
233 /*
234  * Wake up users waiting for IO so they can disconnect from
235  * dead device.
236  */
237 static void joydev_hangup(struct joydev *joydev)
238 {
239         struct joydev_client *client;
240
241         spin_lock(&joydev->client_lock);
242         list_for_each_entry(client, &joydev->client_list, node)
243                 kill_fasync(&client->fasync, SIGIO, POLL_HUP);
244         spin_unlock(&joydev->client_lock);
245
246         wake_up_interruptible(&joydev->wait);
247 }
248
249 static int joydev_release(struct inode *inode, struct file *file)
250 {
251         struct joydev_client *client = file->private_data;
252         struct joydev *joydev = client->joydev;
253
254         joydev_detach_client(joydev, client);
255         kfree(client);
256
257         joydev_close_device(joydev);
258
259         return 0;
260 }
261
262 static int joydev_open(struct inode *inode, struct file *file)
263 {
264         struct joydev *joydev =
265                         container_of(inode->i_cdev, struct joydev, cdev);
266         struct joydev_client *client;
267         int error;
268
269         client = kzalloc(sizeof(struct joydev_client), GFP_KERNEL);
270         if (!client)
271                 return -ENOMEM;
272
273         spin_lock_init(&client->buffer_lock);
274         client->joydev = joydev;
275         joydev_attach_client(joydev, client);
276
277         error = joydev_open_device(joydev);
278         if (error)
279                 goto err_free_client;
280
281         file->private_data = client;
282         nonseekable_open(inode, file);
283
284         return 0;
285
286  err_free_client:
287         joydev_detach_client(joydev, client);
288         kfree(client);
289         return error;
290 }
291
292 static int joydev_generate_startup_event(struct joydev_client *client,
293                                          struct input_dev *input,
294                                          struct js_event *event)
295 {
296         struct joydev *joydev = client->joydev;
297         int have_event;
298
299         spin_lock_irq(&client->buffer_lock);
300
301         have_event = client->startup < joydev->nabs + joydev->nkey;
302
303         if (have_event) {
304
305                 event->time = jiffies_to_msecs(jiffies);
306                 if (client->startup < joydev->nkey) {
307                         event->type = JS_EVENT_BUTTON | JS_EVENT_INIT;
308                         event->number = client->startup;
309                         event->value = !!test_bit(joydev->keypam[event->number],
310                                                   input->key);
311                 } else {
312                         event->type = JS_EVENT_AXIS | JS_EVENT_INIT;
313                         event->number = client->startup - joydev->nkey;
314                         event->value = joydev->abs[event->number];
315                 }
316                 client->startup++;
317         }
318
319         spin_unlock_irq(&client->buffer_lock);
320
321         return have_event;
322 }
323
324 static int joydev_fetch_next_event(struct joydev_client *client,
325                                    struct js_event *event)
326 {
327         int have_event;
328
329         spin_lock_irq(&client->buffer_lock);
330
331         have_event = client->head != client->tail;
332         if (have_event) {
333                 *event = client->buffer[client->tail++];
334                 client->tail &= JOYDEV_BUFFER_SIZE - 1;
335         }
336
337         spin_unlock_irq(&client->buffer_lock);
338
339         return have_event;
340 }
341
342 /*
343  * Old joystick interface
344  */
345 static ssize_t joydev_0x_read(struct joydev_client *client,
346                               struct input_dev *input,
347                               char __user *buf)
348 {
349         struct joydev *joydev = client->joydev;
350         struct JS_DATA_TYPE data;
351         int i;
352
353         spin_lock_irq(&input->event_lock);
354
355         /*
356          * Get device state
357          */
358         for (data.buttons = i = 0; i < 32 && i < joydev->nkey; i++)
359                 data.buttons |=
360                         test_bit(joydev->keypam[i], input->key) ? (1 << i) : 0;
361         data.x = (joydev->abs[0] / 256 + 128) >> joydev->glue.JS_CORR.x;
362         data.y = (joydev->abs[1] / 256 + 128) >> joydev->glue.JS_CORR.y;
363
364         /*
365          * Reset reader's event queue
366          */
367         spin_lock(&client->buffer_lock);
368         client->startup = 0;
369         client->tail = client->head;
370         spin_unlock(&client->buffer_lock);
371
372         spin_unlock_irq(&input->event_lock);
373
374         if (copy_to_user(buf, &data, sizeof(struct JS_DATA_TYPE)))
375                 return -EFAULT;
376
377         return sizeof(struct JS_DATA_TYPE);
378 }
379
380 static inline int joydev_data_pending(struct joydev_client *client)
381 {
382         struct joydev *joydev = client->joydev;
383
384         return client->startup < joydev->nabs + joydev->nkey ||
385                 client->head != client->tail;
386 }
387
388 static ssize_t joydev_read(struct file *file, char __user *buf,
389                            size_t count, loff_t *ppos)
390 {
391         struct joydev_client *client = file->private_data;
392         struct joydev *joydev = client->joydev;
393         struct input_dev *input = joydev->handle.dev;
394         struct js_event event;
395         int retval;
396
397         if (!joydev->exist)
398                 return -ENODEV;
399
400         if (count < sizeof(struct js_event))
401                 return -EINVAL;
402
403         if (count == sizeof(struct JS_DATA_TYPE))
404                 return joydev_0x_read(client, input, buf);
405
406         if (!joydev_data_pending(client) && (file->f_flags & O_NONBLOCK))
407                 return -EAGAIN;
408
409         retval = wait_event_interruptible(joydev->wait,
410                         !joydev->exist || joydev_data_pending(client));
411         if (retval)
412                 return retval;
413
414         if (!joydev->exist)
415                 return -ENODEV;
416
417         while (retval + sizeof(struct js_event) <= count &&
418                joydev_generate_startup_event(client, input, &event)) {
419
420                 if (copy_to_user(buf + retval, &event, sizeof(struct js_event)))
421                         return -EFAULT;
422
423                 retval += sizeof(struct js_event);
424         }
425
426         while (retval + sizeof(struct js_event) <= count &&
427                joydev_fetch_next_event(client, &event)) {
428
429                 if (copy_to_user(buf + retval, &event, sizeof(struct js_event)))
430                         return -EFAULT;
431
432                 retval += sizeof(struct js_event);
433         }
434
435         return retval;
436 }
437
438 /* No kernel lock - fine */
439 static unsigned int joydev_poll(struct file *file, poll_table *wait)
440 {
441         struct joydev_client *client = file->private_data;
442         struct joydev *joydev = client->joydev;
443
444         poll_wait(file, &joydev->wait, wait);
445         return (joydev_data_pending(client) ? (POLLIN | POLLRDNORM) : 0) |
446                 (joydev->exist ?  0 : (POLLHUP | POLLERR));
447 }
448
449 static int joydev_handle_JSIOCSAXMAP(struct joydev *joydev,
450                                      void __user *argp, size_t len)
451 {
452         __u8 *abspam;
453         int i;
454         int retval = 0;
455
456         len = min(len, sizeof(joydev->abspam));
457
458         /* Validate the map. */
459         abspam = memdup_user(argp, len);
460         if (IS_ERR(abspam))
461                 return PTR_ERR(abspam);
462
463         for (i = 0; i < len && i < joydev->nabs; i++) {
464                 if (abspam[i] > ABS_MAX) {
465                         retval = -EINVAL;
466                         goto out;
467                 }
468         }
469
470         memcpy(joydev->abspam, abspam, len);
471
472         for (i = 0; i < joydev->nabs; i++)
473                 joydev->absmap[joydev->abspam[i]] = i;
474
475  out:
476         kfree(abspam);
477         return retval;
478 }
479
480 static int joydev_handle_JSIOCSBTNMAP(struct joydev *joydev,
481                                       void __user *argp, size_t len)
482 {
483         __u16 *keypam;
484         int i;
485         int retval = 0;
486
487         if (len % sizeof(*keypam))
488                 return -EINVAL;
489
490         len = min(len, sizeof(joydev->keypam));
491
492         /* Validate the map. */
493         keypam = memdup_user(argp, len);
494         if (IS_ERR(keypam))
495                 return PTR_ERR(keypam);
496
497         for (i = 0; i < (len / 2) && i < joydev->nkey; i++) {
498                 if (keypam[i] > KEY_MAX || keypam[i] < BTN_MISC) {
499                         retval = -EINVAL;
500                         goto out;
501                 }
502         }
503
504         memcpy(joydev->keypam, keypam, len);
505
506         for (i = 0; i < joydev->nkey; i++)
507                 joydev->keymap[joydev->keypam[i] - BTN_MISC] = i;
508
509  out:
510         kfree(keypam);
511         return retval;
512 }
513
514
515 static int joydev_ioctl_common(struct joydev *joydev,
516                                 unsigned int cmd, void __user *argp)
517 {
518         struct input_dev *dev = joydev->handle.dev;
519         size_t len;
520         int i;
521         const char *name;
522
523         /* Process fixed-sized commands. */
524         switch (cmd) {
525
526         case JS_SET_CAL:
527                 return copy_from_user(&joydev->glue.JS_CORR, argp,
528                                 sizeof(joydev->glue.JS_CORR)) ? -EFAULT : 0;
529
530         case JS_GET_CAL:
531                 return copy_to_user(argp, &joydev->glue.JS_CORR,
532                                 sizeof(joydev->glue.JS_CORR)) ? -EFAULT : 0;
533
534         case JS_SET_TIMEOUT:
535                 return get_user(joydev->glue.JS_TIMEOUT, (s32 __user *) argp);
536
537         case JS_GET_TIMEOUT:
538                 return put_user(joydev->glue.JS_TIMEOUT, (s32 __user *) argp);
539
540         case JSIOCGVERSION:
541                 return put_user(JS_VERSION, (__u32 __user *) argp);
542
543         case JSIOCGAXES:
544                 return put_user(joydev->nabs, (__u8 __user *) argp);
545
546         case JSIOCGBUTTONS:
547                 return put_user(joydev->nkey, (__u8 __user *) argp);
548
549         case JSIOCSCORR:
550                 if (copy_from_user(joydev->corr, argp,
551                               sizeof(joydev->corr[0]) * joydev->nabs))
552                         return -EFAULT;
553
554                 for (i = 0; i < joydev->nabs; i++) {
555                         int val = input_abs_get_val(dev, joydev->abspam[i]);
556                         joydev->abs[i] = joydev_correct(val, &joydev->corr[i]);
557                 }
558                 return 0;
559
560         case JSIOCGCORR:
561                 return copy_to_user(argp, joydev->corr,
562                         sizeof(joydev->corr[0]) * joydev->nabs) ? -EFAULT : 0;
563
564         }
565
566         /*
567          * Process variable-sized commands (the axis and button map commands
568          * are considered variable-sized to decouple them from the values of
569          * ABS_MAX and KEY_MAX).
570          */
571         switch (cmd & ~IOCSIZE_MASK) {
572
573         case (JSIOCSAXMAP & ~IOCSIZE_MASK):
574                 return joydev_handle_JSIOCSAXMAP(joydev, argp, _IOC_SIZE(cmd));
575
576         case (JSIOCGAXMAP & ~IOCSIZE_MASK):
577                 len = min_t(size_t, _IOC_SIZE(cmd), sizeof(joydev->abspam));
578                 return copy_to_user(argp, joydev->abspam, len) ? -EFAULT : len;
579
580         case (JSIOCSBTNMAP & ~IOCSIZE_MASK):
581                 return joydev_handle_JSIOCSBTNMAP(joydev, argp, _IOC_SIZE(cmd));
582
583         case (JSIOCGBTNMAP & ~IOCSIZE_MASK):
584                 len = min_t(size_t, _IOC_SIZE(cmd), sizeof(joydev->keypam));
585                 return copy_to_user(argp, joydev->keypam, len) ? -EFAULT : len;
586
587         case JSIOCGNAME(0):
588                 name = dev->name;
589                 if (!name)
590                         return 0;
591
592                 len = min_t(size_t, _IOC_SIZE(cmd), strlen(name) + 1);
593                 return copy_to_user(argp, name, len) ? -EFAULT : len;
594         }
595
596         return -EINVAL;
597 }
598
599 #ifdef CONFIG_COMPAT
600 static long joydev_compat_ioctl(struct file *file,
601                                 unsigned int cmd, unsigned long arg)
602 {
603         struct joydev_client *client = file->private_data;
604         struct joydev *joydev = client->joydev;
605         void __user *argp = (void __user *)arg;
606         s32 tmp32;
607         struct JS_DATA_SAVE_TYPE_32 ds32;
608         int retval;
609
610         retval = mutex_lock_interruptible(&joydev->mutex);
611         if (retval)
612                 return retval;
613
614         if (!joydev->exist) {
615                 retval = -ENODEV;
616                 goto out;
617         }
618
619         switch (cmd) {
620
621         case JS_SET_TIMELIMIT:
622                 retval = get_user(tmp32, (s32 __user *) arg);
623                 if (retval == 0)
624                         joydev->glue.JS_TIMELIMIT = tmp32;
625                 break;
626
627         case JS_GET_TIMELIMIT:
628                 tmp32 = joydev->glue.JS_TIMELIMIT;
629                 retval = put_user(tmp32, (s32 __user *) arg);
630                 break;
631
632         case JS_SET_ALL:
633                 retval = copy_from_user(&ds32, argp,
634                                         sizeof(ds32)) ? -EFAULT : 0;
635                 if (retval == 0) {
636                         joydev->glue.JS_TIMEOUT    = ds32.JS_TIMEOUT;
637                         joydev->glue.BUSY          = ds32.BUSY;
638                         joydev->glue.JS_EXPIRETIME = ds32.JS_EXPIRETIME;
639                         joydev->glue.JS_TIMELIMIT  = ds32.JS_TIMELIMIT;
640                         joydev->glue.JS_SAVE       = ds32.JS_SAVE;
641                         joydev->glue.JS_CORR       = ds32.JS_CORR;
642                 }
643                 break;
644
645         case JS_GET_ALL:
646                 ds32.JS_TIMEOUT    = joydev->glue.JS_TIMEOUT;
647                 ds32.BUSY          = joydev->glue.BUSY;
648                 ds32.JS_EXPIRETIME = joydev->glue.JS_EXPIRETIME;
649                 ds32.JS_TIMELIMIT  = joydev->glue.JS_TIMELIMIT;
650                 ds32.JS_SAVE       = joydev->glue.JS_SAVE;
651                 ds32.JS_CORR       = joydev->glue.JS_CORR;
652
653                 retval = copy_to_user(argp, &ds32, sizeof(ds32)) ? -EFAULT : 0;
654                 break;
655
656         default:
657                 retval = joydev_ioctl_common(joydev, cmd, argp);
658                 break;
659         }
660
661  out:
662         mutex_unlock(&joydev->mutex);
663         return retval;
664 }
665 #endif /* CONFIG_COMPAT */
666
667 static long joydev_ioctl(struct file *file,
668                          unsigned int cmd, unsigned long arg)
669 {
670         struct joydev_client *client = file->private_data;
671         struct joydev *joydev = client->joydev;
672         void __user *argp = (void __user *)arg;
673         int retval;
674
675         retval = mutex_lock_interruptible(&joydev->mutex);
676         if (retval)
677                 return retval;
678
679         if (!joydev->exist) {
680                 retval = -ENODEV;
681                 goto out;
682         }
683
684         switch (cmd) {
685
686         case JS_SET_TIMELIMIT:
687                 retval = get_user(joydev->glue.JS_TIMELIMIT,
688                                   (long __user *) arg);
689                 break;
690
691         case JS_GET_TIMELIMIT:
692                 retval = put_user(joydev->glue.JS_TIMELIMIT,
693                                   (long __user *) arg);
694                 break;
695
696         case JS_SET_ALL:
697                 retval = copy_from_user(&joydev->glue, argp,
698                                         sizeof(joydev->glue)) ? -EFAULT : 0;
699                 break;
700
701         case JS_GET_ALL:
702                 retval = copy_to_user(argp, &joydev->glue,
703                                       sizeof(joydev->glue)) ? -EFAULT : 0;
704                 break;
705
706         default:
707                 retval = joydev_ioctl_common(joydev, cmd, argp);
708                 break;
709         }
710  out:
711         mutex_unlock(&joydev->mutex);
712         return retval;
713 }
714
715 static const struct file_operations joydev_fops = {
716         .owner          = THIS_MODULE,
717         .read           = joydev_read,
718         .poll           = joydev_poll,
719         .open           = joydev_open,
720         .release        = joydev_release,
721         .unlocked_ioctl = joydev_ioctl,
722 #ifdef CONFIG_COMPAT
723         .compat_ioctl   = joydev_compat_ioctl,
724 #endif
725         .fasync         = joydev_fasync,
726         .llseek         = no_llseek,
727 };
728
729 /*
730  * Mark device non-existent. This disables writes, ioctls and
731  * prevents new users from opening the device. Already posted
732  * blocking reads will stay, however new ones will fail.
733  */
734 static void joydev_mark_dead(struct joydev *joydev)
735 {
736         mutex_lock(&joydev->mutex);
737         joydev->exist = false;
738         mutex_unlock(&joydev->mutex);
739 }
740
741 static void joydev_cleanup(struct joydev *joydev)
742 {
743         struct input_handle *handle = &joydev->handle;
744
745         joydev_mark_dead(joydev);
746         joydev_hangup(joydev);
747
748         /* joydev is marked dead so no one else accesses joydev->open */
749         if (joydev->open)
750                 input_close_device(handle);
751 }
752
753 /*
754  * These codes are copied from from hid-ids.h, unfortunately there is no common
755  * usb_ids/bt_ids.h header.
756  */
757 #define USB_VENDOR_ID_SONY                      0x054c
758 #define USB_DEVICE_ID_SONY_PS3_CONTROLLER               0x0268
759 #define USB_DEVICE_ID_SONY_PS4_CONTROLLER               0x05c4
760 #define USB_DEVICE_ID_SONY_PS4_CONTROLLER_2             0x09cc
761 #define USB_DEVICE_ID_SONY_PS4_CONTROLLER_DONGLE        0x0ba0
762
763 #define USB_VENDOR_ID_THQ                       0x20d6
764 #define USB_DEVICE_ID_THQ_PS3_UDRAW                     0xcb17
765
766 #define ACCEL_DEV(vnd, prd)                                             \
767         {                                                               \
768                 .flags = INPUT_DEVICE_ID_MATCH_VENDOR |                 \
769                                 INPUT_DEVICE_ID_MATCH_PRODUCT |         \
770                                 INPUT_DEVICE_ID_MATCH_PROPBIT,          \
771                 .vendor = (vnd),                                        \
772                 .product = (prd),                                       \
773                 .propbit = { BIT_MASK(INPUT_PROP_ACCELEROMETER) },      \
774         }
775
776 static const struct input_device_id joydev_blacklist[] = {
777         /* Avoid touchpads and touchscreens */
778         {
779                 .flags = INPUT_DEVICE_ID_MATCH_EVBIT |
780                                 INPUT_DEVICE_ID_MATCH_KEYBIT,
781                 .evbit = { BIT_MASK(EV_KEY) },
782                 .keybit = { [BIT_WORD(BTN_TOUCH)] = BIT_MASK(BTN_TOUCH) },
783         },
784         /* Avoid tablets, digitisers and similar devices */
785         {
786                 .flags = INPUT_DEVICE_ID_MATCH_EVBIT |
787                                 INPUT_DEVICE_ID_MATCH_KEYBIT,
788                 .evbit = { BIT_MASK(EV_KEY) },
789                 .keybit = { [BIT_WORD(BTN_DIGI)] = BIT_MASK(BTN_DIGI) },
790         },
791         /* Disable accelerometers on composite devices */
792         ACCEL_DEV(USB_VENDOR_ID_SONY, USB_DEVICE_ID_SONY_PS3_CONTROLLER),
793         ACCEL_DEV(USB_VENDOR_ID_SONY, USB_DEVICE_ID_SONY_PS4_CONTROLLER),
794         ACCEL_DEV(USB_VENDOR_ID_SONY, USB_DEVICE_ID_SONY_PS4_CONTROLLER_2),
795         ACCEL_DEV(USB_VENDOR_ID_SONY, USB_DEVICE_ID_SONY_PS4_CONTROLLER_DONGLE),
796         ACCEL_DEV(USB_VENDOR_ID_THQ, USB_DEVICE_ID_THQ_PS3_UDRAW),
797         { /* sentinel */ }
798 };
799
800 static bool joydev_dev_is_blacklisted(struct input_dev *dev)
801 {
802         const struct input_device_id *id;
803
804         for (id = joydev_blacklist; id->flags; id++) {
805                 if (input_match_device_id(dev, id)) {
806                         dev_dbg(&dev->dev,
807                                 "joydev: blacklisting '%s'\n", dev->name);
808                         return true;
809                 }
810         }
811
812         return false;
813 }
814
815 static bool joydev_dev_is_absolute_mouse(struct input_dev *dev)
816 {
817         DECLARE_BITMAP(jd_scratch, KEY_CNT);
818
819         BUILD_BUG_ON(ABS_CNT > KEY_CNT || EV_CNT > KEY_CNT);
820
821         /*
822          * Virtualization (VMware, etc) and remote management (HP
823          * ILO2) solutions use absolute coordinates for their virtual
824          * pointing devices so that there is one-to-one relationship
825          * between pointer position on the host screen and virtual
826          * guest screen, and so their mice use ABS_X, ABS_Y and 3
827          * primary button events. This clashes with what joydev
828          * considers to be joysticks (a device with at minimum ABS_X
829          * axis).
830          *
831          * Here we are trying to separate absolute mice from
832          * joysticks. A device is, for joystick detection purposes,
833          * considered to be an absolute mouse if the following is
834          * true:
835          *
836          * 1) Event types are exactly EV_ABS, EV_KEY and EV_SYN.
837          * 2) Absolute events are exactly ABS_X and ABS_Y.
838          * 3) Keys are exactly BTN_LEFT, BTN_RIGHT and BTN_MIDDLE.
839          * 4) Device is not on "Amiga" bus.
840          */
841
842         bitmap_zero(jd_scratch, EV_CNT);
843         __set_bit(EV_ABS, jd_scratch);
844         __set_bit(EV_KEY, jd_scratch);
845         __set_bit(EV_SYN, jd_scratch);
846         if (!bitmap_equal(jd_scratch, dev->evbit, EV_CNT))
847                 return false;
848
849         bitmap_zero(jd_scratch, ABS_CNT);
850         __set_bit(ABS_X, jd_scratch);
851         __set_bit(ABS_Y, jd_scratch);
852         if (!bitmap_equal(dev->absbit, jd_scratch, ABS_CNT))
853                 return false;
854
855         bitmap_zero(jd_scratch, KEY_CNT);
856         __set_bit(BTN_LEFT, jd_scratch);
857         __set_bit(BTN_RIGHT, jd_scratch);
858         __set_bit(BTN_MIDDLE, jd_scratch);
859
860         if (!bitmap_equal(dev->keybit, jd_scratch, KEY_CNT))
861                 return false;
862
863         /*
864          * Amiga joystick (amijoy) historically uses left/middle/right
865          * button events.
866          */
867         if (dev->id.bustype == BUS_AMIGA)
868                 return false;
869
870         return true;
871 }
872
873 static bool joydev_match(struct input_handler *handler, struct input_dev *dev)
874 {
875         /* Disable blacklisted devices */
876         if (joydev_dev_is_blacklisted(dev))
877                 return false;
878
879         /* Avoid absolute mice */
880         if (joydev_dev_is_absolute_mouse(dev))
881                 return false;
882
883         return true;
884 }
885
886 static int joydev_connect(struct input_handler *handler, struct input_dev *dev,
887                           const struct input_device_id *id)
888 {
889         struct joydev *joydev;
890         int i, j, t, minor, dev_no;
891         int error;
892
893         minor = input_get_new_minor(JOYDEV_MINOR_BASE, JOYDEV_MINORS, true);
894         if (minor < 0) {
895                 error = minor;
896                 pr_err("failed to reserve new minor: %d\n", error);
897                 return error;
898         }
899
900         joydev = kzalloc(sizeof(struct joydev), GFP_KERNEL);
901         if (!joydev) {
902                 error = -ENOMEM;
903                 goto err_free_minor;
904         }
905
906         INIT_LIST_HEAD(&joydev->client_list);
907         spin_lock_init(&joydev->client_lock);
908         mutex_init(&joydev->mutex);
909         init_waitqueue_head(&joydev->wait);
910         joydev->exist = true;
911
912         dev_no = minor;
913         /* Normalize device number if it falls into legacy range */
914         if (dev_no < JOYDEV_MINOR_BASE + JOYDEV_MINORS)
915                 dev_no -= JOYDEV_MINOR_BASE;
916         dev_set_name(&joydev->dev, "js%d", dev_no);
917
918         joydev->handle.dev = input_get_device(dev);
919         joydev->handle.name = dev_name(&joydev->dev);
920         joydev->handle.handler = handler;
921         joydev->handle.private = joydev;
922
923         for_each_set_bit(i, dev->absbit, ABS_CNT) {
924                 joydev->absmap[i] = joydev->nabs;
925                 joydev->abspam[joydev->nabs] = i;
926                 joydev->nabs++;
927         }
928
929         for (i = BTN_JOYSTICK - BTN_MISC; i < KEY_MAX - BTN_MISC + 1; i++)
930                 if (test_bit(i + BTN_MISC, dev->keybit)) {
931                         joydev->keymap[i] = joydev->nkey;
932                         joydev->keypam[joydev->nkey] = i + BTN_MISC;
933                         joydev->nkey++;
934                 }
935
936         for (i = 0; i < BTN_JOYSTICK - BTN_MISC; i++)
937                 if (test_bit(i + BTN_MISC, dev->keybit)) {
938                         joydev->keymap[i] = joydev->nkey;
939                         joydev->keypam[joydev->nkey] = i + BTN_MISC;
940                         joydev->nkey++;
941                 }
942
943         for (i = 0; i < joydev->nabs; i++) {
944                 j = joydev->abspam[i];
945                 if (input_abs_get_max(dev, j) == input_abs_get_min(dev, j)) {
946                         joydev->corr[i].type = JS_CORR_NONE;
947                         continue;
948                 }
949                 joydev->corr[i].type = JS_CORR_BROKEN;
950                 joydev->corr[i].prec = input_abs_get_fuzz(dev, j);
951
952                 t = (input_abs_get_max(dev, j) + input_abs_get_min(dev, j)) / 2;
953                 joydev->corr[i].coef[0] = t - input_abs_get_flat(dev, j);
954                 joydev->corr[i].coef[1] = t + input_abs_get_flat(dev, j);
955
956                 t = (input_abs_get_max(dev, j) - input_abs_get_min(dev, j)) / 2
957                         - 2 * input_abs_get_flat(dev, j);
958                 if (t) {
959                         joydev->corr[i].coef[2] = (1 << 29) / t;
960                         joydev->corr[i].coef[3] = (1 << 29) / t;
961                 }
962         }
963
964         joydev->dev.devt = MKDEV(INPUT_MAJOR, minor);
965         joydev->dev.class = &input_class;
966         joydev->dev.parent = &dev->dev;
967         joydev->dev.release = joydev_free;
968         device_initialize(&joydev->dev);
969
970         error = input_register_handle(&joydev->handle);
971         if (error)
972                 goto err_free_joydev;
973
974         cdev_init(&joydev->cdev, &joydev_fops);
975
976         error = cdev_device_add(&joydev->cdev, &joydev->dev);
977         if (error)
978                 goto err_cleanup_joydev;
979
980         return 0;
981
982  err_cleanup_joydev:
983         joydev_cleanup(joydev);
984         input_unregister_handle(&joydev->handle);
985  err_free_joydev:
986         put_device(&joydev->dev);
987  err_free_minor:
988         input_free_minor(minor);
989         return error;
990 }
991
992 static void joydev_disconnect(struct input_handle *handle)
993 {
994         struct joydev *joydev = handle->private;
995
996         cdev_device_del(&joydev->cdev, &joydev->dev);
997         joydev_cleanup(joydev);
998         input_free_minor(MINOR(joydev->dev.devt));
999         input_unregister_handle(handle);
1000         put_device(&joydev->dev);
1001 }
1002
1003 static const struct input_device_id joydev_ids[] = {
1004         {
1005                 .flags = INPUT_DEVICE_ID_MATCH_EVBIT |
1006                                 INPUT_DEVICE_ID_MATCH_ABSBIT,
1007                 .evbit = { BIT_MASK(EV_ABS) },
1008                 .absbit = { BIT_MASK(ABS_X) },
1009         },
1010         {
1011                 .flags = INPUT_DEVICE_ID_MATCH_EVBIT |
1012                                 INPUT_DEVICE_ID_MATCH_ABSBIT,
1013                 .evbit = { BIT_MASK(EV_ABS) },
1014                 .absbit = { BIT_MASK(ABS_Z) },
1015         },
1016         {
1017                 .flags = INPUT_DEVICE_ID_MATCH_EVBIT |
1018                                 INPUT_DEVICE_ID_MATCH_ABSBIT,
1019                 .evbit = { BIT_MASK(EV_ABS) },
1020                 .absbit = { BIT_MASK(ABS_WHEEL) },
1021         },
1022         {
1023                 .flags = INPUT_DEVICE_ID_MATCH_EVBIT |
1024                                 INPUT_DEVICE_ID_MATCH_ABSBIT,
1025                 .evbit = { BIT_MASK(EV_ABS) },
1026                 .absbit = { BIT_MASK(ABS_THROTTLE) },
1027         },
1028         {
1029                 .flags = INPUT_DEVICE_ID_MATCH_EVBIT |
1030                                 INPUT_DEVICE_ID_MATCH_KEYBIT,
1031                 .evbit = { BIT_MASK(EV_KEY) },
1032                 .keybit = {[BIT_WORD(BTN_JOYSTICK)] = BIT_MASK(BTN_JOYSTICK) },
1033         },
1034         {
1035                 .flags = INPUT_DEVICE_ID_MATCH_EVBIT |
1036                                 INPUT_DEVICE_ID_MATCH_KEYBIT,
1037                 .evbit = { BIT_MASK(EV_KEY) },
1038                 .keybit = { [BIT_WORD(BTN_GAMEPAD)] = BIT_MASK(BTN_GAMEPAD) },
1039         },
1040         {
1041                 .flags = INPUT_DEVICE_ID_MATCH_EVBIT |
1042                                 INPUT_DEVICE_ID_MATCH_KEYBIT,
1043                 .evbit = { BIT_MASK(EV_KEY) },
1044                 .keybit = { [BIT_WORD(BTN_TRIGGER_HAPPY)] = BIT_MASK(BTN_TRIGGER_HAPPY) },
1045         },
1046         { }     /* Terminating entry */
1047 };
1048
1049 MODULE_DEVICE_TABLE(input, joydev_ids);
1050
1051 static struct input_handler joydev_handler = {
1052         .event          = joydev_event,
1053         .match          = joydev_match,
1054         .connect        = joydev_connect,
1055         .disconnect     = joydev_disconnect,
1056         .legacy_minors  = true,
1057         .minor          = JOYDEV_MINOR_BASE,
1058         .name           = "joydev",
1059         .id_table       = joydev_ids,
1060 };
1061
1062 static int __init joydev_init(void)
1063 {
1064         return input_register_handler(&joydev_handler);
1065 }
1066
1067 static void __exit joydev_exit(void)
1068 {
1069         input_unregister_handler(&joydev_handler);
1070 }
1071
1072 module_init(joydev_init);
1073 module_exit(joydev_exit);