X-Git-Url: https://git.distorted.org.uk/u/mdw/catacomb/blobdiff_plain/bb78535ed9f01dd4b0afa408a0d513c0660c1d30..04d452095ff4435e8b13cdca81fb2d4a13caa2a5:/rspit.c diff --git a/rspit.c b/rspit.c index e10d05f..01ebf4d 100644 --- a/rspit.c +++ b/rspit.c @@ -1,6 +1,6 @@ /* -*-c-*- * - * $Id: rspit.c,v 1.4 2000/06/17 12:08:28 mdw Exp $ + * $Id: rspit.c,v 1.9 2000/08/04 23:24:15 mdw Exp $ * * Spit out random numbers * @@ -30,6 +30,23 @@ /*----- Revision history --------------------------------------------------* * * $Log: rspit.c,v $ + * Revision 1.9 2000/08/04 23:24:15 mdw + * Add a timer and a discard option. + * + * Revision 1.8 2000/07/29 22:05:47 mdw + * Fix error in help message about Maurer test syntax. + * + * Revision 1.7 2000/07/18 23:01:26 mdw + * Improve progress indications, and allow user to choose chunk sizes for + * Maurer's test. + * + * Revision 1.6 2000/07/15 20:53:35 mdw + * Add a load of new ciphers and hashes. + * + * Revision 1.5 2000/07/01 11:27:03 mdw + * Portability fix: don't assume that `stdout' is a constant expression. + * Remove old type name `bbs_param'. + * * Revision 1.4 2000/06/17 12:08:28 mdw * Restructure handling of cipher-based generators. Add counter-mode * ciphers and MGF-1 hash functions. Add FIPS 140-1 and Maurer's tests. @@ -89,30 +106,42 @@ #include "des3-ofb.h" #include "rc2-ofb.h" #include "rc5-ofb.h" +#include "skipjack-ofb.h" +#include "tea-ofb.h" +#include "xtea-ofb.h" #include "blowfish-ofb.h" #include "twofish-ofb.h" #include "idea-ofb.h" #include "cast128-ofb.h" #include "cast256-ofb.h" #include "rijndael-ofb.h" +#include "square-ofb.h" #include "serpent-ofb.h" #include "des-counter.h" #include "des3-counter.h" #include "rc2-counter.h" #include "rc5-counter.h" +#include "skipjack-counter.h" +#include "tea-counter.h" +#include "xtea-counter.h" #include "blowfish-counter.h" #include "twofish-counter.h" #include "idea-counter.h" #include "cast128-counter.h" #include "cast256-counter.h" #include "rijndael-counter.h" +#include "square-counter.h" #include "serpent-counter.h" #include "md4-mgf.h" #include "md5-mgf.h" #include "sha-mgf.h" +#include "tiger-mgf.h" +#include "rmd128-mgf.h" #include "rmd160-mgf.h" +#include "rmd256-mgf.h" +#include "rmd320-mgf.h" #include "rmd160.h" @@ -132,11 +161,15 @@ static gen generators[]; E(DES3, des3) \ E(RC2, rc2) \ E(RC5, rc5) \ + E(SKIPJACK, skipjack) \ + E(TEA, tea) \ + E(XTEA, xtea) \ E(BLOWFISH, blowfish) \ E(TWOFISH, twofish) \ E(IDEA, idea) \ E(CAST128, cast128) \ E(CAST256, cast256) \ + E(SQUARE, square) \ E(RIJNDAEL, rijndael) \ E(SERPENT, serpent) @@ -144,7 +177,11 @@ static gen generators[]; E(MD4, md4) \ E(MD5, md5) \ E(SHA, sha) \ - E(RMD160, rmd160) + E(TIGER, tiger) \ + E(RMD128, rmd128) \ + E(RMD160, rmd160) \ + E(RMD256, rmd256) \ + E(RMD320, rmd320) #define E(PRE, pre) CIPHER_##PRE, enum { CIPHERS CIPHER__bogus }; @@ -179,8 +216,9 @@ static struct { /*----- Miscellaneous static data -----------------------------------------*/ -static FILE *outfp = stdout; +static FILE *outfp; static size_t outsz = 0; +static unsigned maurer_lo = 5, maurer_hi = 8; static int argc; static char **argv; @@ -191,7 +229,9 @@ enum { f_progress = 1, f_file = 2, f_fips = 4, - f_maurer = 8 + f_maurer = 8, + f_timer = 16, + f_discard = 32 }; /*----- Help options ------------------------------------------------------*/ @@ -226,10 +266,12 @@ common core set:\n\ -l, --list Show a list of the supported generators, with\n\ their options.\n\ -f, --fipstest Run the FIPS 140-1 randomness test.\n\ --m, --maurer Run Maurer's universal statistical test.\n\ +-m, --maurer[=LO-HI] Run Maurer's universal statistical test.\n\ -o, --output FILE Write output to FILE, not stdout.\n\ -z, --size SIZE Emit SIZE bytes, not an unlimited number.\n\ -p, --progress Show a little progress meter (on stderr).\n\ +-T, --timer Keep track of the CPU time used by the generator.\n\ +-d, --discard Discard the generated output.\n\ \n\ (A SIZE may be followed by `g' for gigabytes, `m' for megabytes, or\n\ `k' for kilobytes. If unqualified, an amount in bytes is assumed.)\n\ @@ -250,17 +292,19 @@ static struct option opts[] = { { "list", 0, 0, 'l' }, { "fipstest", 0, 0, 'f' }, - { "maurer", 0, 0, 'm' }, + { "maurer", OPTF_ARGOPT, 0, 'm' }, { "output", OPTF_ARGREQ, 0, 'o' }, { "size", OPTF_ARGREQ, 0, 'z' }, { "progress", 0, 0, 'p' }, + { "timer", 0, 0, 'T' }, + { "discard", 0, 0, 'd' }, /* --- End of main table --- */ { 0, 0, 0, 0 } }; -static const char *sopts = "hvu lfmo:z:p"; +static const char *sopts = "hvu lfm::o:z:pTd"; #ifndef OPTION_V DA_DECL(option_v, struct option); @@ -325,6 +369,19 @@ static int opt(void) break; case 'm': flags |= f_maurer; + if (optarg) { + char *p; + unsigned long lo, hi; + lo = strtoul(optarg, &p, 0); + if (*p == '-') + hi = strtoul(p + 1, &p, 0); + else + hi = lo; + if (*p != 0 || hi < lo || lo == 0) + die(EXIT_FAILURE, "bad bit range `%s'", optarg); + maurer_lo = lo; + maurer_hi = hi; + } break; case 'o': if (flags & f_file) @@ -361,6 +418,12 @@ static int opt(void) case 'p': flags |= f_progress; break; + case 'T': + flags |= f_timer; + break; + case 'd': + flags |= f_discard; + break; default: return (i); } @@ -578,7 +641,7 @@ static grand *gen_bbs(unsigned i) m = MP_COPY(kd->u.m); key_close(&kf); } else { - bbs_param bp; + bbs_priv bp; if (bbs_gen(&bp, bits, &rand_global, 0, (flags & f_progress) ? pgen_ev : 0, 0)) @@ -1038,17 +1101,159 @@ static gen generators[] = { static gen optsg = { "options", gen_opts, 0, "This message shouldn't be printed." }; -/*----- Main code ---------------------------------------------------------*/ +/*----- Random number generation ------------------------------------------*/ -int main(int ac, char *av[]) +static int genfile(const void *buf, size_t sz, void *p) { - gen *g = &optsg; - grand *r; + FILE *fp = p; + if (fwrite(buf, 1, sz, fp) != sz) + die(EXIT_FAILURE, "error writing to file: %s", strerror(errno)); + return (0); +} + +static int genbuf(const void *buf, size_t sz, void *p) +{ + octet **pp = p; + memcpy(*pp, buf, sz); + *pp += sz; + return (0); +} + +static int generate(grand *r, size_t outsz, + int (*func)(const void *buf, size_t sz, void *p), + void *p) +{ + static char kmg[] = { ' ', 'k', 'M', 'G', 'T', 'P', 'E', 'Z', 'Y', 0 }; + unsigned percent = 0; size_t kb = 0; time_t last; static char baton[] = "-\\|/"; char *bp; + int rc; + clock_t clk = 0; + + /* --- Spit out random data --- */ + + last = time(0); + bp = baton; + if (flags & f_progress) { + char *errbuf = xmalloc(BUFSIZ); + setvbuf(stderr, errbuf, _IOLBF, BUFSIZ); + if (outsz) + fprintf(stderr, "[%*s] 0%% 0\r[/\b", 50, ""); + else + fputs("[ ] 0\r[/\b", stderr); + fflush(stderr); + } + +#ifdef SIGPIPE + signal(SIGPIPE, SIG_IGN); +#endif + + do { + octet buf[BUFSIZ]; + size_t sz = sizeof(buf); + clock_t c_start, c_stop; + + /* --- Emit a bufferful (or less) of data --- */ + + if (outsz) { + if (sz > outsz - kb) + sz = outsz - kb; + } + c_start = clock(); + r->ops->fill(r, buf, sz); + c_stop = clock(); + clk += c_stop - c_start; + if (func && (rc = func(buf, sz, p)) != 0) + return (rc); + kb += sz; + + /* --- Update the display --- */ + + if (flags & f_progress) { + time_t t = time(0); + unsigned up = 0; + + if (percent > 100) + up = 1; + + if (!outsz) { + if (difftime(t, last) > 1.0) { + up = 1; + } + if (up) + fputs(" ] ", stderr); + } else { + unsigned pc = kb * 100.0 / outsz; + if (pc > percent || percent > 100 || difftime(t, last) > 1.0) { + if (percent > 100) + percent = 0; + percent &= ~1; + for (; percent < (pc & ~1); percent += 2) + putc('.', stderr); + percent = pc; + for (; pc < 100; pc += 2) + putc(' ', stderr); + fprintf(stderr, "] %3i%% ", percent); + up = 1; + } + } + + if (up) { + size_t q = kb; + char *kk = kmg; + while (q > 8192 && kk[1]) { + q >>= 10; + kk++; + } + fprintf(stderr, "%4i%c\r[", q, *kk); + if (outsz) { + unsigned pc; + for (pc = 0; pc < (percent & ~1); pc += 2) + putc('.', stderr); + } + last = t; + } + + if (percent > 100) + percent = 0; + + if (percent < 100) { + putc(*bp++, stderr); + putc('\b', stderr); + if (!*bp) + bp = baton; + } + fflush(stderr); + } + + /* --- Terminate the loop --- */ + + } while (!outsz || kb < outsz); + + if (flags & f_progress) + fputc('\n', stderr); + if (flags & f_timer) { + double sec = (double)clk/CLOCKS_PER_SEC; + double bps = (outsz << 3)/sec; + char *kk; + + for (kk = kmg; bps > 1024 && kk[1]; kk++, bps /= 1024) + ; + fprintf(stderr, "generated %lu bytes in %g secs (%g %cb/s)\n", + (unsigned long)outsz, sec, bps, *kk); + } + return (0); +} + +/*----- Main code ---------------------------------------------------------*/ + +int main(int ac, char *av[]) +{ + gen *g = &optsg; + grand *r; /* --- Initialize mLib --- */ @@ -1065,6 +1270,7 @@ int main(int ac, char *av[]) addopts(sopts, opts); argc = ac; argv = av; + outfp = stdout; /* --- Read the generator out of the first argument --- */ @@ -1104,8 +1310,9 @@ int main(int ac, char *av[]) if (flags & f_fips) { octet buf[FIPSTEST_BUFSZ]; unsigned rc; + octet *p = buf; - r->ops->fill(r, buf, sizeof(buf)); + generate(r, sizeof(buf), genbuf, &p); rc = fipstest(buf); if (rc & FIPSTEST_MONOBIT) moan("failed monobit test"); @@ -1116,18 +1323,20 @@ int main(int ac, char *av[]) if (rc & FIPSTEST_LONGRUNS) moan("failed long runs test"); if (!rc && (flags & f_progress)) - puts("test passed"); + fputs("test passed\n", stderr); return (rc ? EXIT_FAILURE : 0); } /* --- Do Maurer's test --- */ if (flags & f_maurer) { - octet buf[250 * 1024]; + octet *buf; + size_t bufsz; unsigned i; unsigned rc = 0; unsigned f = 0, jj = 0; double maxz = 0; + octet *p; static struct { double x; const char *sig; } sigtab[] = { { 3.2905, "1e-3" }, @@ -1137,9 +1346,14 @@ int main(int ac, char *av[]) { 0 , 0 } }; - r->ops->fill(r, buf, sizeof(buf)); - for (i = 5; i < 8; i++) { - double z = maurer(buf, sizeof(buf), i + 1); + bufsz = (100 * maurer_hi) << maurer_hi; + if ((buf = a_alloc(arena_global, bufsz)) == 0) + die(EXIT_FAILURE, "not enough memory for data buffer"); + p = buf; + generate(r, bufsz, genbuf, &p); + + for (i = maurer_lo; i <= maurer_hi; i++) { + double z = maurer(buf, bufsz, i); double zz = fabs(z); unsigned j; @@ -1147,135 +1361,41 @@ int main(int ac, char *av[]) if (zz > sigtab[j].x) { if (zz > fabs(maxz)) { maxz = z; - f = i + 1; + f = i; jj = j; } rc = EXIT_FAILURE; moan("failed, bits = %u, sig = %s, Z_u = %g", - i + 1, sigtab[j].sig, z); + i, sigtab[j].sig, z); break; } } if (flags & f_progress) - printf("bits = %u, Z_u = %g\n", i + 1, z); + fprintf(stderr, "bits = %u, Z_u = %g\n", i, z); } return (rc); } - /* --- Make sure we don't write to the terminal --- */ + /* --- Discard --- */ + + if (flags & f_discard) { + generate(r, outsz, 0, 0); + return (0); + } + + /* --- Write to a file --- */ #ifndef PORTABLE if (!(flags & f_file) && isatty(STDOUT_FILENO)) die(EXIT_FAILURE, "writing output to a terminal is a bad idea"); #endif - /* --- Spit out random data --- */ - - last = time(0); - bp = baton; - if (flags & f_progress) { - char *errbuf = xmalloc(BUFSIZ); - setvbuf(stderr, errbuf, _IOLBF, BUFSIZ); - if (outsz) - fprintf(stderr, "[%*s] 0%% 0\r[/\b", 50, ""); - else - fputs("[ ] 0\r[/\b", stderr); - fflush(stderr); - } - -#ifdef SIGPIPE - signal(SIGPIPE, SIG_IGN); -#endif - - for (;;) { - octet buf[BUFSIZ]; - size_t sz = sizeof(buf); - - /* --- Emit a bufferful (or less) of data --- */ - - if (outsz) { - if (sz > outsz - kb) - sz = outsz - kb; - } - r->ops->fill(r, buf, sz); - if (fwrite(buf, 1, sz, outfp) != sz) { - if (flags & f_progress) - fputc('\n', stderr); - die(EXIT_FAILURE, "error writing data: %s", strerror(errno)); - } - kb += sz; - - /* --- Update the display --- */ - - if (flags & f_progress) { - time_t t = time(0); - unsigned up = 0; - - if (percent > 100) - up = 1; - - if (!outsz) { - if (difftime(t, last) > 1.0) { - up = 1; - } - if (up) - fputs(" ] ", stderr); - } else { - unsigned pc = kb * 100.0 / outsz; - if (pc > percent || percent > 100 || difftime(t, last) > 1.0) { - if (percent > 100) - percent = 0; - percent &= ~1; - for (; percent < (pc & ~1); percent += 2) - putc('.', stderr); - percent = pc; - for (; pc < 100; pc += 2) - putc(' ', stderr); - fprintf(stderr, "] %3i%% ", percent); - up = 1; - } - } - - if (up) { - size_t q = kb; - char *suff = " KMG"; - while (q > 8192 && suff[1]) { - q >>= 10; - suff++; - } - fprintf(stderr, "%4i%c\r[", q, *suff); - if (outsz) { - unsigned pc; - for (pc = 0; pc < (percent & ~1); pc += 2) - putc('.', stderr); - } - last = t; - } - - if (percent > 100) - percent = 0; - - if (percent < 100) { - putc(*bp++, stderr); - putc('\b', stderr); - if (!*bp) - bp = baton; - } - fflush(stderr); - } - - /* --- Terminate the loop --- */ - - if (outsz && kb >= outsz) - break; - } + generate(r, outsz, genfile, outfp); /* --- Done --- */ r->ops->destroy(r); - if (flags & f_progress) - fputc('\n', stderr); return (0); }