1 /* $FreeBSD: src/sys/dev/rndtest/rndtest.c,v 1.1.4.1 2003/06/04 17:10:30 sam Exp $ */
2 /* $DragonFly: src/sys/dev/crypto/rndtest/rndtest.c,v 1.9 2006/12/18 20:41:01 dillon Exp $ */
6 * Copyright (c) 2002 Jason L. Wright (jason@thought.net)
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 * notice, this list of conditions and the following disclaimer in the
16 * documentation and/or other materials provided with the distribution.
17 * 3. All advertising materials mentioning features or use of this software
18 * must display the following acknowledgement:
19 * This product includes software developed by Jason L. Wright
20 * 4. The name of the author may not be used to endorse or promote products
21 * derived from this software without specific prior written permission.
23 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
24 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
25 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
26 * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
27 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
28 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
29 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
31 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
32 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
33 * POSSIBILITY OF SUCH DAMAGE.
36 #include <sys/param.h>
37 #include <sys/systm.h>
39 #include <sys/callout.h>
40 #include <sys/kernel.h>
41 #include <sys/malloc.h>
42 #include <sys/module.h>
43 #include <sys/random.h>
44 #include <sys/sysctl.h>
45 #include <machine/stdarg.h>
49 static int rndtest_modevent(module_t, int, void *);
50 static void rndtest_report(struct rndtest_state *, int, const char *, ...)
52 static void rndtest_test(struct rndtest_state *);
53 static void rndtest_timeout(void *);
55 /* The tests themselves */
56 static int rndtest_monobit(struct rndtest_state *);
57 static int rndtest_runs(struct rndtest_state *);
58 static int rndtest_longruns(struct rndtest_state *);
59 static int rndtest_chi_4(struct rndtest_state *);
61 static int rndtest_runs_check(struct rndtest_state *, int, int *);
62 static void rndtest_runs_record(struct rndtest_state *, int, int *);
64 static const struct rndtest_testfunc {
65 int (*test)(struct rndtest_state *);
73 #define RNDTEST_NTESTS (sizeof(rndtest_funcs)/sizeof(rndtest_funcs[0]))
75 SYSCTL_NODE(_kern, OID_AUTO, rndtest, CTLFLAG_RD, 0, "RNG test parameters");
77 static int rndtest_retest = 120; /* interval in seconds */
78 SYSCTL_INT(_kern_rndtest, OID_AUTO, retest, CTLFLAG_RW, &rndtest_retest,
79 0, "retest interval (seconds)");
80 static struct rndtest_stats rndstats;
81 SYSCTL_STRUCT(_kern_rndtest, OID_AUTO, stats, CTLFLAG_RD, &rndstats,
82 rndtest_stats, "RNG test statistics");
83 static int rndtest_verbose = 1; /* report only failures */
84 SYSCTL_INT(_kern_rndtest, OID_AUTO, verbose, CTLFLAG_RW, &rndtest_verbose,
85 0, "display results on console");
87 struct rndtest_state *
88 rndtest_attach(device_t dev)
90 struct rndtest_state *rsp;
92 rsp = kmalloc(sizeof (*rsp), M_DEVBUF, M_INTWAIT);
93 rsp->rs_begin = rsp->rs_buf;
94 rsp->rs_end = rsp->rs_buf + sizeof(rsp->rs_buf);
95 rsp->rs_current = rsp->rs_begin;
99 /* NB: 1 means the callout runs w/o Giant locked */
100 callout_init_mp(&rsp->rs_to);
105 rndtest_detach(struct rndtest_state *rsp)
107 callout_stop(&rsp->rs_to);
108 kfree(rsp, M_DEVBUF);
112 rndtest_harvest(struct rndtest_state *rsp, void *buf, u_int len)
116 * If enabled, collect data and run tests when we have enough.
118 if (rsp->rs_collect) {
119 for (i = 0; i < len; i++) {
120 *rsp->rs_current = ((u_char *) buf)[i];
121 if (++rsp->rs_current == rsp->rs_end) {
123 rsp->rs_current = rsp->rs_begin;
125 * If tests passed, turn off collection and
126 * schedule another test. Otherwise we keep
127 * testing until the data looks ok.
129 if (!rsp->rs_discard && rndtest_retest != 0) {
131 callout_reset(&rsp->rs_to,
133 rndtest_timeout, rsp);
140 * Only stir entropy that passes muster into the pool.
143 rndstats.rst_discard += len;
145 #if defined(__DragonFly__) || __FreeBSD_version < 500000
146 /* XXX verify buffer is word aligned */
148 for (len /= sizeof (u_int32_t); len; len--)
149 add_true_randomness(*p++);
151 random_harvest(buf, len, len*NBBY, 0, RANDOM_PURE);
157 rndtest_test(struct rndtest_state *rsp)
161 rndstats.rst_tests++;
162 for (i = 0; i < RNDTEST_NTESTS; i++)
163 rv |= (*rndtest_funcs[i].test)(rsp);
164 rsp->rs_discard = (rv != 0);
168 rndtest_report(struct rndtest_state *rsp, int failure, const char *fmt, ...)
173 if (rndtest_verbose == 0)
175 if (!failure && rndtest_verbose == 1) /* don't report successes */
178 kvsnprintf(buf, sizeof (buf), fmt, ap);
180 device_printf(rsp->rs_parent, "rndtest: %s\n", buf);
183 #define RNDTEST_MONOBIT_MINONES 9725
184 #define RNDTEST_MONOBIT_MAXONES 10275
187 rndtest_monobit(struct rndtest_state *rsp)
192 for (i = 0; i < RNDTEST_NBYTES; i++) {
194 for (j = 0; j < 8; j++, r <<= 1)
198 if (ones > RNDTEST_MONOBIT_MINONES &&
199 ones < RNDTEST_MONOBIT_MAXONES) {
200 if (rndtest_verbose > 1)
201 rndtest_report(rsp, 0, "monobit pass (%d < %d < %d)",
202 RNDTEST_MONOBIT_MINONES, ones,
203 RNDTEST_MONOBIT_MAXONES);
207 rndtest_report(rsp, 1,
208 "monobit failed (%d ones)", ones);
209 rndstats.rst_monobit++;
214 #define RNDTEST_RUNS_NINTERVAL 6
216 static const struct rndtest_runs_tabs {
218 } rndtest_runs_tab[] = {
228 rndtest_runs(struct rndtest_state *rsp)
230 int i, j, ones, zeros, rv = 0;
231 int onei[RNDTEST_RUNS_NINTERVAL], zeroi[RNDTEST_RUNS_NINTERVAL];
234 bzero(onei, sizeof(onei));
235 bzero(zeroi, sizeof(zeroi));
237 for (i = 0; i < RNDTEST_NBYTES; i++) {
239 for (j = 0; j < 8; j++, c <<= 1) {
242 rndtest_runs_record(rsp, zeros, zeroi);
246 rndtest_runs_record(rsp, ones, onei);
251 rndtest_runs_record(rsp, ones, onei);
252 rndtest_runs_record(rsp, zeros, zeroi);
254 rv |= rndtest_runs_check(rsp, 0, zeroi);
255 rv |= rndtest_runs_check(rsp, 1, onei);
264 rndtest_runs_record(struct rndtest_state *rsp, int len, int *intrv)
268 if (len > RNDTEST_RUNS_NINTERVAL)
269 len = RNDTEST_RUNS_NINTERVAL;
275 rndtest_runs_check(struct rndtest_state *rsp, int val, int *src)
279 for (i = 0; i < RNDTEST_RUNS_NINTERVAL; i++) {
280 if (src[i] < rndtest_runs_tab[i].min ||
281 src[i] > rndtest_runs_tab[i].max) {
282 rndtest_report(rsp, 1,
283 "%s interval %d failed (%d, %d-%d)",
284 val ? "ones" : "zeros",
285 i + 1, src[i], rndtest_runs_tab[i].min,
286 rndtest_runs_tab[i].max);
289 rndtest_report(rsp, 0,
290 "runs pass %s interval %d (%d < %d < %d)",
291 val ? "ones" : "zeros",
292 i + 1, rndtest_runs_tab[i].min, src[i],
293 rndtest_runs_tab[i].max);
300 rndtest_longruns(struct rndtest_state *rsp)
302 int i, j, ones = 0, zeros = 0, maxones = 0, maxzeros = 0;
305 for (i = 0; i < RNDTEST_NBYTES; i++) {
307 for (j = 0; j < 8; j++, c <<= 1) {
316 if (zeros > maxzeros)
322 if (maxones < 26 && maxzeros < 26) {
323 rndtest_report(rsp, 0, "longruns pass (%d ones, %d zeros)",
327 rndtest_report(rsp, 1, "longruns fail (%d ones, %d zeros)",
329 rndstats.rst_longruns++;
335 * chi^2 test over 4 bits: (this is called the poker test in FIPS 140-2,
336 * but it is really the chi^2 test over 4 bits (the poker test as described
337 * by Knuth vol 2 is something different, and I take him as authoritative
338 * on nomenclature over NIST).
340 #define RNDTEST_CHI4_K 16
341 #define RNDTEST_CHI4_K_MASK (RNDTEST_CHI4_K - 1)
344 * The unnormalized values are used so that we don't have to worry about
345 * fractional precision. The "real" value is found by:
346 * (V - 1562500) * (16 / 5000) = Vn (where V is the unnormalized value)
348 #define RNDTEST_CHI4_VMIN 1563181 /* 2.1792 */
349 #define RNDTEST_CHI4_VMAX 1576929 /* 46.1728 */
352 rndtest_chi_4(struct rndtest_state *rsp)
354 unsigned int freq[RNDTEST_CHI4_K], i, sum;
356 for (i = 0; i < RNDTEST_CHI4_K; i++)
359 /* Get number of occurances of each 4 bit pattern */
360 for (i = 0; i < RNDTEST_NBYTES; i++) {
361 freq[(rsp->rs_buf[i] >> 4) & RNDTEST_CHI4_K_MASK]++;
362 freq[(rsp->rs_buf[i] >> 0) & RNDTEST_CHI4_K_MASK]++;
365 for (i = 0, sum = 0; i < RNDTEST_CHI4_K; i++)
366 sum += freq[i] * freq[i];
368 if (sum >= 1563181 && sum <= 1576929) {
369 rndtest_report(rsp, 0, "chi^2(4): pass (sum %u)", sum);
372 rndtest_report(rsp, 1, "chi^2(4): failed (sum %u)", sum);
379 rndtest_timeout(void *xrsp)
381 struct rndtest_state *rsp = xrsp;
387 rndtest_modevent(module_t mod, int type, void *unused)
398 static moduledata_t rndtest_mod = {
403 DECLARE_MODULE(rndtest, rndtest_mod, SI_SUB_DRIVERS, SI_ORDER_ANY);
404 MODULE_VERSION(rndtest, 1);