1 /* $OpenBSD: rand.c,v 1.17 2023/03/06 14:32:06 tb 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 static struct { 66 int base64; 67 int hex; 68 char *outfile; 69 } cfg; 70 71 static const struct option rand_options[] = { 72 { 73 .name = "base64", 74 .desc = "Perform base64 encoding on output", 75 .type = OPTION_FLAG, 76 .opt.flag = &cfg.base64, 77 }, 78 { 79 .name = "hex", 80 .desc = "Hexadecimal output", 81 .type = OPTION_FLAG, 82 .opt.flag = &cfg.hex, 83 }, 84 { 85 .name = "out", 86 .argname = "file", 87 .desc = "Write to the given file instead of standard output", 88 .type = OPTION_ARG, 89 .opt.arg = &cfg.outfile, 90 }, 91 {NULL}, 92 }; 93 94 static void 95 rand_usage() 96 { 97 fprintf(stderr, 98 "usage: rand [-base64 | -hex] [-out file] num\n"); 99 options_usage(rand_options); 100 } 101 102 int 103 rand_main(int argc, char **argv) 104 { 105 char *num_bytes = NULL; 106 int ret = 1; 107 int badopt = 0; 108 int num = -1; 109 int i, r; 110 BIO *out = NULL; 111 112 if (pledge("stdio cpath wpath rpath", NULL) == -1) { 113 perror("pledge"); 114 exit(1); 115 } 116 117 memset(&cfg, 0, sizeof(cfg)); 118 119 if (options_parse(argc, argv, rand_options, &num_bytes, NULL) != 0) { 120 rand_usage(); 121 return (1); 122 } 123 124 if (num_bytes != NULL) { 125 r = sscanf(num_bytes, "%d", &num); 126 if (r == 0 || num < 0) 127 badopt = 1; 128 } else 129 badopt = 1; 130 131 if (cfg.hex && cfg.base64) 132 badopt = 1; 133 134 if (badopt) { 135 rand_usage(); 136 goto err; 137 } 138 139 out = BIO_new(BIO_s_file()); 140 if (out == NULL) 141 goto err; 142 if (cfg.outfile != NULL) 143 r = BIO_write_filename(out, cfg.outfile); 144 else 145 r = BIO_set_fp(out, stdout, BIO_NOCLOSE | BIO_FP_TEXT); 146 if (r <= 0) 147 goto err; 148 if (cfg.base64) { 149 BIO *b64 = BIO_new(BIO_f_base64()); 150 if (b64 == NULL) 151 goto err; 152 out = BIO_push(b64, out); 153 } 154 155 while (num > 0) { 156 unsigned char buf[4096]; 157 int chunk; 158 159 chunk = num; 160 if (chunk > (int) sizeof(buf)) 161 chunk = sizeof(buf); 162 arc4random_buf(buf, chunk); 163 if (cfg.hex) { 164 for (i = 0; i < chunk; i++) 165 BIO_printf(out, "%02x", buf[i]); 166 } else 167 BIO_write(out, buf, chunk); 168 num -= chunk; 169 } 170 171 if (cfg.hex) 172 BIO_puts(out, "\n"); 173 (void) BIO_flush(out); 174 175 ret = 0; 176 177 err: 178 ERR_print_errors(bio_err); 179 BIO_free_all(out); 180 181 return (ret); 182 } 183