1 /* $OpenBSD: rand.c,v 1.18 2023/07/23 11:39:29 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
rand_usage(void)95 rand_usage(void)
96 {
97 fprintf(stderr,
98 "usage: rand [-base64 | -hex] [-out file] num\n");
99 options_usage(rand_options);
100 }
101
102 int
rand_main(int argc,char ** argv)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