Add parameter generation option to genpkey.
[openssl.git] / apps / genpkey.c
1 /* apps/genpkey.c */
2 /* Written by Dr Stephen N Henson (shenson@bigfoot.com) for the OpenSSL
3  * project 2006
4  */
5 /* ====================================================================
6  * Copyright (c) 2006 The OpenSSL Project.  All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  *
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions and the following disclaimer. 
14  *
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in
17  *    the documentation and/or other materials provided with the
18  *    distribution.
19  *
20  * 3. All advertising materials mentioning features or use of this
21  *    software must display the following acknowledgment:
22  *    "This product includes software developed by the OpenSSL Project
23  *    for use in the OpenSSL Toolkit. (http://www.OpenSSL.org/)"
24  *
25  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
26  *    endorse or promote products derived from this software without
27  *    prior written permission. For written permission, please contact
28  *    licensing@OpenSSL.org.
29  *
30  * 5. Products derived from this software may not be called "OpenSSL"
31  *    nor may "OpenSSL" appear in their names without prior written
32  *    permission of the OpenSSL Project.
33  *
34  * 6. Redistributions of any form whatsoever must retain the following
35  *    acknowledgment:
36  *    "This product includes software developed by the OpenSSL Project
37  *    for use in the OpenSSL Toolkit (http://www.OpenSSL.org/)"
38  *
39  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
40  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
41  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
42  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
43  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
44  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
45  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
46  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
47  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
48  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
49  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
50  * OF THE POSSIBILITY OF SUCH DAMAGE.
51  * ====================================================================
52  *
53  * This product includes cryptographic software written by Eric Young
54  * (eay@cryptsoft.com).  This product includes software written by Tim
55  * Hudson (tjh@cryptsoft.com).
56  *
57  */
58 #include <stdio.h>
59 #include <string.h>
60 #include "apps.h"
61 #include <openssl/pem.h>
62 #include <openssl/err.h>
63 #include <openssl/evp.h>
64
65 static int init_keygen_file(BIO *err, EVP_PKEY_CTX **pctx,
66                                 const char *file, ENGINE *e);
67 static int init_gen_str(BIO *err, EVP_PKEY_CTX **pctx,
68                                 const char *algname, ENGINE *e, int do_param);
69 static int genpkey_cb(EVP_PKEY_CTX *ctx);
70
71 #define PROG genpkey_main
72
73 int MAIN(int, char **);
74
75 int MAIN(int argc, char **argv)
76         {
77         ENGINE *e = NULL;
78         char **args, *outfile = NULL;
79         char *passarg = NULL;
80         BIO *in = NULL, *out = NULL;
81         const EVP_CIPHER *cipher = NULL;
82         int outformat;
83         int text = 0;
84         EVP_PKEY *pkey=NULL;
85         EVP_PKEY_CTX *ctx = NULL;
86         char *pass = NULL;
87         int badarg = 0;
88         int ret = 1;
89
90         int do_param = -1;
91
92         if (bio_err == NULL)
93                 bio_err = BIO_new_fp (stderr, BIO_NOCLOSE);
94
95         if (!load_config(bio_err, NULL))
96                 goto end;
97
98         outformat=FORMAT_PEM;
99
100         ERR_load_crypto_strings();
101         OpenSSL_add_all_algorithms();
102         args = argv + 1;
103         while (!badarg && *args && *args[0] == '-')
104                 {
105                 if (!strcmp(*args,"-outform"))
106                         {
107                         if (args[1])
108                                 {
109                                 args++;
110                                 outformat=str2fmt(*args);
111                                 }
112                         else badarg = 1;
113                         }
114                 else if (!strcmp(*args,"-pass"))
115                         {
116                         if (!args[1]) goto bad;
117                         passarg= *(++args);
118                         }
119 #ifndef OPENSSL_NO_ENGINE
120                 else if (strcmp(*args,"-engine") == 0)
121                         {
122                         if (!args[1])
123                                 goto bad;
124                         e = setup_engine(bio_err, *(++args), 0);
125                         }
126 #endif
127                 else if (!strcmp (*args, "-paramfile"))
128                         {
129                         if (!args[1])
130                                 goto bad;
131                         args++;
132                         if (do_param == 1)
133                                 goto bad;
134                         if (!init_keygen_file(bio_err, &ctx, *args, e))
135                                 goto end;
136                         }
137                 else if (!strcmp (*args, "-out"))
138                         {
139                         if (args[1])
140                                 {
141                                 args++;
142                                 outfile = *args;
143                                 }
144                         else badarg = 1;
145                         }
146                 else if (strcmp(*args,"-algorithm") == 0)
147                         {
148                         if (!args[1])
149                                 goto bad;
150                         if (do_param == -1)
151                                 do_param = 0;
152                         if (!init_gen_str(bio_err, &ctx, *(++args),e, do_param))
153                                 goto end;
154                         }
155                 else if (strcmp(*args,"-param") == 0)
156                         {
157                         if (!args[1])
158                                 goto bad;
159                         if (!ctx)
160                                 {
161                                 BIO_puts(bio_err, "No keytype specified\n");
162                                 goto bad;
163                                 }
164                         else if (pkey_ctrl_string(ctx, *(++args)) <= 0)
165                                 {
166                                 BIO_puts(bio_err, "parameter setting error\n");
167                                 ERR_print_errors(bio_err);
168                                 goto end;
169                                 }
170                         }
171                 else if (strcmp(*args,"-genparam") == 0)
172                         {
173                         if (ctx)
174                                 goto bad;
175                         do_param = 1;
176                         }
177                 else if (strcmp(*args,"-text") == 0)
178                         text=1;
179                 else
180                         {
181                         cipher = EVP_get_cipherbyname(*args + 1);
182                         if (!cipher)
183                                 {
184                                 BIO_printf(bio_err, "Unknown cipher %s\n",
185                                                                 *args + 1);
186                                 badarg = 1;
187                                 }
188                         if (do_param == 1)
189                                 badarg = 1;
190                         }
191                 args++;
192                 }
193
194         if (!ctx)
195                 badarg = 1;
196
197         if (badarg)
198                 {
199                 bad:
200                 BIO_printf(bio_err, "Usage genpkey [options]\n");
201                 BIO_printf(bio_err, "where options are\n");
202                 BIO_printf(bio_err, "-paramfile file parameter file\n");
203                 BIO_printf(bio_err, "-pass arg       output file pass phrase source\n");
204                 BIO_printf(bio_err, "-outform X      output format (DER or PEM)\n");
205                 BIO_printf(bio_err, "-out file       output file\n");
206 #ifndef OPENSSL_NO_ENGINE
207                 BIO_printf(bio_err, "-engine e       use engine e, possibly a hardware device.\n");
208 #endif
209                 return 1;
210                 }
211
212         if (!app_passwd(bio_err, passarg, NULL, &pass, NULL))
213                 {
214                 BIO_puts(bio_err, "Error getting password\n");
215                 goto end;
216                 }
217
218         if (outfile)
219                 {
220                 if (!(out = BIO_new_file (outfile, "wb")))
221                         {
222                         BIO_printf(bio_err,
223                                  "Can't open output file %s\n", outfile);
224                         goto end;
225                         }
226                 }
227         else
228                 {
229                 out = BIO_new_fp (stdout, BIO_NOCLOSE);
230 #ifdef OPENSSL_SYS_VMS
231                         {
232                         BIO *tmpbio = BIO_new(BIO_f_linebuffer());
233                         out = BIO_push(tmpbio, out);
234                         }
235 #endif
236                 }
237
238         EVP_PKEY_CTX_set_cb(ctx, genpkey_cb);
239         EVP_PKEY_CTX_set_app_data(ctx, bio_err);
240
241         if (do_param)
242                 {
243                 if (EVP_PKEY_paramgen(ctx, &pkey) <= 0)
244                         {
245                         BIO_puts(bio_err, "Error generating key\n");
246                         goto end;
247                         }
248                 }
249         else
250                 {
251                 if (EVP_PKEY_keygen(ctx, &pkey) <= 0)
252                         {
253                         BIO_puts(bio_err, "Error generating key\n");
254                         goto end;
255                         }
256                 }
257
258         if (outformat == FORMAT_PEM) 
259                 PEM_write_bio_PrivateKey(out, pkey, cipher, NULL, 0,
260                                                                 NULL, pass);
261         else if (outformat == FORMAT_ASN1)
262                 i2d_PrivateKey_bio(out, pkey);
263         else
264                 {
265                 BIO_printf(bio_err, "Bad format specified for key\n");
266                 goto end;
267                 }
268
269
270         if (text)
271                 {
272                 if (do_param)
273                         EVP_PKEY_print_params(out, pkey, 0, NULL);
274                 else
275                         EVP_PKEY_print_private(out, pkey, 0, NULL);
276                 }
277
278         ret = 0;
279
280         end:
281         if (pkey)
282                 EVP_PKEY_free(pkey);
283         if (ctx)
284                 EVP_PKEY_CTX_free(ctx);
285         if (out)
286                 BIO_free_all(out);
287         BIO_free(in);
288         if (pass)
289                 OPENSSL_free(pass);
290
291         return ret;
292         }
293
294 static int init_keygen_file(BIO *err, EVP_PKEY_CTX **pctx,
295                                 const char *file, ENGINE *e)
296         {
297         BIO *pbio;
298         EVP_PKEY *pkey = NULL;
299         EVP_PKEY_CTX *ctx = NULL;
300         if (*pctx)
301                 {
302                 BIO_puts(err, "Parameters already set!\n");
303                 return 0;
304                 }
305
306         pbio = BIO_new_file(file, "r");
307         if (!pbio)
308                 {
309                 BIO_printf(err, "Can't open parameter file %s\n", file);
310                 return 0;
311                 }
312
313         pkey = PEM_read_bio_Parameters(pbio, NULL);
314         BIO_free(pbio);
315
316         if (!pkey)
317                 {
318                 BIO_printf(bio_err, "Error reading parameter file %s\n", file);
319                 return 0;
320                 }
321
322         ctx = EVP_PKEY_CTX_new(pkey, e);
323         if (!ctx)
324                 goto err;
325         if (EVP_PKEY_keygen_init(ctx) <= 0)
326                 goto err;
327         EVP_PKEY_free(pkey);
328         *pctx = ctx;
329         return 1;
330
331         err:
332         BIO_puts(err, "Error initializing context\n");
333         ERR_print_errors(err);
334         if (ctx)
335                 EVP_PKEY_CTX_free(ctx);
336         if (pkey)
337                 EVP_PKEY_free(pkey);
338         return 0;
339
340         }
341
342 static int init_gen_str(BIO *err, EVP_PKEY_CTX **pctx,
343                                 const char *algname, ENGINE *e, int do_param)
344         {
345         EVP_PKEY_CTX *ctx = NULL;
346         const EVP_PKEY_ASN1_METHOD *ameth;
347         int pkey_id;
348         if (*pctx)
349                 {
350                 BIO_puts(err, "Algorithm already set!\n");
351                 return 0;
352                 }
353
354         ameth = EVP_PKEY_asn1_find_str(algname, -1);
355         if (!ameth)
356                 {
357                 BIO_printf(bio_err, "Algorithm %s not found\n", algname);
358                 return 0;
359                 }
360
361         EVP_PKEY_asn1_get0_info(&pkey_id, NULL, NULL, NULL, NULL, ameth);
362         ctx = EVP_PKEY_CTX_new_id(pkey_id, e);
363
364         if (!ctx)
365                 goto err;
366         if (do_param)
367                 {
368                 if (EVP_PKEY_paramgen_init(ctx) <= 0)
369                         goto err;
370                 }
371         else
372                 {
373                 if (EVP_PKEY_keygen_init(ctx) <= 0)
374                         goto err;
375                 }
376
377         *pctx = ctx;
378         return 1;
379
380         err:
381         BIO_printf(err, "Error initializing %s context\n", algname);
382         ERR_print_errors(err);
383         if (ctx)
384                 EVP_PKEY_CTX_free(ctx);
385         return 0;
386
387         }
388
389 static int genpkey_cb(EVP_PKEY_CTX *ctx)
390         {
391         char c='*';
392         BIO *b = EVP_PKEY_CTX_get_app_data(ctx);
393         int p;
394         p = EVP_PKEY_CTX_get_keygen_info(ctx, 0);
395         if (p == 0) c='.';
396         if (p == 1) c='+';
397         if (p == 2) c='*';
398         if (p == 3) c='\n';
399         BIO_write(b,&c,1);
400         (void)BIO_flush(b);
401 #ifdef LINT
402         p=n;
403 #endif
404         return 1;
405         }