crypto/openssl/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         }