Add ctrls to clear options and mode.
[openssl.git] / apps / rand.c
1 /* apps/rand.c */
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 "apps.h"
57
58 #include <ctype.h>
59 #include <stdio.h>
60 #include <string.h>
61
62 #include <openssl/bio.h>
63 #include <openssl/err.h>
64 #include <openssl/rand.h>
65
66 #undef PROG
67 #define PROG rand_main
68
69 /* -out file         - write to file
70  * -rand file:file   - PRNG seed files
71  * -base64           - base64 encode output
72  * -hex              - hex encode output
73  * num               - write 'num' bytes
74  */
75
76 int MAIN(int, char **);
77
78 int MAIN(int argc, char **argv)
79         {
80 #ifndef OPENSSL_NO_ENGINE
81         ENGINE *e = NULL;
82 #endif
83         int i, r, ret = 1;
84         int badopt;
85         char *outfile = NULL;
86         char *inrand = NULL;
87         int base64 = 0;
88         int hex = 0;
89         BIO *out = NULL;
90         int num = -1;
91 #ifndef OPENSSL_NO_ENGINE
92         char *engine=NULL;
93 #endif
94
95         apps_startup();
96
97         if (bio_err == NULL)
98                 if ((bio_err = BIO_new(BIO_s_file())) != NULL)
99                         BIO_set_fp(bio_err, stderr, BIO_NOCLOSE|BIO_FP_TEXT);
100
101         if (!load_config(bio_err, NULL))
102                 goto err;
103
104         badopt = 0;
105         i = 0;
106         while (!badopt && argv[++i] != NULL)
107                 {
108                 if (strcmp(argv[i], "-out") == 0)
109                         {
110                         if ((argv[i+1] != NULL) && (outfile == NULL))
111                                 outfile = argv[++i];
112                         else
113                                 badopt = 1;
114                         }
115 #ifndef OPENSSL_NO_ENGINE
116                 else if (strcmp(argv[i], "-engine") == 0)
117                         {
118                         if ((argv[i+1] != NULL) && (engine == NULL))
119                                 engine = argv[++i];
120                         else
121                                 badopt = 1;
122                         }
123 #endif
124                 else if (strcmp(argv[i], "-rand") == 0)
125                         {
126                         if ((argv[i+1] != NULL) && (inrand == NULL))
127                                 inrand = argv[++i];
128                         else
129                                 badopt = 1;
130                         }
131                 else if (strcmp(argv[i], "-base64") == 0)
132                         {
133                         if (!base64)
134                                 base64 = 1;
135                         else
136                                 badopt = 1;
137                         }
138                 else if (strcmp(argv[i], "-hex") == 0)
139                         {
140                         if (!hex)
141                                 hex = 1;
142                         else
143                                 badopt = 1;
144                         }
145                 else if (isdigit((unsigned char)argv[i][0]))
146                         {
147                         if (num < 0)
148                                 {
149                                 r = sscanf(argv[i], "%d", &num);
150                                 if (r == 0 || num < 0)
151                                         badopt = 1;
152                                 }
153                         else
154                                 badopt = 1;
155                         }
156                 else
157                         badopt = 1;
158                 }
159
160         if (hex && base64)
161                 badopt = 1;
162
163         if (num < 0)
164                 badopt = 1;
165         
166         if (badopt) 
167                 {
168                 BIO_printf(bio_err, "Usage: rand [options] num\n");
169                 BIO_printf(bio_err, "where options are\n");
170                 BIO_printf(bio_err, "-out file             - write to file\n");
171 #ifndef OPENSSL_NO_ENGINE
172                 BIO_printf(bio_err, "-engine e             - use engine e, possibly a hardware device.\n");
173 #endif
174                 BIO_printf(bio_err, "-rand file%cfile%c... - seed PRNG from files\n", LIST_SEPARATOR_CHAR, LIST_SEPARATOR_CHAR);
175                 BIO_printf(bio_err, "-base64               - base64 encode output\n");
176                 BIO_printf(bio_err, "-hex                  - hex encode output\n");
177                 goto err;
178                 }
179
180 #ifndef OPENSSL_NO_ENGINE
181         e = setup_engine(bio_err, engine, 0);
182 #endif
183
184         app_RAND_load_file(NULL, bio_err, (inrand != NULL));
185         if (inrand != NULL)
186                 BIO_printf(bio_err,"%ld semi-random bytes loaded\n",
187                         app_RAND_load_files(inrand));
188
189         out = BIO_new(BIO_s_file());
190         if (out == NULL)
191                 goto err;
192         if (outfile != NULL)
193                 r = BIO_write_filename(out, outfile);
194         else
195                 {
196                 r = BIO_set_fp(out, stdout, BIO_NOCLOSE | BIO_FP_TEXT);
197 #ifdef OPENSSL_SYS_VMS
198                 {
199                 BIO *tmpbio = BIO_new(BIO_f_linebuffer());
200                 out = BIO_push(tmpbio, out);
201                 }
202 #endif
203                 }
204         if (r <= 0)
205                 goto err;
206
207         if (base64)
208                 {
209                 BIO *b64 = BIO_new(BIO_f_base64());
210                 if (b64 == NULL)
211                         goto err;
212                 out = BIO_push(b64, out);
213                 }
214         
215         while (num > 0) 
216                 {
217                 unsigned char buf[4096];
218                 int chunk;
219
220                 chunk = num;
221                 if (chunk > (int)sizeof(buf))
222                         chunk = sizeof buf;
223                 r = RAND_bytes(buf, chunk);
224                 if (r <= 0)
225                         goto err;
226                 if (!hex) 
227                         BIO_write(out, buf, chunk);
228                 else
229                         {
230                         for (i = 0; i < chunk; i++)
231                                 BIO_printf(out, "%02x", buf[i]);
232                         }
233                 num -= chunk;
234                 }
235         if (hex)
236                 BIO_puts(out, "\n");
237         (void)BIO_flush(out);
238
239         app_RAND_write_file(NULL, bio_err);
240         ret = 0;
241         
242 err:
243         ERR_print_errors(bio_err);
244         if (out)
245                 BIO_free_all(out);
246         apps_shutdown();
247         OPENSSL_EXIT(ret);
248         }