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