Ensure CRYPTO_mem_leaks is the last thing we do
[oweals/openssl.git] / test / rc4test.c
1 /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
2  * All rights reserved.
3  *
4  * This package is an SSL implementation written
5  * by Eric Young (eay@cryptsoft.com).
6  * The implementation was written so as to conform with Netscapes SSL.
7  *
8  * This library is free for commercial and non-commercial use as long as
9  * the following conditions are aheared to.  The following conditions
10  * apply to all code found in this distribution, be it the RC4, RSA,
11  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
12  * included with this distribution is covered by the same copyright terms
13  * except that the holder is Tim Hudson (tjh@cryptsoft.com).
14  *
15  * Copyright remains Eric Young's, and as such any Copyright notices in
16  * the code are not to be removed.
17  * If this package is used in a product, Eric Young should be given attribution
18  * as the author of the parts of the library used.
19  * This can be in the form of a textual message at program startup or
20  * in documentation (online or textual) provided with the package.
21  *
22  * Redistribution and use in source and binary forms, with or without
23  * modification, are permitted provided that the following conditions
24  * are met:
25  * 1. Redistributions of source code must retain the copyright
26  *    notice, this list of conditions and the following disclaimer.
27  * 2. Redistributions in binary form must reproduce the above copyright
28  *    notice, this list of conditions and the following disclaimer in the
29  *    documentation and/or other materials provided with the distribution.
30  * 3. All advertising materials mentioning features or use of this software
31  *    must display the following acknowledgement:
32  *    "This product includes cryptographic software written by
33  *     Eric Young (eay@cryptsoft.com)"
34  *    The word 'cryptographic' can be left out if the rouines from the library
35  *    being used are not cryptographic related :-).
36  * 4. If you include any Windows specific code (or a derivative thereof) from
37  *    the apps directory (application code) you must include an acknowledgement:
38  *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
39  *
40  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
41  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
42  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
43  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
44  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
45  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
46  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
47  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
48  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
49  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
50  * SUCH DAMAGE.
51  *
52  * The licence and distribution terms for any publically available version or
53  * derivative of this code cannot be changed.  i.e. this code cannot simply be
54  * copied and put under another distribution licence
55  * [including the GNU Public Licence.]
56  */
57
58 #include <stdio.h>
59 #include <stdlib.h>
60 #include <string.h>
61
62 #include "../e_os.h"
63
64 #ifdef OPENSSL_NO_RC4
65 int main(int argc, char *argv[])
66 {
67     printf("No RC4 support\n");
68     return (0);
69 }
70 #else
71 # include <openssl/rc4.h>
72 # include <openssl/sha.h>
73
74 static unsigned char keys[7][30] = {
75     {8, 0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef},
76     {8, 0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef},
77     {8, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00},
78     {4, 0xef, 0x01, 0x23, 0x45},
79     {8, 0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef},
80     {4, 0xef, 0x01, 0x23, 0x45},
81 };
82
83 static unsigned char data_len[7] = { 8, 8, 8, 20, 28, 10 };
84
85 static unsigned char data[7][30] = {
86     {0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef, 0xff},
87     {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xff},
88     {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xff},
89     {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
90      0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
91      0x00, 0x00, 0x00, 0x00, 0xff},
92     {0x12, 0x34, 0x56, 0x78, 0x9A, 0xBC, 0xDE, 0xF0,
93      0x12, 0x34, 0x56, 0x78, 0x9A, 0xBC, 0xDE, 0xF0,
94      0x12, 0x34, 0x56, 0x78, 0x9A, 0xBC, 0xDE, 0xF0,
95      0x12, 0x34, 0x56, 0x78, 0xff},
96     {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xff},
97     {0},
98 };
99
100 static unsigned char output[7][30] = {
101     {0x75, 0xb7, 0x87, 0x80, 0x99, 0xe0, 0xc5, 0x96, 0x00},
102     {0x74, 0x94, 0xc2, 0xe7, 0x10, 0x4b, 0x08, 0x79, 0x00},
103     {0xde, 0x18, 0x89, 0x41, 0xa3, 0x37, 0x5d, 0x3a, 0x00},
104     {0xd6, 0xa1, 0x41, 0xa7, 0xec, 0x3c, 0x38, 0xdf,
105      0xbd, 0x61, 0x5a, 0x11, 0x62, 0xe1, 0xc7, 0xba,
106      0x36, 0xb6, 0x78, 0x58, 0x00},
107     {0x66, 0xa0, 0x94, 0x9f, 0x8a, 0xf7, 0xd6, 0x89,
108      0x1f, 0x7f, 0x83, 0x2b, 0xa8, 0x33, 0xc0, 0x0c,
109      0x89, 0x2e, 0xbe, 0x30, 0x14, 0x3c, 0xe2, 0x87,
110      0x40, 0x01, 0x1e, 0xcf, 0x00},
111     {0xd6, 0xa1, 0x41, 0xa7, 0xec, 0x3c, 0x38, 0xdf, 0xbd, 0x61, 0x00},
112     {0},
113 };
114
115 int main(int argc, char *argv[])
116 {
117     int i, err = 0;
118     int j;
119     unsigned char *p;
120     RC4_KEY key;
121     unsigned char obuf[512];
122
123     for (i = 0; i < 6; i++) {
124         RC4_set_key(&key, keys[i][0], &(keys[i][1]));
125         memset(obuf, 0, sizeof(obuf));
126         RC4(&key, data_len[i], &(data[i][0]), obuf);
127         if (memcmp(obuf, output[i], data_len[i] + 1) != 0) {
128             printf("error calculating RC4\n");
129             printf("output:");
130             for (j = 0; j < data_len[i] + 1; j++)
131                 printf(" %02x", obuf[j]);
132             printf("\n");
133             printf("expect:");
134             p = &(output[i][0]);
135             for (j = 0; j < data_len[i] + 1; j++)
136                 printf(" %02x", *(p++));
137             printf("\n");
138             err++;
139         } else
140             printf("test %d ok\n", i);
141     }
142     printf("test end processing ");
143     for (i = 0; i < data_len[3]; i++) {
144         RC4_set_key(&key, keys[3][0], &(keys[3][1]));
145         memset(obuf, 0, sizeof(obuf));
146         RC4(&key, i, &(data[3][0]), obuf);
147         if ((memcmp(obuf, output[3], i) != 0) || (obuf[i] != 0)) {
148             printf("error in RC4 length processing\n");
149             printf("output:");
150             for (j = 0; j < i + 1; j++)
151                 printf(" %02x", obuf[j]);
152             printf("\n");
153             printf("expect:");
154             p = &(output[3][0]);
155             for (j = 0; j < i; j++)
156                 printf(" %02x", *(p++));
157             printf(" 00\n");
158             err++;
159         } else {
160             printf(".");
161             fflush(stdout);
162         }
163     }
164     printf("done\n");
165     printf("test multi-call ");
166     for (i = 0; i < data_len[3]; i++) {
167         RC4_set_key(&key, keys[3][0], &(keys[3][1]));
168         memset(obuf, 0, sizeof(obuf));
169         RC4(&key, i, &(data[3][0]), obuf);
170         RC4(&key, data_len[3] - i, &(data[3][i]), &(obuf[i]));
171         if (memcmp(obuf, output[3], data_len[3] + 1) != 0) {
172             printf("error in RC4 multi-call processing\n");
173             printf("output:");
174             for (j = 0; j < data_len[3] + 1; j++)
175                 printf(" %02x", obuf[j]);
176             printf("\n");
177             printf("expect:");
178             p = &(output[3][0]);
179             for (j = 0; j < data_len[3] + 1; j++)
180                 printf(" %02x", *(p++));
181             err++;
182         } else {
183             printf(".");
184             fflush(stdout);
185         }
186     }
187     printf("done\n");
188     printf("bulk test ");
189     {
190         unsigned char buf[513];
191         SHA_CTX c;
192         unsigned char md[SHA_DIGEST_LENGTH];
193         static unsigned char expected[] = {
194             0xa4, 0x7b, 0xcc, 0x00, 0x3d, 0xd0, 0xbd, 0xe1, 0xac, 0x5f,
195             0x12, 0x1e, 0x45, 0xbc, 0xfb, 0x1a, 0xa1, 0xf2, 0x7f, 0xc5
196         };
197
198         RC4_set_key(&key, keys[0][0], &(keys[3][1]));
199         memset(buf, 0, sizeof(buf));
200         SHA1_Init(&c);
201         for (i = 0; i < 2571; i++) {
202             RC4(&key, sizeof(buf), buf, buf);
203             SHA1_Update(&c, buf, sizeof(buf));
204         }
205         SHA1_Final(md, &c);
206
207         if (memcmp(md, expected, sizeof(md))) {
208             printf("error in RC4 bulk test\n");
209             printf("output:");
210             for (j = 0; j < (int)sizeof(md); j++)
211                 printf(" %02x", md[j]);
212             printf("\n");
213             printf("expect:");
214             for (j = 0; j < (int)sizeof(md); j++)
215                 printf(" %02x", expected[j]);
216             printf("\n");
217             err++;
218         } else
219             printf("ok\n");
220     }
221 # ifdef OPENSSL_SYS_NETWARE
222     if (err)
223         printf("ERROR: %d\n", err);
224 # endif
225     EXIT(err);
226 }
227 #endif