More code for X509_print_ex() support.
[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  * -base64           - encode output
19  * num               - write 'num' bytes
20  */
21
22 int MAIN(int, char **);
23
24 int MAIN(int argc, char **argv)
25         {
26         int i, r, ret = 1;
27         int badopt;
28         char *outfile = NULL;
29         char *inrand = NULL;
30         int base64 = 0;
31         BIO *out = NULL;
32         int num = -1;
33
34         apps_startup();
35
36         if (bio_err == NULL)
37                 if ((bio_err = BIO_new(BIO_s_file())) != NULL)
38                         BIO_set_fp(bio_err, stderr, BIO_NOCLOSE|BIO_FP_TEXT);
39
40         badopt = 0;
41         i = 0;
42         while (!badopt && argv[++i] != NULL)
43                 {
44                 if (strcmp(argv[i], "-out") == 0)
45                         {
46                         if ((argv[i+1] != NULL) && (outfile == NULL))
47                                 outfile = argv[++i];
48                         else
49                                 badopt = 1;
50                         }
51                 else if (strcmp(argv[i], "-rand") == 0)
52                         {
53                         if ((argv[i+1] != NULL) && (inrand == NULL))
54                                 inrand = argv[++i];
55                         else
56                                 badopt = 1;
57                         }
58                 else if (strcmp(argv[i], "-base64") == 0)
59                         {
60                         if (!base64)
61                                 base64 = 1;
62                         else
63                                 badopt = 1;
64                         }
65                 else if (isdigit((unsigned char)argv[i][0]))
66                         {
67                         if (num < 0)
68                                 {
69                                 r = sscanf(argv[i], "%d", &num);
70                                 if (r == 0 || num < 0)
71                                         badopt = 1;
72                                 }
73                         else
74                                 badopt = 1;
75                         }
76                 else
77                         badopt = 1;
78                 }
79
80         if (num < 0)
81                 badopt = 1;
82         
83         if (badopt) 
84                 {
85                 BIO_printf(bio_err, "Usage: rand [options] num\n");
86                 BIO_printf(bio_err, "where options are\n");
87                 BIO_printf(bio_err, "-out file            - write to file\n");
88                 BIO_printf(bio_err, "-rand file%cfile%c...  - seed PRNG from files\n", LIST_SEPARATOR_CHAR, LIST_SEPARATOR_CHAR);
89                 BIO_printf(bio_err, "-base64              - encode output\n");
90                 goto err;
91                 }
92
93         app_RAND_load_file(NULL, bio_err, (inrand != NULL));
94         if (inrand != NULL)
95                 BIO_printf(bio_err,"%ld semi-random bytes loaded\n",
96                         app_RAND_load_files(inrand));
97
98         out = BIO_new(BIO_s_file());
99         if (out == NULL)
100                 goto err;
101         if (outfile != NULL)
102                 r = BIO_write_filename(out, outfile);
103         else
104                 {
105                 r = BIO_set_fp(out, stdout, BIO_NOCLOSE | BIO_FP_TEXT);
106 #ifdef VMS
107                 {
108                 BIO *tmpbio = BIO_new(BIO_f_linebuffer());
109                 out = BIO_push(tmpbio, out);
110                 }
111 #endif
112                 }
113         if (r <= 0)
114                 goto err;
115
116         if (base64)
117                 {
118                 BIO *b64 = BIO_new(BIO_f_base64());
119                 if (b64 == NULL)
120                         goto err;
121                 out = BIO_push(b64, out);
122                 }
123         
124         while (num > 0) 
125                 {
126                 unsigned char buf[4096];
127                 int chunk;
128
129                 chunk = num;
130                 if (chunk > sizeof buf)
131                         chunk = sizeof buf;
132                 r = RAND_bytes(buf, chunk);
133                 if (r <= 0)
134                         goto err;
135                 BIO_write(out, buf, chunk);
136                 num -= chunk;
137                 }
138         BIO_flush(out);
139
140         app_RAND_write_file(NULL, bio_err);
141         ret = 0;
142         
143 err:
144         ERR_print_errors(bio_err);
145         if (out)
146                 BIO_free_all(out);
147         EXIT(ret);
148         }