bin/openssl/rand.c

   1 /* $OpenBSD: rand.c,v 1.13 2018/02/07 05:47:55 jsing Exp $ */
   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 <ctype.h>
  57 #include <stdio.h>
  58 #include <string.h>
  59
  60 #include "apps.h"
  61
  62 #include <openssl/bio.h>
  63 #include <openssl/err.h>
  64
  65 struct {
  66         int base64;
  67         int hex;
  68         char *outfile;
  69 } rand_config;
  70
  71 struct option rand_options[] = {
  72         {
  73                 .name = "base64",
  74                 .desc = "Perform base64 encoding on output",
  75                 .type = OPTION_FLAG,
  76                 .opt.flag = &rand_config.base64,
  77         },
  78         {
  79                 .name = "hex",
  80                 .desc = "Hexadecimal output",
  81                 .type = OPTION_FLAG,
  82                 .opt.flag = &rand_config.hex,
  83         },
  84         {
  85                 .name = "out",
  86                 .argname = "file",
  87                 .desc = "Write to the given file instead of standard output",
  88                 .type = OPTION_ARG,
  89                 .opt.arg = &rand_config.outfile,
  90         },
  91         {NULL},
  92 };
  93
  94 static void
  95 rand_usage()
  96 {
  97         fprintf(stderr,
  98             "usage: rand [-base64 | -hex] [-out file] num\n");
  99         options_usage(rand_options);
 100 }
 101
 102 int
 103 rand_main(int argc, char **argv)
 104 {
 105         char *num_bytes = NULL;
 106         int ret = 1;
 107         int badopt = 0;
 108         int num = -1;
 109         int i, r;
 110         BIO *out = NULL;
 111
 112         if (single_execution) {
 113                 if (pledge("stdio cpath wpath rpath", NULL) == -1) {
 114                         perror("pledge");
 115                         exit(1);
 116                 }
 117         }
 118
 119         memset(&rand_config, 0, sizeof(rand_config));
 120
 121         if (options_parse(argc, argv, rand_options, &num_bytes, NULL) != 0) {
 122                 rand_usage();
 123                 return (1);
 124         }
 125
 126         if (num_bytes != NULL) {
 127                 r = sscanf(num_bytes, "%d", &num);
 128                 if (r == 0 || num < 0)
 129                         badopt = 1;
 130         } else
 131                 badopt = 1;
 132
 133         if (rand_config.hex && rand_config.base64)
 134                 badopt = 1;
 135
 136         if (badopt) {
 137                 rand_usage();
 138                 goto err;
 139         }
 140
 141         out = BIO_new(BIO_s_file());
 142         if (out == NULL)
 143                 goto err;
 144         if (rand_config.outfile != NULL)
 145                 r = BIO_write_filename(out, rand_config.outfile);
 146         else
 147                 r = BIO_set_fp(out, stdout, BIO_NOCLOSE | BIO_FP_TEXT);
 148         if (r <= 0)
 149                 goto err;
 150         if (rand_config.base64) {
 151                 BIO *b64 = BIO_new(BIO_f_base64());
 152                 if (b64 == NULL)
 153                         goto err;
 154                 out = BIO_push(b64, out);
 155         }
 156
 157         while (num > 0) {
 158                 unsigned char buf[4096];
 159                 int chunk;
 160
 161                 chunk = num;
 162                 if (chunk > (int) sizeof(buf))
 163                         chunk = sizeof(buf);
 164                 arc4random_buf(buf, chunk);
 165                 if (rand_config.hex) {
 166                         for (i = 0; i < chunk; i++)
 167                                 BIO_printf(out, "%02x", buf[i]);
 168                 } else
 169                         BIO_write(out, buf, chunk);
 170                 num -= chunk;
 171         }
 172
 173         if (rand_config.hex)
 174                 BIO_puts(out, "\n");
 175         (void) BIO_flush(out);
 176
 177         ret = 0;
 178
 179  err:
 180         ERR_print_errors(bio_err);
 181         BIO_free_all(out);
 182
 183         return (ret);
 184 }