Make use of the copy-on-leak functionality of the virtio rng driver to expose a mechanism to user space for detecting entropy leak events, such as taking a VM snapshot or restoring from one. The driver setups a single page of memory where it stores in the first word a counter and queues a copy-on-leak command for increasing the counter every time an entropy leak occurs. It exposes the value of the counter in a binary sysfs file per device. The file can be mmap'ed and read and every time a change on the counter is observed, `sysfs_notify` is used to notify processes that are polling it. The mechanism is implemented based on the idea of a VM generation counter that had been before proposed as an extension to the VM Generation ID device, where mmap and poll interfaces can be used on the file containing the counter and changes in its value signal snapshot events. It is worth noting that using mmap is entirely race-free, since changes in the counter are observable by user-space as soon as vcpus are resumed. Instead, using poll is not race-free. There is a race-window between the moment the vcpus are resumed and the used-buffers are handled by the virtio-rng driver. Signed-off-by: Babis Chalios <bchalios@xxxxxxxxx> --- drivers/char/hw_random/virtio-rng.c | 178 +++++++++++++++++++++++++++- 1 file changed, 175 insertions(+), 3 deletions(-) diff --git a/drivers/char/hw_random/virtio-rng.c b/drivers/char/hw_random/virtio-rng.c index 154f68a1e326..9fe9da09f202 100644 --- a/drivers/char/hw_random/virtio-rng.c +++ b/drivers/char/hw_random/virtio-rng.c @@ -4,6 +4,9 @@ * Copyright (C) 2007, 2008 Rusty Russell IBM Corporation */ +#include "linux/gfp.h" +#include "linux/minmax.h" +#include "linux/sysfs.h" #include <linux/hw_random.h> #include <linux/scatterlist.h> #include <linux/spinlock.h> @@ -15,6 +18,10 @@ static DEFINE_IDA(rng_index_ida); +#ifdef CONFIG_SYSFS +static struct kobject *virtio_rng_kobj; +#endif + struct virtrng_info { struct hwrng hwrng; struct virtqueue *vq; @@ -23,6 +30,12 @@ struct virtrng_info { struct virtqueue *leakq[2]; spinlock_t lock; int active_leakq; +#ifdef CONFIG_SYSFS + struct kobject *kobj; + struct bin_attribute vm_gen_counter_attr; + unsigned long map_buffer; + unsigned long next_vm_gen_counter; +#endif char name[25]; int index; @@ -42,6 +55,40 @@ struct virtrng_info { #endif }; +#ifdef CONFIG_SYSFS +static ssize_t virtrng_sysfs_read(struct file *filep, struct kobject *kobj, + struct bin_attribute *attr, char *buf, loff_t pos, size_t len) +{ + struct virtrng_info *vi = attr->private; + unsigned long gen_counter = *(unsigned long *)vi->map_buffer; + + if (!len) + return 0; + + len = min(len, sizeof(gen_counter)); + memcpy(buf, &gen_counter, len); + + return len; +} + +static int virtrng_sysfs_mmap(struct file *filep, struct kobject *kobj, + struct bin_attribute *attr, struct vm_area_struct *vma) +{ + struct virtrng_info *vi = attr->private; + + if (vma->vm_pgoff || vma_pages(vma) > 1) + return -EINVAL; + + if (vma->vm_flags & VM_WRITE) + return -EPERM; + + vma->vm_flags |= VM_DONTEXPAND; + vma->vm_flags &= ~VM_MAYWRITE; + + return vm_insert_page(vma, vma->vm_start, virt_to_page(vi->map_buffer)); +} +#endif + /* Swaps the queues and returns the new active leak queue. */ static struct virtqueue *swap_leakqs(struct virtrng_info *vi) { @@ -81,7 +128,7 @@ static int virtrng_fill_on_leak(struct virtrng_info *vi, void *data, size_t len) vq = get_active_leakq(vi); ret = add_fill_on_leak_request(vi, vq, data, len); - if (ret) + if (!ret) virtqueue_kick(vq); spin_unlock_irqrestore(&vi->lock, flags); @@ -121,7 +168,7 @@ static int virtrng_copy_on_leak(struct virtrng_info *vi, void *to, void *from, s vq = get_active_leakq(vi); ret = add_copy_on_leak_request(vi, vq, to, from, len); - if (ret) + if (!ret) virtqueue_kick(vq); spin_unlock_irqrestore(&vi->lock, flags); @@ -137,6 +184,9 @@ static void entropy_leak_detected(struct virtqueue *vq) unsigned long flags; void *buffer; bool kick_activeq = false; +#ifdef CONFIG_SYSFS + bool notify_sysfs = false; +#endif spin_lock_irqsave(&vi->lock, flags); @@ -158,12 +208,34 @@ static void entropy_leak_detected(struct virtqueue *vq) add_fill_on_leak_request(vi, activeq, vi->leak_data, sizeof(vi->leak_data)); kick_activeq = true; } + +#ifdef CONFIG_SYSFS + if (buffer == (void *)vi->map_buffer) { + notify_sysfs = true; + + /* Add a request to bump the generation counter on the next leak event. + * We have already swapped leak queues, so this will get properly handled + * with the next entropy leak event. + */ + vi->next_vm_gen_counter++; + add_copy_on_leak_request(vi, activeq, (void *)vi->map_buffer, + &vi->next_vm_gen_counter, sizeof(unsigned long)); + + kick_activeq = true; + } +#endif } if (kick_activeq) virtqueue_kick(activeq); spin_unlock_irqrestore(&vi->lock, flags); + +#ifdef CONFIG_SYSFS + /* Notify anyone polling on the sysfs file */ + if (notify_sysfs) + sysfs_notify(vi->kobj, NULL, "vm_gen_counter"); +#endif } static void random_recv_done(struct virtqueue *vq) @@ -300,6 +372,59 @@ static int init_virtqueues(struct virtrng_info *vi, struct virtio_device *vdev) return ret; } +#ifdef CONFIG_SYSFS +static int setup_sysfs(struct virtrng_info *vi) +{ + int err; + + vi->next_vm_gen_counter = 1; + + /* We have one binary file per device under /sys/virtio-rng/<device>/vm_gen_counter */ + vi->vm_gen_counter_attr.attr.name = "vm_gen_counter"; + vi->vm_gen_counter_attr.attr.mode = 0444; + vi->vm_gen_counter_attr.read = virtrng_sysfs_read; + vi->vm_gen_counter_attr.mmap = virtrng_sysfs_mmap; + vi->vm_gen_counter_attr.private = vi; + + vi->map_buffer = get_zeroed_page(GFP_KERNEL); + if (!vi->map_buffer) + return -ENOMEM; + + err = -ENOMEM; + vi->kobj = kobject_create_and_add(vi->name, virtio_rng_kobj); + if (!vi->kobj) + goto err_page; + + err = sysfs_create_bin_file(vi->kobj, &vi->vm_gen_counter_attr); + if (err) + goto err_kobj; + + return 0; + +err_kobj: + kobject_put(vi->kobj); +err_page: + free_pages(vi->map_buffer, 0); + return err; +} + +static void cleanup_sysfs(struct virtrng_info *vi) +{ + sysfs_remove_bin_file(vi->kobj, &vi->vm_gen_counter_attr); + kobject_put(vi->kobj); + free_pages(vi->map_buffer, 0); +} +#else +static int setup_sysfs(struct virtrng_info *vi) +{ + return 0; +} + +static void cleanup_sysfs(struct virtrng_info *vi) +{ +} +#endif + static int probe_common(struct virtio_device *vdev) { int err, index; @@ -330,11 +455,15 @@ static int probe_common(struct virtio_device *vdev) if (vi->has_leakqs) { spin_lock_init(&vi->lock); vi->active_leakq = 0; + + err = setup_sysfs(vi); + if (err) + goto err_find; } err = init_virtqueues(vi, vdev); if (err) - goto err_find; + goto err_sysfs; virtio_device_ready(vdev); @@ -344,8 +473,18 @@ static int probe_common(struct virtio_device *vdev) /* we always have a fill_on_leak request pending */ virtrng_fill_on_leak(vi, vi->leak_data, sizeof(vi->leak_data)); +#ifdef CONFIG_SYSFS + /* also a copy_on_leak request for the generation counter when we have sysfs + * support. + */ + virtrng_copy_on_leak(vi, (void *)vi->map_buffer, &vi->next_vm_gen_counter, + sizeof(unsigned long)); +#endif + return 0; +err_sysfs: + cleanup_sysfs(vi); err_find: ida_simple_remove(&rng_index_ida, index); err_ida: @@ -363,6 +502,8 @@ static void remove_common(struct virtio_device *vdev) complete(&vi->have_data); if (vi->hwrng_register_done) hwrng_unregister(&vi->hwrng); + if (vi->has_leakqs) + cleanup_sysfs(vi); virtio_reset_device(vdev); vdev->config->del_vqs(vdev); ida_simple_remove(&rng_index_ida, vi->index); @@ -445,7 +586,38 @@ static struct virtio_driver virtio_rng_driver = { #endif }; +#ifdef CONFIG_SYSFS +static int __init virtio_rng_init(void) +{ + int ret; + + virtio_rng_kobj = kobject_create_and_add("virtio-rng", NULL); + if (!virtio_rng_kobj) + return -ENOMEM; + + ret = register_virtio_driver(&virtio_rng_driver); + if (ret < 0) + goto err; + + return 0; + +err: + kobject_put(virtio_rng_kobj); + return ret; +} + +static void __exit virtio_rng_fini(void) +{ + kobject_put(virtio_rng_kobj); + unregister_virtio_driver(&virtio_rng_driver); +} + +module_init(virtio_rng_init); +module_exit(virtio_rng_fini); +#else module_virtio_driver(virtio_rng_driver); +#endif + MODULE_DEVICE_TABLE(virtio, id_table); MODULE_DESCRIPTION("Virtio random number driver"); MODULE_LICENSE("GPL"); -- 2.38.1 Amazon Spain Services sociedad limitada unipersonal, Calle Ramirez de Prado 5, 28045 Madrid. Registro Mercantil de Madrid . Tomo 22458 . Folio 102 . Hoja M-401234 . CIF B84570936