1 /* $NetBSD: test_rand.c,v 1.1.1.2 2014/04/24 12:45:30 pettai Exp $ */ 2 3 /* 4 * Copyright (c) 2007 Kungliga Tekniska Högskolan 5 * (Royal Institute of Technology, Stockholm, Sweden). 6 * All rights reserved. 7 * 8 * Portions Copyright (c) 2009 Apple Inc. All rights reserved. 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 * 14 * 1. Redistributions of source code must retain the above copyright 15 * notice, this list of conditions and the following disclaimer. 16 * 17 * 2. Redistributions in binary form must reproduce the above copyright 18 * notice, this list of conditions and the following disclaimer in the 19 * documentation and/or other materials provided with the distribution. 20 * 21 * 3. Neither the name of the Institute nor the names of its contributors 22 * may be used to endorse or promote products derived from this software 23 * without specific prior written permission. 24 * 25 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND 26 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 27 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 28 * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE 29 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 30 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 31 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 32 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 33 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 34 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 35 * SUCH DAMAGE. 36 */ 37 38 #include <config.h> 39 40 #include <stdio.h> 41 42 #include <krb5/roken.h> 43 #include <krb5/getarg.h> 44 45 #include "rand.h" 46 47 48 /* 49 * 50 */ 51 52 static int version_flag; 53 static int help_flag; 54 static int len = 1024 * 1024; 55 static char *rand_method; 56 static char *filename; 57 58 static struct getargs args[] = { 59 { "length", 0, arg_integer, &len, 60 "length", NULL }, 61 { "file", 0, arg_string, &filename, 62 "file name", NULL }, 63 { "method", 0, arg_string, &rand_method, 64 "method", NULL }, 65 { "version", 0, arg_flag, &version_flag, 66 "print version", NULL }, 67 { "help", 0, arg_flag, &help_flag, 68 NULL, NULL } 69 }; 70 71 /* 72 * 73 */ 74 75 /* 76 * 77 */ 78 79 static void 80 usage (int ret) 81 { 82 arg_printusage (args, 83 sizeof(args)/sizeof(args[0]), 84 NULL, 85 ""); 86 exit (ret); 87 } 88 89 int 90 main(int argc, char **argv) 91 { 92 int idx = 0; 93 char *buffer; 94 char path[MAXPATHLEN]; 95 96 setprogname(argv[0]); 97 98 if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &idx)) 99 usage(1); 100 101 if (help_flag) 102 usage(0); 103 104 if(version_flag){ 105 print_version(NULL); 106 exit(0); 107 } 108 109 argc -= idx; 110 argv += idx; 111 112 if (argc != 0) 113 usage(1); 114 115 buffer = emalloc(len); 116 117 if (rand_method) { 118 if (0) { 119 } 120 #ifndef NO_RAND_FORTUNA_METHOD 121 else if (strcasecmp(rand_method, "fortuna") == 0) 122 RAND_set_rand_method(RAND_fortuna_method()); 123 #endif 124 #ifndef NO_RAND_UNIX_METHOD 125 else if (strcasecmp(rand_method, "unix") == 0) 126 RAND_set_rand_method(RAND_unix_method()); 127 #endif 128 #ifndef NO_RAND_EGD_METHOD 129 else if (strcasecmp(rand_method, "egd") == 0) 130 RAND_set_rand_method(RAND_egd_method()); 131 #endif 132 #ifdef WIN32 133 else if (strcasecmp(rand_method, "w32crypto") == 0) 134 RAND_set_rand_method(RAND_w32crypto_method()); 135 #endif 136 else 137 errx(1, "unknown method %s", rand_method); 138 } 139 140 if (RAND_file_name(path, sizeof(path)) == NULL) 141 errx(1, "RAND_file_name failed"); 142 143 if (RAND_status() != 1) 144 errx(1, "random not ready yet"); 145 146 if (RAND_bytes(buffer, len) != 1) 147 errx(1, "RAND_bytes"); 148 149 if (filename) 150 rk_dumpdata(filename, buffer, len); 151 152 /* head vs tail */ 153 if (len >= 100000) { 154 int bit, i; 155 double res; 156 int bits[8] = { 0, 0, 0, 0, 0, 0, 0, 0 }; 157 158 for (i = 0; i < len; i++) { 159 unsigned char c = ((unsigned char *)buffer)[i]; 160 for (bit = 0; bit < 8 && c; bit++) { 161 if (c & 1) 162 bits[bit]++; 163 c = c >> 1; 164 } 165 } 166 167 for (bit = 0; bit < 8; bit++) { 168 169 res = ((double)abs(len - bits[bit] * 2)) / (double)len; 170 if (res > 0.005) 171 errx(1, "head%d vs tail%d > 0.5%%%% %lf == %d vs %d", 172 bit, bit, res, len, bits[bit]); 173 174 printf("head vs tails bit%d: %lf\n", bit, res); 175 } 176 } 177 178 free(buffer); 179 180 /* test write random file */ 181 { 182 static const char *file = "test.file"; 183 if (RAND_write_file(file) != 1) 184 errx(1, "RAND_write_file"); 185 if (RAND_load_file(file, 1024) != 1) 186 errx(1, "RAND_load_file"); 187 unlink(file); 188 } 189 190 return 0; 191 } 192