Check # of arguments for remaining commands.
[openssl.git] / apps / genrsa.c
1 /*
2  * Copyright 1995-2016 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 <openssl/opensslconf.h>
11 #ifdef OPENSSL_NO_RSA
12 NON_EMPTY_TRANSLATION_UNIT
13 #else
14
15 # include <stdio.h>
16 # include <string.h>
17 # include <sys/types.h>
18 # include <sys/stat.h>
19 # include "apps.h"
20 # include <openssl/bio.h>
21 # include <openssl/err.h>
22 # include <openssl/bn.h>
23 # include <openssl/rsa.h>
24 # include <openssl/evp.h>
25 # include <openssl/x509.h>
26 # include <openssl/pem.h>
27 # include <openssl/rand.h>
28
29 # define DEFBITS 2048
30
31 static int genrsa_cb(int p, int n, BN_GENCB *cb);
32
33 typedef enum OPTION_choice {
34     OPT_ERR = -1, OPT_EOF = 0, OPT_HELP,
35     OPT_3, OPT_F4, OPT_ENGINE,
36     OPT_OUT, OPT_PASSOUT, OPT_CIPHER,
37     OPT_R_ENUM
38 } OPTION_CHOICE;
39
40 const OPTIONS genrsa_options[] = {
41     {"help", OPT_HELP, '-', "Display this summary"},
42     {"3", OPT_3, '-', "Use 3 for the E value"},
43     {"F4", OPT_F4, '-', "Use F4 (0x10001) for the E value"},
44     {"f4", OPT_F4, '-', "Use F4 (0x10001) for the E value"},
45     {"out", OPT_OUT, 's', "Output the key to specified file"},
46     OPT_R_OPTIONS,
47     {"passout", OPT_PASSOUT, 's', "Output file pass phrase source"},
48     {"", OPT_CIPHER, '-', "Encrypt the output with any supported cipher"},
49 # ifndef OPENSSL_NO_ENGINE
50     {"engine", OPT_ENGINE, 's', "Use engine, possibly a hardware device"},
51 # endif
52     {NULL}
53 };
54
55 int genrsa_main(int argc, char **argv)
56 {
57     BN_GENCB *cb = BN_GENCB_new();
58     PW_CB_DATA cb_data;
59     ENGINE *eng = NULL;
60     BIGNUM *bn = BN_new();
61     BIO *out = NULL;
62     const BIGNUM *e;
63     RSA *rsa = NULL;
64     const EVP_CIPHER *enc = NULL;
65     int ret = 1, num = DEFBITS, private = 0;
66     unsigned long f4 = RSA_F4;
67     char *outfile = NULL, *passoutarg = NULL, *passout = NULL;
68     char *prog, *hexe, *dece;
69     OPTION_CHOICE o;
70
71     if (bn == NULL || cb == NULL)
72         goto end;
73
74     BN_GENCB_set(cb, genrsa_cb, bio_err);
75
76     prog = opt_init(argc, argv, genrsa_options);
77     while ((o = opt_next()) != OPT_EOF) {
78         switch (o) {
79         case OPT_EOF:
80         case OPT_ERR:
81 opthelp:
82             BIO_printf(bio_err, "%s: Use -help for summary.\n", prog);
83             goto end;
84         case OPT_HELP:
85             ret = 0;
86             opt_help(genrsa_options);
87             goto end;
88         case OPT_3:
89             f4 = 3;
90             break;
91         case OPT_F4:
92             f4 = RSA_F4;
93             break;
94         case OPT_OUT:
95             outfile = opt_arg();
96             break;
97         case OPT_ENGINE:
98             eng = setup_engine(opt_arg(), 0);
99             break;
100         case OPT_R_CASES:
101             if (!opt_rand(o))
102                 goto end;
103             break;
104         case OPT_PASSOUT:
105             passoutarg = opt_arg();
106             break;
107         case OPT_CIPHER:
108             if (!opt_cipher(opt_unknown(), &enc))
109                 goto end;
110             break;
111         }
112     }
113     argc = opt_num_rest();
114     argv = opt_rest();
115
116     if (argc == 1) {
117         if (!opt_int(argv[0], &num) || num <= 0)
118             goto end;
119     } else if (argc > 0) {
120         BIO_printf(bio_err, "Extra arguments given.\n");
121         goto opthelp;
122     }
123
124     private = 1;
125     if (!app_passwd(NULL, passoutarg, NULL, &passout)) {
126         BIO_printf(bio_err, "Error getting password\n");
127         goto end;
128     }
129
130     out = bio_open_owner(outfile, FORMAT_PEM, private);
131     if (out == NULL)
132         goto end;
133
134     BIO_printf(bio_err, "Generating RSA private key, %d bit long modulus\n",
135                num);
136     rsa = eng ? RSA_new_method(eng) : RSA_new();
137     if (rsa == NULL)
138         goto end;
139
140     if (!BN_set_word(bn, f4) || !RSA_generate_key_ex(rsa, num, bn, cb))
141         goto end;
142
143     RSA_get0_key(rsa, NULL, &e, NULL);
144     hexe = BN_bn2hex(e);
145     dece = BN_bn2dec(e);
146     if (hexe && dece) {
147         BIO_printf(bio_err, "e is %s (0x%s)\n", dece, hexe);
148     }
149     OPENSSL_free(hexe);
150     OPENSSL_free(dece);
151     cb_data.password = passout;
152     cb_data.prompt_info = outfile;
153     assert(private);
154     if (!PEM_write_bio_RSAPrivateKey(out, rsa, enc, NULL, 0,
155                                      (pem_password_cb *)password_callback,
156                                      &cb_data))
157         goto end;
158
159     ret = 0;
160  end:
161     BN_free(bn);
162     BN_GENCB_free(cb);
163     RSA_free(rsa);
164     BIO_free_all(out);
165     release_engine(eng);
166     OPENSSL_free(passout);
167     if (ret != 0)
168         ERR_print_errors(bio_err);
169     return (ret);
170 }
171
172 static int genrsa_cb(int p, int n, BN_GENCB *cb)
173 {
174     char c = '*';
175
176     if (p == 0)
177         c = '.';
178     if (p == 1)
179         c = '+';
180     if (p == 2)
181         c = '*';
182     if (p == 3)
183         c = '\n';
184     BIO_write(BN_GENCB_get_arg(cb), &c, 1);
185     (void)BIO_flush(BN_GENCB_get_arg(cb));
186     return 1;
187 }
188 #endif