Home | History | Annotate | Line # | Download | only in rumpkern
hyperentropy.c revision 1.5
      1  1.5    justin /*	$NetBSD: hyperentropy.c,v 1.5 2014/08/15 17:45:00 justin Exp $	*/
      2  1.1     pooka 
      3  1.1     pooka /*
      4  1.1     pooka  * Copyright (c) 2014 Antti Kantee.  All Rights Reserved.
      5  1.1     pooka  *
      6  1.1     pooka  * Redistribution and use in source and binary forms, with or without
      7  1.1     pooka  * modification, are permitted provided that the following conditions
      8  1.1     pooka  * are met:
      9  1.1     pooka  * 1. Redistributions of source code must retain the above copyright
     10  1.1     pooka  *    notice, this list of conditions and the following disclaimer.
     11  1.1     pooka  * 2. Redistributions in binary form must reproduce the above copyright
     12  1.1     pooka  *    notice, this list of conditions and the following disclaimer in the
     13  1.1     pooka  *    documentation and/or other materials provided with the distribution.
     14  1.1     pooka  *
     15  1.1     pooka  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
     16  1.1     pooka  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
     17  1.1     pooka  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
     18  1.1     pooka  * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
     19  1.1     pooka  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     20  1.1     pooka  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
     21  1.1     pooka  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     22  1.1     pooka  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     23  1.1     pooka  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     24  1.1     pooka  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     25  1.1     pooka  * SUCH DAMAGE.
     26  1.1     pooka  */
     27  1.1     pooka 
     28  1.5    justin __KERNEL_RCSID(0, "$NetBSD: hyperentropy.c,v 1.5 2014/08/15 17:45:00 justin Exp $");
     29  1.5    justin 
     30  1.5    justin #include <sys/atomic.h>
     31  1.1     pooka #include <sys/cdefs.h>
     32  1.5    justin #include <sys/kmem.h>
     33  1.1     pooka #include <sys/param.h>
     34  1.1     pooka #include <sys/rnd.h>
     35  1.1     pooka 
     36  1.1     pooka #include <rump/rumpuser.h>
     37  1.1     pooka 
     38  1.1     pooka #include "rump_private.h"
     39  1.1     pooka 
     40  1.1     pooka static krndsource_t rndsrc;
     41  1.4  riastrad static volatile unsigned hyperentropy_wanted;
     42  1.4  riastrad static void *feedrandom_softint;
     43  1.1     pooka 
     44  1.2     pooka #define MAXGET (RND_POOLBITS/NBBY)
     45  1.1     pooka static void
     46  1.4  riastrad feedrandom(size_t bytes)
     47  1.1     pooka {
     48  1.2     pooka 	uint8_t *rnddata;
     49  1.1     pooka 	size_t dsize;
     50  1.1     pooka 
     51  1.2     pooka 	rnddata = kmem_intr_alloc(MAXGET, KM_SLEEP);
     52  1.2     pooka 	if (rumpuser_getrandom(rnddata, MIN(MAXGET, bytes),
     53  1.1     pooka 	    RUMPUSER_RANDOM_HARD|RUMPUSER_RANDOM_NOWAIT, &dsize) == 0)
     54  1.2     pooka 		rnd_add_data(&rndsrc, rnddata, dsize, NBBY*dsize);
     55  1.2     pooka 	kmem_intr_free(rnddata, MAXGET);
     56  1.1     pooka }
     57  1.1     pooka 
     58  1.4  riastrad static void
     59  1.4  riastrad feedrandom_intr(void *cookie __unused)
     60  1.4  riastrad {
     61  1.4  riastrad 
     62  1.4  riastrad 	feedrandom(atomic_swap_uint(&hyperentropy_wanted, 0));
     63  1.4  riastrad }
     64  1.4  riastrad 
     65  1.4  riastrad static void
     66  1.4  riastrad feedrandom_cb(size_t bytes, void *cookie __unused)
     67  1.4  riastrad {
     68  1.4  riastrad 	unsigned old, new;
     69  1.4  riastrad 
     70  1.4  riastrad 	do {
     71  1.4  riastrad 		old = hyperentropy_wanted;
     72  1.4  riastrad 		new = ((MAXGET - old) < bytes? MAXGET : (old + bytes));
     73  1.4  riastrad 	} while (atomic_cas_uint(&hyperentropy_wanted, old, new) != old);
     74  1.4  riastrad 
     75  1.4  riastrad 	softint_schedule(feedrandom_softint);
     76  1.4  riastrad }
     77  1.4  riastrad 
     78  1.1     pooka void
     79  1.1     pooka rump_hyperentropy_init(void)
     80  1.1     pooka {
     81  1.1     pooka 
     82  1.1     pooka 	if (rump_threads) {
     83  1.4  riastrad 		feedrandom_softint =
     84  1.4  riastrad 		    softint_establish(SOFTINT_CLOCK|SOFTINT_MPSAFE,
     85  1.4  riastrad 			feedrandom_intr, NULL);
     86  1.4  riastrad 		KASSERT(feedrandom_softint != NULL);
     87  1.4  riastrad 		rndsource_setcb(&rndsrc, feedrandom_cb, &rndsrc);
     88  1.1     pooka 		rnd_attach_source(&rndsrc, "rump_hyperent", RND_TYPE_VM,
     89  1.3       tls 		    RND_FLAG_COLLECT_VALUE|RND_FLAG_HASCB);
     90  1.1     pooka 	} else {
     91  1.2     pooka 		/* without threads, just fill the pool */
     92  1.1     pooka 		rnd_attach_source(&rndsrc, "rump_hyperent", RND_TYPE_VM,
     93  1.3       tls 		    RND_FLAG_COLLECT_VALUE);
     94  1.4  riastrad 		feedrandom(MAXGET);
     95  1.1     pooka 	}
     96  1.1     pooka }
     97