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