| @@ -0,0 +1,384 @@ |
| 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 *names[] = { "events", "status" }; |
| 174 | + int err; |
| 175 | + |
| 176 | + err = vi->vdev->config->find_vqs(vi->vdev, 2, vqs, cbs, names); |
| 177 | + if (err) |
| 178 | + return err; |
| 179 | + vi->evt = vqs[0]; |
| 180 | + vi->sts = vqs[1]; |
| 181 | + |
| 182 | + return 0; |
| 183 | +} |
| 184 | + |
| 185 | +static void virtinput_fill_evt(struct virtio_input *vi) |
| 186 | +{ |
| 187 | + unsigned long flags; |
| 188 | + int i, size; |
| 189 | + |
| 190 | + spin_lock_irqsave(&vi->lock, flags); |
| 191 | + size = virtqueue_get_vring_size(vi->evt); |
| 192 | + if (size > ARRAY_SIZE(vi->evts)) |
| 193 | + size = ARRAY_SIZE(vi->evts); |
| 194 | + for (i = 0; i < size; i++) |
| 195 | + virtinput_queue_evtbuf(vi, &vi->evts[i]); |
| 196 | + virtqueue_kick(vi->evt); |
| 197 | + spin_unlock_irqrestore(&vi->lock, flags); |
| 198 | +} |
| 199 | + |
| 200 | +static int virtinput_probe(struct virtio_device *vdev) |
| 201 | +{ |
| 202 | + struct virtio_input *vi; |
| 203 | + unsigned long flags; |
| 204 | + size_t size; |
| 205 | + int abs, err; |
| 206 | + |
| 207 | + if (!virtio_has_feature(vdev, VIRTIO_F_VERSION_1)) |
| 208 | + return -ENODEV; |
| 209 | + |
| 210 | + vi = kzalloc(sizeof(*vi), GFP_KERNEL); |
| 211 | + if (!vi) |
| 212 | + return -ENOMEM; |
| 213 | + |
| 214 | + vdev->priv = vi; |
| 215 | + vi->vdev = vdev; |
| 216 | + spin_lock_init(&vi->lock); |
| 217 | + |
| 218 | + err = virtinput_init_vqs(vi); |
| 219 | + if (err) |
| 220 | + goto err_init_vq; |
| 221 | + |
| 222 | + vi->idev = input_allocate_device(); |
| 223 | + if (!vi->idev) { |
| 224 | + err = -ENOMEM; |
| 225 | + goto err_input_alloc; |
| 226 | + } |
| 227 | + input_set_drvdata(vi->idev, vi); |
| 228 | + |
| 229 | + size = virtinput_cfg_select(vi, VIRTIO_INPUT_CFG_ID_NAME, 0); |
| 230 | + virtio_cread_bytes(vi->vdev, offsetof(struct virtio_input_config, |
| 231 | + u.string), |
| 232 | + vi->name, min(size, sizeof(vi->name))); |
| 233 | + size = virtinput_cfg_select(vi, VIRTIO_INPUT_CFG_ID_SERIAL, 0); |
| 234 | + virtio_cread_bytes(vi->vdev, offsetof(struct virtio_input_config, |
| 235 | + u.string), |
| 236 | + vi->serial, min(size, sizeof(vi->serial))); |
| 237 | + snprintf(vi->phys, sizeof(vi->phys), |
| 238 | + "virtio%d/input0", vdev->index); |
| 239 | + vi->idev->name = vi->name; |
| 240 | + vi->idev->phys = vi->phys; |
| 241 | + vi->idev->uniq = vi->serial; |
| 242 | + |
| 243 | + size = virtinput_cfg_select(vi, VIRTIO_INPUT_CFG_ID_DEVIDS, 0); |
| 244 | + if (size >= sizeof(struct virtio_input_devids)) { |
| 245 | + virtio_cread(vi->vdev, struct virtio_input_config, |
| 246 | + u.ids.bustype, &vi->idev->id.bustype); |
| 247 | + virtio_cread(vi->vdev, struct virtio_input_config, |
| 248 | + u.ids.vendor, &vi->idev->id.vendor); |
| 249 | + virtio_cread(vi->vdev, struct virtio_input_config, |
| 250 | + u.ids.product, &vi->idev->id.product); |
| 251 | + virtio_cread(vi->vdev, struct virtio_input_config, |
| 252 | + u.ids.version, &vi->idev->id.version); |
| 253 | + } else { |
| 254 | + vi->idev->id.bustype = BUS_VIRTUAL; |
| 255 | + } |
| 256 | + |
| 257 | + virtinput_cfg_bits(vi, VIRTIO_INPUT_CFG_PROP_BITS, 0, |
| 258 | + vi->idev->propbit, INPUT_PROP_CNT); |
| 259 | + size = virtinput_cfg_select(vi, VIRTIO_INPUT_CFG_EV_BITS, EV_REP); |
| 260 | + if (size) |
| 261 | + __set_bit(EV_REP, vi->idev->evbit); |
| 262 | + |
| 263 | + vi->idev->dev.parent = &vdev->dev; |
| 264 | + vi->idev->event = virtinput_status; |
| 265 | + |
| 266 | + /* device -> kernel */ |
| 267 | + virtinput_cfg_bits(vi, VIRTIO_INPUT_CFG_EV_BITS, EV_KEY, |
| 268 | + vi->idev->keybit, KEY_CNT); |
| 269 | + virtinput_cfg_bits(vi, VIRTIO_INPUT_CFG_EV_BITS, EV_REL, |
| 270 | + vi->idev->relbit, REL_CNT); |
| 271 | + virtinput_cfg_bits(vi, VIRTIO_INPUT_CFG_EV_BITS, EV_ABS, |
| 272 | + vi->idev->absbit, ABS_CNT); |
| 273 | + virtinput_cfg_bits(vi, VIRTIO_INPUT_CFG_EV_BITS, EV_MSC, |
| 274 | + vi->idev->mscbit, MSC_CNT); |
| 275 | + virtinput_cfg_bits(vi, VIRTIO_INPUT_CFG_EV_BITS, EV_SW, |
| 276 | + vi->idev->swbit, SW_CNT); |
| 277 | + |
| 278 | + /* kernel -> device */ |
| 279 | + virtinput_cfg_bits(vi, VIRTIO_INPUT_CFG_EV_BITS, EV_LED, |
| 280 | + vi->idev->ledbit, LED_CNT); |
| 281 | + virtinput_cfg_bits(vi, VIRTIO_INPUT_CFG_EV_BITS, EV_SND, |
| 282 | + vi->idev->sndbit, SND_CNT); |
| 283 | + |
| 284 | + if (test_bit(EV_ABS, vi->idev->evbit)) { |
| 285 | + for (abs = 0; abs < ABS_CNT; abs++) { |
| 286 | + if (!test_bit(abs, vi->idev->absbit)) |
| 287 | + continue; |
| 288 | + virtinput_cfg_abs(vi, abs); |
| 289 | + } |
| 290 | + } |
| 291 | + |
| 292 | + virtio_device_ready(vdev); |
| 293 | + vi->ready = true; |
| 294 | + err = input_register_device(vi->idev); |
| 295 | + if (err) |
| 296 | + goto err_input_register; |
| 297 | + |
| 298 | + virtinput_fill_evt(vi); |
| 299 | + return 0; |
| 300 | + |
| 301 | +err_input_register: |
| 302 | + spin_lock_irqsave(&vi->lock, flags); |
| 303 | + vi->ready = false; |
| 304 | + spin_unlock_irqrestore(&vi->lock, flags); |
| 305 | + input_free_device(vi->idev); |
| 306 | +err_input_alloc: |
| 307 | + vdev->config->del_vqs(vdev); |
| 308 | +err_init_vq: |
| 309 | + kfree(vi); |
| 310 | + return err; |
| 311 | +} |
| 312 | + |
| 313 | +static void virtinput_remove(struct virtio_device *vdev) |
| 314 | +{ |
| 315 | + struct virtio_input *vi = vdev->priv; |
| 316 | + unsigned long flags; |
| 317 | + |
| 318 | + spin_lock_irqsave(&vi->lock, flags); |
| 319 | + vi->ready = false; |
| 320 | + spin_unlock_irqrestore(&vi->lock, flags); |
| 321 | + |
| 322 | + input_unregister_device(vi->idev); |
| 323 | + vdev->config->del_vqs(vdev); |
| 324 | + kfree(vi); |
| 325 | +} |
| 326 | + |
| 327 | +#ifdef CONFIG_PM_SLEEP |
| 328 | +static int virtinput_freeze(struct virtio_device *vdev) |
| 329 | +{ |
| 330 | + struct virtio_input *vi = vdev->priv; |
| 331 | + unsigned long flags; |
| 332 | + |
| 333 | + spin_lock_irqsave(&vi->lock, flags); |
| 334 | + vi->ready = false; |
| 335 | + spin_unlock_irqrestore(&vi->lock, flags); |
| 336 | + |
| 337 | + vdev->config->del_vqs(vdev); |
| 338 | + return 0; |
| 339 | +} |
| 340 | + |
| 341 | +static int virtinput_restore(struct virtio_device *vdev) |
| 342 | +{ |
| 343 | + struct virtio_input *vi = vdev->priv; |
| 344 | + int err; |
| 345 | + |
| 346 | + err = virtinput_init_vqs(vi); |
| 347 | + if (err) |
| 348 | + return err; |
| 349 | + |
| 350 | + virtio_device_ready(vdev); |
| 351 | + vi->ready = true; |
| 352 | + virtinput_fill_evt(vi); |
| 353 | + return 0; |
| 354 | +} |
| 355 | +#endif |
| 356 | + |
| 357 | +static unsigned int features[] = { |
| 358 | + /* none */ |
| 359 | +}; |
| 360 | +static struct virtio_device_id id_table[] = { |
| 361 | + { VIRTIO_ID_INPUT, VIRTIO_DEV_ANY_ID }, |
| 362 | + { 0 }, |
| 363 | +}; |
| 364 | + |
| 365 | +static struct virtio_driver virtio_input_driver = { |
| 366 | + .driver.name = KBUILD_MODNAME, |
| 367 | + .driver.owner = THIS_MODULE, |
| 368 | + .feature_table = features, |
| 369 | + .feature_table_size = ARRAY_SIZE(features), |
| 370 | + .id_table = id_table, |
| 371 | + .probe = virtinput_probe, |
| 372 | + .remove = virtinput_remove, |
| 373 | +#ifdef CONFIG_PM_SLEEP |
| 374 | + .freeze = virtinput_freeze, |
| 375 | + .restore = virtinput_restore, |
| 376 | +#endif |
| 377 | +}; |
| 378 | + |
| 379 | +module_virtio_driver(virtio_input_driver); |
| 380 | +MODULE_DEVICE_TABLE(virtio, id_table); |
| 381 | + |
| 382 | +MODULE_LICENSE("GPL"); |
| 383 | +MODULE_DESCRIPTION("Virtio input device driver"); |
| 384 | +MODULE_AUTHOR("Gerd Hoffmann <kraxel@redhat.com>"); |