1 /* 2 * Copyright 1998-2018 The OpenSSL Project Authors. All Rights Reserved. 3 * 4 * Licensed under the OpenSSL license (the "License"). You may not use 5 * this file except in compliance with the License. You can obtain a copy 6 * in the file LICENSE in the source distribution or at 7 * https://www.openssl.org/source/license.html 8 */ 9 10 #include "apps.h" 11 #include "progs.h" 12 13 #include <ctype.h> 14 #include <stdio.h> 15 #include <string.h> 16 17 #include <openssl/bio.h> 18 #include <openssl/err.h> 19 #include <openssl/rand.h> 20 21 typedef enum OPTION_choice { 22 OPT_ERR = -1, OPT_EOF = 0, OPT_HELP, 23 OPT_OUT, OPT_ENGINE, OPT_BASE64, OPT_HEX, 24 OPT_R_ENUM 25 } OPTION_CHOICE; 26 27 const OPTIONS rand_options[] = { 28 {OPT_HELP_STR, 1, '-', "Usage: %s [flags] num\n"}, 29 {OPT_HELP_STR, 1, '-', "Valid options are:\n"}, 30 {"help", OPT_HELP, '-', "Display this summary"}, 31 {"out", OPT_OUT, '>', "Output file"}, 32 OPT_R_OPTIONS, 33 {"base64", OPT_BASE64, '-', "Base64 encode output"}, 34 {"hex", OPT_HEX, '-', "Hex encode output"}, 35 #ifndef OPENSSL_NO_ENGINE 36 {"engine", OPT_ENGINE, 's', "Use engine, possibly a hardware device"}, 37 #endif 38 {NULL} 39 }; 40 41 int rand_main(int argc, char **argv) 42 { 43 ENGINE *e = NULL; 44 BIO *out = NULL; 45 char *outfile = NULL, *prog; 46 OPTION_CHOICE o; 47 int format = FORMAT_BINARY, i, num = -1, r, ret = 1; 48 49 prog = opt_init(argc, argv, rand_options); 50 while ((o = opt_next()) != OPT_EOF) { 51 switch (o) { 52 case OPT_EOF: 53 case OPT_ERR: 54 opthelp: 55 BIO_printf(bio_err, "%s: Use -help for summary.\n", prog); 56 goto end; 57 case OPT_HELP: 58 opt_help(rand_options); 59 ret = 0; 60 goto end; 61 case OPT_OUT: 62 outfile = opt_arg(); 63 break; 64 case OPT_ENGINE: 65 e = setup_engine(opt_arg(), 0); 66 break; 67 case OPT_R_CASES: 68 if (!opt_rand(o)) 69 goto end; 70 break; 71 case OPT_BASE64: 72 format = FORMAT_BASE64; 73 break; 74 case OPT_HEX: 75 format = FORMAT_TEXT; 76 break; 77 } 78 } 79 argc = opt_num_rest(); 80 argv = opt_rest(); 81 if (argc == 1) { 82 if (!opt_int(argv[0], &num) || num <= 0) 83 goto end; 84 } else if (argc > 0) { 85 BIO_printf(bio_err, "Extra arguments given.\n"); 86 goto opthelp; 87 } 88 89 out = bio_open_default(outfile, 'w', format); 90 if (out == NULL) 91 goto end; 92 93 if (format == FORMAT_BASE64) { 94 BIO *b64 = BIO_new(BIO_f_base64()); 95 if (b64 == NULL) 96 goto end; 97 out = BIO_push(b64, out); 98 } 99 100 while (num > 0) { 101 unsigned char buf[4096]; 102 int chunk; 103 104 chunk = num; 105 if (chunk > (int)sizeof(buf)) 106 chunk = sizeof(buf); 107 r = RAND_bytes(buf, chunk); 108 if (r <= 0) 109 goto end; 110 if (format != FORMAT_TEXT) { 111 if (BIO_write(out, buf, chunk) != chunk) 112 goto end; 113 } else { 114 for (i = 0; i < chunk; i++) 115 if (BIO_printf(out, "%02x", buf[i]) != 2) 116 goto end; 117 } 118 num -= chunk; 119 } 120 if (format == FORMAT_TEXT) 121 BIO_puts(out, "\n"); 122 if (BIO_flush(out) <= 0) 123 goto end; 124 125 ret = 0; 126 127 end: 128 if (ret != 0) 129 ERR_print_errors(bio_err); 130 release_engine(e); 131 BIO_free_all(out); 132 return ret; 133 } 134