1 /* $OpenBSD: rand.c,v 1.5 2014/10/22 13:54:03 jsing Exp $ */ 2 /* ==================================================================== 3 * Copyright (c) 1998-2001 The OpenSSL Project. All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in 14 * the documentation and/or other materials provided with the 15 * distribution. 16 * 17 * 3. All advertising materials mentioning features or use of this 18 * software must display the following acknowledgment: 19 * "This product includes software developed by the OpenSSL Project 20 * for use in the OpenSSL Toolkit. (http://www.openssl.org/)" 21 * 22 * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to 23 * endorse or promote products derived from this software without 24 * prior written permission. For written permission, please contact 25 * openssl-core@openssl.org. 26 * 27 * 5. Products derived from this software may not be called "OpenSSL" 28 * nor may "OpenSSL" appear in their names without prior written 29 * permission of the OpenSSL Project. 30 * 31 * 6. Redistributions of any form whatsoever must retain the following 32 * acknowledgment: 33 * "This product includes software developed by the OpenSSL Project 34 * for use in the OpenSSL Toolkit (http://www.openssl.org/)" 35 * 36 * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY 37 * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 38 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 39 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE OpenSSL PROJECT OR 40 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 41 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 42 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 43 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 44 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, 45 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 46 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED 47 * OF THE POSSIBILITY OF SUCH DAMAGE. 48 * ==================================================================== 49 * 50 * This product includes cryptographic software written by Eric Young 51 * (eay@cryptsoft.com). This product includes software written by Tim 52 * Hudson (tjh@cryptsoft.com). 53 * 54 */ 55 56 #include <ctype.h> 57 #include <stdio.h> 58 #include <string.h> 59 60 #include "apps.h" 61 62 #include <openssl/bio.h> 63 #include <openssl/err.h> 64 65 struct { 66 int base64; 67 char *engine; 68 int hex; 69 char *outfile; 70 } rand_config; 71 72 struct option rand_options[] = { 73 { 74 .name = "base64", 75 .desc = "Perform base64 encoding on output", 76 .type = OPTION_FLAG, 77 .opt.flag = &rand_config.base64, 78 }, 79 #ifndef OPENSSL_NO_ENGINE 80 { 81 .name = "engine", 82 .argname = "id", 83 .desc = "Use the engine specified by the given identifier", 84 .type = OPTION_ARG, 85 .opt.arg = &rand_config.engine, 86 }, 87 #endif 88 { 89 .name = "hex", 90 .desc = "Hexadecimal output", 91 .type = OPTION_FLAG, 92 .opt.flag = &rand_config.hex, 93 }, 94 { 95 .name = "out", 96 .argname = "file", 97 .desc = "Write to the given file instead of standard output", 98 .type = OPTION_ARG, 99 .opt.arg = &rand_config.outfile, 100 }, 101 {NULL}, 102 }; 103 104 static void 105 rand_usage() 106 { 107 fprintf(stderr, 108 "usage: rand [-base64 | -hex] [-engine id] [-out file] num\n"); 109 options_usage(rand_options); 110 } 111 112 int rand_main(int, char **); 113 114 int 115 rand_main(int argc, char **argv) 116 { 117 char *num_bytes = NULL; 118 int ret = 1; 119 int badopt = 0; 120 int num = -1; 121 int i, r; 122 BIO *out = NULL; 123 124 memset(&rand_config, 0, sizeof(rand_config)); 125 126 if (options_parse(argc, argv, rand_options, &num_bytes) != 0) { 127 rand_usage(); 128 return (1); 129 } 130 131 if (num_bytes != NULL) { 132 r = sscanf(num_bytes, "%d", &num); 133 if (r == 0 || num < 0) 134 badopt = 1; 135 } else 136 badopt = 1; 137 138 if (rand_config.hex && rand_config.base64) 139 badopt = 1; 140 141 if (badopt) { 142 rand_usage(); 143 goto err; 144 } 145 146 #ifndef OPENSSL_NO_ENGINE 147 setup_engine(bio_err, rand_config.engine, 0); 148 #endif 149 150 out = BIO_new(BIO_s_file()); 151 if (out == NULL) 152 goto err; 153 if (rand_config.outfile != NULL) 154 r = BIO_write_filename(out, rand_config.outfile); 155 else 156 r = BIO_set_fp(out, stdout, BIO_NOCLOSE | BIO_FP_TEXT); 157 if (r <= 0) 158 goto err; 159 if (rand_config.base64) { 160 BIO *b64 = BIO_new(BIO_f_base64()); 161 if (b64 == NULL) 162 goto err; 163 out = BIO_push(b64, out); 164 } 165 166 while (num > 0) { 167 unsigned char buf[4096]; 168 int chunk; 169 170 chunk = num; 171 if (chunk > (int) sizeof(buf)) 172 chunk = sizeof(buf); 173 arc4random_buf(buf, chunk); 174 if (rand_config.hex) { 175 for (i = 0; i < chunk; i++) 176 BIO_printf(out, "%02x", buf[i]); 177 } else 178 BIO_write(out, buf, chunk); 179 num -= chunk; 180 } 181 182 if (rand_config.hex) 183 BIO_puts(out, "\n"); 184 (void) BIO_flush(out); 185 186 ret = 0; 187 188 err: 189 ERR_print_errors(bio_err); 190 if (out) 191 BIO_free_all(out); 192 193 return (ret); 194 } 195