drivers/virtio/virtio_input.c

   1 #include <linux/module.h>
   2 #include <linux/virtio.h>
   3 #include <linux/virtio_config.h>
   4 #include <linux/input.h>
   5
   6 #include <uapi/linux/virtio_ids.h>
   7 #include <uapi/linux/virtio_input.h>
   8
   9 struct virtio_input {
  10         struct virtio_device       *vdev;
  11         struct input_dev           *idev;
  12         char                       name[64];
  13         char                       serial[64];
  14         char                       phys[64];
  15         struct virtqueue           *evt, *sts;
  16         struct virtio_input_event  evts[64];
  17         spinlock_t                 lock;
  18         bool                       ready;
  19 };
  20
  21 static void virtinput_queue_evtbuf(struct virtio_input *vi,
  22                                    struct virtio_input_event *evtbuf)
  23 {
  24         struct scatterlist sg[1];
  25
  26         sg_init_one(sg, evtbuf, sizeof(*evtbuf));
  27         virtqueue_add_inbuf(vi->evt, sg, 1, evtbuf, GFP_ATOMIC);
  28 }
  29
  30 static void virtinput_recv_events(struct virtqueue *vq)
  31 {
  32         struct virtio_input *vi = vq->vdev->priv;
  33         struct virtio_input_event *event;
  34         unsigned long flags;
  35         unsigned int len;
  36
  37         spin_lock_irqsave(&vi->lock, flags);
  38         if (vi->ready) {
  39                 while ((event = virtqueue_get_buf(vi->evt, &len)) != NULL) {
  40                         spin_unlock_irqrestore(&vi->lock, flags);
  41                         input_event(vi->idev,
  42                                     le16_to_cpu(event->type),
  43                                     le16_to_cpu(event->code),
  44                                     le32_to_cpu(event->value));
  45                         spin_lock_irqsave(&vi->lock, flags);
  46                         virtinput_queue_evtbuf(vi, event);
  47                 }
  48                 virtqueue_kick(vq);
  49         }
  50         spin_unlock_irqrestore(&vi->lock, flags);
  51 }
  52
  53 /*
  54  * On error we are losing the status update, which isn't critical as
  55  * this is typically used for stuff like keyboard leds.
  56  */
  57 static int virtinput_send_status(struct virtio_input *vi,
  58                                  u16 type, u16 code, s32 value)
  59 {
  60         struct virtio_input_event *stsbuf;
  61         struct scatterlist sg[1];
  62         unsigned long flags;
  63         int rc;
  64
  65         stsbuf = kzalloc(sizeof(*stsbuf), GFP_ATOMIC);
  66         if (!stsbuf)
  67                 return -ENOMEM;
  68
  69         stsbuf->type  = cpu_to_le16(type);
  70         stsbuf->code  = cpu_to_le16(code);
  71         stsbuf->value = cpu_to_le32(value);
  72         sg_init_one(sg, stsbuf, sizeof(*stsbuf));
  73
  74         spin_lock_irqsave(&vi->lock, flags);
  75         if (vi->ready) {
  76                 rc = virtqueue_add_outbuf(vi->sts, sg, 1, stsbuf, GFP_ATOMIC);
  77                 virtqueue_kick(vi->sts);
  78         } else {
  79                 rc = -ENODEV;
  80         }
  81         spin_unlock_irqrestore(&vi->lock, flags);
  82
  83         if (rc != 0)
  84                 kfree(stsbuf);
  85         return rc;
  86 }
  87
  88 static void virtinput_recv_status(struct virtqueue *vq)
  89 {
  90         struct virtio_input *vi = vq->vdev->priv;
  91         struct virtio_input_event *stsbuf;
  92         unsigned long flags;
  93         unsigned int len;
  94
  95         spin_lock_irqsave(&vi->lock, flags);
  96         while ((stsbuf = virtqueue_get_buf(vi->sts, &len)) != NULL)
  97                 kfree(stsbuf);
  98         spin_unlock_irqrestore(&vi->lock, flags);
  99 }
 100
 101 static int virtinput_status(struct input_dev *idev, unsigned int type,
 102                             unsigned int code, int value)
 103 {
 104         struct virtio_input *vi = input_get_drvdata(idev);
 105
 106         return virtinput_send_status(vi, type, code, value);
 107 }
 108
 109 static u8 virtinput_cfg_select(struct virtio_input *vi,
 110                                u8 select, u8 subsel)
 111 {
 112         u8 size;
 113
 114         virtio_cwrite(vi->vdev, struct virtio_input_config, select, &select);
 115         virtio_cwrite(vi->vdev, struct virtio_input_config, subsel, &subsel);
 116         virtio_cread(vi->vdev, struct virtio_input_config, size, &size);
 117         return size;
 118 }
 119
 120 static void virtinput_cfg_bits(struct virtio_input *vi, int select, int subsel,
 121                                unsigned long *bits, unsigned int bitcount)
 122 {
 123         unsigned int bit;
 124         u8 *virtio_bits;
 125         u8 bytes;
 126
 127         bytes = virtinput_cfg_select(vi, select, subsel);
 128         if (!bytes)
 129                 return;
 130         if (bitcount > bytes * 8)
 131                 bitcount = bytes * 8;
 132
 133         /*
 134          * Bitmap in virtio config space is a simple stream of bytes,
 135          * with the first byte carrying bits 0-7, second bits 8-15 and
 136          * so on.
 137          */
 138         virtio_bits = kzalloc(bytes, GFP_KERNEL);
 139         if (!virtio_bits)
 140                 return;
 141         virtio_cread_bytes(vi->vdev, offsetof(struct virtio_input_config,
 142                                               u.bitmap),
 143                            virtio_bits, bytes);
 144         for (bit = 0; bit < bitcount; bit++) {
 145                 if (virtio_bits[bit / 8] & (1 << (bit % 8)))
 146                         __set_bit(bit, bits);
 147         }
 148         kfree(virtio_bits);
 149
 150         if (select == VIRTIO_INPUT_CFG_EV_BITS)
 151                 __set_bit(subsel, vi->idev->evbit);
 152 }
 153
 154 static void virtinput_cfg_abs(struct virtio_input *vi, int abs)
 155 {
 156         u32 mi, ma, re, fu, fl;
 157
 158         virtinput_cfg_select(vi, VIRTIO_INPUT_CFG_ABS_INFO, abs);
 159         virtio_cread(vi->vdev, struct virtio_input_config, u.abs.min, &mi);
 160         virtio_cread(vi->vdev, struct virtio_input_config, u.abs.max, &ma);
 161         virtio_cread(vi->vdev, struct virtio_input_config, u.abs.res, &re);
 162         virtio_cread(vi->vdev, struct virtio_input_config, u.abs.fuzz, &fu);
 163         virtio_cread(vi->vdev, struct virtio_input_config, u.abs.flat, &fl);
 164         input_set_abs_params(vi->idev, abs, mi, ma, fu, fl);
 165         input_abs_set_res(vi->idev, abs, re);
 166 }
 167
 168 static int virtinput_init_vqs(struct virtio_input *vi)
 169 {
 170         struct virtqueue *vqs[2];
 171         vq_callback_t *cbs[] = { virtinput_recv_events,
 172                                  virtinput_recv_status };
 173         static const char * const names[] = { "events", "status" };
 174         int err;
 175
 176         err = vi->vdev->config->find_vqs(vi->vdev, 2, vqs, cbs, names,
 177                         NULL);
 178         if (err)
 179                 return err;
 180         vi->evt = vqs[0];
 181         vi->sts = vqs[1];
 182
 183         return 0;
 184 }
 185
 186 static void virtinput_fill_evt(struct virtio_input *vi)
 187 {
 188         unsigned long flags;
 189         int i, size;
 190
 191         spin_lock_irqsave(&vi->lock, flags);
 192         size = virtqueue_get_vring_size(vi->evt);
 193         if (size > ARRAY_SIZE(vi->evts))
 194                 size = ARRAY_SIZE(vi->evts);
 195         for (i = 0; i < size; i++)
 196                 virtinput_queue_evtbuf(vi, &vi->evts[i]);
 197         virtqueue_kick(vi->evt);
 198         spin_unlock_irqrestore(&vi->lock, flags);
 199 }
 200
 201 static int virtinput_probe(struct virtio_device *vdev)
 202 {
 203         struct virtio_input *vi;
 204         unsigned long flags;
 205         size_t size;
 206         int abs, err;
 207
 208         if (!virtio_has_feature(vdev, VIRTIO_F_VERSION_1))
 209                 return -ENODEV;
 210
 211         vi = kzalloc(sizeof(*vi), GFP_KERNEL);
 212         if (!vi)
 213                 return -ENOMEM;
 214
 215         vdev->priv = vi;
 216         vi->vdev = vdev;
 217         spin_lock_init(&vi->lock);
 218
 219         err = virtinput_init_vqs(vi);
 220         if (err)
 221                 goto err_init_vq;
 222
 223         vi->idev = input_allocate_device();
 224         if (!vi->idev) {
 225                 err = -ENOMEM;
 226                 goto err_input_alloc;
 227         }
 228         input_set_drvdata(vi->idev, vi);
 229
 230         size = virtinput_cfg_select(vi, VIRTIO_INPUT_CFG_ID_NAME, 0);
 231         virtio_cread_bytes(vi->vdev, offsetof(struct virtio_input_config,
 232                                               u.string),
 233                            vi->name, min(size, sizeof(vi->name)));
 234         size = virtinput_cfg_select(vi, VIRTIO_INPUT_CFG_ID_SERIAL, 0);
 235         virtio_cread_bytes(vi->vdev, offsetof(struct virtio_input_config,
 236                                               u.string),
 237                            vi->serial, min(size, sizeof(vi->serial)));
 238         snprintf(vi->phys, sizeof(vi->phys),
 239                  "virtio%d/input0", vdev->index);
 240         vi->idev->name = vi->name;
 241         vi->idev->phys = vi->phys;
 242         vi->idev->uniq = vi->serial;
 243
 244         size = virtinput_cfg_select(vi, VIRTIO_INPUT_CFG_ID_DEVIDS, 0);
 245         if (size >= sizeof(struct virtio_input_devids)) {
 246                 virtio_cread(vi->vdev, struct virtio_input_config,
 247                              u.ids.bustype, &vi->idev->id.bustype);
 248                 virtio_cread(vi->vdev, struct virtio_input_config,
 249                              u.ids.vendor, &vi->idev->id.vendor);
 250                 virtio_cread(vi->vdev, struct virtio_input_config,
 251                              u.ids.product, &vi->idev->id.product);
 252                 virtio_cread(vi->vdev, struct virtio_input_config,
 253                              u.ids.version, &vi->idev->id.version);
 254         } else {
 255                 vi->idev->id.bustype = BUS_VIRTUAL;
 256         }
 257
 258         virtinput_cfg_bits(vi, VIRTIO_INPUT_CFG_PROP_BITS, 0,
 259                            vi->idev->propbit, INPUT_PROP_CNT);
 260         size = virtinput_cfg_select(vi, VIRTIO_INPUT_CFG_EV_BITS, EV_REP);
 261         if (size)
 262                 __set_bit(EV_REP, vi->idev->evbit);
 263
 264         vi->idev->dev.parent = &vdev->dev;
 265         vi->idev->event = virtinput_status;
 266
 267         /* device -> kernel */
 268         virtinput_cfg_bits(vi, VIRTIO_INPUT_CFG_EV_BITS, EV_KEY,
 269                            vi->idev->keybit, KEY_CNT);
 270         virtinput_cfg_bits(vi, VIRTIO_INPUT_CFG_EV_BITS, EV_REL,
 271                            vi->idev->relbit, REL_CNT);
 272         virtinput_cfg_bits(vi, VIRTIO_INPUT_CFG_EV_BITS, EV_ABS,
 273                            vi->idev->absbit, ABS_CNT);
 274         virtinput_cfg_bits(vi, VIRTIO_INPUT_CFG_EV_BITS, EV_MSC,
 275                            vi->idev->mscbit, MSC_CNT);
 276         virtinput_cfg_bits(vi, VIRTIO_INPUT_CFG_EV_BITS, EV_SW,
 277                            vi->idev->swbit,  SW_CNT);
 278
 279         /* kernel -> device */
 280         virtinput_cfg_bits(vi, VIRTIO_INPUT_CFG_EV_BITS, EV_LED,
 281                            vi->idev->ledbit, LED_CNT);
 282         virtinput_cfg_bits(vi, VIRTIO_INPUT_CFG_EV_BITS, EV_SND,
 283                            vi->idev->sndbit, SND_CNT);
 284
 285         if (test_bit(EV_ABS, vi->idev->evbit)) {
 286                 for (abs = 0; abs < ABS_CNT; abs++) {
 287                         if (!test_bit(abs, vi->idev->absbit))
 288                                 continue;
 289                         virtinput_cfg_abs(vi, abs);
 290                 }
 291         }
 292
 293         virtio_device_ready(vdev);
 294         vi->ready = true;
 295         err = input_register_device(vi->idev);
 296         if (err)
 297                 goto err_input_register;
 298
 299         virtinput_fill_evt(vi);
 300         return 0;
 301
 302 err_input_register:
 303         spin_lock_irqsave(&vi->lock, flags);
 304         vi->ready = false;
 305         spin_unlock_irqrestore(&vi->lock, flags);
 306         input_free_device(vi->idev);
 307 err_input_alloc:
 308         vdev->config->del_vqs(vdev);
 309 err_init_vq:
 310         kfree(vi);
 311         return err;
 312 }
 313
 314 static void virtinput_remove(struct virtio_device *vdev)
 315 {
 316         struct virtio_input *vi = vdev->priv;
 317         void *buf;
 318         unsigned long flags;
 319
 320         spin_lock_irqsave(&vi->lock, flags);
 321         vi->ready = false;
 322         spin_unlock_irqrestore(&vi->lock, flags);
 323
 324         input_unregister_device(vi->idev);
 325         vdev->config->reset(vdev);
 326         while ((buf = virtqueue_detach_unused_buf(vi->sts)) != NULL)
 327                 kfree(buf);
 328         vdev->config->del_vqs(vdev);
 329         kfree(vi);
 330 }
 331
 332 #ifdef CONFIG_PM_SLEEP
 333 static int virtinput_freeze(struct virtio_device *vdev)
 334 {
 335         struct virtio_input *vi = vdev->priv;
 336         unsigned long flags;
 337
 338         spin_lock_irqsave(&vi->lock, flags);
 339         vi->ready = false;
 340         spin_unlock_irqrestore(&vi->lock, flags);
 341
 342         vdev->config->del_vqs(vdev);
 343         return 0;
 344 }
 345
 346 static int virtinput_restore(struct virtio_device *vdev)
 347 {
 348         struct virtio_input *vi = vdev->priv;
 349         int err;
 350
 351         err = virtinput_init_vqs(vi);
 352         if (err)
 353                 return err;
 354
 355         virtio_device_ready(vdev);
 356         vi->ready = true;
 357         virtinput_fill_evt(vi);
 358         return 0;
 359 }
 360 #endif
 361
 362 static unsigned int features[] = {
 363         /* none */
 364 };
 365 static struct virtio_device_id id_table[] = {
 366         { VIRTIO_ID_INPUT, VIRTIO_DEV_ANY_ID },
 367         { 0 },
 368 };
 369
 370 static struct virtio_driver virtio_input_driver = {
 371         .driver.name         = KBUILD_MODNAME,
 372         .driver.owner        = THIS_MODULE,
 373         .feature_table       = features,
 374         .feature_table_size  = ARRAY_SIZE(features),
 375         .id_table            = id_table,
 376         .probe               = virtinput_probe,
 377         .remove              = virtinput_remove,
 378 #ifdef CONFIG_PM_SLEEP
 379         .freeze              = virtinput_freeze,
 380         .restore             = virtinput_restore,
 381 #endif
 382 };
 383
 384 module_virtio_driver(virtio_input_driver);
 385 MODULE_DEVICE_TABLE(virtio, id_table);
 386
 387 MODULE_LICENSE("GPL");
 388 MODULE_DESCRIPTION("Virtio input device driver");
 389 MODULE_AUTHOR("Gerd Hoffmann <kraxel@redhat.com>");