Free the right thing.
[openssl.git] / apps / dsaparam.c
1 /* apps/dsaparam.c */
2 /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
3  * All rights reserved.
4  *
5  * This package is an SSL implementation written
6  * by Eric Young (eay@cryptsoft.com).
7  * The implementation was written so as to conform with Netscapes SSL.
8  * 
9  * This library is free for commercial and non-commercial use as long as
10  * the following conditions are aheared to.  The following conditions
11  * apply to all code found in this distribution, be it the RC4, RSA,
12  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
13  * included with this distribution is covered by the same copyright terms
14  * except that the holder is Tim Hudson (tjh@cryptsoft.com).
15  * 
16  * Copyright remains Eric Young's, and as such any Copyright notices in
17  * the code are not to be removed.
18  * If this package is used in a product, Eric Young should be given attribution
19  * as the author of the parts of the library used.
20  * This can be in the form of a textual message at program startup or
21  * in documentation (online or textual) provided with the package.
22  * 
23  * Redistribution and use in source and binary forms, with or without
24  * modification, are permitted provided that the following conditions
25  * are met:
26  * 1. Redistributions of source code must retain the copyright
27  *    notice, this list of conditions and the following disclaimer.
28  * 2. Redistributions in binary form must reproduce the above copyright
29  *    notice, this list of conditions and the following disclaimer in the
30  *    documentation and/or other materials provided with the distribution.
31  * 3. All advertising materials mentioning features or use of this software
32  *    must display the following acknowledgement:
33  *    "This product includes cryptographic software written by
34  *     Eric Young (eay@cryptsoft.com)"
35  *    The word 'cryptographic' can be left out if the rouines from the library
36  *    being used are not cryptographic related :-).
37  * 4. If you include any Windows specific code (or a derivative thereof) from 
38  *    the apps directory (application code) you must include an acknowledgement:
39  *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
40  * 
41  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
42  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
43  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
44  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
45  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
46  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
47  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
48  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
49  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
50  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
51  * SUCH DAMAGE.
52  * 
53  * The licence and distribution terms for any publically available version or
54  * derivative of this code cannot be changed.  i.e. this code cannot simply be
55  * copied and put under another distribution licence
56  * [including the GNU Public Licence.]
57  */
58
59 #include <stdio.h>
60 #include <stdlib.h>
61 #include <time.h>
62 #include <string.h>
63 #include "apps.h"
64 #include "bio.h"
65 #include "err.h"
66 #include "bn.h"
67 #include "rand.h"
68 #include "dsa.h"
69 #include "x509.h"
70 #include "pem.h"
71
72 #undef PROG
73 #define PROG    dsaparam_main
74
75 /* -inform arg  - input format - default PEM (one of DER, TXT or PEM)
76  * -outform arg - output format - default PEM
77  * -in arg      - input file - default stdin
78  * -out arg     - output file - default stdout
79  * -noout
80  * -text
81  * -C
82  * -noout
83  * -genkey
84  */
85
86 #ifndef NOPROTO
87 static void MS_CALLBACK dsa_cb(int p, int n, char *arg);
88 #else
89 static void MS_CALLBACK dsa_cb();
90 #endif
91
92 int MAIN(argc, argv)
93 int argc;
94 char **argv;
95         {
96         DSA *dsa=NULL;
97         int i,badops=0,text=0;
98         BIO *in=NULL,*out=NULL;
99         int informat,outformat,noout=0,C=0,ret=1;
100         char *infile,*outfile,*prog,*inrand=NULL;
101         int numbits= -1,num,genkey=0;
102         char buffer[200],*randfile=NULL;
103
104         apps_startup();
105
106         if (bio_err == NULL)
107                 if ((bio_err=BIO_new(BIO_s_file())) != NULL)
108                         BIO_set_fp(bio_err,stderr,BIO_NOCLOSE|BIO_FP_TEXT);
109
110         infile=NULL;
111         outfile=NULL;
112         informat=FORMAT_PEM;
113         outformat=FORMAT_PEM;
114
115         prog=argv[0];
116         argc--;
117         argv++;
118         while (argc >= 1)
119                 {
120                 if      (strcmp(*argv,"-inform") == 0)
121                         {
122                         if (--argc < 1) goto bad;
123                         informat=str2fmt(*(++argv));
124                         }
125                 else if (strcmp(*argv,"-outform") == 0)
126                         {
127                         if (--argc < 1) goto bad;
128                         outformat=str2fmt(*(++argv));
129                         }
130                 else if (strcmp(*argv,"-in") == 0)
131                         {
132                         if (--argc < 1) goto bad;
133                         infile= *(++argv);
134                         }
135                 else if (strcmp(*argv,"-out") == 0)
136                         {
137                         if (--argc < 1) goto bad;
138                         outfile= *(++argv);
139                         }
140                 else if (strcmp(*argv,"-text") == 0)
141                         text=1;
142                 else if (strcmp(*argv,"-C") == 0)
143                         C=1;
144                 else if (strcmp(*argv,"-genkey") == 0)
145                         genkey=1;
146                 else if (strcmp(*argv,"-rand") == 0)
147                         {
148                         if (--argc < 1) goto bad;
149                         inrand= *(++argv);
150                         }
151                 else if (strcmp(*argv,"-noout") == 0)
152                         noout=1;
153                 else if (sscanf(*argv,"%d",&num) == 1)
154                         {
155                         /* generate a key */
156                         numbits=num;
157                         }
158                 else
159                         {
160                         BIO_printf(bio_err,"unknown option %s\n",*argv);
161                         badops=1;
162                         break;
163                         }
164                 argc--;
165                 argv++;
166                 }
167
168         if (badops)
169                 {
170 bad:
171                 BIO_printf(bio_err,"%s [options] [bits] <infile >outfile\n",prog);
172                 BIO_printf(bio_err,"where options are\n");
173                 BIO_printf(bio_err," -inform arg   input format - one of DER TXT PEM\n");
174                 BIO_printf(bio_err," -outform arg  output format - one of DER TXT PEM\n");
175                 BIO_printf(bio_err," -in arg       inout file\n");
176                 BIO_printf(bio_err," -out arg      output file\n");
177                 BIO_printf(bio_err," -text         check the DSA parameters\n");
178                 BIO_printf(bio_err," -C            Output C code\n");
179                 BIO_printf(bio_err," -noout        no output\n");
180                 BIO_printf(bio_err," -rand         files to use for random number input\n");
181                 BIO_printf(bio_err," number        number of bits to use for generating private key\n");
182                 goto end;
183                 }
184
185         ERR_load_crypto_strings();
186
187         in=BIO_new(BIO_s_file());
188         out=BIO_new(BIO_s_file());
189         if ((in == NULL) || (out == NULL))
190                 {
191                 ERR_print_errors(bio_err);
192                 goto end;
193                 }
194
195         if (infile == NULL)
196                 BIO_set_fp(in,stdin,BIO_NOCLOSE);
197         else
198                 {
199                 if (BIO_read_filename(in,infile) <= 0)
200                         {
201                         perror(infile);
202                         goto end;
203                         }
204                 }
205         if (outfile == NULL)
206                 BIO_set_fp(out,stdout,BIO_NOCLOSE);
207         else
208                 {
209                 if (BIO_write_filename(out,outfile) <= 0)
210                         {
211                         perror(outfile);
212                         goto end;
213                         }
214                 }
215
216         if (numbits > 0)
217                 {
218                 randfile=RAND_file_name(buffer,200);
219                 RAND_load_file(randfile,1024L*1024L);
220
221                 BIO_printf(bio_err,"Generating DSA parameters, %d bit long prime\n",num);
222                 BIO_printf(bio_err,"This could take some time\n");
223                 dsa=DSA_generate_parameters(num,NULL,0,NULL,NULL,
224                         dsa_cb,(char *)bio_err);
225                 }
226         else if (informat == FORMAT_ASN1)
227                 dsa=d2i_DSAparams_bio(in,NULL);
228         else if (informat == FORMAT_PEM)
229                 dsa=PEM_read_bio_DSAparams(in,NULL,NULL);
230         else
231                 {
232                 BIO_printf(bio_err,"bad input format specified\n");
233                 goto end;
234                 }
235         if (dsa == NULL)
236                 {
237                 BIO_printf(bio_err,"unable to load DSA parameters\n");
238                 ERR_print_errors(bio_err);
239                 goto end;
240                 }
241
242         if (text)
243                 {
244                 DSAparams_print(out,dsa);
245                 }
246         
247         if (C)
248                 {
249                 unsigned char *data;
250                 int l,len,bits_p,bits_q,bits_g;
251
252                 len=BN_num_bytes(dsa->p);
253                 bits_p=BN_num_bits(dsa->p);
254                 bits_q=BN_num_bits(dsa->q);
255                 bits_g=BN_num_bits(dsa->g);
256                 data=(unsigned char *)Malloc(len+20);
257                 if (data == NULL)
258                         {
259                         perror("Malloc");
260                         goto end;
261                         }
262                 l=BN_bn2bin(dsa->p,data);
263                 printf("static unsigned char dsa%d_p[]={",bits_p);
264                 for (i=0; i<l; i++)
265                         {
266                         if ((i%12) == 0) printf("\n\t");
267                         printf("0x%02X,",data[i]);
268                         }
269                 printf("\n\t};\n");
270
271                 l=BN_bn2bin(dsa->q,data);
272                 printf("static unsigned char dsa%d_q[]={",bits_p);
273                 for (i=0; i<l; i++)
274                         {
275                         if ((i%12) == 0) printf("\n\t");
276                         printf("0x%02X,",data[i]);
277                         }
278                 printf("\n\t};\n");
279
280                 l=BN_bn2bin(dsa->g,data);
281                 printf("static unsigned char dsa%d_g[]={",bits_p);
282                 for (i=0; i<l; i++)
283                         {
284                         if ((i%12) == 0) printf("\n\t");
285                         printf("0x%02X,",data[i]);
286                         }
287                 printf("\n\t};\n\n");
288
289                 printf("DSA *get_dsa%d()\n\t{\n",bits_p);
290                 printf("\tDSA *dsa;\n\n");
291                 printf("\tif ((dsa=DSA_new()) == NULL) return(NULL);\n");
292                 printf("\tdsa->p=BN_bin2bn(dsa%d_p,sizeof(dsa%d_p),NULL);\n",
293                         bits_p,bits_p);
294                 printf("\tdsa->q=BN_bin2bn(dsa%d_q,sizeof(dsa%d_q),NULL);\n",
295                         bits_p,bits_p);
296                 printf("\tdsa->g=BN_bin2bn(dsa%d_g,sizeof(dsa%d_g),NULL);\n",
297                         bits_p,bits_p);
298                 printf("\tif ((dsa->p == NULL) || (dsa->q == NULL) || (dsa->g == NULL))\n");
299                 printf("\t\treturn(NULL);\n");
300                 printf("\treturn(dsa);\n\t}\n");
301                 }
302
303
304         if (!noout)
305                 {
306                 if      (outformat == FORMAT_ASN1)
307                         i=i2d_DSAparams_bio(out,dsa);
308                 else if (outformat == FORMAT_PEM)
309                         i=PEM_write_bio_DSAparams(out,dsa);
310                 else    {
311                         BIO_printf(bio_err,"bad output format specified for outfile\n");
312                         goto end;
313                         }
314                 if (!i)
315                         {
316                         BIO_printf(bio_err,"unable to write DSA paramaters\n");
317                         ERR_print_errors(bio_err);
318                         goto end;
319                         }
320                 }
321         if (genkey)
322                 {
323                 DSA *dsakey;
324
325                 if ((dsakey=DSAparams_dup(dsa)) == NULL) goto end;
326                 if (!DSA_generate_key(dsakey)) goto end;
327                 if      (outformat == FORMAT_ASN1)
328                         i=i2d_DSAPrivateKey_bio(out,dsakey);
329                 else if (outformat == FORMAT_PEM)
330                         i=PEM_write_bio_DSAPrivateKey(out,dsakey,NULL,NULL,0,NULL);
331                 else    {
332                         BIO_printf(bio_err,"bad output format specified for outfile\n");
333                         goto end;
334                         }
335                 DSA_free(dsakey);
336                 }
337         ret=0;
338 end:
339         if (in != NULL) BIO_free(in);
340         if (out != NULL) BIO_free(out);
341         if (dsa != NULL) DSA_free(dsa);
342         EXIT(ret);
343         }
344
345 static void MS_CALLBACK dsa_cb(p, n, arg)
346 int p;
347 int n;
348 char *arg;
349         {
350         char c='*';
351
352         if (p == 0) c='.';
353         if (p == 1) c='+';
354         if (p == 2) c='*';
355         if (p == 3) c='\n';
356         BIO_write((BIO *)arg,&c,1);
357         BIO_flush((BIO *)arg);
358 #ifdef LINT
359         p=n;
360 #endif
361         }