Use apps_shutdown() in all applications, in case someone decides not
[openssl.git] / apps / enc.c
1 /* apps/enc.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 <string.h>
62 #include "apps.h"
63 #include <openssl/bio.h>
64 #include <openssl/err.h>
65 #include <openssl/evp.h>
66 #include <openssl/objects.h>
67 #include <openssl/x509.h>
68 #include <openssl/rand.h>
69 #include <openssl/pem.h>
70 #include <openssl/engine.h>
71 #include <ctype.h>
72
73 int set_hex(char *in,unsigned char *out,int size);
74 #undef SIZE
75 #undef BSIZE
76 #undef PROG
77
78 #define SIZE    (512)
79 #define BSIZE   (8*1024)
80 #define PROG    enc_main
81
82 void show_ciphers(const OBJ_NAME *name,void *bio_)
83         {
84         BIO *bio=bio_;
85         static int n;
86
87         if(!islower((unsigned char)*name->name))
88                 return;
89
90         BIO_printf(bio,"-%-25s",name->name);
91         if(++n == 3)
92                 {
93                 BIO_printf(bio,"\n");
94                 n=0;
95                 }
96         else
97                 BIO_printf(bio," ");
98         }
99
100 int MAIN(int, char **);
101
102 int MAIN(int argc, char **argv)
103         {
104         ENGINE *e = NULL;
105         static const char magic[]="Salted__";
106         char mbuf[8];   /* should be 1 smaller than magic */
107         char *strbuf=NULL;
108         unsigned char *buff=NULL,*bufsize=NULL;
109         int bsize=BSIZE,verbose=0;
110         int ret=1,inl;
111         int nopad = 0;
112         unsigned char key[EVP_MAX_KEY_LENGTH],iv[EVP_MAX_IV_LENGTH];
113         unsigned char salt[PKCS5_SALT_LEN];
114         char *str=NULL, *passarg = NULL, *pass = NULL;
115         char *hkey=NULL,*hiv=NULL,*hsalt = NULL;
116         int enc=1,printkey=0,i,base64=0;
117         int debug=0,olb64=0,nosalt=0;
118         const EVP_CIPHER *cipher=NULL,*c;
119         char *inf=NULL,*outf=NULL;
120         BIO *in=NULL,*out=NULL,*b64=NULL,*benc=NULL,*rbio=NULL,*wbio=NULL;
121 #define PROG_NAME_SIZE  16
122         char pname[PROG_NAME_SIZE];
123         char *engine = NULL;
124
125         apps_startup();
126
127         if (bio_err == NULL)
128                 if ((bio_err=BIO_new(BIO_s_file())) != NULL)
129                         BIO_set_fp(bio_err,stderr,BIO_NOCLOSE|BIO_FP_TEXT);
130
131         /* first check the program name */
132         program_name(argv[0],pname,PROG_NAME_SIZE);
133         if (strcmp(pname,"base64") == 0)
134                 base64=1;
135
136         cipher=EVP_get_cipherbyname(pname);
137         if (!base64 && (cipher == NULL) && (strcmp(pname,"enc") != 0))
138                 {
139                 BIO_printf(bio_err,"%s is an unknown cipher\n",pname);
140                 goto bad;
141                 }
142
143         argc--;
144         argv++;
145         while (argc >= 1)
146                 {
147                 if      (strcmp(*argv,"-e") == 0)
148                         enc=1;
149                 else if (strcmp(*argv,"-in") == 0)
150                         {
151                         if (--argc < 1) goto bad;
152                         inf= *(++argv);
153                         }
154                 else if (strcmp(*argv,"-out") == 0)
155                         {
156                         if (--argc < 1) goto bad;
157                         outf= *(++argv);
158                         }
159                 else if (strcmp(*argv,"-pass") == 0)
160                         {
161                         if (--argc < 1) goto bad;
162                         passarg= *(++argv);
163                         }
164                 else if (strcmp(*argv,"-engine") == 0)
165                         {
166                         if (--argc < 1) goto bad;
167                         engine= *(++argv);
168                         }
169                 else if (strcmp(*argv,"-d") == 0)
170                         enc=0;
171                 else if (strcmp(*argv,"-p") == 0)
172                         printkey=1;
173                 else if (strcmp(*argv,"-v") == 0)
174                         verbose=1;
175                 else if (strcmp(*argv,"-nopad") == 0)
176                         nopad=1;
177                 else if (strcmp(*argv,"-salt") == 0)
178                         nosalt=0;
179                 else if (strcmp(*argv,"-nosalt") == 0)
180                         nosalt=1;
181                 else if (strcmp(*argv,"-debug") == 0)
182                         debug=1;
183                 else if (strcmp(*argv,"-P") == 0)
184                         printkey=2;
185                 else if (strcmp(*argv,"-A") == 0)
186                         olb64=1;
187                 else if (strcmp(*argv,"-a") == 0)
188                         base64=1;
189                 else if (strcmp(*argv,"-base64") == 0)
190                         base64=1;
191                 else if (strcmp(*argv,"-bufsize") == 0)
192                         {
193                         if (--argc < 1) goto bad;
194                         bufsize=(unsigned char *)*(++argv);
195                         }
196                 else if (strcmp(*argv,"-k") == 0)
197                         {
198                         if (--argc < 1) goto bad;
199                         str= *(++argv);
200                         }
201                 else if (strcmp(*argv,"-kfile") == 0)
202                         {
203                         static char buf[128];
204                         FILE *infile;
205                         char *file;
206
207                         if (--argc < 1) goto bad;
208                         file= *(++argv);
209                         infile=fopen(file,"r");
210                         if (infile == NULL)
211                                 {
212                                 BIO_printf(bio_err,"unable to read key from '%s'\n",
213                                         file);
214                                 goto bad;
215                                 }
216                         buf[0]='\0';
217                         fgets(buf,128,infile);
218                         fclose(infile);
219                         i=strlen(buf);
220                         if ((i > 0) &&
221                                 ((buf[i-1] == '\n') || (buf[i-1] == '\r')))
222                                 buf[--i]='\0';
223                         if ((i > 0) &&
224                                 ((buf[i-1] == '\n') || (buf[i-1] == '\r')))
225                                 buf[--i]='\0';
226                         if (i < 1)
227                                 {
228                                 BIO_printf(bio_err,"zero length password\n");
229                                 goto bad;
230                                 }
231                         str=buf;
232                         }
233                 else if (strcmp(*argv,"-K") == 0)
234                         {
235                         if (--argc < 1) goto bad;
236                         hkey= *(++argv);
237                         }
238                 else if (strcmp(*argv,"-S") == 0)
239                         {
240                         if (--argc < 1) goto bad;
241                         hsalt= *(++argv);
242                         }
243                 else if (strcmp(*argv,"-iv") == 0)
244                         {
245                         if (--argc < 1) goto bad;
246                         hiv= *(++argv);
247                         }
248                 else if ((argv[0][0] == '-') &&
249                         ((c=EVP_get_cipherbyname(&(argv[0][1]))) != NULL))
250                         {
251                         cipher=c;
252                         }
253                 else if (strcmp(*argv,"-none") == 0)
254                         cipher=NULL;
255                 else
256                         {
257                         BIO_printf(bio_err,"unknown option '%s'\n",*argv);
258 bad:
259                         BIO_printf(bio_err,"options are\n");
260                         BIO_printf(bio_err,"%-14s input file\n","-in <file>");
261                         BIO_printf(bio_err,"%-14s output file\n","-out <file>");
262                         BIO_printf(bio_err,"%-14s pass phrase source\n","-pass <arg>");
263                         BIO_printf(bio_err,"%-14s encrypt\n","-e");
264                         BIO_printf(bio_err,"%-14s decrypt\n","-d");
265                         BIO_printf(bio_err,"%-14s base64 encode/decode, depending on encryption flag\n","-a/-base64");
266                         BIO_printf(bio_err,"%-14s key is the next argument\n","-k");
267                         BIO_printf(bio_err,"%-14s key is the first line of the file argument\n","-kfile");
268                         BIO_printf(bio_err,"%-14s key/iv in hex is the next argument\n","-K/-iv");
269                         BIO_printf(bio_err,"%-14s print the iv/key (then exit if -P)\n","-[pP]");
270                         BIO_printf(bio_err,"%-14s buffer size\n","-bufsize <n>");
271                         BIO_printf(bio_err,"%-14s use engine e, possibly a hardware device.\n","-engine e");
272
273                         BIO_printf(bio_err,"Cipher Types\n");
274                         OBJ_NAME_do_all_sorted(OBJ_NAME_TYPE_CIPHER_METH,
275                                                show_ciphers,
276                                                bio_err);
277                         BIO_printf(bio_err,"\n");
278
279                         goto end;
280                         }
281                 argc--;
282                 argv++;
283                 }
284
285         e = setup_engine(bio_err, engine, 0);
286
287         if (bufsize != NULL)
288                 {
289                 unsigned long n;
290
291                 for (n=0; *bufsize; bufsize++)
292                         {
293                         i= *bufsize;
294                         if ((i <= '9') && (i >= '0'))
295                                 n=n*10+i-'0';
296                         else if (i == 'k')
297                                 {
298                                 n*=1024;
299                                 bufsize++;
300                                 break;
301                                 }
302                         }
303                 if (*bufsize != '\0')
304                         {
305                         BIO_printf(bio_err,"invalid 'bufsize' specified.\n");
306                         goto end;
307                         }
308
309                 /* It must be large enough for a base64 encoded line */
310                 if (n < 80) n=80;
311
312                 bsize=(int)n;
313                 if (verbose) BIO_printf(bio_err,"bufsize=%d\n",bsize);
314                 }
315
316         strbuf=OPENSSL_malloc(SIZE);
317         buff=(unsigned char *)OPENSSL_malloc(EVP_ENCODE_LENGTH(bsize));
318         if ((buff == NULL) || (strbuf == NULL))
319                 {
320                 BIO_printf(bio_err,"OPENSSL_malloc failure %ld\n",(long)EVP_ENCODE_LENGTH(bsize));
321                 goto end;
322                 }
323
324         in=BIO_new(BIO_s_file());
325         out=BIO_new(BIO_s_file());
326         if ((in == NULL) || (out == NULL))
327                 {
328                 ERR_print_errors(bio_err);
329                 goto end;
330                 }
331         if (debug)
332                 {
333                 BIO_set_callback(in,BIO_debug_callback);
334                 BIO_set_callback(out,BIO_debug_callback);
335                 BIO_set_callback_arg(in,bio_err);
336                 BIO_set_callback_arg(out,bio_err);
337                 }
338
339         if (inf == NULL)
340                 BIO_set_fp(in,stdin,BIO_NOCLOSE);
341         else
342                 {
343                 if (BIO_read_filename(in,inf) <= 0)
344                         {
345                         perror(inf);
346                         goto end;
347                         }
348                 }
349
350         if(!str && passarg) {
351                 if(!app_passwd(bio_err, passarg, NULL, &pass, NULL)) {
352                         BIO_printf(bio_err, "Error getting password\n");
353                         goto end;
354                 }
355                 str = pass;
356         }
357
358         if ((str == NULL) && (cipher != NULL) && (hkey == NULL))
359                 {
360                 for (;;)
361                         {
362                         char buf[200];
363
364                         sprintf(buf,"enter %s %s password:",
365                                 OBJ_nid2ln(EVP_CIPHER_nid(cipher)),
366                                 (enc)?"encryption":"decryption");
367                         strbuf[0]='\0';
368                         i=EVP_read_pw_string((char *)strbuf,SIZE,buf,enc);
369                         if (i == 0)
370                                 {
371                                 if (strbuf[0] == '\0')
372                                         {
373                                         ret=1;
374                                         goto end;
375                                         }
376                                 str=strbuf;
377                                 break;
378                                 }
379                         if (i < 0)
380                                 {
381                                 BIO_printf(bio_err,"bad password read\n");
382                                 goto end;
383                                 }
384                         }
385                 }
386
387
388         if (outf == NULL)
389                 {
390                 BIO_set_fp(out,stdout,BIO_NOCLOSE);
391 #ifdef OPENSSL_SYS_VMS
392                 {
393                 BIO *tmpbio = BIO_new(BIO_f_linebuffer());
394                 out = BIO_push(tmpbio, out);
395                 }
396 #endif
397                 }
398         else
399                 {
400                 if (BIO_write_filename(out,outf) <= 0)
401                         {
402                         perror(outf);
403                         goto end;
404                         }
405                 }
406
407         rbio=in;
408         wbio=out;
409
410         if (base64)
411                 {
412                 if ((b64=BIO_new(BIO_f_base64())) == NULL)
413                         goto end;
414                 if (debug)
415                         {
416                         BIO_set_callback(b64,BIO_debug_callback);
417                         BIO_set_callback_arg(b64,bio_err);
418                         }
419                 if (olb64)
420                         BIO_set_flags(b64,BIO_FLAGS_BASE64_NO_NL);
421                 if (enc)
422                         wbio=BIO_push(b64,wbio);
423                 else
424                         rbio=BIO_push(b64,rbio);
425                 }
426
427         if (cipher != NULL)
428                 {
429                 /* Note that str is NULL if a key was passed on the command
430                  * line, so we get no salt in that case. Is this a bug?
431                  */
432                 if (str != NULL)
433                         {
434                         /* Salt handling: if encrypting generate a salt and
435                          * write to output BIO. If decrypting read salt from
436                          * input BIO.
437                          */
438                         unsigned char *sptr;
439                         if(nosalt) sptr = NULL;
440                         else {
441                                 if(enc) {
442                                         if(hsalt) {
443                                                 if(!set_hex(hsalt,salt,PKCS5_SALT_LEN)) {
444                                                         BIO_printf(bio_err,
445                                                                 "invalid hex salt value\n");
446                                                         goto end;
447                                                 }
448                                         } else if (RAND_pseudo_bytes(salt, PKCS5_SALT_LEN) < 0)
449                                                 goto end;
450                                         /* If -P option then don't bother writing */
451                                         if((printkey != 2)
452                                            && (BIO_write(wbio,magic,
453                                                          sizeof magic-1) != sizeof magic-1
454                                                || BIO_write(wbio,
455                                                             (char *)salt,
456                                                             PKCS5_SALT_LEN) != PKCS5_SALT_LEN)) {
457                                                 BIO_printf(bio_err,"error writing output file\n");
458                                                 goto end;
459                                         }
460                                 } else if(BIO_read(rbio,mbuf,sizeof mbuf) != sizeof mbuf
461                                           || BIO_read(rbio,
462                                                       (unsigned char *)salt,
463                                     PKCS5_SALT_LEN) != PKCS5_SALT_LEN) {
464                                         BIO_printf(bio_err,"error reading input file\n");
465                                         goto end;
466                                 } else if(memcmp(mbuf,magic,sizeof magic-1)) {
467                                     BIO_printf(bio_err,"bad magic number\n");
468                                     goto end;
469                                 }
470
471                                 sptr = salt;
472                         }
473
474                         EVP_BytesToKey(cipher,EVP_md5(),sptr,
475                                 (unsigned char *)str,
476                                 strlen(str),1,key,iv);
477                         /* zero the complete buffer or the string
478                          * passed from the command line
479                          * bug picked up by
480                          * Larry J. Hughes Jr. <hughes@indiana.edu> */
481                         if (str == strbuf)
482                                 memset(str,0,SIZE);
483                         else
484                                 memset(str,0,strlen(str));
485                         }
486                 if ((hiv != NULL) && !set_hex(hiv,iv,sizeof iv))
487                         {
488                         BIO_printf(bio_err,"invalid hex iv value\n");
489                         goto end;
490                         }
491                 if ((hkey != NULL) && !set_hex(hkey,key,sizeof key))
492                         {
493                         BIO_printf(bio_err,"invalid hex key value\n");
494                         goto end;
495                         }
496
497                 if ((benc=BIO_new(BIO_f_cipher())) == NULL)
498                         goto end;
499                 BIO_set_cipher(benc,cipher,key,iv,enc);
500                 if (nopad)
501                         {
502                         EVP_CIPHER_CTX *ctx;
503                         BIO_get_cipher_ctx(benc, &ctx);
504                         EVP_CIPHER_CTX_set_padding(ctx, 0);
505                         }
506                 if (debug)
507                         {
508                         BIO_set_callback(benc,BIO_debug_callback);
509                         BIO_set_callback_arg(benc,bio_err);
510                         }
511
512                 if (printkey)
513                         {
514                         if (!nosalt)
515                                 {
516                                 printf("salt=");
517                                 for (i=0; i<PKCS5_SALT_LEN; i++)
518                                         printf("%02X",salt[i]);
519                                 printf("\n");
520                                 }
521                         if (cipher->key_len > 0)
522                                 {
523                                 printf("key=");
524                                 for (i=0; i<cipher->key_len; i++)
525                                         printf("%02X",key[i]);
526                                 printf("\n");
527                                 }
528                         if (cipher->iv_len > 0)
529                                 {
530                                 printf("iv =");
531                                 for (i=0; i<cipher->iv_len; i++)
532                                         printf("%02X",iv[i]);
533                                 printf("\n");
534                                 }
535                         if (printkey == 2)
536                                 {
537                                 ret=0;
538                                 goto end;
539                                 }
540                         }
541                 }
542
543         /* Only encrypt/decrypt as we write the file */
544         if (benc != NULL)
545                 wbio=BIO_push(benc,wbio);
546
547         for (;;)
548                 {
549                 inl=BIO_read(rbio,(char *)buff,bsize);
550                 if (inl <= 0) break;
551                 if (BIO_write(wbio,(char *)buff,inl) != inl)
552                         {
553                         BIO_printf(bio_err,"error writing output file\n");
554                         goto end;
555                         }
556                 }
557         if (!BIO_flush(wbio))
558                 {
559                 BIO_printf(bio_err,"bad decrypt\n");
560                 goto end;
561                 }
562
563         ret=0;
564         if (verbose)
565                 {
566                 BIO_printf(bio_err,"bytes read   :%8ld\n",BIO_number_read(in));
567                 BIO_printf(bio_err,"bytes written:%8ld\n",BIO_number_written(out));
568                 }
569 end:
570         ERR_print_errors(bio_err);
571         if (strbuf != NULL) OPENSSL_free(strbuf);
572         if (buff != NULL) OPENSSL_free(buff);
573         if (in != NULL) BIO_free(in);
574         if (out != NULL) BIO_free_all(out);
575         if (benc != NULL) BIO_free(benc);
576         if (b64 != NULL) BIO_free(b64);
577         if(pass) OPENSSL_free(pass);
578         apps_shutdown();
579         EXIT(ret);
580         }
581
582 int set_hex(char *in, unsigned char *out, int size)
583         {
584         int i,n;
585         unsigned char j;
586
587         n=strlen(in);
588         if (n > (size*2))
589                 {
590                 BIO_printf(bio_err,"hex string is too long\n");
591                 return(0);
592                 }
593         memset(out,0,size);
594         for (i=0; i<n; i++)
595                 {
596                 j=(unsigned char)*in;
597                 *(in++)='\0';
598                 if (j == 0) break;
599                 if ((j >= '0') && (j <= '9'))
600                         j-='0';
601                 else if ((j >= 'A') && (j <= 'F'))
602                         j=j-'A'+10;
603                 else if ((j >= 'a') && (j <= 'f'))
604                         j=j-'a'+10;
605                 else
606                         {
607                         BIO_printf(bio_err,"non-hex digit\n");
608                         return(0);
609                         }
610                 if (i&1)
611                         out[i/2]|=j;
612                 else
613                         out[i/2]=(j<<4);
614                 }
615         return(1);
616         }