free NULL cleanup -- coda
[openssl.git] / crypto / rc4 / rc4.c
1 /* crypto/rc4/rc4.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 <openssl/rc4.h>
63 #include <openssl/evp.h>
64
65 char *usage[] = {
66     "usage: rc4 args\n",
67     "\n",
68     " -in arg         - input file - default stdin\n",
69     " -out arg        - output file - default stdout\n",
70     " -key key        - password\n",
71     NULL
72 };
73
74 int main(int argc, char *argv[])
75 {
76     FILE *in = NULL, *out = NULL;
77     char *infile = NULL, *outfile = NULL, *keystr = NULL;
78     RC4_KEY key;
79     char buf[BUFSIZ];
80     int badops = 0, i;
81     char **pp;
82     unsigned char md[MD5_DIGEST_LENGTH];
83
84     argc--;
85     argv++;
86     while (argc >= 1) {
87         if (strcmp(*argv, "-in") == 0) {
88             if (--argc < 1)
89                 goto bad;
90             infile = *(++argv);
91         } else if (strcmp(*argv, "-out") == 0) {
92             if (--argc < 1)
93                 goto bad;
94             outfile = *(++argv);
95         } else if (strcmp(*argv, "-key") == 0) {
96             if (--argc < 1)
97                 goto bad;
98             keystr = *(++argv);
99         } else {
100             fprintf(stderr, "unknown option %s\n", *argv);
101             badops = 1;
102             break;
103         }
104         argc--;
105         argv++;
106     }
107
108     if (badops) {
109  bad:
110         for (pp = usage; (*pp != NULL); pp++)
111             fprintf(stderr, "%s", *pp);
112         exit(1);
113     }
114
115     if (infile == NULL)
116         in = stdin;
117     else {
118         in = fopen(infile, "r");
119         if (in == NULL) {
120             perror("open");
121             exit(1);
122         }
123
124     }
125     if (outfile == NULL)
126         out = stdout;
127     else {
128         out = fopen(outfile, "w");
129         if (out == NULL) {
130             perror("open");
131             exit(1);
132         }
133     }
134
135 #ifdef OPENSSL_SYS_MSDOS
136     /* This should set the file to binary mode. */
137     {
138 # include <fcntl.h>
139         setmode(fileno(in), O_BINARY);
140         setmode(fileno(out), O_BINARY);
141     }
142 #endif
143
144     if (keystr == NULL) {       /* get key */
145         i = EVP_read_pw_string(buf, BUFSIZ, "Enter RC4 password:", 0);
146         if (i != 0) {
147             OPENSSL_cleanse(buf, BUFSIZ);
148             fprintf(stderr, "bad password read\n");
149             exit(1);
150         }
151         keystr = buf;
152     }
153
154     EVP_Digest((unsigned char *)keystr, strlen(keystr), md, NULL, EVP_md5(),
155                NULL);
156     OPENSSL_cleanse(keystr, strlen(keystr));
157     RC4_set_key(&key, MD5_DIGEST_LENGTH, md);
158
159     for (;;) {
160         i = fread(buf, 1, BUFSIZ, in);
161         if (i == 0)
162             break;
163         if (i < 0) {
164             perror("read");
165             exit(1);
166         }
167         RC4(&key, (unsigned int)i, (unsigned char *)buf,
168             (unsigned char *)buf);
169         i = fwrite(buf, (unsigned int)i, 1, out);
170         if (i != 1) {
171             perror("write");
172             exit(1);
173         }
174     }
175     fclose(out);
176     fclose(in);
177     exit(0);
178     return (1);
179 }