1 // SPDX-License-Identifier: GPL-2.0
3 * chaoskey - driver for ChaosKey device from Altus Metrum.
5 * This device provides true random numbers using a noise source based
6 * on a reverse-biased p-n junction in avalanche breakdown. More
7 * details can be found at http://chaoskey.org
9 * The driver connects to the kernel hardware RNG interface to provide
10 * entropy for /dev/random and other kernel activities. It also offers
11 * a separate /dev/ entry to allow for direct access to the random
14 * Copyright © 2015 Keith Packard <keithp@keithp.com>
17 #include <linux/module.h>
18 #include <linux/slab.h>
19 #include <linux/usb.h>
20 #include <linux/wait.h>
21 #include <linux/hw_random.h>
22 #include <linux/mutex.h>
23 #include <linux/uaccess.h>
25 static struct usb_driver chaoskey_driver;
26 static struct usb_class_driver chaoskey_class;
27 static int chaoskey_rng_read(struct hwrng *rng, void *data,
28 size_t max, bool wait);
30 #define usb_dbg(usb_if, format, arg...) \
31 dev_dbg(&(usb_if)->dev, format, ## arg)
33 #define usb_err(usb_if, format, arg...) \
34 dev_err(&(usb_if)->dev, format, ## arg)
36 /* Version Information */
37 #define DRIVER_AUTHOR "Keith Packard, keithp@keithp.com"
38 #define DRIVER_DESC "Altus Metrum ChaosKey driver"
39 #define DRIVER_SHORT "chaoskey"
41 MODULE_AUTHOR(DRIVER_AUTHOR);
42 MODULE_DESCRIPTION(DRIVER_DESC);
43 MODULE_LICENSE("GPL");
45 #define CHAOSKEY_VENDOR_ID 0x1d50 /* OpenMoko */
46 #define CHAOSKEY_PRODUCT_ID 0x60c6 /* ChaosKey */
48 #define ALEA_VENDOR_ID 0x12d8 /* Araneus */
49 #define ALEA_PRODUCT_ID 0x0001 /* Alea I */
51 #define CHAOSKEY_BUF_LEN 64 /* max size of USB full speed packet */
53 #define NAK_TIMEOUT (HZ) /* normal stall/wait timeout */
54 #define ALEA_FIRST_TIMEOUT (HZ*3) /* first stall/wait timeout for Alea */
56 #ifdef CONFIG_USB_DYNAMIC_MINORS
57 #define USB_CHAOSKEY_MINOR_BASE 0
60 /* IOWARRIOR_MINOR_BASE + 16, not official yet */
61 #define USB_CHAOSKEY_MINOR_BASE 224
64 static const struct usb_device_id chaoskey_table[] = {
65 { USB_DEVICE(CHAOSKEY_VENDOR_ID, CHAOSKEY_PRODUCT_ID) },
66 { USB_DEVICE(ALEA_VENDOR_ID, ALEA_PRODUCT_ID) },
69 MODULE_DEVICE_TABLE(usb, chaoskey_table);
71 static void chaos_read_callback(struct urb *urb);
73 /* Driver-local specific stuff */
75 struct usb_interface *interface;
78 struct mutex rng_lock;
79 int open; /* open count */
80 bool present; /* device not disconnected */
81 bool reading; /* ongoing IO */
82 bool reads_started; /* track first read for Alea */
83 int size; /* size of buf */
84 int valid; /* bytes of buf read */
85 int used; /* bytes of buf consumed */
86 char *name; /* product + serial */
87 struct hwrng hwrng; /* Embedded struct for hwrng */
88 int hwrng_registered; /* registered with hwrng API */
89 wait_queue_head_t wait_q; /* for timeouts */
90 struct urb *urb; /* for performing IO */
94 static void chaoskey_free(struct chaoskey *dev)
97 usb_dbg(dev->interface, "free");
98 usb_free_urb(dev->urb);
101 usb_put_intf(dev->interface);
106 static int chaoskey_probe(struct usb_interface *interface,
107 const struct usb_device_id *id)
109 struct usb_device *udev = interface_to_usbdev(interface);
110 struct usb_host_interface *altsetting = interface->cur_altsetting;
111 struct usb_endpoint_descriptor *epd;
113 struct chaoskey *dev;
114 int result = -ENOMEM;
118 usb_dbg(interface, "probe %s-%s", udev->product, udev->serial);
120 /* Find the first bulk IN endpoint and its packet size */
121 res = usb_find_bulk_in_endpoint(altsetting, &epd);
123 usb_dbg(interface, "no IN endpoint found");
127 in_ep = usb_endpoint_num(epd);
128 size = usb_endpoint_maxp(epd);
130 /* Validate endpoint and size */
132 usb_dbg(interface, "invalid size (%d)", size);
136 if (size > CHAOSKEY_BUF_LEN) {
137 usb_dbg(interface, "size reduced from %d to %d\n",
138 size, CHAOSKEY_BUF_LEN);
139 size = CHAOSKEY_BUF_LEN;
142 /* Looks good, allocate and initialize */
144 dev = kzalloc(sizeof(struct chaoskey), GFP_KERNEL);
149 dev->interface = usb_get_intf(interface);
151 dev->buf = kmalloc(size, GFP_KERNEL);
153 if (dev->buf == NULL)
156 dev->urb = usb_alloc_urb(0, GFP_KERNEL);
161 usb_fill_bulk_urb(dev->urb,
163 usb_rcvbulkpipe(udev, in_ep),
169 /* Construct a name using the product and serial values. Each
170 * device needs a unique name for the hwrng code
173 if (udev->product && udev->serial) {
174 dev->name = kasprintf(GFP_KERNEL, "%s-%s", udev->product,
176 if (dev->name == NULL)
182 if (le16_to_cpu(udev->descriptor.idVendor) != ALEA_VENDOR_ID)
183 dev->reads_started = true;
188 init_waitqueue_head(&dev->wait_q);
190 mutex_init(&dev->lock);
191 mutex_init(&dev->rng_lock);
193 usb_set_intfdata(interface, dev);
195 result = usb_register_dev(interface, &chaoskey_class);
197 usb_err(interface, "Unable to allocate minor number.");
201 dev->hwrng.name = dev->name ? dev->name : chaoskey_driver.name;
202 dev->hwrng.read = chaoskey_rng_read;
203 dev->hwrng.quality = 1024;
205 dev->hwrng_registered = (hwrng_register(&dev->hwrng) == 0);
206 if (!dev->hwrng_registered)
207 usb_err(interface, "Unable to register with hwrng");
209 usb_enable_autosuspend(udev);
211 usb_dbg(interface, "chaoskey probe success, size %d", dev->size);
215 usb_set_intfdata(interface, NULL);
220 static void chaoskey_disconnect(struct usb_interface *interface)
222 struct chaoskey *dev;
224 usb_dbg(interface, "disconnect");
225 dev = usb_get_intfdata(interface);
227 usb_dbg(interface, "disconnect failed - no dev");
231 if (dev->hwrng_registered)
232 hwrng_unregister(&dev->hwrng);
234 usb_deregister_dev(interface, &chaoskey_class);
236 usb_set_intfdata(interface, NULL);
237 mutex_lock(&dev->lock);
239 dev->present = false;
240 usb_poison_urb(dev->urb);
243 mutex_unlock(&dev->lock);
246 mutex_unlock(&dev->lock);
248 usb_dbg(interface, "disconnect done");
251 static int chaoskey_open(struct inode *inode, struct file *file)
253 struct chaoskey *dev;
254 struct usb_interface *interface;
256 /* get the interface from minor number and driver information */
257 interface = usb_find_interface(&chaoskey_driver, iminor(inode));
261 usb_dbg(interface, "open");
263 dev = usb_get_intfdata(interface);
265 usb_dbg(interface, "open (dev)");
269 file->private_data = dev;
270 mutex_lock(&dev->lock);
272 mutex_unlock(&dev->lock);
274 usb_dbg(interface, "open success");
278 static int chaoskey_release(struct inode *inode, struct file *file)
280 struct chaoskey *dev = file->private_data;
281 struct usb_interface *interface;
286 interface = dev->interface;
288 usb_dbg(interface, "release");
290 mutex_lock(&dev->lock);
292 usb_dbg(interface, "open count at release is %d", dev->open);
294 if (dev->open <= 0) {
295 usb_dbg(interface, "invalid open count (%d)", dev->open);
296 mutex_unlock(&dev->lock);
303 if (dev->open == 0) {
304 mutex_unlock(&dev->lock);
307 mutex_unlock(&dev->lock);
309 mutex_unlock(&dev->lock);
311 usb_dbg(interface, "release success");
315 static void chaos_read_callback(struct urb *urb)
317 struct chaoskey *dev = urb->context;
318 int status = urb->status;
320 usb_dbg(dev->interface, "callback status (%d)", status);
323 dev->valid = urb->actual_length;
329 /* must be seen first before validity is announced */
332 dev->reading = false;
333 wake_up(&dev->wait_q);
336 /* Fill the buffer. Called with dev->lock held
338 static int _chaoskey_fill(struct chaoskey *dev)
344 usb_dbg(dev->interface, "fill");
346 /* Return immediately if someone called before the buffer was
348 if (dev->valid != dev->used) {
349 usb_dbg(dev->interface, "not empty yet (valid %d used %d)",
350 dev->valid, dev->used);
354 /* Bail if the device has been removed */
356 usb_dbg(dev->interface, "device not present");
360 /* Make sure the device is awake */
361 result = usb_autopm_get_interface(dev->interface);
363 usb_dbg(dev->interface, "wakeup failed (result %d)", result);
368 result = usb_submit_urb(dev->urb, GFP_KERNEL);
370 result = usb_translate_errors(result);
371 dev->reading = false;
375 /* The first read on the Alea takes a little under 2 seconds.
376 * Reads after the first read take only a few microseconds
377 * though. Presumably the entropy-generating circuit needs
378 * time to ramp up. So, we wait longer on the first read.
380 started = dev->reads_started;
381 dev->reads_started = true;
382 result = wait_event_interruptible_timeout(
385 (started ? NAK_TIMEOUT : ALEA_FIRST_TIMEOUT) );
395 /* Let the device go back to sleep eventually */
396 usb_autopm_put_interface(dev->interface);
398 usb_dbg(dev->interface, "read %d bytes", dev->valid);
403 static ssize_t chaoskey_read(struct file *file,
408 struct chaoskey *dev;
409 ssize_t read_count = 0;
412 unsigned long remain;
414 dev = file->private_data;
416 if (dev == NULL || !dev->present)
419 usb_dbg(dev->interface, "read %zu", count);
423 /* Grab the rng_lock briefly to ensure that the hwrng interface
424 * gets priority over other user access
426 result = mutex_lock_interruptible(&dev->rng_lock);
429 mutex_unlock(&dev->rng_lock);
431 result = mutex_lock_interruptible(&dev->lock);
434 if (dev->valid == dev->used) {
435 result = _chaoskey_fill(dev);
437 mutex_unlock(&dev->lock);
442 this_time = dev->valid - dev->used;
443 if (this_time > count)
446 remain = copy_to_user(buffer, dev->buf + dev->used, this_time);
450 /* Consume the bytes that were copied so we don't leak
453 dev->used += this_time - remain;
454 mutex_unlock(&dev->lock);
459 read_count += this_time;
461 dev->used += this_time;
462 mutex_unlock(&dev->lock);
466 usb_dbg(dev->interface, "read %zu bytes", read_count);
469 usb_dbg(dev->interface, "empty read, result %d", result);
470 if (result == -ETIMEDOUT)
475 static int chaoskey_rng_read(struct hwrng *rng, void *data,
476 size_t max, bool wait)
478 struct chaoskey *dev = container_of(rng, struct chaoskey, hwrng);
481 usb_dbg(dev->interface, "rng_read max %zu wait %d", max, wait);
484 usb_dbg(dev->interface, "device not present");
488 /* Hold the rng_lock until we acquire the device lock so that
489 * this operation gets priority over other user access to the
492 mutex_lock(&dev->rng_lock);
494 mutex_lock(&dev->lock);
496 mutex_unlock(&dev->rng_lock);
498 /* Try to fill the buffer if empty. It doesn't actually matter
499 * if _chaoskey_fill works; we'll just return zero bytes as
500 * the buffer will still be empty
502 if (dev->valid == dev->used)
503 (void) _chaoskey_fill(dev);
505 this_time = dev->valid - dev->used;
509 memcpy(data, dev->buf + dev->used, this_time);
511 dev->used += this_time;
513 mutex_unlock(&dev->lock);
515 usb_dbg(dev->interface, "rng_read this_time %d\n", this_time);
520 static int chaoskey_suspend(struct usb_interface *interface,
521 pm_message_t message)
523 usb_dbg(interface, "suspend");
527 static int chaoskey_resume(struct usb_interface *interface)
529 usb_dbg(interface, "resume");
533 #define chaoskey_suspend NULL
534 #define chaoskey_resume NULL
537 /* file operation pointers */
538 static const struct file_operations chaoskey_fops = {
539 .owner = THIS_MODULE,
540 .read = chaoskey_read,
541 .open = chaoskey_open,
542 .release = chaoskey_release,
543 .llseek = default_llseek,
546 /* class driver information */
547 static struct usb_class_driver chaoskey_class = {
548 .name = "chaoskey%d",
549 .fops = &chaoskey_fops,
550 .minor_base = USB_CHAOSKEY_MINOR_BASE,
553 /* usb specific object needed to register this driver with the usb subsystem */
554 static struct usb_driver chaoskey_driver = {
555 .name = DRIVER_SHORT,
556 .probe = chaoskey_probe,
557 .disconnect = chaoskey_disconnect,
558 .suspend = chaoskey_suspend,
559 .resume = chaoskey_resume,
560 .reset_resume = chaoskey_resume,
561 .id_table = chaoskey_table,
562 .supports_autosuspend = 1,
565 module_usb_driver(chaoskey_driver);