Home | History | Annotate | Line # | Download | only in kern
subr_cprng.c revision 1.5.2.3.4.2 
      1 /*	$NetBSD: subr_cprng.c,v 1.5.2.3.4.2 2013/03/29 00:46:58 msaitoh Exp $ */
      2 
      3 /*-
      4  * Copyright (c) 2011 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * This code is derived from software contributed to The NetBSD Foundation
      8  * by Thor Lancelot Simon.
      9  *
     10  * Redistribution and use in source and binary forms, with or without
     11  * modification, are permitted provided that the following conditions
     12  * are met:
     13  * 1. Redistributions of source code must retain the above copyright
     14  *    notice, this list of conditions and the following disclaimer.
     15  * 2. Redistributions in binary form must reproduce the above copyright
     16  *    notice, this list of conditions and the following disclaimer in the
     17  *    documentation and/or other materials provided with the distribution.
     18  *
     19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     29  * POSSIBILITY OF SUCH DAMAGE.
     30  */
     31 
     32 #include <sys/types.h>
     33 #include <sys/time.h>
     34 #include <sys/param.h>
     35 #include <sys/kernel.h>
     36 #include <sys/systm.h>
     37 #include <sys/kmem.h>
     38 #include <sys/mutex.h>
     39 #include <sys/rngtest.h>
     40 #include <sys/rnd.h>
     41 #include <dev/rnd_private.h>
     42 
     43 #if defined(__HAVE_CPU_COUNTER)
     44 #include <machine/cpu_counter.h>
     45 #endif
     46 
     47 #include <sys/cprng.h>
     48 
     49 __KERNEL_RCSID(0, "$NetBSD: subr_cprng.c,v 1.5.2.3.4.2 2013/03/29 00:46:58 msaitoh Exp $");
     50 
     51 void
     52 cprng_init(void)
     53 {
     54 	nist_ctr_initialize();
     55 }
     56 
     57 static inline uint32_t
     58 cprng_counter(void)
     59 {
     60 	struct timeval tv;
     61 
     62 #if defined(__HAVE_CPU_COUNTER)
     63 	if (cpu_hascounter())
     64 		return cpu_counter32();
     65 #endif
     66 	if (__predict_false(cold)) {
     67 		/* microtime unsafe if clock not running yet */
     68 		return 0;
     69 	}
     70 	microtime(&tv);
     71 	return (tv.tv_sec * 1000000 + tv.tv_usec);
     72 }
     73 
     74 static void
     75 cprng_strong_doreseed(cprng_strong_t *const c)
     76 {
     77 	uint32_t cc = cprng_counter();
     78 
     79 	KASSERT(mutex_owned(&c->mtx));
     80 	KASSERT(mutex_owned(&c->reseed.mtx));
     81 	KASSERT(c->reseed.len == NIST_BLOCK_KEYLEN_BYTES);
     82 
     83 	if (nist_ctr_drbg_reseed(&c->drbg, c->reseed.data, c->reseed.len,
     84 				 &cc, sizeof(cc))) {
     85 		panic("cprng %s: nist_ctr_drbg_reseed failed.", c->name);
     86 	}
     87 	memset(c->reseed.data, 0, c->reseed.len);
     88 
     89 #ifdef RND_VERBOSE
     90 	printf("cprng %s: reseeded with rnd_filled = %d\n", c->name,
     91 							    rnd_filled);
     92 #endif
     93 	c->entropy_serial = rnd_filled;
     94 	c->reseed.state = RSTATE_IDLE;
     95 	if (c->flags & CPRNG_USE_CV) {
     96 		cv_broadcast(&c->cv);
     97 	}
     98 	selnotify(&c->selq, 0, 0);
     99 }
    100 
    101 static void
    102 cprng_strong_sched_reseed(cprng_strong_t *const c)
    103 {
    104 	KASSERT(mutex_owned(&c->mtx));
    105 	if (mutex_tryenter(&c->reseed.mtx)) {
    106 		switch (c->reseed.state) {
    107 		    case RSTATE_IDLE:
    108 			c->reseed.state = RSTATE_PENDING;
    109 			c->reseed.len = NIST_BLOCK_KEYLEN_BYTES;
    110 			rndsink_attach(&c->reseed);
    111 			break;
    112 		    case RSTATE_HASBITS:
    113 			/* Just rekey the underlying generator now. */
    114 			cprng_strong_doreseed(c);
    115 			break;
    116 		    case RSTATE_PENDING:
    117 			if (c->entropy_serial != rnd_filled) {
    118 				rndsink_detach(&c->reseed);
    119 				rndsink_attach(&c->reseed);
    120 			}
    121 			break;
    122 		    default:
    123 			panic("cprng %s: bad reseed state %d",
    124 			      c->name, c->reseed.state);
    125 			break;
    126 		}
    127 		mutex_spin_exit(&c->reseed.mtx);
    128 	}
    129 #ifdef RND_VERBOSE
    130 	else {
    131 		printf("cprng %s: skipping sched_reseed, sink busy\n",
    132 		       c->name);
    133 	}
    134 #endif
    135 }
    136 
    137 static void
    138 cprng_strong_reseed(void *const arg)
    139 {
    140 	cprng_strong_t *c = arg;
    141 
    142 	KASSERT(mutex_owned(&c->reseed.mtx));
    143 	KASSERT(RSTATE_HASBITS == c->reseed.state);
    144 
    145 	if (!mutex_tryenter(&c->mtx)) {
    146 #ifdef RND_VERBOSE
    147 	    printf("cprng: sink %s cprng busy, no reseed\n", c->reseed.name);
    148 #endif
    149 	    if (c->flags & CPRNG_USE_CV) {	/* XXX if flags change? */
    150         	cv_broadcast(&c->cv);
    151 	    }
    152 	    return;
    153 	}
    154 
    155 	cprng_strong_doreseed(c);
    156 	mutex_exit(&c->mtx);
    157 }
    158 
    159 static size_t
    160 cprng_entropy_try(uint8_t *key, size_t keylen)
    161 {
    162 	int r;
    163 	r = rnd_extract_data(key, keylen, RND_EXTRACT_GOOD);
    164 	if (r != keylen) {	/* Always fill in, for safety */
    165 		rnd_extract_data(key + r, keylen - r, RND_EXTRACT_ANY);
    166 	}
    167 	return r;
    168 }
    169 
    170 cprng_strong_t *
    171 cprng_strong_create(const char *const name, int ipl, int flags)
    172 {
    173 	cprng_strong_t *c;
    174 	uint8_t key[NIST_BLOCK_KEYLEN_BYTES];
    175 	int r, getmore = 0, hard = 0;
    176 	uint32_t cc;
    177 
    178 	c = kmem_alloc(sizeof(*c), KM_NOSLEEP);
    179 	if (c == NULL) {
    180 		return NULL;
    181 	}
    182 	c->flags = flags;
    183 	strlcpy(c->name, name, sizeof(c->name));
    184 	c->reseed.state = RSTATE_IDLE;
    185 	c->reseed.cb = cprng_strong_reseed;
    186 	c->reseed.arg = c;
    187 	c->entropy_serial = rnd_filled;
    188 	mutex_init(&c->reseed.mtx, MUTEX_DEFAULT, IPL_VM);
    189 	strlcpy(c->reseed.name, name, sizeof(c->reseed.name));
    190 
    191 	mutex_init(&c->mtx, MUTEX_DEFAULT, ipl);
    192 
    193 	if (c->flags & CPRNG_USE_CV) {
    194 		cv_init(&c->cv, name);
    195 	}
    196 
    197 	selinit(&c->selq);
    198 
    199 	r = cprng_entropy_try(key, sizeof(key));
    200 	if (r != sizeof(key)) {
    201 		if (c->flags & CPRNG_INIT_ANY) {
    202 #ifdef DEBUG
    203 			printf("cprng %s: WARNING insufficient "
    204 			       "entropy at creation.\n", name);
    205 #endif
    206 		} else {
    207 			hard++;
    208 		}
    209 		getmore++;
    210 	}
    211 
    212 	if (nist_ctr_drbg_instantiate(&c->drbg, key, sizeof(key),
    213 				      &cc, sizeof(cc), name, strlen(name))) {
    214 		panic("cprng %s: instantiation failed.", name);
    215 	}
    216 
    217 	if (getmore) {
    218 		/* Cause readers to wait for rekeying. */
    219 		if (hard) {
    220 			c->drbg.reseed_counter =
    221 			    NIST_CTR_DRBG_RESEED_INTERVAL + 1;
    222 		} else {
    223 			c->drbg.reseed_counter =
    224 			    (NIST_CTR_DRBG_RESEED_INTERVAL / 2) + 1;
    225 		}
    226 	}
    227 	return c;
    228 }
    229 
    230 size_t
    231 cprng_strong(cprng_strong_t *const c, void *const p, size_t len, int flags)
    232 {
    233 	uint32_t cc = cprng_counter();
    234 #ifdef DEBUG
    235 	int testfail = 0;
    236 #endif
    237 	if (len > CPRNG_MAX_LEN) {	/* XXX should we loop? */
    238 		len = CPRNG_MAX_LEN;	/* let the caller loop if desired */
    239 	}
    240 	mutex_enter(&c->mtx);
    241 
    242 	if (nist_ctr_drbg_generate(&c->drbg, p, len, &cc, sizeof(cc))) {
    243 		/* A generator failure really means we hit the hard limit. */
    244 		if (c->flags & CPRNG_REKEY_ANY) {
    245 			uint8_t key[NIST_BLOCK_KEYLEN_BYTES];
    246 
    247 			if (cprng_entropy_try(key, sizeof(key)) !=
    248 			    sizeof(key)) {
    249  				printf("cprng %s: WARNING "
    250 				       "pseudorandom rekeying.\n", c->name);
    251 			}
    252 			cc = cprng_counter();
    253 			if (nist_ctr_drbg_reseed(&c->drbg, key, sizeof(key),
    254 			    &cc, sizeof(cc))) {
    255 				panic("cprng %s: nist_ctr_drbg_reseed "
    256 				      "failed.", c->name);
    257 			}
    258 			memset(key, 0, sizeof(key));
    259 		} else {
    260 			int wr;
    261 
    262 			do {
    263 				cprng_strong_sched_reseed(c);
    264 				if ((flags & FNONBLOCK) ||
    265 				    !(c->flags & CPRNG_USE_CV)) {
    266 					len = 0;
    267 					break;
    268 				}
    269 			/*
    270 			 * XXX There's a race with the cv_broadcast
    271 			 * XXX in cprng_strong_sched_reseed, because
    272 			 * XXX of the use of tryenter in that function.
    273 			 * XXX This "timedwait" hack works around it,
    274 			 * XXX at the expense of occasionaly polling
    275 			 * XXX for success on a /dev/random rekey.
    276 			 */
    277 				wr = cv_timedwait_sig(&c->cv, &c->mtx,
    278 						      mstohz(100));
    279 				if (wr == ERESTART) {
    280 					mutex_exit(&c->mtx);
    281 					return 0;
    282 				}
    283 			} while (nist_ctr_drbg_generate(&c->drbg, p,
    284 							len, &cc,
    285 							sizeof(cc)));
    286 		}
    287 	}
    288 
    289 #ifdef DEBUG
    290 	/*
    291 	 * If the generator has just been keyed, perform
    292 	 * the statistical RNG test.
    293 	 */
    294 	if (__predict_false(c->drbg.reseed_counter == 1)) {
    295 		rngtest_t *rt = kmem_alloc(sizeof(*rt), KM_NOSLEEP);
    296 
    297 		if (rt) {
    298 
    299 			strncpy(rt->rt_name, c->name, sizeof(rt->rt_name));
    300 
    301 			if (nist_ctr_drbg_generate(&c->drbg, rt->rt_b,
    302 		  	    sizeof(rt->rt_b), NULL, 0)) {
    303 				panic("cprng %s: nist_ctr_drbg_generate "
    304 				      "failed!", c->name);
    305 
    306 			}
    307 			testfail = rngtest(rt);
    308 
    309 			if (testfail) {
    310 				printf("cprng %s: failed statistical RNG "
    311 				       "test.\n", c->name);
    312 				c->drbg.reseed_counter =
    313 				    NIST_CTR_DRBG_RESEED_INTERVAL + 1;
    314 				len = 0;
    315 			}
    316 			memset(rt, 0, sizeof(*rt));
    317 			kmem_free(rt, sizeof(*rt));
    318 		}
    319 	}
    320 #endif
    321 	if (__predict_false(c->drbg.reseed_counter >
    322 			     (NIST_CTR_DRBG_RESEED_INTERVAL / 2))) {
    323 		cprng_strong_sched_reseed(c);
    324 	} else if (rnd_full) {
    325 		if (c->entropy_serial != rnd_filled) {
    326 #ifdef RND_VERBOSE
    327 			printf("cprng %s: reseeding from full pool "
    328 			       "(serial %d vs pool %d)\n", c->name,
    329 			       c->entropy_serial, rnd_filled);
    330 #endif
    331 			cprng_strong_sched_reseed(c);
    332 		}
    333 	}
    334 
    335 	mutex_exit(&c->mtx);
    336 	return len;
    337 }
    338 
    339 void
    340 cprng_strong_destroy(cprng_strong_t *c)
    341 {
    342 	mutex_enter(&c->mtx);
    343 	mutex_spin_enter(&c->reseed.mtx);
    344 
    345 	if (c->flags & CPRNG_USE_CV) {
    346 		KASSERT(!cv_has_waiters(&c->cv));
    347 		cv_destroy(&c->cv);
    348 	}
    349 	seldestroy(&c->selq);
    350 
    351 	if (RSTATE_PENDING == c->reseed.state) {
    352 		rndsink_detach(&c->reseed);
    353 	}
    354 	mutex_spin_exit(&c->reseed.mtx);
    355 	mutex_destroy(&c->reseed.mtx);
    356 
    357 	nist_ctr_drbg_destroy(&c->drbg);
    358 
    359 	mutex_exit(&c->mtx);
    360 	mutex_destroy(&c->mtx);
    361 
    362 	memset(c, 0, sizeof(*c));
    363 	kmem_free(c, sizeof(*c));
    364 }
    365 
    366 int
    367 cprng_strong_getflags(cprng_strong_t *const c)
    368 {
    369 	KASSERT(mutex_owned(&c->mtx));
    370 	return c->flags;
    371 }
    372 
    373 void
    374 cprng_strong_setflags(cprng_strong_t *const c, int flags)
    375 {
    376 	KASSERT(mutex_owned(&c->mtx));
    377 	if (flags & CPRNG_USE_CV) {
    378 		if (!(c->flags & CPRNG_USE_CV)) {
    379 			cv_init(&c->cv, (const char *)c->name);
    380 		}
    381 	} else {
    382 		if (c->flags & CPRNG_USE_CV) {
    383 			KASSERT(!cv_has_waiters(&c->cv));
    384 			cv_destroy(&c->cv);
    385 		}
    386 	}
    387 	if (flags & CPRNG_REKEY_ANY) {
    388 		if (!(c->flags & CPRNG_REKEY_ANY)) {
    389 			if (c->flags & CPRNG_USE_CV) {
    390 				cv_broadcast(&c->cv);
    391 			}
    392 			selnotify(&c->selq, 0, 0);
    393 		}
    394 	}
    395 	c->flags = flags;
    396 }
    397