Make it possible for users of the openssl applications to specify the
[openssl.git] / apps / rand.c
1 /* apps/rand.c */
2
3 #include "apps.h"
4
5 #include <ctype.h>
6 #include <stdio.h>
7 #include <string.h>
8
9 #include <openssl/bio.h>
10 #include <openssl/err.h>
11 #include <openssl/rand.h>
12
13 #undef PROG
14 #define PROG rand_main
15
16 /* -out file         - write to file
17  * -rand file:file   - PRNG seed files
18  * -egd file         - PRNG seed from EGD named socket
19  * -base64           - encode output
20  * num               - write 'num' bytes
21  */
22
23 int MAIN(int, char **);
24
25 int MAIN(int argc, char **argv)
26         {
27         int i, r, ret = 1;
28         int badopt;
29         char *outfile = NULL;
30         char *inrand = NULL,*inegd=NULL;
31         int base64 = 0;
32         BIO *out = NULL;
33         int num = -1;
34
35         apps_startup();
36
37         if (bio_err == NULL)
38                 if ((bio_err = BIO_new(BIO_s_file())) != NULL)
39                         BIO_set_fp(bio_err, stderr, BIO_NOCLOSE|BIO_FP_TEXT);
40
41         badopt = 0;
42         i = 0;
43         while (!badopt && argv[++i] != NULL)
44                 {
45                 if (strcmp(argv[i], "-out") == 0)
46                         {
47                         if ((argv[i+1] != NULL) && (outfile == NULL))
48                                 outfile = argv[++i];
49                         else
50                                 badopt = 1;
51                         }
52                 else if (strcmp(argv[i], "-rand") == 0)
53                         {
54                         if ((argv[i+1] != NULL) && (inrand == NULL))
55                                 inrand = argv[++i];
56                         else
57                                 badopt = 1;
58                         }
59                 else if (strcmp(argv[i], "-egd") == 0)
60                         {
61                         if ((argv[i+1] != NULL) && (inegd == NULL))
62                                 inegd = argv[++i];
63                         else
64                                 badopt = 1;
65                         }
66                 else if (strcmp(argv[i], "-base64") == 0)
67                         {
68                         if (!base64)
69                                 base64 = 1;
70                         else
71                                 badopt = 1;
72                         }
73                 else if (isdigit((unsigned char)argv[i][0]))
74                         {
75                         if (num < 0)
76                                 {
77                                 r = sscanf(argv[i], "%d", &num);
78                                 if (r == 0 || num < 0)
79                                         badopt = 1;
80                                 }
81                         else
82                                 badopt = 1;
83                         }
84                 else
85                         badopt = 1;
86                 }
87
88         if (num < 0)
89                 badopt = 1;
90         
91         if (badopt) 
92                 {
93                 BIO_printf(bio_err, "Usage: rand [options] num\n");
94                 BIO_printf(bio_err, "where options are\n");
95                 BIO_printf(bio_err, "-out file            - write to file\n");
96                 BIO_printf(bio_err, "-rand file%cfile%c...  - seed PRNG from files\n", LIST_SEPARATOR_CHAR, LIST_SEPARATOR_CHAR);
97                 BIO_printf(bio_err, "-egd file            - seed PRNG from EGD named socket\n");
98                 BIO_printf(bio_err, "-base64              - encode output\n");
99                 goto err;
100                 }
101
102         app_RAND_load_file(NULL, bio_err, (inrand != NULL || inegd != NULL));
103         if (inrand != NULL)
104                 BIO_printf(bio_err,"%ld semi-random bytes loaded\n",
105                         app_RAND_load_files(inrand));
106         if (inegd != NULL)
107                 BIO_printf(bio_err,"%ld egd bytes loaded\n",
108                         RAND_egd(inegd));
109
110         out = BIO_new(BIO_s_file());
111         if (out == NULL)
112                 goto err;
113         if (outfile != NULL)
114                 r = BIO_write_filename(out, outfile);
115         else
116                 r = BIO_set_fp(out, stdout, BIO_NOCLOSE | BIO_FP_TEXT);
117         if (r <= 0)
118                 goto err;
119
120         if (base64)
121                 {
122                 BIO *b64 = BIO_new(BIO_f_base64());
123                 if (b64 == NULL)
124                         goto err;
125                 out = BIO_push(b64, out);
126                 }
127         
128         while (num > 0) 
129                 {
130                 unsigned char buf[4096];
131                 int chunk;
132
133                 chunk = num;
134                 if (chunk > sizeof buf)
135                         chunk = sizeof buf;
136                 r = RAND_bytes(buf, chunk);
137                 if (r <= 0)
138                         goto err;
139                 BIO_write(out, buf, chunk);
140                 num -= chunk;
141                 }
142         BIO_flush(out);
143
144         app_RAND_write_file(NULL, bio_err);
145         ret = 0;
146         
147 err:
148         ERR_print_errors(bio_err);
149         if (out)
150                 BIO_free_all(out);
151         EXIT(ret);
152         }