2 * chaoskey - driver for ChaosKey device from Altus Metrum.
4 * This device provides true random numbers using a noise source based
5 * on a reverse-biased p-n junction in avalanche breakdown. More
6 * details can be found at http://chaoskey.org
8 * The driver connects to the kernel hardware RNG interface to provide
9 * entropy for /dev/random and other kernel activities. It also offers
10 * a separate /dev/ entry to allow for direct access to the random
13 * Copyright © 2015 Keith Packard <keithp@keithp.com>
15 * This program is free software; you can redistribute it and/or modify
16 * it under the terms of the GNU General Public License as published by
17 * the Free Software Foundation; version 2 of the License.
19 * This program is distributed in the hope that it will be useful, but
20 * WITHOUT ANY WARRANTY; without even the implied warranty of
21 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
22 * General Public License for more details.
25 #include <linux/module.h>
26 #include <linux/slab.h>
27 #include <linux/usb.h>
28 #include <linux/wait.h>
29 #include <linux/hw_random.h>
30 #include <linux/mutex.h>
31 #include <linux/uaccess.h>
33 static struct usb_driver chaoskey_driver
;
34 static struct usb_class_driver chaoskey_class
;
35 static int chaoskey_rng_read(struct hwrng
*rng
, void *data
,
36 size_t max
, bool wait
);
38 #define usb_dbg(usb_if, format, arg...) \
39 dev_dbg(&(usb_if)->dev, format, ## arg)
41 #define usb_err(usb_if, format, arg...) \
42 dev_err(&(usb_if)->dev, format, ## arg)
44 /* Version Information */
45 #define DRIVER_VERSION "v0.1"
46 #define DRIVER_AUTHOR "Keith Packard, keithp@keithp.com"
47 #define DRIVER_DESC "Altus Metrum ChaosKey driver"
48 #define DRIVER_SHORT "chaoskey"
50 MODULE_VERSION(DRIVER_VERSION
);
51 MODULE_AUTHOR(DRIVER_AUTHOR
);
52 MODULE_DESCRIPTION(DRIVER_DESC
);
53 MODULE_LICENSE("GPL");
55 #define CHAOSKEY_VENDOR_ID 0x1d50 /* OpenMoko */
56 #define CHAOSKEY_PRODUCT_ID 0x60c6 /* ChaosKey */
58 #define CHAOSKEY_BUF_LEN 64 /* max size of USB full speed packet */
60 #define NAK_TIMEOUT (HZ) /* stall/wait timeout for device */
62 #ifdef CONFIG_USB_DYNAMIC_MINORS
63 #define USB_CHAOSKEY_MINOR_BASE 0
66 /* IOWARRIOR_MINOR_BASE + 16, not official yet */
67 #define USB_CHAOSKEY_MINOR_BASE 224
70 static const struct usb_device_id chaoskey_table
[] = {
71 { USB_DEVICE(CHAOSKEY_VENDOR_ID
, CHAOSKEY_PRODUCT_ID
) },
74 MODULE_DEVICE_TABLE(usb
, chaoskey_table
);
76 /* Driver-local specific stuff */
78 struct usb_interface
*interface
;
81 struct mutex rng_lock
;
82 int open
; /* open count */
83 int present
; /* device not disconnected */
84 int size
; /* size of buf */
85 int valid
; /* bytes of buf read */
86 int used
; /* bytes of buf consumed */
87 char *name
; /* product + serial */
88 struct hwrng hwrng
; /* Embedded struct for hwrng */
89 int hwrng_registered
; /* registered with hwrng API */
90 wait_queue_head_t wait_q
; /* for timeouts */
94 static void chaoskey_free(struct chaoskey
*dev
)
96 usb_dbg(dev
->interface
, "free");
102 static int chaoskey_probe(struct usb_interface
*interface
,
103 const struct usb_device_id
*id
)
105 struct usb_device
*udev
= interface_to_usbdev(interface
);
106 struct usb_host_interface
*altsetting
= interface
->cur_altsetting
;
109 struct chaoskey
*dev
;
113 usb_dbg(interface
, "probe %s-%s", udev
->product
, udev
->serial
);
115 /* Find the first bulk IN endpoint and its packet size */
116 for (i
= 0; i
< altsetting
->desc
.bNumEndpoints
; i
++) {
117 if (usb_endpoint_is_bulk_in(&altsetting
->endpoint
[i
].desc
)) {
118 in_ep
= usb_endpoint_num(&altsetting
->endpoint
[i
].desc
);
119 size
= usb_endpoint_maxp(&altsetting
->endpoint
[i
].desc
);
124 /* Validate endpoint and size */
126 usb_dbg(interface
, "no IN endpoint found");
130 usb_dbg(interface
, "invalid size (%d)", size
);
134 if (size
> CHAOSKEY_BUF_LEN
) {
135 usb_dbg(interface
, "size reduced from %d to %d\n",
136 size
, CHAOSKEY_BUF_LEN
);
137 size
= CHAOSKEY_BUF_LEN
;
140 /* Looks good, allocate and initialize */
142 dev
= kzalloc(sizeof(struct chaoskey
), GFP_KERNEL
);
147 dev
->buf
= kmalloc(size
, GFP_KERNEL
);
149 if (dev
->buf
== NULL
) {
154 /* Construct a name using the product and serial values. Each
155 * device needs a unique name for the hwrng code
158 if (udev
->product
&& udev
->serial
) {
159 dev
->name
= kmalloc(strlen(udev
->product
) + 1 +
160 strlen(udev
->serial
) + 1, GFP_KERNEL
);
161 if (dev
->name
== NULL
) {
167 strcpy(dev
->name
, udev
->product
);
168 strcat(dev
->name
, "-");
169 strcat(dev
->name
, udev
->serial
);
172 dev
->interface
= interface
;
179 init_waitqueue_head(&dev
->wait_q
);
181 mutex_init(&dev
->lock
);
182 mutex_init(&dev
->rng_lock
);
184 usb_set_intfdata(interface
, dev
);
186 result
= usb_register_dev(interface
, &chaoskey_class
);
188 usb_err(interface
, "Unable to allocate minor number.");
189 usb_set_intfdata(interface
, NULL
);
194 dev
->hwrng
.name
= dev
->name
? dev
->name
: chaoskey_driver
.name
;
195 dev
->hwrng
.read
= chaoskey_rng_read
;
197 /* Set the 'quality' metric. Quality is measured in units of
198 * 1/1024's of a bit ("mills"). This should be set to 1024,
199 * but there is a bug in the hwrng core which masks it with
202 * The patch that has been merged to the crypto development
203 * tree for that bug limits the value to 1024 at most, so by
204 * setting this to 1024 + 1023, we get 1023 before the fix is
205 * merged and 1024 afterwards. We'll patch this driver once
206 * both bits of code are in the same tree.
208 dev
->hwrng
.quality
= 1024 + 1023;
210 dev
->hwrng_registered
= (hwrng_register(&dev
->hwrng
) == 0);
211 if (!dev
->hwrng_registered
)
212 usb_err(interface
, "Unable to register with hwrng");
214 usb_enable_autosuspend(udev
);
216 usb_dbg(interface
, "chaoskey probe success, size %d", dev
->size
);
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
);
242 mutex_unlock(&dev
->lock
);
245 mutex_unlock(&dev
->lock
);
247 usb_dbg(interface
, "disconnect done");
250 static int chaoskey_open(struct inode
*inode
, struct file
*file
)
252 struct chaoskey
*dev
;
253 struct usb_interface
*interface
;
255 /* get the interface from minor number and driver information */
256 interface
= usb_find_interface(&chaoskey_driver
, iminor(inode
));
260 usb_dbg(interface
, "open");
262 dev
= usb_get_intfdata(interface
);
264 usb_dbg(interface
, "open (dev)");
268 file
->private_data
= dev
;
269 mutex_lock(&dev
->lock
);
271 mutex_unlock(&dev
->lock
);
273 usb_dbg(interface
, "open success");
277 static int chaoskey_release(struct inode
*inode
, struct file
*file
)
279 struct chaoskey
*dev
= file
->private_data
;
280 struct usb_interface
*interface
;
285 interface
= dev
->interface
;
287 usb_dbg(interface
, "release");
289 mutex_lock(&dev
->lock
);
291 usb_dbg(interface
, "open count at release is %d", dev
->open
);
293 if (dev
->open
<= 0) {
294 usb_dbg(interface
, "invalid open count (%d)", dev
->open
);
295 mutex_unlock(&dev
->lock
);
302 if (dev
->open
== 0) {
303 mutex_unlock(&dev
->lock
);
306 mutex_unlock(&dev
->lock
);
308 mutex_unlock(&dev
->lock
);
310 usb_dbg(interface
, "release success");
314 /* Fill the buffer. Called with dev->lock held
316 static int _chaoskey_fill(struct chaoskey
*dev
)
321 struct usb_device
*udev
= interface_to_usbdev(dev
->interface
);
323 usb_dbg(dev
->interface
, "fill");
325 /* Return immediately if someone called before the buffer was
327 if (dev
->valid
!= dev
->used
) {
328 usb_dbg(dev
->interface
, "not empty yet (valid %d used %d)",
329 dev
->valid
, dev
->used
);
333 /* Bail if the device has been removed */
335 usb_dbg(dev
->interface
, "device not present");
339 /* Make sure the device is awake */
340 result
= usb_autopm_get_interface(dev
->interface
);
342 usb_dbg(dev
->interface
, "wakeup failed (result %d)", result
);
346 result
= usb_bulk_msg(udev
,
347 usb_rcvbulkpipe(udev
, dev
->in_ep
),
348 dev
->buf
, dev
->size
, &this_read
,
351 /* Let the device go back to sleep eventually */
352 usb_autopm_put_interface(dev
->interface
);
355 dev
->valid
= this_read
;
359 usb_dbg(dev
->interface
, "bulk_msg result %d this_read %d",
365 static ssize_t
chaoskey_read(struct file
*file
,
370 struct chaoskey
*dev
;
371 ssize_t read_count
= 0;
374 unsigned long remain
;
376 dev
= file
->private_data
;
378 if (dev
== NULL
|| !dev
->present
)
381 usb_dbg(dev
->interface
, "read %zu", count
);
385 /* Grab the rng_lock briefly to ensure that the hwrng interface
386 * gets priority over other user access
388 result
= mutex_lock_interruptible(&dev
->rng_lock
);
391 mutex_unlock(&dev
->rng_lock
);
393 result
= mutex_lock_interruptible(&dev
->lock
);
396 if (dev
->valid
== dev
->used
) {
397 result
= _chaoskey_fill(dev
);
399 mutex_unlock(&dev
->lock
);
403 /* Read returned zero bytes */
404 if (dev
->used
== dev
->valid
) {
405 mutex_unlock(&dev
->lock
);
410 this_time
= dev
->valid
- dev
->used
;
411 if (this_time
> count
)
414 remain
= copy_to_user(buffer
, dev
->buf
+ dev
->used
, this_time
);
418 /* Consume the bytes that were copied so we don't leak
421 dev
->used
+= this_time
- remain
;
422 mutex_unlock(&dev
->lock
);
427 read_count
+= this_time
;
429 dev
->used
+= this_time
;
430 mutex_unlock(&dev
->lock
);
434 usb_dbg(dev
->interface
, "read %zu bytes", read_count
);
437 usb_dbg(dev
->interface
, "empty read, result %d", result
);
441 static int chaoskey_rng_read(struct hwrng
*rng
, void *data
,
442 size_t max
, bool wait
)
444 struct chaoskey
*dev
= container_of(rng
, struct chaoskey
, hwrng
);
447 usb_dbg(dev
->interface
, "rng_read max %zu wait %d", max
, wait
);
450 usb_dbg(dev
->interface
, "device not present");
454 /* Hold the rng_lock until we acquire the device lock so that
455 * this operation gets priority over other user access to the
458 mutex_lock(&dev
->rng_lock
);
460 mutex_lock(&dev
->lock
);
462 mutex_unlock(&dev
->rng_lock
);
464 /* Try to fill the buffer if empty. It doesn't actually matter
465 * if _chaoskey_fill works; we'll just return zero bytes as
466 * the buffer will still be empty
468 if (dev
->valid
== dev
->used
)
469 (void) _chaoskey_fill(dev
);
471 this_time
= dev
->valid
- dev
->used
;
475 memcpy(data
, dev
->buf
, this_time
);
477 dev
->used
+= this_time
;
479 mutex_unlock(&dev
->lock
);
481 usb_dbg(dev
->interface
, "rng_read this_time %d\n", this_time
);
486 static int chaoskey_suspend(struct usb_interface
*interface
,
487 pm_message_t message
)
489 usb_dbg(interface
, "suspend");
493 static int chaoskey_resume(struct usb_interface
*interface
)
495 usb_dbg(interface
, "resume");
499 #define chaoskey_suspend NULL
500 #define chaoskey_resume NULL
503 /* file operation pointers */
504 static const struct file_operations chaoskey_fops
= {
505 .owner
= THIS_MODULE
,
506 .read
= chaoskey_read
,
507 .open
= chaoskey_open
,
508 .release
= chaoskey_release
,
509 .llseek
= default_llseek
,
512 /* class driver information */
513 static struct usb_class_driver chaoskey_class
= {
514 .name
= "chaoskey%d",
515 .fops
= &chaoskey_fops
,
516 .minor_base
= USB_CHAOSKEY_MINOR_BASE
,
519 /* usb specific object needed to register this driver with the usb subsystem */
520 static struct usb_driver chaoskey_driver
= {
521 .name
= DRIVER_SHORT
,
522 .probe
= chaoskey_probe
,
523 .disconnect
= chaoskey_disconnect
,
524 .suspend
= chaoskey_suspend
,
525 .resume
= chaoskey_resume
,
526 .reset_resume
= chaoskey_resume
,
527 .id_table
= chaoskey_table
,
528 .supports_autosuspend
= 1,
531 module_usb_driver(chaoskey_driver
);