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 }