Fix comments.
[openssl.git] / demos / maurice / example3.c
1 /* NOCW */
2 /*
3         Please read the README file for condition of use, before
4         using this software.
5
6         Maurice Gittens  <mgittens@gits.nl>   January 1997
7
8 */
9
10 #include <stdio.h>
11 #include <unistd.h>
12 #include <fcntl.h>
13 #include <sys/stat.h>
14 #include <openssl/evp.h>
15
16 #define STDIN           0
17 #define STDOUT          1
18 #define BUFLEN          512 
19 #define INIT_VECTOR     "12345678"
20 #define ENCRYPT         1
21 #define DECRYPT         0
22 #define ALG             EVP_des_ede3_cbc()
23
24 static const char *usage = "Usage: example3 [-d] password\n";
25
26 void do_cipher(char *,int);
27
28 int main(int argc, char *argv[])
29 {
30         if ((argc == 2))        
31         {
32                 do_cipher(argv[1],ENCRYPT);
33         }       
34         else if ((argc == 3) && !strcmp(argv[1],"-d"))
35         {
36                 do_cipher(argv[2],DECRYPT);
37         }
38         else
39         {
40                 fprintf(stderr,"%s", usage);
41                 exit(1);
42         }
43
44         return 0;               
45 }
46
47 void do_cipher(char *pw, int operation)
48 {
49         char buf[BUFLEN];
50         char ebuf[BUFLEN + 8];
51         unsigned int ebuflen; /* rc; */
52         unsigned char iv[EVP_MAX_IV_LENGTH], key[EVP_MAX_KEY_LENGTH];
53         /* unsigned int ekeylen, net_ekeylen;  */
54         EVP_CIPHER_CTX ectx;
55         
56         memcpy(iv, INIT_VECTOR, sizeof(iv));
57
58         EVP_BytesToKey(ALG, EVP_md5(), "salu", pw, strlen(pw), 1, key, iv);
59
60         EVP_CIPHER_CTX_init(&ectx);
61         EVP_CipherInit_ex(&ectx, ALG, NULL, key, iv, operation);
62
63         while(1)
64         {
65                 int readlen = read(STDIN, buf, sizeof(buf));
66         
67                 if (readlen <= 0)
68                 {
69                         if (!readlen)
70                            break;
71                         else
72                         {
73                                 perror("read");
74                                 exit(1);
75                         }
76                 }
77
78                 EVP_CipherUpdate(&ectx, ebuf, &ebuflen, buf, readlen);
79
80                 write(STDOUT, ebuf, ebuflen);
81         }
82
83         EVP_CipherFinal_ex(&ectx, ebuf, &ebuflen); 
84         EVP_CIPHER_CTX_cleanup(&ectx);
85
86         write(STDOUT, ebuf, ebuflen); 
87 }