sys/dev/random/harvest.c

   1 /*-
   2  * Copyright (c) 2000-2004 Mark R V Murray
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer
  10  *    in this position and unchanged.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  18  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25  *
  26  */
  27
  28 #include <sys/cdefs.h>
  29 __FBSDID("$FreeBSD$");
  30
  31 #include <sys/param.h>
  32 #include <sys/kthread.h>
  33 #include <sys/lock.h>
  34 #include <sys/malloc.h>
  35 #include <sys/mutex.h>
  36 #include <sys/poll.h>
  37 #include <sys/queue.h>
  38 #include <sys/random.h>
  39 #include <sys/selinfo.h>
  40 #include <sys/systm.h>
  41 #include <sys/sysctl.h>
  42
  43 #include <machine/cpu.h>
  44
  45 #include <dev/random/randomdev_soft.h>
  46
  47 static int read_random_phony(void *, int);
  48
  49 /* Structure holding the desired entropy sources */
  50 struct harvest_select harvest = { 1, 1, 1, 0 };
  51
  52 /* hold the address of the routine which is actually called if
  53  * the randomdev is loaded
  54  */
  55 static void (*reap_func)(u_int64_t, const void *, u_int, u_int, u_int,
  56     enum esource) = NULL;
  57 static int (*read_func)(void *, int) = read_random_phony;
  58
  59 /* Initialise the harvester at load time */
  60 void
  61 random_yarrow_init_harvester(void (*reaper)(u_int64_t, const void *, u_int,
  62     u_int, u_int, enum esource), int (*reader)(void *, int))
  63 {
  64         reap_func = reaper;
  65         read_func = reader;
  66 }
  67
  68 /* Deinitialise the harvester at unload time */
  69 void
  70 random_yarrow_deinit_harvester(void)
  71 {
  72         reap_func = NULL;
  73         read_func = read_random_phony;
  74 }
  75
  76 /* Entropy harvesting routine. This is supposed to be fast; do
  77  * not do anything slow in here!
  78  * Implemented as in indirect call to allow non-inclusion of
  79  * the entropy device.
  80  *
  81  * XXXRW: get_cyclecount() is cheap on most modern hardware, where cycle
  82  * counters are built in, but on older hardware it will do a real time clock
  83  * read which can be quite expensive.
  84  */
  85 void
  86 random_harvest(void *entropy, u_int count, u_int bits, u_int frac,
  87     enum esource origin)
  88 {
  89         if (reap_func)
  90                 (*reap_func)(get_cyclecount(), entropy, count, bits, frac,
  91                     origin);
  92 }
  93
  94 /* Userland-visible version of read_random */
  95 int
  96 read_random(void *buf, int count)
  97 {
  98         return ((*read_func)(buf, count));
  99 }
 100
 101 /* If the entropy device is not loaded, make a token effort to
 102  * provide _some_ kind of randomness. This should only be used
 103  * inside other RNG's, like arc4random(9).
 104  */
 105 static int
 106 read_random_phony(void *buf, int count)
 107 {
 108         u_long randval;
 109         int size, i;
 110
 111         /* srandom() is called in kern/init_main.c:proc0_post() */
 112
 113         /* Fill buf[] with random(9) output */
 114         for (i = 0; i < count; i+= (int)sizeof(u_long)) {
 115                 randval = random();
 116                 size = MIN(count - i, sizeof(u_long));
 117                 memcpy(&((char *)buf)[i], &randval, (size_t)size);
 118         }
 119
 120         return (count);
 121 }
 122
 123 /* Helper routine to enable kproc_exit() to work while the module is
 124  * being (or has been) unloaded.
 125  * This routine is in this file because it is always linked into the kernel,
 126  * and will thus never be unloaded. This is critical for unloadable modules
 127  * that have threads.
 128  */
 129 void
 130 random_set_wakeup_exit(void *control)
 131 {
 132         wakeup(control);
 133         kproc_exit(0);
 134         /* NOTREACHED */
 135 }