2 * hw_random/core.c: HWRNG core API
4 * Copyright 2006 Michael Buesch <m@bues.ch>
5 * Copyright 2005 (c) MontaVista Software, Inc.
7 * Please read Documentation/hw_random.txt for details on use.
9 * This software may be used and distributed according to the terms
10 * of the GNU General Public License, incorporated herein by reference.
13 #include <linux/delay.h>
14 #include <linux/device.h>
15 #include <linux/err.h>
17 #include <linux/hw_random.h>
18 #include <linux/kernel.h>
19 #include <linux/kthread.h>
20 #include <linux/sched/signal.h>
21 #include <linux/miscdevice.h>
22 #include <linux/module.h>
23 #include <linux/random.h>
24 #include <linux/sched.h>
25 #include <linux/slab.h>
26 #include <linux/uaccess.h>
28 #define RNG_MODULE_NAME "hw_random"
30 static struct hwrng *current_rng;
31 static struct task_struct *hwrng_fill;
32 static LIST_HEAD(rng_list);
33 /* Protects rng_list and current_rng */
34 static DEFINE_MUTEX(rng_mutex);
35 /* Protects rng read functions, data_avail, rng_buffer and rng_fillbuf */
36 static DEFINE_MUTEX(reading_mutex);
37 static int data_avail;
38 static u8 *rng_buffer, *rng_fillbuf;
39 static unsigned short current_quality;
40 static unsigned short default_quality; /* = 0; default to "off" */
42 module_param(current_quality, ushort, 0644);
43 MODULE_PARM_DESC(current_quality,
44 "current hwrng entropy estimation per mill");
45 module_param(default_quality, ushort, 0644);
46 MODULE_PARM_DESC(default_quality,
47 "default entropy content of hwrng per mill");
49 static void drop_current_rng(void);
50 static int hwrng_init(struct hwrng *rng);
51 static void start_khwrngd(void);
53 static inline int rng_get_data(struct hwrng *rng, u8 *buffer, size_t size,
56 static size_t rng_buffer_size(void)
58 return SMP_CACHE_BYTES < 32 ? 32 : SMP_CACHE_BYTES;
61 static void add_early_randomness(struct hwrng *rng)
64 size_t size = min_t(size_t, 16, rng_buffer_size());
66 mutex_lock(&reading_mutex);
67 bytes_read = rng_get_data(rng, rng_buffer, size, 1);
68 mutex_unlock(&reading_mutex);
70 add_device_randomness(rng_buffer, bytes_read);
73 static inline void cleanup_rng(struct kref *kref)
75 struct hwrng *rng = container_of(kref, struct hwrng, ref);
80 complete(&rng->cleanup_done);
83 static int set_current_rng(struct hwrng *rng)
87 BUG_ON(!mutex_is_locked(&rng_mutex));
89 err = hwrng_init(rng);
99 static void drop_current_rng(void)
101 BUG_ON(!mutex_is_locked(&rng_mutex));
105 /* decrease last reference for triggering the cleanup */
106 kref_put(¤t_rng->ref, cleanup_rng);
110 /* Returns ERR_PTR(), NULL or refcounted hwrng */
111 static struct hwrng *get_current_rng(void)
115 if (mutex_lock_interruptible(&rng_mutex))
116 return ERR_PTR(-ERESTARTSYS);
122 mutex_unlock(&rng_mutex);
126 static void put_rng(struct hwrng *rng)
129 * Hold rng_mutex here so we serialize in case they set_current_rng
130 * on rng again immediately.
132 mutex_lock(&rng_mutex);
134 kref_put(&rng->ref, cleanup_rng);
135 mutex_unlock(&rng_mutex);
138 static int hwrng_init(struct hwrng *rng)
140 if (kref_get_unless_zero(&rng->ref))
146 ret = rng->init(rng);
151 kref_init(&rng->ref);
152 reinit_completion(&rng->cleanup_done);
155 add_early_randomness(rng);
157 current_quality = rng->quality ? : default_quality;
158 if (current_quality > 1024)
159 current_quality = 1024;
161 if (current_quality == 0 && hwrng_fill)
162 kthread_stop(hwrng_fill);
163 if (current_quality > 0 && !hwrng_fill)
169 static int rng_dev_open(struct inode *inode, struct file *filp)
171 /* enforce read-only access to this chrdev */
172 if ((filp->f_mode & FMODE_READ) == 0)
174 if (filp->f_mode & FMODE_WRITE)
179 static inline int rng_get_data(struct hwrng *rng, u8 *buffer, size_t size,
183 BUG_ON(!mutex_is_locked(&reading_mutex));
185 return rng->read(rng, (void *)buffer, size, wait);
187 if (rng->data_present)
188 present = rng->data_present(rng, wait);
193 return rng->data_read(rng, (u32 *)buffer);
198 static ssize_t rng_dev_read(struct file *filp, char __user *buf,
199 size_t size, loff_t *offp)
207 rng = get_current_rng();
217 if (mutex_lock_interruptible(&reading_mutex)) {
222 bytes_read = rng_get_data(rng, rng_buffer,
224 !(filp->f_flags & O_NONBLOCK));
225 if (bytes_read < 0) {
227 goto out_unlock_reading;
229 data_avail = bytes_read;
233 if (filp->f_flags & O_NONBLOCK) {
235 goto out_unlock_reading;
244 if (copy_to_user(buf + ret, rng_buffer + data_avail,
247 goto out_unlock_reading;
254 mutex_unlock(&reading_mutex);
258 schedule_timeout_interruptible(1);
260 if (signal_pending(current)) {
269 mutex_unlock(&reading_mutex);
275 static const struct file_operations rng_chrdev_ops = {
276 .owner = THIS_MODULE,
277 .open = rng_dev_open,
278 .read = rng_dev_read,
279 .llseek = noop_llseek,
282 static const struct attribute_group *rng_dev_groups[];
284 static struct miscdevice rng_miscdev = {
285 .minor = HWRNG_MINOR,
286 .name = RNG_MODULE_NAME,
288 .fops = &rng_chrdev_ops,
289 .groups = rng_dev_groups,
292 static ssize_t hwrng_attr_current_store(struct device *dev,
293 struct device_attribute *attr,
294 const char *buf, size_t len)
299 err = mutex_lock_interruptible(&rng_mutex);
303 list_for_each_entry(rng, &rng_list, list) {
304 if (sysfs_streq(rng->name, buf)) {
306 if (rng != current_rng)
307 err = set_current_rng(rng);
311 mutex_unlock(&rng_mutex);
316 static ssize_t hwrng_attr_current_show(struct device *dev,
317 struct device_attribute *attr,
323 rng = get_current_rng();
327 ret = snprintf(buf, PAGE_SIZE, "%s\n", rng ? rng->name : "none");
333 static ssize_t hwrng_attr_available_show(struct device *dev,
334 struct device_attribute *attr,
340 err = mutex_lock_interruptible(&rng_mutex);
344 list_for_each_entry(rng, &rng_list, list) {
345 strlcat(buf, rng->name, PAGE_SIZE);
346 strlcat(buf, " ", PAGE_SIZE);
348 strlcat(buf, "\n", PAGE_SIZE);
349 mutex_unlock(&rng_mutex);
354 static DEVICE_ATTR(rng_current, S_IRUGO | S_IWUSR,
355 hwrng_attr_current_show,
356 hwrng_attr_current_store);
357 static DEVICE_ATTR(rng_available, S_IRUGO,
358 hwrng_attr_available_show,
361 static struct attribute *rng_dev_attrs[] = {
362 &dev_attr_rng_current.attr,
363 &dev_attr_rng_available.attr,
367 ATTRIBUTE_GROUPS(rng_dev);
369 static void __exit unregister_miscdev(void)
371 misc_deregister(&rng_miscdev);
374 static int __init register_miscdev(void)
376 return misc_register(&rng_miscdev);
379 static int hwrng_fillfn(void *unused)
383 while (!kthread_should_stop()) {
386 rng = get_current_rng();
387 if (IS_ERR(rng) || !rng)
389 mutex_lock(&reading_mutex);
390 rc = rng_get_data(rng, rng_fillbuf,
391 rng_buffer_size(), 1);
392 mutex_unlock(&reading_mutex);
395 pr_warn("hwrng: no data available\n");
396 msleep_interruptible(10000);
399 /* Outside lock, sure, but y'know: randomness. */
400 add_hwgenerator_randomness((void *)rng_fillbuf, rc,
401 rc * current_quality * 8 >> 10);
407 static void start_khwrngd(void)
409 hwrng_fill = kthread_run(hwrng_fillfn, NULL, "hwrng");
410 if (IS_ERR(hwrng_fill)) {
411 pr_err("hwrng_fill thread creation failed");
416 int hwrng_register(struct hwrng *rng)
419 struct hwrng *old_rng, *tmp;
421 if (!rng->name || (!rng->data_read && !rng->read))
424 mutex_lock(&rng_mutex);
425 /* Must not register two RNGs with the same name. */
427 list_for_each_entry(tmp, &rng_list, list) {
428 if (strcmp(tmp->name, rng->name) == 0)
432 init_completion(&rng->cleanup_done);
433 complete(&rng->cleanup_done);
435 old_rng = current_rng;
438 err = set_current_rng(rng);
442 list_add_tail(&rng->list, &rng_list);
444 if (old_rng && !rng->init) {
446 * Use a new device's input to add some randomness to
447 * the system. If this rng device isn't going to be
448 * used right away, its init function hasn't been
449 * called yet; so only use the randomness from devices
450 * that don't need an init callback.
452 add_early_randomness(rng);
456 mutex_unlock(&rng_mutex);
460 EXPORT_SYMBOL_GPL(hwrng_register);
462 void hwrng_unregister(struct hwrng *rng)
464 mutex_lock(&rng_mutex);
466 list_del(&rng->list);
467 if (current_rng == rng) {
469 if (!list_empty(&rng_list)) {
472 tail = list_entry(rng_list.prev, struct hwrng, list);
474 set_current_rng(tail);
478 if (list_empty(&rng_list)) {
479 mutex_unlock(&rng_mutex);
481 kthread_stop(hwrng_fill);
483 mutex_unlock(&rng_mutex);
485 wait_for_completion(&rng->cleanup_done);
487 EXPORT_SYMBOL_GPL(hwrng_unregister);
489 static void devm_hwrng_release(struct device *dev, void *res)
491 hwrng_unregister(*(struct hwrng **)res);
494 static int devm_hwrng_match(struct device *dev, void *res, void *data)
496 struct hwrng **r = res;
498 if (WARN_ON(!r || !*r))
504 int devm_hwrng_register(struct device *dev, struct hwrng *rng)
509 ptr = devres_alloc(devm_hwrng_release, sizeof(*ptr), GFP_KERNEL);
513 error = hwrng_register(rng);
520 devres_add(dev, ptr);
523 EXPORT_SYMBOL_GPL(devm_hwrng_register);
525 void devm_hwrng_unregister(struct device *dev, struct hwrng *rng)
527 devres_release(dev, devm_hwrng_release, devm_hwrng_match, rng);
529 EXPORT_SYMBOL_GPL(devm_hwrng_unregister);
531 static int __init hwrng_modinit(void)
535 /* kmalloc makes this safe for virt_to_page() in virtio_rng.c */
536 rng_buffer = kmalloc(rng_buffer_size(), GFP_KERNEL);
540 rng_fillbuf = kmalloc(rng_buffer_size(), GFP_KERNEL);
546 ret = register_miscdev();
555 static void __exit hwrng_modexit(void)
557 mutex_lock(&rng_mutex);
561 mutex_unlock(&rng_mutex);
563 unregister_miscdev();
566 module_init(hwrng_modinit);
567 module_exit(hwrng_modexit);
569 MODULE_DESCRIPTION("H/W Random Number Generator (RNG) driver");
570 MODULE_LICENSE("GPL");