xref: /openbsd-src/usr.bin/openssl/rand.c (revision 440d141494672caa7e8d231fced157bd3b6b7448)
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