1 /* $OpenBSD: rand.c,v 1.7 2015/08/22 16:36:05 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 113 rand_main(int argc, char **argv) 114 { 115 char *num_bytes = NULL; 116 int ret = 1; 117 int badopt = 0; 118 int num = -1; 119 int i, r; 120 BIO *out = NULL; 121 122 memset(&rand_config, 0, sizeof(rand_config)); 123 124 if (options_parse(argc, argv, rand_options, &num_bytes, NULL) != 0) { 125 rand_usage(); 126 return (1); 127 } 128 129 if (num_bytes != NULL) { 130 r = sscanf(num_bytes, "%d", &num); 131 if (r == 0 || num < 0) 132 badopt = 1; 133 } else 134 badopt = 1; 135 136 if (rand_config.hex && rand_config.base64) 137 badopt = 1; 138 139 if (badopt) { 140 rand_usage(); 141 goto err; 142 } 143 144 #ifndef OPENSSL_NO_ENGINE 145 setup_engine(bio_err, rand_config.engine, 0); 146 #endif 147 148 out = BIO_new(BIO_s_file()); 149 if (out == NULL) 150 goto err; 151 if (rand_config.outfile != NULL) 152 r = BIO_write_filename(out, rand_config.outfile); 153 else 154 r = BIO_set_fp(out, stdout, BIO_NOCLOSE | BIO_FP_TEXT); 155 if (r <= 0) 156 goto err; 157 if (rand_config.base64) { 158 BIO *b64 = BIO_new(BIO_f_base64()); 159 if (b64 == NULL) 160 goto err; 161 out = BIO_push(b64, out); 162 } 163 164 while (num > 0) { 165 unsigned char buf[4096]; 166 int chunk; 167 168 chunk = num; 169 if (chunk > (int) sizeof(buf)) 170 chunk = sizeof(buf); 171 arc4random_buf(buf, chunk); 172 if (rand_config.hex) { 173 for (i = 0; i < chunk; i++) 174 BIO_printf(out, "%02x", buf[i]); 175 } else 176 BIO_write(out, buf, chunk); 177 num -= chunk; 178 } 179 180 if (rand_config.hex) 181 BIO_puts(out, "\n"); 182 (void) BIO_flush(out); 183 184 ret = 0; 185 186 err: 187 ERR_print_errors(bio_err); 188 if (out) 189 BIO_free_all(out); 190 191 return (ret); 192 } 193