don't prematurely shut down socket -- use SSL_shutdown
[openssl.git] / demos / b64.c
1 /* demos/b64.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/apps.h"
63 #include <openssl/buffer.h>
64 #include <openssl/err.h>
65 #include <openssl/evp.h>
66 #include <openssl/objects.h>
67 #include <openssl/x509.h>
68 #include <openssl/pem.h>
69
70 #undef SIZE
71 #undef BSIZE
72 #undef PROG
73
74 #define SIZE    (512)
75 #define BSIZE   (8*1024)
76 #define PROG    enc_main
77
78 int main(argc,argv)
79 int argc;
80 char **argv;
81         {
82         char *strbuf=NULL;
83         unsigned char *buff=NULL,*bufsize=NULL;
84         int bsize=BSIZE,verbose=0;
85         int ret=1,inl;
86         unsigned char key[24],iv[MD5_DIGEST_LENGTH];
87         char *str=NULL;
88         char *hkey=NULL,*hiv=NULL;
89         int enc=1,printkey=0,i,base64=0;
90         int debug=0;
91         EVP_CIPHER *cipher=NULL,*c;
92         char *inf=NULL,*outf=NULL;
93         BIO *in=NULL,*out=NULL,*b64=NULL,*benc=NULL,*rbio=NULL,*wbio=NULL;
94 #define PROG_NAME_SIZE  16
95         char pname[PROG_NAME_SIZE];
96
97
98         apps_startup();
99
100         if (bio_err == NULL)
101                 if ((bio_err=BIO_new(BIO_s_file())) != NULL)
102                         BIO_set_fp(bio_err,stderr,BIO_NOCLOSE);
103
104         base64=1;
105
106         argc--;
107         argv++;
108         while (argc >= 1)
109                 {
110                 if      (strcmp(*argv,"-e") == 0)
111                         enc=1;
112                 if (strcmp(*argv,"-in") == 0)
113                         {
114                         if (--argc < 1) goto bad;
115                         inf= *(++argv);
116                         }
117                 else if (strcmp(*argv,"-out") == 0)
118                         {
119                         if (--argc < 1) goto bad;
120                         outf= *(++argv);
121                         }
122                 else if (strcmp(*argv,"-d") == 0)
123                         enc=0;
124                 else if (strcmp(*argv,"-v") == 0)
125                         verbose=1;
126                 else if (strcmp(*argv,"-debug") == 0)
127                         debug=1;
128                 else if (strcmp(*argv,"-bufsize") == 0)
129                         {
130                         if (--argc < 1) goto bad;
131                         bufsize=(unsigned char *)*(++argv);
132                         }
133                 else
134                         {
135                         BIO_printf(bio_err,"unknown option '%s'\n",*argv);
136 bad:
137                         BIO_printf(bio_err,"options are\n");
138                         BIO_printf(bio_err,"%-14s input file\n","-in <file>");
139                         BIO_printf(bio_err,"%-14s output file\n","-out <file>");
140                         BIO_printf(bio_err,"%-14s encode\n","-e");
141                         BIO_printf(bio_err,"%-14s decode\n","-d");
142                         BIO_printf(bio_err,"%-14s buffer size\n","-bufsize <n>");
143
144                         goto end;
145                         }
146                 argc--;
147                 argv++;
148                 }
149
150         if (bufsize != NULL)
151                 {
152                 int i;
153                 unsigned long n;
154
155                 for (n=0; *bufsize; bufsize++)
156                         {
157                         i= *bufsize;
158                         if ((i <= '9') && (i >= '0'))
159                                 n=n*10+i-'0';
160                         else if (i == 'k')
161                                 {
162                                 n*=1024;
163                                 bufsize++;
164                                 break;
165                                 }
166                         }
167                 if (*bufsize != '\0')
168                         {
169                         BIO_printf(bio_err,"invalid 'bufsize' specified.\n");
170                         goto end;
171                         }
172
173                 /* It must be large enough for a base64 encoded line */
174                 if (n < 80) n=80;
175
176                 bsize=(int)n;
177                 if (verbose) BIO_printf(bio_err,"bufsize=%d\n",bsize);
178                 }
179
180         strbuf=Malloc(SIZE);
181         buff=(unsigned char *)Malloc(EVP_ENCODE_LENGTH(bsize));
182         if ((buff == NULL) || (strbuf == NULL))
183                 {
184                 BIO_printf(bio_err,"Malloc failure\n");
185                 goto end;
186                 }
187
188         in=BIO_new(BIO_s_file());
189         out=BIO_new(BIO_s_file());
190         if ((in == NULL) || (out == NULL))
191                 {
192                 ERR_print_errors(bio_err);
193                 goto end;
194                 }
195         if (debug)
196                 {
197                 BIO_set_callback(in,BIO_debug_callback);
198                 BIO_set_callback(out,BIO_debug_callback);
199                 BIO_set_callback_arg(in,bio_err);
200                 BIO_set_callback_arg(out,bio_err);
201                 }
202
203         if (inf == NULL)
204                 BIO_set_fp(in,stdin,BIO_NOCLOSE);
205         else
206                 {
207                 if (BIO_read_filename(in,inf) <= 0)
208                         {
209                         perror(inf);
210                         goto end;
211                         }
212                 }
213
214         if (outf == NULL)
215                 BIO_set_fp(out,stdout,BIO_NOCLOSE);
216         else
217                 {
218                 if (BIO_write_filename(out,outf) <= 0)
219                         {
220                         perror(outf);
221                         goto end;
222                         }
223                 }
224
225         rbio=in;
226         wbio=out;
227
228         if (base64)
229                 {
230                 if ((b64=BIO_new(BIO_f_base64())) == NULL)
231                         goto end;
232                 if (debug)
233                         {
234                         BIO_set_callback(b64,BIO_debug_callback);
235                         BIO_set_callback_arg(b64,bio_err);
236                         }
237                 if (enc)
238                         wbio=BIO_push(b64,wbio);
239                 else
240                         rbio=BIO_push(b64,rbio);
241                 }
242
243         for (;;)
244                 {
245                 inl=BIO_read(rbio,(char *)buff,bsize);
246                 if (inl <= 0) break;
247                 if (BIO_write(wbio,(char *)buff,inl) != inl)
248                         {
249                         BIO_printf(bio_err,"error writing output file\n");
250                         goto end;
251                         }
252                 }
253         BIO_flush(wbio);
254
255         ret=0;
256         if (verbose)
257                 {
258                 BIO_printf(bio_err,"bytes read   :%8ld\n",BIO_number_read(in));
259                 BIO_printf(bio_err,"bytes written:%8ld\n",BIO_number_written(out));
260                 }
261 end:
262         if (strbuf != NULL) Free(strbuf);
263         if (buff != NULL) Free(buff);
264         if (in != NULL) BIO_free(in);
265         if (out != NULL) BIO_free(out);
266         if (benc != NULL) BIO_free(benc);
267         if (b64 != NULL) BIO_free(b64);
268         EXIT(ret);
269         }
270