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